From f0f7130b083c293d9937d2bcc18146e86db4d4ad Mon Sep 17 00:00:00 2001 From: patterniha <71074308+patterniha@users.noreply.github.com> Date: Tue, 9 Sep 2025 18:53:02 +0330 Subject: [PATCH] Update bridge.go --- app/reverse/bridge.go | 48 +++++++++++++++++++++---------------------- 1 file changed, 24 insertions(+), 24 deletions(-) diff --git a/app/reverse/bridge.go b/app/reverse/bridge.go index 7fb5856c..fe3b2c3a 100644 --- a/app/reverse/bridge.go +++ b/app/reverse/bridge.go @@ -55,7 +55,7 @@ func (b *Bridge) cleanup() { activeWorkers = append(activeWorkers, w) } if w.Closed() { - w.timer.SetTimeout(0) + w.Timer.SetTimeout(0) } } @@ -98,11 +98,11 @@ func (b *Bridge) Close() error { } type BridgeWorker struct { - tag string - worker *mux.ServerWorker - dispatcher routing.Dispatcher - state Control_State - timer *signal.ActivityTimer + Tag string + Worker *mux.ServerWorker + Dispatcher routing.Dispatcher + State Control_State + Timer *signal.ActivityTimer } func NewBridgeWorker(domain string, tag string, d routing.Dispatcher) (*BridgeWorker, error) { @@ -120,20 +120,20 @@ func NewBridgeWorker(domain string, tag string, d routing.Dispatcher) (*BridgeWo } w := &BridgeWorker{ - dispatcher: d, - tag: tag, + Dispatcher: d, + Tag: tag, } worker, err := mux.NewServerWorker(context.Background(), w, link) if err != nil { return nil, err } - w.worker = worker + w.Worker = worker terminate := func() { worker.Close() } - w.timer = signal.CancelAfterInactivity(ctx, terminate, 60*time.Second) // prevent leak + w.Timer = signal.CancelAfterInactivity(ctx, terminate, 60*time.Second) return w, nil } @@ -150,15 +150,15 @@ func (w *BridgeWorker) Close() error { } func (w *BridgeWorker) IsActive() bool { - return w.state == Control_ACTIVE && !w.worker.Closed() + return w.State == Control_ACTIVE && !w.Worker.Closed() } func (w *BridgeWorker) Closed() bool { - return w.worker.Closed() + return w.Worker.Closed() } func (w *BridgeWorker) Connections() uint32 { - return w.worker.ActiveConnections() + return w.Worker.ActiveConnections() } func (w *BridgeWorker) handleInternalConn(link *transport.Link) { @@ -167,22 +167,22 @@ func (w *BridgeWorker) handleInternalConn(link *transport.Link) { mb, err := reader.ReadMultiBuffer() if err != nil { if w.Closed() { - w.timer.SetTimeout(0) + w.Timer.SetTimeout(0) } else { - w.timer.SetTimeout(24 * time.Hour) + w.Timer.SetTimeout(24 * time.Hour) } return } - w.timer.Update() + w.Timer.Update() for _, b := range mb { var ctl Control if err := proto.Unmarshal(b.Bytes(), &ctl); err != nil { errors.LogInfoInner(context.Background(), err, "failed to parse proto message") - w.timer.SetTimeout(0) + w.Timer.SetTimeout(0) return } - if ctl.State != w.state { - w.state = ctl.State + if ctl.State != w.State { + w.State = ctl.State } } } @@ -191,9 +191,9 @@ func (w *BridgeWorker) handleInternalConn(link *transport.Link) { func (w *BridgeWorker) Dispatch(ctx context.Context, dest net.Destination) (*transport.Link, error) { if !isInternalDomain(dest) { ctx = session.ContextWithInbound(ctx, &session.Inbound{ - Tag: w.tag, + Tag: w.Tag, }) - return w.dispatcher.Dispatch(ctx, dest) + return w.Dispatcher.Dispatch(ctx, dest) } opt := []pipe.Option{pipe.WithSizeLimit(16 * 1024)} @@ -214,12 +214,12 @@ func (w *BridgeWorker) Dispatch(ctx context.Context, dest net.Destination) (*tra func (w *BridgeWorker) DispatchLink(ctx context.Context, dest net.Destination, link *transport.Link) error { if !isInternalDomain(dest) { ctx = session.ContextWithInbound(ctx, &session.Inbound{ - Tag: w.tag, + Tag: w.Tag, }) - return w.dispatcher.DispatchLink(ctx, dest, link) + return w.Dispatcher.DispatchLink(ctx, dest, link) } - link = w.dispatcher.(*dispatcher.DefaultDispatcher).WrapLink(ctx, link) + link = w.Dispatcher.(*dispatcher.DefaultDispatcher).WrapLink(ctx, link) w.handleInternalConn(link) return nil