/*
 * ZeroTier One - Network Virtualization Everywhere
 * Copyright (C) 2011-2016  ZeroTier, Inc.  https://www.zerotier.com/
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program 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 for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef ZT_jniutils_h_
#define ZT_jniutils_h_

#include <jni.h>
#include <ZeroTierOne.h>

#include <limits> // for numeric_limits
#include <sys/socket.h> // for sockaddr_storage

#if defined(__ANDROID__)

#include <android/log.h>

    #if !defined(NDEBUG)
        #define LOGV(...) ((void)__android_log_print(ANDROID_LOG_VERBOSE, LOG_TAG, __VA_ARGS__))
        #define LOGD(...) ((void)__android_log_print(ANDROID_LOG_DEBUG, LOG_TAG, __VA_ARGS__))
    #else
        #define LOGV(...)
        #define LOGD(...)
    #endif

    #define LOGI(...) ((void)__android_log_print(ANDROID_LOG_INFO, LOG_TAG, __VA_ARGS__))
    #define LOGE(...) ((void)__android_log_print(ANDROID_LOG_ERROR, LOG_TAG, __VA_ARGS__))
#else
    #if !defined(NDEBUG)
        #define LOGV(...) fprintf(stdout, __VA_ARGS__)
        #define LOGD(...) fprintf(stdout, __VA_ARGS__)
    #else
        #define LOGV(...)
        #define LOGD(...)
    #endif

    #define LOGI(...) fprintf(stdout, __VA_ARGS__)
    #define LOGE(...) fprintf(stdout, __VA_ARGS__)
#endif

//
// Call GetEnv and assert if there is an error
//
#define GETENV(env, vm) \
    do { \
        jint getEnvRet; \
        assert(vm); \
        if ((getEnvRet = vm->GetEnv(reinterpret_cast<void**>(&env), JNI_VERSION_1_6)) != JNI_OK) { \
            LOGE("Error calling GetEnv: %d", getEnvRet); \
            assert(false && "Error calling GetEnv"); \
        } \
    } while (false)

//
// Call GetJavaVM and assert if there is an error
//
#define GETJAVAVM(env, vm) \
    do { \
        jint getJavaVMRet; \
        if ((getJavaVMRet = env->GetJavaVM(&vm)) != 0) { \
            LOGE("Error calling GetJavaVM: %d", getJavaVMRet); \
            assert(false && "Error calling GetJavaVM"); \
        } \
    } while (false)


const jsize JSIZE_MAX = std::numeric_limits<jsize>::max();

jobject createResultObject(JNIEnv *env, ZT_ResultCode code);
jobject createVirtualNetworkStatus(JNIEnv *env, ZT_VirtualNetworkStatus status);
jobject createVirtualNetworkType(JNIEnv *env, ZT_VirtualNetworkType type);
jobject createEvent(JNIEnv *env, ZT_Event event);
jobject createPeerRole(JNIEnv *env, ZT_PeerRole role);
jobject createVirtualNetworkConfigOperation(JNIEnv *env, ZT_VirtualNetworkConfigOperation op);

jobject newInetSocketAddress(JNIEnv *env, const sockaddr_storage &addr);
jobject newInetAddress(JNIEnv *env, const sockaddr_storage &addr);
int addressPort(const sockaddr_storage addr);

jobject newPeer(JNIEnv *env, const ZT_Peer &peer);
jobject newPeerPhysicalPath(JNIEnv *env, const ZT_PeerPhysicalPath &ppp);

jobject newNetworkConfig(JNIEnv *env, const ZT_VirtualNetworkConfig &config);

jobject newVersion(JNIEnv *env, int major, int minor, int rev);

jobject newVirtualNetworkRoute(JNIEnv *env, const ZT_VirtualNetworkRoute &route);

jobject newVirtualNetworkDNS(JNIEnv *env, const ZT_VirtualNetworkDNS &dns);

jobject newNodeStatus(JNIEnv *env, const ZT_NodeStatus &status);

jobjectArray newPeerArray(JNIEnv *env, const ZT_Peer *peers, size_t count);

jobjectArray newVirtualNetworkConfigArray(JNIEnv *env, const ZT_VirtualNetworkConfig *networks, size_t count);

jobjectArray newPeerPhysicalPathArray(JNIEnv *env, const ZT_PeerPhysicalPath *paths, size_t count);

jobjectArray newInetSocketAddressArray(JNIEnv *env, const sockaddr_storage *addresses, size_t count);

jobjectArray newVirtualNetworkRouteArray(JNIEnv *env, const ZT_VirtualNetworkRoute *routes, size_t count);

//
// log functions only for newArrayObject below
//
void newArrayObject_logCount(size_t count);
void newArrayObject_log(const char *msg);

//
// function template for creating array objects
//
template <typename T, jobject (*F)(JNIEnv *, const T &)>
jobjectArray newArrayObject(JNIEnv *env, const T *buffer, size_t count, jclass clazz) {

    if (count > JSIZE_MAX) {
        newArrayObject_logCount(count);
        return NULL;
    }

    jsize jCount = static_cast<jsize>(count);

    jobjectArray arrayObj = env->NewObjectArray(jCount, clazz, NULL);
    if (env->ExceptionCheck() || arrayObj == NULL) {
        newArrayObject_log("Error creating array object");
        return NULL;
    }

    for (jsize i = 0; i < jCount; i++) {

        jobject obj = F(env, buffer[i]);
        if(env->ExceptionCheck() || obj == NULL) {
            return NULL;
        }

        env->SetObjectArrayElement(arrayObj, i, obj);
        if(env->ExceptionCheck()) {
            newArrayObject_log("Error assigning object to array");
            return NULL;
        }

        env->DeleteLocalRef(obj);
    }

    return arrayObj;
}

jbyteArray newByteArray(JNIEnv *env, const unsigned char *bytes, size_t count);

jbyteArray newByteArray(JNIEnv *env, size_t count);

bool isSocketAddressEmpty(const sockaddr_storage addr);

sockaddr_storage fromSocketAddressObject(JNIEnv *env, jobject sockAddressObject);

#endif // ZT_jniutils_h_