netmo/web_server.py at master · codebox/netmo · GitHub

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
from SimpleHTTPServer import SimpleHTTPRequestHandler
from BaseHTTPServer import HTTPServer, BaseHTTPRequestHandler
from SocketServer import ThreadingMixIn
import time
import json
from Queue import Queue


class RequestHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        if self.path == '/data':
            self.send_response(200)
            self.send_header("Content-type", "text/event-stream")
            self.end_headers()
            queue = Queue(0)
            self.server.packet_queue.subscribe(queue)

            while True:
                item = queue.get(True)
                sdata = '{"host" : "%s", "port" : "%s", "data" : "%s"}' % (item['src'], item['dport'], item['data'])
                try:
                    self.wfile.write('retry: 1000\ndata: ' + sdata + '\n\n')
                except:
                    self.server.packet_queue.unsubscribe(queue)
                    return

        elif self.path == '/dns':
            self.send_response(200)
            self.send_header("Content-type", "text/event-stream")
            self.end_headers()
            queue = Queue(0)
            self.server.dns_queue.subscribe(queue)

            while True:
                item = queue.get(True)
                sdata = '{"host" : "%s", "name" : "%s"}' % (item['host'], item['name'])
                try:
                    self.wfile.write('retry: 1000\ndata: ' + sdata + '\n\n')
                except:
                    self.server.dns_queue.unsubscribe(queue)
                    return

        else:
            try:
                content, content_type = self.server.file_manager.read(self.path)
                self.send_response(200)
                self.send_header("Content-type", content_type)
                self.end_headers()
                self.wfile.write(content)
            except:
                self.send_response(404)


class ThreadedHTTPServer(ThreadingMixIn, HTTPServer):
    def __init__(self, server_address, RequestHandlerClass, file_manager, packet_queue, dns_queue):
        HTTPServer.__init__(self, server_address, RequestHandlerClass)
        self.file_manager = file_manager
        self.packet_queue = packet_queue
        self.dns_queue    = dns_queue

class WebServer:
    def __init__(self, file_manager, packet_queue, dns_queue):
        self.file_manager = file_manager
        self.packet_queue = packet_queue
        self.dns_queue    = dns_queue

    def start(self, host, port):
        ThreadingMixIn.daemon_threads = True
        httpd = ThreadedHTTPServer((host, port), RequestHandler, self.file_manager, self.packet_queue, self.dns_queue)
        print time.asctime(), "Server Starts - %s:%s" % (host, port)
        try:
            httpd.serve_forever()
        except KeyboardInterrupt:
            pass
        httpd.server_close()
        print time.asctime(), "Server Stops - %s:%s" % (host, port)