/*

 * Copyright 2008-2009 Sun Microsystems, Inc.  All Rights Reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Sun designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Sun in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 */

#include <stdio.h>

#include <stdlib.h>

#include <limits.h>

#include <fcntl.h>

#include <dirent.h>

#include <unistd.h>

#include <pwd.h>

#include <grp.h>

#include <errno.h>

#include <dlfcn.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <sys/statvfs.h>

#include <sys/time.h>

#ifdef __solaris__

#include <strings.h>

#include <sys/mnttab.h>

#include <sys/mkdev.h>

#endif

#ifdef __linux__

#include <string.h>

#include <mntent.h>

#endif

#include "jni.h"

#include "jni_util.h"

#include "jlong.h"

#include "sun_nio_fs_UnixNativeDispatcher.h"

#define RESTARTABLE(_cmd, _result) do { \

  do { \

    _result = _cmd; \

  } while((_result == -1) && (errno == EINTR)); \

} while(0)

static jfieldID attrs_st_mode;

static jfieldID attrs_st_ino;

static jfieldID attrs_st_dev;

static jfieldID attrs_st_rdev;

static jfieldID attrs_st_nlink;

static jfieldID attrs_st_uid;

static jfieldID attrs_st_gid;

static jfieldID attrs_st_size;

static jfieldID attrs_st_atime;

static jfieldID attrs_st_mtime;

static jfieldID attrs_st_ctime;

static jfieldID attrs_f_frsize;

static jfieldID attrs_f_blocks;

static jfieldID attrs_f_bfree;

static jfieldID attrs_f_bavail;

static jfieldID entry_name;

static jfieldID entry_dir;

static jfieldID entry_fstype;

static jfieldID entry_options;

static jfieldID entry_dev;

/**

 */

typedef int openat64_func(int, const char *, int, ...);

typedef int fstatat64_func(int, const char *, struct stat64 *, int);

typedef int unlinkat_func(int, const char*, int);

typedef int renameat_func(int, const char*, int, const char*);

typedef int futimesat_func(int, const char *, const struct timeval *);

static openat64_func* my_openat64_func = NULL;

static fstatat64_func* my_fstatat64_func = NULL;

static unlinkat_func* my_unlinkat_func = NULL;

static renameat_func* my_renameat_func = NULL;

static futimesat_func* my_futimesat_func = NULL;

/**

 * fstatat missing from glibc on Linux. Temporary workaround

 * for x86/x64.

 */

#if defined(__linux__) && defined(__i386)

#define FSTATAT64_SYSCALL_AVAILABLE

static int fstatat64_wrapper(int dfd, const char *path,

                             struct stat64 *statbuf, int flag)

{

    #ifndef __NR_fstatat64

    #define __NR_fstatat64  300

    #endif

    return syscall(__NR_fstatat64, dfd, path, statbuf, flag);

}

#endif

#if defined(__linux__) && defined(__x86_64__)

#define FSTATAT64_SYSCALL_AVAILABLE

static int fstatat64_wrapper(int dfd, const char *path,

                             struct stat64 *statbuf, int flag)

{

    #ifndef __NR_newfstatat

    #define __NR_newfstatat  262

    #endif

    return syscall(__NR_newfstatat, dfd, path, statbuf, flag);

}

#endif

/**

 * Call this to throw an internal UnixException when a system/library

 * call fails

 */

static void throwUnixException(JNIEnv* env, int errnum) {

    jobject x = JNU_NewObjectByName(env, "sun/nio/fs/UnixException",

        "(I)V", errnum);

    if (x != NULL) {

        (*env)->Throw(env, x);

    }

}

/**

 * Initialization

 */

JNIEXPORT jint JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_init(JNIEnv* env, jclass this)

