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 #elif defined(HAVE_INET_ADDR)
  55     return ntohl( inet_addr( cp ) );
  56 #else
  57     return 0;
  58 #endif
  59 }
  60
  61 /*****************************************************************************
  62  *          getnetbyname       [WSOCK32.1101]
  63  */
  64 struct netent * WINAPI WSOCK32_getnetbyname(const char *name)
  65 {
  66 #ifdef HAVE_GETNETBYNAME
  67     return getnetbyname(name);
  68 #else
  69     return NULL;
  70 #endif
  71 }
  72
  73 static DWORD map_service(DWORD wsaflags)
  74 {
  75     DWORD flags = 0;
  76
  77     if (wsaflags & XP1_CONNECTIONLESS)      flags |= XP_CONNECTIONLESS;
  78     if (wsaflags & XP1_GUARANTEED_DELIVERY) flags |= XP_GUARANTEED_DELIVERY;
  79     if (wsaflags & XP1_GUARANTEED_ORDER)    flags |= XP_GUARANTEED_ORDER;
  80     if (wsaflags & XP1_MESSAGE_ORIENTED)    flags |= XP_MESSAGE_ORIENTED;
  81     if (wsaflags & XP1_PSEUDO_STREAM)       flags |= XP_PSEUDO_STREAM;
  82     if (wsaflags & XP1_GRACEFUL_CLOSE)      flags |= XP_GRACEFUL_CLOSE;
  83     if (wsaflags & XP1_EXPEDITED_DATA)      flags |= XP_EXPEDITED_DATA;
  84     if (wsaflags & XP1_CONNECT_DATA)        flags |= XP_CONNECT_DATA;
  85     if (wsaflags & XP1_DISCONNECT_DATA)     flags |= XP_DISCONNECT_DATA;
  86     if (wsaflags & XP1_SUPPORT_BROADCAST)   flags |= XP_SUPPORTS_BROADCAST;
  87     if (wsaflags & XP1_SUPPORT_MULTIPOINT)  flags |= XP_SUPPORTS_MULTICAST;
  88     if (wsaflags & XP1_QOS_SUPPORTED)       flags |= XP_BANDWIDTH_ALLOCATION;
  89     if (wsaflags & XP1_PARTIAL_MESSAGE)     flags |= XP_FRAGMENTATION;
  90     return flags;
  91 }
  92
  93 /*****************************************************************************
  94  *          EnumProtocolsA       [WSOCK32.1111]
  95  */
  96 INT WINAPI EnumProtocolsA(LPINT protocols, LPVOID buffer, LPDWORD buflen)
  97 {
  98     INT ret;
  99     DWORD size, string_size = WSAPROTOCOL_LEN + 1;
 100
 101     TRACE("%p, %p, %p\n", protocols, buffer, buflen);
 102
 103     if (!buflen) return SOCKET_ERROR;
 104
 105     size = 0;
 106     ret = WSAEnumProtocolsA(protocols, NULL, &size);
 107
 108     if (ret == SOCKET_ERROR && WSAGetLastError() == WSAENOBUFS)
 109     {
 110         DWORD num_protocols = size / sizeof(WSAPROTOCOL_INFOA);
 111         if (*buflen < num_protocols * (sizeof(PROTOCOL_INFOA) + string_size))
 112         {
 113             *buflen = num_protocols * (sizeof(PROTOCOL_INFOA) + string_size);
 114             return SOCKET_ERROR;
 115         }
 116         if (buffer)
 117         {
 118             WSAPROTOCOL_INFOA *wsabuf;
 119             PROTOCOL_INFOA *pi = buffer;
 120             unsigned int string_offset;
 121             INT i;
 122
 123             if (!(wsabuf = HeapAlloc(GetProcessHeap(), 0, size))) return SOCKET_ERROR;
 124
 125             ret = WSAEnumProtocolsA(protocols, wsabuf, &size);
 126             string_offset = ret * sizeof(PROTOCOL_INFOA);
 127
 128             for (i = 0; i < ret; i++)
 129             {
 130                 pi[i].dwServiceFlags = map_service(wsabuf[i].dwServiceFlags1);
 131                 pi[i].iAddressFamily = wsabuf[i].iAddressFamily;
 132                 pi[i].iMaxSockAddr   = wsabuf[i].iMaxSockAddr;
 133                 pi[i].iMinSockAddr   = wsabuf[i].iMinSockAddr;
 134                 pi[i].iSocketType    = wsabuf[i].iSocketType;
 135                 pi[i].iProtocol      = wsabuf[i].iProtocol;
 136                 pi[i].dwMessageSize  = wsabuf[i].dwMessageSize;
 137
 138                 memcpy((char *)buffer + string_offset, wsabuf[i].szProtocol, string_size);
 139                 pi[i].lpProtocol = (char *)buffer + string_offset;
 140                 string_offset += string_size;
 141             }
 142             HeapFree(GetProcessHeap(), 0, wsabuf);
 143         }
 144     }
 145     return ret;
 146 }
 147
 148 /*****************************************************************************
 149  *          EnumProtocolsW       [WSOCK32.1112]
 150  */
 151 INT WINAPI EnumProtocolsW(LPINT protocols, LPVOID buffer, LPDWORD buflen)
 152 {
 153     INT ret;
 154     DWORD size, string_size = (WSAPROTOCOL_LEN + 1) * sizeof(WCHAR);
 155
 156     TRACE("%p, %p, %p\n", protocols, buffer, buflen);
 157
 158     if (!buflen) return SOCKET_ERROR;
 159
 160     size = 0;
 161     ret = WSAEnumProtocolsW(protocols, NULL, &size);
 162
 163     if (ret == SOCKET_ERROR && WSAGetLastError() == WSAENOBUFS)
 164     {
 165         DWORD num_protocols = size / sizeof(WSAPROTOCOL_INFOW);
 166         if (*buflen < num_protocols * (sizeof(PROTOCOL_INFOW) + string_size))
 167         {
 168             *buflen = num_protocols * (sizeof(PROTOCOL_INFOW) + string_size);
 169             return SOCKET_ERROR;
 170         }
 171         if (buffer)
 172         {
 173             WSAPROTOCOL_INFOW *wsabuf;
 174             PROTOCOL_INFOW *pi = buffer;
 175             unsigned int string_offset;
 176             INT i;
 177
 178             if (!(wsabuf = HeapAlloc(GetProcessHeap(), 0, size))) return SOCKET_ERROR;
 179
 180             ret = WSAEnumProtocolsW(protocols, wsabuf, &size);
 181             string_offset = ret * sizeof(PROTOCOL_INFOW);
 182
 183             for (i = 0; i < ret; i++)
 184             {
 185                 pi[i].dwServiceFlags = map_service(wsabuf[i].dwServiceFlags1);
 186                 pi[i].iAddressFamily = wsabuf[i].iAddressFamily;
 187                 pi[i].iMaxSockAddr   = wsabuf[i].iMaxSockAddr;
 188                 pi[i].iMinSockAddr   = wsabuf[i].iMinSockAddr;
 189                 pi[i].iSocketType    = wsabuf[i].iSocketType;
 190                 pi[i].iProtocol      = wsabuf[i].iProtocol;
 191                 pi[i].dwMessageSize  = wsabuf[i].dwMessageSize;
 192
 193                 memcpy((char *)buffer + string_offset, wsabuf[i].szProtocol, string_size);
 194                 pi[i].lpProtocol = (WCHAR *)((char *)buffer + string_offset);
 195                 string_offset += string_size;
 196             }
 197             HeapFree(GetProcessHeap(), 0, wsabuf);
 198         }
 199     }
 200     return ret;
 201 }