[Minor] timeout in seconds everywhere (as documentation says)

1 files changed, 8 insertions, 7 deletions

diff --git a/src/lua/lua_tcp.c b/src/lua/lua_tcp.c
index 2cf2919bf..5f67607bb 100644
--- a/src/lua/lua_tcp.c
+++ b/src/lua/lua_tcp.c
@@ -213,7 +213,7 @@ static const struct luaL_reg tcp_libm[] = {
 LUA_FUNCTION_DEF (tcp_sync, close);
 
 /***
- * @method set_timeout(timeout)
+ * @method set_timeout(seconds)
  *
  * Sets timeout for IO operations
  */
@@ -1813,14 +1813,16 @@ lua_tcp_set_timeout (lua_State *L)
 {
 	LUA_TRACE_POINT;
 	struct lua_tcp_cbdata *cbd = lua_check_tcp (L, 1);
-	gdouble ms = lua_tonumber (L, 2);
+	gdouble seconds = lua_tonumber (L, 2);
 
 	if (cbd == NULL) {
 		return luaL_error (L, "invalid arguments");
 	}
+	if (!lua_isnumber (L, 2)) {
+		return luaL_error (L, "invalid arguments: 'seconds' is expected to be number");
+	}
 
-	ms *= 1000.0;
-	double_to_tv (ms, &cbd->tv);
+	double_to_tv (seconds, &cbd->tv);
 
 	return 0;
 }
@@ -2019,7 +2021,7 @@ lua_tcp_sync_set_timeout (lua_State *L)
 {
 	LUA_TRACE_POINT;
 	struct lua_tcp_cbdata *cbd = lua_check_sync_tcp (L, 1);
-	gdouble ms = lua_tonumber (L, 2);
+	gdouble seconds = lua_tonumber (L, 2);
 
 	if (cbd == NULL) {
 		return luaL_error (L, "invalid arguments: self is not rspamd{tcp_sync}");
@@ -2028,8 +2030,7 @@ lua_tcp_sync_set_timeout (lua_State *L)
 		return luaL_error (L, "invalid arguments: second parameter is expected to be number");
 	}
 
-	ms *= 1000.0;
-	double_to_tv (ms, &cbd->tv);
+	double_to_tv (seconds, &cbd->tv);
 
 	return 0;
 }