package org.jruby.puma;
import org.jruby.Ruby; import org.jruby.RubyHash; import org.jruby.util.ByteList;
public class Http11Parser {
Machine
%%{
machine puma_parser; action mark {parser.mark = fpc; } action start_field { parser.field_start = fpc; } action snake_upcase_field { FIXME stub } action write_field { parser.field_len = fpc-parser.field_start; } action start_value { parser.mark = fpc; } action write_value { Http11.http_field(runtime, parser.data, parser.buffer, parser.field_start, parser.field_len, parser.mark, fpc-parser.mark); } action request_method { Http11.request_method(runtime, parser.data, parser.buffer, parser.mark, fpc-parser.mark); } action request_uri { Http11.request_uri(runtime, parser.data, parser.buffer, parser.mark, fpc-parser.mark); } action fragment { Http11.fragment(runtime, parser.data, parser.buffer, parser.mark, fpc-parser.mark); } action start_query {parser.query_start = fpc; } action query_string { Http11.query_string(runtime, parser.data, parser.buffer, parser.query_start, fpc-parser.query_start); } action http_version { Http11.http_version(runtime, parser.data, parser.buffer, parser.mark, fpc-parser.mark); } action request_path { Http11.request_path(runtime, parser.data, parser.buffer, parser.mark, fpc-parser.mark); } action done { parser.body_start = fpc + 1; http.header_done(runtime, parser.data, parser.buffer, fpc + 1, pe - fpc - 1); fbreak; } include puma_parser_common "http11_parser_common.rl";
}%%
/** Data **/ %% write data noentry;
public static interface ElementCB {
public void call(Ruby runtime, RubyHash data, ByteList buffer, int at, int length);
}
public static interface FieldCB {
public void call(Ruby runtime, RubyHash data, ByteList buffer, int field, int flen, int value, int vlen);
}
public static class HttpParser {
int cs;
int body_start;
int content_len;
int nread;
int mark;
int field_start;
int field_len;
int query_start;
RubyHash data;
ByteList buffer;
public void init() {
cs = 0;
%% write init;
body_start = 0;
content_len = 0;
mark = 0;
nread = 0;
field_len = 0;
field_start = 0;
}
}
public final HttpParser parser = new HttpParser();
public int execute(Ruby runtime, Http11 http, ByteList buffer, int off) {
int p, pe;
int cs = parser.cs;
int len = buffer.length();
assert off<=len : "offset past end of buffer";
p = off;
pe = len;
// get a copy of the bytes, since it may not start at 0
// FIXME: figure out how to just use the bytes in-place
byte[] data = buffer.bytes();
parser.buffer = buffer;
%% write exec;
parser.cs = cs;
parser.nread += (p - off);
assert p <= pe : "buffer overflow after parsing execute";
assert parser.nread <= len : "nread longer than length";
assert parser.body_start <= len : "body starts after buffer end";
assert parser.mark < len : "mark is after buffer end";
assert parser.field_len <= len : "field has length longer than whole buffer";
assert parser.field_start < len : "field starts after buffer end";
return parser.nread;
}
public int finish() {
if(has_error()) {
return -1;
} else if(is_finished()) {
return 1;
} else {
return 0;
}
}
public boolean has_error() {
return parser.cs == puma_parser_error;
}
public boolean is_finished() {
return parser.cs == puma_parser_first_final;
}
}