def main():
if not sys.argv[1:] or sys.argv[1] not in ["client", "server"] or sys.argv[3:]:
sys.exit(__doc__)
role = sys.argv[1]
if sys.argv[2:]:
filename = sys.argv[2]
else:
filename = None
if role == "client":
local_uri = URI(session_id="client", use_tls=False)
remote_uri = URI(session_id="server", use_tls=False)
connector = ConnectorDirect(logger=Logger(is_enabled_func=lambda: False))
dest = None
else:
local_uri = URI(session_id="server", use_tls=False)
remote_uri = URI(session_id="client", use_tls=False)
connector = AcceptorDirect(logger=Logger(is_enabled_func=lambda: False))
if filename:
if os.path.exists(filename) and os.path.isfile(filename):
sys.exit("%s already exists. Remove it first or provide another destination" % filename)
dest = file(filename, "w+")
connector.prepare(local_uri)
transport = connector.complete([remote_uri])
main_greenlet = api.getcurrent()
def on_received(chunk=None, error=None):
if chunk is not None:
if chunk.content_type == "text/plain":
print "\nreceived message: %s" % chunk.data
elif chunk.content_type == "application/octet-stream":
fro, to, total = chunk.byte_range
dest.seek(fro - 1)
dest.write(chunk.data)
if total:
percent = " (%d%%)" % (100.0 * (fro - 1 + len(chunk.data)) / total)
else:
percent = ""
speed = (fro - 1 + len(chunk.data)) / float((time() - start_time)) / 1024.0 / 1024
print "\rwrote %s bytes to %s. %s bytes total%s %.2f MB/s " % (
len(chunk.data),
filename,
fro - 1 + len(chunk.data),
percent,
speed,
),
if chunk.contflag == "$":
api.spawn(api.kill, main_greenlet, proc.ProcExit)
if error is not None:
api.spawn(api.kill, main_greenlet, error.type, error.value, error.tb)
session = MSRPSession(transport, on_incoming_cb=on_received)
def sender():
while True:
session.send_message("hello from %s" % role, "text/plain")
api.sleep(3)
sender = proc.spawn_link_exception(proc.wrap_errors(proc.ProcExit, sender))
if filename and role == "client":
if os.path.isfile(filename):
size = os.stat(filename).st_size
else:
size = None
f = file(filename)
outgoing_file = OutgoingFile(f, size, "application/octet-stream")
print "Sending %s %s bytes" % (filename, size)
session.send_file(outgoing_file)
start_time = time()
try:
api.get_hub().switch() # sleep forever
except (proc.ProcExit, ConnectionDone):
pass
finally:
sender.kill()
session.shutdown()
def setup_two_endpoints(self):
server_path = TimeoutEvent()
client_path = TimeoutEvent()
client = proc.spawn_link_exception(_connect_msrp, client_path, server_path, self.get_connector(), self.get_client_uri())
server = proc.spawn_link_exception(_connect_msrp, server_path, client_path, self.get_acceptor(), self.get_server_uri())
return client, server