#include "smss.h"
#include <debug.h>

Include dependency graph for smsessn.c:

Classes
struct	_SMP_SESSION

Macros
#define	NDEBUG

Typedefs
typedef struct _SMP_SESSION	SMP_SESSION

typedef struct _SMP_SESSION *	PSMP_SESSION

Functions
BOOLEAN NTAPI	SmpCheckDuplicateMuSessionId (IN ULONG MuSessionId)

PSMP_SESSION NTAPI	SmpSessionIdToSession (IN ULONG SessionId)

VOID NTAPI	SmpDeleteSession (IN ULONG SessionId)

ULONG NTAPI	SmpAllocateSessionId (IN PSMP_SUBSYSTEM Subsystem, IN PSMP_SUBSYSTEM OtherSubsystem)

NTSTATUS NTAPI	SmpGetProcessMuSessionId (IN HANDLE ProcessHandle, OUT PULONG SessionId)

NTSTATUS NTAPI	SmpSetProcessMuSessionId (IN HANDLE ProcessHandle, IN ULONG SessionId)

Variables
RTL_CRITICAL_SECTION	SmpSessionListLock

LIST_ENTRY	SmpSessionListHead

ULONG	SmpNextSessionId

BOOLEAN	SmpNextSessionIdScanMode

BOOLEAN	SmpDbgSsLoaded

HANDLE	SmpSessionsObjectDirectory

Macro Definition Documentation

◆ NDEBUG

#define NDEBUG

Definition at line 13 of file smsessn.c.

Typedef Documentation

◆ PSMP_SESSION

typedef struct _SMP_SESSION * PSMP_SESSION

◆ SMP_SESSION

typedef struct _SMP_SESSION SMP_SESSION

Function Documentation

◆ SmpAllocateSessionId()

ULONG NTAPI SmpAllocateSessionId	(	IN PSMP_SUBSYSTEM	Subsystem,
		IN PSMP_SUBSYSTEM	OtherSubsystem
	)

Definition at line 123 of file smsessn.c.

{
    ULONG SessionId;
    PSMP_SESSION Session;
 
    /* Allocate a new ID while under the lock */
    RtlEnterCriticalSection(&SmpSessionListLock);
    SessionId = SmpNextSessionId++;
 
    /* Check for overflow */
    if (SmpNextSessionIdScanMode)
    {
        /* Break if it happened */
        UNIMPLEMENTED_DBGBREAK("SMSS: SessionId's Wrapped\n");
    }
    else
    {
        /* Detect it for next time */
        if (!SmpNextSessionId)
            SmpNextSessionIdScanMode = TRUE;
    }
 
    /* Allocate a session structure */
    Session = RtlAllocateHeap(SmpHeap, 0, sizeof(SMP_SESSION));
    if (Session)
    {
        /* Write the session data and insert it into the session list */
        Session->Subsystem = Subsystem;
        Session->SessionId = SessionId;
        Session->OtherSubsystem = OtherSubsystem;
        InsertTailList(&SmpSessionListHead, &Session->Entry);
    }
    else
    {
        DPRINT1("SMSS: Unable to keep track of session ID -- no memory available\n");
    }
 
    /* Release the session lock */
    RtlLeaveCriticalSection(&SmpSessionListLock);
    return SessionId;
}

Referenced by SmpLoadSubSystem(), and SmpSbCreateSession().

◆ SmpCheckDuplicateMuSessionId()

BOOLEAN NTAPI SmpCheckDuplicateMuSessionId ( IN ULONG MuSessionId )

Definition at line 37 of file smsessn.c.

{
    PSMP_SUBSYSTEM Subsystem;
    BOOLEAN FoundDuplicate = FALSE;
    PLIST_ENTRY NextEntry;
 
    /* Lock the subsystem database */
    RtlEnterCriticalSection(&SmpKnownSubSysLock);
 
    /* Scan each entry */
    NextEntry = SmpKnownSubSysHead.Flink;
    while (NextEntry != &SmpKnownSubSysHead)
    {
        /* Check if this entry has the same session ID */
        Subsystem = CONTAINING_RECORD(NextEntry, SMP_SUBSYSTEM, Entry);
        if (Subsystem->MuSessionId == MuSessionId)
        {
            /* Break out of here! */
            FoundDuplicate = TRUE;
            break;
        }
 
        /* Keep going */
        NextEntry = NextEntry->Flink;
    }
 
    /* Release the database and return the result */
    RtlLeaveCriticalSection(&SmpKnownSubSysLock);
    return FoundDuplicate;
}

