ardupilot/libraries/AP_Scripting/generator/src/main.c

1793 lines
56 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <stdarg.h>
#include <assert.h>
#include <string.h>
#include <unistd.h>
#include <getopt.h>
char keyword_alias[] = "alias";
char keyword_comment[] = "--";
char keyword_depends[] = "depends";
char keyword_enum[] = "enum";
char keyword_field[] = "field";
char keyword_include[] = "include";
char keyword_method[] = "method";
char keyword_operator[] = "operator";
char keyword_read[] = "read";
char keyword_semaphore[] = "semaphore";
char keyword_singleton[] = "singleton";
char keyword_userdata[] = "userdata";
char keyword_write[] = "write";
// attributes (should include the leading ' )
char keyword_attr_enum[] = "'enum";
char keyword_attr_literal[] = "'literal";
char keyword_attr_null[] = "'Null";
// type keywords
char keyword_boolean[] = "boolean";
char keyword_float[] = "float";
char keyword_int8_t[] = "int8_t";
char keyword_int16_t[] = "int16_t";
char keyword_int32_t[] = "int32_t";
char keyword_string[] = "string";
char keyword_uint8_t[] = "uint8_t";
char keyword_uint16_t[] = "uint16_t";
char keyword_uint32_t[] = "uint32_t";
char keyword_void[] = "void";
enum error_codes {
ERROR_OUT_OF_MEMORY = 1, // ran out of memory
ERROR_HEADER = 2, // header keyword not followed by a header to include
ERROR_UNKNOWN_KEYWORD = 3, // a keyword we didn't know how to handle
ERROR_USERDATA = 4, // userdata
ERROR_INTERNAL = 5, // internal error of some form
ERROR_GENERAL = 6, // general error
ERROR_SINGLETON = 7, // singletons
ERROR_DEPENDS = 8, // dependencies
};
struct header {
struct header *next;
char *name; // name of the header to include (not sanatized)
int line; // line of the file declared on
};
struct generator_state {
char line[1<<14];
int line_num; // current line read in
int token_num; // current token on the current line
char *token;
};
FILE *description;
FILE *header;
FILE *source;
static struct generator_state state = {};
static struct header * headers = NULL;
enum trace_level {
TRACE_TOKENS = (1 << 0),
TRACE_HEADER = (1 << 1),
TRACE_GENERAL = (1 << 2),
TRACE_USERDATA = (1 << 3),
TRACE_SINGLETON = (1 << 4),
TRACE_DEPENDS = (1 << 5),
};
enum access_flags {
ACCESS_FLAG_READ = (1 << 0),
ACCESS_FLAG_WRITE = (1 << 1),
};
enum field_type {
TYPE_BOOLEAN = 0,
TYPE_FLOAT,
TYPE_INT8_T,
TYPE_INT16_T,
TYPE_INT32_T,
TYPE_UINT8_T,
TYPE_UINT16_T,
TYPE_UINT32_T,
TYPE_NONE,
TYPE_STRING,
TYPE_ENUM,
TYPE_LITERAL,
TYPE_USERDATA,
};
const char * type_labels[TYPE_USERDATA + 1] = { "bool",
"float",
"int8_t",
"int16_t",
"int32_t",
"uint8_t",
"uint16_t",
"void",
"string",
"enum",
"userdata",
};
enum operator_type {
OP_ADD = (1U << 0),
OP_SUB = (1U << 1),
OP_MUL = (1U << 2),
OP_DIV = (1U << 3),
OP_LAST
};
enum access_type {
ACCESS_VALUE = 0,
ACCESS_REFERENCE,
};
struct range_check {
// store the requested range check as a string
// we will check that it's a numeric form of some type, but keep it as a string rather then a casted version
char *low;
char *high;
};
enum type_flags {
TYPE_FLAGS_NULLABLE = (1U << 1),
TYPE_FLAGS_ENUM = (1U << 2),
};
struct type {
struct range_check *range;
enum field_type type;
enum access_type access;
uint32_t flags;
union {
char *userdata_name;
char *enum_name;
char *literal;
} data;
};
int TRACE_LEVEL = 0;
void trace(const int trace, const char *message, ...) {
if (trace & TRACE_LEVEL) {
char * fmt = malloc(strlen(message)+1024);
if (fmt == NULL) {
exit(ERROR_OUT_OF_MEMORY);
}
sprintf(fmt, "TRACE: %s\n", message);
va_list args;
va_start(args, message);
vfprintf(stderr, fmt, args);
va_end(args);
}
}
void error(const int code, const char *message, ...) {
char * fmt = malloc(strlen(message)+1024);
if (fmt == NULL) {
exit(ERROR_OUT_OF_MEMORY);
}
if (state.line_num >= 0) {
sprintf(fmt, "Error (line %d): %s\n", state.line_num, message);
} else {
sprintf(fmt, "Error: %s\n", message);
}
va_list args;
va_start(args, message);
vfprintf(stderr, fmt, args);
va_end(args);
exit(code);
}
char *token_delimiters = " \n";
char * next_token(void) {
state.token = strtok(NULL, token_delimiters);
state.token_num++;
trace(TRACE_TOKENS, "Token %d:%d %s", state.line_num, state.token_num, state.token);
if ((state.token!= NULL) && (strcmp(state.token, keyword_comment) == 0)) {
trace(TRACE_TOKENS, "Detected comment %d", state.line_num);
while (next_token()) {} // burn all the symbols
}
return state.token;
}
char * start_line(void) {
while (fgets(state.line, sizeof(state.line)/sizeof(state.line[0]), description) != NULL) {//state.line = readline(NULL))) {
state.line_num++;
state.token = strtok(state.line, token_delimiters);
state.token_num = 1;
trace(TRACE_TOKENS, "Token %d:%d %s", state.line_num, state.token_num, state.token);
if (state.token != NULL) {
break;
}
}
return state.token;
}
// thin wrapper for malloc that exits if we can't allocate memory, and memsets the allocated chunk
void *allocate(const size_t size) {
void *data = malloc(size);
if (data == NULL) {
error(ERROR_OUT_OF_MEMORY, "Out of memory.");
} else {
memset(data, 0, size);
}
return data;
}
void handle_header(void) {
trace(TRACE_HEADER, "Parsing a header");
// find the new header
char * name = next_token();
if (name == NULL) {
error(ERROR_HEADER, "Header must be followed by the name of the header to include");
}
// search for duplicates
struct header *node = headers;
while (node != NULL && strcmp(node->name, name)) {
node = node->next;
}
if (node != NULL) {
error(ERROR_HEADER, "Header %s was already included on line %d", name, node->line);
}
// add to the list of headers
node = (struct header *)allocate(sizeof(struct header));
node->next = headers;
node->line = state.line_num;
node->name = (char *)allocate(strlen(name) + 1);
strcpy(node->name, name);
headers = node;
trace(TRACE_HEADER, "Added header %s", name);
// ensure no more tokens on the line
if (next_token()) {
error(ERROR_HEADER, "Header contained an unexpected extra token: %s", state.token);
}
}
enum userdata_type {
UD_USERDATA,
UD_SINGLETON,
};
struct argument {
struct argument * next;
struct type type;
int line_num; // line read from
int token_num; // token number on the line
};
struct method {
struct method * next;
char *name;
int line; // line declared on
struct type return_type;
struct argument * arguments;
uint32_t flags; // filled out with TYPE_FLAGS
};
struct userdata_field {
struct userdata_field * next;
char * name;
struct type type; // field type, points to a string
int line; // line declared on
unsigned int access_flags;
};
enum userdata_flags {
UD_FLAG_SEMAPHORE = (1U << 0),
};
struct userdata_enum {
struct userdata_enum * next;
char * name; // enum name
};
struct userdata {
struct userdata * next;
char *name; // name of the C++ singleton
char *alias; // (optional) used for scripting access
struct userdata_field *fields;
struct method *methods;
struct userdata_enum *enums;
enum userdata_type ud_type;
uint32_t operations; // bitset of enum operation_types
int flags; // flags from the userdata_flags enum
};
static struct userdata *parsed_userdata = NULL;
struct dependency {
struct dependency * next;
char *symbol; // dependency symbol to check
char *value; // value to target
char *error_msg; // message if the check fails
};
static struct dependency *parsed_dependencies = NULL;
// lazy helper that allocates a storage buffer and does strcpy for us
void string_copy(char **dest, const char * src) {
*dest = (char *)allocate(strlen(src) + 1);
strcpy(*dest, src);
}
struct range_check *parse_range_check(enum field_type type) {
char * low = next_token();
if (low == NULL) {
error(ERROR_USERDATA, "Missing low value for a range check (type: %s)", type_labels[type]);
}
trace(TRACE_TOKENS, "Range check: Low: %s", low);
char * high = next_token();
if (high == NULL) {
error(ERROR_USERDATA, "Missing high value for a range check");
}
trace(TRACE_TOKENS, "Range check: High: %s", high);
struct range_check *check = allocate(sizeof(struct range_check));
string_copy(&(check->low), low);
string_copy(&(check->high), high);
return check;
}
// parses one or more access flags, leaves the token on the first non access token
// throws an error if no flags were found
unsigned int parse_access_flags(struct type * type) {
unsigned int flags = 0;
next_token();
while(state.token != NULL) {
trace(TRACE_TOKENS, "Possible access: %s", state.token);
if (strcmp(state.token, keyword_read) == 0) {
flags |= ACCESS_FLAG_READ;
} else if (strcmp(state.token, keyword_write) == 0) {
flags |= ACCESS_FLAG_WRITE;
switch (type->type) {
case TYPE_FLOAT:
case TYPE_INT8_T:
case TYPE_INT16_T:
case TYPE_INT32_T:
case TYPE_UINT8_T:
case TYPE_UINT16_T:
case TYPE_UINT32_T:
case TYPE_ENUM:
type->range = parse_range_check(type->type);
break;
case TYPE_USERDATA:
case TYPE_BOOLEAN:
case TYPE_STRING:
case TYPE_LITERAL:
// a range check is illogical
break;
case TYPE_NONE:
error(ERROR_INTERNAL, "Can't access a NONE type");
}
} else {
break;
}
next_token();
}
trace(TRACE_TOKENS, "Parsed access flags: 0x%x", flags);
if (flags == 0) {
error(ERROR_USERDATA, "Expected to find an access specifier");
}
return flags;
}
#define TRUE 1
#define FALSE 0
enum type_restriction {
TYPE_RESTRICTION_NONE = 0,
TYPE_RESTRICTION_OPTIONAL = (1U << 1),
TYPE_RESTRICTION_NOT_NULLABLE = (1U << 2),
};
enum range_check_type {
RANGE_CHECK_NONE,
RANGE_CHECK_MANDATORY,
};
int parse_type(struct type *type, const uint32_t restrictions, enum range_check_type range_type) {
char *data_type = next_token();
if (data_type == NULL) {
if (restrictions & TYPE_RESTRICTION_OPTIONAL) {
return FALSE;
} else {
error(ERROR_USERDATA, "Data type must be specified");
}
}
if (data_type[0] == '&') {
type->access = ACCESS_REFERENCE;
data_type++; // drop the reference character
} else {
type->access = ACCESS_VALUE;
}
char *attribute = strchr(data_type, '\'');
if (attribute != NULL) {
if (strcmp(attribute, keyword_attr_enum) == 0) {
type->flags |= TYPE_FLAGS_ENUM;
} else if (strcmp(attribute, keyword_attr_literal) == 0) {
type->type = TYPE_LITERAL;
} else if (strcmp(attribute, keyword_attr_null) == 0) {
if (restrictions & TYPE_RESTRICTION_NOT_NULLABLE) {
error(ERROR_USERDATA, "%s is not nullable in this context", data_type);
}
type->flags |= TYPE_FLAGS_NULLABLE;
} else {
error(ERROR_USERDATA, "Unknown attribute: %s", attribute);
}
attribute[0] = 0;
}
if (strcmp(data_type, keyword_boolean) == 0) {
type->type = TYPE_BOOLEAN;
} else if (strcmp(data_type, keyword_float) == 0) {
type->type = TYPE_FLOAT;
} else if (strcmp(data_type, keyword_int8_t) == 0) {
type->type = TYPE_INT8_T;
} else if (strcmp(data_type, keyword_int16_t) == 0) {
type->type = TYPE_INT16_T;
} else if (strcmp(data_type, keyword_int32_t) == 0) {
type->type = TYPE_INT32_T;
} else if (strcmp(data_type, keyword_uint8_t) == 0) {
type->type = TYPE_UINT8_T;
} else if (strcmp(data_type, keyword_uint16_t) == 0) {
type->type = TYPE_UINT16_T;
} else if (strcmp(data_type, keyword_uint32_t) == 0) {
type->type = TYPE_UINT32_T;
} else if (strcmp(data_type, keyword_string) == 0) {
type->type = TYPE_STRING;
} else if (strcmp(data_type, keyword_void) == 0) {
type->type = TYPE_NONE;
} else if (type->flags & TYPE_FLAGS_ENUM) {
type->type = TYPE_ENUM;
string_copy(&(type->data.enum_name), data_type);
} else if (type->type == TYPE_LITERAL) {
string_copy(&(type->data.literal), data_type);
} else {
// assume that this is a user data, we can't validate this until later though
type->type = TYPE_USERDATA;
string_copy(&(type->data.userdata_name), data_type);
}
// sanity check that only supported types are nullable
if (type->flags & TYPE_FLAGS_NULLABLE) {
// a switch is a very verbose way to do this, but forces users to consider new types added
switch (type->type) {
case TYPE_FLOAT:
case TYPE_INT8_T:
case TYPE_INT16_T:
case TYPE_INT32_T:
case TYPE_UINT8_T:
case TYPE_UINT16_T:
case TYPE_UINT32_T:
case TYPE_BOOLEAN:
case TYPE_STRING:
case TYPE_ENUM:
case TYPE_USERDATA:
break;
case TYPE_LITERAL:
case TYPE_NONE:
error(ERROR_USERDATA, "%s types cannot be nullable", data_type);
break;
}
}
// add range checks, unless disabled or a nullable type
if (range_type != RANGE_CHECK_NONE && !(type->flags & TYPE_FLAGS_NULLABLE)) {
switch (type->type) {
case TYPE_FLOAT:
case TYPE_INT8_T:
case TYPE_INT16_T:
case TYPE_INT32_T:
case TYPE_UINT8_T:
case TYPE_UINT16_T:
case TYPE_UINT32_T:
case TYPE_ENUM:
type->range = parse_range_check(type->type);
break;
case TYPE_BOOLEAN:
case TYPE_NONE:
case TYPE_STRING:
case TYPE_USERDATA:
case TYPE_LITERAL:
// no sane range checks, so we can ignore this
break;
}
}
return TRUE;
}
void handle_userdata_enum(struct userdata *data) {
trace(TRACE_USERDATA, "Adding a userdata enum");
char * enum_name;
while ((enum_name = next_token()) != NULL) {
trace(TRACE_USERDATA, "Adding enum %s", enum_name);
struct userdata_enum *ud_enum = (struct userdata_enum *) allocate(sizeof(struct userdata_enum));
ud_enum->next = data->enums;
string_copy(&(ud_enum->name), enum_name);
data->enums = ud_enum;
}
}
void handle_userdata_field(struct userdata *data) {
trace(TRACE_USERDATA, "Adding a userdata field");
// find the field name
char * field_name = next_token();
if (field_name == NULL) {
error(ERROR_USERDATA, "Missing a field name for userdata %s", data->name);
}
struct userdata_field * field = data->fields;
while (field != NULL && strcmp(field->name, field_name)) {
field = field-> next;
}
if (field != NULL) {
error(ERROR_USERDATA, "Field %s already exsists in userdata %s (declared on %d)", field_name, data->name, field->line);
}
trace(TRACE_USERDATA, "Adding field %s", field_name);
field = (struct userdata_field *)allocate(sizeof(struct userdata_field));
field->next = data->fields;
data->fields = field;
field->line = state.line_num;
string_copy(&(field->name), field_name);
parse_type(&(field->type), TYPE_RESTRICTION_NOT_NULLABLE, RANGE_CHECK_NONE);
field->access_flags = parse_access_flags(&(field->type));
}
void handle_method(char *parent_name, struct method **methods) {
trace(TRACE_USERDATA, "Adding a method");
// find the field name
char * name = next_token();
if (name == NULL) {
error(ERROR_USERDATA, "Missing method name for %s", parent_name);
}
struct method * method = *methods;
while (method != NULL && strcmp(method->name, name)) {
method = method-> next;
}
if (method != NULL) {
error(ERROR_USERDATA, "Method %s already exsists for %s (declared on %d)", name, parent_name, method->line);
}
trace(TRACE_USERDATA, "Adding method %s", name);
method = allocate(sizeof(struct method));
method->next = *methods;
*methods = method;
string_copy(&(method->name), name);
method->line = state.line_num;
parse_type(&(method->return_type), TYPE_RESTRICTION_NONE, RANGE_CHECK_NONE);
// iterate the arguments
struct type arg_type = {};
while (parse_type(&arg_type, TYPE_RESTRICTION_OPTIONAL, RANGE_CHECK_MANDATORY)) {
if (arg_type.type == TYPE_NONE) {
error(ERROR_USERDATA, "Can't pass an empty argument to a method");
}
if ((method->return_type.type != TYPE_BOOLEAN) && (arg_type.flags & TYPE_FLAGS_NULLABLE)) {
error(ERROR_USERDATA, "Nullable arguments are only available on a boolean method");
}
if (arg_type.flags & TYPE_FLAGS_NULLABLE) {
method->flags |= TYPE_FLAGS_NULLABLE;
}
struct argument * arg = allocate(sizeof(struct argument));
memcpy(&(arg->type), &arg_type, sizeof(struct type));
arg->line_num = state.line_num;
arg->token_num = state.token_num;
if (method->arguments == NULL) {
method->arguments = arg;
} else {
struct argument *tail = method->arguments;
while (tail->next != NULL) {
tail = tail->next;
}
tail->next = arg;
}
// reset the stack arg_type
memset(&arg_type, 0, sizeof(struct type));
}
}
void handle_operator(struct userdata *data) {
trace(TRACE_USERDATA, "Adding a operator");
if (data->ud_type != UD_USERDATA) {
error(ERROR_USERDATA, "Operators are only allowed on userdata objects");
}
char *operator = next_token();
if (operator == NULL) {
error(ERROR_USERDATA, "Needed a symbol for the operator");
}
enum operator_type operation;
if (strcmp(operator, "+") == 0) {
operation = OP_ADD;
} else if (strcmp(operator, "-") == 0) {
operation = OP_SUB;
} else if (strcmp(operator, "*") == 0) {
operation = OP_MUL;
} else if (strcmp(operator, "/") == 0) {
operation = OP_DIV;
} else {
error(ERROR_USERDATA, "Unknown operation type: %s", operator);
}
if ((data->operations) & operation) {
error(ERROR_USERDATA, "Operation %s was already defined for %s", operator, data->name);
}
trace(TRACE_USERDATA, "Adding operation %d to %s", operation, data->name);
data->operations |= operation;
if (next_token() != NULL) {
error(ERROR_USERDATA, "Extra token on operation %s", operator);
}
}
void handle_userdata(void) {
trace(TRACE_USERDATA, "Adding a userdata");
char *name = next_token();
if (name == NULL) {
error(ERROR_USERDATA, "Expected a name for the userdata");
}
struct userdata *node = parsed_userdata;
while (node != NULL && strcmp(node->name, name)) {
node = node->next;
}
if (node == NULL) {
trace(TRACE_USERDATA, "Allocating new userdata for %s", name);
node = (struct userdata *)allocate(sizeof(struct userdata));
node->ud_type = UD_USERDATA;
node->name = (char *)allocate(strlen(name) + 1);
strcpy(node->name, name);
node->next = parsed_userdata;
parsed_userdata = node;
} else {
trace(TRACE_USERDATA, "Found exsisting userdata for %s", name);
}
// read type
char *type = next_token();
if (type == NULL) {
error(ERROR_USERDATA, "Expected a access type for userdata %s", name);
}
// match type
if (strcmp(type, keyword_field) == 0) {
handle_userdata_field(node);
} else if (strcmp(type, keyword_operator) == 0) {
handle_operator(node);
} else if (strcmp(type, keyword_method) == 0) {
handle_method(node->name, &(node->methods));
} else if (strcmp(type, keyword_enum) == 0) {
handle_userdata_enum(node);
} else {
error(ERROR_USERDATA, "Unknown or unsupported type for userdata: %s", type);
}
}
struct userdata *parsed_singletons = NULL;
void handle_singleton(void) {
trace(TRACE_SINGLETON, "Adding a singleton");
char *name = next_token();
if (name == NULL) {
error(ERROR_USERDATA, "Expected a name for the singleton");
}
struct userdata *node = parsed_singletons;
while (node != NULL && strcmp(node->name, name)) {
node = node->next;
}
if (node == NULL) {
trace(TRACE_SINGLETON, "Allocating new singleton for %s", name);
node = (struct userdata *)allocate(sizeof(struct userdata));
node->ud_type = UD_SINGLETON;
node->name = (char *)allocate(strlen(name) + 1);
strcpy(node->name, name);
node->next = parsed_singletons;
parsed_singletons = node;
}
// read type
char *type = next_token();
if (type == NULL) {
error(ERROR_SINGLETON, "Expected a access type for userdata %s", name);
}
if (strcmp(type, keyword_alias) == 0) {
if (node->alias != NULL) {
error(ERROR_SINGLETON, "Alias of %s was already declared for %s", node->alias, node->name);
}
const char *alias = next_token();
if (alias == NULL) {
error(ERROR_SINGLETON, "Missing the name of the alias for %s", node->name);
}
node->alias = (char *)allocate(strlen(alias) + 1);
strcpy(node->alias, alias);
} else if (strcmp(type, keyword_semaphore) == 0) {
node->flags |= UD_FLAG_SEMAPHORE;
} else if (strcmp(type, keyword_method) == 0) {
handle_method(node->name, &(node->methods));
} else if (strcmp(type, keyword_enum) == 0) {
handle_userdata_enum(node);
} else {
error(ERROR_SINGLETON, "Singletons only support aliases, methods or semaphore keyowrds (got %s)", type);
}
// ensure no more tokens on the line
if (next_token()) {
error(ERROR_HEADER, "Singleton contained an unexpected extra token: %s", state.token);
}
}
void handle_depends(void) {
trace(TRACE_DEPENDS, "Adding a dependency");
char *symbol = next_token();
if (symbol == NULL) {
error(ERROR_DEPENDS, "Expected a name symbol for the dependency");
}
// read value
char *value = next_token();
if (value == NULL) {
error(ERROR_DEPENDS, "Expected a required value for dependency on %s", symbol);
}
char *error_msg = strtok(NULL, "");
if (error_msg == NULL) {
error(ERROR_DEPENDS, "Expected a error message for dependency on %s", symbol);
}
trace(TRACE_SINGLETON, "Allocating new dependency for %s", symbol);
struct dependency * node = (struct dependency *)allocate(sizeof(struct dependency));
node->symbol = (char *)allocate(strlen(symbol) + 1);
strcpy(node->symbol, symbol);
node->value = (char *)allocate(strlen(value) + 1);
strcpy(node->value, value);
node->error_msg = (char *)allocate(strlen(error_msg) + 1);
strcpy(node->error_msg, error_msg);
node->next = parsed_dependencies;
parsed_dependencies = node;
}
void sanity_check_userdata(void) {
struct userdata * node = parsed_userdata;
while(node) {
if ((node->fields == NULL) && (node->methods == NULL)) {
error(ERROR_USERDATA, "Userdata %s has no fields or methods", node->name);
}
node = node->next;
}
}
void emit_headers(FILE *f) {
struct header *node = headers;
while (node) {
fprintf(f, "#include <%s>\n", node->name);
node = node->next;
}
}
void emit_dependencies(FILE *f) {
struct dependency *node = parsed_dependencies;
while (node) {
fprintf(f, "#if !defined(%s) || (%s != %s)\n", node->symbol, node->symbol, node->value);
fprintf(f, " #error %s\n", node->error_msg);
fprintf(f, "#endif // !defined(%s) || (%s != %s)\n", node->symbol, node->symbol, node->value);
node = node->next;
}
}
void emit_userdata_allocators(void) {
struct userdata * node = parsed_userdata;
while (node) {
fprintf(source, "int new_%s(lua_State *L) {\n", node->name);
fprintf(source, " luaL_checkstack(L, 2, \"Out of stack\");\n"); // ensure we have sufficent stack to push the return
fprintf(source, " void *ud = lua_newuserdata(L, sizeof(%s));\n", node->name);
fprintf(source, " memset(ud, 0, sizeof(%s));\n", node->name);
fprintf(source, " new (ud) %s();\n", node->name);
fprintf(source, " luaL_getmetatable(L, \"%s\");\n", node->name);
fprintf(source, " lua_setmetatable(L, -2);\n");
fprintf(source, " return 1;\n");
fprintf(source, "}\n\n");
node = node->next;
}
}
void emit_userdata_checkers(void) {
struct userdata * node = parsed_userdata;
while (node) {
fprintf(source, "%s * check_%s(lua_State *L, int arg) {\n", node->name, node->name);
fprintf(source, " void *data = luaL_checkudata(L, arg, \"%s\");\n", node->name);
fprintf(source, " return (%s *)data;\n", node->name);
fprintf(source, "}\n\n");
node = node->next;
}
}
void emit_userdata_declarations(void) {
struct userdata * node = parsed_userdata;
while (node) {
fprintf(header, "int new_%s(lua_State *L);\n", node->name);
fprintf(header, "%s * check_%s(lua_State *L, int arg);\n", node->name, node->name);
node = node->next;
}
}
#define NULLABLE_ARG_COUNT_BASE 5000
void emit_checker(const struct type t, int arg_number, int skipped, const char *indentation, const char *name) {
assert(indentation != NULL);
if (arg_number > NULLABLE_ARG_COUNT_BASE) {
error(ERROR_INTERNAL, "Can't handle more then %d arguments to a function", NULLABLE_ARG_COUNT_BASE);
}
if (t.flags & TYPE_FLAGS_NULLABLE) {
arg_number = arg_number + NULLABLE_ARG_COUNT_BASE;
switch (t.type) {
case TYPE_BOOLEAN:
fprintf(source, "%sbool data_%d = {};\n", indentation, arg_number);
break;
case TYPE_FLOAT:
fprintf(source, "%sfloat data_%d = {};\n", indentation, arg_number);
break;
case TYPE_INT8_T:
fprintf(source, "%sint8_t data_%d = {};\n", indentation, arg_number);
break;
case TYPE_INT16_T:
fprintf(source, "%sint16_t data_%d = {};\n", indentation, arg_number);
break;
case TYPE_INT32_T:
fprintf(source, "%sint32_t data_%d = {};\n", indentation, arg_number);
break;
case TYPE_UINT8_T:
fprintf(source, "%suint8_t data_%d = {};\n", indentation, arg_number);
break;
case TYPE_UINT16_T:
fprintf(source, "%suint16_t data_%d = {};\n", indentation, arg_number);
break;
case TYPE_UINT32_T:
fprintf(source, "%suint32_t data_%d = {};\n", indentation, arg_number);
break;
case TYPE_NONE:
case TYPE_LITERAL:
return; // nothing to do here, this should potentially be checked outside of this, but it makes an easier implementation to accept it
case TYPE_STRING:
fprintf(source, "%schar * data_%d = {};\n", indentation, arg_number);
break;
case TYPE_ENUM:
fprintf(source, "%suint32_t data_%d = {};\n", indentation, arg_number);
break;
case TYPE_USERDATA:
fprintf(source, "%s%s data_%d = {};\n", indentation, t.data.userdata_name, arg_number);
break;
}
} else {
// handle this in four stages
// - figure out any relevant minimum values for range checking
// - emit a non down casted version
// - then run range checks
// - then cast down as appropriate
// select minimums
char * forced_min;
char * forced_max;
switch (t.type) {
case TYPE_FLOAT:
forced_min = "-INFINITY";
forced_max = "INFINITY";
break;
case TYPE_INT8_T:
forced_min = "INT8_MIN";
forced_max = "INT8_MAX";
break;
case TYPE_INT16_T:
forced_min = "INT16_MIN";
forced_max = "INT16_MAX";
break;
case TYPE_INT32_T:
forced_min = "INT32_MIN";
forced_max = "INT32_MAX";
break;
case TYPE_UINT8_T:
forced_min = "0";
forced_max = "UINT8_MAX";
break;
case TYPE_UINT16_T:
forced_min = "0";
forced_max = "UINT16_MAX";
break;
case TYPE_UINT32_T:
forced_min = "0U";
forced_max = "UINT32_MAX";
break;
case TYPE_ENUM:
forced_min = forced_max = NULL;
break;
case TYPE_NONE:
return; // nothing to do here, this should potentially be checked outside of this, but it makes an easier implementation to accept it
case TYPE_STRING:
case TYPE_BOOLEAN:
case TYPE_USERDATA:
case TYPE_LITERAL:
// these don't get range checked, so skip the raw_data phase
assert(t.range == NULL); // we should have caught this during the parse phase
break;
}
// non down cast
switch (t.type) {
case TYPE_FLOAT:
fprintf(source, "%sconst float raw_data_%d = luaL_checknumber(L, %d);\n", indentation, arg_number, arg_number - skipped);
break;
case TYPE_INT8_T:
case TYPE_INT16_T:
case TYPE_INT32_T:
case TYPE_UINT8_T:
case TYPE_UINT16_T:
case TYPE_ENUM:
fprintf(source, "%sconst lua_Integer raw_data_%d = luaL_checkinteger(L, %d);\n", indentation, arg_number, arg_number - skipped);
break;
case TYPE_UINT32_T:
fprintf(source, "%sconst uint32_t raw_data_%d = *check_uint32_t(L, %d);\n", indentation, arg_number, arg_number - skipped);
break;
case TYPE_NONE:
case TYPE_STRING:
case TYPE_BOOLEAN:
case TYPE_USERDATA:
case TYPE_LITERAL:
// these don't get range checked, so skip the raw_data phase
assert(t.range == NULL); // we should have caught this during the parse phase
break;
}
// range check
if (t.range != NULL) {
if ((forced_min != NULL) && (forced_max != NULL)) {
fprintf(source, "%sluaL_argcheck(L, ((raw_data_%d >= MAX(%s, %s)) && (raw_data_%d <= MIN(%s, %s))), %d, \"%s out of range\");\n",
indentation,
arg_number, t.range->low, forced_min,
arg_number, t.range->high, forced_max,
arg_number, name);
} else {
char * cast_target = "";
switch (t.type) {
case TYPE_FLOAT:
cast_target = "float";
break;
case TYPE_INT8_T:
case TYPE_INT16_T:
case TYPE_INT32_T:
case TYPE_UINT8_T:
case TYPE_UINT16_T:
case TYPE_ENUM:
cast_target = "int32_t";
break;
case TYPE_UINT32_T:
cast_target = "uint32_t";
break;
case TYPE_NONE:
case TYPE_STRING:
case TYPE_BOOLEAN:
case TYPE_USERDATA:
case TYPE_LITERAL:
assert(t.range == NULL); // we should have caught this during the parse phase
break;
}
fprintf(source, "%sluaL_argcheck(L, ((raw_data_%d >= static_cast<%s>(%s)) && (raw_data_%d <= static_cast<%s>(%s))), %d, \"%s out of range\");\n",
indentation,
arg_number, cast_target, t.range->low,
arg_number, cast_target, t.range->high,
arg_number - skipped, name);
}
}
// down cast
switch (t.type) {
case TYPE_FLOAT:
// this is a trivial transformation, trust the compiler to resolve it for us
fprintf(source, "%sconst float data_%d = raw_data_%d;\n", indentation, arg_number, arg_number);
break;
case TYPE_INT8_T:
fprintf(source, "%sconst int8_t data_%d = static_cast<int8_t>(raw_data_%d);\n", indentation, arg_number, arg_number);
break;
case TYPE_INT16_T:
fprintf(source, "%sconst int16_t data_%d = static_cast<int16_t>(raw_data_%d);\n", indentation, arg_number, arg_number);
break;
case TYPE_INT32_T:
fprintf(source, "%sconst int32_t data_%d = raw_data_%d;\n", indentation, arg_number, arg_number);
break;
case TYPE_UINT8_T:
fprintf(source, "%sconst uint8_t data_%d = static_cast<uint8_t>(raw_data_%d);\n", indentation, arg_number, arg_number);
break;
case TYPE_UINT16_T:
fprintf(source, "%sconst uint16_t data_%d = static_cast<uint16_t>(raw_data_%d);\n", indentation, arg_number, arg_number);
break;
case TYPE_UINT32_T:
fprintf(source, "%sconst uint32_t data_%d = static_cast<uint32_t>(raw_data_%d);\n", indentation, arg_number, arg_number);
break;
case TYPE_BOOLEAN:
fprintf(source, "%sconst bool data_%d = static_cast<bool>(lua_toboolean(L, %d));\n", indentation, arg_number, arg_number);
break;
case TYPE_STRING:
fprintf(source, "%sconst char * data_%d = luaL_checkstring(L, %d);\n", indentation, arg_number, arg_number);
break;
case TYPE_ENUM:
fprintf(source, "%sconst %s data_%d = static_cast<%s>(raw_data_%d);\n", indentation, t.data.enum_name, arg_number, t.data.enum_name, arg_number);
break;
case TYPE_USERDATA:
fprintf(source, "%s%s & data_%d = *check_%s(L, %d);\n", indentation, t.data.userdata_name, arg_number, t.data.userdata_name, arg_number);
break;
case TYPE_LITERAL:
// literals are expected to be done directly later
break;
case TYPE_NONE:
// nothing to do, we've either already emitted a reasonable value, or returned
break;
}
}
}
void emit_userdata_field(const struct userdata *data, const struct userdata_field *field) {
fprintf(source, "static int %s_%s(lua_State *L) {\n", data->name, field->name);
fprintf(source, " %s *ud = check_%s(L, 1);\n", data->name, data->name);
fprintf(source, " switch(lua_gettop(L)) {\n");
if (field->access_flags & ACCESS_FLAG_READ) {
fprintf(source, " case 1:\n");
switch (field->type.type) {
case TYPE_BOOLEAN:
fprintf(source, " lua_pushinteger(L, ud->%s);\n", field->name);
break;
case TYPE_FLOAT:
fprintf(source, " lua_pushnumber(L, ud->%s);\n", field->name);
break;
case TYPE_INT8_T:
case TYPE_INT16_T:
case TYPE_INT32_T:
case TYPE_UINT8_T:
case TYPE_UINT16_T:
case TYPE_ENUM:
fprintf(source, " lua_pushinteger(L, ud->%s);\n", field->name);
break;
case TYPE_UINT32_T:
fprintf(source, " new_uint32_t(L);\n");
fprintf(source, " *static_cast<uint32_t *>(luaL_checkudata(L, -1, \"uint32_t\")) = ud->%s;\n", field->name);
break;
case TYPE_NONE:
error(ERROR_INTERNAL, "Can't access a NONE field");
break;
case TYPE_LITERAL:
error(ERROR_INTERNAL, "Can't access a literal field");
break;
case TYPE_STRING:
fprintf(source, " lua_pushstring(L, ud->%s);\n", field->name);
break;
case TYPE_USERDATA:
error(ERROR_USERDATA, "Userdata does not currently support accss to userdata field's");
break;
}
fprintf(source, " return 1;\n");
}
if (field->access_flags & ACCESS_FLAG_WRITE) {
fprintf(source, " case 2: {\n");
emit_checker(field->type, 2, 0, " ", field->name);
fprintf(source, " ud->%s = data_2;\n", field->name);
fprintf(source, " return 0;\n");
fprintf(source, " }\n");
}
fprintf(source, " default:\n");
fprintf(source, " return luaL_argerror(L, lua_gettop(L), \"too many arguments\");\n");
fprintf(source, " }\n");
fprintf(source, "}\n\n");
}
void emit_userdata_fields() {
struct userdata * node = parsed_userdata;
while(node) {
struct userdata_field *field = node->fields;
while(field) {
emit_userdata_field(node, field);
field = field->next;
}
node = node->next;
}
}
void emit_userdata_method(const struct userdata *data, const struct method *method) {
int arg_count = 1;
const char *access_name = data->alias ? data->alias : data->name;
// bind ud early if it's a singleton, so that we can use it in the range checks
fprintf(source, "static int %s_%s(lua_State *L) {\n", data->name, method->name);
// emit comments on expected arg/type
struct argument *arg = method->arguments;
if (data->ud_type == UD_SINGLETON) {
// fetch and check the singleton pointer
fprintf(source, " %s * ud = %s::get_singleton();\n", data->name, data->name);
fprintf(source, " if (ud == nullptr) {\n");
fprintf(source, " return luaL_argerror(L, %d, \"%s not supported on this firmware\");\n", arg_count, access_name);
fprintf(source, " }\n\n");
}
// sanity check number of args called with
arg_count = 1;
while (arg != NULL) {
if (!(arg->type.flags & TYPE_FLAGS_NULLABLE) && !(arg->type.type == TYPE_LITERAL)) {
arg_count++;
}
arg = arg->next;
}
fprintf(source, " binding_argcheck(L, %d);\n", arg_count);
switch (data->ud_type) {
case UD_USERDATA:
// extract the userdata
fprintf(source, " %s * ud = check_%s(L, 1);\n", data->name, data->name);
break;
case UD_SINGLETON:
// this was bound early
break;
}
// extract the arguments
arg = method->arguments;
arg_count = 2;
int skipped = 0;
while (arg != NULL) {
if (arg->type.type != TYPE_LITERAL) {
// emit_checker will emit a nullable argument for us
emit_checker(arg->type, arg_count, skipped, " ", "argument");
arg_count++;
}
if (arg->type.type != TYPE_LITERAL || arg->type.flags & TYPE_FLAGS_NULLABLE) {
skipped++;
}
arg = arg->next;
}
if (data->flags & UD_FLAG_SEMAPHORE) {
fprintf(source, " ud->get_semaphore().take_blocking();\n");
}
switch (method->return_type.type) {
case TYPE_BOOLEAN:
fprintf(source, " const bool data = ud->%s(", method->name);
break;
case TYPE_FLOAT:
fprintf(source, " const float data = ud->%s(", method->name);
break;
case TYPE_INT8_T:
fprintf(source, " const int8_t data = ud->%s(", method->name);
break;
case TYPE_INT16_T:
fprintf(source, " const int6_t data = ud->%s(", method->name);
break;
case TYPE_INT32_T:
fprintf(source, " const int32_t data = ud->%s(", method->name);
break;
case TYPE_STRING:
fprintf(source, " const char * data = ud->%s(", method->name);
break;
case TYPE_UINT8_T:
fprintf(source, " const uint8_t data = ud->%s(", method->name);
break;
case TYPE_UINT16_T:
fprintf(source, " const uint16_t data = ud->%s(", method->name);
break;
case TYPE_UINT32_T:
fprintf(source, " const uint32_t data = ud->%s(", method->name);
break;
case TYPE_ENUM:
fprintf(source, " const %s &data = ud->%s(", method->return_type.data.enum_name, method->name);
break;
case TYPE_USERDATA:
fprintf(source, " const %s &data = ud->%s(", method->return_type.data.userdata_name, method->name);
break;
case TYPE_NONE:
fprintf(source, " ud->%s(", method->name);
break;
case TYPE_LITERAL:
error(ERROR_USERDATA, "Can't return a literal from a method");
break;
}
if (arg_count != 2) {
fprintf(source, "\n");
}
arg = method->arguments;
arg_count = 2;
while (arg != NULL) {
switch (arg->type.type) {
case TYPE_BOOLEAN:
case TYPE_FLOAT:
case TYPE_INT8_T:
case TYPE_INT16_T:
case TYPE_INT32_T:
case TYPE_STRING:
case TYPE_UINT8_T:
case TYPE_UINT16_T:
case TYPE_UINT32_T:
case TYPE_ENUM:
case TYPE_USERDATA:
fprintf(source, " data_%d", arg_count + ((arg->type.flags & TYPE_FLAGS_NULLABLE) ? NULLABLE_ARG_COUNT_BASE : 0));
break;
case TYPE_LITERAL:
fprintf(source, " %s", arg->type.data.literal);
break;
case TYPE_NONE:
error(ERROR_INTERNAL, "Can't pass nil as an argument");
break;
}
if (arg->type.type != TYPE_LITERAL) {
arg_count++;
}
arg = arg->next;
if (arg != NULL) {
fprintf(source, ",\n");
}
}
fprintf(source, "%s);\n\n", "");
if (data->flags & UD_FLAG_SEMAPHORE) {
fprintf(source, " ud->get_semaphore().give();\n");
}
int return_count = 1; // number of arguments to return
switch (method->return_type.type) {
case TYPE_BOOLEAN:
if (method->flags & TYPE_FLAGS_NULLABLE) {
fprintf(source, " if (data) {\n");
// we need to emit out nullable arguments, iterate the args again, creating and copying objects, while keeping a new count
return_count = 0;
arg = method->arguments;
int arg_index = NULLABLE_ARG_COUNT_BASE + 2;
while (arg != NULL) {
if (arg->type.flags & TYPE_FLAGS_NULLABLE) {
return_count++;
switch (arg->type.type) {
case TYPE_BOOLEAN:
fprintf(source, " lua_pushboolean(L, data_%d);\n", arg_index);
break;
case TYPE_FLOAT:
fprintf(source, " lua_pushnumber(L, data_%d);\n", arg_index);
break;
case TYPE_INT8_T:
case TYPE_INT16_T:
case TYPE_INT32_T:
case TYPE_UINT8_T:
case TYPE_UINT16_T:
case TYPE_ENUM:
fprintf(source, " lua_pushinteger(L, data_%d);\n", arg_index);
break;
case TYPE_UINT32_T:
fprintf(source, " new_uint32_t(L);\n");
fprintf(source, " *static_cast<uint32_t *>(luaL_checkudata(L, -1, \"uint32_t\")) = data_%d;\n", arg_index);
break;
case TYPE_STRING:
fprintf(source, " lua_pushstring(L, data_%d);\n", arg_index);
break;
case TYPE_USERDATA:
// userdatas must allocate a new container to return
fprintf(source, " new_%s(L);\n", arg->type.data.userdata_name);
fprintf(source, " *check_%s(L, -1) = data_%d;\n", arg->type.data.userdata_name, arg_index);
break;
case TYPE_NONE:
error(ERROR_INTERNAL, "Attempted to emit a nullable argument of type none");
break;
case TYPE_LITERAL:
error(ERROR_INTERNAL, "Attempted to make a nullable literal");
break;
}
}
arg_index++;
arg = arg->next;
}
fprintf(source, " } else {\n");
fprintf(source, " lua_pushnil(L);\n");
fprintf(source, " }\n");
} else {
fprintf(source, " lua_pushboolean(L, data);\n");
}
break;
case TYPE_FLOAT:
fprintf(source, " lua_pushnumber(L, data);\n");
break;
case TYPE_INT8_T:
case TYPE_INT16_T:
case TYPE_INT32_T:
case TYPE_UINT8_T:
case TYPE_UINT16_T:
case TYPE_ENUM:
fprintf(source, " lua_pushinteger(L, data);\n");
break;
case TYPE_UINT32_T:
fprintf(source, " new_uint32_t(L);\n");
fprintf(source, " *static_cast<uint32_t *>(luaL_checkudata(L, -1, \"uint32_t\")) = data;\n");
break;
case TYPE_STRING:
fprintf(source, " lua_pushstring(L, data);\n");
break;
case TYPE_USERDATA:
// userdatas must allocate a new container to return
fprintf(source, " new_%s(L);\n", method->return_type.data.userdata_name);
fprintf(source, " *check_%s(L, -1) = data;\n", method->return_type.data.userdata_name);
break;
case TYPE_NONE:
case TYPE_LITERAL:
// no return value, so don't worry about pushing a value
return_count = 0;
break;
}
fprintf(source, " return %d;\n", return_count);
fprintf(source, "}\n\n");
}
const char * get_name_for_operation(enum operator_type op) {
switch (op) {
case OP_ADD:
return "__add";
case OP_SUB:
return "__sub";
case OP_MUL:
return "__mul";
break;
case OP_DIV:
return "__div";
break;
case OP_LAST:
return NULL;
}
return NULL;
}
void emit_operators(struct userdata *data) {
trace(TRACE_USERDATA, "Emitting operators for %s", data->name);
assert(data->ud_type == UD_USERDATA);
for (uint32_t i = 1; i < OP_LAST; i = (i << 1)) {
const char * op_name = get_name_for_operation((data->operations) & i);
if (op_name == NULL) {
continue;
}
char op_sym;
switch ((data->operations) & i) {
case OP_ADD:
op_sym = '+';
break;
case OP_SUB:
op_sym = '-';
break;
case OP_MUL:
op_sym = '*';
break;
case OP_DIV:
op_sym = '/';
break;
case OP_LAST:
return;
}
fprintf(source, "static int %s_%s(lua_State *L) {\n", data->name, op_name);
// check number of arguments
fprintf(source, " binding_argcheck(L, 2);\n");
// check the pointers
fprintf(source, " %s *ud = check_%s(L, 1);\n", data->name, data->name);
fprintf(source, " %s *ud2 = check_%s(L, 2);\n", data->name, data->name);
// create a container for the result
fprintf(source, " new_%s(L);\n", data->name);
fprintf(source, " *check_%s(L, -1) = *ud %c *ud2;;\n", data->name, op_sym);
// return the first pointer
fprintf(source, " return 1;\n");
fprintf(source, "}\n\n");
}
}
void emit_userdata_methods(struct userdata *node) {
while(node) {
// methods
struct method *method = node->methods;
while(method) {
emit_userdata_method(node, method);
method = method->next;
}
// operators
if (node->operations) {
emit_operators(node);
}
node = node->next;
}
}
void emit_userdata_metatables(void) {
struct userdata * node = parsed_userdata;
while(node) {
fprintf(source, "const luaL_Reg %s_meta[] = {\n", node->name);
struct userdata_field *field = node->fields;
while(field) {
fprintf(source, " {\"%s\", %s_%s},\n", field->name, node->name, field->name);
field = field->next;
}
struct method *method = node->methods;
while(method) {
fprintf(source, " {\"%s\", %s_%s},\n", method->name, node->name, method->name);
method = method->next;
}
for (uint32_t i = 1; i < OP_LAST; i = i << 1) {
const char * op_name = get_name_for_operation((node->operations) & i);
if (op_name == NULL) {
continue;
}
fprintf(source, " {\"%s\", %s_%s},\n", op_name, node->name, op_name);
}
fprintf(source, " {NULL, NULL}\n");
fprintf(source, "};\n\n");
node = node->next;
}
}
void emit_singleton_metatables(void) {
struct userdata * node = parsed_singletons;
while(node) {
fprintf(source, "const luaL_Reg %s_meta[] = {\n", node->name);
struct method *method = node->methods;
while (method) {
fprintf(source, " {\"%s\", %s_%s},\n", method->name, node->name, method->name);
method = method->next;
}
fprintf(source, " {NULL, NULL}\n");
fprintf(source, "};\n\n");
node = node->next;
}
}
void emit_enums(struct userdata * data) {
while (data) {
if (data->enums != NULL) {
fprintf(source, "struct userdata_enum %s_enums[] = {\n", data->name);
struct userdata_enum *ud_enum = data->enums;
while (ud_enum != NULL) {
fprintf(source, " {\"%s\", %s::%s},\n", ud_enum->name, data->name, ud_enum->name);
ud_enum = ud_enum->next;
}
fprintf(source, " {NULL, 0}};\n\n");
}
data = data->next;
}
}
void emit_metas(struct userdata * data, char * meta_name) {
fprintf(source, "const struct userdata_meta %s_fun[] = {\n", meta_name);
while (data) {
if (data->enums) {
fprintf(source, " {\"%s\", %s_meta, %s_enums},\n", data->alias ? data->alias : data->name, data->name, data->name);
} else {
fprintf(source, " {\"%s\", %s_meta, NULL},\n", data->alias ? data->alias : data->name, data->name);
}
data = data->next;
}
fprintf(source, "};\n\n");
}
void emit_loaders(void) {
// emit the enum header
fprintf(source, "struct userdata_enum {\n");
fprintf(source, " const char *name;\n");
fprintf(source, " int value;\n");
fprintf(source, "};\n\n");
emit_enums(parsed_userdata);
emit_enums(parsed_singletons);
// emit the meta table header
fprintf(source, "struct userdata_meta {\n");
fprintf(source, " const char *name;\n");
fprintf(source, " const luaL_Reg *reg;\n");
fprintf(source, " const struct userdata_enum *enums;\n");
fprintf(source, "};\n\n");
emit_metas(parsed_userdata, "userdata");
emit_metas(parsed_singletons, "singleton");
fprintf(source, "void load_generated_bindings(lua_State *L) {\n");
fprintf(source, " luaL_checkstack(L, 5, \"Out of stack\");\n"); // this is more stack space then we need, but should never fail
fprintf(source, " // userdata metatables\n");
fprintf(source, " for (uint32_t i = 0; i < ARRAY_SIZE(userdata_fun); i++) {\n");
fprintf(source, " luaL_newmetatable(L, userdata_fun[i].name);\n");
fprintf(source, " luaL_setfuncs(L, userdata_fun[i].reg, 0);\n");
fprintf(source, " lua_pushstring(L, \"__index\");\n");
fprintf(source, " lua_pushvalue(L, -2);\n");
fprintf(source, " lua_settable(L, -3);\n");
fprintf(source, " lua_pop(L, 1);\n");
fprintf(source, " }\n");
fprintf(source, "\n");
fprintf(source, " // singleton metatables\n");
fprintf(source, " for (uint32_t i = 0; i < ARRAY_SIZE(singleton_fun); i++) {\n");
fprintf(source, " luaL_newmetatable(L, singleton_fun[i].name);\n");
fprintf(source, " luaL_setfuncs(L, singleton_fun[i].reg, 0);\n");
fprintf(source, " lua_pushstring(L, \"__index\");\n");
fprintf(source, " lua_pushvalue(L, -2);\n");
fprintf(source, " lua_settable(L, -3);\n");
fprintf(source, " if (singleton_fun[i].enums != nullptr) {\n");
fprintf(source, " int j = 0;\n");
fprintf(source, " while (singleton_fun[i].enums[j].name != NULL) {\n");
fprintf(source, " lua_pushstring(L, singleton_fun[i].enums[j].name);\n");
fprintf(source, " lua_pushinteger(L, singleton_fun[i].enums[j].value);\n");
fprintf(source, " lua_settable(L, -3);\n");
fprintf(source, " j++;\n");
fprintf(source, " }\n");
fprintf(source, " }\n");
fprintf(source, " lua_pop(L, 1);\n");
fprintf(source, " lua_newuserdata(L, 0);\n");
fprintf(source, " luaL_getmetatable(L, singleton_fun[i].name);\n");
fprintf(source, " lua_setmetatable(L, -2);\n");
fprintf(source, " lua_setglobal(L, singleton_fun[i].name);\n");
fprintf(source, " }\n");
fprintf(source, "\n");
fprintf(source, " load_boxed_numerics(L);\n");
fprintf(source, "}\n\n");
}
void emit_sandbox(void) {
struct userdata *single = parsed_singletons;
fprintf(source, "const char *singletons[] = {\n");
while (single) {
fprintf(source, " \"%s\",\n", single->alias ? single->alias : single->name);
single = single->next;
}
fprintf(source, "};\n\n");
struct userdata *data = parsed_userdata;
fprintf(source, "const struct userdata {\n");
fprintf(source, " const char *name;\n");
fprintf(source, " const lua_CFunction fun;\n");
fprintf(source, "} new_userdata[] = {\n");
while (data) {
fprintf(source, " {\"%s\", new_%s},\n", data->name, data->name);
data = data->next;
}
fprintf(source, "};\n\n");
fprintf(source, "void load_generated_sandbox(lua_State *L) {\n");
// load the singletons
fprintf(source, " for (uint32_t i = 0; i < ARRAY_SIZE(singletons); i++) {\n");
fprintf(source, " lua_pushstring(L, singletons[i]);\n");
fprintf(source, " lua_getglobal(L, singletons[i]);\n");
fprintf(source, " lua_settable(L, -3);\n");
fprintf(source, " }\n");
// load the userdata allactors
fprintf(source, " for (uint32_t i = 0; i < ARRAY_SIZE(new_userdata); i++) {\n");
fprintf(source, " lua_pushstring(L, new_userdata[i].name);\n");
fprintf(source, " lua_pushcfunction(L, new_userdata[i].fun);\n");
fprintf(source, " lua_settable(L, -3);\n");
fprintf(source, " }\n");
fprintf(source, "\n");
fprintf(source, " load_boxed_numerics_sandbox(L);\n");
// load the userdata complex functions
fprintf(source, "}\n");
}
void emit_argcheck_helper(void) {
// tagging this with NOINLINE can save a large amount of flash
// but until we need it we will allow the compilier to choose to inline this for us
fprintf(source, "static int binding_argcheck(lua_State *L, int expected_arg_count) {\n");
fprintf(source, " const int args = lua_gettop(L);\n");
fprintf(source, " if (args > expected_arg_count) {\n");
fprintf(source, " return luaL_argerror(L, args, \"too many arguments\");\n");
fprintf(source, " } else if (args < expected_arg_count) {\n");
fprintf(source, " return luaL_argerror(L, args, \"too few arguments\");\n");
fprintf(source, " }\n");
fprintf(source, " return 0;\n");
fprintf(source, "}\n\n");
}
char * output_path = NULL;
int main(int argc, char **argv) {
state.line_num = -1;
int c;
while ((c = getopt(argc, argv, "i:o:")) != -1) {
switch (c) {
case 'i':
if (description != NULL) {
error(ERROR_GENERAL, "Already loaded a description file");
}
trace(TRACE_GENERAL, "Loading a description file: %s", optarg);
description = fopen(optarg, "r");
if (description == NULL) {
error(ERROR_GENERAL, "Unable to load the description file: %s", optarg);
}
break;
case 'o':
if (output_path != NULL) {
error(ERROR_GENERAL, "An output path was already selected.");
}
output_path = optarg;
trace(TRACE_GENERAL, "Loading an output path of %s", output_path);
break;
}
}
if (output_path == NULL) {
error(ERROR_GENERAL, "An output path must be provided for the generated bindings");
}
state.line_num = 0;
while (start_line()) {
// identify line type
if (state.token != NULL) {
// we have input
if (strcmp(state.token, keyword_comment) == 0) {
while (next_token()) {}
// nothing to do here, jump to the next line
} else if (strcmp(state.token, keyword_include) == 0) {
handle_header();
} else if (strcmp (state.token, keyword_userdata) == 0){
handle_userdata();
} else if (strcmp (state.token, keyword_singleton) == 0){
handle_singleton();
} else if (strcmp (state.token, keyword_depends) == 0){
handle_depends();
} else {
error(ERROR_UNKNOWN_KEYWORD, "Expected a keyword, got: %s", state.token);
}
if (next_token()) {
error(ERROR_UNKNOWN_KEYWORD, "Extra token provided: %s", state.token);
}
}
}
state.line_num = -1;
char *file_name = (char *)allocate(strlen(output_path) + 5);
sprintf(file_name, "%s.cpp", output_path);
source = fopen(file_name, "w");
if (source == NULL) {
error(ERROR_GENERAL, "Unable to open the output source file: %s", file_name);
}
fprintf(source, "// auto generated bindings, don't manually edit\n");
trace(TRACE_GENERAL, "Sanity checking parsed input");
sanity_check_userdata();
fprintf(source, "#include \"lua_generated_bindings.h\"\n");
fprintf(source, "#include \"lua_boxed_numerics.h\"\n");
trace(TRACE_GENERAL, "Starting emission");
emit_headers(source);
fprintf(source, "\n\n");
emit_dependencies(source);
fprintf(source, "\n\n");
emit_argcheck_helper();
emit_userdata_allocators();
emit_userdata_checkers();
emit_userdata_fields();
emit_userdata_methods(parsed_userdata);
emit_userdata_metatables();
emit_userdata_methods(parsed_singletons);
emit_singleton_metatables();
emit_loaders();
emit_sandbox();
fclose(source);
source = NULL;
sprintf(file_name, "%s.h", output_path);
header = fopen(file_name, "w");
if (header == NULL) {
error(ERROR_GENERAL, "Unable to open the output header file: %s", file_name);
}
free(file_name);
fprintf(header, "#pragma once\n");
fprintf(header, "// auto generated bindings, don't manually edit\n");
emit_headers(header);
fprintf(header, "#include \"lua/src/lua.hpp\"\n");
fprintf(header, "#include <new>\n\n");
emit_dependencies(header);
fprintf(header, "\n\n");
emit_userdata_declarations();
fprintf(header, "void load_generated_bindings(lua_State *L);\n");
fprintf(header, "void load_generated_sandbox(lua_State *L);\n");
fclose(header);
header = NULL;
return 0;
}