mirror of
https://gitlab.com/Theopse/fbi-i18n-zh.git
synced 2025-04-06 03:58:02 +08:00
023fec6f46
This adds an interactive mode if no arguments are provided, prompting the user to enter the IP, file, etc., The 'help' display is instead shown if the standard 'help' flags are used (ex: --h, -h, --help, etc.,)t This also allows the user to specify the path to the interpreter. The original version assumed only the script name was used as an argument, and the default interpreter for the environment was being used, so it would fail on "C:\python27\python.exe servefiles.py"
136 lines
4.7 KiB
Python
136 lines
4.7 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 urlparse import urljoin
|
|
from urllib import pathname2url, quote
|
|
except ImportError:
|
|
from http.server import SimpleHTTPRequestHandler
|
|
from socketserver import TCPServer
|
|
from urllib.parse import urljoin, quote
|
|
from urllib.request import pathname2url
|
|
|
|
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')
|
|
hostPort = 8080 # Default value
|
|
|
|
if interactive:
|
|
target_ip = raw_input("The IP of your 3DS: ")
|
|
target_path = raw_input("The file you want to send (.cia, .tik, or .cetk): ")
|
|
|
|
hostIp = raw_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 = raw_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]
|
|
|
|
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')
|
|
|
|
print('Opening HTTP server on port ' + str(hostPort))
|
|
server = TCPServer(('', 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()
|