/*

 * 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.

 */

#include "sun_security_jgss_wrapper_GSSLibStub.h"

#include "NativeUtil.h"

#include "NativeFunc.h"

#include "jlong.h"

#include <jni.h>

/* Constants for indicating what type of info is needed for inquiries */

const int TYPE_CRED_NAME = 10;

const int TYPE_CRED_TIME = 11;

const int TYPE_CRED_USAGE = 12;

/*

 * Class:     sun_security_jgss_wrapper_GSSLibStub

 * Method:    init

 * Signature: (Ljava/lang/String;Z)Z

 */

JNIEXPORT jboolean JNICALL

Java_sun_security_jgss_wrapper_GSSLibStub_init(JNIEnv *env,

                                               jclass jcls,

                                               jstring jlibName,

                                               jboolean jDebug) {

    const char *libName;

    int failed;

    char *error = NULL;

    if (!jDebug) {

      JGSS_DEBUG = 0;

    } else {

      JGSS_DEBUG = 1;

    }

    if (jlibName == NULL) {

        TRACE0("[GSSLibStub_init] GSS lib name is NULL");

        return JNI_FALSE;

    }

    libName = (*env)->GetStringUTFChars(env, jlibName, NULL);

    if (libName == NULL) {

        return JNI_FALSE;

    }

    TRACE1("[GSSLibStub_init] libName=%s", libName);

    /* initialize global function table */

    failed = loadNative(libName);

    (*env)->ReleaseStringUTFChars(env, jlibName, libName);

    if (!failed) {

        return JNI_TRUE;

    } else {

        if (JGSS_DEBUG) {

#ifdef WIN32

            #define MAX_MSG_SIZE 256

            static CHAR szMsgBuf[MAX_MSG_SIZE];

            DWORD dwRes;

            DWORD dwError = GetLastError();

            dwRes = FormatMessage (

                    FORMAT_MESSAGE_FROM_SYSTEM,

                    NULL,

                    dwError,

                    0,

                    szMsgBuf,

                    MAX_MSG_SIZE,

                    NULL);

            if (0 == dwRes) {

                printf("GSS-API: Unknown failure %d\n", dwError);

            } else {

                printf("GSS-API: %s\n",szMsgBuf);

            }

#else

            char* error = dlerror();

            if (error) {

                TRACE0(error);

            }

#endif

        }

        return JNI_FALSE;

    }

}

/*

 * Class:     sun_security_jgss_wrapper_GSSLibStub

 * Method:    getMechPtr

 * Signature: ([B)J

 */

JNIEXPORT jlong JNICALL

Java_sun_security_jgss_wrapper_GSSLibStub_getMechPtr(JNIEnv *env,

                                                     jclass jcls,

                                                     jbyteArray jbytes) {

  gss_OID cOid;

  unsigned int i, len;

  jbyte* bytes;

  int found;

  if (jbytes != NULL) {

    found = 0;

    len = (unsigned int)((*env)->GetArrayLength(env, jbytes) - 2);

    bytes = (*env)->GetByteArrayElements(env, jbytes, NULL);

    if (bytes == NULL) {

      return ptr_to_jlong(NULL);

    }

    for (i = 0; i < ftab->mechs->count; i++) {

      cOid = &(ftab->mechs->elements[i]);

      if (len == cOid->length &&

          (memcmp(cOid->elements, (bytes + 2), len) == 0)) {

        // Found a match

        found = 1;

        break;

      }

    }

    (*env)->ReleaseByteArrayElements(env, jbytes, bytes, 0);

    if (found != 1) {

      checkStatus(env, NULL, GSS_S_BAD_MECH, 0, "[GSSLibStub_getMechPtr]");

      return ptr_to_jlong(NULL);

    } else {

      return ptr_to_jlong(cOid);

    }

  } else {

    return ptr_to_jlong(GSS_C_NO_OID);

  }

}

/*

 * Utility routine which releases the specified gss_channel_bindings_t

 * structure.

 */

void deleteGSSCB(gss_channel_bindings_t cb) {

  if (cb == GSS_C_NO_CHANNEL_BINDINGS) return;

  /* release initiator address */

  if (cb->initiator_addrtype != GSS_C_AF_NULLADDR) {

    resetGSSBuffer(&(cb->initiator_address));

  }

  /* release acceptor address */

  if (cb->acceptor_addrtype != GSS_C_AF_NULLADDR) {

    resetGSSBuffer(&(cb->acceptor_address));

  }

  /* release application data */

  if (cb->application_data.length != 0) {

    resetGSSBuffer(&(cb->application_data));

  }

  free(cb);

}

