Compartilhar via


Usando um identificador para um objeto Registry-Key

A tabela a seguir lista as operações que os drivers podem executar em uma chave aberta, bem como as rotinas apropriadas para chamar.

Operação Rotina para chamar

Examine as propriedades da chave, como seu nome ou o número de suas subchaves.

ZwQueryKey

Itere por meio das subchaves da chave, examinando as propriedades de cada uma delas.

ZwEnumerateKey

Examine as propriedades de um valor de chave, incluindo os dados do valor.

ZwQueryValueKey

Itere por meio dos valores de uma chave, examinando as propriedades de cada uma delas.

ZwEnumerateValueKey

Defina os dados para um valor associado a uma chave.

ZwSetValueKey

Excluir uma chave.

ZwDeleteKey

Exclua um valor de chave.

ZwDeleteValueKey

Depois que o driver tiver terminado suas manipulações, ele deverá chamar ZwClose para fechar o identificador, mesmo que ele já tenha chamado ZwDeleteKey para excluir a chave. (Depois que uma chave é excluída, todas as alças abertas para ela se tornam inválidas, no entanto, o driver ainda deve fechar o identificador.)

O exemplo de código a seguir ilustra como abrir um identificador para uma chave chamada \Registry\Machine\Software\MyCompany\MyApp e, em seguida, recuperar dados de chave e fechar o identificador.

//
// Get the frame location from the registry key
// HKLM\SOFTWARE\MyCompany\MyApp.
// For example: "FrameLocation"="X:\\MyApp\\Frames"
// 
HANDLE              handleRegKey = NULL;
for (int n = 0; n < 1; n++) 
{
    NTSTATUS           status = NULL;
    UNICODE_STRING     RegistryKeyName;
    OBJECT_ATTRIBUTES  ObjectAttributes;

    RtlInitUnicodeString(&RegistryKeyName, L"\\Registry\\Machine\\Software\\MyCompany\\MyApp");
    InitializeObjectAttributes(&ObjectAttributes, 
                               &RegistryKeyName,
                               OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
                               NULL,    // handle
                               NULL);
    status = ZwOpenKey(&handleRegKey, KEY_READ, &ObjectAttributes);

    // If the driver cannot open the key, the driver cannot continue. 
    // In this situation, the driver was probably set up incorrectly 
    // and worst case, the driver cannot stream.
    if( NT_SUCCESS(status) == FALSE ) 
    {
        break;
    }
    // The driver obtained the registry key.
    PKEY_VALUE_FULL_INFORMATION  pKeyInfo = NULL;
    UNICODE_STRING               ValueName;
    ULONG                        ulKeyInfoSize = 0;
    ULONG                        ulKeyInfoSizeNeeded = 0;

    // The driver requires the following value.
    RtlInitUnicodeString(&ValueName, L"FrameLocation");

    // Determine the required size of keyInfo.
    status = ZwQueryValueKey( handleRegKey,
                              &ValueName,
                              KeyValueFullInformation,
                              pKeyInfo,
                              ulKeyInfoSize,
                              &ulKeyInfoSizeNeeded );

    // The driver expects one of the following errors.
    if( (status == STATUS_BUFFER_TOO_SMALL) || (status == STATUS_BUFFER_OVERFLOW) )
    {
        // Allocate the memory required for the key.
        ulKeyInfoSize = ulKeyInfoSizeNeeded;
        pKeyInfo = (PKEY_VALUE_FULL_INFORMATION) ExAllocatePoolWithTag( NonPagedPool, ulKeyInfoSizeNeeded, g_ulTag);
        if( NULL == pKeyInfo )
        {
            break;
        }
        RtlZeroMemory( pKeyInfo, ulKeyInfoSize );

        // Get the key data.
        status = ZwQueryValueKey( handleRegKey,
                                  &ValueName,
                                  KeyValueFullInformation,
                                  pKeyInfo,
                                  ulKeyInfoSize,
                                  &ulKeyInfoSizeNeeded );
        if( (status != STATUS_SUCCESS) || (ulKeyInfoSizeNeeded != ulKeyInfoSize) || (NULL == pKeyInfo) )
        {
            break;
        }

        // Fill in the frame location if it has not been filled in already.
        if ( NULL == m_szwFramePath )
        {
            m_ulFramePathLength = pKeyInfo->DataLength;
            ULONG_PTR   pSrc = NULL;

            pSrc = (ULONG_PTR) ( (PBYTE) pKeyInfo + pKeyInfo->DataOffset);

            m_szwFramePath = (LPWSTR) ExAllocatePoolWithTag( NonPagedPool, m_ulFramePathLength, g_ulTag);
            if ( NULL == m_szwFramePath )
            {
                m_ulFramePathLength = 0;
                break;
            }

            // Copy the frame path.
            RtlCopyMemory(m_szwFramePath, (PVOID) pSrc, m_ulFramePathLength);
        }
        // The driver is done with the pKeyInfo.
        xFreePoolWithTag(pKeyInfo, g_ulTag);

    } // if( (status == STATUS_BUFFER_TOO_SMALL) || (status == STATUS_BUFFER_OVERFLOW) )
} // Get the Frame location from the registry key.

// All done with the registry.
if (NULL != handleRegKey)
{
    ZwClose(handleRegKey);
}

O sistema armazena em cache as principais alterações na memória e as grava no disco a cada poucos segundos. Para forçar uma alteração de chave no disco, chame ZwFlushKey.

Para manipular o registro por meio de uma interface mais simples, os drivers também podem chamar as rotinas xxx do Registro RtlXxx. Para obter mais informações, consulte Rotinas de biblioteca de Run-Time do Registro.