o jR7@sdZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z mZddlmZmZGdddZGdd d eZ  dd dZ  dddZddZedkrdedSdS)a File Download Server Example This example demonstrates how to create a secure file download server using the HTTPServerHandler class. The server allows: - Listing files in a specified directory - Downloading individual files - Protection against directory traversal attacks Usage: python fileserver.py [directory_path] [host] [port] Example: python fileserver.py ./downloads 127.0.0.1 8080 N)Path) UniTargetUniProto)HTTPServerHandler HTTPServerc@seZdZdZ  d#ddZd$d d Zd d Zd dZddZddZ ddZ ddZ ddZ ddZ ddZddZdd Zd!d"ZdS)%MultipartStreamProcessorz Memory-efficient multipart form data processor that handles boundary detection across chunk boundaries and streams file data directly to disk. @NdcCsp||_||_||_||_||_||_d|_d|_d|_d|_ g|_ g|_ d|_ d|j|_ |jd|_||_dS)Nlooking_for_boundaryr s--)boundary_bytes target_pathmax_buffer_size max_file_sizeallowed_extensionsmax_files_per_requestbufferstatecurrent_file_infocurrent_file_handle temp_filescompleted_files file_countboundary_with_crlf end_boundaryprint_cb)selfrrrrrrrr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/asysocks/examples/fileserver.py__init__%s    z!MultipartStreamProcessor.__init__c$|jdurdS||IdHdSNrrmsgrrr print< zMultipartStreamProcessor.printcs|j|7_g}t|j|jkr4|jdkr|IdHnt|j|jdkr4tdt|jd |jdkrM||IdH|jdkrL |Sn+|jdkr]|IdHs\ |Sn|jdkrx| IdH}|rp| ||jdkrx |Sq5) z Process a chunk of data, returning any completed file info. Args: chunk (bytes): Raw chunk data Returns: list: List of completed file info dictionaries reading_file_dataNzBuffer size exceeded:  bytesTr reading_headers) rlenrr_flush_file_data ValueErrorextend_process_boundary_search_process_headers_process_file_dataappend)rchunkr file_inforrr process_chunkAs8          z&MultipartStreamProcessor.process_chunkcsg}|j|j}|dkr,t|jt|jdkr*t|jd}|j| d|_|S|j||t|j|jkrP|IdH|jrN||jd|_|S|t|j}|j|d|_|jdrn|jdd|_n|jdr||jdd|_d|_ |S) z*Search for boundary markers in the buffer.r+ Nr  r-) rfindrr.r_finalize_current_filerr5 startswithr)rr boundary_pos keep_size boundary_endrrr r2is,   z1MultipartStreamProcessor._process_boundary_searchc s0|jd}|dkr4|jd}|dkr"t|jdkr tddS|jd|}|j|dd|_n|jd|}|j|d d|_t|d krOtd z |jd d d}Wntyk}ztd|d}~ww||}|r|jr}|IdH| |}|std|| |IdHd|_ dS)zProcess multipart headers.s r9s i zCMultipart headers too long or malformed (missing header terminator)FNr+izMultipart part headers too longutf-8strict)errorszInvalid header encoding: zInvalid or unsafe filename: r*T) rr=r.r0decodeUnicodeDecodeError_extract_filename_from_headersrr>_sanitize_upload_filename_start_new_filer)r header_endheader_section headers_textefilename safe_filenamerrr r3s<     z)MultipartStreamProcessor._process_headerscs|js d|_dS|j|j}|dkrEt|jt|jdkrCt|jt|jd}|jd|}|j|d|_||IdHdS|jd|}|j|d|_||IdH|IdH}d|_|S)z1Process file data, looking for the next boundary.r Nr9r:)rrrr=rr._write_file_datar>)rnext_boundary_pos write_size data_to_writer7rrr r4s$z+MultipartStreamProcessor._process_file_datacCsn|dD]/}|}|dr4td|tj}|r#|dStd|tj}|r4|dSqdS)z1Extract filename from Content-Disposition header. zcontent-disposition:zfilename="([^"]*)"r<zfilename=([^;\s]+)N)splitstriplowerr?research IGNORECASEgroup)rrNlinefilename_matchrrr rIsz7MultipartStreamProcessor._extract_filename_from_headerscCs|sdStj|}tdd|}tdd|}tdd|}|d}|r*|dvr,dSt|d krFtj|\}}|dd |dd }|S) z^ Sanitize uploaded filename to prevent directory traversal and other attacks. Nz [<>:"|?*]_z [^\w\-_\.]z\.\.+.z._)ra..)ospathbasenamerZsubrXr.splitext)rrP safe_name name_partext_partrrr rJs    z2MultipartStreamProcessor._sanitize_upload_filenamec s|j|jkrtd|jd|jr0tj|d}||jvr0td|dd|jtj|j |}|d}zt |d|_ |j |Wn+ty`}z td |d |d }~wtyu}z td |d |d }~ww|||d d|_|jd7_d S)zStart writing a new file.zToo many files in upload (max: )r<zFile extension 'z' not allowed. Allowed: , z .uploadingwbzCannot create upload file '': Nz(Permission denied creating upload file 'r)rP temp_path final_pathsize)rrr0rrfrgrjrYjoinropenrrr5OSErrorPermissionErrorr)rrPfile_ext file_pathtemp_file_pathrOrrr rKs2   z(MultipartStreamProcessor._start_new_filec s|jrl|rn|jdt|}||jkr&td|jdd|d|jdz|j||jdt|7<Wd StyS}z td|jdd|d }~wtyk}z td |jdd|d }~wwd Sd S) zWrite data to current file.rtzFile 'rPz' exceeds size limit: bytes (max: rnzError writing to file 'rqNzUnexpected error writing file ')rrr.rr0writerw Exception)rdatanew_sizerOrrr rR"s   $ z)MultipartStreamProcessor._write_file_datacs|jr |jdSdS)z'Flush file data to reduce memory usage.N)rflush)rrrr r/2sz)MultipartStreamProcessor._flush_file_datac s*|jsdS|jr|jd|_|jd}|jd}z=t||||jvr-|j||jd|jd|d}|d|dd|dd d IdH|j |d|_|WSt y}z+|d |d |IdHzt |||jvr|j|WnYWYd}~nd}~wwd|_dS) z.Finalize the current file and return its info.NrrrsrPrt)rPrtrgzPLOAD-SUCCESS] Uploaded: z (, bytes)z%[UPLOAD-ERROR] Error finalizing file : ) rrcloserfrenamerremover(rr5r~unlink)rrrrsr7rOrrr r>7sB      (     z/MultipartStreamProcessor._finalize_current_filecsg}|jrK|jdkrK|jdkr|IdHs |Sn(|jdkrC|jr2|jr2||jIdHd|_|IdH}|r@|| |S|jrK|jdks |S)zFinalize any remaining data.r r-Nr*r )rrr3rrRr>r5)rrr7rrr finalizebs&     z!MultipartStreamProcessor.finalizec s|jrz|jWnYd|_|jD]:}ztj|r/t||d|IdHWqtyP}z|d|d|IdHWYd}~qd}~ww|j dS)zClean up any temporary files.Nz$[UPLOAD-CLEANUP] Removed temp file: z*[UPLOAD-CLEANUP] Error removing temp file r) rrrrfrgexistsrr(r~clear)rrrrOrrr cleanupws$   (z MultipartStreamProcessor.cleanup)rr Nr Nr")__name__ __module__ __qualname____doc__r!r(r8r2r3r4rIrJrKrRr/r>rrrrrr rs$ (%, + rcseZdZdZ  d'fdd Zd(d d Zd d Zd)d dZddZddZ ddZ ddZ d(ddZ ddZ ddZddZdd Zd(d!d"Zd)d#d$Zd%d&ZZS)*FileDownloadHandlera0 HTTP server handler that provides secure file download functionality. Features: - File listing for GET requests without parameters - File download for GET requests with 'file' parameter - Directory traversal protection - MIME type detection - HTML directory listing Nr r cst||_||_||_||_||_||_|jr@tj |j|_tj |js1t d|jtj |jsBt d|jdSdS)a Initialize the file download handler. Args: download_directory (str): Directory to serve files from max_upload_size (int): Maximum total upload size per request in bytes (default: 2GB) max_file_size (int): Maximum individual file size in bytes (default: 500MB) allowed_extensions (set): Set of allowed file extensions (None = allow all) max_files_per_request (int): Maximum number of files per upload request (default: 100) z#Download directory does not exist: z"Download path is not a directory: N)superr!download_directorymax_upload_sizerrrrrfrgabspathrr0isdir)rrrrrrr __class__rr r!s zFileDownloadHandler.__init__r"cr#r$r%r&rrr r(r)zFileDownloadHandler.printcs|jsdS|s |jS|dd}g}|dD]%r#dkr$qdkr)qtfdddDr7dS|q|sB|jS|j}|D] tj|}qGtj |}ztj ||jg}||jkriWdSW|St yuYdSw) aW Sanitize the requested path to prevent directory traversal attacks while allowing safe navigation within the allowed directory tree. Args: path_request (str): The requested file or directory path Returns: str or None: Safe absolute path or None if invalid N\/rarbc3s|]}|vVqdSr$r).0char componentrr sz5FileDownloadHandler._sanitize_path..):*?"<>|) rreplacerWrXanyr5rfrgrur commonpathr0)r path_requestnormalized_pathpath_components safe_path common_pathrrr _sanitize_paths<       z"FileDownloadHandler._sanitize_pathc Cs|jsdS|dur |j}z||jkrd}d}ndtj||jtjd}|}g}g}t|D](}tj||}tj|rOtj |}| ||fq2tj |rZ| |q2| | t |} t |} ||} d| d} ||jkrtj|d} | dkrd} tj| }| d|d 7} |dkrtj|dnd}|rd |nd}d }| d |d 7} |D]+}t |}|dkr|}n |dd|}tj|}| d|d|d|d7} q|D]t\}}t |}|dkr|}n |dd|}tj|}||}tj|d}|dvr"d}n*|dvr*d}n"|dvr2d}n|dvr:d}n|dvrBd}n |dvrJd}nd}| d|d |d!|d"|d#|d$ 7} qt|t|}| dgd%t|d&t|d'|d(| d)|d|d*|d|d+|d|d,|d|d-|d|d.|d|d/|d|d07} | WSty}zt|}d1t |d2WYd}~Sd}~ww)3z Generate an HTML page listing files and directories. Args: current_path (str): The current directory path to list Returns: str: HTML content zV