{

    jint flags = 0;

    jclass clazz;

    clazz = (*env)->FindClass(env, "sun/nio/fs/UnixFileAttributes");

    if (clazz == NULL) {

        return 0;

    }

    attrs_st_mode = (*env)->GetFieldID(env, clazz, "st_mode", "I");

    attrs_st_ino = (*env)->GetFieldID(env, clazz, "st_ino", "J");

    attrs_st_dev = (*env)->GetFieldID(env, clazz, "st_dev", "J");

    attrs_st_rdev = (*env)->GetFieldID(env, clazz, "st_rdev", "J");

    attrs_st_nlink = (*env)->GetFieldID(env, clazz, "st_nlink", "I");

    attrs_st_uid = (*env)->GetFieldID(env, clazz, "st_uid", "I");

    attrs_st_gid = (*env)->GetFieldID(env, clazz, "st_gid", "I");

    attrs_st_size = (*env)->GetFieldID(env, clazz, "st_size", "J");

    attrs_st_atime = (*env)->GetFieldID(env, clazz, "st_atime", "J");

    attrs_st_mtime = (*env)->GetFieldID(env, clazz, "st_mtime", "J");

    attrs_st_ctime = (*env)->GetFieldID(env, clazz, "st_ctime", "J");

    clazz = (*env)->FindClass(env, "sun/nio/fs/UnixFileStoreAttributes");

    if (clazz == NULL) {

        return 0;

    }

    attrs_f_frsize = (*env)->GetFieldID(env, clazz, "f_frsize", "J");

    attrs_f_blocks = (*env)->GetFieldID(env, clazz, "f_blocks", "J");

    attrs_f_bfree = (*env)->GetFieldID(env, clazz, "f_bfree", "J");

    attrs_f_bavail = (*env)->GetFieldID(env, clazz, "f_bavail", "J");

    clazz = (*env)->FindClass(env, "sun/nio/fs/UnixMountEntry");

    if (clazz == NULL) {

        return 0;

    }

    entry_name = (*env)->GetFieldID(env, clazz, "name", "[B");

    entry_dir = (*env)->GetFieldID(env, clazz, "dir", "[B");

    entry_fstype = (*env)->GetFieldID(env, clazz, "fstype", "[B");

    entry_options = (*env)->GetFieldID(env, clazz, "opts", "[B");

    entry_dev = (*env)->GetFieldID(env, clazz, "dev", "J");

#if defined(__solaris__) && defined(_LP64)

    /* Solaris 64-bit does not have openat64/fstatat64 */

    my_openat64_func = (openat64_func*)dlsym(RTLD_DEFAULT, "openat");

    my_fstatat64_func = (fstatat64_func*)dlsym(RTLD_DEFAULT, "fstatat");

#else

    my_openat64_func = (openat64_func*) dlsym(RTLD_DEFAULT, "openat64");

    my_fstatat64_func = (fstatat64_func*) dlsym(RTLD_DEFAULT, "fstatat64");

#endif

    my_unlinkat_func = (unlinkat_func*) dlsym(RTLD_DEFAULT, "unlinkat");

    my_renameat_func = (renameat_func*) dlsym(RTLD_DEFAULT, "renameat");

    my_futimesat_func = (futimesat_func*) dlsym(RTLD_DEFAULT, "futimesat");

#if defined(FSTATAT64_SYSCALL_AVAILABLE)

    /* fstatat64 missing from glibc */

    if (my_fstatat64_func == NULL)

        my_fstatat64_func = (fstatat64_func*)&fstatat64_wrapper;

#endif

    if (my_openat64_func != NULL &&  my_fstatat64_func != NULL &&

        my_unlinkat_func != NULL && my_renameat_func != NULL &&

    {

        flags |= sun_nio_fs_UnixNativeDispatcher_HAS_AT_SYSCALLS;

    }

    return flags;

}

JNIEXPORT jbyteArray JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_getcwd(JNIEnv* env, jclass this) {

    jbyteArray result = NULL;

    char buf[PATH_MAX+1];

    /* EINTR not listed as a possible error */

    char* cwd = getcwd(buf, sizeof(buf));

    if (cwd == NULL) {

        throwUnixException(env, errno);

    } else {

        jsize len = (jsize)strlen(buf);

        result = (*env)->NewByteArray(env, len);

        if (result != NULL) {

            (*env)->SetByteArrayRegion(env, result, 0, len, (jbyte*)buf);

        }

    }

    return result;

}

JNIEXPORT jbyteArray

Java_sun_nio_fs_UnixNativeDispatcher_strerror(JNIEnv* env, jclass this, jint error)

{

    char* msg;

    jsize len;

    jbyteArray bytes;

    msg = strerror((int)error);

    len = strlen(msg);

    bytes = (*env)->NewByteArray(env, len);

    if (bytes != NULL) {

        (*env)->SetByteArrayRegion(env, bytes, 0, len, (jbyte*)msg);

    }

    return bytes;

}

JNIEXPORT jint

Java_sun_nio_fs_UnixNativeDispatcher_dup(JNIEnv* env, jclass this, jint fd) {

    int res = -1;

    RESTARTABLE(dup((int)fd), res);

    if (fd == -1) {

        throwUnixException(env, errno);

    }

    return (jint)res;

}

JNIEXPORT jlong JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_fopen0(JNIEnv* env, jclass this,

    jlong pathAddress, jlong modeAddress)

{

    FILE* fp = NULL;

    const char* path = (const char*)jlong_to_ptr(pathAddress);

    const char* mode = (const char*)jlong_to_ptr(modeAddress);

    do {

        fp = fopen(path, mode);

    } while (fp == NULL && errno == EINTR);

    if (fp == NULL) {

        throwUnixException(env, errno);

    }

    return ptr_to_jlong(fp);

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_fclose(JNIEnv* env, jclass this, jlong stream)

{

    int res;

    FILE* fp = jlong_to_ptr(stream);

    do {

        res = fclose(fp);

    } while (res == EOF && errno == EINTR);

    if (res == EOF) {

        throwUnixException(env, errno);

    }

}

JNIEXPORT jint JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_open0(JNIEnv* env, jclass this,

    jlong pathAddress, jint oflags, jint mode)

{

    jint fd;

    const char* path = (const char*)jlong_to_ptr(pathAddress);

    RESTARTABLE(open64(path, (int)oflags, (mode_t)mode), fd);

    if (fd == -1) {

        throwUnixException(env, errno);

    }

    return fd;

}

JNIEXPORT jint JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_openat0(JNIEnv* env, jclass this, jint dfd,

    jlong pathAddress, jint oflags, jint mode)

{

    jint fd;

    const char* path = (const char*)jlong_to_ptr(pathAddress);

    if (my_openat64_func == NULL) {

        JNU_ThrowInternalError(env, "should not reach here");

        return -1;

    }

    RESTARTABLE((*my_openat64_func)(dfd, path, (int)oflags, (mode_t)mode), fd);

    if (fd == -1) {

        throwUnixException(env, errno);

    }

    return fd;

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_close(JNIEnv* env, jclass this, jint fd) {

    int err;

    /* TDB - need to decide if EIO and other errors should cause exception */

    RESTARTABLE(close((int)fd), err);

}

JNIEXPORT jint JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_read(JNIEnv* env, jclass this, jint fd,

    jlong address, jint nbytes)

{

    ssize_t n;

    void* bufp = jlong_to_ptr(address);

    RESTARTABLE(read((int)fd, bufp, (size_t)nbytes), n);

    if (n == -1) {

        throwUnixException(env, errno);

    }

    return (jint)n;

}

JNIEXPORT jint JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_write(JNIEnv* env, jclass this, jint fd,

    jlong address, jint nbytes)

{

    ssize_t n;

    void* bufp = jlong_to_ptr(address);

    RESTARTABLE(write((int)fd, bufp, (size_t)nbytes), n);

    if (n == -1) {

        throwUnixException(env, errno);

    }

    return (jint)n;

}

/**

 * Copy stat64 members into sun.nio.fs.UnixFileAttributes

 */

static void prepAttributes(JNIEnv* env, struct stat64* buf, jobject attrs) {

    (*env)->SetIntField(env, attrs, attrs_st_mode, (jint)buf->st_mode);

    (*env)->SetLongField(env, attrs, attrs_st_ino, (jlong)buf->st_ino);

    (*env)->SetLongField(env, attrs, attrs_st_dev, (jlong)buf->st_dev);

    (*env)->SetLongField(env, attrs, attrs_st_rdev, (jlong)buf->st_rdev);

    (*env)->SetIntField(env, attrs, attrs_st_nlink, (jint)buf->st_nlink);

    (*env)->SetIntField(env, attrs, attrs_st_uid, (jint)buf->st_uid);

    (*env)->SetIntField(env, attrs, attrs_st_gid, (jint)buf->st_gid);

    (*env)->SetLongField(env, attrs, attrs_st_size, (jlong)buf->st_size);

    (*env)->SetLongField(env, attrs, attrs_st_atime, (jlong)buf->st_atime);

    (*env)->SetLongField(env, attrs, attrs_st_mtime, (jlong)buf->st_mtime);

    (*env)->SetLongField(env, attrs, attrs_st_ctime, (jlong)buf->st_ctime);

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_stat0(JNIEnv* env, jclass this,

    jlong pathAddress, jobject attrs)

{

    int err;

    struct stat64 buf;

    const char* path = (const char*)jlong_to_ptr(pathAddress);

    RESTARTABLE(stat64(path, &buf), err);

    if (err == -1) {

        throwUnixException(env, errno);

    } else {

        prepAttributes(env, &buf, attrs);

    }

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_lstat0(JNIEnv* env, jclass this,

    jlong pathAddress, jobject attrs)

{

    int err;

    struct stat64 buf;

    const char* path = (const char*)jlong_to_ptr(pathAddress);

    RESTARTABLE(lstat64(path, &buf), err);

    if (err == -1) {

        throwUnixException(env, errno);

    } else {

        prepAttributes(env, &buf, attrs);

    }

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_fstat(JNIEnv* env, jclass this, jint fd,

    jobject attrs)

{

    int err;

    struct stat64 buf;

    RESTARTABLE(fstat64((int)fd, &buf), err);

    if (err == -1) {

        throwUnixException(env, errno);

    } else {

        prepAttributes(env, &buf, attrs);

    }

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_fstatat0(JNIEnv* env, jclass this, jint dfd,

    jlong pathAddress, jint flag, jobject attrs)

{

    int err;

    struct stat64 buf;

    const char* path = (const char*)jlong_to_ptr(pathAddress);

    if (my_fstatat64_func == NULL) {

        JNU_ThrowInternalError(env, "should not reach here");

        return;

    }

    RESTARTABLE((*my_fstatat64_func)((int)dfd, path, &buf, (int)flag), err);

    if (err == -1) {

        throwUnixException(env, errno);

    } else {

        prepAttributes(env, &buf, attrs);

    }

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_chmod0(JNIEnv* env, jclass this,

    jlong pathAddress, jint mode)

{

    int err;

    const char* path = (const char*)jlong_to_ptr(pathAddress);

    RESTARTABLE(chmod(path, (mode_t)mode), err);

    if (err == -1) {

        throwUnixException(env, errno);

    }

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_fchmod(JNIEnv* env, jclass this, jint filedes,

    jint mode)

{

    int err;

    RESTARTABLE(fchmod((int)filedes, (mode_t)mode), err);

    if (err == -1) {

        throwUnixException(env, errno);

    }

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_chown0(JNIEnv* env, jclass this,

    jlong pathAddress, jint uid, jint gid)

{

    int err;

    const char* path = (const char*)jlong_to_ptr(pathAddress);

    RESTARTABLE(chown(path, (uid_t)uid, (gid_t)gid), err);

    if (err == -1) {

        throwUnixException(env, errno);

    }

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_lchown0(JNIEnv* env, jclass this, jlong pathAddress, jint uid, jint gid)

{

    int err;

    const char* path = (const char*)jlong_to_ptr(pathAddress);

    RESTARTABLE(lchown(path, (uid_t)uid, (gid_t)gid), err);

    if (err == -1) {

        throwUnixException(env, errno);

    }

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_fchown(JNIEnv* env, jclass this, jint filedes, jint uid, jint gid)

{

    int err;

    RESTARTABLE(fchown(filedes, (uid_t)uid, (gid_t)gid), err);

    if (err == -1) {

        throwUnixException(env, errno);

    }

}

JNIEXPORT void JNICALL

Java_sun_nio_fs_UnixNativeDispatcher_utimes0(JNIEnv* env, jclass this,

    jlong pathAddress, jlong accessTime, jlong modificationTime)

{

    int err;

    struct timeval times[2];

    const char* path = (const char*)jlong_to_ptr(pathAddress);

    times[0].tv_sec = accessTime / 1000000;

    times[0].tv_usec = accessTime % 1000000;

    times[1].tv_sec = modificationTime / 1000000;

    times[1].tv_usec = modificationTime % 1000000;