/*

 * Copyright (c) 2001, 2019, 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 <poll.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <string.h>

#include <netinet/in.h>

#include <netinet/tcp.h>

#include <limits.h>

#include "jni.h"

#include "jni_util.h"

#include "jvm.h"

#include "jlong.h"

#include "sun_nio_ch_Net.h"

#include "net_util.h"

#include "net_util_md.h"

#include "nio_util.h"

#include "nio.h"

#ifdef _AIX

#include <sys/utsname.h>

#endif

/**

 * IP_MULTICAST_ALL supported since 2.6.31 but may not be available at

 * build time.

 */

#ifdef __linux__

  #ifndef IP_MULTICAST_ALL

    #define IP_MULTICAST_ALL    49

  #endif

#endif

/**

 * IPV6_ADD_MEMBERSHIP/IPV6_DROP_MEMBERSHIP may not be defined on OSX and AIX

 */

#if defined(__APPLE__) || defined(_AIX)

  #ifndef IPV6_ADD_MEMBERSHIP

    #define IPV6_ADD_MEMBERSHIP     IPV6_JOIN_GROUP

    #define IPV6_DROP_MEMBERSHIP    IPV6_LEAVE_GROUP

  #endif

#endif

#if defined(_AIX)

  #ifndef IP_BLOCK_SOURCE

    #define IP_BLOCK_SOURCE                 58   /* Block data from a given source to a given group */

    #define IP_UNBLOCK_SOURCE               59   /* Unblock data from a given source to a given group */

    #define IP_ADD_SOURCE_MEMBERSHIP        60   /* Join a source-specific group */

    #define IP_DROP_SOURCE_MEMBERSHIP       61   /* Leave a source-specific group */

  #endif

  #ifndef MCAST_BLOCK_SOURCE

    #define MCAST_BLOCK_SOURCE              64

    #define MCAST_UNBLOCK_SOURCE            65

    #define MCAST_JOIN_SOURCE_GROUP         66

    #define MCAST_LEAVE_SOURCE_GROUP        67

    /* This means we're on AIX 5.3 and 'group_source_req' and 'ip_mreq_source' aren't defined as well */

    struct group_source_req {

        uint32_t gsr_interface;

        struct sockaddr_storage gsr_group;

        struct sockaddr_storage gsr_source;

    };

    struct ip_mreq_source {

        struct in_addr  imr_multiaddr;  /* IP multicast address of group */

        struct in_addr  imr_sourceaddr; /* IP address of source */

        struct in_addr  imr_interface;  /* local IP address of interface */

    };

  #endif

#endif /* _AIX */

#define COPY_INET6_ADDRESS(env, source, target) \

    (*env)->GetByteArrayRegion(env, source, 0, 16, target)

/*

 * Copy IPv6 group, interface index, and IPv6 source address

 * into group_source_req structure.

 */

static void initGroupSourceReq(JNIEnv* env, jbyteArray group, jint index,

                               jbyteArray source, struct group_source_req *req)

{

    struct sockaddr_in6* sin6;

    req->gsr_interface = (uint32_t)index;

    sin6 = (struct sockaddr_in6 *)&(req->gsr_group);

    sin6->sin6_family = AF_INET6;

    COPY_INET6_ADDRESS(env, group, (jbyte *)&(sin6->sin6_addr));

    sin6 = (struct sockaddr_in6 *)&(req->gsr_source);

    sin6->sin6_family = AF_INET6;

    COPY_INET6_ADDRESS(env, source, (jbyte *)&(sin6->sin6_addr));

}

#ifdef _AIX

/*

 * Checks whether or not "socket extensions for multicast source filters" is supported.

 * Returns JNI_TRUE if it is supported, JNI_FALSE otherwise

 */

static jboolean isSourceFilterSupported(){

    static jboolean alreadyChecked = JNI_FALSE;

    static jboolean result = JNI_TRUE;

    if (alreadyChecked != JNI_TRUE){

        struct utsname uts;

        memset(&uts, 0, sizeof(uts));

        strcpy(uts.sysname, "?");

        const int utsRes = uname(&uts);

        int major = -1;

        int minor = -1;

        major = atoi(uts.version);

        minor = atoi(uts.release);

        if (strcmp(uts.sysname, "AIX") == 0) {

            if (major < 6 || (major == 6 && minor < 1)) {// unsupported on aix < 6.1

                result = JNI_FALSE;

            }

        }

        alreadyChecked = JNI_TRUE;

    }

    return result;

}

