[ros-diffs] [tkreuzer] 61050: [NTOSKRNL] Move session routines into session.c. No code change.

Author: tkreuzer
Date: Tue Nov 19 21:13:31 2013
New Revision: 61050

URL: http://svn.reactos.org/svn/reactos?rev=61050&view=rev
Log:
[NTOSKRNL]
Move session routines into session.c. No code change.

Added:
    trunk/reactos/ntoskrnl/mm/ARM3/session.c   (with props)
Modified:
    trunk/reactos/ntoskrnl/CMakeLists.txt
    trunk/reactos/ntoskrnl/mm/ARM3/procsup.c

Modified: trunk/reactos/ntoskrnl/CMakeLists.txt
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/CMakeLists.txt?re…
==============================================================================
--- trunk/reactos/ntoskrnl/CMakeLists.txt	[iso-8859-1] (original)
+++ trunk/reactos/ntoskrnl/CMakeLists.txt	[iso-8859-1] Tue Nov 19 21:13:31 2013
@@ -216,6 +216,7 @@
     mm/ARM3/pool.c
     mm/ARM3/procsup.c
     mm/ARM3/section.c
+    mm/ARM3/session.c
     mm/ARM3/special.c
     mm/ARM3/sysldr.c
     mm/ARM3/syspte.c

Modified: trunk/reactos/ntoskrnl/mm/ARM3/procsup.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/mm/ARM3/procsup.c…
==============================================================================
--- trunk/reactos/ntoskrnl/mm/ARM3/procsup.c	[iso-8859-1] (original)
+++ trunk/reactos/ntoskrnl/mm/ARM3/procsup.c	[iso-8859-1] Tue Nov 19 21:13:31 2013
@@ -554,41 +554,6 @@
     // Return the old priority
     //
     return OldPriority;
-}
-
-LCID
-NTAPI
-MmGetSessionLocaleId(VOID)
-{
-    PEPROCESS Process;
-    PAGED_CODE();
-
-    //
-    // Get the current process
-    //
-    Process = PsGetCurrentProcess();
-
-    //
-    // Check if it's the Session Leader
-    //
-    if (Process->Vm.Flags.SessionLeader)
-    {
-        //
-        // Make sure it has a valid Session
-        //
-        if (Process->Session)
-        {
-            //
-            // Get the Locale ID
-            //
-            return ((PMM_SESSION_SPACE)Process->Session)->LocaleId;
-        }
-    }
-
-    //
-    // Not a session leader, return the default
-    //
-    return PsDefaultThreadLocaleId;
 }
 
 NTSTATUS
@@ -1473,653 +1438,6 @@
     Process->Pcb.DirectoryTableBase[1] = 0;
 }
 
