config: writing how the config file should look

This commit is contained in:
2023-05-01 22:31:15 -07:00
parent 7630cd4e7d
commit 482a12b819
10 changed files with 205 additions and 152 deletions

View File

@@ -1,13 +0,0 @@
load("@io_bazel_rules_go//go:def.bzl", "go_library")
go_library(
name = "server",
srcs = [
"config.go",
"packet.go",
"server.go",
],
importpath = "git.wntrmute.dev/kyle/kdhcp/server",
visibility = ["//visibility:public"],
deps = ["//log"],
)

View File

@@ -1,19 +0,0 @@
package server
const (
DefaultPort = 67
DefaultMaxPacketSize = 512
DefaultNetwork = "udp4"
)
type Config struct {
Port int
MaxPacketSize int
}
func DefaultConfig() *Config {
return &Config{
Port: DefaultPort,
MaxPacketSize: DefaultMaxPacketSize,
}
}

View File

@@ -1,14 +0,0 @@
package server
import (
"net"
"github.com/insomniacslk/dhcp/dhcpv4"
)
type Packet struct {
Data []byte
Addr net.Addr
Raw *dhcpv4.DHCPv4
}

View File

@@ -1,67 +0,0 @@
package server
import (
"fmt"
"net"
"git.wntrmute.dev/kyle/kdhcp/log"
"github.com/insomniacslk/dhcp/dhcpv4"
)
type Server struct {
cfg *Config
conn net.PacketConn
}
func (srv *Server) Listen() (err error) {
if srv.conn != nil {
srv.conn.Close()
}
log.Debugf("attempting to set up packet listener on %s 0.0.0.0:%d", DefaultNetwork, srv.cfg.Port)
srv.conn, err = net.ListenPacket(DefaultNetwork, fmt.Sprintf(":%d", srv.cfg.Port))
if err != nil {
return err
}
return nil
}
func (srv *Server) Close() error {
if srv.conn != nil {
return srv.conn.Close()
}
return nil
}
func (srv *Server) ReadFrom() (*Packet, error) {
b := make([]byte, srv.cfg.MaxPacketSize)
n, addr, err := srv.conn.ReadFrom(b)
if err != nil {
return nil, err
}
packet := &Packet{
Data: b[:n],
Addr: addr,
}
packet.Raw, err = dhcpv4.FromBytes(b[:n])
if err != nil {
return packet, err
}
return packet, nil
}
func NewServer(cfg *Config) (*Server, error) {
srv := &Server{
cfg: cfg,
}
err := srv.Listen()
if err != nil {
return nil, err
}
return srv, nil
}