/*++

Copyright (c) 1991  Microsoft Corporation

Module Name:

    cmparse2.c

Abstract:

    This module contains parse routines for the configuration manager, particularly
    the registry.

Author:

    Bryan M. Willman (bryanwi) 10-Sep-1991

Revision History:

--*/

#include    "cmp.h"

#ifdef ALLOC_PRAGMA
#pragma alloc_text(PAGE,CmpDoCreate)
#pragma alloc_text(PAGE,CmpDoCreateChild)
#endif

extern  PCMHIVE CmpMasterHive;
extern  BOOLEAN CmpNoMasterCreates;
extern  PCM_KEY_CONTROL_BLOCK CmpKeyControlBlockRoot;


NTSTATUS
CmpDoCreate(
    IN PHHIVE Hive,
    IN HCELL_INDEX Cell,
    IN PACCESS_STATE AccessState,
    IN PUNICODE_STRING Name,
    IN KPROCESSOR_MODE AccessMode,
    IN PCM_PARSE_CONTEXT Context,
    IN PUNICODE_STRING BaseName,
    IN PUNICODE_STRING KeyName,
    OUT PVOID *Object
    )
/*++

Routine Description:

    Performs the first step in the creation of a registry key.  This
    routine checks to make sure the caller has the proper access to
    create a key here, and allocates space for the child in the parent
    cell.  It then calls CmpDoCreateChild to initialize the key and
    create the key object.

    This two phase creation allows us to share the child creation code
    with the creation of link nodes.

Arguments:

    Hive - supplies a pointer to the hive control structure for the hive

    Cell - supplies index of node to create child under.

    AccessState - Running security access state information for operation.

    Name - supplies pointer to a UNICODE string which is the name of
            the child to be created.

    AccessMode - Access mode of the original caller.

    Context - pointer to CM_PARSE_CONTEXT structure passed through
                the object manager

    BaseName - Name of object create is relative to

    KeyName - Relative name (to BaseName)

    Object - The address of a variable to receive the created key object, if
             any.

Return Value:

    NTSTATUS

--*/
{
    NTSTATUS status;
    PCELL_DATA pparent;
    PCELL_DATA pdata;
    HCELL_INDEX KeyCell;
    ULONG       ParentType;
    ACCESS_MASK AdditionalAccess;
    BOOLEAN CreateAccess;
    UNICODE_STRING FullName;

    CMLOG(CML_FLOW, CMS_PARSE) {
        KdPrint(("CmpDoCreate:\n"));
    }

    if (ARGUMENT_PRESENT(Context)) {

        if (Context->CreateOptions & REG_OPTION_BACKUP_RESTORE) {

            //
            // we're supposed to be opening for a backup/restore
            // operation, but this is a new create, which makes
            // no sense, so fail.
            //
            return STATUS_OBJECT_NAME_NOT_FOUND;
        }

        //
        // Operation is a create, so set Disposition
        //
        Context->Disposition = REG_CREATED_NEW_KEY;
    }

    //
    // Check to make sure the caller can create a sub-key here.
    //
    pparent = HvGetCell(Hive, Cell);
    pdata = HvGetCell(Hive, pparent->u.KeyNode.u1.s1.Security);
    ParentType = HvGetCellType(Cell);

    if ( (ParentType == Volatile) &&
         ((Context->CreateOptions & REG_OPTION_VOLATILE) == 0) )
    {
        //
        // Trying to create stable child under volatile parent, report error
        //
        return STATUS_CHILD_MUST_BE_VOLATILE;
    }

    if (pparent->u.KeyNode.Flags &   KEY_SYM_LINK)
    {
        //
        // Disallow attempts to create anything under a symbolic link
        //
        return STATUS_ACCESS_DENIED;
    }

    AdditionalAccess = (Context->CreateOptions & REG_OPTION_CREATE_LINK) ? KEY_CREATE_LINK : 0;

    FullName.MaximumLength = BaseName->Length + KeyName->Length + sizeof(WCHAR);

    FullName.Buffer = ExAllocatePool(PagedPool,FullName.MaximumLength);
    if (FullName.Buffer == NULL) {
        return(STATUS_INSUFFICIENT_RESOURCES);
    }

    RtlCopyUnicodeString(&FullName,BaseName);
    FullName.Buffer[FullName.Length/sizeof(WCHAR)] = OBJ_NAME_PATH_SEPARATOR;
    FullName.Length += sizeof(WCHAR);
    RtlAppendUnicodeStringToString(&FullName,KeyName);

    CreateAccess = CmpCheckCreateAccess(&FullName,
                                        &(pdata->u.KeySecurity.Descriptor),
                                        AccessState,
                                        AccessMode,
                                        AdditionalAccess,
                                        &status);
    ExFreePool(FullName.Buffer);

    if (CreateAccess) {

        //
        // Security check passed, so we can go ahead and create
        // the sub-key.
        //
        if (! HvMarkCellDirty(Hive, Cell)) {
            return STATUS_NO_LOG_SPACE;
        }

        //
        // Create and initialize the new sub-key
        //
        status = CmpDoCreateChild( Hive,
                                   Cell,
                                   &(pdata->u.KeySecurity.Descriptor),
                                   AccessState,
                                   Name,
                                   AccessMode,
                                   Context,
                                   BaseName,
                                   KeyName,
                                   0,
                                   &KeyCell,
                                   Object );

        if (NT_SUCCESS(status)) {

            //
            // Child successfully created, add to parent's list.
            //
            if (! CmpAddSubKey(Hive, Cell, KeyCell)) {

                //
                // Unable to add child, so free it
                //
                CmpFreeKeyByCell(Hive, KeyCell, FALSE);
                return STATUS_INSUFFICIENT_RESOURCES;
            }

            //
            // Update max keyname and class name length fields
            //
            if (pparent->u.KeyNode.MaxNameLen < KeyName->Length) {
                pparent->u.KeyNode.MaxNameLen = KeyName->Length;
            }

            if (pparent->u.KeyNode.MaxClassLen < Context->Class.Length) {
                pparent->u.KeyNode.MaxClassLen = Context->Class.Length;
            }

            if (Context->CreateOptions & REG_OPTION_CREATE_LINK) {
                pdata = HvGetCell(Hive, KeyCell);
                pdata->u.KeyNode.Flags |= KEY_SYM_LINK;
            }
        }
    }
    return status;
}


