Skip to content

Commit dc68681

Browse files
committed
Small refactoring in cert building subroutines
1 parent 10aa952 commit dc68681

File tree

2 files changed

+7
-18
lines changed

2 files changed

+7
-18
lines changed

Diff for: flasher/nina.go

+2-9
Original file line numberDiff line numberDiff line change
@@ -114,10 +114,11 @@ func (f *NinaFlasher) FlashCertificates(certificatePaths *paths.PathList, URLs [
114114
for _, URL := range URLs {
115115
logrus.Infof("Converting and flashing certificate from %s", URL)
116116
flasherOut.Write([]byte(fmt.Sprintf("Converting and flashing certificate from %s\n", URL)))
117-
data, err := f.certificateFromURL(URL)
117+
rootCertificate, err := certificates.ScrapeRootCertificatesFromURL(URL)
118118
if err != nil {
119119
return err
120120
}
121+
data := pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: rootCertificate.Raw})
121122
certificatesData = append(certificatesData, data...)
122123
}
123124

@@ -143,14 +144,6 @@ func (f *NinaFlasher) FlashCertificates(certificatePaths *paths.PathList, URLs [
143144
return nil
144145
}
145146

146-
func (f *NinaFlasher) certificateFromURL(URL string) ([]byte, error) {
147-
rootCertificate, err := certificates.ScrapeRootCertificatesFromURL(URL)
148-
if err != nil {
149-
return nil, err
150-
}
151-
return pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: rootCertificate.Raw}), nil
152-
}
153-
154147
// Close the port used by this flasher
155148
func (f *NinaFlasher) Close() error {
156149
return f.port.Close()

Diff for: flasher/winc.go

+5-9
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,11 @@ func (f *WincFlasher) FlashCertificates(certificatePaths *paths.PathList, URLs [
9999
for _, URL := range URLs {
100100
logrus.Infof("Converting and flashing certificate from %s", URL)
101101
flasherOut.Write([]byte(fmt.Sprintf("Converting and flashing certificate from %s\n", URL)))
102-
data, err := f.certificateFromURL(URL)
102+
rootCertificate, err := certificates.ScrapeRootCertificatesFromURL(URL)
103+
if err != nil {
104+
return err
105+
}
106+
data, err := f.getCertificateData(rootCertificate)
103107
if err != nil {
104108
return err
105109
}
@@ -116,14 +120,6 @@ func (f *WincFlasher) FlashCertificates(certificatePaths *paths.PathList, URLs [
116120
return nil
117121
}
118122

119-
func (f *WincFlasher) certificateFromURL(URL string) ([]byte, error) {
120-
rootCertificate, err := certificates.ScrapeRootCertificatesFromURL(URL)
121-
if err != nil {
122-
return nil, err
123-
}
124-
return f.getCertificateData(rootCertificate)
125-
}
126-
127123
func (f *WincFlasher) getCertificateData(cert *x509.Certificate) ([]byte, error) {
128124
b := []byte{}
129125
nameSHA1Bytes, err := calculateNameSha1(cert)

0 commit comments

Comments
 (0)