fix(UNetStack): expose the TLSMITMConfig

Also part of ooni/probe#1803
ooni · Sep 6, 2023 · 85d9625 · 85d9625
1 parent c91fe50
commit 85d9625
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/tlsmitm_test.go b/tlsmitm_test.go
@@ -48,7 +48,7 @@ func TestMITMWeCanGenerateAnExpiredCertificate(t *testing.T) {
 	defer tcpConn.Close()
 
 	tlsClientConfig := &tls.Config{
-		RootCAs:    Must1(topology.TLSMITMConfig().CertPool()),
+		RootCAs:    Must1(clientStack.TLSMITMConfig().CertPool()),
 		ServerName: "www.example.com",
 	}
 	tlsConn := tls.Client(tcpConn, tlsClientConfig)

diff --git a/unetstack.go b/unetstack.go
@@ -104,7 +104,12 @@ func NewUNetStack(
 	return stack, nil
 }
 
-// Logger implements HTTPUnderlyingNetwork
+// TLSMITMConfig exposes the underlying [TLSMITMConfig].
+func (gs *UNetStack) TLSMITMConfig() *TLSMITMConfig {
+	return gs.mitmConfig
+}
+
+// Logger implements HTTPUnderlyingNetwork.
 func (gs *UNetStack) Logger() Logger {
 	return gs.ns.logger
 }