mirror of
https://github.com/JannisHeydemann/BoredOS.git
synced 2026-05-30 10:26:59 +00:00
Network commands port to userspace
This commit is contained in:
boreddevnl
376
src/kernel/userland/net.c
Normal file
376
src/kernel/userland/net.c
Normal file
@@ -0,0 +1,376 @@
|
||||
#include <stdlib.h>
|
||||
#include <syscall.h>
|
||||
|
||||
// Helper function to print a MAC address
|
||||
static void print_mac(const net_mac_address_t* mac) {
|
||||
char buf[64];
|
||||
int p = 0;
|
||||
for (int i = 0; i < 6; i++) {
|
||||
int v = mac->bytes[i];
|
||||
int hi = (v >> 4) & 0xF;
|
||||
int lo = v & 0xF;
|
||||
buf[p++] = (hi < 10) ? ('0' + hi) : ('A' + (hi - 10));
|
||||
buf[p++] = (lo < 10) ? ('0' + lo) : ('A' + (lo - 10));
|
||||
if (i < 5) buf[p++] = ':';
|
||||
}
|
||||
buf[p] = 0;
|
||||
printf("%s", buf);
|
||||
}
|
||||
|
||||
// Helper to parse integer from string
|
||||
static int string_to_int(const char *str) {
|
||||
int result = 0;
|
||||
int sign = 1;
|
||||
if (*str == '-') {
|
||||
sign = -1;
|
||||
str++;
|
||||
}
|
||||
while (*str >= '0' && *str <= '9') {
|
||||
result = result * 10 + (*str - '0');
|
||||
str++;
|
||||
}
|
||||
return result * sign;
|
||||
}
|
||||
|
||||
// Helper to get string length
|
||||
static int string_length(const char *str) {
|
||||
int len = 0;
|
||||
while (str[len]) len++;
|
||||
return len;
|
||||
}
|
||||
|
||||
// Command: netinit - Initialize network and acquire DHCP
|
||||
static void cmd_netinit(void) {
|
||||
int r = sys_network_init();
|
||||
if (r == 0) {
|
||||
printf("Network initialized\n");
|
||||
int d = sys_network_dhcp_acquire();
|
||||
if (d == 0) {
|
||||
printf("DHCP acquired\n");
|
||||
} else {
|
||||
printf("DHCP failed\n");
|
||||
}
|
||||
} else {
|
||||
printf("Network init failed\n");
|
||||
}
|
||||
}
|
||||
|
||||
// Command: netinfo - Display network information
|
||||
static void cmd_netinfo(void) {
|
||||
if (!sys_network_is_initialized()) {
|
||||
printf("Error: Network not initialized. Run 'net init' first.\n");
|
||||
return;
|
||||
}
|
||||
|
||||
net_mac_address_t mac;
|
||||
net_ipv4_address_t ip;
|
||||
|
||||
if (sys_network_get_mac(&mac) == 0) {
|
||||
printf("MAC: ");
|
||||
print_mac(&mac);
|
||||
printf("\n");
|
||||
}
|
||||
|
||||
if (sys_network_get_ip(&ip) == 0) {
|
||||
printf("IP: ");
|
||||
for (int i = 0; i < 4; i++) {
|
||||
printf("%d", ip.bytes[i]);
|
||||
if (i < 3) printf(".");
|
||||
}
|
||||
printf("\n");
|
||||
}
|
||||
|
||||
printf("Frames: %d\n", sys_network_get_stat(0));
|
||||
printf("UDP packets: %d\n", sys_network_get_stat(1));
|
||||
printf("UDP callbacks: %d\n", sys_network_get_stat(2));
|
||||
printf("E1000 receive calls: %d\n", sys_network_get_stat(3));
|
||||
printf("E1000 receive empty: %d\n", sys_network_get_stat(4));
|
||||
printf("Process calls: %d\n", sys_network_get_stat(5));
|
||||
}
|
||||
|
||||
// Command: ipset - Set IP address manually
|
||||
static void cmd_ipset(const char *args) {
|
||||
if (!sys_network_is_initialized()) {
|
||||
printf("Error: Network not initialized. Run 'net init' first.\n");
|
||||
return;
|
||||
}
|
||||
|
||||
if (!args || !*args) {
|
||||
printf("Usage: NET IPSET a.b.c.d\n");
|
||||
return;
|
||||
}
|
||||
|
||||
net_ipv4_address_t ip = {{0, 0, 0, 0}};
|
||||
int part = 0;
|
||||
int val = 0;
|
||||
int i = 0;
|
||||
|
||||
while (args[i]) {
|
||||
char ch = args[i++];
|
||||
if (ch >= '0' && ch <= '9') {
|
||||
val = val * 10 + (ch - '0');
|
||||
if (val > 255) {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
} else if (ch == '.') {
|
||||
if (part > 3) {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
ip.bytes[part++] = (uint8_t)val;
|
||||
val = 0;
|
||||
} else {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
if (part != 3) {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
ip.bytes[3] = (uint8_t)val;
|
||||
|
||||
if (sys_network_set_ip(&ip) == 0) {
|
||||
printf("IP set\n");
|
||||
} else {
|
||||
printf("IP set failed\n");
|
||||
}
|
||||
}
|
||||
|
||||
// Command: udpsend - Send UDP packet
|
||||
static void cmd_udpsend(const char *args) {
|
||||
if (!sys_network_is_initialized()) {
|
||||
printf("Error: Network not initialized. Run 'net init' first.\n");
|
||||
return;
|
||||
}
|
||||
|
||||
if (!args || !*args) {
|
||||
printf("Usage: NET UDPSEND ip port data\n");
|
||||
return;
|
||||
}
|
||||
|
||||
// Parse IP address
|
||||
char ipstr[32];
|
||||
int pos = 0;
|
||||
while (args[pos] && args[pos] != ' ') {
|
||||
ipstr[pos] = args[pos];
|
||||
pos++;
|
||||
}
|
||||
ipstr[pos] = 0;
|
||||
|
||||
while (args[pos] == ' ') pos++;
|
||||
|
||||
// Parse port
|
||||
char portstr[16];
|
||||
int p = 0;
|
||||
while (args[pos] && args[pos] != ' ') {
|
||||
portstr[p++] = args[pos++];
|
||||
}
|
||||
portstr[p] = 0;
|
||||
|
||||
while (args[pos] == ' ') pos++;
|
||||
|
||||
// Get data
|
||||
const char *datastr = args + pos;
|
||||
|
||||
// Parse IP
|
||||
net_ipv4_address_t ip = {{0, 0, 0, 0}};
|
||||
int idx = 0;
|
||||
int val = 0;
|
||||
int j = 0;
|
||||
|
||||
while (ipstr[j]) {
|
||||
char ch = ipstr[j++];
|
||||
if (ch >= '0' && ch <= '9') {
|
||||
val = val * 10 + (ch - '0');
|
||||
if (val > 255) {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
} else if (ch == '.') {
|
||||
if (idx > 3) {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
ip.bytes[idx++] = (uint8_t)val;
|
||||
val = 0;
|
||||
} else {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
if (idx != 3) {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
ip.bytes[3] = (uint8_t)val;
|
||||
|
||||
// Parse port
|
||||
int port = 0;
|
||||
int k = 0;
|
||||
while (portstr[k]) {
|
||||
char ch = portstr[k++];
|
||||
if (ch < '0' || ch > '9') {
|
||||
printf("Invalid port\n");
|
||||
return;
|
||||
}
|
||||
port = port * 10 + (ch - '0');
|
||||
}
|
||||
|
||||
if (port <= 0 || port > 65535) {
|
||||
printf("Invalid port\n");
|
||||
return;
|
||||
}
|
||||
|
||||
int len = string_length(datastr);
|
||||
if (len <= 0) {
|
||||
printf("No data\n");
|
||||
return;
|
||||
}
|
||||
|
||||
int r = sys_udp_send(&ip, (uint16_t)port, 12345, datastr, (size_t)len);
|
||||
if (r == 0) {
|
||||
printf("Sent\n");
|
||||
} else {
|
||||
printf("Send failed\n");
|
||||
}
|
||||
}
|
||||
|
||||
// Command: ping - Send ICMP ping request
|
||||
static void cmd_ping(const char *args) {
|
||||
if (!sys_network_is_initialized()) {
|
||||
printf("Error: Network not initialized. Run 'net init' first.\n");
|
||||
return;
|
||||
}
|
||||
|
||||
if (!args || !*args) {
|
||||
printf("Usage: NET PING a.b.c.d\n");
|
||||
return;
|
||||
}
|
||||
|
||||
net_ipv4_address_t ip = {{0, 0, 0, 0}};
|
||||
int idx = 0;
|
||||
int val = 0;
|
||||
int j = 0;
|
||||
|
||||
while (args[j]) {
|
||||
char ch = args[j++];
|
||||
if (ch >= '0' && ch <= '9') {
|
||||
val = val * 10 + (ch - '0');
|
||||
if (val > 255) {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
} else if (ch == '.') {
|
||||
if (idx > 3) {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
ip.bytes[idx++] = (uint8_t)val;
|
||||
val = 0;
|
||||
} else if (ch == ' ' || ch == '\t') {
|
||||
// Skip whitespace
|
||||
while (args[j] == ' ' || args[j] == '\t') j++;
|
||||
j--;
|
||||
} else {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
if (idx != 3) {
|
||||
printf("Invalid IP\n");
|
||||
return;
|
||||
}
|
||||
ip.bytes[3] = (uint8_t)val;
|
||||
|
||||
printf("Pinging ");
|
||||
for (int i = 0; i < 4; i++) {
|
||||
printf("%d", ip.bytes[i]);
|
||||
if (i < 3) printf(".");
|
||||
}
|
||||
printf("...\n");
|
||||
|
||||
int result = sys_icmp_ping(&ip);
|
||||
|
||||
if (result == -2) {
|
||||
printf("Error: Network not initialized. Run 'net init' first.\n");
|
||||
} else if (result < 0) {
|
||||
printf("Error: Failed to send ping request.\n");
|
||||
} else {
|
||||
printf("Ping complete: %d/%d replies received\n", result, 4);
|
||||
}
|
||||
}
|
||||
|
||||
// Command: help
|
||||
static void cmd_help(void) {
|
||||
printf("Network Commands:\n");
|
||||
printf(" NET INIT - Initialize network and acquire DHCP\n");
|
||||
printf(" NET INFO - Display network information\n");
|
||||
printf(" NET IPSET a.b.c.d - Set IP address\n");
|
||||
printf(" NET UDPSEND ip port data - Send UDP packet\n");
|
||||
printf(" NET PING a.b.c.d - Send ICMP ping request\n");
|
||||
printf(" NET HELP - Show this help\n");
|
||||
}
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
if (argc < 2) {
|
||||
cmd_help();
|
||||
return 1;
|
||||
}
|
||||
|
||||
const char *cmd = argv[1];
|
||||
|
||||
// Convert command to uppercase for easier comparison
|
||||
char cmd_upper[32];
|
||||
int i = 0;
|
||||
while (cmd[i] && i < 31) {
|
||||
char c = cmd[i];
|
||||
cmd_upper[i] = (c >= 'a' && c <= 'z') ? (c - 'a' + 'A') : c;
|
||||
i++;
|
||||
}
|
||||
cmd_upper[i] = 0;
|
||||
|
||||
// Parse arguments if provided
|
||||
const char *args = "";
|
||||
if (argc > 2) {
|
||||
args = argv[2];
|
||||
}
|
||||
|
||||
if ((string_length(cmd_upper) == 4 &&
|
||||
cmd_upper[0] == 'I' && cmd_upper[1] == 'N' && cmd_upper[2] == 'I' && cmd_upper[3] == 'T') ||
|
||||
(string_length(cmd_upper) == 7 &&
|
||||
cmd_upper[0] == 'N' && cmd_upper[1] == 'E' && cmd_upper[2] == 'T' &&
|
||||
cmd_upper[3] == 'I' && cmd_upper[4] == 'N' && cmd_upper[5] == 'I' && cmd_upper[6] == 'T')) {
|
||||
cmd_netinit();
|
||||
} else if ((string_length(cmd_upper) == 4 &&
|
||||
cmd_upper[0] == 'I' && cmd_upper[1] == 'N' && cmd_upper[2] == 'F' && cmd_upper[3] == 'O') ||
|
||||
(string_length(cmd_upper) == 7 &&
|
||||
cmd_upper[0] == 'N' && cmd_upper[1] == 'E' && cmd_upper[2] == 'T' &&
|
||||
cmd_upper[3] == 'I' && cmd_upper[4] == 'N' && cmd_upper[5] == 'F' && cmd_upper[6] == 'O')) {
|
||||
cmd_netinfo();
|
||||
} else if (string_length(cmd_upper) == 5 &&
|
||||
cmd_upper[0] == 'I' && cmd_upper[1] == 'P' && cmd_upper[2] == 'S' &&
|
||||
cmd_upper[3] == 'E' && cmd_upper[4] == 'T') {
|
||||
cmd_ipset(args);
|
||||
} else if (string_length(cmd_upper) == 7 &&
|
||||
cmd_upper[0] == 'U' && cmd_upper[1] == 'D' && cmd_upper[2] == 'P' &&
|
||||
cmd_upper[3] == 'S' && cmd_upper[4] == 'E' && cmd_upper[5] == 'N' && cmd_upper[6] == 'D') {
|
||||
cmd_udpsend(args);
|
||||
} else if (string_length(cmd_upper) == 4 &&
|
||||
cmd_upper[0] == 'P' && cmd_upper[1] == 'I' && cmd_upper[2] == 'N' && cmd_upper[3] == 'G') {
|
||||
cmd_ping(args);
|
||||
} else if (string_length(cmd_upper) == 4 &&
|
||||
cmd_upper[0] == 'H' && cmd_upper[1] == 'E' && cmd_upper[2] == 'L' && cmd_upper[3] == 'P') {
|
||||
cmd_help();
|
||||
} else {
|
||||
printf("Unknown command: %s\n", cmd);
|
||||
cmd_help();
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user