refactor core modules, integrate network stats

This commit is contained in:
Username
2025-12-25 11:13:20 +01:00
parent 2201515b10
commit 269fed55ff
8 changed files with 270 additions and 219 deletions

View File

@@ -19,6 +19,7 @@
# https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
import socket, ssl, select, copy, errno
import network_stats
# rs_proxyType
RS_PT_NONE = 0
@@ -310,6 +311,7 @@ class Rocksock():
if self.sock is None:
raise(RocksockException(RS_E_NO_SOCKET, failedproxy=self._failed_proxy(pnum)))
try:
network_stats.add_tx(len(buf))
return self.sock.sendall(buf)
except socket.error as e:
raise(self._translate_socket_error(e, pnum))
@@ -340,6 +342,7 @@ class Rocksock():
raise(RocksockException(RS_E_SSL_GENERIC, failedproxy=s, errortype=RS_ET_SSL))
if len(chunk) == 0:
raise(RocksockException(RS_E_REMOTE_DISCONNECTED, failedproxy=self._failed_proxy(pnum)))
network_stats.add_rx(len(chunk))
data += chunk
if count == -1: break
else: count -= len(chunk)