From 1f06d99a0f563fdc32e9c00392f9875ba3009593 Mon Sep 17 00:00:00 2001 From: Martin Hurton Date: Mon, 31 Aug 2009 13:11:46 +0200 Subject: Add Java binding --- java/Socket.cpp | 276 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 276 insertions(+) create mode 100644 java/Socket.cpp (limited to 'java/Socket.cpp') diff --git a/java/Socket.cpp b/java/Socket.cpp new file mode 100644 index 0000000..eb3c1e3 --- /dev/null +++ b/java/Socket.cpp @@ -0,0 +1,276 @@ +/* + Copyright (c) 2007-2009 FastMQ Inc. + + This file is part of 0MQ. + + 0MQ is free software; you can redistribute it and/or modify it under + the terms of the Lesser GNU General Public License as published by + the Free Software Foundation; either version 3 of the License, or + (at your option) any later version. + + 0MQ 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 + Lesser GNU General Public License for more details. + + You should have received a copy of the Lesser GNU General Public License + along with this program. If not, see . +*/ + +#include +#include +#include +#include + +#include "zmq.h" +#include "org_zmq_Socket.h" + +static jfieldID socket_handle_fid = NULL; +static jclass msg_class = NULL; +static jmethodID msg_constructor; +static jmethodID get_msg_handle_mid = NULL; +static jmethodID create_socket_mid = NULL; + +static void +raise_exception (JNIEnv *env, int err) +{ + // Get exception class. + jclass exception_class = env->FindClass ("java/lang/Exception"); + assert (exception_class); + + // Get text description of the exception. + const char *err_msg = strerror (err); + + // Raise the exception. + int rc = env->ThrowNew (exception_class, err_msg); + assert (rc == 0); + + // Free the local ref. + env->DeleteLocalRef (exception_class); +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_construct (JNIEnv *env, jobject obj, jobject context, + jint type) +{ + if (socket_handle_fid == NULL) { + jclass cls = env->GetObjectClass (obj); + assert (cls); + socket_handle_fid = env->GetFieldID (cls, "socketHandle", "J"); + assert (socket_handle_fid); + env->DeleteLocalRef (cls); + } + + if (create_socket_mid == NULL) { + jclass cls = env->FindClass ("org/zmq/Context"); + assert (cls); + create_socket_mid = env->GetMethodID (cls, "createSocket", "(I)J"); + assert (create_socket_mid); + env->DeleteLocalRef (cls); + } + + if (msg_class == NULL) { + jclass cls = env->FindClass ("org/zmq/Message"); + assert (cls); + + msg_constructor = env->GetMethodID (cls, "", "()V"); + assert (msg_constructor); + + get_msg_handle_mid = env->GetMethodID (cls, "getMsgHandle", "()J"); + assert (get_msg_handle_mid); + + msg_class = (jclass) env->NewGlobalRef (cls); + assert (msg_class); + env->DeleteLocalRef (cls); + } + + void *s = (void *) env->CallLongMethod (context, create_socket_mid, type); + if (env->ExceptionCheck ()) + return; + + env->SetLongField (obj, socket_handle_fid, (jlong) s); +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_finalize (JNIEnv *env, jobject obj) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + int rc = zmq_close (s); + assert (rc == 0); +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_setHwm (JNIEnv *env, jobject obj, jlong hwm) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + int rc = zmq_setsockopt (s, ZMQ_HWM, &hwm, sizeof hwm); + if (rc == -1) + raise_exception (env, errno); +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_setLwm (JNIEnv *env, jobject obj, jlong lwm) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + + int rc = zmq_setsockopt (s, ZMQ_LWM, &lwm, sizeof lwm); + if (rc == -1) + raise_exception (env, errno); +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_setSwap (JNIEnv *env, jobject obj, jlong swap_size) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + + int rc = zmq_setsockopt (s, ZMQ_SWAP, &swap_size, sizeof swap_size); + if (rc == -1) + raise_exception (env, errno); +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_setMask (JNIEnv *env, jobject obj, jlong mask) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + + int rc = zmq_setsockopt (s, ZMQ_MASK, &mask, sizeof mask); + if (rc == -1) + raise_exception (env, errno); +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_setAffinity (JNIEnv *env, jobject obj, jlong affinity) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + + int rc = zmq_setsockopt (s, ZMQ_AFFINITY, &affinity, sizeof affinity); + if (rc == -1) + raise_exception (env, errno); +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_setIdentity (JNIEnv *env, jobject obj, jstring identity) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + + if (identity == NULL) { + raise_exception (env, EINVAL); + return; + } + + const char *c_identity = env->GetStringUTFChars (identity, NULL); + if (c_identity == NULL) + return; + + int rc = zmq_setsockopt (s, ZMQ_IDENTITY, c_identity, sizeof c_identity); + env->ReleaseStringUTFChars (identity, c_identity); + + if (rc == -1) + raise_exception (env, errno); +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_bind (JNIEnv *env, jobject obj, jstring addr) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + + if (addr == NULL) { + raise_exception (env, EINVAL); + return; + } + + const char *c_addr = env->GetStringUTFChars (addr, NULL); + if (c_addr == NULL) + return; + + int rc = zmq_bind (s, c_addr); + env->ReleaseStringUTFChars (addr, c_addr); + + if (rc == -1) + raise_exception (env, errno); +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_connect (JNIEnv *env, jobject obj, jstring addr) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + + if (addr == NULL) { + raise_exception (env, EINVAL); + return; + } + + const char *c_addr = env->GetStringUTFChars (addr, NULL); + if (c_addr == NULL) + return; + + int rc = zmq_connect (s, c_addr); + env->ReleaseStringUTFChars (addr, c_addr); + + if (rc == -1) + raise_exception (env, errno); +} + +JNIEXPORT jint JNICALL +Java_org_zmq_Socket_send (JNIEnv *env, jobject obj, jobject msg, jlong flags) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + + zmq_msg_t *zmq_msg = (zmq_msg_t *) + env->CallLongMethod (msg, get_msg_handle_mid); + + if (env->ExceptionCheck ()) + return -1; + + int rc = zmq_send (s, zmq_msg, (int) flags); + if (rc == -1) { + raise_exception (env, errno); + return -1; + } + + return rc; +} + +JNIEXPORT void JNICALL +Java_org_zmq_Socket_flush (JNIEnv *env, jobject obj) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + + zmq_flush (s); +} + +JNIEXPORT jobject JNICALL +Java_org_zmq_Socket_recv (JNIEnv *env, jobject obj, jlong flags) +{ + void *s = (void*) env->GetLongField (obj, socket_handle_fid); + assert (s); + + jobject msg = env->NewObject (msg_class, msg_constructor); + if (msg == NULL) + return NULL; + + zmq_msg_t *zmq_msg = (zmq_msg_t*) + env->CallLongMethod (msg, get_msg_handle_mid); + + if (env->ExceptionCheck ()) + return NULL; + + int rc = zmq_recv (s, zmq_msg, (int) flags); + if (rc == -1) { + raise_exception (env, errno); + return NULL; + } + + return msg; +} -- cgit v1.2.3