fix constructors

golang-fips · Jan 7, 2025 · 1592b8e · 1592b8e
1 parent a44d077
commit 1592b8e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cshake.go b/cshake.go
@@ -113,6 +113,9 @@ func NewSHAKE256() *SHAKE {
 // cSHAKE is desired. S is a customization byte string used for domain
 // separation. When N and S are both empty, this is equivalent to NewSHAKE128.
 func NewCSHAKE128(N, S []byte) *SHAKE {
+	if len(N) == 0 && len(S) == 0 {
+		return NewSHAKE128()
+	}
 	return nil
 }
 
@@ -122,6 +125,9 @@ func NewCSHAKE128(N, S []byte) *SHAKE {
 // cSHAKE is desired. S is a customization byte string used for domain
 // separation. When N and S are both empty, this is equivalent to NewSHAKE256.
 func NewCSHAKE256(N, S []byte) *SHAKE {
+	if len(N) == 0 && len(S) == 0 {
+		return NewSHAKE256()
+	}
 	return nil
 }