[people/ms/u-boot.git] / drivers / net / sandbox.c

/*
 * Copyright (c) 2015 National Instruments
 *
 * (C) Copyright 2015
 * Joe Hershberger <joe.hershberger@ni.com>
 *
 * SPDX-License-Identifier:	GPL-2.0
 */

#include <common.h>
#include <dm.h>
#include <malloc.h>
#include <net.h>
#include <asm/test.h>

DECLARE_GLOBAL_DATA_PTR;

/**
 * struct eth_sandbox_priv - memory for sandbox mock driver
 *
 * fake_host_hwaddr: MAC address of mocked machine
 * fake_host_ipaddr: IP address of mocked machine
 * recv_packet_buffer: buffer of the packet returned as received
 * recv_packet_length: length of the packet returned as received
 */
struct eth_sandbox_priv {
	uchar fake_host_hwaddr[ARP_HLEN];
	struct in_addr fake_host_ipaddr;
	uchar *recv_packet_buffer;
	int recv_packet_length;
};

static bool disabled[8] = {false};
static bool skip_timeout;

/*
 * sandbox_eth_disable_response()
 *
 * index - The alias index (also DM seq number)
 * disable - If non-zero, ignore sent packets and don't send mock response
 */
void sandbox_eth_disable_response(int index, bool disable)
{
	disabled[index] = disable;
}

/*
 * sandbox_eth_skip_timeout()
 *
 * When the first packet read is attempted, fast-forward time
 */
void sandbox_eth_skip_timeout(void)
{
	skip_timeout = true;
}

static int sb_eth_start(struct udevice *dev)
{
	struct eth_sandbox_priv *priv = dev_get_priv(dev);

	debug("eth_sandbox: Start\n");

	fdtdec_get_byte_array(gd->fdt_blob, dev_of_offset(dev),
			      "fake-host-hwaddr", priv->fake_host_hwaddr,
			      ARP_HLEN);
	priv->recv_packet_buffer = net_rx_packets[0];
	return 0;
}

static int sb_eth_send(struct udevice *dev, void *packet, int length)
{
	struct eth_sandbox_priv *priv = dev_get_priv(dev);
	struct ethernet_hdr *eth = packet;

	debug("eth_sandbox: Send packet %d\n", length);

	if (dev->seq >= 0 && dev->seq < ARRAY_SIZE(disabled) &&
	    disabled[dev->seq])
		return 0;

	if (ntohs(eth->et_protlen) == PROT_ARP) {
		struct arp_hdr *arp = packet + ETHER_HDR_SIZE;

		if (ntohs(arp->ar_op) == ARPOP_REQUEST) {
			struct ethernet_hdr *eth_recv;
			struct arp_hdr *arp_recv;

			/* store this as the assumed IP of the fake host */
			priv->fake_host_ipaddr = net_read_ip(&arp->ar_tpa);
			/* Formulate a fake response */
			eth_recv = (void *)priv->recv_packet_buffer;
			memcpy(eth_recv->et_dest, eth->et_src, ARP_HLEN);
			memcpy(eth_recv->et_src, priv->fake_host_hwaddr,
			       ARP_HLEN);
			eth_recv->et_protlen = htons(PROT_ARP);

			arp_recv = (void *)priv->recv_packet_buffer +
				ETHER_HDR_SIZE;
			arp_recv->ar_hrd = htons(ARP_ETHER);
			arp_recv->ar_pro = htons(PROT_IP);
			arp_recv->ar_hln = ARP_HLEN;
			arp_recv->ar_pln = ARP_PLEN;
			arp_recv->ar_op = htons(ARPOP_REPLY);
			memcpy(&arp_recv->ar_sha, priv->fake_host_hwaddr,
			       ARP_HLEN);
			net_write_ip(&arp_recv->ar_spa, priv->fake_host_ipaddr);
			memcpy(&arp_recv->ar_tha, &arp->ar_sha, ARP_HLEN);
			net_copy_ip(&arp_recv->ar_tpa, &arp->ar_spa);

			priv->recv_packet_length = ETHER_HDR_SIZE +
				ARP_HDR_SIZE;
		}
	} else if (ntohs(eth->et_protlen) == PROT_IP) {
		struct ip_udp_hdr *ip = packet + ETHER_HDR_SIZE;

		if (ip->ip_p == IPPROTO_ICMP) {
			struct icmp_hdr *icmp = (struct icmp_hdr *)&ip->udp_src;

			if (icmp->type == ICMP_ECHO_REQUEST) {
				struct ethernet_hdr *eth_recv;
				struct ip_udp_hdr *ipr;
				struct icmp_hdr *icmpr;

				/* reply to the ping */
				memcpy(priv->recv_packet_buffer, packet,
				       length);
				eth_recv = (void *)priv->recv_packet_buffer;
				ipr = (void *)priv->recv_packet_buffer +
					ETHER_HDR_SIZE;
				icmpr = (struct icmp_hdr *)&ipr->udp_src;
				memcpy(eth_recv->et_dest, eth->et_src,
				       ARP_HLEN);
				memcpy(eth_recv->et_src, priv->fake_host_hwaddr,
				       ARP_HLEN);
				ipr->ip_sum = 0;
				ipr->ip_off = 0;
				net_copy_ip((void *)&ipr->ip_dst, &ip->ip_src);
				net_write_ip((void *)&ipr->ip_src,
					     priv->fake_host_ipaddr);
				ipr->ip_sum = compute_ip_checksum(ipr,
					IP_HDR_SIZE);

				icmpr->type = ICMP_ECHO_REPLY;
				icmpr->checksum = 0;
				icmpr->checksum = compute_ip_checksum(icmpr,
					ICMP_HDR_SIZE);

				priv->recv_packet_length = length;
			}
		}
	}

	return 0;
}

