diff options
Diffstat (limited to 'libnet')
| -rw-r--r-- | libnet/Makefile | 21 | ||||
| -rw-r--r-- | libnet/dns.c | 53 | ||||
| -rw-r--r-- | libnet/inc/net.h | 25 |
3 files changed, 99 insertions, 0 deletions
diff --git a/libnet/Makefile b/libnet/Makefile new file mode 100644 index 0000000..507a564 --- /dev/null +++ b/libnet/Makefile @@ -0,0 +1,21 @@ +# MIT License, Copyright (c) 2020 Marvin Borner + +COBJS = dns.o +CC = ccache ../cross/opt/bin/i686-elf-gcc +LD = ccache ../cross/opt/bin/i686-elf-ld +AR = ccache ../cross/opt/bin/i686-elf-ar + +WARNINGS = -Wall -Wextra -pedantic-errors -Wshadow -Wpointer-arith -Wwrite-strings -Wredundant-decls -Wnested-externs -Wno-long-long +CFLAGS = $(WARNINGS) -nostdlib -nostdinc -fno-builtin -mgeneral-regs-only -std=c99 -m32 -Iinc/ -I../libc/inc/ -fPIE -Duserspace -Ofast + +all: libtxt + +%.o: %.c + @$(CC) -c $(CFLAGS) $< -o $@ + +libtxt: $(COBJS) + @mkdir -p ../build/ + @$(AR) rcs ../build/libnet.a $+ + +clean: + @find . -name "*.o" -type f -delete diff --git a/libnet/dns.c b/libnet/dns.c new file mode 100644 index 0000000..b35eb72 --- /dev/null +++ b/libnet/dns.c @@ -0,0 +1,53 @@ +// MIT License, Copyright (c) 2020 Marvin Borner + +#include <def.h> +#include <mem.h> +#include <random.h> +#include <socket.h> +#include <str.h> + +static u32 dns_ip_addr = ip(1, 1, 1, 1); + +struct dns_packet { + u16 qid; + u16 flags; + u16 questions; + u16 answers; + u16 authorities; + u16 additional; + u8 data[]; +} __attribute__((packed)); + +static void dns_make_packet(struct dns_packet *packet, const char *name, const char *tld) +{ + packet->qid = htons(rand()); + packet->flags = htons(0x0100); // Standard query + packet->questions = htons(1); + packet->answers = htons(0); + packet->authorities = htons(0); + packet->additional = htons(0); + + packet->data[0] = (u8)strlen(name); + memcpy(&packet->data[1], name, (u8)strlen(name)); + packet->data[(u8)strlen(name) + 1] = (u8)strlen(tld); + memcpy(&packet->data[(u8)strlen(name) + 2], tld, (u8)strlen(tld)); + packet->data[(u8)strlen(name) + (u8)strlen(tld) + 2] = 0x00; // Name end + packet->data[(u8)strlen(name) + (u8)strlen(tld) + 4] = 0x01; // A + packet->data[(u8)strlen(name) + (u8)strlen(tld) + 6] = 0x01; // IN +} + +void dns_request(const char *name, const char *tld) +{ + struct socket *socket = net_open(S_UDP); + if (socket) + socket->src_port = 50053; + if (!socket || !net_connect(socket, dns_ip_addr, 53)) + return; + + u32 length = sizeof(struct dns_packet) + strlen(name) + strlen(tld) + 7; // TODO: 7 :) + struct dns_packet *packet = malloc(length); + memset(packet, 0, length); + dns_make_packet(packet, name, tld); + net_send(socket, packet, length); + free(packet); +} diff --git a/libnet/inc/net.h b/libnet/inc/net.h new file mode 100644 index 0000000..836e181 --- /dev/null +++ b/libnet/inc/net.h @@ -0,0 +1,25 @@ +// MIT License, Copyright (c) 2020 Marvin Borner + +#ifndef NET_H +#define NET_H + +#include <socket.h> + +#define htonl(l) \ + ((((l)&0xff) << 24) | (((l)&0xff00) << 8) | (((l)&0xff0000) >> 8) | \ + (((l)&0xff000000) >> 24)) +#define htons(s) ((((s)&0xff) << 8) | (((s)&0xff00) >> 8)) +#define ntohl(l) htonl((l)) +#define ntohs(s) htons((s)) +#define ip(a, b, c, d) \ + ((((a)&0xff) << 24) | (((b)&0xff) << 16) | (((c)&0xff) << 8) | (((d)&0xff) << 0)) + +#define net_open(type) (void *)sys1(SYS_NET_OPEN, (int)(type)) +#define net_close(socket) (void)sys1(SYS_NET_CLOSE, (int)(type)) +#define net_connect(socket, ip_addr, dst_port) \ + (int)sys3(SYS_NET_CONNECT, (int)(socket), (int)(ip_addr), (int)(dst_port)) +#define net_send(socket, data, len) (void)sys3(SYS_NET_SEND, (int)(socket), (int)(data), (int)(len)) + +void dns_request(const char *name, const char *tld); + +#endif |