smclient.c File Reference

#include "precomp.h"
#include <sm/smmsg.h>
#include <debug.h>

Include dependency graph for smclient.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
NTSTATUS NTAPI	SmExecPgm (IN HANDLE SmApiPort, IN PRTL_USER_PROCESS_INFORMATION ProcessInformation, IN BOOLEAN DebugFlag)
NTSTATUS NTAPI	SmConnectToSm (IN PUNICODE_STRING SbApiPortName, IN HANDLE SbApiPort, IN ULONG ImageType, OUT PHANDLE SmApiPort)
NTSTATUS NTAPI	SmSessionComplete (IN HANDLE SmApiPort, IN ULONG SessionId, IN NTSTATUS SessionStatus)
NTSTATUS NTAPI	SmStartCsr (IN HANDLE SmApiPort, OUT PULONG pMuSessionId, IN PUNICODE_STRING CommandLine, OUT PHANDLE pWindowsSubSysProcessId, OUT PHANDLE pInitialCommandProcessId)
NTSTATUS NTAPI	SmStopCsr (IN HANDLE SmApiPort, IN ULONG SessionId)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 14 of file smclient.c.

Function Documentation

SmConnectToSm()

NTSTATUS NTAPI SmConnectToSm	(	IN PUNICODE_STRING	SbApiPortName,
		IN HANDLE	SbApiPort,
		IN ULONG	ImageType,
		OUT PHANDLE	SmApiPort
	)

Definition at line 68 of file smclient.c.

{
    NTSTATUS Status;
    SB_CONNECTION_INFO ConnectInfo;
    UNICODE_STRING PortName;
    SECURITY_QUALITY_OF_SERVICE SecurityQos;
    ULONG ConnectInfoLength = sizeof(ConnectInfo);

    /* Setup the QoS structure */
    SecurityQos.ImpersonationLevel = SecurityIdentification;
    SecurityQos.ContextTrackingMode = SECURITY_DYNAMIC_TRACKING;
    SecurityQos.EffectiveOnly = TRUE;

    /* Set the SM API port name */
    RtlInitUnicodeString(&PortName, L"\\SmApiPort");

    /* Check if this is a client connecting to SMSS, or SMSS to itself */
    if (SbApiPortName)
    {
        /* A client SB port as well as an image type must be present */
        if (!(SbApiPort) || !(ImageType)) return STATUS_INVALID_PARAMETER_MIX;

        /* Copy the client port name, and NULL-terminate it */
        RtlCopyMemory(ConnectInfo.SbApiPortName,
                      SbApiPortName->Buffer,
                      SbApiPortName->Length);
        ConnectInfo.SbApiPortName[SbApiPortName->Length /
                                  sizeof(WCHAR)] = UNICODE_NULL;

        /* Save the subsystem type */
        ConnectInfo.SubsystemType = ImageType;
    }
    else
    {
        /* No client port, and the subsystem type is not set */
        ConnectInfo.SbApiPortName[0] = UNICODE_NULL;
        ConnectInfo.SubsystemType = IMAGE_SUBSYSTEM_UNKNOWN;
    }

    /* Connect to SMSS and exchange connection information */
    Status = NtConnectPort(SmApiPort,
                           &PortName,
                           &SecurityQos,
                           NULL,
                           NULL,
                           NULL,
                           &ConnectInfo,
                           &ConnectInfoLength);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("SmConnectToSm: Connect to Sm failed %lx\n", Status);
    }
    else
    {
        /* Treat a warning or informational status as success */
        Status = STATUS_SUCCESS;
    }

    /* Return if the connection was successful or not */
    return Status;
}

Referenced by CsrServerInitialization(), and SmpExecuteInitialCommand().

SmExecPgm()

NTSTATUS NTAPI SmExecPgm	(	IN HANDLE	SmApiPort,
		IN PRTL_USER_PROCESS_INFORMATION	ProcessInformation,
		IN BOOLEAN	DebugFlag
	)

Definition at line 21 of file smclient.c.

{
    NTSTATUS Status;
    SM_API_MSG SmApiMsg;

#if 0 //def _WIN64 // You can take care of this Timo
    /* 64-bit SMSS needs to talk to 32-bit processes so do the LPC conversion */
    if (SmpIsWow64Process())
    {
        return SmpWow64ExecPgm(SmApiPort, ProcessInformation, DebugFlag);
    }
#endif

    /* Initialize the generic LPC header */
    SmApiMsg.h.u2.ZeroInit = 0;
    SmApiMsg.h.u1.s1.DataLength = sizeof(SM_EXEC_PGM_MSG) + 8;
    SmApiMsg.h.u1.s1.TotalLength = sizeof(SmApiMsg);

    /* Initialize this specific API's parameters */
    SmApiMsg.ApiNumber = SmpExecPgmApi;
    RtlCopyMemory(&SmApiMsg.u.ExecPgm.ProcessInformation,
                  ProcessInformation,
                  sizeof(SmApiMsg.u.ExecPgm.ProcessInformation));
    SmApiMsg.u.ExecPgm.DebugFlag = DebugFlag;

    /* Send the message to SMSS */
    Status = NtRequestWaitReplyPort(SmApiPort, &SmApiMsg.h, &SmApiMsg.h);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("SmExecPgm: NtRequestWaitReply Failed %lx\n", Status);
    }
    else
    {
        /* Upon success, we use the API's return value */
        Status = SmApiMsg.ReturnValue;
    }

    /* Close the handles that the parent passed in and return status */
    NtClose(ProcessInformation->ProcessHandle);
    NtClose(ProcessInformation->ThreadHandle);
    return Status;
}