static int sb_eth_recv(struct udevice *dev, int flags, uchar **packetp)
{
	struct eth_sandbox_priv *priv = dev_get_priv(dev);

	if (skip_timeout) {
		sandbox_timer_add_offset(11000UL);
		skip_timeout = false;
	}

	if (priv->recv_packet_length) {
		int lcl_recv_packet_length = priv->recv_packet_length;

		debug("eth_sandbox: received packet %d\n",
		      priv->recv_packet_length);
		priv->recv_packet_length = 0;
		*packetp = priv->recv_packet_buffer;
		return lcl_recv_packet_length;
	}
	return 0;
}

static void sb_eth_stop(struct udevice *dev)
{
	debug("eth_sandbox: Stop\n");
}

static int sb_eth_write_hwaddr(struct udevice *dev)
{
	struct eth_pdata *pdata = dev_get_platdata(dev);

	debug("eth_sandbox %s: Write HW ADDR - %pM\n", dev->name,
	      pdata->enetaddr);
	return 0;
}

static const struct eth_ops sb_eth_ops = {
	.start			= sb_eth_start,
	.send			= sb_eth_send,
	.recv			= sb_eth_recv,
	.stop			= sb_eth_stop,
	.write_hwaddr		= sb_eth_write_hwaddr,
};

static int sb_eth_remove(struct udevice *dev)
{
	return 0;
}

static int sb_eth_ofdata_to_platdata(struct udevice *dev)
{
	struct eth_pdata *pdata = dev_get_platdata(dev);

	pdata->iobase = devfdt_get_addr(dev);
	return 0;
}

static const struct udevice_id sb_eth_ids[] = {
	{ .compatible = "sandbox,eth" },
	{ }
};

U_BOOT_DRIVER(eth_sandbox) = {
	.name	= "eth_sandbox",
	.id	= UCLASS_ETH,
	.of_match = sb_eth_ids,
	.ofdata_to_platdata = sb_eth_ofdata_to_platdata,
	.remove	= sb_eth_remove,
	.ops	= &sb_eth_ops,
	.priv_auto_alloc_size = sizeof(struct eth_sandbox_priv),
	.platdata_auto_alloc_size = sizeof(struct eth_pdata),
};
Commit	Line	Data
3ea143ab JH	1	/*
	2	* Copyright (c) 2015 National Instruments
	3	*
	4	* (C) Copyright 2015
	5	* Joe Hershberger <joe.hershberger@ni.com>
	6	*
	7	* SPDX-License-Identifier: GPL-2.0
	8	*/
	9
	10	#include <common.h>
	11	#include <dm.h>
	12	#include <malloc.h>
	13	#include <net.h>
6f2707c6	14	#include <asm/test.h>
3ea143ab JH	15
	16	DECLARE_GLOBAL_DATA_PTR;
	17
d87a457b JH	18	/**
	19	* struct eth_sandbox_priv - memory for sandbox mock driver
	20	*
	21	* fake_host_hwaddr: MAC address of mocked machine
	22	* fake_host_ipaddr: IP address of mocked machine
	23	* recv_packet_buffer: buffer of the packet returned as received
	24	* recv_packet_length: length of the packet returned as received
	25	*/
	26	struct eth_sandbox_priv {
	27	uchar fake_host_hwaddr[ARP_HLEN];
049a95a7	28	struct in_addr fake_host_ipaddr;
d87a457b JH	29	uchar *recv_packet_buffer;
	30	int recv_packet_length;
	31	};
	32
