/**
* Constructor.
* @param[in] module The IronBee++ module.
*/
explicit EngineShutdownModule(IronBee::Module module) :
IronBee::ModuleDelegate(module),
m_mode(RUNNING)
{
assert(module);
module.engine().register_hooks()
.transaction_started(
boost::bind(
&EngineShutdownModule::on_transaction_started,
this,
_1,
_2
)
)
.response_header_data(
boost::bind(
&EngineShutdownModule::on_response_header_data,
this,
_1,
_2,
_3,
_4
)
)
.connection_opened(
boost::bind(
&EngineShutdownModule::on_connection_opened,
this,
_1,
_2
)
)
.engine_shutdown_initiated(
boost::bind(
&EngineShutdownModule::on_engine_shutdown_initiated,
this,
_1
)
);
ib_log_info(
module.engine().ib(),
"The engine_shutdown module is deprecated. "
"Use the new graceful_shutdown module instead."
);
}
Delegate::Delegate(IB::Module module) :
IB::ModuleDelegate(module),
m_graph(new P::MergeGraph()),
m_write_debug_report(false)
{
assert(module);
// Call factory.
P::Standard::load(m_call_factory);
// Configuration data.
PerContext base(*this);
module.set_configuration_data<PerContext>(base);
// Ownership Function.
pair<ib_rule_ownership_fn_t, void*> owner =
IB::make_c_trampoline<
ib_status_t(
const ib_engine_t*,
const ib_rule_t*
)
>(bind(&Delegate::ownership, this, _1, _2));
register_trampoline_data(owner.second);
IB::throw_if_error(
ib_rule_register_ownership_fn(
module.engine().ib(),
"predicate",
owner.first, owner.second
)
);
// Injection Functions.
for (unsigned int i = 0; i < c_num_phases; ++i) {
ib_rule_phase_num_t phase = c_phases[i];
pair<ib_rule_injection_fn_t, void*> injection =
IB::make_c_trampoline<
ib_status_t(
const ib_engine_t*,
const ib_rule_exec_t*,
ib_list_t*
)
>(bind(&Delegate::injection, this, _2, _3));
register_trampoline_data(injection.second);
IB::throw_if_error(
ib_rule_register_injection_fn(
module.engine().ib(),
c_module_name,
phase,
injection.first, injection.second
)
);
}
// 'predicate' action.
pair<ib_action_create_fn_t, void*> action_create =
IB::make_c_trampoline<
ib_status_t(
ib_engine_t*,
ib_context_t*,
ib_mpool_t*,
const char*,
ib_action_inst_t*
)
>(bind(&Delegate::action_create, this, _3, _4, _5));
register_trampoline_data(action_create.second);
IB::throw_if_error(
ib_action_register(
module.engine().ib(),
c_predicate_action,
IB_ACT_FLAG_NONE,
action_create.first, action_create.second,
NULL, NULL,
NULL, NULL
)
);
// Hooks
module.engine().register_hooks()
.request_started(
boost::bind(&Delegate::request_started, this, _2)
)
.context_close(
boost::bind(&Delegate::context_close, this, _2)
)
;
// Introspection directives.
module.engine().register_configuration_directives()
.param1(
c_assert_valid_directive,
bind(&Delegate::assert_valid, this, _1, _3)
)
.param1(
c_debug_report_directive,
bind(&Delegate::debug_report, this, _1, _3)
)
;
}