keyusage.Equals

keyusage/keyusage.go

@@ -4,6 +4,7 @@ import (
 	"crypto/x509"
 	"errors"
 	"fmt"
+	"sort"
 
 	"github.com/IPA-CyberLab/kmgm/pb"
 )
@@ -160,3 +161,24 @@ func (p *KeyUsage) UnmarshalFlag(s string) error {
 	*p = ku
 	return nil
 }
+
+func (a KeyUsage) Equals(b KeyUsage) bool {
+	if a.KeyUsage != b.KeyUsage {
+		return false
+	}
+	if len(a.ExtKeyUsages) != len(b.ExtKeyUsages) {
+		return false
+	}
+
+	ekua := append([]x509.ExtKeyUsage{}, a.ExtKeyUsages...)
+	ekub := append([]x509.ExtKeyUsage{}, b.ExtKeyUsages...)
+	sort.Slice(ekua, func(i, j int) bool { return ekua[i] < ekua[j] })
+	sort.Slice(ekub, func(i, j int) bool { return ekub[i] < ekub[j] })
+	for i := range ekua {
+		if ekua[i] != ekub[i] {
+			return false
+		}
+	}
+
+	return true
+}

keyusage/keyusage_test.go

@@ -0,0 +1,25 @@
+package keyusage_test
+
+import (
+	"testing"
+
+	"github.com/IPA-CyberLab/kmgm/keyusage"
+)
+
+func TestKeyUsage_Equals(t *testing.T) {
+	if keyusage.KeyUsageCA.Equals(keyusage.KeyUsageTLSServer) {
+		t.Errorf("Unexpected: CA == TlsServer")
+	}
+	if keyusage.KeyUsageTLSClientServer.Equals(keyusage.KeyUsageTLSServer) {
+		t.Errorf("Unexpected: cs == s")
+	}
+	if keyusage.KeyUsageTLSClient.Equals(keyusage.KeyUsageTLSServer) {
+		t.Errorf("Unexpected: c == s")
+	}
+	if !keyusage.KeyUsageCA.Equals(keyusage.KeyUsageCA) {
+		t.Errorf("Unexpected: CA != CA")
+	}
+	if !keyusage.KeyUsageTLSClientServer.Equals(keyusage.KeyUsageTLSClientServer) {
+		t.Errorf("Unexpected: cs != cs")
+	}
+}