/*
 * Copyright (c) 2014, 2016, 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.
 */

#include <jni.h>
#include <stdio.h>
#include <stdlib.h>
#include <dlfcn.h>
#include <link.h>
#include "nativeFunc.h"

/* standard md5/md/softcrypto method names (ordering is from mapfile) */
static const char MD5_INIT[]                     = "MD5Init";
static const char MD5_UPDATE[]                   = "MD5Update";
static const char MD5_FINAL[]                    = "MD5Final";
static const char SHA1_INIT[]                    = "SHA1Init";
static const char SHA1_UPDATE[]                  = "SHA1Update";
static const char SHA1_FINAL[]                   = "SHA1Final";
static const char SHA2_INIT[]                    = "SHA2Init";
static const char SHA2_UPDATE[]                  = "SHA2Update";
static const char SHA2_FINAL[]                   = "SHA2Final";
static const char UCRYPTO_VERSION[]              = "ucrypto_version";
static const char UCRYPTO_GET_MECHLIST[]         = "ucrypto_get_mechlist";

static const char UCRYPTO_ENCRYPT_INIT[]         = "ucrypto_encrypt_init";
static const char UCRYPTO_ENCRYPT_UPDATE[]       = "ucrypto_encrypt_update";
static const char UCRYPTO_ENCRYPT_FINAL[]        = "ucrypto_encrypt_final";
static const char UCRYPTO_ENCRYPT[]              = "ucrypto_encrypt";

static const char UCRYPTO_DECRYPT_INIT[]         = "ucrypto_decrypt_init";
static const char UCRYPTO_DECRYPT_UPDATE[]       = "ucrypto_decrypt_update";
static const char UCRYPTO_DECRYPT_FINAL[]        = "ucrypto_decrypt_final";
static const char UCRYPTO_DECRYPT[]              = "ucrypto_decrypt";

static const char UCRYPTO_SIGN_INIT[]            = "ucrypto_sign_init";
static const char UCRYPTO_SIGN_UPDATE[]          = "ucrypto_sign_update";
static const char UCRYPTO_SIGN_FINAL[]           = "ucrypto_sign_final";

static const char UCRYPTO_VERIFY_INIT[]          = "ucrypto_verify_init";
static const char UCRYPTO_VERIFY_UPDATE[]        = "ucrypto_verify_update";
static const char UCRYPTO_VERIFY_FINAL[]         = "ucrypto_verify_final";

static const char UCRYPTO_DIGEST_INIT[]          = "ucrypto_digest_init";
static const char UCRYPTO_DIGEST_UPDATE[]        = "ucrypto_digest_update";
static const char UCRYPTO_DIGEST_FINAL[]         = "ucrypto_digest_final";

static const char UCRYPTO_FREE_CONTEXT[]         = "ucrypto_free_context";

static const char UCRYPTO_STRERROR[]             = "ucrypto_strerror";

/**
 * Initialize native T4 crypto function pointers
 */
