Add send_msg()/recv_msg() functions and zmq_msg_t object. Update FFI-based bindings to use latest LuaNativeObjects FFI support.
Robert G. Jakabosky
15 years ago
parent
ed462fe9b8
commit
d0a71ccbae
@ -0,0 +1,160 @@
|
|||||||
|
-- Copyright (c) 2010 by Robert G. Jakabosky <bobby@sharedrealm.com>
|
||||||
|
--
|
||||||
|
-- Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
-- of this software and associated documentation files (the "Software"), to deal
|
||||||
|
-- in the Software without restriction, including without limitation the rights
|
||||||
|
-- to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
-- copies of the Software, and to permit persons to whom the Software is
|
||||||
|
-- furnished to do so, subject to the following conditions:
|
||||||
|
--
|
||||||
|
-- The above copyright notice and this permission notice shall be included in
|
||||||
|
-- all copies or substantial portions of the Software.
|
||||||
|
--
|
||||||
|
-- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
-- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
-- FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
-- AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
-- LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
-- OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
-- THE SOFTWARE.
|
||||||
|
|
||||||
|
object "zmq_msg_t" {
|
||||||
|
-- store the `zmq_msg_t` structure in Lua userdata object
|
||||||
|
userdata_type = "embed",
|
||||||
|
--
|
||||||
|
-- Define zmq_msq_t type & function API for FFI
|
||||||
|
--
|
||||||
|
ffi_cdef[[
|
||||||
|
|
||||||
|
typedef struct zmq_msg_t
|
||||||
|
{
|
||||||
|
void *content;
|
||||||
|
unsigned char flags;
|
||||||
|
unsigned char vsm_size;
|
||||||
|
unsigned char vsm_data [30]; /* that '30' is from 'MAX_VSM_SIZE' */
|
||||||
|
} zmq_msg_t;
|
||||||
|
|
||||||
|
typedef void (zmq_free_fn) (void *data, void *hint);
|
||||||
|
|
||||||
|
int zmq_msg_init (zmq_msg_t *msg);
|
||||||
|
int zmq_msg_init_size (zmq_msg_t *msg, size_t size);
|
||||||
|
int zmq_msg_init_data (zmq_msg_t *msg, void *data, size_t size, zmq_free_fn *ffn, void *hint);
|
||||||
|
|
||||||
|
]],
|
||||||
|
constructor "init" {
|
||||||
|
var_out{ "ZMQ_Error", "err" },
|
||||||
|
c_source[[
|
||||||
|
zmq_msg_t tmp;
|
||||||
|
${this} = &tmp;
|
||||||
|
${err} = zmq_msg_init(${this});
|
||||||
|
]],
|
||||||
|
},
|
||||||
|
constructor "init_size" {
|
||||||
|
var_in{ "size_t", "size" },
|
||||||
|
var_out{ "ZMQ_Error", "err" },
|
||||||
|
c_source[[
|
||||||
|
zmq_msg_t tmp;
|
||||||
|
${this} = &tmp;
|
||||||
|
${err} = zmq_msg_init_size(${this}, ${size});
|
||||||
|
]],
|
||||||
|
},
|
||||||
|
constructor "init_data" {
|
||||||
|
var_in{ "const char *", "data" },
|
||||||
|
var_out{ "ZMQ_Error", "err" },
|
||||||
|
c_source[[
|
||||||
|
zmq_msg_t tmp;
|
||||||
|
${this} = &tmp;
|
||||||
|
${err} = zmq_msg_init_size(${this}, ${data_len});
|
||||||
|
if(0 == ${err}) {
|
||||||
|
/* fill message */
|
||||||
|
memcpy(zmq_msg_data(${this}), ${data}, ${data_len});
|
||||||
|
}
|
||||||
|
]],
|
||||||
|
},
|
||||||
|
destructor {
|
||||||
|
c_method_call "ZMQ_Error" "zmq_msg_close" {}
|
||||||
|
},
|
||||||
|
method "close" {
|
||||||
|
c_method_call "ZMQ_Error" "zmq_msg_close" {}
|
||||||
|
},
|
||||||
|
method "move" {
|
||||||
|
c_method_call "ZMQ_Error" "zmq_msg_move" { "zmq_msg_t *", "src" }
|
||||||
|
},
|
||||||
|
method "copy" {
|
||||||
|
c_method_call "ZMQ_Error" "zmq_msg_copy" { "zmq_msg_t *", "src" }
|
||||||
|
},
|
||||||
|
method "set_data" {
|
||||||
|
var_in{ "const char *", "data" },
|
||||||
|
var_out{ "ZMQ_Error", "err" },
|
||||||
|
c_source[[
|
||||||
|
/* check message data size. */
|
||||||
|
if(zmq_msg_size(${this}) != ${data_len}) {
|
||||||
|
/* need to resize message. */
|
||||||
|
zmq_msg_close(${this}); /* close old message, to free old data. */
|
||||||
|
${err} = zmq_msg_init_size(${this}, ${data_len}); /* re-initialize message. */
|
||||||
|
if(0 != ${err}) {
|
||||||
|
luaL_error(L, "set_data() failed: %s", get_zmq_strerror());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/* copy data into message */
|
||||||
|
memcpy(zmq_msg_data(${this}), ${data}, ${data_len});
|
||||||
|
]],
|
||||||
|
ffi_source[[
|
||||||
|
-- check message data size.
|
||||||
|
if (C.zmq_msg_size(${this}) ~= ${data_len}) then
|
||||||
|
-- need to resize message.
|
||||||
|
C.zmq_msg_close(${this}); -- close old message, to free old data.
|
||||||
|
${err} = C.zmq_msg_init_size(${this}, ${data_len}); -- re-initialize message.
|
||||||
|
if (0 ~= ${err}) then
|
||||||
|
error("set_data() failed: " .. get_zmq_strerror());
|
||||||
|
end
|
||||||
|
end
|
||||||
|
-- copy data into message
|
||||||
|
ffi.copy(C.zmq_msg_data(${this}), ${data}, ${data_len});
|
||||||
|
]],
|
||||||
|
},
|
||||||
|
method "data" {
|
||||||
|
c_method_call "void *" "zmq_msg_data" {}
|
||||||
|
},
|
||||||
|
method "set_size" {
|
||||||
|
var_in{ "size_t", "size" },
|
||||||
|
var_out{ "ZMQ_Error", "err" },
|
||||||
|
c_source[[
|
||||||
|
/* check message data size. */
|
||||||
|
if(zmq_msg_size(${this}) != ${size}) {
|
||||||
|
/* need to resize message. */
|
||||||
|
zmq_msg_close(${this}); /* close old message, to free old data. */
|
||||||
|
${err} = zmq_msg_init_size(${this}, ${size}); /* re-initialize message. */
|
||||||
|
if(0 != ${err}) {
|
||||||
|
luaL_error(L, "set_size() failed: %s", get_zmq_strerror());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
]],
|
||||||
|
ffi_source[[
|
||||||
|
-- check message data size.
|
||||||
|
if (C.zmq_msg_size(${this}) ~= ${size}) then
|
||||||
|
-- need to resize message.
|
||||||
|
C.zmq_msg_close(${this}); -- close old message, to free old data.
|
||||||
|
${err} = C.zmq_msg_init_size(${this}, ${size}); -- re-initialize message.
|
||||||
|
if (0 ~= ${err}) then
|
||||||
|
error("set_size() failed: " .. get_zmq_strerror());
|
||||||
|
end
|
||||||
|
end
|
||||||
|
]],
|
||||||
|
},
|
||||||
|
method "size" {
|
||||||
|
c_method_call "size_t" "zmq_msg_size" {}
|
||||||
|
},
|
||||||
|
method "__tostring" {
|
||||||
|
var_out{ "const char *", "data", has_length = true },
|
||||||
|
c_source[[
|
||||||
|
${data} = zmq_msg_data(${this});
|
||||||
|
${data_len} = zmq_msg_size(${this});
|
||||||
|
]],
|
||||||
|
ffi_source[[
|
||||||
|
${data} = zmq_msg_data(${this});
|
||||||
|
${data_len} = zmq_msg_size(${this});
|
||||||
|
]],
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
File diff suppressed because it is too large
Load Diff
@ -1,347 +0,0 @@
|
|||||||
-- Copyright (c) 2010 by Robert G. Jakabosky <bobby@sharedrealm.com>
|
|
||||||
--
|
|
||||||
-- Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
||||||
-- of this software and associated documentation files (the "Software"), to deal
|
|
||||||
-- in the Software without restriction, including without limitation the rights
|
|
||||||
-- to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
||||||
-- copies of the Software, and to permit persons to whom the Software is
|
|
||||||
-- furnished to do so, subject to the following conditions:
|
|
||||||
--
|
|
||||||
-- The above copyright notice and this permission notice shall be included in
|
|
||||||
-- all copies or substantial portions of the Software.
|
|
||||||
--
|
|
||||||
-- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
||||||
-- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
||||||
-- FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
||||||
-- AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
||||||
-- LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
||||||
-- OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
||||||
-- THE SOFTWARE.
|
|
||||||
|
|
||||||
local zmq = ...
|
|
||||||
|
|
||||||
-- try loading luajit's ffi
|
|
||||||
local stat, ffi=pcall(require,"ffi")
|
|
||||||
if not stat then
|
|
||||||
print("No FFI module: ZMQ using standard Lua api interface.")
|
|
||||||
return
|
|
||||||
end
|
|
||||||
-- check if ffi is disabled.
|
|
||||||
if disable_ffi then
|
|
||||||
print("FFI disabled: ZMQ using standard Lua api interface.")
|
|
||||||
return
|
|
||||||
end
|
|
||||||
|
|
||||||
local setmetatable = setmetatable
|
|
||||||
local getmetatable = getmetatable
|
|
||||||
local print = print
|
|
||||||
local pairs = pairs
|
|
||||||
local error = error
|
|
||||||
local type = type
|
|
||||||
local assert = assert
|
|
||||||
local tostring = tostring
|
|
||||||
local tonumber = tonumber
|
|
||||||
local newproxy = newproxy
|
|
||||||
|
|
||||||
ffi.cdef[[
|
|
||||||
void zmq_version (int *major, int *minor, int *patch);
|
|
||||||
int zmq_errno ();
|
|
||||||
const char *zmq_strerror (int errnum);
|
|
||||||
|
|
||||||
typedef struct zmq_msg_t
|
|
||||||
{
|
|
||||||
void *content;
|
|
||||||
unsigned char flags;
|
|
||||||
unsigned char vsm_size;
|
|
||||||
unsigned char vsm_data [30];
|
|
||||||
} zmq_msg_t;
|
|
||||||
|
|
||||||
typedef void (zmq_free_fn) (void *data, void *hint);
|
|
||||||
|
|
||||||
int zmq_msg_init (zmq_msg_t *msg);
|
|
||||||
int zmq_msg_init_size (zmq_msg_t *msg, size_t size);
|
|
||||||
int zmq_msg_init_data (zmq_msg_t *msg, void *data, size_t size, zmq_free_fn *ffn, void *hint);
|
|
||||||
int zmq_msg_close (zmq_msg_t *msg);
|
|
||||||
int zmq_msg_move (zmq_msg_t *dest, zmq_msg_t *src);
|
|
||||||
int zmq_msg_copy (zmq_msg_t *dest, zmq_msg_t *src);
|
|
||||||
void *zmq_msg_data (zmq_msg_t *msg);
|
|
||||||
size_t zmq_msg_size (zmq_msg_t *msg);
|
|
||||||
|
|
||||||
void *zmq_init (int io_threads);
|
|
||||||
int zmq_term (void *context);
|
|
||||||
|
|
||||||
void *zmq_socket (void *context, int type);
|
|
||||||
int zmq_close (void *s);
|
|
||||||
int zmq_setsockopt (void *s, int option, const void *optval, size_t optvallen);
|
|
||||||
int zmq_getsockopt (void *s, int option, void *optval, size_t *optvallen);
|
|
||||||
int zmq_bind (void *s, const char *addr);
|
|
||||||
int zmq_connect (void *s, const char *addr);
|
|
||||||
int zmq_send (void *s, zmq_msg_t *msg, int flags);
|
|
||||||
int zmq_recv (void *s, zmq_msg_t *msg, int flags);
|
|
||||||
|
|
||||||
int zmq_device (int device, void * insocket, void* outsocket);
|
|
||||||
|
|
||||||
]]
|
|
||||||
|
|
||||||
local C = ffi.load"zmq"
|
|
||||||
|
|
||||||
-- simulate: module(...)
|
|
||||||
zmq._M = zmq
|
|
||||||
setfenv(1, zmq)
|
|
||||||
|
|
||||||
function version()
|
|
||||||
local major = ffi.new('int[1]',0)
|
|
||||||
local minor = ffi.new('int[1]',0)
|
|
||||||
local patch = ffi.new('int[1]',0)
|
|
||||||
C.zmq_version(major, minor, patch)
|
|
||||||
return {major[0], minor[0], patch[0]}
|
|
||||||
end
|
|
||||||
|
|
||||||
local function zmq_error()
|
|
||||||
local errno = C.zmq_errno()
|
|
||||||
local err = ffi.string(C.zmq_strerror(errno))
|
|
||||||
if err == "Resource temporarily unavailable" then err = "timeout" end
|
|
||||||
if err == "Context was terminated" then err = "closed" end
|
|
||||||
return nil, err
|
|
||||||
end
|
|
||||||
|
|
||||||
--
|
|
||||||
-- ZMQ socket
|
|
||||||
--
|
|
||||||
local sock_mt = {}
|
|
||||||
sock_mt.__index = sock_mt
|
|
||||||
|
|
||||||
function sock_mt:close()
|
|
||||||
-- get the true self
|
|
||||||
self=getmetatable(self)
|
|
||||||
local sock = self.sock
|
|
||||||
-- make sure socket is still valid.
|
|
||||||
if not sock then return end
|
|
||||||
-- mark this socket as closed.
|
|
||||||
self.sock = nil
|
|
||||||
-- close zmq socket.
|
|
||||||
if C.zmq_close(sock) ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
local function sock__gc(self)
|
|
||||||
self:close()
|
|
||||||
end
|
|
||||||
|
|
||||||
local option_types = {
|
|
||||||
[zmq.HWM] = 'uint64_t[1]',
|
|
||||||
[zmq.SWAP] = 'int64_t[1]',
|
|
||||||
[zmq.AFFINITY] = 'uint64_t[1]',
|
|
||||||
[zmq.IDENTITY] = 'string',
|
|
||||||
[zmq.SUBSCRIBE] = 'string',
|
|
||||||
[zmq.UNSUBSCRIBE] = 'string',
|
|
||||||
[zmq.RATE] = 'int64_t[1]',
|
|
||||||
[zmq.RECOVERY_IVL] = 'int64_t[1]',
|
|
||||||
[zmq.MCAST_LOOP] = 'int64_t[1]',
|
|
||||||
[zmq.SNDBUF] = 'uint64_t[1]',
|
|
||||||
[zmq.RCVBUF] = 'uint64_t[1]',
|
|
||||||
[zmq.RCVMORE] = 'int64_t[1]',
|
|
||||||
[zmq.FD] = 'int[1]',
|
|
||||||
[zmq.EVENTS] = 'uint32_t[1]',
|
|
||||||
[zmq.TYPE] = 'int[1]',
|
|
||||||
[zmq.LINGER] = 'int[1]',
|
|
||||||
[zmq.RECONNECT_IVL] = 'int[1]',
|
|
||||||
[zmq.BACKLOG] = 'int[1]',
|
|
||||||
}
|
|
||||||
local option_len = {}
|
|
||||||
local option_tmps = {}
|
|
||||||
for k,v in pairs(option_types) do
|
|
||||||
if v ~= 'string' then
|
|
||||||
option_len[k] = ffi.sizeof(v)
|
|
||||||
option_tmps[k] = ffi.new(v, 0)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
function sock_mt:setopt(opt, opt_val)
|
|
||||||
local ctype = option_types[opt]
|
|
||||||
local val_len = 0
|
|
||||||
if ctype == 'string' then
|
|
||||||
--val = ffi.cast('void *', tostring(val))
|
|
||||||
val = tostring(opt_val)
|
|
||||||
val_len = #val
|
|
||||||
else
|
|
||||||
val = option_tmps[opt]
|
|
||||||
val[0] = opt_val
|
|
||||||
val_len = option_len[opt]
|
|
||||||
end
|
|
||||||
local ret = C.zmq_setsockopt(self.sock, opt, val, val_len)
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
local tmp_val_len = ffi.new('size_t[1]', 4)
|
|
||||||
function sock_mt:getopt(opt)
|
|
||||||
local ctype = option_types[opt]
|
|
||||||
local val
|
|
||||||
local val_len = tmp_val_len
|
|
||||||
if ctype == 'string' then
|
|
||||||
val_len[0] = 255
|
|
||||||
val = ffi.new('uint8_t[?]', val_len[0])
|
|
||||||
ffi.fill(val, val_len[0])
|
|
||||||
else
|
|
||||||
val = option_tmps[opt]
|
|
||||||
val[0] = 0
|
|
||||||
val_len[0] = option_len[opt]
|
|
||||||
end
|
|
||||||
local ret = C.zmq_getsockopt(self.sock, opt, val, val_len)
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
if ctype == 'string' then
|
|
||||||
val_len = val_len[0]
|
|
||||||
return ffi.string(val, val_len)
|
|
||||||
else
|
|
||||||
val = val[0]
|
|
||||||
end
|
|
||||||
return tonumber(val)
|
|
||||||
end
|
|
||||||
|
|
||||||
local tmp32 = ffi.new('uint32_t[1]', 0)
|
|
||||||
local tmp32_len = ffi.new('size_t[1]', 4)
|
|
||||||
function sock_mt:events()
|
|
||||||
local val = tmp32
|
|
||||||
local val_len = tmp32_len
|
|
||||||
val[0] = 0
|
|
||||||
val_len[0] = 4
|
|
||||||
local ret = C.zmq_getsockopt(self.sock, 15, val, val_len)
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return val[0]
|
|
||||||
end
|
|
||||||
|
|
||||||
function sock_mt:bind(addr)
|
|
||||||
local ret = C.zmq_bind(self.sock, addr)
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
function sock_mt:connect(addr)
|
|
||||||
local ret = C.zmq_connect(self.sock, addr)
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
local tmp_msg = ffi.new('zmq_msg_t')
|
|
||||||
function sock_mt:send(data, flags)
|
|
||||||
local msg = tmp_msg
|
|
||||||
local msg_len = #data
|
|
||||||
-- initialize message
|
|
||||||
if C.zmq_msg_init_size(msg, msg_len) < 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
-- copy data into message.
|
|
||||||
ffi.copy(C.zmq_msg_data(msg), data, msg_len)
|
|
||||||
|
|
||||||
-- send message
|
|
||||||
local ret = C.zmq_send(self.sock, msg, flags or 0)
|
|
||||||
-- close message before processing return code
|
|
||||||
C.zmq_msg_close(msg)
|
|
||||||
-- now process send return code
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
function sock_mt:recv(flags)
|
|
||||||
local msg = tmp_msg
|
|
||||||
-- initialize blank message.
|
|
||||||
if C.zmq_msg_init(msg) < 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
|
|
||||||
-- receive message
|
|
||||||
local ret = C.zmq_recv(self.sock, msg, flags or 0)
|
|
||||||
if ret ~= 0 then
|
|
||||||
local data, err = zmq_error()
|
|
||||||
C.zmq_msg_close(msg)
|
|
||||||
return data, err
|
|
||||||
end
|
|
||||||
local data = ffi.string(C.zmq_msg_data(msg), C.zmq_msg_size(msg))
|
|
||||||
-- close message
|
|
||||||
C.zmq_msg_close(msg)
|
|
||||||
return data
|
|
||||||
end
|
|
||||||
|
|
||||||
--
|
|
||||||
-- ZMQ context
|
|
||||||
--
|
|
||||||
local ctx_mt = {}
|
|
||||||
ctx_mt.__index = ctx_mt
|
|
||||||
|
|
||||||
function ctx_mt:term()
|
|
||||||
-- get the true self
|
|
||||||
self=getmetatable(self)
|
|
||||||
local ctx = self.ctx
|
|
||||||
self.ctx = nil
|
|
||||||
-- make sure context is valid.
|
|
||||||
if not ctx then return end
|
|
||||||
if C.zmq_term(ctx) ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
function ctx_mt:lightuserdata()
|
|
||||||
return self.ctx
|
|
||||||
end
|
|
||||||
|
|
||||||
function ctx_mt:socket(sock_type)
|
|
||||||
local sock = C.zmq_socket(self.ctx, sock_type)
|
|
||||||
if not sock then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
-- use a wrapper newproxy for __gc support
|
|
||||||
local self=newproxy(true)
|
|
||||||
local meta=getmetatable(self)
|
|
||||||
meta.__index = meta
|
|
||||||
meta.sock = sock
|
|
||||||
meta.__gc = sock__gc
|
|
||||||
setmetatable(meta, sock_mt)
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
|
|
||||||
local function ctx__gc(self)
|
|
||||||
if self.should_free then
|
|
||||||
self:term()
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
function init(io_threads)
|
|
||||||
local should_free = true
|
|
||||||
local ctx
|
|
||||||
print("ZMQ using FFI interface.")
|
|
||||||
if type(io_threads) == 'number' then
|
|
||||||
ctx = C.zmq_init(io_threads)
|
|
||||||
if not ctx then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
else
|
|
||||||
should_free = false
|
|
||||||
-- should be lightuserdata or cdata<void *>
|
|
||||||
ctx = io_threads
|
|
||||||
end
|
|
||||||
-- use a wrapper newproxy for __gc support
|
|
||||||
local self=newproxy(true)
|
|
||||||
local meta=getmetatable(self)
|
|
||||||
meta.__index = meta
|
|
||||||
meta.ctx = ctx
|
|
||||||
meta.should_free = should_free
|
|
||||||
meta.__gc = ctx__gc
|
|
||||||
setmetatable(meta, ctx_mt)
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
|
|
||||||
@ -1,347 +0,0 @@
|
|||||||
--[[
|
|
||||||
--
|
|
||||||
-- This is just an normal LuaJIT2 FFI based bindings for zmq.
|
|
||||||
-- It is only for testing and comparison.
|
|
||||||
--
|
|
||||||
--]]
|
|
||||||
|
|
||||||
local setmetatable = setmetatable
|
|
||||||
local print = print
|
|
||||||
local pairs = pairs
|
|
||||||
local error = error
|
|
||||||
local type = type
|
|
||||||
local assert = assert
|
|
||||||
local tostring = tostring
|
|
||||||
local tonumber = tonumber
|
|
||||||
|
|
||||||
local zmq = {
|
|
||||||
MAX_VSM_SIZE = 30,
|
|
||||||
|
|
||||||
-- message types
|
|
||||||
DELIMITER = 31,
|
|
||||||
VSM = 32,
|
|
||||||
|
|
||||||
-- message flags
|
|
||||||
MSG_MORE = 1,
|
|
||||||
MSG_SHARED = 128,
|
|
||||||
|
|
||||||
-- socket types
|
|
||||||
PAIR = 0,
|
|
||||||
PUB = 1,
|
|
||||||
SUB = 2,
|
|
||||||
REQ = 3,
|
|
||||||
REP = 4,
|
|
||||||
XREQ = 5,
|
|
||||||
XREP = 6,
|
|
||||||
PULL = 7,
|
|
||||||
PUSH = 8,
|
|
||||||
|
|
||||||
-- socket options
|
|
||||||
HWM = 1,
|
|
||||||
SWAP = 3,
|
|
||||||
AFFINITY = 4,
|
|
||||||
IDENTITY = 5,
|
|
||||||
SUBSCRIBE = 6,
|
|
||||||
UNSUBSCRIBE = 7,
|
|
||||||
RATE = 8,
|
|
||||||
RECOVERY_IVL = 9,
|
|
||||||
MCAST_LOOP = 10,
|
|
||||||
SNDBUF = 11,
|
|
||||||
RCVBUF = 12,
|
|
||||||
RCVMORE = 13,
|
|
||||||
FD = 14,
|
|
||||||
EVENTS = 15,
|
|
||||||
TYPE = 16,
|
|
||||||
LINGER = 17,
|
|
||||||
RECONNECT_IVL = 18,
|
|
||||||
BACKLOG = 19,
|
|
||||||
|
|
||||||
-- send/recv flags
|
|
||||||
NOBLOCK = 1,
|
|
||||||
SNDMORE = 2,
|
|
||||||
|
|
||||||
-- poll events
|
|
||||||
POLLIN = 1,
|
|
||||||
POLLOUT = 2,
|
|
||||||
POLLERR = 4,
|
|
||||||
|
|
||||||
-- devices
|
|
||||||
STREAMER = 1,
|
|
||||||
FORWARDER = 2,
|
|
||||||
QUEUE = 3,
|
|
||||||
}
|
|
||||||
|
|
||||||
local z_SUBSCRIBE = zmq.SUBSCRIBE
|
|
||||||
local z_UNSUBSCRIBE = zmq.UNSUBSCRIBE
|
|
||||||
local z_IDENTITY = zmq.IDENTITY
|
|
||||||
local z_NOBLOCK = zmq.NOBLOCK
|
|
||||||
local z_RCVMORE = zmq.RCVMORE
|
|
||||||
local z_SNDMORE = zmq.SNDMORE
|
|
||||||
local z_EVENTS = zmq.EVENTS
|
|
||||||
local z_POLLIN = zmq.POLLIN
|
|
||||||
local z_POLLOUT = zmq.POLLOUT
|
|
||||||
local z_POLLIN_OUT = z_POLLIN + z_POLLOUT
|
|
||||||
|
|
||||||
local ffi=require"ffi"
|
|
||||||
ffi.cdef[[
|
|
||||||
void zmq_version (int *major, int *minor, int *patch);
|
|
||||||
int zmq_errno ();
|
|
||||||
const char *zmq_strerror (int errnum);
|
|
||||||
|
|
||||||
typedef struct zmq_msg_t
|
|
||||||
{
|
|
||||||
void *content;
|
|
||||||
unsigned char flags;
|
|
||||||
unsigned char vsm_size;
|
|
||||||
unsigned char vsm_data [30];
|
|
||||||
} zmq_msg_t;
|
|
||||||
|
|
||||||
typedef void (zmq_free_fn) (void *data, void *hint);
|
|
||||||
|
|
||||||
int zmq_msg_init (zmq_msg_t *msg);
|
|
||||||
int zmq_msg_init_size (zmq_msg_t *msg, size_t size);
|
|
||||||
int zmq_msg_init_data (zmq_msg_t *msg, void *data, size_t size, zmq_free_fn *ffn, void *hint);
|
|
||||||
int zmq_msg_close (zmq_msg_t *msg);
|
|
||||||
int zmq_msg_move (zmq_msg_t *dest, zmq_msg_t *src);
|
|
||||||
int zmq_msg_copy (zmq_msg_t *dest, zmq_msg_t *src);
|
|
||||||
void *zmq_msg_data (zmq_msg_t *msg);
|
|
||||||
size_t zmq_msg_size (zmq_msg_t *msg);
|
|
||||||
|
|
||||||
void *zmq_init (int io_threads);
|
|
||||||
int zmq_term (void *context);
|
|
||||||
|
|
||||||
void *zmq_socket (void *context, int type);
|
|
||||||
int zmq_close (void *s);
|
|
||||||
int zmq_setsockopt (void *s, int option, const void *optval, size_t optvallen);
|
|
||||||
int zmq_getsockopt (void *s, int option, void *optval, size_t *optvallen);
|
|
||||||
int zmq_bind (void *s, const char *addr);
|
|
||||||
int zmq_connect (void *s, const char *addr);
|
|
||||||
int zmq_send (void *s, zmq_msg_t *msg, int flags);
|
|
||||||
int zmq_recv (void *s, zmq_msg_t *msg, int flags);
|
|
||||||
|
|
||||||
int zmq_device (int device, void * insocket, void* outsocket);
|
|
||||||
|
|
||||||
]]
|
|
||||||
|
|
||||||
local c_zmq = ffi.load"zmq"
|
|
||||||
|
|
||||||
module(...)
|
|
||||||
|
|
||||||
-- copy constants
|
|
||||||
for k,v in pairs(zmq) do
|
|
||||||
-- only copy upper-case string values.
|
|
||||||
if type(k) == 'string' and k == k:upper() then
|
|
||||||
_M[k] = v
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
function version()
|
|
||||||
local major = ffi.new('int[1]',0)
|
|
||||||
local minor = ffi.new('int[1]',0)
|
|
||||||
local patch = ffi.new('int[1]',0)
|
|
||||||
c_zmq.zmq_version(major, minor, patch)
|
|
||||||
return {major[0], minor[0], patch[0]}
|
|
||||||
end
|
|
||||||
|
|
||||||
local function zmq_error()
|
|
||||||
local errno = c_zmq.zmq_errno()
|
|
||||||
local err = ffi.string(c_zmq.zmq_strerror(errno))
|
|
||||||
if err == "Resource temporarily unavailable" then err = "timeout" end
|
|
||||||
if err == "Context was terminated" then err = "closed" end
|
|
||||||
return nil, err
|
|
||||||
end
|
|
||||||
|
|
||||||
--
|
|
||||||
-- ZMQ socket
|
|
||||||
--
|
|
||||||
local sock_mt = {}
|
|
||||||
sock_mt.__index = sock_mt
|
|
||||||
|
|
||||||
local function new_socket(ctx, sock_type)
|
|
||||||
local sock = c_zmq.zmq_socket(ctx, sock_type)
|
|
||||||
if not sock then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return setmetatable({ sock = sock }, sock_mt)
|
|
||||||
end
|
|
||||||
|
|
||||||
function sock_mt:close()
|
|
||||||
local ret = c_zmq.zmq_close(self.sock)
|
|
||||||
self.sock = nil
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
local option_types = {
|
|
||||||
[zmq.HWM] = 'uint64_t[1]',
|
|
||||||
[zmq.SWAP] = 'int64_t[1]',
|
|
||||||
[zmq.AFFINITY] = 'uint64_t[1]',
|
|
||||||
[zmq.IDENTITY] = 'string',
|
|
||||||
[zmq.SUBSCRIBE] = 'string',
|
|
||||||
[zmq.UNSUBSCRIBE] = 'string',
|
|
||||||
[zmq.RATE] = 'int64_t[1]',
|
|
||||||
[zmq.RECOVERY_IVL] = 'int64_t[1]',
|
|
||||||
[zmq.MCAST_LOOP] = 'int64_t[1]',
|
|
||||||
[zmq.SNDBUF] = 'uint64_t[1]',
|
|
||||||
[zmq.RCVBUF] = 'uint64_t[1]',
|
|
||||||
[zmq.RCVMORE] = 'int64_t[1]',
|
|
||||||
[zmq.FD] = 'int[1]',
|
|
||||||
[zmq.EVENTS] = 'uint32_t[1]',
|
|
||||||
[zmq.TYPE] = 'int[1]',
|
|
||||||
[zmq.LINGER] = 'int[1]',
|
|
||||||
[zmq.RECONNECT_IVL] = 'int[1]',
|
|
||||||
[zmq.BACKLOG] = 'int[1]',
|
|
||||||
}
|
|
||||||
local option_len = {}
|
|
||||||
local option_tmps = {}
|
|
||||||
for k,v in pairs(option_types) do
|
|
||||||
if v ~= 'string' then
|
|
||||||
option_len[k] = ffi.sizeof(v)
|
|
||||||
option_tmps[k] = ffi.new(v, 0)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
function sock_mt:setopt(opt, opt_val)
|
|
||||||
local ctype = option_types[opt]
|
|
||||||
local val_len = 0
|
|
||||||
if ctype == 'string' then
|
|
||||||
--val = ffi.cast('void *', tostring(val))
|
|
||||||
val = tostring(opt_val)
|
|
||||||
val_len = #val
|
|
||||||
else
|
|
||||||
val = option_tmps[opt]
|
|
||||||
val[0] = opt_val
|
|
||||||
val_len = option_len[opt]
|
|
||||||
end
|
|
||||||
local ret = c_zmq.zmq_setsockopt(self.sock, opt, val, val_len)
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
local tmp_val_len = ffi.new('size_t[1]', 4)
|
|
||||||
function sock_mt:getopt(opt)
|
|
||||||
local ctype = option_types[opt]
|
|
||||||
local val
|
|
||||||
local val_len = tmp_val_len
|
|
||||||
if ctype == 'string' then
|
|
||||||
val_len[0] = 255
|
|
||||||
val = ffi.new('uint8_t[?]', val_len[0])
|
|
||||||
ffi.fill(val, val_len[0])
|
|
||||||
else
|
|
||||||
val = option_tmps[opt]
|
|
||||||
val[0] = 0
|
|
||||||
val_len[0] = option_len[opt]
|
|
||||||
end
|
|
||||||
local ret = c_zmq.zmq_getsockopt(self.sock, opt, val, val_len)
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
if ctype == 'string' then
|
|
||||||
val_len = val_len[0]
|
|
||||||
return ffi.string(val, val_len)
|
|
||||||
else
|
|
||||||
val = val[0]
|
|
||||||
end
|
|
||||||
return tonumber(val)
|
|
||||||
end
|
|
||||||
|
|
||||||
local tmp32 = ffi.new('uint32_t[1]', 0)
|
|
||||||
local tmp32_len = ffi.new('size_t[1]', 4)
|
|
||||||
function sock_mt:events()
|
|
||||||
local val = tmp32
|
|
||||||
local val_len = tmp32_len
|
|
||||||
val[0] = 0
|
|
||||||
val_len[0] = 4
|
|
||||||
local ret = c_zmq.zmq_getsockopt(self.sock, 15, val, val_len)
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return val[0]
|
|
||||||
end
|
|
||||||
|
|
||||||
function sock_mt:bind(addr)
|
|
||||||
local ret = c_zmq.zmq_bind(self.sock, addr)
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
function sock_mt:connect(addr)
|
|
||||||
local ret = c_zmq.zmq_connect(self.sock, addr)
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
local tmp_msg = ffi.new('zmq_msg_t')
|
|
||||||
function sock_mt:send(data, flags)
|
|
||||||
local msg = tmp_msg
|
|
||||||
local msg_len = #data
|
|
||||||
-- initialize message
|
|
||||||
if c_zmq.zmq_msg_init_size(msg, msg_len) < 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
-- copy data into message.
|
|
||||||
ffi.copy(c_zmq.zmq_msg_data(msg), data, msg_len)
|
|
||||||
|
|
||||||
-- send message
|
|
||||||
local ret = c_zmq.zmq_send(self.sock, msg, flags or 0)
|
|
||||||
-- close message before processing return code
|
|
||||||
c_zmq.zmq_msg_close(msg)
|
|
||||||
-- now process send return code
|
|
||||||
if ret ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
function sock_mt:recv(flags)
|
|
||||||
local msg = tmp_msg
|
|
||||||
-- initialize blank message.
|
|
||||||
if c_zmq.zmq_msg_init(msg) < 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
|
|
||||||
-- receive message
|
|
||||||
local ret = c_zmq.zmq_recv(self.sock, msg, flags or 0)
|
|
||||||
if ret ~= 0 then
|
|
||||||
local data, err = zmq_error()
|
|
||||||
c_zmq.zmq_msg_close(msg)
|
|
||||||
return data, err
|
|
||||||
end
|
|
||||||
local data = ffi.string(c_zmq.zmq_msg_data(msg), c_zmq.zmq_msg_size(msg))
|
|
||||||
-- close message
|
|
||||||
c_zmq.zmq_msg_close(msg)
|
|
||||||
return data
|
|
||||||
end
|
|
||||||
|
|
||||||
--
|
|
||||||
-- ZMQ context
|
|
||||||
--
|
|
||||||
local ctx_mt = {}
|
|
||||||
ctx_mt.__index = ctx_mt
|
|
||||||
|
|
||||||
function ctx_mt:term()
|
|
||||||
if c_zmq.zmq_term(self.ctx) ~= 0 then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
function ctx_mt:socket(sock_type)
|
|
||||||
return new_socket(self.ctx, sock_type)
|
|
||||||
end
|
|
||||||
|
|
||||||
function init(io_threads)
|
|
||||||
local ctx = c_zmq.zmq_init(io_threads)
|
|
||||||
if not ctx then
|
|
||||||
return zmq_error()
|
|
||||||
end
|
|
||||||
return setmetatable({ ctx = ctx }, ctx_mt)
|
|
||||||
end
|
|
||||||
|
|
||||||
Loading…
Reference in New Issue