NTSTATUS
CmpDoCreateChild(
    IN PHHIVE Hive,
    IN HCELL_INDEX ParentCell,
    IN PSECURITY_DESCRIPTOR ParentDescriptor OPTIONAL,
    IN PACCESS_STATE AccessState,
    IN PUNICODE_STRING Name,
    IN KPROCESSOR_MODE AccessMode,
    IN PCM_PARSE_CONTEXT Context,
    IN PUNICODE_STRING BaseName,
    IN PUNICODE_STRING KeyName,
    IN USHORT Flags,
    OUT PHCELL_INDEX KeyCell,
    OUT PVOID *Object
    )

/*++

Routine Description:

    Creates a new sub-key.  This is called by CmpDoCreate to create child
    sub-keys and CmpCreateLinkNode to create root sub-keys.

Arguments:

    Hive - supplies a pointer to the hive control structure for the hive

    ParentCell - supplies cell index of parent cell

    ParentDescriptor - Supplies security descriptor of parent key, for use
           in inheriting ACLs.

    AccessState - Running security access state information for operation.

    Name - Supplies pointer to a UNICODE string which is the name of the
           child to be created.

    AccessMode - Access mode of the original caller.

    Context - Supplies pointer to CM_PARSE_CONTEXT structure passed through
           the object manager.

    BaseName - Name of object create is relative to

    KeyName - Relative name (to BaseName)

    Flags - Supplies any flags to be set in the newly created node

    KeyCell - Receives the cell index of the newly created sub-key, if any.

    Object - Receives a pointer to the created key object, if any.

Return Value:

    STATUS_SUCCESS - sub-key successfully created.  New object is returned in
            Object, and the new cell's cell index is returned in KeyCell.

    !STATUS_SUCCESS - appropriate error message.

--*/