jboolean* loadNative() {

  jboolean* buf;
  void *lib;

  buf = malloc(2 * sizeof(jboolean));
  buf[0] = buf[1] = JNI_FALSE;
  ftab = (T4CRYPTO_FUNCTION_TABLE_PTR) calloc(1, sizeof(T4CRYPTO_FUNCTION_TABLE));
  if (ftab == NULL) {
    free(buf);
    return NULL;
  }

  lib = dlopen("libsoftcrypto.so", RTLD_NOW);
  if (lib != NULL) {
    // These APIs aren't available for v0 lib on Solaris 10
    ftab->ucryptoVersion = (UCRYPTO_VERSION_FN_PTR)
      dlsym(lib, UCRYPTO_VERSION);
    ftab->ucryptoGetMechList = (UCRYPTO_GET_MECHLIST_FN_PTR)
      dlsym(lib, UCRYPTO_GET_MECHLIST);
    ftab->ucryptoSignInit = (UCRYPTO_SIGN_INIT_FN_PTR)
      dlsym(lib, UCRYPTO_SIGN_INIT);
    ftab->ucryptoSignUpdate = (UCRYPTO_SIGN_UPDATE_FN_PTR)
      dlsym(lib, UCRYPTO_SIGN_UPDATE);
    ftab->ucryptoSignFinal = (UCRYPTO_SIGN_FINAL_FN_PTR)
      dlsym(lib, UCRYPTO_SIGN_FINAL);
    ftab->ucryptoVerifyInit = (UCRYPTO_VERIFY_INIT_FN_PTR)
      dlsym(lib, UCRYPTO_VERIFY_INIT);
    ftab->ucryptoVerifyUpdate = (UCRYPTO_VERIFY_UPDATE_FN_PTR)
      dlsym(lib, UCRYPTO_VERIFY_UPDATE);
    ftab->ucryptoVerifyFinal = (UCRYPTO_VERIFY_FINAL_FN_PTR)
      dlsym(lib, UCRYPTO_VERIFY_FINAL);

    // These APS are added starting S12
    ftab->ucryptoDigestInit = (UCRYPTO_DIGEST_INIT_FN_PTR)
      dlsym(lib, UCRYPTO_DIGEST_INIT);
    ftab->ucryptoDigestUpdate = (UCRYPTO_DIGEST_UPDATE_FN_PTR)
      dlsym(lib, UCRYPTO_DIGEST_UPDATE);
    ftab->ucryptoDigestFinal = (UCRYPTO_DIGEST_FINAL_FN_PTR)
      dlsym(lib, UCRYPTO_DIGEST_FINAL);

    ftab->ucryptoFreeContext = (UCRYPTO_FREE_CONTEXT_FN_PTR)
      dlsym(lib, UCRYPTO_FREE_CONTEXT);

    ftab->ucryptoStrerror = (UCRYPTO_STRERROR_FN_PTR)
      dlsym(lib, UCRYPTO_STRERROR);


    // These should be avilable for all libsoftcrypto libs
    ftab->ucryptoEncryptInit = (UCRYPTO_ENCRYPT_INIT_FN_PTR)
      dlsym(lib, UCRYPTO_ENCRYPT_INIT);
    ftab->ucryptoEncryptUpdate = (UCRYPTO_ENCRYPT_UPDATE_FN_PTR)
      dlsym(lib, UCRYPTO_ENCRYPT_UPDATE);
    ftab->ucryptoEncryptFinal = (UCRYPTO_ENCRYPT_FINAL_FN_PTR)
      dlsym(lib, UCRYPTO_ENCRYPT_FINAL);
    ftab->ucryptoEncrypt = (UCRYPTO_ENCRYPT_FN_PTR)
      dlsym(lib, UCRYPTO_ENCRYPT);

    ftab->ucryptoDecryptInit = (UCRYPTO_DECRYPT_INIT_FN_PTR)
      dlsym(lib, UCRYPTO_DECRYPT_INIT);
    ftab->ucryptoDecryptUpdate = (UCRYPTO_DECRYPT_UPDATE_FN_PTR)
      dlsym(lib, UCRYPTO_DECRYPT_UPDATE);
    ftab->ucryptoDecryptFinal = (UCRYPTO_DECRYPT_FINAL_FN_PTR)
      dlsym(lib, UCRYPTO_DECRYPT_FINAL);
    ftab->ucryptoDecrypt = (UCRYPTO_DECRYPT_FN_PTR)
      dlsym(lib, UCRYPTO_DECRYPT);

    if (ftab->ucryptoEncryptInit != NULL &&
        ftab->ucryptoEncryptUpdate != NULL &&
        ftab->ucryptoEncryptFinal != NULL &&
        ftab->ucryptoEncrypt != NULL &&
        ftab->ucryptoDecryptInit != NULL &&
        ftab->ucryptoDecryptUpdate != NULL &&
        ftab->ucryptoDecryptFinal != NULL &&
        ftab->ucryptoDecrypt != NULL) {
      buf[1] = JNI_TRUE;
    } else {
      dlclose(lib);
    }

    // proceed with libmd when libucrypto does not support digest operations
    if (ftab->ucryptoDigestInit == NULL ||
        ftab->ucryptoDigestUpdate == NULL ||
        ftab->ucryptoDigestFinal == NULL) {

      lib = dlopen("libmd.so", RTLD_NOW);
      if (lib != NULL) {
        ftab->md5Init = (MD5INIT_FN_PTR) dlsym(lib, MD5_INIT);
        ftab->md5Update = (MD5UPDATE_FN_PTR) dlsym(lib, MD5_UPDATE);
        ftab->md5Final = (MD5FINAL_FN_PTR) dlsym(lib, MD5_FINAL);
        ftab->sha1Init = (SHA1INIT_FN_PTR) dlsym(lib, SHA1_INIT);
        ftab->sha1Update = (SHA1UPDATE_FN_PTR) dlsym(lib, SHA1_UPDATE);
        ftab->sha1Final = (SHA1FINAL_FN_PTR) dlsym(lib, SHA1_FINAL);
        ftab->sha2Init = (SHA2INIT_FN_PTR) dlsym(lib, SHA2_INIT);
        ftab->sha2Update = (SHA2UPDATE_FN_PTR) dlsym(lib, SHA2_UPDATE);
        ftab->sha2Final = (SHA2FINAL_FN_PTR) dlsym(lib, SHA2_FINAL);
        if (ftab->md5Init != NULL && ftab->md5Update != NULL &&
            ftab->md5Final != NULL && ftab->sha1Init != NULL &&
            ftab->sha1Update != NULL && ftab->sha1Final != NULL &&
            ftab->sha2Init != NULL && ftab->sha2Update != NULL &&
            ftab->sha2Final != NULL) {
          buf[0] = JNI_TRUE;
        } else {
          dlclose(lib);
        }
      }
    }
  }

  return buf;
}