Next Previous

ImageReceiveServer.py

#! /usr/bin/python

#   File:       ImageReceiveServer.py

#   Contains:   A basic HTTP server to accept HTTP PUTs and POSTs.

#   Written by: DTS

#   Copyright:  Copyright (c) 2009 Apple Inc. All Rights Reserved.

#   Disclaimer: IMPORTANT: This Apple software is supplied to you by Apple Inc.

#               ("Apple") in consideration of your agreement to the following

#               terms, and your use, installation, modification or

#               redistribution of this Apple software constitutes acceptance of

#               these terms.  If you do not agree with these terms, please do

#               not use, install, modify or redistribute this Apple software.

#               In consideration of your agreement to abide by the following

#               terms, and subject to these terms, Apple grants you a personal,

#               non-exclusive license, under Apple's copyrights in this

#               original Apple software (the "Apple Software"), to use,

#               reproduce, modify and redistribute the Apple Software, with or

#               without modifications, in source and/or binary forms; provided

#               that if you redistribute the Apple Software in its entirety and

#               without modifications, you must retain this notice and the

#               following text and disclaimers in all such redistributions of

#               the Apple Software. Neither the name, trademarks, service marks

#               or logos of Apple Inc. may be used to endorse or promote

#               products derived from the Apple Software without specific prior

#               written permission from Apple.  Except as expressly stated in

#               this notice, no other rights or licenses, express or implied,

#               are granted by Apple herein, including but not limited to any

#               patent rights that may be infringed by your derivative works or

#               by other works in which the Apple Software may be incorporated.

#               The Apple Software is provided by Apple on an "AS IS" basis.

#               APPLE MAKES NO WARRANTIES, EXPRESS OR IMPLIED, INCLUDING

#               WITHOUT LIMITATION THE IMPLIED WARRANTIES OF NON-INFRINGEMENT,

#               MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE, REGARDING

#               THE APPLE SOFTWARE OR ITS USE AND OPERATION ALONE OR IN

#               COMBINATION WITH YOUR PRODUCTS.

#               IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT,

#               INCIDENTAL OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED

#               TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

#               DATA, OR PROFITS; OR BUSINESS INTERRUPTION) ARISING IN ANY WAY

#               OUT OF THE USE, REPRODUCTION, MODIFICATION AND/OR DISTRIBUTION

#               OF THE APPLE SOFTWARE, HOWEVER CAUSED AND WHETHER UNDER THEORY

#               OF CONTRACT, TORT (INCLUDING NEGLIGENCE), STRICT LIABILITY OR

#               OTHERWISE, EVEN IF APPLE HAS BEEN ADVISED OF THE POSSIBILITY OF

#               SUCH DAMAGE.

import os

import sys

import BaseHTTPServer

import string

import email.feedparser

global gImageDirPath

class HTTPError (Exception):

    def __init__(self, statusCode, statusMessage=None):

        self.statusCode    = statusCode

        self.statusMessage = statusMessage

