ojluni/src/main/native/java_util_zip_Deflater.c - platform/libcore - Git at Google

 /*
  * Copyright (c) 1997, 2010, Oracle and/or its affiliates. All rights reserved.
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  *
  * This code is free software; you can redistribute it and/or modify it
  * under the terms of the GNU General Public License version 2 only, as
  * published by the Free Software Foundation.  Oracle designates this
  * particular file as subject to the "Classpath" exception as provided
  * by Oracle in the LICENSE file that accompanied this code.
  *
  * This code is distributed in the hope that it will be useful, but WITHOUT
  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  * version 2 for more details (a copy is included in the LICENSE file that
  * accompanied this code).
  *
  * You should have received a copy of the GNU General Public License version
  * 2 along with this work; if not, write to the Free Software Foundation,
  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  *
  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  * or visit www.oracle.com if you need additional information or have any
  * questions.
  */

 /*
  * Native method support for java.util.zip.Deflater
  */

 #include <stdio.h>
 #include <stdlib.h>
 #include "JNIHelp.h"
 #include "jlong.h"
 #include "jni.h"
 #include "jni_util.h"
 #include <zlib.h>


 #define NATIVE_METHOD(className, functionName, signature) \
 { #functionName, signature, (void*)(className ## _ ## functionName) }

 #define DEF_MEM_LEVEL 8

 static jfieldID levelID;
 static jfieldID strategyID;
 static jfieldID setParamsID;
 static jfieldID finishID;
 static jfieldID finishedID;
 static jfieldID bufID, offID, lenID;

 static void Deflater_initIDs(JNIEnv *env) {
     jclass cls = (*env)->FindClass(env, "java/util/zip/Deflater");
     levelID = (*env)->GetFieldID(env, cls, "level", "I");
     strategyID = (*env)->GetFieldID(env, cls, "strategy", "I");
     setParamsID = (*env)->GetFieldID(env, cls, "setParams", "Z");
     finishID = (*env)->GetFieldID(env, cls, "finish", "Z");
     finishedID = (*env)->GetFieldID(env, cls, "finished", "Z");
     bufID = (*env)->GetFieldID(env, cls, "buf", "[B");
     offID = (*env)->GetFieldID(env, cls, "off", "I");
     lenID = (*env)->GetFieldID(env, cls, "len", "I");
 }

 JNIEXPORT jlong JNICALL
 Deflater_init(JNIEnv *env, jclass cls, jint level,
                                  jint strategy, jboolean nowrap)
 {
     z_stream *strm = calloc(1, sizeof(z_stream));

     if (strm == 0) {
         JNU_ThrowOutOfMemoryError(env, 0);
         return jlong_zero;
     } else {
         char *msg;
         switch (deflateInit2(strm, level, Z_DEFLATED,
                              nowrap ? -MAX_WBITS : MAX_WBITS,
                              DEF_MEM_LEVEL, strategy)) {
           case Z_OK:
             return ptr_to_jlong(strm);
           case Z_MEM_ERROR:
             free(strm);
             JNU_ThrowOutOfMemoryError(env, 0);
             return jlong_zero;
           case Z_STREAM_ERROR:
             free(strm);
             JNU_ThrowIllegalArgumentException(env, 0);
             return jlong_zero;
           default:
             msg = strm->msg;
             free(strm);
             JNU_ThrowInternalError(env, msg);
             return jlong_zero;
         }
     }
 }

 JNIEXPORT void JNICALL
 Deflater_setDictionary(JNIEnv *env, jclass cls, jlong addr,
                                           jarray b, jint off, jint len)
 {
     Bytef *buf = (*env)->GetPrimitiveArrayCritical(env, b, 0);
     int res;
     if (buf == 0) {/* out of memory */
         return;
     }
     res = deflateSetDictionary((z_stream *)jlong_to_ptr(addr), buf + off, len);
     (*env)->ReleasePrimitiveArrayCritical(env, b, buf, 0);
     switch (res) {
     case Z_OK:
         break;
     case Z_STREAM_ERROR:
         JNU_ThrowIllegalArgumentException(env, 0);
         break;
     default:
         JNU_ThrowInternalError(env, ((z_stream *)jlong_to_ptr(addr))->msg);
         break;
     }
 }

 JNIEXPORT jint JNICALL
 Deflater_deflateBytes(JNIEnv *env, jobject this, jlong addr,
                                          jarray b, jint off, jint len, jint flush)
 {
     z_stream *strm = jlong_to_ptr(addr);

     jarray this_buf = (*env)->GetObjectField(env, this, bufID);
     jint this_off = (*env)->GetIntField(env, this, offID);
     jint this_len = (*env)->GetIntField(env, this, lenID);
     jbyte *in_buf;
     jbyte *out_buf;
     int res;
     if ((*env)->GetBooleanField(env, this, setParamsID)) {
         int level = (*env)->GetIntField(env, this, levelID);
         int strategy = (*env)->GetIntField(env, this, strategyID);
         in_buf = (*env)->GetPrimitiveArrayCritical(env, this_buf, 0);
         if (in_buf == NULL) {
             // Throw OOME only when length is not zero
             if (this_len != 0)
                 JNU_ThrowOutOfMemoryError(env, 0);
             return 0;
         }
         out_buf = (*env)->GetPrimitiveArrayCritical(env, b, 0);
         if (out_buf == NULL) {
             (*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);
             if (len != 0)
                 JNU_ThrowOutOfMemoryError(env, 0);
             return 0;
         }

         strm->next_in = (Bytef *) (in_buf + this_off);
         strm->next_out = (Bytef *) (out_buf + off);
         strm->avail_in = this_len;
         strm->avail_out = len;
         res = deflateParams(strm, level, strategy);
         (*env)->ReleasePrimitiveArrayCritical(env, b, out_buf, 0);
         (*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);

         switch (res) {
         case Z_OK:
             (*env)->SetBooleanField(env, this, setParamsID, JNI_FALSE);
             this_off += this_len - strm->avail_in;
             (*env)->SetIntField(env, this, offID, this_off);
             (*env)->SetIntField(env, this, lenID, strm->avail_in);
             return len - strm->avail_out;
         case Z_BUF_ERROR:
             (*env)->SetBooleanField(env, this, setParamsID, JNI_FALSE);
             return 0;
         default:
             JNU_ThrowInternalError(env, strm->msg);
             return 0;
         }
     } else {
         jboolean finish = (*env)->GetBooleanField(env, this, finishID);
         in_buf = (*env)->GetPrimitiveArrayCritical(env, this_buf, 0);
         if (in_buf == NULL) {
             if (this_len != 0)
                 JNU_ThrowOutOfMemoryError(env, 0);
             return 0;
         }
         out_buf = (*env)->GetPrimitiveArrayCritical(env, b, 0);
         if (out_buf == NULL) {
             (*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);
             if (len != 0)
                 JNU_ThrowOutOfMemoryError(env, 0);

             return 0;
         }

         strm->next_in = (Bytef *) (in_buf + this_off);
         strm->next_out = (Bytef *) (out_buf + off);
         strm->avail_in = this_len;
         strm->avail_out = len;
         res = deflate(strm, finish ? Z_FINISH : flush);
         (*env)->ReleasePrimitiveArrayCritical(env, b, out_buf, 0);
         (*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);

         switch (res) {
         case Z_STREAM_END:
             (*env)->SetBooleanField(env, this, finishedID, JNI_TRUE);
             /* fall through */
         case Z_OK:
             this_off += this_len - strm->avail_in;
             (*env)->SetIntField(env, this, offID, this_off);
             (*env)->SetIntField(env, this, lenID, strm->avail_in);
             return len - strm->avail_out;
         case Z_BUF_ERROR:
             return 0;
             default:
             JNU_ThrowInternalError(env, strm->msg);
             return 0;
         }
     }
 }

 JNIEXPORT jint JNICALL
 Deflater_getAdler(JNIEnv *env, jclass cls, jlong addr)
 {
     return ((z_stream *)jlong_to_ptr(addr))->adler;
 }

 JNIEXPORT void JNICALL
 Deflater_reset(JNIEnv *env, jclass cls, jlong addr)
 {
     if (deflateReset((z_stream *)jlong_to_ptr(addr)) != Z_OK) {
         JNU_ThrowInternalError(env, 0);
     }
 }

 JNIEXPORT void JNICALL
 Deflater_end(JNIEnv *env, jclass cls, jlong addr)
 {
     if (deflateEnd((z_stream *)jlong_to_ptr(addr)) == Z_STREAM_ERROR) {
         JNU_ThrowInternalError(env, 0);
     } else {
         free((z_stream *)jlong_to_ptr(addr));
     }
 }

 static JNINativeMethod gMethods[] = {
   NATIVE_METHOD(Deflater, init, "(IIZ)J"),
   NATIVE_METHOD(Deflater, setDictionary, "(J[BII)V"),
   NATIVE_METHOD(Deflater, deflateBytes, "(J[BIII)I"),
   NATIVE_METHOD(Deflater, getAdler, "(J)I"),
   NATIVE_METHOD(Deflater, reset, "(J)V"),
   NATIVE_METHOD(Deflater, end, "(J)V"),
 };

 void register_java_util_zip_Deflater(JNIEnv* env) {
     jniRegisterNativeMethods(env, "java/util/zip/Deflater", gMethods, NELEM(gMethods));

     Deflater_initIDs(env);
 }
	/*
	* Copyright (c) 1997, 2010, Oracle and/or its affiliates. All rights reserved.
	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
	*
	* This code is free software; you can redistribute it and/or modify it
	* under the terms of the GNU General Public License version 2 only, as
	* published by the Free Software Foundation. Oracle designates this
	* particular file as subject to the "Classpath" exception as provided
	* by Oracle in the LICENSE file that accompanied this code.
	*
	* This code is distributed in the hope that it will be useful, but WITHOUT
	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
	* version 2 for more details (a copy is included in the LICENSE file that
	* accompanied this code).
	*
	* You should have received a copy of the GNU General Public License version
	* 2 along with this work; if not, write to the Free Software Foundation,
	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
	*
	* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
	* or visit www.oracle.com if you need additional information or have any
	* questions.
	*/

	/*
	* Native method support for java.util.zip.Deflater
	*/

	#include <stdio.h>
	#include <stdlib.h>
	#include "JNIHelp.h"
	#include "jlong.h"
	#include "jni.h"
	#include "jni_util.h"
	#include <zlib.h>


	#define NATIVE_METHOD(className, functionName, signature) \
	{ #functionName, signature, (void*)(className ## _ ## functionName) }

	#define DEF_MEM_LEVEL 8

	static jfieldID levelID;
	static jfieldID strategyID;
	static jfieldID setParamsID;
	static jfieldID finishID;
	static jfieldID finishedID;
	static jfieldID bufID, offID, lenID;

	static void Deflater_initIDs(JNIEnv *env) {
	jclass cls = (*env)->FindClass(env, "java/util/zip/Deflater");
	levelID = (*env)->GetFieldID(env, cls, "level", "I");
	strategyID = (*env)->GetFieldID(env, cls, "strategy", "I");
	setParamsID = (*env)->GetFieldID(env, cls, "setParams", "Z");
	finishID = (*env)->GetFieldID(env, cls, "finish", "Z");
	finishedID = (*env)->GetFieldID(env, cls, "finished", "Z");
	bufID = (*env)->GetFieldID(env, cls, "buf", "[B");
	offID = (*env)->GetFieldID(env, cls, "off", "I");
	lenID = (*env)->GetFieldID(env, cls, "len", "I");
	}

	JNIEXPORT jlong JNICALL
	Deflater_init(JNIEnv *env, jclass cls, jint level,
	jint strategy, jboolean nowrap)
	{
	z_stream *strm = calloc(1, sizeof(z_stream));

	if (strm == 0) {
	JNU_ThrowOutOfMemoryError(env, 0);
	return jlong_zero;
	} else {
	char *msg;
	switch (deflateInit2(strm, level, Z_DEFLATED,
	nowrap ? -MAX_WBITS : MAX_WBITS,
	DEF_MEM_LEVEL, strategy)) {
	case Z_OK:
	return ptr_to_jlong(strm);
	case Z_MEM_ERROR:
	free(strm);
	JNU_ThrowOutOfMemoryError(env, 0);
	return jlong_zero;
	case Z_STREAM_ERROR:
	free(strm);
	JNU_ThrowIllegalArgumentException(env, 0);
	return jlong_zero;
	default:
	msg = strm->msg;
	free(strm);
	JNU_ThrowInternalError(env, msg);
	return jlong_zero;
	}
	}
	}

	JNIEXPORT void JNICALL
	Deflater_setDictionary(JNIEnv *env, jclass cls, jlong addr,
	jarray b, jint off, jint len)
	{
	Bytef buf = (env)->GetPrimitiveArrayCritical(env, b, 0);
	int res;
	if (buf == 0) {/* out of memory */
	return;
	}
	res = deflateSetDictionary((z_stream *)jlong_to_ptr(addr), buf + off, len);
	(*env)->ReleasePrimitiveArrayCritical(env, b, buf, 0);
	switch (res) {
	case Z_OK:
	break;
	case Z_STREAM_ERROR:
	JNU_ThrowIllegalArgumentException(env, 0);
	break;
	default:
	JNU_ThrowInternalError(env, ((z_stream *)jlong_to_ptr(addr))->msg);
	break;
	}
	}

	JNIEXPORT jint JNICALL
	Deflater_deflateBytes(JNIEnv *env, jobject this, jlong addr,
	jarray b, jint off, jint len, jint flush)
	{
	z_stream *strm = jlong_to_ptr(addr);

	jarray this_buf = (*env)->GetObjectField(env, this, bufID);
	jint this_off = (*env)->GetIntField(env, this, offID);
	jint this_len = (*env)->GetIntField(env, this, lenID);
	jbyte *in_buf;
	jbyte *out_buf;
	int res;
	if ((*env)->GetBooleanField(env, this, setParamsID)) {
	int level = (*env)->GetIntField(env, this, levelID);
	int strategy = (*env)->GetIntField(env, this, strategyID);
	in_buf = (*env)->GetPrimitiveArrayCritical(env, this_buf, 0);
	if (in_buf == NULL) {
	// Throw OOME only when length is not zero
	if (this_len != 0)
	JNU_ThrowOutOfMemoryError(env, 0);
	return 0;
	}
	out_buf = (*env)->GetPrimitiveArrayCritical(env, b, 0);
	if (out_buf == NULL) {
	(*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);
	if (len != 0)
	JNU_ThrowOutOfMemoryError(env, 0);
	return 0;
	}

	strm->next_in = (Bytef *) (in_buf + this_off);
	strm->next_out = (Bytef *) (out_buf + off);
	strm->avail_in = this_len;
	strm->avail_out = len;
	res = deflateParams(strm, level, strategy);
	(*env)->ReleasePrimitiveArrayCritical(env, b, out_buf, 0);
	(*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);

	switch (res) {
	case Z_OK:
	(*env)->SetBooleanField(env, this, setParamsID, JNI_FALSE);
	this_off += this_len - strm->avail_in;
	(*env)->SetIntField(env, this, offID, this_off);
	(*env)->SetIntField(env, this, lenID, strm->avail_in);
	return len - strm->avail_out;
	case Z_BUF_ERROR:
	(*env)->SetBooleanField(env, this, setParamsID, JNI_FALSE);
	return 0;
	default:
	JNU_ThrowInternalError(env, strm->msg);
	return 0;
	}
	} else {
	jboolean finish = (*env)->GetBooleanField(env, this, finishID);
	in_buf = (*env)->GetPrimitiveArrayCritical(env, this_buf, 0);
	if (in_buf == NULL) {
	if (this_len != 0)
	JNU_ThrowOutOfMemoryError(env, 0);
	return 0;
	}
	out_buf = (*env)->GetPrimitiveArrayCritical(env, b, 0);
	if (out_buf == NULL) {
	(*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);
	if (len != 0)
	JNU_ThrowOutOfMemoryError(env, 0);

	return 0;
	}

	strm->next_in = (Bytef *) (in_buf + this_off);
	strm->next_out = (Bytef *) (out_buf + off);
	strm->avail_in = this_len;
	strm->avail_out = len;
	res = deflate(strm, finish ? Z_FINISH : flush);
	(*env)->ReleasePrimitiveArrayCritical(env, b, out_buf, 0);
	(*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);

	switch (res) {
	case Z_STREAM_END:
	(*env)->SetBooleanField(env, this, finishedID, JNI_TRUE);
	/* fall through */
	case Z_OK:
	this_off += this_len - strm->avail_in;
	(*env)->SetIntField(env, this, offID, this_off);
	(*env)->SetIntField(env, this, lenID, strm->avail_in);
	return len - strm->avail_out;
	case Z_BUF_ERROR:
	return 0;
	default:
	JNU_ThrowInternalError(env, strm->msg);
	return 0;
	}
	}
	}

	JNIEXPORT jint JNICALL
	Deflater_getAdler(JNIEnv *env, jclass cls, jlong addr)
	{
	return ((z_stream *)jlong_to_ptr(addr))->adler;
	}

	JNIEXPORT void JNICALL
	Deflater_reset(JNIEnv *env, jclass cls, jlong addr)
	{
	if (deflateReset((z_stream *)jlong_to_ptr(addr)) != Z_OK) {
	JNU_ThrowInternalError(env, 0);
	}
	}

	JNIEXPORT void JNICALL
	Deflater_end(JNIEnv *env, jclass cls, jlong addr)
	{
	if (deflateEnd((z_stream *)jlong_to_ptr(addr)) == Z_STREAM_ERROR) {
	JNU_ThrowInternalError(env, 0);
	} else {
	free((z_stream *)jlong_to_ptr(addr));
	}
	}

	static JNINativeMethod gMethods[] = {
	NATIVE_METHOD(Deflater, init, "(IIZ)J"),
	NATIVE_METHOD(Deflater, setDictionary, "(J[BII)V"),
	NATIVE_METHOD(Deflater, deflateBytes, "(J[BIII)I"),
	NATIVE_METHOD(Deflater, getAdler, "(J)I"),
	NATIVE_METHOD(Deflater, reset, "(J)V"),
	NATIVE_METHOD(Deflater, end, "(J)V"),
	};

	void register_java_util_zip_Deflater(JNIEnv* env) {
	jniRegisterNativeMethods(env, "java/util/zip/Deflater", gMethods, NELEM(gMethods));

	Deflater_initIDs(env);
	}