DWORD __stdcall TestMsgLoopThread( void* lparam )
{
CEmbTestDlg* pExcObj = (CEmbTestDlg*)lparam;
HINSTANCE hInstance = GetSelfModuleHandle();
CFWriteLog(0, TEXT("start RegisterClassEx"));
WNDCLASSEX wndclassex = {0};
wndclassex.cbSize = sizeof(WNDCLASSEX);
wndclassex.style = CS_HREDRAW | CS_VREDRAW;
wndclassex.lpfnWndProc = TestObjectWndProc;
wndclassex.cbClsExtra = 0;
wndclassex.cbWndExtra = 0;
wndclassex.hInstance = hInstance;
wndclassex.hIcon = NULL;
wndclassex.hCursor = NULL;
wndclassex.hbrBackground = (HBRUSH) GetStockObject (WHITE_BRUSH);;
wndclassex.lpszMenuName = NULL;
CString strClsName = Guid2String(TxGenGuid());
wndclassex.lpszClassName = strClsName;
wndclassex.hIconSm = wndclassex.hIcon;
CFWriteLog(0, TEXT("call RegisterClassEx"));
if ( 0 == RegisterClassEx (&wndclassex))
{
HRESULT hr = GetLastError();
if (hr != 0x00000582)
{
CString strMsg;
strMsg.Format(TEXT("reg class error %d"), hr);
OutputDebugString(strMsg);
CFWriteLog(0, strMsg);
ASSERT(FALSE);
return 0;
}
}
HWND& hwnd = pExcObj->m_hwndExcMsg;
ASSERT(hwnd == NULL);
CFWriteLog(0, TEXT("start CreateWindowEx"));
hwnd = CreateWindowEx (WS_EX_OVERLAPPEDWINDOW,
wndclassex.lpszClassName,
TEXT (""),
WS_OVERLAPPEDWINDOW,
0,
0,
50,
50,
NULL,
NULL,
hInstance,0);
CFWriteLog(0, TEXT("end CreateWindowEx"));
if (hwnd == NULL)
{
HRESULT hr = GetLastError();
CString strMsg;
strMsg.Format(TEXT("wndcreate error %d"), hr);
CFWriteLog(0, strMsg);
OutputDebugString(strMsg);
return 0;
}
CFWriteLog(0, TEXT("start show wnd"));
ShowWindow(hwnd, SW_HIDE);
ASSERT(::IsWindow(hwnd));
//::SetWindowLong(hwnd, GWL_USERDATA, (LONG)pMgr);
//CFWriteLog(TEXT("SockmsgWnd = %x, bind sock %d"), hwnd, this->m_hSock);
//ShowWindow (hwnd, SW_HIDE);
UpdateWindow (hwnd);
SetEvent(pExcObj->m_hEventMsgWndCreated);
CFWriteLog(0, TEXT("enter wnd loop"));
MSG msg;
while (GetMessage(&msg, NULL, 0, 0))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
if(IsWindow(hwnd))
{
DestroyWindow(hwnd);
}
BOOL bSuc = UnregisterClass(strClsName, hInstance);
if(!bSuc)
{
HRESULT hr = GetLastError();
CString strMsg;
strMsg.Format(TEXT("wndcreate error %d"), hr);
ASSERT(FALSE);
CFWriteLog(0, strMsg);
}
CFWriteLog(0, TEXT("UnregisterClass"));
return 0;
}
int __stdcall wlanprof(DWORD *WlanStatus)
{
HWND hWnd = NULL;
WCHAR wsProfileXml[1024];
HINSTANCE hInstance;
DWORD dwNicCnt;
DWORD dwServiceVersion;
HMODULE hModule = NULL;
TCHAR NewDevPath[MAX_PATH];
int Found;
int RetryCount = 0;
BOOL LaterThanXP;
int status = 0;
*WlanStatus = 0;
LaterThanXP = TestForLaterThanXP();
//
// Load the entry points into the WLANAPI DLL to access the WLAN stuff...
//
GetSystemDirectory(NewDevPath, SIZECHARS(NewDevPath));
StringCchCat (NewDevPath, SIZECHARS(NewDevPath), TEXT("\\WLANAPI.DLL"));
hModule = LoadLibrary(NewDevPath);
if (!hModule)
{
//@@@ MessageBox(hWnd, "WLANAPI.DLL not found", "Error", MB_OK);
return -1;
}
//
// Found the DLL...start fetching the entry points.
//
pWlanOpenHandle = (PWLANOPENHANDLE)GetProcAddress(hModule, "WlanOpenHandle");
pWlanCloseHandle = (PWLANCLOSEHANDLE)GetProcAddress(hModule, "WlanCloseHandle");
pWlanEnumInterfaces = (PWLANENUMINTERFACES)GetProcAddress(hModule, "WlanEnumInterfaces");
pWlanSetProfile = (PWLANSETPROFILE)GetProcAddress(hModule, "WlanSetProfile");
pWlanReasonCodeToString = (PWLANREASONCODETOSTRING)GetProcAddress(hModule, "WlanReasonCodeToString");
if (!pWlanOpenHandle || !pWlanCloseHandle || !pWlanEnumInterfaces || !pWlanSetProfile || !pWlanReasonCodeToString)
{
//@@@ MessageBox(hWnd, "Error loading Wifi API procedure(s)", "Error", MB_OK);
return -2;
}
//
// Load the XML profile data from the embedded
// string resource. For XP, there is one profile...
// anything later uses the alternate profile.
//
hInstance = GetModuleHandle(NULL);
LoadStringW (hInstance, IDS_XENWIRELESS, wsProfileXml, 1024);
Found = 0;
do
{
if (pWlanOpenHandle (1, NULL, &dwServiceVersion, &hClientHandle) != ERROR_SUCCESS)
{
// Error opening Wlan Handle
//
// Print no error since we are being launched at user login and the WLAN
// service may not be running yet. So just keep retrying...
//
Sleep (5000);
continue;
}
//
// Fetch the current list of WLAN interfaces
//
if (*WlanStatus =
pWlanEnumInterfaces (hClientHandle, NULL, &ppInterfaceList) != ERROR_SUCCESS)
{
// Error Getting Interfaces
//@@@ MessageBox(hWnd, "Error getting list of device interfaces", "Error", MB_OK);
status = -3;
goto done;
}
//
// Check number of Citrix PV wireless adapters found so far
//
for (dwNicCnt = 0; dwNicCnt < ppInterfaceList->dwNumberOfItems; dwNicCnt++)
{
if (wcsstr(ppInterfaceList->InterfaceInfo[dwNicCnt].strInterfaceDescription, L"Citrix PV"))
Found++;
}
//
// If none found, sleep for 5s and try again
//
if (!Found)
Sleep (5000);
} while (!Found);
MessageBeep ((UINT)-1);
#if 0
if (ListGuids)
{
char entry[512];
for (dwNicCnt = 0; dwNicCnt < ppInterfaceList->dwNumberOfItems; dwNicCnt++)
{
sprintf_s (entry, sizeof(entry), "%s / %S",
Guid2String(ppInterfaceList->InterfaceInfo[dwNicCnt].InterfaceGuid),
ppInterfaceList->InterfaceInfo[dwNicCnt].strInterfaceDescription);
MessageBox(hWnd, entry, "Info", MB_OK);
}
return 0;
}
#endif
//
// At this point, we have found at least one Citrix branded wireless
// adapter. So we can go ahead and set the profile.
//
for (dwNicCnt=0; dwNicCnt < ppInterfaceList->dwNumberOfItems; dwNicCnt++)
{
DWORD pdwReasonCode = 0;
DWORD retWlanSetProfile;
*WlanStatus =
retWlanSetProfile =
pWlanSetProfile (
hClientHandle,
&ppInterfaceList->InterfaceInfo[dwNicCnt].InterfaceGuid,
0, wsProfileXml,
NULL, TRUE, NULL,
&pdwReasonCode);
if (retWlanSetProfile != ERROR_SUCCESS)
{
#if 0
WCHAR swcReason[256];
char err[256];
pWlanReasonCodeToString (pdwReasonCode, 256, swcReason, NULL);
sprintf_s (err, sizeof(err), "[%d] %S", pdwReasonCode, swcReason);
MessageBox(hWnd, err, "Error adding AP", MB_OK);
#endif
Sleep (1000);
dwNicCnt -= 1;
RetryCount++;
}
if (RetryCount > 50)
break;
}
if (RetryCount >= 50)
status = -4;
done: // sorry Mattba...
//FreeLibrary (hModule);
pWlanCloseHandle(hClientHandle, NULL);
//
// We have completed our task, so it is time to remove the
// registry entry that kicks off this util.
//
// HKLM\SOFTWARE\Citrix\XenClient\WlanProf
//
RegDeleteKey (HKEY_LOCAL_MACHINE, TEXT("SOFTWARE\\Citrix\\XenClient\\WlanProf"));
return status;
}
