robot0nfire/fl0w-old
robot0nfire/fl0w-old
Archived
1
0
Fork 0
Code Issues 5 Pull requests Projects Releases Packages Wiki Activity Actions

Unfinished; added GetInfo, initial support for new WallabyControl

Browse source
This commit is contained in:
PhilipTrauner 2016-03-29 23:46:46 +02:00
parent 1ee65798b0
commit 2ef7a5c1c1
1 changed files with 31 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

42
Wallaby/Wallaby.py
View file

@ -1,23 +1,31 @@
import fl0w
from ESock import ESock
from Sync import SyncClient
import Routing
import socket
import time
import os
import sys
import platform
CHANNEL = "w"
IS_WALLABY = True if "ARMv7" in platform.uname().version.lower() else False
class WallabyControl(Routing.ClientRoute):
def __init__(self):
self.actions_with_params = {"stop" : self.stop, "run" : self.run}
self.actions_without_params = {"restart" : self.restart, "disconnect" : self.disconnect,
"reboot" : self.reboot, "shutdown" : self.shutdown}
def run(self, data, handler):
commands = {"stop" : self.stop, "restart" : self.restart,
"disconnect" : self.disconnect, "reboot" : self.reboot,
"shutdown" : self.shutdown}
if data in commands:
commands[data](handler)
if data in self.actions_without_params.keys():
self.actions_without_params[action](handler)
elif type(data[address_pair]) is dict:
if action in self.actions_with_params.keys():
self.actions_without_params[action](data[action], handler)
def stop(self, handler):
print("Stop nyi")
os.system("killall -s 2 botball_user_program")
def restart(self, handler):
self.disconnect(handler)
@ -35,6 +43,15 @@ class WallabyControl(Routing.ClientRoute):
handler.sock.close()
class GetInfo(Routing.ClientRoute):
def run(self, data, handler):
if data == "":
handler.sock.send({"type" : CHANNEL, "name" : platform.node()}, "get_info")
elif "name" in data:
if IS_WALLABY:
open("/etc/hostname", "w").write(str(data["name"]))
else:
Logging.info("Hostname change: '%s'" % str(data["name"]))
class WallabyClient:
@ -42,17 +59,20 @@ class WallabyClient:
self.sock = ESock(socket.create_connection(host_port_pair), debug=debug)
self.connected = True
self.debug = debug
self.sync = SyncClient(self.sock, sys.argv[1], "w_sync")
self.routes = {"wallaby_control" : WallabyControl(), "w_sync" : self.sync}
self.sync = SyncClient(self.sock, sys.argv[1], "w_sync", debug=True)
self.routes = {"wallaby_control" : WallabyControl(), "w_sync" : self.sync,
"get_info" : GetInfo()}
def start(self):
self.sock.send("w", "set_type")
self.sync.start()
while 1 and self.connected:
data = self.sock.recv()
if data[1] in self.routes:
try:
if data[1] in self.routes:
self.routes[data[1]].run(data[0], self)
except Exception as e:
def stop(self):