2eede1f3	33	static bool disabled[8] = {false};
6f2707c6	34	static bool skip_timeout;
2eede1f3 JH	35
	36	/*
	37	* sandbox_eth_disable_response()
	38	*
	39	* index - The alias index (also DM seq number)
	40	* disable - If non-zero, ignore sent packets and don't send mock response
	41	*/
	42	void sandbox_eth_disable_response(int index, bool disable)
	43	{
	44	disabled[index] = disable;
	45	}
	46
6f2707c6 JH	47	/*
	48	* sandbox_eth_skip_timeout()
	49	*
	50	* When the first packet read is attempted, fast-forward time
	51	*/
	52	void sandbox_eth_skip_timeout(void)
	53	{
	54	skip_timeout = true;
	55	}
	56
3ea143ab JH	57	static int sb_eth_start(struct udevice *dev)
3ea143ab JH	58	{
d87a457b JH	59	struct eth_sandbox_priv *priv = dev_get_priv(dev);
d87a457b JH	60
3ea143ab JH	61	debug("eth_sandbox: Start\n");
3ea143ab JH	62
e160f7d4 SG	63	fdtdec_get_byte_array(gd->fdt_blob, dev_of_offset(dev),
	64	"fake-host-hwaddr", priv->fake_host_hwaddr,
	65	ARP_HLEN);
d87a457b	66	priv->recv_packet_buffer = net_rx_packets[0];
3ea143ab JH	67	return 0;
	68	}
	69
	70	static int sb_eth_send(struct udevice dev, void packet, int length)
	71	{
d87a457b JH	72	struct eth_sandbox_priv *priv = dev_get_priv(dev);
	73	struct ethernet_hdr *eth = packet;
	74
3ea143ab JH	75	debug("eth_sandbox: Send packet %d\n", length);
3ea143ab JH	76
2eede1f3 JH	77	if (dev->seq >= 0 && dev->seq < ARRAY_SIZE(disabled) &&
	78	disabled[dev->seq])
	79	return 0;
	80
d87a457b JH	81	if (ntohs(eth->et_protlen) == PROT_ARP) {
	82	struct arp_hdr *arp = packet + ETHER_HDR_SIZE;
	83
	84	if (ntohs(arp->ar_op) == ARPOP_REQUEST) {
	85	struct ethernet_hdr *eth_recv;
	86	struct arp_hdr *arp_recv;
	87
	88	/* store this as the assumed IP of the fake host */
049a95a7	89	priv->fake_host_ipaddr = net_read_ip(&arp->ar_tpa);
d87a457b JH	90	/* Formulate a fake response */
	91	eth_recv = (void *)priv->recv_packet_buffer;
	92	memcpy(eth_recv->et_dest, eth->et_src, ARP_HLEN);
	93	memcpy(eth_recv->et_src, priv->fake_host_hwaddr,
	94	ARP_HLEN);
	95	eth_recv->et_protlen = htons(PROT_ARP);
	96
	97	arp_recv = (void *)priv->recv_packet_buffer +
	98	ETHER_HDR_SIZE;
	99	arp_recv->ar_hrd = htons(ARP_ETHER);
	100	arp_recv->ar_pro = htons(PROT_IP);
	101	arp_recv->ar_hln = ARP_HLEN;
	102	arp_recv->ar_pln = ARP_PLEN;
	103	arp_recv->ar_op = htons(ARPOP_REPLY);
	104	memcpy(&arp_recv->ar_sha, priv->fake_host_hwaddr,
	105	ARP_HLEN);
049a95a7	106	net_write_ip(&arp_recv->ar_spa, priv->fake_host_ipaddr);
d87a457b	107	memcpy(&arp_recv->ar_tha, &arp->ar_sha, ARP_HLEN);
049a95a7	108	net_copy_ip(&arp_recv->ar_tpa, &arp->ar_spa);
d87a457b JH	109
	110	priv->recv_packet_length = ETHER_HDR_SIZE +
	111	ARP_HDR_SIZE;
	112	}
	113	} else if (ntohs(eth->et_protlen) == PROT_IP) {
	114	struct ip_udp_hdr *ip = packet + ETHER_HDR_SIZE;
	115
	116	if (ip->ip_p == IPPROTO_ICMP) {
	117	struct icmp_hdr icmp = (struct icmp_hdr )&ip->udp_src;
	118
	119	if (icmp->type == ICMP_ECHO_REQUEST) {
	120	struct ethernet_hdr *eth_recv;
	121	struct ip_udp_hdr *ipr;
	122	struct icmp_hdr *icmpr;
	123
	124	/* reply to the ping */
	125	memcpy(priv->recv_packet_buffer, packet,
	126	length);
	127	eth_recv = (void *)priv->recv_packet_buffer;
	128	ipr = (void *)priv->recv_packet_buffer +
	129	ETHER_HDR_SIZE;
	130	icmpr = (struct icmp_hdr *)&ipr->udp_src;
	131	memcpy(eth_recv->et_dest, eth->et_src,
	132	ARP_HLEN);
	133	memcpy(eth_recv->et_src, priv->fake_host_hwaddr,
	134	ARP_HLEN);
	135	ipr->ip_sum = 0;
	136	ipr->ip_off = 0;
049a95a7 JH	137	net_copy_ip((void *)&ipr->ip_dst, &ip->ip_src);
	138	net_write_ip((void *)&ipr->ip_src,
	139	priv->fake_host_ipaddr);
d87a457b JH	140	ipr->ip_sum = compute_ip_checksum(ipr,
	141	IP_HDR_SIZE);
	142
	143	icmpr->type = ICMP_ECHO_REPLY;
	144	icmpr->checksum = 0;
	145	icmpr->checksum = compute_ip_checksum(icmpr,
	146	ICMP_HDR_SIZE);
	147
	148	priv->recv_packet_length = length;
	149	}
	150	}
	151	}
	152
3ea143ab JH	153	return 0;
	154	}
	155
