mirror of
https://github.com/puma/puma.git
synced 2022-11-09 13:48:40 -05:00
145 lines
3.6 KiB
Ragel
145 lines
3.6 KiB
Ragel
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;
|
|
|
|
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;
|
|
}
|
|
}
|