int yr_parser_reduce_import(
yyscan_t yyscanner,
SIZED_STRING* module_name)
{
YR_COMPILER* compiler = yyget_extra(yyscanner);
YR_OBJECT* module_structure;
char* name;
module_structure = (YR_OBJECT*) yr_hash_table_lookup(
compiler->objects_table,
module_name->c_string,
compiler->current_namespace->name);
// if module already imported, do nothing
if (module_structure != NULL)
return ERROR_SUCCESS;
compiler->last_result = yr_object_create(
OBJECT_TYPE_STRUCTURE,
module_name->c_string,
NULL,
&module_structure);
if (compiler->last_result == ERROR_SUCCESS)
compiler->last_result = yr_hash_table_add(
compiler->objects_table,
module_name->c_string,
compiler->current_namespace->name,
module_structure);
if (compiler->last_result == ERROR_SUCCESS)
{
compiler->last_result = yr_modules_do_declarations(
module_name->c_string,
module_structure);
if (compiler->last_result == ERROR_UNKNOWN_MODULE)
yr_compiler_set_error_extra_info(compiler, module_name->c_string);
}
if (compiler->last_result == ERROR_SUCCESS)
compiler->last_result = yr_arena_write_string(
compiler->sz_arena,
module_name->c_string,
&name);
if (compiler->last_result == ERROR_SUCCESS)
compiler->last_result = yr_parser_emit_with_arg_reloc(
yyscanner,
OP_IMPORT,
PTR_TO_INT64(name),
NULL,
NULL);
return compiler->last_result;
}
int yr_object_function_create(
const char* identifier,
const char* arguments_fmt,
const char* return_fmt,
YR_MODULE_FUNC code,
YR_OBJECT* parent,
YR_OBJECT** function)
{
YR_OBJECT* return_obj;
YR_OBJECT* f;
int8_t return_type;
switch (*return_fmt)
{
case 'i':
return_type = OBJECT_TYPE_INTEGER;
break;
case 's':
return_type = OBJECT_TYPE_STRING;
break;
default:
return ERROR_INVALID_FORMAT;
}
FAIL_ON_ERROR(yr_object_create(
OBJECT_TYPE_FUNCTION,
identifier,
parent,
&f));
FAIL_ON_ERROR_WITH_CLEANUP(
yr_object_create(return_type, "result", f, &return_obj),
yr_object_destroy(f));
((YR_OBJECT_FUNCTION* )f)->arguments_fmt = arguments_fmt;
((YR_OBJECT_FUNCTION* )f)->return_obj = return_obj;
((YR_OBJECT_FUNCTION* )f)->code = code;
if (function != NULL)
*function = f;
return ERROR_SUCCESS;
}
int yr_object_from_external_variable(
YR_EXTERNAL_VARIABLE* external,
YR_OBJECT** object)
{
YR_OBJECT* obj;
int result;
uint8_t obj_type = 0;
switch(external->type)
{
case EXTERNAL_VARIABLE_TYPE_INTEGER:
case EXTERNAL_VARIABLE_TYPE_BOOLEAN:
obj_type = OBJECT_TYPE_INTEGER;
break;
case EXTERNAL_VARIABLE_TYPE_STRING:
case EXTERNAL_VARIABLE_TYPE_MALLOC_STRING:
obj_type = OBJECT_TYPE_STRING;
break;
default:
assert(FALSE);
}
result = yr_object_create(
obj_type,
external->identifier,
NULL,
&obj);
if (result == ERROR_SUCCESS)
{
switch(external->type)
{
case EXTERNAL_VARIABLE_TYPE_INTEGER:
case EXTERNAL_VARIABLE_TYPE_BOOLEAN:
yr_object_set_integer(external->integer, obj, NULL);
break;
case EXTERNAL_VARIABLE_TYPE_STRING:
case EXTERNAL_VARIABLE_TYPE_MALLOC_STRING:
yr_object_set_string(external->string, obj, NULL);
break;
}
*object = obj;
}
return result;
}
int yr_modules_load(
const char* module_name,
YR_SCAN_CONTEXT* context)
{
int i, result;
YR_MODULE_IMPORT mi;
YR_OBJECT* module_structure = (YR_OBJECT*) yr_hash_table_lookup(
context->objects_table,
module_name,
NULL);
// if module_structure != NULL, the module was already
// loaded, return successfully without doing nothing.
if (module_structure != NULL)
return ERROR_SUCCESS;
// not loaded yet
FAIL_ON_ERROR(yr_object_create(
OBJECT_TYPE_STRUCTURE,
module_name,
NULL,
&module_structure));
mi.module_name = module_name;
mi.module_data = NULL;
mi.module_data_size = 0;
result = context->callback(
CALLBACK_MSG_IMPORT_MODULE,
&mi,
context->user_data);
if (result == CALLBACK_ERROR)
{
yr_object_destroy(module_structure);
return ERROR_CALLBACK_ERROR;
}
FAIL_ON_ERROR_WITH_CLEANUP(
yr_modules_do_declarations(module_name, module_structure),
yr_object_destroy(module_structure));
FAIL_ON_ERROR_WITH_CLEANUP(
yr_hash_table_add(
context->objects_table,
module_name,
NULL,
module_structure),
yr_object_destroy(module_structure));
for (i = 0; i < sizeof(yr_modules_table) / sizeof(YR_MODULE); i++)
{
if (strcmp(yr_modules_table[i].name, module_name) == 0)
{
result = yr_modules_table[i].load(
context,
module_structure,
mi.module_data,
mi.module_data_size);
if (result != ERROR_SUCCESS)
return result;
}
}
result = context->callback(
CALLBACK_MSG_MODULE_IMPORTED,
module_structure,
context->user_data);
return ERROR_SUCCESS;
}
int yr_parser_reduce_import(
yyscan_t yyscanner,
SIZED_STRING* module_name)
{
int result;
YR_COMPILER* compiler = yyget_extra(yyscanner);
YR_OBJECT* module_structure;
char* name;
if (!_yr_parser_valid_module_name(module_name))
{
yr_compiler_set_error_extra_info(compiler, module_name->c_string);
return ERROR_INVALID_MODULE_NAME;
}
module_structure = (YR_OBJECT*) yr_hash_table_lookup(
compiler->objects_table,
module_name->c_string,
compiler->current_namespace->name);
// if module already imported, do nothing
if (module_structure != NULL)
return ERROR_SUCCESS;
FAIL_ON_ERROR(yr_object_create(
OBJECT_TYPE_STRUCTURE,
module_name->c_string,
NULL,
&module_structure));
FAIL_ON_ERROR(yr_hash_table_add(
compiler->objects_table,
module_name->c_string,
compiler->current_namespace->name,
module_structure));
result = yr_modules_do_declarations(
module_name->c_string,
module_structure);
if (result == ERROR_UNKNOWN_MODULE)
yr_compiler_set_error_extra_info(compiler, module_name->c_string);
if (result != ERROR_SUCCESS)
return result;
FAIL_ON_ERROR(yr_arena_write_string(
compiler->sz_arena,
module_name->c_string,
&name));
FAIL_ON_ERROR(yr_parser_emit_with_arg_reloc(
yyscanner,
OP_IMPORT,
name,
NULL,
NULL));
return ERROR_SUCCESS;
}
