71 lines
2.1 KiB
C++
71 lines
2.1 KiB
C++
#include "uart_mitm.h"
|
|
#include "esphome/core/log.h"
|
|
#define RX_SUPPRESS_INTERVAL 50 // Time in ms to suppress rx after tx on same bus
|
|
#define RX_BLOCK_INTERVAL 50 // Time in ms to block before preparing a message to ship.
|
|
#define RX_BUF_LEN 64 // Time in ms to block before preparing a message to ship.
|
|
|
|
namespace esphome {
|
|
namespace serial {
|
|
|
|
static const char *const TAG = "uart_mitm";
|
|
static uint32_t rx1_suppress_until = 0;
|
|
static uint32_t rx2_suppress_until = 0;
|
|
|
|
static uint32_t rx1_block_until = 0;
|
|
static uint32_t rx2_block_until = 0;
|
|
|
|
static uint8_t rx1_buffer_idx = 0;
|
|
static uint8_t rx2_buffer_idx = 0;
|
|
|
|
static uint8_t rx1_buffer[RX_BUF_LEN];
|
|
static uint8_t rx2_buffer[RX_BUF_LEN];
|
|
|
|
void UARTMITM::loop() {
|
|
uint8_t c;
|
|
while (this->uart1_->available()) {
|
|
this->uart1_->read_byte(&c);
|
|
if (rx1_block_until < millis()) { // current millis is after our blocking deadline; we can ship
|
|
// ... so ship!
|
|
ESP_LOGD(TAG, "RX1: We're gonna ship rx1!");
|
|
rx1_buffer_idx = 0;
|
|
}
|
|
|
|
rx1_buffer[rx1_buffer_idx++] = c;
|
|
if (rx1_buffer_idx == RX_BUF_LEN) {
|
|
ESP_LOGW(TAG, "RX1: Somehow ran out of buf space.");
|
|
rx1_buffer_idx = 0;
|
|
}
|
|
rx1_block_until = millis() + RX_BLOCK_INTERVAL;
|
|
|
|
if (rx1_suppress_until < millis()) {
|
|
this->uart2_->write_byte(c);
|
|
rx2_suppress_until = millis() + RX_SUPPRESS_INTERVAL;
|
|
}
|
|
}
|
|
while (this->uart2_->available()) {
|
|
this->uart2_->read_byte(&c);
|
|
if (rx2_block_until < millis()) { // current millis is after our blocking deadline; we can ship
|
|
// ... so ship!
|
|
ESP_LOGD(TAG, "RX2: We're gonna ship rx2!");
|
|
rx2_buffer_idx = 0;
|
|
}
|
|
|
|
rx2_buffer[rx2_buffer_idx++] = c;
|
|
if (rx2_buffer_idx == RX_BUF_LEN) {
|
|
ESP_LOGW(TAG, "RX2: Somehow ran out of buf space.");
|
|
rx2_buffer_idx = 0;
|
|
}
|
|
rx2_block_until = millis() + RX_BLOCK_INTERVAL;
|
|
|
|
if (rx2_suppress_until < millis()) {
|
|
this->uart1_->write_byte(c);
|
|
rx1_suppress_until = millis() + RX_SUPPRESS_INTERVAL;
|
|
}
|
|
}
|
|
}
|
|
|
|
void UARTMITM::dump_config() { ESP_LOGCONFIG(TAG, "UART MITM"); }
|
|
|
|
} // namespace serial
|
|
} // namespace esphome
|