Added initial SCGI server and a handler for serving JARs from the filesystem.

[jsvc.git] / src / dolda / jsvc / scgi / LimitInputStream.java

diff --git a/src/dolda/jsvc/scgi/LimitInputStream.java b/src/dolda/jsvc/scgi/LimitInputStream.java
new file mode 100644 (file)
index 0000000..7a37f25
--- /dev/null
+++ b/src/dolda/jsvc/scgi/LimitInputStream.java
@@ -0,0 +1,65 @@
+package dolda.jsvc.scgi;
+
+import java.io.*;
+
+public class LimitInputStream extends InputStream {
+    private final InputStream bk;
+    private final long limit;
+    private long read;
+    
+    public LimitInputStream(InputStream bk, long limit) {
+       this.bk = bk;
+       this.limit = limit;
+    }
+    
+    public void close() throws IOException {
+       bk.close();
+    }
+    
+    public int available() throws IOException {
+       int av = bk.available();
+       synchronized(this) {
+           if(av > limit - read)
+               av = (int)(limit - read);
+           return(av);
+       }
+    }
+    
+    public int read() throws IOException {
+       synchronized(this) {
+           if(read >= limit)
+               return(-1);
+           int ret = bk.read();
+           if(ret >= 0)
+               read++;
+           return(ret);
+       }
+    }
+    
+    public int read(byte[] b) throws IOException {
+       return(read(b, 0, b.length));
+    }
+    
+    public int read(byte[] b, int off, int len) throws IOException {
+       synchronized(this) {
+           if(read >= limit)
+               return(-1);
+           if(len > limit - read)
+               len = (int)(limit - read);
+           int ret = bk.read(b, off, len);
+           if(ret > 0)
+               read += ret;
+           return(ret);
+       }
+    }
+    
+    public long skip(long n) throws IOException {
+       synchronized(this) {
+           if(n > limit - read)
+               n = limit - read;
+           long ret = bk.skip(n);
+           read += ret;
+           return(ret);
+       }
+    }
+}