//package redstone.xmlrpc.util;
/**
* nTorrent - A GUI client to administer a rtorrent process
* over a network connection.
*
* Copyright (C) 2007 Kim Eik
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
import java.io.IOException;
import java.io.InputStream;
import java.nio.ByteBuffer;
import java.nio.charset.Charset;
import java.util.HashMap;
import java.util.Map;
/**
* SCGI connector.<br>
* Version: 1.0<br>
* Home page: http://snippets.dzone.com/posts/show/4304
*/
public class Scgi {
public static class SCGIException extends IOException {
private static final long serialVersionUID = 1L;
public SCGIException(String message) {
super(message);
}
}
/** Used to decode the headers. */
public static final Charset ISO_8859_1 = Charset.forName("ISO-8859-1");
/**
* Read the <a href="http://python.ca/scgi/protocol.txt">SCGI</a> request
* headers.<br>
* After the headers had been loaded, you can read the body of the request
* manually from the same {@code input} stream:
*
* <pre>
* // Load the SCGI headers.
* Socket clientSocket = socket.accept();
* BufferedInputStream bis = new BufferedInputStream(
* clientSocket.getInputStream(), 4096);
* HashMap<String, String> env = SCGI.parse(bis);
* // Read the body of the request.
* bis.read(new byte[Integer.parseInt(env.get("CONTENT_LENGTH"))]);
* </pre>
*
* @param input
* an efficient (buffered) input stream.
* @return strings passed via the SCGI request.
*/
@SuppressWarnings("unchecked")
public static HashMap parse(InputStream input) throws IOException {
StringBuilder lengthString = new StringBuilder(12);
String headers = "";
for (;;) {
char ch = (char) input.read();
if (ch >= '0' && ch <= '9') {
lengthString.append(ch);
} else if (ch == ':') {
int length = Integer.parseInt(lengthString.toString());
byte[] headersBuf = new byte[length];
int read = input.read(headersBuf);
if (read != headersBuf.length)
throw new SCGIException("Couldn't read all the headers ("
+ length + ").");
headers = ISO_8859_1.decode(ByteBuffer.wrap(headersBuf))
.toString();
if (input.read() != ',')
throw new SCGIException("Wrong SCGI header length: "
+ lengthString);
break;
} else {
lengthString.append(ch);
throw new SCGIException("Wrong SCGI header length: "
+ lengthString);
}
}
HashMap<String, String> env = new HashMap<String, String>();
while (headers.length() != 0) {
int sep1 = headers.indexOf(0);
int sep2 = headers.indexOf(0, sep1 + 1);
env.put(headers.substring(0, sep1),
headers.substring(sep1 + 1, sep2));
headers = headers.substring(sep2 + 1);
}
return env;
}
public static String make(HashMap<String, String> header, String body) {
String res = "CONTENT_LENGTH\0" + (body != null ? body.length() : 0)
+ "\0SCGI\0" + "1\0";
if (header != null) {
for (Map.Entry<String, String> entry : header.entrySet())
res += entry.getKey() + '\0' + entry.getValue() + '\0';
}
String size = new Integer(res.getBytes().length) + ":";
res += "," + body;
return size + res;
}
}
|
