add omitted file

contrib/shadow/genconf.py

@@ -0,0 +1,113 @@
+#!/usr/bin/env python3
+
+import configparser
+import sys
+import os
+
+from xml.etree import ElementTree as etree
+
+getSetting = lambda s, name, fallback : name in s and s[name] or fallback
+
+shadowRoot = getSetting(os.environ, "SHADOW_ROOT", os.path.join(os.environ['HOME'], '.shadow'))
+
+libpath = 'libshadow-plugin-llarp.so'
+
+def nodeconf(conf, baseDir, name, ifname, port):
+    conf['netdb'] = {'dir': 'tmp-nodes'}
+    conf['router']  = {}
+    conf['router']['contact-file'] = os.path.join(baseDir, '{}.signed'.format(name))
+    conf['router']['ident-privkey'] = os.path.join(baseDir, '{}-ident.key'.format(name))
+    conf['router']['transport-privkey'] = os.path.join(baseDir, '{}-transport.key'.format(name))
+    conf['iwp-links'] = {ifname : port}
+    conf['iwp-connect'] = {}
+
+def addPeer(conf, baseDir, peer):
+    conf['iwp-connect'][peer] = os.path.join(baseDir, '{}.signed'.format(peer))
+
+def createNode(pluginName, root, peer):
+    node = etree.SubElement(root, 'host')
+    node.attrib['id'] = peer['name']
+    node.attrib['geocodehint'] = 'US'
+    app = etree.SubElement(node, 'process')
+    app.attrib['plugin'] = pluginName
+    app.attrib['time'] = '50'
+    app.attrib['arguments'] = peer['configfile']
+
+
+def makeBase(settings, name, id):
+    return {
+        'id': id,
+        'name' : name,
+        'contact-file' : os.path.join(getSetting(settings, 'baseDir', 'tmp'), '{}.signed'.format(name)),
+        'configfile' : os.path.join(getSetting(settings, 'baseDir', 'tmp'), '{}.ini'.format(name)),
+        'config': configparser.ConfigParser()
+    }
+
+def makeClient(settings, name, id, port):
+    peer = makeBase(settings, name, id)
+    nodeconf(peer['config'], getSetting(settings, 'baseDir', 'tmp'), name, '*', port)
+    return peer
+
+def makeSVCNode(settings, name, id, port):
+    peer = makeBase(settings, name, id)
+    nodeconf(peer['config'], getSetting(settings, 'baseDir', 'tmp'), name, 'eth0', port)
+    return peer
+
+
+def genconf(settings, outf):
+    root = etree.Element('shadow')
+    topology = etree.SubElement(root, 'topology')
+    topology.attrib['path'] = getSetting(settings, 'topology', os.path.join(shadowRoot, 'share', 'topology.graphml.xml'))
+
+
+    pluginName = getSetting(settings, 'name', 'llarpd')
+
+    kill = etree.SubElement(root, 'kill')
+    kill.attrib['time'] = getSetting(settings, 'runFor', '600')
+    
+    baseDir = getSetting(settings, 'baseDir', 'tmp')
+
+    if not os.path.exists(baseDir):
+        os.mkdir(baseDir)
+    
+    plugin = etree.SubElement(root, "plugin")
+    plugin.attrib['id'] = pluginName
+    plugin.attrib['path'] = libpath
+    basePort = getSetting(settings, 'base-port', 19000)
+    svcNodeCount = getSetting(settings, 'service-nodes', 20)
+    peers = list()
+    for nodeid in range(svcNodeCount):
+        peers.append(makeSVCNode(settings, 'svc-node-{}'.format(nodeid), str(nodeid), basePort + 1))
+        basePort += 1
+
+    # make all service nodes know each other
+    for peer in peers:
+        for nodeid in range(svcNodeCount):
+            if str(nodeid) != peer['id']:
+                addPeer(peer['config'], baseDir, 'svc-node-{}'.format(nodeid))
+        
+    # add client nodes
+    for nodeid in range(getSetting(settings, 'client-nodes', 200)):
+        peer = makeClient(settings, 'client-node-{}'.format(nodeid), str(nodeid), basePort +1)
+        peers.append(peer)
+        for p in range(getSetting(settings, 'client-connect-to', 1)):
+            addPeer(peer['config'], baseDir, 'svc-node-{}'.format((p + nodeid) % svcNodeCount))
+        basePort += 1
+        
+
+    # generate xml and settings files
+    for peer in peers:
+        createNode(pluginName, root, peer)
+
+        with open(peer['configfile'], 'w') as f:
+            peer['config'].write(f)
+
+    # render
+    outf.write(etree.tostring(root).decode('utf-8'))
+
+if __name__ == '__main__':
+    settings = {
+        'topology': os.path.join(shadowRoot, 'share', 'topology.graphml.xml')    
+    }
+    with open(sys.argv[1], 'w') as f:
+        genconf(settings, f)