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
120
121
|
"""
Parse a WHOIS research dump and write out (just) the RPKI-relevant
fields in myrpki-format CSV syntax.
NB: The input data for this script comes from ARIN under an agreement
that allows research use but forbids redistribution, so if you think
you need a copy of the data, please talk to ARIN about it, not us.
$Id$
Copyright (C) 2009 Internet Systems Consortium ("ISC")
Permission to use, copy, modify, and distribute this software for any
purpose with or without fee is hereby granted, provided that the above
copyright notice and this permission notice appear in all copies.
THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
AND FITNESS. IN NO EVENT SHALL ISC 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 gzip, rpki.myrpki
class Handle(object):
want_tags = ()
debug = False
def set(self, tag, val):
if tag in self.want_tags:
setattr(self, tag, "".join(val.split(" ")))
def check(self):
for tag in self.want_tags:
if not hasattr(self, tag):
return False
if self.debug:
print repr(self)
return True
class ASHandle(Handle):
want_tags = ("ASHandle", "ASNumber", "OrgID")
def __repr__(self):
return "<%s %s.%s %s>" % (self.__class__.__name__,
self.OrgID, self.ASHandle, self.ASNumber)
def finish(self, ctx):
if self.check():
ctx.asns.writerow((ctx.translations.get(self.OrgID, self.OrgID), self.ASNumber))
class NetHandle(Handle):
NetType = None
want_tags = ("NetHandle", "NetRange", "NetType", "OrgID")
def finish(self, ctx):
if self.NetType in ("allocation", "assignment") and self.check():
ctx.prefixes.writerow((ctx.translations.get(self.OrgID, self.OrgID), self.NetRange))
def __repr__(self):
return "<%s %s.%s %s %s>" % (self.__class__.__name__,
self.OrgID, self.NetHandle,
self.NetType, self.NetRange)
class V6NetHandle(NetHandle):
want_tags = ("V6NetHandle", "NetRange", "NetType", "OrgID")
def __repr__(self):
return "<%s %s.%s %s %s>" % (self.__class__.__name__,
ctx.translations.get(self.OrgID, self.OrgID),
self.V6NetHandle, self.NetType, self.NetRange)
class main(object):
types = {
"ASHandle" : ASHandle,
"NetHandle" : NetHandle,
"V6NetHandle" : V6NetHandle }
translations = {}
@staticmethod
def parseline(line):
tag, sep, val = line.partition(":")
assert sep, "Couldn't find separator in %r" % line
return tag.strip(), val.strip()
def __init__(self):
self.asns = rpki.myrpki.csv_writer("asns.csv")
self.prefixes = rpki.myrpki.csv_writer("prefixes.csv")
try:
self.translations = dict((src, dst) for src, dst in rpki.myrpki.csv_reader("translations.csv", columns = 2))
except IOError:
pass
f = gzip.open("arin_db.txt.gz")
cur = None
for line in f:
line = line.expandtabs().strip()
if not line:
if cur:
cur.finish(self)
cur = None
elif not line.startswith("#"):
tag, val = self.parseline(line)
if cur is None:
cur = self.types[tag]() if tag in self.types else False
if cur:
cur.set(tag, val)
if cur:
cur.finish(self)
main()
|
