network.c File Reference

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

Go to the source code of this file.

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 )

static

Definition at line 29 of file network.c.

{
    /* FIXME: query NDIS for all the interfaces */
    UNIMPLEMENTED;
    return FALSE;
}

Referenced by DealWithDeviceEvent().