mirror of
https://gitlab.com/Theopse/fbi-i18n-zh.git
synced 2025-04-06 03:58:02 +08:00
90 lines
2.1 KiB
Python
90 lines
2.1 KiB
Python
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
|
|
|
|
if len(sys.argv) < 3:
|
|
print("Usage: " + sys.argv[0] + " <ip> <file/directory>")
|
|
sys.exit(1)
|
|
|
|
ip = sys.argv[1]
|
|
directory = sys.argv[2]
|
|
|
|
if not os.path.exists(directory):
|
|
print(directory + ": No such file or directory.")
|
|
sys.exit(1)
|
|
|
|
print("Preparing data...")
|
|
|
|
baseUrl = [(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] + ":8080/"
|
|
payload = ""
|
|
|
|
if os.path.isfile(directory):
|
|
if directory.endswith(('.cia', '.tik')):
|
|
payload += baseUrl + quote(os.path.basename(directory))
|
|
|
|
directory = os.path.dirname(directory)
|
|
else:
|
|
for file in [ file for file in next(os.walk(directory))[2] if file.endswith(('.cia', '.tik')) ]:
|
|
payload += baseUrl + quote(file) + "\n"
|
|
|
|
if len(payload) == 0:
|
|
print("No files to serve.")
|
|
sys.exit(1)
|
|
|
|
if not directory == "":
|
|
os.chdir(directory)
|
|
|
|
print("")
|
|
print("URLS:")
|
|
print(payload)
|
|
print("")
|
|
|
|
print("Opening HTTP server on port 8080...")
|
|
|
|
server = TCPServer(("", 8080), SimpleHTTPRequestHandler)
|
|
thread = threading.Thread(target=server.serve_forever)
|
|
thread.start()
|
|
|
|
try:
|
|
print("Sending URL(s) to " + ip + ":5000...")
|
|
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sock.connect((ip, 5000))
|
|
|
|
payloadBytes = bytes(payload, "ascii")
|
|
networkPayload = struct.pack('!L', len(payloadBytes)) + payloadBytes
|
|
|
|
sentLength = 0
|
|
while sentLength < len(networkPayload):
|
|
sent = sock.send(networkPayload[sentLength:])
|
|
if sent == 0:
|
|
raise RuntimeError("Socket connection broken.")
|
|
|
|
sentLength += sent
|
|
|
|
while len(sock.recv(1)) < 1:
|
|
time.sleep(0.05)
|
|
|
|
sock.close()
|
|
except Exception as e:
|
|
print("Error: " + str(e))
|
|
|
|
print("Shutting down HTTP server...")
|
|
|
|
server.shutdown()
|