|
4 | 4 | "bytes" |
5 | 5 | "context" |
6 | 6 | "crypto/tls" |
| 7 | + "crypto/x509" |
7 | 8 | "fmt" |
8 | 9 | "net/http" |
9 | 10 | "os" |
@@ -136,17 +137,181 @@ var _ = Describe("OCIUtils tests", func() { |
136 | 137 | Expect(repo.PlainHTTP).To(BeFalse()) |
137 | 138 | }) |
138 | 139 | It("return the expected tls client", func() { |
139 | | - client := getHTTPClient(true) |
140 | | - expected := &http.Client{ |
141 | | - Transport: &http.Transport{ |
142 | | - TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, |
143 | | - }, |
144 | | - } |
145 | | - Expect(client).To(Equal(expected)) |
| 140 | + client := getHTTPClient(true, nil) |
| 141 | + |
| 142 | + // Custom path wraps transport in retry.Transport |
| 143 | + retryTransport, ok := client.Transport.(*retry.Transport) |
| 144 | + Expect(ok).To(BeTrue()) |
| 145 | + innerTransport, ok := retryTransport.Base.(*http.Transport) |
| 146 | + Expect(ok).To(BeTrue()) |
| 147 | + Expect(innerTransport.TLSClientConfig).ToNot(BeNil()) |
| 148 | + Expect(innerTransport.TLSClientConfig.InsecureSkipVerify).To(BeTrue()) |
| 149 | + Expect(innerTransport.Proxy).ToNot(BeNil()) |
| 150 | + |
| 151 | + client = getHTTPClient(false, nil) |
| 152 | + Expect(client).To(Equal(retry.DefaultClient)) |
| 153 | + }) |
| 154 | + It("should use custom CA bundle when provided", func() { |
| 155 | + // Use a valid test certificate from the codebase pattern (same as netutils_test.go) |
| 156 | + caBundle := []byte(`-----BEGIN CERTIFICATE----- |
| 157 | +MIICGTCCAZ+gAwIBAgIQCeCTZaz32ci5PhwLBCou8zAKBggqhkjOPQQDAzBOMQsw |
| 158 | +CQYDVQQGEwJVUzEXMBUGA1UEChMORGlnaUNlcnQsIEluYy4xJjAkBgNVBAMTHURp |
| 159 | +Z2lDZXJ0IFRMUyBFQ0MgUDM4NCBSb290IEc1MB4XDTIxMDExNTAwMDAwMFoXDTQ2 |
| 160 | +MDExNDIzNTk1OVowTjELMAkGA1UEBhMCVVMxFzAVBgNVBAoTDkRpZ2lDZXJ0LCBJ |
| 161 | +bmMuMSYwJAYDVQQDEx1EaWdpQ2VydCBUTFMgRUNDIFAzODQgUm9vdCBHNTB2MBAG |
| 162 | +ByqGSM49AgEGBSuBBAAiA2IABMFEoc8Rl1Ca3iOCNQfN0MsYndLxf3c1TzvdlHJS |
| 163 | +7cI7+Oz6e2tYIOyZrsn8aLN1udsJ7MgT9U7GCh1mMEy7H0cKPGEQQil8pQgO4CLp |
| 164 | +0zVozptjn4S1mU1YoI71VOeVyaNCMEAwHQYDVR0OBBYEFMFRRVBZqz7nLFr6ICIS |
| 165 | +B4CIfBFqMA4GA1UdDwEB/wQEAwIBhjAPBgNVHRMBAf8EBTADAQH/MAoGCCqGSM49 |
| 166 | +BAMDA2gAMGUCMQCJao1H5+z8blUD2WdsJk6Dxv3J+ysTvLd6jLRl0mlpYxNjOyZQ |
| 167 | +LgGheQaRnUi/wr4CMEfDFXuxoJGZSZOoPHzoRgaLLPIxAJSdYsiJvRmEFOml+wG4 |
| 168 | +DXZDjC5Ty3zfDBeWUA== |
| 169 | +-----END CERTIFICATE-----`) |
| 170 | + client := getHTTPClient(false, caBundle) |
| 171 | + |
| 172 | + // Verify transport is configured with custom TLS config |
| 173 | + retryT, ok := client.Transport.(*retry.Transport) |
| 174 | + Expect(ok).To(BeTrue()) |
| 175 | + transport, ok := retryT.Base.(*http.Transport) |
| 176 | + Expect(ok).To(BeTrue()) |
| 177 | + Expect(transport.TLSClientConfig).ToNot(BeNil()) |
| 178 | + Expect(transport.TLSClientConfig.RootCAs).ToNot(BeNil()) |
| 179 | + Expect(transport.TLSClientConfig.MinVersion).To(Equal(uint16(tls.VersionTLS12))) |
| 180 | + Expect(transport.TLSClientConfig.InsecureSkipVerify).To(BeFalse()) |
| 181 | + }) |
| 182 | + It("should merge proxy CA bundle from environment variable", func() { |
| 183 | + // Use a valid certificate for proxy CA (same as custom test) |
| 184 | + proxyCA := `-----BEGIN CERTIFICATE----- |
| 185 | +MIICGTCCAZ+gAwIBAgIQCeCTZaz32ci5PhwLBCou8zAKBggqhkjOPQQDAzBOMQsw |
| 186 | +CQYDVQQGEwJVUzEXMBUGA1UEChMORGlnaUNlcnQsIEluYy4xJjAkBgNVBAMTHURp |
| 187 | +Z2lDZXJ0IFRMUyBFQ0MgUDM4NCBSb290IEc1MB4XDTIxMDExNTAwMDAwMFoXDTQ2 |
| 188 | +MDExNDIzNTk1OVowTjELMAkGA1UEBhMCVVMxFzAVBgNVBAoTDkRpZ2lDZXJ0LCBJ |
| 189 | +bmMuMSYwJAYDVQQDEx1EaWdpQ2VydCBUTFMgRUNDIFAzODQgUm9vdCBHNTB2MBAG |
| 190 | +ByqGSM49AgEGBSuBBAAiA2IABMFEoc8Rl1Ca3iOCNQfN0MsYndLxf3c1TzvdlHJS |
| 191 | +7cI7+Oz6e2tYIOyZrsn8aLN1udsJ7MgT9U7GCh1mMEy7H0cKPGEQQil8pQgO4CLp |
| 192 | +0zVozptjn4S1mU1YoI71VOeVyaNCMEAwHQYDVR0OBBYEFMFRRVBZqz7nLFr6ICIS |
| 193 | +B4CIfBFqMA4GA1UdDwEB/wQEAwIBhjAPBgNVHRMBAf8EBTADAQH/MAoGCCqGSM49 |
| 194 | +BAMDA2gAMGUCMQCJao1H5+z8blUD2WdsJk6Dxv3J+ysTvLd6jLRl0mlpYxNjOyZQ |
| 195 | +LgGheQaRnUi/wr4CMEfDFXuxoJGZSZOoPHzoRgaLLPIxAJSdYsiJvRmEFOml+wG4 |
| 196 | +DXZDjC5Ty3zfDBeWUA== |
| 197 | +-----END CERTIFICATE-----` |
| 198 | + os.Setenv("PROXY_CA_BUNDLE", proxyCA) |
| 199 | + defer os.Unsetenv("PROXY_CA_BUNDLE") |
| 200 | + |
| 201 | + // Use the same valid certificate for custom CA (simpler for testing) |
| 202 | + customCA := []byte(proxyCA) |
| 203 | + client := getHTTPClient(false, customCA) |
| 204 | + |
| 205 | + // Should have merged both CAs |
| 206 | + retryT, ok := client.Transport.(*retry.Transport) |
| 207 | + Expect(ok).To(BeTrue()) |
| 208 | + transport, ok := retryT.Base.(*http.Transport) |
| 209 | + Expect(ok).To(BeTrue()) |
| 210 | + Expect(transport.TLSClientConfig).ToNot(BeNil()) |
| 211 | + Expect(transport.TLSClientConfig.RootCAs).ToNot(BeNil()) |
| 212 | + Expect(transport.TLSClientConfig.MinVersion).To(Equal(uint16(tls.VersionTLS12))) |
| 213 | + |
| 214 | + // Verify both CAs are valid and would have been loaded |
| 215 | + testPool := x509.NewCertPool() |
| 216 | + ok = testPool.AppendCertsFromPEM(customCA) |
| 217 | + Expect(ok).To(BeTrue(), "custom CA bundle should be valid PEM") |
| 218 | + ok = testPool.AppendCertsFromPEM([]byte(proxyCA)) |
| 219 | + Expect(ok).To(BeTrue(), "proxy CA bundle should be valid PEM") |
| 220 | + }) |
| 221 | + It("should handle invalid CA bundle gracefully", func() { |
| 222 | + invalidCA := []byte("not a valid PEM certificate") |
| 223 | + client := getHTTPClient(false, invalidCA) |
| 224 | + |
| 225 | + // Should still create a client with TLS config (warning logged) |
| 226 | + retryT, ok := client.Transport.(*retry.Transport) |
| 227 | + Expect(ok).To(BeTrue()) |
| 228 | + transport, ok := retryT.Base.(*http.Transport) |
| 229 | + Expect(ok).To(BeTrue()) |
| 230 | + Expect(transport.TLSClientConfig).ToNot(BeNil()) |
| 231 | + Expect(transport.TLSClientConfig.RootCAs).ToNot(BeNil()) |
| 232 | + }) |
| 233 | + It("should handle invalid proxy CA bundle gracefully", func() { |
| 234 | + os.Setenv("PROXY_CA_BUNDLE", "invalid proxy PEM") |
| 235 | + defer os.Unsetenv("PROXY_CA_BUNDLE") |
| 236 | + |
| 237 | + client := getHTTPClient(false, nil) |
146 | 238 |
|
147 | | - client = getHTTPClient(false) |
| 239 | + // Should return default client when no valid CA and not insecure |
148 | 240 | Expect(client).To(Equal(retry.DefaultClient)) |
149 | 241 | }) |
| 242 | + It("should combine insecureSkipTLS with CA bundle", func() { |
| 243 | + caBundle := []byte(`-----BEGIN CERTIFICATE----- |
| 244 | +MIICGTCCAZ+gAwIBAgIQCeCTZaz32ci5PhwLBCou8zAKBggqhkjOPQQDAzBOMQsw |
| 245 | +CQYDVQQGEwJVUzEXMBUGA1UEChMORGlnaUNlcnQsIEluYy4xJjAkBgNVBAMTHURp |
| 246 | +Z2lDZXJ0IFRMUyBFQ0MgUDM4NCBSb290IEc1MB4XDTIxMDExNTAwMDAwMFoXDTQ2 |
| 247 | +MDExNDIzNTk1OVowTjELMAkGA1UEBhMCVVMxFzAVBgNVBAoTDkRpZ2lDZXJ0LCBJ |
| 248 | +bmMuMSYwJAYDVQQDEx1EaWdpQ2VydCBUTFMgRUNDIFAzODQgUm9vdCBHNTB2MBAG |
| 249 | +ByqGSM49AgEGBSuBBAAiA2IABMFEoc8Rl1Ca3iOCNQfN0MsYndLxf3c1TzvdlHJS |
| 250 | +7cI7+Oz6e2tYIOyZrsn8aLN1udsJ7MgT9U7GCh1mMEy7H0cKPGEQQil8pQgO4CLp |
| 251 | +0zVozptjn4S1mU1YoI71VOeVyaNCMEAwHQYDVR0OBBYEFMFRRVBZqz7nLFr6ICIS |
| 252 | +B4CIfBFqMA4GA1UdDwEB/wQEAwIBhjAPBgNVHRMBAf8EBTADAQH/MAoGCCqGSM49 |
| 253 | +BAMDA2gAMGUCMQCJao1H5+z8blUD2WdsJk6Dxv3J+ysTvLd6jLRl0mlpYxNjOyZQ |
| 254 | +LgGheQaRnUi/wr4CMEfDFXuxoJGZSZOoPHzoRgaLLPIxAJSdYsiJvRmEFOml+wG4 |
| 255 | +DXZDjC5Ty3zfDBeWUA== |
| 256 | +-----END CERTIFICATE-----`) |
| 257 | + client := getHTTPClient(true, caBundle) |
| 258 | + |
| 259 | + retryT, ok := client.Transport.(*retry.Transport) |
| 260 | + Expect(ok).To(BeTrue()) |
| 261 | + transport, ok := retryT.Base.(*http.Transport) |
| 262 | + Expect(ok).To(BeTrue()) |
| 263 | + Expect(transport.TLSClientConfig).ToNot(BeNil()) |
| 264 | + Expect(transport.TLSClientConfig.InsecureSkipVerify).To(BeTrue()) |
| 265 | + Expect(transport.TLSClientConfig.RootCAs).ToNot(BeNil()) |
| 266 | + Expect(transport.TLSClientConfig.MinVersion).To(Equal(uint16(tls.VersionTLS12))) |
| 267 | + |
| 268 | + // Verify the CA bundle is valid PEM (even though InsecureSkipVerify is set) |
| 269 | + testPool := x509.NewCertPool() |
| 270 | + ok = testPool.AppendCertsFromPEM(caBundle) |
| 271 | + Expect(ok).To(BeTrue(), "CA bundle should be valid PEM") |
| 272 | + }) |
| 273 | + It("should not mutate the original CA bundle slice", func() { |
| 274 | + // validPEM is a real certificate so that AppendCertsFromPEM succeeds and |
| 275 | + // getHTTPClient actually reaches the append(caBundle, proxyBytes...) path. |
| 276 | + // An invalid PEM would make the function return early without appending, |
| 277 | + // meaning the defensive copy would never be exercised. |
| 278 | + validPEM := []byte(`-----BEGIN CERTIFICATE----- |
| 279 | +MIICGTCCAZ+gAwIBAgIQCeCTZaz32ci5PhwLBCou8zAKBggqhkjOPQQDAzBOMQsw |
| 280 | +CQYDVQQGEwJVUzEXMBUGA1UEChMORGlnaUNlcnQsIEluYy4xJjAkBgNVBAMTHURp |
| 281 | +Z2lDZXJ0IFRMUyBFQ0MgUDM4NCBSb290IEc1MB4XDTIxMDExNTAwMDAwMFoXDTQ2 |
| 282 | +MDExNDIzNTk1OVowTjELMAkGA1UEBhMCVVMxFzAVBgNVBAoTDkRpZ2lDZXJ0LCBJ |
| 283 | +bmMuMSYwJAYDVQQDEx1EaWdpQ2VydCBUTFMgRUNDIFAzODQgUm9vdCBHNTB2MBAG |
| 284 | +ByqGSM49AgEGBSuBBAAiA2IABMFEoc8Rl1Ca3iOCNQfN0MsYndLxf3c1TzvdlHJS |
| 285 | +7cI7+Oz6e2tYIOyZrsn8aLN1udsJ7MgT9U7GCh1mMEy7H0cKPGEQQil8pQgO4CLp |
| 286 | +0zVozptjn4S1mU1YoI71VOeVyaNCMEAwHQYDVR0OBBYEFMFRRVBZqz7nLFr6ICIS |
| 287 | +B4CIfBFqMA4GA1UdDwEB/wQEAwIBhjAPBgNVHRMBAf8EBTADAQH/MAoGCCqGSM49 |
| 288 | +BAMDA2gAMGUCMQCJao1H5+z8blUD2WdsJk6Dxv3J+ysTvLd6jLRl0mlpYxNjOyZQ |
| 289 | +LgGheQaRnUi/wr4CMEfDFXuxoJGZSZOoPHzoRgaLLPIxAJSdYsiJvRmEFOml+wG4 |
| 290 | +DXZDjC5Ty3zfDBeWUA== |
| 291 | +-----END CERTIFICATE-----`) |
| 292 | + |
| 293 | + os.Setenv("PROXY_CA_BUNDLE", string(validPEM)) |
| 294 | + defer os.Unsetenv("PROXY_CA_BUNDLE") |
| 295 | + |
| 296 | + // Allocate with excess capacity so that append can write into the backing |
| 297 | + // array without reallocating. With len == cap (e.g. []byte("literal")), |
| 298 | + // append always reallocates and the defensive copy makes no observable |
| 299 | + // difference — the test would pass even without it. |
| 300 | + originalCA := make([]byte, len(validPEM), len(validPEM)+512) |
| 301 | + copy(originalCA, validPEM) |
| 302 | + |
| 303 | + // backing covers the full allocation so we can detect writes past len(originalCA). |
| 304 | + backing := originalCA[:cap(originalCA)] |
| 305 | + |
| 306 | + _ = getHTTPClient(false, originalCA) |
| 307 | + |
| 308 | + // The slice header and visible content must be unchanged. |
| 309 | + Expect(originalCA).To(Equal(validPEM)) |
| 310 | + // Without the defensive copy, getHTTPClient would append '\n'+proxyBytes |
| 311 | + // directly into the excess capacity of the caller's backing array. |
| 312 | + Expect(backing[len(validPEM):]).To(Equal(make([]byte, 512)), |
| 313 | + "backing array beyond len(originalCA) must not be written to") |
| 314 | + }) |
150 | 315 | It("return the expected credentials", func() { |
151 | 316 | opts := OCIOpts{ |
152 | 317 | Reference: "test.com", |
|
0 commit comments