class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):

    def do_GET(self):

        self.send_error(404)

    def do_PUT(self):

        global gImageDirPath

        try:

            assert self.path[0] == "/"

            fileName = self.path[1:]

            if fileName.translate(string.maketrans("", ""), string.ascii_letters + string.digits + "_") != ".":

                # Non-alphanumeric characters must be limited to exactly

                # one dot, that is, the extension delimiter.

                raise HTTPError(403)

            # These Content-Type comparisons are too naive (they should be

            # case insensitive, and they should ignore training parameters),

            # but this is only a test server, not a real server.

            contentType = self.headers.get('Content-Type')

            if contentType == None or (contentType not in ["image/png", "image/jpeg", "image/gif"]):

                raise HTTPError(406)

            if contentType == "image/png" and not fileName.endswith(".png"):

                raise HTTPError(406)

            if contentType == "image/jpeg" and not fileName.endswith(".jpg"):

                raise HTTPError(406)

            contentLength = self.headers.get('Content-Length')

            if contentLength == None:

                raise HTTPError(411)

            try:

                fileLength = int(contentLength)

                assert fileLength > 0

            except:

                raise HTTPError(411)

            filePath = os.path.join(gImageDirPath, fileName)

            fileExists = os.path.exists(filePath)

            imageFile = open(filePath, "w")

            # Transfer the data in MB chunks to prevent us being run out of

            # memory with large files.

            bytesReadSoFar = 0

            while bytesReadSoFar != fileLength:

                bytesToRead = (fileLength - bytesReadSoFar)

                if bytesToRead > (1024 * 1024):

                    bytesToRead = (1024 * 1024)

                data = self.rfile.read(bytesToRead)

                bytesReadSoFar += len(data)

                imageFile.write(data)

                data = None

            imageFile.close()

            if fileExists:

                self.send_response(200)

            else:

                self.send_response(201)

            self.end_headers()

        except HTTPError, e:

            self.send_error(e.statusCode, e.statusMessage)

        except:

            self.send_error(500, "Internal Server Error")

    def do_POST(self):

        global gImageDirPath

        try:

            if self.path != "/cgi-bin/PostIt.py":

                raise HTTPError(403)

            # These Content-Type comparisons are too naive (they should be

            # case insensitive, and they should ignore training parameters),

            # but this is only a test server, not a real server.

            contentType = self.headers.get('Content-Type')

            if contentType == None or not contentType.startswith("multipart/form-data"):

                raise HTTPError(406)

            contentLength = self.headers.get('Content-Length')

            if contentLength == None:

                raise HTTPError(411)

            try:

                fileLength = int(contentLength)

                assert fileLength > 0

            except:

                raise HTTPError(411)

            # Set up a MIME parser and feed it the Content-Type header.

            p = email.feedparser.FeedParser()

            p.feed("Content-Type: %s\r\n" % self.headers.get('Content-Type'))

            # Transfer the data in MB chunks to prevent us being run out of

            # memory with large files.  Of course, a large file is probably

            # going to hoark and die because the MIME parser keeps everything

            # in memory.  Good thing this is only a trivial test server.

            bytesReadSoFar = 0

            while bytesReadSoFar != fileLength:

                bytesToRead = (fileLength - bytesReadSoFar)

                if bytesToRead > (1024 * 1024):

                    bytesToRead = (1024 * 1024)

                data = self.rfile.read(bytesToRead)

                bytesReadSoFar += len(data)

                p.feed(data)

                data = None

            rootMessage = p.close()

            # Check some basic facts about the message.

            if not rootMessage.is_multipart() or len(rootMessage.defects) != 0:

                raise HTTPError(400)

            # Look for the "fileContents" part.

            fileContentsMessage = None

            for message in rootMessage.get_payload():

                if message.get_param("name", header="Content-Disposition") == "fileContents":

                    fileContentsMessage = message

                    break

            if fileContentsMessage == None or fileContentsMessage.is_multipart():

                raise HTTPError(400)

            # Extract the file name and check that it's reasonable.

            fileName = fileContentsMessage.get_param("filename", header="Content-Disposition")

            if fileName == None:

                raise HTTPError(400)

            if fileName.translate(string.maketrans("", ""), string.ascii_letters + string.digits + "_") != ".":

                # Non-alphanumeric characters must be limited to exactly

                # one dot, that is, the extension delimiter.

                raise HTTPError(403)

            # Verify the content type.

            contentType = fileContentsMessage.get('Content-Type')

            if contentType == None or (contentType not in ["image/png", "image/jpeg", "image/gif"]):

                raise HTTPError(406)

            if contentType == "image/png" and not fileName.endswith(".png"):

                raise HTTPError(406)

            if contentType == "image/jpeg" and not fileName.endswith(".jpg"):

                raise HTTPError(406)

            # Create the file with the supplied name and write the content to it.

            filePath = os.path.join(gImageDirPath, fileName)

            fileExists = os.path.exists(filePath)

            imageFile = open(filePath, "w")

            imageFile.write(fileContentsMessage.get_payload())

            imageFile.close()

            if fileExists:

                self.send_response(200)

            else:

                self.send_response(201)

            self.end_headers()

        except HTTPError, e:

            self.send_error(e.statusCode, e.statusMessage)

        except:

            self.send_error(500, "Internal Server Error")

def main():

    global gImageDirPath

    gImageDirPath = os.path.join(os.curdir, "images")

    if not os.path.exists(gImageDirPath):

        os.mkdir(gImageDirPath)

    server = BaseHTTPServer.HTTPServer(('', 9000), MyHandler)

    print "Hello Cruel World!"

    try:

        server.serve_forever()

    except KeyboardInterrupt:

        print ""

if __name__ == "__main__":

    main()

Next Previous