106 lines
2.6 KiB
Python
106 lines
2.6 KiB
Python
import os, logging, gevent, time, msgpack
|
|
import zmq.green as zmq
|
|
from cStringIO import StringIO
|
|
from Config import config
|
|
|
|
context = zmq.Context()
|
|
|
|
# Communicate remote peers
|
|
class Peer:
|
|
def __init__(self, ip, port, site):
|
|
self.ip = ip
|
|
self.port = port
|
|
self.site = site
|
|
self.key = "%s:%s" % (ip, port)
|
|
|
|
self.socket = None
|
|
self.last_found = None
|
|
self.added = time.time()
|
|
|
|
self.connection_error = 0
|
|
self.hash_failed = 0
|
|
self.download_bytes = 0
|
|
self.download_time = 0
|
|
|
|
|
|
# Connect to host
|
|
def connect(self):
|
|
self.log = logging.getLogger("Peer:%s:%s" % (self.ip, self.port))
|
|
self.socket = context.socket(zmq.REQ)
|
|
self.socket.setsockopt(zmq.SNDTIMEO, 5000) # Wait for data send
|
|
self.socket.setsockopt(zmq.LINGER, 500) # Wait for socket close
|
|
self.socket.connect('tcp://%s:%s' % (self.ip, self.port))
|
|
|
|
|
|
# Found a peer on tracker
|
|
def found(self):
|
|
self.last_found = time.time()
|
|
|
|
|
|
# Send a command to peer
|
|
def sendCmd(self, cmd, params = {}):
|
|
if not self.socket: self.connect()
|
|
try:
|
|
self.socket.send(msgpack.packb({"cmd": cmd, "params": params}, use_bin_type=True))
|
|
response = msgpack.unpackb(self.socket.recv())
|
|
if "error" in response:
|
|
self.log.debug("%s %s error: %s" % (cmd, params, response["error"]))
|
|
self.onConnectionError()
|
|
else: # Successful request, reset connection error num
|
|
self.connection_error = 0
|
|
return response
|
|
except Exception, err:
|
|
self.onConnectionError()
|
|
self.log.error("%s" % err)
|
|
self.socket.close()
|
|
time.sleep(1)
|
|
self.connect()
|
|
return None
|
|
|
|
|
|
# Get a file content from peer
|
|
def getFile(self, site, inner_path):
|
|
location = 0
|
|
buff = StringIO()
|
|
s = time.time()
|
|
while 1: # Read in 512k parts
|
|
back = self.sendCmd("getFile", {"site": site, "inner_path": inner_path, "location": location}) # Get file content from last location
|
|
if not back or "body" not in back: # Error
|
|
return False
|
|
|
|
buff.write(back["body"])
|
|
if back["location"] == back["size"]: # End of file
|
|
break
|
|
else:
|
|
location = back["location"]
|
|
self.download_bytes += back["location"]
|
|
self.download_time += (time.time() - s)
|
|
buff.seek(0)
|
|
return buff
|
|
|
|
|
|
# Send a ping request
|
|
def ping(self):
|
|
return self.sendCmd("ping")
|
|
|
|
|
|
# Stop and remove from site
|
|
def remove(self):
|
|
self.log.debug("Removing peer...Connection error: %s, Hash failed: %s" % (self.connection_error, self.hash_failed))
|
|
del(self.site.peers[self.key])
|
|
self.socket.close()
|
|
|
|
|
|
# - EVENTS -
|
|
|
|
# On connection error
|
|
def onConnectionError(self):
|
|
self.connection_error += 1
|
|
if self.connection_error > 5: # Dead peer
|
|
self.remove()
|
|
|
|
|
|
# Done working with peer
|
|
def onWorkerDone(self):
|
|
pass
|