6fbf28f3b8
luaL_optinteger will raise an error when the argument is an invalid number. Delay the allocation to avoid a leak. Fixes the test_wslua_nstime test under ASAN. Change-Id: I6856fd218897565a60786d820f43192b41d489f2 Reviewed-on: https://code.wireshark.org/review/31744 Petri-Dish: Peter Wu <peter@lekensteyn.nl> Tested-by: Petri Dish Buildbot Reviewed-by: Anders Broman <a.broman58@gmail.com>
237 lines
6.5 KiB
C
237 lines
6.5 KiB
C
/*
|
|
* wslua_nstime.c
|
|
*
|
|
* Wireshark's interface to the Lua Programming Language
|
|
*
|
|
* (c) 2006, Luis E. Garcia Ontanon <luis@ontanon.org>
|
|
* (c) 2008, Balint Reczey <balint.reczey@ericsson.com>
|
|
* (c) 2011, Stig Bjorlykke <stig@bjorlykke.org>
|
|
* (c) 2014, Hadriel Kaplan <hadrielk@yahoo.com>
|
|
*
|
|
* Wireshark - Network traffic analyzer
|
|
* By Gerald Combs <gerald@wireshark.org>
|
|
* Copyright 1998 Gerald Combs
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include "wslua.h"
|
|
#include <wsutil/nstime.h>
|
|
|
|
/* 1s = 10^9 ns. */
|
|
#define NS_PER_S 1000000000
|
|
|
|
/* WSLUA_CONTINUE_MODULE Pinfo */
|
|
|
|
|
|
WSLUA_CLASS_DEFINE(NSTime,FAIL_ON_NULL("NSTime"));
|
|
/* NSTime represents a nstime_t. This is an object with seconds and nanoseconds. */
|
|
|
|
WSLUA_CONSTRUCTOR NSTime_new(lua_State *L) {
|
|
/* Creates a new NSTime object. */
|
|
#define WSLUA_OPTARG_NSTime_new_SECONDS 1 /* Seconds. */
|
|
#define WSLUA_OPTARG_NSTime_new_NSECONDS 2 /* Nano seconds. */
|
|
lua_Integer secs = luaL_optinteger(L,WSLUA_OPTARG_NSTime_new_SECONDS,0);
|
|
lua_Integer nsecs = luaL_optinteger(L,WSLUA_OPTARG_NSTime_new_NSECONDS,0);
|
|
|
|
NSTime nstime = g_new(nstime_t, 1);
|
|
if (!nstime) return 0;
|
|
nstime->secs = (time_t) secs;
|
|
nstime->nsecs = (int) nsecs;
|
|
|
|
pushNSTime(L,nstime);
|
|
|
|
WSLUA_RETURN(1); /* The new NSTime object. */
|
|
}
|
|
|
|
WSLUA_METAMETHOD NSTime__call(lua_State* L) { /* Creates a NSTime object. */
|
|
#define WSLUA_OPTARG_NSTime__call_SECONDS 1 /* Seconds. */
|
|
#define WSLUA_OPTARG_NSTime__call_NSECONDS 2 /* Nanoseconds. */
|
|
lua_remove(L,1); /* remove the table */
|
|
WSLUA_RETURN(NSTime_new(L)); /* The new NSTime object. */
|
|
}
|
|
|
|
WSLUA_METHOD NSTime_tonumber(lua_State* L) {
|
|
/* Returns a Lua number of the `NSTime` representing seconds from epoch
|
|
@since 2.4.0
|
|
*/
|
|
NSTime nstime = checkNSTime(L,1);
|
|
lua_pushnumber(L, (lua_Number)nstime_to_sec(nstime));
|
|
WSLUA_RETURN(1); /* The Lua number. */
|
|
}
|
|
|
|
WSLUA_METAMETHOD NSTime__tostring(lua_State* L) {
|
|
NSTime nstime = checkNSTime(L,1);
|
|
gchar *str;
|
|
long secs = (long)nstime->secs;
|
|
gint nsecs = nstime->nsecs;
|
|
gboolean negative_zero = FALSE;
|
|
|
|
/* Time is defined as sec + nsec/10^9, both parts can be negative.
|
|
* Translate this into the more familiar sec.nsec notation instead. */
|
|
if (secs > 0 && nsecs < 0) {
|
|
/* sign mismatch: (2, -3ns) -> 1.7 */
|
|
nsecs += NS_PER_S;
|
|
secs--;
|
|
} else if (secs < 0 && nsecs > 0) {
|
|
/* sign mismatch: (-2, 3ns) -> -1.7 */
|
|
nsecs = NS_PER_S - nsecs;
|
|
secs--;
|
|
} else if (nsecs < 0) {
|
|
/* Drop sign, the integer part already has it: (-2, -3ns) -> -2.3 */
|
|
nsecs = -nsecs;
|
|
/* In case the integer part is zero, it does not has a sign, so remember
|
|
* that it must be explicitly added. */
|
|
negative_zero = secs == 0;
|
|
}
|
|
|
|
if (negative_zero) {
|
|
str = wmem_strdup_printf(NULL, "-0.%09d", nsecs);
|
|
} else {
|
|
str = wmem_strdup_printf(NULL, "%ld.%09d", secs, nsecs);
|
|
}
|
|
lua_pushstring(L, str);
|
|
wmem_free(NULL, str);
|
|
|
|
WSLUA_RETURN(1); /* The string representing the nstime. */
|
|
}
|
|
WSLUA_METAMETHOD NSTime__add(lua_State* L) { /* Calculates the sum of two NSTimes. */
|
|
NSTime time1 = checkNSTime(L,1);
|
|
NSTime time2 = checkNSTime(L,2);
|
|
NSTime time3 = (NSTime)g_malloc (sizeof (nstime_t));
|
|
|
|
nstime_sum (time3, time1, time2);
|
|
pushNSTime (L, time3);
|
|
|
|
return 1;
|
|
}
|
|
|
|
WSLUA_METAMETHOD NSTime__sub(lua_State* L) { /* Calculates the diff of two NSTimes. */
|
|
NSTime time1 = checkNSTime(L,1);
|
|
NSTime time2 = checkNSTime(L,2);
|
|
NSTime time3 = (NSTime)g_malloc (sizeof (nstime_t));
|
|
|
|
nstime_delta (time3, time1, time2);
|
|
pushNSTime (L, time3);
|
|
|
|
return 1;
|
|
}
|
|
|
|
WSLUA_METAMETHOD NSTime__unm(lua_State* L) { /* Calculates the negative NSTime. */
|
|
NSTime time1 = checkNSTime(L,1);
|
|
NSTime time2 = (NSTime)g_malloc (sizeof (nstime_t));
|
|
|
|
nstime_set_zero (time2);
|
|
nstime_subtract (time2, time1);
|
|
pushNSTime (L, time2);
|
|
|
|
return 1;
|
|
}
|
|
|
|
WSLUA_METAMETHOD NSTime__eq(lua_State* L) { /* Compares two NSTimes. */
|
|
NSTime time1 = checkNSTime(L,1);
|
|
NSTime time2 = checkNSTime(L,2);
|
|
gboolean result = FALSE;
|
|
|
|
if (nstime_cmp(time1, time2) == 0)
|
|
result = TRUE;
|
|
|
|
lua_pushboolean(L,result);
|
|
|
|
return 1;
|
|
}
|
|
|
|
WSLUA_METAMETHOD NSTime__le(lua_State* L) { /* Compares two NSTimes. */
|
|
NSTime time1 = checkNSTime(L,1);
|
|
NSTime time2 = checkNSTime(L,2);
|
|
gboolean result = FALSE;
|
|
|
|
if (nstime_cmp(time1, time2) <= 0)
|
|
result = TRUE;
|
|
|
|
lua_pushboolean(L,result);
|
|
|
|
return 1;
|
|
}
|
|
|
|
WSLUA_METAMETHOD NSTime__lt(lua_State* L) { /* Compares two NSTimes. */
|
|
NSTime time1 = checkNSTime(L,1);
|
|
NSTime time2 = checkNSTime(L,2);
|
|
gboolean result = FALSE;
|
|
|
|
if (nstime_cmp(time1, time2) < 0)
|
|
result = TRUE;
|
|
|
|
lua_pushboolean(L,result);
|
|
|
|
return 1;
|
|
}
|
|
|
|
|
|
/* WSLUA_ATTRIBUTE NSTime_secs RW The NSTime seconds. */
|
|
WSLUA_ATTRIBUTE_NUMBER_GETTER(NSTime,secs);
|
|
WSLUA_ATTRIBUTE_NUMBER_SETTER(NSTime,secs,time_t);
|
|
|
|
/* WSLUA_ATTRIBUTE NSTime_nsecs RW The NSTime nano seconds. */
|
|
WSLUA_ATTRIBUTE_NUMBER_GETTER(NSTime,nsecs);
|
|
WSLUA_ATTRIBUTE_NUMBER_SETTER(NSTime,nsecs,int);
|
|
|
|
/* Gets registered as metamethod automatically by WSLUA_REGISTER_CLASS/META */
|
|
static int NSTime__gc(lua_State* L) {
|
|
NSTime nstime = toNSTime(L,1);
|
|
|
|
if (!nstime) return 0;
|
|
|
|
g_free (nstime);
|
|
return 0;
|
|
}
|
|
|
|
/* This table is ultimately registered as a sub-table of the class' metatable,
|
|
* and if __index/__newindex is invoked then it calls the appropriate function
|
|
* from this table for getting/setting the members.
|
|
*/
|
|
WSLUA_ATTRIBUTES NSTime_attributes[] = {
|
|
WSLUA_ATTRIBUTE_RWREG(NSTime,secs),
|
|
WSLUA_ATTRIBUTE_RWREG(NSTime,nsecs),
|
|
{ NULL, NULL, NULL }
|
|
};
|
|
|
|
WSLUA_METHODS NSTime_methods[] = {
|
|
WSLUA_CLASS_FNREG(NSTime,new),
|
|
WSLUA_CLASS_FNREG(NSTime,tonumber),
|
|
{ NULL, NULL }
|
|
};
|
|
|
|
WSLUA_META NSTime_meta[] = {
|
|
WSLUA_CLASS_MTREG(NSTime,tostring),
|
|
WSLUA_CLASS_MTREG(NSTime,add),
|
|
WSLUA_CLASS_MTREG(NSTime,sub),
|
|
WSLUA_CLASS_MTREG(NSTime,unm),
|
|
WSLUA_CLASS_MTREG(NSTime,eq),
|
|
WSLUA_CLASS_MTREG(NSTime,le),
|
|
WSLUA_CLASS_MTREG(NSTime,lt),
|
|
WSLUA_CLASS_MTREG(NSTime,call),
|
|
{ NULL, NULL }
|
|
};
|
|
|
|
int NSTime_register(lua_State* L) {
|
|
WSLUA_REGISTER_CLASS_WITH_ATTRS(NSTime);
|
|
return 0;
|
|
}
|
|
|
|
|
|
/*
|
|
* Editor modelines - http://www.wireshark.org/tools/modelines.html
|
|
*
|
|
* Local variables:
|
|
* c-basic-offset: 4
|
|
* tab-width: 8
|
|
* indent-tabs-mode: nil
|
|
* End:
|
|
*
|
|
* vi: set shiftwidth=4 tabstop=8 expandtab:
|
|
* :indentSize=4:tabSize=8:noTabs=true:
|
|
*/
|