[libsigrok.git] / src / hardware / ipdbg-la / protocol.c

/*
 * This file is part of the libsigrok project.
 *
 * Copyright (C) 2016 Eva Kissling <eva.kissling@bluewin.ch>
 *
 * 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>

#ifdef _WIN32
#define _WIN32_WINNT 0x0501
#include <winsock2.h>
#include <ws2tcpip.h>
#else
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <sys/ioctl.h>
#endif
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include "protocol.h"

#define BUFFER_SIZE 4

/* Top-level command opcodes */
#define CMD_SET_TRIGGER            0x00
#define CMD_CFG_TRIGGER            0xF0
#define CMD_CFG_LA                 0x0F
#define CMD_START                  0xFE
#define CMD_RESET                  0xEE

#define CMD_GET_BUS_WIDTHS         0xAA
#define CMD_GET_LA_ID              0xBB
#define CMD_ESCAPE                 0x55

/* Trigger subfunction command opcodes */
#define CMD_TRIG_MASKS             0xF1
#define CMD_TRIG_MASK              0xF3
#define CMD_TRIG_VALUE             0xF7

#define CMD_TRIG_MASKS_LAST        0xF9
#define CMD_TRIG_MASK_LAST         0xFB
#define CMD_TRIG_VALUE_LAST        0xFF

#define CMD_TRIG_SELECT_EDGE_MASK  0xF5
#define CMD_TRIG_SET_EDGE_MASK     0xF6

/* LA subfunction command opcodes */
#define CMD_LA_DELAY               0x1F

