#ifdef OS_WINCE
// BUGBUG -- cleanup sources file in this dir a little (include wincecom.inc)

#include "windows.h"
#include <winsock.h>
// extern "C" {
// The following four files are from the OS, but are needed here
// by the GUID-generation code, so they were copied over.
// Ick...
#include "tdiinfo.h"
#include "tdistat.h"
#include "llinfo.h"
#include "wscntl.h"

#define REGISTRY_ROOT                   TEXT("Software\\Microsoft\\Terminal Server Client")
#define REGISTRY_VALUE_UUID             TEXT("UUID")
#define ETHER_ADDRESS_LENGTH            6
#define DEFAULT_MINIMUM_ENTITIES        32
#define MAX_ADAPTER_DESCRIPTION_LENGTH  128
#define RPC_UUID_TIME_HIGH_MASK         0x0FFF
#define RPC_UUID_VERSION                0x1000
#define RPC_UUID_RESERVED               0x80
#define RPC_UUID_CLOCK_SEQ_HI_MASK      0x3F

typedef struct {
    unsigned long  ulTimeLow;
    unsigned short usTimeMid;
    unsigned short usTimeHiAndVersion;
    unsigned char  ucClockSeqHiAndReserved;
    unsigned char  ucClockSeqLow;
    unsigned char  ucNodeId[6];
} _UUID;

static unsigned int gs_seq = 0;
static LARGE_INTEGER gs_tm = {0};
static unsigned char gs_ucEtherAddr[ETHER_ADDRESS_LENGTH];
static int  gs_fIsAddressInitialized = FALSE;

static void GetAdapterAddress()
{
    if(!gs_fIsAddressInitialized) {
        memset (gs_ucEtherAddr, 0, sizeof(gs_ucEtherAddr));
        // Load WinSock.DLL so we can call into WsControl(...)
        HINSTANCE hInstWinSock = LoadLibrary(TEXT("winsock.dll"));
        if(0 != hInstWinSock) {
            DWORD(*fpfWsControl)(DWORD Protocol, DWORD Action, LPVOID InputBuffer, LPDWORD InputBufferLength, LPVOID OutputBuffer, LPDWORD OutputBufferLength);
            fpfWsControl = (DWORD(*)(DWORD Protocol, DWORD Action, LPVOID InputBuffer, LPDWORD InputBufferLength, LPVOID OutputBuffer, LPDWORD OutputBufferLength))GetProcAddress(hInstWinSock, TEXT("WsControl"));
            if(0 != fpfWsControl) {
                //  First, obtain list of TCP entities...
                TCP_REQUEST_QUERY_INFORMATION_EX req;
                memset(&req, 0, sizeof(req));
                req.ID.toi_entity.tei_entity = GENERIC_ENTITY;
                req.ID.toi_entity.tei_instance = 0;
                req.ID.toi_class = INFO_CLASS_GENERIC;
                req.ID.toi_type = INFO_TYPE_PROVIDER;
                req.ID.toi_id = ENTITY_LIST_ID;
                int iInputLen  = sizeof(req);
                int iOutputLen = sizeof(TDIEntityID) * DEFAULT_MINIMUM_ENTITIES;
                TDIEntityID *pEntity = NULL;
                for ( ; ; ) {
                    int iPrevOutputLen = iOutputLen;
                    pEntity = (TDIEntityID*)LocalAlloc(LPTR, (size_t)iOutputLen);
                    if (!pEntity)
                        break;
                    DWORD status = fpfWsControl(IPPROTO_TCP,
                                       WSCNTL_TCPIP_QUERY_INFO,
                                       (LPVOID)&req,
                                       (ULONG *)&iInputLen,
                                       (LPVOID)pEntity,
                                       (ULONG *)&iOutputLen
                                       );
                    if (status != TDI_SUCCESS) {
                        LocalFree(pEntity);
                        pEntity = NULL;
                        break;
                    }
                    if (iOutputLen <= iPrevOutputLen)
                        break;
                    LocalFree(pEntity);
                }
                if (0 != pEntity) {
                    int iCount = (UINT)(iOutputLen / sizeof(TDIEntityID));
                    //  Second, walk through these in search of adapters
                    TDIEntityID *pRunner = pEntity;
                    for (int i = 0; i < iCount; ++i, ++pRunner) {
                        // IF_ENTITY: this entity/instance describes an adapter
                        if (pRunner->tei_entity == IF_ENTITY) {
                            // find out if this entity supports MIB requests
                            memset(&req, 0, sizeof(req));
                            TDIObjectID id;
                            memset (&id, 0, sizeof(id));
                            id.toi_entity = *pRunner;
                            id.toi_class  = INFO_CLASS_GENERIC;
                            id.toi_type   = INFO_TYPE_PROVIDER;
                            id.toi_id     = ENTITY_TYPE_ID;
                            req.ID = id;
                            DWORD fIsMib = FALSE;
                            iInputLen  = sizeof(req);
                            iOutputLen = sizeof(fIsMib);
                            DWORD status = fpfWsControl(IPPROTO_TCP,
                                               WSCNTL_TCPIP_QUERY_INFO,
                                               (LPVOID)&req,
                                               (ULONG *)&iInputLen,
                                               (LPVOID)&fIsMib,
                                               (ULONG *)&iOutputLen
                                               );
                            if (status != TDI_SUCCESS)
                                break;
                            if (fIsMib != IF_MIB)
                                continue;
                            // MIB requests supported - query the adapter info
                            id.toi_class = INFO_CLASS_PROTOCOL;
                            id.toi_id = IF_MIB_STATS_ID;
                            memset(&req, 0, sizeof(req));
                            req.ID = id;
                            BYTE info[sizeof(IFEntry) + MAX_ADAPTER_DESCRIPTION_LENGTH + 1];
                            iInputLen  = sizeof(req);
                            iOutputLen = sizeof(info);
                            status = fpfWsControl(IPPROTO_TCP,
                                               WSCNTL_TCPIP_QUERY_INFO,
                                               (LPVOID)&req,
                                               (ULONG *)&iInputLen,
                                               (LPVOID)&info,
                                               (ULONG *)&iOutputLen
                                               );
                            if (status != TDI_SUCCESS)
                                break;
                            if (iOutputLen > sizeof(info))
                                continue;
                            IFEntry* pIfEntry = (IFEntry*)info;
                            memcpy (gs_ucEtherAddr, pIfEntry->if_physaddr, sizeof (gs_ucEtherAddr));
                            gs_fIsAddressInitialized = TRUE;
                            break;
                        }
                    }
                    LocalFree (pEntity);
                }
            }
            FreeLibrary(hInstWinSock);
        }
    }
}


