#!/usr/bin/env python
import sys
sys.path.append("../waggle_protocol/")
from utilities import packetmaker
from send import send
"""
A python script that creates, packs, and sends a ping.
"""
packet = packetmaker.make_ping_packet()
print "Ping packet made..."
for pack in packet:
send(pack)
def msg_handler(msg, DEVICE_DICT):
"""
Unpacks and acts on messages sent to the node.
:param string msg: The packed message sent to the node.
"""
try:
#unpacks the header
header = get_header(msg)
except:
print 'Message is corrupt: ', msg #TODO should this send some kind of error response?
#get the major header type
major = chr(header['msg_mj_type'])
minor = chr(header['msg_mi_type'])
#large file transfer
if major == 'F':
#TODO pass to packet reassembler
pass
#small file transfer
elif major == 'f':
#TODO do something with this file
pass
#ping
elif major == 'p':
#ping request.
if minor == 'r':
#send a ping response
packet = packetmaker.make_ping_packet()
for pack_ in packet:
send(pack_)
#ping answer
else:
#unpack the message
ping = unpack(msg)
#print out the body of the message
print 'Node received: ', ping[1]
#time
elif major == 't':
#time request
if minor == 'r':
#send time
packet = packetmaker.make_time_packet()
for pack_ in packet:
send(pack_)
#time answer
else:
#unpack the message
time = unpack(msg)
#print out the body of the message
print 'Node received time: ', time[1]
#sensor data
elif major == 's':
#TODO do stuff here
pass
#registration
elif major =='r':
sender = header['s_uniqid']
if minor == 'r': #registration request
if sender == 0:
#message is from cloud
#unpack the message
reg = unpack(msg)
print 'NC received registration: ', reg[1]
else:
#guest node registration
sender = str(sender) #convert from int to string
with open('/etc/waggle/devices', 'r') as _file:
lines = _file.readlines()
#check if device is already registered
devices = []
#the first line of the file contains a list of already registered nodes
#deconstruct string into list
while not lines[0].find(',')== -1:
device, lines[0] = lines[0].split(',',1)
devices.append(device)
try:
devices.index(sender) #if this works, the device is already registered
#nothing else to be done
except:
#if it fails, the device is not yet registered. Add to list of devices
print 'Adding device ',sender, 'to devices file.'
devices.append(sender)
#Need to find available priorities to assign it
#the second line of the file contains a list of available priorities
priorities = []
while not lines[1].find(',')== -1:
priority, lines[1] = lines[1].split(',',1)
priorities.append(priority)
device_p = priorities.pop()
if not device_p == '': #make sure it isn't empty
#assign the device to its priority
#the third line of the file contains a mapping of device to its priority.
#This is what is used to construct the device_dict
lines[2] = sender + ':' + device_p + ',' + lines[2]
#put the list back together to be written back into the file
for priority in priorities:
lines[1] = priority + ',' + lines[1]
#send GN registration to cloud
header_dict = {
"msg_mj_type" : ord('r'),
"msg_mi_type" : ord('r'),
"s_uniqid" : int(sender)
}
msg = str(QUEUENAME)
try:
packet = pack(header_dict, message_data = msg)
print 'Registration made for node ID ', sender
for pack_ in packet:
send(pack_)
except Exception as e:
print e
else:
print 'Cannot register any more guestnodes. Maximum of four guestnodes can be registered at one time. Please remove any unused guestnode IDs from the devices file.'
#put the list back together to be written back into the file
for device in devices:
lines[0] = device + ',' + lines[0]
#write the lines back into the file
with open('/etc/waggle/devices', 'w') as _file:
_file.writelines(lines)
elif minor == 'd': #de-registration request from guest node
sender = str(sender) #convert from int to string
print 'Received de-registration message from node ', sender
with open('/etc/waggle/devices', 'r') as _file:
lines = _file.readlines()
#check if device is already registered and remove from devices list
devices = []
#the first line of the file contains a list of already registered nodes
#deconstruct string into list
while not lines[0].find(',')== -1:
device, lines[0] = lines[0].split(',',1)
devices.append(device)
#now, need to get the priorities list
priorities = []
while not lines[1].find(',')== -1:
priority, lines[1] = lines[1].split(',',1)
priorities.append(priority)
#need to remove the device mapping
mapping = []
while not lines[2].find(',')== -1:
device_map, lines[2] = lines[2].split(',',1)
mapping.append(device_map)
#try to remove the device from the devices list
devices.remove(sender)
#get the device's priority/location from dictionary
device_p = DEVICE_DICT[sender]
#add the device's priority/location back to the list
priorities.append(str(device_p))
dev_map = sender + ':' + str(device_p)
#remove device mapping from file
mapping.remove(dev_map)
#put everything back together to be written back onto the file
for priority in priorities:
lines[1] = priority + ',' + lines[1]
for device in devices:
lines[0] = device + ',' + lines[0]
for maps in mapping:
lines[2] = maps + ',' + lines[2]
#write the lines back into the file
with open('/etc/waggle/devices', 'w') as _file:
_file.writelines(lines)
#message type unrecognized
else:
print 'Message major type, ' , major, ' unrecognized.'
