misc.c File Reference

#include "diskpart.h"

Include dependency graph for misc.c:

Go to the source code of this file.

Functions
BOOL	IsDecString (_In_ PWSTR pszDecString)
BOOL	IsHexString (_In_ PWSTR pszHexString)
WCHAR	IsDriveLetter (_In_ PWSTR pszString)
BOOL	HasPrefix (_In_ PWSTR pszString, _In_ PWSTR pszPrefix, _Out_opt_ PWSTR *ppszSuffix)
ULONGLONG	RoundingDivide (_In_ ULONGLONG Dividend, _In_ ULONGLONG Divisor)
PWSTR	DuplicateQuotedString (_In_ PWSTR pszInString)
PWSTR	DuplicateString (_In_ PWSTR pszInString)
VOID	CreateGUID (_Out_ GUID *pGuid)
VOID	CreateSignature (_Out_ PDWORD pSignature)
VOID	PrintGUID (_Out_ PWSTR pszBuffer, _In_ GUID *pGuid)
static BYTE	CharToByte (WCHAR Char)
BOOL	StringToGUID (_Out_ GUID *pGuid, _In_ PWSTR pszString)
VOID	PrintBusType (_Out_ PWSTR pszBuffer, _In_ INT cchBufferMax, _In_ STORAGE_BUS_TYPE BusType)

Function Documentation

CharToByte()

static BYTE CharToByte ( WCHAR  Char )

static

Definition at line 226 of file misc.c.

{
    if (iswdigit(Char))
    {
        return (BYTE)((INT)Char - (INT)'0');
    }
    else if (iswalpha(Char))
    {
        if (iswupper(Char))
            return (INT)Char - ((INT)'A' - 10);
        else
            return (INT)Char - ((INT)'a' - 10);
    }
    return 0;
}

Referenced by StringToGUID().

CreateGUID()

VOID CreateGUID

(

_Out_ GUID *

pGuid

)

Definition at line 170 of file misc.c.

{
    RtlGenRandom(pGuid, sizeof(*pGuid));
    /* Clear the version bits and set the version (4) */
    pGuid->Data3 &= 0x0fff;
    pGuid->Data3 |= (4 << 12);
    /* Set the topmost bits of Data4 (clock_seq_hi_and_reserved) as
     * specified in RFC 4122, section 4.4.
     */
    pGuid->Data4[0] &= 0x3f;
    pGuid->Data4[0] |= 0x80;
}

Referenced by ConvertGPT(), and InsertGptPartition().

CreateSignature()

VOID CreateSignature

(

_Out_ PDWORD

pSignature

)

Definition at line 186 of file misc.c.

{
    LARGE_INTEGER SystemTime;
    TIME_FIELDS TimeFields;
    PUCHAR Buffer;
 
    NtQuerySystemTime(&SystemTime);
    RtlTimeToTimeFields(&SystemTime, &TimeFields);
 
    Buffer = (PUCHAR)pSignature;
    Buffer[0] = (UCHAR)(TimeFields.Year & 0xFF) + (UCHAR)(TimeFields.Hour & 0xFF);
    Buffer[1] = (UCHAR)(TimeFields.Year >> 8) + (UCHAR)(TimeFields.Minute & 0xFF);
    Buffer[2] = (UCHAR)(TimeFields.Month & 0xFF) + (UCHAR)(TimeFields.Second & 0xFF);
    Buffer[3] = (UCHAR)(TimeFields.Day & 0xFF) + (UCHAR)(TimeFields.Milliseconds & 0xFF);
}

Referenced by ConvertMBR(), CreateExtendedPartition(), and CreatePrimaryPartition().

DuplicateQuotedString()

PWSTR DuplicateQuotedString

(

_In_ PWSTR

pszInString

)

Definition at line 102 of file misc.c.

