Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 1 | #!/usr/bin/python |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame^] | 2 | # -*- mode: python-mode; py-indent-tabs-mode: nil -*- |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 3 | |
Max | 2e195a9 | 2016-08-01 14:26:19 +0200 | [diff] [blame] | 4 | import sys,os, random |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 5 | from optparse import OptionParser |
| 6 | import socket |
| 7 | import struct |
| 8 | |
| 9 | verbose = False |
| 10 | |
| 11 | def prefix_ipa_ctrl_header(data): |
| 12 | return struct.pack(">HBB", len(data)+1, 0xee, 0) + data |
| 13 | |
Max | 44c4c47 | 2016-08-01 18:54:17 +0200 | [diff] [blame] | 14 | def ipa_ctrl_header(header): |
| 15 | (plen, ipa_proto, osmo_proto) = struct.unpack(">HBB", header) |
| 16 | return None if (ipa_proto != 0xee or osmo_proto != 0) else plen |
| 17 | |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 18 | def remove_ipa_ctrl_header(data): |
| 19 | if (len(data) < 4): |
| 20 | raise BaseException("Answer too short!") |
Max | 44c4c47 | 2016-08-01 18:54:17 +0200 | [diff] [blame] | 21 | plen = ipa_ctrl_header(data[:4]) |
| 22 | if (None == plen): |
| 23 | raise BaseException("Wrong protocol in answer!") |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 24 | if (plen + 3 > len(data)): |
| 25 | print "Warning: Wrong payload length (expected %i, got %i)" % (plen, len(data) - 3) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 26 | return data[4:plen+3], data[plen+3:] |
| 27 | |
| 28 | def connect(host, port): |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame^] | 29 | if verbose: |
| 30 | print "Connecting to host %s:%i" % (host, port) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 31 | |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame^] | 32 | sck = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
| 33 | sck.setblocking(1) |
| 34 | sck.connect((host, port)) |
| 35 | return sck |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 36 | |
| 37 | def send(sck, data): |
| 38 | if verbose: |
| 39 | print "Sending \"%s\"" %(data) |
| 40 | data = prefix_ipa_ctrl_header(data) |
| 41 | sck.send(data) |
| 42 | |
Max | 2e195a9 | 2016-08-01 14:26:19 +0200 | [diff] [blame] | 43 | def do_set(var, value, op_id, sck): |
| 44 | setmsg = "SET %s %s %s" %(op_id, var, value) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 45 | send(sck, setmsg) |
| 46 | |
Max | 2e195a9 | 2016-08-01 14:26:19 +0200 | [diff] [blame] | 47 | def do_get(var, op_id, sck): |
| 48 | getmsg = "GET %s %s" %(op_id, var) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 49 | send(sck, getmsg) |
| 50 | |
Max | 44c4c47 | 2016-08-01 18:54:17 +0200 | [diff] [blame] | 51 | def do_set_get(sck, var, value = None): |
| 52 | r = random.randint(1, sys.maxint) |
| 53 | if (value != None): |
| 54 | s = 'SET_REPLY' |
| 55 | do_set(var, value, r, sck) |
| 56 | else: |
| 57 | s = 'GET_REPLY' |
| 58 | do_get(var, r, sck) |
| 59 | (answer, data) = remove_ipa_ctrl_header(sck.recv(4096)) |
| 60 | x = answer.split() |
| 61 | if (s == x[0] and str(r) == x[1] and var == x[2]): |
| 62 | return None if ('SET_REPLY' == s and value != x[3]) else x[3] |
| 63 | return None |
| 64 | |
| 65 | def set_var(sck, var, val): |
| 66 | return do_set_get(sck, var, val) |
| 67 | |
| 68 | def get_var(sck, var): |
| 69 | return do_set_get(sck, var) |
| 70 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 71 | if __name__ == '__main__': |
Max | 2e195a9 | 2016-08-01 14:26:19 +0200 | [diff] [blame] | 72 | random.seed() |
| 73 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 74 | parser = OptionParser("Usage: %prog [options] var [value]") |
| 75 | parser.add_option("-d", "--host", dest="host", |
| 76 | help="connect to HOST", metavar="HOST") |
| 77 | parser.add_option("-p", "--port", dest="port", type="int", |
| 78 | help="use PORT", metavar="PORT", default=4249) |
| 79 | parser.add_option("-g", "--get", action="store_true", |
| 80 | dest="cmd_get", help="perform GET operation") |
| 81 | parser.add_option("-s", "--set", action="store_true", |
| 82 | dest="cmd_set", help="perform SET operation") |
Max | 2e195a9 | 2016-08-01 14:26:19 +0200 | [diff] [blame] | 83 | parser.add_option("-i", "--id", dest="op_id", default=random.randint(1, sys.maxint), |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 84 | help="set id manually", metavar="ID") |
| 85 | parser.add_option("-v", "--verbose", action="store_true", |
| 86 | dest="verbose", help="be verbose", default=False) |
| 87 | parser.add_option("-m", "--monitor", action="store_true", |
| 88 | dest="monitor", help="monitor the connection for traps", default=False) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 89 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 90 | (options, args) = parser.parse_args() |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 91 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 92 | verbose = options.verbose |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 93 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 94 | if options.cmd_set and options.cmd_get: |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame^] | 95 | parser.error("Get and set options are mutually exclusive!") |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 96 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 97 | if not (options.cmd_get or options.cmd_set or options.monitor): |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame^] | 98 | parser.error("One of -m, -g, or -s must be set") |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 99 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 100 | if not (options.host): |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame^] | 101 | parser.error("Destination host and port required!") |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 102 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 103 | sock = connect(options.host, options.port) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 104 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 105 | if options.cmd_set: |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame^] | 106 | if len(args) < 2: |
| 107 | parser.error("Set requires var and value arguments") |
Max | 2e195a9 | 2016-08-01 14:26:19 +0200 | [diff] [blame] | 108 | do_set(args[0], ' '.join(args[1:]), options.op_id, sock) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 109 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 110 | if options.cmd_get: |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame^] | 111 | if len(args) != 1: |
| 112 | parser.error("Get requires the var argument") |
Max | 2e195a9 | 2016-08-01 14:26:19 +0200 | [diff] [blame] | 113 | do_get(args[0], options.op_id, sock) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 114 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 115 | data = sock.recv(1024) |
| 116 | while (len(data)>0): |
| 117 | (answer, data) = remove_ipa_ctrl_header(data) |
| 118 | print "Got message:", answer |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 119 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 120 | if options.monitor: |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame^] | 121 | while True: |
| 122 | data = sock.recv(1024) |
| 123 | if len(data) == 0: |
| 124 | print "Connection is gone." |
| 125 | break |
Daniel Willmann | 6044162 | 2011-07-28 19:04:57 +0200 | [diff] [blame] | 126 | |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame^] | 127 | while len(data) > 0: |
| 128 | (answer, data) = remove_ipa_ctrl_header(data) |
| 129 | print "Got message:", answer |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 130 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 131 | sock.close() |