Added event-driven server.

[jagi.git] / src / jagi / scgi / Scgi.java

package jagi.scgi;

import jagi.*;
import java.util.*;
import java.io.*;
import java.nio.*;
import java.nio.channels.*;

public class Scgi {
    public static ByteBuffer readns(ReadableByteChannel sk) throws IOException {
	int hln = 0;
	while(true) {
	    int c = Utils.read(sk);
	    if(c == ':')
		break;
	    else if((c >= '0') && (c <= '9'))
		hln = (hln * 10) + (c - '0');
	    else if(c < 0)
		throw(new IOException("unexpected eof in netstring header"));
	    else
		throw(new IOException("invalid netstring length byte: " + (c & 0xff)));
	}
	ByteBuffer data = Utils.readall(sk, ByteBuffer.allocate(hln));
	if(Utils.read(sk) != ',')
	    throw(new IOException("non-terminated netstring"));
	return(data);
    }

    public static Map<ByteBuffer, ByteBuffer> splithead(ByteBuffer ns) {
	Map<ByteBuffer, ByteBuffer> ret = new HashMap<>();
	ByteBuffer k = null;
	for(int i = 0, p = 0; i < ns.limit(); i++) {
	    if(ns.get(i) == 0) {
		ByteBuffer s = ns.duplicate();
		s.position(p).limit(i);
		if(k == null) {
		    k = s;
		} else {
		    ret.put(k, s);
		    k = null;
		}
		p = i + 1;
	    }
	}
	return(ret);
    }

    public static Map<ByteBuffer, ByteBuffer> readhead(ReadableByteChannel sk) throws IOException {
	return(splithead(readns(sk)));
    }
}