/*

 * 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 "jlong.h"

#include "jni.h"

#include "jni_util.h"

#include <zlib.h>

#include "java_util_zip_Deflater.h"

#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;

JNIEXPORT void JNICALL

Java_java_util_zip_Deflater_initIDs(JNIEnv *env, jclass cls)

{

    levelID = (*env)->GetFieldID(env, cls, "level", "I");

    CHECK_NULL(levelID);

    strategyID = (*env)->GetFieldID(env, cls, "strategy", "I");

    CHECK_NULL(strategyID);

    setParamsID = (*env)->GetFieldID(env, cls, "setParams", "Z");

    CHECK_NULL(setParamsID);

    finishID = (*env)->GetFieldID(env, cls, "finish", "Z");

    CHECK_NULL(finishID);

    finishedID = (*env)->GetFieldID(env, cls, "finished", "Z");

    CHECK_NULL(finishedID);

    bufID = (*env)->GetFieldID(env, cls, "buf", "[B");

    CHECK_NULL(bufID);

    offID = (*env)->GetFieldID(env, cls, "off", "I");

    CHECK_NULL(offID);

    lenID = (*env)->GetFieldID(env, cls, "len", "I");

    CHECK_NULL(lenID);

}

JNIEXPORT jlong JNICALL

Java_java_util_zip_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 {

        const char *msg;

        int ret = deflateInit2(strm, level, Z_DEFLATED,

                               nowrap ? -MAX_WBITS : MAX_WBITS,

                               DEF_MEM_LEVEL, strategy);

        switch (ret) {

          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 != NULL) ? strm->msg :

                   (ret == Z_VERSION_ERROR) ?

                   "zlib returned Z_VERSION_ERROR: "

                   "compile time and runtime zlib implementations differ" :

                   "unknown error initializing zlib library");

            free(strm);

            JNU_ThrowInternalError(env, msg);

            return jlong_zero;

        }

    }

}

JNIEXPORT void JNICALL

Java_java_util_zip_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

Java_java_util_zip_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 && (*env)->ExceptionOccurred(env) == NULL)

                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 && (*env)->ExceptionOccurred(env) == NULL)

                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 (jint) (len - strm->avail_out);

        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;

            JNU_ThrowInternalError(env, strm->msg);

            return 0;

        }

    }

}

JNIEXPORT jint JNICALL

Java_java_util_zip_Deflater_getAdler(JNIEnv *env, jclass cls, jlong addr)

{

    return ((z_stream *)jlong_to_ptr(addr))->adler;

}

JNIEXPORT void JNICALL

Java_java_util_zip_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

Java_java_util_zip_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));

    }

}