mirror of
https://github.com/DarkFlippers/unleashed-firmware.git
synced 2024-12-21 04:11:48 +03:00
283 lines
9.0 KiB
C
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,
|
|
};
|