[libsigrok.git] / hardware / uni-t-dmm / protocol.c

/*
 * This file is part of the libsigrok project.
 *
 * Copyright (C) 2012-2013 Uwe Hermann <uwe@hermann-uwe.de>
 *
 * 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 2 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, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
 */

#include <string.h>
#include <glib.h>
#include "libsigrok.h"
#include "libsigrok-internal.h"
#include "protocol.h"

extern struct dmm_info udmms[];

/*
 * Driver for various UNI-T multimeters (and rebranded ones).
 *
 * Most UNI-T DMMs can be used with two (three) different PC interface cables:
 *  - The UT-D04 USB/HID cable, old version with Hoitek HE2325U chip.
 *  - The UT-D04 USB/HID cable, new version with WCH CH9325 chip.
 *  - The UT-D02 RS232 cable.
 *
 * This driver is meant to support all USB/HID cables, and various DMMs that
 * can be attached to a PC via these cables. Currently only the UT-D04 cable
 * (new version) is supported/tested.
 * The UT-D02 RS232 cable is handled by the 'serial-dmm' driver.
 *
 * The data for one DMM packet (e.g. 14 bytes if the respective DMM uses a
 * Fortune Semiconductor FS9922-DMM4 chip) is spread across multiple
 * 8-byte chunks.
 *
 * An 8-byte chunk looks like this:
 *  - Byte 0: 0xfz, where z is the number of actual data bytes in this chunk.
 *  - Bytes 1-7: z data bytes, the rest of the bytes should be ignored.
 *
 * Example:
 *  f0 00 00 00 00 00 00 00 (no data bytes)
 *  f2 55 77 00 00 00 00 00 (2 data bytes, 0x55 and 0x77)
 *  f1 d1 00 00 00 00 00 00 (1 data byte, 0xd1)
 */

static void decode_packet(struct sr_dev_inst *sdi, int dmm, const uint8_t *buf,
			  void *info)
{
	struct dev_context *devc;
	struct sr_datafeed_packet packet;
	struct sr_datafeed_analog analog;
	float floatval;
	int ret;

	devc = sdi->priv;
	memset(&analog, 0, sizeof(struct sr_datafeed_analog));

	/* Parse the protocol packet. */
	ret = udmms[dmm].packet_parse(buf, &floatval, &analog, info);
	if (ret != SR_OK) {
		sr_dbg("Invalid DMM packet, ignoring.");
		return;
	}

	/* If this DMM needs additional handling, call the resp. function. */
	if (udmms[dmm].dmm_details)
		udmms[dmm].dmm_details(&analog, info);

	/* Send a sample packet with one analog value. */
	analog.channels = sdi->channels;
	analog.num_samples = 1;
	analog.data = &floatval;
	packet.type = SR_DF_ANALOG;
	packet.payload = &analog;
	sr_session_send(devc->cb_data, &packet);

	/* Increase sample count. */
	devc->num_samples++;
}

