http_proxy: enable getting MITM CA cert

saucelabs · Sep 27, 2023 · 8442986 · 8442986
1 parent d6ea3fc
commit 8442986
Showing 1 changed file with 15 additions and 8 deletions.
diff --git a/http_proxy.go b/http_proxy.go
@@ -10,6 +10,7 @@ import (
 	"bufio"
 	"context"
 	"crypto/tls"
+	"crypto/x509"
 	"errors"
 	"fmt"
 	"io"
@@ -128,14 +129,15 @@ func (c *HTTPProxyConfig) Validate() error {
 }
 
 type HTTPProxy struct {
-	config    HTTPProxyConfig
-	pac       PACResolver
-	creds     *CredentialsMatcher
-	transport http.RoundTripper
-	log       log.Logger
-	proxy     *martian.Proxy
-	proxyFunc ProxyFunc
-	listener  net.Listener
+	config     HTTPProxyConfig
+	pac        PACResolver
+	creds      *CredentialsMatcher
+	transport  http.RoundTripper
+	log        log.Logger
+	proxy      *martian.Proxy
+	mitmCACert *x509.Certificate
+	proxyFunc  ProxyFunc
+	listener   net.Listener
 
 	TLSConfig *tls.Config
 }
@@ -229,6 +231,7 @@ func (hp *HTTPProxy) configureProxy() error {
 			return fmt.Errorf("mitm: %w", err)
 		}
 		hp.proxy.SetMITM(mc)
+		hp.mitmCACert = mc.CACert()
 
 		if hp.config.MITMDomains != nil {
 			hp.proxy.MITMFilter = func(req *http.Request) bool {
@@ -527,6 +530,10 @@ func setEmptyUserAgent(req *http.Request) error {
 	return nil
 }
 
+func (hp *HTTPProxy) MITMCACert() *x509.Certificate {
+	return hp.mitmCACert
+}
+
 func (hp *HTTPProxy) ProxyFunc() ProxyFunc {
 	return hp.proxyFunc
 }