src/openssl.c: Add bn:mod_exp() function

1 files changed, 28 insertions, 0 deletions

diff --git a/src/openssl.c b/src/openssl.c
index 5ff80fe..ba6e9b6 100644
--- a/src/openssl.c
+++ b/src/openssl.c
@@ -2798,6 +2798,20 @@ static void bn_prepbop(lua_State *L, BIGNUM **r, BIGNUM **a, BIGNUM **b, _Bool c
 } /* bn_prepbop() */
 
 
+/* prepare numbers at top of stack for ternary operation, and push result object onto stack  */
+static void bn_preptop(lua_State *L, BIGNUM **r, BIGNUM **a, BIGNUM **b, BIGNUM **c) {
+	_Bool a_lvalue, b_lvalue, c_lvalue;
+
+	*a = checkbig(L, 1, &a_lvalue);
+	*b = checkbig(L, 2, &b_lvalue);
+	*c = checkbig(L, 3, &c_lvalue);
+
+	bn_push(L);
+
+	*r = *(BIGNUM **)lua_touserdata(L, -1);
+} /* bn_preptop() */
+
+
 static int ctx__gc(lua_State *L) {
 	BN_CTX **ctx = lua_touserdata(L, 1);
 
@@ -2959,6 +2973,19 @@ static int bn__pow(lua_State *L) {
 } /* bn__pow() */
 
 
+static int bn_mod_exp(lua_State *L) {
+	BIGNUM *r, *a, *b, *c;
+
+	lua_settop(L, 3);
+	bn_preptop(L, &r, &a, &b, &c);
+
+	if (!BN_mod_exp(r, a, b, c, getctx(L)))
+		return auxL_error(L, auxL_EOPENSSL, "bignum:mod_exp");
+
+	return 1;
+} /* bn_mod_exp() */
+
+
 static int bn_gcd(lua_State *L) {
 	BIGNUM *r, *a, *b;
 
@@ -3153,6 +3180,7 @@ static const auxL_Reg bn_methods[] = {
 	{ "mod",       &bn__mod },
 	{ "nnmod",     &bn_nnmod },
 	{ "exp",       &bn__pow },
+	{ "mod_exp",   &bn_mod_exp },
 	{ "gcd",       &bn_gcd },
 	{ "lshift",    &bn__shl },
 	{ "rshift",    &bn__shr },