HTTP proxy 24 hours timeout, fix #303

This commit is contained in:
Quentin McGaw
2020-11-21 01:26:02 +00:00
parent 1cc0f5fee9
commit fb62910b17
6 changed files with 17 additions and 28 deletions

View File

@@ -243,7 +243,7 @@ func _main(background context.Context, args []string) int { //nolint:gocognit,go
go publicIPLooper.RunRestartTicker(ctx, wg) go publicIPLooper.RunRestartTicker(ctx, wg)
publicIPLooper.SetPeriod(allSettings.PublicIPPeriod) // call after RunRestartTicker publicIPLooper.SetPeriod(allSettings.PublicIPPeriod) // call after RunRestartTicker
httpProxyLooper := httpproxy.NewLooper(httpClient, logger, allSettings.HTTPProxy) httpProxyLooper := httpproxy.NewLooper(logger, allSettings.HTTPProxy)
wg.Add(1) wg.Add(1)
go httpProxyLooper.Run(ctx, wg) go httpProxyLooper.Run(ctx, wg)

View File

@@ -9,20 +9,18 @@ import (
"github.com/qdm12/golibs/logging" "github.com/qdm12/golibs/logging"
) )
func newHandler(ctx context.Context, wg *sync.WaitGroup, func newHandler(ctx context.Context, wg *sync.WaitGroup, logger logging.Logger,
client *http.Client, logger logging.Logger,
stealth, verbose bool, username, password string) http.Handler { stealth, verbose bool, username, password string) http.Handler {
const relayTimeout = 10 * time.Second const httpTimeout = 24 * time.Hour
return &handler{ return &handler{
ctx: ctx, ctx: ctx,
wg: wg, wg: wg,
client: client, client: &http.Client{Timeout: httpTimeout},
logger: logger, logger: logger,
relayTimeout: relayTimeout, verbose: verbose,
verbose: verbose, stealth: stealth,
stealth: stealth, username: username,
username: username, password: password,
password: password,
} }
} }
@@ -31,7 +29,6 @@ type handler struct {
wg *sync.WaitGroup wg *sync.WaitGroup
client *http.Client client *http.Client
logger logging.Logger logger logging.Logger
relayTimeout time.Duration
verbose, stealth bool verbose, stealth bool
username, password string username, password string
} }

View File

@@ -1,7 +1,6 @@
package httpproxy package httpproxy
import ( import (
"context"
"fmt" "fmt"
"io" "io"
"net" "net"
@@ -18,9 +17,7 @@ func (h *handler) handleHTTP(responseWriter http.ResponseWriter, request *http.R
return return
} }
ctx, cancel := context.WithTimeout(h.ctx, h.relayTimeout) request = request.WithContext(h.ctx)
defer cancel()
request = request.WithContext(ctx)
request.RequestURI = "" request.RequestURI = ""

View File

@@ -9,7 +9,7 @@ import (
) )
func (h *handler) handleHTTPS(responseWriter http.ResponseWriter, request *http.Request) { func (h *handler) handleHTTPS(responseWriter http.ResponseWriter, request *http.Request) {
dialer := net.Dialer{Timeout: h.relayTimeout} dialer := net.Dialer{}
destinationConn, err := dialer.DialContext(h.ctx, "tcp", request.Host) destinationConn, err := dialer.DialContext(h.ctx, "tcp", request.Host)
if err != nil { if err != nil {
http.Error(responseWriter, err.Error(), http.StatusServiceUnavailable) http.Error(responseWriter, err.Error(), http.StatusServiceUnavailable)

View File

@@ -3,7 +3,6 @@ package httpproxy
import ( import (
"context" "context"
"fmt" "fmt"
"net/http"
"sync" "sync"
"github.com/qdm12/gluetun/internal/settings" "github.com/qdm12/gluetun/internal/settings"
@@ -20,7 +19,6 @@ type Looper interface {
} }
type looper struct { type looper struct {
client *http.Client
settings settings.HTTPProxy settings settings.HTTPProxy
settingsMutex sync.RWMutex settingsMutex sync.RWMutex
logger logging.Logger logger logging.Logger
@@ -29,10 +27,8 @@ type looper struct {
stop chan struct{} stop chan struct{}
} }
func NewLooper(client *http.Client, logger logging.Logger, func NewLooper(logger logging.Logger, settings settings.HTTPProxy) Looper {
settings settings.HTTPProxy) Looper {
return &looper{ return &looper{
client: client,
settings: settings, settings: settings,
logger: logger.WithPrefix("http proxy: "), logger: logger.WithPrefix("http proxy: "),
restart: make(chan struct{}), restart: make(chan struct{}),
@@ -104,7 +100,7 @@ func (l *looper) Run(ctx context.Context, wg *sync.WaitGroup) {
settings := l.GetSettings() settings := l.GetSettings()
address := fmt.Sprintf("0.0.0.0:%d", settings.Port) address := fmt.Sprintf("0.0.0.0:%d", settings.Port)
server := New(ctx, address, l.logger, l.client, settings.Stealth, settings.Log, settings.User, settings.Password) server := New(ctx, address, l.logger, settings.Stealth, settings.Log, settings.User, settings.Password)
runCtx, runCancel := context.WithCancel(context.Background()) runCtx, runCancel := context.WithCancel(context.Background())
runWg := &sync.WaitGroup{} runWg := &sync.WaitGroup{}

View File

@@ -20,13 +20,12 @@ type server struct {
internalWG *sync.WaitGroup internalWG *sync.WaitGroup
} }
func New(ctx context.Context, address string, func New(ctx context.Context, address string, logger logging.Logger,
logger logging.Logger, client *http.Client,
stealth, verbose bool, username, password string) Server { stealth, verbose bool, username, password string) Server {
wg := &sync.WaitGroup{} wg := &sync.WaitGroup{}
return &server{ return &server{
address: address, address: address,
handler: newHandler(ctx, wg, client, logger, stealth, verbose, username, password), handler: newHandler(ctx, wg, logger, stealth, verbose, username, password),
logger: logger, logger: logger,
internalWG: wg, internalWG: wg,
} }