IDA VMT Dumper Script

#include <idc.idc>

static main()
{
	auto pAddress, iIndex;
	auto szFilePath, hFile;
	auto skipAmt;

	SetStatus(IDA_STATUS_WORK);

	// User selected vtable block
	pAddress = ScreenEA();

	if (pAddress == BADADDR)
	{
		Message("** No vtable selected! Aborted **");
		Warning("No vtable selected!\nSelect vtable block first.");
		SetStatus(IDA_STATUS_READY);
		return;
	}

	skipAmt = AskLong(1, "Number of vtable entries to ignore for indexing:");

	// Request output header file
	SetStatus(IDA_STATUS_WAITING);
	if ((szFilePath = AskFile(1, "*.txt", "Select output dump file:")) == 0)
	{
		Message("Aborted.");
		SetStatus(IDA_STATUS_READY);
		return;
	}

	// And create it..
	if ((hFile = fopen(szFilePath, "wb")) != 0)
	{
		auto szFuncName, szFullName, BadHits;

		BadHits = 0;

			/* For linux, skip the first entry */
		if (Dword(pAddress) == 0)
		{
			pAddress = pAddress + 4;
		}

		pAddress = pAddress + (skipAmt * 4);

		// Create the header
                        auto className = Demangle(Name(Dword(pAddress)), INF_SHORT_DN);
		fprintf(hFile, "// Auto reconstructed from vtable block @ 0x%08X\n// from \"%s\", by ida_vtables.idc\n// Modified VTable dumper script obviously by t.me/Gaztoof.\nclass %s\n{\npublic:\n//Don't forget the constructor.\n", pAddress, GetInputFile(), className);

		// Loop through the vtable block
		while (pAddress != BADADDR)
		{
			auto real_addr;
			real_addr = Dword(pAddress);

			szFuncName = Name(real_addr);
			if (strlen(szFuncName) == 0)
			{
			break;
			}

			szFullName = demangle_name(szFuncName, 0);

			if (trim(szFullName) == "")
			{
			szFullName = szFuncName;
			}
			if (strstr(szFullName, "_ZN") != -1)
			{
			fclose(hFile);
			Warning("You must toggle GCC v3.x demangled names!\n");
			break;
			}

	fprintf(hFile, "/*%d*/\tvirtual void* %s = 0;\n", iIndex, szFullName);

			pAddress = pAddress + 4;
			iIndex++;
		};

		fprintf(hFile, "};");
		fclose(hFile);
		Message("Successfully wrote %d vtable entries.\n", iIndex);
	}
	else
	{
	Message("** Error opening \"%s\"! Aborted **\n", szFilePath);
	Warning("Error creating \"%s\"!\n", szFilePath);
}

Message("\nDone.\n\n");
SetStatus(IDA_STATUS_READY);
}