def resetwlan(interface='wlan0'):
pilib.log(pilib.networklog, 'Resetting ' + interface + ' . ', 3, pilib.networkloglevel)
try:
subprocess.check_output(['/sbin/ifdown', '--force', interface], stderr=subprocess.PIPE)
subprocess.call(['/sbin/ifup', interface], stderr=subprocess.PIPE)
except Exception, e:
pilib.log(pilib.networklog, 'Error resetting ' + interface + ' : ' + str(e), 0, pilib.networkloglevel)
def killapservices():
pilib.log(pilib.networklog, 'Killing AP Services. ', 1, pilib.networkloglevel)
try:
killhostapd()
except:
pilib.log(pilib.networklog, 'Error killing hostapd. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Killed hostapd. ', 3, pilib.networkloglevel)
try:
killdhcpserver()
except:
pilib.log(pilib.networklog, 'Error killing dhcp server. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Successfully killed dhcp server. ', 3, pilib.networkloglevel)
def readhardwarefileintoversions():
import pilib
devicefile = '/var/wwwsafe/devicedata'
try:
file = open(devicefile)
lines = file.readlines()
devicedict = {}
for line in lines:
split = line.split(':')
try:
devicedict[split[0].strip()] = split[1].strip()
except:
pilib.log(pilib.syslog, 'Device data parse error', 1, pilib.sysloglevel)
pilib.sqlitequery(pilib.systemdatadatabase,
pilib.makesqliteinsert('versions', ['hardware', devicedict['hardware']], ['item', 'version']))
except:
pilib.log(pilib.syslog, 'Cannot find devicedata file to parse', 1,
pilib.sysloglevel)
def runIPTables(mode, flush=True):
import pilib
if flush:
pilib.log(pilib.networklog, 'Flushing IPTables', 2, pilib.networkloglevel)
flushIPTables()
if mode == 'eth0wlan0bridge':
pilib.log(pilib.networklog, 'Running eth0wlan0 bridge IPTables', 2, pilib.networkloglevel)
runeth0wlan0bridgeIPTables()
elif mode == 'wlan0wlan1bridge':
pilib.log(pilib.networklog, 'Running wlan0wlan1 bridge IPTables', 2, pilib.networkloglevel)
runwlan0wlan1bridgeIPTables()
elif mode == 'wlan1wlan0bridge':
pilib.log(pilib.networklog, 'Running wlan1wlan0 bridge IPTables', 2, pilib.networkloglevel)
runwlan1wlan0bridgeIPTables()
def replaceifaceparameters(iffilein, iffileout, iface, parameternames, parametervalues):
file = open(iffilein)
lines = file.readlines()
writestring = ''
ifacename = None
for line in lines:
if line.find('iface') >= 0 > line.find('default'):
# we are at an iface stanza beginning
foundname = line[6:11].strip()
ifacename = foundname
if ifacename == iface:
# do our replace
for parametername, parametervalue in zip(parameternames, parametervalues):
if line.find(parametername) > 0:
split = line.split(' ')
# We find where the parameter is, then assume the value is the
# next position. We then trim everything past the parameter
# This safeguards against whitespace at the end of lines creating problems.
index = split.index(parametername)
split[index + 1] = str(parametervalue) + '\n'
line = ' '.join(split[0:index + 2])
writestring += line
try:
myfile = open(iffileout, 'w')
except:
pilib.log(pilib.networklog, 'Error opening interface file. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Interface file read successfully. ', 3, pilib.networkloglevel)
try:
myfile.write(writestring)
except:
pilib.log(pilib.networklog, 'Error writing interface file. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Interface file written successfully. ', 3, pilib.networkloglevel)
pilib.log(pilib.networklog, 'Write string: ' + writestring, 3, pilib.networkloglevel)
def getwpasupplicantconfig(conffile='/etc/wpa_supplicant/wpa_supplicant.conf'):
class data():
pass
file = open(conffile)
lines = file.readlines()
header = ''
tail = ''
datalines = []
readheader = True
readbody = False
readtail = False
for line in lines:
if readheader:
header = header + line
if '}' in line:
pilib.log(pilib.networklog, 'Ending supplicant parse. ', 5, pilib.networkloglevel)
readbody = False
readtail = True
if readbody:
datalines.append(line)
if readtail:
tail = tail + line
if '{' in line:
pilib.log(pilib.networklog, 'Beginning supplicant parse. ', 5, pilib.networkloglevel)
readheader = False
readbody = True
datadict = {}
for line in datalines:
split = line.split('=')
datadict[split[0].strip()] = split[1].strip()
returndict = data()
returndict.header = header
returndict.tail = tail
returndict.data = datadict
return returndict
def watchdoghamachi(pingip):
from netfun import runping, restarthamachi, killhamachi
import pilib
try:
# Turns out this is not going to work, as when it hangs, it hangs hard
# hamachistatusdata = gethamachistatusdata()
# So instead, we are going to test with a ping to another member on the network that
# should always be online. This of course means that we have to make sure that it is, in fact, always
# online
pingtimes = runping(pingip, numpings=5)
pingmax = max(pingtimes)
pingmin = min(pingtimes)
pingave = sum(pingtimes)/len(pingtimes)
# if hamachistatusdata['status'] not in ['logged in']:
if pingave == 0 or pingave > 3000:
pilib.log(pilib.networklog, 'Pingtime unacceptable: ' + str(pingave) + '. ', 1, pilib.networkloglevel)
pilib.setsinglevalue(pilib.controldatabase, 'systemstatus', 'hamachistatus', 0)
pilib.log(pilib.networklog, 'Restarting Hamachi. ', 1, pilib.networkloglevel)
killhamachi()
restarthamachi()
else:
if pingmax > 3000 or pingmin <= 0:
pilib.log(pilib.syslog, 'Hamachi lives, with issues: ' + str(pingtimes), 3, pilib.sysloglevel)
else:
pilib.setsinglevalue(pilib.controldatabase, 'systemstatus', 'hamachistatus', 1)
pilib.log(pilib.networklog, 'Hamachi appears fine. ', 3, pilib.networkloglevel)
except Exception as e:
pilib.log(pilib.networklog, 'Error checking Hamachi with message: ' + e.message, 1, pilib.networkloglevel)
killhamachi()
restarthamachi()
def processsystemflags(systemflags=None):
import pilib
from pilib import log, syslog, sysloglevel
if not systemflags:
systemflags = pilib.readalldbrows(pilib.systemdatadatabase, 'systemflags')
flagnames = []
flagvalues = []
for flag in systemflags:
flagnames.append(flag['name'])
flagvalues.append(flag['value'])
stop = False
if 'reboot' in flagnames:
if flagvalues[flagnames.index('reboot')]:
stop = True
pilib.setsinglevalue(pilib.systemdatadatabase, 'systemflags', 'value', 0, "name='reboot'")
import subprocess
log(syslog, 'Rebooting for system flag', 0, sysloglevel)
subprocess.call(['/sbin/reboot'])
if 'netconfig' in flagnames:
if flagvalues[flagnames.index('netconfig')]:
stop = True
pilib.setsinglevalue(pilib.systemdatadatabase, 'systemflags', 'value', 0, "name='netconfig'")
from netconfig import runconfig
log(syslog, 'Restarting network configuration', 0, sysloglevel)
runconfig()
if 'updateiicontrollibs' in flagnames and not stop:
if flagvalues[flagnames.index('updateiicontrollibs')]:
stop = True
pilib.setsinglevalue(pilib.systemdatadatabase, 'systemflags', 'value', 0, 'name=\'updateiicontrollibs\'')
from misc.gitupdatelib import updateiicontrollibs
log(syslog, 'Updating iicontrollibs', 0, sysloglevel)
updateiicontrollibs(True)
if 'updatecupidweblib' in flagnames and not stop:
if flagvalues[flagnames.index('updatecupidweblib')]:
stop = True
pilib.setsinglevalue(pilib.systemdatadatabase, 'systemflags', 'value', 0, 'name=\'updatecupidweblib\'')
from misc.gitupdatelib import updatecupidweblib
log(syslog, 'Updating cupidweblib', 0, sysloglevel)
updatecupidweblib(True)
def processapoverride(pin):
pilib.log(pilib.networklog, "Reading GPIO override on pin " + str(pin) + '. ', 3, pilib.networkloglevel)
pilib.log(pilib.syslog, "Reading GPIO override on pin " + str(pin) + '. ', 2, pilib.sysloglevel)
import RPi.GPIO as GPIO
import pilib
try:
GPIO.setmode(GPIO.BCM)
GPIO.setwarnings(False)
GPIO.setup(pin, GPIO.IN, pull_up_down=GPIO.PUD_UP)
except:
pilib.log(pilib.networklog, "Error reading GPIO", 3, pilib.networkloglevel)
else:
# jumper in place = input off, --> AP mode
if not GPIO.input(pin):
pilib.log(pilib.networklog, "GPIO On. Setting AP Mode.", 3, pilib.networkloglevel)
pilib.setsinglevalue(pilib.systemdatadatabase, 'netconfig', 'mode', 'ap')
def startapservices(interface='wlan0'):
from time import sleep
try:
# We name the file by the interfae. This way when we pgrep, we know we're running AP on the right interface
hostapdfilename = '/etc/hostapd/hostapd' + interface + '.conf'
updatehostapd(path=hostapdfilename, interface=interface)
subprocess.call(['/usr/sbin/hostapd', '-B', hostapdfilename])
except:
pilib.log(pilib.networklog, 'Error starting hostapd. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Started hostapd without error. ', 3, pilib.networkloglevel)
sleep(1)
try:
updatedhcpd(path='/etc/dhcp/dhcpd.conf', interface=interface)
subprocess.call(['/usr/sbin/service', 'isc-dhcp-server', 'start'])
except:
pilib.log(pilib.networklog, 'Error starting dhcp server. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Started dhcp server without error. ', 3, pilib.networkloglevel)
def setapmode(interface='wlan0', netconfig=None):
pilib.log(pilib.networklog, 'Setting ap mode for interface ' + interface, 1, pilib.networkloglevel)
try:
if interface == 'wlan0':
subprocess.call(['/bin/cp', '/usr/lib/iicontrollibs/misc/interfaces/interfaces.ap', '/etc/network/interfaces'])
elif interface == 'wlan0wlan1':
subprocess.call(['/bin/cp', '/usr/lib/iicontrollibs/misc/interfaces/interfaces.wlan0dhcp.wlan1cupidwifi', '/etc/network/interfaces'])
elif interface == 'wlan1wlan0':
subprocess.call(['/bin/cp', '/usr/lib/iicontrollibs/misc/interfaces/interfaces.wlan1dhcp.wlan0cupidwifi', '/etc/network/interfaces'])
except:
pilib.log(pilib.networklog, 'Error copying network configuration file. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Copied network configuration file successfully. ', 3, pilib.networkloglevel)
killapservices()
resetwlan()
startapservices(interface)
def writesupplicantfile(filedata, filepath='/etc/wpa_supplicant/wpa_supplicant.conf'):
writestring = ''
supplicantfilepath = '/etc/wpa_supplicant/wpa_supplicant.conf'
# We are hard-coding this, as this will unfortunately propagate a mangled file
subprocess.call(['/bin/cp', '/usr/lib/iicontrollibs/misc/wpa_supplicant.conf', supplicantfilepath])
# iterate over fed data
ssid = 'not found'
psk = 'not found'
for key, value in filedata.data.items():
if key == 'ssid':
ssid=value
elif key == 'psk':
psk = value
file = open(supplicantfilepath)
lines = file.readlines()
writestring = ''
ifacename = None
for line in lines:
if line.find('psk=') >= 0:
# insert our iface parameters
try:
writestring +='psk=' + psk + '\n'
except:
writestring +='psk=error'
elif line.find('ssid=') >= 0:
# insert our iface parameters
try:
writestring +='ssid=' + ssid + '\n'
except:
writestring +='psk=error'
else:
writestring += line
pilib.log(pilib.networklog, 'Writing supplicant file. ', 1, pilib.networkloglevel)
try:
myfile = open(filepath, 'w')
myfile.write(writestring)
except:
pilib.log(pilib.networklog, 'Error writing supplicant file. ', 1, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Supplicant file written. ', 3, pilib.networkloglevel)
def runsystemstatus(runonce=False):
import pilib
import time
import netconfig
from misc.gitupdatelib import updategitversions
# This doesn't update git libraries. It checks current versions and updates the database
try:
pilib.log(pilib.syslog, 'Checking git versions', 3, pilib.sysloglevel)
updategitversions()
except:
pilib.log(pilib.syslog, 'Error in git version check', 0, pilib.sysloglevel)
else:
pilib.log(pilib.syslog, 'Git version check complete', 3, pilib.sysloglevel)
systemstatus = pilib.readalldbrows(pilib.controldatabase, 'systemstatus')[0]
## Read wireless config via iwconfig
# this is breaking systemstatus for some reason
# updateiwstatus()
## Read current netstatus
lastnetstatus={}
try:
lastnetstatus = pilib.readonedbrow(pilib.systemdatadatabase, 'netstatus')[0]
except:
pilib.log(pilib.networklog, 'Error reading network status. ', 1, pilib.networkloglevel)
else:
pilib.log(pilib.syslog, 'Completed network status. ', 3, pilib.networkloglevel)
# Poll netstatus and return data
allnetstatus = updatenetstatus(lastnetstatus)
# wpastatusdict = allnetstatus['wpastatusdict']
# Keep reading system status?
while systemstatus['systemstatusenabled']:
currenttime = pilib.gettimestring()
pilib.setsinglevalue(pilib.controldatabase, 'systemstatus', 'lastsystemstatuspoll', pilib.gettimestring())
starttime = time.time()
pilib.log(pilib.syslog, 'System status routine is starting. ', 3,
pilib.sysloglevel)
"""
Check all network statuses. The goal here is to totally decouple status read and reconfigure
When we need to check all status data, we'll have it either in a dict or dict array, or in a database table
"""
if systemstatus['netstatusenabled']:
pilib.log(pilib.syslog, 'Beginning network routines. ', 3, pilib.sysloglevel)
# Update network interfaces statuses for all interfaces, in database tables as well
# Check on wpa supplicant status as well. Function returns wpastatusdict
try:
pilib.log(pilib.syslog, 'Running updateifacestatus. ', 4, pilib.sysloglevel)
pilib.log(pilib.networklog, 'Running updateifacestatus', 4, pilib.networkloglevel)
allnetstatus = updatenetstatus(lastnetstatus)
except:
pilib.log(pilib.networklog, 'Exception in updateifacestatus. ')
else:
pilib.log(pilib.networklog, 'Updateifacestatus completed. ')
pilib.log(pilib.syslog, 'Completed net status update. ', 4, pilib.sysloglevel)
else:
allnetstatus={'netstatusdict': {}, 'ifacesdictarray': {}}
"""
End network configuration status
"""
"""
Do we want to autoconfig the network? If so, we analyze our netstatus data against what should be going on,
and translate this into a network status
"""
if systemstatus['netconfigenabled'] and systemstatus['netstatusenabled']:
# No need to get this fresh. We have it stored.
netconfigdata = allnetstatus['netconfigdata']
# We are going to hack in a jumper that sets AP configuration. This isn't the worst thing ever.
if netconfigdata['apoverride']:
result = processapoverride(21)
''' Now we check network status depending on the configuration we have selected '''
''' Now we check network status depending on the configuration we have selected '''
pilib.log(pilib.syslog, 'Running interface configuration watchdog. ', 4,
pilib.sysloglevel)
pilib.log(pilib.networklog, 'Running interface configuration. Mode: ' + netconfigdata['mode'], 4,
pilib.networkloglevel)
result = watchdognetstatus()
else:
pilib.log(pilib.syslog, 'Netconfig disabled. ', 1, pilib.sysloglevel)
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'mode', 'manual')
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'statusmsg', 'netconfig is disabled')
if systemstatus['checkhamachistatus']:
pilib.log(pilib.syslog, 'Hamachi watchdog is enabled', 3, pilib.sysloglevel)
pilib.log(pilib.networklog, 'Hamachi watchdog is enabled. ', 3, pilib.networkloglevel)
# Only watchdog haamchi if we are connected to the network.
netstatus = pilib.readonedbrow(pilib.systemdatadatabase, 'netstatus')[0]
if netstatus['WANaccess']:
pilib.log(pilib.syslog, 'We appear to be online. Checking Hamachi Status. ', 3, pilib.sysloglevel)
pilib.log(pilib.networklog, 'We appear to be online. Checking Hamachi Status. ', 3, pilib.networkloglevel)
watchdoghamachi(pingip='25.37.18.7')
else:
pilib.log(pilib.syslog, 'We appear to be offline. Not checking Hamachi Status. ', 3, pilib.sysloglevel)
pilib.log(pilib.networklog, 'We appear to be offline. Not checking Hamachi Status. ', 3, pilib.networkloglevel)
else:
pilib.log(pilib.syslog, 'Hamachi watchdog is disnabled', 3, pilib.sysloglevel)
pilib.log(pilib.syslog, 'Finished interface configuration. ', 4,
pilib.sysloglevel)
# pilib.writedatedlogmsg(pilib.networklog, statusmsg)
pilib.log(pilib.syslog, 'Running updateifacestatus. ', 4, pilib.sysloglevel)
updatenetstatus()
pilib.log(pilib.syslog, 'Completed updateifacestatus. ', 4, pilib.sysloglevel)
pilib.log(pilib.syslog, 'Network routines complete. ', 3, pilib.sysloglevel)
pilib.log(pilib.syslog, 'Checking system flags. ', 3, pilib.sysloglevel)
processsystemflags()
pilib.log(pilib.syslog, 'System flags complete. ', 3, pilib.sysloglevel)
# Get system status again
systemstatus = pilib.readalldbrows(pilib.controldatabase, 'systemstatus')[0]
elapsedtime = int(time.time() - starttime)
pilib.log(pilib.syslog, 'Status routines complete. Elapsed time: ' + str(elapsedtime), 3,
pilib.sysloglevel)
pilib.log(pilib.syslog,
'System status is sleeping for ' + str(systemstatus['systemstatusfreq']) + '. ', 3,
pilib.sysloglevel)
if runonce:
break
time.sleep(systemstatus['systemstatusfreq'])
else:
pilib.log(pilib.syslog, 'System status is disabled. Exiting. ', 0,
pilib.sysloglevel)
def updatenetstatus(lastnetstatus=None):
import pilib
import time
import subprocess
from netfun import getifacestatus, getwpaclientstatus, getifconfigstatus
netconfigdata = pilib.readonedbrow(pilib.systemdatadatabase, 'netconfig')[0]
""" We get last netstatus so that we can save last online times, previous online status, etc. """
if not lastnetstatus:
try:
lastnetstatus = pilib.readonedbrow(pilib.systemdatadatabase, 'netstatus')[0]
except:
pilib.log(pilib.syslog, 'Error reading netstatus. Attempting to recreate netstatus table with default values. ', 1, pilib.networkloglevel)
try:
pilib.emptyandsetdefaults(pilib.systemdatadatabase, 'netstatus')
lastnetstatus = pilib.readonedbrow(pilib.systemdatadatabase, 'netstatus')[0]
except:
pilib.log(pilib.syslog, 'Error recreating netstatus. ', 1, pilib.networkloglevel)
""" Pyiface is one way to read some iface data, but it doesn't always appear to show all interfaces(?!)
So we wrote our own instead. A work in progress but quite functional at the moment. """
pilib.log(pilib.networklog, 'Reading ifaces with ifconfig status. ', 4, pilib.networkloglevel)
ifacesdictarray = getifconfigstatus()
# ifacesdictarray = getifacestatus()
""" We supplement with wpa status on the wlan interfaces """
updateddictarray = []
for interface in ifacesdictarray:
if interface['name'].find('wlan') >= 0:
interface['wpastate'] = pilib.dicttojson(getwpaclientstatus(interface['name']))
else:
interface['wpastate'] = ''
updateddictarray.append(interface)
ifacesdictarray = updateddictarray
""" Then write it to the table """
if ifacesdictarray:
pilib.log(pilib.networklog, 'Sending ifaces query. ', 5, pilib.networkloglevel)
# print(ifacesdictarray)
pilib.insertstringdicttablelist(pilib.systemdatadatabase, 'netifaces', ifacesdictarray)
else:
pilib.log(pilib.networklog, 'Empty ifaces query. ', 2, pilib.networkloglevel)
pilib.log(pilib.networklog, 'Completed pyiface ifaces. ', 4, pilib.networkloglevel)
""" Now we check to see if we can connect to WAN """
pilib.log(pilib.networklog, 'Checking pingtimes. ', 4, pilib.networkloglevel)
okping = float(netconfigdata['pingthreshold'])
from netfun import runping
netstatusdict = {}
querylist=[]
pingresults = runping('8.8.8.8')
# pingresults = [20, 20, 20]
pingresult = sum(pingresults) / float(len(pingresults))
if pingresult == 0:
wanaccess = 0
latency = 0
else:
if pingresult < okping:
wanaccess = 1
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'WANaccess', 1)
if lastnetstatus['WANaccess'] == 0 or not lastnetstatus['onlinetime']:
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'onlinetime', pilib.gettimestring())
else:
wanaccess = 0
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'WANaccess', 0)
if lastnetstatus['WANaccess'] == 1 or not lastnetstatus['offlinetime']:
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'offlinetime', pilib.gettimestring())
latency = pingresult
# we set all the values here, so when we retreive it we get changed and also whatever else happens to be there.
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'latency', latency)
updatetime = pilib.gettimestring()
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'updatetime', updatetime)
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'WANaccess', wanaccess)
pilib.log(pilib.networklog, 'Done checking pings. ', 4, pilib.networkloglevel)
if netconfigdata['netstatslogenabled']:
# print('going to log stuff')
pilib.logtimevaluedata(pilib.logdatabase, 'system_WANping', time.time(), pingresult, 1000,
netconfigdata['netstatslogfreq'])
#This is kinda ugly. Should be fixed.
# netstatusdict = {'WANaccess':wanaccess, 'latency': latency, 'updatetime': updatetime}
netstatusdict = pilib.readonedbrow(pilib.systemdatadatabase, 'netstatus')[0]
return {'netstatusdict': netstatusdict, 'ifacesdictarray': ifacesdictarray, 'netconfigdata':netconfigdata}
def updatewpasupplicant(path='/etc/wpa_supplicant/wpa_supplicant.conf'):
netconfig = pilib.readonedbrow(pilib.systemdatadatabase, 'netconfig')[0]
try:
wirelessauths = pilib.readalldbrows(pilib.safedatabase, 'wireless')
except:
pilib.log(pilib.networklog, 'Error reading wireless data. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Read wireless data. ', 4, pilib.networkloglevel)
psk = ''
ssid = ''
# we only update if we find the credentials
try:
ssid = netconfig['SSID']
except KeyError:
pilib.log(pilib.syslog, 'No SSID found in netconfig. Finding first in wireless', 1, pilib.sysloglevel)
# try to attach to first network by setting SSID to first network in wireless auths
# this can help alleviate some headaches down the line, hopefully
wirelessauths = pilib.readalldbrows(pilib.safedatabase, 'wireless')
try:
defaultauths = wirelessauths[0]
currssid = defaultauths['SSID']
except:
pilib.log(pilib.syslog, 'No SSID in wireless table to default to. ', 1, pilib.sysloglevel)
if ssid:
for auth in wirelessauths:
if auth['SSID'] == netconfig['SSID']:
password = auth['password']
ssid = auth['SSID']
pilib.log(pilib.networklog, 'SSID ' + auth['SSID'] + 'found. ', 1, pilib.networkloglevel)
psk = password
ssid = ssid
if psk:
myfile = open(path, 'w')
filestring = 'ctrl_interface=DIR=/var/run/wpa_supplicant GROUP=netdev\nupdate_config=1\n\n'
filestring += 'network={\n'
filestring += 'psk="' + psk + '"\n'
filestring += 'ssid="' + ssid + '"\n'
filestring += 'proto=RSN\nauth_alg=OPEN\npairwise=CCMP\nkey_mgmt=WPA-PSK\n}'
myfile.write(filestring)
else:
pilib.log(pilib.networklog, 'No auths recovered for SSID, so not writing wpa_supplicant', 1, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'No SSID found to write, so not writing wpa_supplicant', 1, pilib.networkloglevel)
sleep(1)
# Refresh after set
itemstatuses = findprocstatuses(pilib.daemonprocs)
for item in pilib.daemonprocs:
index = pilib.daemonprocs.index(item)
# set status
if itemstatuses[index]:
pilib.sqlitequery(pilib.controldatabase, 'update systemstatus set ' + statustableitemnames[index] + ' = 1')
else:
pilib.sqlitequery(pilib.controldatabase, 'update systemstatus set ' + statustableitemnames[index] + ' = 0')
# Set system message
systemstatusmsg = ''
for name, enabled, status in zip(pilib.daemonprocs, enableditemlist, itemstatuses):
systemstatusmsg += name + ' - Enabled: ' + str(bool(enabled)) + ' Status: ' + str(status) + '. '
pilib.setsinglevalue(pilib.controldatabase, 'systemstatus','systemmessage', systemstatusmsg)
# Rotate all logs
pilib.rotatelogs(pilib.networklog, pilib.numlogs, pilib.maxlogsize)
pilib.rotatelogs(pilib.syslog, pilib.numlogs, pilib.maxlogsize)
pilib.rotatelogs(pilib.iolog, pilib.numlogs, pilib.maxlogsize)
pilib.rotatelogs(pilib.controllog, pilib.numlogs, pilib.maxlogsize)
pilib.rotatelogs(pilib.daemonlog, pilib.numlogs, pilib.maxlogsize)
if __name__ == "__main__":
from pilib import log, daemonlog, daemonloglevel
log(daemonlog, 'Running daemon.', 1, daemonloglevel)
rundaemon()
log(daemonlog, 'Daemon complete.',1,daemonloglevel)
def runpicontrol(runonce=False):
systemstatus = pilib.readalldbrows(pilib.controldatabase, 'systemstatus')[0]
while systemstatus['picontrolenabled']:
pilib.log(pilib.syslog, 'Running picontrol', 3, pilib.sysloglevel)
pilib.log(pilib.controllog, 'Running picontrol', 3, pilib.controlloglevel)
# Set poll date. While intuitively we might want to set this
# after the poll is complete, if we error below, we will know
# from this stamp when it barfed. This is arguably more valuable
# then 'last time we didn't barf'
pilib.sqlitequery(pilib.controldatabase,
"update systemstatus set lastpicontrolpoll='" + pilib.gettimestring() + "'")
channels = pilib.readalldbrows(pilib.controldatabase, 'channels')
outputs = pilib.readalldbrows(pilib.controldatabase, 'outputs')
controlalgorithms = pilib.readalldbrows(pilib.controldatabase, 'controlalgorithms')
algorithmnames=[]
for algorithm in controlalgorithms:
algorithmnames.append(algorithm['name'])
# Cycle through channels and set action based on setpoint
# and algorithm if set to auto mode
for channel in channels:
statusmsg = ''
querylist = []
channelindex = str(int(channel['channelindex']))
channelname = channel['name']
logtablename = 'channel' + '_' + channel['name'] + '_log'
time = pilib.gettimestring()
disableoutputs = True
# Make sure channel is enabled
if channel['enabled']:
# Create log if it doesn't exist
query = 'create table if not exists \'' + logtablename + '\' (time text, controlinput text, controlvalue real, setpointvalue real, action real, algorithm text, enabled real, statusmsg text)'
pilib.sqlitequery(pilib.logdatabase, query)
statusmsg = ''
if 'setpointvalue' in channel:
setpointvalue = float(channel['setpointvalue'])
else:
statusmsg += 'No setpoint. '
# Need to test for age of data. If stale or disconnected, invalidate
if 'controlvalue' in channel:
try:
controlvalue = float(channel['controlvalue'])
except (ValueError, TypeError) as e:
statusmsg += 'Invalid control value. '
controllib.setcontrolvalue(pilib.controldatabase, channelname, 0)
else:
statusmsg += 'No controlvalue. '
# Test to see if key exists and is true
if 'enabled' in channel:
if channel['enabled']:
pass
else:
statusmsg += 'Channel disabled. '
else:
statusmsg += 'Error. Enabled key does not exist'
mode = channel['mode']
channelalgorithmname = channel['controlalgorithm']
controlinput = channel['controlinput']
logpoints = channel['logpoints']
# Move forward if everything is defined for control
if channel['enabled'] and 'controlvalue' in locals() and 'setpointvalue' in locals():
statusmsg += 'Channel Enabled. '
if mode == 'auto':
statusmsg += 'Mode:Auto. '
#print('running auto sequence')
# run algorithm on channel
response = controllib.runalgorithm(pilib.controldatabase, pilib.recipedatabase, channelname)
action = response[0]
message = response[1]
statusmsg += ' ' + response[1] + ' '
statusmsg += 'Action: ' + str(action) + '. '
# Set action in channel
controllib.setaction(pilib.controldatabase, channelname, action)
elif mode == 'manual':
#print('manual mode')
statusmsg += 'Mode:Manual. '
# action = controllib.getaction(pilib.controldatabase, channelname)
else:
#print('error, mode= ' + mode)
statusmsg += 'Mode:Error. '
if systemstatus['enableoutputs']:
statusmsg += 'System outputs enabled. '
if channel['outputsenabled']:
statusmsg += 'Channel outputs enabled. '
disableoutputs = False
# find out whether action is positive or negative or
# not at all.
# and act. for now, this is binary, but in the future
# this will be a duty cycle daemon
outputsetnames = []
outputresetnames = []
if action > 0:
print("set positive output on")
outputsetnames.append(channel['positiveoutput'])
outputresetnames.append(channel['negativeoutput'])
elif action < 0:
print("set negative output on")
outputsetnames.append(channel['negativeoutput'])
outputresetnames.append(channel['positiveoutput'])
elif action == 0:
statusmsg += 'No action. '
outputresetnames.append(channel['positiveoutput'])
outputresetnames.append(channel['negativeoutput'])
else:
statusmsg += 'Algorithm error. Doing nothing.'
outputsetname = None
# Check to see if outputs are ready to enable/disable
# If not, pull them from list of set/reset
outputstoset=[]
for outputname in outputsetnames:
if channelalgorithmname in algorithmnames:
offtime = pilib.sqlitedatumquery(pilib.controldatabase, "select offtime from outputs where name='" + outputname + "'" )
if pilib.timestringtoseconds(pilib.gettimestring()) - pilib.timestringtoseconds(offtime) > controlalgorithms[algorithmnames.index(channelalgorithmname)]['minofftime']:
outputstoset.append(outputname)
else:
statusmsg += 'Output ' + outputname + ' not ready to enable. '
else:
statusmsg += 'Algorithm Error: Not found. '
outputstoreset=[]
for outputname in outputresetnames:
if channelalgorithmname in algorithmnames:
ontime = pilib.sqlitedatumquery(pilib.controldatabase, "select ontime from outputs where name='" + outputname +"'")
if pilib.timestringtoseconds(pilib.gettimestring()) - pilib.timestringtoseconds(ontime) > controlalgorithms[algorithmnames.index(channelalgorithmname)]['minontime']:
outputstoreset.append(outputname)
else:
statusmsg += 'Output ' + outputname + ' not ready to disable. '
else:
statusmsg += 'Algorithm Error: Not found. '
""" TODO: Change reference to controlinputs to name rather than id. Need to double-check
enforcement of no duplicates."""
# Find output in list of outputs if we have one to set
time = pilib.gettimestring()
if len(outputstoset) > 0 or len(outputstoreset) > 0:
for output in outputs:
if output['name'] in outputstoset:
# check current status
currvalue = output['value']
if currvalue == 0: # No need to set if otherwise. Will be different for analog out
# set ontime
querylist.append('update outputs set ontime=\'' + time + '\' ' + 'where id=\'' + output['id'] + '\'')
# set value
querylist.append("update outputs set value = 1 where id='" + output['id'] + '\'')
statusmsg += 'Output ' + output['name'] + ' enabled. '
else:
statusmsg += 'Output ' + output['name'] + ' already enabled. '
if output['name'] in outputstoreset:
# check current status
currvalue = output['value']
if currvalue == 1: # No need to set if otherwise. Will be different for analog out
# set ontime
querylist.append('update outputs set offtime=\'' + time + '\' ' + 'where id=\'' + output['id'] + '\'')
# set value
querylist.append('update outputs set value = 0 where id=\'' + output['id'] + '\'')
statusmsg += 'Output ' + output['name'] + ' disabled. '
else:
statusmsg += 'Output ' + output['name'] + ' already disabled. '
else:
statusmsg += 'Channel outputs disabled. '
action = 0
else:
statusmsg += 'System outputs disabled. '
action = 0
# Insert entry into control log
pilib.makesqliteinsert(pilib.logdatabase, logtablename, [time, controlinput, controlvalue, setpointvalue, action, channelalgorithmname, channel['enabled'], statusmsg])
pilib.sqliteinsertsingle(pilib.logdatabase, logtablename,
[time, controlinput, controlvalue, setpointvalue, action, channelalgorithmname,
channel['enabled'], statusmsg])
# Size log
pilib.sizesqlitetable(pilib.logdatabase, logtablename, logpoints)
# print(statusmsg)
else:
# print('channel not enabled')
statusmsg += 'Channel not enabled. '
# If active reset and we didn't set channel modes, disable outputs
# Active reset is not yet explicitly declared, but implied
if disableoutputs:
statusmsg += 'Disabling Outputs. '
for output in outputs:
if output['name'] in [channel['positiveoutput'], channel['negativeoutput']]:
# set value
querylist.append("update outputs set value = 0 where id='" + output['id'] + '\'')
statusmsg += 'Output ' + output['name'] + ' disabled. '
# Set status message for channel
# print(statusmsg)
querylist.append('update channels set statusmessage=\'' + statusmsg + '\'' + 'where channelindex=' + channelindex)
# Set update time for channel
querylist.append('update channels set controlupdatetime=\'' + time + '\'' + 'where channelindex=' + channelindex)
# Execute query
pilib.sqlitemultquery(pilib.controldatabase, querylist)
# We do this system status again to refresh settings
systemstatus = pilib.readalldbrows(pilib.controldatabase, 'systemstatus')[0]
from actions import processactions
processactions()
# Wait for delay time
#print('sleeping')
# spilights.updatelightsfromdb(pilib.controldatabase, 'indicators')
if runonce:
break
pilib.log(pilib.syslog, 'Picontrol Sleeping for ' + str(systemstatus['picontrolfreq']), 2, pilib.sysloglevel)
pilib.log(pilib.controllog, 'Picontrol Sleeping for ' + str(systemstatus['picontrolfreq']), 2, pilib.sysloglevel)
sleep(systemstatus['picontrolfreq'])
pilib.log(pilib.syslog, 'picontrol not enabled. exiting.', 1, pilib.sysloglevel)
def processGPIOinterface(interface, prevoutputs, prevoutputvalues, prevoutputids, previnputs, previnputids, defaults, logconfig, **kwargs):
import pilib
if 'method' in kwargs:
method = kwargs['method']
else:
method = 'pigpio'
if method == 'rpigpio':
import RPi.GPIO as GPIO
elif method == 'pigpio':
import pigpio
try:
if method == 'rpigpio':
GPIO.setmode(GPIO.BCM)
GPIO.setwarnings(False)
elif method == 'pigpio':
if 'piobject' in kwargs:
pilib.log(pilib.iolog,
'Pigpio object already exists. ', 4, logconfig['iologlevel'])
pi = kwargs['piobject']
else:
pilib.log(pilib.iolog,
'Instantiating pigpio. ', 4, logconfig['iologlevel'])
pi = pigpio.pi()
except:
pilib.log(pilib.iolog,
'Error setting up GPIO. ', 1, logconfig['iologlevel'])
else:
pilib.log(pilib.iolog,
'Done setting up GPIO. ', 4, logconfig['iologlevel'])
options = pilib.parseoptions(interface['options'])
address = int(interface['address'])
pilib.log(pilib.iolog, 'GPIO address' + str(address) + ' enabled', 4, logconfig['iologlevel'])
# Get name from ioinfo table to give it a colloquial name
gpioname = pilib.sqlitedatumquery(pilib.controldatabase, 'select name from ioinfo where id=\'' + interface['id'] + '\'')
polltime = pilib.gettimestring()
querylist = []
# Append to inputs and update name, even if it's an output (can read status as input)
if options['mode'] == 'output':
pilib.log(pilib.iolog, 'Setting output mode for GPIO address' + str(address), 3, logconfig['iologlevel'])
try:
if method == 'rpigpio':
GPIO.setup(address, GPIO.OUT)
elif method == 'pigpio':
pi.set_mode(address, pigpio.OUTPUT)
except TypeError:
pilib.log(pilib.iolog, 'You are trying to set a GPIO with the wrong variable type : ' +
str(type(address)), 0, logconfig['iologlevel'])
pilib.log(pilib.iolog, 'Exiting interface routine for ' + interface['id'], 0, logconfig['iologlevel'])
return
except:
pilib.log(pilib.iolog, 'Error setting GPIO : ' +
str(address), 1, logconfig['iologlevel'])
pilib.log(pilib.iolog, 'Exiting interface routine for ' + interface['id'], 0, logconfig['iologlevel'])
return
# Set the value of the gpio.
# Get previous value if exists
if interface['id'] in prevoutputids:
value = prevoutputvalues[prevoutputids.index(interface['id'])]
else:
value = 0
if value == 1:
if method == 'rpigpio':
GPIO.output(address, True)
elif method == 'pigpio':
pi.write(address, 1)
pilib.log(pilib.iolog, 'Setting output ON for GPIO address' + str(address), 3,
logconfig['iologlevel'])
else:
if method == 'rpigpio':
GPIO.output(address, False)
elif method == 'pigpio':
pi.write(address,0)
pilib.log(pilib.iolog, 'Setting output OFF for GPIO address' + str(address), 3,
logconfig['iologlevel'])
# Get output settings and keep them if the GPIO previously existed
if interface['id'] in prevoutputids:
pollfreq = prevoutputs[prevoutputids.index(interface['id'])]['pollfreq']
ontime = prevoutputs[prevoutputids.index(interface['id'])]['ontime']
offtime = prevoutputs[prevoutputids.index(interface['id'])]['offtime']
else:
try:
pollfreq = defaults['defaultoutputpollfreq']
except:
pollfreq = 60
ontime = ''
offtime = ''
# Add entry to outputs tables
querylist.append('insert into outputs values (\'' + interface['id'] + '\',\'' +
interface['interface'] + '\',\'' + interface['type'] + '\',\'' + str(
address) + '\',\'' +
gpioname + '\',\'' + str(value) + "','','" + str(polltime) + '\',\'' +
str(pollfreq) + "','" + ontime + "','" + offtime + "')")
elif options['mode'] == 'input':
if method == 'rpigpio':
GPIO.setup(address, GPIO.IN)
elif method == 'pigpio':
if 'pullupdown' in options:
if options['pullupdown'] == 'pullup':
pi.set_pull_up_down(address, pigpio.PUD_UP)
elif options['pullupdown'] == 'pulldown':
pi.set_pull_up_down(address, pigpio.PUD_DOWN)
else:
pi.set_pull_up_down(address, pigpio.PUD_OFF)
else:
pi.set_pull_up_down(17, pigpio.PUD_OFF)
pi.set_mode(address, pigpio.INPUT)
if method == 'rpigpio':
value = GPIO.input(address)
elif method == 'pigpio':
value = pi.read(address)
if 'function' in options:
if options['function'] == 'shutdown':
# TODO : The reboot function is still held in a shell script, because it works.
if 'functionstate' in options:
if value == 1 and options['functionstate'] in ['true', 'On', 'True', '1']:
from systemstatus import processsystemflags
processsystemflags()
pilib.log(pilib.iolog, 'Setting input mode for GPIO address' + str(address), 3,
logconfig['iologlevel'])
# Get input settings and keep them if the GPIO previously existed
if interface['id'] in previnputids:
pollfreq = previnputs[previnputids.index(interface['id'])]['pollfreq']
ontime = previnputs[previnputids.index(interface['id'])]['ontime']
offtime = previnputs[previnputids.index(interface['id'])]['offtime']
pilib.log(pilib.iolog,
'Restoring values from previous inputids: pollfreq = ' + str(
pollfreq) + ' ontime = ' + str(ontime) + ' offtime = ' + str(
offtime), 3, logconfig['iologlevel'])
else:
try:
pollfreq = defaults['defaultinputpollfreq']
except:
pollfreq = 60
ontime = ''
offtime = ''
pilib.log(pilib.iolog,
'Setting values to defaults, defaultinputpollfreq = ' + str(
pollfreq), 3, logconfig['iologlevel'])
querylist.append(
'insert into inputs values (\'' + interface['id'] + '\',\'' + interface['interface'] + '\',\'' +
interface['type'] + '\',\'' + str(address) + '\',\'' + gpioname + '\',\'' + str(
value) + "','','" +
str(polltime) + '\',\'' + str(pollfreq) + "','" + ontime + "','" + offtime + "')")
if method == 'pigpio' and 'piobject' not in kwargs:
pi.stop()
return querylist
def runconfig(onboot=False):
import subprocess
try:
netconfigdata = pilib.readonedbrow(pilib.systemdatadatabase, 'netconfig')[0]
# print(netconfigdata)
except:
pilib.log(pilib.networklog, 'Error reading netconfig data. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Successfully read netconfig data', 3, pilib.networkloglevel)
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'mode', netconfigdata['mode'])
pilib.log(pilib.networklog, 'Netconfig is enabled', 3, pilib.networkloglevel)
# This will grab the specified SSID and the credentials and update
# the wpa_supplicant file
updatewpasupplicant()
# Copy the correct interfaces file
if netconfigdata['mode'] == 'station':
setstationmode(netconfigdata)
elif netconfigdata['mode'] in ['ap', 'tempap', 'eth0wlan0bridge']:
pilib.log(pilib.networklog, 'Setting eth0wlan0 bridge (or bare ap mode). ', 0, pilib.networkloglevel)
subprocess.call(['/bin/cp', '/usr/lib/iicontrollibs/misc/interfaces/interfaces.ap', '/etc/network/interfaces'])
killapservices()
resetwlan()
startapservices('wlan0')
# All of these require ipv4 being enabled in /etc/sysctl.conf
# First interface is DHCP, second is CuPIDwifi
elif netconfigdata['mode'] == 'wlan0wlan1bridge':
pilib.log(pilib.networklog, 'Setting wlan0wlan1 bridge', 0, pilib.networkloglevel)
subprocess.call(['/bin/cp', '/usr/lib/iicontrollibs/misc/interfaces/interfaces.wlan0dhcp.wlan1cupidwifi', '/etc/network/interfaces'])
killapservices()
resetwlan('wlan0')
resetwlan('wlan1')
startapservices('wlan1')
elif netconfigdata['mode'] == 'wlan1wlan0bridge':
pilib.log(pilib.networklog, 'Setting wlan1wlan0 bridge', 0, pilib.networkloglevel)
subprocess.call(['/bin/cp', '/usr/lib/iicontrollibs/misc/interfaces/interfaces.wlan1dhcp.wlan0cupidwifi', '/etc/network/interfaces'])
killapservices()
resetwlan('wlan0')
resetwlan('wlan1')
startapservices('wlan0')
runIPTables(netconfigdata['mode'])
def updatewpasupplicantOLD(interface='wlan0'):
# print('I AM UPDATING SUPPLICANT DATA')
try:
suppdata = getwpasupplicantconfig()
except:
pilib.log(pilib.networklog, 'Error getting supplicant data. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Supplicant data retrieved successfully. ', 3, pilib.networkloglevel)
try:
updateddata = updatesupplicantdata(suppdata)
except:
pilib.log(pilib.networklog, 'Error updating supplicant data. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Supplicant data retrieved successfully. ', 3, pilib.networkloglevel)
try:
writesupplicantfile(updateddata)
except:
pilib.log(pilib.networklog, 'Error writing supplicant data. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Supplicant data written successfully. ', 3, pilib.networkloglevel)
def updateiodata(database, **kwargs):
# This recreates all input and output tables based on the interfaces table.
# Thus way we don't keep around stale data values. We could at some point incorporate
# a retention feature that keeps them around in case they disappear temporarily.
# It also reads the elements if they are enabled and it's time to read them
import pilib
import traceback
if 'piobject' in kwargs:
pi = kwargs['piobject']
else:
import pigpio
pi = pigpio.pi()
allowedGPIOaddresses = [18, 23, 24, 25, 4, 17, 27, 22, 5, 6, 13, 19, 26, 16, 20, 21]
logconfig = pilib.getlogconfig()
tables = pilib.gettablenames(pilib.controldatabase)
if 'interfaces' in tables:
interfaces = pilib.readalldbrows(pilib.controldatabase, 'interfaces')
else:
pilib.log(pilib.iolog, 'interfaces table not found. Exiting', 1,
logconfig['iologlevel'])
return
if 'inputs' in tables:
previnputs = pilib.readalldbrows(pilib.controldatabase, 'inputs')
# Make list of IDs for easy indexing
previnputids = []
for input in previnputs:
previnputids.append(input['id'])
else:
previnputs = []
previnputids = []
if 'outputs' in tables:
prevoutputs = pilib.readalldbrows(pilib.controldatabase, 'outputs')
# Make list of IDs for easy indexing
prevoutputids = []
prevoutputvalues = []
for output in prevoutputs:
prevoutputids.append(output['id'])
prevoutputvalues.append(output['value'])
else:
prevoutputs = {}
prevoutputids = []
if 'defaults' in tables:
defaults = pilib.readalldbrows(pilib.controldatabase, 'defaults')[0]
defaultinputpollfreq = defaults['inputpollfreq']
defaultoutputpollfreq = defaults['outputpollfreq']
else:
defaults = []
defaultinputpollfreq = 60
defaultoutputpollfreq = 60
if 'indicators' in tables:
indicatornames = []
previndicators = pilib.readalldbrows(pilib.controldatabase, 'indicators')
for indicator in previndicators:
indicatornames.append(indicator['name'])
else:
previndicators = []
indicatornames = []
# We drop all inputs and outputs and recreate
# Add all into one query so there is no time when the IO don't exist.
querylist = []
querylist.append('delete from inputs')
querylist.append('delete from outputs')
# This is temporary. Clearing the table here and adding entries below can result in a gap in time
# where there are no database indicator entries. This is not too much of a problem with indicators, as we
# update the hardware explicitly after we add the entries. If the interface queries the table during
# this period, however, we could end up with an apparently empty table.
# TODO: FIX update on indicators in updateio
# We drop this table, so that if SP1 has been disabled, the entries do not appear as valid indicators
pilib.sqlitequery(pilib.controldatabase, 'delete from indicators')
owfsupdate = False
for interface in interfaces:
if interface['interface'] == 'I2C':
pilib.log(pilib.iolog, 'Processing I2C interface ' + interface['name'], 3,
logconfig['iologlevel'])
if interface['enabled']:
pilib.log(pilib.iolog, 'I2C Interface ' + interface['name'] + ' enabled', 3,
logconfig['iologlevel'])
if interface['type'] == 'DS2483':
pilib.log(pilib.iolog, 'Interface type is DS2483', 3,
logconfig['iologlevel'])
owfsupdate = True
else:
pilib.log(pilib.iolog, 'I2C Interface ' + interface['name'] + ' disabled', 3,
logconfig['iologlevel'])
elif interface['interface'] == 'USB':
pilib.log(pilib.iolog, 'Processing USB interface ' + interface['name'], 3,
logconfig['iologlevel'])
if interface['enabled']:
pilib.log(pilib.iolog, 'USB Interface ' + interface['name'] + ' enabled', 3,
logconfig['iologlevel'])
if interface['type'] == 'DS9490':
pilib.log(pilib.iolog, 'Interface type is DS9490', 3,
logconfig['iologlevel'])
owfsupdate = True
else:
pilib.log(pilib.iolog, 'USB Interface ' + interface['name'] + ' disabled', 3,
logconfig['iologlevel'])
elif interface['interface'] == 'MOTE':
#determine and then update id based on fields
entryid = interface['interface'] + '_' + interface['type'] + '_' + interface['address']
condition = '"interface"=\'' + interface['interface'] + '\' and "type"=\'' + interface['type'] + '\' and "address"=\'' + interface['address'] + '\''
print(condition)
pilib.setsinglevalue(pilib.controldatabase, 'interfaces', 'id', entryid, condition)
pilib.log(pilib.iolog, 'Processing Mote interface' + interface['name'] + ', id:' + entryid, 3,
logconfig['iologlevel'])
if interface['enabled']:
pilib.log(pilib.iolog, 'Mote Interface ' + interface['name'] + ', id:' + entryid + ' enabled', 3,
logconfig['iologlevel'])
# Grab mote entries from remotes table
# nodeid and keyvalue are keyed into address in remotes table
# keyvalues are :
# channel: channel number
# iovalue: ionumber
# owdev: ROM
# This would look like, for example
# 1:1 for a nodeid:channel scenario for a controller
split = interface['address'].split(':')
nodeid = split[0]
keyvalue = split[1]
# so we used to enable an interface and then take all entries from a node
# Now, we have to explicitly add an interface for each device, unless the keychar * is used as the
# keyvalue. This will allow us to insert all automatically, for example for owdevs or iovals from a
# node.
# This pulls out all mote entries that have nodeid and keyvalue that match the interface address
# We should just find one, ideally
if keyvalue == '*':
pass
else:
condition = "\"nodeid\"='" + nodeid + "' and \"keyvalue\"='" + keyvalue + "'"
nodeentries = pilib.dynamicsqliteread(pilib.controldatabase, 'remotes', condition=condition)
print("WE FOUND MOTE")
print(condition)
print(len(nodeentries))
if interface['type'] == 'channel':
print('channel')
if len(nodeentries) == 1:
print('one entry found')
nodeentry = nodeentries[0]
nodedata = pilib.parseoptions(nodeentry['data'])
# Find existing channel so we can get existing data, settings, etc., and retain channel ordering
newchanneldata = {'name':interface['name'], 'controlvalue':nodedata['pv'], 'setpointvalue':nodedata['sv'],'controlvaluetime':pilib.gettimestring(), 'data':nodeentry['data'], 'type':'remote'}
newchannel = {}
existingchannels = pilib.readalldbrows(pilib.controldatabase, 'channels')
for channel in existingchannels:
if channel['name'] == interface['name']:
print('updating')
print(channel)
newchannel.update(channel)
print(newchannel)
newchannel.update(newchanneldata)
print(newchannel)
keys = []
values = []
for key, value in newchannel.iteritems():
keys.append(key)
values.append(value)
query = pilib.makesqliteinsert('channels',values, keys)
# print(query)
pilib.sqlitequery(pilib.controldatabase,query)
else:
print('multiple entries found for channel. not appropriate')
else:
pass
# Create queries for table insertion
# TODO: process mote pollfreq, ontime, offtime
moteentries = []
for nodeentry in nodeentries:
# THis breaks out all of the strictly json-encoded data.
datadict = pilib.parseoptions(nodeentry['data'])
try:
entrytype = nodeentry['msgtype']
# now treat each mote type entry specially
# if entrytype == 'channel':
entryid = 'MOTE' + str(nodeentry['nodeid']) + '_' + nodeentry['keyvaluename'] + '_' + nodeentry['keyvalue']
entrymetareturn = pilib.dynamicsqliteread(pilib.controldatabase, 'ioinfo', condition="\"id\"='" + entryid + "'")
try:
entrymeta = entrymetareturn[0]
except:
entrymeta = []
# print(entrymeta)
entryoptions={}
if entrymeta:
entryname = entrymeta['name']
if entrymeta['options']:
entryoptions = pilib.parseoptions(entrymeta['options'])
else:
entryname = '[MOTE' + str(nodeentry['nodeid']) + '] ' + nodeentry['keyvaluename'] + ':' + nodeentry['keyvalue']
except KeyError:
print('OOPS KEY ERROR')
else:
if entrytype == 'iovalue':
if 'scale' in entryoptions:
entryvalue = str(float(entryoptions['scale']) * float(datadict['ioval']))
elif 'formula' in entryoptions:
x = float(datadict['ioval'])
try:
entryvalue = eval(entryoptions['formula'])
except:
entryvalue = float(datadict['ioval'])
else:
entryvalue = float(datadict['ioval'])
elif entrytype == 'owdev':
if 'owtmpasc' in datadict:
if 'scale' in entryoptions:
entryvalue = str(float(entryoptions['scale']) * float(datadict['owtmpasc']))
elif 'formula' in entryoptions:
x = float(datadict['owtmpasc'])
try:
entryvalue = eval(entryoptions['formula'])
except:
entryvalue = float(datadict['owtmpasc'])
else:
entryvalue = datadict['owtmpasc']
else:
entryvalue = -1
else:
entryvalue = -1
moteentries.append('insert into inputs values (\'' + entryid + '\',\'' + interface['interface'] + '\',\'' +
interface['type'] + '\',\'' + str(address) + '\',\'' + entryname + '\',\'' + str(entryvalue) + "','','" +
nodeentry['time'] + '\',\'' + str(15) + "','" + '' + "','" + '' + "')")
# print('querylist')
# print(moteentries)
querylist.extend(moteentries)
else:
pilib.log(pilib.iolog, 'Mote Interface ' + interface['name'] + ' disnabled', 3,
logconfig['iologlevel'])
elif interface['interface'] == 'LAN':
pilib.log(pilib.iolog, 'Processing LAN interface' + interface['name'], 3,
logconfig['iologlevel'])
if interface['enabled']:
pilib.log(pilib.iolog, 'LAN Interface ' + interface['name'] + ' enabled', 3,
logconfig['iologlevel'])
if interface['type'] == 'MBTCP':
pilib.log(pilib.iolog, 'Interface ' + interface['name'] + ' type is MBTCP',
3, logconfig['iologlevel'])
try:
mbentries = processMBinterface(interface, prevoutputs, prevoutputids, previnputs, previnputids, defaults, logconfig)
except:
pilib.log(pilib.iolog,
'Error processing MBTCP interface ' + interface['name'], 0,
logconfig['iologlevel'])
errorstring = traceback.format_exc()
pilib.log(pilib.iolog,
'Error of kind: ' + errorstring, 0,
logconfig['iologlevel'])
else:
pilib.log(pilib.iolog,
'Done processing MBTCP interface ' + interface['name'], 3,
logconfig['iologlevel'])
querylist.extend(mbentries)
else:
pilib.log(pilib.iolog, 'LAN Interface ' + interface['name'] + ' disabled', 3,
logconfig['iologlevel'])
elif interface['interface'] == 'GPIO':
try:
address = int(interface['address'])
except KeyError:
pilib.log(pilib.iolog, 'GPIO address key not found for ' + interface['name'], 1,
logconfig['iologlevel'])
continue
if interface['enabled']:
pilib.log(pilib.iolog, 'Processing GPIO interface ' + str(interface['address']), 3,
logconfig['iologlevel'])
if address in allowedGPIOaddresses:
pilib.log(pilib.iolog, 'GPIO address' + str(address) + ' allowed. Processing.', 4,
logconfig['iologlevel'])
GPIOentries = processGPIOinterface(interface, prevoutputs, prevoutputvalues, prevoutputids,
previnputs, previnputids, defaults, logconfig, piobject=pi)
if GPIOentries:
querylist.extend(GPIOentries)
else:
pilib.log(pilib.iolog,
'GPIO address' + str(address) + ' not allowed. Bad things can happen. ', 4,
logconfig['iologlevel'])
else:
pilib.log(pilib.iolog, 'GPIO address' + str(address) + ' disabled. Doing nothing.', 4,
logconfig['iologlevel'])
elif interface['interface'] == 'SPI0':
pilib.log(pilib.iolog, 'Processing SPI0', 1, logconfig['iologlevel'])
if interface['enabled']:
pilib.log(pilib.iolog, 'SPI0 enabled', 1, logconfig['iologlevel'])
if interface['type'] == 'SPITC':
pilib.log(pilib.iolog, 'Processing SPITC on SPI0', 3, logconfig['iologlevel'])
import readspi
tcdict = readspi.getpigpioMAX31855temp(0,0)
# Convert to F for now
spitcentries = readspi.recordspidata(database, {'SPITC1' :tcdict['tctemp']*1.8+32})
querylist.extend(spitcentries)
if interface['type'] == 'CuPIDlights':
import spilights
spilightsentries, setlist = spilights.getCuPIDlightsentries('indicators', 0, previndicators)
querylist.extend(spilightsentries)
spilights.updatelightsfromdb(pilib.controldatabase, 'indicators', 0)
spilights.setspilights(setlist, 0)
else:
pilib.log(pilib.iolog, 'SPI0 not enabled', 1, logconfig['iologlevel'])
elif interface['interface'] == 'SPI1':
pilib.log(pilib.iolog, 'Processing SPI1', 1, logconfig['iologlevel'])
if interface['enabled']:
pilib.log(pilib.iolog, 'SPI1 enabled', 1, logconfig['iologlevel'])
if interface['type'] == 'CuPIDlights':
pilib.log(pilib.iolog, 'Processing CuPID Lights on SPI1', 1, logconfig['iologlevel'])
import spilights
spilightsentries, setlist = spilights.getCuPIDlightsentries('indicators', 1, previndicators)
querylist.extend(spilightsentries)
spilights.setspilights(setlist, 1)
else:
pilib.log(pilib.iolog, 'SPI1 disaabled', 1, logconfig['iologlevel'])
# Set tables
querylist.append(pilib.makesinglevaluequery('systemstatus', 'lastiopoll', pilib.gettimestring()))
if owfsupdate:
from owfslib import runowfsupdate
pilib.log(pilib.iolog, 'Running owfsupdate', 1, logconfig['iologlevel'])
devices, owfsentries = runowfsupdate(execute=False)
querylist.extend(owfsentries)
else:
pilib.log(pilib.iolog, 'owfsupdate disabled', 3, logconfig['iologlevel'])
pilib.log(pilib.iolog, 'Executing query: ' + str(querylist), 5, logconfig['iologlevel'])
try:
# print(querylist)
pilib.sqlitemultquery(pilib.controldatabase, querylist)
except:
errorstring = traceback.format_exc()
pilib.log(pilib.iolog, 'Error executing query, message: ' + errorstring, 0, logconfig['iologlevel'])
pilib.log(pilib.errorlog, 'Error executing updateio query, message: ' + errorstring)
pilib.log(pilib.errorlog, 'Query: ' + str(querylist))
def updatesupplicantdata(configdata):
from pilib import readalldbrows, safedatabase, systemdatadatabase
netconfig = []
try:
netconfig = readalldbrows(systemdatadatabase, 'netconfig')[0]
except:
pilib.log(pilib.networklog, 'Error reading netconfig data. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Read netconfig data. ', 4, pilib.networkloglevel)
try:
wirelessauths = readalldbrows(safedatabase, 'wireless')
except:
pilib.log(pilib.networklog, 'Error reading wireless data. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Read wireless data. ', 4, pilib.networkloglevel)
password = ''
pilib.log(pilib.networklog, 'Netconfig data: ' + str(netconfig), 2, pilib.networkloglevel)
# we only update if we find the credentials
for auth in wirelessauths:
if auth['SSID'] == netconfig['SSID']:
password = '******' + auth['password'] + '"'
ssid = '"' + auth['SSID'] + '"'
pilib.log(pilib.networklog, 'SSID ' + auth['SSID'] + 'found. ', 1, pilib.networkloglevel)
configdata.data['psk'] = password
configdata.data['ssid'] = ssid
return configdata
def processMBinterface(interface, prevoutputs, prevoutputids, previnputs, previnputids, defaults, logconfig):
from netfun import readMBcodedaddresses, MBFCfromaddress
import pilib
# get all modbus reads that have the same address from the modbus table
try:
modbustable = pilib.readalldbrows(pilib.controldatabase, 'modbustcp')
except:
pilib.log(pilib.iolog, 'Error reading modbus table', 0, logconfig['iologlevel'])
else:
pilib.log(pilib.iolog, 'Read modbus table', 4, logconfig['iologlevel'])
querylist = []
for entry in modbustable:
# Get name from ioinfo table to give it a colloquial name
# First we have to give it a unique ID. This is a bit difficult with modbus
if entry['mode'] == 'read':
shortmode = 'R'
elif entry['mode'] == 'write':
shortmode = 'W'
elif entry['mode'] == 'readwrite':
shortmode = 'RW'
else:
pilib.log(pilib.iolog, 'modbus mode error', 1, logconfig['iologlevel'])
try:
mbid = entry['interfaceid'] + '_' + str(entry['register']) + '_' + str(entry['length']) + '_' + shortmode
except KeyError:
pilib.log(pilib.iolog, 'Cannot form mbid due to key error', 0, logconfig['iologlevel'])
return
pilib.log(pilib.iolog, 'Modbus ID: ' + mbid, 4, logconfig['iologlevel'])
mbname = pilib.sqlitedatumquery(pilib.controldatabase, "select name from ioinfo where id='" + mbid + "'")
polltime = pilib.gettimestring()
if entry['interfaceid'] == interface['id']:
# For now, we're going to read them one by one. We'll assemble these into block reads
# in the next iteration
if entry['mode'] == 'read':
# Get previous metadata and data
# Get input settings and keep them if the input previously existed
if mbid in previnputids:
pollfreq = previnputs[previnputids.index(mbid)]['pollfreq']
ontime = previnputs[previnputids.index(mbid)]['ontime']
offtime = previnputs[previnputids.index(mbid)]['offtime']
prevvalue = previnputs[previnputids.index(mbid)]['offtime']
prevpolltime = previnputs[previnputids.index(mbid)]['offtime']
pilib.log(pilib.iolog,
'Restoring values from previous inputids: pollfreq = ' + str(
pollfreq) + ' ontime = ' + str(ontime) + ' offtime = ' + str(
offtime), 3, logconfig['iologlevel'])
else:
# set values to defaults if it did not previously exist
try:
pollfreq = defaults['defaultinputpollfreq']
except:
pollfreq = 60
ontime = ''
offtime = ''
prevvalue = ''
prevpolltime = ''
pilib.log(pilib.iolog,
'Setting values to defaults, defaultinputpollfreq = ' + str(pollfreq), 3, logconfig['iologlevel'])
# Read data
try:
readresult = readMBcodedaddresses(interface['address'], entry['register'], entry['length'])
except:
pilib.log(pilib.iolog, 'Uncaught reror reading modbus value', 0, logconfig['iologlevel'])
else:
if readresult['statuscode'] == 0:
values = readresult['values']
try:
FC = MBFCfromaddress(int(entry['register']))
except ValueError:
pilib.log(pilib.iolog, 'Malformed address for FC determination : ' + str(entry['address']), 0, logconfig['iologlevel'])
else:
pilib.log(pilib.iolog, 'Function code : ' + str(FC), 4, logconfig['iologlevel'])
returnvalue = 0
if len(values) > 0:
pilib.log(pilib.iolog, 'Multiple values returned', 4, logconfig['iologlevel'])
if not entry['bigendian']:
try:
values.reverse()
except AttributeError:
pilib.log(pilib.iolog, 'Error on reverse of MB values: ' + str(values), 0, logconfig['iologlevel'])
if entry['format'] == 'float32':
import struct
byte2 = values[0] % 256
byte1 = (values[0] - byte2)/256
byte4 = values[1] % 256
byte3 = (values[1] - byte4)/256
byte1hex = chr(byte1)
byte2hex = chr(byte2)
byte3hex = chr(byte3)
byte4hex = chr(byte4)
hexstring = byte1hex + byte2hex + byte3hex + byte4hex
returnvalue = struct.unpack('>f',hexstring)[0]
else:
for index, val in enumerate(values):
if FC in [0, 1]:
returnvalue += val * 2 ** index
elif FC in [3, 4]:
returnvalue += val * 256 ** index
else:
pilib.log(pilib.iolog, 'Invalid function code', 0, logconfig['iologlevel'])
else:
returnvalue = values[0]
if entry['options'] != '':
options = pilib.parseoptions(entry['options'])
if 'scale' in options:
# try:
returnvalue = returnvalue * float(options['scale'])
# except:
# pilib.writedatedlogmsg(pilib.iolog, 'Error on scale operation', 0, logconfig['iologlevel'])
if 'precision' in options:
# try:
returnvalue = round(returnvalue, int(options['precision']))
# except:
# pilib.writedatedlogmsg(pilib.iolog, 'Error on precision operation', 0, logconfig['iologlevel'])
pilib.log(pilib.iolog, 'Values read: ' + str(values), 4, logconfig['iologlevel'])
pilib.log(pilib.iolog, 'Value returned: ' + str(returnvalue), 4, logconfig['iologlevel'])
# Contruct entry for newly acquired data
querylist.append('insert into inputs values (\'' + mbid + '\',\'' + interface['id'] + '\',\'' +
interface['type'] + '\',\'' + str(entry['register']) + '\',\'' + mbname + '\',\'' + str(
returnvalue) + "','','" + str(polltime) + '\',\'' + str(pollfreq) + "','" + ontime + "','" + offtime + "')")
else:
pilib.log(pilib.iolog, 'Statuscode ' + str(readresult['statuscode']) + ' on MB read : ' + readresult['message'] , 0, logconfig['iologlevel'])
# restore previous value and construct entry if it existed (or not)
querylist.append('insert into inputs values (\'' + mbid + '\',\'' + interface['interface'] + '\',\'' +
interface['type'] + '\',\'' + str(entry['register']) + '\',\'' + mbname + '\',\'' + str(prevvalue) + "','','" + str(prevpolltime) + '\',\'' + str(pollfreq) + "','" + ontime + "','" + offtime + "')")
pilib.log(pilib.iolog, 'Querylist: ' + str(querylist) , 4, logconfig['iologlevel'])
return querylist
def setstationmode(netconfigdata=None):
pilib.log(pilib.networklog, 'Setting station mode. ', 3, pilib.networkloglevel)
from time import sleep
if not netconfigdata:
pilib.log(pilib.networklog, 'Retrieving unfound netconfig data. ', 3, pilib.networkloglevel)
try:
netconfigdata = pilib.readonedbrow(pilib.systemdatadatabase, 'netconfig')[0]
except:
pilib.log(pilib.networklog, 'Error reading netconfig data. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Read netconfig data. ', 4, pilib.networkloglevel)
killapservices()
if netconfigdata['addtype'] == 'static':
pilib.log(pilib.networklog, 'Configuring static address. ', 3, pilib.networkloglevel)
subprocess.call(['/bin/cp', '/usr/lib/iicontrollibs/misc/interfaces/interfaces.sta.static', '/etc/network/interfaces'])
# update IP from netconfig
pilib.log(pilib.networklog, 'Updating netconfig with ip ' + netconfigdata['address'], 3, pilib.networkloglevel)
replaceifaceparameters('/etc/network/interfaces', '/etc/network/interfaces', 'wlan0', ['address', 'gateway'],
[netconfigdata['address'], netconfigdata['gateway']])
elif netconfigdata['addtype'] == 'dhcp':
pilib.log(pilib.networklog, 'Configuring dhcp. ', 3, pilib.networkloglevel)
subprocess.call(['/bin/cp', '/usr/lib/iicontrollibs/misc/interfaces/interfaces.sta.dhcp', '/etc/network/interfaces'])
pilib.log(pilib.networklog, 'Resetting wlan. ', 3,pilib.networkloglevel)
resetwlan()
sleep(1)
resetwlan()
# specified interval, sends to log, channels and plot dbs
# TODO: Update to include better logging
import pilib
import updateio
from time import sleep
readtime = 10 # default, seconds
# Read from systemstatus to make sure we should be running
updateioenabled = pilib.getsinglevalue(pilib.controldatabase, 'systemstatus', 'updateioenabled')
while updateioenabled:
pilib.log(pilib.iolog, 'Running periodicupdateio', 3, pilib.iologlevel)
pilib.log(pilib.syslog, 'Running periodicupdateio', 3, pilib.sysloglevel)
# Set last run time
pilib.setsinglevalue(pilib.controldatabase, 'systemstatus', 'lastinputpoll', pilib.gettimestring())
pilib.setsinglevalue(pilib.controldatabase, 'systemstatus', 'updateiostatus', '1')
# Read and record everything as specified in controldatabase
# Update database of inputs with read data
# DEBUG
runupdate = True
if runupdate:
reply = updateio.updateiodata(pilib.controldatabase)
else:
pilib.log(pilib.iolog,'DEBUG: Update IO disabled', 1, pilib.iologlevel)
runconfig=True
elif netconfigdata['mode'] == 'wlan0wlan1bridge' or 'wlan1wlan0bridge':
if netconfigdata['mode'] == 'wlan0wlan1bridge':
stationinterface = 'wlan0'
apinterface = 'wlan1'
else:
stationinterface = 'wlan1'
apinterface = 'wlan0'
runconfig = False
# Check station address (not yet implemented) and wpa status (implemented)
try:
stationifacedata = ifacedict[stationinterface]
except KeyError:
pilib.log(pilib.networklog, 'No stationiface data(' + stationinterface + ') present for mode ' + netconfigdata['mode'], 1, pilib.networkloglevel)
statusmsg += 'No stationiface data(' + stationinterface + ') present for mode ' + netconfigdata['mode']
runconfig = True
else:
wpadata = pilib.parseoptions(stationifacedata['wpastate'])
if wpadata['wpa_state'] == 'COMPLETED':
pilib.log(pilib.networklog, 'station interface ' + stationinterface + ' wpastatus appears ok. ', 3, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'station interface for ' + stationinterface + ' does not appear ok judged by wpa_state. ', 1, pilib.networkloglevel)
statusmsg += 'station interface does not appear ok judged by wpa_state. '
runconfig = True
# Check wlan1 dhcp and hostapd status
try:
apifacedata = ifacedict[apinterface]
except KeyError:
def setrawspilights(enabledlists, CS=1, **kwargs):
from pilib import log, syslog, sysloglevel
if "method" in kwargs:
method = kwargs["method"]
else:
method = "spidev"
# Notes:
# Low is on. Cathodes are open drain.
# This unforunately means we need to initialize
# to off on start-up. not a huge deal
spiassignments = []
for enabledlist in enabledlists:
bytesum = 0
for index, bit in enumerate(enabledlist):
bytesum += bit * (2 ** index)
spiassign = 255 - bytesum
spiassignments.append(spiassign)
resp = []
if method == "spidev":
try:
import spidev
except ImportError:
log(syslog, "You have not installed the spidev python package. Exiting.", 0, sysloglevel)
exit
else:
spi = spidev.SpiDev()
try:
spi.open(0, CS) # Port 0, CS1
spi.max_speed_hz = 50000
except:
log(syslog, "Error raised on spi open. exiting.", 0, sysloglevel)
exit
else:
# Transfer bytes
# print('spi assignments')
# print(spiassignments)
resp = spi.xfer2(spiassignments)
# resp = spi.xfer2(spiassignments)
elif method == "pigpio":
import pigpio
import array
if "piobject" in kwargs:
pi = kwargs["piobject"]
else:
pi = pigpio.pi()
handle = pi.spi_open(0, 50000, 0)
resp = pi.spi_write(handle, array.array(spiassignments).tostring())
print(resp)
if method == "pigpio" and "piobject" not in kwargs:
pi.stop()
return resp
def watchdognetstatus(allnetstatus=None):
import pilib
import netconfig
import subprocess
from cupiddaemon import pgrepstatus
"""
And now comes the checking of configuration specific statuses and restarting them if enabled
and necessary
NOW
Set number configuration types:
eth0 is always dhcp
if interface is not specified, it is assumed that the specified configuration applies to wlan0
ap : access point on wlan0.
Currently, address is hard-coded as 192.168.0.1 as set in hostapd (to have dhcp all on same subnet)
station : station mode on wlan0
IP address mode set in 'addtype' field. If DHCP, set as DHCP. If static, set as address field
eth0wlan0bridge :
wlan0 set as AP mode with bridge to eth0
wlan0wlan1bridge :
wlan0 set as station mode, wlan1 as ap mode with bridge
wlan1wlan0bridge
wlan1 set as station mode, wlan0 as ap mode with bridge
IN THE FUTURE
netconfigdata will have a row for each interface that needs to be configured.
Each interface can be either static, dhcp, or ap.
Each interface can also bridge, but this will be limited at first, due to the number of combinations.
If bridge is selected, the mode will be checked, and the appropriate IPTables will be set based on which one is AP
"""
if not allnetstatus:
allnetstatus = updatenetstatus()
netconfigdata = allnetstatus['netconfigdata']
netstatus = allnetstatus['netstatusdict']
# Refactor ifaces array. We could probably do this earlier and pass it around like this
ifacedict = {}
for iface in allnetstatus['ifacesdictarray']:
# print(iface['name'])
# print(iface)
ifacedict[iface['name']] = iface
# print(ifacedict)
statusmsg = ''
currenttime = pilib.gettimestring()
runconfig = False
if netconfigdata['mode'] in ['ap', 'tempap']:
pilib.log(pilib.networklog, 'AP Mode is set. ', 1, pilib.networkloglevel)
"""
THIS NEEDS TO BE REWRITTEN ( or absorbed into eth0wlan0bridge )
"""
if netconfigdata['mode'] == 'tempap':
timesincelastretry = pilib.timestringtoseconds(currenttime) - pilib.timestringtoseconds(netconfigdata['laststationretry'])
pilib.log(pilib.networklog, 'TempAP mode: Time since last retry: ' + str(timesincelastretry) + '. Station retry time: '
+ str(netconfigdata['stationretrytime']), 1, pilib.networkloglevel)
if timesincelastretry > netconfigdata['stationretrytime']:
# We go back to station mode
statusmsg += 'Time to go back to station mode. '
pilib.log(pilib.networklog, 'Time to go back to station mode. ', 1, pilib.networkloglevel)
pilib.setsinglevalue(pilib.systemdatadatabase, 'netconfig', 'mode', 'station')
pilib.setsinglevalue(pilib.systemdatadatabase, 'netconfig', 'laststationretry', '')
netconfig.runconfig()
else:
# If we have ap up, do nothing
if netstatus['dhcpstatus']:
statusmsg += 'AP checked and ok. '
pilib.log(pilib.networklog, 'AP checked and ok. ', 1, pilib.networkloglevel)
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'mode', 'ap')
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'SSID', 'cupidwifi')
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'offlinetime', '')
# If we don't have dhcp up, restart ap mode
# this will currently cause reboot if we don't set onboot=True
# We set status message in case we change our minds and reboot here.
else:
statusmsg += 'Restarting AP. '
pilib.log(pilib.networklog, 'Restarting AP mode. ', 1, pilib.networkloglevel)
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'statusmsg', statusmsg)
netconfig.runconfig()
# Station Mode. Again, like above, this should be modular.
elif netconfigdata['mode'] == 'station':
pilib.log(pilib.networklog, 'Station mode is set. ', 3, pilib.networkloglevel)
# Check station address (not yet implemented) and wpa status (implemented)
stationinterface = 'wlan0'
try:
stationifacedata = ifacedict[stationinterface]
except KeyError:
pilib.log(pilib.networklog, 'No stationiface data(' + stationinterface + ') present for mode ' + netconfigdata['mode'], 1, pilib.networkloglevel)
statusmsg += 'No stationiface data(' + stationinterface + ') present for mode ' + netconfigdata['mode']
runconfig = True
else:
wpadata = pilib.parseoptions(stationifacedata['wpastate'])
if wpadata['wpa_state'] == 'COMPLETED' and stationifacedata['address']:
pilib.log(pilib.networklog, 'station interface ' + stationinterface + ' wpastatus appears ok with address ' + str(stationifacedata['address']), 3, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'station interface for ' + stationinterface + ' does not appear ok judged by wpa_state: wpastate = ' + wpadata['wpa_state'] + ' address= ' + stationifacedata['address'], 1, pilib.networkloglevel)
statusmsg += 'station interface does not appear ok judged by wpa_state. '
runconfig = True
# Should do special handling in here to verify address for static mode.
# # If we have wpa up, do nothing
# if int(netstatus['connected']):
# statusmsg += 'Station wpamode appears ok. '
# pilib.log(pilib.networklog, 'wpamode appears ok. ', 1, pilib.networkloglevel)
#
# # If wpa is not connected
# else:
# statusmsg += 'Station wpamode appears disconnected. '
# pilib.log(pilib.networklog, 'wpamode appears disconnected. ', 1, pilib.networkloglevel)
#
# if netstatus['offlinetime'] == '':
# pilib.log(pilib.networklog, 'Setting offline time for empty value. ', 4,
# pilib.networkloglevel)
# pilib.setsinglevalue('netstatus', 'offlinetime', pilib.gettimestring())
# offlinetime = 0
# else:
# pilib.log(pilib.networklog, 'Calculating offline time. ', 4, pilib.networkloglevel)
# offlinetime = pilib.timestringtoseconds(currenttime) - pilib.timestringtoseconds(
# netstatus['offlinetime'])
#
# pilib.log(pilib.networklog, 'wpa has been offline for ' + str(offlinetime) + '. ', 3,
# pilib.networkloglevel)
# statusmsg += 'We have been offline for ' + str(offlinetime) + '. '
#
# # If aprevert is aprevert or temprevert and we've been offline long enough, flip over to ap
# # then set offline time to now (otherwise this keeps happening)
# if netconfigdata['aprevert'] in ['temprevert', 'aprevert'] and offlinetime > netconfigdata[
# 'apreverttime']:
#
# # set laststationretry to currenttime. This marks when we flipped over to ap
# statusmsg += 'Setting last station retry time. '
# pilib.log(pilib.networklog, 'Reverting to AP mode', 3, pilib.networkloglevel)
# pilib.log(pilib.networklog,
# 'Setting last station retry time to ' + str(currenttime), 0,
# pilib.networkloglevel)
# pilib.setsinglevalue(pilib.systemdatadatabase, 'netconfig', 'laststationretry', currenttime)
# pilib.setsinglevalue('netstatus', 'offlinetime', currenttime)
#
# if netconfigdata['aprevert'] == 'aprevert':
# # set mode to ap
# statusmsg += 'Setting mode to ap. '
# pilib.log(pilib.networklog, 'Setting mode to ap ' + str(currenttime), 3,
# pilib.networkloglevel)
# pilib.setsinglevalue(pilib.systemdatadatabase, 'netconfig', 'mode', 'ap')
# elif netconfigdata['aprevert'] == 'temprevert':
# # set mode to tempap
# statusmsg += 'Setting mode to tempap. '
# pilib.log(pilib.networklog, 'Setting mode to tempap ' + str(currenttime), 3,
# pilib.networkloglevel)
# pilib.setsinglevalue(pilib.systemdatadatabase, 'netconfig', 'mode', 'tempap')
#
# # Unfortunately, to revert to ap mode successfully, we currently have to reboot
# # this is built into the netconfig script - any time you set ap mode except at boot, it reboots
# pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'statusmsg', statusmsg)
# pilib.log(pilib.syslog, 'Running netconfig . ', 4,
# pilib.sysloglevel)
# netconfig.runconfig()
# elif offlinetime > 15:
# pilib.log(pilib.syslog, 'Restarting netconfig on bad wpastatus', 1,
# pilib.sysloglevel)
# runconfig = True
#
# # Here, we need to check the ifaces address. Netstatus address is ambiguous
# if netstatus['ip_address'] != netconfigdata['address']:
# pilib.log(pilib.networklog, 'IP address mismatch ( Configured for ' + netconfigdata['address'] + '. Reporting' + netstatus['ip_address'] + ' ). Running config.', 1, pilib.networkloglevel)
# runconfig = True
# else:
# pilib.log(pilib.networklog, 'IP address match for ' + netconfigdata['address'] + '. ', 3, pilib.networkloglevel)
elif netconfigdata['mode'] == 'eth0wlan0bridge':
# We don't actually check eth0. This is because we shouldn't have to. Also, if we don't check on eth0, we can
# use the same mode for wlan0 AP and eth0wlan0 bridge. Hot plug, works fine.
# Check wlan0 dhcp and hostapd status
try:
wlan0ifacedata =ifacedict['wlan0']
except KeyError:
pilib.log(pilib.networklog, 'No wlan0 data present in configuration of eth0wlan0bridge. ', 1, pilib.networkloglevel)
statusmsg += 'wlan0 data is not present. '
runconfig = True
else:
pilib.log(pilib.networklog, 'Checking dhcp server status on wlan0. ', 4, pilib.networkloglevel)
try:
result = subprocess.check_output(['/usr/sbin/service', 'isc-dhcp-server', 'status'], stderr=subprocess.PIPE)
except Exception, e:
# If not running, the subprocess call will throw an error
pilib.log(pilib.networklog, 'Error in reading dhcp server status. Assumed down. ', 1, pilib.networkloglevel)
statusmsg += 'dhcp server appears down. '
runconfig = True
else:
pilib.log(pilib.networklog, 'DHCP server appears to be up. ', 1, pilib.networkloglevel)
""" Check ifconfig ipaddress for wlan0
This should be programmable. """
if wlan0ifacedata['address'] == '192.168.0.1':
pilib.log(pilib.networklog, 'wlan0 address is appears to be set properly: ' + str(wlan0ifacedata['address']), 3, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'wlan0 address is not set properly: ' + str(wlan0ifacedata['address']), 1, pilib.networkloglevel)
statusmsg += 'wlan0 address does not appear ok. '
runconfig = True
if pgrepstatus('hostapd.*wlan0')['count'] == 1:
pilib.log(pilib.networklog, 'hostapd on wlan0 appears to be ok. ', 3, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'hostapd on wlan0 does NOT appear to be ok. ', 1, pilib.networkloglevel)
statusmsg += 'wlan0 hostpad does not appear ok. '
runconfig=True
def rundaemon(startall=False):
from subprocess import Popen, PIPE
from time import sleep
from pilib import log, daemonlog, daemonloglevel
# Set up list of enabled statuses (whether to restart if
# we find that the process is not currently running
picontrolenabled = pilib.sqlitedatumquery(pilib.controldatabase, 'select picontrolenabled from systemstatus')
updateioenabled = pilib.sqlitedatumquery(pilib.controldatabase, 'select updateioenabled from systemstatus')
systemstatusenabled = pilib.sqlitedatumquery(pilib.controldatabase, 'select systemstatusenabled from systemstatus')
sessioncontrolenabled = pilib.sqlitedatumquery(pilib.controldatabase, 'select sessioncontrolenabled from systemstatus')
serialhandlerenabled = pilib.sqlitedatumquery(pilib.controldatabase, 'select serialhandlerenabled from systemstatus')
enableditemlist = [(int(updateioenabled)), (int(picontrolenabled)), int(systemstatusenabled), int(sessioncontrolenabled), int(serialhandlerenabled)]
itemstatuses = findprocstatuses(pilib.daemonprocs)
# Set system message
systemstatusmsg = ''
for name, enabled, status in zip(pilib.daemonprocs, enableditemlist, itemstatuses):
systemstatusmsg += name + ' - Enabled: ' + str(enabled) + ' Status: ' + str(status) + '. '
if pilib.daemonloglevel > 0:
pilib.log(pilib.daemonlog, name + ' - Enabled: ' + str(enabled) + ' Status: ' + str(status) + '. ')
pilib.setsinglevalue(pilib.controldatabase, 'systemstatus', 'systemmessage', systemstatusmsg)
# Set up list of itemnames in the systemstatus table that
# we assign the values to when we detect if the process
# is running or not
statustableitemnames = ['updateiostatus', 'picontrolstatus', 'systemstatusstatus', 'sessioncontrolstatus', 'serialhandlerstatus']
for index, item in enumerate(pilib.daemonprocs):
# set status
if itemstatuses[index]:
pilib.sqlitequery(pilib.controldatabase, 'update systemstatus set ' + statustableitemnames[index] + ' = 1')
if pilib.daemonloglevel > 0:
pilib.log(pilib.daemonlog, 'Process is running: ' + pilib.baselibdir + pilib.daemonprocs[index])
else:
pilib.sqlitequery(pilib.controldatabase, 'update systemstatus set ' + statustableitemnames[index] + ' = 0')
if pilib.daemonloglevel > 0:
pilib.log(pilib.daemonlog, 'Process is not running: ' + pilib.baselibdir + pilib.daemonprocs[index])
# run if set to enable
if enableditemlist[index]:
# print(pilib.baselibdir + pilib.daemonprocs[index])
if pilib.daemonloglevel > 0:
pilib.log(pilib.daemonlog, 'Starting ' + pilib.baselibdir + pilib.daemonprocs[index])
procresult = Popen([pilib.baselibdir + pilib.daemonprocs[index]], stdout=PIPE, stderr=PIPE)
# if pilib.daemonloglevel > 0:
# pilib.writedatedlogmsg(pilib.daemonproclog, procresult.stdout.read())
sleep(1)
# Refresh after set
itemstatuses = findprocstatuses(pilib.daemonprocs)
for item in pilib.daemonprocs:
index = pilib.daemonprocs.index(item)
# set status
if itemstatuses[index]:
pilib.sqlitequery(pilib.controldatabase, 'update systemstatus set ' + statustableitemnames[index] + ' = 1')
else:
pilib.sqlitequery(pilib.controldatabase, 'update systemstatus set ' + statustableitemnames[index] + ' = 0')
# Set system message
systemstatusmsg = ''
for name, enabled, status in zip(pilib.daemonprocs, enableditemlist, itemstatuses):
systemstatusmsg += name + ' - Enabled: ' + str(bool(enabled)) + ' Status: ' + str(status) + '. '
pilib.setsinglevalue(pilib.controldatabase, 'systemstatus','systemmessage', systemstatusmsg)
# Rotate all logs
pilib.rotatelogs(pilib.networklog, pilib.numlogs, pilib.maxlogsize)
pilib.rotatelogs(pilib.syslog, pilib.numlogs, pilib.maxlogsize)
pilib.rotatelogs(pilib.iolog, pilib.numlogs, pilib.maxlogsize)
pilib.rotatelogs(pilib.controllog, pilib.numlogs, pilib.maxlogsize)
pilib.rotatelogs(pilib.daemonlog, pilib.numlogs, pilib.maxlogsize)
pilib.log(pilib.networklog, 'Copied network configuration file successfully. ', 3, pilib.networkloglevel)
killapservices()
resetwlan()
startapservices(interface)
def resetwlan(interface='wlan0'):
pilib.log(pilib.networklog, 'Resetting ' + interface + ' . ', 3, pilib.networkloglevel)
try:
subprocess.check_output(['/sbin/ifdown', '--force', interface], stderr=subprocess.PIPE)
subprocess.call(['/sbin/ifup', interface], stderr=subprocess.PIPE)
except Exception, e:
pilib.log(pilib.networklog, 'Error resetting ' + interface + ' : ' + str(e), 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Completed resetting '+ interface + '. ', 3, pilib.networkloglevel)
def runconfig(onboot=False):
import subprocess
try:
netconfigdata = pilib.readonedbrow(pilib.systemdatadatabase, 'netconfig')[0]
# print(netconfigdata)
except:
pilib.log(pilib.networklog, 'Error reading netconfig data. ', 0, pilib.networkloglevel)
else:
pilib.log(pilib.networklog, 'Successfully read netconfig data', 3, pilib.networkloglevel)
pilib.setsinglevalue(pilib.systemdatadatabase, 'netstatus', 'mode', netconfigdata['mode'])
pilib.log(pilib.networklog, 'Netconfig is enabled', 3, pilib.networkloglevel)
