/*

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

#include <winsock2.h>

#include "jni.h"

#include "jni_util.h"

#include "jvm.h"

#include "jlong.h"

#include "nio.h"

#include "nio_util.h"

#include "net_util.h"

#include "sun_nio_ch_Net.h"

#include "sun_nio_ch_PollArrayWrapper.h"

/**

 * Definitions to allow for building with older SDK include files.

 */

#ifndef MCAST_BLOCK_SOURCE

#define MCAST_BLOCK_SOURCE          43

#define MCAST_UNBLOCK_SOURCE        44

#define MCAST_JOIN_SOURCE_GROUP     45

#define MCAST_LEAVE_SOURCE_GROUP    46

#endif  /* MCAST_BLOCK_SOURCE */

struct my_ip_mreq_source {

    IN_ADDR imr_multiaddr;

    IN_ADDR imr_sourceaddr;

    IN_ADDR imr_interface;

};

struct my_group_source_req {

    ULONG gsr_interface;

    SOCKADDR_STORAGE gsr_group;

    SOCKADDR_STORAGE gsr_source;

};

/**

 * Copy IPv6 address as jbytearray to target

 */

#define COPY_INET6_ADDRESS(env, source, target) \

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

/**

 * Enable or disable receipt of WSAECONNRESET errors.

 */

static void setConnectionReset(SOCKET s, BOOL enable) {

    DWORD bytesReturned = 0;

    WSAIoctl(s, SIO_UDP_CONNRESET, &enable, sizeof(enable),

             NULL, 0, &bytesReturned, NULL, NULL);

}

jint handleSocketError(JNIEnv *env, int errorValue)

{

    NET_ThrowNew(env, errorValue, NULL);

    return IOS_THROWN;

}

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

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

extern jclass udsa_class;

extern jmethodID udsa_ctorID;

extern jfieldID udsa_pathID;

JNIEXPORT jobject JNICALL

    if (sa->sun_family == AF_UNIX) {

        jstring nstr = JNU_NewStringPlatform(env, name);

        return (*env)->NewObject(env, udsa_class, udsa_ctorID, nstr);

    }

    return NULL;

}

JNIEXPORT jint JNICALL

NET_UnixSocketAddressToSockaddr(JNIEnv *env, jobject uaddr, struct sockaddr_un *sa, int *len)

{

    jstring path = (*env)->GetObjectField(env, uaddr, udsa_pathID);

    jboolean isCopy;

    int ret;

    const char* pname = JNU_GetStringPlatformChars(env, path, &isCopy);

    memset(sa, 0, sizeof(struct sockaddr_un));

    sa->sun_family = AF_UNIX;

    size_t name_len = strlen(pname)+1;

    if (name_len > MAX_UNIX_DOMAIN_PATH_LEN) {

        JNU_ThrowByName(env, JNU_JAVANETPKG "SocketException", "unix domain path too long");

        ret = 1;

        goto finish;

    }

    strncpy(sa->sun_path, pname, name_len);

    *len = (int)(offsetof(struct sockaddr_un, sun_path) + name_len);

    ret = 0;

  finish:

    if (isCopy)

        JNU_ReleaseStringPlatformChars(env, path, pname);

    return ret;

}

JNIEXPORT jboolean JNICALL

Java_sun_nio_ch_Net_unixDomainSocketSupported(JNIEnv *env, jclass cl)

