blob: e9dbedf32680ec05143e63d70cfa335261fafbf2 [file] [log] [blame]
wdenk68ceb292004-08-02 21:11:11 +00001/*
2 * (C) Copyright 2004
3 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4 *
Wolfgang Denk1a459662013-07-08 09:37:19 +02005 * SPDX-License-Identifier: GPL-2.0+
wdenk68ceb292004-08-02 21:11:11 +00006 */
7
8#include <common.h>
wdenk68ceb292004-08-02 21:11:11 +00009#include <command.h>
Jean-Christophe PLAGNIOL-VILLARD52cb4d42009-05-16 12:14:54 +020010#include <stdio_dev.h>
wdenk68ceb292004-08-02 21:11:11 +000011#include <net.h>
12
Wolfgang Denkd87080b2006-03-31 18:32:53 +020013DECLARE_GLOBAL_DATA_PTR;
14
Joe Hershberger2c8fe512012-07-31 06:09:17 +000015#ifndef CONFIG_NETCONSOLE_BUFFER_SIZE
16#define CONFIG_NETCONSOLE_BUFFER_SIZE 512
17#endif
18
19static char input_buffer[CONFIG_NETCONSOLE_BUFFER_SIZE];
Joe Hershbergere1902ac2012-05-15 08:59:13 +000020static int input_size; /* char count in input buffer */
21static int input_offset; /* offset to valid chars in input buffer */
22static int input_recursion;
23static int output_recursion;
wdenk68ceb292004-08-02 21:11:11 +000024static int net_timeout;
Joe Hershbergere1902ac2012-05-15 08:59:13 +000025static uchar nc_ether[6]; /* server enet address */
Joe Hershberger049a95a2015-04-08 01:41:01 -050026static struct in_addr nc_ip; /* server ip */
Joe Hershberger7f518982012-07-31 06:06:41 +000027static short nc_out_port; /* target output port */
28static short nc_in_port; /* source input port */
Joe Hershbergere1902ac2012-05-15 08:59:13 +000029static const char *output_packet; /* used by first send udp */
30static int output_packet_len;
Joe Hershbergerf8be7d62012-08-03 10:59:08 +000031/*
32 * Start with a default last protocol.
33 * We are only interested in NETCONS or not.
34 */
35enum proto_t net_loop_last_protocol = BOOTP;
wdenk68ceb292004-08-02 21:11:11 +000036
Luca Ceresoli03eb1292011-04-18 06:19:50 +000037static void nc_wait_arp_handler(uchar *pkt, unsigned dest,
Joe Hershberger049a95a2015-04-08 01:41:01 -050038 struct in_addr sip, unsigned src,
wdenk68ceb292004-08-02 21:11:11 +000039 unsigned len)
40{
Joe Hershberger22f6e992012-05-23 07:59:14 +000041 net_set_state(NETLOOP_SUCCESS); /* got arp reply - quit net loop */
wdenk68ceb292004-08-02 21:11:11 +000042}
43
Joe Hershberger049a95a2015-04-08 01:41:01 -050044static void nc_handler(uchar *pkt, unsigned dest, struct in_addr sip,
45 unsigned src, unsigned len)
wdenk68ceb292004-08-02 21:11:11 +000046{
wdenkeedcd072004-09-08 22:03:11 +000047 if (input_size)
Joe Hershberger22f6e992012-05-23 07:59:14 +000048 net_set_state(NETLOOP_SUCCESS); /* got input - quit net loop */
wdenk68ceb292004-08-02 21:11:11 +000049}
50
Joe Hershberger6a38a5f2015-04-08 01:41:16 -050051static void nc_timeout_handler(void)
wdenk68ceb292004-08-02 21:11:11 +000052{
Joe Hershberger22f6e992012-05-23 07:59:14 +000053 net_set_state(NETLOOP_SUCCESS);
wdenk68ceb292004-08-02 21:11:11 +000054}
55
Joe Hershberger049a95a2015-04-08 01:41:01 -050056static int is_broadcast(struct in_addr ip)
Joe Hershbergere827fec2012-09-18 10:01:31 +000057{
Joe Hershberger049a95a2015-04-08 01:41:01 -050058 static struct in_addr netmask;
59 static struct in_addr our_ip;
Joe Hershbergere827fec2012-09-18 10:01:31 +000060 static int env_changed_id;
61 int env_id = get_env_id();
62
63 /* update only when the environment has changed */
64 if (env_changed_id != env_id) {
Simon Glass723806c2017-08-03 12:22:15 -060065 netmask = env_get_ip("netmask");
66 our_ip = env_get_ip("ipaddr");
Joe Hershbergere827fec2012-09-18 10:01:31 +000067
68 env_changed_id = env_id;
69 }
70
Joe Hershberger049a95a2015-04-08 01:41:01 -050071 return (ip.s_addr == ~0 || /* 255.255.255.255 (global bcast) */
72 ((netmask.s_addr & our_ip.s_addr) ==
73 (netmask.s_addr & ip.s_addr) && /* on the same net and */
74 (netmask.s_addr | ip.s_addr) == ~0)); /* bcast to our net */
Joe Hershbergere827fec2012-09-18 10:01:31 +000075}
76
77static int refresh_settings_from_env(void)
78{
79 const char *p;
80 static int env_changed_id;
81 int env_id = get_env_id();
82
83 /* update only when the environment has changed */
84 if (env_changed_id != env_id) {
Simon Glass00caae62017-08-03 12:22:12 -060085 if (env_get("ncip")) {
Simon Glass723806c2017-08-03 12:22:15 -060086 nc_ip = env_get_ip("ncip");
Joe Hershberger049a95a2015-04-08 01:41:01 -050087 if (!nc_ip.s_addr)
Joe Hershbergere827fec2012-09-18 10:01:31 +000088 return -1; /* ncip is 0.0.0.0 */
Simon Glass00caae62017-08-03 12:22:12 -060089 p = strchr(env_get("ncip"), ':');
Joe Hershbergere827fec2012-09-18 10:01:31 +000090 if (p != NULL) {
91 nc_out_port = simple_strtoul(p + 1, NULL, 10);
92 nc_in_port = nc_out_port;
93 }
Joe Hershberger6a38a5f2015-04-08 01:41:16 -050094 } else {
Joe Hershberger049a95a2015-04-08 01:41:01 -050095 nc_ip.s_addr = ~0; /* ncip is not set, so broadcast */
Joe Hershberger6a38a5f2015-04-08 01:41:16 -050096 }
Joe Hershbergere827fec2012-09-18 10:01:31 +000097
Simon Glass00caae62017-08-03 12:22:12 -060098 p = env_get("ncoutport");
Joe Hershbergere827fec2012-09-18 10:01:31 +000099 if (p != NULL)
100 nc_out_port = simple_strtoul(p, NULL, 10);
Simon Glass00caae62017-08-03 12:22:12 -0600101 p = env_get("ncinport");
Joe Hershbergere827fec2012-09-18 10:01:31 +0000102 if (p != NULL)
103 nc_in_port = simple_strtoul(p, NULL, 10);
104
105 if (is_broadcast(nc_ip))
106 /* broadcast MAC address */
107 memset(nc_ether, 0xff, sizeof(nc_ether));
108 else
109 /* force arp request */
110 memset(nc_ether, 0, sizeof(nc_ether));
111 }
112 return 0;
113}
114
115/**
Joe Hershbergerbc0571f2015-04-08 01:41:21 -0500116 * Called from net_loop in net/net.c before each packet
Joe Hershbergere827fec2012-09-18 10:01:31 +0000117 */
Joe Hershberger6a38a5f2015-04-08 01:41:16 -0500118void nc_start(void)
wdenk68ceb292004-08-02 21:11:11 +0000119{
Joe Hershbergere827fec2012-09-18 10:01:31 +0000120 refresh_settings_from_env();
Joe Hershberger0adb5b72015-04-08 01:41:04 -0500121 if (!output_packet_len || memcmp(nc_ether, net_null_ethaddr, 6)) {
wdenk68ceb292004-08-02 21:11:11 +0000122 /* going to check for input packet */
Joe Hershbergerece223b2012-05-23 07:59:15 +0000123 net_set_udp_handler(nc_handler);
Joe Hershbergerbc0571f2015-04-08 01:41:21 -0500124 net_set_timeout_handler(net_timeout, nc_timeout_handler);
wdenk68ceb292004-08-02 21:11:11 +0000125 } else {
126 /* send arp request */
wdenkeedcd072004-09-08 22:03:11 +0000127 uchar *pkt;
Joe Hershbergerece223b2012-05-23 07:59:15 +0000128 net_set_arp_handler(nc_wait_arp_handler);
Joe Hershberger1203fcc2015-04-08 01:41:05 -0500129 pkt = (uchar *)net_tx_packet + net_eth_hdr_size() +
130 IP_UDP_HDR_SIZE;
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000131 memcpy(pkt, output_packet, output_packet_len);
Joe Hershberger1203fcc2015-04-08 01:41:05 -0500132 net_send_udp_packet(nc_ether, nc_ip, nc_out_port, nc_in_port,
133 output_packet_len);
wdenk68ceb292004-08-02 21:11:11 +0000134 }
135}
136
Joe Hershberger049a95a2015-04-08 01:41:01 -0500137int nc_input_packet(uchar *pkt, struct in_addr src_ip, unsigned dest_port,
Joe Hershberger8a0eccb2012-09-18 10:01:32 +0000138 unsigned src_port, unsigned len)
wdenk68ceb292004-08-02 21:11:11 +0000139{
wdenkeedcd072004-09-08 22:03:11 +0000140 int end, chunk;
141
Joe Hershberger8a0eccb2012-09-18 10:01:32 +0000142 if (dest_port != nc_in_port || !len)
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000143 return 0; /* not for us */
wdenk68ceb292004-08-02 21:11:11 +0000144
Joe Hershberger049a95a2015-04-08 01:41:01 -0500145 if (src_ip.s_addr != nc_ip.s_addr && !is_broadcast(nc_ip))
Joe Hershberger8a0eccb2012-09-18 10:01:32 +0000146 return 0; /* not from our client */
147
Joe Hershberger4ef8d532012-05-23 08:01:04 +0000148 debug_cond(DEBUG_DEV_PKT, "input: \"%*.*s\"\n", len, len, pkt);
149
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000150 if (input_size == sizeof(input_buffer))
151 return 1; /* no space */
152 if (len > sizeof(input_buffer) - input_size)
153 len = sizeof(input_buffer) - input_size;
wdenkeedcd072004-09-08 22:03:11 +0000154
155 end = input_offset + input_size;
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000156 if (end > sizeof(input_buffer))
157 end -= sizeof(input_buffer);
wdenkeedcd072004-09-08 22:03:11 +0000158
159 chunk = len;
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000160 if (end + len > sizeof(input_buffer)) {
161 chunk = sizeof(input_buffer) - end;
wdenkeedcd072004-09-08 22:03:11 +0000162 memcpy(input_buffer, pkt + chunk, len - chunk);
163 }
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000164 memcpy(input_buffer + end, pkt, chunk);
wdenkeedcd072004-09-08 22:03:11 +0000165
166 input_size += len;
167
wdenk68ceb292004-08-02 21:11:11 +0000168 return 1;
169}
170
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000171static void nc_send_packet(const char *buf, int len)
wdenk68ceb292004-08-02 21:11:11 +0000172{
Bernhard Nortmannc163e432015-09-14 15:29:44 +0200173#ifdef CONFIG_DM_ETH
174 struct udevice *eth;
175#else
wdenk68ceb292004-08-02 21:11:11 +0000176 struct eth_device *eth;
Bernhard Nortmannc163e432015-09-14 15:29:44 +0200177#endif
wdenk68ceb292004-08-02 21:11:11 +0000178 int inited = 0;
179 uchar *pkt;
wdenkeedcd072004-09-08 22:03:11 +0000180 uchar *ether;
Joe Hershberger049a95a2015-04-08 01:41:01 -0500181 struct in_addr ip;
wdenk68ceb292004-08-02 21:11:11 +0000182
Joe Hershberger4ef8d532012-05-23 08:01:04 +0000183 debug_cond(DEBUG_DEV_PKT, "output: \"%*.*s\"\n", len, len, buf);
184
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000185 eth = eth_get_dev();
186 if (eth == NULL)
wdenk68ceb292004-08-02 21:11:11 +0000187 return;
wdenk68ceb292004-08-02 21:11:11 +0000188
Joe Hershberger0adb5b72015-04-08 01:41:04 -0500189 if (!memcmp(nc_ether, net_null_ethaddr, 6)) {
Bernhard Nortmannc163e432015-09-14 15:29:44 +0200190 if (eth_is_active(eth))
wdenkeedcd072004-09-08 22:03:11 +0000191 return; /* inside net loop */
192 output_packet = buf;
193 output_packet_len = len;
Suriyan Ramasamiefd9bb92013-10-16 09:54:24 -0700194 input_recursion = 1;
Joe Hershbergerbc0571f2015-04-08 01:41:21 -0500195 net_loop(NETCONS); /* wait for arp reply and send packet */
Suriyan Ramasamiefd9bb92013-10-16 09:54:24 -0700196 input_recursion = 0;
wdenkeedcd072004-09-08 22:03:11 +0000197 output_packet_len = 0;
198 return;
199 }
200
Bernhard Nortmannc163e432015-09-14 15:29:44 +0200201 if (!eth_is_active(eth)) {
Joe Hershbergerf8be7d62012-08-03 10:59:08 +0000202 if (eth_is_on_demand_init()) {
Joe Hershbergerd2eaec62015-03-22 17:09:06 -0500203 if (eth_init() < 0)
Joe Hershbergerf8be7d62012-08-03 10:59:08 +0000204 return;
205 eth_set_last_protocol(NETCONS);
Joe Hershberger6a38a5f2015-04-08 01:41:16 -0500206 } else {
Joe Hershbergerd2eaec62015-03-22 17:09:06 -0500207 eth_init_state_only();
Joe Hershberger6a38a5f2015-04-08 01:41:16 -0500208 }
Joe Hershbergerf8be7d62012-08-03 10:59:08 +0000209
wdenk68ceb292004-08-02 21:11:11 +0000210 inited = 1;
211 }
Joe Hershberger1203fcc2015-04-08 01:41:05 -0500212 pkt = (uchar *)net_tx_packet + net_eth_hdr_size() + IP_UDP_HDR_SIZE;
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000213 memcpy(pkt, buf, len);
wdenkeedcd072004-09-08 22:03:11 +0000214 ether = nc_ether;
215 ip = nc_ip;
Joe Hershberger1203fcc2015-04-08 01:41:05 -0500216 net_send_udp_packet(ether, ip, nc_out_port, nc_in_port, len);
wdenk68ceb292004-08-02 21:11:11 +0000217
Joe Hershbergerf8be7d62012-08-03 10:59:08 +0000218 if (inited) {
219 if (eth_is_on_demand_init())
220 eth_halt();
221 else
222 eth_halt_state_only();
223 }
wdenk68ceb292004-08-02 21:11:11 +0000224}
225
Joe Hershberger6a38a5f2015-04-08 01:41:16 -0500226static int nc_stdio_start(struct stdio_dev *dev)
wdenk68ceb292004-08-02 21:11:11 +0000227{
Joe Hershbergere827fec2012-09-18 10:01:31 +0000228 int retval;
wdenk68ceb292004-08-02 21:11:11 +0000229
Joe Hershberger7f518982012-07-31 06:06:41 +0000230 nc_out_port = 6666; /* default port */
231 nc_in_port = nc_out_port;
wdenkeedcd072004-09-08 22:03:11 +0000232
Joe Hershbergere827fec2012-09-18 10:01:31 +0000233 retval = refresh_settings_from_env();
234 if (retval != 0)
235 return retval;
wdenkeedcd072004-09-08 22:03:11 +0000236
Joe Hershbergerd7310c72012-05-23 07:59:23 +0000237 /*
238 * Initialize the static IP settings and buffer pointers
Joe Hershbergerbc0571f2015-04-08 01:41:21 -0500239 * incase we call net_send_udp_packet before net_loop
Joe Hershbergerd7310c72012-05-23 07:59:23 +0000240 */
241 net_init();
242
wdenkeedcd072004-09-08 22:03:11 +0000243 return 0;
wdenk68ceb292004-08-02 21:11:11 +0000244}
245
Joe Hershberger6a38a5f2015-04-08 01:41:16 -0500246static void nc_stdio_putc(struct stdio_dev *dev, char c)
wdenk68ceb292004-08-02 21:11:11 +0000247{
248 if (output_recursion)
249 return;
250 output_recursion = 1;
251
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000252 nc_send_packet(&c, 1);
wdenk68ceb292004-08-02 21:11:11 +0000253
254 output_recursion = 0;
255}
256
Joe Hershberger6a38a5f2015-04-08 01:41:16 -0500257static void nc_stdio_puts(struct stdio_dev *dev, const char *s)
wdenk68ceb292004-08-02 21:11:11 +0000258{
wdenkb2323ea2005-04-20 09:28:54 +0000259 int len;
260
wdenk68ceb292004-08-02 21:11:11 +0000261 if (output_recursion)
262 return;
263 output_recursion = 1;
264
Michael Walle1a9845b2011-10-07 12:27:50 +0000265 len = strlen(s);
266 while (len) {
Masahiro Yamadab4141192014-11-07 03:03:31 +0900267 int send_len = min(len, (int)sizeof(input_buffer));
Michael Walle1a9845b2011-10-07 12:27:50 +0000268 nc_send_packet(s, send_len);
269 len -= send_len;
270 s += send_len;
271 }
wdenk68ceb292004-08-02 21:11:11 +0000272
273 output_recursion = 0;
274}
275
Joe Hershberger6a38a5f2015-04-08 01:41:16 -0500276static int nc_stdio_getc(struct stdio_dev *dev)
wdenk68ceb292004-08-02 21:11:11 +0000277{
wdenkb2323ea2005-04-20 09:28:54 +0000278 uchar c;
279
wdenk68ceb292004-08-02 21:11:11 +0000280 input_recursion = 1;
281
282 net_timeout = 0; /* no timeout */
wdenkeedcd072004-09-08 22:03:11 +0000283 while (!input_size)
Joe Hershbergerbc0571f2015-04-08 01:41:21 -0500284 net_loop(NETCONS);
wdenk68ceb292004-08-02 21:11:11 +0000285
286 input_recursion = 0;
287
wdenkb2323ea2005-04-20 09:28:54 +0000288 c = input_buffer[input_offset++];
289
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000290 if (input_offset >= sizeof(input_buffer))
291 input_offset -= sizeof(input_buffer);
wdenkeedcd072004-09-08 22:03:11 +0000292 input_size--;
wdenk68ceb292004-08-02 21:11:11 +0000293
wdenkeedcd072004-09-08 22:03:11 +0000294 return c;
wdenk68ceb292004-08-02 21:11:11 +0000295}
296
Joe Hershberger6a38a5f2015-04-08 01:41:16 -0500297static int nc_stdio_tstc(struct stdio_dev *dev)
wdenk68ceb292004-08-02 21:11:11 +0000298{
Bernhard Nortmannc163e432015-09-14 15:29:44 +0200299#ifdef CONFIG_DM_ETH
300 struct udevice *eth;
301#else
wdenk68ceb292004-08-02 21:11:11 +0000302 struct eth_device *eth;
Bernhard Nortmannc163e432015-09-14 15:29:44 +0200303#endif
wdenk68ceb292004-08-02 21:11:11 +0000304
305 if (input_recursion)
306 return 0;
307
wdenkeedcd072004-09-08 22:03:11 +0000308 if (input_size)
wdenk68ceb292004-08-02 21:11:11 +0000309 return 1;
310
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000311 eth = eth_get_dev();
Bernhard Nortmannc163e432015-09-14 15:29:44 +0200312 if (eth_is_active(eth))
wdenk68ceb292004-08-02 21:11:11 +0000313 return 0; /* inside net loop */
314
315 input_recursion = 1;
316
317 net_timeout = 1;
Joe Hershbergerbc0571f2015-04-08 01:41:21 -0500318 net_loop(NETCONS); /* kind of poll */
wdenk68ceb292004-08-02 21:11:11 +0000319
320 input_recursion = 0;
321
wdenkeedcd072004-09-08 22:03:11 +0000322 return input_size != 0;
wdenk68ceb292004-08-02 21:11:11 +0000323}
324
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000325int drv_nc_init(void)
wdenk68ceb292004-08-02 21:11:11 +0000326{
Jean-Christophe PLAGNIOL-VILLARD52cb4d42009-05-16 12:14:54 +0200327 struct stdio_dev dev;
wdenk68ceb292004-08-02 21:11:11 +0000328 int rc;
329
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000330 memset(&dev, 0, sizeof(dev));
wdenk68ceb292004-08-02 21:11:11 +0000331
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000332 strcpy(dev.name, "nc");
Bin Meng1caf9342015-11-03 23:23:37 -0800333 dev.flags = DEV_FLAGS_OUTPUT | DEV_FLAGS_INPUT;
Joe Hershberger6a38a5f2015-04-08 01:41:16 -0500334 dev.start = nc_stdio_start;
335 dev.putc = nc_stdio_putc;
336 dev.puts = nc_stdio_puts;
337 dev.getc = nc_stdio_getc;
338 dev.tstc = nc_stdio_tstc;
wdenk68ceb292004-08-02 21:11:11 +0000339
Joe Hershbergere1902ac2012-05-15 08:59:13 +0000340 rc = stdio_register(&dev);
wdenk68ceb292004-08-02 21:11:11 +0000341
342 return (rc == 0) ? 1 : rc;
343}