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