def send_file(filename):
"""Initiate sending of a file with RUDP.
Create a RUDP socket for sending. Send the file name to the VS receiver.
Register a handler for input event, which will take care of sending
file data"""
f = open(filename, "rb")
rsock = rudp.rudp_socket(0)
rudp.rudp_event_handler(rsock, eventhandler)
rudp.rudp_recvfrom_handler(rsock, recvfrom_handler)
# Strip of any leading path name and create start packet
vsm = vsftp.vsftp()
vsm.vs_type = vsftp.VS_TYPE_BEGIN
vsm.vs_filename = os.path.basename(filename)
buf = vsm.pack()
for p in peers:
(address, port, flow, scope) = p
if debug:
print "vs_send: send BEGIN \"%s\" (%d bytes) to %s:%s" % (filename, len(buf), address, port)
rudp.rudp_sendto(rsock, buf, p)
# Launch event driven filesender to send file content
event.event_fd(f, filesender, rsock, "filesender")
def filesender(file, rsock):
"""
Callback function for handling sending of the file.
Will be called when data is available on the file (which is always
true, until the file is closed...).
Send file data. Detect end of file and tell VS peers that transfer is
complete"""
str = file.read(vsftp.VS_MAXDATA)
if len(str) == 0: # end of file
vsm = vsftp.vsftp()
vsm.vs_type = vsftp.VS_TYPE_END
buf = vsm.pack()
for p in peers:
if debug:
(address, port, flow, scope) = p
print "vs_send: send END (%d bytes) to %s:%s" % (len(buf), address, port)
rudp.rudp_sendto(rsock, buf, p)
event.event_fd_delete(filesender, rsock)
file.close()
rudp.rudp_close(rsock)
else:
vsm = vsftp.vsftp()
vsm.vs_type = vsftp.VS_TYPE_DATA
vsm.vs_data = str
buf = vsm.pack()
for p in peers:
if debug:
(address, port, flow, scope) = p
print "vs_send: send DATA (%d bytes) to %s:%s" % (len(buf), address, port)
rudp.rudp_sendto(rsock, buf, p)
return 0
