/*

 * 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 "jni.h"

#include "jni_util.h"

#include "jvm.h"

#include "jlong.h"

#include "nio_util.h"

#include "sun_nio_ch_EPoll.h"

#include <dlfcn.h>

#include <unistd.h>

#include <sys/types.h>

#ifdef  __cplusplus

extern "C" {

#endif

/* epoll_wait(2) man page */

typedef union epoll_data {

    void *ptr;

    int fd;

    __uint32_t u32;

    __uint64_t u64;

} epoll_data_t;

struct epoll_event {

    __uint32_t events;  /* Epoll events */

    epoll_data_t data;  /* User data variable */

} __attribute__ ((__packed__));

#ifdef  __cplusplus

}

#endif

/*

 * epoll event notification is new in 2.6 kernel. As the offical build

 * platform for the JDK is on a 2.4-based distribution then we must

 * obtain the addresses of the epoll functions dynamically.

 */

typedef int (*epoll_create_t)(int size);

typedef int (*epoll_ctl_t)   (int epfd, int op, int fd, struct epoll_event *event);

typedef int (*epoll_wait_t)  (int epfd, struct epoll_event *events, int maxevents, int timeout);

static epoll_create_t epoll_create_func;

static epoll_ctl_t    epoll_ctl_func;

static epoll_wait_t   epoll_wait_func;

JNIEXPORT void JNICALL

Java_sun_nio_ch_EPoll_init(JNIEnv *env, jclass this)

{

    epoll_create_func = (epoll_create_t) dlsym(RTLD_DEFAULT, "epoll_create");

    epoll_ctl_func    = (epoll_ctl_t)    dlsym(RTLD_DEFAULT, "epoll_ctl");

    epoll_wait_func   = (epoll_wait_t)   dlsym(RTLD_DEFAULT, "epoll_wait");

    if ((epoll_create_func == NULL) || (epoll_ctl_func == NULL) ||

        (epoll_wait_func == NULL)) {

        JNU_ThrowInternalError(env, "unable to get address of epoll functions, pre-2.6 kernel?");

    }

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_EPoll_eventSize(JNIEnv* env, jclass this)

{

    return sizeof(struct epoll_event);

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_EPoll_eventsOffset(JNIEnv* env, jclass this)

{

    return offsetof(struct epoll_event, events);

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_EPoll_dataOffset(JNIEnv* env, jclass this)

{

    return offsetof(struct epoll_event, data);

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_EPoll_epollCreate(JNIEnv *env, jclass c) {

    /*

     * epoll_create expects a size as a hint to the kernel about how to

     * dimension internal structures. We can't predict the size in advance.

     */

    int epfd = (*epoll_create_func)(256);

    if (epfd < 0) {

       JNU_ThrowIOExceptionWithLastError(env, "epoll_create failed");

    }

    return epfd;

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_EPoll_epollCtl(JNIEnv *env, jclass c, jint epfd,

                                   jint opcode, jint fd, jint events)

{

    struct epoll_event event;

    int res;

    event.events = events;

    event.data.fd = fd;

    RESTARTABLE((*epoll_ctl_func)(epfd, (int)opcode, (int)fd, &event), res);

    return (res == 0) ? 0 : errno;

}

JNIEXPORT jint JNICALL

Java_sun_nio_ch_EPoll_epollWait(JNIEnv *env, jclass c,

                                    jint epfd, jlong address, jint numfds)

{

    struct epoll_event *events = jlong_to_ptr(address);

    int res;

    RESTARTABLE((*epoll_wait_func)(epfd, events, numfds, -1), res);

    if (res < 0) {

        JNU_ThrowIOExceptionWithLastError(env, "epoll_wait failed");

    }

    return res;

}

JNIEXPORT void JNICALL

Java_sun_nio_ch_EPoll_close0(JNIEnv *env, jclass c, jint epfd) {

    int res;

    RESTARTABLE(close(epfd), res);

}