blackteay/fbi-i18n-zh
1
0
Fork 0
mirror of https://gitlab.com/Theopse/fbi-i18n-zh.git synced 2025-09-02 01:39:47 +08:00
Code Issues Projects Releases Wiki Activity

Add .cetk to extensions supported by servefiles, minor string changes.

Browse Source
This commit is contained in:
Steveice10 2017-02-16 09:57:37 -08:00
parent 1b5cc0a268
commit 6b5e4e1a54
2 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
servefiles/README.md
View File

@ -4,5 +4,4 @@ Simple Python script for serving local files to FBI's remote installer. Requires
**Usage**: python servefiles.py (3ds ip) (file / directory) \[host ip\] \[host port\] **Usage**: python servefiles.py (3ds ip) (file / directory) \[host ip\] \[host port\]
- the file can be *.cia or *.tik - Supported file extensions: .cia, .tik, .cetk
- the directory have to contain *.cia or *.tik files

20
servefiles/servefiles.py
View File

@ -24,13 +24,13 @@ if len(sys.argv) < 3 or len(sys.argv) > 5:
print('Usage:', sys.argv[0], '<target ip> <file / directory> [host ip] [host port]') print('Usage:', sys.argv[0], '<target ip> <file / directory> [host ip] [host port]')
sys.exit(1) sys.exit(1)
accepted_extension = ('.cia', '.tik') accepted_extension = ('.cia', '.tik', '.cetk')
target_ip = sys.argv[1] target_ip = sys.argv[1]
target_path = sys.argv[2] target_path = sys.argv[2]
hostPort = 8080 # default value hostPort = 8080 # Default value
if not os.path.exists(target_path): if not os.path.exists(target_path):
print(target_path, ': No such file or directory.', file=sys.stderr) print(target_path + ': No such file or directory.', file=sys.stderr)
sys.exit(1) sys.exit(1)
if len(sys.argv) >= 4: if len(sys.argv) >= 4:
@ -38,10 +38,10 @@ if len(sys.argv) >= 4:
if len(sys.argv) == 5: if len(sys.argv) == 5:
hostPort = int(sys.argv[4]) hostPort = int(sys.argv[4])
else: else:
print('Finding host ip ...') # loop over network address to find one 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] 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]
print('Preparing data ...') print('Preparing data...')
baseUrl = hostIp + ':' + str(hostPort) + '/' baseUrl = hostIp + ':' + str(hostPort) + '/'
if os.path.isfile(target_path): if os.path.isfile(target_path):
@ -49,7 +49,7 @@ if os.path.isfile(target_path):
file_list_payload = baseUrl + quote(os.path.basename(target_path)) file_list_payload = baseUrl + quote(os.path.basename(target_path))
directory = os.path.dirname(target_path) # get file directory directory = os.path.dirname(target_path) # get file directory
else: else:
print('This file has no extension like :', accepted_extension, file=sys.stderr) print('Unsupported file extension. Supported extensions are:', accepted_extension, file=sys.stderr)
sys.exit(1) sys.exit(1)
else: else:
@ -67,16 +67,16 @@ 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 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 os.chdir(directory) # set working directory to the right folder to be able to serve files
print('\nURLS (data) :') print('\nURLs:')
print(file_list_payload, '\n') print(file_list_payload, '\n')
print('Opening HTTP server on port :', hostPort) print('Opening HTTP server on port', hostPort)
server = TCPServer(('', hostPort), SimpleHTTPRequestHandler) server = TCPServer(('', hostPort), SimpleHTTPRequestHandler)
thread = threading.Thread(target=server.serve_forever) thread = threading.Thread(target=server.serve_forever)
thread.start() thread.start()
try: try:
print('Sending URL(s) to', target_ip, 'on port 5000 ...') print('Sending URL(s) to', target_ip, 'on port 5000...')
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect((target_ip, 5000)) sock.connect((target_ip, 5000))
sock.sendall(struct.pack('!L', len(file_list_payloadBytes)) + file_list_payloadBytes) sock.sendall(struct.pack('!L', len(file_list_payloadBytes)) + file_list_payloadBytes)
@ -84,7 +84,7 @@ try:
time.sleep(0.05) time.sleep(0.05)
sock.close() sock.close()
except Exception as e: except Exception as e:
print('An Error occurred :', str(e), file=sys.stderr) print('An error occurred:', str(e), file=sys.stderr)
server.shutdown() server.shutdown()
sys.exit(1) sys.exit(1)