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
|
import random
import socket
import pickle
import zlib
import time
import cv2
from Utilities.db_queries import get_module_data
from Specific.encrypt import Encryption
from sys import exit
try:
info = get_module_data()
stream_settings = []
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind((info[1], int(info[3].split(',')[0])))
s.listen()
except:
exit(0)
def stream_info():
info = '\n'
for index, setting in enumerate(stream_settings):
info += f' - {setting[0]}[{index}]'
if len(stream_settings) > 1 and index + 1 != len(stream_settings):
info += '\n'
if len(stream_settings) == 0:
info = 'False'
return info
def set_stream_settings(index):
global stream_settings
try:
if index == '*':
for stream_setting in stream_settings:
stream_setting[1].close()
return 'Streams closed!'
else:
stream_settings[int(index)][1].close()
return 'Stream closed!'
except:
return 'Stream failed to close.'
def Stream(encoding, path, user):
try:
global stream_settings
FRAMES = 5.0
QUALITY = (1920, 1080)
headersize = 10
new_msg = True
msg_len = 0
full_msg = b''
msg = b'next'
r = random.randint(0, 1000000)
e = Encryption()
client, addr = s.accept()
u = (user, client, addr)
stream_settings.append(u)
real_msg = pickle.dumps(msg)
real_msg = zlib.compress(real_msg, 1)
real_msg = e.do_encrypt(real_msg)
final_msg = bytes(f'{len(msg):<{headersize}}', encoding) + msg
client.send(final_msg)
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter(f'{path}/{time.strftime("%Y-%m-%d (%H-%M-%S")}.avi', fourcc, FRAMES, QUALITY)
while True:
try:
client_msg = client.recv(81920)
if new_msg:
msg_len = int(client_msg[:headersize])
new_msg = False
full_msg += client_msg
if len(full_msg)-headersize == msg_len:
frame = e.do_decrypt(full_msg[headersize:])
frame = zlib.decompress(frame)
frame = pickle.loads(frame)
cv2.namedWindow(f'{user}\'s live stream. {r}', cv2.WINDOW_NORMAL)
cv2.imshow(f'{user}\'s live stream. {r}', frame)
out.write(frame)
if cv2.waitKey(1) == 27:
client.close()
break
real_msg = pickle.dumps(msg)
real_msg = zlib.compress(real_msg, 1)
real_msg = e.do_encrypt(real_msg)
final_msg = bytes(f'{len(real_msg):<{headersize}}', encoding) + real_msg
client.send(final_msg)
new_msg = True
msg_len = 0
full_msg = b''
except:
stream_settings.remove(u)
out.release()
exit(0)
stream_settings.remove(u)
out.release()
except:
exit(0)
|
