extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-06-24 11:41:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

backend request information (#314)

Browse Source
This commit is contained in:
Michael Quigley 2023-04-25 15:16:05 -04:00
parent bf0fbb0e35
commit 980d0dead7
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
cmd/zrok/sharePrivate.go
View File

@ -177,6 +177,7 @@ func (cmd *sharePrivateCommand) run(_ *cobra.Command, args []string) {
IdentityPath: zif, IdentityPath: zif,
EndpointAddress: target, EndpointAddress: target,
ShrToken: resp.Payload.ShrToken, ShrToken: resp.Payload.ShrToken,
RequestsChan: requestsChan,
} }
be, err := tcpTunnel.NewBackend(cfg) be, err := tcpTunnel.NewBackend(cfg)
if err != nil { if err != nil {

9
endpoints/tcpTunnel/backend.go
View File

@ -15,6 +15,7 @@ type BackendConfig struct {
IdentityPath string IdentityPath string
EndpointAddress string EndpointAddress string
ShrToken string ShrToken string
RequestsChan chan *endpoints.Request
} }
type Backend struct { type Backend struct {
@ -56,11 +57,17 @@ func (b *Backend) Run() error {
} }
func (b *Backend) handle(conn net.Conn) { func (b *Backend) handle(conn net.Conn) {
logrus.Infof("handling '%v'", conn.RemoteAddr()) logrus.Debugf("handling '%v'", conn.RemoteAddr())
if rAddr, err := net.ResolveTCPAddr("tcp", b.cfg.EndpointAddress); err == nil { if rAddr, err := net.ResolveTCPAddr("tcp", b.cfg.EndpointAddress); err == nil {
if rConn, err := net.DialTCP("tcp", nil, rAddr); err == nil { if rConn, err := net.DialTCP("tcp", nil, rAddr); err == nil {
go endpoints.TXer(conn, rConn) go endpoints.TXer(conn, rConn)
go endpoints.TXer(rConn, conn) go endpoints.TXer(rConn, conn)
b.cfg.RequestsChan <- &endpoints.Request{
Stamp: time.Now(),
RemoteAddr: conn.RemoteAddr().String(),
Method: "OPEN",
Path: rAddr.String(),
}
} else { } else {
logrus.Errorf("error dialing '%v': %v", b.cfg.EndpointAddress, err) logrus.Errorf("error dialing '%v': %v", b.cfg.EndpointAddress, err)
_ = conn.Close() _ = conn.Close()