rpcrt4: Free the memory for a registered interface that is in use and when WaitForCal...
[wine] / dlls / rpcrt4 / rpc_server.c
1 /*
2  * RPC server API
3  *
4  * Copyright 2001 Ove Kåven, TransGaming Technologies
5  * Copyright 2004 Filip Navara
6  * Copyright 2006-2008 Robert Shearman (for CodeWeavers)
7  *
8  * This library is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * This library is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with this library; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21  */
22
23 #include "config.h"
24 #include "wine/port.h"
25
26 #include <stdarg.h>
27 #include <stdio.h>
28 #include <string.h>
29 #include <assert.h>
30
31 #include "windef.h"
32 #include "winbase.h"
33 #include "winerror.h"
34
35 #include "rpc.h"
36 #include "rpcndr.h"
37 #include "excpt.h"
38
39 #include "wine/debug.h"
40 #include "wine/exception.h"
41
42 #include "rpc_server.h"
43 #include "rpc_assoc.h"
44 #include "rpc_message.h"
45 #include "rpc_defs.h"
46 #include "ncastatus.h"
47
48 WINE_DEFAULT_DEBUG_CHANNEL(rpc);
49
50 typedef struct _RpcPacket
51 {
52   struct _RpcConnection* conn;
53   RpcPktHdr* hdr;
54   RPC_MESSAGE* msg;
55 } RpcPacket;
56
57 typedef struct _RpcObjTypeMap
58 {
59   /* FIXME: a hash table would be better. */
60   struct _RpcObjTypeMap *next;
61   UUID Object;
62   UUID Type;
63 } RpcObjTypeMap;
64
65 static RpcObjTypeMap *RpcObjTypeMaps;
66
67 /* list of type RpcServerProtseq */
68 static struct list protseqs = LIST_INIT(protseqs);
69 static struct list server_interfaces = LIST_INIT(server_interfaces);
70
71 static CRITICAL_SECTION server_cs;
72 static CRITICAL_SECTION_DEBUG server_cs_debug =
73 {
74     0, 0, &server_cs,
75     { &server_cs_debug.ProcessLocksList, &server_cs_debug.ProcessLocksList },
76       0, 0, { (DWORD_PTR)(__FILE__ ": server_cs") }
77 };
78 static CRITICAL_SECTION server_cs = { &server_cs_debug, -1, 0, 0, 0, 0 };
79
80 static CRITICAL_SECTION listen_cs;
81 static CRITICAL_SECTION_DEBUG listen_cs_debug =
82 {
83     0, 0, &listen_cs,
84     { &listen_cs_debug.ProcessLocksList, &listen_cs_debug.ProcessLocksList },
85       0, 0, { (DWORD_PTR)(__FILE__ ": listen_cs") }
86 };
87 static CRITICAL_SECTION listen_cs = { &listen_cs_debug, -1, 0, 0, 0, 0 };
88
89 /* whether the server is currently listening */
90 static BOOL std_listen;
91 /* number of manual listeners (calls to RpcServerListen) */
92 static LONG manual_listen_count;
93 /* total listeners including auto listeners */
94 static LONG listen_count;
95
96 static UUID uuid_nil;
97
98 static inline RpcObjTypeMap *LookupObjTypeMap(UUID *ObjUuid)
99 {
100   RpcObjTypeMap *rslt = RpcObjTypeMaps;
101   RPC_STATUS dummy;
102
103   while (rslt) {
104     if (! UuidCompare(ObjUuid, &rslt->Object, &dummy)) break;
105     rslt = rslt->next;
106   }
107
108   return rslt;
109 }
110
111 static inline UUID *LookupObjType(UUID *ObjUuid)
112 {
113   RpcObjTypeMap *map = LookupObjTypeMap(ObjUuid);
114   if (map)
115     return &map->Type;
116   else
117     return &uuid_nil;
118 }
119
120 static RpcServerInterface* RPCRT4_find_interface(UUID* object,
121                                                  const RPC_SYNTAX_IDENTIFIER* if_id,
122                                                  BOOL check_object)
123 {
124   UUID* MgrType = NULL;
125   RpcServerInterface* cif;
126   RPC_STATUS status;
127
128   if (check_object)
129     MgrType = LookupObjType(object);
130   EnterCriticalSection(&server_cs);
131   LIST_FOR_EACH_ENTRY(cif, &server_interfaces, RpcServerInterface, entry) {
132     if (!memcmp(if_id, &cif->If->InterfaceId, sizeof(RPC_SYNTAX_IDENTIFIER)) &&
133         (check_object == FALSE || UuidEqual(MgrType, &cif->MgrTypeUuid, &status)) &&
134         std_listen) {
135       InterlockedIncrement(&cif->CurrentCalls);
136       break;
137     }
138   }
139   LeaveCriticalSection(&server_cs);
140   if (&cif->entry == &server_interfaces) cif = NULL;
141   TRACE("returning %p for object %s, if_id { %d.%d %s }\n", cif,
142     debugstr_guid(object), if_id->SyntaxVersion.MajorVersion,
143     if_id->SyntaxVersion.MinorVersion, debugstr_guid(&if_id->SyntaxGUID));
144   return cif;
145 }
146
147 static void RPCRT4_release_server_interface(RpcServerInterface *sif)
148 {
149   if (!InterlockedDecrement(&sif->CurrentCalls) &&
150       sif->Delete) {
151     /* sif must have been removed from server_interfaces before
152      * CallsCompletedEvent is set */
153     if (sif->CallsCompletedEvent)
154       SetEvent(sif->CallsCompletedEvent);
155     HeapFree(GetProcessHeap(), 0, sif);
156   }
157 }
158
159 static RPC_STATUS process_bind_packet(RpcConnection *conn, RpcPktBindHdr *hdr, RPC_MESSAGE *msg)
160 {
161   RPC_STATUS status;
162   RpcServerInterface* sif;
163   RpcPktHdr *response = NULL;
164
165   /* FIXME: do more checks! */
166   if (hdr->max_tsize < RPC_MIN_PACKET_SIZE ||
167       !UuidIsNil(&conn->ActiveInterface.SyntaxGUID, &status) ||
168       conn->server_binding) {
169     TRACE("packet size less than min size, or active interface syntax guid non-null\n");
170     sif = NULL;
171   } else {
172     /* create temporary binding */
173     if (RPCRT4_MakeBinding(&conn->server_binding, conn) == RPC_S_OK &&
174         RpcServerAssoc_GetAssociation(rpcrt4_conn_get_name(conn),
175                                       conn->NetworkAddr, conn->Endpoint,
176                                       conn->NetworkOptions,
177                                       hdr->assoc_gid,
178                                       &conn->server_binding->Assoc) == RPC_S_OK)
179       sif = RPCRT4_find_interface(NULL, &hdr->abstract, FALSE);
180     else
181       sif = NULL;
182   }
183   if (sif == NULL) {
184     TRACE("rejecting bind request on connection %p\n", conn);
185     /* Report failure to client. */
186     response = RPCRT4_BuildBindNackHeader(NDR_LOCAL_DATA_REPRESENTATION,
187                                           RPC_VER_MAJOR, RPC_VER_MINOR);
188   } else {
189     TRACE("accepting bind request on connection %p for %s\n", conn,
190           debugstr_guid(&hdr->abstract.SyntaxGUID));
191
192     /* accept. */
193     response = RPCRT4_BuildBindAckHeader(NDR_LOCAL_DATA_REPRESENTATION,
194                                          RPC_MAX_PACKET_SIZE,
195                                          RPC_MAX_PACKET_SIZE,
196                                          conn->server_binding->Assoc->assoc_group_id,
197                                          conn->Endpoint,
198                                          RESULT_ACCEPT, REASON_NONE,
199                                          &sif->If->TransferSyntax);
200
201     /* save the interface for later use */
202     conn->ActiveInterface = hdr->abstract;
203     conn->MaxTransmissionSize = hdr->max_tsize;
204
205     RPCRT4_release_server_interface(sif);
206   }
207
208   if (response)
209     status = RPCRT4_Send(conn, response, NULL, 0);
210   else
211     status = ERROR_OUTOFMEMORY;
212   RPCRT4_FreeHeader(response);
213
214   return status;
215 }
216
217 static RPC_STATUS process_request_packet(RpcConnection *conn, RpcPktRequestHdr *hdr, RPC_MESSAGE *msg)
218 {
219   RPC_STATUS status;
220   RpcPktHdr *response = NULL;
221   RpcServerInterface* sif;
222   RPC_DISPATCH_FUNCTION func;
223   BOOL exception;
224   UUID *object_uuid;
225   NDR_SCONTEXT context_handle;
226   void *buf = msg->Buffer;
227
228   /* fail if the connection isn't bound with an interface */
229   if (UuidIsNil(&conn->ActiveInterface.SyntaxGUID, &status)) {
230     /* FIXME: should send BindNack instead */
231     response = RPCRT4_BuildFaultHeader(NDR_LOCAL_DATA_REPRESENTATION,
232                                        status);
233
234     RPCRT4_Send(conn, response, NULL, 0);
235     RPCRT4_FreeHeader(response);
236     return RPC_S_OK;
237   }
238
239   if (hdr->common.flags & RPC_FLG_OBJECT_UUID) {
240     object_uuid = (UUID*)(hdr + 1);
241   } else {
242     object_uuid = NULL;
243   }
244
245   sif = RPCRT4_find_interface(object_uuid, &conn->ActiveInterface, TRUE);
246   if (!sif) {
247     WARN("interface %s no longer registered, returning fault packet\n", debugstr_guid(&conn->ActiveInterface.SyntaxGUID));
248     response = RPCRT4_BuildFaultHeader(NDR_LOCAL_DATA_REPRESENTATION,
249                                        NCA_S_UNK_IF);
250
251     RPCRT4_Send(conn, response, NULL, 0);
252     RPCRT4_FreeHeader(response);
253     return RPC_S_OK;
254   }
255   msg->RpcInterfaceInformation = sif->If;
256   /* copy the endpoint vector from sif to msg so that midl-generated code will use it */
257   msg->ManagerEpv = sif->MgrEpv;
258   if (object_uuid != NULL) {
259     RPCRT4_SetBindingObject(msg->Handle, object_uuid);
260   }
261
262   /* find dispatch function */
263   msg->ProcNum = hdr->opnum;
264   if (sif->Flags & RPC_IF_OLE) {
265     /* native ole32 always gives us a dispatch table with a single entry
266     * (I assume that's a wrapper for IRpcStubBuffer::Invoke) */
267     func = *sif->If->DispatchTable->DispatchTable;
268   } else {
269     if (msg->ProcNum >= sif->If->DispatchTable->DispatchTableCount) {
270       WARN("invalid procnum (%d/%d)\n", msg->ProcNum, sif->If->DispatchTable->DispatchTableCount);
271       response = RPCRT4_BuildFaultHeader(NDR_LOCAL_DATA_REPRESENTATION,
272                                          NCA_S_OP_RNG_ERROR);
273
274       RPCRT4_Send(conn, response, NULL, 0);
275       RPCRT4_FreeHeader(response);
276     }
277     func = sif->If->DispatchTable->DispatchTable[msg->ProcNum];
278   }
279
280   /* put in the drep. FIXME: is this more universally applicable?
281     perhaps we should move this outward... */
282   msg->DataRepresentation =
283     MAKELONG( MAKEWORD(hdr->common.drep[0], hdr->common.drep[1]),
284               MAKEWORD(hdr->common.drep[2], hdr->common.drep[3]));
285
286   exception = FALSE;
287
288   /* dispatch */
289   RPCRT4_SetThreadCurrentCallHandle(msg->Handle);
290   __TRY {
291     if (func) func(msg);
292   } __EXCEPT_ALL {
293     WARN("exception caught with code 0x%08x = %d\n", GetExceptionCode(), GetExceptionCode());
294     exception = TRUE;
295     if (GetExceptionCode() == STATUS_ACCESS_VIOLATION)
296       status = ERROR_NOACCESS;
297     else
298       status = GetExceptionCode();
299     response = RPCRT4_BuildFaultHeader(msg->DataRepresentation,
300                                        RPC2NCA_STATUS(status));
301   } __ENDTRY
302     RPCRT4_SetThreadCurrentCallHandle(NULL);
303
304   /* release any unmarshalled context handles */
305   while ((context_handle = RPCRT4_PopThreadContextHandle()) != NULL)
306     RpcServerAssoc_ReleaseContextHandle(conn->server_binding->Assoc, context_handle, TRUE);
307
308   if (!exception)
309     response = RPCRT4_BuildResponseHeader(msg->DataRepresentation,
310                                           msg->BufferLength);
311
312   /* send response packet */
313   if (response) {
314     status = RPCRT4_Send(conn, response, exception ? NULL : msg->Buffer,
315                          exception ? 0 : msg->BufferLength);
316     RPCRT4_FreeHeader(response);
317   } else
318     ERR("out of memory\n");
319
320   msg->RpcInterfaceInformation = NULL;
321   RPCRT4_release_server_interface(sif);
322
323   if (msg->Buffer == buf) buf = NULL;
324   TRACE("freeing Buffer=%p\n", buf);
325   I_RpcFree(buf);
326
327   return status;
328 }
329
330 static void RPCRT4_process_packet(RpcConnection* conn, RpcPktHdr* hdr, RPC_MESSAGE* msg)
331 {
332   RPC_STATUS status;
333
334   msg->Handle = (RPC_BINDING_HANDLE)conn->server_binding;
335
336   switch (hdr->common.ptype) {
337     case PKT_BIND:
338       TRACE("got bind packet\n");
339
340       status = process_bind_packet(conn, &hdr->bind, msg);
341       break;
342
343     case PKT_REQUEST:
344       TRACE("got request packet\n");
345
346       status = process_request_packet(conn, &hdr->request, msg);
347       break;
348
349     default:
350       FIXME("unhandled packet type %u\n", hdr->common.ptype);
351       break;
352   }
353
354   /* clean up */
355   I_RpcFreeBuffer(msg);
356   RPCRT4_FreeHeader(hdr);
357   HeapFree(GetProcessHeap(), 0, msg);
358 }
359
360 static DWORD CALLBACK RPCRT4_worker_thread(LPVOID the_arg)
361 {
362   RpcPacket *pkt = the_arg;
363   RPCRT4_process_packet(pkt->conn, pkt->hdr, pkt->msg);
364   HeapFree(GetProcessHeap(), 0, pkt);
365   return 0;
366 }
367
368 static DWORD CALLBACK RPCRT4_io_thread(LPVOID the_arg)
369 {
370   RpcConnection* conn = (RpcConnection*)the_arg;
371   RpcPktHdr *hdr;
372   RPC_MESSAGE *msg;
373   RPC_STATUS status;
374   RpcPacket *packet;
375
376   TRACE("(%p)\n", conn);
377
378   for (;;) {
379     msg = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(RPC_MESSAGE));
380
381     status = RPCRT4_Receive(conn, &hdr, msg);
382     if (status != RPC_S_OK) {
383       WARN("receive failed with error %lx\n", status);
384       HeapFree(GetProcessHeap(), 0, msg);
385       break;
386     }
387
388     packet = HeapAlloc(GetProcessHeap(), 0, sizeof(RpcPacket));
389     if (!packet) {
390       I_RpcFree(msg->Buffer);
391       HeapFree(GetProcessHeap(), 0, msg);
392       break;
393     }
394     packet->conn = conn;
395     packet->hdr = hdr;
396     packet->msg = msg;
397     if (!QueueUserWorkItem(RPCRT4_worker_thread, packet, WT_EXECUTELONGFUNCTION)) {
398       ERR("couldn't queue work item for worker thread, error was %d\n", GetLastError());
399       I_RpcFree(msg->Buffer);
400       HeapFree(GetProcessHeap(), 0, msg);
401       HeapFree(GetProcessHeap(), 0, packet);
402       break;
403     }
404
405     msg = NULL;
406   }
407   RPCRT4_DestroyConnection(conn);
408   return 0;
409 }
410
411 void RPCRT4_new_client(RpcConnection* conn)
412 {
413   HANDLE thread = CreateThread(NULL, 0, RPCRT4_io_thread, conn, 0, NULL);
414   if (!thread) {
415     DWORD err = GetLastError();
416     ERR("failed to create thread, error=%08x\n", err);
417     RPCRT4_DestroyConnection(conn);
418   }
419   /* we could set conn->thread, but then we'd have to make the io_thread wait
420    * for that, otherwise the thread might finish, destroy the connection, and
421    * free the memory we'd write to before we did, causing crashes and stuff -
422    * so let's implement that later, when we really need conn->thread */
423
424   CloseHandle( thread );
425 }
426
427 static DWORD CALLBACK RPCRT4_server_thread(LPVOID the_arg)
428 {
429   int res;
430   unsigned int count;
431   void *objs = NULL;
432   RpcServerProtseq* cps = the_arg;
433   RpcConnection* conn;
434   BOOL set_ready_event = FALSE;
435
436   TRACE("(the_arg == ^%p)\n", the_arg);
437
438   for (;;) {
439     objs = cps->ops->get_wait_array(cps, objs, &count);
440
441     if (set_ready_event)
442     {
443         /* signal to function that changed state that we are now sync'ed */
444         SetEvent(cps->server_ready_event);
445         set_ready_event = FALSE;
446     }
447
448     /* start waiting */
449     res = cps->ops->wait_for_new_connection(cps, count, objs);
450     if (res == -1)
451       break;
452     else if (res == 0)
453     {
454       if (!std_listen)
455       {
456         SetEvent(cps->server_ready_event);
457         break;
458       }
459       set_ready_event = TRUE;
460     }
461   }
462   cps->ops->free_wait_array(cps, objs);
463   EnterCriticalSection(&cps->cs);
464   /* close connections */
465   conn = cps->conn;
466   while (conn) {
467     RPCRT4_CloseConnection(conn);
468     conn = conn->Next;
469   }
470   LeaveCriticalSection(&cps->cs);
471   return 0;
472 }
473
474 /* tells the server thread that the state has changed and waits for it to
475  * make the changes */
476 static void RPCRT4_sync_with_server_thread(RpcServerProtseq *ps)
477 {
478   /* make sure we are the only thread sync'ing the server state, otherwise
479    * there is a race with the server thread setting an older state and setting
480    * the server_ready_event when the new state hasn't yet been applied */
481   WaitForSingleObject(ps->mgr_mutex, INFINITE);
482
483   ps->ops->signal_state_changed(ps);
484
485   /* wait for server thread to make the requested changes before returning */
486   WaitForSingleObject(ps->server_ready_event, INFINITE);
487
488   ReleaseMutex(ps->mgr_mutex);
489 }
490
491 static RPC_STATUS RPCRT4_start_listen_protseq(RpcServerProtseq *ps, BOOL auto_listen)
492 {
493   RPC_STATUS status = RPC_S_OK;
494   HANDLE server_thread;
495
496   EnterCriticalSection(&listen_cs);
497   if (ps->is_listening) goto done;
498
499   if (!ps->mgr_mutex) ps->mgr_mutex = CreateMutexW(NULL, FALSE, NULL);
500   if (!ps->server_ready_event) ps->server_ready_event = CreateEventW(NULL, FALSE, FALSE, NULL);
501   server_thread = CreateThread(NULL, 0, RPCRT4_server_thread, ps, 0, NULL);
502   if (!server_thread)
503   {
504     status = RPC_S_OUT_OF_RESOURCES;
505     goto done;
506   }
507   ps->is_listening = TRUE;
508   CloseHandle(server_thread);
509
510 done:
511   LeaveCriticalSection(&listen_cs);
512   return status;
513 }
514
515 static RPC_STATUS RPCRT4_start_listen(BOOL auto_listen)
516 {
517   RPC_STATUS status = RPC_S_ALREADY_LISTENING;
518   RpcServerProtseq *cps;
519
520   TRACE("\n");
521
522   EnterCriticalSection(&listen_cs);
523   if (auto_listen || (manual_listen_count++ == 0))
524   {
525     status = RPC_S_OK;
526     if (++listen_count == 1)
527       std_listen = TRUE;
528   }
529   LeaveCriticalSection(&listen_cs);
530
531   if (std_listen)
532   {
533     EnterCriticalSection(&server_cs);
534     LIST_FOR_EACH_ENTRY(cps, &protseqs, RpcServerProtseq, entry)
535     {
536       status = RPCRT4_start_listen_protseq(cps, TRUE);
537       if (status != RPC_S_OK)
538         break;
539       
540       /* make sure server is actually listening on the interface before
541        * returning */
542       RPCRT4_sync_with_server_thread(cps);
543     }
544     LeaveCriticalSection(&server_cs);
545   }
546
547   return status;
548 }
549
550 static void RPCRT4_stop_listen(BOOL auto_listen)
551 {
552   EnterCriticalSection(&listen_cs);
553   if (auto_listen || (--manual_listen_count == 0))
554   {
555     if (listen_count != 0 && --listen_count == 0) {
556       RpcServerProtseq *cps;
557
558       std_listen = FALSE;
559       LeaveCriticalSection(&listen_cs);
560
561       LIST_FOR_EACH_ENTRY(cps, &protseqs, RpcServerProtseq, entry)
562         RPCRT4_sync_with_server_thread(cps);
563
564       return;
565     }
566     assert(listen_count >= 0);
567   }
568   LeaveCriticalSection(&listen_cs);
569 }
570
571 static RPC_STATUS RPCRT4_use_protseq(RpcServerProtseq* ps, LPSTR endpoint)
572 {
573   RPC_STATUS status;
574
575   status = ps->ops->open_endpoint(ps, endpoint);
576   if (status != RPC_S_OK)
577     return status;
578
579   if (std_listen)
580   {
581     status = RPCRT4_start_listen_protseq(ps, FALSE);
582     if (status == RPC_S_OK)
583       RPCRT4_sync_with_server_thread(ps);
584   }
585
586   return status;
587 }
588
589 /***********************************************************************
590  *             RpcServerInqBindings (RPCRT4.@)
591  */
592 RPC_STATUS WINAPI RpcServerInqBindings( RPC_BINDING_VECTOR** BindingVector )
593 {
594   RPC_STATUS status;
595   DWORD count;
596   RpcServerProtseq* ps;
597   RpcConnection* conn;
598
599   if (BindingVector)
600     TRACE("(*BindingVector == ^%p)\n", *BindingVector);
601   else
602     ERR("(BindingVector == NULL!!?)\n");
603
604   EnterCriticalSection(&server_cs);
605   /* count connections */
606   count = 0;
607   LIST_FOR_EACH_ENTRY(ps, &protseqs, RpcServerProtseq, entry) {
608     EnterCriticalSection(&ps->cs);
609     conn = ps->conn;
610     while (conn) {
611       count++;
612       conn = conn->Next;
613     }
614     LeaveCriticalSection(&ps->cs);
615   }
616   if (count) {
617     /* export bindings */
618     *BindingVector = HeapAlloc(GetProcessHeap(), 0,
619                               sizeof(RPC_BINDING_VECTOR) +
620                               sizeof(RPC_BINDING_HANDLE)*(count-1));
621     (*BindingVector)->Count = count;
622     count = 0;
623     LIST_FOR_EACH_ENTRY(ps, &protseqs, RpcServerProtseq, entry) {
624       EnterCriticalSection(&ps->cs);
625       conn = ps->conn;
626       while (conn) {
627        RPCRT4_MakeBinding((RpcBinding**)&(*BindingVector)->BindingH[count],
628                           conn);
629        count++;
630        conn = conn->Next;
631       }
632       LeaveCriticalSection(&ps->cs);
633     }
634     status = RPC_S_OK;
635   } else {
636     *BindingVector = NULL;
637     status = RPC_S_NO_BINDINGS;
638   }
639   LeaveCriticalSection(&server_cs);
640   return status;
641 }
642
643 /***********************************************************************
644  *             RpcServerUseProtseqEpA (RPCRT4.@)
645  */
646 RPC_STATUS WINAPI RpcServerUseProtseqEpA( RPC_CSTR Protseq, UINT MaxCalls, RPC_CSTR Endpoint, LPVOID SecurityDescriptor )
647 {
648   RPC_POLICY policy;
649   
650   TRACE( "(%s,%u,%s,%p)\n", Protseq, MaxCalls, Endpoint, SecurityDescriptor );
651   
652   /* This should provide the default behaviour */
653   policy.Length        = sizeof( policy );
654   policy.EndpointFlags = 0;
655   policy.NICFlags      = 0;
656   
657   return RpcServerUseProtseqEpExA( Protseq, MaxCalls, Endpoint, SecurityDescriptor, &policy );
658 }
659
660 /***********************************************************************
661  *             RpcServerUseProtseqEpW (RPCRT4.@)
662  */
663 RPC_STATUS WINAPI RpcServerUseProtseqEpW( RPC_WSTR Protseq, UINT MaxCalls, RPC_WSTR Endpoint, LPVOID SecurityDescriptor )
664 {
665   RPC_POLICY policy;
666   
667   TRACE( "(%s,%u,%s,%p)\n", debugstr_w( Protseq ), MaxCalls, debugstr_w( Endpoint ), SecurityDescriptor );
668   
669   /* This should provide the default behaviour */
670   policy.Length        = sizeof( policy );
671   policy.EndpointFlags = 0;
672   policy.NICFlags      = 0;
673   
674   return RpcServerUseProtseqEpExW( Protseq, MaxCalls, Endpoint, SecurityDescriptor, &policy );
675 }
676
677 /***********************************************************************
678  *             alloc_serverprotoseq (internal)
679  *
680  * Must be called with server_cs held.
681  */
682 static RPC_STATUS alloc_serverprotoseq(UINT MaxCalls, char *Protseq, RpcServerProtseq **ps)
683 {
684   const struct protseq_ops *ops = rpcrt4_get_protseq_ops(Protseq);
685
686   if (!ops)
687   {
688     FIXME("protseq %s not supported\n", debugstr_a(Protseq));
689     return RPC_S_PROTSEQ_NOT_SUPPORTED;
690   }
691
692   *ps = ops->alloc();
693   if (!*ps)
694     return RPC_S_OUT_OF_RESOURCES;
695   (*ps)->MaxCalls = MaxCalls;
696   (*ps)->Protseq = Protseq;
697   (*ps)->ops = ops;
698   (*ps)->MaxCalls = 0;
699   (*ps)->conn = NULL;
700   InitializeCriticalSection(&(*ps)->cs);
701   (*ps)->is_listening = FALSE;
702   (*ps)->mgr_mutex = NULL;
703   (*ps)->server_ready_event = NULL;
704
705   list_add_head(&protseqs, &(*ps)->entry);
706
707   TRACE("new protseq %p created for %s\n", *ps, Protseq);
708
709   return RPC_S_OK;
710 }
711
712 /* Finds a given protseq or creates a new one if one doesn't already exist */
713 static RPC_STATUS RPCRT4_get_or_create_serverprotseq(UINT MaxCalls, char *Protseq, RpcServerProtseq **ps)
714 {
715     RPC_STATUS status;
716     RpcServerProtseq *cps;
717
718     EnterCriticalSection(&server_cs);
719
720     LIST_FOR_EACH_ENTRY(cps, &protseqs, RpcServerProtseq, entry)
721         if (!strcmp(cps->Protseq, Protseq))
722         {
723             TRACE("found existing protseq object for %s\n", Protseq);
724             *ps = cps;
725             LeaveCriticalSection(&server_cs);
726             return S_OK;
727         }
728
729     status = alloc_serverprotoseq(MaxCalls, Protseq, ps);
730
731     LeaveCriticalSection(&server_cs);
732
733     return status;
734 }
735
736 /***********************************************************************
737  *             RpcServerUseProtseqEpExA (RPCRT4.@)
738  */
739 RPC_STATUS WINAPI RpcServerUseProtseqEpExA( RPC_CSTR Protseq, UINT MaxCalls, RPC_CSTR Endpoint, LPVOID SecurityDescriptor,
740                                             PRPC_POLICY lpPolicy )
741 {
742   char *szps = (char*)Protseq, *szep = (char*)Endpoint;
743   RpcServerProtseq* ps;
744   RPC_STATUS status;
745
746   TRACE("(%s,%u,%s,%p,{%u,%lu,%lu})\n", debugstr_a(szps), MaxCalls,
747        debugstr_a(szep), SecurityDescriptor,
748        lpPolicy->Length, lpPolicy->EndpointFlags, lpPolicy->NICFlags );
749
750   status = RPCRT4_get_or_create_serverprotseq(MaxCalls, RPCRT4_strdupA(szps), &ps);
751   if (status != RPC_S_OK)
752     return status;
753
754   return RPCRT4_use_protseq(ps, szep);
755 }
756
757 /***********************************************************************
758  *             RpcServerUseProtseqEpExW (RPCRT4.@)
759  */
760 RPC_STATUS WINAPI RpcServerUseProtseqEpExW( RPC_WSTR Protseq, UINT MaxCalls, RPC_WSTR Endpoint, LPVOID SecurityDescriptor,
761                                             PRPC_POLICY lpPolicy )
762 {
763   RpcServerProtseq* ps;
764   RPC_STATUS status;
765   LPSTR EndpointA;
766
767   TRACE("(%s,%u,%s,%p,{%u,%lu,%lu})\n", debugstr_w( Protseq ), MaxCalls,
768        debugstr_w( Endpoint ), SecurityDescriptor,
769        lpPolicy->Length, lpPolicy->EndpointFlags, lpPolicy->NICFlags );
770
771   status = RPCRT4_get_or_create_serverprotseq(MaxCalls, RPCRT4_strdupWtoA(Protseq), &ps);
772   if (status != RPC_S_OK)
773     return status;
774
775   EndpointA = RPCRT4_strdupWtoA(Endpoint);
776   status = RPCRT4_use_protseq(ps, EndpointA);
777   RPCRT4_strfree(EndpointA);
778   return status;
779 }
780
781 /***********************************************************************
782  *             RpcServerUseProtseqA (RPCRT4.@)
783  */
784 RPC_STATUS WINAPI RpcServerUseProtseqA(RPC_CSTR Protseq, unsigned int MaxCalls, void *SecurityDescriptor)
785 {
786   TRACE("(Protseq == %s, MaxCalls == %d, SecurityDescriptor == ^%p)\n", debugstr_a((char*)Protseq), MaxCalls, SecurityDescriptor);
787   return RpcServerUseProtseqEpA(Protseq, MaxCalls, NULL, SecurityDescriptor);
788 }
789
790 /***********************************************************************
791  *             RpcServerUseProtseqW (RPCRT4.@)
792  */
793 RPC_STATUS WINAPI RpcServerUseProtseqW(RPC_WSTR Protseq, unsigned int MaxCalls, void *SecurityDescriptor)
794 {
795   TRACE("Protseq == %s, MaxCalls == %d, SecurityDescriptor == ^%p)\n", debugstr_w(Protseq), MaxCalls, SecurityDescriptor);
796   return RpcServerUseProtseqEpW(Protseq, MaxCalls, NULL, SecurityDescriptor);
797 }
798
799 /***********************************************************************
800  *             RpcServerRegisterIf (RPCRT4.@)
801  */
802 RPC_STATUS WINAPI RpcServerRegisterIf( RPC_IF_HANDLE IfSpec, UUID* MgrTypeUuid, RPC_MGR_EPV* MgrEpv )
803 {
804   TRACE("(%p,%s,%p)\n", IfSpec, debugstr_guid(MgrTypeUuid), MgrEpv);
805   return RpcServerRegisterIf2( IfSpec, MgrTypeUuid, MgrEpv, 0, RPC_C_LISTEN_MAX_CALLS_DEFAULT, (UINT)-1, NULL );
806 }
807
808 /***********************************************************************
809  *             RpcServerRegisterIfEx (RPCRT4.@)
810  */
811 RPC_STATUS WINAPI RpcServerRegisterIfEx( RPC_IF_HANDLE IfSpec, UUID* MgrTypeUuid, RPC_MGR_EPV* MgrEpv,
812                        UINT Flags, UINT MaxCalls, RPC_IF_CALLBACK_FN* IfCallbackFn )
813 {
814   TRACE("(%p,%s,%p,%u,%u,%p)\n", IfSpec, debugstr_guid(MgrTypeUuid), MgrEpv, Flags, MaxCalls, IfCallbackFn);
815   return RpcServerRegisterIf2( IfSpec, MgrTypeUuid, MgrEpv, Flags, MaxCalls, (UINT)-1, IfCallbackFn );
816 }
817
818 /***********************************************************************
819  *             RpcServerRegisterIf2 (RPCRT4.@)
820  */
821 RPC_STATUS WINAPI RpcServerRegisterIf2( RPC_IF_HANDLE IfSpec, UUID* MgrTypeUuid, RPC_MGR_EPV* MgrEpv,
822                       UINT Flags, UINT MaxCalls, UINT MaxRpcSize, RPC_IF_CALLBACK_FN* IfCallbackFn )
823 {
824   PRPC_SERVER_INTERFACE If = (PRPC_SERVER_INTERFACE)IfSpec;
825   RpcServerInterface* sif;
826   unsigned int i;
827
828   TRACE("(%p,%s,%p,%u,%u,%u,%p)\n", IfSpec, debugstr_guid(MgrTypeUuid), MgrEpv, Flags, MaxCalls,
829          MaxRpcSize, IfCallbackFn);
830   TRACE(" interface id: %s %d.%d\n", debugstr_guid(&If->InterfaceId.SyntaxGUID),
831                                      If->InterfaceId.SyntaxVersion.MajorVersion,
832                                      If->InterfaceId.SyntaxVersion.MinorVersion);
833   TRACE(" transfer syntax: %s %d.%d\n", debugstr_guid(&If->TransferSyntax.SyntaxGUID),
834                                         If->TransferSyntax.SyntaxVersion.MajorVersion,
835                                         If->TransferSyntax.SyntaxVersion.MinorVersion);
836   TRACE(" dispatch table: %p\n", If->DispatchTable);
837   if (If->DispatchTable) {
838     TRACE("  dispatch table count: %d\n", If->DispatchTable->DispatchTableCount);
839     for (i=0; i<If->DispatchTable->DispatchTableCount; i++) {
840       TRACE("   entry %d: %p\n", i, If->DispatchTable->DispatchTable[i]);
841     }
842     TRACE("  reserved: %ld\n", If->DispatchTable->Reserved);
843   }
844   TRACE(" protseq endpoint count: %d\n", If->RpcProtseqEndpointCount);
845   TRACE(" default manager epv: %p\n", If->DefaultManagerEpv);
846   TRACE(" interpreter info: %p\n", If->InterpreterInfo);
847
848   sif = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(RpcServerInterface));
849   sif->If           = If;
850   if (MgrTypeUuid) {
851     sif->MgrTypeUuid = *MgrTypeUuid;
852     sif->MgrEpv       = MgrEpv;
853   } else {
854     memset(&sif->MgrTypeUuid, 0, sizeof(UUID));
855     sif->MgrEpv       = If->DefaultManagerEpv;
856   }
857   sif->Flags        = Flags;
858   sif->MaxCalls     = MaxCalls;
859   sif->MaxRpcSize   = MaxRpcSize;
860   sif->IfCallbackFn = IfCallbackFn;
861
862   EnterCriticalSection(&server_cs);
863   list_add_head(&server_interfaces, &sif->entry);
864   LeaveCriticalSection(&server_cs);
865
866   if (sif->Flags & RPC_IF_AUTOLISTEN)
867       RPCRT4_start_listen(TRUE);
868
869   return RPC_S_OK;
870 }
871
872 /***********************************************************************
873  *             RpcServerUnregisterIf (RPCRT4.@)
874  */
875 RPC_STATUS WINAPI RpcServerUnregisterIf( RPC_IF_HANDLE IfSpec, UUID* MgrTypeUuid, UINT WaitForCallsToComplete )
876 {
877   PRPC_SERVER_INTERFACE If = (PRPC_SERVER_INTERFACE)IfSpec;
878   HANDLE event = NULL;
879   BOOL found = FALSE;
880   BOOL completed = TRUE;
881   RpcServerInterface *cif;
882   RPC_STATUS status;
883
884   TRACE("(IfSpec == (RPC_IF_HANDLE)^%p (%s), MgrTypeUuid == %s, WaitForCallsToComplete == %u)\n",
885     IfSpec, debugstr_guid(&If->InterfaceId.SyntaxGUID), debugstr_guid(MgrTypeUuid), WaitForCallsToComplete);
886
887   EnterCriticalSection(&server_cs);
888   LIST_FOR_EACH_ENTRY(cif, &server_interfaces, RpcServerInterface, entry) {
889     if ((!IfSpec || !memcmp(&If->InterfaceId, &cif->If->InterfaceId, sizeof(RPC_SYNTAX_IDENTIFIER))) &&
890         UuidEqual(MgrTypeUuid, &cif->MgrTypeUuid, &status)) {
891       list_remove(&cif->entry);
892       TRACE("unregistering cif %p\n", cif);
893       if (cif->CurrentCalls) {
894         completed = FALSE;
895         cif->Delete = TRUE;
896         if (WaitForCallsToComplete)
897           cif->CallsCompletedEvent = event = CreateEventW(NULL, FALSE, FALSE, NULL);
898       }
899       found = TRUE;
900       break;
901     }
902   }
903   LeaveCriticalSection(&server_cs);
904
905   if (!found) {
906     ERR("not found for object %s\n", debugstr_guid(MgrTypeUuid));
907     return RPC_S_UNKNOWN_IF;
908   }
909
910   if (completed)
911     HeapFree(GetProcessHeap(), 0, cif);
912   else if (event) {
913     /* sif will be freed when the last call is completed, so be careful not to
914      * touch that memory here as that could happen before we get here */
915     WaitForSingleObject(event, INFINITE);
916     CloseHandle(event);
917   }
918
919   return RPC_S_OK;
920 }
921
922 /***********************************************************************
923  *             RpcServerUnregisterIfEx (RPCRT4.@)
924  */
925 RPC_STATUS WINAPI RpcServerUnregisterIfEx( RPC_IF_HANDLE IfSpec, UUID* MgrTypeUuid, int RundownContextHandles )
926 {
927   FIXME("(IfSpec == (RPC_IF_HANDLE)^%p, MgrTypeUuid == %s, RundownContextHandles == %d): stub\n",
928     IfSpec, debugstr_guid(MgrTypeUuid), RundownContextHandles);
929
930   return RPC_S_OK;
931 }
932
933 /***********************************************************************
934  *             RpcObjectSetType (RPCRT4.@)
935  *
936  * PARAMS
937  *   ObjUuid  [I] "Object" UUID
938  *   TypeUuid [I] "Type" UUID
939  *
940  * RETURNS
941  *   RPC_S_OK                 The call succeeded
942  *   RPC_S_INVALID_OBJECT     The provided object (nil) is not valid
943  *   RPC_S_ALREADY_REGISTERED The provided object is already registered
944  *
945  * Maps "Object" UUIDs to "Type" UUID's.  Passing the nil UUID as the type
946  * resets the mapping for the specified object UUID to nil (the default).
947  * The nil object is always associated with the nil type and cannot be
948  * reassigned.  Servers can support multiple implementations on the same
949  * interface by registering different end-point vectors for the different
950  * types.  There's no need to call this if a server only supports the nil
951  * type, as is typical.
952  */
953 RPC_STATUS WINAPI RpcObjectSetType( UUID* ObjUuid, UUID* TypeUuid )
954 {
955   RpcObjTypeMap *map = RpcObjTypeMaps, *prev = NULL;
956   RPC_STATUS dummy;
957
958   TRACE("(ObjUUID == %s, TypeUuid == %s).\n", debugstr_guid(ObjUuid), debugstr_guid(TypeUuid));
959   if ((! ObjUuid) || UuidIsNil(ObjUuid, &dummy)) {
960     /* nil uuid cannot be remapped */
961     return RPC_S_INVALID_OBJECT;
962   }
963
964   /* find the mapping for this object if there is one ... */
965   while (map) {
966     if (! UuidCompare(ObjUuid, &map->Object, &dummy)) break;
967     prev = map;
968     map = map->next;
969   }
970   if ((! TypeUuid) || UuidIsNil(TypeUuid, &dummy)) {
971     /* ... and drop it from the list */
972     if (map) {
973       if (prev) 
974         prev->next = map->next;
975       else
976         RpcObjTypeMaps = map->next;
977       HeapFree(GetProcessHeap(), 0, map);
978     }
979   } else {
980     /* ... , fail if we found it ... */
981     if (map)
982       return RPC_S_ALREADY_REGISTERED;
983     /* ... otherwise create a new one and add it in. */
984     map = HeapAlloc(GetProcessHeap(), 0, sizeof(RpcObjTypeMap));
985     map->Object = *ObjUuid;
986     map->Type = *TypeUuid;
987     map->next = NULL;
988     if (prev)
989       prev->next = map; /* prev is the last map in the linklist */
990     else
991       RpcObjTypeMaps = map;
992   }
993
994   return RPC_S_OK;
995 }
996
997 /***********************************************************************
998  *             RpcServerRegisterAuthInfoA (RPCRT4.@)
999  */
1000 RPC_STATUS WINAPI RpcServerRegisterAuthInfoA( RPC_CSTR ServerPrincName, ULONG AuthnSvc, RPC_AUTH_KEY_RETRIEVAL_FN GetKeyFn,
1001                             LPVOID Arg )
1002 {
1003   FIXME( "(%s,%u,%p,%p): stub\n", ServerPrincName, AuthnSvc, GetKeyFn, Arg );
1004   
1005   return RPC_S_UNKNOWN_AUTHN_SERVICE; /* We don't know any authentication services */
1006 }
1007
1008 /***********************************************************************
1009  *             RpcServerRegisterAuthInfoW (RPCRT4.@)
1010  */
1011 RPC_STATUS WINAPI RpcServerRegisterAuthInfoW( RPC_WSTR ServerPrincName, ULONG AuthnSvc, RPC_AUTH_KEY_RETRIEVAL_FN GetKeyFn,
1012                             LPVOID Arg )
1013 {
1014   FIXME( "(%s,%u,%p,%p): stub\n", debugstr_w( ServerPrincName ), AuthnSvc, GetKeyFn, Arg );
1015   
1016   return RPC_S_UNKNOWN_AUTHN_SERVICE; /* We don't know any authentication services */
1017 }
1018
1019 /***********************************************************************
1020  *             RpcServerListen (RPCRT4.@)
1021  */
1022 RPC_STATUS WINAPI RpcServerListen( UINT MinimumCallThreads, UINT MaxCalls, UINT DontWait )
1023 {
1024   RPC_STATUS status = RPC_S_OK;
1025
1026   TRACE("(%u,%u,%u)\n", MinimumCallThreads, MaxCalls, DontWait);
1027
1028   if (list_empty(&protseqs))
1029     return RPC_S_NO_PROTSEQS_REGISTERED;
1030
1031   status = RPCRT4_start_listen(FALSE);
1032
1033   if (DontWait || (status != RPC_S_OK)) return status;
1034
1035   return RpcMgmtWaitServerListen();
1036 }
1037
1038 /***********************************************************************
1039  *             RpcMgmtServerWaitListen (RPCRT4.@)
1040  */
1041 RPC_STATUS WINAPI RpcMgmtWaitServerListen( void )
1042 {
1043   TRACE("()\n");
1044
1045   EnterCriticalSection(&listen_cs);
1046
1047   if (!std_listen) {
1048     LeaveCriticalSection(&listen_cs);
1049     return RPC_S_NOT_LISTENING;
1050   }
1051   
1052   LeaveCriticalSection(&listen_cs);
1053
1054   FIXME("not waiting for server calls to finish\n");
1055
1056   return RPC_S_OK;
1057 }
1058
1059 /***********************************************************************
1060  *             RpcMgmtStopServerListening (RPCRT4.@)
1061  */
1062 RPC_STATUS WINAPI RpcMgmtStopServerListening ( RPC_BINDING_HANDLE Binding )
1063 {
1064   TRACE("(Binding == (RPC_BINDING_HANDLE)^%p)\n", Binding);
1065
1066   if (Binding) {
1067     FIXME("client-side invocation not implemented.\n");
1068     return RPC_S_WRONG_KIND_OF_BINDING;
1069   }
1070   
1071   RPCRT4_stop_listen(FALSE);
1072
1073   return RPC_S_OK;
1074 }
1075
1076 /***********************************************************************
1077  *             RpcMgmtEnableIdleCleanup (RPCRT4.@)
1078  */
1079 RPC_STATUS WINAPI RpcMgmtEnableIdleCleanup(void)
1080 {
1081     FIXME("(): stub\n");
1082     return RPC_S_OK;
1083 }
1084
1085 /***********************************************************************
1086  *             I_RpcServerStartListening (RPCRT4.@)
1087  */
1088 RPC_STATUS WINAPI I_RpcServerStartListening( HWND hWnd )
1089 {
1090   FIXME( "(%p): stub\n", hWnd );
1091
1092   return RPC_S_OK;
1093 }
1094
1095 /***********************************************************************
1096  *             I_RpcServerStopListening (RPCRT4.@)
1097  */
1098 RPC_STATUS WINAPI I_RpcServerStopListening( void )
1099 {
1100   FIXME( "(): stub\n" );
1101
1102   return RPC_S_OK;
1103 }
1104
1105 /***********************************************************************
1106  *             I_RpcWindowProc (RPCRT4.@)
1107  */
1108 UINT WINAPI I_RpcWindowProc( void *hWnd, UINT Message, UINT wParam, ULONG lParam )
1109 {
1110   FIXME( "(%p,%08x,%08x,%08x): stub\n", hWnd, Message, wParam, lParam );
1111
1112   return 0;
1113 }
1114
1115 /***********************************************************************
1116  *             RpcMgmtInqIfIds (RPCRT4.@)
1117  */
1118 RPC_STATUS WINAPI RpcMgmtInqIfIds(RPC_BINDING_HANDLE Binding, RPC_IF_ID_VECTOR **IfIdVector)
1119 {
1120   FIXME("(%p,%p): stub\n", Binding, IfIdVector);
1121   return RPC_S_INVALID_BINDING;
1122 }
1123
1124 /***********************************************************************
1125  *             RpcMgmtEpEltInqBegin (RPCRT4.@)
1126  */
1127 RPC_STATUS WINAPI RpcMgmtEpEltInqBegin(RPC_BINDING_HANDLE Binding, ULONG InquiryType,
1128     RPC_IF_ID *IfId, ULONG VersOption, UUID *ObjectUuid, RPC_EP_INQ_HANDLE* InquiryContext)
1129 {
1130   FIXME("(%p,%u,%p,%u,%p,%p): stub\n",
1131         Binding, InquiryType, IfId, VersOption, ObjectUuid, InquiryContext);
1132   return RPC_S_INVALID_BINDING;
1133 }
1134
1135 /***********************************************************************
1136  *             RpcMgmtIsServerListening (RPCRT4.@)
1137  */
1138 RPC_STATUS WINAPI RpcMgmtIsServerListening(RPC_BINDING_HANDLE Binding)
1139 {
1140   FIXME("(%p): stub\n", Binding);
1141   return RPC_S_INVALID_BINDING;
1142 }
1143
1144 /***********************************************************************
1145  *             RpcMgmtSetServerStackSize (RPCRT4.@)
1146  */
1147 RPC_STATUS WINAPI RpcMgmtSetServerStackSize(ULONG ThreadStackSize)
1148 {
1149   FIXME("(0x%x): stub\n", ThreadStackSize);
1150   return RPC_S_OK;
1151 }
1152
1153 /***********************************************************************
1154  *             I_RpcGetCurrentCallHandle (RPCRT4.@)
1155  */
1156 RPC_BINDING_HANDLE WINAPI I_RpcGetCurrentCallHandle(void)
1157 {
1158     TRACE("\n");
1159     return RPCRT4_GetThreadCurrentCallHandle();
1160 }