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 }