// Main lua VM debug hook
void LSProfiler::profileHook(lua_State *L, lua_Debug *ar)
{
// Only process calls from the main thread for now
#ifdef LOOM_ENABLE_JIT
lua_State *mainL = mainthread(G(L));
#else
lua_State *mainL = L->l_G->mainthread;
#endif
if (mainL != L)
return;
MethodBase *methodBase = NULL;
if (ar->event == LUA_HOOKCALL)
{
methodBase = getTopMethod(L);
if (methodBase)
{
methodStack.push(methodBase);
enterMethod(methodBase->getFullMemberName());
}
}
if (ar->event == LUA_HOOKRET)
{
methodBase = getTopMethod(L);
if (methodBase)
{
if (methodStack.size())
{
methodStack.pop();
leaveMethod(methodBase->getFullMemberName());
}
}
}
}
void LSProfiler::getCurrentStack(lua_State *L, utStack<MethodBase *>& stack)
{
int top = lua_gettop(L);
lua_Debug lstack;
int stackFrame = 0;
MethodBase *lastMethod = NULL;
while (true)
{
// if we get a null result here, we are out of stack
if (!lua_getstack(L, stackFrame++, &lstack))
{
lua_settop(L, top);
return;
}
// something bad in denmark
if (!lua_getinfo(L, "f", &lstack))
{
lua_settop(L, top);
return;
}
bool cfunc = false;
if (lua_iscfunction(L, -1))
{
cfunc = true;
}
lua_rawgeti(L, LUA_GLOBALSINDEX, LSINDEXMETHODLOOKUP);
lua_pushvalue(L, -2);
lua_rawget(L, -2);
if (lua_isnil(L, -1))
{
lua_settop(L, top);
continue;
}
MethodBase *methodBase = (MethodBase *)lua_topointer(L, -1);
lua_settop(L, top);
#ifdef LOOM_ENABLE_JIT
// We do not receive line return hooks for native calls under JIT :(
// So, don't add to initial stack
if (cfunc)
{
continue;
}
#endif
if (shouldFilterFunction(methodBase->getFullMemberName()))
{
continue;
}
// we only want the root call, not the pcall wrapper
if (cfunc && (lastMethod == methodBase))
{
continue;
}
lastMethod = methodBase;
stack.push(methodBase);
}
}
void LSProfiler::dumpAllocations(lua_State *L)
{
lua_gc(L, LUA_GCCOLLECT, 0);
#ifdef LOOM_ENABLE_JIT
lmLog(gProfilerLogGroup, "Please note: Profiling under JIT does not include native function calls.");
lmLog(gProfilerLogGroup, "switch to the interpreted VM in order to gather native method allocation");
#endif
utList<LSProfilerTypeAllocation *> allocs;
utHashTable<utPointerHashKey, MethodAllocation> methodAggr;
// Prepare for object allocation dump
for (UTsize i = 0; i < allocations.size(); i++)
{
LSProfilerTypeAllocation* alloc = allocations.at(i);
allocs.push_back(alloc);
}
lmLog(gProfilerLogGroup, "");
lmLog(gProfilerLogGroup, "Object Allocation Dump");
lmLog(gProfilerLogGroup, "----------------------");
while (allocs.size())
{
int best = -1;
LSProfilerTypeAllocation *bestType = NULL;
for (UTsize i = 0; i < allocs.size(); i++)
{
LSProfilerTypeAllocation *alloc = allocs.at(i);
if (alloc->alive > best)
{
best = alloc->alive;
bestType = alloc;
}
}
lmAssert(bestType, "couldn't get bestType");
allocs.erase(bestType);
lmLog(gProfilerLogGroup, "Alive: %i (%i KiB), Total: %i (%i KiB), Type: %s", bestType->alive, bestType->memoryCurrent/1024, bestType->total, bestType->memoryTotal/1024, bestType->type->getFullName().c_str());
if (bestType->anonTotal)
{
lmLog(gProfilerLogGroup, " Alive: %i, Total %i (Anonymous)", bestType->anonTotal, bestType->anonCurrent);
}
for (UTsize j = 0; j < bestType->methodCurrent.size(); j++)
{
MethodBase *methodBase = (MethodBase *)bestType->methodCurrent.keyAt(j).key();
lmAssert(methodBase == (MethodBase *)bestType->methodTotal.keyAt(j).key(), "mismatched methodbase");
int *value1 = bestType->methodCurrent.get(methodBase);
int *value2 = bestType->methodTotal.get(methodBase);
lmAssert(value1 && value2, "bad pointer on allocation");
lmLog(gProfilerLogGroup, " Alive %i, Total %i (%s)", (*value1), (*value2), methodBase->getFullMemberName());
}
}
// Aggregate method allocations
for (UTsize i = 0; i < allocations.size(); i++)
{
LSProfilerTypeAllocation* alloc = allocations.at(i);
utHashTable<utPointerHashKey, int> &methodCurrent = alloc->methodCurrent;
utHashTable<utPointerHashKey, int> &methodTotal = alloc->methodTotal;
for (UTsize j = 0; j < methodCurrent.size(); j++) {
utPointerHashKey key = methodCurrent.keyAt(j);
int *current = methodCurrent.get(key);
lmAssert(current != NULL, "Internal hash table error");
int *total = methodTotal.get(key);
lmAssert(total != NULL, "Internal total hash table inconsistency error");
MethodAllocation *aggr = methodAggr.get(key);
if (aggr == NULL)
{
// First seen method, insert initial values
MethodAllocation ma;
ma.currentCount = *current;
ma.totalCount = *total;
ma.currentBytes = alloc->memoryCurrent;
ma.totalBytes = alloc->memoryTotal;
ma.allocations.push_back(alloc);
methodAggr.insert(key, ma);
}
else
{
// Method already seen, update tracked values
aggr->currentCount += *current;
aggr->totalCount += *total;
aggr->currentBytes += alloc->memoryCurrent;
aggr->totalBytes += alloc->memoryTotal;
aggr->allocations.push_back(alloc);
}
}
}
utArray<int> sortByTotal;
sortByTotal.resize(methodAggr.size());
for (UTsize i = 0; i < sortByTotal.size(); i++) sortByTotal[i] = i;
sortMethods = &methodAggr;
SDL_qsort((void*)sortByTotal.ptr(), sortByTotal.size(), sizeof(int), sortMethodsByTotalCount);
sortMethods = NULL;
lmLog(gProfilerLogGroup, "");
lmLog(gProfilerLogGroup, "Aggregated Method Allocation");
lmLog(gProfilerLogGroup, "----------------------------");
for (UTsize i = 0; i < methodAggr.size(); i++)
{
utPointerHashKey key = methodAggr.keyAt(sortByTotal[i]);
MethodBase *methodBase = (MethodBase *)key.key();
lmAssert(methodBase != NULL, "Internal aggregated method base missing");
MethodAllocation &ma = *methodAggr.get(key);
lmLog(gProfilerLogGroup, "Total: %i (%i KiB), Alive: %i (%i KiB), Method: %s", ma.totalCount, ma.totalBytes/1024, ma.currentCount, ma.currentBytes/1024, methodBase->getFullMemberName());
//SDL_qsort((void*)ma.allocations.ptr(), ma.allocations.size(), sizeof(LSProfilerTypeAllocation*), sortAllocsByTotal);
for (UTsize j = 0; j < ma.allocations.size(); j++)
{
LSProfilerTypeAllocation* alloc = ma.allocations.at(j);
int &methodCurrent = *alloc->methodCurrent.get(key);
int &methodTotal = *alloc->methodTotal.get(key);
lmLog(gProfilerLogGroup, " Total: %i / %i (%i kiB), Alive: %i / %i (%i KiB), Type: %s", methodTotal, alloc->total, alloc->memoryTotal / 1024, methodCurrent, alloc->alive, alloc->memoryCurrent / 1024, alloc->type->getFullName().c_str());
}
}
}
MethodBase *LSProfiler::getTopMethod(lua_State *L)
{
int top = lua_gettop(L);
lua_Debug stack;
// if we get a null result here, we are out of stack
if (!lua_getstack(L, 0, &stack))
{
lua_settop(L, top);
return NULL;
}
if (!lua_getinfo(L, "f", &stack))
{
lua_settop(L, top);
return NULL;
}
bool iscfunc = false;
if (lua_iscfunction(L, -1))
{
iscfunc = true;
}
#ifdef LOOM_ENABLE_JIT
// We do not receive line return hooks for native calls under JIT :(
// So, if we want to profile native methods, we need to be under interpreted VM
if (iscfunc)
{
lua_settop(L, top);
return NULL;
}
#endif
lua_rawgeti(L, LUA_GLOBALSINDEX, LSINDEXMETHODLOOKUP);
lua_pushvalue(L, -2);
lua_rawget(L, -2);
if (lua_isnil(L, -1))
{
lua_settop(L, top);
return NULL;
}
MethodBase *methodBase = (MethodBase *)lua_topointer(L, -1);
lua_settop(L, top);
if (iscfunc && !methodBase->isNative())
{
return NULL;
}
if (shouldFilterFunction(methodBase->getFullMemberName()))
{
return NULL;
}
return methodBase;
}