blob: c2d26817ac3e0f4699c739c28c304f42cbf45e81 [file] [log] [blame]
Thierry Fourniere726b142016-02-11 17:57:57 +01001/*
2 * Lua safe functions
3 *
4 * Copyright 2015-2016 Thierry Fournier <tfournier@arpalert.org>
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version
9 * 2 of the License, or (at your option) any later version.
10 *
11 *
12 * All the functions in this file runs with a Lua stack, and can
Thierry Fournierfb0b5462016-01-21 09:28:58 +010013 * return with a longjmp. All of these function must be launched
14 * in an environment able to catch a longjmp, otherwise a
15 * critical error can be raised.
16 */
Bertrand Jacquinf4c12d42021-01-21 21:14:07 +000017
18#define _GNU_SOURCE
19
Thierry Fournierfb0b5462016-01-21 09:28:58 +010020#include <lauxlib.h>
21#include <lua.h>
22#include <lualib.h>
23
Willy Tarreau83487a82020-06-04 20:19:54 +020024#include <haproxy/cli-t.h>
Willy Tarreau36979d92020-06-05 17:27:29 +020025#include <haproxy/errors.h>
Willy Tarreau86416052020-06-04 09:20:54 +020026#include <haproxy/hlua-t.h>
Willy Tarreaucd72d8c2020-06-02 19:11:26 +020027#include <haproxy/http.h>
Willy Tarreau6131d6a2020-06-02 16:48:09 +020028#include <haproxy/net_helper.h>
Willy Tarreaua264d962020-06-04 22:29:18 +020029#include <haproxy/pattern-t.h>
30#include <haproxy/proxy.h>
Willy Tarreau7cd8b6e2020-06-02 17:32:26 +020031#include <haproxy/regex.h>
Willy Tarreau1e56f922020-06-04 23:20:13 +020032#include <haproxy/server.h>
Willy Tarreau2eec9b52020-06-04 19:58:55 +020033#include <haproxy/stats.h>
Willy Tarreau872f2ea2020-06-04 18:46:44 +020034#include <haproxy/stick_table.h>
Willy Tarreaub2551052020-06-09 09:07:15 +020035#include <haproxy/time.h>
Christopher Faulet29e93262021-02-26 09:39:05 +010036#include <haproxy/tools.h>
Thierry Fournier94ed1c12016-02-24 08:06:32 +010037
Thierry Fournier1de16592016-01-27 09:49:07 +010038/* Contains the class reference of the concat object. */
39static int class_concat_ref;
Thierry Fournierf61aa632016-02-19 20:56:00 +010040static int class_proxy_ref;
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +010041static int class_server_ref;
Thierry Fournierff480422016-02-25 08:36:46 +010042static int class_listener_ref;
Thierry FOURNIER31904272017-10-25 12:59:51 +020043static int class_regex_ref;
Adis Nezirovic8878f8e2018-07-13 12:18:33 +020044static int class_stktable_ref;
Thierry Fournier1de16592016-01-27 09:49:07 +010045
Thierry Fournierf61aa632016-02-19 20:56:00 +010046#define STATS_LEN (MAX((int)ST_F_TOTAL_FIELDS, (int)INF_TOTAL_FIELDS))
Thierry Fourniereea77c02016-03-18 08:47:13 +010047
Thierry FOURNIERffbad792017-07-12 11:39:04 +020048static THREAD_LOCAL struct field stats[STATS_LEN];
Thierry Fourniereea77c02016-03-18 08:47:13 +010049
Thierry FOURNIER / OZON.IO7f3aa8b2016-11-24 20:37:38 +010050int hlua_checkboolean(lua_State *L, int index)
51{
52 if (!lua_isboolean(L, index))
53 luaL_argerror(L, index, "boolean expected");
54 return lua_toboolean(L, index);
55}
56
Adis Nezirovic8878f8e2018-07-13 12:18:33 +020057/* Helper to push unsigned integers to Lua stack, respecting Lua limitations */
58static int hlua_fcn_pushunsigned(lua_State *L, unsigned int val)
59{
60#if (LUA_MAXINTEGER == LLONG_MAX || ((LUA_MAXINTEGER == LONG_MAX) && (__WORDSIZE == 64)))
61 lua_pushinteger(L, val);
62#else
63 if (val > INT_MAX)
64 lua_pushnumber(L, (lua_Number)val);
65 else
66 lua_pushinteger(L, (int)val);
67#endif
68 return 1;
69}
70
71/* Helper to push unsigned long long to Lua stack, respecting Lua limitations */
72static int hlua_fcn_pushunsigned_ll(lua_State *L, unsigned long long val) {
73#if (LUA_MAXINTEGER == LLONG_MAX || ((LUA_MAXINTEGER == LONG_MAX) && (__WORDSIZE == 64)))
74 /* 64 bits case, U64 is supported until LLONG_MAX */
75 if (val > LLONG_MAX)
76 lua_pushnumber(L, (lua_Number)val);
77 else
78 lua_pushinteger(L, val);
79#else
80 /* 32 bits case, U64 is supported until INT_MAX */
81 if (val > INT_MAX)
82 lua_pushnumber(L, (lua_Number)val);
83 else
84 lua_pushinteger(L, (int)val);
85#endif
86 return 1;
87}
88
Joseph Herlantb3d92e32018-11-15 09:35:04 -080089/* This function gets a struct field and converts it in Lua
90 * variable. The variable is pushed at the top of the stack.
Thierry Fournier8b0d6e12016-03-16 18:29:13 +010091 */
92int hlua_fcn_pushfield(lua_State *L, struct field *field)
93{
94 /* The lua_Integer is always signed. Its length depends on
Joseph Herlantb3d92e32018-11-15 09:35:04 -080095 * compilation options, so the following code is conditioned
Thierry Fournier8b0d6e12016-03-16 18:29:13 +010096 * by some macros. Windows maros are not supported.
97 * If the number cannot be represented as integer, we try to
98 * convert to float.
99 */
100 switch (field_format(field, 0)) {
101
102 case FF_EMPTY:
103 lua_pushnil(L);
104 return 1;
105
106 case FF_S32:
107 /* S32 is always supported. */
108 lua_pushinteger(L, field->u.s32);
109 return 1;
110
111 case FF_U32:
112#if (LUA_MAXINTEGER == LLONG_MAX || ((LUA_MAXINTEGER == LONG_MAX) && (__WORDSIZE == 64)))
113 /* 64 bits case, U32 is always supported */
114 lua_pushinteger(L, field->u.u32);
115#else
116 /* 32 bits case, U32 is supported until INT_MAX. */
117 if (field->u.u32 > INT_MAX)
118 lua_pushnumber(L, (lua_Number)field->u.u32);
119 else
120 lua_pushinteger(L, field->u.u32);
121#endif
122 return 1;
123
124 case FF_S64:
125#if (LUA_MAXINTEGER == LLONG_MAX || ((LUA_MAXINTEGER == LONG_MAX) && (__WORDSIZE == 64)))
126 /* 64 bits case, S64 is always supported */
127 lua_pushinteger(L, field->u.s64);
128#else
Ilya Shipitsince7b00f2020-03-23 22:28:40 +0500129 /* 64 bits case, S64 is supported between INT_MIN and INT_MAX */
Thierry Fournier8b0d6e12016-03-16 18:29:13 +0100130 if (field->u.s64 < INT_MIN || field->u.s64 > INT_MAX)
131 lua_pushnumber(L, (lua_Number)field->u.s64);
132 else
133 lua_pushinteger(L, (int)field->u.s64);
134#endif
135 return 1;
136
137 case FF_U64:
138#if (LUA_MAXINTEGER == LLONG_MAX || ((LUA_MAXINTEGER == LONG_MAX) && (__WORDSIZE == 64)))
139 /* 64 bits case, U64 is supported until LLONG_MAX */
140 if (field->u.u64 > LLONG_MAX)
141 lua_pushnumber(L, (lua_Number)field->u.u64);
142 else
143 lua_pushinteger(L, field->u.u64);
144#else
145 /* 64 bits case, U64 is supported until INT_MAX */
146 if (field->u.u64 > INT_MAX)
147 lua_pushnumber(L, (lua_Number)field->u.u64);
148 else
149 lua_pushinteger(L, (int)field->u.u64);
150#endif
151 return 1;
152
153 case FF_STR:
154 lua_pushstring(L, field->u.str);
155 return 1;
156
157 default:
158 break;
159 }
160
161 /* Default case, never reached. */
162 lua_pushnil(L);
163 return 1;
164}
165
Thierry Fournier94ed1c12016-02-24 08:06:32 +0100166/* Some string are started or terminated by blank chars,
167 * this function removes the spaces, tabs, \r and
168 * \n at the begin and at the end of the string "str", and
169 * push the result in the lua stack.
170 * Returns a pointer to the Lua internal copy of the string.
171 */
172const char *hlua_pushstrippedstring(lua_State *L, const char *str)
173{
174 const char *p;
175 const char *e;
176
177 for (p = str; HTTP_IS_LWS(*p); p++);
178 for (e = p + strlen(p) - 1; e > p && HTTP_IS_LWS(*e); e--);
179
180 return lua_pushlstring(L, p, e - p);
181}
182
Thierry Fournierddd89882016-02-22 19:52:08 +0100183/* The three following functions are useful for adding entries
184 * in a table. These functions takes a string and respectively an
185 * integer, a string or a function and add it to the table in the
186 * top of the stack.
187 *
188 * These functions throws an error if no more stack size is
189 * available.
190 */
191void hlua_class_const_int(lua_State *L, const char *name, int value)
192{
Thierry Fournierddd89882016-02-22 19:52:08 +0100193 lua_pushstring(L, name);
194 lua_pushinteger(L, value);
195 lua_rawset(L, -3);
196}
197void hlua_class_const_str(lua_State *L, const char *name, const char *value)
198{
Thierry Fournierddd89882016-02-22 19:52:08 +0100199 lua_pushstring(L, name);
200 lua_pushstring(L, value);
201 lua_rawset(L, -3);
202}
203void hlua_class_function(lua_State *L, const char *name, int (*function)(lua_State *L))
204{
Thierry Fournierddd89882016-02-22 19:52:08 +0100205 lua_pushstring(L, name);
206 lua_pushcclosure(L, function, 0);
207 lua_rawset(L, -3);
208}
209
Joseph Herlantb3d92e32018-11-15 09:35:04 -0800210/* This function returns a string containing the HAProxy object name. */
Thierry Fournierddd89882016-02-22 19:52:08 +0100211int hlua_dump_object(struct lua_State *L)
212{
213 const char *name = (const char *)lua_tostring(L, lua_upvalueindex(1));
214 lua_pushfstring(L, "HAProxy class %s", name);
215 return 1;
216}
217
Thierry Fournier45e78d72016-02-19 18:34:46 +0100218/* This function register a table as metatable and. It names
219 * the metatable, and returns the associated reference.
Ilya Shipitsince7b00f2020-03-23 22:28:40 +0500220 * The original table is popped from the top of the stack.
Thierry Fournier45e78d72016-02-19 18:34:46 +0100221 * "name" is the referenced class name.
222 */
223int hlua_register_metatable(struct lua_State *L, char *name)
224{
225 /* Check the type of the top element. it must be
226 * a table.
227 */
228 if (lua_type(L, -1) != LUA_TTABLE)
229 luaL_error(L, "hlua_register_metatable() requires a type Table "
230 "in the top of the stack");
231
232 /* Add the __tostring function which identify the
233 * created object.
234 */
235 lua_pushstring(L, "__tostring");
236 lua_pushstring(L, name);
237 lua_pushcclosure(L, hlua_dump_object, 1);
238 lua_rawset(L, -3);
239
240 /* Register a named entry for the table. The table
Ilya Shipitsince7b00f2020-03-23 22:28:40 +0500241 * reference is copied first because the function
Thierry Fournier45e78d72016-02-19 18:34:46 +0100242 * lua_setfield() pop the entry.
243 */
244 lua_pushvalue(L, -1);
245 lua_setfield(L, LUA_REGISTRYINDEX, name);
246
247 /* Creates the reference of the object. The
248 * function luaL_ref pop the top of the stack.
249 */
250 return luaL_ref(L, LUA_REGISTRYINDEX);
251}
252
Thierry Fournier9e7e3ea2016-01-27 09:55:30 +0100253/* Return an object of the expected type, or throws an error. */
254void *hlua_checkudata(lua_State *L, int ud, int class_ref)
255{
256 void *p;
Thierry Fournier53518272016-01-27 10:34:09 +0100257 int ret;
Thierry Fournier9e7e3ea2016-01-27 09:55:30 +0100258
259 /* Check if the stack entry is an array. */
260 if (!lua_istable(L, ud))
Thierry Fournier53518272016-01-27 10:34:09 +0100261 luaL_argerror(L, ud, NULL);
262
263 /* pop the metatable of the referencecd object. */
264 if (!lua_getmetatable(L, ud))
265 luaL_argerror(L, ud, NULL);
266
267 /* pop the expected metatable. */
268 lua_rawgeti(L, LUA_REGISTRYINDEX, class_ref);
269
Thierry Fournier9e7e3ea2016-01-27 09:55:30 +0100270 /* Check if the metadata have the expected type. */
Thierry Fournier53518272016-01-27 10:34:09 +0100271 ret = lua_rawequal(L, -1, -2);
272 lua_pop(L, 2);
273 if (!ret)
274 luaL_argerror(L, ud, NULL);
275
Thierry Fournier9e7e3ea2016-01-27 09:55:30 +0100276 /* Push on the stack at the entry [0] of the table. */
277 lua_rawgeti(L, ud, 0);
Thierry Fournier53518272016-01-27 10:34:09 +0100278
Thierry Fournier9e7e3ea2016-01-27 09:55:30 +0100279 /* Check if this entry is userdata. */
280 p = lua_touserdata(L, -1);
281 if (!p)
Thierry Fournier53518272016-01-27 10:34:09 +0100282 luaL_argerror(L, ud, NULL);
283
Thierry Fournier9e7e3ea2016-01-27 09:55:30 +0100284 /* Remove the entry returned by lua_rawgeti(). */
285 lua_pop(L, 1);
Thierry Fournier53518272016-01-27 10:34:09 +0100286
Thierry Fournier9e7e3ea2016-01-27 09:55:30 +0100287 /* Return the associated struct. */
288 return p;
289}
290
Thierry Fournierb1f46562016-01-21 09:46:15 +0100291/* This function return the current date at epoch format in milliseconds. */
292int hlua_now(lua_State *L)
293{
294 lua_newtable(L);
295 lua_pushstring(L, "sec");
296 lua_pushinteger(L, now.tv_sec);
297 lua_rawset(L, -3);
298 lua_pushstring(L, "usec");
299 lua_pushinteger(L, now.tv_usec);
300 lua_rawset(L, -3);
301 return 1;
302}
303
Thierry Fournier1550d5d2016-01-21 09:35:41 +0100304/* This functions expects a Lua string as HTTP date, parse it and
305 * returns an integer containing the epoch format of the date, or
306 * nil if the parsing fails.
307 */
308static int hlua_parse_date(lua_State *L, int (*fcn)(const char *, int, struct tm*))
309{
310 const char *str;
311 size_t len;
312 struct tm tm;
313 time_t time;
314
315 str = luaL_checklstring(L, 1, &len);
316
317 if (!fcn(str, len, &tm)) {
318 lua_pushnil(L);
319 return 1;
320 }
321
322 /* This function considers the content of the broken-down time
323 * is exprimed in the UTC timezone. timegm don't care about
324 * the gnu variable tm_gmtoff. If gmtoff is set, or if you know
325 * the timezone from the broken-down time, it must be fixed
326 * after the conversion.
327 */
Willy Tarreauabd9bb22017-07-19 19:08:48 +0200328 time = my_timegm(&tm);
Thierry Fournier1550d5d2016-01-21 09:35:41 +0100329 if (time == -1) {
330 lua_pushnil(L);
331 return 1;
332 }
333
334 lua_pushinteger(L, (int)time);
335 return 1;
336}
337static int hlua_http_date(lua_State *L)
338{
339 return hlua_parse_date(L, parse_http_date);
340}
341static int hlua_imf_date(lua_State *L)
342{
343 return hlua_parse_date(L, parse_imf_date);
344}
345static int hlua_rfc850_date(lua_State *L)
346{
347 return hlua_parse_date(L, parse_rfc850_date);
348}
349static int hlua_asctime_date(lua_State *L)
350{
351 return hlua_parse_date(L, parse_asctime_date);
352}
353
Thierry Fourniereea77c02016-03-18 08:47:13 +0100354static int hlua_get_info(lua_State *L)
355{
356 int i;
357
358 stats_fill_info(stats, STATS_LEN);
359
360 lua_newtable(L);
361 for (i=0; i<INF_TOTAL_FIELDS; i++) {
Willy Tarreaueaa55372019-10-09 07:39:11 +0200362 lua_pushstring(L, info_fields[i].name);
Thierry Fourniereea77c02016-03-18 08:47:13 +0100363 hlua_fcn_pushfield(L, &stats[i]);
364 lua_settable(L, -3);
365 }
366 return 1;
367}
368
Thierry Fournier49d48422016-02-19 12:09:29 +0100369static struct hlua_concat *hlua_check_concat(lua_State *L, int ud)
Thierry Fournier1de16592016-01-27 09:49:07 +0100370{
Vincent Bernat3c2f2f22016-04-03 13:48:42 +0200371 return (hlua_checkudata(L, ud, class_concat_ref));
Thierry Fournier1de16592016-01-27 09:49:07 +0100372}
373
374static int hlua_concat_add(lua_State *L)
375{
Thierry Fournier49d48422016-02-19 12:09:29 +0100376 struct hlua_concat *b;
377 char *buffer;
378 char *new;
Thierry Fournier1de16592016-01-27 09:49:07 +0100379 const char *str;
380 size_t l;
381
382 /* First arg must be a concat object. */
383 b = hlua_check_concat(L, 1);
384
385 /* Second arg must be a string. */
386 str = luaL_checklstring(L, 2, &l);
387
Thierry Fournier49d48422016-02-19 12:09:29 +0100388 /* Get the buffer. */
389 lua_rawgeti(L, 1, 1);
390 buffer = lua_touserdata(L, -1);
391 lua_pop(L, 1);
392
393 /* Update the buffer size if it s required. The old buffer
394 * is crushed by the new in the object array, so it will
395 * be deleted by the GC.
396 * Note that in the first loop, the "new" variable is only
397 * used as a flag.
398 */
399 new = NULL;
400 while (b->size - b->len < l) {
401 b->size += HLUA_CONCAT_BLOCSZ;
402 new = buffer;
403 }
404 if (new) {
405 new = lua_newuserdata(L, b->size);
406 memcpy(new, buffer, b->len);
407 lua_rawseti(L, 1, 1);
408 buffer = new;
409 }
410
411 /* Copy string, and update metadata. */
412 memcpy(buffer + b->len, str, l);
413 b->len += l;
Thierry Fournier1de16592016-01-27 09:49:07 +0100414 return 0;
415}
416
417static int hlua_concat_dump(lua_State *L)
418{
Thierry Fournier49d48422016-02-19 12:09:29 +0100419 struct hlua_concat *b;
420 char *buffer;
Thierry Fournier1de16592016-01-27 09:49:07 +0100421
422 /* First arg must be a concat object. */
423 b = hlua_check_concat(L, 1);
424
Thierry Fournier49d48422016-02-19 12:09:29 +0100425 /* Get the buffer. */
426 lua_rawgeti(L, 1, 1);
427 buffer = lua_touserdata(L, -1);
428 lua_pop(L, 1);
429
Ilya Shipitsince7b00f2020-03-23 22:28:40 +0500430 /* Push the soncatenated string in the stack. */
Thierry Fournier49d48422016-02-19 12:09:29 +0100431 lua_pushlstring(L, buffer, b->len);
Thierry Fournier1de16592016-01-27 09:49:07 +0100432 return 1;
433}
434
435int hlua_concat_new(lua_State *L)
436{
Thierry Fournier49d48422016-02-19 12:09:29 +0100437 struct hlua_concat *b;
Thierry Fournier1de16592016-01-27 09:49:07 +0100438
439 lua_newtable(L);
Vincent Bernat3c2f2f22016-04-03 13:48:42 +0200440 b = lua_newuserdata(L, sizeof(*b));
Thierry Fournier49d48422016-02-19 12:09:29 +0100441 b->size = HLUA_CONCAT_BLOCSZ;
442 b->len = 0;
Thierry Fournier1de16592016-01-27 09:49:07 +0100443 lua_rawseti(L, -2, 0);
Thierry Fournier49d48422016-02-19 12:09:29 +0100444 lua_newuserdata(L, HLUA_CONCAT_BLOCSZ);
445 lua_rawseti(L, -2, 1);
Thierry Fournier1de16592016-01-27 09:49:07 +0100446
447 lua_rawgeti(L, LUA_REGISTRYINDEX, class_concat_ref);
448 lua_setmetatable(L, -2);
449
Thierry Fournier1de16592016-01-27 09:49:07 +0100450 return 1;
451}
452
453static int concat_tostring(lua_State *L)
454{
455 const void *ptr = lua_topointer(L, 1);
456 lua_pushfstring(L, "Concat object: %p", ptr);
457 return 1;
458}
459
460static int hlua_concat_init(lua_State *L)
461{
462 /* Creates the buffered concat object. */
463 lua_newtable(L);
464
465 lua_pushstring(L, "__tostring");
466 lua_pushcclosure(L, concat_tostring, 0);
467 lua_settable(L, -3);
468
469 lua_pushstring(L, "__index"); /* Creates the index entry. */
470 lua_newtable(L); /* The "__index" content. */
471
472 lua_pushstring(L, "add");
473 lua_pushcclosure(L, hlua_concat_add, 0);
474 lua_settable(L, -3);
475
476 lua_pushstring(L, "dump");
477 lua_pushcclosure(L, hlua_concat_dump, 0);
478 lua_settable(L, -3);
479
480 lua_settable(L, -3); /* Sets the __index entry. */
481 class_concat_ref = luaL_ref(L, LUA_REGISTRYINDEX);
482
483 return 1;
484}
485
Adis Nezirovic8878f8e2018-07-13 12:18:33 +0200486int hlua_fcn_new_stktable(lua_State *L, struct stktable *tbl)
487{
488 lua_newtable(L);
489
490 /* Pop a class stktbl metatable and affect it to the userdata. */
491 lua_rawgeti(L, LUA_REGISTRYINDEX, class_stktable_ref);
492 lua_setmetatable(L, -2);
493
494 lua_pushlightuserdata(L, tbl);
495 lua_rawseti(L, -2, 0);
496 return 1;
497}
498
499static struct stktable *hlua_check_stktable(lua_State *L, int ud)
500{
501 return hlua_checkudata(L, ud, class_stktable_ref);
502}
503
504/* Extract stick table attributes into Lua table */
505int hlua_stktable_info(lua_State *L)
506{
507 struct stktable *tbl;
508 int dt;
509
510 tbl = hlua_check_stktable(L, 1);
511
512 if (!tbl->id) {
513 lua_pushnil(L);
514 return 1;
515 }
516
517 lua_newtable(L);
518
519 lua_pushstring(L, "type");
520 lua_pushstring(L, stktable_types[tbl->type].kw);
521 lua_settable(L, -3);
522
523 lua_pushstring(L, "length");
524 lua_pushinteger(L, tbl->key_size);
525 lua_settable(L, -3);
526
527 lua_pushstring(L, "size");
528 hlua_fcn_pushunsigned(L, tbl->size);
529 lua_settable(L, -3);
530
531 lua_pushstring(L, "used");
532 hlua_fcn_pushunsigned(L, tbl->current);
533 lua_settable(L, -3);
534
535 lua_pushstring(L, "nopurge");
536 lua_pushboolean(L, tbl->nopurge > 0);
537 lua_settable(L, -3);
538
539 lua_pushstring(L, "expire");
540 lua_pushinteger(L, tbl->expire);
541 lua_settable(L, -3);
542
543 /* Save data types periods (if applicable) in 'data' table */
544 lua_pushstring(L, "data");
545 lua_newtable(L);
546
547 for (dt = 0; dt < STKTABLE_DATA_TYPES; dt++) {
548 if (tbl->data_ofs[dt] == 0)
549 continue;
550
551 lua_pushstring(L, stktable_data_types[dt].name);
552
553 if (stktable_data_types[dt].arg_type == ARG_T_DELAY)
554 lua_pushinteger(L, tbl->data_arg[dt].u);
555 else
556 lua_pushinteger(L, -1);
557
558 lua_settable(L, -3);
559 }
560
561 lua_settable(L, -3);
562
563 return 1;
564}
565
566/* Helper to get extract stick table entry into Lua table */
567static void hlua_stktable_entry(lua_State *L, struct stktable *t, struct stksess *ts)
568{
569 int dt;
570 void *ptr;
571
572 for (dt = 0; dt < STKTABLE_DATA_TYPES; dt++) {
573
574 if (t->data_ofs[dt] == 0)
575 continue;
576
577 lua_pushstring(L, stktable_data_types[dt].name);
578
579 ptr = stktable_data_ptr(t, ts, dt);
580 switch (stktable_data_types[dt].std_type) {
581 case STD_T_SINT:
582 lua_pushinteger(L, stktable_data_cast(ptr, std_t_sint));
583 break;
584 case STD_T_UINT:
585 hlua_fcn_pushunsigned(L, stktable_data_cast(ptr, std_t_uint));
586 break;
587 case STD_T_ULL:
588 hlua_fcn_pushunsigned_ll(L, stktable_data_cast(ptr, std_t_ull));
589 break;
590 case STD_T_FRQP:
591 lua_pushinteger(L, read_freq_ctr_period(&stktable_data_cast(ptr, std_t_frqp),
592 t->data_arg[dt].u));
593 break;
Adis Nezirovicad9f9ed2020-05-05 13:57:28 +0200594 case STD_T_DICT: {
595 struct dict_entry *de;
596 de = stktable_data_cast(ptr, std_t_dict);
597 lua_pushstring(L, de ? (char *)de->value.key : "-");
598 break;
599 }
Adis Nezirovic8878f8e2018-07-13 12:18:33 +0200600 }
601
602 lua_settable(L, -3);
603 }
604}
605
606/* Looks in table <t> for a sticky session matching key <key>
607 * Returns table with session data or nil
608 *
609 * The returned table always contains 'use' and 'expire' (integer) fields.
610 * For frequency/rate counters, each data entry is returned as table with
611 * 'value' and 'period' fields.
612 */
613int hlua_stktable_lookup(lua_State *L)
614{
615 struct stktable *t;
616 struct sample smp;
617 struct stktable_key *skey;
618 struct stksess *ts;
619
620 t = hlua_check_stktable(L, 1);
621 smp.data.type = SMP_T_STR;
622 smp.flags = SMP_F_CONST;
Nathan Neulinger31a841c2020-03-03 20:32:47 -0600623 smp.data.u.str.area = (char *)lua_tolstring(L, 2, &smp.data.u.str.data);
Adis Nezirovic8878f8e2018-07-13 12:18:33 +0200624
625 skey = smp_to_stkey(&smp, t);
626 if (!skey) {
627 lua_pushnil(L);
628 return 1;
629 }
630
631 ts = stktable_lookup_key(t, skey);
632 if (!ts) {
633 lua_pushnil(L);
634 return 1;
635 }
636
637 lua_newtable(L);
638 lua_pushstring(L, "use");
639 lua_pushinteger(L, ts->ref_cnt - 1);
640 lua_settable(L, -3);
641
642 lua_pushstring(L, "expire");
643 lua_pushinteger(L, tick_remain(now_ms, ts->expire));
644 lua_settable(L, -3);
645
646 hlua_stktable_entry(L, t, ts);
647 HA_SPIN_LOCK(STK_TABLE_LOCK, &t->lock);
648 ts->ref_cnt--;
649 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &t->lock);
650
651 return 1;
652}
653
654struct stk_filter {
655 long long val;
656 int type;
657 int op;
658};
659
660
661/* Helper for returning errors to callers using Lua convention (nil, err) */
662static int hlua_error(lua_State *L, const char *fmt, ...) {
663 char buf[256];
664 int len;
665 va_list args;
666 va_start(args, fmt);
667 len = vsnprintf(buf, sizeof(buf), fmt, args);
668 va_end(args);
669
670 if (len < 0) {
671 ha_alert("hlua_error(): Could not write error message.\n");
672 lua_pushnil(L);
673 return 1;
674 } else if (len >= sizeof(buf))
675 ha_alert("hlua_error(): Error message was truncated.\n");
676
677 lua_pushnil(L);
678 lua_pushstring(L, buf);
679
680 return 2;
681}
682
683/* Dump the contents of stick table <t>*/
684int hlua_stktable_dump(lua_State *L)
685{
686 struct stktable *t;
687 struct ebmb_node *eb;
688 struct ebmb_node *n;
689 struct stksess *ts;
690 int type;
691 int op;
692 int dt;
693 long long val;
Adis Nezirovic1a693fc2020-01-16 15:19:29 +0100694 struct stk_filter filter[STKTABLE_FILTER_LEN];
Adis Nezirovic8878f8e2018-07-13 12:18:33 +0200695 int filter_count = 0;
696 int i;
697 int skip_entry;
698 void *ptr;
699
700 t = hlua_check_stktable(L, 1);
701 type = lua_type(L, 2);
702
703 switch (type) {
704 case LUA_TNONE:
705 case LUA_TNIL:
706 break;
707 case LUA_TTABLE:
708 lua_pushnil(L);
709 while (lua_next(L, 2) != 0) {
710 int entry_idx = 0;
711
Adis Nezirovic1a693fc2020-01-16 15:19:29 +0100712 if (filter_count >= STKTABLE_FILTER_LEN)
713 return hlua_error(L, "Filter table too large (len > %d)", STKTABLE_FILTER_LEN);
Adis Nezirovic8878f8e2018-07-13 12:18:33 +0200714
715 if (lua_type(L, -1) != LUA_TTABLE || lua_rawlen(L, -1) != 3)
716 return hlua_error(L, "Filter table entry must be a triplet: {\"data_col\", \"op\", val} (entry #%d)", filter_count + 1);
717
718 lua_pushnil(L);
719 while (lua_next(L, -2) != 0) {
720 switch (entry_idx) {
721 case 0:
722 if (lua_type(L, -1) != LUA_TSTRING)
723 return hlua_error(L, "Filter table data column must be string (entry #%d)", filter_count + 1);
724
725 dt = stktable_get_data_type((char *)lua_tostring(L, -1));
726 if (dt < 0 || t->data_ofs[dt] == 0)
727 return hlua_error(L, "Filter table data column not present in stick table (entry #%d)", filter_count + 1);
728 filter[filter_count].type = dt;
729 break;
730 case 1:
731 if (lua_type(L, -1) != LUA_TSTRING)
732 return hlua_error(L, "Filter table operator must be string (entry #%d)", filter_count + 1);
733
734 op = get_std_op(lua_tostring(L, -1));
735 if (op < 0)
736 return hlua_error(L, "Unknown operator in filter table (entry #%d)", filter_count + 1);
737 filter[filter_count].op = op;
738 break;
739 case 2:
740 val = lua_tointeger(L, -1);
741 filter[filter_count].val = val;
742 filter_count++;
743 break;
744 default:
745 break;
746 }
747 entry_idx++;
748 lua_pop(L, 1);
749 }
750 lua_pop(L, 1);
751 }
752 break;
753 default:
754 return hlua_error(L, "filter table expected");
755 }
756
757 lua_newtable(L);
758
759 HA_SPIN_LOCK(STK_TABLE_LOCK, &t->lock);
760 eb = ebmb_first(&t->keys);
761 for (n = eb; n; n = ebmb_next(n)) {
762 ts = ebmb_entry(n, struct stksess, key);
763 if (!ts) {
764 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &t->lock);
765 return 1;
766 }
767 ts->ref_cnt++;
768 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &t->lock);
769
770 /* multi condition/value filter */
771 skip_entry = 0;
772 for (i = 0; i < filter_count; i++) {
773 if (t->data_ofs[filter[i].type] == 0)
774 continue;
775
776 ptr = stktable_data_ptr(t, ts, filter[i].type);
777
778 switch (stktable_data_types[filter[i].type].std_type) {
779 case STD_T_SINT:
780 val = stktable_data_cast(ptr, std_t_sint);
781 break;
782 case STD_T_UINT:
783 val = stktable_data_cast(ptr, std_t_uint);
784 break;
785 case STD_T_ULL:
786 val = stktable_data_cast(ptr, std_t_ull);
787 break;
788 case STD_T_FRQP:
789 val = read_freq_ctr_period(&stktable_data_cast(ptr, std_t_frqp),
790 t->data_arg[filter[i].type].u);
791 break;
792 default:
793 continue;
794 break;
795 }
796
797 op = filter[i].op;
798
799 if ((val < filter[i].val && (op == STD_OP_EQ || op == STD_OP_GT || op == STD_OP_GE)) ||
800 (val == filter[i].val && (op == STD_OP_NE || op == STD_OP_GT || op == STD_OP_LT)) ||
801 (val > filter[i].val && (op == STD_OP_EQ || op == STD_OP_LT || op == STD_OP_LE))) {
802 skip_entry = 1;
803 break;
804 }
805 }
806
807 if (skip_entry) {
808 HA_SPIN_LOCK(STK_TABLE_LOCK, &t->lock);
809 ts->ref_cnt--;
810 continue;
811 }
812
813 if (t->type == SMP_T_IPV4) {
814 char addr[INET_ADDRSTRLEN];
815 inet_ntop(AF_INET, (const void *)&ts->key.key, addr, sizeof(addr));
816 lua_pushstring(L, addr);
817 } else if (t->type == SMP_T_IPV6) {
818 char addr[INET6_ADDRSTRLEN];
819 inet_ntop(AF_INET6, (const void *)&ts->key.key, addr, sizeof(addr));
820 lua_pushstring(L, addr);
821 } else if (t->type == SMP_T_SINT) {
822 lua_pushinteger(L, *ts->key.key);
823 } else if (t->type == SMP_T_STR) {
824 lua_pushstring(L, (const char *)ts->key.key);
825 } else {
826 return hlua_error(L, "Unsupported stick table key type");
827 }
828
829 lua_newtable(L);
830 hlua_stktable_entry(L, t, ts);
831 lua_settable(L, -3);
832 HA_SPIN_LOCK(STK_TABLE_LOCK, &t->lock);
833 ts->ref_cnt--;
834 }
835 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &t->lock);
836
837 return 1;
838}
839
Thierry Fournierff480422016-02-25 08:36:46 +0100840int hlua_fcn_new_listener(lua_State *L, struct listener *lst)
841{
842 lua_newtable(L);
843
844 /* Pop a class sesison metatable and affect it to the userdata. */
845 lua_rawgeti(L, LUA_REGISTRYINDEX, class_listener_ref);
846 lua_setmetatable(L, -2);
847
848 lua_pushlightuserdata(L, lst);
849 lua_rawseti(L, -2, 0);
850 return 1;
851}
852
853static struct listener *hlua_check_listener(lua_State *L, int ud)
854{
Vincent Bernat3c2f2f22016-04-03 13:48:42 +0200855 return hlua_checkudata(L, ud, class_listener_ref);
Thierry Fournierff480422016-02-25 08:36:46 +0100856}
857
858int hlua_listener_get_stats(lua_State *L)
859{
860 struct listener *li;
861 int i;
862
863 li = hlua_check_listener(L, 1);
864
Willy Tarreauc95bad52016-12-22 00:13:31 +0100865 if (!li->bind_conf->frontend) {
Thierry Fournierff480422016-02-25 08:36:46 +0100866 lua_pushnil(L);
867 return 1;
868 }
869
William Dauchy655e14e2021-02-14 23:22:54 +0100870 stats_fill_li_stats(li->bind_conf->frontend, li, STAT_SHLGNDS, stats,
871 STATS_LEN, NULL);
Thierry Fournierff480422016-02-25 08:36:46 +0100872
873 lua_newtable(L);
874 for (i=0; i<ST_F_TOTAL_FIELDS; i++) {
Willy Tarreaueaa55372019-10-09 07:39:11 +0200875 lua_pushstring(L, stat_fields[i].name);
Thierry Fournierff480422016-02-25 08:36:46 +0100876 hlua_fcn_pushfield(L, &stats[i]);
877 lua_settable(L, -3);
878 }
879 return 1;
880
881}
882
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +0100883int hlua_fcn_new_server(lua_State *L, struct server *srv)
884{
Patrick Hemmera62ae7e2018-04-29 14:23:48 -0400885 char buffer[12];
886
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +0100887 lua_newtable(L);
888
889 /* Pop a class sesison metatable and affect it to the userdata. */
890 lua_rawgeti(L, LUA_REGISTRYINDEX, class_server_ref);
891 lua_setmetatable(L, -2);
892
893 lua_pushlightuserdata(L, srv);
894 lua_rawseti(L, -2, 0);
Patrick Hemmera62ae7e2018-04-29 14:23:48 -0400895
896 /* Add server name. */
897 lua_pushstring(L, "name");
898 lua_pushstring(L, srv->id);
899 lua_settable(L, -3);
900
901 /* Add server puid. */
902 lua_pushstring(L, "puid");
903 snprintf(buffer, sizeof(buffer), "%d", srv->puid);
904 lua_pushstring(L, buffer);
905 lua_settable(L, -3);
906
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +0100907 return 1;
908}
909
910static struct server *hlua_check_server(lua_State *L, int ud)
911{
Vincent Bernat3c2f2f22016-04-03 13:48:42 +0200912 return hlua_checkudata(L, ud, class_server_ref);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +0100913}
914
915int hlua_server_get_stats(lua_State *L)
916{
917 struct server *srv;
918 int i;
919
920 srv = hlua_check_server(L, 1);
921
922 if (!srv->proxy) {
923 lua_pushnil(L);
924 return 1;
925 }
926
William Dauchyd3a9a492021-01-25 17:29:03 +0100927 stats_fill_sv_stats(srv->proxy, srv, STAT_SHLGNDS, stats,
928 STATS_LEN, NULL);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +0100929
930 lua_newtable(L);
931 for (i=0; i<ST_F_TOTAL_FIELDS; i++) {
Willy Tarreaueaa55372019-10-09 07:39:11 +0200932 lua_pushstring(L, stat_fields[i].name);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +0100933 hlua_fcn_pushfield(L, &stats[i]);
934 lua_settable(L, -3);
935 }
936 return 1;
937
938}
939
940int hlua_server_get_addr(lua_State *L)
941{
942 struct server *srv;
943 char addr[INET6_ADDRSTRLEN];
944 luaL_Buffer b;
945
946 srv = hlua_check_server(L, 1);
947
948 luaL_buffinit(L, &b);
949
950 switch (srv->addr.ss_family) {
951 case AF_INET:
952 inet_ntop(AF_INET, &((struct sockaddr_in *)&srv->addr)->sin_addr,
953 addr, INET_ADDRSTRLEN);
954 luaL_addstring(&b, addr);
955 luaL_addstring(&b, ":");
Nenad Merdanovic38494732017-07-23 22:04:58 -0400956 snprintf(addr, INET_ADDRSTRLEN, "%d", srv->svc_port);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +0100957 luaL_addstring(&b, addr);
958 break;
959 case AF_INET6:
960 inet_ntop(AF_INET6, &((struct sockaddr_in6 *)&srv->addr)->sin6_addr,
Nenad Merdanovica9f04042017-07-23 22:04:59 -0400961 addr, INET6_ADDRSTRLEN);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +0100962 luaL_addstring(&b, addr);
963 luaL_addstring(&b, ":");
Nenad Merdanovic38494732017-07-23 22:04:58 -0400964 snprintf(addr, INET_ADDRSTRLEN, "%d", srv->svc_port);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +0100965 luaL_addstring(&b, addr);
966 break;
967 case AF_UNIX:
968 luaL_addstring(&b, (char *)((struct sockaddr_un *)&srv->addr)->sun_path);
969 break;
970 default:
971 luaL_addstring(&b, "<unknown>");
972 break;
973 }
974
975 luaL_pushresult(&b);
976 return 1;
977}
978
979int hlua_server_is_draining(lua_State *L)
980{
981 struct server *srv;
982
983 srv = hlua_check_server(L, 1);
984 lua_pushinteger(L, server_is_draining(srv));
985 return 1;
986}
987
Patrick Hemmer32d539f2018-04-29 14:25:46 -0400988int hlua_server_set_maxconn(lua_State *L)
989{
990 struct server *srv;
991 const char *maxconn;
992 const char *err;
993
994 srv = hlua_check_server(L, 1);
995 maxconn = luaL_checkstring(L, 2);
996
997 HA_SPIN_LOCK(SERVER_LOCK, &srv->lock);
998 err = server_parse_maxconn_change_request(srv, maxconn);
999 HA_SPIN_UNLOCK(SERVER_LOCK, &srv->lock);
1000 if (!err)
1001 lua_pushnil(L);
1002 else
1003 hlua_pushstrippedstring(L, err);
1004 return 1;
1005}
1006
1007int hlua_server_get_maxconn(lua_State *L)
1008{
1009 struct server *srv;
1010
1011 srv = hlua_check_server(L, 1);
1012 lua_pushinteger(L, srv->maxconn);
1013 return 1;
1014}
1015
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001016int hlua_server_set_weight(lua_State *L)
1017{
1018 struct server *srv;
1019 const char *weight;
1020 const char *err;
1021
1022 srv = hlua_check_server(L, 1);
1023 weight = luaL_checkstring(L, 2);
1024
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001025 HA_SPIN_LOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001026 err = server_parse_weight_change_request(srv, weight);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001027 HA_SPIN_UNLOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001028 if (!err)
1029 lua_pushnil(L);
1030 else
1031 hlua_pushstrippedstring(L, err);
1032 return 1;
1033}
1034
1035int hlua_server_get_weight(lua_State *L)
1036{
1037 struct server *srv;
1038
1039 srv = hlua_check_server(L, 1);
1040 lua_pushinteger(L, srv->uweight);
1041 return 1;
1042}
1043
1044int hlua_server_set_addr(lua_State *L)
1045{
1046 struct server *srv;
1047 const char *addr;
Joseph C. Sible49bbf522020-05-04 22:20:32 -04001048 const char *port;
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001049 const char *err;
1050
1051 srv = hlua_check_server(L, 1);
1052 addr = luaL_checkstring(L, 2);
Joseph C. Sible49bbf522020-05-04 22:20:32 -04001053 if (lua_gettop(L) >= 3)
1054 port = luaL_checkstring(L, 3);
1055 else
1056 port = NULL;
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001057
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001058 HA_SPIN_LOCK(SERVER_LOCK, &srv->lock);
Christopher Faulet69beaa92021-02-16 12:07:47 +01001059 err = srv_update_addr_port(srv, addr, port, "Lua script");
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001060 HA_SPIN_UNLOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001061 if (!err)
1062 lua_pushnil(L);
1063 else
1064 hlua_pushstrippedstring(L, err);
1065 return 1;
1066}
1067
1068int hlua_server_shut_sess(lua_State *L)
1069{
1070 struct server *srv;
1071
1072 srv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001073 HA_SPIN_LOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001074 srv_shutdown_streams(srv, SF_ERR_KILLED);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001075 HA_SPIN_UNLOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001076 return 0;
1077}
1078
1079int hlua_server_set_drain(lua_State *L)
1080{
1081 struct server *srv;
1082
1083 srv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001084 HA_SPIN_LOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001085 srv_adm_set_drain(srv);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001086 HA_SPIN_UNLOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001087 return 0;
1088}
1089
1090int hlua_server_set_maint(lua_State *L)
1091{
1092 struct server *srv;
1093
1094 srv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001095 HA_SPIN_LOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001096 srv_adm_set_maint(srv);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001097 HA_SPIN_UNLOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001098 return 0;
1099}
1100
1101int hlua_server_set_ready(lua_State *L)
1102{
1103 struct server *srv;
1104
1105 srv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001106 HA_SPIN_LOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001107 srv_adm_set_ready(srv);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001108 HA_SPIN_UNLOCK(SERVER_LOCK, &srv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001109 return 0;
1110}
1111
1112int hlua_server_check_enable(lua_State *L)
1113{
1114 struct server *sv;
1115
1116 sv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001117 HA_SPIN_LOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001118 if (sv->check.state & CHK_ST_CONFIGURED) {
Adis Nezirovicceee9332017-07-26 09:19:06 +02001119 sv->check.state |= CHK_ST_ENABLED;
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001120 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001121 HA_SPIN_UNLOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001122 return 0;
1123}
1124
1125int hlua_server_check_disable(lua_State *L)
1126{
1127 struct server *sv;
1128
1129 sv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001130 HA_SPIN_LOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001131 if (sv->check.state & CHK_ST_CONFIGURED) {
Adis Nezirovicceee9332017-07-26 09:19:06 +02001132 sv->check.state &= ~CHK_ST_ENABLED;
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001133 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001134 HA_SPIN_UNLOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001135 return 0;
1136}
1137
1138int hlua_server_check_force_up(lua_State *L)
1139{
1140 struct server *sv;
1141
1142 sv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001143 HA_SPIN_LOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001144 if (!(sv->track)) {
1145 sv->check.health = sv->check.rise + sv->check.fall - 1;
Emeric Brun5a133512017-10-19 14:42:30 +02001146 srv_set_running(sv, "changed from Lua script", NULL);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001147 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001148 HA_SPIN_UNLOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001149 return 0;
1150}
1151
1152int hlua_server_check_force_nolb(lua_State *L)
1153{
1154 struct server *sv;
1155
1156 sv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001157 HA_SPIN_LOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001158 if (!(sv->track)) {
1159 sv->check.health = sv->check.rise + sv->check.fall - 1;
Emeric Brun5a133512017-10-19 14:42:30 +02001160 srv_set_stopping(sv, "changed from Lua script", NULL);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001161 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001162 HA_SPIN_UNLOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001163 return 0;
1164}
1165
1166int hlua_server_check_force_down(lua_State *L)
1167{
1168 struct server *sv;
1169
1170 sv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001171 HA_SPIN_LOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001172 if (!(sv->track)) {
1173 sv->check.health = 0;
Emeric Brun5a133512017-10-19 14:42:30 +02001174 srv_set_stopped(sv, "changed from Lua script", NULL);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001175 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001176 HA_SPIN_UNLOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001177 return 0;
1178}
1179
1180int hlua_server_agent_enable(lua_State *L)
1181{
1182 struct server *sv;
1183
1184 sv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001185 HA_SPIN_LOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001186 if (sv->agent.state & CHK_ST_CONFIGURED) {
1187 sv->agent.state |= CHK_ST_ENABLED;
1188 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001189 HA_SPIN_UNLOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001190 return 0;
1191}
1192
1193int hlua_server_agent_disable(lua_State *L)
1194{
1195 struct server *sv;
1196
1197 sv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001198 HA_SPIN_LOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001199 if (sv->agent.state & CHK_ST_CONFIGURED) {
1200 sv->agent.state &= ~CHK_ST_ENABLED;
1201 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001202 HA_SPIN_UNLOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001203 return 0;
1204}
1205
1206int hlua_server_agent_force_up(lua_State *L)
1207{
1208 struct server *sv;
1209
1210 sv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001211 HA_SPIN_LOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001212 if (sv->agent.state & CHK_ST_ENABLED) {
1213 sv->agent.health = sv->agent.rise + sv->agent.fall - 1;
Emeric Brun5a133512017-10-19 14:42:30 +02001214 srv_set_running(sv, "changed from Lua script", NULL);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001215 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001216 HA_SPIN_UNLOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001217 return 0;
1218}
1219
1220int hlua_server_agent_force_down(lua_State *L)
1221{
1222 struct server *sv;
1223
1224 sv = hlua_check_server(L, 1);
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001225 HA_SPIN_LOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001226 if (sv->agent.state & CHK_ST_ENABLED) {
1227 sv->agent.health = 0;
Emeric Brun5a133512017-10-19 14:42:30 +02001228 srv_set_stopped(sv, "changed from Lua script", NULL);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001229 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001230 HA_SPIN_UNLOCK(SERVER_LOCK, &sv->lock);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001231 return 0;
1232}
1233
Thierry Fournierf61aa632016-02-19 20:56:00 +01001234int hlua_fcn_new_proxy(lua_State *L, struct proxy *px)
1235{
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001236 struct server *srv;
Thierry Fournierff480422016-02-25 08:36:46 +01001237 struct listener *lst;
1238 int lid;
Willy Tarreau29d69802018-05-06 14:50:09 +02001239 char buffer[17];
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001240
Thierry Fournierf61aa632016-02-19 20:56:00 +01001241 lua_newtable(L);
1242
1243 /* Pop a class sesison metatable and affect it to the userdata. */
1244 lua_rawgeti(L, LUA_REGISTRYINDEX, class_proxy_ref);
1245 lua_setmetatable(L, -2);
1246
1247 lua_pushlightuserdata(L, px);
1248 lua_rawseti(L, -2, 0);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001249
Thierry FOURNIERf2bbe382017-07-24 13:59:22 +02001250 /* Add proxy name. */
1251 lua_pushstring(L, "name");
1252 lua_pushstring(L, px->id);
1253 lua_settable(L, -3);
1254
Baptiste Assmann46c72552017-10-26 21:51:58 +02001255 /* Add proxy uuid. */
1256 lua_pushstring(L, "uuid");
1257 snprintf(buffer, sizeof(buffer), "%d", px->uuid);
1258 lua_pushstring(L, buffer);
1259 lua_settable(L, -3);
1260
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001261 /* Browse and register servers. */
1262 lua_pushstring(L, "servers");
1263 lua_newtable(L);
1264 for (srv = px->srv; srv; srv = srv->next) {
1265 lua_pushstring(L, srv->id);
1266 hlua_fcn_new_server(L, srv);
1267 lua_settable(L, -3);
1268 }
1269 lua_settable(L, -3);
1270
Thierry Fournierff480422016-02-25 08:36:46 +01001271 /* Browse and register listeners. */
1272 lua_pushstring(L, "listeners");
1273 lua_newtable(L);
1274 lid = 1;
1275 list_for_each_entry(lst, &px->conf.listeners, by_fe) {
1276 if (lst->name)
1277 lua_pushstring(L, lst->name);
1278 else {
Willy Tarreau29d69802018-05-06 14:50:09 +02001279 snprintf(buffer, sizeof(buffer), "sock-%d", lid);
Thierry Fournierff480422016-02-25 08:36:46 +01001280 lid++;
1281 lua_pushstring(L, buffer);
1282 }
1283 hlua_fcn_new_listener(L, lst);
1284 lua_settable(L, -3);
1285 }
1286 lua_settable(L, -3);
1287
Frédéric Lécaille1b8e68e2019-03-14 07:07:41 +01001288 if (px->table && px->table->id) {
Adis Nezirovic8878f8e2018-07-13 12:18:33 +02001289 lua_pushstring(L, "stktable");
Frédéric Lécaille1b8e68e2019-03-14 07:07:41 +01001290 hlua_fcn_new_stktable(L, px->table);
Adis Nezirovic8878f8e2018-07-13 12:18:33 +02001291 lua_settable(L, -3);
1292 }
1293
Thierry Fournierf61aa632016-02-19 20:56:00 +01001294 return 1;
1295}
1296
1297static struct proxy *hlua_check_proxy(lua_State *L, int ud)
1298{
Vincent Bernat3c2f2f22016-04-03 13:48:42 +02001299 return hlua_checkudata(L, ud, class_proxy_ref);
Thierry Fournierf61aa632016-02-19 20:56:00 +01001300}
1301
1302int hlua_proxy_pause(lua_State *L)
1303{
1304 struct proxy *px;
1305
1306 px = hlua_check_proxy(L, 1);
1307 pause_proxy(px);
1308 return 0;
1309}
1310
1311int hlua_proxy_resume(lua_State *L)
1312{
1313 struct proxy *px;
1314
1315 px = hlua_check_proxy(L, 1);
1316 resume_proxy(px);
1317 return 0;
1318}
1319
1320int hlua_proxy_stop(lua_State *L)
1321{
1322 struct proxy *px;
1323
1324 px = hlua_check_proxy(L, 1);
1325 stop_proxy(px);
1326 return 0;
1327}
1328
1329int hlua_proxy_get_cap(lua_State *L)
1330{
1331 struct proxy *px;
1332 const char *str;
1333
1334 px = hlua_check_proxy(L, 1);
1335 str = proxy_cap_str(px->cap);
1336 lua_pushstring(L, str);
1337 return 1;
1338}
1339
1340int hlua_proxy_get_stats(lua_State *L)
1341{
1342 struct proxy *px;
1343 int i;
1344
1345 px = hlua_check_proxy(L, 1);
1346 if (px->cap & PR_CAP_BE)
William Dauchyda3b4662021-01-25 17:29:01 +01001347 stats_fill_be_stats(px, STAT_SHLGNDS, stats, STATS_LEN, NULL);
Thierry Fournierf61aa632016-02-19 20:56:00 +01001348 else
William Dauchy0ef54392021-01-17 18:27:45 +01001349 stats_fill_fe_stats(px, stats, STATS_LEN, NULL);
Thierry Fournierf61aa632016-02-19 20:56:00 +01001350 lua_newtable(L);
1351 for (i=0; i<ST_F_TOTAL_FIELDS; i++) {
Willy Tarreaueaa55372019-10-09 07:39:11 +02001352 lua_pushstring(L, stat_fields[i].name);
Thierry Fournierf61aa632016-02-19 20:56:00 +01001353 hlua_fcn_pushfield(L, &stats[i]);
1354 lua_settable(L, -3);
1355 }
1356 return 1;
1357}
1358
1359int hlua_proxy_get_mode(lua_State *L)
1360{
1361 struct proxy *px;
1362 const char *str;
1363
1364 px = hlua_check_proxy(L, 1);
1365 str = proxy_mode_str(px->mode);
1366 lua_pushstring(L, str);
1367 return 1;
1368}
1369
1370int hlua_proxy_shut_bcksess(lua_State *L)
1371{
1372 struct proxy *px;
1373
1374 px = hlua_check_proxy(L, 1);
1375 srv_shutdown_backup_streams(px, SF_ERR_KILLED);
1376 return 0;
1377}
1378
Thierry Fournier3d4a6752016-02-19 20:53:30 +01001379int hlua_fcn_post_init(lua_State *L)
1380{
Thierry Fournierf61aa632016-02-19 20:56:00 +01001381 struct proxy *px;
1382
1383 /* get core array. */
1384 if (lua_getglobal(L, "core") != LUA_TTABLE)
1385 lua_error(L);
1386
1387 /* Create proxies entry. */
1388 lua_pushstring(L, "proxies");
1389 lua_newtable(L);
1390
1391 /* List all proxies. */
Olivier Houchardfbc74e82017-11-24 16:54:05 +01001392 for (px = proxies_list; px; px = px->next) {
Thierry Fournierf61aa632016-02-19 20:56:00 +01001393 lua_pushstring(L, px->id);
1394 hlua_fcn_new_proxy(L, px);
1395 lua_settable(L, -3);
1396 }
1397
1398 /* push "proxies" in "core" */
1399 lua_settable(L, -3);
1400
Thierry FOURNIER9b82a582017-07-24 13:30:43 +02001401 /* Create proxies entry. */
1402 lua_pushstring(L, "frontends");
1403 lua_newtable(L);
1404
1405 /* List all proxies. */
Olivier Houchardfbc74e82017-11-24 16:54:05 +01001406 for (px = proxies_list; px; px = px->next) {
Thierry FOURNIER9b82a582017-07-24 13:30:43 +02001407 if (!(px->cap & PR_CAP_FE))
1408 continue;
1409 lua_pushstring(L, px->id);
1410 hlua_fcn_new_proxy(L, px);
1411 lua_settable(L, -3);
1412 }
1413
1414 /* push "frontends" in "core" */
1415 lua_settable(L, -3);
1416
1417 /* Create proxies entry. */
1418 lua_pushstring(L, "backends");
1419 lua_newtable(L);
1420
1421 /* List all proxies. */
Olivier Houchardfbc74e82017-11-24 16:54:05 +01001422 for (px = proxies_list; px; px = px->next) {
Thierry FOURNIER9b82a582017-07-24 13:30:43 +02001423 if (!(px->cap & PR_CAP_BE))
1424 continue;
1425 lua_pushstring(L, px->id);
1426 hlua_fcn_new_proxy(L, px);
1427 lua_settable(L, -3);
1428 }
1429
1430 /* push "backend" in "core" */
1431 lua_settable(L, -3);
1432
Thierry Fournier3d4a6752016-02-19 20:53:30 +01001433 return 1;
1434}
1435
Thierry FOURNIER / OZON.IO8a1027a2016-11-24 20:48:38 +01001436/* This Lua function take a string, a list of separators.
1437 * It tokenize the input string using the list of separators
1438 * as separator.
1439 *
Ilya Shipitsince7b00f2020-03-23 22:28:40 +05001440 * The functionreturns a table filled with tokens.
Thierry FOURNIER / OZON.IO8a1027a2016-11-24 20:48:38 +01001441 */
1442int hlua_tokenize(lua_State *L)
1443{
1444 const char *str;
1445 const char *sep;
1446 int index;
1447 const char *token;
1448 const char *p;
1449 const char *c;
1450 int ignore_empty;
1451
1452 ignore_empty = 0;
1453
1454 str = luaL_checkstring(L, 1);
1455 sep = luaL_checkstring(L, 2);
1456 if (lua_gettop(L) == 3)
1457 ignore_empty = hlua_checkboolean(L, 3);
1458
1459 lua_newtable(L);
1460 index = 1;
1461 token = str;
1462 p = str;
1463 while(1) {
1464 for (c = sep; *c != '\0'; c++)
1465 if (*p == *c)
1466 break;
1467 if (*p == *c) {
1468 if ((!ignore_empty) || (p - token > 0)) {
1469 lua_pushlstring(L, token, p - token);
1470 lua_rawseti(L, -2, index);
1471 index++;
1472 }
1473 token = p + 1;
1474 }
1475 if (*p == '\0')
1476 break;
1477 p++;
1478 }
1479
1480 return 1;
1481}
1482
Thierry FOURNIER / OZON.IO62fec752016-11-10 20:38:11 +01001483int hlua_parse_addr(lua_State *L)
1484{
Christopher Faulet29e93262021-02-26 09:39:05 +01001485 struct net_addr *addr;
Thierry FOURNIER / OZON.IO62fec752016-11-10 20:38:11 +01001486 const char *str = luaL_checkstring(L, 1);
1487 unsigned char mask;
1488
Christopher Faulet29e93262021-02-26 09:39:05 +01001489 addr = lua_newuserdata(L, sizeof(struct net_addr));
Thierry FOURNIER / OZON.IO62fec752016-11-10 20:38:11 +01001490 if (!addr) {
1491 lua_pushnil(L);
1492 return 1;
1493 }
1494
1495 if (str2net(str, PAT_MF_NO_DNS, &addr->addr.v4.ip, &addr->addr.v4.mask)) {
Christopher Faulet29e93262021-02-26 09:39:05 +01001496 addr->family = AF_INET;
Thierry FOURNIER / OZON.IO62fec752016-11-10 20:38:11 +01001497 return 1;
1498 }
1499
1500 if (str62net(str, &addr->addr.v6.ip, &mask)) {
1501 len2mask6(mask, &addr->addr.v6.mask);
Christopher Faulet29e93262021-02-26 09:39:05 +01001502 addr->family = AF_INET6;
Thierry FOURNIER / OZON.IO62fec752016-11-10 20:38:11 +01001503 return 1;
1504 }
1505
1506 lua_pop(L, 1);
1507 lua_pushnil(L);
1508 return 1;
1509}
1510
1511int hlua_match_addr(lua_State *L)
1512{
Christopher Faulet29e93262021-02-26 09:39:05 +01001513 struct net_addr *addr1;
1514 struct net_addr *addr2;
Thierry FOURNIER / OZON.IO62fec752016-11-10 20:38:11 +01001515
1516 if (!lua_isuserdata(L, 1) ||
1517 !lua_isuserdata(L, 2)) {
1518 lua_pushboolean(L, 0);
1519 return 1;
1520 }
1521
1522 addr1 = lua_touserdata(L, 1);
1523 addr2 = lua_touserdata(L, 2);
1524
Christopher Faulet29e93262021-02-26 09:39:05 +01001525 if (addr1->family != addr2->family) {
Thierry FOURNIER / OZON.IO62fec752016-11-10 20:38:11 +01001526 lua_pushboolean(L, 0);
1527 return 1;
1528 }
1529
Christopher Faulet29e93262021-02-26 09:39:05 +01001530 if (addr1->family == AF_INET) {
Thierry FOURNIER / OZON.IO62fec752016-11-10 20:38:11 +01001531 if ((addr1->addr.v4.ip.s_addr & addr2->addr.v4.mask.s_addr) ==
1532 (addr2->addr.v4.ip.s_addr & addr1->addr.v4.mask.s_addr)) {
1533 lua_pushboolean(L, 1);
1534 return 1;
1535 }
1536 } else {
Thierry FOURNIERde6925e2016-12-23 17:03:25 +01001537 int i;
1538
1539 for (i = 0; i < 16; i += 4) {
Willy Tarreau26474c42020-02-25 10:02:51 +01001540 if ((read_u32(&addr1->addr.v6.ip.s6_addr[i]) &
1541 read_u32(&addr2->addr.v6.mask.s6_addr[i])) !=
1542 (read_u32(&addr2->addr.v6.ip.s6_addr[i]) &
1543 read_u32(&addr1->addr.v6.mask.s6_addr[i])))
Thierry FOURNIERde6925e2016-12-23 17:03:25 +01001544 break;
1545 }
1546 if (i == 16) {
Thierry FOURNIER / OZON.IO62fec752016-11-10 20:38:11 +01001547 lua_pushboolean(L, 1);
1548 return 1;
1549 }
1550 }
1551
1552 lua_pushboolean(L, 0);
1553 return 1;
1554}
1555
Dragan Dosen26743032019-04-30 15:54:36 +02001556static struct my_regex **hlua_check_regex(lua_State *L, int ud)
Thierry FOURNIER31904272017-10-25 12:59:51 +02001557{
1558 return (hlua_checkudata(L, ud, class_regex_ref));
1559}
1560
1561static int hlua_regex_comp(struct lua_State *L)
1562{
Dragan Dosen26743032019-04-30 15:54:36 +02001563 struct my_regex **regex;
Thierry FOURNIER31904272017-10-25 12:59:51 +02001564 const char *str;
1565 int cs;
1566 char *err;
1567
1568 str = luaL_checkstring(L, 1);
1569 luaL_argcheck(L, lua_isboolean(L, 2), 2, NULL);
1570 cs = lua_toboolean(L, 2);
1571
1572 regex = lua_newuserdata(L, sizeof(*regex));
1573
1574 err = NULL;
Dragan Dosen26743032019-04-30 15:54:36 +02001575 if (!(*regex = regex_comp(str, cs, 1, &err))) {
Thierry FOURNIER31904272017-10-25 12:59:51 +02001576 lua_pushboolean(L, 0); /* status error */
1577 lua_pushstring(L, err); /* Reason */
1578 free(err);
1579 return 2;
1580 }
1581
1582 lua_pushboolean(L, 1); /* Status ok */
1583
1584 /* Create object */
1585 lua_newtable(L);
1586 lua_pushvalue(L, -3); /* Get the userdata pointer. */
1587 lua_rawseti(L, -2, 0);
1588 lua_rawgeti(L, LUA_REGISTRYINDEX, class_regex_ref);
1589 lua_setmetatable(L, -2);
1590 return 2;
1591}
1592
1593static int hlua_regex_exec(struct lua_State *L)
1594{
Dragan Dosen26743032019-04-30 15:54:36 +02001595 struct my_regex **regex;
Thierry FOURNIER31904272017-10-25 12:59:51 +02001596 const char *str;
1597 size_t len;
Willy Tarreau83061a82018-07-13 11:56:34 +02001598 struct buffer *tmp;
Thierry FOURNIER31904272017-10-25 12:59:51 +02001599
1600 regex = hlua_check_regex(L, 1);
1601 str = luaL_checklstring(L, 2, &len);
1602
Dragan Dosen26743032019-04-30 15:54:36 +02001603 if (!*regex) {
1604 lua_pushboolean(L, 0);
1605 return 1;
1606 }
1607
Thierry FOURNIER7c210e62017-10-27 14:13:51 +02001608 /* Copy the string because regex_exec2 require a 'char *'
1609 * and not a 'const char *'.
1610 */
1611 tmp = get_trash_chunk();
1612 if (len >= tmp->size) {
1613 lua_pushboolean(L, 0);
1614 return 1;
1615 }
Willy Tarreau843b7cb2018-07-13 10:54:26 +02001616 memcpy(tmp->area, str, len);
Thierry FOURNIER7c210e62017-10-27 14:13:51 +02001617
Dragan Dosen26743032019-04-30 15:54:36 +02001618 lua_pushboolean(L, regex_exec2(*regex, tmp->area, len));
Thierry FOURNIER31904272017-10-25 12:59:51 +02001619
1620 return 1;
1621}
1622
1623static int hlua_regex_match(struct lua_State *L)
1624{
Dragan Dosen26743032019-04-30 15:54:36 +02001625 struct my_regex **regex;
Thierry FOURNIER31904272017-10-25 12:59:51 +02001626 const char *str;
1627 size_t len;
1628 regmatch_t pmatch[20];
1629 int ret;
1630 int i;
Willy Tarreau83061a82018-07-13 11:56:34 +02001631 struct buffer *tmp;
Thierry FOURNIER31904272017-10-25 12:59:51 +02001632
1633 regex = hlua_check_regex(L, 1);
1634 str = luaL_checklstring(L, 2, &len);
1635
Dragan Dosen26743032019-04-30 15:54:36 +02001636 if (!*regex) {
1637 lua_pushboolean(L, 0);
1638 return 1;
1639 }
1640
Thierry FOURNIER7c210e62017-10-27 14:13:51 +02001641 /* Copy the string because regex_exec2 require a 'char *'
1642 * and not a 'const char *'.
1643 */
1644 tmp = get_trash_chunk();
1645 if (len >= tmp->size) {
1646 lua_pushboolean(L, 0);
1647 return 1;
1648 }
Willy Tarreau843b7cb2018-07-13 10:54:26 +02001649 memcpy(tmp->area, str, len);
Thierry FOURNIER7c210e62017-10-27 14:13:51 +02001650
Dragan Dosen26743032019-04-30 15:54:36 +02001651 ret = regex_exec_match2(*regex, tmp->area, len, 20, pmatch, 0);
Thierry FOURNIER31904272017-10-25 12:59:51 +02001652 lua_pushboolean(L, ret);
1653 lua_newtable(L);
1654 if (ret) {
1655 for (i = 0; i < 20 && pmatch[i].rm_so != -1; i++) {
1656 lua_pushlstring(L, str + pmatch[i].rm_so, pmatch[i].rm_eo - pmatch[i].rm_so);
1657 lua_rawseti(L, -2, i + 1);
1658 }
1659 }
1660 return 2;
1661}
1662
1663static int hlua_regex_free(struct lua_State *L)
1664{
Dragan Dosen26743032019-04-30 15:54:36 +02001665 struct my_regex **regex;
Thierry FOURNIER31904272017-10-25 12:59:51 +02001666
1667 regex = hlua_check_regex(L, 1);
Dragan Dosen26743032019-04-30 15:54:36 +02001668 regex_free(*regex);
1669 *regex = NULL;
Thierry FOURNIER31904272017-10-25 12:59:51 +02001670 return 0;
1671}
1672
Thierry Fournierfb0b5462016-01-21 09:28:58 +01001673int hlua_fcn_reg_core_fcn(lua_State *L)
1674{
Thierry Fournier1de16592016-01-27 09:49:07 +01001675 if (!hlua_concat_init(L))
1676 return 0;
1677
Thierry Fournier4f99b272016-02-22 08:40:02 +01001678 hlua_class_function(L, "now", hlua_now);
1679 hlua_class_function(L, "http_date", hlua_http_date);
1680 hlua_class_function(L, "imf_date", hlua_imf_date);
1681 hlua_class_function(L, "rfc850_date", hlua_rfc850_date);
1682 hlua_class_function(L, "asctime_date", hlua_asctime_date);
1683 hlua_class_function(L, "concat", hlua_concat_new);
Thierry Fourniereea77c02016-03-18 08:47:13 +01001684 hlua_class_function(L, "get_info", hlua_get_info);
Thierry FOURNIER / OZON.IO62fec752016-11-10 20:38:11 +01001685 hlua_class_function(L, "parse_addr", hlua_parse_addr);
1686 hlua_class_function(L, "match_addr", hlua_match_addr);
Thierry FOURNIER / OZON.IO8a1027a2016-11-24 20:48:38 +01001687 hlua_class_function(L, "tokenize", hlua_tokenize);
Thierry Fournier4f99b272016-02-22 08:40:02 +01001688
Thierry FOURNIER31904272017-10-25 12:59:51 +02001689 /* Create regex object. */
1690 lua_newtable(L);
1691 hlua_class_function(L, "new", hlua_regex_comp);
1692
1693 lua_newtable(L); /* The metatable. */
1694 lua_pushstring(L, "__index");
1695 lua_newtable(L);
1696 hlua_class_function(L, "exec", hlua_regex_exec);
1697 hlua_class_function(L, "match", hlua_regex_match);
1698 lua_rawset(L, -3); /* -> META["__index"] = TABLE */
1699 hlua_class_function(L, "__gc", hlua_regex_free);
1700
1701 lua_pushvalue(L, -1); /* Duplicate the metatable reference. */
1702 class_regex_ref = hlua_register_metatable(L, CLASS_REGEX);
1703
1704 lua_setmetatable(L, -2);
1705 lua_setglobal(L, CLASS_REGEX); /* Create global object called Regex */
1706
Adis Nezirovic8878f8e2018-07-13 12:18:33 +02001707 /* Create stktable object. */
1708 lua_newtable(L);
1709 lua_pushstring(L, "__index");
1710 lua_newtable(L);
1711 hlua_class_function(L, "info", hlua_stktable_info);
1712 hlua_class_function(L, "lookup", hlua_stktable_lookup);
1713 hlua_class_function(L, "dump", hlua_stktable_dump);
1714 lua_settable(L, -3); /* -> META["__index"] = TABLE */
1715 class_stktable_ref = hlua_register_metatable(L, CLASS_STKTABLE);
1716
Thierry Fournierff480422016-02-25 08:36:46 +01001717 /* Create listener object. */
1718 lua_newtable(L);
1719 lua_pushstring(L, "__index");
1720 lua_newtable(L);
1721 hlua_class_function(L, "get_stats", hlua_listener_get_stats);
1722 lua_settable(L, -3); /* -> META["__index"] = TABLE */
1723 class_listener_ref = hlua_register_metatable(L, CLASS_LISTENER);
1724
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001725 /* Create server object. */
1726 lua_newtable(L);
1727 lua_pushstring(L, "__index");
1728 lua_newtable(L);
1729 hlua_class_function(L, "is_draining", hlua_server_is_draining);
Patrick Hemmer32d539f2018-04-29 14:25:46 -04001730 hlua_class_function(L, "set_maxconn", hlua_server_set_maxconn);
1731 hlua_class_function(L, "get_maxconn", hlua_server_get_maxconn);
Thierry Fournierf2fdc9d2016-02-22 08:21:39 +01001732 hlua_class_function(L, "set_weight", hlua_server_set_weight);
1733 hlua_class_function(L, "get_weight", hlua_server_get_weight);
1734 hlua_class_function(L, "set_addr", hlua_server_set_addr);
1735 hlua_class_function(L, "get_addr", hlua_server_get_addr);
1736 hlua_class_function(L, "get_stats", hlua_server_get_stats);
1737 hlua_class_function(L, "shut_sess", hlua_server_shut_sess);
1738 hlua_class_function(L, "set_drain", hlua_server_set_drain);
1739 hlua_class_function(L, "set_maint", hlua_server_set_maint);
1740 hlua_class_function(L, "set_ready", hlua_server_set_ready);
1741 hlua_class_function(L, "check_enable", hlua_server_check_enable);
1742 hlua_class_function(L, "check_disable", hlua_server_check_disable);
1743 hlua_class_function(L, "check_force_up", hlua_server_check_force_up);
1744 hlua_class_function(L, "check_force_nolb", hlua_server_check_force_nolb);
1745 hlua_class_function(L, "check_force_down", hlua_server_check_force_down);
1746 hlua_class_function(L, "agent_enable", hlua_server_agent_enable);
1747 hlua_class_function(L, "agent_disable", hlua_server_agent_disable);
1748 hlua_class_function(L, "agent_force_up", hlua_server_agent_force_up);
1749 hlua_class_function(L, "agent_force_down", hlua_server_agent_force_down);
1750 lua_settable(L, -3); /* -> META["__index"] = TABLE */
1751 class_server_ref = hlua_register_metatable(L, CLASS_SERVER);
1752
Thierry Fournierf61aa632016-02-19 20:56:00 +01001753 /* Create proxy object. */
1754 lua_newtable(L);
1755 lua_pushstring(L, "__index");
1756 lua_newtable(L);
1757 hlua_class_function(L, "pause", hlua_proxy_pause);
1758 hlua_class_function(L, "resume", hlua_proxy_resume);
1759 hlua_class_function(L, "stop", hlua_proxy_stop);
1760 hlua_class_function(L, "shut_bcksess", hlua_proxy_shut_bcksess);
1761 hlua_class_function(L, "get_cap", hlua_proxy_get_cap);
1762 hlua_class_function(L, "get_mode", hlua_proxy_get_mode);
1763 hlua_class_function(L, "get_stats", hlua_proxy_get_stats);
1764 lua_settable(L, -3); /* -> META["__index"] = TABLE */
1765 class_proxy_ref = hlua_register_metatable(L, CLASS_PROXY);
1766
Thierry Fournier1550d5d2016-01-21 09:35:41 +01001767 return 5;
Thierry Fournierfb0b5462016-01-21 09:28:58 +01001768}