Referenced by SmpLoadDeferedSubsystem(), SmpLoadSubSystem(), and SmpSbCreateSession().

◆ SmpDeleteSession()

VOID NTAPI SmpDeleteSession ( IN ULONG SessionId )

Definition at line 98 of file smsessn.c.

{
    PSMP_SESSION Session;
 
    /* Enter the lock and get the session structure */
    RtlEnterCriticalSection(&SmpSessionListLock);
    Session = SmpSessionIdToSession(SessionId);
    if (Session)
    {
        /* Remove it from the list */
        RemoveEntryList(&Session->Entry);
        RtlLeaveCriticalSection(&SmpSessionListLock);
 
        /* Now free the structure outside of the lock */
        RtlFreeHeap(SmpHeap, 0, Session);
    }
    else
    {
        /* ID doesn't map to one of our structures, nothing to do... */
        RtlLeaveCriticalSection(&SmpSessionListLock);
    }
}

Referenced by SmpLoadSubSystem(), and SmpSbCreateSession().

◆ SmpGetProcessMuSessionId()

NTSTATUS NTAPI SmpGetProcessMuSessionId	(	IN HANDLE	ProcessHandle,
		OUT PULONG	SessionId
	)

Definition at line 168 of file smsessn.c.

{
    NTSTATUS Status;
    ULONG ProcessSession;
 
    /* Query the kernel for the session ID */
    Status = NtQueryInformationProcess(ProcessHandle,
                                       ProcessSessionInformation,
                                       &ProcessSession,
                                       sizeof(ProcessSession),
                                       NULL);
    if (NT_SUCCESS(Status))
    {
        /* Copy it back into the buffer */
        *SessionId = ProcessSession;
    }
    else
    {
        /* Failure -- assume session zero */
        DPRINT1("SMSS: GetProcessMuSessionId, Process=%p, Status=%x\n",
                ProcessHandle, Status);
        *SessionId = 0;
    }
 
    /* Return result */
    return Status;
}

Referenced by SmpHandleConnectionRequest(), SmpLoadDeferedSubsystem(), and SmpSbCreateSession().

◆ SmpSessionIdToSession()

PSMP_SESSION NTAPI SmpSessionIdToSession ( IN ULONG SessionId )

Definition at line 70 of file smsessn.c.

{
    PSMP_SESSION Session, FoundSession = NULL;
    PLIST_ENTRY NextEntry;
 
    /* Loop the session list -- lock must already be held! */
    NextEntry = SmpSessionListHead.Flink;
    while (NextEntry != &SmpSessionListHead)
    {
        /* Check if this session's ID matches */
        Session = CONTAINING_RECORD(NextEntry, SMP_SESSION, Entry);
        if (Session->SessionId == SessionId)
        {
            /* Set this as the found session and break out */
            FoundSession = Session;
            break;
        }
 
        /* Keep going */
        NextEntry = NextEntry->Flink;
    }
 
    /* Return the session that was found and exit */
    return FoundSession;
}

Referenced by SmpDeleteSession().

◆ SmpSetProcessMuSessionId()

NTSTATUS NTAPI SmpSetProcessMuSessionId	(	IN HANDLE	ProcessHandle,
		IN ULONG	SessionId
	)

Definition at line 199 of file smsessn.c.

{
    NTSTATUS Status;
 
    /* Tell the kernel about the session ID */
    Status = NtSetInformationProcess(ProcessHandle,
                                     ProcessSessionInformation,
                                     &SessionId,
                                     sizeof(SessionId));
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("SMSS: SetProcessMuSessionId, Process=%p, Status=%x\n",
                ProcessHandle, Status);
    }
 
    /* Return */
    return Status;
}