build/scripts/bootstrap_daemon.py

   1 #!/usr/bin/env python
   2
   3 import socket
   4 import subprocess
   5 import sys
   6
   7
   8 address = '0.0.0.0'
   9 port = 4242
  10
  11
  12 def receiveExactly(connection, size):
  13         data = '';
  14         while size > 0:
  15                 dataReceived = connection.recv(size)
  16                 if not dataReceived:
  17                         raise EOFError()
  18                 data += dataReceived
  19                 size -= len(dataReceived)
  20         return data
  21
  22
  23 def handleConnection(listenerSocket):
  24         (controlConnection, controlAddress) = listenerSocket.accept()
  25         (stdioConnection, stdioAddress) = listenerSocket.accept()
  26         (stderrConnection, stderrAddress) = listenerSocket.accept()
  27
  28         print 'accepted client connections'
  29
  30         try:
  31                 commandLength = receiveExactly(controlConnection, 8)
  32                 commandToRun = receiveExactly(controlConnection, int(commandLength))
  33
  34                 print 'received command: ' + commandToRun
  35
  36                 exitCode = subprocess.call(commandToRun, stdin=stdioConnection,
  37                         stdout=stdioConnection, stderr=stderrConnection, shell=True)
  38
  39                 controlConnection.send(str(exitCode))
  40         finally:
  41                 controlConnection.close()
  42                 stdioConnection.close()
  43                 stderrConnection.close()
  44
  45         print 'client connections closed'
  46
  47
  48 listenerSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  49 listenerSocket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
  50
  51 try:
  52         listenerSocket.bind((address, port))
  53 except socket.error, msg:
  54         sys.exit('Failed to bind to %s port %d: %s' % (address, port, msg[1]))
  55
  56 listenerSocket.listen(3)
  57
  58 print 'started listening on adddress %s port %s' % (address, port)
  59
  60 while True:
  61         handleConnection(listenerSocket)