119 lines
3.5 KiB
C
119 lines
3.5 KiB
C
#include "modbus.h"
|
||
#include "lwip/tcp.h"
|
||
#include "esp_log.h"
|
||
|
||
static const char *TAG = "modbus tcp";
|
||
|
||
/* 修正：应该将整个包的解析都放到Modbus的处理中，这里有一点纠缠了 */
|
||
static err_t ModBusSlave_recv(void *arg, struct tcp_pcb *newpcb, struct pbuf *p, err_t err) {
|
||
// ESP_LOGI(TAG, "ModBusSlave_recv");
|
||
|
||
/* 处理接收到的包信息（存储在pbuf中），并发送包回应 */
|
||
if (p != NULL) {
|
||
uint8_t *txbuf;
|
||
uint8_t *rxbuf;
|
||
int txlen, rxlen;
|
||
|
||
/* 接收数据 */
|
||
tcp_recved(newpcb, p->tot_len);
|
||
rxbuf = (uint8_t *)p->payload; /* 修正：pbuf的数据是用链表存储的，p->payload仅是该节点的数据，并不是所有的，但是目前只取了第一个节点 */
|
||
rxlen = p->len;
|
||
|
||
/* MBAP报文头7Byte：事务处理标识2Byte + 协议标识2Byte + 长度2Byte(此字节往后的字节个数) + 单元标识符1Byte*/
|
||
/* 这里忽略了前面的6Byte未处理 */
|
||
if (rxlen < 6) {
|
||
ESP_LOGI(TAG, "err: rxlen less 6");
|
||
pbuf_free(p);
|
||
return tcp_close(newpcb); /* 修正：tcp_close的处理是否合理？ */
|
||
}
|
||
|
||
txbuf = (uint8_t *)mem_malloc(1024 + 16);
|
||
if (txbuf == NULL) {
|
||
ESP_LOGI(TAG, "err:tx malloc fault");
|
||
return tcp_close(newpcb);
|
||
}
|
||
|
||
/* 使用Modbus协议解析接收数据，同时准备应答包txbuf数据，发送应答包 */
|
||
/* 包含： */
|
||
/* MBAP：单元标识符1Byte */
|
||
/* PUD（帧结构）：功能码1Byte + 地址2Byte(H+L) + ... */
|
||
txlen = ModbusSlaveProcess(&txbuf[6], &rxbuf[6], (rxlen - 6), 0);
|
||
if (txlen > 0) {
|
||
int i;
|
||
for (i = 0; i < 4; i++) {
|
||
txbuf[i] = rxbuf[i];
|
||
}
|
||
txbuf[4] = (uint8_t)(txlen >> 8);
|
||
txbuf[5] = (uint8_t)(txlen & 0xff);
|
||
tcp_write(newpcb, txbuf, txlen + 6, 1);
|
||
}
|
||
|
||
mem_free(txbuf);
|
||
pbuf_free(p);
|
||
}
|
||
else if (err == ERR_OK) {
|
||
/* 有错误，关闭tcp */
|
||
tcp_err(newpcb, NULL);
|
||
tcp_recv(newpcb, NULL);
|
||
tcp_poll(newpcb, NULL, 120);
|
||
err_t err_p = tcp_close(newpcb);
|
||
if (ERR_OK != err_p) {
|
||
// log_printf(LERROR, LOG_TAG "close error. err:%d\n", err_p);
|
||
}
|
||
return err_p;
|
||
}
|
||
return ERR_OK;
|
||
}
|
||
|
||
|
||
static void ModBusSlave_conn_err(void *arg, err_t err) {
|
||
ESP_LOGI(TAG, "connection error, err:%d\n", err);
|
||
}
|
||
|
||
|
||
/* 修正：这里的实现时关闭tcp，为什么要60s定时关闭一次tcp呢？会不会影响程序？ */
|
||
static err_t ModBusSlave_poll(void *arg, struct tcp_pcb *newpcb) {
|
||
ESP_LOGI(TAG, "ModBusSlave_poll");
|
||
|
||
if (newpcb) {
|
||
ESP_LOGI(TAG, "close pcb");
|
||
tcp_close(newpcb);
|
||
newpcb = NULL;
|
||
}
|
||
return ERR_OK;
|
||
}
|
||
|
||
|
||
/*
|
||
* 有tcp连接请求时触发该回调函数
|
||
* pcb：新分配的用于和请求者通信的控制块
|
||
*/
|
||
static err_t ModBusSlave_accept(void *arg, struct tcp_pcb *pcb, err_t err) {
|
||
ESP_LOGI(TAG, "accept assess");
|
||
|
||
/* 设置各种回调函数 */
|
||
tcp_err(pcb, ModBusSlave_conn_err);
|
||
tcp_recv(pcb, ModBusSlave_recv);
|
||
tcp_poll(pcb, ModBusSlave_poll, 120); /* 轮询函数，定时会触发 120表示60s触发一次 */
|
||
|
||
return ERR_OK;
|
||
}
|
||
|
||
|
||
void ModBusTCPSlave_init(void) {
|
||
ESP_LOGI(TAG, "ModBusTCPSlave_init");
|
||
|
||
struct tcp_pcb *pcb;
|
||
pcb = tcp_new();
|
||
|
||
/* IP_ADDR_ANY表示绑定到主机的所有IP上 502端口是专门分配给Modbus的 */
|
||
tcp_bind(pcb, IP_ADDR_ANY, 502);
|
||
|
||
/* 开启被动监听 */
|
||
pcb = tcp_listen(pcb);
|
||
|
||
/* 设置ModBusSlave_accept回调函数，在有tcp连接请求时触发 */
|
||
tcp_accept(pcb, ModBusSlave_accept);
|
||
}
|
||
|