Function main

view/sharedcache/api/python/generator.cpp:240–617

int main(int argc, char* argv[])
{
	if (argc < 5)
	{
		fprintf(stderr, "Usage: generator <header> <output> <output_template> <output_enum>\n");
		return 1;
	}

	// Parse API header to get type and function information
	map<QualifiedName, Ref<Type>> types, vars, funcs;
	string errors;
	auto arch = new CoreArchitecture(BNGetNativeTypeParserArchitecture());

	// Enable ephemeral settings
	Settings::Instance()->LoadSettingsFile("");
	Settings::Instance()->Set("analysis.types.parserName", "ClangTypeParser");
	bool ok = arch->GetStandalonePlatform()->ParseTypesFromSourceFile(argv[1], types, vars, funcs, errors);

	if (!ok)
	{
		fprintf(stderr, "Errors: %s\n", errors.c_str());
		return 1;
	}

	FILE* out = fopen(argv[2], "w");
	FILE* out_template = fopen(argv[3], "r");
	FILE* enums = fopen(argv[4], "w");

	fprintf(enums, "import enum\n");

	// Copy the content of the template to the output file
	int c;
	while((c = fgetc(out_template)) != EOF)
		fputc(c, out);

	// Create type objects
	fprintf(out, "# Type definitions\n");
	for (auto& i : types)
	{
		string name;
		if (i.first.size() != 1)
			continue;
		name = i.first[0];
		if (name == "BNBinaryView")
		{
			fprintf(out, "from binaryninja._binaryninjacore import BNBinaryView, BNBinaryViewHandle\n");
			continue;
		}
		if (i.second->GetClass() == StructureTypeClass)
		{
			fprintf(out, "class %s(ctypes.Structure):\n", name.c_str());

			// python uses str's, C uses byte-arrays
			bool stringField = false;
			for (auto& arg : i.second->GetStructure()->GetMembers())
			{
				if ((arg.type->GetClass() == PointerTypeClass) &&
					(arg.type->GetChildType()->GetWidth() == 1) &&