#include <windef.h>
#include <winbase.h>
#include <winreg.h>
#include <winsvc.h>
#include <ntddndis.h>
#include <ndk/exfuncs.h>
#include <ntsecapi.h>
#include <debug.h>

Include dependency graph for network.c:

Macros
#define	WIN32_NO_STATUS

#define	NDEBUG

#define	SEED_BUFFER_SIZE 6

Functions
static BOOLEAN	getMacAddress (UCHAR *MacAddress)

static VOID	CookupNodeId (UCHAR *NodeId)

VOID	DealWithDeviceEvent (VOID)

Macro Definition Documentation

◆ NDEBUG

#define NDEBUG

Definition at line 21 of file network.c.

◆ SEED_BUFFER_SIZE

#define SEED_BUFFER_SIZE 6

Definition at line 24 of file network.c.

◆ WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 11 of file network.c.

Function Documentation

◆ CookupNodeId()

static VOID CookupNodeId ( UCHAR * NodeId )

static

Definition at line 37 of file network.c.

 {
     CHAR ComputerName[MAX_COMPUTERNAME_LENGTH + 1];
     CHAR * CurrentChar;
     DWORD Size;
     LARGE_INTEGER PerformanceCount;
     PDWORD NodeBegin, NodeMiddle;
     DWORD dwValue;
     MEMORYSTATUS MemoryStatus;
     LUID Luid;
     WCHAR szSystem[MAX_PATH];
     ULARGE_INTEGER FreeBytesToCaller, TotalBytes, TotalFreeBytes;
 
     /* Initialize node id */
     memset(NodeId, 0x11, SEED_BUFFER_SIZE * sizeof(UCHAR));
 
     /* Randomize it with computer name */
     Size = MAX_COMPUTERNAME_LENGTH + 1;
     if (GetComputerNameA(ComputerName, &Size))
     {
         Size = 0;
         CurrentChar = &ComputerName[0];
         while (*CurrentChar != ANSI_NULL)
         {
             NodeId[Size] ^= *CurrentChar;
             ++CurrentChar;
 
             /* Don't overflow our NodeId */
             if (++Size >= SEED_BUFFER_SIZE)
             {
                 Size = 0;
             }
         }
     }
 
     /* Now, we'll work directly on DWORD values */
     NodeBegin = (DWORD *)&NodeId[0];
     NodeMiddle = (DWORD *)&NodeId[2];
 
     /* Randomize with performance counter */
     if (QueryPerformanceCounter(&PerformanceCount))
     {
         *NodeMiddle = *NodeMiddle ^ PerformanceCount.u.HighPart ^ PerformanceCount.u.LowPart;
 
         dwValue = PerformanceCount.u.HighPart ^ PerformanceCount.u.LowPart ^ *NodeBegin;
     }
     else
     {
         dwValue = *NodeBegin;
     }
 
     *NodeBegin = *NodeBegin ^ dwValue;
     *NodeMiddle = *NodeMiddle ^ *NodeBegin;
 
     /* Then, with memory status */
     MemoryStatus.dwLength = sizeof(MemoryStatus);
     GlobalMemoryStatus(&MemoryStatus);
 
     *NodeBegin = *NodeBegin ^ MemoryStatus.dwMemoryLoad;
     *NodeMiddle = *NodeMiddle ^ MemoryStatus.dwTotalPhys;
     *NodeBegin = *NodeBegin ^ MemoryStatus.dwTotalPageFile ^ MemoryStatus.dwAvailPhys;
     *NodeMiddle = *NodeMiddle ^ MemoryStatus.dwTotalVirtual ^ MemoryStatus.dwAvailPageFile;
     *NodeBegin = *NodeBegin ^ MemoryStatus.dwAvailVirtual;
 
     /* With a LUID */
     if (AllocateLocallyUniqueId(&Luid))
     {
         *NodeBegin = *NodeBegin ^ Luid.LowPart;
         *NodeMiddle = *NodeMiddle ^ Luid.HighPart;
     }
 
     /* Get system directory */
     GetSystemDirectoryW(szSystem, ARRAYSIZE(szSystem));
 
     /* And finally with free disk space */
     if (GetDiskFreeSpaceExW(szSystem, &FreeBytesToCaller, &TotalBytes, &TotalFreeBytes))
     {
         *NodeMiddle ^= FreeBytesToCaller.LowPart ^ TotalFreeBytes.HighPart;
         *NodeMiddle ^= FreeBytesToCaller.HighPart ^ TotalFreeBytes.LowPart;
         *NodeBegin ^= TotalBytes.LowPart ^ TotalFreeBytes.LowPart;
         *NodeBegin ^= TotalBytes.HighPart ^ TotalFreeBytes.HighPart;
     }
 
     /*
      * Because it was locally generated, force the seed to be local
      * setting this, will trigger the check for validness in the kernel
      * and make the seed local
      */
     NodeId[0] |= 0x80u;
 }

Referenced by DealWithDeviceEvent().

◆ DealWithDeviceEvent()

VOID DealWithDeviceEvent ( VOID )

Definition at line 128 of file network.c.

 {
     NTSTATUS Status;
     UCHAR UuidSeed[SEED_BUFFER_SIZE];
 
     /* First, try to get a multicast MAC address */
     if (!getMacAddress(UuidSeed))
     {
         DPRINT1("Failed finding a proper MAC address, will generate seed\n");
         CookupNodeId(UuidSeed);
     }
 
     /* Seed our UUID generator */
     Status = NtSetUuidSeed(UuidSeed);
     if (!NT_SUCCESS(Status))
     {
         DPRINT1("NtSetUuidSeed failed with status: %lx\n", Status);
     }
 }

◆ getMacAddress()

static BOOLEAN getMacAddress ( UCHAR * MacAddress )