File Server

No download directory configured.

NrzRoot Directoryu$ 📁 File Server - Directory Browser
📍 Current Location: z.
r"z@
⬆️ Back to Parent Directory
z?dir=zhttp://YOUR_SERVER_IP:8080u

📤 Upload Files

Uploading... 0%
💡 Drag and drop files or click to browse. Multiple files supported. Large files streamed efficiently.

📂 Directory Contents

zS r<)z.jpgz.jpegz.pngz.gifz.bmpu🖼️)z.mp4z.aviz.mkvz.movu🎬)z.mp3z.wavz.flacz.oggu🎵)z.pdfu📕)z.txtz.mdu📄)z.zipz.rarz.7zz.tarz.gzu📦zT u
📋 Name 📁 Type 📏 Size 🕒 Actions
📁 z Directory - Browse
 z File z9 Download

📊 Directory Statistics

u
📁 Directories
u
📄 Files
ue
📋 Total Items
💾 Current path: u

💻 Command Line Upload Examples

Use these commands to upload files from your terminal or scripts.

# Upload a single file
curl -X POST -F "files=@/path/to/your/file.txt" "z{"

# Upload multiple files
curl -X POST \
  -F "files=@file1.txt" \
  -F "files=@file2.jpg" \
  -F "files=@file3.pdf" \
  "uV"

