Adding asn1_find_tag() for searching for a tag in a asn1 string.

This commit is contained in:
Pol Henarejos
2022-05-30 23:31:17 +02:00
parent ef52ae37d3
commit 950e276ee8
4 changed files with 31 additions and 10 deletions

View File

@@ -553,16 +553,20 @@ int format_tlv_len(size_t len, uint8_t *out) {
return 0; return 0;
} }
int walk_tlv(const uint8_t *cdata, size_t cdata_len, uint8_t **p, uint8_t *tag, size_t *tag_len, uint8_t **data) { int walk_tlv(const uint8_t *cdata, size_t cdata_len, uint8_t **p, uint16_t *tag, size_t *tag_len, uint8_t **data) {
if (!p) if (!p)
return 0; return 0;
if (!*p) if (!*p)
*p = (uint8_t *)cdata; *p = (uint8_t *)cdata;
if (*p-cdata >= cdata_len) if (*p-cdata >= cdata_len)
return 0; return 0;
uint8_t tg = 0x0; uint16_t tg = 0x0;
size_t tgl = 0; size_t tgl = 0;
tg = *(*p)++; tg = *(*p)++;
if ((tg & 0x1f) == 0x1f) {
tg <<= 8;
tg |= *(*p)++;
}
tgl = *(*p)++; tgl = *(*p)++;
if (tgl == 0x82) { if (tgl == 0x82) {
tgl = *(*p)++ << 8; tgl = *(*p)++ << 8;
@@ -581,6 +585,16 @@ int walk_tlv(const uint8_t *cdata, size_t cdata_len, uint8_t **p, uint8_t *tag,
return 1; return 1;
} }
bool asn1_find_tag(const uint8_t *data, size_t data_len, uint16_t itag, size_t *tag_len, uint8_t **tag_data) {
uint16_t tag = 0x0;
uint8_t *p = NULL;
while (walk_tlv(data, data_len, &p, &tag, tag_len, tag_data)) {
if (itag == tag)
return true;
}
return false;
}
void init_rtc() { void init_rtc() {
rtc_init(); rtc_init();

View File

@@ -244,7 +244,8 @@ extern void led_set_blink(uint32_t mode);
#define CCID_WRONG_PADDING -1011 #define CCID_WRONG_PADDING -1011
#define CCID_VERIFICATION_FAILED -1012 #define CCID_VERIFICATION_FAILED -1012
extern int walk_tlv(const uint8_t *cdata, size_t cdata_len, uint8_t **p, uint8_t *tag, size_t *tag_len, uint8_t **data); extern int walk_tlv(const uint8_t *cdata, size_t cdata_len, uint8_t **p, uint16_t *tag, size_t *tag_len, uint8_t **data);
extern int format_tlv_len(size_t len, uint8_t *out); extern int format_tlv_len(size_t len, uint8_t *out);
extern bool asn1_find_tag(const uint8_t *data, size_t data_len, uint16_t itag, size_t *tag_len, uint8_t **tag_data);
#endif #endif

View File

@@ -91,7 +91,8 @@ int sm_unwrap() {
uint8_t *body = NULL; uint8_t *body = NULL;
size_t body_size = 0; size_t body_size = 0;
bool is87 = false; bool is87 = false;
uint8_t tag = 0x0, *tag_data = NULL, *p = NULL; uint16_t tag = 0x0;
uint8_t *tag_data = NULL, *p = NULL;
size_t tag_len = 0; size_t tag_len = 0;
while (walk_tlv(apdu.data, apdu.nc, &p, &tag, &tag_len, &tag_data)) { while (walk_tlv(apdu.data, apdu.nc, &p, &tag, &tag_len, &tag_data)) {
if (tag == 0x87 || tag == 0x85) { if (tag == 0x87 || tag == 0x85) {
@@ -180,7 +181,8 @@ int sm_wrap() {
} }
int sm_get_le() { int sm_get_le() {
uint8_t tag = 0x0, *tag_data = NULL, *p = NULL; uint16_t tag = 0x0;
uint8_t *tag_data = NULL, *p = NULL;
size_t tag_len = 0; size_t tag_len = 0;
while (walk_tlv(apdu.data, apdu.nc, &p, &tag, &tag_len, &tag_data)) { while (walk_tlv(apdu.data, apdu.nc, &p, &tag, &tag_len, &tag_data)) {
if (tag == 0x97) { if (tag == 0x97) {
@@ -231,7 +233,8 @@ int sm_verify() {
bool some_added = false; bool some_added = false;
const uint8_t *mac = NULL; const uint8_t *mac = NULL;
size_t mac_len = 0; size_t mac_len = 0;
uint8_t tag = 0x0, *tag_data = NULL, *p = NULL; uint16_t tag = 0x0;
uint8_t *tag_data = NULL, *p = NULL;
size_t tag_len = 0; size_t tag_len = 0;
while (walk_tlv(apdu.data, apdu.nc, &p, &tag, &tag_len, &tag_data)) { while (walk_tlv(apdu.data, apdu.nc, &p, &tag, &tag_len, &tag_data)) {
if (tag & 0x1) { if (tag & 0x1) {

View File

@@ -305,7 +305,8 @@ int meta_find(uint16_t fid, uint8_t **out) {
file_t *ef = search_by_fid(EF_META, NULL, SPECIFY_EF); file_t *ef = search_by_fid(EF_META, NULL, SPECIFY_EF);
if (!ef) if (!ef)
return CCID_ERR_FILE_NOT_FOUND; return CCID_ERR_FILE_NOT_FOUND;
uint8_t tag = 0x0, *tag_data = NULL, *p = NULL, *data = file_get_data(ef); uint16_t tag = 0x0;
uint8_t *tag_data = NULL, *p = NULL, *data = file_get_data(ef);
size_t tag_len = 0, data_len = file_get_size(ef); size_t tag_len = 0, data_len = file_get_size(ef);
while (walk_tlv(data, data_len, &p, &tag, &tag_len, &tag_data)) { while (walk_tlv(data, data_len, &p, &tag, &tag_len, &tag_data)) {
if (tag_len < 2) if (tag_len < 2)
@@ -323,7 +324,8 @@ int meta_delete(uint16_t fid) {
file_t *ef = search_by_fid(EF_META, NULL, SPECIFY_EF); file_t *ef = search_by_fid(EF_META, NULL, SPECIFY_EF);
if (!ef) if (!ef)
return CCID_ERR_FILE_NOT_FOUND; return CCID_ERR_FILE_NOT_FOUND;
uint8_t tag = 0x0, *tag_data = NULL, *p = NULL, *data = file_get_data(ef); uint16_t tag = 0x0;
uint8_t *tag_data = NULL, *p = NULL, *data = file_get_data(ef);
size_t tag_len = 0, data_len = file_get_size(ef); size_t tag_len = 0, data_len = file_get_size(ef);
uint8_t *fdata = NULL; uint8_t *fdata = NULL;
while (walk_tlv(data, data_len, &p, &tag, &tag_len, &tag_data)) { while (walk_tlv(data, data_len, &p, &tag, &tag_len, &tag_data)) {
@@ -358,7 +360,8 @@ int meta_add(uint16_t fid, const uint8_t *data, uint16_t len) {
uint16_t ef_size = file_get_size(ef); uint16_t ef_size = file_get_size(ef);
uint8_t *fdata = (uint8_t *)calloc(1, ef_size); uint8_t *fdata = (uint8_t *)calloc(1, ef_size);
memcpy(fdata, file_get_data(ef), ef_size); memcpy(fdata, file_get_data(ef), ef_size);
uint8_t tag = 0x0, *tag_data = NULL, *p = NULL; uint16_t tag = 0x0;
uint8_t *tag_data = NULL, *p = NULL;
size_t tag_len = 0; size_t tag_len = 0;
while (walk_tlv(fdata, ef_size, &p, &tag, &tag_len, &tag_data)) { while (walk_tlv(fdata, ef_size, &p, &tag, &tag_len, &tag_data)) {
if (tag_len < 2) if (tag_len < 2)
@@ -399,7 +402,7 @@ int meta_add(uint16_t fid, const uint8_t *data, uint16_t len) {
} }
fdata = (uint8_t *)realloc(fdata, ef_size+1+format_tlv_len(len+2,NULL)+2+len); fdata = (uint8_t *)realloc(fdata, ef_size+1+format_tlv_len(len+2,NULL)+2+len);
uint8_t *f = fdata+ef_size; uint8_t *f = fdata+ef_size;
*f++ = fid & 0xff; *f++ = fid & 0x1f;
f += format_tlv_len(len+2, f); f += format_tlv_len(len+2, f);
*f++ = fid >> 8; *f++ = fid >> 8;
*f++ = fid & 0xff; *f++ = fid & 0xff;