blob: 0cc664153886ab65af1f6e5fb1b1a1b3ff02d146 (
plain)
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
|
/*
* Copyright (C) Yichun Zhang (agentzh)
*/
#ifndef _NGX_HTTP_LUA_SOCKET_TCP_H_INCLUDED_
#define _NGX_HTTP_LUA_SOCKET_TCP_H_INCLUDED_
#include "ngx_http_lua_common.h"
#define NGX_HTTP_LUA_SOCKET_FT_ERROR 0x0001
#define NGX_HTTP_LUA_SOCKET_FT_TIMEOUT 0x0002
#define NGX_HTTP_LUA_SOCKET_FT_CLOSED 0x0004
#define NGX_HTTP_LUA_SOCKET_FT_RESOLVER 0x0008
#define NGX_HTTP_LUA_SOCKET_FT_BUFTOOSMALL 0x0010
#define NGX_HTTP_LUA_SOCKET_FT_NOMEM 0x0020
#define NGX_HTTP_LUA_SOCKET_FT_PARTIALWRITE 0x0040
#define NGX_HTTP_LUA_SOCKET_FT_CLIENTABORT 0x0080
#define NGX_HTTP_LUA_SOCKET_FT_SSL 0x0100
typedef struct ngx_http_lua_socket_tcp_upstream_s
ngx_http_lua_socket_tcp_upstream_t;
typedef struct ngx_http_lua_socket_udata_queue_s
ngx_http_lua_socket_udata_queue_t;
typedef
int (*ngx_http_lua_socket_tcp_retval_handler)(ngx_http_request_t *r,
ngx_http_lua_socket_tcp_upstream_t *u, lua_State *L);
typedef void (*ngx_http_lua_socket_tcp_upstream_handler_pt)
(ngx_http_request_t *r, ngx_http_lua_socket_tcp_upstream_t *u);
typedef struct {
ngx_event_t event;
ngx_queue_t queue;
ngx_str_t host;
ngx_http_cleanup_pt *cleanup;
ngx_http_lua_socket_tcp_upstream_t *u;
in_port_t port;
} ngx_http_lua_socket_tcp_conn_op_ctx_t;
#define ngx_http_lua_socket_tcp_free_conn_op_ctx(conn_op_ctx) \
ngx_free(conn_op_ctx->host.data); \
ngx_free(conn_op_ctx)
typedef struct {
lua_State *lua_vm;
ngx_int_t size;
ngx_queue_t cache_connect_op;
ngx_queue_t wait_connect_op;
/* connections == active connections + pending connect operations,
* while active connections == out-of-pool reused connections
* + in-pool connections */
ngx_int_t connections;
/* queues of ngx_http_lua_socket_pool_item_t: */
ngx_queue_t cache;
ngx_queue_t free;
ngx_int_t backlog;
u_char key[1];
} ngx_http_lua_socket_pool_t;
struct ngx_http_lua_socket_tcp_upstream_s {
ngx_http_lua_socket_tcp_retval_handler read_prepare_retvals;
ngx_http_lua_socket_tcp_retval_handler write_prepare_retvals;
ngx_http_lua_socket_tcp_upstream_handler_pt read_event_handler;
ngx_http_lua_socket_tcp_upstream_handler_pt write_event_handler;
ngx_http_lua_socket_udata_queue_t *udata_queue;
ngx_http_lua_socket_pool_t *socket_pool;
ngx_http_lua_loc_conf_t *conf;
ngx_http_cleanup_pt *cleanup;
ngx_http_request_t *request;
ngx_peer_connection_t peer;
ngx_msec_t read_timeout;
ngx_msec_t send_timeout;
ngx_msec_t connect_timeout;
ngx_http_upstream_resolved_t *resolved;
ngx_chain_t *bufs_in; /* input data buffers */
ngx_chain_t *buf_in; /* last input data buffer */
ngx_buf_t buffer; /* receive buffer */
size_t length;
size_t rest;
ngx_err_t socket_errno;
ngx_int_t (*input_filter)(void *data, ssize_t bytes);
void *input_filter_ctx;
size_t request_len;
ngx_chain_t *request_bufs;
ngx_http_lua_co_ctx_t *read_co_ctx;
ngx_http_lua_co_ctx_t *write_co_ctx;
ngx_uint_t reused;
#if (NGX_HTTP_SSL)
ngx_str_t ssl_name;
ngx_ssl_session_t *ssl_session_ret;
const char *error_ret;
int openssl_error_code_ret;
#endif
ngx_chain_t *busy_bufs;
unsigned ft_type:16;
unsigned no_close:1;
unsigned conn_waiting:1;
unsigned read_waiting:1;
unsigned write_waiting:1;
unsigned eof:1;
unsigned body_downstream:1;
unsigned raw_downstream:1;
unsigned read_closed:1;
unsigned write_closed:1;
unsigned conn_closed:1;
#if (NGX_HTTP_SSL)
unsigned ssl_verify:1;
unsigned ssl_session_reuse:1;
#endif
};
typedef struct ngx_http_lua_dfa_edge_s ngx_http_lua_dfa_edge_t;
struct ngx_http_lua_dfa_edge_s {
u_char chr;
int new_state;
ngx_http_lua_dfa_edge_t *next;
};
typedef struct {
ngx_http_lua_socket_tcp_upstream_t *upstream;
ngx_str_t pattern;
int state;
ngx_http_lua_dfa_edge_t **recovering;
unsigned inclusive:1;
} ngx_http_lua_socket_compiled_pattern_t;
typedef struct {
ngx_http_lua_socket_pool_t *socket_pool;
ngx_queue_t queue;
ngx_connection_t *connection;
socklen_t socklen;
struct sockaddr_storage sockaddr;
ngx_uint_t reused;
ngx_http_lua_socket_udata_queue_t *udata_queue;
} ngx_http_lua_socket_pool_item_t;
struct ngx_http_lua_socket_udata_queue_s {
ngx_pool_t *pool;
ngx_queue_t queue;
ngx_queue_t free;
int len;
int capacity;
};
typedef struct {
ngx_queue_t queue;
uint64_t key;
uint64_t value;
} ngx_http_lua_socket_node_t;
void ngx_http_lua_inject_socket_tcp_api(ngx_log_t *log, lua_State *L);
void ngx_http_lua_inject_req_socket_api(lua_State *L);
void ngx_http_lua_cleanup_conn_pools(lua_State *L);
#endif /* _NGX_HTTP_LUA_SOCKET_TCP_H_INCLUDED_ */
/* vi:set ft=c ts=4 sw=4 et fdm=marker: */
|
