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
|
#!/usr/bin/env python
# $Id$
#
# Helper script to load existing data from csv into the Django DB.
# Primarly useful for the initial load, as the GUI does not sync changes
# made directly to the csv files back into the database.
#
# This script should be run from the directory containing the myrpki.conf
# for the handle you are loading data
#
import os
import csv
import socket # for socket.error
import rpki
import rpki.resource_set
import rpki.ipaddrs
from rpki.myrpki import csv_reader
from rpkigui.myrpki import models
from rpkigui.myrpki.views import update_roas
from rpkigui.myrpki.asnset import asnset
cfg_file = os.getenv("MYRPKI_CONF", "myrpki.conf")
cfg = rpki.config.parser(cfg_file, "myrpki")
handle = cfg.get('handle')
asn_csv = cfg.get('asn_csv')
prefix_csv = cfg.get('prefix_csv')
roa_csv = cfg.get('roa_csv')
print 'processing csv files for resource handle', handle
conf = models.Conf.objects.get(handle=handle)
for child_handle, asn in csv_reader(asn_csv, columns=2):
child = conf.children.get(handle=child_handle)
asn = models.Asn.objects.get(lo=asn, hi=asn,
from_cert__parent__in=conf.parents.all())
child.asn.add(asn)
for child_handle, prefix in csv_reader(prefix_csv, columns=2):
child = conf.children.get(handle=child_handle)
try:
rs = rpki.resource_set.resource_range_ipv4.from_str(prefix)
except socket.error:
rs = rpki.resource_set.resource_range_ipv6.from_str(prefix)
obj = models.AddressRange.objects.get(lo=str(rs.min), hi=str(rs.max),
from_cert__parent__in=conf.parents.all())
child.address_range.add(obj)
for prefix, asn, group in csv_reader(roa_csv, columns=3):
try:
rs = rpki.resource_set.roa_prefix_set_ipv4().parse_str(prefix)
except socket.error:
rs = rpki.resource_set.roa_prefix_set_ipv6().parse_str(prefix)
if rs.prefixlen != rs.max_prefixlen:
raise ValueError, \
"%s: max prefixlen larger than prefixlen is not currently supported." % (prefix,)
print str(rs.min()), str(rs.max())
obj = models.AddressRange.objects.get(lo=str(rs.min()), hi=str(rs.max()),
from_cert__parent__in=conf.parents.all())
roa_asns = asnset(obj.asns)
asid = int(asn)
if asid not in roa_asns:
roa_asns.add(asid)
obj.asns = str(roa_asns)
obj.save()
update_roas(conf, obj)
|