Usage: Replace /path/to/your/file.txt with your actual file path. Use -F "files=@filename" for each file.

# Create multipart form data (limited support)
echo -e "--boundary123\r\nContent-Disposition: form-data; name=\"files\"; filename=\"test.txt\"\r\nContent-Type: text/plain\r\n\r\nfile content here\r\n--boundary123--\r\n" | \
wget --post-data=@- \
     --header="Content-Type: multipart/form-data; boundary=boundary123" \
     "u" -O -

Note: wget has limited multipart support. Recommended: Use cURL for better compatibility and easier syntax.

# Upload a single file
$file = Get-Item "C:\path\to\your\file.txt"
$form = @{'files' = $file}
$response = Invoke-RestMethod -Uri "z" -Method Post -Form $form
Write-Output $response

# Upload multiple files
$files = Get-ChildItem "C:\path\to\files\*" -Include *.txt,*.jpg,*.pdf
$form = @{}
foreach ($file in $files) {
    $form.Add("files", $file)
}
$response = Invoke-RestMethod -Uri "u" -Method Post -Form $form
Write-Output $response

Usage: Replace the file paths with your actual Windows file paths. Works with PowerShell 3.0+.

#!/usr/bin/env python3
import requests
import os

# Upload a single file
def upload_single_file(file_path):
    url = "a7"
    with open(file_path, 'rb') as f:
        files = {'files': f}
        response = requests.post(url, files=files)
        print(f"Status: {response.status_code}")
        print(f"Response: {response.text}")
        return response