static int hid_chip_init(struct sr_dev_inst *sdi, uint16_t baudrate)
{
	int ret;
	uint8_t buf[5];
	struct sr_usb_dev_inst *usb;

	usb = sdi->conn;
	
	/* Detach kernel drivers which grabbed this device (if any). */
	if (libusb_kernel_driver_active(usb->devhdl, 0) == 1) {
		ret = libusb_detach_kernel_driver(usb->devhdl, 0);
		if (ret < 0) {
			sr_err("Failed to detach kernel driver: %s.",
			       libusb_error_name(ret));
			return SR_ERR;
		}
		sr_dbg("Successfully detached kernel driver.");
	} else {
		sr_dbg("No need to detach a kernel driver.");
	}

	/* Claim interface 0. */
	if ((ret = libusb_claim_interface(usb->devhdl, 0)) < 0) {
		sr_err("Failed to claim interface 0: %s.",
		       libusb_error_name(ret));
		return SR_ERR;
	}
	sr_dbg("Successfully claimed interface 0.");

	/* Set data for the HID feature report (e.g. baudrate). */
	buf[0] = baudrate & 0xff;        /* Baudrate, LSB */
	buf[1] = (baudrate >> 8) & 0xff; /* Baudrate, MSB */
	buf[2] = 0x00;                   /* Unknown/unused (?) */
	buf[3] = 0x00;                   /* Unknown/unused (?) */
	buf[4] = 0x03;                   /* Unknown, always 0x03. */

	/* Send HID feature report to setup the baudrate/chip. */
	sr_dbg("Sending initial HID feature report.");
	sr_spew("HID init = 0x%02x 0x%02x 0x%02x 0x%02x 0x%02x (%d baud)",
		buf[0], buf[1], buf[2], buf[3], buf[4], baudrate);
	ret = libusb_control_transfer(
		usb->devhdl, /* libusb device handle */
		LIBUSB_REQUEST_TYPE_CLASS |
		LIBUSB_RECIPIENT_INTERFACE |
		LIBUSB_ENDPOINT_OUT,
		9, /* bRequest: HID set_report */
		0x300, /* wValue: HID feature, report number 0 */
		0, /* wIndex: interface 0 */
		(unsigned char *)&buf, /* payload buffer */
		5, /* wLength: 5 bytes payload */
		1000 /* timeout (ms) */);

	if (ret < 0) {
		sr_err("HID feature report error: %s.", libusb_error_name(ret));
		return SR_ERR;
	}

	if (ret != 5) {
		/* TODO: Handle better by also sending the remaining bytes. */
		sr_err("Short packet: sent %d/5 bytes.", ret);
		return SR_ERR;
	}

	sr_dbg("Successfully sent initial HID feature report.");

	return SR_OK;
}

static void log_8byte_chunk(const uint8_t *buf)
{
	sr_spew("8-byte chunk: %02x %02x %02x %02x %02x %02x %02x %02x "
		"(%d data bytes)", buf[0], buf[1], buf[2], buf[3],
		buf[4], buf[5], buf[6], buf[7], (buf[0] & 0x0f));
}

static void log_dmm_packet(const uint8_t *buf)
{
	sr_dbg("DMM packet:   %02x %02x %02x %02x %02x %02x %02x"
	       " %02x %02x %02x %02x %02x %02x %02x",
	       buf[0], buf[1], buf[2], buf[3], buf[4], buf[5], buf[6],
	       buf[7], buf[8], buf[9], buf[10], buf[11], buf[12], buf[13]);
}