static gboolean data_available(struct ipdbg_la_tcp *tcp)
{
#ifdef _WIN32
	u_long bytes_available;
	if(ioctlsocket(tcp->socket, FIONREAD, &bytes_available) != 0){
#else
	int bytes_available;
	if (ioctl(tcp->socket, FIONREAD, &bytes_available) < 0){ 	// TIOCMGET
#endif
		sr_err("FIONREAD failed: %s\n", g_strerror(errno));
		return FALSE;
	}
	return (bytes_available > 0);
}

SR_PRIV struct ipdbg_la_tcp *ipdbg_la_tcp_new(void)
{
	struct ipdbg_la_tcp *tcp;

	tcp = g_malloc0(sizeof(struct ipdbg_la_tcp));
	tcp->address = NULL;
	tcp->port = NULL;
	tcp->socket = -1;

	return tcp;
}

SR_PRIV void ipdbg_la_tcp_free(struct ipdbg_la_tcp *tcp)
{
	g_free(tcp->address);
	g_free(tcp->port);
}

SR_PRIV int ipdbg_la_tcp_open(struct ipdbg_la_tcp *tcp)
{
	struct addrinfo hints;
	struct addrinfo *results, *res;
	int err;

	memset(&hints, 0, sizeof(hints));
	hints.ai_family = AF_UNSPEC;
	hints.ai_socktype = SOCK_STREAM;
	hints.ai_protocol = IPPROTO_TCP;

	err = getaddrinfo(tcp->address, tcp->port, &hints, &results);

	if (err) {
		sr_err("Address lookup failed: %s:%s: %s", tcp->address,
			tcp->port, gai_strerror(err));
		return SR_ERR;
	}

	for (res = results; res; res = res->ai_next) {
		if ((tcp->socket = socket(res->ai_family, res->ai_socktype,
			res->ai_protocol)) < 0)
			continue;
		if (connect(tcp->socket, res->ai_addr, res->ai_addrlen) != 0) {
			close(tcp->socket);
			tcp->socket = -1;
			continue;
		}
		break;
	}

	freeaddrinfo(results);

	if (tcp->socket < 0) {
		sr_err("Failed to connect to %s:%s: %s", tcp->address, tcp->port,
			g_strerror(errno));
		return SR_ERR;
	}

	return SR_OK;
}

SR_PRIV int ipdbg_la_tcp_close(struct ipdbg_la_tcp *tcp)
{
	int ret = SR_OK;

#ifdef _WIN32
	if (shutdown(tcp->socket, SD_SEND) != SOCKET_ERROR)
	{
		char recvbuf[16];
		int recvbuflen = 16;
		// Receive until the peer closes the connection
		while(recv(tcp->socket, recvbuf, recvbuflen, 0) > 0);
	}
#endif
	if (close(tcp->socket) < 0)
		ret = SR_ERR;

	tcp->socket = -1;

	return ret;
}

static int tcp_send(struct ipdbg_la_tcp *tcp, const uint8_t *buf, size_t len)
{
	int out;
	out = send(tcp->socket, (const char *)buf, len, 0);

	if (out < 0) {
		sr_err("Send error: %s", g_strerror(errno));
		return SR_ERR;
	}

	if (out < (int)len)
		sr_dbg("Only sent %d/%d bytes of data.", out, (int)len);

	return SR_OK;
}

static int tcp_receive_blocking(struct ipdbg_la_tcp *tcp,
	uint8_t *buf, int bufsize)
{
	int received = 0;
	int error_count = 0;

	/* Timeout after 500ms of not receiving data */
	while ((received < bufsize) && (error_count < 500)) {
		if (ipdbg_la_tcp_receive(tcp, buf) > 0) {
			buf++;
			received++;
		} else {
			error_count++;
			g_usleep(1000);  /* Sleep for 1ms */
		}
	}

	return received;
}

SR_PRIV int ipdbg_la_tcp_receive(struct ipdbg_la_tcp *tcp,
	uint8_t *buf)
{
	int received = 0;

	if (data_available(tcp)) {
		while (received < 1) {
			int len = recv(tcp->socket, (char *)buf, 1, 0);

			if (len < 0) {
				sr_err("Receive error: %s", g_strerror(errno));
				return SR_ERR;
			} else
				received += len;
		}

		return received;
	} else
		return -1;
}

SR_PRIV int ipdbg_la_convert_trigger(const struct sr_dev_inst *sdi)
{
	struct dev_context *devc;
	struct sr_trigger *trigger;
	struct sr_trigger_stage *stage;
	struct sr_trigger_match *match;
	const GSList *l, *m;

	devc = sdi->priv;

	devc->num_stages = 0;
	devc->num_transfers = 0;
	devc->raw_sample_buf = NULL;

	for (uint64_t i = 0; i < devc->data_width_bytes; i++) {
		devc->trigger_mask[i] = 0;
		devc->trigger_value[i] = 0;
		devc->trigger_mask_last[i] = 0;
		devc->trigger_value_last[i] = 0;
		devc->trigger_edge_mask[i] = 0;
	}

	if (!(trigger = sr_session_trigger_get(sdi->session)))
		return SR_OK;

	for (l = trigger->stages; l; l = l->next) {
		stage = l->data;
		for (m = stage->matches; m; m = m->next) {
			match = m->data;
			int byte_idx = match->channel->index / 8;
			uint8_t match_bit = 1 << (match->channel->index % 8);

			if (!match->channel->enabled)
				/* Ignore disabled channels with a trigger. */
				continue;

			if (match->match == SR_TRIGGER_ONE) {
				devc->trigger_value[byte_idx] |= match_bit;
				devc->trigger_mask[byte_idx] |= match_bit;
				devc->trigger_mask_last[byte_idx] &= ~match_bit;
				devc->trigger_edge_mask[byte_idx] &= ~match_bit;
			} else if (match->match == SR_TRIGGER_ZERO) {
				devc->trigger_value[byte_idx] &= ~match_bit;
				devc->trigger_mask[byte_idx] |= match_bit;
				devc->trigger_mask_last[byte_idx] &= ~match_bit;
				devc->trigger_edge_mask[byte_idx] &= ~match_bit;
			} else if (match->match == SR_TRIGGER_RISING) {
				devc->trigger_value[byte_idx] |= match_bit;
				devc->trigger_value_last[byte_idx] &=
				    ~match_bit;
				devc->trigger_mask[byte_idx] |= match_bit;
				devc->trigger_mask_last[byte_idx] |= match_bit;
				devc->trigger_edge_mask[byte_idx] &= ~match_bit;
			} else if (match->match == SR_TRIGGER_FALLING) {
				devc->trigger_value[byte_idx] &= ~match_bit;
				devc->trigger_value_last[byte_idx] |= match_bit;
				devc->trigger_mask[byte_idx] |= match_bit;
				devc->trigger_mask_last[byte_idx] |= match_bit;
				devc->trigger_edge_mask[byte_idx] &= ~match_bit;
			} else if (match->match == SR_TRIGGER_EDGE) {
				devc->trigger_mask[byte_idx] &= ~match_bit;
				devc->trigger_mask_last[byte_idx] &= ~match_bit;
				devc->trigger_edge_mask[byte_idx] |= match_bit;
			}
		}
	}

	return SR_OK;
}

SR_PRIV int ipdbg_la_receive_data(int fd, int revents, void *cb_data)
{
	const struct sr_dev_inst *sdi;
	struct dev_context *devc;

	(void)fd;
	(void)revents;

	sdi = cb_data;
	if (!sdi)
		return FALSE;

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

	struct ipdbg_la_tcp *tcp = sdi->conn;
	struct sr_datafeed_packet packet;
	struct sr_datafeed_logic logic;

	if (!devc->raw_sample_buf) {
		devc->raw_sample_buf =
			g_try_malloc(devc->limit_samples * devc->data_width_bytes);
		if (!devc->raw_sample_buf) {
			sr_err("Sample buffer malloc failed.");
			return FALSE;
		}
	}

	if (devc->num_transfers <
		(devc->limit_samples_max * devc->data_width_bytes)) {
		uint8_t byte;

		if (ipdbg_la_tcp_receive(tcp, &byte) == 1) {
			if (devc->num_transfers <
				(devc->limit_samples * devc->data_width_bytes))
				devc->raw_sample_buf[devc->num_transfers] = byte;

			devc->num_transfers++;
		}
	} else {
		if (devc->delay_value > 0) {
			/* There are pre-trigger samples, send those first. */
			packet.type = SR_DF_LOGIC;
			packet.payload = &logic;
			logic.length = devc->delay_value * devc->data_width_bytes;
			logic.unitsize = devc->data_width_bytes;
			logic.data = devc->raw_sample_buf;
			sr_session_send(cb_data, &packet);
		}

		/* Send the trigger. */
		packet.type = SR_DF_TRIGGER;
		sr_session_send(cb_data, &packet);

		/* Send post-trigger samples. */
		packet.type = SR_DF_LOGIC;
		packet.payload = &logic;
		logic.length = (devc->limit_samples - devc->delay_value) *
			devc->data_width_bytes;
		logic.unitsize = devc->data_width_bytes;
		logic.data = devc->raw_sample_buf +
			(devc->delay_value * devc->data_width_bytes);
		sr_session_send(cb_data, &packet);

		g_free(devc->raw_sample_buf);
		devc->raw_sample_buf = NULL;

		ipdbg_la_abort_acquisition(sdi);
	}

	return TRUE;
}

static int send_escaping(struct ipdbg_la_tcp *tcp, uint8_t *data_to_send,
	uint32_t length)
{
	uint8_t escape = CMD_ESCAPE;

	while (length--) {
		uint8_t payload = *data_to_send++;

		if (payload == CMD_RESET)
			if (tcp_send(tcp, &escape, 1) != SR_OK)
				sr_warn("Couldn't send escape");

		if (payload == CMD_ESCAPE)
			if (tcp_send(tcp, &escape, 1) != SR_OK)
				sr_warn("Couldn't send escape");

		if (tcp_send(tcp, &payload, 1) != SR_OK)
			sr_warn("Couldn't send data");
	}

	return SR_OK;
}

SR_PRIV int ipdbg_la_send_delay(struct dev_context *devc,
	struct ipdbg_la_tcp *tcp)
{
	devc->delay_value = (devc->limit_samples / 100.0) * devc->capture_ratio;

	uint8_t buf;
	buf = CMD_CFG_LA;
	tcp_send(tcp, &buf, 1);
	buf = CMD_LA_DELAY;
	tcp_send(tcp, &buf, 1);

	uint8_t delay_buf[4] = { devc->delay_value & 0x000000ff,
		(devc->delay_value >> 8) & 0x000000ff,
		(devc->delay_value >> 16) & 0x000000ff,
		(devc->delay_value >> 24) & 0x000000ff
	};

	for (uint64_t i = 0; i < devc->addr_width_bytes; i++)
		send_escaping(tcp, &(delay_buf[devc->addr_width_bytes - 1 - i]), 1);

	return SR_OK;
}

SR_PRIV int ipdbg_la_send_trigger(struct dev_context *devc,
	struct ipdbg_la_tcp *tcp)
{
	uint8_t buf;

	/* Mask */
	buf = CMD_CFG_TRIGGER;
	tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASKS;
	tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASK;
	tcp_send(tcp, &buf, 1);

	for (size_t i = 0; i < devc->data_width_bytes; i++)
		send_escaping(tcp,
			devc->trigger_mask + devc->data_width_bytes - 1 - i, 1);

	/* Value */
	buf = CMD_CFG_TRIGGER;
	tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASKS;
	tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_VALUE;
	tcp_send(tcp, &buf, 1);

	for (size_t i = 0; i < devc->data_width_bytes; i++)
		send_escaping(tcp,
			devc->trigger_value + devc->data_width_bytes - 1 - i, 1);

	/* Mask_last */
	buf = CMD_CFG_TRIGGER;
	tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASKS_LAST;
	tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASK_LAST;
	tcp_send(tcp, &buf, 1);

	for (size_t i = 0; i < devc->data_width_bytes; i++)
		send_escaping(tcp,
			devc->trigger_mask_last + devc->data_width_bytes - 1 - i, 1);

	/* Value_last */
	buf = CMD_CFG_TRIGGER;
	tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASKS_LAST;
	tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_VALUE_LAST;
	tcp_send(tcp, &buf, 1);

	for (size_t i = 0; i < devc->data_width_bytes; i++)
		send_escaping(tcp,
			devc->trigger_value_last + devc->data_width_bytes - 1 - i, 1);

	/* Edge_mask */
	buf = CMD_CFG_TRIGGER;
	tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_SELECT_EDGE_MASK;
	tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_SET_EDGE_MASK;
	tcp_send(tcp, &buf, 1);

	for (size_t i = 0; i < devc->data_width_bytes; i++)
		send_escaping(tcp,
			devc->trigger_edge_mask + devc->data_width_bytes - 1 - i, 1);

	return SR_OK;
}

SR_PRIV void ipdbg_la_get_addrwidth_and_datawidth(
	struct ipdbg_la_tcp *tcp, struct dev_context *devc)
{
	uint8_t buf[8];
	uint8_t read_cmd = CMD_GET_BUS_WIDTHS;

	if (tcp_send(tcp, &read_cmd, 1) != SR_OK)
		sr_warn("Can't send read command");

	if (tcp_receive_blocking(tcp, buf, 8) != 8)
		sr_warn("Can't get address and data width from device");

	devc->data_width = buf[0] & 0x000000FF;
	devc->data_width |= (buf[1] << 8) & 0x0000FF00;
	devc->data_width |= (buf[2] << 16) & 0x00FF0000;
	devc->data_width |= (buf[3] << 24) & 0xFF000000;

	devc->addr_width = buf[4] & 0x000000FF;
	devc->addr_width |= (buf[5] << 8) & 0x0000FF00;
	devc->addr_width |= (buf[6] << 16) & 0x00FF0000;
	devc->addr_width |= (buf[7] << 24) & 0xFF000000;

	const uint8_t host_word_size = 8;

	devc->data_width_bytes =
		(devc->data_width + host_word_size - 1) / host_word_size;
	devc->addr_width_bytes =
		(devc->addr_width + host_word_size - 1) / host_word_size;

	devc->limit_samples_max = (0x01 << devc->addr_width);
	devc->limit_samples = devc->limit_samples_max;

	devc->trigger_mask = g_malloc0(devc->data_width_bytes);
	devc->trigger_value = g_malloc0(devc->data_width_bytes);
	devc->trigger_mask_last = g_malloc0(devc->data_width_bytes);
	devc->trigger_value_last = g_malloc0(devc->data_width_bytes);
	devc->trigger_edge_mask = g_malloc0(devc->data_width_bytes);
}

SR_PRIV struct dev_context *ipdbg_la_dev_new(void)
{
	struct dev_context *devc;

	devc = g_malloc0(sizeof(struct dev_context));
	devc->capture_ratio = 50;

	return devc;
}

SR_PRIV int ipdbg_la_send_reset(struct ipdbg_la_tcp *tcp)
{
	uint8_t buf = CMD_RESET;
	if (tcp_send(tcp, &buf, 1) != SR_OK)
		sr_warn("Couldn't send reset");

	return SR_OK;
}

SR_PRIV int ipdbg_la_request_id(struct ipdbg_la_tcp *tcp)
{
	uint8_t buf = CMD_GET_LA_ID;
	if (tcp_send(tcp, &buf, 1) != SR_OK)
		sr_warn("Couldn't send ID request");

	char id[4];
	if (tcp_receive_blocking(tcp, (uint8_t *)id, 4) != 4) {
		sr_err("Couldn't read device ID");
		return SR_ERR;
	}

	if (strncmp(id, "IDBG", 4)) {
		sr_err("Invalid device ID: expected 'IDBG', got '%c%c%c%c'.",
			id[0], id[1], id[2], id[3]);
		return SR_ERR;
	}

	return SR_OK;
}

SR_PRIV void ipdbg_la_abort_acquisition(const struct sr_dev_inst *sdi)
{
	struct ipdbg_la_tcp *tcp = sdi->conn;

	sr_session_source_remove(sdi->session, tcp->socket);

	std_session_send_df_end(sdi);
}

SR_PRIV int ipdbg_la_send_start(struct ipdbg_la_tcp *tcp)
{
	uint8_t buf = CMD_START;

	if (tcp_send(tcp, &buf, 1) != SR_OK)
		sr_warn("Couldn't send start");

	return SR_OK;
}
Commit	Line	Data
a4210e18 EK	1	/*
	2	* This file is part of the libsigrok project.
	3	*
932ef10f	4	* Copyright (C) 2016 Eva Kissling <eva.kissling@bluewin.ch>
a4210e18 EK	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>
7f4c9a04	21
a4210e18 EK	22	#ifdef _WIN32
	23	#define _WIN32_WINNT 0x0501
	24	#include <winsock2.h>
	25	#include <ws2tcpip.h>
00f0016c	26	#else
a4210e18 EK	27	#include <sys/socket.h>
	28	#include <netinet/in.h>
	29	#include <arpa/inet.h>
	30	#include <netdb.h>
24fbd9f8	31	#include <sys/ioctl.h>
a4210e18	32	#endif
00f0016c UH	33	#include <string.h>
00f0016c UH	34	#include <unistd.h>
a4210e18 EK	35	#include <errno.h>
	36	#include "protocol.h"
	37
7f4c9a04	38	#define BUFFER_SIZE 4
2f15f5bf	39
7f4c9a04 SA	40	/* Top-level command opcodes */
	41	#define CMD_SET_TRIGGER 0x00
	42	#define CMD_CFG_TRIGGER 0xF0
	43	#define CMD_CFG_LA 0x0F
	44	#define CMD_START 0xFE
	45	#define CMD_RESET 0xEE
2f15f5bf	46
7f4c9a04 SA	47	#define CMD_GET_BUS_WIDTHS 0xAA
	48	#define CMD_GET_LA_ID 0xBB
	49	#define CMD_ESCAPE 0x55
2f15f5bf	50
7f4c9a04 SA	51	/* Trigger subfunction command opcodes */
	52	#define CMD_TRIG_MASKS 0xF1
	53	#define CMD_TRIG_MASK 0xF3
	54	#define CMD_TRIG_VALUE 0xF7
2f15f5bf	55
7f4c9a04 SA	56	#define CMD_TRIG_MASKS_LAST 0xF9
	57	#define CMD_TRIG_MASK_LAST 0xFB
	58	#define CMD_TRIG_VALUE_LAST 0xFF
2f15f5bf	59
7f4c9a04 SA	60	#define CMD_TRIG_SELECT_EDGE_MASK 0xF5
7f4c9a04 SA	61	#define CMD_TRIG_SET_EDGE_MASK 0xF6
2f15f5bf	62
7f4c9a04 SA	63	/* LA subfunction command opcodes */
7f4c9a04 SA	64	#define CMD_LA_DELAY 0x1F
a66099c6	65
4d33f5e1	66	static gboolean data_available(struct ipdbg_la_tcp *tcp)
b51288e3	67	{
00f0016c	68	#ifdef _WIN32
3bfdadf6	69	u_long bytes_available;
c54ca323	70	if(ioctlsocket(tcp->socket, FIONREAD, &bytes_available) != 0){
b51288e3	71	#else
c54ca323	72	int bytes_available;
	73	if (ioctl(tcp->socket, FIONREAD, &bytes_available) < 0){ // TIOCMGET
	74	#endif
4838c6ca	75	sr_err("FIONREAD failed: %s\n", g_strerror(errno));
4465837b	76	return FALSE;
9d2e5483	77	}
c54ca323	78	return (bytes_available > 0);
b51288e3 EK	79	}
b51288e3 EK	80
8a9788e2	81	SR_PRIV struct ipdbg_la_tcp *ipdbg_la_tcp_new(void)
a4210e18	82	{
8a9788e2	83	struct ipdbg_la_tcp *tcp;
a4210e18	84
8a9788e2	85	tcp = g_malloc0(sizeof(struct ipdbg_la_tcp));
9d2e5483 SA	86	tcp->address = NULL;
	87	tcp->port = NULL;
	88	tcp->socket = -1;
a4210e18	89
9d2e5483	90	return tcp;
a4210e18 EK	91	}
a4210e18 EK	92
8a9788e2	93	SR_PRIV void ipdbg_la_tcp_free(struct ipdbg_la_tcp *tcp)
7f4c9a04	94	{
9d2e5483 SA	95	g_free(tcp->address);
9d2e5483 SA	96	g_free(tcp->port);
7f4c9a04 SA	97	}
7f4c9a04 SA	98
8a9788e2	99	SR_PRIV int ipdbg_la_tcp_open(struct ipdbg_la_tcp *tcp)
a4210e18	100	{
9d2e5483 SA	101	struct addrinfo hints;
	102	struct addrinfo results, res;
	103	int err;
	104
	105	memset(&hints, 0, sizeof(hints));
	106	hints.ai_family = AF_UNSPEC;
	107	hints.ai_socktype = SOCK_STREAM;
	108	hints.ai_protocol = IPPROTO_TCP;
	109
	110	err = getaddrinfo(tcp->address, tcp->port, &hints, &results);
	111
	112	if (err) {
	113	sr_err("Address lookup failed: %s:%s: %s", tcp->address,
	114	tcp->port, gai_strerror(err));
	115	return SR_ERR;
	116	}
	117
	118	for (res = results; res; res = res->ai_next) {
	119	if ((tcp->socket = socket(res->ai_family, res->ai_socktype,
	120	res->ai_protocol)) < 0)
	121	continue;
	122	if (connect(tcp->socket, res->ai_addr, res->ai_addrlen) != 0) {
	123	close(tcp->socket);
	124	tcp->socket = -1;
	125	continue;
	126	}
	127	break;
	128	}
	129
	130	freeaddrinfo(results);
	131
	132	if (tcp->socket < 0) {
	133	sr_err("Failed to connect to %s:%s: %s", tcp->address, tcp->port,
	134	g_strerror(errno));
	135	return SR_ERR;
	136	}
	137
	138	return SR_OK;
a4210e18 EK	139	}
a4210e18 EK	140
8a9788e2	141	SR_PRIV int ipdbg_la_tcp_close(struct ipdbg_la_tcp *tcp)
7f4c9a04	142	{
9d2e5483	143	int ret = SR_OK;
7f4c9a04	144
c54ca323	145	#ifdef _WIN32
	146	if (shutdown(tcp->socket, SD_SEND) != SOCKET_ERROR)
	147	{
	148	char recvbuf[16];
	149	int recvbuflen = 16;
	150	// Receive until the peer closes the connection
	151	while(recv(tcp->socket, recvbuf, recvbuflen, 0) > 0);
	152	}
	153	#endif
9d2e5483 SA	154	if (close(tcp->socket) < 0)
9d2e5483 SA	155	ret = SR_ERR;
7f4c9a04	156
9d2e5483	157	tcp->socket = -1;
7f4c9a04	158
9d2e5483	159	return ret;
7f4c9a04 SA	160	}
7f4c9a04 SA	161
4d33f5e1	162	static int tcp_send(struct ipdbg_la_tcp tcp, const uint8_t buf, size_t len)
a4210e18	163	{
9d2e5483	164	int out;
9be587a1	165	out = send(tcp->socket, (const char *)buf, len, 0);
a4210e18	166
9d2e5483 SA	167	if (out < 0) {
	168	sr_err("Send error: %s", g_strerror(errno));
	169	return SR_ERR;
	170	}
a4210e18	171
9d2e5483 SA	172	if (out < (int)len)
9d2e5483 SA	173	sr_dbg("Only sent %d/%d bytes of data.", out, (int)len);
a4210e18	174
9d2e5483	175	return SR_OK;
a4210e18 EK	176	}
a4210e18 EK	177
4d33f5e1	178	static int tcp_receive_blocking(struct ipdbg_la_tcp *tcp,
7f4c9a04	179	uint8_t *buf, int bufsize)
a4210e18	180	{
9d2e5483	181	int received = 0;
ed186484	182	int error_count = 0;
7f4c9a04	183
ed186484 SA	184	/* Timeout after 500ms of not receiving data */
ed186484 SA	185	while ((received < bufsize) && (error_count < 500)) {
8a9788e2	186	if (ipdbg_la_tcp_receive(tcp, buf) > 0) {
9d2e5483 SA	187	buf++;
9d2e5483 SA	188	received++;
ed186484 SA	189	} else {
	190	error_count++;
	191	g_usleep(1000); /* Sleep for 1ms */
9d2e5483 SA	192	}
9d2e5483 SA	193	}
7f4c9a04	194
9d2e5483	195	return received;
a4210e18	196	}
7f4c9a04	197
8a9788e2	198	SR_PRIV int ipdbg_la_tcp_receive(struct ipdbg_la_tcp *tcp,
9d2e5483	199	uint8_t *buf)
b51288e3	200	{
9d2e5483 SA	201	int received = 0;
	202
	203	if (data_available(tcp)) {
	204	while (received < 1) {
9be587a1	205	int len = recv(tcp->socket, (char *)buf, 1, 0);
9d2e5483 SA	206
	207	if (len < 0) {
	208	sr_err("Receive error: %s", g_strerror(errno));
	209	return SR_ERR;
	210	} else
	211	received += len;
	212	}
	213
	214	return received;
	215	} else
	216	return -1;
a4210e18 EK	217	}
a4210e18 EK	218
8a9788e2	219	SR_PRIV int ipdbg_la_convert_trigger(const struct sr_dev_inst *sdi)
a4210e18	220	{
1f15efc1	221	struct dev_context *devc;
9d2e5483 SA	222	struct sr_trigger *trigger;
	223	struct sr_trigger_stage *stage;
	224	struct sr_trigger_match *match;
	225	const GSList l, m;
	226
	227	devc = sdi->priv;
	228
	229	devc->num_stages = 0;
	230	devc->num_transfers = 0;
	231	devc->raw_sample_buf = NULL;
	232
ac3625be	233	for (uint64_t i = 0; i < devc->data_width_bytes; i++) {
9d2e5483 SA	234	devc->trigger_mask[i] = 0;
	235	devc->trigger_value[i] = 0;
	236	devc->trigger_mask_last[i] = 0;
	237	devc->trigger_value_last[i] = 0;
	238	devc->trigger_edge_mask[i] = 0;
	239	}
	240
	241	if (!(trigger = sr_session_trigger_get(sdi->session)))
	242	return SR_OK;
	243
	244	for (l = trigger->stages; l; l = l->next) {
	245	stage = l->data;
	246	for (m = stage->matches; m; m = m->next) {
	247	match = m->data;
	248	int byte_idx = match->channel->index / 8;
	249	uint8_t match_bit = 1 << (match->channel->index % 8);
	250
	251	if (!match->channel->enabled)
	252	/* Ignore disabled channels with a trigger. */
	253	continue;
	254
	255	if (match->match == SR_TRIGGER_ONE) {
	256	devc->trigger_value[byte_idx] \|= match_bit;
	257	devc->trigger_mask[byte_idx] \|= match_bit;
	258	devc->trigger_mask_last[byte_idx] &= ~match_bit;
	259	devc->trigger_edge_mask[byte_idx] &= ~match_bit;
	260	} else if (match->match == SR_TRIGGER_ZERO) {
	261	devc->trigger_value[byte_idx] &= ~match_bit;
	262	devc->trigger_mask[byte_idx] \|= match_bit;
	263	devc->trigger_mask_last[byte_idx] &= ~match_bit;
	264	devc->trigger_edge_mask[byte_idx] &= ~match_bit;
	265	} else if (match->match == SR_TRIGGER_RISING) {
	266	devc->trigger_value[byte_idx] \|= match_bit;
	267	devc->trigger_value_last[byte_idx] &=
	268	~match_bit;
	269	devc->trigger_mask[byte_idx] \|= match_bit;
	270	devc->trigger_mask_last[byte_idx] \|= match_bit;
	271	devc->trigger_edge_mask[byte_idx] &= ~match_bit;
	272	} else if (match->match == SR_TRIGGER_FALLING) {
	273	devc->trigger_value[byte_idx] &= ~match_bit;
	274	devc->trigger_value_last[byte_idx] \|= match_bit;
	275	devc->trigger_mask[byte_idx] \|= match_bit;
	276	devc->trigger_mask_last[byte_idx] \|= match_bit;
	277	devc->trigger_edge_mask[byte_idx] &= ~match_bit;
	278	} else if (match->match == SR_TRIGGER_EDGE) {
	279	devc->trigger_mask[byte_idx] &= ~match_bit;
	280	devc->trigger_mask_last[byte_idx] &= ~match_bit;
	281	devc->trigger_edge_mask[byte_idx] \|= match_bit;
	282	}
	283	}
	284	}
	285
	286	return SR_OK;
a4210e18 EK	287	}
a4210e18 EK	288
8a9788e2	289	SR_PRIV int ipdbg_la_receive_data(int fd, int revents, void *cb_data)
a4210e18	290	{
9d2e5483	291	const struct sr_dev_inst *sdi;
1f15efc1	292	struct dev_context *devc;
9d2e5483 SA	293
	294	(void)fd;
	295	(void)revents;
	296
703fb454	297	sdi = cb_data;
9d2e5483 SA	298	if (!sdi)
	299	return FALSE;
	300
	301	if (!(devc = sdi->priv))
	302	return FALSE;
	303
8a9788e2	304	struct ipdbg_la_tcp *tcp = sdi->conn;
9d2e5483 SA	305	struct sr_datafeed_packet packet;
	306	struct sr_datafeed_logic logic;
	307
	308	if (!devc->raw_sample_buf) {
	309	devc->raw_sample_buf =
ac3625be	310	g_try_malloc(devc->limit_samples * devc->data_width_bytes);
9d2e5483	311	if (!devc->raw_sample_buf) {
fac36d0a	312	sr_err("Sample buffer malloc failed.");
9d2e5483 SA	313	return FALSE;
	314	}
	315	}
	316
	317	if (devc->num_transfers <
ac3625be	318	(devc->limit_samples_max * devc->data_width_bytes)) {
9d2e5483 SA	319	uint8_t byte;
9d2e5483 SA	320
8a9788e2	321	if (ipdbg_la_tcp_receive(tcp, &byte) == 1) {
9d2e5483	322	if (devc->num_transfers <
ac3625be	323	(devc->limit_samples * devc->data_width_bytes))
9d2e5483 SA	324	devc->raw_sample_buf[devc->num_transfers] = byte;
	325
	326	devc->num_transfers++;
	327	}
	328	} else {
	329	if (devc->delay_value > 0) {
	330	/* There are pre-trigger samples, send those first. */
	331	packet.type = SR_DF_LOGIC;
	332	packet.payload = &logic;
ac3625be UH	333	logic.length = devc->delay_value * devc->data_width_bytes;
ac3625be UH	334	logic.unitsize = devc->data_width_bytes;
9d2e5483 SA	335	logic.data = devc->raw_sample_buf;
	336	sr_session_send(cb_data, &packet);
	337	}
	338
	339	/* Send the trigger. */
	340	packet.type = SR_DF_TRIGGER;
	341	sr_session_send(cb_data, &packet);
	342
	343	/* Send post-trigger samples. */
	344	packet.type = SR_DF_LOGIC;
	345	packet.payload = &logic;
	346	logic.length = (devc->limit_samples - devc->delay_value) *
ac3625be UH	347	devc->data_width_bytes;
ac3625be UH	348	logic.unitsize = devc->data_width_bytes;
9d2e5483	349	logic.data = devc->raw_sample_buf +
ac3625be	350	(devc->delay_value * devc->data_width_bytes);
9d2e5483 SA	351	sr_session_send(cb_data, &packet);
	352
	353	g_free(devc->raw_sample_buf);
	354	devc->raw_sample_buf = NULL;
	355
8a9788e2	356	ipdbg_la_abort_acquisition(sdi);
9d2e5483 SA	357	}
	358
	359	return TRUE;
a4210e18 EK	360	}
a4210e18 EK	361
4d33f5e1 UH	362	static int send_escaping(struct ipdbg_la_tcp tcp, uint8_t data_to_send,
	363	uint32_t length)
	364	{
	365	uint8_t escape = CMD_ESCAPE;
	366
	367	while (length--) {
	368	uint8_t payload = *data_to_send++;
	369
	370	if (payload == CMD_RESET)
	371	if (tcp_send(tcp, &escape, 1) != SR_OK)
	372	sr_warn("Couldn't send escape");
	373
	374	if (payload == CMD_ESCAPE)
	375	if (tcp_send(tcp, &escape, 1) != SR_OK)
	376	sr_warn("Couldn't send escape");
	377
	378	if (tcp_send(tcp, &payload, 1) != SR_OK)
	379	sr_warn("Couldn't send data");
	380	}
	381
	382	return SR_OK;
	383	}
	384
1f15efc1	385	SR_PRIV int ipdbg_la_send_delay(struct dev_context *devc,
8a9788e2	386	struct ipdbg_la_tcp *tcp)
a4210e18	387	{
9d2e5483	388	devc->delay_value = (devc->limit_samples / 100.0) * devc->capture_ratio;
a4210e18	389
9d2e5483 SA	390	uint8_t buf;
9d2e5483 SA	391	buf = CMD_CFG_LA;
4d33f5e1	392	tcp_send(tcp, &buf, 1);
9d2e5483	393	buf = CMD_LA_DELAY;
4d33f5e1	394	tcp_send(tcp, &buf, 1);
a4210e18	395
9d2e5483 SA	396	uint8_t delay_buf[4] = { devc->delay_value & 0x000000ff,
	397	(devc->delay_value >> 8) & 0x000000ff,
	398	(devc->delay_value >> 16) & 0x000000ff,
	399	(devc->delay_value >> 24) & 0x000000ff
	400	};
a4210e18	401
ac3625be UH	402	for (uint64_t i = 0; i < devc->addr_width_bytes; i++)
ac3625be UH	403	send_escaping(tcp, &(delay_buf[devc->addr_width_bytes - 1 - i]), 1);
a4210e18	404
9d2e5483	405	return SR_OK;
a4210e18 EK	406	}
a4210e18 EK	407
1f15efc1	408	SR_PRIV int ipdbg_la_send_trigger(struct dev_context *devc,
8a9788e2	409	struct ipdbg_la_tcp *tcp)
a4210e18	410	{
9d2e5483 SA	411	uint8_t buf;
	412
	413	/* Mask */
	414	buf = CMD_CFG_TRIGGER;
4d33f5e1	415	tcp_send(tcp, &buf, 1);
9d2e5483	416	buf = CMD_TRIG_MASKS;
4d33f5e1	417	tcp_send(tcp, &buf, 1);
9d2e5483	418	buf = CMD_TRIG_MASK;
4d33f5e1	419	tcp_send(tcp, &buf, 1);
9d2e5483	420
ac3625be	421	for (size_t i = 0; i < devc->data_width_bytes; i++)
9d2e5483	422	send_escaping(tcp,
ac3625be	423	devc->trigger_mask + devc->data_width_bytes - 1 - i, 1);
9d2e5483 SA	424
	425	/* Value */
	426	buf = CMD_CFG_TRIGGER;
4d33f5e1	427	tcp_send(tcp, &buf, 1);
9d2e5483	428	buf = CMD_TRIG_MASKS;
4d33f5e1	429	tcp_send(tcp, &buf, 1);
9d2e5483	430	buf = CMD_TRIG_VALUE;
4d33f5e1	431	tcp_send(tcp, &buf, 1);
9d2e5483	432
ac3625be	433	for (size_t i = 0; i < devc->data_width_bytes; i++)
9d2e5483	434	send_escaping(tcp,
ac3625be	435	devc->trigger_value + devc->data_width_bytes - 1 - i, 1);
9d2e5483 SA	436
	437	/* Mask_last */
	438	buf = CMD_CFG_TRIGGER;
4d33f5e1	439	tcp_send(tcp, &buf, 1);
9d2e5483	440	buf = CMD_TRIG_MASKS_LAST;
4d33f5e1	441	tcp_send(tcp, &buf, 1);
9d2e5483	442	buf = CMD_TRIG_MASK_LAST;
4d33f5e1	443	tcp_send(tcp, &buf, 1);
9d2e5483	444
ac3625be	445	for (size_t i = 0; i < devc->data_width_bytes; i++)
9d2e5483	446	send_escaping(tcp,
ac3625be	447	devc->trigger_mask_last + devc->data_width_bytes - 1 - i, 1);
9d2e5483 SA	448
	449	/* Value_last */
	450	buf = CMD_CFG_TRIGGER;
4d33f5e1	451	tcp_send(tcp, &buf, 1);
9d2e5483	452	buf = CMD_TRIG_MASKS_LAST;
4d33f5e1	453	tcp_send(tcp, &buf, 1);
9d2e5483	454	buf = CMD_TRIG_VALUE_LAST;
4d33f5e1	455	tcp_send(tcp, &buf, 1);
9d2e5483	456
ac3625be	457	for (size_t i = 0; i < devc->data_width_bytes; i++)
9d2e5483	458	send_escaping(tcp,
ac3625be	459	devc->trigger_value_last + devc->data_width_bytes - 1 - i, 1);
9d2e5483 SA	460
	461	/* Edge_mask */
	462	buf = CMD_CFG_TRIGGER;
4d33f5e1	463	tcp_send(tcp, &buf, 1);
9d2e5483	464	buf = CMD_TRIG_SELECT_EDGE_MASK;
4d33f5e1	465	tcp_send(tcp, &buf, 1);
9d2e5483	466	buf = CMD_TRIG_SET_EDGE_MASK;
4d33f5e1	467	tcp_send(tcp, &buf, 1);
9d2e5483	468
ac3625be	469	for (size_t i = 0; i < devc->data_width_bytes; i++)
9d2e5483	470	send_escaping(tcp,
ac3625be	471	devc->trigger_edge_mask + devc->data_width_bytes - 1 - i, 1);
9d2e5483 SA	472
9d2e5483 SA	473	return SR_OK;
a4210e18 EK	474	}
a4210e18 EK	475
8a9788e2	476	SR_PRIV void ipdbg_la_get_addrwidth_and_datawidth(
1f15efc1	477	struct ipdbg_la_tcp tcp, struct dev_context devc)
a4210e18	478	{
9d2e5483 SA	479	uint8_t buf[8];
9d2e5483 SA	480	uint8_t read_cmd = CMD_GET_BUS_WIDTHS;
a4210e18	481
4d33f5e1	482	if (tcp_send(tcp, &read_cmd, 1) != SR_OK)
9d2e5483	483	sr_warn("Can't send read command");
a4210e18	484
4d33f5e1	485	if (tcp_receive_blocking(tcp, buf, 8) != 8)
9d2e5483	486	sr_warn("Can't get address and data width from device");
b51288e3	487
ac3625be UH	488	devc->data_width = buf[0] & 0x000000FF;
	489	devc->data_width \|= (buf[1] << 8) & 0x0000FF00;
	490	devc->data_width \|= (buf[2] << 16) & 0x00FF0000;
	491	devc->data_width \|= (buf[3] << 24) & 0xFF000000;
a4210e18	492
ac3625be UH	493	devc->addr_width = buf[4] & 0x000000FF;
	494	devc->addr_width \|= (buf[5] << 8) & 0x0000FF00;
	495	devc->addr_width \|= (buf[6] << 16) & 0x00FF0000;
	496	devc->addr_width \|= (buf[7] << 24) & 0xFF000000;
a4210e18	497
ac3625be	498	const uint8_t host_word_size = 8;
a4210e18	499
ac3625be UH	500	devc->data_width_bytes =
	501	(devc->data_width + host_word_size - 1) / host_word_size;
	502	devc->addr_width_bytes =
	503	(devc->addr_width + host_word_size - 1) / host_word_size;
a4210e18	504
ac3625be	505	devc->limit_samples_max = (0x01 << devc->addr_width);
9d2e5483 SA	506	devc->limit_samples = devc->limit_samples_max;
9d2e5483 SA	507
ac3625be UH	508	devc->trigger_mask = g_malloc0(devc->data_width_bytes);
	509	devc->trigger_value = g_malloc0(devc->data_width_bytes);
	510	devc->trigger_mask_last = g_malloc0(devc->data_width_bytes);
	511	devc->trigger_value_last = g_malloc0(devc->data_width_bytes);
	512	devc->trigger_edge_mask = g_malloc0(devc->data_width_bytes);
a4210e18 EK	513	}
a4210e18 EK	514
1f15efc1	515	SR_PRIV struct dev_context *ipdbg_la_dev_new(void)
a4210e18	516	{
1f15efc1	517	struct dev_context *devc;
a4210e18	518
1f15efc1	519	devc = g_malloc0(sizeof(struct dev_context));
9d2e5483	520	devc->capture_ratio = 50;
a4210e18	521
9d2e5483	522	return devc;
a4210e18 EK	523	}
a4210e18 EK	524
8a9788e2	525	SR_PRIV int ipdbg_la_send_reset(struct ipdbg_la_tcp *tcp)
a4210e18	526	{
9d2e5483	527	uint8_t buf = CMD_RESET;
4d33f5e1	528	if (tcp_send(tcp, &buf, 1) != SR_OK)
9d2e5483	529	sr_warn("Couldn't send reset");
7f4c9a04	530
9d2e5483	531	return SR_OK;
a4210e18 EK	532	}
a4210e18 EK	533
8a9788e2	534	SR_PRIV int ipdbg_la_request_id(struct ipdbg_la_tcp *tcp)
a4210e18	535	{
9d2e5483	536	uint8_t buf = CMD_GET_LA_ID;
4d33f5e1	537	if (tcp_send(tcp, &buf, 1) != SR_OK)
9d2e5483	538	sr_warn("Couldn't send ID request");
a4210e18	539
9d2e5483	540	char id[4];
4d33f5e1	541	if (tcp_receive_blocking(tcp, (uint8_t *)id, 4) != 4) {
ed186484 SA	542	sr_err("Couldn't read device ID");
	543	return SR_ERR;
	544	}
a4210e18	545
9d2e5483 SA	546	if (strncmp(id, "IDBG", 4)) {
	547	sr_err("Invalid device ID: expected 'IDBG', got '%c%c%c%c'.",
	548	id[0], id[1], id[2], id[3]);
	549	return SR_ERR;
	550	}
a4210e18	551
9d2e5483	552	return SR_OK;
a4210e18 EK	553	}
a4210e18 EK	554
8a9788e2	555	SR_PRIV void ipdbg_la_abort_acquisition(const struct sr_dev_inst *sdi)
a4210e18	556	{
8a9788e2	557	struct ipdbg_la_tcp *tcp = sdi->conn;
a4210e18	558
9d2e5483	559	sr_session_source_remove(sdi->session, tcp->socket);
a4210e18	560
9d2e5483	561	std_session_send_df_end(sdi);
a4210e18 EK	562	}
a4210e18 EK	563
8a9788e2	564	SR_PRIV int ipdbg_la_send_start(struct ipdbg_la_tcp *tcp)
a4210e18	565	{
9d2e5483	566	uint8_t buf = CMD_START;
a4210e18	567
4d33f5e1	568	if (tcp_send(tcp, &buf, 1) != SR_OK)
9d2e5483	569	sr_warn("Couldn't send start");
7f4c9a04	570
9d2e5483	571	return SR_OK;
a4210e18	572	}