Files
yttrium-virtio-gpu/NetKVM/NotifyObject/utils.cpp
T
2020-11-02 11:11:23 +02:00

104 lines
4.0 KiB
C++

/*
* Copyright (c) 2020 Red Hat, Inc.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met :
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and / or other materials provided with the distribution.
* 3. Neither the names of the copyright holders nor the names of their contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED.IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include "stdafx.h"
#include "netkvmnoif_i.c"
void Trace (LPCSTR szFormat, ...)
{
va_list arglist;
va_start(arglist, szFormat);
CStringA s;
s.FormatV(szFormat, arglist);
s += '\n';
OutputDebugString(s.GetBuffer());
va_end(arglist);
}
class CSupportedAdapters : public CAtlArray<CStringW>
{
public:
CSupportedAdapters()
{
HKEY hKey;
ULONG NoDefaults = false;
if (!RegOpenKeyExA(HKEY_LOCAL_MACHINE,
"System\\CurrentControlSet\\Control\\VioProt",
0, KEY_READ, &hKey)) {
ULONG len;
WCHAR buffer[1024] = {};
len = sizeof(NoDefaults);
RegGetValueW(hKey, NULL, L"NoDefaults", RRF_RT_DWORD, NULL, &NoDefaults, &len);
len = sizeof(buffer);
if (!RegGetValueW(hKey, NULL, L"Supported", RRF_RT_REG_MULTI_SZ, NULL, buffer, &len)) {
for (UINT i = 0; i < len; )
{
CStringW s = buffer + i;
if (s.IsEmpty())
break;
s.MakeLower();
Add(s);
i += s.GetLength() + 1;
}
}
RegCloseKey(hKey);
}
if (!NoDefaults) {
Add(L"ven_8086&dev_1515"); // Intel X540 Virtual Function
Add(L"ven_8086&dev_10ca"); // Intel 82576 Virtual Function
Add(L"ven_8086&dev_15a8"); // Intel Ethernet Connection X552
Add(L"ven_8086&dev_154c"); // Intel 700 Series VF
Add(L"ven_8086&dev_10ed"); // Intel 82599 VF
Add(L"ven_15b3&dev_101a"); // Mellanox MT28800 Family
Add(L"ven_14e4&dev_16dc"); // Broadcom NetXtreme-E Virtual Function
Add(L"ven_14e4&dev_16af"); // Broadcom NetXtreme II BCM57810 Virtual Function
Add(L"bdrv\\l2ndv&pci_16af14e4"); // Broadcom NetXtreme II BCM57810 Virtual Function (QLogic miniport driver)
}
for (UINT i = 0; i < GetCount(); ++i)
{
CStringW& s = GetAt(i);
Trace("NotifyObj: supported %S", s.GetBuffer());
}
}
};
static CSupportedAdapters SupportedAdapters;
bool IsSupportedSRIOVAdapter(LPCWSTR pnpId)
{
CStringW sId = pnpId;
sId.MakeLower();
for (UINT i = 0; i < SupportedAdapters.GetCount(); ++i)
{
CStringW& s = SupportedAdapters.GetAt(i);
if (sId.Find(s) > 0) {
return true;
}
}
return false;
}