config: writing how the config file should look
This commit is contained in:
@@ -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"],
|
||||
)
|
||||
@@ -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,
|
||||
}
|
||||
}
|
||||
@@ -1,14 +0,0 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"net"
|
||||
|
||||
"github.com/insomniacslk/dhcp/dhcpv4"
|
||||
)
|
||||
|
||||
type Packet struct {
|
||||
Data []byte
|
||||
Addr net.Addr
|
||||
|
||||
Raw *dhcpv4.DHCPv4
|
||||
}
|
||||
@@ -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
|
||||
}
|
||||
Reference in New Issue
Block a user