changed the input stream logic to separate initial frame and continuous stream to work() function
parent
0b5e5e425f
commit
bdb1aaba65
|
@ -95,15 +95,9 @@ namespace gr
|
||||||
memset (_key, 0, 32 * sizeof (uint8_t));
|
memset (_key, 0, 32 * sizeof (uint8_t));
|
||||||
memset (_iv, 0, 16 * sizeof (uint8_t));
|
memset (_iv, 0, 16 * sizeof (uint8_t));
|
||||||
#endif
|
#endif
|
||||||
if (_encr_type == ENCR_AES)
|
|
||||||
{
|
|
||||||
set_key (key); // read key
|
|
||||||
*((int32_t *) & _iv[0]) = (uint32_t) time (NULL) - (uint32_t) epoch; //timestamp
|
|
||||||
for (uint8_t i = 4; i < 4 + 10; i++)
|
|
||||||
_iv[i] = 0; //10 random bytes TODO: replace with a rand() or pass through an additional arg
|
|
||||||
}
|
|
||||||
message_port_register_in(pmt::mp("end_of_transmission"));
|
message_port_register_in(pmt::mp("end_of_transmission"));
|
||||||
set_msg_handler(pmt::mp("end_of_transmission"), [this](const pmt::pmt_t& msg) { end_of_transmission(msg); });
|
set_msg_handler(pmt::mp("end_of_transmission"), [this](const pmt::pmt_t& msg) { end_of_transmission(msg); });
|
||||||
|
_init_frame=true;
|
||||||
}
|
}
|
||||||
|
|
||||||
void m17_coder_impl::end_of_transmission(const pmt::pmt_t& msg)
|
void m17_coder_impl::end_of_transmission(const pmt::pmt_t& msg)
|
||||||
|
@ -115,10 +109,10 @@ namespace gr
|
||||||
void m17_coder_impl::set_encr_type (int encr_type)
|
void m17_coder_impl::set_encr_type (int encr_type)
|
||||||
{
|
{
|
||||||
switch (encr_type)
|
switch (encr_type)
|
||||||
{case 0:_encr_type=ENCR_NONE;break;
|
{case 0:_encr_type=ENCR_NONE; break;
|
||||||
case 1:_encr_type=ENCR_SCRAM;break;
|
case 1:_encr_type=ENCR_SCRAM;break;
|
||||||
case 2:_encr_type=ENCR_AES;break;
|
case 2:_encr_type=ENCR_AES; break;
|
||||||
case 3:_encr_type=ENCR_RES;break;
|
case 3:_encr_type=ENCR_RES; break;
|
||||||
default:_encr_type=ENCR_NONE;
|
default:_encr_type=ENCR_NONE;
|
||||||
}
|
}
|
||||||
printf ("new encr type: %x -> ", _encr_type);
|
printf ("new encr type: %x -> ", _encr_type);
|
||||||
|
@ -546,6 +540,31 @@ namespace gr
|
||||||
|
|
||||||
uint8_t data[16], next_data[16]; //raw payload, packed bits
|
uint8_t data[16], next_data[16]; //raw payload, packed bits
|
||||||
|
|
||||||
|
if (_init_frame==true)
|
||||||
|
{
|
||||||
|
if(_encr_type==ENCR_AES)
|
||||||
|
{
|
||||||
|
for(uint8_t i=0; i<4; i++)
|
||||||
|
_iv[i] = ((uint32_t)(time(NULL)&0xFFFFFFFF)-(uint32_t)epoch) >> (24-(i*8));
|
||||||
|
for(uint8_t i=3; i<14; i++)
|
||||||
|
_iv[i] = rand() & 0xFF; //10 random bytes
|
||||||
|
}
|
||||||
|
send_preamble(out, &countout, PREAM_LSF); //0 - LSF preamble, as opposed to 1 - BERT preamble
|
||||||
|
|
||||||
|
if(_encr_type==ENCR_AES)
|
||||||
|
{
|
||||||
|
memcpy(&(_lsf.meta), _iv, 14);
|
||||||
|
_iv[14] = (_fn >> 8) & 0x7F;
|
||||||
|
_iv[15] = (_fn >> 0) & 0xFF;
|
||||||
|
|
||||||
|
//re-calculate LSF CRC with IV insertion
|
||||||
|
uint16_t ccrc=LSF_CRC(&_lsf);
|
||||||
|
_lsf.crc[0]=ccrc>>8;
|
||||||
|
_lsf.crc[1]=ccrc&0xFF;
|
||||||
|
}
|
||||||
|
_init_frame=false;
|
||||||
|
} // end of init frame
|
||||||
|
|
||||||
while ((countout < (uint32_t) noutput_items) && (countin + 16 <= noutput_items))
|
while ((countout < (uint32_t) noutput_items) && (countin + 16 <= noutput_items))
|
||||||
{
|
{
|
||||||
if (!_got_lsf) //stream frames
|
if (!_got_lsf) //stream frames
|
||||||
|
@ -572,178 +591,7 @@ namespace gr
|
||||||
_got_lsf = 1;
|
_got_lsf = 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (_debug == true)
|
if (countin + 16 < noutput_items)
|
||||||
{
|
|
||||||
//destination set to "ALL"
|
|
||||||
memset (_next_lsf.dst, 0xFF, 6 * sizeof (uint8_t));
|
|
||||||
|
|
||||||
//source set to "N0CALL"
|
|
||||||
_next_lsf.src[0] = 0x00;
|
|
||||||
_next_lsf.src[1] = 0x00;
|
|
||||||
_next_lsf.src[2] = 0x4B;
|
|
||||||
_next_lsf.src[3] = 0x13;
|
|
||||||
_next_lsf.src[4] = 0xD1;
|
|
||||||
_next_lsf.src[5] = 0x06;
|
|
||||||
|
|
||||||
if (_encr_type == ENCR_AES) //AES ENC, 3200 voice
|
|
||||||
{
|
|
||||||
_next_lsf.type[0] = 0x03;
|
|
||||||
_next_lsf.type[1] = 0x95;
|
|
||||||
}
|
|
||||||
else if (_encr_type == ENCR_SCRAM) //Scrambler ENC, 3200 Voice
|
|
||||||
{
|
|
||||||
_next_lsf.type[0] = 0x00;
|
|
||||||
_next_lsf.type[1] = 0x00;
|
|
||||||
if (_scrambler_subtype == 0)
|
|
||||||
_next_lsf.type[1] = 0x0D;
|
|
||||||
else if (_scrambler_subtype == 1)
|
|
||||||
_next_lsf.type[1] = 0x2D;
|
|
||||||
else if (_scrambler_subtype == 2)
|
|
||||||
_next_lsf.type[1] = 0x4D;
|
|
||||||
}
|
|
||||||
else //no enc or subtype field, normal 3200 voice
|
|
||||||
{
|
|
||||||
_next_lsf.type[0] = 0x00;
|
|
||||||
_next_lsf.type[1] = 0x05;
|
|
||||||
}
|
|
||||||
|
|
||||||
//a signature key is loaded, OR this bit
|
|
||||||
if (_priv_key_loaded == true)
|
|
||||||
_next_lsf.type[0] |= 0x8;
|
|
||||||
|
|
||||||
_finished = false;
|
|
||||||
|
|
||||||
memset (next_data, 0, sizeof (next_data));
|
|
||||||
memcpy (data, next_data, sizeof (data));
|
|
||||||
// if (_fn == 60)
|
|
||||||
// _finished = true;
|
|
||||||
|
|
||||||
//debug sig with random payloads (don't play the audio)
|
|
||||||
for (uint8_t i = 0; i < 16; i++)
|
|
||||||
data[i] = 0x69; //rand() & 0xFF;
|
|
||||||
|
|
||||||
}
|
|
||||||
/*
|
|
||||||
//TODO: pass some of these through arguments?
|
|
||||||
//read data
|
|
||||||
dummy=fread(&(lsf.dst), 6, 1, stdin);
|
|
||||||
dummy=fread(&(lsf.src), 6, 1, stdin);
|
|
||||||
dummy=fread(&(lsf.type), 2, 1, stdin);
|
|
||||||
dummy=fread(&(lsf.meta), 14, 1, stdin);
|
|
||||||
dummy=fread(data, 16, 1, stdin);
|
|
||||||
*/
|
|
||||||
if (countin + 16 <= noutput_items)
|
|
||||||
{
|
|
||||||
for (int i = 0; i < 16; i++)
|
|
||||||
{
|
|
||||||
data[i] = in[countin];
|
|
||||||
countin++;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//AES encryption enabled - use 112 bits of IV
|
|
||||||
if (_encr_type == ENCR_AES)
|
|
||||||
{
|
|
||||||
memcpy (&(_lsf.meta), _iv, 14);
|
|
||||||
_iv[14] = (_fn >> 8) & 0x7F;
|
|
||||||
_iv[15] = (_fn >> 0) & 0xFF;
|
|
||||||
|
|
||||||
//re-calculate LSF CRC with IV insertion
|
|
||||||
uint16_t ccrc = LSF_CRC (&_lsf);
|
|
||||||
_lsf.crc[0] = ccrc >> 8;
|
|
||||||
_lsf.crc[1] = ccrc & 0xFF;
|
|
||||||
}
|
|
||||||
|
|
||||||
// while (_finished == false)
|
|
||||||
{
|
|
||||||
if (!_got_lsf)
|
|
||||||
{ //debug
|
|
||||||
//fprintf(stderr, "LSF\n");
|
|
||||||
|
|
||||||
//send LSF syncword
|
|
||||||
send_syncword (out, &countout, SYNC_LSF);
|
|
||||||
|
|
||||||
//encode LSF data
|
|
||||||
conv_encode_LSF (enc_bits, &_lsf);
|
|
||||||
|
|
||||||
//reorder bits
|
|
||||||
reorder_bits (rf_bits, enc_bits);
|
|
||||||
|
|
||||||
//randomize
|
|
||||||
randomize_bits (rf_bits);
|
|
||||||
|
|
||||||
//send LSF data
|
|
||||||
send_data (out, &countout, rf_bits);
|
|
||||||
|
|
||||||
//check the SIGNED STREAM flag
|
|
||||||
_signed_str = (_lsf.type[0] >> 3) & 1;
|
|
||||||
|
|
||||||
//set the flag
|
|
||||||
_got_lsf = 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (_debug == true)
|
|
||||||
{
|
|
||||||
//destination set to "ALL"
|
|
||||||
memset (_next_lsf.dst, 0xFF, 6 * sizeof (uint8_t));
|
|
||||||
|
|
||||||
//source set to "N0CALL"
|
|
||||||
_next_lsf.src[0] = 0x00;
|
|
||||||
_next_lsf.src[1] = 0x00;
|
|
||||||
_next_lsf.src[2] = 0x4B;
|
|
||||||
_next_lsf.src[3] = 0x13;
|
|
||||||
_next_lsf.src[4] = 0xD1;
|
|
||||||
_next_lsf.src[5] = 0x06;
|
|
||||||
|
|
||||||
if (_encr_type == ENCR_AES) //AES ENC, 3200 voice
|
|
||||||
{
|
|
||||||
_next_lsf.type[0] = 0x03;
|
|
||||||
_next_lsf.type[1] = 0x95;
|
|
||||||
}
|
|
||||||
else if (_encr_type == ENCR_SCRAM) //Scrambler ENC, 3200 Voice
|
|
||||||
{
|
|
||||||
_next_lsf.type[0] = 0x00;
|
|
||||||
_next_lsf.type[1] = 0x00;
|
|
||||||
if (_scrambler_subtype == 0)
|
|
||||||
_next_lsf.type[1] = 0x0D;
|
|
||||||
else if (_scrambler_subtype == 1)
|
|
||||||
_next_lsf.type[1] = 0x2D;
|
|
||||||
else if (_scrambler_subtype == 2)
|
|
||||||
_next_lsf.type[1] = 0x4D;
|
|
||||||
}
|
|
||||||
else //no enc or subtype field, normal 3200 voice
|
|
||||||
{
|
|
||||||
_next_lsf.type[0] = 0x00;
|
|
||||||
_next_lsf.type[1] = 0x05;
|
|
||||||
}
|
|
||||||
|
|
||||||
//a signature key is loaded, OR this bit
|
|
||||||
if (_priv_key_loaded == true)
|
|
||||||
_next_lsf.type[0] |= 0x8;
|
|
||||||
|
|
||||||
_finished = false;
|
|
||||||
|
|
||||||
memset (next_data, 0, sizeof (next_data));
|
|
||||||
memcpy (data, next_data, sizeof (data));
|
|
||||||
if (_fn == 60)
|
|
||||||
_finished = 1;
|
|
||||||
|
|
||||||
//debug sig with random payloads (don't play the audio)
|
|
||||||
for (uint8_t i = 0; i < 16; i++)
|
|
||||||
data[i] = 0x69; //rand() & 0xFF;
|
|
||||||
|
|
||||||
} // end of debug==true
|
|
||||||
else
|
|
||||||
{
|
|
||||||
/*
|
|
||||||
//check if theres any more data
|
|
||||||
if(fread(&(next_lsf.dst), 6, 1, stdin)<1) finished=1;
|
|
||||||
if(fread(&(next_lsf.src), 6, 1, stdin)<1) finished=1;
|
|
||||||
if(fread(&(next_lsf.type), 2, 1, stdin)<1) finished=1;
|
|
||||||
if(fread(&(next_lsf.meta), 14, 1, stdin)<1) finished=1;
|
|
||||||
if(fread(next_data, 16, 1, stdin)<1) _finished=true;
|
|
||||||
*/
|
|
||||||
if (countin + 16 <= noutput_items)
|
|
||||||
{
|
{
|
||||||
for (int i = 0; i < 16; i++)
|
for (int i = 0; i < 16; i++)
|
||||||
{
|
{
|
||||||
|
@ -751,7 +599,6 @@ namespace gr
|
||||||
countin++;
|
countin++;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
//AES
|
//AES
|
||||||
if (_encr_type == ENCR_AES)
|
if (_encr_type == ENCR_AES)
|
||||||
|
@ -874,7 +721,6 @@ namespace gr
|
||||||
send_eot (out, &countout);
|
send_eot (out, &countout);
|
||||||
//fprintf(stderr, "Stream has ended. Exiting.\n");
|
//fprintf(stderr, "Stream has ended. Exiting.\n");
|
||||||
} // finished == true
|
} // finished == true
|
||||||
} // finished == false
|
|
||||||
} // loop on input data
|
} // loop on input data
|
||||||
// Tell runtime system how many input items we consumed on
|
// Tell runtime system how many input items we consumed on
|
||||||
// each input stream.
|
// each input stream.
|
||||||
|
|
|
@ -62,7 +62,7 @@ namespace gr
|
||||||
bool _priv_key_loaded = false; //do we have a sig key loaded?
|
bool _priv_key_loaded = false; //do we have a sig key loaded?
|
||||||
uint8_t _priv_key[32] = { 0 }; //private key
|
uint8_t _priv_key[32] = { 0 }; //private key
|
||||||
uint8_t _sig[64] = { 0 }; //ECDSA signature
|
uint8_t _sig[64] = { 0 }; //ECDSA signature
|
||||||
|
bool _init_frame;
|
||||||
|
|
||||||
#ifdef ECC
|
#ifdef ECC
|
||||||
//Scrambler
|
//Scrambler
|
||||||
|
|
Loading…
Reference in New Issue