#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 )