1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
#include <gtest/gtest.h>
#include <time.h>
#include "elua.h"
static long mstime()
{
struct timespec now;
clock_gettime(CLOCK_MONOTONIC, &now);
return now.tv_sec * 1000 + now.tv_nsec/1000000;
}
static int count_set(elua_vm *L)
{
int num = 0;
for (int i = 0; i < 10000; i++)
{
for (int j = 0; j < 1000; j++)
{
num++;
}
}
int *count = (int *)elua_get_execute_userdata(L);
*count = num;
return 0;
}
TEST(elua_time_out, lua)
{
elua_vm *L = elua_create_vm("TEST");
const char *ud = "hello world.";
const char *script = "./script/elua_time_out.lua";
char *buff= (char *)"This is a test";
int len = strlen(buff);
elua_data out;
out.type = STRING;
out.integer = 1024;
out.buff = (char *)calloc(1, 1024);
long tt, elapsed;
int ret = 0;
struct elua_script *escript = elua_cache_script_file(L, script, 10);
ret = elua_execute_script(escript, buff, len, (void *)ud, NULL, &out);
EXPECT_EQ(ret, -1);
EXPECT_STREQ(elua_get_last_error_string(L), "[elua_call_script:1074] Lua script killed by time out..");
bzero(out.buff, 1024);
struct elua_script *escript0 = elua_cache_script_file(L, script, 0);
tt = mstime();
ret = elua_execute_script(escript0, buff, len, (void *)ud, NULL, &out);
elapsed = mstime() - tt;
EXPECT_EQ(0, ret);
EXPECT_LE(10, elapsed);
EXPECT_EQ(len, out.len);
EXPECT_EQ(STRING, out.type);
EXPECT_STREQ(buff, out.buff);
bzero(out.buff, 1024);
ret = elua_execute_script(escript, buff, len, (void *)ud, NULL, &out);
EXPECT_EQ(ret, -1);
EXPECT_STREQ(elua_get_last_error_string(L), "[elua_call_script:1074] Lua script killed by time out..");
bzero(out.buff, 1024);
elua_cleanup_script(escript);
elua_cleanup_script(escript0);
elua_destroy_vm(L);
free(out.buff);
}
TEST(elua_time_out, c)
{
elua_vm *L = elua_create_vm("TEST");
int count = 10;
const char *script = "./script/c_time_out.lua";
char *buff= (char *)"This is a test";
int len = strlen(buff);
struct elua_data out;
out.type = STRING;
out.integer = 1024;
out.buff = (char *)calloc(1, 1024);
long tt, elapsed;
elua_register_cbinding(L, NULL, "set_count", count_set);
struct elua_script *escript = elua_cache_script_file(L, script, 10);
tt = mstime();
int ret = elua_execute_script(escript, buff, len, (void *)&count, NULL, &out);
elapsed = mstime() - tt;
EXPECT_EQ(-1, ret);
EXPECT_STREQ(elua_get_last_error_string(L), "[elua_call_script:1074] Lua script killed by time out..");
EXPECT_LE(10, elapsed);
EXPECT_EQ(10000000, count);
bzero(out.buff, 1024);
struct elua_script *escript0 = elua_cache_script_file(L, script, 0);
tt = mstime();
ret = elua_execute_script(escript0, buff, len, (void *)&count, NULL, &out);
elapsed = mstime() - tt;
EXPECT_EQ(0, ret);
EXPECT_LE(10, elapsed);
EXPECT_EQ(len, out.len);
EXPECT_EQ(STRING, out.type);
EXPECT_STREQ(buff, out.buff);
EXPECT_EQ(10000000, count);
bzero(out.buff, 1024);
tt = mstime();
ret = elua_execute_script(escript, buff, len, (void *)&count, NULL, &out);
elapsed = mstime() - tt;
EXPECT_EQ(-1, ret);
EXPECT_STREQ(elua_get_last_error_string(L), "[elua_call_script:1074] Lua script killed by time out..");
EXPECT_LE(10, elapsed);
EXPECT_EQ(10000000, count);
bzero(out.buff, 1024);
elua_cleanup_script(escript0);
elua_cleanup_script(escript);
elua_destroy_vm(L);
free(out.buff);
}
|
