locals.lua 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736
  1. -- $Id: testes/locals.lua $
  2. -- See Copyright Notice in file all.lua
  3. print('testing local variables and environments')
  4. local debug = require"debug"
  5. -- bug in 5.1:
  6. local function f(x) x = nil; return x end
  7. assert(f(10) == nil)
  8. local function f() local x; return x end
  9. assert(f(10) == nil)
  10. local function f(x) x = nil; local y; return x, y end
  11. assert(f(10) == nil and select(2, f(20)) == nil)
  12. do
  13. local i = 10
  14. do local i = 100; assert(i==100) end
  15. do local i = 1000; assert(i==1000) end
  16. assert(i == 10)
  17. if i ~= 10 then
  18. local i = 20
  19. else
  20. local i = 30
  21. assert(i == 30)
  22. end
  23. end
  24. f = nil
  25. local f
  26. x = 1
  27. a = nil
  28. load('local a = {}')()
  29. assert(a == nil)
  30. function f (a)
  31. local _1, _2, _3, _4, _5
  32. local _6, _7, _8, _9, _10
  33. local x = 3
  34. local b = a
  35. local c,d = a,b
  36. if (d == b) then
  37. local x = 'q'
  38. x = b
  39. assert(x == 2)
  40. else
  41. assert(nil)
  42. end
  43. assert(x == 3)
  44. local f = 10
  45. end
  46. local b=10
  47. local a; repeat local b; a,b=1,2; assert(a+1==b); until a+b==3
  48. assert(x == 1)
  49. f(2)
  50. assert(type(f) == 'function')
  51. local function getenv (f)
  52. local a,b = debug.getupvalue(f, 1)
  53. assert(a == '_ENV')
  54. return b
  55. end
  56. -- test for global table of loaded chunks
  57. assert(getenv(load"a=3") == _G)
  58. local c = {}; local f = load("a = 3", nil, nil, c)
  59. assert(getenv(f) == c)
  60. assert(c.a == nil)
  61. f()
  62. assert(c.a == 3)
  63. -- old test for limits for special instructions
  64. do
  65. local i = 2
  66. local p = 4 -- p == 2^i
  67. repeat
  68. for j=-3,3 do
  69. assert(load(string.format([[local a=%s;
  70. a=a+%s;
  71. assert(a ==2^%s)]], j, p-j, i), '')) ()
  72. assert(load(string.format([[local a=%s;
  73. a=a-%s;
  74. assert(a==-2^%s)]], -j, p-j, i), '')) ()
  75. assert(load(string.format([[local a,b=0,%s;
  76. a=b-%s;
  77. assert(a==-2^%s)]], -j, p-j, i), '')) ()
  78. end
  79. p = 2 * p; i = i + 1
  80. until p <= 0
  81. end
  82. print'+'
  83. if rawget(_G, "T") then
  84. -- testing clearing of dead elements from tables
  85. collectgarbage("stop") -- stop GC
  86. local a = {[{}] = 4, [3] = 0, alo = 1,
  87. a1234567890123456789012345678901234567890 = 10}
  88. local t = T.querytab(a)
  89. for k,_ in pairs(a) do a[k] = undef end
  90. collectgarbage() -- restore GC and collect dead fields in 'a'
  91. for i=0,t-1 do
  92. local k = querytab(a, i)
  93. assert(k == nil or type(k) == 'number' or k == 'alo')
  94. end
  95. -- testing allocation errors during table insertions
  96. local a = {}
  97. local function additems ()
  98. a.x = true; a.y = true; a.z = true
  99. a[1] = true
  100. a[2] = true
  101. end
  102. for i = 1, math.huge do
  103. T.alloccount(i)
  104. local st, msg = pcall(additems)
  105. T.alloccount()
  106. local count = 0
  107. for k, v in pairs(a) do
  108. assert(a[k] == v)
  109. count = count + 1
  110. end
  111. if st then assert(count == 5); break end
  112. end
  113. end
  114. -- testing lexical environments
  115. assert(_ENV == _G)
  116. do
  117. local dummy
  118. local _ENV = (function (...) return ... end)(_G, dummy) -- {
  119. do local _ENV = {assert=assert}; assert(true) end
  120. mt = {_G = _G}
  121. local foo,x
  122. A = false -- "declare" A
  123. do local _ENV = mt
  124. function foo (x)
  125. A = x
  126. do local _ENV = _G; A = 1000 end
  127. return function (x) return A .. x end
  128. end
  129. end
  130. assert(getenv(foo) == mt)
  131. x = foo('hi'); assert(mt.A == 'hi' and A == 1000)
  132. assert(x('*') == mt.A .. '*')
  133. do local _ENV = {assert=assert, A=10};
  134. do local _ENV = {assert=assert, A=20};
  135. assert(A==20);x=A
  136. end
  137. assert(A==10 and x==20)
  138. end
  139. assert(x==20)
  140. do -- constants
  141. local a<const>, b, c<const> = 10, 20, 30
  142. b = a + c + b -- 'b' is not constant
  143. assert(a == 10 and b == 60 and c == 30)
  144. local function checkro (name, code)
  145. local st, msg = load(code)
  146. local gab = string.format("attempt to assign to const variable '%s'", name)
  147. assert(not st and string.find(msg, gab))
  148. end
  149. checkro("y", "local x, y <const>, z = 10, 20, 30; x = 11; y = 12")
  150. checkro("x", "local x <const>, y, z <const> = 10, 20, 30; x = 11")
  151. checkro("z", "local x <const>, y, z <const> = 10, 20, 30; y = 10; z = 11")
  152. checkro("z", [[
  153. local a, z <const>, b = 10;
  154. function foo() a = 20; z = 32; end
  155. ]])
  156. checkro("var1", [[
  157. local a, var1 <const> = 10;
  158. function foo() a = 20; z = function () var1 = 12; end end
  159. ]])
  160. end
  161. print"testing to-be-closed variables"
  162. local function stack(n) n = ((n == 0) or stack(n - 1)) end
  163. local function func2close (f, x, y)
  164. local obj = setmetatable({}, {__close = f})
  165. if x then
  166. return x, obj, y
  167. else
  168. return obj
  169. end
  170. end
  171. do
  172. local a = {}
  173. do
  174. local b <close> = false -- not to be closed
  175. local x <close> = setmetatable({"x"}, {__close = function (self)
  176. a[#a + 1] = self[1] end})
  177. local w, y <close>, z = func2close(function (self, err)
  178. assert(err == nil); a[#a + 1] = "y"
  179. end, 10, 20)
  180. local c <close> = nil -- not to be closed
  181. a[#a + 1] = "in"
  182. assert(w == 10 and z == 20)
  183. end
  184. a[#a + 1] = "out"
  185. assert(a[1] == "in" and a[2] == "y" and a[3] == "x" and a[4] == "out")
  186. end
  187. do
  188. local X = false
  189. local x, closescope = func2close(function () stack(10); X = true end, 100)
  190. assert(x == 100); x = 101; -- 'x' is not read-only
  191. -- closing functions do not corrupt returning values
  192. local function foo (x)
  193. local _ <close> = closescope
  194. return x, X, 23
  195. end
  196. local a, b, c = foo(1.5)
  197. assert(a == 1.5 and b == false and c == 23 and X == true)
  198. X = false
  199. foo = function (x)
  200. local _<close> = closescope
  201. local y = 15
  202. return y
  203. end
  204. assert(foo() == 15 and X == true)
  205. X = false
  206. foo = function ()
  207. local x <close> = closescope
  208. return x
  209. end
  210. assert(foo() == closescope and X == true)
  211. end
  212. -- testing to-be-closed x compile-time constants
  213. -- (there were some bugs here in Lua 5.4-rc3, due to a confusion
  214. -- between compile levels and stack levels of variables)
  215. do
  216. local flag = false
  217. local x = setmetatable({},
  218. {__close = function() assert(flag == false); flag = true end})
  219. local y <const> = nil
  220. local z <const> = nil
  221. do
  222. local a <close> = x
  223. end
  224. assert(flag) -- 'x' must be closed here
  225. end
  226. do
  227. -- similar problem, but with implicit close in for loops
  228. local flag = false
  229. local x = setmetatable({},
  230. {__close = function () assert(flag == false); flag = true end})
  231. -- return an empty iterator, nil, nil, and 'x' to be closed
  232. local function a ()
  233. return (function () return nil end), nil, nil, x
  234. end
  235. local v <const> = 1
  236. local w <const> = 1
  237. local x <const> = 1
  238. local y <const> = 1
  239. local z <const> = 1
  240. for k in a() do
  241. a = k
  242. end -- ending the loop must close 'x'
  243. assert(flag) -- 'x' must be closed here
  244. end
  245. do
  246. -- calls cannot be tail in the scope of to-be-closed variables
  247. local X, Y
  248. local function foo ()
  249. local _ <close> = func2close(function () Y = 10 end)
  250. assert(X == true and Y == nil) -- 'X' not closed yet
  251. return 1,2,3
  252. end
  253. local function bar ()
  254. local _ <close> = func2close(function () X = false end)
  255. X = true
  256. do
  257. return foo() -- not a tail call!
  258. end
  259. end
  260. local a, b, c, d = bar()
  261. assert(a == 1 and b == 2 and c == 3 and X == false and Y == 10 and d == nil)
  262. end
  263. -- auxiliary functions for testing warnings in '__close'
  264. local function prepwarn ()
  265. if not T then -- no test library?
  266. warn("@off") -- do not show (lots of) warnings
  267. else
  268. warn("@store") -- to test the warnings
  269. end
  270. end
  271. local function endwarn ()
  272. if not T then
  273. warn("@on") -- back to normal
  274. else
  275. assert(_WARN == false)
  276. warn("@normal")
  277. end
  278. end
  279. local function checkwarn (msg)
  280. if T then
  281. assert(string.find(_WARN, msg))
  282. _WARN = false -- reset variable to check next warning
  283. end
  284. end
  285. warn("@on")
  286. do print("testing errors in __close")
  287. prepwarn()
  288. -- original error is in __close
  289. local function foo ()
  290. local x <close> =
  291. func2close(function (self, msg)
  292. assert(string.find(msg, "@z"))
  293. error("@x")
  294. end)
  295. local x1 <close> =
  296. func2close(function (self, msg)
  297. checkwarn("@y")
  298. assert(string.find(msg, "@z"))
  299. end)
  300. local gc <close> = func2close(function () collectgarbage() end)
  301. local y <close> =
  302. func2close(function (self, msg)
  303. assert(string.find(msg, "@z")) -- first error in 'z'
  304. checkwarn("@z") -- second error in 'z' generated a warning
  305. error("@y")
  306. end)
  307. local first = true
  308. local z <close> =
  309. -- 'z' close is called twice
  310. func2close(function (self, msg)
  311. if first then
  312. assert(msg == nil)
  313. first = false
  314. else
  315. assert(string.find(msg, "@z")) -- own error
  316. end
  317. error("@z")
  318. end)
  319. return 200
  320. end
  321. local stat, msg = pcall(foo, false)
  322. assert(string.find(msg, "@z"))
  323. checkwarn("@x")
  324. -- original error not in __close
  325. local function foo ()
  326. local x <close> =
  327. func2close(function (self, msg)
  328. assert(msg == 4)
  329. end)
  330. local x1 <close> =
  331. func2close(function (self, msg)
  332. checkwarn("@y")
  333. assert(msg == 4)
  334. error("@x1")
  335. end)
  336. local gc <close> = func2close(function () collectgarbage() end)
  337. local y <close> =
  338. func2close(function (self, msg)
  339. assert(msg == 4) -- error in body
  340. checkwarn("@z")
  341. error("@y")
  342. end)
  343. local first = true
  344. local z <close> =
  345. func2close(function (self, msg)
  346. -- 'z' close is called once
  347. assert(first and msg == 4)
  348. first = false
  349. error("@z")
  350. end)
  351. error(4) -- original error
  352. end
  353. local stat, msg = pcall(foo, true)
  354. assert(msg == 4)
  355. checkwarn("@x1") -- last error
  356. -- error leaving a block
  357. local function foo (...)
  358. do
  359. local x1 <close> =
  360. func2close(function ()
  361. checkwarn("@X")
  362. error("@Y")
  363. end)
  364. local x123 <close> =
  365. func2close(function ()
  366. error("@X")
  367. end)
  368. end
  369. os.exit(false) -- should not run
  370. end
  371. local st, msg = xpcall(foo, debug.traceback)
  372. assert(string.match(msg, "^[^ ]* @X"))
  373. assert(string.find(msg, "in metamethod 'close'"))
  374. checkwarn("@Y")
  375. -- error in toclose in vararg function
  376. local function foo (...)
  377. local x123 <close> = func2close(function () error("@x123") end)
  378. end
  379. local st, msg = xpcall(foo, debug.traceback)
  380. assert(string.match(msg, "^[^ ]* @x123"))
  381. assert(string.find(msg, "in metamethod 'close'"))
  382. checkwarn("@x123") -- from second call to close 'x123'
  383. endwarn()
  384. end
  385. do -- errors due to non-closable values
  386. local function foo ()
  387. local x <close> = {}
  388. os.exit(false) -- should not run
  389. end
  390. local stat, msg = pcall(foo)
  391. assert(not stat and
  392. string.find(msg, "variable 'x' got a non%-closable value"))
  393. local function foo ()
  394. local xyz <close> = setmetatable({}, {__close = print})
  395. getmetatable(xyz).__close = nil -- remove metamethod
  396. end
  397. local stat, msg = pcall(foo)
  398. assert(not stat and
  399. string.find(msg, "attempt to close non%-closable variable 'xyz'"))
  400. end
  401. if rawget(_G, "T") then
  402. warn("@off")
  403. -- memory error inside closing function
  404. local function foo ()
  405. local y <close> = func2close(function () T.alloccount() end)
  406. local x <close> = setmetatable({}, {__close = function ()
  407. T.alloccount(0); local x = {} -- force a memory error
  408. end})
  409. error(1000) -- common error inside the function's body
  410. end
  411. stack(5) -- ensure a minimal number of CI structures
  412. -- despite memory error, 'y' will be executed and
  413. -- memory limit will be lifted
  414. local _, msg = pcall(foo)
  415. assert(msg == 1000)
  416. local close = func2close(function (self, msg)
  417. T.alloccount()
  418. assert(msg == "not enough memory")
  419. end)
  420. -- set a memory limit and return a closing object to remove the limit
  421. local function enter (count)
  422. stack(10) -- reserve some stack space
  423. T.alloccount(count)
  424. return close
  425. end
  426. local function test ()
  427. local x <close> = enter(0) -- set a memory limit
  428. -- creation of previous upvalue will raise a memory error
  429. assert(false) -- should not run
  430. end
  431. local _, msg = pcall(test)
  432. assert(msg == "not enough memory")
  433. -- now use metamethod for closing
  434. close = setmetatable({}, {__close = function ()
  435. T.alloccount()
  436. end})
  437. -- repeat test with extra closing upvalues
  438. local function test ()
  439. local xxx <close> = func2close(function (self, msg)
  440. assert(msg == "not enough memory");
  441. error(1000) -- raise another error
  442. end)
  443. local xx <close> = func2close(function (self, msg)
  444. assert(msg == "not enough memory");
  445. end)
  446. local x <close> = enter(0) -- set a memory limit
  447. -- creation of previous upvalue will raise a memory error
  448. os.exit(false) -- should not run
  449. end
  450. local _, msg = pcall(test)
  451. assert(msg == "not enough memory") -- reported error is the first one
  452. do -- testing 'toclose' in C string buffer
  453. collectgarbage()
  454. local s = string.rep('a', 10000) -- large string
  455. local m = T.totalmem()
  456. collectgarbage("stop")
  457. s = string.upper(s) -- allocate buffer + new string (10K each)
  458. -- ensure buffer was deallocated
  459. assert(T.totalmem() - m <= 11000)
  460. collectgarbage("restart")
  461. end
  462. do -- now some tests for freeing buffer in case of errors
  463. local lim = 10000 -- some size larger than the static buffer
  464. local extra = 2000 -- some extra memory (for callinfo, etc.)
  465. local s = string.rep("a", lim)
  466. -- concat this table needs two buffer resizes (one for each 's')
  467. local a = {s, s}
  468. collectgarbage()
  469. m = T.totalmem()
  470. collectgarbage("stop")
  471. -- error in the first buffer allocation
  472. T. totalmem(m + extra)
  473. assert(not pcall(table.concat, a))
  474. -- first buffer was not even allocated
  475. assert(T.totalmem() - m <= extra)
  476. -- error in the second buffer allocation
  477. T. totalmem(m + lim + extra)
  478. assert(not pcall(table.concat, a))
  479. -- first buffer was released by 'toclose'
  480. assert(T.totalmem() - m <= extra)
  481. -- error in creation of final string
  482. T.totalmem(m + 2 * lim + extra)
  483. assert(not pcall(table.concat, a))
  484. -- second buffer was released by 'toclose'
  485. assert(T.totalmem() - m <= extra)
  486. -- userdata, upvalue, buffer, buffer, final string
  487. T.totalmem(m + 4*lim + extra)
  488. assert(#table.concat(a) == 2*lim)
  489. T.totalmem(0) -- remove memory limit
  490. collectgarbage("restart")
  491. print'+'
  492. end
  493. warn("@on")
  494. end
  495. print "to-be-closed variables in coroutines"
  496. do
  497. -- an error in a wrapped coroutine closes variables
  498. local x = false
  499. local y = false
  500. local co = coroutine.wrap(function ()
  501. local xv <close> = func2close(function () x = true end)
  502. do
  503. local yv <close> = func2close(function () y = true end)
  504. coroutine.yield(100) -- yield doesn't close variable
  505. end
  506. coroutine.yield(200) -- yield doesn't close variable
  507. error(23) -- error does
  508. end)
  509. local b = co()
  510. assert(b == 100 and not x and not y)
  511. b = co()
  512. assert(b == 200 and not x and y)
  513. local a, b = pcall(co)
  514. assert(not a and b == 23 and x and y)
  515. end
  516. do
  517. prepwarn()
  518. -- error in a wrapped coroutine raising errors when closing a variable
  519. local x = 0
  520. local co = coroutine.wrap(function ()
  521. local xx <close> = func2close(function () x = x + 1; error("@YYY") end)
  522. local xv <close> = func2close(function () x = x + 1; error("@XXX") end)
  523. coroutine.yield(100)
  524. error(200)
  525. end)
  526. assert(co() == 100); assert(x == 0)
  527. local st, msg = pcall(co); assert(x == 2)
  528. assert(not st and msg == 200) -- should get first error raised
  529. checkwarn("@YYY")
  530. local x = 0
  531. local y = 0
  532. co = coroutine.wrap(function ()
  533. local xx <close> = func2close(function () y = y + 1; error("YYY") end)
  534. local xv <close> = func2close(function () x = x + 1; error("XXX") end)
  535. coroutine.yield(100)
  536. return 200
  537. end)
  538. assert(co() == 100); assert(x == 0)
  539. local st, msg = pcall(co)
  540. assert(x == 2 and y == 1) -- first close is called twice
  541. -- should get first error raised
  542. assert(not st and string.find(msg, "%w+%.%w+:%d+: XXX"))
  543. checkwarn("YYY")
  544. endwarn()
  545. end
  546. -- a suspended coroutine should not close its variables when collected
  547. local co
  548. co = coroutine.wrap(function()
  549. -- should not run
  550. local x <close> = func2close(function () os.exit(false) end)
  551. co = nil
  552. coroutine.yield()
  553. end)
  554. co() -- start coroutine
  555. assert(co == nil) -- eventually it will be collected
  556. collectgarbage()
  557. -- to-be-closed variables in generic for loops
  558. do
  559. local numopen = 0
  560. local function open (x)
  561. numopen = numopen + 1
  562. return
  563. function () -- iteraction function
  564. x = x - 1
  565. if x > 0 then return x end
  566. end,
  567. nil, -- state
  568. nil, -- control variable
  569. func2close(function () numopen = numopen - 1 end) -- closing function
  570. end
  571. local s = 0
  572. for i in open(10) do
  573. s = s + i
  574. end
  575. assert(s == 45 and numopen == 0)
  576. local s = 0
  577. for i in open(10) do
  578. if i < 5 then break end
  579. s = s + i
  580. end
  581. assert(s == 35 and numopen == 0)
  582. local s = 0
  583. for i in open(10) do
  584. for j in open(10) do
  585. if i + j < 5 then goto endloop end
  586. s = s + i
  587. end
  588. end
  589. ::endloop::
  590. assert(s == 375 and numopen == 0)
  591. end
  592. print('OK')
  593. return 5,f
  594. end -- }