Page MenuHomePhabricator
Files F7159266

test_file_sender_receiver2.py
No OneTemporary
Actions

test_file_sender_receiver2.py
View Options

from base64 import b64encode
from twisted.internet.protocol import ClientFactory, Protocol
from twisted.protocols.basic import LineOnlyReceiver
from twisted.internet import reactor
from gnutls.constants import *
from gnutls.crypto import *
from gnutls.errors import *
from gnutls.interfaces.twisted import X509Credentials
from msrp.protocol import *
from msrp.digest import process_www_authenticate
from msrp.responses import *
rand_source = open("/dev/urandom")
zero_source = open("/dev/zero")
FILE_BLOCKS = 4
BLOCK_SIZE = 1024 * 1024
def generate_transaction_id():
return b64encode(rand_source.read(12), "+-")
class MSRPTestClientFactory(ClientFactory):
protocol = MSRPProtocol
def __init__(self, host, port):
self.uri = URI(host, use_tls = True, port = port, session_id = b64encode(rand_source.read(12)))
self.relay_uri = URI("ws1.dns-hosting.info", use_tls = True, port = 10000)
self.do_on_start = None
self.do_on_data = None
self.byte_count = 0
def get_peer(self, protocol):
self.protocol = protocol
reactor.callLater(0, self._send_auth1)
return self
def data_start(self, msrpdata):
print "RECEIVED %s" % str(msrpdata)
print "HEADERS:\n%s" % "\n".join("%s: %s" % (hname, hval.encoded) for hname, hval in msrpdata.headers.iteritems())
if self.do_on_start:
self.do_on_start(msrpdata)
def write_chunk(self, data):
print "GOT DATA"
if self.do_on_data:
self.do_on_data(data)
def data_end(self, msrpdata):
print "GOT END"
def connection_lost(self, reason):
pass
def _send_auth1(self):
msrpdata = MSRPData(generate_transaction_id(), method = "AUTH")
msrpdata.add_header(ToPathHeader([self.relay_uri]))
msrpdata.add_header(FromPathHeader([self.uri]))
print "SENDING:\n\n%s" % msrpdata.encode()
self.protocol.transport.write(msrpdata.encode())
self.do_on_start = self._send_auth2
def _send_auth2(self, msrpdata):
auth, rsp_auth = process_www_authenticate("user2", "pass2", "AUTH", str(self.relay_uri), **msrpdata.headers["WWW-Authenticate"].decoded)
msrpdata = MSRPData(generate_transaction_id(), method = "AUTH")
msrpdata.add_header(ToPathHeader([self.relay_uri]))
msrpdata.add_header(FromPathHeader([self.uri]))
msrpdata.add_header(AuthorizationHeader(auth))
print "SENDING:\n\n%s" % msrpdata.encode()
self.protocol.transport.write(msrpdata.encode())
self.do_on_start = self._get_path
def _get_path(self, msrpdata):
sdp_path = list(reversed(msrpdata.headers["Use-Path"].decoded)) + [self.uri]
print "Path to send in SDP: %s" % " ".join(str(uri) for uri in sdp_path)
self.full_to_path = " ".join(str(uri) for uri in msrpdata.headers["Use-Path"].decoded) + " " + raw_input("Destination path: ")
reactor.callLater(5, self._send_file)
self.do_on_start = None
self.do_on_data = self._receive_data
self.bytes_sent = 0
def _send_file(self):
msrpdata = MSRPData(generate_transaction_id(), method = "SEND")
msrpdata.add_header(FromPathHeader([self.uri]))
msrpdata.add_header(MSRPHeader("To-Path", self.full_to_path))
msrpdata.add_header(MessageIDHeader("1"))
msrpdata.add_header(ByteRangeHeader([1, "*", "*"]))
msrpdata.add_header(ContentTypeHeader("binary/octet-stream"))
msrpdata.add_header(FailureReportHeader("yes"))
self.protocol.transport.write(msrpdata.encode_start())
for i in range(FILE_BLOCKS):
data = zero_source.read(BLOCK_SIZE)
self.protocol.transport.write(data)
self.protocol.transport.write(msrpdata.encode_end("$"))
def _receive_data(self, data):
self.byte_count += len(data)
print "got %d bytes, total %d" % (len(data), self.byte_count)
def clientConnectionFailed(self, connector, err):
print err.value
def clientConnectionLost(self, connector, err):
print err.value
ca = X509Certificate(open("../certs/ca-cert.pem").read())
cred = X509Credentials(None, None, [ca])
cred.verify_peer = True
factory = MSRPTestClientFactory("node03.dns-hosting.info", 12345)
reactor.connectTLS("ws1.dns-hosting.info", 10000, factory, cred, server_name = "dns-hosting.info")
reactor.run()

File Metadata

Mime Type
text/x-python
Expires
Sat, Nov 23, 3:57 AM (20 h, 55 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
3408738
Default Alt Text
test_file_sender_receiver2.py (4 KB)

Event Timeline