1 /* 2 * Copyright (c) 2000, 2018, 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 <stdlib.h> 27 #include <netdb.h> 28 #include <sys/types.h> 29 #include <sys/socket.h> 30 #include <rdma/rsocket.h> 31 #include <netinet/in.h> 32 33 #include "jni.h" 34 #include "jni_util.h" 35 #include "net_util.h" 36 #include "jvm.h" 37 #include "jlong.h" 38 #include "rdma_ch_RdmaServerSocketChannelImpl.h" 39 #include "nio.h" 40 #include "nio_util.h" 41 42 43 static jfieldID fd_fdID; /* java.io.FileDescriptor.fd */ 44 static jclass isa_class; /* java.net.InetSocketAddress */ 45 static jmethodID isa_ctorID; /* .InetSocketAddress(InetAddress, int) */ 46 47 48 JNIEXPORT void JNICALL 49 Java_rdma_ch_RdmaServerSocketChannelImpl_initIDs(JNIEnv *env, jclass c) 50 { 51 jclass cls; 52 53 cls = (*env)->FindClass(env, "java/io/FileDescriptor"); 54 CHECK_NULL(cls); 55 fd_fdID = (*env)->GetFieldID(env, cls, "fd", "I"); 56 CHECK_NULL(fd_fdID); 57 58 cls = (*env)->FindClass(env, "java/net/InetSocketAddress"); 59 CHECK_NULL(cls); 60 isa_class = (*env)->NewGlobalRef(env, cls); 61 if (isa_class == NULL) { 62 JNU_ThrowOutOfMemoryError(env, NULL); 63 return; 64 } 65 isa_ctorID = (*env)->GetMethodID(env, cls, "<init>", 66 "(Ljava/net/InetAddress;I)V"); 67 CHECK_NULL(isa_ctorID); 68 } 69 70 JNIEXPORT jint JNICALL 71 Java_rdma_ch_RdmaServerSocketChannelImpl_accept0(JNIEnv *env, jobject this, 72 jobject ssfdo, jobject newfdo, 73 jobjectArray isaa) 74 { 75 jint ssfd = (*env)->GetIntField(env, ssfdo, fd_fdID); 76 jint newfd; 77 SOCKETADDRESS sa; 78 socklen_t sa_len = sizeof(SOCKETADDRESS); 79 jobject remote_ia = 0; 80 jobject isa; 81 jint remote_port = 0; 82 83 /* 84 * accept connection but ignore ECONNABORTED indicating that 85 * a connection was eagerly accepted but was reset before 86 * accept() was called. 87 */ 88 for (;;) { 89 newfd = raccept(ssfd, &sa.sa, &sa_len); 90 if (newfd >= 0) { 91 break; 92 } 93 if (errno != ECONNABORTED) { 94 break; 95 } 96 /* ECONNABORTED => restart accept */ 97 } 98 99 if (newfd < 0) { 100 if (errno == EAGAIN) 101 return IOS_UNAVAILABLE; 102 if (errno == EINTR) 103 return IOS_INTERRUPTED; 104 JNU_ThrowIOExceptionWithLastError(env, "Accept failed"); 105 return IOS_THROWN; 106 } 107 108 (*env)->SetIntField(env, newfdo, fd_fdID, newfd); 109 remote_ia = NET_SockaddrToInetAddress(env, &sa, (int *)&remote_port); 110 CHECK_NULL_RETURN(remote_ia, IOS_THROWN); 111 isa = (*env)->NewObject(env, isa_class, isa_ctorID, remote_ia, remote_port); 112 CHECK_NULL_RETURN(isa, IOS_THROWN); 113 (*env)->SetObjectArrayElement(env, isaa, 0, isa); 114 return 1; 115 }