# Upload multiple files
def upload_multiple_files(file_paths):
    url = "u"
    files = []
    try:
        for file_path in file_paths:
            files.append(('files', open(file_path, 'rb')))
        
        response = requests.post(url, files=files)
        print(f"Status: {response.status_code}")
        print(f"Response: {response.text}")
        return response
    finally:
        # Close all file handles
        for _, file_handle in files:
            file_handle.close()

# Example usage
if __name__ == "__main__":
    # Single file upload
    upload_single_file("/path/to/your/file.txt")
    
    # Multiple files upload
    file_list = [
        "/path/to/file1.txt",
        "/path/to/file2.jpg",
        "/path/to/file3.pdf"
    ]
    upload_multiple_files(file_list)

Installation: pip install requests
Usage: Replace file paths with your actual file paths. The script includes proper error handling and file cleanup.

z7

Error

Failed to list directory: 

)rrfrgrelpathrseplistdirruisfilegetsizer5rsorthtmlescape_generate_breadcrumbsdirnamerstripurllibparsequotelstrip_format_file_sizerjrYr.r~str) r current_path relative_path display_pathfiles directoriesitem item_pathrtescaped_directoryescaped_display_path breadcrumbs html_contentparent_relativeencoded_parentcurrent_dir_encodedcurrent_dir_param server_urlrescaped_dirnamedir_pathencoded_dirnamerPescaped_filenamerzencoded_filenamesize_strryicon total_itemsrO error_msgrrr _generate_directory_list_htmls&         EE K  "               ))00<<GGPP__ii*z1FileDownloadHandler._generate_directory_list_htmlc Cs|dkrdSdg}|dd}d}t|D]:\}}|sq||r%d|n|7}tj|}t|}|t|dkrE| d|dq| d|d|d qd |S) z Generate breadcrumb navigation HTML. Args: relative_path (str): The current relative path Returns: str: HTML breadcrumb navigation ruH🏠 Rootr"r<u 📁 zz📁 zu' → ) rXrW enumeraterrrrrr.r5ru) rrrpartsripart encoded_path escaped_partrrr rs    z)FileDownloadHandler._generate_breadcrumbscCst|\}}|p dS)z Get the MIME type for a file. Args: filepath (str): Path to the file Returns: str: MIME type zapplication/octet-stream) mimetypes guess_type)rfilepath mime_typer`rrr _get_mime_type%s z"FileDownloadHandler._get_mime_typec szTtj|jd}tj|j}d|vr.|dr.|dd}|||jIdHWdSd|vrK|ddr>|ddnd}| |IdHWdS| dIdHWdSt yv}z| ddt |IdHWYd}~dSd}~ww) z Handle GET requests for file listing, directory browsing, and downloads. Args: event: The HTTP request event asciifilerNdirr"zInternal Server Error: ) rrurlparsetargetrGparse_qsquery _serve_fileheaders_serve_directory_listr~ _serve_errorr)revent url_parts query_paramsrzrrOrrr do_GET2s *zFileDownloadHandler.do_GETc sz&tj|jd}tj|j}|ddgd}|||IdHWdSt yH}z| ddt |IdHWYd}~dSd}~ww)zy Handle POST requests for file uploads. Args: event: The HTTP request event rrr"rNrzUpload Error: ) rrrrrGrrget_handle_file_uploadr~rr)rrrr target_dirrOrrr do_POSTNs*zFileDownloadHandler.do_POSTc sz||}|rtj|s|ddIdHWdSd}d}|jD]\}}|dkr2|d}q"|dkr?t|d}q"|rG| dsS|ddIdHWdS|sa|dd IdHWdSt d |}|su|dd IdHWdS| d  d } d| d} d| dd} ||| | |IdHWdSty} z |d| IdH|ddt| IdHWYd} ~ dSd} ~ ww)z Handle multipart file upload with memory-efficient streaming. Args: event: The HTTP request event target_dir (str): Target directory for upload (relative to download_directory) zInvalid target directoryNs content-typerscontent-lengthzmultipart/form-dataz.Only multipart/form-data uploads are supportedzContent-Length header requiredzboundary=([^;]+)z Missing boundary in Content-Typer<rz--z[UPLOAD-ERROR] rzUpload processing error: )rrfrgrrrrYrGintr?rZr[r]rXencode_process_multipart_uploadr~r(r) rrrsafe_target_path content_typecontent_lengthnamevalueboundary_matchboundaryrend_boundary_bytesrOrrr rbsB     *z'FileDownloadHandler._handle_file_uploadc sg}d}d}d}||jkrtd|d|jdt||||j|j|j|jd} zl||kr~||} t|| } || IdH} | sBn<|t | 7}| | IdH} | | |d krz||d }|ddkrz| d |d d |dd|ddIdH||ks.| IdH}| ||||IdHWdSty}z'| d|IdH| IdH|ddt|IdHWYd}~dSd}~wty}z"| d|IdH| IdH|ddIdHWYd}~dSd}~wty(}z'| d|IdH| IdH|ddt|IdHWYd}~dSd}~wty[}z'| d|IdH| IdH|ddt|IdHWYd}~dSd}~ww)aJ Process multipart upload data with streaming to avoid memory issues. Args: target_path (str): Safe target directory path boundary_bytes (bytes): Multipart boundary end_boundary_bytes (bytes): End boundary content_length (int): Total content length rirzUpload too large: r|rn)rrrrNr z[UPLOAD-PROGRESS] .1f% (rrrz[UPLOAD-VALIDATION] rzUpload validation failed: z[UPLOAD-MEMORY] Out of memory: iz&Upload too large - insufficient memoryz[UPLOAD-DISK] Disk error: izServer storage error: z![UPLOAD-ERROR] Unexpected error: rzUpload processing failed: )rr0rrrrrmin_read_upload_chunkr.r8r1r(r_send_upload_success_responserrr MemoryErrorrwr~)rrrr ruploaded_filesbytes_processed chunk_sizer processor remaining read_sizer6 file_infosprogress final_filesrOrrr rsn        * * **z-FileDownloadHandler._process_multipart_uploadc sz:tj|jddIdH}t|tjr#|jr |jd|WSdWSt|tjr,WdS| dt |IdHWdStj yO| dIdHt dt yl}z| d|IdHt d |d}~wty}z| d |IdHt d |d}~ww) zQ Read a chunk of upload data from the HTTP request with timeout. g>@)timeoutNr z&[UPLOAD-ERROR] Unexpected event type: z*[UPLOAD-ERROR] Timeout reading upload dataz$Upload timeout - connection too slowz![UPLOAD-ERROR] Connection error: zUpload connection lost: z$[UPLOAD-ERROR] Error reading chunk: zError reading upload data: )asynciowait_for_wrapper next_event isinstanceh11Datar EndOfMessager(type TimeoutErrorrwConnectionErrorr~)rrtrrOrrr rs*  z&FileDownloadHandler._read_upload_chunkc s^z||jkr d}n tj||jtjd}dtj|dt |d}|D]}| |d}|dt |dd |d 7}q(|d tj|d 7}| d }|}|dd dfdtt | dfgtjd|d} |j| IdH|jtj|dIdH|jtIdHWdSty} z |d| IdHd} ~ ww)z Send a success response after file upload. Args: uploaded_files (list): List of uploaded file info target_path (str): Target directory path r"rzv Upload Successful

