Веб-консоль для загрузчика U-Boot

ccflags-y += -I$(obj)/../include

obj-y += 
    init.o 
	def.o 
	dns.o 
	inet_chksum.o 
	ip.o 
	mem.o 
	memp.o 

#ifndef __LWIPOPTS_H__
#define __LWIPOPTS_H__

#define NO_SYS                  		1
#define SYS_LIGHTWEIGHT_PROT    		0

#define LWIP_NETCONN                    0
#define LWIP_SOCKET                     0
#define LWIP_DHCP                       1

#define MEM_ALIGNMENT           		4
#define MEM_SIZE                        (8 * 1024 * 1024)

#ifndef __ARCH_CC_H__
#define __ARCH_CC_H__

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

int atoi(const char *str);

#ifdef CONFIG_SYS_BIG_ENDIAN
	#define BYTE_ORDER BIG_ENDIAN
#else
	#define BYTE_ORDER LITTLE_ENDIAN
#endif

#define LWIP_NO_LIMITS_H 1
#define LWIP_NO_CTYPE_H 1

typedef uint8_t     u8_t;
typedef int8_t      s8_t;

void eth_save_packet_lwip(void* packet, int length) {
	if (length > 0) {
		struct pbuf* p = pbuf_alloc(PBUF_RAW, length, PBUF_POOL);
		if (p != NULL) {
			pbuf_take(p, packet, length);
			if (netif.input(p, &netif) != ERR_OK) {
				pbuf_free(p);
			}
		}
	}
}

push_packet = eth_save_packet_lwip;

err_t netif_output(struct netif* netif, struct pbuf* p) {
	unsigned char mac_send_buffer[p->tot_len];
	pbuf_copy_partial(p, (void*)mac_send_buffer, p->tot_len, 0);
	eth_send(mac_send_buffer, p->tot_len);
	return ERR_OK;
}

eth_halt();
eth_init();
	
ip4_addr_t addr;
ip4_addr_t netmask;
ip4_addr_t gw;

IP4_ADDR(&addr, 192, 168, 10, 1);
IP4_ADDR(&netmask, 255, 255, 255, 0);
IP4_ADDR(&gw, 192, 168, 10, 2);

lwip_init();

netif_add(&netif, &addr, &netmask, &gw, NULL, netif_set_opts, netif_input);

netif.name[0] = 'e';
netif.name[1] = '0';
netif_set_default(&netif);

err_t netif_set_opts(struct netif* netif) {
	netif->linkoutput = netif_output;
	netif->output = etharp_output;
	netif->mtu = 1500;
	netif->flags = NETIF_FLAG_BROADCAST | NETIF_FLAG_ETHARP 
    | NETIF_FLAG_ETHERNET | NETIF_FLAG_LINK_UP | NETIF_FLAG_UP;
	netif->hwaddr_len = 6;

	if (env_get("ethaddr"))
		string_to_enetaddr(env_get("ethaddr"), netif->hwaddr);
	else
		memset(netif->hwaddr, 0, 6);

	return ERR_OK;
}

struct udp_pcb *dhcp = udp_new();
udp_bind(dhcp, IP_ADDR_ANY, 67);
udp_recv(dhcp , dhcp_recv, NULL);

