diff options
Diffstat (limited to 'test-wdclient.rb')
-rwxr-xr-x | test-wdclient.rb | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/test-wdclient.rb b/test-wdclient.rb new file mode 100755 index 0000000..bd162f0 --- /dev/null +++ b/test-wdclient.rb @@ -0,0 +1,51 @@ +#!/usr/bin/env ruby +# +# test-wdclient.rb: Test watchdog by sending arbitrary command messages. +# +# Copyright (C) 2011 VMware, Inc. +# +# 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. + +require 'ffi-rzmq' + +def raise_if_error(rc) + unless ZMQ::Util.resultcode_ok?(rc) + raise "ZMQ Error: #{ZMQ::Util.error_string}" + end +end + +if ARGV.length < 2 + print "usage: test-wdclient <command-ep> <command> <args...>\n" + exit 1 +end + +command_ep = ARGV.shift +command = ARGV.shift +args = ARGV.join(' ') + +ctx = ZMQ::Context.new() +command_sock = ctx.socket(ZMQ::PUSH) +rc = command_sock.connect(command_ep) +raise_if_error(rc) + +rc = command_sock.send_string(command + ' ' + args) +raise_if_error(rc) +rc = command_sock.close +raise_if_error(rc) + |