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