Referenced by SmpExecPgm(), and SmpExecuteInitialCommand().

SmSessionComplete()

NTSTATUS NTAPI SmSessionComplete	(	IN HANDLE	SmApiPort,
		IN ULONG	SessionId,
		IN NTSTATUS	SessionStatus
	)

Definition at line 135 of file smclient.c.

{
    NTSTATUS Status;
    SM_API_MSG ApiMessage;
    PSM_SESSION_COMPLETE_MSG SessionComplete = &ApiMessage.u.SessionComplete;

    /* Set the message data */
    SessionComplete->SessionId = SessionId;
    SessionComplete->SessionStatus = SessionStatus;

    /* Set the API Message Port Message header */
    ApiMessage.ApiNumber = SmpSessionCompleteApi;
    ApiMessage.h.u1.s1.DataLength = sizeof(SM_SESSION_COMPLETE_MSG) + 8;
    ApiMessage.h.u1.s1.TotalLength = sizeof(SM_API_MSG);
    ApiMessage.h.u2.ZeroInit = 0;

    /* Sent the message and wait for a reply */
    Status = NtRequestWaitReplyPort(SmApiPort,
                                    &ApiMessage.h,
                                    &ApiMessage.h);
    if (NT_SUCCESS(Status))
    {
        /* Return the real status */
        Status = ApiMessage.ReturnValue;
    }
    else
    {
        DPRINT1("SmCompleteSession: NtRequestWaitReply failed\n");
    }

    /* Return status */
    return Status;
}

Referenced by CsrDereferenceNtSession().

SmStartCsr()

NTSTATUS NTAPI SmStartCsr	(	IN HANDLE	SmApiPort,
		OUT PULONG	pMuSessionId,
		IN PUNICODE_STRING	CommandLine,
		OUT PHANDLE	pWindowsSubSysProcessId,
		OUT PHANDLE	pInitialCommandProcessId
	)

Definition at line 173 of file smclient.c.

{
    NTSTATUS Status;
    SM_API_MSG SmApiMsg;

    /* Initialize the generic LPC header */
    RtlZeroMemory(&SmApiMsg, sizeof(SmApiMsg));
    SmApiMsg.h.u1.s1.DataLength = sizeof(SM_EXEC_PGM_MSG) + 8;
    SmApiMsg.h.u1.s1.TotalLength = sizeof(SmApiMsg);

    /* Initialize this specific API's parameters */
    SmApiMsg.ApiNumber = SmpStartCsrApi;
    if (CommandLine)
    {
        if (CommandLine->Length > ARRAYSIZE(SmApiMsg.u.StartCsr.Buffer))
        {
            DPRINT1("SmStartCsr: Command line too long\n");
            return STATUS_INVALID_PARAMETER;
        }
        RtlCopyMemory(SmApiMsg.u.StartCsr.Buffer, CommandLine->Buffer, CommandLine->Length);
        SmApiMsg.u.StartCsr.Length = CommandLine->Length;
    }

    /* Send the message to SMSS */
    Status = NtRequestWaitReplyPort(SmApiPort, &SmApiMsg.h, &SmApiMsg.h);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("SmStartCsr: NtRequestWaitReply Failed %lx\n", Status);
    }
    else
    {
        /* Upon success, we use the API's return value */
        Status = SmApiMsg.ReturnValue;
    }

    /* Give back informations to caller */
    *pMuSessionId = SmApiMsg.u.StartCsr.MuSessionId;
    *pWindowsSubSysProcessId = SmApiMsg.u.StartCsr.WindowsSubSysProcessId;
    *pInitialCommandProcessId = SmApiMsg.u.StartCsr.SmpInitialCommandProcessId;

    return Status;
}

Referenced by SmpStartCsr().

SmStopCsr()

NTSTATUS NTAPI SmStopCsr	(	IN HANDLE	SmApiPort,
		IN ULONG	SessionId
	)

Definition at line 222 of file smclient.c.

{
    NTSTATUS Status;
    SM_API_MSG SmApiMsg;

    /* Initialize the generic LPC header */
    RtlZeroMemory(&SmApiMsg, sizeof(SmApiMsg));
    SmApiMsg.h.u1.s1.DataLength = sizeof(SM_EXEC_PGM_MSG) + 8;
    SmApiMsg.h.u1.s1.TotalLength = sizeof(SmApiMsg);

    /* Initialize this specific API's parameters */
    SmApiMsg.ApiNumber = SmpStopCsrApi;
    SmApiMsg.u.StopCsr.MuSessionId = SessionId;

    /* Send the message to SMSS */
    Status = NtRequestWaitReplyPort(SmApiPort, &SmApiMsg.h, &SmApiMsg.h);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("SmStopCsr: NtRequestWaitReply Failed %lx\n", Status);
    }
    else
    {
        /* Upon success, we use the API's return value */
        Status = SmApiMsg.ReturnValue;
    }

    return Status;
}