mirror of
https://github.com/lwch/natpass.git
synced 2024-04-21 12:41:54 +00:00
46 lines
981 B
Go
46 lines
981 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"natpass/code/client/global"
|
|
"natpass/code/client/handler"
|
|
"natpass/code/network"
|
|
|
|
"github.com/lwch/logging"
|
|
"github.com/lwch/runtime"
|
|
"google.golang.org/grpc/metadata"
|
|
)
|
|
|
|
func makeContext(secret, id string) context.Context {
|
|
return metadata.AppendToOutgoingContext(context.Background(),
|
|
"secret", secret,
|
|
"id", id)
|
|
}
|
|
|
|
func handshake(cfg *global.Configure, cli network.Natpass_ControlClient) error {
|
|
return cli.Send(&network.ControlData{
|
|
XType: network.ControlData_come,
|
|
From: cfg.ID,
|
|
})
|
|
}
|
|
|
|
func run(cfg *global.Configure, cli network.NatpassClient) {
|
|
ctx := makeContext(cfg.Secret, cfg.ID)
|
|
|
|
ctlCli, err := cli.Control(ctx)
|
|
runtime.Assert(err)
|
|
fwdCli, err := cli.Forward(ctx)
|
|
runtime.Assert(err)
|
|
|
|
err = handshake(cfg, ctlCli)
|
|
runtime.Assert(err)
|
|
logging.Info("connect to server %s ok", cfg.Server)
|
|
|
|
h := handler.New(ctx, ctlCli, fwdCli)
|
|
|
|
h.CreateTunnels(cfg.ID, cfg.Tunnels)
|
|
go h.HandleControl()
|
|
|
|
h.HandleForward()
|
|
}
|