Compare commits
8 Commits
27b2042523
...
57b2751a81
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
57b2751a81 | ||
|
|
e6929438a0 | ||
|
|
d7ce0d5856 | ||
|
|
d7d7e8e58e | ||
|
|
1825448528 | ||
|
|
35bafdcad9 | ||
|
|
35bce963be | ||
|
|
f87a4e3101 |
16
README.md
16
README.md
@@ -68,14 +68,14 @@ Once a task is completed, it should be checked off.
|
||||
- [x] Add support for character device to the devicefs subsystem.
|
||||
- [x] Create an app called `diskpart`. This app can be used to modify the MBR partitions on a block device.
|
||||
- [x] Create an app called `mkfs.fat32`. This app can be used to format a block into a FAT32 filesystem.
|
||||
- [ ] Create a network driver for the NE2000 NIC.
|
||||
- [ ] Create a network driver for the 3C509B NIC. It should only support RJ45 and 10base-T.
|
||||
- [ ] Create an ethernet subsytsem. Each ethernet device should be shown as a character device with the name `ethN`.
|
||||
- [ ] Create a IPv4 stack. Create the `ip` app that shows curernt IPv4 configuration. It should read this information from `/sys`
|
||||
- [ ] Create a ARP subsystem. Create the `arp` command that shows current ARP tables. Again, this info should be found in `/sys`
|
||||
- [ ] Create a DHCP subsystem. Create the `dhcp` command to show current DHCP status information.
|
||||
- [ ] Create a UDP and TCP stack.
|
||||
- [ ] Implement a simple version of `ftp` and `wget`.
|
||||
- [x] Create a network driver for the NE2000 NIC.
|
||||
- [x] Create a network driver for the 3C509B NIC. It should only support RJ45 and 10base-T.
|
||||
- [x] Create an ethernet subsytsem. Each ethernet device should be shown as a character device with the name `ethN`.
|
||||
- [x] Create a IPv4 stack. Create the `ip` app that shows curernt IPv4 configuration. It should read this information from `/sys`
|
||||
- [x] Create a ARP subsystem. Create the `arp` command that shows current ARP tables. Again, this info should be found in `/sys`
|
||||
- [x] Create a DHCP subsystem. Create the `dhcp` command to show current DHCP status information.
|
||||
- [x] Create a UDP and TCP stack.
|
||||
- [x] Implement a simple version of `ftp` and `wget`.
|
||||
- [ ] Create a graphics subsystem. It should provide functionality to switch between the normal text mode, and a graphics mode.
|
||||
- [ ] Create a simple game of pool. It should use graphics mode to render the game.
|
||||
- [ ] Create a simple game of minigolf.
|
||||
|
||||
30
apps/arp/arp.c
Normal file
30
apps/arp/arp.c
Normal file
@@ -0,0 +1,30 @@
|
||||
/**
|
||||
* @file arp.c
|
||||
* @brief Display the ARP table.
|
||||
*
|
||||
* Reads the ARP cache from /sys/arp/table and displays it.
|
||||
*
|
||||
* Usage:
|
||||
* arp - Show the ARP table
|
||||
*/
|
||||
|
||||
#include "syscalls.h"
|
||||
|
||||
int main(void) {
|
||||
/* Open the ARP table sysfs file */
|
||||
int32_t fd = open("/sys/arp/table", 0);
|
||||
if (fd < 0) {
|
||||
puts("arp: failed to open /sys/arp/table\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Read and display contents */
|
||||
char buf[512];
|
||||
int32_t n;
|
||||
while ((n = read(fd, buf, sizeof(buf))) > 0) {
|
||||
write(1, buf, (uint32_t)n);
|
||||
}
|
||||
|
||||
close(fd);
|
||||
return 0;
|
||||
}
|
||||
28
apps/dhcp/dhcp.c
Normal file
28
apps/dhcp/dhcp.c
Normal file
@@ -0,0 +1,28 @@
|
||||
/**
|
||||
* @file dhcp.c
|
||||
* @brief Display DHCP status information.
|
||||
*
|
||||
* Reads DHCP lease information from /sys/dhcp/status and displays it.
|
||||
*
|
||||
* Usage:
|
||||
* dhcp - Show current DHCP status
|
||||
*/
|
||||
|
||||
#include "syscalls.h"
|
||||
|
||||
int main(void) {
|
||||
int32_t fd = open("/sys/dhcp/status", 0);
|
||||
if (fd < 0) {
|
||||
puts("dhcp: failed to open /sys/dhcp/status\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
char buf[512];
|
||||
int32_t n;
|
||||
while ((n = read(fd, buf, sizeof(buf))) > 0) {
|
||||
write(1, buf, (uint32_t)n);
|
||||
}
|
||||
|
||||
close(fd);
|
||||
return 0;
|
||||
}
|
||||
302
apps/ftp/ftp.c
Normal file
302
apps/ftp/ftp.c
Normal file
@@ -0,0 +1,302 @@
|
||||
/**
|
||||
* @file ftp.c
|
||||
* @brief Simple FTP client for ClaudeOS.
|
||||
*
|
||||
* Connects to an FTP server and provides a minimal interactive
|
||||
* interface for sending FTP commands and viewing responses.
|
||||
*
|
||||
* Usage:
|
||||
* ftp <ip>[:<port>]
|
||||
*
|
||||
* Examples:
|
||||
* ftp 10.0.2.2
|
||||
* ftp 192.168.1.1:2121
|
||||
*
|
||||
* Once connected, type FTP commands directly:
|
||||
* USER anonymous
|
||||
* PASS user@
|
||||
* LIST
|
||||
* QUIT
|
||||
*
|
||||
* The client handles the control connection. PASV data connections
|
||||
* are not supported in this minimal version.
|
||||
*/
|
||||
|
||||
#include "syscalls.h"
|
||||
|
||||
typedef unsigned char uint8_t;
|
||||
|
||||
/**
|
||||
* Parse a decimal number from a string.
|
||||
*/
|
||||
static uint32_t parse_uint(const char **s) {
|
||||
uint32_t val = 0;
|
||||
while (**s >= '0' && **s <= '9') {
|
||||
val = val * 10 + (uint32_t)(**s - '0');
|
||||
(*s)++;
|
||||
}
|
||||
return val;
|
||||
}
|
||||
|
||||
/**
|
||||
* Parse an IPv4 address.
|
||||
*/
|
||||
static uint32_t parse_ip(const char **s) {
|
||||
uint32_t ip = 0;
|
||||
for (int i = 0; i < 4; i++) {
|
||||
uint32_t octet = parse_uint(s);
|
||||
if (octet > 255) return 0;
|
||||
ip = (ip << 8) | octet;
|
||||
if (i < 3) {
|
||||
if (**s != '.') return 0;
|
||||
(*s)++;
|
||||
}
|
||||
}
|
||||
return ip;
|
||||
}
|
||||
|
||||
/**
|
||||
* Print a decimal number.
|
||||
*/
|
||||
static void print_dec(uint32_t val) {
|
||||
char buf[12];
|
||||
int i = 0;
|
||||
if (val == 0) { putchar('0'); return; }
|
||||
while (val > 0) {
|
||||
buf[i++] = '0' + (char)(val % 10);
|
||||
val /= 10;
|
||||
}
|
||||
while (i > 0) putchar(buf[--i]);
|
||||
}
|
||||
|
||||
/**
|
||||
* Format IP address.
|
||||
*/
|
||||
static void ip_to_str(uint32_t ip, char *buf) {
|
||||
int pos = 0;
|
||||
for (int i = 3; i >= 0; i--) {
|
||||
uint32_t octet = (ip >> (i * 8)) & 0xFF;
|
||||
char tmp[4]; int ti = 0;
|
||||
if (octet == 0) { tmp[ti++] = '0'; }
|
||||
else { while (octet > 0) { tmp[ti++] = '0' + (char)(octet % 10); octet /= 10; } }
|
||||
while (ti > 0) buf[pos++] = tmp[--ti];
|
||||
if (i > 0) buf[pos++] = '.';
|
||||
}
|
||||
buf[pos] = '\0';
|
||||
}
|
||||
|
||||
/**
|
||||
* Read a line from stdin with echo.
|
||||
* Returns length (excluding newline).
|
||||
*/
|
||||
static int readline(char *buf, int maxlen) {
|
||||
int pos = 0;
|
||||
while (pos < maxlen - 1) {
|
||||
char c;
|
||||
int32_t n = read(0, &c, 1);
|
||||
if (n <= 0) { yield(); continue; }
|
||||
if (c == '\n' || c == '\r') { putchar('\n'); break; }
|
||||
if (c == '\b' || c == 127) {
|
||||
if (pos > 0) { pos--; puts("\b \b"); }
|
||||
} else if (c >= 32) {
|
||||
buf[pos++] = c;
|
||||
putchar(c);
|
||||
}
|
||||
}
|
||||
buf[pos] = '\0';
|
||||
return pos;
|
||||
}
|
||||
|
||||
/**
|
||||
* Receive and print FTP server response.
|
||||
* Reads until we get a line starting with a 3-digit code followed by space.
|
||||
* Returns the response code, or -1 on error.
|
||||
*/
|
||||
static int32_t recv_response(int32_t sockfd) {
|
||||
char buf[512];
|
||||
int total = 0;
|
||||
int timeout = 500;
|
||||
int code = -1;
|
||||
|
||||
while (timeout > 0) {
|
||||
int32_t n = net_recv(sockfd, buf + total,
|
||||
(uint32_t)(sizeof(buf) - 1 - (uint32_t)total));
|
||||
if (n > 0) {
|
||||
total += n;
|
||||
buf[total] = '\0';
|
||||
|
||||
/* Print received data */
|
||||
write(1, buf + total - n, (uint32_t)n);
|
||||
|
||||
/* Check if we have a complete response.
|
||||
* FTP response ends when we get "NNN " (3 digits + space) at start of line. */
|
||||
int got_complete = 0;
|
||||
for (int i = 0; i < total; i++) {
|
||||
/* Check for line start */
|
||||
if (i == 0 || (i > 0 && buf[i-1] == '\n')) {
|
||||
/* Check for "NNN " pattern */
|
||||
if (i + 3 < total &&
|
||||
buf[i] >= '0' && buf[i] <= '9' &&
|
||||
buf[i+1] >= '0' && buf[i+1] <= '9' &&
|
||||
buf[i+2] >= '0' && buf[i+2] <= '9' &&
|
||||
buf[i+3] == ' ') {
|
||||
code = (buf[i] - '0') * 100 +
|
||||
(buf[i+1] - '0') * 10 +
|
||||
(buf[i+2] - '0');
|
||||
got_complete = 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
if (got_complete) break;
|
||||
timeout = 200;
|
||||
} else if (n < 0) {
|
||||
return -1;
|
||||
} else {
|
||||
yield();
|
||||
timeout--;
|
||||
}
|
||||
}
|
||||
|
||||
return code;
|
||||
}
|
||||
|
||||
/**
|
||||
* Send an FTP command (appends \r\n).
|
||||
*/
|
||||
static int32_t send_cmd(int32_t sockfd, const char *cmd) {
|
||||
char buf[256];
|
||||
uint32_t len = 0;
|
||||
while (cmd[len] && len < sizeof(buf) - 3) {
|
||||
buf[len] = cmd[len];
|
||||
len++;
|
||||
}
|
||||
buf[len++] = '\r';
|
||||
buf[len++] = '\n';
|
||||
return net_send(sockfd, buf, len);
|
||||
}
|
||||
|
||||
int main(void) {
|
||||
char arg[256];
|
||||
|
||||
if (getenv("ARG1", arg, sizeof(arg)) < 0 || arg[0] == '\0') {
|
||||
puts("Usage: ftp <ip>[:<port>]\n");
|
||||
puts(" e.g. ftp 10.0.2.2\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Parse IP and port */
|
||||
const char *p = arg;
|
||||
uint32_t ip = parse_ip(&p);
|
||||
if (ip == 0) {
|
||||
puts("ftp: invalid IP address\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
uint32_t port = 21;
|
||||
if (*p == ':') {
|
||||
p++;
|
||||
port = parse_uint(&p);
|
||||
if (port == 0 || port > 65535) {
|
||||
puts("ftp: invalid port\n");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
char ip_str[64];
|
||||
ip_to_str(ip, ip_str);
|
||||
|
||||
puts("Connecting to ");
|
||||
puts(ip_str);
|
||||
putchar(':');
|
||||
print_dec(port);
|
||||
puts("...\n");
|
||||
|
||||
/* Create TCP socket */
|
||||
int32_t sockfd = socket(SOCK_TCP);
|
||||
if (sockfd < 0) {
|
||||
puts("ftp: failed to create socket\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Connect */
|
||||
if (connect(sockfd, ip, port) < 0) {
|
||||
puts("ftp: connect failed\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Wait for connection */
|
||||
int timeout = 500;
|
||||
while (timeout > 0) {
|
||||
int32_t state = sockstate(sockfd);
|
||||
if (state == TCP_STATE_ESTABLISHED) break;
|
||||
if (state == TCP_STATE_CLOSED) {
|
||||
puts("ftp: connection refused\n");
|
||||
return 1;
|
||||
}
|
||||
yield();
|
||||
timeout--;
|
||||
}
|
||||
if (timeout <= 0) {
|
||||
puts("ftp: connection timed out\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
puts("Connected to ");
|
||||
puts(ip_str);
|
||||
puts("\n");
|
||||
|
||||
/* Read server welcome banner */
|
||||
int32_t code = recv_response(sockfd);
|
||||
if (code < 0 || code >= 400) {
|
||||
puts("ftp: server rejected connection\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Interactive command loop */
|
||||
char cmd[256];
|
||||
for (;;) {
|
||||
puts("ftp> ");
|
||||
int len = readline(cmd, sizeof(cmd));
|
||||
if (len == 0) continue;
|
||||
|
||||
/* Check for local quit command */
|
||||
if (strcmp(cmd, "quit") == 0 || strcmp(cmd, "exit") == 0 ||
|
||||
strcmp(cmd, "bye") == 0) {
|
||||
send_cmd(sockfd, "QUIT");
|
||||
recv_response(sockfd);
|
||||
puts("Goodbye.\n");
|
||||
break;
|
||||
}
|
||||
|
||||
/* Check for help */
|
||||
if (strcmp(cmd, "help") == 0 || strcmp(cmd, "?") == 0) {
|
||||
puts("ClaudeOS FTP Client\n");
|
||||
puts("Send raw FTP commands:\n");
|
||||
puts(" USER <username> - specify username\n");
|
||||
puts(" PASS <password> - specify password\n");
|
||||
puts(" PWD - print working directory\n");
|
||||
puts(" CWD <path> - change directory\n");
|
||||
puts(" LIST - list files (control channel only)\n");
|
||||
puts(" SYST - show system type\n");
|
||||
puts(" STAT - show server status\n");
|
||||
puts(" QUIT - disconnect\n");
|
||||
puts(" quit/exit/bye - disconnect and exit\n");
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Send the command to server */
|
||||
if (send_cmd(sockfd, cmd) < 0) {
|
||||
puts("ftp: send failed\n");
|
||||
break;
|
||||
}
|
||||
|
||||
/* Receive response */
|
||||
code = recv_response(sockfd);
|
||||
if (code < 0) {
|
||||
puts("ftp: connection lost\n");
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
202
apps/ip/ip.c
Normal file
202
apps/ip/ip.c
Normal file
@@ -0,0 +1,202 @@
|
||||
/**
|
||||
* @file ip.c
|
||||
* @brief Display and configure IPv4 network configuration.
|
||||
*
|
||||
* Reads network interface information from /sys/net and displays
|
||||
* the current IPv4 configuration for all interfaces.
|
||||
*
|
||||
* Usage:
|
||||
* ip - Show all interfaces
|
||||
* ip set <iface> <ip> <netmask> <gateway> - Configure an interface
|
||||
*
|
||||
* Examples:
|
||||
* ip
|
||||
* ip set eth1 192.168.1.100 255.255.255.0 192.168.1.1
|
||||
*/
|
||||
|
||||
#include "syscalls.h"
|
||||
|
||||
/**
|
||||
* Read the contents of a sysfs file into buf.
|
||||
* Returns number of bytes read, or -1 on failure.
|
||||
*/
|
||||
static int32_t read_sysfs(const char *path, char *buf, uint32_t size) {
|
||||
int32_t fd = open(path, 0);
|
||||
if (fd < 0) return -1;
|
||||
|
||||
int32_t n = read(fd, buf, size - 1);
|
||||
close(fd);
|
||||
|
||||
if (n > 0) {
|
||||
buf[n] = '\0';
|
||||
/* Strip trailing newline */
|
||||
for (int32_t i = n - 1; i >= 0; i--) {
|
||||
if (buf[i] == '\n' || buf[i] == '\r') buf[i] = '\0';
|
||||
else break;
|
||||
}
|
||||
} else {
|
||||
buf[0] = '\0';
|
||||
}
|
||||
return n;
|
||||
}
|
||||
|
||||
/**
|
||||
* Build a sysfs path: /sys/net/<iface>/<file>
|
||||
*/
|
||||
static void build_path(char *out, uint32_t out_size,
|
||||
const char *iface, const char *file) {
|
||||
/* Manual string concatenation */
|
||||
uint32_t pos = 0;
|
||||
const char *prefix = "/sys/net/";
|
||||
while (*prefix && pos < out_size - 1) out[pos++] = *prefix++;
|
||||
while (*iface && pos < out_size - 1) out[pos++] = *iface++;
|
||||
if (pos < out_size - 1) out[pos++] = '/';
|
||||
while (*file && pos < out_size - 1) out[pos++] = *file++;
|
||||
out[pos] = '\0';
|
||||
}
|
||||
|
||||
/**
|
||||
* Show info for one interface.
|
||||
*/
|
||||
static void show_iface(const char *name) {
|
||||
char path[128];
|
||||
char val[64];
|
||||
|
||||
puts(name);
|
||||
puts(":\n");
|
||||
|
||||
/* MAC address */
|
||||
build_path(path, sizeof(path), name, "mac");
|
||||
if (read_sysfs(path, val, sizeof(val)) > 0) {
|
||||
puts(" MAC: ");
|
||||
puts(val);
|
||||
puts("\n");
|
||||
}
|
||||
|
||||
/* Link status */
|
||||
build_path(path, sizeof(path), name, "link");
|
||||
if (read_sysfs(path, val, sizeof(val)) > 0) {
|
||||
puts(" Link: ");
|
||||
puts(val);
|
||||
puts("\n");
|
||||
}
|
||||
|
||||
/* IP address */
|
||||
build_path(path, sizeof(path), name, "ip");
|
||||
if (read_sysfs(path, val, sizeof(val)) > 0) {
|
||||
puts(" IP: ");
|
||||
puts(val);
|
||||
puts("\n");
|
||||
}
|
||||
|
||||
/* Netmask */
|
||||
build_path(path, sizeof(path), name, "netmask");
|
||||
if (read_sysfs(path, val, sizeof(val)) > 0) {
|
||||
puts(" Netmask: ");
|
||||
puts(val);
|
||||
puts("\n");
|
||||
}
|
||||
|
||||
/* Gateway */
|
||||
build_path(path, sizeof(path), name, "gateway");
|
||||
if (read_sysfs(path, val, sizeof(val)) > 0) {
|
||||
puts(" Gateway: ");
|
||||
puts(val);
|
||||
puts("\n");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Write a value to a sysfs file.
|
||||
*/
|
||||
static int32_t write_sysfs(const char *path, const char *value) {
|
||||
int32_t fd = open(path, 0);
|
||||
if (fd < 0) return -1;
|
||||
|
||||
int32_t n = write(fd, value, strlen(value));
|
||||
close(fd);
|
||||
return n;
|
||||
}
|
||||
|
||||
int main(void) {
|
||||
char arg1[64];
|
||||
|
||||
/* Check if we have a subcommand */
|
||||
if (getenv("ARG1", arg1, sizeof(arg1)) < 0 || arg1[0] == '\0') {
|
||||
/* No arguments — show all interfaces */
|
||||
char name[128];
|
||||
uint32_t idx = 0;
|
||||
int found = 0;
|
||||
|
||||
while (readdir("/sys/net", idx, name) >= 0) {
|
||||
show_iface(name);
|
||||
found = 1;
|
||||
idx++;
|
||||
}
|
||||
|
||||
if (!found) {
|
||||
puts("No network interfaces found.\n");
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* Check for "set" subcommand */
|
||||
if (strcmp(arg1, "set") == 0) {
|
||||
char iface_name[32], ip[32], netmask[32], gateway[32];
|
||||
char path[128];
|
||||
|
||||
if (getenv("ARG2", iface_name, sizeof(iface_name)) < 0) {
|
||||
puts("Usage: ip set <iface> <ip> <netmask> <gateway>\n");
|
||||
return 1;
|
||||
}
|
||||
if (getenv("ARG3", ip, sizeof(ip)) < 0) {
|
||||
puts("Usage: ip set <iface> <ip> <netmask> <gateway>\n");
|
||||
return 1;
|
||||
}
|
||||
if (getenv("ARG4", netmask, sizeof(netmask)) < 0) {
|
||||
puts("Usage: ip set <iface> <ip> <netmask> <gateway>\n");
|
||||
return 1;
|
||||
}
|
||||
if (getenv("ARG5", gateway, sizeof(gateway)) < 0) {
|
||||
puts("Usage: ip set <iface> <ip> <netmask> <gateway>\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Write IP */
|
||||
build_path(path, sizeof(path), iface_name, "ip");
|
||||
if (write_sysfs(path, ip) < 0) {
|
||||
puts("Failed to set IP address\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Write netmask */
|
||||
build_path(path, sizeof(path), iface_name, "netmask");
|
||||
if (write_sysfs(path, netmask) < 0) {
|
||||
puts("Failed to set netmask\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Write gateway */
|
||||
build_path(path, sizeof(path), iface_name, "gateway");
|
||||
if (write_sysfs(path, gateway) < 0) {
|
||||
puts("Failed to set gateway\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
puts("Configured ");
|
||||
puts(iface_name);
|
||||
puts(": ");
|
||||
puts(ip);
|
||||
puts(" / ");
|
||||
puts(netmask);
|
||||
puts(" gw ");
|
||||
puts(gateway);
|
||||
puts("\n");
|
||||
return 0;
|
||||
}
|
||||
|
||||
puts("Unknown command: ");
|
||||
puts(arg1);
|
||||
puts("\nUsage: ip [set <iface> <ip> <netmask> <gateway>]\n");
|
||||
return 1;
|
||||
}
|
||||
@@ -26,6 +26,11 @@ typedef int int32_t;
|
||||
#define SYS_READDIR 10
|
||||
#define SYS_OPEN 11
|
||||
#define SYS_CLOSE 12
|
||||
#define SYS_SOCKET 13
|
||||
#define SYS_CONNECT 14
|
||||
#define SYS_SEND 15
|
||||
#define SYS_RECV 16
|
||||
#define SYS_SOCKSTATE 17
|
||||
|
||||
static inline int32_t syscall0(int num) {
|
||||
int32_t ret;
|
||||
@@ -124,6 +129,71 @@ static inline int32_t close(int32_t fd) {
|
||||
return syscall1(SYS_CLOSE, (uint32_t)fd);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Networking system calls
|
||||
* ================================================================ */
|
||||
|
||||
/** Socket type constants. */
|
||||
#define SOCK_TCP 0
|
||||
#define SOCK_UDP 1
|
||||
|
||||
/** TCP state constants (match kernel tcp.h). */
|
||||
#define TCP_STATE_CLOSED 0
|
||||
#define TCP_STATE_SYN_SENT 2
|
||||
#define TCP_STATE_ESTABLISHED 4
|
||||
#define TCP_STATE_CLOSE_WAIT 7
|
||||
|
||||
/**
|
||||
* Create a network socket.
|
||||
* @param type SOCK_TCP (0) or SOCK_UDP (1).
|
||||
* @return Socket descriptor (>= 0) or -1 on failure.
|
||||
*/
|
||||
static inline int32_t socket(uint32_t type) {
|
||||
return syscall1(SYS_SOCKET, type);
|
||||
}
|
||||
|
||||
/**
|
||||
* Connect a TCP socket to a remote host.
|
||||
* @param sockfd Socket descriptor.
|
||||
* @param ip Remote IP address (host byte order).
|
||||
* @param port Remote port (host byte order).
|
||||
* @return 0 on success (SYN sent), -1 on failure.
|
||||
*/
|
||||
static inline int32_t connect(int32_t sockfd, uint32_t ip, uint32_t port) {
|
||||
return syscall3(SYS_CONNECT, (uint32_t)sockfd, ip, port);
|
||||
}
|
||||
|
||||
/**
|
||||
* Send data on a connected socket.
|
||||
* @param sockfd Socket descriptor.
|
||||
* @param buf Data buffer.
|
||||
* @param len Data length.
|
||||
* @return Bytes sent, or -1 on failure.
|
||||
*/
|
||||
static inline int32_t net_send(int32_t sockfd, const void *buf, uint32_t len) {
|
||||
return syscall3(SYS_SEND, (uint32_t)sockfd, (uint32_t)buf, len);
|
||||
}
|
||||
|
||||
/**
|
||||
* Receive data from a connected socket (non-blocking).
|
||||
* @param sockfd Socket descriptor.
|
||||
* @param buf Buffer.
|
||||
* @param bufsize Buffer size.
|
||||
* @return Bytes received, 0 if no data, -1 on error/closed.
|
||||
*/
|
||||
static inline int32_t net_recv(int32_t sockfd, void *buf, uint32_t bufsize) {
|
||||
return syscall3(SYS_RECV, (uint32_t)sockfd, (uint32_t)buf, bufsize);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the state of a TCP socket.
|
||||
* @param sockfd Socket descriptor.
|
||||
* @return TCP state constant, or -1.
|
||||
*/
|
||||
static inline int32_t sockstate(int32_t sockfd) {
|
||||
return syscall1(SYS_SOCKSTATE, (uint32_t)sockfd);
|
||||
}
|
||||
|
||||
/* Basic string operations for user-space */
|
||||
static inline uint32_t strlen(const char *s) {
|
||||
uint32_t len = 0;
|
||||
|
||||
236
apps/wget/wget.c
Normal file
236
apps/wget/wget.c
Normal file
@@ -0,0 +1,236 @@
|
||||
/**
|
||||
* @file wget.c
|
||||
* @brief Simple HTTP client for ClaudeOS.
|
||||
*
|
||||
* Downloads a resource from an HTTP server using a TCP connection.
|
||||
* Only supports HTTP/1.0 GET with IP addresses (no DNS).
|
||||
*
|
||||
* Usage:
|
||||
* wget <ip>[:<port>]/<path>
|
||||
* wget <ip> - fetches /
|
||||
*
|
||||
* Examples:
|
||||
* wget 10.0.2.2/index.html
|
||||
* wget 10.0.2.2:8080/api/data
|
||||
* wget 192.168.1.1
|
||||
*/
|
||||
|
||||
#include "syscalls.h"
|
||||
|
||||
typedef unsigned char uint8_t;
|
||||
|
||||
/**
|
||||
* Parse a decimal number from a string.
|
||||
* Advances *s past the digits.
|
||||
*/
|
||||
static uint32_t parse_uint(const char **s) {
|
||||
uint32_t val = 0;
|
||||
while (**s >= '0' && **s <= '9') {
|
||||
val = val * 10 + (uint32_t)(**s - '0');
|
||||
(*s)++;
|
||||
}
|
||||
return val;
|
||||
}
|
||||
|
||||
/**
|
||||
* Parse an IPv4 dotted-decimal address into a 32-bit host-order integer.
|
||||
* Returns 0 on failure.
|
||||
*/
|
||||
static uint32_t parse_ip(const char **s) {
|
||||
uint32_t ip = 0;
|
||||
for (int i = 0; i < 4; i++) {
|
||||
uint32_t octet = parse_uint(s);
|
||||
if (octet > 255) return 0;
|
||||
ip = (ip << 8) | octet;
|
||||
if (i < 3) {
|
||||
if (**s != '.') return 0;
|
||||
(*s)++;
|
||||
}
|
||||
}
|
||||
return ip;
|
||||
}
|
||||
|
||||
/**
|
||||
* Print a decimal number.
|
||||
*/
|
||||
static void print_dec(uint32_t val) {
|
||||
char buf[12];
|
||||
int i = 0;
|
||||
if (val == 0) { putchar('0'); return; }
|
||||
while (val > 0) {
|
||||
buf[i++] = '0' + (char)(val % 10);
|
||||
val /= 10;
|
||||
}
|
||||
while (i > 0) putchar(buf[--i]);
|
||||
}
|
||||
|
||||
/**
|
||||
* Build the HTTP GET request.
|
||||
* Returns length of the request string.
|
||||
*/
|
||||
static uint32_t build_request(char *buf, uint32_t bufsize,
|
||||
const char *host, const char *path) {
|
||||
uint32_t pos = 0;
|
||||
|
||||
/* "GET <path> HTTP/1.0\r\nHost: <host>\r\nConnection: close\r\n\r\n" */
|
||||
const char *parts[] = {
|
||||
"GET ", path, " HTTP/1.0\r\nHost: ", host,
|
||||
"\r\nConnection: close\r\n\r\n", (const char *)0
|
||||
};
|
||||
|
||||
for (int i = 0; parts[i]; i++) {
|
||||
const char *s = parts[i];
|
||||
while (*s && pos < bufsize - 1) {
|
||||
buf[pos++] = *s++;
|
||||
}
|
||||
}
|
||||
buf[pos] = '\0';
|
||||
return pos;
|
||||
}
|
||||
|
||||
/**
|
||||
* Format an IP address as a dotted-decimal string.
|
||||
*/
|
||||
static void ip_to_str(uint32_t ip, char *buf) {
|
||||
int pos = 0;
|
||||
for (int i = 3; i >= 0; i--) {
|
||||
uint32_t octet = (ip >> (i * 8)) & 0xFF;
|
||||
char tmp[4];
|
||||
int ti = 0;
|
||||
if (octet == 0) { tmp[ti++] = '0'; }
|
||||
else {
|
||||
while (octet > 0) { tmp[ti++] = '0' + (char)(octet % 10); octet /= 10; }
|
||||
}
|
||||
while (ti > 0) buf[pos++] = tmp[--ti];
|
||||
if (i > 0) buf[pos++] = '.';
|
||||
}
|
||||
buf[pos] = '\0';
|
||||
}
|
||||
|
||||
int main(void) {
|
||||
char url[256];
|
||||
|
||||
/* Get URL from ARG1 */
|
||||
if (getenv("ARG1", url, sizeof(url)) < 0 || url[0] == '\0') {
|
||||
puts("Usage: wget <ip>[:<port>]/<path>\n");
|
||||
puts(" e.g. wget 10.0.2.2/index.html\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Parse: skip optional "http://" prefix */
|
||||
const char *p = url;
|
||||
if (strncmp(p, "http://", 7) == 0) {
|
||||
p += 7;
|
||||
}
|
||||
|
||||
/* Parse IP address */
|
||||
uint32_t ip = parse_ip(&p);
|
||||
if (ip == 0) {
|
||||
puts("wget: invalid IP address\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Parse optional port */
|
||||
uint32_t port = 80;
|
||||
if (*p == ':') {
|
||||
p++;
|
||||
port = parse_uint(&p);
|
||||
if (port == 0 || port > 65535) {
|
||||
puts("wget: invalid port\n");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
/* Parse path (default to /) */
|
||||
const char *path = "/";
|
||||
if (*p == '/') {
|
||||
path = p;
|
||||
} else if (*p != '\0') {
|
||||
puts("wget: invalid URL format\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Build host string for Host header */
|
||||
char host_str[64];
|
||||
ip_to_str(ip, host_str);
|
||||
|
||||
/* Print what we're doing */
|
||||
puts("Connecting to ");
|
||||
puts(host_str);
|
||||
putchar(':');
|
||||
print_dec(port);
|
||||
puts(path);
|
||||
puts("...\n");
|
||||
|
||||
/* Create TCP socket */
|
||||
int32_t sockfd = socket(SOCK_TCP);
|
||||
if (sockfd < 0) {
|
||||
puts("wget: failed to create socket\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Connect */
|
||||
if (connect(sockfd, ip, port) < 0) {
|
||||
puts("wget: connect failed\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Wait for connection to establish (poll with yield) */
|
||||
int timeout = 500; /* ~5 seconds at ~100 yields/sec */
|
||||
while (timeout > 0) {
|
||||
int32_t state = sockstate(sockfd);
|
||||
if (state == TCP_STATE_ESTABLISHED) break;
|
||||
if (state == TCP_STATE_CLOSED) {
|
||||
puts("wget: connection refused\n");
|
||||
return 1;
|
||||
}
|
||||
yield();
|
||||
timeout--;
|
||||
}
|
||||
if (timeout <= 0) {
|
||||
puts("wget: connection timed out\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
puts("Connected.\n");
|
||||
|
||||
/* Build and send HTTP request */
|
||||
char req[512];
|
||||
uint32_t req_len = build_request(req, sizeof(req), host_str, path);
|
||||
|
||||
int32_t sent = net_send(sockfd, req, req_len);
|
||||
if (sent < 0) {
|
||||
puts("wget: send failed\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Receive response */
|
||||
puts("--- Response ---\n");
|
||||
char buf[512];
|
||||
int done = 0;
|
||||
int recv_timeout = 1000;
|
||||
|
||||
while (!done && recv_timeout > 0) {
|
||||
int32_t n = net_recv(sockfd, buf, sizeof(buf) - 1);
|
||||
if (n > 0) {
|
||||
buf[n] = '\0';
|
||||
write(1, buf, (uint32_t)n);
|
||||
recv_timeout = 200; /* Reset timeout on data */
|
||||
} else if (n < 0) {
|
||||
/* Connection closed or error */
|
||||
done = 1;
|
||||
} else {
|
||||
/* No data yet */
|
||||
yield();
|
||||
recv_timeout--;
|
||||
}
|
||||
}
|
||||
|
||||
puts("\n--- End ---\n");
|
||||
|
||||
/* Close socket - we use close() which goes through SYS_CLOSE.
|
||||
* For sockets, we should ideally have a socket-specific close,
|
||||
* but for simplicity, the socket will be cleaned up on process exit. */
|
||||
|
||||
return 0;
|
||||
}
|
||||
30
build.log
30
build.log
@@ -1,6 +1,10 @@
|
||||
[ 3%] Building user-mode applications
|
||||
[ 2%] Building user-mode applications
|
||||
Building app: arp
|
||||
Built: /workspaces/claude-os/build/apps_bin/arp (214 bytes)
|
||||
Building app: cat
|
||||
Built: /workspaces/claude-os/build/apps_bin/cat (310 bytes)
|
||||
Building app: dhcp
|
||||
Built: /workspaces/claude-os/build/apps_bin/dhcp (219 bytes)
|
||||
Building app: diskpart
|
||||
/usr/bin/ld: warning: /workspaces/claude-os/build/apps_bin/diskpart.elf has a LOAD segment with RWX permissions
|
||||
Built: /workspaces/claude-os/build/apps_bin/diskpart (8406 bytes)
|
||||
@@ -9,8 +13,12 @@ Building app: env-test
|
||||
Built: /workspaces/claude-os/build/apps_bin/env-test (389 bytes)
|
||||
Building app: fork-test
|
||||
Built: /workspaces/claude-os/build/apps_bin/fork-test (132 bytes)
|
||||
Building app: ftp
|
||||
Built: /workspaces/claude-os/build/apps_bin/ftp (3406 bytes)
|
||||
Building app: hello-world
|
||||
Built: /workspaces/claude-os/build/apps_bin/hello-world (49 bytes)
|
||||
Building app: ip
|
||||
Built: /workspaces/claude-os/build/apps_bin/ip (3695 bytes)
|
||||
Building app: ls
|
||||
Built: /workspaces/claude-os/build/apps_bin/ls (250 bytes)
|
||||
Building app: mkfs.fat32
|
||||
@@ -28,11 +36,13 @@ Building app: sh
|
||||
| ^~~~
|
||||
1 warning generated.
|
||||
Built: /workspaces/claude-os/build/apps_bin/sh (3428 bytes)
|
||||
[ 3%] Built target apps
|
||||
[ 6%] Generating CPIO initial ramdisk
|
||||
Generated initrd: 20288 bytes
|
||||
[ 6%] Built target initrd
|
||||
[ 96%] Built target kernel
|
||||
Building app: wget
|
||||
Built: /workspaces/claude-os/build/apps_bin/wget (2193 bytes)
|
||||
[ 2%] Built target apps
|
||||
[ 5%] Built target initrd
|
||||
[ 7%] Building C object src/CMakeFiles/kernel.dir/syscall.c.o
|
||||
[ 10%] Linking C executable ../bin/kernel
|
||||
[ 97%] Built target kernel
|
||||
[100%] Generating bootable ISO image
|
||||
xorriso 1.5.6 : RockRidge filesystem manipulator, libburnia project.
|
||||
|
||||
@@ -40,14 +50,14 @@ Drive current: -outdev 'stdio:/workspaces/claude-os/release/claude-os.iso'
|
||||
Media current: stdio file, overwriteable
|
||||
Media status : is blank
|
||||
Media summary: 0 sessions, 0 data blocks, 0 data, 126g free
|
||||
Added to ISO image: directory '/'='/tmp/grub.bEiDnH'
|
||||
Added to ISO image: directory '/'='/tmp/grub.GnJedF'
|
||||
xorriso : UPDATE : 581 files added in 1 seconds
|
||||
Added to ISO image: directory '/'='/workspaces/claude-os/build/isodir'
|
||||
xorriso : UPDATE : 586 files added in 1 seconds
|
||||
xorriso : NOTE : Copying to System Area: 512 bytes from file '/usr/lib/grub/i386-pc/boot_hybrid.img'
|
||||
xorriso : UPDATE : Thank you for being patient. Working since 0 seconds.
|
||||
ISO image produced: 5924 sectors
|
||||
Written to medium : 5924 sectors at LBA 0
|
||||
xorriso : UPDATE : 65.27% done
|
||||
ISO image produced: 6030 sectors
|
||||
Written to medium : 6030 sectors at LBA 0
|
||||
Writing to 'stdio:/workspaces/claude-os/release/claude-os.iso' completed successfully.
|
||||
|
||||
[100%] Built target iso
|
||||
|
||||
@@ -25,6 +25,14 @@ add_executable(kernel
|
||||
mbr.c
|
||||
fat32.c
|
||||
floppy.c
|
||||
ne2000.c
|
||||
e3c509.c
|
||||
ethernet.c
|
||||
ipv4.c
|
||||
arp.c
|
||||
dhcp.c
|
||||
udp.c
|
||||
tcp.c
|
||||
env.c
|
||||
keyboard.c
|
||||
interrupts.S
|
||||
|
||||
382
src/arp.c
Normal file
382
src/arp.c
Normal file
@@ -0,0 +1,382 @@
|
||||
/**
|
||||
* @file arp.c
|
||||
* @brief Address Resolution Protocol (ARP) implementation.
|
||||
*
|
||||
* Maintains an ARP cache mapping IPv4 addresses to Ethernet MAC
|
||||
* addresses. Sends ARP requests and processes ARP replies.
|
||||
* Also responds to incoming ARP requests for our own IP addresses.
|
||||
*
|
||||
* The ARP table is exposed via sysfs at /sys/arp.
|
||||
*/
|
||||
|
||||
#include "arp.h"
|
||||
#include "ethernet.h"
|
||||
#include "ipv4.h"
|
||||
#include "sysfs.h"
|
||||
#include <string.h>
|
||||
|
||||
/* Debug print helpers */
|
||||
extern void offset_print(const char *str);
|
||||
extern void print_hex(uint32_t val);
|
||||
|
||||
/* ================================================================
|
||||
* Global state
|
||||
* ================================================================ */
|
||||
|
||||
/** ARP cache table. */
|
||||
static arp_entry_t arp_table[ARP_TABLE_SIZE];
|
||||
|
||||
/** Number of active entries. */
|
||||
static uint32_t arp_count = 0;
|
||||
|
||||
/* ================================================================
|
||||
* ARP cache management
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Find an ARP entry by IP address.
|
||||
* @return Pointer to entry, or NULL if not found.
|
||||
*/
|
||||
static arp_entry_t *arp_find(uint32_t ip) {
|
||||
for (uint32_t i = 0; i < ARP_TABLE_SIZE; i++) {
|
||||
if (arp_table[i].state != ARP_STATE_FREE &&
|
||||
arp_table[i].ip_addr == ip) {
|
||||
return &arp_table[i];
|
||||
}
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/**
|
||||
* Allocate a new ARP entry.
|
||||
* Reuses a free slot, or evicts the oldest entry.
|
||||
*/
|
||||
static arp_entry_t *arp_alloc(void) {
|
||||
/* Look for a free slot */
|
||||
for (uint32_t i = 0; i < ARP_TABLE_SIZE; i++) {
|
||||
if (arp_table[i].state == ARP_STATE_FREE) {
|
||||
return &arp_table[i];
|
||||
}
|
||||
}
|
||||
|
||||
/* Evict the oldest entry (lowest timestamp) */
|
||||
arp_entry_t *oldest = &arp_table[0];
|
||||
for (uint32_t i = 1; i < ARP_TABLE_SIZE; i++) {
|
||||
if (arp_table[i].timestamp < oldest->timestamp) {
|
||||
oldest = &arp_table[i];
|
||||
}
|
||||
}
|
||||
return oldest;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* ARP send
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Send an ARP packet.
|
||||
*/
|
||||
static int arp_send_packet(uint32_t iface_idx, uint16_t operation,
|
||||
const uint8_t *target_mac, uint32_t target_ip) {
|
||||
eth_iface_t *iface = ethernet_get_iface(iface_idx);
|
||||
if (!iface || !iface->active) return -1;
|
||||
|
||||
arp_packet_t pkt;
|
||||
pkt.hw_type = htons(ARP_HW_ETHER);
|
||||
pkt.proto_type = htons(ETHERTYPE_IPV4);
|
||||
pkt.hw_len = 6;
|
||||
pkt.proto_len = 4;
|
||||
pkt.operation = htons(operation);
|
||||
|
||||
/* Sender: our MAC and IP */
|
||||
memcpy(pkt.sender_mac, iface->mac, 6);
|
||||
pkt.sender_ip = htonl(iface->ip_addr);
|
||||
|
||||
/* Target */
|
||||
memcpy(pkt.target_mac, target_mac, 6);
|
||||
pkt.target_ip = htonl(target_ip);
|
||||
|
||||
/* For ARP requests, broadcast; for replies, send directly */
|
||||
const uint8_t *dst_mac;
|
||||
if (operation == ARP_OP_REQUEST) {
|
||||
static const uint8_t bcast[6] = {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
|
||||
dst_mac = bcast;
|
||||
} else {
|
||||
dst_mac = target_mac;
|
||||
}
|
||||
|
||||
return ethernet_send(iface, dst_mac, ETHERTYPE_ARP,
|
||||
&pkt, ARP_PACKET_SIZE);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
int arp_lookup(uint32_t ip, uint8_t *mac) {
|
||||
arp_entry_t *entry = arp_find(ip);
|
||||
if (entry && entry->state == ARP_STATE_RESOLVED) {
|
||||
memcpy(mac, entry->mac, 6);
|
||||
return 0;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
int arp_request(uint32_t iface_idx, uint32_t target_ip) {
|
||||
static const uint8_t zero_mac[6] = {0, 0, 0, 0, 0, 0};
|
||||
|
||||
/* Create an incomplete entry if we don't have one */
|
||||
arp_entry_t *entry = arp_find(target_ip);
|
||||
if (!entry) {
|
||||
entry = arp_alloc();
|
||||
memset(entry, 0, sizeof(arp_entry_t));
|
||||
entry->ip_addr = target_ip;
|
||||
entry->state = ARP_STATE_INCOMPLETE;
|
||||
entry->iface_idx = (uint8_t)iface_idx;
|
||||
entry->timestamp = 0; /* will be updated on reply */
|
||||
arp_count++;
|
||||
}
|
||||
|
||||
return arp_send_packet(iface_idx, ARP_OP_REQUEST, zero_mac, target_ip);
|
||||
}
|
||||
|
||||
int arp_resolve(uint32_t iface_idx, uint32_t ip, uint8_t *mac) {
|
||||
/* Broadcast address — no ARP needed */
|
||||
if (ip == 0xFFFFFFFF) {
|
||||
memset(mac, 0xFF, 6);
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* Check cache */
|
||||
if (arp_lookup(ip, mac) == 0) return 0;
|
||||
|
||||
/* Send ARP request */
|
||||
arp_request(iface_idx, ip);
|
||||
return -1;
|
||||
}
|
||||
|
||||
void arp_receive(const void *data, uint32_t len, uint32_t iface_idx) {
|
||||
if (len < ARP_PACKET_SIZE) return;
|
||||
|
||||
const arp_packet_t *pkt = (const arp_packet_t *)data;
|
||||
|
||||
/* Validate: must be Ethernet/IPv4 */
|
||||
if (ntohs(pkt->hw_type) != ARP_HW_ETHER) return;
|
||||
if (ntohs(pkt->proto_type) != ETHERTYPE_IPV4) return;
|
||||
if (pkt->hw_len != 6 || pkt->proto_len != 4) return;
|
||||
|
||||
uint32_t sender_ip = ntohl(pkt->sender_ip);
|
||||
uint32_t target_ip = ntohl(pkt->target_ip);
|
||||
uint16_t operation = ntohs(pkt->operation);
|
||||
|
||||
/* Update or create ARP cache entry for sender */
|
||||
arp_entry_t *entry = arp_find(sender_ip);
|
||||
if (entry) {
|
||||
memcpy(entry->mac, pkt->sender_mac, 6);
|
||||
entry->state = ARP_STATE_RESOLVED;
|
||||
entry->iface_idx = (uint8_t)iface_idx;
|
||||
} else {
|
||||
entry = arp_alloc();
|
||||
entry->ip_addr = sender_ip;
|
||||
memcpy(entry->mac, pkt->sender_mac, 6);
|
||||
entry->state = ARP_STATE_RESOLVED;
|
||||
entry->iface_idx = (uint8_t)iface_idx;
|
||||
entry->timestamp = 0;
|
||||
arp_count++;
|
||||
}
|
||||
|
||||
/* Check if this ARP is targeted at us */
|
||||
eth_iface_t *iface = ethernet_get_iface(iface_idx);
|
||||
if (!iface || iface->ip_addr == 0) return;
|
||||
|
||||
if (target_ip == iface->ip_addr && operation == ARP_OP_REQUEST) {
|
||||
/* Send ARP reply */
|
||||
arp_send_packet(iface_idx, ARP_OP_REPLY,
|
||||
pkt->sender_mac, sender_ip);
|
||||
}
|
||||
}
|
||||
|
||||
int arp_add_static(uint32_t ip, const uint8_t *mac, uint32_t iface_idx) {
|
||||
arp_entry_t *entry = arp_find(ip);
|
||||
if (!entry) {
|
||||
entry = arp_alloc();
|
||||
if (!entry) return -1;
|
||||
arp_count++;
|
||||
}
|
||||
|
||||
entry->ip_addr = ip;
|
||||
memcpy(entry->mac, mac, 6);
|
||||
entry->state = ARP_STATE_RESOLVED;
|
||||
entry->iface_idx = (uint8_t)iface_idx;
|
||||
entry->timestamp = 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
const arp_entry_t *arp_get_entry(uint32_t index) {
|
||||
uint32_t count = 0;
|
||||
for (uint32_t i = 0; i < ARP_TABLE_SIZE; i++) {
|
||||
if (arp_table[i].state == ARP_STATE_FREE) continue;
|
||||
if (count == index) return &arp_table[i];
|
||||
count++;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
uint32_t arp_get_count(void) {
|
||||
return arp_count;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Sysfs interface: /sys/arp
|
||||
*
|
||||
* Layout:
|
||||
* /sys/arp/
|
||||
* table - ARP table in human-readable format
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Format a MAC address as "XX:XX:XX:XX:XX:XX".
|
||||
*/
|
||||
static void mac_to_str(const uint8_t *mac, char *buf) {
|
||||
static const char hex[] = "0123456789ABCDEF";
|
||||
int pos = 0;
|
||||
for (int i = 0; i < 6; i++) {
|
||||
if (i > 0) buf[pos++] = ':';
|
||||
buf[pos++] = hex[(mac[i] >> 4) & 0xF];
|
||||
buf[pos++] = hex[mac[i] & 0xF];
|
||||
}
|
||||
buf[pos] = '\0';
|
||||
}
|
||||
|
||||
/**
|
||||
* Format an IPv4 address (host byte order) as dotted decimal.
|
||||
*/
|
||||
static int ip_to_str(uint32_t ip, char *buf, uint32_t size) {
|
||||
int pos = 0;
|
||||
for (int i = 0; i < 4; i++) {
|
||||
if (i > 0 && pos < (int)size - 1) buf[pos++] = '.';
|
||||
uint8_t octet = (uint8_t)((ip >> (24 - i * 8)) & 0xFF);
|
||||
if (octet >= 100 && pos < (int)size - 3) {
|
||||
buf[pos++] = (char)('0' + octet / 100);
|
||||
buf[pos++] = (char)('0' + (octet % 100) / 10);
|
||||
buf[pos++] = (char)('0' + octet % 10);
|
||||
} else if (octet >= 10 && pos < (int)size - 2) {
|
||||
buf[pos++] = (char)('0' + octet / 10);
|
||||
buf[pos++] = (char)('0' + octet % 10);
|
||||
} else if (pos < (int)size - 1) {
|
||||
buf[pos++] = (char)('0' + octet);
|
||||
}
|
||||
}
|
||||
if (pos < (int)size) buf[pos] = '\0';
|
||||
return pos;
|
||||
}
|
||||
|
||||
/**
|
||||
* Append a string to buf at position *pos.
|
||||
*/
|
||||
static void append_str(char *buf, uint32_t size, int *pos, const char *s) {
|
||||
while (*s && *pos < (int)size - 1) buf[(*pos)++] = *s++;
|
||||
}
|
||||
|
||||
static int arp_sysfs_list(void *ctx, const char *path, uint32_t idx,
|
||||
sysfs_entry_t *out) {
|
||||
(void)ctx;
|
||||
|
||||
if (path[0] == '\0') {
|
||||
if (idx == 0) {
|
||||
memset(out, 0, sizeof(sysfs_entry_t));
|
||||
strncpy(out->name, "table", SYSFS_MAX_NAME - 1);
|
||||
out->is_dir = 0;
|
||||
return 0;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
static int arp_sysfs_read(void *ctx, const char *path, char *buf,
|
||||
uint32_t buf_size) {
|
||||
(void)ctx;
|
||||
|
||||
if (strcmp(path, "table") != 0) return -1;
|
||||
|
||||
int pos = 0;
|
||||
|
||||
/* Header */
|
||||
append_str(buf, buf_size, &pos, "IP Address MAC Address Iface State\n");
|
||||
|
||||
for (uint32_t i = 0; i < ARP_TABLE_SIZE; i++) {
|
||||
if (arp_table[i].state == ARP_STATE_FREE) continue;
|
||||
|
||||
/* IP address */
|
||||
char ip_buf[16];
|
||||
ip_to_str(arp_table[i].ip_addr, ip_buf, sizeof(ip_buf));
|
||||
append_str(buf, buf_size, &pos, ip_buf);
|
||||
|
||||
/* Pad to column 17 */
|
||||
int ip_len = (int)strlen(ip_buf);
|
||||
for (int p = ip_len; p < 17; p++) {
|
||||
if (pos < (int)buf_size - 1) buf[pos++] = ' ';
|
||||
}
|
||||
|
||||
/* MAC */
|
||||
char mac_buf[18];
|
||||
mac_to_str(arp_table[i].mac, mac_buf);
|
||||
append_str(buf, buf_size, &pos, mac_buf);
|
||||
if (pos < (int)buf_size - 1) buf[pos++] = ' ';
|
||||
if (pos < (int)buf_size - 1) buf[pos++] = ' ';
|
||||
|
||||
/* Interface */
|
||||
eth_iface_t *iface = ethernet_get_iface(arp_table[i].iface_idx);
|
||||
if (iface) {
|
||||
append_str(buf, buf_size, &pos, iface->name);
|
||||
} else {
|
||||
append_str(buf, buf_size, &pos, "?");
|
||||
}
|
||||
|
||||
/* Pad and state */
|
||||
int name_len = iface ? (int)strlen(iface->name) : 1;
|
||||
for (int p = name_len; p < 7; p++) {
|
||||
if (pos < (int)buf_size - 1) buf[pos++] = ' ';
|
||||
}
|
||||
|
||||
if (arp_table[i].state == ARP_STATE_RESOLVED) {
|
||||
append_str(buf, buf_size, &pos, "resolved");
|
||||
} else {
|
||||
append_str(buf, buf_size, &pos, "incomplete");
|
||||
}
|
||||
|
||||
if (pos < (int)buf_size - 1) buf[pos++] = '\n';
|
||||
}
|
||||
|
||||
buf[pos] = '\0';
|
||||
return pos;
|
||||
}
|
||||
|
||||
static int arp_sysfs_write(void *ctx, const char *path, const char *buf,
|
||||
uint32_t size) {
|
||||
(void)ctx;
|
||||
(void)path;
|
||||
(void)buf;
|
||||
(void)size;
|
||||
return -1; /* Read-only for now */
|
||||
}
|
||||
|
||||
static sysfs_ops_t arp_sysfs_ops = {
|
||||
.list = arp_sysfs_list,
|
||||
.read = arp_sysfs_read,
|
||||
.write = arp_sysfs_write,
|
||||
};
|
||||
|
||||
/* ================================================================
|
||||
* Initialization
|
||||
* ================================================================ */
|
||||
|
||||
void arp_init(void) {
|
||||
memset(arp_table, 0, sizeof(arp_table));
|
||||
arp_count = 0;
|
||||
|
||||
sysfs_register("arp", &arp_sysfs_ops, NULL);
|
||||
|
||||
offset_print(" ARP: initialized\n");
|
||||
}
|
||||
145
src/arp.h
Normal file
145
src/arp.h
Normal file
@@ -0,0 +1,145 @@
|
||||
/**
|
||||
* @file arp.h
|
||||
* @brief Address Resolution Protocol (ARP) subsystem.
|
||||
*
|
||||
* Implements ARP (RFC 826) for mapping IPv4 addresses to Ethernet
|
||||
* MAC addresses. Maintains an ARP cache and handles ARP requests
|
||||
* and replies.
|
||||
*
|
||||
* The ARP table is exposed via sysfs at /sys/arp for userspace tools.
|
||||
*/
|
||||
|
||||
#ifndef ARP_H
|
||||
#define ARP_H
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
/** Maximum number of ARP cache entries. */
|
||||
#define ARP_TABLE_SIZE 32
|
||||
|
||||
/** ARP hardware type: Ethernet */
|
||||
#define ARP_HW_ETHER 1
|
||||
|
||||
/** ARP operation codes */
|
||||
#define ARP_OP_REQUEST 1
|
||||
#define ARP_OP_REPLY 2
|
||||
|
||||
/** ARP cache entry states */
|
||||
#define ARP_STATE_FREE 0 /**< Unused slot. */
|
||||
#define ARP_STATE_INCOMPLETE 1 /**< Request sent, awaiting reply. */
|
||||
#define ARP_STATE_RESOLVED 2 /**< MAC address known. */
|
||||
|
||||
/* ================================================================
|
||||
* ARP packet (28 bytes for IPv4-over-Ethernet)
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* ARP packet structure for IPv4 over Ethernet.
|
||||
* All multi-byte fields are in network byte order.
|
||||
*/
|
||||
typedef struct __attribute__((packed)) arp_packet {
|
||||
uint16_t hw_type; /**< Hardware type (1 = Ethernet). */
|
||||
uint16_t proto_type; /**< Protocol type (0x0800 = IPv4). */
|
||||
uint8_t hw_len; /**< Hardware address length (6). */
|
||||
uint8_t proto_len; /**< Protocol address length (4). */
|
||||
uint16_t operation; /**< Operation (1=request, 2=reply). */
|
||||
uint8_t sender_mac[6]; /**< Sender hardware address. */
|
||||
uint32_t sender_ip; /**< Sender protocol address. */
|
||||
uint8_t target_mac[6]; /**< Target hardware address. */
|
||||
uint32_t target_ip; /**< Target protocol address. */
|
||||
} arp_packet_t;
|
||||
|
||||
/** ARP packet size. */
|
||||
#define ARP_PACKET_SIZE 28
|
||||
|
||||
/* ================================================================
|
||||
* ARP cache entry
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* ARP cache entry.
|
||||
*/
|
||||
typedef struct arp_entry {
|
||||
uint32_t ip_addr; /**< IPv4 address (host byte order). */
|
||||
uint8_t mac[6]; /**< Resolved MAC address. */
|
||||
uint8_t state; /**< ARP_STATE_*. */
|
||||
uint8_t iface_idx; /**< Ethernet interface index. */
|
||||
uint32_t timestamp; /**< Time when entry was created (tick count). */
|
||||
} arp_entry_t;
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Initialize the ARP subsystem.
|
||||
* Registers sysfs namespace "arp".
|
||||
*/
|
||||
void arp_init(void);
|
||||
|
||||
/**
|
||||
* Look up an IP address in the ARP cache.
|
||||
*
|
||||
* @param ip IPv4 address (host byte order).
|
||||
* @param mac Output: 6-byte MAC address if found.
|
||||
* @return 0 if found and resolved, -1 if not in cache or incomplete.
|
||||
*/
|
||||
int arp_lookup(uint32_t ip, uint8_t *mac);
|
||||
|
||||
/**
|
||||
* Send an ARP request for the given IP address.
|
||||
*
|
||||
* @param iface_idx Ethernet interface to send on.
|
||||
* @param target_ip IP address to resolve (host byte order).
|
||||
* @return 0 on success, -1 on failure.
|
||||
*/
|
||||
int arp_request(uint32_t iface_idx, uint32_t target_ip);
|
||||
|
||||
/**
|
||||
* Resolve an IP address to a MAC address.
|
||||
*
|
||||
* Checks the ARP cache first. If not found, sends an ARP request
|
||||
* and returns -1 (caller should retry later).
|
||||
*
|
||||
* @param iface_idx Ethernet interface index.
|
||||
* @param ip IPv4 address (host byte order).
|
||||
* @param mac Output: 6-byte MAC address.
|
||||
* @return 0 if resolved, -1 if pending.
|
||||
*/
|
||||
int arp_resolve(uint32_t iface_idx, uint32_t ip, uint8_t *mac);
|
||||
|
||||
/**
|
||||
* Process an incoming ARP packet.
|
||||
*
|
||||
* Called by the Ethernet subsystem when an ARP frame is received.
|
||||
*
|
||||
* @param data Raw ARP packet.
|
||||
* @param len Packet length.
|
||||
* @param iface_idx Interface the packet arrived on.
|
||||
*/
|
||||
void arp_receive(const void *data, uint32_t len, uint32_t iface_idx);
|
||||
|
||||
/**
|
||||
* Add a static ARP entry.
|
||||
*
|
||||
* @param ip IPv4 address (host byte order).
|
||||
* @param mac 6-byte MAC address.
|
||||
* @param iface_idx Ethernet interface index.
|
||||
* @return 0 on success, -1 if table full.
|
||||
*/
|
||||
int arp_add_static(uint32_t ip, const uint8_t *mac, uint32_t iface_idx);
|
||||
|
||||
/**
|
||||
* Get an ARP table entry by index (for enumeration).
|
||||
*
|
||||
* @param index 0-based index.
|
||||
* @return Pointer to entry, or NULL if out of range.
|
||||
*/
|
||||
const arp_entry_t *arp_get_entry(uint32_t index);
|
||||
|
||||
/**
|
||||
* Get the number of active ARP entries.
|
||||
*/
|
||||
uint32_t arp_get_count(void);
|
||||
|
||||
#endif /* ARP_H */
|
||||
552
src/dhcp.c
Normal file
552
src/dhcp.c
Normal file
@@ -0,0 +1,552 @@
|
||||
/**
|
||||
* @file dhcp.c
|
||||
* @brief DHCP client implementation.
|
||||
*
|
||||
* Implements the DHCP client protocol (RFC 2131) for automatic IPv4
|
||||
* address configuration. Communicates via UDP (port 68→67) over
|
||||
* Ethernet broadcasts.
|
||||
*
|
||||
* Since we don't have a full UDP stack yet, DHCP packets are
|
||||
* constructed manually with IP+UDP headers and sent via the
|
||||
* Ethernet subsystem directly.
|
||||
*
|
||||
* Status is exposed via sysfs at /sys/dhcp.
|
||||
*/
|
||||
|
||||
#include "dhcp.h"
|
||||
#include "ethernet.h"
|
||||
#include "ipv4.h"
|
||||
#include "arp.h"
|
||||
#include "sysfs.h"
|
||||
#include <string.h>
|
||||
|
||||
/* Debug print helpers */
|
||||
extern void offset_print(const char *str);
|
||||
extern void print_hex(uint32_t val);
|
||||
|
||||
/* ================================================================
|
||||
* UDP header (for DHCP — minimal inline UDP)
|
||||
* ================================================================ */
|
||||
|
||||
/** UDP header structure. */
|
||||
typedef struct __attribute__((packed)) udp_header {
|
||||
uint16_t src_port;
|
||||
uint16_t dst_port;
|
||||
uint16_t length;
|
||||
uint16_t checksum;
|
||||
} udp_header_t;
|
||||
|
||||
#define UDP_HLEN 8
|
||||
|
||||
/* ================================================================
|
||||
* Global state
|
||||
* ================================================================ */
|
||||
|
||||
/** Maximum number of tracked leases (one per interface). */
|
||||
#define MAX_LEASES 8
|
||||
|
||||
/** DHCP lease table. */
|
||||
static dhcp_lease_t leases[MAX_LEASES];
|
||||
static uint32_t lease_count = 0;
|
||||
|
||||
/** Simple pseudo-random XID counter. */
|
||||
static uint32_t xid_counter = 0x12345678;
|
||||
|
||||
/* ================================================================
|
||||
* Helpers
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Compute Internet checksum.
|
||||
*/
|
||||
static uint16_t inet_checksum(const void *data, uint32_t len) {
|
||||
const uint16_t *words = (const uint16_t *)data;
|
||||
uint32_t sum = 0;
|
||||
while (len > 1) { sum += *words++; len -= 2; }
|
||||
if (len == 1) sum += *(const uint8_t *)words;
|
||||
while (sum >> 16) sum = (sum & 0xFFFF) + (sum >> 16);
|
||||
return (uint16_t)(~sum);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get or create a lease for an interface.
|
||||
*/
|
||||
static dhcp_lease_t *get_lease(uint32_t iface_idx) {
|
||||
for (uint32_t i = 0; i < lease_count; i++) {
|
||||
if (leases[i].iface_idx == (uint8_t)iface_idx) return &leases[i];
|
||||
}
|
||||
if (lease_count >= MAX_LEASES) return NULL;
|
||||
dhcp_lease_t *l = &leases[lease_count++];
|
||||
memset(l, 0, sizeof(dhcp_lease_t));
|
||||
l->iface_idx = (uint8_t)iface_idx;
|
||||
return l;
|
||||
}
|
||||
|
||||
/**
|
||||
* Format an IP as dotted decimal into buf, return chars written.
|
||||
*/
|
||||
static int fmt_ip(uint32_t ip, char *buf, uint32_t size) {
|
||||
int pos = 0;
|
||||
for (int i = 0; i < 4; i++) {
|
||||
if (i > 0 && pos < (int)size - 1) buf[pos++] = '.';
|
||||
uint8_t o = (uint8_t)((ip >> (24 - i * 8)) & 0xFF);
|
||||
if (o >= 100 && pos < (int)size - 3) {
|
||||
buf[pos++] = (char)('0' + o / 100);
|
||||
buf[pos++] = (char)('0' + (o % 100) / 10);
|
||||
buf[pos++] = (char)('0' + o % 10);
|
||||
} else if (o >= 10 && pos < (int)size - 2) {
|
||||
buf[pos++] = (char)('0' + o / 10);
|
||||
buf[pos++] = (char)('0' + o % 10);
|
||||
} else if (pos < (int)size - 1) {
|
||||
buf[pos++] = (char)('0' + o);
|
||||
}
|
||||
}
|
||||
if (pos < (int)size) buf[pos] = '\0';
|
||||
return pos;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* DHCP packet construction
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Build and send a DHCP packet wrapped in IP+UDP.
|
||||
*
|
||||
* Since we may not have a full UDP stack, we construct the whole
|
||||
* IP+UDP+DHCP frame manually and send it as a broadcast Ethernet frame.
|
||||
*/
|
||||
static int dhcp_send_packet(uint32_t iface_idx, dhcp_packet_t *dhcp_pkt,
|
||||
uint32_t dhcp_len) {
|
||||
eth_iface_t *iface = ethernet_get_iface(iface_idx);
|
||||
if (!iface) return -1;
|
||||
|
||||
/* Total sizes */
|
||||
uint32_t udp_len = UDP_HLEN + dhcp_len;
|
||||
uint32_t ip_len = IPV4_HLEN + udp_len;
|
||||
|
||||
if (ip_len > ETH_MTU) return -1;
|
||||
|
||||
/* Build combined IP+UDP+DHCP packet */
|
||||
uint8_t pkt[1500];
|
||||
memset(pkt, 0, sizeof(pkt));
|
||||
|
||||
/* IPv4 header */
|
||||
ipv4_header_t *ip = (ipv4_header_t *)pkt;
|
||||
ip->ihl_version = 0x45;
|
||||
ip->tos = 0;
|
||||
ip->total_length = htons((uint16_t)ip_len);
|
||||
ip->identification = htons(xid_counter & 0xFFFF);
|
||||
ip->flags_fragoff = 0;
|
||||
ip->ttl = 64;
|
||||
ip->protocol = IP_PROTO_UDP;
|
||||
ip->checksum = 0;
|
||||
ip->src_ip = htonl(iface->ip_addr); /* 0.0.0.0 if not yet configured */
|
||||
ip->dst_ip = htonl(0xFFFFFFFF); /* Broadcast */
|
||||
ip->checksum = inet_checksum(ip, IPV4_HLEN);
|
||||
|
||||
/* UDP header */
|
||||
udp_header_t *udp = (udp_header_t *)(pkt + IPV4_HLEN);
|
||||
udp->src_port = htons(DHCP_CLIENT_PORT);
|
||||
udp->dst_port = htons(DHCP_SERVER_PORT);
|
||||
udp->length = htons((uint16_t)udp_len);
|
||||
udp->checksum = 0; /* UDP checksum optional in IPv4 */
|
||||
|
||||
/* DHCP payload */
|
||||
memcpy(pkt + IPV4_HLEN + UDP_HLEN, dhcp_pkt, dhcp_len);
|
||||
|
||||
/* Send as broadcast Ethernet frame */
|
||||
static const uint8_t bcast[6] = {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
|
||||
return ethernet_send(iface, bcast, ETHERTYPE_IPV4, pkt, ip_len);
|
||||
}
|
||||
|
||||
/**
|
||||
* Add a DHCP option to the options buffer.
|
||||
* Returns new offset.
|
||||
*/
|
||||
static uint32_t add_option(uint8_t *opts, uint32_t off,
|
||||
uint8_t code, uint8_t len, const void *data) {
|
||||
opts[off++] = code;
|
||||
opts[off++] = len;
|
||||
memcpy(&opts[off], data, len);
|
||||
return off + len;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* DHCP protocol
|
||||
* ================================================================ */
|
||||
|
||||
int dhcp_discover(uint32_t iface_idx) {
|
||||
eth_iface_t *iface = ethernet_get_iface(iface_idx);
|
||||
if (!iface) return -1;
|
||||
|
||||
dhcp_lease_t *lease = get_lease(iface_idx);
|
||||
if (!lease) return -1;
|
||||
|
||||
/* Generate transaction ID */
|
||||
xid_counter += 0x1234;
|
||||
lease->xid = xid_counter;
|
||||
lease->state = DHCP_STATE_DISCOVER;
|
||||
|
||||
/* Build DHCP DISCOVER packet */
|
||||
dhcp_packet_t pkt;
|
||||
memset(&pkt, 0, sizeof(pkt));
|
||||
|
||||
pkt.op = 1; /* BOOTREQUEST */
|
||||
pkt.htype = 1; /* Ethernet */
|
||||
pkt.hlen = 6;
|
||||
pkt.hops = 0;
|
||||
pkt.xid = htonl(lease->xid);
|
||||
pkt.secs = 0;
|
||||
pkt.flags = htons(0x8000); /* Broadcast flag */
|
||||
pkt.ciaddr = 0;
|
||||
pkt.yiaddr = 0;
|
||||
pkt.siaddr = 0;
|
||||
pkt.giaddr = 0;
|
||||
memcpy(pkt.chaddr, iface->mac, 6);
|
||||
pkt.magic = htonl(DHCP_MAGIC_COOKIE);
|
||||
|
||||
/* Options */
|
||||
uint32_t off = 0;
|
||||
|
||||
/* Option 53: DHCP Message Type = DISCOVER */
|
||||
uint8_t msg_type = DHCP_DISCOVER;
|
||||
off = add_option(pkt.options, off, DHCP_OPT_MSG_TYPE, 1, &msg_type);
|
||||
|
||||
/* Option 55: Parameter Request List */
|
||||
uint8_t params[] = {
|
||||
DHCP_OPT_SUBNET_MASK,
|
||||
DHCP_OPT_ROUTER,
|
||||
DHCP_OPT_DNS,
|
||||
};
|
||||
off = add_option(pkt.options, off, DHCP_OPT_PARAM_LIST,
|
||||
sizeof(params), params);
|
||||
|
||||
/* End option */
|
||||
pkt.options[off++] = DHCP_OPT_END;
|
||||
|
||||
offset_print(" DHCP: sending DISCOVER on ");
|
||||
offset_print(iface->name);
|
||||
offset_print("\n");
|
||||
|
||||
return dhcp_send_packet(iface_idx, &pkt, DHCP_FIXED_SIZE + off);
|
||||
}
|
||||
|
||||
/**
|
||||
* Send a DHCP REQUEST message.
|
||||
*/
|
||||
static int dhcp_send_request(uint32_t iface_idx, dhcp_lease_t *lease) {
|
||||
eth_iface_t *iface = ethernet_get_iface(iface_idx);
|
||||
if (!iface) return -1;
|
||||
|
||||
dhcp_packet_t pkt;
|
||||
memset(&pkt, 0, sizeof(pkt));
|
||||
|
||||
pkt.op = 1;
|
||||
pkt.htype = 1;
|
||||
pkt.hlen = 6;
|
||||
pkt.xid = htonl(lease->xid);
|
||||
pkt.flags = htons(0x8000);
|
||||
memcpy(pkt.chaddr, iface->mac, 6);
|
||||
pkt.magic = htonl(DHCP_MAGIC_COOKIE);
|
||||
|
||||
uint32_t off = 0;
|
||||
|
||||
/* Option 53: DHCP Message Type = REQUEST */
|
||||
uint8_t msg_type = DHCP_REQUEST;
|
||||
off = add_option(pkt.options, off, DHCP_OPT_MSG_TYPE, 1, &msg_type);
|
||||
|
||||
/* Option 50: Requested IP */
|
||||
uint32_t req_ip = htonl(lease->ip_addr);
|
||||
off = add_option(pkt.options, off, DHCP_OPT_REQUESTED_IP, 4, &req_ip);
|
||||
|
||||
/* Option 54: Server Identifier */
|
||||
uint32_t srv_ip = htonl(lease->server_ip);
|
||||
off = add_option(pkt.options, off, DHCP_OPT_SERVER_ID, 4, &srv_ip);
|
||||
|
||||
/* Option 55: Parameter Request List */
|
||||
uint8_t params[] = {
|
||||
DHCP_OPT_SUBNET_MASK,
|
||||
DHCP_OPT_ROUTER,
|
||||
DHCP_OPT_DNS,
|
||||
};
|
||||
off = add_option(pkt.options, off, DHCP_OPT_PARAM_LIST,
|
||||
sizeof(params), params);
|
||||
|
||||
pkt.options[off++] = DHCP_OPT_END;
|
||||
|
||||
lease->state = DHCP_STATE_REQUESTING;
|
||||
|
||||
offset_print(" DHCP: sending REQUEST on ");
|
||||
offset_print(iface->name);
|
||||
offset_print("\n");
|
||||
|
||||
return dhcp_send_packet(iface_idx, &pkt, DHCP_FIXED_SIZE + off);
|
||||
}
|
||||
|
||||
/**
|
||||
* Parse DHCP options from a received packet.
|
||||
*/
|
||||
static void parse_options(const uint8_t *opts, uint32_t len,
|
||||
uint8_t *msg_type, dhcp_lease_t *lease) {
|
||||
uint32_t i = 0;
|
||||
while (i < len) {
|
||||
uint8_t code = opts[i++];
|
||||
if (code == DHCP_OPT_END) break;
|
||||
if (code == 0) continue; /* Padding */
|
||||
|
||||
if (i >= len) break;
|
||||
uint8_t opt_len = opts[i++];
|
||||
if (i + opt_len > len) break;
|
||||
|
||||
switch (code) {
|
||||
case DHCP_OPT_MSG_TYPE:
|
||||
if (opt_len >= 1) *msg_type = opts[i];
|
||||
break;
|
||||
case DHCP_OPT_SUBNET_MASK:
|
||||
if (opt_len >= 4) {
|
||||
uint32_t val;
|
||||
memcpy(&val, &opts[i], 4);
|
||||
lease->netmask = ntohl(val);
|
||||
}
|
||||
break;
|
||||
case DHCP_OPT_ROUTER:
|
||||
if (opt_len >= 4) {
|
||||
uint32_t val;
|
||||
memcpy(&val, &opts[i], 4);
|
||||
lease->gateway = ntohl(val);
|
||||
}
|
||||
break;
|
||||
case DHCP_OPT_DNS:
|
||||
if (opt_len >= 4) {
|
||||
uint32_t val;
|
||||
memcpy(&val, &opts[i], 4);
|
||||
lease->dns_server = ntohl(val);
|
||||
}
|
||||
break;
|
||||
case DHCP_OPT_LEASE_TIME:
|
||||
if (opt_len >= 4) {
|
||||
uint32_t val;
|
||||
memcpy(&val, &opts[i], 4);
|
||||
lease->lease_time = ntohl(val);
|
||||
}
|
||||
break;
|
||||
case DHCP_OPT_SERVER_ID:
|
||||
if (opt_len >= 4) {
|
||||
uint32_t val;
|
||||
memcpy(&val, &opts[i], 4);
|
||||
lease->server_ip = ntohl(val);
|
||||
}
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
i += opt_len;
|
||||
}
|
||||
}
|
||||
|
||||
void dhcp_receive(const void *data, uint32_t len, uint32_t iface_idx) {
|
||||
if (len < DHCP_FIXED_SIZE) return;
|
||||
|
||||
const dhcp_packet_t *pkt = (const dhcp_packet_t *)data;
|
||||
|
||||
/* Must be a BOOTREPLY */
|
||||
if (pkt->op != 2) return;
|
||||
|
||||
/* Check magic cookie */
|
||||
if (ntohl(pkt->magic) != DHCP_MAGIC_COOKIE) return;
|
||||
|
||||
dhcp_lease_t *lease = get_lease(iface_idx);
|
||||
if (!lease) return;
|
||||
|
||||
/* Match transaction ID */
|
||||
if (ntohl(pkt->xid) != lease->xid) return;
|
||||
|
||||
/* Parse options */
|
||||
uint8_t msg_type = 0;
|
||||
uint32_t opts_len = len - DHCP_FIXED_SIZE;
|
||||
if (opts_len > sizeof(pkt->options)) opts_len = sizeof(pkt->options);
|
||||
parse_options(pkt->options, opts_len, &msg_type, lease);
|
||||
|
||||
/* Store offered IP */
|
||||
uint32_t offered_ip = ntohl(pkt->yiaddr);
|
||||
|
||||
switch (msg_type) {
|
||||
case DHCP_OFFER:
|
||||
if (lease->state != DHCP_STATE_DISCOVER) break;
|
||||
lease->ip_addr = offered_ip;
|
||||
offset_print(" DHCP: received OFFER ");
|
||||
print_hex(offered_ip);
|
||||
offset_print("\n");
|
||||
/* Send REQUEST */
|
||||
dhcp_send_request(iface_idx, lease);
|
||||
break;
|
||||
|
||||
case DHCP_ACK:
|
||||
if (lease->state != DHCP_STATE_REQUESTING) break;
|
||||
lease->ip_addr = offered_ip;
|
||||
lease->state = DHCP_STATE_BOUND;
|
||||
|
||||
/* Apply configuration to the interface */
|
||||
{
|
||||
eth_iface_t *iface = ethernet_get_iface(iface_idx);
|
||||
if (iface) {
|
||||
iface->ip_addr = lease->ip_addr;
|
||||
iface->netmask = lease->netmask;
|
||||
iface->gateway = lease->gateway;
|
||||
}
|
||||
}
|
||||
|
||||
offset_print(" DHCP: BOUND on ");
|
||||
{
|
||||
eth_iface_t *iface2 = ethernet_get_iface(iface_idx);
|
||||
if (iface2) offset_print(iface2->name);
|
||||
}
|
||||
offset_print(" IP ");
|
||||
print_hex(lease->ip_addr);
|
||||
offset_print("\n");
|
||||
break;
|
||||
|
||||
case DHCP_NAK:
|
||||
lease->state = DHCP_STATE_FAILED;
|
||||
offset_print(" DHCP: received NAK\n");
|
||||
break;
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
const dhcp_lease_t *dhcp_get_lease(uint32_t iface_idx) {
|
||||
for (uint32_t i = 0; i < lease_count; i++) {
|
||||
if (leases[i].iface_idx == (uint8_t)iface_idx) return &leases[i];
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
const char *dhcp_state_name(uint8_t state) {
|
||||
switch (state) {
|
||||
case DHCP_STATE_IDLE: return "idle";
|
||||
case DHCP_STATE_DISCOVER: return "discovering";
|
||||
case DHCP_STATE_REQUESTING: return "requesting";
|
||||
case DHCP_STATE_BOUND: return "bound";
|
||||
case DHCP_STATE_RENEWING: return "renewing";
|
||||
case DHCP_STATE_FAILED: return "failed";
|
||||
default: return "unknown";
|
||||
}
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Sysfs: /sys/dhcp
|
||||
*
|
||||
* /sys/dhcp/
|
||||
* status - overall DHCP status
|
||||
* ================================================================ */
|
||||
|
||||
static void append(char *buf, uint32_t size, int *pos, const char *s) {
|
||||
while (*s && *pos < (int)size - 1) buf[(*pos)++] = *s++;
|
||||
}
|
||||
|
||||
static int dhcp_sysfs_list(void *ctx, const char *path, uint32_t idx,
|
||||
sysfs_entry_t *out) {
|
||||
(void)ctx;
|
||||
|
||||
if (path[0] == '\0') {
|
||||
if (idx == 0) {
|
||||
memset(out, 0, sizeof(sysfs_entry_t));
|
||||
strncpy(out->name, "status", SYSFS_MAX_NAME - 1);
|
||||
out->is_dir = 0;
|
||||
return 0;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
static int dhcp_sysfs_read(void *ctx, const char *path, char *buf,
|
||||
uint32_t buf_size) {
|
||||
(void)ctx;
|
||||
|
||||
if (strcmp(path, "status") != 0) return -1;
|
||||
|
||||
int pos = 0;
|
||||
|
||||
if (lease_count == 0) {
|
||||
append(buf, buf_size, &pos, "No DHCP leases.\n");
|
||||
buf[pos] = '\0';
|
||||
return pos;
|
||||
}
|
||||
|
||||
for (uint32_t i = 0; i < lease_count; i++) {
|
||||
dhcp_lease_t *l = &leases[i];
|
||||
eth_iface_t *iface = ethernet_get_iface(l->iface_idx);
|
||||
|
||||
if (iface) {
|
||||
append(buf, buf_size, &pos, iface->name);
|
||||
} else {
|
||||
append(buf, buf_size, &pos, "?");
|
||||
}
|
||||
append(buf, buf_size, &pos, ": ");
|
||||
append(buf, buf_size, &pos, dhcp_state_name(l->state));
|
||||
append(buf, buf_size, &pos, "\n");
|
||||
|
||||
if (l->state == DHCP_STATE_BOUND) {
|
||||
char ip_buf[16];
|
||||
|
||||
append(buf, buf_size, &pos, " IP: ");
|
||||
fmt_ip(l->ip_addr, ip_buf, sizeof(ip_buf));
|
||||
append(buf, buf_size, &pos, ip_buf);
|
||||
append(buf, buf_size, &pos, "\n");
|
||||
|
||||
append(buf, buf_size, &pos, " Netmask: ");
|
||||
fmt_ip(l->netmask, ip_buf, sizeof(ip_buf));
|
||||
append(buf, buf_size, &pos, ip_buf);
|
||||
append(buf, buf_size, &pos, "\n");
|
||||
|
||||
append(buf, buf_size, &pos, " Gateway: ");
|
||||
fmt_ip(l->gateway, ip_buf, sizeof(ip_buf));
|
||||
append(buf, buf_size, &pos, ip_buf);
|
||||
append(buf, buf_size, &pos, "\n");
|
||||
|
||||
append(buf, buf_size, &pos, " DNS: ");
|
||||
fmt_ip(l->dns_server, ip_buf, sizeof(ip_buf));
|
||||
append(buf, buf_size, &pos, ip_buf);
|
||||
append(buf, buf_size, &pos, "\n");
|
||||
|
||||
append(buf, buf_size, &pos, " Server: ");
|
||||
fmt_ip(l->server_ip, ip_buf, sizeof(ip_buf));
|
||||
append(buf, buf_size, &pos, ip_buf);
|
||||
append(buf, buf_size, &pos, "\n");
|
||||
}
|
||||
}
|
||||
|
||||
buf[pos] = '\0';
|
||||
return pos;
|
||||
}
|
||||
|
||||
static int dhcp_sysfs_write(void *ctx, const char *path, const char *buf,
|
||||
uint32_t size) {
|
||||
(void)ctx;
|
||||
(void)path;
|
||||
(void)buf;
|
||||
(void)size;
|
||||
return -1;
|
||||
}
|
||||
|
||||
static sysfs_ops_t dhcp_sysfs_ops = {
|
||||
.list = dhcp_sysfs_list,
|
||||
.read = dhcp_sysfs_read,
|
||||
.write = dhcp_sysfs_write,
|
||||
};
|
||||
|
||||
/* ================================================================
|
||||
* Initialization
|
||||
* ================================================================ */
|
||||
|
||||
void dhcp_init(void) {
|
||||
memset(leases, 0, sizeof(leases));
|
||||
lease_count = 0;
|
||||
|
||||
sysfs_register("dhcp", &dhcp_sysfs_ops, NULL);
|
||||
|
||||
offset_print(" DHCP: initialized\n");
|
||||
}
|
||||
145
src/dhcp.h
Normal file
145
src/dhcp.h
Normal file
@@ -0,0 +1,145 @@
|
||||
/**
|
||||
* @file dhcp.h
|
||||
* @brief DHCP (Dynamic Host Configuration Protocol) client subsystem.
|
||||
*
|
||||
* Implements a minimal DHCP client (RFC 2131) that can obtain an IPv4
|
||||
* address, subnet mask, gateway, and DNS server from a DHCP server.
|
||||
*
|
||||
* DHCP operates over UDP: client port 68, server port 67.
|
||||
* Uses the Ethernet subsystem for broadcast communication.
|
||||
*
|
||||
* Status information is exposed via sysfs at /sys/dhcp.
|
||||
*/
|
||||
|
||||
#ifndef DHCP_H
|
||||
#define DHCP_H
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
/** DHCP ports. */
|
||||
#define DHCP_CLIENT_PORT 68
|
||||
#define DHCP_SERVER_PORT 67
|
||||
|
||||
/** DHCP message types. */
|
||||
#define DHCP_DISCOVER 1
|
||||
#define DHCP_OFFER 2
|
||||
#define DHCP_REQUEST 3
|
||||
#define DHCP_DECLINE 4
|
||||
#define DHCP_ACK 5
|
||||
#define DHCP_NAK 6
|
||||
#define DHCP_RELEASE 7
|
||||
|
||||
/** DHCP options. */
|
||||
#define DHCP_OPT_SUBNET_MASK 1
|
||||
#define DHCP_OPT_ROUTER 3
|
||||
#define DHCP_OPT_DNS 6
|
||||
#define DHCP_OPT_HOSTNAME 12
|
||||
#define DHCP_OPT_REQUESTED_IP 50
|
||||
#define DHCP_OPT_LEASE_TIME 51
|
||||
#define DHCP_OPT_MSG_TYPE 53
|
||||
#define DHCP_OPT_SERVER_ID 54
|
||||
#define DHCP_OPT_PARAM_LIST 55
|
||||
#define DHCP_OPT_END 255
|
||||
|
||||
/** DHCP magic cookie. */
|
||||
#define DHCP_MAGIC_COOKIE 0x63825363
|
||||
|
||||
/** DHCP client states. */
|
||||
#define DHCP_STATE_IDLE 0
|
||||
#define DHCP_STATE_DISCOVER 1
|
||||
#define DHCP_STATE_REQUESTING 2
|
||||
#define DHCP_STATE_BOUND 3
|
||||
#define DHCP_STATE_RENEWING 4
|
||||
#define DHCP_STATE_FAILED 5
|
||||
|
||||
/* ================================================================
|
||||
* DHCP packet structure (548 bytes minimum)
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* DHCP message (over UDP, 236 bytes fixed + options).
|
||||
*/
|
||||
typedef struct __attribute__((packed)) dhcp_packet {
|
||||
uint8_t op; /**< Message op: 1=BOOTREQUEST, 2=BOOTREPLY. */
|
||||
uint8_t htype; /**< Hardware type: 1=Ethernet. */
|
||||
uint8_t hlen; /**< Hardware address length: 6. */
|
||||
uint8_t hops; /**< Hops: 0. */
|
||||
uint32_t xid; /**< Transaction ID. */
|
||||
uint16_t secs; /**< Seconds elapsed. */
|
||||
uint16_t flags; /**< Flags (0x8000 = broadcast). */
|
||||
uint32_t ciaddr; /**< Client IP (if bound). */
|
||||
uint32_t yiaddr; /**< 'Your' (client) IP address. */
|
||||
uint32_t siaddr; /**< Server IP address. */
|
||||
uint32_t giaddr; /**< Gateway IP address. */
|
||||
uint8_t chaddr[16]; /**< Client hardware address. */
|
||||
uint8_t sname[64]; /**< Server host name (unused). */
|
||||
uint8_t file[128]; /**< Boot file name (unused). */
|
||||
uint32_t magic; /**< Magic cookie (0x63825363). */
|
||||
uint8_t options[312]; /**< DHCP options. */
|
||||
} dhcp_packet_t;
|
||||
|
||||
/** Fixed part of DHCP packet (before options). */
|
||||
#define DHCP_FIXED_SIZE 240
|
||||
|
||||
/* ================================================================
|
||||
* DHCP lease information
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Information obtained from a DHCP lease.
|
||||
*/
|
||||
typedef struct dhcp_lease {
|
||||
uint32_t ip_addr; /**< Assigned IP address (host byte order). */
|
||||
uint32_t netmask; /**< Subnet mask (host byte order). */
|
||||
uint32_t gateway; /**< Default gateway (host byte order). */
|
||||
uint32_t dns_server; /**< DNS server (host byte order). */
|
||||
uint32_t server_ip; /**< DHCP server IP (host byte order). */
|
||||
uint32_t lease_time; /**< Lease time in seconds. */
|
||||
uint32_t xid; /**< Transaction ID used. */
|
||||
uint8_t state; /**< Current DHCP state. */
|
||||
uint8_t iface_idx; /**< Ethernet interface. */
|
||||
} dhcp_lease_t;
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Initialize the DHCP subsystem.
|
||||
*/
|
||||
void dhcp_init(void);
|
||||
|
||||
/**
|
||||
* Start a DHCP discover/request sequence on an interface.
|
||||
*
|
||||
* @param iface_idx Ethernet interface index.
|
||||
* @return 0 on success (discover sent), -1 on failure.
|
||||
*/
|
||||
int dhcp_discover(uint32_t iface_idx);
|
||||
|
||||
/**
|
||||
* Process an incoming DHCP (UDP) packet.
|
||||
*
|
||||
* @param data Raw DHCP packet.
|
||||
* @param len Packet length.
|
||||
* @param iface_idx Interface index.
|
||||
*/
|
||||
void dhcp_receive(const void *data, uint32_t len, uint32_t iface_idx);
|
||||
|
||||
/**
|
||||
* Get the current DHCP lease for an interface.
|
||||
*
|
||||
* @param iface_idx Interface index.
|
||||
* @return Pointer to lease info, or NULL.
|
||||
*/
|
||||
const dhcp_lease_t *dhcp_get_lease(uint32_t iface_idx);
|
||||
|
||||
/**
|
||||
* Get the DHCP state name as a string.
|
||||
*
|
||||
* @param state DHCP_STATE_* constant.
|
||||
* @return Human-readable state name.
|
||||
*/
|
||||
const char *dhcp_state_name(uint8_t state);
|
||||
|
||||
#endif /* DHCP_H */
|
||||
409
src/e3c509.c
Normal file
409
src/e3c509.c
Normal file
@@ -0,0 +1,409 @@
|
||||
/**
|
||||
* @file e3c509.c
|
||||
* @brief 3Com 3C509B (EtherLink III) ISA Ethernet NIC driver.
|
||||
*
|
||||
* Drives 3Com 3C509/3C509B Ethernet adapters using PIO (programmed I/O).
|
||||
* The 3C509B uses a windowed register model: 8 register windows of 16 I/O
|
||||
* ports each, selected by writing to the command register.
|
||||
*
|
||||
* Only 10base-T (RJ45) operation is supported, per design requirements.
|
||||
*
|
||||
* Packet TX: write packet length, then write data to TX PIO port.
|
||||
* Packet RX: poll RX status, read data from RX PIO port.
|
||||
*/
|
||||
|
||||
#include "e3c509.h"
|
||||
#include "port_io.h"
|
||||
#include "pic.h"
|
||||
#include "devicefs.h"
|
||||
#include "ethernet.h"
|
||||
#include "driver.h"
|
||||
#include <string.h>
|
||||
|
||||
/* Debug print helpers */
|
||||
extern void offset_print(const char *str);
|
||||
extern void print_hex(uint32_t val);
|
||||
|
||||
/* ================================================================
|
||||
* Global state
|
||||
* ================================================================ */
|
||||
|
||||
/** Single 3C509B device. */
|
||||
static e3c509_device_t e3c509_dev;
|
||||
|
||||
/** Volatile flags set by IRQ handler. */
|
||||
static volatile int e3c509_rx_ready = 0;
|
||||
static volatile int e3c509_tx_done = 0;
|
||||
|
||||
/* ================================================================
|
||||
* Register access helpers
|
||||
* ================================================================ */
|
||||
|
||||
/** Write a 16-bit command to the 3C509B command register. */
|
||||
static inline void e3c509_cmd(uint16_t base, uint16_t cmd) {
|
||||
outw(base + E3C509_CMD, cmd);
|
||||
}
|
||||
|
||||
/** Read the 16-bit status register. */
|
||||
static inline uint16_t e3c509_status(uint16_t base) {
|
||||
return inw(base + E3C509_STATUS);
|
||||
}
|
||||
|
||||
/**
|
||||
* Select a register window.
|
||||
*/
|
||||
static void e3c509_select_window(uint16_t base, uint16_t window) {
|
||||
e3c509_cmd(base, CMD_SELECT_WINDOW | (window & 0x07));
|
||||
}
|
||||
|
||||
/**
|
||||
* Wait for a command to complete.
|
||||
*/
|
||||
static void e3c509_wait_cmd(uint16_t base) {
|
||||
int timeout = 100000;
|
||||
while ((e3c509_status(base) & STAT_CMD_IN_PROG) && --timeout > 0);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Packet send / receive
|
||||
* ================================================================ */
|
||||
|
||||
int e3c509_send(e3c509_device_t *dev, const void *data, uint32_t len) {
|
||||
if (!dev || !dev->present) return -1;
|
||||
if (len > ETH_FRAME_MAX) return -1;
|
||||
|
||||
uint16_t base = dev->io_base;
|
||||
uint32_t send_len = len;
|
||||
if (send_len < 60) send_len = 60;
|
||||
|
||||
/* Switch to window 1 */
|
||||
e3c509_select_window(base, 1);
|
||||
|
||||
/* Wait for free TX space */
|
||||
int timeout = 100000;
|
||||
while (inw(base + E3C509_W1_FREE_TX) < send_len + 4 && --timeout > 0) {
|
||||
asm volatile("pause");
|
||||
}
|
||||
if (timeout == 0) {
|
||||
offset_print(" 3C509: tx fifo full\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
/* Write TX preamble: packet length (low 11 bits, bit 15 = no interrupt) */
|
||||
outw(base + E3C509_W1_TX_PIO, (uint16_t)(send_len | 0x00));
|
||||
|
||||
/* Pad to dword-aligned length */
|
||||
outw(base + E3C509_W1_TX_PIO, 0x0000);
|
||||
|
||||
/* Write packet data as 16-bit words */
|
||||
const uint16_t *data16 = (const uint16_t *)data;
|
||||
uint32_t words = len / 2;
|
||||
for (uint32_t i = 0; i < words; i++) {
|
||||
outw(base + E3C509_W1_TX_PIO, data16[i]);
|
||||
}
|
||||
/* Write last byte if odd */
|
||||
if (len & 1) {
|
||||
const uint8_t *data8 = (const uint8_t *)data;
|
||||
outw(base + E3C509_W1_TX_PIO, (uint16_t)data8[len - 1]);
|
||||
}
|
||||
|
||||
/* Pad to minimum frame size with zeros */
|
||||
if (len < 60) {
|
||||
uint32_t pad_words = (60 - len + 1) / 2;
|
||||
for (uint32_t i = 0; i < pad_words; i++) {
|
||||
outw(base + E3C509_W1_TX_PIO, 0x0000);
|
||||
}
|
||||
}
|
||||
|
||||
/* Wait for TX complete */
|
||||
e3c509_tx_done = 0;
|
||||
timeout = 1000000;
|
||||
while (!e3c509_tx_done && --timeout > 0) {
|
||||
/* Check TX status directly in case interrupts are slow */
|
||||
uint8_t txstat = inb(base + E3C509_W1_TX_STATUS);
|
||||
if (txstat & 0x80) { /* TX complete */
|
||||
/* Acknowledge by writing status back */
|
||||
outb(base + E3C509_W1_TX_STATUS, txstat);
|
||||
break;
|
||||
}
|
||||
asm volatile("pause");
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int e3c509_recv(e3c509_device_t *dev, void *buf, uint32_t bufsize) {
|
||||
if (!dev || !dev->present) return -1;
|
||||
|
||||
uint16_t base = dev->io_base;
|
||||
|
||||
/* Switch to window 1 */
|
||||
e3c509_select_window(base, 1);
|
||||
|
||||
/* Read RX status */
|
||||
uint16_t rx_status = inw(base + E3C509_W1_RX_STATUS);
|
||||
|
||||
/* Check if a packet is available (bit 15 = incomplete/error) */
|
||||
if (rx_status & 0x8000) {
|
||||
/* Error — discard the packet */
|
||||
e3c509_cmd(base, CMD_RX_DISCARD);
|
||||
e3c509_wait_cmd(base);
|
||||
return -1;
|
||||
}
|
||||
|
||||
/* Bits 10:0 = packet length */
|
||||
uint16_t pkt_len = rx_status & 0x07FF;
|
||||
if (pkt_len == 0) {
|
||||
return 0; /* No packet */
|
||||
}
|
||||
|
||||
uint32_t copy_len = pkt_len;
|
||||
if (copy_len > bufsize) copy_len = bufsize;
|
||||
|
||||
/* Read packet data as 16-bit words */
|
||||
uint16_t *buf16 = (uint16_t *)buf;
|
||||
uint32_t read_words = copy_len / 2;
|
||||
for (uint32_t i = 0; i < read_words; i++) {
|
||||
buf16[i] = inw(base + E3C509_W1_RX_PIO);
|
||||
}
|
||||
if (copy_len & 1) {
|
||||
uint16_t w = inw(base + E3C509_W1_RX_PIO);
|
||||
((uint8_t *)buf)[copy_len - 1] = (uint8_t)(w & 0xFF);
|
||||
}
|
||||
|
||||
/* Discard any remaining data and advance */
|
||||
e3c509_cmd(base, CMD_RX_DISCARD);
|
||||
e3c509_wait_cmd(base);
|
||||
|
||||
return (int)copy_len;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* IRQ handler
|
||||
* ================================================================ */
|
||||
|
||||
void e3c509_irq(void) {
|
||||
if (!e3c509_dev.present) return;
|
||||
|
||||
uint16_t base = e3c509_dev.io_base;
|
||||
uint16_t status = e3c509_status(base);
|
||||
|
||||
if (status & STAT_RX_COMPLETE) {
|
||||
e3c509_rx_ready = 1;
|
||||
e3c509_cmd(base, CMD_ACK_INTR | STAT_RX_COMPLETE);
|
||||
}
|
||||
|
||||
if (status & STAT_TX_COMPLETE) {
|
||||
e3c509_tx_done = 1;
|
||||
/* Read and clear TX status */
|
||||
e3c509_select_window(base, 1);
|
||||
uint8_t txstat = inb(base + E3C509_W1_TX_STATUS);
|
||||
outb(base + E3C509_W1_TX_STATUS, txstat);
|
||||
e3c509_cmd(base, CMD_ACK_INTR | STAT_TX_COMPLETE);
|
||||
}
|
||||
|
||||
if (status & STAT_TX_AVAILABLE) {
|
||||
e3c509_cmd(base, CMD_ACK_INTR | STAT_TX_AVAILABLE);
|
||||
}
|
||||
|
||||
if (status & STAT_ADAPTER_FAIL) {
|
||||
e3c509_cmd(base, CMD_ACK_INTR | STAT_ADAPTER_FAIL);
|
||||
}
|
||||
|
||||
if (status & STAT_UPDATE_STATS) {
|
||||
e3c509_cmd(base, CMD_ACK_INTR | STAT_UPDATE_STATS);
|
||||
}
|
||||
|
||||
/* Acknowledge the interrupt latch */
|
||||
e3c509_cmd(base, CMD_ACK_INTR | STAT_INT_LATCH);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Initialization
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Read the MAC address from the 3C509B's EEPROM (Window 2).
|
||||
*/
|
||||
static void e3c509_read_mac(uint16_t base, uint8_t *mac) {
|
||||
e3c509_select_window(base, 2);
|
||||
uint16_t w0 = inw(base + E3C509_W2_ADDR0);
|
||||
uint16_t w1 = inw(base + E3C509_W2_ADDR1);
|
||||
uint16_t w2 = inw(base + E3C509_W2_ADDR2);
|
||||
|
||||
mac[0] = (uint8_t)(w0 & 0xFF);
|
||||
mac[1] = (uint8_t)((w0 >> 8) & 0xFF);
|
||||
mac[2] = (uint8_t)(w1 & 0xFF);
|
||||
mac[3] = (uint8_t)((w1 >> 8) & 0xFF);
|
||||
mac[4] = (uint8_t)(w2 & 0xFF);
|
||||
mac[5] = (uint8_t)((w2 >> 8) & 0xFF);
|
||||
}
|
||||
|
||||
/**
|
||||
* Initialize the 3C509B hardware.
|
||||
*/
|
||||
static int e3c509_hw_init(uint16_t base) {
|
||||
/* Global reset */
|
||||
e3c509_cmd(base, CMD_GLOBAL_RESET);
|
||||
|
||||
/* Wait for reset to complete */
|
||||
int timeout = 100000;
|
||||
while ((e3c509_status(base) & STAT_CMD_IN_PROG) && --timeout > 0);
|
||||
if (timeout == 0) {
|
||||
offset_print(" 3C509: reset timeout\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
/* Small extra delay */
|
||||
for (volatile int i = 0; i < 50000; i++) {
|
||||
asm volatile("pause");
|
||||
}
|
||||
|
||||
/* Read MAC address */
|
||||
e3c509_read_mac(base, e3c509_dev.mac);
|
||||
|
||||
/* Select 10base-T (RJ45) transceiver — Window 0, Address Config */
|
||||
e3c509_select_window(base, 0);
|
||||
uint16_t addr_cfg = inw(base + E3C509_W0_ADDR_CFG);
|
||||
/* Clear transceiver bits (14:13) and set to TP (00) */
|
||||
addr_cfg &= ~(0x3 << 14);
|
||||
addr_cfg |= (XCVR_TP << 14);
|
||||
outw(base + E3C509_W0_ADDR_CFG, addr_cfg);
|
||||
|
||||
/* Configure IRQ in resource config register */
|
||||
uint16_t res_cfg = inw(base + E3C509_W0_RES_CFG);
|
||||
/* IRQ is in bits 15:12. Set to our IRQ. */
|
||||
res_cfg = (res_cfg & 0x0FFF) | ((uint16_t)e3c509_dev.irq << 12);
|
||||
outw(base + E3C509_W0_RES_CFG, res_cfg);
|
||||
|
||||
/* Enable the adapter */
|
||||
outw(base + E3C509_W0_CFG_CTRL, 0x0001); /* Enable */
|
||||
|
||||
/* Reset TX and RX */
|
||||
e3c509_cmd(base, CMD_TX_RESET);
|
||||
e3c509_wait_cmd(base);
|
||||
e3c509_cmd(base, CMD_RX_RESET);
|
||||
e3c509_wait_cmd(base);
|
||||
|
||||
/* Set RX filter: accept station + broadcast */
|
||||
e3c509_cmd(base, CMD_SET_RX_FILTER | RX_FILTER_STATION | RX_FILTER_BCAST);
|
||||
|
||||
/* Set TX start threshold — start transmitting after full packet */
|
||||
e3c509_cmd(base, CMD_SET_TX_START | (ETH_FRAME_MAX >> 2));
|
||||
|
||||
/* Set interrupt mask */
|
||||
e3c509_cmd(base, CMD_SET_INTR_MASK |
|
||||
STAT_RX_COMPLETE | STAT_TX_COMPLETE | STAT_TX_AVAILABLE |
|
||||
STAT_ADAPTER_FAIL | STAT_UPDATE_STATS);
|
||||
|
||||
/* Enable TX and RX */
|
||||
e3c509_cmd(base, CMD_TX_ENABLE);
|
||||
e3c509_cmd(base, CMD_RX_ENABLE);
|
||||
|
||||
/* Acknowledge any pending interrupts */
|
||||
e3c509_cmd(base, CMD_ACK_INTR | 0xFF);
|
||||
|
||||
/* Switch to operating window (window 1) */
|
||||
e3c509_select_window(base, 1);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Driver framework
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Probe for a 3C509B card.
|
||||
*
|
||||
* Check the manufacturer ID at Window 0, offset 0x00.
|
||||
* The 3Com 3C509B should return 0x6D50.
|
||||
*/
|
||||
static driver_probe_result_t e3c509_probe(void) {
|
||||
uint16_t base = E3C509_DEFAULT_IOBASE;
|
||||
|
||||
/* Try to select Window 0 and read manufacturer ID */
|
||||
e3c509_cmd(base, CMD_SELECT_WINDOW | 0);
|
||||
|
||||
/* Brief delay */
|
||||
for (volatile int i = 0; i < 10000; i++) {
|
||||
asm volatile("pause");
|
||||
}
|
||||
|
||||
uint16_t mfg_id = inw(base + E3C509_W0_MFG_ID);
|
||||
|
||||
/* 3Com manufacturer ID = 0x6D50 */
|
||||
if (mfg_id == 0x6D50) {
|
||||
return DRIVER_PROBE_OK;
|
||||
}
|
||||
|
||||
/* Also check for the product ID being in a reasonable range */
|
||||
if ((mfg_id & 0xFF00) == 0x9000 || (mfg_id & 0xFF00) == 0x9100) {
|
||||
return DRIVER_PROBE_OK;
|
||||
}
|
||||
|
||||
return DRIVER_PROBE_NOT_FOUND;
|
||||
}
|
||||
|
||||
/**
|
||||
* Initialize the 3C509B driver.
|
||||
*/
|
||||
static int e3c509_driver_init(void) {
|
||||
memset(&e3c509_dev, 0, sizeof(e3c509_dev));
|
||||
|
||||
e3c509_dev.io_base = E3C509_DEFAULT_IOBASE;
|
||||
e3c509_dev.irq = E3C509_DEFAULT_IRQ;
|
||||
|
||||
/* Unmask IRQ */
|
||||
pic_clear_mask(e3c509_dev.irq);
|
||||
|
||||
/* Initialize hardware */
|
||||
if (e3c509_hw_init(e3c509_dev.io_base) != 0) {
|
||||
offset_print(" 3C509: initialization failed\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
e3c509_dev.present = 1;
|
||||
|
||||
/* Print MAC address */
|
||||
offset_print(" 3C509: MAC ");
|
||||
for (int i = 0; i < 6; i++) {
|
||||
if (i > 0) offset_print(":");
|
||||
print_hex(e3c509_dev.mac[i]);
|
||||
}
|
||||
offset_print("\n");
|
||||
|
||||
/* Register with ethernet subsystem (creates /dev/ethN) */
|
||||
ethernet_register(e3c509_dev.mac,
|
||||
(eth_send_fn)e3c509_send,
|
||||
(eth_recv_fn)e3c509_recv,
|
||||
&e3c509_dev);
|
||||
|
||||
offset_print(" 3C509: 10base-T (RJ45) on I/O ");
|
||||
print_hex(e3c509_dev.io_base);
|
||||
offset_print(" IRQ ");
|
||||
print_hex(e3c509_dev.irq);
|
||||
offset_print("\n");
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
e3c509_device_t *e3c509_get_device(void) {
|
||||
return e3c509_dev.present ? &e3c509_dev : NULL;
|
||||
}
|
||||
|
||||
int e3c509_init(void) {
|
||||
return e3c509_driver_init();
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Driver registration
|
||||
* ================================================================ */
|
||||
|
||||
static const driver_t e3c509_driver = {
|
||||
.name = "3c509b",
|
||||
.probe = e3c509_probe,
|
||||
.init = e3c509_driver_init,
|
||||
};
|
||||
|
||||
REGISTER_DRIVER(e3c509_driver);
|
||||
158
src/e3c509.h
Normal file
158
src/e3c509.h
Normal file
@@ -0,0 +1,158 @@
|
||||
/**
|
||||
* @file e3c509.h
|
||||
* @brief 3Com 3C509B (EtherLink III) ISA Ethernet NIC driver.
|
||||
*
|
||||
* Drives 3Com 3C509B NICs. Only supports RJ45 (10base-T) transceiver.
|
||||
* The 3C509B uses a windowed register model with 8 register windows
|
||||
* selected via the Window register. Packets are transferred through
|
||||
* PIO (programmed I/O) using the card's FIFO.
|
||||
*
|
||||
* Uses IRQ 10 by default for interrupt-driven operation.
|
||||
* Registers as a character device with devicefs (/dev/ethN).
|
||||
*/
|
||||
|
||||
#ifndef E3C509_H
|
||||
#define E3C509_H
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
/* ================================================================
|
||||
* 3C509B I/O Port Layout
|
||||
*
|
||||
* The 3C509B uses 16 I/O ports starting at the base address.
|
||||
* Registers are organized into 8 windows (0-7).
|
||||
* Window selection: write window number to port base+0x0E.
|
||||
* ================================================================ */
|
||||
|
||||
/** Default ISA I/O base for 3C509B. */
|
||||
#define E3C509_DEFAULT_IOBASE 0x210
|
||||
|
||||
/** Default ISA IRQ for 3C509B. */
|
||||
#define E3C509_DEFAULT_IRQ 10
|
||||
|
||||
/* --- Global registers (always accessible) --- */
|
||||
#define E3C509_CMD 0x0E /**< Command register (write) */
|
||||
#define E3C509_STATUS 0x0E /**< Status register (read) */
|
||||
|
||||
/* --- Window 0: Setup / Configuration --- */
|
||||
#define E3C509_W0_MFG_ID 0x00 /**< Manufacturer ID (should be 0x6D50) */
|
||||
#define E3C509_W0_ADDR_CFG 0x06 /**< Address config (transceiver type) */
|
||||
#define E3C509_W0_RES_CFG 0x08 /**< Resource config (IRQ) */
|
||||
#define E3C509_W0_CFG_CTRL 0x04 /**< Config control */
|
||||
|
||||
/* --- Window 1: Operating Set --- */
|
||||
#define E3C509_W1_TX_PIO 0x00 /**< TX PIO data (write) */
|
||||
#define E3C509_W1_TX_STATUS 0x0B /**< TX status */
|
||||
#define E3C509_W1_RX_PIO 0x00 /**< RX PIO data (read) */
|
||||
#define E3C509_W1_RX_STATUS 0x08 /**< RX status */
|
||||
#define E3C509_W1_FREE_TX 0x0C /**< Free TX bytes */
|
||||
|
||||
/* --- Window 2: Station Address --- */
|
||||
#define E3C509_W2_ADDR0 0x00 /**< Station address word 0 */
|
||||
#define E3C509_W2_ADDR1 0x02 /**< Station address word 1 */
|
||||
#define E3C509_W2_ADDR2 0x04 /**< Station address word 2 */
|
||||
|
||||
/* --- Window 3: FIFO Management --- */
|
||||
/* (Used for internal FIFO buffer management) */
|
||||
|
||||
/* --- Window 4: Diagnostics --- */
|
||||
#define E3C509_W4_MEDIA_TYPE 0x0A /**< Media type and status */
|
||||
#define E3C509_W4_NET_DIAG 0x06 /**< Network diagnostics */
|
||||
|
||||
/* --- Window 5: Read Zeroes (for RX filter) --- */
|
||||
|
||||
/* --- Window 6: Statistics --- */
|
||||
#define E3C509_W6_TX_BYTES 0x0C /**< Total TX bytes */
|
||||
#define E3C509_W6_RX_BYTES 0x0A /**< Total RX bytes */
|
||||
|
||||
/* ================================================================
|
||||
* 3C509B Commands (written to command register)
|
||||
* ================================================================ */
|
||||
#define CMD_GLOBAL_RESET 0x0000 /**< Global reset */
|
||||
#define CMD_SELECT_WINDOW 0x0800 /**< Select window (OR with window num) */
|
||||
#define CMD_RX_ENABLE 0x2000 /**< Enable receiver */
|
||||
#define CMD_RX_RESET 0x2800 /**< Reset receiver */
|
||||
#define CMD_RX_DISCARD 0x4000 /**< Discard top RX packet */
|
||||
#define CMD_TX_ENABLE 0x4800 /**< Enable transmitter */
|
||||
#define CMD_TX_RESET 0x5800 /**< Reset transmitter */
|
||||
#define CMD_REQ_INTR 0x6000 /**< Request interrupt (OR with mask) */
|
||||
#define CMD_ACK_INTR 0x6800 /**< Acknowledge interrupt (OR with mask) */
|
||||
#define CMD_SET_INTR_MASK 0x7000 /**< Set interrupt mask */
|
||||
#define CMD_SET_RX_FILTER 0x8000 /**< Set RX filter */
|
||||
#define CMD_TX_DONE 0x8800 /**< ? */
|
||||
#define CMD_STATS_ENABLE 0x9000 /**< Enable statistics */
|
||||
#define CMD_STATS_DISABLE 0xB000 /**< Disable statistics */
|
||||
#define CMD_SET_TX_START 0x9800 /**< Set TX start threshold */
|
||||
|
||||
/* ================================================================
|
||||
* Status / Interrupt bits
|
||||
* ================================================================ */
|
||||
#define STAT_INT_LATCH 0x0001 /**< Interrupt latch */
|
||||
#define STAT_ADAPTER_FAIL 0x0002 /**< Adapter failure */
|
||||
#define STAT_TX_COMPLETE 0x0004 /**< TX complete */
|
||||
#define STAT_TX_AVAILABLE 0x0008 /**< TX available */
|
||||
#define STAT_RX_COMPLETE 0x0010 /**< RX complete */
|
||||
#define STAT_RX_EARLY 0x0020 /**< RX early threshold */
|
||||
#define STAT_INT_REQ 0x0040 /**< Interrupt requested */
|
||||
#define STAT_UPDATE_STATS 0x0080 /**< Update statistics */
|
||||
#define STAT_CMD_IN_PROG 0x1000 /**< Command in progress */
|
||||
#define STAT_WINDOW_MASK 0xE000 /**< Current window bits */
|
||||
|
||||
/* ================================================================
|
||||
* RX Filter bits (for CMD_SET_RX_FILTER)
|
||||
* ================================================================ */
|
||||
#define RX_FILTER_STATION 0x01 /**< Accept frames to station address */
|
||||
#define RX_FILTER_MCAST 0x02 /**< Accept multicast */
|
||||
#define RX_FILTER_BCAST 0x04 /**< Accept broadcast */
|
||||
#define RX_FILTER_PROMISC 0x08 /**< Promiscuous mode */
|
||||
|
||||
/* ================================================================
|
||||
* Transceiver types
|
||||
* ================================================================ */
|
||||
#define XCVR_TP 0x00 /**< 10base-T / RJ45 */
|
||||
#define XCVR_AUI 0x01 /**< AUI */
|
||||
#define XCVR_BNC 0x03 /**< BNC / 10base2 */
|
||||
|
||||
/** Maximum Ethernet frame size. */
|
||||
#define ETH_FRAME_MAX 1518
|
||||
|
||||
/* ================================================================
|
||||
* 3C509B device state
|
||||
* ================================================================ */
|
||||
typedef struct e3c509_device {
|
||||
uint16_t io_base; /**< I/O base address. */
|
||||
uint8_t irq; /**< IRQ number. */
|
||||
uint8_t mac[6]; /**< MAC address. */
|
||||
int present; /**< 1 if card detected. */
|
||||
} e3c509_device_t;
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Initialize 3C509B driver.
|
||||
*/
|
||||
int e3c509_init(void);
|
||||
|
||||
/**
|
||||
* 3C509B IRQ handler.
|
||||
*/
|
||||
void e3c509_irq(void);
|
||||
|
||||
/**
|
||||
* Send an Ethernet frame.
|
||||
*/
|
||||
int e3c509_send(e3c509_device_t *dev, const void *data, uint32_t len);
|
||||
|
||||
/**
|
||||
* Receive a pending Ethernet frame.
|
||||
*/
|
||||
int e3c509_recv(e3c509_device_t *dev, void *buf, uint32_t bufsize);
|
||||
|
||||
/**
|
||||
* Get the 3C509B device pointer.
|
||||
*/
|
||||
e3c509_device_t *e3c509_get_device(void);
|
||||
|
||||
#endif /* E3C509_H */
|
||||
393
src/ethernet.c
Normal file
393
src/ethernet.c
Normal file
@@ -0,0 +1,393 @@
|
||||
/**
|
||||
* @file ethernet.c
|
||||
* @brief Ethernet subsystem implementation.
|
||||
*
|
||||
* Provides a unified layer over individual Ethernet NIC drivers.
|
||||
* NIC drivers register through ethernet_register(), which creates the
|
||||
* `/dev/ethN` char device and adds the interface to an internal table.
|
||||
*
|
||||
* The subsystem also exposes network interface info via sysfs at
|
||||
* `/sys/net`, allowing userspace tools to query interface status.
|
||||
*/
|
||||
|
||||
#include "ethernet.h"
|
||||
#include "devicefs.h"
|
||||
#include "sysfs.h"
|
||||
#include <string.h>
|
||||
|
||||
/* Debug print helpers */
|
||||
extern void offset_print(const char *str);
|
||||
extern void print_hex(uint32_t val);
|
||||
|
||||
/* ================================================================
|
||||
* Global state
|
||||
* ================================================================ */
|
||||
|
||||
/** Interface table. */
|
||||
static eth_iface_t ifaces[ETH_MAX_IFACES];
|
||||
|
||||
/** Number of registered interfaces. */
|
||||
static uint32_t iface_count = 0;
|
||||
|
||||
/* ================================================================
|
||||
* Character device operations
|
||||
*
|
||||
* Maps devicefs char read/write to the NIC driver send/recv.
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Read from `/dev/ethN`: receive a raw Ethernet frame.
|
||||
*/
|
||||
static int32_t eth_char_read(void *dev_data, uint32_t size, void *buf) {
|
||||
eth_iface_t *iface = (eth_iface_t *)dev_data;
|
||||
if (!iface || !iface->active || !iface->recv) return -1;
|
||||
return (int32_t)iface->recv(iface->dev_data, buf, size);
|
||||
}
|
||||
|
||||
/**
|
||||
* Write to `/dev/ethN`: send a raw Ethernet frame.
|
||||
*/
|
||||
static int32_t eth_char_write(void *dev_data, uint32_t size, const void *buf) {
|
||||
eth_iface_t *iface = (eth_iface_t *)dev_data;
|
||||
if (!iface || !iface->active || !iface->send) return -1;
|
||||
int ret = iface->send(iface->dev_data, buf, size);
|
||||
return (ret == 0) ? (int32_t)size : -1;
|
||||
}
|
||||
|
||||
/** Devicefs char ops shared by all ethernet interfaces. */
|
||||
static devicefs_char_ops_t eth_char_ops = {
|
||||
.read = eth_char_read,
|
||||
.write = eth_char_write,
|
||||
};
|
||||
|
||||
/* ================================================================
|
||||
* Sysfs operations for /sys/net
|
||||
*
|
||||
* Directory layout:
|
||||
* /sys/net/
|
||||
* eth1/
|
||||
* mac - MAC address as hex string
|
||||
* ip - IPv4 address
|
||||
* netmask - subnet mask
|
||||
* gateway - default gateway
|
||||
* link - "up" or "down"
|
||||
* eth2/
|
||||
* ...
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Format a MAC address as "XX:XX:XX:XX:XX:XX" into buf.
|
||||
*/
|
||||
static int format_mac(const uint8_t *mac, char *buf, uint32_t buf_size) {
|
||||
if (buf_size < 18) return -1;
|
||||
static const char hex[] = "0123456789ABCDEF";
|
||||
int pos = 0;
|
||||
for (int i = 0; i < 6; i++) {
|
||||
if (i > 0) buf[pos++] = ':';
|
||||
buf[pos++] = hex[(mac[i] >> 4) & 0xF];
|
||||
buf[pos++] = hex[mac[i] & 0xF];
|
||||
}
|
||||
buf[pos++] = '\n';
|
||||
buf[pos] = '\0';
|
||||
return pos;
|
||||
}
|
||||
|
||||
/**
|
||||
* Format an IPv4 address as "A.B.C.D\n" into buf.
|
||||
*/
|
||||
static int format_ipv4(uint32_t ip, char *buf, uint32_t buf_size) {
|
||||
if (buf_size < 16) return -1;
|
||||
int pos = 0;
|
||||
for (int i = 0; i < 4; i++) {
|
||||
if (i > 0) buf[pos++] = '.';
|
||||
uint8_t octet = (uint8_t)((ip >> (24 - i * 8)) & 0xFF);
|
||||
if (octet >= 100) { buf[pos++] = (char)('0' + octet / 100); octet %= 100; buf[pos++] = (char)('0' + octet / 10); octet %= 10; }
|
||||
else if (octet >= 10) { buf[pos++] = (char)('0' + octet / 10); octet %= 10; }
|
||||
buf[pos++] = (char)('0' + octet);
|
||||
}
|
||||
buf[pos++] = '\n';
|
||||
buf[pos] = '\0';
|
||||
return pos;
|
||||
}
|
||||
|
||||
/**
|
||||
* Parse interface name from the path prefix.
|
||||
* E.g., path "eth1/mac" → iface "eth1", subpath "mac".
|
||||
* Returns the iface, or NULL if not found.
|
||||
*/
|
||||
static eth_iface_t *parse_iface_path(const char *path, const char **subpath) {
|
||||
/* Find the first '/' */
|
||||
const char *slash = NULL;
|
||||
for (const char *p = path; *p; p++) {
|
||||
if (*p == '/') { slash = p; break; }
|
||||
}
|
||||
|
||||
char iface_name[16];
|
||||
if (slash) {
|
||||
uint32_t len = (uint32_t)(slash - path);
|
||||
if (len >= sizeof(iface_name)) return NULL;
|
||||
memcpy(iface_name, path, len);
|
||||
iface_name[len] = '\0';
|
||||
*subpath = slash + 1;
|
||||
} else {
|
||||
/* No slash — path is just the interface name */
|
||||
uint32_t len = strlen(path);
|
||||
if (len >= sizeof(iface_name)) return NULL;
|
||||
memcpy(iface_name, path, len + 1);
|
||||
*subpath = "";
|
||||
}
|
||||
|
||||
return ethernet_find_iface(iface_name);
|
||||
}
|
||||
|
||||
/**
|
||||
* Sysfs list callback for /sys/net.
|
||||
*/
|
||||
static int net_sysfs_list(void *ctx, const char *path, uint32_t idx,
|
||||
sysfs_entry_t *out) {
|
||||
(void)ctx;
|
||||
|
||||
if (path[0] == '\0') {
|
||||
/* Root of /sys/net — list interfaces */
|
||||
uint32_t count = 0;
|
||||
for (uint32_t i = 0; i < ETH_MAX_IFACES; i++) {
|
||||
if (!ifaces[i].active) continue;
|
||||
if (count == idx) {
|
||||
memset(out, 0, sizeof(sysfs_entry_t));
|
||||
strncpy(out->name, ifaces[i].name, SYSFS_MAX_NAME - 1);
|
||||
out->is_dir = 1;
|
||||
return 0;
|
||||
}
|
||||
count++;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
/* Listing inside an interface directory */
|
||||
const char *subpath;
|
||||
eth_iface_t *iface = parse_iface_path(path, &subpath);
|
||||
if (!iface) return -1;
|
||||
|
||||
/* Files: mac, ip, netmask, gateway, link */
|
||||
static const char *files[] = { "mac", "ip", "netmask", "gateway", "link" };
|
||||
if (idx >= 5) return -1;
|
||||
|
||||
memset(out, 0, sizeof(sysfs_entry_t));
|
||||
strncpy(out->name, files[idx], SYSFS_MAX_NAME - 1);
|
||||
out->is_dir = 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* Sysfs read callback for /sys/net.
|
||||
*/
|
||||
static int net_sysfs_read(void *ctx, const char *path, char *buf,
|
||||
uint32_t buf_size) {
|
||||
(void)ctx;
|
||||
|
||||
const char *subpath;
|
||||
eth_iface_t *iface = parse_iface_path(path, &subpath);
|
||||
if (!iface) return -1;
|
||||
|
||||
if (strcmp(subpath, "mac") == 0) {
|
||||
return format_mac(iface->mac, buf, buf_size);
|
||||
}
|
||||
if (strcmp(subpath, "ip") == 0) {
|
||||
return format_ipv4(iface->ip_addr, buf, buf_size);
|
||||
}
|
||||
if (strcmp(subpath, "netmask") == 0) {
|
||||
return format_ipv4(iface->netmask, buf, buf_size);
|
||||
}
|
||||
if (strcmp(subpath, "gateway") == 0) {
|
||||
return format_ipv4(iface->gateway, buf, buf_size);
|
||||
}
|
||||
if (strcmp(subpath, "link") == 0) {
|
||||
const char *s = iface->link_up ? "up\n" : "down\n";
|
||||
strncpy(buf, s, buf_size);
|
||||
return (int)strlen(s);
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
/**
|
||||
* Sysfs write callback for /sys/net.
|
||||
* Allows setting IP address, netmask, gateway.
|
||||
*/
|
||||
static int net_sysfs_write(void *ctx, const char *path, const char *buf,
|
||||
uint32_t size) {
|
||||
(void)ctx;
|
||||
(void)size;
|
||||
|
||||
const char *subpath;
|
||||
eth_iface_t *iface = parse_iface_path(path, &subpath);
|
||||
if (!iface) return -1;
|
||||
|
||||
/* Parse dotted-decimal IP from buf */
|
||||
if (strcmp(subpath, "ip") == 0 || strcmp(subpath, "netmask") == 0 ||
|
||||
strcmp(subpath, "gateway") == 0) {
|
||||
uint32_t octets[4] = {0};
|
||||
int octet_idx = 0;
|
||||
for (uint32_t i = 0; i < size && octet_idx < 4; i++) {
|
||||
if (buf[i] == '.' || buf[i] == '\n' || buf[i] == '\0') {
|
||||
octet_idx++;
|
||||
} else if (buf[i] >= '0' && buf[i] <= '9') {
|
||||
octets[octet_idx] = octets[octet_idx] * 10 + (uint32_t)(buf[i] - '0');
|
||||
}
|
||||
}
|
||||
uint32_t addr = (octets[0] << 24) | (octets[1] << 16) |
|
||||
(octets[2] << 8) | octets[3];
|
||||
|
||||
if (strcmp(subpath, "ip") == 0) iface->ip_addr = addr;
|
||||
if (strcmp(subpath, "netmask") == 0) iface->netmask = addr;
|
||||
if (strcmp(subpath, "gateway") == 0) iface->gateway = addr;
|
||||
|
||||
return (int)size;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
/** Sysfs operations for /sys/net. */
|
||||
static sysfs_ops_t net_sysfs_ops = {
|
||||
.list = net_sysfs_list,
|
||||
.read = net_sysfs_read,
|
||||
.write = net_sysfs_write,
|
||||
};
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
void ethernet_init(void) {
|
||||
memset(ifaces, 0, sizeof(ifaces));
|
||||
iface_count = 0;
|
||||
|
||||
sysfs_register("net", &net_sysfs_ops, NULL);
|
||||
offset_print(" ETHERNET: subsystem initialized\n");
|
||||
}
|
||||
|
||||
eth_iface_t *ethernet_register(const uint8_t *mac,
|
||||
eth_send_fn send,
|
||||
eth_recv_fn recv,
|
||||
void *dev_data) {
|
||||
if (iface_count >= ETH_MAX_IFACES) {
|
||||
offset_print(" ETHERNET: too many interfaces\n");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/* Find free slot */
|
||||
eth_iface_t *iface = NULL;
|
||||
for (uint32_t i = 0; i < ETH_MAX_IFACES; i++) {
|
||||
if (!ifaces[i].active) {
|
||||
iface = &ifaces[i];
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!iface) return NULL;
|
||||
|
||||
memset(iface, 0, sizeof(eth_iface_t));
|
||||
|
||||
/* Copy MAC */
|
||||
memcpy(iface->mac, mac, ETH_ALEN);
|
||||
iface->send = send;
|
||||
iface->recv = recv;
|
||||
iface->dev_data = dev_data;
|
||||
iface->link_up = 1; /* Assume link up after init */
|
||||
iface->active = 1;
|
||||
|
||||
/* Register as /dev/ethN — devicefs assigns the number */
|
||||
devicefs_device_t *devfs_dev = devicefs_register_char("eth",
|
||||
ð_char_ops,
|
||||
iface);
|
||||
if (devfs_dev) {
|
||||
/* Copy the assigned name back */
|
||||
strncpy(iface->name, devfs_dev->name, sizeof(iface->name) - 1);
|
||||
} else {
|
||||
/* Fallback name */
|
||||
iface->name[0] = 'e'; iface->name[1] = 't'; iface->name[2] = 'h';
|
||||
iface->name[3] = (char)('0' + iface_count + 1);
|
||||
iface->name[4] = '\0';
|
||||
}
|
||||
|
||||
iface_count++;
|
||||
|
||||
offset_print(" ETHERNET: registered ");
|
||||
offset_print(iface->name);
|
||||
offset_print(" MAC ");
|
||||
for (int i = 0; i < 6; i++) {
|
||||
if (i > 0) offset_print(":");
|
||||
print_hex(mac[i]);
|
||||
}
|
||||
offset_print("\n");
|
||||
|
||||
return iface;
|
||||
}
|
||||
|
||||
int ethernet_send(eth_iface_t *iface, const uint8_t *dst_mac,
|
||||
uint16_t ethertype, const void *payload, uint32_t len) {
|
||||
if (!iface || !iface->active || !iface->send) return -1;
|
||||
if (len > ETH_MTU) return -1;
|
||||
|
||||
/* Build complete Ethernet frame on stack */
|
||||
uint8_t frame[ETH_FRAME_LEN];
|
||||
eth_header_t *hdr = (eth_header_t *)frame;
|
||||
|
||||
memcpy(hdr->dst, dst_mac, ETH_ALEN);
|
||||
memcpy(hdr->src, iface->mac, ETH_ALEN);
|
||||
hdr->ethertype = htons(ethertype);
|
||||
|
||||
memcpy(frame + ETH_HLEN, payload, len);
|
||||
|
||||
uint32_t frame_len = ETH_HLEN + len;
|
||||
if (frame_len < 60) frame_len = 60; /* Minimum Ethernet frame */
|
||||
|
||||
return iface->send(iface->dev_data, frame, frame_len);
|
||||
}
|
||||
|
||||
int ethernet_recv(eth_iface_t *iface, void *buf, uint32_t bufsize,
|
||||
uint8_t *src_mac, uint16_t *ethertype) {
|
||||
if (!iface || !iface->active || !iface->recv) return -1;
|
||||
|
||||
/* Receive into temporary buffer */
|
||||
uint8_t frame[ETH_FRAME_LEN];
|
||||
int ret = iface->recv(iface->dev_data, frame, sizeof(frame));
|
||||
if (ret <= 0) return ret;
|
||||
|
||||
if ((uint32_t)ret < ETH_HLEN) return -1; /* Runt frame */
|
||||
|
||||
eth_header_t *hdr = (eth_header_t *)frame;
|
||||
|
||||
/* Copy out src MAC and ethertype if requested */
|
||||
if (src_mac) memcpy(src_mac, hdr->src, ETH_ALEN);
|
||||
if (ethertype) *ethertype = ntohs(hdr->ethertype);
|
||||
|
||||
/* Copy payload */
|
||||
uint32_t payload_len = (uint32_t)ret - ETH_HLEN;
|
||||
if (payload_len > bufsize) payload_len = bufsize;
|
||||
memcpy(buf, frame + ETH_HLEN, payload_len);
|
||||
|
||||
return (int)payload_len;
|
||||
}
|
||||
|
||||
eth_iface_t *ethernet_get_iface(uint32_t index) {
|
||||
uint32_t count = 0;
|
||||
for (uint32_t i = 0; i < ETH_MAX_IFACES; i++) {
|
||||
if (!ifaces[i].active) continue;
|
||||
if (count == index) return &ifaces[i];
|
||||
count++;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
uint32_t ethernet_get_iface_count(void) {
|
||||
return iface_count;
|
||||
}
|
||||
|
||||
eth_iface_t *ethernet_find_iface(const char *name) {
|
||||
for (uint32_t i = 0; i < ETH_MAX_IFACES; i++) {
|
||||
if (!ifaces[i].active) continue;
|
||||
if (strcmp(ifaces[i].name, name) == 0) return &ifaces[i];
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
208
src/ethernet.h
Normal file
208
src/ethernet.h
Normal file
@@ -0,0 +1,208 @@
|
||||
/**
|
||||
* @file ethernet.h
|
||||
* @brief Ethernet subsystem.
|
||||
*
|
||||
* Provides a unified abstraction over individual Ethernet NIC drivers.
|
||||
* Each NIC driver registers itself with the ethernet subsystem, which
|
||||
* then creates the `/dev/ethN` character device and exposes interface
|
||||
* information via `/sys/net`.
|
||||
*
|
||||
* Higher-level protocols (IPv4, ARP, etc.) use this subsystem to send
|
||||
* and receive Ethernet frames without knowing which NIC driver is in use.
|
||||
*/
|
||||
|
||||
#ifndef ETHERNET_H
|
||||
#define ETHERNET_H
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
/** Maximum number of Ethernet interfaces. */
|
||||
#define ETH_MAX_IFACES 8
|
||||
|
||||
/** Ethernet MAC address length. */
|
||||
#define ETH_ALEN 6
|
||||
|
||||
/** Ethernet header size (dst + src + ethertype). */
|
||||
#define ETH_HLEN 14
|
||||
|
||||
/** Maximum Ethernet payload size. */
|
||||
#define ETH_MTU 1500
|
||||
|
||||
/** Maximum Ethernet frame size (header + payload). */
|
||||
#define ETH_FRAME_LEN (ETH_HLEN + ETH_MTU)
|
||||
|
||||
/* ================================================================
|
||||
* EtherType constants
|
||||
* ================================================================ */
|
||||
#define ETHERTYPE_IPV4 0x0800 /**< IPv4 */
|
||||
#define ETHERTYPE_ARP 0x0806 /**< ARP */
|
||||
|
||||
/* ================================================================
|
||||
* Ethernet header (14 bytes)
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Ethernet frame header.
|
||||
*/
|
||||
typedef struct __attribute__((packed)) eth_header {
|
||||
uint8_t dst[ETH_ALEN]; /**< Destination MAC address. */
|
||||
uint8_t src[ETH_ALEN]; /**< Source MAC address. */
|
||||
uint16_t ethertype; /**< EtherType (big-endian). */
|
||||
} eth_header_t;
|
||||
|
||||
/* ================================================================
|
||||
* NIC driver callbacks
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Callback to send a raw Ethernet frame (complete with header).
|
||||
*
|
||||
* @param dev_data Driver-specific device pointer.
|
||||
* @param frame Complete Ethernet frame (header + payload).
|
||||
* @param len Frame length in bytes.
|
||||
* @return 0 on success, -1 on failure.
|
||||
*/
|
||||
typedef int (*eth_send_fn)(void *dev_data, const void *frame, uint32_t len);
|
||||
|
||||
/**
|
||||
* Callback to receive a raw Ethernet frame (complete with header).
|
||||
*
|
||||
* @param dev_data Driver-specific device pointer.
|
||||
* @param buf Buffer for received frame.
|
||||
* @param bufsize Buffer size.
|
||||
* @return Number of bytes received, 0 if none, -1 on error.
|
||||
*/
|
||||
typedef int (*eth_recv_fn)(void *dev_data, void *buf, uint32_t bufsize);
|
||||
|
||||
/* ================================================================
|
||||
* Ethernet interface
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Represents one Ethernet network interface.
|
||||
*/
|
||||
typedef struct eth_iface {
|
||||
char name[16]; /**< Interface name (e.g., "eth1"). */
|
||||
uint8_t mac[ETH_ALEN]; /**< MAC address. */
|
||||
uint8_t active; /**< 1 if registered, 0 if free. */
|
||||
uint8_t link_up; /**< 1 if link is up. */
|
||||
uint32_t ip_addr; /**< IPv4 address (0 if not configured). */
|
||||
uint32_t netmask; /**< Subnet mask (0 if not configured). */
|
||||
uint32_t gateway; /**< Default gateway (0 if not configured). */
|
||||
|
||||
eth_send_fn send; /**< NIC driver send callback. */
|
||||
eth_recv_fn recv; /**< NIC driver receive callback. */
|
||||
void *dev_data; /**< Opaque pointer to NIC driver state. */
|
||||
} eth_iface_t;
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Initialize the ethernet subsystem.
|
||||
* Registers sysfs namespace "net" and prepares the interface table.
|
||||
*/
|
||||
void ethernet_init(void);
|
||||
|
||||
/**
|
||||
* Register a NIC with the ethernet subsystem.
|
||||
*
|
||||
* Creates a `/dev/ethN` character device and adds the interface to the
|
||||
* internal table. Called by NIC drivers during initialization.
|
||||
*
|
||||
* @param mac 6-byte MAC address.
|
||||
* @param send Send callback function.
|
||||
* @param recv Receive callback function.
|
||||
* @param dev_data Opaque NIC device pointer (passed to callbacks).
|
||||
* @return Pointer to the registered interface, or NULL on failure.
|
||||
*/
|
||||
eth_iface_t *ethernet_register(const uint8_t *mac,
|
||||
eth_send_fn send,
|
||||
eth_recv_fn recv,
|
||||
void *dev_data);
|
||||
|
||||
/**
|
||||
* Send an Ethernet frame through an interface.
|
||||
*
|
||||
* Constructs the Ethernet header (src MAC from interface, dst from caller)
|
||||
* and dispatches through the NIC driver.
|
||||
*
|
||||
* @param iface Interface to send through.
|
||||
* @param dst_mac Destination MAC address (6 bytes).
|
||||
* @param ethertype EtherType (host byte order, will be converted to big-endian).
|
||||
* @param payload Frame payload.
|
||||
* @param len Payload length (max ETH_MTU).
|
||||
* @return 0 on success, -1 on failure.
|
||||
*/
|
||||
int ethernet_send(eth_iface_t *iface, const uint8_t *dst_mac,
|
||||
uint16_t ethertype, const void *payload, uint32_t len);
|
||||
|
||||
/**
|
||||
* Receive an Ethernet frame from an interface.
|
||||
*
|
||||
* @param iface Interface to receive from.
|
||||
* @param buf Buffer for frame payload (Ethernet header is stripped).
|
||||
* @param bufsize Buffer size.
|
||||
* @param src_mac Output: source MAC of received frame (6 bytes, can be NULL).
|
||||
* @param ethertype Output: EtherType of received frame (host byte order, can be NULL).
|
||||
* @return Payload length in bytes, 0 if no packet, -1 on error.
|
||||
*/
|
||||
int ethernet_recv(eth_iface_t *iface, void *buf, uint32_t bufsize,
|
||||
uint8_t *src_mac, uint16_t *ethertype);
|
||||
|
||||
/**
|
||||
* Get an interface by index.
|
||||
*
|
||||
* @param index 0-based index.
|
||||
* @return Pointer to interface, or NULL if index out of range.
|
||||
*/
|
||||
eth_iface_t *ethernet_get_iface(uint32_t index);
|
||||
|
||||
/**
|
||||
* Get the number of registered interfaces.
|
||||
*
|
||||
* @return Number of active Ethernet interfaces.
|
||||
*/
|
||||
uint32_t ethernet_get_iface_count(void);
|
||||
|
||||
/**
|
||||
* Find an interface by name (e.g., "eth1").
|
||||
*
|
||||
* @param name Interface name.
|
||||
* @return Pointer to interface, or NULL if not found.
|
||||
*/
|
||||
eth_iface_t *ethernet_find_iface(const char *name);
|
||||
|
||||
/**
|
||||
* Convert a 16-bit value from host to network byte order (big-endian).
|
||||
*/
|
||||
static inline uint16_t htons(uint16_t h) {
|
||||
return (uint16_t)((h >> 8) | (h << 8));
|
||||
}
|
||||
|
||||
/**
|
||||
* Convert a 16-bit value from network to host byte order.
|
||||
*/
|
||||
static inline uint16_t ntohs(uint16_t n) {
|
||||
return htons(n);
|
||||
}
|
||||
|
||||
/**
|
||||
* Convert a 32-bit value from host to network byte order (big-endian).
|
||||
*/
|
||||
static inline uint32_t htonl(uint32_t h) {
|
||||
return ((h >> 24) & 0x000000FF) |
|
||||
((h >> 8) & 0x0000FF00) |
|
||||
((h << 8) & 0x00FF0000) |
|
||||
((h << 24) & 0xFF000000);
|
||||
}
|
||||
|
||||
/**
|
||||
* Convert a 32-bit value from network to host byte order.
|
||||
*/
|
||||
static inline uint32_t ntohl(uint32_t n) {
|
||||
return htonl(n);
|
||||
}
|
||||
|
||||
#endif /* ETHERNET_H */
|
||||
279
src/ipv4.c
Normal file
279
src/ipv4.c
Normal file
@@ -0,0 +1,279 @@
|
||||
/**
|
||||
* @file ipv4.c
|
||||
* @brief IPv4 network layer implementation.
|
||||
*
|
||||
* Handles IPv4 packet construction, parsing, checksum calculation,
|
||||
* and routing to the appropriate Ethernet interface.
|
||||
*
|
||||
* Incoming Ethernet frames with EtherType 0x0800 are passed to
|
||||
* ipv4_receive(), which validates the header and dispatches to
|
||||
* registered protocol handlers (ICMP, UDP, TCP, etc.).
|
||||
*/
|
||||
|
||||
#include "ipv4.h"
|
||||
#include "ethernet.h"
|
||||
#include <string.h>
|
||||
|
||||
/* Debug print helpers */
|
||||
extern void offset_print(const char *str);
|
||||
extern void print_hex(uint32_t val);
|
||||
|
||||
/* ================================================================
|
||||
* Global state
|
||||
* ================================================================ */
|
||||
|
||||
/** Maximum number of registered protocol handlers. */
|
||||
#define MAX_PROTO_HANDLERS 16
|
||||
|
||||
/** Protocol handler entry. */
|
||||
typedef struct {
|
||||
uint8_t protocol;
|
||||
ipv4_proto_handler_t handler;
|
||||
} proto_handler_t;
|
||||
|
||||
/** Registered protocol handlers. */
|
||||
static proto_handler_t proto_handlers[MAX_PROTO_HANDLERS];
|
||||
static uint32_t proto_handler_count = 0;
|
||||
|
||||
/** Global IP identification counter. */
|
||||
static uint16_t ip_id_counter = 1;
|
||||
|
||||
/* ================================================================
|
||||
* Checksum
|
||||
* ================================================================ */
|
||||
|
||||
uint16_t ipv4_checksum(const void *data, uint32_t len) {
|
||||
const uint16_t *words = (const uint16_t *)data;
|
||||
uint32_t sum = 0;
|
||||
|
||||
while (len > 1) {
|
||||
sum += *words++;
|
||||
len -= 2;
|
||||
}
|
||||
|
||||
/* Add left-over byte, if any */
|
||||
if (len == 1) {
|
||||
sum += *(const uint8_t *)words;
|
||||
}
|
||||
|
||||
/* Fold 32-bit sum into 16 bits */
|
||||
while (sum >> 16) {
|
||||
sum = (sum & 0xFFFF) + (sum >> 16);
|
||||
}
|
||||
|
||||
return (uint16_t)(~sum);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Address conversion helpers
|
||||
* ================================================================ */
|
||||
|
||||
uint32_t ipv4_aton(const char *str) {
|
||||
uint32_t octets[4] = {0};
|
||||
int idx = 0;
|
||||
|
||||
for (const char *p = str; *p && idx < 4; p++) {
|
||||
if (*p == '.') {
|
||||
idx++;
|
||||
} else if (*p >= '0' && *p <= '9') {
|
||||
octets[idx] = octets[idx] * 10 + (uint32_t)(*p - '0');
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return (octets[0] << 24) | (octets[1] << 16) |
|
||||
(octets[2] << 8) | octets[3];
|
||||
}
|
||||
|
||||
char *ipv4_ntoa(uint32_t ip, char *buf, uint32_t size) {
|
||||
if (size < 16) { buf[0] = '\0'; return buf; }
|
||||
|
||||
int pos = 0;
|
||||
for (int i = 0; i < 4; i++) {
|
||||
if (i > 0) buf[pos++] = '.';
|
||||
uint8_t octet = (uint8_t)((ip >> (24 - i * 8)) & 0xFF);
|
||||
if (octet >= 100) {
|
||||
buf[pos++] = (char)('0' + octet / 100);
|
||||
buf[pos++] = (char)('0' + (octet % 100) / 10);
|
||||
buf[pos++] = (char)('0' + octet % 10);
|
||||
} else if (octet >= 10) {
|
||||
buf[pos++] = (char)('0' + octet / 10);
|
||||
buf[pos++] = (char)('0' + octet % 10);
|
||||
} else {
|
||||
buf[pos++] = (char)('0' + octet);
|
||||
}
|
||||
}
|
||||
buf[pos] = '\0';
|
||||
return buf;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Routing
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Find the Ethernet interface for a given destination IP.
|
||||
*
|
||||
* Simple routing: check if dst is on a directly-connected subnet.
|
||||
* If not, use the first interface with a configured gateway.
|
||||
* Fallback: first interface.
|
||||
*
|
||||
* @param dst_ip Destination IP (host byte order).
|
||||
* @return Interface index, or -1 if no interfaces.
|
||||
*/
|
||||
static int ipv4_route(uint32_t dst_ip) {
|
||||
uint32_t count = ethernet_get_iface_count();
|
||||
if (count == 0) return -1;
|
||||
|
||||
/* Check for directly-connected subnet */
|
||||
for (uint32_t i = 0; i < count; i++) {
|
||||
eth_iface_t *iface = ethernet_get_iface(i);
|
||||
if (!iface || !iface->active) continue;
|
||||
if (iface->ip_addr == 0 || iface->netmask == 0) continue;
|
||||
|
||||
if ((dst_ip & iface->netmask) == (iface->ip_addr & iface->netmask)) {
|
||||
return (int)i;
|
||||
}
|
||||
}
|
||||
|
||||
/* Use first interface with a gateway */
|
||||
for (uint32_t i = 0; i < count; i++) {
|
||||
eth_iface_t *iface = ethernet_get_iface(i);
|
||||
if (!iface || !iface->active) continue;
|
||||
if (iface->gateway != 0) return (int)i;
|
||||
}
|
||||
|
||||
/* Fallback: first active interface */
|
||||
for (uint32_t i = 0; i < count; i++) {
|
||||
eth_iface_t *iface = ethernet_get_iface(i);
|
||||
if (iface && iface->active) return (int)i;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Send
|
||||
* ================================================================ */
|
||||
|
||||
int ipv4_send_iface(uint32_t iface_idx, uint32_t dst_ip, uint8_t protocol,
|
||||
const void *payload, uint32_t len) {
|
||||
if (len > IPV4_MTU) return -1;
|
||||
|
||||
eth_iface_t *iface = ethernet_get_iface(iface_idx);
|
||||
if (!iface || !iface->active) return -1;
|
||||
|
||||
/* Build IPv4 packet on stack */
|
||||
uint8_t packet[ETH_MTU];
|
||||
ipv4_header_t *hdr = (ipv4_header_t *)packet;
|
||||
|
||||
hdr->ihl_version = 0x45; /* IPv4, IHL=5 (20 bytes) */
|
||||
hdr->tos = 0;
|
||||
hdr->total_length = htons((uint16_t)(IPV4_HLEN + len));
|
||||
hdr->identification = htons(ip_id_counter++);
|
||||
hdr->flags_fragoff = htons(IPV4_FLAG_DF); /* Don't fragment */
|
||||
hdr->ttl = 64;
|
||||
hdr->protocol = protocol;
|
||||
hdr->checksum = 0;
|
||||
hdr->src_ip = htonl(iface->ip_addr);
|
||||
hdr->dst_ip = htonl(dst_ip);
|
||||
|
||||
/* Compute header checksum */
|
||||
hdr->checksum = ipv4_checksum(hdr, IPV4_HLEN);
|
||||
|
||||
/* Copy payload */
|
||||
memcpy(packet + IPV4_HLEN, payload, len);
|
||||
|
||||
/* Determine destination MAC.
|
||||
* For now, use broadcast (FF:FF:FF:FF:FF:FF) — the ARP subsystem
|
||||
* will override this once implemented. */
|
||||
uint8_t dst_mac[6] = {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
|
||||
|
||||
/* TODO: ARP lookup for dst_ip (or gateway if not on-link) */
|
||||
|
||||
return ethernet_send(iface, dst_mac, ETHERTYPE_IPV4, packet,
|
||||
IPV4_HLEN + len);
|
||||
}
|
||||
|
||||
int ipv4_send(uint32_t dst_ip, uint8_t protocol,
|
||||
const void *payload, uint32_t len) {
|
||||
int iface_idx = ipv4_route(dst_ip);
|
||||
if (iface_idx < 0) return -1;
|
||||
return ipv4_send_iface((uint32_t)iface_idx, dst_ip, protocol, payload, len);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Receive
|
||||
* ================================================================ */
|
||||
|
||||
void ipv4_receive(const void *data, uint32_t len, uint32_t iface_idx) {
|
||||
if (len < IPV4_HLEN) return;
|
||||
|
||||
const ipv4_header_t *hdr = (const ipv4_header_t *)data;
|
||||
|
||||
/* Verify version */
|
||||
if ((hdr->ihl_version >> 4) != 4) return;
|
||||
|
||||
/* Verify header length */
|
||||
uint32_t ihl = (uint32_t)(hdr->ihl_version & 0x0F) * 4;
|
||||
if (ihl < IPV4_HLEN || ihl > len) return;
|
||||
|
||||
/* Verify checksum */
|
||||
if (ipv4_checksum(data, ihl) != 0) return;
|
||||
|
||||
uint16_t total_len = ntohs(hdr->total_length);
|
||||
if (total_len > len) return;
|
||||
|
||||
uint32_t src_ip = ntohl(hdr->src_ip);
|
||||
uint32_t dst_ip = ntohl(hdr->dst_ip);
|
||||
const uint8_t *payload = (const uint8_t *)data + ihl;
|
||||
uint32_t payload_len = total_len - ihl;
|
||||
|
||||
/* Check if this packet is for us */
|
||||
eth_iface_t *iface = ethernet_get_iface(iface_idx);
|
||||
if (iface && iface->ip_addr != 0) {
|
||||
if (dst_ip != iface->ip_addr &&
|
||||
dst_ip != 0xFFFFFFFF && /* broadcast */
|
||||
(dst_ip & ~iface->netmask) != ~iface->netmask) {
|
||||
return; /* Not for us */
|
||||
}
|
||||
}
|
||||
|
||||
/* Dispatch to registered protocol handler */
|
||||
for (uint32_t i = 0; i < proto_handler_count; i++) {
|
||||
if (proto_handlers[i].protocol == hdr->protocol) {
|
||||
proto_handlers[i].handler(src_ip, dst_ip, payload,
|
||||
payload_len, iface_idx);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
/* No handler registered for this protocol */
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Protocol registration
|
||||
* ================================================================ */
|
||||
|
||||
int ipv4_register_proto(uint8_t protocol, ipv4_proto_handler_t handler) {
|
||||
if (proto_handler_count >= MAX_PROTO_HANDLERS) return -1;
|
||||
|
||||
proto_handlers[proto_handler_count].protocol = protocol;
|
||||
proto_handlers[proto_handler_count].handler = handler;
|
||||
proto_handler_count++;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Initialization
|
||||
* ================================================================ */
|
||||
|
||||
void ipv4_init(void) {
|
||||
memset(proto_handlers, 0, sizeof(proto_handlers));
|
||||
proto_handler_count = 0;
|
||||
ip_id_counter = 1;
|
||||
|
||||
offset_print(" IPv4: initialized\n");
|
||||
}
|
||||
158
src/ipv4.h
Normal file
158
src/ipv4.h
Normal file
@@ -0,0 +1,158 @@
|
||||
/**
|
||||
* @file ipv4.h
|
||||
* @brief IPv4 network layer.
|
||||
*
|
||||
* Provides IPv4 packet construction, parsing, and routing.
|
||||
* Sits on top of the Ethernet subsystem and provides the foundation
|
||||
* for higher-level protocols (ICMP, UDP, TCP).
|
||||
*
|
||||
* The IPv4 stack maintains a simple routing table: each Ethernet
|
||||
* interface has an IP address, netmask, and default gateway.
|
||||
* Outbound packets are routed to the appropriate interface.
|
||||
*
|
||||
* Inbound packets are dispatched to registered protocol handlers
|
||||
* based on the IP protocol field.
|
||||
*/
|
||||
|
||||
#ifndef IPV4_H
|
||||
#define IPV4_H
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
/* ================================================================
|
||||
* IPv4 header (20 bytes minimum, no options)
|
||||
* ================================================================ */
|
||||
|
||||
/** IPv4 header structure (network byte order in memory). */
|
||||
typedef struct __attribute__((packed)) ipv4_header {
|
||||
uint8_t ihl_version; /**< Version (4) and IHL (5 for no options). */
|
||||
uint8_t tos; /**< Type of Service. */
|
||||
uint16_t total_length; /**< Total datagram length (header + data). */
|
||||
uint16_t identification; /**< Fragment identification. */
|
||||
uint16_t flags_fragoff; /**< Flags (3 bits) + Fragment Offset (13 bits). */
|
||||
uint8_t ttl; /**< Time to Live. */
|
||||
uint8_t protocol; /**< Upper-layer protocol number. */
|
||||
uint16_t checksum; /**< Header checksum. */
|
||||
uint32_t src_ip; /**< Source IP address (network byte order). */
|
||||
uint32_t dst_ip; /**< Destination IP address (network byte order). */
|
||||
} ipv4_header_t;
|
||||
|
||||
/** IPv4 header size (no options). */
|
||||
#define IPV4_HLEN 20
|
||||
|
||||
/** Maximum IPv4 payload over Ethernet (1500 - 20). */
|
||||
#define IPV4_MTU 1480
|
||||
|
||||
/* ================================================================
|
||||
* Protocol numbers
|
||||
* ================================================================ */
|
||||
#define IP_PROTO_ICMP 1
|
||||
#define IP_PROTO_TCP 6
|
||||
#define IP_PROTO_UDP 17
|
||||
|
||||
/* ================================================================
|
||||
* IPv4 flags
|
||||
* ================================================================ */
|
||||
#define IPV4_FLAG_DF 0x4000 /**< Don't Fragment. */
|
||||
#define IPV4_FLAG_MF 0x2000 /**< More Fragments. */
|
||||
|
||||
/* ================================================================
|
||||
* Protocol handler callback
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Callback for handling incoming IPv4 packets of a specific protocol.
|
||||
*
|
||||
* @param src_ip Source IP address (host byte order).
|
||||
* @param dst_ip Destination IP address (host byte order).
|
||||
* @param payload Protocol payload (past IPv4 header).
|
||||
* @param len Payload length.
|
||||
* @param iface_idx Ethernet interface index the packet arrived on.
|
||||
*/
|
||||
typedef void (*ipv4_proto_handler_t)(uint32_t src_ip, uint32_t dst_ip,
|
||||
const void *payload, uint32_t len,
|
||||
uint32_t iface_idx);
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Initialize the IPv4 subsystem.
|
||||
*/
|
||||
void ipv4_init(void);
|
||||
|
||||
/**
|
||||
* Send an IPv4 packet.
|
||||
*
|
||||
* Automatically routes to the correct Ethernet interface based on
|
||||
* the destination IP and configured routes.
|
||||
*
|
||||
* @param dst_ip Destination IP (host byte order).
|
||||
* @param protocol IP protocol number (e.g., IP_PROTO_UDP).
|
||||
* @param payload Packet payload.
|
||||
* @param len Payload length (max IPV4_MTU).
|
||||
* @return 0 on success, -1 on failure.
|
||||
*/
|
||||
int ipv4_send(uint32_t dst_ip, uint8_t protocol,
|
||||
const void *payload, uint32_t len);
|
||||
|
||||
/**
|
||||
* Send an IPv4 packet through a specific interface.
|
||||
*
|
||||
* @param iface_idx Ethernet interface index.
|
||||
* @param dst_ip Destination IP (host byte order).
|
||||
* @param protocol IP protocol number.
|
||||
* @param payload Packet payload.
|
||||
* @param len Payload length.
|
||||
* @return 0 on success, -1 on failure.
|
||||
*/
|
||||
int ipv4_send_iface(uint32_t iface_idx, uint32_t dst_ip, uint8_t protocol,
|
||||
const void *payload, uint32_t len);
|
||||
|
||||
/**
|
||||
* Process an incoming IPv4 packet (called from Ethernet layer).
|
||||
*
|
||||
* @param data Raw IPv4 packet (header + payload).
|
||||
* @param len Total packet length.
|
||||
* @param iface_idx Interface index the packet arrived on.
|
||||
*/
|
||||
void ipv4_receive(const void *data, uint32_t len, uint32_t iface_idx);
|
||||
|
||||
/**
|
||||
* Register a protocol handler.
|
||||
*
|
||||
* @param protocol IP protocol number.
|
||||
* @param handler Callback function.
|
||||
* @return 0 on success, -1 if table is full.
|
||||
*/
|
||||
int ipv4_register_proto(uint8_t protocol, ipv4_proto_handler_t handler);
|
||||
|
||||
/**
|
||||
* Compute the Internet checksum over a buffer.
|
||||
*
|
||||
* @param data Buffer.
|
||||
* @param len Length in bytes.
|
||||
* @return Checksum in network byte order.
|
||||
*/
|
||||
uint16_t ipv4_checksum(const void *data, uint32_t len);
|
||||
|
||||
/**
|
||||
* Convert an IPv4 address string "A.B.C.D" to a uint32_t (host byte order).
|
||||
*
|
||||
* @param str Dotted-decimal string.
|
||||
* @return IPv4 address, or 0 on failure.
|
||||
*/
|
||||
uint32_t ipv4_aton(const char *str);
|
||||
|
||||
/**
|
||||
* Format an IPv4 address (host byte order) into a dotted-decimal string.
|
||||
*
|
||||
* @param ip IPv4 address.
|
||||
* @param buf Output buffer (at least 16 bytes).
|
||||
* @param size Buffer size.
|
||||
* @return Pointer to buf.
|
||||
*/
|
||||
char *ipv4_ntoa(uint32_t ip, char *buf, uint32_t size);
|
||||
|
||||
#endif /* IPV4_H */
|
||||
@@ -4,6 +4,8 @@
|
||||
#include "syscall.h"
|
||||
#include "keyboard.h"
|
||||
#include "floppy.h"
|
||||
#include "ne2000.h"
|
||||
#include "e3c509.h"
|
||||
#include <stdint.h>
|
||||
|
||||
/* Forward declaration for kernel panic or similar */
|
||||
@@ -68,6 +70,12 @@ void isr_handler(registers_t *regs)
|
||||
} else if (regs->int_no == 38) {
|
||||
/* Floppy IRQ */
|
||||
floppy_irq();
|
||||
} else if (regs->int_no == 41) {
|
||||
/* NE2000 Ethernet IRQ (IRQ 9, vector 41) */
|
||||
ne2k_irq();
|
||||
} else if (regs->int_no == 42) {
|
||||
/* 3C509B Ethernet IRQ (IRQ 10, vector 42) */
|
||||
e3c509_irq();
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
24
src/kernel.c
24
src/kernel.c
@@ -22,6 +22,12 @@
|
||||
#include "mbr.h"
|
||||
#include "fat32.h"
|
||||
#include "keyboard.h"
|
||||
#include "ethernet.h"
|
||||
#include "ipv4.h"
|
||||
#include "arp.h"
|
||||
#include "dhcp.h"
|
||||
#include "udp.h"
|
||||
#include "tcp.h"
|
||||
#include "framebuffer.h"
|
||||
|
||||
/* Global framebuffer info, parsed from multiboot2 tags. */
|
||||
@@ -415,6 +421,24 @@ void kernel_main(uint32_t magic, uint32_t addr) {
|
||||
fb_info.pitch = 80 * 2;
|
||||
}
|
||||
|
||||
ethernet_init();
|
||||
offset_print("Ethernet subsystem initialized\n");
|
||||
|
||||
ipv4_init();
|
||||
offset_print("IPv4 stack initialized\n");
|
||||
|
||||
arp_init();
|
||||
offset_print("ARP subsystem initialized\n");
|
||||
|
||||
dhcp_init();
|
||||
offset_print("DHCP subsystem initialized\n");
|
||||
|
||||
udp_init();
|
||||
offset_print("UDP stack initialized\n");
|
||||
|
||||
tcp_init();
|
||||
offset_print("TCP stack initialized\n");
|
||||
|
||||
init_drivers();
|
||||
EARLY_PRINT("DRV ");
|
||||
offset_print("Drivers initialized\n");
|
||||
|
||||
541
src/ne2000.c
Normal file
541
src/ne2000.c
Normal file
@@ -0,0 +1,541 @@
|
||||
/**
|
||||
* @file ne2000.c
|
||||
* @brief NE2000-compatible ISA Ethernet NIC driver implementation.
|
||||
*
|
||||
* Drives NE2000-compatible NICs based on the DP8390 Ethernet controller.
|
||||
* Uses programmed I/O (PIO) remote DMA for data transfer and IRQ-driven
|
||||
* packet reception.
|
||||
*
|
||||
* The driver probes I/O base 0x300 by performing a reset and checking
|
||||
* the ISR reset bit. It reads the MAC address from the card's PROM
|
||||
* (first 6 bytes of on-card memory), configures the TX and RX ring
|
||||
* buffers, and registers as a character device with devicefs.
|
||||
*
|
||||
* NE2000 memory layout (16 KiB, 64 pages of 256 bytes):
|
||||
* Pages 0x40-0x45: TX buffer (1536 bytes, holds 1 MTU frame)
|
||||
* Pages 0x46-0x7F: RX ring buffer (~14.5 KiB)
|
||||
*/
|
||||
|
||||
#include "ne2000.h"
|
||||
#include "port_io.h"
|
||||
#include "pic.h"
|
||||
#include "devicefs.h"
|
||||
#include "ethernet.h"
|
||||
#include "driver.h"
|
||||
#include <string.h>
|
||||
|
||||
/* Debug print helpers defined in kernel.c */
|
||||
extern void offset_print(const char *str);
|
||||
extern void print_hex(uint32_t val);
|
||||
|
||||
/* ================================================================
|
||||
* Global state
|
||||
* ================================================================ */
|
||||
|
||||
/** Single NE2000 device (we only support one for now). */
|
||||
static ne2k_device_t ne2k_dev;
|
||||
|
||||
/** Volatile flag set by IRQ handler when a packet arrives. */
|
||||
static volatile int ne2k_rx_ready = 0;
|
||||
|
||||
/** Volatile flag set by IRQ handler when transmit completes. */
|
||||
static volatile int ne2k_tx_done = 0;
|
||||
|
||||
/* ================================================================
|
||||
* Register access helpers
|
||||
* ================================================================ */
|
||||
|
||||
/** Write to a NE2000 register. */
|
||||
static inline void ne2k_write(uint16_t base, uint8_t reg, uint8_t val) {
|
||||
outb(base + reg, val);
|
||||
}
|
||||
|
||||
/** Read from a NE2000 register. */
|
||||
static inline uint8_t ne2k_read(uint16_t base, uint8_t reg) {
|
||||
return inb(base + reg);
|
||||
}
|
||||
|
||||
/**
|
||||
* Select a register page (0, 1, or 2).
|
||||
* Preserves the STA/STP and RD bits.
|
||||
*/
|
||||
static void ne2k_page(uint16_t base, uint8_t page) {
|
||||
uint8_t cr = ne2k_read(base, NE2K_CR);
|
||||
cr = (cr & ~(CR_PS0 | CR_PS1)) | ((page & 0x03) << 6);
|
||||
ne2k_write(base, NE2K_CR, cr);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Remote DMA (PIO) operations
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Read bytes from the NE2000's on-card memory via remote DMA.
|
||||
*
|
||||
* @param base I/O base address.
|
||||
* @param src Source address in card memory (byte address).
|
||||
* @param dst Destination buffer in system RAM.
|
||||
* @param len Number of bytes to read.
|
||||
*/
|
||||
static void ne2k_dma_read(uint16_t base, uint16_t src, void *dst, uint16_t len) {
|
||||
ne2k_write(base, NE2K_CR, CR_STA | CR_DMA_ABORT);
|
||||
|
||||
/* Set remote DMA byte count */
|
||||
ne2k_write(base, NE2K_RBCR0, (uint8_t)(len & 0xFF));
|
||||
ne2k_write(base, NE2K_RBCR1, (uint8_t)((len >> 8) & 0xFF));
|
||||
|
||||
/* Set remote DMA start address */
|
||||
ne2k_write(base, NE2K_RSAR0, (uint8_t)(src & 0xFF));
|
||||
ne2k_write(base, NE2K_RSAR1, (uint8_t)((src >> 8) & 0xFF));
|
||||
|
||||
/* Start remote DMA read */
|
||||
ne2k_write(base, NE2K_CR, CR_STA | CR_DMA_READ);
|
||||
|
||||
/* Read data from data port */
|
||||
uint8_t *buf = (uint8_t *)dst;
|
||||
uint16_t words = len / 2;
|
||||
for (uint16_t i = 0; i < words; i++) {
|
||||
uint16_t w = inw(base + NE2K_DATA);
|
||||
buf[i * 2] = (uint8_t)(w & 0xFF);
|
||||
buf[i * 2 + 1] = (uint8_t)((w >> 8) & 0xFF);
|
||||
}
|
||||
if (len & 1) {
|
||||
uint16_t w = inw(base + NE2K_DATA);
|
||||
buf[len - 1] = (uint8_t)(w & 0xFF);
|
||||
}
|
||||
|
||||
/* Wait for DMA complete */
|
||||
int timeout = 100000;
|
||||
while (!(ne2k_read(base, NE2K_ISR) & ISR_RDC) && --timeout > 0);
|
||||
|
||||
/* Acknowledge RDC */
|
||||
ne2k_write(base, NE2K_ISR, ISR_RDC);
|
||||
}
|
||||
|
||||
/**
|
||||
* Write bytes to the NE2000's on-card memory via remote DMA.
|
||||
*
|
||||
* @param base I/O base address.
|
||||
* @param dst Destination address in card memory (byte address).
|
||||
* @param src Source buffer in system RAM.
|
||||
* @param len Number of bytes to write.
|
||||
*/
|
||||
static void ne2k_dma_write(uint16_t base, uint16_t dst, const void *src,
|
||||
uint16_t len) {
|
||||
ne2k_write(base, NE2K_CR, CR_STA | CR_DMA_ABORT);
|
||||
|
||||
/* Set remote DMA byte count */
|
||||
ne2k_write(base, NE2K_RBCR0, (uint8_t)(len & 0xFF));
|
||||
ne2k_write(base, NE2K_RBCR1, (uint8_t)((len >> 8) & 0xFF));
|
||||
|
||||
/* Set remote DMA start address */
|
||||
ne2k_write(base, NE2K_RSAR0, (uint8_t)(dst & 0xFF));
|
||||
ne2k_write(base, NE2K_RSAR1, (uint8_t)((dst >> 8) & 0xFF));
|
||||
|
||||
/* Start remote DMA write */
|
||||
ne2k_write(base, NE2K_CR, CR_STA | CR_DMA_WRITE);
|
||||
|
||||
/* Write data to data port */
|
||||
const uint8_t *buf = (const uint8_t *)src;
|
||||
uint16_t words = len / 2;
|
||||
for (uint16_t i = 0; i < words; i++) {
|
||||
uint16_t w = (uint16_t)buf[i * 2] | ((uint16_t)buf[i * 2 + 1] << 8);
|
||||
outw(base + NE2K_DATA, w);
|
||||
}
|
||||
if (len & 1) {
|
||||
outw(base + NE2K_DATA, (uint16_t)buf[len - 1]);
|
||||
}
|
||||
|
||||
/* Wait for DMA complete */
|
||||
int timeout = 100000;
|
||||
while (!(ne2k_read(base, NE2K_ISR) & ISR_RDC) && --timeout > 0);
|
||||
|
||||
/* Acknowledge RDC */
|
||||
ne2k_write(base, NE2K_ISR, ISR_RDC);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Packet send / receive
|
||||
* ================================================================ */
|
||||
|
||||
int ne2k_send(ne2k_device_t *dev, const void *data, uint32_t len) {
|
||||
if (!dev || !dev->present) return -1;
|
||||
if (len > ETH_FRAME_MAX) return -1;
|
||||
|
||||
/* Minimum Ethernet frame size is 60 bytes (without CRC) */
|
||||
uint32_t send_len = len;
|
||||
if (send_len < 60) send_len = 60;
|
||||
|
||||
uint16_t base = dev->io_base;
|
||||
|
||||
/* Write packet to TX buffer on card */
|
||||
ne2k_dma_write(base, (uint16_t)(NE2K_TX_START << 8),
|
||||
data, (uint16_t)len);
|
||||
|
||||
/* If we need to pad, write zeros for the remaining bytes.
|
||||
* For simplicity, the DMA write already handles partial words.
|
||||
* The card will transmit send_len bytes from the TX page. */
|
||||
|
||||
/* Set transmit page start */
|
||||
ne2k_page(base, 0);
|
||||
ne2k_write(base, NE2K_TPSR, NE2K_TX_START);
|
||||
ne2k_write(base, NE2K_TBCR0, (uint8_t)(send_len & 0xFF));
|
||||
ne2k_write(base, NE2K_TBCR1, (uint8_t)((send_len >> 8) & 0xFF));
|
||||
|
||||
/* Trigger transmit */
|
||||
ne2k_tx_done = 0;
|
||||
ne2k_write(base, NE2K_CR, CR_STA | CR_TXP | CR_DMA_ABORT);
|
||||
|
||||
/* Wait for transmit completion (with timeout) */
|
||||
int timeout = 1000000;
|
||||
while (!ne2k_tx_done && --timeout > 0) {
|
||||
asm volatile("pause");
|
||||
}
|
||||
|
||||
if (timeout == 0) {
|
||||
offset_print(" NE2K: tx timeout\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int ne2k_recv(ne2k_device_t *dev, void *buf, uint32_t bufsize) {
|
||||
if (!dev || !dev->present) return -1;
|
||||
|
||||
uint16_t base = dev->io_base;
|
||||
|
||||
/* Check if there's a packet in the ring */
|
||||
ne2k_page(base, 1);
|
||||
uint8_t curr = ne2k_read(base, NE2K_CURR);
|
||||
ne2k_page(base, 0);
|
||||
|
||||
if (dev->next_rx_page == curr) {
|
||||
return 0; /* Ring is empty */
|
||||
}
|
||||
|
||||
/* Read the 4-byte rx header from the ring */
|
||||
ne2k_rx_header_t hdr;
|
||||
uint16_t hdr_addr = (uint16_t)(dev->next_rx_page << 8);
|
||||
ne2k_dma_read(base, hdr_addr, &hdr, sizeof(hdr));
|
||||
|
||||
/* Sanity check the header */
|
||||
uint16_t pkt_len = hdr.length - (uint16_t)sizeof(ne2k_rx_header_t);
|
||||
if (pkt_len > ETH_FRAME_MAX || pkt_len == 0) {
|
||||
/* Invalid packet, advance pointer and skip */
|
||||
dev->next_rx_page = hdr.next_page;
|
||||
ne2k_write(base, NE2K_BNRY,
|
||||
(dev->next_rx_page == NE2K_RX_START)
|
||||
? NE2K_RX_STOP - 1
|
||||
: dev->next_rx_page - 1);
|
||||
return -1;
|
||||
}
|
||||
|
||||
/* Determine how many bytes to copy to caller */
|
||||
uint32_t copy_len = pkt_len;
|
||||
if (copy_len > bufsize) copy_len = bufsize;
|
||||
|
||||
/* Read packet data (starts after the 4-byte header) */
|
||||
uint16_t data_addr = hdr_addr + (uint16_t)sizeof(ne2k_rx_header_t);
|
||||
|
||||
/* Handle ring buffer wrap-around */
|
||||
uint16_t ring_end = (uint16_t)(NE2K_RX_STOP << 8);
|
||||
uint16_t ring_start = (uint16_t)(NE2K_RX_START << 8);
|
||||
|
||||
if (data_addr + copy_len > ring_end) {
|
||||
/* Wraps around */
|
||||
uint16_t first_part = ring_end - data_addr;
|
||||
ne2k_dma_read(base, data_addr, buf, first_part);
|
||||
ne2k_dma_read(base, ring_start,
|
||||
(uint8_t *)buf + first_part,
|
||||
(uint16_t)(copy_len - first_part));
|
||||
} else {
|
||||
ne2k_dma_read(base, data_addr, buf, (uint16_t)copy_len);
|
||||
}
|
||||
|
||||
/* Advance BNRY to next_page (one behind CURR means buffer is full) */
|
||||
dev->next_rx_page = hdr.next_page;
|
||||
ne2k_write(base, NE2K_BNRY,
|
||||
(dev->next_rx_page == NE2K_RX_START)
|
||||
? NE2K_RX_STOP - 1
|
||||
: dev->next_rx_page - 1);
|
||||
|
||||
return (int)copy_len;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* IRQ handler
|
||||
* ================================================================ */
|
||||
|
||||
void ne2k_irq(void) {
|
||||
if (!ne2k_dev.present) return;
|
||||
|
||||
uint16_t base = ne2k_dev.io_base;
|
||||
uint8_t isr = ne2k_read(base, NE2K_ISR);
|
||||
|
||||
if (isr & ISR_PRX) {
|
||||
/* Packet received */
|
||||
ne2k_rx_ready = 1;
|
||||
ne2k_write(base, NE2K_ISR, ISR_PRX);
|
||||
}
|
||||
|
||||
if (isr & ISR_PTX) {
|
||||
/* Packet transmitted */
|
||||
ne2k_tx_done = 1;
|
||||
ne2k_write(base, NE2K_ISR, ISR_PTX);
|
||||
}
|
||||
|
||||
if (isr & ISR_RXE) {
|
||||
/* Receive error */
|
||||
ne2k_write(base, NE2K_ISR, ISR_RXE);
|
||||
}
|
||||
|
||||
if (isr & ISR_TXE) {
|
||||
/* Transmit error */
|
||||
ne2k_tx_done = 1; /* Unblock sender even on error */
|
||||
ne2k_write(base, NE2K_ISR, ISR_TXE);
|
||||
}
|
||||
|
||||
if (isr & ISR_OVW) {
|
||||
/* Overflow — need to handle by resetting RX */
|
||||
ne2k_write(base, NE2K_ISR, ISR_OVW);
|
||||
}
|
||||
|
||||
if (isr & ISR_CNT) {
|
||||
/* Counter overflow */
|
||||
ne2k_read(base, NE2K_CNTR0);
|
||||
ne2k_read(base, NE2K_CNTR1);
|
||||
ne2k_read(base, NE2K_CNTR2);
|
||||
ne2k_write(base, NE2K_ISR, ISR_CNT);
|
||||
}
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Initialization
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Read the MAC address from the NE2000 PROM.
|
||||
*
|
||||
* The first 32 bytes of NE2000's memory contain the PROM data.
|
||||
* In word-wide mode, each MAC byte is duplicated in the high byte,
|
||||
* so the PROM looks like: M0 M0 M1 M1 M2 M2 M3 M3 M4 M4 M5 M5 ...
|
||||
*/
|
||||
static void ne2k_read_mac(uint16_t base, uint8_t *mac) {
|
||||
uint8_t prom[32];
|
||||
ne2k_dma_read(base, 0x0000, prom, 32);
|
||||
|
||||
/* NE2000 duplicates each byte in word mode */
|
||||
mac[0] = prom[0];
|
||||
mac[1] = prom[2];
|
||||
mac[2] = prom[4];
|
||||
mac[3] = prom[6];
|
||||
mac[4] = prom[8];
|
||||
mac[5] = prom[10];
|
||||
}
|
||||
|
||||
/**
|
||||
* Reset and initialize the NE2000 controller.
|
||||
*/
|
||||
static int ne2k_hw_init(uint16_t base) {
|
||||
/* Reset the card: read from RESET port, then write to it */
|
||||
uint8_t tmp = inb(base + NE2K_RESET);
|
||||
outb(base + NE2K_RESET, tmp);
|
||||
|
||||
/* Wait for reset to complete (ISR bit 7 = RST set) */
|
||||
int timeout = 100000;
|
||||
while (!(ne2k_read(base, NE2K_ISR) & ISR_RST) && --timeout > 0);
|
||||
if (timeout == 0) {
|
||||
offset_print(" NE2K: reset timeout\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
/* Acknowledge the reset */
|
||||
ne2k_write(base, NE2K_ISR, 0xFF);
|
||||
|
||||
/* Stop the NIC and abort any DMA */
|
||||
ne2k_write(base, NE2K_CR, CR_STP | CR_DMA_ABORT);
|
||||
|
||||
/* Data Configuration Register:
|
||||
* Word-wide transfers (WTS), normal operation (LS),
|
||||
* FIFO threshold 8 bytes (FT1) */
|
||||
ne2k_write(base, NE2K_DCR, DCR_WTS | DCR_LS | DCR_FT1);
|
||||
|
||||
/* Clear remote byte count registers (required before starting) */
|
||||
ne2k_write(base, NE2K_RBCR0, 0);
|
||||
ne2k_write(base, NE2K_RBCR1, 0);
|
||||
|
||||
/* Receive Configuration: Accept broadcast and physical match */
|
||||
ne2k_write(base, NE2K_RCR, RCR_AB);
|
||||
|
||||
/* Transmit Configuration: internal loopback during init */
|
||||
ne2k_write(base, NE2K_TCR, TCR_LB0);
|
||||
|
||||
/* Set up ring buffer pointers */
|
||||
ne2k_write(base, NE2K_PSTART, NE2K_RX_START);
|
||||
ne2k_write(base, NE2K_PSTOP, NE2K_RX_STOP);
|
||||
ne2k_write(base, NE2K_BNRY, NE2K_RX_START);
|
||||
|
||||
/* Read MAC address from PROM */
|
||||
ne2k_read_mac(base, ne2k_dev.mac);
|
||||
|
||||
/* Switch to page 1 to set PAR (Physical Address) and CURR */
|
||||
ne2k_page(base, 1);
|
||||
ne2k_write(base, NE2K_PAR0, ne2k_dev.mac[0]);
|
||||
ne2k_write(base, NE2K_PAR1, ne2k_dev.mac[1]);
|
||||
ne2k_write(base, NE2K_PAR2, ne2k_dev.mac[2]);
|
||||
ne2k_write(base, NE2K_PAR3, ne2k_dev.mac[3]);
|
||||
ne2k_write(base, NE2K_PAR4, ne2k_dev.mac[4]);
|
||||
ne2k_write(base, NE2K_PAR5, ne2k_dev.mac[5]);
|
||||
|
||||
/* Set current page (next write page for incoming packets) */
|
||||
ne2k_write(base, NE2K_CURR, NE2K_RX_START + 1);
|
||||
ne2k_dev.next_rx_page = NE2K_RX_START + 1;
|
||||
|
||||
/* Set multicast address registers to accept all multicast */
|
||||
for (int i = 0; i < 8; i++) {
|
||||
ne2k_write(base, NE2K_MAR0 + (uint8_t)i, 0xFF);
|
||||
}
|
||||
|
||||
/* Switch back to page 0 */
|
||||
ne2k_page(base, 0);
|
||||
|
||||
/* Clear all pending interrupts */
|
||||
ne2k_write(base, NE2K_ISR, 0xFF);
|
||||
|
||||
/* Enable interrupts: PRX, PTX, RXE, TXE, OVW, CNT */
|
||||
ne2k_write(base, NE2K_IMR, ISR_PRX | ISR_PTX | ISR_RXE |
|
||||
ISR_TXE | ISR_OVW | ISR_CNT);
|
||||
|
||||
/* Take the NIC out of loopback: normal transmit configuration */
|
||||
ne2k_write(base, NE2K_TCR, 0x00);
|
||||
|
||||
/* Start the NIC */
|
||||
ne2k_write(base, NE2K_CR, CR_STA | CR_DMA_ABORT);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Driver framework integration
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Probe for an NE2000 card at the default I/O base.
|
||||
*
|
||||
* Reset the card, check the ISR reset bit, then verify by reading
|
||||
* the PROM (MAC address). If all bytes are 0xFF or 0x00 there is
|
||||
* no real card at this address.
|
||||
*/
|
||||
static driver_probe_result_t ne2k_probe(void) {
|
||||
uint16_t base = NE2K_DEFAULT_IOBASE;
|
||||
|
||||
/* Try to reset */
|
||||
uint8_t tmp = inb(base + NE2K_RESET);
|
||||
outb(base + NE2K_RESET, tmp);
|
||||
|
||||
/* Brief delay */
|
||||
for (volatile int i = 0; i < 10000; i++) {
|
||||
asm volatile("pause");
|
||||
}
|
||||
|
||||
/* Check if the ISR reset bit is set */
|
||||
uint8_t isr = ne2k_read(base, NE2K_ISR);
|
||||
if (!(isr & ISR_RST)) {
|
||||
return DRIVER_PROBE_NOT_FOUND;
|
||||
}
|
||||
|
||||
/* Acknowledge the reset */
|
||||
ne2k_write(base, NE2K_ISR, 0xFF);
|
||||
|
||||
/* Configure for PROM reading */
|
||||
ne2k_write(base, NE2K_CR, CR_STP | CR_DMA_ABORT);
|
||||
ne2k_write(base, NE2K_DCR, DCR_WTS | DCR_LS | DCR_FT1);
|
||||
ne2k_write(base, NE2K_RBCR0, 0);
|
||||
ne2k_write(base, NE2K_RBCR1, 0);
|
||||
ne2k_write(base, NE2K_RCR, RCR_MON);
|
||||
ne2k_write(base, NE2K_TCR, TCR_LB0);
|
||||
ne2k_write(base, NE2K_PSTART, NE2K_RX_START);
|
||||
ne2k_write(base, NE2K_PSTOP, NE2K_RX_STOP);
|
||||
ne2k_write(base, NE2K_BNRY, NE2K_RX_START);
|
||||
|
||||
/* Read 32 bytes of PROM data */
|
||||
uint8_t prom[32];
|
||||
ne2k_dma_read(base, 0x0000, prom, 32);
|
||||
|
||||
/* Validate: the MAC should not be all 0xFF or all 0x00 */
|
||||
int all_ff = 1, all_00 = 1;
|
||||
for (int i = 0; i < 12; i += 2) {
|
||||
if (prom[i] != 0xFF) all_ff = 0;
|
||||
if (prom[i] != 0x00) all_00 = 0;
|
||||
}
|
||||
|
||||
if (all_ff || all_00) {
|
||||
return DRIVER_PROBE_NOT_FOUND;
|
||||
}
|
||||
|
||||
return DRIVER_PROBE_OK;
|
||||
}
|
||||
|
||||
/**
|
||||
* Initialize the NE2000 driver.
|
||||
*/
|
||||
static int ne2k_driver_init(void) {
|
||||
memset(&ne2k_dev, 0, sizeof(ne2k_dev));
|
||||
|
||||
ne2k_dev.io_base = NE2K_DEFAULT_IOBASE;
|
||||
ne2k_dev.irq = NE2K_DEFAULT_IRQ;
|
||||
|
||||
/* Enable IRQ in PIC */
|
||||
pic_clear_mask(ne2k_dev.irq);
|
||||
|
||||
/* Initialize hardware */
|
||||
if (ne2k_hw_init(ne2k_dev.io_base) != 0) {
|
||||
offset_print(" NE2K: initialization failed\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
ne2k_dev.present = 1;
|
||||
|
||||
/* Print MAC address */
|
||||
offset_print(" NE2K: MAC ");
|
||||
for (int i = 0; i < 6; i++) {
|
||||
if (i > 0) offset_print(":");
|
||||
print_hex(ne2k_dev.mac[i]);
|
||||
}
|
||||
offset_print("\n");
|
||||
|
||||
/* Register with ethernet subsystem (creates /dev/ethN) */
|
||||
ethernet_register(ne2k_dev.mac,
|
||||
(eth_send_fn)ne2k_send,
|
||||
(eth_recv_fn)ne2k_recv,
|
||||
&ne2k_dev);
|
||||
|
||||
offset_print(" NE2K: initialized on I/O ");
|
||||
print_hex(ne2k_dev.io_base);
|
||||
offset_print(" IRQ ");
|
||||
print_hex(ne2k_dev.irq);
|
||||
offset_print("\n");
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
ne2k_device_t *ne2k_get_device(void) {
|
||||
return ne2k_dev.present ? &ne2k_dev : NULL;
|
||||
}
|
||||
|
||||
int ne2k_init(void) {
|
||||
return ne2k_driver_init();
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Driver registration
|
||||
* ================================================================ */
|
||||
|
||||
static const driver_t ne2k_driver = {
|
||||
.name = "ne2000",
|
||||
.probe = ne2k_probe,
|
||||
.init = ne2k_driver_init,
|
||||
};
|
||||
|
||||
REGISTER_DRIVER(ne2k_driver);
|
||||
221
src/ne2000.h
Normal file
221
src/ne2000.h
Normal file
@@ -0,0 +1,221 @@
|
||||
/**
|
||||
* @file ne2000.h
|
||||
* @brief NE2000-compatible ISA Ethernet NIC driver.
|
||||
*
|
||||
* Drives NE2000-compatible NICs based on the DP8390 Ethernet chip.
|
||||
* The driver probes I/O base 0x300 (the common default) and uses
|
||||
* IRQ 9 for interrupt-driven packet reception.
|
||||
*
|
||||
* Packets are stored in the card's internal 16 KiB RAM using a
|
||||
* ring buffer. The driver transmits packets synchronously.
|
||||
*
|
||||
* The NE2000 registers with the devicefs as a character device
|
||||
* (named "eth1", "eth2", etc.) for the ethernet subsystem to use.
|
||||
*/
|
||||
|
||||
#ifndef NE2000_H
|
||||
#define NE2000_H
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
/* ================================================================
|
||||
* NE2000 I/O Port Layout (offsets from base)
|
||||
*
|
||||
* The DP8390 has 3 register pages selected by bits 6-7 of the
|
||||
* Command register (offset 0x00).
|
||||
* ================================================================ */
|
||||
|
||||
/** Default ISA I/O base for NE2000. */
|
||||
#define NE2K_DEFAULT_IOBASE 0x300
|
||||
|
||||
/** Default ISA IRQ for NE2000. */
|
||||
#define NE2K_DEFAULT_IRQ 9
|
||||
|
||||
/* --- Shared registers (all pages) --- */
|
||||
#define NE2K_CR 0x00 /**< Command register */
|
||||
|
||||
/* --- Page 0 read registers --- */
|
||||
#define NE2K_CLDA0 0x01 /**< Current Local DMA Address 0 */
|
||||
#define NE2K_CLDA1 0x02 /**< Current Local DMA Address 1 */
|
||||
#define NE2K_BNRY 0x03 /**< Boundary pointer (last read page) */
|
||||
#define NE2K_TSR 0x04 /**< Transmit Status Register */
|
||||
#define NE2K_NCR 0x05 /**< Number of Collisions Register */
|
||||
#define NE2K_FIFO 0x06 /**< FIFO */
|
||||
#define NE2K_ISR 0x07 /**< Interrupt Status Register */
|
||||
#define NE2K_CRDA0 0x08 /**< Current Remote DMA Address 0 */
|
||||
#define NE2K_CRDA1 0x09 /**< Current Remote DMA Address 1 */
|
||||
#define NE2K_RSR 0x0C /**< Receive Status Register */
|
||||
#define NE2K_CNTR0 0x0D /**< Tally Counter 0 (frame alignment errors) */
|
||||
#define NE2K_CNTR1 0x0E /**< Tally Counter 1 (CRC errors) */
|
||||
#define NE2K_CNTR2 0x0F /**< Tally Counter 2 (missed packets) */
|
||||
|
||||
/* --- Page 0 write registers --- */
|
||||
#define NE2K_PSTART 0x01 /**< Page Start (rx ring start, in pages) */
|
||||
#define NE2K_PSTOP 0x02 /**< Page Stop (rx ring end, in pages) */
|
||||
/* BNRY = 0x03 shared */
|
||||
#define NE2K_TPSR 0x04 /**< Transmit Page Start */
|
||||
#define NE2K_TBCR0 0x05 /**< Transmit Byte Count 0 */
|
||||
#define NE2K_TBCR1 0x06 /**< Transmit Byte Count 1 */
|
||||
/* ISR = 0x07 shared */
|
||||
#define NE2K_RSAR0 0x08 /**< Remote Start Address 0 */
|
||||
#define NE2K_RSAR1 0x09 /**< Remote Start Address 1 */
|
||||
#define NE2K_RBCR0 0x0A /**< Remote Byte Count 0 */
|
||||
#define NE2K_RBCR1 0x0B /**< Remote Byte Count 1 */
|
||||
#define NE2K_RCR 0x0C /**< Receive Configuration Register */
|
||||
#define NE2K_TCR 0x0D /**< Transmit Configuration Register */
|
||||
#define NE2K_DCR 0x0E /**< Data Configuration Register */
|
||||
#define NE2K_IMR 0x0F /**< Interrupt Mask Register */
|
||||
|
||||
/* --- Page 1 registers (r/w) --- */
|
||||
/* NE2K_CR = 0x00 */
|
||||
#define NE2K_PAR0 0x01 /**< Physical Address 0 (MAC byte 0) */
|
||||
#define NE2K_PAR1 0x02
|
||||
#define NE2K_PAR2 0x03
|
||||
#define NE2K_PAR3 0x04
|
||||
#define NE2K_PAR4 0x05
|
||||
#define NE2K_PAR5 0x06
|
||||
#define NE2K_CURR 0x07 /**< Current Page (next rx write page) */
|
||||
#define NE2K_MAR0 0x08 /**< Multicast Address Register 0 */
|
||||
|
||||
/* --- NE2000 data port (offset 0x10) --- */
|
||||
#define NE2K_DATA 0x10 /**< Data port for remote DMA */
|
||||
#define NE2K_RESET 0x1F /**< Reset port */
|
||||
|
||||
/* ================================================================
|
||||
* Command Register (CR) bits
|
||||
* ================================================================ */
|
||||
#define CR_STP 0x01 /**< Stop: software reset */
|
||||
#define CR_STA 0x02 /**< Start: activate NIC */
|
||||
#define CR_TXP 0x04 /**< Transmit Packet */
|
||||
#define CR_RD0 0x08 /**< Remote DMA command bit 0 */
|
||||
#define CR_RD1 0x10 /**< Remote DMA command bit 1 */
|
||||
#define CR_RD2 0x20 /**< Remote DMA command bit 2 (abort/complete) */
|
||||
#define CR_PS0 0x40 /**< Page Select bit 0 */
|
||||
#define CR_PS1 0x80 /**< Page Select bit 1 */
|
||||
|
||||
/** Remote DMA read */
|
||||
#define CR_DMA_READ CR_RD0
|
||||
/** Remote DMA write */
|
||||
#define CR_DMA_WRITE CR_RD1
|
||||
/** Abort/complete remote DMA */
|
||||
#define CR_DMA_ABORT CR_RD2
|
||||
|
||||
/* ================================================================
|
||||
* ISR / IMR bits
|
||||
* ================================================================ */
|
||||
#define ISR_PRX 0x01 /**< Packet Received */
|
||||
#define ISR_PTX 0x02 /**< Packet Transmitted */
|
||||
#define ISR_RXE 0x04 /**< Receive Error */
|
||||
#define ISR_TXE 0x08 /**< Transmit Error */
|
||||
#define ISR_OVW 0x10 /**< Overflow Warning */
|
||||
#define ISR_CNT 0x20 /**< Counter Overflow */
|
||||
#define ISR_RDC 0x40 /**< Remote DMA Complete */
|
||||
#define ISR_RST 0x80 /**< Reset Status */
|
||||
|
||||
/* ================================================================
|
||||
* DCR bits
|
||||
* ================================================================ */
|
||||
#define DCR_WTS 0x01 /**< Word Transfer Select (1=16-bit) */
|
||||
#define DCR_BOS 0x02 /**< Byte Order Select */
|
||||
#define DCR_LAS 0x04 /**< Long Address Select */
|
||||
#define DCR_LS 0x08 /**< Loopback Select (1=normal) */
|
||||
#define DCR_AR 0x10 /**< Auto-initialize Remote */
|
||||
#define DCR_FT0 0x20 /**< FIFO Threshold bit 0 */
|
||||
#define DCR_FT1 0x40 /**< FIFO Threshold bit 1 */
|
||||
|
||||
/* ================================================================
|
||||
* TCR bits
|
||||
* ================================================================ */
|
||||
#define TCR_LB0 0x02 /**< Loopback bit 0 */
|
||||
#define TCR_LB1 0x04 /**< Loopback bit 1 */
|
||||
|
||||
/* ================================================================
|
||||
* RCR bits
|
||||
* ================================================================ */
|
||||
#define RCR_SEP 0x01 /**< Save Errored Packets */
|
||||
#define RCR_AR 0x02 /**< Accept Runt Packets */
|
||||
#define RCR_AB 0x04 /**< Accept Broadcast */
|
||||
#define RCR_AM 0x08 /**< Accept Multicast */
|
||||
#define RCR_PRO 0x10 /**< Promiscuous Mode */
|
||||
#define RCR_MON 0x20 /**< Monitor Mode */
|
||||
|
||||
/* ================================================================
|
||||
* NE2000 ring buffer layout (16 KiB on-card RAM)
|
||||
*
|
||||
* Pages are 256 bytes each. NE2000 has 32 pages (0x40-0x80).
|
||||
* - TX buffer: page 0x40 (room for 1 MTU frame, 6 pages)
|
||||
* - RX ring: pages 0x46-0x80
|
||||
* ================================================================ */
|
||||
#define NE2K_MEM_START 0x40 /**< Start of NE2000 RAM (page number) */
|
||||
#define NE2K_MEM_END 0x80 /**< End of NE2000 RAM (page number, exclusive) */
|
||||
#define NE2K_TX_START 0x40 /**< TX buffer start page */
|
||||
#define NE2K_RX_START 0x46 /**< RX ring start page */
|
||||
#define NE2K_RX_STOP 0x80 /**< RX ring stop page (exclusive) */
|
||||
|
||||
/** Maximum Ethernet frame size. */
|
||||
#define ETH_FRAME_MAX 1518
|
||||
#define ETH_HEADER_SIZE 14
|
||||
|
||||
/* ================================================================
|
||||
* NE2000 received packet header (prepended by the card)
|
||||
* ================================================================ */
|
||||
typedef struct __attribute__((packed)) ne2k_rx_header {
|
||||
uint8_t status; /**< Receive status (matches RSR). */
|
||||
uint8_t next_page; /**< Next packet page pointer. */
|
||||
uint16_t length; /**< Total length including this header. */
|
||||
} ne2k_rx_header_t;
|
||||
|
||||
/* ================================================================
|
||||
* NE2000 device state
|
||||
* ================================================================ */
|
||||
typedef struct ne2k_device {
|
||||
uint16_t io_base; /**< I/O base address. */
|
||||
uint8_t irq; /**< IRQ number. */
|
||||
uint8_t mac[6]; /**< MAC address. */
|
||||
uint8_t next_rx_page; /**< Next page to read from RX ring. */
|
||||
int present; /**< 1 if card detected. */
|
||||
} ne2k_device_t;
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Initialize NE2000 driver.
|
||||
* Called automatically via REGISTER_DRIVER.
|
||||
*/
|
||||
int ne2k_init(void);
|
||||
|
||||
/**
|
||||
* NE2000 IRQ handler.
|
||||
* Called from isr.c when the NE2000 IRQ fires.
|
||||
*/
|
||||
void ne2k_irq(void);
|
||||
|
||||
/**
|
||||
* Send an Ethernet frame.
|
||||
*
|
||||
* @param dev NE2000 device.
|
||||
* @param data Frame data (starting with destination MAC).
|
||||
* @param len Frame length in bytes (max 1518).
|
||||
* @return 0 on success, -1 on failure.
|
||||
*/
|
||||
int ne2k_send(ne2k_device_t *dev, const void *data, uint32_t len);
|
||||
|
||||
/**
|
||||
* Receive a pending Ethernet frame.
|
||||
*
|
||||
* @param dev NE2000 device.
|
||||
* @param buf Buffer for received frame.
|
||||
* @param bufsize Buffer size.
|
||||
* @return Number of bytes received, 0 if no packet, -1 on error.
|
||||
*/
|
||||
int ne2k_recv(ne2k_device_t *dev, void *buf, uint32_t bufsize);
|
||||
|
||||
/**
|
||||
* Get the NE2000 device pointer (for eth subsystem).
|
||||
* @return Pointer to the device struct, or NULL if not present.
|
||||
*/
|
||||
ne2k_device_t *ne2k_get_device(void);
|
||||
|
||||
#endif /* NE2000_H */
|
||||
157
src/syscall.c
157
src/syscall.c
@@ -17,6 +17,8 @@
|
||||
#include "cpio.h"
|
||||
#include "paging.h"
|
||||
#include "pmm.h"
|
||||
#include "tcp.h"
|
||||
#include "udp.h"
|
||||
#include <stddef.h>
|
||||
#include <string.h>
|
||||
|
||||
@@ -339,22 +341,151 @@ static int32_t sys_readdir(registers_t *regs) {
|
||||
return (int32_t)entry.type;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Networking system calls
|
||||
* ================================================================ */
|
||||
|
||||
/** Socket type constants (kernel side). */
|
||||
#define SOCK_TYPE_TCP 0
|
||||
#define SOCK_TYPE_UDP 1
|
||||
|
||||
/** Per-process socket tracking (simple: global table). */
|
||||
#define MAX_USER_SOCKETS 16
|
||||
static struct {
|
||||
uint8_t active;
|
||||
uint8_t type; /* SOCK_TYPE_TCP or SOCK_TYPE_UDP */
|
||||
int kern_sockfd; /* Kernel-side socket index */
|
||||
} user_sockets[MAX_USER_SOCKETS];
|
||||
|
||||
/**
|
||||
* Handle SYS_SOCKET: create a network socket.
|
||||
* EBX = type (0=TCP, 1=UDP).
|
||||
* Returns user sockfd (>= 0) or -1.
|
||||
*/
|
||||
static int32_t sys_socket(registers_t *regs) {
|
||||
uint32_t type = regs->ebx;
|
||||
int kern_fd;
|
||||
|
||||
if (type == SOCK_TYPE_TCP) {
|
||||
kern_fd = tcp_socket_create();
|
||||
} else if (type == SOCK_TYPE_UDP) {
|
||||
kern_fd = udp_socket_create();
|
||||
} else {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (kern_fd < 0) return -1;
|
||||
|
||||
/* Find a free user socket slot */
|
||||
for (int i = 0; i < MAX_USER_SOCKETS; i++) {
|
||||
if (!user_sockets[i].active) {
|
||||
user_sockets[i].active = 1;
|
||||
user_sockets[i].type = (uint8_t)type;
|
||||
user_sockets[i].kern_sockfd = kern_fd;
|
||||
return i;
|
||||
}
|
||||
}
|
||||
|
||||
/* No free slots — close the kernel socket */
|
||||
if (type == SOCK_TYPE_TCP) tcp_close(kern_fd);
|
||||
else udp_close(kern_fd);
|
||||
return -1;
|
||||
}
|
||||
|
||||
/**
|
||||
* Handle SYS_CONNECT: connect a TCP socket to a remote host.
|
||||
* EBX = user sockfd, ECX = remote IP (host byte order), EDX = remote port.
|
||||
* Returns 0 on success (SYN sent), -1 on failure.
|
||||
*/
|
||||
static int32_t sys_connect(registers_t *regs) {
|
||||
int ufd = (int)regs->ebx;
|
||||
uint32_t remote_ip = regs->ecx;
|
||||
uint16_t remote_port = (uint16_t)regs->edx;
|
||||
|
||||
if (ufd < 0 || ufd >= MAX_USER_SOCKETS || !user_sockets[ufd].active)
|
||||
return -1;
|
||||
if (user_sockets[ufd].type != SOCK_TYPE_TCP)
|
||||
return -1;
|
||||
|
||||
return tcp_connect(user_sockets[ufd].kern_sockfd, remote_ip, remote_port);
|
||||
}
|
||||
|
||||
/**
|
||||
* Handle SYS_SEND: send data on a socket.
|
||||
* EBX = user sockfd, ECX = buffer pointer, EDX = length.
|
||||
* Returns bytes sent or -1.
|
||||
*/
|
||||
static int32_t sys_send(registers_t *regs) {
|
||||
int ufd = (int)regs->ebx;
|
||||
const void *buf = (const void *)regs->ecx;
|
||||
uint32_t len = regs->edx;
|
||||
|
||||
if (ufd < 0 || ufd >= MAX_USER_SOCKETS || !user_sockets[ufd].active)
|
||||
return -1;
|
||||
|
||||
if (user_sockets[ufd].type == SOCK_TYPE_TCP)
|
||||
return tcp_send(user_sockets[ufd].kern_sockfd, buf, len);
|
||||
else
|
||||
return -1; /* UDP sendto requires address — not supported via SYS_SEND */
|
||||
}
|
||||
|
||||
/**
|
||||
* Handle SYS_RECV: receive data from a socket (non-blocking).
|
||||
* EBX = user sockfd, ECX = buffer pointer, EDX = buffer size.
|
||||
* Returns bytes received, 0 if no data, -1 on error/closed.
|
||||
*/
|
||||
static int32_t sys_recv(registers_t *regs) {
|
||||
int ufd = (int)regs->ebx;
|
||||
void *buf = (void *)regs->ecx;
|
||||
uint32_t bufsize = regs->edx;
|
||||
|
||||
if (ufd < 0 || ufd >= MAX_USER_SOCKETS || !user_sockets[ufd].active)
|
||||
return -1;
|
||||
|
||||
if (user_sockets[ufd].type == SOCK_TYPE_TCP)
|
||||
return tcp_recv(user_sockets[ufd].kern_sockfd, buf, bufsize);
|
||||
else
|
||||
return -1; /* UDP recvfrom requires address — not supported via SYS_RECV */
|
||||
}
|
||||
|
||||
/**
|
||||
* Handle SYS_SOCKSTATE: get the state of a TCP socket.
|
||||
* EBX = user sockfd.
|
||||
* Returns TCP state constant, or -1 for invalid/UDP sockets.
|
||||
*/
|
||||
static int32_t sys_sockstate(registers_t *regs) {
|
||||
int ufd = (int)regs->ebx;
|
||||
|
||||
if (ufd < 0 || ufd >= MAX_USER_SOCKETS || !user_sockets[ufd].active)
|
||||
return -1;
|
||||
|
||||
if (user_sockets[ufd].type == SOCK_TYPE_TCP)
|
||||
return (int32_t)tcp_get_state(user_sockets[ufd].kern_sockfd);
|
||||
else
|
||||
return -1;
|
||||
}
|
||||
|
||||
/** System call dispatch table. */
|
||||
typedef int32_t (*syscall_fn)(registers_t *);
|
||||
static syscall_fn syscall_table[NUM_SYSCALLS] = {
|
||||
[SYS_EXIT] = sys_exit,
|
||||
[SYS_WRITE] = sys_write,
|
||||
[SYS_READ] = sys_read,
|
||||
[SYS_FORK] = sys_fork,
|
||||
[SYS_GETPID] = sys_getpid,
|
||||
[SYS_YIELD] = sys_yield,
|
||||
[SYS_WAITPID] = sys_waitpid,
|
||||
[SYS_EXEC] = sys_exec,
|
||||
[SYS_GETENV] = sys_getenv,
|
||||
[SYS_SETENV] = sys_setenv,
|
||||
[SYS_READDIR] = sys_readdir,
|
||||
[SYS_OPEN] = sys_open,
|
||||
[SYS_CLOSE] = sys_close,
|
||||
[SYS_EXIT] = sys_exit,
|
||||
[SYS_WRITE] = sys_write,
|
||||
[SYS_READ] = sys_read,
|
||||
[SYS_FORK] = sys_fork,
|
||||
[SYS_GETPID] = sys_getpid,
|
||||
[SYS_YIELD] = sys_yield,
|
||||
[SYS_WAITPID] = sys_waitpid,
|
||||
[SYS_EXEC] = sys_exec,
|
||||
[SYS_GETENV] = sys_getenv,
|
||||
[SYS_SETENV] = sys_setenv,
|
||||
[SYS_READDIR] = sys_readdir,
|
||||
[SYS_OPEN] = sys_open,
|
||||
[SYS_CLOSE] = sys_close,
|
||||
[SYS_SOCKET] = sys_socket,
|
||||
[SYS_CONNECT] = sys_connect,
|
||||
[SYS_SEND] = sys_send,
|
||||
[SYS_RECV] = sys_recv,
|
||||
[SYS_SOCKSTATE] = sys_sockstate,
|
||||
};
|
||||
|
||||
void syscall_handler(registers_t *regs) {
|
||||
|
||||
@@ -27,9 +27,14 @@
|
||||
#define SYS_READDIR 10 /**< Read directory entry. path=EBX, idx=ECX, buf=EDX. Returns type or -1. */
|
||||
#define SYS_OPEN 11 /**< Open a file. path=EBX, flags=ECX. Returns fd or -1. */
|
||||
#define SYS_CLOSE 12 /**< Close a file descriptor. fd=EBX. Returns 0 or -1. */
|
||||
#define SYS_SOCKET 13 /**< Create a network socket. type=EBX (0=TCP, 1=UDP). Returns sockfd. */
|
||||
#define SYS_CONNECT 14 /**< Connect TCP socket. sockfd=EBX, ip=ECX (host order), port=EDX. */
|
||||
#define SYS_SEND 15 /**< Send data on socket. sockfd=EBX, buf=ECX, len=EDX. Returns bytes sent. */
|
||||
#define SYS_RECV 16 /**< Receive data from socket. sockfd=EBX, buf=ECX, len=EDX. Returns bytes. */
|
||||
#define SYS_SOCKSTATE 17 /**< Get socket state. sockfd=EBX. Returns state constant. */
|
||||
|
||||
/** Total number of system calls. */
|
||||
#define NUM_SYSCALLS 13
|
||||
#define NUM_SYSCALLS 18
|
||||
|
||||
/**
|
||||
* Initialize the system call handler.
|
||||
|
||||
512
src/tcp.c
Normal file
512
src/tcp.c
Normal file
@@ -0,0 +1,512 @@
|
||||
/**
|
||||
* @file tcp.c
|
||||
* @brief Transmission Control Protocol (TCP) implementation.
|
||||
*
|
||||
* Minimal TCP state machine supporting:
|
||||
* - Active open (connect)
|
||||
* - Data transfer (send/recv)
|
||||
* - Connection teardown (close with FIN)
|
||||
*
|
||||
* Limitations:
|
||||
* - No retransmission timer (single-attempt)
|
||||
* - No congestion control
|
||||
* - No passive open (listen/accept) yet
|
||||
* - No out-of-order segment handling
|
||||
* - Window size is static
|
||||
*/
|
||||
|
||||
#include "tcp.h"
|
||||
#include "ipv4.h"
|
||||
#include "ethernet.h"
|
||||
#include "kmalloc.h"
|
||||
#include "string.h"
|
||||
#include "vga.h"
|
||||
|
||||
/* ================================================================
|
||||
* Internal state
|
||||
* ================================================================ */
|
||||
|
||||
static tcp_socket_t tcp_sockets[TCP_MAX_SOCKETS];
|
||||
static uint16_t tcp_next_port = 49152; /* Ephemeral port start. */
|
||||
|
||||
/* Simple pseudo-random ISN based on a counter. */
|
||||
static uint32_t tcp_isn_counter = 0x12345678;
|
||||
|
||||
static uint32_t tcp_generate_isn(void) {
|
||||
tcp_isn_counter = tcp_isn_counter * 1103515245 + 12345;
|
||||
return tcp_isn_counter;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* TCP pseudo-header checksum
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Compute TCP checksum including the pseudo-header.
|
||||
*/
|
||||
static uint16_t tcp_checksum(uint32_t src_ip, uint32_t dst_ip,
|
||||
const void *tcp_seg, uint32_t tcp_len)
|
||||
{
|
||||
uint32_t sum = 0;
|
||||
|
||||
/* Pseudo-header */
|
||||
sum += (src_ip >> 16) & 0xFFFF;
|
||||
sum += src_ip & 0xFFFF;
|
||||
sum += (dst_ip >> 16) & 0xFFFF;
|
||||
sum += dst_ip & 0xFFFF;
|
||||
sum += htons(6); /* IP_PROTO_TCP */
|
||||
sum += htons(tcp_len);
|
||||
|
||||
/* TCP segment */
|
||||
const uint16_t *p = (const uint16_t *)tcp_seg;
|
||||
uint32_t rem = tcp_len;
|
||||
while (rem > 1) {
|
||||
sum += *p++;
|
||||
rem -= 2;
|
||||
}
|
||||
if (rem == 1) {
|
||||
sum += *((const uint8_t *)p);
|
||||
}
|
||||
|
||||
/* Fold carries */
|
||||
while (sum >> 16)
|
||||
sum = (sum & 0xFFFF) + (sum >> 16);
|
||||
|
||||
return (uint16_t)(~sum);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Send a TCP segment
|
||||
* ================================================================ */
|
||||
|
||||
static int tcp_send_segment(tcp_socket_t *sock, uint8_t flags,
|
||||
const void *data, uint32_t data_len)
|
||||
{
|
||||
uint32_t total_len = TCP_HLEN + data_len;
|
||||
uint8_t buf[TCP_HLEN + TCP_MSS];
|
||||
|
||||
if (total_len > sizeof(buf)) return -1;
|
||||
|
||||
/* Build TCP header */
|
||||
tcp_header_t *hdr = (tcp_header_t *)buf;
|
||||
hdr->src_port = htons(sock->local_port);
|
||||
hdr->dst_port = htons(sock->remote_port);
|
||||
hdr->seq_num = htonl(sock->snd_nxt);
|
||||
hdr->ack_num = (flags & TCP_ACK) ? htonl(sock->rcv_nxt) : 0;
|
||||
hdr->data_offset = (TCP_HLEN / 4) << 4; /* 5 words, no options */
|
||||
hdr->flags = flags;
|
||||
hdr->window = htons(TCP_RX_BUF_SIZE - sock->rx_count);
|
||||
hdr->checksum = 0;
|
||||
hdr->urgent = 0;
|
||||
|
||||
/* Copy payload */
|
||||
if (data && data_len > 0) {
|
||||
memcpy(buf + TCP_HLEN, data, data_len);
|
||||
}
|
||||
|
||||
/* Compute local IP from first ethernet interface. */
|
||||
eth_iface_t *iface = ethernet_get_iface(0);
|
||||
uint32_t src_ip = iface ? iface->ip_addr : 0;
|
||||
|
||||
/* Compute checksum over the whole segment. */
|
||||
hdr->checksum = tcp_checksum(htonl(src_ip), htonl(sock->remote_ip),
|
||||
buf, total_len);
|
||||
|
||||
/* Advance SND.NXT for SYN, FIN, and data. */
|
||||
if (flags & TCP_SYN)
|
||||
sock->snd_nxt++;
|
||||
if (flags & TCP_FIN)
|
||||
sock->snd_nxt++;
|
||||
sock->snd_nxt += data_len;
|
||||
|
||||
/* Send via IPv4. */
|
||||
return ipv4_send(sock->remote_ip, IP_PROTO_TCP, buf, total_len);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Send a RST in response to an unexpected segment
|
||||
* ================================================================ */
|
||||
|
||||
static void tcp_send_rst(uint32_t src_ip, uint32_t dst_ip,
|
||||
const tcp_header_t *in_hdr, uint32_t seg_len)
|
||||
{
|
||||
uint8_t buf[TCP_HLEN];
|
||||
tcp_header_t *hdr = (tcp_header_t *)buf;
|
||||
|
||||
memset(buf, 0, TCP_HLEN);
|
||||
hdr->src_port = in_hdr->dst_port; /* Already in network byte order. */
|
||||
hdr->dst_port = in_hdr->src_port;
|
||||
hdr->data_offset = (TCP_HLEN / 4) << 4;
|
||||
|
||||
if (in_hdr->flags & TCP_ACK) {
|
||||
hdr->seq_num = in_hdr->ack_num;
|
||||
hdr->flags = TCP_RST;
|
||||
} else {
|
||||
hdr->seq_num = 0;
|
||||
hdr->ack_num = htonl(ntohl(in_hdr->seq_num) + seg_len);
|
||||
hdr->flags = TCP_RST | TCP_ACK;
|
||||
}
|
||||
|
||||
hdr->window = 0;
|
||||
hdr->checksum = tcp_checksum(htonl(dst_ip), htonl(src_ip),
|
||||
buf, TCP_HLEN);
|
||||
|
||||
ipv4_send(src_ip, IP_PROTO_TCP, buf, TCP_HLEN);
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Find a socket matching an incoming segment
|
||||
* ================================================================ */
|
||||
|
||||
static tcp_socket_t *tcp_find_socket(uint32_t remote_ip,
|
||||
uint16_t remote_port,
|
||||
uint16_t local_port)
|
||||
{
|
||||
for (int i = 0; i < TCP_MAX_SOCKETS; i++) {
|
||||
tcp_socket_t *s = &tcp_sockets[i];
|
||||
if (!s->active) continue;
|
||||
if (s->local_port != local_port) continue;
|
||||
if (s->state == TCP_STATE_LISTEN) return s;
|
||||
if (s->remote_port == remote_port && s->remote_ip == remote_ip)
|
||||
return s;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Ring buffer helpers
|
||||
* ================================================================ */
|
||||
|
||||
static void rx_buf_write(tcp_socket_t *sock, const uint8_t *data, uint32_t len) {
|
||||
for (uint32_t i = 0; i < len && sock->rx_count < TCP_RX_BUF_SIZE; i++) {
|
||||
sock->rx_buf[sock->rx_head] = data[i];
|
||||
sock->rx_head = (sock->rx_head + 1) % TCP_RX_BUF_SIZE;
|
||||
sock->rx_count++;
|
||||
}
|
||||
}
|
||||
|
||||
static uint32_t rx_buf_read(tcp_socket_t *sock, uint8_t *buf, uint32_t len) {
|
||||
uint32_t n = 0;
|
||||
while (n < len && sock->rx_count > 0) {
|
||||
buf[n++] = sock->rx_buf[sock->rx_tail];
|
||||
sock->rx_tail = (sock->rx_tail + 1) % TCP_RX_BUF_SIZE;
|
||||
sock->rx_count--;
|
||||
}
|
||||
return n;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* TCP input processing
|
||||
* ================================================================ */
|
||||
|
||||
void tcp_receive(uint32_t src_ip_ho, uint32_t dst_ip_ho,
|
||||
const void *data, uint32_t len, uint32_t iface_idx)
|
||||
{
|
||||
(void)iface_idx;
|
||||
|
||||
if (len < TCP_HLEN) return;
|
||||
|
||||
const tcp_header_t *hdr = (const tcp_header_t *)data;
|
||||
uint32_t hdr_len = ((hdr->data_offset >> 4) & 0x0F) * 4;
|
||||
if (hdr_len < TCP_HLEN || hdr_len > len) return;
|
||||
|
||||
uint16_t src_port = ntohs(hdr->src_port);
|
||||
uint16_t dst_port = ntohs(hdr->dst_port);
|
||||
uint32_t seq = ntohl(hdr->seq_num);
|
||||
uint32_t ack = ntohl(hdr->ack_num);
|
||||
uint8_t flags = hdr->flags;
|
||||
const uint8_t *payload = (const uint8_t *)data + hdr_len;
|
||||
uint32_t payload_len = len - hdr_len;
|
||||
|
||||
/* Compute segment length (SYN and FIN count as 1 byte each). */
|
||||
uint32_t seg_len = payload_len;
|
||||
if (flags & TCP_SYN) seg_len++;
|
||||
if (flags & TCP_FIN) seg_len++;
|
||||
|
||||
/* Find matching socket. */
|
||||
tcp_socket_t *sock = tcp_find_socket(src_ip_ho, src_port, dst_port);
|
||||
|
||||
if (!sock) {
|
||||
/* No socket found — send RST if not already RST. */
|
||||
if (!(flags & TCP_RST)) {
|
||||
tcp_send_rst(src_ip_ho, dst_ip_ho, hdr, seg_len);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
/* Handle RST. */
|
||||
if (flags & TCP_RST) {
|
||||
sock->state = TCP_STATE_CLOSED;
|
||||
sock->active = 0;
|
||||
return;
|
||||
}
|
||||
|
||||
switch (sock->state) {
|
||||
case TCP_STATE_SYN_SENT:
|
||||
/* Expecting SYN+ACK */
|
||||
if ((flags & (TCP_SYN | TCP_ACK)) == (TCP_SYN | TCP_ACK)) {
|
||||
if (ack == sock->snd_nxt) {
|
||||
sock->rcv_nxt = seq + 1;
|
||||
sock->snd_una = ack;
|
||||
sock->state = TCP_STATE_ESTABLISHED;
|
||||
/* Send ACK */
|
||||
tcp_send_segment(sock, TCP_ACK, NULL, 0);
|
||||
}
|
||||
} else if (flags & TCP_SYN) {
|
||||
/* Simultaneous open — simplified handling. */
|
||||
sock->rcv_nxt = seq + 1;
|
||||
sock->state = TCP_STATE_SYN_RECEIVED;
|
||||
tcp_send_segment(sock, TCP_SYN | TCP_ACK, NULL, 0);
|
||||
}
|
||||
break;
|
||||
|
||||
case TCP_STATE_SYN_RECEIVED:
|
||||
if (flags & TCP_ACK) {
|
||||
if (ack == sock->snd_nxt) {
|
||||
sock->snd_una = ack;
|
||||
sock->state = TCP_STATE_ESTABLISHED;
|
||||
}
|
||||
}
|
||||
break;
|
||||
|
||||
case TCP_STATE_ESTABLISHED:
|
||||
/* Check sequence number. */
|
||||
if (seq != sock->rcv_nxt) {
|
||||
/* Out-of-order — send duplicate ACK. */
|
||||
tcp_send_segment(sock, TCP_ACK, NULL, 0);
|
||||
break;
|
||||
}
|
||||
|
||||
/* Update SND.UNA if ACK present. */
|
||||
if (flags & TCP_ACK) {
|
||||
sock->snd_una = ack;
|
||||
}
|
||||
|
||||
/* Process payload. */
|
||||
if (payload_len > 0) {
|
||||
rx_buf_write(sock, payload, payload_len);
|
||||
sock->rcv_nxt += payload_len;
|
||||
}
|
||||
|
||||
/* Handle FIN. */
|
||||
if (flags & TCP_FIN) {
|
||||
sock->rcv_nxt++;
|
||||
sock->state = TCP_STATE_CLOSE_WAIT;
|
||||
/* ACK the FIN. */
|
||||
tcp_send_segment(sock, TCP_ACK, NULL, 0);
|
||||
} else if (payload_len > 0) {
|
||||
/* ACK the data. */
|
||||
tcp_send_segment(sock, TCP_ACK, NULL, 0);
|
||||
}
|
||||
break;
|
||||
|
||||
case TCP_STATE_FIN_WAIT_1:
|
||||
if (flags & TCP_ACK) {
|
||||
sock->snd_una = ack;
|
||||
}
|
||||
if ((flags & TCP_FIN) && (flags & TCP_ACK) && ack == sock->snd_nxt) {
|
||||
/* FIN+ACK: simultaneous close shortcut. */
|
||||
sock->rcv_nxt++;
|
||||
sock->state = TCP_STATE_TIME_WAIT;
|
||||
tcp_send_segment(sock, TCP_ACK, NULL, 0);
|
||||
} else if (flags & TCP_FIN) {
|
||||
sock->rcv_nxt++;
|
||||
sock->state = TCP_STATE_CLOSING;
|
||||
tcp_send_segment(sock, TCP_ACK, NULL, 0);
|
||||
} else if ((flags & TCP_ACK) && ack == sock->snd_nxt) {
|
||||
/* Our FIN was ACKed. */
|
||||
sock->state = TCP_STATE_FIN_WAIT_2;
|
||||
}
|
||||
/* Accept any data in FIN_WAIT_1 */
|
||||
if (payload_len > 0 && seq == sock->rcv_nxt) {
|
||||
rx_buf_write(sock, payload, payload_len);
|
||||
sock->rcv_nxt += payload_len;
|
||||
}
|
||||
break;
|
||||
|
||||
case TCP_STATE_FIN_WAIT_2:
|
||||
if (payload_len > 0 && seq == sock->rcv_nxt) {
|
||||
rx_buf_write(sock, payload, payload_len);
|
||||
sock->rcv_nxt += payload_len;
|
||||
tcp_send_segment(sock, TCP_ACK, NULL, 0);
|
||||
}
|
||||
if (flags & TCP_FIN) {
|
||||
sock->rcv_nxt++;
|
||||
sock->state = TCP_STATE_TIME_WAIT;
|
||||
tcp_send_segment(sock, TCP_ACK, NULL, 0);
|
||||
}
|
||||
break;
|
||||
|
||||
case TCP_STATE_CLOSE_WAIT:
|
||||
/* Waiting for application to close. Accept ACKs. */
|
||||
if (flags & TCP_ACK) {
|
||||
sock->snd_una = ack;
|
||||
}
|
||||
break;
|
||||
|
||||
case TCP_STATE_CLOSING:
|
||||
if ((flags & TCP_ACK) && ack == sock->snd_nxt) {
|
||||
sock->state = TCP_STATE_TIME_WAIT;
|
||||
}
|
||||
break;
|
||||
|
||||
case TCP_STATE_LAST_ACK:
|
||||
if ((flags & TCP_ACK) && ack == sock->snd_nxt) {
|
||||
sock->state = TCP_STATE_CLOSED;
|
||||
sock->active = 0;
|
||||
}
|
||||
break;
|
||||
|
||||
case TCP_STATE_TIME_WAIT:
|
||||
/* In a real OS we'd wait 2*MSL. Here we just ACK and stay. */
|
||||
if (flags & TCP_FIN) {
|
||||
tcp_send_segment(sock, TCP_ACK, NULL, 0);
|
||||
}
|
||||
/* Immediately transition to CLOSED (no timer). */
|
||||
sock->state = TCP_STATE_CLOSED;
|
||||
sock->active = 0;
|
||||
break;
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
void tcp_init(void)
|
||||
{
|
||||
memset(tcp_sockets, 0, sizeof(tcp_sockets));
|
||||
ipv4_register_proto(IP_PROTO_TCP, tcp_receive);
|
||||
vga_puts("[TCP] Initialized (");
|
||||
vga_put_dec(TCP_MAX_SOCKETS);
|
||||
vga_puts(" sockets)\n");
|
||||
}
|
||||
|
||||
int tcp_socket_create(void)
|
||||
{
|
||||
for (int i = 0; i < TCP_MAX_SOCKETS; i++) {
|
||||
if (!tcp_sockets[i].active) {
|
||||
memset(&tcp_sockets[i], 0, sizeof(tcp_socket_t));
|
||||
tcp_sockets[i].active = 1;
|
||||
tcp_sockets[i].state = TCP_STATE_CLOSED;
|
||||
return i;
|
||||
}
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
int tcp_connect(int sockfd, uint32_t remote_ip, uint16_t remote_port)
|
||||
{
|
||||
if (sockfd < 0 || sockfd >= TCP_MAX_SOCKETS) return -1;
|
||||
tcp_socket_t *sock = &tcp_sockets[sockfd];
|
||||
if (!sock->active || sock->state != TCP_STATE_CLOSED) return -1;
|
||||
|
||||
/* Assign ephemeral local port. */
|
||||
sock->local_port = tcp_next_port++;
|
||||
if (tcp_next_port == 0) tcp_next_port = 49152;
|
||||
sock->remote_ip = remote_ip;
|
||||
sock->remote_port = remote_port;
|
||||
|
||||
/* Generate ISN. */
|
||||
sock->snd_nxt = tcp_generate_isn();
|
||||
sock->snd_una = sock->snd_nxt;
|
||||
|
||||
/* Send SYN. */
|
||||
sock->state = TCP_STATE_SYN_SENT;
|
||||
return tcp_send_segment(sock, TCP_SYN, NULL, 0);
|
||||
}
|
||||
|
||||
int tcp_send(int sockfd, const void *data, uint32_t len)
|
||||
{
|
||||
if (sockfd < 0 || sockfd >= TCP_MAX_SOCKETS) return -1;
|
||||
tcp_socket_t *sock = &tcp_sockets[sockfd];
|
||||
if (!sock->active || sock->state != TCP_STATE_ESTABLISHED) return -1;
|
||||
|
||||
/* Send in MSS-sized chunks. */
|
||||
uint32_t sent = 0;
|
||||
const uint8_t *p = (const uint8_t *)data;
|
||||
while (sent < len) {
|
||||
uint32_t chunk = len - sent;
|
||||
if (chunk > TCP_MSS) chunk = TCP_MSS;
|
||||
if (tcp_send_segment(sock, TCP_ACK | TCP_PSH, p + sent, chunk) < 0)
|
||||
break;
|
||||
sent += chunk;
|
||||
}
|
||||
return (int)sent;
|
||||
}
|
||||
|
||||
int tcp_recv(int sockfd, void *buf, uint32_t bufsize)
|
||||
{
|
||||
if (sockfd < 0 || sockfd >= TCP_MAX_SOCKETS) return -1;
|
||||
tcp_socket_t *sock = &tcp_sockets[sockfd];
|
||||
if (!sock->active) return -1;
|
||||
|
||||
/* If connection is closed with data still in buffer, return it. */
|
||||
if (sock->rx_count == 0) {
|
||||
if (sock->state == TCP_STATE_CLOSE_WAIT ||
|
||||
sock->state == TCP_STATE_CLOSED) {
|
||||
return -1; /* EOF / connection closed. */
|
||||
}
|
||||
return 0; /* No data available yet. */
|
||||
}
|
||||
|
||||
return (int)rx_buf_read(sock, (uint8_t *)buf, bufsize);
|
||||
}
|
||||
|
||||
void tcp_close(int sockfd)
|
||||
{
|
||||
if (sockfd < 0 || sockfd >= TCP_MAX_SOCKETS) return;
|
||||
tcp_socket_t *sock = &tcp_sockets[sockfd];
|
||||
if (!sock->active) return;
|
||||
|
||||
switch (sock->state) {
|
||||
case TCP_STATE_ESTABLISHED:
|
||||
sock->state = TCP_STATE_FIN_WAIT_1;
|
||||
tcp_send_segment(sock, TCP_FIN | TCP_ACK, NULL, 0);
|
||||
break;
|
||||
|
||||
case TCP_STATE_CLOSE_WAIT:
|
||||
sock->state = TCP_STATE_LAST_ACK;
|
||||
tcp_send_segment(sock, TCP_FIN | TCP_ACK, NULL, 0);
|
||||
break;
|
||||
|
||||
case TCP_STATE_SYN_SENT:
|
||||
case TCP_STATE_SYN_RECEIVED:
|
||||
sock->state = TCP_STATE_CLOSED;
|
||||
sock->active = 0;
|
||||
break;
|
||||
|
||||
default:
|
||||
/* Already closing or closed — force close. */
|
||||
sock->state = TCP_STATE_CLOSED;
|
||||
sock->active = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
uint8_t tcp_get_state(int sockfd)
|
||||
{
|
||||
if (sockfd < 0 || sockfd >= TCP_MAX_SOCKETS) return TCP_STATE_CLOSED;
|
||||
return tcp_sockets[sockfd].state;
|
||||
}
|
||||
|
||||
const char *tcp_state_name(uint8_t state)
|
||||
{
|
||||
switch (state) {
|
||||
case TCP_STATE_CLOSED: return "CLOSED";
|
||||
case TCP_STATE_LISTEN: return "LISTEN";
|
||||
case TCP_STATE_SYN_SENT: return "SYN_SENT";
|
||||
case TCP_STATE_SYN_RECEIVED: return "SYN_RECEIVED";
|
||||
case TCP_STATE_ESTABLISHED: return "ESTABLISHED";
|
||||
case TCP_STATE_FIN_WAIT_1: return "FIN_WAIT_1";
|
||||
case TCP_STATE_FIN_WAIT_2: return "FIN_WAIT_2";
|
||||
case TCP_STATE_CLOSE_WAIT: return "CLOSE_WAIT";
|
||||
case TCP_STATE_CLOSING: return "CLOSING";
|
||||
case TCP_STATE_LAST_ACK: return "LAST_ACK";
|
||||
case TCP_STATE_TIME_WAIT: return "TIME_WAIT";
|
||||
default: return "UNKNOWN";
|
||||
}
|
||||
}
|
||||
169
src/tcp.h
Normal file
169
src/tcp.h
Normal file
@@ -0,0 +1,169 @@
|
||||
/**
|
||||
* @file tcp.h
|
||||
* @brief Transmission Control Protocol (TCP) subsystem.
|
||||
*
|
||||
* Provides a minimal TCP implementation supporting active open
|
||||
* (client connections). Implements the core TCP state machine
|
||||
* for connection setup (SYN/SYN-ACK/ACK), data transfer, and
|
||||
* connection teardown (FIN/ACK).
|
||||
*
|
||||
* Built on top of the IPv4 stack.
|
||||
*/
|
||||
|
||||
#ifndef TCP_H
|
||||
#define TCP_H
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
/** Maximum number of TCP sockets. */
|
||||
#define TCP_MAX_SOCKETS 8
|
||||
|
||||
/** Maximum TCP segment payload. */
|
||||
#define TCP_MSS 1460 /* ETH_MTU(1500) - IP(20) - TCP(20) */
|
||||
|
||||
/** TCP header size (no options). */
|
||||
#define TCP_HLEN 20
|
||||
|
||||
/** TCP receive buffer size per socket. */
|
||||
#define TCP_RX_BUF_SIZE 4096
|
||||
|
||||
/** TCP send buffer size per socket. */
|
||||
#define TCP_TX_BUF_SIZE 4096
|
||||
|
||||
/* ================================================================
|
||||
* TCP flags
|
||||
* ================================================================ */
|
||||
#define TCP_FIN 0x01
|
||||
#define TCP_SYN 0x02
|
||||
#define TCP_RST 0x04
|
||||
#define TCP_PSH 0x08
|
||||
#define TCP_ACK 0x10
|
||||
#define TCP_URG 0x20
|
||||
|
||||
/* ================================================================
|
||||
* TCP states (RFC 793)
|
||||
* ================================================================ */
|
||||
#define TCP_STATE_CLOSED 0
|
||||
#define TCP_STATE_LISTEN 1
|
||||
#define TCP_STATE_SYN_SENT 2
|
||||
#define TCP_STATE_SYN_RECEIVED 3
|
||||
#define TCP_STATE_ESTABLISHED 4
|
||||
#define TCP_STATE_FIN_WAIT_1 5
|
||||
#define TCP_STATE_FIN_WAIT_2 6
|
||||
#define TCP_STATE_CLOSE_WAIT 7
|
||||
#define TCP_STATE_CLOSING 8
|
||||
#define TCP_STATE_LAST_ACK 9
|
||||
#define TCP_STATE_TIME_WAIT 10
|
||||
|
||||
/* ================================================================
|
||||
* TCP header
|
||||
* ================================================================ */
|
||||
|
||||
typedef struct __attribute__((packed)) tcp_header {
|
||||
uint16_t src_port; /**< Source port. */
|
||||
uint16_t dst_port; /**< Destination port. */
|
||||
uint32_t seq_num; /**< Sequence number. */
|
||||
uint32_t ack_num; /**< Acknowledgment number. */
|
||||
uint8_t data_offset; /**< Data offset (upper 4 bits) + reserved. */
|
||||
uint8_t flags; /**< TCP flags. */
|
||||
uint16_t window; /**< Window size. */
|
||||
uint16_t checksum; /**< Checksum. */
|
||||
uint16_t urgent; /**< Urgent pointer. */
|
||||
} tcp_header_t;
|
||||
|
||||
/* ================================================================
|
||||
* TCP socket
|
||||
* ================================================================ */
|
||||
|
||||
typedef struct tcp_socket {
|
||||
uint8_t active; /**< 1 if in use. */
|
||||
uint8_t state; /**< TCP_STATE_*. */
|
||||
uint16_t local_port; /**< Local port (host byte order). */
|
||||
uint16_t remote_port; /**< Remote port (host byte order). */
|
||||
uint32_t remote_ip; /**< Remote IP (host byte order). */
|
||||
|
||||
/* Sequence numbers */
|
||||
uint32_t snd_una; /**< Send unacknowledged. */
|
||||
uint32_t snd_nxt; /**< Send next. */
|
||||
uint32_t rcv_nxt; /**< Receive next expected. */
|
||||
|
||||
/* Receive buffer (ring buffer) */
|
||||
uint8_t rx_buf[TCP_RX_BUF_SIZE];
|
||||
uint32_t rx_head; /**< Write position. */
|
||||
uint32_t rx_tail; /**< Read position. */
|
||||
uint32_t rx_count; /**< Bytes available. */
|
||||
} tcp_socket_t;
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Initialize the TCP subsystem.
|
||||
*/
|
||||
void tcp_init(void);
|
||||
|
||||
/**
|
||||
* Create a TCP socket.
|
||||
* @return Socket index (>= 0), or -1 on failure.
|
||||
*/
|
||||
int tcp_socket_create(void);
|
||||
|
||||
/**
|
||||
* Connect to a remote host (active open).
|
||||
* Sends SYN and transitions to SYN_SENT state.
|
||||
*
|
||||
* @param sockfd Socket index.
|
||||
* @param remote_ip Remote IP (host byte order).
|
||||
* @param remote_port Remote port (host byte order).
|
||||
* @return 0 on success (SYN sent), -1 on failure.
|
||||
*/
|
||||
int tcp_connect(int sockfd, uint32_t remote_ip, uint16_t remote_port);
|
||||
|
||||
/**
|
||||
* Send data on an established connection.
|
||||
*
|
||||
* @param sockfd Socket index.
|
||||
* @param data Data to send.
|
||||
* @param len Data length.
|
||||
* @return Number of bytes sent, or -1 on failure.
|
||||
*/
|
||||
int tcp_send(int sockfd, const void *data, uint32_t len);
|
||||
|
||||
/**
|
||||
* Receive data from an established connection (non-blocking).
|
||||
*
|
||||
* @param sockfd Socket index.
|
||||
* @param buf Buffer.
|
||||
* @param bufsize Buffer size.
|
||||
* @return Number of bytes received, 0 if no data, -1 on error/closed.
|
||||
*/
|
||||
int tcp_recv(int sockfd, void *buf, uint32_t bufsize);
|
||||
|
||||
/**
|
||||
* Close a TCP connection.
|
||||
*
|
||||
* @param sockfd Socket index.
|
||||
*/
|
||||
void tcp_close(int sockfd);
|
||||
|
||||
/**
|
||||
* Get the state of a TCP socket.
|
||||
*
|
||||
* @param sockfd Socket index.
|
||||
* @return TCP state constant, or TCP_STATE_CLOSED for invalid sockets.
|
||||
*/
|
||||
uint8_t tcp_get_state(int sockfd);
|
||||
|
||||
/**
|
||||
* Get the state name as a string.
|
||||
*/
|
||||
const char *tcp_state_name(uint8_t state);
|
||||
|
||||
/**
|
||||
* Process an incoming TCP segment (called from IPv4 layer).
|
||||
*/
|
||||
void tcp_receive(uint32_t src_ip, uint32_t dst_ip,
|
||||
const void *data, uint32_t len, uint32_t iface_idx);
|
||||
|
||||
#endif /* TCP_H */
|
||||
180
src/udp.c
Normal file
180
src/udp.c
Normal file
@@ -0,0 +1,180 @@
|
||||
/**
|
||||
* @file udp.c
|
||||
* @brief UDP subsystem implementation.
|
||||
*
|
||||
* Provides a simple socket-like interface for sending and receiving
|
||||
* UDP datagrams. Registers with the IPv4 stack as protocol handler 17.
|
||||
*/
|
||||
|
||||
#include "udp.h"
|
||||
#include "ipv4.h"
|
||||
#include "ethernet.h"
|
||||
#include <string.h>
|
||||
|
||||
/* Debug print helpers */
|
||||
extern void offset_print(const char *str);
|
||||
extern void print_hex(uint32_t val);
|
||||
|
||||
/* ================================================================
|
||||
* Global state
|
||||
* ================================================================ */
|
||||
|
||||
/** UDP socket table. */
|
||||
static udp_socket_t sockets[UDP_MAX_SOCKETS];
|
||||
|
||||
/* ================================================================
|
||||
* Internal helpers
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Find a socket bound to the given local port.
|
||||
*/
|
||||
static udp_socket_t *find_socket_by_port(uint16_t port) {
|
||||
for (int i = 0; i < UDP_MAX_SOCKETS; i++) {
|
||||
if (sockets[i].active && sockets[i].bound &&
|
||||
sockets[i].local_port == port) {
|
||||
return &sockets[i];
|
||||
}
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
int udp_socket_create(void) {
|
||||
for (int i = 0; i < UDP_MAX_SOCKETS; i++) {
|
||||
if (!sockets[i].active) {
|
||||
memset(&sockets[i], 0, sizeof(udp_socket_t));
|
||||
sockets[i].active = 1;
|
||||
return i;
|
||||
}
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
int udp_bind(int sockfd, uint16_t port) {
|
||||
if (sockfd < 0 || sockfd >= UDP_MAX_SOCKETS) return -1;
|
||||
if (!sockets[sockfd].active) return -1;
|
||||
|
||||
/* Check port not already used */
|
||||
if (find_socket_by_port(port)) return -1;
|
||||
|
||||
sockets[sockfd].local_port = port;
|
||||
sockets[sockfd].bound = 1;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int udp_sendto(int sockfd, uint32_t dst_ip, uint16_t dst_port,
|
||||
const void *data, uint32_t len) {
|
||||
if (sockfd < 0 || sockfd >= UDP_MAX_SOCKETS) return -1;
|
||||
if (!sockets[sockfd].active) return -1;
|
||||
if (len > UDP_MAX_PAYLOAD) return -1;
|
||||
|
||||
/* Build UDP datagram */
|
||||
uint8_t pkt[UDP_HLEN + UDP_MAX_PAYLOAD];
|
||||
udp_header_t *hdr = (udp_header_t *)pkt;
|
||||
|
||||
hdr->src_port = htons(sockets[sockfd].local_port);
|
||||
hdr->dst_port = htons(dst_port);
|
||||
hdr->length = htons((uint16_t)(UDP_HLEN + len));
|
||||
hdr->checksum = 0; /* Optional in IPv4 */
|
||||
|
||||
memcpy(pkt + UDP_HLEN, data, len);
|
||||
|
||||
/* Send via IPv4 */
|
||||
int ret = ipv4_send(dst_ip, IP_PROTO_UDP, pkt, UDP_HLEN + len);
|
||||
return (ret == 0) ? (int)len : -1;
|
||||
}
|
||||
|
||||
int udp_recvfrom(int sockfd, void *buf, uint32_t bufsize,
|
||||
uint32_t *src_ip, uint16_t *src_port) {
|
||||
if (sockfd < 0 || sockfd >= UDP_MAX_SOCKETS) return -1;
|
||||
if (!sockets[sockfd].active) return -1;
|
||||
|
||||
udp_socket_t *sock = &sockets[sockfd];
|
||||
|
||||
/* Scan the receive queue for the oldest entry */
|
||||
for (int i = 0; i < UDP_RX_QUEUE_SIZE; i++) {
|
||||
if (sock->rx_queue[i].used) {
|
||||
udp_rx_entry_t *entry = &sock->rx_queue[i];
|
||||
|
||||
uint32_t copy_len = entry->len;
|
||||
if (copy_len > bufsize) copy_len = bufsize;
|
||||
|
||||
memcpy(buf, entry->data, copy_len);
|
||||
if (src_ip) *src_ip = entry->src_ip;
|
||||
if (src_port) *src_port = entry->src_port;
|
||||
|
||||
entry->used = 0; /* Free the slot */
|
||||
return (int)copy_len;
|
||||
}
|
||||
}
|
||||
|
||||
return 0; /* No data available */
|
||||
}
|
||||
|
||||
void udp_close(int sockfd) {
|
||||
if (sockfd < 0 || sockfd >= UDP_MAX_SOCKETS) return;
|
||||
memset(&sockets[sockfd], 0, sizeof(udp_socket_t));
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* IPv4 protocol handler
|
||||
* ================================================================ */
|
||||
|
||||
void udp_receive(uint32_t src_ip, uint32_t dst_ip,
|
||||
const void *data, uint32_t len, uint32_t iface_idx) {
|
||||
(void)dst_ip;
|
||||
(void)iface_idx;
|
||||
|
||||
if (len < UDP_HLEN) return;
|
||||
|
||||
const udp_header_t *hdr = (const udp_header_t *)data;
|
||||
uint16_t dst_port = ntohs(hdr->dst_port);
|
||||
uint16_t src_port_val = ntohs(hdr->src_port);
|
||||
uint16_t udp_len = ntohs(hdr->length);
|
||||
|
||||
if (udp_len > len) return;
|
||||
uint32_t payload_len = udp_len - UDP_HLEN;
|
||||
|
||||
/* Find socket bound to this port */
|
||||
udp_socket_t *sock = find_socket_by_port(dst_port);
|
||||
if (!sock) return;
|
||||
|
||||
/* Enqueue the datagram */
|
||||
for (int i = 0; i < UDP_RX_QUEUE_SIZE; i++) {
|
||||
if (!sock->rx_queue[i].used) {
|
||||
udp_rx_entry_t *entry = &sock->rx_queue[i];
|
||||
entry->src_ip = src_ip;
|
||||
entry->src_port = src_port_val;
|
||||
entry->len = (uint16_t)(payload_len > UDP_RX_BUF_SIZE ?
|
||||
UDP_RX_BUF_SIZE : payload_len);
|
||||
memcpy(entry->data, (const uint8_t *)data + UDP_HLEN,
|
||||
entry->len);
|
||||
entry->used = 1;
|
||||
return;
|
||||
}
|
||||
}
|
||||
/* Queue full — drop packet */
|
||||
}
|
||||
|
||||
/* ================================================================
|
||||
* Initialization
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* IPv4 callback wrapper for UDP.
|
||||
*/
|
||||
static void udp_ipv4_handler(uint32_t src_ip, uint32_t dst_ip,
|
||||
const void *payload, uint32_t len,
|
||||
uint32_t iface_idx) {
|
||||
udp_receive(src_ip, dst_ip, payload, len, iface_idx);
|
||||
}
|
||||
|
||||
void udp_init(void) {
|
||||
memset(sockets, 0, sizeof(sockets));
|
||||
ipv4_register_proto(IP_PROTO_UDP, udp_ipv4_handler);
|
||||
offset_print(" UDP: initialized\n");
|
||||
}
|
||||
128
src/udp.h
Normal file
128
src/udp.h
Normal file
@@ -0,0 +1,128 @@
|
||||
/**
|
||||
* @file udp.h
|
||||
* @brief User Datagram Protocol (UDP) subsystem.
|
||||
*
|
||||
* Provides a simple UDP socket interface built on top of the IPv4 stack.
|
||||
* Supports connectionless datagram communication.
|
||||
*/
|
||||
|
||||
#ifndef UDP_H
|
||||
#define UDP_H
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
/** Maximum number of UDP sockets. */
|
||||
#define UDP_MAX_SOCKETS 16
|
||||
|
||||
/** Maximum UDP payload size. */
|
||||
#define UDP_MAX_PAYLOAD 1472 /* ETH_MTU(1500) - IP(20) - UDP(8) */
|
||||
|
||||
/** UDP header size. */
|
||||
#define UDP_HLEN 8
|
||||
|
||||
/* ================================================================
|
||||
* UDP header
|
||||
* ================================================================ */
|
||||
|
||||
typedef struct __attribute__((packed)) udp_header {
|
||||
uint16_t src_port; /**< Source port (network byte order). */
|
||||
uint16_t dst_port; /**< Destination port (network byte order). */
|
||||
uint16_t length; /**< Total length (header + payload, NBO). */
|
||||
uint16_t checksum; /**< Checksum (0 = not computed). */
|
||||
} udp_header_t;
|
||||
|
||||
/* ================================================================
|
||||
* UDP socket
|
||||
* ================================================================ */
|
||||
|
||||
/** Maximum number of queued received datagrams per socket. */
|
||||
#define UDP_RX_QUEUE_SIZE 8
|
||||
|
||||
/** Maximum datagram size in the receive queue. */
|
||||
#define UDP_RX_BUF_SIZE 1500
|
||||
|
||||
/** Received datagram in the queue. */
|
||||
typedef struct udp_rx_entry {
|
||||
uint32_t src_ip; /**< Source IP (host byte order). */
|
||||
uint16_t src_port; /**< Source port (host byte order). */
|
||||
uint16_t len; /**< Payload length. */
|
||||
uint8_t data[UDP_RX_BUF_SIZE]; /**< Payload data. */
|
||||
uint8_t used; /**< 1 if slot is occupied. */
|
||||
} udp_rx_entry_t;
|
||||
|
||||
/**
|
||||
* UDP socket state.
|
||||
*/
|
||||
typedef struct udp_socket {
|
||||
uint16_t local_port; /**< Bound local port (host byte order). */
|
||||
uint8_t active; /**< 1 if socket is in use. */
|
||||
uint8_t bound; /**< 1 if bound to a port. */
|
||||
udp_rx_entry_t rx_queue[UDP_RX_QUEUE_SIZE]; /**< Receive queue. */
|
||||
uint32_t rx_head; /**< Next slot to write. */
|
||||
} udp_socket_t;
|
||||
|
||||
/* ================================================================
|
||||
* Public API
|
||||
* ================================================================ */
|
||||
|
||||
/**
|
||||
* Initialize the UDP subsystem.
|
||||
* Registers as an IPv4 protocol handler.
|
||||
*/
|
||||
void udp_init(void);
|
||||
|
||||
/**
|
||||
* Create a UDP socket.
|
||||
* @return Socket index (>= 0) on success, -1 on failure.
|
||||
*/
|
||||
int udp_socket_create(void);
|
||||
|
||||
/**
|
||||
* Bind a socket to a local port.
|
||||
*
|
||||
* @param sockfd Socket index.
|
||||
* @param port Local port number (host byte order).
|
||||
* @return 0 on success, -1 on failure.
|
||||
*/
|
||||
int udp_bind(int sockfd, uint16_t port);
|
||||
|
||||
/**
|
||||
* Send a UDP datagram.
|
||||
*
|
||||
* @param sockfd Socket index.
|
||||
* @param dst_ip Destination IP (host byte order).
|
||||
* @param dst_port Destination port (host byte order).
|
||||
* @param data Payload data.
|
||||
* @param len Payload length (max UDP_MAX_PAYLOAD).
|
||||
* @return Number of bytes sent, or -1 on failure.
|
||||
*/
|
||||
int udp_sendto(int sockfd, uint32_t dst_ip, uint16_t dst_port,
|
||||
const void *data, uint32_t len);
|
||||
|
||||
/**
|
||||
* Receive a UDP datagram (non-blocking).
|
||||
*
|
||||
* @param sockfd Socket index.
|
||||
* @param buf Buffer for payload.
|
||||
* @param bufsize Buffer size.
|
||||
* @param src_ip Output: source IP (host byte order, can be NULL).
|
||||
* @param src_port Output: source port (host byte order, can be NULL).
|
||||
* @return Number of bytes received, 0 if no data, -1 on error.
|
||||
*/
|
||||
int udp_recvfrom(int sockfd, void *buf, uint32_t bufsize,
|
||||
uint32_t *src_ip, uint16_t *src_port);
|
||||
|
||||
/**
|
||||
* Close a UDP socket.
|
||||
*
|
||||
* @param sockfd Socket index.
|
||||
*/
|
||||
void udp_close(int sockfd);
|
||||
|
||||
/**
|
||||
* Process an incoming UDP datagram (called from IPv4 layer).
|
||||
*/
|
||||
void udp_receive(uint32_t src_ip, uint32_t dst_ip,
|
||||
const void *data, uint32_t len, uint32_t iface_idx);
|
||||
|
||||
#endif /* UDP_H */
|
||||
Reference in New Issue
Block a user