{

    SOCKET fd = socket(PF_UNIX, SOCK_STREAM, 0);

    if (fd == INVALID_SOCKET) {

        return JNI_FALSE;

    }

    closesocket(fd);

    return JNI_TRUE;

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_Net_unixDomainSocket0(JNIEnv *env, jclass cl)

{

    SOCKET fd = socket(PF_UNIX, SOCK_STREAM, 0);

    if (fd == INVALID_SOCKET) {

        return handleSocketError(env, errno);

    }

    return (int)fd;

}

JNIEXPORT void JNICALL

Java_sun_nio_ch_Net_unixDomainBind(JNIEnv *env, jclass clazz, jobject fdo, jobject uaddr)

{

    struct sockaddr_un sa;

    int sa_len = 0;

    int rv = 0;

    if (NET_UnixSocketAddressToSockaddr(env, uaddr, &sa, &sa_len) != 0)

        return;

    int fd = fdval(env, fdo);

    rv = bind(fdval(env, fdo), (struct sockaddr *)&sa, sa_len);

    if (rv != 0) {

        int err = WSAGetLastError();

        NET_ThrowNew(env, err, "bind");

    }

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_Net_unixDomainConnect(JNIEnv *env, jclass clazz, jobject fdo, jobject usa)

{

    struct sockaddr_un sa;

    int sa_len = 0;

    int rv;

    if (NET_UnixSocketAddressToSockaddr(env, usa, &sa, &sa_len) != 0) {

        return IOS_THROWN;

    }

    rv = connect(fdval(env, fdo), (const struct sockaddr *)&sa, sa_len);

    if (rv != 0) {

        int err = WSAGetLastError();

        if (err == WSAEINPROGRESS || err == WSAEWOULDBLOCK) {

            return IOS_UNAVAILABLE;

        }

        NET_ThrowNew(env, err, "connect");

        return IOS_THROWN;

    }

    return 1;

}

JNIEXPORT jint JNICALL

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

                           jobjectArray usaa)

{

    jint fd = fdval(env, fdo);

    jint newfd;

    struct sockaddr_un sa;

    socklen_t sa_len = sizeof(sa);

    jobject usa;

    /* accept connection but ignore ECONNABORTED */

    for (;;) {

        newfd = (jint)accept(fd, (struct sockaddr *)&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);

    CHECK_NULL_RETURN(usa, IOS_THROWN);

    (*env)->SetObjectArrayElement(env, usaa, 0, usa);

    return 1;

}

JNIEXPORT jobject JNICALL

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

{

    struct sockaddr_un sa;

    socklen_t sa_len = sizeof(sa);

    if (getsockname(fdval(env, fdo), (struct sockaddr *)&sa, &sa_len) < 0) {

        handleSocketError(env, errno);

        return NULL;

    }

}

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 true if IPv6 is configured

     */

    return ipv6_available() ? JNI_TRUE : JNI_FALSE;

}

JNIEXPORT jboolean JNICALL

Java_sun_nio_ch_Net_isReusePortAvailable0(JNIEnv* env, jclass c1)

{

    // SO_REUSEPORT is not supported on Windows

    return 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)

{

    /* IPv6 sockets can join IPv4 multicast groups */

    return JNI_TRUE;

}

JNIEXPORT jboolean JNICALL

Java_sun_nio_ch_Net_canJoin6WithIPv4Group0(JNIEnv* env, jclass cl)

{

    /* IPV6_ADD_MEMBERSHIP cannot be used to join IPv4 multicast groups */

    return JNI_FALSE;

}

JNIEXPORT jboolean JNICALL

Java_sun_nio_ch_Net_canUseIPv6OptionsWithIPv4LocalAddress0(JNIEnv* env, jclass cl)

{

    /* IPV6_XXX socket options cannot be used on IPv6 sockets bound to IPv4 address */

    return JNI_FALSE;

}

JNIEXPORT jint JNICALL

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

                            jboolean stream, jboolean reuse, jboolean fastLoopback)

{

    SOCKET s;

    int domain = (preferIPv6) ? AF_INET6 : AF_INET;

    s = socket(domain, (stream ? SOCK_STREAM : SOCK_DGRAM), 0);

    if (s != INVALID_SOCKET) {

        SetHandleInformation((HANDLE)s, HANDLE_FLAG_INHERIT, 0);

        /* IPV6_V6ONLY is true by default */

        if (domain == AF_INET6) {

            int opt = 0;

            setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY,

                       (const char *)&opt, sizeof(opt));

        }

        /* Disable WSAECONNRESET errors for initially unconnected UDP sockets */

        if (!stream) {

            setConnectionReset(s, FALSE);

        }

    } else {

        NET_ThrowNew(env, WSAGetLastError(), "socket");

    }

    if (stream && fastLoopback) {

        static int loopback_available = 1;

        if (loopback_available) {

            int rv = NET_EnableFastTcpLoopback((jint)s);

            if (rv) {

                if (rv == WSAEOPNOTSUPP) {

                    loopback_available = 0;

                } else {

                    NET_ThrowNew(env, rv, "fastLoopback");

                }

            }

        }

    }

    return (jint)s;

}

