#! /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() |
