forked from astravaganza/wvproxy
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathwvproxy.py
More file actions
executable file
·148 lines (118 loc) · 3.73 KB
/
Copy pathwvproxy.py
File metadata and controls
executable file
·148 lines (118 loc) · 3.73 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
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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
#!/usr/bin/env python3
import base64
import sys
import time
import uuid
from collections import defaultdict
from flask import Flask, render_template, request
from flask_socketio import SocketIO
from pymp4.parser import Box
app = Flask(__name__)
socketio = SocketIO(app, async_mode='threading')
sessions = defaultdict(dict)
def log(message, *args, **kwargs):
kwargs.setdefault('file', sys.stderr)
kwargs.setdefault('flush', True)
print(message, *args, **kwargs)
@app.route('/', methods=['GET'])
def index():
return render_template('index.html')
@app.route('/wvproxy.js', methods=['GET'])
def wvproxy_js():
return render_template('wvproxy.js')
@app.route('/api', methods=['POST'])
def api():
req = request.get_json()
method = req['method']
authorized = False
with open('authorized_users.txt') as fd:
for line in fd.read().splitlines():
name, token = line.split()
if token == req['token']:
log(f'[{name}] {req}')
authorized = True
break
if not authorized:
log(f'[anonymous] {req}')
return {
'status_code': 403,
'message': 'Invalid token',
}
del req['token']
if method == 'GetChallenge':
session_id = str(uuid.uuid4())
req['params']['session_id'] = session_id
try:
Box.parse(base64.b64decode(req['params']['init']))
except OSError:
req['params']['init'] = base64.b64encode(Box.build(dict(
type=b'pssh',
version=0,
flags=0,
system_ID=uuid.UUID('edef8ba9-79d6-4ace-a3c8-27dcd51d21ed'),
init_data=base64.b64decode(req['params']['init']),
))).decode()
socketio.emit('GetChallenge', req)
start = time.time()
log('Waiting for challenge', end='')
while not sessions[session_id]:
if time.time() - start > 15:
return {
'status_code': 504,
'message': 'Request timed out',
}
log('.', end='')
time.sleep(0.5)
return {
'status_code': 200,
'message': {
'session_id': session_id,
'challenge': base64.b64encode(sessions[session_id]['challenge']).decode(),
}
}
elif method == 'GetKeys':
session_id = req['params']['session_id']
if session_id not in sessions:
return {
'status_code': 400,
'message': 'Invalid session ID',
}
socketio.emit('GetKeys', req)
start = time.time()
log('Waiting for keys', end='')
while 'keys' not in sessions[session_id]:
if time.time() - start > 15:
return {
'status_code': 504,
'message': 'Request timed out',
}
log('.', end='')
time.sleep(0.5)
return {
'status_code': 200,
'message': {
'keys': sessions[session_id]['keys'],
},
}
elif method == 'GetKeysX':
return {
'status_code': 400,
'message': 'Not implemented',
}
else:
return {
'status_code': 400,
'message': 'Unknown method',
}
@socketio.on('SetChallenge')
def on_set_challenge(res):
log('\nGot challenge')
log(base64.b64encode(res['challenge']))
sessions[res['session_id']]['challenge'] = res['challenge']
@socketio.on('SetKeys')
def on_set_keys(res):
log('\nGot keys')
log(res['keys'])
sessions[res['session_id']]['keys'] = res['keys']
if __name__ == '__main__':
socketio.run(app)