Peter Kurfer
7c2a41ad25
- apply changes in proxy plugin and TLS interceptor - add HTTPS proxy support - move ca-generation command to main app - minor refactoring to improve API stability - move mocks to extra packages to avoid cycling imports - fix bug in multi-port configuration - change HTTP proxy to redirect to HTTP mock instead of maintaining custom rules
93 lines
2.6 KiB
Go
93 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"github.com/baez90/inetmock/pkg/logging"
|
|
"go.uber.org/zap"
|
|
"gopkg.in/elazarl/goproxy.v1"
|
|
"net/http"
|
|
"net/url"
|
|
)
|
|
|
|
type proxyHttpHandler struct {
|
|
options httpProxyOptions
|
|
logger logging.Logger
|
|
}
|
|
|
|
type proxyHttpsHandler struct {
|
|
tlsConfig *tls.Config
|
|
logger logging.Logger
|
|
}
|
|
|
|
func (p *proxyHttpsHandler) HandleConnect(req string, _ *goproxy.ProxyCtx) (*goproxy.ConnectAction, string) {
|
|
p.logger.Info(
|
|
"Intercepting HTTPS proxy request",
|
|
zap.String("request", req),
|
|
)
|
|
|
|
return &goproxy.ConnectAction{
|
|
Action: goproxy.ConnectMitm,
|
|
TLSConfig: func(host string, ctx *goproxy.ProxyCtx) (*tls.Config, error) {
|
|
return p.tlsConfig, nil
|
|
},
|
|
}, ""
|
|
}
|
|
|
|
func (p *proxyHttpHandler) Handle(req *http.Request, ctx *goproxy.ProxyCtx) (retReq *http.Request, resp *http.Response) {
|
|
retReq = req
|
|
p.logger.Info(
|
|
"Handling request",
|
|
zap.String("source", req.RemoteAddr),
|
|
zap.String("host", req.Host),
|
|
zap.String("method", req.Method),
|
|
zap.String("protocol", req.Proto),
|
|
zap.String("path", req.RequestURI),
|
|
zap.Reflect("headers", req.Header),
|
|
)
|
|
|
|
var err error
|
|
if resp, err = ctx.RoundTrip(p.redirectHTTPRequest(req)); err != nil {
|
|
p.logger.Error(
|
|
"error while doing roundtrip",
|
|
zap.Error(err),
|
|
)
|
|
return req, nil
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (p proxyHttpHandler) redirectHTTPRequest(originalRequest *http.Request) (redirectReq *http.Request) {
|
|
redirectReq = &http.Request{
|
|
Method: originalRequest.Method,
|
|
URL: &url.URL{
|
|
Host: p.options.redirectionTarget.host(),
|
|
Scheme: p.options.redirectionTarget.scheme,
|
|
Path: originalRequest.URL.Path,
|
|
ForceQuery: originalRequest.URL.ForceQuery,
|
|
Fragment: originalRequest.URL.Fragment,
|
|
Opaque: originalRequest.URL.Opaque,
|
|
RawPath: originalRequest.URL.RawPath,
|
|
RawQuery: originalRequest.URL.RawQuery,
|
|
User: originalRequest.URL.User,
|
|
},
|
|
Proto: originalRequest.Proto,
|
|
ProtoMajor: originalRequest.ProtoMajor,
|
|
ProtoMinor: originalRequest.ProtoMinor,
|
|
Header: originalRequest.Header,
|
|
Body: originalRequest.Body,
|
|
GetBody: originalRequest.GetBody,
|
|
ContentLength: originalRequest.ContentLength,
|
|
TransferEncoding: originalRequest.TransferEncoding,
|
|
Close: false,
|
|
Host: originalRequest.Host,
|
|
Form: originalRequest.Form,
|
|
PostForm: originalRequest.PostForm,
|
|
MultipartForm: originalRequest.MultipartForm,
|
|
Trailer: originalRequest.Trailer,
|
|
}
|
|
redirectReq = redirectReq.WithContext(context.Background())
|
|
|
|
return
|
|
}
|