JNIEXPORT void JNICALL

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

                          jboolean isExclBind, jobject iao, jint port)

{

    SOCKETADDRESS sa;

    int rv;

    int sa_len = 0;

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

        return;

    }

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

    if (rv == SOCKET_ERROR)

        NET_ThrowNew(env, WSAGetLastError(), "bind");

}

JNIEXPORT void JNICALL

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

{

    if (listen(fdval(env,fdo), backlog) == SOCKET_ERROR) {

        NET_ThrowNew(env, WSAGetLastError(), "listen");

    }

}

JNIEXPORT jint JNICALL

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

                             jobject iao, jint port)

{

    SOCKETADDRESS sa;

    int rv;

    int sa_len = 0;

    SOCKET s = (SOCKET)fdval(env, fdo);

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

        return IOS_THROWN;

    }

    rv = connect(s, &sa.sa, sa_len);

    if (rv != 0) {

        int err = WSAGetLastError();

        if (err == WSAEINPROGRESS || err == WSAEWOULDBLOCK) {

            return IOS_UNAVAILABLE;

        }

        NET_ThrowNew(env, err, "connect");

        return IOS_THROWN;

    } else {

        /* Enable WSAECONNRESET errors when a UDP socket is connected */

        int type = 0, optlen = sizeof(type);

        rv = getsockopt(s, SOL_SOCKET, SO_TYPE, (char*)&type, &optlen);

        if (rv == 0 && type == SOCK_DGRAM) {

            setConnectionReset(s, TRUE);

        }

    }

    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;

    int addrlen = sizeof(sa);

    jobject remote_ia;

    jint remote_port = 0;

    jobject isa;

    memset((char *)&sa, 0, sizeof(sa));

    newfd = (jint) accept(fd, &sa.sa, &addrlen);

    if (newfd == INVALID_SOCKET) {

        int theErr = (jint)WSAGetLastError();

        if (theErr == WSAEWOULDBLOCK) {

            return IOS_UNAVAILABLE;

        }

        JNU_ThrowIOExceptionWithLastError(env, "Accept failed");

        return IOS_THROWN;

    }

    SetHandleInformation((HANDLE)(UINT_PTR)newfd, HANDLE_FLAG_INHERIT, 0);

    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;

    int sa_len = sizeof(sa);

    if (getsockname(fdval(env, fdo), &sa.sa, &sa_len) == SOCKET_ERROR) {

        int error = WSAGetLastError();

        if (error == WSAEINVAL) {

            return 0;

        }

        NET_ThrowNew(env, error, "getsockname");

        return IOS_THROWN;

    }

    return NET_GetPortFromSockaddr(&sa);

}

JNIEXPORT jobject JNICALL

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

{

    SOCKETADDRESS sa;

    int sa_len = sizeof(sa);

    int port;

    if (getsockname(fdval(env, fdo), &sa.sa, &sa_len) == SOCKET_ERROR) {

        NET_ThrowNew(env, WSAGetLastError(), "getsockname");

        return NULL;

    }

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

}

JNIEXPORT jint JNICALL

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

{

    SOCKETADDRESS sa;

    int sa_len = sizeof(sa);

    if (getpeername(fdval(env, fdo), &sa.sa, &sa_len) == SOCKET_ERROR) {

        int error = WSAGetLastError();

        if (error == WSAEINVAL) {

            return 0;

        }

        NET_ThrowNew(env, error, "getsockname");

        return IOS_THROWN;

    }

    return NET_GetPortFromSockaddr(&sa);