summaryrefslogtreecommitdiffstats
path: root/woof
diff options
context:
space:
mode:
Diffstat (limited to 'woof')
-rwxr-xr-xwoof429
1 files changed, 0 insertions, 429 deletions
diff --git a/woof b/woof
deleted file mode 100755
index 03de063..0000000
--- a/woof
+++ /dev/null
@@ -1,429 +0,0 @@
-#!/usr/bin/python2
-# -*- encoding: utf-8 -*-
-#
-# woof -- an ad-hoc single file webserver
-# Copyright (C) 2004-2009 Simon Budig <simon@budig.de>
-#
-# This program is free software; you can redistribute it and/or modify
-# it under the terms of the GNU General Public License as published by
-# the Free Software Foundation; either version 2 of the License, or
-# (at your option) any later version.
-#
-# This program is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-# GNU General Public License for more details.
-#
-# A copy of the GNU General Public License is available at
-# http://www.fsf.org/licenses/gpl.txt, you can also write to the
-# Free Software Foundation, Inc., 59 Temple Place - Suite 330,
-# Boston, MA 02111-1307, USA.
-
-# Darwin support with the help from Mat Caughron, <mat@phpconsulting.com>
-# Solaris support by Colin Marquardt, <colin.marquardt@zmd.de>
-# FreeBSD support with the help from Andy Gimblett, <A.M.Gimblett@swansea.ac.uk>
-# Cygwin support by Stefan Reichör <stefan@xsteve.at>
-# tarfile usage suggested by Morgan Lefieux <comete@geekandfree.org>
-
-import sys, os, socket, getopt, commands
-import urllib, BaseHTTPServer
-import ConfigParser
-import shutil, tarfile, zipfile
-import struct
-
-maxdownloads = 1
-TM = object
-cpid = -1
-compressed = 'gz'
-
-
-class EvilZipStreamWrapper(TM):
- def __init__ (self, victim):
- self.victim_fd = victim
- self.position = 0
- self.tells = []
- self.in_file_data = 0
-
- def tell (self):
- self.tells.append (self.position)
- return self.position
-
- def seek (self, offset, whence = 0):
- if offset != 0:
- if offset == self.tells[0] + 14:
- # the zipfile module tries to fix up the file header.
- # write Data descriptor header instead,
- # the next write from zipfile
- # is CRC, compressed_size and file_size (as required)
- self.write ("PK\007\010")
- elif offset == self.tells[1]:
- # the zipfile module goes to the end of the file. The next
- # data written definitely is infrastructure (in_file_data = 0)
- self.tells = []
- self.in_file_data = 0
- else:
- raise "unexpected seek for EvilZipStreamWrapper"
-
- def write (self, data):
- # only test for headers if we know that we're not writing
- # (potentially compressed) data.
- if self.in_file_data == 0:
- if data[:4] == zipfile.stringFileHeader:
- # fix the file header for extra Data descriptor
- hdr = list (struct.unpack (zipfile.structFileHeader, data[:30]))
- hdr[3] |= (1 << 3)
- data = struct.pack (zipfile.structFileHeader, *hdr) + data[30:]
- self.in_file_data = 1
- elif data[:4] == zipfile.stringCentralDir:
- # fix the directory entry to match file header.
- hdr = list (struct.unpack (zipfile.structCentralDir, data[:46]))
- hdr[5] |= (1 << 3)
- data = struct.pack (zipfile.structCentralDir, *hdr) + data[46:]
-
- self.position += len (data)
- self.victim_fd.write (data)
-
- def __getattr__ (self, name):
- return getattr (self.victim_fd, name)
-
-
-# Utility function to guess the IP (as a string) where the server can be
-# reached from the outside. Quite nasty problem actually.
-
-def find_ip ():
- if sys.platform == "cygwin":
- ipcfg = os.popen("ipconfig").readlines()
- for l in ipcfg:
- try:
- candidat = l.split(":")[1].strip()
- if candidat[0].isdigit():
- break
- except:
- pass
- return candidat
-
- os.environ["PATH"] = "/sbin:/usr/sbin:/usr/local/sbin:" + os.environ["PATH"]
- platform = os.uname()[0];
-
- if platform == "Linux":
- netstat = commands.getoutput ("LC_MESSAGES=C netstat -rn")
- defiface = [i.split ()[-1] for i in netstat.split ('\n')
- if i.split ()[0] == "0.0.0.0"]
- elif platform in ("Darwin", "FreeBSD", "NetBSD"):
- netstat = commands.getoutput ("LC_MESSAGES=C netstat -rn")
- defiface = [i.split ()[-1] for i in netstat.split ('\n')
- if len(i) > 2 and i.split ()[0] == "default"]
- elif platform == "SunOS":
- netstat = commands.getoutput ("LC_MESSAGES=C netstat -arn")
- defiface = [i.split ()[-1] for i in netstat.split ('\n')
- if len(i) > 2 and i.split ()[0] == "0.0.0.0"]
- else:
- print >>sys.stderr, "Unsupported platform; please add support for your platform in find_ip().";
- return None
-
- if not defiface:
- return None
-
- if platform == "Linux":
- ifcfg = commands.getoutput ("LC_MESSAGES=C ifconfig "
- + defiface[0]).split ("inet addr:")
- elif platform in ("Darwin", "FreeBSD", "SunOS", "NetBSD"):
- ifcfg = commands.getoutput ("LC_MESSAGES=C ifconfig "
- + defiface[0]).split ("inet ")
-
- if len (ifcfg) != 2:
- return None
- ip_addr = ifcfg[1].split ()[0]
-
- # sanity check
- try:
- ints = [ i for i in ip_addr.split (".") if 0 <= int(i) <= 255]
- if len (ints) != 4:
- return None
- except ValueError:
- return None
-
- return ip_addr
-
-
-# Main class implementing an HTTP-Requesthandler, that serves just a single
-# file and redirects all other requests to this file (this passes the actual
-# filename to the client).
-# Currently it is impossible to serve different files with different
-# instances of this class.
-
-class FileServHTTPRequestHandler (BaseHTTPServer.BaseHTTPRequestHandler):
- server_version = "Simons FileServer"
- protocol_version = "HTTP/1.0"
-
- filename = "."
-
- def log_request (self, code='-', size='-'):
- if code == 200:
- BaseHTTPServer.BaseHTTPRequestHandler.log_request (self, code, size)
-
-
- def do_GET (self):
- global maxdownloads, cpid, compressed
-
- # Redirect any request to the filename of the file to serve.
- # This hands over the filename to the client.
-
- self.path = urllib.quote (urllib.unquote (self.path))
- location = "/" + urllib.quote (os.path.basename (self.filename))
- if os.path.isdir (self.filename):
- if compressed == 'gz':
- location += ".tar.gz"
- elif compressed == 'bz2':
- location += ".tar.bz2"
- elif compressed == 'zip':
- location += ".zip"
- else:
- location += ".tar"
-
- if self.path != location:
- txt = """\
- <html>
- <head><title>302 Found</title></head>
- <body>302 Found <a href="%s">here</a>.</body>
- </html>\n""" % location
- self.send_response (302)
- self.send_header ("Location", location)
- self.send_header ("Content-type", "text/html")
- self.send_header ("Content-Length", str (len (txt)))
- self.end_headers ()
- self.wfile.write (txt)
- return
-
- maxdownloads -= 1
-
- # let a separate process handle the actual download, so that
- # multiple downloads can happen simultaneously.
-
- cpid = os.fork ()
-
- if cpid == 0:
- # Child process
- child = None
- type = None
-
- if os.path.isfile (self.filename):
- type = "file"
- elif os.path.isdir (self.filename):
- type = "dir"
-
- if not type:
- print >> sys.stderr, "can only serve files or directories. Aborting."
- sys.exit (1)
-
- self.send_response (200)
- self.send_header ("Content-type", "application/octet-stream")
- if os.path.isfile (self.filename):
- self.send_header ("Content-Length",
- os.path.getsize (self.filename))
- self.end_headers ()
-
- try:
- if type == "file":
- datafile = file (self.filename)
- shutil.copyfileobj (datafile, self.wfile)
- datafile.close ()
- elif type == "dir":
- if compressed == 'zip':
- ezfile = EvilZipStreamWrapper (self.wfile)
- zfile = zipfile.ZipFile (ezfile, 'w', zipfile.ZIP_DEFLATED)
- stripoff = os.path.dirname (self.filename) + os.sep
-
- for root, dirs, files in os.walk (self.filename):
- for f in files:
- filename = os.path.join (root, f)
- if filename[:len (stripoff)] != stripoff:
- raise RuntimeException, "invalid filename assumptions, please report!"
- zfile.write (filename, filename[len (stripoff):])
- zfile.close ()
- else:
- tfile = tarfile.open (mode=('w|' + compressed),
- fileobj=self.wfile)
- tfile.add (self.filename,
- arcname=os.path.basename(self.filename))
- tfile.close ()
- except Exception, e:
- print e
- print >>sys.stderr, "Connection broke. Aborting"
-
-
-def serve_files (filename, maxdown = 1, ip_addr = '', port = 8080):
- global maxdownloads
-
- maxdownloads = maxdown
-
- # We have to somehow push the filename of the file to serve to the
- # class handling the requests. This is an evil way to do this...
-
- FileServHTTPRequestHandler.filename = filename
-
- try:
- httpd = BaseHTTPServer.HTTPServer ((ip_addr, port),
- FileServHTTPRequestHandler)
- except socket.error:
- print >>sys.stderr, "cannot bind to IP address '%s' port %d" % (ip_addr, port)
- sys.exit (1)
-
- if not ip_addr:
- ip_addr = find_ip ()
- if ip_addr:
- print "Now serving on http://%s:%s/" % (ip_addr, httpd.server_port)
-
- while cpid != 0 and maxdownloads > 0:
- httpd.handle_request ()
-
-
-
-def usage (defport, defmaxdown, errmsg = None):
- name = os.path.basename (sys.argv[0])
- print >>sys.stderr, """
- Usage: %s [-i <ip_addr>] [-p <port>] [-c <count>] <file>
- %s [-i <ip_addr>] [-p <port>] [-c <count>] [-z|-j|-Z|-u] <dir>
- %s [-i <ip_addr>] [-p <port>] [-c <count>] -s
-
- Serves a single file <count> times via http on port <port> on IP
- address <ip_addr>.
- When a directory is specified, an tar archive gets served. By default
- it is gzip compressed. You can specify -z for gzip compression,
- -j for bzip2 compression, -Z for ZIP compression or -u for no compression.
- You can configure your default compression method in the configuration
- file described below.
-
- When -s is specified instead of a filename, %s distributes itself.
-
- defaults: count = %d, port = %d
-
- You can specify different defaults in two locations: /etc/woofrc
- and ~/.woofrc can be INI-style config files containing the default
- port and the default count. The file in the home directory takes
- precedence. The compression methods are "off", "gz", "bz2" or "zip".
-
- Sample file:
-
- [main]
- port = 8008
- count = 2
- ip = 127.0.0.1
- compressed = gz
- """ % (name, name, name, name, defmaxdown, defport)
- if errmsg:
- print >>sys.stderr, errmsg
- print >>sys.stderr
- sys.exit (1)
-
-
-
-def main ():
- global cpid, compressed
-
- maxdown = 1
- port = 8080
- ip_addr = ''
-
- config = ConfigParser.ConfigParser()
- config.read (['/etc/woofrc', os.path.expanduser('~/.woofrc')])
-
- if config.has_option ('main', 'port'):
- port = config.getint ('main', 'port')
-
- if config.has_option ('main', 'count'):
- maxdown = config.getint ('main', 'count')
-
- if config.has_option ('main', 'ip'):
- ip_addr = config.get ('main', 'ip')
-
- if config.has_option ('main', 'compressed'):
- formats = { 'gz' : 'gz',
- 'true' : 'gz',
- 'bz' : 'bz2',
- 'bz2' : 'bz2',
- 'zip' : 'zip',
- 'off' : '',
- 'false' : '' }
- compressed = config.get ('main', 'compressed')
- compressed = formats.get (compressed, 'gz')
-
- defaultport = port
- defaultmaxdown = maxdown
-
- try:
- options, filenames = getopt.getopt (sys.argv[1:], "hszjZui:c:p:")
- except getopt.GetoptError, desc:
- usage (defaultport, defaultmaxdown, desc)
-
- for option, val in options:
- if option == '-c':
- try:
- maxdown = int (val)
- if maxdown <= 0:
- raise ValueError
- except ValueError:
- usage (defaultport, defaultmaxdown,
- "invalid download count: %r. "
- "Please specify an integer >= 0." % val)
-
- elif option == '-i':
- ip_addr = val
-
- elif option == '-p':
- try:
- port = int (val)
- except ValueError:
- usage (defaultport, defaultmaxdown,
- "invalid port number: %r. Please specify an integer" % val)
-
- elif option == '-s':
- filenames.append (__file__)
-
- elif option == '-h':
- usage (defaultport, defaultmaxdown)
-
- elif option == '-z':
- compressed = 'gz'
- elif option == '-j':
- compressed = 'bz2'
- elif option == '-Z':
- compressed = 'zip'
- elif option == '-u':
- compressed = ''
-
- else:
- usage (defaultport, defaultmaxdown, "Unknown option: %r" % option)
-
- if len (filenames) == 1:
- filename = os.path.abspath (filenames[0])
- else:
- usage (defaultport, defaultmaxdown,
- "Can only serve single files/directories.")
-
- if not os.path.exists (filename):
- usage (defaultport, defaultmaxdown,
- "%s: No such file or directory" % filenames[0])
-
- if not (os.path.isfile (filename) or os.path.isdir (filename)):
- usage (defaultport, defaultmaxdown,
- "%s: Neither file nor directory" % filenames[0])
-
- serve_files (filename, maxdown, ip_addr, port)
-
- # wait for child processes to terminate
- if cpid != 0:
- try:
- while 1:
- os.wait ()
- except OSError:
- pass
-
-
-
-if __name__=='__main__':
- try:
- main ()
- except KeyboardInterrupt:
- pass
-