move the certPool loader function into pkg/certs (#10239)
parent
f8f290e848
commit
0dd3a08169
@ -0,0 +1,57 @@ |
|||||||
|
/* |
||||||
|
* MinIO Cloud Storage, (C) 2020 MinIO, Inc. |
||||||
|
* |
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||||
|
* you may not use this file except in compliance with the License. |
||||||
|
* You may obtain a copy of the License at |
||||||
|
* |
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* |
||||||
|
* Unless required by applicable law or agreed to in writing, software |
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||||
|
* See the License for the specific language governing permissions and |
||||||
|
* limitations under the License. |
||||||
|
*/ |
||||||
|
|
||||||
|
package certs |
||||||
|
|
||||||
|
import ( |
||||||
|
"crypto/x509" |
||||||
|
"io/ioutil" |
||||||
|
"os" |
||||||
|
"path" |
||||||
|
) |
||||||
|
|
||||||
|
// GetRootCAs - returns all the root CAs into certPool
|
||||||
|
// at the input certsCADir
|
||||||
|
func GetRootCAs(certsCAsDir string) (*x509.CertPool, error) { |
||||||
|
rootCAs, _ := x509.SystemCertPool() |
||||||
|
if rootCAs == nil { |
||||||
|
// In some systems (like Windows) system cert pool is
|
||||||
|
// not supported or no certificates are present on the
|
||||||
|
// system - so we create a new cert pool.
|
||||||
|
rootCAs = x509.NewCertPool() |
||||||
|
} |
||||||
|
|
||||||
|
fis, err := ioutil.ReadDir(certsCAsDir) |
||||||
|
if err != nil { |
||||||
|
if os.IsNotExist(err) || os.IsPermission(err) { |
||||||
|
// Return success if CA's directory is missing or permission denied.
|
||||||
|
err = nil |
||||||
|
} |
||||||
|
return rootCAs, err |
||||||
|
} |
||||||
|
|
||||||
|
// Load all custom CA files.
|
||||||
|
for _, fi := range fis { |
||||||
|
caCert, err := ioutil.ReadFile(path.Join(certsCAsDir, fi.Name())) |
||||||
|
if err != nil { |
||||||
|
// ignore files which are not readable.
|
||||||
|
continue |
||||||
|
} |
||||||
|
rootCAs.AppendCertsFromPEM(caCert) |
||||||
|
} |
||||||
|
|
||||||
|
return rootCAs, nil |
||||||
|
} |
@ -0,0 +1,78 @@ |
|||||||
|
/* |
||||||
|
* MinIO Cloud Storage, (C) 2020 MinIO, Inc. |
||||||
|
* |
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||||
|
* you may not use this file except in compliance with the License. |
||||||
|
* You may obtain a copy of the License at |
||||||
|
* |
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* |
||||||
|
* Unless required by applicable law or agreed to in writing, software |
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||||
|
* See the License for the specific language governing permissions and |
||||||
|
* limitations under the License. |
||||||
|
*/ |
||||||
|
|
||||||
|
package certs |
||||||
|
|
||||||
|
import ( |
||||||
|
"io/ioutil" |
||||||
|
"os" |
||||||
|
"path/filepath" |
||||||
|
"testing" |
||||||
|
) |
||||||
|
|
||||||
|
func TestGetRootCAs(t *testing.T) { |
||||||
|
emptydir, err := ioutil.TempDir("", "test-get-root-cas") |
||||||
|
if err != nil { |
||||||
|
t.Fatalf("Unable create temp directory. %v", emptydir) |
||||||
|
} |
||||||
|
defer os.RemoveAll(emptydir) |
||||||
|
|
||||||
|
dir1, err := ioutil.TempDir("", "test-get-root-cas") |
||||||
|
if err != nil { |
||||||
|
t.Fatalf("Unable create temp directory. %v", dir1) |
||||||
|
} |
||||||
|
defer os.RemoveAll(dir1) |
||||||
|
if err = os.Mkdir(filepath.Join(dir1, "empty-dir"), 0755); err != nil { |
||||||
|
t.Fatalf("Unable create empty dir. %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
dir2, err := ioutil.TempDir("", "test-get-root-cas") |
||||||
|
if err != nil { |
||||||
|
t.Fatalf("Unable create temp directory. %v", dir2) |
||||||
|
} |
||||||
|
defer os.RemoveAll(dir2) |
||||||
|
if err = ioutil.WriteFile(filepath.Join(dir2, "empty-file"), []byte{}, 0644); err != nil { |
||||||
|
t.Fatalf("Unable create test file. %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
testCases := []struct { |
||||||
|
certCAsDir string |
||||||
|
expectedErr error |
||||||
|
}{ |
||||||
|
// ignores non-existent directories.
|
||||||
|
{"nonexistent-dir", nil}, |
||||||
|
// Ignores directories.
|
||||||
|
{dir1, nil}, |
||||||
|
// Ignore empty directory.
|
||||||
|
{emptydir, nil}, |
||||||
|
// Loads the cert properly.
|
||||||
|
{dir2, nil}, |
||||||
|
} |
||||||
|
|
||||||
|
for _, testCase := range testCases { |
||||||
|
_, err := GetRootCAs(testCase.certCAsDir) |
||||||
|
|
||||||
|
if testCase.expectedErr == nil { |
||||||
|
if err != nil { |
||||||
|
t.Fatalf("error: expected = <nil>, got = %v", err) |
||||||
|
} |
||||||
|
} else if err == nil { |
||||||
|
t.Fatalf("error: expected = %v, got = <nil>", testCase.expectedErr) |
||||||
|
} else if testCase.expectedErr.Error() != err.Error() { |
||||||
|
t.Fatalf("error: expected = %v, got = %v", testCase.expectedErr, err) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue