dlls/wsock32/protocol.c

   1 /*
   2  * WSOCK32 specific functions
   3  *
   4  * Copyright (C) 2001 Stefan Leichter
   5  * Copyright (C) 2008 Hans Leidekker
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #define USE_WS_PREFIX
  23
  24 #include "config.h"
  25
  26 #include <stdarg.h>
  27 #include <stdio.h>
  28 #include <string.h>
  29
  30 #ifdef HAVE_ARPA_INET_H
  31 #include <arpa/inet.h>
  32 #endif
  33 #ifdef HAVE_NETDB_H
  34 #include <netdb.h>
  35 #endif
  36
  37 #include "windef.h"
  38 #include "winbase.h"
  39 #include "winsock2.h"
  40 #include "nspapi.h"
  41
  42 #include "wine/debug.h"
  43 #include "wine/unicode.h"
  44
  45 WINE_DEFAULT_DEBUG_CHANNEL(winsock);
  46
  47 /*****************************************************************************
  48  *          inet_network       [WSOCK32.1100]
  49  */
  50 UINT WINAPI WSOCK32_inet_network(const char *cp)
  51 {
  52 #ifdef HAVE_INET_NETWORK
  53     return inet_network(cp);
  54 #else
  55     return 0;
  56 #endif
  57 }
  58
  59 /*****************************************************************************
  60  *          getnetbyname       [WSOCK32.1101]
  61  */
  62 struct netent * WINAPI WSOCK32_getnetbyname(const char *name)
  63 {
  64 #ifdef HAVE_GETNETBYNAME
  65     return getnetbyname(name);
  66 #else
  67     return NULL;
  68 #endif
  69 }
  70
  71 static DWORD map_service(DWORD wsaflags)
  72 {
  73     DWORD flags = 0;
  74
  75     if (wsaflags & XP1_CONNECTIONLESS)      flags |= XP_CONNECTIONLESS;
  76     if (wsaflags & XP1_GUARANTEED_DELIVERY) flags |= XP_GUARANTEED_DELIVERY;
  77     if (wsaflags & XP1_GUARANTEED_ORDER)    flags |= XP_GUARANTEED_ORDER;
  78     if (wsaflags & XP1_MESSAGE_ORIENTED)    flags |= XP_MESSAGE_ORIENTED;
  79     if (wsaflags & XP1_PSEUDO_STREAM)       flags |= XP_PSEUDO_STREAM;
  80     if (wsaflags & XP1_GRACEFUL_CLOSE)      flags |= XP_GRACEFUL_CLOSE;
  81     if (wsaflags & XP1_EXPEDITED_DATA)      flags |= XP_EXPEDITED_DATA;
  82     if (wsaflags & XP1_CONNECT_DATA)        flags |= XP_CONNECT_DATA;
  83     if (wsaflags & XP1_DISCONNECT_DATA)     flags |= XP_DISCONNECT_DATA;
  84     if (wsaflags & XP1_SUPPORT_BROADCAST)   flags |= XP_SUPPORTS_BROADCAST;
  85     if (wsaflags & XP1_SUPPORT_MULTIPOINT)  flags |= XP_SUPPORTS_MULTICAST;
  86     if (wsaflags & XP1_QOS_SUPPORTED)       flags |= XP_BANDWIDTH_ALLOCATION;
  87     if (wsaflags & XP1_PARTIAL_MESSAGE)     flags |= XP_FRAGMENTATION;
  88     return flags;
  89 }
  90
  91 /*****************************************************************************
  92  *          EnumProtocolsA       [WSOCK32.1111]
  93  */
  94 INT WINAPI EnumProtocolsA(LPINT protocols, LPVOID buffer, LPDWORD buflen)
  95 {
  96     INT ret;
  97     DWORD size, string_size = WSAPROTOCOL_LEN + 1;
  98
  99     TRACE("%p, %p, %p\n", protocols, buffer, buflen);
 100
 101     if (!buflen) return SOCKET_ERROR;
 102
 103     size = 0;
 104     ret = WSAEnumProtocolsA(protocols, NULL, &size);
 105
 106     if (ret == SOCKET_ERROR && WSAGetLastError() == WSAENOBUFS)
 107     {
 108         DWORD num_protocols = size / sizeof(WSAPROTOCOL_INFOA);
 109         if (*buflen < num_protocols * (sizeof(PROTOCOL_INFOA) + string_size))
 110         {
 111             *buflen = num_protocols * (sizeof(PROTOCOL_INFOA) + string_size);
 112             return SOCKET_ERROR;
 113         }
 114         if (buffer)
 115         {
 116             WSAPROTOCOL_INFOA *wsabuf;
 117             PROTOCOL_INFOA *pi = buffer;
 118             unsigned int i, string_offset;
 119
 120             if (!(wsabuf = HeapAlloc(GetProcessHeap(), 0, size))) return SOCKET_ERROR;
 121
 122             ret = WSAEnumProtocolsA(protocols, wsabuf, &size);
 123             string_offset = ret * sizeof(PROTOCOL_INFOA);
 124
 125             for (i = 0; i < ret; i++)
 126             {
 127                 pi[i].dwServiceFlags = map_service(wsabuf[i].dwServiceFlags1);
 128                 pi[i].iAddressFamily = wsabuf[i].iAddressFamily;
 129                 pi[i].iMaxSockAddr   = wsabuf[i].iMaxSockAddr;
 130                 pi[i].iMinSockAddr   = wsabuf[i].iMinSockAddr;
 131                 pi[i].iSocketType    = wsabuf[i].iSocketType;
 132                 pi[i].iProtocol      = wsabuf[i].iProtocol;
 133                 pi[i].dwMessageSize  = wsabuf[i].dwMessageSize;
 134
 135                 memcpy((char *)buffer + string_offset, wsabuf[i].szProtocol, string_size);
 136                 pi[i].lpProtocol = (char *)buffer + string_offset;
 137                 string_offset += string_size;
 138             }
 139             HeapFree(GetProcessHeap(), 0, wsabuf);
 140         }
 141     }
 142     return ret;
 143 }
 144
 145 /*****************************************************************************
 146  *          EnumProtocolsW       [WSOCK32.1112]
 147  */
 148 INT WINAPI EnumProtocolsW(LPINT protocols, LPVOID buffer, LPDWORD buflen)
 149 {
 150     INT ret;
 151     DWORD size, string_size = (WSAPROTOCOL_LEN + 1) * sizeof(WCHAR);
 152
 153     TRACE("%p, %p, %p\n", protocols, buffer, buflen);
 154
 155     if (!buflen) return SOCKET_ERROR;
 156
 157     size = 0;
 158     ret = WSAEnumProtocolsW(protocols, NULL, &size);
 159
 160     if (ret == SOCKET_ERROR && WSAGetLastError() == WSAENOBUFS)
 161     {
 162         DWORD num_protocols = size / sizeof(WSAPROTOCOL_INFOW);
 163         if (*buflen < num_protocols * (sizeof(PROTOCOL_INFOW) + string_size))
 164         {
 165             *buflen = num_protocols * (sizeof(PROTOCOL_INFOW) + string_size);
 166             return SOCKET_ERROR;
 167         }
 168         if (buffer)
 169         {
 170             WSAPROTOCOL_INFOW *wsabuf;
 171             PROTOCOL_INFOW *pi = buffer;
 172             unsigned int i, string_offset;
 173
 174             if (!(wsabuf = HeapAlloc(GetProcessHeap(), 0, size))) return SOCKET_ERROR;
 175
 176             ret = WSAEnumProtocolsW(protocols, wsabuf, &size);
 177             string_offset = ret * sizeof(PROTOCOL_INFOW);
 178
 179             for (i = 0; i < ret; i++)
 180             {
 181                 pi[i].dwServiceFlags = map_service(wsabuf[i].dwServiceFlags1);
 182                 pi[i].iAddressFamily = wsabuf[i].iAddressFamily;
 183                 pi[i].iMaxSockAddr   = wsabuf[i].iMaxSockAddr;
 184                 pi[i].iMinSockAddr   = wsabuf[i].iMinSockAddr;
 185                 pi[i].iSocketType    = wsabuf[i].iSocketType;
 186                 pi[i].iProtocol      = wsabuf[i].iProtocol;
 187                 pi[i].dwMessageSize  = wsabuf[i].dwMessageSize;
 188
 189                 memcpy((char *)buffer + string_offset, wsabuf[i].szProtocol, string_size);
 190                 pi[i].lpProtocol = (WCHAR *)(char *)buffer + string_offset;
 191                 string_offset += string_size;
 192             }
 193             HeapFree(GetProcessHeap(), 0, wsabuf);
 194         }
 195     }
 196     return ret;
 197 }