-/* SESSION CODE TO MOVE TO SESSION.C ******************************************/
-
-PMM_SESSION_SPACE MmSessionSpace;
-PFN_NUMBER MiSessionDataPages, MiSessionTagPages, MiSessionTagSizePages;
-PFN_NUMBER MiSessionBigPoolPages, MiSessionCreateCharge;
-KGUARDED_MUTEX MiSessionIdMutex;
-LONG MmSessionDataPages;
-PRTL_BITMAP MiSessionIdBitmap;
-volatile LONG MiSessionLeaderExists;
-
-VOID
-NTAPI
-MiInitializeSessionIds(VOID)
-{
-    ULONG Size, BitmapSize;
-    PFN_NUMBER TotalPages;
-
-    /* Setup the total number of data pages needed for the structure */
-    TotalPages = MI_SESSION_DATA_PAGES_MAXIMUM;
-    MiSessionDataPages = ROUND_TO_PAGES(sizeof(MM_SESSION_SPACE)) >> PAGE_SHIFT;
-    ASSERT(MiSessionDataPages <= MI_SESSION_DATA_PAGES_MAXIMUM - 3);
-    TotalPages -= MiSessionDataPages;
-
-    /* Setup the number of pages needed for session pool tags */
-    MiSessionTagSizePages = 2;
-    MiSessionBigPoolPages = 1;
-    MiSessionTagPages = MiSessionTagSizePages + MiSessionBigPoolPages;
-    ASSERT(MiSessionTagPages <= TotalPages);
-    ASSERT(MiSessionTagPages < MI_SESSION_TAG_PAGES_MAXIMUM);
-
-    /* Total pages needed for a session (FIXME: Probably different on PAE/x64) */
-    MiSessionCreateCharge = 1 + MiSessionDataPages + MiSessionTagPages;
-
-    /* Initialize the lock */
-    KeInitializeGuardedMutex(&MiSessionIdMutex);
-
-    /* Allocate the bitmap */
-    Size = MI_INITIAL_SESSION_IDS;
-    BitmapSize = ((Size + 31) / 32) * sizeof(ULONG);
-    MiSessionIdBitmap = ExAllocatePoolWithTag(PagedPool,
-                                              sizeof(RTL_BITMAP) + BitmapSize,
-                                              '  mM');
-    if (MiSessionIdBitmap)
-    {
-        /* Free all the bits */
-        RtlInitializeBitMap(MiSessionIdBitmap,
-                            (PVOID)(MiSessionIdBitmap + 1),
-                            Size);
-        RtlClearAllBits(MiSessionIdBitmap);
-    }
-    else
-    {
-        /* Die if we couldn't allocate the bitmap */
-        KeBugCheckEx(INSTALL_MORE_MEMORY,
-                     MmNumberOfPhysicalPages,
-                     MmLowestPhysicalPage,
-                     MmHighestPhysicalPage,
-                     0x200);
-    }
-}
-
-VOID
-NTAPI
-MiSessionLeader(IN PEPROCESS Process)
-{
-    KIRQL OldIrql;
-
-    /* Set the flag while under the expansion lock */
-    OldIrql = KeAcquireQueuedSpinLock(LockQueueExpansionLock);
-    Process->Vm.Flags.SessionLeader = TRUE;
-    KeReleaseQueuedSpinLock(LockQueueExpansionLock, OldIrql);
-}
-
-ULONG
-NTAPI
-MmGetSessionId(IN PEPROCESS Process)
-{
-    PMM_SESSION_SPACE SessionGlobal;
-
-    /* The session leader is always session zero */
-    if (Process->Vm.Flags.SessionLeader == 1) return 0;
-
-    /* Otherwise, get the session global, and read the session ID from it */
-    SessionGlobal = (PMM_SESSION_SPACE)Process->Session;
-    if (!SessionGlobal) return 0;
-    return SessionGlobal->SessionId;
-}
-
-ULONG
-NTAPI
-MmGetSessionIdEx(IN PEPROCESS Process)
-{
-    PMM_SESSION_SPACE SessionGlobal;
-
-    /* The session leader is always session zero */
-    if (Process->Vm.Flags.SessionLeader == 1) return 0;
-
-    /* Otherwise, get the session global, and read the session ID from it */
-    SessionGlobal = (PMM_SESSION_SPACE)Process->Session;
-    if (!SessionGlobal) return -1;
-    return SessionGlobal->SessionId;
-}
-
-VOID
-NTAPI
-MiReleaseProcessReferenceToSessionDataPage(IN PMM_SESSION_SPACE SessionGlobal)
-{
-    ULONG i, SessionId;
-    PMMPTE PointerPte;
-    PFN_NUMBER PageFrameIndex[MI_SESSION_DATA_PAGES_MAXIMUM];
-    PMMPFN Pfn1;
-    KIRQL OldIrql;
-
-    /* Is there more than just this reference? If so, bail out */
-    if (InterlockedDecrement(&SessionGlobal->ProcessReferenceToSession)) return;
-
-    /* Get the session ID */
-    SessionId = SessionGlobal->SessionId;
-    DPRINT1("Last process in session %lu going down!!!\n", SessionId);
-
-    /* Free the session page tables */
-#ifndef _M_AMD64
-    ExFreePoolWithTag(SessionGlobal->PageTables, 'tHmM');
-#endif
-    ASSERT(!MI_IS_PHYSICAL_ADDRESS(SessionGlobal));
-
-    /* Capture the data page PFNs */
-    PointerPte = MiAddressToPte(SessionGlobal);
-    for (i = 0; i < MiSessionDataPages; i++)
-    {
-        PageFrameIndex[i] = PFN_FROM_PTE(PointerPte + i);
-    }
-
-    /* Release them */
-    MiReleaseSystemPtes(PointerPte, MiSessionDataPages, SystemPteSpace);
-
-    /* Mark them as deleted */
-    for (i = 0; i < MiSessionDataPages; i++)
-    {
-        Pfn1 = MI_PFN_ELEMENT(PageFrameIndex[i]);
-        MI_SET_PFN_DELETED(Pfn1);
-    }
-
-    /* Loop every data page and drop a reference count */
-    OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
-    for (i = 0; i < MiSessionDataPages; i++)
-    {
-        /* Sanity check that the page is correct, then decrement it */
-        Pfn1 = MI_PFN_ELEMENT(PageFrameIndex[i]);
-        ASSERT(Pfn1->u2.ShareCount == 1);
-        ASSERT(Pfn1->u3.e2.ReferenceCount == 1);
-        MiDecrementShareCount(Pfn1, PageFrameIndex[i]);
-    }
-
-    /* Done playing with pages, release the lock */
-    KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
-
-    /* Decrement the number of data pages */
-    InterlockedDecrement(&MmSessionDataPages);
-
-    /* Free this session ID from the session bitmap */
-    KeAcquireGuardedMutex(&MiSessionIdMutex);
-    ASSERT(RtlCheckBit(MiSessionIdBitmap, SessionId));
-    RtlClearBit(MiSessionIdBitmap, SessionId);
-    KeReleaseGuardedMutex(&MiSessionIdMutex);
-}
-
-VOID
-NTAPI
-MiSessionRemoveProcess(VOID)
-{
-    PEPROCESS CurrentProcess = PsGetCurrentProcess();
-
-    /* If the process isn't already in a session, or if it's the leader... */
-    if (!(CurrentProcess->Flags & PSF_PROCESS_IN_SESSION_BIT) ||
-        (CurrentProcess->Vm.Flags.SessionLeader))
-    {
-        /* Then there's nothing to do */
-        return;
-    }
-
-    /* Sanity check */
-    ASSERT(MmIsAddressValid(MmSessionSpace) == TRUE);
-
-    /* Remove the process from the list ,and dereference the session */
-    // DO NOT ENABLE THIS UNLESS YOU FIXED THE NP POOL CORRUPTION THAT IT CAUSES!!!
-    //RemoveEntryList(&CurrentProcess->SessionProcessLinks);
-    //MiDereferenceSession();
-}
-
-VOID
-NTAPI
-MiSessionAddProcess(IN PEPROCESS NewProcess)
-{
-    PMM_SESSION_SPACE SessionGlobal;
-
-    /* The current process must already be in a session */
-    if (!(PsGetCurrentProcess()->Flags & PSF_PROCESS_IN_SESSION_BIT)) return;
-
-    /* Sanity check */
-    ASSERT(MmIsAddressValid(MmSessionSpace) == TRUE);
-
-    /* Get the global session */
-    SessionGlobal = MmSessionSpace->GlobalVirtualAddress;
-
-    /* Increment counters */
-    InterlockedIncrement((PLONG)&SessionGlobal->ReferenceCount);
-    InterlockedIncrement(&SessionGlobal->ResidentProcessCount);
-    InterlockedIncrement(&SessionGlobal->ProcessReferenceToSession);
-
-    /* Set the session pointer */
-    ASSERT(NewProcess->Session == NULL);
-    NewProcess->Session = SessionGlobal;
-
-    /* Insert it into the process list */
-    // DO NOT ENABLE THIS UNLESS YOU FIXED THE NP POOL CORRUPTION THAT IT CAUSES!!!
-    //InsertTailList(&SessionGlobal->ProcessList,
&NewProcess->SessionProcessLinks);
-
-    /* Set the flag */
-    PspSetProcessFlag(NewProcess, PSF_PROCESS_IN_SESSION_BIT);
-}
-
-NTSTATUS
-NTAPI
-MiSessionInitializeWorkingSetList(VOID)
-{
-    KIRQL OldIrql;
-    PMMPTE PointerPte, PointerPde;
-    MMPTE TempPte;
-    ULONG Color, Index;
-    PFN_NUMBER PageFrameIndex;
-    PMM_SESSION_SPACE SessionGlobal;
-    BOOLEAN AllocatedPageTable;
-    PMMWSL WorkingSetList;
-
-    /* Get pointers to session global and the session working set list */
-    SessionGlobal = MmSessionSpace->GlobalVirtualAddress;
-    WorkingSetList = (PMMWSL)MiSessionSpaceWs;
-
-    /* Fill out the two pointers */
-    MmSessionSpace->Vm.VmWorkingSetList = WorkingSetList;
-    MmSessionSpace->Wsle = (PMMWSLE)WorkingSetList->UsedPageTableEntries;
-
-    /* Get the PDE for the working set, and check if it's already allocated */
-    PointerPde = MiAddressToPde(WorkingSetList);
-    if (PointerPde->u.Hard.Valid == 1)
-    {
-        /* Nope, we'll have to do it */
-        ASSERT(PointerPde->u.Hard.Global == 0);
-        AllocatedPageTable = FALSE;
-    }
-    else
-    {
-        /* Yep, that makes our job easier */
-        AllocatedPageTable = TRUE;
-    }
-
-    /* Get the PTE for the working set */
-    PointerPte = MiAddressToPte(WorkingSetList);
-
-    /* Initialize the working set lock, and lock the PFN database */
-    ExInitializePushLock(&SessionGlobal->Vm.WorkingSetMutex);
-    //MmLockPageableSectionByHandle(ExPageLockHandle);
-    OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
-
-    /* Check if we need a page table */
-    if (AllocatedPageTable == TRUE)
-    {
-        /* Get a zeroed colored zero page */
-        Color = MI_GET_NEXT_COLOR();
-        PageFrameIndex = MiRemoveZeroPageSafe(Color);
-        if (!PageFrameIndex)
-        {
-            /* No zero pages, grab a free one */
-            PageFrameIndex = MiRemoveAnyPage(Color);
-
-            /* Zero it outside the PFN lock */
-            KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
-            MiZeroPhysicalPage(PageFrameIndex);
-            OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
-        }
-
-        /* Write a valid PDE for it */
-        TempPte.u.Long = ValidKernelPdeLocal.u.Long;
-        TempPte.u.Hard.PageFrameNumber = PageFrameIndex;
-        MI_WRITE_VALID_PTE(PointerPde, TempPte);
-
-        /* Add this into the list */
-        Index = ((ULONG_PTR)WorkingSetList - (ULONG_PTR)MmSessionBase) >> 22;
-#ifndef _M_AMD64
-        MmSessionSpace->PageTables[Index] = TempPte;
-#endif
-        /* Initialize the page directory page, and now zero the working set list itself
*/
-        MiInitializePfnForOtherProcess(PageFrameIndex,
-                                       PointerPde,
-                                       MmSessionSpace->SessionPageDirectoryIndex);
-        KeZeroPages(PointerPte, PAGE_SIZE);
-    }
-
-    /* Get a zeroed colored zero page */
-    Color = MI_GET_NEXT_COLOR();
-    PageFrameIndex = MiRemoveZeroPageSafe(Color);
-    if (!PageFrameIndex)
-    {
-        /* No zero pages, grab a free one */
-        PageFrameIndex = MiRemoveAnyPage(Color);
-
-        /* Zero it outside the PFN lock */
-        KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
-        MiZeroPhysicalPage(PageFrameIndex);
-        OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
-    }
-
-    /* Write a valid PTE for it */
-    TempPte.u.Long = ValidKernelPteLocal.u.Long;
-    TempPte.u.Hard.Dirty = TRUE;
-    TempPte.u.Hard.PageFrameNumber = PageFrameIndex;
-
-    /* Initialize the working set list page */
-    MiInitializePfnAndMakePteValid(PageFrameIndex, PointerPte, TempPte);
-
-    /* Now we can release the PFN database lock */
-    KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
-
-    /* Fill out the working set structure */
-    MmSessionSpace->Vm.Flags.SessionSpace = 1;
-    MmSessionSpace->Vm.MinimumWorkingSetSize = 20;
-    MmSessionSpace->Vm.MaximumWorkingSetSize = 384;
-    WorkingSetList->LastEntry = 20;
-    WorkingSetList->HashTable = NULL;
-    WorkingSetList->HashTableSize = 0;
-    WorkingSetList->Wsle = MmSessionSpace->Wsle;
-
-    /* FIXME: Handle list insertions */
-    ASSERT(SessionGlobal->WsListEntry.Flink == NULL);
-    ASSERT(SessionGlobal->WsListEntry.Blink == NULL);
-    ASSERT(SessionGlobal->Vm.WorkingSetExpansionLinks.Flink == NULL);
-    ASSERT(SessionGlobal->Vm.WorkingSetExpansionLinks.Blink == NULL);
-
-    /* All done, return */
-    //MmUnlockPageableImageSection(ExPageLockHandle);
-    return STATUS_SUCCESS;
-}
-
-NTSTATUS
-NTAPI
-MiSessionCreateInternal(OUT PULONG SessionId)
-{
-    PEPROCESS Process = PsGetCurrentProcess();
-    ULONG NewFlags, Flags, Size, i, Color;
-    KIRQL OldIrql;
-    PMMPTE PointerPte, PageTables, SessionPte;
-    PMMPDE PointerPde;
-    PMM_SESSION_SPACE SessionGlobal;
-    MMPTE TempPte;
-    NTSTATUS Status;
-    BOOLEAN Result;
-    PFN_NUMBER SessionPageDirIndex;
-    PFN_NUMBER TagPage[MI_SESSION_TAG_PAGES_MAXIMUM];
-    PFN_NUMBER DataPage[MI_SESSION_DATA_PAGES_MAXIMUM];
-
-    /* This should not exist yet */
-    ASSERT(MmIsAddressValid(MmSessionSpace) == FALSE);
-
-    /* Loop so we can set the session-is-creating flag */
-    Flags = Process->Flags;
-    while (TRUE)
-    {
-        /* Check if it's already set */
-        if (Flags & PSF_SESSION_CREATION_UNDERWAY_BIT)
-        {
-            /* Bail out */
-            DPRINT1("Lost session race\n");
-            return STATUS_ALREADY_COMMITTED;
-        }
-
-        /* Now try to set it */
-        NewFlags = InterlockedCompareExchange((PLONG)&Process->Flags,
-                                              Flags | PSF_SESSION_CREATION_UNDERWAY_BIT,
-                                              Flags);
-        if (NewFlags == Flags) break;
-
-        /* It changed, try again */
-        Flags = NewFlags;
-    }
-
-    /* Now we should own the flag */
-    ASSERT(Process->Flags & PSF_SESSION_CREATION_UNDERWAY_BIT);
-
-    /*
-     * Session space covers everything from 0xA0000000 to 0xC0000000.
-     * Allocate enough page tables to describe the entire region
-     */
-    Size = (0x20000000 / PDE_MAPPED_VA) * sizeof(MMPTE);
-    PageTables = ExAllocatePoolWithTag(NonPagedPool, Size, 'tHmM');
-    ASSERT(PageTables != NULL);
-    RtlZeroMemory(PageTables, Size);
-
-    /* Lock the session ID creation mutex */
-    KeAcquireGuardedMutex(&MiSessionIdMutex);
-
-    /* Allocate a new Session ID */
-    *SessionId = RtlFindClearBitsAndSet(MiSessionIdBitmap, 1, 0);
-    if (*SessionId == 0xFFFFFFFF)
-    {
-        /* We ran out of session IDs, we should expand */
-        DPRINT1("Too many sessions created. Expansion not yet supported\n");
-        ExFreePoolWithTag(PageTables, 'tHmM');
-        return STATUS_NO_MEMORY;
-    }
-
-    /* Unlock the session ID creation mutex */
-    KeReleaseGuardedMutex(&MiSessionIdMutex);
-
-    /* Reserve the global PTEs */
-    SessionPte = MiReserveSystemPtes(MiSessionDataPages, SystemPteSpace);
-    ASSERT(SessionPte != NULL);
-
-    /* Acquire the PFN lock while we set everything up */
-    OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
-
-    /* Loop the global PTEs */
-    TempPte.u.Long = ValidKernelPte.u.Long;
-    for (i = 0; i < MiSessionDataPages; i++)
-    {
-        /* Get a zeroed colored zero page */
-        Color = MI_GET_NEXT_COLOR();
-        DataPage[i] = MiRemoveZeroPageSafe(Color);
-        if (!DataPage[i])
-        {
-            /* No zero pages, grab a free one */
-            DataPage[i] = MiRemoveAnyPage(Color);
-
-            /* Zero it outside the PFN lock */
-            KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
-            MiZeroPhysicalPage(DataPage[i]);
-            OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
-        }
-
-        /* Fill the PTE out */
-        TempPte.u.Hard.PageFrameNumber = DataPage[i];
-        MI_WRITE_VALID_PTE(SessionPte + i, TempPte);
-    }
-
-    /* Set the pointer to global space */
-    SessionGlobal = MiPteToAddress(SessionPte);
-
-    /* Get a zeroed colored zero page */
-    Color = MI_GET_NEXT_COLOR();
-    SessionPageDirIndex = MiRemoveZeroPageSafe(Color);
-    if (!SessionPageDirIndex)
-    {
-        /* No zero pages, grab a free one */
-        SessionPageDirIndex = MiRemoveAnyPage(Color);
-
-        /* Zero it outside the PFN lock */
-        KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
-        MiZeroPhysicalPage(SessionPageDirIndex);
-        OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
-    }
-
-    /* Fill the PTE out */
-    TempPte.u.Long = ValidKernelPdeLocal.u.Long;
-    TempPte.u.Hard.PageFrameNumber = SessionPageDirIndex;
-
-    /* Setup, allocate, fill out the MmSessionSpace PTE */
-    PointerPde = MiAddressToPde(MmSessionSpace);
-    ASSERT(PointerPde->u.Long == 0);
-    MI_WRITE_VALID_PTE(PointerPde, TempPte);
-    MiInitializePfnForOtherProcess(SessionPageDirIndex,
-                                   PointerPde,
-                                   SessionPageDirIndex);
-    ASSERT(MI_PFN_ELEMENT(SessionPageDirIndex)->u1.WsIndex == 0);
-
-     /* Loop all the local PTEs for it */
-    TempPte.u.Long = ValidKernelPteLocal.u.Long;
-    PointerPte = MiAddressToPte(MmSessionSpace);
-    for (i = 0; i < MiSessionDataPages; i++)
-    {
-        /* And fill them out */
-        TempPte.u.Hard.PageFrameNumber = DataPage[i];
-        MiInitializePfnAndMakePteValid(DataPage[i], PointerPte + i, TempPte);
-        ASSERT(MI_PFN_ELEMENT(DataPage[i])->u1.WsIndex == 0);
-    }
-
-     /* Finally loop all of the session pool tag pages */
-    for (i = 0; i < MiSessionTagPages; i++)
-    {
-        /* Grab a zeroed colored page */
-        Color = MI_GET_NEXT_COLOR();
-        TagPage[i] = MiRemoveZeroPageSafe(Color);
-        if (!TagPage[i])
-        {
-            /* No zero pages, grab a free one */
-            TagPage[i] = MiRemoveAnyPage(Color);
-
-            /* Zero it outside the PFN lock */
-            KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
-            MiZeroPhysicalPage(TagPage[i]);
-            OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
-        }
-
-        /* Fill the PTE out */
-        TempPte.u.Hard.PageFrameNumber = TagPage[i];
-        MiInitializePfnAndMakePteValid(TagPage[i],
-                                       PointerPte + MiSessionDataPages + i,
-                                       TempPte);
-    }
-
-    /* PTEs have been setup, release the PFN lock */
-    KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
-
-    /* Fill out the session space structure now */
-    MmSessionSpace->GlobalVirtualAddress = SessionGlobal;
-    MmSessionSpace->ReferenceCount = 1;
-    MmSessionSpace->ResidentProcessCount = 1;
-    MmSessionSpace->u.LongFlags = 0;
-    MmSessionSpace->SessionId = *SessionId;
-    MmSessionSpace->LocaleId = PsDefaultSystemLocaleId;
-    MmSessionSpace->SessionPageDirectoryIndex = SessionPageDirIndex;
-    MmSessionSpace->Color = Color;
-    MmSessionSpace->NonPageablePages = MiSessionCreateCharge;
-    MmSessionSpace->CommittedPages = MiSessionCreateCharge;
-#ifndef _M_AMD64
-    MmSessionSpace->PageTables = PageTables;
-    MmSessionSpace->PageTables[PointerPde - MiAddressToPde(MmSessionBase)] =
*PointerPde;
-#endif
-    InitializeListHead(&MmSessionSpace->ImageList);
-    DPRINT1("Session %lu is ready to go: 0x%p 0x%p, %lx 0x%p\n",
-            *SessionId, MmSessionSpace, SessionGlobal, SessionPageDirIndex, PageTables);
-
-    /* Initialize session pool */
-    //Status = MiInitializeSessionPool();
-    Status = STATUS_SUCCESS;
-    ASSERT(NT_SUCCESS(Status) == TRUE);
-
-    /* Initialize system space */
-    Result = MiInitializeSystemSpaceMap(&SessionGlobal->Session);
-    ASSERT(Result == TRUE);
-
-    /* Initialize the process list, make sure the workign set list is empty */
-    ASSERT(SessionGlobal->WsListEntry.Flink == NULL);
-    ASSERT(SessionGlobal->WsListEntry.Blink == NULL);
-    InitializeListHead(&SessionGlobal->ProcessList);
-
-    /* We're done, clear the flag */
-    ASSERT(Process->Flags & PSF_SESSION_CREATION_UNDERWAY_BIT);
-    PspClearProcessFlag(Process, PSF_SESSION_CREATION_UNDERWAY_BIT);
-
-    /* Insert the process into the session  */
-    ASSERT(Process->Session == NULL);
-    ASSERT(SessionGlobal->ProcessReferenceToSession == 0);
-    SessionGlobal->ProcessReferenceToSession = 1;
-
-    /* We're done */
-    InterlockedIncrement(&MmSessionDataPages);
-    return STATUS_SUCCESS;
-}
-
-NTSTATUS
-NTAPI
-MmSessionCreate(OUT PULONG SessionId)
-{
-    PEPROCESS Process = PsGetCurrentProcess();
-    ULONG SessionLeaderExists;
-    NTSTATUS Status;
-
-    /* Fail if the process is already in a session */
-    if (Process->Flags & PSF_PROCESS_IN_SESSION_BIT)
-    {
-        DPRINT1("Process already in session\n");
-        return STATUS_ALREADY_COMMITTED;
-    }
-
-    /* Check if the process is already the session leader */
-    if (!Process->Vm.Flags.SessionLeader)
-    {
-        /* Atomically set it as the leader */
-        SessionLeaderExists = InterlockedCompareExchange(&MiSessionLeaderExists, 1,
0);
-        if (SessionLeaderExists)
-        {
-            DPRINT1("Session leader race\n");
-            return STATUS_INVALID_SYSTEM_SERVICE;
-        }
-
-        /* Do the work required to upgrade him */
-        MiSessionLeader(Process);
-    }
-
-    /* Create the session */
-    KeEnterCriticalRegion();
-    Status = MiSessionCreateInternal(SessionId);
-    if (!NT_SUCCESS(Status))
-    {
-        KeLeaveCriticalRegion();
-        return Status;
-    }
-
-    /* Set up the session working set */
-    Status = MiSessionInitializeWorkingSetList();
-    if (!NT_SUCCESS(Status))
-    {
-        /* Fail */
-        //MiDereferenceSession();
-        ASSERT(FALSE);
-        KeLeaveCriticalRegion();
-        return Status;
-    }
-
-    /* All done */
-    KeLeaveCriticalRegion();
-
-    /* Set and assert the flags, and return */
-    MmSessionSpace->u.Flags.Initialized = 1;
-    PspSetProcessFlag(Process, PSF_PROCESS_IN_SESSION_BIT);
-    ASSERT(MiSessionLeaderExists == 1);
-    return Status;
-}
-
-NTSTATUS
-NTAPI
-MmSessionDelete(IN ULONG SessionId)
-{
-    PEPROCESS Process = PsGetCurrentProcess();
-
-    /* Process must be in a session */
-    if (!(Process->Flags & PSF_PROCESS_IN_SESSION_BIT))
-    {
-        DPRINT1("Not in a session!\n");
-        return STATUS_UNABLE_TO_FREE_VM;
-    }
-
-    /* It must be the session leader */
-    if (!Process->Vm.Flags.SessionLeader)
-    {
-        DPRINT1("Not a session leader!\n");
-        return STATUS_UNABLE_TO_FREE_VM;
-    }
-
-    /* Remove one reference count */
-    KeEnterCriticalRegion();
-    /* FIXME: Do it */
-    KeLeaveCriticalRegion();
-
-    /* All done */
-    return STATUS_SUCCESS;
-}
 
 /* SYSTEM CALLS ***************************************************************/
 

