1 /* 2 * Copyright (c) 2004, 2011, Oracle and/or its affiliates. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 8 * - Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 11 * - Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * - Neither the name of Oracle nor the names of its 16 * contributors may be used to endorse or promote products derived 17 * from this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS 20 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, 21 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR 23 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 24 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 25 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 26 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 27 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 28 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 29 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32 /* 33 * This source code is provided to illustrate the usage of a given feature 34 * or technique and has been deliberately simplified. Additional steps 35 * required for a production-quality application, such as security checks, 36 * input validation and proper error handling, might not be present in 37 * this sample code. 38 */ 39 40 41 import java.io.*; 42 import java.net.*; 43 import java.nio.channels.*; 44 import java.nio.charset.*; 45 46 /** 47 * A Content type that provides for transferring files. 48 * 49 * @author Mark Reinhold 50 * @author Brad R. Wetmore 51 */ 52 class FileContent implements Content { 53 54 private static File ROOT = new File("root"); 55 56 private File fn; 57 58 FileContent(URI uri) { 59 fn = new File(ROOT, 60 uri.getPath() 61 .replace('/', 62 File.separatorChar)); 63 } 64 65 private String type = null; 66 67 public String type() { 68 if (type != null) 69 return type; 70 String nm = fn.getName(); 71 if (nm.endsWith(".html")) 72 type = "text/html; charset=iso-8859-1"; 73 else if ((nm.indexOf('.') < 0) || nm.endsWith(".txt")) 74 type = "text/plain; charset=iso-8859-1"; 75 else 76 type = "application/octet-stream"; 77 return type; 78 } 79 80 private FileChannel fc = null; 81 private long length = -1; 82 private long position = -1; // NB only; >= 0 if transferring 83 84 public long length() { 85 return length; 86 } 87 88 public void prepare() throws IOException { 89 if (fc == null) 90 fc = new RandomAccessFile(fn, "r").getChannel(); 91 length = fc.size(); 92 position = 0; // NB only 93 } 94 95 public boolean send(ChannelIO cio) throws IOException { 96 if (fc == null) 97 throw new IllegalStateException(); 98 if (position < 0) // NB only 99 throw new IllegalStateException(); 100 101 /* 102 * Short-circuit if we're already done. 103 */ 104 if (position >= length) { 105 return false; 106 } 107 108 position += cio.transferTo(fc, position, length - position); 109 return (position < length); 110 } 111 112 public void release() throws IOException { 113 if (fc != null) { 114 fc.close(); 115 fc = null; 116 } 117 } 118 } 119