1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495 |
- /* Copyright (c) 2014, Vsevolod Stakhov
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are met:
- * * Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * * Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- *
- * THIS SOFTWARE IS PROVIDED ''AS IS'' AND ANY
- * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
- * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL AUTHOR BE LIABLE FOR ANY
- * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
- * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
- * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
- * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
-
- /**
- * @file lua ucl bindings
- */
-
- #include "ucl.h"
- #include "ucl_internal.h"
- #include "lua_ucl.h"
- #include <strings.h>
-
- /***
- * @module ucl
- * This lua module allows to parse objects from strings and to store data into
- * ucl objects. It uses `libucl` C library to parse and manipulate with ucl objects.
- * @example
- local ucl = require("ucl")
-
- local parser = ucl.parser()
- local res,err = parser:parse_string('{key=value}')
-
- if not res then
- print('parser error: ' .. err)
- else
- local obj = parser:get_object()
- local got = ucl.to_format(obj, 'json')
- endif
-
- local table = {
- str = 'value',
- num = 100500,
- null = ucl.null,
- func = function ()
- return 'huh'
- end
- }
-
- print(ucl.to_format(table, 'ucl'))
- -- Output:
- --[[
- num = 100500;
- str = "value";
- null = null;
- func = "huh";
- --]]
- */
-
- #define PARSER_META "ucl.parser.meta"
- #define EMITTER_META "ucl.emitter.meta"
- #define NULL_META "ucl.null.meta"
- #define OBJECT_META "ucl.object.meta"
- #define UCL_OBJECT_TYPE_META "ucl.type.object"
- #define UCL_ARRAY_TYPE_META "ucl.type.array"
- #define UCL_IMPL_ARRAY_TYPE_META "ucl.type.impl_array"
-
- static int ucl_object_lua_push_array (lua_State *L, const ucl_object_t *obj);
- static int ucl_object_lua_push_scalar (lua_State *L, const ucl_object_t *obj, bool allow_array);
- static ucl_object_t* ucl_object_lua_fromtable (lua_State *L, int idx, ucl_string_flags_t flags);
- static ucl_object_t* ucl_object_lua_fromelt (lua_State *L, int idx, ucl_string_flags_t flags);
-
- static void *ucl_null;
-
- /**
- * Push a single element of an object to lua
- * @param L
- * @param key
- * @param obj
- */
- static void
- ucl_object_lua_push_element (lua_State *L, const char *key,
- const ucl_object_t *obj)
- {
- lua_pushstring (L, key);
- ucl_object_push_lua (L, obj, true);
- lua_settable (L, -3);
- }
-
- static void
- lua_ucl_userdata_dtor (void *ud)
- {
- struct ucl_lua_funcdata *fd = (struct ucl_lua_funcdata *)ud;
-
- luaL_unref (fd->L, LUA_REGISTRYINDEX, fd->idx);
- if (fd->ret != NULL) {
- free (fd->ret);
- }
- free (fd);
- }
-
- static const char *
- lua_ucl_userdata_emitter (void *ud)
- {
- struct ucl_lua_funcdata *fd = (struct ucl_lua_funcdata *)ud;
- const char *out = "";
-
- lua_rawgeti (fd->L, LUA_REGISTRYINDEX, fd->idx);
-
- lua_pcall (fd->L, 0, 1, 0);
- out = lua_tostring (fd->L, -1);
-
- if (out != NULL) {
- /* We need to store temporary string in a more appropriate place */
- if (fd->ret) {
- free (fd->ret);
- }
- fd->ret = strdup (out);
- }
-
- lua_settop (fd->L, 0);
-
- return fd->ret;
- }
-
- /**
- * Push a single object to lua
- * @param L
- * @param obj
- * @return
- */
- static int
- ucl_object_lua_push_object (lua_State *L, const ucl_object_t *obj,
- bool allow_array)
- {
- const ucl_object_t *cur;
- ucl_object_iter_t it = NULL;
-
- if (allow_array && obj->next != NULL) {
- /* Actually we need to push this as an array */
- return ucl_object_lua_push_array (L, obj);
- }
-
- lua_createtable (L, 0, obj->len);
- it = NULL;
-
- while ((cur = ucl_object_iterate (obj, &it, true)) != NULL) {
- ucl_object_lua_push_element (L, ucl_object_key (cur), cur);
- }
-
- luaL_getmetatable (L, UCL_OBJECT_TYPE_META);
- lua_setmetatable (L, -2);
-
- return 1;
- }
-
- /**
- * Push an array to lua as table indexed by integers
- * @param L
- * @param obj
- * @return
- */
- static int
- ucl_object_lua_push_array (lua_State *L, const ucl_object_t *obj)
- {
- const ucl_object_t *cur;
- ucl_object_iter_t it;
- int i = 1, nelt = 0;
-
- if (obj->type == UCL_ARRAY) {
- nelt = obj->len;
- it = ucl_object_iterate_new (obj);
- lua_createtable (L, nelt, 0);
-
- while ((cur = ucl_object_iterate_safe (it, true))) {
- ucl_object_push_lua (L, cur, false);
- lua_rawseti (L, -2, i);
- i ++;
- }
-
- luaL_getmetatable (L, UCL_ARRAY_TYPE_META);
- lua_setmetatable (L, -2);
-
- ucl_object_iterate_free (it);
- }
- else {
- /* Optimize allocation by preallocation of table */
- LL_FOREACH (obj, cur) {
- nelt ++;
- }
-
- lua_createtable (L, nelt, 0);
-
- LL_FOREACH (obj, cur) {
- ucl_object_push_lua (L, cur, false);
- lua_rawseti (L, -2, i);
- i ++;
- }
-
- luaL_getmetatable (L, UCL_IMPL_ARRAY_TYPE_META);
- lua_setmetatable (L, -2);
- }
-
- return 1;
- }
-
- /**
- * Push a simple object to lua depending on its actual type
- */
- static int
- ucl_object_lua_push_scalar (lua_State *L, const ucl_object_t *obj,
- bool allow_array)
- {
- struct ucl_lua_funcdata *fd;
-
- if (allow_array && obj->next != NULL) {
- /* Actually we need to push this as an array */
- return ucl_object_lua_push_array (L, obj);
- }
-
- switch (obj->type) {
- case UCL_BOOLEAN:
- lua_pushboolean (L, ucl_obj_toboolean (obj));
- break;
- case UCL_STRING:
- lua_pushstring (L, ucl_obj_tostring (obj));
- break;
- case UCL_INT:
- #if LUA_VERSION_NUM >= 501
- lua_pushinteger (L, ucl_obj_toint (obj));
- #else
- lua_pushnumber (L, ucl_obj_toint (obj));
- #endif
- break;
- case UCL_FLOAT:
- case UCL_TIME:
- lua_pushnumber (L, ucl_obj_todouble (obj));
- break;
- case UCL_NULL:
- lua_getfield (L, LUA_REGISTRYINDEX, "ucl.null");
- break;
- case UCL_USERDATA:
- fd = (struct ucl_lua_funcdata *)obj->value.ud;
- lua_rawgeti (L, LUA_REGISTRYINDEX, fd->idx);
- break;
- default:
- lua_pushnil (L);
- break;
- }
-
- return 1;
- }
-
- /***
- * @function ucl_object_push_lua(L, obj, allow_array)
- * This is a `C` function to push `UCL` object as lua variable. This function
- * converts `obj` to lua representation using the following conversions:
- *
- * - *scalar* values are directly presented by lua objects
- * - *userdata* values are converted to lua function objects using `LUA_REGISTRYINDEX`,
- * this can be used to pass functions from lua to c and vice-versa
- * - *arrays* are converted to lua tables with numeric indicies suitable for `ipairs` iterations
- * - *objects* are converted to lua tables with string indicies
- * @param {lua_State} L lua state pointer
- * @param {ucl_object_t} obj object to push
- * @param {bool} allow_array expand implicit arrays (should be true for all but partial arrays)
- * @return {int} `1` if an object is pushed to lua
- */
- int
- ucl_object_push_lua (lua_State *L, const ucl_object_t *obj, bool allow_array)
- {
- switch (obj->type) {
- case UCL_OBJECT:
- return ucl_object_lua_push_object (L, obj, allow_array);
- case UCL_ARRAY:
- return ucl_object_lua_push_array (L, obj);
- default:
- return ucl_object_lua_push_scalar (L, obj, allow_array);
- }
- }
-
- /**
- * Parse lua table into object top
- * @param L
- * @param top
- * @param idx
- */
- static ucl_object_t *
- ucl_object_lua_fromtable (lua_State *L, int idx, ucl_string_flags_t flags)
- {
- ucl_object_t *obj, *top = NULL, *cur;
- size_t keylen;
- const char *k;
- bool is_array = true, is_implicit = false, found_mt = false;
- size_t max = 0, nelts = 0;
-
- if (idx < 0) {
- /* For negative indicies we want to invert them */
- idx = lua_gettop (L) + idx + 1;
- }
-
- /* First, we check from metatable */
- if (luaL_getmetafield (L, idx, "class") != 0) {
-
- if (lua_type (L, -1) == LUA_TSTRING) {
- const char *classname = lua_tostring (L, -1);
-
- if (strcmp (classname, UCL_OBJECT_TYPE_META) == 0) {
- is_array = false;
- found_mt = true;
- } else if (strcmp (classname, UCL_ARRAY_TYPE_META) == 0) {
- is_array = true;
- found_mt = true;
- #if LUA_VERSION_NUM >= 502
- max = lua_rawlen (L, idx);
- #else
- max = lua_objlen (L, idx);
- #endif
- nelts = max;
- } else if (strcmp (classname, UCL_IMPL_ARRAY_TYPE_META) == 0) {
- is_array = true;
- is_implicit = true;
- found_mt = true;
- #if LUA_VERSION_NUM >= 502
- max = lua_rawlen (L, idx);
- #else
- max = lua_objlen (L, idx);
- #endif
- nelts = max;
- }
- }
-
- lua_pop (L, 1);
- }
-
- if (!found_mt) {
- /* Check for array */
- lua_pushnil (L);
- while (lua_next (L, idx) != 0) {
- if (lua_type (L, -2) == LUA_TNUMBER) {
- double num = lua_tonumber (L, -2);
- if (num == (int) num) {
- if (num > max) {
- max = num;
- }
- } else {
- /* Keys are not integer */
- is_array = false;
- }
- } else {
- /* Keys are not numeric */
- is_array = false;
- }
- lua_pop (L, 1);
- nelts ++;
- }
- }
-
- /* Table iterate */
- if (is_array) {
- int i;
-
- if (!is_implicit) {
- top = ucl_object_typed_new (UCL_ARRAY);
- ucl_object_reserve (top, nelts);
- }
- else {
- top = NULL;
- }
-
- for (i = 1; i <= max; i ++) {
- lua_pushinteger (L, i);
- lua_gettable (L, idx);
-
- obj = ucl_object_lua_fromelt (L, lua_gettop (L), flags);
-
- if (obj != NULL) {
- if (is_implicit) {
- DL_APPEND (top, obj);
- }
- else {
- ucl_array_append (top, obj);
- }
- }
- lua_pop (L, 1);
- }
- }
- else {
- lua_pushnil (L);
- top = ucl_object_typed_new (UCL_OBJECT);
- ucl_object_reserve (top, nelts);
-
- while (lua_next (L, idx) != 0) {
- /* copy key to avoid modifications */
- lua_pushvalue (L, -2);
- k = lua_tolstring (L, -1, &keylen);
- obj = ucl_object_lua_fromelt (L, lua_gettop (L) - 1, flags);
-
- if (obj != NULL) {
- ucl_object_insert_key (top, obj, k, keylen, true);
-
- DL_FOREACH (obj, cur) {
- if (cur->keylen == 0) {
- cur->keylen = obj->keylen;
- cur->key = obj->key;
- }
- }
- }
- lua_pop (L, 2);
- }
- }
-
- return top;
- }
-
- /**
- * Get a single element from lua to object obj
- * @param L
- * @param obj
- * @param idx
- */
- static ucl_object_t *
- ucl_object_lua_fromelt (lua_State *L, int idx, ucl_string_flags_t flags)
- {
- int type;
- double num;
- ucl_object_t *obj = NULL;
- struct ucl_lua_funcdata *fd;
- const char *str;
- size_t sz;
-
- type = lua_type (L, idx);
-
- switch (type) {
- case LUA_TSTRING:
- str = lua_tolstring (L, idx, &sz);
-
- if (str) {
- obj = ucl_object_fromstring_common (str, sz, flags);
- }
- else {
- obj = ucl_object_typed_new (UCL_NULL);
- }
- break;
- case LUA_TNUMBER:
- num = lua_tonumber (L, idx);
- if (num == (int64_t)num) {
- obj = ucl_object_fromint (num);
- }
- else {
- obj = ucl_object_fromdouble (num);
- }
- break;
- case LUA_TBOOLEAN:
- obj = ucl_object_frombool (lua_toboolean (L, idx));
- break;
- case LUA_TUSERDATA:
- if (lua_topointer (L, idx) == ucl_null) {
- obj = ucl_object_typed_new (UCL_NULL);
- }
- break;
- case LUA_TTABLE:
- case LUA_TFUNCTION:
- case LUA_TTHREAD:
- if (luaL_getmetafield (L, idx, "__gen_ucl")) {
- if (lua_isfunction (L, -1)) {
- lua_settop (L, 3); /* gen, obj, func */
- lua_insert (L, 1); /* func, gen, obj */
- lua_insert (L, 2); /* func, obj, gen */
- lua_call(L, 2, 1);
- obj = ucl_object_lua_fromelt (L, 1, flags);
- }
- lua_pop (L, 2);
- }
- else {
- if (type == LUA_TTABLE) {
- obj = ucl_object_lua_fromtable (L, idx, flags);
- }
- else if (type == LUA_TFUNCTION) {
- fd = malloc (sizeof (*fd));
- if (fd != NULL) {
- lua_pushvalue (L, idx);
- fd->L = L;
- fd->ret = NULL;
- fd->idx = luaL_ref (L, LUA_REGISTRYINDEX);
-
- obj = ucl_object_new_userdata (lua_ucl_userdata_dtor,
- lua_ucl_userdata_emitter, (void *)fd);
- }
- }
- }
- break;
- }
-
- return obj;
- }
-
- /**
- * @function ucl_object_lua_import(L, idx)
- * Extracts ucl object from lua variable at `idx` position,
- * @see ucl_object_push_lua for conversion definitions
- * @param {lua_state} L lua state machine pointer
- * @param {int} idx index where the source variable is placed
- * @return {ucl_object_t} new ucl object extracted from lua variable. Reference count of this object is 1,
- * this object thus needs to be unref'ed after usage.
- */
- ucl_object_t *
- ucl_object_lua_import (lua_State *L, int idx)
- {
- ucl_object_t *obj;
- int t;
-
- t = lua_type (L, idx);
- switch (t) {
- case LUA_TTABLE:
- obj = ucl_object_lua_fromtable (L, idx, 0);
- break;
- default:
- obj = ucl_object_lua_fromelt (L, idx, 0);
- break;
- }
-
- return obj;
- }
-
- /**
- * @function ucl_object_lua_import_escape(L, idx)
- * Extracts ucl object from lua variable at `idx` position escaping JSON strings
- * @see ucl_object_push_lua for conversion definitions
- * @param {lua_state} L lua state machine pointer
- * @param {int} idx index where the source variable is placed
- * @return {ucl_object_t} new ucl object extracted from lua variable. Reference count of this object is 1,
- * this object thus needs to be unref'ed after usage.
- */
- ucl_object_t *
- ucl_object_lua_import_escape (lua_State *L, int idx)
- {
- ucl_object_t *obj;
- int t;
-
- t = lua_type (L, idx);
- switch (t) {
- case LUA_TTABLE:
- obj = ucl_object_lua_fromtable (L, idx, UCL_STRING_RAW);
- break;
- default:
- obj = ucl_object_lua_fromelt (L, idx, UCL_STRING_RAW);
- break;
- }
-
- return obj;
- }
-
- static int
- lua_ucl_to_string (lua_State *L, const ucl_object_t *obj, enum ucl_emitter type)
- {
- unsigned char *result;
-
- result = ucl_object_emit (obj, type);
-
- if (result != NULL) {
- lua_pushstring (L, (const char *)result);
- free (result);
- }
- else {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
- static int
- lua_ucl_parser_init (lua_State *L)
- {
- struct ucl_parser *parser, **pparser;
- int flags = UCL_PARSER_NO_FILEVARS;
-
- if (lua_gettop (L) >= 1) {
- flags = lua_tonumber (L, 1);
- }
-
- parser = ucl_parser_new (flags);
- if (parser == NULL) {
- lua_pushnil (L);
- }
-
- pparser = lua_newuserdata (L, sizeof (parser));
- *pparser = parser;
- luaL_getmetatable (L, PARSER_META);
- lua_setmetatable (L, -2);
-
- return 1;
- }
-
- static struct ucl_parser *
- lua_ucl_parser_get (lua_State *L, int index)
- {
- return *((struct ucl_parser **) luaL_checkudata(L, index, PARSER_META));
- }
-
- static ucl_object_t *
- lua_ucl_object_get (lua_State *L, int index)
- {
- return *((ucl_object_t **) luaL_checkudata(L, index, OBJECT_META));
- }
-
- static void
- lua_ucl_push_opaque (lua_State *L, ucl_object_t *obj)
- {
- ucl_object_t **pobj;
-
- pobj = lua_newuserdata (L, sizeof (*pobj));
- *pobj = obj;
- luaL_getmetatable (L, OBJECT_META);
- lua_setmetatable (L, -2);
- }
-
- static inline enum ucl_parse_type
- lua_ucl_str_to_parse_type (const char *str)
- {
- enum ucl_parse_type type = UCL_PARSE_UCL;
-
- if (str != NULL) {
- if (strcasecmp (str, "msgpack") == 0) {
- type = UCL_PARSE_MSGPACK;
- }
- else if (strcasecmp (str, "sexp") == 0 ||
- strcasecmp (str, "csexp") == 0) {
- type = UCL_PARSE_CSEXP;
- }
- else if (strcasecmp (str, "auto") == 0) {
- type = UCL_PARSE_AUTO;
- }
- }
-
- return type;
- }
-
- /***
- * @method parser:parse_file(name)
- * Parse UCL object from file.
- * @param {string} name filename to parse
- * @return {bool[, string]} if res is `true` then file has been parsed successfully, otherwise an error string is also returned
- @example
- local parser = ucl.parser()
- local res,err = parser:parse_file('/some/file.conf')
-
- if not res then
- print('parser error: ' .. err)
- else
- -- Do something with object
- end
- */
- static int
- lua_ucl_parser_parse_file (lua_State *L)
- {
- struct ucl_parser *parser;
- const char *file;
- int ret = 2;
-
- parser = lua_ucl_parser_get (L, 1);
- file = luaL_checkstring (L, 2);
-
- if (parser != NULL && file != NULL) {
- if (ucl_parser_add_file (parser, file)) {
- lua_pushboolean (L, true);
- ret = 1;
- }
- else {
- lua_pushboolean (L, false);
- lua_pushstring (L, ucl_parser_get_error (parser));
- }
- }
- else {
- lua_pushboolean (L, false);
- lua_pushstring (L, "invalid arguments");
- }
-
- return ret;
- }
-
- /***
- * @method parser:register_variable(name, value)
- * Register parser variable
- * @param {string} name name of variable
- * @param {string} value value of variable
- * @return {bool} success
- @example
- local parser = ucl.parser()
- local res = parser:register_variable('CONFDIR', '/etc/foo')
- */
- static int
- lua_ucl_parser_register_variable (lua_State *L)
- {
- struct ucl_parser *parser;
- const char *name, *value;
- int ret = 2;
-
- parser = lua_ucl_parser_get (L, 1);
- name = luaL_checkstring (L, 2);
- value = luaL_checkstring (L, 3);
-
- if (parser != NULL && name != NULL && value != NULL) {
- ucl_parser_register_variable (parser, name, value);
- lua_pushboolean (L, true);
- ret = 1;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return ret;
- }
-
- /***
- * @method parser:register_variables(vars)
- * Register parser variables
- * @param {table} vars names/values of variables
- * @return {bool} success
- @example
- local parser = ucl.parser()
- local res = parser:register_variables({CONFDIR = '/etc/foo', VARDIR = '/var'})
- */
- static int
- lua_ucl_parser_register_variables (lua_State *L)
- {
- struct ucl_parser *parser;
- const char *name, *value;
- int ret = 2;
-
- parser = lua_ucl_parser_get (L, 1);
-
- if (parser != NULL && lua_type (L, 2) == LUA_TTABLE) {
- for (lua_pushnil (L); lua_next (L, 2); lua_pop (L, 1)) {
- lua_pushvalue (L, -2);
- name = luaL_checkstring (L, -1);
- value = luaL_checkstring (L, -2);
- ucl_parser_register_variable (parser, name, value);
- lua_pop (L, 1);
- }
-
- lua_pushboolean (L, true);
- ret = 1;
- }
- else {
- return luaL_error (L, "invalid arguments");
- }
-
- return ret;
- }
-
- /***
- * @method parser:parse_string(input)
- * Parse UCL object from file.
- * @param {string} input string to parse
- * @return {bool[, string]} if res is `true` then file has been parsed successfully, otherwise an error string is also returned
- */
- static int
- lua_ucl_parser_parse_string (lua_State *L)
- {
- struct ucl_parser *parser;
- const char *string;
- size_t llen;
- enum ucl_parse_type type = UCL_PARSE_UCL;
- int ret = 2;
-
- parser = lua_ucl_parser_get (L, 1);
- string = luaL_checklstring (L, 2, &llen);
-
- if (lua_type (L, 3) == LUA_TSTRING) {
- type = lua_ucl_str_to_parse_type (lua_tostring (L, 3));
- }
-
- if (parser != NULL && string != NULL) {
- if (ucl_parser_add_chunk_full (parser, (const unsigned char *)string,
- llen, 0, UCL_DUPLICATE_APPEND, type)) {
- lua_pushboolean (L, true);
- ret = 1;
- }
- else {
- lua_pushboolean (L, false);
- lua_pushstring (L, ucl_parser_get_error (parser));
- }
- }
- else {
- lua_pushboolean (L, false);
- lua_pushstring (L, "invalid arguments");
- }
-
- return ret;
- }
-
- struct _rspamd_lua_text {
- const char *start;
- unsigned int len;
- unsigned int flags;
- };
-
- /***
- * @method parser:parse_text(input)
- * Parse UCL object from file.
- * @param {string} input string to parse
- * @return {bool[, string]} if res is `true` then file has been parsed successfully, otherwise an error string is also returned
- */
- static int
- lua_ucl_parser_parse_text (lua_State *L)
- {
- struct ucl_parser *parser;
- struct _rspamd_lua_text *t;
- enum ucl_parse_type type = UCL_PARSE_UCL;
- int ret = 2;
-
- parser = lua_ucl_parser_get (L, 1);
- t = luaL_checkudata (L, 2, "rspamd{text}");
-
- if (lua_type (L, 3) == LUA_TSTRING) {
- type = lua_ucl_str_to_parse_type (lua_tostring (L, 3));
- }
-
- if (parser != NULL && t != NULL) {
- if (ucl_parser_add_chunk_full (parser, (const unsigned char *)t->start,
- t->len, 0, UCL_DUPLICATE_APPEND, type)) {
- lua_pushboolean (L, true);
- ret = 1;
- }
- else {
- lua_pushboolean (L, false);
- lua_pushstring (L, ucl_parser_get_error (parser));
- }
- }
- else {
- lua_pushboolean (L, false);
- lua_pushstring (L, "invalid arguments");
- }
-
- return ret;
- }
-
- /***
- * @method parser:get_object()
- * Get top object from parser and export it to lua representation.
- * @return {variant or nil} ucl object as lua native variable
- */
- static int
- lua_ucl_parser_get_object (lua_State *L)
- {
- struct ucl_parser *parser;
- ucl_object_t *obj;
- int ret = 1;
-
- parser = lua_ucl_parser_get (L, 1);
- obj = ucl_parser_get_object (parser);
-
- if (obj != NULL) {
- ret = ucl_object_push_lua (L, obj, false);
- /* no need to keep reference */
- ucl_object_unref (obj);
- }
- else {
- lua_pushnil (L);
- }
-
- return ret;
- }
-
- /***
- * @method parser:get_object_wrapped()
- * Get top object from parser and export it to userdata object without
- * unwrapping to lua.
- * @return {ucl.object or nil} ucl object wrapped variable
- */
- static int
- lua_ucl_parser_get_object_wrapped (lua_State *L)
- {
- struct ucl_parser *parser;
- ucl_object_t *obj;
- int ret = 1;
-
- parser = lua_ucl_parser_get (L, 1);
- obj = ucl_parser_get_object (parser);
-
- if (obj != NULL) {
- lua_ucl_push_opaque (L, obj);
- }
- else {
- lua_pushnil (L);
- }
-
- return ret;
- }
-
- /***
- * @method parser:validate(schema)
- * Validates the top object in the parser against schema. Schema might be
- * another object or a string that represents file to load schema from.
- *
- * @param {string/table} schema input schema
- * @return {result,err} two values: boolean result and the corresponding error
- *
- */
- static int
- lua_ucl_parser_validate (lua_State *L)
- {
- struct ucl_parser *parser, *schema_parser;
- ucl_object_t *schema;
- const char *schema_file;
- struct ucl_schema_error err;
-
- parser = lua_ucl_parser_get (L, 1);
-
- if (parser && parser->top_obj) {
- if (lua_type (L, 2) == LUA_TTABLE) {
- schema = ucl_object_lua_import (L, 2);
-
- if (schema == NULL) {
- lua_pushboolean (L, false);
- lua_pushstring (L, "cannot load schema from lua table");
-
- return 2;
- }
- }
- else if (lua_type (L, 2) == LUA_TSTRING) {
- schema_parser = ucl_parser_new (0);
- schema_file = luaL_checkstring (L, 2);
-
- if (!ucl_parser_add_file (schema_parser, schema_file)) {
- lua_pushboolean (L, false);
- lua_pushfstring (L, "cannot parse schema file \"%s\": "
- "%s", schema_file, ucl_parser_get_error (parser));
- ucl_parser_free (schema_parser);
-
- return 2;
- }
-
- schema = ucl_parser_get_object (schema_parser);
- ucl_parser_free (schema_parser);
- }
- else {
- lua_pushboolean (L, false);
- lua_pushstring (L, "invalid schema argument");
-
- return 2;
- }
-
- if (!ucl_object_validate (schema, parser->top_obj, &err)) {
- lua_pushboolean (L, false);
- lua_pushfstring (L, "validation error: "
- "%s", err.msg);
- }
- else {
- lua_pushboolean (L, true);
- lua_pushnil (L);
- }
-
- ucl_object_unref (schema);
- }
- else {
- lua_pushboolean (L, false);
- lua_pushstring (L, "invalid parser or empty top object");
- }
-
- return 2;
- }
-
- static int
- lua_ucl_parser_gc (lua_State *L)
- {
- struct ucl_parser *parser;
-
- parser = lua_ucl_parser_get (L, 1);
- ucl_parser_free (parser);
-
- return 0;
- }
-
- /***
- * @method object:unwrap()
- * Unwraps opaque ucl object to the native lua object (performing copying)
- * @return {variant} any lua object
- */
- static int
- lua_ucl_object_unwrap (lua_State *L)
- {
- ucl_object_t *obj;
-
- obj = lua_ucl_object_get (L, 1);
-
- if (obj) {
- ucl_object_push_lua (L, obj, true);
- }
- else {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
- static inline enum ucl_emitter
- lua_ucl_str_to_emit_type (const char *strtype)
- {
- enum ucl_emitter format = UCL_EMIT_JSON_COMPACT;
-
- if (strcasecmp (strtype, "json") == 0) {
- format = UCL_EMIT_JSON;
- }
- else if (strcasecmp (strtype, "json-compact") == 0) {
- format = UCL_EMIT_JSON_COMPACT;
- }
- else if (strcasecmp (strtype, "yaml") == 0) {
- format = UCL_EMIT_YAML;
- }
- else if (strcasecmp (strtype, "config") == 0 ||
- strcasecmp (strtype, "ucl") == 0) {
- format = UCL_EMIT_CONFIG;
- }
-
- return format;
- }
-
- /***
- * @method object:tostring(type)
- * Unwraps opaque ucl object to string (json by default). Optionally you can
- * specify output format:
- *
- * - `json` - fine printed json
- * - `json-compact` - compacted json
- * - `config` - fine printed configuration
- * - `ucl` - same as `config`
- * - `yaml` - embedded yaml
- * @param {string} type optional
- * @return {string} string representation of the opaque ucl object
- */
- static int
- lua_ucl_object_tostring (lua_State *L)
- {
- ucl_object_t *obj;
- enum ucl_emitter format = UCL_EMIT_JSON_COMPACT;
-
- obj = lua_ucl_object_get (L, 1);
-
- if (obj) {
- if (lua_gettop (L) > 1) {
- if (lua_type (L, 2) == LUA_TSTRING) {
- const char *strtype = lua_tostring (L, 2);
-
- format = lua_ucl_str_to_emit_type (strtype);
- }
- }
-
- return lua_ucl_to_string (L, obj, format);
- }
- else {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
- /***
- * @method object:validate(schema[, path[, ext_refs]])
- * Validates the given ucl object using schema object represented as another
- * opaque ucl object. You can also specify path in the form `#/path/def` to
- * specify the specific schema element to perform validation.
- *
- * @param {ucl.object} schema schema object
- * @param {string} path optional path for validation procedure
- * @return {result,err} two values: boolean result and the corresponding
- * error, if `ext_refs` are also specified, then they are returned as opaque
- * ucl object as {result,err,ext_refs}
- */
- static int
- lua_ucl_object_validate (lua_State *L)
- {
- ucl_object_t *obj, *schema, *ext_refs = NULL;
- const ucl_object_t *schema_elt;
- bool res = false;
- struct ucl_schema_error err;
- const char *path = NULL;
-
- obj = lua_ucl_object_get (L, 1);
- schema = lua_ucl_object_get (L, 2);
-
- if (schema && obj && ucl_object_type (schema) == UCL_OBJECT) {
- if (lua_gettop (L) > 2) {
- if (lua_type (L, 3) == LUA_TSTRING) {
- path = lua_tostring (L, 3);
- if (path[0] == '#') {
- path++;
- }
- }
- else if (lua_type (L, 3) == LUA_TUSERDATA || lua_type (L, 3) ==
- LUA_TTABLE) {
- /* External refs */
- ext_refs = lua_ucl_object_get (L, 3);
- }
-
- if (lua_gettop (L) > 3) {
- if (lua_type (L, 4) == LUA_TUSERDATA || lua_type (L, 4) ==
- LUA_TTABLE) {
- /* External refs */
- ext_refs = lua_ucl_object_get (L, 4);
- }
- }
- }
-
- if (path) {
- schema_elt = ucl_object_lookup_path_char (schema, path, '/');
- }
- else {
- /* Use the top object */
- schema_elt = schema;
- }
-
- if (schema_elt) {
- res = ucl_object_validate_root_ext (schema_elt, obj, schema,
- ext_refs, &err);
-
- if (res) {
- lua_pushboolean (L, res);
- lua_pushnil (L);
-
- if (ext_refs) {
- lua_ucl_push_opaque (L, ext_refs);
- }
- }
- else {
- lua_pushboolean (L, res);
- lua_pushfstring (L, "validation error: %s", err.msg);
-
- if (ext_refs) {
- lua_ucl_push_opaque (L, ext_refs);
- }
- }
- }
- else {
- lua_pushboolean (L, res);
-
- lua_pushfstring (L, "cannot find the requested path: %s", path);
-
- if (ext_refs) {
- lua_ucl_push_opaque (L, ext_refs);
- }
- }
- }
- else {
- lua_pushboolean (L, res);
- lua_pushstring (L, "invalid object or schema");
- }
-
- if (ext_refs) {
- return 3;
- }
-
- return 2;
- }
-
- static int
- lua_ucl_object_gc (lua_State *L)
- {
- ucl_object_t *obj;
-
- obj = lua_ucl_object_get (L, 1);
-
- ucl_object_unref (obj);
-
- return 0;
- }
-
- static void
- lua_ucl_parser_mt (lua_State *L)
- {
- luaL_newmetatable (L, PARSER_META);
-
- lua_pushvalue(L, -1);
- lua_setfield(L, -2, "__index");
-
- lua_pushcfunction (L, lua_ucl_parser_gc);
- lua_setfield (L, -2, "__gc");
-
- lua_pushcfunction (L, lua_ucl_parser_parse_file);
- lua_setfield (L, -2, "parse_file");
-
- lua_pushcfunction (L, lua_ucl_parser_parse_string);
- lua_setfield (L, -2, "parse_string");
-
- lua_pushcfunction (L, lua_ucl_parser_parse_text);
- lua_setfield (L, -2, "parse_text");
-
- lua_pushcfunction (L, lua_ucl_parser_register_variable);
- lua_setfield (L, -2, "register_variable");
-
- lua_pushcfunction (L, lua_ucl_parser_register_variables);
- lua_setfield (L, -2, "register_variables");
-
- lua_pushcfunction (L, lua_ucl_parser_get_object);
- lua_setfield (L, -2, "get_object");
-
- lua_pushcfunction (L, lua_ucl_parser_get_object_wrapped);
- lua_setfield (L, -2, "get_object_wrapped");
-
- lua_pushcfunction (L, lua_ucl_parser_validate);
- lua_setfield (L, -2, "validate");
-
- lua_pop (L, 1);
- }
-
- static void
- lua_ucl_object_mt (lua_State *L)
- {
- luaL_newmetatable (L, OBJECT_META);
-
- lua_pushvalue(L, -1);
- lua_setfield(L, -2, "__index");
-
- lua_pushcfunction (L, lua_ucl_object_gc);
- lua_setfield (L, -2, "__gc");
-
- lua_pushcfunction (L, lua_ucl_object_tostring);
- lua_setfield (L, -2, "__tostring");
-
- lua_pushcfunction (L, lua_ucl_object_tostring);
- lua_setfield (L, -2, "tostring");
-
- lua_pushcfunction (L, lua_ucl_object_unwrap);
- lua_setfield (L, -2, "unwrap");
-
- lua_pushcfunction (L, lua_ucl_object_unwrap);
- lua_setfield (L, -2, "tolua");
-
- lua_pushcfunction (L, lua_ucl_object_validate);
- lua_setfield (L, -2, "validate");
-
- lua_pushstring (L, OBJECT_META);
- lua_setfield (L, -2, "class");
-
- lua_pop (L, 1);
- }
-
- static void
- lua_ucl_types_mt (lua_State *L)
- {
- luaL_newmetatable (L, UCL_OBJECT_TYPE_META);
-
- lua_pushcfunction (L, lua_ucl_object_tostring);
- lua_setfield (L, -2, "__tostring");
-
- lua_pushcfunction (L, lua_ucl_object_tostring);
- lua_setfield (L, -2, "tostring");
-
- lua_pushstring (L, UCL_OBJECT_TYPE_META);
- lua_setfield (L, -2, "class");
-
- lua_pop (L, 1);
-
- luaL_newmetatable (L, UCL_ARRAY_TYPE_META);
-
- lua_pushcfunction (L, lua_ucl_object_tostring);
- lua_setfield (L, -2, "__tostring");
-
- lua_pushcfunction (L, lua_ucl_object_tostring);
- lua_setfield (L, -2, "tostring");
-
- lua_pushstring (L, UCL_ARRAY_TYPE_META);
- lua_setfield (L, -2, "class");
-
- lua_pop (L, 1);
-
- luaL_newmetatable (L, UCL_IMPL_ARRAY_TYPE_META);
-
- lua_pushcfunction (L, lua_ucl_object_tostring);
- lua_setfield (L, -2, "__tostring");
-
- lua_pushcfunction (L, lua_ucl_object_tostring);
- lua_setfield (L, -2, "tostring");
-
- lua_pushstring (L, UCL_IMPL_ARRAY_TYPE_META);
- lua_setfield (L, -2, "class");
-
- lua_pop (L, 1);
- }
-
- static int
- lua_ucl_to_json (lua_State *L)
- {
- ucl_object_t *obj;
- int format = UCL_EMIT_JSON;
-
- if (lua_gettop (L) > 1) {
- if (lua_toboolean (L, 2)) {
- format = UCL_EMIT_JSON_COMPACT;
- }
- }
-
- obj = ucl_object_lua_import (L, 1);
- if (obj != NULL) {
- lua_ucl_to_string (L, obj, format);
- ucl_object_unref (obj);
- }
- else {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
- static int
- lua_ucl_to_config (lua_State *L)
- {
- ucl_object_t *obj;
-
- obj = ucl_object_lua_import (L, 1);
- if (obj != NULL) {
- lua_ucl_to_string (L, obj, UCL_EMIT_CONFIG);
- ucl_object_unref (obj);
- }
- else {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
- /***
- * @function ucl.to_format(var, format)
- * Converts lua variable `var` to the specified `format`. Formats supported are:
- *
- * - `json` - fine printed json
- * - `json-compact` - compacted json
- * - `config` - fine printed configuration
- * - `ucl` - same as `config`
- * - `yaml` - embedded yaml
- *
- * If `var` contains function, they are called during output formatting and if
- * they return string value, then this value is used for output.
- * @param {variant} var any sort of lua variable (if userdata then metafield `__to_ucl` is searched for output)
- * @param {string} format any available format
- * @return {string} string representation of `var` in the specific `format`.
- * @example
- local table = {
- str = 'value',
- num = 100500,
- null = ucl.null,
- func = function ()
- return 'huh'
- end
- }
-
- print(ucl.to_format(table, 'ucl'))
- -- Output:
- --[[
- num = 100500;
- str = "value";
- null = null;
- func = "huh";
- --]]
- */
- static int
- lua_ucl_to_format (lua_State *L)
- {
- ucl_object_t *obj;
- int format = UCL_EMIT_JSON;
- bool sort = false;
-
- if (lua_gettop (L) > 1) {
- if (lua_type (L, 2) == LUA_TNUMBER) {
- format = lua_tonumber (L, 2);
- if (format < 0 || format >= UCL_EMIT_YAML) {
- lua_pushnil (L);
- return 1;
- }
- }
- else if (lua_type (L, 2) == LUA_TSTRING) {
- const char *strtype = lua_tostring (L, 2);
-
- if (strcasecmp (strtype, "json") == 0) {
- format = UCL_EMIT_JSON;
- }
- else if (strcasecmp (strtype, "json-compact") == 0) {
- format = UCL_EMIT_JSON_COMPACT;
- }
- else if (strcasecmp (strtype, "yaml") == 0) {
- format = UCL_EMIT_YAML;
- }
- else if (strcasecmp (strtype, "config") == 0 ||
- strcasecmp (strtype, "ucl") == 0) {
- format = UCL_EMIT_CONFIG;
- }
- else if (strcasecmp (strtype, "msgpack") == 0) {
- format = UCL_EMIT_MSGPACK;
- }
- }
-
- if (lua_isboolean (L, 3)) {
- sort = lua_toboolean (L, 3);
- }
- }
-
- obj = ucl_object_lua_import (L, 1);
-
- if (obj != NULL) {
-
- if (sort) {
- if (ucl_object_type (obj) == UCL_OBJECT) {
- ucl_object_sort_keys (obj, UCL_SORT_KEYS_RECURSIVE);
- }
- }
-
- lua_ucl_to_string (L, obj, format);
- ucl_object_unref (obj);
- }
- else {
- lua_pushnil (L);
- }
-
- return 1;
- }
-
- static int
- lua_ucl_null_tostring (lua_State* L)
- {
- lua_pushstring (L, "null");
- return 1;
- }
-
- static void
- lua_ucl_null_mt (lua_State *L)
- {
- luaL_newmetatable (L, NULL_META);
-
- lua_pushcfunction (L, lua_ucl_null_tostring);
- lua_setfield (L, -2, "__tostring");
-
- lua_pop (L, 1);
- }
-
- int
- luaopen_ucl (lua_State *L)
- {
- lua_ucl_parser_mt (L);
- lua_ucl_null_mt (L);
- lua_ucl_object_mt (L);
- lua_ucl_types_mt (L);
-
- /* Create the refs weak table: */
- lua_createtable (L, 0, 2);
- lua_pushliteral (L, "v"); /* tbl, "v" */
- lua_setfield (L, -2, "__mode");
- lua_pushvalue (L, -1); /* tbl, tbl */
- lua_setmetatable (L, -2); /* tbl */
- lua_setfield (L, LUA_REGISTRYINDEX, "ucl.refs");
-
- lua_newtable (L);
-
- lua_pushcfunction (L, lua_ucl_parser_init);
- lua_setfield (L, -2, "parser");
-
- lua_pushcfunction (L, lua_ucl_to_json);
- lua_setfield (L, -2, "to_json");
-
- lua_pushcfunction (L, lua_ucl_to_config);
- lua_setfield (L, -2, "to_config");
-
- lua_pushcfunction (L, lua_ucl_to_format);
- lua_setfield (L, -2, "to_format");
-
- ucl_null = lua_newuserdata (L, 0);
- luaL_getmetatable (L, NULL_META);
- lua_setmetatable (L, -2);
-
- lua_pushvalue (L, -1);
- lua_setfield (L, LUA_REGISTRYINDEX, "ucl.null");
-
- lua_setfield (L, -2, "null");
-
- return 1;
- }
-
- struct ucl_lua_funcdata*
- ucl_object_toclosure (const ucl_object_t *obj)
- {
- if (obj == NULL || obj->type != UCL_USERDATA) {
- return NULL;
- }
-
- return (struct ucl_lua_funcdata*)obj->value.ud;
- }
|