mirror of
https://github.com/meshcore-dev/MeshCore.git
synced 2026-04-20 22:13:47 +00:00
Rewrite KISS modem to be fully spec-compliant
This commit is contained in:
parent
bcb7a8067e
commit
5dcc377b77
5 changed files with 533 additions and 352 deletions
|
|
@ -9,10 +9,20 @@ KissModem::KissModem(Stream& serial, mesh::LocalIdentity& identity, mesh::RNG& r
|
|||
_rx_active = false;
|
||||
_has_pending_tx = false;
|
||||
_pending_tx_len = 0;
|
||||
_txdelay = KISS_DEFAULT_TXDELAY;
|
||||
_persistence = KISS_DEFAULT_PERSISTENCE;
|
||||
_slottime = KISS_DEFAULT_SLOTTIME;
|
||||
_txtail = 0;
|
||||
_fullduplex = 0;
|
||||
_tx_state = TX_IDLE;
|
||||
_tx_timer = 0;
|
||||
_setRadioCallback = nullptr;
|
||||
_setTxPowerCallback = nullptr;
|
||||
_getCurrentRssiCallback = nullptr;
|
||||
_getStatsCallback = nullptr;
|
||||
_sendPacketCallback = nullptr;
|
||||
_isSendCompleteCallback = nullptr;
|
||||
_onSendFinishedCallback = nullptr;
|
||||
_config = {0, 0, 0, 0, 0};
|
||||
}
|
||||
|
||||
|
|
@ -21,6 +31,7 @@ void KissModem::begin() {
|
|||
_rx_escaped = false;
|
||||
_rx_active = false;
|
||||
_has_pending_tx = false;
|
||||
_tx_state = TX_IDLE;
|
||||
}
|
||||
|
||||
void KissModem::writeByte(uint8_t b) {
|
||||
|
|
@ -35,23 +46,33 @@ void KissModem::writeByte(uint8_t b) {
|
|||
}
|
||||
}
|
||||
|
||||
void KissModem::writeFrame(uint8_t cmd, const uint8_t* data, uint16_t len) {
|
||||
void KissModem::writeFrame(uint8_t type, const uint8_t* data, uint16_t len) {
|
||||
_serial.write(KISS_FEND);
|
||||
writeByte(cmd);
|
||||
writeByte(type);
|
||||
for (uint16_t i = 0; i < len; i++) {
|
||||
writeByte(data[i]);
|
||||
}
|
||||
_serial.write(KISS_FEND);
|
||||
}
|
||||
|
||||
void KissModem::writeErrorFrame(uint8_t error_code) {
|
||||
writeFrame(RESP_ERROR, &error_code, 1);
|
||||
void KissModem::writeHardwareFrame(uint8_t sub_cmd, const uint8_t* data, uint16_t len) {
|
||||
_serial.write(KISS_FEND);
|
||||
writeByte(KISS_CMD_SETHARDWARE);
|
||||
writeByte(sub_cmd);
|
||||
for (uint16_t i = 0; i < len; i++) {
|
||||
writeByte(data[i]);
|
||||
}
|
||||
_serial.write(KISS_FEND);
|
||||
}
|
||||
|
||||
void KissModem::writeHardwareError(uint8_t error_code) {
|
||||
writeHardwareFrame(HW_RESP_ERROR, &error_code, 1);
|
||||
}
|
||||
|
||||
void KissModem::loop() {
|
||||
while (_serial.available()) {
|
||||
uint8_t b = _serial.read();
|
||||
|
||||
|
||||
if (b == KISS_FEND) {
|
||||
if (_rx_active && _rx_len > 0) {
|
||||
processFrame();
|
||||
|
|
@ -61,283 +82,368 @@ void KissModem::loop() {
|
|||
_rx_active = true;
|
||||
continue;
|
||||
}
|
||||
|
||||
|
||||
if (!_rx_active) continue;
|
||||
|
||||
|
||||
if (b == KISS_FESC) {
|
||||
_rx_escaped = true;
|
||||
continue;
|
||||
}
|
||||
|
||||
|
||||
if (_rx_escaped) {
|
||||
_rx_escaped = false;
|
||||
if (b == KISS_TFEND) b = KISS_FEND;
|
||||
else if (b == KISS_TFESC) b = KISS_FESC;
|
||||
else continue;
|
||||
}
|
||||
|
||||
|
||||
if (_rx_len < KISS_MAX_FRAME_SIZE) {
|
||||
_rx_buf[_rx_len++] = b;
|
||||
}
|
||||
}
|
||||
|
||||
processTx();
|
||||
}
|
||||
|
||||
void KissModem::processFrame() {
|
||||
if (_rx_len < 1) return;
|
||||
|
||||
uint8_t cmd = _rx_buf[0];
|
||||
|
||||
uint8_t type_byte = _rx_buf[0];
|
||||
|
||||
if (type_byte == KISS_CMD_RETURN) return;
|
||||
|
||||
uint8_t port = (type_byte >> 4) & 0x0F;
|
||||
uint8_t cmd = type_byte & 0x0F;
|
||||
|
||||
if (port != 0) return;
|
||||
|
||||
const uint8_t* data = &_rx_buf[1];
|
||||
uint16_t data_len = _rx_len - 1;
|
||||
|
||||
|
||||
switch (cmd) {
|
||||
case CMD_DATA:
|
||||
if (data_len < 2) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
} else if (data_len > KISS_MAX_PACKET_SIZE) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
} else if (_has_pending_tx) {
|
||||
writeErrorFrame(ERR_TX_PENDING);
|
||||
} else {
|
||||
case KISS_CMD_DATA:
|
||||
if (data_len > 0 && data_len <= KISS_MAX_PACKET_SIZE && !_has_pending_tx) {
|
||||
memcpy(_pending_tx, data, data_len);
|
||||
_pending_tx_len = data_len;
|
||||
_has_pending_tx = true;
|
||||
}
|
||||
break;
|
||||
case CMD_GET_IDENTITY:
|
||||
handleGetIdentity();
|
||||
|
||||
case KISS_CMD_TXDELAY:
|
||||
if (data_len >= 1) _txdelay = data[0];
|
||||
break;
|
||||
case CMD_GET_RANDOM:
|
||||
handleGetRandom(data, data_len);
|
||||
|
||||
case KISS_CMD_PERSISTENCE:
|
||||
if (data_len >= 1) _persistence = data[0];
|
||||
break;
|
||||
case CMD_VERIFY_SIGNATURE:
|
||||
handleVerifySignature(data, data_len);
|
||||
|
||||
case KISS_CMD_SLOTTIME:
|
||||
if (data_len >= 1) _slottime = data[0];
|
||||
break;
|
||||
case CMD_SIGN_DATA:
|
||||
handleSignData(data, data_len);
|
||||
|
||||
case KISS_CMD_TXTAIL:
|
||||
if (data_len >= 1) _txtail = data[0];
|
||||
break;
|
||||
case CMD_ENCRYPT_DATA:
|
||||
handleEncryptData(data, data_len);
|
||||
|
||||
case KISS_CMD_FULLDUPLEX:
|
||||
if (data_len >= 1) _fullduplex = data[0];
|
||||
break;
|
||||
case CMD_DECRYPT_DATA:
|
||||
handleDecryptData(data, data_len);
|
||||
break;
|
||||
case CMD_KEY_EXCHANGE:
|
||||
handleKeyExchange(data, data_len);
|
||||
break;
|
||||
case CMD_HASH:
|
||||
handleHash(data, data_len);
|
||||
break;
|
||||
case CMD_SET_RADIO:
|
||||
handleSetRadio(data, data_len);
|
||||
break;
|
||||
case CMD_SET_TX_POWER:
|
||||
handleSetTxPower(data, data_len);
|
||||
break;
|
||||
case CMD_GET_RADIO:
|
||||
handleGetRadio();
|
||||
break;
|
||||
case CMD_GET_TX_POWER:
|
||||
handleGetTxPower();
|
||||
break;
|
||||
case CMD_GET_VERSION:
|
||||
handleGetVersion();
|
||||
break;
|
||||
case CMD_GET_CURRENT_RSSI:
|
||||
handleGetCurrentRssi();
|
||||
break;
|
||||
case CMD_IS_CHANNEL_BUSY:
|
||||
handleIsChannelBusy();
|
||||
break;
|
||||
case CMD_GET_AIRTIME:
|
||||
handleGetAirtime(data, data_len);
|
||||
break;
|
||||
case CMD_GET_NOISE_FLOOR:
|
||||
handleGetNoiseFloor();
|
||||
break;
|
||||
case CMD_GET_STATS:
|
||||
handleGetStats();
|
||||
break;
|
||||
case CMD_GET_BATTERY:
|
||||
handleGetBattery();
|
||||
break;
|
||||
case CMD_PING:
|
||||
handlePing();
|
||||
break;
|
||||
case CMD_GET_SENSORS:
|
||||
handleGetSensors(data, data_len);
|
||||
|
||||
case KISS_CMD_SETHARDWARE:
|
||||
if (data_len >= 1) {
|
||||
handleHardwareCommand(data[0], data + 1, data_len - 1);
|
||||
}
|
||||
break;
|
||||
|
||||
default:
|
||||
writeErrorFrame(ERR_UNKNOWN_CMD);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void KissModem::handleHardwareCommand(uint8_t sub_cmd, const uint8_t* data, uint16_t len) {
|
||||
switch (sub_cmd) {
|
||||
case HW_CMD_GET_IDENTITY:
|
||||
handleGetIdentity();
|
||||
break;
|
||||
case HW_CMD_GET_RANDOM:
|
||||
handleGetRandom(data, len);
|
||||
break;
|
||||
case HW_CMD_VERIFY_SIGNATURE:
|
||||
handleVerifySignature(data, len);
|
||||
break;
|
||||
case HW_CMD_SIGN_DATA:
|
||||
handleSignData(data, len);
|
||||
break;
|
||||
case HW_CMD_ENCRYPT_DATA:
|
||||
handleEncryptData(data, len);
|
||||
break;
|
||||
case HW_CMD_DECRYPT_DATA:
|
||||
handleDecryptData(data, len);
|
||||
break;
|
||||
case HW_CMD_KEY_EXCHANGE:
|
||||
handleKeyExchange(data, len);
|
||||
break;
|
||||
case HW_CMD_HASH:
|
||||
handleHash(data, len);
|
||||
break;
|
||||
case HW_CMD_SET_RADIO:
|
||||
handleSetRadio(data, len);
|
||||
break;
|
||||
case HW_CMD_SET_TX_POWER:
|
||||
handleSetTxPower(data, len);
|
||||
break;
|
||||
case HW_CMD_GET_RADIO:
|
||||
handleGetRadio();
|
||||
break;
|
||||
case HW_CMD_GET_TX_POWER:
|
||||
handleGetTxPower();
|
||||
break;
|
||||
case HW_CMD_GET_VERSION:
|
||||
handleGetVersion();
|
||||
break;
|
||||
case HW_CMD_GET_CURRENT_RSSI:
|
||||
handleGetCurrentRssi();
|
||||
break;
|
||||
case HW_CMD_IS_CHANNEL_BUSY:
|
||||
handleIsChannelBusy();
|
||||
break;
|
||||
case HW_CMD_GET_AIRTIME:
|
||||
handleGetAirtime(data, len);
|
||||
break;
|
||||
case HW_CMD_GET_NOISE_FLOOR:
|
||||
handleGetNoiseFloor();
|
||||
break;
|
||||
case HW_CMD_GET_STATS:
|
||||
handleGetStats();
|
||||
break;
|
||||
case HW_CMD_GET_BATTERY:
|
||||
handleGetBattery();
|
||||
break;
|
||||
case HW_CMD_PING:
|
||||
handlePing();
|
||||
break;
|
||||
case HW_CMD_GET_SENSORS:
|
||||
handleGetSensors(data, len);
|
||||
break;
|
||||
default:
|
||||
writeHardwareError(HW_ERR_UNKNOWN_CMD);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void KissModem::processTx() {
|
||||
switch (_tx_state) {
|
||||
case TX_IDLE:
|
||||
if (_has_pending_tx) {
|
||||
if (_fullduplex) {
|
||||
_tx_timer = millis();
|
||||
_tx_state = TX_DELAY;
|
||||
} else {
|
||||
_tx_state = TX_WAIT_CLEAR;
|
||||
}
|
||||
}
|
||||
break;
|
||||
|
||||
case TX_WAIT_CLEAR:
|
||||
if (!_radio.isReceiving()) {
|
||||
uint8_t rand_val;
|
||||
_rng.random(&rand_val, 1);
|
||||
if (rand_val <= _persistence) {
|
||||
_tx_timer = millis();
|
||||
_tx_state = TX_DELAY;
|
||||
} else {
|
||||
_tx_timer = millis();
|
||||
_tx_state = TX_SLOT_WAIT;
|
||||
}
|
||||
}
|
||||
break;
|
||||
|
||||
case TX_SLOT_WAIT:
|
||||
if (millis() - _tx_timer >= (uint32_t)_slottime * 10) {
|
||||
_tx_state = TX_WAIT_CLEAR;
|
||||
}
|
||||
break;
|
||||
|
||||
case TX_DELAY:
|
||||
if (millis() - _tx_timer >= (uint32_t)_txdelay * 10) {
|
||||
if (_sendPacketCallback) {
|
||||
_sendPacketCallback(_pending_tx, _pending_tx_len);
|
||||
_tx_state = TX_SENDING;
|
||||
} else {
|
||||
_has_pending_tx = false;
|
||||
_tx_state = TX_IDLE;
|
||||
}
|
||||
}
|
||||
break;
|
||||
|
||||
case TX_SENDING:
|
||||
if (_isSendCompleteCallback && _isSendCompleteCallback()) {
|
||||
if (_onSendFinishedCallback) _onSendFinishedCallback();
|
||||
uint8_t result = 0x01;
|
||||
writeHardwareFrame(HW_RESP_TX_DONE, &result, 1);
|
||||
_has_pending_tx = false;
|
||||
_tx_state = TX_IDLE;
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void KissModem::onPacketReceived(int8_t snr, int8_t rssi, const uint8_t* packet, uint16_t len) {
|
||||
writeFrame(KISS_CMD_DATA, packet, len);
|
||||
uint8_t meta[2] = { (uint8_t)snr, (uint8_t)rssi };
|
||||
writeHardwareFrame(HW_RESP_RX_META, meta, 2);
|
||||
}
|
||||
|
||||
void KissModem::handleGetIdentity() {
|
||||
writeFrame(RESP_IDENTITY, _identity.pub_key, PUB_KEY_SIZE);
|
||||
writeHardwareFrame(HW_RESP_IDENTITY, _identity.pub_key, PUB_KEY_SIZE);
|
||||
}
|
||||
|
||||
void KissModem::handleGetRandom(const uint8_t* data, uint16_t len) {
|
||||
if (len < 1) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
uint8_t requested = data[0];
|
||||
if (requested < 1 || requested > 64) {
|
||||
writeErrorFrame(ERR_INVALID_PARAM);
|
||||
writeHardwareError(HW_ERR_INVALID_PARAM);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
uint8_t buf[64];
|
||||
_rng.random(buf, requested);
|
||||
writeFrame(RESP_RANDOM, buf, requested);
|
||||
writeHardwareFrame(HW_RESP_RANDOM, buf, requested);
|
||||
}
|
||||
|
||||
void KissModem::handleVerifySignature(const uint8_t* data, uint16_t len) {
|
||||
if (len < PUB_KEY_SIZE + SIGNATURE_SIZE + 1) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
mesh::Identity signer(data);
|
||||
const uint8_t* signature = data + PUB_KEY_SIZE;
|
||||
const uint8_t* msg = data + PUB_KEY_SIZE + SIGNATURE_SIZE;
|
||||
uint16_t msg_len = len - PUB_KEY_SIZE - SIGNATURE_SIZE;
|
||||
|
||||
|
||||
uint8_t result = signer.verify(signature, msg, msg_len) ? 0x01 : 0x00;
|
||||
writeFrame(RESP_VERIFY, &result, 1);
|
||||
writeHardwareFrame(HW_RESP_VERIFY, &result, 1);
|
||||
}
|
||||
|
||||
void KissModem::handleSignData(const uint8_t* data, uint16_t len) {
|
||||
if (len < 1) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
uint8_t signature[SIGNATURE_SIZE];
|
||||
_identity.sign(signature, data, len);
|
||||
writeFrame(RESP_SIGNATURE, signature, SIGNATURE_SIZE);
|
||||
writeHardwareFrame(HW_RESP_SIGNATURE, signature, SIGNATURE_SIZE);
|
||||
}
|
||||
|
||||
void KissModem::handleEncryptData(const uint8_t* data, uint16_t len) {
|
||||
if (len < PUB_KEY_SIZE + 1) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
const uint8_t* key = data;
|
||||
const uint8_t* plaintext = data + PUB_KEY_SIZE;
|
||||
uint16_t plaintext_len = len - PUB_KEY_SIZE;
|
||||
|
||||
|
||||
uint8_t buf[KISS_MAX_FRAME_SIZE];
|
||||
int encrypted_len = mesh::Utils::encryptThenMAC(key, buf, plaintext, plaintext_len);
|
||||
|
||||
|
||||
if (encrypted_len > 0) {
|
||||
writeFrame(RESP_ENCRYPTED, buf, encrypted_len);
|
||||
writeHardwareFrame(HW_RESP_ENCRYPTED, buf, encrypted_len);
|
||||
} else {
|
||||
writeErrorFrame(ERR_ENCRYPT_FAILED);
|
||||
writeHardwareError(HW_ERR_ENCRYPT_FAILED);
|
||||
}
|
||||
}
|
||||
|
||||
void KissModem::handleDecryptData(const uint8_t* data, uint16_t len) {
|
||||
if (len < PUB_KEY_SIZE + CIPHER_MAC_SIZE + 1) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
const uint8_t* key = data;
|
||||
const uint8_t* ciphertext = data + PUB_KEY_SIZE;
|
||||
uint16_t ciphertext_len = len - PUB_KEY_SIZE;
|
||||
|
||||
|
||||
uint8_t buf[KISS_MAX_FRAME_SIZE];
|
||||
int decrypted_len = mesh::Utils::MACThenDecrypt(key, buf, ciphertext, ciphertext_len);
|
||||
|
||||
|
||||
if (decrypted_len > 0) {
|
||||
writeFrame(RESP_DECRYPTED, buf, decrypted_len);
|
||||
writeHardwareFrame(HW_RESP_DECRYPTED, buf, decrypted_len);
|
||||
} else {
|
||||
writeErrorFrame(ERR_MAC_FAILED);
|
||||
writeHardwareError(HW_ERR_MAC_FAILED);
|
||||
}
|
||||
}
|
||||
|
||||
void KissModem::handleKeyExchange(const uint8_t* data, uint16_t len) {
|
||||
if (len < PUB_KEY_SIZE) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
uint8_t shared_secret[PUB_KEY_SIZE];
|
||||
_identity.calcSharedSecret(shared_secret, data);
|
||||
writeFrame(RESP_SHARED_SECRET, shared_secret, PUB_KEY_SIZE);
|
||||
writeHardwareFrame(HW_RESP_SHARED_SECRET, shared_secret, PUB_KEY_SIZE);
|
||||
}
|
||||
|
||||
void KissModem::handleHash(const uint8_t* data, uint16_t len) {
|
||||
if (len < 1) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
uint8_t hash[32];
|
||||
mesh::Utils::sha256(hash, 32, data, len);
|
||||
writeFrame(RESP_HASH, hash, 32);
|
||||
}
|
||||
|
||||
bool KissModem::getPacketToSend(uint8_t* packet, uint16_t* len) {
|
||||
if (!_has_pending_tx) return false;
|
||||
|
||||
memcpy(packet, _pending_tx, _pending_tx_len);
|
||||
*len = _pending_tx_len;
|
||||
_has_pending_tx = false;
|
||||
return true;
|
||||
}
|
||||
|
||||
void KissModem::onPacketReceived(int8_t snr, int8_t rssi, const uint8_t* packet, uint16_t len) {
|
||||
uint8_t buf[2 + KISS_MAX_PACKET_SIZE];
|
||||
buf[0] = (uint8_t)snr;
|
||||
buf[1] = (uint8_t)rssi;
|
||||
memcpy(&buf[2], packet, len);
|
||||
writeFrame(CMD_DATA, buf, 2 + len);
|
||||
writeHardwareFrame(HW_RESP_HASH, hash, 32);
|
||||
}
|
||||
|
||||
void KissModem::handleSetRadio(const uint8_t* data, uint16_t len) {
|
||||
if (len < 10) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
if (!_setRadioCallback) {
|
||||
writeErrorFrame(ERR_NO_CALLBACK);
|
||||
writeHardwareError(HW_ERR_NO_CALLBACK);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
uint32_t freq_hz, bw_hz;
|
||||
memcpy(&freq_hz, data, 4);
|
||||
memcpy(&bw_hz, data + 4, 4);
|
||||
uint8_t sf = data[8];
|
||||
uint8_t cr = data[9];
|
||||
|
||||
|
||||
_config.freq_hz = freq_hz;
|
||||
_config.bw_hz = bw_hz;
|
||||
_config.sf = sf;
|
||||
_config.cr = cr;
|
||||
|
||||
|
||||
float freq = freq_hz / 1000000.0f;
|
||||
float bw = bw_hz / 1000.0f;
|
||||
|
||||
|
||||
_setRadioCallback(freq, bw, sf, cr);
|
||||
writeFrame(RESP_OK, nullptr, 0);
|
||||
writeHardwareFrame(HW_RESP_OK, nullptr, 0);
|
||||
}
|
||||
|
||||
void KissModem::handleSetTxPower(const uint8_t* data, uint16_t len) {
|
||||
if (len < 1) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
if (!_setTxPowerCallback) {
|
||||
writeErrorFrame(ERR_NO_CALLBACK);
|
||||
writeHardwareError(HW_ERR_NO_CALLBACK);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
_config.tx_power = data[0];
|
||||
_setTxPowerCallback(data[0]);
|
||||
writeFrame(RESP_OK, nullptr, 0);
|
||||
writeHardwareFrame(HW_RESP_OK, nullptr, 0);
|
||||
}
|
||||
|
||||
void KissModem::handleGetRadio() {
|
||||
|
|
@ -346,92 +452,87 @@ void KissModem::handleGetRadio() {
|
|||
memcpy(buf + 4, &_config.bw_hz, 4);
|
||||
buf[8] = _config.sf;
|
||||
buf[9] = _config.cr;
|
||||
writeFrame(RESP_RADIO, buf, 10);
|
||||
writeHardwareFrame(HW_RESP_RADIO, buf, 10);
|
||||
}
|
||||
|
||||
void KissModem::handleGetTxPower() {
|
||||
writeFrame(RESP_TX_POWER, &_config.tx_power, 1);
|
||||
writeHardwareFrame(HW_RESP_TX_POWER, &_config.tx_power, 1);
|
||||
}
|
||||
|
||||
void KissModem::handleGetVersion() {
|
||||
uint8_t buf[2];
|
||||
buf[0] = KISS_FIRMWARE_VERSION;
|
||||
buf[1] = 0;
|
||||
writeFrame(RESP_VERSION, buf, 2);
|
||||
}
|
||||
|
||||
void KissModem::onTxComplete(bool success) {
|
||||
uint8_t result = success ? 0x01 : 0x00;
|
||||
writeFrame(RESP_TX_DONE, &result, 1);
|
||||
writeHardwareFrame(HW_RESP_VERSION, buf, 2);
|
||||
}
|
||||
|
||||
void KissModem::handleGetCurrentRssi() {
|
||||
if (!_getCurrentRssiCallback) {
|
||||
writeErrorFrame(ERR_NO_CALLBACK);
|
||||
writeHardwareError(HW_ERR_NO_CALLBACK);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
float rssi = _getCurrentRssiCallback();
|
||||
int8_t rssi_byte = (int8_t)rssi;
|
||||
writeFrame(RESP_CURRENT_RSSI, (uint8_t*)&rssi_byte, 1);
|
||||
writeHardwareFrame(HW_RESP_CURRENT_RSSI, (uint8_t*)&rssi_byte, 1);
|
||||
}
|
||||
|
||||
void KissModem::handleIsChannelBusy() {
|
||||
uint8_t busy = _radio.isReceiving() ? 0x01 : 0x00;
|
||||
writeFrame(RESP_CHANNEL_BUSY, &busy, 1);
|
||||
writeHardwareFrame(HW_RESP_CHANNEL_BUSY, &busy, 1);
|
||||
}
|
||||
|
||||
void KissModem::handleGetAirtime(const uint8_t* data, uint16_t len) {
|
||||
if (len < 1) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
uint8_t packet_len = data[0];
|
||||
uint32_t airtime = _radio.getEstAirtimeFor(packet_len);
|
||||
writeFrame(RESP_AIRTIME, (uint8_t*)&airtime, 4);
|
||||
writeHardwareFrame(HW_RESP_AIRTIME, (uint8_t*)&airtime, 4);
|
||||
}
|
||||
|
||||
void KissModem::handleGetNoiseFloor() {
|
||||
int16_t noise_floor = _radio.getNoiseFloor();
|
||||
writeFrame(RESP_NOISE_FLOOR, (uint8_t*)&noise_floor, 2);
|
||||
writeHardwareFrame(HW_RESP_NOISE_FLOOR, (uint8_t*)&noise_floor, 2);
|
||||
}
|
||||
|
||||
void KissModem::handleGetStats() {
|
||||
if (!_getStatsCallback) {
|
||||
writeErrorFrame(ERR_NO_CALLBACK);
|
||||
writeHardwareError(HW_ERR_NO_CALLBACK);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
uint32_t rx, tx, errors;
|
||||
_getStatsCallback(&rx, &tx, &errors);
|
||||
uint8_t buf[12];
|
||||
memcpy(buf, &rx, 4);
|
||||
memcpy(buf + 4, &tx, 4);
|
||||
memcpy(buf + 8, &errors, 4);
|
||||
writeFrame(RESP_STATS, buf, 12);
|
||||
writeHardwareFrame(HW_RESP_STATS, buf, 12);
|
||||
}
|
||||
|
||||
void KissModem::handleGetBattery() {
|
||||
uint16_t mv = _board.getBattMilliVolts();
|
||||
writeFrame(RESP_BATTERY, (uint8_t*)&mv, 2);
|
||||
writeHardwareFrame(HW_RESP_BATTERY, (uint8_t*)&mv, 2);
|
||||
}
|
||||
|
||||
void KissModem::handlePing() {
|
||||
writeFrame(RESP_PONG, nullptr, 0);
|
||||
writeHardwareFrame(HW_RESP_PONG, nullptr, 0);
|
||||
}
|
||||
|
||||
void KissModem::handleGetSensors(const uint8_t* data, uint16_t len) {
|
||||
if (len < 1) {
|
||||
writeErrorFrame(ERR_INVALID_LENGTH);
|
||||
writeHardwareError(HW_ERR_INVALID_LENGTH);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
uint8_t permissions = data[0];
|
||||
CayenneLPP telemetry(255);
|
||||
if (_sensors.querySensors(permissions, telemetry)) {
|
||||
writeFrame(RESP_SENSORS, telemetry.getBuffer(), telemetry.getSize());
|
||||
writeHardwareFrame(HW_RESP_SENSORS, telemetry.getBuffer(), telemetry.getSize());
|
||||
} else {
|
||||
writeFrame(RESP_SENSORS, nullptr, 0);
|
||||
writeHardwareFrame(HW_RESP_SENSORS, nullptr, 0);
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -11,62 +11,74 @@
|
|||
#define KISS_TFEND 0xDC
|
||||
#define KISS_TFESC 0xDD
|
||||
|
||||
#define KISS_MAX_FRAME_SIZE 512
|
||||
#define KISS_MAX_FRAME_SIZE 512
|
||||
#define KISS_MAX_PACKET_SIZE 255
|
||||
|
||||
#define CMD_DATA 0x00
|
||||
#define CMD_GET_IDENTITY 0x01
|
||||
#define CMD_GET_RANDOM 0x02
|
||||
#define CMD_VERIFY_SIGNATURE 0x03
|
||||
#define CMD_SIGN_DATA 0x04
|
||||
#define CMD_ENCRYPT_DATA 0x05
|
||||
#define CMD_DECRYPT_DATA 0x06
|
||||
#define CMD_KEY_EXCHANGE 0x07
|
||||
#define CMD_HASH 0x08
|
||||
#define CMD_SET_RADIO 0x09
|
||||
#define CMD_SET_TX_POWER 0x0A
|
||||
#define CMD_GET_RADIO 0x0C
|
||||
#define CMD_GET_TX_POWER 0x0D
|
||||
#define CMD_GET_VERSION 0x0F
|
||||
#define CMD_GET_CURRENT_RSSI 0x10
|
||||
#define CMD_IS_CHANNEL_BUSY 0x11
|
||||
#define CMD_GET_AIRTIME 0x12
|
||||
#define CMD_GET_NOISE_FLOOR 0x13
|
||||
#define CMD_GET_STATS 0x14
|
||||
#define CMD_GET_BATTERY 0x15
|
||||
#define CMD_PING 0x16
|
||||
#define CMD_GET_SENSORS 0x17
|
||||
#define KISS_CMD_DATA 0x00
|
||||
#define KISS_CMD_TXDELAY 0x01
|
||||
#define KISS_CMD_PERSISTENCE 0x02
|
||||
#define KISS_CMD_SLOTTIME 0x03
|
||||
#define KISS_CMD_TXTAIL 0x04
|
||||
#define KISS_CMD_FULLDUPLEX 0x05
|
||||
#define KISS_CMD_SETHARDWARE 0x06
|
||||
#define KISS_CMD_RETURN 0xFF
|
||||
|
||||
#define RESP_IDENTITY 0x21
|
||||
#define RESP_RANDOM 0x22
|
||||
#define RESP_VERIFY 0x23
|
||||
#define RESP_SIGNATURE 0x24
|
||||
#define RESP_ENCRYPTED 0x25
|
||||
#define RESP_DECRYPTED 0x26
|
||||
#define RESP_SHARED_SECRET 0x27
|
||||
#define RESP_HASH 0x28
|
||||
#define RESP_OK 0x29
|
||||
#define RESP_RADIO 0x2A
|
||||
#define RESP_TX_POWER 0x2B
|
||||
#define RESP_VERSION 0x2D
|
||||
#define RESP_ERROR 0x2E
|
||||
#define RESP_TX_DONE 0x2F
|
||||
#define RESP_CURRENT_RSSI 0x30
|
||||
#define RESP_CHANNEL_BUSY 0x31
|
||||
#define RESP_AIRTIME 0x32
|
||||
#define RESP_NOISE_FLOOR 0x33
|
||||
#define RESP_STATS 0x34
|
||||
#define RESP_BATTERY 0x35
|
||||
#define RESP_PONG 0x36
|
||||
#define RESP_SENSORS 0x37
|
||||
#define KISS_DEFAULT_TXDELAY 50
|
||||
#define KISS_DEFAULT_PERSISTENCE 63
|
||||
#define KISS_DEFAULT_SLOTTIME 10
|
||||
|
||||
#define ERR_INVALID_LENGTH 0x01
|
||||
#define ERR_INVALID_PARAM 0x02
|
||||
#define ERR_NO_CALLBACK 0x03
|
||||
#define ERR_MAC_FAILED 0x04
|
||||
#define ERR_UNKNOWN_CMD 0x05
|
||||
#define ERR_ENCRYPT_FAILED 0x06
|
||||
#define ERR_TX_PENDING 0x07
|
||||
#define HW_CMD_GET_IDENTITY 0x01
|
||||
#define HW_CMD_GET_RANDOM 0x02
|
||||
#define HW_CMD_VERIFY_SIGNATURE 0x03
|
||||
#define HW_CMD_SIGN_DATA 0x04
|
||||
#define HW_CMD_ENCRYPT_DATA 0x05
|
||||
#define HW_CMD_DECRYPT_DATA 0x06
|
||||
#define HW_CMD_KEY_EXCHANGE 0x07
|
||||
#define HW_CMD_HASH 0x08
|
||||
#define HW_CMD_SET_RADIO 0x09
|
||||
#define HW_CMD_SET_TX_POWER 0x0A
|
||||
#define HW_CMD_GET_RADIO 0x0C
|
||||
#define HW_CMD_GET_TX_POWER 0x0D
|
||||
#define HW_CMD_GET_VERSION 0x0F
|
||||
#define HW_CMD_GET_CURRENT_RSSI 0x10
|
||||
#define HW_CMD_IS_CHANNEL_BUSY 0x11
|
||||
#define HW_CMD_GET_AIRTIME 0x12
|
||||
#define HW_CMD_GET_NOISE_FLOOR 0x13
|
||||
#define HW_CMD_GET_STATS 0x14
|
||||
#define HW_CMD_GET_BATTERY 0x15
|
||||
#define HW_CMD_PING 0x16
|
||||
#define HW_CMD_GET_SENSORS 0x17
|
||||
|
||||
#define HW_RESP_IDENTITY 0x21
|
||||
#define HW_RESP_RANDOM 0x22
|
||||
#define HW_RESP_VERIFY 0x23
|
||||
#define HW_RESP_SIGNATURE 0x24
|
||||
#define HW_RESP_ENCRYPTED 0x25
|
||||
#define HW_RESP_DECRYPTED 0x26
|
||||
#define HW_RESP_SHARED_SECRET 0x27
|
||||
#define HW_RESP_HASH 0x28
|
||||
#define HW_RESP_OK 0x29
|
||||
#define HW_RESP_RADIO 0x2A
|
||||
#define HW_RESP_TX_POWER 0x2B
|
||||
#define HW_RESP_VERSION 0x2D
|
||||
#define HW_RESP_ERROR 0x2E
|
||||
#define HW_RESP_TX_DONE 0x2F
|
||||
#define HW_RESP_CURRENT_RSSI 0x30
|
||||
#define HW_RESP_CHANNEL_BUSY 0x31
|
||||
#define HW_RESP_AIRTIME 0x32
|
||||
#define HW_RESP_NOISE_FLOOR 0x33
|
||||
#define HW_RESP_STATS 0x34
|
||||
#define HW_RESP_BATTERY 0x35
|
||||
#define HW_RESP_PONG 0x36
|
||||
#define HW_RESP_SENSORS 0x37
|
||||
#define HW_RESP_RX_META 0x38
|
||||
|
||||
#define HW_ERR_INVALID_LENGTH 0x01
|
||||
#define HW_ERR_INVALID_PARAM 0x02
|
||||
#define HW_ERR_NO_CALLBACK 0x03
|
||||
#define HW_ERR_MAC_FAILED 0x04
|
||||
#define HW_ERR_UNKNOWN_CMD 0x05
|
||||
#define HW_ERR_ENCRYPT_FAILED 0x06
|
||||
|
||||
#define KISS_FIRMWARE_VERSION 1
|
||||
|
||||
|
|
@ -74,6 +86,9 @@ typedef void (*SetRadioCallback)(float freq, float bw, uint8_t sf, uint8_t cr);
|
|||
typedef void (*SetTxPowerCallback)(uint8_t power);
|
||||
typedef float (*GetCurrentRssiCallback)();
|
||||
typedef void (*GetStatsCallback)(uint32_t* rx, uint32_t* tx, uint32_t* errors);
|
||||
typedef void (*SendPacketCallback)(const uint8_t* data, uint16_t len);
|
||||
typedef bool (*IsSendCompleteCallback)();
|
||||
typedef void (*OnSendFinishedCallback)();
|
||||
|
||||
struct RadioConfig {
|
||||
uint32_t freq_hz;
|
||||
|
|
@ -83,6 +98,14 @@ struct RadioConfig {
|
|||
uint8_t tx_power;
|
||||
};
|
||||
|
||||
enum TxState {
|
||||
TX_IDLE,
|
||||
TX_WAIT_CLEAR,
|
||||
TX_SLOT_WAIT,
|
||||
TX_DELAY,
|
||||
TX_SENDING
|
||||
};
|
||||
|
||||
class KissModem {
|
||||
Stream& _serial;
|
||||
mesh::LocalIdentity& _identity;
|
||||
|
|
@ -90,28 +113,43 @@ class KissModem {
|
|||
mesh::Radio& _radio;
|
||||
mesh::MainBoard& _board;
|
||||
SensorManager& _sensors;
|
||||
|
||||
|
||||
uint8_t _rx_buf[KISS_MAX_FRAME_SIZE];
|
||||
uint16_t _rx_len;
|
||||
bool _rx_escaped;
|
||||
bool _rx_active;
|
||||
|
||||
|
||||
uint8_t _pending_tx[KISS_MAX_PACKET_SIZE];
|
||||
uint16_t _pending_tx_len;
|
||||
bool _has_pending_tx;
|
||||
|
||||
uint8_t _txdelay;
|
||||
uint8_t _persistence;
|
||||
uint8_t _slottime;
|
||||
uint8_t _txtail;
|
||||
uint8_t _fullduplex;
|
||||
|
||||
TxState _tx_state;
|
||||
uint32_t _tx_timer;
|
||||
|
||||
SetRadioCallback _setRadioCallback;
|
||||
SetTxPowerCallback _setTxPowerCallback;
|
||||
GetCurrentRssiCallback _getCurrentRssiCallback;
|
||||
GetStatsCallback _getStatsCallback;
|
||||
|
||||
SendPacketCallback _sendPacketCallback;
|
||||
IsSendCompleteCallback _isSendCompleteCallback;
|
||||
OnSendFinishedCallback _onSendFinishedCallback;
|
||||
|
||||
RadioConfig _config;
|
||||
|
||||
void writeByte(uint8_t b);
|
||||
void writeFrame(uint8_t cmd, const uint8_t* data, uint16_t len);
|
||||
void writeErrorFrame(uint8_t error_code);
|
||||
void writeFrame(uint8_t type, const uint8_t* data, uint16_t len);
|
||||
void writeHardwareFrame(uint8_t sub_cmd, const uint8_t* data, uint16_t len);
|
||||
void writeHardwareError(uint8_t error_code);
|
||||
void processFrame();
|
||||
|
||||
void handleHardwareCommand(uint8_t sub_cmd, const uint8_t* data, uint16_t len);
|
||||
void processTx();
|
||||
|
||||
void handleGetIdentity();
|
||||
void handleGetRandom(const uint8_t* data, uint16_t len);
|
||||
void handleVerifySignature(const uint8_t* data, uint16_t len);
|
||||
|
|
@ -137,16 +175,18 @@ class KissModem {
|
|||
public:
|
||||
KissModem(Stream& serial, mesh::LocalIdentity& identity, mesh::RNG& rng,
|
||||
mesh::Radio& radio, mesh::MainBoard& board, SensorManager& sensors);
|
||||
|
||||
|
||||
void begin();
|
||||
void loop();
|
||||
|
||||
|
||||
void setRadioCallback(SetRadioCallback cb) { _setRadioCallback = cb; }
|
||||
void setTxPowerCallback(SetTxPowerCallback cb) { _setTxPowerCallback = cb; }
|
||||
void setGetCurrentRssiCallback(GetCurrentRssiCallback cb) { _getCurrentRssiCallback = cb; }
|
||||
void setGetStatsCallback(GetStatsCallback cb) { _getStatsCallback = cb; }
|
||||
|
||||
bool getPacketToSend(uint8_t* packet, uint16_t* len);
|
||||
void setSendPacketCallback(SendPacketCallback cb) { _sendPacketCallback = cb; }
|
||||
void setIsSendCompleteCallback(IsSendCompleteCallback cb) { _isSendCompleteCallback = cb; }
|
||||
void setOnSendFinishedCallback(OnSendFinishedCallback cb) { _onSendFinishedCallback = cb; }
|
||||
|
||||
void onPacketReceived(int8_t snr, int8_t rssi, const uint8_t* packet, uint16_t len);
|
||||
void onTxComplete(bool success);
|
||||
bool isTxBusy() const { return _tx_state != TX_IDLE; }
|
||||
};
|
||||
|
|
|
|||
|
|
@ -12,14 +12,9 @@
|
|||
#include <SPIFFS.h>
|
||||
#endif
|
||||
|
||||
#define NOISE_FLOOR_CALIB_INTERVAL_MS 2000
|
||||
#define AGC_RESET_INTERVAL_MS 30000
|
||||
|
||||
StdRNG rng;
|
||||
mesh::LocalIdentity identity;
|
||||
KissModem* modem;
|
||||
static uint32_t next_noise_floor_calib_ms = 0;
|
||||
static uint32_t next_agc_reset_ms = 0;
|
||||
|
||||
void halt() {
|
||||
while (1) ;
|
||||
|
|
@ -67,6 +62,18 @@ void onGetStats(uint32_t* rx, uint32_t* tx, uint32_t* errors) {
|
|||
*errors = radio_driver.getPacketsRecvErrors();
|
||||
}
|
||||
|
||||
void onSendPacket(const uint8_t* data, uint16_t len) {
|
||||
radio_driver.startSendRaw(data, len);
|
||||
}
|
||||
|
||||
bool onIsSendComplete() {
|
||||
return radio_driver.isSendComplete();
|
||||
}
|
||||
|
||||
void onSendFinished() {
|
||||
radio_driver.onSendFinished();
|
||||
}
|
||||
|
||||
void setup() {
|
||||
board.begin();
|
||||
|
||||
|
|
@ -91,40 +98,25 @@ void setup() {
|
|||
modem->setTxPowerCallback(onSetTxPower);
|
||||
modem->setGetCurrentRssiCallback(onGetCurrentRssi);
|
||||
modem->setGetStatsCallback(onGetStats);
|
||||
modem->setSendPacketCallback(onSendPacket);
|
||||
modem->setIsSendCompleteCallback(onIsSendComplete);
|
||||
modem->setOnSendFinishedCallback(onSendFinished);
|
||||
modem->begin();
|
||||
}
|
||||
|
||||
void loop() {
|
||||
modem->loop();
|
||||
|
||||
uint8_t packet[KISS_MAX_PACKET_SIZE];
|
||||
uint16_t len;
|
||||
if (!modem->isTxBusy()) {
|
||||
uint8_t rx_buf[256];
|
||||
int rx_len = radio_driver.recvRaw(rx_buf, sizeof(rx_buf));
|
||||
|
||||
// trigger noise floor calibration
|
||||
if ((uint32_t)(millis() - next_noise_floor_calib_ms) >= NOISE_FLOOR_CALIB_INTERVAL_MS) {
|
||||
radio_driver.triggerNoiseFloorCalibrate(0);
|
||||
next_noise_floor_calib_ms = millis();
|
||||
}
|
||||
radio_driver.loop();
|
||||
|
||||
if (modem->getPacketToSend(packet, &len)) {
|
||||
radio_driver.startSendRaw(packet, len);
|
||||
while (!radio_driver.isSendComplete()) {
|
||||
delay(1);
|
||||
if (rx_len > 0) {
|
||||
int8_t snr = (int8_t)(radio_driver.getLastSNR() * 4);
|
||||
int8_t rssi = (int8_t)radio_driver.getLastRSSI();
|
||||
modem->onPacketReceived(snr, rssi, rx_buf, rx_len);
|
||||
}
|
||||
radio_driver.onSendFinished();
|
||||
modem->onTxComplete(true);
|
||||
}
|
||||
|
||||
if ((uint32_t)(millis() - next_agc_reset_ms) >= AGC_RESET_INTERVAL_MS) {
|
||||
radio_driver.resetAGC();
|
||||
next_agc_reset_ms = millis();
|
||||
}
|
||||
uint8_t rx_buf[256];
|
||||
int rx_len = radio_driver.recvRaw(rx_buf, sizeof(rx_buf));
|
||||
if (rx_len > 0) {
|
||||
int8_t snr = (int8_t)(radio_driver.getLastSNR() * 4);
|
||||
int8_t rssi = (int8_t)radio_driver.getLastRSSI();
|
||||
modem->onPacketReceived(snr, rssi, rx_buf, rx_len);
|
||||
}
|
||||
radio_driver.loop();
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue