PirateBoxScripts_Webserver/piratebox/piratebox/bin/droopy

1157 lines
40 KiB
Python
Executable File
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Droopy (http://stackp.online.fr/droopy)
# Copyright 2008-2013 (c) Pierre Duquesne <stackp@online.fr>
# Licensed under the New BSD License.
# Changelog
# 20131121 * Update HTML/CSS for mobile devices
# * Add HTTPS support
# * Add HTTP basic authentication
# * Add option to change uploaded file permissions
# * Add support for HTML5 multiple file upload
# 20120108 * Taiwanese translation by Li-cheng Hsu.
# 20110928 * Correctly save message with --save-config. Fix by Sven Radde.
# 20110708 * Polish translation by Jacek Politowski.
# 20110625 * Fix bug regarding filesystem name encoding.
# * Save the --dl option when --save-config is passed.
# 20110501 * Add the --dl option to let clients download files.
# * CSS speech bubble.
# 20101130 * CSS and HTML update. Switch to the new BSD License.
# 20100523 * Simplified Chinese translation by Ye Wei.
# 20100521 * Hungarian translation by Csaba Szigetvári.
# * Russian translation by muromec.
# * Use %APPDATA% Windows environment variable -- fix by Maik.
# 20091229 * Brazilian Portuguese translation by
# Carlos Eduardo Moreira dos Santos and Toony Poony.
# * IE layout fix by Carlos Eduardo Moreira dos Santos.
# * Galician translation by Miguel Anxo Bouzada.
# 20090721 * Indonesian translation by Kemas.
# 20090205 * Japanese translation by Satoru Matsumoto.
# * Slovak translation by CyberBoBaK.
# 20090203 * Norwegian translation by Preben Olav Pedersen.
# 20090202 * Korean translation by xissy.
# * Fix for unicode filenames by xissy.
# * Relies on 127.0.0.1 instead of "localhost" hostname.
# 20090129 * Serbian translation by kotnik.
# 20090125 * Danish translation by jan.
# 20081210 * Greek translation by n2j3.
# 20081128 * Slovene translation by david.
# * Romanian translation by Licaon.
# 20081022 * Swedish translation by David Eurenius.
# 20081001 * Droopy gets pretty (css and html rework).
# * Finnish translation by ipppe.
# 20080926 * Configuration saving and loading.
# 20080906 * Extract the file base name (some browsers send the full path).
# 20080905 * File is uploaded directly into the specified directory.
# 20080904 * Arabic translation by Djalel Chefrour.
# * Italian translation by fabius and d1s4st3r.
# * Dutch translation by Tonio Voerman.
# * Portuguese translation by Pedro Palma.
# * Turkish translation by Heartsmagic.
# 20080727 * Spanish translation by Federico Kereki.
# 20080624 * Option -d or --directory to specify the upload directory.
# 20080622 * File numbering to avoid overwriting.
# 20080620 * Czech translation by Jiří.
# * German translation by Michael.
# 20080408 * First release.
import BaseHTTPServer
import SocketServer
import cgi
import os
import posixpath
import macpath
import ntpath
import sys
import getopt
import mimetypes
import copy
import shutil
import tempfile
import socket
import locale
import urllib
import base64
import string
from pickle import load,dump
from ctypes import CDLL, c_char_p
LOGO = '''\
_____
| \.----.-----.-----.-----.--.--.
| -- | _| _ | _ | _ | | |
|_____/|__| |_____|_____| __|___ |
|__| |_____|
'''
USAGE='''\
Usage: droopy [options] [PORT]
Options:
-h, --help show this help message and exit
-d DIRECTORY, --directory DIRECTORY set the directory to upload files to
-m MESSAGE, --message MESSAGE set the message
-p PICTURE, --picture PICTURE set the picture
--dl provide download links
-a USER:PASS, --auth USER:PASS set the authentication credentials
--ssl PEMFILE set up https using the certificate file
--chmod MODE set the file permissions (octal value)
--save-config save options in a configuration file
--delete-config delete the configuration file and exit
Example:
droopy -m "Hi, this is Bob. You can send me a file." -p avatar.png
'''
picture = None
message = ""
port = 8000
directory = os.curdir
must_save_options = False
publish_files = False
auth = None
certfile = None
file_mode = None
## User supplied subdirectory stuff
userdirs= False # disabled by default, overridden by command line
# where to store the password file for users
#password_file='/opt/piratebox/passwd'
password_file='/tmp/passwd'
# What name will be used if subdir isn't specified
# must be lower case
anondir="anonymous"
# If no subdir is specified, put in root (0), or anondir(1)?
useAnondir=1
userdirform='''
<br><A HREF="javascript:show('advanced_fields')">%(advanced)s</a>
<div height="150px" class="box">
<fieldset id="advanced_fields" style="hidden:true;display:none">
<br>%(userdirname)s <input name="subdir" type="textbox">
<p>%(userdirpassword)s <input name="secret" type="textbox">
</fieldset>
</div>
'''
# -- HTML templates
style = '''
<meta name="viewport"
content="width=device-width,initial-scale=1,maximum-scale=1" />
<style type="text/css">
<!--
* {margin: 0; padding: 0;}
body {text-align: center; font-family: sans-serif;
color:#777;}
div {word-wrap: break-word;}
img {max-width: 100%%;}
a {color: #4499cc; text-decoration: none;}
.container {max-width: 700px; margin: auto; background-color: #fff;}
#linkurl {background-color: #333;}
#linkurl a {color: #ddd; text-decoration: none;}
#linkurl a:hover {color: #fff;}
#message {padding: 5px 0; font-size: 2em; font-weight: lighter;
letter-spacing: -2px; line-height: 50px; color: #aaa;}
#sending {display: none; font-style: italic;}
#sending .text {padding-top: 10px; color: #bbb; font-size: 0.8em;}
#wrapform { height: 30px; padding-top:30px;}
#progress {display: inline; border-collapse: separate; empty-cells: show;
border-spacing: 24px 0; padding: 0; vertical-align: bottom;}
#progress td {height: 17px; width: 17px; background-color: #eee;
padding: 0px; border-radius: 90px; box-shadow: 0 0 3px #bbb;}
#userinfo {padding-bottom: 20px;}
#files {
margin: auto;
padding: 13px 0;
text-align: left;
overflow: auto;
margin-bottom: 20px;
}
#files a {text-decoration: none; display: block; padding: 10px 20px;}
#files a:nth-child(2n+1) {background-color: #F7F7F7;}
#files a:link {color: #4499cc}
#files a:visited {color: #a0c0e0}
#files a:hover {background-color:#f0f0f0}
--></style>'''
userinfo = '''
<div id="userinfo">
%(message)s
%(divpicture)s
</div>
'''
maintmpl = '''
<!doctype html>
<html>
<head>
<title>%(maintitle)s</title>
''' + style + '''
<script language="JavaScript">
function swap() {
document.getElementById("form").style.display = "none";
document.getElementById("sending").style.display = "block";
pulse(0);
}
function pulse(i) {
var NUMCELL = 5;
var cell = document.getElementById("cell-" + (i %% NUMCELL));
var prev = document.getElementById("cell-"+((i - 1 + NUMCELL) %% NUMCELL));
cell.style.backgroundColor = "#7ac";
prev.style.backgroundColor = "#eee";
setTimeout(function() {pulse(i+1);}, 300);
}
function onunload() {
document.getElementById("form").style.display = "block";
document.getElementById("sending").style.display = "none";
}
function show(elem_id) {
elem = document.getElementById(elem_id);
if (elem) elem.style.display = 'block';
}
</script></head>
<body>
%(linkurl)s
<div class="container">
<div id="wrapform">
<div id="form" class="box">
<form method="post" enctype="multipart/form-data" action="">
<input name="upfile" type="file" multiple="yes">
<input value="%(submit)s" onclick="swap()" type="submit">
%(userdiropts)s
</form>
</div>
<div id="sending" class="box">
<table id="progress">
<tr>
<td id="cell-0"/><td id="cell-1"/><td id="cell-2"/>
<td id="cell-3"/><td id="cell-4"/>
</tr>
</table>
<div class="text">%(sending)s</div>
</div>
</div>
''' + userinfo + '''
%(files)s
</div>
</body>
</html>
'''
successtmpl = '''
<!doctype html>
<html>
<head><title> %(successtitle)s </title>
''' + style + '''
</head>
<body>
<div class="container">
<div id="wrapform">
<div class="box">
%(received)s
<a href="/"> %(another)s </a>
</div>
</div>
''' + userinfo + '''
</div>
</body>
</html>
'''
errortmpl = '''
<!doctype html>
<html>
<head><title> %(errortitle)s </title>
''' + style + '''
</head>
<body>
<div class="container">
<div id="wrapform">
<div class="box">
%(problem)s
<a href="/"> %(retry)s </a>
</div>
</div>
''' + userinfo + '''
</div>
</body>
</html>
'''
linkurltmpl = '''<div id="linkurl" class="box">
<a href="http://stackp.online.fr/droopy-ip.php?port=%(port)d&ssl=%(ssl)d"> %(discover)s
</a></div>'''
templates = {"main": maintmpl, "success": successtmpl, "error": errortmpl}
# -- Translations
ar = {"maintitle": u"إرسال ملف",
"submit": u"إرسال",
"sending": u"الملف قيد الإرسال",
"successtitle": u"تم استقبال الملف",
"received": u"تم استقبال الملف !",
"another": u"إرسال ملف آخر",
"errortitle": u"مشكلة",
"problem": u"حدثت مشكلة !",
"retry": u"إعادة المحاولة",
"discover": u"اكتشاف عنوان هذه الصفحة"}
cs = {"maintitle": u"Poslat soubor",
"submit": u"Poslat",
"sending": u"Posílám",
"successtitle": u"Soubor doručen",
"received": u"Soubor doručen !",
"another": u"Poslat další soubor",
"errortitle": u"Chyba",
"problem": u"Stala se chyba !",
"retry": u"Zkusit znova.",
"discover": u"Zjistit adresu stránky"}
da = {"maintitle": u"Send en fil",
"submit": u"Send",
"sending": u"Sender",
"successtitle": u"Fil modtaget",
"received": u"Fil modtaget!",
"another": u"Send en fil til.",
"errortitle": u"Problem",
"problem": u"Det er opstået en fejl!",
"retry": u"Forsøg igen.",
"discover": u"Find adressen til denne side"}
de = {"maintitle": "Datei senden",
"submit": "Senden",
"sending": "Sendet",
"successtitle": "Datei empfangen",
"received": "Datei empfangen!",
"another": "Weitere Datei senden",
"errortitle": "Fehler",
"problem": "Ein Fehler ist aufgetreten!",
"retry": "Wiederholen",
"discover": "Internet-Adresse dieser Seite feststellen"}
el = {"maintitle": u"Στείλε ένα αρχείο",
"submit": u"Αποστολή",
"sending": u"Αποστέλλεται...",
"successtitle": u"Επιτυχής λήψη αρχείου ",
"received": u"Λήψη αρχείου ολοκληρώθηκε",
"another": u"Στείλε άλλο ένα αρχείο",
"errortitle": u"Σφάλμα",
"problem": u"Παρουσιάστηκε σφάλμα",
"retry": u"Επανάληψη",
"discover": u"Βρες την διεύθυνση της σελίδας"}
en = {"maintitle": "Send a file",
"submit": "Send",
"sending": "Sending",
"successtitle": "File received",
"received": "File received!",
"another": "Send another file.",
"errortitle": "Problem",
"problem": "There has been a problem!",
"retry": "Retry.",
"discover": "Discover the address of this page",
"subdirerror": "There is a problem uploading to that subdirectory.<br> Check if that subdirectory already exits.<br> If it does not, make sure you are only using characters a-zA-Z0-9 for the username and password. <br>If the directory exists and the characters are valid, you probably are using the wrong password<p>",
"advanced": "advanced options",
"userdirname": "Name of directory to place files in ",
"userdirpassword": "Password for saving files to that directory (optional)"
}
es = {"maintitle": u"Enviar un archivo",
"submit": u"Enviar",
"sending": u"Enviando",
"successtitle": u"Archivo recibido",
"received": u"¡Archivo recibido!",
"another": u"Enviar otro archivo.",
"errortitle": u"Error",
"problem": u"¡Hubo un problema!",
"retry": u"Reintentar",
"discover": u"Descubrir la dirección de esta página"}
fi = {"maintitle": u"Lähetä tiedosto",
"submit": u"Lähetä",
"sending": u"Lähettää",
"successtitle": u"Tiedosto vastaanotettu",
"received": u"Tiedosto vastaanotettu!",
"another": u"Lähetä toinen tiedosto.",
"errortitle": u"Virhe",
"problem": u"Virhe lahetettäessä tiedostoa!",
"retry": u"Uudelleen.",
"discover": u"Näytä tämän sivun osoite"}
fr = {"maintitle": u"Envoyer un fichier",
"submit": u"Envoyer",
"sending": u"Envoi en cours",
"successtitle": u"Fichier reçu",
"received": u"Fichier reçu !",
"another": u"Envoyer un autre fichier.",
"errortitle": u"Problème",
"problem": u"Il y a eu un problème !",
"retry": u"Réessayer.",
"discover": u"Découvrir l'adresse de cette page"}
gl = {"maintitle": u"Enviar un ficheiro",
"submit": u"Enviar",
"sending": u"Enviando",
"successtitle": u"Ficheiro recibido",
"received": u"Ficheiro recibido!",
"another": u"Enviar outro ficheiro.",
"errortitle": u"Erro",
"problem": u"Xurdíu un problema!",
"retry": u"Reintentar",
"discover": u"Descubrir o enderezo desta páxina"}
hu = {"maintitle": u"Állomány küldése",
"submit": u"Küldés",
"sending": u"Küldés folyamatban",
"successtitle": u"Az állomány beérkezett",
"received": u"Az állomány beérkezett!",
"another": u"További állományok küldése",
"errortitle": u"Hiba",
"problem": u"Egy hiba lépett fel!",
"retry": u"Megismételni",
"discover": u"Az oldal Internet-címének megállapítása"}
id = {"maintitle": "Kirim sebuah berkas",
"submit": "Kirim",
"sending": "Mengirim",
"successtitle": "Berkas diterima",
"received": "Berkas diterima!",
"another": "Kirim berkas yang lain.",
"errortitle": "Permasalahan",
"problem": "Telah ditemukan sebuah kesalahan!",
"retry": "Coba kembali.",
"discover": "Kenali alamat IP dari halaman ini"}
it = {"maintitle": u"Invia un file",
"submit": u"Invia",
"sending": u"Invio in corso",
"successtitle": u"File ricevuto",
"received": u"File ricevuto!",
"another": u"Invia un altro file.",
"errortitle": u"Errore",
"problem": u"Si è verificato un errore!",
"retry": u"Riprova.",
"discover": u"Scopri lindirizzo di questa pagina"}
ja = {"maintitle": u"ファイル送信",
"submit": u"送信",
"sending": u"送信中",
"successtitle": u"受信完了",
"received": u"ファイルを受信しました!",
"another": u"他のファイルを送信する",
"errortitle": u"問題発生",
"problem": u"問題が発生しました!",
"retry": u"リトライ",
"discover": u"このページのアドレスを確認する"}
ko = {"maintitle": u"파일 보내기",
"submit": u"보내기",
"sending": u"보내는 중",
"successtitle": u"파일이 받아졌습니다",
"received": u"파일이 받아졌습니다!",
"another": u"다른 파일 보내기",
"errortitle": u"문제가 발생했습니다",
"problem": u"문제가 발생했습니다!",
"retry": u"다시 시도",
"discover": u"이 페이지 주소 알아보기"}
nl = {"maintitle": "Verstuur een bestand",
"submit": "Verstuur",
"sending": "Bezig met versturen",
"successtitle": "Bestand ontvangen",
"received": "Bestand ontvangen!",
"another": "Verstuur nog een bestand.",
"errortitle": "Fout",
"problem": "Er is een fout opgetreden!",
"retry": "Nog eens.",
"discover": "Vind het adres van deze pagina"}
no = {"maintitle": u"Send en fil",
"submit": u"Send",
"sending": u"Sender",
"successtitle": u"Fil mottatt",
"received": u"Fil mottatt !",
"another": u"Send en ny fil.",
"errortitle": u"Feil",
"problem": u"Det har skjedd en feil !",
"retry": u"Send på nytt.",
"discover": u"Finn addressen til denne siden"}
pl = {"maintitle": u"Wyślij plik",
"submit": u"Wyślij",
"sending": u"Wysyłanie",
"successtitle": u"Plik wysłany",
"received": u"Plik wysłany!",
"another": u"Wyślij kolejny plik.",
"errortitle": u"Problem",
"problem": u"Wystąpił błąd!",
"retry": u"Spróbuj ponownie.",
"discover": u"Znajdź adres tej strony"}
pt = {"maintitle": u"Enviar um ficheiro",
"submit": u"Enviar",
"sending": u"A enviar",
"successtitle": u"Ficheiro recebido",
"received": u"Ficheiro recebido !",
"another": u"Enviar outro ficheiro.",
"errortitle": u"Erro",
"problem": u"Ocorreu um erro !",
"retry": u"Tentar novamente.",
"discover": u"Descobrir o endereço desta página"}
pt_br = {
"maintitle": u"Enviar um arquivo",
"submit": u"Enviar",
"sending": u"Enviando",
"successtitle": u"Arquivo recebido",
"received": u"Arquivo recebido!",
"another": u"Enviar outro arquivo.",
"errortitle": u"Erro",
"problem": u"Ocorreu um erro!",
"retry": u"Tentar novamente.",
"discover": u"Descobrir o endereço desta página"}
ro = {"maintitle": u"Trimite un fişier",
"submit": u"Trimite",
"sending": u"Se trimite",
"successtitle": u"Fişier recepţionat",
"received": u"Fişier recepţionat !",
"another": u"Trimite un alt fişier.",
"errortitle": u"Problemă",
"problem": u"A intervenit o problemă !",
"retry": u"Reîncearcă.",
"discover": u"Descoperă adresa acestei pagini"}
ru = {"maintitle": u"Отправить файл",
"submit": u"Отправить",
"sending": u"Отправляю",
"successtitle": u"Файл получен",
"received": u"Файл получен !",
"another": u"Отправить другой файл.",
"errortitle": u"Ошибка",
"problem": u"Произошла ошибка !",
"retry": u"Повторить.",
"discover": u"Посмотреть адрес этой страницы"}
sk = {"maintitle": u"Pošli súbor",
"submit": u"Pošli",
"sending": u"Posielam",
"successtitle": u"Súbor prijatý",
"received": u"Súbor prijatý !",
"another": u"Poslať ďalší súbor.",
"errortitle": u"Chyba",
"problem": u"Vyskytla sa chyba!",
"retry": u"Skúsiť znova.",
"discover": u"Zisti adresu tejto stránky"}
sl = {"maintitle": u"Pošlji datoteko",
"submit": u"Pošlji",
"sending": u"Pošiljam",
"successtitle": u"Datoteka prejeta",
"received": u"Datoteka prejeta !",
"another": u"Pošlji novo datoteko.",
"errortitle": u"Napaka",
"problem": u"Prišlo je do napake !",
"retry": u"Poizkusi ponovno.",
"discover": u"Poišči naslov na tej strani"}
sr = {"maintitle": u"Pošalji fajl",
"submit": u"Pošalji",
"sending": u"Šaljem",
"successtitle": u"Fajl primljen",
"received": u"Fajl primljen !",
"another": u"Pošalji još jedan fajl.",
"errortitle": u"Problem",
"problem": u"Desio se problem !",
"retry": u"Pokušaj ponovo.",
"discover": u"Otkrij adresu ove stranice"}
sv = {"maintitle": u"Skicka en fil",
"submit": u"Skicka",
"sending": u"Skickar...",
"successtitle": u"Fil mottagen",
"received": u"Fil mottagen !",
"another": u"Skicka en fil till.",
"errortitle": u"Fel",
"problem": u"Det har uppstått ett fel !",
"retry": u"Försök igen.",
"discover": u"Ta reda på adressen till denna sida"}
tr = {"maintitle": u"Dosya gönder",
"submit": u"Gönder",
"sending": u"Gönderiliyor...",
"successtitle": u"Gönderildi",
"received": u"Gönderildi",
"another": u"Başka bir dosya gönder.",
"errortitle": u"Problem.",
"problem": u"Bir problem oldu !",
"retry": u"Yeniden dene.",
"discover": u"Bu sayfanın adresini bul"}
zh_cn = {
"maintitle": u"发送文件",
"submit": u"发送",
"sending": u"发送中",
"successtitle": u"文件已收到",
"received": u"文件已收到!",
"another": u"发送另一个文件。",
"errortitle": u"问题",
"problem": u"出现问题!",
"retry": u"重试。",
"discover": u"查看本页面的地址"}
zh_tw = {
"maintitle": u"上傳檔案",
"submit": u"上傳",
"sending": u"傳送中...",
"successtitle": u"已收到檔案",
"received": u"已收到檔案!",
"another": u"上傳另一個檔案。",
"errortitle": u"錯誤",
"problem": u"出現錯誤!",
"retry": u"重試。",
"discover": u"查閱本網頁的網址"}
translations = {"ar": ar, "cs": cs, "da": da, "de": de, "el": el, "en": en,
"es": es, "fi": fi, "fr": fr, "gl": gl, "hu": hu, "id": id,
"it": it, "ja": ja, "ko": ko, "nl": nl, "no": no, "pl": pl,
"pt": pt, "pt-br": pt_br, "ro": ro, "ru": ru, "sk": sk,
"sl": sl, "sr": sr, "sv": sv, "tr": tr, "zh-cn": zh_cn,
"zh-tw": zh_tw}
class DroopyFieldStorage(cgi.FieldStorage):
"""The file is created in the destination directory and its name is
stored in the tmpfilename attribute.
"""
TMPPREFIX = 'tmpdroopy'
def make_file(self, binary=None):
fd, name = tempfile.mkstemp(dir=directory, prefix=self.TMPPREFIX)
self.tmpfile = os.fdopen(fd, 'w+b')
self.tmpfilename = name
return self.tmpfile
class HTTPUploadHandler(BaseHTTPServer.BaseHTTPRequestHandler):
protocol_version = 'HTTP/1.0'
form_field = 'upfile'
divpicture = '<div class="box"><img src="/__droopy/picture"/></div>'
def _safername(self,name):
""" Checking whether we can make a directory-name out of this name
Thanks to stackoverflow, Vinko Vrsalovic
"""
valid_chars = frozenset("-_() %s%s" % (string.letters, string.digits))
name2 = ''.join(c for c in name if c in valid_chars)
if len(name) == 0:
return anondir
if name2 == name:
return name.encode('utf8')
else:
print('WARNING: name too complex; try %s'%name2)
return False
def _mkuserpath(self,userdir,filename):
mydir = directory + "/" + userdir
if (os.path.exists(mydir)):
localpath = os.path.join(mydir, filename).encode('utf-8')
else:
try:
os.mkdir(mydir)
except IOError:
print ("can't make directory")
else:
localpath = os.path.join(mydir, filename).encode('utf-8')
return localpath
def _login(self,subdir, passwd ):
""" Allow user to provide subdirectory & write access password.
This offers little security, but helps organization
returns success,NAME
"""
name = self._safername( subdir.lower() )
password = passwd #safename( passwd )
if (name == "" or name == anondir) :
return (True, anondir ) # don't let people set password on anon
try:
pairs = load(open(password_file))
except IOError:
print('WARNING: No password file found. Starting from scratch.')
pairs = {}
if name in pairs:
if pairs[name] == password:
return True,name
else:
print('ERROR: wrong password')
return False,"BadPassword"
else:
print('INFO: Adding subdir %s to the system'%name)
pairs[name] = password
dump(pairs, open(password_file, 'w'))
return True,name
def html(self, page, errormsg=""):
"""
page can be "main", "success", or "error"
returns an html page (in the appropriate language) as a string
"""
# -- Parse accept-language header
if not self.headers.has_key("accept-language"):
a = []
else:
a = self.headers["accept-language"]
a = a.split(',')
a = [e.split(';q=') for e in a]
a = [(lambda x: len(x)==1 and (1, x[0]) or
(float(x[1]), x[0])) (e) for e in a]
a.sort()
a.reverse()
a = [x[1] for x in a]
# now a is an ordered list of preferred languages
# -- Choose the appropriate translation dictionary (default is english)
lang = "en"
for l in a:
if translations.has_key(l):
lang = l
break
dico = copy.copy(translations[lang])
# -- Set message and picture
if message:
dico["message"] = ('<div id="message">%s</div>' %
message)
else:
dico["message"] = ""
if picture != None:
dico["divpicture"] = self.divpicture
else:
dico["divpicture"] = ""
# -- Possibly provide download links
links = ""
names = self.published_files()
if names:
for name in names:
links += '<a href="/%s">%s</a>' % (
urllib.quote(name.encode('utf-8')),
name)
links = '<div id="files">' + links + '</div>'
dico["files"] = links
# -- Add a link to discover the url
if self.client_address[0] == "127.0.0.1":
dico["port"] = self.server.server_port
dico["ssl"] = int(certfile is not None)
dico["linkurl"] = linkurltmpl % dico
else:
dico["linkurl"] = ""
if (userdirs == True):
dico["userdiropts"] = userdirform % dico
else:
dico["userdiropts"] = ''
dico["problem"] = ""
if (errormsg != ""):
if dico[errormsg]:
dico["problem"]=dico[errormsg]
return templates[page] % dico
def check_auth(method):
def decorated(self, *args):
if auth and self.headers.getheader('Authorization') != \
'Basic ' + base64.b64encode(auth):
self.send_response(401)
self.send_header('WWW-Authenticate', 'Basic realm=\"Droopy\"')
self.send_header('Content-type', 'text/html')
self.end_headers()
else:
method(self, *args)
return decorated
@check_auth
def do_GET(self):
name = self.path.lstrip('/')
name = urllib.unquote(name)
name = name.decode('utf-8')
if picture != None and self.path == '/__droopy/picture':
# send the picture
self.send_file(picture)
elif name in self.published_files():
localpath = os.path.join(directory, name)
self.send_file(localpath)
else:
self.send_html(self.html("main"))
@check_auth
def do_POST(self):
# Do some browsers /really/ use multipart ? maybe Opera ?
try:
self.log_message("Started file transfer")
# -- Set up environment for cgi.FieldStorage
env = {}
env['REQUEST_METHOD'] = self.command
if self.headers.typeheader is None:
env['CONTENT_TYPE'] = self.headers.type
else:
env['CONTENT_TYPE'] = self.headers.typeheader
# -- Save file (numbered to avoid overwriting, ex: foo-3.png)
form = DroopyFieldStorage(fp = self.rfile, environ = env);
file_items = form[self.form_field]
#-- Handle multiple file upload
if not isinstance(file_items, list):
file_items = [file_items]
for item in file_items:
filename = self.basename(item.filename).decode('utf-8')
if filename == "":
continue
if filename.lower() == "index.html" or filename.lower() == "index.htm":
self.send_response(303)
self.send_header('Location', '/')
self.end_headers()
continue
localpath =''
if (userdirs == True ):
subdir = form['subdir'].value
passwd = form['secret'].value
success,userdir=self._login(subdir, passwd )
if success == False:
self.send_html(self.html("error","subdirerror"))
#os.unlink(fileitem.tmpfilename)
return
else:
if (userdir == anondir or userdir == ''):
if ( useAnondir == 0):
localpath = None
else:
localpath=self._mkuserpath(anondir,filename)
else:
localpath=self._mkuserpath(userdir,filename)
localpath = os.path.join(directory, filename).encode('utf-8')
root, ext = os.path.splitext(localpath)
i = 1
# race condition, but hey...
while (os.path.exists(localpath)):
localpath = "%s-%d%s" % (root, i, ext)
i = i + 1
if hasattr(item, 'tmpfile'):
# DroopyFieldStorage.make_file() has been called
item.tmpfile.close()
shutil.move(item.tmpfilename, localpath)
else:
# no temporary file, self.file is a StringIO()
# see cgi.FieldStorage.read_lines()
fout = file(localpath, 'wb')
shutil.copyfileobj(item.file, fout)
fout.close()
if file_mode is not None:
os.chmod(localpath, file_mode)
self.log_message("Received: %s", os.path.basename(localpath))
# -- Reply
if publish_files:
# The file list gives a feedback for the upload
# success
self.send_response(301)
self.send_header("Location", "/")
self.end_headers()
else:
self.send_html(self.html("success"))
except Exception, e:
self.log_message(repr(e))
self.send_html(self.html("error"))
def send_html(self, htmlstr):
self.send_response(200)
self.send_header('Content-type','text/html; charset=utf-8')
self.end_headers()
self.wfile.write(htmlstr.encode('utf-8'))
def send_file(self, localpath):
f = open(localpath, 'rb')
self.send_response(200)
self.send_header('Content-type',
mimetypes.guess_type(localpath)[0])
self.send_header('Content-length', os.fstat(f.fileno())[6])
self.end_headers()
shutil.copyfileobj(f, self.wfile)
def basename(self, path):
"""Extract the file base name (some browsers send the full file path).
"""
for mod in posixpath, macpath, ntpath:
path = mod.basename(path)
return path
def published_files(self):
"""Returns the list of files that should appear as download links.
The returned filenames are unicode strings.
"""
if publish_files:
# os.listdir() returns a list of unicode strings when the
# directory is passed as an unicode string itself.
names = [name for name in os.listdir(unicode(directory))
if os.path.isfile(os.path.join(directory, name))
and not name.startswith(DroopyFieldStorage.TMPPREFIX)]
names.sort(key=lambda s: s.lower())
else:
names = []
return names
def handle(self):
try:
BaseHTTPServer.BaseHTTPRequestHandler.handle(self)
except socket.error, e:
self.log_message(str(e))
raise Abort()
class Abort(Exception): pass
class ThreadedHTTPServer(SocketServer.ThreadingMixIn,
BaseHTTPServer.HTTPServer):
def handle_error(self, request, client_address):
# Override SocketServer.handle_error
exctype = sys.exc_info()[0]
if not exctype is Abort:
BaseHTTPServer.HTTPServer.handle_error(self,request,client_address)
# -- Options
def fullpath(path):
return os.path.abspath(os.path.expanduser(path))
def configfile():
appname = 'droopy'
# os.name is 'posix', 'nt', 'os2', 'mac', 'ce' or 'riscos'
if os.name == 'posix':
getenv = CDLL("libc.so.6").getenv
getenv.restype = c_char_p
filename = "%s/.%s" % (getenv("HOME"), appname)
elif os.name == 'mac':
filename = ("%s/Library/Application Support/%s" %
(os.environ["HOME"], appname))
elif os.name == 'nt':
filename = ("%s\%s" % (os.environ["APPDATA"], appname))
else:
filename = None
return filename
def save_options():
opt = []
if message:
opt.append('--message=%s' % message.replace('\n', '\\n'))
if picture:
opt.append('--picture=%s' % picture)
if directory:
opt.append('--directory=%s' % directory)
if auth:
opt.append('--auth=%s' % auth)
if certfile:
opt.append('--ssl=%s' % certfile)
if file_mode:
opt.append('--chmod=%s' % file_mode)
if publish_files:
opt.append('--dl')
if port:
opt.append('%d' % port)
f = open(configfile(), 'w')
f.write('\n'.join(opt).encode('utf8'))
f.close()
def load_options():
try:
f = open(configfile())
cmd = [line.strip().decode('utf8').replace('\\n', '\n')
for line in f.readlines()]
parse_args(cmd)
f.close()
return True
except IOError, e:
return False
def parse_args(cmd=None):
"""Parse command-line arguments.
Parse sys.argv[1:] if no argument is passed.
"""
global picture, message, port, directory, must_save_options, publish_files
global auth, certfile, file_mode
global userdirs
if cmd == None:
cmd = sys.argv[1:]
lang, encoding = locale.getdefaultlocale()
if encoding != None:
cmd = [a.decode(encoding) for a in cmd]
opts, args = None, None
try:
opts, args = getopt.gnu_getopt(cmd, "p:m:d:a:u:h",
["picture=","message=", "directory=",
"auth=", "ssl=", "chmod=", "help", "userdirs",
"save-config", "delete-config", "dl"])
except Exception, e:
print e
sys.exit(1)
for o, a in opts:
if o in ["-p", "--picture"]:
if os.path.exists(a):
picture = fullpath(a)
else:
print "Picture not found: '%s'" % a
elif o in ["-m", "--message"]:
message = a
elif o in ['-d', '--directory']:
directory = fullpath(a)
elif o in ["-u", "--userdirs" ] :
userdirs = True
print "userdirs are %s" %userdirs
elif o in ['--save-config']:
must_save_options = True
elif o in ['--delete-config']:
try:
filename = configfile()
os.remove(filename)
print 'Deleted ' + filename
except Exception, e:
print e
sys.exit(0)
elif o in ['--dl']:
publish_files = True
elif o in ['-a', '--auth']:
if ':' not in a:
print "Error: authentication credentials must be " \
"specified as USER:PASSWORD"
sys.exit(1)
auth = a
elif o in ['--ssl']:
if not os.path.isfile(a):
print "PEM file not found: '%s'" % a
sys.exit(1)
certfile = fullpath(a)
elif o in ['--chmod']:
try:
file_mode = int(a, 8)
except ValueError:
print "Invalid octal value passed to chmod option: '%s'" % a
sys.exit(1)
elif o in ['-h', '--help']:
print USAGE
sys.exit(0)
# port number
try:
if args[0:]:
port = int(args[0])
except ValueError:
print args[0], "is not a valid port number"
sys.exit(1)
# --
def run():
"""Run the webserver."""
socket.setdefaulttimeout(3*60)
server_address = ('', port)
httpd = ThreadedHTTPServer(server_address, HTTPUploadHandler)
if certfile:
try:
import ssl
except:
print "Error: Could not import module 'ssl', exiting."
sys.exit(2)
httpd.socket = ssl.wrap_socket(httpd.socket, certfile=certfile,
server_side=True)
httpd.serve_forever()
if __name__ == '__main__':
print LOGO
config_found = load_options()
parse_args()
if config_found:
print 'Configuration found in %s' % configfile()
else:
print "No configuration file found."
if must_save_options:
save_options()
print "Options saved in %s" % configfile()
proto = 'https' if certfile else 'http'
print "Files will be uploaded to %s" % directory
print
print ("HTTP server running... Check it out at %s://localhost:%d"
% (proto, port))
try:
run()
except KeyboardInterrupt:
print '^C received, shutting down server'
# some threads may run until they terminate