[LWIP] Fix src/core/init.c a bit (#1620)
[reactos.git] / sdk / lib / smlib / smclient.c
1 /*
2 * PROJECT: ReactOS Windows-Compatible Session Manager
3 * LICENSE: BSD 2-Clause License
4 * FILE: lib/smlib/smclient.c
5 * PURPOSE: SMSS Client Library Stubs for calling SM APIs from a client
6 * PROGRAMMERS: Alex Ionescu
7 */
8
9 /* INCLUDES *******************************************************************/
10
11 #include "precomp.h"
12 #include <sm/smmsg.h> // To go in precomp.h after
13
14 #define NDEBUG
15 #include <debug.h>
16
17 /* FUNCTIONS ******************************************************************/
18
19 NTSTATUS
20 NTAPI
21 SmExecPgm(IN HANDLE SmApiPort,
22 IN PRTL_USER_PROCESS_INFORMATION ProcessInformation,
23 IN BOOLEAN DebugFlag)
24 {
25 NTSTATUS Status;
26 SM_API_MSG SmApiMsg;
27
28 #if 0 //def _WIN64 // You can take care of this Timo
29 /* 64-bit SMSS needs to talk to 32-bit processes so do the LPC conversion */
30 if (SmpIsWow64Process())
31 {
32 return SmpWow64ExecPgm(SmApiPort, ProcessInformation, DebugFlag);
33 }
34 #endif
35
36 /* Initialize the generic LPC header */
37 SmApiMsg.h.u2.ZeroInit = 0;
38 SmApiMsg.h.u1.s1.DataLength = sizeof(SM_EXEC_PGM_MSG) + 8;
39 SmApiMsg.h.u1.s1.TotalLength = sizeof(SmApiMsg);
40
41 /* Initialize this specific API's parameters */
42 SmApiMsg.ApiNumber = SmpExecPgmApi;
43 RtlCopyMemory(&SmApiMsg.u.ExecPgm.ProcessInformation,
44 ProcessInformation,
45 sizeof(SmApiMsg.u.ExecPgm.ProcessInformation));
46 SmApiMsg.u.ExecPgm.DebugFlag = DebugFlag;
47
48 /* Send the message to SMSS */
49 Status = NtRequestWaitReplyPort(SmApiPort, &SmApiMsg.h, &SmApiMsg.h);
50 if (!NT_SUCCESS(Status))
51 {
52 DPRINT1("SmExecPgm: NtRequestWaitReply Failed %lx\n", Status);
53 }
54 else
55 {
56 /* Upon success, we use the API's return value */
57 Status = SmApiMsg.ReturnValue;
58 }
59
60 /* Close the handles that the parent passed in and return status */
61 NtClose(ProcessInformation->ProcessHandle);
62 NtClose(ProcessInformation->ThreadHandle);
63 return Status;
64 }
65
66 NTSTATUS
67 NTAPI
68 SmConnectToSm(IN PUNICODE_STRING SbApiPortName,
69 IN HANDLE SbApiPort,
70 IN ULONG ImageType,
71 OUT PHANDLE SmApiPort)
72 {
73 NTSTATUS Status;
74 SB_CONNECTION_INFO ConnectInfo;
75 UNICODE_STRING PortName;
76 SECURITY_QUALITY_OF_SERVICE SecurityQos;
77 ULONG ConnectInfoLength = sizeof(ConnectInfo);
78
79 /* Setup the QoS structure */
80 SecurityQos.ImpersonationLevel = SecurityIdentification;
81 SecurityQos.ContextTrackingMode = SECURITY_DYNAMIC_TRACKING;
82 SecurityQos.EffectiveOnly = TRUE;
83
84 /* Set the SM API port name */
85 RtlInitUnicodeString(&PortName, L"\\SmApiPort");
86
87 /* Check if this is a client connecting to SMSS, or SMSS to itself */
88 if (SbApiPortName)
89 {
90 /* A client SB port as well as an image type must be present */
91 if (!(SbApiPort) || !(ImageType)) return STATUS_INVALID_PARAMETER_MIX;
92
93 /* Copy the client port name, and NULL-terminate it */
94 RtlCopyMemory(ConnectInfo.SbApiPortName,
95 SbApiPortName->Buffer,
96 SbApiPortName->Length);
97 ConnectInfo.SbApiPortName[SbApiPortName->Length /
98 sizeof(WCHAR)] = UNICODE_NULL;
99
100 /* Save the subsystem type */
101 ConnectInfo.SubsystemType = ImageType;
102 }
103 else
104 {
105 /* No client port, and the subsystem type is not set */
106 ConnectInfo.SbApiPortName[0] = UNICODE_NULL;
107 ConnectInfo.SubsystemType = IMAGE_SUBSYSTEM_UNKNOWN;
108 }
109
110 /* Connect to SMSS and exchange connection information */
111 Status = NtConnectPort(SmApiPort,
112 &PortName,
113 &SecurityQos,
114 NULL,
115 NULL,
116 NULL,
117 &ConnectInfo,
118 &ConnectInfoLength);
119 if (!NT_SUCCESS(Status))
120 {
121 DPRINT1("SmConnectToSm: Connect to Sm failed %lx\n", Status);
122 }
123 else
124 {
125 /* Treat a warning or informational status as success */
126 Status = STATUS_SUCCESS;
127 }
128
129 /* Return if the connection was successful or not */
130 return Status;
131 }
132
133 NTSTATUS
134 NTAPI
135 SmSessionComplete(IN HANDLE SmApiPort,
136 IN ULONG SessionId,
137 IN NTSTATUS SessionStatus)
138 {
139 NTSTATUS Status;
140 SM_API_MSG ApiMessage;
141 PSM_SESSION_COMPLETE_MSG SessionComplete = &ApiMessage.u.SessionComplete;
142
143 /* Set the message data */
144 SessionComplete->SessionId = SessionId;
145 SessionComplete->SessionStatus = SessionStatus;
146
147 /* Set the API Message Port Message header */
148 ApiMessage.ApiNumber = SmpSessionCompleteApi;
149 ApiMessage.h.u1.s1.DataLength = sizeof(SM_SESSION_COMPLETE_MSG) + 8;
150 ApiMessage.h.u1.s1.TotalLength = sizeof(SM_API_MSG);
151 ApiMessage.h.u2.ZeroInit = 0;
152
153 /* Sent the message and wait for a reply */
154 Status = NtRequestWaitReplyPort(SmApiPort,
155 &ApiMessage.h,
156 &ApiMessage.h);
157 if (NT_SUCCESS(Status))
158 {
159 /* Return the real status */
160 Status = ApiMessage.ReturnValue;
161 }
162 else
163 {
164 DPRINT1("SmCompleteSession: NtRequestWaitReply failed\n");
165 }
166
167 /* Return status */
168 return Status;
169 }