added datamgmt and retiolum

1 files changed, 101 insertions, 0 deletions

diff --git a/retiolum/hosts/.scripts/datamgmt.py b/retiolum/hosts/.scripts/datamgmt.py
new file mode 100755
index 00000000..0d8f056d
--- /dev/null
+++ b/retiolum/hosts/.scripts/datamgmt.py
@@ -0,0 +1,101 @@
+#!/usr/bin/python2
+import sqlite3
+import os, sys
+
+def create_db(netname):
+    conn = sqlite3.connect("/etc/tinc/"+ netname + "/hosts.sqlite")
+    db = conn.cursor()
+    db.execute('''create table hosts(hostname text, subnet text, address text, port text, r_pub text, t_pub text)''')
+    conn.commit()
+    db.close()
+
+def PubInDb(netname, hostname):
+    pubfile = open("/etc/tinc/" + netname + "/hosts/.pubkeys/" + hostname, "r")
+    publines = pubfile.readlines()
+    pubfile.close()
+    pubkey = ""
+    for i in range(len(publines)):
+        pubkey += publines[i]
+    print(pubkey)
+    conn = sqlite3.connect("/etc/tinc/" + netname + "/hosts.sqlite")
+    c = conn.cursor()
+    tupel = [pubkey, hostname]
+    c.execute('UPDATE hosts SET r_pub=? WHERE hostname=?', tupel)
+    conn.commit()
+    c.close()     
+
+def HostInDb(netname, hostname):
+    hostFile = open("/etc/tinc/"+ netname + "/hosts/" + hostname, "r")
+    hostlines = hostFile.readlines()
+    hostFile.close()
+    conn = sqlite3.connect("/etc/tinc/"+ netname + "/hosts.sqlite")
+    db = conn.cursor()
+
+    lines = 0
+    Subnet = ""
+    Tinc_pub_key = ""
+    Address = "#Address = none\n"
+    Port = "Port =  655\n"
+    while lines < len(hostlines):
+        if hostlines[lines][0:4] == "Addr":
+            Address = hostlines[lines]
+            print Address
+        if hostlines[lines][0:4] == "Subn":
+            Subnet = hostlines[lines]
+            print Subnet
+        if hostlines[lines][0:4] == "Port":
+            Port =  hostlines[lines]
+            print Port
+        if hostlines[lines][0:10]  == "-----BEGIN":
+            Tinc_pub_key_array = hostlines[lines:lines+8]
+            line = 0
+            Tinc_pub_key = ""
+            while line < len(Tinc_pub_key_array):
+                Tinc_pub_key += Tinc_pub_key_array[line]
+                line += 1
+            print Tinc_pub_key
+        lines += 1
+    if not(Subnet == "" or Tinc_pub_key == ""):
+        tupel = (hostname, Subnet, Address, Port, Tinc_pub_key )
+        conn = sqlite3.connect("/etc/tinc/"+ netname + "/hosts.sqlite")
+        db = conn.cursor()
+        db.execute('insert into hosts values(?,?,?,?,"none",?)', tupel)
+        conn.commit()
+        db.close()
+    
+def ShowDb(netname):
+    conn = sqlite3.connect("/etc/tinc/" + netname + "/hosts.sqlite")
+    db = conn.cursor()
+    db.execute("select * from hosts")
+    for hosts in db:
+        for i in range(len(hosts)):
+            print hosts[i]
+    db.close
+
+def DirInDb(netname):
+    #the normal tinc files are read into the sqlite
+    files = os.listdir("/etc/tinc/" + netname + "/hosts/")
+    file_n = 0
+    while file_n < len(files):
+        if files[file_n][0] == ".":
+            files.remove(files[file_n])
+            file_n -= 1
+        file_n += 1
+    for filename in files:
+        HostInDb(netname, filename)
+    #the pubkeys are included into the sqlite
+    pubfiles = os.listdir("/etc/tinc/" + netname + "/hosts/.pubkeys/")
+    for pubname in pubfiles:
+        PubInDb(netname, pubname)
+
+
+#Program start here
+netname = sys.argv[1]
+
+try: 
+    os.remove("/etc/tinc/" + netname + "/hosts.sqlite")
+except: 
+    print("no hosts.sqlite found")
+
+create_db(netname)
+DirInDb(netname)