1 /*
   2  * Copyright (c) 2000, 2020, 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 <sys/mman.h>
  27 #include <sys/stat.h>
  28 #include <fcntl.h>
  29 #include <sys/types.h>
  30 #include <unistd.h>
  31 
  32 #if defined(__linux__)
  33 #include <sys/sendfile.h>
  34 #elif defined(_AIX)
  35 #include <string.h>
  36 #include <sys/socket.h>
  37 #elif defined(_ALLBSD_SOURCE)
  38 #include <sys/socket.h>
  39 #include <sys/uio.h>
  40 #define lseek64 lseek
  41 #define mmap64 mmap
  42 #endif
  43 
  44 #include "jni.h"
  45 #include "jni_util.h"
  46 #include "jlong.h"
  47 #include "nio.h"
  48 #include "nio_util.h"
  49 #include "sun_nio_ch_FileChannelImpl.h"
  50 #include "java_lang_Integer.h"
  51 #include <assert.h>
  52 
  53 static jfieldID chan_fd;        /* jobject 'fd' in sun.nio.ch.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, jboolean map_sync)
  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     // should never be called with map_sync and prot == PRIVATE
  86     assert((prot != sun_nio_ch_FileChannelImpl_MAP_PV) || !map_sync);
  87 
  88     if (prot == sun_nio_ch_FileChannelImpl_MAP_RO) {
  89         protections = PROT_READ;
  90         flags = MAP_SHARED;
  91     } else if (prot == sun_nio_ch_FileChannelImpl_MAP_RW) {
  92         protections = PROT_WRITE | PROT_READ;
  93         flags = MAP_SHARED;
  94     } else if (prot == sun_nio_ch_FileChannelImpl_MAP_PV) {
  95         protections =  PROT_WRITE | PROT_READ;
  96         flags = MAP_PRIVATE;
  97     }
  98 
  99     // if MAP_SYNC and MAP_SHARED_VALIDATE are not defined then it is
 100     // best to define them here. This ensures the code compiles on old
 101     // OS releases which do not provide the relevant headers. If run
 102     // on the same machine then it will work if the kernel contains
 103     // the necessary support otherwise mmap should fail with an
 104     // invalid argument error
 105 
 106 #ifndef MAP_SYNC
 107 #define MAP_SYNC 0x80000
 108 #endif
 109 #ifndef MAP_SHARED_VALIDATE
 110 #define MAP_SHARED_VALIDATE 0x03
 111 #endif
 112 
 113     if (map_sync) {
 114         // ensure
 115         //  1) this is Linux on AArch64, x86_64, or PPC64 LE
 116         //  2) the mmap APIs are available at compile time
 117 #if !defined(LINUX) || ! (defined(aarch64) || (defined(amd64) && defined(_LP64)) || defined(ppc64le))
 118         // TODO - implement for solaris/AIX/BSD/WINDOWS and for 32 bit
 119         JNU_ThrowInternalError(env, "should never call map on platform where MAP_SYNC is unimplemented");
 120         return IOS_THROWN;
 121 #else
 122         flags |= MAP_SYNC | MAP_SHARED_VALIDATE;
 123 #endif
 124     }
 125 
 126     mapAddress = mmap64(
 127         0,                    /* Let OS decide location */
 128         len,                  /* Number of bytes to map */
 129         protections,          /* File permissions */
 130         flags,                /* Changes are shared */
 131         fd,                   /* File descriptor of mapped file */
 132         off);                 /* Offset into file */
 133 
 134     if (mapAddress == MAP_FAILED) {
 135         if (map_sync && errno == ENOTSUP) {
 136             JNU_ThrowIOExceptionWithLastError(env, "map with mode MAP_SYNC unsupported");
 137             return IOS_THROWN;
 138         }
 139 
 140         if (errno == ENOMEM) {
 141             JNU_ThrowOutOfMemoryError(env, "Map failed");
 142             return IOS_THROWN;
 143         }
 144         return handle(env, -1, "Map failed");
 145     }
 146 
 147     return ((jlong) (unsigned long) mapAddress);
 148 }
 149 
 150 
 151 JNIEXPORT jint JNICALL
 152 Java_sun_nio_ch_FileChannelImpl_unmap0(JNIEnv *env, jobject this,
 153                                        jlong address, jlong len)
 154 {
 155     void *a = (void *)jlong_to_ptr(address);
 156     return handle(env,
 157                   munmap(a, (size_t)len),
 158                   "Unmap failed");
 159 }
 160 
 161 JNIEXPORT jlong JNICALL
 162 Java_sun_nio_ch_FileChannelImpl_transferTo0(JNIEnv *env, jobject this,
 163                                             jobject srcFDO,
 164                                             jlong position, jlong count,
 165                                             jobject dstFDO)
 166 {
 167     jint srcFD = fdval(env, srcFDO);
 168     jint dstFD = fdval(env, dstFDO);
 169 
 170 #if defined(__linux__)
 171     off64_t offset = (off64_t)position;
 172     jlong n = sendfile64(dstFD, srcFD, &offset, (size_t)count);
 173     if (n < 0) {
 174         if (errno == EAGAIN)
 175             return IOS_UNAVAILABLE;
 176         if ((errno == EINVAL) && ((ssize_t)count >= 0))
 177             return IOS_UNSUPPORTED_CASE;
 178         if (errno == EINTR) {
 179             return IOS_INTERRUPTED;
 180         }
 181         JNU_ThrowIOExceptionWithLastError(env, "Transfer failed");
 182         return IOS_THROWN;
 183     }
 184     return n;
 185 #elif defined(__APPLE__)
 186     off_t numBytes;
 187     int result;
 188 
 189     numBytes = count;
 190 
 191     result = sendfile(srcFD, dstFD, position, &numBytes, NULL, 0);
 192 
 193     if (numBytes > 0)
 194         return numBytes;
 195 
 196     if (result == -1) {
 197         if (errno == EAGAIN)
 198             return IOS_UNAVAILABLE;
 199         if (errno == EOPNOTSUPP || errno == ENOTSOCK || errno == ENOTCONN)
 200             return IOS_UNSUPPORTED_CASE;
 201         if ((errno == EINVAL) && ((ssize_t)count >= 0))
 202             return IOS_UNSUPPORTED_CASE;
 203         if (errno == EINTR)
 204             return IOS_INTERRUPTED;
 205         JNU_ThrowIOExceptionWithLastError(env, "Transfer failed");
 206         return IOS_THROWN;
 207     }
 208 
 209     return result;
 210 
 211 #elif defined(_AIX)
 212     jlong max = (jlong)java_lang_Integer_MAX_VALUE;
 213     struct sf_parms sf_iobuf;
 214     jlong result;
 215 
 216     if (position > max)
 217         return IOS_UNSUPPORTED_CASE;
 218 
 219     if (count > max)
 220         count = max;
 221 
 222     memset(&sf_iobuf, 0, sizeof(sf_iobuf));
 223     sf_iobuf.file_descriptor = srcFD;
 224     sf_iobuf.file_offset = (off_t)position;
 225     sf_iobuf.file_bytes = count;
 226 
 227     result = send_file(&dstFD, &sf_iobuf, SF_SYNC_CACHE);
 228 
 229     /* AIX send_file() will return 0 when this operation complete successfully,
 230      * return 1 when partial bytes transfered and return -1 when an error has
 231      * Occured.
 232      */
 233     if (result == -1) {
 234         if (errno == EWOULDBLOCK)
 235             return IOS_UNAVAILABLE;
 236         if ((errno == EINVAL) && ((ssize_t)count >= 0))
 237             return IOS_UNSUPPORTED_CASE;
 238         if (errno == EINTR)
 239             return IOS_INTERRUPTED;
 240         if (errno == ENOTSOCK)
 241             return IOS_UNSUPPORTED;
 242         JNU_ThrowIOExceptionWithLastError(env, "Transfer failed");
 243         return IOS_THROWN;
 244     }
 245 
 246     if (sf_iobuf.bytes_sent > 0)
 247         return (jlong)sf_iobuf.bytes_sent;
 248 
 249     return IOS_UNSUPPORTED_CASE;
 250 #else
 251     return IOS_UNSUPPORTED_CASE;
 252 #endif
 253 }
 254