Files
cpp3ds/scripts/sendfile.py
T
2016-05-27 14:28:58 -04:00

45 lines
1000 B
Python

#!/usr/bin/env python
import os, socket, sys, struct, getopt
def sendfile(filename, ip):
statinfo = os.stat(filename)
with open(filename, 'rb') as f:
sock = socket.socket()
sock.connect((ip, 5000))
sock.send(struct.pack('!i', 1))
if struct.unpack("!b", sock.recv(1))[0] == 1:
sock.send(struct.pack('!q', statinfo.st_size))
while True:
chunk = f.read(1024 * 256)
if not chunk:
break # EOF
sock.sendall(chunk)
else:
print("Canceled by FBI")
sock.close()
def show_usage_exit():
print('sendfile.py -f <inputfile> -i <ip address>')
sys.exit(2)
def main(argv):
filename = None
ip = None
try:
opts, args = getopt.getopt(argv, "hf:i:")
except getopt.GetoptError:
show_usage_exit()
for opt, arg in opts:
if opt == '-h':
show_usage_exit()
elif opt in ("-f", "--file"):
filename = arg
elif opt in ("-i", "--ip"):
ip = arg
if not (filename and ip):
show_usage_exit()
sendfile(filename, ip)
if __name__ == "__main__":
main(sys.argv[1:])