static int
protected_gk_assign_lcores(lua_State *l)
{
uint32_t ctypeid;
struct gk_config *gk_conf;
lua_Integer i, n;
unsigned int *lcores;
gk_conf = *(struct gk_config **)
luaL_checkcdata(l, 1, &ctypeid, CTYPE_STRUCT_GK_CONFIG_PTR);
n = lua_objlen(l, 2);
lcores = *(unsigned int **)lua_touserdata(l, 3);
for (i = 1; i <= n; i++) {
lua_pushinteger(l, i); /* Push i. */
lua_gettable(l, 2); /* Pop i, Push t[i]. */
/* Check that t[i] is a number. */
if (!lua_isnumber(l, -1))
luaL_error(l, "Index %i is not a number", i);
lcores[i - 1] = lua_tointeger(l, -1);
lua_pop(l, 1); /* Pop t[i]. */
}
gk_conf->lcores = lcores;
gk_conf->num_lcores = n;
return 0; /* No results. */
}
static inline int
luaL_convertint64(lua_State *L, int idx, bool unsignd, int64_t *result)
{
uint32_t ctypeid;
void *cdata;
/*
* This code looks mostly like luaL_tofield(), but has less
* cases and optimized for numbers.
*/
switch (lua_type(L, idx)) {
case LUA_TNUMBER:
*result = lua_tointeger(L, idx);
return 0;
case LUA_TCDATA:
cdata = luaL_checkcdata(L, 1, &ctypeid);
switch (ctypeid) {
case CTID_CCHAR:
case CTID_INT8:
*result = *(int8_t *) cdata;
return 0;
case CTID_INT16:
*result = *(int16_t *) cdata;
return 0;
case CTID_INT32:
*result = *(int32_t *) cdata;
return 0;
case CTID_INT64:
*result = *(int64_t *) cdata;
return 0;
case CTID_UINT8:
*result = *(uint8_t *) cdata;
return 0;
case CTID_UINT16:
*result = *(uint16_t *) cdata;
return 0;
case CTID_UINT32:
*result = *(uint32_t *) cdata;
return 0;
case CTID_UINT64:
*result = *(uint64_t *) cdata;
return 0;
}
*result = 0;
return -1;
case LUA_TSTRING:
{
const char *arg = luaL_checkstring(L, idx);
char *arge;
errno = 0;
*result = (unsignd ? (long long) strtoull(arg, &arge, 10) :
strtoll(arg, &arge, 10));
if (errno == 0 && arge != arg)
return 0;
return 1;
}
}
*result = 0;
return -1;
}
box_tuple_t *
luaT_istuple(struct lua_State *L, int narg)
{
assert(CTID_STRUCT_TUPLE_REF != 0);
uint32_t ctypeid;
void *data;
if (lua_type(L, narg) != LUA_TCDATA)
return NULL;
data = luaL_checkcdata(L, narg, &ctypeid);
if (ctypeid != CTID_STRUCT_TUPLE_REF)
return NULL;
return *(struct tuple **) data;
}
struct error *
luaL_iserror(struct lua_State *L, int narg)
{
assert(CTID_CONST_STRUCT_ERROR_REF != 0);
if (lua_type(L, narg) != LUA_TCDATA)
return NULL;
uint32_t ctypeid;
void *data = luaL_checkcdata(L, narg, &ctypeid);
if (ctypeid != (uint32_t) CTID_CONST_STRUCT_ERROR_REF)
return NULL;
struct error *e = *(struct error **) data;
assert(e->refs);
return e;
}
static int
l_gk_assign_lcores(lua_State *l)
{
uint32_t ctypeid;
lua_Integer n;
unsigned int *lcores, **ud;
uint32_t correct_ctypeid = luaL_get_ctypeid(l,
CTYPE_STRUCT_GK_CONFIG_PTR);
/* First argument must be of type CTYPE_STRUCT_GK_CONFIG_PTR. */
luaL_checkcdata(l, 1, &ctypeid, CTYPE_STRUCT_GK_CONFIG_PTR);
if (ctypeid != correct_ctypeid)
luaL_error(l, "Expected `%s' as first argument",
CTYPE_STRUCT_GK_CONFIG_PTR);
/* Second argument must be a table. */
luaL_checktype(l, 2, LUA_TTABLE);
n = lua_objlen(l, 2); /* Get size of the table. */
if (n <= 0)
return 0; /* No results. */
ud = lua_newuserdata(l, sizeof(lcores));
lua_pushcfunction(l, protected_gk_assign_lcores);
lua_insert(l, 1);
lcores = rte_malloc("gk_conf.lcores", n * sizeof(*lcores), 0);
if (lcores == NULL)
luaL_error(l, "DPDK has run out memory");
*ud = lcores;
/* lua_pcall() is used here to avoid leaking @lcores. */
if (lua_pcall(l, 3, 0, 0)) {
rte_free(lcores);
lua_error(l);
}
return 0;
}
uint32_t
luaL_ctypeid(struct lua_State *L, const char *ctypename)
{
int idx = lua_gettop(L);
/* This function calls ffi.typeof to determine CDataType */
/* Get ffi.typeof function */
luaL_loadstring(L, "return require('ffi').typeof");
lua_call(L, 0, 1);
/* FFI must exist */
assert(lua_gettop(L) == idx + 1 && lua_isfunction(L, idx + 1));
/* Push the first argument to ffi.typeof */
lua_pushstring(L, ctypename);
/* Call ffi.typeof() */
lua_call(L, 1, 1);
/* Returned type must be LUA_TCDATA with CTID_CTYPEID */
uint32_t ctypetypeid;
CTypeID ctypeid = *(CTypeID *)luaL_checkcdata(L, idx + 1, &ctypetypeid);
assert(ctypetypeid == CTID_CTYPEID);
lua_settop(L, idx);
return ctypeid;
}
void
luaL_tofield(struct lua_State *L, struct luaL_serializer *cfg, int index,
struct luaL_field *field)
{
if (index < 0)
index = lua_gettop(L) + index + 1;
double num;
double intpart;
size_t size;
#define CHECK_NUMBER(x) ({\
if (!isfinite(x) && !cfg->encode_invalid_numbers) { \
if (!cfg->encode_invalid_as_nil) \
luaL_error(L, "number must not be NaN or Inf"); \
field->type = MP_NIL; \
}})
switch (lua_type(L, index)) {
case LUA_TNUMBER:
num = lua_tonumber(L, index);
if (isfinite(num) && modf(num, &intpart) != 0.0) {
field->type = MP_DOUBLE;
field->dval = num;
} else if (num >= 0 && num <= UINT64_MAX) {
field->type = MP_UINT;
field->ival = (uint64_t) num;
} else if (num >= INT64_MIN && num <= INT64_MAX) {
field->type = MP_INT;
field->ival = (int64_t) num;
} else {
field->type = MP_DOUBLE;
field->dval = num;
CHECK_NUMBER(num);
}
return;
case LUA_TCDATA:
{
uint32_t ctypeid = 0;
void *cdata = luaL_checkcdata(L, index, &ctypeid);
int64_t ival;
switch (ctypeid) {
case CTID_BOOL:
field->type = MP_BOOL;
field->bval = *(bool*) cdata;
return;
case CTID_CCHAR:
case CTID_INT8:
ival = *(int8_t *) cdata;
field->type = (ival >= 0) ? MP_UINT : MP_INT;
field->ival = ival;
return;
case CTID_INT16:
ival = *(int16_t *) cdata;
field->type = (ival >= 0) ? MP_UINT : MP_INT;
field->ival = ival;
return;
case CTID_INT32:
ival = *(int32_t *) cdata;
field->type = (ival >= 0) ? MP_UINT : MP_INT;
field->ival = ival;
return;
case CTID_INT64:
ival = *(int64_t *) cdata;
field->type = (ival >= 0) ? MP_UINT : MP_INT;
field->ival = ival;
return;
case CTID_UINT8:
field->type = MP_UINT;
field->ival = *(uint8_t *) cdata;
return;
case CTID_UINT16:
field->type = MP_UINT;
field->ival = *(uint16_t *) cdata;
return;
case CTID_UINT32:
field->type = MP_UINT;
field->ival = *(uint32_t *) cdata;
return;
case CTID_UINT64:
field->type = MP_UINT;
field->ival = *(uint64_t *) cdata;
return;
case CTID_FLOAT:
field->type = MP_FLOAT;
field->fval = *(float *) cdata;
CHECK_NUMBER(field->fval);
return;
case CTID_DOUBLE:
field->type = MP_DOUBLE;
field->dval = *(double *) cdata;
CHECK_NUMBER(field->dval);
return;
case CTID_P_CVOID:
case CTID_P_VOID:
if (*(void **) cdata == NULL) {
field->type = MP_NIL;
return;
}
/* Fall through */
default:
field->type = MP_EXT;
return;
}
return;
}
case LUA_TBOOLEAN:
field->type = MP_BOOL;
field->bval = lua_toboolean(L, index);
return;
case LUA_TNIL:
field->type = MP_NIL;
return;
case LUA_TSTRING:
field->sval.data = lua_tolstring(L, index, &size);
field->sval.len = (uint32_t) size;
field->type = MP_STR;
return;
case LUA_TTABLE:
{
field->compact = false;
lua_field_inspect_table(L, cfg, index, field);
return;
}
case LUA_TLIGHTUSERDATA:
case LUA_TUSERDATA:
field->sval.data = NULL;
field->sval.len = 0;
if (lua_touserdata(L, index) == NULL) {
field->type = MP_NIL;
return;
}
/* Fall through */
default:
field->type = MP_EXT;
return;
}
#undef CHECK_NUMBER
}