static int get_and_handle_data(struct sr_dev_inst *sdi, int dmm, void *info)
{
	struct dev_context *devc;
	uint8_t buf[CHUNK_SIZE], *pbuf;
	int i, ret, len, num_databytes_in_chunk;
	struct sr_usb_dev_inst *usb;

	devc = sdi->priv;
	usb = sdi->conn;
	pbuf = devc->protocol_buf;

	/* On the first run, we need to init the HID chip. */
	if (devc->first_run) {
		if ((ret = hid_chip_init(sdi, udmms[dmm].baudrate)) != SR_OK) {
			sr_err("HID chip init failed: %d.", ret);
			return SR_ERR;
		}
		memset(pbuf, 0x00, DMM_BUFSIZE);
		devc->first_run = FALSE;
	}

	memset(&buf, 0x00, CHUNK_SIZE);

	/* Get data from EP2 using an interrupt transfer. */
	ret = libusb_interrupt_transfer(
		usb->devhdl, /* libusb device handle */
		LIBUSB_ENDPOINT_IN | 2, /* EP2, IN */
		(unsigned char *)&buf, /* receive buffer */
		CHUNK_SIZE, /* wLength */
		&len, /* actually received byte count */
		1000 /* timeout (ms) */);

	if (ret < 0) {
		sr_err("USB receive error: %s.", libusb_error_name(ret));
		return SR_ERR;
	}

	if (len != CHUNK_SIZE) {
		sr_err("Short packet: received %d/%d bytes.", len, CHUNK_SIZE);
		/* TODO: Print the bytes? */
		return SR_ERR;
	}

	log_8byte_chunk((const uint8_t *)&buf);

	/* If there are no data bytes just return (without error). */
	if (buf[0] == 0xf0)
		return SR_OK;

	devc->bufoffset = 0;

	/*
	 * Append the 1-7 data bytes of this chunk to pbuf.
	 *
	 * Special case:
	 * DMMs with Cyrustek ES51922 chip need serial settings of
	 * 19230/7o1. The WCH CH9325 UART to USB/HID chip used in (some
	 * versions of) the UNI-T UT-D04 cable however, will also send
	 * the parity bit to the host in the 8-byte data chunks. This bit
	 * is encoded in bit 7 of each of the 1-7 data bytes and must thus
	 * be removed in order for the actual ES51922 protocol parser to
	 * work properly.
	 */
	num_databytes_in_chunk = buf[0] & 0x0f;
	for (i = 0; i < num_databytes_in_chunk; i++, devc->buflen++) {
		pbuf[devc->buflen] = buf[1 + i];
		if (udmms[dmm].packet_parse == sr_es519xx_19200_14b_parse)
			pbuf[devc->buflen] &= ~(1 << 7);
	}

	/* Now look for packets in that data. */
	while ((devc->buflen - devc->bufoffset) >= udmms[dmm].packet_size) {
		if (udmms[dmm].packet_valid(pbuf + devc->bufoffset)) {
			log_dmm_packet(pbuf + devc->bufoffset);
			decode_packet(sdi, dmm, pbuf + devc->bufoffset, info);
			devc->bufoffset += udmms[dmm].packet_size;
		} else {
			devc->bufoffset++;
		}
	}

	/* Move remaining bytes to beginning of buffer. */
	for (i = 0; i < devc->buflen - devc->bufoffset; i++)
		pbuf[i] = pbuf[devc->bufoffset + i];
	devc->buflen -= devc->bufoffset;

	return SR_OK;
}

static int receive_data(int fd, int revents, int dmm, void *info, void *cb_data)
{
	int ret;
	struct sr_dev_inst *sdi;
	struct dev_context *devc;
	int64_t time_ms;

	(void)fd;
	(void)revents;

	sdi = cb_data;
	devc = sdi->priv;

	if ((ret = get_and_handle_data(sdi, dmm, info)) != SR_OK)
		return FALSE;

	/* Abort acquisition if we acquired enough samples. */
	if (devc->limit_samples && devc->num_samples >= devc->limit_samples) {
		sr_info("Requested number of samples reached.");
		sdi->driver->dev_acquisition_stop(sdi, cb_data);
	}

	if (devc->limit_msec) {
		time_ms = (g_get_monotonic_time() - devc->starttime) / 1000;
		if (time_ms > (int64_t)devc->limit_msec) {
			sr_info("Requested time limit reached.");
			sdi->driver->dev_acquisition_stop(sdi, cb_data);
			return TRUE;
		}
	}

	return TRUE;
}

#define RECEIVE_DATA(ID_UPPER, DMM_DRIVER) \
SR_PRIV int receive_data_##ID_UPPER(int fd, int revents, void *cb_data) { \
	struct DMM_DRIVER##_info info; \
	return receive_data(fd, revents, ID_UPPER, &info, cb_data); }

/* Driver-specific receive_data() wrappers */
RECEIVE_DATA(TECPEL_DMM_8061, fs9721)
RECEIVE_DATA(UNI_T_UT60A, fs9721)
RECEIVE_DATA(UNI_T_UT60E, fs9721)
RECEIVE_DATA(UNI_T_UT60G, es519xx)
RECEIVE_DATA(UNI_T_UT61B, fs9922)
RECEIVE_DATA(UNI_T_UT61C, fs9922)
RECEIVE_DATA(UNI_T_UT61D, fs9922)
RECEIVE_DATA(UNI_T_UT61E, es519xx)
RECEIVE_DATA(VOLTCRAFT_VC820, fs9721)
RECEIVE_DATA(VOLTCRAFT_VC830, fs9922)
RECEIVE_DATA(VOLTCRAFT_VC840, fs9721)
RECEIVE_DATA(TENMA_72_7745, es519xx)
RECEIVE_DATA(TENMA_72_7750, es519xx)
Commit	Line	Data
79081ec8	1	/*
50985c20	2	* This file is part of the libsigrok project.
79081ec8	3	*
bc653a56	4	* Copyright (C) 2012-2013 Uwe Hermann <uwe@hermann-uwe.de>
79081ec8 UH	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 2 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, write to the Free Software
	18	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
	19	*/
	20
	21	#include <string.h>
	22	#include <glib.h>
	23	#include "libsigrok.h"
	24	#include "libsigrok-internal.h"
	25	#include "protocol.h"
	26
