1 /*
2  * Copyright (c) 2000, 2010, Oracle and/or its affiliates. All rights reserved.
3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4  *
5  * This code is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 only, as
7  * published by the Free Software Foundation.  Oracle designates this
8  * particular file as subject to the "Classpath" exception as provided
9  * by Oracle in the LICENSE file that accompanied this code.
10  *
11  * This code is distributed in the hope that it will be useful, but WITHOUT
12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14  * version 2 for more details (a copy is included in the LICENSE file that
15  * accompanied this code).
16  *
17  * You should have received a copy of the GNU General Public License version
18  * 2 along with this work; if not, write to the Free Software Foundation,
19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20  *
21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22  * or visit www.oracle.com if you need additional information or have any
23  * questions.
24  */
25 
26 #include <netdb.h>
27 #include <sys/types.h>
28 #include <sys/socket.h>
29 #include <stdlib.h>
30 #include <errno.h>
31 #include <string.h>
32 #include <poll.h>
33 
34 #if __linux__
35 #include <netinet/in.h>
36 #endif
37 
38 #include "jni.h"
39 #include "jni_util.h"
40 #include "net_util.h"
41 #include "jvm.h"
42 #include "jlong.h"
43 #include "nio_util.h"
44 #include "nio.h"
45 
46 #include <nativehelper/JNIHelp.h>
47 
48 #define NATIVE_METHOD(className, functionName, signature) \
49 { #functionName, signature, (void*)(Java_sun_nio_ch_ ## className ## _ ## functionName) }
50 
51 JNIEXPORT jint JNICALL
Java_sun_nio_ch_SocketChannelImpl_checkConnect(JNIEnv * env,jobject this,jobject fdo,jboolean block,jboolean ready)52 Java_sun_nio_ch_SocketChannelImpl_checkConnect(JNIEnv *env, jobject this,
53                                                jobject fdo, jboolean block,
54                                                jboolean ready)
55 {
56     int error = 0;
57     socklen_t n = sizeof(int);
58     jint fd = fdval(env, fdo);
59     int result = 0;
60     struct pollfd poller;
61 
62     poller.revents = 1;
63     if (!ready) {
64         poller.fd = fd;
65         poller.events = POLLOUT;
66         poller.revents = 0;
67         result = poll(&poller, 1, block ? -1 : 0);
68         if (result < 0) {
69             JNU_ThrowIOExceptionWithLastError(env, "Poll failed");
70             return IOS_THROWN;
71         }
72         if (!block && (result == 0))
73             return IOS_UNAVAILABLE;
74     }
75 
76     if (poller.revents) {
77         errno = 0;
78         result = getsockopt(fd, SOL_SOCKET, SO_ERROR, &error, &n);
79         if (result < 0) {
80             handleSocketError(env, errno);
81             return JNI_FALSE;
82         } else if (error) {
83             handleSocketError(env, error);
84             return JNI_FALSE;
85         }
86         return 1;
87     }
88     return 0;
89 }
90 
91 JNIEXPORT jint JNICALL
Java_sun_nio_ch_SocketChannelImpl_sendOutOfBandData(JNIEnv * env,jclass this,jobject fdo,jbyte b)92 Java_sun_nio_ch_SocketChannelImpl_sendOutOfBandData(JNIEnv* env, jclass this,
93                                                     jobject fdo, jbyte b)
94 {
95     int n = send(fdval(env, fdo), (const void*)&b, 1, MSG_OOB);
96     return convertReturnVal(env, n, JNI_FALSE);
97 }
98 
99 static JNINativeMethod gMethods[] = {
100   NATIVE_METHOD(SocketChannelImpl, checkConnect, "(Ljava/io/FileDescriptor;ZZ)I"),
101   NATIVE_METHOD(SocketChannelImpl, sendOutOfBandData, "(Ljava/io/FileDescriptor;B)I"),
102 };
103 
register_sun_nio_ch_SocketChannelImpl(JNIEnv * env)104 void register_sun_nio_ch_SocketChannelImpl(JNIEnv* env) {
105   jniRegisterNativeMethods(env, "sun/nio/ch/SocketChannelImpl", gMethods, NELEM(gMethods));
106 }
107