3 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
5 * SPDX-License-Identifier: GPL-2.0+
10 #include <stdio_dev.h>
13 DECLARE_GLOBAL_DATA_PTR
;
15 #ifndef CONFIG_NETCONSOLE_BUFFER_SIZE
16 #define CONFIG_NETCONSOLE_BUFFER_SIZE 512
19 static char input_buffer
[CONFIG_NETCONSOLE_BUFFER_SIZE
];
20 static int input_size
; /* char count in input buffer */
21 static int input_offset
; /* offset to valid chars in input buffer */
22 static int input_recursion
;
23 static int output_recursion
;
24 static int net_timeout
;
25 static uchar nc_ether
[6]; /* server enet address */
26 static struct in_addr nc_ip
; /* server ip */
27 static short nc_out_port
; /* target output port */
28 static short nc_in_port
; /* source input port */
29 static const char *output_packet
; /* used by first send udp */
30 static int output_packet_len
;
32 * Start with a default last protocol.
33 * We are only interested in NETCONS or not.
35 enum proto_t net_loop_last_protocol
= BOOTP
;
37 static void nc_wait_arp_handler(uchar
*pkt
, unsigned dest
,
38 struct in_addr sip
, unsigned src
,
41 net_set_state(NETLOOP_SUCCESS
); /* got arp reply - quit net loop */
44 static void nc_handler(uchar
*pkt
, unsigned dest
, struct in_addr sip
,
45 unsigned src
, unsigned len
)
48 net_set_state(NETLOOP_SUCCESS
); /* got input - quit net loop */
51 static void nc_timeout(void)
53 net_set_state(NETLOOP_SUCCESS
);
56 static int is_broadcast(struct in_addr ip
)
58 static struct in_addr netmask
;
59 static struct in_addr our_ip
;
60 static int env_changed_id
;
61 int env_id
= get_env_id();
63 /* update only when the environment has changed */
64 if (env_changed_id
!= env_id
) {
65 netmask
= getenv_ip("netmask");
66 our_ip
= getenv_ip("ipaddr");
68 env_changed_id
= env_id
;
71 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 */
77 static int refresh_settings_from_env(void)
80 static int env_changed_id
;
81 int env_id
= get_env_id();
83 /* update only when the environment has changed */
84 if (env_changed_id
!= env_id
) {
86 nc_ip
= getenv_ip("ncip");
88 return -1; /* ncip is 0.0.0.0 */
89 p
= strchr(getenv("ncip"), ':');
91 nc_out_port
= simple_strtoul(p
+ 1, NULL
, 10);
92 nc_in_port
= nc_out_port
;
95 nc_ip
.s_addr
= ~0; /* ncip is not set, so broadcast */
97 p
= getenv("ncoutport");
99 nc_out_port
= simple_strtoul(p
, NULL
, 10);
100 p
= getenv("ncinport");
102 nc_in_port
= simple_strtoul(p
, NULL
, 10);
104 if (is_broadcast(nc_ip
))
105 /* broadcast MAC address */
106 memset(nc_ether
, 0xff, sizeof(nc_ether
));
108 /* force arp request */
109 memset(nc_ether
, 0, sizeof(nc_ether
));
115 * Called from NetLoop in net/net.c before each packet
119 refresh_settings_from_env();
120 if (!output_packet_len
|| memcmp(nc_ether
, net_null_ethaddr
, 6)) {
121 /* going to check for input packet */
122 net_set_udp_handler(nc_handler
);
123 NetSetTimeout(net_timeout
, nc_timeout
);
125 /* send arp request */
127 net_set_arp_handler(nc_wait_arp_handler
);
128 pkt
= (uchar
*)net_tx_packet
+ net_eth_hdr_size() +
130 memcpy(pkt
, output_packet
, output_packet_len
);
131 net_send_udp_packet(nc_ether
, nc_ip
, nc_out_port
, nc_in_port
,
136 int nc_input_packet(uchar
*pkt
, struct in_addr src_ip
, unsigned dest_port
,
137 unsigned src_port
, unsigned len
)
141 if (dest_port
!= nc_in_port
|| !len
)
142 return 0; /* not for us */
144 if (src_ip
.s_addr
!= nc_ip
.s_addr
&& !is_broadcast(nc_ip
))
145 return 0; /* not from our client */
147 debug_cond(DEBUG_DEV_PKT
, "input: \"%*.*s\"\n", len
, len
, pkt
);
149 if (input_size
== sizeof(input_buffer
))
150 return 1; /* no space */
151 if (len
> sizeof(input_buffer
) - input_size
)
152 len
= sizeof(input_buffer
) - input_size
;
154 end
= input_offset
+ input_size
;
155 if (end
> sizeof(input_buffer
))
156 end
-= sizeof(input_buffer
);
159 if (end
+ len
> sizeof(input_buffer
)) {
160 chunk
= sizeof(input_buffer
) - end
;
161 memcpy(input_buffer
, pkt
+ chunk
, len
- chunk
);
163 memcpy(input_buffer
+ end
, pkt
, chunk
);
170 static void nc_send_packet(const char *buf
, int len
)
172 struct eth_device
*eth
;
178 debug_cond(DEBUG_DEV_PKT
, "output: \"%*.*s\"\n", len
, len
, buf
);
184 if (!memcmp(nc_ether
, net_null_ethaddr
, 6)) {
185 if (eth
->state
== ETH_STATE_ACTIVE
)
186 return; /* inside net loop */
188 output_packet_len
= len
;
190 NetLoop(NETCONS
); /* wait for arp reply and send packet */
192 output_packet_len
= 0;
196 if (eth
->state
!= ETH_STATE_ACTIVE
) {
197 if (eth_is_on_demand_init()) {
200 eth_set_last_protocol(NETCONS
);
202 eth_init_state_only();
206 pkt
= (uchar
*)net_tx_packet
+ net_eth_hdr_size() + IP_UDP_HDR_SIZE
;
207 memcpy(pkt
, buf
, len
);
210 net_send_udp_packet(ether
, ip
, nc_out_port
, nc_in_port
, len
);
213 if (eth_is_on_demand_init())
216 eth_halt_state_only();
220 static int nc_start(struct stdio_dev
*dev
)
224 nc_out_port
= 6666; /* default port */
225 nc_in_port
= nc_out_port
;
227 retval
= refresh_settings_from_env();
232 * Initialize the static IP settings and buffer pointers
233 * incase we call net_send_udp_packet before NetLoop
240 static void nc_putc(struct stdio_dev
*dev
, char c
)
242 if (output_recursion
)
244 output_recursion
= 1;
246 nc_send_packet(&c
, 1);
248 output_recursion
= 0;
251 static void nc_puts(struct stdio_dev
*dev
, const char *s
)
255 if (output_recursion
)
257 output_recursion
= 1;
261 int send_len
= min(len
, (int)sizeof(input_buffer
));
262 nc_send_packet(s
, send_len
);
267 output_recursion
= 0;
270 static int nc_getc(struct stdio_dev
*dev
)
276 net_timeout
= 0; /* no timeout */
282 c
= input_buffer
[input_offset
++];
284 if (input_offset
>= sizeof(input_buffer
))
285 input_offset
-= sizeof(input_buffer
);
291 static int nc_tstc(struct stdio_dev
*dev
)
293 struct eth_device
*eth
;
302 if (eth
&& eth
->state
== ETH_STATE_ACTIVE
)
303 return 0; /* inside net loop */
308 NetLoop(NETCONS
); /* kind of poll */
312 return input_size
!= 0;
315 int drv_nc_init(void)
317 struct stdio_dev dev
;
320 memset(&dev
, 0, sizeof(dev
));
322 strcpy(dev
.name
, "nc");
323 dev
.flags
= DEV_FLAGS_OUTPUT
| DEV_FLAGS_INPUT
| DEV_FLAGS_SYSTEM
;
324 dev
.start
= nc_start
;
330 rc
= stdio_register(&dev
);
332 return (rc
== 0) ? 1 : rc
;