From 32803d4577151e18252c7e2102738b544e99fcbc Mon Sep 17 00:00:00 2001 From: siddharth ravikumar Date: Mon, 14 Oct 2024 19:16:11 -0400 Subject: lib: update `RSAGenKey` Add `size` argument. --- lib/rsa.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'lib/rsa.go') diff --git a/lib/rsa.go b/lib/rsa.go index 772b898..9353824 100644 --- a/lib/rsa.go +++ b/lib/rsa.go @@ -68,7 +68,8 @@ func InvMod(a, n *big.Int) (*big.Int, error) { return t0, nil } -func RSAGenKey() (*RSAPair, error) { +// RSAGenKey generates a rsa key pair with N equal to size bits. +func RSAGenKey(size int) (*RSAPair, error) { // Initialize. e := big.NewInt(3) d := big.NewInt(0) @@ -77,13 +78,13 @@ func RSAGenKey() (*RSAPair, error) { // Compute n and d. for { // Generate prime p. - p, err := rand.Prime(rand.Reader, 1024) + p, err := rand.Prime(rand.Reader, size/2) if err != nil { return nil, CPError{"unable to generate p"} } // Generate prime q. - q, err := rand.Prime(rand.Reader, 1024) + q, err := rand.Prime(rand.Reader, size/2) if err != nil { return nil, CPError{"unable to generate q"} } -- cgit v1.2.3