#include #include #include #include #include #include #include #include char keyword_alias[] = "alias"; char keyword_ap_object[] = "ap_object"; 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_scheduler_semaphore[] = "scheduler-semaphore"; 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; 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, TYPE_AP_OBJECT }; const char * type_labels[TYPE_USERDATA + 1] = { "bool", "float", "int8_t", "int16_t", "int32_t", "uint8_t", "uint16_t", "void", "string", "enum", "userdata", "ap_object", }; 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 { struct ud { char *name; char *sanatized_name; } ud; 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); free(fmt); fmt = NULL; } } void error(const int code, const char *message, ...) __attribute__ ((noreturn)); 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); free(fmt); fmt = NULL; 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); state.token = NULL; // burn the line } 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++; const size_t length = strlen(state.line); if (length > 1 && state.line[length - 2] == '\r') { trace(TRACE_TOKENS, "Discarding carriage return"); if (length == 2) { // empty line of just carriage return, loop again continue; } state.line[length - 2] = '\0'; } else if (length > 0 && state.line[length - 1] == '\n') { trace(TRACE_TOKENS, "Discarding newline"); if (length == 1) { // empty line of just carriage return, loop again continue; } state.line[length - 1] = '\0'; } state.token = strtok(state.line, token_delimiters); state.token_num = 1; trace(TRACE_TOKENS, "Start of line 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, UD_AP_OBJECT, }; 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), UD_FLAG_SCHEDULER_SEMAPHORE = (1U << 1), }; struct userdata_enum { struct userdata_enum * next; char * name; // enum name }; struct userdata { struct userdata * next; char *name; // name of the C++ singleton char *sanatized_name; // sanatized 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; static struct userdata *parsed_ap_objects; 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; // 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); } void sanatize_name(char **dest, char *src) { *dest = (char *)allocate(strlen(src) + 1); strcpy(*dest, src); char *position = strchr(*dest, ':'); while (position) { *position = '_'; position = strchr(position, ':'); } }; 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_AP_OBJECT: 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 { // this must be a user data or an ap_object, check if it's already been declared as an object struct userdata *node = parsed_ap_objects; while (node != NULL && strcmp(node->name, data_type)) { node = node->next; } if (node != NULL) { type->type = TYPE_AP_OBJECT; } else { // assume that this is a user data, we can't validate this until later though type->type = TYPE_USERDATA; } string_copy(&(type->data.ud.name), data_type); sanatize_name(&(type->data.ud.sanatized_name), type->data.ud.name); } // 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_AP_OBJECT: 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_AP_OBJECT: 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 = OP_ADD; 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); sanatize_name(&(node->sanatized_name), node->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); sanatize_name(&(node->sanatized_name), node->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_scheduler_semaphore) == 0) { node->flags |= UD_FLAG_SCHEDULER_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_ap_object(void) { trace(TRACE_SINGLETON, "Adding a ap_object"); char *name = next_token(); if (name == NULL) { error(ERROR_USERDATA, "Expected a name for the ap_object"); } struct userdata *node = parsed_ap_objects; while (node != NULL && strcmp(node->name, name)) { node = node->next; } if (node == NULL) { trace(TRACE_USERDATA, "Allocating new ap_object for %s", name); node = (struct userdata *)allocate(sizeof(struct userdata)); node->ud_type = UD_AP_OBJECT; node->name = (char *)allocate(strlen(name) + 1); strcpy(node->name, name); sanatize_name(&(node->sanatized_name), node->name); node->next = parsed_ap_objects; parsed_ap_objects = node; } // read type char *type = next_token(); if (type == NULL) { error(ERROR_SINGLETON, "Expected a access type for ap_object %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_scheduler_semaphore) == 0) { node->flags |= UD_FLAG_SCHEDULER_SEMAPHORE; } else if (strcmp(type, keyword_method) == 0) { handle_method(node->name, &(node->methods)); } else { error(ERROR_SINGLETON, "AP_Objects only support aliases, methods or semaphore keyowrds (got %s)", type); } // check that we didn't just add 2 singleton flags if ((node->flags & UD_FLAG_SEMAPHORE) && (node->flags & UD_FLAG_SCHEDULER_SEMAPHORE)) { error(ERROR_SINGLETON, "Taking both a library semaphore and scheduler semaphore is prohibited"); } // 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->sanatized_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_ap_object_allocators(void) { struct userdata * node = parsed_ap_objects; while (node) { fprintf(source, "int new_%s(lua_State *L) {\n", node->sanatized_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); // FIXME: memset is a ridiculously large hammer here 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->sanatized_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_ap_object_checkers(void) { struct userdata * node = parsed_ap_objects; while (node) { fprintf(source, "%s ** check_%s(lua_State *L, int arg) {\n", node->name, node->sanatized_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->sanatized_name); fprintf(header, "%s * check_%s(lua_State *L, int arg);\n", node->name, node->sanatized_name); node = node->next; } } void emit_ap_object_declarations(void) { struct userdata * node = parsed_ap_objects; while (node) { fprintf(header, "int new_%s(lua_State *L);\n", node->sanatized_name); fprintf(header, "%s ** check_%s(lua_State *L, int arg);\n", node->name, node->sanatized_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_AP_OBJECT: 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.ud.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_AP_OBJECT: 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 = coerce_to_uint32_t(L, %d);\n", indentation, arg_number, arg_number - skipped); break; case TYPE_AP_OBJECT: 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_AP_OBJECT: 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(raw_data_%d);\n", indentation, arg_number, arg_number); break; case TYPE_INT16_T: fprintf(source, "%sconst int16_t data_%d = static_cast(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(raw_data_%d);\n", indentation, arg_number, arg_number); break; case TYPE_UINT16_T: fprintf(source, "%sconst uint16_t data_%d = static_cast(raw_data_%d);\n", indentation, arg_number, arg_number); break; case TYPE_UINT32_T: fprintf(source, "%sconst uint32_t data_%d = static_cast(raw_data_%d);\n", indentation, arg_number, arg_number); break; case TYPE_BOOLEAN: fprintf(source, "%sconst bool data_%d = static_cast(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.ud.name, arg_number, t.data.ud.sanatized_name, arg_number); break; case TYPE_AP_OBJECT: fprintf(source, "%s%s * data_%d = *check_%s(L, %d);\n", indentation, t.data.ud.name, arg_number, t.data.ud.sanatized_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->sanatized_name, field->name); fprintf(source, " %s *ud = check_%s(L, 1);\n", data->name, data->sanatized_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(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; case TYPE_AP_OBJECT: // FIXME: collapse the identical cases here, and use the type string function error(ERROR_USERDATA, "AP_Object 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->sanatized_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->sanatized_name); break; case UD_SINGLETON: // this was bound early break; case UD_AP_OBJECT: // extract the userdata, it was a pointer, so we need to grab it fprintf(source, " %s * ud = *check_%s(L, 1);\n", data->name, data->sanatized_name); fprintf(source, " if (ud == NULL) {\n"); fprintf(source, " return luaL_error(L, \"Internal error, null pointer\");\n"); fprintf(source, " }\n"); 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"); } if (data->flags & UD_FLAG_SCHEDULER_SEMAPHORE) { fprintf(source, " AP::scheduler().get_semaphore().take_blocking();\n"); } int static_cast = TRUE; switch (method->return_type.type) { case TYPE_STRING: fprintf(source, " const char * data = ud->%s(", method->name); static_cast = FALSE; break; case TYPE_ENUM: fprintf(source, " const %s &data = ud->%s(", method->return_type.data.enum_name, method->name); static_cast = FALSE; break; case TYPE_USERDATA: fprintf(source, " const %s &data = ud->%s(", method->return_type.data.ud.name, method->name); static_cast = FALSE; break; case TYPE_AP_OBJECT: fprintf(source, " %s *data = ud->%s(", method->return_type.data.ud.name, method->name); static_cast = FALSE; break; case TYPE_NONE: fprintf(source, " ud->%s(", method->name); static_cast = FALSE; break; case TYPE_LITERAL: error(ERROR_USERDATA, "Can't return a literal from a method"); break; case TYPE_BOOLEAN: 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: break; } if (static_cast) { char *var_type_name; switch (method->return_type.type) { case TYPE_BOOLEAN: var_type_name = "bool"; break; case TYPE_FLOAT: var_type_name = "float"; break; case TYPE_INT8_T: var_type_name = "int8_t"; break; case TYPE_INT16_T: var_type_name = "int16_t"; break; case TYPE_INT32_T: var_type_name = "int32_t"; break; case TYPE_UINT8_T: var_type_name = "uint8_t"; break; case TYPE_UINT16_T: var_type_name = "uint16_t"; break; case TYPE_UINT32_T: var_type_name = "uint32_t"; break; case TYPE_STRING: case TYPE_ENUM: case TYPE_USERDATA: case TYPE_AP_OBJECT: case TYPE_NONE: case TYPE_LITERAL: error(ERROR_USERDATA, "Unexpected type"); break; } fprintf(source, " const %s data = static_cast<%s>(ud->%s(", var_type_name, var_type_name, method->name); } 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: case TYPE_AP_OBJECT: 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"); } } if (static_cast) { fprintf(source, "%s));\n\n", ""); } else { fprintf(source, "%s);\n\n", ""); } if (data->flags & UD_FLAG_SEMAPHORE) { fprintf(source, " ud->get_semaphore().give();\n"); } if (data->flags & UD_FLAG_SCHEDULER_SEMAPHORE) { fprintf(source, " AP::scheduler().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(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.ud.sanatized_name); fprintf(source, " *check_%s(L, -1) = data_%d;\n", arg->type.data.ud.sanatized_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; case TYPE_AP_OBJECT: // FIXME: collapse these to a single failure case error(ERROR_INTERNAL, "Attempted to make a nullable ap_object"); break; } } arg_index++; arg = arg->next; } fprintf(source, " return %d;\n", return_count); fprintf(source, " } else {\n"); fprintf(source, " return 0;\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(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.ud.sanatized_name); fprintf(source, " *check_%s(L, -1) = data;\n", method->return_type.data.ud.sanatized_name); break; case TYPE_AP_OBJECT: fprintf(source, " if (data == NULL) {\n"); fprintf(source, " lua_pushnil(L);\n"); fprintf(source, " } else {\n"); fprintf(source, " new_%s(L);\n", method->return_type.data.ud.sanatized_name); fprintf(source, " *check_%s(L, -1) = data;\n", method->return_type.data.ud.sanatized_name); fprintf(source, " }\n"); break; case TYPE_NONE: case TYPE_LITERAL: // no return value, so don't worry about pushing a value return_count = 0; break; } if ((method->return_type.type != TYPE_BOOLEAN) || ((method->flags & TYPE_FLAGS_NULLABLE) == 0)) { 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->sanatized_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->sanatized_name); fprintf(source, " %s *ud2 = check_%s(L, 2);\n", data->name, data->sanatized_name); // create a container for the result fprintf(source, " new_%s(L);\n", data->sanatized_name); fprintf(source, " *check_%s(L, -1) = *ud %c *ud2;;\n", data->sanatized_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->sanatized_name); struct userdata_field *field = node->fields; while(field) { fprintf(source, " {\"%s\", %s_%s},\n", field->name, node->sanatized_name, field->name); field = field->next; } struct method *method = node->methods; while(method) { fprintf(source, " {\"%s\", %s_%s},\n", method->name, node->sanatized_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->sanatized_name, op_name); } fprintf(source, " {NULL, NULL}\n"); fprintf(source, "};\n\n"); node = node->next; } } void emit_singleton_metatables(struct userdata *head) { struct userdata * node = head; while(node) { fprintf(source, "const luaL_Reg %s_meta[] = {\n", node->sanatized_name); struct method *method = node->methods; while (method) { fprintf(source, " {\"%s\", %s_%s},\n", method->name, node->sanatized_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->sanatized_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->sanatized_name); } else { fprintf(source, " {\"%s\", %s_meta, NULL},\n", data->alias ? data->alias : data->name, data->sanatized_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"); emit_metas(parsed_ap_objects, "ap_object"); 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, " // ap object metatables\n"); fprintf(source, " for (uint32_t i = 0; i < ARRAY_SIZE(ap_object_fun); i++) {\n"); fprintf(source, " luaL_newmetatable(L, ap_object_fun[i].name);\n"); fprintf(source, " luaL_setfuncs(L, ap_object_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->sanatized_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->sanatized_name); data = data->next; } data = parsed_ap_objects; while (data) { fprintf(source, " {\"%s\", new_%s},\n", data->name, data->sanatized_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_ap_object) == 0){ handle_ap_object(); } 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. See README.md for details.\n"); trace(TRACE_GENERAL, "Sanity checking parsed input"); sanity_check_userdata(); fprintf(source, "#include \"lua_generated_bindings.h\"\n"); fprintf(source, "#include \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_ap_object_allocators(); emit_ap_object_checkers(); emit_userdata_fields(); emit_userdata_methods(parsed_userdata); emit_userdata_metatables(); emit_userdata_methods(parsed_singletons); emit_singleton_metatables(parsed_singletons); emit_userdata_methods(parsed_ap_objects); emit_singleton_metatables(parsed_ap_objects); 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. See README.md for details.\n"); emit_headers(header); fprintf(header, "#include \n"); fprintf(header, "#include \n\n"); emit_dependencies(header); fprintf(header, "\n\n"); emit_userdata_declarations(); emit_ap_object_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; }