1 ## handling malformed programs 2 3 container environment [ 4 recipe-errors:text 5 ] 6 7 # copy code from recipe editor, persist to disk, load, save any errors 8 def! update-recipes env:&:environment, resources:&:resources, screen:&:screen -> errors-found?:bool, env:&:environment, resources:&:resources, screen:&:screen [ 9 local-scope 10 load-ingredients 11 recipes:&:editor <- get *env, recipes:offset 12 in:text <- editor-contents recipes 13 resources <- dump resources, [lesson/recipes.mu], in 14 recipe-errors:text <- reload in 15 *env <- put *env, recipe-errors:offset, recipe-errors 16 # if recipe editor has errors, stop 17 { 18 ¦ break-unless recipe-errors 19 ¦ update-status screen, [errors found ], 1/red 20 ¦ errors-found? <- copy 1/true 21 ¦ return 22 } 23 errors-found? <- copy 0/false 24 ] 25 26 after <begin-run-sandboxes-on-F4> [ 27 old-recipe-errors:text <- get *env, recipe-errors:offset 28 ] 29 before <end-run-sandboxes-on-F4> [ 30 # if there were recipe errors before, check if we can clear them 31 { 32 ¦ break-unless old-recipe-errors 33 ¦ screen <- render-recipes screen, env, render 34 } 35 screen <- render-recipe-errors env, screen 36 ] 37 38 before <end-render-recipe-components> [ 39 screen <- render-recipe-errors env, screen 40 ] 41 42 def render-recipe-errors env:&:environment, screen:&:screen -> screen:&:screen [ 43 local-scope 44 load-ingredients 45 recipe-errors:text <- get *env, recipe-errors:offset 46 return-unless recipe-errors 47 recipes:&:editor <- get *env, recipes:offset 48 left:num <- get *recipes, left:offset 49 right:num <- get *recipes, right:offset 50 row:num <- get *recipes, bottom:offset 51 row, screen <- render-text screen, recipe-errors, left, right, 1/red, row 52 # draw dotted line after recipes 53 draw-horizontal screen, row, left, right, 9480/horizontal-dotted 54 row <- add row, 1 55 clear-screen-from screen, row, left, left, right 56 ] 57 58 container environment [ 59 error-index:num # index of first sandbox with an error (or -1 if none) 60 ] 61 62 after <programming-environment-initialization> [ 63 *result <- put *result, error-index:offset, -1 64 ] 65 66 after <begin-run-sandboxes> [ 67 *env <- put *env, error-index:offset, -1 68 ] 69 70 before <end-run-sandboxes> [ 71 { 72 ¦ error-index:num <- get *env, error-index:offset 73 ¦ sandboxes-completed-successfully?:bool <- equal error-index, -1 74 ¦ break-if sandboxes-completed-successfully? 75 ¦ errors-found? <- copy 1/true 76 } 77 ] 78 79 before <end-run-sandboxes-on-F4> [ 80 { 81 ¦ break-unless error? 82 ¦ recipe-errors:text <- get *env, recipe-errors:offset 83 ¦ break-if recipe-errors 84 ¦ error-index:num <- get *env, error-index:offset 85 ¦ sandboxes-completed-successfully?:bool <- equal error-index, -1 86 ¦ break-if sandboxes-completed-successfully? 87 ¦ error-index-text:text <- to-text error-index 88 ¦ status:text <- interpolate [errors found (_) ], error-index-text 89 ¦ update-status screen, status, 1/red 90 } 91 ] 92 93 container sandbox [ 94 errors:text 95 ] 96 97 def! update-sandbox sandbox:&:sandbox, env:&:environment, idx:num -> sandbox:&:sandbox, env:&:environment [ 98 local-scope 99 load-ingredients 100 data:text <- get *sandbox, data:offset 101 response:text, errors:text, fake-screen:&:screen, trace:text, completed?:bool <- run-sandboxed data 102 *sandbox <- put *sandbox, response:offset, response 103 *sandbox <- put *sandbox, errors:offset, errors 104 *sandbox <- put *sandbox, screen:offset, fake-screen 105 *sandbox <- put *sandbox, trace:offset, trace 106 { 107 ¦ break-if errors 108 ¦ break-if completed? 109 ¦ errors <- new [took too long! 110 ] 111 ¦ *sandbox <- put *sandbox, errors:offset, errors 112 } 113 { 114 ¦ break-unless errors 115 ¦ error-index:num <- get *env, error-index:offset 116 ¦ error-not-set?:bool <- equal error-index, -1 117 ¦ break-unless error-not-set? 118 ¦ *env <- put *env, error-index:offset, idx 119 } 120 ] 121 122 # make sure we render any trace 123 after <render-sandbox-trace-done> [ 124 { 125 ¦ sandbox-errors:text <- get *sandbox, errors:offset 126 ¦ break-unless sandbox-errors 127 ¦ *sandbox <- put *sandbox, response-starting-row-on-screen:offset, 0 # no response 128 ¦ row, screen <- render-text screen, sandbox-errors, left, right, 1/red, row 129 ¦ # don't try to print anything more for this sandbox 130 ¦ jump +render-sandbox-end 131 } 132 ] 133 134 scenario run-shows-errors-in-get [ 135 local-scope 136 trace-until 100/app # trace too long 137 assume-screen 100/width, 15/height 138 assume-resources [ 139 ¦ [lesson/recipes.mu] <- [ 140 ¦ ¦ |recipe foo [| 141 ¦ ¦ | get 123:num, foo:offset| 142 ¦ ¦ |]| 143 ¦ ] 144 ] 145 env:&:environment <- new-programming-environment resources, screen, [foo] 146 render-all screen, env, render 147 screen-should-contain [ 148 ¦ . run (F4) . 149 ¦ .recipe foo [ ╎foo . 150 ¦ . get 123:num, foo:offset ╎─────────────────────────────────────────────────. 151 ¦ .] ╎ . 152 ¦ . ╎ . 153 ¦ .╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╎ . 154 ¦ . ╎ . 155 ] 156 assume-console [ 157 ¦ press F4 158 ] 159 run [ 160 ¦ event-loop screen, console, env, resources 161 ] 162 screen-should-contain [ 163 ¦ . errors found run (F4) . 164 ¦ .recipe foo [ ╎foo . 165 ¦ . get 123:num, foo:offset ╎─────────────────────────────────────────────────. 166 ¦ .] ╎ . 167 ¦ .foo: unknown element 'foo' in container 'number' ╎ . 168 ¦ .foo: first ingredient of 'get' should be a contai↩╎ . 169 ¦ .ner, but got '123:num' ╎ . 170 ¦ .╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╎ . 171 ¦ . ╎ . 172 ] 173 screen-should-contain-in-color 1/red, [ 174 ¦ . errors found . 175 ¦ . . 176 ¦ . . 177 ¦ . . 178 ¦ .foo: unknown element 'foo' in container 'number' . 179 ¦ .foo: first ingredient of 'get' should be a contai . 180 ¦ .ner, but got '123:num' . 181 ¦ . . 182 ] 183 ] 184 185 scenario run-updates-status-with-first-erroneous-sandbox [ 186 local-scope 187 trace-until 100/app # trace too long 188 assume-screen 100/width, 15/height 189 assume-resources [ 190 ] 191 env:&:environment <- new-programming-environment resources, screen, [] 192 render-all screen, env, render 193 assume-console [ 194 ¦ left-click 3, 80 195 ¦ # create invalid sandbox 1 196 ¦ type [get foo, x:offset] 197 ¦ press F4 198 ¦ # create invalid sandbox 0 199 ¦ type [get foo, x:offset] 200 ¦ press F4 201 ] 202 run [ 203 ¦ event-loop screen, console, env, resources 204 ] 205 # status line shows that error is in first sandbox 206 screen-should-contain [ 207 ¦ . errors found (0) run (F4) . 208 ] 209 ] 210 211 scenario run-updates-status-with-first-erroneous-sandbox-2 [ 212 local-scope 213 trace-until 100/app # trace too long 214 assume-screen 100/width, 15/height 215 assume-resources [ 216 ] 217 env:&:environment <- new-programming-environment resources, screen, [] 218 render-all screen, env, render 219 assume-console [ 220 ¦ left-click 3, 80 221 ¦ # create invalid sandbox 2 222 ¦ type [get foo, x:offset] 223 ¦ press F4 224 ¦ # create invalid sandbox 1 225 ¦ type [get foo, x:offset] 226 ¦ press F4 227 ¦ # create valid sandbox 0 228 ¦ type [add 2, 2] 229 ¦ press F4 230 ] 231 run [ 232 ¦ event-loop screen, console, env, resources 233 ] 234 # status line shows that error is in second sandbox 235 screen-should-contain [ 236 ¦ . errors found (1) run (F4) . 237 ] 238 ] 239 240 scenario run-hides-errors-from-past-sandboxes [ 241 local-scope 242 trace-until 100/app # trace too long 243 assume-screen 100/width, 15/height 244 assume-resources [ 245 ] 246 env:&:environment <- new-programming-environment resources, screen, [get foo, x:offset] # invalid 247 render-all screen, env, render 248 assume-console [ 249 ¦ press F4 # generate error 250 ] 251 event-loop screen, console, env, resources 252 assume-console [ 253 ¦ left-click 3, 58 254 ¦ press ctrl-k 255 ¦ type [add 2, 2] # valid code 256 ¦ press F4 # update sandbox 257 ] 258 run [ 259 ¦ event-loop screen, console, env, resources 260 ] 261 # error should disappear 262 screen-should-contain [ 263 ¦ . run (F4) . 264 ¦ . ╎ . 265 ¦ .╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╎─────────────────────────────────────────────────. 266 ¦ . ╎0 edit copy to recipe delete . 267 ¦ . ╎add 2, 2 . 268 ¦ . ╎4 . 269 ¦ . ╎─────────────────────────────────────────────────. 270 ¦ . ╎ . 271 ] 272 ] 273 274 scenario run-updates-errors-for-shape-shifting-recipes [ 275 local-scope 276 trace-until 100/app # trace too long 277 assume-screen 100/width, 15/height 278 # define a shape-shifting recipe with an error 279 assume-resources [ 280 ¦ [lesson/recipes.mu] <- [ 281 ¦ ¦ |recipe foo x:_elem -> z:_elem [| 282 ¦ ¦ | local-scope| 283 ¦ ¦ | load-ingredients| 284 ¦ ¦ | y:&:num <- copy 0| 285 ¦ ¦ | z <- add x, y| 286 ¦ ¦ |]| 287 ¦ ] 288 ] 289 env:&:environment <- new-programming-environment resources, screen, [foo 2] 290 render-all screen, env, render 291 assume-console [ 292 ¦ press F4 293 ] 294 event-loop screen, console, env, resources-- -- Collect memory reference info. -- https://github.com/yaukeywang/LuaMemorySnapshotDump -- -- @filename MemoryReferenceInfo.lua -- @author WangYaoqi -- @date 2016-02-03 -- The global config of the mri. local cConfig = { m_bAllMemoryRefFileAddTime = true, m_bSingleMemoryRefFileAddTime = true, m_bComparedMemoryRefFileAddTime = true } -- Get the format string of date time. local function FormatDateTimeNow() local cDateTime = os.date("*t") local strDateTime = string.format("%04d%02d%02d-%02d%02d%02d", tostring(cDateTime.year), tostring(cDateTime.month), tostring(cDateTime.day), tostring(cDateTime.hour), tostring(cDateTime.min), tostring(cDateTime.sec)) return strDateTime end -- Get the string result without overrided __tostring. local function GetOriginalToStringResult(cObject) if not cObject then return "" end local cMt = getmetatable(cObject) if not cMt then return tostring(cObject) end -- Check tostring override. local strName = "" local cToString = rawget(cMt, "__tostring") if cToString then print('tostring overridden:', tostring(cObject)) --? rawset(cMt, "__tostring", nil) --? strName = tostring(cObject) --? rawset(cMt, "__tostring", cToString) --? else --? strName = tostring(cObject) end strName = tostring(cObject) return strName end -- Create a container to collect the mem ref info results. local function CreateObjectReferenceInfoContainer() -- Create new container. local cContainer = {} -- Contain [table/function] - [reference count] info. local cObjectReferenceCount = {} setmetatable(cObjectReferenceCount, {__mode = "k"}) -- Contain [table/function] - [name] info. local cObjectAddressToName = {} setmetatable(cObjectAddressToName, {__mode = "k"}) -- Set members. cContainer.m_cObjectReferenceCount = cObjectReferenceCount cContainer.m_cObjectAddressToName = cObjectAddressToName -- For stack info. cContainer.m_nStackLevel = -1 cContainer.m_strShortSrc = "None" cContainer.m_nCurrentLine = -1 return cContainer end -- Create a container to collect the mem ref info results from a dumped file. -- strFilePath - The file path. local function CreateObjectReferenceInfoContainerFromFile(strFilePath) -- Create a empty container. local cContainer = CreateObjectReferenceInfoContainer() cContainer.m_strShortSrc = strFilePath -- Cache ref info. local cRefInfo = cContainer.m_cObjectReferenceCount local cNameInfo = cContainer.m_cObjectAddressToName -- Read each line from file. local cFile = assert(io.open(strFilePath, "rb")) for strLine in cFile:lines() do local strHeader = string.sub(strLine, 1, 2) if "--" ~= strHeader then local _, _, strAddr, strName, strRefCount= string.find(strLine, "(.+)\t(.*)\t(%d+)") if strAddr then cRefInfo[strAddr] = strRefCount cNameInfo[strAddr] = strName end end end -- Close and clear file handler. io.close(cFile) cFile = nil return cContainer end -- Create a container to collect the mem ref info results from a dumped file. -- strObjectName - The object name you need to collect info. -- cObject - The object you need to collect info. local function CreateSingleObjectReferenceInfoContainer(strObjectName, cObject) -- Create new container. local cContainer = {} -- Contain [address] - [true] info. local cObjectExistTag = {} setmetatable(cObjectExistTag, {__mode = "k"}) -- Contain [name] - [true] info. local cObjectAliasName = {} -- Contain [access] - [true] info. local cObjectAccessTag = {} setmetatable(cObjectAccessTag, {__mode = "k"}) -- Set members. cContainer.m_cObjectExistTag = cObjectExistTag cContainer.m_cObjectAliasName = cObjectAliasName cContainer.m_cObjectAccessTag = cObjectAccessTag -- For stack info. cContainer.m_nStackLevel = -1 cContainer.m_strShortSrc = "None" cContainer.m_nCurrentLine = -1 -- Init with object values. cContainer.m_strObjectName = strObjectName cContainer.m_strAddressName = (("string" == type(cObject)) and ("\"" .. tostring(cObject) .. "\"")) or GetOriginalToStringResult(cObject) cContainer.m_cObjectExistTag[cObject] = true return cContainer end -- Collect memory reference info from a root table or function. -- strName - The root object name that start to search, default is "_G" if leave this to nil. -- cObject - The root object that start to search, default is _G if leave this to nil. -- cDumpInfoContainer - The container of the dump result info. local function CollectObjectReferenceInMemory(strName, cObject, cDumpInfoContainer) if not cObject then return end if not strName then strName = "" end -- Check container. if (not cDumpInfoContainer) then cDumpInfoContainer = CreateObjectReferenceInfoContainer() end -- Check stack. if cDumpInfoContainer.m_nStackLevel > 0 then local cStackInfo = debug.getinfo(cDumpInfoContainer.m_nStackLevel, "Sl") if cStackInfo then cDumpInfoContainer.m_strShortSrc = cStackInfo.short_src cDumpInfoContainer.m_nCurrentLine = cStackInfo.currentline end cDumpInfoContainer.m_nStackLevel = -1 end -- Get ref and name info. local cRefInfoContainer = cDumpInfoContainer.m_cObjectReferenceCount local cNameInfoContainer = cDumpInfoContainer.m_cObjectAddressToName local strType = type(cObject) if "table" == strType then -- Check table with class name. if rawget(cObject, "__cname") then if "string" == type(cObject.__cname) then strName = strName .. "[class:" .. cObject.__cname .. "]" end elseif rawget(cObject, "class") then if "string" == type(cObject.class) then strName = strName .. "[class:" .. cObject.class .. "]" end elseif rawget(cObject, "_className") then if "string" == type(cObject._className) then strName = strName .. "[class:" .. cObject._className .. "]" end end -- Check if table is _G. if cObject == _G then strName = strName .. "[_G]" end -- Get metatable. local bWeakK = false local bWeakV = false local cMt = getmetatable(cObject) if cMt then -- Check mode. local strMode = rawget(cMt, "__mode") if strMode then if "k" == strMode then bWeakK = true elseif "v" == strMode then bWeakV = true elseif "kv" == strMode then bWeakK = true bWeakV = true end end end -- Add reference and name. cRefInfoContainer[cObject] = (cRefInfoContainer[cObject] and (cRefInfoContainer[cObject] + 1)) or 1 if cNameInfoContainer[cObject] then return end -- Set name. cNameInfoContainer[cObject] = strName -- Dump table key and value. for k, v in pairs(cObject) do -- Check key type. local strKeyType = type(k) if "table" == strKeyType then if not bWeakK then CollectObjectReferenceInMemory(strName .. ".[table:key.table]", k, cDumpInfoContainer) end if not bWeakV then CollectObjectReferenceInMemory(strName .. ".[table:value]", v, cDumpInfoContainer) end elseif "function" == strKeyType then if not bWeakK then CollectObjectReferenceInMemory(strName .. ".[table:key.function]", k, cDumpInfoContainer) end if not bWeakV then CollectObjectReferenceInMemory(strName .. ".[table:value]", v, cDumpInfoContainer) end elseif "thread" == strKeyType then if not bWeakK then CollectObjectReferenceInMemory(strName .. ".[table:key.thread]", k, cDumpInfoContainer) end if not bWeakV then CollectObjectReferenceInMemory(strName .. ".[table:value]", v, cDumpInfoContainer) end elseif "userdata" == strKeyType then if not bWeakK then CollectObjectReferenceInMemory(strName .. ".[table:key.userdata]", k, cDumpInfoContainer) end if not bWeakV then CollectObjectReferenceInMemory(strName .. ".[table:value]", v, cDumpInfoContainer) end else CollectObjectReferenceInMemory(strName .. "." .. tostring(k), v, cDumpInfoContainer) end end -- Dump metatable. if cMt then CollectObjectReferenceInMemory(strName ..".[metatable]", cMt, cDumpInfoContainer) end elseif "function" == strType then -- Get function info. local cDInfo = debug.getinfo(cObject, "Su") -- Write this info. cRefInfoContainer[cObject] = (cRefInfoContainer[cObject] and (cRefInfoContainer[cObject] + 1)) or 1 if cNameInfoContainer[cObject] then return end -- Set name. cNameInfoContainer[cObject] = strName .. "[line:" .. tostring(cDInfo.linedefined) .. "@file:" .. cDInfo.short_src .. "]" -- Get upvalues. local nUpsNum = cDInfo.nups for i = 1, nUpsNum do local strUpName, cUpValue = debug.getupvalue(cObject, i) local strUpValueType = type(cUpValue) --print(strUpName, cUpValue) if "table" == strUpValueType then CollectObjectReferenceInMemory(strName .. ".[ups:table:" .. strUpName .. "]", cUpValue, cDumpInfoContainer) elseif "function" == strUpValueType then CollectObjectReferenceInMemory(strName .. ".[ups:function:" .. strUpName .. "]", cUpValue, cDumpInfoContainer) elseif "thread" == strUpValueType then CollectObjectReferenceInMemory(strName .. ".[ups:thread:" .. strUpName .. "]", cUpValue, cDumpInfoContainer) elseif "userdata" == strUpValueType then CollectObjectReferenceInMemory(strName .. ".[ups:userdata:" .. strUpName .. "]", cUpValue, cDumpInfoContainer) end end -- Dump environment table. local getfenv = debug.getfenv if getfenv then local cEnv = getfenv(cObject) if cEnv then CollectObjectReferenceInMemory(strName ..".[function:environment]", cEnv, cDumpInfoContainer) end end elseif "thread" == strType then -- Add reference and name. cRefInfoContainer[cObject] = (cRefInfoContainer[cObject] and (cRefInfoContainer[cObject] + 1)) or 1 if cNameInfoContainer[cObject] then return end -- Set name. cNameInfoContainer[cObject] = strName -- Dump environment table. local getfenv = debug.getfenv if getfenv then local cEnv = getfenv(cObject) if cEnv then CollectObjectReferenceInMemory(strName ..".[thread:environment]", cEnv, cDumpInfoContainer) end end -- Dump metatable. local cMt = getmetatable(cObject) if cMt then CollectObjectReferenceInMemory(strName ..".[thread:metatable]", cMt, cDumpInfoContainer) end elseif "userdata" == strType then -- Add reference and name. cRefInfoContainer[cObject] = (cRefInfoContainer[cObject] and (cRefInfoContainer[cObject] + 1)) or 1 if cNameInfoContainer[cObject] then return end -- Set name. cNameInfoContainer[cObject] = strName -- Dump environment table. local getfenv = debug.getfenv if getfenv then local cEnv = getfenv(cObject) if cEnv then CollectObjectReferenceInMemory(strName ..".[userdata:environment]", cEnv, cDumpInfoContainer) end end -- Dump metatable. local cMt = getmetatable(cObject) if cMt then CollectObjectReferenceInMemory(strName ..".[userdata:metatable]", cMt, cDumpInfoContainer) end elseif "string" == strType then -- Add reference and name. cRefInfoContainer[cObject] = (cRefInfoContainer[cObject] and (cRefInfoContainer[cObject] + 1)) or 1 if cNameInfoContainer[cObject] then return end -- Set name. cNameInfoContainer[cObject] = strName .. "[" .. strType .. "]" else -- For "number" and "boolean". (If you want to dump them, uncomment the followed lines.) -- -- Add reference and name. -- cRefInfoContainer[cObject] = (cRefInfoContainer[cObject] and (cRefInfoContainer[cObject] + 1)) or 1 -- if cNameInfoContainer[cObject] then -- return -- end -- -- Set name. -- cNameInfoContainer[cObject] = strName .. "[" .. strType .. ":" .. tostring(cObject) .. "]" end end -- Collect memory reference info of a single object from a root table or function. -- strName - The root object name that start to search, can not be nil. -- cObject - The root object that start to search, can not be nil. -- cDumpInfoContainer - The container of the dump result info. local function CollectSingleObjectReferenceInMemory(strName, cObject, cDumpInfoContainer) if not cObject then return end if not strName then strName = "" end -- Check container. if (not cDumpInfoContainer) then cDumpInfoContainer = CreateObjectReferenceInfoContainer() end -- Check stack. if cDumpInfoContainer.m_nStackLevel > 0 then local cStackInfo = debug.getinfo(cDumpInfoContainer.m_nStackLevel, "Sl") if cStackInfo then cDumpInfoContainer.m_strShortSrc = cStackInfo.short_src cDumpInfoContainer.m_nCurrentLine = cStackInfo.currentline end cDumpInfoContainer.m_nStackLevel = -1 end local cExistTag = cDumpInfoContainer.m_cObjectExistTag local cNameAllAlias = cDumpInfoContainer.m_cObjectAliasName local cAccessTag = cDumpInfoContainer.m_cObjectAccessTag local strType = type(cObject) if "table" == strType then -- Check table with class name. if rawget(cObject, "__cname") then if "string" == type(cObject.__cname) then strName = strName .. "[class:" .. cObject.__cname .. "]" end elseif rawget(cObject, "class") then if "string" == type(cObject.class) then strName = strName .. "[class:" .. cObject.class .. "]" end elseif rawget(cObject, "_className") then if "string" == type(cObject._className) then strName = strName .. "[class:" .. cObject._className .. "]" end end -- Check if table is _G. if cObject == _G then strName = strName .. "[_G]" end -- Get metatable. local bWeakK = false local bWeakV = false local cMt = getmetatable(cObject) if cMt then -- Check mode. local strMode = rawget(cMt, "__mode") if strMode then if "k" == strMode then bWeakK = true elseif "v" == strMode then bWeakV = true elseif "kv" == strMode then bWeakK = true bWeakV = true end end end -- Check if the specified object. if cExistTag[cObject] and (not cNameAllAlias[strName]) then cNameAllAlias[strName] = true end -- Add reference and name. if cAccessTag[cObject] then return end -- Get this name. cAccessTag[cObject] = true -- Dump table key and value. for k, v in pairs(cObject) do -- Check key type. local strKeyType = type(k) if "table" == strKeyType then if not bWeakK then CollectSingleObjectReferenceInMemory(strName .. ".[table:key.table]", k, cDumpInfoContainer) end if not bWeakV then CollectSingleObjectReferenceInMemory(strName .. ".[table:value]", v, cDumpInfoContainer) end elseif "function" == strKeyType then if not bWeakK then CollectSingleObjectReferenceInMemory(strName .. ".[table:key.function]", k, cDumpInfoContainer) end if not bWeakV then CollectSingleObjectReferenceInMemory(strName .. ".[table:value]", v, cDumpInfoContainer) end elseif "thread" == strKeyType then if not bWeakK then CollectSingleObjectReferenceInMemory(strName .. ".[table:key.thread]", k, cDumpInfoContainer) end if not bWeakV then CollectSingleObjectReferenceInMemory(strName .. ".[table:value]", v, cDumpInfoContainer) end elseif "userdata" == strKeyType then if not bWeakK then CollectSingleObjectReferenceInMemory(strName .. ".[table:key.userdata]", k, cDumpInfoContainer) end if not bWeakV then CollectSingleObjectReferenceInMemory(strName .. ".[table:value]", v, cDumpInfoContainer) end else CollectSingleObjectReferenceInMemory(strName .. "." .. k, v, cDumpInfoContainer) end end -- Dump metatable. if cMt then CollectSingleObjectReferenceInMemory(strName ..".[metatable]", cMt, cDumpInfoContainer) end elseif "function" == strType then -- Get function info. local cDInfo = debug.getinfo(cObject, "Su") local cCombinedName = strName .. "[line:" .. tostring(cDInfo.linedefined) .. "@file:" .. cDInfo.short_src .. "]" -- Check if the specified object. if cExistTag[cObject] and (not cNameAllAlias[cCombinedName]) then cNameAllAlias[cCombinedName] = true end -- Write this info. if cAccessTag[cObject] then return end -- Set name. cAccessTag[cObject] = true -- Get upvalues. local nUpsNum = cDInfo.nups for i = 1, nUpsNum do local strUpName, cUpValue = debug.getupvalue(cObject, i) local strUpValueType = type(cUpValue) --print(strUpName, cUpValue) if "table" == strUpValueType then CollectSingleObjectReferenceInMemory(strName .. ".[ups:table:" .. strUpName .. "]", cUpValue, cDumpInfoContainer) elseif "function" == strUpValueType then CollectSingleObjectReferenceInMemory(strName .. ".[ups:function:" .. strUpName .. "]", cUpValue, cDumpInfoContainer) elseif "thread" == strUpValueType then CollectSingleObjectReferenceInMemory(strName .. ".[ups:thread:" .. strUpName .. "]", cUpValue, cDumpInfoContainer) elseif "userdata" == strUpValueType then CollectSingleObjectReferenceInMemory(strName .. ".[ups:userdata:" .. strUpName .. "]", cUpValue, cDumpInfoContainer) end end -- Dump environment table. local getfenv = debug.getfenv if getfenv then local cEnv = getfenv(cObject) if cEnv then CollectSingleObjectReferenceInMemory(strName ..".[function:environment]", cEnv, cDumpInfoContainer) end end elseif "thread" == strType then -- Check if the specified object. if cExistTag[cObject] and (not cNameAllAlias[strName]) then cNameAllAlias[strName] = true end -- Add reference and name. if cAccessTag[cObject] then return end -- Get this name. cAccessTag[cObject] = true -- Dump environment table. local getfenv = debug.getfenv if getfenv then local cEnv = getfenv(cObject) if cEnv then CollectSingleObjectReferenceInMemory(strName ..".[thread:environment]", cEnv, cDumpInfoContainer) end end -- Dump metatable. local cMt = getmetatable(cObject) if cMt then CollectSingleObjectReferenceInMemory(strName ..".[thread:metatable]", cMt, cDumpInfoContainer) end elseif "userdata" == strType then -- Check if the specified object. if cExistTag[cObject] and (not cNameAllAlias[strName]) then cNameAllAlias[strName] = true end -- Add reference and name. if cAccessTag[cObject] then return end -- Get this name. cAccessTag[cObject] = true -- Dump environment table. local getfenv = debug.getfenv if getfenv then local cEnv = getfenv(cObject) if cEnv then CollectSingleObjectReferenceInMemory(strName ..".[userdata:environment]", cEnv, cDumpInfoContainer) end end -- Dump metatable. local cMt = getmetatable(cObject) if cMt then CollectSingleObjectReferenceInMemory(strName ..".[userdata:metatable]", cMt, cDumpInfoContainer) end elseif "string" == strType then -- Check if the specified object. if cExistTag[cObject] and (not cNameAllAlias[strName]) then cNameAllAlias[strName] = true end -- Add reference and name. if cAccessTag[cObject] then return end -- Get this name. cAccessTag[cObject] = true else -- For "number" and "boolean" type, they are not object type, skip. end end -- The base method to dump a mem ref info result into a file. -- strSavePath - The save path of the file to store the result, must be a directory path, If nil or "" then the result will output to console as print does. -- strExtraFileName - If you want to add extra info append to the end of the result file, give a string, nothing will do if set to nil or "". -- nMaxRescords - How many rescords of the results in limit to save in the file or output to the console, -1 will give all the result. -- strRootObjectName - The header info to show the root object name, can be nil. -- cRootObject - The header info to show the root object address, can be nil. -- cDumpInfoResultsBase - The base dumped mem info result, nil means no compare and only output cDumpInfoResults, otherwise to compare with cDumpInfoResults. -- cDumpInfoResults - The compared dumped mem info result, dump itself only if cDumpInfoResultsBase is nil, otherwise dump compared results with cDumpInfoResultsBase. local function OutputMemorySnapshot(strSavePath, strExtraFileName, nMaxRescords, strRootObjectName, cRootObject, cDumpInfoResultsBase, cDumpInfoResults) -- Check results. if not cDumpInfoResults then return end -- Get time format string. local strDateTime = FormatDateTimeNow() -- Collect memory info. local cRefInfoBase = (cDumpInfoResultsBase and cDumpInfoResultsBase.m_cObjectReferenceCount) or nil local cNameInfoBase = (cDumpInfoResultsBase and cDumpInfoResultsBase.m_cObjectAddressToName) or nil local cRefInfo = cDumpInfoResults.m_cObjectReferenceCount local cNameInfo = cDumpInfoResults.m_cObjectAddressToName -- Create a cache result to sort by ref count. local cRes = {} local nIdx = 0 for k in pairs(cRefInfo) do nIdx = nIdx + 1 cRes[nIdx] = k end -- Sort result. table.sort(cRes, function (l, r) return cRefInfo[l] > cRefInfo[r] end) -- Save result to file. local bOutputFile = strSavePath and (string.len(strSavePath) > 0) local cOutputHandle = nil local cOutputEntry = print if bOutputFile then -- Check save path affix. local strAffix = string.sub(strSavePath, -1) if ("/" ~= strAffix) and ("\\" ~= strAffix) then strSavePath = strSavePath .. "/" end -- Combine file name. local strFileName = strSavePath .. "LuaMemRefInfo-All" if (not strExtraFileName) or (0 == string.len(strExtraFileName)) then if cDumpInfoResultsBase then if cConfig.m_bComparedMemoryRefFileAddTime then strFileName = strFileName .. "-[" .. strDateTime .. "].txt" else strFileName = strFileName .. ".txt" end else if cConfig.m_bAllMemoryRefFileAddTime then strFileName = strFileName .. "-[" .. strDateTime .. "].txt" else strFileName = strFileName .. ".txt" end end else if cDumpInfoResultsBase then if cConfig.m_bComparedMemoryRefFileAddTime then strFileName = strFileName .. "-[" .. strDateTime .. "]-[" .. strExtraFileName .. "].txt" else strFileName = strFileName .. "-[" .. strExtraFileName .. "].txt" end else if cConfig.m_bAllMemoryRefFileAddTime then strFileName = strFileName .. "-[" .. strDateTime .. "]-[" .. strExtraFileName .. "].txt" else strFileName = strFileName .. "-[" .. strExtraFileName .. "].txt" end end end local cFile = assert(io.open(strFileName, "w")) cOutputHandle = cFile cOutputEntry = cFile.write end local cOutputer = function (strContent) if cOutputHandle then cOutputEntry(cOutputHandle, strContent) else cOutputEntry(strContent) end end -- Write table header. if cDumpInfoResultsBase then cOutputer("--------------------------------------------------------\n") cOutputer("-- This is compared memory information.\n") cOutputer("--------------------------------------------------------\n") cOutputer("-- Collect base memory reference at line:" .. tostring(cDumpInfoResultsBase.m_nCurrentLine) .. "@file:" .. cDumpInfoResultsBase.m_strShortSrc .. "\n") cOutputer("-- Collect compared memory reference at line:" .. tostring(cDumpInfoResults.m_nCurrentLine) .. "@file:" .. cDumpInfoResults.m_strShortSrc .. "\n") else cOutputer("--------------------------------------------------------\n") cOutputer("-- Collect memory reference at line:" .. tostring(cDumpInfoResults.m_nCurrentLine) .. "@file:" .. cDumpInfoResults.m_strShortSrc .. "\n") end cOutputer("--------------------------------------------------------\n") cOutputer("-- [Table/Function/String Address/Name]\t[Reference Path]\t[Reference Count]\n") cOutputer("--------------------------------------------------------\n") if strRootObjectName and cRootObject then if "string" == type(cRootObject) then cOutputer("-- From Root Object: \"" .. tostring(cRootObject) .. "\" (" .. strRootObjectName .. ")\n") else cOutputer("-- From Root Object: " .. GetOriginalToStringResult(cRootObject) .. " (" .. strRootObjectName .. ")\n") end end -- Save each info. for i, v in ipairs(cRes) do if (not cDumpInfoResultsBase) or (not cRefInfoBase[v]) then if (nMaxRescords > 0) then if (i <= nMaxRescords) then if "string" == type(v) then local strOrgString = tostring(v) local nPattenBegin, nPattenEnd = string.find(strOrgString, "string: \".*\"") if ((not cDumpInfoResultsBase) and ((nil == nPattenBegin) or (nil == nPattenEnd))) then local strRepString = string.gsub(strOrgString, "([\n\r])", "\\n") cOutputer("string: \"" .. strRepString .. "\"\t" .. cNameInfo[v] .. "\t" .. tostring(cRefInfo[v]) .. "\n") else cOutputer(tostring(v) .. "\t" .. cNameInfo[v] .. "\t" .. tostring(cRefInfo[v]) .. "\n") end else cOutputer(GetOriginalToStringResult(v) .. "\t" .. cNameInfo[v] .. "\t" .. tostring(cRefInfo[v]) .. "\n") end end else if "string" == type(v) then local strOrgString = tostring(v) local nPattenBegin, nPattenEnd = string.find(strOrgString, "string: \".*\"") if ((not cDumpInfoResultsBase) and ((nil == nPattenBegin) or (nil == nPattenEnd))) then local strRepString = string.gsub(strOrgString, "([\n\r])", "\\n") cOutputer("string: \"" .. strRepString .. "\"\t" .. cNameInfo[v] .. "\t" .. tostring(cRefInfo[v]) .. "\n") else cOutputer(tostring(v) .. "\t" .. cNameInfo[v] .. "\t" .. tostring(cRefInfo[v]) .. "\n") end else cOutputer(GetOriginalToStringResult(v) .. "\t" .. cNameInfo[v] .. "\t" .. tostring(cRefInfo[v]) .. "\n") end end end end if bOutputFile then io.close(cOutputHandle) cOutputHandle = nil end end -- The base method to dump a mem ref info result of a single object into a file. -- strSavePath - The save path of the file to store the result, must be a directory path, If nil or "" then the result will output to console as print does. -- strExtraFileName - If you want to add extra info append to the end of the result file, give a string, nothing will do if set to nil or "". -- nMaxRescords - How many rescords of the results in limit to save in the file or output to the console, -1 will give all the result. -- cDumpInfoResults - The dumped results. local function OutputMemorySnapshotSingleObject(strSavePath, strExtraFileName, nMaxRescords, cDumpInfoResults) -- Check results. if not cDumpInfoResults then return end -- Get time format string. local strDateTime = FormatDateTimeNow() -- Collect memory info. local cObjectAliasName = cDumpInfoResults.m_cObjectAliasName -- Save result to file. local bOutputFile = strSavePath and (string.len(strSavePath) > 0) local cOutputHandle = nil local cOutputEntry = print if bOutputFile then -- Check save path affix. local strAffix = string.sub(strSavePath, -1) if ("/" ~= strAffix) and ("\\" ~= strAffix) then strSavePath = strSavePath .. "/" end -- Combine file name. local strFileName = strSavePath .. "LuaMemRefInfo-Single" if (not strExtraFileName) or (0 == string.len(strExtraFileName)) then if cConfig.m_bSingleMemoryRefFileAddTime then strFileName = strFileName .. "-[" .. strDateTime .. "].txt" else strFileName = strFileName .. ".txt" end else if cConfig.m_bSingleMemoryRefFileAddTime then strFileName = strFileName .. "-[" .. strDateTime .. "]-[" .. strExtraFileName .. "].txt" else strFileName = strFileName .. "-[" .. strExtraFileName .. "].txt" end end local cFile = assert(io.open(strFileName, "w")) cOutputHandle = cFile cOutputEntry = cFile.write end local cOutputer = function (strContent) if cOutputHandle then cOutputEntry(cOutputHandle, strContent) else cOutputEntry(strContent) end end -- Write table header. cOutputer("--------------------------------------------------------\n") cOutputer("-- Collect single object memory reference at line:" .. tostring(cDumpInfoResults.m_nCurrentLine) .. "@file:" .. cDumpInfoResults.m_strShortSrc .. "\n") cOutputer("--------------------------------------------------------\n") -- Calculate reference count. local nCount = 0 for k in pairs(cObjectAliasName) do nCount = nCount + 1 end -- Output reference count. cOutputer("-- For Object: " .. cDumpInfoResults.m_strAddressName .. " (" .. cDumpInfoResults.m_strObjectName .. "), have " .. tostring(nCount) .. " reference in total.\n") cOutputer("--------------------------------------------------------\n") -- Save each info. for k in pairs(cObjectAliasName) do if (nMaxRescords > 0) then if (i <= nMaxRescords) then cOutputer(k .. "\n") end else cOutputer(k .. "\n") end end if bOutputFile then io.close(cOutputHandle) cOutputHandle = nil end end -- Fileter an existing result file and output it. -- strFilePath - The existing result file. -- strFilter - The filter string. -- bIncludeFilter - Include(true) or exclude(false) the filter. -- bOutputFile - Output to file(true) or console(false). local function OutputFilteredResult(strFilePath, strFilter, bIncludeFilter, bOutputFile) if (not strFilePath) or (0 == string.len(strFilePath)) then print("You need to specify a file path.") return end if (not strFilter) or (0 == string.len(strFilter)) then print("You need to specify a filter string.") return end -- Read file. local cFilteredResult = {} local cReadFile = assert(io.open(strFilePath, "rb")) for strLine in cReadFile:lines() do local nBegin, nEnd = string.find(strLine, strFilter) if nBegin and nEnd then if bIncludeFilter then nBegin, nEnd = string.find(strLine, "[\r\n]") if nBegin and nEnd and (string.len(strLine) == nEnd) then table.insert(cFilteredResult, string.sub(strLine, 1, nBegin - 1)) else table.insert(cFilteredResult, strLine) end end else if not bIncludeFilter then nBegin, nEnd = string.find(strLine, "[\r\n]") if nBegin and nEnd and (string.len(strLine) == nEnd) then table.insert(cFilteredResult, string.sub(strLine, 1, nBegin - 1)) else table.insert(cFilteredResult, strLine) end end end end -- Close and clear read file handle. io.close(cReadFile) cReadFile = nil -- Write filtered result. local cOutputHandle = nil local cOutputEntry = print if bOutputFile then -- Combine file name. local _, _, strResFileName = string.find(strFilePath, "(.*)%.txt") strResFileName = strResFileName .. "-Filter-" .. ((bIncludeFilter and "I") or "E") .. "-[" .. strFilter .. "].txt" local cFile = assert(io.open(strResFileName, "w")) cOutputHandle = cFile cOutputEntry = cFile.write end local cOutputer = function (strContent) if cOutputHandle then cOutputEntry(cOutputHandle, strContent) else cOutputEntry(strContent) end end -- Output result. for i, v in ipairs(cFilteredResult) do cOutputer(v .. "\n") end if bOutputFile then io.close(cOutputHandle) cOutputHandle = nil end end -- Dump memory reference at current time. -- strSavePath - The save path of the file to store the result, must be a directory path, If nil or "" then the result will output to console as print does. -- strExtraFileName - If you want to add extra info append to the end of the result file, give a string, nothing will do if set to nil or "". -- nMaxRescords - How many rescords of the results in limit to save in the file or output to the console, -1 will give all the result. -- strRootObjectName - The root object name that start to search, default is "_G" if leave this to nil. -- cRootObject - The root object that start to search, default is _G if leave this to nil. local function DumpMemorySnapshot(strSavePath, strExtraFileName, nMaxRescords, strRootObjectName, cRootObject) -- Get time format string. local strDateTime = FormatDateTimeNow() -- Check root object. if cRootObject then if (not strRootObjectName) or (0 == string.len(strRootObjectName)) then strRootObjectName = tostring(cRootObject) end else cRootObject = debug.getregistry() strRootObjectName = "registry" end -- Create container. local cDumpInfoContainer = CreateObjectReferenceInfoContainer() local cStackInfo = debug.getinfo(2, "Sl") if cStackInfo then cDumpInfoContainer.m_strShortSrc = cStackInfo.short_src cDumpInfoContainer.m_nCurrentLine = cStackInfo.currentline end -- Collect memory info. CollectObjectReferenceInMemory(strRootObjectName, cRootObject, cDumpInfoContainer) -- Dump the result. OutputMemorySnapshot(strSavePath, strExtraFileName, nMaxRescords, strRootObjectName, cRootObject, nil, cDumpInfoContainer) end -- Dump compared memory reference results generated by DumpMemorySnapshot. -- strSavePath - The save path of the file to store the result, must be a directory path, If nil or "" then the result will output to console as print does. -- strExtraFileName - If you want to add extra info append to the end of the result file, give a string, nothing will do if set to nil or "". -- nMaxRescords - How many rescords of the results in limit to save in the file or output to the console, -1 will give all the result. -- cResultBefore - The base dumped results. -- cResultAfter - The compared dumped results. local function DumpMemorySnapshotCompared(strSavePath, strExtraFileName, nMaxRescords, cResultBefore, cResultAfter) -- Dump the result. OutputMemorySnapshot(strSavePath, strExtraFileName, nMaxRescords, nil, nil, cResultBefore, cResultAfter) end -- Dump compared memory reference file results generated by DumpMemorySnapshot. -- strSavePath - The save path of the file to store the result, must be a directory path, If nil or "" then the result will output to console as print does. -- strExtraFileName - If you want to add extra info append to the end of the result file, give a string, nothing will do if set to nil or "". -- nMaxRescords - How many rescords of the results in limit to save in the file or output to the console, -1 will give all the result. -- strResultFilePathBefore - The base dumped results file. -- strResultFilePathAfter - The compared dumped results file. local function DumpMemorySnapshotComparedFile(strSavePath, strExtraFileName, nMaxRescords, strResultFilePathBefore, strResultFilePathAfter) -- Read results from file. local cResultBefore = CreateObjectReferenceInfoContainerFromFile(strResultFilePathBefore) local cResultAfter = CreateObjectReferenceInfoContainerFromFile(strResultFilePathAfter) -- Dump the result. OutputMemorySnapshot(strSavePath, strExtraFileName, nMaxRescords, nil, nil, cResultBefore, cResultAfter) end -- Dump memory reference of a single object at current time. -- strSavePath - The save path of the file to store the result, must be a directory path, If nil or "" then the result will output to console as print does. -- strExtraFileName - If you want to add extra info append to the end of the result file, give a string, nothing will do if set to nil or "". -- nMaxRescords - How many rescords of the results in limit to save in the file or output to the console, -1 will give all the result. -- strObjectName - The object name reference you want to dump. -- cObject - The object reference you want to dump. local function DumpMemorySnapshotSingleObject(strSavePath, strExtraFileName, nMaxRescords, strObjectName, cObject) -- Check object. if not cObject then return end if (not strObjectName) or (0 == string.len(strObjectName)) then strObjectName = GetOriginalToStringResult(cObject) end -- Get time format string. local strDateTime = FormatDateTimeNow() -- Create container. local cDumpInfoContainer = CreateSingleObjectReferenceInfoContainer(strObjectName, cObject) local cStackInfo = debug.getinfo(2, "Sl") if cStackInfo then cDumpInfoContainer.m_strShortSrc = cStackInfo.short_src cDumpInfoContainer.m_nCurrentLine = cStackInfo.currentline end -- Collect memory info. CollectSingleObjectReferenceInMemory("registry", debug.getregistry(), cDumpInfoContainer) -- Dump the result. OutputMemorySnapshotSingleObject(strSavePath, strExtraFileName, nMaxRescords, cDumpInfoContainer) end -- Return methods. local cPublications = {m_cConfig = nil, m_cMethods = {}, m_cHelpers = {}, m_cBases = {}} cPublications.m_cConfig = cConfig cPublications.m_cMethods.DumpMemorySnapshot = DumpMemorySnapshot cPublications.m_cMethods.DumpMemorySnapshotCompared = DumpMemorySnapshotCompared cPublications.m_cMethods.DumpMemorySnapshotComparedFile = DumpMemorySnapshotComparedFile cPublications.m_cMethods.DumpMemorySnapshotSingleObject = DumpMemorySnapshotSingleObject cPublications.m_cHelpers.FormatDateTimeNow = FormatDateTimeNow cPublications.m_cHelpers.GetOriginalToStringResult = GetOriginalToStringResult cPublications.m_cBases.CreateObjectReferenceInfoContainer = CreateObjectReferenceInfoContainer cPublications.m_cBases.CreateObjectReferenceInfoContainerFromFile = CreateObjectReferenceInfoContainerFromFile cPublications.m_cBases.CreateSingleObjectReferenceInfoContainer = CreateSingleObjectReferenceInfoContainer cPublications.m_cBases.CollectObjectReferenceInMemory = CollectObjectReferenceInMemory cPublications.m_cBases.CollectSingleObjectReferenceInMemory = CollectSingleObjectReferenceInMemory cPublications.m_cBases.OutputMemorySnapshot = OutputMemorySnapshot cPublications.m_cBases.OutputMemorySnapshotSingleObject = OutputMemorySnapshotSingleObject cPublications.m_cBases.OutputFilteredResult = OutputFilteredResult return cPublications