83 lines
2.3 KiB
C
83 lines
2.3 KiB
C
// Copyright 2022 Framework Computer
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
#include "quantum.h"
|
|
#include "dyn_serial.h"
|
|
#include "usb_descriptor.h"
|
|
#include "usb_descriptor_common.h"
|
|
|
|
// Prefix string literal with L for descriptors
|
|
#ifdef SERIAL_NUMBER
|
|
USB_Descriptor_String_t PROGMEM SerialNumberString = {
|
|
.Header = {
|
|
.Size = sizeof(FALLBACK_SERIAL_NUMBER),
|
|
.Type = DTYPE_String
|
|
},
|
|
.UnicodeString = FALLBACK_SERIAL_NUMBER
|
|
};
|
|
|
|
char ascii_serialnum[SERIALNUM_LEN+1];
|
|
|
|
struct __attribute__((__packed__)) serialnum_raw {
|
|
uint8_t sn_rev;
|
|
uint8_t serialnum[SERIALNUM_LEN];
|
|
uint8_t crc32[4];
|
|
};
|
|
|
|
uint32_t crc32b(uint8_t *message, size_t msg_len) {
|
|
uint8_t byte;
|
|
uint32_t crc;
|
|
uint32_t mask;
|
|
|
|
crc = 0xFFFFFFFF;
|
|
for (size_t i = 0; i < msg_len; i++) {
|
|
byte = message[i];
|
|
crc = crc ^ byte;
|
|
for (int j = 0; j < 8; j++) {
|
|
mask = -(crc & 1);
|
|
crc = (crc >> 1) ^ (0xEDB88320 & mask);
|
|
}
|
|
}
|
|
return ~crc;
|
|
}
|
|
|
|
void *dyn_serial_number_string(void) {
|
|
// Read ASCII serial number from memory-mapped flash
|
|
void *serialnum_ptr = (void*) (FLASH_OFFSET + LAST_4K_BLOCK);
|
|
struct serialnum_raw *serialnum_raw = serialnum_ptr;
|
|
|
|
// Only rev 1 supported
|
|
if (serialnum_raw->sn_rev != 1) {
|
|
return &SerialNumberString;
|
|
}
|
|
|
|
uint32_t crc32 = crc32b(serialnum_ptr, SERIALNUM_LEN+1);
|
|
// Make sure it's little-endian
|
|
uint32_t crc32_read = (serialnum_raw->crc32[3] << 24) +
|
|
(serialnum_raw->crc32[2] << 16) +
|
|
(serialnum_raw->crc32[1] << 8) +
|
|
serialnum_raw->crc32[0];
|
|
|
|
// CRC invalid, fall back to defult serial number
|
|
if (crc32_read != crc32) {
|
|
return &SerialNumberString;
|
|
}
|
|
|
|
// Convert to UCS-2, which is equivalent to UTF-16, if the input is ASCII
|
|
memcpy(ascii_serialnum, serialnum_raw->serialnum, SERIALNUM_LEN);
|
|
for (int i = 0; i < SERIALNUM_LEN; i++) {
|
|
if (ascii_serialnum[i] > 128) {
|
|
dprintf("Serial number character %d is not valid ASCII.", ascii_serialnum[i]);
|
|
SerialNumberString.Header.Size = i * 2;
|
|
break;
|
|
}
|
|
SerialNumberString.UnicodeString[i] = ascii_serialnum[i];
|
|
}
|
|
|
|
return &SerialNumberString;
|
|
}
|
|
uint16_t dyn_serial_number_string_len(void) {
|
|
return pgm_read_byte(&SerialNumberString.Header.Size);
|
|
}
|
|
#endif
|