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
|
/*
* %CopyrightBegin%
*
* Copyright Ericsson AB 2008-2009. All Rights Reserved.
*
* The contents of this file are subject to the Erlang Public License,
* Version 1.1, (the "License"); you may not use this file except in
* compliance with the License. You should have received a copy of the
* Erlang Public License along with this software. If not, it can be
* retrieved online at http://www.erlang.org/.
*
* Software distributed under the License is distributed on an "AS IS"
* basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
* the License for the specific language governing rights and limitations
* under the License.
*
* %CopyrightEnd%
*/
/* A protocol decoder. Simple packet length extraction as well as packet
* body parsing with protocol specific callback interfaces (http and ssl).
*/
#ifndef __PACKET_PARSER_H__
#define __PACKET_PARSER_H__
#include <erl_driver.h>
#include "sys.h"
/* INET_LOPT_PACKET options */
enum PacketParseType {
TCP_PB_RAW = 0,
TCP_PB_1 = 1,
TCP_PB_2 = 2,
TCP_PB_4 = 3,
TCP_PB_ASN1 = 4,
TCP_PB_RM = 5,
TCP_PB_CDR = 6,
TCP_PB_FCGI = 7,
TCP_PB_LINE_LF = 8,
TCP_PB_TPKT = 9,
TCP_PB_HTTP = 10,
TCP_PB_HTTPH = 11,
TCP_PB_SSL_TLS = 12,
TCP_PB_HTTP_BIN = 13,
TCP_PB_HTTPH_BIN = 14
};
typedef struct http_atom {
struct http_atom* next; /* next in bucket */
unsigned long h; /* stored hash value */
const char* name;
int len;
int index; /* index in table + bit-pos */
ErlDrvTermData atom; /* erlang atom rep */
} http_atom_t;
typedef struct {
enum {
URI_STAR, /* '*' */
URI_STRING, /* "string(s1)" */
URI_ABS_PATH,/* {abs_path, "path(s1)"} */
URI_SCHEME, /* {scheme, "scheme(s1)", "string(s2)"} */
URI_HTTP, /* {absoluteURI, http, "host(s1)", Port, "path(s2)"} */
URI_HTTPS /* {absoluteURI, https, ... */
} type;
const char* s1_ptr;
int s1_len;
const char* s2_ptr;
int s2_len;
int port; /* 0=undefined */
}PacketHttpURI;
typedef int HttpResponseMessageFn(void* arg, int major, int minor, int status,
const char* phrase, int phrase_len);
typedef int HttpRequestMessageFn(void* arg, const http_atom_t* meth, const char* meth_ptr,
int meth_len, const PacketHttpURI*, int major, int minor);
typedef int HttpEohMessageFn(void *arg);
typedef int HttpHeaderMessageFn(void* arg, const http_atom_t* name, const char* name_ptr,
int name_len, const char* value_ptr, int value_len);
typedef int HttpErrorMessageFn(void* arg, const char* buf, int len);
typedef int SslTlsFn(void* arg, unsigned type, unsigned major, unsigned minor,
const char* data, int len, const char* prefix, int plen);
typedef struct {
HttpResponseMessageFn* http_response;
HttpRequestMessageFn* http_request;
HttpEohMessageFn* http_eoh;
HttpHeaderMessageFn* http_header;
HttpErrorMessageFn* http_error;
SslTlsFn* ssl_tls;
}PacketCallbacks;
/* Called once at emulator start
*/
void packet_parser_init(void);
/* Returns > 0 Total packet length.
* = 0 Length unknown, need more data.
* < 0 Error, invalid format.
*/
int packet_get_length(enum PacketParseType htype,
const char* ptr, unsigned n, /* Bytes read so far */
unsigned max_plen, /* Packet max length, 0=no limit */
unsigned trunc_len, /* Truncate (lines) if longer, 0=no limit */
int* statep); /* Internal protocol state */
ERTS_GLB_INLINE
void packet_get_body(enum PacketParseType htype,
const char** bufp, /* In: Packet header, Out: Packet body */
int* lenp); /* In: Packet length, Out: Body length */
/* Returns 1 = Packet parsed and handled by callbacks.
** 0 = No parsing support for this packet type
** -1 = Error
*/
ERTS_GLB_INLINE
int packet_parse(enum PacketParseType htype,
const char* buf, int len, /* Total packet */
int* statep, PacketCallbacks* pcb, void* arg);
/* Internals for the inlines below: */
#define FCGI_VERSION_1 1
struct fcgi_head {
unsigned char version;
unsigned char type;
unsigned char requestIdB1;
unsigned char requestIdB0;
unsigned char contentLengthB1;
unsigned char contentLengthB0;
unsigned char paddingLength;
unsigned char reserved;
/* char data[] */
/* char padding[paddingLength] */
};
int packet_parse_http(const char*, int, int*, PacketCallbacks*, void*);
int packet_parse_ssl(const char*, int, PacketCallbacks*, void*);
#if ERTS_GLB_INLINE_INCL_FUNC_DEF
ERTS_GLB_INLINE
void packet_get_body(enum PacketParseType htype, const char** bufp, int* lenp)
{
switch (htype) {
case TCP_PB_1: *bufp += 1; *lenp -= 1; break;
case TCP_PB_2: *bufp += 2; *lenp -= 2; break;
case TCP_PB_4: *bufp += 4; *lenp -= 4; break;
case TCP_PB_FCGI:
*lenp -= ((struct fcgi_head*)*bufp)->paddingLength;
break;
default:
;/* Return other packets "as is" */
}
}
ERTS_GLB_INLINE
int packet_parse(enum PacketParseType htype, const char* buf, int len,
int* statep, PacketCallbacks* pcb, void* arg)
{
switch (htype) {
case TCP_PB_HTTP:
case TCP_PB_HTTPH:
case TCP_PB_HTTP_BIN:
case TCP_PB_HTTPH_BIN:
if (packet_parse_http(buf, len, statep, pcb, arg) < 0)
pcb->http_error(arg, buf, len);
return 1;
case TCP_PB_SSL_TLS:
return packet_parse_ssl(buf, len, pcb, arg);
default:;
}
return 0;
}
#endif /* ERTS_GLB_INLINE_INCL_FUNC_DEF */
#endif /* !__PACKET_PARSER_H__ */
|
