flipperzero-firmware/lib/lfrfid/protocols/protocol_pyramid.c
2023-01-07 01:06:50 +09:00

283 lines
9.0 KiB
C

#include <furi.h>
#include <toolbox/protocols/protocol.h>
#include <lfrfid/tools/fsk_demod.h>
#include <lfrfid/tools/fsk_osc.h>
#include "lfrfid_protocols.h"
#include <lfrfid/tools/bit_lib.h>
#define JITTER_TIME (20)
#define MIN_TIME (64 - JITTER_TIME)
#define MAX_TIME (80 + JITTER_TIME)
#define PYRAMID_DATA_SIZE 13
#define PYRAMID_PREAMBLE_SIZE 3
#define PYRAMID_ENCODED_DATA_SIZE \
(PYRAMID_PREAMBLE_SIZE + PYRAMID_DATA_SIZE + PYRAMID_PREAMBLE_SIZE)
#define PYRAMID_ENCODED_BIT_SIZE ((PYRAMID_PREAMBLE_SIZE + PYRAMID_DATA_SIZE) * 8)
#define PYRAMID_DECODED_DATA_SIZE (4)
#define PYRAMID_DECODED_BIT_SIZE ((PYRAMID_ENCODED_BIT_SIZE - PYRAMID_PREAMBLE_SIZE * 8) / 2)
typedef struct {
FSKDemod* fsk_demod;
} ProtocolPyramidDecoder;
typedef struct {
FSKOsc* fsk_osc;
uint8_t encoded_index;
uint32_t pulse;
} ProtocolPyramidEncoder;
typedef struct {
ProtocolPyramidDecoder decoder;
ProtocolPyramidEncoder encoder;
uint8_t encoded_data[PYRAMID_ENCODED_DATA_SIZE];
uint8_t data[PYRAMID_DECODED_DATA_SIZE];
} ProtocolPyramid;
ProtocolPyramid* protocol_pyramid_alloc(void) {
ProtocolPyramid* protocol = malloc(sizeof(ProtocolPyramid));
protocol->decoder.fsk_demod = fsk_demod_alloc(MIN_TIME, 6, MAX_TIME, 5);
protocol->encoder.fsk_osc = fsk_osc_alloc(8, 10, 50);
return protocol;
};
void protocol_pyramid_free(ProtocolPyramid* protocol) {
fsk_demod_free(protocol->decoder.fsk_demod);
fsk_osc_free(protocol->encoder.fsk_osc);
free(protocol);
};
uint8_t* protocol_pyramid_get_data(ProtocolPyramid* protocol) {
return protocol->data;
};
void protocol_pyramid_decoder_start(ProtocolPyramid* protocol) {
memset(protocol->encoded_data, 0, PYRAMID_ENCODED_DATA_SIZE);
};
static bool protocol_pyramid_can_be_decoded(uint8_t* data) {
// check preamble
if(bit_lib_get_bits_16(data, 0, 16) != 0b0000000000000001 ||
bit_lib_get_bits(data, 16, 8) != 0b00000001) {
return false;
}
if(bit_lib_get_bits_16(data, 128, 16) != 0b0000000000000001 ||
bit_lib_get_bits(data, 136, 8) != 0b00000001) {
return false;
}
uint8_t checksum = bit_lib_get_bits(data, 120, 8);
uint8_t checksum_data[13] = {0x00};
for(uint8_t i = 0; i < 13; i++) {
checksum_data[i] = bit_lib_get_bits(data, 16 + (i * 8), 8);
}
uint8_t calc_checksum = bit_lib_crc8(checksum_data, 13, 0x31, 0x00, true, true, 0x00);
if(checksum != calc_checksum) return false;
// Remove parity
bit_lib_remove_bit_every_nth(data, 8, 15 * 8, 8);
// Determine Startbit and format
int j;
for(j = 0; j < 105; ++j) {
if(bit_lib_get_bit(data, j)) break;
}
uint8_t fmt_len = 105 - j;
// Only support 26bit format for now
if(fmt_len != 26) return false;
return true;
}
static void protocol_pyramid_decode(ProtocolPyramid* protocol) {
// Format
bit_lib_set_bits(protocol->data, 0, 26, 8);
// Facility Code
bit_lib_copy_bits(protocol->data, 8, 8, protocol->encoded_data, 73 + 8);
// Card Number
bit_lib_copy_bits(protocol->data, 16, 16, protocol->encoded_data, 81 + 8);
}
bool protocol_pyramid_decoder_feed(ProtocolPyramid* protocol, bool level, uint32_t duration) {
bool value;
uint32_t count;
bool result = false;
fsk_demod_feed(protocol->decoder.fsk_demod, level, duration, &value, &count);
if(count > 0) {
for(size_t i = 0; i < count; i++) {
bit_lib_push_bit(protocol->encoded_data, PYRAMID_ENCODED_DATA_SIZE, value);
if(protocol_pyramid_can_be_decoded(protocol->encoded_data)) {
protocol_pyramid_decode(protocol);
result = true;
}
}
}
return result;
};
bool protocol_pyramid_get_parity(const uint8_t* bits, uint8_t type, int length) {
int x;
for(x = 0; length > 0; --length) x += bit_lib_get_bit(bits, length - 1);
x %= 2;
return x ^ type;
}
void protocol_pyramid_add_wiegand_parity(
uint8_t* target,
uint8_t target_position,
uint8_t* source,
uint8_t length) {
bit_lib_set_bit(
target, target_position, protocol_pyramid_get_parity(source, 0 /* even */, length / 2));
bit_lib_copy_bits(target, target_position + 1, length, source, 0);
bit_lib_set_bit(
target,
target_position + length + 1,
protocol_pyramid_get_parity(source + length / 2, 1 /* odd */, length / 2));
}
static void protocol_pyramid_encode(ProtocolPyramid* protocol) {
memset(protocol->encoded_data, 0, sizeof(protocol->encoded_data));
uint8_t pre[16];
memset(pre, 0, sizeof(pre));
// Format start bit
bit_lib_set_bit(pre, 79, 1);
uint8_t wiegand[3];
memset(wiegand, 0, sizeof(wiegand));
// FC
bit_lib_copy_bits(wiegand, 0, 8, protocol->data, 8);
// CardNum
bit_lib_copy_bits(wiegand, 8, 16, protocol->data, 16);
// Wiegand parity
protocol_pyramid_add_wiegand_parity(pre, 80, wiegand, 24);
bit_lib_add_parity(pre, 8, protocol->encoded_data, 8, 102, 8, 1);
// Add checksum
uint8_t checksum_buffer[13];
for(uint8_t i = 0; i < 13; i++)
checksum_buffer[i] = bit_lib_get_bits(protocol->encoded_data, 16 + (i * 8), 8);
uint8_t crc = bit_lib_crc8(checksum_buffer, 13, 0x31, 0x00, true, true, 0x00);
bit_lib_set_bits(protocol->encoded_data, 120, crc, 8);
}
bool protocol_pyramid_encoder_start(ProtocolPyramid* protocol) {
protocol->encoder.encoded_index = 0;
protocol->encoder.pulse = 0;
protocol_pyramid_encode(protocol);
return true;
};
LevelDuration protocol_pyramid_encoder_yield(ProtocolPyramid* protocol) {
bool level = 0;
uint32_t duration = 0;
// if pulse is zero, we need to output high, otherwise we need to output low
if(protocol->encoder.pulse == 0) {
// get bit
uint8_t bit = bit_lib_get_bit(protocol->encoded_data, protocol->encoder.encoded_index);
// get pulse from oscillator
bool advance = fsk_osc_next(protocol->encoder.fsk_osc, bit, &duration);
if(advance) {
bit_lib_increment_index(protocol->encoder.encoded_index, PYRAMID_ENCODED_BIT_SIZE);
}
// duration diveded by 2 because we need to output high and low
duration = duration / 2;
protocol->encoder.pulse = duration;
level = true;
} else {
// output low half and reset pulse
duration = protocol->encoder.pulse;
protocol->encoder.pulse = 0;
level = false;
}
return level_duration_make(level, duration);
};
bool protocol_pyramid_write_data(ProtocolPyramid* protocol, void* data) {
LFRFIDWriteRequest* request = (LFRFIDWriteRequest*)data;
bool result = false;
// Correct protocol data by redecoding
protocol_pyramid_encode(protocol);
bit_lib_remove_bit_every_nth(protocol->encoded_data, 8, 15 * 8, 8);
protocol_pyramid_decode(protocol);
protocol_pyramid_encoder_start(protocol);
if(request->write_type == LFRFIDWriteTypeT5577) {
request->t5577.block[0] = LFRFID_T5577_MODULATION_FSK2a | LFRFID_T5577_BITRATE_RF_50 |
(4 << LFRFID_T5577_MAXBLOCK_SHIFT);
request->t5577.block[1] = bit_lib_get_bits_32(protocol->encoded_data, 0, 32);
request->t5577.block[2] = bit_lib_get_bits_32(protocol->encoded_data, 32, 32);
request->t5577.block[3] = bit_lib_get_bits_32(protocol->encoded_data, 64, 32);
request->t5577.block[4] = bit_lib_get_bits_32(protocol->encoded_data, 96, 32);
request->t5577.blocks_to_write = 5;
result = true;
}
return result;
};
void protocol_pyramid_render_data(ProtocolPyramid* protocol, FuriString* result) {
uint8_t* decoded_data = protocol->data;
uint8_t format_length = decoded_data[0];
furi_string_cat_printf(result, "Format: %d\r\n", format_length);
if(format_length == 26) {
uint8_t facility;
bit_lib_copy_bits(&facility, 0, 8, decoded_data, 8);
uint16_t card_id;
bit_lib_copy_bits((uint8_t*)&card_id, 8, 8, decoded_data, 16);
bit_lib_copy_bits((uint8_t*)&card_id, 0, 8, decoded_data, 24);
furi_string_cat_printf(result, "FC: %03u, Card: %05u", facility, card_id);
} else {
furi_string_cat_printf(result, "Data: unknown");
}
};
const ProtocolBase protocol_pyramid = {
.name = "Pyramid",
.manufacturer = "Farpointe",
.data_size = PYRAMID_DECODED_DATA_SIZE,
.features = LFRFIDFeatureASK,
.validate_count = 3,
.alloc = (ProtocolAlloc)protocol_pyramid_alloc,
.free = (ProtocolFree)protocol_pyramid_free,
.get_data = (ProtocolGetData)protocol_pyramid_get_data,
.decoder =
{
.start = (ProtocolDecoderStart)protocol_pyramid_decoder_start,
.feed = (ProtocolDecoderFeed)protocol_pyramid_decoder_feed,
},
.encoder =
{
.start = (ProtocolEncoderStart)protocol_pyramid_encoder_start,
.yield = (ProtocolEncoderYield)protocol_pyramid_encoder_yield,
},
.render_data = (ProtocolRenderData)protocol_pyramid_render_data,
.render_brief_data = (ProtocolRenderData)protocol_pyramid_render_data,
.write_data = (ProtocolWriteData)protocol_pyramid_write_data,
};