{
    PWSTR pszOutString = NULL;
    PWSTR pStart, pEnd;
    INT nLength;
 
    if ((pszInString == NULL) || (pszInString[0] == UNICODE_NULL))
        return NULL;
 
    if (pszInString[0] == L'"')
    {
        if (pszInString[1] == UNICODE_NULL)
            return NULL;
 
        pStart = &pszInString[1];
        pEnd = wcschr(pStart, '"');
        if (pEnd == NULL)
        {
            nLength = wcslen(pStart);
        }
        else
        {
            nLength = (pEnd - pStart);
        }
    }
    else
    {
        pStart = pszInString;
        nLength = wcslen(pStart);
    }
 
    pszOutString = RtlAllocateHeap(RtlGetProcessHeap(),
                                   HEAP_ZERO_MEMORY,
                                   (nLength + 1) * sizeof(WCHAR));
    if (pszOutString == NULL)
        return NULL;
 
    wcsncpy(pszOutString, pStart, nLength);
 
    return pszOutString;
}

DuplicateString()

PWSTR DuplicateString

(

_In_ PWSTR

pszInString

)

Definition at line 147 of file misc.c.

{
    PWSTR pszOutString = NULL;
    INT nLength;
 
    if ((pszInString == NULL) || (pszInString[0] == UNICODE_NULL))
        return NULL;
 
    nLength = wcslen(pszInString);
    pszOutString = RtlAllocateHeap(RtlGetProcessHeap(),
                                   HEAP_ZERO_MEMORY,
                                   (nLength + 1) * sizeof(WCHAR));
    if (pszOutString == NULL)
        return NULL;
 
    wcscpy(pszOutString, pszInString);
 
    return pszOutString;
}

HasPrefix()

BOOL HasPrefix	(	_In_ PWSTR	pszString,
		_In_ PWSTR	pszPrefix,
		_Out_opt_ PWSTR *	ppszSuffix
	)

Definition at line 76 of file misc.c.

{
    INT nPrefixLength, ret;
 
    nPrefixLength = wcslen(pszPrefix);
    ret = _wcsnicmp(pszString, pszPrefix, nPrefixLength);
    if ((ret == 0) && (ppszSuffix != NULL))
        *ppszSuffix = &pszString[nPrefixLength];
 
    return (ret == 0);
}

Referenced by AcpiExGetNameString(), assign_main(), CreateEfiPartition(), CreateExtendedPartition(), CreateLogicalPartition(), CreateMsrPartition(), CreatePrimaryPartition(), format_main(), gpt_main(), remove_main(), setid_main(), and UniqueIdDisk().

IsDecString()

BOOL IsDecString

(

_In_ PWSTR

pszDecString

)

Definition at line 14 of file misc.c.

{
    PWSTR ptr;
 
    if ((pszDecString == NULL) || (*pszDecString == UNICODE_NULL))
        return FALSE;
 
    ptr = pszDecString;
    while (*ptr != UNICODE_NULL)
    {
        if (!iswdigit(*ptr))
            return FALSE;
 
        ptr++;
    }
 
    return TRUE;
}

Referenced by SelectDisk(), SelectPartition(), and SelectVolume().

IsDriveLetter()

WCHAR IsDriveLetter

(

_In_ PWSTR

pszString

)

Definition at line 57 of file misc.c.

{
    INT nLength;
 
    nLength = wcslen(pszString);
    if ((nLength == 0) || (nLength > 2))
        return UNICODE_NULL;
 
    if (!iswalpha(pszString[0]))
        return UNICODE_NULL;
 
    if ((nLength > 1) && (pszString[1] != L':'))
        return UNICODE_NULL;
 
    return pszString[0];
}

Referenced by DeleteDriveLetterRoutine(), HasDriveLetter(), MountMgrCreatePointWorker(), MountMgrDeletePoints(), MountMgrDeletePointsDbOnly(), MountMgrMountedDeviceArrival(), MountMgrNextDriveLetterWorker(), and SelectVolume().

IsHexString()

BOOL IsHexString

(

_In_ PWSTR

pszHexString

)

Definition at line 36 of file misc.c.

