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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
# $Id$
#
# Copyright (C) 2015-2016 Parsons Government Services ("PARSONS")
# Portions copyright (C) 2013-2014 Dragon Research Labs ("DRL")
# Portions copyright (C) 2009-2012 Internet Systems Consortium ("ISC")
# Portions copyright (C) 2007-2008 American Registry for Internet Numbers ("ARIN")
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notices and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND PARSONS, DRL, ISC, AND ARIN
# DISCLAIM ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT
# SHALL PARSONS, DRL, ISC, OR ARIN BE LIABLE FOR ANY SPECIAL, DIRECT,
# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
# RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
# CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
# CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
import glob
import lxml.etree
import rpki.up_down
import rpki.left_right
import rpki.publication
import rpki.publication_control
import rpki.relaxng
verbose = False
def test(fileglob, rng, parser, encoding, tester = None):
files = glob.glob(fileglob)
files.sort()
for f in files:
print "<!--", f, "-->"
elt_in = lxml.etree.parse(f).getroot()
if verbose:
print "<!-- Input -->"
print lxml.etree.tostring(elt_in, pretty_print = True, encoding = encoding, xml_declaration = True)
rng.assertValid(elt_in)
parsed = parser.fromXML(elt_in)
elt_out = parsed.toXML()
if verbose:
print "<!-- Output -->"
print lxml.etree.tostring(elt_out, pretty_print = True, encoding = encoding, xml_declaration = True)
rng.assertValid(elt_out)
if tester:
tester(elt_in, elt_out, parsed)
if verbose:
print
def pprint(pairs):
if verbose:
for thing, name in pairs:
if thing is not None:
print "[%s]" % name
print thing.get_POW().pprint()
def ud_tester(elt_in, elt_out, msg):
assert isinstance(msg, rpki.up_down.message_pdu)
if isinstance(msg.payload, rpki.up_down.list_response_pdu):
for c in msg.payload.classes:
pprint([(c.certs[i].cert, ("%s certificate #%d" % (c.class_name, i))) for i in xrange(len(c.certs))] + [(c.issuer, ("%s issuer" % c.class_name))])
def lr_tester(elt_in, elt_out, msg):
assert isinstance(msg, rpki.left_right.msg)
for obj in msg:
if isinstance(obj, rpki.left_right.self_elt):
pprint(((obj.bpki_cert, "BPKI cert"),
(obj.bpki_glue, "BPKI glue")))
if isinstance(obj, rpki.left_right.bsc_elt):
pprint(((obj.signing_cert, "Signing certificate"),
(obj.signing_cert_crl, "Signing certificate CRL")))
# (obj.pkcs10_request, "PKCS #10 request")
if isinstance(obj, rpki.left_right.parent_elt):
pprint(((obj.bpki_cert, "BPKI certificate"),
(obj.bpki_glue, "BPKI glue")))
if isinstance(obj, (rpki.left_right.child_elt, rpki.left_right.repository_elt)):
pprint(((obj.bpki_cert, "BPKI certificate"),
(obj.bpki_glue, "BPKI glue")))
def pp_tester(elt_in, elt_out, msg):
assert isinstance(msg, rpki.publication.msg)
for obj in msg:
if isinstance(obj, rpki.publication.publish_elt):
pprint(((obj.payload, "Publish object"),))
if isinstance(obj, rpki.publication.withdraw_elt):
pprint(((None, "Withdraw object"),))
def pc_tester(elt_in, elt_out, msg):
assert isinstance(msg, rpki.publication_control.msg)
for obj in msg:
if isinstance(obj, rpki.publication_control.client_elt):
pprint(((obj.bpki_cert, "BPKI cert"),
(obj.bpki_glue, "BPKI glue")))
test(fileglob = "up-down-protocol-samples/*.xml",
rng = rpki.relaxng.up_down,
parser = rpki.up_down.msg,
encoding = "utf-8",
tester = ud_tester)
test(fileglob = "left-right-protocol-samples/*.xml",
rng = rpki.relaxng.left_right,
parser = rpki.left_right.msg,
encoding = "us-ascii",
tester = lr_tester)
test(fileglob = "publication-protocol-samples/*.xml",
rng = rpki.relaxng.publication,
parser = rpki.publication.msg,
encoding = "us-ascii",
tester = pp_tester)
test(fileglob = "publication-control-protocol-samples/*.xml",
rng = rpki.relaxng.publication_control,
parser = rpki.publication_control.msg,
encoding = "us-ascii",
tester = pc_tester)
|