#endif  /* _AIX */

static jclass isa_class;        /* java.net.InetSocketAddress */

static jmethodID isa_ctorID;    /* InetSocketAddress(InetAddress, int) */

JNIEXPORT void JNICALL

Java_sun_nio_ch_Net_initIDs(JNIEnv *env, jclass clazz)

{

     jclass cls = (*env)->FindClass(env, "java/net/InetSocketAddress");

     CHECK_NULL(cls);

     isa_class = (*env)->NewGlobalRef(env, cls);

     if (isa_class == NULL) {

         JNU_ThrowOutOfMemoryError(env, NULL);

         return;

     }

     isa_ctorID = (*env)->GetMethodID(env, cls, "<init>", "(Ljava/net/InetAddress;I)V");

     CHECK_NULL(isa_ctorID);

     initInetAddressIDs(env);

}

JNIEXPORT jboolean JNICALL

Java_sun_nio_ch_Net_isIPv6Available0(JNIEnv* env, jclass cl)

{

    return (ipv6_available()) ? JNI_TRUE : JNI_FALSE;

}

JNIEXPORT jboolean JNICALL

Java_sun_nio_ch_Net_isReusePortAvailable0(JNIEnv* env, jclass c1)

{

    return (reuseport_available()) ? JNI_TRUE : JNI_FALSE;

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_Net_isExclusiveBindAvailable(JNIEnv *env, jclass clazz) {

    return -1;

}

JNIEXPORT jboolean JNICALL

Java_sun_nio_ch_Net_canIPv6SocketJoinIPv4Group0(JNIEnv* env, jclass cl)

{

#if defined(__APPLE__) || defined(_AIX)

    /* for now IPv6 sockets cannot join IPv4 multicast groups */

    return JNI_FALSE;

#else

    return JNI_TRUE;

#endif

}

JNIEXPORT jboolean JNICALL

Java_sun_nio_ch_Net_canJoin6WithIPv4Group0(JNIEnv* env, jclass cl)

{

#ifdef __solaris__

    return JNI_TRUE;

#else

    return JNI_FALSE;

#endif

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_Net_socket0(JNIEnv *env, jclass cl, jboolean preferIPv6,

                            jboolean stream, jboolean reuse, jboolean ignored)

{

    int fd;

    int type = (stream ? SOCK_STREAM : SOCK_DGRAM);

    int domain = (ipv6_available() && preferIPv6) ? AF_INET6 : AF_INET;

    fd = socket(domain, type, 0);

    if (fd < 0) {

        return handleSocketError(env, errno);

    }

    /* Disable IPV6_V6ONLY to ensure dual-socket support */

    if (domain == AF_INET6) {

        int arg = 0;

        if (setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, (char*)&arg,

                       sizeof(int)) < 0) {

            JNU_ThrowByNameWithLastError(env,

                                         JNU_JAVANETPKG "SocketException",

                                         "Unable to set IPV6_V6ONLY");

            close(fd);

            return -1;

        }

    }

    if (reuse) {

        int arg = 1;

        if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char*)&arg,

                       sizeof(arg)) < 0) {

            JNU_ThrowByNameWithLastError(env,

                                         JNU_JAVANETPKG "SocketException",

                                         "Unable to set SO_REUSEADDR");

            close(fd);

            return -1;

        }

    }

