/
CTParser.m
225 lines (192 loc) · 6.22 KB
/
CTParser.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
/*
* This file is covered by the Ruby license. See COPYING for more details.
* Copyright (C) 2009-2010, Apple Inc. All rights reserved.
*/
#import "CTParser.h"
#pragma mark Parser Callbacks
#define DEF_MAX_LENGTH(N, val) const size_t MAX_##N##_LENGTH = val
#define VALIDATE_MAX_LENGTH(len, N) \
if (len > MAX_##N##_LENGTH) { \
[NSException raise:@"ParserFieldLengthError" \
format:@"HTTP element " # N " is longer than the " \
# len " character allowed length."]; \
}
#define PARSE_FIELD(field) \
static void \
parse_##field(void *env, const char *at, size_t length) \
{ \
VALIDATE_MAX_LENGTH(length, field) \
NSString *val = [[NSString alloc] initWithBytes:at length:length \
encoding:NSUTF8StringEncoding]; \
[(NSMutableDictionary *)env setObject:val forKey:@"" #field]; \
}
// Max field lengths
DEF_MAX_LENGTH(FIELD_NAME, 256);
DEF_MAX_LENGTH(FIELD_VALUE, 80 * 1024);
DEF_MAX_LENGTH(REQUEST_METHOD, 256);
DEF_MAX_LENGTH(REQUEST_URI, 1024 * 12);
DEF_MAX_LENGTH(FRAGMENT, 1024);
DEF_MAX_LENGTH(PATH_INFO, 1024);
DEF_MAX_LENGTH(QUERY_STRING, (1024 * 10));
DEF_MAX_LENGTH(HTTP_VERSION, 256);
DEF_MAX_LENGTH(HEADER, (1024 * (80 + 32)));
static void
parse_HTTP_FIELD(void *env, const char *field, size_t flen, const char *value,
size_t vlen)
{
VALIDATE_MAX_LENGTH(flen, FIELD_NAME);
VALIDATE_MAX_LENGTH(vlen, FIELD_VALUE);
NSString *val = [[NSString alloc] initWithBytes:value length:vlen
encoding:NSUTF8StringEncoding];
NSString *key;
if (strncmp(field, "HOST", 4) == 0) {
key = @"HTTP_HOST";
}
else if (strncmp(field, "REFERER", 4) == 0) {
key = @"HTTP_REFERER";
}
else if (strncmp(field, "CACHE_CONTROL", 4) == 0) {
key = @"HTTP_CACHE_CONTROL";
}
else if (strncmp(field, "COOKIE", 4) == 0) {
key = @"HTTP_COOKIE";
}
else if (strncmp(field, "CONNECTION", 4) == 0) {
key = @"HTTP_CONNECTION";
}
else {
key = [@"HTTP_" stringByAppendingString:[[NSString alloc]
initWithBytes:field length:flen encoding:NSUTF8StringEncoding]];
}
[(NSMutableDictionary *)env setObject:val forKey:key];
}
// Parsing callback functions
PARSE_FIELD(REQUEST_METHOD);
PARSE_FIELD(REQUEST_URI);
PARSE_FIELD(FRAGMENT);
PARSE_FIELD(PATH_INFO);
PARSE_FIELD(QUERY_STRING);
PARSE_FIELD(HTTP_VERSION);
static void
header_done(void *env, const char *at, size_t length)
{
NSMutableDictionary *environment = (NSMutableDictionary *)env;
NSString *contentLength = [environment objectForKey:@"HTTP_CONTENT_LENGTH"];
if (contentLength != nil) {
[environment setObject:contentLength forKey:@"CONTENT_LENGTH"];
}
NSString *contentType = [environment objectForKey:@"HTTP_CONTENT_TYPE"];
if (contentType != nil) {
[environment setObject:contentType forKey:@"CONTENT_TYPE"];
}
[environment setObject:@"CGI/1.2" forKey:@"GATEWAY_INTERFACE"];
NSString *hostString = [environment objectForKey:@"HTTP_HOST"];
NSString *serverName = nil;
NSString *serverPort = nil;
if (hostString != nil) {
NSRange colon_pos = [hostString rangeOfString:@":"];
if (colon_pos.location != NSNotFound) {
serverName = [hostString substringToIndex:colon_pos.location];
serverPort = [hostString substringFromIndex:colon_pos.location+1];
}
else {
serverName = [NSString stringWithString:hostString];
serverPort = @"80";
}
[environment setObject:serverName forKey:@"SERVER_NAME"];
[environment setObject:serverPort forKey:@"SERVER_PORT"];
}
[environment setObject:@"HTTP/1.1" forKey:@"SERVER_PROTOCOL"];
[environment setObject:SERVER_SOFTWARE forKey:@"SERVER_SOFTWARE"];
[environment setObject:@"" forKey:@"SCRIPT_NAME"];
// We don't do tls yet
[environment setObject:@"http" forKey:@"rack.url_scheme"];
// To satisfy Rack specs...
if ([environment objectForKey:@"QUERY_STRING"] == nil) {
[environment setObject:@"" forKey:@"QUERY_STRING"];
}
// If we've been given any part of the body, put it here
NSMutableArray *body = [environment objectForKey:@"rack.input"];
if (body != nil) {
[body addObject:[NSData dataWithBytes:at length:length]];
}
else {
NSLog(@"Hmm...you seem to have body data but no where to put it. That's probably an error.");
}
}
@implementation CTParser
- (id)init
{
self = [super init];
if (self != nil) {
_parser = malloc(sizeof(http_parser));
assert(_parser != NULL);
// Setup the callbacks
_parser->http_field = parse_HTTP_FIELD;
_parser->request_method = parse_REQUEST_METHOD;
_parser->request_uri = parse_REQUEST_URI;
_parser->fragment = parse_FRAGMENT;
_parser->request_path = parse_PATH_INFO;
_parser->query_string = parse_QUERY_STRING;
_parser->http_version = parse_HTTP_VERSION;
_parser->header_done = header_done;
http_parser_init(_parser);
}
return self;
}
- (void)reset
{
http_parser_init(_parser);
}
- (NSNumber *)parseData:(NSData *)dataBuf
forEnvironment:(NSMutableDictionary *)env
startingAt:(NSNumber *)startingPos
{
NSMutableData *dataForParser = [NSMutableData dataWithLength:
[dataBuf length] + 1];
[dataForParser setData:dataBuf];
[dataForParser appendData:'\0'];
const char *data = [dataForParser bytes];
size_t length = [dataForParser length];
size_t offset = [startingPos unsignedLongValue];
_parser->data = env;
http_parser_execute(_parser, data, length, offset);
if (http_parser_has_error(_parser)) {
[NSException raise:@"CTParserError"
format:@"Invalid HTTP format, parsing failed."];
}
NSNumber *headerLength = [NSNumber numberWithUnsignedLong:_parser->nread];
VALIDATE_MAX_LENGTH([headerLength unsignedLongValue], HEADER);
return headerLength;
}
- (NSNumber *)parseData:(NSData *)dataBuf forEnvironment:(NSDictionary *)env
{
return [self parseData:dataBuf forEnvironment:env startingAt:0];
}
- (BOOL)errorCond
{
return http_parser_has_error(_parser);
}
- (BOOL)finished
{
return http_parser_is_finished(_parser);
}
- (NSNumber *)nread
{
return [NSNumber numberWithInt:_parser->nread];
}
- (void)finalize
{
if (_parser != NULL) {
free(_parser);
_parser = NULL;
}
[super finalize];
}
@end
void
Init_CTParser(void)
{
// Do nothing. This function is required by the MacRuby runtime when this
// file is compiled as a C extension bundle.
}