[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>
#endif

#include <string.h>
#include <unistd.h>

#ifndef _WIN32
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#endif

#include <errno.h>
#include "protocol.h"

#include <sys/ioctl.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

SR_PRIV gboolean data_available(struct ipdbg_la_tcp *tcp)
{
#ifdef __WIN32__
	ioctlsocket(tcp->socket, FIONREAD, &bytes_available);
#else
	int status;

	if (ioctl(tcp->socket, FIONREAD, &status) < 0) {	// TIOCMGET
		sr_err("FIONREAD failed: %s\n", g_strerror(errno));
		return FALSE;
	}

	return (status < 1) ? FALSE : TRUE;
#endif  // __WIN32__
}

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;

	if (close(tcp->socket) < 0)
		ret = SR_ERR;

	tcp->socket = -1;

	return ret;
}

SR_PRIV int ipdbg_la_tcp_send(struct ipdbg_la_tcp *tcp,
	const uint8_t *buf, size_t len)
{
	int out;
	out = send(tcp->socket, (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;
}

SR_PRIV int ipdbg_la_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, 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 = (const struct sr_dev_inst *)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;
}

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;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_LA_DELAY;
	ipdbg_la_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;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASKS;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASK;
	ipdbg_la_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;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASKS;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_VALUE;
	ipdbg_la_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;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASKS_LAST;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASK_LAST;
	ipdbg_la_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;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_MASKS_LAST;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_VALUE_LAST;
	ipdbg_la_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;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_SELECT_EDGE_MASK;
	ipdbg_la_tcp_send(tcp, &buf, 1);
	buf = CMD_TRIG_SET_EDGE_MASK;
	ipdbg_la_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 int send_escaping(struct ipdbg_la_tcp *tcp, uint8_t *dataToSend,
	uint32_t length)
{
	uint8_t escape = CMD_ESCAPE;

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

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

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

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

	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 (ipdbg_la_tcp_send(tcp, &read_cmd, 1) != SR_OK)
		sr_warn("Can't send read command");

	if (ipdbg_la_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;

	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 (ipdbg_la_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 (ipdbg_la_tcp_send(tcp, &buf, 1) != SR_OK)
		sr_warn("Couldn't send ID request");

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