You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.3 KiB
58 lines
1.3 KiB
12 years ago
|
import zlib
|
||
|
|
import struct
|
||
12 years ago
|
import socket
|
||
|
|
import ssl
|
||
|
|
|
||
|
|
from synchronousdeluge import rencode
|
||
|
|
|
||
|
12 years ago
|
|
||
|
|
__all__ = ["DelugeTransfer"]
|
||
|
|
|
||
|
|
class DelugeTransfer(object):
|
||
|
|
def __init__(self):
|
||
|
|
self.sock = None
|
||
|
|
self.conn = None
|
||
|
|
self.connected = False
|
||
|
|
|
||
|
|
def connect(self, hostport):
|
||
|
|
if self.connected:
|
||
|
|
self.disconnect()
|
||
|
|
|
||
|
|
self.sock = socket.create_connection(hostport)
|
||
|
|
self.conn = ssl.wrap_socket(self.sock)
|
||
|
|
self.connected = True
|
||
|
|
|
||
|
|
def disconnect(self):
|
||
|
|
if self.conn:
|
||
|
|
self.conn.close()
|
||
|
|
self.connected = False
|
||
|
|
|
||
|
|
def send_request(self, request):
|
||
|
|
data = (request.format(),)
|
||
|
|
payload = zlib.compress(rencode.dumps(data))
|
||
|
|
self.conn.sendall(payload)
|
||
|
|
|
||
|
|
buf = b""
|
||
|
|
|
||
|
|
while True:
|
||
|
|
data = self.conn.recv(1024)
|
||
|
|
|
||
|
|
if not data:
|
||
|
|
self.connected = False
|
||
|
|
break
|
||
|
|
|
||
|
|
buf += data
|
||
|
|
dobj = zlib.decompressobj()
|
||
|
|
|
||
|
|
try:
|
||
|
|
message = rencode.loads(dobj.decompress(buf))
|
||
|
|
except (ValueError, zlib.error, struct.error):
|
||
|
|
# Probably incomplete data, read more
|
||
|
|
continue
|
||
|
|
else:
|
||
|
|
buf = dobj.unused_data
|
||
|
|
|
||
|
|
yield message
|
||
|
|
|
||
|
|
|