[libsigrok.git] / src / hardware / rdtech-um / protocol.c

/*
 * This file is part of the libsigrok project.
 *
 * Copyright (C) 2018-2020 Andreas Sandberg <andreas@sandberg.pp.se>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <config.h>

#include <glib.h>
#include <libsigrok/libsigrok.h>
#include <math.h>
#include <stdlib.h>
#include <string.h>

#include "libsigrok-internal.h"
#include "protocol.h"

/* Read/write timeouts, poll request intervals. */
#define PROBE_TO_MS 1000
#define WRITE_TO_MS 1
#define POLL_PERIOD_MS 100

/* Expected receive data size for poll responses. */
#define POLL_RECV_LEN 130

/* Command code to request another poll response. */
#define UM_CMD_POLL 0xf0

static const struct rdtech_um_channel_desc default_channels[] = {
	{ "V", { 2, BVT_BE_UINT16, 1, }, { 10, 1e3, }, 2, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	{ "I", { 4, BVT_BE_UINT16, 1, }, { 1, 1e3, }, 3, SR_MQ_CURRENT, SR_UNIT_AMPERE },
	{ "D+", { 96, BVT_BE_UINT16, 1, }, { 10, 1e3, }, 2, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	{ "D-", { 98, BVT_BE_UINT16, 1, }, { 10, 1e3, }, 2, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	{ "T", { 10, BVT_BE_UINT16, 1, }, { 1, 1, }, 0, SR_MQ_TEMPERATURE, SR_UNIT_CELSIUS },
	/* Threshold-based recording (mWh) */
	{ "E", { 106, BVT_BE_UINT32, 1, }, { 1, 1e3, }, 3, SR_MQ_ENERGY, SR_UNIT_WATT_HOUR },
};

static const struct rdtech_um_channel_desc um25c_channels[] = {
	{ "V", { 2, BVT_BE_UINT16, 1, }, { 1, 1e3, }, 3, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	{ "I", { 4, BVT_BE_UINT16, 1, }, { 100, 1e6, }, 4, SR_MQ_CURRENT, SR_UNIT_AMPERE },
	{ "D+", { 96, BVT_BE_UINT16, 1, }, { 10, 1e3, }, 2, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	{ "D-", { 98, BVT_BE_UINT16, 1, }, { 10, 1e3, }, 2, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	{ "T", { 10, BVT_BE_UINT16, 1, }, { 1, 1, }, 0, SR_MQ_TEMPERATURE, SR_UNIT_CELSIUS },
	/* Threshold-based recording (mWh) */
	{ "E", { 106, BVT_BE_UINT32, 1, }, { 1, 1e3, }, 3, SR_MQ_ENERGY, SR_UNIT_WATT_HOUR },
};

static gboolean csum_ok_fff1(const uint8_t *buf, size_t len)
{
	uint16_t csum_recv;

	if (len != POLL_RECV_LEN)
		return FALSE;

	csum_recv = read_u16be(&buf[len - sizeof(uint16_t)]);
	if (csum_recv != 0xfff1)
		return FALSE;

	return TRUE;
}

static gboolean csum_ok_um34c(const uint8_t *buf, size_t len)
{
	static const int positions[] = {
		1, 3, 7, 9, 15, 17, 19, 23, 31, 39, 41, 45, 49, 53,
		55, 57, 59, 63, 67, 69, 73, 79, 83, 89, 97, 99, 109,
		111, 113, 119, 121, 127,
	};

	size_t i;
	uint8_t csum_calc, csum_recv;

	if (len != POLL_RECV_LEN)
		return FALSE;

	csum_calc = 0;
	for (i = 0; i < ARRAY_SIZE(positions); i++)
		csum_calc ^= buf[positions[i]];
	csum_recv = read_u8(&buf[len - sizeof(uint8_t)]);
	if (csum_recv != csum_calc)
		return FALSE;

	return TRUE;
}

static const struct rdtech_um_profile um_profiles[] = {
	{ "UM24C", RDTECH_UM24C, ARRAY_AND_SIZE(default_channels), csum_ok_fff1, },
	{ "UM25C", RDTECH_UM25C, ARRAY_AND_SIZE(um25c_channels), csum_ok_fff1, },
	{ "UM34C", RDTECH_UM34C, ARRAY_AND_SIZE(default_channels), csum_ok_um34c, },
};

static const struct rdtech_um_profile *find_profile(uint16_t id)
{
	size_t i;
	const struct rdtech_um_profile *profile;

	for (i = 0; i < ARRAY_SIZE(um_profiles); i++) {
		profile = &um_profiles[i];
		if (profile->model_id == id)
			return profile;
	}
	return NULL;
}

SR_PRIV const struct rdtech_um_profile *rdtech_um_probe(struct sr_serial_dev_inst *serial)
{
	const struct rdtech_um_profile *p;
	uint8_t req;
	int ret;
	uint8_t buf[RDTECH_UM_BUFSIZE];
	int rcvd;
	uint16_t model_id;

	req = UM_CMD_POLL;
	ret = serial_write_blocking(serial, &req, sizeof(req), WRITE_TO_MS);
	if (ret < 0) {
		sr_err("Failed to send probe request.");
		return NULL;
	}

	rcvd = serial_read_blocking(serial, buf, POLL_RECV_LEN, PROBE_TO_MS);
	if (rcvd != POLL_RECV_LEN) {
		sr_err("Failed to read probe response.");
		return NULL;
	}

	model_id = read_u16be(&buf[0]);
	p = find_profile(model_id);
	if (!p) {
		sr_err("Unrecognized UM device (0x%.4" PRIx16 ").", model_id);
		return NULL;
	}

	if (!p->csum_ok(buf, rcvd)) {
		sr_err("Probe response fails checksum verification.");
		return NULL;
	}

	return p;
}

SR_PRIV int rdtech_um_poll(const struct sr_dev_inst *sdi, gboolean force)
{
	struct dev_context *devc;
	int64_t now, elapsed;
	struct sr_serial_dev_inst *serial;
	uint8_t req;
	int ret;

	/* Don't send request when receive data is being accumulated. */
	devc = sdi->priv;
	if (!force && devc->buflen)
		return SR_OK;

	/* Check for expired intervals or forced requests. */
	now = g_get_monotonic_time() / 1000;
	elapsed = now - devc->cmd_sent_at;
	if (!force && elapsed < POLL_PERIOD_MS)
		return SR_OK;

	/* Send another poll request. Update interval only on success. */
	serial = sdi->conn;
	req = UM_CMD_POLL;
	ret = serial_write_blocking(serial, &req, sizeof(req), WRITE_TO_MS);
	if (ret < 0) {
		sr_err("Unable to send poll request.");
		return SR_ERR;
	}
	devc->cmd_sent_at = now;

	return SR_OK;
}

static int process_data(struct sr_dev_inst *sdi,
	const uint8_t *data, size_t dlen)
{
	struct dev_context *devc;
	const struct rdtech_um_profile *p;
	size_t ch_idx;
	float v;
	int ret;

	devc = sdi->priv;
	p = devc->profile;

	sr_spew("Received poll packet (len: %zu).", dlen);
	if (dlen < POLL_RECV_LEN) {
		sr_err("Insufficient response data length: %zu", dlen);
		return SR_ERR_DATA;
	}

	if (!p->csum_ok(data, POLL_RECV_LEN)) {
		sr_err("Packet checksum verification failed.");
		return SR_ERR_DATA;
	}

	for (ch_idx = 0; ch_idx < p->channel_count; ch_idx++) {
		ret = bv_get_value(&v, &p->channels[ch_idx].spec, data, dlen);
		if (ret != SR_OK)
			return ret;
		ret = feed_queue_analog_submit(devc->feeds[ch_idx], v, 1);
		if (ret != SR_OK)
			return ret;
	}

	sr_sw_limits_update_samples_read(&devc->limits, 1);
	if (sr_sw_limits_check(&devc->limits))
		sr_dev_acquisition_stop(sdi);

	return SR_OK;
}

static int accum_data(struct sr_dev_inst *sdi, struct sr_serial_dev_inst *serial)
{
	struct dev_context *devc;
	const struct rdtech_um_profile *p;
	uint8_t *rdptr;
	size_t space, rcvd, rdlen;
	int ret;
	gboolean do_sync_check;
	size_t sync_len, sync_idx;

	/*
	 * Receive data became available. Drain the serial transport.
	 * Grab incoming data in as large a chunk as possible. Also
	 * copes with zero receive data length, as some transports may
	 * trigger periodically without data really being available.
	 */
	devc = sdi->priv;
	p = devc->profile;
	rdptr = &devc->buf[devc->buflen];
	space = sizeof(devc->buf) - devc->buflen;
	do_sync_check = FALSE;
	sync_len = sizeof(uint16_t);
	while (space) {
		ret = serial_read_nonblocking(serial, rdptr, space);
		if (ret < 0)
			return SR_ERR_IO;
		rcvd = (size_t)ret;
		if (rcvd == 0)
			break;
		if (rcvd > space)
			return SR_ERR_BUG;
		if (devc->buflen < sync_len)
			do_sync_check = TRUE;
		devc->buflen += rcvd;
		if (devc->buflen < sync_len)
			do_sync_check = FALSE;
		space -= rcvd;
		rdptr += rcvd;
	}

	/*
	 * Synchronize to the packetized input stream. Check the model
	 * ID at the start of receive data. Which is a weak condition,
	 * but going out of sync should be rare, and repeated attempts
	 * to synchronize should eventually succeed. Try to rate limit
	 * the emission of diagnostics messages. (Re-)run this logic
	 * at the first reception which makes enough data available,
	 * but not during subsequent accumulation of more data.
	 *
	 * Reducing redundancy in the implementation at the same time as
	 * increasing robustness would involve the creation of a checker
	 * routine, which just gets called for every byte position until
	 * it succeeds. Similar to what a previous implementation of the
	 * read loop did, which was expensive on the serial transport.
	 */
	sync_idx = 0;
	if (do_sync_check && read_u16be(&devc->buf[sync_idx]) != p->model_id)
		sr_warn("Unexpected response data, trying to synchronize.");
	while (do_sync_check) {
		if (sync_idx + sync_len >= devc->buflen)
			break;
		if (read_u16be(&devc->buf[sync_idx]) == p->model_id)
			break;
		sync_idx++;
	}
	if (do_sync_check && sync_idx) {
		sr_dbg("Skipping %zu bytes in attempt to sync.", sync_idx);
		sync_len = devc->buflen - sync_idx;
		if (sync_len)
			memmove(&devc->buf[0], &devc->buf[sync_idx], sync_len);
		devc->buflen -= sync_idx;
	}

	/*
	 * Process packets as their reception completes. Periodically
	 * re-transmit poll requests. Discard consumed data after all
	 * processing has completed.
	 */
	rdptr = devc->buf;
	rdlen = devc->buflen;
	ret = SR_OK;
	while (ret == SR_OK && rdlen >= POLL_RECV_LEN) {
		ret = process_data(sdi, rdptr, rdlen);
		if (ret != SR_OK) {
			sr_err("Processing response packet failed.");
			break;
		}
		rdptr += POLL_RECV_LEN;
		rdlen -= POLL_RECV_LEN;

		if (0 && !sr_sw_limits_check(&devc->limits))
			(void)rdtech_um_poll(sdi, FALSE);
	}
	rcvd = rdptr - devc->buf;
	devc->buflen -= rcvd;
	if (devc->buflen)
		memmove(&devc->buf[0], rdptr, devc->buflen);

	return ret;
}

SR_PRIV int rdtech_um_receive_data(int fd, int revents, void *cb_data)
{
	struct sr_dev_inst *sdi;
	struct dev_context *devc;
	struct sr_serial_dev_inst *serial;
	int ret;

	(void)fd;

	if (!(sdi = cb_data))
		return TRUE;
	if (!(devc = sdi->priv))
		return TRUE;

	/*
	 * Drain and process receive data as it becomes available.
	 * Terminate acquisition upon receive or processing error.
	 */
	serial = sdi->conn;
	if (revents == G_IO_IN) {
		ret = accum_data(sdi, serial);
		if (ret != SR_OK) {
			sr_dev_acquisition_stop(sdi);
			return TRUE;
		}
	}

	/* Check configured acquisition limits. */
	if (sr_sw_limits_check(&devc->limits)) {
		sr_dev_acquisition_stop(sdi);
		return TRUE;
	}

	/* Periodically retransmit measurement requests. */
	(void)rdtech_um_poll(sdi, FALSE);

	return TRUE;
}
Commit	Line	Data
8b607a24 AS	1	/*
	2	* This file is part of the libsigrok project.
	3	*
	4	* Copyright (C) 2018-2020 Andreas Sandberg <andreas@sandberg.pp.se>
	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 <config.h>
dca972f8	21
8b607a24 AS	22	#include <glib.h>
8b607a24 AS	23	#include <libsigrok/libsigrok.h>
dca972f8 GS	24	#include <math.h>
	25	#include <stdlib.h>
	26	#include <string.h>
	27
8b607a24 AS	28	#include "libsigrok-internal.h"
	29	#include "protocol.h"
	30
ae33433b GS	31	/* Read/write timeouts, poll request intervals. */
	32	#define PROBE_TO_MS 1000
	33	#define WRITE_TO_MS 1
	34	#define POLL_PERIOD_MS 100
8b607a24	35
ae33433b GS	36	/* Expected receive data size for poll responses. */
ae33433b GS	37	#define POLL_RECV_LEN 130
8b607a24	38
ae33433b	39	/* Command code to request another poll response. */
8b607a24 AS	40	#define UM_CMD_POLL 0xf0
8b607a24 AS	41
1b43eb29 GS	42	static const struct rdtech_um_channel_desc default_channels[] = {
	43	{ "V", { 2, BVT_BE_UINT16, 1, }, { 10, 1e3, }, 2, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	44	{ "I", { 4, BVT_BE_UINT16, 1, }, { 1, 1e3, }, 3, SR_MQ_CURRENT, SR_UNIT_AMPERE },
	45	{ "D+", { 96, BVT_BE_UINT16, 1, }, { 10, 1e3, }, 2, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	46	{ "D-", { 98, BVT_BE_UINT16, 1, }, { 10, 1e3, }, 2, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	47	{ "T", { 10, BVT_BE_UINT16, 1, }, { 1, 1, }, 0, SR_MQ_TEMPERATURE, SR_UNIT_CELSIUS },
8b607a24	48	/* Threshold-based recording (mWh) */
1b43eb29	49	{ "E", { 106, BVT_BE_UINT32, 1, }, { 1, 1e3, }, 3, SR_MQ_ENERGY, SR_UNIT_WATT_HOUR },
8b607a24 AS	50	};
8b607a24 AS	51
1b43eb29 GS	52	static const struct rdtech_um_channel_desc um25c_channels[] = {
	53	{ "V", { 2, BVT_BE_UINT16, 1, }, { 1, 1e3, }, 3, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	54	{ "I", { 4, BVT_BE_UINT16, 1, }, { 100, 1e6, }, 4, SR_MQ_CURRENT, SR_UNIT_AMPERE },
	55	{ "D+", { 96, BVT_BE_UINT16, 1, }, { 10, 1e3, }, 2, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	56	{ "D-", { 98, BVT_BE_UINT16, 1, }, { 10, 1e3, }, 2, SR_MQ_VOLTAGE, SR_UNIT_VOLT },
	57	{ "T", { 10, BVT_BE_UINT16, 1, }, { 1, 1, }, 0, SR_MQ_TEMPERATURE, SR_UNIT_CELSIUS },
8b607a24	58	/* Threshold-based recording (mWh) */
1b43eb29	59	{ "E", { 106, BVT_BE_UINT32, 1, }, { 1, 1e3, }, 3, SR_MQ_ENERGY, SR_UNIT_WATT_HOUR },
8b607a24 AS	60	};
8b607a24 AS	61
85accbc2	62	static gboolean csum_ok_fff1(const uint8_t *buf, size_t len)
8b607a24	63	{
85accbc2 GS	64	uint16_t csum_recv;
85accbc2 GS	65
ae33433b	66	if (len != POLL_RECV_LEN)
85accbc2 GS	67	return FALSE;
	68
	69	csum_recv = read_u16be(&buf[len - sizeof(uint16_t)]);
	70	if (csum_recv != 0xfff1)
	71	return FALSE;
	72
	73	return TRUE;
8b607a24 AS	74	}
8b607a24 AS	75
85accbc2	76	static gboolean csum_ok_um34c(const uint8_t *buf, size_t len)
8b607a24 AS	77	{
	78	static const int positions[] = {
	79	1, 3, 7, 9, 15, 17, 19, 23, 31, 39, 41, 45, 49, 53,
	80	55, 57, 59, 63, 67, 69, 73, 79, 83, 89, 97, 99, 109,
	81	111, 113, 119, 121, 127,
	82	};
fda34e5a	83
85accbc2 GS	84	size_t i;
85accbc2 GS	85	uint8_t csum_calc, csum_recv;
8b607a24	86
ae33433b	87	if (len != POLL_RECV_LEN)
85accbc2	88	return FALSE;
8b607a24	89
85accbc2	90	csum_calc = 0;
8b607a24	91	for (i = 0; i < ARRAY_SIZE(positions); i++)
85accbc2 GS	92	csum_calc ^= buf[positions[i]];
	93	csum_recv = read_u8(&buf[len - sizeof(uint8_t)]);
	94	if (csum_recv != csum_calc)
	95	return FALSE;
8b607a24	96
85accbc2	97	return TRUE;
8b607a24 AS	98	}
	99
	100	static const struct rdtech_um_profile um_profiles[] = {
7cfa93b0 GS	101	{ "UM24C", RDTECH_UM24C, ARRAY_AND_SIZE(default_channels), csum_ok_fff1, },
	102	{ "UM25C", RDTECH_UM25C, ARRAY_AND_SIZE(um25c_channels), csum_ok_fff1, },
	103	{ "UM34C", RDTECH_UM34C, ARRAY_AND_SIZE(default_channels), csum_ok_um34c, },
8b607a24 AS	104	};
	105
	106	static const struct rdtech_um_profile *find_profile(uint16_t id)
	107	{
4441e573 GS	108	size_t i;
4441e573 GS	109	const struct rdtech_um_profile *profile;
fda34e5a	110
8b607a24	111	for (i = 0; i < ARRAY_SIZE(um_profiles); i++) {
4441e573 GS	112	profile = &um_profiles[i];
	113	if (profile->model_id == id)
	114	return profile;
8b607a24 AS	115	}
	116	return NULL;
	117	}
	118
	119	SR_PRIV const struct rdtech_um_profile rdtech_um_probe(struct sr_serial_dev_inst serial)
	120	{
	121	const struct rdtech_um_profile *p;
ae33433b GS	122	uint8_t req;
ae33433b GS	123	int ret;
85accbc2	124	uint8_t buf[RDTECH_UM_BUFSIZE];
4441e573	125	int rcvd;
bcbf2b59	126	uint16_t model_id;
8b607a24	127
ae33433b GS	128	req = UM_CMD_POLL;
	129	ret = serial_write_blocking(serial, &req, sizeof(req), WRITE_TO_MS);
	130	if (ret < 0) {
	131	sr_err("Failed to send probe request.");
8b607a24 AS	132	return NULL;
	133	}
	134
ae33433b GS	135	rcvd = serial_read_blocking(serial, buf, POLL_RECV_LEN, PROBE_TO_MS);
ae33433b GS	136	if (rcvd != POLL_RECV_LEN) {
8b607a24 AS	137	sr_err("Failed to read probe response.");
	138	return NULL;
	139	}
	140
bcbf2b59 GS	141	model_id = read_u16be(&buf[0]);
bcbf2b59 GS	142	p = find_profile(model_id);
8b607a24	143	if (!p) {
bcbf2b59	144	sr_err("Unrecognized UM device (0x%.4" PRIx16 ").", model_id);
8b607a24 AS	145	return NULL;
	146	}
	147
4441e573	148	if (!p->csum_ok(buf, rcvd)) {
bcbf2b59	149	sr_err("Probe response fails checksum verification.");
8b607a24 AS	150	return NULL;
	151	}
	152
	153	return p;
	154	}
	155
c1f4f589	156	SR_PRIV int rdtech_um_poll(const struct sr_dev_inst *sdi, gboolean force)
8b607a24	157	{
fda34e5a	158	struct dev_context *devc;
c1f4f589	159	int64_t now, elapsed;
fda34e5a	160	struct sr_serial_dev_inst *serial;
ae33433b GS	161	uint8_t req;
ae33433b GS	162	int ret;
8b607a24	163
481f894f	164	/* Don't send request when receive data is being accumulated. */
c1f4f589	165	devc = sdi->priv;
481f894f GS	166	if (!force && devc->buflen)
	167	return SR_OK;
	168
	169	/* Check for expired intervals or forced requests. */
c1f4f589 GS	170	now = g_get_monotonic_time() / 1000;
c1f4f589 GS	171	elapsed = now - devc->cmd_sent_at;
ae33433b	172	if (!force && elapsed < POLL_PERIOD_MS)
c1f4f589 GS	173	return SR_OK;
	174
	175	/* Send another poll request. Update interval only on success. */
fda34e5a	176	serial = sdi->conn;
ae33433b GS	177	req = UM_CMD_POLL;
	178	ret = serial_write_blocking(serial, &req, sizeof(req), WRITE_TO_MS);
	179	if (ret < 0) {
8b607a24 AS	180	sr_err("Unable to send poll request.");
	181	return SR_ERR;
	182	}
c1f4f589	183	devc->cmd_sent_at = now;
8b607a24 AS	184
	185	return SR_OK;
	186	}
	187
481f894f GS	188	static int process_data(struct sr_dev_inst *sdi,
481f894f GS	189	const uint8_t *data, size_t dlen)
8b607a24	190	{
fda34e5a	191	struct dev_context *devc;
481f894f	192	const struct rdtech_um_profile *p;
4441e573	193	size_t ch_idx;
1b43eb29	194	float v;
481f894f	195	int ret;
8b607a24	196
fda34e5a	197	devc = sdi->priv;
481f894f GS	198	p = devc->profile;
	199
	200	sr_spew("Received poll packet (len: %zu).", dlen);
	201	if (dlen < POLL_RECV_LEN) {
	202	sr_err("Insufficient response data length: %zu", dlen);
	203	return SR_ERR_DATA;
	204	}
	205
	206	if (!p->csum_ok(data, POLL_RECV_LEN)) {
	207	sr_err("Packet checksum verification failed.");
	208	return SR_ERR_DATA;
8b607a24 AS	209	}
8b607a24 AS	210
1b43eb29 GS	211	for (ch_idx = 0; ch_idx < p->channel_count; ch_idx++) {
	212	ret = bv_get_value(&v, &p->channels[ch_idx].spec, data, dlen);
	213	if (ret != SR_OK)
	214	return ret;
	215	ret = feed_queue_analog_submit(devc->feeds[ch_idx], v, 1);
481f894f GS	216	if (ret != SR_OK)
481f894f GS	217	return ret;
323b32f4	218	}
8b607a24 AS	219
8b607a24 AS	220	sr_sw_limits_update_samples_read(&devc->limits, 1);
481f894f GS	221	if (sr_sw_limits_check(&devc->limits))
	222	sr_dev_acquisition_stop(sdi);
	223
	224	return SR_OK;
8b607a24 AS	225	}
8b607a24 AS	226
481f894f	227	static int accum_data(struct sr_dev_inst sdi, struct sr_serial_dev_inst serial)
8b607a24	228	{
fda34e5a GS	229	struct dev_context *devc;
fda34e5a GS	230	const struct rdtech_um_profile *p;
481f894f GS	231	uint8_t *rdptr;
	232	size_t space, rcvd, rdlen;
	233	int ret;
	234	gboolean do_sync_check;
	235	size_t sync_len, sync_idx;
	236
	237	/*
	238	* Receive data became available. Drain the serial transport.
	239	* Grab incoming data in as large a chunk as possible. Also
	240	* copes with zero receive data length, as some transports may
	241	* trigger periodically without data really being available.
	242	*/
fda34e5a GS	243	devc = sdi->priv;
fda34e5a GS	244	p = devc->profile;
481f894f GS	245	rdptr = &devc->buf[devc->buflen];
	246	space = sizeof(devc->buf) - devc->buflen;
	247	do_sync_check = FALSE;
	248	sync_len = sizeof(uint16_t);
	249	while (space) {
	250	ret = serial_read_nonblocking(serial, rdptr, space);
	251	if (ret < 0)
	252	return SR_ERR_IO;
	253	rcvd = (size_t)ret;
	254	if (rcvd == 0)
	255	break;
	256	if (rcvd > space)
	257	return SR_ERR_BUG;
	258	if (devc->buflen < sync_len)
	259	do_sync_check = TRUE;
	260	devc->buflen += rcvd;
	261	if (devc->buflen < sync_len)
	262	do_sync_check = FALSE;
	263	space -= rcvd;
	264	rdptr += rcvd;
	265	}
	266
	267	/*
	268	* Synchronize to the packetized input stream. Check the model
	269	* ID at the start of receive data. Which is a weak condition,
	270	* but going out of sync should be rare, and repeated attempts
	271	* to synchronize should eventually succeed. Try to rate limit
	272	* the emission of diagnostics messages. (Re-)run this logic
	273	* at the first reception which makes enough data available,
	274	* but not during subsequent accumulation of more data.
	275	*
	276	* Reducing redundancy in the implementation at the same time as
	277	* increasing robustness would involve the creation of a checker
	278	* routine, which just gets called for every byte position until
	279	* it succeeds. Similar to what a previous implementation of the
	280	* read loop did, which was expensive on the serial transport.
	281	*/
	282	sync_idx = 0;
	283	if (do_sync_check && read_u16be(&devc->buf[sync_idx]) != p->model_id)
	284	sr_warn("Unexpected response data, trying to synchronize.");
	285	while (do_sync_check) {
	286	if (sync_idx + sync_len >= devc->buflen)
	287	break;
	288	if (read_u16be(&devc->buf[sync_idx]) == p->model_id)
	289	break;
	290	sync_idx++;
	291	}
	292	if (do_sync_check && sync_idx) {
	293	sr_dbg("Skipping %zu bytes in attempt to sync.", sync_idx);
	294	sync_len = devc->buflen - sync_idx;
	295	if (sync_len)
	296	memmove(&devc->buf[0], &devc->buf[sync_idx], sync_len);
	297	devc->buflen -= sync_idx;
	298	}
	299
	300	/*
	301	* Process packets as their reception completes. Periodically
	302	* re-transmit poll requests. Discard consumed data after all
	303	* processing has completed.
	304	*/
	305	rdptr = devc->buf;
	306	rdlen = devc->buflen;
	307	ret = SR_OK;
	308	while (ret == SR_OK && rdlen >= POLL_RECV_LEN) {
309	ret = process_data(sdi, rdptr, rdlen);
310	if (ret != SR_OK) {
311	sr_err("Processing response packet failed.");
312	break;
8b607a24	313	}
481f894f GS	314	rdptr += POLL_RECV_LEN;
	315	rdlen -= POLL_RECV_LEN;
	316
	317	if (0 && !sr_sw_limits_check(&devc->limits))
	318	(void)rdtech_um_poll(sdi, FALSE);
8b607a24	319	}
481f894f GS	320	rcvd = rdptr - devc->buf;
	321	devc->buflen -= rcvd;
	322	if (devc->buflen)
	323	memmove(&devc->buf[0], rdptr, devc->buflen);
8b607a24	324
481f894f	325	return ret;
8b607a24 AS	326	}
	327
	328	SR_PRIV int rdtech_um_receive_data(int fd, int revents, void *cb_data)
	329	{
	330	struct sr_dev_inst *sdi;
	331	struct dev_context *devc;
	332	struct sr_serial_dev_inst *serial;
481f894f	333	int ret;
8b607a24 AS	334
	335	(void)fd;
	336
	337	if (!(sdi = cb_data))
	338	return TRUE;
8b607a24 AS	339	if (!(devc = sdi->priv))
	340	return TRUE;
	341
481f894f GS	342	/*
	343	* Drain and process receive data as it becomes available.
	344	* Terminate acquisition upon receive or processing error.
	345	*/
8b607a24	346	serial = sdi->conn;
481f894f GS	347	if (revents == G_IO_IN) {
	348	ret = accum_data(sdi, serial);
	349	if (ret != SR_OK) {
	350	sr_dev_acquisition_stop(sdi);
	351	return TRUE;
	352	}
	353	}
8b607a24	354
ae33433b	355	/* Check configured acquisition limits. */
8b607a24 AS	356	if (sr_sw_limits_check(&devc->limits)) {
	357	sr_dev_acquisition_stop(sdi);
	358	return TRUE;
	359	}
	360
481f894f	361	/* Periodically retransmit measurement requests. */
c1f4f589	362	(void)rdtech_um_poll(sdi, FALSE);
8b607a24 AS	363
	364	return TRUE;
	365	}