2020-06-15 10:04:08 +00:00
|
|
|
package http_proxy
|
2020-04-08 22:59:22 +00:00
|
|
|
|
|
|
|
import (
|
2020-04-12 01:51:41 +00:00
|
|
|
"fmt"
|
2020-04-25 22:22:45 +00:00
|
|
|
"github.com/baez90/inetmock/pkg/api"
|
2020-04-27 22:26:15 +00:00
|
|
|
"github.com/baez90/inetmock/pkg/config"
|
2020-04-13 22:14:56 +00:00
|
|
|
"github.com/baez90/inetmock/pkg/logging"
|
2020-04-08 22:59:22 +00:00
|
|
|
"go.uber.org/zap"
|
2020-04-12 01:51:41 +00:00
|
|
|
"gopkg.in/elazarl/goproxy.v1"
|
|
|
|
"net/http"
|
2020-04-08 22:59:22 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
name = "http_proxy"
|
|
|
|
)
|
|
|
|
|
|
|
|
type httpProxy struct {
|
2020-04-13 22:14:56 +00:00
|
|
|
logger logging.Logger
|
2020-04-12 01:51:41 +00:00
|
|
|
proxy *goproxy.ProxyHttpServer
|
|
|
|
server *http.Server
|
2020-04-08 22:59:22 +00:00
|
|
|
}
|
|
|
|
|
2020-04-27 22:26:15 +00:00
|
|
|
func (h *httpProxy) Start(config config.HandlerConfig) (err error) {
|
|
|
|
options := loadFromConfig(config.Options)
|
|
|
|
addr := fmt.Sprintf("%s:%d", config.ListenAddress, config.Port)
|
2020-04-12 01:51:41 +00:00
|
|
|
h.server = &http.Server{Addr: addr, Handler: h.proxy}
|
|
|
|
h.logger = h.logger.With(
|
|
|
|
zap.String("address", addr),
|
|
|
|
)
|
|
|
|
|
2020-04-25 22:22:45 +00:00
|
|
|
tlsConfig := api.ServicesInstance().CertStore().TLSConfig()
|
|
|
|
|
2020-04-12 01:51:41 +00:00
|
|
|
proxyHandler := &proxyHttpHandler{
|
|
|
|
options: options,
|
|
|
|
logger: h.logger,
|
|
|
|
}
|
2020-04-25 22:22:45 +00:00
|
|
|
|
|
|
|
proxyHttpsHandler := &proxyHttpsHandler{
|
|
|
|
tlsConfig: tlsConfig,
|
|
|
|
logger: h.logger,
|
|
|
|
}
|
|
|
|
|
2020-04-12 01:51:41 +00:00
|
|
|
h.proxy.OnRequest().Do(proxyHandler)
|
2020-04-25 22:22:45 +00:00
|
|
|
h.proxy.OnRequest().HandleConnect(proxyHttpsHandler)
|
2020-04-12 01:51:41 +00:00
|
|
|
go h.startProxy()
|
2020-04-13 22:14:56 +00:00
|
|
|
return
|
2020-04-12 01:51:41 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpProxy) startProxy() {
|
|
|
|
if err := h.server.ListenAndServe(); err != nil {
|
|
|
|
h.logger.Error(
|
|
|
|
"failed to start proxy server",
|
|
|
|
zap.Error(err),
|
|
|
|
)
|
|
|
|
}
|
2020-04-08 22:59:22 +00:00
|
|
|
}
|
|
|
|
|
2020-04-13 22:14:56 +00:00
|
|
|
func (h *httpProxy) Shutdown() (err error) {
|
2020-04-12 01:51:41 +00:00
|
|
|
h.logger.Info("Shutting down HTTP proxy")
|
2020-04-13 22:14:56 +00:00
|
|
|
if err = h.server.Close(); err != nil {
|
2020-04-12 01:51:41 +00:00
|
|
|
h.logger.Error(
|
|
|
|
"failed to shutdown proxy endpoint",
|
|
|
|
zap.Error(err),
|
|
|
|
)
|
2020-04-13 22:14:56 +00:00
|
|
|
|
|
|
|
err = fmt.Errorf(
|
|
|
|
"failed to shutdown proxy endpoint: %w",
|
|
|
|
err,
|
|
|
|
)
|
2020-04-12 01:51:41 +00:00
|
|
|
}
|
2020-04-13 22:14:56 +00:00
|
|
|
return
|
2020-04-08 22:59:22 +00:00
|
|
|
}
|