79b9a237 BV	27	extern struct dmm_info udmms[];
79b9a237 BV	28
79081ec8 UH	29	/*
	30	* Driver for various UNI-T multimeters (and rebranded ones).
	31	*
	32	* Most UNI-T DMMs can be used with two (three) different PC interface cables:
	33	* - The UT-D04 USB/HID cable, old version with Hoitek HE2325U chip.
	34	* - The UT-D04 USB/HID cable, new version with WCH CH9325 chip.
c8852687	35	* - The UT-D02 RS232 cable.
79081ec8	36	*
c8852687	37	* This driver is meant to support all USB/HID cables, and various DMMs that
79081ec8	38	* can be attached to a PC via these cables. Currently only the UT-D04 cable
c8852687 UH	39	* (new version) is supported/tested.
c8852687 UH	40	* The UT-D02 RS232 cable is handled by the 'serial-dmm' driver.
79081ec8 UH	41	*
	42	* The data for one DMM packet (e.g. 14 bytes if the respective DMM uses a
	43	* Fortune Semiconductor FS9922-DMM4 chip) is spread across multiple
	44	* 8-byte chunks.
	45	*
	46	* An 8-byte chunk looks like this:
	47	* - Byte 0: 0xfz, where z is the number of actual data bytes in this chunk.
	48	* - Bytes 1-7: z data bytes, the rest of the bytes should be ignored.
	49	*
	50	* Example:
	51	* f0 00 00 00 00 00 00 00 (no data bytes)
	52	* f2 55 77 00 00 00 00 00 (2 data bytes, 0x55 and 0x77)
	53	* f1 d1 00 00 00 00 00 00 (1 data byte, 0xd1)
79081ec8 UH	54	*/
79081ec8 UH	55
c8852687 UH	56	static void decode_packet(struct sr_dev_inst sdi, int dmm, const uint8_t buf,
c8852687 UH	57	void *info)
79081ec8	58	{
bc653a56	59	struct dev_context *devc;
79081ec8 UH	60	struct sr_datafeed_packet packet;
	61	struct sr_datafeed_analog analog;
	62	float floatval;
	63	int ret;
	64
69e19dd7	65	devc = sdi->priv;
79081ec8 UH	66	memset(&analog, 0, sizeof(struct sr_datafeed_analog));
	67
	68	/* Parse the protocol packet. */
bc653a56	69	ret = udmms[dmm].packet_parse(buf, &floatval, &analog, info);
fdbcb86d	70	if (ret != SR_OK) {
bc653a56	71	sr_dbg("Invalid DMM packet, ignoring.");
79081ec8 UH	72	return;
	73	}
	74
c8852687 UH	75	/* If this DMM needs additional handling, call the resp. function. */
	76	if (udmms[dmm].dmm_details)
	77	udmms[dmm].dmm_details(&analog, info);
	78
79081ec8	79	/* Send a sample packet with one analog value. */
ba7dd8bb	80	analog.channels = sdi->channels;
79081ec8 UH	81	analog.num_samples = 1;
	82	analog.data = &floatval;
	83	packet.type = SR_DF_ANALOG;
	84	packet.payload = &analog;
	85	sr_session_send(devc->cb_data, &packet);
	86
	87	/* Increase sample count. */
	88	devc->num_samples++;
	89	}
	90
