]>
Commit | Line | Data |
---|---|---|
4c938fc2 AJ |
1 | /* |
2 | * This file is part of the libsigrok project. | |
3 | * | |
4 | * Copyright (C) 2015 Aurelien Jacobs <aurel@gnuage.org> | |
5 | * | |
6 | * This program is free software: you can redistribute it and/or modify | |
7 | * it under the terms of the GNU General Public License as published by | |
8 | * the Free Software Foundation, either version 3 of the License, or | |
9 | * (at your option) any later version. | |
10 | * | |
11 | * This program is distributed in the hope that it will be useful, | |
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
14 | * GNU General Public License for more details. | |
15 | * | |
16 | * You should have received a copy of the GNU General Public License | |
17 | * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
18 | */ | |
19 | ||
20 | #include "libsigrok.h" | |
21 | #include "libsigrok-internal.h" | |
22 | ||
23 | #include <glib.h> | |
24 | #include <stdlib.h> | |
25 | #include <string.h> | |
26 | ||
27 | #define LOG_PREFIX "modbus_serial" | |
28 | ||
29 | #define BUFFER_SIZE 1024 | |
30 | ||
31 | struct modbus_serial_rtu { | |
32 | struct sr_serial_dev_inst *serial; | |
33 | uint8_t slave_addr; | |
34 | uint16_t crc; | |
35 | }; | |
36 | ||
37 | static int modbus_serial_rtu_dev_inst_new(void *priv, const char *resource, | |
38 | char **params, const char *serialcomm, int modbusaddr) | |
39 | { | |
40 | struct modbus_serial_rtu *modbus = priv; | |
41 | ||
42 | (void) params; | |
43 | ||
44 | modbus->serial = sr_serial_dev_inst_new(resource, serialcomm); | |
45 | modbus->slave_addr = modbusaddr; | |
46 | ||
47 | return SR_OK; | |
48 | } | |
49 | ||
50 | static int modbus_serial_rtu_open(void *priv) | |
51 | { | |
52 | struct modbus_serial_rtu *modbus = priv; | |
53 | struct sr_serial_dev_inst *serial = modbus->serial; | |
54 | ||
55 | if (serial_open(serial, SERIAL_RDWR) != SR_OK) | |
56 | return SR_ERR; | |
57 | ||
58 | if (serial_flush(serial) != SR_OK) | |
59 | return SR_ERR; | |
60 | ||
61 | return SR_OK; | |
62 | } | |
63 | ||
64 | static int modbus_serial_rtu_source_add(struct sr_session *session, void *priv, | |
65 | int events, int timeout, sr_receive_data_callback cb, void *cb_data) | |
66 | { | |
67 | struct modbus_serial_rtu *modbus = priv; | |
68 | struct sr_serial_dev_inst *serial = modbus->serial; | |
69 | ||
70 | return serial_source_add(session, serial, events, timeout, cb, cb_data); | |
71 | } | |
72 | ||
73 | static int modbus_serial_rtu_source_remove(struct sr_session *session, | |
74 | void *priv) | |
75 | { | |
76 | struct modbus_serial_rtu *modbus = priv; | |
77 | struct sr_serial_dev_inst *serial = modbus->serial; | |
78 | ||
79 | return serial_source_remove(session, serial); | |
80 | } | |
81 | ||
82 | static uint16_t modbus_serial_rtu_crc(uint16_t crc, | |
83 | const uint8_t *buffer, int len) | |
84 | { | |
85 | int i; | |
86 | ||
87 | if (!buffer || len < 0) | |
88 | return crc; | |
89 | ||
90 | while (len--) { | |
91 | crc ^= *buffer++; | |
92 | for (i = 0; i < 8; i++) { | |
93 | int carry = crc & 1; | |
94 | crc >>= 1; | |
95 | if (carry) | |
96 | crc ^= 0xA001; | |
97 | } | |
98 | } | |
99 | return crc; | |
100 | } | |
101 | ||
102 | static int modbus_serial_rtu_send(void *priv, | |
103 | const uint8_t *buffer, int buffer_size) | |
104 | { | |
105 | int result; | |
106 | struct modbus_serial_rtu *modbus = priv; | |
107 | struct sr_serial_dev_inst *serial = modbus->serial; | |
108 | uint8_t slave_addr = modbus->slave_addr; | |
109 | uint16_t crc; | |
110 | ||
111 | result = serial_write_nonblocking(serial, &slave_addr, sizeof(slave_addr)); | |
112 | if (result < 0) | |
113 | return result; | |
114 | ||
115 | result = serial_write_nonblocking(serial, buffer, buffer_size); | |
116 | if (result < 0) | |
117 | return result; | |
118 | ||
119 | crc = modbus_serial_rtu_crc(0xFFFF, &slave_addr, sizeof(slave_addr)); | |
120 | crc = modbus_serial_rtu_crc(crc, buffer, buffer_size); | |
121 | ||
122 | result = serial_write_nonblocking(serial, &crc, sizeof(crc)); | |
123 | if (result < 0) | |
124 | return result; | |
125 | ||
126 | return SR_OK; | |
127 | } | |
128 | ||
129 | static int modbus_serial_rtu_read_begin(void *priv, uint8_t *function_code) | |
130 | { | |
131 | struct modbus_serial_rtu *modbus = priv; | |
132 | uint8_t slave_addr; | |
133 | int ret; | |
134 | ||
135 | ret = serial_read_blocking(modbus->serial, &slave_addr, 1, 100); | |
136 | if (ret != 1 || slave_addr != modbus->slave_addr) | |
137 | return ret; | |
138 | ||
139 | ret = serial_read_blocking(modbus->serial, function_code, 1, 100); | |
140 | if (ret != 1) | |
141 | return ret; | |
142 | ||
143 | modbus->crc = modbus_serial_rtu_crc(0xFFFF, &slave_addr, sizeof(slave_addr)); | |
144 | modbus->crc = modbus_serial_rtu_crc(modbus->crc, function_code, 1); | |
145 | ||
146 | return SR_OK; | |
147 | } | |
148 | ||
149 | static int modbus_serial_rtu_read_data(void *priv, uint8_t *buf, int maxlen) | |
150 | { | |
151 | struct modbus_serial_rtu *modbus = priv; | |
152 | int ret; | |
153 | ||
154 | ret = serial_read_nonblocking(modbus->serial, buf, maxlen); | |
155 | if (ret < 0) | |
156 | return ret; | |
157 | modbus->crc = modbus_serial_rtu_crc(modbus->crc, buf, ret); | |
158 | return ret; | |
159 | } | |
160 | ||
161 | static int modbus_serial_rtu_read_end(void *priv) | |
162 | { | |
163 | struct modbus_serial_rtu *modbus = priv; | |
164 | uint16_t crc; | |
165 | int ret; | |
166 | ||
167 | ret = serial_read_blocking(modbus->serial, &crc, sizeof(crc), 100); | |
168 | if (ret != 2) | |
169 | return ret; | |
170 | ||
171 | if (crc != modbus->crc) { | |
172 | sr_err("CRC error (0x%04X vs 0x%04X).", crc, modbus->crc); | |
173 | return SR_ERR_DATA; | |
174 | } | |
175 | ||
176 | return SR_OK; | |
177 | } | |
178 | ||
179 | static int modbus_serial_rtu_close(void *priv) | |
180 | { | |
181 | struct modbus_serial_rtu *modbus = priv; | |
182 | ||
183 | return serial_close(modbus->serial); | |
184 | } | |
185 | ||
186 | static void modbus_serial_rtu_free(void *priv) | |
187 | { | |
188 | struct modbus_serial_rtu *modbus = priv; | |
189 | ||
190 | sr_serial_dev_inst_free(modbus->serial); | |
191 | } | |
192 | ||
193 | SR_PRIV const struct sr_modbus_dev_inst modbus_serial_rtu_dev = { | |
194 | .name = "serial_rtu", | |
195 | .prefix = "", | |
196 | .priv_size = sizeof(struct modbus_serial_rtu), | |
197 | .scan = NULL, | |
198 | .dev_inst_new = modbus_serial_rtu_dev_inst_new, | |
199 | .open = modbus_serial_rtu_open, | |
200 | .source_add = modbus_serial_rtu_source_add, | |
201 | .source_remove = modbus_serial_rtu_source_remove, | |
202 | .send = modbus_serial_rtu_send, | |
203 | .read_begin = modbus_serial_rtu_read_begin, | |
204 | .read_data = modbus_serial_rtu_read_data, | |
205 | .read_end = modbus_serial_rtu_read_end, | |
206 | .close = modbus_serial_rtu_close, | |
207 | .free = modbus_serial_rtu_free, | |
208 | }; |