NanoHTTPD.java revision c7fd739559d0f33a7fdbc831183acc5e2b0b15bb
1269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawkepackage fi.iki.elonen; 2269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 3b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.BufferedReader; 4b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.ByteArrayInputStream; 5b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.Closeable; 6b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.File; 7b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.FileInputStream; 8b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.FileOutputStream; 9b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.IOException; 10b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.InputStream; 11b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.InputStreamReader; 12b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.OutputStream; 13b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.PrintWriter; 14b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.RandomAccessFile; 15b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.SequenceInputStream; 16b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.io.UnsupportedEncodingException; 17b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.InetAddress; 18b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.InetSocketAddress; 19b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.ServerSocket; 20b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.Socket; 21b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.SocketException; 2213736e18ec88e3df74d055e061fb324e04778ad6Paul Hawkeimport java.net.SocketTimeoutException; 23b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.net.URLDecoder; 24dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawkeimport java.nio.ByteBuffer; 25dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawkeimport java.nio.channels.FileChannel; 267b88819e82b89ac3476ce060903468076a73de8fPaul Hawkeimport java.text.SimpleDateFormat; 27b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.ArrayList; 28b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Calendar; 29b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Date; 30b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.HashMap; 31b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.HashSet; 32b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Iterator; 33b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.List; 34b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Locale; 35b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Map; 36b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.Set; 37b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.StringTokenizer; 38b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursiimport java.util.TimeZone; 39269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 40269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke/** 415b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * A simple, tiny, nicely embeddable HTTP server in Java 42d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 43d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 44b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke * NanoHTTPD 4584dcf74395d1817c4a2a7861bd5d3787daa6700aPaul Hawke * <p></p>Copyright (c) 2012-2013 by Paul S. Hawke, 2001,2005-2013 by Jarno Elonen, 2010 by Konstantinos Togias</p> 46d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 47d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 48269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <b>Features + limitations: </b> 49269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <ul> 50d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 51269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Only one Java file</li> 5201ecef8887ed32be34ef7442a656a0b3330ecb8dPaul Hawke * <li>Java 5 compatible</li> 53269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Released as open source, Modified BSD licence</li> 54269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>No fixed config files, logging, authorization etc. (Implement yourself if you need them.)</li> 55269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Supports parameter parsing of GET and POST methods (+ rudimentary PUT support in 1.25)</li> 56269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Supports both dynamic content and file serving</li> 57269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Supports file upload (since version 1.2, 2010)</li> 58269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Supports partial content (streaming)</li> 59269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Supports ETags</li> 60269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Never caches anything</li> 61269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Doesn't limit bandwidth, request time or simultaneous connections</li> 62269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Default code serves files and shows all HTTP parameters and headers</li> 63269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server supports directory listing, index.html and index.htm</li> 64269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server supports partial content (streaming)</li> 65269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server supports ETags</li> 66269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server does the 301 redirection trick for directories without '/'</li> 67269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server supports simple skipping for files (continue download)</li> 68269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>File server serves also very long files without memory overhead</li> 69269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>Contains a built-in list of most common mime types</li> 70269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <li>All header names are converted lowercase so they don't vary between browsers/clients</li> 71d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 72269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * </ul> 73d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 74d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 7501ecef8887ed32be34ef7442a656a0b3330ecb8dPaul Hawke * <b>How to use: </b> 76269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * <ul> 77d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 7801ecef8887ed32be34ef7442a656a0b3330ecb8dPaul Hawke * <li>Subclass and implement serve() and embed to your own program</li> 79d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 80269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * </ul> 81d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 8250b70cd2a7250fd85394edc863b219a3bfb738e6Paul Hawke * See the separate "LICENSE.md" file for the distribution license (Modified BSD licence) 83269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 84d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawkepublic abstract class NanoHTTPD { 85269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 86293e04675f3d4f427c125a26e831d70d5011c79bhflicka * Maximum time to wait on Socket.getInputStream().read() (in milliseconds) 87293e04675f3d4f427c125a26e831d70d5011c79bhflicka * This is required as the Keep-Alive HTTP connections would otherwise 88293e04675f3d4f427c125a26e831d70d5011c79bhflicka * block the socket reading thread forever (or as long the browser is open). 89293e04675f3d4f427c125a26e831d70d5011c79bhflicka */ 90293e04675f3d4f427c125a26e831d70d5011c79bhflicka public static final int SOCKET_READ_TIMEOUT = 5000; 91293e04675f3d4f427c125a26e831d70d5011c79bhflicka /** 925b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Common mime type for dynamic content: plain text 93269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 94269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke public static final String MIME_PLAINTEXT = "text/plain"; 955b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 965b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Common mime type for dynamic content: html 975b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 98269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke public static final String MIME_HTML = "text/html"; 995b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 1008dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Pseudo-Parameter to use to store the actual query string in the parameters map for later re-processing. 1018dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 1028dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private static final String QUERY_STRING_PARAMETER = "NanoHttpd.QUERY_STRING"; 103f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke private final String hostname; 1047b88819e82b89ac3476ce060903468076a73de8fPaul Hawke private final int myPort; 1057b88819e82b89ac3476ce060903468076a73de8fPaul Hawke private ServerSocket myServerSocket; 1060a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander private Set<Socket> openConnections = new HashSet<Socket>(); 1077b88819e82b89ac3476ce060903468076a73de8fPaul Hawke private Thread myThread; 1085b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 1098dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Pluggable strategy for asynchronously executing requests. 1105b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 1118dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private AsyncRunner asyncRunner; 1128dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 1138dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Pluggable strategy for creating and cleaning up temporary files. 1148dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 1158dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private TempFileManagerFactory tempFileManagerFactory; 116269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 117269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 118d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * Constructs an HTTP server on given port. 119269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 120df674ca7f11cee6336b66271c50476f389e0ff4bPaul Hawke public NanoHTTPD(int port) { 121f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke this(null, port); 122f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke } 123f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke 1245b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 1256e419652da8eb07f46c768c083c46c206e0e2762Martin M Reed * Constructs an HTTP server on given hostname and port. 1265b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 127f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke public NanoHTTPD(String hostname, int port) { 128f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke this.hostname = hostname; 1297b88819e82b89ac3476ce060903468076a73de8fPaul Hawke this.myPort = port; 1305b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke setTempFileManagerFactory(new DefaultTempFileManagerFactory()); 1315b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke setAsyncRunner(new DefaultAsyncRunner()); 1323f2aa2e2e0e02fe7e0151eb64d31265b578885ecPaul Hawke } 1333f2aa2e2e0e02fe7e0151eb64d31265b578885ecPaul Hawke 1348dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private static final void safeClose(Closeable closeable) { 1358dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (closeable != null) { 1368dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke try { 1378dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke closeable.close(); 1388dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } catch (IOException e) { 1398dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 1408dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 1418dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 1428dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 143c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke private static final void safeClose(Socket closeable) { 144c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke if (closeable != null) { 145c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke try { 146c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke closeable.close(); 147c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } catch (IOException e) { 148c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 149c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 150c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 151c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke 152c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke private static final void safeClose(ServerSocket closeable) { 153c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke if (closeable != null) { 154c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke try { 155c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke closeable.close(); 156c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } catch (IOException e) { 157c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 158c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 159c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke } 160c7fd739559d0f33a7fdbc831183acc5e2b0b15bbPaul Hawke 1613f2aa2e2e0e02fe7e0151eb64d31265b578885ecPaul Hawke /** 1625b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Start the server. 1638dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 1645b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @throws IOException if the socket is in use. 1653f2aa2e2e0e02fe7e0151eb64d31265b578885ecPaul Hawke */ 1667b88819e82b89ac3476ce060903468076a73de8fPaul Hawke public void start() throws IOException { 167f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke myServerSocket = new ServerSocket(); 168f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke myServerSocket.bind((hostname != null) ? new InetSocketAddress(hostname, myPort) : new InetSocketAddress(myPort)); 169f639f201819cb2652cfcad67ed8633e0609cb440Paul Hawke 170269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke myThread = new Thread(new Runnable() { 171269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke @Override 172269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke public void run() { 1736200253df7b934c807eb3436b51a3b039406457dPaul Hawke do { 1746200253df7b934c807eb3436b51a3b039406457dPaul Hawke try { 1750ad8e8a948f2a3d347ff633eccc875363fdbbb19Björn Heinrichs final Socket finalAccept = myServerSocket.accept(); 1760a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander registerConnection(finalAccept); 177293e04675f3d4f427c125a26e831d70d5011c79bhflicka finalAccept.setSoTimeout(SOCKET_READ_TIMEOUT); 1786e419652da8eb07f46c768c083c46c206e0e2762Martin M Reed final InputStream inputStream = finalAccept.getInputStream(); 179864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs asyncRunner.exec(new Runnable() { 180864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs @Override 181864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs public void run() { 182864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs OutputStream outputStream = null; 183864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs try { 184864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs outputStream = finalAccept.getOutputStream(); 185864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs TempFileManager tempFileManager = tempFileManagerFactory.create(); 186864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs HTTPSession session = new HTTPSession(tempFileManager, inputStream, outputStream, finalAccept.getInetAddress()); 187864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs while (!finalAccept.isClosed()) { 188864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs session.execute(); 1896e419652da8eb07f46c768c083c46c206e0e2762Martin M Reed } 190864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs } catch (Exception e) { 191864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs // When the socket is closed by the client, we throw our own SocketException 192864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs // to break the "keep alive" loop above. 193864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs if (!(e instanceof SocketException && "NanoHttpd Shutdown".equals(e.getMessage()))) { 194864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs e.printStackTrace(); 195864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs } 196864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs } finally { 197864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs safeClose(outputStream); 198864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs safeClose(inputStream); 199864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs safeClose(finalAccept); 200864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs unRegisterConnection(finalAccept); 2016e419652da8eb07f46c768c083c46c206e0e2762Martin M Reed } 202864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs } 203864df3792cf52d9a4367fdbbeb9d730a4c24f629Björn Heinrichs }); 2046200253df7b934c807eb3436b51a3b039406457dPaul Hawke } catch (IOException e) { 2056200253df7b934c807eb3436b51a3b039406457dPaul Hawke } 2062dc83f968760f835a138f0320db9eca60f8e0ecaBjörn Heinrichs } while (!myServerSocket.isClosed()); 207269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 208269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke }); 209269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke myThread.setDaemon(true); 2101ca771092a5a39cfea9c4cf843ba7f7c90a1dbccPaul Hawke myThread.setName("NanoHttpd Main Listener"); 211269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke myThread.start(); 212269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 213d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 214269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 2155b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Stop the server. 216269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 217269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke public void stop() { 218269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke try { 2199cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed safeClose(myServerSocket); 2200a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander closeAllConnections(); 221269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke myThread.join(); 2225b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } catch (Exception e) { 2237b88819e82b89ac3476ce060903468076a73de8fPaul Hawke e.printStackTrace(); 224269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 225269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 226269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 2270a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander /** 2280a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * Registers that a new connection has been set up. 2290a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * 2300a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * @param socket 2310a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * the {@link Socket} for the connection. 2320a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander */ 2330a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander public synchronized void registerConnection(Socket socket) { 2340a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander openConnections.add(socket); 2350a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander } 2360a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander 2370a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander /** 2380a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * Registers that a connection has been closed 2390a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * 2400a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * @param socket 2410a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * the {@link Socket} for the connection. 2420a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander */ 2430a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander public synchronized void unRegisterConnection(Socket socket) { 2440a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander openConnections.remove(socket); 2450a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander } 2460a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander 2470a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander /** 2480a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander * Forcibly closes all connections that are open. 2490a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander */ 2500a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander public synchronized void closeAllConnections() { 2510a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander for (Socket socket : openConnections) { 2520a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander safeClose(socket); 2530a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander } 2540a35219489d8f19d13197bc60eec168dd6b3089dAlbin Theander } 2558dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 2568dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public final int getListeningPort() { 2578dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return myServerSocket == null ? -1 : myServerSocket.getLocalPort(); 2588dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 2598dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 2608dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public final boolean wasStarted() { 2618dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return myServerSocket != null && myThread != null; 2628dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 2638dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 2648dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public final boolean isAlive() { 2658dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return wasStarted() && !myServerSocket.isClosed() && myThread.isAlive(); 2668dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 2678dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 268a6619fe65dd3a997960724faf7c51685de793f91Paul Hawke /** 269d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * Override this to customize the server. 270d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 271d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * <p/> 272d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * (By default, this delegates to serveFile() and allows directory listing.) 2739788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * 2748dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param uri Percent-decoded URI without parameters, for example "/index.cgi" 2758dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param method "GET", "POST" etc. 2768dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param parms Parsed, percent decoded parameters from URI and, in case of POST, data. 2778af508ee58e0abca6dbb6c30532410a1dd99866bMartin M Reed * @param headers Header entries, percent decoded 278d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * @return HTTP response, see class Response for details 279269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 2808dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke @Deprecated 2818dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public Response serve(String uri, Method method, Map<String, String> headers, Map<String, String> parms, 2828dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke Map<String, String> files) { 2838dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return new Response(Response.Status.NOT_FOUND, MIME_PLAINTEXT, "Not Found"); 2848dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 2859788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 286f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke /** 28793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * Override this to customize the server. 28893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * <p/> 28993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * <p/> 29093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * (By default, this delegates to serveFile() and allows directory listing.) 29193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * 29293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * @param session The HTTP session 29393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed * @return HTTP response, see class Response for details 29493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed */ 295249def0afcd823cd21315973bf197f2bef4a089fSleekWeasel public Response serve(IHTTPSession session) { 29693441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed Map<String, String> files = new HashMap<String, String>(); 2978dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke Method method = session.getMethod(); 2988dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (Method.PUT.equals(method) || Method.POST.equals(method)) { 2998dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke try { 3008dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke session.parseBody(files); 3018dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } catch (IOException ioe) { 3028dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return new Response(Response.Status.INTERNAL_ERROR, MIME_PLAINTEXT, "SERVER INTERNAL ERROR: IOException: " + ioe.getMessage()); 3038dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } catch (ResponseException re) { 3048dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return new Response(re.getStatus(), MIME_PLAINTEXT, re.getMessage()); 3058dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 30693441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 30793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 308fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke Map<String, String> parms = session.getParms(); 309fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke parms.put(QUERY_STRING_PARAMETER, session.getQueryParameterString()); 310fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke return serve(session.getUri(), method, session.getHeaders(), parms, files); 31193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 31293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 31393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed /** 3145b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Decode percent encoded <code>String</code> values. 3158dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 3165b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @param str the percent encoded <code>String</code> 3175b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @return expanded form of the input, for example "foo%20bar" becomes "foo bar" 318f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke */ 319b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke protected String decodePercent(String str) { 32049f777c1de22de7f0918439582d79d98d0b6e70aPaul Hawke String decoded = null; 32149f777c1de22de7f0918439582d79d98d0b6e70aPaul Hawke try { 32249f777c1de22de7f0918439582d79d98d0b6e70aPaul Hawke decoded = URLDecoder.decode(str, "UTF8"); 32349f777c1de22de7f0918439582d79d98d0b6e70aPaul Hawke } catch (UnsupportedEncodingException ignored) { 324f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 32549f777c1de22de7f0918439582d79d98d0b6e70aPaul Hawke return decoded; 326f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 327f639f201819cb2652cfcad67ed8633e0609cb440Paul 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 parms original <b>NanoHttpd</b> parameters values, as passed to the <code>serve()</code> method. 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(Map<String, String> parms) { 337f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke return this.decodeParameters(parms.get(QUERY_STRING_PARAMETER)); 338f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 339f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke 3405b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 3415b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Decode parameters from a URL, handing the case where a single parameter name might have been 3425b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * supplied several times, by return lists of values. In general these lists will contain a single 3435b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * element. 3445b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * 3455b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @param queryString a query string pulled from the URL. 3465b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @return a map of <code>String</code> (parameter name) to <code>List<String></code> (a list of the values supplied). 3475b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 348f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke protected Map<String, List<String>> decodeParameters(String queryString) { 349f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke Map<String, List<String>> parms = new HashMap<String, List<String>>(); 350f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke if (queryString != null) { 351f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke StringTokenizer st = new StringTokenizer(queryString, "&"); 352f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke while (st.hasMoreTokens()) { 353f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke String e = st.nextToken(); 354f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke int sep = e.indexOf('='); 355b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke String propertyName = (sep >= 0) ? decodePercent(e.substring(0, sep)).trim() : decodePercent(e).trim(); 356b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke if (!parms.containsKey(propertyName)) { 357b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke parms.put(propertyName, new ArrayList<String>()); 358b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke } 359b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke String propertyValue = (sep >= 0) ? decodePercent(e.substring(sep + 1)) : null; 360b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke if (propertyValue != null) { 361b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke parms.get(propertyName).add(propertyValue); 362f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 363f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 364f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 365f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke return parms; 366f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 367f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke 3685b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // ------------------------------------------------------------------------------- // 3695b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // 3705b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // Threading Strategy. 3715b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // 3725b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // ------------------------------------------------------------------------------- // 3735b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 3745b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 3755b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Pluggable strategy for asynchronously executing requests. 3768dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 3775b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @param asyncRunner new strategy for handling threads. 3785b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 3795b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public void setAsyncRunner(AsyncRunner asyncRunner) { 3805b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke this.asyncRunner = asyncRunner; 3815b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 3825b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 3835b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // ------------------------------------------------------------------------------- // 3845b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // 3855b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // Temp file handling strategy. 3865b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // 3875b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke // ------------------------------------------------------------------------------- // 3885b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 3895b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 3905b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Pluggable strategy for creating and cleaning up temporary files. 3918dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 3925b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * @param tempFileManagerFactory new strategy for handling temp files. 3935b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 3945b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public void setTempFileManagerFactory(TempFileManagerFactory tempFileManagerFactory) { 3955b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke this.tempFileManagerFactory = tempFileManagerFactory; 3965b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 3975b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 3985b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 3998dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * HTTP Request methods, with the ability to decode a <code>String</code> back to its enum value. 4008dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 4018dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public enum Method { 402ef698c852a24dfa42ab25054291286feb01ec871Paul Hawke GET, PUT, POST, DELETE, HEAD, OPTIONS; 4038dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 4048dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke static Method lookup(String method) { 4058dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke for (Method m : Method.values()) { 4068dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (m.toString().equalsIgnoreCase(method)) { 4078dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return m; 4088dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 4098dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 4108dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return null; 4118dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 4128dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 4138dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 4148dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 4158dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Pluggable strategy for asynchronously executing requests. 4168dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 4178dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public interface AsyncRunner { 4188dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke void exec(Runnable code); 4198dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 4208dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 4218dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 4225b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Factory to create temp file managers. 4235b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 4246200253df7b934c807eb3436b51a3b039406457dPaul Hawke public interface TempFileManagerFactory { 4256200253df7b934c807eb3436b51a3b039406457dPaul Hawke TempFileManager create(); 4266200253df7b934c807eb3436b51a3b039406457dPaul Hawke } 4276200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4288dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke // ------------------------------------------------------------------------------- // 4298dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 4305b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 4315b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Temp file manager. 4328dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p/> 4335b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * <p>Temp file managers are created 1-to-1 with incoming requests, to create and cleanup 4345b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * temporary files created as a result of handling the request.</p> 4355b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 4366200253df7b934c807eb3436b51a3b039406457dPaul Hawke public interface TempFileManager { 4376200253df7b934c807eb3436b51a3b039406457dPaul Hawke TempFile createTempFile() throws Exception; 4386200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4396200253df7b934c807eb3436b51a3b039406457dPaul Hawke void clear(); 4406200253df7b934c807eb3436b51a3b039406457dPaul Hawke } 4416200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4425b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 4435b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * A temp file. 4448dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p/> 4455b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * <p>Temp files are responsible for managing the actual temporary storage and cleaning 4465b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * themselves up when no longer needed.</p> 4475b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 4486200253df7b934c807eb3436b51a3b039406457dPaul Hawke public interface TempFile { 4496200253df7b934c807eb3436b51a3b039406457dPaul Hawke OutputStream open() throws Exception; 4506200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4516200253df7b934c807eb3436b51a3b039406457dPaul Hawke void delete() throws Exception; 4526200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4536200253df7b934c807eb3436b51a3b039406457dPaul Hawke String getName(); 4546200253df7b934c807eb3436b51a3b039406457dPaul Hawke } 4556200253df7b934c807eb3436b51a3b039406457dPaul Hawke 4569788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 4578dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Default threading strategy for NanoHttpd. 4588dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p/> 4598dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p>By default, the server spawns a new Thread for every incoming request. These are set 4608dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * to <i>daemon</i> status, and named according to the request number. The name is 4618dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * useful when profiling the application.</p> 4625b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 4638dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public static class DefaultAsyncRunner implements AsyncRunner { 4648dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private long requestCount; 4658dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 4665b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 4678dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public void exec(Runnable code) { 4688dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke ++requestCount; 4698dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke Thread t = new Thread(code); 4708dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke t.setDaemon(true); 4718dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke t.setName("NanoHttpd Request Processor (#" + requestCount + ")"); 4728dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke t.start(); 4735b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 4745b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 4755b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 4765b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 4775b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Default strategy for creating and cleaning up temporary files. 4788dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p/> 4795b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * <p></p>This class stores its files in the standard location (that is, 4805b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * wherever <code>java.io.tmpdir</code> points to). Files are added 4815b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * to an internal list, and deleted when no longer needed (that is, 4825b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * when <code>clear()</code> is invoked at the end of processing a 4835b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * request).</p> 4845b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 4855b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public static class DefaultTempFileManager implements TempFileManager { 4865b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke private final String tmpdir; 4875b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke private final List<TempFile> tempFiles; 4885b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 4895b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public DefaultTempFileManager() { 4905b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke tmpdir = System.getProperty("java.io.tmpdir"); 4915b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke tempFiles = new ArrayList<TempFile>(); 4925b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 4935b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 4945b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 4955b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public TempFile createTempFile() throws Exception { 4965b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke DefaultTempFile tempFile = new DefaultTempFile(tmpdir); 4975b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke tempFiles.add(tempFile); 4985b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke return tempFile; 4995b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5005b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5015b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 5025b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public void clear() { 5035b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke for (TempFile file : tempFiles) { 5045b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke try { 5055b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke file.delete(); 5065b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } catch (Exception ignored) { 5075b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5085b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5095b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke tempFiles.clear(); 5105b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5115b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5125b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5135b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 5145b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * Default strategy for creating and cleaning up temporary files. 5158dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * <p/> 5165b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * <p></p></[>By default, files are created by <code>File.createTempFile()</code> in 5175b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke * the directory specified.</p> 5185b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke */ 5195b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public static class DefaultTempFile implements TempFile { 5205b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke private File file; 5215b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke private OutputStream fstream; 5225b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5235b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public DefaultTempFile(String tempdir) throws IOException { 5245b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke file = File.createTempFile("NanoHTTPD-", "", new File(tempdir)); 5255b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke fstream = new FileOutputStream(file); 5265b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5275b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5285b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 5295b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public OutputStream open() throws Exception { 5305b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke return fstream; 5315b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5325b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5335b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 5345b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public void delete() throws Exception { 5359cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed safeClose(fstream); 5365b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke file.delete(); 5375b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5385b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5395b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke @Override 5405b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke public String getName() { 5415b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke return file.getAbsolutePath(); 5425b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5435b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke } 5445b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke 5455b4919a7af60ff39d1d47dea7e668fbde333dd84Paul Hawke /** 5469788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * HTTP response. Return one of these from serve(). 5479788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 5489788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public static class Response { 5499788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5509788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * HTTP status code after processing, e.g. "200 OK", HTTP_OK 5519788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 552c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke private Status status; 5539788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5549788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * MIME type of content, e.g. "text/html" 5559788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 556c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke private String mimeType; 5579788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5589788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Data of the response, may be null. 5599788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 5609d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private InputStream data; 5619788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5629788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Headers for the HTTP response. Use addHeader() to add lines. 5639788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 564c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke private Map<String, String> header = new HashMap<String, String>(); 5659d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke /** 5669d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke * The request method that spawned this response. 5679d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke */ 5689d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private Method requestMethod; 569517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel /** 570517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel * Use chunkedTransfer 571517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel */ 572517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel private boolean chunkedTransfer; 5738dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 5749788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5759788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Default constructor: response = HTTP_OK, mime = MIME_HTML and your supplied message 5769788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 5779788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public Response(String msg) { 5789788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this(Status.OK, MIME_HTML, msg); 5799788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 5809788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 5819788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5829788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Basic constructor. 5839788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 5849788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public Response(Status status, String mimeType, InputStream data) { 5859788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.status = status; 5869788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.mimeType = mimeType; 5879788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.data = data; 5889788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 5899788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 5909788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 5919788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Convenience method that makes an InputStream out of given text. 5929788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 5939788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public Response(Status status, String mimeType, String txt) { 5949788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.status = status; 5959788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.mimeType = mimeType; 5969788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke try { 5977162954f3d9927a455a9f6e806c8cecf5327dcc6Paul Hawke this.data = txt != null ? new ByteArrayInputStream(txt.getBytes("UTF-8")) : null; 5989788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } catch (java.io.UnsupportedEncodingException uee) { 5999788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke uee.printStackTrace(); 6009788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 6019788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 6029788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 6039788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 6049788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Adds given line to the header. 6059788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 6069788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public void addHeader(String name, String value) { 6079788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke header.put(name, value); 6089788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 6099788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 6109788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke /** 6117f0727787957c2f093412c01d165846842ec2425Paul Hawke * Sends given response to the socket. 6127f0727787957c2f093412c01d165846842ec2425Paul Hawke */ 6137f0727787957c2f093412c01d165846842ec2425Paul Hawke private void send(OutputStream outputStream) { 6147f0727787957c2f093412c01d165846842ec2425Paul Hawke String mime = mimeType; 6157f0727787957c2f093412c01d165846842ec2425Paul Hawke SimpleDateFormat gmtFrmt = new SimpleDateFormat("E, d MMM yyyy HH:mm:ss 'GMT'", Locale.US); 6167f0727787957c2f093412c01d165846842ec2425Paul Hawke gmtFrmt.setTimeZone(TimeZone.getTimeZone("GMT")); 6177f0727787957c2f093412c01d165846842ec2425Paul Hawke 6187f0727787957c2f093412c01d165846842ec2425Paul Hawke try { 6197f0727787957c2f093412c01d165846842ec2425Paul Hawke if (status == null) { 6207f0727787957c2f093412c01d165846842ec2425Paul Hawke throw new Error("sendResponse(): Status can't be null."); 6217f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6227f0727787957c2f093412c01d165846842ec2425Paul Hawke PrintWriter pw = new PrintWriter(outputStream); 6236b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke pw.print("HTTP/1.1 " + status.getDescription() + " \r\n"); 6247f0727787957c2f093412c01d165846842ec2425Paul Hawke 6257f0727787957c2f093412c01d165846842ec2425Paul Hawke if (mime != null) { 6267f0727787957c2f093412c01d165846842ec2425Paul Hawke pw.print("Content-Type: " + mime + "\r\n"); 6277f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6287f0727787957c2f093412c01d165846842ec2425Paul Hawke 6297f0727787957c2f093412c01d165846842ec2425Paul Hawke if (header == null || header.get("Date") == null) { 6307f0727787957c2f093412c01d165846842ec2425Paul Hawke pw.print("Date: " + gmtFrmt.format(new Date()) + "\r\n"); 6317f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6327f0727787957c2f093412c01d165846842ec2425Paul Hawke 6337f0727787957c2f093412c01d165846842ec2425Paul Hawke if (header != null) { 6347f0727787957c2f093412c01d165846842ec2425Paul Hawke for (String key : header.keySet()) { 6357f0727787957c2f093412c01d165846842ec2425Paul Hawke String value = header.get(key); 6367f0727787957c2f093412c01d165846842ec2425Paul Hawke pw.print(key + ": " + value + "\r\n"); 6377f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6387f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6397f0727787957c2f093412c01d165846842ec2425Paul Hawke 640a7a9272e5ecaad16422356a8be7ae46641351ba7Éric ALBER pw.print("Connection: keep-alive\r\n"); 641b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke 642517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel if (requestMethod != Method.HEAD && chunkedTransfer) { 643fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel sendAsChunked(outputStream, pw); 644fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } else { 645fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel sendAsFixedLength(outputStream, pw); 6467f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6477f0727787957c2f093412c01d165846842ec2425Paul Hawke outputStream.flush(); 6489cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed safeClose(data); 6497f0727787957c2f093412c01d165846842ec2425Paul Hawke } catch (IOException ioe) { 6507f0727787957c2f093412c01d165846842ec2425Paul Hawke // Couldn't write? No can do. 6517f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6527f0727787957c2f093412c01d165846842ec2425Paul Hawke } 6537f0727787957c2f093412c01d165846842ec2425Paul Hawke 654fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel private void sendAsChunked(OutputStream outputStream, PrintWriter pw) throws IOException { 655fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pw.print("Transfer-Encoding: chunked\r\n"); 656fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pw.print("\r\n"); 657fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pw.flush(); 658fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel int BUFFER_SIZE = 16 * 1024; 659fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel byte[] CRLF = "\r\n".getBytes(); 660fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel byte[] buff = new byte[BUFFER_SIZE]; 661fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel int read; 662fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel while ((read = data.read(buff)) > 0) { 663fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel outputStream.write(String.format("%x\r\n", read).getBytes()); 664fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel outputStream.write(buff, 0, read); 665fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel outputStream.write(CRLF); 666fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 667fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel outputStream.write(String.format("0\r\n\r\n").getBytes()); 668fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 669fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel 670fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel private void sendAsFixedLength(OutputStream outputStream, PrintWriter pw) throws IOException { 671fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel int pending = data != null ? data.available() : 0; // This is to support partial sends, see serveFile() 672fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pw.print("Content-Length: "+pending+"\r\n"); 673fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel 674fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pw.print("\r\n"); 675fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pw.flush(); 676fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel 677fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel if (requestMethod != Method.HEAD && data != null) { 678fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel int BUFFER_SIZE = 16 * 1024; 679fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel byte[] buff = new byte[BUFFER_SIZE]; 680fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel while (pending > 0) { 681fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel int read = data.read(buff, 0, ((pending > BUFFER_SIZE) ? BUFFER_SIZE : pending)); 682fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel if (read <= 0) { 683fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel break; 684fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 685fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel outputStream.write(buff, 0, read); 686fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel 687fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel pending -= read; 688fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 689fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 690fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel } 691fe1961e39809c90a76422332e91af67a5c65c31cSleekWeasel 692c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public Status getStatus() { 693c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke return status; 694c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 695c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 696c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public void setStatus(Status status) { 697c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke this.status = status; 698c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 699c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 700c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public String getMimeType() { 701c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke return mimeType; 702c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 703c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 704c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public void setMimeType(String mimeType) { 705c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke this.mimeType = mimeType; 706c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 707c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 708c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public InputStream getData() { 709c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke return data; 710c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 711c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 712c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public void setData(InputStream data) { 713c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke this.data = data; 714c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 715c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 716c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public Method getRequestMethod() { 717c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke return requestMethod; 718c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 719c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 720c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke public void setRequestMethod(Method requestMethod) { 721c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke this.requestMethod = requestMethod; 722c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke } 723c6f3f15bbf5a0ccbff5a4f17e53804c0d3888446Paul Hawke 724517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel public void setChunkedTransfer(boolean chunkedTransfer) { 725517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel this.chunkedTransfer = chunkedTransfer; 726517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel } 727517987ff0fe07af253c4cf6163ee5b9da17faf64SleekWeasel 7287f0727787957c2f093412c01d165846842ec2425Paul Hawke /** 7299788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke * Some HTTP response status codes 7309788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke */ 7319788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public enum Status { 7327162954f3d9927a455a9f6e806c8cecf5327dcc6Paul Hawke OK(200, "OK"), CREATED(201, "Created"), ACCEPTED(202, "Accepted"), NO_CONTENT(204, "No Content"), PARTIAL_CONTENT(206, "Partial Content"), REDIRECT(301, 7338dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke "Moved Permanently"), NOT_MODIFIED(304, "Not Modified"), BAD_REQUEST(400, "Bad Request"), UNAUTHORIZED(401, 734866241d4c51b84a7e4c000d76865f100b0878655synapticloop "Unauthorized"), FORBIDDEN(403, "Forbidden"), NOT_FOUND(404, "Not Found"), METHOD_NOT_ALLOWED(405, "Method Not Allowed"), RANGE_NOT_SATISFIABLE(416, 7358dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke "Requested Range Not Satisfiable"), INTERNAL_ERROR(500, "Internal Server Error"); 7369d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private final int requestStatus; 7379d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private final String description; 7389788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 7399d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke Status(int requestStatus, String description) { 7409788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke this.requestStatus = requestStatus; 7419d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke this.description = description; 7429788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 7439788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 7449788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public int getRequestStatus() { 7459788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke return this.requestStatus; 7469788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 7479788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke 7489788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke public String getDescription() { 7499d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke return "" + this.requestStatus + " " + description; 7509788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 7519788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 7529788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 753269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 7548dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public static final class ResponseException extends Exception { 7558dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 7568dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private final Response.Status status; 7578dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 7588dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public ResponseException(Response.Status status, String message) { 7598dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke super(message); 7608dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke this.status = status; 7618dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7628dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 7638dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public ResponseException(Response.Status status, String message, Exception e) { 7648dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke super(message, e); 7658dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke this.status = status; 7668dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7678dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 7688dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public Response.Status getStatus() { 7698dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return status; 7708dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7718dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7728dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 7738dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 7748dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Default strategy for creating and cleaning up temporary files. 7758dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 7768dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private class DefaultTempFileManagerFactory implements TempFileManagerFactory { 7778dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke @Override 7788dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public TempFileManager create() { 7798dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return new DefaultTempFileManager(); 7808dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7818dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 7828dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 783d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke /** 784d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke * Handles one session, i.e. parses the HTTP request and returns the response. 785d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 786249def0afcd823cd21315973bf197f2bef4a089fSleekWeasel public interface IHTTPSession { 787bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel void execute() throws IOException; 788bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 789bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel Map<String, String> getParms(); 790bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 791bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel Map<String, String> getHeaders(); 792bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 793bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel /** 794bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel * @return the path part of the URL. 795bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel */ 796bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel String getUri(); 797bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 798fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke String getQueryParameterString(); 799fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke 800bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel Method getMethod(); 801bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 802bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel InputStream getInputStream(); 803bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 804bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel CookieHandler getCookies(); 805bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 806bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel /** 807bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel * Adds the files in the request body to the files map. 808bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel * @arg files - map to modify 809bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel */ 810bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel void parseBody(Map<String, String> files) throws IOException, ResponseException; 811bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel } 812bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel 813249def0afcd823cd21315973bf197f2bef4a089fSleekWeasel protected class HTTPSession implements IHTTPSession { 814249def0afcd823cd21315973bf197f2bef4a089fSleekWeasel public static final int BUFSIZE = 8192; 8154e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke private final TempFileManager tempFileManager; 8169d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private final OutputStream outputStream; 8178dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private InputStream inputStream; 81893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private int splitbyte; 81993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private int rlen; 82093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private String uri; 82193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private Method method; 82293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private Map<String, String> parms; 82393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private Map<String, String> headers; 8246a28d482af06a03ea88a1b131d2eac3b8777badblordfokas private CookieHandler cookies; 825fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke private String queryParameterString; 8268dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 827249def0afcd823cd21315973bf197f2bef4a089fSleekWeasel public HTTPSession(TempFileManager tempFileManager, InputStream inputStream, OutputStream outputStream) { 8286200253df7b934c807eb3436b51a3b039406457dPaul Hawke this.tempFileManager = tempFileManager; 8297f0727787957c2f093412c01d165846842ec2425Paul Hawke this.inputStream = inputStream; 8306200253df7b934c807eb3436b51a3b039406457dPaul Hawke this.outputStream = outputStream; 831d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke } 832d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 8337e4e4ae652c755faea626f1e2538d495f96e648esynapticloop public HTTPSession(TempFileManager tempFileManager, InputStream inputStream, OutputStream outputStream, InetAddress inetAddress) { 8347e4e4ae652c755faea626f1e2538d495f96e648esynapticloop this.tempFileManager = tempFileManager; 8357e4e4ae652c755faea626f1e2538d495f96e648esynapticloop this.inputStream = inputStream; 8367e4e4ae652c755faea626f1e2538d495f96e648esynapticloop this.outputStream = outputStream; 8377e4e4ae652c755faea626f1e2538d495f96e648esynapticloop String remoteIp = inetAddress.isLoopbackAddress() || inetAddress.isAnyLocalAddress() ? "127.0.0.1" : inetAddress.getHostAddress().toString(); 8387e4e4ae652c755faea626f1e2538d495f96e648esynapticloop headers = new HashMap<String, String>(); 8397e4e4ae652c755faea626f1e2538d495f96e648esynapticloop 8407e4e4ae652c755faea626f1e2538d495f96e648esynapticloop headers.put("remote-addr", remoteIp); 8417e4e4ae652c755faea626f1e2538d495f96e648esynapticloop headers.put("http-client-ip", remoteIp); 8427e4e4ae652c755faea626f1e2538d495f96e648esynapticloop } 8437e4e4ae652c755faea626f1e2538d495f96e648esynapticloop 844bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 84567708a714f4c7883a0f623272383d1e9be3f4de9douzifly public void execute() throws IOException { 846d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke try { 847d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Read the first 8192 bytes. 848d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // The full header should fit in here. 849d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Apache's default header limit is 8KB. 850d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Do NOT assume that a single read will get the entire header at once! 851745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed byte[] buf = new byte[BUFSIZE]; 852745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed splitbyte = 0; 853745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed rlen = 0; 854d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke { 855b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursi int read = -1; 856b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursi try { 857b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursi read = inputStream.read(buf, 0, BUFSIZE); 85868394ef81a80928d25e584475be59a2e937a5ae0Paul Hawke } catch (Exception e) { 85968394ef81a80928d25e584475be59a2e937a5ae0Paul Hawke safeClose(inputStream); 86068394ef81a80928d25e584475be59a2e937a5ae0Paul Hawke safeClose(outputStream); 861b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursi throw new SocketException("NanoHttpd Shutdown"); 862b01d2c9086c3df2b63bd1eca602320cd7f45d24cSherif Moursi } 8638dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (read == -1) { 86467708a714f4c7883a0f623272383d1e9be3f4de9douzifly // socket was been closed 86568394ef81a80928d25e584475be59a2e937a5ae0Paul Hawke safeClose(inputStream); 86668394ef81a80928d25e584475be59a2e937a5ae0Paul Hawke safeClose(outputStream); 8678195cac53b34938cb9ac6732be91b0e1fbea2193Paul Hawke throw new SocketException("NanoHttpd Shutdown"); 86867708a714f4c7883a0f623272383d1e9be3f4de9douzifly } 869d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke while (read > 0) { 870d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke rlen += read; 871d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke splitbyte = findHeaderEnd(buf, rlen); 872d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke if (splitbyte > 0) 873d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke break; 8747f0727787957c2f093412c01d165846842ec2425Paul Hawke read = inputStream.read(buf, rlen, BUFSIZE - rlen); 875d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke } 876d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke } 877d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 878745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed if (splitbyte < rlen) { 879745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed ByteArrayInputStream splitInputStream = new ByteArrayInputStream(buf, splitbyte, rlen - splitbyte); 880745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed SequenceInputStream sequenceInputStream = new SequenceInputStream(splitInputStream, inputStream); 881745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed inputStream = sequenceInputStream; 882745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed } 883745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed 88493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed parms = new HashMap<String, String>(); 8857e4e4ae652c755faea626f1e2538d495f96e648esynapticloop if(null == headers) { 8867e4e4ae652c755faea626f1e2538d495f96e648esynapticloop headers = new HashMap<String, String>(); 8877e4e4ae652c755faea626f1e2538d495f96e648esynapticloop } 88893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 889d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Create a BufferedReader for parsing the header. 8907f0727787957c2f093412c01d165846842ec2425Paul Hawke BufferedReader hin = new BufferedReader(new InputStreamReader(new ByteArrayInputStream(buf, 0, rlen))); 891d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 892d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Decode the header into parms and header java properties 89393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed Map<String, String> pre = new HashMap<String, String>(); 8948af508ee58e0abca6dbb6c30532410a1dd99866bMartin M Reed decodeHeader(hin, pre, parms, headers); 89593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 89693441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed method = Method.lookup(pre.get("method")); 8970cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (method == null) { 898964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.BAD_REQUEST, "BAD REQUEST: Syntax error."); 899d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke } 90093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 90193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed uri = pre.get("uri"); 9028dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 9036a28d482af06a03ea88a1b131d2eac3b8777badblordfokas cookies = new CookieHandler(headers); 90493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 90593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed // Ok, now do the serve() 90693441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed Response r = serve(this); 90793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed if (r == null) { 90893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed throw new ResponseException(Response.Status.INTERNAL_ERROR, "SERVER INTERNAL ERROR: Serve() returned a null response."); 90993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } else { 9108dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke cookies.unloadQueue(r); 91193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed r.setRequestMethod(method); 91293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed r.send(outputStream); 91393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 9148dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } catch (SocketException e) { 91567708a714f4c7883a0f623272383d1e9be3f4de9douzifly // throw it out to close socket object (finalAccept) 91667708a714f4c7883a0f623272383d1e9be3f4de9douzifly throw e; 917895b2e657db482a93d6ee2c4e3b3fdaddec1278bWorxFR } catch (SocketTimeoutException ste) { 918895b2e657db482a93d6ee2c4e3b3fdaddec1278bWorxFR throw ste; 91993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } catch (IOException ioe) { 92093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed Response r = new Response(Response.Status.INTERNAL_ERROR, MIME_PLAINTEXT, "SERVER INTERNAL ERROR: IOException: " + ioe.getMessage()); 92193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed r.send(outputStream); 9226b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke safeClose(outputStream); 92393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } catch (ResponseException re) { 92493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed Response r = new Response(re.getStatus(), MIME_PLAINTEXT, re.getMessage()); 92593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed r.send(outputStream); 9266b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke safeClose(outputStream); 927ac42f7c14c9da6efb92e38270c9cc44879913ee1Martin M Reed } finally { 928ac42f7c14c9da6efb92e38270c9cc44879913ee1Martin M Reed tempFileManager.clear(); 92993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 93093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 93193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 932bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 933bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel public void parseBody(Map<String, String> files) throws IOException, ResponseException { 93493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed RandomAccessFile randomAccessFile = null; 93593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed BufferedReader in = null; 93693441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed try { 93793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 938412e2e136193da510e859f73f9ff2ac21565df69james.mcclure randomAccessFile = getTmpBucket(); 939d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 940745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed long size; 941745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed if (headers.containsKey("content-length")) { 942745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed size = Integer.parseInt(headers.get("content-length")); 943745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed } else if (splitbyte < rlen) { 944745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed size = rlen - splitbyte; 945745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed } else { 946d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke size = 0; 9470cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 948d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke 949d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke // Now read all the body and write it to f 950745614b606ebcca5f69f5c84bde1774ca2f11cc5Martin M Reed byte[] buf = new byte[512]; 951d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke while (rlen >= 0 && size > 0) { 9527f0727787957c2f093412c01d165846842ec2425Paul Hawke rlen = inputStream.read(buf, 0, 512); 953269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke size -= rlen; 9540cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (rlen > 0) { 955412e2e136193da510e859f73f9ff2ac21565df69james.mcclure randomAccessFile.write(buf, 0, rlen); 9560cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 957269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 958269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 959269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Get the raw body as a byte [] 960412e2e136193da510e859f73f9ff2ac21565df69james.mcclure ByteBuffer fbuf = randomAccessFile.getChannel().map(FileChannel.MapMode.READ_ONLY, 0, randomAccessFile.length()); 961412e2e136193da510e859f73f9ff2ac21565df69james.mcclure randomAccessFile.seek(0); 962269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 963269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Create a BufferedReader for easily reading it as string. 964412e2e136193da510e859f73f9ff2ac21565df69james.mcclure InputStream bin = new FileInputStream(randomAccessFile.getFD()); 96593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed in = new BufferedReader(new InputStreamReader(bin)); 966269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 967269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // If the method is POST, there may be parameters 968269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // in data section, too, read it: 969df674ca7f11cee6336b66271c50476f389e0ff4bPaul Hawke if (Method.POST.equals(method)) { 970269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String contentType = ""; 9718af508ee58e0abca6dbb6c30532410a1dd99866bMartin M Reed String contentTypeHeader = headers.get("content-type"); 9722b54eda5fee8430268a99d73b061e790362db544Tom Hermann 9732c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke StringTokenizer st = null; 9742c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke if (contentTypeHeader != null) { 9752c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke st = new StringTokenizer(contentTypeHeader, ",; "); 9762c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke if (st.hasMoreTokens()) { 9772c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke contentType = st.nextToken(); 9782c48103a2fc2ef4ccbd5d441da96dc77c7736109Paul Hawke } 979269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 980269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 9810277bbd03983f15110fcf0d039a3e276d0808e29Paul Hawke if ("multipart/form-data".equalsIgnoreCase(contentType)) { 982269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Handle multipart/form-data 9830cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (!st.hasMoreTokens()) { 984964393fd7e5ed49088882a126cf82507184467efMartin 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"); 9850cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 9862b54eda5fee8430268a99d73b061e790362db544Tom Hermann 9874e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke String boundaryStartString = "boundary="; 9882b54eda5fee8430268a99d73b061e790362db544Tom Hermann int boundaryContentStart = contentTypeHeader.indexOf(boundaryStartString) + boundaryStartString.length(); 9894e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke String boundary = contentTypeHeader.substring(boundaryContentStart, contentTypeHeader.length()); 990220e1a21e7bbb831d06551c72799dfedc1db979fPaul Hawke if (boundary.startsWith("\"") && boundary.endsWith("\"")) { 9914e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke boundary = boundary.substring(1, boundary.length() - 1); 9924e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke } 993269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 994269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke decodeMultipartData(boundary, fbuf, in, parms, files); 995269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } else { 996269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Handle application/x-www-form-urlencoded 997269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String postLine = ""; 998269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke char pbuf[] = new char[512]; 999269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int read = in.read(pbuf); 1000269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (read >= 0 && !postLine.endsWith("\r\n")) { 1001269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke postLine += String.valueOf(pbuf, 0, read); 1002269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke read = in.read(pbuf); 1003269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1004269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke postLine = postLine.trim(); 1005269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke decodeParms(postLine, parms); 1006269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1007b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke } else if (Method.PUT.equals(method)) { 1008dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke files.put("content", saveTmpFile(fbuf, 0, fbuf.limit())); 1009b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke } 10104e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke } finally { 10119cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed safeClose(randomAccessFile); 101293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed safeClose(in); 1013269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1014269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1015269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1016269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 10177f0727787957c2f093412c01d165846842ec2425Paul Hawke * Decodes the sent headers and loads the data into Key/value pairs 1018d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 10198af508ee58e0abca6dbb6c30532410a1dd99866bMartin M Reed private void decodeHeader(BufferedReader in, Map<String, String> pre, Map<String, String> parms, Map<String, String> headers) 10208dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke throws ResponseException { 1021269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke try { 1022269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Read the request line 1023269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String inLine = in.readLine(); 10240cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (inLine == null) { 1025269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return; 10260cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 10270cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke 1028269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke StringTokenizer st = new StringTokenizer(inLine); 10290cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (!st.hasMoreTokens()) { 1030964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.BAD_REQUEST, "BAD REQUEST: Syntax error. Usage: GET /example/file.html"); 10310cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 1032269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 10330277bbd03983f15110fcf0d039a3e276d0808e29Paul Hawke pre.put("method", st.nextToken()); 1034269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 10350cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke if (!st.hasMoreTokens()) { 1036964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.BAD_REQUEST, "BAD REQUEST: Missing URI. Usage: GET /example/file.html"); 10370cb0846c31d1eaf3fcd5d2382ca595abfb2dda38Paul Hawke } 1038269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1039269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String uri = st.nextToken(); 1040269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1041269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // Decode parameters from the URI 1042269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int qmi = uri.indexOf('?'); 1043269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (qmi >= 0) { 1044269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke decodeParms(uri.substring(qmi + 1), parms); 1045269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke uri = decodePercent(uri.substring(0, qmi)); 10469788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } else { 1047269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke uri = decodePercent(uri); 10489788c5b7900b4c7ff27355f69fee4e14630eb412Paul Hawke } 1049269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1050269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // If there's another token, it's protocol version, 1051269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // followed by HTTP headers. Ignore version but parse headers. 1052269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // NOTE: this now forces header names lowercase since they are 1053269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke // case insensitive and vary by client. 1054269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (st.hasMoreTokens()) { 1055269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String line = in.readLine(); 1056269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (line != null && line.trim().length() > 0) { 1057269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int p = line.indexOf(':'); 1058269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (p >= 0) 1059eb71c8999ff29f7dc3fe8dcdba81084f7f6a5facDavid Masover headers.put(line.substring(0, p).trim().toLowerCase(Locale.US), line.substring(p + 1).trim()); 1060269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke line = in.readLine(); 1061269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1062269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1063269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1064269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke pre.put("uri", uri); 1065269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } catch (IOException ioe) { 1066964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.INTERNAL_ERROR, "SERVER INTERNAL ERROR: IOException: " + ioe.getMessage(), ioe); 1067269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1068269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1069269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1070269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 10717f0727787957c2f093412c01d165846842ec2425Paul Hawke * Decodes the Multipart Body data and put it into Key/Value pairs. 1072d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 1073dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke private void decodeMultipartData(String boundary, ByteBuffer fbuf, BufferedReader in, Map<String, String> parms, 1074964393fd7e5ed49088882a126cf82507184467efMartin M Reed Map<String, String> files) throws ResponseException { 1075269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke try { 1076269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int[] bpositions = getBoundaryPositions(fbuf, boundary.getBytes()); 1077269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int boundarycount = 1; 1078269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String mpline = in.readLine(); 1079269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (mpline != null) { 10807b88819e82b89ac3476ce060903468076a73de8fPaul Hawke if (!mpline.contains(boundary)) { 1081964393fd7e5ed49088882a126cf82507184467efMartin 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"); 10827b88819e82b89ac3476ce060903468076a73de8fPaul Hawke } 1083269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke boundarycount++; 1084269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke Map<String, String> item = new HashMap<String, String>(); 1085269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke mpline = in.readLine(); 1086269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (mpline != null && mpline.trim().length() > 0) { 1087269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int p = mpline.indexOf(':'); 1088269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (p != -1) { 1089eb71c8999ff29f7dc3fe8dcdba81084f7f6a5facDavid Masover item.put(mpline.substring(0, p).trim().toLowerCase(Locale.US), mpline.substring(p + 1).trim()); 1090269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1091269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke mpline = in.readLine(); 1092269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1093269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (mpline != null) { 1094269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String contentDisposition = item.get("content-disposition"); 1095269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (contentDisposition == null) { 1096964393fd7e5ed49088882a126cf82507184467efMartin 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"); 1097269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1098269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke StringTokenizer st = new StringTokenizer(contentDisposition, "; "); 1099269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke Map<String, String> disposition = new HashMap<String, String>(); 1100269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (st.hasMoreTokens()) { 1101269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String token = st.nextToken(); 1102269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int p = token.indexOf('='); 1103269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (p != -1) { 1104eb71c8999ff29f7dc3fe8dcdba81084f7f6a5facDavid Masover disposition.put(token.substring(0, p).trim().toLowerCase(Locale.US), token.substring(p + 1).trim()); 1105269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1106269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1107269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String pname = disposition.get("name"); 1108269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke pname = pname.substring(1, pname.length() - 1); 1109269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1110269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String value = ""; 1111269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (item.get("content-type") == null) { 11127b88819e82b89ac3476ce060903468076a73de8fPaul Hawke while (mpline != null && !mpline.contains(boundary)) { 1113269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke mpline = in.readLine(); 1114269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (mpline != null) { 1115269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int d = mpline.indexOf(boundary); 1116269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (d == -1) { 1117269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke value += mpline; 1118269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } else { 1119269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke value += mpline.substring(0, d - 2); 1120269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1121269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1122269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1123269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } else { 1124269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (boundarycount > bpositions.length) { 1125964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.INTERNAL_ERROR, "Error processing request"); 1126269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1127269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int offset = stripMultipartHeaders(fbuf, bpositions[boundarycount - 2]); 1128269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String path = saveTmpFile(fbuf, offset, bpositions[boundarycount - 1] - offset - 4); 1129269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke files.put(pname, path); 1130269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke value = disposition.get("filename"); 1131269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke value = value.substring(1, value.length() - 1); 1132269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke do { 1133269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke mpline = in.readLine(); 11347b88819e82b89ac3476ce060903468076a73de8fPaul Hawke } while (mpline != null && !mpline.contains(boundary)); 1135269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1136269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke parms.put(pname, value); 1137269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1138269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1139269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } catch (IOException ioe) { 1140964393fd7e5ed49088882a126cf82507184467efMartin M Reed throw new ResponseException(Response.Status.INTERNAL_ERROR, "SERVER INTERNAL ERROR: IOException: " + ioe.getMessage(), ioe); 1141269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1142269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1143269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1144269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 1145269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * Find byte index separating header from body. It must be the last byte of the first two sequential new lines. 1146d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 1147269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke private int findHeaderEnd(final byte[] buf, int rlen) { 1148269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int splitbyte = 0; 1149269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke while (splitbyte + 3 < rlen) { 1150269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (buf[splitbyte] == '\r' && buf[splitbyte + 1] == '\n' && buf[splitbyte + 2] == '\r' && buf[splitbyte + 3] == '\n') { 1151269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return splitbyte + 4; 1152269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1153269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke splitbyte++; 1154269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1155269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return 0; 1156269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1157269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1158269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 1159269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * Find the byte positions where multipart boundaries start. 1160d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 116193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed private int[] getBoundaryPositions(ByteBuffer b, byte[] boundary) { 1162269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int matchcount = 0; 1163269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int matchbyte = -1; 1164269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke List<Integer> matchbytes = new ArrayList<Integer>(); 11658dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke for (int i = 0; i < b.limit(); i++) { 1166dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke if (b.get(i) == boundary[matchcount]) { 1167269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (matchcount == 0) 1168269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchbyte = i; 1169269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchcount++; 1170269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (matchcount == boundary.length) { 1171269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchbytes.add(matchbyte); 1172269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchcount = 0; 1173269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchbyte = -1; 1174269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1175269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } else { 1176269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke i -= matchcount; 1177269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchcount = 0; 1178269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke matchbyte = -1; 1179269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1180269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1181269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int[] ret = new int[matchbytes.size()]; 1182269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke for (int i = 0; i < ret.length; i++) { 11837b88819e82b89ac3476ce060903468076a73de8fPaul Hawke ret[i] = matchbytes.get(i); 1184269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1185269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return ret; 1186269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1187269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1188269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 1189269baff79590de5abd70e8e05bf46547e4a28ee6Micah 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. 1190d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke */ 11918dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private String saveTmpFile(ByteBuffer b, int offset, int len) { 1192269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke String path = ""; 1193269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke if (len > 0) { 11946b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke FileOutputStream fileOutputStream = null; 1195269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke try { 11964e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke TempFile tempFile = tempFileManager.createTempFile(); 1197dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke ByteBuffer src = b.duplicate(); 11986b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke fileOutputStream = new FileOutputStream(tempFile.getName()); 11996b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke FileChannel dest = fileOutputStream.getChannel(); 1200dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke src.position(offset).limit(offset + len); 1201dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke dest.write(src.slice()); 12024e1ae19daee9a8e9437476af6b07a6aba950802cPaul Hawke path = tempFile.getName(); 1203269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } catch (Exception e) { // Catch exception if any 1204d22b56d68169cbf5fcf9ddfb30b54fb5f155cc40Paul Hawke System.err.println("Error: " + e.getMessage()); 1205062b50e32b22f4c419bb4b8e3d781b9eac056380Martin M Reed } finally { 12066b018cee02f3517742880f80c293f5ae0a905b87Paul Hawke safeClose(fileOutputStream); 1207269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1208269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1209269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return path; 1210269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1211269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 12129d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private RandomAccessFile getTmpBucket() { 1213dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke try { 1214dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke TempFile tempFile = tempFileManager.createTempFile(); 1215dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke return new RandomAccessFile(tempFile.getName(), "rw"); 1216dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke } catch (Exception e) { 1217dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke System.err.println("Error: " + e.getMessage()); 1218dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke } 1219dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke return null; 1220dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke } 1221dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke 1222269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 1223269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke * It returns the offset separating multipart file headers from the file's data. 1224269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 1225dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke private int stripMultipartHeaders(ByteBuffer b, int offset) { 1226269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke int i; 12278dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke for (i = offset; i < b.limit(); i++) { 1228dfea30a36949bc24a698ac39d8a714fc8f1c9e82Paul Hawke if (b.get(i) == '\r' && b.get(++i) == '\n' && b.get(++i) == '\r' && b.get(++i) == '\n') { 1229269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke break; 1230269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1231269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1232269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return i + 1; 1233269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1234269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1235269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke /** 12367f0727787957c2f093412c01d165846842ec2425Paul Hawke * Decodes parameters in percent-encoded URI-format ( e.g. "name=Jack%20Daniels&pass=Single%20Malt" ) and 12377f0727787957c2f093412c01d165846842ec2425Paul Hawke * adds them to given Map. NOTE: this doesn't support multiple identical keys due to the simplicity of Map. 1238269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke */ 12399d9bd98292f8f38cbd3450d1abe94009b281b6e7Paul Hawke private void decodeParms(String parms, Map<String, String> p) { 1240f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke if (parms == null) { 1241fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke queryParameterString = ""; 1242269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke return; 1243f29e141959a4bbb738453c3462518e52ebb218cePaul Hawke } 1244269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke 1245fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke queryParameterString = parms; 1246269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke StringTokenizer st = new StringTokenizer(parms, "&"); 1247b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke while (st.hasMoreTokens()) { 1248b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke String e = st.nextToken(); 1249b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke int sep = e.indexOf('='); 1250b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke if (sep >= 0) { 1251b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke p.put(decodePercent(e.substring(0, sep)).trim(), 12528dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke decodePercent(e.substring(sep + 1))); 1253b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke } else { 1254b814145cf66e94b7ce5511077ca0ebcff6e67268Paul Hawke p.put(decodePercent(e).trim(), ""); 1255269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1256269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 1257269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke } 125893441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 1259bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 126093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed public final Map<String, String> getParms() { 126193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed return parms; 126293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 126393441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 1264fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke public String getQueryParameterString() { 1265fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke return queryParameterString; 1266fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke } 1267fbe92e0eb77710addafab0d48e7be5e147f17d00Paul Hawke 1268bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 126993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed public final Map<String, String> getHeaders() { 127093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed return headers; 127193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 127293441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 1273bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 127493441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed public final String getUri() { 127593441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed return uri; 127693441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 127793441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed 1278bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 127993441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed public final Method getMethod() { 128093441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed return method; 128193441a0455c683a17622ec2da6768c326e6d7a48Martin M Reed } 12828cdc631dca8d05f64ea8da466ecfe759a2f6e429Martin M Reed 1283bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 12848cdc631dca8d05f64ea8da466ecfe759a2f6e429Martin M Reed public final InputStream getInputStream() { 12858cdc631dca8d05f64ea8da466ecfe759a2f6e429Martin M Reed return inputStream; 12868cdc631dca8d05f64ea8da466ecfe759a2f6e429Martin M Reed } 1287964393fd7e5ed49088882a126cf82507184467efMartin M Reed 1288bc0de3e38a4edadd6704a6c8ccb0c5ee738b8f54SleekWeasel @Override 12898dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public CookieHandler getCookies() { 12908dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return cookies; 12918dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 12928dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 1293964393fd7e5ed49088882a126cf82507184467efMartin M Reed 12945820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public static class Cookie { 12955820f308631a825a05feddddd0aafb962f272b9bPaul Hawke private String n, v, e; 12965820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 12975820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public Cookie(String name, String value, String expires) { 12985820f308631a825a05feddddd0aafb962f272b9bPaul Hawke n = name; 12995820f308631a825a05feddddd0aafb962f272b9bPaul Hawke v = value; 13005820f308631a825a05feddddd0aafb962f272b9bPaul Hawke e = expires; 13015820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13025820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13035820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public Cookie(String name, String value) { 13045820f308631a825a05feddddd0aafb962f272b9bPaul Hawke this(name, value, 30); 13055820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13065820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13075820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public Cookie(String name, String value, int numDays) { 13085820f308631a825a05feddddd0aafb962f272b9bPaul Hawke n = name; 13095820f308631a825a05feddddd0aafb962f272b9bPaul Hawke v = value; 13105820f308631a825a05feddddd0aafb962f272b9bPaul Hawke e = getHTTPTime(numDays); 13115820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13125820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13135820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public String getHTTPHeader() { 13145820f308631a825a05feddddd0aafb962f272b9bPaul Hawke String fmt = "%s=%s; expires=%s"; 13155820f308631a825a05feddddd0aafb962f272b9bPaul Hawke return String.format(fmt, n, v, e); 13165820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13175820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13185820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public static String getHTTPTime(int days) { 13195820f308631a825a05feddddd0aafb962f272b9bPaul Hawke Calendar calendar = Calendar.getInstance(); 13205820f308631a825a05feddddd0aafb962f272b9bPaul Hawke SimpleDateFormat dateFormat = new SimpleDateFormat("EEE, dd MMM yyyy HH:mm:ss z", Locale.US); 13215820f308631a825a05feddddd0aafb962f272b9bPaul Hawke dateFormat.setTimeZone(TimeZone.getTimeZone("GMT")); 13225820f308631a825a05feddddd0aafb962f272b9bPaul Hawke calendar.add(Calendar.DAY_OF_MONTH, days); 13235820f308631a825a05feddddd0aafb962f272b9bPaul Hawke return dateFormat.format(calendar.getTime()); 13245820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13255820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13265820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13278dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 13288dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Provides rudimentary support for cookies. 13298dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Doesn't support 'path', 'secure' nor 'httpOnly'. 13308dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Feel free to improve it and/or add unsupported features. 13318dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 13328dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @author LordFokas 13338dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 13345820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public class CookieHandler implements Iterable<String> { 13358dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private HashMap<String, String> cookies = new HashMap<String, String>(); 13368dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke private ArrayList<Cookie> queue = new ArrayList<Cookie>(); 13378dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke 13388dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public CookieHandler(Map<String, String> httpHeaders) { 13398dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke String raw = httpHeaders.get("cookie"); 13408dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (raw != null) { 13418dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke String[] tokens = raw.split(";"); 13428dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke for (String token : tokens) { 13438dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke String[] data = token.trim().split("="); 13448dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke if (data.length == 2) { 13458dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke cookies.put(data[0], data[1]); 13468dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 13478dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 13488dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke } 1349964393fd7e5ed49088882a126cf82507184467efMartin M Reed } 1350964393fd7e5ed49088882a126cf82507184467efMartin M Reed 13515820f308631a825a05feddddd0aafb962f272b9bPaul Hawke @Override public Iterator<String> iterator() { 13525820f308631a825a05feddddd0aafb962f272b9bPaul Hawke return cookies.keySet().iterator(); 13535820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13545820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13558dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 13568dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Read a cookie from the HTTP Headers. 13578dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 13588dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param name The cookie's name. 13598dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @return The cookie's value if it exists, null otherwise. 13608dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 13618dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public String read(String name) { 13628dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke return cookies.get(name); 1363964393fd7e5ed49088882a126cf82507184467efMartin M Reed } 1364964393fd7e5ed49088882a126cf82507184467efMartin M Reed 13658dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 13668dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Sets a cookie. 13678dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 13688dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param name The cookie's name. 13698dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param value The cookie's value. 13708dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param expires How many days until the cookie expires. 13718dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 13728dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public void set(String name, String value, int expires) { 13735820f308631a825a05feddddd0aafb962f272b9bPaul Hawke queue.add(new Cookie(name, value, Cookie.getHTTPTime(expires))); 13745820f308631a825a05feddddd0aafb962f272b9bPaul Hawke } 13755820f308631a825a05feddddd0aafb962f272b9bPaul Hawke 13765820f308631a825a05feddddd0aafb962f272b9bPaul Hawke public void set(Cookie cookie) { 13775820f308631a825a05feddddd0aafb962f272b9bPaul Hawke queue.add(cookie); 1378964393fd7e5ed49088882a126cf82507184467efMartin M Reed } 1379964393fd7e5ed49088882a126cf82507184467efMartin M Reed 13808dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 13818dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Set a cookie with an expiration date from a month ago, effectively deleting it on the client side. 13828dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 13838dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param name The cookie name. 13848dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 13858dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public void delete(String name) { 13868dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke set(name, "-delete-", -30); 13879cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed } 13889cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed 13898dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke /** 13908dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * Internally used by the webserver to add all queued cookies into the Response's HTTP Headers. 13918dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * 13928dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke * @param response The Response object to which headers the queued cookies will be added. 13938dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke */ 13948dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke public void unloadQueue(Response response) { 13958dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke for (Cookie cookie : queue) { 13968dd183554e56783ecf7cd6b8529cf807b9eb813cPaul Hawke response.addHeader("Set-Cookie", cookie.getHTTPHeader()); 13979cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed } 13989cd5d3b4438667394ab98895c75b4e1a2f8e76a0Martin M Reed } 13997e423dde0fe863fadbd363de8a94746a30fca0eeMartin M Reed } 1400269baff79590de5abd70e8e05bf46547e4a28ee6Micah Hainline and Paul Hawke} 1401