BOOL generate_guid (GUID *guid) {
    GetAdapterAddress();  // It may fail, but gs_ucEtherAddr will still be all 0
    _UUID *p_uuid = (_UUID *)guid;
    SYSTEMTIME st;
    GetLocalTime(&st);
    LARGE_INTEGER tm2;
    SystemTimeToFileTime (&st, (FILETIME *)&tm2);
    if (gs_tm.QuadPart < tm2.QuadPart)
        gs_tm = tm2;
    else
        ++gs_tm.QuadPart;
    if (gs_tm.QuadPart > tm2.QuadPart + 1000)   { // Clock reset or super heavy usage
        gs_tm = tm2;
        ++gs_seq;
    }
    unsigned int uiLowPart  = gs_tm.LowPart;
    unsigned int uiHighPart = gs_tm.HighPart;
    unsigned int uiSeq      = gs_seq;
    p_uuid->ulTimeLow = (unsigned long)uiLowPart;
    p_uuid->usTimeMid = (unsigned short)(uiHighPart & 0x0000FFFF);
    p_uuid->usTimeHiAndVersion = (unsigned short)(( (unsigned short)(uiHighPart >> 16) & RPC_UUID_TIME_HIGH_MASK) | RPC_UUID_VERSION);
    p_uuid->ucClockSeqHiAndReserved = RPC_UUID_RESERVED | (((unsigned char) (uiSeq >> 8)) & (unsigned char) RPC_UUID_CLOCK_SEQ_HI_MASK);
    p_uuid->ucClockSeqLow = (unsigned char) (uiSeq & 0x00FF);
    p_uuid->ucNodeId[0] = gs_ucEtherAddr[0];
    p_uuid->ucNodeId[1] = gs_ucEtherAddr[1];
    p_uuid->ucNodeId[2] = gs_ucEtherAddr[2];
    p_uuid->ucNodeId[3] = gs_ucEtherAddr[3];
    p_uuid->ucNodeId[4] = gs_ucEtherAddr[4];
    p_uuid->ucNodeId[5] = gs_ucEtherAddr[5];
    return TRUE;
}

BOOL OEMGetUUID(GUID* pGuid)
{
    DWORD len;
    HKEY hKey;
    BOOL fRetVal = FALSE;

    len = sizeof(UUID);

    // Try to read the UUID from the registry - if we find one, use it
    if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, REGISTRY_ROOT, 0, KEY_READ, &hKey) == ERROR_SUCCESS)
    {
        DWORD dwType;

        if ( ( RegQueryValueEx(hKey, REGISTRY_VALUE_UUID, 0, &dwType, 
                (BYTE*)pGuid, &len) == ERROR_SUCCESS) && (sizeof(UUID) == len) )
        {
            fRetVal = TRUE;
        }

        RegCloseKey(hKey);
    }

    if (!fRetVal)
    {
        // We didn't find a UUID in the registry, so we need to generate one now

        // First, try asking the hardware for a UUID...
        fRetVal = KernelIoControl(IOCTL_HAL_GET_UUID, NULL, 0, pGuid, len, &len);

        // If the hardware was unable to provide a UUID, generate one now.
        if (!fRetVal)
        {
            fRetVal = generate_guid(pGuid);
        }

        // Save the UUID (however we got it) in the registry so that we will always use this UUID
        if(fRetVal) {
            DWORD dwDisposition;
            BOOL fSavedKey = FALSE;
            if (RegCreateKeyEx(HKEY_LOCAL_MACHINE, REGISTRY_ROOT, 0, 0, 0, KEY_ALL_ACCESS, 0, &hKey, &dwDisposition) == ERROR_SUCCESS)
            {
                if (RegSetValueEx(hKey, REGISTRY_VALUE_UUID, 0, REG_BINARY, (BYTE*)pGuid, len) == ERROR_SUCCESS)
                {
                    fSavedKey = TRUE;
                }
                RegCloseKey(hKey);
            }
            // If we can't save the registry key, we have to return failure because we don't want to leak licenses
            fRetVal = fSavedKey;
        }
    }

    if (!fRetVal)
    {
        // We failed to generate a UUID.
        MessageBox(NULL, 
                   TEXT("Can't read or generate licensing information.  (Unable to generate a UUID or read one from the registry.)"), 
                   TEXT("Error"), MB_OK);

    }

    return fRetVal;
}

// }; // extern "C"

#endif // OS_WINCE