#if defined(__linux__)

    if (type == SOCK_DGRAM) {

        int arg = 0;

        int level = (domain == AF_INET6) ? IPPROTO_IPV6 : IPPROTO_IP;

        if ((setsockopt(fd, level, IP_MULTICAST_ALL, (char*)&arg, sizeof(arg)) < 0) &&

            (errno != ENOPROTOOPT)) {

            JNU_ThrowByNameWithLastError(env,

                                         JNU_JAVANETPKG "SocketException",

                                         "Unable to set IP_MULTICAST_ALL");

            close(fd);

            return -1;

        }

    }

    /* By default, Linux uses the route default */

    if (domain == AF_INET6 && type == SOCK_DGRAM) {

        int arg = 1;

        if (setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &arg,

                       sizeof(arg)) < 0) {

            JNU_ThrowByNameWithLastError(env,

                                         JNU_JAVANETPKG "SocketException",

                                         "Unable to set IPV6_MULTICAST_HOPS");

            close(fd);

            return -1;

        }

    }

#endif

    return fd;

}

JNIEXPORT void JNICALL

Java_sun_nio_ch_Net_bind0(JNIEnv *env, jclass clazz, jobject fdo, jboolean preferIPv6,

                          jboolean useExclBind, jobject iao, int port)

{

    SOCKETADDRESS sa;

    int sa_len = 0;

    int rv = 0;

    if (NET_InetAddressToSockaddr(env, iao, port, &sa, &sa_len,

                                  preferIPv6) != 0) {

        return;

    }

    rv = NET_Bind(fdval(env, fdo), &sa, sa_len);

    if (rv != 0) {

        handleSocketError(env, errno);

    }

}

JNIEXPORT void JNICALL

Java_sun_nio_ch_Net_listen(JNIEnv *env, jclass cl, jobject fdo, jint backlog)