/*

 * Utility routine which creates a gss_channel_bindings_t structure

 * using the specified org.ietf.jgss.ChannelBinding object.

 * NOTE: must call deleteGSSCB() to free up the resources.

 */

gss_channel_bindings_t newGSSCB(JNIEnv *env, jobject jcb) {

  gss_channel_bindings_t cb;

  jobject jinetAddr;

  jbyteArray value;

  if (jcb == NULL) {

    return GSS_C_NO_CHANNEL_BINDINGS;

  }

  cb = malloc(sizeof(struct gss_channel_bindings_struct));

  if (cb == NULL) {

    throwOutOfMemoryError(env,NULL);

    return NULL;

  }

  // initialize addrtype in CB first

  cb->initiator_addrtype = GSS_C_AF_NULLADDR;

  cb->acceptor_addrtype = GSS_C_AF_NULLADDR;

  // addresses needs to be initialized to empty

  memset(&cb->initiator_address, 0, sizeof(cb->initiator_address));

  memset(&cb->acceptor_address, 0, sizeof(cb->acceptor_address));

  /* set up initiator address */

  jinetAddr = (*env)->CallObjectMethod(env, jcb,

      MID_ChannelBinding_getInitiatorAddr);

  if ((*env)->ExceptionCheck(env)) {

    goto cleanup;

  }

  if (jinetAddr != NULL) {

    value = (*env)->CallObjectMethod(env, jinetAddr,

                                     MID_InetAddress_getAddr);

    if ((*env)->ExceptionCheck(env)) {

      goto cleanup;

    }

    cb->initiator_addrtype = GSS_C_AF_INET;

    initGSSBuffer(env, value, &(cb->initiator_address));

    if ((*env)->ExceptionCheck(env)) {

      goto cleanup;

    }

  }

  /* set up acceptor address */

  jinetAddr = (*env)->CallObjectMethod(env, jcb,

      MID_ChannelBinding_getAcceptorAddr);

  if ((*env)->ExceptionCheck(env)) {

    goto cleanup;

  }

  if (jinetAddr != NULL) {

    value = (*env)->CallObjectMethod(env, jinetAddr,

                                     MID_InetAddress_getAddr);

    if ((*env)->ExceptionCheck(env)) {

      goto cleanup;

    }

    cb->acceptor_addrtype = GSS_C_AF_INET;

    initGSSBuffer(env, value, &(cb->acceptor_address));

    if ((*env)->ExceptionCheck(env)) {

      goto cleanup;

    }

  }

  /* set up application data */

  value = (*env)->CallObjectMethod(env, jcb,

                                   MID_ChannelBinding_getAppData);

  if ((*env)->ExceptionCheck(env)) {

    goto cleanup;

  }

  initGSSBuffer(env, value, &(cb->application_data));

  if ((*env)->ExceptionCheck(env)) {

    goto cleanup;

  }

  return cb;

cleanup:

  deleteGSSCB(cb);

  return NULL;

}

/*

 * Utility routine for storing the supplementary information

 * into the specified org.ietf.jgss.MessageProp object.

 */

void setSupplementaryInfo(JNIEnv *env, jobject jstub, jobject jprop,

                          int suppInfo, int minor) {

  jboolean isDuplicate, isOld, isUnseq, hasGap;

  jstring minorMsg;

  if (suppInfo != GSS_S_COMPLETE) {

    isDuplicate = ((suppInfo & GSS_S_DUPLICATE_TOKEN) != 0);

    isOld = ((suppInfo & GSS_S_OLD_TOKEN) != 0);

    isUnseq = ((suppInfo & GSS_S_UNSEQ_TOKEN) != 0);

    hasGap = ((suppInfo & GSS_S_GAP_TOKEN) != 0);

    minorMsg = getMinorMessage(env, jstub, minor);

    if ((*env)->ExceptionCheck(env)) {

      return;

    }

    (*env)->CallVoidMethod(env, jprop, MID_MessageProp_setSupplementaryStates,

                           isDuplicate, isOld, isUnseq, hasGap, minor,

                           minorMsg);

  }

}

/*

 * Class:     sun_security_jgss_wrapper_GSSLibStub

 * Method:    indicateMechs

 * Signature: ()[Lorg/ietf/jgss/Oid;

 */

JNIEXPORT jobjectArray JNICALL

