289 lines
8.1 KiB
C++
289 lines
8.1 KiB
C++
#include <pch.h>
|
|
#pragma hdrstop
|
|
|
|
#include "nsbase.h"
|
|
|
|
#include "ncreg.h"
|
|
#include "ncsetup.h"
|
|
#include "ndispnp.h"
|
|
#include "resource.h"
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: GetLocationInfo
|
|
//
|
|
// Purpose: Gets the slot and port number of a device and formats
|
|
// a display string into a buffer.
|
|
//
|
|
// Arguments:
|
|
// pszDevNodeId [in] The device isntance id of the adapter.
|
|
// pszBuffer [in] Buffer to add location string.
|
|
// (must be preallocated)
|
|
//
|
|
// Returns:
|
|
//
|
|
// Author: billbe 2 Aug 1999
|
|
//
|
|
// Notes: Slot and/or port number may not exist so the buffer
|
|
// may not be modified.
|
|
//
|
|
VOID
|
|
GetLocationInfo (
|
|
IN PCWSTR pszDevNodeId,
|
|
OUT PWSTR pszLocation)
|
|
{
|
|
HDEVINFO hdi;
|
|
SP_DEVINFO_DATA deid;
|
|
HRESULT hr;
|
|
|
|
// Create the device info set needed to access SetupDi fcns.
|
|
//
|
|
hr = HrSetupDiCreateDeviceInfoList (&GUID_DEVCLASS_NET, NULL, &hdi);
|
|
|
|
if (S_OK == hr)
|
|
{
|
|
TraceTag (ttidLanUi, "Opening %S", pszDevNodeId);
|
|
// Open the device info for the adapter.
|
|
//
|
|
hr = HrSetupDiOpenDeviceInfo (hdi, pszDevNodeId, NULL, 0, &deid);
|
|
|
|
if (S_OK == hr)
|
|
{
|
|
BOOL fHaveSlotNumber;
|
|
DWORD dwSlotNumber;
|
|
DWORD dwPortNumber;
|
|
BOOL fHavePortNumber;
|
|
|
|
// Slot number is stored as the UINumber registry property.
|
|
//
|
|
hr = HrSetupDiGetDeviceRegistryProperty (hdi, &deid,
|
|
SPDRP_UI_NUMBER, NULL, (BYTE*)&dwSlotNumber,
|
|
sizeof (dwSlotNumber), NULL);
|
|
|
|
TraceTag (ttidLanUi, "Getting ui number result %lX %d",
|
|
hr, dwSlotNumber);
|
|
|
|
fHaveSlotNumber = (S_OK == hr);
|
|
|
|
// Port information is stored by the class installer in the
|
|
// device key.
|
|
//
|
|
HKEY hkey;
|
|
fHavePortNumber = FALSE;
|
|
hr = HrSetupDiOpenDevRegKey (hdi, &deid, DICS_FLAG_GLOBAL, 0,
|
|
DIREG_DEV, KEY_READ, &hkey);
|
|
|
|
if (S_OK == hr)
|
|
{
|
|
hr = HrRegQueryDword(hkey, L"Port", &dwPortNumber);
|
|
|
|
fHavePortNumber = (S_OK == hr);
|
|
|
|
RegCloseKey (hkey);
|
|
}
|
|
|
|
// Format the string according to what information
|
|
// we were able to retrieve.
|
|
//
|
|
HINSTANCE hinst = _Module.GetResourceInstance();
|
|
if (fHaveSlotNumber && fHavePortNumber)
|
|
{
|
|
swprintf (pszLocation,
|
|
SzLoadString (hinst, IDS_SLOT_PORT_LOCATION),
|
|
dwSlotNumber, dwPortNumber);
|
|
TraceTag (ttidLanUi, "Found slot and port. %S", pszLocation);
|
|
}
|
|
else if (fHaveSlotNumber)
|
|
{
|
|
swprintf (pszLocation,
|
|
SzLoadString (hinst, IDS_SLOT_LOCATION),
|
|
dwSlotNumber);
|
|
TraceTag (ttidLanUi, "Found slot. %S", pszLocation);
|
|
}
|
|
else if (fHavePortNumber)
|
|
{
|
|
swprintf (pszLocation,
|
|
SzLoadString (hinst, IDS_PORT_LOCATION),
|
|
dwPortNumber);;
|
|
TraceTag (ttidLanUi, "Found port. %S", pszLocation);
|
|
}
|
|
}
|
|
SetupDiDestroyDeviceInfoList (hdi);
|
|
}
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: AppendMacAddress
|
|
//
|
|
// Purpose: Appends the MAC address of a LAN adapter to a buffer for
|
|
// display in UI.
|
|
//
|
|
// Arguments:
|
|
// pszBindName [in] Bind name of adapter.
|
|
// pszBuffer [in] Buffer to add MAC address string.
|
|
// (must be preallocated)
|
|
//
|
|
// Returns:
|
|
//
|
|
// Author: tongl 17 Sept 1998
|
|
// billbe 3 Aug 1999 Modified for datatip
|
|
//
|
|
// Notes:
|
|
//
|
|
VOID
|
|
AppendMacAddress (
|
|
IN PCWSTR pszBindName,
|
|
IN OUT PWSTR pszBuffer)
|
|
{
|
|
Assert (pszBindName);
|
|
Assert (pszBuffer);
|
|
|
|
WCHAR szExport[_MAX_PATH];
|
|
|
|
if (pszBindName)
|
|
{
|
|
wcscpy (szExport, L"\\Device\\");
|
|
wcscat (szExport, pszBindName);
|
|
|
|
UNICODE_STRING ustrDevice;
|
|
RtlInitUnicodeString(&ustrDevice, szExport);
|
|
|
|
// Get the Mac Address
|
|
UINT uiRet;
|
|
UCHAR MacAddr[6];
|
|
UCHAR PMacAddr[6];
|
|
UCHAR VendorId[3];
|
|
uiRet = NdisQueryHwAddress(&ustrDevice, MacAddr, PMacAddr, VendorId);
|
|
|
|
if (uiRet)
|
|
{
|
|
// Succeeded
|
|
WCHAR pszNumber[32];
|
|
*pszNumber = 0;
|
|
|
|
WCHAR szBuff[4];
|
|
|
|
for (INT i=0; i<=5; i++)
|
|
{
|
|
wsprintfW(szBuff, L"%02X", MacAddr[i]);
|
|
wcscat(pszNumber, szBuff);
|
|
|
|
if (i != 5)
|
|
{
|
|
wcscat(pszNumber, L"-");
|
|
}
|
|
}
|
|
|
|
if (*pszBuffer)
|
|
{
|
|
wcscat (pszBuffer, L"\n");
|
|
}
|
|
DwFormatString(SzLoadString (_Module.GetResourceInstance(), IDS_MAC_ADDRESS),
|
|
pszBuffer + wcslen (pszBuffer),
|
|
_MAX_PATH,
|
|
pszNumber);
|
|
}
|
|
}
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CreateDeviceDataTip
|
|
//
|
|
// Purpose: Creates a data tip that will display device specific
|
|
// information when the user hovers over nIdTool.
|
|
//
|
|
// Arguments:
|
|
// hwndParent [in] hwnd to parent window.
|
|
// phwndDataTip [in] pointer to the hwnd of data tip. Must be
|
|
// preallocated and the hwnd assigned to NULL if
|
|
// data tip has not been created.
|
|
// nIdTool [in] resource is of tool to add datatip to.
|
|
// pszDevNodeId [in] The device isntance id of the adapter.
|
|
// pszBindName [in] Bind name of adapter.
|
|
//
|
|
// Returns: nothing
|
|
//
|
|
// Author: billbe 2 Aug 1999
|
|
//
|
|
// Notes:
|
|
//
|
|
VOID
|
|
CreateDeviceDataTip (
|
|
IN HWND hwndParent,
|
|
IN OUT HWND* phwndDataTip,
|
|
IN UINT nIdTool,
|
|
IN PCWSTR pszDevNodeId,
|
|
IN PCWSTR pszBindName)
|
|
{
|
|
if (!*phwndDataTip)
|
|
{
|
|
TraceTag (ttidLanUi, "Creating device datatip!!!");
|
|
|
|
*phwndDataTip = CreateWindowExW (0, TOOLTIPS_CLASS, NULL,
|
|
WS_POPUP | TTS_ALWAYSTIP,
|
|
CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
|
|
CW_USEDEFAULT, hwndParent, NULL, NULL, NULL);
|
|
|
|
if (*phwndDataTip)
|
|
{
|
|
SetWindowPos (*phwndDataTip, HWND_TOPMOST, CW_USEDEFAULT,
|
|
CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
|
|
SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE);
|
|
}
|
|
}
|
|
|
|
if (*phwndDataTip)
|
|
{
|
|
TTTOOLINFOW toolinfo = {0};
|
|
toolinfo.cbSize = sizeof (toolinfo);
|
|
toolinfo.uId = (UINT_PTR)GetDlgItem (hwndParent, nIdTool);
|
|
toolinfo.uFlags = TTF_SUBCLASS | TTF_IDISHWND;
|
|
|
|
WCHAR szDataTip[_MAX_PATH] = {0};
|
|
|
|
// Get location info.
|
|
if (pszDevNodeId)
|
|
{
|
|
GetLocationInfo (pszDevNodeId, szDataTip);
|
|
}
|
|
|
|
// Append Mac address.
|
|
if (pszBindName)
|
|
{
|
|
AppendMacAddress (pszBindName, szDataTip);
|
|
}
|
|
|
|
// If there is anything to display, set the data tip.
|
|
//
|
|
if (*szDataTip)
|
|
{
|
|
toolinfo.lpszText = szDataTip;
|
|
SendMessage (*phwndDataTip, TTM_ADDTOOL, 0, (LPARAM)&toolinfo);
|
|
|
|
// In order to use '\n' to move to the next line of the data tip,
|
|
// we need to set the width. We will set it to have of the
|
|
// promary monitor's screen size.
|
|
//
|
|
DWORD dwToolTipWidth = GetSystemMetrics (SM_CXSCREEN) / 2;
|
|
if (dwToolTipWidth)
|
|
{
|
|
SendMessage (*phwndDataTip, TTM_SETMAXTIPWIDTH, 0, dwToolTipWidth);
|
|
}
|
|
|
|
// Keep the tip up for 30 seconds.
|
|
SendMessage (*phwndDataTip, TTM_SETDELAYTIME, TTDT_AUTOPOP,
|
|
MAKELONG (30000, 0));
|
|
TraceTag (ttidLanUi, "Creating device datatip complete!!!");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
TraceTag (ttidError, "Creating datatip failed");
|
|
}
|
|
}
|
|
|
|
|