Add OIDs to ML-DSA

cloudflare · Oct 16, 2024 · 2635e5c · 2635e5c
1 parent 1310edf
commit 2635e5c
Show file tree

Hide file tree

Showing 8 changed files with 47 additions and 7 deletions.
diff --git a/sign/dilithium/gen.go b/sign/dilithium/gen.go
@@ -7,6 +7,7 @@ package main
 
 import (
 	"bytes"
+	"encoding/asn1"
 	"fmt"
 	"go/format"
 	"os"
@@ -29,6 +30,7 @@ type Mode struct {
 	Gamma2        int
 	TRSize        int
 	CTildeSize    int
+	Oid           asn1.ObjectIdentifier
 }
 
 func (m Mode) Pkg() string {
@@ -59,6 +61,20 @@ func (m Mode) NIST() bool {
 	return strings.HasPrefix(m.Name, "ML-DSA-")
 }
 
+func (m Mode) OidGo() string {
+	ret := "asn1.ObjectIdentifier{"
+	first := true
+	for _, b := range m.Oid {
+		if first {
+			first = false
+		} else {
+			ret += ", "
+		}
+		ret += fmt.Sprintf("%d", b)
+	}
+	return ret + "}"
+}
+
 var (
 	Modes = []Mode{
 		{
@@ -112,6 +128,7 @@ var (
 			Gamma2:        (params.Q - 1) / 88,
 			TRSize:        64,
 			CTildeSize:    32,
+			Oid:           asn1.ObjectIdentifier{2, 16, 840, 1, 101, 3, 4, 17},
 		},
 		{
 			Name:          "ML-DSA-65",
@@ -125,6 +142,7 @@ var (
 			Gamma2:        (params.Q - 1) / 32,
 			TRSize:        64,
 			CTildeSize:    48,
+			Oid:           asn1.ObjectIdentifier{2, 16, 840, 1, 101, 3, 4, 18},
 		},
 		{
 			Name:          "ML-DSA-87",
@@ -138,6 +156,7 @@ var (
 			Gamma2:        (params.Q - 1) / 32,
 			TRSize:        64,
 			CTildeSize:    64,
+			Oid:           asn1.ObjectIdentifier{2, 16, 840, 1, 101, 3, 4, 19},
 		},
 	}
 	TemplateWarning = "// Code generated from"

diff --git a/sign/dilithium/mode2/dilithium.go b/sign/dilithium/mode2/dilithium.go
diff --git a/sign/dilithium/mode3/dilithium.go b/sign/dilithium/mode3/dilithium.go
diff --git a/sign/dilithium/mode5/dilithium.go b/sign/dilithium/mode5/dilithium.go
diff --git a/sign/dilithium/templates/pkg.templ.go b/sign/dilithium/templates/pkg.templ.go
diff --git a/sign/mldsa/mldsa44/dilithium.go b/sign/mldsa/mldsa44/dilithium.go
diff --git a/sign/mldsa/mldsa65/dilithium.go b/sign/mldsa/mldsa65/dilithium.go
diff --git a/sign/mldsa/mldsa87/dilithium.go b/sign/mldsa/mldsa87/dilithium.go