From 1a655045951c29edda0b616640b9dac507cda164 Mon Sep 17 00:00:00 2001 From: malosek Date: Wed, 16 Sep 2009 17:06:59 +0200 Subject: ruby directory moved into bindings dir --- ruby/rbzmq.cpp | 277 --------------------------------------------------------- 1 file changed, 277 deletions(-) delete mode 100644 ruby/rbzmq.cpp (limited to 'ruby/rbzmq.cpp') diff --git a/ruby/rbzmq.cpp b/ruby/rbzmq.cpp deleted file mode 100644 index bf0d9bc..0000000 --- a/ruby/rbzmq.cpp +++ /dev/null @@ -1,277 +0,0 @@ -/* - 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 "../c/zmq.h" - -static void context_free (void *ctx) -{ - if (ctx) { - int rc = zmq_term (ctx); - assert (rc == 0); - } -} - -static VALUE context_alloc (VALUE class_) -{ - return rb_data_object_alloc (class_, NULL, 0, context_free); -} - -static VALUE context_initialize (VALUE self_, VALUE app_threads_, - VALUE io_threads_) -{ - assert (!DATA_PTR (self_)); - void *ctx = zmq_init (NUM2INT (app_threads_), NUM2INT (io_threads_)); - if (!ctx) { - rb_raise (rb_eRuntimeError, strerror (errno)); - return Qnil; - } - - DATA_PTR (self_) = (void*) ctx; - return self_; -} - -static void socket_free (void *s) -{ - if (s) { - int rc = zmq_close (s); - assert (rc == 0); - } -} - -static VALUE socket_alloc (VALUE class_) -{ - return rb_data_object_alloc (class_, NULL, 0, socket_free); -} - -static VALUE socket_initialize (VALUE self_, VALUE context_, VALUE type_) -{ - assert (!DATA_PTR (self_)); - - if (strcmp (rb_obj_classname (context_), "Context") != 0) { - rb_raise (rb_eArgError, "expected Context object"); - return Qnil; - } - - void *s = zmq_socket (DATA_PTR (context_), NUM2INT (type_)); - if (!s) { - rb_raise (rb_eRuntimeError, strerror (errno)); - return Qnil; - } - - DATA_PTR (self_) = (void*) s; - return self_; -} - - -static VALUE socket_setsockopt (VALUE self_, VALUE option_, - VALUE optval_) -{ - - int rc = 0; - - switch (NUM2INT (option_)) { - case ZMQ_HWM: - case ZMQ_LWM: - case ZMQ_SWAP: - case ZMQ_AFFINITY: - case ZMQ_RATE: - case ZMQ_RECOVERY_IVL: - case ZMQ_MCAST_LOOP: - { - long optval = FIX2LONG (optval_); - - // Forward the code to native 0MQ library. - rc = zmq_setsockopt (DATA_PTR (self_), NUM2INT (option_), - (void *) &optval, 4); - } - - break; - case ZMQ_IDENTITY: - case ZMQ_SUBSCRIBE: - case ZMQ_UNSUBSCRIBE: - - // Forward the code to native 0MQ library. - rc = zmq_setsockopt (DATA_PTR (self_), NUM2INT (option_), - (void *) StringValueCStr (optval_), RSTRING_LEN (optval_)); - break; - - default: - rc = -1; - errno = EINVAL; - } - - if (rc != 0) { - rb_raise (rb_eRuntimeError, strerror (errno)); - return Qnil; - } - - return self_; -} - - -static VALUE socket_bind (VALUE self_, VALUE addr_) -{ - assert (DATA_PTR (self_)); - - int rc = zmq_bind (DATA_PTR (self_), rb_string_value_cstr (&addr_)); - if (rc != 0) { - rb_raise (rb_eRuntimeError, strerror (errno)); - return Qnil; - } - - return Qnil; -} - -static VALUE socket_connect (VALUE self_, VALUE addr_) -{ - assert (DATA_PTR (self_)); - - int rc = zmq_connect (DATA_PTR (self_), rb_string_value_cstr (&addr_)); - if (rc != 0) { - rb_raise (rb_eRuntimeError, strerror (errno)); - return Qnil; - } - - return Qnil; -} - -static VALUE socket_send (VALUE self_, VALUE msg_, VALUE flags_) -{ - assert (DATA_PTR (self_)); - - Check_Type (msg_, T_STRING); - - zmq_msg_t msg; - int rc = zmq_msg_init_size (&msg, RSTRING_LEN (msg_)); - if (rc != 0) { - rb_raise (rb_eRuntimeError, strerror (errno)); - return Qnil; - } - memcpy (zmq_msg_data (&msg), RSTRING_PTR (msg_), RSTRING_LEN (msg_)); - - rc = zmq_send (DATA_PTR (self_), &msg, NUM2INT (flags_)); - if (rc != 0 && errno == EAGAIN) { - rc = zmq_msg_close (&msg); - assert (rc == 0); - return Qfalse; - } - - if (rc != 0) { - rb_raise (rb_eRuntimeError, strerror (errno)); - rc = zmq_msg_close (&msg); - assert (rc == 0); - return Qnil; - } - - rc = zmq_msg_close (&msg); - assert (rc == 0); - return Qtrue; -} - -static VALUE socket_flush (VALUE self_) -{ - assert (DATA_PTR (self_)); - - int rc = zmq_flush (DATA_PTR (self_)); - if (rc != 0) { - rb_raise (rb_eRuntimeError, strerror (errno)); - return Qnil; - } - - return Qnil; -} - -static VALUE socket_recv (VALUE self_, VALUE flags_) -{ - assert (DATA_PTR (self_)); - - zmq_msg_t msg; - int rc = zmq_msg_init (&msg); - assert (rc == 0); - - rc = zmq_recv (DATA_PTR (self_), &msg, NUM2INT (flags_)); - if (rc != 0 && errno == EAGAIN) { - rc = zmq_msg_close (&msg); - assert (rc == 0); - return Qnil; - } - - if (rc != 0) { - rb_raise (rb_eRuntimeError, strerror (errno)); - rc = zmq_msg_close (&msg); - assert (rc == 0); - return Qnil; - } - - VALUE message = rb_str_new ((char*) zmq_msg_data (&msg), - zmq_msg_size (&msg)); - rc = zmq_msg_close (&msg); - assert (rc == 0); - return message; -} - -extern "C" void Init_librbzmq () -{ - VALUE context_type = rb_define_class ("Context", rb_cObject); - rb_define_alloc_func (context_type, context_alloc); - rb_define_method (context_type, "initialize", - (VALUE(*)(...)) context_initialize, 2); - - VALUE socket_type = rb_define_class ("Socket", rb_cObject); - rb_define_alloc_func (socket_type, socket_alloc); - rb_define_method (socket_type, "initialize", - (VALUE(*)(...)) socket_initialize, 2); - rb_define_method (socket_type, "setsockopt", - (VALUE(*)(...)) socket_setsockopt, 2); - rb_define_method (socket_type, "bind", - (VALUE(*)(...)) socket_bind, 1); - rb_define_method (socket_type, "connect", - (VALUE(*)(...)) socket_connect, 1); - rb_define_method (socket_type, "send", - (VALUE(*)(...)) socket_send, 2); - rb_define_method (socket_type, "flush", - (VALUE(*)(...)) socket_flush, 0); - rb_define_method (socket_type, "recv", - (VALUE(*)(...)) socket_recv, 1); - - rb_define_global_const ("HWM", INT2NUM (ZMQ_HWM)); - rb_define_global_const ("LWM", INT2NUM (ZMQ_LWM)); - rb_define_global_const ("SWAP", INT2NUM (ZMQ_SWAP)); - rb_define_global_const ("AFFINITY", INT2NUM (ZMQ_AFFINITY)); - rb_define_global_const ("IDENTITY", INT2NUM (ZMQ_IDENTITY)); - rb_define_global_const ("SUBSCRIBE", INT2NUM (ZMQ_SUBSCRIBE)); - rb_define_global_const ("UNSUBSCRIBE", INT2NUM (ZMQ_UNSUBSCRIBE)); - rb_define_global_const ("RATE", INT2NUM (ZMQ_RATE)); - rb_define_global_const ("RECOVERY_IVL", INT2NUM (ZMQ_RECOVERY_IVL)); - rb_define_global_const ("MCAST_LOOP", INT2NUM (ZMQ_MCAST_LOOP)); - - rb_define_global_const ("NOBLOCK", INT2NUM (ZMQ_NOBLOCK)); - rb_define_global_const ("NOFLUSH", INT2NUM (ZMQ_NOFLUSH)); - - rb_define_global_const ("P2P", INT2NUM (ZMQ_P2P)); - rb_define_global_const ("SUB", INT2NUM (ZMQ_SUB)); - rb_define_global_const ("PUB", INT2NUM (ZMQ_PUB)); - rb_define_global_const ("REQ", INT2NUM (ZMQ_REQ)); - rb_define_global_const ("REP", INT2NUM (ZMQ_REP)); -} -- cgit v1.2.3