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   unsigned char *auth_data;
  56   ULONG auth_length;
  57 } RpcPacket;
  58
  59 typedef struct _RpcObjTypeMap
  60 {
  61   /* FIXME: a hash table would be better. */
  62   struct _RpcObjTypeMap *next;
  63   UUID Object;
  64   UUID Type;
  65 } RpcObjTypeMap;
  66
  67 static RpcObjTypeMap *RpcObjTypeMaps;
  68
  69 /* list of type RpcServerProtseq */
  70 static struct list protseqs = LIST_INIT(protseqs);
  71 static struct list server_interfaces = LIST_INIT(server_interfaces);
  72 static struct list server_registered_auth_info = LIST_INIT(server_registered_auth_info);
  73
  74 static CRITICAL_SECTION server_cs;
  75 static CRITICAL_SECTION_DEBUG server_cs_debug =
  76 {
  77     0, 0, &server_cs,
  78     { &server_cs_debug.ProcessLocksList, &server_cs_debug.ProcessLocksList },
  79       0, 0, { (DWORD_PTR)(__FILE__ ": server_cs") }
  80 };
  81 static CRITICAL_SECTION server_cs = { &server_cs_debug, -1, 0, 0, 0, 0 };
  82
  83 static CRITICAL_SECTION listen_cs;
  84 static CRITICAL_SECTION_DEBUG listen_cs_debug =
  85 {
  86     0, 0, &listen_cs,
  87     { &listen_cs_debug.ProcessLocksList, &listen_cs_debug.ProcessLocksList },
  88       0, 0, { (DWORD_PTR)(__FILE__ ": listen_cs") }
  89 };
  90 static CRITICAL_SECTION listen_cs = { &listen_cs_debug, -1, 0, 0, 0, 0 };
  91
  92 static CRITICAL_SECTION server_auth_info_cs;
  93 static CRITICAL_SECTION_DEBUG server_auth_info_cs_debug =
  94 {
  95     0, 0, &server_auth_info_cs,
  96     { &server_auth_info_cs_debug.ProcessLocksList, &server_auth_info_cs_debug.ProcessLocksList },
  97       0, 0, { (DWORD_PTR)(__FILE__ ": server_auth_info_cs") }
  98 };
  99 static CRITICAL_SECTION server_auth_info_cs = { &server_auth_info_cs_debug, -1, 0, 0, 0, 0 };
 100
 101 /* whether the server is currently listening */
 102 static BOOL std_listen;
 103 /* number of manual listeners (calls to RpcServerListen) */
 104 static LONG manual_listen_count;
 105 /* total listeners including auto listeners */
 106 static LONG listen_count;
 107 /* event set once all listening is finished */
 108 static HANDLE listen_done_event;
 109
 110 static UUID uuid_nil;
 111
 112 static inline RpcObjTypeMap *LookupObjTypeMap(UUID *ObjUuid)
 113 {
 114   RpcObjTypeMap *rslt = RpcObjTypeMaps;
 115   RPC_STATUS dummy;
 116
 117   while (rslt) {
 118     if (! UuidCompare(ObjUuid, &rslt->Object, &dummy)) break;
 119     rslt = rslt->next;
 120   }
 121
 122   return rslt;
 123 }
 124
 125 static inline UUID *LookupObjType(UUID *ObjUuid)
 126 {
 127   RpcObjTypeMap *map = LookupObjTypeMap(ObjUuid);
 128   if (map)
 129     return &map->Type;
 130   else
 131     return &uuid_nil;
 132 }
 133
 134 static RpcServerInterface* RPCRT4_find_interface(UUID* object,
 135                                                  const RPC_SYNTAX_IDENTIFIER *if_id,
 136                                                  const RPC_SYNTAX_IDENTIFIER *transfer_syntax,
 137                                                  BOOL check_object)
 138 {
 139   UUID* MgrType = NULL;
 140   RpcServerInterface* cif;
 141   RPC_STATUS status;
 142
 143   if (check_object)
 144     MgrType = LookupObjType(object);
 145   EnterCriticalSection(&server_cs);
 146   LIST_FOR_EACH_ENTRY(cif, &server_interfaces, RpcServerInterface, entry) {
 147     if (!memcmp(if_id, &cif->If->InterfaceId, sizeof(RPC_SYNTAX_IDENTIFIER)) &&
 148         (!transfer_syntax || !memcmp(transfer_syntax, &cif->If->TransferSyntax, sizeof(RPC_SYNTAX_IDENTIFIER))) &&
 149         (check_object == FALSE || UuidEqual(MgrType, &cif->MgrTypeUuid, &status)) &&
 150         std_listen) {
 151       InterlockedIncrement(&cif->CurrentCalls);
 152       break;
 153     }
 154   }
 155   LeaveCriticalSection(&server_cs);
 156   if (&cif->entry == &server_interfaces) cif = NULL;
 157   TRACE("returning %p for object %s, if_id { %d.%d %s }\n", cif,
 158     debugstr_guid(object), if_id->SyntaxVersion.MajorVersion,
 159     if_id->SyntaxVersion.MinorVersion, debugstr_guid(&if_id->SyntaxGUID));
 160   return cif;
 161 }
 162
 163 static void RPCRT4_release_server_interface(RpcServerInterface *sif)
 164 {
 165   if (!InterlockedDecrement(&sif->CurrentCalls) &&
 166       sif->Delete) {
 167     /* sif must have been removed from server_interfaces before
 168      * CallsCompletedEvent is set */
 169     if (sif->CallsCompletedEvent)
 170       SetEvent(sif->CallsCompletedEvent);
 171     HeapFree(GetProcessHeap(), 0, sif);
 172   }
 173 }
 174
 175 static RpcPktHdr *handle_bind_error(RpcConnection *conn, RPC_STATUS error)
 176 {
 177     unsigned int reject_reason;
 178     switch (error)
 179     {
 180     case RPC_S_SERVER_TOO_BUSY:
 181         reject_reason = REJECT_TEMPORARY_CONGESTION;
 182         break;
 183     case ERROR_OUTOFMEMORY:
 184     case RPC_S_OUT_OF_RESOURCES:
 185         reject_reason = REJECT_LOCAL_LIMIT_EXCEEDED;
 186         break;
 187     case RPC_S_PROTOCOL_ERROR:
 188         reject_reason = REJECT_PROTOCOL_VERSION_NOT_SUPPORTED;
 189         break;
 190     case RPC_S_UNKNOWN_AUTHN_SERVICE:
 191         reject_reason = REJECT_UNKNOWN_AUTHN_SERVICE;
 192         break;
 193     case ERROR_ACCESS_DENIED:
 194         reject_reason = REJECT_INVALID_CHECKSUM;
 195         break;
 196     default:
 197         FIXME("unexpected status value %d\n", error);
 198         /* fall through */
 199     case RPC_S_INVALID_BOUND:
 200         reject_reason = REJECT_REASON_NOT_SPECIFIED;
 201         break;
 202     }
 203     return RPCRT4_BuildBindNackHeader(NDR_LOCAL_DATA_REPRESENTATION,
 204                                       RPC_VER_MAJOR, RPC_VER_MINOR,
 205                                       reject_reason);
 206 }
 207
 208 static RPC_STATUS process_bind_packet_no_send(
 209     RpcConnection *conn, RpcPktBindHdr *hdr, RPC_MESSAGE *msg,
 210     unsigned char *auth_data, ULONG auth_length, RpcPktHdr **ack_response,
 211     unsigned char **auth_data_out, ULONG *auth_length_out)
 212 {
 213   RPC_STATUS status;
 214   RpcContextElement *ctxt_elem;
 215   unsigned int i;
 216   RpcResult *results;
 217
 218   /* validate data */
 219   for (i = 0, ctxt_elem = msg->Buffer;
 220        i < hdr->num_elements;
 221        i++, ctxt_elem = (RpcContextElement *)&ctxt_elem->transfer_syntaxes[ctxt_elem->num_syntaxes])
 222   {
 223       if (((char *)ctxt_elem - (char *)msg->Buffer) > msg->BufferLength ||
 224           ((char *)&ctxt_elem->transfer_syntaxes[ctxt_elem->num_syntaxes] - (char *)msg->Buffer) > msg->BufferLength)
 225       {
 226           ERR("inconsistent data in packet - packet length %d, num elements %d\n",
 227               msg->BufferLength, hdr->num_elements);
 228           return RPC_S_INVALID_BOUND;
 229       }
 230   }
 231
 232   if (hdr->max_tsize < RPC_MIN_PACKET_SIZE ||
 233       !UuidIsNil(&conn->ActiveInterface.SyntaxGUID, &status) ||
 234       conn->server_binding)
 235   {
 236     TRACE("packet size less than min size, or active interface syntax guid non-null\n");
 237
 238     return RPC_S_INVALID_BOUND;
 239   }
 240
 241   results = HeapAlloc(GetProcessHeap(), 0,
 242                       hdr->num_elements * sizeof(*results));
 243   if (!results)
 244     return RPC_S_OUT_OF_RESOURCES;
 245
 246   for (i = 0, ctxt_elem = (RpcContextElement *)msg->Buffer;
 247        i < hdr->num_elements;
 248        i++, ctxt_elem = (RpcContextElement *)&ctxt_elem->transfer_syntaxes[ctxt_elem->num_syntaxes])
 249   {
 250       RpcServerInterface* sif = NULL;
 251       unsigned int j;
 252
 253       for (j = 0; !sif && j < ctxt_elem->num_syntaxes; j++)
 254       {
 255           sif = RPCRT4_find_interface(NULL, &ctxt_elem->abstract_syntax,
 256                                       &ctxt_elem->transfer_syntaxes[j], FALSE);
 257           if (sif)
 258               break;
 259       }
 260       if (sif)
 261       {
 262           RPCRT4_release_server_interface(sif);
 263           TRACE("accepting bind request on connection %p for %s\n", conn,
 264                 debugstr_guid(&ctxt_elem->abstract_syntax.SyntaxGUID));
 265           results[i].result = RESULT_ACCEPT;
 266           results[i].reason = REASON_NONE;
 267           results[i].transfer_syntax = ctxt_elem->transfer_syntaxes[j];
 268
 269           /* save the interface for later use */
 270           /* FIXME: save linked list */
 271           conn->ActiveInterface = ctxt_elem->abstract_syntax;
 272       }
 273       else if ((sif = RPCRT4_find_interface(NULL, &ctxt_elem->abstract_syntax,
 274                                             NULL, FALSE)) != NULL)
 275       {
 276           RPCRT4_release_server_interface(sif);
 277           TRACE("not accepting bind request on connection %p for %s - no transfer syntaxes supported\n",
 278                 conn, debugstr_guid(&ctxt_elem->abstract_syntax.SyntaxGUID));
 279           results[i].result = RESULT_PROVIDER_REJECTION;
 280           results[i].reason = REASON_TRANSFER_SYNTAXES_NOT_SUPPORTED;
 281           memset(&results[i].transfer_syntax, 0, sizeof(results[i].transfer_syntax));
 282       }
 283       else
 284       {
 285           TRACE("not accepting bind request on connection %p for %s - abstract syntax not supported\n",
 286                 conn, debugstr_guid(&ctxt_elem->abstract_syntax.SyntaxGUID));
 287           results[i].result = RESULT_PROVIDER_REJECTION;
 288           results[i].reason = REASON_ABSTRACT_SYNTAX_NOT_SUPPORTED;
 289           memset(&results[i].transfer_syntax, 0, sizeof(results[i].transfer_syntax));
 290       }
 291   }
 292
 293   /* create temporary binding */
 294   status = RPCRT4_MakeBinding(&conn->server_binding, conn);
 295   if (status != RPC_S_OK)
 296   {
 297       HeapFree(GetProcessHeap(), 0, results);
 298       return status;
 299   }
 300
 301   status = RpcServerAssoc_GetAssociation(rpcrt4_conn_get_name(conn),
 302                                          conn->NetworkAddr, conn->Endpoint,
 303                                          conn->NetworkOptions,
 304                                          hdr->assoc_gid,
 305                                          &conn->server_binding->Assoc);
 306   if (status != RPC_S_OK)
 307   {
 308       HeapFree(GetProcessHeap(), 0, results);
 309       return status;
 310   }
 311
 312   if (auth_length)
 313   {
 314       status = RPCRT4_ServerConnectionAuth(conn, TRUE,
 315                                            (RpcAuthVerifier *)auth_data,
 316                                            auth_length, auth_data_out,
 317                                            auth_length_out);
 318       if (status != RPC_S_OK)
 319       {
 320           HeapFree(GetProcessHeap(), 0, results);
 321           return status;
 322       }
 323   }
 324
 325   *ack_response = RPCRT4_BuildBindAckHeader(NDR_LOCAL_DATA_REPRESENTATION,
 326                                             RPC_MAX_PACKET_SIZE,
 327                                             RPC_MAX_PACKET_SIZE,
 328                                             conn->server_binding->Assoc->assoc_group_id,
 329                                             conn->Endpoint, hdr->num_elements,
 330                                             results);
 331   HeapFree(GetProcessHeap(), 0, results);
 332
 333   if (*ack_response)
 334       conn->MaxTransmissionSize = hdr->max_tsize;
 335   else
 336       status = RPC_S_OUT_OF_RESOURCES;
 337
 338   return status;
 339 }
 340
 341 static RPC_STATUS process_bind_packet(RpcConnection *conn, RpcPktBindHdr *hdr,
 342                                       RPC_MESSAGE *msg,
 343                                       unsigned char *auth_data,
 344                                       ULONG auth_length)
 345 {
 346     RPC_STATUS status;
 347     RpcPktHdr *response = NULL;
 348     unsigned char *auth_data_out = NULL;
 349     ULONG auth_length_out = 0;
 350
 351     status = process_bind_packet_no_send(conn, hdr, msg, auth_data, auth_length,
 352                                          &response, &auth_data_out,
 353                                          &auth_length_out);
 354     if (status != RPC_S_OK)
 355         response = handle_bind_error(conn, status);
 356     if (response)
 357         status = RPCRT4_SendWithAuth(conn, response, NULL, 0, auth_data_out, auth_length_out);
 358     else
 359         status = ERROR_OUTOFMEMORY;
 360     RPCRT4_FreeHeader(response);
 361
 362     return status;
 363 }
 364
 365
 366 static RPC_STATUS process_request_packet(RpcConnection *conn, RpcPktRequestHdr *hdr, RPC_MESSAGE *msg)
 367 {
 368   RPC_STATUS status;
 369   RpcPktHdr *response = NULL;
 370   RpcServerInterface* sif;
 371   RPC_DISPATCH_FUNCTION func;
 372   BOOL exception;
 373   UUID *object_uuid;
 374   NDR_SCONTEXT context_handle;
 375   void *buf = msg->Buffer;
 376
 377   /* fail if the connection isn't bound with an interface */
 378   if (UuidIsNil(&conn->ActiveInterface.SyntaxGUID, &status)) {
 379     /* FIXME: should send BindNack instead */
 380     response = RPCRT4_BuildFaultHeader(NDR_LOCAL_DATA_REPRESENTATION,
 381                                        status);
 382
 383     RPCRT4_Send(conn, response, NULL, 0);
 384     RPCRT4_FreeHeader(response);
 385     return RPC_S_OK;
 386   }
 387
 388   if (hdr->common.flags & RPC_FLG_OBJECT_UUID) {
 389     object_uuid = (UUID*)(hdr + 1);
 390   } else {
 391     object_uuid = NULL;
 392   }
 393
 394   sif = RPCRT4_find_interface(object_uuid, &conn->ActiveInterface, NULL, TRUE);
 395   if (!sif) {
 396     WARN("interface %s no longer registered, returning fault packet\n", debugstr_guid(&conn->ActiveInterface.SyntaxGUID));
 397     response = RPCRT4_BuildFaultHeader(NDR_LOCAL_DATA_REPRESENTATION,
 398                                        NCA_S_UNK_IF);
 399
 400     RPCRT4_Send(conn, response, NULL, 0);
 401     RPCRT4_FreeHeader(response);
 402     return RPC_S_OK;
 403   }
 404   msg->RpcInterfaceInformation = sif->If;
 405   /* copy the endpoint vector from sif to msg so that midl-generated code will use it */
 406   msg->ManagerEpv = sif->MgrEpv;
 407   if (object_uuid != NULL) {
 408     RPCRT4_SetBindingObject(msg->Handle, object_uuid);
 409   }
 410
 411   /* find dispatch function */
 412   msg->ProcNum = hdr->opnum;
 413   if (sif->Flags & RPC_IF_OLE) {
 414     /* native ole32 always gives us a dispatch table with a single entry
 415     * (I assume that's a wrapper for IRpcStubBuffer::Invoke) */
 416     func = *sif->If->DispatchTable->DispatchTable;
 417   } else {
 418     if (msg->ProcNum >= sif->If->DispatchTable->DispatchTableCount) {
 419       WARN("invalid procnum (%d/%d)\n", msg->ProcNum, sif->If->DispatchTable->DispatchTableCount);
 420       response = RPCRT4_BuildFaultHeader(NDR_LOCAL_DATA_REPRESENTATION,
 421                                          NCA_S_OP_RNG_ERROR);
 422
 423       RPCRT4_Send(conn, response, NULL, 0);
 424       RPCRT4_FreeHeader(response);
 425     }
 426     func = sif->If->DispatchTable->DispatchTable[msg->ProcNum];
 427   }
 428
 429   /* put in the drep. FIXME: is this more universally applicable?
 430     perhaps we should move this outward... */
 431   msg->DataRepresentation =
 432     MAKELONG( MAKEWORD(hdr->common.drep[0], hdr->common.drep[1]),
 433               MAKEWORD(hdr->common.drep[2], hdr->common.drep[3]));
 434
 435   exception = FALSE;
 436
 437   /* dispatch */
 438   RPCRT4_SetThreadCurrentCallHandle(msg->Handle);
 439   __TRY {
 440     if (func) func(msg);
 441   } __EXCEPT_ALL {
 442     WARN("exception caught with code 0x%08x = %d\n", GetExceptionCode(), GetExceptionCode());
 443     exception = TRUE;
 444     if (GetExceptionCode() == STATUS_ACCESS_VIOLATION)
 445       status = ERROR_NOACCESS;
 446     else
 447       status = GetExceptionCode();
 448     response = RPCRT4_BuildFaultHeader(msg->DataRepresentation,
 449                                        RPC2NCA_STATUS(status));
 450   } __ENDTRY
 451     RPCRT4_SetThreadCurrentCallHandle(NULL);
 452
 453   /* release any unmarshalled context handles */
 454   while ((context_handle = RPCRT4_PopThreadContextHandle()) != NULL)
 455     RpcServerAssoc_ReleaseContextHandle(conn->server_binding->Assoc, context_handle, TRUE);
 456
 457   if (!exception)
 458     response = RPCRT4_BuildResponseHeader(msg->DataRepresentation,
 459                                           msg->BufferLength);
 460
 461   /* send response packet */
 462   if (response) {
 463     status = RPCRT4_Send(conn, response, exception ? NULL : msg->Buffer,
 464                          exception ? 0 : msg->BufferLength);
 465     RPCRT4_FreeHeader(response);
 466   } else
 467     ERR("out of memory\n");
 468
 469   msg->RpcInterfaceInformation = NULL;
 470   RPCRT4_release_server_interface(sif);
 471
 472   if (msg->Buffer == buf) buf = NULL;
 473   TRACE("freeing Buffer=%p\n", buf);
 474   I_RpcFree(buf);
 475
 476   return status;
 477 }
 478
 479 static RPC_STATUS process_auth3_packet(RpcConnection *conn,
 480                                        RpcPktCommonHdr *hdr,
 481                                        RPC_MESSAGE *msg,
 482                                        unsigned char *auth_data,
 483                                        ULONG auth_length)
 484 {
 485     RPC_STATUS status;
 486
 487     if (UuidIsNil(&conn->ActiveInterface.SyntaxGUID, &status) ||
 488         !auth_length || msg->BufferLength != 0)
 489         status = RPC_S_PROTOCOL_ERROR;
 490     else
 491     {
 492         status = RPCRT4_ServerConnectionAuth(conn, FALSE,
 493                                              (RpcAuthVerifier *)auth_data,
 494                                              auth_length, NULL, NULL);
 495     }
 496
 497     /* FIXME: client doesn't expect a response to this message so must store
 498      * status in connection so that fault packet can be returned when next
 499      * packet is received */
 500
 501     return RPC_S_OK;
 502 }
 503
 504 static void RPCRT4_process_packet(RpcConnection* conn, RpcPktHdr* hdr,
 505                                   RPC_MESSAGE* msg, unsigned char *auth_data,
 506                                   ULONG auth_length)
 507 {
 508   msg->Handle = (RPC_BINDING_HANDLE)conn->server_binding;
 509
 510   switch (hdr->common.ptype) {
 511     case PKT_BIND:
 512       TRACE("got bind packet\n");
 513       process_bind_packet(conn, &hdr->bind, msg, auth_data, auth_length);
 514       break;
 515
 516     case PKT_REQUEST:
 517       TRACE("got request packet\n");
 518       process_request_packet(conn, &hdr->request, msg);
 519       break;
 520
 521     case PKT_AUTH3:
 522       TRACE("got auth3 packet\n");
 523       process_auth3_packet(conn, &hdr->common, msg, auth_data, auth_length);
 524       break;
 525     default:
 526       FIXME("unhandled packet type %u\n", hdr->common.ptype);
 527       break;
 528   }
 529
 530   /* clean up */
 531   I_RpcFree(msg->Buffer);
 532   RPCRT4_FreeHeader(hdr);
 533   HeapFree(GetProcessHeap(), 0, msg);
 534   HeapFree(GetProcessHeap(), 0, auth_data);
 535 }
 536
 537 static DWORD CALLBACK RPCRT4_worker_thread(LPVOID the_arg)
 538 {
 539   RpcPacket *pkt = the_arg;
 540   RPCRT4_process_packet(pkt->conn, pkt->hdr, pkt->msg, pkt->auth_data,
 541                         pkt->auth_length);
 542   RPCRT4_ReleaseConnection(pkt->conn);
 543   HeapFree(GetProcessHeap(), 0, pkt);
 544   return 0;
 545 }
 546
 547 static DWORD CALLBACK RPCRT4_io_thread(LPVOID the_arg)
 548 {
 549   RpcConnection* conn = the_arg;
 550   RpcPktHdr *hdr;
 551   RPC_MESSAGE *msg;
 552   RPC_STATUS status;
 553   RpcPacket *packet;
 554   unsigned char *auth_data;
 555   ULONG auth_length;
 556
 557   TRACE("(%p)\n", conn);
 558
 559   for (;;) {
 560     msg = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(RPC_MESSAGE));
 561     if (!msg) break;
 562
 563     status = RPCRT4_ReceiveWithAuth(conn, &hdr, msg, &auth_data, &auth_length);
 564     if (status != RPC_S_OK) {
 565       WARN("receive failed with error %x\n", status);
 566       HeapFree(GetProcessHeap(), 0, msg);
 567       break;
 568     }
 569
 570     switch (hdr->common.ptype) {
 571     case PKT_BIND:
 572       TRACE("got bind packet\n");
 573
 574       status = process_bind_packet(conn, &hdr->bind, msg, auth_data,
 575                                    auth_length);
 576       break;
 577
 578     case PKT_REQUEST:
 579       TRACE("got request packet\n");
 580
 581       packet = HeapAlloc(GetProcessHeap(), 0, sizeof(RpcPacket));
 582       if (!packet) {
 583         I_RpcFree(msg->Buffer);
 584         RPCRT4_FreeHeader(hdr);
 585         HeapFree(GetProcessHeap(), 0, msg);
 586         HeapFree(GetProcessHeap(), 0, auth_data);
 587         goto exit;
 588       }
 589       packet->conn = RPCRT4_GrabConnection( conn );
 590       packet->hdr = hdr;
 591       packet->msg = msg;
 592       packet->auth_data = auth_data;
 593       packet->auth_length = auth_length;
 594       if (!QueueUserWorkItem(RPCRT4_worker_thread, packet, WT_EXECUTELONGFUNCTION)) {
 595         ERR("couldn't queue work item for worker thread, error was %d\n", GetLastError());
 596         HeapFree(GetProcessHeap(), 0, packet);
 597         status = RPC_S_OUT_OF_RESOURCES;
 598       } else {
 599         continue;
 600       }
 601       break;
 602
 603     case PKT_AUTH3:
 604       TRACE("got auth3 packet\n");
 605
 606       status = process_auth3_packet(conn, &hdr->common, msg, auth_data,
 607                                     auth_length);
 608       break;
 609     default:
 610       FIXME("unhandled packet type %u\n", hdr->common.ptype);
 611       break;
 612     }
 613
 614     I_RpcFree(msg->Buffer);
 615     RPCRT4_FreeHeader(hdr);
 616     HeapFree(GetProcessHeap(), 0, msg);
 617     HeapFree(GetProcessHeap(), 0, auth_data);
 618
 619     if (status != RPC_S_OK) {
 620       WARN("processing packet failed with error %u\n", status);
 621       break;
 622     }
 623   }
 624 exit:
 625   RPCRT4_ReleaseConnection(conn);
 626   return 0;
 627 }
 628
 629 void RPCRT4_new_client(RpcConnection* conn)
 630 {
 631   HANDLE thread = CreateThread(NULL, 0, RPCRT4_io_thread, conn, 0, NULL);
 632   if (!thread) {
 633     DWORD err = GetLastError();
 634     ERR("failed to create thread, error=%08x\n", err);
 635     RPCRT4_ReleaseConnection(conn);
 636   }
 637   /* we could set conn->thread, but then we'd have to make the io_thread wait
 638    * for that, otherwise the thread might finish, destroy the connection, and
 639    * free the memory we'd write to before we did, causing crashes and stuff -
 640    * so let's implement that later, when we really need conn->thread */
 641
 642   CloseHandle( thread );
 643 }
 644
 645 static DWORD CALLBACK RPCRT4_server_thread(LPVOID the_arg)
 646 {
 647   int res;
 648   unsigned int count;
 649   void *objs = NULL;
 650   RpcServerProtseq* cps = the_arg;
 651   RpcConnection* conn;
 652   BOOL set_ready_event = FALSE;
 653
 654   TRACE("(the_arg == ^%p)\n", the_arg);
 655
 656   for (;;) {
 657     objs = cps->ops->get_wait_array(cps, objs, &count);
 658
 659     if (set_ready_event)
 660     {
 661         /* signal to function that changed state that we are now sync'ed */
 662         SetEvent(cps->server_ready_event);
 663         set_ready_event = FALSE;
 664     }
 665
 666     /* start waiting */
 667     res = cps->ops->wait_for_new_connection(cps, count, objs);
 668
 669     if (res == -1 || (res == 0 && !std_listen))
 670     {
 671       /* cleanup */
 672       cps->ops->free_wait_array(cps, objs);
 673       EnterCriticalSection(&cps->cs);
 674       for (conn = cps->conn; conn; conn = conn->Next)
 675         RPCRT4_CloseConnection(conn);
 676       LeaveCriticalSection(&cps->cs);
 677
 678       if (res == 0 && !std_listen)
 679         SetEvent(cps->server_ready_event);
 680       break;
 681     }
 682     else if (res == 0)
 683       set_ready_event = TRUE;
 684   }
 685   return 0;
 686 }
 687
 688 /* tells the server thread that the state has changed and waits for it to
 689  * make the changes */
 690 static void RPCRT4_sync_with_server_thread(RpcServerProtseq *ps)
 691 {
 692   /* make sure we are the only thread sync'ing the server state, otherwise
 693    * there is a race with the server thread setting an older state and setting
 694    * the server_ready_event when the new state hasn't yet been applied */
 695   WaitForSingleObject(ps->mgr_mutex, INFINITE);
 696
 697   ps->ops->signal_state_changed(ps);
 698
 699   /* wait for server thread to make the requested changes before returning */
 700   WaitForSingleObject(ps->server_ready_event, INFINITE);
 701
 702   ReleaseMutex(ps->mgr_mutex);
 703 }
 704
 705 static RPC_STATUS RPCRT4_start_listen_protseq(RpcServerProtseq *ps, BOOL auto_listen)
 706 {
 707   RPC_STATUS status = RPC_S_OK;
 708   HANDLE server_thread;
 709
 710   EnterCriticalSection(&listen_cs);
 711   if (ps->is_listening) goto done;
 712
 713   if (!ps->mgr_mutex) ps->mgr_mutex = CreateMutexW(NULL, FALSE, NULL);
 714   if (!ps->server_ready_event) ps->server_ready_event = CreateEventW(NULL, FALSE, FALSE, NULL);
 715   server_thread = CreateThread(NULL, 0, RPCRT4_server_thread, ps, 0, NULL);
 716   if (!server_thread)
 717   {
 718     status = RPC_S_OUT_OF_RESOURCES;
 719     goto done;
 720   }
 721   ps->is_listening = TRUE;
 722   CloseHandle(server_thread);
 723
 724 done:
 725   LeaveCriticalSection(&listen_cs);
 726   return status;
 727 }
 728
 729 static RPC_STATUS RPCRT4_start_listen(BOOL auto_listen)
 730 {
 731   RPC_STATUS status = RPC_S_ALREADY_LISTENING;
 732   RpcServerProtseq *cps;
 733
 734   TRACE("\n");
 735
 736   EnterCriticalSection(&listen_cs);
 737   if (auto_listen || (manual_listen_count++ == 0))
 738   {
 739     status = RPC_S_OK;
 740     if (++listen_count == 1)
 741       std_listen = TRUE;
 742   }
 743   LeaveCriticalSection(&listen_cs);
 744
 745   if (std_listen)
 746   {
 747     EnterCriticalSection(&server_cs);
 748     LIST_FOR_EACH_ENTRY(cps, &protseqs, RpcServerProtseq, entry)
 749     {
 750       status = RPCRT4_start_listen_protseq(cps, TRUE);
 751       if (status != RPC_S_OK)
 752         break;
 753
 754       /* make sure server is actually listening on the interface before
 755        * returning */
 756       RPCRT4_sync_with_server_thread(cps);
 757     }
 758     LeaveCriticalSection(&server_cs);
 759   }
 760
 761   return status;
 762 }
 763
 764 static void RPCRT4_stop_listen(BOOL auto_listen)
 765 {
 766   EnterCriticalSection(&listen_cs);
 767   if (auto_listen || (--manual_listen_count == 0))
 768   {
 769     if (listen_count != 0 && --listen_count == 0) {
 770       RpcServerProtseq *cps;
 771
 772       std_listen = FALSE;
 773       LeaveCriticalSection(&listen_cs);
 774
 775       LIST_FOR_EACH_ENTRY(cps, &protseqs, RpcServerProtseq, entry)
 776         RPCRT4_sync_with_server_thread(cps);
 777
 778       EnterCriticalSection(&listen_cs);
 779       if (listen_done_event) SetEvent( listen_done_event );
 780       listen_done_event = 0;
 781       LeaveCriticalSection(&listen_cs);
 782       return;
 783     }
 784     assert(listen_count >= 0);
 785   }
 786   LeaveCriticalSection(&listen_cs);
 787 }
 788
 789 static BOOL RPCRT4_protseq_is_endpoint_registered(RpcServerProtseq *protseq, const char *endpoint)
 790 {
 791   RpcConnection *conn;
 792   EnterCriticalSection(&protseq->cs);
 793   for (conn = protseq->conn; conn; conn = conn->Next)
 794   {
 795     if (!endpoint || !strcmp(endpoint, conn->Endpoint))
 796       break;
 797   }
 798   LeaveCriticalSection(&protseq->cs);
 799   return (conn != NULL);
 800 }
 801
 802 static RPC_STATUS RPCRT4_use_protseq(RpcServerProtseq* ps, const char *endpoint)
 803 {
 804   RPC_STATUS status;
 805
 806   EnterCriticalSection(&ps->cs);
 807
 808   if (RPCRT4_protseq_is_endpoint_registered(ps, endpoint))
 809     status = RPC_S_OK;
 810   else
 811     status = ps->ops->open_endpoint(ps, endpoint);
 812
 813   LeaveCriticalSection(&ps->cs);
 814
 815   if (status != RPC_S_OK)
 816     return status;
 817
 818   if (std_listen)
 819   {
 820     status = RPCRT4_start_listen_protseq(ps, FALSE);
 821     if (status == RPC_S_OK)
 822       RPCRT4_sync_with_server_thread(ps);
 823   }
 824
 825   return status;
 826 }
 827
 828 /***********************************************************************
 829  *             RpcServerInqBindings (RPCRT4.@)
 830  */
 831 RPC_STATUS WINAPI RpcServerInqBindings( RPC_BINDING_VECTOR** BindingVector )
 832 {
 833   RPC_STATUS status;
 834   DWORD count;
 835   RpcServerProtseq* ps;
 836   RpcConnection* conn;
 837
 838   if (BindingVector)
 839     TRACE("(*BindingVector == ^%p)\n", *BindingVector);
 840   else
 841     ERR("(BindingVector == NULL!!?)\n");
 842
 843   EnterCriticalSection(&server_cs);
 844   /* count connections */
 845   count = 0;
 846   LIST_FOR_EACH_ENTRY(ps, &protseqs, RpcServerProtseq, entry) {
 847     EnterCriticalSection(&ps->cs);
 848     for (conn = ps->conn; conn; conn = conn->Next)
 849       count++;
 850     LeaveCriticalSection(&ps->cs);
 851   }
 852   if (count) {
 853     /* export bindings */
 854     *BindingVector = HeapAlloc(GetProcessHeap(), 0,
 855                               sizeof(RPC_BINDING_VECTOR) +
 856                               sizeof(RPC_BINDING_HANDLE)*(count-1));
 857     (*BindingVector)->Count = count;
 858     count = 0;
 859     LIST_FOR_EACH_ENTRY(ps, &protseqs, RpcServerProtseq, entry) {
 860       EnterCriticalSection(&ps->cs);
 861       for (conn = ps->conn; conn; conn = conn->Next) {
 862        RPCRT4_MakeBinding((RpcBinding**)&(*BindingVector)->BindingH[count],
 863                           conn);
 864        count++;
 865       }
 866       LeaveCriticalSection(&ps->cs);
 867     }
 868     status = RPC_S_OK;
 869   } else {
 870     *BindingVector = NULL;
 871     status = RPC_S_NO_BINDINGS;
 872   }
 873   LeaveCriticalSection(&server_cs);
 874   return status;
 875 }
 876
 877 /***********************************************************************
 878  *             RpcServerUseProtseqEpA (RPCRT4.@)
 879  */
 880 RPC_STATUS WINAPI RpcServerUseProtseqEpA( RPC_CSTR Protseq, UINT MaxCalls, RPC_CSTR Endpoint, LPVOID SecurityDescriptor )
 881 {
 882   RPC_POLICY policy;
 883
 884   TRACE( "(%s,%u,%s,%p)\n", Protseq, MaxCalls, Endpoint, SecurityDescriptor );
 885
 886   /* This should provide the default behaviour */
 887   policy.Length        = sizeof( policy );
 888   policy.EndpointFlags = 0;
 889   policy.NICFlags      = 0;
 890
 891   return RpcServerUseProtseqEpExA( Protseq, MaxCalls, Endpoint, SecurityDescriptor, &policy );
 892 }
 893
 894 /***********************************************************************
 895  *             RpcServerUseProtseqEpW (RPCRT4.@)
 896  */
 897 RPC_STATUS WINAPI RpcServerUseProtseqEpW( RPC_WSTR Protseq, UINT MaxCalls, RPC_WSTR Endpoint, LPVOID SecurityDescriptor )
 898 {
 899   RPC_POLICY policy;
 900
 901   TRACE( "(%s,%u,%s,%p)\n", debugstr_w( Protseq ), MaxCalls, debugstr_w( Endpoint ), SecurityDescriptor );
 902
 903   /* This should provide the default behaviour */
 904   policy.Length        = sizeof( policy );
 905   policy.EndpointFlags = 0;
 906   policy.NICFlags      = 0;
 907
 908   return RpcServerUseProtseqEpExW( Protseq, MaxCalls, Endpoint, SecurityDescriptor, &policy );
 909 }
 910
 911 /***********************************************************************
 912  *             alloc_serverprotoseq (internal)
 913  *
 914  * Must be called with server_cs held.
 915  */
 916 static RPC_STATUS alloc_serverprotoseq(UINT MaxCalls, const char *Protseq, RpcServerProtseq **ps)
 917 {
 918   const struct protseq_ops *ops = rpcrt4_get_protseq_ops(Protseq);
 919
 920   if (!ops)
 921   {
 922     FIXME("protseq %s not supported\n", debugstr_a(Protseq));
 923     return RPC_S_PROTSEQ_NOT_SUPPORTED;
 924   }
 925
 926   *ps = ops->alloc();
 927   if (!*ps)
 928     return RPC_S_OUT_OF_RESOURCES;
 929   (*ps)->MaxCalls = MaxCalls;
 930   (*ps)->Protseq = RPCRT4_strdupA(Protseq);
 931   (*ps)->ops = ops;
 932   (*ps)->MaxCalls = 0;
 933   (*ps)->conn = NULL;
 934   InitializeCriticalSection(&(*ps)->cs);
 935   (*ps)->cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": RpcServerProtseq.cs");
 936   (*ps)->is_listening = FALSE;
 937   (*ps)->mgr_mutex = NULL;
 938   (*ps)->server_ready_event = NULL;
 939
 940   list_add_head(&protseqs, &(*ps)->entry);
 941
 942   TRACE("new protseq %p created for %s\n", *ps, Protseq);
 943
 944   return RPC_S_OK;
 945 }
 946
 947 /* must be called with server_cs held */
 948 static void destroy_serverprotoseq(RpcServerProtseq *ps)
 949 {
 950     RPCRT4_strfree(ps->Protseq);
 951     ps->cs.DebugInfo->Spare[0] = 0;
 952     DeleteCriticalSection(&ps->cs);
 953     CloseHandle(ps->mgr_mutex);
 954     CloseHandle(ps->server_ready_event);
 955     list_remove(&ps->entry);
 956     HeapFree(GetProcessHeap(), 0, ps);
 957 }
 958
 959 /* Finds a given protseq or creates a new one if one doesn't already exist */
 960 static RPC_STATUS RPCRT4_get_or_create_serverprotseq(UINT MaxCalls, const char *Protseq, RpcServerProtseq **ps)
 961 {
 962     RPC_STATUS status;
 963     RpcServerProtseq *cps;
 964
 965     EnterCriticalSection(&server_cs);
 966
 967     LIST_FOR_EACH_ENTRY(cps, &protseqs, RpcServerProtseq, entry)
 968         if (!strcmp(cps->Protseq, Protseq))
 969         {
 970             TRACE("found existing protseq object for %s\n", Protseq);
 971             *ps = cps;
 972             LeaveCriticalSection(&server_cs);
 973             return S_OK;
 974         }
 975
 976     status = alloc_serverprotoseq(MaxCalls, Protseq, ps);
 977
 978     LeaveCriticalSection(&server_cs);
 979
 980     return status;
 981 }
 982
 983 /***********************************************************************
 984  *             RpcServerUseProtseqEpExA (RPCRT4.@)
 985  */
 986 RPC_STATUS WINAPI RpcServerUseProtseqEpExA( RPC_CSTR Protseq, UINT MaxCalls, RPC_CSTR Endpoint, LPVOID SecurityDescriptor,
 987                                             PRPC_POLICY lpPolicy )
 988 {
 989   RpcServerProtseq* ps;
 990   RPC_STATUS status;
 991
 992   TRACE("(%s,%u,%s,%p,{%u,%u,%u})\n", debugstr_a((const char *)Protseq),
 993        MaxCalls, debugstr_a((const char *)Endpoint), SecurityDescriptor,
 994        lpPolicy->Length, lpPolicy->EndpointFlags, lpPolicy->NICFlags );
 995
 996   status = RPCRT4_get_or_create_serverprotseq(MaxCalls, (const char *)Protseq, &ps);
 997   if (status != RPC_S_OK)
 998     return status;
 999
1000   return RPCRT4_use_protseq(ps, (const char *)Endpoint);
1001 }
1002
1003 /***********************************************************************
1004  *             RpcServerUseProtseqEpExW (RPCRT4.@)
1005  */
1006 RPC_STATUS WINAPI RpcServerUseProtseqEpExW( RPC_WSTR Protseq, UINT MaxCalls, RPC_WSTR Endpoint, LPVOID SecurityDescriptor,
1007                                             PRPC_POLICY lpPolicy )
1008 {
1009   RpcServerProtseq* ps;
1010   RPC_STATUS status;
1011   LPSTR ProtseqA;
1012   LPSTR EndpointA;
1013
1014   TRACE("(%s,%u,%s,%p,{%u,%u,%u})\n", debugstr_w( Protseq ), MaxCalls,
1015        debugstr_w( Endpoint ), SecurityDescriptor,
1016        lpPolicy->Length, lpPolicy->EndpointFlags, lpPolicy->NICFlags );
1017
1018   ProtseqA = RPCRT4_strdupWtoA(Protseq);
1019   status = RPCRT4_get_or_create_serverprotseq(MaxCalls, ProtseqA, &ps);
1020   RPCRT4_strfree(ProtseqA);
1021   if (status != RPC_S_OK)
1022     return status;
1023
1024   EndpointA = RPCRT4_strdupWtoA(Endpoint);
1025   status = RPCRT4_use_protseq(ps, EndpointA);
1026   RPCRT4_strfree(EndpointA);
1027   return status;
1028 }
1029
1030 /***********************************************************************
1031  *             RpcServerUseProtseqA (RPCRT4.@)
1032  */
1033 RPC_STATUS WINAPI RpcServerUseProtseqA(RPC_CSTR Protseq, unsigned int MaxCalls, void *SecurityDescriptor)
1034 {
1035   RPC_STATUS status;
1036   RpcServerProtseq* ps;
1037
1038   TRACE("(Protseq == %s, MaxCalls == %d, SecurityDescriptor == ^%p)\n", debugstr_a((char*)Protseq), MaxCalls, SecurityDescriptor);
1039
1040   status = RPCRT4_get_or_create_serverprotseq(MaxCalls, (const char *)Protseq, &ps);
1041   if (status != RPC_S_OK)
1042     return status;
1043
1044   return RPCRT4_use_protseq(ps, NULL);
1045 }
1046
1047 /***********************************************************************
1048  *             RpcServerUseProtseqW (RPCRT4.@)
1049  */
1050 RPC_STATUS WINAPI RpcServerUseProtseqW(RPC_WSTR Protseq, unsigned int MaxCalls, void *SecurityDescriptor)
1051 {
1052   RPC_STATUS status;
1053   RpcServerProtseq* ps;
1054   LPSTR ProtseqA;
1055
1056   TRACE("Protseq == %s, MaxCalls == %d, SecurityDescriptor == ^%p)\n", debugstr_w(Protseq), MaxCalls, SecurityDescriptor);
1057
1058   ProtseqA = RPCRT4_strdupWtoA(Protseq);
1059   status = RPCRT4_get_or_create_serverprotseq(MaxCalls, ProtseqA, &ps);
1060   RPCRT4_strfree(ProtseqA);
1061   if (status != RPC_S_OK)
1062     return status;
1063
1064   return RPCRT4_use_protseq(ps, NULL);
1065 }
1066
1067 void RPCRT4_destroy_all_protseqs(void)
1068 {
1069     RpcServerProtseq *cps, *cursor2;
1070
1071     if (listen_count != 0)
1072         std_listen = FALSE;
1073
1074     EnterCriticalSection(&server_cs);
1075     LIST_FOR_EACH_ENTRY_SAFE(cps, cursor2, &protseqs, RpcServerProtseq, entry)
1076     {
1077         if (listen_count != 0)
1078             RPCRT4_sync_with_server_thread(cps);
1079         destroy_serverprotoseq(cps);
1080     }
1081     LeaveCriticalSection(&server_cs);
1082     DeleteCriticalSection(&server_cs);
1083     DeleteCriticalSection(&listen_cs);
1084 }
1085
1086 /***********************************************************************
1087  *             RpcServerRegisterIf (RPCRT4.@)
1088  */
1089 RPC_STATUS WINAPI RpcServerRegisterIf( RPC_IF_HANDLE IfSpec, UUID* MgrTypeUuid, RPC_MGR_EPV* MgrEpv )
1090 {
1091   TRACE("(%p,%s,%p)\n", IfSpec, debugstr_guid(MgrTypeUuid), MgrEpv);
1092   return RpcServerRegisterIf2( IfSpec, MgrTypeUuid, MgrEpv, 0, RPC_C_LISTEN_MAX_CALLS_DEFAULT, (UINT)-1, NULL );
1093 }
1094
1095 /***********************************************************************
1096  *             RpcServerRegisterIfEx (RPCRT4.@)
1097  */
1098 RPC_STATUS WINAPI RpcServerRegisterIfEx( RPC_IF_HANDLE IfSpec, UUID* MgrTypeUuid, RPC_MGR_EPV* MgrEpv,
1099                        UINT Flags, UINT MaxCalls, RPC_IF_CALLBACK_FN* IfCallbackFn )
1100 {
1101   TRACE("(%p,%s,%p,%u,%u,%p)\n", IfSpec, debugstr_guid(MgrTypeUuid), MgrEpv, Flags, MaxCalls, IfCallbackFn);
1102   return RpcServerRegisterIf2( IfSpec, MgrTypeUuid, MgrEpv, Flags, MaxCalls, (UINT)-1, IfCallbackFn );
1103 }
1104
1105 /***********************************************************************
1106  *             RpcServerRegisterIf2 (RPCRT4.@)
1107  */
1108 RPC_STATUS WINAPI RpcServerRegisterIf2( RPC_IF_HANDLE IfSpec, UUID* MgrTypeUuid, RPC_MGR_EPV* MgrEpv,
1109                       UINT Flags, UINT MaxCalls, UINT MaxRpcSize, RPC_IF_CALLBACK_FN* IfCallbackFn )
1110 {
1111   PRPC_SERVER_INTERFACE If = IfSpec;
1112   RpcServerInterface* sif;
1113   unsigned int i;
1114
1115   TRACE("(%p,%s,%p,%u,%u,%u,%p)\n", IfSpec, debugstr_guid(MgrTypeUuid), MgrEpv, Flags, MaxCalls,
1116          MaxRpcSize, IfCallbackFn);
1117   TRACE(" interface id: %s %d.%d\n", debugstr_guid(&If->InterfaceId.SyntaxGUID),
1118                                      If->InterfaceId.SyntaxVersion.MajorVersion,
1119                                      If->InterfaceId.SyntaxVersion.MinorVersion);
1120   TRACE(" transfer syntax: %s %d.%d\n", debugstr_guid(&If->TransferSyntax.SyntaxGUID),
1121                                         If->TransferSyntax.SyntaxVersion.MajorVersion,
1122                                         If->TransferSyntax.SyntaxVersion.MinorVersion);
1123   TRACE(" dispatch table: %p\n", If->DispatchTable);
1124   if (If->DispatchTable) {
1125     TRACE("  dispatch table count: %d\n", If->DispatchTable->DispatchTableCount);
1126     for (i=0; i<If->DispatchTable->DispatchTableCount; i++) {
1127       TRACE("   entry %d: %p\n", i, If->DispatchTable->DispatchTable[i]);
1128     }
1129     TRACE("  reserved: %ld\n", If->DispatchTable->Reserved);
1130   }
1131   TRACE(" protseq endpoint count: %d\n", If->RpcProtseqEndpointCount);
1132   TRACE(" default manager epv: %p\n", If->DefaultManagerEpv);
1133   TRACE(" interpreter info: %p\n", If->InterpreterInfo);
1134
1135   sif = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(RpcServerInterface));
1136   sif->If           = If;
1137   if (MgrTypeUuid) {
1138     sif->MgrTypeUuid = *MgrTypeUuid;
1139     sif->MgrEpv       = MgrEpv;
1140   } else {
1141     memset(&sif->MgrTypeUuid, 0, sizeof(UUID));
1142     sif->MgrEpv       = If->DefaultManagerEpv;
1143   }
1144   sif->Flags        = Flags;
1145   sif->MaxCalls     = MaxCalls;
1146   sif->MaxRpcSize   = MaxRpcSize;
1147   sif->IfCallbackFn = IfCallbackFn;
1148
1149   EnterCriticalSection(&server_cs);
1150   list_add_head(&server_interfaces, &sif->entry);
1151   LeaveCriticalSection(&server_cs);
1152
1153   if (sif->Flags & RPC_IF_AUTOLISTEN)
1154       RPCRT4_start_listen(TRUE);
1155
1156   return RPC_S_OK;
1157 }
1158
1159 /***********************************************************************
1160  *             RpcServerUnregisterIf (RPCRT4.@)
1161  */
1162 RPC_STATUS WINAPI RpcServerUnregisterIf( RPC_IF_HANDLE IfSpec, UUID* MgrTypeUuid, UINT WaitForCallsToComplete )
1163 {
1164   PRPC_SERVER_INTERFACE If = IfSpec;
1165   HANDLE event = NULL;
1166   BOOL found = FALSE;
1167   BOOL completed = TRUE;
1168   RpcServerInterface *cif;
1169   RPC_STATUS status;
1170
1171   TRACE("(IfSpec == (RPC_IF_HANDLE)^%p (%s), MgrTypeUuid == %s, WaitForCallsToComplete == %u)\n",
1172     IfSpec, debugstr_guid(&If->InterfaceId.SyntaxGUID), debugstr_guid(MgrTypeUuid), WaitForCallsToComplete);
1173
1174   EnterCriticalSection(&server_cs);
1175   LIST_FOR_EACH_ENTRY(cif, &server_interfaces, RpcServerInterface, entry) {
1176     if ((!IfSpec || !memcmp(&If->InterfaceId, &cif->If->InterfaceId, sizeof(RPC_SYNTAX_IDENTIFIER))) &&
1177         UuidEqual(MgrTypeUuid, &cif->MgrTypeUuid, &status)) {
1178       list_remove(&cif->entry);
1179       TRACE("unregistering cif %p\n", cif);
1180       if (cif->CurrentCalls) {
1181         completed = FALSE;
1182         cif->Delete = TRUE;
1183         if (WaitForCallsToComplete)
1184           cif->CallsCompletedEvent = event = CreateEventW(NULL, FALSE, FALSE, NULL);
1185       }
1186       found = TRUE;
1187       break;
1188     }
1189   }
1190   LeaveCriticalSection(&server_cs);
1191
1192   if (!found) {
1193     ERR("not found for object %s\n", debugstr_guid(MgrTypeUuid));
1194     return RPC_S_UNKNOWN_IF;
1195   }
1196
1197   if (completed)
1198     HeapFree(GetProcessHeap(), 0, cif);
1199   else if (event) {
1200     /* sif will be freed when the last call is completed, so be careful not to
1201      * touch that memory here as that could happen before we get here */
1202     WaitForSingleObject(event, INFINITE);
1203     CloseHandle(event);
1204   }
1205
1206   return RPC_S_OK;
1207 }
1208
1209 /***********************************************************************
1210  *             RpcServerUnregisterIfEx (RPCRT4.@)
1211  */
1212 RPC_STATUS WINAPI RpcServerUnregisterIfEx( RPC_IF_HANDLE IfSpec, UUID* MgrTypeUuid, int RundownContextHandles )
1213 {
1214   FIXME("(IfSpec == (RPC_IF_HANDLE)^%p, MgrTypeUuid == %s, RundownContextHandles == %d): stub\n",
1215     IfSpec, debugstr_guid(MgrTypeUuid), RundownContextHandles);
1216
1217   return RPC_S_OK;
1218 }
1219
1220 /***********************************************************************
1221  *             RpcObjectSetType (RPCRT4.@)
1222  *
1223  * PARAMS
1224  *   ObjUuid  [I] "Object" UUID
1225  *   TypeUuid [I] "Type" UUID
1226  *
1227  * RETURNS
1228  *   RPC_S_OK                 The call succeeded
1229  *   RPC_S_INVALID_OBJECT     The provided object (nil) is not valid
1230  *   RPC_S_ALREADY_REGISTERED The provided object is already registered
1231  *
1232  * Maps "Object" UUIDs to "Type" UUIDs.  Passing the nil UUID as the type
1233  * resets the mapping for the specified object UUID to nil (the default).
1234  * The nil object is always associated with the nil type and cannot be
1235  * reassigned.  Servers can support multiple implementations on the same
1236  * interface by registering different end-point vectors for the different
1237  * types.  There's no need to call this if a server only supports the nil
1238  * type, as is typical.
1239  */
1240 RPC_STATUS WINAPI RpcObjectSetType( UUID* ObjUuid, UUID* TypeUuid )
1241 {
1242   RpcObjTypeMap *map = RpcObjTypeMaps, *prev = NULL;
1243   RPC_STATUS dummy;
1244
1245   TRACE("(ObjUUID == %s, TypeUuid == %s).\n", debugstr_guid(ObjUuid), debugstr_guid(TypeUuid));
1246   if ((! ObjUuid) || UuidIsNil(ObjUuid, &dummy)) {
1247     /* nil uuid cannot be remapped */
1248     return RPC_S_INVALID_OBJECT;
1249   }
1250
1251   /* find the mapping for this object if there is one ... */
1252   while (map) {
1253     if (! UuidCompare(ObjUuid, &map->Object, &dummy)) break;
1254     prev = map;
1255     map = map->next;
1256   }
1257   if ((! TypeUuid) || UuidIsNil(TypeUuid, &dummy)) {
1258     /* ... and drop it from the list */
1259     if (map) {
1260       if (prev)
1261         prev->next = map->next;
1262       else
1263         RpcObjTypeMaps = map->next;
1264       HeapFree(GetProcessHeap(), 0, map);
1265     }
1266   } else {
1267     /* ... , fail if we found it ... */
1268     if (map)
1269       return RPC_S_ALREADY_REGISTERED;
1270     /* ... otherwise create a new one and add it in. */
1271     map = HeapAlloc(GetProcessHeap(), 0, sizeof(RpcObjTypeMap));
1272     map->Object = *ObjUuid;
1273     map->Type = *TypeUuid;
1274     map->next = NULL;
1275     if (prev)
1276       prev->next = map; /* prev is the last map in the linklist */
1277     else
1278       RpcObjTypeMaps = map;
1279   }
1280
1281   return RPC_S_OK;
1282 }
1283
1284 struct rpc_server_registered_auth_info
1285 {
1286     struct list entry;
1287     TimeStamp exp;
1288     CredHandle cred;
1289     ULONG max_token;
1290     USHORT auth_type;
1291 };
1292
1293 RPC_STATUS RPCRT4_ServerGetRegisteredAuthInfo(
1294     USHORT auth_type, CredHandle *cred, TimeStamp *exp, ULONG *max_token)
1295 {
1296     RPC_STATUS status = RPC_S_UNKNOWN_AUTHN_SERVICE;
1297     struct rpc_server_registered_auth_info *auth_info;
1298
1299     EnterCriticalSection(&server_auth_info_cs);
1300     LIST_FOR_EACH_ENTRY(auth_info, &server_registered_auth_info, struct rpc_server_registered_auth_info, entry)
1301     {
1302         if (auth_info->auth_type == auth_type)
1303         {
1304             *cred = auth_info->cred;
1305             *exp = auth_info->exp;
1306             *max_token = auth_info->max_token;
1307             status = RPC_S_OK;
1308             break;
1309         }
1310     }
1311     LeaveCriticalSection(&server_auth_info_cs);
1312
1313     return status;
1314 }
1315
1316 void RPCRT4_ServerFreeAllRegisteredAuthInfo(void)
1317 {
1318     struct rpc_server_registered_auth_info *auth_info, *cursor2;
1319
1320     EnterCriticalSection(&server_auth_info_cs);
1321     LIST_FOR_EACH_ENTRY_SAFE(auth_info, cursor2, &server_registered_auth_info, struct rpc_server_registered_auth_info, entry)
1322     {
1323         FreeCredentialsHandle(&auth_info->cred);
1324         HeapFree(GetProcessHeap(), 0, auth_info);
1325     }
1326     LeaveCriticalSection(&server_auth_info_cs);
1327     DeleteCriticalSection(&server_auth_info_cs);
1328 }
1329
1330 /***********************************************************************
1331  *             RpcServerRegisterAuthInfoA (RPCRT4.@)
1332  */
1333 RPC_STATUS WINAPI RpcServerRegisterAuthInfoA( RPC_CSTR ServerPrincName, ULONG AuthnSvc, RPC_AUTH_KEY_RETRIEVAL_FN GetKeyFn,
1334                             LPVOID Arg )
1335 {
1336     SECURITY_STATUS sec_status;
1337     CredHandle cred;
1338     TimeStamp exp;
1339     ULONG package_count;
1340     ULONG i;
1341     PSecPkgInfoA packages;
1342     ULONG max_token;
1343     struct rpc_server_registered_auth_info *auth_info;
1344
1345     TRACE("(%s,%u,%p,%p)\n", ServerPrincName, AuthnSvc, GetKeyFn, Arg);
1346
1347     sec_status = EnumerateSecurityPackagesA(&package_count, &packages);
1348     if (sec_status != SEC_E_OK)
1349     {
1350         ERR("EnumerateSecurityPackagesA failed with error 0x%08x\n",
1351             sec_status);
1352         return RPC_S_SEC_PKG_ERROR;
1353     }
1354
1355     for (i = 0; i < package_count; i++)
1356         if (packages[i].wRPCID == AuthnSvc)
1357             break;
1358
1359     if (i == package_count)
1360     {
1361         WARN("unsupported AuthnSvc %u\n", AuthnSvc);
1362         FreeContextBuffer(packages);
1363         return RPC_S_UNKNOWN_AUTHN_SERVICE;
1364     }
1365     TRACE("found package %s for service %u\n", packages[i].Name,
1366           AuthnSvc);
1367     sec_status = AcquireCredentialsHandleA((SEC_CHAR *)ServerPrincName,
1368                                            packages[i].Name,
1369                                            SECPKG_CRED_INBOUND, NULL, NULL,
1370                                            NULL, NULL, &cred, &exp);
1371     max_token = packages[i].cbMaxToken;
1372     FreeContextBuffer(packages);
1373     if (sec_status != SEC_E_OK)
1374         return RPC_S_SEC_PKG_ERROR;
1375
1376     auth_info = HeapAlloc(GetProcessHeap(), 0, sizeof(*auth_info));
1377     if (!auth_info)
1378     {
1379         FreeCredentialsHandle(&cred);
1380         return RPC_S_OUT_OF_RESOURCES;
1381     }
1382
1383     auth_info->exp = exp;
1384     auth_info->cred = cred;
1385     auth_info->max_token = max_token;
1386     auth_info->auth_type = AuthnSvc;
1387
1388     EnterCriticalSection(&server_auth_info_cs);
1389     list_add_tail(&server_registered_auth_info, &auth_info->entry);
1390     LeaveCriticalSection(&server_auth_info_cs);
1391
1392     return RPC_S_OK;
1393 }
1394
1395 /***********************************************************************
1396  *             RpcServerRegisterAuthInfoW (RPCRT4.@)
1397  */
1398 RPC_STATUS WINAPI RpcServerRegisterAuthInfoW( RPC_WSTR ServerPrincName, ULONG AuthnSvc, RPC_AUTH_KEY_RETRIEVAL_FN GetKeyFn,
1399                             LPVOID Arg )
1400 {
1401     SECURITY_STATUS sec_status;
1402     CredHandle cred;
1403     TimeStamp exp;
1404     ULONG package_count;
1405     ULONG i;
1406     PSecPkgInfoW packages;
1407     ULONG max_token;
1408     struct rpc_server_registered_auth_info *auth_info;
1409
1410     TRACE("(%s,%u,%p,%p)\n", debugstr_w(ServerPrincName), AuthnSvc, GetKeyFn, Arg);
1411
1412     sec_status = EnumerateSecurityPackagesW(&package_count, &packages);
1413     if (sec_status != SEC_E_OK)
1414     {
1415         ERR("EnumerateSecurityPackagesW failed with error 0x%08x\n",
1416             sec_status);
1417         return RPC_S_SEC_PKG_ERROR;
1418     }
1419
1420     for (i = 0; i < package_count; i++)
1421         if (packages[i].wRPCID == AuthnSvc)
1422             break;
1423
1424     if (i == package_count)
1425     {
1426         WARN("unsupported AuthnSvc %u\n", AuthnSvc);
1427         FreeContextBuffer(packages);
1428         return RPC_S_UNKNOWN_AUTHN_SERVICE;
1429     }
1430     TRACE("found package %s for service %u\n", debugstr_w(packages[i].Name),
1431           AuthnSvc);
1432     sec_status = AcquireCredentialsHandleW((SEC_WCHAR *)ServerPrincName,
1433                                            packages[i].Name,
1434                                            SECPKG_CRED_INBOUND, NULL, NULL,
1435                                            NULL, NULL, &cred, &exp);
1436     max_token = packages[i].cbMaxToken;
1437     FreeContextBuffer(packages);
1438     if (sec_status != SEC_E_OK)
1439         return RPC_S_SEC_PKG_ERROR;
1440
1441     auth_info = HeapAlloc(GetProcessHeap(), 0, sizeof(*auth_info));
1442     if (!auth_info)
1443     {
1444         FreeCredentialsHandle(&cred);
1445         return RPC_S_OUT_OF_RESOURCES;
1446     }
1447
1448     auth_info->exp = exp;
1449     auth_info->cred = cred;
1450     auth_info->max_token = max_token;
1451     auth_info->auth_type = AuthnSvc;
1452
1453     EnterCriticalSection(&server_auth_info_cs);
1454     list_add_tail(&server_registered_auth_info, &auth_info->entry);
1455     LeaveCriticalSection(&server_auth_info_cs);
1456
1457     return RPC_S_OK;
1458 }
1459
1460 /***********************************************************************
1461  *             RpcServerListen (RPCRT4.@)
1462  */
1463 RPC_STATUS WINAPI RpcServerListen( UINT MinimumCallThreads, UINT MaxCalls, UINT DontWait )
1464 {
1465   RPC_STATUS status = RPC_S_OK;
1466
1467   TRACE("(%u,%u,%u)\n", MinimumCallThreads, MaxCalls, DontWait);
1468
1469   if (list_empty(&protseqs))
1470     return RPC_S_NO_PROTSEQS_REGISTERED;
1471
1472   status = RPCRT4_start_listen(FALSE);
1473
1474   if (DontWait || (status != RPC_S_OK)) return status;
1475
1476   return RpcMgmtWaitServerListen();
1477 }
1478
1479 /***********************************************************************
1480  *             RpcMgmtServerWaitListen (RPCRT4.@)
1481  */
1482 RPC_STATUS WINAPI RpcMgmtWaitServerListen( void )
1483 {
1484   HANDLE event;
1485
1486   TRACE("()\n");
1487
1488   EnterCriticalSection(&listen_cs);
1489
1490   if (!std_listen) {
1491     LeaveCriticalSection(&listen_cs);
1492     return RPC_S_NOT_LISTENING;
1493   }
1494   if (listen_done_event) {
1495     LeaveCriticalSection(&listen_cs);
1496     return RPC_S_ALREADY_LISTENING;
1497   }
1498   event = CreateEventW( NULL, TRUE, FALSE, NULL );
1499   listen_done_event = event;
1500
1501   LeaveCriticalSection(&listen_cs);
1502
1503   TRACE( "waiting for server calls to finish\n" );
1504   WaitForSingleObject( event, INFINITE );
1505   TRACE( "done waiting\n" );
1506
1507   CloseHandle( event );
1508   return RPC_S_OK;
1509 }
1510
1511 /***********************************************************************
1512  *             RpcMgmtStopServerListening (RPCRT4.@)
1513  */
1514 RPC_STATUS WINAPI RpcMgmtStopServerListening ( RPC_BINDING_HANDLE Binding )
1515 {
1516   TRACE("(Binding == (RPC_BINDING_HANDLE)^%p)\n", Binding);
1517
1518   if (Binding) {
1519     FIXME("client-side invocation not implemented.\n");
1520     return RPC_S_WRONG_KIND_OF_BINDING;
1521   }
1522
1523   RPCRT4_stop_listen(FALSE);
1524
1525   return RPC_S_OK;
1526 }
1527
1528 /***********************************************************************
1529  *             RpcMgmtEnableIdleCleanup (RPCRT4.@)
1530  */
1531 RPC_STATUS WINAPI RpcMgmtEnableIdleCleanup(void)
1532 {
1533     FIXME("(): stub\n");
1534     return RPC_S_OK;
1535 }
1536
1537 /***********************************************************************
1538  *             I_RpcServerStartListening (RPCRT4.@)
1539  */
1540 RPC_STATUS WINAPI I_RpcServerStartListening( HWND hWnd )
1541 {
1542   FIXME( "(%p): stub\n", hWnd );
1543
1544   return RPC_S_OK;
1545 }
1546
1547 /***********************************************************************
1548  *             I_RpcServerStopListening (RPCRT4.@)
1549  */
1550 RPC_STATUS WINAPI I_RpcServerStopListening( void )
1551 {
1552   FIXME( "(): stub\n" );
1553
1554   return RPC_S_OK;
1555 }
1556
1557 /***********************************************************************
1558  *             I_RpcWindowProc (RPCRT4.@)
1559  */
1560 UINT WINAPI I_RpcWindowProc( void *hWnd, UINT Message, UINT wParam, ULONG lParam )
1561 {
1562   FIXME( "(%p,%08x,%08x,%08x): stub\n", hWnd, Message, wParam, lParam );
1563
1564   return 0;
1565 }
1566
1567 /***********************************************************************
1568  *             RpcMgmtInqIfIds (RPCRT4.@)
1569  */
1570 RPC_STATUS WINAPI RpcMgmtInqIfIds(RPC_BINDING_HANDLE Binding, RPC_IF_ID_VECTOR **IfIdVector)
1571 {
1572   FIXME("(%p,%p): stub\n", Binding, IfIdVector);
1573   return RPC_S_INVALID_BINDING;
1574 }
1575
1576 /***********************************************************************
1577  *             RpcMgmtInqStats (RPCRT4.@)
1578  */
1579 RPC_STATUS WINAPI RpcMgmtInqStats(RPC_BINDING_HANDLE Binding, RPC_STATS_VECTOR **Statistics)
1580 {
1581   RPC_STATS_VECTOR *stats;
1582
1583   FIXME("(%p,%p)\n", Binding, Statistics);
1584
1585   if ((stats = HeapAlloc(GetProcessHeap(), 0, sizeof(RPC_STATS_VECTOR))))
1586   {
1587     stats->Count = 1;
1588     stats->Stats[0] = 0;
1589     *Statistics = stats;
1590     return RPC_S_OK;
1591   }
1592   return RPC_S_OUT_OF_RESOURCES;
1593 }
1594
1595 /***********************************************************************
1596  *             RpcMgmtStatsVectorFree (RPCRT4.@)
1597  */
1598 RPC_STATUS WINAPI RpcMgmtStatsVectorFree(RPC_STATS_VECTOR **StatsVector)
1599 {
1600   FIXME("(%p)\n", StatsVector);
1601
1602   if (StatsVector)
1603   {
1604     HeapFree(GetProcessHeap(), 0, *StatsVector);
1605     *StatsVector = NULL;
1606   }
1607   return RPC_S_OK;
1608 }
1609
1610 /***********************************************************************
1611  *             RpcMgmtEpEltInqBegin (RPCRT4.@)
1612  */
1613 RPC_STATUS WINAPI RpcMgmtEpEltInqBegin(RPC_BINDING_HANDLE Binding, ULONG InquiryType,
1614     RPC_IF_ID *IfId, ULONG VersOption, UUID *ObjectUuid, RPC_EP_INQ_HANDLE* InquiryContext)
1615 {
1616   FIXME("(%p,%u,%p,%u,%p,%p): stub\n",
1617         Binding, InquiryType, IfId, VersOption, ObjectUuid, InquiryContext);
1618   return RPC_S_INVALID_BINDING;
1619 }
1620
1621 /***********************************************************************
1622  *             RpcMgmtIsServerListening (RPCRT4.@)
1623  */
1624 RPC_STATUS WINAPI RpcMgmtIsServerListening(RPC_BINDING_HANDLE Binding)
1625 {
1626   FIXME("(%p): stub\n", Binding);
1627   return RPC_S_INVALID_BINDING;
1628 }
1629
1630 /***********************************************************************
1631  *             RpcMgmtSetAuthorizationFn (RPCRT4.@)
1632  */
1633 RPC_STATUS WINAPI RpcMgmtSetAuthorizationFn(RPC_MGMT_AUTHORIZATION_FN fn)
1634 {
1635   FIXME("(%p): stub\n", fn);
1636   return RPC_S_OK;
1637 }
1638
1639 /***********************************************************************
1640  *             RpcMgmtSetServerStackSize (RPCRT4.@)
1641  */
1642 RPC_STATUS WINAPI RpcMgmtSetServerStackSize(ULONG ThreadStackSize)
1643 {
1644   FIXME("(0x%x): stub\n", ThreadStackSize);
1645   return RPC_S_OK;
1646 }
1647
1648 /***********************************************************************
1649  *             I_RpcGetCurrentCallHandle (RPCRT4.@)
1650  */
1651 RPC_BINDING_HANDLE WINAPI I_RpcGetCurrentCallHandle(void)
1652 {
1653     TRACE("\n");
1654     return RPCRT4_GetThreadCurrentCallHandle();
1655 }