xyou
efa626ed17
- 用户态共享网络协议栈 CODE/tools/net_stack.h（eth/ARP/IPv4/ICMP/UDP-DNS/TCP static inline，轮询模型） - shell 内建 ping/curl/run 命令；ping.elf/curl.elf 对齐 net_stack.h 真实协议栈 - shell dev_mode 串口镜像（g_serial_mirror）便于自动化验证 - DSK 启动流程：firstInit.txt 双行格式（firstInit + dev_mode），dev_mode=1 触发 shell 自动回归测试 - FirstInit 完成→持久化 USER.CONF + 翻转 firstInit 0->1（保留 dev_mode） - firstInit=1 时 DSK 加载 login.elf 进行用户选择 + 密码校验后再进 desktop - 新增 CODE/login/ 登录应用、CODE/DKM/dkm_shared.h、Logo.png - ahci/e1000/virtio_net/fat32/console_fb 驱动更新；utsm serial/block/dsk_loader 调整 - build.ps1 打包 ping.elf/curl.elf 进 SATA FAT32 根目录 - 移除 CODE/tools/bash（与 shell.elf 重复）
696 lines
26 KiB
C
696 lines
26 KiB
C
/* netman — Deshab network manager initializer
|
||
* 读取 /system/deshab64/network/conf/conf.conf（SATA 早期镜像里为 root 目录 NETCONF.CNF），
|
||
* 解析网络配置；枚举内核 netdev 注册表；对具备收发能力的接口通过 e1000 真实收发
|
||
* 跑最小 DHCP 客户端拿到 IP。
|
||
*/
|
||
|
||
#include "../UTSM/include/utsm/dsk.h"
|
||
|
||
typedef signed char i8;
|
||
typedef unsigned char u8;
|
||
typedef unsigned short u16;
|
||
typedef unsigned int u32;
|
||
typedef unsigned long long u64;
|
||
|
||
typedef int (*block_read_fn)(u32 index, u64 lba, u32 count, void *buf);
|
||
typedef u32 (*net_device_count_fn)(void);
|
||
typedef int (*net_device_info_fn)(u32 index, void *out);
|
||
typedef int (*net_tx_fn)(u32 index, const void *packet, u32 length);
|
||
typedef int (*net_rx_poll_fn)(u32 index, void *buffer, u32 capacity, u32 *out_length);
|
||
|
||
#define DKM_NET_F_LINK_UP (1u << 0)
|
||
#define DKM_NET_F_TX_READY (1u << 1)
|
||
#define DKM_NET_F_RX_READY (1u << 2)
|
||
|
||
typedef struct {
|
||
const char *name;
|
||
u8 mac[6];
|
||
u32 flags;
|
||
} nm_net_device_info;
|
||
|
||
#define COM1 0x3F8
|
||
|
||
static inline void outb(u16 port, u8 value) { __asm__ volatile("outb %0,%1"::"a"(value),"Nd"(port)); }
|
||
static inline u8 inb(u16 port) { u8 v; __asm__ volatile("inb %1,%0":"=a"(v):"Nd"(port)); return v; }
|
||
|
||
static void sputc(char c) {
|
||
for (u32 i=0;i<100000;i++) if (inb(COM1+5)&0x20) break;
|
||
outb(COM1,(u8)c);
|
||
}
|
||
static void swrite(const char *s) { while(*s){ if(*s=='\n') sputc('\r'); sputc(*s++);} }
|
||
static void logl(const char *s) { swrite(s); swrite("\n"); }
|
||
|
||
static void log_hex(const char *p, u64 v) {
|
||
static const char h[]="0123456789abcdef";
|
||
char b[19]; int i=0; b[i++]='0'; b[i++]='x';
|
||
for(int j=15;j>=0;j--) b[i++]=h[(v>>(j*4))&0xf];
|
||
b[i]=0; swrite(p); swrite(b); swrite("\n");
|
||
}
|
||
|
||
static void *memset_nm(void *d, int c, u64 n) { u8 *p=(u8*)d; while(n--)*p++=(u8)c; return d; }
|
||
static int streqn(const char *a, const char *b, u32 n) { for(u32 i=0;i<n;i++) if(a[i]!=b[i]) return 0; return 1; }
|
||
static int neq11(const char *a, const char *b) { for(int i=0;i<11;i++) if(a[i]!=b[i]) return 0; return 1; }
|
||
static int str_eq(const char *a, const char *b) {
|
||
while (*a && *b) {
|
||
if (*a != *b) return 0;
|
||
a++;
|
||
b++;
|
||
}
|
||
return *a == 0 && *b == 0;
|
||
}
|
||
static u16 r16(const u8 *p) { return (u16)p[0]|((u16)p[1]<<8); }
|
||
static u32 r32(const u8 *p) { return (u32)p[0]|((u32)p[1]<<8)|((u32)p[2]<<16)|((u32)p[3]<<24); }
|
||
|
||
#pragma pack(push,1)
|
||
typedef struct {
|
||
u8 jmp[3]; char oem[8]; u16 bps; u8 spc; u16 rsvd; u8 fc; u16 root_ent;
|
||
u16 ts16; u8 media; u16 spf16; u16 spt; u16 heads; u32 hidden; u32 ts32;
|
||
u32 spf; u16 flags; u16 ver; u32 root_clus; u16 fsi; u16 bkboot;
|
||
u8 res[12]; u8 drv; u8 ntfl; u8 sig; u32 ser; char lbl[11]; char typ[8];
|
||
u8 code[420]; u16 boot_sig;
|
||
} fat32_bpb;
|
||
|
||
typedef struct {
|
||
char name[11]; u8 attr; u8 ntr; u8 ctenth;
|
||
u16 ctime; u16 cdate; u16 adate; u16 chigh;
|
||
u16 wtime; u16 wdate; u16 clow; u32 fsize;
|
||
} fat32_de;
|
||
#pragma pack(pop)
|
||
|
||
static block_read_fn g_block_read;
|
||
static net_device_count_fn g_net_device_count;
|
||
static net_device_info_fn g_net_device_info;
|
||
static net_tx_fn g_net_tx;
|
||
static net_rx_poll_fn g_net_rx_poll;
|
||
static u8 g_disk[131072];
|
||
static u8 g_cluster[4096];
|
||
static char g_conf[4096];
|
||
|
||
static int read_sectors(u32 lba, u32 count, u8 *out) {
|
||
return g_block_read ? g_block_read(0, lba, count, out) : -1;
|
||
}
|
||
|
||
static int find_entry(const u8 *clus, u32 clus_sectors, const char *target, u32 *out_clus, u32 *out_size) {
|
||
const fat32_de *dir = (const fat32_de *)clus;
|
||
for (u32 e=0; e*32 < clus_sectors*512; e++) {
|
||
if (dir[e].name[0]==0) break;
|
||
if ((u8)dir[e].name[0]==0xE5) continue;
|
||
if (dir[e].attr==0x0F) continue;
|
||
if (dir[e].attr & 0x08) continue;
|
||
if (neq11(dir[e].name, target)) {
|
||
*out_clus = ((u32)r16((const u8*)&dir[e].chigh)<<16) | r16((const u8*)&dir[e].clow);
|
||
*out_size = dir[e].fsize;
|
||
return 0;
|
||
}
|
||
}
|
||
return -1;
|
||
}
|
||
|
||
static int fat32_read_root_file(const char *name11, char *out, u32 out_cap, u32 *out_size) {
|
||
if (read_sectors(0, 256, g_disk) != 0) return -1;
|
||
const fat32_bpb *bpb = (const fat32_bpb *)g_disk;
|
||
if (bpb->boot_sig != 0xAA55 || bpb->bps != 512 || bpb->spf == 0) return -2;
|
||
u32 data_lba = bpb->rsvd + (u32)bpb->fc * bpb->spf;
|
||
u32 fat_byte_off = bpb->rsvd * 512;
|
||
u32 spc = bpb->spc;
|
||
u32 clus = bpb->root_clus;
|
||
u32 found_clus=0, found_size=0;
|
||
int found=0;
|
||
while (clus >= 2 && clus < 0x0FFFFFF8 && !found) {
|
||
u32 lba = data_lba + (clus - 2) * spc;
|
||
const u8 *cb;
|
||
if ((u64)lba * 512 + (u64)spc * 512 <= 256ULL * 512) cb = g_disk + (u64)lba * 512;
|
||
else { if (read_sectors(lba, spc, g_cluster) != 0) return -3; cb = g_cluster; }
|
||
if (find_entry(cb, spc, name11, &found_clus, &found_size) == 0) { found=1; break; }
|
||
u32 fo = fat_byte_off + clus * 4;
|
||
if (fo + 4 > sizeof(g_disk)) break;
|
||
clus = r32(g_disk + fo) & 0x0FFFFFFF;
|
||
}
|
||
if (!found) return -4;
|
||
if (found_size + 1 > out_cap) return -5;
|
||
u32 remaining = found_size;
|
||
u32 fc = found_clus;
|
||
char *dst = out;
|
||
while (fc >= 2 && fc < 0x0FFFFFF8 && remaining > 0) {
|
||
u32 lba = data_lba + (fc - 2) * spc;
|
||
u32 bytes = spc * 512;
|
||
if (bytes > remaining) bytes = remaining;
|
||
if (read_sectors(lba, spc, g_cluster) != 0) return -6;
|
||
for (u32 i=0;i<bytes;i++) dst[i]=(char)g_cluster[i];
|
||
dst += bytes; remaining -= bytes;
|
||
u32 fo = fat_byte_off + fc * 4;
|
||
if (fo + 4 > sizeof(g_disk)) break;
|
||
fc = r32(g_disk + fo) & 0x0FFFFFFF;
|
||
}
|
||
out[found_size] = 0;
|
||
*out_size = found_size;
|
||
return 0;
|
||
}
|
||
|
||
typedef struct {
|
||
char mode[16];
|
||
char device[16];
|
||
char name[64];
|
||
char password[64];
|
||
char ip[16];
|
||
char dns[16];
|
||
} net_conf;
|
||
|
||
static void copy_value(char *dst, u32 cap, const char *src, u32 len) {
|
||
if (!cap) return;
|
||
u32 n = len < cap-1 ? len : cap-1;
|
||
for (u32 i=0;i<n;i++) dst[i]=src[i];
|
||
dst[n]=0;
|
||
}
|
||
|
||
static void parse_conf(const char *txt, net_conf *cfg) {
|
||
copy_value(cfg->mode, sizeof(cfg->mode), "dhcp", 4);
|
||
copy_value(cfg->device, sizeof(cfg->device), "auto", 4);
|
||
copy_value(cfg->ip, sizeof(cfg->ip), "dhcp", 4);
|
||
copy_value(cfg->dns, sizeof(cfg->dns), "auto", 4);
|
||
cfg->name[0]=0; cfg->password[0]=0;
|
||
const char *p = txt;
|
||
while (*p) {
|
||
const char *line = p;
|
||
u32 len = 0;
|
||
while (p[len] && p[len]!='\n' && p[len]!='\r') len++;
|
||
const char *eq = 0;
|
||
for (u32 i=0;i<len;i++) if (line[i]=='=') { eq = line + i; break; }
|
||
if (eq) {
|
||
u32 klen = (u32)(eq - line);
|
||
const char *v = eq + 1;
|
||
u32 vlen = len - klen - 1;
|
||
if (klen==12 && streqn(line,"network.mode",12)) copy_value(cfg->mode,sizeof(cfg->mode),v,vlen);
|
||
else if (klen==14 && streqn(line,"network.device",14)) copy_value(cfg->device,sizeof(cfg->device),v,vlen);
|
||
else if (klen==12 && streqn(line,"network.name",12)) copy_value(cfg->name,sizeof(cfg->name),v,vlen);
|
||
else if (klen==16 && streqn(line,"network.password",16)) copy_value(cfg->password,sizeof(cfg->password),v,vlen);
|
||
else if (klen==10 && streqn(line,"network.ip",10)) copy_value(cfg->ip,sizeof(cfg->ip),v,vlen);
|
||
else if (klen==11 && streqn(line,"network.dns",11)) copy_value(cfg->dns,sizeof(cfg->dns),v,vlen);
|
||
}
|
||
p += len;
|
||
while (*p=='\n' || *p=='\r') p++;
|
||
}
|
||
}
|
||
|
||
static void log_cfg(const net_conf *cfg) {
|
||
logl("[netman] config loaded");
|
||
swrite("[netman] mode="); logl(cfg->mode);
|
||
swrite("[netman] device="); logl(cfg->device);
|
||
swrite("[netman] name="); logl(cfg->name[0] ? cfg->name : "<empty>");
|
||
swrite("[netman] ip="); logl(cfg->ip);
|
||
swrite("[netman] dns="); logl(cfg->dns);
|
||
u32 pwlen=0; while(cfg->password[pwlen]) pwlen++;
|
||
log_hex("[netman] password length=", pwlen);
|
||
}
|
||
|
||
static void log_mac(const u8 mac[6]) {
|
||
static const char h[]="0123456789abcdef";
|
||
char b[18];
|
||
int p=0;
|
||
for (int i=0;i<6;i++) {
|
||
if (i) b[p++]=':';
|
||
b[p++]=h[(mac[i]>>4)&0xf];
|
||
b[p++]=h[mac[i]&0xf];
|
||
}
|
||
b[p]=0;
|
||
swrite(" mac=");
|
||
swrite(b);
|
||
}
|
||
|
||
/* ---- 最小 DHCP 客户端：Ethernet + IPv4 + UDP，通过 e1000 真实收发 ---- */
|
||
|
||
static u8 g_pktbuf[2048];
|
||
static u8 g_rxbuf[2048];
|
||
static u8 g_local_mac[6];
|
||
static u32 g_lease_ip;
|
||
static u32 g_server_ip;
|
||
static u32 g_gateway_ip;
|
||
static u32 g_dns_ip;
|
||
static u32 g_dhcp_xid = 0x21030201u;
|
||
|
||
/* 累加和，用于 IPv4 头部校验 */
|
||
static u16 ip_checksum(const u8 *data, u32 len) {
|
||
u32 sum = 0;
|
||
for (u32 i=0;i+1<len;i+=2) sum += (u32)((data[i]<<8)|data[i+1]);
|
||
if (len & 1) sum += (u32)(data[len-1]<<8);
|
||
while (sum >> 16) sum = (sum & 0xffff) + (sum >> 16);
|
||
return (u16)(~sum);
|
||
}
|
||
|
||
static u16 udp_checksum_ipv4(const u8 *ip, const u8 *udp, u32 udp_len) {
|
||
u32 sum = 0;
|
||
for (u32 i=12;i<20;i+=2) sum += (u32)((ip[i]<<8)|ip[i+1]);
|
||
sum += 17;
|
||
sum += udp_len;
|
||
for (u32 i=0;i+1<udp_len;i+=2) sum += (u32)((udp[i]<<8)|udp[i+1]);
|
||
if (udp_len & 1) sum += (u32)(udp[udp_len-1]<<8);
|
||
while (sum >> 16) sum = (sum & 0xffff) + (sum >> 16);
|
||
u16 out = (u16)(~sum);
|
||
return out ? out : 0xffff;
|
||
}
|
||
|
||
/* ---- TSC-based timing (实机要求: 用 CPU 频率计算, 不用循环) ---- */
|
||
static u64 g_tsc_per_ms = 0;
|
||
|
||
static inline u64 rdtsc_nm(void) {
|
||
u32 lo, hi;
|
||
__asm__ volatile("rdtsc" : "=a"(lo), "=d"(hi));
|
||
return ((u64)hi << 32) | lo;
|
||
}
|
||
|
||
static void tsc_calibrate_nm(void) {
|
||
outb(0x43, 0x30); /* ch0, lo+hi, mode 0, binary */
|
||
outb(0x40, 0x7c); /* 11932 low = ~10ms */
|
||
outb(0x40, 0x2e); /* 11932 high */
|
||
u64 tsc_start = rdtsc_nm();
|
||
u16 prev = 0; u64 loops = 0;
|
||
for (;;) {
|
||
outb(0x43, 0x00);
|
||
u16 cur = (u16)inb(0x40) | ((u16)inb(0x40) << 8);
|
||
if (cur > prev && loops > 10) break;
|
||
prev = cur; loops++;
|
||
}
|
||
u64 tsc_end = rdtsc_nm();
|
||
g_tsc_per_ms = (tsc_end - tsc_start) / 10;
|
||
}
|
||
|
||
static void delay_ms_nm(u32 ms) {
|
||
if (!g_tsc_per_ms) {
|
||
for (volatile u32 i = 0; i < 100000 * ms; i++) __asm__ volatile("pause");
|
||
return;
|
||
}
|
||
u64 target = g_tsc_per_ms * ms;
|
||
u64 start = rdtsc_nm();
|
||
while (rdtsc_nm() - start < target) __asm__ volatile("pause");
|
||
}
|
||
|
||
static void delay_spin(u32 loops) {
|
||
/* 兼容旧调用: 粗略转换为 ms (实机以 TSC 为准) */
|
||
if (g_tsc_per_ms) {
|
||
delay_ms_nm(loops / 10000 + 1);
|
||
} else {
|
||
for (volatile u32 i=0;i<loops;i++) { __asm__ volatile("pause"); }
|
||
}
|
||
}
|
||
|
||
/* 构造并发送一个 DHCP 报文。type: 1=DISCOVER 3=REQUEST */
|
||
static int dhcp_send(u32 index, u8 type, u32 requested_ip, u32 server_ip) {
|
||
memset_nm(g_pktbuf, 0, sizeof(g_pktbuf));
|
||
u8 *p = g_pktbuf;
|
||
|
||
/* Ethernet (14) */
|
||
for (int i=0;i<6;i++) p[i] = 0xff; /* dst broadcast */
|
||
for (int i=0;i<6;i++) p[6+i] = g_local_mac[i];
|
||
p[12]=0x08; p[13]=0x00; /* IPv4 */
|
||
|
||
u8 *ip = p + 14;
|
||
u8 *udp = ip + 20;
|
||
u8 *dhcp = udp + 8;
|
||
|
||
/* DHCP/bootp */
|
||
dhcp[0]=1; dhcp[1]=1; dhcp[2]=6; dhcp[3]=0;
|
||
dhcp[4]=(u8)(g_dhcp_xid>>24); dhcp[5]=(u8)(g_dhcp_xid>>16);
|
||
dhcp[6]=(u8)(g_dhcp_xid>>8); dhcp[7]=(u8)(g_dhcp_xid);
|
||
dhcp[10]=0x80; dhcp[11]=0x00; /* broadcast flag */
|
||
for (int i=0;i<6;i++) dhcp[28+i]=g_local_mac[i]; /* chaddr */
|
||
dhcp[236]=99; dhcp[237]=130; dhcp[238]=83; dhcp[239]=99; /* magic cookie */
|
||
|
||
u32 opt = 240;
|
||
dhcp[opt++]=53; dhcp[opt++]=1; dhcp[opt++]=type;
|
||
if (type == 3) {
|
||
dhcp[opt++]=50; dhcp[opt++]=4;
|
||
dhcp[opt++]=(u8)(requested_ip>>24); dhcp[opt++]=(u8)(requested_ip>>16);
|
||
dhcp[opt++]=(u8)(requested_ip>>8); dhcp[opt++]=(u8)(requested_ip);
|
||
dhcp[opt++]=54; dhcp[opt++]=4;
|
||
dhcp[opt++]=(u8)(server_ip>>24); dhcp[opt++]=(u8)(server_ip>>16);
|
||
dhcp[opt++]=(u8)(server_ip>>8); dhcp[opt++]=(u8)(server_ip);
|
||
}
|
||
dhcp[opt++]=55; dhcp[opt++]=3; dhcp[opt++]=1; dhcp[opt++]=3; dhcp[opt++]=6;
|
||
dhcp[opt++]=255;
|
||
|
||
u32 dhcp_len = opt;
|
||
if (dhcp_len < 548) dhcp_len = 548;
|
||
u32 udp_len = 8 + dhcp_len;
|
||
u32 ip_len = 20 + udp_len;
|
||
|
||
/* UDP (8) */
|
||
udp[0]=0x00; udp[1]=68; udp[2]=0x00; udp[3]=67;
|
||
udp[4]=(u8)(udp_len>>8); udp[5]=(u8)(udp_len);
|
||
udp[6]=0; udp[7]=0;
|
||
|
||
/* IPv4 (20) */
|
||
ip[0]=0x45; ip[1]=0x00;
|
||
ip[2]=(u8)(ip_len>>8); ip[3]=(u8)(ip_len);
|
||
ip[4]=0; ip[5]=0; ip[6]=0x00; ip[7]=0x00;
|
||
ip[8]=64; ip[9]=17; ip[10]=0; ip[11]=0;
|
||
ip[12]=0;ip[13]=0;ip[14]=0;ip[15]=0;
|
||
ip[16]=255;ip[17]=255;ip[18]=255;ip[19]=255;
|
||
u16 ipck = ip_checksum(ip, 20);
|
||
ip[10]=(u8)(ipck>>8); ip[11]=(u8)(ipck);
|
||
|
||
u16 udpck = udp_checksum_ipv4(ip, udp, udp_len);
|
||
udp[6]=(u8)(udpck>>8); udp[7]=(u8)(udpck);
|
||
|
||
u32 frame_len = 14 + ip_len;
|
||
if (frame_len < 60) frame_len = 60;
|
||
return g_net_tx(index, g_pktbuf, frame_len);
|
||
}
|
||
|
||
/* 解析收到的帧，若为本机 xid 的 DHCP 回复则返回 message type。 */
|
||
static int dhcp_parse(const u8 *frame, u32 len, u32 *out_yiaddr, u32 *out_server, u32 *out_router, u32 *out_dns) {
|
||
if (len < 14+20+8+240) return -1;
|
||
if (frame[12]!=0x08 || frame[13]!=0x00) return -1;
|
||
const u8 *ip = frame + 14;
|
||
if ((ip[0]>>4)!=4) return -1;
|
||
u32 ihl = (ip[0]&0xf)*4;
|
||
if (ihl < 20 || len < 14 + ihl + 8 + 240) return -1;
|
||
u16 ip_total = (u16)((ip[2]<<8)|ip[3]);
|
||
if (ip_total < ihl + 8 + 240 || len < 14 + ip_total) return -1;
|
||
if (ip[9]!=17) return -1;
|
||
const u8 *udp = ip + ihl;
|
||
u16 udp_len = (u16)((udp[4]<<8)|udp[5]);
|
||
if (udp_len < 8 + 240 || udp_len > ip_total - ihl) return -1;
|
||
u16 dport = (u16)((udp[2]<<8)|udp[3]);
|
||
if (dport != 68) return -1;
|
||
const u8 *dhcp = udp + 8;
|
||
u32 xid = ((u32)dhcp[4]<<24)|((u32)dhcp[5]<<16)|((u32)dhcp[6]<<8)|dhcp[7];
|
||
if (xid != g_dhcp_xid) return -1;
|
||
u32 yi = ((u32)dhcp[16]<<24)|((u32)dhcp[17]<<16)|((u32)dhcp[18]<<8)|dhcp[19];
|
||
if (!(dhcp[236]==99&&dhcp[237]==130&&dhcp[238]==83&&dhcp[239]==99)) return -1;
|
||
u32 o = 240;
|
||
u8 msg_type = 0;
|
||
u32 server = 0;
|
||
u32 router = 0;
|
||
u32 dns = 0;
|
||
u32 cap = udp_len - 8;
|
||
while (o < cap) {
|
||
u8 code = dhcp[o++];
|
||
if (code == 255) break;
|
||
if (code == 0) continue;
|
||
if (o >= cap) break;
|
||
u8 l = dhcp[o++];
|
||
if (o + l > cap) break;
|
||
if (code == 53 && l >= 1) msg_type = dhcp[o];
|
||
if (code == 54 && l >= 4)
|
||
server = ((u32)dhcp[o]<<24)|((u32)dhcp[o+1]<<16)|((u32)dhcp[o+2]<<8)|dhcp[o+3];
|
||
if (code == 3 && l >= 4)
|
||
router = ((u32)dhcp[o]<<24)|((u32)dhcp[o+1]<<16)|((u32)dhcp[o+2]<<8)|dhcp[o+3];
|
||
if (code == 6 && l >= 4)
|
||
dns = ((u32)dhcp[o]<<24)|((u32)dhcp[o+1]<<16)|((u32)dhcp[o+2]<<8)|dhcp[o+3];
|
||
o += l;
|
||
}
|
||
*out_yiaddr = yi;
|
||
*out_server = server;
|
||
if (out_router) *out_router = router;
|
||
if (out_dns) *out_dns = dns;
|
||
return msg_type;
|
||
}
|
||
|
||
static void log_ipv4(const char *pre, u32 ip_hostorder) {
|
||
char b[16]; int p=0;
|
||
for (int s=3;s>=0;s--) {
|
||
u32 oct = (ip_hostorder >> (s*8)) & 0xff;
|
||
if (oct==0) { b[p++]='0'; }
|
||
else {
|
||
int started=0;
|
||
for (u32 d=100; d>=1; d/=10) {
|
||
u32 dig=(oct/d)%10;
|
||
if (dig||started){ b[p++]=(char)('0'+dig); started=1; }
|
||
if (d==1) break;
|
||
}
|
||
}
|
||
if (s) b[p++]='.';
|
||
}
|
||
b[p]=0;
|
||
swrite(pre); swrite(b); swrite("\n");
|
||
}
|
||
|
||
/* 对指定 netdev 跑一次 DHCP DISCOVER/OFFER/REQUEST/ACK。成功返回 0。 */
|
||
static int dhcp_run(u32 index, const u8 mac[6]) {
|
||
if (!g_net_tx || !g_net_rx_poll) { logl("[netman] DHCP: net tx/rx api missing"); return -1; }
|
||
for (int i=0;i<6;i++) g_local_mac[i]=mac[i];
|
||
g_lease_ip = 0; g_server_ip = 0; g_gateway_ip = 0; g_dns_ip = 0;
|
||
|
||
u32 offer_ip = 0, offer_srv = 0, offer_router = 0, offer_dns = 0;
|
||
int got_offer = 0;
|
||
u32 rx_frames = 0;
|
||
for (u32 round=0; round<4 && !got_offer; round++) {
|
||
logl("[netman] DHCP: sending DISCOVER");
|
||
int txrc = dhcp_send(index, 1, 0, 0);
|
||
log_hex("[netman] DHCP DISCOVER tx rc=", (u64)(i8)txrc);
|
||
if (txrc != 0) { logl("[netman] DHCP: DISCOVER tx failed"); return -1; }
|
||
|
||
for (u32 attempt=0; attempt<120000 && !got_offer; attempt++) {
|
||
u32 rlen = 0;
|
||
int rc = g_net_rx_poll(index, g_rxbuf, sizeof(g_rxbuf), &rlen);
|
||
if (rc == 0 && rlen > 0) {
|
||
rx_frames++;
|
||
int t = dhcp_parse(g_rxbuf, rlen, &offer_ip, &offer_srv, &offer_router, &offer_dns);
|
||
if (t == 2) { got_offer = 1; break; }
|
||
}
|
||
delay_spin(300);
|
||
}
|
||
}
|
||
log_hex("[netman] DHCP rx frames seen=", rx_frames);
|
||
if (!got_offer) { logl("[netman] DHCP: no OFFER received"); return -2; }
|
||
log_ipv4("[netman] DHCP OFFER ip=", offer_ip);
|
||
|
||
u32 ack_ip=0, ack_srv=0, ack_router=0, ack_dns=0;
|
||
int got_ack=0;
|
||
for (u32 round=0; round<4 && !got_ack; round++) {
|
||
logl("[netman] DHCP: sending REQUEST");
|
||
if (dhcp_send(index, 3, offer_ip, offer_srv) != 0) { logl("[netman] DHCP: REQUEST tx failed"); return -3; }
|
||
|
||
for (u32 attempt=0; attempt<120000 && !got_ack; attempt++) {
|
||
u32 rlen=0;
|
||
int rc = g_net_rx_poll(index, g_rxbuf, sizeof(g_rxbuf), &rlen);
|
||
if (rc == 0 && rlen > 0) {
|
||
int t = dhcp_parse(g_rxbuf, rlen, &ack_ip, &ack_srv, &ack_router, &ack_dns);
|
||
if (t == 5) { got_ack=1; break; }
|
||
}
|
||
delay_spin(300);
|
||
}
|
||
}
|
||
if (!got_ack) { logl("[netman] DHCP: no ACK received"); return -4; }
|
||
|
||
g_lease_ip = ack_ip; g_server_ip = ack_srv;
|
||
g_gateway_ip = ack_router ? ack_router : (offer_router ? offer_router : 0x0a000202u);
|
||
g_dns_ip = ack_dns ? ack_dns : (offer_dns ? offer_dns : g_gateway_ip);
|
||
log_ipv4("[netman] DHCP ACK, leased ip=", ack_ip);
|
||
log_ipv4("[netman] DHCP server=", ack_srv);
|
||
log_ipv4("[netman] gateway=", g_gateway_ip);
|
||
log_ipv4("[netman] dns=", g_dns_ip);
|
||
logl("[netman] DHCP: connected");
|
||
return 0;
|
||
}
|
||
|
||
static int arp_resolve_gateway(u32 index, u32 target_ip, u8 out_mac[6]) {
|
||
memset_nm(g_pktbuf, 0, sizeof(g_pktbuf));
|
||
u8 *p = g_pktbuf;
|
||
for (int i=0;i<6;i++) p[i]=0xff;
|
||
for (int i=0;i<6;i++) p[6+i]=g_local_mac[i];
|
||
p[12]=0x08; p[13]=0x06;
|
||
u8 *a = p + 14;
|
||
a[0]=0x00; a[1]=0x01;
|
||
a[2]=0x08; a[3]=0x00;
|
||
a[4]=6; a[5]=4;
|
||
a[6]=0x00; a[7]=0x01;
|
||
for (int i=0;i<6;i++) a[8+i]=g_local_mac[i];
|
||
a[14]=(u8)(g_lease_ip>>24); a[15]=(u8)(g_lease_ip>>16); a[16]=(u8)(g_lease_ip>>8); a[17]=(u8)g_lease_ip;
|
||
a[24]=(u8)(target_ip>>24); a[25]=(u8)(target_ip>>16); a[26]=(u8)(target_ip>>8); a[27]=(u8)target_ip;
|
||
logl("[netman] ARP: who-has gateway");
|
||
if (g_net_tx(index, g_pktbuf, 60) != 0) return -1;
|
||
for (u32 attempt=0; attempt<160000; attempt++) {
|
||
u32 rlen=0;
|
||
int rc = g_net_rx_poll(index, g_rxbuf, sizeof(g_rxbuf), &rlen);
|
||
if (rc == 0 && rlen >= 42 && g_rxbuf[12]==0x08 && g_rxbuf[13]==0x06) {
|
||
const u8 *r = g_rxbuf + 14;
|
||
u16 op = (u16)((r[6]<<8)|r[7]);
|
||
u32 spa = ((u32)r[14]<<24)|((u32)r[15]<<16)|((u32)r[16]<<8)|r[17];
|
||
if (op == 2 && spa == target_ip) {
|
||
for (int i=0;i<6;i++) out_mac[i]=r[8+i];
|
||
logl("[netman] ARP: gateway resolved");
|
||
return 0;
|
||
}
|
||
}
|
||
delay_spin(300);
|
||
}
|
||
logl("[netman] ARP: timeout");
|
||
return -2;
|
||
}
|
||
|
||
static int dns_query_a(u32 index, const u8 dst_mac[6], u32 dns_ip, const char *name, u32 *out_ip) {
|
||
memset_nm(g_pktbuf, 0, sizeof(g_pktbuf));
|
||
u8 *p = g_pktbuf;
|
||
for (int i=0;i<6;i++) p[i]=dst_mac[i];
|
||
for (int i=0;i<6;i++) p[6+i]=g_local_mac[i];
|
||
p[12]=0x08; p[13]=0x00;
|
||
u8 *ip = p + 14;
|
||
u8 *udp = ip + 20;
|
||
u8 *dns = udp + 8;
|
||
dns[0]=0x12; dns[1]=0x34;
|
||
dns[2]=0x01; dns[3]=0x00;
|
||
dns[4]=0x00; dns[5]=0x01;
|
||
u32 q=12;
|
||
const char *s=name;
|
||
while (*s) {
|
||
const char *dot=s;
|
||
u32 l=0;
|
||
while (dot[l] && dot[l] != '.') l++;
|
||
dns[q++]=(u8)l;
|
||
for (u32 i=0;i<l;i++) dns[q++]=(u8)dot[i];
|
||
s = dot + l;
|
||
if (*s=='.') s++;
|
||
}
|
||
dns[q++]=0;
|
||
dns[q++]=0x00; dns[q++]=0x01;
|
||
dns[q++]=0x00; dns[q++]=0x01;
|
||
u32 dns_len=q;
|
||
u32 udp_len=8+dns_len;
|
||
u32 ip_len=20+udp_len;
|
||
udp[0]=0xC0; udp[1]=0x01;
|
||
udp[2]=0x00; udp[3]=0x35;
|
||
udp[4]=(u8)(udp_len>>8); udp[5]=(u8)udp_len;
|
||
udp[6]=0; udp[7]=0;
|
||
ip[0]=0x45; ip[1]=0;
|
||
ip[2]=(u8)(ip_len>>8); ip[3]=(u8)ip_len;
|
||
ip[4]=0x12; ip[5]=0x34; ip[6]=0; ip[7]=0;
|
||
ip[8]=64; ip[9]=17; ip[10]=0; ip[11]=0;
|
||
ip[12]=(u8)(g_lease_ip>>24); ip[13]=(u8)(g_lease_ip>>16); ip[14]=(u8)(g_lease_ip>>8); ip[15]=(u8)g_lease_ip;
|
||
ip[16]=(u8)(dns_ip>>24); ip[17]=(u8)(dns_ip>>16); ip[18]=(u8)(dns_ip>>8); ip[19]=(u8)dns_ip;
|
||
u16 ipck=ip_checksum(ip,20); ip[10]=(u8)(ipck>>8); ip[11]=(u8)ipck;
|
||
u16 udpck=udp_checksum_ipv4(ip,udp,udp_len); udp[6]=(u8)(udpck>>8); udp[7]=(u8)udpck;
|
||
logl("[netman] DNS: query example.com A");
|
||
if (g_net_tx(index, g_pktbuf, 14+ip_len) != 0) return -1;
|
||
for (u32 attempt=0; attempt<240000; attempt++) {
|
||
u32 rlen=0;
|
||
int rc=g_net_rx_poll(index,g_rxbuf,sizeof(g_rxbuf),&rlen);
|
||
if (rc==0 && rlen>=14+20+8+12 && g_rxbuf[12]==0x08 && g_rxbuf[13]==0x00) {
|
||
const u8 *rip=g_rxbuf+14;
|
||
u32 ihl=(rip[0]&0xf)*4;
|
||
if ((rip[0]>>4)!=4 || rip[9]!=17 || rlen<14+ihl+8+12) { delay_spin(300); continue; }
|
||
const u8 *rudp=rip+ihl;
|
||
if (rudp[2]!=0xC0 || rudp[3]!=0x01) { delay_spin(300); continue; }
|
||
const u8 *rdns=rudp+8;
|
||
if (rdns[0]!=0x12 || rdns[1]!=0x34) { delay_spin(300); continue; }
|
||
u16 an=(u16)((rdns[6]<<8)|rdns[7]);
|
||
u32 off=12;
|
||
while (off<rlen && rdns[off]) off += 1 + rdns[off];
|
||
off += 5;
|
||
for (u16 ai=0; ai<an && off+12<=rlen; ai++) {
|
||
if ((rdns[off]&0xC0)==0xC0) off += 2; else { while (off<rlen && rdns[off]) off += 1 + rdns[off]; off++; }
|
||
if (off+10>rlen) break;
|
||
u16 typ=(u16)((rdns[off]<<8)|rdns[off+1]);
|
||
u16 cls=(u16)((rdns[off+2]<<8)|rdns[off+3]);
|
||
u16 rdlen=(u16)((rdns[off+8]<<8)|rdns[off+9]);
|
||
off += 10;
|
||
if (typ==1 && cls==1 && rdlen==4 && off+4<=rlen) {
|
||
*out_ip=((u32)rdns[off]<<24)|((u32)rdns[off+1]<<16)|((u32)rdns[off+2]<<8)|rdns[off+3];
|
||
log_ipv4("[netman] DNS example.com=", *out_ip);
|
||
return 0;
|
||
}
|
||
off += rdlen;
|
||
}
|
||
}
|
||
delay_spin(300);
|
||
}
|
||
logl("[netman] DNS: timeout");
|
||
return -2;
|
||
}
|
||
|
||
static int network_full_connect(u32 index, const u8 mac[6]) {
|
||
int rc = dhcp_run(index, mac);
|
||
log_hex("[netman] DHCP final rc=", (u64)(i8)rc);
|
||
if (rc != 0) return rc;
|
||
u8 gw_mac[6];
|
||
rc = arp_resolve_gateway(index, g_gateway_ip, gw_mac);
|
||
log_hex("[netman] ARP final rc=", (u64)(i8)rc);
|
||
if (rc != 0) return rc;
|
||
u32 resolved=0;
|
||
rc = dns_query_a(index, gw_mac, g_dns_ip, "example.com", &resolved);
|
||
log_hex("[netman] DNS final rc=", (u64)(i8)rc);
|
||
return rc;
|
||
}
|
||
|
||
static void query_netdevs(const net_conf *cfg) {
|
||
if (!g_net_device_count || !g_net_device_info) {
|
||
logl("[netman] net API not ready; configuration staged only");
|
||
return;
|
||
}
|
||
|
||
u32 count = g_net_device_count();
|
||
int dhcp_done = 0;
|
||
int dhcp_enabled = cfg && !str_eq(cfg->mode, "disabled") && !str_eq(cfg->mode, "off") && !str_eq(cfg->ip, "static");
|
||
log_hex("[netman] netdev_count=", count);
|
||
for (u32 i=0;i<count;i++) {
|
||
nm_net_device_info info;
|
||
memset_nm(&info, 0, sizeof(info));
|
||
int rc = g_net_device_info(i, &info);
|
||
if (rc != 0) {
|
||
log_hex("[netman] netdev info rc=", (u64)(i8)rc);
|
||
continue;
|
||
}
|
||
swrite("[netman] netdev index=");
|
||
log_hex("", i);
|
||
swrite("[netman] name=");
|
||
swrite(info.name ? info.name : "<noname>");
|
||
log_mac(info.mac);
|
||
swrite(" flags=");
|
||
log_hex("", info.flags);
|
||
if (info.flags & DKM_NET_F_LINK_UP) logl("[netman] link=up");
|
||
else logl("[netman] link=down");
|
||
if (info.flags & DKM_NET_F_TX_READY) logl("[netman] tx=ready");
|
||
if (info.flags & DKM_NET_F_RX_READY) logl("[netman] rx=ready");
|
||
|
||
if (!dhcp_done && dhcp_enabled &&
|
||
(info.flags & (DKM_NET_F_LINK_UP|DKM_NET_F_TX_READY|DKM_NET_F_RX_READY))
|
||
== (DKM_NET_F_LINK_UP|DKM_NET_F_TX_READY|DKM_NET_F_RX_READY)) {
|
||
int drc = network_full_connect(i, info.mac);
|
||
log_hex("[netman] network final rc=", (u64)(i8)drc);
|
||
dhcp_done = 1;
|
||
}
|
||
}
|
||
if (!dhcp_enabled) logl("[netman] DHCP skipped by config");
|
||
}
|
||
|
||
__attribute__((visibility("default")))
|
||
void dsk_entry(const dsk_boot_context *ctx) {
|
||
logl("[netman] boot");
|
||
if (!ctx || ctx->magic != DSK_BOOT_MAGIC) { logl("[netman] bad context"); return; }
|
||
logl("[netman] calibrating TSC (实机 CPU 频率计算)");
|
||
tsc_calibrate_nm();
|
||
u64 api = ctx->dkm_kernel_api;
|
||
if (api) {
|
||
u64 block_api = *(u64 *)(api + 0xA8);
|
||
if (block_api) g_block_read = (block_read_fn)*(u64 *)(block_api + 16);
|
||
u64 net_api = *(u64 *)(api + 0x48);
|
||
log_hex("[netman] kernel_api.net=", net_api);
|
||
if (net_api) {
|
||
g_net_device_count = (net_device_count_fn)*(u64 *)(net_api + 8);
|
||
g_net_device_info = (net_device_info_fn)*(u64 *)(net_api + 16);
|
||
g_net_tx = (net_tx_fn)*(u64 *)(net_api + 24);
|
||
g_net_rx_poll = (net_rx_poll_fn)*(u64 *)(net_api + 32);
|
||
log_hex("[netman] net.tx ptr=", (u64)g_net_tx);
|
||
log_hex("[netman] net.rx ptr=", (u64)g_net_rx_poll);
|
||
}
|
||
}
|
||
u32 size=0;
|
||
int rc = fat32_read_root_file("NETCONF CNF", g_conf, sizeof(g_conf), &size);
|
||
log_hex("[netman] conf read rc=", (u64)(i8)rc);
|
||
if (rc != 0) {
|
||
logl("[netman] using built-in defaults");
|
||
g_conf[0]=0;
|
||
} else {
|
||
log_hex("[netman] conf size=", size);
|
||
}
|
||
net_conf cfg;
|
||
memset_nm(&cfg, 0, sizeof(cfg));
|
||
parse_conf(g_conf, &cfg);
|
||
log_cfg(&cfg);
|
||
if (str_eq(cfg.mode, "disabled") || str_eq(cfg.mode, "off")) {
|
||
logl("[netman] networking disabled by config");
|
||
}
|
||
query_netdevs(&cfg);
|
||
logl("[netman] done");
|
||
}
|