main.go

package main

import (
	"embed"
	"net"
	"os"
	"os/signal"
	"syscall"
)

//go:embed tftpboot
var tftpRoot embed.FS

func main() {
	// config
	Conf := &Config{}
	Conf.ParseConfig("config.yaml")
	// logger
	Initial(Conf.Logger.Level, InitialFD(Conf.Logger.File))
	// dhcp
	ip := net.ParseIP(Conf.IPAddr)
	dhcpServer := &DHCPServer{
		Handler: &DHCPHandler{
			DHCPAddr: ip,
			TFTPAddr: ip,
		},
		Iface: Conf.Iface,
		Port:  Conf.DHCP.Port,
	}
	go dhcpServer.Run()
	// tftp
	tftpServer := &TFTPServer{
		Handler: &TFTPHandler{
			EmbedRoot: tftpRoot,
			TftpAddr:  ip,
			PXEConfig: Conf.PXE,
		},
		Port:     Conf.TFTP.Port,
		External: Conf.TFTP.External,
	}
	go tftpServer.Run()

	sigs := make(chan os.Signal, 1)
	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
	done := make(chan bool, 1)
	go func() {
		<-sigs
		done <- true
	}()
	Info("Awaiting signal ......")
	<-done
	Info("Except signal, exiting ......")
}