3ece1dff	91	static int hid_chip_init(struct sr_dev_inst *sdi, uint16_t baudrate)
79081ec8 UH	92	{
	93	int ret;
	94	uint8_t buf[5];
3ece1dff	95	struct sr_usb_dev_inst *usb;
79081ec8	96
3ece1dff UH	97	usb = sdi->conn;
3ece1dff UH	98
79081ec8	99	/* Detach kernel drivers which grabbed this device (if any). */
3ece1dff UH	100	if (libusb_kernel_driver_active(usb->devhdl, 0) == 1) {
3ece1dff UH	101	ret = libusb_detach_kernel_driver(usb->devhdl, 0);
79081ec8	102	if (ret < 0) {
d4928d71 PS	103	sr_err("Failed to detach kernel driver: %s.",
d4928d71 PS	104	libusb_error_name(ret));
79081ec8 UH	105	return SR_ERR;
	106	}
	107	sr_dbg("Successfully detached kernel driver.");
	108	} else {
	109	sr_dbg("No need to detach a kernel driver.");
	110	}
	111
	112	/* Claim interface 0. */
3ece1dff	113	if ((ret = libusb_claim_interface(usb->devhdl, 0)) < 0) {
d4928d71 PS	114	sr_err("Failed to claim interface 0: %s.",
d4928d71 PS	115	libusb_error_name(ret));
79081ec8 UH	116	return SR_ERR;
	117	}
	118	sr_dbg("Successfully claimed interface 0.");
	119
bc653a56	120	/* Set data for the HID feature report (e.g. baudrate). */
79081ec8 UH	121	buf[0] = baudrate & 0xff; /* Baudrate, LSB */
	122	buf[1] = (baudrate >> 8) & 0xff; /* Baudrate, MSB */
	123	buf[2] = 0x00; /* Unknown/unused (?) */
	124	buf[3] = 0x00; /* Unknown/unused (?) */
	125	buf[4] = 0x03; /* Unknown, always 0x03. */
	126
	127	/* Send HID feature report to setup the baudrate/chip. */
	128	sr_dbg("Sending initial HID feature report.");
	129	sr_spew("HID init = 0x%02x 0x%02x 0x%02x 0x%02x 0x%02x (%d baud)",
	130	buf[0], buf[1], buf[2], buf[3], buf[4], baudrate);
	131	ret = libusb_control_transfer(
3ece1dff	132	usb->devhdl, /* libusb device handle */
79081ec8 UH	133	LIBUSB_REQUEST_TYPE_CLASS \|
	134	LIBUSB_RECIPIENT_INTERFACE \|
	135	LIBUSB_ENDPOINT_OUT,
	136	9, /* bRequest: HID set_report */
	137	0x300, /* wValue: HID feature, report number 0 */
	138	0, /* wIndex: interface 0 */
	139	(unsigned char )&buf, / payload buffer */
	140	5, /* wLength: 5 bytes payload */
	141	1000 /* timeout (ms) */);
	142
	143	if (ret < 0) {
d4928d71	144	sr_err("HID feature report error: %s.", libusb_error_name(ret));
79081ec8 UH	145	return SR_ERR;
	146	}
	147
	148	if (ret != 5) {
	149	/* TODO: Handle better by also sending the remaining bytes. */
	150	sr_err("Short packet: sent %d/5 bytes.", ret);
	151	return SR_ERR;
	152	}
	153
	154	sr_dbg("Successfully sent initial HID feature report.");
	155
	156	return SR_OK;
	157	}
	158
	159	static void log_8byte_chunk(const uint8_t *buf)
	160	{
	161	sr_spew("8-byte chunk: %02x %02x %02x %02x %02x %02x %02x %02x "
	162	"(%d data bytes)", buf[0], buf[1], buf[2], buf[3],
	163	buf[4], buf[5], buf[6], buf[7], (buf[0] & 0x0f));
	164	}
	165
	166	static void log_dmm_packet(const uint8_t *buf)
	167	{
	168	sr_dbg("DMM packet: %02x %02x %02x %02x %02x %02x %02x"
	169	" %02x %02x %02x %02x %02x %02x %02x",
	170	buf[0], buf[1], buf[2], buf[3], buf[4], buf[5], buf[6],
	171	buf[7], buf[8], buf[9], buf[10], buf[11], buf[12], buf[13]);
	172	}
	173
