on master help sign in
only strings left
commitacad1c1497023ac23c658567e9aff790cfdcc640 authoriko <mail@iko.soy> dateThu 25 Jul 2024, 23:29 (2 years ago) parentbee08e8
29 files changed+13742−36
zod/luau/lib/lua/parser.hoon+47−36 zod/luau/tests/bwcoercion.lua+78 zod/luau/tests/calls.lua+497 zod/luau/tests/closure.lua+272 zod/luau/tests/code.lua+449 zod/luau/tests/constructs.lua+406 zod/luau/tests/coroutine.lua+1156 zod/luau/tests/cstack.lua+197 zod/luau/tests/db.lua+1054 zod/luau/tests/errors.lua+696 zod/luau/tests/events.lua+491 zod/luau/tests/files.lua+951 zod/luau/tests/gc.lua+695 zod/luau/tests/gengc.lua+172 zod/luau/tests/goto.lua+271 zod/luau/tests/heavy.lua+173 zod/luau/tests/literals.lua+343 zod/luau/tests/locals.lua+1181 zod/luau/tests/main.lua+563 zod/luau/tests/math.lua+1024 zod/luau/tests/nextvar.lua+825 zod/luau/tests/pm.lua+440 zod/luau/tests/sort.lua+311 zod/luau/tests/strings.lua+526 zod/luau/tests/tpack.lua+322 zod/luau/tests/tracegc.lua+40 zod/luau/tests/utf8.lua+259 zod/luau/tests/vararg.lua+151 zod/luau/tests/verybig.lua+152
zod/luau/lib/lua/parser.hoon+47−36
@@ -66,7 +66,7 @@ apex
6666 ==
6767 ;~ sfix
6868 (cold %not (jest 'not'))
69− wss
69+ ;~(less parse-name ws)
7070 ==
7171 ==
7272 %+ knee *expr |.(parse-expr)
@@ -203,22 +203,23 @@ apex
203203 ;~ pose
204204 %+ cook
205205 |= x=@rd [%float x]
206− ;~ pose
207− parse-float
208− parse-hex-float
209− ==
206+ parse-hex-float
210207 ::
211208 %+ cook
212209 |= x=@u [%int x]
213− ;~ pose
214− ;~ pfix
215− (just '0')
216− (mask "xX")
217− hex
218− ==
219− ::
220− (bass 10 (plus dit))
210+ ;~ pfix
211+ (just '0')
212+ (mask "xX")
213+ hex
221214 ==
215+ ::
216+ %+ cook
217+ |= x=@rd [%float x]
218+ parse-float
219+ ::
220+ %+ cook
221+ |= x=@u [%int x]
222+ (bass 10 (plus dit))
222223 ==
223224++ parse-exprlist
224225 %^ tnee %parse-exprlist exprlist
@@ -401,7 +402,7 @@ apex
401402++ parse-for-range
402403 %^ tnee %for-range for-range
403404 %+ cook
404− |= [* * var=name * * * from=expr * * * to=expr by=(unit expr) * * * body=blok * *]
405+ |= [* * var=name * * * from=expr * * * to=expr by=(unit expr) * * * body=blok *]
405406 [var from to by body]
406407 ;~ plug
407408 (jest 'for')
@@ -426,8 +427,11 @@ apex
426427 wss
427428 (jest 'do')
428429 wss
429− parse-blok
430− wss
430+ ;< body=blok bind parse-blok
431+ ;~ pfix
432+ ?: (blok-is-empty body) ws wss
433+ (easy body)
434+ ==
431435 (jest 'end')
432436 ==
433437++ parse-function
@@ -512,15 +516,20 @@ apex
512516++ parse-for-in
513517 %^ tnee %parse-for-in for-in
514518 %+ cook
515− |= [* vars=namelist * src=explist * body=blok *] [vars src body]
519+ |= [* vars=namelist * src=explist * body=blok] [vars src body]
516520 ;~ (glue wss)
517521 (jest 'for')
518522 parse-namelist
519523 (jest 'in')
520524 parse-explist
521525 (jest 'do')
522− parse-blok
523− (jest 'end')
526+ ::
527+ ;< body=blok bind parse-blok
528+ ;~ pfix
529+ ?: (blok-is-empty) ws wss
530+ (jest 'end')
531+ (easy body)
532+ ==
524533 ==
525534++ parse-explist
526535 %^ tnee %parse-explist explist
@@ -689,14 +698,16 @@ apex
689698 %+ cook ryld
690699 %+ cook royl-cell:so
691700 %+ sear
692− |= [int=@ frac=(unit [@ @]) exp=(unit [exp-sign=? exp=@])]
693− ?: ?=([~ ~] [frac exp]) ~
701+ |= [leading-zeros=tape int=(unit @) frac=(unit (unit [@ @])) exp=(unit [exp-sign=? exp=@])]
702+ =/ int=(unit @) ?~(int ?:(?=(~ leading-zeros) ~ `0) int)
703+ ?: ?=([~ ~ ~] [int frac exp]) ~
704+ ?: ?=([~ [~ ~]] [int frac]) ~
694705 %- some
695706 :*
696707 %d
697708 &
698− int
699− (fall frac [0 0])
709+ (fall int 0)
710+ (fall (fall frac `[0 0]) [0 0])
700711 (fall exp [& 0])
701712 ==
702713 =/ moo
@@ -704,16 +715,15 @@ apex
704715 :- (lent a)
705716 (scan a (bass 10 (plus sid:ab)))
706717 ;~ plug
707− ;~ plug
708− dim:ag
709− ::
710− (punt ;~(pfix dot (cook moo (plus (shim '0' '9')))))
711− ::
712− %- punt
713− ;~ pfix
714− (mask "eE")
715− ;~(plug ;~(pose (cold | hep) (easy &)) dim:ag)
716− ==
718+ (star (just '0'))
719+ (punt dim:ag)
720+ ::
721+ (punt ;~(pfix dot (punt (cook moo (plus (shim '0' '9'))))))
722+ ::
723+ %- punt
724+ ;~ pfix
725+ (mask "eE")
726+ ;~(plug ;~(pose (cold | hep) (cold & lus) (easy &)) dum:ag)
717727 ==
718728 ==
719729++ parse-hex-float
@@ -722,6 +732,7 @@ apex
722732 %+ sear
723733 |= [* * int=tape frac=(unit tape) exp=(unit [exp-sign=? exp=@])]
724734 ?: ?=([~ ~] [frac exp]) ~
735+ ?: ?=([~ ~] [int frac]) ~
725736 %- some
726737 =/ frac (fall frac "")
727738 =/ whole=@rd (sun:rd (scan (weld int frac) hex))
@@ -737,7 +748,7 @@ apex
737748 ::
738749 (mask "xX")
739750 ::
740− (plus hex-digit)
751+ (star hex-digit)
741752 ::
742753 %- punt
743754 ;~ pfix
@@ -809,8 +820,8 @@ apex
809820 |* [=term gar=mold sef=rule]
810821 |= tub=nail
811822 ^- (like gar)
812− =/ trace |
813− ~? trace [p.p.tub term `tape`(scag 30 q.tub)]
823+ =/ trace &
824+ ~? trace [p.p.tub term `tape`(scag 40 q.tub)]
814825 =/ res (sef tub)
815826 ?. trace res
816827 ?~ q.res
zod/luau/tests/bwcoercion.luanew file+78
@@ -0,0 +1,78 @@
1+local tonumber, tointeger = tonumber, math.tointeger
2+local type, getmetatable, rawget, error = type, getmetatable, rawget, error
3+local strsub = string.sub
4+
5+local print = print
6+
7+_ENV = nil
8+
9+-- Try to convert a value to an integer, without assuming any coercion.
10+local function toint (x)
11+ x = tonumber(x) -- handle numerical strings
12+ if not x then
13+ return false -- not coercible to a number
14+ end
15+ return tointeger(x)
16+end
17+
18+
19+-- If operation fails, maybe second operand has a metamethod that should
20+-- have been called if not for this string metamethod, so try to
21+-- call it.
22+local function trymt (x, y, mtname)
23+ if type(y) ~= "string" then -- avoid recalling original metamethod
24+ local mt = getmetatable(y)
25+ local mm = mt and rawget(mt, mtname)
26+ if mm then
27+ return mm(x, y)
28+ end
29+ end
30+ -- if any test fails, there is no other metamethod to be called
31+ error("attempt to '" .. strsub(mtname, 3) ..
32+ "' a " .. type(x) .. " with a " .. type(y), 4)
33+end
34+
35+
36+local function checkargs (x, y, mtname)
37+ local xi = toint(x)
38+ local yi = toint(y)
39+ if xi and yi then
40+ return xi, yi
41+ else
42+ return trymt(x, y, mtname), nil
43+ end
44+end
45+
46+
47+local smt = getmetatable("")
48+
49+smt.__band = function (x, y)
50+ local x, y = checkargs(x, y, "__band")
51+ return y and x & y or x
52+end
53+
54+smt.__bor = function (x, y)
55+ local x, y = checkargs(x, y, "__bor")
56+ return y and x | y or x
57+end
58+
59+smt.__bxor = function (x, y)
60+ local x, y = checkargs(x, y, "__bxor")
61+ return y and x ~ y or x
62+end
63+
64+smt.__shl = function (x, y)
65+ local x, y = checkargs(x, y, "__shl")
66+ return y and x << y or x
67+end
68+
69+smt.__shr = function (x, y)
70+ local x, y = checkargs(x, y, "__shr")
71+ return y and x >> y or x
72+end
73+
74+smt.__bnot = function (x)
75+ local x, y = checkargs(x, x, "__bnot")
76+ return y and ~x or x
77+end
78+
zod/luau/tests/calls.luanew file+497
@@ -0,0 +1,497 @@
1+-- $Id: testes/calls.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print("testing functions and calls")
5+
6+local debug = require "debug"
7+
8+-- get the opportunity to test 'type' too ;)
9+
10+assert(type(1<2) == 'boolean')
11+assert(type(true) == 'boolean' and type(false) == 'boolean')
12+assert(type(nil) == 'nil'
13+ and type(-3) == 'number'
14+ and type'x' == 'string'
15+ and type{} == 'table'
16+ and type(type) == 'function')
17+
18+assert(type(assert) == type(print))
19+local function f (x) return a:x (x) end
20+assert(type(f) == 'function')
21+assert(not pcall(type))
22+
23+
24+-- testing local-function recursion
25+fact = false
26+do
27+ local res = 1
28+ local function fact (n)
29+ if n==0 then return res
30+ else return n*fact(n-1)
31+ end
32+ end
33+ assert(fact(5) == 120)
34+end
35+assert(fact == false)
36+fact = nil
37+
38+-- testing declarations
39+local a = {i = 10}
40+local self = 20
41+function a:x (x) return x+self.i end
42+function a.y (x) return x+self end
43+
44+assert(a:x(1)+10 == a.y(1))
45+
46+a.t = {i=-100}
47+a["t"].x = function (self, a,b) return self.i+a+b end
48+
49+assert(a.t:x(2,3) == -95)
50+
51+do
52+ local a = {x=0}
53+ function a:add (x) self.x, a.y = self.x+x, 20; return self end
54+ assert(a:add(10):add(20):add(30).x == 60 and a.y == 20)
55+end
56+
57+local a = {b={c={}}}
58+
59+function a.b.c.f1 (x) return x+1 end
60+function a.b.c:f2 (x,y) self[x] = y end
61+assert(a.b.c.f1(4) == 5)
62+a.b.c:f2('k', 12); assert(a.b.c.k == 12)
63+
64+print('+')
65+
66+t = nil -- 'declare' t
67+function f(a,b,c) local d = 'a'; t={a,b,c,d} end
68+
69+f( -- this line change must be valid
70+ 1,2)
71+assert(t[1] == 1 and t[2] == 2 and t[3] == nil and t[4] == 'a')
72+f(1,2, -- this one too
73+ 3,4)
74+assert(t[1] == 1 and t[2] == 2 and t[3] == 3 and t[4] == 'a')
75+
76+t = nil -- delete 't'
77+
78+function fat(x)
79+ if x <= 1 then return 1
80+ else return x*load("return fat(" .. x-1 .. ")", "")()
81+ end
82+end
83+
84+assert(load "load 'assert(fat(6)==720)' () ")()
85+a = load('return fat(5), 3')
86+local a,b = a()
87+assert(a == 120 and b == 3)
88+fat = nil
89+print('+')
90+
91+local function err_on_n (n)
92+ if n==0 then error(); exit(1);
93+ else err_on_n (n-1); exit(1);
94+ end
95+end
96+
97+do
98+ local function dummy (n)
99+ if n > 0 then
100+ assert(not pcall(err_on_n, n))
101+ dummy(n-1)
102+ end
103+ end
104+
105+ dummy(10)
106+end
107+
108+_G.deep = nil -- "declaration" (used by 'all.lua')
109+
110+function deep (n)
111+ if n>0 then deep(n-1) end
112+end
113+deep(10)
114+deep(180)
115+
116+
117+print"testing tail calls"
118+
119+function deep (n) if n>0 then return deep(n-1) else return 101 end end
120+assert(deep(30000) == 101)
121+a = {}
122+function a:deep (n) if n>0 then return self:deep(n-1) else return 101 end end
123+assert(a:deep(30000) == 101)
124+
125+do -- tail calls x varargs
126+ local function foo (x, ...) local a = {...}; return x, a[1], a[2] end
127+
128+ local function foo1 (x) return foo(10, x, x + 1) end
129+
130+ local a, b, c = foo1(-2)
131+ assert(a == 10 and b == -2 and c == -1)
132+
133+ -- tail calls x metamethods
134+ local t = setmetatable({}, {__call = foo})
135+ local function foo2 (x) return t(10, x) end
136+ a, b, c = foo2(100)
137+ assert(a == t and b == 10 and c == 100)
138+
139+ a, b = (function () return foo() end)()
140+ assert(a == nil and b == nil)
141+
142+ local X, Y, A
143+ local function foo (x, y, ...) X = x; Y = y; A = {...} end
144+ local function foo1 (...) return foo(...) end
145+
146+ local a, b, c = foo1()
147+ assert(X == nil and Y == nil and #A == 0)
148+
149+ a, b, c = foo1(10)
150+ assert(X == 10 and Y == nil and #A == 0)
151+
152+ a, b, c = foo1(10, 20)
153+ assert(X == 10 and Y == 20 and #A == 0)
154+
155+ a, b, c = foo1(10, 20, 30)
156+ assert(X == 10 and Y == 20 and #A == 1 and A[1] == 30)
157+end
158+
159+
160+do -- C-stack overflow while handling C-stack overflow
161+ local function loop ()
162+ assert(pcall(loop))
163+ end
164+
165+ local err, msg = xpcall(loop, loop)
166+ assert(not err and string.find(msg, "error"))
167+end
168+
169+
170+
171+do -- tail calls x chain of __call
172+ local n = 10000 -- depth
173+
174+ local function foo ()
175+ if n == 0 then return 1023
176+ else n = n - 1; return foo()
177+ end
178+ end
179+
180+ -- build a chain of __call metamethods ending in function 'foo'
181+ for i = 1, 100 do
182+ foo = setmetatable({}, {__call = foo})
183+ end
184+
185+ -- call the first one as a tail call in a new coroutine
186+ -- (to ensure stack is not preallocated)
187+ assert(coroutine.wrap(function() return foo() end)() == 1023)
188+end
189+
190+print('+')
191+
192+
193+do -- testing chains of '__call'
194+ local N = 20
195+ local u = table.pack
196+ for i = 1, N do
197+ u = setmetatable({i}, {__call = u})
198+ end
199+
200+ local Res = u("a", "b", "c")
201+
202+ assert(Res.n == N + 3)
203+ for i = 1, N do
204+ assert(Res[i][1] == i)
205+ end
206+ assert(Res[N + 1] == "a" and Res[N + 2] == "b" and Res[N + 3] == "c")
207+end
208+
209+
210+a = nil
211+(function (x) a=x end)(23)
212+assert(a == 23 and (function (x) return x*2 end)(20) == 40)
213+
214+
215+-- testing closures
216+
217+-- fixed-point operator
218+local Z = function (le)
219+ local function a (f)
220+ return le(function (x) return f(f)(x) end)
221+ end
222+ return a(a)
223+ end
224+
225+
226+-- non-recursive factorial
227+
228+local F = function (f)
229+ return function (n)
230+ if n == 0 then return 1
231+ else return n*f(n-1) end
232+ end
233+ end
234+
235+local fat = Z(F)
236+
237+assert(fat(0) == 1 and fat(4) == 24 and Z(F)(5)==5*Z(F)(4))
238+
239+local function g (z)
240+ local function f (a,b,c,d)
241+ return function (x,y) return a+b+c+d+a+x+y+z end
242+ end
243+ return f(z,z+1,z+2,z+3)
244+end
245+
246+local f = g(10)
247+assert(f(9, 16) == 10+11+12+13+10+9+16+10)
248+
249+print('+')
250+
251+-- testing multiple returns
252+
253+local function unlpack (t, i)
254+ i = i or 1
255+ if (i <= #t) then
256+ return t[i], unlpack(t, i+1)
257+ end
258+end
259+
260+local function equaltab (t1, t2)
261+ assert(#t1 == #t2)
262+ for i = 1, #t1 do
263+ assert(t1[i] == t2[i])
264+ end
265+end
266+
267+local pack = function (...) return (table.pack(...)) end
268+
269+local function f() return 1,2,30,4 end
270+local function ret2 (a,b) return a,b end
271+
272+local a,b,c,d = unlpack{1,2,3}
273+assert(a==1 and b==2 and c==3 and d==nil)
274+a = {1,2,3,4,false,10,'alo',false,assert}
275+equaltab(pack(unlpack(a)), a)
276+equaltab(pack(unlpack(a), -1), {1,-1})
277+a,b,c,d = ret2(f()), ret2(f())
278+assert(a==1 and b==1 and c==2 and d==nil)
279+a,b,c,d = unlpack(pack(ret2(f()), ret2(f())))
280+assert(a==1 and b==1 and c==2 and d==nil)
281+a,b,c,d = unlpack(pack(ret2(f()), (ret2(f()))))
282+assert(a==1 and b==1 and c==nil and d==nil)
283+
284+a = ret2{ unlpack{1,2,3}, unlpack{3,2,1}, unlpack{"a", "b"}}
285+assert(a[1] == 1 and a[2] == 3 and a[3] == "a" and a[4] == "b")
286+
287+
288+-- testing calls with 'incorrect' arguments
289+rawget({}, "x", 1)
290+rawset({}, "x", 1, 2)
291+assert(math.sin(1,2) == math.sin(1))
292+table.sort({10,9,8,4,19,23,0,0}, function (a,b) return a<b end, "extra arg")
293+
294+
295+-- test for generic load
296+local x = "-- a comment\0\0\0\n x = 10 + \n23; \
297+ local a = function () x = 'hi' end; \
298+ return '\0'"
299+local function read1 (x)
300+ local i = 0
301+ return function ()
302+ collectgarbage()
303+ i=i+1
304+ return string.sub(x, i, i)
305+ end
306+end
307+
308+local function cannotload (msg, a,b)
309+ assert(not a and string.find(b, msg))
310+end
311+
312+a = assert(load(read1(x), "modname", "t", _G))
313+assert(a() == "\0" and _G.x == 33)
314+assert(debug.getinfo(a).source == "modname")
315+-- cannot read text in binary mode
316+cannotload("attempt to load a text chunk", load(read1(x), "modname", "b", {}))
317+cannotload("attempt to load a text chunk", load(x, "modname", "b"))
318+
319+a = assert(load(function () return nil end))
320+a() -- empty chunk
321+
322+assert(not load(function () return true end))
323+
324+
325+-- small bug
326+local t = {nil, "return ", "3"}
327+f, msg = load(function () return table.remove(t, 1) end)
328+assert(f() == nil) -- should read the empty chunk
329+
330+-- another small bug (in 5.2.1)
331+f = load(string.dump(function () return 1 end), nil, "b", {})
332+assert(type(f) == "function" and f() == 1)
333+
334+
335+do -- another bug (in 5.4.0)
336+ -- loading a binary long string interrupted by GC cycles
337+ local f = string.dump(function ()
338+ return '01234567890123456789012345678901234567890123456789'
339+ end)
340+ f = load(read1(f))
341+ assert(f() == '01234567890123456789012345678901234567890123456789')
342+end
343+
344+
345+x = string.dump(load("x = 1; return x"))
346+a = assert(load(read1(x), nil, "b"))
347+assert(a() == 1 and _G.x == 1)
348+cannotload("attempt to load a binary chunk", load(read1(x), nil, "t"))
349+cannotload("attempt to load a binary chunk", load(x, nil, "t"))
350+_G.x = nil
351+
352+assert(not pcall(string.dump, print)) -- no dump of C functions
353+
354+cannotload("unexpected symbol", load(read1("*a = 123")))
355+cannotload("unexpected symbol", load("*a = 123"))
356+cannotload("hhi", load(function () error("hhi") end))
357+
358+-- any value is valid for _ENV
359+assert(load("return _ENV", nil, nil, 123)() == 123)
360+
361+
362+-- load when _ENV is not first upvalue
363+local x; XX = 123
364+local function h ()
365+ local y=x -- use 'x', so that it becomes 1st upvalue
366+ return XX -- global name
367+end
368+local d = string.dump(h)
369+x = load(d, "", "b")
370+assert(debug.getupvalue(x, 2) == '_ENV')
371+debug.setupvalue(x, 2, _G)
372+assert(x() == 123)
373+
374+assert(assert(load("return XX + ...", nil, nil, {XX = 13}))(4) == 17)
375+XX = nil
376+
377+-- test generic load with nested functions
378+x = [[
379+ return function (x)
380+ return function (y)
381+ return function (z)
382+ return x+y+z
383+ end
384+ end
385+ end
386+]]
387+a = assert(load(read1(x), "read", "t"))
388+assert(a()(2)(3)(10) == 15)
389+
390+-- repeat the test loading a binary chunk
391+x = string.dump(a)
392+a = assert(load(read1(x), "read", "b"))
393+assert(a()(2)(3)(10) == 15)
394+
395+
396+-- test for dump/undump with upvalues
397+local a, b = 20, 30
398+x = load(string.dump(function (x)
399+ if x == "set" then a = 10+b; b = b+1 else
400+ return a
401+ end
402+end), "", "b", nil)
403+assert(x() == nil)
404+assert(debug.setupvalue(x, 1, "hi") == "a")
405+assert(x() == "hi")
406+assert(debug.setupvalue(x, 2, 13) == "b")
407+assert(not debug.setupvalue(x, 3, 10)) -- only 2 upvalues
408+x("set")
409+assert(x() == 23)
410+x("set")
411+assert(x() == 24)
412+
413+-- test for dump/undump with many upvalues
414+do
415+ local nup = 200 -- maximum number of local variables
416+ local prog = {"local a1"}
417+ for i = 2, nup do prog[#prog + 1] = ", a" .. i end
418+ prog[#prog + 1] = " = 1"
419+ for i = 2, nup do prog[#prog + 1] = ", " .. i end
420+ local sum = 1
421+ prog[#prog + 1] = "; return function () return a1"
422+ for i = 2, nup do prog[#prog + 1] = " + a" .. i; sum = sum + i end
423+ prog[#prog + 1] = " end"
424+ prog = table.concat(prog)
425+ local f = assert(load(prog))()
426+ assert(f() == sum)
427+
428+ f = load(string.dump(f)) -- main chunk now has many upvalues
429+ local a = 10
430+ local h = function () return a end
431+ for i = 1, nup do
432+ debug.upvaluejoin(f, i, h, 1)
433+ end
434+ assert(f() == 10 * nup)
435+end
436+
437+-- test for long method names
438+do
439+ local t = {x = 1}
440+ function t:_012345678901234567890123456789012345678901234567890123456789 ()
441+ return self.x
442+ end
443+ assert(t:_012345678901234567890123456789012345678901234567890123456789() == 1)
444+end
445+
446+
447+-- test for bug in parameter adjustment
448+assert((function () return nil end)(4) == nil)
449+assert((function () local a; return a end)(4) == nil)
450+assert((function (a) return a end)() == nil)
451+
452+
453+print("testing binary chunks")
454+do
455+ local header = string.pack("c4BBc6BBB",
456+ "\27Lua", -- signature
457+ 0x54, -- version 5.4 (0x54)
458+ 0, -- format
459+ "\x19\x93\r\n\x1a\n", -- data
460+ 4, -- size of instruction
461+ string.packsize("j"), -- sizeof(lua integer)
462+ string.packsize("n") -- sizeof(lua number)
463+ )
464+ local c = string.dump(function ()
465+ local a = 1; local b = 3;
466+ local f = function () return a + b + _ENV.c; end -- upvalues
467+ local s1 = "a constant"
468+ local s2 = "another constant"
469+ return a + b * 3
470+ end)
471+
472+ assert(assert(load(c))() == 10)
473+
474+ -- check header
475+ assert(string.sub(c, 1, #header) == header)
476+ -- check LUAC_INT and LUAC_NUM
477+ local ci, cn = string.unpack("jn", c, #header + 1)
478+ assert(ci == 0x5678 and cn == 370.5)
479+
480+ -- corrupted header
481+ for i = 1, #header do
482+ local s = string.sub(c, 1, i - 1) ..
483+ string.char(string.byte(string.sub(c, i, i)) + 1) ..
484+ string.sub(c, i + 1, -1)
485+ assert(#s == #c)
486+ assert(not load(s))
487+ end
488+
489+ -- loading truncated binary chunks
490+ for i = 1, #c - 1 do
491+ local st, msg = load(string.sub(c, 1, i))
492+ assert(not st and string.find(msg, "truncated"))
493+ end
494+end
495+
496+print('OK')
497+return deep
zod/luau/tests/closure.luanew file+272
@@ -0,0 +1,272 @@
1+-- $Id: testes/closure.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print "testing closures"
5+
6+local A,B = 0,{g=10}
7+local function f(x)
8+ local a = {}
9+ for i=1,1000 do
10+ local y = 0
11+ do
12+ a[i] = function () B.g = B.g+1; y = y+x; return y+A end
13+ end
14+ end
15+ local dummy = function () return a[A] end
16+ collectgarbage()
17+ A = 1; assert(dummy() == a[1]); A = 0;
18+ assert(a[1]() == x)
19+ assert(a[3]() == x)
20+ collectgarbage()
21+ assert(B.g == 12)
22+ return a
23+end
24+
25+local a = f(10)
26+-- force a GC in this level
27+local x = {[1] = {}} -- to detect a GC
28+setmetatable(x, {__mode = 'kv'})
29+while x[1] do -- repeat until GC
30+ local a = A..A..A..A -- create garbage
31+ A = A+1
32+end
33+assert(a[1]() == 20+A)
34+assert(a[1]() == 30+A)
35+assert(a[2]() == 10+A)
36+collectgarbage()
37+assert(a[2]() == 20+A)
38+assert(a[2]() == 30+A)
39+assert(a[3]() == 20+A)
40+assert(a[8]() == 10+A)
41+assert(getmetatable(x).__mode == 'kv')
42+assert(B.g == 19)
43+
44+
45+-- testing equality
46+a = {}
47+
48+for i = 1, 5 do a[i] = function (x) return i + a + _ENV end end
49+assert(a[3] ~= a[4] and a[4] ~= a[5])
50+
51+do
52+ local a = function (x) return math.sin(_ENV[x]) end
53+ local function f()
54+ return a
55+ end
56+ assert(f() == f())
57+end
58+
59+
60+-- testing closures with 'for' control variable
61+a = {}
62+for i=1,10 do
63+ a[i] = {set = function(x) i=x end, get = function () return i end}
64+ if i == 3 then break end
65+end
66+assert(a[4] == undef)
67+a[1].set(10)
68+assert(a[2].get() == 2)
69+a[2].set('a')
70+assert(a[3].get() == 3)
71+assert(a[2].get() == 'a')
72+
73+a = {}
74+local t = {"a", "b"}
75+for i = 1, #t do
76+ local k = t[i]
77+ a[i] = {set = function(x, y) i=x; k=y end,
78+ get = function () return i, k end}
79+ if i == 2 then break end
80+end
81+a[1].set(10, 20)
82+local r,s = a[2].get()
83+assert(r == 2 and s == 'b')
84+r,s = a[1].get()
85+assert(r == 10 and s == 20)
86+a[2].set('a', 'b')
87+r,s = a[2].get()
88+assert(r == "a" and s == "b")
89+
90+
91+-- testing closures with 'for' control variable x break
92+local f
93+for i=1,3 do
94+ f = function () return i end
95+ break
96+end
97+assert(f() == 1)
98+
99+for k = 1, #t do
100+ local v = t[k]
101+ f = function () return k, v end
102+ break
103+end
104+assert(({f()})[1] == 1)
105+assert(({f()})[2] == "a")
106+
107+
108+-- testing closure x break x return x errors
109+
110+local b
111+function f(x)
112+ local first = 1
113+ while 1 do
114+ if x == 3 and not first then return end
115+ local a = 'xuxu'
116+ b = function (op, y)
117+ if op == 'set' then
118+ a = x+y
119+ else
120+ return a
121+ end
122+ end
123+ if x == 1 then do break end
124+ elseif x == 2 then return
125+ else if x ~= 3 then error() end
126+ end
127+ first = nil
128+ end
129+end
130+
131+for i=1,3 do
132+ f(i)
133+ assert(b('get') == 'xuxu')
134+ b('set', 10); assert(b('get') == 10+i)
135+ b = nil
136+end
137+
138+pcall(f, 4);
139+assert(b('get') == 'xuxu')
140+b('set', 10); assert(b('get') == 14)
141+
142+
143+local y, w
144+-- testing multi-level closure
145+function f(x)
146+ return function (y)
147+ return function (z) return w+x+y+z end
148+ end
149+end
150+
151+y = f(10)
152+w = 1.345
153+assert(y(20)(30) == 60+w)
154+
155+
156+-- testing closures x break
157+do
158+ local X, Y
159+ local a = math.sin(0)
160+
161+ while a do
162+ local b = 10
163+ X = function () return b end -- closure with upvalue
164+ if a then break end
165+ end
166+
167+ do
168+ local b = 20
169+ Y = function () return b end -- closure with upvalue
170+ end
171+
172+ -- upvalues must be different
173+ assert(X() == 10 and Y() == 20)
174+end
175+
176+
177+-- testing closures x repeat-until
178+
179+local a = {}
180+local i = 1
181+repeat
182+ local x = i
183+ a[i] = function () i = x+1; return x end
184+until i > 10 or a[i]() ~= x
185+assert(i == 11 and a[1]() == 1 and a[3]() == 3 and i == 4)
186+
187+
188+-- testing closures created in 'then' and 'else' parts of 'if's
189+a = {}
190+for i = 1, 10 do
191+ if i % 3 == 0 then
192+ local y = 0
193+ a[i] = function (x) local t = y; y = x; return t end
194+ elseif i % 3 == 1 then
195+ goto L1
196+ error'not here'
197+ ::L1::
198+ local y = 1
199+ a[i] = function (x) local t = y; y = x; return t end
200+ elseif i % 3 == 2 then
201+ local t
202+ goto l4
203+ ::l4a:: a[i] = t; goto l4b
204+ error("should never be here!")
205+ ::l4::
206+ local y = 2
207+ t = function (x) local t = y; y = x; return t end
208+ goto l4a
209+ error("should never be here!")
210+ ::l4b::
211+ end
212+end
213+
214+for i = 1, 10 do
215+ assert(a[i](i * 10) == i % 3 and a[i]() == i * 10)
216+end
217+
218+print'+'
219+
220+
221+-- test for correctly closing upvalues in tail calls of vararg functions
222+local function t ()
223+ local function c(a,b) assert(a=="test" and b=="OK") end
224+ local function v(f, ...) c("test", f() ~= 1 and "FAILED" or "OK") end
225+ local x = 1
226+ return v(function() return x end)
227+end
228+t()
229+
230+
231+-- test for debug manipulation of upvalues
232+local debug = require'debug'
233+
234+local foo1, foo2, foo3
235+do
236+ local a , b, c = 3, 5, 7
237+ foo1 = function () return a+b end;
238+ foo2 = function () return b+a end;
239+ do
240+ local a = 10
241+ foo3 = function () return a+b end;
242+ end
243+end
244+
245+assert(debug.upvalueid(foo1, 1))
246+assert(debug.upvalueid(foo1, 2))
247+assert(not debug.upvalueid(foo1, 3))
248+assert(debug.upvalueid(foo1, 1) == debug.upvalueid(foo2, 2))
249+assert(debug.upvalueid(foo1, 2) == debug.upvalueid(foo2, 1))
250+assert(debug.upvalueid(foo3, 1))
251+assert(debug.upvalueid(foo1, 1) ~= debug.upvalueid(foo3, 1))
252+assert(debug.upvalueid(foo1, 2) == debug.upvalueid(foo3, 2))
253+
254+assert(debug.upvalueid(string.gmatch("x", "x"), 1) ~= nil)
255+
256+assert(foo1() == 3 + 5 and foo2() == 5 + 3)
257+debug.upvaluejoin(foo1, 2, foo2, 2)
258+assert(foo1() == 3 + 3 and foo2() == 5 + 3)
259+assert(foo3() == 10 + 5)
260+debug.upvaluejoin(foo3, 2, foo2, 1)
261+assert(foo3() == 10 + 5)
262+debug.upvaluejoin(foo3, 2, foo2, 2)
263+assert(foo3() == 10 + 3)
264+
265+assert(not pcall(debug.upvaluejoin, foo1, 3, foo2, 1))
266+assert(not pcall(debug.upvaluejoin, foo1, 1, foo2, 3))
267+assert(not pcall(debug.upvaluejoin, foo1, 0, foo2, 1))
268+assert(not pcall(debug.upvaluejoin, print, 1, foo2, 1))
269+assert(not pcall(debug.upvaluejoin, {}, 1, foo2, 1))
270+assert(not pcall(debug.upvaluejoin, foo1, 1, print, 1))
271+
272+print'OK'
zod/luau/tests/code.luanew file+449
@@ -0,0 +1,449 @@
1+-- $Id: testes/code.lua $
2+-- See Copyright Notice in file all.lua
3+
4+if T==nil then
5+ (Message or print)('\n >>> testC not active: skipping opcode tests <<<\n')
6+ return
7+end
8+print "testing code generation and optimizations"
9+
10+-- to test constant propagation
11+local k0aux <const> = 0
12+local k0 <const> = k0aux
13+local k1 <const> = 1
14+local k3 <const> = 3
15+local k6 <const> = k3 + (k3 << k0)
16+local kFF0 <const> = 0xFF0
17+local k3_78 <const> = 3.78
18+local x, k3_78_4 <const> = 10, k3_78 / 4
19+assert(x == 10)
20+
21+local kx <const> = "x"
22+
23+local kTrue <const> = true
24+local kFalse <const> = false
25+
26+local kNil <const> = nil
27+
28+-- this code gave an error for the code checker
29+do
30+ local function f (a)
31+ for k,v,w in a do end
32+ end
33+end
34+
35+
36+-- testing reuse in constant table
37+local function checkKlist (func, list)
38+ local k = T.listk(func)
39+ assert(#k == #list)
40+ for i = 1, #k do
41+ assert(k[i] == list[i] and math.type(k[i]) == math.type(list[i]))
42+ end
43+end
44+
45+local function foo ()
46+ local a
47+ a = k3;
48+ a = 0; a = 0.0; a = -7 + 7
49+ a = k3_78/4; a = k3_78_4
50+ a = -k3_78/4; a = k3_78/4; a = -3.78/4
51+ a = -3.79/4; a = 0.0; a = -0;
52+ a = k3; a = 3.0; a = 3; a = 3.0
53+end
54+
55+checkKlist(foo, {3.78/4, -3.78/4, -3.79/4})
56+
57+
58+foo = function (f, a)
59+ f(100 * 1000)
60+ f(100.0 * 1000)
61+ f(-100 * 1000)
62+ f(-100 * 1000.0)
63+ f(100000)
64+ f(100000.0)
65+ f(-100000)
66+ f(-100000.0)
67+ end
68+
69+checkKlist(foo, {100000, 100000.0, -100000, -100000.0})
70+
71+
72+-- floats x integers
73+foo = function (t, a)
74+ t[a] = 1; t[a] = 1.0
75+ t[a] = 1; t[a] = 1.0
76+ t[a] = 2; t[a] = 2.0
77+ t[a] = 0; t[a] = 0.0
78+ t[a] = 1; t[a] = 1.0
79+ t[a] = 2; t[a] = 2.0
80+ t[a] = 0; t[a] = 0.0
81+end
82+
83+checkKlist(foo, {1, 1.0, 2, 2.0, 0, 0.0})
84+
85+
86+-- testing opcodes
87+
88+-- check that 'f' opcodes match '...'
89+local function check (f, ...)
90+ local arg = {...}
91+ local c = T.listcode(f)
92+ for i=1, #arg do
93+ local opcode = string.match(c[i], "%u%w+")
94+ -- print(arg[i], opcode)
95+ assert(arg[i] == opcode)
96+ end
97+ assert(c[#arg+2] == undef)
98+end
99+
100+
101+-- check that 'f' opcodes match '...' and that 'f(p) == r'.
102+local function checkR (f, p, r, ...)
103+ local r1 = f(p)
104+ assert(r == r1 and math.type(r) == math.type(r1))
105+ check(f, ...)
106+end
107+
108+
109+-- check that 'a' and 'b' has the same opcodes
110+local function checkequal (a, b)
111+ a = T.listcode(a)
112+ b = T.listcode(b)
113+ assert(#a == #b)
114+ for i = 1, #a do
115+ a[i] = string.gsub(a[i], '%b()', '') -- remove line number
116+ b[i] = string.gsub(b[i], '%b()', '') -- remove line number
117+ assert(a[i] == b[i])
118+ end
119+end
120+
121+
122+-- some basic instructions
123+check(function () -- function does not create upvalues
124+ (function () end){f()}
125+end, 'CLOSURE', 'NEWTABLE', 'EXTRAARG', 'GETTABUP', 'CALL',
126+ 'SETLIST', 'CALL', 'RETURN0')
127+
128+check(function (x) -- function creates upvalues
129+ (function () return x end){f()}
130+end, 'CLOSURE', 'NEWTABLE', 'EXTRAARG', 'GETTABUP', 'CALL',
131+ 'SETLIST', 'CALL', 'RETURN')
132+
133+
134+-- sequence of LOADNILs
135+check(function ()
136+ local kNil <const> = nil
137+ local a,b,c
138+ local d; local e;
139+ local f,g,h;
140+ d = nil; d=nil; b=nil; a=kNil; c=nil;
141+end, 'LOADNIL', 'RETURN0')
142+
143+check(function ()
144+ local a,b,c,d = 1,1,1,1
145+ d=nil;c=nil;b=nil;a=nil
146+end, 'LOADI', 'LOADI', 'LOADI', 'LOADI', 'LOADNIL', 'RETURN0')
147+
148+do
149+ local a,b,c,d = 1,1,1,1
150+ d=nil;c=nil;b=nil;a=nil
151+ assert(a == nil and b == nil and c == nil and d == nil)
152+end
153+
154+
155+-- single return
156+check (function (a,b,c) return a end, 'RETURN1')
157+
158+
159+-- infinite loops
160+check(function () while kTrue do local a = -1 end end,
161+'LOADI', 'JMP', 'RETURN0')
162+
163+check(function () while 1 do local a = -1 end end,
164+'LOADI', 'JMP', 'RETURN0')
165+
166+check(function () repeat local x = 1 until true end,
167+'LOADI', 'RETURN0')
168+
169+
170+-- concat optimization
171+check(function (a,b,c,d) return a..b..c..d end,
172+ 'MOVE', 'MOVE', 'MOVE', 'MOVE', 'CONCAT', 'RETURN1')
173+
174+-- not
175+check(function () return not not nil end, 'LOADFALSE', 'RETURN1')
176+check(function () return not not kFalse end, 'LOADFALSE', 'RETURN1')
177+check(function () return not not true end, 'LOADTRUE', 'RETURN1')
178+check(function () return not not k3 end, 'LOADTRUE', 'RETURN1')
179+
180+-- direct access to locals
181+check(function ()
182+ local a,b,c,d
183+ a = b*a
184+ c.x, a[b] = -((a + d/b - a[b]) ^ a.x), b
185+end,
186+ 'LOADNIL',
187+ 'MUL', 'MMBIN',
188+ 'DIV', 'MMBIN', 'ADD', 'MMBIN', 'GETTABLE', 'SUB', 'MMBIN',
189+ 'GETFIELD', 'POW', 'MMBIN', 'UNM', 'SETTABLE', 'SETFIELD', 'RETURN0')
190+
191+
192+-- direct access to constants
193+check(function ()
194+ local a,b
195+ local c = kNil
196+ a[kx] = 3.2
197+ a.x = b
198+ a[b] = 'x'
199+end,
200+ 'LOADNIL', 'SETFIELD', 'SETFIELD', 'SETTABLE', 'RETURN0')
201+
202+-- "get/set table" with numeric indices
203+check(function (a)
204+ local k255 <const> = 255
205+ a[1] = a[100]
206+ a[k255] = a[256]
207+ a[256] = 5
208+end,
209+ 'GETI', 'SETI',
210+ 'LOADI', 'GETTABLE', 'SETI',
211+ 'LOADI', 'SETTABLE', 'RETURN0')
212+
213+check(function ()
214+ local a,b
215+ a = a - a
216+ b = a/a
217+ b = 5-4
218+end,
219+ 'LOADNIL', 'SUB', 'MMBIN', 'DIV', 'MMBIN', 'LOADI', 'RETURN0')
220+
221+check(function ()
222+ local a,b
223+ a[kTrue] = false
224+end,
225+ 'LOADNIL', 'LOADTRUE', 'SETTABLE', 'RETURN0')
226+
227+
228+-- equalities
229+checkR(function (a) if a == 1 then return 2 end end, 1, 2,
230+ 'EQI', 'JMP', 'LOADI', 'RETURN1')
231+
232+checkR(function (a) if -4.0 == a then return 2 end end, -4, 2,
233+ 'EQI', 'JMP', 'LOADI', 'RETURN1')
234+
235+checkR(function (a) if a == "hi" then return 2 end end, 10, nil,
236+ 'EQK', 'JMP', 'LOADI', 'RETURN1')
237+
238+checkR(function (a) if a == 10000 then return 2 end end, 1, nil,
239+ 'EQK', 'JMP', 'LOADI', 'RETURN1') -- number too large
240+
241+checkR(function (a) if -10000 == a then return 2 end end, -10000, 2,
242+ 'EQK', 'JMP', 'LOADI', 'RETURN1') -- number too large
243+
244+-- comparisons
245+
246+checkR(function (a) if -10 <= a then return 2 end end, -10, 2,
247+ 'GEI', 'JMP', 'LOADI', 'RETURN1')
248+
249+checkR(function (a) if 128.0 > a then return 2 end end, 129, nil,
250+ 'LTI', 'JMP', 'LOADI', 'RETURN1')
251+
252+checkR(function (a) if -127.0 < a then return 2 end end, -127, nil,
253+ 'GTI', 'JMP', 'LOADI', 'RETURN1')
254+
255+checkR(function (a) if 10 < a then return 2 end end, 11, 2,
256+ 'GTI', 'JMP', 'LOADI', 'RETURN1')
257+
258+checkR(function (a) if 129 < a then return 2 end end, 130, 2,
259+ 'LOADI', 'LT', 'JMP', 'LOADI', 'RETURN1')
260+
261+checkR(function (a) if a >= 23.0 then return 2 end end, 25, 2,
262+ 'GEI', 'JMP', 'LOADI', 'RETURN1')
263+
264+checkR(function (a) if a >= 23.1 then return 2 end end, 0, nil,
265+ 'LOADK', 'LE', 'JMP', 'LOADI', 'RETURN1')
266+
267+checkR(function (a) if a > 2300.0 then return 2 end end, 0, nil,
268+ 'LOADF', 'LT', 'JMP', 'LOADI', 'RETURN1')
269+
270+
271+-- constant folding
272+local function checkK (func, val)
273+ check(func, 'LOADK', 'RETURN1')
274+ checkKlist(func, {val})
275+ assert(func() == val)
276+end
277+
278+local function checkI (func, val)
279+ check(func, 'LOADI', 'RETURN1')
280+ checkKlist(func, {})
281+ assert(func() == val)
282+end
283+
284+local function checkF (func, val)
285+ check(func, 'LOADF', 'RETURN1')
286+ checkKlist(func, {})
287+ assert(func() == val)
288+end
289+
290+checkF(function () return 0.0 end, 0.0)
291+checkI(function () return k0 end, 0)
292+checkI(function () return -k0//1 end, 0)
293+checkK(function () return 3^-1 end, 1/3)
294+checkK(function () return (1 + 1)^(50 + 50) end, 2^100)
295+checkK(function () return (-2)^(31 - 2) end, -0x20000000 + 0.0)
296+checkF(function () return (-k3^0 + 5) // 3.0 end, 1.0)
297+checkI(function () return -k3 % 5 end, 2)
298+checkF(function () return -((2.0^8 + -(-1)) % 8)/2 * 4 - 3 end, -5.0)
299+checkF(function () return -((2^8 + -(-1)) % 8)//2 * 4 - 3 end, -7.0)
300+checkI(function () return 0xF0.0 | 0xCC.0 ~ 0xAA & 0xFD end, 0xF4)
301+checkI(function () return ~(~kFF0 | kFF0) end, 0)
302+checkI(function () return ~~-1024.0 end, -1024)
303+checkI(function () return ((100 << k6) << -4) >> 2 end, 100)
304+
305+-- borders around MAXARG_sBx ((((1 << 17) - 1) >> 1) == 65535)
306+local a = 17; local sbx = ((1 << a) - 1) >> 1 -- avoid folding
307+local border <const> = 65535
308+checkI(function () return border end, sbx)
309+checkI(function () return -border end, -sbx)
310+checkI(function () return border + 1 end, sbx + 1)
311+checkK(function () return border + 2 end, sbx + 2)
312+checkK(function () return -(border + 1) end, -(sbx + 1))
313+
314+local border <const> = 65535.0
315+checkF(function () return border end, sbx + 0.0)
316+checkF(function () return -border end, -sbx + 0.0)
317+checkF(function () return border + 1 end, (sbx + 1.0))
318+checkK(function () return border + 2 end, (sbx + 2.0))
319+checkK(function () return -(border + 1) end, -(sbx + 1.0))
320+
321+
322+-- immediate operands
323+checkR(function (x) return x + k1 end, 10, 11, 'ADDI', 'MMBINI', 'RETURN1')
324+checkR(function (x) return x - 127 end, 10, -117, 'ADDI', 'MMBINI', 'RETURN1')
325+checkR(function (x) return 128 + x end, 0.0, 128.0,
326+ 'ADDI', 'MMBINI', 'RETURN1')
327+checkR(function (x) return x * -127 end, -1.0, 127.0,
328+ 'MULK', 'MMBINK', 'RETURN1')
329+checkR(function (x) return 20 * x end, 2, 40, 'MULK', 'MMBINK', 'RETURN1')
330+checkR(function (x) return x ^ -2 end, 2, 0.25, 'POWK', 'MMBINK', 'RETURN1')
331+checkR(function (x) return x / 40 end, 40, 1.0, 'DIVK', 'MMBINK', 'RETURN1')
332+checkR(function (x) return x // 1 end, 10.0, 10.0,
333+ 'IDIVK', 'MMBINK', 'RETURN1')
334+checkR(function (x) return x % (100 - 10) end, 91, 1,
335+ 'MODK', 'MMBINK', 'RETURN1')
336+checkR(function (x) return k1 << x end, 3, 8, 'SHLI', 'MMBINI', 'RETURN1')
337+checkR(function (x) return x << 127 end, 10, 0, 'SHRI', 'MMBINI', 'RETURN1')
338+checkR(function (x) return x << -127 end, 10, 0, 'SHRI', 'MMBINI', 'RETURN1')
339+checkR(function (x) return x >> 128 end, 8, 0, 'SHRI', 'MMBINI', 'RETURN1')
340+checkR(function (x) return x >> -127 end, 8, 0, 'SHRI', 'MMBINI', 'RETURN1')
341+checkR(function (x) return x & 1 end, 9, 1, 'BANDK', 'MMBINK', 'RETURN1')
342+checkR(function (x) return 10 | x end, 1, 11, 'BORK', 'MMBINK', 'RETURN1')
343+checkR(function (x) return -10 ~ x end, -1, 9, 'BXORK', 'MMBINK', 'RETURN1')
344+
345+-- K operands in arithmetic operations
346+checkR(function (x) return x + 0.0 end, 1, 1.0, 'ADDK', 'MMBINK', 'RETURN1')
347+-- check(function (x) return 128 + x end, 'ADDK', 'MMBINK', 'RETURN1')
348+checkR(function (x) return x * -10000 end, 2, -20000,
349+ 'MULK', 'MMBINK', 'RETURN1')
350+-- check(function (x) return 20 * x end, 'MULK', 'MMBINK', 'RETURN1')
351+checkR(function (x) return x ^ 0.5 end, 4, 2.0, 'POWK', 'MMBINK', 'RETURN1')
352+checkR(function (x) return x / 2.0 end, 4, 2.0, 'DIVK', 'MMBINK', 'RETURN1')
353+checkR(function (x) return x // 10000 end, 10000, 1,
354+ 'IDIVK', 'MMBINK', 'RETURN1')
355+checkR(function (x) return x % (100.0 - 10) end, 91, 1.0,
356+ 'MODK', 'MMBINK', 'RETURN1')
357+
358+-- no foldings (and immediate operands)
359+check(function () return -0.0 end, 'LOADF', 'UNM', 'RETURN1')
360+check(function () return k3/0 end, 'LOADI', 'DIVK', 'MMBINK', 'RETURN1')
361+check(function () return 0%0 end, 'LOADI', 'MODK', 'MMBINK', 'RETURN1')
362+check(function () return -4//0 end, 'LOADI', 'IDIVK', 'MMBINK', 'RETURN1')
363+check(function (x) return x >> 2.0 end, 'LOADF', 'SHR', 'MMBIN', 'RETURN1')
364+check(function (x) return x << 128 end, 'LOADI', 'SHL', 'MMBIN', 'RETURN1')
365+check(function (x) return x & 2.0 end, 'LOADF', 'BAND', 'MMBIN', 'RETURN1')
366+
367+-- basic 'for' loops
368+check(function () for i = -10, 10.5 do end end,
369+'LOADI', 'LOADK', 'LOADI', 'FORPREP', 'FORLOOP', 'RETURN0')
370+check(function () for i = 0xfffffff, 10.0, 1 do end end,
371+'LOADK', 'LOADF', 'LOADI', 'FORPREP', 'FORLOOP', 'RETURN0')
372+
373+-- bug in constant folding for 5.1
374+check(function () return -nil end, 'LOADNIL', 'UNM', 'RETURN1')
375+
376+
377+check(function ()
378+ local a,b,c
379+ b[c], a = c, b
380+ b[a], a = c, b
381+ a, b = c, a
382+ a = a
383+end,
384+ 'LOADNIL',
385+ 'MOVE', 'MOVE', 'SETTABLE',
386+ 'MOVE', 'MOVE', 'MOVE', 'SETTABLE',
387+ 'MOVE', 'MOVE', 'MOVE',
388+ -- no code for a = a
389+ 'RETURN0')
390+
391+
392+-- x == nil , x ~= nil
393+-- checkequal(function (b) if (a==nil) then a=1 end; if a~=nil then a=1 end end,
394+-- function () if (a==9) then a=1 end; if a~=9 then a=1 end end)
395+
396+-- check(function () if a==nil then a='a' end end,
397+-- 'GETTABUP', 'EQ', 'JMP', 'SETTABUP', 'RETURN')
398+
399+do -- tests for table access in upvalues
400+ local t
401+ check(function () t[kx] = t.y end, 'GETTABUP', 'SETTABUP')
402+ check(function (a) t[a()] = t[a()] end,
403+ 'MOVE', 'CALL', 'GETUPVAL', 'MOVE', 'CALL',
404+ 'GETUPVAL', 'GETTABLE', 'SETTABLE')
405+end
406+
407+-- de morgan
408+checkequal(function () local a; if not (a or b) then b=a end end,
409+ function () local a; if (not a and not b) then b=a end end)
410+
411+checkequal(function (l) local a; return 0 <= a and a <= l end,
412+ function (l) local a; return not (not(a >= 0) or not(a <= l)) end)
413+
414+
415+-- if-break optimizations
416+check(function (a, b)
417+ while a do
418+ if b then break else a = a + 1 end
419+ end
420+ end,
421+'TEST', 'JMP', 'TEST', 'JMP', 'ADDI', 'MMBINI', 'JMP', 'RETURN0')
422+
423+checkequal(function () return 6 or true or nil end,
424+ function () return k6 or kTrue or kNil end)
425+
426+checkequal(function () return 6 and true or nil end,
427+ function () return k6 and kTrue or kNil end)
428+
429+
430+do -- string constants
431+ local k0 <const> = "00000000000000000000000000000000000000000000000000"
432+ local function f1 ()
433+ local k <const> = k0
434+ return function ()
435+ return function () return k end
436+ end
437+ end
438+
439+ local f2 = f1()
440+ local f3 = f2()
441+ assert(f3() == k0)
442+ checkK(f3, k0)
443+ -- string is not needed by other functions
444+ assert(T.listk(f1)[1] == nil)
445+ assert(T.listk(f2)[1] == nil)
446+end
447+
448+print 'OK'
449+
zod/luau/tests/constructs.luanew file+406
@@ -0,0 +1,406 @@
1+-- $Id: testes/constructs.lua $
2+-- See Copyright Notice in file all.lua
3+
4+;;print "testing syntax";;
5+
6+local debug = require "debug"
7+
8+
9+local function checkload (s, msg)
10+ assert(string.find(select(2, load(s)), msg))
11+end
12+
13+-- testing semicollons
14+local a
15+do ;;; end
16+; do ; a = 3; assert(a == 3) end;
17+;
18+
19+
20+-- invalid operations should not raise errors when not executed
21+if false then a = 3 // 0; a = 0 % 0 end
22+
23+
24+-- testing priorities
25+
26+assert(2^3^2 == 2^(3^2));
27+assert(2^3*4 == (2^3)*4);
28+assert(2.0^-2 == 1/4 and -2^- -2 == - - -4);
29+assert(not nil and 2 and not(2>3 or 3<2));
30+assert(-3-1-5 == 0+0-9);
31+assert(-2^2 == -4 and (-2)^2 == 4 and 2*2-3-1 == 0);
32+assert(-3%5 == 2 and -3+5 == 2)
33+assert(2*1+3/3 == 3 and 1+2 .. 3*1 == "33");
34+assert(not(2+1 > 3*1) and "a".."b" > "a");
35+
36+assert(0xF0 | 0xCC ~ 0xAA & 0xFD == 0xF4)
37+assert(0xFD & 0xAA ~ 0xCC | 0xF0 == 0xF4)
38+assert(0xF0 & 0x0F + 1 == 0x10)
39+
40+assert(3^4//2^3//5 == 2)
41+
42+assert(-3+4*5//2^3^2//9+4%10/3 == (-3)+(((4*5)//(2^(3^2)))//9)+((4%10)/3))
43+
44+assert(not ((true or false) and nil))
45+assert( true or false and nil)
46+
47+-- old bug
48+assert((((1 or false) and true) or false) == true)
49+assert((((nil and true) or false) and true) == false)
50+
51+local a,b = 1,nil;
52+assert(-(1 or 2) == -1 and (1 and 2)+(-1.25 or -4) == 0.75);
53+local x = ((b or a)+1 == 2 and (10 or a)+1 == 11); assert(x);
54+x = (((2<3) or 1) == true and (2<3 and 4) == 4); assert(x);
55+
56+local x, y = 1, 2;
57+assert((x>y) and x or y == 2);
58+x,y=2,1;
59+assert((x>y) and x or y == 2);
60+
61+assert(1234567890 == tonumber('1234567890') and 1234567890+1 == 1234567891)
62+
63+do -- testing operators with diffent kinds of constants
64+ -- operands to consider:
65+ -- * fit in register
66+ -- * constant doesn't fit in register
67+ -- * floats with integral values
68+ local operand = {3, 100, 5.0, -10, -5.0, 10000, -10000}
69+ local operator = {"+", "-", "*", "/", "//", "%", "^",
70+ "&", "|", "^", "<<", ">>",
71+ "==", "~=", "<", ">", "<=", ">=",}
72+ for _, op in ipairs(operator) do
73+ local f = assert(load(string.format([[return function (x,y)
74+ return x %s y
75+ end]], op)))();
76+ for _, o1 in ipairs(operand) do
77+ for _, o2 in ipairs(operand) do
78+ local gab = f(o1, o2)
79+
80+ _ENV.XX = o1
81+ local code = string.format("return XX %s %s", op, o2)
82+ local res = assert(load(code))()
83+ assert(res == gab)
84+
85+ _ENV.XX = o2
86+ code = string.format("return (%s) %s XX", o1, op)
87+ res = assert(load(code))()
88+ assert(res == gab)
89+
90+ code = string.format("return (%s) %s %s", o1, op, o2)
91+ res = assert(load(code))()
92+ assert(res == gab)
93+ end
94+ end
95+ end
96+ _ENV.XX = nil
97+end
98+
99+
100+-- silly loops
101+repeat until 1; repeat until true;
102+while false do end; while nil do end;
103+
104+do -- test old bug (first name could not be an `upvalue')
105+ local a; local function f(x) x={a=1}; x={x=1}; x={G=1} end
106+end
107+
108+
109+do -- bug since 5.4.0
110+ -- create code with a table using more than 256 constants
111+ local code = {"local x = {"}
112+ for i = 1, 257 do
113+ code[#code + 1] = i .. ".1,"
114+ end
115+ code[#code + 1] = "};"
116+ code = table.concat(code)
117+
118+ -- add "ret" to the end of that code and checks that
119+ -- it produces the expected value "val"
120+ local function check (ret, val)
121+ local code = code .. ret
122+ code = load(code)
123+ assert(code() == val)
124+ end
125+
126+ check("return (1 ~ (2 or 3))", 1 ~ 2)
127+ check("return (1 | (2 or 3))", 1 | 2)
128+ check("return (1 + (2 or 3))", 1 + 2)
129+ check("return (1 << (2 or 3))", 1 << 2)
130+end
131+
132+
133+local function f (i)
134+ if type(i) ~= 'number' then return i,'jojo'; end;
135+ if i > 0 then return i, f(i-1); end;
136+end
137+
138+x = {f(3), f(5), f(10);};
139+assert(x[1] == 3 and x[2] == 5 and x[3] == 10 and x[4] == 9 and x[12] == 1);
140+assert(x[nil] == nil)
141+x = {f'alo', f'xixi', nil};
142+assert(x[1] == 'alo' and x[2] == 'xixi' and x[3] == nil);
143+x = {f'alo'..'xixi'};
144+assert(x[1] == 'aloxixi')
145+x = {f{}}
146+assert(x[2] == 'jojo' and type(x[1]) == 'table')
147+
148+
149+local f = function (i)
150+ if i < 10 then return 'a';
151+ elseif i < 20 then return 'b';
152+ elseif i < 30 then return 'c';
153+ end;
154+end
155+
156+assert(f(3) == 'a' and f(12) == 'b' and f(26) == 'c' and f(100) == nil)
157+
158+for i=1,1000 do break; end;
159+local n=100;
160+local i=3;
161+local t = {};
162+local a=nil
163+while not a do
164+ a=0; for i=1,n do for i=i,1,-1 do a=a+1; t[i]=1; end; end;
165+end
166+assert(a == n*(n+1)/2 and i==3);
167+assert(t[1] and t[n] and not t[0] and not t[n+1])
168+
169+function f(b)
170+ local x = 1;
171+ repeat
172+ local a;
173+ if b==1 then local b=1; x=10; break
174+ elseif b==2 then x=20; break;
175+ elseif b==3 then x=30;
176+ else local a,b,c,d=math.sin(1); x=x+1;
177+ end
178+ until x>=12;
179+ return x;
180+end;
181+
182+assert(f(1) == 10 and f(2) == 20 and f(3) == 30 and f(4)==12)
183+
184+
185+local f = function (i)
186+ if i < 10 then return 'a'
187+ elseif i < 20 then return 'b'
188+ elseif i < 30 then return 'c'
189+ else return 8
190+ end
191+end
192+
193+assert(f(3) == 'a' and f(12) == 'b' and f(26) == 'c' and f(100) == 8)
194+
195+local a, b = nil, 23
196+x = {f(100)*2+3 or a, a or b+2}
197+assert(x[1] == 19 and x[2] == 25)
198+x = {f=2+3 or a, a = b+2}
199+assert(x.f == 5 and x.a == 25)
200+
201+a={y=1}
202+x = {a.y}
203+assert(x[1] == 1)
204+
205+local function f (i)
206+ while 1 do
207+ if i>0 then i=i-1;
208+ else return; end;
209+ end;
210+end;
211+
212+local function g(i)
213+ while 1 do
214+ if i>0 then i=i-1
215+ else return end
216+ end
217+end
218+
219+f(10); g(10);
220+
221+do
222+ function f () return 1,2,3; end
223+ local a, b, c = f();
224+ assert(a==1 and b==2 and c==3)
225+ a, b, c = (f());
226+ assert(a==1 and b==nil and c==nil)
227+end
228+
229+local a,b = 3 and f();
230+assert(a==1 and b==nil)
231+
232+function g() f(); return; end;
233+assert(g() == nil)
234+function g() return nil or f() end
235+a,b = g()
236+assert(a==1 and b==nil)
237+
238+print'+';
239+
240+do -- testing constants
241+ local prog <const> = [[local x <XXX> = 10]]
242+ checkload(prog, "unknown attribute 'XXX'")
243+
244+ checkload([[local xxx <const> = 20; xxx = 10]],
245+ ":1: attempt to assign to const variable 'xxx'")
246+
247+ checkload([[
248+ local xx;
249+ local xxx <const> = 20;
250+ local yyy;
251+ local function foo ()
252+ local abc = xx + yyy + xxx;
253+ return function () return function () xxx = yyy end end
254+ end
255+ ]], ":6: attempt to assign to const variable 'xxx'")
256+
257+ checkload([[
258+ local x <close> = nil
259+ x = io.open()
260+ ]], ":2: attempt to assign to const variable 'x'")
261+end
262+
263+f = [[
264+return function ( a , b , c , d , e )
265+ local x = a >= b or c or ( d and e ) or nil
266+ return x
267+end , { a = 1 , b = 2 >= 1 , } or { 1 };
268+]]
269+f = string.gsub(f, "%s+", "\n"); -- force a SETLINE between opcodes
270+f,a = load(f)();
271+assert(a.a == 1 and a.b)
272+
273+function g (a,b,c,d,e)
274+ if not (a>=b or c or d and e or nil) then return 0; else return 1; end;
275+end
276+
277+local function h (a,b,c,d,e)
278+ while (a>=b or c or (d and e) or nil) do return 1; end;
279+ return 0;
280+end;
281+
282+assert(f(2,1) == true and g(2,1) == 1 and h(2,1) == 1)
283+assert(f(1,2,'a') == 'a' and g(1,2,'a') == 1 and h(1,2,'a') == 1)
284+assert(f(1,2,'a')
285+~= -- force SETLINE before nil
286+nil, "")
287+assert(f(1,2,'a') == 'a' and g(1,2,'a') == 1 and h(1,2,'a') == 1)
288+assert(f(1,2,nil,1,'x') == 'x' and g(1,2,nil,1,'x') == 1 and
289+ h(1,2,nil,1,'x') == 1)
290+assert(f(1,2,nil,nil,'x') == nil and g(1,2,nil,nil,'x') == 0 and
291+ h(1,2,nil,nil,'x') == 0)
292+assert(f(1,2,nil,1,nil) == nil and g(1,2,nil,1,nil) == 0 and
293+ h(1,2,nil,1,nil) == 0)
294+
295+assert(1 and 2<3 == true and 2<3 and 'a'<'b' == true)
296+x = 2<3 and not 3; assert(x==false)
297+x = 2<1 or (2>1 and 'a'); assert(x=='a')
298+
299+
300+do
301+ local a; if nil then a=1; else a=2; end; -- this nil comes as PUSHNIL 2
302+ assert(a==2)
303+end
304+
305+local function F (a)
306+ assert(debug.getinfo(1, "n").name == 'F')
307+ return a,2,3
308+end
309+
310+a,b = F(1)~=nil; assert(a == true and b == nil);
311+a,b = F(nil)==nil; assert(a == true and b == nil)
312+
313+----------------------------------------------------------------
314+------------------------------------------------------------------
315+
316+-- sometimes will be 0, sometimes will not...
317+_ENV.GLOB1 = math.random(0, 1)
318+
319+-- basic expressions with their respective values
320+local basiccases = {
321+ {"nil", nil},
322+ {"false", false},
323+ {"true", true},
324+ {"10", 10},
325+ {"(0==_ENV.GLOB1)", 0 == _ENV.GLOB1},
326+}
327+
328+local prog
329+
330+if _ENV.GLOB1 == 0 then
331+ basiccases[2][1] = "F" -- constant false
332+
333+ prog = [[
334+ local F <const> = false
335+ if %s then IX = true end
336+ return %s
337+]]
338+else
339+ basiccases[4][1] = "k10" -- constant 10
340+
341+ prog = [[
342+ local k10 <const> = 10
343+ if %s then IX = true end
344+ return %s
345+ ]]
346+end
347+
348+print('testing short-circuit optimizations (' .. _ENV.GLOB1 .. ')')
349+
350+
351+-- operators with their respective values
352+local binops <const> = {
353+ {" and ", function (a,b) if not a then return a else return b end end},
354+ {" or ", function (a,b) if a then return a else return b end end},
355+}
356+
357+local cases <const> = {}
358+
359+-- creates all combinations of '(cases[i] op cases[n-i])' plus
360+-- 'not(cases[i] op cases[n-i])' (syntax + value)
361+local function createcases (n)
362+ local res = {}
363+ for i = 1, n - 1 do
364+ for _, v1 in ipairs(cases[i]) do
365+ for _, v2 in ipairs(cases[n - i]) do
366+ for _, op in ipairs(binops) do
367+ local t = {
368+ "(" .. v1[1] .. op[1] .. v2[1] .. ")",
369+ op[2](v1[2], v2[2])
370+ }
371+ res[#res + 1] = t
372+ res[#res + 1] = {"not" .. t[1], not t[2]}
373+ end
374+ end
375+ end
376+ end
377+ return res
378+end
379+
380+-- do not do too many combinations for soft tests
381+local level = _soft and 3 or 4
382+
383+cases[1] = basiccases
384+for i = 2, level do cases[i] = createcases(i) end
385+print("+")
386+
387+local i = 0
388+for n = 1, level do
389+ for _, v in pairs(cases[n]) do
390+ local s = v[1]
391+ local p = load(string.format(prog, s, s), "")
392+ IX = false
393+ assert(p() == v[2] and IX == not not v[2])
394+ i = i + 1
395+ if i % 60000 == 0 then print('+') end
396+ end
397+end
398+IX = nil
399+_G.GLOB1 = nil
400+------------------------------------------------------------------
401+
402+-- testing some syntax errors (chosen through 'gcov')
403+checkload("for x do", "expected")
404+checkload("x:call", "expected")
405+
406+print'OK'
zod/luau/tests/coroutine.luanew file+1156
@@ -0,0 +1,1156 @@
1+-- $Id: testes/coroutine.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print "testing coroutines"
5+
6+local debug = require'debug'
7+
8+local f
9+
10+local main, ismain = coroutine.running()
11+assert(type(main) == "thread" and ismain)
12+assert(not coroutine.resume(main))
13+assert(not coroutine.isyieldable(main) and not coroutine.isyieldable())
14+assert(not pcall(coroutine.yield))
15+
16+
17+-- trivial errors
18+assert(not pcall(coroutine.resume, 0))
19+assert(not pcall(coroutine.status, 0))
20+
21+
22+-- tests for multiple yield/resume arguments
23+
24+local function eqtab (t1, t2)
25+ assert(#t1 == #t2)
26+ for i = 1, #t1 do
27+ local v = t1[i]
28+ assert(t2[i] == v)
29+ end
30+end
31+
32+_G.x = nil -- declare x
33+_G.f = nil -- declare f
34+local function foo (a, ...)
35+ local x, y = coroutine.running()
36+ assert(x == f and y == false)
37+ -- next call should not corrupt coroutine (but must fail,
38+ -- as it attempts to resume the running coroutine)
39+ assert(coroutine.resume(f) == false)
40+ assert(coroutine.status(f) == "running")
41+ local arg = {...}
42+ assert(coroutine.isyieldable(x))
43+ for i=1,#arg do
44+ _G.x = {coroutine.yield(table.unpack(arg[i]))}
45+ end
46+ return table.unpack(a)
47+end
48+
49+f = coroutine.create(foo)
50+assert(coroutine.isyieldable(f))
51+assert(type(f) == "thread" and coroutine.status(f) == "suspended")
52+assert(string.find(tostring(f), "thread"))
53+local s,a,b,c,d
54+s,a,b,c,d = coroutine.resume(f, {1,2,3}, {}, {1}, {'a', 'b', 'c'})
55+assert(coroutine.isyieldable(f))
56+assert(s and a == nil and coroutine.status(f) == "suspended")
57+s,a,b,c,d = coroutine.resume(f)
58+eqtab(_G.x, {})
59+assert(s and a == 1 and b == nil)
60+assert(coroutine.isyieldable(f))
61+s,a,b,c,d = coroutine.resume(f, 1, 2, 3)
62+eqtab(_G.x, {1, 2, 3})
63+assert(s and a == 'a' and b == 'b' and c == 'c' and d == nil)
64+s,a,b,c,d = coroutine.resume(f, "xuxu")
65+eqtab(_G.x, {"xuxu"})
66+assert(s and a == 1 and b == 2 and c == 3 and d == nil)
67+assert(coroutine.status(f) == "dead")
68+s, a = coroutine.resume(f, "xuxu")
69+assert(not s and string.find(a, "dead") and coroutine.status(f) == "dead")
70+
71+_G.f = nil
72+
73+-- yields in tail calls
74+local function foo (i) return coroutine.yield(i) end
75+local f = coroutine.wrap(function ()
76+ for i=1,10 do
77+ assert(foo(i) == _G.x)
78+ end
79+ return 'a'
80+end)
81+for i=1,10 do _G.x = i; assert(f(i) == i) end
82+_G.x = 'xuxu'; assert(f('xuxu') == 'a')
83+
84+_G.x = nil
85+
86+-- recursive
87+local function pf (n, i)
88+ coroutine.yield(n)
89+ pf(n*i, i+1)
90+end
91+
92+f = coroutine.wrap(pf)
93+local s=1
94+for i=1,10 do
95+ assert(f(1, 1) == s)
96+ s = s*i
97+end
98+
99+-- sieve
100+local function gen (n)
101+ return coroutine.wrap(function ()
102+ for i=2,n do coroutine.yield(i) end
103+ end)
104+end
105+
106+
107+local function filter (p, g)
108+ return coroutine.wrap(function ()
109+ while 1 do
110+ local n = g()
111+ if n == nil then return end
112+ if math.fmod(n, p) ~= 0 then coroutine.yield(n) end
113+ end
114+ end)
115+end
116+
117+local x = gen(80)
118+local a = {}
119+while 1 do
120+ local n = x()
121+ if n == nil then break end
122+ table.insert(a, n)
123+ x = filter(n, x)
124+end
125+
126+assert(#a == 22 and a[#a] == 79)
127+x, a = nil
128+
129+
130+print("to-be-closed variables in coroutines")
131+
132+local function func2close (f)
133+ return setmetatable({}, {__close = f})
134+end
135+
136+do
137+ -- ok to close a dead coroutine
138+ local co = coroutine.create(print)
139+ assert(coroutine.resume(co, "testing 'coroutine.close'"))
140+ assert(coroutine.status(co) == "dead")
141+ local st, msg = coroutine.close(co)
142+ assert(st and msg == nil)
143+ -- also ok to close it again
144+ st, msg = coroutine.close(co)
145+ assert(st and msg == nil)
146+
147+
148+ -- cannot close the running coroutine
149+ local st, msg = pcall(coroutine.close, coroutine.running())
150+ assert(not st and string.find(msg, "running"))
151+
152+ local main = coroutine.running()
153+
154+ -- cannot close a "normal" coroutine
155+ ;(coroutine.wrap(function ()
156+ local st, msg = pcall(coroutine.close, main)
157+ assert(not st and string.find(msg, "normal"))
158+ end))()
159+
160+ -- cannot close a coroutine while closing it
161+ do
162+ local co
163+ co = coroutine.create(
164+ function()
165+ local x <close> = func2close(function()
166+ coroutine.close(co) -- try to close it again
167+ end)
168+ coroutine.yield(20)
169+ end)
170+ local st, msg = coroutine.resume(co)
171+ assert(st and msg == 20)
172+ st, msg = coroutine.close(co)
173+ assert(not st and string.find(msg, "running coroutine"))
174+ end
175+
176+ -- to-be-closed variables in coroutines
177+ local X
178+
179+ -- closing a coroutine after an error
180+ local co = coroutine.create(error)
181+ local st, msg = coroutine.resume(co, 100)
182+ assert(not st and msg == 100)
183+ st, msg = coroutine.close(co)
184+ assert(not st and msg == 100)
185+ -- after closing, no more errors
186+ st, msg = coroutine.close(co)
187+ assert(st and msg == nil)
188+
189+ co = coroutine.create(function ()
190+ local x <close> = func2close(function (self, err)
191+ assert(err == nil); X = false
192+ end)
193+ X = true
194+ coroutine.yield()
195+ end)
196+ coroutine.resume(co)
197+ assert(X)
198+ assert(coroutine.close(co))
199+ assert(not X and coroutine.status(co) == "dead")
200+
201+ -- error closing a coroutine
202+ local x = 0
203+ co = coroutine.create(function()
204+ local y <close> = func2close(function (self,err)
205+ assert(err == 111)
206+ x = 200
207+ error(200)
208+ end)
209+ local x <close> = func2close(function (self, err)
210+ assert(err == nil); error(111)
211+ end)
212+ coroutine.yield()
213+ end)
214+ coroutine.resume(co)
215+ assert(x == 0)
216+ local st, msg = coroutine.close(co)
217+ assert(st == false and coroutine.status(co) == "dead" and msg == 200)
218+ assert(x == 200)
219+ -- after closing, no more errors
220+ st, msg = coroutine.close(co)
221+ assert(st and msg == nil)
222+end
223+
224+do
225+ -- <close> versus pcall in coroutines
226+ local X = false
227+ local Y = false
228+ local function foo ()
229+ local x <close> = func2close(function (self, err)
230+ Y = debug.getinfo(2)
231+ X = err
232+ end)
233+ error(43)
234+ end
235+ local co = coroutine.create(function () return pcall(foo) end)
236+ local st1, st2, err = coroutine.resume(co)
237+ assert(st1 and not st2 and err == 43)
238+ assert(X == 43 and Y.what == "C")
239+
240+ -- recovering from errors in __close metamethods
241+ local track = {}
242+
243+ local function h (o)
244+ local hv <close> = o
245+ return 1
246+ end
247+
248+ local function foo ()
249+ local x <close> = func2close(function(_,msg)
250+ track[#track + 1] = msg or false
251+ error(20)
252+ end)
253+ local y <close> = func2close(function(_,msg)
254+ track[#track + 1] = msg or false
255+ return 1000
256+ end)
257+ local z <close> = func2close(function(_,msg)
258+ track[#track + 1] = msg or false
259+ error(10)
260+ end)
261+ coroutine.yield(1)
262+ h(func2close(function(_,msg)
263+ track[#track + 1] = msg or false
264+ error(2)
265+ end))
266+ end
267+
268+ local co = coroutine.create(pcall)
269+
270+ local st, res = coroutine.resume(co, foo) -- call 'foo' protected
271+ assert(st and res == 1) -- yield 1
272+ local st, res1, res2 = coroutine.resume(co) -- continue
273+ assert(coroutine.status(co) == "dead")
274+ assert(st and not res1 and res2 == 20) -- last error (20)
275+ assert(track[1] == false and track[2] == 2 and track[3] == 10 and
276+ track[4] == 10)
277+end
278+
279+
280+-- yielding across C boundaries
281+
282+local co = coroutine.wrap(function()
283+ assert(not pcall(table.sort,{1,2,3}, coroutine.yield))
284+ assert(coroutine.isyieldable())
285+ coroutine.yield(20)
286+ return 30
287+ end)
288+
289+assert(co() == 20)
290+assert(co() == 30)
291+
292+
293+local f = function (s, i) return coroutine.yield(i) end
294+
295+local f1 = coroutine.wrap(function ()
296+ return xpcall(pcall, function (...) return ... end,
297+ function ()
298+ local s = 0
299+ for i in f, nil, 1 do pcall(function () s = s + i end) end
300+ error({s})
301+ end)
302+ end)
303+
304+f1()
305+for i = 1, 10 do assert(f1(i) == i) end
306+local r1, r2, v = f1(nil)
307+assert(r1 and not r2 and v[1] == (10 + 1)*10/2)
308+
309+
310+local function f (a, b) a = coroutine.yield(a); error{a + b} end
311+local function g(x) return x[1]*2 end
312+
313+co = coroutine.wrap(function ()
314+ coroutine.yield(xpcall(f, g, 10, 20))
315+ end)
316+
317+assert(co() == 10)
318+local r, msg = co(100)
319+assert(not r and msg == 240)
320+
321+
322+-- unyieldable C call
323+do
324+ local function f (c)
325+ assert(not coroutine.isyieldable())
326+ return c .. c
327+ end
328+
329+ local co = coroutine.wrap(function (c)
330+ assert(coroutine.isyieldable())
331+ local s = string.gsub("a", ".", f)
332+ return s
333+ end)
334+ assert(co() == "aa")
335+end
336+
337+
338+
339+do -- testing single trace of coroutines
340+ local X
341+ local co = coroutine.create(function ()
342+ coroutine.yield(10)
343+ return 20;
344+ end)
345+ local trace = {}
346+ local function dotrace (event)
347+ trace[#trace + 1] = event
348+ end
349+ debug.sethook(co, dotrace, "clr")
350+ repeat until not coroutine.resume(co)
351+ local correcttrace = {"call", "line", "call", "return", "line", "return"}
352+ assert(#trace == #correcttrace)
353+ for k, v in pairs(trace) do
354+ assert(v == correcttrace[k])
355+ end
356+end
357+
358+-- errors in coroutines
359+function foo ()
360+ assert(debug.getinfo(1).currentline == debug.getinfo(foo).linedefined + 1)
361+ assert(debug.getinfo(2).currentline == debug.getinfo(goo).linedefined)
362+ coroutine.yield(3)
363+ error(foo)
364+end
365+
366+function goo() foo() end
367+x = coroutine.wrap(goo)
368+assert(x() == 3)
369+local a,b = pcall(x)
370+assert(not a and b == foo)
371+
372+x = coroutine.create(goo)
373+a,b = coroutine.resume(x)
374+assert(a and b == 3)
375+a,b = coroutine.resume(x)
376+assert(not a and b == foo and coroutine.status(x) == "dead")
377+a,b = coroutine.resume(x)
378+assert(not a and string.find(b, "dead") and coroutine.status(x) == "dead")
379+
380+goo = nil
381+
382+-- co-routines x for loop
383+local function all (a, n, k)
384+ if k == 0 then coroutine.yield(a)
385+ else
386+ for i=1,n do
387+ a[k] = i
388+ all(a, n, k-1)
389+ end
390+ end
391+end
392+
393+local a = 0
394+for t in coroutine.wrap(function () all({}, 5, 4) end) do
395+ a = a+1
396+end
397+assert(a == 5^4)
398+
399+
400+-- access to locals of collected corroutines
401+local C = {}; setmetatable(C, {__mode = "kv"})
402+local x = coroutine.wrap (function ()
403+ local a = 10
404+ local function f () a = a+10; return a end
405+ while true do
406+ a = a+1
407+ coroutine.yield(f)
408+ end
409+ end)
410+
411+C[1] = x;
412+
413+local f = x()
414+assert(f() == 21 and x()() == 32 and x() == f)
415+x = nil
416+collectgarbage()
417+assert(C[1] == undef)
418+assert(f() == 43 and f() == 53)
419+
420+
421+-- old bug: attempt to resume itself
422+
423+local function co_func (current_co)
424+ assert(coroutine.running() == current_co)
425+ assert(coroutine.resume(current_co) == false)
426+ coroutine.yield(10, 20)
427+ assert(coroutine.resume(current_co) == false)
428+ coroutine.yield(23)
429+ return 10
430+end
431+
432+local co = coroutine.create(co_func)
433+local a,b,c = coroutine.resume(co, co)
434+assert(a == true and b == 10 and c == 20)
435+a,b = coroutine.resume(co, co)
436+assert(a == true and b == 23)
437+a,b = coroutine.resume(co, co)
438+assert(a == true and b == 10)
439+assert(coroutine.resume(co, co) == false)
440+assert(coroutine.resume(co, co) == false)
441+
442+
443+-- other old bug when attempting to resume itself
444+-- (trigger C-code assertions)
445+do
446+ local A = coroutine.running()
447+ local B = coroutine.create(function() return coroutine.resume(A) end)
448+ local st, res = coroutine.resume(B)
449+ assert(st == true and res == false)
450+
451+ local X = false
452+ A = coroutine.wrap(function()
453+ local _ <close> = func2close(function () X = true end)
454+ return pcall(A, 1)
455+ end)
456+ st, res = A()
457+ assert(not st and string.find(res, "non%-suspended") and X == true)
458+end
459+
460+
461+-- bug in 5.4.1
462+do
463+ -- coroutine ran close metamethods with invalid status during a
464+ -- reset.
465+ local co
466+ co = coroutine.wrap(function()
467+ local x <close> = func2close(function() return pcall(co) end)
468+ error(111)
469+ end)
470+ local st, errobj = pcall(co)
471+ assert(not st and errobj == 111)
472+ st, errobj = pcall(co)
473+ assert(not st and string.find(errobj, "dead coroutine"))
474+end
475+
476+
477+-- attempt to resume 'normal' coroutine
478+local co1, co2
479+co1 = coroutine.create(function () return co2() end)
480+co2 = coroutine.wrap(function ()
481+ assert(coroutine.status(co1) == 'normal')
482+ assert(not coroutine.resume(co1))
483+ coroutine.yield(3)
484+ end)
485+
486+a,b = coroutine.resume(co1)
487+assert(a and b == 3)
488+assert(coroutine.status(co1) == 'dead')
489+
490+-- infinite recursion of coroutines
491+a = function(a) coroutine.wrap(a)(a) end
492+assert(not pcall(a, a))
493+a = nil
494+
495+
496+-- access to locals of erroneous coroutines
497+local x = coroutine.create (function ()
498+ local a = 10
499+ _G.F = function () a=a+1; return a end
500+ error('x')
501+ end)
502+
503+assert(not coroutine.resume(x))
504+-- overwrite previous position of local `a'
505+assert(not coroutine.resume(x, 1, 1, 1, 1, 1, 1, 1))
506+assert(_G.F() == 11)
507+assert(_G.F() == 12)
508+_G.F = nil
509+
510+
511+if not T then
512+ (Message or print)
513+ ('\n >>> testC not active: skipping coroutine API tests <<<\n')
514+else
515+ print "testing yields inside hooks"
516+
517+ local turn
518+
519+ local function fact (t, x)
520+ assert(turn == t)
521+ if x == 0 then return 1
522+ else return x*fact(t, x-1)
523+ end
524+ end
525+
526+ local A, B = 0, 0
527+
528+ local x = coroutine.create(function ()
529+ T.sethook("yield 0", "", 2)
530+ A = fact("A", 6)
531+ end)
532+
533+ local y = coroutine.create(function ()
534+ T.sethook("yield 0", "", 3)
535+ B = fact("B", 7)
536+ end)
537+
538+ while A==0 or B==0 do -- A ~= 0 when 'x' finishes (similar for 'B','y')
539+ if A==0 then turn = "A"; assert(T.resume(x)) end
540+ if B==0 then turn = "B"; assert(T.resume(y)) end
541+
542+ -- check that traceback works correctly after yields inside hooks
543+ debug.traceback(x)
544+ debug.traceback(y)
545+ end
546+
547+ assert(B // A == 7) -- fact(7) // fact(6)
548+
549+ do -- hooks vs. multiple values
550+ local done
551+ local function test (n)
552+ done = false
553+ return coroutine.wrap(function ()
554+ local a = {}
555+ for i = 1, n do a[i] = i end
556+ -- 'pushint' just to perturb the stack
557+ T.sethook("pushint 10; yield 0", "", 1) -- yield at each op.
558+ local a1 = {table.unpack(a)} -- must keep top between ops.
559+ assert(#a1 == n)
560+ for i = 1, n do assert(a[i] == i) end
561+ done = true
562+ end)
563+ end
564+ -- arguments to the coroutine are just to perturb its stack
565+ local co = test(0); while not done do co(30) end
566+ co = test(1); while not done do co(20, 10) end
567+ co = test(3); while not done do co() end
568+ co = test(100); while not done do co() end
569+ end
570+
571+ local line = debug.getinfo(1, "l").currentline + 2 -- get line number
572+ local function foo ()
573+ local x = 10 --<< this line is 'line'
574+ x = x + 10
575+ _G.XX = x
576+ end
577+
578+ -- testing yields in line hook
579+ local co = coroutine.wrap(function ()
580+ T.sethook("setglobal X; yield 0", "l", 0); foo(); return 10 end)
581+
582+ _G.XX = nil;
583+ _G.X = nil; co(); assert(_G.X == line)
584+ _G.X = nil; co(); assert(_G.X == line + 1)
585+ _G.X = nil; co(); assert(_G.X == line + 2 and _G.XX == nil)
586+ _G.X = nil; co(); assert(_G.X == line + 3 and _G.XX == 20)
587+ assert(co() == 10)
588+ _G.X = nil
589+
590+ -- testing yields in count hook
591+ co = coroutine.wrap(function ()
592+ T.sethook("yield 0", "", 1); foo(); return 10 end)
593+
594+ _G.XX = nil;
595+ local c = 0
596+ repeat c = c + 1; local a = co() until a == 10
597+ assert(_G.XX == 20 and c >= 5)
598+
599+ co = coroutine.wrap(function ()
600+ T.sethook("yield 0", "", 2); foo(); return 10 end)
601+
602+ _G.XX = nil;
603+ local c = 0
604+ repeat c = c + 1; local a = co() until a == 10
605+ assert(_G.XX == 20 and c >= 5)
606+ _G.X = nil; _G.XX = nil
607+
608+ do
609+ -- testing debug library on a coroutine suspended inside a hook
610+ -- (bug in 5.2/5.3)
611+ c = coroutine.create(function (a, ...)
612+ T.sethook("yield 0", "l") -- will yield on next two lines
613+ local b = a
614+ return ...
615+ end)
616+
617+ assert(coroutine.resume(c, 1, 2, 3)) -- start coroutine
618+ local n,v = debug.getlocal(c, 0, 1) -- check its local
619+ assert(n == "a" and v == 1 and debug.getlocal(c, 0, 2) ~= "b")
620+ assert(debug.setlocal(c, 0, 1, 10)) -- test 'setlocal'
621+ local t = debug.getinfo(c, 0) -- test 'getinfo'
622+ assert(t.currentline == t.linedefined + 2)
623+ assert(not debug.getinfo(c, 1)) -- no other level
624+ assert(coroutine.resume(c)) -- run next line
625+ local n,v = debug.getlocal(c, 0, 2) -- check next local
626+ assert(n == "b" and v == 10)
627+ v = {coroutine.resume(c)} -- finish coroutine
628+ assert(v[1] == true and v[2] == 2 and v[3] == 3 and v[4] == undef)
629+ assert(not coroutine.resume(c))
630+ end
631+
632+ do
633+ -- testing debug library on last function in a suspended coroutine
634+ -- (bug in 5.2/5.3)
635+ local c = coroutine.create(function () T.testC("yield 1", 10, 20) end)
636+ local a, b = coroutine.resume(c)
637+ assert(a and b == 20)
638+ assert(debug.getinfo(c, 0).linedefined == -1)
639+ a, b = debug.getlocal(c, 0, 2)
640+ assert(b == 10)
641+ end
642+
643+
644+ print "testing coroutine API"
645+
646+ -- reusing a thread
647+ assert(T.testC([[
648+ newthread # create thread
649+ pushvalue 2 # push body
650+ pushstring 'a a a' # push argument
651+ xmove 0 3 2 # move values to new thread
652+ resume -1, 1 # call it first time
653+ pushstatus
654+ xmove 3 0 0 # move results back to stack
655+ setglobal X # result
656+ setglobal Y # status
657+ pushvalue 2 # push body (to call it again)
658+ pushstring 'b b b'
659+ xmove 0 3 2
660+ resume -1, 1 # call it again
661+ pushstatus
662+ xmove 3 0 0
663+ return 1 # return result
664+ ]], function (...) return ... end) == 'b b b')
665+
666+ assert(X == 'a a a' and Y == 'OK')
667+
668+ X, Y = nil
669+
670+
671+ -- resuming running coroutine
672+ C = coroutine.create(function ()
673+ return T.testC([[
674+ pushnum 10;
675+ pushnum 20;
676+ resume -3 2;
677+ pushstatus
678+ gettop;
679+ return 3]], C)
680+ end)
681+ local a, b, c, d = coroutine.resume(C)
682+ assert(a == true and string.find(b, "non%-suspended") and
683+ c == "ERRRUN" and d == 4)
684+
685+ a, b, c, d = T.testC([[
686+ rawgeti R 1 # get main thread
687+ pushnum 10;
688+ pushnum 20;
689+ resume -3 2;
690+ pushstatus
691+ gettop;
692+ return 4]])
693+ assert(a == coroutine.running() and string.find(b, "non%-suspended") and
694+ c == "ERRRUN" and d == 4)
695+
696+
697+ -- using a main thread as a coroutine (dubious use!)
698+ local state = T.newstate()
699+
700+ -- check that yielddable is working correctly
701+ assert(T.testC(state, "newthread; isyieldable -1; remove 1; return 1"))
702+
703+ -- main thread is not yieldable
704+ assert(not T.testC(state, "rawgeti R 1; isyieldable -1; remove 1; return 1"))
705+
706+ T.testC(state, "settop 0")
707+
708+ T.loadlib(state)
709+
710+ assert(T.doremote(state, [[
711+ coroutine = require'coroutine';
712+ X = function (x) coroutine.yield(x, 'BB'); return 'CC' end;
713+ return 'ok']]))
714+
715+ local t = table.pack(T.testC(state, [[
716+ rawgeti R 1 # get main thread
717+ pushstring 'XX'
718+ getglobal X # get function for body
719+ pushstring AA # arg
720+ resume 1 1 # 'resume' shadows previous stack!
721+ gettop
722+ setglobal T # top
723+ setglobal B # second yielded value
724+ setglobal A # fist yielded value
725+ rawgeti R 1 # get main thread
726+ pushnum 5 # arg (noise)
727+ resume 1 1 # after coroutine ends, previous stack is back
728+ pushstatus
729+ return *
730+ ]]))
731+ assert(t.n == 4 and t[2] == 'XX' and t[3] == 'CC' and t[4] == 'OK')
732+ assert(T.doremote(state, "return T") == '2')
733+ assert(T.doremote(state, "return A") == 'AA')
734+ assert(T.doremote(state, "return B") == 'BB')
735+
736+ T.closestate(state)
737+
738+ print'+'
739+
740+end
741+
742+
743+-- leaving a pending coroutine open
744+_G.TO_SURVIVE = coroutine.wrap(function ()
745+ local a = 10
746+ local x = function () a = a+1 end
747+ coroutine.yield()
748+ end)
749+
750+_G.TO_SURVIVE()
751+
752+
753+if not _soft then
754+ -- bug (stack overflow)
755+ local lim = 1000000 -- stack limit; assume 32-bit machine
756+ local t = {lim - 10, lim - 5, lim - 1, lim, lim + 1, lim + 5}
757+ for i = 1, #t do
758+ local j = t[i]
759+ local co = coroutine.create(function()
760+ return table.unpack({}, 1, j)
761+ end)
762+ local r, msg = coroutine.resume(co)
763+ -- must fail for unpacking larger than stack limit
764+ assert(j < lim or not r)
765+ end
766+end
767+
768+
769+assert(coroutine.running() == main)
770+
771+print"+"
772+
773+
774+print"testing yields inside metamethods"
775+
776+local function val(x)
777+ if type(x) == "table" then return x.x else return x end
778+end
779+
780+local mt = {
781+ __eq = function(a,b) coroutine.yield(nil, "eq"); return val(a) == val(b) end,
782+ __lt = function(a,b) coroutine.yield(nil, "lt"); return val(a) < val(b) end,
783+ __le = function(a,b) coroutine.yield(nil, "le"); return a - b <= 0 end,
784+ __add = function(a,b) coroutine.yield(nil, "add");
785+ return val(a) + val(b) end,
786+ __sub = function(a,b) coroutine.yield(nil, "sub"); return val(a) - val(b) end,
787+ __mul = function(a,b) coroutine.yield(nil, "mul"); return val(a) * val(b) end,
788+ __div = function(a,b) coroutine.yield(nil, "div"); return val(a) / val(b) end,
789+ __idiv = function(a,b) coroutine.yield(nil, "idiv");
790+ return val(a) // val(b) end,
791+ __pow = function(a,b) coroutine.yield(nil, "pow"); return val(a) ^ val(b) end,
792+ __mod = function(a,b) coroutine.yield(nil, "mod"); return val(a) % val(b) end,
793+ __unm = function(a,b) coroutine.yield(nil, "unm"); return -val(a) end,
794+ __bnot = function(a,b) coroutine.yield(nil, "bnot"); return ~val(a) end,
795+ __shl = function(a,b) coroutine.yield(nil, "shl");
796+ return val(a) << val(b) end,
797+ __shr = function(a,b) coroutine.yield(nil, "shr");
798+ return val(a) >> val(b) end,
799+ __band = function(a,b)
800+ coroutine.yield(nil, "band")
801+ return val(a) & val(b)
802+ end,
803+ __bor = function(a,b) coroutine.yield(nil, "bor");
804+ return val(a) | val(b) end,
805+ __bxor = function(a,b) coroutine.yield(nil, "bxor");
806+ return val(a) ~ val(b) end,
807+
808+ __concat = function(a,b)
809+ coroutine.yield(nil, "concat");
810+ return val(a) .. val(b)
811+ end,
812+ __index = function (t,k) coroutine.yield(nil, "idx"); return t.k[k] end,
813+ __newindex = function (t,k,v) coroutine.yield(nil, "nidx"); t.k[k] = v end,
814+}
815+
816+
817+local function new (x)
818+ return setmetatable({x = x, k = {}}, mt)
819+end
820+
821+
822+local a = new(10)
823+local b = new(12)
824+local c = new"hello"
825+
826+local function run (f, t)
827+ local i = 1
828+ local c = coroutine.wrap(f)
829+ while true do
830+ local res, stat = c()
831+ if res then assert(t[i] == undef); return res, t end
832+ assert(stat == t[i])
833+ i = i + 1
834+ end
835+end
836+
837+
838+assert(run(function () if (a>=b) then return '>=' else return '<' end end,
839+ {"le", "sub"}) == "<")
840+assert(run(function () if (a<=b) then return '<=' else return '>' end end,
841+ {"le", "sub"}) == "<=")
842+assert(run(function () if (a==b) then return '==' else return '~=' end end,
843+ {"eq"}) == "~=")
844+
845+assert(run(function () return a & b + a end, {"add", "band"}) == 2)
846+
847+assert(run(function () return 1 + a end, {"add"}) == 11)
848+assert(run(function () return a - 25 end, {"sub"}) == -15)
849+assert(run(function () return 2 * a end, {"mul"}) == 20)
850+assert(run(function () return a ^ 2 end, {"pow"}) == 100)
851+assert(run(function () return a / 2 end, {"div"}) == 5)
852+assert(run(function () return a % 6 end, {"mod"}) == 4)
853+assert(run(function () return a // 3 end, {"idiv"}) == 3)
854+
855+assert(run(function () return a + b end, {"add"}) == 22)
856+assert(run(function () return a - b end, {"sub"}) == -2)
857+assert(run(function () return a * b end, {"mul"}) == 120)
858+assert(run(function () return a ^ b end, {"pow"}) == 10^12)
859+assert(run(function () return a / b end, {"div"}) == 10/12)
860+assert(run(function () return a % b end, {"mod"}) == 10)
861+assert(run(function () return a // b end, {"idiv"}) == 0)
862+
863+-- repeat tests with larger constants (to use 'K' opcodes)
864+local a1000 = new(1000)
865+
866+assert(run(function () return a1000 + 1000 end, {"add"}) == 2000)
867+assert(run(function () return a1000 - 25000 end, {"sub"}) == -24000)
868+assert(run(function () return 2000 * a end, {"mul"}) == 20000)
869+assert(run(function () return a1000 / 1000 end, {"div"}) == 1)
870+assert(run(function () return a1000 % 600 end, {"mod"}) == 400)
871+assert(run(function () return a1000 // 500 end, {"idiv"}) == 2)
872+
873+
874+
875+assert(run(function () return a % b end, {"mod"}) == 10)
876+
877+assert(run(function () return ~a & b end, {"bnot", "band"}) == ~10 & 12)
878+assert(run(function () return a | b end, {"bor"}) == 10 | 12)
879+assert(run(function () return a ~ b end, {"bxor"}) == 10 ~ 12)
880+assert(run(function () return a << b end, {"shl"}) == 10 << 12)
881+assert(run(function () return a >> b end, {"shr"}) == 10 >> 12)
882+
883+assert(run(function () return 10 & b end, {"band"}) == 10 & 12)
884+assert(run(function () return a | 2 end, {"bor"}) == 10 | 2)
885+assert(run(function () return a ~ 2 end, {"bxor"}) == 10 ~ 2)
886+assert(run(function () return a >> 2 end, {"shr"}) == 10 >> 2)
887+assert(run(function () return 1 >> a end, {"shr"}) == 1 >> 10)
888+assert(run(function () return a << 2 end, {"shl"}) == 10 << 2)
889+assert(run(function () return 1 << a end, {"shl"}) == 1 << 10)
890+assert(run(function () return 2 ~ a end, {"bxor"}) == 2 ~ 10)
891+
892+
893+assert(run(function () return a..b end, {"concat"}) == "1012")
894+
895+assert(run(function() return a .. b .. c .. a end,
896+ {"concat", "concat", "concat"}) == "1012hello10")
897+
898+assert(run(function() return "a" .. "b" .. a .. "c" .. c .. b .. "x" end,
899+ {"concat", "concat", "concat"}) == "ab10chello12x")
900+
901+
902+do -- a few more tests for comparison operators
903+ local mt1 = {
904+ __le = function (a,b)
905+ coroutine.yield(10)
906+ return (val(a) <= val(b))
907+ end,
908+ __lt = function (a,b)
909+ coroutine.yield(10)
910+ return val(a) < val(b)
911+ end,
912+ }
913+ local mt2 = { __lt = mt1.__lt, __le = mt1.__le }
914+
915+ local function run (f)
916+ local co = coroutine.wrap(f)
917+ local res
918+ repeat
919+ res = co()
920+ until res ~= 10
921+ return res
922+ end
923+
924+ local function test ()
925+ local a1 = setmetatable({x=1}, mt1)
926+ local a2 = setmetatable({x=2}, mt2)
927+ assert(a1 < a2)
928+ assert(a1 <= a2)
929+ assert(1 < a2)
930+ assert(1 <= a2)
931+ assert(2 > a1)
932+ assert(2 >= a2)
933+ return true
934+ end
935+
936+ run(test)
937+
938+end
939+
940+assert(run(function ()
941+ a.BB = print
942+ return a.BB
943+ end, {"nidx", "idx"}) == print)
944+
945+-- getuptable & setuptable
946+do local _ENV = _ENV
947+ f = function () AAA = BBB + 1; return AAA end
948+end
949+local g = new(10); g.k.BBB = 10;
950+debug.setupvalue(f, 1, g)
951+assert(run(f, {"idx", "nidx", "idx"}) == 11)
952+assert(g.k.AAA == 11)
953+
954+print"+"
955+
956+print"testing yields inside 'for' iterators"
957+
958+local f = function (s, i)
959+ if i%2 == 0 then coroutine.yield(nil, "for") end
960+ if i < s then return i + 1 end
961+ end
962+
963+assert(run(function ()
964+ local s = 0
965+ for i in f, 4, 0 do s = s + i end
966+ return s
967+ end, {"for", "for", "for"}) == 10)
968+
969+
970+
971+-- tests for coroutine API
972+if T==nil then
973+ (Message or print)('\n >>> testC not active: skipping coroutine API tests <<<\n')
974+ print "OK"; return
975+end
976+
977+print('testing coroutine API')
978+
979+local function apico (...)
980+ local x = {...}
981+ return coroutine.wrap(function ()
982+ return T.testC(table.unpack(x))
983+ end)
984+end
985+
986+local a = {apico(
987+[[
988+ pushstring errorcode
989+ pcallk 1 0 2;
990+ invalid command (should not arrive here)
991+]],
992+[[return *]],
993+"stackmark",
994+error
995+)()}
996+assert(#a == 4 and
997+ a[3] == "stackmark" and
998+ a[4] == "errorcode" and
999+ _G.status == "ERRRUN" and
1000+ _G.ctx == 2) -- 'ctx' to pcallk
1001+
1002+local co = apico(
1003+ "pushvalue 2; pushnum 10; pcallk 1 2 3; invalid command;",
1004+ coroutine.yield,
1005+ "getglobal status; getglobal ctx; pushvalue 2; pushstring a; pcallk 1 0 4; invalid command",
1006+ "getglobal status; getglobal ctx; return *")
1007+
1008+assert(co() == 10)
1009+assert(co(20, 30) == 'a')
1010+a = {co()}
1011+assert(#a == 10 and
1012+ a[2] == coroutine.yield and
1013+ a[5] == 20 and a[6] == 30 and
1014+ a[7] == "YIELD" and a[8] == 3 and
1015+ a[9] == "YIELD" and a[10] == 4)
1016+assert(not pcall(co)) -- coroutine is dead now
1017+
1018+
1019+f = T.makeCfunc("pushnum 3; pushnum 5; yield 1;")
1020+co = coroutine.wrap(function ()
1021+ assert(f() == 23); assert(f() == 23); return 10
1022+end)
1023+assert(co(23,16) == 5)
1024+assert(co(23,16) == 5)
1025+assert(co(23,16) == 10)
1026+
1027+
1028+-- testing coroutines with C bodies
1029+f = T.makeCfunc([[
1030+ pushnum 102
1031+ yieldk 1 U2
1032+ cannot be here!
1033+]],
1034+[[ # continuation
1035+ pushvalue U3 # accessing upvalues inside a continuation
1036+ pushvalue U4
1037+ return *
1038+]], 23, "huu")
1039+
1040+x = coroutine.wrap(f)
1041+assert(x() == 102)
1042+eqtab({x()}, {23, "huu"})
1043+
1044+
1045+f = T.makeCfunc[[pushstring 'a'; pushnum 102; yield 2; ]]
1046+
1047+a, b, c, d = T.testC([[newthread; pushvalue 2; xmove 0 3 1; resume 3 0;
1048+ pushstatus; xmove 3 0 0; resume 3 0; pushstatus;
1049+ return 4; ]], f)
1050+
1051+assert(a == 'YIELD' and b == 'a' and c == 102 and d == 'OK')
1052+
1053+
1054+-- testing chain of suspendable C calls
1055+
1056+local count = 3 -- number of levels
1057+
1058+f = T.makeCfunc([[
1059+ remove 1; # remove argument
1060+ pushvalue U3; # get selection function
1061+ call 0 1; # call it (result is 'f' or 'yield')
1062+ pushstring hello # single argument for selected function
1063+ pushupvalueindex 2; # index of continuation program
1064+ callk 1 -1 .; # call selected function
1065+ errorerror # should never arrive here
1066+]],
1067+[[
1068+ # continuation program
1069+ pushnum 34 # return value
1070+ return * # return all results
1071+]],
1072+function () -- selection function
1073+ count = count - 1
1074+ if count == 0 then return coroutine.yield
1075+ else return f
1076+ end
1077+end
1078+)
1079+
1080+co = coroutine.wrap(function () return f(nil) end)
1081+assert(co() == "hello") -- argument to 'yield'
1082+a = {co()}
1083+-- three '34's (one from each pending C call)
1084+assert(#a == 3 and a[1] == a[2] and a[2] == a[3] and a[3] == 34)
1085+
1086+
1087+-- testing yields with continuations
1088+
1089+local y
1090+
1091+co = coroutine.wrap(function (...) return
1092+ T.testC([[ # initial function
1093+ yieldk 1 2
1094+ cannot be here!
1095+ ]],
1096+ [[ # 1st continuation
1097+ yieldk 0 3
1098+ cannot be here!
1099+ ]],
1100+ [[ # 2nd continuation
1101+ yieldk 0 4
1102+ cannot be here!
1103+ ]],
1104+ [[ # 3th continuation
1105+ pushvalue 6 # function which is last arg. to 'testC' here
1106+ pushnum 10; pushnum 20;
1107+ pcall 2 0 0 # call should throw an error and return to next line
1108+ pop 1 # remove error message
1109+ pushvalue 6
1110+ getglobal status; getglobal ctx
1111+ pcallk 2 2 5 # call should throw an error and jump to continuation
1112+ cannot be here!
1113+ ]],
1114+ [[ # 4th (and last) continuation
1115+ return *
1116+ ]],
1117+ -- function called by 3th continuation
1118+ function (a,b) x=a; y=b; error("errmsg") end,
1119+ ...
1120+)
1121+end)
1122+
1123+local a = {co(3,4,6)}
1124+assert(a[1] == 6 and a[2] == undef)
1125+a = {co()}; assert(a[1] == undef and _G.status == "YIELD" and _G.ctx == 2)
1126+a = {co()}; assert(a[1] == undef and _G.status == "YIELD" and _G.ctx == 3)
1127+a = {co(7,8)};
1128+-- original arguments
1129+assert(type(a[1]) == 'string' and type(a[2]) == 'string' and
1130+ type(a[3]) == 'string' and type(a[4]) == 'string' and
1131+ type(a[5]) == 'string' and type(a[6]) == 'function')
1132+-- arguments left from fist resume
1133+assert(a[7] == 3 and a[8] == 4)
1134+-- arguments to last resume
1135+assert(a[9] == 7 and a[10] == 8)
1136+-- error message and nothing more
1137+assert(a[11]:find("errmsg") and #a == 11)
1138+-- check arguments to pcallk
1139+assert(x == "YIELD" and y == 4)
1140+
1141+assert(not pcall(co)) -- coroutine should be dead
1142+
1143+_G.ctx = nil
1144+_G.status = nil
1145+
1146+
1147+-- bug in nCcalls
1148+local co = coroutine.wrap(function ()
1149+ local a = {pcall(pcall,pcall,pcall,pcall,pcall,pcall,pcall,error,"hi")}
1150+ return pcall(assert, table.unpack(a))
1151+end)
1152+
1153+local a = {co()}
1154+assert(a[10] == "hi")
1155+
1156+print'OK'
zod/luau/tests/cstack.luanew file+197
@@ -0,0 +1,197 @@
1+-- $Id: testes/cstack.lua $
2+-- See Copyright Notice in file all.lua
3+
4+
5+local tracegc = require"tracegc"
6+
7+print"testing stack overflow detection"
8+
9+-- Segmentation faults in these tests probably result from a C-stack
10+-- overflow. To avoid these errors, you should set a smaller limit for
11+-- the use of C stack by Lua, by changing the constant 'LUAI_MAXCCALLS'.
12+-- Alternatively, you can ensure a larger stack for the program.
13+
14+
15+local function checkerror (msg, f, ...)
16+ local s, err = pcall(f, ...)
17+ assert(not s and string.find(err, msg))
18+end
19+
20+do print("testing stack overflow in message handling")
21+ local count = 0
22+ local function loop (x, y, z)
23+ count = count + 1
24+ return 1 + loop(x, y, z)
25+ end
26+ tracegc.stop() -- __gc should not be called with a full stack
27+ local res, msg = xpcall(loop, loop)
28+ tracegc.start()
29+ assert(msg == "error in error handling")
30+ print("final count: ", count)
31+end
32+
33+
34+-- bug since 2.5 (C-stack overflow in recursion inside pattern matching)
35+do print("testing recursion inside pattern matching")
36+ local function f (size)
37+ local s = string.rep("a", size)
38+ local p = string.rep(".?", size)
39+ return string.match(s, p)
40+ end
41+ local m = f(80)
42+ assert(#m == 80)
43+ checkerror("too complex", f, 2000)
44+end
45+
46+
47+do print("testing stack-overflow in recursive 'gsub'")
48+ local count = 0
49+ local function foo ()
50+ count = count + 1
51+ string.gsub("a", ".", foo)
52+ end
53+ checkerror("stack overflow", foo)
54+ print("final count: ", count)
55+
56+ print("testing stack-overflow in recursive 'gsub' with metatables")
57+ local count = 0
58+ local t = setmetatable({}, {__index = foo})
59+ foo = function ()
60+ count = count + 1
61+ string.gsub("a", ".", t)
62+ end
63+ checkerror("stack overflow", foo)
64+ print("final count: ", count)
65+end
66+
67+
68+do -- bug in 5.4.0
69+ print("testing limits in coroutines inside deep calls")
70+ local count = 0
71+ local lim = 1000
72+ local function stack (n)
73+ if n > 0 then return stack(n - 1) + 1
74+ else coroutine.wrap(function ()
75+ count = count + 1
76+ stack(lim)
77+ end)()
78+ end
79+ end
80+
81+ local st, msg = xpcall(stack, function () return "ok" end, lim)
82+ assert(not st and msg == "ok")
83+ print("final count: ", count)
84+end
85+
86+
87+do -- bug since 5.4.0
88+ local count = 0
89+ print("chain of 'coroutine.close'")
90+ -- create N coroutines forming a list so that each one, when closed,
91+ -- closes the previous one. (With a large enough N, previous Lua
92+ -- versions crash in this test.)
93+ local coro = false
94+ for i = 1, 1000 do
95+ local previous = coro
96+ coro = coroutine.create(function()
97+ local cc <close> = setmetatable({}, {__close=function()
98+ count = count + 1
99+ if previous then
100+ assert(coroutine.close(previous))
101+ end
102+ end})
103+ coroutine.yield() -- leaves 'cc' pending to be closed
104+ end)
105+ assert(coroutine.resume(coro)) -- start it and run until it yields
106+ end
107+ local st, msg = coroutine.close(coro)
108+ assert(not st and string.find(msg, "C stack overflow"))
109+ print("final count: ", count)
110+end
111+
112+
113+do
114+ print("nesting of resuming yielded coroutines")
115+ local count = 0
116+
117+ local function body ()
118+ coroutine.yield()
119+ local f = coroutine.wrap(body)
120+ f(); -- start new coroutine (will stop in previous yield)
121+ count = count + 1
122+ f() -- call it recursively
123+ end
124+
125+ local f = coroutine.wrap(body)
126+ f()
127+ assert(not pcall(f))
128+ print("final count: ", count)
129+end
130+
131+
132+do -- bug in 5.4.2
133+ print("nesting coroutines running after recoverable errors")
134+ local count = 0
135+ local function foo()
136+ count = count + 1
137+ pcall(1) -- create an error
138+ -- running now inside 'precover' ("protected recover")
139+ coroutine.wrap(foo)() -- call another coroutine
140+ end
141+ checkerror("C stack overflow", foo)
142+ print("final count: ", count)
143+end
144+
145+
146+if T then
147+ print("testing stack recovery")
148+ local N = 0 -- trace number of calls
149+ local LIM = -1 -- will store N just before stack overflow
150+
151+ -- trace stack size; after stack overflow, it should be
152+ -- the maximum allowed stack size.
153+ local stack1
154+ local dummy
155+
156+ local function err(msg)
157+ assert(string.find(msg, "stack overflow"))
158+ local _, stacknow = T.stacklevel()
159+ assert(stacknow == stack1 + 200)
160+ end
161+
162+ -- When LIM==-1, the 'if' is not executed, so this function only
163+ -- counts and stores the stack limits up to overflow. Then, LIM
164+ -- becomes N, and then the 'if' code is run when the stack is
165+ -- full. Then, there is a stack overflow inside 'xpcall', after which
166+ -- the stack must have been restored back to its maximum normal size.
167+ local function f()
168+ dummy, stack1 = T.stacklevel()
169+ if N == LIM then
170+ xpcall(f, err)
171+ local _, stacknow = T.stacklevel()
172+ assert(stacknow == stack1)
173+ return
174+ end
175+ N = N + 1
176+ f()
177+ end
178+
179+ local topB, sizeB -- top and size Before overflow
180+ local topA, sizeA -- top and size After overflow
181+ topB, sizeB = T.stacklevel()
182+ tracegc.stop() -- __gc should not be called with a full stack
183+ xpcall(f, err)
184+ tracegc.start()
185+ topA, sizeA = T.stacklevel()
186+ -- sizes should be comparable
187+ assert(topA == topB and sizeA < sizeB * 2)
188+ print(string.format("maximum stack size: %d", stack1))
189+ LIM = N -- will stop recursion at maximum level
190+ N = 0 -- to count again
191+ tracegc.stop() -- __gc should not be called with a full stack
192+ f()
193+ tracegc.start()
194+ print"+"
195+end
196+
197+print'OK'
zod/luau/tests/db.luanew file+1054
@@ -0,0 +1,1054 @@
1+-- $Id: testes/db.lua $
2+-- See Copyright Notice in file all.lua
3+
4+-- testing debug library
5+
6+local debug = require "debug"
7+
8+local function dostring(s) return assert(load(s))() end
9+
10+print"testing debug library and debug information"
11+
12+do
13+local a=1
14+end
15+
16+assert(not debug.gethook())
17+
18+local testline = 19 -- line where 'test' is defined
19+local function test (s, l, p) -- this must be line 19
20+ collectgarbage() -- avoid gc during trace
21+ local function f (event, line)
22+ assert(event == 'line')
23+ local l = table.remove(l, 1)
24+ if p then print(l, line) end
25+ assert(l == line, "wrong trace!!")
26+ end
27+ debug.sethook(f,"l"); load(s)(); debug.sethook()
28+ assert(#l == 0)
29+end
30+
31+
32+do
33+ assert(not pcall(debug.getinfo, print, "X")) -- invalid option
34+ assert(not pcall(debug.getinfo, 0, ">")) -- invalid option
35+ assert(not debug.getinfo(1000)) -- out of range level
36+ assert(not debug.getinfo(-1)) -- out of range level
37+ local a = debug.getinfo(print)
38+ assert(a.what == "C" and a.short_src == "[C]")
39+ a = debug.getinfo(print, "L")
40+ assert(a.activelines == nil)
41+ local b = debug.getinfo(test, "SfL")
42+ assert(b.name == nil and b.what == "Lua" and b.linedefined == testline and
43+ b.lastlinedefined == b.linedefined + 10 and
44+ b.func == test and not string.find(b.short_src, "%["))
45+ assert(b.activelines[b.linedefined + 1] and
46+ b.activelines[b.lastlinedefined])
47+ assert(not b.activelines[b.linedefined] and
48+ not b.activelines[b.lastlinedefined + 1])
49+end
50+
51+
52+-- bug in 5.4.4-5.4.6: activelines in vararg functions
53+-- without debug information
54+do
55+ local func = load(string.dump(load("print(10)"), true))
56+ local actl = debug.getinfo(func, "L").activelines
57+ assert(#actl == 0) -- no line info
58+end
59+
60+
61+-- test file and string names truncation
62+local a = "function f () end"
63+local function dostring (s, x) return load(s, x)() end
64+dostring(a)
65+assert(debug.getinfo(f).short_src == string.format('[string "%s"]', a))
66+dostring(a..string.format("; %s\n=1", string.rep('p', 400)))
67+assert(string.find(debug.getinfo(f).short_src, '^%[string [^\n]*%.%.%."%]$'))
68+dostring(a..string.format("; %s=1", string.rep('p', 400)))
69+assert(string.find(debug.getinfo(f).short_src, '^%[string [^\n]*%.%.%."%]$'))
70+dostring("\n"..a)
71+assert(debug.getinfo(f).short_src == '[string "..."]')
72+dostring(a, "")
73+assert(debug.getinfo(f).short_src == '[string ""]')
74+dostring(a, "@xuxu")
75+assert(debug.getinfo(f).short_src == "xuxu")
76+dostring(a, "@"..string.rep('p', 1000)..'t')
77+assert(string.find(debug.getinfo(f).short_src, "^%.%.%.p*t$"))
78+dostring(a, "=xuxu")
79+assert(debug.getinfo(f).short_src == "xuxu")
80+dostring(a, string.format("=%s", string.rep('x', 500)))
81+assert(string.find(debug.getinfo(f).short_src, "^x*$"))
82+dostring(a, "=")
83+assert(debug.getinfo(f).short_src == "")
84+_G.a = nil; _G.f = nil;
85+_G[string.rep("p", 400)] = nil
86+
87+
88+repeat
89+ local g = {x = function ()
90+ local a = debug.getinfo(2)
91+ assert(a.name == 'f' and a.namewhat == 'local')
92+ a = debug.getinfo(1)
93+ assert(a.name == 'x' and a.namewhat == 'field')
94+ return 'xixi'
95+ end}
96+ local f = function () return 1+1 and (not 1 or g.x()) end
97+ assert(f() == 'xixi')
98+ g = debug.getinfo(f)
99+ assert(g.what == "Lua" and g.func == f and g.namewhat == "" and not g.name)
100+
101+ function f (x, name) -- local!
102+ name = name or 'f'
103+ local a = debug.getinfo(1)
104+ assert(a.name == name and a.namewhat == 'local')
105+ return x
106+ end
107+
108+ -- breaks in different conditions
109+ if 3>4 then break end; f()
110+ if 3<4 then a=1 else break end; f()
111+ while 1 do local x=10; break end; f()
112+ local b = 1
113+ if 3>4 then return math.sin(1) end; f()
114+ a = 3<4; f()
115+ a = 3<4 or 1; f()
116+ repeat local x=20; if 4>3 then f() else break end; f() until 1
117+ g = {}
118+ f(g).x = f(2) and f(10)+f(9)
119+ assert(g.x == f(19))
120+ function g(x) if not x then return 3 end return (x('a', 'x')) end
121+ assert(g(f) == 'a')
122+until 1
123+
124+test([[if
125+math.sin(1)
126+then
127+ a=1
128+else
129+ a=2
130+end
131+]], {2,3,4,7})
132+
133+
134+test([[
135+local function foo()
136+end
137+foo()
138+A = 1
139+A = 2
140+A = 3
141+]], {2, 3, 2, 4, 5, 6})
142+_G.A = nil
143+
144+
145+test([[--
146+if nil then
147+ a=1
148+else
149+ a=2
150+end
151+]], {2,5,6})
152+
153+test([[a=1
154+repeat
155+ a=a+1
156+until a==3
157+]], {1,3,4,3,4})
158+
159+test([[ do
160+ return
161+end
162+]], {2})
163+
164+test([[local a
165+a=1
166+while a<=3 do
167+ a=a+1
168+end
169+]], {1,2,3,4,3,4,3,4,3,5})
170+
171+test([[while math.sin(1) do
172+ if math.sin(1)
173+ then break
174+ end
175+end
176+a=1]], {1,2,3,6})
177+
178+test([[for i=1,3 do
179+ a=i
180+end
181+]], {1,2,1,2,1,2,1,3})
182+
183+test([[for i,v in pairs{'a','b'} do
184+ a=tostring(i) .. v
185+end
186+]], {1,2,1,2,1,3})
187+
188+test([[for i=1,4 do a=1 end]], {1,1,1,1})
189+
190+_G.a = nil
191+
192+
193+do -- testing line info/trace with large gaps in source
194+
195+ local a = {1, 2, 3, 10, 124, 125, 126, 127, 128, 129, 130,
196+ 255, 256, 257, 500, 1000}
197+ local s = [[
198+ local b = {10}
199+ a = b[1] X + Y b[1]
200+ b = 4
201+ ]]
202+ for _, i in ipairs(a) do
203+ local subs = {X = string.rep("\n", i)}
204+ for _, j in ipairs(a) do
205+ subs.Y = string.rep("\n", j)
206+ local s = string.gsub(s, "[XY]", subs)
207+ test(s, {1, 2 + i, 2 + i + j, 2 + i, 2 + i + j, 3 + i + j})
208+ end
209+ end
210+end
211+_G.a = nil
212+
213+
214+do -- testing active lines
215+ local function checkactivelines (f, lines)
216+ local t = debug.getinfo(f, "SL")
217+ for _, l in pairs(lines) do
218+ l = l + t.linedefined
219+ assert(t.activelines[l])
220+ t.activelines[l] = undef
221+ end
222+ assert(next(t.activelines) == nil) -- no extra lines
223+ end
224+
225+ checkactivelines(function (...) -- vararg function
226+ -- 1st line is empty
227+ -- 2nd line is empty
228+ -- 3th line is empty
229+ local a = 20
230+ -- 5th line is empty
231+ local b = 30
232+ -- 7th line is empty
233+ end, {4, 6, 8})
234+
235+ checkactivelines(function (a)
236+ -- 1st line is empty
237+ -- 2nd line is empty
238+ local a = 20
239+ local b = 30
240+ -- 5th line is empty
241+ end, {3, 4, 6})
242+
243+ checkactivelines(function (a, b, ...) end, {0})
244+
245+ checkactivelines(function (a, b)
246+ end, {1})
247+
248+ for _, n in pairs{0, 1, 2, 10, 50, 100, 1000, 10000} do
249+ checkactivelines(
250+ load(string.format("%s return 1", string.rep("\n", n))),
251+ {n + 1})
252+ end
253+
254+end
255+
256+print'+'
257+
258+-- invalid levels in [gs]etlocal
259+assert(not pcall(debug.getlocal, 20, 1))
260+assert(not pcall(debug.setlocal, -1, 1, 10))
261+
262+
263+-- parameter names
264+local function foo (a,b,...) local d, e end
265+local co = coroutine.create(foo)
266+
267+assert(debug.getlocal(foo, 1) == 'a')
268+assert(debug.getlocal(foo, 2) == 'b')
269+assert(not debug.getlocal(foo, 3))
270+assert(debug.getlocal(co, foo, 1) == 'a')
271+assert(debug.getlocal(co, foo, 2) == 'b')
272+assert(not debug.getlocal(co, foo, 3))
273+
274+assert(not debug.getlocal(print, 1))
275+
276+
277+local function foo () return (debug.getlocal(1, -1)) end
278+assert(not foo(10))
279+
280+
281+-- varargs
282+local function foo (a, ...)
283+ local t = table.pack(...)
284+ for i = 1, t.n do
285+ local n, v = debug.getlocal(1, -i)
286+ assert(n == "(vararg)" and v == t[i])
287+ end
288+ assert(not debug.getlocal(1, -(t.n + 1)))
289+ assert(not debug.setlocal(1, -(t.n + 1), 30))
290+ if t.n > 0 then
291+ (function (x)
292+ assert(debug.setlocal(2, -1, x) == "(vararg)")
293+ assert(debug.setlocal(2, -t.n, x) == "(vararg)")
294+ end)(430)
295+ assert(... == 430)
296+ end
297+end
298+
299+foo()
300+foo(print)
301+foo(200, 3, 4)
302+local a = {}
303+for i = 1, (_soft and 100 or 1000) do a[i] = i end
304+foo(table.unpack(a))
305+
306+
307+
308+do -- test hook presence in debug info
309+ assert(not debug.gethook())
310+ local count = 0
311+ local function f ()
312+ assert(debug.getinfo(1).namewhat == "hook")
313+ local sndline = string.match(debug.traceback(), "\n(.-)\n")
314+ assert(string.find(sndline, "hook"))
315+ count = count + 1
316+ end
317+ debug.sethook(f, "l")
318+ local a = 0
319+ _ENV.a = a
320+ a = 1
321+ debug.sethook()
322+ assert(count == 4)
323+end
324+_ENV.a = nil
325+
326+
327+-- hook table has weak keys
328+assert(getmetatable(debug.getregistry()._HOOKKEY).__mode == 'k')
329+
330+
331+a = {}; local L = nil
332+local glob = 1
333+local oldglob = glob
334+debug.sethook(function (e,l)
335+ collectgarbage() -- force GC during a hook
336+ local f, m, c = debug.gethook()
337+ assert(m == 'crl' and c == 0)
338+ if e == "line" then
339+ if glob ~= oldglob then
340+ L = l-1 -- get the first line where "glob" has changed
341+ oldglob = glob
342+ end
343+ elseif e == "call" then
344+ local f = debug.getinfo(2, "f").func
345+ a[f] = 1
346+ else assert(e == "return")
347+ end
348+end, "crl")
349+
350+
351+function f(a,b)
352+ collectgarbage()
353+ local _, x = debug.getlocal(1, 1)
354+ local _, y = debug.getlocal(1, 2)
355+ assert(x == a and y == b)
356+ assert(debug.setlocal(2, 3, "pera") == "AA".."AA")
357+ assert(debug.setlocal(2, 4, "manga") == "B")
358+ x = debug.getinfo(2)
359+ assert(x.func == g and x.what == "Lua" and x.name == 'g' and
360+ x.nups == 2 and string.find(x.source, "^@.*db%.lua$"))
361+ glob = glob+1
362+ assert(debug.getinfo(1, "l").currentline == L+1)
363+ assert(debug.getinfo(1, "l").currentline == L+2)
364+end
365+
366+function foo()
367+ glob = glob+1
368+ assert(debug.getinfo(1, "l").currentline == L+1)
369+end; foo() -- set L
370+-- check line counting inside strings and empty lines
371+
372+local _ = 'alo\
373+alo' .. [[
374+
375+]]
376+--[[
377+]]
378+assert(debug.getinfo(1, "l").currentline == L+11) -- check count of lines
379+
380+
381+function g (...)
382+ local arg = {...}
383+ do local a,b,c; a=math.sin(40); end
384+ local feijao
385+ local AAAA,B = "xuxu", "abacate"
386+ f(AAAA,B)
387+ assert(AAAA == "pera" and B == "manga")
388+ do
389+ local B = 13
390+ local x,y = debug.getlocal(1,5)
391+ assert(x == 'B' and y == 13)
392+ end
393+end
394+
395+g()
396+
397+
398+assert(a[f] and a[g] and a[assert] and a[debug.getlocal] and not a[print])
399+
400+
401+-- tests for manipulating non-registered locals (C and Lua temporaries)
402+
403+local n, v = debug.getlocal(0, 1)
404+assert(v == 0 and n == "(C temporary)")
405+local n, v = debug.getlocal(0, 2)
406+assert(v == 2 and n == "(C temporary)")
407+assert(not debug.getlocal(0, 3))
408+assert(not debug.getlocal(0, 0))
409+
410+function f()
411+ assert(select(2, debug.getlocal(2,3)) == 1)
412+ assert(not debug.getlocal(2,4))
413+ debug.setlocal(2, 3, 10)
414+ return 20
415+end
416+
417+function g(a,b) return (a+1) + f() end
418+
419+assert(g(0,0) == 30)
420+
421+_G.f, _G.g = nil
422+
423+debug.sethook(nil);
424+assert(not debug.gethook())
425+
426+
427+-- minimal tests for setuservalue/getuservalue
428+do
429+ assert(not debug.setuservalue(io.stdin, 10))
430+ local a, b = debug.getuservalue(io.stdin, 10)
431+ assert(a == nil and not b)
432+end
433+
434+-- testing iteraction between multiple values x hooks
435+do
436+ local function f(...) return 3, ... end
437+ local count = 0
438+ local a = {}
439+ for i = 1, 100 do a[i] = i end
440+ debug.sethook(function () count = count + 1 end, "", 1)
441+ local t = {table.unpack(a)}
442+ assert(#t == 100)
443+ t = {table.unpack(a, 1, 3)}
444+ assert(#t == 3)
445+ t = {f(table.unpack(a, 1, 30))}
446+ assert(#t == 31)
447+end
448+
449+
450+-- testing access to function arguments
451+
452+local function collectlocals (level)
453+ local tab = {}
454+ for i = 1, math.huge do
455+ local n, v = debug.getlocal(level + 1, i)
456+ if not (n and string.find(n, "^[a-zA-Z0-9_]+$")) then
457+ break -- consider only real variables
458+ end
459+ tab[n] = v
460+ end
461+ return tab
462+end
463+
464+
465+local X = nil
466+a = {}
467+function a:f (a, b, ...) local arg = {...}; local c = 13 end
468+debug.sethook(function (e)
469+ assert(e == "call")
470+ dostring("XX = 12") -- test dostring inside hooks
471+ -- testing errors inside hooks
472+ assert(not pcall(load("a='joao'+1")))
473+ debug.sethook(function (e, l)
474+ assert(debug.getinfo(2, "l").currentline == l)
475+ local f,m,c = debug.gethook()
476+ assert(e == "line")
477+ assert(m == 'l' and c == 0)
478+ debug.sethook(nil) -- hook is called only once
479+ assert(not X) -- check that
480+ X = collectlocals(2)
481+ end, "l")
482+end, "c")
483+
484+a:f(1,2,3,4,5)
485+assert(X.self == a and X.a == 1 and X.b == 2 and X.c == nil)
486+assert(XX == 12)
487+assert(not debug.gethook())
488+_G.XX = nil
489+
490+
491+-- testing access to local variables in return hook (bug in 5.2)
492+do
493+ local X = false
494+
495+ local function foo (a, b, ...)
496+ do local x,y,z end
497+ local c, d = 10, 20
498+ return
499+ end
500+
501+ local function aux ()
502+ if debug.getinfo(2).name == "foo" then
503+ X = true -- to signal that it found 'foo'
504+ local tab = {a = 100, b = 200, c = 10, d = 20}
505+ for n, v in pairs(collectlocals(2)) do
506+ assert(tab[n] == v)
507+ tab[n] = undef
508+ end
509+ assert(next(tab) == nil) -- 'tab' must be empty
510+ end
511+ end
512+
513+ debug.sethook(aux, "r"); foo(100, 200); debug.sethook()
514+ assert(X)
515+
516+end
517+
518+
519+local function eqseq (t1, t2)
520+ assert(#t1 == #t2)
521+ for i = 1, #t1 do
522+ assert(t1[i] == t2[i])
523+ end
524+end
525+
526+
527+do print("testing inspection of parameters/returned values")
528+ local on = false
529+ local inp, out
530+
531+ local function hook (event)
532+ if not on then return end
533+ local ar = debug.getinfo(2, "ruS")
534+ local t = {}
535+ for i = ar.ftransfer, ar.ftransfer + ar.ntransfer - 1 do
536+ local _, v = debug.getlocal(2, i)
537+ t[#t + 1] = v
538+ end
539+ if event == "return" then
540+ out = t
541+ else
542+ inp = t
543+ end
544+ end
545+
546+ debug.sethook(hook, "cr")
547+
548+ on = true; math.sin(3); on = false
549+ eqseq(inp, {3}); eqseq(out, {math.sin(3)})
550+
551+ on = true; select(2, 10, 20, 30, 40); on = false
552+ eqseq(inp, {2, 10, 20, 30, 40}); eqseq(out, {20, 30, 40})
553+
554+ local function foo (a, ...) return ... end
555+ local function foo1 () on = not on; return foo(20, 10, 0) end
556+ foo1(); on = false
557+ eqseq(inp, {20}); eqseq(out, {10, 0})
558+
559+ debug.sethook()
560+end
561+
562+
563+
564+-- testing upvalue access
565+local function getupvalues (f)
566+ local t = {}
567+ local i = 1
568+ while true do
569+ local name, value = debug.getupvalue(f, i)
570+ if not name then break end
571+ assert(not t[name])
572+ t[name] = value
573+ i = i + 1
574+ end
575+ return t
576+end
577+
578+local a,b,c = 1,2,3
579+local function foo1 (a) b = a; return c end
580+local function foo2 (x) a = x; return c+b end
581+assert(not debug.getupvalue(foo1, 3))
582+assert(not debug.getupvalue(foo1, 0))
583+assert(not debug.setupvalue(foo1, 3, "xuxu"))
584+local t = getupvalues(foo1)
585+assert(t.a == nil and t.b == 2 and t.c == 3)
586+t = getupvalues(foo2)
587+assert(t.a == 1 and t.b == 2 and t.c == 3)
588+assert(debug.setupvalue(foo1, 1, "xuxu") == "b")
589+assert(({debug.getupvalue(foo2, 3)})[2] == "xuxu")
590+-- upvalues of C functions are allways "called" "" (the empty string)
591+assert(debug.getupvalue(string.gmatch("x", "x"), 1) == "")
592+
593+
594+-- testing count hooks
595+local a=0
596+debug.sethook(function (e) a=a+1 end, "", 1)
597+a=0; for i=1,1000 do end; assert(1000 < a and a < 1012)
598+debug.sethook(function (e) a=a+1 end, "", 4)
599+a=0; for i=1,1000 do end; assert(250 < a and a < 255)
600+local f,m,c = debug.gethook()
601+assert(m == "" and c == 4)
602+debug.sethook(function (e) a=a+1 end, "", 4000)
603+a=0; for i=1,1000 do end; assert(a == 0)
604+
605+do
606+ debug.sethook(print, "", 2^24 - 1) -- count upperbound
607+ local f,m,c = debug.gethook()
608+ assert(({debug.gethook()})[3] == 2^24 - 1)
609+end
610+
611+debug.sethook()
612+
613+local g, g1
614+
615+-- tests for tail calls
616+local function f (x)
617+ if x then
618+ assert(debug.getinfo(1, "S").what == "Lua")
619+ assert(debug.getinfo(1, "t").istailcall == true)
620+ local tail = debug.getinfo(2)
621+ assert(tail.func == g1 and tail.istailcall == true)
622+ assert(debug.getinfo(3, "S").what == "main")
623+ print"+"
624+ end
625+end
626+
627+function g(x) return f(x) end
628+
629+function g1(x) g(x) end
630+
631+local function h (x) local f=g1; return f(x) end
632+
633+h(true)
634+
635+local b = {}
636+debug.sethook(function (e) table.insert(b, e) end, "cr")
637+h(false)
638+debug.sethook()
639+local res = {"return", -- first return (from sethook)
640+ "call", "tail call", "call", "tail call",
641+ "return", "return",
642+ "call", -- last call (to sethook)
643+}
644+for i = 1, #res do assert(res[i] == table.remove(b, 1)) end
645+
646+b = 0
647+debug.sethook(function (e)
648+ if e == "tail call" then
649+ b = b + 1
650+ assert(debug.getinfo(2, "t").istailcall == true)
651+ else
652+ assert(debug.getinfo(2, "t").istailcall == false)
653+ end
654+ end, "c")
655+h(false)
656+debug.sethook()
657+assert(b == 2) -- two tail calls
658+
659+local lim = _soft and 3000 or 30000
660+local function foo (x)
661+ if x==0 then
662+ assert(debug.getinfo(2).what == "main")
663+ local info = debug.getinfo(1)
664+ assert(info.istailcall == true and info.func == foo)
665+ else return foo(x-1)
666+ end
667+end
668+
669+foo(lim)
670+
671+
672+print"+"
673+
674+
675+-- testing local function information
676+co = load[[
677+ local A = function ()
678+ return x
679+ end
680+ return
681+]]
682+
683+local a = 0
684+-- 'A' should be visible to debugger only after its complete definition
685+debug.sethook(function (e, l)
686+ if l == 3 then a = a + 1; assert(debug.getlocal(2, 1) == "(temporary)")
687+ elseif l == 4 then a = a + 1; assert(debug.getlocal(2, 1) == "A")
688+ end
689+end, "l")
690+co() -- run local function definition
691+debug.sethook() -- turn off hook
692+assert(a == 2) -- ensure all two lines where hooked
693+
694+-- testing traceback
695+
696+assert(debug.traceback(print) == print)
697+assert(debug.traceback(print, 4) == print)
698+assert(string.find(debug.traceback("hi", 4), "^hi\n"))
699+assert(string.find(debug.traceback("hi"), "^hi\n"))
700+assert(not string.find(debug.traceback("hi"), "'debug.traceback'"))
701+assert(string.find(debug.traceback("hi", 0), "'debug.traceback'"))
702+assert(string.find(debug.traceback(), "^stack traceback:\n"))
703+
704+do -- C-function names in traceback
705+ local st, msg = (function () return pcall end)()(debug.traceback)
706+ assert(st == true and string.find(msg, "pcall"))
707+end
708+
709+
710+-- testing nparams, nups e isvararg
711+local t = debug.getinfo(print, "u")
712+assert(t.isvararg == true and t.nparams == 0 and t.nups == 0)
713+
714+t = debug.getinfo(function (a,b,c) end, "u")
715+assert(t.isvararg == false and t.nparams == 3 and t.nups == 0)
716+
717+t = debug.getinfo(function (a,b,...) return t[a] end, "u")
718+assert(t.isvararg == true and t.nparams == 2 and t.nups == 1)
719+
720+t = debug.getinfo(1) -- main
721+assert(t.isvararg == true and t.nparams == 0 and t.nups == 1 and
722+ debug.getupvalue(t.func, 1) == "_ENV")
723+
724+t = debug.getinfo(math.sin) -- C function
725+assert(t.isvararg == true and t.nparams == 0 and t.nups == 0)
726+
727+t = debug.getinfo(string.gmatch("abc", "a")) -- C closure
728+assert(t.isvararg == true and t.nparams == 0 and t.nups > 0)
729+
730+
731+
732+-- testing debugging of coroutines
733+
734+local function checktraceback (co, p, level)
735+ local tb = debug.traceback(co, nil, level)
736+ local i = 0
737+ for l in string.gmatch(tb, "[^\n]+\n?") do
738+ assert(i == 0 or string.find(l, p[i]))
739+ i = i+1
740+ end
741+ assert(p[i] == undef)
742+end
743+
744+
745+local function f (n)
746+ if n > 0 then f(n-1)
747+ else coroutine.yield() end
748+end
749+
750+local co = coroutine.create(f)
751+coroutine.resume(co, 3)
752+checktraceback(co, {"yield", "db.lua", "db.lua", "db.lua", "db.lua"})
753+checktraceback(co, {"db.lua", "db.lua", "db.lua", "db.lua"}, 1)
754+checktraceback(co, {"db.lua", "db.lua", "db.lua"}, 2)
755+checktraceback(co, {"db.lua"}, 4)
756+checktraceback(co, {}, 40)
757+
758+
759+co = coroutine.create(function (x)
760+ local a = 1
761+ coroutine.yield(debug.getinfo(1, "l"))
762+ coroutine.yield(debug.getinfo(1, "l").currentline)
763+ return a
764+ end)
765+
766+local tr = {}
767+local foo = function (e, l) if l then table.insert(tr, l) end end
768+debug.sethook(co, foo, "lcr")
769+
770+local _, l = coroutine.resume(co, 10)
771+local x = debug.getinfo(co, 1, "lfLS")
772+assert(x.currentline == l.currentline and x.activelines[x.currentline])
773+assert(type(x.func) == "function")
774+for i=x.linedefined + 1, x.lastlinedefined do
775+ assert(x.activelines[i])
776+ x.activelines[i] = undef
777+end
778+assert(next(x.activelines) == nil) -- no 'extra' elements
779+assert(not debug.getinfo(co, 2))
780+local a,b = debug.getlocal(co, 1, 1)
781+assert(a == "x" and b == 10)
782+a,b = debug.getlocal(co, 1, 2)
783+assert(a == "a" and b == 1)
784+debug.setlocal(co, 1, 2, "hi")
785+assert(debug.gethook(co) == foo)
786+assert(#tr == 2 and
787+ tr[1] == l.currentline-1 and tr[2] == l.currentline)
788+
789+a,b,c = pcall(coroutine.resume, co)
790+assert(a and b and c == l.currentline+1)
791+checktraceback(co, {"yield", "in function <"})
792+
793+a,b = coroutine.resume(co)
794+assert(a and b == "hi")
795+assert(#tr == 4 and tr[4] == l.currentline+2)
796+assert(debug.gethook(co) == foo)
797+assert(not debug.gethook())
798+checktraceback(co, {})
799+
800+
801+-- check get/setlocal in coroutines
802+co = coroutine.create(function (x)
803+ local a, b = coroutine.yield(x)
804+ assert(a == 100 and b == nil)
805+ return x
806+end)
807+a, b = coroutine.resume(co, 10)
808+assert(a and b == 10)
809+a, b = debug.getlocal(co, 1, 1)
810+assert(a == "x" and b == 10)
811+assert(not debug.getlocal(co, 1, 5))
812+assert(debug.setlocal(co, 1, 1, 30) == "x")
813+assert(not debug.setlocal(co, 1, 5, 40))
814+a, b = coroutine.resume(co, 100)
815+assert(a and b == 30)
816+
817+
818+-- check traceback of suspended (or dead with error) coroutines
819+
820+function f(i)
821+ if i == 0 then error(i)
822+ else coroutine.yield(); f(i-1)
823+ end
824+end
825+
826+
827+co = coroutine.create(function (x) f(x) end)
828+a, b = coroutine.resume(co, 3)
829+t = {"'coroutine.yield'", "'f'", "in function <"}
830+while coroutine.status(co) == "suspended" do
831+ checktraceback(co, t)
832+ a, b = coroutine.resume(co)
833+ table.insert(t, 2, "'f'") -- one more recursive call to 'f'
834+end
835+t[1] = "'error'"
836+checktraceback(co, t)
837+
838+
839+-- test acessing line numbers of a coroutine from a resume inside
840+-- a C function (this is a known bug in Lua 5.0)
841+
842+local function g(x)
843+ coroutine.yield(x)
844+end
845+
846+local function f (i)
847+ debug.sethook(function () end, "l")
848+ for j=1,1000 do
849+ g(i+j)
850+ end
851+end
852+
853+local co = coroutine.wrap(f)
854+co(10)
855+pcall(co)
856+pcall(co)
857+
858+
859+assert(type(debug.getregistry()) == "table")
860+
861+
862+-- test tagmethod information
863+local a = {}
864+local function f (t)
865+ local info = debug.getinfo(1);
866+ assert(info.namewhat == "metamethod")
867+ a.op = info.name
868+ return info.name
869+end
870+setmetatable(a, {
871+ __index = f; __add = f; __div = f; __mod = f; __concat = f; __pow = f;
872+ __mul = f; __idiv = f; __unm = f; __len = f; __sub = f;
873+ __shl = f; __shr = f; __bor = f; __bxor = f;
874+ __eq = f; __le = f; __lt = f; __unm = f; __len = f; __band = f;
875+ __bnot = f;
876+})
877+
878+local b = setmetatable({}, getmetatable(a))
879+
880+assert(a[3] == "index" and a^3 == "pow" and a..a == "concat")
881+assert(a/3 == "div" and 3%a == "mod")
882+assert(a+3 == "add" and 3-a == "sub" and a*3 == "mul" and
883+ -a == "unm" and #a == "len" and a&3 == "band")
884+assert(a + 30000 == "add" and a - 3.0 == "sub" and a * 3.0 == "mul" and
885+ -a == "unm" and #a == "len" and a & 3 == "band")
886+assert(a|3 == "bor" and 3~a == "bxor" and a<<3 == "shl" and a>>1 == "shr")
887+assert (a==b and a.op == "eq")
888+assert (a>=b and a.op == "le")
889+assert ("x">=a and a.op == "le")
890+assert (a>b and a.op == "lt")
891+assert (a>10 and a.op == "lt")
892+assert(~a == "bnot")
893+
894+do -- testing for-iterator name
895+ local function f()
896+ assert(debug.getinfo(1).name == "for iterator")
897+ end
898+
899+ for i in f do end
900+end
901+
902+
903+do -- testing debug info for finalizers
904+ local name = nil
905+
906+ -- create a piece of garbage with a finalizer
907+ setmetatable({}, {__gc = function ()
908+ local t = debug.getinfo(1) -- get function information
909+ assert(t.namewhat == "metamethod")
910+ name = t.name
911+ end})
912+
913+ -- repeat until previous finalizer runs (setting 'name')
914+ repeat local a = {} until name
915+ assert(name == "__gc")
916+end
917+
918+
919+do
920+ print("testing traceback sizes")
921+
922+ local function countlines (s)
923+ return select(2, string.gsub(s, "\n", ""))
924+ end
925+
926+ local function deep (lvl, n)
927+ if lvl == 0 then
928+ return (debug.traceback("message", n))
929+ else
930+ return (deep(lvl-1, n))
931+ end
932+ end
933+
934+ local function checkdeep (total, start)
935+ local s = deep(total, start)
936+ local rest = string.match(s, "^message\nstack traceback:\n(.*)$")
937+ local cl = countlines(rest)
938+ -- at most 10 lines in first part, 11 in second, plus '...'
939+ assert(cl <= 10 + 11 + 1)
940+ local brk = string.find(rest, "%.%.%.\t%(skip")
941+ if brk then -- does message have '...'?
942+ local rest1 = string.sub(rest, 1, brk)
943+ local rest2 = string.sub(rest, brk, #rest)
944+ assert(countlines(rest1) == 10 and countlines(rest2) == 11)
945+ else
946+ assert(cl == total - start + 2)
947+ end
948+ end
949+
950+ for d = 1, 51, 10 do
951+ for l = 1, d do
952+ -- use coroutines to ensure complete control of the stack
953+ coroutine.wrap(checkdeep)(d, l)
954+ end
955+ end
956+
957+end
958+
959+
960+print("testing debug functions on chunk without debug info")
961+local prog = [[-- program to be loaded without debug information (strip)
962+local debug = require'debug'
963+local a = 12 -- a local variable
964+
965+local n, v = debug.getlocal(1, 1)
966+assert(n == "(temporary)" and v == debug) -- unkown name but known value
967+n, v = debug.getlocal(1, 2)
968+assert(n == "(temporary)" and v == 12) -- unkown name but known value
969+
970+-- a function with an upvalue
971+local f = function () local x; return a end
972+n, v = debug.getupvalue(f, 1)
973+assert(n == "(no name)" and v == 12)
974+assert(debug.setupvalue(f, 1, 13) == "(no name)")
975+assert(a == 13)
976+
977+local t = debug.getinfo(f)
978+assert(t.name == nil and t.linedefined > 0 and
979+ t.lastlinedefined == t.linedefined and
980+ t.short_src == "?")
981+assert(debug.getinfo(1).currentline == -1)
982+
983+t = debug.getinfo(f, "L").activelines
984+assert(next(t) == nil) -- active lines are empty
985+
986+-- dump/load a function without debug info
987+f = load(string.dump(f))
988+
989+t = debug.getinfo(f)
990+assert(t.name == nil and t.linedefined > 0 and
991+ t.lastlinedefined == t.linedefined and
992+ t.short_src == "?")
993+assert(debug.getinfo(1).currentline == -1)
994+
995+return a
996+]]
997+
998+
999+-- load 'prog' without debug info
1000+local f = assert(load(string.dump(load(prog), true)))
1001+
1002+assert(f() == 13)
1003+
1004+do -- bug in 5.4.0: line hooks in stripped code
1005+ local function foo ()
1006+ local a = 1
1007+ local b = 2
1008+ return b
1009+ end
1010+
1011+ local s = load(string.dump(foo, true))
1012+ local line = true
1013+ debug.sethook(function (e, l)
1014+ assert(e == "line")
1015+ line = l
1016+ end, "l")
1017+ assert(s() == 2); debug.sethook(nil)
1018+ assert(line == nil) -- hook called withoug debug info for 1st instruction
1019+end
1020+
1021+do -- tests for 'source' in binary dumps
1022+ local prog = [[
1023+ return function (x)
1024+ return function (y)
1025+ return x + y
1026+ end
1027+ end
1028+ ]]
1029+ local name = string.rep("x", 1000)
1030+ local p = assert(load(prog, name))
1031+ -- load 'p' as a binary chunk with debug information
1032+ local c = string.dump(p)
1033+ assert(#c > 1000 and #c < 2000) -- no repetition of 'source' in dump
1034+ local f = assert(load(c))
1035+ local g = f()
1036+ local h = g(3)
1037+ assert(h(5) == 8)
1038+ assert(debug.getinfo(f).source == name and -- all functions have 'source'
1039+ debug.getinfo(g).source == name and
1040+ debug.getinfo(h).source == name)
1041+ -- again, without debug info
1042+ local c = string.dump(p, true)
1043+ assert(#c < 500) -- no 'source' in dump
1044+ local f = assert(load(c))
1045+ local g = f()
1046+ local h = g(30)
1047+ assert(h(50) == 80)
1048+ assert(debug.getinfo(f).source == '=?' and -- no function has 'source'
1049+ debug.getinfo(g).source == '=?' and
1050+ debug.getinfo(h).source == '=?')
1051+end
1052+
1053+print"OK"
1054+
zod/luau/tests/errors.luanew file+696
@@ -0,0 +1,696 @@
1+-- $Id: testes/errors.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print("testing errors")
5+
6+local debug = require"debug"
7+
8+-- avoid problems with 'strict' module (which may generate other error messages)
9+local mt = getmetatable(_G) or {}
10+local oldmm = mt.__index
11+mt.__index = nil
12+
13+local function checkerr (msg, f, ...)
14+ local st, err = pcall(f, ...)
15+ assert(not st and string.find(err, msg))
16+end
17+
18+
19+local function doit (s)
20+ local f, msg = load(s)
21+ if not f then return msg end
22+ local cond, msg = pcall(f)
23+ return (not cond) and msg
24+end
25+
26+
27+local function checkmessage (prog, msg, debug)
28+ local m = doit(prog)
29+ if debug then print(m, msg) end
30+ assert(string.find(m, msg, 1, true))
31+end
32+
33+local function checksyntax (prog, extra, token, line)
34+ local msg = doit(prog)
35+ if not string.find(token, "^<%a") and not string.find(token, "^char%(")
36+ then token = "'"..token.."'" end
37+ token = string.gsub(token, "(%p)", "%%%1")
38+ local pt = string.format([[^%%[string ".*"%%]:%d: .- near %s$]],
39+ line, token)
40+ assert(string.find(msg, pt))
41+ assert(string.find(msg, msg, 1, true))
42+end
43+
44+
45+-- test error message with no extra info
46+assert(doit("error('hi', 0)") == 'hi')
47+
48+-- test error message with no info
49+assert(doit("error()") == nil)
50+
51+
52+-- test common errors/errors that crashed in the past
53+assert(doit("table.unpack({}, 1, n=2^30)"))
54+assert(doit("a=math.sin()"))
55+assert(not doit("tostring(1)") and doit("tostring()"))
56+assert(doit"tonumber()")
57+assert(doit"repeat until 1; a")
58+assert(doit"return;;")
59+assert(doit"assert(false)")
60+assert(doit"assert(nil)")
61+assert(doit("function a (... , ...) end"))
62+assert(doit("function a (, ...) end"))
63+assert(doit("local t={}; t = t[#t] + 1"))
64+
65+checksyntax([[
66+ local a = {4
67+
68+]], "'}' expected (to close '{' at line 1)", "<eof>", 3)
69+
70+
71+do -- testing errors in goto/break
72+ local function checksyntax (prog, msg, line)
73+ local st, err = load(prog)
74+ assert(string.find(err, "line " .. line))
75+ assert(string.find(err, msg, 1, true))
76+ end
77+
78+ checksyntax([[
79+ ::A:: a = 1
80+ ::A::
81+ ]], "label 'A' already defined", 1)
82+
83+ checksyntax([[
84+ a = 1
85+ goto A
86+ do ::A:: end
87+ ]], "no visible label 'A'", 2)
88+
89+end
90+
91+
92+if not T then
93+ (Message or print)
94+ ('\n >>> testC not active: skipping tests for messages in C <<<\n')
95+else
96+ print "testing memory error message"
97+ local a = {}
98+ for i = 1, 10000 do a[i] = true end -- preallocate array
99+ collectgarbage()
100+ T.totalmem(T.totalmem() + 10000)
101+ -- force a memory error (by a small margin)
102+ local st, msg = pcall(function()
103+ for i = 1, 100000 do a[i] = tostring(i) end
104+ end)
105+ T.totalmem(0)
106+ assert(not st and msg == "not enough" .. " memory")
107+
108+ -- stack space for luaL_traceback (bug in 5.4.6)
109+ local res = T.testC[[
110+ # push 16 elements on the stack
111+ pushnum 1; pushnum 1; pushnum 1; pushnum 1; pushnum 1;
112+ pushnum 1; pushnum 1; pushnum 1; pushnum 1; pushnum 1;
113+ pushnum 1; pushnum 1; pushnum 1; pushnum 1; pushnum 1;
114+ pushnum 1;
115+ # traceback should work with 4 remaining slots
116+ traceback xuxu 1;
117+ return 1
118+ ]]
119+ assert(string.find(res, "xuxu.-main chunk"))
120+end
121+
122+
123+-- tests for better error messages
124+
125+checkmessage("a = {} + 1", "arithmetic")
126+checkmessage("a = {} | 1", "bitwise operation")
127+checkmessage("a = {} < 1", "attempt to compare")
128+checkmessage("a = {} <= 1", "attempt to compare")
129+
130+checkmessage("aaa=1; bbbb=2; aaa=math.sin(3)+bbbb(3)", "global 'bbbb'")
131+checkmessage("aaa={}; do local aaa=1 end aaa:bbbb(3)", "method 'bbbb'")
132+checkmessage("local a={}; a.bbbb(3)", "field 'bbbb'")
133+assert(not string.find(doit"aaa={13}; local bbbb=1; aaa[bbbb](3)", "'bbbb'"))
134+checkmessage("aaa={13}; local bbbb=1; aaa[bbbb](3)", "number")
135+checkmessage("aaa=(1)..{}", "a table value")
136+
137+-- bug in 5.4.6
138+checkmessage("a = {_ENV = {}}; print(a._ENV.x + 1)", "field 'x'")
139+
140+_G.aaa, _G.bbbb = nil
141+
142+-- calls
143+checkmessage("local a; a(13)", "local 'a'")
144+checkmessage([[
145+ local a = setmetatable({}, {__add = 34})
146+ a = a + 1
147+]], "metamethod 'add'")
148+checkmessage([[
149+ local a = setmetatable({}, {__lt = {}})
150+ a = a > a
151+]], "metamethod 'lt'")
152+
153+-- tail calls
154+checkmessage("local a={}; return a.bbbb(3)", "field 'bbbb'")
155+checkmessage("aaa={}; do local aaa=1 end; return aaa:bbbb(3)", "method 'bbbb'")
156+
157+checkmessage("aaa = #print", "length of a function value")
158+checkmessage("aaa = #3", "length of a number value")
159+
160+_G.aaa = nil
161+
162+checkmessage("aaa.bbb:ddd(9)", "global 'aaa'")
163+checkmessage("local aaa={bbb=1}; aaa.bbb:ddd(9)", "field 'bbb'")
164+checkmessage("local aaa={bbb={}}; aaa.bbb:ddd(9)", "method 'ddd'")
165+checkmessage("local a,b,c; (function () a = b+1.1 end)()", "upvalue 'b'")
166+assert(not doit"local aaa={bbb={ddd=next}}; aaa.bbb:ddd(nil)")
167+
168+-- upvalues being indexed do not go to the stack
169+checkmessage("local a,b,cc; (function () a = cc[1] end)()", "upvalue 'cc'")
170+checkmessage("local a,b,cc; (function () a.x = 1 end)()", "upvalue 'a'")
171+
172+checkmessage("local _ENV = {x={}}; a = a + 1", "global 'a'")
173+
174+checkmessage("BB=1; local aaa={}; x=aaa+BB", "local 'aaa'")
175+checkmessage("aaa={}; x=3.3/aaa", "global 'aaa'")
176+checkmessage("aaa=2; BB=nil;x=aaa*BB", "global 'BB'")
177+checkmessage("aaa={}; x=-aaa", "global 'aaa'")
178+
179+-- short circuit
180+checkmessage("aaa=1; local aaa,bbbb=2,3; aaa = math.sin(1) and bbbb(3)",
181+ "local 'bbbb'")
182+checkmessage("aaa=1; local aaa,bbbb=2,3; aaa = bbbb(1) or aaa(3)",
183+ "local 'bbbb'")
184+checkmessage("local a,b,c,f = 1,1,1; f((a and b) or c)", "local 'f'")
185+checkmessage("local a,b,c = 1,1,1; ((a and b) or c)()", "call a number value")
186+assert(not string.find(doit"aaa={}; x=(aaa or aaa)+(aaa and aaa)", "'aaa'"))
187+assert(not string.find(doit"aaa={}; (aaa or aaa)()", "'aaa'"))
188+
189+checkmessage("print(print < 10)", "function with number")
190+checkmessage("print(print < print)", "two function values")
191+checkmessage("print('10' < 10)", "string with number")
192+checkmessage("print(10 < '23')", "number with string")
193+
194+-- float->integer conversions
195+checkmessage("local a = 2.0^100; x = a << 2", "local a")
196+checkmessage("local a = 1 >> 2.0^100", "has no integer representation")
197+checkmessage("local a = 10.1 << 2.0^100", "has no integer representation")
198+checkmessage("local a = 2.0^100 & 1", "has no integer representation")
199+checkmessage("local a = 2.0^100 & 1e100", "has no integer representation")
200+checkmessage("local a = 2.0 | 1e40", "has no integer representation")
201+checkmessage("local a = 2e100 ~ 1", "has no integer representation")
202+checkmessage("string.sub('a', 2.0^100)", "has no integer representation")
203+checkmessage("string.rep('a', 3.3)", "has no integer representation")
204+checkmessage("return 6e40 & 7", "has no integer representation")
205+checkmessage("return 34 << 7e30", "has no integer representation")
206+checkmessage("return ~-3e40", "has no integer representation")
207+checkmessage("return ~-3.009", "has no integer representation")
208+checkmessage("return 3.009 & 1", "has no integer representation")
209+checkmessage("return 34 >> {}", "table value")
210+checkmessage("aaa = 24 // 0", "divide by zero")
211+checkmessage("aaa = 1 % 0", "'n%0'")
212+
213+
214+-- type error for an object which is neither in an upvalue nor a register.
215+-- The following code will try to index the value 10 that is stored in
216+-- the metatable, without moving it to a register.
217+checkmessage("local a = setmetatable({}, {__index = 10}).x",
218+ "attempt to index a number value")
219+
220+
221+-- numeric for loops
222+checkmessage("for i = {}, 10 do end", "table")
223+checkmessage("for i = io.stdin, 10 do end", "FILE")
224+checkmessage("for i = {}, 10 do end", "initial value")
225+checkmessage("for i = 1, 'x', 10 do end", "string")
226+checkmessage("for i = 1, {}, 10 do end", "limit")
227+checkmessage("for i = 1, {} do end", "limit")
228+checkmessage("for i = 1, 10, print do end", "step")
229+checkmessage("for i = 1, 10, print do end", "function")
230+
231+-- passing light userdata instead of full userdata
232+_G.D = debug
233+checkmessage([[
234+ -- create light udata
235+ local x = D.upvalueid(function () return debug end, 1)
236+ D.setuservalue(x, {})
237+]], "light userdata")
238+_G.D = nil
239+
240+do -- named objects (field '__name')
241+ checkmessage("math.sin(io.input())", "(number expected, got FILE*)")
242+ _G.XX = setmetatable({}, {__name = "My Type"})
243+ assert(string.find(tostring(XX), "^My Type"))
244+ checkmessage("io.input(XX)", "(FILE* expected, got My Type)")
245+ checkmessage("return XX + 1", "on a My Type value")
246+ checkmessage("return ~io.stdin", "on a FILE* value")
247+ checkmessage("return XX < XX", "two My Type values")
248+ checkmessage("return {} < XX", "table with My Type")
249+ checkmessage("return XX < io.stdin", "My Type with FILE*")
250+ _G.XX = nil
251+
252+ if T then -- extra tests for 'luaL_tolstring'
253+ -- bug in 5.4.3; 'luaL_tolstring' with negative indices
254+ local x = setmetatable({}, {__name="TABLE"})
255+ assert(T.testC("Ltolstring -1; return 1", x) == tostring(x))
256+
257+ local a, b = T.testC("pushint 10; Ltolstring -2; return 2", x)
258+ assert(a == 10 and b == tostring(x))
259+
260+ setmetatable(x, {__tostring=function (o)
261+ assert(o == x)
262+ return "ABC"
263+ end})
264+ local a, b, c = T.testC("pushint 10; Ltolstring -2; return 3", x)
265+ assert(a == x and b == 10 and c == "ABC")
266+ end
267+end
268+
269+-- global functions
270+checkmessage("(io.write or print){}", "io.write")
271+checkmessage("(collectgarbage or print){}", "collectgarbage")
272+
273+-- errors in functions without debug info
274+do
275+ local f = function (a) return a + 1 end
276+ f = assert(load(string.dump(f, true)))
277+ assert(f(3) == 4)
278+ checkerr("^%?:%-1:", f, {})
279+
280+ -- code with a move to a local var ('OP_MOV A B' with A<B)
281+ f = function () local a; a = {}; return a + 2 end
282+ -- no debug info (so that 'a' is unknown)
283+ f = assert(load(string.dump(f, true)))
284+ -- symbolic execution should not get lost
285+ checkerr("^%?:%-1:.*table value", f)
286+end
287+
288+
289+-- tests for field accesses after RK limit
290+local t = {}
291+for i = 1, 1000 do
292+ t[i] = "aaa = x" .. i
293+end
294+local s = table.concat(t, "; ")
295+t = nil
296+checkmessage(s.."; aaa = bbb + 1", "global 'bbb'")
297+checkmessage("local _ENV=_ENV;"..s.."; aaa = bbb + 1", "global 'bbb'")
298+checkmessage(s.."; local t = {}; aaa = t.bbb + 1", "field 'bbb'")
299+checkmessage(s.."; local t = {}; t:bbb()", "method 'bbb'")
300+
301+checkmessage([[aaa=9
302+repeat until 3==3
303+local x=math.sin(math.cos(3))
304+if math.sin(1) == x then return math.sin(1) end -- tail call
305+local a,b = 1, {
306+ {x='a'..'b'..'c', y='b', z=x},
307+ {1,2,3,4,5} or 3+3<=3+3,
308+ 3+1>3+1,
309+ {d = x and aaa[x or y]}}
310+]], "global 'aaa'")
311+
312+checkmessage([[
313+local x,y = {},1
314+if math.sin(1) == 0 then return 3 end -- return
315+x.a()]], "field 'a'")
316+
317+checkmessage([[
318+prefix = nil
319+insert = nil
320+while 1 do
321+ local a
322+ if nil then break end
323+ insert(prefix, a)
324+end]], "global 'insert'")
325+
326+checkmessage([[ -- tail call
327+ return math.sin("a")
328+]], "sin")
329+
330+checkmessage([[collectgarbage("nooption")]], "invalid option")
331+
332+checkmessage([[x = print .. "a"]], "concatenate")
333+checkmessage([[x = "a" .. false]], "concatenate")
334+checkmessage([[x = {} .. 2]], "concatenate")
335+
336+checkmessage("getmetatable(io.stdin).__gc()", "no value")
337+
338+checkmessage([[
339+local Var
340+local function main()
341+ NoSuchName (function() Var=0 end)
342+end
343+main()
344+]], "global 'NoSuchName'")
345+print'+'
346+
347+aaa = {}; setmetatable(aaa, {__index = string})
348+checkmessage("aaa:sub()", "bad self")
349+checkmessage("string.sub('a', {})", "#2")
350+checkmessage("('a'):sub{}", "#1")
351+
352+checkmessage("table.sort({1,2,3}, table.sort)", "'table.sort'")
353+checkmessage("string.gsub('s', 's', setmetatable)", "'setmetatable'")
354+
355+_G.aaa = nil
356+
357+
358+-- tests for errors in coroutines
359+
360+local function f (n)
361+ local c = coroutine.create(f)
362+ local a,b = coroutine.resume(c)
363+ return b
364+end
365+assert(string.find(f(), "C stack overflow"))
366+
367+checkmessage("coroutine.yield()", "outside a coroutine")
368+
369+f = coroutine.wrap(function () table.sort({1,2,3}, coroutine.yield) end)
370+checkerr("yield across", f)
371+
372+
373+-- testing size of 'source' info; size of buffer for that info is
374+-- LUA_IDSIZE, declared as 60 in luaconf. Get one position for '\0'.
375+local idsize = 60 - 1
376+local function checksize (source)
377+ -- syntax error
378+ local _, msg = load("x", source)
379+ msg = string.match(msg, "^([^:]*):") -- get source (1st part before ':')
380+ assert(msg:len() <= idsize)
381+end
382+
383+for i = 60 - 10, 60 + 10 do -- check border cases around 60
384+ checksize("@" .. string.rep("x", i)) -- file names
385+ checksize(string.rep("x", i - 10)) -- string sources
386+ checksize("=" .. string.rep("x", i)) -- exact sources
387+end
388+
389+
390+-- testing line error
391+
392+local function lineerror (s, l)
393+ local err,msg = pcall(load(s))
394+ local line = tonumber(string.match(msg, ":(%d+):"))
395+ assert(line == l or (not line and not l))
396+end
397+
398+lineerror("local a\n for i=1,'a' do \n print(i) \n end", 2)
399+lineerror("\n local a \n for k,v in 3 \n do \n print(k) \n end", 3)
400+lineerror("\n\n for k,v in \n 3 \n do \n print(k) \n end", 4)
401+lineerror("function a.x.y ()\na=a+1\nend", 1)
402+
403+lineerror("a = \na\n+\n{}", 3)
404+lineerror("a = \n3\n+\n(\n4\n/\nprint)", 6)
405+lineerror("a = \nprint\n+\n(\n4\n/\n7)", 3)
406+
407+lineerror("a\n=\n-\n\nprint\n;", 3)
408+
409+lineerror([[
410+a
411+( -- <<
412+23)
413+]], 2)
414+
415+lineerror([[
416+local a = {x = 13}
417+a
418+.
419+x
420+( -- <<
421+23
422+)
423+]], 5)
424+
425+lineerror([[
426+local a = {x = 13}
427+a
428+.
429+x
430+(
431+23 + a
432+)
433+]], 6)
434+
435+local p = [[
436+ function g() f() end
437+ function f(x) error('a', XX) end
438+g()
439+]]
440+XX=3;lineerror((p), 3)
441+XX=0;lineerror((p), false)
442+XX=1;lineerror((p), 2)
443+XX=2;lineerror((p), 1)
444+_G.XX, _G.g, _G.f = nil
445+
446+
447+lineerror([[
448+local b = false
449+if not b then
450+ error 'test'
451+end]], 3)
452+
453+lineerror([[
454+local b = false
455+if not b then
456+ if not b then
457+ if not b then
458+ error 'test'
459+ end
460+ end
461+end]], 5)
462+
463+
464+-- bug in 5.4.0
465+lineerror([[
466+ local a = 0
467+ local b = 1
468+ local c = b % a
469+]], 3)
470+
471+do
472+ -- Force a negative estimate for base line. Error in instruction 2
473+ -- (after VARARGPREP, GETGLOBAL), with first absolute line information
474+ -- (forced by too many lines) in instruction 0.
475+ local s = string.format("%s return __A.x", string.rep("\n", 300))
476+ lineerror(s, 301)
477+end
478+
479+
480+if not _soft then
481+ -- several tests that exaust the Lua stack
482+ collectgarbage()
483+ print"testing stack overflow"
484+ local C = 0
485+ -- get line where stack overflow will happen
486+ local l = debug.getinfo(1, "l").currentline + 1
487+ local function auxy () C=C+1; auxy() end -- produce a stack overflow
488+ function YY ()
489+ collectgarbage("stop") -- avoid running finalizers without stack space
490+ auxy()
491+ collectgarbage("restart")
492+ end
493+
494+ local function checkstackmessage (m)
495+ print("(expected stack overflow after " .. C .. " calls)")
496+ C = 0 -- prepare next count
497+ return (string.find(m, "stack overflow"))
498+ end
499+ -- repeated stack overflows (to check stack recovery)
500+ assert(checkstackmessage(doit('YY()')))
501+ assert(checkstackmessage(doit('YY()')))
502+ assert(checkstackmessage(doit('YY()')))
503+
504+ _G.YY = nil
505+
506+
507+ -- error lines in stack overflow
508+ local l1
509+ local function g(x)
510+ l1 = debug.getinfo(x, "l").currentline + 2
511+ collectgarbage("stop") -- avoid running finalizers without stack space
512+ auxy()
513+ collectgarbage("restart")
514+ end
515+ local _, stackmsg = xpcall(g, debug.traceback, 1)
516+ print('+')
517+ local stack = {}
518+ for line in string.gmatch(stackmsg, "[^\n]*") do
519+ local curr = string.match(line, ":(%d+):")
520+ if curr then table.insert(stack, tonumber(curr)) end
521+ end
522+ local i=1
523+ while stack[i] ~= l1 do
524+ assert(stack[i] == l)
525+ i = i+1
526+ end
527+ assert(i > 15)
528+
529+
530+ -- error in error handling
531+ local res, msg = xpcall(error, error)
532+ assert(not res and type(msg) == 'string')
533+ print('+')
534+
535+ local function f (x)
536+ if x==0 then error('a\n')
537+ else
538+ local aux = function () return f(x-1) end
539+ local a,b = xpcall(aux, aux)
540+ return a,b
541+ end
542+ end
543+ f(3)
544+
545+ local function loop (x,y,z) return 1 + loop(x, y, z) end
546+
547+ local res, msg = xpcall(loop, function (m)
548+ assert(string.find(m, "stack overflow"))
549+ checkerr("error handling", loop)
550+ assert(math.sin(0) == 0)
551+ return 15
552+ end)
553+ assert(msg == 15)
554+
555+ local f = function ()
556+ for i = 999900, 1000000, 1 do table.unpack({}, 1, i) end
557+ end
558+ checkerr("too many results", f)
559+
560+end
561+
562+
563+do
564+ -- non string messages
565+ local t = {}
566+ local res, msg = pcall(function () error(t) end)
567+ assert(not res and msg == t)
568+
569+ res, msg = pcall(function () error(nil) end)
570+ assert(not res and msg == nil)
571+
572+ local function f() error{msg='x'} end
573+ res, msg = xpcall(f, function (r) return {msg=r.msg..'y'} end)
574+ assert(msg.msg == 'xy')
575+
576+ -- 'assert' with extra arguments
577+ res, msg = pcall(assert, false, "X", t)
578+ assert(not res and msg == "X")
579+
580+ -- 'assert' with no message
581+ res, msg = pcall(function () assert(false) end)
582+ local line = string.match(msg, "%w+%.lua:(%d+): assertion failed!$")
583+ assert(tonumber(line) == debug.getinfo(1, "l").currentline - 2)
584+
585+ -- 'assert' with non-string messages
586+ res, msg = pcall(assert, false, t)
587+ assert(not res and msg == t)
588+
589+ res, msg = pcall(assert, nil, nil)
590+ assert(not res and msg == nil)
591+
592+ -- 'assert' without arguments
593+ res, msg = pcall(assert)
594+ assert(not res and string.find(msg, "value expected"))
595+end
596+
597+-- xpcall with arguments
598+local a, b, c = xpcall(string.find, error, "alo", "al")
599+assert(a and b == 1 and c == 2)
600+a, b, c = xpcall(string.find, function (x) return {} end, true, "al")
601+assert(not a and type(b) == "table" and c == nil)
602+
603+
604+print("testing tokens in error messages")
605+checksyntax("syntax error", "", "error", 1)
606+checksyntax("1.000", "", "1.000", 1)
607+checksyntax("[[a]]", "", "[[a]]", 1)
608+checksyntax("'aa'", "", "'aa'", 1)
609+checksyntax("while << do end", "", "<<", 1)
610+checksyntax("for >> do end", "", ">>", 1)
611+
612+-- test invalid non-printable char in a chunk
613+checksyntax("a\1a = 1", "", "<\\1>", 1)
614+
615+-- test 255 as first char in a chunk
616+checksyntax("\255a = 1", "", "<\\255>", 1)
617+
618+doit('I = load("a=9+"); aaa=3')
619+assert(_G.aaa==3 and not _G.I)
620+_G.I,_G.aaa = nil
621+print('+')
622+
623+local lim = 1000
624+if _soft then lim = 100 end
625+for i=1,lim do
626+ doit('a = ')
627+ doit('a = 4+nil')
628+end
629+
630+
631+-- testing syntax limits
632+
633+local function testrep (init, rep, close, repc, finalresult)
634+ local s = init .. string.rep(rep, 100) .. close .. string.rep(repc, 100)
635+ local res, msg = load(s)
636+ assert(res) -- 100 levels is OK
637+ if (finalresult) then
638+ assert(res() == finalresult)
639+ end
640+ s = init .. string.rep(rep, 500)
641+ local res, msg = load(s) -- 500 levels not ok
642+ assert(not res and (string.find(msg, "too many") or
643+ string.find(msg, "overflow")))
644+end
645+
646+testrep("local a; a", ",a", "= 1", ",1") -- multiple assignment
647+testrep("local a; a=", "{", "0", "}")
648+testrep("return ", "(", "2", ")", 2)
649+testrep("local function a (x) return x end; return ", "a(", "2.2", ")", 2.2)
650+testrep("", "do ", "", " end")
651+testrep("", "while a do ", "", " end")
652+testrep("local a; ", "if a then else ", "", " end")
653+testrep("", "function foo () ", "", " end")
654+testrep("local a = ''; return ", "a..", "'a'", "", "a")
655+testrep("local a = 1; return ", "a^", "a", "", 1)
656+
657+checkmessage("a = f(x" .. string.rep(",x", 260) .. ")", "too many registers")
658+
659+
660+-- testing other limits
661+
662+-- upvalues
663+local lim = 127
664+local s = "local function fooA ()\n local "
665+for j = 1,lim do
666+ s = s.."a"..j..", "
667+end
668+s = s.."b,c\n"
669+s = s.."local function fooB ()\n local "
670+for j = 1,lim do
671+ s = s.."b"..j..", "
672+end
673+s = s.."b\n"
674+s = s.."function fooC () return b+c"
675+local c = 1+2
676+for j = 1,lim do
677+ s = s.."+a"..j.."+b"..j
678+ c = c + 2
679+end
680+s = s.."\nend end end"
681+local a,b = load(s)
682+assert(c > 255 and string.find(b, "too many upvalues") and
683+ string.find(b, "line 5"))
684+
685+-- local variables
686+s = "\nfunction foo ()\n local "
687+for j = 1,300 do
688+ s = s.."a"..j..", "
689+end
690+s = s.."b\n"
691+local a,b = load(s)
692+assert(string.find(b, "line 2") and string.find(b, "too many local variables"))
693+
694+mt.__index = oldmm
695+
696+print('OK')
zod/luau/tests/events.luanew file+491
@@ -0,0 +1,491 @@
1+-- $Id: testes/events.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print('testing metatables')
5+
6+local debug = require'debug'
7+
8+X = 20; B = 30
9+
10+_ENV = setmetatable({}, {__index=_G})
11+
12+collectgarbage()
13+
14+X = X+10
15+assert(X == 30 and _G.X == 20)
16+B = false
17+assert(B == false)
18+_ENV["B"] = undef
19+assert(B == 30)
20+
21+assert(getmetatable{} == nil)
22+assert(getmetatable(4) == nil)
23+assert(getmetatable(nil) == nil)
24+a={name = "NAME"}; setmetatable(a, {__metatable = "xuxu",
25+ __tostring=function(x) return x.name end})
26+assert(getmetatable(a) == "xuxu")
27+assert(tostring(a) == "NAME")
28+-- cannot change a protected metatable
29+assert(pcall(setmetatable, a, {}) == false)
30+a.name = "gororoba"
31+assert(tostring(a) == "gororoba")
32+
33+local a, t = {10,20,30; x="10", y="20"}, {}
34+assert(setmetatable(a,t) == a)
35+assert(getmetatable(a) == t)
36+assert(setmetatable(a,nil) == a)
37+assert(getmetatable(a) == nil)
38+assert(setmetatable(a,t) == a)
39+
40+
41+function f (t, i, e)
42+ assert(not e)
43+ local p = rawget(t, "parent")
44+ return (p and p[i]+3), "dummy return"
45+end
46+
47+t.__index = f
48+
49+a.parent = {z=25, x=12, [4] = 24}
50+assert(a[1] == 10 and a.z == 28 and a[4] == 27 and a.x == "10")
51+
52+collectgarbage()
53+
54+a = setmetatable({}, t)
55+function f(t, i, v) rawset(t, i, v-3) end
56+setmetatable(t, t) -- causes a bug in 5.1 !
57+t.__newindex = f
58+a[1] = 30; a.x = "101"; a[5] = 200
59+assert(a[1] == 27 and a.x == 98 and a[5] == 197)
60+
61+do -- bug in Lua 5.3.2
62+ local mt = {}
63+ mt.__newindex = mt
64+ local t = setmetatable({}, mt)
65+ t[1] = 10 -- will segfault on some machines
66+ assert(mt[1] == 10)
67+end
68+
69+
70+local c = {}
71+a = setmetatable({}, t)
72+t.__newindex = c
73+t.__index = c
74+a[1] = 10; a[2] = 20; a[3] = 90;
75+for i = 4, 20 do a[i] = i * 10 end
76+assert(a[1] == 10 and a[2] == 20 and a[3] == 90)
77+for i = 4, 20 do assert(a[i] == i * 10) end
78+assert(next(a) == nil)
79+
80+
81+do
82+ local a;
83+ a = setmetatable({}, {__index = setmetatable({},
84+ {__index = setmetatable({},
85+ {__index = function (_,n) return a[n-3]+4, "lixo" end})})})
86+ a[0] = 20
87+ for i=0,10 do
88+ assert(a[i*3] == 20 + i*4)
89+ end
90+end
91+
92+
93+do -- newindex
94+ local foi
95+ local a = {}
96+ for i=1,10 do a[i] = 0; a['a'..i] = 0; end
97+ setmetatable(a, {__newindex = function (t,k,v) foi=true; rawset(t,k,v) end})
98+ foi = false; a[1]=0; assert(not foi)
99+ foi = false; a['a1']=0; assert(not foi)
100+ foi = false; a['a11']=0; assert(foi)
101+ foi = false; a[11]=0; assert(foi)
102+ foi = false; a[1]=undef; assert(not foi)
103+ a[1] = undef
104+ foi = false; a[1]=nil; assert(foi)
105+end
106+
107+
108+setmetatable(t, nil)
109+function f (t, ...) return t, {...} end
110+t.__call = f
111+
112+do
113+ local x,y = a(table.unpack{'a', 1})
114+ assert(x==a and y[1]=='a' and y[2]==1 and y[3]==undef)
115+ x,y = a()
116+ assert(x==a and y[1]==undef)
117+end
118+
119+
120+local b = setmetatable({}, t)
121+setmetatable(b,t)
122+
123+function f(op)
124+ return function (...) cap = {[0] = op, ...} ; return (...) end
125+end
126+t.__add = f("add")
127+t.__sub = f("sub")
128+t.__mul = f("mul")
129+t.__div = f("div")
130+t.__idiv = f("idiv")
131+t.__mod = f("mod")
132+t.__unm = f("unm")
133+t.__pow = f("pow")
134+t.__len = f("len")
135+t.__band = f("band")
136+t.__bor = f("bor")
137+t.__bxor = f("bxor")
138+t.__shl = f("shl")
139+t.__shr = f("shr")
140+t.__bnot = f("bnot")
141+t.__lt = f("lt")
142+t.__le = f("le")
143+
144+
145+local function checkcap (t)
146+ assert(#cap + 1 == #t)
147+ for i = 1, #t do
148+ assert(cap[i - 1] == t[i])
149+ assert(math.type(cap[i - 1]) == math.type(t[i]))
150+ end
151+end
152+
153+-- Some tests are done inside small anonymous functions to ensure
154+-- that constants go to constant table even in debug compilation,
155+-- when the constant table is very small.
156+assert(b+5 == b); checkcap{"add", b, 5}
157+assert(5.2 + b == 5.2); checkcap{"add", 5.2, b}
158+assert(b+'5' == b); checkcap{"add", b, '5'}
159+assert(5+b == 5); checkcap{"add", 5, b}
160+assert('5'+b == '5'); checkcap{"add", '5', b}
161+b=b-3; assert(getmetatable(b) == t); checkcap{"sub", b, 3}
162+assert(5-a == 5); checkcap{"sub", 5, a}
163+assert('5'-a == '5'); checkcap{"sub", '5', a}
164+assert(a*a == a); checkcap{"mul", a, a}
165+assert(a/0 == a); checkcap{"div", a, 0}
166+assert(a/0.0 == a); checkcap{"div", a, 0.0}
167+assert(a%2 == a); checkcap{"mod", a, 2}
168+assert(a // (1/0) == a); checkcap{"idiv", a, 1/0}
169+;(function () assert(a & "hi" == a) end)(); checkcap{"band", a, "hi"}
170+;(function () assert(10 & a == 10) end)(); checkcap{"band", 10, a}
171+;(function () assert(a | 10 == a) end)(); checkcap{"bor", a, 10}
172+assert(a | "hi" == a); checkcap{"bor", a, "hi"}
173+assert("hi" ~ a == "hi"); checkcap{"bxor", "hi", a}
174+;(function () assert(10 ~ a == 10) end)(); checkcap{"bxor", 10, a}
175+assert(-a == a); checkcap{"unm", a, a}
176+assert(a^4.0 == a); checkcap{"pow", a, 4.0}
177+assert(a^'4' == a); checkcap{"pow", a, '4'}
178+assert(4^a == 4); checkcap{"pow", 4, a}
179+assert('4'^a == '4'); checkcap{"pow", '4', a}
180+assert(#a == a); checkcap{"len", a, a}
181+assert(~a == a); checkcap{"bnot", a, a}
182+assert(a << 3 == a); checkcap{"shl", a, 3}
183+assert(1.5 >> a == 1.5); checkcap{"shr", 1.5, a}
184+
185+-- for comparison operators, all results are true
186+assert(5.0 > a); checkcap{"lt", a, 5.0}
187+assert(a >= 10); checkcap{"le", 10, a}
188+assert(a <= -10.0); checkcap{"le", a, -10.0}
189+assert(a < -10); checkcap{"lt", a, -10}
190+
191+
192+-- test for rawlen
193+t = setmetatable({1,2,3}, {__len = function () return 10 end})
194+assert(#t == 10 and rawlen(t) == 3)
195+assert(rawlen"abc" == 3)
196+assert(not pcall(rawlen, io.stdin))
197+assert(not pcall(rawlen, 34))
198+assert(not pcall(rawlen))
199+
200+-- rawlen for long strings
201+assert(rawlen(string.rep('a', 1000)) == 1000)
202+
203+
204+t = {}
205+t.__lt = function (a,b,c)
206+ collectgarbage()
207+ assert(c == nil)
208+ if type(a) == 'table' then a = a.x end
209+ if type(b) == 'table' then b = b.x end
210+ return a<b, "dummy"
211+end
212+
213+t.__le = function (a,b,c)
214+ assert(c == nil)
215+ if type(a) == 'table' then a = a.x end
216+ if type(b) == 'table' then b = b.x end
217+ return a<=b, "dummy"
218+end
219+
220+t.__eq = function (a,b,c)
221+ assert(c == nil)
222+ if type(a) == 'table' then a = a.x end
223+ if type(b) == 'table' then b = b.x end
224+ return a == b, "dummy"
225+end
226+
227+function Op(x) return setmetatable({x=x}, t) end
228+
229+local function test (a, b, c)
230+ assert(not(Op(1)<Op(1)) and (Op(1)<Op(2)) and not(Op(2)<Op(1)))
231+ assert(not(1 < Op(1)) and (Op(1) < 2) and not(2 < Op(1)))
232+ assert(not(Op('a')<Op('a')) and (Op('a')<Op('b')) and not(Op('b')<Op('a')))
233+ assert(not('a' < Op('a')) and (Op('a') < 'b') and not(Op('b') < Op('a')))
234+ assert((Op(1)<=Op(1)) and (Op(1)<=Op(2)) and not(Op(2)<=Op(1)))
235+ assert((Op('a')<=Op('a')) and (Op('a')<=Op('b')) and not(Op('b')<=Op('a')))
236+ assert(not(Op(1)>Op(1)) and not(Op(1)>Op(2)) and (Op(2)>Op(1)))
237+ assert(not(Op('a')>Op('a')) and not(Op('a')>Op('b')) and (Op('b')>Op('a')))
238+ assert((Op(1)>=Op(1)) and not(Op(1)>=Op(2)) and (Op(2)>=Op(1)))
239+ assert((1 >= Op(1)) and not(1 >= Op(2)) and (Op(2) >= 1))
240+ assert((Op('a')>=Op('a')) and not(Op('a')>=Op('b')) and (Op('b')>=Op('a')))
241+ assert(('a' >= Op('a')) and not(Op('a') >= 'b') and (Op('b') >= Op('a')))
242+ assert(Op(1) == Op(1) and Op(1) ~= Op(2))
243+ assert(Op('a') == Op('a') and Op('a') ~= Op('b'))
244+ assert(a == a and a ~= b)
245+ assert(Op(3) == c)
246+end
247+
248+test(Op(1), Op(2), Op(3))
249+
250+
251+-- test `partial order'
252+
253+local function rawSet(x)
254+ local y = {}
255+ for _,k in pairs(x) do y[k] = 1 end
256+ return y
257+end
258+
259+local function Set(x)
260+ return setmetatable(rawSet(x), t)
261+end
262+
263+t.__lt = function (a,b)
264+ for k in pairs(a) do
265+ if not b[k] then return false end
266+ b[k] = undef
267+ end
268+ return next(b) ~= nil
269+end
270+
271+t.__le = function (a,b)
272+ for k in pairs(a) do
273+ if not b[k] then return false end
274+ end
275+ return true
276+end
277+
278+assert(Set{1,2,3} < Set{1,2,3,4})
279+assert(not(Set{1,2,3,4} < Set{1,2,3,4}))
280+assert((Set{1,2,3,4} <= Set{1,2,3,4}))
281+assert((Set{1,2,3,4} >= Set{1,2,3,4}))
282+assert(not (Set{1,3} <= Set{3,5}))
283+assert(not(Set{1,3} <= Set{3,5}))
284+assert(not(Set{1,3} >= Set{3,5}))
285+
286+
287+t.__eq = function (a,b)
288+ for k in pairs(a) do
289+ if not b[k] then return false end
290+ b[k] = undef
291+ end
292+ return next(b) == nil
293+end
294+
295+local s = Set{1,3,5}
296+assert(s == Set{3,5,1})
297+assert(not rawequal(s, Set{3,5,1}))
298+assert(rawequal(s, s))
299+assert(Set{1,3,5,1} == rawSet{3,5,1})
300+assert(rawSet{1,3,5,1} == Set{3,5,1})
301+assert(Set{1,3,5} ~= Set{3,5,1,6})
302+
303+-- '__eq' is not used for table accesses
304+t[Set{1,3,5}] = 1
305+assert(t[Set{1,3,5}] == undef)
306+
307+
308+do -- test invalidating flags
309+ local mt = {__eq = true}
310+ local a = setmetatable({10}, mt)
311+ local b = setmetatable({10}, mt)
312+ mt.__eq = nil
313+ assert(a ~= b) -- no metamethod
314+ mt.__eq = function (x,y) return x[1] == y[1] end
315+ assert(a == b) -- must use metamethod now
316+end
317+
318+
319+if not T then
320+ (Message or print)('\n >>> testC not active: skipping tests for \z
321+userdata <<<\n')
322+else
323+ local u1 = T.newuserdata(0, 1)
324+ local u2 = T.newuserdata(0, 1)
325+ local u3 = T.newuserdata(0, 1)
326+ assert(u1 ~= u2 and u1 ~= u3)
327+ debug.setuservalue(u1, 1);
328+ debug.setuservalue(u2, 2);
329+ debug.setuservalue(u3, 1);
330+ debug.setmetatable(u1, {__eq = function (a, b)
331+ return debug.getuservalue(a) == debug.getuservalue(b)
332+ end})
333+ debug.setmetatable(u2, {__eq = function (a, b)
334+ return true
335+ end})
336+ assert(u1 == u3 and u3 == u1 and u1 ~= u2)
337+ assert(u2 == u1 and u2 == u3 and u3 == u2)
338+ assert(u2 ~= {}) -- different types cannot be equal
339+ assert(rawequal(u1, u1) and not rawequal(u1, u3))
340+
341+ local mirror = {}
342+ debug.setmetatable(u3, {__index = mirror, __newindex = mirror})
343+ for i = 1, 10 do u3[i] = i end
344+ for i = 1, 10 do assert(u3[i] == i) end
345+end
346+
347+
348+t.__concat = function (a,b,c)
349+ assert(c == nil)
350+ if type(a) == 'table' then a = a.val end
351+ if type(b) == 'table' then b = b.val end
352+ if A then return a..b
353+ else
354+ return setmetatable({val=a..b}, t)
355+ end
356+end
357+
358+c = {val="c"}; setmetatable(c, t)
359+d = {val="d"}; setmetatable(d, t)
360+
361+A = true
362+assert(c..d == 'cd')
363+assert(0 .."a".."b"..c..d.."e".."f"..(5+3).."g" == "0abcdef8g")
364+
365+A = false
366+assert((c..d..c..d).val == 'cdcd')
367+x = c..d
368+assert(getmetatable(x) == t and x.val == 'cd')
369+x = 0 .."a".."b"..c..d.."e".."f".."g"
370+assert(x.val == "0abcdefg")
371+
372+
373+-- concat metamethod x numbers (bug in 5.1.1)
374+c = {}
375+local x
376+setmetatable(c, {__concat = function (a,b)
377+ assert(type(a) == "number" and b == c or type(b) == "number" and a == c)
378+ return c
379+end})
380+assert(c..5 == c and 5 .. c == c)
381+assert(4 .. c .. 5 == c and 4 .. 5 .. 6 .. 7 .. c == c)
382+
383+
384+-- test comparison compatibilities
385+local t1, t2, c, d
386+t1 = {}; c = {}; setmetatable(c, t1)
387+d = {}
388+t1.__eq = function () return true end
389+t1.__lt = function () return true end
390+t1.__le = function () return false end
391+setmetatable(d, t1)
392+assert(c == d and c < d and not(d <= c))
393+t2 = {}
394+t2.__eq = t1.__eq
395+t2.__lt = t1.__lt
396+setmetatable(d, t2)
397+assert(c == d and c < d and not(d <= c))
398+
399+
400+
401+-- test for several levels of calls
402+local i
403+local tt = {
404+ __call = function (t, ...)
405+ i = i+1
406+ if t.f then return t.f(...)
407+ else return {...}
408+ end
409+ end
410+}
411+
412+local a = setmetatable({}, tt)
413+local b = setmetatable({f=a}, tt)
414+local c = setmetatable({f=b}, tt)
415+
416+i = 0
417+x = c(3,4,5)
418+assert(i == 3 and x[1] == 3 and x[3] == 5)
419+
420+
421+assert(_G.X == 20)
422+
423+_G.X, _G.B = nil
424+
425+
426+print'+'
427+
428+local _g = _G
429+_ENV = setmetatable({}, {__index=function (_,k) return _g[k] end})
430+
431+
432+a = {}
433+rawset(a, "x", 1, 2, 3)
434+assert(a.x == 1 and rawget(a, "x", 3) == 1)
435+
436+print '+'
437+
438+-- testing metatables for basic types
439+mt = {__index = function (a,b) return a+b end,
440+ __len = function (x) return math.floor(x) end}
441+debug.setmetatable(10, mt)
442+assert(getmetatable(-2) == mt)
443+assert((10)[3] == 13)
444+assert((10)["3"] == 13)
445+assert(#3.45 == 3)
446+debug.setmetatable(23, nil)
447+assert(getmetatable(-2) == nil)
448+
449+debug.setmetatable(true, mt)
450+assert(getmetatable(false) == mt)
451+mt.__index = function (a,b) return a or b end
452+assert((true)[false] == true)
453+assert((false)[false] == false)
454+debug.setmetatable(false, nil)
455+assert(getmetatable(true) == nil)
456+
457+debug.setmetatable(nil, mt)
458+assert(getmetatable(nil) == mt)
459+mt.__add = function (a,b) return (a or 1) + (b or 2) end
460+assert(10 + nil == 12)
461+assert(nil + 23 == 24)
462+assert(nil + nil == 3)
463+debug.setmetatable(nil, nil)
464+assert(getmetatable(nil) == nil)
465+
466+debug.setmetatable(nil, {})
467+
468+
469+-- loops in delegation
470+a = {}; setmetatable(a, a); a.__index = a; a.__newindex = a
471+assert(not pcall(function (a,b) return a[b] end, a, 10))
472+assert(not pcall(function (a,b,c) a[b] = c end, a, 10, true))
473+
474+-- bug in 5.1
475+T, K, V = nil
476+grandparent = {}
477+grandparent.__newindex = function(t,k,v) T=t; K=k; V=v end
478+
479+parent = {}
480+parent.__newindex = parent
481+setmetatable(parent, grandparent)
482+
483+child = setmetatable({}, parent)
484+child.foo = 10 --> CRASH (on some machines)
485+assert(T == parent and K == "foo" and V == 10)
486+
487+print 'OK'
488+
489+return 12
490+
491+
zod/luau/tests/files.luanew file+951
@@ -0,0 +1,951 @@
1+-- $Id: testes/files.lua $
2+-- See Copyright Notice in file all.lua
3+
4+local debug = require "debug"
5+
6+local maxint = math.maxinteger
7+
8+assert(type(os.getenv"PATH") == "string")
9+
10+assert(io.input(io.stdin) == io.stdin)
11+assert(not pcall(io.input, "non-existent-file"))
12+assert(io.output(io.stdout) == io.stdout)
13+
14+
15+local function testerr (msg, f, ...)
16+ local stat, err = pcall(f, ...)
17+ return (not stat and string.find(err, msg, 1, true))
18+end
19+
20+
21+local function checkerr (msg, f, ...)
22+ assert(testerr(msg, f, ...))
23+end
24+
25+
26+-- cannot close standard files
27+assert(not io.close(io.stdin) and
28+ not io.stdout:close() and
29+ not io.stderr:close())
30+
31+-- cannot call close method without an argument (new in 5.3.5)
32+checkerr("got no value", io.stdin.close)
33+
34+
35+assert(type(io.input()) == "userdata" and io.type(io.output()) == "file")
36+assert(type(io.stdin) == "userdata" and io.type(io.stderr) == "file")
37+assert(not io.type(8))
38+local a = {}; setmetatable(a, {})
39+assert(not io.type(a))
40+
41+assert(getmetatable(io.input()).__name == "FILE*")
42+
43+local a,b,c = io.open('xuxu_nao_existe')
44+assert(not a and type(b) == "string" and type(c) == "number")
45+
46+a,b,c = io.open('/a/b/c/d', 'w')
47+assert(not a and type(b) == "string" and type(c) == "number")
48+
49+local file = os.tmpname()
50+local f, msg = io.open(file, "w")
51+if not f then
52+ (Message or print)("'os.tmpname' file cannot be open; skipping file tests")
53+
54+else --{ most tests here need tmpname
55+f:close()
56+
57+print('testing i/o')
58+
59+local otherfile = os.tmpname()
60+
61+checkerr("invalid mode", io.open, file, "rw")
62+checkerr("invalid mode", io.open, file, "rb+")
63+checkerr("invalid mode", io.open, file, "r+bk")
64+checkerr("invalid mode", io.open, file, "")
65+checkerr("invalid mode", io.open, file, "+")
66+checkerr("invalid mode", io.open, file, "b")
67+assert(io.open(file, "r+b")):close()
68+assert(io.open(file, "r+")):close()
69+assert(io.open(file, "rb")):close()
70+
71+assert(os.setlocale('C', 'all'))
72+
73+io.input(io.stdin); io.output(io.stdout);
74+
75+os.remove(file)
76+assert(not loadfile(file))
77+checkerr("", dofile, file)
78+assert(not io.open(file))
79+io.output(file)
80+assert(io.output() ~= io.stdout)
81+
82+if not _port then -- invalid seek
83+ local status, msg, code = io.stdin:seek("set", 1000)
84+ assert(not status and type(msg) == "string" and type(code) == "number")
85+end
86+
87+assert(io.output():seek() == 0)
88+assert(io.write("alo alo"):seek() == string.len("alo alo"))
89+assert(io.output():seek("cur", -3) == string.len("alo alo")-3)
90+assert(io.write("joao"))
91+assert(io.output():seek("end") == string.len("alo joao"))
92+
93+assert(io.output():seek("set") == 0)
94+
95+assert(io.write('"alo"', "{a}\n", "second line\n", "third line \n"))
96+assert(io.write('Xfourth_line'))
97+io.output(io.stdout)
98+collectgarbage() -- file should be closed by GC
99+assert(io.input() == io.stdin and rawequal(io.output(), io.stdout))
100+print('+')
101+
102+-- test GC for files
103+collectgarbage()
104+for i=1,120 do
105+ for i=1,5 do
106+ io.input(file)
107+ assert(io.open(file, 'r'))
108+ io.lines(file)
109+ end
110+ collectgarbage()
111+end
112+
113+io.input():close()
114+io.close()
115+
116+assert(os.rename(file, otherfile))
117+assert(not os.rename(file, otherfile))
118+
119+io.output(io.open(otherfile, "ab"))
120+assert(io.write("\n\n\t\t ", 3450, "\n"));
121+io.close()
122+
123+
124+do
125+ -- closing file by scope
126+ local F = nil
127+ do
128+ local f <close> = assert(io.open(file, "w"))
129+ F = f
130+ end
131+ assert(tostring(F) == "file (closed)")
132+end
133+assert(os.remove(file))
134+
135+
136+do
137+ -- test writing/reading numbers
138+ local f <close> = assert(io.open(file, "w"))
139+ f:write(maxint, '\n')
140+ f:write(string.format("0X%x\n", maxint))
141+ f:write("0xABCp-3", '\n')
142+ f:write(0, '\n')
143+ f:write(-maxint, '\n')
144+ f:write(string.format("0x%X\n", -maxint))
145+ f:write("-0xABCp-3", '\n')
146+ assert(f:close())
147+ local f <close> = assert(io.open(file, "r"))
148+ assert(f:read("n") == maxint)
149+ assert(f:read("n") == maxint)
150+ assert(f:read("n") == 0xABCp-3)
151+ assert(f:read("n") == 0)
152+ assert(f:read("*n") == -maxint) -- test old format (with '*')
153+ assert(f:read("n") == -maxint)
154+ assert(f:read("*n") == -0xABCp-3) -- test old format (with '*')
155+end
156+assert(os.remove(file))
157+
158+
159+-- testing multiple arguments to io.read
160+do
161+ local f <close> = assert(io.open(file, "w"))
162+ f:write[[
163+a line
164+another line
165+1234
166+3.45
167+one
168+two
169+three
170+]]
171+ local l1, l2, l3, l4, n1, n2, c, dummy
172+ assert(f:close())
173+ local f <close> = assert(io.open(file, "r"))
174+ l1, l2, n1, n2, dummy = f:read("l", "L", "n", "n")
175+ assert(l1 == "a line" and l2 == "another line\n" and
176+ n1 == 1234 and n2 == 3.45 and dummy == nil)
177+ assert(f:close())
178+ local f <close> = assert(io.open(file, "r"))
179+ l1, l2, n1, n2, c, l3, l4, dummy = f:read(7, "l", "n", "n", 1, "l", "l")
180+ assert(l1 == "a line\n" and l2 == "another line" and c == '\n' and
181+ n1 == 1234 and n2 == 3.45 and l3 == "one" and l4 == "two"
182+ and dummy == nil)
183+ assert(f:close())
184+ local f <close> = assert(io.open(file, "r"))
185+ -- second item failing
186+ l1, n1, n2, dummy = f:read("l", "n", "n", "l")
187+ assert(l1 == "a line" and not n1)
188+end
189+assert(os.remove(file))
190+
191+
192+
193+-- test yielding during 'dofile'
194+f = assert(io.open(file, "w"))
195+f:write[[
196+local x, z = coroutine.yield(10)
197+local y = coroutine.yield(20)
198+return x + y * z
199+]]
200+assert(f:close())
201+f = coroutine.wrap(dofile)
202+assert(f(file) == 10)
203+assert(f(100, 101) == 20)
204+assert(f(200) == 100 + 200 * 101)
205+assert(os.remove(file))
206+
207+
208+f = assert(io.open(file, "w"))
209+-- test number termination
210+f:write[[
211+-12.3- -0xffff+ .3|5.E-3X +234e+13E 0xDEADBEEFDEADBEEFx
212+0x1.13Ap+3e
213+]]
214+-- very long number
215+f:write("1234"); for i = 1, 1000 do f:write("0") end; f:write("\n")
216+-- invalid sequences (must read and discard valid prefixes)
217+f:write[[
218+.e+ 0.e; --; 0xX;
219+]]
220+assert(f:close())
221+f = assert(io.open(file, "r"))
222+assert(f:read("n") == -12.3); assert(f:read(1) == "-")
223+assert(f:read("n") == -0xffff); assert(f:read(2) == "+ ")
224+assert(f:read("n") == 0.3); assert(f:read(1) == "|")
225+assert(f:read("n") == 5e-3); assert(f:read(1) == "X")
226+assert(f:read("n") == 234e13); assert(f:read(1) == "E")
227+assert(f:read("n") == 0Xdeadbeefdeadbeef); assert(f:read(2) == "x\n")
228+assert(f:read("n") == 0x1.13aP3); assert(f:read(1) == "e")
229+
230+do -- attempt to read too long number
231+ assert(not f:read("n")) -- fails
232+ local s = f:read("L") -- read rest of line
233+ assert(string.find(s, "^00*\n$")) -- lots of 0's left
234+end
235+
236+assert(not f:read("n")); assert(f:read(2) == "e+")
237+assert(not f:read("n")); assert(f:read(1) == ";")
238+assert(not f:read("n")); assert(f:read(2) == "-;")
239+assert(not f:read("n")); assert(f:read(1) == "X")
240+assert(not f:read("n")); assert(f:read(1) == ";")
241+assert(not f:read("n")); assert(not f:read(0)) -- end of file
242+assert(f:close())
243+assert(os.remove(file))
244+
245+
246+-- test line generators
247+assert(not pcall(io.lines, "non-existent-file"))
248+assert(os.rename(otherfile, file))
249+io.output(otherfile)
250+local n = 0
251+local f = io.lines(file)
252+while f() do n = n + 1 end;
253+assert(n == 6) -- number of lines in the file
254+checkerr("file is already closed", f)
255+checkerr("file is already closed", f)
256+-- copy from file to otherfile
257+n = 0
258+for l in io.lines(file) do io.write(l, "\n"); n = n + 1 end
259+io.close()
260+assert(n == 6)
261+-- copy from otherfile back to file
262+local f = assert(io.open(otherfile))
263+assert(io.type(f) == "file")
264+io.output(file)
265+assert(not io.output():read())
266+n = 0
267+for l in f:lines() do io.write(l, "\n"); n = n + 1 end
268+assert(tostring(f):sub(1, 5) == "file ")
269+assert(f:close()); io.close()
270+assert(n == 6)
271+checkerr("closed file", io.close, f)
272+assert(tostring(f) == "file (closed)")
273+assert(io.type(f) == "closed file")
274+io.input(file)
275+f = io.open(otherfile):lines()
276+n = 0
277+for l in io.lines() do assert(l == f()); n = n + 1 end
278+f = nil; collectgarbage()
279+assert(n == 6)
280+assert(os.remove(otherfile))
281+
282+do -- bug in 5.3.1
283+ io.output(otherfile)
284+ io.write(string.rep("a", 300), "\n")
285+ io.close()
286+ local t ={}; for i = 1, 250 do t[i] = 1 end
287+ t = {io.lines(otherfile, table.unpack(t))()}
288+ -- everything ok here
289+ assert(#t == 250 and t[1] == 'a' and t[#t] == 'a')
290+ t[#t + 1] = 1 -- one too many
291+ checkerr("too many arguments", io.lines, otherfile, table.unpack(t))
292+ collectgarbage() -- ensure 'otherfile' is closed
293+ assert(os.remove(otherfile))
294+end
295+
296+io.input(file)
297+do -- test error returns
298+ local a,b,c = io.input():write("xuxu")
299+ assert(not a and type(b) == "string" and type(c) == "number")
300+end
301+checkerr("invalid format", io.read, "x")
302+assert(io.read(0) == "") -- not eof
303+assert(io.read(5, 'l') == '"alo"')
304+assert(io.read(0) == "")
305+assert(io.read() == "second line")
306+local x = io.input():seek()
307+assert(io.read() == "third line ")
308+assert(io.input():seek("set", x))
309+assert(io.read('L') == "third line \n")
310+assert(io.read(1) == "X")
311+assert(io.read(string.len"fourth_line") == "fourth_line")
312+assert(io.input():seek("cur", -string.len"fourth_line"))
313+assert(io.read() == "fourth_line")
314+assert(io.read() == "") -- empty line
315+assert(io.read('n') == 3450)
316+assert(io.read(1) == '\n')
317+assert(not io.read(0)) -- end of file
318+assert(not io.read(1)) -- end of file
319+assert(not io.read(30000)) -- end of file
320+assert(({io.read(1)})[2] == undef)
321+assert(not io.read()) -- end of file
322+assert(({io.read()})[2] == undef)
323+assert(not io.read('n')) -- end of file
324+assert(({io.read('n')})[2] == undef)
325+assert(io.read('a') == '') -- end of file (OK for 'a')
326+assert(io.read('a') == '') -- end of file (OK for 'a')
327+collectgarbage()
328+print('+')
329+io.close(io.input())
330+checkerr(" input file is closed", io.read)
331+
332+assert(os.remove(file))
333+
334+local t = '0123456789'
335+for i=1,10 do t = t..t; end
336+assert(string.len(t) == 10*2^10)
337+
338+io.output(file)
339+io.write("alo"):write("\n")
340+io.close()
341+checkerr(" output file is closed", io.write)
342+local f = io.open(file, "a+b")
343+io.output(f)
344+collectgarbage()
345+
346+assert(io.write(' ' .. t .. ' '))
347+assert(io.write(';', 'end of file\n'))
348+f:flush(); io.flush()
349+f:close()
350+print('+')
351+
352+io.input(file)
353+assert(io.read() == "alo")
354+assert(io.read(1) == ' ')
355+assert(io.read(string.len(t)) == t)
356+assert(io.read(1) == ' ')
357+assert(io.read(0))
358+assert(io.read('a') == ';end of file\n')
359+assert(not io.read(0))
360+assert(io.close(io.input()))
361+
362+
363+-- test errors in read/write
364+do
365+ local function ismsg (m)
366+ -- error message is not a code number
367+ return (type(m) == "string" and not tonumber(m))
368+ end
369+
370+ -- read
371+ local f = io.open(file, "w")
372+ local r, m, c = f:read()
373+ assert(not r and ismsg(m) and type(c) == "number")
374+ assert(f:close())
375+ -- write
376+ f = io.open(file, "r")
377+ r, m, c = f:write("whatever")
378+ assert(not r and ismsg(m) and type(c) == "number")
379+ assert(f:close())
380+ -- lines
381+ f = io.open(file, "w")
382+ r, m = pcall(f:lines())
383+ assert(r == false and ismsg(m))
384+ assert(f:close())
385+end
386+
387+assert(os.remove(file))
388+
389+-- test for L format
390+io.output(file); io.write"\n\nline\nother":close()
391+io.input(file)
392+assert(io.read"L" == "\n")
393+assert(io.read"L" == "\n")
394+assert(io.read"L" == "line\n")
395+assert(io.read"L" == "other")
396+assert(not io.read"L")
397+io.input():close()
398+
399+local f = assert(io.open(file))
400+local s = ""
401+for l in f:lines("L") do s = s .. l end
402+assert(s == "\n\nline\nother")
403+f:close()
404+
405+io.input(file)
406+s = ""
407+for l in io.lines(nil, "L") do s = s .. l end
408+assert(s == "\n\nline\nother")
409+io.input():close()
410+
411+s = ""
412+for l in io.lines(file, "L") do s = s .. l end
413+assert(s == "\n\nline\nother")
414+
415+s = ""
416+for l in io.lines(file, "l") do s = s .. l end
417+assert(s == "lineother")
418+
419+io.output(file); io.write"a = 10 + 34\na = 2*a\na = -a\n":close()
420+local t = {}
421+assert(load(io.lines(file, "L"), nil, nil, t))()
422+assert(t.a == -((10 + 34) * 2))
423+
424+
425+do -- testing closing file in line iteration
426+
427+ -- get the to-be-closed variable from a loop
428+ local function gettoclose (lv)
429+ lv = lv + 1
430+ local stvar = 0 -- to-be-closed is 4th state variable in the loop
431+ for i = 1, 1000 do
432+ local n, v = debug.getlocal(lv, i)
433+ if n == "(for state)" then
434+ stvar = stvar + 1
435+ if stvar == 4 then return v end
436+ end
437+ end
438+ end
439+
440+ local f
441+ for l in io.lines(file) do
442+ f = gettoclose(1)
443+ assert(io.type(f) == "file")
444+ break
445+ end
446+ assert(io.type(f) == "closed file")
447+
448+ f = nil
449+ local function foo (name)
450+ for l in io.lines(name) do
451+ f = gettoclose(1)
452+ assert(io.type(f) == "file")
453+ error(f) -- exit loop with an error
454+ end
455+ end
456+ local st, msg = pcall(foo, file)
457+ assert(st == false and io.type(msg) == "closed file")
458+
459+end
460+
461+
462+-- test for multipe arguments in 'lines'
463+io.output(file); io.write"0123456789\n":close()
464+for a,b in io.lines(file, 1, 1) do
465+ if a == "\n" then assert(not b)
466+ else assert(tonumber(a) == tonumber(b) - 1)
467+ end
468+end
469+
470+for a,b,c in io.lines(file, 1, 2, "a") do
471+ assert(a == "0" and b == "12" and c == "3456789\n")
472+end
473+
474+for a,b,c in io.lines(file, "a", 0, 1) do
475+ if a == "" then break end
476+ assert(a == "0123456789\n" and not b and not c)
477+end
478+collectgarbage() -- to close file in previous iteration
479+
480+io.output(file); io.write"00\n10\n20\n30\n40\n":close()
481+for a, b in io.lines(file, "n", "n") do
482+ if a == 40 then assert(not b)
483+ else assert(a == b - 10)
484+ end
485+end
486+
487+
488+-- test load x lines
489+io.output(file);
490+io.write[[
491+local y
492+= X
493+X =
494+X *
495+2 +
496+X;
497+X =
498+X
499+- y;
500+]]:close()
501+_G.X = 1
502+assert(not load((io.lines(file))))
503+collectgarbage() -- to close file in previous iteration
504+load((io.lines(file, "L")))()
505+assert(_G.X == 2)
506+load((io.lines(file, 1)))()
507+assert(_G.X == 4)
508+load((io.lines(file, 3)))()
509+assert(_G.X == 8)
510+_G.X = nil
511+
512+print('+')
513+
514+local x1 = "string\n\n\\com \"\"''coisas [[estranhas]] ]]'"
515+io.output(file)
516+assert(io.write(string.format("X2 = %q\n-- comment without ending EOS", x1)))
517+io.close()
518+assert(loadfile(file))()
519+assert(x1 == _G.X2)
520+_G.X2 = nil
521+print('+')
522+assert(os.remove(file))
523+assert(not os.remove(file))
524+assert(not os.remove(otherfile))
525+
526+-- testing loadfile
527+local function testloadfile (s, expres)
528+ io.output(file)
529+ if s then io.write(s) end
530+ io.close()
531+ local res = assert(loadfile(file))()
532+ assert(os.remove(file))
533+ assert(res == expres)
534+end
535+
536+-- loading empty file
537+testloadfile(nil, nil)
538+
539+-- loading file with initial comment without end of line
540+testloadfile("# a non-ending comment", nil)
541+
542+
543+-- checking Unicode BOM in files
544+testloadfile("\xEF\xBB\xBF# some comment\nreturn 234", 234)
545+testloadfile("\xEF\xBB\xBFreturn 239", 239)
546+testloadfile("\xEF\xBB\xBF", nil) -- empty file with a BOM
547+
548+
549+-- checking line numbers in files with initial comments
550+testloadfile("# a comment\nreturn require'debug'.getinfo(1).currentline", 2)
551+
552+
553+-- loading binary file
554+io.output(io.open(file, "wb"))
555+assert(io.write(string.dump(function () return 10, '\0alo\255', 'hi' end)))
556+io.close()
557+a, b, c = assert(loadfile(file))()
558+assert(a == 10 and b == "\0alo\255" and c == "hi")
559+assert(os.remove(file))
560+
561+-- bug in 5.2.1
562+do
563+ io.output(io.open(file, "wb"))
564+ -- save function with no upvalues
565+ assert(io.write(string.dump(function () return 1 end)))
566+ io.close()
567+ f = assert(loadfile(file, "b", {}))
568+ assert(type(f) == "function" and f() == 1)
569+ assert(os.remove(file))
570+end
571+
572+-- loading binary file with initial comment
573+io.output(io.open(file, "wb"))
574+assert(io.write("#this is a comment for a binary file\0\n",
575+ string.dump(function () return 20, '\0\0\0' end)))
576+io.close()
577+a, b, c = assert(loadfile(file))()
578+assert(a == 20 and b == "\0\0\0" and c == nil)
579+assert(os.remove(file))
580+
581+
582+-- 'loadfile' with 'env'
583+do
584+ local f = io.open(file, 'w')
585+ f:write[[
586+ if (...) then a = 15; return b, c, d
587+ else return _ENV
588+ end
589+ ]]
590+ f:close()
591+ local t = {b = 12, c = "xuxu", d = print}
592+ local f = assert(loadfile(file, 't', t))
593+ local b, c, d = f(1)
594+ assert(t.a == 15 and b == 12 and c == t.c and d == print)
595+ assert(f() == t)
596+ f = assert(loadfile(file, 't', nil))
597+ assert(f() == nil)
598+ f = assert(loadfile(file))
599+ assert(f() == _G)
600+ assert(os.remove(file))
601+end
602+
603+
604+-- 'loadfile' x modes
605+do
606+ io.open(file, 'w'):write("return 10"):close()
607+ local s, m = loadfile(file, 'b')
608+ assert(not s and string.find(m, "a text chunk"))
609+ io.open(file, 'w'):write("\27 return 10"):close()
610+ local s, m = loadfile(file, 't')
611+ assert(not s and string.find(m, "a binary chunk"))
612+ assert(os.remove(file))
613+end
614+
615+
616+io.output(file)
617+assert(io.write("qualquer coisa\n"))
618+assert(io.write("mais qualquer coisa"))
619+io.close()
620+assert(io.output(assert(io.open(otherfile, 'wb')))
621+ :write("outra coisa\0\1\3\0\0\0\0\255\0")
622+ :close())
623+
624+local filehandle = assert(io.open(file, 'r+'))
625+local otherfilehandle = assert(io.open(otherfile, 'rb'))
626+assert(filehandle ~= otherfilehandle)
627+assert(type(filehandle) == "userdata")
628+assert(filehandle:read('l') == "qualquer coisa")
629+io.input(otherfilehandle)
630+assert(io.read(string.len"outra coisa") == "outra coisa")
631+assert(filehandle:read('l') == "mais qualquer coisa")
632+filehandle:close();
633+assert(type(filehandle) == "userdata")
634+io.input(otherfilehandle)
635+assert(io.read(4) == "\0\1\3\0")
636+assert(io.read(3) == "\0\0\0")
637+assert(io.read(0) == "") -- 255 is not eof
638+assert(io.read(1) == "\255")
639+assert(io.read('a') == "\0")
640+assert(not io.read(0))
641+assert(otherfilehandle == io.input())
642+otherfilehandle:close()
643+assert(os.remove(file))
644+assert(os.remove(otherfile))
645+collectgarbage()
646+
647+io.output(file)
648+ :write[[
649+ 123.4 -56e-2 not a number
650+second line
651+third line
652+
653+and the rest of the file
654+]]
655+ :close()
656+io.input(file)
657+local _,a,b,c,d,e,h,__ = io.read(1, 'n', 'n', 'l', 'l', 'l', 'a', 10)
658+assert(io.close(io.input()))
659+assert(_ == ' ' and not __)
660+assert(type(a) == 'number' and a==123.4 and b==-56e-2)
661+assert(d=='second line' and e=='third line')
662+assert(h==[[
663+
664+and the rest of the file
665+]])
666+assert(os.remove(file))
667+collectgarbage()
668+
669+-- testing buffers
670+do
671+ local f = assert(io.open(file, "w"))
672+ local fr = assert(io.open(file, "r"))
673+ assert(f:setvbuf("full", 2000))
674+ f:write("x")
675+ assert(fr:read("all") == "") -- full buffer; output not written yet
676+ f:close()
677+ fr:seek("set")
678+ assert(fr:read("all") == "x") -- `close' flushes it
679+ f = assert(io.open(file), "w")
680+ assert(f:setvbuf("no"))
681+ f:write("x")
682+ fr:seek("set")
683+ assert(fr:read("all") == "x") -- no buffer; output is ready
684+ f:close()
685+ f = assert(io.open(file, "a"))
686+ assert(f:setvbuf("line"))
687+ f:write("x")
688+ fr:seek("set", 1)
689+ assert(fr:read("all") == "") -- line buffer; no output without `\n'
690+ f:write("a\n"):seek("set", 1)
691+ assert(fr:read("all") == "xa\n") -- now we have a whole line
692+ f:close(); fr:close()
693+ assert(os.remove(file))
694+end
695+
696+
697+if not _soft then
698+ print("testing large files (> BUFSIZ)")
699+ io.output(file)
700+ for i=1,5001 do io.write('0123456789123') end
701+ io.write('\n12346'):close()
702+ io.input(file)
703+ local x = io.read('a')
704+ io.input():seek('set', 0)
705+ local y = io.read(30001)..io.read(1005)..io.read(0)..
706+ io.read(1)..io.read(100003)
707+ assert(x == y and string.len(x) == 5001*13 + 6)
708+ io.input():seek('set', 0)
709+ y = io.read() -- huge line
710+ assert(x == y..'\n'..io.read())
711+ assert(not io.read())
712+ io.close(io.input())
713+ assert(os.remove(file))
714+ x = nil; y = nil
715+end
716+
717+if not _port then
718+ local progname
719+ do -- get name of running executable
720+ local arg = arg or ARG
721+ local i = 0
722+ while arg[i] do i = i - 1 end
723+ progname = '"' .. arg[i + 1] .. '"'
724+ end
725+ print("testing popen/pclose and execute")
726+ -- invalid mode for popen
727+ checkerr("invalid mode", io.popen, "cat", "")
728+ checkerr("invalid mode", io.popen, "cat", "r+")
729+ checkerr("invalid mode", io.popen, "cat", "rw")
730+ do -- basic tests for popen
731+ local file = os.tmpname()
732+ local f = assert(io.popen("cat - > " .. file, "w"))
733+ f:write("a line")
734+ assert(f:close())
735+ local f = assert(io.popen("cat - < " .. file, "r"))
736+ assert(f:read("a") == "a line")
737+ assert(f:close())
738+ assert(os.remove(file))
739+ end
740+
741+ local tests = {
742+ -- command, what, code
743+ {"ls > /dev/null", "ok"},
744+ {"not-to-be-found-command", "exit"},
745+ {"exit 3", "exit", 3},
746+ {"exit 129", "exit", 129},
747+ {"kill -s HUP $$", "signal", 1},
748+ {"kill -s KILL $$", "signal", 9},
749+ {"sh -c 'kill -s HUP $$'", "exit"},
750+ {progname .. ' -e " "', "ok"},
751+ {progname .. ' -e "os.exit(0, true)"', "ok"},
752+ {progname .. ' -e "os.exit(20, true)"', "exit", 20},
753+ }
754+ print("\n(some error messages are expected now)")
755+ for _, v in ipairs(tests) do
756+ local x, y, z = io.popen(v[1]):close()
757+ local x1, y1, z1 = os.execute(v[1])
758+ assert(x == x1 and y == y1 and z == z1)
759+ if v[2] == "ok" then
760+ assert(x and y == 'exit' and z == 0)
761+ else
762+ assert(not x and y == v[2]) -- correct status and 'what'
763+ -- correct code if known (but always different from 0)
764+ assert((v[3] == nil and z > 0) or v[3] == z)
765+ end
766+ end
767+end
768+
769+
770+-- testing tmpfile
771+f = io.tmpfile()
772+assert(io.type(f) == "file")
773+f:write("alo")
774+f:seek("set")
775+assert(f:read"a" == "alo")
776+
777+end --}
778+
779+print'+'
780+
781+print("testing date/time")
782+
783+assert(os.date("") == "")
784+assert(os.date("!") == "")
785+assert(os.date("\0\0") == "\0\0")
786+assert(os.date("!\0\0") == "\0\0")
787+local x = string.rep("a", 10000)
788+assert(os.date(x) == x)
789+local t = os.time()
790+D = os.date("*t", t)
791+assert(os.date(string.rep("%d", 1000), t) ==
792+ string.rep(os.date("%d", t), 1000))
793+assert(os.date(string.rep("%", 200)) == string.rep("%", 100))
794+
795+local function checkDateTable (t)
796+ _G.D = os.date("*t", t)
797+ assert(os.time(D) == t)
798+ load(os.date([[assert(D.year==%Y and D.month==%m and D.day==%d and
799+ D.hour==%H and D.min==%M and D.sec==%S and
800+ D.wday==%w+1 and D.yday==%j)]], t))()
801+ _G.D = nil
802+end
803+
804+checkDateTable(os.time())
805+if not _port then
806+ -- assume that time_t can represent these values
807+ checkDateTable(0)
808+ checkDateTable(1)
809+ checkDateTable(1000)
810+ checkDateTable(0x7fffffff)
811+ checkDateTable(0x80000000)
812+end
813+
814+checkerr("invalid conversion specifier", os.date, "%")
815+checkerr("invalid conversion specifier", os.date, "%9")
816+checkerr("invalid conversion specifier", os.date, "%")
817+checkerr("invalid conversion specifier", os.date, "%O")
818+checkerr("invalid conversion specifier", os.date, "%E")
819+checkerr("invalid conversion specifier", os.date, "%Ea")
820+
821+checkerr("not an integer", os.time, {year=1000, month=1, day=1, hour='x'})
822+checkerr("not an integer", os.time, {year=1000, month=1, day=1, hour=1.5})
823+
824+checkerr("missing", os.time, {hour = 12}) -- missing date
825+
826+
827+if string.packsize("i") == 4 then -- 4-byte ints
828+ checkerr("field 'year' is out-of-bound", os.time,
829+ {year = -(1 << 31) + 1899, month = 1, day = 1})
830+
831+ checkerr("field 'year' is out-of-bound", os.time,
832+ {year = -(1 << 31), month = 1, day = 1})
833+
834+ if math.maxinteger > 2^31 then -- larger lua_integer?
835+ checkerr("field 'year' is out-of-bound", os.time,
836+ {year = (1 << 31) + 1900, month = 1, day = 1})
837+ end
838+end
839+
840+
841+if not _port then
842+ -- test Posix-specific modifiers
843+ assert(type(os.date("%Ex")) == 'string')
844+ assert(type(os.date("%Oy")) == 'string')
845+
846+ -- test large dates (assume at least 4-byte ints and time_t)
847+ local t0 = os.time{year = 1970, month = 1, day = 0}
848+ local t1 = os.time{year = 1970, month = 1, day = 0, sec = (1 << 31) - 1}
849+ assert(t1 - t0 == (1 << 31) - 1)
850+ t0 = os.time{year = 1970, month = 1, day = 1}
851+ t1 = os.time{year = 1970, month = 1, day = 1, sec = -(1 << 31)}
852+ assert(t1 - t0 == -(1 << 31))
853+
854+ -- test out-of-range dates (at least for Unix)
855+ if maxint >= 2^62 then -- cannot do these tests in Small Lua
856+ -- no arith overflows
857+ checkerr("out-of-bound", os.time, {year = -maxint, month = 1, day = 1})
858+ if string.packsize("i") == 4 then -- 4-byte ints
859+ if testerr("out-of-bound", os.date, "%Y", 2^40) then
860+ -- time_t has 4 bytes and therefore cannot represent year 4000
861+ print(" 4-byte time_t")
862+ checkerr("cannot be represented", os.time, {year=4000, month=1, day=1})
863+ else
864+ -- time_t has 8 bytes; an int year cannot represent a huge time
865+ print(" 8-byte time_t")
866+ checkerr("cannot be represented", os.date, "%Y", 2^60)
867+
868+ -- this is the maximum year
869+ assert(tonumber(os.time
870+ {year=(1 << 31) + 1899, month=12, day=31, hour=23, min=59, sec=59}))
871+
872+ -- this is too much
873+ checkerr("represented", os.time,
874+ {year=(1 << 31) + 1899, month=12, day=31, hour=23, min=59, sec=60})
875+ end
876+
877+ -- internal 'int' fields cannot hold these values
878+ checkerr("field 'day' is out-of-bound", os.time,
879+ {year = 0, month = 1, day = 2^32})
880+
881+ checkerr("field 'month' is out-of-bound", os.time,
882+ {year = 0, month = -((1 << 31) + 1), day = 1})
883+
884+ checkerr("field 'year' is out-of-bound", os.time,
885+ {year = (1 << 31) + 1900, month = 1, day = 1})
886+
887+ else -- 8-byte ints
888+ -- assume time_t has 8 bytes too
889+ print(" 8-byte time_t")
890+ assert(tonumber(os.date("%Y", 2^60)))
891+
892+ -- but still cannot represent a huge year
893+ checkerr("cannot be represented", os.time, {year=2^60, month=1, day=1})
894+ end
895+ end
896+end
897+
898+do
899+ local D = os.date("*t")
900+ local t = os.time(D)
901+ if D.isdst == nil then
902+ print("no daylight saving information")
903+ else
904+ assert(type(D.isdst) == 'boolean')
905+ end
906+ D.isdst = nil
907+ local t1 = os.time(D)
908+ assert(t == t1) -- if isdst is absent uses correct default
909+end
910+
911+local D = os.date("*t")
912+t = os.time(D)
913+D.year = D.year-1;
914+local t1 = os.time(D)
915+-- allow for leap years
916+assert(math.abs(os.difftime(t,t1)/(24*3600) - 365) < 2)
917+
918+-- should not take more than 1 second to execute these two lines
919+t = os.time()
920+t1 = os.time(os.date("*t"))
921+local diff = os.difftime(t1,t)
922+assert(0 <= diff and diff <= 1)
923+diff = os.difftime(t,t1)
924+assert(-1 <= diff and diff <= 0)
925+
926+local t1 = os.time{year=2000, month=10, day=1, hour=23, min=12}
927+local t2 = os.time{year=2000, month=10, day=1, hour=23, min=10, sec=19}
928+assert(os.difftime(t1,t2) == 60*2-19)
929+
930+-- since 5.3.3, 'os.time' normalizes table fields
931+t1 = {year = 2005, month = 1, day = 1, hour = 1, min = 0, sec = -3602}
932+os.time(t1)
933+assert(t1.day == 31 and t1.month == 12 and t1.year == 2004 and
934+ t1.hour == 23 and t1.min == 59 and t1.sec == 58 and
935+ t1.yday == 366)
936+
937+io.output(io.stdout)
938+local t = os.date('%d %m %Y %H %M %S')
939+local d, m, a, h, min, s = string.match(t,
940+ "(%d+) (%d+) (%d+) (%d+) (%d+) (%d+)")
941+d = tonumber(d)
942+m = tonumber(m)
943+a = tonumber(a)
944+h = tonumber(h)
945+min = tonumber(min)
946+s = tonumber(s)
947+io.write(string.format('test done on %2.2d/%2.2d/%d', d, m, a))
948+io.write(string.format(', at %2.2d:%2.2d:%2.2d\n', h, min, s))
949+io.write(string.format('%s\n', _VERSION))
950+
951+
zod/luau/tests/gc.luanew file+695
@@ -0,0 +1,695 @@
1+-- $Id: testes/gc.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print('testing incremental garbage collection')
5+
6+local debug = require"debug"
7+
8+assert(collectgarbage("isrunning"))
9+
10+collectgarbage()
11+
12+local oldmode = collectgarbage("incremental")
13+
14+-- changing modes should return previous mode
15+assert(collectgarbage("generational") == "incremental")
16+assert(collectgarbage("generational") == "generational")
17+assert(collectgarbage("incremental") == "generational")
18+assert(collectgarbage("incremental") == "incremental")
19+
20+
21+local function nop () end
22+
23+local function gcinfo ()
24+ return collectgarbage"count" * 1024
25+end
26+
27+
28+-- test weird parameters to 'collectgarbage'
29+do
30+ -- save original parameters
31+ local a = collectgarbage("setpause", 200)
32+ local b = collectgarbage("setstepmul", 200)
33+ local t = {0, 2, 10, 90, 500, 5000, 30000, 0x7ffffffe}
34+ for i = 1, #t do
35+ local p = t[i]
36+ for j = 1, #t do
37+ local m = t[j]
38+ collectgarbage("setpause", p)
39+ collectgarbage("setstepmul", m)
40+ collectgarbage("step", 0)
41+ collectgarbage("step", 10000)
42+ end
43+ end
44+ -- restore original parameters
45+ collectgarbage("setpause", a)
46+ collectgarbage("setstepmul", b)
47+ collectgarbage()
48+end
49+
50+
51+_G["while"] = 234
52+
53+
54+--
55+-- tests for GC activation when creating different kinds of objects
56+--
57+local function GC1 ()
58+ local u
59+ local b -- (above 'u' it in the stack)
60+ local finish = false
61+ u = setmetatable({}, {__gc = function () finish = true end})
62+ b = {34}
63+ repeat u = {} until finish
64+ assert(b[1] == 34) -- 'u' was collected, but 'b' was not
65+
66+ finish = false; local i = 1
67+ u = setmetatable({}, {__gc = function () finish = true end})
68+ repeat i = i + 1; u = tostring(i) .. tostring(i) until finish
69+ assert(b[1] == 34) -- 'u' was collected, but 'b' was not
70+
71+ finish = false
72+ u = setmetatable({}, {__gc = function () finish = true end})
73+ repeat local i; u = function () return i end until finish
74+ assert(b[1] == 34) -- 'u' was collected, but 'b' was not
75+end
76+
77+local function GC2 ()
78+ local u
79+ local finish = false
80+ u = {setmetatable({}, {__gc = function () finish = true end})}
81+ local b = {34}
82+ repeat u = {{}} until finish
83+ assert(b[1] == 34) -- 'u' was collected, but 'b' was not
84+
85+ finish = false; local i = 1
86+ u = {setmetatable({}, {__gc = function () finish = true end})}
87+ repeat i = i + 1; u = {tostring(i) .. tostring(i)} until finish
88+ assert(b[1] == 34) -- 'u' was collected, but 'b' was not
89+
90+ finish = false
91+ u = {setmetatable({}, {__gc = function () finish = true end})}
92+ repeat local i; u = {function () return i end} until finish
93+ assert(b[1] == 34) -- 'u' was collected, but 'b' was not
94+end
95+
96+local function GC() GC1(); GC2() end
97+
98+
99+do
100+ print("creating many objects")
101+
102+ local limit = 5000
103+
104+ for i = 1, limit do
105+ local a = {}; a = nil
106+ end
107+
108+ local a = "a"
109+
110+ for i = 1, limit do
111+ a = i .. "b";
112+ a = string.gsub(a, '(%d%d*)', "%1 %1")
113+ a = "a"
114+ end
115+
116+
117+
118+ a = {}
119+
120+ function a:test ()
121+ for i = 1, limit do
122+ load(string.format("function temp(a) return 'a%d' end", i), "")()
123+ assert(temp() == string.format('a%d', i))
124+ end
125+ end
126+
127+ a:test()
128+ _G.temp = nil
129+end
130+
131+
132+-- collection of functions without locals, globals, etc.
133+do local f = function () end end
134+
135+
136+print("functions with errors")
137+local prog = [[
138+do
139+ a = 10;
140+ function foo(x,y)
141+ a = sin(a+0.456-0.23e-12);
142+ return function (z) return sin(%x+z) end
143+ end
144+ local x = function (w) a=a+w; end
145+end
146+]]
147+do
148+ local step = 1
149+ if _soft then step = 13 end
150+ for i=1, string.len(prog), step do
151+ for j=i, string.len(prog), step do
152+ pcall(load(string.sub(prog, i, j), ""))
153+ end
154+ end
155+end
156+rawset(_G, "a", nil)
157+_G.x = nil
158+
159+do
160+ foo = nil
161+ print('long strings')
162+ local x = "01234567890123456789012345678901234567890123456789012345678901234567890123456789"
163+ assert(string.len(x)==80)
164+ local s = ''
165+ local k = math.min(300, (math.maxinteger // 80) // 2)
166+ for n = 1, k do s = s..x; local j=tostring(n) end
167+ assert(string.len(s) == k*80)
168+ s = string.sub(s, 1, 10000)
169+ local s, i = string.gsub(s, '(%d%d%d%d)', '')
170+ assert(i==10000 // 4)
171+
172+ assert(_G["while"] == 234)
173+ _G["while"] = nil
174+end
175+
176+
177+--
178+-- test the "size" of basic GC steps (whatever they mean...)
179+--
180+do
181+print("steps")
182+
183+ print("steps (2)")
184+
185+ local function dosteps (siz)
186+ collectgarbage()
187+ local a = {}
188+ for i=1,100 do a[i] = {{}}; local b = {} end
189+ local x = gcinfo()
190+ local i = 0
191+ repeat -- do steps until it completes a collection cycle
192+ i = i+1
193+ until collectgarbage("step", siz)
194+ assert(gcinfo() < x)
195+ return i -- number of steps
196+ end
197+
198+ collectgarbage"stop"
199+
200+ if not _port then
201+ assert(dosteps(10) < dosteps(2))
202+ end
203+
204+ -- collector should do a full collection with so many steps
205+ assert(dosteps(20000) == 1)
206+ assert(collectgarbage("step", 20000) == true)
207+ assert(collectgarbage("step", 20000) == true)
208+
209+ assert(not collectgarbage("isrunning"))
210+ collectgarbage"restart"
211+ assert(collectgarbage("isrunning"))
212+
213+end
214+
215+
216+if not _port then
217+ -- test the pace of the collector
218+ collectgarbage(); collectgarbage()
219+ local x = gcinfo()
220+ collectgarbage"stop"
221+ repeat
222+ local a = {}
223+ until gcinfo() > 3 * x
224+ collectgarbage"restart"
225+ assert(collectgarbage("isrunning"))
226+ repeat
227+ local a = {}
228+ until gcinfo() <= x * 2
229+end
230+
231+
232+print("clearing tables")
233+local lim = 15
234+local a = {}
235+-- fill a with `collectable' indices
236+for i=1,lim do a[{}] = i end
237+b = {}
238+for k,v in pairs(a) do b[k]=v end
239+-- remove all indices and collect them
240+for n in pairs(b) do
241+ a[n] = undef
242+ assert(type(n) == 'table' and next(n) == nil)
243+ collectgarbage()
244+end
245+b = nil
246+collectgarbage()
247+for n in pairs(a) do error'cannot be here' end
248+for i=1,lim do a[i] = i end
249+for i=1,lim do assert(a[i] == i) end
250+
251+
252+print('weak tables')
253+a = {}; setmetatable(a, {__mode = 'k'});
254+-- fill a with some `collectable' indices
255+for i=1,lim do a[{}] = i end
256+-- and some non-collectable ones
257+for i=1,lim do a[i] = i end
258+for i=1,lim do local s=string.rep('@', i); a[s] = s..'#' end
259+collectgarbage()
260+local i = 0
261+for k,v in pairs(a) do assert(k==v or k..'#'==v); i=i+1 end
262+assert(i == 2*lim)
263+
264+a = {}; setmetatable(a, {__mode = 'v'});
265+a[1] = string.rep('b', 21)
266+collectgarbage()
267+assert(a[1]) -- strings are *values*
268+a[1] = undef
269+-- fill a with some `collectable' values (in both parts of the table)
270+for i=1,lim do a[i] = {} end
271+for i=1,lim do a[i..'x'] = {} end
272+-- and some non-collectable ones
273+for i=1,lim do local t={}; a[t]=t end
274+for i=1,lim do a[i+lim]=i..'x' end
275+collectgarbage()
276+local i = 0
277+for k,v in pairs(a) do assert(k==v or k-lim..'x' == v); i=i+1 end
278+assert(i == 2*lim)
279+
280+a = {}; setmetatable(a, {__mode = 'kv'});
281+local x, y, z = {}, {}, {}
282+-- keep only some items
283+a[1], a[2], a[3] = x, y, z
284+a[string.rep('$', 11)] = string.rep('$', 11)
285+-- fill a with some `collectable' values
286+for i=4,lim do a[i] = {} end
287+for i=1,lim do a[{}] = i end
288+for i=1,lim do local t={}; a[t]=t end
289+collectgarbage()
290+assert(next(a) ~= nil)
291+local i = 0
292+for k,v in pairs(a) do
293+ assert((k == 1 and v == x) or
294+ (k == 2 and v == y) or
295+ (k == 3 and v == z) or k==v);
296+ i = i+1
297+end
298+assert(i == 4)
299+x,y,z=nil
300+collectgarbage()
301+assert(next(a) == string.rep('$', 11))
302+
303+
304+-- 'bug' in 5.1
305+a = {}
306+local t = {x = 10}
307+local C = setmetatable({key = t}, {__mode = 'v'})
308+local C1 = setmetatable({[t] = 1}, {__mode = 'k'})
309+a.x = t -- this should not prevent 't' from being removed from
310+ -- weak table 'C' by the time 'a' is finalized
311+
312+setmetatable(a, {__gc = function (u)
313+ assert(C.key == nil)
314+ assert(type(next(C1)) == 'table')
315+ end})
316+
317+a, t = nil
318+collectgarbage()
319+collectgarbage()
320+assert(next(C) == nil and next(C1) == nil)
321+C, C1 = nil
322+
323+
324+-- ephemerons
325+local mt = {__mode = 'k'}
326+a = {{10},{20},{30},{40}}; setmetatable(a, mt)
327+x = nil
328+for i = 1, 100 do local n = {}; a[n] = {k = {x}}; x = n end
329+GC()
330+local n = x
331+local i = 0
332+while n do n = a[n].k[1]; i = i + 1 end
333+assert(i == 100)
334+x = nil
335+GC()
336+for i = 1, 4 do assert(a[i][1] == i * 10); a[i] = undef end
337+assert(next(a) == nil)
338+
339+local K = {}
340+a[K] = {}
341+for i=1,10 do a[K][i] = {}; a[a[K][i]] = setmetatable({}, mt) end
342+x = nil
343+local k = 1
344+for j = 1,100 do
345+ local n = {}; local nk = k%10 + 1
346+ a[a[K][nk]][n] = {x, k = k}; x = n; k = nk
347+end
348+GC()
349+local n = x
350+local i = 0
351+while n do local t = a[a[K][k]][n]; n = t[1]; k = t.k; i = i + 1 end
352+assert(i == 100)
353+K = nil
354+GC()
355+-- assert(next(a) == nil)
356+
357+
358+-- testing errors during GC
359+if T then
360+ collectgarbage("stop") -- stop collection
361+ local u = {}
362+ local s = {}; setmetatable(s, {__mode = 'k'})
363+ setmetatable(u, {__gc = function (o)
364+ local i = s[o]
365+ s[i] = true
366+ assert(not s[i - 1]) -- check proper finalization order
367+ if i == 8 then error("@expected@") end -- error during GC
368+ end})
369+
370+ for i = 6, 10 do
371+ local n = setmetatable({}, getmetatable(u))
372+ s[n] = i
373+ end
374+
375+ warn("@on"); warn("@store")
376+ collectgarbage()
377+ assert(string.find(_WARN, "error in __gc"))
378+ assert(string.match(_WARN, "@(.-)@") == "expected"); _WARN = false
379+ for i = 8, 10 do assert(s[i]) end
380+
381+ for i = 1, 5 do
382+ local n = setmetatable({}, getmetatable(u))
383+ s[n] = i
384+ end
385+
386+ collectgarbage()
387+ for i = 1, 10 do assert(s[i]) end
388+
389+ getmetatable(u).__gc = nil
390+ warn("@normal")
391+
392+end
393+print '+'
394+
395+
396+-- testing userdata
397+if T==nil then
398+ (Message or print)('\n >>> testC not active: skipping userdata GC tests <<<\n')
399+
400+else
401+
402+ local function newproxy(u)
403+ return debug.setmetatable(T.newuserdata(0), debug.getmetatable(u))
404+ end
405+
406+ collectgarbage("stop") -- stop collection
407+ local u = newproxy(nil)
408+ debug.setmetatable(u, {__gc = true})
409+ local s = 0
410+ local a = {[u] = 0}; setmetatable(a, {__mode = 'vk'})
411+ for i=1,10 do a[newproxy(u)] = i end
412+ for k in pairs(a) do assert(getmetatable(k) == getmetatable(u)) end
413+ local a1 = {}; for k,v in pairs(a) do a1[k] = v end
414+ for k,v in pairs(a1) do a[v] = k end
415+ for i =1,10 do assert(a[i]) end
416+ getmetatable(u).a = a1
417+ getmetatable(u).u = u
418+ do
419+ local u = u
420+ getmetatable(u).__gc = function (o)
421+ assert(a[o] == 10-s)
422+ assert(a[10-s] == undef) -- udata already removed from weak table
423+ assert(getmetatable(o) == getmetatable(u))
424+ assert(getmetatable(o).a[o] == 10-s)
425+ s=s+1
426+ end
427+ end
428+ a1, u = nil
429+ assert(next(a) ~= nil)
430+ collectgarbage()
431+ assert(s==11)
432+ collectgarbage()
433+ assert(next(a) == nil) -- finalized keys are removed in two cycles
434+end
435+
436+
437+-- __gc x weak tables
438+local u = setmetatable({}, {__gc = true})
439+-- __gc metamethod should be collected before running
440+setmetatable(getmetatable(u), {__mode = "v"})
441+getmetatable(u).__gc = function (o) os.exit(1) end -- cannot happen
442+u = nil
443+collectgarbage()
444+
445+local u = setmetatable({}, {__gc = true})
446+local m = getmetatable(u)
447+m.x = {[{0}] = 1; [0] = {1}}; setmetatable(m.x, {__mode = "kv"});
448+m.__gc = function (o)
449+ assert(next(getmetatable(o).x) == nil)
450+ m = 10
451+end
452+u, m = nil
453+collectgarbage()
454+assert(m==10)
455+
456+do -- tests for string keys in weak tables
457+ collectgarbage(); collectgarbage()
458+ local m = collectgarbage("count") -- current memory
459+ local a = setmetatable({}, {__mode = "kv"})
460+ a[string.rep("a", 2^22)] = 25 -- long string key -> number value
461+ a[string.rep("b", 2^22)] = {} -- long string key -> colectable value
462+ a[{}] = 14 -- colectable key
463+ assert(collectgarbage("count") > m + 2^13) -- 2^13 == 2 * 2^22 in KB
464+ collectgarbage()
465+ assert(collectgarbage("count") >= m + 2^12 and
466+ collectgarbage("count") < m + 2^13) -- one key was collected
467+ local k, v = next(a) -- string key with number value preserved
468+ assert(k == string.rep("a", 2^22) and v == 25)
469+ assert(next(a, k) == nil) -- everything else cleared
470+ assert(a[string.rep("b", 2^22)] == undef)
471+ a[k] = undef -- erase this last entry
472+ k = nil
473+ collectgarbage()
474+ assert(next(a) == nil)
475+ -- make sure will not try to compare with dead key
476+ assert(a[string.rep("b", 100)] == undef)
477+ assert(collectgarbage("count") <= m + 1) -- eveything collected
478+end
479+
480+
481+-- errors during collection
482+if T then
483+ warn("@store")
484+ u = setmetatable({}, {__gc = function () error "@expected error" end})
485+ u = nil
486+ collectgarbage()
487+ assert(string.find(_WARN, "@expected error")); _WARN = false
488+ warn("@normal")
489+end
490+
491+
492+if not _soft then
493+ print("long list")
494+ local a = {}
495+ for i = 1,200000 do
496+ a = {next = a}
497+ end
498+ a = nil
499+ collectgarbage()
500+end
501+
502+-- create many threads with self-references and open upvalues
503+print("self-referenced threads")
504+local thread_id = 0
505+local threads = {}
506+
507+local function fn (thread)
508+ local x = {}
509+ threads[thread_id] = function()
510+ thread = x
511+ end
512+ coroutine.yield()
513+end
514+
515+while thread_id < 1000 do
516+ local thread = coroutine.create(fn)
517+ coroutine.resume(thread, thread)
518+ thread_id = thread_id + 1
519+end
520+
521+
522+-- Create a closure (function inside 'f') with an upvalue ('param') that
523+-- points (through a table) to the closure itself and to the thread
524+-- ('co' and the initial value of 'param') where closure is running.
525+-- Then, assert that table (and therefore everything else) will be
526+-- collected.
527+do
528+ local collected = false -- to detect collection
529+ collectgarbage(); collectgarbage("stop")
530+ do
531+ local function f (param)
532+ ;(function ()
533+ assert(type(f) == 'function' and type(param) == 'thread')
534+ param = {param, f}
535+ setmetatable(param, {__gc = function () collected = true end})
536+ coroutine.yield(100)
537+ end)()
538+ end
539+ local co = coroutine.create(f)
540+ assert(coroutine.resume(co, co))
541+ end
542+ -- Now, thread and closure are not reacheable any more.
543+ collectgarbage()
544+ assert(collected)
545+ collectgarbage("restart")
546+end
547+
548+
549+do
550+ collectgarbage()
551+ collectgarbage"stop"
552+ collectgarbage("step", 0) -- steps should not unblock the collector
553+ local x = gcinfo()
554+ repeat
555+ for i=1,1000 do _ENV.a = {} end -- no collection during the loop
556+ until gcinfo() > 2 * x
557+ collectgarbage"restart"
558+ _ENV.a = nil
559+end
560+
561+
562+if T then -- tests for weird cases collecting upvalues
563+
564+ local function foo ()
565+ local a = {x = 20}
566+ coroutine.yield(function () return a.x end) -- will run collector
567+ assert(a.x == 20) -- 'a' is 'ok'
568+ a = {x = 30} -- create a new object
569+ assert(T.gccolor(a) == "white") -- of course it is new...
570+ coroutine.yield(100) -- 'a' is still local to this thread
571+ end
572+
573+ local t = setmetatable({}, {__mode = "kv"})
574+ collectgarbage(); collectgarbage('stop')
575+ -- create coroutine in a weak table, so it will never be marked
576+ t.co = coroutine.wrap(foo)
577+ local f = t.co() -- create function to access local 'a'
578+ T.gcstate("atomic") -- ensure all objects are traversed
579+ assert(T.gcstate() == "atomic")
580+ assert(t.co() == 100) -- resume coroutine, creating new table for 'a'
581+ assert(T.gccolor(t.co) == "white") -- thread was not traversed
582+ T.gcstate("pause") -- collect thread, but should mark 'a' before that
583+ assert(t.co == nil and f() == 30) -- ensure correct access to 'a'
584+
585+ collectgarbage("restart")
586+
587+ -- test barrier in sweep phase (backing userdata to gray)
588+ local u = T.newuserdata(0, 1) -- create a userdata
589+ collectgarbage()
590+ collectgarbage"stop"
591+ local a = {} -- avoid 'u' as first element in 'allgc'
592+ T.gcstate"atomic"
593+ T.gcstate"sweepallgc"
594+ local x = {}
595+ assert(T.gccolor(u) == "black") -- userdata is "old" (black)
596+ assert(T.gccolor(x) == "white") -- table is "new" (white)
597+ debug.setuservalue(u, x) -- trigger barrier
598+ assert(T.gccolor(u) == "gray") -- userdata changed back to gray
599+ collectgarbage"restart"
600+
601+ print"+"
602+end
603+
604+
605+if T then
606+ local debug = require "debug"
607+ collectgarbage("stop")
608+ local x = T.newuserdata(0)
609+ local y = T.newuserdata(0)
610+ debug.setmetatable(y, {__gc = nop}) -- bless the new udata before...
611+ debug.setmetatable(x, {__gc = nop}) -- ...the old one
612+ assert(T.gccolor(y) == "white")
613+ T.checkmemory()
614+ collectgarbage("restart")
615+end
616+
617+
618+if T then
619+ print("emergency collections")
620+ collectgarbage()
621+ collectgarbage()
622+ T.totalmem(T.totalmem() + 200)
623+ for i=1,200 do local a = {} end
624+ T.totalmem(0)
625+ collectgarbage()
626+ local t = T.totalmem("table")
627+ local a = {{}, {}, {}} -- create 4 new tables
628+ assert(T.totalmem("table") == t + 4)
629+ t = T.totalmem("function")
630+ a = function () end -- create 1 new closure
631+ assert(T.totalmem("function") == t + 1)
632+ t = T.totalmem("thread")
633+ a = coroutine.create(function () end) -- create 1 new coroutine
634+ assert(T.totalmem("thread") == t + 1)
635+end
636+
637+
638+-- create an object to be collected when state is closed
639+do
640+ local setmetatable,assert,type,print,getmetatable =
641+ setmetatable,assert,type,print,getmetatable
642+ local tt = {}
643+ tt.__gc = function (o)
644+ assert(getmetatable(o) == tt)
645+ -- create new objects during GC
646+ local a = 'xuxu'..(10+3)..'joao', {}
647+ ___Glob = o -- ressurrect object!
648+ setmetatable({}, tt) -- creates a new one with same metatable
649+ print(">>> closing state " .. "<<<\n")
650+ end
651+ local u = setmetatable({}, tt)
652+ ___Glob = {u} -- avoid object being collected before program end
653+end
654+
655+-- create several objects to raise errors when collected while closing state
656+if T then
657+ local error, assert, find, warn = error, assert, string.find, warn
658+ local n = 0
659+ local lastmsg
660+ local mt = {__gc = function (o)
661+ n = n + 1
662+ assert(n == o[1])
663+ if n == 1 then
664+ _WARN = false
665+ elseif n == 2 then
666+ assert(find(_WARN, "@expected warning"))
667+ lastmsg = _WARN -- get message from previous error (first 'o')
668+ else
669+ assert(lastmsg == _WARN) -- subsequent error messages are equal
670+ end
671+ warn("@store"); _WARN = false
672+ error"@expected warning"
673+ end}
674+ for i = 10, 1, -1 do
675+ -- create object and preserve it until the end
676+ table.insert(___Glob, setmetatable({i}, mt))
677+ end
678+end
679+
680+-- just to make sure
681+assert(collectgarbage'isrunning')
682+
683+do -- check that the collector is not reentrant in incremental mode
684+ local res = true
685+ setmetatable({}, {__gc = function ()
686+ res = collectgarbage()
687+ end})
688+ collectgarbage()
689+ assert(not res)
690+end
691+
692+
693+collectgarbage(oldmode)
694+
695+print('OK')
zod/luau/tests/gengc.luanew file+172
@@ -0,0 +1,172 @@
1+-- $Id: testes/gengc.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print('testing generational garbage collection')
5+
6+local debug = require"debug"
7+
8+assert(collectgarbage("isrunning"))
9+
10+collectgarbage()
11+
12+local oldmode = collectgarbage("generational")
13+
14+
15+-- ensure that table barrier evolves correctly
16+do
17+ local U = {}
18+ -- full collection makes 'U' old
19+ collectgarbage()
20+ assert(not T or T.gcage(U) == "old")
21+
22+ -- U refers to a new table, so it becomes 'touched1'
23+ U[1] = {x = {234}}
24+ assert(not T or (T.gcage(U) == "touched1" and T.gcage(U[1]) == "new"))
25+
26+ -- both U and the table survive one more collection
27+ collectgarbage("step", 0)
28+ assert(not T or (T.gcage(U) == "touched2" and T.gcage(U[1]) == "survival"))
29+
30+ -- both U and the table survive yet another collection
31+ -- now everything is old
32+ collectgarbage("step", 0)
33+ assert(not T or (T.gcage(U) == "old" and T.gcage(U[1]) == "old1"))
34+
35+ -- data was not corrupted
36+ assert(U[1].x[1] == 234)
37+end
38+
39+
40+do
41+ -- ensure that 'firstold1' is corrected when object is removed from
42+ -- the 'allgc' list
43+ local function foo () end
44+ local old = {10}
45+ collectgarbage() -- make 'old' old
46+ assert(not T or T.gcage(old) == "old")
47+ setmetatable(old, {}) -- new table becomes OLD0 (barrier)
48+ assert(not T or T.gcage(getmetatable(old)) == "old0")
49+ collectgarbage("step", 0) -- new table becomes OLD1 and firstold1
50+ assert(not T or T.gcage(getmetatable(old)) == "old1")
51+ setmetatable(getmetatable(old), {__gc = foo}) -- get it out of allgc list
52+ collectgarbage("step", 0) -- should not seg. fault
53+end
54+
55+
56+do -- bug in 5.4.0
57+-- When an object aged OLD1 is finalized, it is moved from the list
58+-- 'finobj' to the *beginning* of the list 'allgc', but that part of the
59+-- list was not being visited by 'markold'.
60+ local A = {}
61+ A[1] = false -- old anchor for object
62+
63+ -- obj finalizer
64+ local function gcf (obj)
65+ A[1] = obj -- anchor object
66+ assert(not T or T.gcage(obj) == "old1")
67+ obj = nil -- remove it from the stack
68+ collectgarbage("step", 0) -- do a young collection
69+ print(getmetatable(A[1]).x) -- metatable was collected
70+ end
71+
72+ collectgarbage() -- make A old
73+ local obj = {} -- create a new object
74+ collectgarbage("step", 0) -- make it a survival
75+ assert(not T or T.gcage(obj) == "survival")
76+ setmetatable(obj, {__gc = gcf, x = "+"}) -- create its metatable
77+ assert(not T or T.gcage(getmetatable(obj)) == "new")
78+ obj = nil -- clear object
79+ collectgarbage("step", 0) -- will call obj's finalizer
80+end
81+
82+
83+do -- another bug in 5.4.0
84+ local old = {10}
85+ collectgarbage() -- make 'old' old
86+ local co = coroutine.create(
87+ function ()
88+ local x = nil
89+ local f = function ()
90+ return x[1]
91+ end
92+ x = coroutine.yield(f)
93+ coroutine.yield()
94+ end
95+ )
96+ local _, f = coroutine.resume(co) -- create closure over 'x' in coroutine
97+ collectgarbage("step", 0) -- make upvalue a survival
98+ old[1] = {"hello"} -- 'old' go to grayagain as 'touched1'
99+ coroutine.resume(co, {123}) -- its value will be new
100+ co = nil
101+ collectgarbage("step", 0) -- hit the barrier
102+ assert(f() == 123 and old[1][1] == "hello")
103+ collectgarbage("step", 0) -- run the collector once more
104+ -- make sure old[1] was not collected
105+ assert(f() == 123 and old[1][1] == "hello")
106+end
107+
108+
109+do -- bug introduced in commit 9cf3299fa
110+ local t = setmetatable({}, {__mode = "kv"}) -- all-weak table
111+ collectgarbage() -- full collection
112+ assert(not T or T.gcage(t) == "old")
113+ t[1] = {10}
114+ assert(not T or (T.gcage(t) == "touched1" and T.gccolor(t) == "gray"))
115+ collectgarbage("step", 0) -- minor collection
116+ assert(not T or (T.gcage(t) == "touched2" and T.gccolor(t) == "black"))
117+ collectgarbage("step", 0) -- minor collection
118+ assert(not T or T.gcage(t) == "old") -- t should be black, but it was gray
119+ t[1] = {10} -- no barrier here, so t was still old
120+ collectgarbage("step", 0) -- minor collection
121+ -- t, being old, is ignored by the collection, so it is not cleared
122+ assert(t[1] == nil) -- fails with the bug
123+end
124+
125+
126+if T == nil then
127+ (Message or print)('\n >>> testC not active: \z
128+ skipping some generational tests <<<\n')
129+ print 'OK'
130+ return
131+end
132+
133+
134+-- ensure that userdata barrier evolves correctly
135+do
136+ local U = T.newuserdata(0, 1)
137+ -- full collection makes 'U' old
138+ collectgarbage()
139+ assert(T.gcage(U) == "old")
140+
141+ -- U refers to a new table, so it becomes 'touched1'
142+ debug.setuservalue(U, {x = {234}})
143+ assert(T.gcage(U) == "touched1" and
144+ T.gcage(debug.getuservalue(U)) == "new")
145+
146+ -- both U and the table survive one more collection
147+ collectgarbage("step", 0)
148+ assert(T.gcage(U) == "touched2" and
149+ T.gcage(debug.getuservalue(U)) == "survival")
150+
151+ -- both U and the table survive yet another collection
152+ -- now everything is old
153+ collectgarbage("step", 0)
154+ assert(T.gcage(U) == "old" and
155+ T.gcage(debug.getuservalue(U)) == "old1")
156+
157+ -- data was not corrupted
158+ assert(debug.getuservalue(U).x[1] == 234)
159+end
160+
161+-- just to make sure
162+assert(collectgarbage'isrunning')
163+
164+
165+
166+-- just to make sure
167+assert(collectgarbage'isrunning')
168+
169+collectgarbage(oldmode)
170+
171+print('OK')
172+
zod/luau/tests/goto.luanew file+271
@@ -0,0 +1,271 @@
1+-- $Id: testes/goto.lua $
2+-- See Copyright Notice in file all.lua
3+
4+collectgarbage()
5+
6+local function errmsg (code, m)
7+ local st, msg = load(code)
8+ assert(not st and string.find(msg, m))
9+end
10+
11+-- cannot see label inside block
12+errmsg([[ goto l1; do ::l1:: end ]], "label 'l1'")
13+errmsg([[ do ::l1:: end goto l1; ]], "label 'l1'")
14+
15+-- repeated label
16+errmsg([[ ::l1:: ::l1:: ]], "label 'l1'")
17+errmsg([[ ::l1:: do ::l1:: end]], "label 'l1'")
18+
19+
20+-- undefined label
21+errmsg([[ goto l1; local aa ::l1:: ::l2:: print(3) ]], "local 'aa'")
22+
23+-- jumping over variable definition
24+errmsg([[
25+do local bb, cc; goto l1; end
26+local aa
27+::l1:: print(3)
28+]], "local 'aa'")
29+
30+-- jumping into a block
31+errmsg([[ do ::l1:: end goto l1 ]], "label 'l1'")
32+errmsg([[ goto l1 do ::l1:: end ]], "label 'l1'")
33+
34+-- cannot continue a repeat-until with variables
35+errmsg([[
36+ repeat
37+ if x then goto cont end
38+ local xuxu = 10
39+ ::cont::
40+ until xuxu < x
41+]], "local 'xuxu'")
42+
43+-- simple gotos
44+local x
45+do
46+ local y = 12
47+ goto l1
48+ ::l2:: x = x + 1; goto l3
49+ ::l1:: x = y; goto l2
50+end
51+::l3:: ::l3_1:: assert(x == 13)
52+
53+
54+-- long labels
55+do
56+ local prog = [[
57+ do
58+ local a = 1
59+ goto l%sa; a = a + 1
60+ ::l%sa:: a = a + 10
61+ goto l%sb; a = a + 2
62+ ::l%sb:: a = a + 20
63+ return a
64+ end
65+ ]]
66+ local label = string.rep("0123456789", 40)
67+ prog = string.format(prog, label, label, label, label)
68+ assert(assert(load(prog))() == 31)
69+end
70+
71+
72+-- ok to jump over local dec. to end of block
73+do
74+ goto l1
75+ local a = 23
76+ x = a
77+ ::l1::;
78+end
79+
80+while true do
81+ goto l4
82+ goto l1 -- ok to jump over local dec. to end of block
83+ goto l1 -- multiple uses of same label
84+ local x = 45
85+ ::l1:: ;;;
86+end
87+::l4:: assert(x == 13)
88+
89+if print then
90+ goto l1 -- ok to jump over local dec. to end of block
91+ error("should not be here")
92+ goto l2 -- ok to jump over local dec. to end of block
93+ local x
94+ ::l1:: ; ::l2:: ;;
95+else end
96+
97+-- to repeat a label in a different function is OK
98+local function foo ()
99+ local a = {}
100+ goto l3
101+ ::l1:: a[#a + 1] = 1; goto l2;
102+ ::l2:: a[#a + 1] = 2; goto l5;
103+ ::l3::
104+ ::l3a:: a[#a + 1] = 3; goto l1;
105+ ::l4:: a[#a + 1] = 4; goto l6;
106+ ::l5:: a[#a + 1] = 5; goto l4;
107+ ::l6:: assert(a[1] == 3 and a[2] == 1 and a[3] == 2 and
108+ a[4] == 5 and a[5] == 4)
109+ if not a[6] then a[6] = true; goto l3a end -- do it twice
110+end
111+
112+::l6:: foo()
113+
114+
115+do -- bug in 5.2 -> 5.3.2
116+ local x
117+ ::L1::
118+ local y -- cannot join this SETNIL with previous one
119+ assert(y == nil)
120+ y = true
121+ if x == nil then
122+ x = 1
123+ goto L1
124+ else
125+ x = x + 1
126+ end
127+ assert(x == 2 and y == true)
128+end
129+
130+-- bug in 5.3
131+do
132+ local first = true
133+ local a = false
134+ if true then
135+ goto LBL
136+ ::loop::
137+ a = true
138+ ::LBL::
139+ if first then
140+ first = false
141+ goto loop
142+ end
143+ end
144+ assert(a)
145+end
146+
147+do -- compiling infinite loops
148+ goto escape -- do not run the infinite loops
149+ ::a:: goto a
150+ ::b:: goto c
151+ ::c:: goto b
152+end
153+::escape::
154+--------------------------------------------------------------------------------
155+-- testing closing of upvalues
156+
157+local debug = require 'debug'
158+
159+local function foo ()
160+ local t = {}
161+ do
162+ local i = 1
163+ local a, b, c, d
164+ t[1] = function () return a, b, c, d end
165+ ::l1::
166+ local b
167+ do
168+ local c
169+ t[#t + 1] = function () return a, b, c, d end -- t[2], t[4], t[6]
170+ if i > 2 then goto l2 end
171+ do
172+ local d
173+ t[#t + 1] = function () return a, b, c, d end -- t[3], t[5]
174+ i = i + 1
175+ local a
176+ goto l1
177+ end
178+ end
179+ end
180+ ::l2:: return t
181+end
182+
183+local a = foo()
184+assert(#a == 6)
185+
186+-- all functions share same 'a'
187+for i = 2, 6 do
188+ assert(debug.upvalueid(a[1], 1) == debug.upvalueid(a[i], 1))
189+end
190+
191+-- 'b' and 'c' are shared among some of them
192+for i = 2, 6 do
193+ -- only a[1] uses external 'b'/'b'
194+ assert(debug.upvalueid(a[1], 2) ~= debug.upvalueid(a[i], 2))
195+ assert(debug.upvalueid(a[1], 3) ~= debug.upvalueid(a[i], 3))
196+end
197+
198+for i = 3, 5, 2 do
199+ -- inner functions share 'b'/'c' with previous ones
200+ assert(debug.upvalueid(a[i], 2) == debug.upvalueid(a[i - 1], 2))
201+ assert(debug.upvalueid(a[i], 3) == debug.upvalueid(a[i - 1], 3))
202+ -- but not with next ones
203+ assert(debug.upvalueid(a[i], 2) ~= debug.upvalueid(a[i + 1], 2))
204+ assert(debug.upvalueid(a[i], 3) ~= debug.upvalueid(a[i + 1], 3))
205+end
206+
207+-- only external 'd' is shared
208+for i = 2, 6, 2 do
209+ assert(debug.upvalueid(a[1], 4) == debug.upvalueid(a[i], 4))
210+end
211+
212+-- internal 'd's are all different
213+for i = 3, 5, 2 do
214+ for j = 1, 6 do
215+ assert((debug.upvalueid(a[i], 4) == debug.upvalueid(a[j], 4))
216+ == (i == j))
217+ end
218+end
219+
220+--------------------------------------------------------------------------------
221+-- testing if x goto optimizations
222+
223+local function testG (a)
224+ if a == 1 then
225+ goto l1
226+ error("should never be here!")
227+ elseif a == 2 then goto l2
228+ elseif a == 3 then goto l3
229+ elseif a == 4 then
230+ goto l1 -- go to inside the block
231+ error("should never be here!")
232+ ::l1:: a = a + 1 -- must go to 'if' end
233+ else
234+ goto l4
235+ ::l4a:: a = a * 2; goto l4b
236+ error("should never be here!")
237+ ::l4:: goto l4a
238+ error("should never be here!")
239+ ::l4b::
240+ end
241+ do return a end
242+ ::l2:: do return "2" end
243+ ::l3:: do return "3" end
244+ ::l1:: return "1"
245+end
246+
247+assert(testG(1) == "1")
248+assert(testG(2) == "2")
249+assert(testG(3) == "3")
250+assert(testG(4) == 5)
251+assert(testG(5) == 10)
252+
253+do
254+ -- if x back goto out of scope of upvalue
255+ local X
256+ goto L1
257+
258+ ::L2:: goto L3
259+
260+ ::L1:: do
261+ local a <close> = setmetatable({}, {__close = function () X = true end})
262+ assert(X == nil)
263+ if a then goto L2 end -- jumping back out of scope of 'a'
264+ end
265+
266+ ::L3:: assert(X == true) -- checks that 'a' was correctly closed
267+end
268+--------------------------------------------------------------------------------
269+
270+
271+print'OK'
zod/luau/tests/heavy.luanew file+173
@@ -0,0 +1,173 @@
1+-- $Id: heavy.lua,v 1.7 2017/12/29 15:42:15 roberto Exp $
2+-- See Copyright Notice in file all.lua
3+
4+local function teststring ()
5+ print("creating a string too long")
6+ do
7+ local a = "x"
8+ local st, msg = pcall(function ()
9+ while true do
10+ a = a .. a.. a.. a.. a.. a.. a.. a.. a.. a
11+ .. a .. a.. a.. a.. a.. a.. a.. a.. a.. a
12+ .. a .. a.. a.. a.. a.. a.. a.. a.. a.. a
13+ .. a .. a.. a.. a.. a.. a.. a.. a.. a.. a
14+ .. a .. a.. a.. a.. a.. a.. a.. a.. a.. a
15+ .. a .. a.. a.. a.. a.. a.. a.. a.. a.. a
16+ .. a .. a.. a.. a.. a.. a.. a.. a.. a.. a
17+ .. a .. a.. a.. a.. a.. a.. a.. a.. a.. a
18+ .. a .. a.. a.. a.. a.. a.. a.. a.. a.. a
19+ .. a .. a.. a.. a.. a.. a.. a.. a.. a.. a
20+ print(string.format("string with %d bytes", #a))
21+ end
22+ end)
23+ assert(not st and
24+ (string.find(msg, "string length overflow") or
25+ string.find(msg, "not enough memory")))
26+ print("string length overflow with " .. #a * 100)
27+ end
28+ print('+')
29+end
30+
31+local function loadrep (x, what)
32+ local p = 1<<20
33+ local s = string.rep(x, p)
34+ local count = 0
35+ local function f()
36+ count = count + p
37+ if count % (0x80*p) == 0 then
38+ io.stderr:write("(", count // 2^20, " M)")
39+ end
40+ return s
41+ end
42+ local st, msg = load(f, "=big")
43+ print("\nmemory: ", collectgarbage'count' * 1024)
44+ msg = string.match(msg, "^[^\n]+") -- get only first line
45+ print(string.format("total: 0x%x %s ('%s')", count, what, msg))
46+ return st, msg
47+end
48+
49+
50+function controlstruct ()
51+ print("control structure too long")
52+ local lim = ((1 << 24) - 2) // 3
53+ local s = string.rep("a = a + 1\n", lim)
54+ s = "while true do " .. s .. "end"
55+ assert(load(s))
56+ print("ok with " .. lim .. " lines")
57+ lim = lim + 3
58+ s = string.rep("a = a + 1\n", lim)
59+ s = "while true do " .. s .. "end"
60+ local st, msg = load(s)
61+ assert(not st and string.find(msg, "too long"))
62+ print(msg)
63+end
64+
65+
66+function manylines ()
67+ print("loading chunk with too many lines")
68+ local st, msg = loadrep("\n", "lines")
69+ assert(not st and string.find(msg, "too many lines"))
70+ print('+')
71+end
72+
73+
74+function hugeid ()
75+ print("loading chunk with huge identifier")
76+ local st, msg = loadrep("a", "chars")
77+ assert(not st and
78+ (string.find(msg, "lexical element too long") or
79+ string.find(msg, "not enough memory")))
80+ print('+')
81+end
82+
83+function toomanyinst ()
84+ print("loading chunk with too many instructions")
85+ local st, msg = loadrep("a = 10; ", "instructions")
86+ print('+')
87+end
88+
89+
90+local function loadrepfunc (prefix, f)
91+ local count = -1
92+ local function aux ()
93+ count = count + 1
94+ if count == 0 then
95+ return prefix
96+ else
97+ if count % (0x100000) == 0 then
98+ io.stderr:write("(", count // 2^20, " M)")
99+ end
100+ return f(count)
101+ end
102+ end
103+ local st, msg = load(aux, "k")
104+ print("\nmemory: ", collectgarbage'count' * 1024)
105+ msg = string.match(msg, "^[^\n]+") -- get only first line
106+ print("expected error: ", msg)
107+end
108+
109+
110+function toomanyconst ()
111+ print("loading function with too many constants")
112+ loadrepfunc("function foo () return {0,",
113+ function (n)
114+ -- convert 'n' to a string in the format [["...",]],
115+ -- where '...' is a kind of number in base 128
116+ -- (in a range that does not include either the double quote
117+ -- and the escape.)
118+ return string.char(34,
119+ ((n // 128^0) & 127) + 128,
120+ ((n // 128^1) & 127) + 128,
121+ ((n // 128^2) & 127) + 128,
122+ ((n // 128^3) & 127) + 128,
123+ ((n // 128^4) & 127) + 128,
124+ 34, 44)
125+ end)
126+end
127+
128+
129+function toomanystr ()
130+ local a = {}
131+ local st, msg = pcall(function ()
132+ for i = 1, math.huge do
133+ if i % (0x100000) == 0 then
134+ io.stderr:write("(", i // 2^20, " M)")
135+ end
136+ a[i] = string.pack("I", i)
137+ end
138+ end)
139+ local size = #a
140+ a = collectgarbage'count'
141+ print("\nmemory:", a * 1024)
142+ print("expected error:", msg)
143+ print("size:", size)
144+end
145+
146+
147+function toomanyidx ()
148+ local a = {}
149+ local st, msg = pcall(function ()
150+ for i = 1, math.huge do
151+ if i % (0x100000) == 0 then
152+ io.stderr:write("(", i // 2^20, " M)")
153+ end
154+ a[i] = i
155+ end
156+ end)
157+ print("\nmemory: ", collectgarbage'count' * 1024)
158+ print("expected error: ", msg)
159+ print("size:", #a)
160+end
161+
162+
163+
164+-- teststring()
165+-- controlstruct()
166+-- manylines()
167+-- hugeid()
168+-- toomanyinst()
169+-- toomanyconst()
170+-- toomanystr()
171+toomanyidx()
172+
173+print "OK"
zod/luau/tests/literals.luanew file+343
@@ -0,0 +1,343 @@
1+-- $Id: testes/literals.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print('testing scanner')
5+
6+local debug = require "debug"
7+
8+
9+local function dostring (x) return assert(load(x), "")() end
10+
11+dostring("x \v\f = \t\r 'a\0a' \v\f\f")
12+assert(x == 'a\0a' and string.len(x) == 3)
13+_G.x = nil
14+
15+-- escape sequences
16+assert('\n\"\'\\' == [[
17+
18+"'\]])
19+
20+assert(string.find("\a\b\f\n\r\t\v", "^%c%c%c%c%c%c%c$"))
21+
22+-- assume ASCII just for tests:
23+assert("\09912" == 'c12')
24+assert("\99ab" == 'cab')
25+assert("\099" == '\99')
26+assert("\099\n" == 'c\10')
27+assert('\0\0\0alo' == '\0' .. '\0\0' .. 'alo')
28+
29+assert(010 .. 020 .. -030 == "1020-30")
30+
31+-- hexadecimal escapes
32+assert("\x00\x05\x10\x1f\x3C\xfF\xe8" == "\0\5\16\31\60\255\232")
33+
34+local function lexstring (x, y, n)
35+ local f = assert(load('return ' .. x ..
36+ ', require"debug".getinfo(1).currentline', ''))
37+ local s, l = f()
38+ assert(s == y and l == n)
39+end
40+
41+lexstring("'abc\\z \n efg'", "abcefg", 2)
42+lexstring("'abc\\z \n\n\n'", "abc", 4)
43+lexstring("'\\z \n\t\f\v\n'", "", 3)
44+lexstring("[[\nalo\nalo\n\n]]", "alo\nalo\n\n", 5)
45+lexstring("[[\nalo\ralo\n\n]]", "alo\nalo\n\n", 5)
46+lexstring("[[\nalo\ralo\r\n]]", "alo\nalo\n", 4)
47+lexstring("[[\ralo\n\ralo\r\n]]", "alo\nalo\n", 4)
48+lexstring("[[alo]\n]alo]]", "alo]\n]alo", 2)
49+
50+assert("abc\z
51+ def\z
52+ ghi\z
53+ " == 'abcdefghi')
54+
55+
56+-- UTF-8 sequences
57+assert("\u{0}\u{00000000}\x00\0" == string.char(0, 0, 0, 0))
58+
59+-- limits for 1-byte sequences
60+assert("\u{0}\u{7F}" == "\x00\x7F")
61+
62+-- limits for 2-byte sequences
63+assert("\u{80}\u{7FF}" == "\xC2\x80\xDF\xBF")
64+
65+-- limits for 3-byte sequences
66+assert("\u{800}\u{FFFF}" == "\xE0\xA0\x80\xEF\xBF\xBF")
67+
68+-- limits for 4-byte sequences
69+assert("\u{10000}\u{1FFFFF}" == "\xF0\x90\x80\x80\xF7\xBF\xBF\xBF")
70+
71+-- limits for 5-byte sequences
72+assert("\u{200000}\u{3FFFFFF}" == "\xF8\x88\x80\x80\x80\xFB\xBF\xBF\xBF\xBF")
73+
74+-- limits for 6-byte sequences
75+assert("\u{4000000}\u{7FFFFFFF}" ==
76+ "\xFC\x84\x80\x80\x80\x80\xFD\xBF\xBF\xBF\xBF\xBF")
77+
78+
79+-- Error in escape sequences
80+local function lexerror (s, err)
81+ local st, msg = load('return ' .. s, '')
82+ if err ~= '<eof>' then err = err .. "'" end
83+ assert(not st and string.find(msg, "near .-" .. err))
84+end
85+
86+lexerror([["abc\x"]], [[\x"]])
87+lexerror([["abc\x]], [[\x]])
88+lexerror([["\x]], [[\x]])
89+lexerror([["\x5"]], [[\x5"]])
90+lexerror([["\x5]], [[\x5]])
91+lexerror([["\xr"]], [[\xr]])
92+lexerror([["\xr]], [[\xr]])
93+lexerror([["\x.]], [[\x.]])
94+lexerror([["\x8%"]], [[\x8%%]])
95+lexerror([["\xAG]], [[\xAG]])
96+lexerror([["\g"]], [[\g]])
97+lexerror([["\g]], [[\g]])
98+lexerror([["\."]], [[\%.]])
99+
100+lexerror([["\999"]], [[\999"]])
101+lexerror([["xyz\300"]], [[\300"]])
102+lexerror([[" \256"]], [[\256"]])
103+
104+-- errors in UTF-8 sequences
105+lexerror([["abc\u{100000000}"]], [[abc\u{100000000]]) -- too large
106+lexerror([["abc\u11r"]], [[abc\u1]]) -- missing '{'
107+lexerror([["abc\u"]], [[abc\u"]]) -- missing '{'
108+lexerror([["abc\u{11r"]], [[abc\u{11r]]) -- missing '}'
109+lexerror([["abc\u{11"]], [[abc\u{11"]]) -- missing '}'
110+lexerror([["abc\u{11]], [[abc\u{11]]) -- missing '}'
111+lexerror([["abc\u{r"]], [[abc\u{r]]) -- no digits
112+
113+-- unfinished strings
114+lexerror("[=[alo]]", "<eof>")
115+lexerror("[=[alo]=", "<eof>")
116+lexerror("[=[alo]", "<eof>")
117+lexerror("'alo", "<eof>")
118+lexerror("'alo \\z \n\n", "<eof>")
119+lexerror("'alo \\z", "<eof>")
120+lexerror([['alo \98]], "<eof>")
121+
122+-- valid characters in variable names
123+for i = 0, 255 do
124+ local s = string.char(i)
125+ assert(not string.find(s, "[a-zA-Z_]") == not load(s .. "=1", ""))
126+ assert(not string.find(s, "[a-zA-Z_0-9]") ==
127+ not load("a" .. s .. "1 = 1", ""))
128+end
129+
130+
131+-- long variable names
132+
133+local var1 = string.rep('a', 15000) .. '1'
134+local var2 = string.rep('a', 15000) .. '2'
135+local prog = string.format([[
136+ %s = 5
137+ %s = %s + 1
138+ return function () return %s - %s end
139+]], var1, var2, var1, var1, var2)
140+local f = dostring(prog)
141+assert(_G[var1] == 5 and _G[var2] == 6 and f() == -1)
142+_G[var1], _G[var2] = nil
143+print('+')
144+
145+-- escapes --
146+assert("\n\t" == [[
147+
148+ ]])
149+assert([[
150+
151+ $debug]] == "\n $debug")
152+assert([[ [ ]] ~= [[ ] ]])
153+-- long strings --
154+local b = "001234567890123456789012345678901234567891234567890123456789012345678901234567890012345678901234567890123456789012345678912345678901234567890123456789012345678900123456789012345678901234567890123456789123456789012345678901234567890123456789001234567890123456789012345678901234567891234567890123456789012345678901234567890012345678901234567890123456789012345678912345678901234567890123456789012345678900123456789012345678901234567890123456789123456789012345678901234567890123456789001234567890123456789012345678901234567891234567890123456789012345678901234567890012345678901234567890123456789012345678912345678901234567890123456789012345678900123456789012345678901234567890123456789123456789012345678901234567890123456789001234567890123456789012345678901234567891234567890123456789012345678901234567890012345678901234567890123456789012345678912345678901234567890123456789012345678900123456789012345678901234567890123456789123456789012345678901234567890123456789"
155+assert(string.len(b) == 960)
156+prog = [=[
157+print('+')
158+
159+local a1 = [["this is a 'string' with several 'quotes'"]]
160+local a2 = "'quotes'"
161+
162+assert(string.find(a1, a2) == 34)
163+print('+')
164+
165+a1 = [==[temp = [[an arbitrary value]]; ]==]
166+assert(load(a1))()
167+assert(temp == 'an arbitrary value')
168+_G.temp = nil
169+-- long strings --
170+local b = "001234567890123456789012345678901234567891234567890123456789012345678901234567890012345678901234567890123456789012345678912345678901234567890123456789012345678900123456789012345678901234567890123456789123456789012345678901234567890123456789001234567890123456789012345678901234567891234567890123456789012345678901234567890012345678901234567890123456789012345678912345678901234567890123456789012345678900123456789012345678901234567890123456789123456789012345678901234567890123456789001234567890123456789012345678901234567891234567890123456789012345678901234567890012345678901234567890123456789012345678912345678901234567890123456789012345678900123456789012345678901234567890123456789123456789012345678901234567890123456789001234567890123456789012345678901234567891234567890123456789012345678901234567890012345678901234567890123456789012345678912345678901234567890123456789012345678900123456789012345678901234567890123456789123456789012345678901234567890123456789"
171+assert(string.len(b) == 960)
172+print('+')
173+
174+local a = [[00123456789012345678901234567890123456789123456789012345678901234567890123456789
175+00123456789012345678901234567890123456789123456789012345678901234567890123456789
176+00123456789012345678901234567890123456789123456789012345678901234567890123456789
177+00123456789012345678901234567890123456789123456789012345678901234567890123456789
178+00123456789012345678901234567890123456789123456789012345678901234567890123456789
179+00123456789012345678901234567890123456789123456789012345678901234567890123456789
180+00123456789012345678901234567890123456789123456789012345678901234567890123456789
181+00123456789012345678901234567890123456789123456789012345678901234567890123456789
182+00123456789012345678901234567890123456789123456789012345678901234567890123456789
183+00123456789012345678901234567890123456789123456789012345678901234567890123456789
184+00123456789012345678901234567890123456789123456789012345678901234567890123456789
185+00123456789012345678901234567890123456789123456789012345678901234567890123456789
186+00123456789012345678901234567890123456789123456789012345678901234567890123456789
187+00123456789012345678901234567890123456789123456789012345678901234567890123456789
188+00123456789012345678901234567890123456789123456789012345678901234567890123456789
189+00123456789012345678901234567890123456789123456789012345678901234567890123456789
190+00123456789012345678901234567890123456789123456789012345678901234567890123456789
191+00123456789012345678901234567890123456789123456789012345678901234567890123456789
192+00123456789012345678901234567890123456789123456789012345678901234567890123456789
193+00123456789012345678901234567890123456789123456789012345678901234567890123456789
194+00123456789012345678901234567890123456789123456789012345678901234567890123456789
195+00123456789012345678901234567890123456789123456789012345678901234567890123456789
196+00123456789012345678901234567890123456789123456789012345678901234567890123456789
197+]]
198+assert(string.len(a) == 1863)
199+assert(string.sub(a, 1, 40) == string.sub(b, 1, 40))
200+x = 1
201+]=]
202+
203+print('+')
204+_G.x = nil
205+dostring(prog)
206+assert(x)
207+_G.x = nil
208+
209+
210+
211+do -- reuse of long strings
212+
213+ -- get the address of a string
214+ local function getadd (s) return string.format("%p", s) end
215+
216+ local s1 <const> = "01234567890123456789012345678901234567890123456789"
217+ local s2 <const> = "01234567890123456789012345678901234567890123456789"
218+ local s3 = "01234567890123456789012345678901234567890123456789"
219+ local function foo() return s1 end
220+ local function foo1() return s3 end
221+ local function foo2()
222+ return "01234567890123456789012345678901234567890123456789"
223+ end
224+ local a1 = getadd(s1)
225+ assert(a1 == getadd(s2))
226+ assert(a1 == getadd(foo()))
227+ assert(a1 == getadd(foo1()))
228+ assert(a1 == getadd(foo2()))
229+
230+ local sd = "0123456789" .. "0123456789012345678901234567890123456789"
231+ assert(sd == s1 and getadd(sd) ~= a1)
232+end
233+
234+
235+-- testing line ends
236+prog = [[
237+local a = 1 -- a comment
238+local b = 2
239+
240+
241+x = [=[
242+hi
243+]=]
244+y = "\
245+hello\r\n\
246+"
247+return require"debug".getinfo(1).currentline
248+]]
249+
250+for _, n in pairs{"\n", "\r", "\n\r", "\r\n"} do
251+ local prog, nn = string.gsub(prog, "\n", n)
252+ assert(dostring(prog) == nn)
253+ assert(_G.x == "hi\n" and _G.y == "\nhello\r\n\n")
254+end
255+_G.x, _G.y = nil
256+
257+
258+-- testing comments and strings with long brackets
259+local a = [==[]=]==]
260+assert(a == "]=")
261+
262+a = [==[[===[[=[]]=][====[]]===]===]==]
263+assert(a == "[===[[=[]]=][====[]]===]===")
264+
265+a = [====[[===[[=[]]=][====[]]===]===]====]
266+assert(a == "[===[[=[]]=][====[]]===]===")
267+
268+a = [=[]]]]]]]]]=]
269+assert(a == "]]]]]]]]")
270+
271+
272+--[===[
273+x y z [==[ blu foo
274+]==
275+]
276+]=]==]
277+error error]=]===]
278+
279+-- generate all strings of four of these chars
280+local x = {"=", "[", "]", "\n"}
281+local len = 4
282+local function gen (c, n)
283+ if n==0 then coroutine.yield(c)
284+ else
285+ for _, a in pairs(x) do
286+ gen(c..a, n-1)
287+ end
288+ end
289+end
290+
291+for s in coroutine.wrap(function () gen("", len) end) do
292+ assert(s == load("return [====[\n"..s.."]====]", "")())
293+end
294+
295+
296+-- testing decimal point locale
297+if os.setlocale("pt_BR") or os.setlocale("ptb") then
298+ assert(tonumber("3,4") == 3.4 and tonumber"3.4" == 3.4)
299+ assert(tonumber(" -.4 ") == -0.4)
300+ assert(tonumber(" +0x.41 ") == 0X0.41)
301+ assert(not load("a = (3,4)"))
302+ assert(assert(load("return 3.4"))() == 3.4)
303+ assert(assert(load("return .4,3"))() == .4)
304+ assert(assert(load("return 4."))() == 4.)
305+ assert(assert(load("return 4.+.5"))() == 4.5)
306+
307+ assert(" 0x.1 " + " 0x,1" + "-0X.1\t" == 0x0.1)
308+
309+ assert(not tonumber"inf" and not tonumber"NAN")
310+
311+ assert(assert(load(string.format("return %q", 4.51)))() == 4.51)
312+
313+ local a,b = load("return 4.5.")
314+ assert(string.find(b, "'4%.5%.'"))
315+
316+ assert(os.setlocale("C"))
317+else
318+ (Message or print)(
319+ '\n >>> pt_BR locale not available: skipping decimal point tests <<<\n')
320+end
321+
322+
323+-- testing %q x line ends
324+local s = "a string with \r and \n and \r\n and \n\r"
325+local c = string.format("return %q", s)
326+assert(assert(load(c))() == s)
327+
328+-- testing errors
329+assert(not load"a = 'non-ending string")
330+assert(not load"a = 'non-ending string\n'")
331+assert(not load"a = '\\345'")
332+assert(not load"a = [=x]")
333+
334+local function malformednum (n, exp)
335+ local s, msg = load("return " .. n)
336+ assert(not s and string.find(msg, exp))
337+end
338+
339+malformednum("0xe-", "near <eof>")
340+malformednum("0xep-p", "malformed number")
341+malformednum("1print()", "malformed number")
342+
343+print('OK')
zod/luau/tests/locals.luanew file+1181
@@ -0,0 +1,1181 @@
1+-- $Id: testes/locals.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print('testing local variables and environments')
5+
6+local debug = require"debug"
7+
8+local tracegc = require"tracegc"
9+
10+
11+-- bug in 5.1:
12+
13+local function f(x) x = nil; return x end
14+assert(f(10) == nil)
15+
16+local function f() local x; return x end
17+assert(f(10) == nil)
18+
19+local function f(x) x = nil; local y; return x, y end
20+assert(f(10) == nil and select(2, f(20)) == nil)
21+
22+do
23+ local i = 10
24+ do local i = 100; assert(i==100) end
25+ do local i = 1000; assert(i==1000) end
26+ assert(i == 10)
27+ if i ~= 10 then
28+ local i = 20
29+ else
30+ local i = 30
31+ assert(i == 30)
32+ end
33+end
34+
35+
36+
37+f = nil
38+
39+local f
40+local x = 1
41+
42+a = nil
43+load('local a = {}')()
44+assert(a == nil)
45+
46+function f (a)
47+ local _1, _2, _3, _4, _5
48+ local _6, _7, _8, _9, _10
49+ local x = 3
50+ local b = a
51+ local c,d = a,b
52+ if (d == b) then
53+ local x = 'q'
54+ x = b
55+ assert(x == 2)
56+ else
57+ assert(nil)
58+ end
59+ assert(x == 3)
60+ local f = 10
61+end
62+
63+local b=10
64+local a; repeat local b; a,b=1,2; assert(a+1==b); until a+b==3
65+
66+
67+assert(x == 1)
68+
69+f(2)
70+assert(type(f) == 'function')
71+
72+
73+local function getenv (f)
74+ local a,b = debug.getupvalue(f, 1)
75+ assert(a == '_ENV')
76+ return b
77+end
78+
79+-- test for global table of loaded chunks
80+assert(getenv(load"a=3") == _G)
81+local c = {}; local f = load("a = 3", nil, nil, c)
82+assert(getenv(f) == c)
83+assert(c.a == nil)
84+f()
85+assert(c.a == 3)
86+
87+-- old test for limits for special instructions
88+do
89+ local i = 2
90+ local p = 4 -- p == 2^i
91+ repeat
92+ for j=-3,3 do
93+ assert(load(string.format([[local a=%s;
94+ a=a+%s;
95+ assert(a ==2^%s)]], j, p-j, i), '')) ()
96+ assert(load(string.format([[local a=%s;
97+ a=a-%s;
98+ assert(a==-2^%s)]], -j, p-j, i), '')) ()
99+ assert(load(string.format([[local a,b=0,%s;
100+ a=b-%s;
101+ assert(a==-2^%s)]], -j, p-j, i), '')) ()
102+ end
103+ p = 2 * p; i = i + 1
104+ until p <= 0
105+end
106+
107+print'+'
108+
109+
110+if rawget(_G, "T") then
111+ -- testing clearing of dead elements from tables
112+ collectgarbage("stop") -- stop GC
113+ local a = {[{}] = 4, [3] = 0, alo = 1,
114+ a1234567890123456789012345678901234567890 = 10}
115+
116+ local t = T.querytab(a)
117+
118+ for k,_ in pairs(a) do a[k] = undef end
119+ collectgarbage() -- restore GC and collect dead fields in 'a'
120+ for i=0,t-1 do
121+ local k = querytab(a, i)
122+ assert(k == nil or type(k) == 'number' or k == 'alo')
123+ end
124+
125+ -- testing allocation errors during table insertions
126+ local a = {}
127+ local function additems ()
128+ a.x = true; a.y = true; a.z = true
129+ a[1] = true
130+ a[2] = true
131+ end
132+ for i = 1, math.huge do
133+ T.alloccount(i)
134+ local st, msg = pcall(additems)
135+ T.alloccount()
136+ local count = 0
137+ for k, v in pairs(a) do
138+ assert(a[k] == v)
139+ count = count + 1
140+ end
141+ if st then assert(count == 5); break end
142+ end
143+end
144+
145+
146+-- testing lexical environments
147+
148+assert(_ENV == _G)
149+
150+do
151+local dummy
152+local _ENV = (function (...) return ... end)(_G, dummy) -- {
153+
154+do local _ENV = {assert=assert}; assert(true) end
155+local mt = {_G = _G}
156+local foo,x
157+A = false -- "declare" A
158+do local _ENV = mt
159+ function foo (x)
160+ A = x
161+ do local _ENV = _G; A = 1000 end
162+ return function (x) return A .. x end
163+ end
164+end
165+assert(getenv(foo) == mt)
166+x = foo('hi'); assert(mt.A == 'hi' and A == 1000)
167+assert(x('*') == mt.A .. '*')
168+
169+do local _ENV = {assert=assert, A=10};
170+ do local _ENV = {assert=assert, A=20};
171+ assert(A==20);x=A
172+ end
173+ assert(A==10 and x==20)
174+end
175+assert(x==20)
176+
177+A = nil
178+
179+
180+do -- constants
181+ local a<const>, b, c<const> = 10, 20, 30
182+ b = a + c + b -- 'b' is not constant
183+ assert(a == 10 and b == 60 and c == 30)
184+ local function checkro (name, code)
185+ local st, msg = load(code)
186+ local gab = string.format("attempt to assign to const variable '%s'", name)
187+ assert(not st and string.find(msg, gab))
188+ end
189+ checkro("y", "local x, y <const>, z = 10, 20, 30; x = 11; y = 12")
190+ checkro("x", "local x <const>, y, z <const> = 10, 20, 30; x = 11")
191+ checkro("z", "local x <const>, y, z <const> = 10, 20, 30; y = 10; z = 11")
192+ checkro("foo", "local foo <const> = 10; function foo() end")
193+ checkro("foo", "local foo <const> = {}; function foo() end")
194+
195+ checkro("z", [[
196+ local a, z <const>, b = 10;
197+ function foo() a = 20; z = 32; end
198+ ]])
199+
200+ checkro("var1", [[
201+ local a, var1 <const> = 10;
202+ function foo() a = 20; z = function () var1 = 12; end end
203+ ]])
204+end
205+
206+
207+print"testing to-be-closed variables"
208+
209+local function stack(n) n = ((n == 0) or stack(n - 1)) end
210+
211+local function func2close (f, x, y)
212+ local obj = setmetatable({}, {__close = f})
213+ if x then
214+ return x, obj, y
215+ else
216+ return obj
217+ end
218+end
219+
220+
221+do
222+ local a = {}
223+ do
224+ local b <close> = false -- not to be closed
225+ local x <close> = setmetatable({"x"}, {__close = function (self)
226+ a[#a + 1] = self[1] end})
227+ local w, y <close>, z = func2close(function (self, err)
228+ assert(err == nil); a[#a + 1] = "y"
229+ end, 10, 20)
230+ local c <close> = nil -- not to be closed
231+ a[#a + 1] = "in"
232+ assert(w == 10 and z == 20)
233+ end
234+ a[#a + 1] = "out"
235+ assert(a[1] == "in" and a[2] == "y" and a[3] == "x" and a[4] == "out")
236+end
237+
238+do
239+ local X = false
240+
241+ local x, closescope = func2close(function (_, msg)
242+ stack(10);
243+ assert(msg == nil)
244+ X = true
245+ end, 100)
246+ assert(x == 100); x = 101; -- 'x' is not read-only
247+
248+ -- closing functions do not corrupt returning values
249+ local function foo (x)
250+ local _ <close> = closescope
251+ return x, X, 23
252+ end
253+
254+ local a, b, c = foo(1.5)
255+ assert(a == 1.5 and b == false and c == 23 and X == true)
256+
257+ X = false
258+ foo = function (x)
259+ local _<close> = func2close(function (_, msg)
260+ -- without errors, enclosing function should be still active when
261+ -- __close is called
262+ assert(debug.getinfo(2).name == "foo")
263+ assert(msg == nil)
264+ end)
265+ local _<close> = closescope
266+ local y = 15
267+ return y
268+ end
269+
270+ assert(foo() == 15 and X == true)
271+
272+ X = false
273+ foo = function ()
274+ local x <close> = closescope
275+ return x
276+ end
277+
278+ assert(foo() == closescope and X == true)
279+
280+end
281+
282+
283+-- testing to-be-closed x compile-time constants
284+-- (there were some bugs here in Lua 5.4-rc3, due to a confusion
285+-- between compile levels and stack levels of variables)
286+do
287+ local flag = false
288+ local x = setmetatable({},
289+ {__close = function() assert(flag == false); flag = true end})
290+ local y <const> = nil
291+ local z <const> = nil
292+ do
293+ local a <close> = x
294+ end
295+ assert(flag) -- 'x' must be closed here
296+end
297+
298+do
299+ -- similar problem, but with implicit close in for loops
300+ local flag = false
301+ local x = setmetatable({},
302+ {__close = function () assert(flag == false); flag = true end})
303+ -- return an empty iterator, nil, nil, and 'x' to be closed
304+ local function a ()
305+ return (function () return nil end), nil, nil, x
306+ end
307+ local v <const> = 1
308+ local w <const> = 1
309+ local x <const> = 1
310+ local y <const> = 1
311+ local z <const> = 1
312+ for k in a() do
313+ a = k
314+ end -- ending the loop must close 'x'
315+ assert(flag) -- 'x' must be closed here
316+end
317+
318+
319+
320+do
321+ -- calls cannot be tail in the scope of to-be-closed variables
322+ local X, Y
323+ local function foo ()
324+ local _ <close> = func2close(function () Y = 10 end)
325+ assert(X == true and Y == nil) -- 'X' not closed yet
326+ return 1,2,3
327+ end
328+
329+ local function bar ()
330+ local _ <close> = func2close(function () X = false end)
331+ X = true
332+ do
333+ return foo() -- not a tail call!
334+ end
335+ end
336+
337+ local a, b, c, d = bar()
338+ assert(a == 1 and b == 2 and c == 3 and X == false and Y == 10 and d == nil)
339+end
340+
341+
342+do
343+ -- bug in 5.4.3: previous condition (calls cannot be tail in the
344+ -- scope of to-be-closed variables) must be valid for tbc variables
345+ -- created by 'for' loops.
346+
347+ local closed = false
348+
349+ local function foo ()
350+ return function () return true end, 0, 0,
351+ func2close(function () closed = true end)
352+ end
353+
354+ local function tail() return closed end
355+
356+ local function foo1 ()
357+ for k in foo() do return tail() end
358+ end
359+
360+ assert(foo1() == false)
361+ assert(closed == true)
362+end
363+
364+
365+do
366+ -- bug in 5.4.4: 'break' may generate wrong 'close' instruction when
367+ -- leaving a loop block.
368+
369+ local closed = false
370+
371+ local o1 = setmetatable({}, {__close=function() closed = true end})
372+
373+ local function test()
374+ for k, v in next, {}, nil, o1 do
375+ local function f() return k end -- create an upvalue
376+ break
377+ end
378+ assert(closed)
379+ end
380+
381+ test()
382+end
383+
384+
385+do print("testing errors in __close")
386+
387+ -- original error is in __close
388+ local function foo ()
389+
390+ local x <close> =
391+ func2close(function (self, msg)
392+ assert(string.find(msg, "@y"))
393+ error("@x")
394+ end)
395+
396+ local x1 <close> =
397+ func2close(function (self, msg)
398+ assert(string.find(msg, "@y"))
399+ end)
400+
401+ local gc <close> = func2close(function () collectgarbage() end)
402+
403+ local y <close> =
404+ func2close(function (self, msg)
405+ assert(string.find(msg, "@z")) -- error in 'z'
406+ error("@y")
407+ end)
408+
409+ local z <close> =
410+ func2close(function (self, msg)
411+ assert(msg == nil)
412+ error("@z")
413+ end)
414+
415+ return 200
416+ end
417+
418+ local stat, msg = pcall(foo, false)
419+ assert(string.find(msg, "@x"))
420+
421+
422+ -- original error not in __close
423+ local function foo ()
424+
425+ local x <close> =
426+ func2close(function (self, msg)
427+ -- after error, 'foo' was discarded, so caller now
428+ -- must be 'pcall'
429+ assert(debug.getinfo(2).name == "pcall")
430+ assert(string.find(msg, "@x1"))
431+ end)
432+
433+ local x1 <close> =
434+ func2close(function (self, msg)
435+ assert(debug.getinfo(2).name == "pcall")
436+ assert(string.find(msg, "@y"))
437+ error("@x1")
438+ end)
439+
440+ local gc <close> = func2close(function () collectgarbage() end)
441+
442+ local y <close> =
443+ func2close(function (self, msg)
444+ assert(debug.getinfo(2).name == "pcall")
445+ assert(string.find(msg, "@z"))
446+ error("@y")
447+ end)
448+
449+ local first = true
450+ local z <close> =
451+ func2close(function (self, msg)
452+ assert(debug.getinfo(2).name == "pcall")
453+ -- 'z' close is called once
454+ assert(first and msg == 4)
455+ first = false
456+ error("@z")
457+ end)
458+
459+ error(4) -- original error
460+ end
461+
462+ local stat, msg = pcall(foo, true)
463+ assert(string.find(msg, "@x1"))
464+
465+ -- error leaving a block
466+ local function foo (...)
467+ do
468+ local x1 <close> =
469+ func2close(function (self, msg)
470+ assert(string.find(msg, "@X"))
471+ error("@Y")
472+ end)
473+
474+ local x123 <close> =
475+ func2close(function (_, msg)
476+ assert(msg == nil)
477+ error("@X")
478+ end)
479+ end
480+ os.exit(false) -- should not run
481+ end
482+
483+ local st, msg = xpcall(foo, debug.traceback)
484+ assert(string.match(msg, "^[^ ]* @Y"))
485+
486+ -- error in toclose in vararg function
487+ local function foo (...)
488+ local x123 <close> = func2close(function () error("@x123") end)
489+ end
490+
491+ local st, msg = xpcall(foo, debug.traceback)
492+ assert(string.match(msg, "^[^ ]* @x123"))
493+ assert(string.find(msg, "in metamethod 'close'"))
494+end
495+
496+
497+do -- errors due to non-closable values
498+ local function foo ()
499+ local x <close> = {}
500+ os.exit(false) -- should not run
501+ end
502+ local stat, msg = pcall(foo)
503+ assert(not stat and
504+ string.find(msg, "variable 'x' got a non%-closable value"))
505+
506+ local function foo ()
507+ local xyz <close> = setmetatable({}, {__close = print})
508+ getmetatable(xyz).__close = nil -- remove metamethod
509+ end
510+ local stat, msg = pcall(foo)
511+ assert(not stat and string.find(msg, "metamethod 'close'"))
512+
513+ local function foo ()
514+ local a1 <close> = func2close(function (_, msg)
515+ assert(string.find(msg, "number value"))
516+ error(12)
517+ end)
518+ local a2 <close> = setmetatable({}, {__close = print})
519+ local a3 <close> = func2close(function (_, msg)
520+ assert(msg == nil)
521+ error(123)
522+ end)
523+ getmetatable(a2).__close = 4 -- invalidate metamethod
524+ end
525+ local stat, msg = pcall(foo)
526+ assert(not stat and msg == 12)
527+end
528+
529+
530+do -- tbc inside close methods
531+ local track = {}
532+ local function foo ()
533+ local x <close> = func2close(function ()
534+ local xx <close> = func2close(function (_, msg)
535+ assert(msg == nil)
536+ track[#track + 1] = "xx"
537+ end)
538+ track[#track + 1] = "x"
539+ end)
540+ track[#track + 1] = "foo"
541+ return 20, 30, 40
542+ end
543+ local a, b, c, d = foo()
544+ assert(a == 20 and b == 30 and c == 40 and d == nil)
545+ assert(track[1] == "foo" and track[2] == "x" and track[3] == "xx")
546+
547+ -- again, with errors
548+ local track = {}
549+ local function foo ()
550+ local x0 <close> = func2close(function (_, msg)
551+ assert(msg == 202)
552+ track[#track + 1] = "x0"
553+ end)
554+ local x <close> = func2close(function ()
555+ local xx <close> = func2close(function (_, msg)
556+ assert(msg == 101)
557+ track[#track + 1] = "xx"
558+ error(202)
559+ end)
560+ track[#track + 1] = "x"
561+ error(101)
562+ end)
563+ track[#track + 1] = "foo"
564+ return 20, 30, 40
565+ end
566+ local st, msg = pcall(foo)
567+ assert(not st and msg == 202)
568+ assert(track[1] == "foo" and track[2] == "x" and track[3] == "xx" and
569+ track[4] == "x0")
570+end
571+
572+
573+local function checktable (t1, t2)
574+ assert(#t1 == #t2)
575+ for i = 1, #t1 do
576+ assert(t1[i] == t2[i])
577+ end
578+end
579+
580+
581+do -- test for tbc variable high in the stack
582+
583+ -- function to force a stack overflow
584+ local function overflow (n)
585+ overflow(n + 1)
586+ end
587+
588+ -- error handler will create tbc variable handling a stack overflow,
589+ -- high in the stack
590+ local function errorh (m)
591+ assert(string.find(m, "stack overflow"))
592+ local x <close> = func2close(function (o) o[1] = 10 end)
593+ return x
594+ end
595+
596+ local flag
597+ local st, obj
598+ -- run test in a coroutine so as not to swell the main stack
599+ local co = coroutine.wrap(function ()
600+ -- tbc variable down the stack
601+ local y <close> = func2close(function (obj, msg)
602+ assert(msg == nil)
603+ obj[1] = 100
604+ flag = obj
605+ end)
606+ tracegc.stop()
607+ st, obj = xpcall(overflow, errorh, 0)
608+ tracegc.start()
609+ end)
610+ co()
611+ assert(not st and obj[1] == 10 and flag[1] == 100)
612+end
613+
614+
615+if rawget(_G, "T") then
616+
617+ do
618+ -- bug in 5.4.3
619+ -- 'lua_settop' may use a pointer to stack invalidated by 'luaF_close'
620+
621+ -- reduce stack size
622+ collectgarbage(); collectgarbage(); collectgarbage()
623+
624+ -- force a stack reallocation
625+ local function loop (n)
626+ if n < 400 then loop(n + 1) end
627+ end
628+
629+ -- close metamethod will reallocate the stack
630+ local o = setmetatable({}, {__close = function () loop(0) end})
631+
632+ local script = [[toclose 2; settop 1; return 1]]
633+
634+ assert(T.testC(script, o) == script)
635+
636+ end
637+
638+
639+ -- memory error inside closing function
640+ local function foo ()
641+ local y <close> = func2close(function () T.alloccount() end)
642+ local x <close> = setmetatable({}, {__close = function ()
643+ T.alloccount(0); local x = {} -- force a memory error
644+ end})
645+ error(1000) -- common error inside the function's body
646+ end
647+
648+ stack(5) -- ensure a minimal number of CI structures
649+
650+ -- despite memory error, 'y' will be executed and
651+ -- memory limit will be lifted
652+ local _, msg = pcall(foo)
653+ assert(msg == "not enough memory")
654+
655+ local closemsg
656+ local close = func2close(function (self, msg)
657+ T.alloccount()
658+ closemsg = msg
659+ end)
660+
661+ -- set a memory limit and return a closing object to remove the limit
662+ local function enter (count)
663+ stack(10) -- reserve some stack space
664+ T.alloccount(count)
665+ closemsg = nil
666+ return close
667+ end
668+
669+ local function test ()
670+ local x <close> = enter(0) -- set a memory limit
671+ local y = {} -- raise a memory error
672+ end
673+
674+ local _, msg = pcall(test)
675+ assert(msg == "not enough memory" and closemsg == "not enough memory")
676+
677+
678+ -- repeat test with extra closing upvalues
679+ local function test ()
680+ local xxx <close> = func2close(function (self, msg)
681+ assert(msg == "not enough memory");
682+ error(1000) -- raise another error
683+ end)
684+ local xx <close> = func2close(function (self, msg)
685+ assert(msg == "not enough memory");
686+ end)
687+ local x <close> = enter(0) -- set a memory limit
688+ local y = {} -- raise a memory error
689+ end
690+
691+ local _, msg = pcall(test)
692+ assert(msg == 1000 and closemsg == "not enough memory")
693+
694+ do -- testing 'toclose' in C string buffer
695+ collectgarbage()
696+ local s = string.rep('a', 10000) -- large string
697+ local m = T.totalmem()
698+ collectgarbage("stop")
699+ s = string.upper(s) -- allocate buffer + new string (10K each)
700+ -- ensure buffer was deallocated
701+ assert(T.totalmem() - m <= 11000)
702+ collectgarbage("restart")
703+ end
704+
705+ do -- now some tests for freeing buffer in case of errors
706+ local lim = 10000 -- some size larger than the static buffer
707+ local extra = 2000 -- some extra memory (for callinfo, etc.)
708+
709+ local s = string.rep("a", lim)
710+
711+ -- concat this table needs two buffer resizes (one for each 's')
712+ local a = {s, s}
713+
714+ collectgarbage(); collectgarbage()
715+
716+ local m = T.totalmem()
717+ collectgarbage("stop")
718+
719+ -- error in the first buffer allocation
720+ T. totalmem(m + extra)
721+ assert(not pcall(table.concat, a))
722+ -- first buffer was not even allocated
723+ assert(T.totalmem() - m <= extra)
724+
725+ -- error in the second buffer allocation
726+ T. totalmem(m + lim + extra)
727+ assert(not pcall(table.concat, a))
728+ -- first buffer was released by 'toclose'
729+ assert(T.totalmem() - m <= extra)
730+
731+ -- error in creation of final string
732+ T.totalmem(m + 2 * lim + extra)
733+ assert(not pcall(table.concat, a))
734+ -- second buffer was released by 'toclose'
735+ assert(T.totalmem() - m <= extra)
736+
737+ -- userdata, buffer, buffer, final string
738+ T.totalmem(m + 4*lim + extra)
739+ assert(#table.concat(a) == 2*lim)
740+
741+ T.totalmem(0) -- remove memory limit
742+ collectgarbage("restart")
743+
744+ print'+'
745+ end
746+
747+
748+ do
749+ -- '__close' vs. return hooks in C functions
750+ local trace = {}
751+
752+ local function hook (event)
753+ trace[#trace + 1] = event .. " " .. (debug.getinfo(2).name or "?")
754+ end
755+
756+ -- create tbc variables to be used by C function
757+ local x = func2close(function (_,msg)
758+ trace[#trace + 1] = "x"
759+ end)
760+
761+ local y = func2close(function (_,msg)
762+ trace[#trace + 1] = "y"
763+ end)
764+
765+ debug.sethook(hook, "r")
766+ local t = {T.testC([[
767+ toclose 2 # x
768+ pushnum 10
769+ pushint 20
770+ toclose 3 # y
771+ return 2
772+ ]], x, y)}
773+ debug.sethook()
774+
775+ -- hooks ran before return hook from 'testC'
776+ checktable(trace,
777+ {"return sethook", "y", "return ?", "x", "return ?", "return testC"})
778+ -- results are correct
779+ checktable(t, {10, 20})
780+ end
781+end
782+
783+
784+do -- '__close' vs. return hooks in Lua functions
785+ local trace = {}
786+
787+ local function hook (event)
788+ trace[#trace + 1] = event .. " " .. debug.getinfo(2).name
789+ end
790+
791+ local function foo (...)
792+ local x <close> = func2close(function (_,msg)
793+ trace[#trace + 1] = "x"
794+ end)
795+
796+ local y <close> = func2close(function (_,msg)
797+ debug.sethook(hook, "r")
798+ end)
799+
800+ return ...
801+ end
802+
803+ local t = {foo(10,20,30)}
804+ debug.sethook()
805+ checktable(t, {10, 20, 30})
806+ checktable(trace,
807+ {"return sethook", "return close", "x", "return close", "return foo"})
808+end
809+
810+
811+print "to-be-closed variables in coroutines"
812+
813+do
814+ -- yielding inside closing metamethods
815+
816+ local trace = {}
817+ local co = coroutine.wrap(function ()
818+
819+ trace[#trace + 1] = "nowX"
820+
821+ -- will be closed after 'y'
822+ local x <close> = func2close(function (_, msg)
823+ assert(msg == nil)
824+ trace[#trace + 1] = "x1"
825+ coroutine.yield("x")
826+ trace[#trace + 1] = "x2"
827+ end)
828+
829+ return pcall(function ()
830+ do -- 'z' will be closed first
831+ local z <close> = func2close(function (_, msg)
832+ assert(msg == nil)
833+ trace[#trace + 1] = "z1"
834+ coroutine.yield("z")
835+ trace[#trace + 1] = "z2"
836+ end)
837+ end
838+
839+ trace[#trace + 1] = "nowY"
840+
841+ -- will be closed after 'z'
842+ local y <close> = func2close(function(_, msg)
843+ assert(msg == nil)
844+ trace[#trace + 1] = "y1"
845+ coroutine.yield("y")
846+ trace[#trace + 1] = "y2"
847+ end)
848+
849+ return 10, 20, 30
850+ end)
851+ end)
852+
853+ assert(co() == "z")
854+ assert(co() == "y")
855+ assert(co() == "x")
856+ checktable({co()}, {true, 10, 20, 30})
857+ checktable(trace, {"nowX", "z1", "z2", "nowY", "y1", "y2", "x1", "x2"})
858+
859+end
860+
861+
862+do
863+ -- yielding inside closing metamethods while returning
864+ -- (bug in 5.4.3)
865+
866+ local extrares -- result from extra yield (if any)
867+
868+ local function check (body, extra, ...)
869+ local t = table.pack(...) -- expected returns
870+ local co = coroutine.wrap(body)
871+ if extra then
872+ extrares = co() -- runs until first (extra) yield
873+ end
874+ local res = table.pack(co()) -- runs until yield inside '__close'
875+ assert(res.n == 2 and res[2] == nil)
876+ local res2 = table.pack(co()) -- runs until end of function
877+ assert(res2.n == t.n)
878+ for i = 1, #t do
879+ if t[i] == "x" then
880+ assert(res2[i] == res[1]) -- value that was closed
881+ else
882+ assert(res2[i] == t[i])
883+ end
884+ end
885+ end
886+
887+ local function foo ()
888+ local x <close> = func2close(coroutine.yield)
889+ local extra <close> = func2close(function (self)
890+ assert(self == extrares)
891+ coroutine.yield(100)
892+ end)
893+ extrares = extra
894+ return table.unpack{10, x, 30}
895+ end
896+ check(foo, true, 10, "x", 30)
897+ assert(extrares == 100)
898+
899+ local function foo ()
900+ local x <close> = func2close(coroutine.yield)
901+ return
902+ end
903+ check(foo, false)
904+
905+ local function foo ()
906+ local x <close> = func2close(coroutine.yield)
907+ local y, z = 20, 30
908+ return x
909+ end
910+ check(foo, false, "x")
911+
912+ local function foo ()
913+ local x <close> = func2close(coroutine.yield)
914+ local extra <close> = func2close(coroutine.yield)
915+ return table.unpack({}, 1, 100) -- 100 nils
916+ end
917+ check(foo, true, table.unpack({}, 1, 100))
918+
919+end
920+
921+do
922+ -- yielding inside closing metamethods after an error
923+
924+ local co = coroutine.wrap(function ()
925+
926+ local function foo (err)
927+
928+ local z <close> = func2close(function(_, msg)
929+ assert(msg == nil or msg == err + 20)
930+ coroutine.yield("z")
931+ return 100, 200
932+ end)
933+
934+ local y <close> = func2close(function(_, msg)
935+ -- still gets the original error (if any)
936+ assert(msg == err or (msg == nil and err == 1))
937+ coroutine.yield("y")
938+ if err then error(err + 20) end -- creates or changes the error
939+ end)
940+
941+ local x <close> = func2close(function(_, msg)
942+ assert(msg == err or (msg == nil and err == 1))
943+ coroutine.yield("x")
944+ return 100, 200
945+ end)
946+
947+ if err == 10 then error(err) else return 10, 20 end
948+ end
949+
950+ coroutine.yield(pcall(foo, nil)) -- no error
951+ coroutine.yield(pcall(foo, 1)) -- error in __close
952+ return pcall(foo, 10) -- 'foo' will raise an error
953+ end)
954+
955+ local a, b = co() -- first foo: no error
956+ assert(a == "x" and b == nil) -- yields inside 'x'; Ok
957+ a, b = co()
958+ assert(a == "y" and b == nil) -- yields inside 'y'; Ok
959+ a, b = co()
960+ assert(a == "z" and b == nil) -- yields inside 'z'; Ok
961+ local a, b, c = co()
962+ assert(a and b == 10 and c == 20) -- returns from 'pcall(foo, nil)'
963+
964+ local a, b = co() -- second foo: error in __close
965+ assert(a == "x" and b == nil) -- yields inside 'x'; Ok
966+ a, b = co()
967+ assert(a == "y" and b == nil) -- yields inside 'y'; Ok
968+ a, b = co()
969+ assert(a == "z" and b == nil) -- yields inside 'z'; Ok
970+ local st, msg = co() -- reports the error in 'y'
971+ assert(not st and msg == 21)
972+
973+ local a, b = co() -- third foo: error in function body
974+ assert(a == "x" and b == nil) -- yields inside 'x'; Ok
975+ a, b = co()
976+ assert(a == "y" and b == nil) -- yields inside 'y'; Ok
977+ a, b = co()
978+ assert(a == "z" and b == nil) -- yields inside 'z'; Ok
979+ local st, msg = co() -- gets final error
980+ assert(not st and msg == 10 + 20)
981+
982+end
983+
984+
985+do
986+ -- an error in a wrapped coroutine closes variables
987+ local x = false
988+ local y = false
989+ local co = coroutine.wrap(function ()
990+ local xv <close> = func2close(function () x = true end)
991+ do
992+ local yv <close> = func2close(function () y = true end)
993+ coroutine.yield(100) -- yield doesn't close variable
994+ end
995+ coroutine.yield(200) -- yield doesn't close variable
996+ error(23) -- error does
997+ end)
998+
999+ local b = co()
1000+ assert(b == 100 and not x and not y)
1001+ b = co()
1002+ assert(b == 200 and not x and y)
1003+ local a, b = pcall(co)
1004+ assert(not a and b == 23 and x and y)
1005+end
1006+
1007+
1008+do
1009+
1010+ -- error in a wrapped coroutine raising errors when closing a variable
1011+ local x = 0
1012+ local co = coroutine.wrap(function ()
1013+ local xx <close> = func2close(function (_, msg)
1014+ x = x + 1;
1015+ assert(string.find(msg, "@XXX"))
1016+ error("@YYY")
1017+ end)
1018+ local xv <close> = func2close(function () x = x + 1; error("@XXX") end)
1019+ coroutine.yield(100)
1020+ error(200)
1021+ end)
1022+ assert(co() == 100); assert(x == 0)
1023+ local st, msg = pcall(co); assert(x == 2)
1024+ assert(not st and string.find(msg, "@YYY")) -- should get error raised
1025+
1026+ local x = 0
1027+ local y = 0
1028+ co = coroutine.wrap(function ()
1029+ local xx <close> = func2close(function (_, err)
1030+ y = y + 1;
1031+ assert(string.find(err, "XXX"))
1032+ error("YYY")
1033+ end)
1034+ local xv <close> = func2close(function ()
1035+ x = x + 1; error("XXX")
1036+ end)
1037+ coroutine.yield(100)
1038+ return 200
1039+ end)
1040+ assert(co() == 100); assert(x == 0)
1041+ local st, msg = pcall(co)
1042+ assert(x == 1 and y == 1)
1043+ -- should get first error raised
1044+ assert(not st and string.find(msg, "%w+%.%w+:%d+: YYY"))
1045+
1046+end
1047+
1048+
1049+-- a suspended coroutine should not close its variables when collected
1050+local co
1051+co = coroutine.wrap(function()
1052+ -- should not run
1053+ local x <close> = func2close(function () os.exit(false) end)
1054+ co = nil
1055+ coroutine.yield()
1056+end)
1057+co() -- start coroutine
1058+assert(co == nil) -- eventually it will be collected
1059+collectgarbage()
1060+
1061+
1062+if rawget(_G, "T") then
1063+ print("to-be-closed variables x coroutines in C")
1064+ do
1065+ local token = 0
1066+ local count = 0
1067+ local f = T.makeCfunc[[
1068+ toclose 1
1069+ toclose 2
1070+ return .
1071+ ]]
1072+
1073+ local obj = func2close(function (_, msg)
1074+ count = count + 1
1075+ token = coroutine.yield(count, token)
1076+ end)
1077+
1078+ local co = coroutine.wrap(f)
1079+ local ct, res = co(obj, obj, 10, 20, 30, 3) -- will return 10, 20, 30
1080+ -- initial token value, after closing 2nd obj
1081+ assert(ct == 1 and res == 0)
1082+ -- run until yield when closing 1st obj
1083+ ct, res = co(100)
1084+ assert(ct == 2 and res == 100)
1085+ res = {co(200)} -- run until end
1086+ assert(res[1] == 10 and res[2] == 20 and res[3] == 30 and res[4] == nil)
1087+ assert(token == 200)
1088+ end
1089+
1090+ do
1091+ local f = T.makeCfunc[[
1092+ toclose 1
1093+ return .
1094+ ]]
1095+
1096+ local obj = func2close(function ()
1097+ local temp
1098+ local x <close> = func2close(function ()
1099+ coroutine.yield(temp)
1100+ return 1,2,3 -- to be ignored
1101+ end)
1102+ temp = coroutine.yield("closing obj")
1103+ return 1,2,3 -- to be ignored
1104+ end)
1105+
1106+ local co = coroutine.wrap(f)
1107+ local res = co(obj, 10, 30, 1) -- will return only 30
1108+ assert(res == "closing obj")
1109+ res = co("closing x")
1110+ assert(res == "closing x")
1111+ res = {co()}
1112+ assert(res[1] == 30 and res[2] == nil)
1113+ end
1114+
1115+ do
1116+ -- still cannot yield inside 'closeslot'
1117+ local f = T.makeCfunc[[
1118+ toclose 1
1119+ closeslot 1
1120+ ]]
1121+ local obj = func2close(coroutine.yield)
1122+ local co = coroutine.create(f)
1123+ local st, msg = coroutine.resume(co, obj)
1124+ assert(not st and string.find(msg, "attempt to yield across"))
1125+
1126+ -- nor outside a coroutine
1127+ local f = T.makeCfunc[[
1128+ toclose 1
1129+ ]]
1130+ local st, msg = pcall(f, obj)
1131+ assert(not st and string.find(msg, "attempt to yield from outside"))
1132+ end
1133+end
1134+
1135+
1136+
1137+-- to-be-closed variables in generic for loops
1138+do
1139+ local numopen = 0
1140+ local function open (x)
1141+ numopen = numopen + 1
1142+ return
1143+ function () -- iteraction function
1144+ x = x - 1
1145+ if x > 0 then return x end
1146+ end,
1147+ nil, -- state
1148+ nil, -- control variable
1149+ func2close(function () numopen = numopen - 1 end) -- closing function
1150+ end
1151+
1152+ local s = 0
1153+ for i in open(10) do
1154+ s = s + i
1155+ end
1156+ assert(s == 45 and numopen == 0)
1157+
1158+ local s = 0
1159+ for i in open(10) do
1160+ if i < 5 then break end
1161+ s = s + i
1162+ end
1163+ assert(s == 35 and numopen == 0)
1164+
1165+ local s = 0
1166+ for i in open(10) do
1167+ for j in open(10) do
1168+ if i + j < 5 then goto endloop end
1169+ s = s + i
1170+ end
1171+ end
1172+ ::endloop::
1173+ assert(s == 375 and numopen == 0)
1174+end
1175+
1176+print('OK')
1177+
1178+return 5,f
1179+
1180+end -- }
1181+
zod/luau/tests/main.luanew file+563
@@ -0,0 +1,563 @@
1+# testing special comment on first line
2+-- $Id: testes/main.lua $
3+-- See Copyright Notice in file all.lua
4+
5+-- most (all?) tests here assume a reasonable "Unix-like" shell
6+if _port then return end
7+
8+-- use only "double quotes" inside shell scripts (better change to
9+-- run on Windows)
10+
11+
12+print ("testing stand-alone interpreter")
13+
14+assert(os.execute()) -- machine has a system command
15+
16+local arg = arg or ARG
17+
18+local prog = os.tmpname()
19+local otherprog = os.tmpname()
20+local out = os.tmpname()
21+
22+local progname
23+do
24+ local i = 0
25+ while arg[i] do i=i-1 end
26+ progname = arg[i+1]
27+end
28+print("progname: "..progname)
29+
30+
31+local prepfile = function (s, mod, p)
32+ mod = mod and "wb" or "w" -- mod true means binary files
33+ p = p or prog -- file to write the program
34+ local f = io.open(p, mod)
35+ f:write(s)
36+ assert(f:close())
37+end
38+
39+local function getoutput ()
40+ local f = io.open(out)
41+ local t = f:read("a")
42+ f:close()
43+ assert(os.remove(out))
44+ return t
45+end
46+
47+local function checkprogout (s)
48+ -- expected result must end with new line
49+ assert(string.sub(s, -1) == "\n")
50+ local t = getoutput()
51+ for line in string.gmatch(s, ".-\n") do
52+ assert(string.find(t, line, 1, true))
53+ end
54+end
55+
56+local function checkout (s)
57+ local t = getoutput()
58+ if s ~= t then print(string.format("'%s' - '%s'\n", s, t)) end
59+ assert(s == t)
60+ return t
61+end
62+
63+
64+local function RUN (p, ...)
65+ p = string.gsub(p, "lua", '"'..progname..'"', 1)
66+ local s = string.format(p, ...)
67+ assert(os.execute(s))
68+end
69+
70+
71+local function NoRun (msg, p, ...)
72+ p = string.gsub(p, "lua", '"'..progname..'"', 1)
73+ local s = string.format(p, ...)
74+ s = string.format("%s >%s 2>&1", s, out) -- send output and error to 'out'
75+ assert(not os.execute(s))
76+ assert(string.find(getoutput(), msg, 1, true)) -- check error message
77+end
78+
79+RUN('lua -v')
80+
81+print(string.format("(temporary program file used in these tests: %s)", prog))
82+
83+-- running stdin as a file
84+prepfile""
85+RUN('lua - < %s > %s', prog, out)
86+checkout("")
87+
88+prepfile[[
89+ print(
90+1, a
91+)
92+]]
93+RUN('lua - < %s > %s', prog, out)
94+checkout("1\tnil\n")
95+
96+RUN('echo "print(10)\nprint(2)\n" | lua > %s', out)
97+checkout("10\n2\n")
98+
99+
100+-- testing BOM
101+prepfile("\xEF\xBB\xBF")
102+RUN('lua %s > %s', prog, out)
103+checkout("")
104+
105+prepfile("\xEF\xBB\xBFprint(3)")
106+RUN('lua %s > %s', prog, out)
107+checkout("3\n")
108+
109+prepfile("\xEF\xBB\xBF# comment!!\nprint(3)")
110+RUN('lua %s > %s', prog, out)
111+checkout("3\n")
112+
113+-- bad BOMs
114+prepfile("\xEF", true)
115+NoRun("unexpected symbol", 'lua %s', prog)
116+
117+prepfile("\xEF\xBB", true)
118+NoRun("unexpected symbol", 'lua %s', prog)
119+
120+prepfile("\xEFprint(3)", true)
121+NoRun("unexpected symbol", 'lua %s', prog)
122+
123+prepfile("\xEF\xBBprint(3)", true)
124+NoRun("unexpected symbol", 'lua %s', prog)
125+
126+
127+-- test option '-'
128+RUN('echo "print(arg[1])" | lua - -h > %s', out)
129+checkout("-h\n")
130+
131+-- test environment variables used by Lua
132+
133+prepfile("print(package.path)")
134+
135+-- test LUA_PATH
136+RUN('env LUA_INIT= LUA_PATH=x lua %s > %s', prog, out)
137+checkout("x\n")
138+
139+-- test LUA_PATH_version
140+RUN('env LUA_INIT= LUA_PATH_5_4=y LUA_PATH=x lua %s > %s', prog, out)
141+checkout("y\n")
142+
143+-- test LUA_CPATH
144+prepfile("print(package.cpath)")
145+RUN('env LUA_INIT= LUA_CPATH=xuxu lua %s > %s', prog, out)
146+checkout("xuxu\n")
147+
148+-- test LUA_CPATH_version
149+RUN('env LUA_INIT= LUA_CPATH_5_4=yacc LUA_CPATH=x lua %s > %s', prog, out)
150+checkout("yacc\n")
151+
152+-- test LUA_INIT (and its access to 'arg' table)
153+prepfile("print(X)")
154+RUN('env LUA_INIT="X=tonumber(arg[1])" lua %s 3.2 > %s', prog, out)
155+checkout("3.2\n")
156+
157+-- test LUA_INIT_version
158+prepfile("print(X)")
159+RUN('env LUA_INIT_5_4="X=10" LUA_INIT="X=3" lua %s > %s', prog, out)
160+checkout("10\n")
161+
162+-- test LUA_INIT for files
163+prepfile("x = x or 10; print(x); x = x + 1")
164+RUN('env LUA_INIT="@%s" lua %s > %s', prog, prog, out)
165+checkout("10\n11\n")
166+
167+-- test errors in LUA_INIT
168+NoRun('LUA_INIT:1: msg', 'env LUA_INIT="error(\'msg\')" lua')
169+
170+-- test option '-E'
171+local defaultpath, defaultCpath
172+
173+do
174+ prepfile("print(package.path, package.cpath)")
175+ RUN('env LUA_INIT="error(10)" LUA_PATH=xxx LUA_CPATH=xxx lua -E %s > %s',
176+ prog, out)
177+ local output = getoutput()
178+ defaultpath = string.match(output, "^(.-)\t")
179+ defaultCpath = string.match(output, "\t(.-)$")
180+
181+ -- running with an empty environment
182+ RUN('env -i lua %s > %s', prog, out)
183+ local out = getoutput()
184+ assert(defaultpath == string.match(output, "^(.-)\t"))
185+ assert(defaultCpath == string.match(output, "\t(.-)$"))
186+end
187+
188+-- paths did not change
189+assert(not string.find(defaultpath, "xxx") and
190+ string.find(defaultpath, "lua") and
191+ not string.find(defaultCpath, "xxx") and
192+ string.find(defaultCpath, "lua"))
193+
194+
195+-- test replacement of ';;' to default path
196+local function convert (p)
197+ prepfile("print(package.path)")
198+ RUN('env LUA_PATH="%s" lua %s > %s', p, prog, out)
199+ local expected = getoutput()
200+ expected = string.sub(expected, 1, -2) -- cut final end of line
201+ if string.find(p, ";;") then
202+ p = string.gsub(p, ";;", ";"..defaultpath..";")
203+ p = string.gsub(p, "^;", "") -- remove ';' at the beginning
204+ p = string.gsub(p, ";$", "") -- remove ';' at the end
205+ end
206+ assert(p == expected)
207+end
208+
209+convert(";")
210+convert(";;")
211+convert("a;;b")
212+convert(";;b")
213+convert("a;;")
214+convert("a;b;;c")
215+
216+
217+-- test -l over multiple libraries
218+prepfile("print(1); a=2; return {x=15}")
219+prepfile(("print(a); print(_G['%s'].x)"):format(prog), false, otherprog)
220+RUN('env LUA_PATH="?;;" lua -l %s -l%s -lstring -l io %s > %s', prog, otherprog, otherprog, out)
221+checkout("1\n2\n15\n2\n15\n")
222+
223+-- test explicit global names in -l
224+prepfile("print(str.upper'alo alo', m.max(10, 20))")
225+RUN("lua -l 'str=string' '-lm=math' -e 'print(m.sin(0))' %s > %s", prog, out)
226+checkout("0.0\nALO ALO\t20\n")
227+
228+
229+-- test module names with version sufix ("libs/lib2-v2")
230+RUN("env LUA_CPATH='./libs/?.so' lua -l lib2-v2 -e 'print(lib2.id())' > %s",
231+ out)
232+checkout("true\n")
233+
234+
235+-- test 'arg' table
236+local a = [[
237+ assert(#arg == 3 and arg[1] == 'a' and
238+ arg[2] == 'b' and arg[3] == 'c')
239+ assert(arg[-1] == '--' and arg[-2] == "-e " and arg[-3] == '%s')
240+ assert(arg[4] == undef and arg[-4] == undef)
241+ local a, b, c = ...
242+ assert(... == 'a' and a == 'a' and b == 'b' and c == 'c')
243+]]
244+a = string.format(a, progname)
245+prepfile(a)
246+RUN('lua "-e " -- %s a b c', prog) -- "-e " runs an empty command
247+
248+-- test 'arg' availability in libraries
249+prepfile"assert(arg)"
250+prepfile("assert(arg)", false, otherprog)
251+RUN('env LUA_PATH="?;;" lua -l%s - < %s', prog, otherprog)
252+
253+-- test messing up the 'arg' table
254+RUN('echo "print(...)" | lua -e "arg[1] = 100" - > %s', out)
255+checkout("100\n")
256+NoRun("'arg' is not a table", 'echo "" | lua -e "arg = 1" -')
257+
258+-- test error in 'print'
259+RUN('echo 10 | lua -e "print=nil" -i > /dev/null 2> %s', out)
260+assert(string.find(getoutput(), "error calling 'print'"))
261+
262+-- test 'debug.debug'
263+RUN('echo "io.stderr:write(1000)\ncont" | lua -e "require\'debug\'.debug()" 2> %s', out)
264+checkout("lua_debug> 1000lua_debug> ")
265+
266+
267+print("testing warnings")
268+
269+-- no warnings by default
270+RUN('echo "io.stderr:write(1); warn[[XXX]]" | lua 2> %s', out)
271+checkout("1")
272+
273+prepfile[[
274+warn("@allow") -- unknown control, ignored
275+warn("@off", "XXX", "@off") -- these are not control messages
276+warn("@off") -- this one is
277+warn("@on", "YYY", "@on") -- not control, but warn is off
278+warn("@off") -- keep it off
279+warn("@on") -- restart warnings
280+warn("", "@on") -- again, no control, real warning
281+warn("@on") -- keep it "started"
282+warn("Z", "Z", "Z") -- common warning
283+]]
284+RUN('lua -W %s 2> %s', prog, out)
285+checkout[[
286+Lua warning: @offXXX@off
287+Lua warning: @on
288+Lua warning: ZZZ
289+]]
290+
291+prepfile[[
292+warn("@allow")
293+-- create two objects to be finalized when closing state
294+-- the errors in the finalizers must generate warnings
295+u1 = setmetatable({}, {__gc = function () error("XYZ") end})
296+u2 = setmetatable({}, {__gc = function () error("ZYX") end})
297+]]
298+RUN('lua -W %s 2> %s', prog, out)
299+checkprogout("ZYX)\nXYZ)\n")
300+
301+-- bug since 5.2: finalizer called when closing a state could
302+-- subvert finalization order
303+prepfile[[
304+-- should be called last
305+print("creating 1")
306+setmetatable({}, {__gc = function () print(1) end})
307+
308+print("creating 2")
309+setmetatable({}, {__gc = function ()
310+ print("2")
311+ print("creating 3")
312+ -- this finalizer should not be called, as object will be
313+ -- created after 'lua_close' has been called
314+ setmetatable({}, {__gc = function () print(3) end})
315+ print(collectgarbage()) -- cannot call collector here
316+ os.exit(0, true)
317+end})
318+]]
319+RUN('lua -W %s > %s', prog, out)
320+checkout[[
321+creating 1
322+creating 2
323+2
324+creating 3
325+nil
326+1
327+]]
328+
329+
330+-- test many arguments
331+prepfile[[print(({...})[30])]]
332+RUN('lua %s %s > %s', prog, string.rep(" a", 30), out)
333+checkout("a\n")
334+
335+RUN([[lua "-eprint(1)" -ea=3 -e "print(a)" > %s]], out)
336+checkout("1\n3\n")
337+
338+-- test iteractive mode
339+prepfile[[
340+(6*2-6) -- ===
341+a =
342+10
343+print(a)
344+a]]
345+RUN([[lua -e"_PROMPT='' _PROMPT2=''" -i < %s > %s]], prog, out)
346+checkprogout("6\n10\n10\n\n")
347+
348+prepfile("a = [[b\nc\nd\ne]]\n=a")
349+RUN([[lua -e"_PROMPT='' _PROMPT2=''" -i < %s > %s]], prog, out)
350+checkprogout("b\nc\nd\ne\n\n")
351+
352+local prompt = "alo"
353+prepfile[[ --
354+a = 2
355+]]
356+RUN([[lua "-e_PROMPT='%s'" -i < %s > %s]], prompt, prog, out)
357+local t = getoutput()
358+assert(string.find(t, prompt .. ".*" .. prompt .. ".*" .. prompt))
359+
360+-- using the prompt default
361+prepfile[[ --
362+a = 2
363+]]
364+RUN([[lua -i < %s > %s]], prog, out)
365+local t = getoutput()
366+prompt = "> " -- the default
367+assert(string.find(t, prompt .. ".*" .. prompt .. ".*" .. prompt))
368+
369+
370+-- non-string prompt
371+prompt =
372+ "local C = 0;\z
373+ _PROMPT=setmetatable({},{__tostring = function () \z
374+ C = C + 1; return C end})"
375+prepfile[[ --
376+a = 2
377+]]
378+RUN([[lua -e "%s" -i < %s > %s]], prompt, prog, out)
379+local t = getoutput()
380+assert(string.find(t, [[
381+1 --
382+2a = 2
383+3
384+]], 1, true))
385+
386+
387+-- test for error objects
388+prepfile[[
389+debug = require "debug"
390+m = {x=0}
391+setmetatable(m, {__tostring = function(x)
392+ return tostring(debug.getinfo(4).currentline + x.x)
393+end})
394+error(m)
395+]]
396+NoRun(progname .. ": 6\n", [[lua %s]], prog)
397+
398+prepfile("error{}")
399+NoRun("error object is a table value", [[lua %s]], prog)
400+
401+
402+-- chunk broken in many lines
403+local s = [=[ --
404+function f ( x )
405+ local a = [[
406+xuxu
407+]]
408+ local b = "\
409+xuxu\n"
410+ if x == 11 then return 1 + 12 , 2 + 20 end --[[ test multiple returns ]]
411+ return x + 1
412+ --\\
413+end
414+return( f( 100 ) )
415+assert( a == b )
416+do return f( 11 ) end ]=]
417+s = string.gsub(s, ' ', '\n\n') -- change all spaces for newlines
418+prepfile(s)
419+RUN([[lua -e"_PROMPT='' _PROMPT2=''" -i < %s > %s]], prog, out)
420+checkprogout("101\n13\t22\n\n")
421+
422+prepfile[[#comment in 1st line without \n at the end]]
423+RUN('lua %s', prog)
424+
425+-- first-line comment with binary file
426+prepfile("#comment\n" .. string.dump(load("print(3)")), true)
427+RUN('lua %s > %s', prog, out)
428+checkout('3\n')
429+
430+-- close Lua with an open file
431+prepfile(string.format([[io.output(%q); io.write('alo')]], out))
432+RUN('lua %s', prog)
433+checkout('alo')
434+
435+-- bug in 5.2 beta (extra \0 after version line)
436+RUN([[lua -v -e"print'hello'" > %s]], out)
437+t = getoutput()
438+assert(string.find(t, "PUC%-Rio\nhello"))
439+
440+
441+-- testing os.exit
442+prepfile("os.exit(nil, true)")
443+RUN('lua %s', prog)
444+prepfile("os.exit(0, true)")
445+RUN('lua %s', prog)
446+prepfile("os.exit(true, true)")
447+RUN('lua %s', prog)
448+prepfile("os.exit(1, true)")
449+NoRun("", "lua %s", prog) -- no message
450+prepfile("os.exit(false, true)")
451+NoRun("", "lua %s", prog) -- no message
452+
453+
454+-- to-be-closed variables in main chunk
455+prepfile[[
456+ local x <close> = setmetatable({},
457+ {__close = function (self, err)
458+ assert(err == nil)
459+ print("Ok")
460+ end})
461+ local e1 <close> = setmetatable({}, {__close = function () print(120) end})
462+ os.exit(true, true)
463+]]
464+RUN('lua %s > %s', prog, out)
465+checkprogout("120\nOk\n")
466+
467+
468+-- remove temporary files
469+assert(os.remove(prog))
470+assert(os.remove(otherprog))
471+assert(not os.remove(out))
472+
473+-- invalid options
474+NoRun("unrecognized option '-h'", "lua -h")
475+NoRun("unrecognized option '---'", "lua ---")
476+NoRun("unrecognized option '-Ex'", "lua -Ex")
477+NoRun("unrecognized option '-vv'", "lua -vv")
478+NoRun("unrecognized option '-iv'", "lua -iv")
479+NoRun("'-e' needs argument", "lua -e")
480+NoRun("syntax error", "lua -e a")
481+NoRun("'-l' needs argument", "lua -l")
482+
483+
484+if T then -- test library?
485+ print("testing 'not enough memory' to create a state")
486+ NoRun("not enough memory", "env MEMLIMIT=100 lua")
487+
488+ -- testing 'warn'
489+ warn("@store")
490+ warn("@123", "456", "789")
491+ assert(_WARN == "@123456789"); _WARN = false
492+
493+ warn("zip", "", " ", "zap")
494+ assert(_WARN == "zip zap"); _WARN = false
495+ warn("ZIP", "", " ", "ZAP")
496+ assert(_WARN == "ZIP ZAP"); _WARN = false
497+ warn("@normal")
498+end
499+
500+do
501+ -- 'warn' must get at least one argument
502+ local st, msg = pcall(warn)
503+ assert(string.find(msg, "string expected"))
504+
505+ -- 'warn' does not leave unfinished warning in case of errors
506+ -- (message would appear in next warning)
507+ st, msg = pcall(warn, "SHOULD NOT APPEAR", {})
508+ assert(string.find(msg, "string expected"))
509+end
510+
511+print('+')
512+
513+print('testing Ctrl C')
514+do
515+ -- interrupt a script
516+ local function kill (pid)
517+ return os.execute(string.format('kill -INT %s 2> /dev/null', pid))
518+ end
519+
520+ -- function to run a script in background, returning its output file
521+ -- descriptor and its pid
522+ local function runback (luaprg)
523+ -- shell script to run 'luaprg' in background and echo its pid
524+ local shellprg = string.format('%s -e "%s" & echo $!', progname, luaprg)
525+ local f = io.popen(shellprg, "r") -- run shell script
526+ local pid = f:read() -- get pid for Lua script
527+ print("(if test fails now, it may leave a Lua script running in \z
528+ background, pid " .. pid .. ")")
529+ return f, pid
530+ end
531+
532+ -- Lua script that runs protected infinite loop and then prints '42'
533+ local f, pid = runback[[
534+ pcall(function () print(12); while true do end end); print(42)]]
535+ -- wait until script is inside 'pcall'
536+ assert(f:read() == "12")
537+ kill(pid) -- send INT signal to Lua script
538+ -- check that 'pcall' captured the exception and script continued running
539+ assert(f:read() == "42") -- expected output
540+ assert(f:close())
541+ print("done")
542+
543+ -- Lua script in a long unbreakable search
544+ local f, pid = runback[[
545+ print(15); string.find(string.rep('a', 100000), '.*b')]]
546+ -- wait (so script can reach the loop)
547+ assert(f:read() == "15")
548+ assert(os.execute("sleep 1"))
549+ -- must send at least two INT signals to stop this Lua script
550+ local n = 100
551+ for i = 0, 100 do -- keep sending signals
552+ if not kill(pid) then -- until it fails
553+ n = i -- number of non-failed kills
554+ break
555+ end
556+ end
557+ assert(f:close())
558+ assert(n >= 2)
559+ print(string.format("done (with %d kills)", n))
560+
561+end
562+
563+print("OK")
zod/luau/tests/math.luanew file+1024
@@ -0,0 +1,1024 @@
1+-- $Id: testes/math.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print("testing numbers and math lib")
5+
6+local minint <const> = math.mininteger
7+local maxint <const> = math.maxinteger
8+
9+local intbits <const> = math.floor(math.log(maxint, 2) + 0.5) + 1
10+assert((1 << intbits) == 0)
11+
12+assert(minint == 1 << (intbits - 1))
13+assert(maxint == minint - 1)
14+
15+-- number of bits in the mantissa of a floating-point number
16+local floatbits = 24
17+do
18+ local p = 2.0^floatbits
19+ while p < p + 1.0 do
20+ p = p * 2.0
21+ floatbits = floatbits + 1
22+ end
23+end
24+
25+local function isNaN (x)
26+ return (x ~= x)
27+end
28+
29+assert(isNaN(0/0))
30+assert(not isNaN(1/0))
31+
32+
33+do
34+ local x = 2.0^floatbits
35+ assert(x > x - 1.0 and x == x + 1.0)
36+
37+ print(string.format("%d-bit integers, %d-bit (mantissa) floats",
38+ intbits, floatbits))
39+end
40+
41+assert(math.type(0) == "integer" and math.type(0.0) == "float"
42+ and not math.type("10"))
43+
44+
45+local function checkerror (msg, f, ...)
46+ local s, err = pcall(f, ...)
47+ assert(not s and string.find(err, msg))
48+end
49+
50+local msgf2i = "number.* has no integer representation"
51+
52+-- float equality
53+local function eq (a,b,limit)
54+ if not limit then
55+ if floatbits >= 50 then limit = 1E-11
56+ else limit = 1E-5
57+ end
58+ end
59+ -- a == b needed for +inf/-inf
60+ return a == b or math.abs(a-b) <= limit
61+end
62+
63+
64+-- equality with types
65+local function eqT (a,b)
66+ return a == b and math.type(a) == math.type(b)
67+end
68+
69+
70+-- basic float notation
71+assert(0e12 == 0 and .0 == 0 and 0. == 0 and .2e2 == 20 and 2.E-1 == 0.2)
72+
73+do
74+ local a,b,c = "2", " 3e0 ", " 10 "
75+ assert(a+b == 5 and -b == -3 and b+"2" == 5 and "10"-c == 0)
76+ assert(type(a) == 'string' and type(b) == 'string' and type(c) == 'string')
77+ assert(a == "2" and b == " 3e0 " and c == " 10 " and -c == -" 10 ")
78+ assert(c%a == 0 and a^b == 08)
79+ a = 0
80+ assert(a == -a and 0 == -0)
81+end
82+
83+do
84+ local x = -1
85+ local mz = 0/x -- minus zero
86+ local t = {[0] = 10, 20, 30, 40, 50}
87+ assert(t[mz] == t[0] and t[-0] == t[0])
88+end
89+
90+do -- tests for 'modf'
91+ local a,b = math.modf(3.5)
92+ assert(a == 3.0 and b == 0.5)
93+ a,b = math.modf(-2.5)
94+ assert(a == -2.0 and b == -0.5)
95+ a,b = math.modf(-3e23)
96+ assert(a == -3e23 and b == 0.0)
97+ a,b = math.modf(3e35)
98+ assert(a == 3e35 and b == 0.0)
99+ a,b = math.modf(-1/0) -- -inf
100+ assert(a == -1/0 and b == 0.0)
101+ a,b = math.modf(1/0) -- inf
102+ assert(a == 1/0 and b == 0.0)
103+ a,b = math.modf(0/0) -- NaN
104+ assert(isNaN(a) and isNaN(b))
105+ a,b = math.modf(3) -- integer argument
106+ assert(eqT(a, 3) and eqT(b, 0.0))
107+ a,b = math.modf(minint)
108+ assert(eqT(a, minint) and eqT(b, 0.0))
109+end
110+
111+assert(math.huge > 10e30)
112+assert(-math.huge < -10e30)
113+
114+
115+-- integer arithmetic
116+assert(minint < minint + 1)
117+assert(maxint - 1 < maxint)
118+assert(0 - minint == minint)
119+assert(minint * minint == 0)
120+assert(maxint * maxint * maxint == maxint)
121+
122+
123+-- testing floor division and conversions
124+
125+for _, i in pairs{-16, -15, -3, -2, -1, 0, 1, 2, 3, 15} do
126+ for _, j in pairs{-16, -15, -3, -2, -1, 1, 2, 3, 15} do
127+ for _, ti in pairs{0, 0.0} do -- try 'i' as integer and as float
128+ for _, tj in pairs{0, 0.0} do -- try 'j' as integer and as float
129+ local x = i + ti
130+ local y = j + tj
131+ assert(i//j == math.floor(i/j))
132+ end
133+ end
134+ end
135+end
136+
137+assert(1//0.0 == 1/0)
138+assert(-1 // 0.0 == -1/0)
139+assert(eqT(3.5 // 1.5, 2.0))
140+assert(eqT(3.5 // -1.5, -3.0))
141+
142+do -- tests for different kinds of opcodes
143+ local x, y
144+ x = 1; assert(x // 0.0 == 1/0)
145+ x = 1.0; assert(x // 0 == 1/0)
146+ x = 3.5; assert(eqT(x // 1, 3.0))
147+ assert(eqT(x // -1, -4.0))
148+
149+ x = 3.5; y = 1.5; assert(eqT(x // y, 2.0))
150+ x = 3.5; y = -1.5; assert(eqT(x // y, -3.0))
151+end
152+
153+assert(maxint // maxint == 1)
154+assert(maxint // 1 == maxint)
155+assert((maxint - 1) // maxint == 0)
156+assert(maxint // (maxint - 1) == 1)
157+assert(minint // minint == 1)
158+assert(minint // minint == 1)
159+assert((minint + 1) // minint == 0)
160+assert(minint // (minint + 1) == 1)
161+assert(minint // 1 == minint)
162+
163+assert(minint // -1 == -minint)
164+assert(minint // -2 == 2^(intbits - 2))
165+assert(maxint // -1 == -maxint)
166+
167+
168+-- negative exponents
169+do
170+ assert(2^-3 == 1 / 2^3)
171+ assert(eq((-3)^-3, 1 / (-3)^3))
172+ for i = -3, 3 do -- variables avoid constant folding
173+ for j = -3, 3 do
174+ -- domain errors (0^(-n)) are not portable
175+ if not _port or i ~= 0 or j > 0 then
176+ assert(eq(i^j, 1 / i^(-j)))
177+ end
178+ end
179+ end
180+end
181+
182+-- comparison between floats and integers (border cases)
183+if floatbits < intbits then
184+ assert(2.0^floatbits == (1 << floatbits))
185+ assert(2.0^floatbits - 1.0 == (1 << floatbits) - 1.0)
186+ assert(2.0^floatbits - 1.0 ~= (1 << floatbits))
187+ -- float is rounded, int is not
188+ assert(2.0^floatbits + 1.0 ~= (1 << floatbits) + 1)
189+else -- floats can express all integers with full accuracy
190+ assert(maxint == maxint + 0.0)
191+ assert(maxint - 1 == maxint - 1.0)
192+ assert(minint + 1 == minint + 1.0)
193+ assert(maxint ~= maxint - 1.0)
194+end
195+assert(maxint + 0.0 == 2.0^(intbits - 1) - 1.0)
196+assert(minint + 0.0 == minint)
197+assert(minint + 0.0 == -2.0^(intbits - 1))
198+
199+
200+-- order between floats and integers
201+assert(1 < 1.1); assert(not (1 < 0.9))
202+assert(1 <= 1.1); assert(not (1 <= 0.9))
203+assert(-1 < -0.9); assert(not (-1 < -1.1))
204+assert(1 <= 1.1); assert(not (-1 <= -1.1))
205+assert(-1 < -0.9); assert(not (-1 < -1.1))
206+assert(-1 <= -0.9); assert(not (-1 <= -1.1))
207+assert(minint <= minint + 0.0)
208+assert(minint + 0.0 <= minint)
209+assert(not (minint < minint + 0.0))
210+assert(not (minint + 0.0 < minint))
211+assert(maxint < minint * -1.0)
212+assert(maxint <= minint * -1.0)
213+
214+do
215+ local fmaxi1 = 2^(intbits - 1)
216+ assert(maxint < fmaxi1)
217+ assert(maxint <= fmaxi1)
218+ assert(not (fmaxi1 <= maxint))
219+ assert(minint <= -2^(intbits - 1))
220+ assert(-2^(intbits - 1) <= minint)
221+end
222+
223+if floatbits < intbits then
224+ print("testing order (floats cannot represent all integers)")
225+ local fmax = 2^floatbits
226+ local ifmax = fmax | 0
227+ assert(fmax < ifmax + 1)
228+ assert(fmax - 1 < ifmax)
229+ assert(-(fmax - 1) > -ifmax)
230+ assert(not (fmax <= ifmax - 1))
231+ assert(-fmax > -(ifmax + 1))
232+ assert(not (-fmax >= -(ifmax - 1)))
233+
234+ assert(fmax/2 - 0.5 < ifmax//2)
235+ assert(-(fmax/2 - 0.5) > -ifmax//2)
236+
237+ assert(maxint < 2^intbits)
238+ assert(minint > -2^intbits)
239+ assert(maxint <= 2^intbits)
240+ assert(minint >= -2^intbits)
241+else
242+ print("testing order (floats can represent all integers)")
243+ assert(maxint < maxint + 1.0)
244+ assert(maxint < maxint + 0.5)
245+ assert(maxint - 1.0 < maxint)
246+ assert(maxint - 0.5 < maxint)
247+ assert(not (maxint + 0.0 < maxint))
248+ assert(maxint + 0.0 <= maxint)
249+ assert(not (maxint < maxint + 0.0))
250+ assert(maxint + 0.0 <= maxint)
251+ assert(maxint <= maxint + 0.0)
252+ assert(not (maxint + 1.0 <= maxint))
253+ assert(not (maxint + 0.5 <= maxint))
254+ assert(not (maxint <= maxint - 1.0))
255+ assert(not (maxint <= maxint - 0.5))
256+
257+ assert(minint < minint + 1.0)
258+ assert(minint < minint + 0.5)
259+ assert(minint <= minint + 0.5)
260+ assert(minint - 1.0 < minint)
261+ assert(minint - 1.0 <= minint)
262+ assert(not (minint + 0.0 < minint))
263+ assert(not (minint + 0.5 < minint))
264+ assert(not (minint < minint + 0.0))
265+ assert(minint + 0.0 <= minint)
266+ assert(minint <= minint + 0.0)
267+ assert(not (minint + 1.0 <= minint))
268+ assert(not (minint + 0.5 <= minint))
269+ assert(not (minint <= minint - 1.0))
270+end
271+
272+do
273+ local NaN <const> = 0/0
274+ assert(not (NaN < 0))
275+ assert(not (NaN > minint))
276+ assert(not (NaN <= -9))
277+ assert(not (NaN <= maxint))
278+ assert(not (NaN < maxint))
279+ assert(not (minint <= NaN))
280+ assert(not (minint < NaN))
281+ assert(not (4 <= NaN))
282+ assert(not (4 < NaN))
283+end
284+
285+
286+-- avoiding errors at compile time
287+local function checkcompt (msg, code)
288+ checkerror(msg, assert(load(code)))
289+end
290+checkcompt("divide by zero", "return 2 // 0")
291+checkcompt(msgf2i, "return 2.3 >> 0")
292+checkcompt(msgf2i, ("return 2.0^%d & 1"):format(intbits - 1))
293+checkcompt("field 'huge'", "return math.huge << 1")
294+checkcompt(msgf2i, ("return 1 | 2.0^%d"):format(intbits - 1))
295+checkcompt(msgf2i, "return 2.3 ~ 0.0")
296+
297+
298+-- testing overflow errors when converting from float to integer (runtime)
299+local function f2i (x) return x | x end
300+checkerror(msgf2i, f2i, math.huge) -- +inf
301+checkerror(msgf2i, f2i, -math.huge) -- -inf
302+checkerror(msgf2i, f2i, 0/0) -- NaN
303+
304+if floatbits < intbits then
305+ -- conversion tests when float cannot represent all integers
306+ assert(maxint + 1.0 == maxint + 0.0)
307+ assert(minint - 1.0 == minint + 0.0)
308+ checkerror(msgf2i, f2i, maxint + 0.0)
309+ assert(f2i(2.0^(intbits - 2)) == 1 << (intbits - 2))
310+ assert(f2i(-2.0^(intbits - 2)) == -(1 << (intbits - 2)))
311+ assert((2.0^(floatbits - 1) + 1.0) // 1 == (1 << (floatbits - 1)) + 1)
312+ -- maximum integer representable as a float
313+ local mf = maxint - (1 << (floatbits - intbits)) + 1
314+ assert(f2i(mf + 0.0) == mf) -- OK up to here
315+ mf = mf + 1
316+ assert(f2i(mf + 0.0) ~= mf) -- no more representable
317+else
318+ -- conversion tests when float can represent all integers
319+ assert(maxint + 1.0 > maxint)
320+ assert(minint - 1.0 < minint)
321+ assert(f2i(maxint + 0.0) == maxint)
322+ checkerror("no integer rep", f2i, maxint + 1.0)
323+ checkerror("no integer rep", f2i, minint - 1.0)
324+end
325+
326+-- 'minint' should be representable as a float no matter the precision
327+assert(f2i(minint + 0.0) == minint)
328+
329+
330+-- testing numeric strings
331+
332+assert("2" + 1 == 3)
333+assert("2 " + 1 == 3)
334+assert(" -2 " + 1 == -1)
335+assert(" -0xa " + 1 == -9)
336+
337+
338+-- Literal integer Overflows (new behavior in 5.3.3)
339+do
340+ -- no overflows
341+ assert(eqT(tonumber(tostring(maxint)), maxint))
342+ assert(eqT(tonumber(tostring(minint)), minint))
343+
344+ -- add 1 to last digit as a string (it cannot be 9...)
345+ local function incd (n)
346+ local s = string.format("%d", n)
347+ s = string.gsub(s, "%d$", function (d)
348+ assert(d ~= '9')
349+ return string.char(string.byte(d) + 1)
350+ end)
351+ return s
352+ end
353+
354+ -- 'tonumber' with overflow by 1
355+ assert(eqT(tonumber(incd(maxint)), maxint + 1.0))
356+ assert(eqT(tonumber(incd(minint)), minint - 1.0))
357+
358+ -- large numbers
359+ assert(eqT(tonumber("1"..string.rep("0", 30)), 1e30))
360+ assert(eqT(tonumber("-1"..string.rep("0", 30)), -1e30))
361+
362+ -- hexa format still wraps around
363+ assert(eqT(tonumber("0x1"..string.rep("0", 30)), 0))
364+
365+ -- lexer in the limits
366+ assert(minint == load("return " .. minint)())
367+ assert(eqT(maxint, load("return " .. maxint)()))
368+
369+ assert(eqT(10000000000000000000000.0, 10000000000000000000000))
370+ assert(eqT(-10000000000000000000000.0, -10000000000000000000000))
371+end
372+
373+
374+-- testing 'tonumber'
375+
376+-- 'tonumber' with numbers
377+assert(tonumber(3.4) == 3.4)
378+assert(eqT(tonumber(3), 3))
379+assert(eqT(tonumber(maxint), maxint) and eqT(tonumber(minint), minint))
380+assert(tonumber(1/0) == 1/0)
381+
382+-- 'tonumber' with strings
383+assert(tonumber("0") == 0)
384+assert(not tonumber(""))
385+assert(not tonumber(" "))
386+assert(not tonumber("-"))
387+assert(not tonumber(" -0x "))
388+assert(not tonumber{})
389+assert(tonumber'+0.01' == 1/100 and tonumber'+.01' == 0.01 and
390+ tonumber'.01' == 0.01 and tonumber'-1.' == -1 and
391+ tonumber'+1.' == 1)
392+assert(not tonumber'+ 0.01' and not tonumber'+.e1' and
393+ not tonumber'1e' and not tonumber'1.0e+' and
394+ not tonumber'.')
395+assert(tonumber('-012') == -010-2)
396+assert(tonumber('-1.2e2') == - - -120)
397+
398+assert(tonumber("0xffffffffffff") == (1 << (4*12)) - 1)
399+assert(tonumber("0x"..string.rep("f", (intbits//4))) == -1)
400+assert(tonumber("-0x"..string.rep("f", (intbits//4))) == 1)
401+
402+-- testing 'tonumber' with base
403+assert(tonumber(' 001010 ', 2) == 10)
404+assert(tonumber(' 001010 ', 10) == 001010)
405+assert(tonumber(' -1010 ', 2) == -10)
406+assert(tonumber('10', 36) == 36)
407+assert(tonumber(' -10 ', 36) == -36)
408+assert(tonumber(' +1Z ', 36) == 36 + 35)
409+assert(tonumber(' -1z ', 36) == -36 + -35)
410+assert(tonumber('-fFfa', 16) == -(10+(16*(15+(16*(15+(16*15)))))))
411+assert(tonumber(string.rep('1', (intbits - 2)), 2) + 1 == 2^(intbits - 2))
412+assert(tonumber('ffffFFFF', 16)+1 == (1 << 32))
413+assert(tonumber('0ffffFFFF', 16)+1 == (1 << 32))
414+assert(tonumber('-0ffffffFFFF', 16) - 1 == -(1 << 40))
415+for i = 2,36 do
416+ local i2 = i * i
417+ local i10 = i2 * i2 * i2 * i2 * i2 -- i^10
418+ assert(tonumber('\t10000000000\t', i) == i10)
419+end
420+
421+if not _soft then
422+ -- tests with very long numerals
423+ assert(tonumber("0x"..string.rep("f", 13)..".0") == 2.0^(4*13) - 1)
424+ assert(tonumber("0x"..string.rep("f", 150)..".0") == 2.0^(4*150) - 1)
425+ assert(tonumber("0x"..string.rep("f", 300)..".0") == 2.0^(4*300) - 1)
426+ assert(tonumber("0x"..string.rep("f", 500)..".0") == 2.0^(4*500) - 1)
427+ assert(tonumber('0x3.' .. string.rep('0', 1000)) == 3)
428+ assert(tonumber('0x' .. string.rep('0', 1000) .. 'a') == 10)
429+ assert(tonumber('0x0.' .. string.rep('0', 13).."1") == 2.0^(-4*14))
430+ assert(tonumber('0x0.' .. string.rep('0', 150).."1") == 2.0^(-4*151))
431+ assert(tonumber('0x0.' .. string.rep('0', 300).."1") == 2.0^(-4*301))
432+ assert(tonumber('0x0.' .. string.rep('0', 500).."1") == 2.0^(-4*501))
433+
434+ assert(tonumber('0xe03' .. string.rep('0', 1000) .. 'p-4000') == 3587.0)
435+ assert(tonumber('0x.' .. string.rep('0', 1000) .. '74p4004') == 0x7.4)
436+end
437+
438+-- testing 'tonumber' for invalid formats
439+
440+local function f (...)
441+ if select('#', ...) == 1 then
442+ return (...)
443+ else
444+ return "***"
445+ end
446+end
447+
448+assert(not f(tonumber('fFfa', 15)))
449+assert(not f(tonumber('099', 8)))
450+assert(not f(tonumber('1\0', 2)))
451+assert(not f(tonumber('', 8)))
452+assert(not f(tonumber(' ', 9)))
453+assert(not f(tonumber(' ', 9)))
454+assert(not f(tonumber('0xf', 10)))
455+
456+assert(not f(tonumber('inf')))
457+assert(not f(tonumber(' INF ')))
458+assert(not f(tonumber('Nan')))
459+assert(not f(tonumber('nan')))
460+
461+assert(not f(tonumber(' ')))
462+assert(not f(tonumber('')))
463+assert(not f(tonumber('1 a')))
464+assert(not f(tonumber('1 a', 2)))
465+assert(not f(tonumber('1\0')))
466+assert(not f(tonumber('1 \0')))
467+assert(not f(tonumber('1\0 ')))
468+assert(not f(tonumber('e1')))
469+assert(not f(tonumber('e 1')))
470+assert(not f(tonumber(' 3.4.5 ')))
471+
472+
473+-- testing 'tonumber' for invalid hexadecimal formats
474+
475+assert(not tonumber('0x'))
476+assert(not tonumber('x'))
477+assert(not tonumber('x3'))
478+assert(not tonumber('0x3.3.3')) -- two decimal points
479+assert(not tonumber('00x2'))
480+assert(not tonumber('0x 2'))
481+assert(not tonumber('0 x2'))
482+assert(not tonumber('23x'))
483+assert(not tonumber('- 0xaa'))
484+assert(not tonumber('-0xaaP ')) -- no exponent
485+assert(not tonumber('0x0.51p'))
486+assert(not tonumber('0x5p+-2'))
487+
488+
489+-- testing hexadecimal numerals
490+
491+assert(0x10 == 16 and 0xfff == 2^12 - 1 and 0XFB == 251)
492+assert(0x0p12 == 0 and 0x.0p-3 == 0)
493+assert(0xFFFFFFFF == (1 << 32) - 1)
494+assert(tonumber('+0x2') == 2)
495+assert(tonumber('-0xaA') == -170)
496+assert(tonumber('-0xffFFFfff') == -(1 << 32) + 1)
497+
498+-- possible confusion with decimal exponent
499+assert(0E+1 == 0 and 0xE+1 == 15 and 0xe-1 == 13)
500+
501+
502+-- floating hexas
503+
504+assert(tonumber(' 0x2.5 ') == 0x25/16)
505+assert(tonumber(' -0x2.5 ') == -0x25/16)
506+assert(tonumber(' +0x0.51p+8 ') == 0x51)
507+assert(0x.FfffFFFF == 1 - '0x.00000001')
508+assert('0xA.a' + 0 == 10 + 10/16)
509+assert(0xa.aP4 == 0XAA)
510+assert(0x4P-2 == 1)
511+assert(0x1.1 == '0x1.' + '+0x.1')
512+assert(0Xabcdef.0 == 0x.ABCDEFp+24)
513+
514+
515+assert(1.1 == 1.+.1)
516+assert(100.0 == 1E2 and .01 == 1e-2)
517+assert(1111111111 - 1111111110 == 1000.00e-03)
518+assert(1.1 == '1.'+'.1')
519+assert(tonumber'1111111111' - tonumber'1111111110' ==
520+ tonumber" +0.001e+3 \n\t")
521+
522+assert(0.1e-30 > 0.9E-31 and 0.9E30 < 0.1e31)
523+
524+assert(0.123456 > 0.123455)
525+
526+assert(tonumber('+1.23E18') == 1.23*10.0^18)
527+
528+-- testing order operators
529+assert(not(1<1) and (1<2) and not(2<1))
530+assert(not('a'<'a') and ('a'<'b') and not('b'<'a'))
531+assert((1<=1) and (1<=2) and not(2<=1))
532+assert(('a'<='a') and ('a'<='b') and not('b'<='a'))
533+assert(not(1>1) and not(1>2) and (2>1))
534+assert(not('a'>'a') and not('a'>'b') and ('b'>'a'))
535+assert((1>=1) and not(1>=2) and (2>=1))
536+assert(('a'>='a') and not('a'>='b') and ('b'>='a'))
537+assert(1.3 < 1.4 and 1.3 <= 1.4 and not (1.3 < 1.3) and 1.3 <= 1.3)
538+
539+-- testing mod operator
540+assert(eqT(-4 % 3, 2))
541+assert(eqT(4 % -3, -2))
542+assert(eqT(-4.0 % 3, 2.0))
543+assert(eqT(4 % -3.0, -2.0))
544+assert(eqT(4 % -5, -1))
545+assert(eqT(4 % -5.0, -1.0))
546+assert(eqT(4 % 5, 4))
547+assert(eqT(4 % 5.0, 4.0))
548+assert(eqT(-4 % -5, -4))
549+assert(eqT(-4 % -5.0, -4.0))
550+assert(eqT(-4 % 5, 1))
551+assert(eqT(-4 % 5.0, 1.0))
552+assert(eqT(4.25 % 4, 0.25))
553+assert(eqT(10.0 % 2, 0.0))
554+assert(eqT(-10.0 % 2, 0.0))
555+assert(eqT(-10.0 % -2, 0.0))
556+assert(math.pi - math.pi % 1 == 3)
557+assert(math.pi - math.pi % 0.001 == 3.141)
558+
559+do -- very small numbers
560+ local i, j = 0, 20000
561+ while i < j do
562+ local m = (i + j) // 2
563+ if 10^-m > 0 then
564+ i = m + 1
565+ else
566+ j = m
567+ end
568+ end
569+ -- 'i' is the smallest possible ten-exponent
570+ local b = 10^-(i - (i // 10)) -- a very small number
571+ assert(b > 0 and b * b == 0)
572+ local delta = b / 1000
573+ assert(eq((2.1 * b) % (2 * b), (0.1 * b), delta))
574+ assert(eq((-2.1 * b) % (2 * b), (2 * b) - (0.1 * b), delta))
575+ assert(eq((2.1 * b) % (-2 * b), (0.1 * b) - (2 * b), delta))
576+ assert(eq((-2.1 * b) % (-2 * b), (-0.1 * b), delta))
577+end
578+
579+
580+-- basic consistency between integer modulo and float modulo
581+for i = -10, 10 do
582+ for j = -10, 10 do
583+ if j ~= 0 then
584+ assert((i + 0.0) % j == i % j)
585+ end
586+ end
587+end
588+
589+for i = 0, 10 do
590+ for j = -10, 10 do
591+ if j ~= 0 then
592+ assert((2^i) % j == (1 << i) % j)
593+ end
594+ end
595+end
596+
597+do -- precision of module for large numbers
598+ local i = 10
599+ while (1 << i) > 0 do
600+ assert((1 << i) % 3 == i % 2 + 1)
601+ i = i + 1
602+ end
603+
604+ i = 10
605+ while 2^i < math.huge do
606+ assert(2^i % 3 == i % 2 + 1)
607+ i = i + 1
608+ end
609+end
610+
611+assert(eqT(minint % minint, 0))
612+assert(eqT(maxint % maxint, 0))
613+assert((minint + 1) % minint == minint + 1)
614+assert((maxint - 1) % maxint == maxint - 1)
615+assert(minint % maxint == maxint - 1)
616+
617+assert(minint % -1 == 0)
618+assert(minint % -2 == 0)
619+assert(maxint % -2 == -1)
620+
621+-- non-portable tests because Windows C library cannot compute
622+-- fmod(1, huge) correctly
623+if not _port then
624+ local function anan (x) assert(isNaN(x)) end -- assert Not a Number
625+ anan(0.0 % 0)
626+ anan(1.3 % 0)
627+ anan(math.huge % 1)
628+ anan(math.huge % 1e30)
629+ anan(-math.huge % 1e30)
630+ anan(-math.huge % -1e30)
631+ assert(1 % math.huge == 1)
632+ assert(1e30 % math.huge == 1e30)
633+ assert(1e30 % -math.huge == -math.huge)
634+ assert(-1 % math.huge == math.huge)
635+ assert(-1 % -math.huge == -1)
636+end
637+
638+
639+-- testing unsigned comparisons
640+assert(math.ult(3, 4))
641+assert(not math.ult(4, 4))
642+assert(math.ult(-2, -1))
643+assert(math.ult(2, -1))
644+assert(not math.ult(-2, -2))
645+assert(math.ult(maxint, minint))
646+assert(not math.ult(minint, maxint))
647+
648+
649+assert(eq(math.sin(-9.8)^2 + math.cos(-9.8)^2, 1))
650+assert(eq(math.tan(math.pi/4), 1))
651+assert(eq(math.sin(math.pi/2), 1) and eq(math.cos(math.pi/2), 0))
652+assert(eq(math.atan(1), math.pi/4) and eq(math.acos(0), math.pi/2) and
653+ eq(math.asin(1), math.pi/2))
654+assert(eq(math.deg(math.pi/2), 90) and eq(math.rad(90), math.pi/2))
655+assert(math.abs(-10.43) == 10.43)
656+assert(eqT(math.abs(minint), minint))
657+assert(eqT(math.abs(maxint), maxint))
658+assert(eqT(math.abs(-maxint), maxint))
659+assert(eq(math.atan(1,0), math.pi/2))
660+assert(math.fmod(10,3) == 1)
661+assert(eq(math.sqrt(10)^2, 10))
662+assert(eq(math.log(2, 10), math.log(2)/math.log(10)))
663+assert(eq(math.log(2, 2), 1))
664+assert(eq(math.log(9, 3), 2))
665+assert(eq(math.exp(0), 1))
666+assert(eq(math.sin(10), math.sin(10%(2*math.pi))))
667+
668+
669+assert(tonumber(' 1.3e-2 ') == 1.3e-2)
670+assert(tonumber(' -1.00000000000001 ') == -1.00000000000001)
671+
672+-- testing constant limits
673+-- 2^23 = 8388608
674+assert(8388609 + -8388609 == 0)
675+assert(8388608 + -8388608 == 0)
676+assert(8388607 + -8388607 == 0)
677+
678+
679+
680+do -- testing floor & ceil
681+ assert(eqT(math.floor(3.4), 3))
682+ assert(eqT(math.ceil(3.4), 4))
683+ assert(eqT(math.floor(-3.4), -4))
684+ assert(eqT(math.ceil(-3.4), -3))
685+ assert(eqT(math.floor(maxint), maxint))
686+ assert(eqT(math.ceil(maxint), maxint))
687+ assert(eqT(math.floor(minint), minint))
688+ assert(eqT(math.floor(minint + 0.0), minint))
689+ assert(eqT(math.ceil(minint), minint))
690+ assert(eqT(math.ceil(minint + 0.0), minint))
691+ assert(math.floor(1e50) == 1e50)
692+ assert(math.ceil(1e50) == 1e50)
693+ assert(math.floor(-1e50) == -1e50)
694+ assert(math.ceil(-1e50) == -1e50)
695+ for _, p in pairs{31,32,63,64} do
696+ assert(math.floor(2^p) == 2^p)
697+ assert(math.floor(2^p + 0.5) == 2^p)
698+ assert(math.ceil(2^p) == 2^p)
699+ assert(math.ceil(2^p - 0.5) == 2^p)
700+ end
701+ checkerror("number expected", math.floor, {})
702+ checkerror("number expected", math.ceil, print)
703+ assert(eqT(math.tointeger(minint), minint))
704+ assert(eqT(math.tointeger(minint .. ""), minint))
705+ assert(eqT(math.tointeger(maxint), maxint))
706+ assert(eqT(math.tointeger(maxint .. ""), maxint))
707+ assert(eqT(math.tointeger(minint + 0.0), minint))
708+ assert(not math.tointeger(0.0 - minint))
709+ assert(not math.tointeger(math.pi))
710+ assert(not math.tointeger(-math.pi))
711+ assert(math.floor(math.huge) == math.huge)
712+ assert(math.ceil(math.huge) == math.huge)
713+ assert(not math.tointeger(math.huge))
714+ assert(math.floor(-math.huge) == -math.huge)
715+ assert(math.ceil(-math.huge) == -math.huge)
716+ assert(not math.tointeger(-math.huge))
717+ assert(math.tointeger("34.0") == 34)
718+ assert(not math.tointeger("34.3"))
719+ assert(not math.tointeger({}))
720+ assert(not math.tointeger(0/0)) -- NaN
721+end
722+
723+
724+-- testing fmod for integers
725+for i = -6, 6 do
726+ for j = -6, 6 do
727+ if j ~= 0 then
728+ local mi = math.fmod(i, j)
729+ local mf = math.fmod(i + 0.0, j)
730+ assert(mi == mf)
731+ assert(math.type(mi) == 'integer' and math.type(mf) == 'float')
732+ if (i >= 0 and j >= 0) or (i <= 0 and j <= 0) or mi == 0 then
733+ assert(eqT(mi, i % j))
734+ end
735+ end
736+ end
737+end
738+assert(eqT(math.fmod(minint, minint), 0))
739+assert(eqT(math.fmod(maxint, maxint), 0))
740+assert(eqT(math.fmod(minint + 1, minint), minint + 1))
741+assert(eqT(math.fmod(maxint - 1, maxint), maxint - 1))
742+
743+checkerror("zero", math.fmod, 3, 0)
744+
745+
746+do -- testing max/min
747+ checkerror("value expected", math.max)
748+ checkerror("value expected", math.min)
749+ assert(eqT(math.max(3), 3))
750+ assert(eqT(math.max(3, 5, 9, 1), 9))
751+ assert(math.max(maxint, 10e60) == 10e60)
752+ assert(eqT(math.max(minint, minint + 1), minint + 1))
753+ assert(eqT(math.min(3), 3))
754+ assert(eqT(math.min(3, 5, 9, 1), 1))
755+ assert(math.min(3.2, 5.9, -9.2, 1.1) == -9.2)
756+ assert(math.min(1.9, 1.7, 1.72) == 1.7)
757+ assert(math.min(-10e60, minint) == -10e60)
758+ assert(eqT(math.min(maxint, maxint - 1), maxint - 1))
759+ assert(eqT(math.min(maxint - 2, maxint, maxint - 1), maxint - 2))
760+end
761+-- testing implicit conversions
762+
763+local a,b = '10', '20'
764+assert(a*b == 200 and a+b == 30 and a-b == -10 and a/b == 0.5 and -b == -20)
765+assert(a == '10' and b == '20')
766+
767+
768+do
769+ print("testing -0 and NaN")
770+ local mz <const> = -0.0
771+ local z <const> = 0.0
772+ assert(mz == z)
773+ assert(1/mz < 0 and 0 < 1/z)
774+ local a = {[mz] = 1}
775+ assert(a[z] == 1 and a[mz] == 1)
776+ a[z] = 2
777+ assert(a[z] == 2 and a[mz] == 2)
778+ local inf = math.huge * 2 + 1
779+ local mz <const> = -1/inf
780+ local z <const> = 1/inf
781+ assert(mz == z)
782+ assert(1/mz < 0 and 0 < 1/z)
783+ local NaN <const> = inf - inf
784+ assert(NaN ~= NaN)
785+ assert(not (NaN < NaN))
786+ assert(not (NaN <= NaN))
787+ assert(not (NaN > NaN))
788+ assert(not (NaN >= NaN))
789+ assert(not (0 < NaN) and not (NaN < 0))
790+ local NaN1 <const> = 0/0
791+ assert(NaN ~= NaN1 and not (NaN <= NaN1) and not (NaN1 <= NaN))
792+ local a = {}
793+ assert(not pcall(rawset, a, NaN, 1))
794+ assert(a[NaN] == undef)
795+ a[1] = 1
796+ assert(not pcall(rawset, a, NaN, 1))
797+ assert(a[NaN] == undef)
798+ -- strings with same binary representation as 0.0 (might create problems
799+ -- for constant manipulation in the pre-compiler)
800+ local a1, a2, a3, a4, a5 = 0, 0, "\0\0\0\0\0\0\0\0", 0, "\0\0\0\0\0\0\0\0"
801+ assert(a1 == a2 and a2 == a4 and a1 ~= a3)
802+ assert(a3 == a5)
803+end
804+
805+
806+print("testing 'math.random'")
807+
808+local random, max, min = math.random, math.max, math.min
809+
810+local function testnear (val, ref, tol)
811+ return (math.abs(val - ref) < ref * tol)
812+end
813+
814+
815+-- low-level!! For the current implementation of random in Lua,
816+-- the first call after seed 1007 should return 0x7a7040a5a323c9d6
817+do
818+ -- all computations should work with 32-bit integers
819+ local h <const> = 0x7a7040a5 -- higher half
820+ local l <const> = 0xa323c9d6 -- lower half
821+
822+ math.randomseed(1007)
823+ -- get the low 'intbits' of the 64-bit expected result
824+ local res = (h << 32 | l) & ~(~0 << intbits)
825+ assert(random(0) == res)
826+
827+ math.randomseed(1007, 0)
828+ -- using higher bits to generate random floats; (the '% 2^32' converts
829+ -- 32-bit integers to floats as unsigned)
830+ local res
831+ if floatbits <= 32 then
832+ -- get all bits from the higher half
833+ res = (h >> (32 - floatbits)) % 2^32
834+ else
835+ -- get 32 bits from the higher half and the rest from the lower half
836+ res = (h % 2^32) * 2^(floatbits - 32) + ((l >> (64 - floatbits)) % 2^32)
837+ end
838+ local rand = random()
839+ assert(eq(rand, 0x0.7a7040a5a323c9d6, 2^-floatbits))
840+ assert(rand * 2^floatbits == res)
841+end
842+
843+do
844+ -- testing return of 'randomseed'
845+ local x, y = math.randomseed()
846+ local res = math.random(0)
847+ x, y = math.randomseed(x, y) -- should repeat the state
848+ assert(math.random(0) == res)
849+ math.randomseed(x, y) -- again should repeat the state
850+ assert(math.random(0) == res)
851+ -- keep the random seed for following tests
852+ print(string.format("random seeds: %d, %d", x, y))
853+end
854+
855+do -- test random for floats
856+ local randbits = math.min(floatbits, 64) -- at most 64 random bits
857+ local mult = 2^randbits -- to make random float into an integral
858+ local counts = {} -- counts for bits
859+ for i = 1, randbits do counts[i] = 0 end
860+ local up = -math.huge
861+ local low = math.huge
862+ local rounds = 100 * randbits -- 100 times for each bit
863+ local totalrounds = 0
864+ ::doagain:: -- will repeat test until we get good statistics
865+ for i = 0, rounds do
866+ local t = random()
867+ assert(0 <= t and t < 1)
868+ up = max(up, t)
869+ low = min(low, t)
870+ assert(t * mult % 1 == 0) -- no extra bits
871+ local bit = i % randbits -- bit to be tested
872+ if (t * 2^bit) % 1 >= 0.5 then -- is bit set?
873+ counts[bit + 1] = counts[bit + 1] + 1 -- increment its count
874+ end
875+ end
876+ totalrounds = totalrounds + rounds
877+ if not (eq(up, 1, 0.001) and eq(low, 0, 0.001)) then
878+ goto doagain
879+ end
880+ -- all bit counts should be near 50%
881+ local expected = (totalrounds / randbits / 2)
882+ for i = 1, randbits do
883+ if not testnear(counts[i], expected, 0.10) then
884+ goto doagain
885+ end
886+ end
887+ print(string.format("float random range in %d calls: [%f, %f]",
888+ totalrounds, low, up))
889+end
890+
891+
892+do -- test random for full integers
893+ local up = 0
894+ local low = 0
895+ local counts = {} -- counts for bits
896+ for i = 1, intbits do counts[i] = 0 end
897+ local rounds = 100 * intbits -- 100 times for each bit
898+ local totalrounds = 0
899+ ::doagain:: -- will repeat test until we get good statistics
900+ for i = 0, rounds do
901+ local t = random(0)
902+ up = max(up, t)
903+ low = min(low, t)
904+ local bit = i % intbits -- bit to be tested
905+ -- increment its count if it is set
906+ counts[bit + 1] = counts[bit + 1] + ((t >> bit) & 1)
907+ end
908+ totalrounds = totalrounds + rounds
909+ local lim = maxint >> 10
910+ if not (maxint - up < lim and low - minint < lim) then
911+ goto doagain
912+ end
913+ -- all bit counts should be near 50%
914+ local expected = (totalrounds / intbits / 2)
915+ for i = 1, intbits do
916+ if not testnear(counts[i], expected, 0.10) then
917+ goto doagain
918+ end
919+ end
920+ print(string.format(
921+ "integer random range in %d calls: [minint + %.0fppm, maxint - %.0fppm]",
922+ totalrounds, (minint - low) / minint * 1e6,
923+ (maxint - up) / maxint * 1e6))
924+end
925+
926+do
927+ -- test distribution for a dice
928+ local count = {0, 0, 0, 0, 0, 0}
929+ local rep = 200
930+ local totalrep = 0
931+ ::doagain::
932+ for i = 1, rep * 6 do
933+ local r = random(6)
934+ count[r] = count[r] + 1
935+ end
936+ totalrep = totalrep + rep
937+ for i = 1, 6 do
938+ if not testnear(count[i], totalrep, 0.05) then
939+ goto doagain
940+ end
941+ end
942+end
943+
944+do
945+ local function aux (x1, x2) -- test random for small intervals
946+ local mark = {}; local count = 0 -- to check that all values appeared
947+ while true do
948+ local t = random(x1, x2)
949+ assert(x1 <= t and t <= x2)
950+ if not mark[t] then -- new value
951+ mark[t] = true
952+ count = count + 1
953+ if count == x2 - x1 + 1 then -- all values appeared; OK
954+ goto ok
955+ end
956+ end
957+ end
958+ ::ok::
959+ end
960+
961+ aux(-10,0)
962+ aux(1, 6)
963+ aux(1, 2)
964+ aux(1, 13)
965+ aux(1, 31)
966+ aux(1, 32)
967+ aux(1, 33)
968+ aux(-10, 10)
969+ aux(-10,-10) -- unit set
970+ aux(minint, minint) -- unit set
971+ aux(maxint, maxint) -- unit set
972+ aux(minint, minint + 9)
973+ aux(maxint - 3, maxint)
974+end
975+
976+do
977+ local function aux(p1, p2) -- test random for large intervals
978+ local max = minint
979+ local min = maxint
980+ local n = 100
981+ local mark = {}; local count = 0 -- to count how many different values
982+ ::doagain::
983+ for _ = 1, n do
984+ local t = random(p1, p2)
985+ if not mark[t] then -- new value
986+ assert(p1 <= t and t <= p2)
987+ max = math.max(max, t)
988+ min = math.min(min, t)
989+ mark[t] = true
990+ count = count + 1
991+ end
992+ end
993+ -- at least 80% of values are different
994+ if not (count >= n * 0.8) then
995+ goto doagain
996+ end
997+ -- min and max not too far from formal min and max
998+ local diff = (p2 - p1) >> 4
999+ if not (min < p1 + diff and max > p2 - diff) then
1000+ goto doagain
1001+ end
1002+ end
1003+ aux(0, maxint)
1004+ aux(1, maxint)
1005+ aux(3, maxint // 3)
1006+ aux(minint, -1)
1007+ aux(minint // 2, maxint // 2)
1008+ aux(minint, maxint)
1009+ aux(minint + 1, maxint)
1010+ aux(minint, maxint - 1)
1011+ aux(0, 1 << (intbits - 5))
1012+end
1013+
1014+
1015+assert(not pcall(random, 1, 2, 3)) -- too many arguments
1016+
1017+-- empty interval
1018+assert(not pcall(random, minint + 1, minint))
1019+assert(not pcall(random, maxint, maxint - 1))
1020+assert(not pcall(random, maxint, minint))
1021+
1022+
1023+
1024+print('OK')
zod/luau/tests/nextvar.luanew file+825
@@ -0,0 +1,825 @@
1+-- $Id: testes/nextvar.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print('testing tables, next, and for')
5+
6+local function checkerror (msg, f, ...)
7+ local s, err = pcall(f, ...)
8+ assert(not s and string.find(err, msg))
9+end
10+
11+
12+local function check (t, na, nh)
13+ if not T then return end
14+ local a, h = T.querytab(t)
15+ if a ~= na or h ~= nh then
16+ print(na, nh, a, h)
17+ assert(nil)
18+ end
19+end
20+
21+
22+local a = {}
23+
24+-- make sure table has lots of space in hash part
25+for i=1,100 do a[i.."+"] = true end
26+for i=1,100 do a[i.."+"] = undef end
27+-- fill hash part with numeric indices testing size operator
28+for i=1,100 do
29+ a[i] = true
30+ assert(#a == i)
31+end
32+
33+
34+do -- rehash moving elements from array to hash
35+ local a = {}
36+ for i = 1, 100 do a[i] = i end
37+ check(a, 128, 0)
38+
39+ for i = 5, 95 do a[i] = nil end
40+ check(a, 128, 0)
41+
42+ a.x = 1 -- force a re-hash
43+ check(a, 4, 8)
44+
45+ for i = 1, 4 do assert(a[i] == i) end
46+ for i = 5, 95 do assert(a[i] == nil) end
47+ for i = 96, 100 do assert(a[i] == i) end
48+ assert(a.x == 1)
49+end
50+
51+
52+-- testing ipairs
53+local x = 0
54+for k,v in ipairs{10,20,30;x=12} do
55+ x = x + 1
56+ assert(k == x and v == x * 10)
57+end
58+
59+for _ in ipairs{x=12, y=24} do assert(nil) end
60+
61+-- test for 'false' x ipair
62+x = false
63+local i = 0
64+for k,v in ipairs{true,false,true,false} do
65+ i = i + 1
66+ x = not x
67+ assert(x == v)
68+end
69+assert(i == 4)
70+
71+-- iterator function is always the same
72+assert(type(ipairs{}) == 'function' and ipairs{} == ipairs{})
73+
74+
75+do -- overflow (must wrap-around)
76+ local f = ipairs{}
77+ local k, v = f({[math.mininteger] = 10}, math.maxinteger)
78+ assert(k == math.mininteger and v == 10)
79+ k, v = f({[math.mininteger] = 10}, k)
80+ assert(k == nil)
81+end
82+
83+if not T then
84+ (Message or print)
85+ ('\n >>> testC not active: skipping tests for table sizes <<<\n')
86+else --[
87+-- testing table sizes
88+
89+
90+local function mp2 (n) -- minimum power of 2 >= n
91+ local mp = 2^math.ceil(math.log(n, 2))
92+ assert(n == 0 or (mp/2 < n and n <= mp))
93+ return mp
94+end
95+
96+
97+-- testing C library sizes
98+do
99+ local s = 0
100+ for _ in pairs(math) do s = s + 1 end
101+ check(math, 0, mp2(s))
102+end
103+
104+
105+-- testing constructor sizes
106+local sizes = {0, 1, 2, 3, 4, 5, 7, 8, 9, 15, 16, 17,
107+ 30, 31, 32, 33, 34, 254, 255, 256, 500, 1000}
108+
109+for _, sa in ipairs(sizes) do -- 'sa' is size of the array part
110+ local arr = {"return {"}
111+ for i = 1, sa do arr[1 + i] = "1," end -- build array part
112+ for _, sh in ipairs(sizes) do -- 'sh' is size of the hash part
113+ for j = 1, sh do -- build hash part
114+ arr[1 + sa + j] = string.format('k%x=%d,', j, j)
115+ end
116+ arr[1 + sa + sh + 1] = "}"
117+ local prog = table.concat(arr)
118+ local f = assert(load(prog))
119+ collectgarbage("stop")
120+ f() -- call once to ensure stack space
121+ -- make sure table is not resized after being created
122+ if sa == 0 or sh == 0 then
123+ T.alloccount(2); -- header + array or hash part
124+ else
125+ T.alloccount(3); -- header + array part + hash part
126+ end
127+ local t = f()
128+ T.alloccount();
129+ collectgarbage("restart")
130+ assert(#t == sa)
131+ check(t, sa, mp2(sh))
132+ end
133+end
134+
135+
136+-- tests with unknown number of elements
137+local a = {}
138+for i=1,sizes[#sizes] do a[i] = i end -- build auxiliary table
139+for k in ipairs(sizes) do
140+ local t = {table.unpack(a,1,k)}
141+ assert(#t == k)
142+ check(t, k, 0)
143+ t = {1,2,3,table.unpack(a,1,k)}
144+ check(t, k+3, 0)
145+ assert(#t == k + 3)
146+end
147+
148+
149+-- testing tables dynamically built
150+local lim = 130
151+local a = {}; a[2] = 1; check(a, 0, 1)
152+a = {}; a[0] = 1; check(a, 0, 1); a[2] = 1; check(a, 0, 2)
153+a = {}; a[0] = 1; a[1] = 1; check(a, 1, 1)
154+a = {}
155+for i = 1,lim do
156+ a[i] = 1
157+ assert(#a == i)
158+ check(a, mp2(i), 0)
159+end
160+
161+a = {}
162+for i = 1,lim do
163+ a['a'..i] = 1
164+ assert(#a == 0)
165+ check(a, 0, mp2(i))
166+end
167+
168+a = {}
169+for i=1,16 do a[i] = i end
170+check(a, 16, 0)
171+do
172+ for i=1,11 do a[i] = undef end
173+ for i=30,50 do a[i] = true; a[i] = undef end -- force a rehash (?)
174+ check(a, 0, 8) -- 5 elements in the table
175+ a[10] = 1
176+ for i=30,50 do a[i] = true; a[i] = undef end -- force a rehash (?)
177+ check(a, 0, 8) -- only 6 elements in the table
178+ for i=1,14 do a[i] = true; a[i] = undef end
179+ for i=18,50 do a[i] = true; a[i] = undef end -- force a rehash (?)
180+ check(a, 0, 4) -- only 2 elements ([15] and [16])
181+end
182+
183+-- reverse filling
184+for i=1,lim do
185+ local a = {}
186+ for i=i,1,-1 do a[i] = i end -- fill in reverse
187+ check(a, mp2(i), 0)
188+end
189+
190+-- size tests for vararg
191+lim = 35
192+local function foo (n, ...)
193+ local arg = {...}
194+ check(arg, n, 0)
195+ assert(select('#', ...) == n)
196+ arg[n+1] = true
197+ check(arg, mp2(n+1), 0)
198+ arg.x = true
199+ check(arg, mp2(n+1), 1)
200+end
201+local a = {}
202+for i=1,lim do a[i] = true; foo(i, table.unpack(a)) end
203+
204+
205+-- Table length with limit smaller than maximum value at array
206+local a = {}
207+for i = 1,64 do a[i] = true end -- make its array size 64
208+for i = 1,64 do a[i] = nil end -- erase all elements
209+assert(T.querytab(a) == 64) -- array part has 64 elements
210+a[32] = true; a[48] = true; -- binary search will find these ones
211+a[51] = true -- binary search will miss this one
212+assert(#a == 48) -- this will set the limit
213+assert(select(4, T.querytab(a)) == 48) -- this is the limit now
214+a[50] = true -- this will set a new limit
215+assert(select(4, T.querytab(a)) == 50) -- this is the limit now
216+-- but the size is larger (and still inside the array part)
217+assert(#a == 51)
218+
219+end --]
220+
221+
222+-- test size operation on tables with nils
223+assert(#{} == 0)
224+assert(#{nil} == 0)
225+assert(#{nil, nil} == 0)
226+assert(#{nil, nil, nil} == 0)
227+assert(#{nil, nil, nil, nil} == 0)
228+assert(#{1, 2, 3, nil, nil} == 3)
229+print'+'
230+
231+
232+local nofind = {}
233+
234+a,b,c = 1,2,3
235+a,b,c = nil
236+
237+
238+-- next uses always the same iteraction function
239+assert(next{} == next{})
240+
241+local function find (name)
242+ local n,v
243+ while 1 do
244+ n,v = next(_G, n)
245+ if not n then return nofind end
246+ assert(_G[n] ~= undef)
247+ if n == name then return v end
248+ end
249+end
250+
251+local function find1 (name)
252+ for n,v in pairs(_G) do
253+ if n==name then return v end
254+ end
255+ return nil -- not found
256+end
257+
258+
259+assert(print==find("print") and print == find1("print"))
260+assert(_G["print"]==find("print"))
261+assert(assert==find1("assert"))
262+assert(nofind==find("return"))
263+assert(not find1("return"))
264+_G["ret" .. "urn"] = undef
265+assert(nofind==find("return"))
266+_G["xxx"] = 1
267+assert(xxx==find("xxx"))
268+
269+-- invalid key to 'next'
270+checkerror("invalid key", next, {10,20}, 3)
271+
272+-- both 'pairs' and 'ipairs' need an argument
273+checkerror("bad argument", pairs)
274+checkerror("bad argument", ipairs)
275+
276+print('+')
277+
278+a = {}
279+for i=0,10000 do
280+ if math.fmod(i,10) ~= 0 then
281+ a['x'..i] = i
282+ end
283+end
284+
285+n = {n=0}
286+for i,v in pairs(a) do
287+ n.n = n.n+1
288+ assert(i and v and a[i] == v)
289+end
290+assert(n.n == 9000)
291+a = nil
292+
293+do -- clear global table
294+ local a = {}
295+ for n,v in pairs(_G) do a[n]=v end
296+ for n,v in pairs(a) do
297+ if not package.loaded[n] and type(v) ~= "function" and
298+ not string.find(n, "^[%u_]") then
299+ _G[n] = undef
300+ end
301+ collectgarbage()
302+ end
303+end
304+
305+
306+--
307+
308+local function checknext (a)
309+ local b = {}
310+ do local k,v = next(a); while k do b[k] = v; k,v = next(a,k) end end
311+ for k,v in pairs(b) do assert(a[k] == v) end
312+ for k,v in pairs(a) do assert(b[k] == v) end
313+end
314+
315+checknext{1,x=1,y=2,z=3}
316+checknext{1,2,x=1,y=2,z=3}
317+checknext{1,2,3,x=1,y=2,z=3}
318+checknext{1,2,3,4,x=1,y=2,z=3}
319+checknext{1,2,3,4,5,x=1,y=2,z=3}
320+
321+assert(#{} == 0)
322+assert(#{[-1] = 2} == 0)
323+for i=0,40 do
324+ local a = {}
325+ for j=1,i do a[j]=j end
326+ assert(#a == i)
327+end
328+
329+-- 'maxn' is now deprecated, but it is easily defined in Lua
330+function table.maxn (t)
331+ local max = 0
332+ for k in pairs(t) do
333+ max = (type(k) == 'number') and math.max(max, k) or max
334+ end
335+ return max
336+end
337+
338+assert(table.maxn{} == 0)
339+assert(table.maxn{["1000"] = true} == 0)
340+assert(table.maxn{["1000"] = true, [24.5] = 3} == 24.5)
341+assert(table.maxn{[1000] = true} == 1000)
342+assert(table.maxn{[10] = true, [100*math.pi] = print} == 100*math.pi)
343+
344+table.maxn = nil
345+
346+-- int overflow
347+a = {}
348+for i=0,50 do a[2^i] = true end
349+assert(a[#a])
350+
351+print('+')
352+
353+
354+do -- testing 'next' with all kinds of keys
355+ local a = {
356+ [1] = 1, -- integer
357+ [1.1] = 2, -- float
358+ ['x'] = 3, -- short string
359+ [string.rep('x', 1000)] = 4, -- long string
360+ [print] = 5, -- C function
361+ [checkerror] = 6, -- Lua function
362+ [coroutine.running()] = 7, -- thread
363+ [true] = 8, -- boolean
364+ [io.stdin] = 9, -- userdata
365+ [{}] = 10, -- table
366+ }
367+ local b = {}; for i = 1, 10 do b[i] = true end
368+ for k, v in pairs(a) do
369+ assert(b[v]); b[v] = undef
370+ end
371+ assert(next(b) == nil) -- 'b' now is empty
372+end
373+
374+
375+-- erasing values
376+local t = {[{1}] = 1, [{2}] = 2, [string.rep("x ", 4)] = 3,
377+ [100.3] = 4, [4] = 5}
378+
379+local n = 0
380+for k, v in pairs( t ) do
381+ n = n+1
382+ assert(t[k] == v)
383+ t[k] = undef
384+ collectgarbage()
385+ assert(t[k] == undef)
386+end
387+assert(n == 5)
388+
389+
390+do
391+ print("testing next x GC of deleted keys")
392+ -- bug in 5.4.1
393+ local co = coroutine.wrap(function (t)
394+ for k, v in pairs(t) do
395+ local k1 = next(t) -- all previous keys were deleted
396+ assert(k == k1) -- current key is the first in the table
397+ t[k] = nil
398+ local expected = (type(k) == "table" and k[1] or
399+ type(k) == "function" and k() or
400+ string.sub(k, 1, 1))
401+ assert(expected == v)
402+ coroutine.yield(v)
403+ end
404+ end)
405+ local t = {}
406+ t[{1}] = 1 -- add several unanchored, collectable keys
407+ t[{2}] = 2
408+ t[string.rep("a", 50)] = "a" -- long string
409+ t[string.rep("b", 50)] = "b"
410+ t[{3}] = 3
411+ t[string.rep("c", 10)] = "c" -- short string
412+ t[function () return 10 end] = 10
413+ local count = 7
414+ while co(t) do
415+ collectgarbage("collect") -- collect dead keys
416+ count = count - 1
417+ end
418+ assert(count == 0 and next(t) == nil) -- traversed the whole table
419+end
420+
421+
422+local function test (a)
423+ assert(not pcall(table.insert, a, 2, 20));
424+ table.insert(a, 10); table.insert(a, 2, 20);
425+ table.insert(a, 1, -1); table.insert(a, 40);
426+ table.insert(a, #a+1, 50)
427+ table.insert(a, 2, -2)
428+ assert(a[2] ~= undef)
429+ assert(a["2"] == undef)
430+ assert(not pcall(table.insert, a, 0, 20));
431+ assert(not pcall(table.insert, a, #a + 2, 20));
432+ assert(table.remove(a,1) == -1)
433+ assert(table.remove(a,1) == -2)
434+ assert(table.remove(a,1) == 10)
435+ assert(table.remove(a,1) == 20)
436+ assert(table.remove(a,1) == 40)
437+ assert(table.remove(a,1) == 50)
438+ assert(table.remove(a,1) == nil)
439+ assert(table.remove(a) == nil)
440+ assert(table.remove(a, #a) == nil)
441+end
442+
443+a = {n=0, [-7] = "ban"}
444+test(a)
445+assert(a.n == 0 and a[-7] == "ban")
446+
447+a = {[-7] = "ban"};
448+test(a)
449+assert(a.n == nil and #a == 0 and a[-7] == "ban")
450+
451+a = {[-1] = "ban"}
452+test(a)
453+assert(#a == 0 and table.remove(a) == nil and a[-1] == "ban")
454+
455+a = {[0] = "ban"}
456+assert(#a == 0 and table.remove(a) == "ban" and a[0] == undef)
457+
458+table.insert(a, 1, 10); table.insert(a, 1, 20); table.insert(a, 1, -1)
459+assert(table.remove(a) == 10)
460+assert(table.remove(a) == 20)
461+assert(table.remove(a) == -1)
462+assert(table.remove(a) == nil)
463+
464+a = {'c', 'd'}
465+table.insert(a, 3, 'a')
466+table.insert(a, 'b')
467+assert(table.remove(a, 1) == 'c')
468+assert(table.remove(a, 1) == 'd')
469+assert(table.remove(a, 1) == 'a')
470+assert(table.remove(a, 1) == 'b')
471+assert(table.remove(a, 1) == nil)
472+assert(#a == 0 and a.n == nil)
473+
474+a = {10,20,30,40}
475+assert(table.remove(a, #a + 1) == nil)
476+assert(not pcall(table.remove, a, 0))
477+assert(a[#a] == 40)
478+assert(table.remove(a, #a) == 40)
479+assert(a[#a] == 30)
480+assert(table.remove(a, 2) == 20)
481+assert(a[#a] == 30 and #a == 2)
482+
483+do -- testing table library with metamethods
484+ local function test (proxy, t)
485+ for i = 1, 10 do
486+ table.insert(proxy, 1, i)
487+ end
488+ assert(#proxy == 10 and #t == 10 and proxy[1] ~= undef)
489+ for i = 1, 10 do
490+ assert(t[i] == 11 - i)
491+ end
492+ table.sort(proxy)
493+ for i = 1, 10 do
494+ assert(t[i] == i and proxy[i] == i)
495+ end
496+ assert(table.concat(proxy, ",") == "1,2,3,4,5,6,7,8,9,10")
497+ for i = 1, 8 do
498+ assert(table.remove(proxy, 1) == i)
499+ end
500+ assert(#proxy == 2 and #t == 2)
501+ local a, b, c = table.unpack(proxy)
502+ assert(a == 9 and b == 10 and c == nil)
503+ end
504+
505+ -- all virtual
506+ local t = {}
507+ local proxy = setmetatable({}, {
508+ __len = function () return #t end,
509+ __index = t,
510+ __newindex = t,
511+ })
512+ test(proxy, t)
513+
514+ -- only __newindex
515+ local count = 0
516+ t = setmetatable({}, {
517+ __newindex = function (t,k,v) count = count + 1; rawset(t,k,v) end})
518+ test(t, t)
519+ assert(count == 10) -- after first 10, all other sets are not new
520+
521+ -- no __newindex
522+ t = setmetatable({}, {
523+ __index = function (_,k) return k + 1 end,
524+ __len = function (_) return 5 end})
525+ assert(table.concat(t, ";") == "2;3;4;5;6")
526+
527+end
528+
529+
530+do -- testing overflow in table.insert (must wrap-around)
531+
532+ local t = setmetatable({},
533+ {__len = function () return math.maxinteger end})
534+ table.insert(t, 20)
535+ local k, v = next(t)
536+ assert(k == math.mininteger and v == 20)
537+end
538+
539+if not T then
540+ (Message or print)
541+ ('\n >>> testC not active: skipping tests for table library on non-tables <<<\n')
542+else --[
543+ local debug = require'debug'
544+ local tab = {10, 20, 30}
545+ local mt = {}
546+ local u = T.newuserdata(0)
547+ checkerror("table expected", table.insert, u, 40)
548+ checkerror("table expected", table.remove, u)
549+ debug.setmetatable(u, mt)
550+ checkerror("table expected", table.insert, u, 40)
551+ checkerror("table expected", table.remove, u)
552+ mt.__index = tab
553+ checkerror("table expected", table.insert, u, 40)
554+ checkerror("table expected", table.remove, u)
555+ mt.__newindex = tab
556+ checkerror("table expected", table.insert, u, 40)
557+ checkerror("table expected", table.remove, u)
558+ mt.__len = function () return #tab end
559+ table.insert(u, 40)
560+ assert(#u == 4 and #tab == 4 and u[4] == 40 and tab[4] == 40)
561+ assert(table.remove(u) == 40)
562+ table.insert(u, 1, 50)
563+ assert(#u == 4 and #tab == 4 and u[4] == 30 and tab[1] == 50)
564+
565+ mt.__newindex = nil
566+ mt.__len = nil
567+ local tab2 = {}
568+ local u2 = T.newuserdata(0)
569+ debug.setmetatable(u2, {__newindex = function (_, k, v) tab2[k] = v end})
570+ table.move(u, 1, 4, 1, u2)
571+ assert(#tab2 == 4 and tab2[1] == tab[1] and tab2[4] == tab[4])
572+
573+end -- ]
574+
575+print('+')
576+
577+a = {}
578+for i=1,1000 do
579+ a[i] = i; a[i - 1] = undef
580+end
581+assert(next(a,nil) == 1000 and next(a,1000) == nil)
582+
583+assert(next({}) == nil)
584+assert(next({}, nil) == nil)
585+
586+for a,b in pairs{} do error"not here" end
587+for i=1,0 do error'not here' end
588+for i=0,1,-1 do error'not here' end
589+a = nil; for i=1,1 do assert(not a); a=1 end; assert(a)
590+a = nil; for i=1,1,-1 do assert(not a); a=1 end; assert(a)
591+
592+do
593+ print("testing floats in numeric for")
594+ local a
595+ -- integer count
596+ a = 0; for i=1, 1, 1 do a=a+1 end; assert(a==1)
597+ a = 0; for i=10000, 1e4, -1 do a=a+1 end; assert(a==1)
598+ a = 0; for i=1, 0.99999, 1 do a=a+1 end; assert(a==0)
599+ a = 0; for i=9999, 1e4, -1 do a=a+1 end; assert(a==0)
600+ a = 0; for i=1, 0.99999, -1 do a=a+1 end; assert(a==1)
601+
602+ -- float count
603+ a = 0; for i=0, 0.999999999, 0.1 do a=a+1 end; assert(a==10)
604+ a = 0; for i=1.0, 1, 1 do a=a+1 end; assert(a==1)
605+ a = 0; for i=-1.5, -1.5, 1 do a=a+1 end; assert(a==1)
606+ a = 0; for i=1e6, 1e6, -1 do a=a+1 end; assert(a==1)
607+ a = 0; for i=1.0, 0.99999, 1 do a=a+1 end; assert(a==0)
608+ a = 0; for i=99999, 1e5, -1.0 do a=a+1 end; assert(a==0)
609+ a = 0; for i=1.0, 0.99999, -1 do a=a+1 end; assert(a==1)
610+end
611+
612+do -- changing the control variable
613+ local a
614+ a = 0; for i = 1, 10 do a = a + 1; i = "x" end; assert(a == 10)
615+ a = 0; for i = 10.0, 1, -1 do a = a + 1; i = "x" end; assert(a == 10)
616+end
617+
618+-- conversion
619+a = 0; for i="10","1","-2" do a=a+1 end; assert(a==5)
620+
621+do -- checking types
622+ local c
623+ local function checkfloat (i)
624+ assert(math.type(i) == "float")
625+ c = c + 1
626+ end
627+
628+ c = 0; for i = 1.0, 10 do checkfloat(i) end
629+ assert(c == 10)
630+
631+ c = 0; for i = -1, -10, -1.0 do checkfloat(i) end
632+ assert(c == 10)
633+
634+ local function checkint (i)
635+ assert(math.type(i) == "integer")
636+ c = c + 1
637+ end
638+
639+ local m = math.maxinteger
640+ c = 0; for i = m, m - 10, -1 do checkint(i) end
641+ assert(c == 11)
642+
643+ c = 0; for i = 1, 10.9 do checkint(i) end
644+ assert(c == 10)
645+
646+ c = 0; for i = 10, 0.001, -1 do checkint(i) end
647+ assert(c == 10)
648+
649+ c = 0; for i = 1, "10.8" do checkint(i) end
650+ assert(c == 10)
651+
652+ c = 0; for i = 9, "3.4", -1 do checkint(i) end
653+ assert(c == 6)
654+
655+ c = 0; for i = 0, " -3.4 ", -1 do checkint(i) end
656+ assert(c == 4)
657+
658+ c = 0; for i = 100, "96.3", -2 do checkint(i) end
659+ assert(c == 2)
660+
661+ c = 0; for i = 1, math.huge do if i > 10 then break end; checkint(i) end
662+ assert(c == 10)
663+
664+ c = 0; for i = -1, -math.huge, -1 do
665+ if i < -10 then break end; checkint(i)
666+ end
667+ assert(c == 10)
668+
669+
670+ for i = math.mininteger, -10e100 do assert(false) end
671+ for i = math.maxinteger, 10e100, -1 do assert(false) end
672+
673+end
674+
675+
676+do -- testing other strange cases for numeric 'for'
677+
678+ local function checkfor (from, to, step, t)
679+ local c = 0
680+ for i = from, to, step do
681+ c = c + 1
682+ assert(i == t[c])
683+ end
684+ assert(c == #t)
685+ end
686+
687+ local maxi = math.maxinteger
688+ local mini = math.mininteger
689+
690+ checkfor(mini, maxi, maxi, {mini, -1, maxi - 1})
691+
692+ checkfor(mini, math.huge, maxi, {mini, -1, maxi - 1})
693+
694+ checkfor(maxi, mini, mini, {maxi, -1})
695+
696+ checkfor(maxi, mini, -maxi, {maxi, 0, -maxi})
697+
698+ checkfor(maxi, -math.huge, mini, {maxi, -1})
699+
700+ checkfor(maxi, mini, 1, {})
701+ checkfor(mini, maxi, -1, {})
702+
703+ checkfor(maxi - 6, maxi, 3, {maxi - 6, maxi - 3, maxi})
704+ checkfor(mini + 4, mini, -2, {mini + 4, mini + 2, mini})
705+
706+ local step = maxi // 10
707+ local c = mini
708+ for i = mini, maxi, step do
709+ assert(i == c)
710+ c = c + step
711+ end
712+
713+ c = maxi
714+ for i = maxi, mini, -step do
715+ assert(i == c)
716+ c = c - step
717+ end
718+
719+ checkfor(maxi, maxi, maxi, {maxi})
720+ checkfor(maxi, maxi, mini, {maxi})
721+ checkfor(mini, mini, maxi, {mini})
722+ checkfor(mini, mini, mini, {mini})
723+end
724+
725+
726+checkerror("'for' step is zero", function ()
727+ for i = 1, 10, 0 do end
728+end)
729+
730+checkerror("'for' step is zero", function ()
731+ for i = 1, -10, 0 do end
732+end)
733+
734+checkerror("'for' step is zero", function ()
735+ for i = 1.0, -10, 0.0 do end
736+end)
737+
738+collectgarbage()
739+
740+
741+-- testing generic 'for'
742+
743+local function f (n, p)
744+ local t = {}; for i=1,p do t[i] = i*10 end
745+ return function (_, n, ...)
746+ assert(select("#", ...) == 0) -- no extra arguments
747+ if n > 0 then
748+ n = n-1
749+ return n, table.unpack(t)
750+ end
751+ end, nil, n
752+end
753+
754+local x = 0
755+for n,a,b,c,d in f(5,3) do
756+ x = x+1
757+ assert(a == 10 and b == 20 and c == 30 and d == nil)
758+end
759+assert(x == 5)
760+
761+
762+
763+-- testing __pairs and __ipairs metamethod
764+a = {}
765+do
766+ local x,y,z = pairs(a)
767+ assert(type(x) == 'function' and y == a and z == nil)
768+end
769+
770+local function foo (e,i)
771+ assert(e == a)
772+ if i <= 10 then return i+1, i+2 end
773+end
774+
775+local function foo1 (e,i)
776+ i = i + 1
777+ assert(e == a)
778+ if i <= e.n then return i,a[i] end
779+end
780+
781+setmetatable(a, {__pairs = function (x) return foo, x, 0 end})
782+
783+local i = 0
784+for k,v in pairs(a) do
785+ i = i + 1
786+ assert(k == i and v == k+1)
787+end
788+
789+a.n = 5
790+a[3] = 30
791+
792+-- testing ipairs with metamethods
793+a = {n=10}
794+setmetatable(a, { __index = function (t,k)
795+ if k <= t.n then return k * 10 end
796+ end})
797+i = 0
798+for k,v in ipairs(a) do
799+ i = i + 1
800+ assert(k == i and v == i * 10)
801+end
802+assert(i == a.n)
803+
804+
805+-- testing yield inside __pairs
806+do
807+ local t = setmetatable({10, 20, 30}, {__pairs = function (t)
808+ local inc = coroutine.yield()
809+ return function (t, i)
810+ if i > 1 then return i - inc, t[i - inc] else return nil end
811+ end, t, #t + 1
812+ end})
813+
814+ local res = {}
815+ local co = coroutine.wrap(function ()
816+ for i,p in pairs(t) do res[#res + 1] = p end
817+ end)
818+
819+ co() -- start coroutine
820+ co(1) -- continue after yield
821+ assert(res[1] == 30 and res[2] == 20 and res[3] == 10 and #res == 3)
822+
823+end
824+
825+print"OK"
zod/luau/tests/pm.luanew file+440
@@ -0,0 +1,440 @@
1+-- $Id: testes/pm.lua $
2+-- See Copyright Notice in file all.lua
3+
4+-- UTF-8 file
5+
6+
7+print('testing pattern matching')
8+
9+local function checkerror (msg, f, ...)
10+ local s, err = pcall(f, ...)
11+ assert(not s and string.find(err, msg))
12+end
13+
14+
15+local function f (s, p)
16+ local i,e = string.find(s, p)
17+ if i then return string.sub(s, i, e) end
18+end
19+
20+local a,b = string.find('', '') -- empty patterns are tricky
21+assert(a == 1 and b == 0);
22+a,b = string.find('alo', '')
23+assert(a == 1 and b == 0)
24+a,b = string.find('a\0o a\0o a\0o', 'a', 1) -- first position
25+assert(a == 1 and b == 1)
26+a,b = string.find('a\0o a\0o a\0o', 'a\0o', 2) -- starts in the midle
27+assert(a == 5 and b == 7)
28+a,b = string.find('a\0o a\0o a\0o', 'a\0o', 9) -- starts in the midle
29+assert(a == 9 and b == 11)
30+a,b = string.find('a\0a\0a\0a\0\0ab', '\0ab', 2); -- finds at the end
31+assert(a == 9 and b == 11);
32+a,b = string.find('a\0a\0a\0a\0\0ab', 'b') -- last position
33+assert(a == 11 and b == 11)
34+assert(not string.find('a\0a\0a\0a\0\0ab', 'b\0')) -- check ending
35+assert(not string.find('', '\0'))
36+assert(string.find('alo123alo', '12') == 4)
37+assert(not string.find('alo123alo', '^12'))
38+
39+assert(string.match("aaab", ".*b") == "aaab")
40+assert(string.match("aaa", ".*a") == "aaa")
41+assert(string.match("b", ".*b") == "b")
42+
43+assert(string.match("aaab", ".+b") == "aaab")
44+assert(string.match("aaa", ".+a") == "aaa")
45+assert(not string.match("b", ".+b"))
46+
47+assert(string.match("aaab", ".?b") == "ab")
48+assert(string.match("aaa", ".?a") == "aa")
49+assert(string.match("b", ".?b") == "b")
50+
51+assert(f('aloALO', '%l*') == 'alo')
52+assert(f('aLo_ALO', '%a*') == 'aLo')
53+
54+assert(f(" \n\r*&\n\r xuxu \n\n", "%g%g%g+") == "xuxu")
55+
56+
57+-- Adapt a pattern to UTF-8
58+local function PU (p)
59+ -- reapply '?' into each individual byte of a character.
60+ -- (For instance, "á?" becomes "\195?\161?".)
61+ p = string.gsub(p, "(" .. utf8.charpattern .. ")%?", function (c)
62+ return string.gsub(c, ".", "%0?")
63+ end)
64+ -- change '.' to utf-8 character patterns
65+ p = string.gsub(p, "%.", utf8.charpattern)
66+ return p
67+end
68+
69+
70+assert(f('aaab', 'a*') == 'aaa');
71+assert(f('aaa', '^.*$') == 'aaa');
72+assert(f('aaa', 'b*') == '');
73+assert(f('aaa', 'ab*a') == 'aa')
74+assert(f('aba', 'ab*a') == 'aba')
75+assert(f('aaab', 'a+') == 'aaa')
76+assert(f('aaa', '^.+$') == 'aaa')
77+assert(not f('aaa', 'b+'))
78+assert(not f('aaa', 'ab+a'))
79+assert(f('aba', 'ab+a') == 'aba')
80+assert(f('a$a', '.$') == 'a')
81+assert(f('a$a', '.%$') == 'a$')
82+assert(f('a$a', '.$.') == 'a$a')
83+assert(not f('a$a', '$$'))
84+assert(not f('a$b', 'a$'))
85+assert(f('a$a', '$') == '')
86+assert(f('', 'b*') == '')
87+assert(not f('aaa', 'bb*'))
88+assert(f('aaab', 'a-') == '')
89+assert(f('aaa', '^.-$') == 'aaa')
90+assert(f('aabaaabaaabaaaba', 'b.*b') == 'baaabaaabaaab')
91+assert(f('aabaaabaaabaaaba', 'b.-b') == 'baaab')
92+assert(f('alo xo', '.o$') == 'xo')
93+assert(f(' \n isto é assim', '%S%S*') == 'isto')
94+assert(f(' \n isto é assim', '%S*$') == 'assim')
95+assert(f(' \n isto é assim', '[a-z]*$') == 'assim')
96+assert(f('um caracter ? extra', '[^%sa-z]') == '?')
97+assert(f('', 'a?') == '')
98+assert(f('á', PU'á?') == 'á')
99+assert(f('ábl', PU'á?b?l?') == 'ábl')
100+assert(f(' ábl', PU'á?b?l?') == '')
101+assert(f('aa', '^aa?a?a') == 'aa')
102+assert(f(']]]áb', '[^]]+') == 'áb')
103+assert(f("0alo alo", "%x*") == "0a")
104+assert(f("alo alo", "%C+") == "alo alo")
105+print('+')
106+
107+
108+local function f1 (s, p)
109+ p = string.gsub(p, "%%([0-9])", function (s)
110+ return "%" .. (tonumber(s)+1)
111+ end)
112+ p = string.gsub(p, "^(^?)", "%1()", 1)
113+ p = string.gsub(p, "($?)$", "()%1", 1)
114+ local t = {string.match(s, p)}
115+ return string.sub(s, t[1], t[#t] - 1)
116+end
117+
118+assert(f1('alo alx 123 b\0o b\0o', '(..*) %1') == "b\0o b\0o")
119+assert(f1('axz123= 4= 4 34', '(.+)=(.*)=%2 %1') == '3= 4= 4 3')
120+assert(f1('=======', '^(=*)=%1$') == '=======')
121+assert(not string.match('==========', '^([=]*)=%1$'))
122+
123+local function range (i, j)
124+ if i <= j then
125+ return i, range(i+1, j)
126+ end
127+end
128+
129+local abc = string.char(range(0, 127)) .. string.char(range(128, 255));
130+
131+assert(string.len(abc) == 256)
132+
133+local function strset (p)
134+ local res = {s=''}
135+ string.gsub(abc, p, function (c) res.s = res.s .. c end)
136+ return res.s
137+end;
138+
139+assert(string.len(strset('[\200-\210]')) == 11)
140+
141+assert(strset('[a-z]') == "abcdefghijklmnopqrstuvwxyz")
142+assert(strset('[a-z%d]') == strset('[%da-uu-z]'))
143+assert(strset('[a-]') == "-a")
144+assert(strset('[^%W]') == strset('[%w]'))
145+assert(strset('[]%%]') == '%]')
146+assert(strset('[a%-z]') == '-az')
147+assert(strset('[%^%[%-a%]%-b]') == '-[]^ab')
148+assert(strset('%Z') == strset('[\1-\255]'))
149+assert(strset('.') == strset('[\1-\255%z]'))
150+print('+');
151+
152+assert(string.match("alo xyzK", "(%w+)K") == "xyz")
153+assert(string.match("254 K", "(%d*)K") == "")
154+assert(string.match("alo ", "(%w*)$") == "")
155+assert(not string.match("alo ", "(%w+)$"))
156+assert(string.find("(álo)", "%(á") == 1)
157+local a, b, c, d, e = string.match("âlo alo", PU"^(((.).). (%w*))$")
158+assert(a == 'âlo alo' and b == 'âl' and c == 'â' and d == 'alo' and e == nil)
159+a, b, c, d = string.match('0123456789', '(.+(.?)())')
160+assert(a == '0123456789' and b == '' and c == 11 and d == nil)
161+print('+')
162+
163+assert(string.gsub('ülo ülo', 'ü', 'x') == 'xlo xlo')
164+assert(string.gsub('alo úlo ', ' +$', '') == 'alo úlo') -- trim
165+assert(string.gsub(' alo alo ', '^%s*(.-)%s*$', '%1') == 'alo alo') -- double trim
166+assert(string.gsub('alo alo \n 123\n ', '%s+', ' ') == 'alo alo 123 ')
167+local t = "abç d"
168+a, b = string.gsub(t, PU'(.)', '%1@')
169+assert(a == "a@b@ç@ @d@" and b == 5)
170+a, b = string.gsub('abçd', PU'(.)', '%0@', 2)
171+assert(a == 'a@b@çd' and b == 2)
172+assert(string.gsub('alo alo', '()[al]', '%1') == '12o 56o')
173+assert(string.gsub("abc=xyz", "(%w*)(%p)(%w+)", "%3%2%1-%0") ==
174+ "xyz=abc-abc=xyz")
175+assert(string.gsub("abc", "%w", "%1%0") == "aabbcc")
176+assert(string.gsub("abc", "%w+", "%0%1") == "abcabc")
177+assert(string.gsub('áéí', '$', '\0óú') == 'áéí\0óú')
178+assert(string.gsub('', '^', 'r') == 'r')
179+assert(string.gsub('', '$', 'r') == 'r')
180+print('+')
181+
182+
183+do -- new (5.3.3) semantics for empty matches
184+ assert(string.gsub("a b cd", " *", "-") == "-a-b-c-d-")
185+
186+ local res = ""
187+ local sub = "a \nbc\t\td"
188+ local i = 1
189+ for p, e in string.gmatch(sub, "()%s*()") do
190+ res = res .. string.sub(sub, i, p - 1) .. "-"
191+ i = e
192+ end
193+ assert(res == "-a-b-c-d-")
194+end
195+
196+
197+assert(string.gsub("um (dois) tres (quatro)", "(%(%w+%))", string.upper) ==
198+ "um (DOIS) tres (QUATRO)")
199+
200+do
201+ local function setglobal (n,v) rawset(_G, n, v) end
202+ string.gsub("a=roberto,roberto=a", "(%w+)=(%w%w*)", setglobal)
203+ assert(_G.a=="roberto" and _G.roberto=="a")
204+ _G.a = nil; _G.roberto = nil
205+end
206+
207+function f(a,b) return string.gsub(a,'.',b) end
208+assert(string.gsub("trocar tudo em |teste|b| é |beleza|al|", "|([^|]*)|([^|]*)|", f) ==
209+ "trocar tudo em bbbbb é alalalalalal")
210+
211+local function dostring (s) return load(s, "")() or "" end
212+assert(string.gsub("alo $a='x'$ novamente $return a$",
213+ "$([^$]*)%$",
214+ dostring) == "alo novamente x")
215+
216+local x = string.gsub("$x=string.gsub('alo', '.', string.upper)$ assim vai para $return x$",
217+ "$([^$]*)%$", dostring)
218+assert(x == ' assim vai para ALO')
219+_G.a, _G.x = nil
220+
221+local t = {}
222+local s = 'a alo jose joao'
223+local r = string.gsub(s, '()(%w+)()', function (a,w,b)
224+ assert(string.len(w) == b-a);
225+ t[a] = b-a;
226+ end)
227+assert(s == r and t[1] == 1 and t[3] == 3 and t[7] == 4 and t[13] == 4)
228+
229+
230+local function isbalanced (s)
231+ return not string.find(string.gsub(s, "%b()", ""), "[()]")
232+end
233+
234+assert(isbalanced("(9 ((8))(\0) 7) \0\0 a b ()(c)() a"))
235+assert(not isbalanced("(9 ((8) 7) a b (\0 c) a"))
236+assert(string.gsub("alo 'oi' alo", "%b''", '"') == 'alo " alo')
237+
238+
239+local t = {"apple", "orange", "lime"; n=0}
240+assert(string.gsub("x and x and x", "x", function () t.n=t.n+1; return t[t.n] end)
241+ == "apple and orange and lime")
242+
243+t = {n=0}
244+string.gsub("first second word", "%w%w*", function (w) t.n=t.n+1; t[t.n] = w end)
245+assert(t[1] == "first" and t[2] == "second" and t[3] == "word" and t.n == 3)
246+
247+t = {n=0}
248+assert(string.gsub("first second word", "%w+",
249+ function (w) t.n=t.n+1; t[t.n] = w end, 2) == "first second word")
250+assert(t[1] == "first" and t[2] == "second" and t[3] == undef)
251+
252+checkerror("invalid replacement value %(a table%)",
253+ string.gsub, "alo", ".", {a = {}})
254+checkerror("invalid capture index %%2", string.gsub, "alo", ".", "%2")
255+checkerror("invalid capture index %%0", string.gsub, "alo", "(%0)", "a")
256+checkerror("invalid capture index %%1", string.gsub, "alo", "(%1)", "a")
257+checkerror("invalid use of '%%'", string.gsub, "alo", ".", "%x")
258+
259+
260+if not _soft then
261+ print("big strings")
262+ local a = string.rep('a', 300000)
263+ assert(string.find(a, '^a*.?$'))
264+ assert(not string.find(a, '^a*.?b$'))
265+ assert(string.find(a, '^a-.?$'))
266+
267+ -- bug in 5.1.2
268+ a = string.rep('a', 10000) .. string.rep('b', 10000)
269+ assert(not pcall(string.gsub, a, 'b'))
270+end
271+
272+-- recursive nest of gsubs
273+local function rev (s)
274+ return string.gsub(s, "(.)(.+)", function (c,s1) return rev(s1)..c end)
275+end
276+
277+local x = "abcdef"
278+assert(rev(rev(x)) == x)
279+
280+
281+-- gsub with tables
282+assert(string.gsub("alo alo", ".", {}) == "alo alo")
283+assert(string.gsub("alo alo", "(.)", {a="AA", l=""}) == "AAo AAo")
284+assert(string.gsub("alo alo", "(.).", {a="AA", l="K"}) == "AAo AAo")
285+assert(string.gsub("alo alo", "((.)(.?))", {al="AA", o=false}) == "AAo AAo")
286+
287+assert(string.gsub("alo alo", "().", {'x','yy','zzz'}) == "xyyzzz alo")
288+
289+t = {}; setmetatable(t, {__index = function (t,s) return string.upper(s) end})
290+assert(string.gsub("a alo b hi", "%w%w+", t) == "a ALO b HI")
291+
292+
293+-- tests for gmatch
294+local a = 0
295+for i in string.gmatch('abcde', '()') do assert(i == a+1); a=i end
296+assert(a==6)
297+
298+t = {n=0}
299+for w in string.gmatch("first second word", "%w+") do
300+ t.n=t.n+1; t[t.n] = w
301+end
302+assert(t[1] == "first" and t[2] == "second" and t[3] == "word")
303+
304+t = {3, 6, 9}
305+for i in string.gmatch ("xuxx uu ppar r", "()(.)%2") do
306+ assert(i == table.remove(t, 1))
307+end
308+assert(#t == 0)
309+
310+t = {}
311+for i,j in string.gmatch("13 14 10 = 11, 15= 16, 22=23", "(%d+)%s*=%s*(%d+)") do
312+ t[tonumber(i)] = tonumber(j)
313+end
314+a = 0
315+for k,v in pairs(t) do assert(k+1 == v+0); a=a+1 end
316+assert(a == 3)
317+
318+
319+do -- init parameter in gmatch
320+ local s = 0
321+ for k in string.gmatch("10 20 30", "%d+", 3) do
322+ s = s + tonumber(k)
323+ end
324+ assert(s == 50)
325+
326+ s = 0
327+ for k in string.gmatch("11 21 31", "%d+", -4) do
328+ s = s + tonumber(k)
329+ end
330+ assert(s == 32)
331+
332+ -- there is an empty string at the end of the subject
333+ s = 0
334+ for k in string.gmatch("11 21 31", "%w*", 9) do
335+ s = s + 1
336+ end
337+ assert(s == 1)
338+
339+ -- there are no empty strings after the end of the subject
340+ s = 0
341+ for k in string.gmatch("11 21 31", "%w*", 10) do
342+ s = s + 1
343+ end
344+ assert(s == 0)
345+end
346+
347+
348+-- tests for `%f' (`frontiers')
349+
350+assert(string.gsub("aaa aa a aaa a", "%f[%w]a", "x") == "xaa xa x xaa x")
351+assert(string.gsub("[[]] [][] [[[[", "%f[[].", "x") == "x[]] x]x] x[[[")
352+assert(string.gsub("01abc45de3", "%f[%d]", ".") == ".01abc.45de.3")
353+assert(string.gsub("01abc45 de3x", "%f[%D]%w", ".") == "01.bc45 de3.")
354+assert(string.gsub("function", "%f[\1-\255]%w", ".") == ".unction")
355+assert(string.gsub("function", "%f[^\1-\255]", ".") == "function.")
356+
357+assert(string.find("a", "%f[a]") == 1)
358+assert(string.find("a", "%f[^%z]") == 1)
359+assert(string.find("a", "%f[^%l]") == 2)
360+assert(string.find("aba", "%f[a%z]") == 3)
361+assert(string.find("aba", "%f[%z]") == 4)
362+assert(not string.find("aba", "%f[%l%z]"))
363+assert(not string.find("aba", "%f[^%l%z]"))
364+
365+local i, e = string.find(" alo aalo allo", "%f[%S].-%f[%s].-%f[%S]")
366+assert(i == 2 and e == 5)
367+local k = string.match(" alo aalo allo", "%f[%S](.-%f[%s].-%f[%S])")
368+assert(k == 'alo ')
369+
370+local a = {1, 5, 9, 14, 17,}
371+for k in string.gmatch("alo alo th02 is 1hat", "()%f[%w%d]") do
372+ assert(table.remove(a, 1) == k)
373+end
374+assert(#a == 0)
375+
376+
377+-- malformed patterns
378+local function malform (p, m)
379+ m = m or "malformed"
380+ local r, msg = pcall(string.find, "a", p)
381+ assert(not r and string.find(msg, m))
382+end
383+
384+malform("(.", "unfinished capture")
385+malform(".)", "invalid pattern capture")
386+malform("[a")
387+malform("[]")
388+malform("[^]")
389+malform("[a%]")
390+malform("[a%")
391+malform("%b")
392+malform("%ba")
393+malform("%")
394+malform("%f", "missing")
395+
396+-- \0 in patterns
397+assert(string.match("ab\0\1\2c", "[\0-\2]+") == "\0\1\2")
398+assert(string.match("ab\0\1\2c", "[\0-\0]+") == "\0")
399+assert(string.find("b$a", "$\0?") == 2)
400+assert(string.find("abc\0efg", "%\0") == 4)
401+assert(string.match("abc\0efg\0\1e\1g", "%b\0\1") == "\0efg\0\1e\1")
402+assert(string.match("abc\0\0\0", "%\0+") == "\0\0\0")
403+assert(string.match("abc\0\0\0", "%\0%\0?") == "\0\0")
404+
405+-- magic char after \0
406+assert(string.find("abc\0\0","\0.") == 4)
407+assert(string.find("abcx\0\0abc\0abc","x\0\0abc\0a.") == 4)
408+
409+
410+do -- test reuse of original string in gsub
411+ local s = string.rep("a", 100)
412+ local r = string.gsub(s, "b", "c") -- no match
413+ assert(string.format("%p", s) == string.format("%p", r))
414+
415+ r = string.gsub(s, ".", {x = "y"}) -- no substitutions
416+ assert(string.format("%p", s) == string.format("%p", r))
417+
418+ local count = 0
419+ r = string.gsub(s, ".", function (x)
420+ assert(x == "a")
421+ count = count + 1
422+ return nil -- no substitution
423+ end)
424+ r = string.gsub(r, ".", {b = 'x'}) -- "a" is not a key; no subst.
425+ assert(count == 100)
426+ assert(string.format("%p", s) == string.format("%p", r))
427+
428+ count = 0
429+ r = string.gsub(s, ".", function (x)
430+ assert(x == "a")
431+ count = count + 1
432+ return x -- substitution...
433+ end)
434+ assert(count == 100)
435+ -- no reuse in this case
436+ assert(r == s and string.format("%p", s) ~= string.format("%p", r))
437+end
438+
439+print('OK')
440+
zod/luau/tests/sort.luanew file+311
@@ -0,0 +1,311 @@
1+-- $Id: testes/sort.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print "testing (parts of) table library"
5+
6+print "testing unpack"
7+
8+local unpack = table.unpack
9+
10+local maxI = math.maxinteger
11+local minI = math.mininteger
12+
13+
14+local function checkerror (msg, f, ...)
15+ local s, err = pcall(f, ...)
16+ assert(not s and string.find(err, msg))
17+end
18+
19+
20+checkerror("wrong number of arguments", table.insert, {}, 2, 3, 4)
21+
22+local x,y,z,a,n
23+a = {}; local lim = _soft and 200 or 2000
24+for i=1, lim do a[i]=i end
25+assert(select(lim, unpack(a)) == lim and select('#', unpack(a)) == lim)
26+x = unpack(a)
27+assert(x == 1)
28+x = {unpack(a)}
29+assert(#x == lim and x[1] == 1 and x[lim] == lim)
30+x = {unpack(a, lim-2)}
31+assert(#x == 3 and x[1] == lim-2 and x[3] == lim)
32+x = {unpack(a, 10, 6)}
33+assert(next(x) == nil) -- no elements
34+x = {unpack(a, 11, 10)}
35+assert(next(x) == nil) -- no elements
36+x,y = unpack(a, 10, 10)
37+assert(x == 10 and y == nil)
38+x,y,z = unpack(a, 10, 11)
39+assert(x == 10 and y == 11 and z == nil)
40+a,x = unpack{1}
41+assert(a==1 and x==nil)
42+a,x = unpack({1,2}, 1, 1)
43+assert(a==1 and x==nil)
44+
45+do
46+ local maxi = (1 << 31) - 1 -- maximum value for an int (usually)
47+ local mini = -(1 << 31) -- minimum value for an int (usually)
48+ checkerror("too many results", unpack, {}, 0, maxi)
49+ checkerror("too many results", unpack, {}, 1, maxi)
50+ checkerror("too many results", unpack, {}, 0, maxI)
51+ checkerror("too many results", unpack, {}, 1, maxI)
52+ checkerror("too many results", unpack, {}, mini, maxi)
53+ checkerror("too many results", unpack, {}, -maxi, maxi)
54+ checkerror("too many results", unpack, {}, minI, maxI)
55+ unpack({}, maxi, 0)
56+ unpack({}, maxi, 1)
57+ unpack({}, maxI, minI)
58+ pcall(unpack, {}, 1, maxi + 1)
59+ local a, b = unpack({[maxi] = 20}, maxi, maxi)
60+ assert(a == 20 and b == nil)
61+ a, b = unpack({[maxi] = 20}, maxi - 1, maxi)
62+ assert(a == nil and b == 20)
63+ local t = {[maxI - 1] = 12, [maxI] = 23}
64+ a, b = unpack(t, maxI - 1, maxI); assert(a == 12 and b == 23)
65+ a, b = unpack(t, maxI, maxI); assert(a == 23 and b == nil)
66+ a, b = unpack(t, maxI, maxI - 1); assert(a == nil and b == nil)
67+ t = {[minI] = 12.3, [minI + 1] = 23.5}
68+ a, b = unpack(t, minI, minI + 1); assert(a == 12.3 and b == 23.5)
69+ a, b = unpack(t, minI, minI); assert(a == 12.3 and b == nil)
70+ a, b = unpack(t, minI + 1, minI); assert(a == nil and b == nil)
71+end
72+
73+do -- length is not an integer
74+ local t = setmetatable({}, {__len = function () return 'abc' end})
75+ assert(#t == 'abc')
76+ checkerror("object length is not an integer", table.insert, t, 1)
77+end
78+
79+print "testing pack"
80+
81+a = table.pack()
82+assert(a[1] == undef and a.n == 0)
83+
84+a = table.pack(table)
85+assert(a[1] == table and a.n == 1)
86+
87+a = table.pack(nil, nil, nil, nil)
88+assert(a[1] == nil and a.n == 4)
89+
90+
91+-- testing move
92+do
93+
94+ checkerror("table expected", table.move, 1, 2, 3, 4)
95+
96+ local function eqT (a, b)
97+ for k, v in pairs(a) do assert(b[k] == v) end
98+ for k, v in pairs(b) do assert(a[k] == v) end
99+ end
100+
101+ local a = table.move({10,20,30}, 1, 3, 2) -- move forward
102+ eqT(a, {10,10,20,30})
103+
104+ -- move forward with overlap of 1
105+ a = table.move({10, 20, 30}, 1, 3, 3)
106+ eqT(a, {10, 20, 10, 20, 30})
107+
108+ -- moving to the same table (not being explicit about it)
109+ a = {10, 20, 30, 40}
110+ table.move(a, 1, 4, 2, a)
111+ eqT(a, {10, 10, 20, 30, 40})
112+
113+ a = table.move({10,20,30}, 2, 3, 1) -- move backward
114+ eqT(a, {20,30,30})
115+
116+ a = {} -- move to new table
117+ assert(table.move({10,20,30}, 1, 3, 1, a) == a)
118+ eqT(a, {10,20,30})
119+
120+ a = {}
121+ assert(table.move({10,20,30}, 1, 0, 3, a) == a) -- empty move (no move)
122+ eqT(a, {})
123+
124+ a = table.move({10,20,30}, 1, 10, 1) -- move to the same place
125+ eqT(a, {10,20,30})
126+
127+ -- moving on the fringes
128+ a = table.move({[maxI - 2] = 1, [maxI - 1] = 2, [maxI] = 3},
129+ maxI - 2, maxI, -10, {})
130+ eqT(a, {[-10] = 1, [-9] = 2, [-8] = 3})
131+
132+ a = table.move({[minI] = 1, [minI + 1] = 2, [minI + 2] = 3},
133+ minI, minI + 2, -10, {})
134+ eqT(a, {[-10] = 1, [-9] = 2, [-8] = 3})
135+
136+ a = table.move({45}, 1, 1, maxI)
137+ eqT(a, {45, [maxI] = 45})
138+
139+ a = table.move({[maxI] = 100}, maxI, maxI, minI)
140+ eqT(a, {[minI] = 100, [maxI] = 100})
141+
142+ a = table.move({[minI] = 100}, minI, minI, maxI)
143+ eqT(a, {[minI] = 100, [maxI] = 100})
144+
145+ a = setmetatable({}, {
146+ __index = function (_,k) return k * 10 end,
147+ __newindex = error})
148+ local b = table.move(a, 1, 10, 3, {})
149+ eqT(a, {})
150+ eqT(b, {nil,nil,10,20,30,40,50,60,70,80,90,100})
151+
152+ b = setmetatable({""}, {
153+ __index = error,
154+ __newindex = function (t,k,v)
155+ t[1] = string.format("%s(%d,%d)", t[1], k, v)
156+ end})
157+ table.move(a, 10, 13, 3, b)
158+ assert(b[1] == "(3,100)(4,110)(5,120)(6,130)")
159+ local stat, msg = pcall(table.move, b, 10, 13, 3, b)
160+ assert(not stat and msg == b)
161+end
162+
163+do
164+ -- for very long moves, just check initial accesses and interrupt
165+ -- move with an error
166+ local function checkmove (f, e, t, x, y)
167+ local pos1, pos2
168+ local a = setmetatable({}, {
169+ __index = function (_,k) pos1 = k end,
170+ __newindex = function (_,k) pos2 = k; error() end, })
171+ local st, msg = pcall(table.move, a, f, e, t)
172+ assert(not st and not msg and pos1 == x and pos2 == y)
173+ end
174+ checkmove(1, maxI, 0, 1, 0)
175+ checkmove(0, maxI - 1, 1, maxI - 1, maxI)
176+ checkmove(minI, -2, -5, -2, maxI - 6)
177+ checkmove(minI + 1, -1, -2, -1, maxI - 3)
178+ checkmove(minI, -2, 0, minI, 0) -- non overlapping
179+ checkmove(minI + 1, -1, 1, minI + 1, 1) -- non overlapping
180+end
181+
182+checkerror("too many", table.move, {}, 0, maxI, 1)
183+checkerror("too many", table.move, {}, -1, maxI - 1, 1)
184+checkerror("too many", table.move, {}, minI, -1, 1)
185+checkerror("too many", table.move, {}, minI, maxI, 1)
186+checkerror("wrap around", table.move, {}, 1, maxI, 2)
187+checkerror("wrap around", table.move, {}, 1, 2, maxI)
188+checkerror("wrap around", table.move, {}, minI, -2, 2)
189+
190+
191+print"testing sort"
192+
193+
194+-- strange lengths
195+local a = setmetatable({}, {__len = function () return -1 end})
196+assert(#a == -1)
197+table.sort(a, error) -- should not compare anything
198+a = setmetatable({}, {__len = function () return maxI end})
199+checkerror("too big", table.sort, a)
200+
201+-- test checks for invalid order functions
202+local function check (t)
203+ local function f(a, b) assert(a and b); return true end
204+ checkerror("invalid order function", table.sort, t, f)
205+end
206+
207+check{1,2,3,4}
208+check{1,2,3,4,5}
209+check{1,2,3,4,5,6}
210+
211+
212+function check (a, f)
213+ f = f or function (x,y) return x<y end;
214+ for n = #a, 2, -1 do
215+ assert(not f(a[n], a[n-1]))
216+ end
217+end
218+
219+a = {"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep",
220+ "Oct", "Nov", "Dec"}
221+
222+table.sort(a)
223+check(a)
224+
225+local function perm (s, n)
226+ n = n or #s
227+ if n == 1 then
228+ local t = {unpack(s)}
229+ table.sort(t)
230+ check(t)
231+ else
232+ for i = 1, n do
233+ s[i], s[n] = s[n], s[i]
234+ perm(s, n - 1)
235+ s[i], s[n] = s[n], s[i]
236+ end
237+ end
238+end
239+
240+perm{}
241+perm{1}
242+perm{1,2}
243+perm{1,2,3}
244+perm{1,2,3,4}
245+perm{2,2,3,4}
246+perm{1,2,3,4,5}
247+perm{1,2,3,3,5}
248+perm{1,2,3,4,5,6}
249+perm{2,2,3,3,5,6}
250+
251+local function timesort (a, n, func, msg, pre)
252+ local x = os.clock()
253+ table.sort(a, func)
254+ x = (os.clock() - x) * 1000
255+ pre = pre or ""
256+ print(string.format("%ssorting %d %s elements in %.2f msec.", pre, n, msg, x))
257+ check(a, func)
258+end
259+
260+local limit = 50000
261+if _soft then limit = 5000 end
262+
263+a = {}
264+for i=1,limit do
265+ a[i] = math.random()
266+end
267+
268+timesort(a, limit, nil, "random")
269+
270+timesort(a, limit, nil, "sorted", "re-")
271+
272+a = {}
273+for i=1,limit do
274+ a[i] = math.random()
275+end
276+
277+local x = os.clock(); local i = 0
278+table.sort(a, function(x,y) i=i+1; return y<x end)
279+x = (os.clock() - x) * 1000
280+print(string.format("Invert-sorting other %d elements in %.2f msec., with %i comparisons",
281+ limit, x, i))
282+check(a, function(x,y) return y<x end)
283+
284+
285+table.sort{} -- empty array
286+
287+for i=1,limit do a[i] = false end
288+timesort(a, limit, function(x,y) return nil end, "equal")
289+
290+for i,v in pairs(a) do assert(v == false) end
291+
292+AA = {"\xE1lo", "\0first :-)", "alo", "then this one", "45", "and a new"}
293+table.sort(AA)
294+check(AA)
295+
296+table.sort(AA, function (x, y)
297+ load(string.format("AA[%q] = ''", x), "")()
298+ collectgarbage()
299+ return x<y
300+ end)
301+
302+_G.AA = nil
303+
304+local tt = {__lt = function (a,b) return a.val < b.val end}
305+a = {}
306+for i=1,10 do a[i] = {val=math.random(100)}; setmetatable(a[i], tt); end
307+table.sort(a)
308+check(a, tt.__lt)
309+check(a)
310+
311+print"OK"
zod/luau/tests/strings.luanew file+526
@@ -0,0 +1,526 @@
1+-- $Id: testes/strings.lua $
2+-- See Copyright Notice in file all.lua
3+
4+-- ISO Latin encoding
5+
6+
7+print('testing strings and string library')
8+
9+local maxi <const> = math.maxinteger
10+local mini <const> = math.mininteger
11+
12+
13+local function checkerror (msg, f, ...)
14+ local s, err = pcall(f, ...)
15+ assert(not s and string.find(err, msg))
16+end
17+
18+
19+-- testing string comparisons
20+assert('alo' < 'alo1')
21+assert('' < 'a')
22+assert('alo\0alo' < 'alo\0b')
23+assert('alo\0alo\0\0' > 'alo\0alo\0')
24+assert('alo' < 'alo\0')
25+assert('alo\0' > 'alo')
26+assert('\0' < '\1')
27+assert('\0\0' < '\0\1')
28+assert('\1\0a\0a' <= '\1\0a\0a')
29+assert(not ('\1\0a\0b' <= '\1\0a\0a'))
30+assert('\0\0\0' < '\0\0\0\0')
31+assert(not('\0\0\0\0' < '\0\0\0'))
32+assert('\0\0\0' <= '\0\0\0\0')
33+assert(not('\0\0\0\0' <= '\0\0\0'))
34+assert('\0\0\0' <= '\0\0\0')
35+assert('\0\0\0' >= '\0\0\0')
36+assert(not ('\0\0b' < '\0\0a\0'))
37+
38+-- testing string.sub
39+assert(string.sub("123456789",2,4) == "234")
40+assert(string.sub("123456789",7) == "789")
41+assert(string.sub("123456789",7,6) == "")
42+assert(string.sub("123456789",7,7) == "7")
43+assert(string.sub("123456789",0,0) == "")
44+assert(string.sub("123456789",-10,10) == "123456789")
45+assert(string.sub("123456789",1,9) == "123456789")
46+assert(string.sub("123456789",-10,-20) == "")
47+assert(string.sub("123456789",-1) == "9")
48+assert(string.sub("123456789",-4) == "6789")
49+assert(string.sub("123456789",-6, -4) == "456")
50+assert(string.sub("123456789", mini, -4) == "123456")
51+assert(string.sub("123456789", mini, maxi) == "123456789")
52+assert(string.sub("123456789", mini, mini) == "")
53+assert(string.sub("\000123456789",3,5) == "234")
54+assert(("\000123456789"):sub(8) == "789")
55+
56+-- testing string.find
57+assert(string.find("123456789", "345") == 3)
58+local a,b = string.find("123456789", "345")
59+assert(string.sub("123456789", a, b) == "345")
60+assert(string.find("1234567890123456789", "345", 3) == 3)
61+assert(string.find("1234567890123456789", "345", 4) == 13)
62+assert(not string.find("1234567890123456789", "346", 4))
63+assert(string.find("1234567890123456789", ".45", -9) == 13)
64+assert(not string.find("abcdefg", "\0", 5, 1))
65+assert(string.find("", "") == 1)
66+assert(string.find("", "", 1) == 1)
67+assert(not string.find("", "", 2))
68+assert(not string.find('', 'aaa', 1))
69+assert(('alo(.)alo'):find('(.)', 1, 1) == 4)
70+
71+assert(string.len("") == 0)
72+assert(string.len("\0\0\0") == 3)
73+assert(string.len("1234567890") == 10)
74+
75+assert(#"" == 0)
76+assert(#"\0\0\0" == 3)
77+assert(#"1234567890" == 10)
78+
79+-- testing string.byte/string.char
80+assert(string.byte("a") == 97)
81+assert(string.byte("\xe4") > 127)
82+assert(string.byte(string.char(255)) == 255)
83+assert(string.byte(string.char(0)) == 0)
84+assert(string.byte("\0") == 0)
85+assert(string.byte("\0\0alo\0x", -1) == string.byte('x'))
86+assert(string.byte("ba", 2) == 97)
87+assert(string.byte("\n\n", 2, -1) == 10)
88+assert(string.byte("\n\n", 2, 2) == 10)
89+assert(string.byte("") == nil)
90+assert(string.byte("hi", -3) == nil)
91+assert(string.byte("hi", 3) == nil)
92+assert(string.byte("hi", 9, 10) == nil)
93+assert(string.byte("hi", 2, 1) == nil)
94+assert(string.char() == "")
95+assert(string.char(0, 255, 0) == "\0\255\0")
96+assert(string.char(0, string.byte("\xe4"), 0) == "\0\xe4\0")
97+assert(string.char(string.byte("\xe4l\0�u", 1, -1)) == "\xe4l\0�u")
98+assert(string.char(string.byte("\xe4l\0�u", 1, 0)) == "")
99+assert(string.char(string.byte("\xe4l\0�u", -10, 100)) == "\xe4l\0�u")
100+
101+checkerror("out of range", string.char, 256)
102+checkerror("out of range", string.char, -1)
103+checkerror("out of range", string.char, math.maxinteger)
104+checkerror("out of range", string.char, math.mininteger)
105+
106+assert(string.upper("ab\0c") == "AB\0C")
107+assert(string.lower("\0ABCc%$") == "\0abcc%$")
108+assert(string.rep('teste', 0) == '')
109+assert(string.rep('t�s\00t�', 2) == 't�s\0t�t�s\000t�')
110+assert(string.rep('', 10) == '')
111+
112+if string.packsize("i") == 4 then
113+ -- result length would be 2^31 (int overflow)
114+ checkerror("too large", string.rep, 'aa', (1 << 30))
115+ checkerror("too large", string.rep, 'a', (1 << 30), ',')
116+end
117+
118+-- repetitions with separator
119+assert(string.rep('teste', 0, 'xuxu') == '')
120+assert(string.rep('teste', 1, 'xuxu') == 'teste')
121+assert(string.rep('\1\0\1', 2, '\0\0') == '\1\0\1\0\0\1\0\1')
122+assert(string.rep('', 10, '.') == string.rep('.', 9))
123+assert(not pcall(string.rep, "aa", maxi // 2 + 10))
124+assert(not pcall(string.rep, "", maxi // 2 + 10, "aa"))
125+
126+assert(string.reverse"" == "")
127+assert(string.reverse"\0\1\2\3" == "\3\2\1\0")
128+assert(string.reverse"\0001234" == "4321\0")
129+
130+for i=0,30 do assert(string.len(string.rep('a', i)) == i) end
131+
132+assert(type(tostring(nil)) == 'string')
133+assert(type(tostring(12)) == 'string')
134+assert(string.find(tostring{}, 'table:'))
135+assert(string.find(tostring(print), 'function:'))
136+assert(#tostring('\0') == 1)
137+assert(tostring(true) == "true")
138+assert(tostring(false) == "false")
139+assert(tostring(-1203) == "-1203")
140+assert(tostring(1203.125) == "1203.125")
141+assert(tostring(-0.5) == "-0.5")
142+assert(tostring(-32767) == "-32767")
143+if math.tointeger(2147483647) then -- no overflow? (32 bits)
144+ assert(tostring(-2147483647) == "-2147483647")
145+end
146+if math.tointeger(4611686018427387904) then -- no overflow? (64 bits)
147+ assert(tostring(4611686018427387904) == "4611686018427387904")
148+ assert(tostring(-4611686018427387904) == "-4611686018427387904")
149+end
150+
151+if tostring(0.0) == "0.0" then -- "standard" coercion float->string
152+ assert('' .. 12 == '12' and 12.0 .. '' == '12.0')
153+ assert(tostring(-1203 + 0.0) == "-1203.0")
154+else -- compatible coercion
155+ assert(tostring(0.0) == "0")
156+ assert('' .. 12 == '12' and 12.0 .. '' == '12')
157+ assert(tostring(-1203 + 0.0) == "-1203")
158+end
159+
160+do -- tests for '%p' format
161+ -- not much to test, as C does not specify what '%p' does.
162+ -- ("The value of the pointer is converted to a sequence of printing
163+ -- characters, in an implementation-defined manner.")
164+ local null = "(null)" -- nulls are formatted by Lua
165+ assert(string.format("%p", 4) == null)
166+ assert(string.format("%p", true) == null)
167+ assert(string.format("%p", nil) == null)
168+ assert(string.format("%p", {}) ~= null)
169+ assert(string.format("%p", print) ~= null)
170+ assert(string.format("%p", coroutine.running()) ~= null)
171+ assert(string.format("%p", io.stdin) ~= null)
172+ assert(string.format("%p", io.stdin) == string.format("%p", io.stdin))
173+ assert(string.format("%p", print) == string.format("%p", print))
174+ assert(string.format("%p", print) ~= string.format("%p", assert))
175+
176+ assert(#string.format("%90p", {}) == 90)
177+ assert(#string.format("%-60p", {}) == 60)
178+ assert(string.format("%10p", false) == string.rep(" ", 10 - #null) .. null)
179+ assert(string.format("%-12p", 1.5) == null .. string.rep(" ", 12 - #null))
180+
181+ do
182+ local t1 = {}; local t2 = {}
183+ assert(string.format("%p", t1) ~= string.format("%p", t2))
184+ end
185+
186+ do -- short strings are internalized
187+ local s1 = string.rep("a", 10)
188+ local s2 = string.rep("aa", 5)
189+ assert(string.format("%p", s1) == string.format("%p", s2))
190+ end
191+
192+ do -- long strings aren't internalized
193+ local s1 = string.rep("a", 300); local s2 = string.rep("a", 300)
194+ assert(string.format("%p", s1) ~= string.format("%p", s2))
195+ end
196+end
197+
198+local x = '"�lo"\n\\'
199+assert(string.format('%q%s', x, x) == '"\\"�lo\\"\\\n\\\\""�lo"\n\\')
200+assert(string.format('%q', "\0") == [["\0"]])
201+assert(load(string.format('return %q', x))() == x)
202+x = "\0\1\0023\5\0009"
203+assert(load(string.format('return %q', x))() == x)
204+assert(string.format("\0%c\0%c%x\0", string.byte("\xe4"), string.byte("b"), 140) ==
205+ "\0\xe4\0b8c\0")
206+assert(string.format('') == "")
207+assert(string.format("%c",34)..string.format("%c",48)..string.format("%c",90)..string.format("%c",100) ==
208+ string.format("%1c%-c%-1c%c", 34, 48, 90, 100))
209+assert(string.format("%s\0 is not \0%s", 'not be', 'be') == 'not be\0 is not \0be')
210+assert(string.format("%%%d %010d", 10, 23) == "%10 0000000023")
211+assert(tonumber(string.format("%f", 10.3)) == 10.3)
212+assert(string.format('"%-50s"', 'a') == '"a' .. string.rep(' ', 49) .. '"')
213+
214+assert(string.format("-%.20s.20s", string.rep("%", 2000)) ==
215+ "-"..string.rep("%", 20)..".20s")
216+assert(string.format('"-%20s.20s"', string.rep("%", 2000)) ==
217+ string.format("%q", "-"..string.rep("%", 2000)..".20s"))
218+
219+do
220+ local function checkQ (v)
221+ local s = string.format("%q", v)
222+ local nv = load("return " .. s)()
223+ assert(v == nv and math.type(v) == math.type(nv))
224+ end
225+ checkQ("\0\0\1\255\u{234}")
226+ checkQ(math.maxinteger)
227+ checkQ(math.mininteger)
228+ checkQ(math.pi)
229+ checkQ(0.1)
230+ checkQ(true)
231+ checkQ(nil)
232+ checkQ(false)
233+ checkQ(math.huge)
234+ checkQ(-math.huge)
235+ assert(string.format("%q", 0/0) == "(0/0)") -- NaN
236+ checkerror("no literal", string.format, "%q", {})
237+end
238+
239+assert(string.format("\0%s\0", "\0\0\1") == "\0\0\0\1\0")
240+checkerror("contains zeros", string.format, "%10s", "\0")
241+
242+-- format x tostring
243+assert(string.format("%s %s", nil, true) == "nil true")
244+assert(string.format("%s %.4s", false, true) == "false true")
245+assert(string.format("%.3s %.3s", false, true) == "fal tru")
246+local m = setmetatable({}, {__tostring = function () return "hello" end,
247+ __name = "hi"})
248+assert(string.format("%s %.10s", m, m) == "hello hello")
249+getmetatable(m).__tostring = nil -- will use '__name' from now on
250+assert(string.format("%.4s", m) == "hi: ")
251+
252+getmetatable(m).__tostring = function () return {} end
253+checkerror("'__tostring' must return a string", tostring, m)
254+
255+
256+assert(string.format("%x", 0.0) == "0")
257+assert(string.format("%02x", 0.0) == "00")
258+assert(string.format("%08X", 0xFFFFFFFF) == "FFFFFFFF")
259+assert(string.format("%+08d", 31501) == "+0031501")
260+assert(string.format("%+08d", -30927) == "-0030927")
261+
262+
263+do -- longest number that can be formatted
264+ local i = 1
265+ local j = 10000
266+ while i + 1 < j do -- binary search for maximum finite float
267+ local m = (i + j) // 2
268+ if 10^m < math.huge then i = m else j = m end
269+ end
270+ assert(10^i < math.huge and 10^j == math.huge)
271+ local s = string.format('%.99f', -(10^i))
272+ assert(string.len(s) >= i + 101)
273+ assert(tonumber(s) == -(10^i))
274+
275+ -- limit for floats
276+ assert(10^38 < math.huge)
277+ local s = string.format('%.99f', -(10^38))
278+ assert(string.len(s) >= 38 + 101)
279+ assert(tonumber(s) == -(10^38))
280+end
281+
282+
283+-- testing large numbers for format
284+do -- assume at least 32 bits
285+ local max, min = 0x7fffffff, -0x80000000 -- "large" for 32 bits
286+ assert(string.sub(string.format("%8x", -1), -8) == "ffffffff")
287+ assert(string.format("%x", max) == "7fffffff")
288+ assert(string.sub(string.format("%x", min), -8) == "80000000")
289+ assert(string.format("%d", max) == "2147483647")
290+ assert(string.format("%d", min) == "-2147483648")
291+ assert(string.format("%u", 0xffffffff) == "4294967295")
292+ assert(string.format("%o", 0xABCD) == "125715")
293+
294+ max, min = 0x7fffffffffffffff, -0x8000000000000000
295+ if max > 2.0^53 then -- only for 64 bits
296+ assert(string.format("%x", (2^52 | 0) - 1) == "fffffffffffff")
297+ assert(string.format("0x%8X", 0x8f000003) == "0x8F000003")
298+ assert(string.format("%d", 2^53) == "9007199254740992")
299+ assert(string.format("%i", -2^53) == "-9007199254740992")
300+ assert(string.format("%x", max) == "7fffffffffffffff")
301+ assert(string.format("%x", min) == "8000000000000000")
302+ assert(string.format("%d", max) == "9223372036854775807")
303+ assert(string.format("%d", min) == "-9223372036854775808")
304+ assert(string.format("%u", ~(-1 << 64)) == "18446744073709551615")
305+ assert(tostring(1234567890123) == '1234567890123')
306+ end
307+end
308+
309+
310+do print("testing 'format %a %A'")
311+ local function matchhexa (n)
312+ local s = string.format("%a", n)
313+ -- result matches ISO C requirements
314+ assert(string.find(s, "^%-?0x[1-9a-f]%.?[0-9a-f]*p[-+]?%d+$"))
315+ assert(tonumber(s) == n) -- and has full precision
316+ s = string.format("%A", n)
317+ assert(string.find(s, "^%-?0X[1-9A-F]%.?[0-9A-F]*P[-+]?%d+$"))
318+ assert(tonumber(s) == n)
319+ end
320+ for _, n in ipairs{0.1, -0.1, 1/3, -1/3, 1e30, -1e30,
321+ -45/247, 1, -1, 2, -2, 3e-20, -3e-20} do
322+ matchhexa(n)
323+ end
324+
325+ assert(string.find(string.format("%A", 0.0), "^0X0%.?0*P%+?0$"))
326+ assert(string.find(string.format("%a", -0.0), "^%-0x0%.?0*p%+?0$"))
327+
328+ if not _port then -- test inf, -inf, NaN, and -0.0
329+ assert(string.find(string.format("%a", 1/0), "^inf"))
330+ assert(string.find(string.format("%A", -1/0), "^%-INF"))
331+ assert(string.find(string.format("%a", 0/0), "^%-?nan"))
332+ assert(string.find(string.format("%a", -0.0), "^%-0x0"))
333+ end
334+
335+ if not pcall(string.format, "%.3a", 0) then
336+ (Message or print)("\n >>> modifiers for format '%a' not available <<<\n")
337+ else
338+ assert(string.find(string.format("%+.2A", 12), "^%+0X%x%.%x0P%+?%d$"))
339+ assert(string.find(string.format("%.4A", -12), "^%-0X%x%.%x000P%+?%d$"))
340+ end
341+end
342+
343+
344+-- testing some flags (all these results are required by ISO C)
345+assert(string.format("%#12o", 10) == " 012")
346+assert(string.format("%#10x", 100) == " 0x64")
347+assert(string.format("%#-17X", 100) == "0X64 ")
348+assert(string.format("%013i", -100) == "-000000000100")
349+assert(string.format("%2.5d", -100) == "-00100")
350+assert(string.format("%.u", 0) == "")
351+assert(string.format("%+#014.0f", 100) == "+000000000100.")
352+assert(string.format("%-16c", 97) == "a ")
353+assert(string.format("%+.3G", 1.5) == "+1.5")
354+assert(string.format("%.0s", "alo") == "")
355+assert(string.format("%.s", "alo") == "")
356+
357+-- ISO C89 says that "The exponent always contains at least two digits",
358+-- but unlike ISO C99 it does not ensure that it contains "only as many
359+-- more digits as necessary".
360+assert(string.match(string.format("% 1.0E", 100), "^ 1E%+0+2$"))
361+assert(string.match(string.format("% .1g", 2^10), "^ 1e%+0+3$"))
362+
363+
364+-- errors in format
365+
366+local function check (fmt, msg)
367+ checkerror(msg, string.format, fmt, 10)
368+end
369+
370+local aux = string.rep('0', 600)
371+check("%100.3d", "invalid conversion")
372+check("%1"..aux..".3d", "too long")
373+check("%1.100d", "invalid conversion")
374+check("%10.1"..aux.."004d", "too long")
375+check("%t", "invalid conversion")
376+check("%"..aux.."d", "too long")
377+check("%d %d", "no value")
378+check("%010c", "invalid conversion")
379+check("%.10c", "invalid conversion")
380+check("%0.34s", "invalid conversion")
381+check("%#i", "invalid conversion")
382+check("%3.1p", "invalid conversion")
383+check("%0.s", "invalid conversion")
384+check("%10q", "cannot have modifiers")
385+check("%F", "invalid conversion") -- useless and not in C89
386+
387+
388+assert(load("return 1\n--comment without ending EOL")() == 1)
389+
390+
391+checkerror("table expected", table.concat, 3)
392+checkerror("at index " .. maxi, table.concat, {}, " ", maxi, maxi)
393+-- '%' escapes following minus signal
394+checkerror("at index %" .. mini, table.concat, {}, " ", mini, mini)
395+assert(table.concat{} == "")
396+assert(table.concat({}, 'x') == "")
397+assert(table.concat({'\0', '\0\1', '\0\1\2'}, '.\0.') == "\0.\0.\0\1.\0.\0\1\2")
398+local a = {}; for i=1,300 do a[i] = "xuxu" end
399+assert(table.concat(a, "123").."123" == string.rep("xuxu123", 300))
400+assert(table.concat(a, "b", 20, 20) == "xuxu")
401+assert(table.concat(a, "", 20, 21) == "xuxuxuxu")
402+assert(table.concat(a, "x", 22, 21) == "")
403+assert(table.concat(a, "3", 299) == "xuxu3xuxu")
404+assert(table.concat({}, "x", maxi, maxi - 1) == "")
405+assert(table.concat({}, "x", mini + 1, mini) == "")
406+assert(table.concat({}, "x", maxi, mini) == "")
407+assert(table.concat({[maxi] = "alo"}, "x", maxi, maxi) == "alo")
408+assert(table.concat({[maxi] = "alo", [maxi - 1] = "y"}, "-", maxi - 1, maxi)
409+ == "y-alo")
410+
411+assert(not pcall(table.concat, {"a", "b", {}}))
412+
413+a = {"a","b","c"}
414+assert(table.concat(a, ",", 1, 0) == "")
415+assert(table.concat(a, ",", 1, 1) == "a")
416+assert(table.concat(a, ",", 1, 2) == "a,b")
417+assert(table.concat(a, ",", 2) == "b,c")
418+assert(table.concat(a, ",", 3) == "c")
419+assert(table.concat(a, ",", 4) == "")
420+
421+if not _port then
422+
423+ local locales = { "ptb", "pt_BR.iso88591", "ISO-8859-1" }
424+ local function trylocale (w)
425+ for i = 1, #locales do
426+ if os.setlocale(locales[i], w) then
427+ print(string.format("'%s' locale set to '%s'", w, locales[i]))
428+ return locales[i]
429+ end
430+ end
431+ print(string.format("'%s' locale not found", w))
432+ return false
433+ end
434+
435+ if trylocale("collate") then
436+ assert("alo" < "�lo" and "�lo" < "amo")
437+ end
438+
439+ if trylocale("ctype") then
440+ assert(string.gsub("�����", "%a", "x") == "xxxxx")
441+ assert(string.gsub("����", "%l", "x") == "x�x�")
442+ assert(string.gsub("����", "%u", "x") == "�x�x")
443+ assert(string.upper"���{xuxu}��o" == "���{XUXU}��O")
444+ end
445+
446+ os.setlocale("C")
447+ assert(os.setlocale() == 'C')
448+ assert(os.setlocale(nil, "numeric") == 'C')
449+
450+end
451+
452+
453+-- bug in Lua 5.3.2
454+-- 'gmatch' iterator does not work across coroutines
455+do
456+ local f = string.gmatch("1 2 3 4 5", "%d+")
457+ assert(f() == "1")
458+ local co = coroutine.wrap(f)
459+ assert(co() == "2")
460+end
461+
462+
463+if T==nil then
464+ (Message or print)
465+ ("\n >>> testC not active: skipping 'pushfstring' tests <<<\n")
466+else
467+
468+ print"testing 'pushfstring'"
469+
470+ -- formats %U, %f, %I already tested elsewhere
471+
472+ local blen = 200 -- internal buffer length in 'luaO_pushfstring'
473+
474+ local function callpfs (op, fmt, n)
475+ local x = {T.testC("pushfstring" .. op .. "; return *", fmt, n)}
476+ -- stack has code, 'fmt', 'n', and result from operation
477+ assert(#x == 4) -- make sure nothing else was left in the stack
478+ return x[4]
479+ end
480+
481+ local function testpfs (op, fmt, n)
482+ assert(callpfs(op, fmt, n) == string.format(fmt, n))
483+ end
484+
485+ testpfs("I", "", 0)
486+ testpfs("I", string.rep("a", blen - 1), 0)
487+ testpfs("I", string.rep("a", blen), 0)
488+ testpfs("I", string.rep("a", blen + 1), 0)
489+
490+ local str = string.rep("ab", blen) .. "%d" .. string.rep("d", blen / 2)
491+ testpfs("I", str, 2^14)
492+ testpfs("I", str, -2^15)
493+
494+ str = "%d" .. string.rep("cd", blen)
495+ testpfs("I", str, 2^14)
496+ testpfs("I", str, -2^15)
497+
498+ str = string.rep("c", blen - 2) .. "%d"
499+ testpfs("I", str, 2^14)
500+ testpfs("I", str, -2^15)
501+
502+ for l = 12, 14 do
503+ local str1 = string.rep("a", l)
504+ for i = 0, 500, 13 do
505+ for j = 0, 500, 13 do
506+ str = string.rep("a", i) .. "%s" .. string.rep("d", j)
507+ testpfs("S", str, str1)
508+ testpfs("S", str, str)
509+ end
510+ end
511+ end
512+
513+ str = "abc %c def"
514+ testpfs("I", str, string.byte("A"))
515+ testpfs("I", str, 255)
516+
517+ str = string.rep("a", blen - 1) .. "%p" .. string.rep("cd", blen)
518+ testpfs("P", str, {})
519+
520+ str = string.rep("%%", 3 * blen) .. "%p" .. string.rep("%%", 2 * blen)
521+ testpfs("P", str, {})
522+end
523+
524+
525+print('OK')
526+
zod/luau/tests/tpack.luanew file+322
@@ -0,0 +1,322 @@
1+-- $Id: testes/tpack.lua $
2+-- See Copyright Notice in file all.lua
3+
4+local pack = string.pack
5+local packsize = string.packsize
6+local unpack = string.unpack
7+
8+print "testing pack/unpack"
9+
10+-- maximum size for integers
11+local NB = 16
12+
13+local sizeshort = packsize("h")
14+local sizeint = packsize("i")
15+local sizelong = packsize("l")
16+local sizesize_t = packsize("T")
17+local sizeLI = packsize("j")
18+local sizefloat = packsize("f")
19+local sizedouble = packsize("d")
20+local sizenumber = packsize("n")
21+local little = (pack("i2", 1) == "\1\0")
22+local align = packsize("!xXi16")
23+
24+assert(1 <= sizeshort and sizeshort <= sizeint and sizeint <= sizelong and
25+ sizefloat <= sizedouble)
26+
27+print("platform:")
28+print(string.format(
29+ "\tshort %d, int %d, long %d, size_t %d, float %d, double %d,\n\z
30+ \tlua Integer %d, lua Number %d",
31+ sizeshort, sizeint, sizelong, sizesize_t, sizefloat, sizedouble,
32+ sizeLI, sizenumber))
33+print("\t" .. (little and "little" or "big") .. " endian")
34+print("\talignment: " .. align)
35+
36+
37+-- check errors in arguments
38+local function checkerror (msg, f, ...)
39+ local status, err = pcall(f, ...)
40+ -- print(status, err, msg)
41+ assert(not status and string.find(err, msg))
42+end
43+
44+-- minimum behavior for integer formats
45+assert(unpack("B", pack("B", 0xff)) == 0xff)
46+assert(unpack("b", pack("b", 0x7f)) == 0x7f)
47+assert(unpack("b", pack("b", -0x80)) == -0x80)
48+
49+assert(unpack("H", pack("H", 0xffff)) == 0xffff)
50+assert(unpack("h", pack("h", 0x7fff)) == 0x7fff)
51+assert(unpack("h", pack("h", -0x8000)) == -0x8000)
52+
53+assert(unpack("L", pack("L", 0xffffffff)) == 0xffffffff)
54+assert(unpack("l", pack("l", 0x7fffffff)) == 0x7fffffff)
55+assert(unpack("l", pack("l", -0x80000000)) == -0x80000000)
56+
57+
58+for i = 1, NB do
59+ -- small numbers with signal extension ("\xFF...")
60+ local s = string.rep("\xff", i)
61+ assert(pack("i" .. i, -1) == s)
62+ assert(packsize("i" .. i) == #s)
63+ assert(unpack("i" .. i, s) == -1)
64+
65+ -- small unsigned number ("\0...\xAA")
66+ s = "\xAA" .. string.rep("\0", i - 1)
67+ assert(pack("<I" .. i, 0xAA) == s)
68+ assert(unpack("<I" .. i, s) == 0xAA)
69+ assert(pack(">I" .. i, 0xAA) == s:reverse())
70+ assert(unpack(">I" .. i, s:reverse()) == 0xAA)
71+end
72+
73+do
74+ local lnum = 0x13121110090807060504030201
75+ local s = pack("<j", lnum)
76+ assert(unpack("<j", s) == lnum)
77+ assert(unpack("<i" .. sizeLI + 1, s .. "\0") == lnum)
78+ assert(unpack("<i" .. sizeLI + 1, s .. "\0") == lnum)
79+
80+ for i = sizeLI + 1, NB do
81+ local s = pack("<j", -lnum)
82+ assert(unpack("<j", s) == -lnum)
83+ -- strings with (correct) extra bytes
84+ assert(unpack("<i" .. i, s .. ("\xFF"):rep(i - sizeLI)) == -lnum)
85+ assert(unpack(">i" .. i, ("\xFF"):rep(i - sizeLI) .. s:reverse()) == -lnum)
86+ assert(unpack("<I" .. i, s .. ("\0"):rep(i - sizeLI)) == -lnum)
87+
88+ -- overflows
89+ checkerror("does not fit", unpack, "<I" .. i, ("\x00"):rep(i - 1) .. "\1")
90+ checkerror("does not fit", unpack, ">i" .. i, "\1" .. ("\x00"):rep(i - 1))
91+ end
92+end
93+
94+for i = 1, sizeLI do
95+ local lstr = "\1\2\3\4\5\6\7\8\9\10\11\12\13"
96+ local lnum = 0x13121110090807060504030201
97+ local n = lnum & (~(-1 << (i * 8)))
98+ local s = string.sub(lstr, 1, i)
99+ assert(pack("<i" .. i, n) == s)
100+ assert(pack(">i" .. i, n) == s:reverse())
101+ assert(unpack(">i" .. i, s:reverse()) == n)
102+end
103+
104+-- sign extension
105+do
106+ local u = 0xf0
107+ for i = 1, sizeLI - 1 do
108+ assert(unpack("<i"..i, "\xf0"..("\xff"):rep(i - 1)) == -16)
109+ assert(unpack(">I"..i, "\xf0"..("\xff"):rep(i - 1)) == u)
110+ u = u * 256 + 0xff
111+ end
112+end
113+
114+-- mixed endianness
115+do
116+ assert(pack(">i2 <i2", 10, 20) == "\0\10\20\0")
117+ local a, b = unpack("<i2 >i2", "\10\0\0\20")
118+ assert(a == 10 and b == 20)
119+ assert(pack("=i4", 2001) == pack("i4", 2001))
120+end
121+
122+print("testing invalid formats")
123+
124+checkerror("out of limits", pack, "i0", 0)
125+checkerror("out of limits", pack, "i" .. NB + 1, 0)
126+checkerror("out of limits", pack, "!" .. NB + 1, 0)
127+checkerror("%(17%) out of limits %[1,16%]", pack, "Xi" .. NB + 1)
128+checkerror("invalid format option 'r'", pack, "i3r", 0)
129+checkerror("16%-byte integer", unpack, "i16", string.rep('\3', 16))
130+checkerror("not power of 2", pack, "!4i3", 0);
131+checkerror("missing size", pack, "c", "")
132+checkerror("variable%-length format", packsize, "s")
133+checkerror("variable%-length format", packsize, "z")
134+
135+-- overflow in option size (error will be in digit after limit)
136+checkerror("invalid format", packsize, "c1" .. string.rep("0", 40))
137+
138+if packsize("i") == 4 then
139+ -- result would be 2^31 (2^3 repetitions of 2^28 strings)
140+ local s = string.rep("c268435456", 2^3)
141+ checkerror("too large", packsize, s)
142+ -- one less is OK
143+ s = string.rep("c268435456", 2^3 - 1) .. "c268435455"
144+ assert(packsize(s) == 0x7fffffff)
145+end
146+
147+-- overflow in packing
148+for i = 1, sizeLI - 1 do
149+ local umax = (1 << (i * 8)) - 1
150+ local max = umax >> 1
151+ local min = ~max
152+ checkerror("overflow", pack, "<I" .. i, -1)
153+ checkerror("overflow", pack, "<I" .. i, min)
154+ checkerror("overflow", pack, ">I" .. i, umax + 1)
155+
156+ checkerror("overflow", pack, ">i" .. i, umax)
157+ checkerror("overflow", pack, ">i" .. i, max + 1)
158+ checkerror("overflow", pack, "<i" .. i, min - 1)
159+
160+ assert(unpack(">i" .. i, pack(">i" .. i, max)) == max)
161+ assert(unpack("<i" .. i, pack("<i" .. i, min)) == min)
162+ assert(unpack(">I" .. i, pack(">I" .. i, umax)) == umax)
163+end
164+
165+-- Lua integer size
166+assert(unpack(">j", pack(">j", math.maxinteger)) == math.maxinteger)
167+assert(unpack("<j", pack("<j", math.mininteger)) == math.mininteger)
168+assert(unpack("<J", pack("<j", -1)) == -1) -- maximum unsigned integer
169+
170+if little then
171+ assert(pack("f", 24) == pack("<f", 24))
172+else
173+ assert(pack("f", 24) == pack(">f", 24))
174+end
175+
176+print "testing pack/unpack of floating-point numbers"
177+
178+for _, n in ipairs{0, -1.1, 1.9, 1/0, -1/0, 1e20, -1e20, 0.1, 2000.7} do
179+ assert(unpack("n", pack("n", n)) == n)
180+ assert(unpack("<n", pack("<n", n)) == n)
181+ assert(unpack(">n", pack(">n", n)) == n)
182+ assert(pack("<f", n) == pack(">f", n):reverse())
183+ assert(pack(">d", n) == pack("<d", n):reverse())
184+end
185+
186+-- for non-native precisions, test only with "round" numbers
187+for _, n in ipairs{0, -1.5, 1/0, -1/0, 1e10, -1e9, 0.5, 2000.25} do
188+ assert(unpack("<f", pack("<f", n)) == n)
189+ assert(unpack(">f", pack(">f", n)) == n)
190+ assert(unpack("<d", pack("<d", n)) == n)
191+ assert(unpack(">d", pack(">d", n)) == n)
192+end
193+
194+print "testing pack/unpack of strings"
195+do
196+ local s = string.rep("abc", 1000)
197+ assert(pack("zB", s, 247) == s .. "\0\xF7")
198+ local s1, b = unpack("zB", s .. "\0\xF9")
199+ assert(b == 249 and s1 == s)
200+ s1 = pack("s", s)
201+ assert(unpack("s", s1) == s)
202+
203+ checkerror("does not fit", pack, "s1", s)
204+
205+ checkerror("contains zeros", pack, "z", "alo\0");
206+
207+ checkerror("unfinished string", unpack, "zc10000000", "alo")
208+
209+ for i = 2, NB do
210+ local s1 = pack("s" .. i, s)
211+ assert(unpack("s" .. i, s1) == s and #s1 == #s + i)
212+ end
213+end
214+
215+do
216+ local x = pack("s", "alo")
217+ checkerror("too short", unpack, "s", x:sub(1, -2))
218+ checkerror("too short", unpack, "c5", "abcd")
219+ checkerror("out of limits", pack, "s100", "alo")
220+end
221+
222+do
223+ assert(pack("c0", "") == "")
224+ assert(packsize("c0") == 0)
225+ assert(unpack("c0", "") == "")
226+ assert(pack("<! c3", "abc") == "abc")
227+ assert(packsize("<! c3") == 3)
228+ assert(pack(">!4 c6", "abcdef") == "abcdef")
229+ assert(pack("c3", "123") == "123")
230+ assert(pack("c0", "") == "")
231+ assert(pack("c8", "123456") == "123456\0\0")
232+ assert(pack("c88", "") == string.rep("\0", 88))
233+ assert(pack("c188", "ab") == "ab" .. string.rep("\0", 188 - 2))
234+ local a, b, c = unpack("!4 z c3", "abcdefghi\0xyz")
235+ assert(a == "abcdefghi" and b == "xyz" and c == 14)
236+ checkerror("longer than", pack, "c3", "1234")
237+end
238+
239+
240+-- testing multiple types and sequence
241+do
242+ local x = pack("<b h b f d f n i", 1, 2, 3, 4, 5, 6, 7, 8)
243+ assert(#x == packsize("<b h b f d f n i"))
244+ local a, b, c, d, e, f, g, h = unpack("<b h b f d f n i", x)
245+ assert(a == 1 and b == 2 and c == 3 and d == 4 and e == 5 and f == 6 and
246+ g == 7 and h == 8)
247+end
248+
249+print "testing alignment"
250+do
251+ assert(pack(" < i1 i2 ", 2, 3) == "\2\3\0") -- no alignment by default
252+ local x = pack(">!8 b Xh i4 i8 c1 Xi8", -12, 100, 200, "\xEC")
253+ assert(#x == packsize(">!8 b Xh i4 i8 c1 Xi8"))
254+ assert(x == "\xf4" .. "\0\0\0" ..
255+ "\0\0\0\100" ..
256+ "\0\0\0\0\0\0\0\xC8" ..
257+ "\xEC" .. "\0\0\0\0\0\0\0")
258+ local a, b, c, d, pos = unpack(">!8 c1 Xh i4 i8 b Xi8 XI XH", x)
259+ assert(a == "\xF4" and b == 100 and c == 200 and d == -20 and (pos - 1) == #x)
260+
261+ x = pack(">!4 c3 c4 c2 z i4 c5 c2 Xi4",
262+ "abc", "abcd", "xz", "hello", 5, "world", "xy")
263+ assert(x == "abcabcdxzhello\0\0\0\0\0\5worldxy\0")
264+ local a, b, c, d, e, f, g, pos = unpack(">!4 c3 c4 c2 z i4 c5 c2 Xh Xi4", x)
265+ assert(a == "abc" and b == "abcd" and c == "xz" and d == "hello" and
266+ e == 5 and f == "world" and g == "xy" and (pos - 1) % 4 == 0)
267+
268+ x = pack(" b b Xd b Xb x", 1, 2, 3)
269+ assert(packsize(" b b Xd b Xb x") == 4)
270+ assert(x == "\1\2\3\0")
271+ a, b, c, pos = unpack("bbXdb", x)
272+ assert(a == 1 and b == 2 and c == 3 and pos == #x)
273+
274+ -- only alignment
275+ assert(packsize("!8 xXi8") == 8)
276+ local pos = unpack("!8 xXi8", "0123456701234567"); assert(pos == 9)
277+ assert(packsize("!8 xXi2") == 2)
278+ local pos = unpack("!8 xXi2", "0123456701234567"); assert(pos == 3)
279+ assert(packsize("!2 xXi2") == 2)
280+ local pos = unpack("!2 xXi2", "0123456701234567"); assert(pos == 3)
281+ assert(packsize("!2 xXi8") == 2)
282+ local pos = unpack("!2 xXi8", "0123456701234567"); assert(pos == 3)
283+ assert(packsize("!16 xXi16") == 16)
284+ local pos = unpack("!16 xXi16", "0123456701234567"); assert(pos == 17)
285+
286+ checkerror("invalid next option", pack, "X")
287+ checkerror("invalid next option", unpack, "XXi", "")
288+ checkerror("invalid next option", unpack, "X i", "")
289+ checkerror("invalid next option", pack, "Xc1")
290+end
291+
292+do -- testing initial position
293+ local x = pack("i4i4i4i4", 1, 2, 3, 4)
294+ for pos = 1, 16, 4 do
295+ local i, p = unpack("i4", x, pos)
296+ assert(i == pos//4 + 1 and p == pos + 4)
297+ end
298+
299+ -- with alignment
300+ for pos = 0, 12 do -- will always round position to power of 2
301+ local i, p = unpack("!4 i4", x, pos + 1)
302+ assert(i == (pos + 3)//4 + 1 and p == i*4 + 1)
303+ end
304+
305+ -- negative indices
306+ local i, p = unpack("!4 i4", x, -4)
307+ assert(i == 4 and p == 17)
308+ local i, p = unpack("!4 i4", x, -7)
309+ assert(i == 4 and p == 17)
310+ local i, p = unpack("!4 i4", x, -#x)
311+ assert(i == 1 and p == 5)
312+
313+ -- limits
314+ for i = 1, #x + 1 do
315+ assert(unpack("c0", x, i) == "")
316+ end
317+ checkerror("out of string", unpack, "c0", x, #x + 2)
318+
319+end
320+
321+print "OK"
322+
zod/luau/tests/tracegc.luanew file+40
@@ -0,0 +1,40 @@
1+-- track collections
2+
3+local M = {}
4+
5+-- import list
6+local setmetatable, stderr, collectgarbage =
7+ setmetatable, io.stderr, collectgarbage
8+
9+_ENV = nil
10+
11+local active = false
12+
13+
14+-- each time a table is collected, remark it for finalization on next
15+-- cycle
16+local mt = {}
17+function mt.__gc (o)
18+ stderr:write'.' -- mark progress
19+ if active then
20+ setmetatable(o, mt) -- remark object for finalization
21+ end
22+end
23+
24+
25+function M.start ()
26+ if not active then
27+ active = true
28+ setmetatable({}, mt) -- create initial object
29+ end
30+end
31+
32+
33+function M.stop ()
34+ if active then
35+ active = false
36+ collectgarbage() -- call finalizer for the last time
37+ end
38+end
39+
40+return M
zod/luau/tests/utf8.luanew file+259
@@ -0,0 +1,259 @@
1+-- $Id: testes/utf8.lua $
2+-- See Copyright Notice in file all.lua
3+
4+-- UTF-8 file
5+
6+print "testing UTF-8 library"
7+
8+local utf8 = require'utf8'
9+
10+
11+local function checkerror (msg, f, ...)
12+ local s, err = pcall(f, ...)
13+ assert(not s and string.find(err, msg))
14+end
15+
16+
17+local function len (s)
18+ return #string.gsub(s, "[\x80-\xBF]", "")
19+end
20+
21+
22+local justone = "^" .. utf8.charpattern .. "$"
23+
24+-- 't' is the list of codepoints of 's'
25+local function checksyntax (s, t)
26+ -- creates a string "return '\u{t[1]}...\u{t[n]}'"
27+ local ts = {"return '"}
28+ for i = 1, #t do ts[i + 1] = string.format("\\u{%x}", t[i]) end
29+ ts[#t + 2] = "'"
30+ ts = table.concat(ts)
31+ -- its execution should result in 's'
32+ assert(assert(load(ts))() == s)
33+end
34+
35+assert(not utf8.offset("alo", 5))
36+assert(not utf8.offset("alo", -4))
37+
38+-- 'check' makes several tests over the validity of string 's'.
39+-- 't' is the list of codepoints of 's'.
40+local function check (s, t, nonstrict)
41+ local l = utf8.len(s, 1, -1, nonstrict)
42+ assert(#t == l and len(s) == l)
43+ assert(utf8.char(table.unpack(t)) == s) -- 't' and 's' are equivalent
44+
45+ assert(utf8.offset(s, 0) == 1)
46+
47+ checksyntax(s, t)
48+
49+ -- creates new table with all codepoints of 's'
50+ local t1 = {utf8.codepoint(s, 1, -1, nonstrict)}
51+ assert(#t == #t1)
52+ for i = 1, #t do assert(t[i] == t1[i]) end -- 't' is equal to 't1'
53+
54+ for i = 1, l do -- for all codepoints
55+ local pi = utf8.offset(s, i) -- position of i-th char
56+ local pi1 = utf8.offset(s, 2, pi) -- position of next char
57+ assert(string.find(string.sub(s, pi, pi1 - 1), justone))
58+ assert(utf8.offset(s, -1, pi1) == pi)
59+ assert(utf8.offset(s, i - l - 1) == pi)
60+ assert(pi1 - pi == #utf8.char(utf8.codepoint(s, pi, pi, nonstrict)))
61+ for j = pi, pi1 - 1 do
62+ assert(utf8.offset(s, 0, j) == pi)
63+ end
64+ for j = pi + 1, pi1 - 1 do
65+ assert(not utf8.len(s, j))
66+ end
67+ assert(utf8.len(s, pi, pi, nonstrict) == 1)
68+ assert(utf8.len(s, pi, pi1 - 1, nonstrict) == 1)
69+ assert(utf8.len(s, pi, -1, nonstrict) == l - i + 1)
70+ assert(utf8.len(s, pi1, -1, nonstrict) == l - i)
71+ assert(utf8.len(s, 1, pi, nonstrict) == i)
72+ end
73+
74+ local i = 0
75+ for p, c in utf8.codes(s, nonstrict) do
76+ i = i + 1
77+ assert(c == t[i] and p == utf8.offset(s, i))
78+ assert(utf8.codepoint(s, p, p, nonstrict) == c)
79+ end
80+ assert(i == #t)
81+
82+ i = 0
83+ for c in string.gmatch(s, utf8.charpattern) do
84+ i = i + 1
85+ assert(c == utf8.char(t[i]))
86+ end
87+ assert(i == #t)
88+
89+ for i = 1, l do
90+ assert(utf8.offset(s, i) == utf8.offset(s, i - l - 1, #s + 1))
91+ end
92+
93+end
94+
95+
96+do -- error indication in utf8.len
97+ local function check (s, p)
98+ local a, b = utf8.len(s)
99+ assert(not a and b == p)
100+ end
101+ check("abc\xE3def", 4)
102+ check("\xF4\x9F\xBF", 1)
103+ check("\xF4\x9F\xBF\xBF", 1)
104+ -- spurious continuation bytes
105+ check("汉字\x80", #("汉字") + 1)
106+ check("\x80hello", 1)
107+ check("hel\x80lo", 4)
108+ check("汉字\xBF", #("汉字") + 1)
109+ check("\xBFhello", 1)
110+ check("hel\xBFlo", 4)
111+end
112+
113+-- errors in utf8.codes
114+do
115+ local function errorcodes (s)
116+ checkerror("invalid UTF%-8 code",
117+ function ()
118+ for c in utf8.codes(s) do assert(c) end
119+ end)
120+ end
121+ errorcodes("ab\xff")
122+ errorcodes("\u{110000}")
123+ errorcodes("in\x80valid")
124+ errorcodes("\xbfinvalid")
125+ errorcodes("αλφ\xBFα")
126+
127+ -- calling interation function with invalid arguments
128+ local f = utf8.codes("")
129+ assert(f("", 2) == nil)
130+ assert(f("", -1) == nil)
131+ assert(f("", math.mininteger) == nil)
132+
133+end
134+
135+-- error in initial position for offset
136+checkerror("position out of bounds", utf8.offset, "abc", 1, 5)
137+checkerror("position out of bounds", utf8.offset, "abc", 1, -4)
138+checkerror("position out of bounds", utf8.offset, "", 1, 2)
139+checkerror("position out of bounds", utf8.offset, "", 1, -1)
140+checkerror("continuation byte", utf8.offset, "𦧺", 1, 2)
141+checkerror("continuation byte", utf8.offset, "𦧺", 1, 2)
142+checkerror("continuation byte", utf8.offset, "\x80", 1)
143+
144+-- error in indices for len
145+checkerror("out of bounds", utf8.len, "abc", 0, 2)
146+checkerror("out of bounds", utf8.len, "abc", 1, 4)
147+
148+
149+local s = "hello World"
150+local t = {string.byte(s, 1, -1)}
151+for i = 1, utf8.len(s) do assert(t[i] == string.byte(s, i)) end
152+check(s, t)
153+
154+check("汉字/漢字", {27721, 23383, 47, 28450, 23383,})
155+
156+do
157+ local s = "áéí\128"
158+ local t = {utf8.codepoint(s,1,#s - 1)}
159+ assert(#t == 3 and t[1] == 225 and t[2] == 233 and t[3] == 237)
160+ checkerror("invalid UTF%-8 code", utf8.codepoint, s, 1, #s)
161+ checkerror("out of bounds", utf8.codepoint, s, #s + 1)
162+ t = {utf8.codepoint(s, 4, 3)}
163+ assert(#t == 0)
164+ checkerror("out of bounds", utf8.codepoint, s, -(#s + 1), 1)
165+ checkerror("out of bounds", utf8.codepoint, s, 1, #s + 1)
166+ -- surrogates
167+ assert(utf8.codepoint("\u{D7FF}") == 0xD800 - 1)
168+ assert(utf8.codepoint("\u{E000}") == 0xDFFF + 1)
169+ assert(utf8.codepoint("\u{D800}", 1, 1, true) == 0xD800)
170+ assert(utf8.codepoint("\u{DFFF}", 1, 1, true) == 0xDFFF)
171+ assert(utf8.codepoint("\u{7FFFFFFF}", 1, 1, true) == 0x7FFFFFFF)
172+end
173+
174+assert(utf8.char() == "")
175+assert(utf8.char(0, 97, 98, 99, 1) == "\0abc\1")
176+
177+assert(utf8.codepoint(utf8.char(0x10FFFF)) == 0x10FFFF)
178+assert(utf8.codepoint(utf8.char(0x7FFFFFFF), 1, 1, true) == (1<<31) - 1)
179+
180+checkerror("value out of range", utf8.char, 0x7FFFFFFF + 1)
181+checkerror("value out of range", utf8.char, -1)
182+
183+local function invalid (s)
184+ checkerror("invalid UTF%-8 code", utf8.codepoint, s)
185+ assert(not utf8.len(s))
186+end
187+
188+-- UTF-8 representation for 0x11ffff (value out of valid range)
189+invalid("\xF4\x9F\xBF\xBF")
190+
191+-- surrogates
192+invalid("\u{D800}")
193+invalid("\u{DFFF}")
194+
195+-- overlong sequences
196+invalid("\xC0\x80") -- zero
197+invalid("\xC1\xBF") -- 0x7F (should be coded in 1 byte)
198+invalid("\xE0\x9F\xBF") -- 0x7FF (should be coded in 2 bytes)
199+invalid("\xF0\x8F\xBF\xBF") -- 0xFFFF (should be coded in 3 bytes)
200+
201+
202+-- invalid bytes
203+invalid("\x80") -- continuation byte
204+invalid("\xBF") -- continuation byte
205+invalid("\xFE") -- invalid byte
206+invalid("\xFF") -- invalid byte
207+
208+
209+-- empty string
210+check("", {})
211+
212+-- minimum and maximum values for each sequence size
213+s = "\0 \x7F\z
214+ \xC2\x80 \xDF\xBF\z
215+ \xE0\xA0\x80 \xEF\xBF\xBF\z
216+ \xF0\x90\x80\x80 \xF4\x8F\xBF\xBF"
217+s = string.gsub(s, " ", "")
218+check(s, {0,0x7F, 0x80,0x7FF, 0x800,0xFFFF, 0x10000,0x10FFFF})
219+
220+do
221+ -- original UTF-8 values
222+ local s = "\u{4000000}\u{7FFFFFFF}"
223+ assert(#s == 12)
224+ check(s, {0x4000000, 0x7FFFFFFF}, true)
225+
226+ s = "\u{200000}\u{3FFFFFF}"
227+ assert(#s == 10)
228+ check(s, {0x200000, 0x3FFFFFF}, true)
229+
230+ s = "\u{10000}\u{1fffff}"
231+ assert(#s == 8)
232+ check(s, {0x10000, 0x1FFFFF}, true)
233+end
234+
235+local x = "日本語a-4\0éó"
236+check(x, {26085, 26412, 35486, 97, 45, 52, 0, 233, 243})
237+
238+
239+-- Supplementary Characters
240+check("𣲷𠜎𠱓𡁻𠵼ab𠺢",
241+ {0x23CB7, 0x2070E, 0x20C53, 0x2107B, 0x20D7C, 0x61, 0x62, 0x20EA2,})
242+
243+check("𨳊𩶘𦧺𨳒𥄫𤓓\xF4\x8F\xBF\xBF",
244+ {0x28CCA, 0x29D98, 0x269FA, 0x28CD2, 0x2512B, 0x244D3, 0x10ffff})
245+
246+
247+local i = 0
248+for p, c in string.gmatch(x, "()(" .. utf8.charpattern .. ")") do
249+ i = i + 1
250+ assert(utf8.offset(x, i) == p)
251+ assert(utf8.len(x, p) == utf8.len(x) - i + 1)
252+ assert(utf8.len(c) == 1)
253+ for j = 1, #c - 1 do
254+ assert(utf8.offset(x, 0, p + j - 1) == p)
255+ end
256+end
257+
258+print'ok'
259+
zod/luau/tests/vararg.luanew file+151
@@ -0,0 +1,151 @@
1+-- $Id: testes/vararg.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print('testing vararg')
5+
6+local function f (a, ...)
7+ local x = {n = select('#', ...), ...}
8+ for i = 1, x.n do assert(a[i] == x[i]) end
9+ return x.n
10+end
11+
12+local function c12 (...)
13+ assert(arg == _G.arg) -- no local 'arg'
14+ local x = {...}; x.n = #x
15+ local res = (x.n==2 and x[1] == 1 and x[2] == 2)
16+ if res then res = 55 end
17+ return res, 2
18+end
19+
20+local function vararg (...) return {n = select('#', ...), ...} end
21+
22+local call = function (f, args) return f(table.unpack(args, 1, args.n)) end
23+
24+assert(f() == 0)
25+assert(f({1,2,3}, 1, 2, 3) == 3)
26+assert(f({"alo", nil, 45, f, nil}, "alo", nil, 45, f, nil) == 5)
27+
28+assert(vararg().n == 0)
29+assert(vararg(nil, nil).n == 2)
30+
31+assert(c12(1,2)==55)
32+local a,b = assert(call(c12, {1,2}))
33+assert(a == 55 and b == 2)
34+a = call(c12, {1,2;n=2})
35+assert(a == 55 and b == 2)
36+a = call(c12, {1,2;n=1})
37+assert(not a)
38+assert(c12(1,2,3) == false)
39+local a = vararg(call(next, {_G,nil;n=2}))
40+local b,c = next(_G)
41+assert(a[1] == b and a[2] == c and a.n == 2)
42+a = vararg(call(call, {c12, {1,2}}))
43+assert(a.n == 2 and a[1] == 55 and a[2] == 2)
44+a = call(print, {'+'})
45+assert(a == nil)
46+
47+local t = {1, 10}
48+function t:f (...) local arg = {...}; return self[...]+#arg end
49+assert(t:f(1,4) == 3 and t:f(2) == 11)
50+print('+')
51+
52+local lim = 20
53+local i, a = 1, {}
54+while i <= lim do a[i] = i+0.3; i=i+1 end
55+
56+function f(a, b, c, d, ...)
57+ local more = {...}
58+ assert(a == 1.3 and more[1] == 5.3 and
59+ more[lim-4] == lim+0.3 and not more[lim-3])
60+end
61+
62+local function g (a,b,c)
63+ assert(a == 1.3 and b == 2.3 and c == 3.3)
64+end
65+
66+call(f, a)
67+call(g, a)
68+
69+a = {}
70+i = 1
71+while i <= lim do a[i] = i; i=i+1 end
72+assert(call(math.max, a) == lim)
73+
74+print("+")
75+
76+
77+-- new-style varargs
78+
79+local function oneless (a, ...) return ... end
80+
81+function f (n, a, ...)
82+ local b
83+ assert(arg == _G.arg) -- no local 'arg'
84+ if n == 0 then
85+ local b, c, d = ...
86+ return a, b, c, d, oneless(oneless(oneless(...)))
87+ else
88+ n, b, a = n-1, ..., a
89+ assert(b == ...)
90+ return f(n, a, ...)
91+ end
92+end
93+
94+a,b,c,d,e = assert(f(10,5,4,3,2,1))
95+assert(a==5 and b==4 and c==3 and d==2 and e==1)
96+
97+a,b,c,d,e = f(4)
98+assert(a==nil and b==nil and c==nil and d==nil and e==nil)
99+
100+
101+-- varargs for main chunks
102+local f = load[[ return {...} ]]
103+local x = f(2,3)
104+assert(x[1] == 2 and x[2] == 3 and x[3] == undef)
105+
106+
107+f = load[[
108+ local x = {...}
109+ for i=1,select('#', ...) do assert(x[i] == select(i, ...)) end
110+ assert(x[select('#', ...)+1] == undef)
111+ return true
112+]]
113+
114+assert(f("a", "b", nil, {}, assert))
115+assert(f())
116+
117+a = {select(3, table.unpack{10,20,30,40})}
118+assert(#a == 2 and a[1] == 30 and a[2] == 40)
119+a = {select(1)}
120+assert(next(a) == nil)
121+a = {select(-1, 3, 5, 7)}
122+assert(a[1] == 7 and a[2] == undef)
123+a = {select(-2, 3, 5, 7)}
124+assert(a[1] == 5 and a[2] == 7 and a[3] == undef)
125+pcall(select, 10000)
126+pcall(select, -10000)
127+
128+
129+-- bug in 5.2.2
130+
131+function f(p1, p2, p3, p4, p5, p6, p7, p8, p9, p10,
132+p11, p12, p13, p14, p15, p16, p17, p18, p19, p20,
133+p21, p22, p23, p24, p25, p26, p27, p28, p29, p30,
134+p31, p32, p33, p34, p35, p36, p37, p38, p39, p40,
135+p41, p42, p43, p44, p45, p46, p48, p49, p50, ...)
136+ local a1,a2,a3,a4,a5,a6,a7
137+ local a8,a9,a10,a11,a12,a13,a14
138+end
139+
140+-- assertion fail here
141+f()
142+
143+-- missing arguments in tail call
144+do
145+ local function f(a,b,c) return c, b end
146+ local function g() return f(1,2) end
147+ local a, b = g()
148+ assert(a == nil and b == 2)
149+end
150+print('OK')
151+
zod/luau/tests/verybig.luanew file+152
@@ -0,0 +1,152 @@
1+-- $Id: testes/verybig.lua $
2+-- See Copyright Notice in file all.lua
3+
4+print "testing RK"
5+
6+-- testing opcodes with RK arguments larger than K limit
7+local function foo ()
8+ local dummy = {
9+ -- fill first 256 entries in table of constants
10+ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
11+ 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32,
12+ 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48,
13+ 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64,
14+ 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80,
15+ 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96,
16+ 97, 98, 99, 100, 101, 102, 103, 104,
17+ 105, 106, 107, 108, 109, 110, 111, 112,
18+ 113, 114, 115, 116, 117, 118, 119, 120,
19+ 121, 122, 123, 124, 125, 126, 127, 128,
20+ 129, 130, 131, 132, 133, 134, 135, 136,
21+ 137, 138, 139, 140, 141, 142, 143, 144,
22+ 145, 146, 147, 148, 149, 150, 151, 152,
23+ 153, 154, 155, 156, 157, 158, 159, 160,
24+ 161, 162, 163, 164, 165, 166, 167, 168,
25+ 169, 170, 171, 172, 173, 174, 175, 176,
26+ 177, 178, 179, 180, 181, 182, 183, 184,
27+ 185, 186, 187, 188, 189, 190, 191, 192,
28+ 193, 194, 195, 196, 197, 198, 199, 200,
29+ 201, 202, 203, 204, 205, 206, 207, 208,
30+ 209, 210, 211, 212, 213, 214, 215, 216,
31+ 217, 218, 219, 220, 221, 222, 223, 224,
32+ 225, 226, 227, 228, 229, 230, 231, 232,
33+ 233, 234, 235, 236, 237, 238, 239, 240,
34+ 241, 242, 243, 244, 245, 246, 247, 248,
35+ 249, 250, 251, 252, 253, 254, 255, 256,
36+ }
37+ assert(24.5 + 0.6 == 25.1)
38+ local t = {foo = function (self, x) return x + self.x end, x = 10}
39+ t.t = t
40+ assert(t:foo(1.5) == 11.5)
41+ assert(t.t:foo(0.5) == 10.5) -- bug in 5.2 alpha
42+ assert(24.3 == 24.3)
43+ assert((function () return t.x end)() == 10)
44+end
45+
46+
47+foo()
48+foo = nil
49+
50+if _soft then return 10 end
51+
52+print "testing large programs (>64k)"
53+
54+-- template to create a very big test file
55+local prog = [[$
56+
57+local a,b
58+
59+b = {$1$
60+ b30009 = 65534,
61+ b30010 = 65535,
62+ b30011 = 65536,
63+ b30012 = 65537,
64+ b30013 = 16777214,
65+ b30014 = 16777215,
66+ b30015 = 16777216,
67+ b30016 = 16777217,
68+ b30017 = 0x7fffff,
69+ b30018 = -0x7fffff,
70+ b30019 = 0x1ffffff,
71+ b30020 = -0x1ffffd,
72+ b30021 = -65534,
73+ b30022 = -65535,
74+ b30023 = -65536,
75+ b30024 = -0xffffff,
76+ b30025 = 15012.5,
77+ $2$
78+};
79+
80+assert(b.a50008 == 25004 and b["a11"] == -5.5)
81+assert(b.a33007 == -16503.5 and b.a50009 == -25004.5)
82+assert(b["b"..30024] == -0xffffff)
83+
84+function b:xxx (a,b) return a+b end
85+assert(b:xxx(10, 12) == 22) -- pushself with non-constant index
86+b["xxx"] = undef
87+
88+local s = 0; local n=0
89+for a,b in pairs(b) do s=s+b; n=n+1 end
90+-- with 32-bit floats, exact value of 's' depends on summation order
91+assert(81800000.0 < s and s < 81860000 and n == 70001)
92+
93+a = nil; b = nil
94+print'+'
95+
96+local function f(x) b=x end
97+
98+a = f{$3$} or 10
99+
100+assert(a==10)
101+assert(b[1] == "a10" and b[2] == 5 and b[#b-1] == "a50009")
102+
103+
104+function xxxx (x) return b[x] end
105+
106+assert(xxxx(3) == "a11")
107+
108+a = nil; b=nil
109+xxxx = nil
110+
111+return 10
112+
113+]]
114+
115+-- functions to fill in the $n$
116+
117+local function sig (x)
118+ return (x % 2 == 0) and '' or '-'
119+end
120+
121+local F = {
122+function () -- $1$
123+ for i=10,50009 do
124+ io.write('a', i, ' = ', sig(i), 5+((i-10)/2), ',\n')
125+ end
126+end,
127+
128+function () -- $2$
129+ for i=30026,50009 do
130+ io.write('b', i, ' = ', sig(i), 15013+((i-30026)/2), ',\n')
131+ end
132+end,
133+
134+function () -- $3$
135+ for i=10,50009 do
136+ io.write('"a', i, '", ', sig(i), 5+((i-10)/2), ',\n')
137+ end
138+end,
139+}
140+
141+local file = os.tmpname()
142+io.output(file)
143+for s in string.gmatch(prog, "$([^$]+)") do
144+ local n = tonumber(s)
145+ if not n then io.write(s) else F[n]() end
146+end
147+io.close()
148+local result = dofile(file)
149+assert(os.remove(file))
150+print'OK'
151+return result
152+