void dhcp_recv(void *arg, struct udp_pcb *pcb, struct pbuf *p, 
               const ip_addr_t *addr, u16_t port) {
	if(p == NULL)
		return;
	
	dhcps_msg dhcp_rec;
	
	int data_len = 	p->tot_len;
	pbuf_copy_partial(p, (void*)&dhcp_rec, data_len, 0);
	pbuf_free(p);
	
	int i = 4;
	while(dhcp_rec.options[i] != 255 && dhcp_rec.options[i] != 53) {
		i += dhcp_rec.options[i+1] + 2;
	}

unsigned char http_ans[] = {
  0x3c, 0x68, 0x74, 0x6d, 0x6c, 0x3e, 0x0a, 0x3c, 0x68, 0x65, 0x61, 0x64,
  0x3e, 0x0a, 0x09, 0x3c, 0x73, 0x74, 0x79, 0x6c, 0x65, 0x3e, 0x0a, 0x09,
  0x2e, 0x78, 0x74, 0x65, 0x72, 0x6d, 0x7b, 0x66, 0x6f, 0x6e, 0x74, 0x2d,
  0x66, 0x65, 0x61, 0x74, 0x75, 0x72, 0x65, 0x2d, 0x73, 0x65, 0x74, 0x74,
  0x69, 0x6e, 0x67, 0x73, 0x3a, 0x22, 0x6c, 0x69, 0x67, 0x61, 0x22, 0x20,
  0x30, 0x3b, 0x70, 0x6f, 0x73, 0x69, 0x74, 0x69, 0x6f, 0x6e, 0x3a, 0x72,
  0x65, 0x6c, 0x61, 0x74, 0x69, 0x76, 0x65, 0x3b, 0x75, 0x73, 0x65, 0x72,
  0x2d, 0x73, 0x65, 0x6c, 0x65, 0x63, 0x74, 0x3a, 0x6e, 0x6f, 0x6e, 0x65,
  0x3b, 0x2d, 0x6d, 0x73, 0x2d, 0x75, 0x73, 0x65, 0x72, 0x2d, 0x73, 0x65,
  0x6c, 0x65, 0x63, 0x74, 0x3a, 0x6e, 0x6f, 0x6e, 0x65, 0x3b, 0x2d, 0x77,
  0x65, 0x62, 0x6b, 0x69, 0x74, 0x2d, 0x75, 0x73, 0x65, 0x72, 0x2d, 0x73,
  0x65, 0x6c, 0x65, 0x63, 0x74, 0x3a, 0x6e, 0x6f, 0x6e, 0x65, 0x7d, 0x2e,
  0x78, 0x74, 0x65, 0x72, 0x6d, 0x2e, 0x66, 0x6f, 0x63, 0x75, 0x73, 0x2c,
  0x2e, 0x78, 0x74, 0x65, 0x72, 0x6d, 0x3a, 0x66, 0x6f, 0x63, 0x75, 0x73,
  0x7b, 0x6f, 0x75, 0x74, 0x6c, 0x69, 0x6e, 0x65, 0x3a, 0x30, 0x7d, 0x2e,
  0x78, 0x74, 0x65, 0x72, 0x6d, 0x20, 0x2e, 0x78, 0x74, 0x65, 0x72, 0x6d,
  0x2d, 0x68, 0x65, 0x6c, 0x70, 0x65, 0x72, 0x73, 0x7b, 0x70, 0x6f, 0x73,
  0x69, 0x74, 0x69, 0x6f, 0x6e, 0x3a, 0x61, 0x62, 0x73, 0x6f, 0x6c, 0x75,
  0x74, 0x65, 0x3b, 0x74, 0x6f, 0x70, 0x3a, 0x30, 0x3b, 0x7a, 0x2d, 0x69,
  0x6e, 0x64, 0x65, 0x78, 0x3a, 0x31, 0x30, 0x7d, 0x2e, 0x78, 0x74, 0x65,

unsigned int http_ans_len = 227384;

<html>
<head>
	<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/xterm/3.14.5/xterm.min.css" />
    <script src="https://cdnjs.cloudflare.com/ajax/libs/xterm/3.14.5/xterm.min.js"></script>
</head>
<body>
    <div id="terminal"></div>
    <script>
        var opts = {
			cols: 80,
			rows: 54,
			convertEol : 1
        }

        var term = new Terminal(opts);
        term.open(document.getElementById('terminal'));

		var ws = new WebSocket("ws://192.168.10.1:3000");
		ws.binaryType = "arraybuffer";

		ws.addEventListener('message', function (event) {
			term.write(event.data);
		});

        term.on("key", function(key, ev) {
			if (ev.keyCode === 13) {
				ws.send("n");
			} else if (ev.keyCode === 8) {
				ws.send("b");
			} else {
				ws.send(key);
			}
        });
    </script>
</body>
</html>

struct tcp_pcb* http = tcp_new();
tcp_bind(http, IP_ADDR_ANY, 80);
http = tcp_listen_with_backlog(http, TCP_DEFAULT_LISTEN_BACKLOG);
tcp_accept(http, http_accept);

err_t http_recv(void* arg, struct tcp_pcb* tpcb, struct pbuf* p,
                err_t err) {
	int data_len = 	p->tot_len;
	char tcp_rec[data_len];
	pbuf_copy_partial(p, (void*)tcp_rec, data_len, 0);
	tcp_recved(tpcb, data_len);
	pbuf_free(p);

	char answer_html[1000];
	sprintf(answer_html, HTTP_RSP, http_ans_len);

	tcp_write(tpcb, answer_html, strlen(answer_html), 0x01);

	http_ans_sended = 0;

	return ERR_OK;
}

const char HTTP_RSP[] = 
	"HTTP/1.1 200 OKrn" 
	"Content-Length: %drn" 
	"Content-Type: text/htmlrnrn";

const char WS_RSP[] = 
	"HTTP/1.1 101 Switching Protocolsrn" 
	"Upgrade: websocketrn" 
	"Connection: Upgradern" 
	"Sec-WebSocket-Accept: %srnrn";
const char WS_GUID[] = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
const char WS_KEY[] = "Sec-WebSocket-Key: ";

err_t websocket_recv(void* arg, struct tcp_pcb* tpcb, struct pbuf* p, 
                    err_t err) {
	if (p == NULL) {
		web_socket_open = 0;
		globa_tcp = NULL;
		tcp_get = 0;
		return ERR_OK;
	}
	
	int data_len = 	p->tot_len;
	char tcp_rec[data_len];
	pbuf_copy_partial(p, (void*)tcp_rec, data_len, 0);
	tcp_recved(tpcb, data_len);
	
	if(web_socket_open == 0){
		char * sec_websocket_position_start = strstr(tcp_rec, WS_KEY);
		if(sec_websocket_position_start){

	lwip_u_boot_port();

	cli_loop();

#ifndef CONFIG_SPL_BUILD
			if(push_packet) {
				eth_rx();
				char tmp = tcp_get;
				tcp_get = 0;
				if(tmp)
					return tmp;
			}
#endif

#ifndef CONFIG_SPL_BUILD
	if(globa_tcp) {
		unsigned char buf[3];
		buf[0] = 0x80 | 0x01;
		buf[1] = 1;
		buf[2] = c;
		tcp_write(globa_tcp, buf, 3, 1);
		tcp_output(globa_tcp);
	} else {
#endif

#ifndef CONFIG_SPL_BUILD
	if(globa_tcp) {
		int len = strlen(s);
		unsigned char buf[150];
		while (len) {
			int send_len = min(len, 125);
			buf[0] = 0x80 | 0x01;
			buf[1] = send_len;
			memcpy(&buf[2], s, send_len);
			tcp_write(globa_tcp, buf, send_len + 2, 1);
			len -= send_len;
			s += send_len;
		}
		tcp_output(globa_tcp);
	} else {
#endif