dlls/rpcrt4/rpc_binding.c

   1 /*
   2  * RPC binding API
   3  *
   4  * Copyright 2001 Ove Kåven, TransGaming Technologies
   5  * Copyright 2003 Mike Hearn
   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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20  *
  21  * TODO:
  22  *  - a whole lot
  23  */
  24
  25 #include <stdarg.h>
  26 #include <stdio.h>
  27 #include <string.h>
  28 #include <assert.h>
  29
  30 #include "windef.h"
  31 #include "winbase.h"
  32 #include "winnls.h"
  33 #include "winerror.h"
  34 #include "winreg.h"
  35 #include "winternl.h"
  36 #include "wine/unicode.h"
  37
  38 #include "rpc.h"
  39
  40 #include "wine/debug.h"
  41
  42 #include "rpc_binding.h"
  43
  44 WINE_DEFAULT_DEBUG_CHANNEL(ole);
  45
  46 static RpcConnection* conn_cache;
  47
  48 static CRITICAL_SECTION conn_cache_cs;
  49 static CRITICAL_SECTION_DEBUG critsect_debug =
  50 {
  51     0, 0, &conn_cache_cs,
  52     { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
  53       0, 0, { 0, (DWORD)(__FILE__ ": conn_cache_cs") }
  54 };
  55 static CRITICAL_SECTION conn_cache_cs = { &critsect_debug, -1, 0, 0, 0, 0 };
  56
  57 LPSTR RPCRT4_strndupA(LPCSTR src, INT slen)
  58 {
  59   DWORD len;
  60   LPSTR s;
  61   if (!src) return NULL;
  62   if (slen == -1) slen = strlen(src);
  63   len = slen;
  64   s = HeapAlloc(GetProcessHeap(), 0, len+1);
  65   memcpy(s, src, len);
  66   s[len] = 0;
  67   return s;
  68 }
  69
  70 LPSTR RPCRT4_strdupWtoA(LPWSTR src)
  71 {
  72   DWORD len;
  73   LPSTR s;
  74   if (!src) return NULL;
  75   len = WideCharToMultiByte(CP_ACP, 0, src, -1, NULL, 0, NULL, NULL);
  76   s = HeapAlloc(GetProcessHeap(), 0, len);
  77   WideCharToMultiByte(CP_ACP, 0, src, -1, s, len, NULL, NULL);
  78   return s;
  79 }
  80
  81 LPWSTR RPCRT4_strdupAtoW(LPSTR src)
  82 {
  83   DWORD len;
  84   LPWSTR s;
  85   if (!src) return NULL;
  86   len = MultiByteToWideChar(CP_ACP, 0, src, -1, NULL, 0);
  87   s = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR));
  88   MultiByteToWideChar(CP_ACP, 0, src, -1, s, len);
  89   return s;
  90 }
  91
  92 LPWSTR RPCRT4_strndupW(LPWSTR src, INT slen)
  93 {
  94   DWORD len;
  95   LPWSTR s;
  96   if (!src) return NULL;
  97   if (slen == -1) slen = strlenW(src);
  98   len = slen;
  99   s = HeapAlloc(GetProcessHeap(), 0, (len+1)*sizeof(WCHAR));
 100   memcpy(s, src, len*sizeof(WCHAR));
 101   s[len] = 0;
 102   return s;
 103 }
 104
 105 void RPCRT4_strfree(LPSTR src)
 106 {
 107   if (src) HeapFree(GetProcessHeap(), 0, src);
 108 }
 109
 110 RPC_STATUS RPCRT4_CreateConnection(RpcConnection** Connection, BOOL server, LPSTR Protseq, LPSTR NetworkAddr, LPSTR Endpoint, LPSTR NetworkOptions, RpcBinding* Binding)
 111 {
 112   RpcConnection* NewConnection;
 113
 114   NewConnection = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(RpcConnection));
 115   NewConnection->server = server;
 116   NewConnection->Protseq = RPCRT4_strdupA(Protseq);
 117   NewConnection->NetworkAddr = RPCRT4_strdupA(NetworkAddr);
 118   NewConnection->Endpoint = RPCRT4_strdupA(Endpoint);
 119   NewConnection->Used = Binding;
 120
 121   EnterCriticalSection(&conn_cache_cs);
 122   NewConnection->Next = conn_cache;
 123   conn_cache = NewConnection;
 124   LeaveCriticalSection(&conn_cache_cs);
 125
 126   TRACE("connection: %p\n", NewConnection);
 127   *Connection = NewConnection;
 128
 129   return RPC_S_OK;
 130 }
 131
 132 RPC_STATUS RPCRT4_DestroyConnection(RpcConnection* Connection)
 133 {
 134   RpcConnection* PrevConnection;
 135
 136   TRACE("connection: %p\n", Connection);
 137   if (Connection->Used) ERR("connection is still in use\n");
 138
 139   EnterCriticalSection(&conn_cache_cs);
 140   PrevConnection = conn_cache;
 141   if (PrevConnection == Connection) {
 142     conn_cache = Connection->Next;
 143   } else {
 144     while (PrevConnection && PrevConnection->Next != Connection)
 145       PrevConnection = PrevConnection->Next;
 146     if (PrevConnection) PrevConnection->Next = Connection->Next;
 147   }
 148   LeaveCriticalSection(&conn_cache_cs);
 149
 150   RPCRT4_CloseConnection(Connection);
 151   RPCRT4_strfree(Connection->Endpoint);
 152   RPCRT4_strfree(Connection->NetworkAddr);
 153   RPCRT4_strfree(Connection->Protseq);
 154   HeapFree(GetProcessHeap(), 0, Connection);
 155   return RPC_S_OK;
 156 }
 157
 158 RPC_STATUS RPCRT4_GetConnection(RpcConnection** Connection, BOOL server, LPSTR Protseq, LPSTR NetworkAddr, LPSTR Endpoint, LPSTR NetworkOptions, RpcBinding* Binding)
 159 {
 160   RpcConnection* NewConnection;
 161
 162   if (!server) {
 163     EnterCriticalSection(&conn_cache_cs);
 164     for (NewConnection = conn_cache; NewConnection; NewConnection = NewConnection->Next) {
 165       if (NewConnection->Used) continue;
 166       if (NewConnection->server != server) continue;
 167       if (Protseq && strcmp(NewConnection->Protseq, Protseq)) continue;
 168       if (NetworkAddr && strcmp(NewConnection->NetworkAddr, NetworkAddr)) continue;
 169       if (Endpoint && strcmp(NewConnection->Endpoint, Endpoint)) continue;
 170       /* this connection fits the bill */
 171       NewConnection->Used = Binding;
 172       break;
 173     }
 174     LeaveCriticalSection(&conn_cache_cs);
 175     if (NewConnection) {
 176       TRACE("cached connection: %p\n", NewConnection);
 177       *Connection = NewConnection;
 178       return RPC_S_OK;
 179     }
 180   }
 181   return RPCRT4_CreateConnection(Connection, server, Protseq, NetworkAddr, Endpoint, NetworkOptions, Binding);
 182 }
 183
 184 RPC_STATUS RPCRT4_ReleaseConnection(RpcConnection* Connection)
 185 {
 186   TRACE("connection: %p\n", Connection);
 187   Connection->Used = NULL;
 188   if (!Connection->server) {
 189     /* cache the open connection for reuse later */
 190     /* FIXME: we should probably clean the cache someday */
 191     return RPC_S_OK;
 192   }
 193   return RPCRT4_DestroyConnection(Connection);
 194 }
 195
 196 RPC_STATUS RPCRT4_OpenConnection(RpcConnection* Connection)
 197 {
 198   TRACE("(Connection == ^%p)\n", Connection);
 199   if (!Connection->conn) {
 200     if (Connection->server) { /* server */
 201       /* protseq=ncalrpc: supposed to use NT LPC ports,
 202        * but we'll implement it with named pipes for now */
 203       if (strcmp(Connection->Protseq, "ncalrpc") == 0) {
 204         static LPCSTR prefix = "\\\\.\\pipe\\lrpc\\";
 205         LPSTR pname;
 206         pname = HeapAlloc(GetProcessHeap(), 0, strlen(prefix) + strlen(Connection->Endpoint) + 1);
 207         strcat(strcpy(pname, prefix), Connection->Endpoint);
 208         TRACE("listening on %s\n", pname);
 209         Connection->conn = CreateNamedPipeA(pname, PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED,
 210                                          0, PIPE_UNLIMITED_INSTANCES, 0, 0, 5000, NULL);
 211         HeapFree(GetProcessHeap(), 0, pname);
 212         memset(&Connection->ovl, 0, sizeof(Connection->ovl));
 213         Connection->ovl.hEvent = CreateEventA(NULL, TRUE, FALSE, NULL);
 214         if (!ConnectNamedPipe(Connection->conn, &Connection->ovl)) {
 215           WARN("Couldn't ConnectNamedPipe (error was %ld)\n", GetLastError());
 216           if (GetLastError() == ERROR_PIPE_CONNECTED) {
 217             SetEvent(Connection->ovl.hEvent);
 218             return RPC_S_OK;
 219           }
 220           return RPC_S_SERVER_UNAVAILABLE;
 221         }
 222       }
 223       /* protseq=ncacn_np: named pipes */
 224       else if (strcmp(Connection->Protseq, "ncacn_np") == 0) {
 225         static LPCSTR prefix = "\\\\.";
 226         LPSTR pname;
 227         pname = HeapAlloc(GetProcessHeap(), 0, strlen(prefix) + strlen(Connection->Endpoint) + 1);
 228         strcat(strcpy(pname, prefix), Connection->Endpoint);
 229         TRACE("listening on %s\n", pname);
 230         Connection->conn = CreateNamedPipeA(pname, PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED,
 231                                          0, PIPE_UNLIMITED_INSTANCES, 0, 0, 5000, NULL);
 232         HeapFree(GetProcessHeap(), 0, pname);
 233         memset(&Connection->ovl, 0, sizeof(Connection->ovl));
 234         Connection->ovl.hEvent = CreateEventA(NULL, TRUE, FALSE, NULL);
 235         if (!ConnectNamedPipe(Connection->conn, &Connection->ovl)) {
 236           WARN("Couldn't ConnectNamedPipe (error was %ld)\n", GetLastError());
 237           if (GetLastError() == ERROR_PIPE_CONNECTED) {
 238             SetEvent(Connection->ovl.hEvent);
 239             return RPC_S_OK;
 240           }
 241           return RPC_S_SERVER_UNAVAILABLE;
 242         }
 243       }
 244       else {
 245         ERR("protseq %s not supported\n", Connection->Protseq);
 246         return RPC_S_PROTSEQ_NOT_SUPPORTED;
 247       }
 248     }
 249     else { /* client */
 250       /* protseq=ncalrpc: supposed to use NT LPC ports,
 251        * but we'll implement it with named pipes for now */
 252       if (strcmp(Connection->Protseq, "ncalrpc") == 0) {
 253         static LPCSTR prefix = "\\\\.\\pipe\\lrpc\\";
 254         LPSTR pname;
 255         HANDLE conn;
 256         DWORD err;
 257
 258         pname = HeapAlloc(GetProcessHeap(), 0, strlen(prefix) + strlen(Connection->Endpoint) + 1);
 259         strcat(strcpy(pname, prefix), Connection->Endpoint);
 260         TRACE("connecting to %s\n", pname);
 261         while (TRUE) {
 262           if (WaitNamedPipeA(pname, NMPWAIT_WAIT_FOREVER)) {
 263             conn = CreateFileA(pname, GENERIC_READ|GENERIC_WRITE, 0, NULL,
 264                                OPEN_EXISTING, FILE_FLAG_OVERLAPPED, 0);
 265             if (conn != INVALID_HANDLE_VALUE) break;
 266             err = GetLastError();
 267             if (err == ERROR_PIPE_BUSY) continue;
 268             TRACE("connection failed, error=%lx\n", err);
 269             HeapFree(GetProcessHeap(), 0, pname);
 270             return RPC_S_SERVER_TOO_BUSY;
 271           } else {
 272             err = GetLastError();
 273             TRACE("connection failed, error=%lx\n", err);
 274             HeapFree(GetProcessHeap(), 0, pname);
 275             return RPC_S_SERVER_UNAVAILABLE;
 276           }
 277         }
 278
 279         /* success */
 280         HeapFree(GetProcessHeap(), 0, pname);
 281         memset(&Connection->ovl, 0, sizeof(Connection->ovl));
 282         Connection->ovl.hEvent = CreateEventA(NULL, TRUE, FALSE, NULL);
 283         Connection->conn = conn;
 284       }
 285       /* protseq=ncacn_np: named pipes */
 286       else if (strcmp(Connection->Protseq, "ncacn_np") == 0) {
 287         static LPCSTR prefix = "\\\\.";
 288         LPSTR pname;
 289         HANDLE conn;
 290         DWORD err;
 291
 292         pname = HeapAlloc(GetProcessHeap(), 0, strlen(prefix) + strlen(Connection->Endpoint) + 1);
 293         strcat(strcpy(pname, prefix), Connection->Endpoint);
 294         TRACE("connecting to %s\n", pname);
 295         conn = CreateFileA(pname, GENERIC_READ|GENERIC_WRITE, 0, NULL,
 296                            OPEN_EXISTING, FILE_FLAG_OVERLAPPED, 0);
 297         if (conn == INVALID_HANDLE_VALUE) {
 298           err = GetLastError();
 299           /* we don't need to handle ERROR_PIPE_BUSY here,
 300            * the doc says that it is returned to the app */
 301           TRACE("connection failed, error=%lx\n", err);
 302           HeapFree(GetProcessHeap(), 0, pname);
 303           if (err == ERROR_PIPE_BUSY)
 304             return RPC_S_SERVER_TOO_BUSY;
 305           else
 306             return RPC_S_SERVER_UNAVAILABLE;
 307         }
 308
 309         /* success */
 310         HeapFree(GetProcessHeap(), 0, pname);
 311         memset(&Connection->ovl, 0, sizeof(Connection->ovl));
 312         Connection->ovl.hEvent = CreateEventA(NULL, TRUE, FALSE, NULL);
 313         Connection->conn = conn;
 314       } else {
 315         ERR("protseq %s not supported\n", Connection->Protseq);
 316         return RPC_S_PROTSEQ_NOT_SUPPORTED;
 317       }
 318     }
 319   }
 320   return RPC_S_OK;
 321 }
 322
 323 RPC_STATUS RPCRT4_CloseConnection(RpcConnection* Connection)
 324 {
 325   TRACE("(Connection == ^%p)\n", Connection);
 326   if (Connection->conn) {
 327     CancelIo(Connection->conn);
 328     CloseHandle(Connection->conn);
 329     Connection->conn = 0;
 330   }
 331   if (Connection->ovl.hEvent) {
 332     CloseHandle(Connection->ovl.hEvent);
 333     Connection->ovl.hEvent = 0;
 334   }
 335   return RPC_S_OK;
 336 }
 337
 338 RPC_STATUS RPCRT4_SpawnConnection(RpcConnection** Connection, RpcConnection* OldConnection)
 339 {
 340   RpcConnection* NewConnection;
 341   RPC_STATUS err = RPCRT4_CreateConnection(&NewConnection, OldConnection->server, OldConnection->Protseq,
 342                                            OldConnection->NetworkAddr, OldConnection->Endpoint, NULL, NULL);
 343   if (err == RPC_S_OK) {
 344     /* because of the way named pipes work, we'll transfer the connected pipe
 345      * to the child, then reopen the server binding to continue listening */
 346     NewConnection->conn = OldConnection->conn;
 347     NewConnection->ovl = OldConnection->ovl;
 348     OldConnection->conn = 0;
 349     memset(&OldConnection->ovl, 0, sizeof(OldConnection->ovl));
 350     *Connection = NewConnection;
 351     RPCRT4_OpenConnection(OldConnection);
 352   }
 353   return err;
 354 }
 355
 356 RPC_STATUS RPCRT4_AllocBinding(RpcBinding** Binding, BOOL server)
 357 {
 358   RpcBinding* NewBinding;
 359
 360   NewBinding = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(RpcBinding));
 361   NewBinding->refs = 1;
 362   NewBinding->server = server;
 363
 364   *Binding = NewBinding;
 365
 366   return RPC_S_OK;
 367 }
 368
 369 RPC_STATUS RPCRT4_CreateBindingA(RpcBinding** Binding, BOOL server, LPSTR Protseq)
 370 {
 371   RpcBinding* NewBinding;
 372
 373   RPCRT4_AllocBinding(&NewBinding, server);
 374   NewBinding->Protseq = RPCRT4_strdupA(Protseq);
 375
 376   TRACE("binding: %p\n", NewBinding);
 377   *Binding = NewBinding;
 378
 379   return RPC_S_OK;
 380 }
 381
 382 RPC_STATUS RPCRT4_CreateBindingW(RpcBinding** Binding, BOOL server, LPWSTR Protseq)
 383 {
 384   RpcBinding* NewBinding;
 385
 386   RPCRT4_AllocBinding(&NewBinding, server);
 387   NewBinding->Protseq = RPCRT4_strdupWtoA(Protseq);
 388
 389   TRACE("binding: %p\n", NewBinding);
 390   *Binding = NewBinding;
 391
 392   return RPC_S_OK;
 393 }
 394
 395 RPC_STATUS RPCRT4_CompleteBindingA(RpcBinding* Binding, LPSTR NetworkAddr,  LPSTR Endpoint,  LPSTR NetworkOptions)
 396 {
 397   TRACE("(RpcBinding == ^%p, NetworkAddr == \"%s\", EndPoint == \"%s\", NetworkOptions == \"%s\")\n", Binding,
 398    debugstr_a(NetworkAddr), debugstr_a(Endpoint), debugstr_a(NetworkOptions));
 399
 400   RPCRT4_strfree(Binding->NetworkAddr);
 401   Binding->NetworkAddr = RPCRT4_strdupA(NetworkAddr);
 402   RPCRT4_strfree(Binding->Endpoint);
 403   if (Endpoint) {
 404     Binding->Endpoint = RPCRT4_strdupA(Endpoint);
 405   } else {
 406     Binding->Endpoint = RPCRT4_strdupA("");
 407   }
 408   if (!Binding->Endpoint) ERR("out of memory?\n");
 409
 410   return RPC_S_OK;
 411 }
 412
 413 RPC_STATUS RPCRT4_CompleteBindingW(RpcBinding* Binding, LPWSTR NetworkAddr, LPWSTR Endpoint, LPWSTR NetworkOptions)
 414 {
 415   TRACE("(RpcBinding == ^%p, NetworkAddr == \"%s\", EndPoint == \"%s\", NetworkOptions == \"%s\")\n", Binding,
 416    debugstr_w(NetworkAddr), debugstr_w(Endpoint), debugstr_w(NetworkOptions));
 417
 418   RPCRT4_strfree(Binding->NetworkAddr);
 419   Binding->NetworkAddr = RPCRT4_strdupWtoA(NetworkAddr);
 420   RPCRT4_strfree(Binding->Endpoint);
 421   if (Endpoint) {
 422     Binding->Endpoint = RPCRT4_strdupWtoA(Endpoint);
 423   } else {
 424     Binding->Endpoint = RPCRT4_strdupA("");
 425   }
 426   if (!Binding->Endpoint) ERR("out of memory?\n");
 427
 428   return RPC_S_OK;
 429 }
 430
 431 RPC_STATUS RPCRT4_ResolveBinding(RpcBinding* Binding, LPSTR Endpoint)
 432 {
 433   TRACE("(RpcBinding == ^%p, EndPoint == \"%s\"\n", Binding, Endpoint);
 434
 435   RPCRT4_strfree(Binding->Endpoint);
 436   Binding->Endpoint = RPCRT4_strdupA(Endpoint);
 437
 438   return RPC_S_OK;
 439 }
 440
 441 RPC_STATUS RPCRT4_SetBindingObject(RpcBinding* Binding, UUID* ObjectUuid)
 442 {
 443   TRACE("(*RpcBinding == ^%p, UUID == %s)\n", Binding, debugstr_guid(ObjectUuid));
 444   if (ObjectUuid) memcpy(&Binding->ObjectUuid, ObjectUuid, sizeof(UUID));
 445   else UuidCreateNil(&Binding->ObjectUuid);
 446   return RPC_S_OK;
 447 }
 448
 449 RPC_STATUS RPCRT4_MakeBinding(RpcBinding** Binding, RpcConnection* Connection)
 450 {
 451   RpcBinding* NewBinding;
 452   TRACE("(*RpcBinding == ^%p, Connection == ^%p)\n", *Binding, Connection);
 453
 454   RPCRT4_AllocBinding(&NewBinding, Connection->server);
 455   NewBinding->Protseq = RPCRT4_strdupA(Connection->Protseq);
 456   NewBinding->NetworkAddr = RPCRT4_strdupA(Connection->NetworkAddr);
 457   NewBinding->Endpoint = RPCRT4_strdupA(Connection->Endpoint);
 458   NewBinding->FromConn = Connection;
 459
 460   TRACE("binding: %p\n", NewBinding);
 461   *Binding = NewBinding;
 462
 463   return RPC_S_OK;
 464 }
 465
 466 RPC_STATUS RPCRT4_ExportBinding(RpcBinding** Binding, RpcBinding* OldBinding)
 467 {
 468   InterlockedIncrement(&OldBinding->refs);
 469   *Binding = OldBinding;
 470   return RPC_S_OK;
 471 }
 472
 473 RPC_STATUS RPCRT4_DestroyBinding(RpcBinding* Binding)
 474 {
 475   if (InterlockedDecrement(&Binding->refs))
 476     return RPC_S_OK;
 477
 478   TRACE("binding: %p\n", Binding);
 479   /* FIXME: release connections */
 480   RPCRT4_strfree(Binding->Endpoint);
 481   RPCRT4_strfree(Binding->NetworkAddr);
 482   RPCRT4_strfree(Binding->Protseq);
 483   HeapFree(GetProcessHeap(), 0, Binding);
 484   return RPC_S_OK;
 485 }
 486
 487 RPC_STATUS RPCRT4_OpenBinding(RpcBinding* Binding, RpcConnection** Connection)
 488 {
 489   RpcConnection* NewConnection;
 490   TRACE("(Binding == ^%p)\n", Binding);
 491   if (Binding->FromConn) {
 492     *Connection = Binding->FromConn;
 493     return RPC_S_OK;
 494   }
 495
 496   RPCRT4_GetConnection(&NewConnection, Binding->server, Binding->Protseq, Binding->NetworkAddr, Binding->Endpoint, NULL, Binding);
 497   *Connection = NewConnection;
 498   return RPCRT4_OpenConnection(NewConnection);
 499 }
 500
 501 RPC_STATUS RPCRT4_CloseBinding(RpcBinding* Binding, RpcConnection* Connection)
 502 {
 503   TRACE("(Binding == ^%p)\n", Binding);
 504   if (!Connection) return RPC_S_OK;
 505   if (Binding->FromConn == Connection) return RPC_S_OK;
 506   return RPCRT4_ReleaseConnection(Connection);
 507 }
 508
 509 /* utility functions for string composing and parsing */
 510 static unsigned RPCRT4_strcopyA(LPSTR data, LPCSTR src)
 511 {
 512   unsigned len = strlen(src);
 513   memcpy(data, src, len*sizeof(CHAR));
 514   return len;
 515 }
 516
 517 static unsigned RPCRT4_strcopyW(LPWSTR data, LPCWSTR src)
 518 {
 519   unsigned len = strlenW(src);
 520   memcpy(data, src, len*sizeof(WCHAR));
 521   return len;
 522 }
 523
 524 static LPSTR RPCRT4_strconcatA(LPSTR dst, LPCSTR src)
 525 {
 526   DWORD len = strlen(dst), slen = strlen(src);
 527   LPSTR ndst = HeapReAlloc(GetProcessHeap(), 0, dst, (len+slen+2)*sizeof(CHAR));
 528   if (!ndst)
 529   {
 530     HeapFree(GetProcessHeap(), 0, dst);
 531     return NULL;
 532   }
 533   ndst[len] = ',';
 534   memcpy(ndst+len+1, src, slen+1);
 535   return ndst;
 536 }
 537
 538 static LPWSTR RPCRT4_strconcatW(LPWSTR dst, LPCWSTR src)
 539 {
 540   DWORD len = strlenW(dst), slen = strlenW(src);
 541   LPWSTR ndst = HeapReAlloc(GetProcessHeap(), 0, dst, (len+slen+2)*sizeof(WCHAR));
 542   if (!ndst)
 543   {
 544     HeapFree(GetProcessHeap(), 0, dst);
 545     return NULL;
 546   }
 547   ndst[len] = ',';
 548   memcpy(ndst+len+1, src, (slen+1)*sizeof(WCHAR));
 549   return ndst;
 550 }
 551
 552
 553 /***********************************************************************
 554  *             RpcStringBindingComposeA (RPCRT4.@)
 555  */
 556 RPC_STATUS WINAPI RpcStringBindingComposeA( LPSTR ObjUuid, LPSTR Protseq,
 557                                            LPSTR NetworkAddr, LPSTR Endpoint,
 558                                            LPSTR Options, LPSTR* StringBinding )
 559 {
 560   DWORD len = 1;
 561   LPSTR data;
 562
 563   TRACE( "(%s,%s,%s,%s,%s,%p)\n",
 564         debugstr_a( ObjUuid ), debugstr_a( Protseq ),
 565         debugstr_a( NetworkAddr ), debugstr_a( Endpoint ),
 566         debugstr_a( Options ), StringBinding );
 567
 568   if (ObjUuid && *ObjUuid) len += strlen(ObjUuid) + 1;
 569   if (Protseq && *Protseq) len += strlen(Protseq) + 1;
 570   if (NetworkAddr && *NetworkAddr) len += strlen(NetworkAddr);
 571   if (Endpoint && *Endpoint) len += strlen(Endpoint) + 2;
 572   if (Options && *Options) len += strlen(Options) + 2;
 573
 574   data = HeapAlloc(GetProcessHeap(), 0, len);
 575   *StringBinding = data;
 576
 577   if (ObjUuid && *ObjUuid) {
 578     data += RPCRT4_strcopyA(data, ObjUuid);
 579     *data++ = '@';
 580   }
 581   if (Protseq && *Protseq) {
 582     data += RPCRT4_strcopyA(data, Protseq);
 583     *data++ = ':';
 584   }
 585   if (NetworkAddr && *NetworkAddr)
 586     data += RPCRT4_strcopyA(data, NetworkAddr);
 587
 588   if ((Endpoint && *Endpoint) ||
 589       (Options && *Options)) {
 590     *data++ = '[';
 591     if (Endpoint && *Endpoint) {
 592       data += RPCRT4_strcopyA(data, Endpoint);
 593       if (Options && *Options) *data++ = ',';
 594     }
 595     if (Options && *Options) {
 596       data += RPCRT4_strcopyA(data, Options);
 597     }
 598     *data++ = ']';
 599   }
 600   *data = 0;
 601
 602   return RPC_S_OK;
 603 }
 604
 605 /***********************************************************************
 606  *             RpcStringBindingComposeW (RPCRT4.@)
 607  */
 608 RPC_STATUS WINAPI RpcStringBindingComposeW( LPWSTR ObjUuid, LPWSTR Protseq,
 609                                             LPWSTR NetworkAddr, LPWSTR Endpoint,
 610                                             LPWSTR Options, LPWSTR* StringBinding )
 611 {
 612   DWORD len = 1;
 613   LPWSTR data;
 614
 615   TRACE("(%s,%s,%s,%s,%s,%p)\n",
 616        debugstr_w( ObjUuid ), debugstr_w( Protseq ),
 617        debugstr_w( NetworkAddr ), debugstr_w( Endpoint ),
 618        debugstr_w( Options ), StringBinding);
 619
 620   if (ObjUuid && *ObjUuid) len += strlenW(ObjUuid) + 1;
 621   if (Protseq && *Protseq) len += strlenW(Protseq) + 1;
 622   if (NetworkAddr && *NetworkAddr) len += strlenW(NetworkAddr);
 623   if (Endpoint && *Endpoint) len += strlenW(Endpoint) + 2;
 624   if (Options && *Options) len += strlenW(Options) + 2;
 625
 626   data = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR));
 627   *StringBinding = data;
 628
 629   if (ObjUuid && *ObjUuid) {
 630     data += RPCRT4_strcopyW(data, ObjUuid);
 631     *data++ = '@';
 632   }
 633   if (Protseq && *Protseq) {
 634     data += RPCRT4_strcopyW(data, Protseq);
 635     *data++ = ':';
 636   }
 637   if (NetworkAddr && *NetworkAddr) {
 638     data += RPCRT4_strcopyW(data, NetworkAddr);
 639   }
 640   if ((Endpoint && *Endpoint) ||
 641       (Options && *Options)) {
 642     *data++ = '[';
 643     if (Endpoint && *Endpoint) {
 644       data += RPCRT4_strcopyW(data, Endpoint);
 645       if (Options && *Options) *data++ = ',';
 646     }
 647     if (Options && *Options) {
 648       data += RPCRT4_strcopyW(data, Options);
 649     }
 650     *data++ = ']';
 651   }
 652   *data = 0;
 653
 654   return RPC_S_OK;
 655 }
 656
 657
 658 /***********************************************************************
 659  *             RpcStringBindingParseA (RPCRT4.@)
 660  */
 661 RPC_STATUS WINAPI RpcStringBindingParseA( LPSTR StringBinding, LPSTR *ObjUuid,
 662                                           LPSTR *Protseq, LPSTR *NetworkAddr,
 663                                           LPSTR *Endpoint, LPSTR *Options)
 664 {
 665   CHAR *data, *next;
 666   static const char ep_opt[] = "endpoint=";
 667
 668   TRACE("(%s,%p,%p,%p,%p,%p)\n", debugstr_a(StringBinding),
 669        ObjUuid, Protseq, NetworkAddr, Endpoint, Options);
 670
 671   if (ObjUuid) *ObjUuid = NULL;
 672   if (Protseq) *Protseq = NULL;
 673   if (NetworkAddr) *NetworkAddr = NULL;
 674   if (Endpoint) *Endpoint = NULL;
 675   if (Options) *Options = NULL;
 676
 677   data = StringBinding;
 678
 679   next = strchr(data, '@');
 680   if (next) {
 681     if (ObjUuid) *ObjUuid = RPCRT4_strndupA(data, next - data);
 682     data = next+1;
 683   }
 684
 685   next = strchr(data, ':');
 686   if (next) {
 687     if (Protseq) *Protseq = RPCRT4_strndupA(data, next - data);
 688     data = next+1;
 689   }
 690
 691   next = strchr(data, '[');
 692   if (next) {
 693     CHAR *close, *opt;
 694
 695     if (NetworkAddr) *NetworkAddr = RPCRT4_strndupA(data, next - data);
 696     data = next+1;
 697     close = strchr(data, ']');
 698     if (!close) goto fail;
 699
 700     /* tokenize options */
 701     while (data < close) {
 702       next = strchr(data, ',');
 703       if (!next || next > close) next = close;
 704       /* FIXME: this is kind of inefficient */
 705       opt = RPCRT4_strndupA(data, next - data);
 706       data = next+1;
 707
 708       /* parse option */
 709       next = strchr(opt, '=');
 710       if (!next) {
 711         /* not an option, must be an endpoint */
 712         if (*Endpoint) goto fail;
 713         *Endpoint = opt;
 714       } else {
 715         if (strncmp(opt, ep_opt, strlen(ep_opt)) == 0) {
 716           /* endpoint option */
 717           if (*Endpoint) goto fail;
 718           *Endpoint = RPCRT4_strdupA(next+1);
 719           HeapFree(GetProcessHeap(), 0, opt);
 720         } else {
 721           /* network option */
 722           if (*Options) {
 723             /* FIXME: this is kind of inefficient */
 724             *Options = RPCRT4_strconcatA(*Options, opt);
 725             HeapFree(GetProcessHeap(), 0, opt);
 726           } else
 727             *Options = opt;
 728         }
 729       }
 730     }
 731
 732     data = close+1;
 733     if (*data) goto fail;
 734   }
 735   else if (NetworkAddr)
 736     *NetworkAddr = RPCRT4_strdupA(data);
 737
 738   return RPC_S_OK;
 739
 740 fail:
 741   if (ObjUuid) RpcStringFreeA((unsigned char**)ObjUuid);
 742   if (Protseq) RpcStringFreeA((unsigned char**)Protseq);
 743   if (NetworkAddr) RpcStringFreeA((unsigned char**)NetworkAddr);
 744   if (Endpoint) RpcStringFreeA((unsigned char**)Endpoint);
 745   if (Options) RpcStringFreeA((unsigned char**)Options);
 746   return RPC_S_INVALID_STRING_BINDING;
 747 }
 748
 749 /***********************************************************************
 750  *             RpcStringBindingParseW (RPCRT4.@)
 751  */
 752 RPC_STATUS WINAPI RpcStringBindingParseW( LPWSTR StringBinding, LPWSTR *ObjUuid,
 753                                           LPWSTR *Protseq, LPWSTR *NetworkAddr,
 754                                           LPWSTR *Endpoint, LPWSTR *Options)
 755 {
 756   WCHAR *data, *next;
 757   static const WCHAR ep_opt[] = {'e','n','d','p','o','i','n','t','=',0};
 758
 759   TRACE("(%s,%p,%p,%p,%p,%p)\n", debugstr_w(StringBinding),
 760        ObjUuid, Protseq, NetworkAddr, Endpoint, Options);
 761
 762   if (ObjUuid) *ObjUuid = NULL;
 763   if (Protseq) *Protseq = NULL;
 764   if (NetworkAddr) *NetworkAddr = NULL;
 765   if (Endpoint) *Endpoint = NULL;
 766   if (Options) *Options = NULL;
 767
 768   data = StringBinding;
 769
 770   next = strchrW(data, '@');
 771   if (next) {
 772     if (ObjUuid) *ObjUuid = RPCRT4_strndupW(data, next - data);
 773     data = next+1;
 774   }
 775
 776   next = strchrW(data, ':');
 777   if (next) {
 778     if (Protseq) *Protseq = RPCRT4_strndupW(data, next - data);
 779     data = next+1;
 780   }
 781
 782   next = strchrW(data, '[');
 783   if (next) {
 784     WCHAR *close, *opt;
 785
 786     if (NetworkAddr) *NetworkAddr = RPCRT4_strndupW(data, next - data);
 787     data = next+1;
 788     close = strchrW(data, ']');
 789     if (!close) goto fail;
 790
 791     /* tokenize options */
 792     while (data < close) {
 793       next = strchrW(data, ',');
 794       if (!next || next > close) next = close;
 795       /* FIXME: this is kind of inefficient */
 796       opt = RPCRT4_strndupW(data, next - data);
 797       data = next+1;
 798
 799       /* parse option */
 800       next = strchrW(opt, '=');
 801       if (!next) {
 802         /* not an option, must be an endpoint */
 803         if (*Endpoint) goto fail;
 804         *Endpoint = opt;
 805       } else {
 806         if (strncmpW(opt, ep_opt, strlenW(ep_opt)) == 0) {
 807           /* endpoint option */
 808           if (*Endpoint) goto fail;
 809           *Endpoint = RPCRT4_strdupW(next+1);
 810           HeapFree(GetProcessHeap(), 0, opt);
 811         } else {
 812           /* network option */
 813           if (*Options) {
 814             /* FIXME: this is kind of inefficient */
 815             *Options = RPCRT4_strconcatW(*Options, opt);
 816             HeapFree(GetProcessHeap(), 0, opt);
 817           } else
 818             *Options = opt;
 819         }
 820       }
 821     }
 822
 823     data = close+1;
 824     if (*data) goto fail;
 825   } else if (NetworkAddr)
 826     *NetworkAddr = RPCRT4_strdupW(data);
 827
 828   return RPC_S_OK;
 829
 830 fail:
 831   if (ObjUuid) RpcStringFreeW(ObjUuid);
 832   if (Protseq) RpcStringFreeW(Protseq);
 833   if (NetworkAddr) RpcStringFreeW(NetworkAddr);
 834   if (Endpoint) RpcStringFreeW(Endpoint);
 835   if (Options) RpcStringFreeW(Options);
 836   return RPC_S_INVALID_STRING_BINDING;
 837 }
 838
 839 /***********************************************************************
 840  *             RpcBindingFree (RPCRT4.@)
 841  */
 842 RPC_STATUS WINAPI RpcBindingFree( RPC_BINDING_HANDLE* Binding )
 843 {
 844   RPC_STATUS status;
 845   TRACE("(%p) = %p\n", Binding, *Binding);
 846   status = RPCRT4_DestroyBinding(*Binding);
 847   if (status == RPC_S_OK) *Binding = 0;
 848   return status;
 849 }
 850
 851 /***********************************************************************
 852  *             RpcBindingVectorFree (RPCRT4.@)
 853  */
 854 RPC_STATUS WINAPI RpcBindingVectorFree( RPC_BINDING_VECTOR** BindingVector )
 855 {
 856   RPC_STATUS status;
 857   unsigned long c;
 858
 859   TRACE("(%p)\n", BindingVector);
 860   for (c=0; c<(*BindingVector)->Count; c++) {
 861     status = RpcBindingFree(&(*BindingVector)->BindingH[c]);
 862   }
 863   HeapFree(GetProcessHeap(), 0, *BindingVector);
 864   *BindingVector = NULL;
 865   return RPC_S_OK;
 866 }
 867
 868 /***********************************************************************
 869  *             RpcBindingInqObject (RPCRT4.@)
 870  */
 871 RPC_STATUS WINAPI RpcBindingInqObject( RPC_BINDING_HANDLE Binding, UUID* ObjectUuid )
 872 {
 873   RpcBinding* bind = (RpcBinding*)Binding;
 874
 875   TRACE("(%p,%p) = %s\n", Binding, ObjectUuid, debugstr_guid(&bind->ObjectUuid));
 876   memcpy(ObjectUuid, &bind->ObjectUuid, sizeof(UUID));
 877   return RPC_S_OK;
 878 }
 879
 880 /***********************************************************************
 881  *             RpcBindingSetObject (RPCRT4.@)
 882  */
 883 RPC_STATUS WINAPI RpcBindingSetObject( RPC_BINDING_HANDLE Binding, UUID* ObjectUuid )
 884 {
 885   RpcBinding* bind = (RpcBinding*)Binding;
 886
 887   TRACE("(%p,%s)\n", Binding, debugstr_guid(ObjectUuid));
 888   if (bind->server) return RPC_S_WRONG_KIND_OF_BINDING;
 889   return RPCRT4_SetBindingObject(Binding, ObjectUuid);
 890 }
 891
 892 /***********************************************************************
 893  *             RpcBindingFromStringBindingA (RPCRT4.@)
 894  */
 895 RPC_STATUS WINAPI RpcBindingFromStringBindingA( LPSTR StringBinding, RPC_BINDING_HANDLE* Binding )
 896 {
 897   RPC_STATUS ret;
 898   RpcBinding* bind = NULL;
 899   LPSTR ObjectUuid, Protseq, NetworkAddr, Endpoint, Options;
 900   UUID Uuid;
 901
 902   TRACE("(%s,%p)\n", debugstr_a(StringBinding), Binding);
 903
 904   ret = RpcStringBindingParseA(StringBinding, &ObjectUuid, &Protseq,
 905                               &NetworkAddr, &Endpoint, &Options);
 906   if (ret != RPC_S_OK) return ret;
 907
 908   ret = UuidFromStringA(ObjectUuid, &Uuid);
 909
 910   if (ret == RPC_S_OK)
 911     ret = RPCRT4_CreateBindingA(&bind, FALSE, Protseq);
 912   if (ret == RPC_S_OK)
 913     ret = RPCRT4_SetBindingObject(bind, &Uuid);
 914   if (ret == RPC_S_OK)
 915     ret = RPCRT4_CompleteBindingA(bind, NetworkAddr, Endpoint, Options);
 916
 917   RpcStringFreeA((unsigned char**)&Options);
 918   RpcStringFreeA((unsigned char**)&Endpoint);
 919   RpcStringFreeA((unsigned char**)&NetworkAddr);
 920   RpcStringFreeA((unsigned char**)&Protseq);
 921   RpcStringFreeA((unsigned char**)&ObjectUuid);
 922
 923   if (ret == RPC_S_OK)
 924     *Binding = (RPC_BINDING_HANDLE)bind;
 925   else
 926     RPCRT4_DestroyBinding(bind);
 927
 928   return ret;
 929 }
 930
 931 /***********************************************************************
 932  *             RpcBindingFromStringBindingW (RPCRT4.@)
 933  */
 934 RPC_STATUS WINAPI RpcBindingFromStringBindingW( LPWSTR StringBinding, RPC_BINDING_HANDLE* Binding )
 935 {
 936   RPC_STATUS ret;
 937   RpcBinding* bind = NULL;
 938   LPWSTR ObjectUuid, Protseq, NetworkAddr, Endpoint, Options;
 939   UUID Uuid;
 940
 941   TRACE("(%s,%p)\n", debugstr_w(StringBinding), Binding);
 942
 943   ret = RpcStringBindingParseW(StringBinding, &ObjectUuid, &Protseq,
 944                               &NetworkAddr, &Endpoint, &Options);
 945   if (ret != RPC_S_OK) return ret;
 946
 947   ret = UuidFromStringW(ObjectUuid, &Uuid);
 948
 949   if (ret == RPC_S_OK)
 950     ret = RPCRT4_CreateBindingW(&bind, FALSE, Protseq);
 951   if (ret == RPC_S_OK)
 952     ret = RPCRT4_SetBindingObject(bind, &Uuid);
 953   if (ret == RPC_S_OK)
 954     ret = RPCRT4_CompleteBindingW(bind, NetworkAddr, Endpoint, Options);
 955
 956   RpcStringFreeW(&Options);
 957   RpcStringFreeW(&Endpoint);
 958   RpcStringFreeW(&NetworkAddr);
 959   RpcStringFreeW(&Protseq);
 960   RpcStringFreeW(&ObjectUuid);
 961
 962   if (ret == RPC_S_OK)
 963     *Binding = (RPC_BINDING_HANDLE)bind;
 964   else
 965     RPCRT4_DestroyBinding(bind);
 966
 967   return ret;
 968 }
 969
 970 /***********************************************************************
 971  *             RpcBindingToStringBindingA (RPCRT4.@)
 972  */
 973 RPC_STATUS WINAPI RpcBindingToStringBindingA( RPC_BINDING_HANDLE Binding, LPSTR* StringBinding )
 974 {
 975   RPC_STATUS ret;
 976   RpcBinding* bind = (RpcBinding*)Binding;
 977   LPSTR ObjectUuid;
 978
 979   TRACE("(%p,%p)\n", Binding, StringBinding);
 980
 981   ret = UuidToStringA(&bind->ObjectUuid, (unsigned char**)&ObjectUuid);
 982   if (ret != RPC_S_OK) return ret;
 983
 984   ret = RpcStringBindingComposeA(ObjectUuid, bind->Protseq, bind->NetworkAddr,
 985                                  bind->Endpoint, NULL, StringBinding);
 986
 987   RpcStringFreeA((unsigned char**)&ObjectUuid);
 988
 989   return ret;
 990 }
 991
 992 /***********************************************************************
 993  *             RpcBindingToStringBindingW (RPCRT4.@)
 994  */
 995 RPC_STATUS WINAPI RpcBindingToStringBindingW( RPC_BINDING_HANDLE Binding, LPWSTR* StringBinding )
 996 {
 997   RPC_STATUS ret;
 998   LPSTR str = NULL;
 999   TRACE("(%p,%p)\n", Binding, StringBinding);
1000   ret = RpcBindingToStringBindingA(Binding, &str);
1001   *StringBinding = RPCRT4_strdupAtoW(str);
1002   RpcStringFreeA((unsigned char**)&str);
1003   return ret;
1004 }
1005
1006 /***********************************************************************
1007  *             I_RpcBindingSetAsync (RPCRT4.@)
1008  * NOTES
1009  *  Exists in win9x and winNT, but with different number of arguments
1010  *  (9x version has 3 arguments, NT has 2).
1011  */
1012 RPC_STATUS WINAPI I_RpcBindingSetAsync( RPC_BINDING_HANDLE Binding, RPC_BLOCKING_FN BlockingFn)
1013 {
1014   RpcBinding* bind = (RpcBinding*)Binding;
1015
1016   TRACE( "(%p,%p): stub\n", Binding, BlockingFn );
1017
1018   bind->BlockingFn = BlockingFn;
1019
1020   return RPC_S_OK;
1021 }
1022
1023 /***********************************************************************
1024  *             RpcNetworkIsProtSeqValidA (RPCRT4.@)
1025  */
1026 RPC_STATUS RPC_ENTRY RpcNetworkIsProtSeqValidA(unsigned char *protseq) {
1027   UNICODE_STRING protseqW;
1028
1029   if (!protseq) return RPC_S_INVALID_RPC_PROTSEQ; /* ? */
1030
1031   if (RtlCreateUnicodeStringFromAsciiz(&protseqW, protseq)) {
1032     RPC_STATUS ret = RpcNetworkIsProtSeqValidW(protseqW.Buffer);
1033     RtlFreeUnicodeString(&protseqW);
1034     return ret;
1035   } else return RPC_S_OUT_OF_MEMORY;
1036 }
1037
1038 /***********************************************************************
1039  *             RpcNetworkIsProtSeqValidW (RPCRT4.@)
1040  *
1041  * Checks if the given protocol sequence is known by the RPC system.
1042  * If it is, returns RPC_S_OK, otherwise RPC_S_PROTSEQ_NOT_SUPPORTED.
1043  *
1044  * We currently support:
1045  *   ncalrpc   local-only rpc over LPC (LPC is not really used)
1046  *   ncacn_np  rpc over named pipes
1047  */
1048 RPC_STATUS RPC_ENTRY RpcNetworkIsProtSeqValidW(LPWSTR protseq) {
1049   static const WCHAR protseqsW[][15] = {
1050     {'n','c','a','l','r','p','c',0},
1051     {'n','c','a','c','n','_','n','p',0}
1052   };
1053   static const int count = sizeof(protseqsW) / sizeof(protseqsW[0]);
1054   int i;
1055
1056   if (!protseq) return RPC_S_INVALID_RPC_PROTSEQ; /* ? */
1057
1058   for (i = 0; i < count; i++) {
1059     if (!strcmpW(protseq, protseqsW[i])) return RPC_S_OK;
1060   }
1061
1062   FIXME("Unknown protseq %s - we probably need to implement it one day\n", debugstr_w(protseq));
1063   return RPC_S_PROTSEQ_NOT_SUPPORTED;
1064 }