226 lines
10 KiB
C++
226 lines
10 KiB
C++
/*
|
|
Copyright 2015-2016 Amebis
|
|
Copyright 2016 GÉANT
|
|
|
|
This file is part of GEANTLink.
|
|
|
|
GEANTLink is free software: you can redistribute it and/or modify it
|
|
under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
GEANTLink is distributed in the hope that it will be useful, but
|
|
WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with GEANTLink. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <StdAfx.h>
|
|
|
|
|
|
#if EAPMETHOD_TYPE==21
|
|
#define _EAPMETHOD_PEER_UI eap::peer_ttls_ui
|
|
#else
|
|
#error Unknown EAP Method type.
|
|
#endif
|
|
|
|
_EAPMETHOD_PEER_UI g_peer;
|
|
|
|
|
|
///
|
|
/// DLL main entry point
|
|
///
|
|
/// \sa [DllMain entry point](https://msdn.microsoft.com/en-us/library/windows/desktop/ms682583.aspx)
|
|
///
|
|
BOOL WINAPI DllMain(_In_ HINSTANCE hinstDLL, _In_ DWORD fdwReason, _In_ LPVOID lpvReserved)
|
|
{
|
|
UNREFERENCED_PARAMETER(hinstDLL);
|
|
UNREFERENCED_PARAMETER(lpvReserved);
|
|
|
|
if (fdwReason == DLL_PROCESS_ATTACH) {
|
|
#ifdef _DEBUG
|
|
//MessageBox(NULL, _T("Attach debugger!"), _T(__FUNCTION__), MB_OK);
|
|
#endif
|
|
if (g_peer.create() != ERROR_SUCCESS)
|
|
return FALSE;
|
|
} else if (fdwReason == DLL_PROCESS_DETACH)
|
|
assert(!_CrtDumpMemoryLeaks());
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
///
|
|
/// Releases all memory associated with an opaque user interface context data buffer.
|
|
///
|
|
/// \sa [EapPeerFreeMemory function](https://msdn.microsoft.com/en-us/library/windows/desktop/aa363606.aspx)
|
|
///
|
|
VOID WINAPI EapPeerFreeMemory(_In_ void *pUIContextData)
|
|
{
|
|
g_peer.free_memory(pUIContextData);
|
|
}
|
|
|
|
|
|
///
|
|
/// Releases error-specific memory allocated by the EAP peer method.
|
|
///
|
|
/// \sa [EapPeerFreeErrorMemory function](https://msdn.microsoft.com/en-us/library/windows/desktop/aa363605.aspx)
|
|
///
|
|
VOID WINAPI EapPeerFreeErrorMemory(_In_ EAP_ERROR *ppEapError)
|
|
{
|
|
g_peer.free_error_memory(ppEapError);
|
|
}
|
|
|
|
|
|
///
|
|
/// Raises the EAP method's specific connection configuration user interface dialog on the client.
|
|
///
|
|
/// \sa [EapPeerInvokeConfigUI function](https://msdn.microsoft.com/en-us/library/windows/desktop/aa363614.aspx)
|
|
///
|
|
DWORD WINAPI EapPeerInvokeConfigUI(
|
|
_In_ /*const*/ EAP_METHOD_TYPE *pEapType,
|
|
_In_ HWND hwndParent,
|
|
_In_ DWORD dwFlags,
|
|
_In_ DWORD dwSizeOfConnectionDataIn,
|
|
_In_count_(dwSizeOfConnectionDataIn) /*const*/ BYTE *pConnectionDataIn,
|
|
_Out_ DWORD *pdwSizeOfConnectionDataOut,
|
|
_Out_ BYTE **ppConnectionDataOut,
|
|
_Out_ EAP_ERROR **ppEapError)
|
|
{
|
|
DWORD dwResult = NO_ERROR;
|
|
#ifdef _DEBUG
|
|
//MessageBox(NULL, _T("Attach debugger!"), _T(__FUNCTION__), MB_OK);
|
|
#endif
|
|
|
|
// Parameter check
|
|
if (!ppEapError)
|
|
dwResult = ERROR_INVALID_PARAMETER;
|
|
else if (!pEapType)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_INVALID_PARAMETER, 0, NULL, NULL, NULL, _T(__FUNCTION__) _T(" pEapType is NULL."), NULL);
|
|
else if (pEapType->eapType.type != EAPMETHOD_TYPE)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_NOT_SUPPORTED, 0, NULL, NULL, NULL, winstd::wstring_printf(_T(__FUNCTION__) _T(" Input EAP type (%d) does not match the supported EAP type (%d)."), (int)pEapType->eapType.type, (int)EAPMETHOD_TYPE).c_str(), NULL);
|
|
else if (pEapType->dwAuthorId != 67532)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_NOT_SUPPORTED, 0, NULL, NULL, NULL, winstd::wstring_printf(_T(__FUNCTION__) _T(" EAP author (%d) does not match the supported author (%d)."), (int)pEapType->dwAuthorId, (int)67532).c_str(), NULL);
|
|
else if (!pdwSizeOfConnectionDataOut)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_INVALID_PARAMETER, 0, NULL, NULL, NULL, _T(__FUNCTION__) _T(" pdwSizeOfConnectionDataOut is NULL."), NULL);
|
|
else if (!ppConnectionDataOut)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_INVALID_PARAMETER, 0, NULL, NULL, NULL, _T(__FUNCTION__) _T(" ppConnectionDataOut is NULL."), NULL);
|
|
else
|
|
dwResult = g_peer.invoke_config_ui(
|
|
pEapType,
|
|
hwndParent,
|
|
dwFlags,
|
|
dwSizeOfConnectionDataIn,
|
|
pConnectionDataIn,
|
|
pdwSizeOfConnectionDataOut,
|
|
ppConnectionDataOut,
|
|
ppEapError);
|
|
|
|
return dwResult;
|
|
}
|
|
|
|
|
|
///
|
|
/// Raises a custom interactive user interface dialog to obtain user identity information for the EAP method on the client.
|
|
///
|
|
/// \sa [EapPeerInvokeIdentityUI function](https://msdn.microsoft.com/en-us/library/windows/desktop/aa363615.aspx)
|
|
///
|
|
DWORD WINAPI EapPeerInvokeIdentityUI(
|
|
_In_ /*const*/ EAP_METHOD_TYPE *pEapType,
|
|
_In_ DWORD dwFlags,
|
|
_In_ HWND hwndParent,
|
|
_In_ DWORD dwSizeOfConnectionData,
|
|
_In_count_(dwSizeOfConnectionData) const BYTE *pConnectionData,
|
|
_In_ DWORD dwSizeOfUserData,
|
|
_In_count_(dwSizeOfUserData) const BYTE *pUserData,
|
|
_Out_ DWORD *pdwSizeOfUserDataOut,
|
|
_Out_ BYTE **ppUserDataOut,
|
|
_Out_ LPWSTR *ppwszIdentity,
|
|
_Out_ EAP_ERROR **ppEapError)
|
|
{
|
|
DWORD dwResult = NO_ERROR;
|
|
#ifdef _DEBUG
|
|
//MessageBox(NULL, _T("Attach debugger!"), _T(__FUNCTION__), MB_OK);
|
|
#endif
|
|
|
|
// Parameter check
|
|
if (!ppEapError)
|
|
dwResult = ERROR_INVALID_PARAMETER;
|
|
else if (!pEapType)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_INVALID_PARAMETER, 0, NULL, NULL, NULL, _T(__FUNCTION__) _T(" pEapType is NULL."), NULL);
|
|
else if (pEapType->eapType.type != EAPMETHOD_TYPE)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_NOT_SUPPORTED, 0, NULL, NULL, NULL, winstd::wstring_printf(_T(__FUNCTION__) _T(" Input EAP type (%d) does not match the supported EAP type (%d)."), (int)pEapType->eapType.type, (int)EAPMETHOD_TYPE).c_str(), NULL);
|
|
else if (pEapType->dwAuthorId != 67532)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_NOT_SUPPORTED, 0, NULL, NULL, NULL, winstd::wstring_printf(_T(__FUNCTION__) _T(" EAP author (%d) does not match the supported author (%d)."), (int)pEapType->dwAuthorId, (int)67532).c_str(), NULL);
|
|
else if (!pdwSizeOfUserDataOut)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_INVALID_PARAMETER, 0, NULL, NULL, NULL, _T(__FUNCTION__) _T(" pdwSizeOfUserDataOut is NULL."), NULL);
|
|
else if (!ppUserDataOut)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_INVALID_PARAMETER, 0, NULL, NULL, NULL, _T(__FUNCTION__) _T(" ppUserDataOut is NULL."), NULL);
|
|
else if (!ppwszIdentity)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_INVALID_PARAMETER, 0, NULL, NULL, NULL, _T(__FUNCTION__) _T(" ppwszIdentity is NULL."), NULL);
|
|
else
|
|
dwResult = g_peer.invoke_identity_ui(
|
|
pEapType,
|
|
dwFlags,
|
|
hwndParent,
|
|
dwSizeOfConnectionData,
|
|
pConnectionData,
|
|
dwSizeOfUserData,
|
|
pUserData,
|
|
pdwSizeOfUserDataOut,
|
|
ppUserDataOut,
|
|
ppwszIdentity,
|
|
ppEapError);
|
|
|
|
return dwResult;
|
|
}
|
|
|
|
|
|
///
|
|
/// Raises a custom interactive user interface dialog for the EAP method on the client.
|
|
///
|
|
/// \sa [EapPeerInvokeInteractiveUI function](https://msdn.microsoft.com/en-us/library/windows/desktop/aa363616.aspx)
|
|
///
|
|
DWORD WINAPI EapPeerInvokeInteractiveUI(
|
|
_In_ /*const*/ EAP_METHOD_TYPE *pEapType,
|
|
_In_ HWND hwndParent,
|
|
_In_ DWORD dwSizeofUIContextData,
|
|
_In_count_(dwSizeofUIContextData) /*const*/ BYTE *pUIContextData,
|
|
_Out_ DWORD *pdwSizeOfDataFromInteractiveUI,
|
|
_Out_ BYTE **ppDataFromInteractiveUI,
|
|
_Out_ EAP_ERROR **ppEapError)
|
|
{
|
|
DWORD dwResult = NO_ERROR;
|
|
#ifdef _DEBUG
|
|
//MessageBox(NULL, _T("Attach debugger!"), _T(__FUNCTION__), MB_OK);
|
|
#endif
|
|
|
|
// Parameter check
|
|
if (!ppEapError)
|
|
dwResult = ERROR_INVALID_PARAMETER;
|
|
else if (!pEapType)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_INVALID_PARAMETER, 0, NULL, NULL, NULL, _T(__FUNCTION__) _T(" pEapType is NULL."), NULL);
|
|
else if (pEapType->eapType.type != EAPMETHOD_TYPE)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_NOT_SUPPORTED, 0, NULL, NULL, NULL, winstd::wstring_printf(_T(__FUNCTION__) _T(" Input EAP type (%d) does not match the supported EAP type (%d)."), (int)pEapType->eapType.type, (int)EAPMETHOD_TYPE).c_str(), NULL);
|
|
else if (pEapType->dwAuthorId != 67532)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_NOT_SUPPORTED, 0, NULL, NULL, NULL, winstd::wstring_printf(_T(__FUNCTION__) _T(" EAP author (%d) does not match the supported author (%d)."), (int)pEapType->dwAuthorId, (int)67532).c_str(), NULL);
|
|
else if (!pdwSizeOfDataFromInteractiveUI)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_INVALID_PARAMETER, 0, NULL, NULL, NULL, _T(__FUNCTION__) _T(" pdwSizeOfDataFromInteractiveUI is NULL."), NULL);
|
|
else if (!ppDataFromInteractiveUI)
|
|
*ppEapError = g_peer.make_error(dwResult = ERROR_INVALID_PARAMETER, 0, NULL, NULL, NULL, _T(__FUNCTION__) _T(" ppDataFromInteractiveUI is NULL."), NULL);
|
|
else
|
|
dwResult = g_peer.invoke_interactive_ui(
|
|
pEapType,
|
|
hwndParent,
|
|
dwSizeofUIContextData,
|
|
pUIContextData,
|
|
pdwSizeOfDataFromInteractiveUI,
|
|
ppDataFromInteractiveUI,
|
|
ppEapError);
|
|
|
|
return dwResult;
|
|
}
|