struct sieve_binary *sieve_binary_open
(struct sieve_instance *svinst, const char *path, struct sieve_script *script,
enum sieve_error *error_r)
{
struct sieve_binary_extension_reg *const *regs;
unsigned int ext_count, i;
struct sieve_binary *sbin;
struct sieve_binary_file *file;
i_assert( script == NULL || sieve_script_svinst(script) == svinst );
//file = _file_memory_open(path);
if ( (file=_file_lazy_open(svinst, path, error_r)) == NULL )
return NULL;
/* Create binary object */
sbin = sieve_binary_create(svinst, script);
sbin->path = p_strdup(sbin->pool, path);
sbin->file = file;
if ( !_sieve_binary_open(sbin) ) {
sieve_binary_unref(&sbin);
if ( error_r != NULL )
*error_r = SIEVE_ERROR_NOT_VALID;
return NULL;
}
sieve_binary_activate(sbin);
/* Signal open event to extensions */
regs = array_get(&sbin->extensions, &ext_count);
for ( i = 0; i < ext_count; i++ ) {
const struct sieve_binary_extension *binext = regs[i]->binext;
if ( binext != NULL && binext->binary_open != NULL &&
!binext->binary_open(regs[i]->extension, sbin, regs[i]->context) ) {
/* Extension thinks its corrupt */
if ( error_r != NULL )
*error_r = SIEVE_ERROR_NOT_VALID;
sieve_binary_unref(&sbin);
return NULL;
}
}
return sbin;
}
struct sieve_binary *sieve_generator_run
(struct sieve_generator *gentr, struct sieve_binary_block **sblock_r)
{
bool topmost = ( sblock_r == NULL || *sblock_r == NULL );
struct sieve_binary *sbin;
struct sieve_binary_block *sblock, *debug_block;
const struct sieve_extension *const *extensions;
unsigned int i, ext_count;
bool result = TRUE;
/* Initialize */
if ( topmost ) {
sbin = sieve_binary_create_new(sieve_ast_script(gentr->genenv.ast));
sblock = sieve_binary_block_get(sbin, SBIN_SYSBLOCK_MAIN_PROGRAM);
} else {
sblock = *sblock_r;
sbin = sieve_binary_block_get_binary(sblock);
}
i_assert(sbin != NULL);
sieve_binary_ref(sbin);
gentr->genenv.sbin = sbin;
gentr->genenv.sblock = sblock;
/* Create debug block */
debug_block = sieve_binary_block_create(sbin);
gentr->dwriter = sieve_binary_debug_writer_init(debug_block);
(void)sieve_binary_emit_unsigned
(sblock, sieve_binary_block_get_id(debug_block));
/* Load extensions linked to the AST and emit a list in code */
extensions = sieve_ast_extensions_get(gentr->genenv.ast, &ext_count);
(void) sieve_binary_emit_unsigned(sblock, ext_count);
for ( i = 0; i < ext_count; i++ ) {
const struct sieve_extension *ext = extensions[i];
/* Link to binary */
(void)sieve_binary_extension_link(sbin, ext);
/* Emit */
sieve_binary_emit_extension(sblock, ext, 0);
/* Load */
if ( ext->def != NULL && ext->def->generator_load != NULL &&
!ext->def->generator_load(ext, &gentr->genenv) )
result = FALSE;
}
/* Generate code */
if ( result ) {
if ( !sieve_generate_block
(&gentr->genenv, sieve_ast_root(gentr->genenv.ast)))
result = FALSE;
else if ( topmost )
sieve_binary_activate(sbin);
}
/* Cleanup */
gentr->genenv.sbin = NULL;
gentr->genenv.sblock = NULL;
sieve_binary_unref(&sbin);
if ( !result ) {
if ( topmost ) {
sieve_binary_unref(&sbin);
if ( sblock_r != NULL )
*sblock_r = NULL;
}
sbin = NULL;
} else {
if ( sblock_r != NULL )
*sblock_r = sblock;
}
return sbin;
}
