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