wifictld-analyzer/cmd/controller.go

package cmd

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

	"dev.sum7.eu/genofire/golang-lib/file"
	"github.com/bdlm/log"
	"github.com/spf13/cobra"

	"dev.sum7.eu/genofire/wifictld-analyzer/capture"
	"dev.sum7.eu/genofire/wifictld-analyzer/config"
	"dev.sum7.eu/genofire/wifictld-analyzer/controller"
	"dev.sum7.eu/genofire/wifictld-analyzer/data"
	"dev.sum7.eu/genofire/wifictld-analyzer/database"
	"dev.sum7.eu/genofire/wifictld-analyzer/web"
)

// queryCmd represents the query command
var controllerCmd = &cobra.Command{
	Use:     "controller <interfaces>",
	Short:   "simulate a wifictld controller",
	Example: `analyzer controller "/etc/wifictld.conf"`,
	Args:    cobra.ExactArgs(1),
	Run: func(cmd *cobra.Command, args []string) {
		configObj := &config.Config{}

		file.ReadTOML(args[0], configObj)

		db := database.NewDB(configObj.StatePath)

		ctr := controller.NewController(db)
		defer ctr.Close()

		var handlers []data.Handler

		if configObj.Webserver.Enable {
			log.Infof("starting webserver on %s", configObj.Webserver.Bind)
			srv := web.New(configObj.Webserver)
			go srv.Start()
			handlers = append(handlers, srv.Handler)
			defer srv.Close()
		}

		coll := capture.NewCollector(func(addr *net.UDPAddr, msg *data.SocketMSG) (*data.SocketMSG, error) {
			for _, a := range handlers {
				a(addr, msg)
			}
			if !configObj.Answer {
				ctr.Handler(addr, msg)
				return nil, nil
			}
			return ctr.Handler(addr, msg)
		}, configObj.Interfaces)
		defer coll.Close()

		ctr.Send = coll.Send
		ctr.SendTo = coll.SendTo

		// Wait for INT/TERM
		sigs := make(chan os.Signal, 1)
		signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
		sig := <-sigs
		log.Println("received", sig)

	},
}

func init() {
	RootCMD.AddCommand(controllerCmd)
}