{
    ULONG clean=0;
    ULONG alloc=0;
    NTSTATUS Status;
    PCM_KEY_BODY KeyBody;
    HCELL_INDEX ClassCell=HCELL_NIL;
    PCM_KEY_NODE KeyNode;
    PCELL_DATA CellData;
    PCM_KEY_CONTROL_BLOCK kcb;
    PCM_KEY_CONTROL_BLOCK fkcb;
    LONG found;
    ULONG StorageType;
    PSECURITY_DESCRIPTOR NewDescriptor = NULL;
    LARGE_INTEGER systemtime;


    CMLOG(CML_FLOW, CMS_PARSE) {
        KdPrint(("CmpDoCreateChild:\n"));
    }
    try {
        //
        // Get allocation type
        //
        StorageType = Stable;
        if (Context->CreateOptions & REG_OPTION_VOLATILE) {
            StorageType = Volatile;
        }

        //
        // Allocate child cell
        //
        *KeyCell = HvAllocateCell(
                        Hive,
                        CmpHKeyNodeSize(Hive, Name),
                        StorageType
                        );
        if (*KeyCell == HCELL_NIL) {
            return STATUS_INSUFFICIENT_RESOURCES;
        }
        alloc = 1;
        KeyNode = (PCM_KEY_NODE)HvGetCell(Hive, *KeyCell);

        //
        // Allocate cell for class name
        //
        if (Context->Class.Length > 0) {
            ClassCell = HvAllocateCell(Hive, Context->Class.Length, StorageType);
            if (ClassCell == HCELL_NIL) {
                return STATUS_INSUFFICIENT_RESOURCES;
            }
        }
        alloc = 2;

        //
        // Allocate a key control block
        //
        kcb = CmpCreateKeyControlBlock(Hive, *KeyCell, KeyNode, BaseName, KeyName);
        if (kcb == NULL) {
            return STATUS_INSUFFICIENT_RESOURCES;
        }
        ASSERT(kcb->RefCount == 1);
        alloc = 3;

        //
        // Check to see if parent is marked for delete, we do this
        // by computing parent's name, and looking for its kcb.  If no kcb,
        // no way parent can be marked for delete.  If there is a kcb,
        // it will tell us outright.
        //
        if (ParentCell != HCELL_NIL) {
            if (Flags & KEY_HIVE_ENTRY) {
                found = CmpFindKeyControlBlock(CmpKeyControlBlockRoot,
                                               &CmpMasterHive->Hive,
                                               ParentCell,
                                               &fkcb);
            } else {
                found = CmpFindKeyControlBlock(CmpKeyControlBlockRoot,
                                               Hive,
                                               ParentCell,
                                               &fkcb);

            }
            if ((found == 0) && (fkcb != kcb)) {
                ASSERT(fkcb->KeyCell == ParentCell);
                if (fkcb->Delete == TRUE) {
                    return STATUS_KEY_DELETED;
                }
            }
        }

        //
        // Allocate the object manager object
        //
        Status = ObCreateObject(
                    AccessMode,
                    CmpKeyObjectType,
                    NULL,
                    UserMode,
                    NULL,
                    sizeof(CM_KEY_BODY),
                    0,
                    0,
                    Object
                    );

        if (NT_SUCCESS(Status)) {

            KeyBody = (PCM_KEY_BODY)(*Object);

            //
            // We have managed to allocate all of the objects we need to,
            // so initialize them
            //

            //
            // Fill in the class name
            //
            if (Context->Class.Length > 0) {

                CellData = HvGetCell(Hive, ClassCell);

                try {

                    RtlMoveMemory(
                        &(CellData->u.KeyString[0]),
                        Context->Class.Buffer,
                        Context->Class.Length
                        );

                } except(EXCEPTION_EXECUTE_HANDLER) {
                    return GetExceptionCode();
                }
            }

            //
            // Fill in the new key itself
            //
            KeyNode->Signature = CM_KEY_NODE_SIGNATURE;
            KeyNode->Flags = Flags;

            KeQuerySystemTime(&systemtime);
            KeyNode->LastWriteTime = systemtime;

            KeyNode->Spare = 0;
            KeyNode->Parent = ParentCell;
            KeyNode->SubKeyCounts[Stable] = 0;
            KeyNode->SubKeyCounts[Volatile] = 0;
            KeyNode->SubKeyLists[Stable] = HCELL_NIL;
            KeyNode->SubKeyLists[Volatile] = HCELL_NIL;
            KeyNode->ValueList.Count = 0;
            KeyNode->ValueList.List = HCELL_NIL;
            KeyNode->u1.s1.Security = HCELL_NIL;
            KeyNode->u1.s1.Class = ClassCell;
            KeyNode->ClassLength = Context->Class.Length;

            KeyNode->MaxValueDataLen = 0;
            KeyNode->MaxNameLen = 0;
            KeyNode->MaxValueNameLen = 0;
            KeyNode->MaxClassLen = 0;

            KeyNode->NameLength = CmpCopyName(Hive,
                                              KeyNode->Name,
                                              Name);
            if (KeyNode->NameLength < Name->Length) {
                KeyNode->Flags |= KEY_COMP_NAME;
            }

            if (Context->CreateOptions & REG_OPTION_PREDEF_HANDLE) {
                KeyNode->ValueList.Count = (ULONG)Context->PredefinedHandle;
                KeyNode->Flags |= KEY_PREDEF_HANDLE;
            }

            //
            // Fill in CM specific fields in the object
            //
            KeyBody->Type = KEY_BODY_TYPE;
            KeyBody->KeyControlBlock = kcb;
            KeyBody->NotifyBlock = NULL;

            //
            // Assign a security descriptor to the object.  Note that since
            // registry keys are container objects, and ObAssignSecurity
            // assumes that the only container object in the world is
            // the ObpDirectoryObjectType, we have to call SeAssignSecurity
            // directly in order to get the right inheritance.
            //

            Status = SeAssignSecurity(ParentDescriptor,
                                      AccessState->SecurityDescriptor,
                                      &NewDescriptor,
                                      TRUE,             // container object
                                      &AccessState->SubjectSecurityContext,
                                      &CmpKeyObjectType->TypeInfo.GenericMapping,
                                      CmpKeyObjectType->TypeInfo.PoolType);
            if (NT_SUCCESS(Status)) {
                Status = CmpSecurityMethod(*Object,
                                           AssignSecurityDescriptor,
                                           NULL,
                                           NewDescriptor,
                                           NULL,
                                           NULL,
                                           CmpKeyObjectType->TypeInfo.PoolType,
                                           &CmpKeyObjectType->TypeInfo.GenericMapping);
            }

            //
            // Since the security descriptor now lives in the hive,
            // free the in-memory copy
            //
            SeDeassignSecurity( &NewDescriptor );

            if (!NT_SUCCESS(Status)) {

                //
                // Note that the dereference will clean up the kcb, so
                // make sure and decrement the allocation count here.
                //
                ObDereferenceObject(*Object);
                alloc = 2;

            } else {
                CmpReportNotify(
                    kcb->FullName,
                    kcb->KeyHive,
                    kcb->KeyCell,
                    REG_NOTIFY_CHANGE_NAME
                    );
            }
        }

    } finally {

        if (!NT_SUCCESS(Status)) {

            //
            // Clean up allocations
            //
            switch (alloc) {
            case 3:
                CmpDereferenceKeyControlBlock(kcb);
                // DELIBERATE FALL

            case 2:
                if (Context->Class.Length > 0) {
                    HvFreeCell(Hive, ClassCell);
                }
                // DELIBERATE FALL

            case 1:
                HvFreeCell(Hive, *KeyCell);
                // DELIBERATE FALL
            }
        }
    }
    return(Status);
}