Compare commits
2 Commits
7e3820fda0
...
45156ab4ae
Author | SHA1 | Date | |
---|---|---|---|
45156ab4ae | |||
231a9282ea |
39
main.py
39
main.py
@@ -44,16 +44,26 @@ class WebServer:
|
|||||||
async def handle(self, reader: asyncio.StreamReader, writer: asyncio.StreamWriter) -> None:
|
async def handle(self, reader: asyncio.StreamReader, writer: asyncio.StreamWriter) -> None:
|
||||||
addr = writer.get_extra_info("peername")
|
addr = writer.get_extra_info("peername")
|
||||||
await self.log(self.conn_msg, addr)
|
await self.log(self.conn_msg, addr)
|
||||||
data = await reader.read(self._read_buffer)
|
rdata = await reader.read(self._read_buffer)
|
||||||
data = unquote(data.decode())
|
data = rdata.decode()
|
||||||
if not data: return
|
if not data: return
|
||||||
|
|
||||||
real_addr = "".join([l.replace("X-Real-IP: ", "") for l in data.split("\n") if l.startswith("X-Real-IP: ")][0]).strip() # ТАКОЙ ГАВНАКОД XD
|
real_addr = None
|
||||||
|
for line in data.split("\n"):
|
||||||
|
if line.startswith("X-Real-IP: "):
|
||||||
|
real_addr = line[len("X-Real-IP: "):].strip()
|
||||||
|
break
|
||||||
|
|
||||||
if real_addr and self.proxied:
|
if real_addr and self.proxied:
|
||||||
addr = (real_addr, addr[1])
|
addr = (real_addr, addr[1])
|
||||||
|
|
||||||
request = data.split("\n")[0]
|
request = data.split("\n")[0]
|
||||||
file_name = request.split()[1][1:]
|
parts = request.split()
|
||||||
file_path = os.path.join(self.directory, file_name)
|
if len(parts) < 2: return
|
||||||
|
|
||||||
|
path = unquote(parts[1])
|
||||||
|
file_name = path[1:] if path.startswith('/') else path
|
||||||
|
file_path = os.path.abspath(os.path.join(self.directory, file_name))
|
||||||
|
|
||||||
if os.path.isfile(file_path):
|
if os.path.isfile(file_path):
|
||||||
mime, _ = mimetypes.guess_type(file_path)
|
mime, _ = mimetypes.guess_type(file_path)
|
||||||
@@ -90,22 +100,25 @@ class WebServer:
|
|||||||
resp = await f.read()
|
resp = await f.read()
|
||||||
|
|
||||||
files = ""
|
files = ""
|
||||||
base_path = file_name.strip('/')
|
norm_path = file_path.replace('\\', '/')
|
||||||
|
base_path = norm_path.replace(self.directory.replace('\\', '/'), '').strip('/')
|
||||||
|
|
||||||
for item in sorted(os.listdir(file_path)):
|
for item in sorted(os.listdir(file_path)):
|
||||||
item_path = os.path.join(file_path, item)
|
item_path = os.path.join(file_path, item)
|
||||||
is_dir = os.path.isdir(item_path)
|
is_dir = os.path.isdir(item_path)
|
||||||
|
|
||||||
if base_path: rel_path = f"{base_path}/{item}"
|
rel_path = f"{base_path}/{item}" if base_path else item
|
||||||
else: rel_path = item
|
rel_path = rel_path.replace('\\', '/')
|
||||||
|
|
||||||
if is_dir: rel_path += "/"
|
if is_dir:
|
||||||
|
rel_path += "/"
|
||||||
|
item += "/"
|
||||||
|
|
||||||
modify_time = os.path.getmtime(item_path)
|
modify_time = os.path.getmtime(item_path)
|
||||||
modify_datetime = datetime.datetime.fromtimestamp(modify_time)
|
modify_datetime = datetime.datetime.fromtimestamp(modify_time)
|
||||||
formatted_time = modify_datetime.strftime("%d.%m.%Y %H:%M:%S")
|
formatted_time = modify_datetime.strftime("%d.%m.%Y %H:%M:%S")
|
||||||
|
|
||||||
files += f'<a href="/{rel_path}">{item}</a> | {formatted_time}<br>' # TODO: добавить кастом для инфы про файлики
|
files += f'<a class={"dir" if is_dir else "file"} href="/{rel_path}">{item}</a> | {formatted_time}<br>\n'
|
||||||
|
|
||||||
resp = resp.replace("<FILES>", files)
|
resp = resp.replace("<FILES>", files)
|
||||||
resp = resp.encode()
|
resp = resp.encode()
|
||||||
@@ -117,15 +130,17 @@ class WebServer:
|
|||||||
"\r\n"
|
"\r\n"
|
||||||
)
|
)
|
||||||
|
|
||||||
writer.write(headers + resp)
|
writer.write(headers.encode() + resp)
|
||||||
else:
|
else:
|
||||||
await self.log(self.e404_msg, addr, file_path)
|
await self.log(self.e404_msg, addr, file_path)
|
||||||
|
async with aiofiles.open(self._e404_file, "r", encoding="utf-8") as f:
|
||||||
|
content = await f.read()
|
||||||
response = (
|
response = (
|
||||||
"HTTP/1.1 404 Not Found\r\n"
|
"HTTP/1.1 404 Not Found\r\n"
|
||||||
"Content-Type: text/html; charset=utf-8\r\n"
|
"Content-Type: text/html; charset=utf-8\r\n"
|
||||||
f"Server: {self.name}\r\n"
|
f"Server: {self.name}\r\n"
|
||||||
"\r\n"
|
"\r\n"
|
||||||
f"{open(self._e404_file, encoding="utf-8").read()}"
|
f"{content}"
|
||||||
)
|
)
|
||||||
writer.write(response.encode())
|
writer.write(response.encode())
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
|
Reference in New Issue
Block a user