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

/*
 * (C) Copyright 2004
 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
 *
 * See file CREDITS for list of people who contributed to this
 * project.
 *
 * 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., 59 Temple Place, Suite 330, Boston,
 * MA 02111-1307 USA
 */

#include <common.h>

#ifdef CONFIG_NETCONSOLE

#include <command.h>
#include <devices.h>
#include <net.h>

static char input_buffer[512];
static int input_size = 0;		/* char count in input buffer */
static int input_offset = 0;		/* offset to valid chars in input buffer */
static int input_recursion = 0;
static int output_recursion = 0;
static int net_timeout;
static uchar nc_ether[6];		/* server enet address */
static IPaddr_t nc_ip;			/* server ip */
static short nc_port;			/* source/target port */
static const char *output_packet;	/* used by first send udp */
static int output_packet_len = 0;

static void nc_wait_arp_handler (uchar * pkt, unsigned dest, unsigned src,
				 unsigned len)
{
	NetState = NETLOOP_SUCCESS;	/* got arp reply - quit net loop */
}

static void nc_handler (uchar * pkt, unsigned dest, unsigned src,
			unsigned len)
{
	if (input_size)
		NetState = NETLOOP_SUCCESS;	/* got input - quit net loop */
}

static void nc_timeout (void)
{
	NetState = NETLOOP_SUCCESS;
}

void NcStart (void)
{
	if (!output_packet_len || memcmp (nc_ether, NetEtherNullAddr, 6)) {
		/* going to check for input packet */
		NetSetHandler (nc_handler);
		NetSetTimeout (net_timeout, nc_timeout);
	} else {
		/* send arp request */
		uchar *pkt;
		NetSetHandler (nc_wait_arp_handler);
		pkt = (uchar *) NetTxPacket + NetEthHdrSize () + IP_HDR_SIZE;
		memcpy (pkt, output_packet, output_packet_len);
		NetSendUDPPacket (nc_ether, nc_ip, nc_port, nc_port, output_packet_len);
	}
}

int nc_input_packet (uchar * pkt, unsigned dest, unsigned src, unsigned len)
{
	int end, chunk;

	if (dest != nc_port || !len)
		return 0;		/* not for us */

	if (input_size == sizeof input_buffer)
		return 1;		/* no space */
	if (len > sizeof input_buffer - input_size)
		len = sizeof input_buffer - input_size;

	end = input_offset + input_size;
	if (end > sizeof input_buffer)
		end -= sizeof input_buffer;

	chunk = len;
	if (end + len > sizeof input_buffer) {
		chunk = sizeof input_buffer - end;
		memcpy(input_buffer, pkt + chunk, len - chunk);
	}
	memcpy (input_buffer + end, pkt, chunk);

	input_size += len;

	return 1;
}

static void nc_send_packet (const char *buf, int len)
{
	DECLARE_GLOBAL_DATA_PTR;

	struct eth_device *eth;
	int inited = 0;
	uchar *pkt;
	uchar *ether;
	IPaddr_t ip;

	if ((eth = eth_get_dev ()) == NULL) {
		return;
	}

	if (!memcmp (nc_ether, NetEtherNullAddr, 6)) {
		if (eth->state == ETH_STATE_ACTIVE)
			return;	/* inside net loop */
		output_packet = buf;
		output_packet_len = len;
		NetLoop (NETCONS);	/* wait for arp reply and send packet */
		output_packet_len = 0;
		return;
	}

	if (eth->state != ETH_STATE_ACTIVE) {
		if (eth_init (gd->bd) < 0)
			return;
		inited = 1;
	}
	pkt = (uchar *) NetTxPacket + NetEthHdrSize () + IP_HDR_SIZE;
	memcpy (pkt, buf, len);
	ether = nc_ether;
	ip = nc_ip;
	NetSendUDPPacket (ether, ip, nc_port, nc_port, len);

	if (inited)
		eth_halt ();
}

