wt/mongoose
1
0 Fork 0
mirror of https://github.com/cesanta/mongoose.git synced 2026-10-24 11:54:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add modbus-tcp support

Browse Source
This commit is contained in:
cpq authored and robert committed 2026-04-06 03:54:58 -04:00
1 parent 1bb85799ce
commit d67b30aba3
7 files changed
+885 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

test/Makefile
+1 -2
View File
@@ -251,7 +251,7 @@ mongoose.c: Makefile $(wildcard ../src/*.c) $(wildcard ../src/drivers/*.c)
cd .. && (export LC_ALL=C ; cat src/license.h; echo; echo '#include "mongoose.h"' ; (for F in src/*.c src/drivers/*.c ; do echo; echo '#ifdef MG_ENABLE_LINES'; echo "#line 1 \"$$F\""; echo '#endif'; cat $$F | sed -e 's,#include ".*,,'; done))> $@
mongoose.h: $(HDRS) Makefile
cd .. && (cat src/license.h; echo; echo '#ifndef MONGOOSE_H'; echo '#define MONGOOSE_H'; echo; cat src/version.h ; echo; echo '#ifdef __cplusplus'; echo 'extern "C" {'; echo '#endif'; cat src/arch.h src/arch_*.h src/net_ft.h src/net_lwip.h src/net_rl.h src/config.h src/profile.h src/str.h src/queue.h src/fmt.h src/printf.h src/log.h src/timer.h src/fs.h src/util.h src/url.h src/iobuf.h src/base64.h src/md5.h src/sha1.h src/sha256.h src/event.h src/net.h src/http.h src/ssi.h src/tls.h src/tls_x25519.h src/tls_aes128.h src/tls_uecc.h src/tls_chacha20.h src/tls_rsa.h src/tls_mbed.h src/tls_openssl.h src/ws.h src/sntp.h src/mqtt.h src/dns.h src/json.h src/rpc.h src/ota.h src/flash.h src/wifi.h src/l2.h src/net_builtin.h src/drivers/*.h | sed -e '/keep/! s,#include ".*,,' -e 's,^#pragma once,,'; echo; echo '#ifdef __cplusplus'; echo '}'; echo '#endif'; echo '#endif // MONGOOSE_H')> $@
cd .. && (cat src/license.h; echo; echo '#ifndef MONGOOSE_H'; echo '#define MONGOOSE_H'; echo; cat src/version.h ; echo; echo '#ifdef __cplusplus'; echo 'extern "C" {'; echo '#endif'; cat src/arch.h src/arch_*.h src/net_ft.h src/net_lwip.h src/net_rl.h src/config.h src/profile.h src/str.h src/queue.h src/fmt.h src/printf.h src/log.h src/timer.h src/fs.h src/util.h src/url.h src/iobuf.h src/base64.h src/md5.h src/sha1.h src/sha256.h src/event.h src/net.h src/http.h src/ssi.h src/tls.h src/tls_x25519.h src/tls_aes128.h src/tls_uecc.h src/tls_chacha20.h src/tls_rsa.h src/tls_mbed.h src/tls_openssl.h src/ws.h src/sntp.h src/mqtt.h src/dns.h src/modbus.h src/json.h src/rpc.h src/ota.h src/flash.h src/wifi.h src/l2.h src/net_builtin.h src/drivers/*.h | sed -e '/keep/! s,#include ".*,,' -e 's,^#pragma once,,'; echo; echo '#ifdef __cplusplus'; echo '}'; echo '#endif'; echo '#endif // MONGOOSE_H')> $@
clean: clean_tutorials clean_tutorials_embedded
@@ -285,4 +285,3 @@ tutorials_embedded:
@for X in $(TUTORIALS_EMBEDDED); do test -f $$X/Makefile || continue; $(MAKE) -C $$X build || exit 1; done
clean_tutorials_embedded:
for X in $(TUTORIALS_EMBEDDED); do test -f $$X/Makefile || continue; $(MAKE) -C $$X clean || exit 1; done
test/unit_test.c
+352 -2
View File
@@ -768,7 +768,8 @@ static void test_mqtt_ver(uint8_t mqtt_version) {
memset(mbuf + 1, 0, sizeof(mbuf) - 1);
test_data.flags = 0;
opts.props = 0; opts.num_props = 0;
opts.props = 0;
opts.num_props = 0;
mg_mqtt_unsub(c, &opts);
for (i = 0; i < 500 && test_data.flags == 0; i++) mg_mgr_poll(&mgr, 10);
ASSERT(test_data.flags == flags_unsubscribed);
@@ -1466,7 +1467,8 @@ static void test_tls(void) {
"%s",
bd.len, bd.buf) == 200);
ASSERT(cmpbody(buf, bd.buf) == 0);
#if MG_TLS == MG_TLS_BUILTIN && defined(__linux__) && MG_ENABLE_CHACHA20 // skip for non-CHACHA tests
#if MG_TLS == MG_TLS_BUILTIN && defined(__linux__) && \
MG_ENABLE_CHACHA20 // skip for non-CHACHA tests
// fire patched server, test multiple TLS records per TCP segment handling
// skip other TLS stacks to avoid "bad client hello", we are 1.3 only
if (access("tls_multirec/server", X_OK) == 0) {
@@ -4041,11 +4043,359 @@ static void test_crypto(void) {
#define DASHBOARD(x) \
printf("HEALTH_DASHBOARD\t\"%s\": %s,\n", x, s_error ? "false" : "true");
static uint8_t coils_database[25] = {
0x4D, 0x9C, 0xAA, 0x55, 0xCC,
0xA3, 0x66, 0xAF, 0x60, 0xBC,
0xCC, 0x6C, 0x53, 0xFF, 0x00,
0x55, 0x3C, 0x0F, 0xF0, 0x8F,
0x54, 0x99, 0xF8, 0x0D, 0x2A,
};
static bool modbus_read_coil(uint16_t i) {
uint16_t start_byte = i / 8;
int bit_pos = i % 8;
uint8_t data = ((coils_database[start_byte] >> bit_pos) & 0x01);
return data;
}
static void modbus_srv(struct mg_connection *c, int ev, void *ev_data) {
struct mg_modbus_req *mr = (struct mg_modbus_req *) ev_data;
uint16_t i, r;
size_t tc = *(size_t *) c->fn_data + 1;
static bool coil_1[] = {1};
static bool coil_7[] = {1, 0, 1, 0, 1, 0, 1};
static uint16_t reg_1[] = {0x0001};
static uint16_t reg_7[] = {0x0001, 0x0002, 0x0003, 0x0004,
0x0005, 0x0006, 0x0007};
if (ev == MG_EV_READ) {
*(size_t *) c->fn_data += 1;
} else if (ev == MG_EV_MODBUS_REQ) {
if (mr->addr == 0xFFFF) {
mr->error = MG_MODBUS_ERR_ILLEGAL_ADDRESS;
return;
}
switch (mr->func) {
case MG_MODBUS_FUNC_READ_COILS:
case MG_MODBUS_FUNC_READ_DISCRETE_INPUTS:
if (tc != 0x1D) {
for (i = 0; i < mr->len; i++) {
mr->u.bits[i] = (mr->len == 1 ? coil_1[i] : coil_7[i]);
}
} else {
for (i = 0; i < mr->len; i++) {
mr->u.bits[i] = modbus_read_coil(i);
}
}
break;
case MG_MODBUS_FUNC_READ_HOLDING_REGISTERS:
case MG_MODBUS_FUNC_READ_INPUT_REGISTERS:
if (tc == 0x1c) {
mr->error = MG_MODBUS_ERR_DEVICE_FAILURE;
break;
}
if (mr->len == 1) {
mr->u.regs[0] = reg_1[0];
} else if (mr->len == 7) {
for (i = 0; i < 7; i++) mr->u.regs[i] = reg_7[i];
} else {
mr->error = MG_MODBUS_ERR_ILLEGAL_ADDRESS;
}
break;
case MG_MODBUS_FUNC_WRITE_SINGLE_COIL:
switch (tc) {
case 0x05:
ASSERT(mr->len == 1);
ASSERT(mr->addr == 0x14);
ASSERT(mr->u.bits != NULL);
ASSERT(mr->u.bits[0] == true);
break;
default:
ASSERT(false);
return;
}
break;
case MG_MODBUS_FUNC_WRITE_SINGLE_REGISTER:
switch (tc) {
case 0x06:
ASSERT(mr->len == 1);
ASSERT(mr->addr == 0x15);
ASSERT(mr->u.regs != NULL);
ASSERT(mr->u.regs[0] == 1);
break;
default:
ASSERT(false);
return;
}
break;
case MG_MODBUS_FUNC_WRITE_MULTIPLE_COILS:
switch (tc) {
case 0x14:
ASSERT(mr->addr == 0x23);
ASSERT(mr->len == 1);
ASSERT(mr->u.bits != NULL);
ASSERT(mr->u.bits[0] == true);
break;
case 0x15:
ASSERT(mr->addr == 0x24);
ASSERT(mr->len == 7);
ASSERT(mr->u.bits != NULL);
for (i = 0; i < 7; i++) {
ASSERT(mr->u.bits[i] == coil_7[i]);
}
break;
default:
ASSERT(false);
return;
}
break;
case MG_MODBUS_FUNC_WRITE_MULTIPLE_REGISTERS:
switch (tc) {
case 0x17:
r = 0;
ASSERT(mr->addr == 0x26);
ASSERT(mr->len == 1);
ASSERT(mr->u.regs != NULL);
r = mr->u.regs[0];
ASSERT(r == 1);
break;
case 0x18:
ASSERT(mr->addr == 0x27);
ASSERT(mr->len == 7);
ASSERT(mr->u.regs != NULL);
for (i = 0; i < 7; i++) {
r = mr->u.regs[i];
ASSERT(r == reg_7[i]);
}
break;
default:
ASSERT(false);
return;
}
break;
default:
mr->error = MG_MODBUS_ERR_ILLEGAL_FUNCTION;
break;
}
}
(void) c, (void) ev_data;
}
static void test_modbus(void) {
struct modbus_test {
const char *name;
uint8_t req[256];
size_t req_len;
uint8_t resp[256];
size_t resp_len;
} tests[] = {
{"read coils - 1 @ 0x0010",
{0x00, 0x01, 0x00, 0x00, 0x00, 0x06, 0x01, 0x01, 0x00, 0x10, 0x00, 0x01},
12,
{0x00, 0x01, 0x00, 0x00, 0x00, 0x04, 0x01, 0x01, 0x01, 0x01},
10},
{"read discrete inputs - 1 @ 0x0011",
{0x00, 0x02, 0x00, 0x00, 0x00, 0x06, 0x01, 0x02, 0x00, 0x11, 0x00, 0x01},
12,
{0x00, 0x02, 0x00, 0x00, 0x00, 0x04, 0x01, 0x02, 0x01, 0x01},
10},
{"read holding registers - 1 @ 0x0012",
{0x00, 0x03, 0x00, 0x00, 0x00, 0x06, 0x01, 0x03, 0x00, 0x12, 0x00, 0x01},
12,
{0x00, 0x03, 0x00, 0x00, 0x00, 0x05, 0x01, 0x03, 0x02, 0x00, 0x01},
11},
{"read input registers - 1 @ 0x0013",
{0x00, 0x04, 0x00, 0x00, 0x00, 0x06, 0x01, 0x04, 0x00, 0x13, 0x00, 0x01},
12,
{0x00, 0x04, 0x00, 0x00, 0x00, 0x05, 0x01, 0x04, 0x02, 0x00, 0x01},
11},
{"write single coil @ 0x0014 = 1",
{0x00, 0x05, 0x00, 0x00, 0x00, 0x06, 0x01, 0x05, 0x00, 0x14, 0xFF, 0x00},
12,
{0x00, 0x05, 0x00, 0x00, 0x00, 0x06, 0x01, 0x05, 0x00, 0x14, 0xFF, 0x00},
12},
{"write single register @ 0x0015 = 1",
{0x00, 0x06, 0x00, 0x00, 0x00, 0x06, 0x01, 0x06, 0x00, 0x15, 0x00, 0x01},
12,
{0x00, 0x06, 0x00, 0x00, 0x00, 0x06, 0x01, 0x06, 0x00, 0x15, 0x00, 0x01},
12},
/* --- Multi-value reads: 0, 1, 7 --- */
{"read coils - 0 @ 0x0016",
{0x00, 0x07, 0x00, 0x00, 0x00, 0x06, 0x01, 0x01, 0x00, 0x16, 0x00, 0x00},
12,
{0x00, 0x07, 0x00, 0x00, 0x00, 0x03, 0x01, 0x81, 0x03},
9},
{"read coils - 1 @ 0x0017",
{0x00, 0x08, 0x00, 0x00, 0x00, 0x06, 0x01, 0x01, 0x00, 0x17, 0x00, 0x01},
12,
{0x00, 0x08, 0x00, 0x00, 0x00, 0x04, 0x01, 0x01, 0x01, 0x01},
10},
{"read coils - 7 @ 0x0018",
{0x00, 0x09, 0x00, 0x00, 0x00, 0x06, 0x01, 0x01, 0x00, 0x18, 0x00, 0x07},
12,
{0x00, 0x09, 0x00, 0x00, 0x00, 0x04, 0x01, 0x01, 0x01, 0x55},
10},
{"read discrete inputs - 0 @ 0x0019",
{0x00, 0x0A, 0x00, 0x00, 0x00, 0x06, 0x01, 0x02, 0x00, 0x19, 0x00, 0x00},
12,
{0x00, 0x0A, 0x00, 0x00, 0x00, 0x03, 0x01, 0x82, 0x03},
9},
{"read discrete inputs - 1 @ 0x001A",
{0x00, 0x0B, 0x00, 0x00, 0x00, 0x06, 0x01, 0x02, 0x00, 0x1A, 0x00, 0x01},
12,
{0x00, 0x0B, 0x00, 0x00, 0x00, 0x04, 0x01, 0x02, 0x01, 0x01},
10},
{"read discrete inputs - 7 @ 0x001B",
{0x00, 0x0C, 0x00, 0x00, 0x00, 0x06, 0x01, 0x02, 0x00, 0x1B, 0x00, 0x07},
12,
{0x00, 0x0C, 0x00, 0x00, 0x00, 0x04, 0x01, 0x02, 0x01, 0x55},
10},
{"read holding registers - 0 @ 0x001C",
{0x00, 0x0D, 0x00, 0x00, 0x00, 0x06, 0x01, 0x03, 0x00, 0x1C, 0x00, 0x00},
12,
{0x00, 0x0D, 0x00, 0x00, 0x00, 0x03, 0x01, 0x83, 0x03},
9},
{"read holding registers - 1 @ 0x001D",
{0x00, 0x0E, 0x00, 0x00, 0x00, 0x06, 0x01, 0x03, 0x00, 0x1D, 0x00, 0x01},
12,
{0x00, 0x0E, 0x00, 0x00, 0x00, 0x05, 0x01, 0x03, 0x02, 0x00, 0x01},
11},
{"read holding registers - 7 @ 0x001E",
{0x00, 0x0F, 0x00, 0x00, 0x00, 0x06, 0x01, 0x03, 0x00, 0x1E, 0x00, 0x07},
12,
{0x00, 0x0F, 0x00, 0x00, 0x00, 0x11, 0x01, 0x03, 0x0E, 0x00, 0x01, 0x00,
0x02, 0x00, 0x03, 0x00, 0x04, 0x00, 0x05, 0x00, 0x06, 0x00, 0x07},
23},
{"read input registers - 0 @ 0x001F",
{0x00, 0x10, 0x00, 0x00, 0x00, 0x06, 0x01, 0x04, 0x00, 0x1F, 0x00, 0x00},
12,
{0x00, 0x10, 0x00, 0x00, 0x00, 0x03, 0x01, 0x84, 0x03},
9},
{"read input registers - 1 @ 0x0020",
{0x00, 0x11, 0x00, 0x00, 0x00, 0x06, 0x01, 0x04, 0x00, 0x20, 0x00, 0x01},
12,
{0x00, 0x11, 0x00, 0x00, 0x00, 0x05, 0x01, 0x04, 0x02, 0x00, 0x01},
11},
{"read input registers - 7 @ 0x0021",
{0x00, 0x12, 0x00, 0x00, 0x00, 0x06, 0x01, 0x04, 0x00, 0x21, 0x00, 0x07},
12,
{0x00, 0x12, 0x00, 0x00, 0x00, 0x11, 0x01, 0x04, 0x0E, 0x00, 0x01, 0x00,
0x02, 0x00, 0x03, 0x00, 0x04, 0x00, 0x05, 0x00, 0x06, 0x00, 0x07},
23},
{"write multiple coils - 0 @ 0x0022",
{0x00, 0x13, 0x00, 0x00, 0x00, 0x07, 0x01, 0x0F, 0x00, 0x22, 0x00, 0x00,
0x00},
13,
{0x00, 0x13, 0x00, 0x00, 0x00, 0x03, 0x01, 0x8F, 0x03},
9},
{"write multiple coils - 1 @ 0x0023",
{0x00, 0x14, 0x00, 0x00, 0x00, 0x08, 0x01, 0x0F, 0x00, 0x23, 0x00, 0x01,
0x01, 0x01},
14,
{0x00, 0x14, 0x00, 0x00, 0x00, 0x06, 0x01, 0x0F, 0x00, 0x23, 0x00, 0x01},
12},
{"write multiple coils - 7 @ 0x0024",
{0x00, 0x15, 0x00, 0x00, 0x00, 0x08, 0x01, 0x0F, 0x00, 0x24, 0x00, 0x07,
0x01, 0x55},
14,
{0x00, 0x15, 0x00, 0x00, 0x00, 0x06, 0x01, 0x0F, 0x00, 0x24, 0x00, 0x07},
12},
{"write multiple registers - 0 @ 0x0025",
{0x00, 0x16, 0x00, 0x00, 0x00, 0x07, 0x01, 0x10, 0x00, 0x25, 0x00, 0x00,
0x00},
13,
{0x00, 0x16, 0x00, 0x00, 0x00, 0x03, 0x01, 0x90, 0x03},
9},
{"write multiple registers - 1 @ 0x0026",
{0x00, 0x17, 0x00, 0x00, 0x00, 0x09, 0x01, 0x10, 0x00, 0x26, 0x00, 0x01,
0x02, 0x00, 0x01},
15,
{0x00, 0x17, 0x00, 0x00, 0x00, 0x06, 0x01, 0x10, 0x00, 0x26, 0x00, 0x01},
12},
{"write multiple registers - 7 @ 0x0027",
{0x00, 0x18, 0x00, 0x00, 0x00, 0x15, 0x01, 0x10, 0x00,
0x27, 0x00, 0x07, 0x0E, 0x00, 0x01, 0x00, 0x02, 0x00,
0x03, 0x00, 0x04, 0x00, 0x05, 0x00, 0x06, 0x00, 0x07},
27,
{0x00, 0x18, 0x00, 0x00, 0x00, 0x06, 0x01, 0x10, 0x00, 0x27, 0x00, 0x07},
12},
/* --- Standard Modbus exception cases --- */
{"illegal function",
{0x00, 0x19, 0x00, 0x00, 0x00, 0x06, 0x01, 0x11, 0x00, 0x28, 0x00, 0x01},
12,
{0x00, 0x19, 0x00, 0x00, 0x00, 0x03, 0x01, 0x91, 0x01},
9},
{"illegal data address",
{0x00, 0x1A, 0x00, 0x00, 0x00, 0x06, 0x01, 0x03, 0xFF, 0xFF, 0x00, 0x01},
12,
{0x00, 0x1A, 0x00, 0x00, 0x00, 0x03, 0x01, 0x83, 0x02},
9},
{"illegal data value - bad coil value",
{0x00, 0x1B, 0x00, 0x00, 0x00, 0x06, 0x01, 0x05, 0x00, 0x29, 0x12, 0x34},
12,
{0x00, 0x1B, 0x00, 0x00, 0x00, 0x03, 0x01, 0x85, 0x03},
9},
{"server device failure",
{0x00, 0x1C, 0x00, 0x00, 0x00, 0x06, 0x01, 0x04, 0x00, 0x2A, 0x00, 0x01},
12,
{0x00, 0x1C, 0x00, 0x00, 0x00, 0x03, 0x01, 0x84, 0x04},
9},
/* --- Stress Test ---*/
{"read coils - 200 @ 0x0000",
{0x00, 0x1D, 0x00, 0x00, 0x00, 0x06, 0x01, 0x01, 0x00, 0x00, 0x00, 0xC8},
12,
{0x00, 0x1D, 0x00, 0x00, 0x00, 0x1C, 0x01, 0x01, 0x19,
0x4D, 0x9C, 0xAA, 0x55, 0xCC,
0xA3, 0x66, 0xAF, 0x60, 0xBC,
0xCC, 0x6C, 0x53, 0xFF, 0x00,
0x55, 0x3C, 0x0F, 0xF0, 0x8F,
0x54, 0x99, 0xF8, 0x0D, 0x2A},
34},
};
struct mg_mgr mgr;
struct mg_connection *c;
const char *url = "tcp://localhost:32625";
size_t i, j, total = 0, num_tests = sizeof(tests) / sizeof(tests[0]);
mg_mgr_init(&mgr);
mg_modbus_listen(&mgr, url, modbus_srv, &total);
c = mg_connect(&mgr, url, NULL, NULL);
for (i = 0; i < num_tests; i++) {
MG_INFO(("TEST %s", tests[i].name));
mg_send(c, tests[i].req, tests[i].req_len);
for (j = 0; j < 20; j++) mg_mgr_poll(&mgr, 1);
mg_hexdump(tests[i].req, tests[i].req_len);
mg_hexdump(tests[i].resp, tests[i].resp_len);
mg_hexdump(c->recv.buf, c->recv.len);
ASSERT(c->recv.len == tests[i].resp_len);
ASSERT(memcmp(c->recv.buf, tests[i].resp, c->recv.len) == 0);
c->recv.len = 0;
}
ASSERT(total == num_tests);
mg_mgr_free(&mgr);
}
int main(void) {
const char *debug_level = getenv("V");
if (debug_level == NULL) debug_level = "3";
mg_log_set(atoi(debug_level));
s_error = false;
test_modbus();
DASHBOARD("modbus");
s_error = false;
test_crypto();
DASHBOARD("crypto");