1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
require 'rbtrace'
require 'ffi'
module MsgQ
extend FFI::Library
ffi_lib FFI::CURRENT_PROCESS
class EventMsg < FFI::Struct
BUF_SIZE = RBTrace::BUF_SIZE
IPC_NOWAIT = 004000
layout :mtype, :long,
:buf, [:char, BUF_SIZE]
def self.send_cmd(q, str)
msg = EventMsg.new
msg[:mtype] = 1
msg[:buf].to_ptr.put_string(0, str)
ret = MsgQ.msgsnd(q, msg, BUF_SIZE, 0)
FFI::LastError.raise if ret == -1
end
def self.recv_cmd(q, block=true)
MsgQ.rb_enable_interrupt if RUBY_VERSION > '1.9' && RUBY_VERSION < '2.0'
msg = EventMsg.new
ret = MsgQ.msgrcv(q, msg, BUF_SIZE, 0, block ? 0 : IPC_NOWAIT)
if ret == -1
if !block and [Errno::EAGAIN, Errno::ENOMSG].include?(FFI::LastError.exception)
return nil
end
FFI::LastError.raise
end
msg[:buf].to_ptr.read_string_length(BUF_SIZE)
ensure
MsgQ.rb_disable_interrupt if RUBY_VERSION > '1.9' && RUBY_VERSION < '2.0'
end
end
attach_function :msgget, [:int, :int], :int
attach_function :msgrcv, [:int, EventMsg.ptr, :size_t, :long, :int], :int
attach_function :msgsnd, [:int, EventMsg.ptr, :size_t, :int], :int
if RUBY_VERSION > '1.9' && RUBY_VERSION < '2.0'
attach_function :rb_enable_interrupt, [], :void
attach_function :rb_disable_interrupt, [], :void
end
end
|
