project init

cmd/config.go

@@ -0,0 +1,57 @@
+package cmd
+
+import (
+	"encoding/json"
+	"github.com/pion/ice/v2"
+	"github.com/wiretrustee/wiretrustee/connection"
+	"io/ioutil"
+	"os"
+)
+
+type Config struct {
+	// Wireguard private key of local peer
+	PrivateKey   string
+	Peers        []connection.Peer
+	StunTurnURLs []*ice.URL
+	// host:port of the signal server
+	SignalAddr string
+	WgAddr     string
+	WgIface    string
+}
+
+//Write writes configPath to a file
+func (cfg *Config) Write(path string) error {
+	bs, err := json.Marshal(cfg)
+	if err != nil {
+		return err
+	}
+
+	err = ioutil.WriteFile(path, bs, 0600)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+//Read reads configPath from a file
+func Read(path string) (*Config, error) {
+	f, err := os.Open(path)
+	if err != nil {
+		return nil, err
+	}
+	defer f.Close()
+
+	bs, err := ioutil.ReadAll(f)
+	if err != nil {
+		return nil, err
+	}
+
+	var cfg Config
+	err = json.Unmarshal(bs, &cfg)
+	if err != nil {
+		return nil, err
+	}
+
+	return &cfg, nil
+}

cmd/root.go

@@ -0,0 +1,35 @@
+package cmd
+
+import (
+	"fmt"
+	"github.com/spf13/cobra"
+	"os"
+	"os/signal"
+	"syscall"
+)
+
+var (
+	rootCmd = &cobra.Command{
+		Use:   "wiretrustee",
+		Short: "",
+		Long:  "",
+	}
+)
+
+// Execute executes the root command.
+func Execute() error {
+	return rootCmd.Execute()
+}
+
+func init() {
+	rootCmd.AddCommand(upCmd)
+	rootCmd.AddCommand(signalCmd)
+}
+
+func SetupCloseHandler() {
+	c := make(chan os.Signal)
+	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
+	<-c
+	fmt.Println("\r- Ctrl+C pressed in Terminal")
+	os.Exit(0)
+}

cmd/signal.go

@@ -0,0 +1,46 @@
+package cmd
+
+import (
+	"flag"
+	"fmt"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	sig "github.com/wiretrustee/wiretrustee/signal"
+	sProto "github.com/wiretrustee/wiretrustee/signal/proto"
+	"google.golang.org/grpc"
+	"net"
+)
+
+var (
+	port int
+
+	signalCmd = &cobra.Command{
+		Use:   "signal",
+		Short: "start Wiretrustee Signal Server",
+		Run: func(cmd *cobra.Command, args []string) {
+			flag.Parse()
+
+			lis, err := net.Listen("tcp", fmt.Sprintf(":%d", port))
+			if err != nil {
+				log.Fatalf("failed to listen: %v", err)
+			}
+
+			if err != nil {
+				log.Fatalf("failed to listen: %v", err)
+			}
+			var opts []grpc.ServerOption
+			grpcServer := grpc.NewServer(opts...)
+			sProto.RegisterSignalExchangeServer(grpcServer, sig.NewServer())
+			log.Printf("started server: localhost:%v", port)
+			if err := grpcServer.Serve(lis); err != nil {
+				log.Fatalf("failed to serve: %v", err)
+			}
+
+			SetupCloseHandler()
+		},
+	}
+)
+
+func init() {
+	upCmd.PersistentFlags().IntVar(&port, "port", 10000, "Server port to listen on (e.g. 10000)")
+}

cmd/up.go

@@ -0,0 +1,57 @@
+package cmd
+
+import (
+	"context"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"github.com/wiretrustee/wiretrustee/connection"
+	sig "github.com/wiretrustee/wiretrustee/signal"
+	"os"
+)
+
+const (
+	ExitSetupFailed = 1
+)
+
+var (
+	configPath string
+	logLevel   string
+
+	upCmd = &cobra.Command{
+		Use:   "up",
+		Short: "start wiretrustee",
+		Run: func(cmd *cobra.Command, args []string) {
+			level, err := log.ParseLevel(logLevel)
+			if err != nil {
+				log.Errorf("efailed parsing log-level %s: %s", logLevel, err)
+				os.Exit(ExitSetupFailed)
+			}
+			log.SetLevel(level)
+
+			config, _ := Read(configPath)
+
+			ctx := context.Background()
+			signalClient, err := sig.NewClient(config.SignalAddr, ctx)
+			if err != nil {
+				log.Errorf("error while connecting to the Signal Exchange Service %s: %s", config.SignalAddr, err)
+				os.Exit(ExitSetupFailed)
+			}
+			//todo proper close handling
+			defer func() { signalClient.Close() }()
+
+			engine := connection.NewEngine(signalClient, config.StunTurnURLs, config.WgIface, config.WgAddr)
+
+			err = engine.Start(config.PrivateKey, config.Peers)
+
+			//signalClient.WaitConnected()
+
+			SetupCloseHandler()
+		},
+	}
+)
+
+func init() {
+	upCmd.PersistentFlags().StringVar(&configPath, "config", "", "")
+	upCmd.PersistentFlags().StringVar(&logLevel, "log-level", "info", "")
+	upCmd.MarkPersistentFlagRequired("config")
+}