Merge pull request #1 from mlindgren/mitchl/fix_fips204_to_bytes

Fix to_bytes in fips204.py (compatibility with Python < 3.11)
mjosaarinen · Sep 13, 2024 · 0382529 · 0382529
2 parents 9eddd2e + 424c6ec
commit 0382529
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fips204.py b/fips204.py
@@ -174,7 +174,7 @@ def keygen_internal(self, xi, param=None):
             self.__init__(param)
         # print('# keygen_internal()', param)
         # print('# seed:', xi.hex())
-        se = self.h(xi + self.k.to_bytes() + self.ell.to_bytes(), 128)
+        se = self.h(xi + self.integer_to_bytes(self.k, 1) + self.integer_to_bytes(self.ell, 1), 128)
         rho = se[0:32]
         rhop = se[32:96]
         kk = se[96:128]
@@ -659,7 +659,7 @@ def expand_a(self, rho):
         a = [ [None]*self.ell for _ in range(self.k) ]
         for r in range(self.k):
             for s in range(self.ell):
-                rhop = rho + s.to_bytes() + r.to_bytes()
+                rhop = rho + self.integer_to_bytes(s, 1) + self.integer_to_bytes(r, 1)
                 a[r][s] = self.rej_ntt_poly(rhop)
         return a