ffho_net.py 41 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255
  1. #!/usr/bin/python
  2. import collections
  3. import re
  4. from copy import deepcopy
  5. mac_prefix = "f2"
  6. # VRF configuration map
  7. vrf_info = {
  8. 'vrf_external' : {
  9. 'table' : 1023,
  10. 'fwmark' : [ '0x1', '0x1023' ],
  11. },
  12. }
  13. #
  14. # Default parameters added to any given bonding interface,
  15. # if not specified at the interface configuration.
  16. default_bond_config = {
  17. 'bond-mode': '802.3ad',
  18. 'bond-min-links': '1',
  19. 'bond-xmit-hash-policy': 'layer3+4'
  20. }
  21. #
  22. # Default parameters added to any given bonding interface,
  23. # if not specified at the interface configuration.
  24. default_bridge_config = {
  25. 'bridge-fd' : '0',
  26. 'bridge-stp' : 'no'
  27. }
  28. #
  29. # Hop penalty to set if none is explicitly specified
  30. # Check if one of these roles is configured for any given node, use first match.
  31. default_hop_penalty_by_role = {
  32. 'bbr' : 5,
  33. 'bras' : 50,
  34. 'batman_gw' : 5,
  35. 'batman_ext': 50,
  36. }
  37. batman_role_evaluation_order = [ 'bbr', 'batman_gw', 'bras' ]
  38. #
  39. # Default interface attributes to be added to GRE interface to AS201701 when
  40. # not already present in pillar interface configuration.
  41. GRE_FFRL_attrs = {
  42. 'mode' : 'gre',
  43. 'method' : 'tunnel',
  44. 'mtu' : '1400',
  45. 'ttl' : '64',
  46. }
  47. # The IPv4/IPv6 prefix used for Loopback IPs
  48. loopback_prefix = {
  49. 'v4' : '10.132.255.',
  50. 'v6' : '2a03:2260:2342:ffff::',
  51. }
  52. # The DNS zone base names used for generating zone files from IP address
  53. # configured on nodes interfaces.
  54. DNS_zone_names = {
  55. 'forward' : 'ffho.net',
  56. 'rev_v4' : [
  57. '132.10.in-addr.arpa',
  58. '30.172.in-addr.arpa',
  59. ],
  60. 'rev_v6' : [
  61. '2.4.3.2.0.6.2.2.3.0.a.2.ip6.arpa',
  62. ]
  63. }
  64. # MTU configuration
  65. MTU = {
  66. # The default MTU for any interface which does not have a MTU configured
  67. # explicitly in the pillar node config or does not get a MTU configured
  68. # by any means of this SDN stuff here.
  69. 'default' : 1500,
  70. # A batman underlay device, probably a VXLAN or VLAN interface.
  71. #
  72. # 1500
  73. # + 60 B.A.T.M.A.N. adv header + network coding (activated by default by Debian)
  74. 'batman_underlay_iface' : 1560,
  75. # VXLAN underlay device, probably a VLAN within $POP or between two BBRs.
  76. #
  77. # 1560
  78. # + 14 Inner Ethernet Frame
  79. # + 8 VXLAN Header
  80. # + 8 UDP Header
  81. # + 20 IPv4 Header
  82. 'vxlan_underlay_iface' : 1610,
  83. }
  84. ################################################################################
  85. # #
  86. # Internal functions #
  87. # #
  88. # Touching anything below will void any warranty you never had ;) #
  89. # #
  90. ################################################################################
  91. sites = None
  92. def _get_site_no (sites_config, site_name):
  93. global sites
  94. if sites == None:
  95. sites = {}
  96. for site in sites_config:
  97. if site.startswith ("_"):
  98. continue
  99. sites[site] = sites_config[site].get ("site_no", -2)
  100. return sites.get (site_name, -1)
  101. #
  102. # Generate a MAC address after the format f2:dd:dd:ss:nn:nn where
  103. # dd:dd is the hexadecimal reprensentation of the nodes device_id
  104. # ff:ff representing the gluon nodes
  105. #
  106. # ss is the hexadecimal reprensentation of the site_id the interface is connected to
  107. #
  108. # nn:nn is the decimal representation of the network the interface is connected to, with
  109. # 00:00 being the BATMAN interface
  110. # 00:0d being the dummy interface
  111. # 00:0f being the VEth internal side interface
  112. # 00:e0 being an external instance BATMAN interface
  113. # 00:ed being an external instance dummy interface
  114. # 00:e1 being an inter-gw-vpn interface
  115. # 00:e4 being an nodes fastd tunnel interface of IPv4 transport
  116. # 00:e6 being an nodes fastd tunnel interface of IPv6 transport
  117. # 00:ef being an extenral instance VEth interface side
  118. # 02:xx being a connection to local Vlan 2xx
  119. # xx:xx being a VXLAN tunnel for site ss, with xx being the underlay VLAN ID (1xyz, 2xyz)
  120. # ff:ff being the gluon next-node interface
  121. def gen_batman_iface_mac (site_no, device_no, network):
  122. net_type_map = {
  123. 'bat' : "00:00",
  124. 'dummy' : "00:0d",
  125. 'int2ext' : "00:0f",
  126. 'bat-e' : "00:e0",
  127. 'intergw' : "00:e1",
  128. 'nodes4' : "00:e4",
  129. 'nodes6' : "00:e6",
  130. 'dummy-e' : "00:ed",
  131. 'ext2int' : "00:ef",
  132. }
  133. # Well-known network type?
  134. if network in net_type_map:
  135. last = net_type_map[network]
  136. elif type (network) == int:
  137. last = re.sub (r'(\d{2})(\d{2})', '\g<1>:\g<2>', "%04d" % network)
  138. else:
  139. last = "ee:ee"
  140. # Convert device_no to hex, format number to 4 digits with leading zeros and : betwwen 2nd and 3rd digit
  141. device_no_hex = re.sub (r'([0-9a-fA-F]{2})([0-9a-fA-F]{2})', '\g<1>:\g<2>', "%04x" % int (device_no))
  142. # Format site_no to two digit number with leading zero
  143. site_no_hex = "%02d" % int (site_no)
  144. return "%s:%s:%s:%s" % (mac_prefix, device_no_hex, site_no_hex, last)
  145. # Gather B.A.T.M.A.N. related config options for real batman devices (e.g. bat0)
  146. # as well as for batman member interfaces (e.g. eth0.100, fastd ifaces etc.)
  147. def _update_batman_config (node_config, iface, sites_config):
  148. try:
  149. node_batman_hop_penalty = int (node_config['batman']['hop-penalty'])
  150. except (KeyError,ValueError):
  151. node_batman_hop_penalty = None
  152. iface_config = node_config['ifaces'][iface]
  153. iface_type = iface_config.get ('type', 'inet')
  154. batman_config = {}
  155. for item, value in iface_config.items ():
  156. if item.startswith ('batman-'):
  157. batman_config[item] = value
  158. iface_config.pop (item)
  159. # B.A.T.M.A.N. device (e.g. bat0)
  160. if iface_type == 'batman':
  161. if 'batman-hop-penalty' not in batman_config:
  162. # If there's a hop penalty set for the node, but not for the interface
  163. # apply the nodes hop penalty
  164. if node_batman_hop_penalty:
  165. batman_config['batman-hop-penalty'] = node_batman_hop_penalty
  166. # If there's no hop penalty set for the node, use a default hop penalty
  167. # for the roles the node might have, if any
  168. else:
  169. node_roles = node_config.get ('roles', [])
  170. for role in batman_role_evaluation_order:
  171. if role in node_roles:
  172. batman_config['batman-hop-penalty'] = default_hop_penalty_by_role[role]
  173. if 'batman_ext' in node_roles and iface.endswith('-ext'):
  174. batman_config['batman-hop-penalty'] = default_hop_penalty_by_role['batman_ext']
  175. # If batman ifaces were specified as a list - which they should -
  176. # generate a sorted list of interface names as string representation
  177. if 'batman-ifaces' in batman_config and type (batman_config['batman-ifaces']) == list:
  178. batman_iface_str = " ".join (sorted (batman_config['batman-ifaces']))
  179. batman_config['batman-ifaces'] = batman_iface_str
  180. # B.A.T.M.A.N. member interface (e.g. eth.100, fastd ifaces, etc.)
  181. elif iface_type == 'batman_iface':
  182. # Generate unique MAC address for every batman iface, as B.A.T.M.A.N.
  183. # will get puzzled with multiple interfaces having the same MAC and
  184. # do nasty things.
  185. site = iface_config.get ('site')
  186. site_no = _get_site_no (sites_config, site)
  187. device_no = node_config.get ('id')
  188. network = 1234
  189. # Generate a unique BATMAN-MAC for this interfaces
  190. match = re.search (r'^vlan(\d+)', iface)
  191. if match:
  192. network = int (match.group (1))
  193. iface_config['hwaddress'] = gen_batman_iface_mac (site_no, device_no, network)
  194. iface_config['batman'] = batman_config
  195. # Mangle bond specific config items with default values and store them in
  196. # separate sub-dict for easier access and configuration.
  197. def _update_bond_config (config):
  198. bond_config = default_bond_config.copy ()
  199. to_pop = []
  200. for item, value in config.items ():
  201. if item.startswith ('bond-'):
  202. bond_config[item] = value
  203. to_pop.append (item)
  204. for item in to_pop:
  205. config.pop (item)
  206. if bond_config['bond-mode'] not in ['2', 'balance-xor', '4', '802.3ad']:
  207. bond_config.pop ('bond-xmit-hash-policy')
  208. config['bond'] = bond_config
  209. # Mangle bridge specific config items with default values and store them in
  210. # separate sub-dict for easier access and configuration.
  211. def _update_bridge_config (config):
  212. bridge_config = default_bridge_config.copy ()
  213. for item, value in config.items ():
  214. if item.startswith ('bridge-'):
  215. bridge_config[item] = value
  216. config.pop (item)
  217. # Fix and salt mangled string interpretation back to real string.
  218. if type (value) == bool:
  219. bridge_config[item] = "yes" if value else "no"
  220. # If bridge ports were specified as a list - which they should -
  221. # generate a sorted list of interface names as string representation
  222. if 'bridge-ports' in bridge_config and type (bridge_config['bridge-ports']) == list:
  223. bridge_ports_str = " ".join (sorted (bridge_config['bridge-ports']))
  224. bridge_config['bridge-ports'] = bridge_ports_str
  225. config['bridge'] = bridge_config
  226. # Move vlan specific config items into a sub-dict for easier access and pretty-printing
  227. # in the configuration file
  228. def _update_vlan_config (config):
  229. vlan_config = {}
  230. for item, value in config.items ():
  231. if item.startswith ('vlan-'):
  232. vlan_config[item] = value
  233. config.pop (item)
  234. config['vlan'] = vlan_config
  235. # Pimp Veth interfaces
  236. # * Add peer interface name IF not present
  237. # * Add link-type veth IF not present
  238. def _update_veth_config (interface, config):
  239. veth_peer_name = {
  240. 'veth_ext2int' : 'veth_int2ext',
  241. 'veth_int2ext' : 'veth_ext2int'
  242. }
  243. if interface not in veth_peer_name:
  244. return
  245. if 'link-type' not in config:
  246. config['link-type'] = 'veth'
  247. if 'veth-peer-name' not in config:
  248. config['veth-peer-name'] = veth_peer_name[interface]
  249. # The the given MTU to the given interface - presented by it's interface config dict -
  250. # IFF no MTU has already been set in the node pillar.
  251. #
  252. # @param ifaces: All interface configuration (as dict)
  253. # @param iface_name: Name of the interface to set MTU for
  254. # @param mtu: The MTU value to set (integer)
  255. # When <mtu> is <= 0, the <mtu> configured for <iface_name>
  256. # will be used to set the MTU of the upper interface, and the
  257. # default MTU if none is configured explicitly.
  258. def _set_mtu_to_iface_and_upper (ifaces, iface_name, mtu):
  259. iface_config = ifaces.get (iface_name)
  260. # By default we assume that we should set the given MTU value as the 'automtu'
  261. # attribute to allow distinction between manually set and autogenerated MTU
  262. # values.
  263. set_automtu = True
  264. # If a mtu values <= 0 is given, use the MTU configured for this interface
  265. # or, if none is set, the default value when configuring the vlan-raw-device.
  266. if mtu <= 0:
  267. set_automtu = False
  268. mtu = iface_config.get ('mtu', MTU['default'])
  269. # If this interface already has a MTU set - probably because someone manually
  270. # specified one in the node pillar - we do not touch the MTU of this interface.
  271. # Nevertheless it's worth looking at any underlying interface.
  272. if 'mtu' in iface_config:
  273. set_automtu = False
  274. # There might be - read: "we have" - a situation where on top of e.g. bond0
  275. # there are vlans holding VXLAN communicaton as well a vlans directly carrying
  276. # BATMAN traffic. Now depending on which interface is evaluated first, the upper
  277. # MTU is either correct, or maybe to small.
  278. #
  279. # If any former autogenerated MTU is greater-or-equal than the one we want to
  280. # set now, we'll ignore it, and go for the greater one.
  281. elif 'automtu' in iface_config and iface_config['automtu'] >= mtu:
  282. set_automtu = False
  283. # If we still consider this a good move, set given MTU to this device.
  284. if set_automtu:
  285. iface_config['automtu'] = mtu
  286. # If this is a VLAN - which it probably is - fix the MTU of the underlying interface, too.
  287. # Check for 'vlan-raw-device' in iface_config and in vlan subconfig (yeah, that's not ideal).
  288. vlan_raw_device = None
  289. if 'vlan-raw-device' in iface_config:
  290. vlan_raw_device = iface_config['vlan-raw-device']
  291. elif 'vlan' in iface_config and 'vlan-raw-device' in iface_config['vlan']:
  292. vlan_raw_device = iface_config['vlan']['vlan-raw-device']
  293. if vlan_raw_device:
  294. vlan_raw_device_config = ifaces.get (vlan_raw_device, None)
  295. # vlan-raw-device might point to ethX which usually isn't configured explicitly
  296. # as ifupdown2 simply will bring it up anyway by itself. To set the MTU of such
  297. # an interface we have to add a configuration stanza for it here.
  298. if vlan_raw_device_config == None:
  299. vlan_raw_device_config = {}
  300. ifaces[vlan_raw_device] = vlan_raw_device_config
  301. # If there is a manually set MTU for this device, we don't do nothin'
  302. if 'mtu' in vlan_raw_device_config:
  303. return
  304. if 'automtu' in vlan_raw_device_config and vlan_raw_device_config['automtu'] >= mtu:
  305. return
  306. vlan_raw_device_config['automtu'] = mtu
  307. # Generate configuration entries for any batman related interfaces not
  308. # configured explicitly, but asked for implicitly by role batman and a
  309. # (list of) site(s) specified in the node config.
  310. def _generate_batman_interface_config (node_config, ifaces, sites_config):
  311. # No role 'batman', nothing to do
  312. roles = node_config.get ('roles', [])
  313. if 'batman' not in roles:
  314. return
  315. # Should there be a 2nd external BATMAN instance?
  316. batman_ext = 'batman_ext' in roles or 'bras' in roles
  317. device_no = node_config.get ('id', -1)
  318. for site in node_config.get ('sites', []):
  319. site_no = _get_site_no (sites_config, site)
  320. # Predefine interface names for regular/external BATMAN instance
  321. # and possible VEth link pair for connecting both instances.
  322. bat_site_if = "bat-%s" % site
  323. dummy_site_if = "dummy-%s" % site
  324. bat_site_if_ext = "bat-%s-ext" % site
  325. dummy_site_if_ext = "dummy-%s-e" % site
  326. int2ext_site_if = "i2e-%s" % site
  327. ext2int_site_if = "e2i-%s" % site
  328. site_ifaces = {
  329. # Regular BATMAN interface, always present
  330. bat_site_if : {
  331. 'type' : 'batman',
  332. # int2ext_site_if will be added automagically if requred
  333. 'batman-ifaces' : [ dummy_site_if ],
  334. 'batman-ifaces-ignore-regex': '.*_.*',
  335. 'hwaddress' : gen_batman_iface_mac (site_no, device_no, 'bat'),
  336. },
  337. # Dummy interface always present in regular BATMAN instance
  338. dummy_site_if : {
  339. 'link-type' : 'dummy',
  340. 'hwaddress' : gen_batman_iface_mac (site_no, device_no, 'dummy'),
  341. 'mtu' : MTU['batman_underlay_iface'],
  342. },
  343. # Optional 2nd "external" BATMAN instance
  344. bat_site_if_ext : {
  345. 'type' : 'batman',
  346. 'batman-ifaces' : [ dummy_site_if_ext, ext2int_site_if ],
  347. 'batman-ifaces-ignore-regex': '.*_.*',
  348. 'hwaddress' : gen_batman_iface_mac (site_no, device_no, 'bat-e'),
  349. 'ext_only' : True,
  350. },
  351. # Optional dummy interface always present in 2nd "external" BATMAN instance
  352. dummy_site_if_ext : {
  353. 'link-type' : 'dummy',
  354. 'hwaddress' : gen_batman_iface_mac (site_no, device_no, 'dummy-e'),
  355. 'ext_only' : True,
  356. 'mtu' : MTU['batman_underlay_iface'],
  357. },
  358. # Optional VEth interface pair - internal side
  359. int2ext_site_if : {
  360. 'link-type' : 'veth',
  361. 'veth-peer-name' : ext2int_site_if,
  362. 'hwaddress' : gen_batman_iface_mac (site_no, device_no, 'int2ext'),
  363. 'mtu' : MTU['batman_underlay_iface'],
  364. 'ext_only' : True,
  365. },
  366. # Optional VEth interface pair - "external" side
  367. ext2int_site_if : {
  368. 'link-type' : 'veth',
  369. 'veth-peer-name' : int2ext_site_if,
  370. 'hwaddress' : gen_batman_iface_mac (site_no, device_no, 'ext2int'),
  371. 'mtu' : MTU['batman_underlay_iface'],
  372. 'ext_only' : True,
  373. },
  374. }
  375. for iface, iface_config_tmpl in site_ifaces.items ():
  376. # Ignore any interface only relevant when role batman_ext is set
  377. # but it isn't
  378. if not batman_ext and iface_config_tmpl.get ('ext_only', False):
  379. continue
  380. # Remove ext_only key so we don't leak it into ifaces dict
  381. if 'ext_only' in iface_config_tmpl:
  382. del iface_config_tmpl['ext_only']
  383. # If there is no trace of the desired iface config yet...
  384. if iface not in ifaces:
  385. # ... just place our template there.
  386. ifaces[iface] = iface_config_tmpl
  387. # If there should be an 2nd external BATMAN instance make sure
  388. # the internal side of the VEth iface pair is connected to the
  389. # internal BATMAN instance.
  390. if batman_ext and iface == bat_site_if:
  391. iface_config_tmpl['batman-ifaces'].append (int2ext_site_if)
  392. # If there already is an interface configuration try to enhance it with
  393. # meaningful values from our template and force correct hwaddress to be
  394. # used.
  395. else:
  396. iface_config = ifaces[iface]
  397. # Force hwaddress to be what we expect.
  398. if 'hwaddress' in iface_config_tmpl:
  399. iface_config['hwaddress'] = iface_config_tmpl['hwaddress']
  400. # Copy every attribute of the config template missing in iface config
  401. for attr in iface_config_tmpl:
  402. if attr not in iface_config:
  403. iface_config[attr] = iface_config_tmpl[attr]
  404. # Make sure there is a bridge present for every site where a mesh_breakout
  405. # interface should be configured.
  406. for iface, config in ifaces.items ():
  407. iface_type = config.get ('type', 'inet')
  408. if iface_type not in ['mesh_breakout', 'batman_iface']:
  409. continue
  410. site = config.get ('site')
  411. site_bridge = "br-%s" % site
  412. batman_site_if = "bat-%s" % site
  413. if iface_type == 'mesh_breakout':
  414. # If the bridge has already been defined (with an IP maybe) make
  415. # sure that the corresbonding batman device is part of the bridge-
  416. # ports.
  417. if site_bridge in ifaces:
  418. bridge_config = ifaces.get (site_bridge)
  419. # If there already is/are (a) bridge-port(s) defined, add
  420. # the batman and the breakout interfaces if not present...
  421. bridge_ports = bridge_config.get ('bridge-ports', None)
  422. if bridge_ports:
  423. for dev in (batman_site_if, iface):
  424. if not dev in bridge_ports:
  425. if type (bridge_ports) == list:
  426. bridge_ports.append (dev)
  427. else:
  428. bridge_config['bridge-ports'] += ' ' + dev
  429. # ...if there is no bridge-port defined yet, just used
  430. # the batman and breakout iface.
  431. else:
  432. bridge_config['bridge-ports'] = [ iface, batman_site_if ]
  433. # If the bridge isn't present alltogether, add it.
  434. else:
  435. ifaces[site_bridge] = {
  436. 'bridge-ports' : [ iface, batman_site_if ],
  437. }
  438. elif iface_type == 'batman_iface':
  439. batman_ifaces = ifaces[batman_site_if]['batman-ifaces']
  440. if iface not in batman_ifaces:
  441. if type (batman_ifaces) == list:
  442. batman_ifaces.append (iface)
  443. else:
  444. batman_ifaces += ' ' + iface
  445. # Use the MTU configured for this interface or, if none is set,
  446. # the default value for batman underlay iface.
  447. mtu = config.get('mtu', MTU['batman_underlay_iface'])
  448. _set_mtu_to_iface_and_upper (ifaces, iface, mtu)
  449. #
  450. # Generate any implicitly defined VXLAN interfaces defined in the nodes iface
  451. # defined in pillar.
  452. # The keyword "batman_connect_sites" on an interface will trigger the
  453. # generation of a VXLAN overlay interfaces.
  454. def _generate_vxlan_interface_config (node_config, ifaces, sites_config):
  455. # No role 'batman', nothing to do
  456. if 'batman' not in node_config.get ('roles', []):
  457. return
  458. # Sites configured on this node. Nothing to do, if none.
  459. my_sites = node_config.get ('sites', [])
  460. if len (my_sites) == 0:
  461. return
  462. # As we're still here we can now safely assume that a B.A.T.M.A.N.
  463. # device has been configured for every site specified in sites list.
  464. device_no = node_config.get ('id', -1)
  465. for iface, iface_config in ifaces.items ():
  466. batman_connect_sites = iface_config.get ('batman_connect_sites', [])
  467. # If we got a string, convert it to a list with a single element
  468. if type (batman_connect_sites) == str:
  469. batman_connect_sites = [ batman_connect_sites ]
  470. # If there the list of sites to connect is empty, there's nothing to do here.
  471. if len (batman_connect_sites) == 0:
  472. continue
  473. # Set the MTU of this (probably) VLAN device to the MTU required for a VXLAN underlay
  474. # device, where B.A.T.M.A.N. adv. is to be expected within the VXLAN overlay.
  475. _set_mtu_to_iface_and_upper (ifaces, iface, MTU['vxlan_underlay_iface'])
  476. # If the string 'all' is part of the list, blindly use all sites configured for this node
  477. if 'all' in batman_connect_sites:
  478. batman_connect_sites = my_sites
  479. for site in batman_connect_sites:
  480. # Silenty ignore sites not configured on this node
  481. if site not in my_sites:
  482. continue
  483. # iface_name := vx_<last 5 chars of underlay iface>_<site> stripped to 15 chars
  484. vx_iface = ("vx_%s_%s" % (re.sub ('vlan', 'v', iface)[-5:], re.sub (r'[_-]', '', site)))[:15]
  485. site_no = _get_site_no (sites_config, site)
  486. vni = 100 + site_no
  487. bat_iface = "bat-%s" % site
  488. try:
  489. iface_id = int (re.sub ('vlan', '', iface))
  490. # Gather interface specific mcast address.
  491. # The address is derived from the vlan-id of the underlying interface,
  492. # assuming that it in fact is a vlan interface.
  493. # Mangle the vlan-id into two 2 digit values, eliminating any leading zeros.
  494. iface_id_4digit = "%04d" % iface_id
  495. octet2 = int (iface_id_4digit[0:2])
  496. octet3 = int (iface_id_4digit[2:4])
  497. mcast_ip = "225.%s.%s.%s" % (octet2, octet3, site_no)
  498. vni = octet2 * 256 * 256 + octet3 * 256 + site_no
  499. except ValueError:
  500. iface_id = 9999
  501. mcast_ip = "225.0.0.%s" % site_no
  502. vni = site_no
  503. # bail out if VXLAN tunnel already configured
  504. if vx_iface in ifaces:
  505. continue
  506. # If there's no batman interface for this site, there's no point
  507. # in setting up a VXLAN interfaces
  508. if bat_iface not in ifaces:
  509. continue
  510. # Add the VXLAN interface
  511. ifaces[vx_iface] = {
  512. 'vxlan' : {
  513. 'vxlan-id' : vni,
  514. 'vxlan-svcnodeip' : mcast_ip,
  515. 'vxlan-physdev' : iface,
  516. },
  517. 'hwaddress' : gen_batman_iface_mac (site_no, device_no, iface_id),
  518. 'mtu' : MTU['batman_underlay_iface'],
  519. }
  520. # If the batman interface for this site doesn't have any interfaces
  521. # set up - which basicly cannot happen - add this VXLAN tunnel as
  522. # the first in the list.
  523. if not 'batman-ifaces' in ifaces[bat_iface]:
  524. ifaces[bat_iface]['batman-ifaces'] = [ vx_iface ]
  525. continue
  526. # In the hope there already are interfaces for batman set up already
  527. # add this VXLAN tunnel to the list
  528. batman_ifaces = ifaces[bat_iface]['batman-ifaces']
  529. if vx_iface not in batman_ifaces:
  530. if type (batman_ifaces) == list:
  531. batman_ifaces.append (vx_iface)
  532. else:
  533. batman_ifaces += ' ' + vx_iface
  534. #
  535. # Generate implicitly defined VRFs according to the vrf_info dict at the top
  536. # of this file
  537. def _generate_vrfs (ifaces):
  538. for iface, iface_config in ifaces.items ():
  539. vrf = iface_config.get ('vrf', None)
  540. if vrf and vrf not in ifaces:
  541. conf = vrf_info.get (vrf, {})
  542. table = conf.get ('table', 1234)
  543. fwmark = conf.get ('fwmark', None)
  544. ifaces[vrf] = {
  545. 'vrf-table' : table,
  546. }
  547. # Create ip rule's for any fwmarks defined
  548. if fwmark:
  549. up = []
  550. # Make sure we are dealing with a list even if there is only one mark to be set up
  551. if type (fwmark) in (str, int):
  552. fwmark = [ fwmark ]
  553. # Create ip rule entries for IPv4 and IPv6 for every fwmark
  554. for mark in fwmark:
  555. up.append ("ip rule add fwmark %s table %s" % (mark, table))
  556. up.append ("ip -6 rule add fwmark %s table %s" % (mark, table))
  557. ifaces[vrf]['up'] = up
  558. def _generate_ffrl_gre_tunnels (ifaces):
  559. for iface, iface_config in ifaces.items ():
  560. # We only care for GRE_FFRL type interfaces
  561. if iface_config.get ('type', '') != 'GRE_FFRL':
  562. continue
  563. # Copy default values to interface config
  564. for attr, val in GRE_FFRL_attrs.items ():
  565. if not attr in iface_config:
  566. iface_config[attr] = val
  567. # Guesstimate local IPv4 tunnel endpoint address from tunnel-physdev
  568. if not 'local' in iface_config and 'tunnel-physdev' in iface_config:
  569. try:
  570. physdev_prefixes = [p.split ('/')[0] for p in ifaces[iface_config['tunnel-physdev']]['prefixes'] if '.' in p]
  571. if len (physdev_prefixes) == 1:
  572. iface_config['local'] = physdev_prefixes[0]
  573. except KeyError:
  574. pass
  575. def _generate_loopback_ips (ifaces, node_config, node_id):
  576. v4_ip = "%s/32" % get_loopback_ip (node_config, node_id, 'v4')
  577. v6_ip = "%s/128" % get_loopback_ip (node_config, node_id, 'v6')
  578. # Interface lo already present?
  579. if 'lo' not in ifaces:
  580. ifaces['lo'] = { 'prefixes' : [] }
  581. # Add 'prefixes' list if not present
  582. if 'prefixes' not in ifaces['lo']:
  583. ifaces['lo']['prefixes'] = []
  584. prefixes = ifaces['lo']['prefixes']
  585. if v4_ip not in prefixes:
  586. prefixes.append (v4_ip)
  587. if v6_ip not in prefixes:
  588. prefixes.append (v6_ip)
  589. # Generate interface descriptions / aliases for auto generated or manually
  590. # created interfaces. Currently this only is done for bridges associated
  591. # with BATMAN instanzes.
  592. #
  593. # @param node_config: The configuration of the given node (as dict)
  594. # @param sites_config Global sites configuration (as dict)
  595. def _update_interface_desc (node_config, sites_config):
  596. # Currently we only care for nodes with batman role.
  597. if 'batman' not in node_config.get ('roles', []):
  598. return
  599. for iface, iface_config in node_config.get ('ifaces', {}).items ():
  600. if 'desc' in sites_config:
  601. continue
  602. # If the interface name looks like a bridge for a BATMAN instance
  603. # try to get the name of the corresponding site
  604. match = re.search (r'^br-([a-z_-]+)$', iface)
  605. if match and match.group (1) in sites_config:
  606. try:
  607. iface_config['desc'] = sites_config[match.group (1)]['name']
  608. except KeyError:
  609. pass
  610. ################################################################################
  611. # Public functions #
  612. ################################################################################
  613. # Generate network interface configuration for given node.
  614. #
  615. # This function will read the network configuration from pillar and will
  616. # * enhance it with all default values configured at the top this file
  617. # * auto generate any implicitly configured
  618. # * VRFs
  619. # * B.A.T.M.A.N. instances and interfaces
  620. # * VXLAN interfaces to connect B.A.T.M.A.N. sites
  621. # * Loopback IPs derived from numeric node ID
  622. #
  623. # @param: node_config Pillar node configuration (as dict)
  624. # @param: sites_config Pillar sites configuration (as dict)
  625. # @param: node_id Minion name / Pillar node configuration key
  626. def get_interface_config (node_config, sites_config, node_id = ""):
  627. # Make a copy of the node_config dictionary to suppress side-effects.
  628. # This function deletes some keys from the node_config which will break
  629. # any re-run of this function or other functions relying on the node_config
  630. # to be complete.
  631. node_config = deepcopy (node_config)
  632. # Get config of this node and dict of all configured ifaces
  633. ifaces = node_config.get ('ifaces', {})
  634. # Generate configuration entries for any batman related interfaces not
  635. # configured explicitly, but asked for implicitly by role <batman> and
  636. # a (list of) site(s) specified in the node config.
  637. _generate_batman_interface_config (node_config, ifaces, sites_config)
  638. # Generate VXLAN tunnels for every interfaces specifying 'batman_connect_sites'
  639. _generate_vxlan_interface_config (node_config, ifaces, sites_config)
  640. # Enhance ifaces configuration with some meaningful defaults for
  641. # bonding, bridge and vlan interfaces, MAC address for batman ifaces, etc.
  642. for interface, config in ifaces.items ():
  643. # if type (config) not in [ dict, collections.OrderedDict ]:
  644. # raise Exception ("Configuration for interface %s on node %s seems broken: Type %s" % (interface, node_id, type (config)))
  645. iface_type = config.get ('type', 'inet')
  646. if 'batman-ifaces' in config or iface_type.startswith ('batman'):
  647. _update_batman_config (node_config, interface, sites_config)
  648. if 'bond-slaves' in config:
  649. _update_bond_config (config)
  650. # FIXME: This maybe will not match on bridges without any member ports configured!
  651. if 'bridge-ports' in config or interface.startswith ('br-'):
  652. _update_bridge_config (config)
  653. if 'vlan-raw-device' in config or 'vlan-id' in config:
  654. _update_vlan_config (config)
  655. _set_mtu_to_iface_and_upper (ifaces, interface, 0)
  656. # Pimp configuration for VEth link pairs
  657. if interface.startswith ('veth_'):
  658. _update_veth_config (interface, config)
  659. # Auto generate Loopback IPs IFF not present
  660. _generate_loopback_ips (ifaces, node_config, node_id)
  661. # Auto generated VRF devices for any VRF found in ifaces and not already configured.
  662. _generate_vrfs (ifaces)
  663. # Pimp GRE_FFRL type inteface configuration with default values
  664. _generate_ffrl_gre_tunnels (ifaces)
  665. # Drop any config parameters used in node interface configuration not
  666. # relevant anymore for config file generation.
  667. for interface, config in ifaces.items ():
  668. # Set default MTU if not already set manually or by any earlier function
  669. if interface != 'lo' and ('mtu' not in config):
  670. # Set the MTU value of this interface to the autogenerated value (if any)
  671. # or set the default, when no automtu is present.
  672. config['mtu'] = config.get ('automtu', MTU['default'])
  673. for key in [ 'automtu', 'batman_connect_sites', 'has_gateway', 'ospf', 'site', 'type', 'tagged_vlans' ]:
  674. if key in config:
  675. config.pop (key)
  676. # This leaves 'auto', 'prefixes' and 'desc' as keys which should not be directly
  677. # printed into the remaining configuration. These are handled within the jinja
  678. # interface template.
  679. # Generate meaningful interface descriptions / aliases where useful
  680. _update_interface_desc (node_config, sites_config)
  681. return ifaces
  682. # Generate entries for /etc/bat-hosts for every batman interface we will configure on any node.
  683. # For readability purposes superflous/redundant information is being stripped/supressed.
  684. # As these names will only show up in batctl calls with a specific site, site_names in interfaces
  685. # are stripped. Dummy interfaces are stripped as well.
  686. def gen_bat_hosts (nodes_config, sites_config):
  687. bat_hosts = {}
  688. for node_id in sorted (nodes_config.keys ()):
  689. node_config = nodes_config.get (node_id)
  690. node_name = node_id.split ('.')[0]
  691. ifaces = get_interface_config (node_config, sites_config, node_id)
  692. for iface in sorted (ifaces):
  693. iface_config = ifaces.get (iface)
  694. hwaddress = iface_config.get ('hwaddress', None)
  695. if hwaddress == None:
  696. continue
  697. entry_name = node_name
  698. match = re.search (r'^dummy-(.+)(-e)?$', iface)
  699. if match:
  700. if match.group (2):
  701. entry_name += "-e"
  702. # Append site to make name unique
  703. entry_name += "/%s" % match.group (1)
  704. else:
  705. entry_name += "/%s" % re.sub (r'^(vx_.*|i2e|e2i)[_-](.*)$', '\g<1>/\g<2>', iface)
  706. bat_hosts[hwaddress] = entry_name
  707. if 'fastd' in node_config.get ('roles', []):
  708. device_no = node_config.get ('id')
  709. for site in node_config.get ('sites', []):
  710. site_no = _get_site_no (sites_config, site)
  711. for network in ('intergw', 'nodes4', 'nodes6'):
  712. hwaddress = gen_batman_iface_mac (site_no, device_no, network)
  713. bat_hosts[hwaddress] = "%s/%s/%s" % (node_name, network, site)
  714. return bat_hosts
  715. # Generate eBGP session parameters for FFRL Transit from nodes pillar information.
  716. def get_ffrl_bgp_config (ifaces, proto):
  717. from ipcalc import IP
  718. _generate_ffrl_gre_tunnels (ifaces)
  719. sessions = {}
  720. for iface in sorted (ifaces):
  721. # We only care for GRE tunnels to the FFRL Backbone
  722. if not iface.startswith ('gre_ffrl_'):
  723. continue
  724. iface_config = ifaces.get (iface)
  725. # Search for IPv4/IPv6 prefix as defined by proto parameter
  726. local = None
  727. neighbor = None
  728. for prefix in iface_config.get ('prefixes', []):
  729. if (proto == 'v4' and '.' in prefix) or (proto == 'v6' and ':' in prefix):
  730. local = prefix.split ('/')[0]
  731. # Calculate neighbor IP as <local IP> - 1
  732. if proto == 'v4':
  733. neighbor = str (IP (int (IP (local)) - 1, version = 4))
  734. else:
  735. neighbor = str (IP (int (IP (local)) - 1, version = 6))
  736. break
  737. # Strip gre_ prefix iface name and use it as identifier for the eBGP session.
  738. name = re.sub ('gre_ffrl_', 'ffrl_', iface)
  739. sessions[name] = {
  740. 'local' : local,
  741. 'neighbor' : neighbor,
  742. 'bgp_local_pref' : iface_config.get ('bgp_local_pref', None),
  743. }
  744. return sessions
  745. # Get list of IP address configured on given interface on given node.
  746. #
  747. # @param: node_config Pillar node configuration (as dict)
  748. # @param: iface_name Name of the interface defined in pillar node config
  749. # OR name of VRF ("vrf_<something>") whichs ifaces are
  750. # to be examined.
  751. # @param: with_mask Don't strip the netmask from the prefix. (Default false)
  752. def get_node_iface_ips (node_config, iface_name, with_mask = False):
  753. ips = {
  754. 'v4' : [],
  755. 'v6' : [],
  756. }
  757. ifaces = node_config.get ('ifaces', {})
  758. ifaces_names = [ iface_name ]
  759. if iface_name.startswith ('vrf_'):
  760. # Reset list of ifaces_names to consider
  761. ifaces_names = []
  762. vrf = iface_name
  763. for iface, iface_config in ifaces.items ():
  764. # Ignore any iface NOT in the given VRF
  765. if iface_config.get ('vrf', None) != vrf:
  766. continue
  767. # Ignore any VEth pairs
  768. if iface.startswith ('veth'):
  769. continue
  770. ifaces_names.append (iface)
  771. try:
  772. for iface in ifaces_names:
  773. for prefix in ifaces[iface]['prefixes']:
  774. ip_ver = 'v6' if ':' in prefix else 'v4'
  775. if not with_mask:
  776. prefix = prefix.split ('/')[0]
  777. ips[ip_ver].append (prefix)
  778. except KeyError:
  779. pass
  780. return ips
  781. #
  782. # Get the lookback IP of the given node for the given proto
  783. #
  784. # @param node_config: Pillar node configuration (as dict)
  785. # @param node_id: Minion name / Pillar node configuration key
  786. # @param proto: { 'v4', 'v6' }
  787. def get_loopback_ip (node_config, node_id, proto):
  788. if proto not in [ 'v4', 'v6' ]:
  789. raise Exception ("get_loopback_ip(): Invalid proto: \"%s\"." % proto)
  790. if not proto in loopback_prefix:
  791. raise Exception ("get_loopback_ip(): No loopback_prefix configured for IP%s in ffno_net module!" % proto)
  792. if not 'id' in node_config:
  793. raise Exception ("get_loopback_ip(): No 'id' configured in pillar for node \"%s\"!" % node_id)
  794. # Every rule has an exception.
  795. # If there is a loopback_overwrite configuration for this node, use this instead of
  796. # the generated IPs.
  797. if 'loopback_override' in node_config:
  798. if proto not in node_config['loopback_override']:
  799. raise Exception ("get_loopback_ip(): No loopback_prefix configured for IP%s in node config / loopback_override!" % proto)
  800. return node_config['loopback_override'][proto]
  801. return "%s%s" % (loopback_prefix.get (proto), node_config.get ('id'))
  802. #
  803. # Get the router id (read: IPv4 Lo-IP) out of the given node config.
  804. def get_router_id (node_config, node_id):
  805. return get_loopback_ip (node_config, node_id, 'v4')
  806. # Compute minions OSPF interface configuration according to FFHO routing policy
  807. # See https://wiki.ffho.net/infrastruktur:vlans for information about Vlans
  808. def get_ospf_interface_config (node_config, grains_id):
  809. ospf_node_config = node_config.get ('ospf', {})
  810. ospf_interfaces = {}
  811. for iface, iface_config in node_config.get ('ifaces', {}).items ():
  812. # By default we don't speak OSPF on interfaces
  813. ospf_on = False
  814. # Defaults for OSPF interfaces
  815. ospf_config = {
  816. 'stub' : True, # Active/Passive interface
  817. 'cost' : 12345,
  818. # 'type' # Area type
  819. }
  820. # OSPF configuration for interface present?
  821. ospf_config_pillar = iface_config.get ('ospf', {})
  822. # Should be completely ignore this interface?
  823. if ospf_config_pillar.get ('ignore', False):
  824. continue
  825. # Local Gigabit Ethernet based connections (PTP or L2 subnets), cost 10
  826. if re.search (r'^(br-?|br\d+\.|vlan)10\d\d$', iface):
  827. ospf_on = True
  828. ospf_config['stub'] = False
  829. ospf_config['cost'] = 10
  830. ospf_config['desc'] = "Wired Gigabit connection"
  831. # VLL connection
  832. elif re.search (r'^vlan15\d\d$', iface):
  833. ospf_on = True
  834. ospf_config['stub'] = False
  835. ospf_config['cost'] = 20
  836. ospf_config['desc'] = "VLL connection"
  837. # WBBL connection
  838. elif re.search (r'^vlan20\d\d$', iface):
  839. ospf_on = True
  840. ospf_config['stub'] = False
  841. ospf_config['cost'] = 100
  842. ospf_config['desc'] = "WBBL connection"
  843. # Legacy WBBL connection
  844. elif re.search (r'^vlan22\d\d$', iface):
  845. ospf_on = True
  846. ospf_config['stub'] = False
  847. ospf_config['cost'] = 1000
  848. ospf_config['desc'] = "WBBL connection"
  849. # Management Vlans
  850. elif re.search (r'^vlan30\d\d$', iface):
  851. ospf_on = True
  852. ospf_config['stub'] = True
  853. ospf_config['cost'] = 10
  854. # OPS Vlans
  855. elif re.search (r'^vlan39\d\d$', iface):
  856. ospf_on = True
  857. ospf_config['stub'] = True
  858. ospf_config['cost'] = 10
  859. # Active OSPF on OpenVPN tunnels, cost 10000
  860. elif iface.startswith ('ovpn-'):
  861. ospf_on = True
  862. ospf_config['stub'] = False
  863. ospf_config['cost'] = 10000
  864. # Inter-Core links should have cost 5000
  865. if iface.startswith ('ovpn-cr') and grains_id.startswith ('cr'):
  866. ospf_config['cost'] = 5000
  867. # OpenVPN tunnels to EdgeRouters
  868. elif iface.startswith ('ovpn-er-'):
  869. ospf_config['type'] = 'broadcast'
  870. # Configure Out-of-band OpenVPN tunnels as stub interfaces,
  871. # so recursive next-hop lookups for OOB-BGP-session will work.
  872. elif iface.startswith ('oob-'):
  873. ospf_on = True
  874. ospf_config['stub'] = True
  875. ospf_config['cost'] = 1000
  876. # OSPF explicitly enabled for interface
  877. elif 'ospf' in iface_config:
  878. ospf_on = True
  879. # iface ospf parameters will be applied later
  880. # Go on if OSPF should not be actived
  881. if not ospf_on:
  882. continue
  883. # Explicit OSPF interface configuration parameters take precendence over generated ones
  884. for attr, val in ospf_config_pillar.items ():
  885. ospf_config[attr] = val
  886. # Convert boolean values to 'yes' / 'no' string values
  887. for attr, val in ospf_config.items ():
  888. if type (val) == bool:
  889. ospf_config[attr] = 'yes' if val else 'no'
  890. # Store interface configuration
  891. ospf_interfaces[iface] = ospf_config
  892. return ospf_interfaces
  893. # Return (possibly empty) subset of Traffic Engineering entries from 'te' pillar entry
  894. # relevenant for this minion and protocol (IPv4 / IPv6)
  895. def get_te_prefixes (te_node_config, grains_id, proto):
  896. te_config = {}
  897. for prefix, prefix_config in te_node_config.get ('prefixes', {}).items ():
  898. prefix_proto = 'v6' if ':' in prefix else 'v4'
  899. # Should this TE policy be applied on this node and is the prefix
  900. # of the proto we are looking for?
  901. if grains_id in prefix_config.get ('nodes', []) and prefix_proto == proto:
  902. te_config[prefix] = prefix_config
  903. return te_config
  904. def generate_DNS_entries (nodes_config, sites_config):
  905. import ipaddress
  906. forward_zone_name = ""
  907. forward_zone = []
  908. zones = {
  909. # <forward_zone_name>: [],
  910. # <rev_zone1_name>: [],
  911. # <rev_zone2_name>: [],
  912. # ...
  913. }
  914. # Fill zones dict with zones configured in DNS_zone_names at the top of this file.
  915. # Make sure the zone base names provided start with a leading . so the string
  916. # operations later can be done easily and safely. Proceed with fingers crossed.
  917. for entry, value in DNS_zone_names.items ():
  918. if entry == "forward":
  919. zone = value
  920. if not zone.startswith ('.'):
  921. zone = ".%s" % zone
  922. zones[zone] = forward_zone
  923. forward_zone_name = zone
  924. if entry in [ 'rev_v4', 'rev_v6' ]:
  925. for zone in value:
  926. if not zone.startswith ('.'):
  927. zone = ".%s" % zone
  928. zones[zone] = []
  929. # Process all interfaace of all nodes defined in pillar and generate forward
  930. # and reverse entries for all zones defined in DNS_zone_names. Automagically
  931. # put reverse entries into correct zone.
  932. for node_id in sorted (nodes_config):
  933. node_config = nodes_config.get (node_id)
  934. ifaces = get_interface_config (node_config, sites_config, node_id)
  935. for iface in sorted (ifaces):
  936. iface_config = ifaces.get (iface)
  937. # We only care for interfaces with IPs configured
  938. prefixes = iface_config.get ("prefixes", None)
  939. if prefixes == None:
  940. continue
  941. # Ignore any interface in $VRF
  942. if iface_config.get ('vrf', "") in [ 'vrf_external' ]:
  943. continue
  944. for prefix in sorted (prefixes):
  945. ip = ipaddress.ip_address (u'%s' % prefix.split ('/')[0])
  946. proto = 'v%s' % ip.version
  947. # The entry name is
  948. # <node_id> when interface 'lo'
  949. # <node_name>.srv.<residual> when interface 'srv' (or magically detected internal srv record)
  950. # <interface>.<node_id> else
  951. entry_name = node_id
  952. if iface != "lo":
  953. entry_name = "%s.%s" % (iface, node_id)
  954. elif iface == 'srv' or re.search (r'^(10.132.251|2a03:2260:2342:f251:)', prefix):
  955. entry_name = re.sub (r'^([^.]+)\.(.+)$', r'\g<1>.srv.\g<2>', entry_name)
  956. # Strip forward zone name from entry_name and store forward entry
  957. # with correct entry type for found IP address.
  958. forward_entry_name = re.sub (forward_zone_name, "", entry_name)
  959. forward_entry_name = re.sub (forward_zone_name, "", entry_name)
  960. forward_entry_typ = "A" if ip.version == 4 else "AAAA"
  961. forward_zone.append ("%s IN %s %s" % (forward_entry_name, forward_entry_typ, ip))
  962. # Find correct reverse zone, if configured and strip reverse zone name
  963. # from calculated reverse pointer name. Store reverse entry if we found
  964. # a zone for it. If no configured reverse zone did match, this reverse
  965. # entry will be ignored.
  966. for zone in zones:
  967. if ip.reverse_pointer.find (zone) > 0:
  968. PTR_entry = re.sub (zone, "", ip.reverse_pointer)
  969. zones[zone].append ("%s IN PTR %s." % (PTR_entry, entry_name))
  970. break
  971. return zones
  972. # Convert the CIDR network from the given prefix into a dotted netmask
  973. def cidr_to_dotted_mask (prefix):
  974. from ipcalc import Network
  975. return str (Network (prefix).netmask ())
  976. def is_subprefix (prefix, subprefix):
  977. from ipcalc import Network
  978. return subprefix in Network(prefix)
  979. # Return the network address of the given prefix
  980. def get_network_address (prefix, with_prefixlen = False):
  981. from ipaddress import ip_network
  982. net_h = ip_network (u'%s' % prefix, strict = False)
  983. network = str (net_h.network_address)
  984. if with_prefixlen:
  985. network += "/%s" % net_h.prefixlen
  986. return network