blob: 86fe28283a0a60e808fad25874eb724c26e2052b [file] [log] [blame]
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001#include <sys/socket.h>
2
Thierry FOURNIER6f1fd482015-01-23 14:06:13 +01003#include <lauxlib.h>
4#include <lua.h>
5#include <lualib.h>
6
Thierry FOURNIER463119c2015-03-10 00:35:36 +01007#if !defined(LUA_VERSION_NUM) || LUA_VERSION_NUM < 503
8#error "Requires Lua 5.3 or later."
Cyril Bontédc0306e2015-03-02 00:08:40 +01009#endif
10
Thierry FOURNIER380d0932015-01-23 14:27:52 +010011#include <ebpttree.h>
12
13#include <common/cfgparse.h>
14
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +010015#include <types/connection.h>
Thierry FOURNIER380d0932015-01-23 14:27:52 +010016#include <types/hlua.h>
Thierry FOURNIER65f34c62015-02-16 20:11:43 +010017#include <types/proto_tcp.h>
Thierry FOURNIER380d0932015-01-23 14:27:52 +010018#include <types/proxy.h>
19
Thierry FOURNIER55da1652015-01-23 11:36:30 +010020#include <proto/arg.h>
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +010021#include <proto/channel.h>
Thierry FOURNIER9a819e72015-02-16 20:22:55 +010022#include <proto/hdr_idx.h>
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +010023#include <proto/hlua.h>
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +010024#include <proto/obj_type.h>
Thierry FOURNIER83758bb2015-02-04 13:21:04 +010025#include <proto/pattern.h>
Thierry FOURNIERd0fa5382015-02-16 20:14:51 +010026#include <proto/payload.h>
27#include <proto/proto_http.h>
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +010028#include <proto/proto_tcp.h>
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +010029#include <proto/raw_sock.h>
Thierry FOURNIERd0fa5382015-02-16 20:14:51 +010030#include <proto/sample.h>
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +010031#include <proto/server.h>
32#include <proto/session.h>
33#include <proto/ssl_sock.h>
34#include <proto/stream_interface.h>
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +010035#include <proto/task.h>
36
Thierry FOURNIER6f1fd482015-01-23 14:06:13 +010037/* Lua uses longjmp to perform yield or throwing errors. This
38 * macro is used only for identifying the function that can
39 * not return because a longjmp is executed.
40 * __LJMP marks a prototype of hlua file that can use longjmp.
41 * WILL_LJMP() marks an lua function that will use longjmp.
42 * MAY_LJMP() marks an lua function that may use longjmp.
43 */
44#define __LJMP
45#define WILL_LJMP(func) func
46#define MAY_LJMP(func) func
47
Thierry FOURNIER380d0932015-01-23 14:27:52 +010048/* The main Lua execution context. */
49struct hlua gL;
50
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +010051/* This is the memory pool containing all the signal structs. These
52 * struct are used to store each requiered signal between two tasks.
53 */
54struct pool_head *pool2_hlua_com;
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +010055struct pool_head *pool2_hlua_sleep;
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +010056
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +010057/* Used for Socket connection. */
58static struct proxy socket_proxy;
59static struct server socket_tcp;
60#ifdef USE_OPENSSL
61static struct server socket_ssl;
62#endif
63
Thierry FOURNIERa4a0f3d2015-01-23 12:08:30 +010064/* List head of the function called at the initialisation time. */
65struct list hlua_init_functions = LIST_HEAD_INIT(hlua_init_functions);
66
Thierry FOURNIER2ba18a22015-01-23 14:07:08 +010067/* The following variables contains the reference of the different
68 * Lua classes. These references are useful for identify metadata
69 * associated with an object.
70 */
Thierry FOURNIER65f34c62015-02-16 20:11:43 +010071static int class_txn_ref;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +010072static int class_socket_ref;
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +010073static int class_channel_ref;
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +010074static int class_fetches_ref;
Thierry FOURNIER594afe72015-03-10 23:58:30 +010075static int class_converters_ref;
Thierry FOURNIER2ba18a22015-01-23 14:07:08 +010076
Thierry FOURNIERbd413492015-03-03 16:52:26 +010077/* Global Lua execution timeout. By default Lua, execution linked
78 * with session (actions, sample-fetches and converters) have a
79 * short timeout. Lua linked with tasks doesn't have a timeout
80 * because a task may remain alive during all the haproxy execution.
81 */
82static unsigned int hlua_timeout_session = 4000; /* session timeout. */
83static unsigned int hlua_timeout_task = TICK_ETERNITY; /* task timeout. */
84
Thierry FOURNIERee9f8022015-03-03 17:37:37 +010085/* Interrupts the Lua processing each "hlua_nb_instruction" instructions.
86 * it is used for preventing infinite loops.
87 *
88 * I test the scheer with an infinite loop containing one incrementation
89 * and one test. I run this loop between 10 seconds, I raise a ceil of
90 * 710M loops from one interrupt each 9000 instructions, so I fix the value
91 * to one interrupt each 10 000 instructions.
92 *
93 * configured | Number of
94 * instructions | loops executed
95 * between two | in milions
96 * forced yields |
97 * ---------------+---------------
98 * 10 | 160
99 * 500 | 670
100 * 1000 | 680
101 * 5000 | 700
102 * 7000 | 700
103 * 8000 | 700
104 * 9000 | 710 <- ceil
105 * 10000 | 710
106 * 100000 | 710
107 * 1000000 | 710
108 *
109 */
110static unsigned int hlua_nb_instruction = 10000;
111
Thierry FOURNIER55da1652015-01-23 11:36:30 +0100112/* These functions converts types between HAProxy internal args or
113 * sample and LUA types. Another function permits to check if the
114 * LUA stack contains arguments according with an required ARG_T
115 * format.
116 */
117static int hlua_arg2lua(lua_State *L, const struct arg *arg);
118static int hlua_lua2arg(lua_State *L, int ud, struct arg *arg);
119__LJMP static int hlua_lua2arg_check(lua_State *L, int first, struct arg *argp, unsigned int mask);
Willy Tarreau5eadada2015-03-10 17:28:54 +0100120static int hlua_smp2lua(lua_State *L, struct sample *smp);
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +0100121static int hlua_smp2lua_str(lua_State *L, struct sample *smp);
Thierry FOURNIER55da1652015-01-23 11:36:30 +0100122static int hlua_lua2smp(lua_State *L, int ud, struct sample *smp);
123
Thierry FOURNIERe8b9a402015-02-25 18:48:12 +0100124/* Used to check an Lua function type in the stack. It creates and
125 * returns a reference of the function. This function throws an
126 * error if the rgument is not a "function".
127 */
128__LJMP unsigned int hlua_checkfunction(lua_State *L, int argno)
129{
130 if (!lua_isfunction(L, argno)) {
131 const char *msg = lua_pushfstring(L, "function expected, got %s", luaL_typename(L, -1));
132 WILL_LJMP(luaL_argerror(L, argno, msg));
133 }
134 lua_pushvalue(L, argno);
135 return luaL_ref(L, LUA_REGISTRYINDEX);
136}
137
138/* The three following functions are useful for adding entries
139 * in a table. These functions takes a string and respectively an
140 * integer, a string or a function and add it to the table in the
141 * top of the stack.
142 *
143 * These functions throws an error if no more stack size is
144 * available.
145 */
146__LJMP static inline void hlua_class_const_int(lua_State *L, const char *name,
Thierry FOURNIERf90838b2015-03-06 13:48:32 +0100147 int value)
Thierry FOURNIERe8b9a402015-02-25 18:48:12 +0100148{
149 if (!lua_checkstack(L, 2))
150 WILL_LJMP(luaL_error(L, "full stack"));
151 lua_pushstring(L, name);
Thierry FOURNIERf90838b2015-03-06 13:48:32 +0100152 lua_pushinteger(L, value);
Thierry FOURNIERe8b9a402015-02-25 18:48:12 +0100153 lua_settable(L, -3);
154}
155__LJMP static inline void hlua_class_const_str(lua_State *L, const char *name,
156 const char *value)
157{
158 if (!lua_checkstack(L, 2))
159 WILL_LJMP(luaL_error(L, "full stack"));
160 lua_pushstring(L, name);
161 lua_pushstring(L, value);
162 lua_settable(L, -3);
163}
164__LJMP static inline void hlua_class_function(lua_State *L, const char *name,
165 int (*function)(lua_State *L))
166{
167 if (!lua_checkstack(L, 2))
168 WILL_LJMP(luaL_error(L, "full stack"));
169 lua_pushstring(L, name);
170 lua_pushcclosure(L, function, 0);
171 lua_settable(L, -3);
172}
173
174/* This function check the number of arguments available in the
175 * stack. If the number of arguments available is not the same
176 * then <nb> an error is throwed.
177 */
178__LJMP static inline void check_args(lua_State *L, int nb, char *fcn)
179{
180 if (lua_gettop(L) == nb)
181 return;
182 WILL_LJMP(luaL_error(L, "'%s' needs %d arguments", fcn, nb));
183}
184
185/* Return true if the data in stack[<ud>] is an object of
186 * type <class_ref>.
187 */
Thierry FOURNIER2297bc22015-03-11 17:43:33 +0100188static int hlua_metaistype(lua_State *L, int ud, int class_ref)
Thierry FOURNIERe8b9a402015-02-25 18:48:12 +0100189{
Thierry FOURNIERe8b9a402015-02-25 18:48:12 +0100190 if (!lua_getmetatable(L, ud))
191 return 0;
192
193 lua_rawgeti(L, LUA_REGISTRYINDEX, class_ref);
194 if (!lua_rawequal(L, -1, -2)) {
195 lua_pop(L, 2);
196 return 0;
197 }
198
199 lua_pop(L, 2);
200 return 1;
201}
202
203/* Return an object of the expected type, or throws an error. */
204__LJMP static void *hlua_checkudata(lua_State *L, int ud, int class_ref)
205{
Thierry FOURNIER2297bc22015-03-11 17:43:33 +0100206 void *p;
207
208 /* Check if the stack entry is an array. */
209 if (!lua_istable(L, ud))
210 WILL_LJMP(luaL_argerror(L, ud, NULL));
211 /* Check if the metadata have the expected type. */
212 if (!hlua_metaistype(L, ud, class_ref))
213 WILL_LJMP(luaL_argerror(L, ud, NULL));
214 /* Push on the stack at the entry [0] of the table. */
215 lua_rawgeti(L, ud, 0);
216 /* Check if this entry is userdata. */
217 p = lua_touserdata(L, -1);
218 if (!p)
219 WILL_LJMP(luaL_argerror(L, ud, NULL));
220 /* Remove the entry returned by lua_rawgeti(). */
221 lua_pop(L, 1);
222 /* Return the associated struct. */
223 return p;
Thierry FOURNIERe8b9a402015-02-25 18:48:12 +0100224}
225
226/* This fucntion push an error string prefixed by the file name
227 * and the line number where the error is encountered.
228 */
229static int hlua_pusherror(lua_State *L, const char *fmt, ...)
230{
231 va_list argp;
232 va_start(argp, fmt);
233 luaL_where(L, 1);
234 lua_pushvfstring(L, fmt, argp);
235 va_end(argp);
236 lua_concat(L, 2);
237 return 1;
238}
239
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +0100240/* This function register a new signal. "lua" is the current lua
241 * execution context. It contains a pointer to the associated task.
242 * "link" is a list head attached to an other task that must be wake
243 * the lua task if an event occurs. This is useful with external
244 * events like TCP I/O or sleep functions. This funcion allocate
245 * memory for the signal.
246 */
247static int hlua_com_new(struct hlua *lua, struct list *link)
248{
249 struct hlua_com *com = pool_alloc2(pool2_hlua_com);
250 if (!com)
251 return 0;
252 LIST_ADDQ(&lua->com, &com->purge_me);
253 LIST_ADDQ(link, &com->wake_me);
254 com->task = lua->task;
255 return 1;
256}
257
258/* This function purge all the pending signals when the LUA execution
259 * is finished. This prevent than a coprocess try to wake a deleted
260 * task. This function remove the memory associated to the signal.
261 */
262static void hlua_com_purge(struct hlua *lua)
263{
264 struct hlua_com *com, *back;
265
266 /* Delete all pending communication signals. */
267 list_for_each_entry_safe(com, back, &lua->com, purge_me) {
268 LIST_DEL(&com->purge_me);
269 LIST_DEL(&com->wake_me);
270 pool_free2(pool2_hlua_com, com);
271 }
272}
273
274/* This function sends signals. It wakes all the tasks attached
275 * to a list head, and remove the signal, and free the used
276 * memory.
277 */
278static void hlua_com_wake(struct list *wake)
279{
280 struct hlua_com *com, *back;
281
282 /* Wake task and delete all pending communication signals. */
283 list_for_each_entry_safe(com, back, wake, wake_me) {
284 LIST_DEL(&com->purge_me);
285 LIST_DEL(&com->wake_me);
286 task_wakeup(com->task, TASK_WOKEN_MSG);
287 pool_free2(pool2_hlua_com, com);
288 }
289}
290
Thierry FOURNIER55da1652015-01-23 11:36:30 +0100291/* This functions is used with sample fetch and converters. It
292 * converts the HAProxy configuration argument in a lua stack
293 * values.
294 *
295 * It takes an array of "arg", and each entry of the array is
296 * converted and pushed in the LUA stack.
297 */
298static int hlua_arg2lua(lua_State *L, const struct arg *arg)
299{
300 switch (arg->type) {
301 case ARGT_SINT:
302 lua_pushinteger(L, arg->data.sint);
303 break;
304
305 case ARGT_UINT:
306 case ARGT_TIME:
307 case ARGT_SIZE:
Thierry FOURNIERf90838b2015-03-06 13:48:32 +0100308 lua_pushinteger(L, arg->data.sint);
Thierry FOURNIER55da1652015-01-23 11:36:30 +0100309 break;
310
311 case ARGT_STR:
312 lua_pushlstring(L, arg->data.str.str, arg->data.str.len);
313 break;
314
315 case ARGT_IPV4:
316 case ARGT_IPV6:
317 case ARGT_MSK4:
318 case ARGT_MSK6:
319 case ARGT_FE:
320 case ARGT_BE:
321 case ARGT_TAB:
322 case ARGT_SRV:
323 case ARGT_USR:
324 case ARGT_MAP:
325 default:
326 lua_pushnil(L);
327 break;
328 }
329 return 1;
330}
331
332/* This function take one entrie in an LUA stack at the index "ud",
333 * and try to convert it in an HAProxy argument entry. This is useful
334 * with sample fetch wrappers. The input arguments are gived to the
335 * lua wrapper and converted as arg list by thi function.
336 */
337static int hlua_lua2arg(lua_State *L, int ud, struct arg *arg)
338{
339 switch (lua_type(L, ud)) {
340
341 case LUA_TNUMBER:
342 case LUA_TBOOLEAN:
343 arg->type = ARGT_SINT;
344 arg->data.sint = lua_tointeger(L, ud);
345 break;
346
347 case LUA_TSTRING:
348 arg->type = ARGT_STR;
349 arg->data.str.str = (char *)lua_tolstring(L, ud, (size_t *)&arg->data.str.len);
350 break;
351
352 case LUA_TUSERDATA:
353 case LUA_TNIL:
354 case LUA_TTABLE:
355 case LUA_TFUNCTION:
356 case LUA_TTHREAD:
357 case LUA_TLIGHTUSERDATA:
358 arg->type = ARGT_SINT;
359 arg->data.uint = 0;
360 break;
361 }
362 return 1;
363}
364
365/* the following functions are used to convert a struct sample
366 * in Lua type. This useful to convert the return of the
367 * fetchs or converters.
368 */
Willy Tarreau5eadada2015-03-10 17:28:54 +0100369static int hlua_smp2lua(lua_State *L, struct sample *smp)
Thierry FOURNIER55da1652015-01-23 11:36:30 +0100370{
371 switch (smp->type) {
372 case SMP_T_SINT:
Thierry FOURNIER55da1652015-01-23 11:36:30 +0100373 case SMP_T_BOOL:
374 case SMP_T_UINT:
Thierry FOURNIERf90838b2015-03-06 13:48:32 +0100375 lua_pushinteger(L, smp->data.sint);
Thierry FOURNIER55da1652015-01-23 11:36:30 +0100376 break;
377
378 case SMP_T_BIN:
379 case SMP_T_STR:
380 lua_pushlstring(L, smp->data.str.str, smp->data.str.len);
381 break;
382
383 case SMP_T_METH:
384 switch (smp->data.meth.meth) {
385 case HTTP_METH_OPTIONS: lua_pushstring(L, "OPTIONS"); break;
386 case HTTP_METH_GET: lua_pushstring(L, "GET"); break;
387 case HTTP_METH_HEAD: lua_pushstring(L, "HEAD"); break;
388 case HTTP_METH_POST: lua_pushstring(L, "POST"); break;
389 case HTTP_METH_PUT: lua_pushstring(L, "PUT"); break;
390 case HTTP_METH_DELETE: lua_pushstring(L, "DELETE"); break;
391 case HTTP_METH_TRACE: lua_pushstring(L, "TRACE"); break;
392 case HTTP_METH_CONNECT: lua_pushstring(L, "CONNECT"); break;
393 case HTTP_METH_OTHER:
394 lua_pushlstring(L, smp->data.meth.str.str, smp->data.meth.str.len);
395 break;
396 default:
397 lua_pushnil(L);
398 break;
399 }
400 break;
401
402 case SMP_T_IPV4:
403 case SMP_T_IPV6:
404 case SMP_T_ADDR: /* This type is never used to qualify a sample. */
Willy Tarreau5eadada2015-03-10 17:28:54 +0100405 if (sample_casts[smp->type][SMP_T_STR] &&
406 sample_casts[smp->type][SMP_T_STR](smp))
407 lua_pushlstring(L, smp->data.str.str, smp->data.str.len);
408 else
409 lua_pushnil(L);
410 break;
Thierry FOURNIER55da1652015-01-23 11:36:30 +0100411 default:
412 lua_pushnil(L);
413 break;
414 }
415 return 1;
416}
417
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +0100418/* the following functions are used to convert a struct sample
419 * in Lua strings. This is useful to convert the return of the
420 * fetchs or converters.
421 */
422static int hlua_smp2lua_str(lua_State *L, struct sample *smp)
423{
424 switch (smp->type) {
425
426 case SMP_T_BIN:
427 case SMP_T_STR:
428 lua_pushlstring(L, smp->data.str.str, smp->data.str.len);
429 break;
430
431 case SMP_T_METH:
432 switch (smp->data.meth.meth) {
433 case HTTP_METH_OPTIONS: lua_pushstring(L, "OPTIONS"); break;
434 case HTTP_METH_GET: lua_pushstring(L, "GET"); break;
435 case HTTP_METH_HEAD: lua_pushstring(L, "HEAD"); break;
436 case HTTP_METH_POST: lua_pushstring(L, "POST"); break;
437 case HTTP_METH_PUT: lua_pushstring(L, "PUT"); break;
438 case HTTP_METH_DELETE: lua_pushstring(L, "DELETE"); break;
439 case HTTP_METH_TRACE: lua_pushstring(L, "TRACE"); break;
440 case HTTP_METH_CONNECT: lua_pushstring(L, "CONNECT"); break;
441 case HTTP_METH_OTHER:
442 lua_pushlstring(L, smp->data.meth.str.str, smp->data.meth.str.len);
443 break;
444 default:
445 lua_pushstring(L, "");
446 break;
447 }
448 break;
449
450 case SMP_T_SINT:
451 case SMP_T_BOOL:
452 case SMP_T_UINT:
453 case SMP_T_IPV4:
454 case SMP_T_IPV6:
455 case SMP_T_ADDR: /* This type is never used to qualify a sample. */
456 if (sample_casts[smp->type][SMP_T_STR] &&
457 sample_casts[smp->type][SMP_T_STR](smp))
458 lua_pushlstring(L, smp->data.str.str, smp->data.str.len);
459 else
460 lua_pushstring(L, "");
461 break;
462 default:
463 lua_pushstring(L, "");
464 break;
465 }
466 return 1;
467}
468
Thierry FOURNIER55da1652015-01-23 11:36:30 +0100469/* the following functions are used to convert an Lua type in a
470 * struct sample. This is useful to provide data from a converter
471 * to the LUA code.
472 */
473static int hlua_lua2smp(lua_State *L, int ud, struct sample *smp)
474{
475 switch (lua_type(L, ud)) {
476
477 case LUA_TNUMBER:
478 smp->type = SMP_T_SINT;
479 smp->data.sint = lua_tointeger(L, ud);
480 break;
481
482
483 case LUA_TBOOLEAN:
484 smp->type = SMP_T_BOOL;
485 smp->data.uint = lua_toboolean(L, ud);
486 break;
487
488 case LUA_TSTRING:
489 smp->type = SMP_T_STR;
490 smp->flags |= SMP_F_CONST;
491 smp->data.str.str = (char *)lua_tolstring(L, ud, (size_t *)&smp->data.str.len);
492 break;
493
494 case LUA_TUSERDATA:
495 case LUA_TNIL:
496 case LUA_TTABLE:
497 case LUA_TFUNCTION:
498 case LUA_TTHREAD:
499 case LUA_TLIGHTUSERDATA:
500 smp->type = SMP_T_BOOL;
501 smp->data.uint = 0;
502 break;
503 }
504 return 1;
505}
506
507/* This function check the "argp" builded by another conversion function
508 * is in accord with the expected argp defined by the "mask". The fucntion
509 * returns true or false. It can be adjust the types if there compatibles.
510 */
511__LJMP int hlua_lua2arg_check(lua_State *L, int first, struct arg *argp, unsigned int mask)
512{
513 int min_arg;
514 int idx;
515
516 idx = 0;
517 min_arg = ARGM(mask);
518 mask >>= ARGM_BITS;
519
520 while (1) {
521
522 /* Check oversize. */
523 if (idx >= ARGM_NBARGS && argp[idx].type != ARGT_STOP) {
Cyril Bonté577a36a2015-03-02 00:08:38 +0100524 WILL_LJMP(luaL_argerror(L, first + idx, "Malformed argument mask"));
Thierry FOURNIER55da1652015-01-23 11:36:30 +0100525 }
526
527 /* Check for mandatory arguments. */
528 if (argp[idx].type == ARGT_STOP) {
529 if (idx + 1 < min_arg)
530 WILL_LJMP(luaL_argerror(L, first + idx, "Mandatory argument expected"));
531 return 0;
532 }
533
534 /* Check for exceed the number of requiered argument. */
535 if ((mask & ARGT_MASK) == ARGT_STOP &&
536 argp[idx].type != ARGT_STOP) {
537 WILL_LJMP(luaL_argerror(L, first + idx, "Last argument expected"));
538 }
539
540 if ((mask & ARGT_MASK) == ARGT_STOP &&
541 argp[idx].type == ARGT_STOP) {
542 return 0;
543 }
544
545 /* Compatibility mask. */
546 switch (argp[idx].type) {
547 case ARGT_SINT:
548 switch (mask & ARGT_MASK) {
549 case ARGT_UINT: argp[idx].type = mask & ARGT_MASK; break;
550 case ARGT_TIME: argp[idx].type = mask & ARGT_MASK; break;
551 case ARGT_SIZE: argp[idx].type = mask & ARGT_MASK; break;
552 }
553 break;
554 }
555
556 /* Check for type of argument. */
557 if ((mask & ARGT_MASK) != argp[idx].type) {
558 const char *msg = lua_pushfstring(L, "'%s' expected, got '%s'",
559 arg_type_names[(mask & ARGT_MASK)],
560 arg_type_names[argp[idx].type & ARGT_MASK]);
561 WILL_LJMP(luaL_argerror(L, first + idx, msg));
562 }
563
564 /* Next argument. */
565 mask >>= ARGT_BITS;
566 idx++;
567 }
568}
569
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100570/*
571 * The following functions are used to make correspondance between the the
572 * executed lua pointer and the "struct hlua *" that contain the context.
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100573 *
574 * - hlua_gethlua : return the hlua context associated with an lua_State.
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100575 * - hlua_sethlua : create the association between hlua context and lua_state.
576 */
577static inline struct hlua *hlua_gethlua(lua_State *L)
578{
Thierry FOURNIER38c5fd62015-03-10 02:40:29 +0100579 struct hlua **hlua = lua_getextraspace(L);
580 return *hlua;
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100581}
582static inline void hlua_sethlua(struct hlua *hlua)
583{
Thierry FOURNIER38c5fd62015-03-10 02:40:29 +0100584 struct hlua **hlua_store = lua_getextraspace(hlua->T);
585 *hlua_store = hlua;
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100586}
587
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +0100588/* This function just ensure that the yield will be always
589 * returned with a timeout and permit to set some flags
590 */
591__LJMP void hlua_yieldk(lua_State *L, int nresults, int ctx,
Thierry FOURNIERf90838b2015-03-06 13:48:32 +0100592 lua_KFunction k, int timeout, unsigned int flags)
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +0100593{
594 struct hlua *hlua = hlua_gethlua(L);
595
596 /* Set the wake timeout. If timeout is required, we set
597 * the expiration time.
598 */
Thierry FOURNIERdc9ca962015-03-05 11:16:52 +0100599 hlua->wake_time = tick_first(timeout, hlua->expire);
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +0100600
Thierry FOURNIER4abd3ae2015-03-03 17:29:06 +0100601 hlua->flags |= flags;
602
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +0100603 /* Process the yield. */
604 WILL_LJMP(lua_yieldk(L, nresults, ctx, k));
605}
606
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100607/* This function initialises the Lua environment stored in the session.
608 * It must be called at the start of the session. This function creates
609 * an LUA coroutine. It can not be use to crete the main LUA context.
610 */
611int hlua_ctx_init(struct hlua *lua, struct task *task)
612{
613 lua->Mref = LUA_REFNIL;
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +0100614 lua->flags = 0;
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +0100615 LIST_INIT(&lua->com);
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100616 lua->T = lua_newthread(gL.T);
617 if (!lua->T) {
618 lua->Tref = LUA_REFNIL;
619 return 0;
620 }
621 hlua_sethlua(lua);
622 lua->Tref = luaL_ref(gL.T, LUA_REGISTRYINDEX);
623 lua->task = task;
624 return 1;
625}
626
627/* Used to destroy the Lua coroutine when the attached session or task
628 * is destroyed. The destroy also the memory context. The struct "lua"
629 * is not freed.
630 */
631void hlua_ctx_destroy(struct hlua *lua)
632{
Thierry FOURNIERa718b292015-03-04 16:48:34 +0100633 if (!lua->T)
634 return;
635
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +0100636 /* Purge all the pending signals. */
637 hlua_com_purge(lua);
638
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100639 /* The thread is garbage collected by Lua. */
640 luaL_unref(lua->T, LUA_REGISTRYINDEX, lua->Mref);
641 luaL_unref(gL.T, LUA_REGISTRYINDEX, lua->Tref);
642}
643
644/* This function is used to restore the Lua context when a coroutine
645 * fails. This function copy the common memory between old coroutine
646 * and the new coroutine. The old coroutine is destroyed, and its
647 * replaced by the new coroutine.
648 * If the flag "keep_msg" is set, the last entry of the old is assumed
649 * as string error message and it is copied in the new stack.
650 */
651static int hlua_ctx_renew(struct hlua *lua, int keep_msg)
652{
653 lua_State *T;
654 int new_ref;
655
656 /* Renew the main LUA stack doesn't have sense. */
657 if (lua == &gL)
658 return 0;
659
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100660 /* New Lua coroutine. */
661 T = lua_newthread(gL.T);
662 if (!T)
663 return 0;
664
665 /* Copy last error message. */
666 if (keep_msg)
667 lua_xmove(lua->T, T, 1);
668
669 /* Copy data between the coroutines. */
670 lua_rawgeti(lua->T, LUA_REGISTRYINDEX, lua->Mref);
671 lua_xmove(lua->T, T, 1);
672 new_ref = luaL_ref(T, LUA_REGISTRYINDEX); /* Valur poped. */
673
674 /* Destroy old data. */
675 luaL_unref(lua->T, LUA_REGISTRYINDEX, lua->Mref);
676
677 /* The thread is garbage collected by Lua. */
678 luaL_unref(gL.T, LUA_REGISTRYINDEX, lua->Tref);
679
680 /* Fill the struct with the new coroutine values. */
681 lua->Mref = new_ref;
682 lua->T = T;
683 lua->Tref = luaL_ref(gL.T, LUA_REGISTRYINDEX);
684
685 /* Set context. */
686 hlua_sethlua(lua);
687
688 return 1;
689}
690
Thierry FOURNIERee9f8022015-03-03 17:37:37 +0100691void hlua_hook(lua_State *L, lua_Debug *ar)
692{
Thierry FOURNIERcae49c92015-03-06 14:05:24 +0100693 struct hlua *hlua = hlua_gethlua(L);
694
695 /* Lua cannot yield when its returning from a function,
696 * so, we can fix the interrupt hook to 1 instruction,
697 * expecting that the function is finnished.
698 */
699 if (lua_gethookmask(L) & LUA_MASKRET) {
700 lua_sethook(hlua->T, hlua_hook, LUA_MASKCOUNT, 1);
701 return;
702 }
703
704 /* restore the interrupt condition. */
705 lua_sethook(hlua->T, hlua_hook, LUA_MASKCOUNT, hlua_nb_instruction);
706
707 /* If we interrupt the Lua processing in yieldable state, we yield.
708 * If the state is not yieldable, trying yield causes an error.
709 */
710 if (lua_isyieldable(L))
711 WILL_LJMP(hlua_yieldk(L, 0, 0, NULL, TICK_ETERNITY, HLUA_CTRLYIELD));
712
713 /* If we cannot yield, check the timeout. */
714 if (tick_is_expired(hlua->expire, now_ms)) {
715 lua_pushfstring(L, "execution timeout");
716 WILL_LJMP(lua_error(L));
717 }
718
719 /* Try to interrupt the process at the end of the current
720 * unyieldable function.
721 */
722 lua_sethook(hlua->T, hlua_hook, LUA_MASKRET|LUA_MASKCOUNT, hlua_nb_instruction);
Thierry FOURNIERee9f8022015-03-03 17:37:37 +0100723}
724
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100725/* This function start or resumes the Lua stack execution. If the flag
726 * "yield_allowed" if no set and the LUA stack execution returns a yield
727 * The function return an error.
728 *
729 * The function can returns 4 values:
730 * - HLUA_E_OK : The execution is terminated without any errors.
731 * - HLUA_E_AGAIN : The execution must continue at the next associated
732 * task wakeup.
733 * - HLUA_E_ERRMSG : An error has occured, an error message is set in
734 * the top of the stack.
735 * - HLUA_E_ERR : An error has occured without error message.
736 *
737 * If an error occured, the stack is renewed and it is ready to run new
738 * LUA code.
739 */
740static enum hlua_exec hlua_ctx_resume(struct hlua *lua, int yield_allowed)
741{
742 int ret;
743 const char *msg;
744
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +0100745 HLUA_SET_RUN(lua);
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100746
Thierry FOURNIERdc9ca962015-03-05 11:16:52 +0100747 /* If we want to resume the task, then check first the execution timeout.
748 * if it is reached, we can interrupt the Lua processing.
749 */
750 if (tick_is_expired(lua->expire, now_ms))
751 goto timeout_reached;
752
Thierry FOURNIERee9f8022015-03-03 17:37:37 +0100753resume_execution:
754
755 /* This hook interrupts the Lua processing each 'hlua_nb_instruction'
756 * instructions. it is used for preventing infinite loops.
757 */
758 lua_sethook(lua->T, hlua_hook, LUA_MASKCOUNT, hlua_nb_instruction);
759
Thierry FOURNIER1bfc09b2015-03-05 17:10:14 +0100760 /* Remove all flags except the running flags. */
761 lua->flags = HLUA_RUN;
762
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100763 /* Call the function. */
764 ret = lua_resume(lua->T, gL.T, lua->nargs);
765 switch (ret) {
766
767 case LUA_OK:
768 ret = HLUA_E_OK;
769 break;
770
771 case LUA_YIELD:
Thierry FOURNIERdc9ca962015-03-05 11:16:52 +0100772 /* Check if the execution timeout is expired. It it is the case, we
773 * break the Lua execution.
Thierry FOURNIERee9f8022015-03-03 17:37:37 +0100774 */
Thierry FOURNIERdc9ca962015-03-05 11:16:52 +0100775 if (tick_is_expired(lua->expire, now_ms)) {
776
777timeout_reached:
778
779 lua_settop(lua->T, 0); /* Empty the stack. */
780 if (!lua_checkstack(lua->T, 1)) {
781 ret = HLUA_E_ERR;
Thierry FOURNIERee9f8022015-03-03 17:37:37 +0100782 break;
783 }
Thierry FOURNIERdc9ca962015-03-05 11:16:52 +0100784 lua_pushfstring(lua->T, "execution timeout");
785 ret = HLUA_E_ERRMSG;
786 break;
787 }
788 /* Process the forced yield. if the general yield is not allowed or
789 * if no task were associated this the current Lua execution
790 * coroutine, we resume the execution. Else we want to return in the
791 * scheduler and we want to be waked up again, to continue the
792 * current Lua execution. So we schedule our own task.
793 */
794 if (HLUA_IS_CTRLYIELDING(lua)) {
Thierry FOURNIERee9f8022015-03-03 17:37:37 +0100795 if (!yield_allowed || !lua->task)
796 goto resume_execution;
Thierry FOURNIERee9f8022015-03-03 17:37:37 +0100797 task_wakeup(lua->task, TASK_WOKEN_MSG);
Thierry FOURNIERee9f8022015-03-03 17:37:37 +0100798 }
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100799 if (!yield_allowed) {
800 lua_settop(lua->T, 0); /* Empty the stack. */
801 if (!lua_checkstack(lua->T, 1)) {
802 ret = HLUA_E_ERR;
803 break;
804 }
805 lua_pushfstring(lua->T, "yield not allowed");
806 ret = HLUA_E_ERRMSG;
807 break;
808 }
809 ret = HLUA_E_AGAIN;
810 break;
811
812 case LUA_ERRRUN:
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +0100813 lua->wake_time = TICK_ETERNITY;
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100814 if (!lua_checkstack(lua->T, 1)) {
815 ret = HLUA_E_ERR;
816 break;
817 }
818 msg = lua_tostring(lua->T, -1);
819 lua_settop(lua->T, 0); /* Empty the stack. */
820 lua_pop(lua->T, 1);
821 if (msg)
822 lua_pushfstring(lua->T, "runtime error: %s", msg);
823 else
824 lua_pushfstring(lua->T, "unknown runtime error");
825 ret = HLUA_E_ERRMSG;
826 break;
827
828 case LUA_ERRMEM:
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +0100829 lua->wake_time = TICK_ETERNITY;
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100830 lua_settop(lua->T, 0); /* Empty the stack. */
831 if (!lua_checkstack(lua->T, 1)) {
832 ret = HLUA_E_ERR;
833 break;
834 }
835 lua_pushfstring(lua->T, "out of memory error");
836 ret = HLUA_E_ERRMSG;
837 break;
838
839 case LUA_ERRERR:
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +0100840 lua->wake_time = TICK_ETERNITY;
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100841 if (!lua_checkstack(lua->T, 1)) {
842 ret = HLUA_E_ERR;
843 break;
844 }
845 msg = lua_tostring(lua->T, -1);
846 lua_settop(lua->T, 0); /* Empty the stack. */
847 lua_pop(lua->T, 1);
848 if (msg)
849 lua_pushfstring(lua->T, "message handler error: %s", msg);
850 else
851 lua_pushfstring(lua->T, "message handler error");
852 ret = HLUA_E_ERRMSG;
853 break;
854
855 default:
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +0100856 lua->wake_time = TICK_ETERNITY;
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100857 lua_settop(lua->T, 0); /* Empty the stack. */
858 if (!lua_checkstack(lua->T, 1)) {
859 ret = HLUA_E_ERR;
860 break;
861 }
862 lua_pushfstring(lua->T, "unknonwn error");
863 ret = HLUA_E_ERRMSG;
864 break;
865 }
866
867 switch (ret) {
868 case HLUA_E_AGAIN:
869 break;
870
871 case HLUA_E_ERRMSG:
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +0100872 hlua_com_purge(lua);
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100873 hlua_ctx_renew(lua, 1);
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +0100874 HLUA_CLR_RUN(lua);
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100875 break;
876
877 case HLUA_E_ERR:
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +0100878 HLUA_CLR_RUN(lua);
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +0100879 hlua_com_purge(lua);
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100880 hlua_ctx_renew(lua, 0);
881 break;
882
883 case HLUA_E_OK:
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +0100884 HLUA_CLR_RUN(lua);
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +0100885 hlua_com_purge(lua);
Thierry FOURNIER380d0932015-01-23 14:27:52 +0100886 break;
887 }
888
889 return ret;
890}
891
Thierry FOURNIER83758bb2015-02-04 13:21:04 +0100892/* This function is an LUA binding. It provides a function
893 * for deleting ACL from a referenced ACL file.
894 */
895__LJMP static int hlua_del_acl(lua_State *L)
896{
897 const char *name;
898 const char *key;
899 struct pat_ref *ref;
900
901 MAY_LJMP(check_args(L, 2, "del_acl"));
902
903 name = MAY_LJMP(luaL_checkstring(L, 1));
904 key = MAY_LJMP(luaL_checkstring(L, 2));
905
906 ref = pat_ref_lookup(name);
907 if (!ref)
908 WILL_LJMP(luaL_error(L, "'del_acl': unkown acl file '%s'", name));
909
910 pat_ref_delete(ref, key);
911 return 0;
912}
913
914/* This function is an LUA binding. It provides a function
915 * for deleting map entry from a referenced map file.
916 */
917static int hlua_del_map(lua_State *L)
918{
919 const char *name;
920 const char *key;
921 struct pat_ref *ref;
922
923 MAY_LJMP(check_args(L, 2, "del_map"));
924
925 name = MAY_LJMP(luaL_checkstring(L, 1));
926 key = MAY_LJMP(luaL_checkstring(L, 2));
927
928 ref = pat_ref_lookup(name);
929 if (!ref)
930 WILL_LJMP(luaL_error(L, "'del_map': unkown acl file '%s'", name));
931
932 pat_ref_delete(ref, key);
933 return 0;
934}
935
936/* This function is an LUA binding. It provides a function
937 * for adding ACL pattern from a referenced ACL file.
938 */
939static int hlua_add_acl(lua_State *L)
940{
941 const char *name;
942 const char *key;
943 struct pat_ref *ref;
944
945 MAY_LJMP(check_args(L, 2, "add_acl"));
946
947 name = MAY_LJMP(luaL_checkstring(L, 1));
948 key = MAY_LJMP(luaL_checkstring(L, 2));
949
950 ref = pat_ref_lookup(name);
951 if (!ref)
952 WILL_LJMP(luaL_error(L, "'add_acl': unkown acl file '%s'", name));
953
954 if (pat_ref_find_elt(ref, key) == NULL)
955 pat_ref_add(ref, key, NULL, NULL);
956 return 0;
957}
958
959/* This function is an LUA binding. It provides a function
960 * for setting map pattern and sample from a referenced map
961 * file.
962 */
963static int hlua_set_map(lua_State *L)
964{
965 const char *name;
966 const char *key;
967 const char *value;
968 struct pat_ref *ref;
969
970 MAY_LJMP(check_args(L, 3, "set_map"));
971
972 name = MAY_LJMP(luaL_checkstring(L, 1));
973 key = MAY_LJMP(luaL_checkstring(L, 2));
974 value = MAY_LJMP(luaL_checkstring(L, 3));
975
976 ref = pat_ref_lookup(name);
977 if (!ref)
978 WILL_LJMP(luaL_error(L, "'set_map': unkown map file '%s'", name));
979
980 if (pat_ref_find_elt(ref, key) != NULL)
981 pat_ref_set(ref, key, value, NULL);
982 else
983 pat_ref_add(ref, key, value, NULL);
984 return 0;
985}
986
Thierry FOURNIER65f34c62015-02-16 20:11:43 +0100987/* A class is a lot of memory that contain data. This data can be a table,
988 * an integer or user data. This data is associated with a metatable. This
989 * metatable have an original version registred in the global context with
990 * the name of the object (_G[<name>] = <metable> ).
991 *
992 * A metable is a table that modify the standard behavior of a standard
993 * access to the associated data. The entries of this new metatable are
994 * defined as is:
995 *
996 * http://lua-users.org/wiki/MetatableEvents
997 *
998 * __index
999 *
1000 * we access an absent field in a table, the result is nil. This is
1001 * true, but it is not the whole truth. Actually, such access triggers
1002 * the interpreter to look for an __index metamethod: If there is no
1003 * such method, as usually happens, then the access results in nil;
1004 * otherwise, the metamethod will provide the result.
1005 *
1006 * Control 'prototype' inheritance. When accessing "myTable[key]" and
1007 * the key does not appear in the table, but the metatable has an __index
1008 * property:
1009 *
1010 * - if the value is a function, the function is called, passing in the
1011 * table and the key; the return value of that function is returned as
1012 * the result.
1013 *
1014 * - if the value is another table, the value of the key in that table is
1015 * asked for and returned (and if it doesn't exist in that table, but that
1016 * table's metatable has an __index property, then it continues on up)
1017 *
1018 * - Use "rawget(myTable,key)" to skip this metamethod.
1019 *
1020 * http://www.lua.org/pil/13.4.1.html
1021 *
1022 * __newindex
1023 *
1024 * Like __index, but control property assignment.
1025 *
1026 * __mode - Control weak references. A string value with one or both
1027 * of the characters 'k' and 'v' which specifies that the the
1028 * keys and/or values in the table are weak references.
1029 *
1030 * __call - Treat a table like a function. When a table is followed by
1031 * parenthesis such as "myTable( 'foo' )" and the metatable has
1032 * a __call key pointing to a function, that function is invoked
1033 * (passing any specified arguments) and the return value is
1034 * returned.
1035 *
1036 * __metatable - Hide the metatable. When "getmetatable( myTable )" is
1037 * called, if the metatable for myTable has a __metatable
1038 * key, the value of that key is returned instead of the
1039 * actual metatable.
1040 *
1041 * __tostring - Control string representation. When the builtin
1042 * "tostring( myTable )" function is called, if the metatable
1043 * for myTable has a __tostring property set to a function,
1044 * that function is invoked (passing myTable to it) and the
1045 * return value is used as the string representation.
1046 *
1047 * __len - Control table length. When the table length is requested using
1048 * the length operator ( '#' ), if the metatable for myTable has
1049 * a __len key pointing to a function, that function is invoked
1050 * (passing myTable to it) and the return value used as the value
1051 * of "#myTable".
1052 *
1053 * __gc - Userdata finalizer code. When userdata is set to be garbage
1054 * collected, if the metatable has a __gc field pointing to a
1055 * function, that function is first invoked, passing the userdata
1056 * to it. The __gc metamethod is not called for tables.
1057 * (See http://lua-users.org/lists/lua-l/2006-11/msg00508.html)
1058 *
1059 * Special metamethods for redefining standard operators:
1060 * http://www.lua.org/pil/13.1.html
1061 *
1062 * __add "+"
1063 * __sub "-"
1064 * __mul "*"
1065 * __div "/"
1066 * __unm "!"
1067 * __pow "^"
1068 * __concat ".."
1069 *
1070 * Special methods for redfining standar relations
1071 * http://www.lua.org/pil/13.2.html
1072 *
1073 * __eq "=="
1074 * __lt "<"
1075 * __le "<="
1076 */
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001077
1078/*
1079 *
1080 *
1081 * Class Socket
1082 *
1083 *
1084 */
1085
1086__LJMP static struct hlua_socket *hlua_checksocket(lua_State *L, int ud)
1087{
1088 return (struct hlua_socket *)MAY_LJMP(hlua_checkudata(L, ud, class_socket_ref));
1089}
1090
1091/* This function is the handler called for each I/O on the established
1092 * connection. It is used for notify space avalaible to send or data
1093 * received.
1094 */
1095static void hlua_socket_handler(struct stream_interface *si)
1096{
1097 struct appctx *appctx = objt_appctx(si->end);
Willy Tarreau50fe03b2014-11-28 13:59:31 +01001098 struct connection *c = objt_conn(si_opposite(si)->end);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001099
1100 /* Wakeup the main session if the client connection is closed. */
Willy Tarreau2bb4a962014-11-28 11:11:05 +01001101 if (!c || channel_output_closed(si_ic(si)) || channel_input_closed(si_oc(si))) {
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001102 if (appctx->ctx.hlua.socket) {
1103 appctx->ctx.hlua.socket->s = NULL;
1104 appctx->ctx.hlua.socket = NULL;
1105 }
1106 si_shutw(si);
1107 si_shutr(si);
Willy Tarreau2bb4a962014-11-28 11:11:05 +01001108 si_ic(si)->flags |= CF_READ_NULL;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001109 hlua_com_wake(&appctx->ctx.hlua.wake_on_read);
1110 hlua_com_wake(&appctx->ctx.hlua.wake_on_write);
1111 return;
1112 }
1113
1114 if (!(c->flags & CO_FL_CONNECTED))
1115 return;
1116
1117 /* This function is called after the connect. */
1118 appctx->ctx.hlua.connected = 1;
1119
1120 /* Wake the tasks which wants to write if the buffer have avalaible space. */
Willy Tarreau2bb4a962014-11-28 11:11:05 +01001121 if (channel_may_recv(si_oc(si)))
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001122 hlua_com_wake(&appctx->ctx.hlua.wake_on_write);
1123
1124 /* Wake the tasks which wants to read if the buffer contains data. */
Willy Tarreau2bb4a962014-11-28 11:11:05 +01001125 if (channel_is_empty(si_ic(si)))
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001126 hlua_com_wake(&appctx->ctx.hlua.wake_on_read);
1127}
1128
1129/* This function is called when the "struct session" is destroyed.
1130 * Remove the link from the object to this session.
1131 * Wake all the pending signals.
1132 */
1133static void hlua_socket_release(struct stream_interface *si)
1134{
1135 struct appctx *appctx = objt_appctx(si->end);
1136
1137 /* Remove my link in the original object. */
1138 if (appctx->ctx.hlua.socket)
1139 appctx->ctx.hlua.socket->s = NULL;
1140
1141 /* Wake all the task waiting for me. */
1142 hlua_com_wake(&appctx->ctx.hlua.wake_on_read);
1143 hlua_com_wake(&appctx->ctx.hlua.wake_on_write);
1144}
1145
1146/* If the garbage collectio of the object is launch, nobody
1147 * uses this object. If the session does not exists, just quit.
1148 * Send the shutdown signal to the session. In some cases,
1149 * pending signal can rest in the read and write lists. destroy
1150 * it.
1151 */
1152__LJMP static int hlua_socket_gc(lua_State *L)
1153{
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001154 struct hlua_socket *socket;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001155 struct appctx *appctx;
1156
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001157 MAY_LJMP(check_args(L, 1, "__gc"));
1158
1159 socket = MAY_LJMP(hlua_checksocket(L, 1));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001160 if (!socket->s)
1161 return 0;
1162
1163 /* Remove all reference between the Lua stack and the coroutine session. */
1164 appctx = objt_appctx(socket->s->si[0].end);
1165 session_shutdown(socket->s, SN_ERR_KILLED);
1166 socket->s = NULL;
1167 appctx->ctx.hlua.socket = NULL;
1168
1169 return 0;
1170}
1171
1172/* The close function send shutdown signal and break the
1173 * links between the session and the object.
1174 */
1175__LJMP static int hlua_socket_close(lua_State *L)
1176{
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001177 struct hlua_socket *socket;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001178 struct appctx *appctx;
1179
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001180 MAY_LJMP(check_args(L, 1, "close"));
1181
1182 socket = MAY_LJMP(hlua_checksocket(L, 1));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001183 if (!socket->s)
1184 return 0;
1185
1186 /* Close the session and remove the associated stop task. */
1187 session_shutdown(socket->s, SN_ERR_KILLED);
1188 appctx = objt_appctx(socket->s->si[0].end);
1189 appctx->ctx.hlua.socket = NULL;
1190 socket->s = NULL;
1191
1192 return 0;
1193}
1194
1195/* This Lua function assumes that the stack contain three parameters.
1196 * 1 - USERDATA containing a struct socket
1197 * 2 - INTEGER with values of the macro defined below
1198 * If the integer is -1, we must read at most one line.
1199 * If the integer is -2, we ust read all the data until the
1200 * end of the stream.
1201 * If the integer is positive value, we must read a number of
1202 * bytes corresponding to this value.
1203 */
1204#define HLSR_READ_LINE (-1)
1205#define HLSR_READ_ALL (-2)
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001206__LJMP static int hlua_socket_receive_yield(struct lua_State *L, int status, lua_KContext ctx)
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001207{
1208 struct hlua_socket *socket = MAY_LJMP(hlua_checksocket(L, 1));
1209 int wanted = lua_tointeger(L, 2);
1210 struct hlua *hlua = hlua_gethlua(L);
1211 struct appctx *appctx;
1212 int len;
1213 int nblk;
1214 char *blk1;
1215 int len1;
1216 char *blk2;
1217 int len2;
Thierry FOURNIER00543922015-03-09 18:35:06 +01001218 int skip_at_end = 0;
Willy Tarreau81389672015-03-10 12:03:52 +01001219 struct channel *oc;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001220
1221 /* Check if this lua stack is schedulable. */
1222 if (!hlua || !hlua->task)
1223 WILL_LJMP(luaL_error(L, "The 'receive' function is only allowed in "
1224 "'frontend', 'backend' or 'task'"));
1225
1226 /* check for connection closed. If some data where read, return it. */
1227 if (!socket->s)
1228 goto connection_closed;
1229
Willy Tarreau81389672015-03-10 12:03:52 +01001230 oc = si_oc(&socket->s->si[0]);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001231 if (wanted == HLSR_READ_LINE) {
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001232 /* Read line. */
Willy Tarreau81389672015-03-10 12:03:52 +01001233 nblk = bo_getline_nc(oc, &blk1, &len1, &blk2, &len2);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001234 if (nblk < 0) /* Connection close. */
1235 goto connection_closed;
1236 if (nblk == 0) /* No data avalaible. */
1237 goto connection_empty;
Thierry FOURNIER00543922015-03-09 18:35:06 +01001238
1239 /* remove final \r\n. */
1240 if (nblk == 1) {
1241 if (blk1[len1-1] == '\n') {
1242 len1--;
1243 skip_at_end++;
1244 if (blk1[len1-1] == '\r') {
1245 len1--;
1246 skip_at_end++;
1247 }
1248 }
1249 }
1250 else {
1251 if (blk2[len2-1] == '\n') {
1252 len2--;
1253 skip_at_end++;
1254 if (blk2[len2-1] == '\r') {
1255 len2--;
1256 skip_at_end++;
1257 }
1258 }
1259 }
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001260 }
1261
1262 else if (wanted == HLSR_READ_ALL) {
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001263 /* Read all the available data. */
Willy Tarreau81389672015-03-10 12:03:52 +01001264 nblk = bo_getblk_nc(oc, &blk1, &len1, &blk2, &len2);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001265 if (nblk < 0) /* Connection close. */
1266 goto connection_closed;
1267 if (nblk == 0) /* No data avalaible. */
1268 goto connection_empty;
1269 }
1270
1271 else {
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001272 /* Read a block of data. */
Willy Tarreau81389672015-03-10 12:03:52 +01001273 nblk = bo_getblk_nc(oc, &blk1, &len1, &blk2, &len2);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001274 if (nblk < 0) /* Connection close. */
1275 goto connection_closed;
1276 if (nblk == 0) /* No data avalaible. */
1277 goto connection_empty;
1278
1279 if (len1 > wanted) {
1280 nblk = 1;
1281 len1 = wanted;
1282 } if (nblk == 2 && len1 + len2 > wanted)
1283 len2 = wanted - len1;
1284 }
1285
1286 len = len1;
1287
1288 luaL_addlstring(&socket->b, blk1, len1);
1289 if (nblk == 2) {
1290 len += len2;
1291 luaL_addlstring(&socket->b, blk2, len2);
1292 }
1293
1294 /* Consume data. */
Willy Tarreau81389672015-03-10 12:03:52 +01001295 bo_skip(oc, len + skip_at_end);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001296
1297 /* Don't wait anything. */
1298 si_update(&socket->s->si[0]);
1299
1300 /* If the pattern reclaim to read all the data
1301 * in the connection, got out.
1302 */
1303 if (wanted == HLSR_READ_ALL)
1304 goto connection_empty;
1305 else if (wanted >= 0 && len < wanted)
1306 goto connection_empty;
1307
1308 /* Return result. */
1309 luaL_pushresult(&socket->b);
1310 return 1;
1311
1312connection_closed:
1313
1314 /* If the buffer containds data. */
1315 if (socket->b.n > 0) {
1316 luaL_pushresult(&socket->b);
1317 return 1;
1318 }
1319 lua_pushnil(L);
1320 lua_pushstring(L, "connection closed.");
1321 return 2;
1322
1323connection_empty:
1324
1325 appctx = objt_appctx(socket->s->si[0].end);
1326 if (!hlua_com_new(hlua, &appctx->ctx.hlua.wake_on_read))
1327 WILL_LJMP(luaL_error(L, "out of memory"));
Thierry FOURNIER4abd3ae2015-03-03 17:29:06 +01001328 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_socket_receive_yield, TICK_ETERNITY, 0));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001329 return 0;
1330}
1331
1332/* This Lus function gets two parameters. The first one can be string
1333 * or a number. If the string is "*l", the user require one line. If
1334 * the string is "*a", the user require all the content of the stream.
1335 * If the value is a number, the user require a number of bytes equal
1336 * to the value. The default value is "*l" (a line).
1337 *
1338 * This paraeter with a variable type is converted in integer. This
1339 * integer takes this values:
1340 * -1 : read a line
1341 * -2 : read all the stream
1342 * >0 : amount if bytes.
1343 *
1344 * The second parameter is optinal. It contains a string that must be
1345 * concatenated with the read data.
1346 */
1347__LJMP static int hlua_socket_receive(struct lua_State *L)
1348{
1349 int wanted = HLSR_READ_LINE;
1350 const char *pattern;
1351 int type;
1352 char *error;
1353 size_t len;
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001354 struct hlua_socket *socket;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001355
1356 if (lua_gettop(L) < 1 || lua_gettop(L) > 3)
1357 WILL_LJMP(luaL_error(L, "The 'receive' function requires between 1 and 3 arguments."));
1358
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001359 socket = MAY_LJMP(hlua_checksocket(L, 1));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001360
1361 /* check for pattern. */
1362 if (lua_gettop(L) >= 2) {
1363 type = lua_type(L, 2);
1364 if (type == LUA_TSTRING) {
1365 pattern = lua_tostring(L, 2);
1366 if (strcmp(pattern, "*a") == 0)
1367 wanted = HLSR_READ_ALL;
1368 else if (strcmp(pattern, "*l") == 0)
1369 wanted = HLSR_READ_LINE;
1370 else {
1371 wanted = strtoll(pattern, &error, 10);
1372 if (*error != '\0')
1373 WILL_LJMP(luaL_error(L, "Unsupported pattern."));
1374 }
1375 }
1376 else if (type == LUA_TNUMBER) {
1377 wanted = lua_tointeger(L, 2);
1378 if (wanted < 0)
1379 WILL_LJMP(luaL_error(L, "Unsupported size."));
1380 }
1381 }
1382
1383 /* Set pattern. */
1384 lua_pushinteger(L, wanted);
1385 lua_replace(L, 2);
1386
1387 /* init bufffer, and fiil it wih prefix. */
1388 luaL_buffinit(L, &socket->b);
1389
1390 /* Check prefix. */
1391 if (lua_gettop(L) >= 3) {
1392 if (lua_type(L, 3) != LUA_TSTRING)
1393 WILL_LJMP(luaL_error(L, "Expect a 'string' for the prefix"));
1394 pattern = lua_tolstring(L, 3, &len);
1395 luaL_addlstring(&socket->b, pattern, len);
1396 }
1397
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001398 return __LJMP(hlua_socket_receive_yield(L, 0, 0));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001399}
1400
1401/* Write the Lua input string in the output buffer.
1402 * This fucntion returns a yield if no space are available.
1403 */
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001404static int hlua_socket_write_yield(struct lua_State *L,int status, lua_KContext ctx)
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001405{
1406 struct hlua_socket *socket;
1407 struct hlua *hlua = hlua_gethlua(L);
1408 struct appctx *appctx;
1409 size_t buf_len;
1410 const char *buf;
1411 int len;
1412 int send_len;
1413 int sent;
1414
1415 /* Check if this lua stack is schedulable. */
1416 if (!hlua || !hlua->task)
1417 WILL_LJMP(luaL_error(L, "The 'write' function is only allowed in "
1418 "'frontend', 'backend' or 'task'"));
1419
1420 /* Get object */
1421 socket = MAY_LJMP(hlua_checksocket(L, 1));
1422 buf = MAY_LJMP(luaL_checklstring(L, 2, &buf_len));
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001423 sent = MAY_LJMP(luaL_checkinteger(L, 3));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001424
1425 /* Check for connection close. */
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001426 if (!socket->s || channel_output_closed(&socket->s->req)) {
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001427 lua_pushinteger(L, -1);
1428 return 1;
1429 }
1430
1431 /* Update the input buffer data. */
1432 buf += sent;
1433 send_len = buf_len - sent;
1434
1435 /* All the data are sent. */
1436 if (sent >= buf_len)
1437 return 1; /* Implicitly return the length sent. */
1438
Thierry FOURNIER486d52a2015-03-09 17:51:43 +01001439 /* Check if the buffer is avalaible because HAProxy doesn't allocate
1440 * the request buffer if its not required.
1441 */
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001442 if (socket->s->req.buf->size == 0) {
Willy Tarreau78955f42014-12-28 13:09:02 +01001443 if (!session_alloc_recv_buffer(&socket->s->req)) {
Willy Tarreau350f4872014-11-28 14:42:25 +01001444 socket->s->si[0].flags |= SI_FL_WAIT_ROOM;
Thierry FOURNIER486d52a2015-03-09 17:51:43 +01001445 goto hlua_socket_write_yield_return;
1446 }
1447 }
1448
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001449 /* Check for avalaible space. */
Willy Tarreau4e4292b2014-11-28 12:18:45 +01001450 len = buffer_total_space(si_ib(&socket->s->si[0]));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001451 if (len <= 0)
1452 goto hlua_socket_write_yield_return;
1453
1454 /* send data */
1455 if (len < send_len)
1456 send_len = len;
Willy Tarreau2bb4a962014-11-28 11:11:05 +01001457 len = bi_putblk(si_ic(&socket->s->si[0]), buf+sent, send_len);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001458
1459 /* "Not enough space" (-1), "Buffer too little to contain
1460 * the data" (-2) are not expected because the available length
1461 * is tested.
1462 * Other unknown error are also not expected.
1463 */
1464 if (len <= 0) {
1465 MAY_LJMP(hlua_socket_close(L));
1466 lua_pop(L, 1);
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001467 lua_pushinteger(L, -1);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001468 return 1;
1469 }
1470
1471 /* update buffers. */
1472 si_update(&socket->s->si[0]);
Willy Tarreau2bb4a962014-11-28 11:11:05 +01001473 si_ic(&socket->s->si[0])->rex = TICK_ETERNITY;
1474 si_oc(&socket->s->si[0])->wex = TICK_ETERNITY;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001475
1476 /* Update length sent. */
1477 lua_pop(L, 1);
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001478 lua_pushinteger(L, sent + len);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001479
1480 /* All the data buffer is sent ? */
1481 if (sent + len >= buf_len)
1482 return 1;
1483
1484hlua_socket_write_yield_return:
1485 appctx = objt_appctx(socket->s->si[0].end);
1486 if (!hlua_com_new(hlua, &appctx->ctx.hlua.wake_on_write))
1487 WILL_LJMP(luaL_error(L, "out of memory"));
Thierry FOURNIER4abd3ae2015-03-03 17:29:06 +01001488 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_socket_write_yield, TICK_ETERNITY, 0));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001489 return 0;
1490}
1491
1492/* This function initiate the send of data. It just check the input
1493 * parameters and push an integer in the Lua stack that contain the
1494 * amount of data writed in the buffer. This is used by the function
1495 * "hlua_socket_write_yield" that can yield.
1496 *
1497 * The Lua function gets between 3 and 4 parameters. The first one is
1498 * the associated object. The second is a string buffer. The third is
1499 * a facultative integer that represents where is the buffer position
1500 * of the start of the data that can send. The first byte is the
1501 * position "1". The default value is "1". The fourth argument is a
1502 * facultative integer that represents where is the buffer position
1503 * of the end of the data that can send. The default is the last byte.
1504 */
1505static int hlua_socket_send(struct lua_State *L)
1506{
1507 int i;
1508 int j;
1509 const char *buf;
1510 size_t buf_len;
1511
1512 /* Check number of arguments. */
1513 if (lua_gettop(L) < 2 || lua_gettop(L) > 4)
1514 WILL_LJMP(luaL_error(L, "'send' needs between 2 and 4 arguments"));
1515
1516 /* Get the string. */
1517 buf = MAY_LJMP(luaL_checklstring(L, 2, &buf_len));
1518
1519 /* Get and check j. */
1520 if (lua_gettop(L) == 4) {
1521 j = MAY_LJMP(luaL_checkinteger(L, 4));
1522 if (j < 0)
1523 j = buf_len + j + 1;
1524 if (j > buf_len)
1525 j = buf_len + 1;
1526 lua_pop(L, 1);
1527 }
1528 else
1529 j = buf_len;
1530
1531 /* Get and check i. */
1532 if (lua_gettop(L) == 3) {
1533 i = MAY_LJMP(luaL_checkinteger(L, 3));
1534 if (i < 0)
1535 i = buf_len + i + 1;
1536 if (i > buf_len)
1537 i = buf_len + 1;
1538 lua_pop(L, 1);
1539 } else
1540 i = 1;
1541
1542 /* Check bth i and j. */
1543 if (i > j) {
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001544 lua_pushinteger(L, 0);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001545 return 1;
1546 }
1547 if (i == 0 && j == 0) {
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001548 lua_pushinteger(L, 0);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001549 return 1;
1550 }
1551 if (i == 0)
1552 i = 1;
1553 if (j == 0)
1554 j = 1;
1555
1556 /* Pop the string. */
1557 lua_pop(L, 1);
1558
1559 /* Update the buffer length. */
1560 buf += i - 1;
1561 buf_len = j - i + 1;
1562 lua_pushlstring(L, buf, buf_len);
1563
1564 /* This unsigned is used to remember the amount of sent data. */
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001565 lua_pushinteger(L, 0);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001566
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001567 return MAY_LJMP(hlua_socket_write_yield(L, 0, 0));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001568}
1569
1570#define SOCKET_INFO_EXPANDED_FORM "[0000:0000:0000:0000:0000:0000:0000:0000]:12345"
1571static char _socket_info_expanded_form[] = SOCKET_INFO_EXPANDED_FORM;
1572#define SOCKET_INFO_MAX_LEN (sizeof(_socket_info_expanded_form))
1573__LJMP static inline int hlua_socket_info(struct lua_State *L, struct sockaddr_storage *addr)
1574{
1575 static char buffer[SOCKET_INFO_MAX_LEN];
1576 int ret;
1577 int len;
1578 char *p;
1579
1580 ret = addr_to_str(addr, buffer+1, SOCKET_INFO_MAX_LEN-1);
1581 if (ret <= 0) {
1582 lua_pushnil(L);
1583 return 1;
1584 }
1585
1586 if (ret == AF_UNIX) {
1587 lua_pushstring(L, buffer+1);
1588 return 1;
1589 }
1590 else if (ret == AF_INET6) {
1591 buffer[0] = '[';
1592 len = strlen(buffer);
1593 buffer[len] = ']';
1594 len++;
1595 buffer[len] = ':';
1596 len++;
1597 p = buffer;
1598 }
1599 else if (ret == AF_INET) {
1600 p = buffer + 1;
1601 len = strlen(p);
1602 p[len] = ':';
1603 len++;
1604 }
1605 else {
1606 lua_pushnil(L);
1607 return 1;
1608 }
1609
1610 if (port_to_str(addr, p + len, SOCKET_INFO_MAX_LEN-1 - len) <= 0) {
1611 lua_pushnil(L);
1612 return 1;
1613 }
1614
1615 lua_pushstring(L, p);
1616 return 1;
1617}
1618
1619/* Returns information about the peer of the connection. */
1620__LJMP static int hlua_socket_getpeername(struct lua_State *L)
1621{
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001622 struct hlua_socket *socket;
1623 struct connection *conn;
1624
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001625 MAY_LJMP(check_args(L, 1, "getpeername"));
1626
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001627 socket = MAY_LJMP(hlua_checksocket(L, 1));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001628
1629 /* Check if the tcp object is avalaible. */
1630 if (!socket->s) {
1631 lua_pushnil(L);
1632 return 1;
1633 }
1634
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001635 conn = objt_conn(socket->s->si[1].end);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001636 if (!conn) {
1637 lua_pushnil(L);
1638 return 1;
1639 }
1640
1641 if (!(conn->flags & CO_FL_ADDR_TO_SET)) {
1642 unsigned int salen = sizeof(conn->addr.to);
1643 if (getpeername(conn->t.sock.fd, (struct sockaddr *)&conn->addr.to, &salen) == -1) {
1644 lua_pushnil(L);
1645 return 1;
1646 }
1647 conn->flags |= CO_FL_ADDR_TO_SET;
1648 }
1649
1650 return MAY_LJMP(hlua_socket_info(L, &conn->addr.to));
1651}
1652
1653/* Returns information about my connection side. */
1654static int hlua_socket_getsockname(struct lua_State *L)
1655{
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001656 struct hlua_socket *socket;
1657 struct connection *conn;
1658
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001659 MAY_LJMP(check_args(L, 1, "getsockname"));
1660
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001661 socket = MAY_LJMP(hlua_checksocket(L, 1));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001662
1663 /* Check if the tcp object is avalaible. */
1664 if (!socket->s) {
1665 lua_pushnil(L);
1666 return 1;
1667 }
1668
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001669 conn = objt_conn(socket->s->si[1].end);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001670 if (!conn) {
1671 lua_pushnil(L);
1672 return 1;
1673 }
1674
1675 if (!(conn->flags & CO_FL_ADDR_FROM_SET)) {
1676 unsigned int salen = sizeof(conn->addr.from);
1677 if (getsockname(conn->t.sock.fd, (struct sockaddr *)&conn->addr.from, &salen) == -1) {
1678 lua_pushnil(L);
1679 return 1;
1680 }
1681 conn->flags |= CO_FL_ADDR_FROM_SET;
1682 }
1683
1684 return hlua_socket_info(L, &conn->addr.from);
1685}
1686
1687/* This struct define the applet. */
1688static struct si_applet update_applet = {
1689 .obj_type = OBJ_TYPE_APPLET,
1690 .name = "<LUA_TCP>",
1691 .fct = hlua_socket_handler,
1692 .release = hlua_socket_release,
1693};
1694
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001695__LJMP static int hlua_socket_connect_yield(struct lua_State *L, int status, lua_KContext ctx)
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001696{
1697 struct hlua_socket *socket = MAY_LJMP(hlua_checksocket(L, 1));
1698 struct hlua *hlua = hlua_gethlua(L);
1699 struct appctx *appctx;
1700
1701 /* Check for connection close. */
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001702 if (!hlua || !socket->s || channel_output_closed(&socket->s->req)) {
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001703 lua_pushnil(L);
1704 lua_pushstring(L, "Can't connect");
1705 return 2;
1706 }
1707
1708 appctx = objt_appctx(socket->s->si[0].end);
1709
1710 /* Check for connection established. */
1711 if (appctx->ctx.hlua.connected) {
1712 lua_pushinteger(L, 1);
1713 return 1;
1714 }
1715
1716 if (!hlua_com_new(hlua, &appctx->ctx.hlua.wake_on_write))
1717 WILL_LJMP(luaL_error(L, "out of memory error"));
Thierry FOURNIER4abd3ae2015-03-03 17:29:06 +01001718 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_socket_connect_yield, TICK_ETERNITY, 0));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001719 return 0;
1720}
1721
1722/* This function fail or initite the connection. */
1723__LJMP static int hlua_socket_connect(struct lua_State *L)
1724{
1725 struct hlua_socket *socket;
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001726 int port;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001727 const char *ip;
1728 struct connection *conn;
Thierry FOURNIER95ad96a2015-03-09 18:12:40 +01001729 struct hlua *hlua;
1730 struct appctx *appctx;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001731
1732 MAY_LJMP(check_args(L, 3, "connect"));
1733
1734 /* Get args. */
1735 socket = MAY_LJMP(hlua_checksocket(L, 1));
1736 ip = MAY_LJMP(luaL_checkstring(L, 2));
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001737 port = MAY_LJMP(luaL_checkinteger(L, 3));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001738
Willy Tarreau350f4872014-11-28 14:42:25 +01001739 conn = si_alloc_conn(&socket->s->si[1], 0);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001740 if (!conn)
1741 WILL_LJMP(luaL_error(L, "connect: internal error"));
1742
1743 /* Parse ip address. */
1744 conn->addr.to.ss_family = AF_UNSPEC;
1745 if (!str2ip2(ip, &conn->addr.to, 0))
1746 WILL_LJMP(luaL_error(L, "connect: cannot parse ip address '%s'", ip));
1747
1748 /* Set port. */
1749 if (conn->addr.to.ss_family == AF_INET)
1750 ((struct sockaddr_in *)&conn->addr.to)->sin_port = htons(port);
1751 else if (conn->addr.to.ss_family == AF_INET6)
1752 ((struct sockaddr_in6 *)&conn->addr.to)->sin6_port = htons(port);
1753
1754 /* it is important not to call the wakeup function directly but to
1755 * pass through task_wakeup(), because this one knows how to apply
1756 * priorities to tasks.
1757 */
1758 task_wakeup(socket->s->task, TASK_WOKEN_INIT);
1759
Thierry FOURNIER95ad96a2015-03-09 18:12:40 +01001760 hlua = hlua_gethlua(L);
1761 appctx = objt_appctx(socket->s->si[0].end);
1762 if (!hlua_com_new(hlua, &appctx->ctx.hlua.wake_on_write))
1763 WILL_LJMP(luaL_error(L, "out of memory"));
Thierry FOURNIER4abd3ae2015-03-03 17:29:06 +01001764 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_socket_connect_yield, TICK_ETERNITY, 0));
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001765
1766 return 0;
1767}
1768
Baptiste Assmann84bb4932015-03-02 21:40:06 +01001769#ifdef USE_OPENSSL
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001770__LJMP static int hlua_socket_connect_ssl(struct lua_State *L)
1771{
1772 struct hlua_socket *socket;
1773
1774 MAY_LJMP(check_args(L, 3, "connect_ssl"));
1775 socket = MAY_LJMP(hlua_checksocket(L, 1));
1776 socket->s->target = &socket_ssl.obj_type;
1777 return MAY_LJMP(hlua_socket_connect(L));
1778}
Baptiste Assmann84bb4932015-03-02 21:40:06 +01001779#endif
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001780
1781__LJMP static int hlua_socket_setoption(struct lua_State *L)
1782{
1783 return 0;
1784}
1785
1786__LJMP static int hlua_socket_settimeout(struct lua_State *L)
1787{
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001788 struct hlua_socket *socket;
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001789 int tmout;
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001790
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001791 MAY_LJMP(check_args(L, 2, "settimeout"));
1792
Willy Tarreau80f5fae2015-02-27 16:38:20 +01001793 socket = MAY_LJMP(hlua_checksocket(L, 1));
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01001794 tmout = MAY_LJMP(luaL_checkinteger(L, 2)) * 1000;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001795
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001796 socket->s->req.rto = tmout;
1797 socket->s->req.wto = tmout;
1798 socket->s->res.rto = tmout;
1799 socket->s->res.wto = tmout;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001800
1801 return 0;
1802}
1803
1804__LJMP static int hlua_socket_new(lua_State *L)
1805{
1806 struct hlua_socket *socket;
1807 struct appctx *appctx;
1808
1809 /* Check stack size. */
Thierry FOURNIER2297bc22015-03-11 17:43:33 +01001810 if (!lua_checkstack(L, 3)) {
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001811 hlua_pusherror(L, "socket: full stack");
1812 goto out_fail_conf;
1813 }
1814
Thierry FOURNIER2297bc22015-03-11 17:43:33 +01001815 /* Create the object: obj[0] = userdata. */
1816 lua_newtable(L);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001817 socket = MAY_LJMP(lua_newuserdata(L, sizeof(*socket)));
Thierry FOURNIER2297bc22015-03-11 17:43:33 +01001818 lua_rawseti(L, -2, 0);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001819 memset(socket, 0, sizeof(*socket));
1820
Thierry FOURNIER4a6170c2015-03-09 17:07:10 +01001821 /* Check if the various memory pools are intialized. */
Willy Tarreaua27dc192014-11-27 22:10:04 +01001822 if (!pool2_session || !pool2_buffer) {
Thierry FOURNIER4a6170c2015-03-09 17:07:10 +01001823 hlua_pusherror(L, "socket: uninitialized pools.");
1824 goto out_fail_conf;
1825 }
1826
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001827 /* Pop a class session metatable and affect it to the userdata. */
1828 lua_rawgeti(L, LUA_REGISTRYINDEX, class_socket_ref);
1829 lua_setmetatable(L, -2);
1830
1831 /*
1832 *
1833 * Get memory for the request.
1834 *
1835 */
1836
1837 socket->s = pool_alloc2(pool2_session);
1838 if (!socket->s) {
1839 hlua_pusherror(L, "socket: out of memory");
1840 goto out_fail_conf;
1841 }
1842
1843 socket->s->task = task_new();
1844 if (!socket->s->task) {
1845 hlua_pusherror(L, "socket: out of memory");
1846 goto out_free_session;
1847 }
1848
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001849 socket->s->req.buf = pool_alloc2(pool2_buffer);
1850 if (!socket->s->req.buf) {
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001851 hlua_pusherror(L, "socket: out of memory");
1852 goto out_fail_req_buf;
1853 }
1854
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001855 socket->s->res.buf = pool_alloc2(pool2_buffer);
1856 if (!socket->s->res.buf) {
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001857 hlua_pusherror(L, "socket: out of memory");
1858 goto out_fail_rep_buf;
1859 }
1860
1861 /* Configura empty Lua for the session. */
1862 socket->s->hlua.T = NULL;
1863 socket->s->hlua.Tref = LUA_REFNIL;
1864 socket->s->hlua.Mref = LUA_REFNIL;
1865 socket->s->hlua.nargs = 0;
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +01001866 socket->s->hlua.flags = 0;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001867 LIST_INIT(&socket->s->hlua.com);
1868
1869 /* session initialisation. */
1870 session_init_srv_conn(socket->s);
1871
1872 /*
1873 *
1874 * Configure the associated task.
1875 *
1876 */
1877
1878 /* This is the dedicated function to process the session. This function
1879 * is able to establish the conection, process the timeouts, etc ...
1880 */
1881 socket->s->task->process = process_session;
1882
1883 /* Back reference to session. This is used by process_session(). */
1884 socket->s->task->context = socket->s;
1885
1886 /* The priority of the task is normal. */
1887 socket->s->task->nice = 0;
1888
1889 /* Init the next run to eternity. Later in this function, this task is
1890 * waked.
1891 */
1892 socket->s->task->expire = TICK_ETERNITY;
1893
1894 /*
1895 *
1896 * Initialize the attached buffers
1897 *
1898 */
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001899 socket->s->req.buf->size = global.tune.bufsize;
1900 socket->s->res.buf->size = global.tune.bufsize;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001901
1902 /*
1903 *
1904 * Initialize channels.
1905 *
1906 */
1907
1908 /* This function reset the struct. It must be called
1909 * before the configuration.
1910 */
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001911 channel_init(&socket->s->req);
1912 channel_init(&socket->s->res);
Willy Tarreauef573c02014-11-28 14:17:09 +01001913 socket->s->res.flags |= CF_ISRESP;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001914
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001915 socket->s->req.analysers = 0;
1916 socket->s->req.rto = socket_proxy.timeout.client;
1917 socket->s->req.wto = socket_proxy.timeout.server;
1918 socket->s->req.rex = TICK_ETERNITY;
1919 socket->s->req.wex = TICK_ETERNITY;
1920 socket->s->req.analyse_exp = TICK_ETERNITY;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001921
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001922 socket->s->res.analysers = 0;
1923 socket->s->res.rto = socket_proxy.timeout.server;
1924 socket->s->res.wto = socket_proxy.timeout.client;
1925 socket->s->res.rex = TICK_ETERNITY;
1926 socket->s->res.wex = TICK_ETERNITY;
1927 socket->s->res.analyse_exp = TICK_ETERNITY;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001928
1929 /*
1930 *
1931 * Configure the session.
1932 *
1933 */
1934
1935 /* The session dont have listener. The listener is used with real
1936 * proxies.
1937 */
1938 socket->s->listener = NULL;
1939
1940 /* The flags are initialized to 0. Values are setted later. */
1941 socket->s->flags = 0;
1942
1943 /* Assign the configured proxy to the new session. */
1944 socket->s->be = &socket_proxy;
1945 socket->s->fe = &socket_proxy;
1946
1947 /* XXX: Set namy variables */
1948 socket->s->store_count = 0;
1949 memset(socket->s->stkctr, 0, sizeof(socket->s->stkctr));
1950
1951 /* Configure logs. */
1952 socket->s->logs.logwait = 0;
1953 socket->s->logs.level = 0;
1954 socket->s->logs.accept_date = date; /* user-visible date for logging */
1955 socket->s->logs.tv_accept = now; /* corrected date for internal use */
1956 socket->s->do_log = NULL;
1957
1958 /* Function used if an error is occured. */
1959 socket->s->srv_error = default_srv_error;
1960
1961 /* Init the list of buffers. */
1962 LIST_INIT(&socket->s->buffer_wait);
1963
1964 /* Dont configure the unique ID. */
1965 socket->s->uniq_id = 0;
1966 socket->s->unique_id = NULL;
1967
1968 /* XXX: ? */
1969 socket->s->pend_pos = NULL;
1970
1971 /* XXX: See later. */
1972 socket->s->txn.sessid = NULL;
1973 socket->s->txn.srv_cookie = NULL;
1974 socket->s->txn.cli_cookie = NULL;
1975 socket->s->txn.uri = NULL;
1976 socket->s->txn.req.cap = NULL;
1977 socket->s->txn.rsp.cap = NULL;
1978 socket->s->txn.hdr_idx.v = NULL;
1979 socket->s->txn.hdr_idx.size = 0;
1980 socket->s->txn.hdr_idx.used = 0;
1981
1982 /* Configure "left" stream interface as applet. This "si" produce
1983 * and use the data received from the server. The applet is initialized
1984 * and is attached to the stream interface.
1985 */
1986
1987 /* The data producer is already connected. It is the applet. */
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001988 socket->s->req.flags = CF_READ_ATTACHED;
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001989
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01001990 channel_auto_connect(&socket->s->req); /* don't wait to establish connection */
1991 channel_auto_close(&socket->s->req); /* let the producer forward close requests */
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001992
Willy Tarreaua5f5d8d2014-11-28 11:26:07 +01001993 socket->s->si[0].flags = SI_FL_NONE;
Willy Tarreau819d3322014-11-28 12:12:34 +01001994 si_reset(&socket->s->si[0]);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01001995 si_set_state(&socket->s->si[0], SI_ST_EST); /* connection established (resource exists) */
1996
1997 appctx = stream_int_register_handler(&socket->s->si[0], &update_applet);
1998 if (!appctx)
1999 goto out_fail_conn1;
2000 appctx->ctx.hlua.socket = socket;
2001 appctx->ctx.hlua.connected = 0;
2002 LIST_INIT(&appctx->ctx.hlua.wake_on_write);
2003 LIST_INIT(&appctx->ctx.hlua.wake_on_read);
2004
2005 /* Configure "right" stream interface. this "si" is used to connect
2006 * and retrieve data from the server. The connection is initialized
2007 * with the "struct server".
2008 */
Willy Tarreaua5f5d8d2014-11-28 11:26:07 +01002009 socket->s->si[1].flags = SI_FL_ISBACK;
Willy Tarreau819d3322014-11-28 12:12:34 +01002010 si_reset(&socket->s->si[1]);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01002011 si_set_state(&socket->s->si[1], SI_ST_INI);
2012 socket->s->si[1].conn_retries = socket_proxy.conn_retries;
2013
2014 /* Force destination server. */
2015 socket->s->flags |= SN_DIRECT | SN_ASSIGNED | SN_ADDR_SET | SN_BE_ASSIGNED;
2016 socket->s->target = &socket_tcp.obj_type;
2017
2018 /* This session is added to te lists of alive sessions. */
2019 LIST_ADDQ(&sessions, &socket->s->list);
2020
2021 /* XXX: I think that this list is used by stats. */
2022 LIST_INIT(&socket->s->back_refs);
2023
2024 /* Update statistics counters. */
2025 socket_proxy.feconn++; /* beconn will be increased later */
2026 jobs++;
2027 totalconn++;
2028
2029 /* Return yield waiting for connection. */
2030 return 1;
2031
2032out_fail_conn1:
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01002033 pool_free2(pool2_buffer, socket->s->res.buf);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01002034out_fail_rep_buf:
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01002035 pool_free2(pool2_buffer, socket->s->req.buf);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01002036out_fail_req_buf:
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01002037 task_free(socket->s->task);
2038out_free_session:
2039 pool_free2(pool2_session, socket->s);
2040out_fail_conf:
2041 WILL_LJMP(lua_error(L));
2042 return 0;
2043}
Thierry FOURNIER65f34c62015-02-16 20:11:43 +01002044
2045/*
2046 *
2047 *
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002048 * Class Channel
2049 *
2050 *
2051 */
2052
2053/* Returns the struct hlua_channel join to the class channel in the
2054 * stack entry "ud" or throws an argument error.
2055 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002056__LJMP static struct channel *hlua_checkchannel(lua_State *L, int ud)
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002057{
Willy Tarreau47860ed2015-03-10 14:07:50 +01002058 return (struct channel *)MAY_LJMP(hlua_checkudata(L, ud, class_channel_ref));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002059}
2060
Willy Tarreau47860ed2015-03-10 14:07:50 +01002061/* Pushes the channel onto the top of the stack. If the stask does not have a
2062 * free slots, the function fails and returns 0;
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002063 */
Willy Tarreau2a71af42015-03-10 13:51:50 +01002064static int hlua_channel_new(lua_State *L, struct channel *channel)
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002065{
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002066 /* Check stack size. */
Thierry FOURNIER2297bc22015-03-11 17:43:33 +01002067 if (!lua_checkstack(L, 3))
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002068 return 0;
2069
Thierry FOURNIER2297bc22015-03-11 17:43:33 +01002070 lua_newtable(L);
Willy Tarreau47860ed2015-03-10 14:07:50 +01002071 lua_pushlightuserdata(L, channel);
Thierry FOURNIER2297bc22015-03-11 17:43:33 +01002072 lua_rawseti(L, -2, 0);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002073
2074 /* Pop a class sesison metatable and affect it to the userdata. */
2075 lua_rawgeti(L, LUA_REGISTRYINDEX, class_channel_ref);
2076 lua_setmetatable(L, -2);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002077 return 1;
2078}
2079
2080/* Duplicate all the data present in the input channel and put it
2081 * in a string LUA variables. Returns -1 and push a nil value in
2082 * the stack if the channel is closed and all the data are consumed,
2083 * returns 0 if no data are available, otherwise it returns the length
2084 * of the builded string.
2085 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002086static inline int _hlua_channel_dup(struct channel *chn, lua_State *L)
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002087{
2088 char *blk1;
2089 char *blk2;
2090 int len1;
2091 int len2;
2092 int ret;
2093 luaL_Buffer b;
2094
Willy Tarreau47860ed2015-03-10 14:07:50 +01002095 ret = bi_getblk_nc(chn, &blk1, &len1, &blk2, &len2);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002096 if (unlikely(ret == 0))
2097 return 0;
2098
2099 if (unlikely(ret < 0)) {
2100 lua_pushnil(L);
2101 return -1;
2102 }
2103
2104 luaL_buffinit(L, &b);
2105 luaL_addlstring(&b, blk1, len1);
2106 if (unlikely(ret == 2))
2107 luaL_addlstring(&b, blk2, len2);
2108 luaL_pushresult(&b);
2109
2110 if (unlikely(ret == 2))
2111 return len1 + len2;
2112 return len1;
2113}
2114
2115/* "_hlua_channel_dup" wrapper. If no data are available, it returns
2116 * a yield. This function keep the data in the buffer.
2117 */
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002118__LJMP static int hlua_channel_dup_yield(lua_State *L, int status, lua_KContext ctx)
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002119{
Willy Tarreau47860ed2015-03-10 14:07:50 +01002120 struct channel *chn;
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002121
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002122 chn = MAY_LJMP(hlua_checkchannel(L, 1));
2123
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002124 if (_hlua_channel_dup(chn, L) == 0)
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002125 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_channel_dup_yield, TICK_ETERNITY, 0));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002126 return 1;
2127}
2128
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002129/* Check arguments for the function "hlua_channel_dup_yield". */
2130__LJMP static int hlua_channel_dup(lua_State *L)
2131{
2132 MAY_LJMP(check_args(L, 1, "dup"));
2133 MAY_LJMP(hlua_checkchannel(L, 1));
2134 return MAY_LJMP(hlua_channel_dup_yield(L, 0, 0));
2135}
2136
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002137/* "_hlua_channel_dup" wrapper. If no data are available, it returns
2138 * a yield. This function consumes the data in the buffer. It returns
2139 * a string containing the data or a nil pointer if no data are available
2140 * and the channel is closed.
2141 */
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002142__LJMP static int hlua_channel_get_yield(lua_State *L, int status, lua_KContext ctx)
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002143{
Willy Tarreau47860ed2015-03-10 14:07:50 +01002144 struct channel *chn;
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002145 int ret;
2146
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002147 chn = MAY_LJMP(hlua_checkchannel(L, 1));
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002148
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002149 ret = _hlua_channel_dup(chn, L);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002150 if (unlikely(ret == 0))
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002151 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_channel_get_yield, TICK_ETERNITY, 0));
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002152
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002153 if (unlikely(ret == -1))
2154 return 1;
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002155
Willy Tarreau47860ed2015-03-10 14:07:50 +01002156 chn->buf->i -= ret;
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002157 return 1;
2158}
2159
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002160/* Check arguments for the fucntion "hlua_channel_get_yield". */
2161__LJMP static int hlua_channel_get(lua_State *L)
2162{
2163 MAY_LJMP(check_args(L, 1, "get"));
2164 MAY_LJMP(hlua_checkchannel(L, 1));
2165 return MAY_LJMP(hlua_channel_get_yield(L, 0, 0));
2166}
2167
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002168/* This functions consumes and returns one line. If the channel is closed,
2169 * and the last data does not contains a final '\n', the data are returned
2170 * without the final '\n'. When no more data are avalaible, it returns nil
2171 * value.
2172 */
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002173__LJMP static int hlua_channel_getline_yield(lua_State *L, int status, lua_KContext ctx)
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002174{
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002175 char *blk1;
2176 char *blk2;
2177 int len1;
2178 int len2;
2179 int len;
Willy Tarreau47860ed2015-03-10 14:07:50 +01002180 struct channel *chn;
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002181 int ret;
2182 luaL_Buffer b;
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002183
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002184 chn = MAY_LJMP(hlua_checkchannel(L, 1));
2185
Willy Tarreau47860ed2015-03-10 14:07:50 +01002186 ret = bi_getline_nc(chn, &blk1, &len1, &blk2, &len2);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002187 if (ret == 0)
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002188 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_channel_getline_yield, TICK_ETERNITY, 0));
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002189
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002190 if (ret == -1) {
2191 lua_pushnil(L);
2192 return 1;
2193 }
2194
2195 luaL_buffinit(L, &b);
2196 luaL_addlstring(&b, blk1, len1);
2197 len = len1;
2198 if (unlikely(ret == 2)) {
2199 luaL_addlstring(&b, blk2, len2);
2200 len += len2;
2201 }
2202 luaL_pushresult(&b);
Willy Tarreau47860ed2015-03-10 14:07:50 +01002203 buffer_replace2(chn->buf, chn->buf->p, chn->buf->p + len, NULL, 0);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002204 return 1;
2205}
2206
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002207/* Check arguments for the fucntion "hlua_channel_getline_yield". */
2208__LJMP static int hlua_channel_getline(lua_State *L)
2209{
2210 MAY_LJMP(check_args(L, 1, "getline"));
2211 MAY_LJMP(hlua_checkchannel(L, 1));
2212 return MAY_LJMP(hlua_channel_getline_yield(L, 0, 0));
2213}
2214
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002215/* This function takes a string as input, and append it at the
2216 * input side of channel. If the data is too big, but a space
2217 * is probably available after sending some data, the function
2218 * yield. If the data is bigger than the buffer, or if the
2219 * channel is closed, it returns -1. otherwise, it returns the
2220 * amount of data writed.
2221 */
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002222__LJMP static int hlua_channel_append_yield(lua_State *L, int status, lua_KContext ctx)
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002223{
Willy Tarreau47860ed2015-03-10 14:07:50 +01002224 struct channel *chn = MAY_LJMP(hlua_checkchannel(L, 1));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002225 size_t len;
2226 const char *str = MAY_LJMP(luaL_checklstring(L, 2, &len));
2227 int l = MAY_LJMP(luaL_checkinteger(L, 3));
2228 int ret;
2229 int max;
2230
Willy Tarreau47860ed2015-03-10 14:07:50 +01002231 max = channel_recv_limit(chn) - buffer_len(chn->buf);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002232 if (max > len - l)
2233 max = len - l;
2234
Willy Tarreau47860ed2015-03-10 14:07:50 +01002235 ret = bi_putblk(chn, str + l, max);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002236 if (ret == -2 || ret == -3) {
2237 lua_pushinteger(L, -1);
2238 return 1;
2239 }
2240 if (ret == -1)
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002241 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_channel_append_yield, TICK_ETERNITY, 0));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002242 l += ret;
2243 lua_pop(L, 1);
2244 lua_pushinteger(L, l);
2245
Willy Tarreau47860ed2015-03-10 14:07:50 +01002246 max = channel_recv_limit(chn) - buffer_len(chn->buf);
2247 if (max == 0 && chn->buf->o == 0) {
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002248 /* There are no space avalaible, and the output buffer is empty.
2249 * in this case, we cannot add more data, so we cannot yield,
2250 * we return the amount of copyied data.
2251 */
2252 return 1;
2253 }
2254 if (l < len)
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002255 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_channel_append_yield, TICK_ETERNITY, 0));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002256 return 1;
2257}
2258
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002259/* just a wrapper of "hlua_channel_append_yield". It returns the length
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002260 * of the writed string, or -1 if the channel is closed or if the
2261 * buffer size is too little for the data.
2262 */
2263__LJMP static int hlua_channel_append(lua_State *L)
2264{
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002265 size_t len;
2266
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002267 MAY_LJMP(check_args(L, 2, "append"));
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002268 MAY_LJMP(hlua_checkchannel(L, 1));
2269 MAY_LJMP(luaL_checklstring(L, 2, &len));
2270 MAY_LJMP(luaL_checkinteger(L, 3));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002271 lua_pushinteger(L, 0);
2272
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002273 return MAY_LJMP(hlua_channel_append_yield(L, 0, 0));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002274}
2275
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002276/* just a wrapper of "hlua_channel_append_yield". This wrapper starts
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002277 * his process by cleaning the buffer. The result is a replacement
2278 * of the current data. It returns the length of the writed string,
2279 * or -1 if the channel is closed or if the buffer size is too
2280 * little for the data.
2281 */
2282__LJMP static int hlua_channel_set(lua_State *L)
2283{
Willy Tarreau47860ed2015-03-10 14:07:50 +01002284 struct channel *chn;
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002285
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002286 MAY_LJMP(check_args(L, 2, "set"));
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002287 chn = MAY_LJMP(hlua_checkchannel(L, 1));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002288 lua_pushinteger(L, 0);
2289
Willy Tarreau47860ed2015-03-10 14:07:50 +01002290 chn->buf->i = 0;
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002291
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002292 return MAY_LJMP(hlua_channel_append_yield(L, 0, 0));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002293}
2294
2295/* Append data in the output side of the buffer. This data is immediatly
2296 * sent. The fcuntion returns the ammount of data writed. If the buffer
2297 * cannot contains the data, the function yield. The function returns -1
2298 * if the channel is closed.
2299 */
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002300__LJMP static int hlua_channel_send_yield(lua_State *L, int status, lua_KContext ctx)
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002301{
Willy Tarreau47860ed2015-03-10 14:07:50 +01002302 struct channel *chn = MAY_LJMP(hlua_checkchannel(L, 1));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002303 size_t len;
2304 const char *str = MAY_LJMP(luaL_checklstring(L, 2, &len));
2305 int l = MAY_LJMP(luaL_checkinteger(L, 3));
2306 int max;
Thierry FOURNIERef6a2112015-03-05 17:45:34 +01002307 struct hlua *hlua = hlua_gethlua(L);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002308
Willy Tarreau47860ed2015-03-10 14:07:50 +01002309 if (unlikely(channel_output_closed(chn))) {
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002310 lua_pushinteger(L, -1);
2311 return 1;
2312 }
2313
Thierry FOURNIER3e3a6082015-03-05 17:06:12 +01002314 /* Check if the buffer is avalaible because HAProxy doesn't allocate
2315 * the request buffer if its not required.
2316 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002317 if (chn->buf->size == 0) {
2318 if (!session_alloc_recv_buffer(chn)) {
2319 chn_prod(chn)->flags |= SI_FL_WAIT_ROOM;
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002320 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_channel_send_yield, TICK_ETERNITY, 0));
Thierry FOURNIER3e3a6082015-03-05 17:06:12 +01002321 }
2322 }
2323
Thierry FOURNIERdeb5d732015-03-06 01:07:45 +01002324 /* the writed data will be immediatly sent, so we can check
2325 * the avalaible space without taking in account the reserve.
2326 * The reserve is guaranted for the processing of incoming
2327 * data, because the buffer will be flushed.
2328 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002329 max = chn->buf->size - buffer_len(chn->buf);
Thierry FOURNIERdeb5d732015-03-06 01:07:45 +01002330
2331 /* If there are no space avalaible, and the output buffer is empty.
2332 * in this case, we cannot add more data, so we cannot yield,
2333 * we return the amount of copyied data.
2334 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002335 if (max == 0 && chn->buf->o == 0)
Thierry FOURNIERdeb5d732015-03-06 01:07:45 +01002336 return 1;
2337
2338 /* Adjust the real required length. */
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002339 if (max > len - l)
2340 max = len - l;
2341
Thierry FOURNIERdeb5d732015-03-06 01:07:45 +01002342 /* The buffer avalaible size may be not contiguous. This test
2343 * detects a non contiguous buffer and realign it.
2344 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002345 if (bi_space_for_replace(chn->buf) < max)
2346 buffer_slow_realign(chn->buf);
Thierry FOURNIERdeb5d732015-03-06 01:07:45 +01002347
2348 /* Copy input data in the buffer. */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002349 max = buffer_replace2(chn->buf, chn->buf->p, chn->buf->p, str + l, max);
Thierry FOURNIERdeb5d732015-03-06 01:07:45 +01002350
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002351 /* buffer replace considers that the input part is filled.
2352 * so, I must forward these new data in the output part.
2353 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002354 b_adv(chn->buf, max);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002355
2356 l += max;
2357 lua_pop(L, 1);
2358 lua_pushinteger(L, l);
2359
Thierry FOURNIERdeb5d732015-03-06 01:07:45 +01002360 /* If there are no space avalaible, and the output buffer is empty.
2361 * in this case, we cannot add more data, so we cannot yield,
2362 * we return the amount of copyied data.
2363 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002364 max = chn->buf->size - buffer_len(chn->buf);
2365 if (max == 0 && chn->buf->o == 0)
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002366 return 1;
Thierry FOURNIERdeb5d732015-03-06 01:07:45 +01002367
Thierry FOURNIERef6a2112015-03-05 17:45:34 +01002368 if (l < len) {
2369 /* If we are waiting for space in the response buffer, we
2370 * must set the flag WAKERESWR. This flag required the task
2371 * wake up if any activity is detected on the response buffer.
2372 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002373 if (chn->flags & CF_ISRESP)
Thierry FOURNIERef6a2112015-03-05 17:45:34 +01002374 HLUA_SET_WAKERESWR(hlua);
Thierry FOURNIER53e08ec2015-03-06 00:35:53 +01002375 else
2376 HLUA_SET_WAKEREQWR(hlua);
2377 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_channel_send_yield, TICK_ETERNITY, 0));
Thierry FOURNIERef6a2112015-03-05 17:45:34 +01002378 }
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002379
2380 return 1;
2381}
2382
2383/* Just a wraper of "_hlua_channel_send". This wrapper permits
2384 * yield the LUA process, and resume it without checking the
2385 * input arguments.
2386 */
2387__LJMP static int hlua_channel_send(lua_State *L)
2388{
2389 MAY_LJMP(check_args(L, 2, "send"));
2390 lua_pushinteger(L, 0);
2391
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002392 return MAY_LJMP(hlua_channel_send_yield(L, 0, 0));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002393}
2394
2395/* This function forward and amount of butes. The data pass from
2396 * the input side of the buffer to the output side, and can be
2397 * forwarded. This function never fails.
2398 *
2399 * The Lua function takes an amount of bytes to be forwarded in
2400 * imput. It returns the number of bytes forwarded.
2401 */
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002402__LJMP static int hlua_channel_forward_yield(lua_State *L, int status, lua_KContext ctx)
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002403{
Willy Tarreau47860ed2015-03-10 14:07:50 +01002404 struct channel *chn;
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002405 int len;
2406 int l;
2407 int max;
Thierry FOURNIERef6a2112015-03-05 17:45:34 +01002408 struct hlua *hlua = hlua_gethlua(L);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002409
2410 chn = MAY_LJMP(hlua_checkchannel(L, 1));
2411 len = MAY_LJMP(luaL_checkinteger(L, 2));
2412 l = MAY_LJMP(luaL_checkinteger(L, -1));
2413
2414 max = len - l;
Willy Tarreau47860ed2015-03-10 14:07:50 +01002415 if (max > chn->buf->i)
2416 max = chn->buf->i;
2417 channel_forward(chn, max);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002418 l += max;
2419
2420 lua_pop(L, 1);
2421 lua_pushinteger(L, l);
2422
2423 /* Check if it miss bytes to forward. */
2424 if (l < len) {
2425 /* The the input channel or the output channel are closed, we
2426 * must return the amount of data forwarded.
2427 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002428 if (channel_input_closed(chn) || channel_output_closed(chn))
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002429 return 1;
2430
Thierry FOURNIERef6a2112015-03-05 17:45:34 +01002431 /* If we are waiting for space data in the response buffer, we
2432 * must set the flag WAKERESWR. This flag required the task
2433 * wake up if any activity is detected on the response buffer.
2434 */
Willy Tarreau47860ed2015-03-10 14:07:50 +01002435 if (chn->flags & CF_ISRESP)
Thierry FOURNIERef6a2112015-03-05 17:45:34 +01002436 HLUA_SET_WAKERESWR(hlua);
Thierry FOURNIER53e08ec2015-03-06 00:35:53 +01002437 else
2438 HLUA_SET_WAKEREQWR(hlua);
Thierry FOURNIERef6a2112015-03-05 17:45:34 +01002439
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002440 /* Otherwise, we can yield waiting for new data in the inpout side. */
Thierry FOURNIER4abd3ae2015-03-03 17:29:06 +01002441 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_channel_forward_yield, TICK_ETERNITY, 0));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002442 }
2443
2444 return 1;
2445}
2446
2447/* Just check the input and prepare the stack for the previous
2448 * function "hlua_channel_forward_yield"
2449 */
2450__LJMP static int hlua_channel_forward(lua_State *L)
2451{
2452 MAY_LJMP(check_args(L, 2, "forward"));
2453 MAY_LJMP(hlua_checkchannel(L, 1));
2454 MAY_LJMP(luaL_checkinteger(L, 2));
2455
2456 lua_pushinteger(L, 0);
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002457 return MAY_LJMP(hlua_channel_forward_yield(L, 0, 0));
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002458}
2459
2460/* Just returns the number of bytes available in the input
2461 * side of the buffer. This function never fails.
2462 */
2463__LJMP static int hlua_channel_get_in_len(lua_State *L)
2464{
Willy Tarreau47860ed2015-03-10 14:07:50 +01002465 struct channel *chn;
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002466
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002467 MAY_LJMP(check_args(L, 1, "get_in_len"));
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002468 chn = MAY_LJMP(hlua_checkchannel(L, 1));
Willy Tarreau47860ed2015-03-10 14:07:50 +01002469 lua_pushinteger(L, chn->buf->i);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002470 return 1;
2471}
2472
2473/* Just returns the number of bytes available in the output
2474 * side of the buffer. This function never fails.
2475 */
2476__LJMP static int hlua_channel_get_out_len(lua_State *L)
2477{
Willy Tarreau47860ed2015-03-10 14:07:50 +01002478 struct channel *chn;
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002479
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002480 MAY_LJMP(check_args(L, 1, "get_out_len"));
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002481 chn = MAY_LJMP(hlua_checkchannel(L, 1));
Willy Tarreau47860ed2015-03-10 14:07:50 +01002482 lua_pushinteger(L, chn->buf->o);
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002483 return 1;
2484}
2485
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002486/*
2487 *
2488 *
2489 * Class Fetches
2490 *
2491 *
2492 */
2493
2494/* Returns a struct hlua_session if the stack entry "ud" is
2495 * a class session, otherwise it throws an error.
2496 */
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002497__LJMP static struct hlua_smp *hlua_checkfetches(lua_State *L, int ud)
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002498{
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002499 return (struct hlua_smp *)MAY_LJMP(hlua_checkudata(L, ud, class_fetches_ref));
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002500}
2501
2502/* This function creates and push in the stack a fetch object according
2503 * with a current TXN.
2504 */
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002505static int hlua_fetches_new(lua_State *L, struct hlua_txn *txn, int stringsafe)
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002506{
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002507 struct hlua_smp *hs;
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002508
2509 /* Check stack size. */
2510 if (!lua_checkstack(L, 3))
2511 return 0;
2512
2513 /* Create the object: obj[0] = userdata.
2514 * Note that the base of the Fetches object is the
2515 * transaction object.
2516 */
2517 lua_newtable(L);
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002518 hs = lua_newuserdata(L, sizeof(struct hlua_smp));
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002519 lua_rawseti(L, -2, 0);
2520
2521 hs->s = txn->s;
2522 hs->p = txn->p;
2523 hs->l7 = txn->l7;
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002524 hs->stringsafe = stringsafe;
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002525
2526 /* Pop a class sesison metatable and affect it to the userdata. */
2527 lua_rawgeti(L, LUA_REGISTRYINDEX, class_fetches_ref);
2528 lua_setmetatable(L, -2);
2529
2530 return 1;
2531}
2532
2533/* This function is an LUA binding. It is called with each sample-fetch.
2534 * It uses closure argument to store the associated sample-fetch. It
2535 * returns only one argument or throws an error. An error is thrown
2536 * only if an error is encountered during the argument parsing. If
2537 * the "sample-fetch" function fails, nil is returned.
2538 */
2539__LJMP static int hlua_run_sample_fetch(lua_State *L)
2540{
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002541 struct hlua_smp *s;
Willy Tarreau2ec22742015-03-10 14:27:20 +01002542 struct sample_fetch *f;
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002543 struct arg args[ARGM_NBARGS + 1];
2544 int i;
2545 struct sample smp;
2546
2547 /* Get closure arguments. */
Willy Tarreau2ec22742015-03-10 14:27:20 +01002548 f = (struct sample_fetch *)lua_touserdata(L, lua_upvalueindex(1));
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002549
2550 /* Get traditionnal arguments. */
2551 s = MAY_LJMP(hlua_checkfetches(L, 1));
2552
2553 /* Get extra arguments. */
2554 for (i = 0; i < lua_gettop(L) - 1; i++) {
2555 if (i >= ARGM_NBARGS)
2556 break;
2557 hlua_lua2arg(L, i + 2, &args[i]);
2558 }
2559 args[i].type = ARGT_STOP;
2560
2561 /* Check arguments. */
Willy Tarreau2ec22742015-03-10 14:27:20 +01002562 MAY_LJMP(hlua_lua2arg_check(L, 1, args, f->arg_mask));
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002563
2564 /* Run the special args checker. */
Willy Tarreau2ec22742015-03-10 14:27:20 +01002565 if (f->val_args && !f->val_args(args, NULL)) {
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002566 lua_pushfstring(L, "error in arguments");
2567 WILL_LJMP(lua_error(L));
2568 }
2569
2570 /* Initialise the sample. */
2571 memset(&smp, 0, sizeof(smp));
2572
2573 /* Run the sample fetch process. */
Willy Tarreau2ec22742015-03-10 14:27:20 +01002574 if (!f->process(s->p, s->s, s->l7, 0, args, &smp, f->kw, f->private)) {
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002575 if (s->stringsafe)
2576 lua_pushstring(L, "");
2577 else
2578 lua_pushnil(L);
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002579 return 1;
2580 }
2581
2582 /* Convert the returned sample in lua value. */
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002583 if (s->stringsafe)
2584 hlua_smp2lua_str(L, &smp);
2585 else
2586 hlua_smp2lua(L, &smp);
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002587 return 1;
2588}
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01002589
2590/*
2591 *
2592 *
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002593 * Class Converters
2594 *
2595 *
2596 */
2597
2598/* Returns a struct hlua_session if the stack entry "ud" is
2599 * a class session, otherwise it throws an error.
2600 */
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002601__LJMP static struct hlua_smp *hlua_checkconverters(lua_State *L, int ud)
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002602{
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002603 return (struct hlua_smp *)MAY_LJMP(hlua_checkudata(L, ud, class_converters_ref));
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002604}
2605
2606/* This function creates and push in the stack a Converters object
2607 * according with a current TXN.
2608 */
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002609static int hlua_converters_new(lua_State *L, struct hlua_txn *txn, int stringsafe)
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002610{
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002611 struct hlua_smp *hs;
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002612
2613 /* Check stack size. */
2614 if (!lua_checkstack(L, 3))
2615 return 0;
2616
2617 /* Create the object: obj[0] = userdata.
2618 * Note that the base of the Converters object is the
2619 * same than the TXN object.
2620 */
2621 lua_newtable(L);
2622 hs = lua_newuserdata(L, sizeof(struct hlua_txn));
2623 lua_rawseti(L, -2, 0);
2624
2625 hs->s = txn->s;
2626 hs->p = txn->p;
2627 hs->l7 = txn->l7;
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002628 hs->stringsafe = stringsafe;
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002629
2630 /* Pop a class session metatable and affect it to the table. */
2631 lua_rawgeti(L, LUA_REGISTRYINDEX, class_converters_ref);
2632 lua_setmetatable(L, -2);
2633
2634 return 1;
2635}
2636
2637/* This function is an LUA binding. It is called with each converter.
2638 * It uses closure argument to store the associated converter. It
2639 * returns only one argument or throws an error. An error is thrown
2640 * only if an error is encountered during the argument parsing. If
2641 * the converter function function fails, nil is returned.
2642 */
2643__LJMP static int hlua_run_sample_conv(lua_State *L)
2644{
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002645 struct hlua_smp *sc;
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002646 struct sample_conv *conv;
2647 struct arg args[ARGM_NBARGS + 1];
2648 int i;
2649 struct sample smp;
2650
2651 /* Get closure arguments. */
2652 conv = (struct sample_conv *)lua_touserdata(L, lua_upvalueindex(1));
2653
2654 /* Get traditionnal arguments. */
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002655 sc = MAY_LJMP(hlua_checkconverters(L, 1));
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002656
2657 /* Get extra arguments. */
2658 for (i = 0; i < lua_gettop(L) - 2; i++) {
2659 if (i >= ARGM_NBARGS)
2660 break;
2661 hlua_lua2arg(L, i + 3, &args[i]);
2662 }
2663 args[i].type = ARGT_STOP;
2664
2665 /* Check arguments. */
2666 MAY_LJMP(hlua_lua2arg_check(L, 1, args, conv->arg_mask));
2667
2668 /* Run the special args checker. */
2669 if (conv->val_args && !conv->val_args(args, conv, "", 0, NULL)) {
2670 hlua_pusherror(L, "error in arguments");
2671 WILL_LJMP(lua_error(L));
2672 }
2673
2674 /* Initialise the sample. */
2675 if (!hlua_lua2smp(L, 2, &smp)) {
2676 hlua_pusherror(L, "error in the input argument");
2677 WILL_LJMP(lua_error(L));
2678 }
2679
2680 /* Apply expected cast. */
2681 if (!sample_casts[smp.type][conv->in_type]) {
2682 hlua_pusherror(L, "invalid input argument: cannot cast '%s' to '%s'",
2683 smp_to_type[smp.type], smp_to_type[conv->in_type]);
2684 WILL_LJMP(lua_error(L));
2685 }
2686 if (sample_casts[smp.type][conv->in_type] != c_none &&
2687 !sample_casts[smp.type][conv->in_type](&smp)) {
2688 hlua_pusherror(L, "error during the input argument casting");
2689 WILL_LJMP(lua_error(L));
2690 }
2691
2692 /* Run the sample conversion process. */
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002693 if (!conv->process(sc->s, args, &smp, conv->private)) {
2694 if (sc->stringsafe)
2695 lua_pushstring(L, "");
2696 else
2697 lua_pushnil(L);
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002698 return 1;
2699 }
2700
2701 /* Convert the returned sample in lua value. */
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002702 if (sc->stringsafe)
2703 hlua_smp2lua_str(L, &smp);
2704 else
2705 hlua_smp2lua(L, &smp);
2706 return 1;
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002707}
2708
2709/*
2710 *
2711 *
Thierry FOURNIER65f34c62015-02-16 20:11:43 +01002712 * Class TXN
2713 *
2714 *
2715 */
2716
2717/* Returns a struct hlua_session if the stack entry "ud" is
2718 * a class session, otherwise it throws an error.
2719 */
2720__LJMP static struct hlua_txn *hlua_checktxn(lua_State *L, int ud)
2721{
2722 return (struct hlua_txn *)MAY_LJMP(hlua_checkudata(L, ud, class_txn_ref));
2723}
2724
Willy Tarreau59551662015-03-10 14:23:13 +01002725__LJMP static int hlua_set_priv(lua_State *L)
Thierry FOURNIER05c0b8a2015-02-25 11:43:21 +01002726{
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002727 struct hlua *hlua;
2728
Thierry FOURNIER05c0b8a2015-02-25 11:43:21 +01002729 MAY_LJMP(check_args(L, 2, "set_priv"));
2730
2731 /* It is useles to retrieve the session, but this function
2732 * runs only in a session context.
2733 */
2734 MAY_LJMP(hlua_checktxn(L, 1));
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002735 hlua = hlua_gethlua(L);
Thierry FOURNIER05c0b8a2015-02-25 11:43:21 +01002736
2737 /* Remove previous value. */
2738 if (hlua->Mref != -1)
2739 luaL_unref(L, hlua->Mref, LUA_REGISTRYINDEX);
2740
2741 /* Get and store new value. */
2742 lua_pushvalue(L, 2); /* Copy the element 2 at the top of the stack. */
2743 hlua->Mref = luaL_ref(L, LUA_REGISTRYINDEX); /* pop the previously pushed value. */
2744
2745 return 0;
2746}
2747
Willy Tarreau59551662015-03-10 14:23:13 +01002748__LJMP static int hlua_get_priv(lua_State *L)
Thierry FOURNIER05c0b8a2015-02-25 11:43:21 +01002749{
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002750 struct hlua *hlua;
2751
Thierry FOURNIER05c0b8a2015-02-25 11:43:21 +01002752 MAY_LJMP(check_args(L, 1, "get_priv"));
2753
2754 /* It is useles to retrieve the session, but this function
2755 * runs only in a session context.
2756 */
2757 MAY_LJMP(hlua_checktxn(L, 1));
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002758 hlua = hlua_gethlua(L);
Thierry FOURNIER05c0b8a2015-02-25 11:43:21 +01002759
2760 /* Push configuration index in the stack. */
2761 lua_rawgeti(L, LUA_REGISTRYINDEX, hlua->Mref);
2762
2763 return 1;
2764}
2765
Thierry FOURNIER65f34c62015-02-16 20:11:43 +01002766/* Create stack entry containing a class TXN. This function
2767 * return 0 if the stack does not contains free slots,
2768 * otherwise it returns 1.
2769 */
2770static int hlua_txn_new(lua_State *L, struct session *s, struct proxy *p, void *l7)
2771{
2772 struct hlua_txn *hs;
2773
2774 /* Check stack size. */
Thierry FOURNIER2297bc22015-03-11 17:43:33 +01002775 if (!lua_checkstack(L, 3))
Thierry FOURNIER65f34c62015-02-16 20:11:43 +01002776 return 0;
2777
2778 /* NOTE: The allocation never fails. The failure
2779 * throw an error, and the function never returns.
2780 * if the throw is not avalaible, the process is aborted.
2781 */
Thierry FOURNIER2297bc22015-03-11 17:43:33 +01002782 /* Create the object: obj[0] = userdata. */
2783 lua_newtable(L);
Thierry FOURNIER65f34c62015-02-16 20:11:43 +01002784 hs = lua_newuserdata(L, sizeof(struct hlua_txn));
Thierry FOURNIER2297bc22015-03-11 17:43:33 +01002785 lua_rawseti(L, -2, 0);
2786
Thierry FOURNIER65f34c62015-02-16 20:11:43 +01002787 hs->s = s;
2788 hs->p = p;
2789 hs->l7 = l7;
2790
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002791 /* Create the "f" field that contains a list of fetches. */
2792 lua_pushstring(L, "f");
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002793 if (!hlua_fetches_new(L, hs, 0))
2794 return 0;
2795 lua_settable(L, -3);
2796
2797 /* Create the "sf" field that contains a list of stringsafe fetches. */
2798 lua_pushstring(L, "sf");
2799 if (!hlua_fetches_new(L, hs, 1))
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01002800 return 0;
2801 lua_settable(L, -3);
2802
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002803 /* Create the "c" field that contains a list of converters. */
2804 lua_pushstring(L, "c");
Thierry FOURNIER2694a1a2015-03-11 20:13:36 +01002805 if (!hlua_converters_new(L, hs, 0))
2806 return 0;
2807 lua_settable(L, -3);
2808
2809 /* Create the "sc" field that contains a list of stringsafe converters. */
2810 lua_pushstring(L, "sc");
2811 if (!hlua_converters_new(L, hs, 1))
Thierry FOURNIER594afe72015-03-10 23:58:30 +01002812 return 0;
2813 lua_settable(L, -3);
2814
Thierry FOURNIER397826a2015-03-11 19:39:09 +01002815 /* Create the "req" field that contains the request channel object. */
2816 lua_pushstring(L, "req");
Willy Tarreau2a71af42015-03-10 13:51:50 +01002817 if (!hlua_channel_new(L, &s->req))
Thierry FOURNIER397826a2015-03-11 19:39:09 +01002818 return 0;
2819 lua_settable(L, -3);
2820
2821 /* Create the "res" field that contains the response channel object. */
2822 lua_pushstring(L, "res");
Willy Tarreau2a71af42015-03-10 13:51:50 +01002823 if (!hlua_channel_new(L, &s->res))
Thierry FOURNIER397826a2015-03-11 19:39:09 +01002824 return 0;
2825 lua_settable(L, -3);
2826
Thierry FOURNIER65f34c62015-02-16 20:11:43 +01002827 /* Pop a class sesison metatable and affect it to the userdata. */
2828 lua_rawgeti(L, LUA_REGISTRYINDEX, class_txn_ref);
2829 lua_setmetatable(L, -2);
2830
2831 return 1;
2832}
2833
Thierry FOURNIER893bfa32015-02-17 18:42:34 +01002834/* This function is an Lua binding that send pending data
2835 * to the client, and close the stream interface.
2836 */
2837__LJMP static int hlua_txn_close(lua_State *L)
2838{
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002839 struct hlua_txn *s;
Willy Tarreau81389672015-03-10 12:03:52 +01002840 struct channel *ic, *oc;
Thierry FOURNIER893bfa32015-02-17 18:42:34 +01002841
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002842 MAY_LJMP(check_args(L, 1, "close"));
2843 s = MAY_LJMP(hlua_checktxn(L, 1));
Thierry FOURNIER893bfa32015-02-17 18:42:34 +01002844
Willy Tarreau81389672015-03-10 12:03:52 +01002845 ic = si_ic(&s->s->si[0]);
2846 oc = si_oc(&s->s->si[0]);
2847
2848 channel_abort(ic);
2849 channel_auto_close(ic);
2850 channel_erase(ic);
2851 channel_auto_read(oc);
2852 channel_auto_close(oc);
2853 channel_shutr_now(oc);
Thierry FOURNIER893bfa32015-02-17 18:42:34 +01002854
2855 return 0;
2856}
2857
Thierry FOURNIER9a819e72015-02-16 20:22:55 +01002858/* This function is an LUA binding. It creates ans returns
2859 * an array of HTTP headers. This function does not fails.
2860 */
Willy Tarreau59551662015-03-10 14:23:13 +01002861static int hlua_session_get_headers(lua_State *L)
Thierry FOURNIER9a819e72015-02-16 20:22:55 +01002862{
2863 struct hlua_txn *s = MAY_LJMP(hlua_checktxn(L, 1));
2864 struct session *sess = s->s;
2865 const char *cur_ptr, *cur_next, *p;
2866 int old_idx, cur_idx;
2867 struct hdr_idx_elem *cur_hdr;
2868 const char *hn, *hv;
2869 int hnl, hvl;
2870
2871 /* Create the table. */
2872 lua_newtable(L);
2873
2874 /* Build array of headers. */
2875 old_idx = 0;
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01002876 cur_next = sess->req.buf->p + hdr_idx_first_pos(&sess->txn.hdr_idx);
Thierry FOURNIER9a819e72015-02-16 20:22:55 +01002877
2878 while (1) {
2879 cur_idx = sess->txn.hdr_idx.v[old_idx].next;
2880 if (!cur_idx)
2881 break;
2882 old_idx = cur_idx;
2883
2884 cur_hdr = &sess->txn.hdr_idx.v[cur_idx];
2885 cur_ptr = cur_next;
2886 cur_next = cur_ptr + cur_hdr->len + cur_hdr->cr + 1;
2887
2888 /* Now we have one full header at cur_ptr of len cur_hdr->len,
2889 * and the next header starts at cur_next. We'll check
2890 * this header in the list as well as against the default
2891 * rule.
2892 */
2893
2894 /* look for ': *'. */
2895 hn = cur_ptr;
2896 for (p = cur_ptr; p < cur_ptr + cur_hdr->len && *p != ':'; p++);
2897 if (p >= cur_ptr+cur_hdr->len)
2898 continue;
2899 hnl = p - hn;
2900 p++;
2901 while (p < cur_ptr+cur_hdr->len && ( *p == ' ' || *p == '\t' ))
2902 p++;
2903 if (p >= cur_ptr+cur_hdr->len)
2904 continue;
2905 hv = p;
2906 hvl = cur_ptr+cur_hdr->len-p;
2907
2908 /* Push values in the table. */
2909 lua_pushlstring(L, hn, hnl);
2910 lua_pushlstring(L, hv, hvl);
2911 lua_settable(L, -3);
2912 }
2913
2914 return 1;
2915}
2916
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002917__LJMP static int hlua_sleep_yield(lua_State *L, int status, lua_KContext ctx)
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002918{
2919 int wakeup_ms = lua_tointeger(L, -1);
2920 if (now_ms < wakeup_ms)
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002921 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_sleep_yield, wakeup_ms, 0));
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002922 return 0;
2923}
2924
2925__LJMP static int hlua_sleep(lua_State *L)
2926{
2927 unsigned int delay;
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002928 unsigned int wakeup_ms;
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002929
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002930 MAY_LJMP(check_args(L, 1, "sleep"));
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002931
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002932 delay = MAY_LJMP(luaL_checkinteger(L, 1)) * 1000;
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002933 wakeup_ms = tick_add(now_ms, delay);
2934 lua_pushinteger(L, wakeup_ms);
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002935
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002936 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_sleep_yield, wakeup_ms, 0));
2937 return 0;
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002938}
2939
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002940__LJMP static int hlua_msleep(lua_State *L)
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002941{
2942 unsigned int delay;
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002943 unsigned int wakeup_ms;
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002944
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002945 MAY_LJMP(check_args(L, 1, "msleep"));
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002946
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002947 delay = MAY_LJMP(luaL_checkinteger(L, 1));
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002948 wakeup_ms = tick_add(now_ms, delay);
2949 lua_pushinteger(L, wakeup_ms);
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002950
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002951 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_sleep_yield, wakeup_ms, 0));
2952 return 0;
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01002953}
2954
Thierry FOURNIER13416fe2015-02-17 15:01:59 +01002955/* This functionis an LUA binding. it permits to give back
2956 * the hand at the HAProxy scheduler. It is used when the
2957 * LUA processing consumes a lot of time.
2958 */
Thierry FOURNIERf90838b2015-03-06 13:48:32 +01002959__LJMP static int hlua_yield_yield(lua_State *L, int status, lua_KContext ctx)
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002960{
2961 return 0;
2962}
2963
Thierry FOURNIER13416fe2015-02-17 15:01:59 +01002964__LJMP static int hlua_yield(lua_State *L)
2965{
Thierry FOURNIERd44731f2015-03-04 15:51:09 +01002966 WILL_LJMP(hlua_yieldk(L, 0, 0, hlua_yield_yield, TICK_ETERNITY, HLUA_CTRLYIELD));
2967 return 0;
Thierry FOURNIER13416fe2015-02-17 15:01:59 +01002968}
2969
Thierry FOURNIER37196f42015-02-16 19:34:56 +01002970/* This function change the nice of the currently executed
2971 * task. It is used set low or high priority at the current
2972 * task.
2973 */
Willy Tarreau59551662015-03-10 14:23:13 +01002974__LJMP static int hlua_set_nice(lua_State *L)
Thierry FOURNIER37196f42015-02-16 19:34:56 +01002975{
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002976 struct hlua *hlua;
2977 int nice;
Thierry FOURNIER37196f42015-02-16 19:34:56 +01002978
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002979 MAY_LJMP(check_args(L, 1, "set_nice"));
2980 hlua = hlua_gethlua(L);
2981 nice = MAY_LJMP(luaL_checkinteger(L, 1));
Thierry FOURNIER37196f42015-02-16 19:34:56 +01002982
2983 /* If he task is not set, I'm in a start mode. */
2984 if (!hlua || !hlua->task)
2985 return 0;
2986
2987 if (nice < -1024)
2988 nice = -1024;
Willy Tarreau80f5fae2015-02-27 16:38:20 +01002989 else if (nice > 1024)
Thierry FOURNIER37196f42015-02-16 19:34:56 +01002990 nice = 1024;
2991
2992 hlua->task->nice = nice;
2993 return 0;
2994}
2995
Thierry FOURNIER24f33532015-01-23 12:13:00 +01002996/* This function is used as a calback of a task. It is called by the
2997 * HAProxy task subsystem when the task is awaked. The LUA runtime can
2998 * return an E_AGAIN signal, the emmiter of this signal must set a
2999 * signal to wake the task.
3000 */
3001static struct task *hlua_process_task(struct task *task)
3002{
3003 struct hlua *hlua = task->context;
3004 enum hlua_exec status;
3005
3006 /* We need to remove the task from the wait queue before executing
3007 * the Lua code because we don't know if it needs to wait for
3008 * another timer or not in the case of E_AGAIN.
3009 */
3010 task_delete(task);
3011
Thierry FOURNIERbd413492015-03-03 16:52:26 +01003012 /* If it is the first call to the task, we must initialize the
3013 * execution timeouts.
3014 */
3015 if (!HLUA_IS_RUNNING(hlua))
3016 hlua->expire = tick_add(now_ms, hlua_timeout_task);
3017
Thierry FOURNIER24f33532015-01-23 12:13:00 +01003018 /* Execute the Lua code. */
3019 status = hlua_ctx_resume(hlua, 1);
3020
3021 switch (status) {
3022 /* finished or yield */
3023 case HLUA_E_OK:
3024 hlua_ctx_destroy(hlua);
3025 task_delete(task);
3026 task_free(task);
3027 break;
3028
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +01003029 case HLUA_E_AGAIN: /* co process or timeout wake me later. */
3030 if (hlua->wake_time != TICK_ETERNITY)
3031 task_schedule(task, hlua->wake_time);
Thierry FOURNIER24f33532015-01-23 12:13:00 +01003032 break;
3033
3034 /* finished with error. */
3035 case HLUA_E_ERRMSG:
3036 send_log(NULL, LOG_ERR, "Lua task: %s.", lua_tostring(hlua->T, -1));
3037 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3038 Alert("Lua task: %s.\n", lua_tostring(hlua->T, -1));
3039 hlua_ctx_destroy(hlua);
3040 task_delete(task);
3041 task_free(task);
3042 break;
3043
3044 case HLUA_E_ERR:
3045 default:
3046 send_log(NULL, LOG_ERR, "Lua task: unknown error.");
3047 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3048 Alert("Lua task: unknown error.\n");
3049 hlua_ctx_destroy(hlua);
3050 task_delete(task);
3051 task_free(task);
3052 break;
3053 }
3054 return NULL;
3055}
3056
Thierry FOURNIERa4a0f3d2015-01-23 12:08:30 +01003057/* This function is an LUA binding that register LUA function to be
3058 * executed after the HAProxy configuration parsing and before the
3059 * HAProxy scheduler starts. This function expect only one LUA
3060 * argument that is a function. This function returns nothing, but
3061 * throws if an error is encountered.
3062 */
3063__LJMP static int hlua_register_init(lua_State *L)
3064{
3065 struct hlua_init_function *init;
3066 int ref;
3067
3068 MAY_LJMP(check_args(L, 1, "register_init"));
3069
3070 ref = MAY_LJMP(hlua_checkfunction(L, 1));
3071
3072 init = malloc(sizeof(*init));
3073 if (!init)
3074 WILL_LJMP(luaL_error(L, "lua out of memory error."));
3075
3076 init->function_ref = ref;
3077 LIST_ADDQ(&hlua_init_functions, &init->l);
3078 return 0;
3079}
3080
Thierry FOURNIER24f33532015-01-23 12:13:00 +01003081/* This functio is an LUA binding. It permits to register a task
3082 * executed in parallel of the main HAroxy activity. The task is
3083 * created and it is set in the HAProxy scheduler. It can be called
3084 * from the "init" section, "post init" or during the runtime.
3085 *
3086 * Lua prototype:
3087 *
3088 * <none> core.register_task(<function>)
3089 */
3090static int hlua_register_task(lua_State *L)
3091{
3092 struct hlua *hlua;
3093 struct task *task;
3094 int ref;
3095
3096 MAY_LJMP(check_args(L, 1, "register_task"));
3097
3098 ref = MAY_LJMP(hlua_checkfunction(L, 1));
3099
3100 hlua = malloc(sizeof(*hlua));
3101 if (!hlua)
3102 WILL_LJMP(luaL_error(L, "lua out of memory error."));
3103
3104 task = task_new();
3105 task->context = hlua;
3106 task->process = hlua_process_task;
3107
3108 if (!hlua_ctx_init(hlua, task))
3109 WILL_LJMP(luaL_error(L, "lua out of memory error."));
3110
3111 /* Restore the function in the stack. */
3112 lua_rawgeti(hlua->T, LUA_REGISTRYINDEX, ref);
3113 hlua->nargs = 0;
3114
3115 /* Schedule task. */
3116 task_schedule(task, now_ms);
3117
3118 return 0;
3119}
3120
Thierry FOURNIER9be813f2015-02-16 20:21:12 +01003121/* Wrapper called by HAProxy to execute an LUA converter. This wrapper
3122 * doesn't allow "yield" functions because the HAProxy engine cannot
3123 * resume converters.
3124 */
3125static int hlua_sample_conv_wrapper(struct session *session, const struct arg *arg_p,
3126 struct sample *smp, void *private)
3127{
3128 struct hlua_function *fcn = (struct hlua_function *)private;
3129
Thierry FOURNIER05ac4242015-02-27 18:37:27 +01003130 /* In the execution wrappers linked with a session, the
3131 * Lua context can be not initialized. This behavior
3132 * permits to save performances because a systematic
3133 * Lua initialization cause 5% performances loss.
3134 */
3135 if (!session->hlua.T && !hlua_ctx_init(&session->hlua, session->task)) {
3136 send_log(session->be, LOG_ERR, "Lua converter '%s': can't initialize Lua context.", fcn->name);
3137 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3138 Alert("Lua converter '%s': can't initialize Lua context.\n", fcn->name);
3139 return 0;
3140 }
3141
Thierry FOURNIER9be813f2015-02-16 20:21:12 +01003142 /* If it is the first run, initialize the data for the call. */
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +01003143 if (!HLUA_IS_RUNNING(&session->hlua)) {
Thierry FOURNIER9be813f2015-02-16 20:21:12 +01003144 /* Check stack available size. */
3145 if (!lua_checkstack(session->hlua.T, 1)) {
3146 send_log(session->be, LOG_ERR, "Lua converter '%s': full stack.", fcn->name);
3147 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3148 Alert("Lua converter '%s': full stack.\n", fcn->name);
3149 return 0;
3150 }
3151
3152 /* Restore the function in the stack. */
3153 lua_rawgeti(session->hlua.T, LUA_REGISTRYINDEX, fcn->function_ref);
3154
3155 /* convert input sample and pust-it in the stack. */
3156 if (!lua_checkstack(session->hlua.T, 1)) {
3157 send_log(session->be, LOG_ERR, "Lua converter '%s': full stack.", fcn->name);
3158 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3159 Alert("Lua converter '%s': full stack.\n", fcn->name);
3160 return 0;
3161 }
3162 hlua_smp2lua(session->hlua.T, smp);
3163 session->hlua.nargs = 2;
3164
3165 /* push keywords in the stack. */
3166 if (arg_p) {
3167 for (; arg_p->type != ARGT_STOP; arg_p++) {
3168 if (!lua_checkstack(session->hlua.T, 1)) {
3169 send_log(session->be, LOG_ERR, "Lua converter '%s': full stack.", fcn->name);
3170 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3171 Alert("Lua converter '%s': full stack.\n", fcn->name);
3172 return 0;
3173 }
3174 hlua_arg2lua(session->hlua.T, arg_p);
3175 session->hlua.nargs++;
3176 }
3177 }
3178
Thierry FOURNIERbd413492015-03-03 16:52:26 +01003179 /* We must initialize the execution timeouts. */
3180 session->hlua.expire = tick_add(now_ms, hlua_timeout_session);
3181
Thierry FOURNIER9be813f2015-02-16 20:21:12 +01003182 /* Set the currently running flag. */
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +01003183 HLUA_SET_RUN(&session->hlua);
Thierry FOURNIER9be813f2015-02-16 20:21:12 +01003184 }
3185
3186 /* Execute the function. */
3187 switch (hlua_ctx_resume(&session->hlua, 0)) {
3188 /* finished. */
3189 case HLUA_E_OK:
3190 /* Convert the returned value in sample. */
3191 hlua_lua2smp(session->hlua.T, -1, smp);
3192 lua_pop(session->hlua.T, 1);
3193 return 1;
3194
3195 /* yield. */
3196 case HLUA_E_AGAIN:
3197 send_log(session->be, LOG_ERR, "Lua converter '%s': cannot use yielded functions.", fcn->name);
3198 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3199 Alert("Lua converter '%s': cannot use yielded functions.\n", fcn->name);
3200 return 0;
3201
3202 /* finished with error. */
3203 case HLUA_E_ERRMSG:
3204 /* Display log. */
3205 send_log(session->be, LOG_ERR, "Lua converter '%s': %s.", fcn->name, lua_tostring(session->hlua.T, -1));
3206 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3207 Alert("Lua converter '%s': %s.\n", fcn->name, lua_tostring(session->hlua.T, -1));
3208 lua_pop(session->hlua.T, 1);
3209 return 0;
3210
3211 case HLUA_E_ERR:
3212 /* Display log. */
3213 send_log(session->be, LOG_ERR, "Lua converter '%s' returns an unknown error.", fcn->name);
3214 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3215 Alert("Lua converter '%s' returns an unknown error.\n", fcn->name);
3216
3217 default:
3218 return 0;
3219 }
3220}
3221
Thierry FOURNIERfa0e5dd2015-02-16 20:19:18 +01003222/* Wrapper called by HAProxy to execute a sample-fetch. this wrapper
3223 * doesn't allow "yield" functions because the HAProxy engine cannot
3224 * resume sample-fetches.
3225 */
3226static int hlua_sample_fetch_wrapper(struct proxy *px, struct session *s, void *l7,
3227 unsigned int opt, const struct arg *arg_p,
3228 struct sample *smp, const char *kw, void *private)
3229{
3230 struct hlua_function *fcn = (struct hlua_function *)private;
3231
Thierry FOURNIER05ac4242015-02-27 18:37:27 +01003232 /* In the execution wrappers linked with a session, the
3233 * Lua context can be not initialized. This behavior
3234 * permits to save performances because a systematic
3235 * Lua initialization cause 5% performances loss.
3236 */
3237 if (!s->hlua.T && !hlua_ctx_init(&s->hlua, s->task)) {
3238 send_log(s->be, LOG_ERR, "Lua sample-fetch '%s': can't initialize Lua context.", fcn->name);
3239 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3240 Alert("Lua sample-fetch '%s': can't initialize Lua context.\n", fcn->name);
3241 return 0;
3242 }
3243
Thierry FOURNIERfa0e5dd2015-02-16 20:19:18 +01003244 /* If it is the first run, initialize the data for the call. */
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +01003245 if (!HLUA_IS_RUNNING(&s->hlua)) {
Thierry FOURNIERfa0e5dd2015-02-16 20:19:18 +01003246 /* Check stack available size. */
3247 if (!lua_checkstack(s->hlua.T, 2)) {
3248 send_log(px, LOG_ERR, "Lua sample-fetch '%s': full stack.", fcn->name);
3249 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3250 Alert("Lua sample-fetch '%s': full stack.\n", fcn->name);
3251 return 0;
3252 }
3253
3254 /* Restore the function in the stack. */
3255 lua_rawgeti(s->hlua.T, LUA_REGISTRYINDEX, fcn->function_ref);
3256
3257 /* push arguments in the stack. */
3258 if (!hlua_txn_new(s->hlua.T, s, px, l7)) {
3259 send_log(px, LOG_ERR, "Lua sample-fetch '%s': full stack.", fcn->name);
3260 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3261 Alert("Lua sample-fetch '%s': full stack.\n", fcn->name);
3262 return 0;
3263 }
3264 s->hlua.nargs = 1;
3265
3266 /* push keywords in the stack. */
3267 for (; arg_p && arg_p->type != ARGT_STOP; arg_p++) {
3268 /* Check stack available size. */
3269 if (!lua_checkstack(s->hlua.T, 1)) {
3270 send_log(px, LOG_ERR, "Lua sample-fetch '%s': full stack.", fcn->name);
3271 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3272 Alert("Lua sample-fetch '%s': full stack.\n", fcn->name);
3273 return 0;
3274 }
3275 if (!lua_checkstack(s->hlua.T, 1)) {
3276 send_log(px, LOG_ERR, "Lua sample-fetch '%s': full stack.", fcn->name);
3277 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3278 Alert("Lua sample-fetch '%s': full stack.\n", fcn->name);
3279 return 0;
3280 }
3281 hlua_arg2lua(s->hlua.T, arg_p);
3282 s->hlua.nargs++;
3283 }
3284
Thierry FOURNIERbd413492015-03-03 16:52:26 +01003285 /* We must initialize the execution timeouts. */
3286 s->hlua.expire = tick_add(now_ms, hlua_timeout_session);
3287
Thierry FOURNIERfa0e5dd2015-02-16 20:19:18 +01003288 /* Set the currently running flag. */
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +01003289 HLUA_SET_RUN(&s->hlua);
Thierry FOURNIERfa0e5dd2015-02-16 20:19:18 +01003290 }
3291
3292 /* Execute the function. */
3293 switch (hlua_ctx_resume(&s->hlua, 0)) {
3294 /* finished. */
3295 case HLUA_E_OK:
3296 /* Convert the returned value in sample. */
3297 hlua_lua2smp(s->hlua.T, -1, smp);
3298 lua_pop(s->hlua.T, 1);
3299
3300 /* Set the end of execution flag. */
3301 smp->flags &= ~SMP_F_MAY_CHANGE;
3302 return 1;
3303
3304 /* yield. */
3305 case HLUA_E_AGAIN:
3306 send_log(px, LOG_ERR, "Lua sample-fetch '%s': cannot use yielded functions.", fcn->name);
3307 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3308 Alert("Lua sample-fetch '%s': cannot use yielded functions.\n", fcn->name);
3309 return 0;
3310
3311 /* finished with error. */
3312 case HLUA_E_ERRMSG:
3313 /* Display log. */
3314 send_log(px, LOG_ERR, "Lua sample-fetch '%s': %s.", fcn->name, lua_tostring(s->hlua.T, -1));
3315 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3316 Alert("Lua sample-fetch '%s': %s.\n", fcn->name, lua_tostring(s->hlua.T, -1));
3317 lua_pop(s->hlua.T, 1);
3318 return 0;
3319
3320 case HLUA_E_ERR:
3321 /* Display log. */
3322 send_log(px, LOG_ERR, "Lua sample-fetch '%s' returns an unknown error.", fcn->name);
3323 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3324 Alert("Lua sample-fetch '%s': returns an unknown error.\n", fcn->name);
3325
3326 default:
3327 return 0;
3328 }
3329}
3330
Thierry FOURNIER9be813f2015-02-16 20:21:12 +01003331/* This function is an LUA binding used for registering
3332 * "sample-conv" functions. It expects a converter name used
3333 * in the haproxy configuration file, and an LUA function.
3334 */
3335__LJMP static int hlua_register_converters(lua_State *L)
3336{
3337 struct sample_conv_kw_list *sck;
3338 const char *name;
3339 int ref;
3340 int len;
3341 struct hlua_function *fcn;
3342
3343 MAY_LJMP(check_args(L, 2, "register_converters"));
3344
3345 /* First argument : converter name. */
3346 name = MAY_LJMP(luaL_checkstring(L, 1));
3347
3348 /* Second argument : lua function. */
3349 ref = MAY_LJMP(hlua_checkfunction(L, 2));
3350
3351 /* Allocate and fill the sample fetch keyword struct. */
3352 sck = malloc(sizeof(struct sample_conv_kw_list) +
3353 sizeof(struct sample_conv) * 2);
3354 if (!sck)
3355 WILL_LJMP(luaL_error(L, "lua out of memory error."));
3356 fcn = malloc(sizeof(*fcn));
3357 if (!fcn)
3358 WILL_LJMP(luaL_error(L, "lua out of memory error."));
3359
3360 /* Fill fcn. */
3361 fcn->name = strdup(name);
3362 if (!fcn->name)
3363 WILL_LJMP(luaL_error(L, "lua out of memory error."));
3364 fcn->function_ref = ref;
3365
3366 /* List head */
3367 sck->list.n = sck->list.p = NULL;
3368
3369 /* converter keyword. */
3370 len = strlen("lua.") + strlen(name) + 1;
3371 sck->kw[0].kw = malloc(len);
3372 if (!sck->kw[0].kw)
3373 WILL_LJMP(luaL_error(L, "lua out of memory error."));
3374
3375 snprintf((char *)sck->kw[0].kw, len, "lua.%s", name);
3376 sck->kw[0].process = hlua_sample_conv_wrapper;
3377 sck->kw[0].arg_mask = ARG5(0,STR,STR,STR,STR,STR);
3378 sck->kw[0].val_args = NULL;
3379 sck->kw[0].in_type = SMP_T_STR;
3380 sck->kw[0].out_type = SMP_T_STR;
3381 sck->kw[0].private = fcn;
3382
3383 /* End of array. */
3384 memset(&sck->kw[1], 0, sizeof(struct sample_conv));
3385
3386 /* Register this new converter */
3387 sample_register_convs(sck);
3388
3389 return 0;
3390}
3391
Thierry FOURNIERfa0e5dd2015-02-16 20:19:18 +01003392/* This fucntion is an LUA binding used for registering
3393 * "sample-fetch" functions. It expects a converter name used
3394 * in the haproxy configuration file, and an LUA function.
3395 */
3396__LJMP static int hlua_register_fetches(lua_State *L)
3397{
3398 const char *name;
3399 int ref;
3400 int len;
3401 struct sample_fetch_kw_list *sfk;
3402 struct hlua_function *fcn;
3403
3404 MAY_LJMP(check_args(L, 2, "register_fetches"));
3405
3406 /* First argument : sample-fetch name. */
3407 name = MAY_LJMP(luaL_checkstring(L, 1));
3408
3409 /* Second argument : lua function. */
3410 ref = MAY_LJMP(hlua_checkfunction(L, 2));
3411
3412 /* Allocate and fill the sample fetch keyword struct. */
3413 sfk = malloc(sizeof(struct sample_fetch_kw_list) +
3414 sizeof(struct sample_fetch) * 2);
3415 if (!sfk)
3416 WILL_LJMP(luaL_error(L, "lua out of memory error."));
3417 fcn = malloc(sizeof(*fcn));
3418 if (!fcn)
3419 WILL_LJMP(luaL_error(L, "lua out of memory error."));
3420
3421 /* Fill fcn. */
3422 fcn->name = strdup(name);
3423 if (!fcn->name)
3424 WILL_LJMP(luaL_error(L, "lua out of memory error."));
3425 fcn->function_ref = ref;
3426
3427 /* List head */
3428 sfk->list.n = sfk->list.p = NULL;
3429
3430 /* sample-fetch keyword. */
3431 len = strlen("lua.") + strlen(name) + 1;
3432 sfk->kw[0].kw = malloc(len);
3433 if (!sfk->kw[0].kw)
3434 return luaL_error(L, "lua out of memory error.");
3435
3436 snprintf((char *)sfk->kw[0].kw, len, "lua.%s", name);
3437 sfk->kw[0].process = hlua_sample_fetch_wrapper;
3438 sfk->kw[0].arg_mask = ARG5(0,STR,STR,STR,STR,STR);
3439 sfk->kw[0].val_args = NULL;
3440 sfk->kw[0].out_type = SMP_T_STR;
3441 sfk->kw[0].use = SMP_USE_HTTP_ANY;
3442 sfk->kw[0].val = 0;
3443 sfk->kw[0].private = fcn;
3444
3445 /* End of array. */
3446 memset(&sfk->kw[1], 0, sizeof(struct sample_fetch));
3447
3448 /* Register this new fetch. */
3449 sample_register_fetches(sfk);
3450
3451 return 0;
3452}
3453
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003454/* global {tcp|http}-request parser. Return 1 in succes case, else return 0. */
3455static int hlua_parse_rule(const char **args, int *cur_arg, struct proxy *px,
3456 struct hlua_rule **rule_p, char **err)
3457{
3458 struct hlua_rule *rule;
3459
3460 /* Memory for the rule. */
3461 rule = malloc(sizeof(*rule));
3462 if (!rule) {
3463 memprintf(err, "out of memory error");
3464 return 0;
3465 }
3466 *rule_p = rule;
3467
3468 /* The requiered arg is a function name. */
3469 if (!args[*cur_arg]) {
3470 memprintf(err, "expect Lua function name");
3471 return 0;
3472 }
3473
3474 /* Lookup for the symbol, and check if it is a function. */
3475 lua_getglobal(gL.T, args[*cur_arg]);
3476 if (lua_isnil(gL.T, -1)) {
3477 lua_pop(gL.T, 1);
3478 memprintf(err, "Lua function '%s' not found", args[*cur_arg]);
3479 return 0;
3480 }
3481 if (!lua_isfunction(gL.T, -1)) {
3482 lua_pop(gL.T, 1);
3483 memprintf(err, "'%s' is not a function", args[*cur_arg]);
3484 return 0;
3485 }
3486
3487 /* Reference the Lua function and store the reference. */
3488 rule->fcn.function_ref = luaL_ref(gL.T, LUA_REGISTRYINDEX);
3489 rule->fcn.name = strdup(args[*cur_arg]);
3490 if (!rule->fcn.name) {
3491 memprintf(err, "out of memory error.");
3492 return 0;
3493 }
3494 (*cur_arg)++;
3495
3496 /* TODO: later accept arguments. */
3497 rule->args = NULL;
3498
3499 return 1;
3500}
3501
3502/* This function is a wrapper to execute each LUA function declared
3503 * as an action wrapper during the initialisation period. This function
3504 * return 1 if the processing is finished (with oe without error) and
3505 * return 0 if the function must be called again because the LUA
3506 * returns a yield.
3507 */
3508static int hlua_request_act_wrapper(struct hlua_rule *rule, struct proxy *px,
3509 struct session *s, struct http_txn *http_txn,
3510 unsigned int analyzer)
3511{
3512 char **arg;
3513
Thierry FOURNIER05ac4242015-02-27 18:37:27 +01003514 /* In the execution wrappers linked with a session, the
3515 * Lua context can be not initialized. This behavior
3516 * permits to save performances because a systematic
3517 * Lua initialization cause 5% performances loss.
3518 */
3519 if (!s->hlua.T && !hlua_ctx_init(&s->hlua, s->task)) {
3520 send_log(px, LOG_ERR, "Lua action '%s': can't initialize Lua context.", rule->fcn.name);
3521 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3522 Alert("Lua action '%s': can't initialize Lua context.\n", rule->fcn.name);
3523 return 0;
3524 }
3525
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003526 /* If it is the first run, initialize the data for the call. */
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +01003527 if (!HLUA_IS_RUNNING(&s->hlua)) {
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003528 /* Check stack available size. */
3529 if (!lua_checkstack(s->hlua.T, 1)) {
3530 send_log(px, LOG_ERR, "Lua function '%s': full stack.", rule->fcn.name);
3531 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3532 Alert("Lua function '%s': full stack.\n", rule->fcn.name);
3533 return 0;
3534 }
3535
3536 /* Restore the function in the stack. */
3537 lua_rawgeti(s->hlua.T, LUA_REGISTRYINDEX, rule->fcn.function_ref);
3538
3539 /* Create and and push object session in the stack. */
3540 if (!hlua_txn_new(s->hlua.T, s, px, http_txn)) {
3541 send_log(px, LOG_ERR, "Lua function '%s': full stack.", rule->fcn.name);
3542 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3543 Alert("Lua function '%s': full stack.\n", rule->fcn.name);
3544 return 0;
3545 }
3546 s->hlua.nargs = 1;
3547
3548 /* push keywords in the stack. */
3549 for (arg = rule->args; arg && *arg; arg++) {
3550 if (!lua_checkstack(s->hlua.T, 1)) {
3551 send_log(px, LOG_ERR, "Lua function '%s': full stack.", rule->fcn.name);
3552 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3553 Alert("Lua function '%s': full stack.\n", rule->fcn.name);
3554 return 0;
3555 }
3556 lua_pushstring(s->hlua.T, *arg);
3557 s->hlua.nargs++;
3558 }
3559
Thierry FOURNIERbd413492015-03-03 16:52:26 +01003560 /* We must initialize the execution timeouts. */
3561 s->hlua.expire = tick_add(now_ms, hlua_timeout_session);
3562
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003563 /* Set the currently running flag. */
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +01003564 HLUA_SET_RUN(&s->hlua);
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003565 }
3566
3567 /* Execute the function. */
3568 switch (hlua_ctx_resume(&s->hlua, 1)) {
3569 /* finished. */
3570 case HLUA_E_OK:
3571 return 1;
3572
3573 /* yield. */
3574 case HLUA_E_AGAIN:
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +01003575 /* Set timeout in the required channel. */
3576 if (s->hlua.wake_time != TICK_ETERNITY) {
3577 if (analyzer & (AN_REQ_INSPECT_FE|AN_REQ_HTTP_PROCESS_FE))
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01003578 s->req.analyse_exp = s->hlua.wake_time;
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +01003579 else if (analyzer & (AN_RES_INSPECT|AN_RES_HTTP_PROCESS_BE))
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01003580 s->res.analyse_exp = s->hlua.wake_time;
Thierry FOURNIERc42c1ae2015-03-03 17:17:55 +01003581 }
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003582 /* Some actions can be wake up when a "write" event
3583 * is detected on a response channel. This is useful
3584 * only for actions targetted on the requests.
3585 */
Thierry FOURNIERef6a2112015-03-05 17:45:34 +01003586 if (HLUA_IS_WAKERESWR(&s->hlua)) {
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01003587 s->res.flags |= CF_WAKE_WRITE;
Willy Tarreau76bd97f2015-03-10 17:16:10 +01003588 if ((analyzer & (AN_REQ_INSPECT_FE|AN_REQ_HTTP_PROCESS_FE)))
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01003589 s->res.analysers |= analyzer;
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003590 }
Thierry FOURNIER53e08ec2015-03-06 00:35:53 +01003591 if (HLUA_IS_WAKEREQWR(&s->hlua))
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01003592 s->req.flags |= CF_WAKE_WRITE;
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003593 return 0;
3594
3595 /* finished with error. */
3596 case HLUA_E_ERRMSG:
3597 /* Display log. */
3598 send_log(px, LOG_ERR, "Lua function '%s': %s.", rule->fcn.name, lua_tostring(s->hlua.T, -1));
3599 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3600 Alert("Lua function '%s': %s.\n", rule->fcn.name, lua_tostring(s->hlua.T, -1));
3601 lua_pop(s->hlua.T, 1);
3602 return 1;
3603
3604 case HLUA_E_ERR:
3605 /* Display log. */
3606 send_log(px, LOG_ERR, "Lua function '%s' return an unknown error.", rule->fcn.name);
3607 if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
3608 Alert("Lua function '%s' return an unknown error.\n", rule->fcn.name);
3609
3610 default:
3611 return 1;
3612 }
3613}
3614
3615/* Lua execution wrapper for "tcp-request". This function uses
3616 * "hlua_request_act_wrapper" for executing the LUA code.
3617 */
3618int hlua_tcp_req_act_wrapper(struct tcp_rule *tcp_rule, struct proxy *px,
3619 struct session *s)
3620{
3621 return hlua_request_act_wrapper((struct hlua_rule *)tcp_rule->act_prm.data,
3622 px, s, NULL, AN_REQ_INSPECT_FE);
3623}
3624
3625/* Lua execution wrapper for "tcp-response". This function uses
3626 * "hlua_request_act_wrapper" for executing the LUA code.
3627 */
3628int hlua_tcp_res_act_wrapper(struct tcp_rule *tcp_rule, struct proxy *px,
3629 struct session *s)
3630{
3631 return hlua_request_act_wrapper((struct hlua_rule *)tcp_rule->act_prm.data,
3632 px, s, NULL, AN_RES_INSPECT);
3633}
3634
3635/* Lua execution wrapper for http-request.
3636 * This function uses "hlua_request_act_wrapper" for executing
3637 * the LUA code.
3638 */
3639int hlua_http_req_act_wrapper(struct http_req_rule *rule, struct proxy *px,
3640 struct session *s, struct http_txn *http_txn)
3641{
3642 return hlua_request_act_wrapper((struct hlua_rule *)rule->arg.data, px,
3643 s, http_txn, AN_REQ_HTTP_PROCESS_FE);
3644}
3645
3646/* Lua execution wrapper for http-response.
3647 * This function uses "hlua_request_act_wrapper" for executing
3648 * the LUA code.
3649 */
3650int hlua_http_res_act_wrapper(struct http_res_rule *rule, struct proxy *px,
3651 struct session *s, struct http_txn *http_txn)
3652{
3653 return hlua_request_act_wrapper((struct hlua_rule *)rule->arg.data, px,
3654 s, http_txn, AN_RES_HTTP_PROCESS_BE);
3655}
3656
3657/* tcp-request <*> configuration wrapper. */
3658static int tcp_req_action_register_lua(const char **args, int *cur_arg, struct proxy *px,
3659 struct tcp_rule *rule, char **err)
3660{
3661 if (!hlua_parse_rule(args, cur_arg, px, (struct hlua_rule **)&rule->act_prm.data, err))
Thierry FOURNIER893bfa32015-02-17 18:42:34 +01003662 return 0;
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003663 rule->action = TCP_ACT_CUSTOM;
3664 rule->action_ptr = hlua_tcp_req_act_wrapper;
3665 return 1;
3666}
3667
3668/* tcp-response <*> configuration wrapper. */
3669static int tcp_res_action_register_lua(const char **args, int *cur_arg, struct proxy *px,
3670 struct tcp_rule *rule, char **err)
3671{
3672 if (!hlua_parse_rule(args, cur_arg, px, (struct hlua_rule **)&rule->act_prm.data, err))
Thierry FOURNIER893bfa32015-02-17 18:42:34 +01003673 return 0;
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003674 rule->action = TCP_ACT_CUSTOM;
3675 rule->action_ptr = hlua_tcp_res_act_wrapper;
3676 return 1;
3677}
3678
3679/* http-request <*> configuration wrapper. */
3680static int http_req_action_register_lua(const char **args, int *cur_arg, struct proxy *px,
3681 struct http_req_rule *rule, char **err)
3682{
3683 if (!hlua_parse_rule(args, cur_arg, px, (struct hlua_rule **)&rule->arg.data, err))
3684 return -1;
3685 rule->action = HTTP_REQ_ACT_CUSTOM_CONT;
3686 rule->action_ptr = hlua_http_req_act_wrapper;
3687 return 1;
3688}
3689
3690/* http-response <*> configuration wrapper. */
3691static int http_res_action_register_lua(const char **args, int *cur_arg, struct proxy *px,
3692 struct http_res_rule *rule, char **err)
3693{
3694 if (!hlua_parse_rule(args, cur_arg, px, (struct hlua_rule **)&rule->arg.data, err))
3695 return -1;
3696 rule->action = HTTP_RES_ACT_CUSTOM_CONT;
3697 rule->action_ptr = hlua_http_res_act_wrapper;
3698 return 1;
3699}
3700
Thierry FOURNIERbd413492015-03-03 16:52:26 +01003701static int hlua_read_timeout(char **args, int section_type, struct proxy *curpx,
3702 struct proxy *defpx, const char *file, int line,
3703 char **err, unsigned int *timeout)
3704{
3705 const char *error;
3706
3707 error = parse_time_err(args[1], timeout, TIME_UNIT_MS);
3708 if (error && *error != '\0') {
3709 memprintf(err, "%s: invalid timeout", args[0]);
3710 return -1;
3711 }
3712 return 0;
3713}
3714
3715static int hlua_session_timeout(char **args, int section_type, struct proxy *curpx,
3716 struct proxy *defpx, const char *file, int line,
3717 char **err)
3718{
3719 return hlua_read_timeout(args, section_type, curpx, defpx,
3720 file, line, err, &hlua_timeout_session);
3721}
3722
3723static int hlua_task_timeout(char **args, int section_type, struct proxy *curpx,
3724 struct proxy *defpx, const char *file, int line,
3725 char **err)
3726{
3727 return hlua_read_timeout(args, section_type, curpx, defpx,
3728 file, line, err, &hlua_timeout_task);
3729}
3730
Thierry FOURNIERee9f8022015-03-03 17:37:37 +01003731static int hlua_forced_yield(char **args, int section_type, struct proxy *curpx,
3732 struct proxy *defpx, const char *file, int line,
3733 char **err)
3734{
3735 char *error;
3736
3737 hlua_nb_instruction = strtoll(args[1], &error, 10);
3738 if (*error != '\0') {
3739 memprintf(err, "%s: invalid number", args[0]);
3740 return -1;
3741 }
3742 return 0;
3743}
3744
Thierry FOURNIER6c9b52c2015-01-23 15:57:06 +01003745/* This function is called by the main configuration key "lua-load". It loads and
3746 * execute an lua file during the parsing of the HAProxy configuration file. It is
3747 * the main lua entry point.
3748 *
3749 * This funtion runs with the HAProxy keywords API. It returns -1 if an error is
3750 * occured, otherwise it returns 0.
3751 *
3752 * In some error case, LUA set an error message in top of the stack. This function
3753 * returns this error message in the HAProxy logs and pop it from the stack.
3754 */
3755static int hlua_load(char **args, int section_type, struct proxy *curpx,
3756 struct proxy *defpx, const char *file, int line,
3757 char **err)
3758{
3759 int error;
3760
3761 /* Just load and compile the file. */
3762 error = luaL_loadfile(gL.T, args[1]);
3763 if (error) {
3764 memprintf(err, "error in lua file '%s': %s", args[1], lua_tostring(gL.T, -1));
3765 lua_pop(gL.T, 1);
3766 return -1;
3767 }
3768
3769 /* If no syntax error where detected, execute the code. */
3770 error = lua_pcall(gL.T, 0, LUA_MULTRET, 0);
3771 switch (error) {
3772 case LUA_OK:
3773 break;
3774 case LUA_ERRRUN:
3775 memprintf(err, "lua runtime error: %s\n", lua_tostring(gL.T, -1));
3776 lua_pop(gL.T, 1);
3777 return -1;
3778 case LUA_ERRMEM:
3779 memprintf(err, "lua out of memory error\n");
3780 return -1;
3781 case LUA_ERRERR:
3782 memprintf(err, "lua message handler error: %s\n", lua_tostring(gL.T, -1));
3783 lua_pop(gL.T, 1);
3784 return -1;
3785 case LUA_ERRGCMM:
3786 memprintf(err, "lua garbage collector error: %s\n", lua_tostring(gL.T, -1));
3787 lua_pop(gL.T, 1);
3788 return -1;
3789 default:
3790 memprintf(err, "lua unknonwn error: %s\n", lua_tostring(gL.T, -1));
3791 lua_pop(gL.T, 1);
3792 return -1;
3793 }
3794
3795 return 0;
3796}
3797
3798/* configuration keywords declaration */
3799static struct cfg_kw_list cfg_kws = {{ },{
Thierry FOURNIERbd413492015-03-03 16:52:26 +01003800 { CFG_GLOBAL, "lua-load", hlua_load },
3801 { CFG_GLOBAL, "tune.lua.session-timeout", hlua_session_timeout },
3802 { CFG_GLOBAL, "tune.lua.task-timeout", hlua_task_timeout },
Thierry FOURNIERee9f8022015-03-03 17:37:37 +01003803 { CFG_GLOBAL, "tune.lua.forced-yield", hlua_forced_yield },
Thierry FOURNIER6c9b52c2015-01-23 15:57:06 +01003804 { 0, NULL, NULL },
3805}};
3806
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003807static struct http_req_action_kw_list http_req_kws = {"lua", { }, {
3808 { "lua", http_req_action_register_lua },
3809 { NULL, NULL }
3810}};
3811
3812static struct http_res_action_kw_list http_res_kws = {"lua", { }, {
3813 { "lua", http_res_action_register_lua },
3814 { NULL, NULL }
3815}};
3816
3817static struct tcp_action_kw_list tcp_req_cont_kws = {"lua", { }, {
3818 { "lua", tcp_req_action_register_lua },
3819 { NULL, NULL }
3820}};
3821
3822static struct tcp_action_kw_list tcp_res_cont_kws = {"lua", { }, {
3823 { "lua", tcp_res_action_register_lua },
3824 { NULL, NULL }
3825}};
3826
Thierry FOURNIERa4a0f3d2015-01-23 12:08:30 +01003827int hlua_post_init()
3828{
3829 struct hlua_init_function *init;
3830 const char *msg;
3831 enum hlua_exec ret;
3832
3833 list_for_each_entry(init, &hlua_init_functions, l) {
3834 lua_rawgeti(gL.T, LUA_REGISTRYINDEX, init->function_ref);
3835 ret = hlua_ctx_resume(&gL, 0);
3836 switch (ret) {
3837 case HLUA_E_OK:
3838 lua_pop(gL.T, -1);
3839 return 1;
3840 case HLUA_E_AGAIN:
3841 Alert("lua init: yield not allowed.\n");
3842 return 0;
3843 case HLUA_E_ERRMSG:
3844 msg = lua_tostring(gL.T, -1);
3845 Alert("lua init: %s.\n", msg);
3846 return 0;
3847 case HLUA_E_ERR:
3848 default:
3849 Alert("lua init: unknown runtime error.\n");
3850 return 0;
3851 }
3852 }
3853 return 1;
3854}
3855
Thierry FOURNIER6f1fd482015-01-23 14:06:13 +01003856void hlua_init(void)
3857{
Thierry FOURNIER2ba18a22015-01-23 14:07:08 +01003858 int i;
Thierry FOURNIERd0fa5382015-02-16 20:14:51 +01003859 int idx;
3860 struct sample_fetch *sf;
Thierry FOURNIER594afe72015-03-10 23:58:30 +01003861 struct sample_conv *sc;
Thierry FOURNIERd0fa5382015-02-16 20:14:51 +01003862 char *p;
Willy Tarreau80f5fae2015-02-27 16:38:20 +01003863#ifdef USE_OPENSSL
3864 char *args[4];
3865 struct srv_kw *kw;
3866 int tmp_error;
3867 char *error;
3868#endif
Thierry FOURNIER2ba18a22015-01-23 14:07:08 +01003869
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +01003870 /* Initialise com signals pool session. */
3871 pool2_hlua_com = create_pool("hlua_com", sizeof(struct hlua_com), MEM_F_SHARED);
3872
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01003873 /* Initialise sleep pool. */
3874 pool2_hlua_sleep = create_pool("hlua_sleep", sizeof(struct hlua_sleep), MEM_F_SHARED);
3875
Thierry FOURNIER6c9b52c2015-01-23 15:57:06 +01003876 /* Register configuration keywords. */
3877 cfg_register_keywords(&cfg_kws);
3878
Thierry FOURNIER258d8aa2015-02-16 20:23:40 +01003879 /* Register custom HTTP rules. */
3880 http_req_keywords_register(&http_req_kws);
3881 http_res_keywords_register(&http_res_kws);
3882 tcp_req_cont_keywords_register(&tcp_req_cont_kws);
3883 tcp_res_cont_keywords_register(&tcp_res_cont_kws);
3884
Thierry FOURNIER380d0932015-01-23 14:27:52 +01003885 /* Init main lua stack. */
3886 gL.Mref = LUA_REFNIL;
Thierry FOURNIERa097fdf2015-03-03 15:17:35 +01003887 gL.flags = 0;
Thierry FOURNIER9ff7e6e2015-01-23 11:08:20 +01003888 LIST_INIT(&gL.com);
Thierry FOURNIER380d0932015-01-23 14:27:52 +01003889 gL.T = luaL_newstate();
3890 hlua_sethlua(&gL);
3891 gL.Tref = LUA_REFNIL;
3892 gL.task = NULL;
3893
3894 /* Initialise lua. */
3895 luaL_openlibs(gL.T);
Thierry FOURNIER2ba18a22015-01-23 14:07:08 +01003896
3897 /*
3898 *
3899 * Create "core" object.
3900 *
3901 */
3902
Thierry FOURNIERa2d8c652015-03-11 17:29:39 +01003903 /* This table entry is the object "core" base. */
Thierry FOURNIER2ba18a22015-01-23 14:07:08 +01003904 lua_newtable(gL.T);
3905
3906 /* Push the loglevel constants. */
Willy Tarreau80f5fae2015-02-27 16:38:20 +01003907 for (i = 0; i < NB_LOG_LEVELS; i++)
Thierry FOURNIER2ba18a22015-01-23 14:07:08 +01003908 hlua_class_const_int(gL.T, log_levels[i], i);
3909
Thierry FOURNIERa4a0f3d2015-01-23 12:08:30 +01003910 /* Register special functions. */
3911 hlua_class_function(gL.T, "register_init", hlua_register_init);
Thierry FOURNIER24f33532015-01-23 12:13:00 +01003912 hlua_class_function(gL.T, "register_task", hlua_register_task);
Thierry FOURNIERfa0e5dd2015-02-16 20:19:18 +01003913 hlua_class_function(gL.T, "register_fetches", hlua_register_fetches);
Thierry FOURNIER9be813f2015-02-16 20:21:12 +01003914 hlua_class_function(gL.T, "register_converters", hlua_register_converters);
Thierry FOURNIER13416fe2015-02-17 15:01:59 +01003915 hlua_class_function(gL.T, "yield", hlua_yield);
Willy Tarreau59551662015-03-10 14:23:13 +01003916 hlua_class_function(gL.T, "set_nice", hlua_set_nice);
Thierry FOURNIER5b8608f2015-02-16 19:43:25 +01003917 hlua_class_function(gL.T, "sleep", hlua_sleep);
3918 hlua_class_function(gL.T, "msleep", hlua_msleep);
Thierry FOURNIER83758bb2015-02-04 13:21:04 +01003919 hlua_class_function(gL.T, "add_acl", hlua_add_acl);
3920 hlua_class_function(gL.T, "del_acl", hlua_del_acl);
3921 hlua_class_function(gL.T, "set_map", hlua_set_map);
3922 hlua_class_function(gL.T, "del_map", hlua_del_map);
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01003923 hlua_class_function(gL.T, "tcp", hlua_socket_new);
Thierry FOURNIERa4a0f3d2015-01-23 12:08:30 +01003924
Thierry FOURNIER2ba18a22015-01-23 14:07:08 +01003925 lua_setglobal(gL.T, "core");
Thierry FOURNIER65f34c62015-02-16 20:11:43 +01003926
3927 /*
3928 *
Thierry FOURNIER5a6d3fd2015-02-09 16:38:34 +01003929 * Register class Channel
3930 *
3931 */
3932
3933 /* Create and fill the metatable. */
3934 lua_newtable(gL.T);
3935
3936 /* Create and fille the __index entry. */
3937 lua_pushstring(gL.T, "__index");
3938 lua_newtable(gL.T);
3939
3940 /* Register . */
3941 hlua_class_function(gL.T, "get", hlua_channel_get);
3942 hlua_class_function(gL.T, "dup", hlua_channel_dup);
3943 hlua_class_function(gL.T, "getline", hlua_channel_getline);
3944 hlua_class_function(gL.T, "set", hlua_channel_set);
3945 hlua_class_function(gL.T, "append", hlua_channel_append);
3946 hlua_class_function(gL.T, "send", hlua_channel_send);
3947 hlua_class_function(gL.T, "forward", hlua_channel_forward);
3948 hlua_class_function(gL.T, "get_in_len", hlua_channel_get_in_len);
3949 hlua_class_function(gL.T, "get_out_len", hlua_channel_get_out_len);
3950
3951 lua_settable(gL.T, -3);
3952
3953 /* Register previous table in the registry with reference and named entry. */
3954 lua_pushvalue(gL.T, -1); /* Copy the -1 entry and push it on the stack. */
3955 lua_setfield(gL.T, LUA_REGISTRYINDEX, CLASS_CHANNEL); /* register class session. */
3956 class_channel_ref = luaL_ref(gL.T, LUA_REGISTRYINDEX); /* reference class session. */
3957
3958 /*
3959 *
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01003960 * Register class Fetches
Thierry FOURNIER65f34c62015-02-16 20:11:43 +01003961 *
3962 */
3963
3964 /* Create and fill the metatable. */
3965 lua_newtable(gL.T);
3966
3967 /* Create and fille the __index entry. */
3968 lua_pushstring(gL.T, "__index");
3969 lua_newtable(gL.T);
3970
Thierry FOURNIERd0fa5382015-02-16 20:14:51 +01003971 /* Browse existing fetches and create the associated
3972 * object method.
3973 */
3974 sf = NULL;
3975 while ((sf = sample_fetch_getnext(sf, &idx)) != NULL) {
3976
3977 /* Dont register the keywork if the arguments check function are
3978 * not safe during the runtime.
3979 */
3980 if ((sf->val_args != NULL) &&
3981 (sf->val_args != val_payload_lv) &&
3982 (sf->val_args != val_hdr))
3983 continue;
3984
3985 /* gL.Tua doesn't support '.' and '-' in the function names, replace it
3986 * by an underscore.
3987 */
3988 strncpy(trash.str, sf->kw, trash.size);
3989 trash.str[trash.size - 1] = '\0';
3990 for (p = trash.str; *p; p++)
3991 if (*p == '.' || *p == '-' || *p == '+')
3992 *p = '_';
3993
3994 /* Register the function. */
3995 lua_pushstring(gL.T, trash.str);
Willy Tarreau2ec22742015-03-10 14:27:20 +01003996 lua_pushlightuserdata(gL.T, sf);
Thierry FOURNIERd0fa5382015-02-16 20:14:51 +01003997 lua_pushcclosure(gL.T, hlua_run_sample_fetch, 1);
3998 lua_settable(gL.T, -3);
3999 }
4000
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01004001 lua_settable(gL.T, -3);
4002
4003 /* Register previous table in the registry with reference and named entry. */
4004 lua_pushvalue(gL.T, -1); /* Copy the -1 entry and push it on the stack. */
4005 lua_setfield(gL.T, LUA_REGISTRYINDEX, CLASS_FETCHES); /* register class session. */
4006 class_fetches_ref = luaL_ref(gL.T, LUA_REGISTRYINDEX); /* reference class session. */
4007
4008 /*
4009 *
Thierry FOURNIER594afe72015-03-10 23:58:30 +01004010 * Register class Converters
4011 *
4012 */
4013
4014 /* Create and fill the metatable. */
4015 lua_newtable(gL.T);
4016
4017 /* Create and fill the __index entry. */
4018 lua_pushstring(gL.T, "__index");
4019 lua_newtable(gL.T);
4020
4021 /* Browse existing converters and create the associated
4022 * object method.
4023 */
4024 sc = NULL;
4025 while ((sc = sample_conv_getnext(sc, &idx)) != NULL) {
4026 /* Dont register the keywork if the arguments check function are
4027 * not safe during the runtime.
4028 */
4029 if (sc->val_args != NULL)
4030 continue;
4031
4032 /* gL.Tua doesn't support '.' and '-' in the function names, replace it
4033 * by an underscore.
4034 */
4035 strncpy(trash.str, sc->kw, trash.size);
4036 trash.str[trash.size - 1] = '\0';
4037 for (p = trash.str; *p; p++)
4038 if (*p == '.' || *p == '-' || *p == '+')
4039 *p = '_';
4040
4041 /* Register the function. */
4042 lua_pushstring(gL.T, trash.str);
4043 lua_pushlightuserdata(gL.T, sc);
4044 lua_pushcclosure(gL.T, hlua_run_sample_conv, 1);
4045 lua_settable(gL.T, -3);
4046 }
4047
4048 lua_settable(gL.T, -3);
4049
4050 /* Register previous table in the registry with reference and named entry. */
4051 lua_pushvalue(gL.T, -1); /* Copy the -1 entry and push it on the stack. */
4052 lua_setfield(gL.T, LUA_REGISTRYINDEX, CLASS_CONVERTERS); /* register class session. */
4053 class_converters_ref = luaL_ref(gL.T, LUA_REGISTRYINDEX); /* reference class session. */
4054
4055 /*
4056 *
Thierry FOURNIERbb53c7b2015-03-11 18:28:02 +01004057 * Register class TXN
4058 *
4059 */
4060
4061 /* Create and fill the metatable. */
4062 lua_newtable(gL.T);
4063
4064 /* Create and fille the __index entry. */
4065 lua_pushstring(gL.T, "__index");
4066 lua_newtable(gL.T);
4067
Thierry FOURNIER05c0b8a2015-02-25 11:43:21 +01004068 /* Register Lua functions. */
Willy Tarreau59551662015-03-10 14:23:13 +01004069 hlua_class_function(gL.T, "get_headers", hlua_session_get_headers);
4070 hlua_class_function(gL.T, "set_priv", hlua_set_priv);
4071 hlua_class_function(gL.T, "get_priv", hlua_get_priv);
Thierry FOURNIER893bfa32015-02-17 18:42:34 +01004072 hlua_class_function(gL.T, "close", hlua_txn_close);
Thierry FOURNIER05c0b8a2015-02-25 11:43:21 +01004073
Thierry FOURNIER65f34c62015-02-16 20:11:43 +01004074 lua_settable(gL.T, -3);
4075
4076 /* Register previous table in the registry with reference and named entry. */
4077 lua_pushvalue(gL.T, -1); /* Copy the -1 entry and push it on the stack. */
4078 lua_setfield(gL.T, LUA_REGISTRYINDEX, CLASS_TXN); /* register class session. */
4079 class_txn_ref = luaL_ref(gL.T, LUA_REGISTRYINDEX); /* reference class session. */
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01004080
4081 /*
4082 *
4083 * Register class Socket
4084 *
4085 */
4086
4087 /* Create and fill the metatable. */
4088 lua_newtable(gL.T);
4089
4090 /* Create and fille the __index entry. */
4091 lua_pushstring(gL.T, "__index");
4092 lua_newtable(gL.T);
4093
Baptiste Assmann84bb4932015-03-02 21:40:06 +01004094#ifdef USE_OPENSSL
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01004095 hlua_class_function(gL.T, "connect_ssl", hlua_socket_connect_ssl);
Baptiste Assmann84bb4932015-03-02 21:40:06 +01004096#endif
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01004097 hlua_class_function(gL.T, "connect", hlua_socket_connect);
4098 hlua_class_function(gL.T, "send", hlua_socket_send);
4099 hlua_class_function(gL.T, "receive", hlua_socket_receive);
4100 hlua_class_function(gL.T, "close", hlua_socket_close);
4101 hlua_class_function(gL.T, "getpeername", hlua_socket_getpeername);
4102 hlua_class_function(gL.T, "getsockname", hlua_socket_getsockname);
4103 hlua_class_function(gL.T, "setoption", hlua_socket_setoption);
4104 hlua_class_function(gL.T, "settimeout", hlua_socket_settimeout);
4105
4106 lua_settable(gL.T, -3); /* Push the last 2 entries in the table at index -3 */
4107
4108 /* Register the garbage collector entry. */
4109 lua_pushstring(gL.T, "__gc");
4110 lua_pushcclosure(gL.T, hlua_socket_gc, 0);
4111 lua_settable(gL.T, -3); /* Push the last 2 entries in the table at index -3 */
4112
4113 /* Register previous table in the registry with reference and named entry. */
4114 lua_pushvalue(gL.T, -1); /* Copy the -1 entry and push it on the stack. */
4115 lua_pushvalue(gL.T, -1); /* Copy the -1 entry and push it on the stack. */
4116 lua_setfield(gL.T, LUA_REGISTRYINDEX, CLASS_SOCKET); /* register class socket. */
4117 class_socket_ref = luaL_ref(gL.T, LUA_REGISTRYINDEX); /* reference class socket. */
4118
4119 /* Proxy and server configuration initialisation. */
4120 memset(&socket_proxy, 0, sizeof(socket_proxy));
4121 init_new_proxy(&socket_proxy);
4122 socket_proxy.parent = NULL;
4123 socket_proxy.last_change = now.tv_sec;
4124 socket_proxy.id = "LUA-SOCKET";
4125 socket_proxy.cap = PR_CAP_FE | PR_CAP_BE;
4126 socket_proxy.maxconn = 0;
4127 socket_proxy.accept = NULL;
4128 socket_proxy.options2 |= PR_O2_INDEPSTR;
4129 socket_proxy.srv = NULL;
4130 socket_proxy.conn_retries = 0;
4131 socket_proxy.timeout.connect = 5000; /* By default the timeout connection is 5s. */
4132
4133 /* Init TCP server: unchanged parameters */
4134 memset(&socket_tcp, 0, sizeof(socket_tcp));
4135 socket_tcp.next = NULL;
4136 socket_tcp.proxy = &socket_proxy;
4137 socket_tcp.obj_type = OBJ_TYPE_SERVER;
4138 LIST_INIT(&socket_tcp.actconns);
4139 LIST_INIT(&socket_tcp.pendconns);
4140 socket_tcp.state = SRV_ST_RUNNING; /* early server setup */
4141 socket_tcp.last_change = 0;
4142 socket_tcp.id = "LUA-TCP-CONN";
4143 socket_tcp.check.state &= ~CHK_ST_ENABLED; /* Disable health checks. */
4144 socket_tcp.agent.state &= ~CHK_ST_ENABLED; /* Disable health checks. */
4145 socket_tcp.pp_opts = 0; /* Remove proxy protocol. */
4146
4147 /* XXX: Copy default parameter from default server,
4148 * but the default server is not initialized.
4149 */
4150 socket_tcp.maxqueue = socket_proxy.defsrv.maxqueue;
4151 socket_tcp.minconn = socket_proxy.defsrv.minconn;
4152 socket_tcp.maxconn = socket_proxy.defsrv.maxconn;
4153 socket_tcp.slowstart = socket_proxy.defsrv.slowstart;
4154 socket_tcp.onerror = socket_proxy.defsrv.onerror;
4155 socket_tcp.onmarkeddown = socket_proxy.defsrv.onmarkeddown;
4156 socket_tcp.onmarkedup = socket_proxy.defsrv.onmarkedup;
4157 socket_tcp.consecutive_errors_limit = socket_proxy.defsrv.consecutive_errors_limit;
4158 socket_tcp.uweight = socket_proxy.defsrv.iweight;
4159 socket_tcp.iweight = socket_proxy.defsrv.iweight;
4160
4161 socket_tcp.check.status = HCHK_STATUS_INI;
4162 socket_tcp.check.rise = socket_proxy.defsrv.check.rise;
4163 socket_tcp.check.fall = socket_proxy.defsrv.check.fall;
4164 socket_tcp.check.health = socket_tcp.check.rise; /* socket, but will fall down at first failure */
4165 socket_tcp.check.server = &socket_tcp;
4166
4167 socket_tcp.agent.status = HCHK_STATUS_INI;
4168 socket_tcp.agent.rise = socket_proxy.defsrv.agent.rise;
4169 socket_tcp.agent.fall = socket_proxy.defsrv.agent.fall;
4170 socket_tcp.agent.health = socket_tcp.agent.rise; /* socket, but will fall down at first failure */
4171 socket_tcp.agent.server = &socket_tcp;
4172
4173 socket_tcp.xprt = &raw_sock;
4174
4175#ifdef USE_OPENSSL
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01004176 /* Init TCP server: unchanged parameters */
4177 memset(&socket_ssl, 0, sizeof(socket_ssl));
4178 socket_ssl.next = NULL;
4179 socket_ssl.proxy = &socket_proxy;
4180 socket_ssl.obj_type = OBJ_TYPE_SERVER;
4181 LIST_INIT(&socket_ssl.actconns);
4182 LIST_INIT(&socket_ssl.pendconns);
4183 socket_ssl.state = SRV_ST_RUNNING; /* early server setup */
4184 socket_ssl.last_change = 0;
4185 socket_ssl.id = "LUA-SSL-CONN";
4186 socket_ssl.check.state &= ~CHK_ST_ENABLED; /* Disable health checks. */
4187 socket_ssl.agent.state &= ~CHK_ST_ENABLED; /* Disable health checks. */
4188 socket_ssl.pp_opts = 0; /* Remove proxy protocol. */
4189
4190 /* XXX: Copy default parameter from default server,
4191 * but the default server is not initialized.
4192 */
4193 socket_ssl.maxqueue = socket_proxy.defsrv.maxqueue;
4194 socket_ssl.minconn = socket_proxy.defsrv.minconn;
4195 socket_ssl.maxconn = socket_proxy.defsrv.maxconn;
4196 socket_ssl.slowstart = socket_proxy.defsrv.slowstart;
4197 socket_ssl.onerror = socket_proxy.defsrv.onerror;
4198 socket_ssl.onmarkeddown = socket_proxy.defsrv.onmarkeddown;
4199 socket_ssl.onmarkedup = socket_proxy.defsrv.onmarkedup;
4200 socket_ssl.consecutive_errors_limit = socket_proxy.defsrv.consecutive_errors_limit;
4201 socket_ssl.uweight = socket_proxy.defsrv.iweight;
4202 socket_ssl.iweight = socket_proxy.defsrv.iweight;
4203
4204 socket_ssl.check.status = HCHK_STATUS_INI;
4205 socket_ssl.check.rise = socket_proxy.defsrv.check.rise;
4206 socket_ssl.check.fall = socket_proxy.defsrv.check.fall;
4207 socket_ssl.check.health = socket_ssl.check.rise; /* socket, but will fall down at first failure */
4208 socket_ssl.check.server = &socket_ssl;
4209
4210 socket_ssl.agent.status = HCHK_STATUS_INI;
4211 socket_ssl.agent.rise = socket_proxy.defsrv.agent.rise;
4212 socket_ssl.agent.fall = socket_proxy.defsrv.agent.fall;
4213 socket_ssl.agent.health = socket_ssl.agent.rise; /* socket, but will fall down at first failure */
4214 socket_ssl.agent.server = &socket_ssl;
4215
4216 socket_ssl.xprt = &raw_sock;
4217
4218 args[0] = "ssl";
4219 args[1] = "verify";
4220 args[2] = "none";
4221 args[3] = NULL;
4222
Willy Tarreau80f5fae2015-02-27 16:38:20 +01004223 for (idx = 0; idx < 3; idx++) {
Thierry FOURNIER7e7ac322015-02-16 19:27:16 +01004224 if ((kw = srv_find_kw(args[idx])) != NULL) { /* Maybe it's registered server keyword */
4225 /*
4226 *
4227 * If the keyword is not known, we can search in the registered
4228 * server keywords. This is usefull to configure special SSL
4229 * features like client certificates and ssl_verify.
4230 *
4231 */
4232 tmp_error = kw->parse(args, &idx, &socket_proxy, &socket_ssl, &error);
4233 if (tmp_error != 0) {
4234 fprintf(stderr, "INTERNAL ERROR: %s\n", error);
4235 abort(); /* This must be never arrives because the command line
4236 not editable by the user. */
4237 }
4238 idx += kw->skip;
4239 }
4240 }
4241
4242 /* Initialize SSL server. */
4243 if (socket_ssl.xprt == &ssl_sock) {
4244 socket_ssl.use_ssl = 1;
4245 ssl_sock_prepare_srv_ctx(&socket_ssl, &socket_proxy);
4246 }
4247#endif
Thierry FOURNIER6f1fd482015-01-23 14:06:13 +01004248}