2021-05-01 12:45:37 +02:00
|
|
|
package cmd
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/wiretrustee/wiretrustee/connection"
|
|
|
|
sig "github.com/wiretrustee/wiretrustee/signal"
|
2021-05-01 18:29:59 +02:00
|
|
|
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
2021-05-01 12:45:37 +02:00
|
|
|
"os"
|
|
|
|
)
|
|
|
|
|
2021-05-05 10:40:53 +02:00
|
|
|
func toByte32(key wgtypes.Key) *[32]byte {
|
|
|
|
return (*[32]byte)(&key)
|
|
|
|
}
|
|
|
|
|
2021-05-01 12:45:37 +02:00
|
|
|
var (
|
|
|
|
upCmd = &cobra.Command{
|
|
|
|
Use: "up",
|
|
|
|
Short: "start wiretrustee",
|
|
|
|
Run: func(cmd *cobra.Command, args []string) {
|
2021-05-01 15:47:24 +02:00
|
|
|
InitLog(logLevel)
|
2021-05-01 12:45:37 +02:00
|
|
|
|
|
|
|
config, _ := Read(configPath)
|
|
|
|
|
2021-05-01 18:29:59 +02:00
|
|
|
myKey, err := wgtypes.ParseKey(config.PrivateKey)
|
|
|
|
if err != nil {
|
|
|
|
log.Errorf("failed parsing Wireguard key %s: [%s]", config.PrivateKey, err.Error())
|
|
|
|
os.Exit(ExitSetupFailed)
|
|
|
|
}
|
|
|
|
|
2021-05-01 12:45:37 +02:00
|
|
|
ctx := context.Background()
|
2021-05-01 18:29:59 +02:00
|
|
|
signalClient, err := sig.NewClient(config.SignalAddr, myKey, ctx)
|
2021-05-01 12:45:37 +02:00
|
|
|
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)
|
|
|
|
|
2021-05-01 18:29:59 +02:00
|
|
|
err = engine.Start(myKey, config.Peers)
|
2021-05-01 12:45:37 +02:00
|
|
|
|
|
|
|
//signalClient.WaitConnected()
|
|
|
|
|
|
|
|
SetupCloseHandler()
|
|
|
|
},
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
}
|