NanoHTTPD.java revision fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1
1269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawkepackage fi.iki.elonen; 2269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 3fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawkeimport java.io.*; 4b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.InetAddress; 5b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.InetSocketAddress; 6b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.ServerSocket; 7b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.Socket; 8b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.SocketException; 913736e18ec88e3df74d055e061fb324e04778ad6Paul Hawkeimport java.net.SocketTimeoutException; 10b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.URLDecoder; 11dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawkeimport java.nio.ByteBuffer; 12dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawkeimport java.nio.channels.FileChannel; 137b88819e82b89ac3476ce060903468076a73de8fPaul Hawkeimport java.text.SimpleDateFormat; 14b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.ArrayList; 15b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Calendar; 16b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Date; 17b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.HashMap; 18b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.HashSet; 19b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Iterator; 20b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.List; 21b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Locale; 22b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Map; 23b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Set; 24b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.StringTokenizer; 25b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.TimeZone; 26269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 27269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke/** 285b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * A simple, tiny, nicely embeddable HTTP server in Java 29d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 30d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 31b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke * NanoHTTPD 3284dcf74395d1817c4a2a7861bd5d3787daa6700aPaul Hawke * <p></p>Copyright (c) 2012-2013 by Paul S. Hawke, 2001,2005-2013 by Jarno Elonen, 2010 by Konstantinos Togias</p> 33d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 34d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 35269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <b>Features + limitations: </b> 36269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <ul> 37d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 38269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Only one Java file</li> 3901ecef8887ed32be34ef7442a656a0b3330ecb8dPaul Hawke * <li>Java 5 compatible</li> 40269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Released as open source, Modified BSD licence</li> 41269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>No fixed config files, logging, authorization etc. (Implement yourself if you need them.)</li> 42269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Supports parameter parsing of GET and POST methods (+ rudimentary PUT support in 1.25)</li> 43269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Supports both dynamic content and file serving</li> 44269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Supports file upload (since version 1.2, 2010)</li> 45269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Supports partial content (streaming)</li> 46269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Supports ETags</li> 47269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Never caches anything</li> 48269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Doesn't limit bandwidth, request time or simultaneous connections</li> 49269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Default code serves files and shows all HTTP parameters and headers</li> 50269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server supports directory listing, index.html and index.htm</li> 51269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server supports partial content (streaming)</li> 52269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server supports ETags</li> 53269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server does the 301 redirection trick for directories without '/'</li> 54269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server supports simple skipping for files (continue download)</li> 55269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server serves also very long files without memory overhead</li> 56269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Contains a built-in list of most common mime types</li> 57269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>All header names are converted lowercase so they don't vary between browsers/clients</li> 58d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 59269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * </ul> 60d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 61d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 6201ecef8887ed32be34ef7442a656a0b3330ecb8dPaul Hawke * <b>How to use: </b> 63269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <ul> 64d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 6501ecef8887ed32be34ef7442a656a0b3330ecb8dPaul Hawke * <li>Subclass and implement serve() and embed to your own program</li> 66d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 67269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * </ul> 68d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 6950b70cd2a7250fd85394edc863b219a3bfb738e6Paul Hawke * See the separate "LICENSE.md" file for the distribution license (Modified BSD licence) 70269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 71d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawkepublic abstract class NanoHTTPD { 72269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 73293e04675f3d4f427c125a26e831d70d5011c79bhflicka * Maximum time to wait on Socket.getInputStream().read() (in milliseconds) 74293e04675f3d4f427c125a26e831d70d5011c79bhflicka * This is required as the Keep-Alive HTTP connections would otherwise 75293e04675f3d4f427c125a26e831d70d5011c79bhflicka * block the socket reading thread forever (or as long the browser is open). 76293e04675f3d4f427c125a26e831d70d5011c79bhflicka */ 77293e04675f3d4f427c125a26e831d70d5011c79bhflicka public static final int SOCKET_READ_TIMEOUT = 5000; 78293e04675f3d4f427c125a26e831d70d5011c79bhflicka /** 795b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Common mime type for dynamic content: plain text 80269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 81269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke public static final String MIME_PLAINTEXT = "text/plain"; 825b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 835b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Common mime type for dynamic content: html 845b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 85269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke public static final String MIME_HTML = "text/html"; 865b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 878dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Pseudo-Parameter to use to store the actual query string in the parameters map for later re-processing. 888dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 898dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private static final String QUERY_STRING_PARAMETER = "NanoHttpd.QUERY_STRING"; 90f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke private final String hostname; 917b88819e82b89ac3476ce060903468076a73de8fPaul Hawke private final int myPort; 927b88819e82b89ac3476ce060903468076a73de8fPaul Hawke private ServerSocket myServerSocket; 930a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander private Set<Socket> openConnections = new HashSet<Socket>(); 947b88819e82b89ac3476ce060903468076a73de8fPaul Hawke private Thread myThread; 955b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 968dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Pluggable strategy for asynchronously executing requests. 975b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 988dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private AsyncRunner asyncRunner; 998dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 1008dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Pluggable strategy for creating and cleaning up temporary files. 1018dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 1028dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private TempFileManagerFactory tempFileManagerFactory; 103269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 104269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 105d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * Constructs an HTTP server on given port. 106269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 107df674ca7f11cee6336b66271c50476f389e0ff4bPaul Hawke public NanoHTTPD(int port) { 108f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke this(null, port); 109f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke } 110f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke 1115b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 1126e419652da8eb07f46c768c083c46c206e0e2762Martin M Reed * Constructs an HTTP server on given hostname and port. 1135b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 114f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke public NanoHTTPD(String hostname, int port) { 115f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke this.hostname = hostname; 1167b88819e82b89ac3476ce060903468076a73de8fPaul Hawke this.myPort = port; 1175b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke setTempFileManagerFactory(new DefaultTempFileManagerFactory()); 1185b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke setAsyncRunner(new DefaultAsyncRunner()); 1193f2aa2e2e0e02fe7e0151eb64d31265b578885ecPaul Hawke } 1203f2aa2e2e0e02fe7e0151eb64d31265b578885ecPaul Hawke 1218dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private static final void safeClose(Closeable closeable) { 1228dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (closeable != null) { 1238dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke try { 1248dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke closeable.close(); 1258dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } catch (IOException e) { 1268dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 1278dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 1288dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 1298dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 130c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke private static final void safeClose(Socket closeable) { 131c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke if (closeable != null) { 132c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke try { 133c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke closeable.close(); 134c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } catch (IOException e) { 135c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 136c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 137c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 138c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke 139c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke private static final void safeClose(ServerSocket closeable) { 140c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke if (closeable != null) { 141c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke try { 142c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke closeable.close(); 143c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } catch (IOException e) { 144c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 145c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 146c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 147c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke 1483f2aa2e2e0e02fe7e0151eb64d31265b578885ecPaul Hawke /** 1495b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Start the server. 1508dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 1515b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @throws IOException if the socket is in use. 1523f2aa2e2e0e02fe7e0151eb64d31265b578885ecPaul Hawke */ 1537b88819e82b89ac3476ce060903468076a73de8fPaul Hawke public void start() throws IOException { 154f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke myServerSocket = new ServerSocket(); 155f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke myServerSocket.bind((hostname != null) ? new InetSocketAddress(hostname, myPort) : new InetSocketAddress(myPort)); 156f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke 157269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke myThread = new Thread(new Runnable() { 158269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke @Override 159269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke public void run() { 1606200253df7b934c807eb3436b51a3b039406457dPaul Hawke do { 1616200253df7b934c807eb3436b51a3b039406457dPaul Hawke try { 1620ad8e8a948f2a3d347ff633eccc875363fdbbb19Björn Heinrichs final Socket finalAccept = myServerSocket.accept(); 1630a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander registerConnection(finalAccept); 164293e04675f3d4f427c125a26e831d70d5011c79bhflicka finalAccept.setSoTimeout(SOCKET_READ_TIMEOUT); 1656e419652da8eb07f46c768c083c46c206e0e2762Martin M Reed final InputStream inputStream = finalAccept.getInputStream(); 166864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs asyncRunner.exec(new Runnable() { 167864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs @Override 168864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs public void run() { 169864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs OutputStream outputStream = null; 170864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs try { 171864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs outputStream = finalAccept.getOutputStream(); 172864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs TempFileManager tempFileManager = tempFileManagerFactory.create(); 173864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs HTTPSession session = new HTTPSession(tempFileManager, inputStream, outputStream, finalAccept.getInetAddress()); 174864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs while (!finalAccept.isClosed()) { 175864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs session.execute(); 1766e419652da8eb07f46c768c083c46c206e0e2762Martin M Reed } 177864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs } catch (Exception e) { 178864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs // When the socket is closed by the client, we throw our own SocketException 179864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs // to break the "keep alive" loop above. 180864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs if (!(e instanceof SocketException && "NanoHttpd Shutdown".equals(e.getMessage()))) { 181864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs e.printStackTrace(); 182864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs } 183864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs } finally { 184864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs safeClose(outputStream); 185864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs safeClose(inputStream); 186864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs safeClose(finalAccept); 187864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs unRegisterConnection(finalAccept); 1886e419652da8eb07f46c768c083c46c206e0e2762Martin M Reed } 189864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs } 190864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs }); 1916200253df7b934c807eb3436b51a3b039406457dPaul Hawke } catch (IOException e) { 1926200253df7b934c807eb3436b51a3b039406457dPaul Hawke } 1932dc83f968760f835a138f0320db9eca60f8e0ecaBjörn Heinrichs } while (!myServerSocket.isClosed()); 194269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 195269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke }); 196269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke myThread.setDaemon(true); 1971ca771092a5a39cfea9c4cf843ba7f7c90a1dbccPaul Hawke myThread.setName("NanoHttpd Main Listener"); 198269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke myThread.start(); 199269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 200d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 201269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 2025b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Stop the server. 203269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 204269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke public void stop() { 205269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke try { 2069cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed safeClose(myServerSocket); 2070a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander closeAllConnections(); 208fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke if (myThread != null) { 209fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke myThread.join(); 210fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke } 2115b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } catch (Exception e) { 2127b88819e82b89ac3476ce060903468076a73de8fPaul Hawke e.printStackTrace(); 213269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 214269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 215269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 2160a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander /** 2170a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * Registers that a new connection has been set up. 2180a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * 219fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke * @param socket the {@link Socket} for the connection. 2200a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander */ 2210a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander public synchronized void registerConnection(Socket socket) { 2220a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander openConnections.add(socket); 2230a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander } 2240a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander 2250a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander /** 2260a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * Registers that a connection has been closed 2270a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * 2280a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * @param socket 2290a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * the {@link Socket} for the connection. 2300a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander */ 2310a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander public synchronized void unRegisterConnection(Socket socket) { 2320a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander openConnections.remove(socket); 2330a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander } 2340a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander 2350a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander /** 2360a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * Forcibly closes all connections that are open. 2370a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander */ 2380a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander public synchronized void closeAllConnections() { 2390a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander for (Socket socket : openConnections) { 2400a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander safeClose(socket); 2410a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander } 2420a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander } 2438dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 2448dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public final int getListeningPort() { 2458dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return myServerSocket == null ? -1 : myServerSocket.getLocalPort(); 2468dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 2478dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 2488dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public final boolean wasStarted() { 2498dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return myServerSocket != null && myThread != null; 2508dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 2518dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 2528dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public final boolean isAlive() { 2538dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return wasStarted() && !myServerSocket.isClosed() && myThread.isAlive(); 2548dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 2558dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 256a6619fe65dd3a997960724faf7c51685de793f91Paul Hawke /** 257d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * Override this to customize the server. 258d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 259d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 260d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * (By default, this delegates to serveFile() and allows directory listing.) 2619788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * 2628dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param uri Percent-decoded URI without parameters, for example "/index.cgi" 2638dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param method "GET", "POST" etc. 2648dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param parms Parsed, percent decoded parameters from URI and, in case of POST, data. 2658af508ee58e0abca6dbb6c30532410a1dd99866bMartin M Reed * @param headers Header entries, percent decoded 266d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * @return HTTP response, see class Response for details 267269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 2688dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke @Deprecated 2698dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public Response serve(String uri, Method method, Map<String, String> headers, Map<String, String> parms, 2708dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke Map<String, String> files) { 2718dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return new Response(Response.Status.NOT_FOUND, MIME_PLAINTEXT, "Not Found"); 2728dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 2739788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 274f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke /** 27593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * Override this to customize the server. 27693441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * <p/> 27793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * <p/> 27893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * (By default, this delegates to serveFile() and allows directory listing.) 27993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * 28093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * @param session The HTTP session 28193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * @return HTTP response, see class Response for details 28293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed */ 283249def0afcd823cd21315973bf197f2bef4a089fSleekWeasel public Response serve(IHTTPSession session) { 28493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed Map<String, String> files = new HashMap<String, String>(); 2858dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke Method method = session.getMethod(); 2868dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (Method.PUT.equals(method) || Method.POST.equals(method)) { 2878dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke try { 2888dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke session.parseBody(files); 2898dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } catch (IOException ioe) { 2908dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return new Response(Response.Status.INTERNAL_ERROR, MIME_PLAINTEXT, "SERVER INTERNAL ERROR: IOException: " + ioe.getMessage()); 2918dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } catch (ResponseException re) { 2928dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return new Response(re.getStatus(), MIME_PLAINTEXT, re.getMessage()); 2938dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 29493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 29593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 296fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke Map<String, String> parms = session.getParms(); 297fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke parms.put(QUERY_STRING_PARAMETER, session.getQueryParameterString()); 298fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke return serve(session.getUri(), method, session.getHeaders(), parms, files); 29993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 30093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 30193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed /** 3025b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Decode percent encoded <code>String</code> values. 3038dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 3045b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @param str the percent encoded <code>String</code> 3055b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @return expanded form of the input, for example "foo%20bar" becomes "foo bar" 306f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke */ 307b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke protected String decodePercent(String str) { 30849f777c1de22de7f0918439582d79d98d0b6e70aPaul Hawke String decoded = null; 30949f777c1de22de7f0918439582d79d98d0b6e70aPaul Hawke try { 31049f777c1de22de7f0918439582d79d98d0b6e70aPaul Hawke decoded = URLDecoder.decode(str, "UTF8"); 31149f777c1de22de7f0918439582d79d98d0b6e70aPaul Hawke } catch (UnsupportedEncodingException ignored) { 312f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 31349f777c1de22de7f0918439582d79d98d0b6e70aPaul Hawke return decoded; 314f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 315f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke 3165b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 3175b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Decode parameters from a URL, handing the case where a single parameter name might have been 3185b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * supplied several times, by return lists of values. In general these lists will contain a single 3195b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * element. 3205b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * 3215b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @param parms original <b>NanoHttpd</b> parameters values, as passed to the <code>serve()</code> method. 3225b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @return a map of <code>String</code> (parameter name) to <code>List<String></code> (a list of the values supplied). 3235b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 324f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke protected Map<String, List<String>> decodeParameters(Map<String, String> parms) { 325f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke return this.decodeParameters(parms.get(QUERY_STRING_PARAMETER)); 326f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 327f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke 3285b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 3295b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Decode parameters from a URL, handing the case where a single parameter name might have been 3305b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * supplied several times, by return lists of values. In general these lists will contain a single 3315b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * element. 3325b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * 3335b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @param queryString a query string pulled from the URL. 3345b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @return a map of <code>String</code> (parameter name) to <code>List<String></code> (a list of the values supplied). 3355b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 336f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke protected Map<String, List<String>> decodeParameters(String queryString) { 337f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke Map<String, List<String>> parms = new HashMap<String, List<String>>(); 338f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke if (queryString != null) { 339f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke StringTokenizer st = new StringTokenizer(queryString, "&"); 340f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke while (st.hasMoreTokens()) { 341f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke String e = st.nextToken(); 342f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke int sep = e.indexOf('='); 343b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke String propertyName = (sep >= 0) ? decodePercent(e.substring(0, sep)).trim() : decodePercent(e).trim(); 344b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke if (!parms.containsKey(propertyName)) { 345b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke parms.put(propertyName, new ArrayList<String>()); 346b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke } 347b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke String propertyValue = (sep >= 0) ? decodePercent(e.substring(sep + 1)) : null; 348b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke if (propertyValue != null) { 349b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke parms.get(propertyName).add(propertyValue); 350f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 351f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 352f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 353f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke return parms; 354f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 355f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke 3565b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // ------------------------------------------------------------------------------- // 3575b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // 3585b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // Threading Strategy. 3595b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // 3605b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // ------------------------------------------------------------------------------- // 3615b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 3625b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 3635b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Pluggable strategy for asynchronously executing requests. 3648dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 3655b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @param asyncRunner new strategy for handling threads. 3665b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 3675b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public void setAsyncRunner(AsyncRunner asyncRunner) { 3685b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke this.asyncRunner = asyncRunner; 3695b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 3705b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 3715b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // ------------------------------------------------------------------------------- // 3725b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // 3735b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // Temp file handling strategy. 3745b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // 3755b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // ------------------------------------------------------------------------------- // 3765b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 3775b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 3785b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Pluggable strategy for creating and cleaning up temporary files. 3798dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 3805b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @param tempFileManagerFactory new strategy for handling temp files. 3815b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 3825b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public void setTempFileManagerFactory(TempFileManagerFactory tempFileManagerFactory) { 3835b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke this.tempFileManagerFactory = tempFileManagerFactory; 3845b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 3855b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 3865b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 3878dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * HTTP Request methods, with the ability to decode a <code>String</code> back to its enum value. 3888dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 3898dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public enum Method { 390ef698c852a24dfa42ab25054291286feb01ec871Paul Hawke GET, PUT, POST, DELETE, HEAD, OPTIONS; 3918dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 3928dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke static Method lookup(String method) { 3938dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke for (Method m : Method.values()) { 3948dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (m.toString().equalsIgnoreCase(method)) { 3958dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return m; 3968dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 3978dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 3988dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return null; 3998dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 4008dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 4018dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 4028dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 4038dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Pluggable strategy for asynchronously executing requests. 4048dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 4058dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public interface AsyncRunner { 4068dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke void exec(Runnable code); 4078dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 4088dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 4098dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 4105b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Factory to create temp file managers. 4115b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 4126200253df7b934c807eb3436b51a3b039406457dPaul Hawke public interface TempFileManagerFactory { 4136200253df7b934c807eb3436b51a3b039406457dPaul Hawke TempFileManager create(); 4146200253df7b934c807eb3436b51a3b039406457dPaul Hawke } 4156200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4168dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke // ------------------------------------------------------------------------------- // 4178dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 4185b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 4195b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Temp file manager. 4208dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p/> 4215b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * <p>Temp file managers are created 1-to-1 with incoming requests, to create and cleanup 4225b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * temporary files created as a result of handling the request.</p> 4235b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 4246200253df7b934c807eb3436b51a3b039406457dPaul Hawke public interface TempFileManager { 4256200253df7b934c807eb3436b51a3b039406457dPaul Hawke TempFile createTempFile() throws Exception; 4266200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4276200253df7b934c807eb3436b51a3b039406457dPaul Hawke void clear(); 4286200253df7b934c807eb3436b51a3b039406457dPaul Hawke } 4296200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4305b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 4315b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * A temp file. 4328dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p/> 4335b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * <p>Temp files are responsible for managing the actual temporary storage and cleaning 4345b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * themselves up when no longer needed.</p> 4355b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 4366200253df7b934c807eb3436b51a3b039406457dPaul Hawke public interface TempFile { 4376200253df7b934c807eb3436b51a3b039406457dPaul Hawke OutputStream open() throws Exception; 4386200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4396200253df7b934c807eb3436b51a3b039406457dPaul Hawke void delete() throws Exception; 4406200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4416200253df7b934c807eb3436b51a3b039406457dPaul Hawke String getName(); 4426200253df7b934c807eb3436b51a3b039406457dPaul Hawke } 4436200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4449788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 4458dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Default threading strategy for NanoHttpd. 4468dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p/> 4478dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p>By default, the server spawns a new Thread for every incoming request. These are set 4488dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * to <i>daemon</i> status, and named according to the request number. The name is 4498dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * useful when profiling the application.</p> 4505b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 4518dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public static class DefaultAsyncRunner implements AsyncRunner { 4528dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private long requestCount; 4538dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 4545b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 4558dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public void exec(Runnable code) { 4568dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke ++requestCount; 4578dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke Thread t = new Thread(code); 4588dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke t.setDaemon(true); 4598dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke t.setName("NanoHttpd Request Processor (#" + requestCount + ")"); 4608dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke t.start(); 4615b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 4625b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 4635b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 4645b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 4655b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Default strategy for creating and cleaning up temporary files. 4668dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p/> 4675b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * <p></p>This class stores its files in the standard location (that is, 4685b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * wherever <code>java.io.tmpdir</code> points to). Files are added 4695b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * to an internal list, and deleted when no longer needed (that is, 4705b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * when <code>clear()</code> is invoked at the end of processing a 4715b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * request).</p> 4725b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 4735b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public static class DefaultTempFileManager implements TempFileManager { 4745b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke private final String tmpdir; 4755b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke private final List<TempFile> tempFiles; 4765b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 4775b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public DefaultTempFileManager() { 4785b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke tmpdir = System.getProperty("java.io.tmpdir"); 4795b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke tempFiles = new ArrayList<TempFile>(); 4805b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 4815b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 4825b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 4835b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public TempFile createTempFile() throws Exception { 4845b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke DefaultTempFile tempFile = new DefaultTempFile(tmpdir); 4855b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke tempFiles.add(tempFile); 4865b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke return tempFile; 4875b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 4885b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 4895b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 4905b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public void clear() { 4915b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke for (TempFile file : tempFiles) { 4925b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke try { 4935b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke file.delete(); 4945b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } catch (Exception ignored) { 4955b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 4965b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 4975b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke tempFiles.clear(); 4985b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 4995b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5005b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5015b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 5025b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Default strategy for creating and cleaning up temporary files. 5038dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p/> 5045b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * <p></p></[>By default, files are created by <code>File.createTempFile()</code> in 5055b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * the directory specified.</p> 5065b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 5075b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public static class DefaultTempFile implements TempFile { 5085b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke private File file; 5095b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke private OutputStream fstream; 5105b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5115b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public DefaultTempFile(String tempdir) throws IOException { 5125b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke file = File.createTempFile("NanoHTTPD-", "", new File(tempdir)); 5135b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke fstream = new FileOutputStream(file); 5145b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5155b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5165b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 5175b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public OutputStream open() throws Exception { 5185b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke return fstream; 5195b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5205b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5215b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 5225b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public void delete() throws Exception { 5239cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed safeClose(fstream); 5245b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke file.delete(); 5255b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5265b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5275b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 5285b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public String getName() { 5295b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke return file.getAbsolutePath(); 5305b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5315b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5325b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5335b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 5349788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * HTTP response. Return one of these from serve(). 5359788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 5369788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public static class Response { 5379788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5389788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * HTTP status code after processing, e.g. "200 OK", HTTP_OK 5399788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 54067e24e482b132ff0a8618f96b43097c9ce5f2776yankee private IStatus status; 5419788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5429788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * MIME type of content, e.g. "text/html" 5439788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 544c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke private String mimeType; 5459788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5469788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Data of the response, may be null. 5479788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 5489d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private InputStream data; 5499788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5509788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Headers for the HTTP response. Use addHeader() to add lines. 5519788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 552c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke private Map<String, String> header = new HashMap<String, String>(); 5539d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke /** 5549d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke * The request method that spawned this response. 5559d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke */ 5569d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private Method requestMethod; 557517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel /** 558517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel * Use chunkedTransfer 559517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel */ 560517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel private boolean chunkedTransfer; 5618dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 5629788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5639788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Default constructor: response = HTTP_OK, mime = MIME_HTML and your supplied message 5649788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 5659788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public Response(String msg) { 5669788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this(Status.OK, MIME_HTML, msg); 5679788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 5689788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 5699788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5709788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Basic constructor. 5719788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 57267e24e482b132ff0a8618f96b43097c9ce5f2776yankee public Response(IStatus status, String mimeType, InputStream data) { 5739788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.status = status; 5749788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.mimeType = mimeType; 5759788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.data = data; 5769788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 5779788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 5789788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5799788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Convenience method that makes an InputStream out of given text. 5809788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 58167e24e482b132ff0a8618f96b43097c9ce5f2776yankee public Response(IStatus status, String mimeType, String txt) { 5829788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.status = status; 5839788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.mimeType = mimeType; 5849788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke try { 5857162954f3d9927a455a9f6e806c8cecf5327dcc6Paul Hawke this.data = txt != null ? new ByteArrayInputStream(txt.getBytes("UTF-8")) : null; 5869788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } catch (java.io.UnsupportedEncodingException uee) { 5879788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke uee.printStackTrace(); 5889788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 5899788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 5909788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 5919788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5929788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Adds given line to the header. 5939788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 5949788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public void addHeader(String name, String value) { 5959788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke header.put(name, value); 5969788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 597fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke 59844f52c7b241f235d887ab57bf0ffd97a8ce0f5c4yankee public String getHeader(String name) { 59944f52c7b241f235d887ab57bf0ffd97a8ce0f5c4yankee return header.get(name); 60044f52c7b241f235d887ab57bf0ffd97a8ce0f5c4yankee } 6019788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 6029788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 6037f0727787957c2f093412c01d165846842ec2425Paul Hawke * Sends given response to the socket. 6047f0727787957c2f093412c01d165846842ec2425Paul Hawke */ 6056d3737c76afdb1c991bf8d96e3dd2cc622435f09Niko Fink protected void send(OutputStream outputStream) { 6067f0727787957c2f093412c01d165846842ec2425Paul Hawke String mime = mimeType; 6077f0727787957c2f093412c01d165846842ec2425Paul Hawke SimpleDateFormat gmtFrmt = new SimpleDateFormat("E, d MMM yyyy HH:mm:ss 'GMT'", Locale.US); 6087f0727787957c2f093412c01d165846842ec2425Paul Hawke gmtFrmt.setTimeZone(TimeZone.getTimeZone("GMT")); 6097f0727787957c2f093412c01d165846842ec2425Paul Hawke 6107f0727787957c2f093412c01d165846842ec2425Paul Hawke try { 6117f0727787957c2f093412c01d165846842ec2425Paul Hawke if (status == null) { 6127f0727787957c2f093412c01d165846842ec2425Paul Hawke throw new Error("sendResponse(): Status can't be null."); 6137f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6147f0727787957c2f093412c01d165846842ec2425Paul Hawke PrintWriter pw = new PrintWriter(outputStream); 6156b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke pw.print("HTTP/1.1 " + status.getDescription() + " \r\n"); 6167f0727787957c2f093412c01d165846842ec2425Paul Hawke 6177f0727787957c2f093412c01d165846842ec2425Paul Hawke if (mime != null) { 6187f0727787957c2f093412c01d165846842ec2425Paul Hawke pw.print("Content-Type: " + mime + "\r\n"); 6197f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6207f0727787957c2f093412c01d165846842ec2425Paul Hawke 6217f0727787957c2f093412c01d165846842ec2425Paul Hawke if (header == null || header.get("Date") == null) { 6227f0727787957c2f093412c01d165846842ec2425Paul Hawke pw.print("Date: " + gmtFrmt.format(new Date()) + "\r\n"); 6237f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6247f0727787957c2f093412c01d165846842ec2425Paul Hawke 6257f0727787957c2f093412c01d165846842ec2425Paul Hawke if (header != null) { 6267f0727787957c2f093412c01d165846842ec2425Paul Hawke for (String key : header.keySet()) { 6277f0727787957c2f093412c01d165846842ec2425Paul Hawke String value = header.get(key); 6287f0727787957c2f093412c01d165846842ec2425Paul Hawke pw.print(key + ": " + value + "\r\n"); 6297f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6307f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6317f0727787957c2f093412c01d165846842ec2425Paul Hawke 632c6c080a6f4377fb18fc869111714e9514c36331aPaul Hawke sendConnectionHeaderIfNotAlreadyPresent(pw, header); 633b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke 634517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel if (requestMethod != Method.HEAD && chunkedTransfer) { 635fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel sendAsChunked(outputStream, pw); 636fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } else { 637fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke int pending = data != null ? data.available() : 0; 638fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke sendContentLengthHeaderIfNotAlreadyPresent(pw, header, pending); 639fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke pw.print("\r\n"); 640fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke pw.flush(); 641fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke sendAsFixedLength(outputStream, pending); 6427f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6437f0727787957c2f093412c01d165846842ec2425Paul Hawke outputStream.flush(); 6449cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed safeClose(data); 6457f0727787957c2f093412c01d165846842ec2425Paul Hawke } catch (IOException ioe) { 6467f0727787957c2f093412c01d165846842ec2425Paul Hawke // Couldn't write? No can do. 6477f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6487f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6497f0727787957c2f093412c01d165846842ec2425Paul Hawke 650fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke protected void sendContentLengthHeaderIfNotAlreadyPresent(PrintWriter pw, Map<String, String> header, int size) { 651fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke if (!headerAlreadySent(header, "content-length")) { 652fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke pw.print("Content-Length: "+ size +"\r\n"); 653c6c080a6f4377fb18fc869111714e9514c36331aPaul Hawke } 654fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke } 655fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke 656fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke protected void sendConnectionHeaderIfNotAlreadyPresent(PrintWriter pw, Map<String, String> header) { 657fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke if (!headerAlreadySent(header, "connection")) { 658c6c080a6f4377fb18fc869111714e9514c36331aPaul Hawke pw.print("Connection: keep-alive\r\n"); 659c6c080a6f4377fb18fc869111714e9514c36331aPaul Hawke } 660c6c080a6f4377fb18fc869111714e9514c36331aPaul Hawke } 661c6c080a6f4377fb18fc869111714e9514c36331aPaul Hawke 662fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke private boolean headerAlreadySent(Map<String, String> header, String name) { 663fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke boolean alreadySent = false; 664fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke for (String headerName : header.keySet()) { 665fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke alreadySent |= headerName.equalsIgnoreCase(name); 666fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke } 667fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke return alreadySent; 668fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke } 669fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke 670fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel private void sendAsChunked(OutputStream outputStream, PrintWriter pw) throws IOException { 671fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pw.print("Transfer-Encoding: chunked\r\n"); 672fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pw.print("\r\n"); 673fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pw.flush(); 674fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel int BUFFER_SIZE = 16 * 1024; 675fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel byte[] CRLF = "\r\n".getBytes(); 676fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel byte[] buff = new byte[BUFFER_SIZE]; 677fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel int read; 678fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel while ((read = data.read(buff)) > 0) { 679fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel outputStream.write(String.format("%x\r\n", read).getBytes()); 680fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel outputStream.write(buff, 0, read); 681fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel outputStream.write(CRLF); 682fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 683fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel outputStream.write(String.format("0\r\n\r\n").getBytes()); 684fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 685fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel 686fe8be79fa8d8b131d5d39e77ae0baabd177d5ec1Paul Hawke private void sendAsFixedLength(OutputStream outputStream, int pending) throws IOException { 687fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel if (requestMethod != Method.HEAD && data != null) { 688fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel int BUFFER_SIZE = 16 * 1024; 689fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel byte[] buff = new byte[BUFFER_SIZE]; 690fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel while (pending > 0) { 691fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel int read = data.read(buff, 0, ((pending > BUFFER_SIZE) ? BUFFER_SIZE : pending)); 692fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel if (read <= 0) { 693fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel break; 694fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 695fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel outputStream.write(buff, 0, read); 696fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pending -= read; 697fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 698fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 699fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 700fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel 70167e24e482b132ff0a8618f96b43097c9ce5f2776yankee public IStatus getStatus() { 702c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke return status; 703c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 704c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 705c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public void setStatus(Status status) { 706c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke this.status = status; 707c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 708c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 709c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public String getMimeType() { 710c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke return mimeType; 711c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 712c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 713c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public void setMimeType(String mimeType) { 714c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke this.mimeType = mimeType; 715c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 716c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 717c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public InputStream getData() { 718c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke return data; 719c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 720c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 721c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public void setData(InputStream data) { 722c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke this.data = data; 723c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 724c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 725c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public Method getRequestMethod() { 726c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke return requestMethod; 727c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 728c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 729c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public void setRequestMethod(Method requestMethod) { 730c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke this.requestMethod = requestMethod; 731c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 732c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 733517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel public void setChunkedTransfer(boolean chunkedTransfer) { 734517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel this.chunkedTransfer = chunkedTransfer; 735517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel } 736517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel 73767e24e482b132ff0a8618f96b43097c9ce5f2776yankee public interface IStatus { 73867e24e482b132ff0a8618f96b43097c9ce5f2776yankee int getRequestStatus(); 73967e24e482b132ff0a8618f96b43097c9ce5f2776yankee String getDescription(); 74067e24e482b132ff0a8618f96b43097c9ce5f2776yankee } 741517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel 7427f0727787957c2f093412c01d165846842ec2425Paul Hawke /** 7439788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Some HTTP response status codes 7449788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 74567e24e482b132ff0a8618f96b43097c9ce5f2776yankee public enum Status implements IStatus { 7466d3737c76afdb1c991bf8d96e3dd2cc622435f09Niko Fink SWITCH_PROTOCOL(101, "Switching Protocols"), OK(200, "OK"), CREATED(201, "Created"), ACCEPTED(202, "Accepted"), NO_CONTENT(204, "No Content"), PARTIAL_CONTENT(206, "Partial Content"), REDIRECT(301, 7478dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke "Moved Permanently"), NOT_MODIFIED(304, "Not Modified"), BAD_REQUEST(400, "Bad Request"), UNAUTHORIZED(401, 748866241d4c51b84a7e4c000d76865f100b0878655synapticloop "Unauthorized"), FORBIDDEN(403, "Forbidden"), NOT_FOUND(404, "Not Found"), METHOD_NOT_ALLOWED(405, "Method Not Allowed"), RANGE_NOT_SATISFIABLE(416, 7498dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke "Requested Range Not Satisfiable"), INTERNAL_ERROR(500, "Internal Server Error"); 7509d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private final int requestStatus; 7519d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private final String description; 7529788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 7539d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke Status(int requestStatus, String description) { 7549788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.requestStatus = requestStatus; 7559d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke this.description = description; 7569788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 7579788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 75867e24e482b132ff0a8618f96b43097c9ce5f2776yankee @Override 7599788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public int getRequestStatus() { 7609788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke return this.requestStatus; 7619788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 7629788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 76367e24e482b132ff0a8618f96b43097c9ce5f2776yankee @Override 7649788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public String getDescription() { 7659d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke return "" + this.requestStatus + " " + description; 7669788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 7679788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 7689788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 769269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 7708dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public static final class ResponseException extends Exception { 7718dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 7728dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private final Response.Status status; 7738dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 7748dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public ResponseException(Response.Status status, String message) { 7758dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke super(message); 7768dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke this.status = status; 7778dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7788dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 7798dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public ResponseException(Response.Status status, String message, Exception e) { 7808dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke super(message, e); 7818dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke this.status = status; 7828dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7838dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 7848dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public Response.Status getStatus() { 7858dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return status; 7868dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7878dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7888dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 7898dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 7908dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Default strategy for creating and cleaning up temporary files. 7918dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 7928dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private class DefaultTempFileManagerFactory implements TempFileManagerFactory { 7938dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke @Override 7948dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public TempFileManager create() { 7958dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return new DefaultTempFileManager(); 7968dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7978dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7988dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 799d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke /** 800d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * Handles one session, i.e. parses the HTTP request and returns the response. 801d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 802249def0afcd823cd21315973bf197f2bef4a089fSleekWeasel public interface IHTTPSession { 803bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel void execute() throws IOException; 804bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 805bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel Map<String, String> getParms(); 806bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 807bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel Map<String, String> getHeaders(); 808bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 809bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel /** 810bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel * @return the path part of the URL. 811bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel */ 812bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel String getUri(); 813bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 814fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke String getQueryParameterString(); 815fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke 816bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel Method getMethod(); 817bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 818bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel InputStream getInputStream(); 819bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 820bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel CookieHandler getCookies(); 821bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 822bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel /** 823bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel * Adds the files in the request body to the files map. 824bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel * @arg files - map to modify 825bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel */ 826bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel void parseBody(Map<String, String> files) throws IOException, ResponseException; 827bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel } 828bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 829249def0afcd823cd21315973bf197f2bef4a089fSleekWeasel protected class HTTPSession implements IHTTPSession { 830249def0afcd823cd21315973bf197f2bef4a089fSleekWeasel public static final int BUFSIZE = 8192; 8314e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke private final TempFileManager tempFileManager; 8329d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private final OutputStream outputStream; 833da2a813333f77d6134a891cd19ad5f635c0b427bSteven te Brinke private PushbackInputStream inputStream; 83493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private int splitbyte; 83593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private int rlen; 83693441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private String uri; 83793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private Method method; 83893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private Map<String, String> parms; 83993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private Map<String, String> headers; 8406a28d482af06a03ea88a1b131d2eac3b8777badblordfokas private CookieHandler cookies; 841fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke private String queryParameterString; 8428dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 843249def0afcd823cd21315973bf197f2bef4a089fSleekWeasel public HTTPSession(TempFileManager tempFileManager, InputStream inputStream, OutputStream outputStream) { 8446200253df7b934c807eb3436b51a3b039406457dPaul Hawke this.tempFileManager = tempFileManager; 845da2a813333f77d6134a891cd19ad5f635c0b427bSteven te Brinke this.inputStream = new PushbackInputStream(inputStream, BUFSIZE); 8466200253df7b934c807eb3436b51a3b039406457dPaul Hawke this.outputStream = outputStream; 847d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke } 848d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 8497e4e4ae652c755faea626f1e2538d495f96e648esynapticloop public HTTPSession(TempFileManager tempFileManager, InputStream inputStream, OutputStream outputStream, InetAddress inetAddress) { 8507e4e4ae652c755faea626f1e2538d495f96e648esynapticloop this.tempFileManager = tempFileManager; 851da2a813333f77d6134a891cd19ad5f635c0b427bSteven te Brinke this.inputStream = new PushbackInputStream(inputStream, BUFSIZE); 8527e4e4ae652c755faea626f1e2538d495f96e648esynapticloop this.outputStream = outputStream; 8537e4e4ae652c755faea626f1e2538d495f96e648esynapticloop String remoteIp = inetAddress.isLoopbackAddress() || inetAddress.isAnyLocalAddress() ? "127.0.0.1" : inetAddress.getHostAddress().toString(); 8547e4e4ae652c755faea626f1e2538d495f96e648esynapticloop headers = new HashMap<String, String>(); 8557e4e4ae652c755faea626f1e2538d495f96e648esynapticloop 8567e4e4ae652c755faea626f1e2538d495f96e648esynapticloop headers.put("remote-addr", remoteIp); 8577e4e4ae652c755faea626f1e2538d495f96e648esynapticloop headers.put("http-client-ip", remoteIp); 8587e4e4ae652c755faea626f1e2538d495f96e648esynapticloop } 8597e4e4ae652c755faea626f1e2538d495f96e648esynapticloop 860bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 86167708a714f4c7883a0f623272383d1e9be3f4de9douzifly public void execute() throws IOException { 862d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke try { 863d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Read the first 8192 bytes. 864d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // The full header should fit in here. 865d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Apache's default header limit is 8KB. 866d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Do NOT assume that a single read will get the entire header at once! 867745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed byte[] buf = new byte[BUFSIZE]; 868745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed splitbyte = 0; 869745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed rlen = 0; 870d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke { 871b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursi int read = -1; 872b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursi try { 873b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursi read = inputStream.read(buf, 0, BUFSIZE); 87468394ef81a80928d25e584475be59a2e937a5ae0Paul Hawke } catch (Exception e) { 87568394ef81a80928d25e584475be59a2e937a5ae0Paul Hawke safeClose(inputStream); 87668394ef81a80928d25e584475be59a2e937a5ae0Paul Hawke safeClose(outputStream); 877b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursi throw new SocketException("NanoHttpd Shutdown"); 878b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursi } 8798dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (read == -1) { 88067708a714f4c7883a0f623272383d1e9be3f4de9douzifly // socket was been closed 88168394ef81a80928d25e584475be59a2e937a5ae0Paul Hawke safeClose(inputStream); 88268394ef81a80928d25e584475be59a2e937a5ae0Paul Hawke safeClose(outputStream); 8838195cac53b34938cb9ac6732be91b0e1fbea2193Paul Hawke throw new SocketException("NanoHttpd Shutdown"); 88467708a714f4c7883a0f623272383d1e9be3f4de9douzifly } 885d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke while (read > 0) { 886d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke rlen += read; 887d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke splitbyte = findHeaderEnd(buf, rlen); 888d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke if (splitbyte > 0) 889d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke break; 8907f0727787957c2f093412c01d165846842ec2425Paul Hawke read = inputStream.read(buf, rlen, BUFSIZE - rlen); 891d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke } 892d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke } 893d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 894745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed if (splitbyte < rlen) { 895da2a813333f77d6134a891cd19ad5f635c0b427bSteven te Brinke inputStream.unread(buf, splitbyte, rlen - splitbyte); 896745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed } 897745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed 89893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed parms = new HashMap<String, String>(); 8997e4e4ae652c755faea626f1e2538d495f96e648esynapticloop if(null == headers) { 9007e4e4ae652c755faea626f1e2538d495f96e648esynapticloop headers = new HashMap<String, String>(); 9017e4e4ae652c755faea626f1e2538d495f96e648esynapticloop } 90293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 903d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Create a BufferedReader for parsing the header. 9047f0727787957c2f093412c01d165846842ec2425Paul Hawke BufferedReader hin = new BufferedReader(new InputStreamReader(new ByteArrayInputStream(buf, 0, rlen))); 905d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 906d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Decode the header into parms and header java properties 90793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed Map<String, String> pre = new HashMap<String, String>(); 9088af508ee58e0abca6dbb6c30532410a1dd99866bMartin M Reed decodeHeader(hin, pre, parms, headers); 90993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 91093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed method = Method.lookup(pre.get("method")); 9110cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (method == null) { 912964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.BAD_REQUEST, "BAD REQUEST: Syntax error."); 913d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke } 91493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 91593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed uri = pre.get("uri"); 9168dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 9176a28d482af06a03ea88a1b131d2eac3b8777badblordfokas cookies = new CookieHandler(headers); 91893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 91993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed // Ok, now do the serve() 92093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed Response r = serve(this); 92193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed if (r == null) { 92293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed throw new ResponseException(Response.Status.INTERNAL_ERROR, "SERVER INTERNAL ERROR: Serve() returned a null response."); 92393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } else { 9248dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke cookies.unloadQueue(r); 92593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed r.setRequestMethod(method); 92693441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed r.send(outputStream); 92793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 9288dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } catch (SocketException e) { 92967708a714f4c7883a0f623272383d1e9be3f4de9douzifly // throw it out to close socket object (finalAccept) 93067708a714f4c7883a0f623272383d1e9be3f4de9douzifly throw e; 931895b2e657db482a93d6ee2c4e3b3fdaddec1278bWorxFR } catch (SocketTimeoutException ste) { 932895b2e657db482a93d6ee2c4e3b3fdaddec1278bWorxFR throw ste; 93393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } catch (IOException ioe) { 93493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed Response r = new Response(Response.Status.INTERNAL_ERROR, MIME_PLAINTEXT, "SERVER INTERNAL ERROR: IOException: " + ioe.getMessage()); 93593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed r.send(outputStream); 9366b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke safeClose(outputStream); 93793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } catch (ResponseException re) { 93893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed Response r = new Response(re.getStatus(), MIME_PLAINTEXT, re.getMessage()); 93993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed r.send(outputStream); 9406b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke safeClose(outputStream); 941ac42f7c14c9da6efb92e38270c9cc44879913ee1Martin M Reed } finally { 942ac42f7c14c9da6efb92e38270c9cc44879913ee1Martin M Reed tempFileManager.clear(); 94393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 94493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 94593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 946bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 947bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel public void parseBody(Map<String, String> files) throws IOException, ResponseException { 94893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed RandomAccessFile randomAccessFile = null; 94993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed BufferedReader in = null; 95093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed try { 95193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 952412e2e136193da510e859f73f9ff2ac21565df69james.mcclure randomAccessFile = getTmpBucket(); 953d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 954745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed long size; 955745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed if (headers.containsKey("content-length")) { 956745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed size = Integer.parseInt(headers.get("content-length")); 957745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed } else if (splitbyte < rlen) { 958745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed size = rlen - splitbyte; 959745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed } else { 960d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke size = 0; 9610cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 962d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 963d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Now read all the body and write it to f 964745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed byte[] buf = new byte[512]; 965d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke while (rlen >= 0 && size > 0) { 966da2a813333f77d6134a891cd19ad5f635c0b427bSteven te Brinke rlen = inputStream.read(buf, 0, (int)Math.min(size, 512)); 967269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke size -= rlen; 9680cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (rlen > 0) { 969412e2e136193da510e859f73f9ff2ac21565df69james.mcclure randomAccessFile.write(buf, 0, rlen); 9700cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 971269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 972269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 973269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Get the raw body as a byte [] 974412e2e136193da510e859f73f9ff2ac21565df69james.mcclure ByteBuffer fbuf = randomAccessFile.getChannel().map(FileChannel.MapMode.READ_ONLY, 0, randomAccessFile.length()); 975412e2e136193da510e859f73f9ff2ac21565df69james.mcclure randomAccessFile.seek(0); 976269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 977269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Create a BufferedReader for easily reading it as string. 978412e2e136193da510e859f73f9ff2ac21565df69james.mcclure InputStream bin = new FileInputStream(randomAccessFile.getFD()); 97993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed in = new BufferedReader(new InputStreamReader(bin)); 980269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 981269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // If the method is POST, there may be parameters 982269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // in data section, too, read it: 983df674ca7f11cee6336b66271c50476f389e0ff4bPaul Hawke if (Method.POST.equals(method)) { 984269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String contentType = ""; 9858af508ee58e0abca6dbb6c30532410a1dd99866bMartin M Reed String contentTypeHeader = headers.get("content-type"); 9862b54eda5fee8430268a99d73b061e790362db544Tom Hermann 9872c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke StringTokenizer st = null; 9882c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke if (contentTypeHeader != null) { 9892c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke st = new StringTokenizer(contentTypeHeader, ",; "); 9902c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke if (st.hasMoreTokens()) { 9912c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke contentType = st.nextToken(); 9922c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke } 993269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 994269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 9950277bbd03983f15110fcf0d039a3e276d0808e29Paul Hawke if ("multipart/form-data".equalsIgnoreCase(contentType)) { 996269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Handle multipart/form-data 9970cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (!st.hasMoreTokens()) { 998964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.BAD_REQUEST, "BAD REQUEST: Content type is multipart/form-data but boundary missing. Usage: GET /example/file.html"); 9990cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 10002b54eda5fee8430268a99d73b061e790362db544Tom Hermann 10014e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke String boundaryStartString = "boundary="; 10022b54eda5fee8430268a99d73b061e790362db544Tom Hermann int boundaryContentStart = contentTypeHeader.indexOf(boundaryStartString) + boundaryStartString.length(); 10034e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke String boundary = contentTypeHeader.substring(boundaryContentStart, contentTypeHeader.length()); 1004220e1a21e7bbb831d06551c72799dfedc1db979fPaul Hawke if (boundary.startsWith("\"") && boundary.endsWith("\"")) { 10054e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke boundary = boundary.substring(1, boundary.length() - 1); 10064e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke } 1007269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1008269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke decodeMultipartData(boundary, fbuf, in, parms, files); 1009269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } else { 1010269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String postLine = ""; 101135b077b8e55a8de374f4f314e151c9cb65eb050aTim Wright StringBuilder postLineBuffer = new StringBuilder(); 1012269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke char pbuf[] = new char[512]; 1013269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int read = in.read(pbuf); 1014269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (read >= 0 && !postLine.endsWith("\r\n")) { 101535b077b8e55a8de374f4f314e151c9cb65eb050aTim Wright postLine = String.valueOf(pbuf, 0, read); 101635b077b8e55a8de374f4f314e151c9cb65eb050aTim Wright postLineBuffer.append(postLine); 1017269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke read = in.read(pbuf); 1018269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 101935b077b8e55a8de374f4f314e151c9cb65eb050aTim Wright postLine = postLineBuffer.toString().trim(); 1020e9a1b5310ed98d7c80095fbefd856a0b1277a1d2unknown // Handle application/x-www-form-urlencoded 1021e9a1b5310ed98d7c80095fbefd856a0b1277a1d2unknown if ("application/x-www-form-urlencoded".equalsIgnoreCase(contentType)) { 1022e9a1b5310ed98d7c80095fbefd856a0b1277a1d2unknown decodeParms(postLine, parms); 1023e9a1b5310ed98d7c80095fbefd856a0b1277a1d2unknown } else if (postLine.length() != 0) { 1024e9a1b5310ed98d7c80095fbefd856a0b1277a1d2unknown // Special case for raw POST data => create a special files entry "postData" with raw content data 1025e9a1b5310ed98d7c80095fbefd856a0b1277a1d2unknown files.put("postData", postLine); 1026e9a1b5310ed98d7c80095fbefd856a0b1277a1d2unknown } 1027269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1028b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke } else if (Method.PUT.equals(method)) { 1029dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke files.put("content", saveTmpFile(fbuf, 0, fbuf.limit())); 1030b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke } 10314e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke } finally { 10329cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed safeClose(randomAccessFile); 103393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed safeClose(in); 1034269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1035269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1036269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1037269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 10387f0727787957c2f093412c01d165846842ec2425Paul Hawke * Decodes the sent headers and loads the data into Key/value pairs 1039d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 10408af508ee58e0abca6dbb6c30532410a1dd99866bMartin M Reed private void decodeHeader(BufferedReader in, Map<String, String> pre, Map<String, String> parms, Map<String, String> headers) 10418dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke throws ResponseException { 1042269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke try { 1043269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Read the request line 1044269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String inLine = in.readLine(); 10450cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (inLine == null) { 1046269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return; 10470cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 10480cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke 1049269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke StringTokenizer st = new StringTokenizer(inLine); 10500cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (!st.hasMoreTokens()) { 1051964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.BAD_REQUEST, "BAD REQUEST: Syntax error. Usage: GET /example/file.html"); 10520cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 1053269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 10540277bbd03983f15110fcf0d039a3e276d0808e29Paul Hawke pre.put("method", st.nextToken()); 1055269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 10560cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (!st.hasMoreTokens()) { 1057964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.BAD_REQUEST, "BAD REQUEST: Missing URI. Usage: GET /example/file.html"); 10580cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 1059269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1060269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String uri = st.nextToken(); 1061269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1062269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Decode parameters from the URI 1063269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int qmi = uri.indexOf('?'); 1064269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (qmi >= 0) { 1065269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke decodeParms(uri.substring(qmi + 1), parms); 1066269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke uri = decodePercent(uri.substring(0, qmi)); 10679788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } else { 1068269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke uri = decodePercent(uri); 10699788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 1070269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1071269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // If there's another token, it's protocol version, 1072269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // followed by HTTP headers. Ignore version but parse headers. 1073269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // NOTE: this now forces header names lowercase since they are 1074269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // case insensitive and vary by client. 1075269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (st.hasMoreTokens()) { 1076269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String line = in.readLine(); 1077269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (line != null && line.trim().length() > 0) { 1078269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int p = line.indexOf(':'); 1079269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (p >= 0) 1080eb71c8999ff29f7dc3fe8dcdba81084f7f6a5facDavid Masover headers.put(line.substring(0, p).trim().toLowerCase(Locale.US), line.substring(p + 1).trim()); 1081269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke line = in.readLine(); 1082269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1083269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1084269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1085269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke pre.put("uri", uri); 1086269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } catch (IOException ioe) { 1087964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.INTERNAL_ERROR, "SERVER INTERNAL ERROR: IOException: " + ioe.getMessage(), ioe); 1088269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1089269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1090269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1091269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 10927f0727787957c2f093412c01d165846842ec2425Paul Hawke * Decodes the Multipart Body data and put it into Key/Value pairs. 1093d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 1094dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke private void decodeMultipartData(String boundary, ByteBuffer fbuf, BufferedReader in, Map<String, String> parms, 1095964393fd7e5ed49088882a126cf82507184467efMartin M Reed Map<String, String> files) throws ResponseException { 1096269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke try { 1097269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int[] bpositions = getBoundaryPositions(fbuf, boundary.getBytes()); 1098269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int boundarycount = 1; 1099269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String mpline = in.readLine(); 1100269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (mpline != null) { 11017b88819e82b89ac3476ce060903468076a73de8fPaul Hawke if (!mpline.contains(boundary)) { 1102964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.BAD_REQUEST, "BAD REQUEST: Content type is multipart/form-data but next chunk does not start with boundary. Usage: GET /example/file.html"); 11037b88819e82b89ac3476ce060903468076a73de8fPaul Hawke } 1104269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke boundarycount++; 1105269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke Map<String, String> item = new HashMap<String, String>(); 1106269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke mpline = in.readLine(); 1107269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (mpline != null && mpline.trim().length() > 0) { 1108269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int p = mpline.indexOf(':'); 1109269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (p != -1) { 1110eb71c8999ff29f7dc3fe8dcdba81084f7f6a5facDavid Masover item.put(mpline.substring(0, p).trim().toLowerCase(Locale.US), mpline.substring(p + 1).trim()); 1111269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1112269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke mpline = in.readLine(); 1113269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1114269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (mpline != null) { 1115269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String contentDisposition = item.get("content-disposition"); 1116269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (contentDisposition == null) { 1117964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.BAD_REQUEST, "BAD REQUEST: Content type is multipart/form-data but no content-disposition info found. Usage: GET /example/file.html"); 1118269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 11196bed46385ee916d2e82c7a31d76a0bfb1079a8d9Gaziz Tazhenov StringTokenizer st = new StringTokenizer(contentDisposition, ";"); 1120269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke Map<String, String> disposition = new HashMap<String, String>(); 1121269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (st.hasMoreTokens()) { 11226bed46385ee916d2e82c7a31d76a0bfb1079a8d9Gaziz Tazhenov String token = st.nextToken().trim(); 1123269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int p = token.indexOf('='); 1124269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (p != -1) { 1125eb71c8999ff29f7dc3fe8dcdba81084f7f6a5facDavid Masover disposition.put(token.substring(0, p).trim().toLowerCase(Locale.US), token.substring(p + 1).trim()); 1126269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1127269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1128269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String pname = disposition.get("name"); 1129269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke pname = pname.substring(1, pname.length() - 1); 1130269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1131269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String value = ""; 1132269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (item.get("content-type") == null) { 11337b88819e82b89ac3476ce060903468076a73de8fPaul Hawke while (mpline != null && !mpline.contains(boundary)) { 1134269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke mpline = in.readLine(); 1135269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (mpline != null) { 1136269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int d = mpline.indexOf(boundary); 1137269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (d == -1) { 1138269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke value += mpline; 1139269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } else { 1140269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke value += mpline.substring(0, d - 2); 1141269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1142269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1143269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1144269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } else { 1145269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (boundarycount > bpositions.length) { 1146964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.INTERNAL_ERROR, "Error processing request"); 1147269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1148269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int offset = stripMultipartHeaders(fbuf, bpositions[boundarycount - 2]); 1149269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String path = saveTmpFile(fbuf, offset, bpositions[boundarycount - 1] - offset - 4); 1150269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke files.put(pname, path); 1151269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke value = disposition.get("filename"); 1152269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke value = value.substring(1, value.length() - 1); 1153269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke do { 1154269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke mpline = in.readLine(); 11557b88819e82b89ac3476ce060903468076a73de8fPaul Hawke } while (mpline != null && !mpline.contains(boundary)); 1156269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1157269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke parms.put(pname, value); 1158269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1159269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1160269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } catch (IOException ioe) { 1161964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.INTERNAL_ERROR, "SERVER INTERNAL ERROR: IOException: " + ioe.getMessage(), ioe); 1162269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1163269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1164269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1165269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 1166269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * Find byte index separating header from body. It must be the last byte of the first two sequential new lines. 1167d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 1168269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke private int findHeaderEnd(final byte[] buf, int rlen) { 1169269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int splitbyte = 0; 1170269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (splitbyte + 3 < rlen) { 1171269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (buf[splitbyte] == '\r' && buf[splitbyte + 1] == '\n' && buf[splitbyte + 2] == '\r' && buf[splitbyte + 3] == '\n') { 1172269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return splitbyte + 4; 1173269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1174269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke splitbyte++; 1175269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1176269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return 0; 1177269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1178269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1179269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 1180269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * Find the byte positions where multipart boundaries start. 1181d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 118293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private int[] getBoundaryPositions(ByteBuffer b, byte[] boundary) { 1183269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int matchcount = 0; 1184269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int matchbyte = -1; 1185269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke List<Integer> matchbytes = new ArrayList<Integer>(); 11868dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke for (int i = 0; i < b.limit(); i++) { 1187dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke if (b.get(i) == boundary[matchcount]) { 1188269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (matchcount == 0) 1189269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchbyte = i; 1190269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchcount++; 1191269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (matchcount == boundary.length) { 1192269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchbytes.add(matchbyte); 1193269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchcount = 0; 1194269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchbyte = -1; 1195269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1196269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } else { 1197269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke i -= matchcount; 1198269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchcount = 0; 1199269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchbyte = -1; 1200269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1201269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1202269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int[] ret = new int[matchbytes.size()]; 1203269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke for (int i = 0; i < ret.length; i++) { 12047b88819e82b89ac3476ce060903468076a73de8fPaul Hawke ret[i] = matchbytes.get(i); 1205269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1206269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return ret; 1207269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1208269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1209269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 1210269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * Retrieves the content of a sent file and saves it to a temporary file. The full path to the saved file is returned. 1211d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 12128dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private String saveTmpFile(ByteBuffer b, int offset, int len) { 1213269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String path = ""; 1214269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (len > 0) { 12156b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke FileOutputStream fileOutputStream = null; 1216269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke try { 12174e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke TempFile tempFile = tempFileManager.createTempFile(); 1218dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke ByteBuffer src = b.duplicate(); 12196b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke fileOutputStream = new FileOutputStream(tempFile.getName()); 12206b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke FileChannel dest = fileOutputStream.getChannel(); 1221dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke src.position(offset).limit(offset + len); 1222dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke dest.write(src.slice()); 12234e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke path = tempFile.getName(); 1224269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } catch (Exception e) { // Catch exception if any 1225ad3ccf027ec060e1adb3c9b6c577a8ebf32866b2yankee throw new Error(e); // we won't recover, so throw an error 1226062b50e32b22f4c419bb4b8e3d781b9eac056380Martin M Reed } finally { 12276b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke safeClose(fileOutputStream); 1228269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1229269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1230269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return path; 1231269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1232269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 12339d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private RandomAccessFile getTmpBucket() { 1234dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke try { 1235dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke TempFile tempFile = tempFileManager.createTempFile(); 1236dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke return new RandomAccessFile(tempFile.getName(), "rw"); 1237dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke } catch (Exception e) { 1238ad3ccf027ec060e1adb3c9b6c577a8ebf32866b2yankee throw new Error(e); // we won't recover, so throw an error 1239dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke } 1240dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke } 1241dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke 1242269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 1243269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * It returns the offset separating multipart file headers from the file's data. 1244269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 1245dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke private int stripMultipartHeaders(ByteBuffer b, int offset) { 1246269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int i; 12478dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke for (i = offset; i < b.limit(); i++) { 1248dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke if (b.get(i) == '\r' && b.get(++i) == '\n' && b.get(++i) == '\r' && b.get(++i) == '\n') { 1249269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke break; 1250269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1251269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1252269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return i + 1; 1253269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1254269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1255269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 12567f0727787957c2f093412c01d165846842ec2425Paul Hawke * Decodes parameters in percent-encoded URI-format ( e.g. "name=Jack%20Daniels&pass=Single%20Malt" ) and 12577f0727787957c2f093412c01d165846842ec2425Paul Hawke * adds them to given Map. NOTE: this doesn't support multiple identical keys due to the simplicity of Map. 1258269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 12599d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private void decodeParms(String parms, Map<String, String> p) { 1260f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke if (parms == null) { 1261fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke queryParameterString = ""; 1262269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return; 1263f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 1264269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1265fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke queryParameterString = parms; 1266269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke StringTokenizer st = new StringTokenizer(parms, "&"); 1267b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke while (st.hasMoreTokens()) { 1268b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke String e = st.nextToken(); 1269b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke int sep = e.indexOf('='); 1270b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke if (sep >= 0) { 1271b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke p.put(decodePercent(e.substring(0, sep)).trim(), 12728dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke decodePercent(e.substring(sep + 1))); 1273b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke } else { 1274b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke p.put(decodePercent(e).trim(), ""); 1275269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1276269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1277269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 127893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 1279bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 128093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed public final Map<String, String> getParms() { 128193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed return parms; 128293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 128393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 1284e9a1b5310ed98d7c80095fbefd856a0b1277a1d2unknown public String getQueryParameterString() { 1285fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke return queryParameterString; 1286fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke } 1287fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke 1288bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 128993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed public final Map<String, String> getHeaders() { 129093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed return headers; 129193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 129293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 1293bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 129493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed public final String getUri() { 129593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed return uri; 129693441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 129793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 1298bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 129993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed public final Method getMethod() { 130093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed return method; 130193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 13028cdc631dca8d05f64ea8da466ecfe759a2f6e429Martin M Reed 1303bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 13048cdc631dca8d05f64ea8da466ecfe759a2f6e429Martin M Reed public final InputStream getInputStream() { 13058cdc631dca8d05f64ea8da466ecfe759a2f6e429Martin M Reed return inputStream; 13068cdc631dca8d05f64ea8da466ecfe759a2f6e429Martin M Reed } 1307964393fd7e5ed49088882a126cf82507184467efMartin M Reed 1308bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 13098dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public CookieHandler getCookies() { 13108dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return cookies; 13118dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 13128dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 1313964393fd7e5ed49088882a126cf82507184467efMartin M Reed 13145820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public static class Cookie { 13155820f308631a825a05feddddd0aafb962f272b9bPaul Hawke private String n, v, e; 13165820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13175820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public Cookie(String name, String value, String expires) { 13185820f308631a825a05feddddd0aafb962f272b9bPaul Hawke n = name; 13195820f308631a825a05feddddd0aafb962f272b9bPaul Hawke v = value; 13205820f308631a825a05feddddd0aafb962f272b9bPaul Hawke e = expires; 13215820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13225820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13235820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public Cookie(String name, String value) { 13245820f308631a825a05feddddd0aafb962f272b9bPaul Hawke this(name, value, 30); 13255820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13265820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13275820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public Cookie(String name, String value, int numDays) { 13285820f308631a825a05feddddd0aafb962f272b9bPaul Hawke n = name; 13295820f308631a825a05feddddd0aafb962f272b9bPaul Hawke v = value; 13305820f308631a825a05feddddd0aafb962f272b9bPaul Hawke e = getHTTPTime(numDays); 13315820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13325820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13335820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public String getHTTPHeader() { 13345820f308631a825a05feddddd0aafb962f272b9bPaul Hawke String fmt = "%s=%s; expires=%s"; 13355820f308631a825a05feddddd0aafb962f272b9bPaul Hawke return String.format(fmt, n, v, e); 13365820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13375820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13385820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public static String getHTTPTime(int days) { 13395820f308631a825a05feddddd0aafb962f272b9bPaul Hawke Calendar calendar = Calendar.getInstance(); 13405820f308631a825a05feddddd0aafb962f272b9bPaul Hawke SimpleDateFormat dateFormat = new SimpleDateFormat("EEE, dd MMM yyyy HH:mm:ss z", Locale.US); 13415820f308631a825a05feddddd0aafb962f272b9bPaul Hawke dateFormat.setTimeZone(TimeZone.getTimeZone("GMT")); 13425820f308631a825a05feddddd0aafb962f272b9bPaul Hawke calendar.add(Calendar.DAY_OF_MONTH, days); 13435820f308631a825a05feddddd0aafb962f272b9bPaul Hawke return dateFormat.format(calendar.getTime()); 13445820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13455820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13465820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13478dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 13488dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Provides rudimentary support for cookies. 13498dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Doesn't support 'path', 'secure' nor 'httpOnly'. 13508dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Feel free to improve it and/or add unsupported features. 13518dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 13528dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @author LordFokas 13538dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 13545820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public class CookieHandler implements Iterable<String> { 13558dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private HashMap<String, String> cookies = new HashMap<String, String>(); 13568dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private ArrayList<Cookie> queue = new ArrayList<Cookie>(); 13578dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 13588dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public CookieHandler(Map<String, String> httpHeaders) { 13598dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke String raw = httpHeaders.get("cookie"); 13608dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (raw != null) { 13618dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke String[] tokens = raw.split(";"); 13628dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke for (String token : tokens) { 13638dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke String[] data = token.trim().split("="); 13648dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (data.length == 2) { 13658dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke cookies.put(data[0], data[1]); 13668dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 13678dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 13688dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 1369964393fd7e5ed49088882a126cf82507184467efMartin M Reed } 1370964393fd7e5ed49088882a126cf82507184467efMartin M Reed 13715820f308631a825a05feddddd0aafb962f272b9bPaul Hawke @Override public Iterator<String> iterator() { 13725820f308631a825a05feddddd0aafb962f272b9bPaul Hawke return cookies.keySet().iterator(); 13735820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13745820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13758dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 13768dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Read a cookie from the HTTP Headers. 13778dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 13788dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param name The cookie's name. 13798dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @return The cookie's value if it exists, null otherwise. 13808dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 13818dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public String read(String name) { 13828dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return cookies.get(name); 1383964393fd7e5ed49088882a126cf82507184467efMartin M Reed } 1384964393fd7e5ed49088882a126cf82507184467efMartin M Reed 13858dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 13868dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Sets a cookie. 13878dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 13888dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param name The cookie's name. 13898dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param value The cookie's value. 13908dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param expires How many days until the cookie expires. 13918dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 13928dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public void set(String name, String value, int expires) { 13935820f308631a825a05feddddd0aafb962f272b9bPaul Hawke queue.add(new Cookie(name, value, Cookie.getHTTPTime(expires))); 13945820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13955820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13965820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public void set(Cookie cookie) { 13975820f308631a825a05feddddd0aafb962f272b9bPaul Hawke queue.add(cookie); 1398964393fd7e5ed49088882a126cf82507184467efMartin M Reed } 1399964393fd7e5ed49088882a126cf82507184467efMartin M Reed 14008dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 14018dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Set a cookie with an expiration date from a month ago, effectively deleting it on the client side. 14028dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 14038dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param name The cookie name. 14048dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 14058dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public void delete(String name) { 14068dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke set(name, "-delete-", -30); 14079cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed } 14089cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed 14098dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 14108dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Internally used by the webserver to add all queued cookies into the Response's HTTP Headers. 14118dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 14128dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param response The Response object to which headers the queued cookies will be added. 14138dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 14148dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public void unloadQueue(Response response) { 14158dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke for (Cookie cookie : queue) { 14168dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke response.addHeader("Set-Cookie", cookie.getHTTPHeader()); 14179cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed } 14189cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed } 14197e423dde0fe863fadbd363de8a94746a30fca0eeMartin M Reed } 1420269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke} 1421