-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathserver.py
More file actions
56 lines (51 loc) · 1.67 KB
/
server.py
File metadata and controls
56 lines (51 loc) · 1.67 KB
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
#! /usr/bin/env python
import SimpleHTTPServer
import SocketServer
import pdb
import time
import re
import collections
import os
import subprocess
import datetime
import json
reporting_time_seconds = 4*7*24*60*60
def dict_for(id, field, value):
return {
'data': {
'id': id,
field: value,
'updatedAt': time.time(),
}
}
def get_data():
since_time = time.strftime("%d%b%y", time.localtime(time.time() -
reporting_time_seconds))
subprocess.check_output(("""gtacl -c "\$system.system.showlog """
""" \$system.zssh.sshlog2 * \\"{} 00:01:00\\"" > /tmp/log """).
format(since_time), shell=True)
user_count = int(subprocess.check_output("grep successful /tmp/log | wc -l",
shell=True).split()[0])
log_blob = subprocess.check_output("grep 'password verification successful "
"for user' /tmp/log", shell=True)
# ^password verification successful for user 'username'$
users = map(lambda v: re.search("'([^']+)'", v).group(1),
log_blob.splitlines())
top_users = collections.Counter(users).most_common(5)
return json.dumps([
dict_for("ssh-sessions", "value", user_count),
dict_for("top-users", "items",
[{"label": u[0], "value": u[1]} for u in top_users])
])
class MyRequestHandler(SimpleHTTPServer.SimpleHTTPRequestHandler):
def do_GET(self):
if self.path == '/data':
self.send_response(200)
self.send_header('Content-Type', 'application/json')
self.end_headers()
self.wfile.write(get_data())
return
return SimpleHTTPServer.SimpleHTTPRequestHandler.do_GET(self)
Handler = MyRequestHandler
server = SocketServer.TCPServer(('0.0.0.0', 8080), Handler)
server.serve_forever()