/*

 * 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 <sys/socket.h>

#include <string.h>

#include <errno.h>

#include <unistd.h>

#include <jni.h>

#include <netinet/tcp.h>

#include <netinet/in.h>

#include "jni_util.h"

/*

 * Class:     jdk_net_LinuxSocketOptions

 * Method:    setQuickAck

 * Signature: (II)V

 */

JNIEXPORT void JNICALL Java_jdk_net_LinuxSocketOptions_setQuickAck0

(JNIEnv *env, jobject unused, jint fd, jboolean on) {

    int optval;

    int rv;

    optval = (on ? 1 : 0);

    rv = setsockopt(fd, SOL_SOCKET, TCP_QUICKACK, &optval, sizeof (optval));

    if (rv < 0) {

        if (errno == ENOPROTOOPT) {

            JNU_ThrowByName(env, "java/lang/UnsupportedOperationException",

                            "unsupported socket option");

        } else {

            JNU_ThrowByNameWithLastError(env, "java/net/SocketException",

                                        "set option TCP_QUICKACK failed");

        }

    }

}

/*

 * Class:     jdk_net_LinuxSocketOptions

 * Method:    getQuickAck

 * Signature: (I)Z;

 */

JNIEXPORT jboolean JNICALL Java_jdk_net_LinuxSocketOptions_getQuickAck0

(JNIEnv *env, jobject unused, jint fd) {

    int on;

    socklen_t sz = sizeof (on);

    int rv = getsockopt(fd, SOL_SOCKET, TCP_QUICKACK, &on, &sz);

    if (rv < 0) {

        if (errno == ENOPROTOOPT) {

            JNU_ThrowByName(env, "java/lang/UnsupportedOperationException",

                            "unsupported socket option");

        } else {

            JNU_ThrowByNameWithLastError(env, "java/net/SocketException",

                                        "get option TCP_QUICKACK failed");

        }

    }

    return on != 0;

}

/*

 * Class:     jdk_net_LinuxSocketOptions

 * Method:    quickAckSupported

 * Signature: ()Z

 */

JNIEXPORT jboolean JNICALL Java_jdk_net_LinuxSocketOptions_quickAckSupported0

(JNIEnv *env, jobject unused) {

    int one = 1;

    int rv, s;

    s = socket(PF_INET, SOCK_STREAM, 0);

    if (s < 0) {

        return JNI_FALSE;

    }

    rv = setsockopt(s, SOL_SOCKET, TCP_QUICKACK, (void *) &one, sizeof (one));

    if (rv != 0 && errno == ENOPROTOOPT) {

        rv = JNI_FALSE;

    } else {

        rv = JNI_TRUE;

    }

    close(s);

    return rv;

}

static jint socketOptionSupported(jint sockopt) {

    jint one = 1;

    jint rv, s;

    s = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);

    if (s < 0) {

        return 0;

    }

    rv = setsockopt(s, SOL_TCP, sockopt, (void *) &one, sizeof (one));

    if (rv != 0 && errno == ENOPROTOOPT) {

        rv = 0;

    } else {

        rv = 1;

    }

    close(s);

    return rv;

}

static void handleError(JNIEnv *env, jint rv, const char *errmsg) {

    if (rv < 0) {

        if (errno == ENOPROTOOPT) {

            JNU_ThrowByName(env, "java/lang/UnsupportedOperationException",

                    "unsupported socket option");

        } else {

            JNU_ThrowByNameWithLastError(env, "java/net/SocketException", errmsg);

        }

    }

}

/*

 * Class:     jdk_net_LinuxSocketOptions

 * Method:    keepAliveOptionsSupported0

 * Signature: ()Z

 */

JNIEXPORT jboolean JNICALL Java_jdk_net_LinuxSocketOptions_keepAliveOptionsSupported0

(JNIEnv *env, jobject unused) {

    return socketOptionSupported(TCP_KEEPIDLE) && socketOptionSupported(TCP_KEEPCNT)

            && socketOptionSupported(TCP_KEEPINTVL);

}

/*

 * Class:     jdk_net_LinuxSocketOptions

 * Method:    setTcpkeepAliveProbes0

 * Signature: (II)V

 */

JNIEXPORT void JNICALL Java_jdk_net_LinuxSocketOptions_setTcpkeepAliveProbes0

(JNIEnv *env, jobject unused, jint fd, jint optval) {

    jint rv = setsockopt(fd, SOL_TCP, TCP_KEEPCNT, &optval, sizeof (optval));

    handleError(env, rv, "set option TCP_KEEPCNT failed");

}

/*

 * Class:     jdk_net_LinuxSocketOptions

 * Method:    setTcpKeepAliveTime0

 * Signature: (II)V

 */

JNIEXPORT void JNICALL Java_jdk_net_LinuxSocketOptions_setTcpKeepAliveTime0

(JNIEnv *env, jobject unused, jint fd, jint optval) {

    jint rv = setsockopt(fd, SOL_TCP, TCP_KEEPIDLE, &optval, sizeof (optval));

    handleError(env, rv, "set option TCP_KEEPIDLE failed");

}

/*

 * Class:     jdk_net_LinuxSocketOptions

 * Method:    setTcpKeepAliveIntvl0

 * Signature: (II)V

 */

JNIEXPORT void JNICALL Java_jdk_net_LinuxSocketOptions_setTcpKeepAliveIntvl0

(JNIEnv *env, jobject unused, jint fd, jint optval) {

    jint rv = setsockopt(fd, SOL_TCP, TCP_KEEPINTVL, &optval, sizeof (optval));

    handleError(env, rv, "set option TCP_KEEPINTVL failed");

}

/*

 * Class:     jdk_net_LinuxSocketOptions

 * Method:    getTcpkeepAliveProbes0

 * Signature: (I)I;

 */

JNIEXPORT jint JNICALL Java_jdk_net_LinuxSocketOptions_getTcpkeepAliveProbes0

(JNIEnv *env, jobject unused, jint fd) {

    jint optval, rv;

    socklen_t sz = sizeof (optval);

    rv = getsockopt(fd, SOL_TCP, TCP_KEEPCNT, &optval, &sz);

    handleError(env, rv, "get option TCP_KEEPCNT failed");

    return optval;

}

/*

 * Class:     jdk_net_LinuxSocketOptions

 * Method:    getTcpKeepAliveTime0

 * Signature: (I)I;

 */

JNIEXPORT jint JNICALL Java_jdk_net_LinuxSocketOptions_getTcpKeepAliveTime0

(JNIEnv *env, jobject unused, jint fd) {

    jint optval, rv;

    socklen_t sz = sizeof (optval);

    rv = getsockopt(fd, SOL_TCP, TCP_KEEPIDLE, &optval, &sz);

    handleError(env, rv, "get option TCP_KEEPIDLE failed");

    return optval;

}

/*

 * Class:     jdk_net_LinuxSocketOptions

 * Method:    getTcpKeepAliveIntvl0

 * Signature: (I)I;

 */

JNIEXPORT jint JNICALL Java_jdk_net_LinuxSocketOptions_getTcpKeepAliveIntvl0

(JNIEnv *env, jobject unused, jint fd) {

    jint optval, rv;

    socklen_t sz = sizeof (optval);

    rv = getsockopt(fd, SOL_TCP, TCP_KEEPINTVL, &optval, &sz);

    handleError(env, rv, "get option TCP_KEEPINTVL failed");

    return optval;

}