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 IPaddr_t 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 IPaddr_t sip
, unsigned src
,
41 net_set_state(NETLOOP_SUCCESS
); /* got arp reply - quit net loop */
44 static void nc_handler(uchar
*pkt
, unsigned dest
, IPaddr_t sip
, unsigned src
,
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(IPaddr_t ip
)
58 static IPaddr_t netmask
;
59 static IPaddr_t 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_IPaddr("netmask");
66 our_ip
= getenv_IPaddr("ipaddr");
68 env_changed_id
= env_id
;
71 return (ip
== ~0 || /* 255.255.255.255 */
72 ((netmask
& our_ip
) == (netmask
& ip
) && /* on the same net */
73 (netmask
| ip
) == ~0)); /* broadcast to our net */
76 static int refresh_settings_from_env(void)
79 static int env_changed_id
;
80 int env_id
= get_env_id();
82 /* update only when the environment has changed */
83 if (env_changed_id
!= env_id
) {
85 nc_ip
= getenv_IPaddr("ncip");
87 return -1; /* ncip is 0.0.0.0 */
88 p
= strchr(getenv("ncip"), ':');
90 nc_out_port
= simple_strtoul(p
+ 1, NULL
, 10);
91 nc_in_port
= nc_out_port
;
94 nc_ip
= ~0; /* ncip is not set, so broadcast */
96 p
= getenv("ncoutport");
98 nc_out_port
= simple_strtoul(p
, NULL
, 10);
99 p
= getenv("ncinport");
101 nc_in_port
= simple_strtoul(p
, NULL
, 10);
103 if (is_broadcast(nc_ip
))
104 /* broadcast MAC address */
105 memset(nc_ether
, 0xff, sizeof(nc_ether
));
107 /* force arp request */
108 memset(nc_ether
, 0, sizeof(nc_ether
));
114 * Called from NetLoop in net/net.c before each packet
118 refresh_settings_from_env();
119 if (!output_packet_len
|| memcmp(nc_ether
, NetEtherNullAddr
, 6)) {
120 /* going to check for input packet */
121 net_set_udp_handler(nc_handler
);
122 NetSetTimeout(net_timeout
, nc_timeout
);
124 /* send arp request */
126 net_set_arp_handler(nc_wait_arp_handler
);
127 pkt
= (uchar
*)NetTxPacket
+ NetEthHdrSize() + IP_UDP_HDR_SIZE
;
128 memcpy(pkt
, output_packet
, output_packet_len
);
129 NetSendUDPPacket(nc_ether
, nc_ip
, nc_out_port
, nc_in_port
,
134 int nc_input_packet(uchar
*pkt
, IPaddr_t src_ip
, unsigned dest_port
,
135 unsigned src_port
, unsigned len
)
139 if (dest_port
!= nc_in_port
|| !len
)
140 return 0; /* not for us */
142 if (src_ip
!= nc_ip
&& !is_broadcast(nc_ip
))
143 return 0; /* not from our client */
145 debug_cond(DEBUG_DEV_PKT
, "input: \"%*.*s\"\n", len
, len
, pkt
);
147 if (input_size
== sizeof(input_buffer
))
148 return 1; /* no space */
149 if (len
> sizeof(input_buffer
) - input_size
)
150 len
= sizeof(input_buffer
) - input_size
;
152 end
= input_offset
+ input_size
;
153 if (end
> sizeof(input_buffer
))
154 end
-= sizeof(input_buffer
);
157 if (end
+ len
> sizeof(input_buffer
)) {
158 chunk
= sizeof(input_buffer
) - end
;
159 memcpy(input_buffer
, pkt
+ chunk
, len
- chunk
);
161 memcpy(input_buffer
+ end
, pkt
, chunk
);
168 static void nc_send_packet(const char *buf
, int len
)
170 struct eth_device
*eth
;
176 debug_cond(DEBUG_DEV_PKT
, "output: \"%*.*s\"\n", len
, len
, buf
);
182 if (!memcmp(nc_ether
, NetEtherNullAddr
, 6)) {
183 if (eth
->state
== ETH_STATE_ACTIVE
)
184 return; /* inside net loop */
186 output_packet_len
= len
;
188 NetLoop(NETCONS
); /* wait for arp reply and send packet */
190 output_packet_len
= 0;
194 if (eth
->state
!= ETH_STATE_ACTIVE
) {
195 if (eth_is_on_demand_init()) {
196 if (eth_init(gd
->bd
) < 0)
198 eth_set_last_protocol(NETCONS
);
200 eth_init_state_only(gd
->bd
);
204 pkt
= (uchar
*)NetTxPacket
+ NetEthHdrSize() + IP_UDP_HDR_SIZE
;
205 memcpy(pkt
, buf
, len
);
208 NetSendUDPPacket(ether
, ip
, nc_out_port
, nc_in_port
, len
);
211 if (eth_is_on_demand_init())
214 eth_halt_state_only();
218 static int nc_start(struct stdio_dev
*dev
)
222 nc_out_port
= 6666; /* default port */
223 nc_in_port
= nc_out_port
;
225 retval
= refresh_settings_from_env();
230 * Initialize the static IP settings and buffer pointers
231 * incase we call NetSendUDPPacket before NetLoop
238 static void nc_putc(struct stdio_dev
*dev
, char c
)
240 if (output_recursion
)
242 output_recursion
= 1;
244 nc_send_packet(&c
, 1);
246 output_recursion
= 0;
249 static void nc_puts(struct stdio_dev
*dev
, const char *s
)
253 if (output_recursion
)
255 output_recursion
= 1;
259 int send_len
= min(len
, sizeof(input_buffer
));
260 nc_send_packet(s
, send_len
);
265 output_recursion
= 0;
268 static int nc_getc(struct stdio_dev
*dev
)
274 net_timeout
= 0; /* no timeout */
280 c
= input_buffer
[input_offset
++];
282 if (input_offset
>= sizeof(input_buffer
))
283 input_offset
-= sizeof(input_buffer
);
289 static int nc_tstc(struct stdio_dev
*dev
)
291 struct eth_device
*eth
;
300 if (eth
&& eth
->state
== ETH_STATE_ACTIVE
)
301 return 0; /* inside net loop */
306 NetLoop(NETCONS
); /* kind of poll */
310 return input_size
!= 0;
313 int drv_nc_init(void)
315 struct stdio_dev dev
;
318 memset(&dev
, 0, sizeof(dev
));
320 strcpy(dev
.name
, "nc");
321 dev
.flags
= DEV_FLAGS_OUTPUT
| DEV_FLAGS_INPUT
| DEV_FLAGS_SYSTEM
;
322 dev
.start
= nc_start
;
328 rc
= stdio_register(&dev
);
330 return (rc
== 0) ? 1 : rc
;