68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net"
|
|
"time"
|
|
|
|
pb "git.h3cx.dev/NulTNL/control-plane-proto-go"
|
|
"golang.zx2c4.com/wireguard/wgctrl"
|
|
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type server struct {
|
|
pb.UnimplementedControlServiceServer
|
|
wgClient *wgctrl.Client
|
|
}
|
|
|
|
func (s *server) Ping(ctx context.Context, req *pb.PingRequest) (*pb.PingResponse, error) {
|
|
key, _ := wgtypes.GeneratePrivateKey()
|
|
newcfg(s.wgClient, key.PublicKey())
|
|
return &pb.PingResponse{Reply: "Pong: " + req.Message}, nil
|
|
}
|
|
|
|
func main() {
|
|
client, _ := wgctrl.New()
|
|
defer client.Close()
|
|
|
|
lis, err := net.Listen("tcp", ":50051")
|
|
if err != nil {
|
|
log.Fatalf("failed to listen: %v", err)
|
|
}
|
|
|
|
grpcServer := grpc.NewServer()
|
|
srv := &server{wgClient: client}
|
|
pb.RegisterControlServiceServer(grpcServer, srv)
|
|
|
|
log.Println("gRPC server running on :50051")
|
|
if err := grpcServer.Serve(lis); err != nil {
|
|
log.Fatalf("failed to server: %v", err)
|
|
}
|
|
|
|
}
|
|
|
|
func newcfg(client *wgctrl.Client, pubkey wgtypes.Key) {
|
|
|
|
allowedIP1 := net.IPNet{IP: net.ParseIP("10.0.0.3"), Mask: net.CIDRMask(32, 32)}
|
|
persistentKeepalive := 25 * time.Second
|
|
|
|
peerCfg := wgtypes.PeerConfig{
|
|
PublicKey: pubkey,
|
|
AllowedIPs: []net.IPNet{allowedIP1},
|
|
PersistentKeepaliveInterval: &persistentKeepalive,
|
|
}
|
|
|
|
cfg := wgtypes.Config{
|
|
Peers: []wgtypes.PeerConfig{peerCfg},
|
|
}
|
|
|
|
if err := client.ConfigureDevice("wg0", cfg); err != nil {
|
|
log.Fatalf("failed to add peer: %v", err)
|
|
}
|
|
|
|
log.Println("Peer added")
|
|
|
|
}
|