/usr/lib/python2.7/dist-packages/lavapdu/socketserver.py is in lavapdu-daemon 0.0.5-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 | #! /usr/bin/python
# Copyright 2013 Linaro Limited
# Author Matt Hart <matthew.hart@linaro.org>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
# MA 02110-1301, USA.
import SocketServer
import logging
import socket
import time
import json
from lavapdu.dbhandler import DBHandler
class ListenerServer(object):
def __init__(self, settings):
listen_host = settings["hostname"]
listen_port = settings["port"]
logging.getLogger().name = "ListenerServer"
logging.getLogger().setLevel(settings["logging_level"])
logging.debug("ListenerServer __init__")
logging.info("listening on %s:%s", listen_host, listen_port)
self.server = TCPServer((listen_host, listen_port), TCPRequestHandler)
self.server.settings = settings
dbh = DBHandler(settings)
dbh.create_db()
dbh.close()
del dbh
def start(self):
logging.info("Starting the ListenerServer")
self.server.serve_forever()
class TCPRequestHandler(SocketServer.BaseRequestHandler):
#"One instance per connection. Override handle(self) to customize action."
def insert_request(self, data):
logging.getLogger().name = "TCPRequestHandler"
logging.getLogger().setLevel(self.server.settings["logging_level"])
array = data.split(" ")
delay = 10
custom_delay = False
now = int(time.time())
if (len(array) < 3) or (len(array) > 4):
logging.info("Wrong data size")
raise Exception("Unexpected data")
if len(array) == 4:
delay = int(array[3])
custom_delay = True
hostname = array[0]
port = int(array[1])
request = array[2]
if not (request in ["reboot", "on", "off"]):
logging.info("Unknown request: %s", request)
raise Exception("Unknown request: %s", request)
if request == "reboot":
logging.debug("reboot requested, submitting off/on")
self.queue_request(hostname, port, "off", now)
self.queue_request(hostname, port, "on", now+delay)
else:
if custom_delay:
logging.debug("using delay as requested")
self.queue_request(hostname, port, request, now+delay)
else:
self.queue_request(hostname, port, request, now)
def queue_request(self, hostname, port, request, exectime):
dbhandler = DBHandler(self.server.settings)
sql = "insert into pdu_queue (hostname,port,request,exectime) " \
"values ('%s',%i,'%s',%i)" % (hostname, port, request, exectime)
dbhandler.do_sql(sql)
dbhandler.close()
del dbhandler
def handle(self):
logging.getLogger().name = "TCPRequestHandler"
request_ip = self.client_address[0]
try:
data = self.request.recv(4096).strip()
socket.setdefaulttimeout(2)
try:
request_host = socket.gethostbyaddr(request_ip)[0]
except socket.herror as e: #pylint: disable=invalid-name
#logging.debug("Unable to resolve: %s error: %s" % (ip,e))
request_host = request_ip
logging.info("Received a request from %s: '%s'", request_host, data)
self.insert_request(data)
self.request.sendall("ack\n")
except Exception as e: #pylint: disable=invalid-name
logging.debug(e.__class__)
logging.debug(e.message)
self.request.sendall("nack\n")
self.request.close()
class TCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
allow_reuse_address = True
daemon_threads = True
#pass
if __name__ == "__main__":
logging.basicConfig(level=logging.DEBUG)
logging.getLogger().setLevel(logging.DEBUG)
logging.debug("Executing from __main__")
filename = "/etc/lavapdu/lavapdu.conf"
print("Reading settings from %s" % filename)
with open(filename) as stream:
jobdata = stream.read()
json_data = json.loads(jobdata)
ss = ListenerServer(json_data["daemon"])
ss.start()
|