Added: trunk/reactos/ntoskrnl/mm/ARM3/session.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/mm/ARM3/session.c…
==============================================================================
--- trunk/reactos/ntoskrnl/mm/ARM3/session.c	(added)
+++ trunk/reactos/ntoskrnl/mm/ARM3/session.c	[iso-8859-1] Tue Nov 19 21:13:31 2013
@@ -0,0 +1,702 @@
+/*
+ * PROJECT:         ReactOS Kernel
+ * LICENSE:         BSD - See COPYING.ARM in the top level directory
+ * FILE:            ntoskrnl/mm/ARM3/session.c
+ * PURPOSE:         Session support routines
+ * PROGRAMMERS:     ReactOS Portable Systems Group
+ */
+
+/* INCLUDES *******************************************************************/
+
+#include <ntoskrnl.h>
+#define NDEBUG
+#include <debug.h>
+
+#define MODULE_INVOLVED_IN_ARM3
+#include "../ARM3/miarm.h"
+
+/* GLOBALS ********************************************************************/
+
+PMM_SESSION_SPACE MmSessionSpace;
+PFN_NUMBER MiSessionDataPages, MiSessionTagPages, MiSessionTagSizePages;
+PFN_NUMBER MiSessionBigPoolPages, MiSessionCreateCharge;
+KGUARDED_MUTEX MiSessionIdMutex;
+LONG MmSessionDataPages;
+PRTL_BITMAP MiSessionIdBitmap;
+volatile LONG MiSessionLeaderExists;
+
+
+/* PRIVATE FUNCTIONS **********************************************************/
+
+LCID
+NTAPI
+MmGetSessionLocaleId(VOID)
+{
+    PEPROCESS Process;
+    PAGED_CODE();
+
+    //
+    // Get the current process
+    //
+    Process = PsGetCurrentProcess();
+
+    //
+    // Check if it's the Session Leader
+    //
+    if (Process->Vm.Flags.SessionLeader)
+    {
+        //
+        // Make sure it has a valid Session
+        //
+        if (Process->Session)
+        {
+            //
+            // Get the Locale ID
+            //
+            return ((PMM_SESSION_SPACE)Process->Session)->LocaleId;
+        }
+    }
+
+    //
+    // Not a session leader, return the default
+    //
+    return PsDefaultThreadLocaleId;
+}
+
+VOID
+NTAPI
+MiInitializeSessionIds(VOID)
+{
+    ULONG Size, BitmapSize;
+    PFN_NUMBER TotalPages;
+
+    /* Setup the total number of data pages needed for the structure */
+    TotalPages = MI_SESSION_DATA_PAGES_MAXIMUM;
+    MiSessionDataPages = ROUND_TO_PAGES(sizeof(MM_SESSION_SPACE)) >> PAGE_SHIFT;
+    ASSERT(MiSessionDataPages <= MI_SESSION_DATA_PAGES_MAXIMUM - 3);
+    TotalPages -= MiSessionDataPages;
+
+    /* Setup the number of pages needed for session pool tags */
+    MiSessionTagSizePages = 2;
+    MiSessionBigPoolPages = 1;
+    MiSessionTagPages = MiSessionTagSizePages + MiSessionBigPoolPages;
+    ASSERT(MiSessionTagPages <= TotalPages);
+    ASSERT(MiSessionTagPages < MI_SESSION_TAG_PAGES_MAXIMUM);
+
+    /* Total pages needed for a session (FIXME: Probably different on PAE/x64) */
+    MiSessionCreateCharge = 1 + MiSessionDataPages + MiSessionTagPages;
+
+    /* Initialize the lock */
+    KeInitializeGuardedMutex(&MiSessionIdMutex);
+
+    /* Allocate the bitmap */
+    Size = MI_INITIAL_SESSION_IDS;
+    BitmapSize = ((Size + 31) / 32) * sizeof(ULONG);
+    MiSessionIdBitmap = ExAllocatePoolWithTag(PagedPool,
+                                              sizeof(RTL_BITMAP) + BitmapSize,
+                                              '  mM');
+    if (MiSessionIdBitmap)
+    {
+        /* Free all the bits */
+        RtlInitializeBitMap(MiSessionIdBitmap,
+                            (PVOID)(MiSessionIdBitmap + 1),
+                            Size);
+        RtlClearAllBits(MiSessionIdBitmap);
+    }
+    else
+    {
+        /* Die if we couldn't allocate the bitmap */
+        KeBugCheckEx(INSTALL_MORE_MEMORY,
+                     MmNumberOfPhysicalPages,
+                     MmLowestPhysicalPage,
+                     MmHighestPhysicalPage,
+                     0x200);
+    }
+}
+
+VOID
+NTAPI
+MiSessionLeader(IN PEPROCESS Process)
+{
+    KIRQL OldIrql;
+
+    /* Set the flag while under the expansion lock */
+    OldIrql = KeAcquireQueuedSpinLock(LockQueueExpansionLock);
+    Process->Vm.Flags.SessionLeader = TRUE;
+    KeReleaseQueuedSpinLock(LockQueueExpansionLock, OldIrql);
+}
+
+ULONG
+NTAPI
+MmGetSessionId(IN PEPROCESS Process)
+{
+    PMM_SESSION_SPACE SessionGlobal;
+
+    /* The session leader is always session zero */
+    if (Process->Vm.Flags.SessionLeader == 1) return 0;
+
+    /* Otherwise, get the session global, and read the session ID from it */
+    SessionGlobal = (PMM_SESSION_SPACE)Process->Session;
+    if (!SessionGlobal) return 0;
+    return SessionGlobal->SessionId;
+}
+
+ULONG
+NTAPI
+MmGetSessionIdEx(IN PEPROCESS Process)
+{
+    PMM_SESSION_SPACE SessionGlobal;
+
+    /* The session leader is always session zero */
+    if (Process->Vm.Flags.SessionLeader == 1) return 0;
+
+    /* Otherwise, get the session global, and read the session ID from it */
+    SessionGlobal = (PMM_SESSION_SPACE)Process->Session;
+    if (!SessionGlobal) return -1;
+    return SessionGlobal->SessionId;
+}
+
+VOID
+NTAPI
+MiReleaseProcessReferenceToSessionDataPage(IN PMM_SESSION_SPACE SessionGlobal)
+{
+    ULONG i, SessionId;
+    PMMPTE PointerPte;
+    PFN_NUMBER PageFrameIndex[MI_SESSION_DATA_PAGES_MAXIMUM];
+    PMMPFN Pfn1;
+    KIRQL OldIrql;
+
+    /* Is there more than just this reference? If so, bail out */
+    if (InterlockedDecrement(&SessionGlobal->ProcessReferenceToSession)) return;
+
+    /* Get the session ID */
+    SessionId = SessionGlobal->SessionId;
+    DPRINT1("Last process in session %lu going down!!!\n", SessionId);
+
+    /* Free the session page tables */
+#ifndef _M_AMD64
+    ExFreePoolWithTag(SessionGlobal->PageTables, 'tHmM');
+#endif
+    ASSERT(!MI_IS_PHYSICAL_ADDRESS(SessionGlobal));
+
+    /* Capture the data page PFNs */
+    PointerPte = MiAddressToPte(SessionGlobal);
+    for (i = 0; i < MiSessionDataPages; i++)
+    {
+        PageFrameIndex[i] = PFN_FROM_PTE(PointerPte + i);
+    }
+
+    /* Release them */
+    MiReleaseSystemPtes(PointerPte, MiSessionDataPages, SystemPteSpace);
+
+    /* Mark them as deleted */
+    for (i = 0; i < MiSessionDataPages; i++)
+    {
+        Pfn1 = MI_PFN_ELEMENT(PageFrameIndex[i]);
+        MI_SET_PFN_DELETED(Pfn1);
+    }
+
+    /* Loop every data page and drop a reference count */
+    OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
+    for (i = 0; i < MiSessionDataPages; i++)
+    {
+        /* Sanity check that the page is correct, then decrement it */
+        Pfn1 = MI_PFN_ELEMENT(PageFrameIndex[i]);
+        ASSERT(Pfn1->u2.ShareCount == 1);
+        ASSERT(Pfn1->u3.e2.ReferenceCount == 1);
+        MiDecrementShareCount(Pfn1, PageFrameIndex[i]);
+    }
+
+    /* Done playing with pages, release the lock */
+    KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
+
+    /* Decrement the number of data pages */
+    InterlockedDecrement(&MmSessionDataPages);
+
+    /* Free this session ID from the session bitmap */
+    KeAcquireGuardedMutex(&MiSessionIdMutex);
+    ASSERT(RtlCheckBit(MiSessionIdBitmap, SessionId));
+    RtlClearBit(MiSessionIdBitmap, SessionId);
+    KeReleaseGuardedMutex(&MiSessionIdMutex);
+}
+
+VOID
+NTAPI
+MiSessionRemoveProcess(VOID)
+{
+    PEPROCESS CurrentProcess = PsGetCurrentProcess();
+
+    /* If the process isn't already in a session, or if it's the leader... */
+    if (!(CurrentProcess->Flags & PSF_PROCESS_IN_SESSION_BIT) ||
+        (CurrentProcess->Vm.Flags.SessionLeader))
+    {
+        /* Then there's nothing to do */
+        return;
+    }
+
+    /* Sanity check */
+    ASSERT(MmIsAddressValid(MmSessionSpace) == TRUE);
+
+    /* Remove the process from the list ,and dereference the session */
+    // DO NOT ENABLE THIS UNLESS YOU FIXED THE NP POOL CORRUPTION THAT IT CAUSES!!!
+    //RemoveEntryList(&CurrentProcess->SessionProcessLinks);
+    //MiDereferenceSession();
+}
+
+VOID
+NTAPI
+MiSessionAddProcess(IN PEPROCESS NewProcess)
+{
+    PMM_SESSION_SPACE SessionGlobal;
+
+    /* The current process must already be in a session */
+    if (!(PsGetCurrentProcess()->Flags & PSF_PROCESS_IN_SESSION_BIT)) return;
+
+    /* Sanity check */
+    ASSERT(MmIsAddressValid(MmSessionSpace) == TRUE);
+
+    /* Get the global session */
+    SessionGlobal = MmSessionSpace->GlobalVirtualAddress;
+
+    /* Increment counters */
+    InterlockedIncrement((PLONG)&SessionGlobal->ReferenceCount);
+    InterlockedIncrement(&SessionGlobal->ResidentProcessCount);
+    InterlockedIncrement(&SessionGlobal->ProcessReferenceToSession);
+
+    /* Set the session pointer */
+    ASSERT(NewProcess->Session == NULL);
+    NewProcess->Session = SessionGlobal;
+
+    /* Insert it into the process list */
+    // DO NOT ENABLE THIS UNLESS YOU FIXED THE NP POOL CORRUPTION THAT IT CAUSES!!!
+    //InsertTailList(&SessionGlobal->ProcessList,
&NewProcess->SessionProcessLinks);
+
+    /* Set the flag */
+    PspSetProcessFlag(NewProcess, PSF_PROCESS_IN_SESSION_BIT);
+}
+
+NTSTATUS
+NTAPI
+MiSessionInitializeWorkingSetList(VOID)
+{
+    KIRQL OldIrql;
+    PMMPTE PointerPte, PointerPde;
+    MMPTE TempPte;
+    ULONG Color, Index;
+    PFN_NUMBER PageFrameIndex;
+    PMM_SESSION_SPACE SessionGlobal;
+    BOOLEAN AllocatedPageTable;
+    PMMWSL WorkingSetList;
+
+    /* Get pointers to session global and the session working set list */
+    SessionGlobal = MmSessionSpace->GlobalVirtualAddress;
+    WorkingSetList = (PMMWSL)MiSessionSpaceWs;
+
+    /* Fill out the two pointers */
+    MmSessionSpace->Vm.VmWorkingSetList = WorkingSetList;
+    MmSessionSpace->Wsle = (PMMWSLE)WorkingSetList->UsedPageTableEntries;
+
+    /* Get the PDE for the working set, and check if it's already allocated */
+    PointerPde = MiAddressToPde(WorkingSetList);
+    if (PointerPde->u.Hard.Valid == 1)
+    {
+        /* Nope, we'll have to do it */
+        ASSERT(PointerPde->u.Hard.Global == 0);
+        AllocatedPageTable = FALSE;
+    }
+    else
+    {
+        /* Yep, that makes our job easier */
+        AllocatedPageTable = TRUE;
+    }
+
+    /* Get the PTE for the working set */
+    PointerPte = MiAddressToPte(WorkingSetList);
+
+    /* Initialize the working set lock, and lock the PFN database */
+    ExInitializePushLock(&SessionGlobal->Vm.WorkingSetMutex);
+    //MmLockPageableSectionByHandle(ExPageLockHandle);
+    OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
+
+    /* Check if we need a page table */
+    if (AllocatedPageTable == TRUE)
+    {
+        /* Get a zeroed colored zero page */
+        Color = MI_GET_NEXT_COLOR();
+        PageFrameIndex = MiRemoveZeroPageSafe(Color);
+        if (!PageFrameIndex)
+        {
+            /* No zero pages, grab a free one */
+            PageFrameIndex = MiRemoveAnyPage(Color);
+
+            /* Zero it outside the PFN lock */
+            KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
+            MiZeroPhysicalPage(PageFrameIndex);
+            OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
+        }
+
+        /* Write a valid PDE for it */
+        TempPte.u.Long = ValidKernelPdeLocal.u.Long;
+        TempPte.u.Hard.PageFrameNumber = PageFrameIndex;
+        MI_WRITE_VALID_PTE(PointerPde, TempPte);
+
+        /* Add this into the list */
+        Index = ((ULONG_PTR)WorkingSetList - (ULONG_PTR)MmSessionBase) >> 22;
+#ifndef _M_AMD64
+        MmSessionSpace->PageTables[Index] = TempPte;
+#endif
+        /* Initialize the page directory page, and now zero the working set list itself
*/
+        MiInitializePfnForOtherProcess(PageFrameIndex,
+                                       PointerPde,
+                                       MmSessionSpace->SessionPageDirectoryIndex);
+        KeZeroPages(PointerPte, PAGE_SIZE);
+    }
+
+    /* Get a zeroed colored zero page */
+    Color = MI_GET_NEXT_COLOR();
+    PageFrameIndex = MiRemoveZeroPageSafe(Color);
+    if (!PageFrameIndex)
+    {
+        /* No zero pages, grab a free one */
+        PageFrameIndex = MiRemoveAnyPage(Color);
+
+        /* Zero it outside the PFN lock */
+        KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
+        MiZeroPhysicalPage(PageFrameIndex);
+        OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
+    }
+
+    /* Write a valid PTE for it */
+    TempPte.u.Long = ValidKernelPteLocal.u.Long;
+    TempPte.u.Hard.Dirty = TRUE;
+    TempPte.u.Hard.PageFrameNumber = PageFrameIndex;
+
+    /* Initialize the working set list page */
+    MiInitializePfnAndMakePteValid(PageFrameIndex, PointerPte, TempPte);
+
+    /* Now we can release the PFN database lock */
+    KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
+
+    /* Fill out the working set structure */
+    MmSessionSpace->Vm.Flags.SessionSpace = 1;
+    MmSessionSpace->Vm.MinimumWorkingSetSize = 20;
+    MmSessionSpace->Vm.MaximumWorkingSetSize = 384;
+    WorkingSetList->LastEntry = 20;
+    WorkingSetList->HashTable = NULL;
+    WorkingSetList->HashTableSize = 0;
+    WorkingSetList->Wsle = MmSessionSpace->Wsle;
+
+    /* FIXME: Handle list insertions */
+    ASSERT(SessionGlobal->WsListEntry.Flink == NULL);
+    ASSERT(SessionGlobal->WsListEntry.Blink == NULL);
+    ASSERT(SessionGlobal->Vm.WorkingSetExpansionLinks.Flink == NULL);
+    ASSERT(SessionGlobal->Vm.WorkingSetExpansionLinks.Blink == NULL);
+
+    /* All done, return */
+    //MmUnlockPageableImageSection(ExPageLockHandle);
+    return STATUS_SUCCESS;
+}
+
+NTSTATUS
+NTAPI
+MiSessionCreateInternal(OUT PULONG SessionId)
+{
+    PEPROCESS Process = PsGetCurrentProcess();
+    ULONG NewFlags, Flags, Size, i, Color;
+    KIRQL OldIrql;
+    PMMPTE PointerPte, PageTables, SessionPte;
+    PMMPDE PointerPde;
+    PMM_SESSION_SPACE SessionGlobal;
+    MMPTE TempPte;
+    NTSTATUS Status;
+    BOOLEAN Result;
+    PFN_NUMBER SessionPageDirIndex;
+    PFN_NUMBER TagPage[MI_SESSION_TAG_PAGES_MAXIMUM];
+    PFN_NUMBER DataPage[MI_SESSION_DATA_PAGES_MAXIMUM];
+
+    /* This should not exist yet */
+    ASSERT(MmIsAddressValid(MmSessionSpace) == FALSE);
+
+    /* Loop so we can set the session-is-creating flag */
+    Flags = Process->Flags;
+    while (TRUE)
+    {
+        /* Check if it's already set */
+        if (Flags & PSF_SESSION_CREATION_UNDERWAY_BIT)
+        {
+            /* Bail out */
+            DPRINT1("Lost session race\n");
+            return STATUS_ALREADY_COMMITTED;
+        }
+
+        /* Now try to set it */
+        NewFlags = InterlockedCompareExchange((PLONG)&Process->Flags,
+                                              Flags | PSF_SESSION_CREATION_UNDERWAY_BIT,
+                                              Flags);
+        if (NewFlags == Flags) break;
+
+        /* It changed, try again */
+        Flags = NewFlags;
+    }
+
+    /* Now we should own the flag */
+    ASSERT(Process->Flags & PSF_SESSION_CREATION_UNDERWAY_BIT);
+
+    /*
+     * Session space covers everything from 0xA0000000 to 0xC0000000.
+     * Allocate enough page tables to describe the entire region
+     */
+    Size = (0x20000000 / PDE_MAPPED_VA) * sizeof(MMPTE);
+    PageTables = ExAllocatePoolWithTag(NonPagedPool, Size, 'tHmM');
+    ASSERT(PageTables != NULL);
+    RtlZeroMemory(PageTables, Size);
+
+    /* Lock the session ID creation mutex */
+    KeAcquireGuardedMutex(&MiSessionIdMutex);
+
+    /* Allocate a new Session ID */
+    *SessionId = RtlFindClearBitsAndSet(MiSessionIdBitmap, 1, 0);
+    if (*SessionId == 0xFFFFFFFF)
+    {
+        /* We ran out of session IDs, we should expand */
+        DPRINT1("Too many sessions created. Expansion not yet supported\n");
+        ExFreePoolWithTag(PageTables, 'tHmM');
+        return STATUS_NO_MEMORY;
+    }
+
+    /* Unlock the session ID creation mutex */
+    KeReleaseGuardedMutex(&MiSessionIdMutex);
+
+    /* Reserve the global PTEs */
+    SessionPte = MiReserveSystemPtes(MiSessionDataPages, SystemPteSpace);
+    ASSERT(SessionPte != NULL);
+
+    /* Acquire the PFN lock while we set everything up */
+    OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
+
+    /* Loop the global PTEs */
+    TempPte.u.Long = ValidKernelPte.u.Long;
+    for (i = 0; i < MiSessionDataPages; i++)
+    {
+        /* Get a zeroed colored zero page */
+        Color = MI_GET_NEXT_COLOR();
+        DataPage[i] = MiRemoveZeroPageSafe(Color);
+        if (!DataPage[i])
+        {
+            /* No zero pages, grab a free one */
+            DataPage[i] = MiRemoveAnyPage(Color);
+
+            /* Zero it outside the PFN lock */
+            KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
+            MiZeroPhysicalPage(DataPage[i]);
+            OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
+        }
+
+        /* Fill the PTE out */
+        TempPte.u.Hard.PageFrameNumber = DataPage[i];
+        MI_WRITE_VALID_PTE(SessionPte + i, TempPte);
+    }
+
+    /* Set the pointer to global space */
+    SessionGlobal = MiPteToAddress(SessionPte);
+
+    /* Get a zeroed colored zero page */
+    Color = MI_GET_NEXT_COLOR();
+    SessionPageDirIndex = MiRemoveZeroPageSafe(Color);
+    if (!SessionPageDirIndex)
+    {
+        /* No zero pages, grab a free one */
+        SessionPageDirIndex = MiRemoveAnyPage(Color);
+
+        /* Zero it outside the PFN lock */
+        KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
+        MiZeroPhysicalPage(SessionPageDirIndex);
+        OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
+    }
+
+    /* Fill the PTE out */
+    TempPte.u.Long = ValidKernelPdeLocal.u.Long;
+    TempPte.u.Hard.PageFrameNumber = SessionPageDirIndex;
+
+    /* Setup, allocate, fill out the MmSessionSpace PTE */
+    PointerPde = MiAddressToPde(MmSessionSpace);
+    ASSERT(PointerPde->u.Long == 0);
+    MI_WRITE_VALID_PTE(PointerPde, TempPte);
+    MiInitializePfnForOtherProcess(SessionPageDirIndex,
+                                   PointerPde,
+                                   SessionPageDirIndex);
+    ASSERT(MI_PFN_ELEMENT(SessionPageDirIndex)->u1.WsIndex == 0);
+
+     /* Loop all the local PTEs for it */
+    TempPte.u.Long = ValidKernelPteLocal.u.Long;
+    PointerPte = MiAddressToPte(MmSessionSpace);
+    for (i = 0; i < MiSessionDataPages; i++)
+    {
+        /* And fill them out */
+        TempPte.u.Hard.PageFrameNumber = DataPage[i];
+        MiInitializePfnAndMakePteValid(DataPage[i], PointerPte + i, TempPte);
+        ASSERT(MI_PFN_ELEMENT(DataPage[i])->u1.WsIndex == 0);
+    }
+
+     /* Finally loop all of the session pool tag pages */
+    for (i = 0; i < MiSessionTagPages; i++)
+    {
+        /* Grab a zeroed colored page */
+        Color = MI_GET_NEXT_COLOR();
+        TagPage[i] = MiRemoveZeroPageSafe(Color);
+        if (!TagPage[i])
+        {
+            /* No zero pages, grab a free one */
+            TagPage[i] = MiRemoveAnyPage(Color);
+
+            /* Zero it outside the PFN lock */
+            KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
+            MiZeroPhysicalPage(TagPage[i]);
+            OldIrql = KeAcquireQueuedSpinLock(LockQueuePfnLock);
+        }
+
+        /* Fill the PTE out */
+        TempPte.u.Hard.PageFrameNumber = TagPage[i];
+        MiInitializePfnAndMakePteValid(TagPage[i],
+                                       PointerPte + MiSessionDataPages + i,
+                                       TempPte);
+    }
+
+    /* PTEs have been setup, release the PFN lock */
+    KeReleaseQueuedSpinLock(LockQueuePfnLock, OldIrql);
+
+    /* Fill out the session space structure now */
+    MmSessionSpace->GlobalVirtualAddress = SessionGlobal;
+    MmSessionSpace->ReferenceCount = 1;
+    MmSessionSpace->ResidentProcessCount = 1;
+    MmSessionSpace->u.LongFlags = 0;
+    MmSessionSpace->SessionId = *SessionId;
+    MmSessionSpace->LocaleId = PsDefaultSystemLocaleId;
+    MmSessionSpace->SessionPageDirectoryIndex = SessionPageDirIndex;
+    MmSessionSpace->Color = Color;
+    MmSessionSpace->NonPageablePages = MiSessionCreateCharge;
+    MmSessionSpace->CommittedPages = MiSessionCreateCharge;
+#ifndef _M_AMD64
+    MmSessionSpace->PageTables = PageTables;
+    MmSessionSpace->PageTables[PointerPde - MiAddressToPde(MmSessionBase)] =
*PointerPde;
+#endif
+    InitializeListHead(&MmSessionSpace->ImageList);
+    DPRINT1("Session %lu is ready to go: 0x%p 0x%p, %lx 0x%p\n",
+            *SessionId, MmSessionSpace, SessionGlobal, SessionPageDirIndex, PageTables);
+
+    /* Initialize session pool */
+    //Status = MiInitializeSessionPool();
+    Status = STATUS_SUCCESS;
+    ASSERT(NT_SUCCESS(Status) == TRUE);
+
+    /* Initialize system space */
+    Result = MiInitializeSystemSpaceMap(&SessionGlobal->Session);
+    ASSERT(Result == TRUE);
+
+    /* Initialize the process list, make sure the workign set list is empty */
+    ASSERT(SessionGlobal->WsListEntry.Flink == NULL);
+    ASSERT(SessionGlobal->WsListEntry.Blink == NULL);
+    InitializeListHead(&SessionGlobal->ProcessList);
+
+    /* We're done, clear the flag */
+    ASSERT(Process->Flags & PSF_SESSION_CREATION_UNDERWAY_BIT);
+    PspClearProcessFlag(Process, PSF_SESSION_CREATION_UNDERWAY_BIT);
+
+    /* Insert the process into the session  */
+    ASSERT(Process->Session == NULL);
+    ASSERT(SessionGlobal->ProcessReferenceToSession == 0);
+    SessionGlobal->ProcessReferenceToSession = 1;
+
+    /* We're done */
+    InterlockedIncrement(&MmSessionDataPages);
+    return STATUS_SUCCESS;
+}
+
+NTSTATUS
+NTAPI
+MmSessionCreate(OUT PULONG SessionId)
+{
+    PEPROCESS Process = PsGetCurrentProcess();
+    ULONG SessionLeaderExists;
+    NTSTATUS Status;
+
+    /* Fail if the process is already in a session */
+    if (Process->Flags & PSF_PROCESS_IN_SESSION_BIT)
+    {
+        DPRINT1("Process already in session\n");
+        return STATUS_ALREADY_COMMITTED;
+    }
+
+    /* Check if the process is already the session leader */
+    if (!Process->Vm.Flags.SessionLeader)
+    {
+        /* Atomically set it as the leader */
+        SessionLeaderExists = InterlockedCompareExchange(&MiSessionLeaderExists, 1,
0);
+        if (SessionLeaderExists)
+        {
+            DPRINT1("Session leader race\n");
+            return STATUS_INVALID_SYSTEM_SERVICE;
+        }
+
+        /* Do the work required to upgrade him */
+        MiSessionLeader(Process);
+    }
+
+    /* Create the session */
+    KeEnterCriticalRegion();
+    Status = MiSessionCreateInternal(SessionId);
+    if (!NT_SUCCESS(Status))
+    {
+        KeLeaveCriticalRegion();
+        return Status;
+    }
+
+    /* Set up the session working set */
+    Status = MiSessionInitializeWorkingSetList();
+    if (!NT_SUCCESS(Status))
+    {
+        /* Fail */
+        //MiDereferenceSession();
+        ASSERT(FALSE);
+        KeLeaveCriticalRegion();
+        return Status;
+    }
+
+    /* All done */
+    KeLeaveCriticalRegion();
+
+    /* Set and assert the flags, and return */
+    MmSessionSpace->u.Flags.Initialized = 1;
+    PspSetProcessFlag(Process, PSF_PROCESS_IN_SESSION_BIT);
+    ASSERT(MiSessionLeaderExists == 1);
+    return Status;
+}
+
+NTSTATUS
+NTAPI
+MmSessionDelete(IN ULONG SessionId)
+{
+    PEPROCESS Process = PsGetCurrentProcess();
+
+    /* Process must be in a session */
+    if (!(Process->Flags & PSF_PROCESS_IN_SESSION_BIT))
+    {
+        DPRINT1("Not in a session!\n");
+        return STATUS_UNABLE_TO_FREE_VM;
+    }
+
+    /* It must be the session leader */
+    if (!Process->Vm.Flags.SessionLeader)
+    {
+        DPRINT1("Not a session leader!\n");
+        return STATUS_UNABLE_TO_FREE_VM;
+    }
+
+    /* Remove one reference count */
+    KeEnterCriticalRegion();
+    /* FIXME: Do it */
+    KeLeaveCriticalRegion();
+
+    /* All done */
+    return STATUS_SUCCESS;
+}

Propchange: trunk/reactos/ntoskrnl/mm/ARM3/session.c
------------------------------------------------------------------------------
    svn:eol-style = native