bc653a56	174	static int get_and_handle_data(struct sr_dev_inst sdi, int dmm, void info)
79081ec8	175	{
79081ec8	176	struct dev_context *devc;
bc653a56	177	uint8_t buf[CHUNK_SIZE], *pbuf;
79081ec8	178	int i, ret, len, num_databytes_in_chunk;
3ece1dff	179	struct sr_usb_dev_inst *usb;
79081ec8	180
79081ec8	181	devc = sdi->priv;
3ece1dff	182	usb = sdi->conn;
79081ec8 UH	183	pbuf = devc->protocol_buf;
	184
	185	/* On the first run, we need to init the HID chip. */
bc653a56	186	if (devc->first_run) {
3ece1dff	187	if ((ret = hid_chip_init(sdi, udmms[dmm].baudrate)) != SR_OK) {
79081ec8	188	sr_err("HID chip init failed: %d.", ret);
bc653a56	189	return SR_ERR;
79081ec8	190	}
bc653a56 UH	191	memset(pbuf, 0x00, DMM_BUFSIZE);
bc653a56 UH	192	devc->first_run = FALSE;
79081ec8 UH	193	}
	194
	195	memset(&buf, 0x00, CHUNK_SIZE);
	196
	197	/* Get data from EP2 using an interrupt transfer. */
	198	ret = libusb_interrupt_transfer(
3ece1dff	199	usb->devhdl, /* libusb device handle */
79081ec8 UH	200	LIBUSB_ENDPOINT_IN \| 2, /* EP2, IN */
	201	(unsigned char )&buf, / receive buffer */
	202	CHUNK_SIZE, /* wLength */
	203	&len, /* actually received byte count */
	204	1000 /* timeout (ms) */);
	205
	206	if (ret < 0) {
d4928d71	207	sr_err("USB receive error: %s.", libusb_error_name(ret));
bc653a56	208	return SR_ERR;
79081ec8 UH	209	}
	210
	211	if (len != CHUNK_SIZE) {
	212	sr_err("Short packet: received %d/%d bytes.", len, CHUNK_SIZE);
	213	/* TODO: Print the bytes? */
bc653a56	214	return SR_ERR;
79081ec8 UH	215	}
	216
	217	log_8byte_chunk((const uint8_t *)&buf);
	218
bc653a56 UH	219	/* If there are no data bytes just return (without error). */
	220	if (buf[0] == 0xf0)
	221	return SR_OK;
	222
	223	devc->bufoffset = 0;
	224
4d7ddff7 UH	225	/*
	226	* Append the 1-7 data bytes of this chunk to pbuf.
	227	*
	228	* Special case:
	229	* DMMs with Cyrustek ES51922 chip need serial settings of
	230	* 19230/7o1. The WCH CH9325 UART to USB/HID chip used in (some
	231	* versions of) the UNI-T UT-D04 cable however, will also send
	232	* the parity bit to the host in the 8-byte data chunks. This bit
	233	* is encoded in bit 7 of each of the 1-7 data bytes and must thus
	234	* be removed in order for the actual ES51922 protocol parser to
	235	* work properly.
	236	*/
bc653a56	237	num_databytes_in_chunk = buf[0] & 0x0f;
4d7ddff7 UH	238	for (i = 0; i < num_databytes_in_chunk; i++, devc->buflen++) {
4d7ddff7 UH	239	pbuf[devc->buflen] = buf[1 + i];
d97824e5	240	if (udmms[dmm].packet_parse == sr_es519xx_19200_14b_parse)
4d7ddff7 UH	241	pbuf[devc->buflen] &= ~(1 << 7);
4d7ddff7 UH	242	}
bc653a56 UH	243
	244	/* Now look for packets in that data. */
	245	while ((devc->buflen - devc->bufoffset) >= udmms[dmm].packet_size) {
	246	if (udmms[dmm].packet_valid(pbuf + devc->bufoffset)) {
	247	log_dmm_packet(pbuf + devc->bufoffset);
	248	decode_packet(sdi, dmm, pbuf + devc->bufoffset, info);
	249	devc->bufoffset += udmms[dmm].packet_size;
	250	} else {
	251	devc->bufoffset++;
79081ec8	252	}
bc653a56	253	}
79081ec8	254
bc653a56 UH	255	/* Move remaining bytes to beginning of buffer. */
	256	for (i = 0; i < devc->buflen - devc->bufoffset; i++)
	257	pbuf[i] = pbuf[devc->bufoffset + i];
	258	devc->buflen -= devc->bufoffset;
79081ec8	259
bc653a56 UH	260	return SR_OK;
	261	}
	262
	263	static int receive_data(int fd, int revents, int dmm, void info, void cb_data)
	264	{
	265	int ret;
	266	struct sr_dev_inst *sdi;
	267	struct dev_context *devc;
c5ffac41	268	int64_t time_ms;
c8852687	269
bc653a56 UH	270	(void)fd;
bc653a56 UH	271	(void)revents;
c8852687	272
bc653a56 UH	273	sdi = cb_data;
bc653a56 UH	274	devc = sdi->priv;
c8852687	275
bc653a56 UH	276	if ((ret = get_and_handle_data(sdi, dmm, info)) != SR_OK)
bc653a56 UH	277	return FALSE;
79081ec8 UH	278
79081ec8 UH	279	/* Abort acquisition if we acquired enough samples. */
35e199da	280	if (devc->limit_samples && devc->num_samples >= devc->limit_samples) {
79081ec8 UH	281	sr_info("Requested number of samples reached.");
	282	sdi->driver->dev_acquisition_stop(sdi, cb_data);
	283	}
	284
c5ffac41 UH	285	if (devc->limit_msec) {
	286	time_ms = (g_get_monotonic_time() - devc->starttime) / 1000;
	287	if (time_ms > (int64_t)devc->limit_msec) {
	288	sr_info("Requested time limit reached.");
	289	sdi->driver->dev_acquisition_stop(sdi, cb_data);
	290	return TRUE;
	291	}
	292	}
	293
79081ec8 UH	294	return TRUE;
79081ec8 UH	295	}
fdbcb86d	296
c8852687 UH	297	#define RECEIVE_DATA(ID_UPPER, DMM_DRIVER) \
	298	SR_PRIV int receive_data_##ID_UPPER(int fd, int revents, void *cb_data) { \
	299	struct DMM_DRIVER##_info info; \
	300	return receive_data(fd, revents, ID_UPPER, &info, cb_data); }