✅ Upload Successful!

Successfully uploaded z file(s):

    rtu
  • 📄 rPz (z)
    • z`

Redirecting back to directory listing in 3 seconds...

Click here if not redirected automatically

rD Content-Typetext/html; charset=utf-8rContent-Length status_coderNrz/[UPLOAD-ERROR] Error sending success response: )rrfrgrrrrrrr.rrrr basic_headersr1rr Responsersendr!r"r~r() rrr redirect_pathrr7rbodyrresponserOrrr rsB   "   z1FileDownloadHandler._send_upload_success_responsecCsR|dkr |dddS|dkr|dddS|dkr$|dddS|dS) z*Format file size in human-readable format.i@r z GBr MBz KBr,r)rrtrrr r4s z%FileDownloadHandler._format_file_sizec sPz||}|s|ddIdHWdStj|s'|ddIdHWdStj|s9|ddIdHWdS||}|d}|}| dd d fd t t |d fgt j d |d }|j|IdH|jt j|dIdH|jt IdHWdSty}z|ddt |IdHWYd}~dSd}~ww)z Serve the HTML directory listing page. Args: dir_path (str): The directory path to list (relative to download_directory) rzInvalid directory pathNzDirectory not foundzNot a directoryrDr&r'rr(r)r*r,rz!Error generating directory list: )rrrfrgrrrrr-r1rr.r r.rr/r!r"r~)rrrrr1rr2rOrrr r?s6      *z)FileDownloadHandler._serve_directory_listc sn||}|s|ddIdHdStj|s$|ddIdHdStj|s5|ddIdHdSz^tj|}||}d}|rZ|D]\}}|dkrY| d}nqHd } |d } d } |r| d rz"|d d} d| vr| dd \} }| rt | } |rt |} d} Wn t tfyYnw| | d }|}|d|dfdt|dfdddtj|ddfg| dkr|dd| d| d|dftj| |d}|j|IdHd}d }|}t|d}| d kr|| |d krwt||}||}|snezB|jtj|dIdH|t|7}|t|8}|dkrS|dd krS||d}|d |d!d"|d#d|d#d$IdHWnt yq}z|d%|d#d&|IdHd}~ww|d ksWdn 1swY|jt!IdHWdSt y}z|d'd(t|IdHWYd}~dSd}~ww))a Serve a specific file for download with support for large files. Args: file_path (str): The requested file path (can include directories) request_headers (dict): HTTP request headers for range support rzInvalid filenameNr5zFile not foundz Not a filesrangerrr<r)zbytes=-r&r()z Accept-RangessbyteszContent-Dispositionzattachment; filename="rz Content-Rangezbytes rr*irbr,i@r r z![FILE-SERVER] Download progress: r r rrz%[FILE-SERVER] Error sending chunk at z bytes: rzError serving file: )"rrrfrgrrrrrYrGr?rWrr0 IndexErrorr-r1rrrhr5r r.rr/rvseekr readr!r.r(r~r")rrzrequest_headersr file_sizer range_headerrr start_byteend_byter+ range_specstartendrrr2r bytes_sentbytes_remainingfrr6rrOrrr rgs           &        * *zFileDownloadHandler._serve_filec szSd|dt|dd}|}|dddfdtt|dfgtj||d }|j |Id H|j tj |d Id H|j t Id HWd St yr}z|d |Id HWYd }~d Sd }~ww) z Serve an error response. Args: status_code (int): HTTP status code message (str): Error message z

Error z

rrDr&r'rr(r*Nr,zError serving error response: )rrrr-r1rr.r r.rr/r!r"r~r()rr+messager1rr2rOrrr rs  $z FileDownloadHandler._serve_error)Nrr Nr Nrr$)rrrrr!r(rrrrrrrrrrrrrr __classcell__rrrr rs4   7$  1O7 (nrrr r c sbzfdd}t|||d}|IdHWSty0} z d| fWYd} ~ Sd} ~ ww)aD Run the file download server from a target. Args: target: Server target configuration download_directory (str): Directory to serve files from log_callback: Logging callback function max_upload_size (int): Maximum total upload size per request in bytes (default: 2GB) max_file_size (int): Maximum individual file size in bytes (default: 500MB) allowed_extensions (set): Set of allowed file extensions (None = allow all) max_files_per_request (int): Maximum number of files per upload request (default: 100) cstdS)Nrrrr)rrrrrrrrr sz-run_file_server_from_target..) log_callbackN)rserver~) rrrMrrrrhandler_factoryserverrOrrKr run_file_server_from_targetsrQ 127.0.0.1Fc s<zzZd}|r dd}|r#td|d|tdtj|n td|d|d|r4tdt||tj} t| ||||||d IdH\} } | durVtd | | IdHWn%tyitd Ynt y} z td | WYd} ~ nd} ~ wwW| dur| td dS| dur| td w) aM Run the file download server. Args: download_directory (str): Directory to serve files from host (str): Host to bind to port (int): Port to bind to debug (bool): Enable debug logging max_upload_size (int): Maximum total upload size per request in bytes (default: 2GB) max_file_size (int): Maximum individual file size in bytes (default: 500MB) allowed_extensions (set): Set of allowed file extensions (None = allow all) max_files_per_request (int): Maximum number of files per upload request (default: 100) Ncstd|dS)Nz[FILE-SERVER] )r()r'rrr rMsz%run_file_server..log_callbackzStarting file server on rzServing files from: z (no download directory)z+Debug mode enabled - verbose logging active)rrMrrrrzServer error:  Server stopped by userzServer stopped) r(rfrgrrr SERVER_TCPrQKeyboardInterruptr~cancel) rhostportdebugrrrrrMr server_taskerrrOrrr run_file_server sJ     r]c Cslddl}ddl}|jd|jdd}|jdddd |jd d d d d|jddtddd|jddddd|jdtddd|jdtddd|jdtd d!d|jd"td#d$|jd%d&d'd(d)|}|jrt j |js}t d*|jd+| d,t j |jst d-|jd.| d,|jd,ks|jd/krt d0|j| d,|jd1krt d2|j| d,|jd1krt d3|j| d,|jd,krt d4|j| d,d}|jrt}|jd5D]}|}|d6sd6|}||qt d7t d8|jrt d9t j |jnt d:t d;|jd<|j|jr6t d=t d>|jd?d@dAt dB|jd?d@dAt dC|j|rft dDdEt|nt dFt d8zt t!|j|j|j|j|j|j||jdGWdSt"yt dHYdSt#y}zt dI|| d,WYd}~dSd}~ww)Jz/ Main entry point for the file server. rNzLAsysocks File Server - Secure file upload/download server with web interfaceu Examples: %(prog)s # Start server on 127.0.0.1:8080 (no directory) %(prog)s /home/user/files # Serve files from directory %(prog)s /home/user/files --host 0.0.0.0 # Bind to all interfaces %(prog)s /home/user/files --port 9000 # Use custom port %(prog)s /home/user/files --debug # Enable debug logging %(prog)s --help # Show this help message Features: • Modern web interface with drag-and-drop uploads • Directory browsing with breadcrumb navigation • Real-time upload progress bars • Command-line examples (cURL, wget, PowerShell, Python) • Secure file handling with directory traversal protection • Large file streaming support ) descriptionformatter_classepilog directoryrz(Directory to serve files from (optional))nargshelpz--hostz-HrRz$Host to bind to (default: 127.0.0.1))defaultrcz--portz-prSzPort to bind to (default: 8080))r#rdrcz--debugz-d store_truez5Enable debug logging (shows detailed server activity))actionrcz--max-upload-sizerz=Maximum total upload size per request in bytes (default: 2GB)z--max-file-sizer z6Maximum individual file size in bytes (default: 500MB)z --max-filesr z9Maximum number of files per upload request (default: 100)z--allowed-extensionszHComma-separated list of allowed file extensions (e.g., ".txt,.jpg,.pdf"))r#rcz --versionz-vversionzAsysocks File Server 1.0.0)rfrgzError: Directory 'z' does not existr<zError: 'z' is not a directoryiz-Error: Port must be between 1 and 65535, got r4z8Error: max-upload-size must be at least 1024 bytes, got z6Error: max-file-size must be at least 1024 bytes, got z)Error: max-files must be at least 1, got rrazAsysocks File Serverz2==================================================z Directory: z"Directory: None (upload-only mode)zAddress: http://rzDebug: EnabledzMax upload size: rz.0fr3zMax file size: zMax files per request: zAllowed extensions: rozAllowed extensions: AllrJrTzFailed to start server: )$argparsesysArgumentParserRawDescriptionHelpFormatter add_argumentrr parse_argsrarfrgrr(exitrrYrr max_filesrsetrWrXr?addrYrrXrZrusortedrrunr]rVr~)rhriparserargsrextrOrrr mainAs           rw__main__)NNrr Nr )NrRrSFrr Nr )rrrf urllib.parserrrr tempfilerZpathlibrasysocks.unicomm.common.targetrr0asysocks.unicomm.protocol.server.http.httpserverrrrrrQr]rwrrrrr sN nm  7