1 /*
   2  * Copyright (c) 2000, 2017, 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 "jni.h"
  27 #include "jni_util.h"
  28 #include "jvm.h"
  29 #include "jvm_md.h"
  30 #include "jlong.h"
  31 #include <sys/mman.h>
  32 #include <sys/stat.h>
  33 #include <fcntl.h>
  34 #include "sun_nio_ch_FileChannelImpl.h"
  35 #include "java_lang_Integer.h"
  36 #include "nio.h"
  37 #include "nio_util.h"
  38 #include <dlfcn.h>
  39 
  40 #if defined(__linux__) || defined(__solaris__)
  41 #include <sys/sendfile.h>
  42 #elif defined(_AIX)
  43 #include <sys/socket.h>
  44 #elif defined(_ALLBSD_SOURCE)
  45 #include <sys/types.h>
  46 #include <sys/socket.h>
  47 #include <sys/uio.h>
  48 
  49 #define lseek64 lseek
  50 #define mmap64 mmap
  51 #endif
  52 
  53 static jfieldID chan_fd;        /* jobject 'fd' in sun.io.FileChannelImpl */
  54 
  55 JNIEXPORT jlong JNICALL
  56 Java_sun_nio_ch_FileChannelImpl_initIDs(JNIEnv *env, jclass clazz)
  57 {
  58     jlong pageSize = sysconf(_SC_PAGESIZE);
  59     chan_fd = (*env)->GetFieldID(env, clazz, "fd", "Ljava/io/FileDescriptor;");
  60     return pageSize;
  61 }
  62 
  63 static jlong
  64 handle(JNIEnv *env, jlong rv, char *msg)
  65 {
  66     if (rv >= 0)
  67         return rv;
  68     if (errno == EINTR)
  69         return IOS_INTERRUPTED;
  70     JNU_ThrowIOExceptionWithLastError(env, msg);
  71     return IOS_THROWN;
  72 }
  73 
  74 
  75 JNIEXPORT jlong JNICALL
  76 Java_sun_nio_ch_FileChannelImpl_map0(JNIEnv *env, jobject this,
  77                                      jint prot, jlong off, jlong len)
  78 {
  79     void *mapAddress = 0;
  80     jobject fdo = (*env)->GetObjectField(env, this, chan_fd);
  81     jint fd = fdval(env, fdo);
  82     int protections = 0;
  83     int flags = 0;
  84 
  85     if (prot == sun_nio_ch_FileChannelImpl_MAP_RO) {
  86         protections = PROT_READ;
  87         flags = MAP_SHARED;
  88     } else if (prot == sun_nio_ch_FileChannelImpl_MAP_RW) {
  89         protections = PROT_WRITE | PROT_READ;
  90         flags = MAP_SHARED;
  91     } else if (prot == sun_nio_ch_FileChannelImpl_MAP_PV) {
  92         protections =  PROT_WRITE | PROT_READ;
  93         flags = MAP_PRIVATE;
  94     }
  95 
  96     mapAddress = mmap64(
  97         0,                    /* Let OS decide location */
  98         len,                  /* Number of bytes to map */
  99         protections,          /* File permissions */
 100         flags,                /* Changes are shared */
 101         fd,                   /* File descriptor of mapped file */
 102         off);                 /* Offset into file */
 103 
 104     if (mapAddress == MAP_FAILED) {
 105         if (errno == ENOMEM) {
 106             JNU_ThrowOutOfMemoryError(env, "Map failed");
 107             return IOS_THROWN;
 108         }
 109         return handle(env, -1, "Map failed");
 110     }
 111 
 112     return ((jlong) (unsigned long) mapAddress);
 113 }
 114 
 115 
 116 JNIEXPORT jint JNICALL
 117 Java_sun_nio_ch_FileChannelImpl_unmap0(JNIEnv *env, jobject this,
 118                                        jlong address, jlong len)
 119 {
 120     void *a = (void *)jlong_to_ptr(address);
 121     return handle(env,
 122                   munmap(a, (size_t)len),
 123                   "Unmap failed");
 124 }
 125 
 126 
 127 JNIEXPORT jlong JNICALL
 128 Java_sun_nio_ch_FileChannelImpl_position0(JNIEnv *env, jobject this,
 129                                           jobject fdo, jlong offset)
 130 {
 131     jint fd = fdval(env, fdo);
 132     jlong result = 0;
 133 
 134     if (offset < 0) {
 135         result = lseek64(fd, 0, SEEK_CUR);
 136     } else {
 137         result = lseek64(fd, offset, SEEK_SET);
 138     }
 139     return handle(env, result, "Position failed");
 140 }
 141 
 142 
 143 JNIEXPORT jlong JNICALL
 144 Java_sun_nio_ch_FileChannelImpl_transferTo0(JNIEnv *env, jobject this,
 145                                             jobject srcFDO,
 146                                             jlong position, jlong count,
 147                                             jobject dstFDO)
 148 {
 149     jint srcFD = fdval(env, srcFDO);
 150     jint dstFD = fdval(env, dstFDO);
 151 
 152 #if defined(__linux__)
 153     off64_t offset = (off64_t)position;
 154     jlong n = sendfile64(dstFD, srcFD, &offset, (size_t)count);
 155     if (n < 0) {
 156         if (errno == EAGAIN)
 157             return IOS_UNAVAILABLE;
 158         if ((errno == EINVAL) && ((ssize_t)count >= 0))
 159             return IOS_UNSUPPORTED_CASE;
 160         if (errno == EINTR) {
 161             return IOS_INTERRUPTED;
 162         }
 163         JNU_ThrowIOExceptionWithLastError(env, "Transfer failed");
 164         return IOS_THROWN;
 165     }
 166     return n;
 167 #elif defined (__solaris__)
 168     sendfilevec64_t sfv;
 169     size_t numBytes = 0;
 170     jlong result;
 171 
 172     sfv.sfv_fd = srcFD;
 173     sfv.sfv_flag = 0;
 174     sfv.sfv_off = (off64_t)position;
 175     sfv.sfv_len = count;
 176 
 177     result = sendfilev64(dstFD, &sfv, 1, &numBytes);
 178 
 179     /* Solaris sendfilev() will return -1 even if some bytes have been
 180      * transferred, so we check numBytes first.
 181      */
 182     if (numBytes > 0)
 183         return numBytes;
 184     if (result < 0) {
 185         if (errno == EAGAIN)
 186             return IOS_UNAVAILABLE;
 187         if (errno == EOPNOTSUPP)
 188             return IOS_UNSUPPORTED_CASE;
 189         if ((errno == EINVAL) && ((ssize_t)count >= 0))
 190             return IOS_UNSUPPORTED_CASE;
 191         if (errno == EINTR)
 192             return IOS_INTERRUPTED;
 193         JNU_ThrowIOExceptionWithLastError(env, "Transfer failed");
 194         return IOS_THROWN;
 195     }
 196     return result;
 197 #elif defined(__APPLE__)
 198     off_t numBytes;
 199     int result;
 200 
 201     numBytes = count;
 202 
 203     result = sendfile(srcFD, dstFD, position, &numBytes, NULL, 0);
 204 
 205     if (numBytes > 0)
 206         return numBytes;
 207 
 208     if (result == -1) {
 209         if (errno == EAGAIN)
 210             return IOS_UNAVAILABLE;
 211         if (errno == EOPNOTSUPP || errno == ENOTSOCK || errno == ENOTCONN)
 212             return IOS_UNSUPPORTED_CASE;
 213         if ((errno == EINVAL) && ((ssize_t)count >= 0))
 214             return IOS_UNSUPPORTED_CASE;
 215         if (errno == EINTR)
 216             return IOS_INTERRUPTED;
 217         JNU_ThrowIOExceptionWithLastError(env, "Transfer failed");
 218         return IOS_THROWN;
 219     }
 220 
 221     return result;
 222 
 223 #elif defined(_AIX)
 224     jlong max = (jlong)java_lang_Integer_MAX_VALUE;
 225     struct sf_parms sf_iobuf;
 226     jlong result;
 227 
 228     if (position > max)
 229         return IOS_UNSUPPORTED_CASE;
 230 
 231     if (count > max)
 232         count = max;
 233 
 234     memset(&sf_iobuf, 0, sizeof(sf_iobuf));
 235     sf_iobuf.file_descriptor = srcFD;
 236     sf_iobuf.file_offset = (off_t)position;
 237     sf_iobuf.file_bytes = count;
 238 
 239     result = send_file(&dstFD, &sf_iobuf, SF_SYNC_CACHE);
 240 
 241     /* AIX send_file() will return 0 when this operation complete successfully,
 242      * return 1 when partial bytes transfered and return -1 when an error has
 243      * Occured.
 244      */
 245     if (result == -1) {
 246         if (errno == EWOULDBLOCK)
 247             return IOS_UNAVAILABLE;
 248         if ((errno == EINVAL) && ((ssize_t)count >= 0))
 249             return IOS_UNSUPPORTED_CASE;
 250         if (errno == EINTR)
 251             return IOS_INTERRUPTED;
 252         if (errno == ENOTSOCK)
 253             return IOS_UNSUPPORTED;
 254         JNU_ThrowIOExceptionWithLastError(env, "Transfer failed");
 255         return IOS_THROWN;
 256     }
 257 
 258     if (sf_iobuf.bytes_sent > 0)
 259         return (jlong)sf_iobuf.bytes_sent;
 260 
 261     return IOS_UNSUPPORTED_CASE;
 262 #else
 263     return IOS_UNSUPPORTED_CASE;
 264 #endif
 265 }
 266