[Project] Add some missing methods for spf record

diff --git a/src/lua/lua_spf.c b/src/lua/lua_spf.c
index a6daa26c60a9808151539f667ad9a41c6c768055..8a01b0ef78aee9fffcdf57a2e52be3b1dfa0aa93 100644 (file)
--- a/src/lua/lua_spf.c
+++ b/src/lua/lua_spf.c
@@ -26,8 +26,14 @@
 
 LUA_FUNCTION_DEF (spf, resolve);
 LUA_FUNCTION_DEF (spf, config);
+
 LUA_FUNCTION_DEF (spf_record, check_ip);
 LUA_FUNCTION_DEF (spf_record, dtor);
+LUA_FUNCTION_DEF (spf_record, get_domain);
+LUA_FUNCTION_DEF (spf_record, get_elts);
+LUA_FUNCTION_DEF (spf_record, get_ttl);
+LUA_FUNCTION_DEF (spf_record, get_timestamp);
+LUA_FUNCTION_DEF (spf_record, get_digest);
 
 static luaL_reg rspamd_spf_f[] = {
                LUA_INTERFACE_DEF (spf, resolve),
@@ -37,6 +43,10 @@ static luaL_reg rspamd_spf_f[] = {
 
 static luaL_reg rspamd_spf_record_m[] = {
                LUA_INTERFACE_DEF (spf_record, check_ip),
+               LUA_INTERFACE_DEF (spf_record, get_domain),
+               LUA_INTERFACE_DEF (spf_record, get_ttl),
+               LUA_INTERFACE_DEF (spf_record, get_digest),
+               LUA_INTERFACE_DEF (spf_record, get_elts),
                {"__gc", lua_spf_record_dtor},
                {NULL, NULL},
 };
@@ -355,6 +365,93 @@ lua_spf_record_check_ip (lua_State *L)
        return 3;
 }
 
+/***
+ * @method rspamd_spf_record:get_domain()
+ * Returns domain for the specific spf record
+*/
+static gint
+lua_spf_record_get_domain (lua_State *L)
+{
+       struct spf_resolved *record =
+                       *(struct spf_resolved **) rspamd_lua_check_udata (L, 1,
+                                       SPF_RECORD_CLASS);
+
+       if (record) {
+               lua_pushstring (L, record->domain);
+       }
+       else {
+               return luaL_error (L, "invalid arguments");
+       }
+
+       return 1;
+}
+
+/***
+ * @method rspamd_spf_record:get_ttl()
+ * Returns ttl for the specific spf record
+*/
+static gint
+lua_spf_record_get_ttl (lua_State *L)
+{
+       struct spf_resolved *record =
+                       *(struct spf_resolved **) rspamd_lua_check_udata (L, 1,
+                                       SPF_RECORD_CLASS);
+
+       if (record) {
+               lua_pushinteger (L, record->ttl);
+       }
+       else {
+               return luaL_error (L, "invalid arguments");
+       }
+
+       return 1;
+}
+
+/***
+ * @method rspamd_spf_record:get_timestamp()
+ * Returns ttl for the specific spf record
+*/
+static gint
+lua_spf_record_get_timestamp (lua_State *L)
+{
+       struct spf_resolved *record =
+                       *(struct spf_resolved **) rspamd_lua_check_udata (L, 1,
+                                       SPF_RECORD_CLASS);
+
+       if (record) {
+               lua_pushnumber (L, record->timestamp);
+       }
+       else {
+               return luaL_error (L, "invalid arguments");
+       }
+
+       return 1;
+}
+
+/***
+ * @method rspamd_spf_record:get_digest()
+ * Returns string hex representation of the record digest (fast hash function)
+*/
+static gint
+lua_spf_record_get_digest (lua_State *L)
+{
+       struct spf_resolved *record =
+                       *(struct spf_resolved **) rspamd_lua_check_udata (L, 1,
+                                       SPF_RECORD_CLASS);
+
+       if (record) {
+               gchar hexbuf[64];
+
+               rspamd_snprintf (hexbuf, sizeof (hexbuf), "%xuL", record->digest);
+               lua_pushstring (L, hexbuf);
+       }
+       else {
+               return luaL_error (L, "invalid arguments");
+       }
+
+       return 1;
+}
+
 /***
  * @function rspamd_spf.config(object)
  * Configures SPF library according to the UCL config