{

    if (listen(fdval(env, fdo), backlog) < 0)

        handleSocketError(env, errno);

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_Net_connect0(JNIEnv *env, jclass clazz, jboolean preferIPv6,

                             jobject fdo, jobject iao, jint port)

{

    SOCKETADDRESS sa;

    int sa_len = 0;

    int rv;

    if (NET_InetAddressToSockaddr(env, iao, port, &sa, &sa_len, preferIPv6) != 0) {

        return IOS_THROWN;

    }

    rv = connect(fdval(env, fdo), &sa.sa, sa_len);

    if (rv != 0) {

        if (errno == EINPROGRESS) {

            return IOS_UNAVAILABLE;

        } else if (errno == EINTR) {

            return IOS_INTERRUPTED;

        }

        return handleSocketError(env, errno);

    }

    return 1;

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_Net_accept(JNIEnv *env, jclass clazz, jobject fdo, jobject newfdo,

                           jobjectArray isaa)

{

    jint fd = fdval(env, fdo);

    jint newfd;

    SOCKETADDRESS sa;

    socklen_t sa_len = sizeof(SOCKETADDRESS);

    jobject remote_ia;

    jint remote_port = 0;

    jobject isa;

    /* accept connection but ignore ECONNABORTED */

    for (;;) {

        newfd = accept(fd, &sa.sa, &sa_len);

        if (newfd >= 0) {

            break;

        }

        if (errno != ECONNABORTED) {

            break;

        }

        /* ECONNABORTED => restart accept */

    }

    if (newfd < 0) {

        if (errno == EAGAIN || errno == EWOULDBLOCK)

            return IOS_UNAVAILABLE;

        if (errno == EINTR)

            return IOS_INTERRUPTED;

        JNU_ThrowIOExceptionWithLastError(env, "Accept failed");

        return IOS_THROWN;

    }

    setfdval(env, newfdo, newfd);

    remote_ia = NET_SockaddrToInetAddress(env, &sa, (int *)&remote_port);

    CHECK_NULL_RETURN(remote_ia, IOS_THROWN);

    isa = (*env)->NewObject(env, isa_class, isa_ctorID, remote_ia, remote_port);

    CHECK_NULL_RETURN(isa, IOS_THROWN);

    (*env)->SetObjectArrayElement(env, isaa, 0, isa);

    return 1;

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_Net_localPort(JNIEnv *env, jclass clazz, jobject fdo)

{

    SOCKETADDRESS sa;

    socklen_t sa_len = sizeof(SOCKETADDRESS);

    if (getsockname(fdval(env, fdo), &sa.sa, &sa_len) < 0) {

#ifdef _ALLBSD_SOURCE

        /*

         * XXXBSD:

         * ECONNRESET is specific to the BSDs. We can not return an error,

         * as the calling Java code with raise a java.lang.Error given the expectation

         * that getsockname() will never fail. According to the Single UNIX Specification,

         * it shouldn't fail. As such, we just fill in generic Linux-compatible values.

         */

        if (errno == ECONNRESET) {

            bzero(&sa.sa4, sizeof(sa));

            sa.sa4.sin_len = sizeof(struct sockaddr_in);

            sa.sa4.sin_family = AF_INET;

            sa.sa4.sin_port = htonl(0);

            sa.sa4.sin_addr.s_addr = INADDR_ANY;

        } else {

            handleSocketError(env, errno);

            return -1;

        }

#else /* _ALLBSD_SOURCE */

        handleSocketError(env, errno);

        return -1;

#endif /* _ALLBSD_SOURCE */

    }

    return NET_GetPortFromSockaddr(&sa);

}

JNIEXPORT jobject JNICALL

Java_sun_nio_ch_Net_localInetAddress(JNIEnv *env, jclass clazz, jobject fdo)

{

    SOCKETADDRESS sa;

    socklen_t sa_len = sizeof(SOCKETADDRESS);

    int port;

    if (getsockname(fdval(env, fdo), &sa.sa, &sa_len) < 0) {

#ifdef _ALLBSD_SOURCE

        /*

         * XXXBSD:

         * ECONNRESET is specific to the BSDs. We can not return an error,

         * as the calling Java code with raise a java.lang.Error with the expectation

         * that getsockname() will never fail. According to the Single UNIX Specification,

         * it shouldn't fail. As such, we just fill in generic Linux-compatible values.

         */

        if (errno == ECONNRESET) {

            bzero(&sa.sa4, sizeof(sa));

            sa.sa4.sin_len  = sizeof(struct sockaddr_in);

            sa.sa4.sin_family = AF_INET;

            sa.sa4.sin_port = htonl(0);

            sa.sa4.sin_addr.s_addr = INADDR_ANY;

        } else {

            handleSocketError(env, errno);

            return NULL;

        }

#else /* _ALLBSD_SOURCE */

        handleSocketError(env, errno);

        return NULL;

#endif /* _ALLBSD_SOURCE */

    }

    return NET_SockaddrToInetAddress(env, &sa, &port);

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_Net_remotePort(JNIEnv *env, jclass clazz, jobject fdo)

{

    SOCKETADDRESS sa;

    socklen_t sa_len = sizeof(sa);

    if (getpeername(fdval(env, fdo), &sa.sa, &sa_len) < 0) {

        handleSocketError(env, errno);

        return IOS_THROWN;

    }

    return NET_GetPortFromSockaddr(&sa);

}

JNIEXPORT jobject JNICALL

Java_sun_nio_ch_Net_remoteInetAddress(JNIEnv *env, jclass clazz, jobject fdo)

{

    SOCKETADDRESS sa;

    socklen_t sa_len = sizeof(sa);

    int port;

    if (getpeername(fdval(env, fdo), &sa.sa, &sa_len) < 0) {

        handleSocketError(env, errno);

        return NULL;

    }

    return NET_SockaddrToInetAddress(env, &sa, &port);

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_Net_getIntOption0(JNIEnv *env, jclass clazz, jobject fdo,

                                  jboolean mayNeedConversion, jint level, jint opt)

{

    int result;

    struct linger linger;

    u_char carg;

    void *arg;

    socklen_t arglen;

    int n;

    /* Option value is an int except for a few specific cases */

    arg = (void *)&result;

    arglen = sizeof(result);

    if (level == IPPROTO_IP &&

        (opt == IP_MULTICAST_TTL || opt == IP_MULTICAST_LOOP)) {

        arg = (void*)&carg;

        arglen = sizeof(carg);

    }

    if (level == SOL_SOCKET && opt == SO_LINGER) {

        arg = (void *)&linger;

        arglen = sizeof(linger);

    }

    if (mayNeedConversion) {

        n = NET_GetSockOpt(fdval(env, fdo), level, opt, arg, (int*)&arglen);

    } else {

        n = getsockopt(fdval(env, fdo), level, opt, arg, &arglen);

    }