bignum: Add generatePrime as new constructor

1 files changed, 26 insertions, 3 deletions

diff --git a/src/openssl.c b/src/openssl.c
index c311ba6..1ea5d16 100644
--- a/src/openssl.c
+++ b/src/openssl.c
@@ -223,6 +223,14 @@ static const char *xitoa(char *dst, size_t lim, long i) {
 } /* xitoa() */
 
 
+static _Bool optbool(lua_State *L, int idx, _Bool d) {
+        if (lua_isnoneornil(L, idx))
+                return d;
+        luaL_checktype(L, idx, LUA_TBOOLEAN);
+        return lua_toboolean(L, idx);
+} /* optbool() */
+
+
 static void *prepudata(lua_State *L, size_t size, const char *tname, int (*gc)(lua_State *)) {
         void *p = memset(lua_newuserdata(L, size), 0, size);
 
@@ -1976,6 +1984,20 @@ static int bn__gc(lua_State *L) {
 } /* bn__gc() */
 
 
+static int bn_generatePrime(lua_State *L) {
+        int bits = luaL_checkinteger(L, 1);
+        _Bool safe = optbool(L, 2, 0);
+        const BIGNUM *add = lua_isnoneornil(L, 3) ? NULL : checkbig(L, 3);
+        const BIGNUM *rem = lua_isnoneornil(L, 4) ? NULL : checkbig(L, 4);
+        BIGNUM *bn = bn_push(L);
+
+        if (!BN_generate_prime_ex(bn, bits, safe, add, rem, NULL))
+                return auxL_error(L, auxL_EOPENSSL, "bignum.generatePrime");
+
+        return 1;
+} /* bn_generatePrime() */
+
+
 static int bn_isPrime(lua_State *L) {
         BIGNUM *bn = checksimple(L, 1, BIGNUM_CLASS);
         int nchecks = luaL_optinteger(L, 2, BN_prime_checks);
@@ -2090,9 +2112,10 @@ static const luaL_Reg bn_metatable[] = {
 
 
 static const luaL_Reg bn_globals[] = {
-        { "new",       &bn_new },
-        { "interpose", &bn_interpose },
-        { NULL,        NULL },
+        { "new",           &bn_new },
+        { "interpose",     &bn_interpose },
+        { "generatePrime", &bn_generatePrime },
+        { NULL,            NULL },
 };
 
 int luaopen__openssl_bignum(lua_State *L) {