Get object name for handle

get_object_name.cc

// Returns an OBJECT_NAME_INFORMATION object pointed by name
// Caller must free `name` after usage
UNICODE_STRING* GetObjectNameInformation(HANDLE object_handle)
{
    ULONG length = 0;
    UNICODE_STRING* obj = (UNICODE_STRING*)malloc(sizeof(UNICODE_STRING));
    NTSTATUS(*myNtQueryObject)(HANDLE, ObjectInfoClass, UNICODE_STRING*, uint32_t, PULONG) = (NTSTATUS(*)(HANDLE, ObjectInfoClass, UNICODE_STRING*, uint32_t, PULONG))GetProcAddress(GetModuleHandle("ntdll"), "NtQueryObject");
    NTSTATUS status = myNtQueryObject(object_handle, ObjectNameInformation, obj, sizeof(UNICODE_STRING), &length);
    if (!NT_SUCCESS(status) && (status == 0xc0000004 || status == 0x80000005))
    {
        // Buffer too small, reallocating.
        obj = (UNICODE_STRING*)realloc(obj, length);
    }
    else if (!NT_SUCCESS(status) && !(status == 0xc0000004 || status == 0x80000005))
    {
        free(obj);
        return nullptr;
    }
    status = myNtQueryObject(object_handle, ObjectNameInformation, obj, length, nullptr);
    if (NT_SUCCESS(status))
    {
        return obj;
    }
    free(obj);
    return nullptr;
}