fdbcb86d	301
c8852687	302	/* Driver-specific receive_data() wrappers */
b6bad47c	303	RECEIVE_DATA(TECPEL_DMM_8061, fs9721)
a7a163a7 FK	304	RECEIVE_DATA(UNI_T_UT60A, fs9721)
a7a163a7 FK	305	RECEIVE_DATA(UNI_T_UT60E, fs9721)
4104ef81	306	RECEIVE_DATA(UNI_T_UT60G, es519xx)
7cb69b18	307	RECEIVE_DATA(UNI_T_UT61B, fs9922)
5f985df2	308	RECEIVE_DATA(UNI_T_UT61C, fs9922)
c8852687	309	RECEIVE_DATA(UNI_T_UT61D, fs9922)
d97824e5	310	RECEIVE_DATA(UNI_T_UT61E, es519xx)
c8852687	311	RECEIVE_DATA(VOLTCRAFT_VC820, fs9721)
c1345749	312	RECEIVE_DATA(VOLTCRAFT_VC830, fs9922)
695d0e1e	313	RECEIVE_DATA(VOLTCRAFT_VC840, fs9721)
d9e79c51 UH	314	RECEIVE_DATA(TENMA_72_7745, es519xx)
d9e79c51 UH	315	RECEIVE_DATA(TENMA_72_7750, es519xx)