able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/net/layer2svc/eapol/service/eldeviceio.c

2851 lines
82 KiB
C
Raw Normal View History

Add source files
2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 2000, Microsoft Corporation
Module Name:
eldeviceio.c
Abstract:
This module contains implementations for media-management and device I/O.
The routines declared here operate asynchronously on the handles
associated with an I/O completion port opened on the ndis uio driver.
Revision History:
sachins, Apr 23 2000, Created
--*/
#include "pcheapol.h"
#pragma hdrstop
// NDISUIO constants
CHAR NdisuioDevice[] = "\\\\.\\\\Ndisuio";
CHAR * pNdisuioDevice = &NdisuioDevice[0];
WCHAR cwszNDISUIOProtocolName[] = L"NDISUIO";
WORD g_wEtherType8021X= 0x8E88;
//
// ElMediaInit
//
// Description:
//
// Called on EAPOL service startup to initialize all the media related events
// and callback functions
//
//
// Arguments:
//
// Return Values:
//
DWORD
ElMediaInit (
)
{
DWORD dwIndex = 0;
DWORD dwRetCode = NO_ERROR;
TRACE0 (INIT, "ElMediaInit: Entered");
do
{
// Create Global Interface lock
if (dwRetCode = CREATE_READ_WRITE_LOCK(&(g_ITFLock), "ITF") != NO_ERROR)
{
TRACE1(EAPOL, "ElMediaInit: Error (%ld) in creating g_ITFLock read-write-lock", dwRetCode);
break;
}
// Initialize NLA locks
if (dwRetCode = CREATE_READ_WRITE_LOCK(&(g_NLALock), "NLA") != NO_ERROR)
{
TRACE1(EAPOL, "ElMediaInit: Error (%ld) in creating g_NLALock read-write-lock", dwRetCode);
break;
}
// Initialize EAPOL structures
if ((dwRetCode = ElInitializeEAPOL()) != NO_ERROR)
{
TRACE1(INIT, "ElMediaInit: ElInitializeEAPOL failed with dwRetCode = %d",
dwRetCode );
break;
}
else
{
// TRACE0(INIT, "ElMediaInit: ElInitializeEAPOL successful");
g_dwModulesStarted |= EAPOL_MODULE_STARTED;
}
// Initialize interface hash bucket table
g_ITFTable.pITFBuckets = (ITF_BUCKET *) MALLOC (INTF_TABLE_BUCKETS * sizeof (ITF_BUCKET));
if (g_ITFTable.pITFBuckets == NULL)
{
TRACE0 (DEVICE, "Error in allocation memory for ITF buckets");
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
break;
}
for (dwIndex=0; dwIndex < INTF_TABLE_BUCKETS; dwIndex++)
{
g_ITFTable.pITFBuckets[dwIndex].pItf=NULL;
}
// Indicate logon/logoff notifications can be accepted
g_dwModulesStarted |= LOGON_MODULE_STARTED;
// Check if service was delayed in starting, start user logon
ElCheckUserLoggedOn ();
// Check if the user-context process is ready to be notified
if ((dwRetCode = ElCheckUserModuleReady ()) == ERROR_BAD_IMPERSONATION_LEVEL)
{
break;
}
// Enumerate all the interfaces and start EAPOL state machine
// on interfaces which are of LAN type
if ((dwRetCode = ElEnumAndOpenInterfaces (NULL, NULL, 0, NULL)) != NO_ERROR)
{
TRACE1(INIT, "ElMediaInit: ElEnumAndOpenInterfaces failed with dwRetCode = %d",
dwRetCode );
break;
}
else
{
// TRACE0(INIT, "ElMediaInit: ElEnumAndOpenInterfaces successful");
}
#ifndef ZEROCONFIG_LINKED
// Register for Media Sense detection of MEDIA_CONNECT and
// MEDIA_DISCONNECT of interfaces
if ((dwRetCode = ElMediaSenseRegister (TRUE)) != NO_ERROR)
{
TRACE1(INIT, "ElMediaInit: ElMediaSenseRegister failed with dwRetCode = %d",
dwRetCode );
break;
}
else
{
g_dwModulesStarted |= WMI_MODULE_STARTED;
// TRACE0(INIT, "ElMediaInit: ElMediaSenseRegister successful");
}
// Register for detecting protocol BIND and UNBIND
if ((dwRetCode = ElBindingsNotificationRegister (TRUE)) != NO_ERROR)
{
TRACE1(INIT, "ElMediaInit: ElBindingsNotificationRegister failed with dwRetCode = %d",
dwRetCode );
break;
}
else
{
g_dwModulesStarted |= BINDINGS_MODULE_STARTED;
// TRACE0(INIT, "ElMediaInit: ElBindingsNotificationRegister successful");
}
// Register for device notifications. We are interested in LAN
// interfaces coming and going.
if ((dwRetCode = ElDeviceNotificationRegister (TRUE)) != NO_ERROR)
{
TRACE1(INIT, "ElMediaInit: ElDeviceNotificationRegister failed with dwRetCode = %d",
dwRetCode );
break;
}
else
{
g_dwModulesStarted |= DEVICE_NOTIF_STARTED;
// TRACE0(INIT, "ElMediaInit: ElDeviceNotificationRegister successful");
}
#endif // ZEROCONFIG_LINKED
} while (FALSE);
if (dwRetCode == NO_ERROR)
{
TRACE0(INIT, "ElMediaInit successful");
}
else
{
TRACE1(INIT, "ElMediaInit failed with error %ld",
dwRetCode);
}
return dwRetCode;
}
//
// ElMediaDeInit
//
// Description:
//
// Called on EAPOL service shutdown to de-initialize all the media
// related events and callback functions
//
//
// Arguments:
//
// Return Values:
//
DWORD
ElMediaDeInit (
)
{
LONG lLocalWorkerThreads = 0;
DWORD dwIndex = 0;
EAPOL_ITF *pITFWalker = NULL, *pITF = NULL;
DWORD dwRetCode = NO_ERROR;
TRACE0 (INIT, "ElMediaDeInit: Entered");
// Indicate logon/logoff notifications will not be accepted anymore
g_dwModulesStarted &= ~LOGON_MODULE_STARTED;
#ifndef ZEROCONFIG_LINKED
// DeRegister Media Sense detection of MEDIA_CONNECT and MEDIA_DISCONNECT
// of interfaces
if (g_dwModulesStarted & WMI_MODULE_STARTED)
{
if ((dwRetCode = ElMediaSenseRegister (FALSE)) != NO_ERROR )
{
TRACE1(INIT, "ElMediaDeInit: ElMediaSenseRegister failed with dwRetCode = %d",
dwRetCode );
// log
}
else
{
// TRACE0(INIT, "ElMediaDeInit: ElMediaSenseRegister successful");
}
g_dwModulesStarted &= ~WMI_MODULE_STARTED;
}
// Deregister detecting protocol BIND and UNBIND
if (g_dwModulesStarted & BINDINGS_MODULE_STARTED)
{
if ((dwRetCode = ElBindingsNotificationRegister (FALSE)) != NO_ERROR)
{
TRACE1(INIT, "ElMediaDeInit: ElBindingsNotificationRegister failed with dwRetCode = %d",
dwRetCode );
// log
}
else
{
g_dwModulesStarted &= ~BINDINGS_MODULE_STARTED;
// TRACE0(INIT, "ElMediaDeInit: ElBindingsNotificationRegister successful");
}
}
// Deregister device notifications that may have been posted
if (g_dwModulesStarted & DEVICE_NOTIF_STARTED)
{
if ((dwRetCode = ElDeviceNotificationRegister (FALSE)) != NO_ERROR)
{
TRACE1(INIT, "ElMediaDeInit: ElDeviceNotificationRegister failed with dwRetCode = %d",
dwRetCode );
// log
}
else
{
// TRACE0(INIT, "ElMediaDeInit: ElDeviceNotificationRegister successful");
}
g_dwModulesStarted &= ~DEVICE_NOTIF_STARTED;
}
#endif // ZEROCONFIG_LINKED
// Wait for all the related threads to die
// viz. MediaSense, BindingsNotification, DeviceNotification,
// Registry-watch for EAP-configuration change,
// Registry-watch for EAPOL-parameter change
do
{
lLocalWorkerThreads = 0;
lLocalWorkerThreads = InterlockedCompareExchange (
&g_lWorkerThreads,
0,
0);
if (lLocalWorkerThreads == 0)
{
TRACE0 (INIT, "ElMediaDeInit: No worker threads alive, exiting");
TRACE2 (INIT, "ElMediaDeInit: (%ld) - (%ld) worker threads still alive",
lLocalWorkerThreads, g_lWorkerThreads);
break;
}
TRACE2 (INIT, "ElMediaDeInit: (%ld) - (%ld) worker threads still alive, sleeping zzz... ",
lLocalWorkerThreads, g_lWorkerThreads);
Sleep (1000);
}
while (TRUE);
// Shutdown EAPOL state machine
if (g_dwModulesStarted & EAPOL_MODULE_STARTED)
{
if ((dwRetCode = ElEAPOLDeInit()) != NO_ERROR)
{
TRACE1(INIT, "ElMediaDeInit: ElEAPOLDeInit failed with dwRetCode = %d",
dwRetCode );
// log
}
else
{
TRACE0(INIT, "ElMediaDeInit: ElEAPOLDeInit successful");
}
g_dwModulesStarted &= ~EAPOL_MODULE_STARTED;
}
// Free the interface table
if (READ_WRITE_LOCK_CREATED(&(g_ITFLock)))
{
ACQUIRE_WRITE_LOCK (&(g_ITFLock));
if (g_ITFTable.pITFBuckets != NULL)
{
for (dwIndex = 0; dwIndex < INTF_TABLE_BUCKETS; dwIndex++)
{
for (pITFWalker = g_ITFTable.pITFBuckets[dwIndex].pItf;
pITFWalker != NULL;
/* NOTHING */
)
{
pITF = pITFWalker;
pITFWalker = pITFWalker->pNext;
if (pITF->pwszInterfaceDesc)
{
FREE (pITF->pwszInterfaceDesc);
}
if (pITF->pwszInterfaceGUID)
{
FREE (pITF->pwszInterfaceGUID);
}
if (pITF)
{
FREE (pITF);
}
}
}
FREE(g_ITFTable.pITFBuckets);
}
ZeroMemory (&g_ITFTable, sizeof (g_ITFTable));
RELEASE_WRITE_LOCK (&(g_ITFLock));
// Delete ITF table lock
DELETE_READ_WRITE_LOCK(&(g_ITFLock));
}
if (READ_WRITE_LOCK_CREATED(&(g_NLALock)))
{
// Delete NLA lock
DELETE_READ_WRITE_LOCK(&(g_NLALock));
}
TRACE0(INIT, "ElMediaDeInit completed");
return dwRetCode;
}
#ifdef ZEROCONFIG_LINKED
//
// ElMediaEventsHandler
//
// Description:
//
// Function called by WZC Service to signal various media events
//
// Arguments:
// pwzcDeviceNotif - Pointer to WZC_DEVICE_NOTIF structure
//
// Return values:
// NO_ERROR - Successful
// non-zero - Error
//
DWORD
ElMediaEventsHandler (
IN PWZC_DEVICE_NOTIF pwzcDeviceNotif
)
{
DWORD dwDummyValue = NO_ERROR;
DWORD dwRetCode = NO_ERROR;
do
{
TRACE0 (DEVICE, "ElMediaEventsHandler entered");
if (pwzcDeviceNotif == NULL)
{
break;
}
switch (pwzcDeviceNotif->dwEventType)
{
case WZCNOTIF_DEVICE_ARRIVAL:
TRACE0 (DEVICE, "ElMediaEventsHandler: Calling ElDeviceNotificationHandler ");
ElDeviceNotificationHandler (
(VOID *)&(pwzcDeviceNotif->dbDeviceIntf),
DBT_DEVICEARRIVAL
);
break;
case WZCNOTIF_DEVICE_REMOVAL:
TRACE0 (DEVICE, "ElMediaEventsHandler: Calling ElDeviceNotificationHandler ");
ElDeviceNotificationHandler (
(VOID *)&(pwzcDeviceNotif->dbDeviceIntf),
DBT_DEVICEREMOVECOMPLETE
);
break;
case WZCNOTIF_ADAPTER_BIND:
TRACE0 (DEVICE, "ElMediaEventsHandler: Calling ElBindingsNotificationCallback ");
ElBindingsNotificationCallback (
&(pwzcDeviceNotif->wmiNodeHdr),
0
);
break;
case WZCNOTIF_ADAPTER_UNBIND:
TRACE0 (DEVICE, "ElMediaEventsHandler: Calling ElBindingsNotificationCallback ");
ElBindingsNotificationCallback (
&(pwzcDeviceNotif->wmiNodeHdr),
0
);
break;
case WZCNOTIF_MEDIA_CONNECT:
TRACE0 (DEVICE, "ElMediaEventsHandler: Calling ElMediaSenseCallback ");
ElMediaSenseCallback (
&(pwzcDeviceNotif->wmiNodeHdr),
0
);
break;
case WZCNOTIF_MEDIA_DISCONNECT:
TRACE0 (DEVICE, "ElMediaEventsHandler: Calling ElMediaSenseCallback ");
ElMediaSenseCallback (
&(pwzcDeviceNotif->wmiNodeHdr),
0
);
break;
case WZCNOTIF_WZC_CONNECT:
TRACE0 (DEVICE, "ElMediaEventsHandler: Calling ElZeroConfigEvent ");
ElZeroConfigEvent (
pwzcDeviceNotif->wzcConfig.dwSessionHdl,
pwzcDeviceNotif->wzcConfig.wszGuid,
pwzcDeviceNotif->wzcConfig.ndSSID,
&(pwzcDeviceNotif->wzcConfig.rdEventData)
);
break;
default:
break;
}
}
while (FALSE);
return dwRetCode;
}
#endif // ZEROCONFIG_LINKED
//
// ElMediaSenseRegister
//
// Description:
//
// Function called to register CallBack function with WMI
// for MEDIA_CONNECT/MEDIA_DISCONNECT events
//
// Arguments:
// fRegister - True = Register for Media Sense
// False = Deregister Media Sense requests
// Return values:
// NO_ERROR - Successful
// non-zero - Error
//
DWORD
ElMediaSenseRegister (
IN BOOL fRegister
)
{
DWORD dwRetCode = NO_ERROR;
PVOID pvDeliveryInfo = ElMediaSenseCallback;
dwRetCode = WmiNotificationRegistration (
(LPGUID)(&GUID_NDIS_STATUS_MEDIA_CONNECT),
(BOOLEAN)fRegister,
pvDeliveryInfo,
(ULONG_PTR)NULL,
NOTIFICATION_CALLBACK_DIRECT );
if (dwRetCode != NO_ERROR)
{
TRACE1(INIT, "ElMediaSenseRegister: Error %d in WmiNotificationRegistration:GUID_NDIS_STATUS_MEDIA_CONNECT", dwRetCode);
return( dwRetCode );
}
dwRetCode = WmiNotificationRegistration (
(LPGUID)(&GUID_NDIS_STATUS_MEDIA_DISCONNECT),
(BOOLEAN)fRegister,
pvDeliveryInfo,
(ULONG_PTR)NULL,
NOTIFICATION_CALLBACK_DIRECT );
if (dwRetCode != NO_ERROR)
{
TRACE1(INIT, "ElMediaSenseRegister: Error %d in WmiNotificationRegistration:GUID_NDIS_STATUS_MEDIA_DISCONNECT", dwRetCode);
return( dwRetCode );
}
TRACE1 (INIT, "ElMediaSenseRegister - completed with RetCode %d", dwRetCode);
return( dwRetCode );
}
//
// ElDeviceNotificationRegister
//
// Description:
//
// Function called to register for device addition/removal notifications
//
// Arguments:
// fRegister - True = Register for Device Notifications
// False = Deregister Device Notifications
//
// Return values:
// NO_ERROR - Successful
// non-zero - Error
//
DWORD
ElDeviceNotificationRegister (
IN BOOL fRegister
)
{
HANDLE hDeviceNotification = NULL;
DWORD dwRetCode = NO_ERROR;
#ifdef EAPOL_SERVICE
DEV_BROADCAST_DEVICEINTERFACE PnPFilter;
if (fRegister)
{
ZeroMemory (&PnPFilter, sizeof(PnPFilter));
PnPFilter.dbcc_size = sizeof(PnPFilter);
PnPFilter.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
PnPFilter.dbcc_classguid = GUID_NDIS_LAN_CLASS;
g_hDeviceNotification = RegisterDeviceNotification (
(HANDLE)g_hServiceStatus,
&PnPFilter,
DEVICE_NOTIFY_SERVICE_HANDLE );
if (g_hDeviceNotification == NULL)
{
dwRetCode = GetLastError();
TRACE1 (DEVICE, "ElDeviceNotificationRegister failed with error %ld",
dwRetCode);
}
}
else
{
if (g_hDeviceNotification != NULL)
{
if (!UnregisterDeviceNotification (
g_hDeviceNotification
))
{
dwRetCode = GetLastError();
TRACE1 (DEVICE, "ElDeviceNotificationRegister: Unregister failed with error (%ld)",
dwRetCode);
}
}
}
#endif
return dwRetCode;
}
//
// ElBindingsNotificationRegister
//
// Description:
//
// Function called to register CallBack function with WMI
// for protocol bind/unbind
//
// Arguments:
// fRegister - True = Register for Media Sense
// False = Deregister Media Sense requests
// Return values:
// NO_ERROR - Successful
// non-zero - Error
//
DWORD
ElBindingsNotificationRegister (
IN BOOL fRegister
)
{
DWORD dwRetCode = NO_ERROR;
PVOID pvDeliveryInfo = ElBindingsNotificationCallback;
dwRetCode = WmiNotificationRegistration (
(LPGUID)(&GUID_NDIS_NOTIFY_BIND),
(BOOLEAN)fRegister,
pvDeliveryInfo,
(ULONG_PTR)NULL,
NOTIFICATION_CALLBACK_DIRECT );
if (dwRetCode != NO_ERROR)
{
TRACE1(INIT, "ElBindingsNotificationRegister: Error %d in WmiNotificationRegistration:GUID_NDIS_NOTIFY_BIND", dwRetCode);
return( dwRetCode );
}
dwRetCode = WmiNotificationRegistration (
(LPGUID)(&GUID_NDIS_NOTIFY_UNBIND),
(BOOLEAN)fRegister,
pvDeliveryInfo,
(ULONG_PTR)NULL,
NOTIFICATION_CALLBACK_DIRECT );
if (dwRetCode != NO_ERROR)
{
TRACE1(INIT, "ElBindingsNotificationRegister: Error %d in WmiNotificationRegistration:GUID_NDIS_NOTIFY_BIND", dwRetCode);
return( dwRetCode );
}
TRACE1 (INIT, "ElBindingsNotificationRegister - completed with RetCode %d", dwRetCode);
return( dwRetCode );
}
//
// ElDeviceNotificationHandler
//
// Description:
//
// Function called to handle device notifications for interface addition/
// removal
//
// Arguments:
// lpEventData - interface information
// dwEventType - notification type
//
DWORD
ElDeviceNotificationHandler (
IN VOID *lpEventData,
IN DWORD dwEventType
)
{
DWORD dwEventStatus = 0;
DEV_BROADCAST_DEVICEINTERFACE *pInfo =
(DEV_BROADCAST_DEVICEINTERFACE *) lpEventData;
PVOID pvBuffer = NULL;
BOOLEAN fDecrWorkerThreadCount = TRUE;
DWORD dwRetCode = NO_ERROR;
InterlockedIncrement (&g_lWorkerThreads);
TRACE0 (DEVICE, "ElDeviceNotificationHandler entered");
do
{
if (g_hEventTerminateEAPOL == NULL)
{
break;
}
if (!(g_dwModulesStarted & ALL_MODULES_STARTED))
{
TRACE0 (DEVICE, "ElDeviceNotificationHandler: Received notification before module started");
break;
}
// Check if have already gone through EAPOLCleanUp before
if ((dwEventStatus = WaitForSingleObject (
g_hEventTerminateEAPOL,
0)) == WAIT_FAILED)
{
dwRetCode = GetLastError ();
TRACE1(INIT, "ElDeviceNotificationHandler: WaitForSingleObject failed with error %ld, Terminating !!!",
dwRetCode);
// log
break;
}
if (dwEventStatus == WAIT_OBJECT_0)
{
TRACE0(INIT, "ElDeviceNotificationHandler: g_hEventTerminateEAPOL already signaled, returning");
break;
}
if (lpEventData == NULL)
{
dwRetCode = ERROR_INVALID_DATA;
TRACE0 (DEVICE, "ElDeviceNotificationHandler: lpEventData == NULL");
break;
}
if (pInfo->dbcc_devicetype == DBT_DEVTYP_DEVICEINTERFACE)
{
TRACE0 (DEVICE, "ElDeviceNotificationHandler: Event for Interface type");
if ((pvBuffer = MALLOC (pInfo->dbcc_size + 16)) == NULL)
{
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
TRACE0 (DEVICE, "ElDeviceNotificationHandler: MALLOC failed for pvBuffer");
break;
}
*((DWORD *)pvBuffer) = dwEventType;
memcpy ((PBYTE)pvBuffer + 8, (PBYTE)pInfo, pInfo->dbcc_size);
if (!QueueUserWorkItem (
(LPTHREAD_START_ROUTINE)ElDeviceNotificationHandlerWorker,
pvBuffer,
WT_EXECUTELONGFUNCTION))
{
dwRetCode = GetLastError();
TRACE1 (DEVICE, "ElDeviceNotificationHandler: QueueUserWorkItem failed with error %ld",
dwRetCode);
break;
}
else
{
fDecrWorkerThreadCount = FALSE;
}
}
else
{
TRACE0 (DEVICE, "ElDeviceNotificationHandler: Event NOT for Interface type");
}
}
while (FALSE);
TRACE1 (DEVICE, "ElDeviceNotificationHandler completed with retcode %ld",
dwRetCode);
if (dwRetCode != NO_ERROR)
{
if (pvBuffer != NULL)
{
FREE (pvBuffer);
}
}
if (fDecrWorkerThreadCount)
{
InterlockedDecrement (&g_lWorkerThreads);
}
return dwRetCode;
}
//
// ElDeviceNotificationHandlerWorker
//
// Description:
//
// Worker function for ElDeviceNotificationHandlerWorker
//
// Arguments:
// pvContext - interface information
//
DWORD
WINAPI
ElDeviceNotificationHandlerWorker (
IN PVOID pvContext
)
{
HANDLE hDevice = NULL;
DEV_BROADCAST_DEVICEINTERFACE *pInfo = NULL;
DWORD dwEventType = 0;
DWORD dwRetCode = NO_ERROR;
TRACE0 (DEVICE, "ElDeviceNotificationHandlerWorker: Entered");
do
{
if (pvContext == NULL)
{
TRACE0 (DEVICE, "ElDeviceNotificationHandlerWorker: pvContext == NULL");
break;
}
if (!(g_dwModulesStarted & ALL_MODULES_STARTED))
{
TRACE0 (DEVICE, "ElDeviceNotificationHandlerWorker: Received notification before module started");
break;
}
dwEventType = *((DWORD *) pvContext);
pInfo = (DEV_BROADCAST_DEVICEINTERFACE*)((PBYTE)pvContext + 8);
if ((dwEventType == DBT_DEVICEARRIVAL) ||
(dwEventType == DBT_DEVICEREMOVECOMPLETE))
{
// Extract GUID from the \Device\GUID string
WCHAR *pwszGUIDStart = NULL;
WCHAR *pwszGUIDEnd = NULL;
TRACE0 (DEVICE, "ElDeviceNotificationHandlerWorker: Interface arr/rem");
pwszGUIDStart = wcsrchr( pInfo->dbcc_name, L'{' );
pwszGUIDEnd = wcsrchr( pInfo->dbcc_name, L'}' );
if ((pwszGUIDStart != NULL) && (pwszGUIDEnd != NULL) &&
((pwszGUIDEnd- pwszGUIDStart) == (GUID_STRING_LEN_WITH_TERM-2)))
{
*(pwszGUIDEnd + 1) = L'\0';
TRACE1 (DEVICE, "ElDeviceNotificationHandlerWorker: For interface %ws",
pwszGUIDStart);
// Interface was added
if (dwEventType == DBT_DEVICEARRIVAL)
{
TRACE0(DEVICE, "ElDeviceNotificationHandlerWorker: Callback for device addition");
if ((dwRetCode = ElEnumAndOpenInterfaces (
NULL, pwszGUIDStart, 0, NULL)) != NO_ERROR)
{
TRACE1 (DEVICE, "ElDeviceNotificationHandlerWorker: ElEnumAndOpenInterfaces returned error %ld",
dwRetCode);
}
}
else
{
TRACE0(DEVICE, "ElDeviceNotificationHandlerWorker: Callback for device removal");
if ((dwRetCode = ElShutdownInterface (pwszGUIDStart))
!= NO_ERROR)
{
TRACE1 (DEVICE, "ElDeviceNotificationHandlerWorker: ElShutdownInterface failed with error %ld",
dwRetCode);
}
if ((dwRetCode = ElEnumAndUpdateRegistryInterfaceList ()) != NO_ERROR)
{
TRACE1 (DEVICE, "ElDeviceNotificationHandlerWorker: ElEnumAndUpdateRegistryInterfaceList failed with error %ld",
dwRetCode);
}
}
}
else
{
dwRetCode = ERROR_INVALID_PARAMETER;
break;
}
}
else
{
TRACE0 (DEVICE, "ElDeviceNotificationHandlerWorker: Event type is is NOT device arr/rem");
}
}
while (FALSE);
if (pvContext != NULL)
{
FREE (pvContext);
}
TRACE1 (DEVICE, "ElDeviceNotificationHandlerWorker completed with retcode %ld",
dwRetCode);
InterlockedDecrement (&g_lWorkerThreads);
return 0;
}
//
// ElMediaSenseCallback
//
// Description:
//
// Callback function called by WMI on MEDIA_CONNECT/MEDIA_DISCONNECT
// events
//
// Arguments:
// pWnodeHeader - Pointer to information returned by the event
// uiNotificationContext - unused
//
// Return values:
// NO_ERROR - Success
// non-zero - Error
//
VOID
CALLBACK
ElMediaSenseCallback (
IN PWNODE_HEADER pWnodeHeader,
IN UINT_PTR uiNotificationContext
)
{
DWORD dwEventStatus = 0;
PVOID pvBuffer = NULL;
BOOLEAN fDecrWorkerThreadCount = TRUE;
DWORD dwRetCode = NO_ERROR;
InterlockedIncrement (&g_lWorkerThreads);
TRACE0 (DEVICE, "ElMediaSenseCallback: Entered");
do
{
if (g_hEventTerminateEAPOL == NULL)
{
break;
}
if (!(g_dwModulesStarted & ALL_MODULES_STARTED))
{
TRACE0 (DEVICE, "ElMediaSenseCallback: Received notification before module started");
break;
}
// Check if have already gone through EAPOLCleanUp before
if (( dwEventStatus = WaitForSingleObject (
g_hEventTerminateEAPOL,
0)) == WAIT_FAILED)
{
dwRetCode = GetLastError ();
TRACE1 (INIT, "ElMediaSenseCallback: WaitForSingleObject failed with error %ld, Terminating !!!",
dwRetCode);
// log
break;
}
if (dwEventStatus == WAIT_OBJECT_0)
{
dwRetCode = NO_ERROR;
TRACE0 (INIT, "ElMediaSenseCallback: g_hEventTerminateEAPOL already signaled, returning");
break;
}
if (pWnodeHeader == NULL)
{
dwRetCode = ERROR_INVALID_DATA;
TRACE0 (DEVICE, "ElMediaSenseCallback: pWnodeHeader == NULL");
break;
}
if ((pvBuffer = MALLOC (pWnodeHeader->BufferSize)) == NULL)
{
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
TRACE0 (DEVICE, "ElMediaSenseCallback: MALLOC failed for pvBuffer");
break;
}
memcpy ((PVOID)pvBuffer, (PVOID)pWnodeHeader, pWnodeHeader->BufferSize);
if (!QueueUserWorkItem (
(LPTHREAD_START_ROUTINE)ElMediaSenseCallbackWorker,
pvBuffer,
WT_EXECUTELONGFUNCTION))
{
dwRetCode = GetLastError();
TRACE1 (DEVICE, "ElMediaSenseCallback: QueueUserWorkItem failed with error %ld",
dwRetCode);
// log
break;
}
else
{
fDecrWorkerThreadCount = FALSE;
}
}
while (FALSE);
if (dwRetCode != NO_ERROR)
{
TRACE1 (DEVICE, "ElMediaSenseCallback: Failed with error %ld",
dwRetCode);
if (pvBuffer != NULL)
{
FREE (pvBuffer);
}
}
if (fDecrWorkerThreadCount)
{
InterlockedDecrement (&g_lWorkerThreads);
}
}
//
// ElMediaSenseCallbackWorker
//
// Description:
//
// Worker function for ElMediaSenseCallback and executes in a separate
// thread
//
// Arguments:
// pvContext - Pointer to information returned by the media-sense event
//
// Return values:
// NO_ERROR - Success
// non-zero - Error
//
DWORD
WINAPI
ElMediaSenseCallbackWorker (
IN PVOID pvContext
)
{
PWNODE_HEADER pWnodeHeader = (PWNODE_HEADER)pvContext;
PWNODE_SINGLE_INSTANCE pWnode = (PWNODE_SINGLE_INSTANCE)pWnodeHeader;
WCHAR *pwsName = NULL;
WCHAR *pwszDeviceName = NULL;
WCHAR *pwsGUIDString = NULL;
WCHAR *pwszDeviceGUID = NULL;
WCHAR *pwszGUIDStart = NULL, *pwszGUIDEnd = NULL;
DWORD dwGUIDLen = 0;
USHORT cpsLength;
EAPOL_ITF *pITF;
EAPOL_PCB *pPCB = NULL;
DWORD dwRetCode = NO_ERROR;
do
{
#ifdef EAPOL_SERVICE
if ((g_ServiceStatus.dwCurrentState == SERVICE_STOP_PENDING)
||
(g_ServiceStatus.dwCurrentState == SERVICE_STOPPED))
{
TRACE0 (DEVICE, "ElMediaSenseCallbackWorker: Callback received while service was stopping");
break;
}
#endif // EAPOL_SERVICE
if (pWnodeHeader == NULL)
{
TRACE0 (DEVICE, "ElMediaSenseCallbackWorker: Callback received with NULL NDIS interface details");
break;
}
pwsName = (PWCHAR)RtlOffsetToPointer(
pWnode,
pWnode->OffsetInstanceName );
pwsGUIDString = (PWCHAR)RtlOffsetToPointer(
pWnode,
pWnode->DataBlockOffset );
cpsLength = (SHORT)( *((SHORT *)pwsName) );
if (!(pwszDeviceName = (WCHAR *) MALLOC ((cpsLength+1)*sizeof(WCHAR))))
{
TRACE0 (DEVICE, "ElMediaSenseCallbackWorker: Error in Memory allocation for pszDeviceName");
break;
}
memcpy ((CHAR *)pwszDeviceName, (CHAR *)pwsName+sizeof(SHORT), cpsLength);
pwszDeviceName[cpsLength] = L'\0';
pwszGUIDStart = wcschr (pwsGUIDString, L'{');
pwszGUIDEnd = wcschr (pwsGUIDString, L'}');
if ((pwszGUIDStart == NULL) || (pwszGUIDEnd == NULL) || ((pwszGUIDEnd - pwszGUIDStart) != (GUID_STRING_LEN_WITH_TERM-2)))
{
TRACE0 (DEVICE, "ElMediaSenseCallbackWorker: GUID not constructed correctly");
dwRetCode = ERROR_INVALID_PARAMETER;
break;
}
dwGUIDLen = GUID_STRING_LEN_WITH_TERM;
pwszDeviceGUID = NULL;
if ((pwszDeviceGUID = MALLOC (dwGUIDLen * sizeof (WCHAR))) == NULL)
{
TRACE0 (DEVICE, "ElMediaSenseCallbackWorker: MALLOC failed for pwszDeviceGUID");
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
break;
}
memcpy ((VOID *)pwszDeviceGUID, (VOID *)pwszGUIDStart, ((dwGUIDLen-1)*sizeof(WCHAR)));
pwszDeviceGUID[dwGUIDLen-1] = L'\0';
TRACE3 (DEVICE, "ElMediaSenseCallbackWorker: For interface (%ws), GUID (%ws), length of block = %d",
pwszDeviceName, pwszDeviceGUID, cpsLength);
//
// Get the information for the media disconnect.
//
if (memcmp( &(pWnodeHeader->Guid),
&GUID_NDIS_STATUS_MEDIA_DISCONNECT,
sizeof(GUID)) == 0)
{
// MEDIA DISCONNECT callback
DbLogPCBEvent (DBLOG_CATEG_INFO, NULL, EAPOL_MEDIA_DISCONNECT, pwszDeviceName);
// Check if EAPOL was actually started on this interface
// Verify by checking existence of corresponding entry in hash table
TRACE0(DEVICE, "ElMediaSenseCallbackWorker: Callback for sense disconnect");
ACQUIRE_WRITE_LOCK (&(g_PCBLock));
pPCB = ElGetPCBPointerFromPortGUID (pwszDeviceGUID);
if (pPCB != NULL)
{
ACQUIRE_WRITE_LOCK (&(pPCB->rwLock));
if ((dwRetCode = FSMDisconnected (pPCB, NULL)) != NO_ERROR)
{
TRACE1 (DEVICE, "ElMediaSenseCallbackWorker: FSMDisconnected failed with error %ld",
dwRetCode);
}
else
{
TRACE1 (DEVICE, "ElMediaSenseCallbackWorker: Port marked disconnected %ws",
pwszDeviceName);
}
RELEASE_WRITE_LOCK (&(pPCB->rwLock));
}
RELEASE_WRITE_LOCK (&(g_PCBLock));
}
else
{
if (memcmp( &(pWnodeHeader->Guid),
&GUID_NDIS_STATUS_MEDIA_CONNECT,
sizeof(GUID)) == 0)
{
// MEDIA CONNECT callback
DbLogPCBEvent (DBLOG_CATEG_INFO, NULL, EAPOL_MEDIA_CONNECT, pwszDeviceName);
TRACE0(DEVICE, "ElMediaSenseCallbackWorker: Callback for sense connect");
if ((dwRetCode = ElEnumAndOpenInterfaces (
NULL, pwszDeviceGUID, 0, NULL))
!= NO_ERROR)
{
TRACE1 (DEVICE, "ElMediaSenseCallbackWorker: ElEnumAndOpenInterfaces returned error %ld",
dwRetCode);
}
}
}
}
while (FALSE);
TRACE1 (DEVICE, "ElMediaSenseCallbackWorker: processed, RetCode = %ld", dwRetCode);
if (pWnodeHeader != NULL)
{
FREE (pWnodeHeader);
}
if (pwszDeviceName != NULL)
{
FREE (pwszDeviceName);
}
if (pwszDeviceGUID != NULL)
{
FREE (pwszDeviceGUID);
}
InterlockedDecrement (&g_lWorkerThreads);
return 0;
}
//
// ElBindingsNotificationCallback
//
// Description:
//
// Callback function called by WMI on protocol bind/unbind
// events
//
// Arguments:
// pWnodeHeader - Pointer to information returned by the event
// uiNotificationContext - unused
//
// Return values:
// NO_ERROR - Success
// non-zero - Error
//
VOID
CALLBACK
ElBindingsNotificationCallback (
IN PWNODE_HEADER pWnodeHeader,
IN UINT_PTR uiNotificationContext
)
{
DWORD dwEventStatus = 0;
PVOID pvBuffer = NULL;
BOOLEAN fDecrWorkerThreadCount = TRUE;
DWORD dwRetCode = NO_ERROR;
InterlockedIncrement (&g_lWorkerThreads);
TRACE0 (DEVICE, "ElBindingsNotificationCallback: Entered");
do
{
if (g_hEventTerminateEAPOL == NULL)
{
break;
}
if (!(g_dwModulesStarted & ALL_MODULES_STARTED))
{
TRACE0 (DEVICE, "ElBindingsNotificationCallback: Received notification before module started");
break;
}
// Check if have already gone through EAPOLCleanUp before
if (( dwEventStatus = WaitForSingleObject (
g_hEventTerminateEAPOL,
0)) == WAIT_FAILED)
{
dwRetCode = GetLastError ();
TRACE1 (INIT, "ElBindingsNotificationCallback: WaitForSingleObject failed with error %ld, Terminating !!!",
dwRetCode);
break;
}
if (dwEventStatus == WAIT_OBJECT_0)
{
dwRetCode = NO_ERROR;
TRACE0 (INIT, "ElBindingsNotificationCallback: g_hEventTerminateEAPOL already signaled, returning");
break;
}
if (pWnodeHeader == NULL)
{
dwRetCode = ERROR_INVALID_DATA;
TRACE0 (DEVICE, "ElBindingsNotificationCallback: pWnodeHeader == NULL");
break;
}
if ((pvBuffer = MALLOC (pWnodeHeader->BufferSize)) == NULL)
{
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
TRACE0 (DEVICE, "ElBindingsNotificationCallback: MALLOC failed for pvBuffer");
break;
}
memcpy ((PVOID)pvBuffer, (PVOID)pWnodeHeader, pWnodeHeader->BufferSize);
if (!QueueUserWorkItem (
(LPTHREAD_START_ROUTINE)ElBindingsNotificationCallbackWorker,
pvBuffer,
WT_EXECUTELONGFUNCTION))
{
dwRetCode = GetLastError();
TRACE1 (DEVICE, "ElBindingsNotificationCallback: QueueUserWorkItem failed with error %ld",
dwRetCode);
// log
break;
}
else
{
fDecrWorkerThreadCount = FALSE;
}
}
while (FALSE);
if (dwRetCode != NO_ERROR)
{
TRACE1 (DEVICE, "ElBindingsNotificationCallback: Failed with error %ld",
dwRetCode);
if (pvBuffer != NULL)
{
FREE (pvBuffer);
}
}
if (fDecrWorkerThreadCount)
{
InterlockedDecrement (&g_lWorkerThreads);
}
}
//
// ElBindingsNotificationCallbackWorker
//
// Description:
//
// Worker function for ElBindingsNotificationCallback and executes in a separate
// thread
//
// Arguments:
// pvContext - Pointer to information returned by the protocol bind/unbind
// event
//
// Return values:
// NO_ERROR - Success
// non-zero - Error
//
DWORD
WINAPI
ElBindingsNotificationCallbackWorker (
IN PVOID pvContext
)
{
PWNODE_HEADER pWnodeHeader = (PWNODE_HEADER)pvContext;
PWNODE_SINGLE_INSTANCE pWnode = (PWNODE_SINGLE_INSTANCE)pWnodeHeader;
WCHAR *pwsName = NULL;
WCHAR *pwszDeviceGUID = NULL;
WCHAR *pwszGUIDStart = NULL, *pwszGUIDEnd = NULL;
DWORD dwGUIDLen = 0;
WCHAR *pwsTransportName = NULL;
WCHAR *pwszDeviceName = NULL;
USHORT cpsLength;
EAPOL_ITF *pITF = NULL;
EAPOL_PCB *pPCB = NULL;
DWORD dwRetCode = NO_ERROR;
do
{
#ifdef EAPOL_SERVICE
if ((g_ServiceStatus.dwCurrentState == SERVICE_STOP_PENDING)
||
(g_ServiceStatus.dwCurrentState == SERVICE_STOPPED))
{
TRACE0 (DEVICE, "ElBindingsNotificationCallbackWorker: Callback received while service was stopping");
break;
}
#endif // EAPOL_SERVICE
if (pWnodeHeader == NULL)
{
TRACE0 (DEVICE, "ElBindingsNotificationCallbackWorker: Callback received with NULL NDIS interface details");
break;
}
pwsName = (PWCHAR)RtlOffsetToPointer(
pWnode,
pWnode->OffsetInstanceName );
pwsTransportName = (PWCHAR)RtlOffsetToPointer(
pWnode,
pWnode->DataBlockOffset );
if (wcsncmp (cwszNDISUIOProtocolName, pwsTransportName, wcslen (cwszNDISUIOProtocolName)))
{
TRACE1 (DEVICE, "ElBindingsNotificationCallbackWorker: Protocol binding (%ws) not for NDISUIO",
pwsTransportName);
break;
}
// Get the length of the device name string and null terminate it
cpsLength = (SHORT)( *((SHORT *)pwsName) );
if (!(pwszDeviceName = (WCHAR *) MALLOC ((cpsLength+1)*sizeof(WCHAR))))
{
TRACE0 (DEVICE, "ElBindingsNotificationCallbackWorker: Error in Memory allocation for pwszDeviceName");
break;
}
memcpy ((CHAR *)pwszDeviceName, (CHAR *)pwsName+sizeof(SHORT), cpsLength);
pwszDeviceName[cpsLength] = L'\0';
pwszDeviceGUID = pwsTransportName + wcslen(cwszNDISUIOProtocolName) + 1;
pwszGUIDStart = wcschr (pwszDeviceGUID, L'{');
pwszGUIDEnd = wcschr (pwszDeviceGUID, L'}');
pwszDeviceGUID = NULL;
if ((pwszGUIDStart == NULL) || (pwszGUIDEnd == NULL) || ((pwszGUIDEnd - pwszGUIDStart) != (GUID_STRING_LEN_WITH_TERM-2)))
{
TRACE0 (DEVICE, "ElBindingsNotificationCallbackWorker: GUID not constructed correctly");
dwRetCode = ERROR_INVALID_PARAMETER;
break;
}
dwGUIDLen = GUID_STRING_LEN_WITH_TERM;
if ((pwszDeviceGUID = MALLOC (dwGUIDLen * sizeof (WCHAR))) == NULL)
{
TRACE0 (DEVICE, "ElBindingsNotificationCallbackWorker: MALLOC failed for pwszDeviceGUID");
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
break;
}
memcpy ((VOID *)pwszDeviceGUID, (VOID *)pwszGUIDStart, ((dwGUIDLen-1)*sizeof(WCHAR)));
pwszDeviceGUID[dwGUIDLen-1] = L'\0';
TRACE2 (DEVICE, "ElBindingsNotificationCallbackWorker: For interface = %ws, guid=%ws",
pwszDeviceName, pwszDeviceGUID);
//
// Get the information for the protocol UNBIND
//
if (memcmp( &(pWnodeHeader->Guid),
&GUID_NDIS_NOTIFY_UNBIND,
sizeof(GUID)) == 0)
{
// Protocol UNBIND callback
DbLogPCBEvent (DBLOG_CATEG_INFO, NULL, EAPOL_NDISUIO_UNBIND, pwszDeviceName);
TRACE0(DEVICE, "ElBindingsNotificationCallbackWorker: Callback for protocol unbind");
if ((dwRetCode = ElShutdownInterface (pwszDeviceGUID)) != ERROR)
{
TRACE2 (DEVICE, "ElBindingsNotificationCallbackWorker: ElShutdownInterface failed with error %ld for (%ws)",
dwRetCode, pwszDeviceGUID);
}
if ((dwRetCode = ElEnumAndUpdateRegistryInterfaceList ()) != NO_ERROR)
{
TRACE1 (DEVICE, "ElBindingsNotificationCallbackWorker: ElEnumAndUpdateRegistryInterfaceList failed with error %ld",
dwRetCode);
}
}
else
{
if (memcmp( &(pWnodeHeader->Guid),
&GUID_NDIS_NOTIFY_BIND,
sizeof(GUID)) == 0)
{
// protocol BIND callback
DbLogPCBEvent (DBLOG_CATEG_INFO, NULL, EAPOL_NDISUIO_BIND, pwszDeviceName);
TRACE0(DEVICE, "ElBindingsNotificationCallbackWorker: Callback for protocol BIND");
if ((dwRetCode = ElEnumAndOpenInterfaces (
NULL, pwszDeviceGUID, 0, NULL))
!= NO_ERROR)
{
TRACE1 (DEVICE, "ElBindingsNotificationCallbackWorker: ElEnumAndOpenInterfaces returned error %ld",
dwRetCode);
}
}
}
}
while (FALSE);
TRACE1 (DEVICE, "ElBindingsNotificationCallbackWorker: processed, RetCode = %ld", dwRetCode);
if (pWnodeHeader != NULL)
{
FREE (pWnodeHeader);
}
if (pwszDeviceName != NULL)
{
FREE (pwszDeviceName);
}
if (pwszDeviceGUID != NULL)
{
FREE (pwszDeviceGUID);
}
InterlockedDecrement (&g_lWorkerThreads);
return 0;
}
//
// ElEnumAndOpenInterfaces
//
// Description:
//
// Enumerates interfaces and intializes EAPOL on desired ones.
//
// If EAPOL is to be started on an interface, it opens a handle to
// the NDISUIO driver, calls EAPOL to create and initialize PCB for the
// interface, and finally adds an entry to the interface hashtable.
//
// If pwszDesiredGUID is not NULL, all interfaces are enumerated, but
// EAPOL will be initialized only on the interface whose GUID matches.
//
// If pwszDesiredDescription is not NULL, all interfaces are enumerated, but
// EAPOL will be initialized only on the interface whose description matches.
//
// If pwszDesiredGUID and pwszDescription are both NULL, all interfaces are
// enumerated. EAPOL will be initialized only on all interfaces that
// does have an entry in the interface hashtable.
//
//
// Arguments:
// pwszDesiredDescription - Interface Description on which EAPOL is to
// be started
// pwszDesiredGUID - Interface GUID on which EAPOL is to be started
//
// Return values:
// NO_ERROR - Success
// non-zero - Error
//
DWORD
ElEnumAndOpenInterfaces (
WCHAR *pwszDesiredDescription,
WCHAR *pwszDesiredGUID,
DWORD dwHandle,
PRAW_DATA prdUserData
)
{
CHAR EnumerateBuffer[256];
PNDIS_ENUM_INTF Interfaces = NULL;
BYTE *pbNdisuioEnumBuffer = NULL;
DWORD dwNdisuioEnumBufferSize = 0;
HANDLE hDevice = NULL;
BOOL fSearchByDescription = FALSE;
BOOL fSearchByGUID = FALSE;
DWORD dwEapTypeToBeUsed = DEFAULT_EAP_TYPE;
WCHAR cwsDummyBuffer[256], *pDummyPtr = NULL;
WCHAR *pwszGUIDStart = NULL;
EAPOL_PCB *pPCB = NULL;
BOOL fPCBExists = FALSE;
BOOL fPCBReferenced = FALSE;
DWORD dwAvailableInterfaces = 0;
EAPOL_INTF_PARAMS EapolIntfParams;
DWORD dwRetCode = NO_ERROR;
TRACE2 (DEVICE, "ElEnumAndOpenInterfaces: DeviceDesc = %ws, GUID = %ws",
pwszDesiredDescription, pwszDesiredGUID);
ACQUIRE_WRITE_LOCK (&(g_ITFLock));
if (pwszDesiredGUID == NULL)
{
if (pwszDesiredDescription != NULL)
{
fSearchByDescription = TRUE;
}
}
else
{
if (pwszDesiredDescription != NULL)
{
RELEASE_WRITE_LOCK (&(g_ITFLock));
return ERROR;
}
fSearchByGUID = TRUE;
}
ZeroMemory (EnumerateBuffer, 256);
Interfaces = (PNDIS_ENUM_INTF)EnumerateBuffer;
//
// Allocate amount of memory as instructed by NdisEnumerateInterfaces
// once the API allows querying of bytes required
//
Interfaces->TotalInterfaces = 0;
Interfaces->AvailableInterfaces = 0;
Interfaces->BytesNeeded = 0;
if (!NdisEnumerateInterfaces(Interfaces, 256))
{
RELEASE_WRITE_LOCK (&(g_ITFLock));
dwRetCode = GetLastError ();
TRACE1 (DEVICE, "ElEnumAndOpenInterfaces: NdisEnumerateInterfaces failed with error %ld",
dwRetCode);
return dwRetCode;
}
dwNdisuioEnumBufferSize = (Interfaces->BytesNeeded + 7) & 0xfffffff8;
dwAvailableInterfaces = Interfaces->AvailableInterfaces;
if (dwNdisuioEnumBufferSize == 0)
{
RELEASE_WRITE_LOCK (&(g_ITFLock));
TRACE0 (DEVICE, "ElEnumAndOpenInterfaces: MALLOC skipped for pbNdisuioEnumBuffer as dwNdisuioEnumBufferSize == 0");
dwRetCode = NO_ERROR;
return dwRetCode;
}
pbNdisuioEnumBuffer = (BYTE *) MALLOC (4*dwNdisuioEnumBufferSize);
if (pbNdisuioEnumBuffer == NULL)
{
RELEASE_WRITE_LOCK (&(g_ITFLock));
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
TRACE0 (DEVICE, "ElEnumAndOpenInterfaces: MALLOC failed for pbNdisuioEnumBuffer");
return dwRetCode;
}
Interfaces = (PNDIS_ENUM_INTF)pbNdisuioEnumBuffer;
// Enumerate all the interfaces present on the machine
if ((dwRetCode = ElNdisuioEnumerateInterfaces (
Interfaces,
dwAvailableInterfaces,
4*dwNdisuioEnumBufferSize)) == NO_ERROR)
{
UNICODE_STRING *pInterfaceName = NULL;
UNICODE_STRING *pInterfaceDescription = NULL;
DWORD i;
// Update the interface list in the registry that NDISUIO has bound to.
// The current interface list is just overwritten into the registry.
if ((dwRetCode = ElUpdateRegistryInterfaceList (Interfaces))
!= NO_ERROR)
{
TRACE1 (DEVICE, "ElEnumAndOpenInterfaces: ElUpdateInterfaceList failed with error =%ld",
dwRetCode);
dwRetCode = NO_ERROR;
// log
}
for (i=0; i < Interfaces->TotalInterfaces; i++)
{
fPCBExists = fPCBReferenced = FALSE;
if ((dwRetCode != NO_ERROR) &&
(fSearchByDescription || fSearchByGUID))
{
break;
}
else
{
dwRetCode = NO_ERROR;
}
if (Interfaces->Interface[i].DeviceName.Buffer != NULL)
{
pInterfaceName = &(Interfaces->Interface[i].DeviceName);
}
else
{
TRACE0(INIT, "NdisEnumerateInterfaces: Device Name was NULL");
continue;
}
TRACE1(INIT, "Device: %ws", pInterfaceName->Buffer);
if (Interfaces->Interface[i].DeviceDescription.Buffer != NULL)
{
pInterfaceDescription = &(Interfaces->Interface[i].DeviceDescription);
}
else
{
TRACE0(INIT, "NdisEnumerateInterfaces: Device Description was NULL");
continue;
}
TRACE1(INIT, "Description: %ws", pInterfaceDescription->Buffer);
// EAPOL requested be started only a particular
// interface
if (fSearchByDescription)
{
if (wcscmp (pInterfaceDescription->Buffer,
pwszDesiredDescription)
!= 0)
{
// No match, continue with next interface
continue;
}
TRACE1 (DEVICE, "ElEnumAndOpenInterfaces: Found interface after enumeration %ws", pInterfaceDescription->Buffer);
}
if (fSearchByGUID)
{
if (wcsstr (pInterfaceName->Buffer,
pwszDesiredGUID)
== NULL)
{
// No match, continue with next interface
continue;
}
TRACE1 (DEVICE, "ElEnumAndOpenInterfaces: Found interface after enumeration %ws", pInterfaceName->Buffer);
}
{
// Extract GUID-string out of device name
WCHAR *pwszGUIDEnd = NULL;
WCHAR *pwszGUID = NULL;
WCHAR wchGUIDSaveLast;
pwszGUID = pInterfaceName->Buffer;
pwszGUIDStart = wcschr( pwszGUID, L'{' );
pwszGUIDEnd = wcschr( pwszGUID, L'}' );
if (pwszGUIDStart != NULL)
{
wchGUIDSaveLast = *(pwszGUIDEnd+1);
*(pwszGUIDEnd+1) = (WCHAR)NULL;
}
// Verify if a PCB already exists for the interface
// This is possible if no media disconnect was received
// after the initial media connect
pPCB = NULL;
hDevice = NULL;
ACQUIRE_WRITE_LOCK (&(g_PCBLock));
if ((pPCB = ElGetPCBPointerFromPortGUID (pwszGUIDStart)) != NULL)
{
if (EAPOL_REFERENCE_PORT (pPCB))
{
fPCBReferenced = TRUE;
}
}
RELEASE_WRITE_LOCK (&(g_PCBLock));
// Restore interface buffer
*(pwszGUIDEnd+1) = wchGUIDSaveLast;
if (pPCB != NULL)
{
// Point to existing handle
hDevice = pPCB->hPort;
fPCBExists = TRUE;
dwRetCode = NO_ERROR;
TRACE0 (INIT, "ElEnumAndOpenInterfaces: Found PCB already existing for interface");
}
else
{
TRACE0 (INIT, "ElEnumAndOpenInterfaces: Did NOT find PCB already existing for interface");
// Open handle to ndisuio driver
if ((dwRetCode = ElOpenInterfaceHandle (
pInterfaceName->Buffer,
&hDevice
)) != NO_ERROR)
{
TRACE1 (INIT, "ElEnumAndOpenInterfaces: ElOpenInterfaceHandle failed with error = %d\n",
dwRetCode );
}
}
*(pwszGUIDEnd+1) = (CHAR)NULL;
}
if (dwRetCode != NO_ERROR)
{
TRACE0 (INIT, "ElEnumAndOpenInterfaces: Failed to open handle");
continue;
}
else
{
// Create EAPOL PCB and start state machine
if ((dwRetCode = ElCreatePort (
hDevice,
pwszGUIDStart,
pInterfaceDescription->Buffer,
dwHandle,
prdUserData
)) != NO_ERROR)
{
TRACE1 (DEVICE, "ElEnumAndOpenInterfaces: Error in CreatePort = %d", dwRetCode);
if (fPCBExists)
{
if (dwRetCode = ElShutdownInterface (
pPCB->pwszDeviceGUID
) != NO_ERROR)
{
TRACE1 (DEVICE, "ElEnumAndOpenInterfaces: ElShutdownInterface handle 1 failed with error %ld",
dwRetCode);
}
if (fPCBReferenced)
{
EAPOL_DEREFERENCE_PORT (pPCB);
}
}
else
{
// Close the handle just opened to the ndisuio driver
if ((dwRetCode = ElCloseInterfaceHandle (
hDevice,
pwszGUIDStart)) != NO_ERROR)
{
TRACE1 (DEVICE,
"ElEnumAndOpenInterfaces: Error in ElCloseInterfaceHandle %d",
dwRetCode);
}
}
// Continue with the next interface
continue;
}
else
{
TRACE0 (DEVICE, "ElEnumAndOpenInterfaces: CreatePort successful");
// If PCB already existed, do not add to the hash
// table
if (fPCBExists)
{
TRACE0 (DEVICE, "ElEnumAndOpenInterfaces: PCB already existed, skipping Interface hash table addition");
fPCBExists = FALSE;
if (fPCBReferenced)
{
EAPOL_DEREFERENCE_PORT (pPCB);
}
continue;
}
if ((dwRetCode = ElCreateInterfaceEntry (
pwszGUIDStart,
pInterfaceDescription->Buffer
)) != NO_ERROR)
{
// Could not create new interface entry
// Delete Port entry created for this GUID
if ((dwRetCode = ElDeletePort (
pwszGUIDStart,
&hDevice)) != NO_ERROR)
{
TRACE1 (DEVICE, "ElEnumAndOpenInterfaces: Error in deleting port for %ws",
pwszGUIDStart);
// log
}
// Close the handle to the NDISUIO driver
if ((dwRetCode = ElCloseInterfaceHandle (
hDevice,
pwszGUIDStart)) != NO_ERROR)
{
TRACE1 (DEVICE,
"ElEnumAndOpenInterfaces: Error in ElCloseInterfaceHandle %d",
dwRetCode);
// log
}
}
}
}
} // for (i=0; i < Interfaces
}
else
{
TRACE1(INIT, "ElEnumAndOpenInterfaces: ElNdisuioEnumerateInterfaces failed with error %d",
dwRetCode);
}
TRACE1(INIT, "ElEnumAndOpenInterfaces: Completed with retcode = %d",
dwRetCode);
if (pbNdisuioEnumBuffer != NULL)
{
FREE(pbNdisuioEnumBuffer);
}
RELEASE_WRITE_LOCK (&(g_ITFLock));
return dwRetCode;
}
//
// ElOpenInterfaceHandle
//
// Description:
//
// Function called to open handle to the NDISUIO driver for an interface.
//
// Arguments:
// DeviceName - Identifier for the interface is of the
// form \Device\{GUID String}
// phDevice - Output pointer to handle of NDISUIO driver for
// the interface
//
// Return values:
// NO_ERROR - success
// non-zero - error
//
DWORD
ElOpenInterfaceHandle (
IN WCHAR *pwszDeviceName,
OUT HANDLE *phDevice
)
{
DWORD dwDesiredAccess;
DWORD dwShareMode;
LPSECURITY_ATTRIBUTES lpSecurityAttributes = NULL;
DWORD dwCreationDistribution;
DWORD dwFlagsAndAttributes;
HANDLE hTemplateFile;
HANDLE hHandle = INVALID_HANDLE_VALUE;
DWORD dwRetCode = NO_ERROR;
WCHAR wNdisDeviceName[MAX_NDIS_DEVICE_NAME_LEN];
INT wNameLength;
INT NameLength = wcslen(pwszDeviceName);
DWORD dwBytesReturned;
USHORT wEthernetType = g_wEtherType8021X;
INT i;
dwDesiredAccess = GENERIC_READ | GENERIC_WRITE;
dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;
dwCreationDistribution = OPEN_EXISTING;
dwFlagsAndAttributes = FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED;
hTemplateFile = (HANDLE)INVALID_HANDLE_VALUE;
TRACE1 (INIT, "ElOpenInterfaceHandle: Opening handle for %ws",
pwszDeviceName);
do
{
// Convert to unicode string - non-localized...
wNameLength = 0;
for (i = 0; (i < NameLength) && (i < MAX_NDIS_DEVICE_NAME_LEN-1); i++)
{
wNdisDeviceName[i] = (WCHAR)pwszDeviceName[i];
wNameLength++;
}
wNdisDeviceName[i] = L'\0';
TRACE1(DEVICE, "ElOpenInterfaceHandle: Trying to access NDIS Device: %ws\n",
wNdisDeviceName);
// --ft: Replace these calls to Ndisuio with the calls to the opened handles hash.
//hHandle = CreateFileA(
// pNdisuioDevice,
// dwDesiredAccess,
// dwShareMode,
// lpSecurityAttributes,
// dwCreationDistribution,
// dwFlagsAndAttributes,
// hTemplateFile
// );
//
//if (hHandle == INVALID_HANDLE_VALUE)
//{
// *phDevice = NULL;
// dwRetCode = GetLastError();
// TRACE1 (INIT, "ElOpenInterfaceHandle: Failed in CreateFile with error %d", dwRetCode);
// break;
//}
//else
//{
// *phDevice = hHandle;
//}
//
//if (!(DeviceIoControl(
// *phDevice,
// IOCTL_NDISUIO_OPEN_DEVICE,
// (LPVOID)&wNdisDeviceName[0],
// wNameLength*sizeof(WCHAR),
// NULL,
// 0,
// &dwBytesReturned,
// NULL)))
//
//{
// *phDevice = NULL;
// if ((dwRetCode = GetLastError()) == 0)
// {
// dwRetCode = ERROR_IO_DEVICE;
// }
// TRACE1(DEVICE, "ElOpenInterfaceHandle: Error in accessing NDIS Device: %ws", wNdisDeviceName);
// break;
//}
// The call below goes to the opened handles hash which takes care of
// sharing the handles. EAPOL doesn't have to care about anyone else
// using this handle - the sharing hash keeps a ref count for the handle
// such that the callers can just call OpenIntfHandle & CloseIntfHandle
// whenever they wish.
dwRetCode = OpenIntfHandle(
wNdisDeviceName,
&hHandle);
if (dwRetCode != ERROR_SUCCESS)
{
TRACE1(DEVICE, "ElOpenInterfaceHandle: Error in OpenIntfHandle(%ws)", wNdisDeviceName);
break;
}
*phDevice = hHandle;
TRACE2(DEVICE, "ElOpenInterfaceHandle: OpenIntfHandle(%ws) = %d", wNdisDeviceName, *phDevice);
// IOCTL down the Ethernet type
if (!(DeviceIoControl(
*phDevice,
IOCTL_NDISUIO_SET_ETHER_TYPE,
(LPVOID)&wEthernetType,
sizeof(USHORT),
NULL,
0,
&dwBytesReturned,
NULL)))
{
*phDevice = NULL;
if ((dwRetCode = GetLastError()) == 0)
{
dwRetCode = ERROR_IO_DEVICE;
}
TRACE1(DEVICE, "ElOpenInterfaceHandle: Error in ioctling ETHER type : %ws", wNdisDeviceName);
break;
}
// Bind for asynchronous I/O handling of Read/Write data
// Depending on whether it is completion for Readfile() or WriteFile()
// ElIoCompletionRoutine will call ElReadCompletionRoutine
// or ElWriteCompletionRoutine
if (!BindIoCompletionCallback(
*phDevice,
ElIoCompletionRoutine,
0
))
{
dwRetCode = GetLastError();
if (dwRetCode != ERROR_INVALID_PARAMETER)
{
*phDevice = NULL;
TRACE1 (DEVICE, "ElOpenInterfaceHandle: Error in BindIoCompletionCallBac %d", dwRetCode);
break;
}
else
{
TRACE0 (DEVICE, "ElOpenInterfaceHandle: BindIoCompletionCallback already done !!!");
dwRetCode = NO_ERROR;
}
}
} while (FALSE);
// Cleanup if there is error
if (dwRetCode != NO_ERROR)
{
if (hHandle != INVALID_HANDLE_VALUE)
{
// --ft: if anything bad happened, don't overwrite the dwRetCode - we're interested
// what the first error was, not the error that might have happened when we
// tried to close the hHandle.
// Note: ElCloseInterfaceHandle understands the Guid both decorated and un-decorated
if (ElCloseInterfaceHandle(hHandle, pwszDeviceName) != ERROR_SUCCESS)
{
TRACE1 (INIT, "ElOpenInterfaceHandle: Error in CloseHandle %d", dwRetCode);
}
}
}
TRACE2 (INIT, "ElOpenInterfaceHandle: Opened handle %p with dwRetCode %d", *phDevice, dwRetCode);
return (dwRetCode);
}
//
// ElCloseInterfaceHandle
//
// Description:
//
// Function called to close handle to NDISUIO driver for an interface
//
// Arguments:
// hDevice - Handle to NDISUIO device for the interface
//
// Return values:
// NO_ERROR - success
// non-zero - error
//
DWORD
ElCloseInterfaceHandle (
IN HANDLE hDevice,
IN LPWSTR pwszDeviceGuid
)
{
DWORD dwRetCode = ERROR_SUCCESS;
WCHAR wNdisDeviceName[MAX_NDIS_DEVICE_NAME_LEN];
TRACE2 (DEVICE, "ElCloseInterfaceHandle(0x%x,%ws) entered", hDevice, pwszDeviceGuid);
ZeroMemory (wNdisDeviceName, MAX_NDIS_DEVICE_NAME_LEN);
// if first char in the Guid is '\' then we assume the GUID format is
// '\DEVICE\{...}'. We do just the UNICODE conversion then
if (pwszDeviceGuid[0] == '\\')
{
wcscpy (wNdisDeviceName, pwszDeviceGuid);
}
// else, we assume the Guid is un-decorated, and we add the decorations.
else
{
wcscpy(wNdisDeviceName, L"\\DEVICE\\");
wcsncat(wNdisDeviceName, pwszDeviceGuid, MAX_NDIS_DEVICE_NAME_LEN - 8);
wNdisDeviceName[MAX_NDIS_DEVICE_NAME_LEN-1]=L'\0';
}
// --ft: For now, don't go directly to Ndisuio to close handles. Instead,
// go to the opened handles hash. This takes care of all the handle sharing
// problem.
dwRetCode = CloseIntfHandle(wNdisDeviceName);
//if (!CloseHandle(hDevice))
//{
// dwRetCode = GetLastError();
//}
if (dwRetCode != ERROR_SUCCESS)
{
TRACE1 (INIT, "ElCloseInterfaceHandle: Error in CloseHandle %d",
dwRetCode);
}
return dwRetCode;
}
//
// ElReadFromInterface
//
// Description:
//
// Function called to perform Overlapped read on handle to NDISUIO driver
//
// Arguments:
// hDevice - Handle to NDISUIO driver for this interface
// pElBuffer - Context buffer
// dwBufferLength - Bytes to be read
//
// Return values:
// NO_ERROR - success
// non-zero - error
//
DWORD
ElReadFromInterface (
IN HANDLE hDevice,
IN PEAPOL_BUFFER pElBuffer,
IN DWORD dwBufferLength
)
{
DWORD dwRetCode = NO_ERROR;
if (!ReadFile (
hDevice,
pElBuffer->pBuffer,
dwBufferLength,
NULL,
&pElBuffer->Overlapped
))
{
dwRetCode = GetLastError();
if (dwRetCode == ERROR_IO_PENDING)
{
// Pending status is fine, we are doing OVERLAPPED read
dwRetCode = NO_ERROR;
}
else
{
TRACE1 (DEVICE, "ElReadFromInterface: ReadFile failed with error %d",
dwRetCode);
}
}
return dwRetCode;
}
//
// ElWriteToInterface
//
// Description:
//
// Function called to perform Overlapped write on handle to NDISUIO driver
//
// Arguments:
// hDevice - Handle to NDISUIO device for this interface
// pElBuffer - Context buffer
// dwBufferLength - Bytes to be written
//
// Return values:
// NO_ERROR - success
// non-zero - error
//
DWORD
ElWriteToInterface (
IN HANDLE hDevice,
IN PEAPOL_BUFFER pElBuffer,
IN DWORD dwBufferLength
)
{
DWORD dwRetCode = NO_ERROR;
TRACE0 (DEVICE, "ElWriteToInterface entered");
if (!WriteFile (
hDevice,
pElBuffer->pBuffer,
dwBufferLength,
NULL,
&pElBuffer->Overlapped
))
{
dwRetCode = GetLastError();
if (dwRetCode == ERROR_IO_PENDING)
{
// Pending status is fine, we are doing OVERLAPPED write
dwRetCode = NO_ERROR;
}
else
{
TRACE1 (DEVICE, "ElWriteToInterface: WriteFile failed with error %d",
dwRetCode);
}
}
TRACE1 (DEVICE, "ElWriteToInterface completed, RetCode = %d", dwRetCode);
return dwRetCode;
}
//
// ElHashInterfaceDescToBucket
//
// Description:
//
// Function called to convert Friendly name of interface into interface hash
// table index.
//
// Arguments:
// pwszInterfaceDesc - Friendly name of the interface
//
// Return values:
// Hash table index between from 0 to INTF_TABLE_BUCKETS-1
//
DWORD
ElHashInterfaceDescToBucket (
IN WCHAR *pwszInterfaceDesc
)
{
return ((DWORD)((_wtol(pwszInterfaceDesc)) % INTF_TABLE_BUCKETS));
}
//
// ElGetITFPointerFromInterfaceDesc
//
// Description:
//
// Function called to convert Friendly name of interface to ITF entry pointer
//
// Arguments:
// pwszInterfaceDesc - Friendly name of the interface
//
// Return values:
// Pointer to interface entry in hash table
//
PEAPOL_ITF
ElGetITFPointerFromInterfaceDesc (
IN WCHAR *pwszInterfaceDesc
)
{
EAPOL_ITF *pITFWalker = NULL;
DWORD dwIndex;
INT i=0;
TRACE1 (DEVICE, "ElGetITFPointerFromInterfaceDesc: Desc = %ws", pwszInterfaceDesc);
if (pwszInterfaceDesc == NULL)
{
return (NULL);
}
dwIndex = ElHashInterfaceDescToBucket (pwszInterfaceDesc);
TRACE1 (DEVICE, "ElGetITFPointerFromItfDesc: Index %d", dwIndex);
for (pITFWalker = g_ITFTable.pITFBuckets[dwIndex].pItf;
pITFWalker != NULL;
pITFWalker = pITFWalker->pNext
)
{
if (wcsncmp (pITFWalker->pwszInterfaceDesc, pwszInterfaceDesc, wcslen(pwszInterfaceDesc)) == 0)
{
return pITFWalker;
}
}
return (NULL);
}
//
// ElRemoveITFFromTable
//
// Description:
//
// Function called to remove an interface entry from the interface hash
// table
//
// Arguments:
// pITF - Point to the Interface entry in the hash table
//
// Return values:
//
VOID
ElRemoveITFFromTable (
IN EAPOL_ITF *pITF
)
{
DWORD dwIndex;
EAPOL_ITF *pITFWalker = NULL;
EAPOL_ITF *pITFTemp = NULL;
if (pITF == NULL)
{
TRACE0 (EAPOL, "ElRemoveITFFromTable: Deleting NULL ITF, returning");
return;
}
dwIndex = ElHashInterfaceDescToBucket (pITF->pwszInterfaceDesc);
pITFWalker = g_ITFTable.pITFBuckets[dwIndex].pItf;
pITFTemp = pITFWalker;
while (pITFTemp != NULL)
{
if (wcsncmp (pITFTemp->pwszInterfaceGUID,
pITF->pwszInterfaceGUID, wcslen(pITF->pwszInterfaceGUID)) == 0)
{
// Entry is at head of list in table
if (pITFTemp == g_ITFTable.pITFBuckets[dwIndex].pItf)
{
g_ITFTable.pITFBuckets[dwIndex].pItf = pITFTemp->pNext;
}
else
{
// Entry is inside list in table
pITFWalker->pNext = pITFTemp->pNext;
}
break;
}
pITFWalker = pITFTemp;
pITFTemp = pITFWalker->pNext;
}
return;
}
//
// ElNdisuioEnumerateInterfaces
//
// Description:
//
// Function called to enumerate the interfaces on which NDISUIO is bound
//
// Arguments:
// pItfBuffer - Pointer to buffer which will hold interface details
// dwAvailableInterfaces - Number of interfaces for which details can
// be held in pItfBuffer
// dwBufferSize - Number of bytes in pItfBuffer
//
// Return values:
//
DWORD
ElNdisuioEnumerateInterfaces (
IN OUT PNDIS_ENUM_INTF pItfBuffer,
IN DWORD dwAvailableInterfaces,
IN DWORD dwBufferSize
)
{
DWORD dwDesiredAccess;
DWORD dwShareMode;
LPSECURITY_ATTRIBUTES lpSecurityAttributes = NULL;
DWORD dwCreationDistribution;
DWORD dwFlagsAndAttributes;
HANDLE hTemplateFile;
HANDLE hHandle;
DWORD dwBytesReturned = 0;
INT i;
CHAR Buf[1024];
DWORD BufLength = sizeof(Buf);
DWORD BytesWritten = 0;
PNDISUIO_QUERY_BINDING pQueryBinding = NULL;
PCHAR pTempBuf = NULL;
DWORD dwRetCode = NO_ERROR;
dwDesiredAccess = GENERIC_READ | GENERIC_WRITE;
dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;
dwCreationDistribution = OPEN_EXISTING;
dwFlagsAndAttributes = FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED;
hTemplateFile = (HANDLE)INVALID_HANDLE_VALUE;
TRACE0 (DEVICE, "ElNdisuioEnumerateInterfaces: Opening handle");
do
{
hHandle = CreateFileA (
pNdisuioDevice,
dwDesiredAccess,
dwShareMode,
lpSecurityAttributes,
dwCreationDistribution,
0,
NULL
);
if (hHandle == INVALID_HANDLE_VALUE)
{
dwRetCode = GetLastError();
TRACE1 (DEVICE, "ElNdisuioEnumerateInterfaces: Failed in CreateFile with error %d", dwRetCode);
break;
}
// Send IOCTL to ensure NDISUIO binds to all relevant interfaces
if (!DeviceIoControl (
hHandle,
IOCTL_NDISUIO_BIND_WAIT,
NULL,
0,
NULL,
0,
&dwBytesReturned,
NULL))
{
dwRetCode = GetLastError();
TRACE1 (DEVICE, "ElNdisuioEnumerateInterfaces: Failed in DeviceIoCoontrol NDISUIO_BIND_WAIT with error %d", dwRetCode);
break;
}
pQueryBinding = (PNDISUIO_QUERY_BINDING)Buf;
pTempBuf = (PBYTE)pItfBuffer + dwBufferSize;
i = 0;
for (pQueryBinding->BindingIndex = i;
pQueryBinding->BindingIndex < dwAvailableInterfaces;
pQueryBinding->BindingIndex = ++i)
{
// Query for one interface at a time
if (DeviceIoControl (
hHandle,
IOCTL_NDISUIO_QUERY_BINDING,
pQueryBinding,
sizeof(NDISUIO_QUERY_BINDING),
Buf,
BufLength,
&BytesWritten,
NULL))
{
TRACE3 (DEVICE, "NdisuioEnumerateInterfaces: NDISUIO bound to: (%ld) %ws\n - %ws\n",
pQueryBinding->BindingIndex,
(PUCHAR)pQueryBinding + pQueryBinding->DeviceNameOffset,
(PUCHAR)pQueryBinding + pQueryBinding->DeviceDescrOffset);
pTempBuf = pTempBuf - ((pQueryBinding->DeviceNameLength + 7) & 0xfffffff8);
if (((PBYTE)pTempBuf - (PBYTE)&pItfBuffer->Interface[pItfBuffer->TotalInterfaces]) <= 0)
{
// Going beyond start of buffer, Error
TRACE0 (DEVICE, "NdisuioEnumerateInterfaces: DeviceName: Memory being corrupted !!!");
dwRetCode = ERROR_INVALID_DATA;
break;
}
pItfBuffer->Interface[pItfBuffer->TotalInterfaces].DeviceName.Buffer = (PWCHAR) pTempBuf;
memcpy ((BYTE *)(pItfBuffer->Interface[pItfBuffer->TotalInterfaces].DeviceName.Buffer), (BYTE *)((PUCHAR)pQueryBinding + pQueryBinding->DeviceNameOffset), (pQueryBinding->DeviceNameLength ));
pItfBuffer->Interface[pItfBuffer->TotalInterfaces].DeviceName.Length = (SHORT) ( pQueryBinding->DeviceNameLength );
pItfBuffer->Interface[pItfBuffer->TotalInterfaces].DeviceName.MaximumLength = (SHORT) ( pQueryBinding->DeviceNameLength );
pTempBuf = pTempBuf - ((pQueryBinding->DeviceDescrLength + 7) & 0xfffffff8);;
if (((PBYTE)pTempBuf - (PBYTE)&pItfBuffer->Interface[pItfBuffer->TotalInterfaces]) <= 0)
{
// Going beyond start of buffer, Error
TRACE0 (DEVICE, "NdisuioEnumerateInterfaces: DeviceDescr: Memory being corrupted !!!");
dwRetCode = ERROR_INVALID_DATA;
break;
}
pItfBuffer->Interface[pItfBuffer->TotalInterfaces].DeviceDescription.Buffer = (PWCHAR) pTempBuf;
memcpy ((pItfBuffer->Interface[pItfBuffer->TotalInterfaces].DeviceDescription.Buffer), (PWCHAR)((PUCHAR)pQueryBinding + pQueryBinding->DeviceDescrOffset), (pQueryBinding->DeviceDescrLength ));
pItfBuffer->Interface[pItfBuffer->TotalInterfaces].DeviceDescription.Length = (SHORT) (pQueryBinding->DeviceDescrLength );
pItfBuffer->Interface[pItfBuffer->TotalInterfaces].DeviceDescription.MaximumLength = (SHORT) (pQueryBinding->DeviceDescrLength );
pItfBuffer->TotalInterfaces++;
memset(Buf, 0, BufLength);
}
else
{
dwRetCode = GetLastError ();
if (dwRetCode != ERROR_NO_MORE_ITEMS)
{
TRACE1 (DEVICE, "ElNdisuioEnumerateInterfaces: DeviceIoControl terminated for with IOCTL_NDISUIO_QUERY_BINDING with error %ld",
dwRetCode);
}
else
{
// Reset error, since it only indicates end-of-list
dwRetCode = NO_ERROR;
TRACE0 (DEVICE, "ElNdisuioEnumerateInterfaces: DeviceIoControl IOCTL_NDISUIO_QUERY_BINDING has no more entries");
}
break;
}
}
} while (FALSE);
// Cleanup
if (hHandle != INVALID_HANDLE_VALUE)
{
if (!CloseHandle(hHandle))
{
dwRetCode = GetLastError();
TRACE1 (DEVICE, "ElNdisuioEnumerateInterfaces: Error in CloseHandle %d", dwRetCode);
}
}
return dwRetCode;
}
//
// ElShutdownInterface
//
// Description:
//
// Function called to stop EAPOL state machine, close handle to NDISUIO and
// remove existence of the interface from the module
//
// Arguments:
// pwszDeviceGUID - Pointer to interface GUID
//
// Return values:
// NO_ERROR - success
// non-zero - error
//
DWORD
ElShutdownInterface (
IN WCHAR *pwszDeviceGUID
)
{
WCHAR *pwszGUID = NULL;
EAPOL_PCB *pPCB = NULL;
EAPOL_ITF *pITF = NULL;
HANDLE hDevice = NULL;
DWORD dwRetCode = NO_ERROR;
do
{
TRACE1 (DEVICE, "ElShutdownInterface: Called for interface removal for %ws",
pwszGUID);
pwszGUID = MALLOC ( (wcslen (pwszDeviceGUID) + 1) * sizeof(WCHAR));
if (pwszGUID == NULL)
{
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
TRACE0 (DEVICE, "ElShutdownInterface: MALLOC failed for pwszGUID");
break;
}
wcscpy (pwszGUID, pwszDeviceGUID);
ACQUIRE_WRITE_LOCK (&(g_ITFLock));
// Check if EAPOL was actually started on this interface
// Verify by checking existence of corresponding
// entry in hash table
ACQUIRE_WRITE_LOCK (&(g_PCBLock));
if ((pPCB = ElGetPCBPointerFromPortGUID(pwszGUID))
!= NULL)
{
RELEASE_WRITE_LOCK (&(g_PCBLock));
TRACE0 (DEVICE, "ElShutdownInterface: Found PCB entry for interface");
if ((pITF = ElGetITFPointerFromInterfaceDesc(
pPCB->pwszFriendlyName))
== NULL)
{
TRACE0 (DEVICE, "ElShutdownInterface: Did not find ITF entry when PCB exits, HOW BIZARRE !!!");
}
if ((dwRetCode = ElDeletePort (
pwszGUID,
&hDevice)) != NO_ERROR)
{
TRACE1 (DEVICE, "ElShutdownInterface: Error in deleting port for %ws",
pPCB->pwszDeviceGUID);
}
// Remove interface entry from interface table
if (pITF != NULL)
{
ElRemoveITFFromTable(pITF);
if (pITF->pwszInterfaceDesc)
{
FREE (pITF->pwszInterfaceDesc);
}
if (pITF->pwszInterfaceGUID)
{
FREE (pITF->pwszInterfaceGUID);
}
if (pITF)
{
FREE (pITF);
}
}
// Close the handle to the NDISUIO driver
if (hDevice != NULL)
{
if ((dwRetCode = ElCloseInterfaceHandle (hDevice, pwszGUID))
!= NO_ERROR)
{
TRACE1 (DEVICE,
"ElShutdownInterface: Error in ElCloseInterfaceHandle %d",
dwRetCode);
}
}
TRACE1 (DEVICE, "ElShutdownInterface: Port deleted (%ws)",
pwszGUID);
}
else
{
RELEASE_WRITE_LOCK (&(g_PCBLock));
// Ignore device removal
TRACE0 (DEVICE, "ElShutdownInterface: ElGetPCBPointerFromPortGUID did not find any matching entry, ignoring interface REMOVAL");
}
RELEASE_WRITE_LOCK (&(g_ITFLock));
} while (FALSE);
if (pwszGUID != NULL)
{
FREE (pwszGUID);
}
return dwRetCode;
}
//
// ElCreateInterfaceEntry
//
// Description:
//
// Function called to create an entry in the global interface table
//
// Arguments:
// pwszInterfaceGUID - Pointer to interface GUID
// pwszInterfaceDescription - Pointer to interface Description
//
// Return values:
// NO_ERROR - success
// non-zero - error
//
DWORD
ElCreateInterfaceEntry (
IN WCHAR *pwszInterfaceGUID,
IN WCHAR *pwszInterfaceDescription
)
{
EAPOL_ITF * pNewITF = NULL;
DWORD dwIndex = 0;
DWORD dwRetCode = NO_ERROR;
do
{
dwIndex = ElHashInterfaceDescToBucket (pwszInterfaceDescription);
pNewITF = (PEAPOL_ITF) MALLOC (sizeof (EAPOL_ITF));
if (pNewITF == NULL)
{
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
break;
}
pNewITF->pwszInterfaceGUID =
(WCHAR *) MALLOC ((wcslen(pwszInterfaceGUID) + 1)*sizeof(WCHAR));
if (pNewITF->pwszInterfaceGUID == NULL)
{
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
break;
}
pNewITF->pwszInterfaceDesc =
(WCHAR *) MALLOC ((wcslen(pwszInterfaceDescription) + 1)*sizeof(WCHAR));
if (pNewITF->pwszInterfaceDesc == NULL)
{
dwRetCode = ERROR_NOT_ENOUGH_MEMORY;
break;
}
wcscpy (pNewITF->pwszInterfaceGUID, pwszInterfaceGUID);
wcscpy (pNewITF->pwszInterfaceDesc, pwszInterfaceDescription);
pNewITF->pNext = g_ITFTable.pITFBuckets[dwIndex].pItf;
g_ITFTable.pITFBuckets[dwIndex].pItf = pNewITF;
TRACE3 (DEVICE, "ElCreateInterfaceEntry: Added to hash table GUID= %ws : Desc= %ws at Index=%d",
pNewITF->pwszInterfaceGUID,
pNewITF->pwszInterfaceDesc,
dwIndex
);
}
while (FALSE);
if (dwRetCode != NO_ERROR)
{
if (pNewITF)
{
if (pNewITF->pwszInterfaceDesc)
{
FREE (pNewITF->pwszInterfaceDesc);
}
if (pNewITF->pwszInterfaceGUID)
{
FREE (pNewITF->pwszInterfaceGUID);
}
FREE (pNewITF);
}
}
return dwRetCode;
}
Reference in a new issue Copy permalink