1 files changed, 146 insertions, 0 deletions

diff --git a/generate_tox_bootstrap.py b/generate_tox_bootstrap.py
new file mode 100644
index 0000000..44281c9
--- /dev/null
+++ b/generate_tox_bootstrap.py
@@ -0,0 +1,146 @@
+#!/usr/bin/python
+# pip install jinja2 requests
+
+import datetime
+import jinja2
+import json
+import requests
+import socket
+
+json_url = 'https://nodes.tox.chat/json'
+
+tox_bootstrap_template = """
+/*
+ * Generated with generate_tox_bootstrap.py by GDR!
+ * from {{ json_url }} on {{ now }}
+ */
+struct bootstrap_node {
+    char *address;
+    uint16_t port;
+    uint8_t key[32];
+} bootstrap_nodes[] = {
+{% for node in nodes %}
+    {
+        "{{ node.ipv4 }}",
+        {{ node.port }},
+        {
+            {{ node.public_key|toxtoc }}
+        }
+    },
+{% endfor %}
+    {
+        "176.31.28.63",
+        6881,
+        {
+            0x0B, 0x6C, 0x7A, 0x93, 0xA6, 0xD8, 0xC0, 0xEB, 0x44, 0x96, 0x5C, 0x4A, 0x85, 0xCB, 0x88, 0xBA, 
+            0x75, 0x71, 0x17, 0x0F, 0xE2, 0xDB, 0x3D, 0xEA, 0x10, 0x71, 0x3E, 0x48, 0x55, 0x9A, 0x55, 0x4D
+        }
+    },
+};
+
+struct bootstrap_node tcp_relays[] = {
+{% for node in relays %}
+    {
+        "{{ node.ipv4 }}",
+        {{ node.port }},
+        {
+            {{ node.public_key|toxtoc }}
+        }
+    },
+{% endfor %}
+    {
+        "176.31.28.63",
+        465,
+        {
+            0x0B, 0x6C, 0x7A, 0x93, 0xA6, 0xD8, 0xC0, 0xEB, 0x44, 0x96, 0x5C, 0x4A, 0x85, 0xCB, 0x88, 0xBA, 
+            0x75, 0x71, 0x17, 0x0F, 0xE2, 0xDB, 0x3D, 0xEA, 0x10, 0x71, 0x3E, 0x48, 0x55, 0x9A, 0x55, 0x4D
+        }
+    },
+};
+"""
+
+def toxtoc(value):
+    """
+    A Jinja2 filter to turn a ToxID into two lines of C bytes
+    """
+    def get_16_bytes(value):
+        """
+        Generate 1 line of C code - 16 bytes
+        @param value a hex string of length 32 (32 hex chars)
+        """
+        if len(value) <> 32:
+            raise ValueError('%r is not a 32-char string')
+
+        rv = ""
+
+        for i in range(16):
+            rv += "0x%s" % value[2*i : 2*i+2]
+            if i < 15:
+                rv += ", "
+
+        return rv
+
+    rv = get_16_bytes(value[:32]) + \
+         ",\n" + (12*' ') + \
+         get_16_bytes(value[32:])
+
+    return rv
+
+class Loader(jinja2.BaseLoader):
+    def get_source(self, environment, template):
+        return tox_bootstrap_template, 'tox_bootstrap_template', True
+
+if __name__ == "__main__":
+    r = requests.get(json_url)
+    data = r.json()
+    if 'nodes' not in data:
+        raise ValueError('nodes element not in JSON')
+
+    nodes = []
+    tcp_relays = []
+
+    for elem in data['nodes']:
+        node = {}
+        if 'ipv4' not in elem or 'port' not in elem or 'public_key' not in elem:
+            print "SKIPPING", elem
+            continue
+        
+        if len(elem['public_key']) <> 64:
+            print "Bad public key %s, skipping!" % elem['public_key']
+            continue
+
+        node['port'] = int(elem['port'])
+        node['public_key'] = elem['public_key']
+
+        try:
+            socket.inet_aton(elem['ipv4'])
+            node['ipv4'] = elem['ipv4']
+        except socket.error:
+            # IPv4 is not numeric, let's try resolving
+            try:
+                print "RESOLVING", elem['ipv4']
+                node['ipv4'] = socket.gethostbyname(elem['ipv4'])
+            except socket.error:
+                print "Could not resolve ipv4: %s, skipping!" % elem['ipv4']
+                continue
+
+        if 'status_udp' in elem and elem['status_udp']:
+            nodes.append(node)
+
+        if 'tcp_ports' in elem and elem['tcp_ports'] and \
+           'status_tcp' in elem and elem['status_tcp']:
+            for port in elem['tcp_ports']:
+                relay = dict(node)
+                try:
+                    relay['port'] = int(port)
+                except ValueError:
+                    continue
+
+                tcp_relays.append(relay)
+
+    env = jinja2.Environment(loader=Loader())
+    env.filters['toxtoc'] = toxtoc
+    template = env.get_template('tox_bootstrap_template')
+    tox_bootstrap_h = template.render(nodes=nodes, now=datetime.datetime.now(), json_url=json_url, relays=tcp_relays)
+    open('tox_bootstrap.h', 'w').write(tox_bootstrap_h)
+