65 lines
1.4 KiB
C++
65 lines
1.4 KiB
C++
#include "crypto_module_du_handler.h"
|
|
|
|
#include "abstract_data_unit.h"
|
|
|
|
#include <boost/format.hpp>
|
|
#include <sstream>
|
|
#include <stdio.h>
|
|
|
|
crypto_module_du_handler::crypto_module_du_handler(data_unit_handler_sptr next, crypto_module::sptr crypto_mod)
|
|
: data_unit_handler(next)
|
|
, d_crypto_mod(crypto_mod)
|
|
{
|
|
}
|
|
|
|
void
|
|
crypto_module_du_handler::handle(data_unit_sptr du)
|
|
{
|
|
if (!d_crypto_mod)
|
|
{
|
|
data_unit_handler::handle(du);
|
|
return;
|
|
}
|
|
|
|
crypto_state_provider* p = dynamic_cast<crypto_state_provider*>(du.get());
|
|
if (p == NULL)
|
|
{
|
|
data_unit_handler::handle(du);
|
|
return;
|
|
}
|
|
|
|
CryptoState state = p->crypto_state();
|
|
|
|
///////////////////////////////////
|
|
|
|
if (d_crypto_mod->logging_enabled())
|
|
{
|
|
std::string duid_str("?");
|
|
abstract_data_unit* adu = dynamic_cast<abstract_data_unit*>(du.get());
|
|
if (adu)
|
|
duid_str = adu->duid_str();
|
|
|
|
std::stringstream ss;
|
|
for (size_t n = 0; n < state.mi.size(); ++n)
|
|
ss << (boost::format("%02x") % (int)state.mi[n]);
|
|
|
|
fprintf(stderr, "%s:\tAlgID: 0x%02x, KID: 0x%04x, MI: %s\n", duid_str.c_str(), state.algid, state.kid, ss.str().c_str());
|
|
}
|
|
|
|
///////////////////////////////////
|
|
|
|
crypto_algorithm::sptr algorithm = d_crypto_mod->algorithm(state.algid);
|
|
if (!algorithm)
|
|
{
|
|
data_unit_handler::handle(du);
|
|
return;
|
|
}
|
|
|
|
// TODO: Could do key management & selection here with 'state.kid'
|
|
// Assuming we're only using one key (ignoring 'kid')
|
|
|
|
algorithm->update(state);
|
|
|
|
data_unit_handler::handle(du);
|
|
}
|