aboutsummaryrefslogtreecommitdiffstats
path: root/helpers
diff options
context:
space:
mode:
authorficus <ficus@robocracy.org>2012-09-15 19:46:49 +0200
committerficus <ficus@robocracy.org>2012-09-15 19:46:49 +0200
commit42725426ec02d359b67cd8d665afe17882d38958 (patch)
tree148c4082998f7d0b82ec4f22515d7626fca6a0e7 /helpers
parenta304d7a56563b16ca98d9b45ebed521c52f6347c (diff)
downloadtui-42725426ec02d359b67cd8d665afe17882d38958.tar.gz
tui-42725426ec02d359b67cd8d665afe17882d38958.zip
more WIP
Diffstat (limited to 'helpers')
-rw-r--r--helpers/netif.py188
-rw-r--r--helpers/sysstatus.py26
-rw-r--r--helpers/tor.py11
-rw-r--r--helpers/util.py2
4 files changed, 218 insertions, 9 deletions
diff --git a/helpers/netif.py b/helpers/netif.py
index 6d7a194..77f710c 100644
--- a/helpers/netif.py
+++ b/helpers/netif.py
@@ -1,5 +1,6 @@
import os
+import augeas
from util import *
@@ -20,6 +21,10 @@ def parse_ip(ifname):
for k, v in val_indexes.iteritems():
if len(iplinkl) > v and iplinkl[v] == k:
d[k] = iplinkl[v+1]
+ if 'LOWER_UP' in iplinkl[2][1:-1].split(","):
+ d['state'] = "RUNNING"
+ elif 'NO-CARRIER' in iplinkl[2][1:-1].split(","):
+ d['state'] = "DISCONNECTED"
d['ipv4addrs'] = list()
d['ipv6addrs'] = list()
for l in ipinfo[1:]:
@@ -38,15 +43,121 @@ def parse_ip(ifname):
addr=l[1].split('/')[0],
prefix=int(l[1].split('/')[1]),
scope=l[3]))
- print d['ipv6addrs']
return d
-def get_wan_status(ifname='eth1'):
+def parse_iw(ifname):
+ """
+ Example `iw dev wlan0 link` string (sic):
+
+ Connected to c0:25:06:51:22:9b (on wlan0)
+ SSID: fleischfressendepflanze
+ freq: 2427
+ RX: 73744193 bytes (456838 packets)
+ TX: 3269174 bytes (19709 packets)
+ signal: -44 dBm
+ tx bitrate: 72.2 MBit/s MCS 7 short GI
+
+ bss flags:
+ dtim period: 0
+ beacon int: 100
+ """
+ d = dict()
+ iwinfo = cli_read_lines('iw dev %s link' % ifname)
+ if 0 == len(iwinfo):
+ raise KeyError('No such interface: ' + ifname)
+ if iwinfo[0].strip() == "Not connected.":
+ d['radio_state'] = "disabled"
+ return d
+ else:
+ d['radio_state'] = "enabled"
+ for l in iwinfo:
+ l = l.strip()
+ if l.startswith("SSID:"):
+ d['ssid'] = l[6:].strip()
+ elif l.startswith("freq:"):
+ d['freq'] = "%sMHz" % l.split()[-1].strip()
+ elif l.startswith("signal:"):
+ d['signal_dbm'] = l.split()[1]
+ elif l.startswith("tx bitrate:"):
+ d['signal_throughput'] = ' '.join(l.split()[2:4])
+ return d
+
+def read_augeas_ifinfo(ifname):
+ d = dict()
+ aug = augeas.Augeas(flags=augeas.Augeas.NO_MODL_AUTOLOAD)
+ aug.set("/augeas/load/Interfaces/lens", "Interfaces.lns")
+ aug.set("/augeas/load/Interfaces/incl", "/etc/network/interfaces")
+ aug.load()
+ for iface in aug.match("/files/etc/network/interfaces/iface"):
+ if aug.get(iface) == ifname:
+ if aug.get(iface + "/family") == 'inet':
+ d['ipv4method'] = aug.get(iface + "/method")
+ if d['ipv4method'] == 'manual':
+ d['ipv4method'] = 'disabled'
+ d['ipv4addr'] = aug.get(iface + "/address")
+ d['ipv4netmask'] = aug.get(iface + "/netmask")
+ d['ipv4gateway'] = aug.get(iface + "/gateway")
+ d['ipv4mtu'] = aug.get(iface + "/mtu")
+ d['ipv4mac'] = aug.get(iface + "/hwaddress")
+ elif aug.get(iface + "/family") == 'inet6':
+ # handle ipv6 stuff
+ pass
+ aug.close()
+ return d
+ aug.close()
+ return None
+
+def write_augeas_ifinfo(ifname, settings, method='disabled'):
+ d = dict()
+ aug = augeas.Augeas(flags=augeas.Augeas.NO_MODL_AUTOLOAD)
+ aug.set("/augeas/load/Interfaces/lens", "Interfaces.lns")
+ aug.set("/augeas/load/Interfaces/incl", "/etc/network/interfaces")
+ aug.load()
+ path = None
+ for iface in aug.match("/files/etc/network/interfaces/iface"):
+ if aug.get(iface) == ifname and aug.get(iface + "/family") == 'inet':
+ path = iface
+ if not path:
+ # insert iface
+ if len(aug.match("/files/etc/network/interfaces/iface")) == 0:
+ # no interfaces at all, insert wherever
+ path = "/files/etc/network/interfaces/iface"
+ aug.set(path, ifname)
+ else:
+ aug.insert("/files/etc/network/interfaces/iface", "iface",
+ before=False)
+ path = aug.match("/files/etc/network/interfaces/iface")[-1]
+ aug.set(path, ifname)
+ assert path, "require path to be set"
+ aug.set(path + "/family", 'inet')
+ if method == 'disabled':
+ aug.set(path + "/method", 'manual')
+ aug.remove(path + "/address")
+ aug.remove(path + "/netmask")
+ aug.remove(path + "/gateway")
+ elif method == 'dhcp':
+ aug.set(path + "/method", 'dhcp')
+ aug.remove(path + "/address")
+ aug.remove(path + "/netmask")
+ aug.remove(path + "/gateway")
+ elif method == 'static':
+ aug.set(path + "/method", 'static')
+ aug.set(path + "/address", str(settings['ipv4addr']))
+ aug.set(path + "/netmask", str(settings['ipv4netmask']))
+ aug.set(path + "/gateway", str(settings['ipv4gateway']))
+ else:
+ raise ValueError("unrecognized network interface method: " + method)
+ print "committing with augeas..."
+ aug.save()
+ print "augeas errors: %s" % aug.get("/augeas/error")
+ aug.close()
+
+def get_wan_status(ifname='eth0'):
d = dict()
try:
d.update(parse_ip(ifname))
except KeyError:
- None
+ return None
return d
def get_lan_status(ifname='eth0'):
@@ -54,15 +165,80 @@ def get_lan_status(ifname='eth0'):
try:
d.update(parse_ip(ifname))
except KeyError:
- None
+ return None
return d
-def get_wireless_status(ifname='wlan0'):
+def get_wifi_status(ifname='wlan0'):
d = dict()
try:
d.update(parse_ip(ifname))
except KeyError, ke:
- raise ke
return None
+ d.update(parse_iw(ifname))
return d
+def get_wan_settings(ifname='eth0'):
+ return read_augeas_ifinfo(ifname)
+
+def save_wan_settings(form, ifname='eth0'):
+ write_augeas_ifinfo(ifname, method=form['ipv4method'], settings=form)
+ if form['ipv4method'] == 'disabled':
+ print "ifdown..."
+ os.system("ifdown %s" % ifname)
+ else:
+ print "ifup..."
+ os.system("ifdown %s" % ifname)
+ os.system("ifup %s &" % ifname)
+
+def get_lan_settings(ifname='eth0'):
+ d = read_augeas_ifinfo(ifname)
+ return d
+
+def save_lan_settings(ifname='eth0'):
+ write_augeas_ifinfo(ifname, method=form['ipv4method'], settings=form)
+ if form['ipv4method'] == 'disabled':
+ print "ifdown..."
+ os.system("ifdown %s" % ifname)
+ else:
+ print "ifup..."
+ os.system("ifdown %s" % ifname)
+ os.system("ifup %s &" % ifname)
+
+def get_wifi_settings(ifname='wlan0'):
+ #d = read_augeas_ifinfo(ifname)
+ d = dict()
+ if not d:
+ return d
+ d.update(dict()) # extra wireless settings
+ return d
+
+def save_wifi_settings(ifname='eth0'):
+ pass
+
+def is_valid_ipv4(s):
+ # TODO: this is a hack
+ l = s.split('.')
+ if not len(l) == 4:
+ return False
+ try:
+ l = map(int, l)
+ except ValueError:
+ return False
+ if l[0] > 255 or l[1] > 255 or l[2] > 255 or l[3] > 255:
+ return False
+ if l[0] == 0 or l[3] == 0:
+ return False
+ return True
+
+def is_valid_ipv4mask(s):
+ # TODO: this is a hack
+ l = s.split('.')
+ if not len(l) == 4:
+ return False
+ try:
+ l = map(int, l)
+ except ValueError:
+ return False
+ if l[0] > 255 or l[1] > 255 or l[2] > 255 or l[3] > 255:
+ return False
+ return True
diff --git a/helpers/sysstatus.py b/helpers/sysstatus.py
index b2eec6f..933f625 100644
--- a/helpers/sysstatus.py
+++ b/helpers/sysstatus.py
@@ -12,6 +12,28 @@ def get_system_status():
d['uptime'] = cli_read('uptime')
return d
+def get_resources_status():
+ d = dict()
+ disk_info = cli_read_lines('df -h /home')[1].split()
+ d['disk_used'] = disk_info[2]
+ d['disk_avail'] = disk_info[1]
+ d['disk_percent'] = int(disk_info[4][:-1])
+
+ ram_info = cli_read_lines('free -m')[1].split()
+ d['ram_used'] = "%sMB" % ram_info[2]
+ d['ram_avail'] = "%sMB" % ram_info[1]
+ d['ram_percent'] = int(float(ram_info[2])/float(ram_info[1]) * 100.0)
+
+ d['cpu_cores'] = 1
+ for l in open('/proc/cpuinfo', 'r'):
+ if l.split(":")[0].strip() == "cpu cores":
+ d['cpu_cores'] = int(l.split(":")[-1].strip())
+ break
+ cpu_info = cli_read('uptime')
+ d['cpu_load'] = cpu_info.split()[-3].strip(',')
+ d['cpu_percent'] = int(float(d['cpu_load'])/float(d['cpu_cores']) * 100.0)
+ return d
+
def get_dmesg():
try:
return cli_read('dmesg')
@@ -21,14 +43,14 @@ def get_dmesg():
def get_authlog():
try:
with open('/var/log/auth.log') as f:
- return '\n'.join(f.readlines())
+ return ''.join(f.readlines())
except IOError:
return None
def get_syslog():
try:
with open('/var/log/syslog') as f:
- return '\n'.join(f.readlines())
+ return ''.join(f.readlines())
except IOError:
return None
diff --git a/helpers/tor.py b/helpers/tor.py
index e69de29..438cb5d 100644
--- a/helpers/tor.py
+++ b/helpers/tor.py
@@ -0,0 +1,11 @@
+
+def get_tor_status():
+ d = dict()
+ d['state'] = 'DISABLED'
+ return d
+
+def get_tor_settings():
+ return dict()
+
+def save_tor_settings():
+ pass
diff --git a/helpers/util.py b/helpers/util.py
index 46030bf..229d975 100644
--- a/helpers/util.py
+++ b/helpers/util.py
@@ -11,7 +11,7 @@ def cli_read_lines(cmd):
def fs_read(path):
with open(path, 'r') as f:
- return '\n'.join(f.readlines()).strip()
+ return ''.join(f.readlines())
def prefix_to_ipv4_mask(prefixlen):
assert(prefixlen >= 0)