1 files changed, 173 insertions, 0 deletions
diff --git a/flup/client/scgi_app.py b/flup/client/scgi_app.py
new file mode 100644
index 0000000..dc08743
--- /dev/null
+++ b/flup/client/scgi_app.py
@@ -0,0 +1,173 @@
+# Copyright (c) 2006 Allan Saddi <allan@saddi.com>
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+# 1. Redistributions of source code must retain the above copyright
+#    notice, this list of conditions and the following disclaimer.
+# 2. Redistributions in binary form must reproduce the above copyright
+#    notice, this list of conditions and the following disclaimer in the
+#    documentation and/or other materials provided with the distribution.
+#
+# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
+# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+# SUCH DAMAGE.
+#
+# $Id$
+
+__author__ = 'Allan Saddi <allan@saddi.com>'
+__version__ = '$Revision$'
+
+import select
+import struct
+import socket
+import errno
+
+__all__ = ['SCGIApp']
+
+def encodeNetstring(s):
+    return ''.join([str(len(s)), ':', s, ','])
+
+class SCGIApp(object):
+    def __init__(self, connect=None, host=None, port=None,
+                 filterEnviron=True):
+        if host is not None:
+            assert port is not None
+            connect=(host, port)
+
+        assert connect is not None
+        self._connect = connect
+
+        self._filterEnviron = filterEnviron
+        
+    def __call__(self, environ, start_response):
+        sock = self._getConnection()
+
+        outfile = sock.makefile('w')
+        infile = sock.makefile('r')
+
+        sock.close()
+
+        # Filter WSGI environ and send as request headers
+        if self._filterEnviron:
+            headers = self._defaultFilterEnviron(environ)
+        else:
+            headers = self._lightFilterEnviron(environ)
+        # TODO: Anything not from environ that needs to be sent also?
+
+        content_length = int(environ.get('CONTENT_LENGTH') or 0)
+        if headers.has_key('CONTENT_LENGTH'):
+            del headers['CONTENT_LENGTH']
+            
+        headers_out = ['CONTENT_LENGTH', str(content_length), 'SCGI', '1']
+        for k,v in headers.items():
+            headers_out.append(k)
+            headers_out.append(v)
+        headers_out.append('') # For trailing NUL
+        outfile.write(encodeNetstring('\x00'.join(headers_out)))
+
+        # Transfer wsgi.input to outfile
+        while True:
+            chunk_size = min(content_length, 4096)
+            s = environ['wsgi.input'].read(chunk_size)
+            content_length -= len(s)
+            outfile.write(s)
+
+            if not s: break
+
+        outfile.close()
+        
+        # Read result from SCGI server
+        result = []
+        while True:
+            buf = infile.read(4096)
+            if not buf: break
+
+            result.append(buf)
+
+        infile.close()
+        
+        result = ''.join(result)
+
+        # Parse response headers
+        status = '200 OK'
+        headers = []
+        pos = 0
+        while True:
+            eolpos = result.find('\n', pos)
+            if eolpos < 0: break
+            line = result[pos:eolpos-1]
+            pos = eolpos + 1
+
+            # strip in case of CR. NB: This will also strip other
+            # whitespace...
+            line = line.strip()
+            
+            # Empty line signifies end of headers
+            if not line: break
+
+            # TODO: Better error handling
+            header, value = line.split(':', 1)
+            header = header.strip().lower()
+            value = value.strip()
+
+            if header == 'status':
+                # Special handling of Status header
+                status = value
+                if status.find(' ') < 0:
+                    # Append a dummy reason phrase if one was not provided
+                    status += ' SCGIApp'
+            else:
+                headers.append((header, value))
+
+        result = result[pos:]
+
+        # Set WSGI status, headers, and return result.
+        start_response(status, headers)
+        return [result]
+
+    def _getConnection(self):
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        sock.connect(self._connect)
+        return sock
+    
+    _environPrefixes = ['SERVER_', 'HTTP_', 'REQUEST_', 'REMOTE_', 'PATH_',
+                        'CONTENT_']
+    _environCopies = ['SCRIPT_NAME', 'QUERY_STRING', 'AUTH_TYPE']
+    _environRenames = {}
+
+    def _defaultFilterEnviron(self, environ):
+        result = {}
+        for n in environ.keys():
+            for p in self._environPrefixes:
+                if n.startswith(p):
+                    result[n] = environ[n]
+            if n in self._environCopies:
+                result[n] = environ[n]
+            if n in self._environRenames:
+                result[self._environRenames[n]] = environ[n]
+                
+        return result
+
+    def _lightFilterEnviron(self, environ):
+        result = {}
+        for n in environ.keys():
+            if n.upper() == n:
+                result[n] = environ[n]
+        return result
+
+if __name__ == '__main__':
+    from flup.server.ajp import WSGIServer
+    app = SCGIApp(connect=('localhost', 4000))
+    #import paste.lint
+    #app = paste.lint.middleware(app)
+    WSGIServer(app).run()