added woof

1 files changed, 429 insertions, 0 deletions

diff --git a/woof b/woof
new file mode 100644
index 0000000..b1f509b
--- /dev/null
+++ b/woof
@@ -0,0 +1,429 @@
+#!/usr/bin/env python
+# -*- 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
+