kernel32: Add GetCPInfo test.
[wine] / dlls / rpcrt4 / rpc_epmap.c
1 /*
2  * RPC endpoint mapper
3  *
4  * Copyright 2002 Greg Turner
5  * Copyright 2001 Ove Kåven, TransGaming Technologies
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  *
21  * TODO:
22  *  - actually do things right
23  */
24
25 #include <stdarg.h>
26 #include <stdio.h>
27 #include <string.h>
28
29 #include "windef.h"
30 #include "winbase.h"
31 #include "winerror.h"
32 #include "winreg.h"
33
34 #include "rpc.h"
35
36 #include "wine/debug.h"
37
38 #include "rpc_binding.h"
39 #include "epm_towers.h"
40
41 WINE_DEFAULT_DEBUG_CHANNEL(ole);
42
43 /* The "real" RPC portmapper endpoints that I know of are:
44  *
45  *  ncadg_ip_udp: 135
46  *  ncacn_ip_tcp: 135
47  *  ncacn_np: \\pipe\epmapper (?)
48  *  ncalrpc: epmapper
49  *
50  * If the user's machine ran a DCE RPC daemon, it would
51  * probably be possible to connect to it, but there are many
52  * reasons not to, like:
53  *  - the user probably does *not* run one, and probably
54  *    shouldn't be forced to run one just for local COM
55  *  - very few Unix systems use DCE RPC... if they run a RPC
56  *    daemon at all, it's usually Sun RPC
57  *  - DCE RPC registrations are persistent and saved on disk,
58  *    while MS-RPC registrations are documented as non-persistent
59  *    and stored only in RAM, and auto-destroyed when the process
60  *    dies (something DCE RPC can't do)
61  *
62  * Of course, if the user *did* want to run a DCE RPC daemon anyway,
63  * there would be interoperability advantages, like the possibility
64  * of running a fully functional DCOM server using Wine...
65  */
66
67 /***********************************************************************
68  *             RpcEpRegisterA (RPCRT4.@)
69  */
70 RPC_STATUS WINAPI RpcEpRegisterA( RPC_IF_HANDLE IfSpec, RPC_BINDING_VECTOR *BindingVector,
71                                   UUID_VECTOR *UuidVector, RPC_CSTR Annotation )
72 {
73   RPCSS_NP_MESSAGE msg;
74   RPCSS_NP_REPLY reply;
75   char *vardata_payload, *vp;
76   PRPC_SERVER_INTERFACE If = (PRPC_SERVER_INTERFACE)IfSpec;
77   unsigned long c;
78   RPC_STATUS rslt = RPC_S_OK;
79
80   TRACE("(%p,%p,%p,%s)\n", IfSpec, BindingVector, UuidVector, debugstr_a((char*)Annotation));
81   TRACE(" ifid=%s\n", debugstr_guid(&If->InterfaceId.SyntaxGUID));
82   for (c=0; c<BindingVector->Count; c++) {
83     RpcBinding* bind = (RpcBinding*)(BindingVector->BindingH[c]);
84     TRACE(" protseq[%ld]=%s\n", c, debugstr_a(bind->Protseq));
85     TRACE(" endpoint[%ld]=%s\n", c, debugstr_a(bind->Endpoint));
86   }
87   if (UuidVector) {
88     for (c=0; c<UuidVector->Count; c++)
89       TRACE(" obj[%ld]=%s\n", c, debugstr_guid(UuidVector->Uuid[c]));
90   }
91
92   /* FIXME: Do something with annotation. */
93
94   /* construct the message to rpcss */
95   msg.message_type = RPCSS_NP_MESSAGE_TYPEID_REGISTEREPMSG;
96   msg.message.registerepmsg.iface = If->InterfaceId;
97   msg.message.registerepmsg.no_replace = 0;
98
99   msg.message.registerepmsg.object_count = (UuidVector) ? UuidVector->Count : 0;
100   msg.message.registerepmsg.binding_count = BindingVector->Count;
101
102   /* calculate vardata payload size */
103   msg.vardata_payload_size = msg.message.registerepmsg.object_count * sizeof(UUID);
104   for (c=0; c < msg.message.registerepmsg.binding_count; c++) {
105     RpcBinding *bind = (RpcBinding *)(BindingVector->BindingH[c]);
106     msg.vardata_payload_size += strlen(bind->Protseq) + 1;
107     msg.vardata_payload_size += strlen(bind->Endpoint) + 1;
108   }
109
110   /* allocate the payload buffer */
111   vp = vardata_payload = LocalAlloc(LPTR, msg.vardata_payload_size);
112   if (!vardata_payload)
113     return RPC_S_OUT_OF_MEMORY;
114
115   /* populate the payload data */
116   for (c=0; c < msg.message.registerepmsg.object_count; c++) {
117     CopyMemory(vp, UuidVector->Uuid[c], sizeof(UUID));
118     vp += sizeof(UUID);
119   }
120
121   for (c=0; c < msg.message.registerepmsg.binding_count; c++) {
122     RpcBinding *bind = (RpcBinding*)(BindingVector->BindingH[c]);
123     unsigned long pslen = strlen(bind->Protseq) + 1, eplen = strlen(bind->Endpoint) + 1;
124     CopyMemory(vp, bind->Protseq, pslen);
125     vp += pslen;
126     CopyMemory(vp, bind->Endpoint, eplen);
127     vp += eplen;
128   }
129
130   /* send our request */
131   if (!RPCRT4_RPCSSOnDemandCall(&msg, vardata_payload, &reply))
132     rslt = RPC_S_OUT_OF_MEMORY;
133
134   /* free the payload buffer */
135   LocalFree(vardata_payload);
136
137   return rslt;
138 }
139
140 /***********************************************************************
141  *             RpcEpUnregister (RPCRT4.@)
142  */
143 RPC_STATUS WINAPI RpcEpUnregister( RPC_IF_HANDLE IfSpec, RPC_BINDING_VECTOR *BindingVector,
144                                    UUID_VECTOR *UuidVector )
145 {
146   RPCSS_NP_MESSAGE msg;
147   RPCSS_NP_REPLY reply;
148   char *vardata_payload, *vp;
149   PRPC_SERVER_INTERFACE If = (PRPC_SERVER_INTERFACE)IfSpec;
150   unsigned long c;
151   RPC_STATUS rslt = RPC_S_OK;
152
153   TRACE("(%p,%p,%p)\n", IfSpec, BindingVector, UuidVector);
154   TRACE(" ifid=%s\n", debugstr_guid(&If->InterfaceId.SyntaxGUID));
155   for (c=0; c<BindingVector->Count; c++) {
156     RpcBinding* bind = (RpcBinding*)(BindingVector->BindingH[c]);
157     TRACE(" protseq[%ld]=%s\n", c, debugstr_a(bind->Protseq));
158     TRACE(" endpoint[%ld]=%s\n", c, debugstr_a(bind->Endpoint));
159   }
160   if (UuidVector) {
161     for (c=0; c<UuidVector->Count; c++)
162       TRACE(" obj[%ld]=%s\n", c, debugstr_guid(UuidVector->Uuid[c]));
163   }
164
165   /* construct the message to rpcss */
166   msg.message_type = RPCSS_NP_MESSAGE_TYPEID_UNREGISTEREPMSG;
167   msg.message.unregisterepmsg.iface = If->InterfaceId;
168
169   msg.message.unregisterepmsg.object_count = (UuidVector) ? UuidVector->Count : 0;
170   msg.message.unregisterepmsg.binding_count = BindingVector->Count;
171
172   /* calculate vardata payload size */
173   msg.vardata_payload_size = msg.message.unregisterepmsg.object_count * sizeof(UUID);
174   for (c=0; c < msg.message.unregisterepmsg.binding_count; c++) {
175     RpcBinding *bind = (RpcBinding *)(BindingVector->BindingH[c]);
176     msg.vardata_payload_size += strlen(bind->Protseq) + 1;
177     msg.vardata_payload_size += strlen(bind->Endpoint) + 1;
178   }
179
180   /* allocate the payload buffer */
181   vp = vardata_payload = LocalAlloc(LPTR, msg.vardata_payload_size);
182   if (!vardata_payload)
183     return RPC_S_OUT_OF_MEMORY;
184
185   /* populate the payload data */
186   for (c=0; c < msg.message.unregisterepmsg.object_count; c++) {
187     CopyMemory(vp, UuidVector->Uuid[c], sizeof(UUID));
188     vp += sizeof(UUID);
189   }
190
191   for (c=0; c < msg.message.unregisterepmsg.binding_count; c++) {
192     RpcBinding *bind = (RpcBinding*)(BindingVector->BindingH[c]);
193     unsigned long pslen = strlen(bind->Protseq) + 1, eplen = strlen(bind->Endpoint) + 1;
194     CopyMemory(vp, bind->Protseq, pslen);
195     vp += pslen;
196     CopyMemory(vp, bind->Endpoint, eplen);
197     vp += eplen;
198   }
199
200   /* send our request */
201   if (!RPCRT4_RPCSSOnDemandCall(&msg, vardata_payload, &reply))
202     rslt = RPC_S_OUT_OF_MEMORY;
203
204   /* free the payload buffer */
205   LocalFree(vardata_payload);
206
207   return rslt;
208 }
209
210 /***********************************************************************
211  *             RpcEpResolveBinding (RPCRT4.@)
212  */
213 RPC_STATUS WINAPI RpcEpResolveBinding( RPC_BINDING_HANDLE Binding, RPC_IF_HANDLE IfSpec )
214 {
215   RPCSS_NP_MESSAGE msg;
216   RPCSS_NP_REPLY reply;
217   PRPC_CLIENT_INTERFACE If = (PRPC_CLIENT_INTERFACE)IfSpec;
218   RpcBinding* bind = (RpcBinding*)Binding;
219
220   TRACE("(%p,%p)\n", Binding, IfSpec);
221   TRACE(" protseq=%s\n", debugstr_a(bind->Protseq));
222   TRACE(" obj=%s\n", debugstr_guid(&bind->ObjectUuid));
223   TRACE(" ifid=%s\n", debugstr_guid(&If->InterfaceId.SyntaxGUID));
224
225   /* FIXME: totally untested */
226
227   /* just return for fully bound handles */
228   if (bind->Endpoint && (bind->Endpoint[0] != '\0'))
229     return RPC_S_OK;
230
231   /* construct the message to rpcss */
232   msg.message_type = RPCSS_NP_MESSAGE_TYPEID_RESOLVEEPMSG;
233   msg.message.resolveepmsg.iface = If->InterfaceId;
234   msg.message.resolveepmsg.object = bind->ObjectUuid;
235  
236   msg.vardata_payload_size = strlen(bind->Protseq) + 1;
237
238   /* send the message */
239   if (!RPCRT4_RPCSSOnDemandCall(&msg, bind->Protseq, &reply))
240     return RPC_S_OUT_OF_MEMORY;
241
242   /* empty-string result means not registered */
243   if (reply.as_string[0] == '\0')
244     return EPT_S_NOT_REGISTERED;
245   
246   /* otherwise we fully bind the handle & return RPC_S_OK */
247   return RPCRT4_ResolveBinding(Binding, reply.as_string);
248 }
249
250 typedef unsigned int unsigned32;
251 typedef struct twr_t
252     {
253     unsigned32 tower_length;
254     /* [size_is] */ BYTE tower_octet_string[ 1 ];
255     }   twr_t;
256
257 /***********************************************************************
258  *             TowerExplode (RPCRT4.@)
259  */
260 RPC_STATUS WINAPI TowerExplode(
261     const twr_t *tower, PRPC_SYNTAX_IDENTIFIER object, PRPC_SYNTAX_IDENTIFIER syntax,
262     char **protseq, char **endpoint, char **address)
263 {
264     size_t tower_size;
265     RPC_STATUS status;
266     const unsigned char *p;
267     u_int16 floor_count;
268     const twr_uuid_floor_t *object_floor;
269     const twr_uuid_floor_t *syntax_floor;
270
271     if (protseq)
272         *protseq = NULL;
273     if (endpoint)
274         *endpoint = NULL;
275     if (address)
276         *address = NULL;
277
278     tower_size = tower->tower_length;
279
280     if (tower_size < sizeof(u_int16))
281         return EPT_S_NOT_REGISTERED;
282
283     p = &tower->tower_octet_string[0];
284
285     floor_count = *(const u_int16 *)p;
286     p += sizeof(u_int16);
287     tower_size -= sizeof(u_int16);
288     TRACE("floor_count: %d\n", floor_count);
289     /* FIXME: should we do something with the floor count? at the moment we don't */
290
291     if (tower_size < sizeof(*object_floor) + sizeof(*syntax_floor))
292         return EPT_S_NOT_REGISTERED;
293
294     object_floor = (const twr_uuid_floor_t *)p;
295     p += sizeof(*object_floor);
296     tower_size -= sizeof(*object_floor);
297     syntax_floor = (const twr_uuid_floor_t *)p;
298     p += sizeof(*syntax_floor);
299     tower_size -= sizeof(*syntax_floor);
300
301     if ((object_floor->count_lhs != sizeof(object_floor->protid) +
302         sizeof(object_floor->uuid) + sizeof(object_floor->major_version)) ||
303         (object_floor->protid != EPM_PROTOCOL_UUID) ||
304         (object_floor->count_rhs != sizeof(object_floor->minor_version)))
305         return EPT_S_NOT_REGISTERED;
306
307     if ((syntax_floor->count_lhs != sizeof(syntax_floor->protid) +
308         sizeof(syntax_floor->uuid) + sizeof(syntax_floor->major_version)) ||
309         (syntax_floor->protid != EPM_PROTOCOL_UUID) ||
310         (syntax_floor->count_rhs != sizeof(syntax_floor->minor_version)))
311         return EPT_S_NOT_REGISTERED;
312
313     status = RpcTransport_ParseTopOfTower(p, tower_size, protseq, address, endpoint);
314     if ((status == RPC_S_OK) && syntax && object)
315     {
316         syntax->SyntaxGUID = syntax_floor->uuid;
317         syntax->SyntaxVersion.MajorVersion = syntax_floor->major_version;
318         syntax->SyntaxVersion.MinorVersion = syntax_floor->minor_version;
319         object->SyntaxGUID = object_floor->uuid;
320         object->SyntaxVersion.MajorVersion = object_floor->major_version;
321         object->SyntaxVersion.MinorVersion = object_floor->minor_version;
322     }
323     return status;
324 }
325
326 /***********************************************************************
327  *             TowerConstruct (RPCRT4.@)
328  */
329 RPC_STATUS WINAPI TowerConstruct(
330     const RPC_SYNTAX_IDENTIFIER *object, const RPC_SYNTAX_IDENTIFIER *syntax,
331     const char *protseq, const char *endpoint, const char *address,
332     twr_t **tower)
333 {
334     size_t tower_size;
335     RPC_STATUS status;
336     unsigned char *p;
337     twr_uuid_floor_t *object_floor;
338     twr_uuid_floor_t *syntax_floor;
339
340     *tower = NULL;
341
342     status = RpcTransport_GetTopOfTower(NULL, &tower_size, protseq, address, endpoint);
343
344     if (status != RPC_S_OK)
345         return status;
346
347     tower_size += sizeof(u_int16) + sizeof(*object_floor) + sizeof(*syntax_floor);
348     *tower = I_RpcAllocate(FIELD_OFFSET(twr_t, tower_octet_string[tower_size]));
349     if (!*tower)
350         return RPC_S_OUT_OF_RESOURCES;
351
352     (*tower)->tower_length = tower_size;
353     p = &(*tower)->tower_octet_string[0];
354     *(u_int16 *)p = 5; /* number of floors */
355     p += sizeof(u_int16);
356     object_floor = (twr_uuid_floor_t *)p;
357     p += sizeof(*object_floor);
358     syntax_floor = (twr_uuid_floor_t *)p;
359     p += sizeof(*syntax_floor);
360
361     object_floor->count_lhs = sizeof(object_floor->protid) + sizeof(object_floor->uuid) +
362                               sizeof(object_floor->major_version);
363     object_floor->protid = EPM_PROTOCOL_UUID;
364     object_floor->count_rhs = sizeof(object_floor->minor_version);
365     object_floor->uuid = object->SyntaxGUID;
366     object_floor->major_version = object->SyntaxVersion.MajorVersion;
367     object_floor->minor_version = object->SyntaxVersion.MinorVersion;
368
369     syntax_floor->count_lhs = sizeof(syntax_floor->protid) + sizeof(syntax_floor->uuid) +
370                               sizeof(syntax_floor->major_version);
371     syntax_floor->protid = EPM_PROTOCOL_UUID;
372     syntax_floor->count_rhs = sizeof(syntax_floor->minor_version);
373     syntax_floor->uuid = syntax->SyntaxGUID;
374     syntax_floor->major_version = syntax->SyntaxVersion.MajorVersion;
375     syntax_floor->minor_version = syntax->SyntaxVersion.MinorVersion;
376
377     status = RpcTransport_GetTopOfTower(p, &tower_size, protseq, address, endpoint);
378     if (status != RPC_S_OK)
379     {
380         I_RpcFree(*tower);
381         *tower = NULL;
382         return status;
383     }
384     return RPC_S_OK;
385 }