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,
 | 
						|
};
 |