{
    PWSTR ptr;
 
    if ((pszHexString == NULL) || (*pszHexString == UNICODE_NULL))
        return FALSE;
 
    ptr = pszHexString;
    while (*ptr != UNICODE_NULL)
    {
        if (!iswxdigit(*ptr))
            return FALSE;
 
        ptr++;
    }
 
    return TRUE;
}

Referenced by UniqueIdDisk().

PrintBusType()

VOID PrintBusType	(	_Out_ PWSTR	pszBuffer,
		_In_ INT	cchBufferMax,
		_In_ STORAGE_BUS_TYPE	BusType
	)

Definition at line 313 of file misc.c.

{
    if (BusType <= BusTypeMax)
        LoadStringW(GetModuleHandle(NULL),
                    IDS_BUSTYPE_UNKNOWN + BusType,
                    pszBuffer,
                    cchBufferMax);
    else
        LoadStringW(GetModuleHandle(NULL),
                    IDS_BUSTYPE_OTHER,
                    pszBuffer,
                    cchBufferMax);
}

Referenced by DetailDisk().

PrintGUID()

VOID PrintGUID	(	_Out_ PWSTR	pszBuffer,
		_In_ GUID *	pGuid
	)

Definition at line 205 of file misc.c.

{
    _swprintf(pszBuffer,
              L"%08lx-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x",
              pGuid->Data1,
              pGuid->Data2,
              pGuid->Data3,
              pGuid->Data4[0],
              pGuid->Data4[1],
              pGuid->Data4[2],
              pGuid->Data4[3],
              pGuid->Data4[4],
              pGuid->Data4[5],
              pGuid->Data4[6],
              pGuid->Data4[7]);
}

Referenced by DetailDisk(), DetailPartition(), and UniqueIdDisk().

RoundingDivide()

ULONGLONG RoundingDivide	(	_In_ ULONGLONG	Dividend,
		_In_ ULONGLONG	Divisor
	)

Definition at line 93 of file misc.c.

{
    return (Dividend + Divisor / 2) / Divisor;
}

StringToGUID()

BOOL StringToGUID	(	_Out_ GUID *	pGuid,
		_In_ PWSTR	pszString
	)

Definition at line 245 of file misc.c.

{
    INT i;
 
    if (pszString == NULL)
        return FALSE;
 
    pGuid->Data1 = 0;
    for (i = 0; i < 8; i++)
    {
        if (!iswxdigit(pszString[i]))
            return FALSE;
 
        pGuid->Data1 = (pGuid->Data1 << 4) | CharToByte(pszString[i]);
    }
 
    if (pszString[8] != L'-')
        return FALSE;
 
    pGuid->Data2 = 0;
    for (i = 9; i < 13; i++)
    {
        if (!iswxdigit(pszString[i]))
            return FALSE;
 
        pGuid->Data2 = (pGuid->Data2 << 4) | CharToByte(pszString[i]);
    }
 
    if (pszString[13] != L'-')
        return FALSE;
 
    pGuid->Data3 = 0;
    for (i = 14; i < 18; i++)
    {
        if (!iswxdigit(pszString[i]))
            return FALSE;
 
        pGuid->Data3 = (pGuid->Data3 << 4) | CharToByte(pszString[i]);
    }
 
    if (pszString[18] != L'-')
        return FALSE;
 
    for (i = 19; i < 36; i += 2)
    {
        if (i == 23)
        {
            if (pszString[i] != L'-')
                return FALSE;
            i++;
        }
 
        if (!iswxdigit(pszString[i]) || !iswxdigit(pszString[i + 1]))
            return FALSE;
 
        pGuid->Data4[(i - 19) / 2] = CharToByte(pszString[i]) << 4 | CharToByte(pszString[i + 1]);
    }
 
    if (pszString[36] == L'\0')
        return TRUE;
 
    return FALSE;
}

Referenced by CreatePrimaryGptPartition(), setid_main(), and UniqueIdDisk().