mirror of
https://gitlab.com/Theopse/fbi-i18n-zh.git
synced 2025-04-05 09:52:44 +08:00
c427e08f95
e.g. prevents issue with whitespace being added when a user drags a file to the terminal on mac
142 lines
4.8 KiB
Python
142 lines
4.8 KiB
Python
#!/usr/bin/env python
|
|
# coding: utf-8 -*-
|
|
|
|
import os
|
|
import socket
|
|
import struct
|
|
import sys
|
|
import threading
|
|
import time
|
|
import urllib
|
|
|
|
try:
|
|
from SimpleHTTPServer import SimpleHTTPRequestHandler
|
|
from SocketServer import TCPServer
|
|
from urllib import quote
|
|
input = raw_input
|
|
except ImportError:
|
|
from http.server import SimpleHTTPRequestHandler
|
|
from socketserver import TCPServer
|
|
from urllib.parse import quote
|
|
|
|
interactive = False
|
|
|
|
if len(sys.argv) <= 2:
|
|
# If there aren't enough variables, use interactive mode
|
|
if len(sys.argv) == 2:
|
|
if sys.argv[1].lower() in ('--help', '-help', 'help', 'h', '-h', '--h'):
|
|
print('Usage: ' + sys.argv[0] + ' <target ip> <file / directory> [host ip] [host port]')
|
|
sys.exit(1)
|
|
|
|
interactive = True
|
|
|
|
elif len(sys.argv) < 3 or len(sys.argv) > 6:
|
|
print('Usage: ' + sys.argv[0] + ' <target ip> <file / directory> [host ip] [host port]')
|
|
sys.exit(1)
|
|
|
|
accepted_extension = ('.cia', '.tik', '.cetk', '.3dsx')
|
|
hostPort = 8080 # Default value
|
|
|
|
if interactive:
|
|
target_ip = input("The IP of your 3DS: ")
|
|
target_path = input("The file you want to send (.cia, .tik, .cetk, or .3dsx): ")
|
|
|
|
hostIp = input("Host IP (or press Enter to have the script detect host IP):")
|
|
if hostIp == '':
|
|
print('Detecting host IP...')
|
|
hostIp = [(s.connect(('8.8.8.8', 53)), s.getsockname()[0], s.close()) for s in [socket.socket(socket.AF_INET, socket.SOCK_DGRAM)]][0][1]
|
|
else:
|
|
hostPort = input("Host port (or press Enter to keep default, 8080):")
|
|
if hostPort == '':
|
|
hostPort = 8080 # Default
|
|
|
|
|
|
else:
|
|
# (if the script is being run using a full python path; ex: "path/to/python script_name.py foo foo..")
|
|
if sys.argv[1] == os.path.basename(__file__):
|
|
target_ip = sys.argv[2]
|
|
target_path = sys.argv[3]
|
|
|
|
if len(sys.argv) >= 5:
|
|
hostIp = sys.argv[4]
|
|
if len(sys.argv) == 6:
|
|
hostPort = int(sys.argv[5])
|
|
else:
|
|
print('Detecting host IP...')
|
|
hostIp = [(s.connect(('8.8.8.8', 53)), s.getsockname()[0], s.close()) for s in [socket.socket(socket.AF_INET, socket.SOCK_DGRAM)]][0][1]
|
|
|
|
# (if the script is being run using just the script name and default executable for python scripts; ex: "script_name.py foo foo..")
|
|
else:
|
|
target_ip = sys.argv[1]
|
|
target_path = sys.argv[2]
|
|
|
|
if len(sys.argv) >= 4:
|
|
hostIp = sys.argv[3]
|
|
if len(sys.argv) == 5:
|
|
hostPort = int(sys.argv[4])
|
|
else:
|
|
print('Detecting host IP...')
|
|
hostIp = [(s.connect(('8.8.8.8', 53)), s.getsockname()[0], s.close()) for s in [socket.socket(socket.AF_INET, socket.SOCK_DGRAM)]][0][1]
|
|
|
|
target_path = target_path.strip()
|
|
if not os.path.exists(target_path):
|
|
print(target_path + ': No such file or directory.')
|
|
sys.exit(1)
|
|
|
|
|
|
print('Preparing data...')
|
|
baseUrl = hostIp + ':' + str(hostPort) + '/'
|
|
|
|
if os.path.isfile(target_path):
|
|
if target_path.endswith(accepted_extension):
|
|
file_list_payload = baseUrl + quote(os.path.basename(target_path))
|
|
directory = os.path.dirname(target_path) # get file directory
|
|
else:
|
|
print('Unsupported file extension. Supported extensions are: ' + accepted_extension)
|
|
sys.exit(1)
|
|
|
|
else:
|
|
directory = target_path # it's a directory
|
|
file_list_payload = '' # init the payload before adding lines
|
|
for file in [file for file in next(os.walk(target_path))[2] if file.endswith(accepted_extension)]:
|
|
file_list_payload += baseUrl + quote(file) + '\n'
|
|
|
|
if len(file_list_payload) == 0:
|
|
print('No files to serve.')
|
|
sys.exit(1)
|
|
|
|
file_list_payloadBytes = file_list_payload.encode('ascii')
|
|
|
|
if directory and directory != '.': # doesn't need to move if it's already the current working directory
|
|
os.chdir(directory) # set working directory to the right folder to be able to serve files
|
|
|
|
print('\nURLs:')
|
|
print(file_list_payload + '\n')
|
|
|
|
class MyServer(TCPServer):
|
|
def server_bind(self):
|
|
import socket
|
|
self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
self.socket.bind(self.server_address)
|
|
|
|
print('Opening HTTP server on port ' + str(hostPort))
|
|
server = MyServer(('', hostPort), SimpleHTTPRequestHandler)
|
|
thread = threading.Thread(target=server.serve_forever)
|
|
thread.start()
|
|
|
|
try:
|
|
print('Sending URL(s) to ' + target_ip + ' on port 5000...')
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sock.connect((target_ip, 5000))
|
|
sock.sendall(struct.pack('!L', len(file_list_payloadBytes)) + file_list_payloadBytes)
|
|
while len(sock.recv(1)) < 1:
|
|
time.sleep(0.05)
|
|
sock.close()
|
|
except Exception as e:
|
|
print('An error occurred: ' + str(e))
|
|
server.shutdown()
|
|
sys.exit(1)
|
|
|
|
print('Shutting down HTTP server...')
|
|
server.shutdown()
|