Java_sun_security_jgss_wrapper_GSSLibStub_indicateMechs(JNIEnv *env,

                                                        jclass jcls)

{

  if (ftab->mechs != NULL && ftab->mechs != GSS_C_NO_OID_SET) {

    return getJavaOIDArray(env, ftab->mechs);

  } else return NULL;

}

/*

 * Class:     sun_security_jgss_wrapper_GSSLibStub

 * Method:    inquireNamesForMech

 * Signature: ()[Lorg/ietf/jgss/Oid;

 */

JNIEXPORT jobjectArray JNICALL

Java_sun_security_jgss_wrapper_GSSLibStub_inquireNamesForMech(JNIEnv *env,

                                                              jobject jobj)

{

  OM_uint32 minor, major;

  gss_OID mech;

  gss_OID_set nameTypes;

  jobjectArray result;

  if (ftab->inquireNamesForMech != NULL) {

    mech = (gss_OID)jlong_to_ptr((*env)->GetLongField(env, jobj, FID_GSSLibStub_pMech));

    nameTypes = GSS_C_NO_OID_SET;

    /* gss_inquire_names_for_mech(...) => N/A */

    major = (*ftab->inquireNamesForMech)(&minor, mech, &nameTypes);

    /* release intermediate buffers before checking status */

    result = getJavaOIDArray(env, nameTypes);

    deleteGSSOIDSet(nameTypes);

    if ((*env)->ExceptionCheck(env)) {

      return NULL;

    }

    checkStatus(env, jobj, major, minor, "[GSSLibStub_inquireNamesForMech]");

    if ((*env)->ExceptionCheck(env)) {

      return NULL;

    }

    return result;

  }

  return NULL;

}

/*

 * Class:     sun_security_jgss_wrapper_GSSLibStub

 * Method:    releaseName

 * Signature: (J)V

 */

JNIEXPORT void JNICALL

Java_sun_security_jgss_wrapper_GSSLibStub_releaseName(JNIEnv *env,

                                                      jobject jobj,

                                                      jlong pName)

{

  OM_uint32 minor, major;

  gss_name_t nameHdl;

  nameHdl = (gss_name_t) jlong_to_ptr(pName);

  TRACE1("[GSSLibStub_releaseName] %ld", (long) pName);

  if (nameHdl != GSS_C_NO_NAME) {

    /* gss_release_name(...) => GSS_S_BAD_NAME */

    major = (*ftab->releaseName)(&minor, &nameHdl);

    checkStatus(env, jobj, major, minor, "[GSSLibStub_releaseName]");

  }

}

/*

 * Class:     sun_security_jgss_wrapper_GSSLibStub

 * Method:    importName

 * Signature: ([BLorg/ietf/jgss/Oid;)J

 */

JNIEXPORT jlong JNICALL

Java_sun_security_jgss_wrapper_GSSLibStub_importName(JNIEnv *env,

                                                     jobject jobj,

                                                     jbyteArray jnameVal,

                                                     jobject jnameType)

{

  OM_uint32 minor, major;

  gss_buffer_desc nameVal;

  gss_OID nameType;

  gss_name_t nameHdl;

  nameHdl = GSS_C_NO_NAME;

  TRACE0("[GSSLibStub_importName]");

  initGSSBuffer(env, jnameVal, &nameVal);

  if ((*env)->ExceptionCheck(env)) {

      return jlong_zero;

  }

  nameType = newGSSOID(env, jnameType);

  if ((*env)->ExceptionCheck(env)) {

    resetGSSBuffer(&nameVal);

    return jlong_zero;

  }

  /* gss_import_name(...) => GSS_S_BAD_NAMETYPE, GSS_S_BAD_NAME,

     GSS_S_BAD_MECH */

  major = (*ftab->importName)(&minor, &nameVal, nameType, &nameHdl);

  TRACE1("[GSSLibStub_importName] %" PRIuPTR  "", (uintptr_t) nameHdl);

  /* release intermediate buffers */

  deleteGSSOID(nameType);

  resetGSSBuffer(&nameVal);

  checkStatus(env, jobj, major, minor, "[GSSLibStub_importName]");

  if ((*env)->ExceptionCheck(env)) {

    return jlong_zero;

  }

  return ptr_to_jlong(nameHdl);

}

/*

 * Class:     sun_security_jgss_wrapper_GSSLibStub

 * Method:    compareName

 * Signature: (JJ)Z

 */

JNIEXPORT jboolean JNICALL

