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 | 942c223 | 2016-11-17 17:29:45 +0100 | [diff] [blame] | 4 | import random |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 5 | from optparse import OptionParser |
Max | 942c223 | 2016-11-17 17:29:45 +0100 | [diff] [blame] | 6 | from ipa import Ctrl |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 7 | import socket |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 8 | |
| 9 | verbose = False |
| 10 | |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 11 | def connect(host, port): |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame] | 12 | if verbose: |
| 13 | print "Connecting to host %s:%i" % (host, port) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 14 | |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame] | 15 | sck = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
| 16 | sck.setblocking(1) |
| 17 | sck.connect((host, port)) |
| 18 | return sck |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 19 | |
Max | 44c4c47 | 2016-08-01 18:54:17 +0200 | [diff] [blame] | 20 | def do_set_get(sck, var, value = None): |
Max | 942c223 | 2016-11-17 17:29:45 +0100 | [diff] [blame] | 21 | (r, c) = Ctrl().cmd(var, value) |
| 22 | sck.send(c) |
| 23 | answer = Ctrl().rem_header(sck.recv(4096)) |
| 24 | return (answer,) + Ctrl().verify(answer, r, var, value) |
Max | 44c4c47 | 2016-08-01 18:54:17 +0200 | [diff] [blame] | 25 | |
| 26 | def set_var(sck, var, val): |
Max | 942c223 | 2016-11-17 17:29:45 +0100 | [diff] [blame] | 27 | (a, _, _) = do_set_get(sck, var, val) |
| 28 | return a |
Max | 44c4c47 | 2016-08-01 18:54:17 +0200 | [diff] [blame] | 29 | |
| 30 | def get_var(sck, var): |
Max | 942c223 | 2016-11-17 17:29:45 +0100 | [diff] [blame] | 31 | (_, _, v) = do_set_get(sck, var) |
| 32 | return v |
| 33 | |
| 34 | def _leftovers(sck): |
| 35 | data = sck.recv(1024) |
| 36 | if len(data) != 0: |
| 37 | tail = data |
| 38 | while True: |
| 39 | (head, tail) = Ctrl().split_combined(tail) |
| 40 | print "Got message:", Ctrl().rem_header(head) |
| 41 | if len(tail) == 0: |
| 42 | break |
| 43 | return True |
| 44 | return False |
Max | 44c4c47 | 2016-08-01 18:54:17 +0200 | [diff] [blame] | 45 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 46 | if __name__ == '__main__': |
Max | 2e195a9 | 2016-08-01 14:26:19 +0200 | [diff] [blame] | 47 | random.seed() |
| 48 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 49 | parser = OptionParser("Usage: %prog [options] var [value]") |
| 50 | parser.add_option("-d", "--host", dest="host", |
| 51 | help="connect to HOST", metavar="HOST") |
| 52 | parser.add_option("-p", "--port", dest="port", type="int", |
| 53 | help="use PORT", metavar="PORT", default=4249) |
| 54 | parser.add_option("-g", "--get", action="store_true", |
| 55 | dest="cmd_get", help="perform GET operation") |
| 56 | parser.add_option("-s", "--set", action="store_true", |
| 57 | dest="cmd_set", help="perform SET operation") |
Max | 2e195a9 | 2016-08-01 14:26:19 +0200 | [diff] [blame] | 58 | 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] | 59 | help="set id manually", metavar="ID") |
| 60 | parser.add_option("-v", "--verbose", action="store_true", |
| 61 | dest="verbose", help="be verbose", default=False) |
| 62 | parser.add_option("-m", "--monitor", action="store_true", |
| 63 | dest="monitor", help="monitor the connection for traps", default=False) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 64 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 65 | (options, args) = parser.parse_args() |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 66 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 67 | verbose = options.verbose |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 68 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 69 | if options.cmd_set and options.cmd_get: |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame] | 70 | parser.error("Get and set options are mutually exclusive!") |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 71 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 72 | if not (options.cmd_get or options.cmd_set or options.monitor): |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame] | 73 | parser.error("One of -m, -g, or -s must be set") |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 74 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 75 | if not (options.host): |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame] | 76 | parser.error("Destination host and port required!") |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 77 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 78 | sock = connect(options.host, options.port) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 79 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 80 | if options.cmd_set: |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame] | 81 | if len(args) < 2: |
| 82 | parser.error("Set requires var and value arguments") |
Max | 942c223 | 2016-11-17 17:29:45 +0100 | [diff] [blame] | 83 | _leftovers(sock) |
| 84 | print "Got message:", set_var(sock, args[0], ' '.join(args[1:])) |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 85 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 86 | if options.cmd_get: |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame] | 87 | if len(args) != 1: |
| 88 | parser.error("Get requires the var argument") |
Max | 942c223 | 2016-11-17 17:29:45 +0100 | [diff] [blame] | 89 | _leftovers(sock) |
| 90 | (a, _, _) = do_set_get(sock, args[0]) |
| 91 | print "Got message:", a |
Daniel Willmann | 59073a1 | 2011-04-29 18:40:29 +0200 | [diff] [blame] | 92 | |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 93 | if options.monitor: |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame] | 94 | while True: |
Max | 942c223 | 2016-11-17 17:29:45 +0100 | [diff] [blame] | 95 | if not _leftovers(sock): |
Max | dbfe68a | 2016-11-17 17:24:53 +0100 | [diff] [blame] | 96 | print "Connection is gone." |
| 97 | break |
Max | 28b39b9 | 2016-07-29 18:19:25 +0200 | [diff] [blame] | 98 | sock.close() |