kryptools 0.5__tar.gz → 0.6__tar.gz

{kryptools-0.5 → kryptools-0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kryptools
-Version: 0.5
+Version: 0.6
 Summary: Implemenation of same basic algorithms used in cryptography.
 Author-email: Gerald Teschl <gerald.teschl@univie.ac.at>
 Project-URL: Homepage, https://github.com/teschlg/kryptools

{kryptools-0.5 → kryptools-0.6}/kryptools/__init__.py

@@ -8,6 +8,6 @@ from .factor import factorint
 from .dlp import dlog
 from .ec import EC_Weierstrass
 from .la import Matrix, zeros, eye
-from .lat import gram_det, hadamard_ratio, hermite_nf, gram_schmidt, babai_round_cvp, babai_plane_cvp, lagrange_lr, lll, random_unimodular_matrix
+from .lat import gram_det, hadamard_ratio, hermite_nf, gram_schmidt, babai_round_cvp, babai_round_bnd, babai_plane_cvp, babai_plane_bnd, lagrange_lr, lll, random_unimodular_matrix
 from .poly import Poly
 from .Zmod import Zmod

{kryptools-0.5 → kryptools-0.6}/kryptools/la.py

@@ -4,7 +4,7 @@ Linear algebra
 
 from math import inf, sqrt, prod
 from numbers import Number
-
+from fractions import Fraction
 
 class Matrix:
     """
@@ -148,6 +148,8 @@ class Matrix:
             for j in range(other.cols):
                 for k in range(other.rows):
                     result.matrix[i][j] += self.matrix[i][k] * other.matrix[k][j]
+        if self.rows == 1 and other.cols == 1:
+            return result.matrix[0][0]
         return result
 
     def __add__(self, other) -> "Matrix":
@@ -176,7 +178,7 @@ class Matrix:
 
     def __rmul__(self, other) -> "Matrix":
         if isinstance(other, Number) or type(other) == type(self.matrix[0][0]):
-             return Matrix([ [item * other for item in row] for row in self.matrix ])
+            return Matrix([ [item * other for item in row] for row in self.matrix ])
         return NotImplemented
 
     def rref(self) -> "Matrix":
@@ -246,6 +248,16 @@ class Matrix:
             raise ValueError("Matrix is not invertible!")
         return MM[:,n:]
 
+    def is_unimodular(self) -> bool:
+        "Test if the matrix is unimodular."
+        if self.rows != self.rows:
+            return False
+        def is_integer(i):
+            if isinstance(i, int) or (isinstance(i, Fraction) and i.denominator == 1):
+                return True
+            return False
+        return all([is_integer(i) for i in self]) and self.det()**2 == 1
+
     def zeros(self, m: int = None, n: int = None):
         "Returns a zero matrix of the same dimension"
         if not m and not n:

{kryptools-0.5 → kryptools-0.6}/kryptools/lat.py

@@ -2,7 +2,7 @@
 Lattice tools
 """
 
-from math import prod
+from math import prod, floor
 from fractions import Fraction
 from random import choice, sample
 from .la import Matrix, zeros
@@ -73,19 +73,28 @@ def hadamard_ratio(M: Matrix) -> float:
     return (gram_det(M) / prod([M[:, i].norm() for i in range(m)])) ** (1 / m)
 
 def babai_round_cvp(x: Matrix, U: Matrix) -> Matrix:
-    "Babai's rounding algorithm for solving the CVP."
+    "Babai's rounding algorithm for approximately solving the CVP."
     s = U.inv() * x
     k = s.applyfunc(round)
     return U * k
 
+def babai_round_bnd(U: Matrix) -> float:
+    "Bound for Babai's rounding algorithm for solving the CVP."
+    return floor(1 / (2 * max([ U.inv()[i,:].norm(1) for i in range(U.rows)])))
+
 def babai_plane_cvp(x: Matrix, U: Matrix) -> Matrix:
-    "Babai's closest plane algorithm for solving the CVP."
+    "Babai's closest plane algorithm for approximately solving the CVP."
     Us = gram_schmidt(U)[0]
     y = x
     for k in range(U.cols - 1, -1, -1):
         y = y - round(y.dot(Us[:, k]) / norm2(Us[:, k])) * U[:, k]
     return (x - y).applyfunc(round)
 
+def babai_plane_bnd(U: Matrix, p = 2) -> float:
+    "Bound for Babai's closest plane algorithm for solving the CVP."
+    Us = gram_schmidt(U)[0]
+    return float(0.5 * min([Us[:, i].norm(p) for i in range(Us.rows)]))
+
 def lagrange_lr(V: Matrix) -> Matrix:
     "Lagrange lattice reduction."
     assert (V.rows, V.cols) == (2, 2)
@@ -99,7 +108,7 @@ def lagrange_lr(V: Matrix) -> Matrix:
     return Matrix([list(v1), list(v3)]).transpose()
 
 def lll(V: Matrix, delta: float = 0.75, sort: bool = True) -> Matrix:
-    "lll algorithm for lattice reduction"
+    "LLL algorithm for lattice reduction."
 
     assert 0 < delta <= 1, f"LLL reqires 0 < delta={delta} <= 1"
     j = 1

{kryptools-0.5 → kryptools-0.6}/kryptools/poly.py

@@ -94,10 +94,13 @@ class Poly:
         "Apply a given function to all coefficients."
         self.coeff = list(map(func, self.coeff))
 
+    def _check_type(self, other):
+        return isinstance(other, int) or (isinstance(other, Number) and isinstance(self.coeff[0], Number)) or type(other) == type(self.coeff[0])
+
     def __add__(self, other: "Poly") -> "Poly":
         if not isinstance(other, self.__class__):
-            tmp = self.coeff[:]
-            if isinstance(other, int) or type(other) == type(tmp[0]):
+            if self._check_type(other):
+                tmp = self.coeff[:]
                 tmp[0] += other
                 return self.__class__(tmp, modulus=self.modulus)
             return NotImplemented
@@ -117,7 +120,7 @@ class Poly:
 
     def __radd__(self, other: "Poly") -> "Poly":
         if not isinstance(other, self.__class__):
-            if isinstance(other, Number) or type(other) == type(self.coeff[0]):
+            if self._check_type(other):
                 tmp = self.coeff[:]
                 tmp[0] += other
                 return self.__class__(tmp, modulus=self.modulus)
@@ -128,7 +131,7 @@ class Poly:
 
     def __sub__(self, other: "Poly") -> "Poly":
         if not isinstance(other, self.__class__):
-            if isinstance(other, int) or type(other) == type(self.coeff[0]):
+            if self._check_type(other):
                 tmp = self.coeff[:]
                 tmp[0] -= other
                 return self.__class__(tmp, modulus=self.modulus)
@@ -149,7 +152,7 @@ class Poly:
 
     def __rsub__(self, other: "Poly") -> "Poly":
         if not isinstance(other, self.__class__):
-            if isinstance(other, int) or type(other) == type(self.coeff[0]):
+            if self._check_type(other):
                 tmp = self.coeff[:]
                 tmp[0] -= other
                 return self.__class__(tmp, modulus=self.modulus)
@@ -157,7 +160,7 @@ class Poly:
 
     def __mul__(self, other: "Poly") -> "Poly":
         if not isinstance(other, self.__class__):
-            if isinstance(other, int) or type(other) == type(self.coeff[0]):
+            if self._check_type(other):
                 return Poly([other * s for s in self.coeff], modulus = self.modulus)
             return NotImplemented
         ls, lo = len(self.coeff), len(other.coeff)
@@ -174,11 +177,25 @@ class Poly:
             modulus = other.modulus
         return self.__class__(coeff, modulus = modulus)
 
-    def __rmul__(self, other: int) -> "Poly":
-        if isinstance(other, int) or type(other) == type(self.coeff[0]):
+    def __rmul__(self, other) -> "Poly":
+        if self._check_type(other):
             return self.__class__([other * s for s in self.coeff], modulus=self.modulus)
         return NotImplemented
 
+    def __truediv__(self, other) -> "Poly":
+        if self._check_type(other):
+            return self.__class__([s / other for s in self.coeff], modulus=self.modulus)
+        if isinstance(other, self.__class__):
+            if not other.modulus:
+                raise NotImplementedError("Cannot invert polynomials without modulus.")
+            return self * other.inv()
+        return NotImplemented
+
+    def __rtruediv__(self, other) -> "Poly":
+        if not self.modulus:
+            raise NotImplementedError("Cannot invert polynomials without modulus.")
+        return other * self.inv()
+
     def __pow__(self, i: int) -> "Poly":
         if not isinstance(i, int):
             return NotImplemented

{kryptools-0.5 → kryptools-0.6}/kryptools.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kryptools
-Version: 0.5
+Version: 0.6
 Summary: Implemenation of same basic algorithms used in cryptography.
 Author-email: Gerald Teschl <gerald.teschl@univie.ac.at>
 Project-URL: Homepage, https://github.com/teschlg/kryptools

{kryptools-0.5 → kryptools-0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kryptools"
-version = "0.5"
+version = "0.6"
 authors = [
   { name="Gerald Teschl", email="gerald.teschl@univie.ac.at" },
 ]