a1ca92ea	156	static int sb_eth_recv(struct udevice dev, int flags, uchar *packetp)
3ea143ab	157	{
d87a457b JH	158	struct eth_sandbox_priv *priv = dev_get_priv(dev);
d87a457b JH	159
6f2707c6	160	if (skip_timeout) {
c5a75339	161	sandbox_timer_add_offset(11000UL);
6f2707c6 JH	162	skip_timeout = false;
	163	}
	164
d87a457b JH	165	if (priv->recv_packet_length) {
	166	int lcl_recv_packet_length = priv->recv_packet_length;
	167
	168	debug("eth_sandbox: received packet %d\n",
	169	priv->recv_packet_length);
	170	priv->recv_packet_length = 0;
	171	*packetp = priv->recv_packet_buffer;
	172	return lcl_recv_packet_length;
	173	}
3ea143ab JH	174	return 0;
	175	}
	176
	177	static void sb_eth_stop(struct udevice *dev)
	178	{
	179	debug("eth_sandbox: Stop\n");
	180	}
	181
	182	static int sb_eth_write_hwaddr(struct udevice *dev)
	183	{
	184	struct eth_pdata *pdata = dev_get_platdata(dev);
	185
	186	debug("eth_sandbox %s: Write HW ADDR - %pM\n", dev->name,
	187	pdata->enetaddr);
	188	return 0;
	189	}
	190
	191	static const struct eth_ops sb_eth_ops = {
	192	.start = sb_eth_start,
	193	.send = sb_eth_send,
	194	.recv = sb_eth_recv,
	195	.stop = sb_eth_stop,
	196	.write_hwaddr = sb_eth_write_hwaddr,
	197	};
	198
	199	static int sb_eth_remove(struct udevice *dev)
	200	{
	201	return 0;
	202	}
	203
	204	static int sb_eth_ofdata_to_platdata(struct udevice *dev)
	205	{
	206	struct eth_pdata *pdata = dev_get_platdata(dev);
	207
a821c4af	208	pdata->iobase = devfdt_get_addr(dev);
3ea143ab JH	209	return 0;
	210	}
	211
	212	static const struct udevice_id sb_eth_ids[] = {
	213	{ .compatible = "sandbox,eth" },
	214	{ }
	215	};
	216
	217	U_BOOT_DRIVER(eth_sandbox) = {
	218	.name = "eth_sandbox",
	219	.id = UCLASS_ETH,
	220	.of_match = sb_eth_ids,
	221	.ofdata_to_platdata = sb_eth_ofdata_to_platdata,
	222	.remove = sb_eth_remove,
	223	.ops = &sb_eth_ops,
d87a457b	224	.priv_auto_alloc_size = sizeof(struct eth_sandbox_priv),
3ea143ab JH	225	.platdata_auto_alloc_size = sizeof(struct eth_pdata),
3ea143ab JH	226	};