int nc_start (void)
{
	int netmask, our_ip;

	nc_port = 6666;		/* default port */

	if (getenv ("ncip")) {
		char *p;

		nc_ip = getenv_IPaddr ("ncip");
		if (!nc_ip)
			return -1;	/* ncip is 0.0.0.0 */
		if ((p = strchr (getenv ("ncip"), ':')) != NULL)
			nc_port = simple_strtoul (p + 1, NULL, 10);
	} else
		nc_ip = ~0;		/* ncip is not set */

	our_ip = getenv_IPaddr ("ipaddr");
	netmask = getenv_IPaddr ("netmask");

	if (nc_ip == ~0 ||				/* 255.255.255.255 */
	    ((netmask & our_ip) == (netmask & nc_ip) &&	/* on the same net */
	    (netmask | nc_ip) == ~0))			/* broadcast to our net */
		memset (nc_ether, 0xff, sizeof nc_ether);
	else
		memset (nc_ether, 0, sizeof nc_ether);	/* force arp request */

	return 0;
}

void nc_putc (char c)
{
	if (output_recursion)
		return;
	output_recursion = 1;

	nc_send_packet (&c, 1);

	output_recursion = 0;
}

void nc_puts (const char *s)
{
	int len;

	if (output_recursion)
		return;
	output_recursion = 1;

	if ((len = strlen (s)) > 512)
		len = 512;

	nc_send_packet (s, len);

	output_recursion = 0;
}

int nc_getc (void)
{
	uchar c;

	input_recursion = 1;

	net_timeout = 0;	/* no timeout */
	while (!input_size)
		NetLoop (NETCONS);

	input_recursion = 0;

	c = input_buffer[input_offset++];

	if (input_offset >= sizeof input_buffer)
		input_offset -= sizeof input_buffer;
	input_size--;

	return c;
}

int nc_tstc (void)
{
	struct eth_device *eth;

	if (input_recursion)
		return 0;

	if (input_size)
		return 1;

	eth = eth_get_dev ();
	if (eth && eth->state == ETH_STATE_ACTIVE)
		return 0;	/* inside net loop */

	input_recursion = 1;

	net_timeout = 1;
	NetLoop (NETCONS);	/* kind of poll */

	input_recursion = 0;

	return input_size != 0;
}

int drv_nc_init (void)
{
	device_t dev;
	int rc;

	memset (&dev, 0, sizeof (dev));

	strcpy (dev.name, "nc");
	dev.flags = DEV_FLAGS_OUTPUT | DEV_FLAGS_INPUT | DEV_FLAGS_SYSTEM;
	dev.start = nc_start;
	dev.putc = nc_putc;
	dev.puts = nc_puts;
	dev.getc = nc_getc;
	dev.tstc = nc_tstc;

	rc = device_register (&dev);

	return (rc == 0) ? 1 : rc;
}

#endif	/* CONFIG_NETCONSOLE */
Commit	Line	Data
68ceb29e WD	1	/*
	2	* (C) Copyright 2004
	3	* Wolfgang Denk, DENX Software Engineering, wd@denx.de.
	4	*
	5	* See file CREDITS for list of people who contributed to this
	6	* project.
	7	*
	8	* This program is free software; you can redistribute it and/or
	9	* modify it under the terms of the GNU General Public License as
	10	* published by the Free Software Foundation; either version 2 of
	11	* the License, or (at your option) any later version.
	12	*
	13	* This program is distributed in the hope that it will be useful,
	14	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	16	* GNU General Public License for more details.
	17	*
	18	* You should have received a copy of the GNU General Public License
	19	* along with this program; if not, write to the Free Software
	20	* Foundation, Inc., 59 Temple Place, Suite 330, Boston,
	21	* MA 02111-1307 USA
	22	*/
	23
	24	#include <common.h>
	25
	26	#ifdef CONFIG_NETCONSOLE
	27
	28	#include <command.h>
	29	#include <devices.h>
	30	#include <net.h>
	31
eedcd078 WD	32	static char input_buffer[512];
	33	static int input_size = 0; /* char count in input buffer */
	34	static int input_offset = 0; /* offset to valid chars in input buffer */
68ceb29e WD	35	static int input_recursion = 0;
	36	static int output_recursion = 0;
	37	static int net_timeout;
eedcd078 WD	38	static uchar nc_ether[6]; /* server enet address */
	39	static IPaddr_t nc_ip; /* server ip */
	40	static short nc_port; /* source/target port */
	41	static const char output_packet; / used by first send udp */
	42	static int output_packet_len = 0;
