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
52
53
54
55
56
57
58
59
60
61
62
63
|
# $Id$
"""
Command line program to simulate behavior of the IR back-end.
"""
import glob, rpki.left_right, rpki.relaxng, getopt, sys, lxml.etree
class command(object):
def getopt(self, argv):
if self.options:
opts, args = getopt.getopt(argv, "", [x[2:] + "=" for x in self.options])
for o, a in opts:
getattr(self, o[2:])(a)
return args
else:
return argv
def process(self, msg):
msg.append(self.pdu)
def action(self, arg):
self.pdu.action = arg
self.pdu.type = "query"
def self_id(self, arg):
self.pdu.self_id = arg
class help(command):
options = ()
def process(self, msg):
print "Usage:", sys.argv[0]
for k,v in dispatch.iteritems():
print " ", " ".join((k,) + v.options)
class self(command):
options = ("--action", "--self_id", "--extension")
def __init__(self):
self.pdu = rpki.left_right.self_elt()
def extension(self, arg):
kv = arg.split(":", 1)
self.pdu.extensions[k] = v
dispatch = dict((x.__name__, x) for x in (help, self))
rng = rpki.relaxng.RelaxNG("left-right-schema.rng")
msg = rpki.left_right.msg()
argv = sys.argv[1:]
while argv:
cmd = dispatch[argv[0]]()
argv = cmd.getopt(argv[1:])
cmd.process(msg)
if msg:
elt = msg.toXML()
rng.assertValid(elt)
print lxml.etree.tostring(elt, pretty_print=True, encoding="us-ascii", xml_declaration=True)
|