Java_sun_security_jgss_wrapper_GSSLibStub_compareName(JNIEnv *env,

                                                      jobject jobj,

                                                      jlong pName1,

                                                      jlong pName2)

{

  OM_uint32 minor, major;

  gss_name_t nameHdl1, nameHdl2;

  int isEqual;

  isEqual = 0;

  nameHdl1 = (gss_name_t) jlong_to_ptr(pName1);

  nameHdl2 = (gss_name_t) jlong_to_ptr(pName2);

  TRACE2("[GSSLibStub_compareName] %ld %ld", (long)pName1, (long)pName2);

  if ((nameHdl1 != GSS_C_NO_NAME) && (nameHdl2 != GSS_C_NO_NAME)) {

    /* gss_compare_name(...) => GSS_S_BAD_NAMETYPE, GSS_S_BAD_NAME(!) */

    major = (*ftab->compareName)(&minor, nameHdl1, nameHdl2, &isEqual);

    checkStatus(env, jobj, major, minor, "[GSSLibStub_compareName]");

  }

  return (isEqual != 0);

}

/*

 * Class:     sun_security_jgss_wrapper_GSSLibStub

 * Method:    canonicalizeName

 * Signature: (J)J

 */

JNIEXPORT jlong JNICALL

Java_sun_security_jgss_wrapper_GSSLibStub_canonicalizeName(JNIEnv *env,

                                                           jobject jobj,

                                                           jlong pName)

{

  OM_uint32 minor, major;

  gss_name_t nameHdl, mnNameHdl;

  gss_OID mech;

  nameHdl = (gss_name_t) jlong_to_ptr(pName);

  TRACE1("[GSSLibStub_canonicalizeName] %ld", (long) pName);

  if (nameHdl != GSS_C_NO_NAME) {

    mech = (gss_OID) jlong_to_ptr((*env)->GetLongField(env, jobj, FID_GSSLibStub_pMech));

    mnNameHdl = GSS_C_NO_NAME;

    /* gss_canonicalize_name(...) may return GSS_S_BAD_NAMETYPE,

       GSS_S_BAD_NAME, GSS_S_BAD_MECH */

    major = (*ftab->canonicalizeName)(&minor, nameHdl, mech, &mnNameHdl);

    TRACE1("[GSSLibStub_canonicalizeName] MN=%" PRIuPTR "", (uintptr_t)mnNameHdl);

    checkStatus(env, jobj, major, minor, "[GSSLibStub_canonicalizeName]");

    if ((*env)->ExceptionCheck(env)) {

      return ptr_to_jlong(GSS_C_NO_NAME);

    }

    return ptr_to_jlong(mnNameHdl);

  }

  return ptr_to_jlong(GSS_C_NO_NAME);

}

/*

 * Class:     sun_security_jgss_wrapper_GSSLibStub

 * Method:    exportName

 * Signature: (J)[B

 */

JNIEXPORT jbyteArray JNICALL

Java_sun_security_jgss_wrapper_GSSLibStub_exportName(JNIEnv *env,

                                                     jobject jobj,

                                                     jlong pName) {

  OM_uint32 minor, major;

  gss_name_t nameHdl, mNameHdl;

  gss_buffer_desc outBuf;

  jbyteArray jresult;

  nameHdl = (gss_name_t) jlong_to_ptr(pName);

  TRACE1("[GSSLibStub_exportName] %ld", (long) pName);

  /* gss_export_name(...) => GSS_S_NAME_NOT_MN, GSS_S_BAD_NAMETYPE,

     GSS_S_BAD_NAME */

  major = (*ftab->exportName)(&minor, nameHdl, &outBuf);

  /* canonicalize the internal name to MN and retry */

  if (major == GSS_S_NAME_NOT_MN) {

    /* release intermediate buffers before retrying */

    (*ftab->releaseBuffer)(&minor, &outBuf);

    TRACE0("[GSSLibStub_exportName] canonicalize and re-try");

    mNameHdl = (gss_name_t)jlong_to_ptr(

        Java_sun_security_jgss_wrapper_GSSLibStub_canonicalizeName

                                        (env, jobj, pName));

    if ((*env)->ExceptionCheck(env)) {

        return NULL;

    }

    major = (*ftab->exportName)(&minor, mNameHdl, &outBuf);

    Java_sun_security_jgss_wrapper_GSSLibStub_releaseName

                                        (env, jobj, ptr_to_jlong(mNameHdl));

    if ((*env)->ExceptionCheck(env)) {