1 /*
   2  * Copyright (c) 1997, 2014, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 /*
  27  * Native method support for java.util.zip.Deflater
  28  */
  29 
  30 #include <stdio.h>
  31 #include <stdlib.h>
  32 #include "jlong.h"
  33 #include "jni.h"
  34 #include "jni_util.h"
  35 #include <zlib.h>
  36 
  37 #include "java_util_zip_Deflater.h"
  38 
  39 #define DEF_MEM_LEVEL 8
  40 
  41 static jfieldID levelID;
  42 static jfieldID strategyID;
  43 static jfieldID setParamsID;
  44 static jfieldID finishID;
  45 static jfieldID finishedID;
  46 static jfieldID bufID, offID, lenID;
  47 
  48 JNIEXPORT void JNICALL
  49 Java_java_util_zip_Deflater_initIDs(JNIEnv *env, jclass cls)
  50 {
  51     levelID = (*env)->GetFieldID(env, cls, "level", "I");
  52     CHECK_NULL(levelID);
  53     strategyID = (*env)->GetFieldID(env, cls, "strategy", "I");
  54     CHECK_NULL(strategyID);
  55     setParamsID = (*env)->GetFieldID(env, cls, "setParams", "Z");
  56     CHECK_NULL(setParamsID);
  57     finishID = (*env)->GetFieldID(env, cls, "finish", "Z");
  58     CHECK_NULL(finishID);
  59     finishedID = (*env)->GetFieldID(env, cls, "finished", "Z");
  60     CHECK_NULL(finishedID);
  61     bufID = (*env)->GetFieldID(env, cls, "buf", "[B");
  62     CHECK_NULL(bufID);
  63     offID = (*env)->GetFieldID(env, cls, "off", "I");
  64     CHECK_NULL(offID);
  65     lenID = (*env)->GetFieldID(env, cls, "len", "I");
  66     CHECK_NULL(lenID);
  67 }
  68 
  69 JNIEXPORT jlong JNICALL
  70 Java_java_util_zip_Deflater_init(JNIEnv *env, jclass cls, jint level,
  71                                  jint strategy, jboolean nowrap)
  72 {
  73     z_stream *strm = calloc(1, sizeof(z_stream));
  74 
  75     if (strm == 0) {
  76         JNU_ThrowOutOfMemoryError(env, 0);
  77         return jlong_zero;
  78     } else {
  79         char *msg;
  80         switch (deflateInit2(strm, level, Z_DEFLATED,
  81                              nowrap ? -MAX_WBITS : MAX_WBITS,
  82                              DEF_MEM_LEVEL, strategy)) {
  83           case Z_OK:
  84             return ptr_to_jlong(strm);
  85           case Z_MEM_ERROR:
  86             free(strm);
  87             JNU_ThrowOutOfMemoryError(env, 0);
  88             return jlong_zero;
  89           case Z_STREAM_ERROR:
  90             free(strm);
  91             JNU_ThrowIllegalArgumentException(env, 0);
  92             return jlong_zero;
  93           default:
  94             msg = strm->msg;
  95             free(strm);
  96             JNU_ThrowInternalError(env, msg);
  97             return jlong_zero;
  98         }
  99     }
 100 }
 101 
 102 JNIEXPORT void JNICALL
 103 Java_java_util_zip_Deflater_setDictionary(JNIEnv *env, jclass cls, jlong addr,
 104                                           jarray b, jint off, jint len)
 105 {
 106     Bytef *buf = (*env)->GetPrimitiveArrayCritical(env, b, 0);
 107     int res;
 108     if (buf == 0) {/* out of memory */
 109         return;
 110     }
 111     res = deflateSetDictionary((z_stream *)jlong_to_ptr(addr), buf + off, len);
 112     (*env)->ReleasePrimitiveArrayCritical(env, b, buf, 0);
 113     switch (res) {
 114     case Z_OK:
 115         break;
 116     case Z_STREAM_ERROR:
 117         JNU_ThrowIllegalArgumentException(env, 0);
 118         break;
 119     default:
 120         JNU_ThrowInternalError(env, ((z_stream *)jlong_to_ptr(addr))->msg);
 121         break;
 122     }
 123 }
 124 
 125 JNIEXPORT jint JNICALL
 126 Java_java_util_zip_Deflater_deflateBytes(JNIEnv *env, jobject this, jlong addr,
 127                                          jarray b, jint off, jint len, jint flush)
 128 {
 129     z_stream *strm = jlong_to_ptr(addr);
 130 
 131     jarray this_buf = (*env)->GetObjectField(env, this, bufID);
 132     jint this_off = (*env)->GetIntField(env, this, offID);
 133     jint this_len = (*env)->GetIntField(env, this, lenID);
 134     jbyte *in_buf;
 135     jbyte *out_buf;
 136     int res;
 137     if ((*env)->GetBooleanField(env, this, setParamsID)) {
 138         int level = (*env)->GetIntField(env, this, levelID);
 139         int strategy = (*env)->GetIntField(env, this, strategyID);
 140         in_buf = (*env)->GetPrimitiveArrayCritical(env, this_buf, 0);
 141         if (in_buf == NULL) {
 142             // Throw OOME only when length is not zero
 143             if (this_len != 0 && (*env)->ExceptionOccurred(env) == NULL)
 144                 JNU_ThrowOutOfMemoryError(env, 0);
 145             return 0;
 146         }
 147         out_buf = (*env)->GetPrimitiveArrayCritical(env, b, 0);
 148         if (out_buf == NULL) {
 149             (*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);
 150             if (len != 0 && (*env)->ExceptionOccurred(env) == NULL)
 151                 JNU_ThrowOutOfMemoryError(env, 0);
 152             return 0;
 153         }
 154 
 155         strm->next_in = (Bytef *) (in_buf + this_off);
 156         strm->next_out = (Bytef *) (out_buf + off);
 157         strm->avail_in = this_len;
 158         strm->avail_out = len;
 159         res = deflateParams(strm, level, strategy);
 160         (*env)->ReleasePrimitiveArrayCritical(env, b, out_buf, 0);
 161         (*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);
 162 
 163         switch (res) {
 164         case Z_OK:
 165             (*env)->SetBooleanField(env, this, setParamsID, JNI_FALSE);
 166             this_off += this_len - strm->avail_in;
 167             (*env)->SetIntField(env, this, offID, this_off);
 168             (*env)->SetIntField(env, this, lenID, strm->avail_in);
 169             return (jint) (len - strm->avail_out);
 170         case Z_BUF_ERROR:
 171             (*env)->SetBooleanField(env, this, setParamsID, JNI_FALSE);
 172             return 0;
 173         default:
 174             JNU_ThrowInternalError(env, strm->msg);
 175             return 0;
 176         }
 177     } else {
 178         jboolean finish = (*env)->GetBooleanField(env, this, finishID);
 179         in_buf = (*env)->GetPrimitiveArrayCritical(env, this_buf, 0);
 180         if (in_buf == NULL) {
 181             if (this_len != 0)
 182                 JNU_ThrowOutOfMemoryError(env, 0);
 183             return 0;
 184         }
 185         out_buf = (*env)->GetPrimitiveArrayCritical(env, b, 0);
 186         if (out_buf == NULL) {
 187             (*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);
 188             if (len != 0)
 189                 JNU_ThrowOutOfMemoryError(env, 0);
 190 
 191             return 0;
 192         }
 193 
 194         strm->next_in = (Bytef *) (in_buf + this_off);
 195         strm->next_out = (Bytef *) (out_buf + off);
 196         strm->avail_in = this_len;
 197         strm->avail_out = len;
 198         res = deflate(strm, finish ? Z_FINISH : flush);
 199         (*env)->ReleasePrimitiveArrayCritical(env, b, out_buf, 0);
 200         (*env)->ReleasePrimitiveArrayCritical(env, this_buf, in_buf, 0);
 201 
 202         switch (res) {
 203         case Z_STREAM_END:
 204             (*env)->SetBooleanField(env, this, finishedID, JNI_TRUE);
 205             /* fall through */
 206         case Z_OK:
 207             this_off += this_len - strm->avail_in;
 208             (*env)->SetIntField(env, this, offID, this_off);
 209             (*env)->SetIntField(env, this, lenID, strm->avail_in);
 210             return len - strm->avail_out;
 211         case Z_BUF_ERROR:
 212             return 0;
 213             default:
 214             JNU_ThrowInternalError(env, strm->msg);
 215             return 0;
 216         }
 217     }
 218 }
 219 
 220 JNIEXPORT jint JNICALL
 221 Java_java_util_zip_Deflater_getAdler(JNIEnv *env, jclass cls, jlong addr)
 222 {
 223     return ((z_stream *)jlong_to_ptr(addr))->adler;
 224 }
 225 
 226 JNIEXPORT void JNICALL
 227 Java_java_util_zip_Deflater_reset(JNIEnv *env, jclass cls, jlong addr)
 228 {
 229     if (deflateReset((z_stream *)jlong_to_ptr(addr)) != Z_OK) {
 230         JNU_ThrowInternalError(env, 0);
 231     }
 232 }
 233 
 234 JNIEXPORT void JNICALL
 235 Java_java_util_zip_Deflater_end(JNIEnv *env, jclass cls, jlong addr)
 236 {
 237     if (deflateEnd((z_stream *)jlong_to_ptr(addr)) == Z_STREAM_ERROR) {
 238         JNU_ThrowInternalError(env, 0);
 239     } else {
 240         free((z_stream *)jlong_to_ptr(addr));
 241     }
 242 }