tuerd

   1 #!/usr/bin/python3
   2 import time, socket, os, stat, atexit
   3 from datetime import datetime
   4 import RPi.GPIO as GPIO
   5 GPIO.setmode(GPIO.BOARD)
   6
   7 # ******** definitions *********
   8 # logging function
   9 # TODO: loglevel, log like a real daemon
  10 def log (what):
  11         print (datetime.now(),what)
  12
  13 # send to client for information but don't care if it arrives
  14 def waynesend (conn, what):
  15         try:
  16                 conn.send(what)
  17         except:
  18                 log("Couldn't send %s" % str(what))
  19
  20 # for command not found: do nothing with the pins and send a "0" to the client
  21 def doNothing (conn):
  22         log ("doing nothing")
  23         waynesend(conn,b"0")
  24         pass
  25
  26 # commands: on a pin do a series of timed on/off switches
  27 class Pinoutput:
  28         # name is for logging and also used for mapping command names to instances of this class
  29         # actionsanddelays is a list of pairs: (bool to set on pin, delay in seconds to wait afterwards)
  30         def __init__ (self, name, pinnumber, actionsanddelays):
  31                 self.name = name
  32                 self.pin = pinnumber
  33                 self.todo = actionsanddelays
  34                 GPIO.setup(pinnumber, GPIO.OUT)
  35                 log ("Pin %d set to be an output pin for %s." % (pinnumber,name))
  36         # actually send the signal to the pins
  37         def __call__ (self, conn):
  38                 for (value,delay) in self.todo:
  39                         GPIO.output(self.pin, value)
  40                         log ("%s: Pin %d set to %s." % (self.name,self.pin,str(value)))
  41                         time.sleep(delay)
  42                 # notify success
  43                 waynesend(conn,b"1")
  44
  45 # ******** configuration *********
  46
  47 tuergroupid = 1005
  48 socketname = "/run/tuer.sock"
  49 pinlist = [Pinoutput("open", 12, [(True, 0.3), (False, 5.0)]),
  50         Pinoutput("close", 16, [(True, 0.3), (False, 5.0)]),
  51         Pinoutput("buzz", 22, [(True, 2.0), (False, 0.1)])]
  52
  53
  54 # ******** main *********
  55 # at the end do a cleanup
  56 atexit.register(GPIO.cleanup);
  57
  58 # convert list of pin objects to dictionary for command lookup
  59 pindict = {}
  60 for pin in pinlist:
  61         pindict[pin.name.encode()] = pin
  62
  63 # create socket
  64 sock = socket.socket (socket.AF_UNIX, socket.SOCK_STREAM)
  65 # delete old socket file and don't bitch around if it's not there
  66 try:
  67         os.unlink (socketname)
  68 except OSError:
  69         pass
  70 # bind socket to file name
  71 sock.bind (socketname)
  72 # allow only users in the tuergroup to write to the socket
  73 os.chown (socketname, 0, tuergroupid)
  74 os.chmod (socketname, stat.S_IRUSR | stat.S_IWUSR | stat.S_IRGRP | stat.S_IWGRP)
  75 # listen to the people, but not too many at once
  76 sock.listen(1)
  77
  78 # main loop
  79 # FIXME: DoS by opening socket but not sending data, because this loop is single threaded; maybe settimeout helps a bit.
  80 while True:
  81         # accept connections
  82         conn, addr = sock.accept()
  83         # TODO: use addr to determine the client for logging
  84         # get some data from the client (enough to hold any valid command)
  85         data = conn.recv (32)
  86         # log the command
  87         log("received command: %s" % str(data))
  88         # lookup the command, if it's not in the dict, use the doNothing function instead
  89         # and execute the looked up command or doNothing with the connection, so it can respond to the client
  90         pindict.get(data,doNothing)(conn)
  91         # close connection cleanly
  92         conn.close()
  93