68ceb29e WD	43
	44	static void nc_wait_arp_handler (uchar * pkt, unsigned dest, unsigned src,
	45	unsigned len)
	46	{
	47	NetState = NETLOOP_SUCCESS; /* got arp reply - quit net loop */
	48	}
	49
	50	static void nc_handler (uchar * pkt, unsigned dest, unsigned src,
	51	unsigned len)
	52	{
eedcd078	53	if (input_size)
68ceb29e WD	54	NetState = NETLOOP_SUCCESS; /* got input - quit net loop */
	55	}
	56
	57	static void nc_timeout (void)
	58	{
	59	NetState = NETLOOP_SUCCESS;
	60	}
	61
	62	void NcStart (void)
	63	{
eedcd078	64	if (!output_packet_len \|\| memcmp (nc_ether, NetEtherNullAddr, 6)) {
68ceb29e WD	65	/* going to check for input packet */
	66	NetSetHandler (nc_handler);
	67	NetSetTimeout (net_timeout, nc_timeout);
	68	} else {
	69	/* send arp request */
eedcd078	70	uchar *pkt;
68ceb29e	71	NetSetHandler (nc_wait_arp_handler);
eedcd078 WD	72	pkt = (uchar *) NetTxPacket + NetEthHdrSize () + IP_HDR_SIZE;
	73	memcpy (pkt, output_packet, output_packet_len);
	74	NetSendUDPPacket (nc_ether, nc_ip, nc_port, nc_port, output_packet_len);
68ceb29e WD	75	}
	76	}
	77
	78	int nc_input_packet (uchar * pkt, unsigned dest, unsigned src, unsigned len)
	79	{
eedcd078 WD	80	int end, chunk;
	81
	82	if (dest != nc_port \|\| !len)
68ceb29e WD	83	return 0; /* not for us */
68ceb29e WD	84
eedcd078 WD	85	if (input_size == sizeof input_buffer)
	86	return 1; /* no space */
	87	if (len > sizeof input_buffer - input_size)
	88	len = sizeof input_buffer - input_size;
	89
	90	end = input_offset + input_size;
	91	if (end > sizeof input_buffer)
	92	end -= sizeof input_buffer;
	93
	94	chunk = len;
	95	if (end + len > sizeof input_buffer) {
	96	chunk = sizeof input_buffer - end;
	97	memcpy(input_buffer, pkt + chunk, len - chunk);
	98	}
	99	memcpy (input_buffer + end, pkt, chunk);
	100
	101	input_size += len;
	102
68ceb29e WD	103	return 1;
	104	}
	105
	106	static void nc_send_packet (const char *buf, int len)
	107	{
	108	DECLARE_GLOBAL_DATA_PTR;
	109
	110	struct eth_device *eth;
	111	int inited = 0;
	112	uchar *pkt;
eedcd078 WD	113	uchar *ether;
eedcd078 WD	114	IPaddr_t ip;
68ceb29e	115
eedcd078	116	if ((eth = eth_get_dev ()) == NULL) {
68ceb29e	117	return;
eedcd078	118	}
68ceb29e	119
eedcd078 WD	120	if (!memcmp (nc_ether, NetEtherNullAddr, 6)) {
	121	if (eth->state == ETH_STATE_ACTIVE)
	122	return; /* inside net loop */
	123	output_packet = buf;
	124	output_packet_len = len;
	125	NetLoop (NETCONS); /* wait for arp reply and send packet */
	126	output_packet_len = 0;
68ceb29e WD	127	return;
	128	}
	129
	130	if (eth->state != ETH_STATE_ACTIVE) {
	131	if (eth_init (gd->bd) < 0)
	132	return;
	133	inited = 1;
	134	}
	135	pkt = (uchar *) NetTxPacket + NetEthHdrSize () + IP_HDR_SIZE;
	136	memcpy (pkt, buf, len);
eedcd078 WD	137	ether = nc_ether;
	138	ip = nc_ip;
	139	NetSendUDPPacket (ether, ip, nc_port, nc_port, len);
68ceb29e WD	140
	141	if (inited)
	142	eth_halt ();
	143	}
	144
	145	int nc_start (void)
	146	{
eedcd078 WD	147	int netmask, our_ip;
	148
	149	nc_port = 6666; /* default port */
	150
	151	if (getenv ("ncip")) {
b2323ea6 WD	152	char *p;
b2323ea6 WD	153
eedcd078 WD	154	nc_ip = getenv_IPaddr ("ncip");
	155	if (!nc_ip)
	156	return -1; /* ncip is 0.0.0.0 */
b2323ea6	157	if ((p = strchr (getenv ("ncip"), ':')) != NULL)
eedcd078 WD	158	nc_port = simple_strtoul (p + 1, NULL, 10);
	159	} else
	160	nc_ip = ~0; /* ncip is not set */
	161
	162	our_ip = getenv_IPaddr ("ipaddr");
	163	netmask = getenv_IPaddr ("netmask");
	164
	165	if (nc_ip == ~0 \|\| /* 255.255.255.255 */
	166	((netmask & our_ip) == (netmask & nc_ip) && /* on the same net */
	167	(netmask \| nc_ip) == ~0)) /* broadcast to our net */
	168	memset (nc_ether, 0xff, sizeof nc_ether);
	169	else
	170	memset (nc_ether, 0, sizeof nc_ether); /* force arp request */
	171
	172	return 0;
68ceb29e WD	173	}
	174
	175	void nc_putc (char c)
	176	{
	177	if (output_recursion)
	178	return;
	179	output_recursion = 1;
	180
	181	nc_send_packet (&c, 1);
	182
	183	output_recursion = 0;
	184	}
	185
	186	void nc_puts (const char *s)
	187	{
b2323ea6 WD	188	int len;
b2323ea6 WD	189
68ceb29e WD	190	if (output_recursion)
	191	return;
	192	output_recursion = 1;
	193
b2323ea6	194	if ((len = strlen (s)) > 512)
68ceb29e WD	195	len = 512;
	196
	197	nc_send_packet (s, len);
	198
	199	output_recursion = 0;
	200	}
	201
	202	int nc_getc (void)
	203	{
b2323ea6 WD	204	uchar c;
b2323ea6 WD	205
68ceb29e WD	206	input_recursion = 1;
	207
	208	net_timeout = 0; /* no timeout */
eedcd078	209	while (!input_size)
68ceb29e WD	210	NetLoop (NETCONS);
	211
	212	input_recursion = 0;
	213
b2323ea6 WD	214	c = input_buffer[input_offset++];
b2323ea6 WD	215
eedcd078 WD	216	if (input_offset >= sizeof input_buffer)
	217	input_offset -= sizeof input_buffer;
	218	input_size--;
68ceb29e	219
eedcd078	220	return c;
68ceb29e WD	221	}
	222
	223	int nc_tstc (void)
	224	{
	225	struct eth_device *eth;
	226
	227	if (input_recursion)
	228	return 0;
	229
eedcd078	230	if (input_size)
68ceb29e WD	231	return 1;
	232
	233	eth = eth_get_dev ();
	234	if (eth && eth->state == ETH_STATE_ACTIVE)
	235	return 0; /* inside net loop */
	236
	237	input_recursion = 1;
	238
	239	net_timeout = 1;
eedcd078	240	NetLoop (NETCONS); /* kind of poll */
68ceb29e WD	241
	242	input_recursion = 0;
	243
eedcd078	244	return input_size != 0;
68ceb29e WD	245	}
	246
	247	int drv_nc_init (void)
	248	{
	249	device_t dev;
	250	int rc;
	251
	252	memset (&dev, 0, sizeof (dev));
	253
	254	strcpy (dev.name, "nc");
	255	dev.flags = DEV_FLAGS_OUTPUT \| DEV_FLAGS_INPUT \| DEV_FLAGS_SYSTEM;
	256	dev.start = nc_start;
	257	dev.putc = nc_putc;
	258	dev.puts = nc_puts;
	259	dev.getc = nc_getc;
	260	dev.tstc = nc_tstc;
	261
	262	rc = device_register (&dev);
	263
	264	return (rc == 0) ? 1 : rc;
	265	}
	266
	267	#endif /* CONFIG_NETCONSOLE */