summaryrefslogtreecommitdiff
path: root/java/Socket.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'java/Socket.cpp')
-rw-r--r--java/Socket.cpp276
1 files changed, 276 insertions, 0 deletions
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 <http://www.gnu.org/licenses/>.
+*/
+
+#include <stdlib.h>
+#include <string.h>
+#include <assert.h>
+#include <errno.h>
+
+#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, "<init>", "()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;
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-02 23:38:57 +0000