/* * based on Windows Sockets 1.1 specs * * Copyright (C) 1993,1994,1996,1997 John Brezak, Erik Bos, Alex Korobka. * Copyright (C) 2001 Stefan Leichter * Copyright (C) 2004 Hans Leidekker * Copyright (C) 2005 Marcus Meissner * Copyright (C) 2006-2008 Kai Blin * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA * * NOTE: If you make any changes to fix a particular app, make sure * they don't break something else like Netscape or telnet and ftp * clients and servers (www.winsite.com got a lot of those). */ #include "config.h" #include "wine/port.h" #include #include #include #include #include #ifdef HAVE_SYS_IPC_H # include #endif #ifdef HAVE_SYS_IOCTL_H # include #endif #ifdef HAVE_SYS_FILIO_H # include #endif #ifdef HAVE_SYS_SOCKIO_H # include #endif #if defined(__EMX__) # include #endif #ifdef HAVE_SYS_PARAM_H # include #endif #ifdef HAVE_SYS_MSG_H # include #endif #ifdef HAVE_SYS_WAIT_H # include #endif #ifdef HAVE_SYS_UIO_H # include #endif #ifdef HAVE_SYS_SOCKET_H #include #endif #ifdef HAVE_NETINET_IN_H # include #endif #ifdef HAVE_NETINET_TCP_H # include #endif #ifdef HAVE_ARPA_INET_H # include #endif #include #include #include #ifdef HAVE_NETDB_H #include #endif #ifdef HAVE_UNISTD_H # include #endif #include #ifdef HAVE_ARPA_NAMESER_H # include #endif #ifdef HAVE_RESOLV_H # include #endif #ifdef HAVE_NET_IF_H # include #endif #ifdef HAVE_LINUX_FILTER_H # include #endif #ifdef HAVE_NETIPX_IPX_H # include #elif defined(HAVE_LINUX_IPX_H) # ifdef HAVE_ASM_TYPES_H # include # endif # ifdef HAVE_LINUX_TYPES_H # include # endif # include #endif #if defined(SOL_IPX) || defined(SO_DEFAULT_HEADERS) # define HAS_IPX #endif #ifdef HAVE_LINUX_IRDA_H # ifdef HAVE_LINUX_TYPES_H # include # endif # include # define HAS_IRDA #endif #ifdef HAVE_POLL_H #include #endif #ifdef HAVE_SYS_POLL_H # include #endif #ifdef HAVE_SYS_TIME_H # include #endif #define NONAMELESSUNION #define NONAMELESSSTRUCT #include "ntstatus.h" #define WIN32_NO_STATUS #include "windef.h" #include "winbase.h" #include "wingdi.h" #include "winuser.h" #include "winerror.h" #include "winnls.h" #include "winsock2.h" #include "mswsock.h" #include "ws2tcpip.h" #include "ws2spi.h" #include "wsipx.h" #include "wsnwlink.h" #include "wshisotp.h" #include "mstcpip.h" #include "af_irda.h" #include "winnt.h" #define USE_WC_PREFIX /* For CMSG_DATA */ #include "iphlpapi.h" #include "ip2string.h" #include "wine/afd.h" #include "wine/server.h" #include "wine/debug.h" #include "wine/exception.h" #include "wine/unicode.h" #include "wine/heap.h" #if defined(linux) && !defined(IP_UNICAST_IF) #define IP_UNICAST_IF 50 #endif #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__) || defined(__DragonFly__) # define sipx_network sipx_addr.x_net # define sipx_node sipx_addr.x_host.c_host #endif /* __FreeBSD__ */ #ifndef INADDR_NONE #define INADDR_NONE ~0UL #endif #if !defined(TCP_KEEPIDLE) && defined(TCP_KEEPALIVE) /* TCP_KEEPALIVE is the Mac OS name for TCP_KEEPIDLE */ #define TCP_KEEPIDLE TCP_KEEPALIVE #endif #define FILE_USE_FILE_POINTER_POSITION ((LONGLONG)-2) WINE_DEFAULT_DEBUG_CHANNEL(winsock); WINE_DECLARE_DEBUG_CHANNEL(winediag); static const WSAPROTOCOL_INFOW supported_protocols[] = { { .dwServiceFlags1 = XP1_IFS_HANDLES | XP1_EXPEDITED_DATA | XP1_GRACEFUL_CLOSE | XP1_GUARANTEED_ORDER | XP1_GUARANTEED_DELIVERY, .dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO, .ProviderId = {0xe70f1aa0, 0xab8b, 0x11cf, {0x8c, 0xa3, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92}}, .dwCatalogEntryId = 1001, .ProtocolChain.ChainLen = 1, .iVersion = 2, .iAddressFamily = WS_AF_INET, .iMaxSockAddr = sizeof(struct WS_sockaddr_in), .iMinSockAddr = sizeof(struct WS_sockaddr_in), .iSocketType = WS_SOCK_STREAM, .iProtocol = WS_IPPROTO_TCP, .szProtocol = {'T','C','P','/','I','P',0}, }, { .dwServiceFlags1 = XP1_IFS_HANDLES | XP1_SUPPORT_BROADCAST | XP1_SUPPORT_MULTIPOINT | XP1_MESSAGE_ORIENTED | XP1_CONNECTIONLESS, .dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO, .ProviderId = {0xe70f1aa0, 0xab8b, 0x11cf, {0x8c, 0xa3, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92}}, .dwCatalogEntryId = 1002, .ProtocolChain.ChainLen = 1, .iVersion = 2, .iAddressFamily = WS_AF_INET, .iMaxSockAddr = sizeof(struct WS_sockaddr_in), .iMinSockAddr = sizeof(struct WS_sockaddr_in), .iSocketType = WS_SOCK_DGRAM, .iProtocol = WS_IPPROTO_UDP, .dwMessageSize = 0xffbb, .szProtocol = {'U','D','P','/','I','P',0}, }, { .dwServiceFlags1 = XP1_IFS_HANDLES | XP1_EXPEDITED_DATA | XP1_GRACEFUL_CLOSE | XP1_GUARANTEED_ORDER | XP1_GUARANTEED_DELIVERY, .dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO, .ProviderId = {0xf9eab0c0, 0x26d4, 0x11d0, {0xbb, 0xbf, 0x00, 0xaa, 0x00, 0x6c, 0x34, 0xe4}}, .dwCatalogEntryId = 1004, .ProtocolChain.ChainLen = 1, .iVersion = 2, .iAddressFamily = WS_AF_INET6, .iMaxSockAddr = sizeof(struct WS_sockaddr_in6), .iMinSockAddr = sizeof(struct WS_sockaddr_in6), .iSocketType = WS_SOCK_STREAM, .iProtocol = WS_IPPROTO_TCP, .szProtocol = {'T','C','P','/','I','P','v','6',0}, }, { .dwServiceFlags1 = XP1_IFS_HANDLES | XP1_SUPPORT_BROADCAST | XP1_SUPPORT_MULTIPOINT | XP1_MESSAGE_ORIENTED | XP1_CONNECTIONLESS, .dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO, .ProviderId = {0xf9eab0c0, 0x26d4, 0x11d0, {0xbb, 0xbf, 0x00, 0xaa, 0x00, 0x6c, 0x34, 0xe4}}, .dwCatalogEntryId = 1005, .ProtocolChain.ChainLen = 1, .iVersion = 2, .iAddressFamily = WS_AF_INET6, .iMaxSockAddr = sizeof(struct WS_sockaddr_in6), .iMinSockAddr = sizeof(struct WS_sockaddr_in6), .iSocketType = WS_SOCK_DGRAM, .iProtocol = WS_IPPROTO_UDP, .dwMessageSize = 0xffbb, .szProtocol = {'U','D','P','/','I','P','v','6',0}, }, { .dwServiceFlags1 = XP1_PARTIAL_MESSAGE | XP1_SUPPORT_BROADCAST | XP1_SUPPORT_MULTIPOINT | XP1_MESSAGE_ORIENTED | XP1_CONNECTIONLESS, .dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO, .ProviderId = {0x11058240, 0xbe47, 0x11cf, {0x95, 0xc8, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92}}, .dwCatalogEntryId = 1030, .ProtocolChain.ChainLen = 1, .iVersion = 2, .iAddressFamily = WS_AF_IPX, .iMaxSockAddr = sizeof(struct WS_sockaddr), .iMinSockAddr = sizeof(struct WS_sockaddr_ipx), .iSocketType = WS_SOCK_DGRAM, .iProtocol = WS_NSPROTO_IPX, .iProtocolMaxOffset = 255, .dwMessageSize = 0x240, .szProtocol = {'I','P','X',0}, }, { .dwServiceFlags1 = XP1_IFS_HANDLES | XP1_PSEUDO_STREAM | XP1_MESSAGE_ORIENTED | XP1_GUARANTEED_ORDER | XP1_GUARANTEED_DELIVERY, .dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO, .ProviderId = {0x11058241, 0xbe47, 0x11cf, {0x95, 0xc8, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92}}, .dwCatalogEntryId = 1031, .ProtocolChain.ChainLen = 1, .iVersion = 2, .iAddressFamily = WS_AF_IPX, .iMaxSockAddr = sizeof(struct WS_sockaddr), .iMinSockAddr = sizeof(struct WS_sockaddr_ipx), .iSocketType = WS_SOCK_SEQPACKET, .iProtocol = WS_NSPROTO_SPX, .dwMessageSize = UINT_MAX, .szProtocol = {'S','P','X',0}, }, { .dwServiceFlags1 = XP1_IFS_HANDLES | XP1_GRACEFUL_CLOSE | XP1_PSEUDO_STREAM | XP1_MESSAGE_ORIENTED | XP1_GUARANTEED_ORDER | XP1_GUARANTEED_DELIVERY, .dwProviderFlags = PFL_MATCHES_PROTOCOL_ZERO, .ProviderId = {0x11058241, 0xbe47, 0x11cf, {0x95, 0xc8, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92}}, .dwCatalogEntryId = 1033, .ProtocolChain.ChainLen = 1, .iVersion = 2, .iAddressFamily = WS_AF_IPX, .iMaxSockAddr = sizeof(struct WS_sockaddr), .iMinSockAddr = sizeof(struct WS_sockaddr_ipx), .iSocketType = WS_SOCK_SEQPACKET, .iProtocol = WS_NSPROTO_SPXII, .dwMessageSize = UINT_MAX, .szProtocol = {'S','P','X',' ','I','I',0}, }, }; #define IS_IPX_PROTO(X) ((X) >= WS_NSPROTO_IPX && (X) <= WS_NSPROTO_IPX + 255) #if defined(IP_UNICAST_IF) && defined(SO_ATTACH_FILTER) # define LINUX_BOUND_IF struct interface_filter { struct sock_filter iface_memaddr; struct sock_filter iface_rule; struct sock_filter ip_memaddr; struct sock_filter ip_rule; struct sock_filter return_keep; struct sock_filter return_dump; }; # define FILTER_JUMP_DUMP(here) (u_char)(offsetof(struct interface_filter, return_dump) \ -offsetof(struct interface_filter, here)-sizeof(struct sock_filter)) \ /sizeof(struct sock_filter) # define FILTER_JUMP_KEEP(here) (u_char)(offsetof(struct interface_filter, return_keep) \ -offsetof(struct interface_filter, here)-sizeof(struct sock_filter)) \ /sizeof(struct sock_filter) # define FILTER_JUMP_NEXT() (u_char)(0) # define SKF_NET_DESTIP 16 /* offset in the network header to the destination IP */ static struct interface_filter generic_interface_filter = { /* This filter rule allows incoming packets on the specified interface, which works for all * remotely generated packets and for locally generated broadcast packets. */ BPF_STMT(BPF_LD+BPF_W+BPF_ABS, SKF_AD_OFF+SKF_AD_IFINDEX), BPF_JUMP(BPF_JMP+BPF_JEQ+BPF_K, 0xdeadbeef, FILTER_JUMP_KEEP(iface_rule), FILTER_JUMP_NEXT()), /* This rule allows locally generated packets targeted at the specific IP address of the chosen * adapter (local packets not destined for the broadcast address do not have IFINDEX set) */ BPF_STMT(BPF_LD+BPF_W+BPF_ABS, SKF_NET_OFF+SKF_NET_DESTIP), BPF_JUMP(BPF_JMP+BPF_JEQ+BPF_K, 0xdeadbeef, FILTER_JUMP_KEEP(ip_rule), FILTER_JUMP_DUMP(ip_rule)), BPF_STMT(BPF_RET+BPF_K, (u_int)-1), /* keep packet */ BPF_STMT(BPF_RET+BPF_K, 0) /* dump packet */ }; #endif /* LINUX_BOUND_IF */ extern ssize_t CDECL __wine_locked_recvmsg( int fd, struct msghdr *hdr, int flags ); /* * The actual definition of WSASendTo, wrapped in a different function name * so that internal calls from ws2_32 itself will not trigger programs like * Garena, which hooks WSASendTo/WSARecvFrom calls. */ static int WS2_sendto( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount, LPDWORD lpNumberOfBytesSent, DWORD dwFlags, const struct WS_sockaddr *to, int tolen, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine ); /* * Internal fundamental receive function, essentially WSARecvFrom with an * additional parameter to support message control headers. */ static int WS2_recv_base( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount, LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags, struct WS_sockaddr *lpFrom, LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine, LPWSABUF lpControlBuffer ); #define DECLARE_CRITICAL_SECTION(cs) \ static CRITICAL_SECTION cs; \ static CRITICAL_SECTION_DEBUG cs##_debug = \ { 0, 0, &cs, { &cs##_debug.ProcessLocksList, &cs##_debug.ProcessLocksList }, \ 0, 0, { (DWORD_PTR)(__FILE__ ": " # cs) }}; \ static CRITICAL_SECTION cs = { &cs##_debug, -1, 0, 0, 0, 0 } DECLARE_CRITICAL_SECTION(csWSgetXXXbyYYY); DECLARE_CRITICAL_SECTION(cs_if_addr_cache); DECLARE_CRITICAL_SECTION(cs_socket_list); static in_addr_t *if_addr_cache; static unsigned int if_addr_cache_size; static SOCKET *socket_list; static unsigned int socket_list_size; union generic_unix_sockaddr { struct sockaddr addr; char data[128]; /* should be big enough for all families */ }; static inline const char *debugstr_sockaddr( const struct WS_sockaddr *a ) { if (!a) return "(nil)"; switch (a->sa_family) { case WS_AF_INET: { char buf[16]; const char *p; struct WS_sockaddr_in *sin = (struct WS_sockaddr_in *)a; p = WS_inet_ntop( WS_AF_INET, &sin->sin_addr, buf, sizeof(buf) ); if (!p) p = "(unknown IPv4 address)"; return wine_dbg_sprintf("{ family AF_INET, address %s, port %d }", p, ntohs(sin->sin_port)); } case WS_AF_INET6: { char buf[46]; const char *p; struct WS_sockaddr_in6 *sin = (struct WS_sockaddr_in6 *)a; p = WS_inet_ntop( WS_AF_INET6, &sin->sin6_addr, buf, sizeof(buf) ); if (!p) p = "(unknown IPv6 address)"; return wine_dbg_sprintf("{ family AF_INET6, address %s, port %d }", p, ntohs(sin->sin6_port)); } case WS_AF_IPX: { int i; char netnum[16], nodenum[16]; struct WS_sockaddr_ipx *sin = (struct WS_sockaddr_ipx *)a; for (i = 0;i < 4; i++) sprintf(netnum + i * 2, "%02X", (unsigned char) sin->sa_netnum[i]); for (i = 0;i < 6; i++) sprintf(nodenum + i * 2, "%02X", (unsigned char) sin->sa_nodenum[i]); return wine_dbg_sprintf("{ family AF_IPX, address %s.%s, ipx socket %d }", netnum, nodenum, sin->sa_socket); } case WS_AF_IRDA: { DWORD addr; memcpy( &addr, ((const SOCKADDR_IRDA *)a)->irdaDeviceID, sizeof(addr) ); addr = ntohl( addr ); return wine_dbg_sprintf("{ family AF_IRDA, addr %08x, name %s }", addr, ((const SOCKADDR_IRDA *)a)->irdaServiceName); } default: return wine_dbg_sprintf("{ family %d }", a->sa_family); } } static inline const char *debugstr_sockopt(int level, int optname) { const char *stropt = NULL, *strlevel = NULL; #define DEBUG_SOCKLEVEL(x) case (x): strlevel = #x #define DEBUG_SOCKOPT(x) case (x): stropt = #x; break switch(level) { DEBUG_SOCKLEVEL(WS_SOL_SOCKET); switch(optname) { DEBUG_SOCKOPT(WS_SO_ACCEPTCONN); DEBUG_SOCKOPT(WS_SO_BROADCAST); DEBUG_SOCKOPT(WS_SO_BSP_STATE); DEBUG_SOCKOPT(WS_SO_CONDITIONAL_ACCEPT); DEBUG_SOCKOPT(WS_SO_CONNECT_TIME); DEBUG_SOCKOPT(WS_SO_DEBUG); DEBUG_SOCKOPT(WS_SO_DONTLINGER); DEBUG_SOCKOPT(WS_SO_DONTROUTE); DEBUG_SOCKOPT(WS_SO_ERROR); DEBUG_SOCKOPT(WS_SO_EXCLUSIVEADDRUSE); DEBUG_SOCKOPT(WS_SO_GROUP_ID); DEBUG_SOCKOPT(WS_SO_GROUP_PRIORITY); DEBUG_SOCKOPT(WS_SO_KEEPALIVE); DEBUG_SOCKOPT(WS_SO_LINGER); DEBUG_SOCKOPT(WS_SO_MAX_MSG_SIZE); DEBUG_SOCKOPT(WS_SO_OOBINLINE); DEBUG_SOCKOPT(WS_SO_OPENTYPE); DEBUG_SOCKOPT(WS_SO_PROTOCOL_INFOA); DEBUG_SOCKOPT(WS_SO_PROTOCOL_INFOW); DEBUG_SOCKOPT(WS_SO_RCVBUF); DEBUG_SOCKOPT(WS_SO_RCVTIMEO); DEBUG_SOCKOPT(WS_SO_REUSEADDR); DEBUG_SOCKOPT(WS_SO_SNDBUF); DEBUG_SOCKOPT(WS_SO_SNDTIMEO); DEBUG_SOCKOPT(WS_SO_TYPE); DEBUG_SOCKOPT(WS_SO_UPDATE_CONNECT_CONTEXT); } break; DEBUG_SOCKLEVEL(WS_NSPROTO_IPX); switch(optname) { DEBUG_SOCKOPT(WS_IPX_PTYPE); DEBUG_SOCKOPT(WS_IPX_FILTERPTYPE); DEBUG_SOCKOPT(WS_IPX_DSTYPE); DEBUG_SOCKOPT(WS_IPX_RECVHDR); DEBUG_SOCKOPT(WS_IPX_MAXSIZE); DEBUG_SOCKOPT(WS_IPX_ADDRESS); DEBUG_SOCKOPT(WS_IPX_MAX_ADAPTER_NUM); } break; DEBUG_SOCKLEVEL(WS_SOL_IRLMP); switch(optname) { DEBUG_SOCKOPT(WS_IRLMP_ENUMDEVICES); } break; DEBUG_SOCKLEVEL(WS_IPPROTO_TCP); switch(optname) { DEBUG_SOCKOPT(WS_TCP_BSDURGENT); DEBUG_SOCKOPT(WS_TCP_EXPEDITED_1122); DEBUG_SOCKOPT(WS_TCP_NODELAY); } break; DEBUG_SOCKLEVEL(WS_IPPROTO_IP); switch(optname) { DEBUG_SOCKOPT(WS_IP_ADD_MEMBERSHIP); DEBUG_SOCKOPT(WS_IP_DONTFRAGMENT); DEBUG_SOCKOPT(WS_IP_DROP_MEMBERSHIP); DEBUG_SOCKOPT(WS_IP_HDRINCL); DEBUG_SOCKOPT(WS_IP_MULTICAST_IF); DEBUG_SOCKOPT(WS_IP_MULTICAST_LOOP); DEBUG_SOCKOPT(WS_IP_MULTICAST_TTL); DEBUG_SOCKOPT(WS_IP_OPTIONS); DEBUG_SOCKOPT(WS_IP_PKTINFO); DEBUG_SOCKOPT(WS_IP_RECEIVE_BROADCAST); DEBUG_SOCKOPT(WS_IP_TOS); DEBUG_SOCKOPT(WS_IP_TTL); DEBUG_SOCKOPT(WS_IP_UNICAST_IF); } break; DEBUG_SOCKLEVEL(WS_IPPROTO_IPV6); switch(optname) { DEBUG_SOCKOPT(WS_IPV6_ADD_MEMBERSHIP); DEBUG_SOCKOPT(WS_IPV6_DROP_MEMBERSHIP); DEBUG_SOCKOPT(WS_IPV6_MULTICAST_IF); DEBUG_SOCKOPT(WS_IPV6_MULTICAST_HOPS); DEBUG_SOCKOPT(WS_IPV6_MULTICAST_LOOP); DEBUG_SOCKOPT(WS_IPV6_UNICAST_HOPS); DEBUG_SOCKOPT(WS_IPV6_V6ONLY); DEBUG_SOCKOPT(WS_IPV6_UNICAST_IF); DEBUG_SOCKOPT(WS_IPV6_DONTFRAG); } break; } #undef DEBUG_SOCKLEVEL #undef DEBUG_SOCKOPT if (!strlevel) strlevel = wine_dbg_sprintf("WS_0x%x", level); if (!stropt) stropt = wine_dbg_sprintf("WS_0x%x", optname); return wine_dbg_sprintf("level %s, name %s", strlevel + 3, stropt + 3); } static inline const char *debugstr_optval(const char *optval, int optlenval) { if (optval && !IS_INTRESOURCE(optval) && optlenval >= 1 && optlenval <= sizeof(DWORD)) { DWORD value = 0; memcpy(&value, optval, optlenval); return wine_dbg_sprintf("%p (%u)", optval, value); } return wine_dbg_sprintf("%p", optval); } /* HANDLE<->SOCKET conversion (SOCKET is UINT_PTR). */ #define SOCKET2HANDLE(s) ((HANDLE)(s)) #define HANDLE2SOCKET(h) ((SOCKET)(h)) static BOOL socket_list_add(SOCKET socket) { unsigned int i, new_size; SOCKET *new_array; EnterCriticalSection(&cs_socket_list); for (i = 0; i < socket_list_size; ++i) { if (!socket_list[i]) { socket_list[i] = socket; LeaveCriticalSection(&cs_socket_list); return TRUE; } } new_size = max(socket_list_size * 2, 8); if (!(new_array = heap_realloc(socket_list, new_size * sizeof(*socket_list)))) { LeaveCriticalSection(&cs_socket_list); return FALSE; } socket_list = new_array; memset(socket_list + socket_list_size, 0, (new_size - socket_list_size) * sizeof(*socket_list)); socket_list[socket_list_size] = socket; socket_list_size = new_size; LeaveCriticalSection(&cs_socket_list); return TRUE; } static void socket_list_remove(SOCKET socket) { unsigned int i; EnterCriticalSection(&cs_socket_list); for (i = 0; i < socket_list_size; ++i) { if (socket_list[i] == socket) { socket_list[i] = 0; break; } } LeaveCriticalSection(&cs_socket_list); } /**************************************************************** * Async IO declarations ****************************************************************/ typedef NTSTATUS async_callback_t( void *user, IO_STATUS_BLOCK *io, NTSTATUS status ); struct ws2_async_io { async_callback_t *callback; /* must be the first field */ struct ws2_async_io *next; }; struct ws2_async_shutdown { struct ws2_async_io io; HANDLE hSocket; IO_STATUS_BLOCK iosb; int type; }; struct ws2_async { struct ws2_async_io io; HANDLE hSocket; LPWSAOVERLAPPED user_overlapped; LPWSAOVERLAPPED_COMPLETION_ROUTINE completion_func; IO_STATUS_BLOCK local_iosb; struct WS_sockaddr *addr; union { int val; /* for send operations */ int *ptr; /* for recv operations */ } addrlen; DWORD flags; DWORD *lpFlags; WSABUF *control; unsigned int n_iovecs; unsigned int first_iovec; struct iovec iovec[1]; }; struct ws2_accept_async { struct ws2_async_io io; HANDLE listen_socket; HANDLE accept_socket; LPOVERLAPPED user_overlapped; ULONG_PTR cvalue; PVOID buf; /* buffer to write data to */ int data_len; int local_len; int remote_len; struct ws2_async *read; }; struct ws2_transmitfile_async { struct ws2_async_io io; char *buffer; HANDLE file; DWORD file_read; DWORD file_bytes; DWORD bytes_per_send; TRANSMIT_FILE_BUFFERS buffers; DWORD flags; LARGE_INTEGER offset; struct ws2_async write; }; static struct ws2_async_io *async_io_freelist; static void release_async_io( struct ws2_async_io *io ) { for (;;) { struct ws2_async_io *next = async_io_freelist; io->next = next; if (InterlockedCompareExchangePointer( (void **)&async_io_freelist, io, next ) == next) return; } } static struct ws2_async_io *alloc_async_io( DWORD size, async_callback_t callback ) { /* first free remaining previous fileinfos */ struct ws2_async_io *io = InterlockedExchangePointer( (void **)&async_io_freelist, NULL ); while (io) { struct ws2_async_io *next = io->next; HeapFree( GetProcessHeap(), 0, io ); io = next; } io = HeapAlloc( GetProcessHeap(), 0, size ); if (io) io->callback = callback; return io; } static NTSTATUS register_async( int type, HANDLE handle, struct ws2_async_io *async, HANDLE event, PIO_APC_ROUTINE apc, void *apc_context, IO_STATUS_BLOCK *io ) { NTSTATUS status; SERVER_START_REQ( register_async ) { req->type = type; req->async.handle = wine_server_obj_handle( handle ); req->async.user = wine_server_client_ptr( async ); req->async.iosb = wine_server_client_ptr( io ); req->async.event = wine_server_obj_handle( event ); req->async.apc = wine_server_client_ptr( apc ); req->async.apc_context = wine_server_client_ptr( apc_context ); status = wine_server_call( req ); } SERVER_END_REQ; return status; } /****************************************************************/ /* ----------------------------------- internal data */ /* ws_... struct conversion flags */ typedef struct /* WSAAsyncSelect() control struct */ { HANDLE service, event, sock; HWND hWnd; UINT uMsg; LONG lEvent; } ws_select_info; #define WS_MAX_SOCKETS_PER_PROCESS 128 /* reasonable guess */ #define WS_MAX_UDP_DATAGRAM 1024 static INT WINAPI WSA_DefaultBlockingHook( FARPROC x ); /* hostent's, servent's and protent's are stored in one buffer per thread, * as documented on MSDN for the functions that return any of the buffers */ struct per_thread_data { int opentype; struct WS_hostent *he_buffer; struct WS_servent *se_buffer; struct WS_protoent *pe_buffer; struct pollfd *fd_cache; unsigned int fd_count; int he_len; int se_len; int pe_len; char ntoa_buffer[16]; /* 4*3 digits + 3 '.' + 1 '\0' */ }; /* internal: routing description information */ struct route { struct in_addr addr; IF_INDEX interface; DWORD metric, default_route; }; static INT num_startup; /* reference counter */ static FARPROC blocking_hook = (FARPROC)WSA_DefaultBlockingHook; /* function prototypes */ static struct WS_hostent *WS_create_he(char *name, int aliases, int aliases_size, int addresses, int address_length); static struct WS_hostent *WS_dup_he(const struct hostent* p_he); static struct WS_protoent *WS_create_pe( const char *name, char **aliases, int prot ); static struct WS_servent *WS_dup_se(const struct servent* p_se); static int ws_protocol_info(SOCKET s, int unicode, WSAPROTOCOL_INFOW *buffer, int *size); int WSAIOCTL_GetInterfaceCount(void); int WSAIOCTL_GetInterfaceName(int intNumber, char *intName); static void WS_AddCompletion( SOCKET sock, ULONG_PTR CompletionValue, NTSTATUS CompletionStatus, ULONG Information, BOOL force ); #define MAP_OPTION(opt) { WS_##opt, opt } static const int ws_flags_map[][2] = { MAP_OPTION( MSG_OOB ), MAP_OPTION( MSG_PEEK ), MAP_OPTION( MSG_DONTROUTE ), MAP_OPTION( MSG_WAITALL ), { WS_MSG_PARTIAL, 0 }, }; static const int ws_sock_map[][2] = { MAP_OPTION( SO_DEBUG ), MAP_OPTION( SO_ACCEPTCONN ), MAP_OPTION( SO_REUSEADDR ), MAP_OPTION( SO_KEEPALIVE ), MAP_OPTION( SO_DONTROUTE ), MAP_OPTION( SO_BROADCAST ), MAP_OPTION( SO_LINGER ), MAP_OPTION( SO_OOBINLINE ), MAP_OPTION( SO_SNDBUF ), MAP_OPTION( SO_RCVBUF ), MAP_OPTION( SO_ERROR ), MAP_OPTION( SO_TYPE ), #ifdef SO_RCVTIMEO MAP_OPTION( SO_RCVTIMEO ), #endif #ifdef SO_SNDTIMEO MAP_OPTION( SO_SNDTIMEO ), #endif }; static const int ws_tcp_map[][2] = { #ifdef TCP_NODELAY MAP_OPTION( TCP_NODELAY ), #endif }; static const int ws_ip_map[][2] = { MAP_OPTION( IP_MULTICAST_IF ), MAP_OPTION( IP_MULTICAST_TTL ), MAP_OPTION( IP_MULTICAST_LOOP ), MAP_OPTION( IP_ADD_MEMBERSHIP ), MAP_OPTION( IP_DROP_MEMBERSHIP ), MAP_OPTION( IP_ADD_SOURCE_MEMBERSHIP ), MAP_OPTION( IP_DROP_SOURCE_MEMBERSHIP ), MAP_OPTION( IP_BLOCK_SOURCE ), MAP_OPTION( IP_UNBLOCK_SOURCE ), MAP_OPTION( IP_OPTIONS ), #ifdef IP_HDRINCL MAP_OPTION( IP_HDRINCL ), #endif MAP_OPTION( IP_TOS ), MAP_OPTION( IP_TTL ), #if defined(IP_PKTINFO) MAP_OPTION( IP_PKTINFO ), #elif defined(IP_RECVDSTADDR) { WS_IP_PKTINFO, IP_RECVDSTADDR }, #endif #ifdef IP_UNICAST_IF MAP_OPTION( IP_UNICAST_IF ), #endif }; static const int ws_ipv6_map[][2] = { #ifdef IPV6_ADD_MEMBERSHIP MAP_OPTION( IPV6_ADD_MEMBERSHIP ), #endif #ifdef IPV6_DROP_MEMBERSHIP MAP_OPTION( IPV6_DROP_MEMBERSHIP ), #endif MAP_OPTION( IPV6_MULTICAST_IF ), MAP_OPTION( IPV6_MULTICAST_HOPS ), MAP_OPTION( IPV6_MULTICAST_LOOP ), MAP_OPTION( IPV6_UNICAST_HOPS ), MAP_OPTION( IPV6_V6ONLY ), #ifdef IPV6_UNICAST_IF MAP_OPTION( IPV6_UNICAST_IF ), #endif }; static const int ws_af_map[][2] = { MAP_OPTION( AF_UNSPEC ), MAP_OPTION( AF_INET ), MAP_OPTION( AF_INET6 ), #ifdef HAS_IPX MAP_OPTION( AF_IPX ), #endif #ifdef AF_IRDA MAP_OPTION( AF_IRDA ), #endif {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO}, }; static const int ws_socktype_map[][2] = { MAP_OPTION( SOCK_DGRAM ), MAP_OPTION( SOCK_STREAM ), MAP_OPTION( SOCK_RAW ), {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO}, }; static const int ws_proto_map[][2] = { MAP_OPTION( IPPROTO_IP ), MAP_OPTION( IPPROTO_TCP ), MAP_OPTION( IPPROTO_UDP ), MAP_OPTION( IPPROTO_IPV6 ), MAP_OPTION( IPPROTO_ICMP ), MAP_OPTION( IPPROTO_IGMP ), MAP_OPTION( IPPROTO_RAW ), MAP_OPTION( IPPROTO_IPIP ), {FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO}, }; static const int ws_aiflag_map[][2] = { MAP_OPTION( AI_PASSIVE ), MAP_OPTION( AI_CANONNAME ), MAP_OPTION( AI_NUMERICHOST ), #ifdef AI_NUMERICSERV MAP_OPTION( AI_NUMERICSERV ), #endif #ifdef AI_V4MAPPED MAP_OPTION( AI_V4MAPPED ), #endif MAP_OPTION( AI_ALL ), MAP_OPTION( AI_ADDRCONFIG ), }; static const int ws_niflag_map[][2] = { MAP_OPTION( NI_NOFQDN ), MAP_OPTION( NI_NUMERICHOST ), MAP_OPTION( NI_NAMEREQD ), MAP_OPTION( NI_NUMERICSERV ), MAP_OPTION( NI_DGRAM ), }; static const int ws_eai_map[][2] = { MAP_OPTION( EAI_AGAIN ), MAP_OPTION( EAI_BADFLAGS ), MAP_OPTION( EAI_FAIL ), MAP_OPTION( EAI_FAMILY ), MAP_OPTION( EAI_MEMORY ), /* Note: EAI_NODATA is deprecated, but still * used by Windows and Linux... We map the newer * EAI_NONAME to EAI_NODATA for now until Windows * changes too. */ #ifdef EAI_NODATA MAP_OPTION( EAI_NODATA ), #endif #ifdef EAI_NONAME { WS_EAI_NODATA, EAI_NONAME }, #endif MAP_OPTION( EAI_SERVICE ), MAP_OPTION( EAI_SOCKTYPE ), { 0, 0 } }; static const int ws_poll_map[][2] = { MAP_OPTION( POLLERR ), MAP_OPTION( POLLHUP ), MAP_OPTION( POLLNVAL ), MAP_OPTION( POLLWRNORM ), MAP_OPTION( POLLWRBAND ), MAP_OPTION( POLLRDNORM ), { WS_POLLRDBAND, POLLPRI } }; static const char magic_loopback_addr[] = {127, 12, 34, 56}; #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS #if defined(IP_PKTINFO) || defined(IP_RECVDSTADDR) static inline WSACMSGHDR *fill_control_message(int level, int type, WSACMSGHDR *current, ULONG *maxsize, void *data, int len) { ULONG msgsize = sizeof(WSACMSGHDR) + WSA_CMSG_ALIGN(len); char *ptr = (char *) current + sizeof(WSACMSGHDR); /* Make sure there is at least enough room for this entry */ if (msgsize > *maxsize) return NULL; *maxsize -= msgsize; /* Fill in the entry */ current->cmsg_len = sizeof(WSACMSGHDR) + len; current->cmsg_level = level; current->cmsg_type = type; memcpy(ptr, data, len); /* Return the pointer to where next entry should go */ return (WSACMSGHDR *) (ptr + WSA_CMSG_ALIGN(len)); } #endif /* defined(IP_PKTINFO) || defined(IP_RECVDSTADDR) */ static inline int convert_control_headers(struct msghdr *hdr, WSABUF *control) { #if defined(IP_PKTINFO) || defined(IP_RECVDSTADDR) WSACMSGHDR *cmsg_win = (WSACMSGHDR *) control->buf, *ptr; ULONG ctlsize = control->len; struct cmsghdr *cmsg_unix; ptr = cmsg_win; /* Loop over all the headers, converting as appropriate */ for (cmsg_unix = CMSG_FIRSTHDR(hdr); cmsg_unix != NULL; cmsg_unix = CMSG_NXTHDR(hdr, cmsg_unix)) { switch(cmsg_unix->cmsg_level) { case IPPROTO_IP: switch(cmsg_unix->cmsg_type) { #if defined(IP_PKTINFO) case IP_PKTINFO: { /* Convert the Unix IP_PKTINFO structure to the Windows version */ struct in_pktinfo *data_unix = (struct in_pktinfo *) CMSG_DATA(cmsg_unix); struct WS_in_pktinfo data_win; memcpy(&data_win.ipi_addr,&data_unix->ipi_addr.s_addr,4); /* 4 bytes = 32 address bits */ data_win.ipi_ifindex = data_unix->ipi_ifindex; ptr = fill_control_message(WS_IPPROTO_IP, WS_IP_PKTINFO, ptr, &ctlsize, (void*)&data_win, sizeof(data_win)); if (!ptr) goto error; } break; #elif defined(IP_RECVDSTADDR) case IP_RECVDSTADDR: { struct in_addr *addr_unix = (struct in_addr *) CMSG_DATA(cmsg_unix); struct WS_in_pktinfo data_win; memcpy(&data_win.ipi_addr, &addr_unix->s_addr, 4); /* 4 bytes = 32 address bits */ data_win.ipi_ifindex = 0; /* FIXME */ ptr = fill_control_message(WS_IPPROTO_IP, WS_IP_PKTINFO, ptr, &ctlsize, (void*)&data_win, sizeof(data_win)); if (!ptr) goto error; } break; #endif /* IP_PKTINFO */ default: FIXME("Unhandled IPPROTO_IP message header type %d\n", cmsg_unix->cmsg_type); break; } break; default: FIXME("Unhandled message header level %d\n", cmsg_unix->cmsg_level); break; } } /* Set the length of the returned control headers */ control->len = (char*)ptr - (char*)cmsg_win; return 1; error: control->len = 0; return 0; #else /* defined(IP_PKTINFO) || defined(IP_RECVDSTADDR) */ control->len = 0; return 1; #endif /* defined(IP_PKTINFO) || defined(IP_RECVDSTADDR) */ } #endif /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */ /* ----------------------------------- error handling */ static NTSTATUS sock_get_ntstatus( int err ) { switch ( err ) { case EBADF: return STATUS_INVALID_HANDLE; case EBUSY: return STATUS_DEVICE_BUSY; case EPERM: case EACCES: return STATUS_ACCESS_DENIED; case EFAULT: return STATUS_NO_MEMORY; case EINVAL: return STATUS_INVALID_PARAMETER; case ENFILE: case EMFILE: return STATUS_TOO_MANY_OPENED_FILES; case EWOULDBLOCK: return STATUS_CANT_WAIT; case EINPROGRESS: return STATUS_PENDING; case EALREADY: return STATUS_NETWORK_BUSY; case ENOTSOCK: return STATUS_OBJECT_TYPE_MISMATCH; case EDESTADDRREQ: return STATUS_INVALID_PARAMETER; case EMSGSIZE: return STATUS_BUFFER_OVERFLOW; case EPROTONOSUPPORT: case ESOCKTNOSUPPORT: case EPFNOSUPPORT: case EAFNOSUPPORT: case EPROTOTYPE: return STATUS_NOT_SUPPORTED; case ENOPROTOOPT: return STATUS_INVALID_PARAMETER; case EOPNOTSUPP: return STATUS_NOT_SUPPORTED; case EADDRINUSE: return STATUS_ADDRESS_ALREADY_ASSOCIATED; case EADDRNOTAVAIL: return STATUS_INVALID_PARAMETER; case ECONNREFUSED: return STATUS_CONNECTION_REFUSED; case ESHUTDOWN: return STATUS_PIPE_DISCONNECTED; case ENOTCONN: return STATUS_CONNECTION_DISCONNECTED; case ETIMEDOUT: return STATUS_IO_TIMEOUT; case ENETUNREACH: return STATUS_NETWORK_UNREACHABLE; case ENETDOWN: return STATUS_NETWORK_BUSY; case EPIPE: case ECONNRESET: return STATUS_CONNECTION_RESET; case ECONNABORTED: return STATUS_CONNECTION_ABORTED; case 0: return STATUS_SUCCESS; default: WARN("Unknown errno %d!\n", err); return STATUS_UNSUCCESSFUL; } } static UINT sock_get_error( int err ) { switch(err) { case EINTR: return WSAEINTR; case EPERM: case EACCES: return WSAEACCES; case EFAULT: return WSAEFAULT; case EINVAL: return WSAEINVAL; case EMFILE: return WSAEMFILE; case EWOULDBLOCK: return WSAEWOULDBLOCK; case EINPROGRESS: return WSAEINPROGRESS; case EALREADY: return WSAEALREADY; case EBADF: case ENOTSOCK: return WSAENOTSOCK; case EDESTADDRREQ: return WSAEDESTADDRREQ; case EMSGSIZE: return WSAEMSGSIZE; case EPROTOTYPE: return WSAEPROTOTYPE; case ENOPROTOOPT: return WSAENOPROTOOPT; case EPROTONOSUPPORT: return WSAEPROTONOSUPPORT; case ESOCKTNOSUPPORT: return WSAESOCKTNOSUPPORT; case EOPNOTSUPP: return WSAEOPNOTSUPP; case EPFNOSUPPORT: return WSAEPFNOSUPPORT; case EAFNOSUPPORT: return WSAEAFNOSUPPORT; case EADDRINUSE: return WSAEADDRINUSE; case EADDRNOTAVAIL: return WSAEADDRNOTAVAIL; case ENETDOWN: return WSAENETDOWN; case ENETUNREACH: return WSAENETUNREACH; case ENETRESET: return WSAENETRESET; case ECONNABORTED: return WSAECONNABORTED; case EPIPE: case ECONNRESET: return WSAECONNRESET; case ENOBUFS: return WSAENOBUFS; case EISCONN: return WSAEISCONN; case ENOTCONN: return WSAENOTCONN; case ESHUTDOWN: return WSAESHUTDOWN; case ETOOMANYREFS: return WSAETOOMANYREFS; case ETIMEDOUT: return WSAETIMEDOUT; case ECONNREFUSED: return WSAECONNREFUSED; case ELOOP: return WSAELOOP; case ENAMETOOLONG: return WSAENAMETOOLONG; case EHOSTDOWN: return WSAEHOSTDOWN; case EHOSTUNREACH: return WSAEHOSTUNREACH; case ENOTEMPTY: return WSAENOTEMPTY; #ifdef EPROCLIM case EPROCLIM: return WSAEPROCLIM; #endif #ifdef EUSERS case EUSERS: return WSAEUSERS; #endif #ifdef EDQUOT case EDQUOT: return WSAEDQUOT; #endif #ifdef ESTALE case ESTALE: return WSAESTALE; #endif #ifdef EREMOTE case EREMOTE: return WSAEREMOTE; #endif /* just in case we ever get here and there are no problems */ case 0: return 0; default: WARN("Unknown errno %d!\n", err); return WSAEOPNOTSUPP; } } static UINT wsaErrno(void) { int loc_errno = errno; WARN("errno %d, (%s).\n", loc_errno, strerror(loc_errno)); return sock_get_error( loc_errno ); } /* most ws2 overlapped functions return an ntstatus-based error code */ static NTSTATUS wsaErrStatus(void) { int loc_errno = errno; WARN("errno %d, (%s).\n", loc_errno, strerror(loc_errno)); return sock_get_ntstatus(loc_errno); } static UINT wsaHerrno(int loc_errno) { WARN("h_errno %d.\n", loc_errno); switch(loc_errno) { case HOST_NOT_FOUND: return WSAHOST_NOT_FOUND; case TRY_AGAIN: return WSATRY_AGAIN; case NO_RECOVERY: return WSANO_RECOVERY; case NO_DATA: return WSANO_DATA; case ENOBUFS: return WSAENOBUFS; case 0: return 0; default: WARN("Unknown h_errno %d!\n", loc_errno); return WSAEOPNOTSUPP; } } static NTSTATUS sock_error_to_ntstatus( DWORD err ) { switch (err) { case 0: return STATUS_SUCCESS; case WSAEBADF: return STATUS_INVALID_HANDLE; case WSAEACCES: return STATUS_ACCESS_DENIED; case WSAEFAULT: return STATUS_NO_MEMORY; case WSAEINVAL: return STATUS_INVALID_PARAMETER; case WSAEMFILE: return STATUS_TOO_MANY_OPENED_FILES; case WSAEWOULDBLOCK: return STATUS_CANT_WAIT; case WSAEINPROGRESS: return STATUS_PENDING; case WSAEALREADY: return STATUS_NETWORK_BUSY; case WSAENOTSOCK: return STATUS_OBJECT_TYPE_MISMATCH; case WSAEDESTADDRREQ: return STATUS_INVALID_PARAMETER; case WSAEMSGSIZE: return STATUS_BUFFER_OVERFLOW; case WSAEPROTONOSUPPORT: case WSAESOCKTNOSUPPORT: case WSAEPFNOSUPPORT: case WSAEAFNOSUPPORT: case WSAEPROTOTYPE: return STATUS_NOT_SUPPORTED; case WSAENOPROTOOPT: return STATUS_INVALID_PARAMETER; case WSAEOPNOTSUPP: return STATUS_NOT_SUPPORTED; case WSAEADDRINUSE: return STATUS_ADDRESS_ALREADY_ASSOCIATED; case WSAEADDRNOTAVAIL: return STATUS_INVALID_PARAMETER; case WSAECONNREFUSED: return STATUS_CONNECTION_REFUSED; case WSAESHUTDOWN: return STATUS_PIPE_DISCONNECTED; case WSAENOTCONN: return STATUS_CONNECTION_DISCONNECTED; case WSAETIMEDOUT: return STATUS_IO_TIMEOUT; case WSAENETUNREACH: return STATUS_NETWORK_UNREACHABLE; case WSAENETDOWN: return STATUS_NETWORK_BUSY; case WSAECONNRESET: return STATUS_CONNECTION_RESET; case WSAECONNABORTED: return STATUS_CONNECTION_ABORTED; default: FIXME("unmapped error %u\n", err); return STATUS_UNSUCCESSFUL; } } static DWORD NtStatusToWSAError( DWORD status ) { switch ( status ) { case STATUS_SUCCESS: return 0; case STATUS_PENDING: return WSA_IO_PENDING; case STATUS_INVALID_HANDLE: case STATUS_OBJECT_TYPE_MISMATCH: return WSAENOTSOCK; case STATUS_INVALID_PARAMETER: return WSAEINVAL; case STATUS_PIPE_DISCONNECTED: return WSAESHUTDOWN; case STATUS_NETWORK_BUSY: return WSAEALREADY; case STATUS_NETWORK_UNREACHABLE: return WSAENETUNREACH; case STATUS_CONNECTION_REFUSED: return WSAECONNREFUSED; case STATUS_CONNECTION_DISCONNECTED: return WSAENOTCONN; case STATUS_CONNECTION_RESET: return WSAECONNRESET; case STATUS_CONNECTION_ABORTED: return WSAECONNABORTED; case STATUS_CANCELLED: return WSA_OPERATION_ABORTED; case STATUS_ADDRESS_ALREADY_ASSOCIATED: return WSAEADDRINUSE; case STATUS_IO_TIMEOUT: case STATUS_TIMEOUT: return WSAETIMEDOUT; case STATUS_NO_MEMORY: return WSAEFAULT; case STATUS_ACCESS_DENIED: return WSAEACCES; case STATUS_TOO_MANY_OPENED_FILES: return WSAEMFILE; case STATUS_CANT_WAIT: return WSAEWOULDBLOCK; case STATUS_BUFFER_OVERFLOW: return WSAEMSGSIZE; case STATUS_NOT_SUPPORTED: return WSAEOPNOTSUPP; case STATUS_HOST_UNREACHABLE: return WSAEHOSTUNREACH; default: return RtlNtStatusToDosError( status ); } } /* set last error code from NT status without mapping WSA errors */ static inline unsigned int set_error( unsigned int err ) { if (err) { err = NtStatusToWSAError( err ); SetLastError( err ); } return err; } static inline int get_sock_fd( SOCKET s, DWORD access, unsigned int *options ) { int fd; if (set_error( wine_server_handle_to_fd( SOCKET2HANDLE(s), access, &fd, options ) )) return -1; return fd; } static inline void release_sock_fd( SOCKET s, int fd ) { wine_server_release_fd( SOCKET2HANDLE(s), fd ); } static void _enable_event( HANDLE s, unsigned int event, unsigned int sstate, unsigned int cstate ) { SERVER_START_REQ( enable_socket_event ) { req->handle = wine_server_obj_handle( s ); req->mask = event; req->sstate = sstate; req->cstate = cstate; wine_server_call( req ); } SERVER_END_REQ; } static DWORD sock_is_blocking(SOCKET s, BOOL *ret) { DWORD err; SERVER_START_REQ( get_socket_event ) { req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) ); req->service = FALSE; req->c_event = 0; err = NtStatusToWSAError( wine_server_call( req )); *ret = (reply->state & FD_WINE_NONBLOCKING) == 0; } SERVER_END_REQ; return err; } static unsigned int _get_sock_mask(SOCKET s) { unsigned int ret; SERVER_START_REQ( get_socket_event ) { req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) ); req->service = FALSE; req->c_event = 0; wine_server_call( req ); ret = reply->mask; } SERVER_END_REQ; return ret; } static void _sync_sock_state(SOCKET s) { BOOL dummy; /* do a dummy wineserver request in order to let the wineserver run through its select loop once */ sock_is_blocking(s, &dummy); } static void _get_sock_errors(SOCKET s, int *events) { SERVER_START_REQ( get_socket_event ) { req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) ); req->service = FALSE; req->c_event = 0; wine_server_set_reply( req, events, sizeof(int) * FD_MAX_EVENTS ); wine_server_call( req ); } SERVER_END_REQ; } static int get_sock_error(SOCKET s, unsigned int bit) { int events[FD_MAX_EVENTS]; _get_sock_errors(s, events); return events[bit]; } static int _get_fd_type(int fd) { int sock_type = -1; socklen_t optlen = sizeof(sock_type); getsockopt(fd, SOL_SOCKET, SO_TYPE, (char*) &sock_type, &optlen); return sock_type; } static BOOL set_dont_fragment(SOCKET s, int level, BOOL value) { int fd, optname; if (level == IPPROTO_IP) { #ifdef IP_DONTFRAG optname = IP_DONTFRAG; #elif defined(IP_MTU_DISCOVER) && defined(IP_PMTUDISC_DO) && defined(IP_PMTUDISC_DONT) optname = IP_MTU_DISCOVER; value = value ? IP_PMTUDISC_DO : IP_PMTUDISC_DONT; #else static int once; if (!once++) FIXME("IP_DONTFRAGMENT for IPv4 not supported in this platform\n"); return TRUE; /* fake success */ #endif } else { #ifdef IPV6_DONTFRAG optname = IPV6_DONTFRAG; #elif defined(IPV6_MTU_DISCOVER) && defined(IPV6_PMTUDISC_DO) && defined(IPV6_PMTUDISC_DONT) optname = IPV6_MTU_DISCOVER; value = value ? IPV6_PMTUDISC_DO : IPV6_PMTUDISC_DONT; #else static int once; if (!once++) FIXME("IP_DONTFRAGMENT for IPv6 not supported in this platform\n"); return TRUE; /* fake success */ #endif } fd = get_sock_fd(s, 0, NULL); if (fd == -1) return FALSE; if (!setsockopt(fd, level, optname, &value, sizeof(value))) value = TRUE; else { WSASetLastError(wsaErrno()); value = FALSE; } release_sock_fd(s, fd); return value; } static BOOL get_dont_fragment(SOCKET s, int level, BOOL *out) { int fd, optname, value, not_expected; socklen_t optlen = sizeof(value); if (level == IPPROTO_IP) { #ifdef IP_DONTFRAG optname = IP_DONTFRAG; not_expected = 0; #elif defined(IP_MTU_DISCOVER) && defined(IP_PMTUDISC_DONT) optname = IP_MTU_DISCOVER; not_expected = IP_PMTUDISC_DONT; #else static int once; if (!once++) FIXME("IP_DONTFRAGMENT for IPv4 not supported in this platform\n"); return TRUE; /* fake success */ #endif } else { #ifdef IPV6_DONTFRAG optname = IPV6_DONTFRAG; not_expected = 0; #elif defined(IPV6_MTU_DISCOVER) && defined(IPV6_PMTUDISC_DONT) optname = IPV6_MTU_DISCOVER; not_expected = IPV6_PMTUDISC_DONT; #else static int once; if (!once++) FIXME("IP_DONTFRAGMENT for IPv6 not supported in this platform\n"); return TRUE; /* fake success */ #endif } fd = get_sock_fd(s, 0, NULL); if (fd == -1) return FALSE; if (!getsockopt(fd, level, optname, &value, &optlen)) { *out = value != not_expected; value = TRUE; } else { WSASetLastError(wsaErrno()); value = FALSE; } release_sock_fd(s, fd); return value; } static struct per_thread_data *get_per_thread_data(void) { struct per_thread_data * ptb = NtCurrentTeb()->WinSockData; /* lazy initialization */ if (!ptb) { ptb = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*ptb) ); NtCurrentTeb()->WinSockData = ptb; } return ptb; } static void free_per_thread_data(void) { struct per_thread_data * ptb = NtCurrentTeb()->WinSockData; if (!ptb) return; /* delete scratch buffers */ HeapFree( GetProcessHeap(), 0, ptb->he_buffer ); HeapFree( GetProcessHeap(), 0, ptb->se_buffer ); HeapFree( GetProcessHeap(), 0, ptb->pe_buffer ); HeapFree( GetProcessHeap(), 0, ptb->fd_cache ); HeapFree( GetProcessHeap(), 0, ptb ); NtCurrentTeb()->WinSockData = NULL; } /*********************************************************************** * DllMain (WS2_32.init) */ BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID fImpLoad) { TRACE("%p 0x%x %p\n", hInstDLL, fdwReason, fImpLoad); switch (fdwReason) { case DLL_PROCESS_ATTACH: break; case DLL_PROCESS_DETACH: if (fImpLoad) break; free_per_thread_data(); DeleteCriticalSection(&csWSgetXXXbyYYY); break; case DLL_THREAD_DETACH: free_per_thread_data(); break; } return TRUE; } /*********************************************************************** * convert_flags() * * Converts send/recv flags from Windows format. * Return the converted flag bits, unsupported flags remain unchanged. */ static int convert_flags(int flags) { int i, out; if (!flags) return 0; for (out = i = 0; flags && i < ARRAY_SIZE(ws_flags_map); i++) { if (ws_flags_map[i][0] & flags) { out |= ws_flags_map[i][1]; flags &= ~ws_flags_map[i][0]; } } if (flags) { FIXME("Unknown send/recv flags 0x%x, using anyway...\n", flags); out |= flags; } return out; } /*********************************************************************** * convert_sockopt() * * Converts socket flags from Windows format. * Return 1 if converted, 0 if not (error). */ static int convert_sockopt(INT *level, INT *optname) { unsigned int i; switch (*level) { case WS_SOL_SOCKET: *level = SOL_SOCKET; for(i = 0; i < ARRAY_SIZE(ws_sock_map); i++) { if( ws_sock_map[i][0] == *optname ) { *optname = ws_sock_map[i][1]; return 1; } } FIXME("Unknown SOL_SOCKET optname 0x%x\n", *optname); break; case WS_IPPROTO_TCP: *level = IPPROTO_TCP; for(i = 0; i < ARRAY_SIZE(ws_tcp_map); i++) { if ( ws_tcp_map[i][0] == *optname ) { *optname = ws_tcp_map[i][1]; return 1; } } FIXME("Unknown IPPROTO_TCP optname 0x%x\n", *optname); break; case WS_IPPROTO_IP: *level = IPPROTO_IP; for(i = 0; i < ARRAY_SIZE(ws_ip_map); i++) { if (ws_ip_map[i][0] == *optname ) { *optname = ws_ip_map[i][1]; return 1; } } FIXME("Unknown IPPROTO_IP optname 0x%x\n", *optname); break; case WS_IPPROTO_IPV6: *level = IPPROTO_IPV6; for(i = 0; i < ARRAY_SIZE(ws_ipv6_map); i++) { if (ws_ipv6_map[i][0] == *optname ) { *optname = ws_ipv6_map[i][1]; return 1; } } FIXME("Unknown IPPROTO_IPV6 optname 0x%x\n", *optname); break; default: FIXME("Unimplemented or unknown socket level\n"); } return 0; } /* ----------------------------------- Per-thread info (or per-process?) */ static char *strdup_lower(const char *str) { int i; char *ret = HeapAlloc( GetProcessHeap(), 0, strlen(str) + 1 ); if (ret) { for (i = 0; str[i]; i++) ret[i] = tolower(str[i]); ret[i] = 0; } else SetLastError(WSAENOBUFS); return ret; } /* Utility: get the SO_RCVTIMEO or SO_SNDTIMEO socket option * from an fd and return the value converted to milli seconds * or 0 if there is an infinite time out */ static inline INT64 get_rcvsnd_timeo( int fd, BOOL is_recv) { struct timeval tv; socklen_t len = sizeof(tv); int optname, res; if (is_recv) #ifdef SO_RCVTIMEO optname = SO_RCVTIMEO; #else return 0; #endif else #ifdef SO_SNDTIMEO optname = SO_SNDTIMEO; #else return 0; #endif res = getsockopt(fd, SOL_SOCKET, optname, &tv, &len); if (res < 0) return 0; return (UINT64)tv.tv_sec * 1000 + tv.tv_usec / 1000; } /* utility: given an fd, will block until one of the events occurs */ static inline int do_block( int fd, int events, int timeout ) { struct pollfd pfd; int ret; pfd.fd = fd; pfd.events = events; while ((ret = poll(&pfd, 1, timeout)) < 0) { if (errno != EINTR) return -1; } if( ret == 0 ) return 0; return pfd.revents; } static int convert_af_w2u(int windowsaf) { unsigned int i; for (i = 0; i < ARRAY_SIZE(ws_af_map); i++) if (ws_af_map[i][0] == windowsaf) return ws_af_map[i][1]; FIXME("unhandled Windows address family %d\n", windowsaf); return -1; } static int convert_af_u2w(int unixaf) { unsigned int i; for (i = 0; i < ARRAY_SIZE(ws_af_map); i++) if (ws_af_map[i][1] == unixaf) return ws_af_map[i][0]; FIXME("unhandled UNIX address family %d\n", unixaf); return -1; } static int convert_proto_w2u(int windowsproto) { unsigned int i; for (i = 0; i < ARRAY_SIZE(ws_proto_map); i++) if (ws_proto_map[i][0] == windowsproto) return ws_proto_map[i][1]; /* check for extended IPX */ if (IS_IPX_PROTO(windowsproto)) return windowsproto; FIXME("unhandled Windows socket protocol %d\n", windowsproto); return -1; } static int convert_proto_u2w(int unixproto) { unsigned int i; for (i = 0; i < ARRAY_SIZE(ws_proto_map); i++) if (ws_proto_map[i][1] == unixproto) return ws_proto_map[i][0]; /* if value is inside IPX range just return it - the kernel simply * echoes the value used in the socket() function */ if (IS_IPX_PROTO(unixproto)) return unixproto; FIXME("unhandled UNIX socket protocol %d\n", unixproto); return -1; } static int convert_socktype_w2u(int windowssocktype) { unsigned int i; for (i = 0; i < ARRAY_SIZE(ws_socktype_map); i++) if (ws_socktype_map[i][0] == windowssocktype) return ws_socktype_map[i][1]; FIXME("unhandled Windows socket type %d\n", windowssocktype); return -1; } static int convert_socktype_u2w(int unixsocktype) { unsigned int i; for (i = 0; i < ARRAY_SIZE(ws_socktype_map); i++) if (ws_socktype_map[i][1] == unixsocktype) return ws_socktype_map[i][0]; FIXME("unhandled UNIX socket type %d\n", unixsocktype); return -1; } static int convert_poll_w2u(int events) { int i, ret; for (i = ret = 0; events && i < ARRAY_SIZE(ws_poll_map); i++) { if (ws_poll_map[i][0] & events) { ret |= ws_poll_map[i][1]; events &= ~ws_poll_map[i][0]; } } if (events) FIXME("Unsupported WSAPoll() flags 0x%x\n", events); return ret; } static int convert_poll_u2w(int events) { int i, ret; for (i = ret = 0; events && i < ARRAY_SIZE(ws_poll_map); i++) { if (ws_poll_map[i][1] & events) { ret |= ws_poll_map[i][0]; events &= ~ws_poll_map[i][1]; } } if (events) FIXME("Unsupported poll() flags 0x%x\n", events); return ret; } static int set_ipx_packettype(int sock, int ptype) { #ifdef HAS_IPX int fd = get_sock_fd( sock, 0, NULL ), ret = 0; TRACE("trying to set IPX_PTYPE: %d (fd: %d)\n", ptype, fd); if (fd == -1) return SOCKET_ERROR; /* We try to set the ipx type on ipx socket level. */ #ifdef SOL_IPX if(setsockopt(fd, SOL_IPX, IPX_TYPE, &ptype, sizeof(ptype)) == -1) { ERR("IPX: could not set ipx option type; expect weird behaviour\n"); ret = SOCKET_ERROR; } #else { struct ipx val; /* Should we retrieve val using a getsockopt call and then * set the modified one? */ val.ipx_pt = ptype; setsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, sizeof(struct ipx)); } #endif release_sock_fd( sock, fd ); return ret; #else WARN("IPX support is not enabled, can't set packet type\n"); return SOCKET_ERROR; #endif } /* ----------------------------------- API ----- * * Init / cleanup / error checking. */ /*********************************************************************** * WSAStartup (WS2_32.115) */ int WINAPI WSAStartup(WORD wVersionRequested, LPWSADATA lpWSAData) { TRACE("verReq=%x\n", wVersionRequested); if (LOBYTE(wVersionRequested) < 1) return WSAVERNOTSUPPORTED; if (!lpWSAData) return WSAEINVAL; num_startup++; /* that's the whole of the negotiation for now */ lpWSAData->wVersion = wVersionRequested; /* return winsock information */ lpWSAData->wHighVersion = 0x0202; strcpy(lpWSAData->szDescription, "WinSock 2.0" ); strcpy(lpWSAData->szSystemStatus, "Running" ); lpWSAData->iMaxSockets = WS_MAX_SOCKETS_PER_PROCESS; lpWSAData->iMaxUdpDg = WS_MAX_UDP_DATAGRAM; /* don't do anything with lpWSAData->lpVendorInfo */ /* (some apps don't allocate the space for this field) */ TRACE("succeeded starts: %d\n", num_startup); return 0; } /*********************************************************************** * WSACleanup (WS2_32.116) */ INT WINAPI WSACleanup(void) { TRACE("decreasing startup count from %d\n", num_startup); if (num_startup) { if (!--num_startup) { unsigned int i; for (i = 0; i < socket_list_size; ++i) CloseHandle(SOCKET2HANDLE(socket_list[i])); memset(socket_list, 0, socket_list_size * sizeof(*socket_list)); } return 0; } SetLastError(WSANOTINITIALISED); return SOCKET_ERROR; } /*********************************************************************** * WSAGetLastError (WS2_32.111) */ INT WINAPI WSAGetLastError(void) { return GetLastError(); } /*********************************************************************** * WSASetLastError (WS2_32.112) */ void WINAPI WSASetLastError(INT iError) { SetLastError(iError); } static struct WS_hostent *check_buffer_he(int size) { struct per_thread_data * ptb = get_per_thread_data(); if (ptb->he_buffer) { if (ptb->he_len >= size ) return ptb->he_buffer; HeapFree( GetProcessHeap(), 0, ptb->he_buffer ); } ptb->he_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->he_len = size) ); if (!ptb->he_buffer) SetLastError(WSAENOBUFS); return ptb->he_buffer; } static struct WS_servent *check_buffer_se(int size) { struct per_thread_data * ptb = get_per_thread_data(); if (ptb->se_buffer) { if (ptb->se_len >= size ) return ptb->se_buffer; HeapFree( GetProcessHeap(), 0, ptb->se_buffer ); } ptb->se_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->se_len = size) ); if (!ptb->se_buffer) SetLastError(WSAENOBUFS); return ptb->se_buffer; } static struct WS_protoent *check_buffer_pe(int size) { struct per_thread_data * ptb = get_per_thread_data(); if (ptb->pe_buffer) { if (ptb->pe_len >= size ) return ptb->pe_buffer; HeapFree( GetProcessHeap(), 0, ptb->pe_buffer ); } ptb->pe_buffer = HeapAlloc( GetProcessHeap(), 0, (ptb->pe_len = size) ); if (!ptb->pe_buffer) SetLastError(WSAENOBUFS); return ptb->pe_buffer; } /* ----------------------------------- i/o APIs */ static inline BOOL supported_pf(int pf) { switch (pf) { case WS_AF_INET: case WS_AF_INET6: return TRUE; #ifdef HAS_IPX case WS_AF_IPX: return TRUE; #endif #ifdef HAS_IRDA case WS_AF_IRDA: return TRUE; #endif default: return FALSE; } } /**********************************************************************/ /* Returns the length of the converted address if successful, 0 if it was too * small to start with or unknown family or invalid address buffer. */ static unsigned int ws_sockaddr_ws2u(const struct WS_sockaddr* wsaddr, int wsaddrlen, union generic_unix_sockaddr *uaddr) { unsigned int uaddrlen = 0; if (!wsaddr) return 0; switch (wsaddr->sa_family) { #ifdef HAS_IPX case WS_AF_IPX: { const struct WS_sockaddr_ipx* wsipx=(const struct WS_sockaddr_ipx*)wsaddr; struct sockaddr_ipx* uipx = (struct sockaddr_ipx *)uaddr; if (wsaddrlensipx_family=AF_IPX; uipx->sipx_port=wsipx->sa_socket; /* copy sa_netnum and sa_nodenum to sipx_network and sipx_node * in one go */ memcpy(&uipx->sipx_network,wsipx->sa_netnum,sizeof(uipx->sipx_network)+sizeof(uipx->sipx_node)); #ifdef IPX_FRAME_NONE uipx->sipx_type=IPX_FRAME_NONE; #endif break; } #endif case WS_AF_INET6: { struct sockaddr_in6* uin6 = (struct sockaddr_in6 *)uaddr; const struct WS_sockaddr_in6* win6 = (const struct WS_sockaddr_in6*)wsaddr; /* Note: Windows has 2 versions of the sockaddr_in6 struct, one with * scope_id, one without. */ if (wsaddrlen >= sizeof(struct WS_sockaddr_in6_old)) { uaddrlen = sizeof(struct sockaddr_in6); memset( uaddr, 0, uaddrlen ); uin6->sin6_family = AF_INET6; uin6->sin6_port = win6->sin6_port; uin6->sin6_flowinfo = win6->sin6_flowinfo; #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID if (wsaddrlen >= sizeof(struct WS_sockaddr_in6)) uin6->sin6_scope_id = win6->sin6_scope_id; #endif memcpy(&uin6->sin6_addr,&win6->sin6_addr,16); /* 16 bytes = 128 address bits */ break; } FIXME("bad size %d for WS_sockaddr_in6\n",wsaddrlen); return 0; } case WS_AF_INET: { struct sockaddr_in* uin = (struct sockaddr_in *)uaddr; const struct WS_sockaddr_in* win = (const struct WS_sockaddr_in*)wsaddr; if (wsaddrlensin_family = AF_INET; uin->sin_port = win->sin_port; memcpy(&uin->sin_addr,&win->sin_addr,4); /* 4 bytes = 32 address bits */ break; } #ifdef HAS_IRDA case WS_AF_IRDA: { struct sockaddr_irda *uin = (struct sockaddr_irda *)uaddr; const SOCKADDR_IRDA *win = (const SOCKADDR_IRDA *)wsaddr; if (wsaddrlen < sizeof(SOCKADDR_IRDA)) return 0; uaddrlen = sizeof(struct sockaddr_irda); memset( uaddr, 0, uaddrlen ); uin->sir_family = AF_IRDA; if (!strncmp( win->irdaServiceName, "LSAP-SEL", strlen( "LSAP-SEL" ) )) { unsigned int lsap_sel = 0; sscanf( win->irdaServiceName, "LSAP-SEL%u", &lsap_sel ); uin->sir_lsap_sel = lsap_sel; } else { uin->sir_lsap_sel = LSAP_ANY; memcpy( uin->sir_name, win->irdaServiceName, 25 ); } memcpy( &uin->sir_addr, win->irdaDeviceID, sizeof(uin->sir_addr) ); break; } #endif case WS_AF_UNSPEC: { /* Try to determine the needed space by the passed windows sockaddr space */ switch (wsaddrlen) { default: /* likely an ipv4 address */ case sizeof(struct WS_sockaddr_in): uaddrlen = sizeof(struct sockaddr_in); break; #ifdef HAS_IPX case sizeof(struct WS_sockaddr_ipx): uaddrlen = sizeof(struct sockaddr_ipx); break; #endif #ifdef HAS_IRDA case sizeof(SOCKADDR_IRDA): uaddrlen = sizeof(struct sockaddr_irda); break; #endif case sizeof(struct WS_sockaddr_in6): case sizeof(struct WS_sockaddr_in6_old): uaddrlen = sizeof(struct sockaddr_in6); break; } memset( uaddr, 0, uaddrlen ); break; } default: FIXME("Unknown address family %d, return NULL.\n", wsaddr->sa_family); return 0; } return uaddrlen; } static BOOL is_sockaddr_bound(const struct sockaddr *uaddr, int uaddrlen) { switch (uaddr->sa_family) { #ifdef HAS_IPX case AF_IPX: { static const struct sockaddr_ipx emptyAddr; struct sockaddr_ipx *ipx = (struct sockaddr_ipx*) uaddr; return ipx->sipx_port || memcmp(&ipx->sipx_network, &emptyAddr.sipx_network, sizeof(emptyAddr.sipx_network)) || memcmp(&ipx->sipx_node, &emptyAddr.sipx_node, sizeof(emptyAddr.sipx_node)); } #endif case AF_INET6: { static const struct sockaddr_in6 emptyAddr; const struct sockaddr_in6 *in6 = (const struct sockaddr_in6*) uaddr; return in6->sin6_port || memcmp(&in6->sin6_addr, &emptyAddr.sin6_addr, sizeof(struct in6_addr)); } case AF_INET: { static const struct sockaddr_in emptyAddr; const struct sockaddr_in *in = (const struct sockaddr_in*) uaddr; return in->sin_port || memcmp(&in->sin_addr, &emptyAddr.sin_addr, sizeof(struct in_addr)); } case AF_UNSPEC: return FALSE; default: FIXME("unknown address family %d\n", uaddr->sa_family); return TRUE; } } /* Returns -1 if getsockname fails, 0 if not bound, 1 otherwise */ static int is_fd_bound(int fd, union generic_unix_sockaddr *uaddr, socklen_t *uaddrlen) { union generic_unix_sockaddr inaddr; socklen_t inlen; int res; if (!uaddr) uaddr = &inaddr; if (!uaddrlen) uaddrlen = &inlen; *uaddrlen = sizeof(inaddr); res = getsockname(fd, &uaddr->addr, uaddrlen); if (!res) res = is_sockaddr_bound(&uaddr->addr, *uaddrlen); return res; } /* Returns 0 if successful, -1 if the buffer is too small */ static int ws_sockaddr_u2ws(const struct sockaddr* uaddr, struct WS_sockaddr* wsaddr, int* wsaddrlen) { int res; switch(uaddr->sa_family) { #ifdef HAS_IPX case AF_IPX: { const struct sockaddr_ipx* uipx=(const struct sockaddr_ipx*)uaddr; struct WS_sockaddr_ipx* wsipx=(struct WS_sockaddr_ipx*)wsaddr; res=-1; switch (*wsaddrlen) /* how much can we copy? */ { default: res=0; /* enough */ *wsaddrlen = sizeof(*wsipx); wsipx->sa_socket=uipx->sipx_port; /* fall through */ case 13: case 12: memcpy(wsipx->sa_nodenum,uipx->sipx_node,sizeof(wsipx->sa_nodenum)); /* fall through */ case 11: case 10: case 9: case 8: case 7: case 6: memcpy(wsipx->sa_netnum,&uipx->sipx_network,sizeof(wsipx->sa_netnum)); /* fall through */ case 5: case 4: case 3: case 2: wsipx->sa_family=WS_AF_IPX; /* fall through */ case 1: case 0: /* way too small */ break; } } break; #endif #ifdef HAS_IRDA case AF_IRDA: { const struct sockaddr_irda *uin = (const struct sockaddr_irda *)uaddr; SOCKADDR_IRDA *win = (SOCKADDR_IRDA *)wsaddr; if (*wsaddrlen < sizeof(SOCKADDR_IRDA)) return -1; win->irdaAddressFamily = WS_AF_IRDA; memcpy( win->irdaDeviceID, &uin->sir_addr, sizeof(win->irdaDeviceID) ); if (uin->sir_lsap_sel != LSAP_ANY) sprintf( win->irdaServiceName, "LSAP-SEL%u", uin->sir_lsap_sel ); else memcpy( win->irdaServiceName, uin->sir_name, sizeof(win->irdaServiceName) ); return 0; } #endif case AF_INET6: { const struct sockaddr_in6* uin6 = (const struct sockaddr_in6*)uaddr; struct WS_sockaddr_in6_old* win6old = (struct WS_sockaddr_in6_old*)wsaddr; if (*wsaddrlen < sizeof(struct WS_sockaddr_in6_old)) return -1; win6old->sin6_family = WS_AF_INET6; win6old->sin6_port = uin6->sin6_port; win6old->sin6_flowinfo = uin6->sin6_flowinfo; memcpy(&win6old->sin6_addr,&uin6->sin6_addr,16); /* 16 bytes = 128 address bits */ #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID if (*wsaddrlen >= sizeof(struct WS_sockaddr_in6)) { struct WS_sockaddr_in6* win6 = (struct WS_sockaddr_in6*)wsaddr; win6->sin6_scope_id = uin6->sin6_scope_id; *wsaddrlen = sizeof(struct WS_sockaddr_in6); } else *wsaddrlen = sizeof(struct WS_sockaddr_in6_old); #else *wsaddrlen = sizeof(struct WS_sockaddr_in6_old); #endif return 0; } case AF_INET: { const struct sockaddr_in* uin = (const struct sockaddr_in*)uaddr; struct WS_sockaddr_in* win = (struct WS_sockaddr_in*)wsaddr; if (*wsaddrlen < sizeof(struct WS_sockaddr_in)) return -1; win->sin_family = WS_AF_INET; win->sin_port = uin->sin_port; memcpy(&win->sin_addr,&uin->sin_addr,4); /* 4 bytes = 32 address bits */ memset(win->sin_zero, 0, 8); /* Make sure the null padding is null */ *wsaddrlen = sizeof(struct WS_sockaddr_in); return 0; } case AF_UNSPEC: { memset(wsaddr,0,*wsaddrlen); return 0; } default: FIXME("Unknown address family %d\n", uaddr->sa_family); return -1; } return res; } static INT WS_DuplicateSocket(BOOL unicode, SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOW lpProtocolInfo) { HANDLE hProcess; int size; WSAPROTOCOL_INFOW infow; TRACE("(unicode %d, socket %04lx, processid %x, buffer %p)\n", unicode, s, dwProcessId, lpProtocolInfo); if (!ws_protocol_info(s, unicode, &infow, &size)) return SOCKET_ERROR; if (!(hProcess = OpenProcess(PROCESS_DUP_HANDLE, FALSE, dwProcessId))) { SetLastError(WSAEINVAL); return SOCKET_ERROR; } if (!lpProtocolInfo) { CloseHandle(hProcess); SetLastError(WSAEFAULT); return SOCKET_ERROR; } /* I don't know what the real Windoze does next, this is a hack */ /* ...we could duplicate and then use ConvertToGlobalHandle on the duplicate, then let * the target use the global duplicate, or we could copy a reference to us to the structure * and let the target duplicate it from us, but let's do it as simple as possible */ memcpy(lpProtocolInfo, &infow, size); DuplicateHandle(GetCurrentProcess(), SOCKET2HANDLE(s), hProcess, (LPHANDLE)&lpProtocolInfo->dwServiceFlags3, 0, FALSE, DUPLICATE_SAME_ACCESS); CloseHandle(hProcess); lpProtocolInfo->dwServiceFlags4 = 0xff00ff00; /* magic */ return 0; } static BOOL ws_protocol_info(SOCKET s, int unicode, WSAPROTOCOL_INFOW *buffer, int *size) { NTSTATUS status; int address_family; int socket_type; int protocol; unsigned int i; *size = unicode ? sizeof(WSAPROTOCOL_INFOW) : sizeof(WSAPROTOCOL_INFOA); memset(buffer, 0, *size); SERVER_START_REQ( get_socket_info ) { req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) ); status = wine_server_call( req ); if (!status) { address_family = reply->family; socket_type = reply->type; protocol = reply->protocol; } } SERVER_END_REQ; if (status) { unsigned int err = NtStatusToWSAError( status ); SetLastError( err == WSAEBADF ? WSAENOTSOCK : err ); return FALSE; } for (i = 0; i < ARRAY_SIZE(supported_protocols); ++i) { const WSAPROTOCOL_INFOW *info = &supported_protocols[i]; if (address_family == info->iAddressFamily && socket_type == info->iSocketType && protocol >= info->iProtocol && protocol <= info->iProtocol + info->iProtocolMaxOffset) { if (unicode) *buffer = *info; else { WSAPROTOCOL_INFOA *bufferA = (WSAPROTOCOL_INFOA *)buffer; memcpy( bufferA, info, offsetof( WSAPROTOCOL_INFOW, szProtocol ) ); WideCharToMultiByte( CP_ACP, 0, info->szProtocol, -1, bufferA->szProtocol, sizeof(bufferA->szProtocol), NULL, NULL ); } buffer->iProtocol = protocol; return TRUE; } } FIXME("Could not fill protocol information for family %d, type %d, protocol %d.\n", address_family, socket_type, protocol); return TRUE; } /************************************************************************** * Functions for handling overlapped I/O **************************************************************************/ /* user APC called upon async completion */ static void WINAPI ws2_async_apc( void *arg, IO_STATUS_BLOCK *iosb, ULONG reserved ) { struct ws2_async *wsa = arg; if (wsa->completion_func) wsa->completion_func( NtStatusToWSAError(iosb->u.Status), iosb->Information, wsa->user_overlapped, wsa->flags ); release_async_io( &wsa->io ); } /*********************************************************************** * WS2_recv (INTERNAL) * * Workhorse for both synchronous and asynchronous recv() operations. */ static int WS2_recv( int fd, struct ws2_async *wsa, int flags ) { #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS char pktbuf[512]; #endif struct msghdr hdr; union generic_unix_sockaddr unix_sockaddr; int n; hdr.msg_name = NULL; if (wsa->addr) { hdr.msg_namelen = sizeof(unix_sockaddr); hdr.msg_name = &unix_sockaddr; } else hdr.msg_namelen = 0; hdr.msg_iov = wsa->iovec + wsa->first_iovec; hdr.msg_iovlen = wsa->n_iovecs - wsa->first_iovec; #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS hdr.msg_accrights = NULL; hdr.msg_accrightslen = 0; #else hdr.msg_control = pktbuf; hdr.msg_controllen = sizeof(pktbuf); hdr.msg_flags = 0; #endif while ((n = __wine_locked_recvmsg( fd, &hdr, flags )) == -1) { if (errno != EINTR) return -1; } #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS if (wsa->control) { ERR("Message control headers cannot be properly supported on this system.\n"); wsa->control->len = 0; } #else if (wsa->control && !convert_control_headers(&hdr, wsa->control)) { WARN("Application passed insufficient room for control headers.\n"); *wsa->lpFlags |= WS_MSG_CTRUNC; errno = EMSGSIZE; return -1; } #endif /* if this socket is connected and lpFrom is not NULL, Linux doesn't give us * msg_name and msg_namelen from recvmsg, but it does set msg_namelen to zero. * * quoting linux 2.6 net/ipv4/tcp.c: * "According to UNIX98, msg_name/msg_namelen are ignored * on connected socket. I was just happy when found this 8) --ANK" * * likewise MSDN says that lpFrom and lpFromlen are ignored for * connection-oriented sockets, so don't try to update lpFrom. */ if (wsa->addr && hdr.msg_namelen) ws_sockaddr_u2ws( &unix_sockaddr.addr, wsa->addr, wsa->addrlen.ptr ); return n; } /*********************************************************************** * WS2_async_recv (INTERNAL) * * Handler for overlapped recv() operations. */ static NTSTATUS WS2_async_recv( void *user, IO_STATUS_BLOCK *iosb, NTSTATUS status ) { struct ws2_async *wsa = user; int result = 0, fd; switch (status) { case STATUS_ALERTED: if ((status = wine_server_handle_to_fd( wsa->hSocket, FILE_READ_DATA, &fd, NULL ) )) break; result = WS2_recv( fd, wsa, convert_flags(wsa->flags) ); wine_server_release_fd( wsa->hSocket, fd ); if (result >= 0) { status = STATUS_SUCCESS; _enable_event( wsa->hSocket, FD_READ, 0, 0 ); } else { if (errno == EAGAIN) { status = STATUS_PENDING; _enable_event( wsa->hSocket, FD_READ, 0, 0 ); } else { result = 0; status = wsaErrStatus(); } } break; } if (status != STATUS_PENDING) { iosb->u.Status = status; iosb->Information = result; if (!wsa->completion_func) release_async_io( &wsa->io ); } return status; } /*********************************************************************** * WS2_send (INTERNAL) * * Workhorse for both synchronous and asynchronous send() operations. */ static int WS2_send( int fd, struct ws2_async *wsa, int flags ) { struct msghdr hdr; union generic_unix_sockaddr unix_addr; int n, ret; hdr.msg_name = NULL; hdr.msg_namelen = 0; if (wsa->addr) { hdr.msg_name = &unix_addr; hdr.msg_namelen = ws_sockaddr_ws2u( wsa->addr, wsa->addrlen.val, &unix_addr ); if ( !hdr.msg_namelen ) { errno = EFAULT; return -1; } #if defined(HAS_IPX) && defined(SOL_IPX) if(wsa->addr->sa_family == WS_AF_IPX) { struct sockaddr_ipx* uipx = (struct sockaddr_ipx*)hdr.msg_name; int val=0; socklen_t len = sizeof(int); /* The packet type is stored at the ipx socket level; At least the linux kernel seems * to do something with it in case hdr.msg_name is NULL. Nonetheless can we use it to store * the packet type and then we can retrieve it using getsockopt. After that we can set the * ipx type in the sockaddr_opx structure with the stored value. */ if(getsockopt(fd, SOL_IPX, IPX_TYPE, &val, &len) != -1) uipx->sipx_type = val; } #endif } hdr.msg_iov = wsa->iovec + wsa->first_iovec; hdr.msg_iovlen = wsa->n_iovecs - wsa->first_iovec; #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS hdr.msg_accrights = NULL; hdr.msg_accrightslen = 0; #else hdr.msg_control = NULL; hdr.msg_controllen = 0; hdr.msg_flags = 0; #endif while ((ret = sendmsg(fd, &hdr, flags)) == -1) { if (errno == EISCONN) { hdr.msg_name = 0; hdr.msg_namelen = 0; continue; } if (errno != EINTR) return -1; } n = ret; while (wsa->first_iovec < wsa->n_iovecs && wsa->iovec[wsa->first_iovec].iov_len <= n) n -= wsa->iovec[wsa->first_iovec++].iov_len; if (wsa->first_iovec < wsa->n_iovecs) { wsa->iovec[wsa->first_iovec].iov_base = (char*)wsa->iovec[wsa->first_iovec].iov_base + n; wsa->iovec[wsa->first_iovec].iov_len -= n; } return ret; } /*********************************************************************** * WS2_async_send (INTERNAL) * * Handler for overlapped send() operations. */ static NTSTATUS WS2_async_send( void *user, IO_STATUS_BLOCK *iosb, NTSTATUS status ) { struct ws2_async *wsa = user; int result = 0, fd; switch (status) { case STATUS_ALERTED: if ( wsa->n_iovecs <= wsa->first_iovec ) { /* Nothing to do */ status = STATUS_SUCCESS; break; } if ((status = wine_server_handle_to_fd( wsa->hSocket, FILE_WRITE_DATA, &fd, NULL ) )) break; /* check to see if the data is ready (non-blocking) */ result = WS2_send( fd, wsa, convert_flags(wsa->flags) ); wine_server_release_fd( wsa->hSocket, fd ); if (result >= 0) { if (wsa->first_iovec < wsa->n_iovecs) status = STATUS_PENDING; else status = STATUS_SUCCESS; iosb->Information += result; } else if (errno == EAGAIN) { status = STATUS_PENDING; } else { status = wsaErrStatus(); } break; } if (status != STATUS_PENDING) { iosb->u.Status = status; if (!wsa->completion_func) release_async_io( &wsa->io ); } return status; } /*********************************************************************** * WS2_async_shutdown (INTERNAL) * * Handler for shutdown() operations on overlapped sockets. */ static NTSTATUS WS2_async_shutdown( void *user, IO_STATUS_BLOCK *iosb, NTSTATUS status ) { struct ws2_async_shutdown *wsa = user; int fd, err = 1; switch (status) { case STATUS_ALERTED: if ((status = wine_server_handle_to_fd( wsa->hSocket, 0, &fd, NULL ) )) break; switch ( wsa->type ) { case ASYNC_TYPE_READ: err = shutdown( fd, 0 ); break; case ASYNC_TYPE_WRITE: err = shutdown( fd, 1 ); break; } status = err ? wsaErrStatus() : STATUS_SUCCESS; wine_server_release_fd( wsa->hSocket, fd ); break; } iosb->u.Status = status; iosb->Information = 0; release_async_io( &wsa->io ); return status; } /*********************************************************************** * WS2_register_async_shutdown (INTERNAL) * * Helper function for WS_shutdown() on overlapped sockets. */ static int WS2_register_async_shutdown( SOCKET s, int type ) { struct ws2_async_shutdown *wsa; NTSTATUS status; TRACE("socket %04lx type %d\n", s, type); wsa = (struct ws2_async_shutdown *)alloc_async_io( sizeof(*wsa), WS2_async_shutdown ); if ( !wsa ) return WSAEFAULT; wsa->hSocket = SOCKET2HANDLE(s); wsa->type = type; status = register_async( type, wsa->hSocket, &wsa->io, 0, NULL, NULL, &wsa->iosb ); if (status != STATUS_PENDING) { HeapFree( GetProcessHeap(), 0, wsa ); return NtStatusToWSAError( status ); } return 0; } /*********************************************************************** * accept (WS2_32.1) */ SOCKET WINAPI WS_accept( SOCKET s, struct WS_sockaddr *addr, int *len ) { IO_STATUS_BLOCK io; NTSTATUS status; obj_handle_t accept_handle; HANDLE sync_event; SOCKET ret; TRACE("%#lx\n", s); if (!(sync_event = CreateEventW( NULL, TRUE, FALSE, NULL ))) return INVALID_SOCKET; status = NtDeviceIoControlFile( SOCKET2HANDLE(s), (HANDLE)((ULONG_PTR)sync_event | 0), NULL, NULL, &io, IOCTL_AFD_ACCEPT, NULL, 0, &accept_handle, sizeof(accept_handle) ); if (status == STATUS_PENDING) { if (WaitForSingleObject( sync_event, INFINITE ) == WAIT_FAILED) { CloseHandle( sync_event ); return SOCKET_ERROR; } status = io.u.Status; } CloseHandle( sync_event ); if (status) { WARN("failed; status %#x\n", status); WSASetLastError( NtStatusToWSAError( status ) ); return INVALID_SOCKET; } ret = HANDLE2SOCKET(wine_server_ptr_handle( accept_handle )); if (!socket_list_add( ret )) { CloseHandle( SOCKET2HANDLE(ret) ); return INVALID_SOCKET; } if (addr && len && WS_getpeername( ret, addr, len )) { WS_closesocket( ret ); return INVALID_SOCKET; } TRACE("returning %#lx\n", ret); return ret; } /*********************************************************************** * AcceptEx */ static BOOL WINAPI WS2_AcceptEx( SOCKET listener, SOCKET acceptor, void *dest, DWORD recv_len, DWORD local_len, DWORD remote_len, DWORD *ret_len, OVERLAPPED *overlapped) { struct afd_accept_into_params params = { .accept_handle = acceptor, .recv_len = recv_len, .local_len = local_len, }; void *cvalue = NULL; NTSTATUS status; TRACE( "listener %#lx, acceptor %#lx, dest %p, recv_len %u, local_len %u, remote_len %u, ret_len %p, " "overlapped %p\n", listener, acceptor, dest, recv_len, local_len, remote_len, ret_len, overlapped ); if (!overlapped) { SetLastError(WSA_INVALID_PARAMETER); return FALSE; } if (!((ULONG_PTR)overlapped->hEvent & 1)) cvalue = overlapped; overlapped->Internal = STATUS_PENDING; overlapped->InternalHigh = 0; if (!dest) { SetLastError(WSAEINVAL); return FALSE; } if (!remote_len) { SetLastError(WSAEFAULT); return FALSE; } status = NtDeviceIoControlFile( SOCKET2HANDLE(listener), overlapped->hEvent, NULL, cvalue, (IO_STATUS_BLOCK *)overlapped, IOCTL_AFD_ACCEPT_INTO, ¶ms, sizeof(params), dest, recv_len + local_len + remote_len ); if (ret_len) *ret_len = overlapped->InternalHigh; WSASetLastError( NtStatusToWSAError(status) ); return !status; } /*********************************************************************** * WS2_ReadFile (INTERNAL) * * Perform an APC-safe ReadFile operation */ static NTSTATUS WS2_ReadFile(HANDLE hFile, PIO_STATUS_BLOCK io_status, char* buffer, ULONG length, PLARGE_INTEGER offset) { int result = -1, unix_handle; unsigned int options; NTSTATUS status; TRACE( "(%p,%p,0x%08x)\n", hFile, buffer,length ); status = wine_server_handle_to_fd( hFile, FILE_READ_DATA, &unix_handle, &options ); if (status) return status; while (result == -1) { if (offset->QuadPart != FILE_USE_FILE_POINTER_POSITION) result = pread( unix_handle, buffer, length, offset->QuadPart ); else result = read( unix_handle, buffer, length ); if (errno != EINTR) break; } if (!result) status = (length ? STATUS_END_OF_FILE : STATUS_SUCCESS); else if (result != -1) status = STATUS_SUCCESS; else if (errno != EAGAIN) status = wsaErrStatus(); else status = STATUS_PENDING; wine_server_release_fd( hFile, unix_handle ); TRACE("= 0x%08x (%d)\n", status, result); if (status == STATUS_SUCCESS || status == STATUS_END_OF_FILE) { io_status->u.Status = status; io_status->Information = result; } return status; } /*********************************************************************** * WS2_transmitfile_getbuffer (INTERNAL) * * Pick the appropriate buffer for a TransmitFile send operation. */ static NTSTATUS WS2_transmitfile_getbuffer( int fd, struct ws2_transmitfile_async *wsa ) { /* send any incomplete writes from a previous iteration */ if (wsa->write.first_iovec < wsa->write.n_iovecs) return STATUS_PENDING; /* process the header (if applicable) */ if (wsa->buffers.Head) { wsa->write.first_iovec = 0; wsa->write.n_iovecs = 1; wsa->write.iovec[0].iov_base = wsa->buffers.Head; wsa->write.iovec[0].iov_len = wsa->buffers.HeadLength; wsa->buffers.Head = NULL; return STATUS_PENDING; } /* process the main file */ if (wsa->file) { DWORD bytes_per_send = wsa->bytes_per_send; IO_STATUS_BLOCK iosb; NTSTATUS status; iosb.Information = 0; /* when the size of the transfer is limited ensure that we don't go past that limit */ if (wsa->file_bytes != 0) bytes_per_send = min(bytes_per_send, wsa->file_bytes - wsa->file_read); status = WS2_ReadFile( wsa->file, &iosb, wsa->buffer, bytes_per_send, &wsa->offset ); if (wsa->offset.QuadPart != FILE_USE_FILE_POINTER_POSITION) wsa->offset.QuadPart += iosb.Information; if (status == STATUS_END_OF_FILE) wsa->file = NULL; /* continue on to the footer */ else if (status != STATUS_SUCCESS) return status; else { if (iosb.Information) { wsa->write.first_iovec = 0; wsa->write.n_iovecs = 1; wsa->write.iovec[0].iov_base = wsa->buffer; wsa->write.iovec[0].iov_len = iosb.Information; wsa->file_read += iosb.Information; } if (wsa->file_bytes != 0 && wsa->file_read >= wsa->file_bytes) wsa->file = NULL; return STATUS_PENDING; } } /* send the footer (if applicable) */ if (wsa->buffers.Tail) { wsa->write.first_iovec = 0; wsa->write.n_iovecs = 1; wsa->write.iovec[0].iov_base = wsa->buffers.Tail; wsa->write.iovec[0].iov_len = wsa->buffers.TailLength; wsa->buffers.Tail = NULL; return STATUS_PENDING; } return STATUS_SUCCESS; } /*********************************************************************** * WS2_transmitfile_base (INTERNAL) * * Shared implementation for both synchronous and asynchronous TransmitFile. */ static NTSTATUS WS2_transmitfile_base( int fd, struct ws2_transmitfile_async *wsa ) { NTSTATUS status; status = WS2_transmitfile_getbuffer( fd, wsa ); if (status == STATUS_PENDING) { IO_STATUS_BLOCK *iosb = (IO_STATUS_BLOCK *)wsa->write.user_overlapped; int n; n = WS2_send( fd, &wsa->write, convert_flags(wsa->write.flags) ); if (n >= 0) { if (iosb) iosb->Information += n; } else if (errno != EAGAIN) return wsaErrStatus(); } return status; } /*********************************************************************** * WS2_async_transmitfile (INTERNAL) * * Asynchronous callback for overlapped TransmitFile operations. */ static NTSTATUS WS2_async_transmitfile( void *user, IO_STATUS_BLOCK *iosb, NTSTATUS status ) { struct ws2_transmitfile_async *wsa = user; int fd; if (status == STATUS_ALERTED) { if (!(status = wine_server_handle_to_fd( wsa->write.hSocket, FILE_WRITE_DATA, &fd, NULL ))) { status = WS2_transmitfile_base( fd, wsa ); wine_server_release_fd( wsa->write.hSocket, fd ); } if (status == STATUS_PENDING) return status; } iosb->u.Status = status; release_async_io( &wsa->io ); return status; } /*********************************************************************** * TransmitFile */ static BOOL WINAPI WS2_TransmitFile( SOCKET s, HANDLE h, DWORD file_bytes, DWORD bytes_per_send, LPOVERLAPPED overlapped, LPTRANSMIT_FILE_BUFFERS buffers, DWORD flags ) { union generic_unix_sockaddr uaddr; socklen_t uaddrlen = sizeof(uaddr); struct ws2_transmitfile_async *wsa; NTSTATUS status; int fd; TRACE("(%lx, %p, %d, %d, %p, %p, %d)\n", s, h, file_bytes, bytes_per_send, overlapped, buffers, flags ); fd = get_sock_fd( s, FILE_WRITE_DATA, NULL ); if (fd == -1) return FALSE; if (getpeername( fd, &uaddr.addr, &uaddrlen ) != 0) { release_sock_fd( s, fd ); WSASetLastError( WSAENOTCONN ); return FALSE; } if (flags) FIXME("Flags are not currently supported (0x%x).\n", flags); if (h && GetFileType( h ) != FILE_TYPE_DISK) { FIXME("Non-disk file handles are not currently supported.\n"); release_sock_fd( s, fd ); WSASetLastError( WSAEOPNOTSUPP ); return FALSE; } /* set reasonable defaults when requested */ if (!bytes_per_send) bytes_per_send = (1 << 16); /* Depends on OS version: PAGE_SIZE, 2*PAGE_SIZE, or 2^16 */ if (!(wsa = (struct ws2_transmitfile_async *)alloc_async_io( sizeof(*wsa) + bytes_per_send, WS2_async_transmitfile ))) { release_sock_fd( s, fd ); WSASetLastError( WSAEFAULT ); return FALSE; } if (buffers) wsa->buffers = *buffers; else memset(&wsa->buffers, 0x0, sizeof(wsa->buffers)); wsa->buffer = (char *)(wsa + 1); wsa->file = h; wsa->file_read = 0; wsa->file_bytes = file_bytes; wsa->bytes_per_send = bytes_per_send; wsa->flags = flags; wsa->offset.QuadPart = FILE_USE_FILE_POINTER_POSITION; wsa->write.hSocket = SOCKET2HANDLE(s); wsa->write.addr = NULL; wsa->write.addrlen.val = 0; wsa->write.flags = 0; wsa->write.lpFlags = &wsa->flags; wsa->write.control = NULL; wsa->write.n_iovecs = 0; wsa->write.first_iovec = 0; wsa->write.user_overlapped = overlapped; if (overlapped) { IO_STATUS_BLOCK *iosb = (IO_STATUS_BLOCK *)overlapped; int status; wsa->offset.u.LowPart = overlapped->u.s.Offset; wsa->offset.u.HighPart = overlapped->u.s.OffsetHigh; iosb->u.Status = STATUS_PENDING; iosb->Information = 0; status = register_async( ASYNC_TYPE_WRITE, SOCKET2HANDLE(s), &wsa->io, overlapped->hEvent, NULL, NULL, iosb ); if(status != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa ); release_sock_fd( s, fd ); WSASetLastError( NtStatusToWSAError(status) ); return FALSE; } do { status = WS2_transmitfile_base( fd, wsa ); if (status == STATUS_PENDING) { /* block here */ do_block(fd, POLLOUT, -1); _sync_sock_state(s); /* let wineserver notice connection */ } } while (status == STATUS_PENDING); release_sock_fd( s, fd ); if (status != STATUS_SUCCESS) WSASetLastError( NtStatusToWSAError(status) ); HeapFree( GetProcessHeap(), 0, wsa ); return (status == STATUS_SUCCESS); } /*********************************************************************** * GetAcceptExSockaddrs */ static void WINAPI WS2_GetAcceptExSockaddrs(PVOID buffer, DWORD data_size, DWORD local_size, DWORD remote_size, struct WS_sockaddr **local_addr, LPINT local_addr_len, struct WS_sockaddr **remote_addr, LPINT remote_addr_len) { char *cbuf = buffer; TRACE("(%p, %d, %d, %d, %p, %p, %p, %p)\n", buffer, data_size, local_size, remote_size, local_addr, local_addr_len, remote_addr, remote_addr_len ); cbuf += data_size; *local_addr_len = *(int *) cbuf; *local_addr = (struct WS_sockaddr *)(cbuf + sizeof(int)); cbuf += local_size; *remote_addr_len = *(int *) cbuf; *remote_addr = (struct WS_sockaddr *)(cbuf + sizeof(int)); } /*********************************************************************** * WSASendMsg */ int WINAPI WSASendMsg( SOCKET s, LPWSAMSG msg, DWORD dwFlags, LPDWORD lpNumberOfBytesSent, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine) { if (!msg) { SetLastError( WSAEFAULT ); return SOCKET_ERROR; } return WS2_sendto( s, msg->lpBuffers, msg->dwBufferCount, lpNumberOfBytesSent, dwFlags, msg->name, msg->namelen, lpOverlapped, lpCompletionRoutine ); } /*********************************************************************** * WSARecvMsg * * Perform a receive operation that is capable of returning message * control headers. It is important to note that the WSAMSG parameter * must remain valid throughout the operation, even when an overlapped * receive is performed. */ static int WINAPI WS2_WSARecvMsg( SOCKET s, LPWSAMSG msg, LPDWORD lpNumberOfBytesRecvd, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine ) { if (!msg) { SetLastError( WSAEFAULT ); return SOCKET_ERROR; } return WS2_recv_base( s, msg->lpBuffers, msg->dwBufferCount, lpNumberOfBytesRecvd, &msg->dwFlags, msg->name, &msg->namelen, lpOverlapped, lpCompletionRoutine, &msg->Control ); } /*********************************************************************** * interface_bind (INTERNAL) * * Take bind() calls on any name corresponding to a local network adapter and restrict the given socket to * operating only on the specified interface. This restriction consists of two components: * 1) An outgoing packet restriction suggesting the egress interface for all packets. * 2) An incoming packet restriction dropping packets not meant for the interface. * If the function succeeds in placing these restrictions (returns TRUE) then the name for the bind() may * safely be changed to INADDR_ANY, permitting the transmission and receipt of broadcast packets on the * socket. This behavior is only relevant to UDP sockets and is needed for applications that expect to be able * to receive broadcast packets on a socket that is bound to a specific network interface. */ static BOOL interface_bind( SOCKET s, int fd, struct sockaddr *addr ) { struct sockaddr_in *in_sock = (struct sockaddr_in *) addr; in_addr_t bind_addr = in_sock->sin_addr.s_addr; PIP_ADAPTER_INFO adapters = NULL, adapter; BOOL ret = FALSE; DWORD adap_size; int enable = 1; if (bind_addr == htonl(INADDR_ANY) || bind_addr == htonl(INADDR_LOOPBACK)) return FALSE; /* Not binding to a network adapter, special interface binding unnecessary. */ if (_get_fd_type(fd) != SOCK_DGRAM) return FALSE; /* Special interface binding is only necessary for UDP datagrams. */ if (GetAdaptersInfo(NULL, &adap_size) != ERROR_BUFFER_OVERFLOW) goto cleanup; adapters = HeapAlloc(GetProcessHeap(), 0, adap_size); if (adapters == NULL || GetAdaptersInfo(adapters, &adap_size) != NO_ERROR) goto cleanup; /* Search the IPv4 adapter list for the appropriate binding interface */ for (adapter = adapters; adapter != NULL; adapter = adapter->Next) { in_addr_t adapter_addr = (in_addr_t) inet_addr(adapter->IpAddressList.IpAddress.String); if (bind_addr == adapter_addr) { #if defined(IP_BOUND_IF) /* IP_BOUND_IF sets both the incoming and outgoing restriction at once */ if (setsockopt(fd, IPPROTO_IP, IP_BOUND_IF, &adapter->Index, sizeof(adapter->Index)) != 0) goto cleanup; ret = TRUE; #elif defined(LINUX_BOUND_IF) in_addr_t ifindex = (in_addr_t) htonl(adapter->Index); struct interface_filter specific_interface_filter; struct sock_fprog filter_prog; if (setsockopt(fd, IPPROTO_IP, IP_UNICAST_IF, &ifindex, sizeof(ifindex)) != 0) goto cleanup; /* Failed to suggest egress interface */ specific_interface_filter = generic_interface_filter; specific_interface_filter.iface_rule.k = adapter->Index; specific_interface_filter.ip_rule.k = htonl(adapter_addr); filter_prog.len = sizeof(generic_interface_filter)/sizeof(struct sock_filter); filter_prog.filter = (struct sock_filter *) &specific_interface_filter; if (setsockopt(fd, SOL_SOCKET, SO_ATTACH_FILTER, &filter_prog, sizeof(filter_prog)) != 0) goto cleanup; /* Failed to specify incoming packet filter */ ret = TRUE; #else FIXME("Broadcast packets on interface-bound sockets are not currently supported on this platform, " "receiving broadcast packets will not work on socket %04lx.\n", s); #endif if (ret) { EnterCriticalSection(&cs_if_addr_cache); if (if_addr_cache_size <= adapter->Index) { unsigned int new_size; in_addr_t *new; new_size = max(if_addr_cache_size * 2, adapter->Index + 1); if (!(new = heap_realloc(if_addr_cache, sizeof(*if_addr_cache) * new_size))) { ERR("No memory.\n"); ret = FALSE; LeaveCriticalSection(&cs_if_addr_cache); break; } memset(new + if_addr_cache_size, 0, sizeof(*if_addr_cache) * (new_size - if_addr_cache_size)); if_addr_cache = new; if_addr_cache_size = new_size; } if (if_addr_cache[adapter->Index] && if_addr_cache[adapter->Index] != adapter_addr) WARN("Adapter addr for iface index %u has changed.\n", adapter->Index); if_addr_cache[adapter->Index] = adapter_addr; LeaveCriticalSection(&cs_if_addr_cache); } break; } } /* Will soon be switching to INADDR_ANY: permit address reuse */ if (ret && setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &enable, sizeof(enable)) == 0) TRACE("Socket %04lx bound to interface index %d\n", s, adapter->Index); else ret = FALSE; cleanup: if(!ret) ERR("Failed to bind to interface, receiving broadcast packets will not work on socket %04lx.\n", s); HeapFree(GetProcessHeap(), 0, adapters); return ret; } /*********************************************************************** * bind (WS2_32.2) */ int WINAPI WS_bind(SOCKET s, const struct WS_sockaddr* name, int namelen) { int fd = get_sock_fd( s, 0, NULL ); int res = SOCKET_ERROR; TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen); if (fd != -1) { if (!name || (name->sa_family && !supported_pf(name->sa_family))) { SetLastError(WSAEAFNOSUPPORT); } else { union generic_unix_sockaddr uaddr; unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr); if (!uaddrlen) { SetLastError(WSAEFAULT); } else { if (name->sa_family == WS_AF_INET) { struct sockaddr_in *in4 = (struct sockaddr_in*) &uaddr; if (memcmp(&in4->sin_addr, magic_loopback_addr, 4) == 0) { /* Trying to bind to the default host interface, using * INADDR_ANY instead*/ WARN("Trying to bind to magic IP address, using " "INADDR_ANY instead.\n"); in4->sin_addr.s_addr = htonl(INADDR_ANY); } else if (interface_bind(s, fd, &uaddr.addr)) in4->sin_addr.s_addr = htonl(INADDR_ANY); } if (bind(fd, &uaddr.addr, uaddrlen) < 0) { int loc_errno = errno; WARN("\tfailure - errno = %i\n", errno); errno = loc_errno; switch (errno) { case EADDRNOTAVAIL: SetLastError(WSAEINVAL); break; case EADDRINUSE: { int optval = 0; socklen_t optlen = sizeof(optval); /* Windows >= 2003 will return different results depending on * SO_REUSEADDR, WSAEACCES may be returned representing that * the socket hijacking protection prevented the bind */ if (!getsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char *)&optval, &optlen) && optval) { SetLastError(WSAEACCES); break; } /* fall through */ } default: SetLastError(wsaErrno()); break; } } else { res=0; /* success */ } } } release_sock_fd( s, fd ); } return res; } /*********************************************************************** * closesocket (WS2_32.3) */ int WINAPI WS_closesocket(SOCKET s) { int res = SOCKET_ERROR, fd; if (num_startup) { fd = get_sock_fd(s, FILE_READ_DATA, NULL); if (fd >= 0) { release_sock_fd(s, fd); socket_list_remove(s); if (CloseHandle(SOCKET2HANDLE(s))) res = 0; } } else SetLastError(WSANOTINITIALISED); TRACE("(socket %04lx) -> %d\n", s, res); return res; } static int do_connect(int fd, const struct WS_sockaddr* name, int namelen) { union generic_unix_sockaddr uaddr; unsigned int uaddrlen = ws_sockaddr_ws2u(name, namelen, &uaddr); if (!uaddrlen) return WSAEFAULT; if (name->sa_family == WS_AF_INET) { struct sockaddr_in *in4 = (struct sockaddr_in*) &uaddr; if (memcmp(&in4->sin_addr, magic_loopback_addr, 4) == 0) { /* Trying to connect to magic replace-loopback address, * assuming we really want to connect to localhost */ TRACE("Trying to connect to magic IP address, using " "INADDR_LOOPBACK instead.\n"); in4->sin_addr.s_addr = htonl(INADDR_LOOPBACK); } } if (connect(fd, &uaddr.addr, uaddrlen) == 0) return 0; return wsaErrno(); } /*********************************************************************** * connect (WS2_32.4) */ int WINAPI WS_connect(SOCKET s, const struct WS_sockaddr* name, int namelen) { int fd = get_sock_fd( s, FILE_READ_DATA, NULL ); TRACE("socket %04lx, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen); if (fd != -1) { BOOL is_blocking; int ret = do_connect(fd, name, namelen); if (ret == 0) goto connect_success; if (ret == WSAEINPROGRESS) { /* tell wineserver that a connection is in progress */ _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE, FD_CONNECT, FD_WINE_CONNECTED|FD_WINE_LISTENING); ret = sock_is_blocking( s, &is_blocking ); if (!ret) { if (is_blocking) { do_block(fd, POLLIN | POLLOUT, -1); _sync_sock_state(s); /* let wineserver notice connection */ /* retrieve any error codes from it */ if (!(ret = get_sock_error(s, FD_CONNECT_BIT))) goto connect_success; } else ret = WSAEWOULDBLOCK; } } release_sock_fd( s, fd ); SetLastError(ret); } return SOCKET_ERROR; connect_success: release_sock_fd( s, fd ); _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE, FD_WINE_CONNECTED|FD_READ|FD_WRITE, FD_CONNECT|FD_WINE_LISTENING); TRACE("\tconnected %04lx\n", s); return 0; } /*********************************************************************** * WSAConnect (WS2_32.30) */ int WINAPI WSAConnect( SOCKET s, const struct WS_sockaddr* name, int namelen, LPWSABUF lpCallerData, LPWSABUF lpCalleeData, LPQOS lpSQOS, LPQOS lpGQOS ) { if ( lpCallerData || lpCalleeData || lpSQOS || lpGQOS ) FIXME("unsupported parameters!\n"); return WS_connect( s, name, namelen ); } /*********************************************************************** * ConnectEx */ static BOOL WINAPI WS2_ConnectEx(SOCKET s, const struct WS_sockaddr* name, int namelen, PVOID sendBuf, DWORD sendBufLen, LPDWORD sent, LPOVERLAPPED ov) { int fd, ret, status; if (!ov) { SetLastError( ERROR_INVALID_PARAMETER ); return FALSE; } fd = get_sock_fd( s, FILE_READ_DATA, NULL ); if (fd == -1) return FALSE; TRACE("socket %04lx, ptr %p %s, length %d, sendptr %p, len %d, ov %p\n", s, name, debugstr_sockaddr(name), namelen, sendBuf, sendBufLen, ov); ret = is_fd_bound(fd, NULL, NULL); if (ret <= 0) { SetLastError(ret == -1 ? wsaErrno() : WSAEINVAL); release_sock_fd( s, fd ); return FALSE; } ret = do_connect(fd, name, namelen); if (ret == 0) { WSABUF wsabuf; _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE, FD_WINE_CONNECTED|FD_READ|FD_WRITE, FD_CONNECT|FD_WINE_LISTENING); wsabuf.len = sendBufLen; wsabuf.buf = (char*) sendBuf; /* WSASend takes care of completion if need be */ if (WSASend(s, &wsabuf, sendBuf ? 1 : 0, sent, 0, ov, NULL) != SOCKET_ERROR) goto connection_success; } else if (ret == WSAEINPROGRESS) { struct ws2_async *wsa; ULONG_PTR cvalue = (((ULONG_PTR)ov->hEvent & 1) == 0) ? (ULONG_PTR)ov : 0; _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE, FD_CONNECT, FD_WINE_CONNECTED|FD_WINE_LISTENING); /* Indirectly call WSASend */ if (!(wsa = (struct ws2_async *)alloc_async_io( offsetof( struct ws2_async, iovec[1] ), WS2_async_send ))) { SetLastError(WSAEFAULT); } else { IO_STATUS_BLOCK *iosb = (IO_STATUS_BLOCK *)ov; iosb->u.Status = STATUS_PENDING; iosb->Information = 0; wsa->hSocket = SOCKET2HANDLE(s); wsa->addr = NULL; wsa->addrlen.val = 0; wsa->flags = 0; wsa->lpFlags = &wsa->flags; wsa->control = NULL; wsa->n_iovecs = sendBuf ? 1 : 0; wsa->first_iovec = 0; wsa->completion_func = NULL; wsa->iovec[0].iov_base = sendBuf; wsa->iovec[0].iov_len = sendBufLen; status = register_async( ASYNC_TYPE_WRITE, wsa->hSocket, &wsa->io, ov->hEvent, NULL, (void *)cvalue, iosb ); if (status != STATUS_PENDING) HeapFree(GetProcessHeap(), 0, wsa); /* If the connect already failed */ if (status == STATUS_PIPE_DISCONNECTED) { ov->Internal = sock_error_to_ntstatus( get_sock_error( s, FD_CONNECT_BIT )); ov->InternalHigh = 0; if (cvalue) WS_AddCompletion( s, cvalue, ov->Internal, ov->InternalHigh, TRUE ); if (ov->hEvent) NtSetEvent( ov->hEvent, NULL ); status = STATUS_PENDING; } SetLastError( NtStatusToWSAError(status) ); } } else { SetLastError(ret); } release_sock_fd( s, fd ); return FALSE; connection_success: release_sock_fd( s, fd ); return TRUE; } /*********************************************************************** * DisconnectEx */ static BOOL WINAPI WS2_DisconnectEx( SOCKET s, LPOVERLAPPED ov, DWORD flags, DWORD reserved ) { TRACE( "socket %04lx, ov %p, flags 0x%x, reserved 0x%x\n", s, ov, flags, reserved ); if (flags & TF_REUSE_SOCKET) FIXME( "Reusing socket not supported yet\n" ); if (ov) { ov->Internal = STATUS_PENDING; ov->InternalHigh = 0; } return !WS_shutdown( s, SD_BOTH ); } /*********************************************************************** * getpeername (WS2_32.5) */ int WINAPI WS_getpeername(SOCKET s, struct WS_sockaddr *name, int *namelen) { int fd; int res; TRACE("socket %04lx, ptr %p, len %08x\n", s, name, namelen ? *namelen : 0); fd = get_sock_fd( s, 0, NULL ); res = SOCKET_ERROR; if (fd != -1) { union generic_unix_sockaddr uaddr; socklen_t uaddrlen = sizeof(uaddr); if (getpeername(fd, &uaddr.addr, &uaddrlen) == 0) { if (!name || !namelen) SetLastError(WSAEFAULT); else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0) /* The buffer was too small */ SetLastError(WSAEFAULT); else { res = 0; TRACE("=> %s\n", debugstr_sockaddr(name)); } } else SetLastError(wsaErrno()); release_sock_fd( s, fd ); } return res; } /* When binding to an UDP address with filter support the getsockname call on the socket * will always return 0.0.0.0 instead of the filtered interface address. This function * checks if the socket is interface-bound on UDP and return the correct address. * This is required because applications often do a bind() with port zero followed by a * getsockname() to retrieve the port and address acquired. */ static void interface_bind_check(int fd, struct sockaddr_in *addr) { #if !defined(IP_BOUND_IF) && !defined(LINUX_BOUND_IF) return; #else unsigned int ifindex; int ret; socklen_t len; /* Check for IPv4, address 0.0.0.0 and UDP socket */ if (addr->sin_family != AF_INET || addr->sin_addr.s_addr != 0) return; if (_get_fd_type(fd) != SOCK_DGRAM) return; len = sizeof(ifindex); #if defined(IP_BOUND_IF) ret = getsockopt(fd, IPPROTO_IP, IP_BOUND_IF, &ifindex, &len); #elif defined(LINUX_BOUND_IF) ret = getsockopt(fd, IPPROTO_IP, IP_UNICAST_IF, &ifindex, &len); if (!ret) ifindex = ntohl(ifindex); #endif if (!ret && ifindex) { EnterCriticalSection(&cs_if_addr_cache); if (ifindex < if_addr_cache_size) addr->sin_addr.s_addr = if_addr_cache[ifindex]; else ERR("No cache entry for ifindex %u.\n", ifindex); LeaveCriticalSection(&cs_if_addr_cache); } #endif } /*********************************************************************** * getsockname (WS2_32.6) */ int WINAPI WS_getsockname(SOCKET s, struct WS_sockaddr *name, int *namelen) { int fd; int res; TRACE("socket %04lx, ptr %p, len %08x\n", s, name, namelen ? *namelen : 0); /* Check if what we've received is valid. Should we use IsBadReadPtr? */ if( (name == NULL) || (namelen == NULL) ) { SetLastError( WSAEFAULT ); return SOCKET_ERROR; } fd = get_sock_fd( s, 0, NULL ); res = SOCKET_ERROR; if (fd != -1) { union generic_unix_sockaddr uaddr; socklen_t uaddrlen; int bound = is_fd_bound(fd, &uaddr, &uaddrlen); if (bound <= 0) { SetLastError(bound == -1 ? wsaErrno() : WSAEINVAL); } else if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0) { /* The buffer was too small */ SetLastError(WSAEFAULT); } else { interface_bind_check(fd, (struct sockaddr_in*) &uaddr); if (ws_sockaddr_u2ws(&uaddr.addr, name, namelen) != 0) { /* The buffer was too small */ SetLastError(WSAEFAULT); } else { res = 0; TRACE("=> %s\n", debugstr_sockaddr(name)); } } release_sock_fd( s, fd ); } return res; } /*********************************************************************** * getsockopt (WS2_32.7) */ INT WINAPI WS_getsockopt(SOCKET s, INT level, INT optname, char *optval, INT *optlen) { int fd; INT ret = 0; TRACE("(socket %04lx, %s, optval %s, optlen %p (%d))\n", s, debugstr_sockopt(level, optname), debugstr_optval(optval, 0), optlen, optlen ? *optlen : 0); switch(level) { case WS_SOL_SOCKET: { switch(optname) { /* Handle common cases. The special cases are below, sorted * alphabetically */ case WS_SO_BROADCAST: case WS_SO_DEBUG: case WS_SO_KEEPALIVE: case WS_SO_OOBINLINE: case WS_SO_RCVBUF: case WS_SO_REUSEADDR: case WS_SO_SNDBUF: if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; convert_sockopt(&level, &optname); if (getsockopt(fd, level, optname, optval, (socklen_t *)optlen) != 0 ) { SetLastError(wsaErrno()); ret = SOCKET_ERROR; } release_sock_fd( s, fd ); return ret; case WS_SO_ACCEPTCONN: if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; if (getsockopt(fd, SOL_SOCKET, SO_ACCEPTCONN, optval, (socklen_t *)optlen) != 0 ) { SetLastError(wsaErrno()); ret = SOCKET_ERROR; } else { /* BSD returns != 0 while Windows return exact == 1 */ if (*(int *)optval) *(int *)optval = 1; } release_sock_fd( s, fd ); return ret; case WS_SO_BSP_STATE: { int req_size, addr_size; WSAPROTOCOL_INFOW infow; CSADDR_INFO *csinfo; ret = ws_protocol_info(s, TRUE, &infow, &addr_size); if (ret) { if (infow.iAddressFamily == WS_AF_INET) addr_size = sizeof(struct sockaddr_in); else if (infow.iAddressFamily == WS_AF_INET6) addr_size = sizeof(struct sockaddr_in6); else { FIXME("Family %d is unsupported for SO_BSP_STATE\n", infow.iAddressFamily); SetLastError(WSAEAFNOSUPPORT); return SOCKET_ERROR; } req_size = sizeof(CSADDR_INFO) + addr_size * 2; if (*optlen < req_size) { ret = 0; SetLastError(WSAEFAULT); } else { union generic_unix_sockaddr uaddr; socklen_t uaddrlen; if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; csinfo = (CSADDR_INFO*) optval; /* Check if the sock is bound */ if (is_fd_bound(fd, &uaddr, &uaddrlen) == 1) { csinfo->LocalAddr.lpSockaddr = (LPSOCKADDR) (optval + sizeof(CSADDR_INFO)); ws_sockaddr_u2ws(&uaddr.addr, csinfo->LocalAddr.lpSockaddr, &addr_size); csinfo->LocalAddr.iSockaddrLength = addr_size; } else { csinfo->LocalAddr.lpSockaddr = NULL; csinfo->LocalAddr.iSockaddrLength = 0; } /* Check if the sock is connected */ if (!getpeername(fd, &uaddr.addr, &uaddrlen) && is_sockaddr_bound(&uaddr.addr, uaddrlen)) { csinfo->RemoteAddr.lpSockaddr = (LPSOCKADDR) (optval + sizeof(CSADDR_INFO) + addr_size); ws_sockaddr_u2ws(&uaddr.addr, csinfo->RemoteAddr.lpSockaddr, &addr_size); csinfo->RemoteAddr.iSockaddrLength = addr_size; } else { csinfo->RemoteAddr.lpSockaddr = NULL; csinfo->RemoteAddr.iSockaddrLength = 0; } csinfo->iSocketType = infow.iSocketType; csinfo->iProtocol = infow.iProtocol; release_sock_fd( s, fd ); } } return ret ? 0 : SOCKET_ERROR; } case WS_SO_DONTLINGER: { struct linger lingval; socklen_t len = sizeof(struct linger); if (!optlen || *optlen < sizeof(BOOL)|| !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0 ) { SetLastError(wsaErrno()); ret = SOCKET_ERROR; } else { *(BOOL *)optval = !lingval.l_onoff; *optlen = sizeof(BOOL); } release_sock_fd( s, fd ); return ret; } case WS_SO_CONNECT_TIME: { static int pretendtime = 0; struct WS_sockaddr addr; int len = sizeof(addr); if (!optlen || *optlen < sizeof(DWORD) || !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if (WS_getpeername(s, &addr, &len) == SOCKET_ERROR) *(DWORD *)optval = ~0u; else { if (!pretendtime) FIXME("WS_SO_CONNECT_TIME - faking results\n"); *(DWORD *)optval = pretendtime++; } *optlen = sizeof(DWORD); return ret; } /* As mentioned in setsockopt, Windows ignores this, so we * always return true here */ case WS_SO_DONTROUTE: if (!optlen || *optlen < sizeof(BOOL) || !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } *(BOOL *)optval = TRUE; *optlen = sizeof(BOOL); return 0; case WS_SO_ERROR: { if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; if (getsockopt(fd, SOL_SOCKET, SO_ERROR, optval, (socklen_t *)optlen) != 0 ) { SetLastError(wsaErrno()); ret = SOCKET_ERROR; } release_sock_fd( s, fd ); /* The wineserver may have swallowed the error before us */ if (!ret && *(int*) optval == 0) { int i, events[FD_MAX_EVENTS]; _get_sock_errors(s, events); for (i = 0; i < FD_MAX_EVENTS; i++) { if(events[i]) { TRACE("returning SO_ERROR %d from wine server\n", events[i]); *(int*) optval = events[i]; break; } } } return ret; } case WS_SO_LINGER: { struct linger lingval; socklen_t len = sizeof(struct linger); /* struct linger and LINGER have different sizes */ if (!optlen || *optlen < sizeof(LINGER) || !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; if (_get_fd_type(fd) == SOCK_DGRAM) { SetLastError(WSAENOPROTOOPT); ret = SOCKET_ERROR; } else if (getsockopt(fd, SOL_SOCKET, SO_LINGER, &lingval, &len) != 0) { SetLastError(wsaErrno()); ret = SOCKET_ERROR; } else { ((LINGER *)optval)->l_onoff = lingval.l_onoff; ((LINGER *)optval)->l_linger = lingval.l_linger; *optlen = sizeof(struct linger); } release_sock_fd( s, fd ); return ret; } case WS_SO_MAX_MSG_SIZE: if (!optlen || *optlen < sizeof(int) || !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } TRACE("getting global SO_MAX_MSG_SIZE = 65507\n"); *(int *)optval = 65507; *optlen = sizeof(int); return 0; /* SO_OPENTYPE does not require a valid socket handle. */ case WS_SO_OPENTYPE: if (!optlen || *optlen < sizeof(int) || !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } *(int *)optval = get_per_thread_data()->opentype; *optlen = sizeof(int); TRACE("getting global SO_OPENTYPE = 0x%x\n", *((int*)optval) ); return 0; case WS_SO_PROTOCOL_INFOA: case WS_SO_PROTOCOL_INFOW: { int size; WSAPROTOCOL_INFOW infow; ret = ws_protocol_info(s, optname == WS_SO_PROTOCOL_INFOW, &infow, &size); if (ret) { if (!optlen || !optval || *optlen < size) { if(optlen) *optlen = size; ret = 0; SetLastError(WSAEFAULT); } else memcpy(optval, &infow, size); } return ret ? 0 : SOCKET_ERROR; } case WS_SO_RCVTIMEO: case WS_SO_SNDTIMEO: { INT64 timeout; if (!optlen || *optlen < sizeof(int)|| !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; timeout = get_rcvsnd_timeo(fd, optname == WS_SO_RCVTIMEO); *(int *)optval = timeout <= UINT_MAX ? timeout : UINT_MAX; release_sock_fd( s, fd ); return ret; } case WS_SO_TYPE: { int sock_type; if (!optlen || *optlen < sizeof(int) || !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; sock_type = _get_fd_type(fd); if (sock_type == -1) { SetLastError(wsaErrno()); ret = SOCKET_ERROR; } else (*(int *)optval) = convert_socktype_u2w(sock_type); release_sock_fd( s, fd ); return ret; } default: TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname); SetLastError(WSAENOPROTOOPT); return SOCKET_ERROR; } /* end switch(optname) */ }/* end case WS_SOL_SOCKET */ #ifdef HAS_IPX case WS_NSPROTO_IPX: { struct WS_sockaddr_ipx addr; IPX_ADDRESS_DATA *data; int namelen; switch(optname) { case WS_IPX_PTYPE: if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; #ifdef SOL_IPX if(getsockopt(fd, SOL_IPX, IPX_TYPE, optval, (socklen_t *)optlen) == -1) { ret = SOCKET_ERROR; } #else { struct ipx val; socklen_t len=sizeof(struct ipx); if(getsockopt(fd, 0, SO_DEFAULT_HEADERS, &val, &len) == -1 ) ret = SOCKET_ERROR; else *optval = (int)val.ipx_pt; } #endif TRACE("ptype: %d (fd: %d)\n", *(int*)optval, fd); release_sock_fd( s, fd ); return ret; case WS_IPX_ADDRESS: /* * On a Win2000 system with one network card there are usually * three ipx devices one with a speed of 28.8kbps, 10Mbps and 100Mbps. * Using this call you can then retrieve info about this all. * In case of Linux it is a bit different. Usually you have * only "one" device active and further it is not possible to * query things like the linkspeed. */ FIXME("IPX_ADDRESS\n"); namelen = sizeof(struct WS_sockaddr_ipx); memset(&addr, 0, sizeof(struct WS_sockaddr_ipx)); WS_getsockname(s, (struct WS_sockaddr*)&addr, &namelen); data = (IPX_ADDRESS_DATA*)optval; memcpy(data->nodenum,addr.sa_nodenum,sizeof(data->nodenum)); memcpy(data->netnum,addr.sa_netnum,sizeof(data->netnum)); data->adapternum = 0; data->wan = FALSE; /* We are not on a wan for now .. */ data->status = FALSE; /* Since we are not on a wan, the wan link isn't up */ data->maxpkt = 1467; /* This value is the default one, at least on Win2k/WinXP */ data->linkspeed = 100000; /* Set the line speed in 100bit/s to 10 Mbit; * note 1MB = 1000kB in this case */ return 0; case WS_IPX_MAX_ADAPTER_NUM: FIXME("IPX_MAX_ADAPTER_NUM\n"); *(int*)optval = 1; /* As noted under IPX_ADDRESS we have just one card. */ return 0; default: FIXME("IPX optname:%x\n", optname); return SOCKET_ERROR; }/* end switch(optname) */ } /* end case WS_NSPROTO_IPX */ #endif #ifdef HAS_IRDA #define MAX_IRDA_DEVICES 10 case WS_SOL_IRLMP: switch(optname) { case WS_IRLMP_ENUMDEVICES: { char buf[sizeof(struct irda_device_list) + (MAX_IRDA_DEVICES - 1) * sizeof(struct irda_device_info)]; int res; socklen_t len = sizeof(buf); if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; res = getsockopt( fd, SOL_IRLMP, IRLMP_ENUMDEVICES, buf, &len ); release_sock_fd( s, fd ); if (res < 0) { SetLastError(wsaErrno()); return SOCKET_ERROR; } else { struct irda_device_list *src = (struct irda_device_list *)buf; DEVICELIST *dst = (DEVICELIST *)optval; INT needed = sizeof(DEVICELIST); unsigned int i; if (src->len > 0) needed += (src->len - 1) * sizeof(IRDA_DEVICE_INFO); if (*optlen < needed) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } *optlen = needed; TRACE("IRLMP_ENUMDEVICES: %d devices found:\n", src->len); dst->numDevice = src->len; for (i = 0; i < src->len; i++) { TRACE("saddr = %08x, daddr = %08x, info = %s, hints = %02x%02x\n", src->dev[i].saddr, src->dev[i].daddr, src->dev[i].info, src->dev[i].hints[0], src->dev[i].hints[1]); memcpy( dst->Device[i].irdaDeviceID, &src->dev[i].daddr, sizeof(dst->Device[i].irdaDeviceID) ) ; memcpy( dst->Device[i].irdaDeviceName, src->dev[i].info, sizeof(dst->Device[i].irdaDeviceName) ) ; memcpy( &dst->Device[i].irdaDeviceHints1, &src->dev[i].hints[0], sizeof(dst->Device[i].irdaDeviceHints1) ) ; memcpy( &dst->Device[i].irdaDeviceHints2, &src->dev[i].hints[1], sizeof(dst->Device[i].irdaDeviceHints2) ) ; dst->Device[i].irdaCharSet = src->dev[i].charset; } return 0; } } default: FIXME("IrDA optname:0x%x\n", optname); return SOCKET_ERROR; } break; /* case WS_SOL_IRLMP */ #undef MAX_IRDA_DEVICES #endif /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */ case WS_IPPROTO_TCP: switch(optname) { case WS_TCP_NODELAY: if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; convert_sockopt(&level, &optname); if (getsockopt(fd, level, optname, optval, (socklen_t *)optlen) != 0 ) { SetLastError(wsaErrno()); ret = SOCKET_ERROR; } release_sock_fd( s, fd ); return ret; } FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname); return SOCKET_ERROR; case WS_IPPROTO_IP: switch(optname) { case WS_IP_ADD_MEMBERSHIP: case WS_IP_DROP_MEMBERSHIP: #ifdef IP_HDRINCL case WS_IP_HDRINCL: #endif case WS_IP_MULTICAST_IF: case WS_IP_MULTICAST_LOOP: case WS_IP_MULTICAST_TTL: case WS_IP_OPTIONS: #if defined(IP_PKTINFO) || defined(IP_RECVDSTADDR) case WS_IP_PKTINFO: #endif case WS_IP_TOS: case WS_IP_TTL: #ifdef IP_UNICAST_IF case WS_IP_UNICAST_IF: #endif if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; convert_sockopt(&level, &optname); if (getsockopt(fd, level, optname, optval, (socklen_t *)optlen) != 0 ) { SetLastError(wsaErrno()); ret = SOCKET_ERROR; } release_sock_fd( s, fd ); return ret; case WS_IP_DONTFRAGMENT: return get_dont_fragment(s, IPPROTO_IP, (BOOL *)optval) ? 0 : SOCKET_ERROR; } FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname); return SOCKET_ERROR; case WS_IPPROTO_IPV6: switch(optname) { #ifdef IPV6_ADD_MEMBERSHIP case WS_IPV6_ADD_MEMBERSHIP: #endif #ifdef IPV6_DROP_MEMBERSHIP case WS_IPV6_DROP_MEMBERSHIP: #endif case WS_IPV6_MULTICAST_IF: case WS_IPV6_MULTICAST_HOPS: case WS_IPV6_MULTICAST_LOOP: case WS_IPV6_UNICAST_HOPS: case WS_IPV6_V6ONLY: #ifdef IPV6_UNICAST_IF case WS_IPV6_UNICAST_IF: #endif if ( (fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; convert_sockopt(&level, &optname); if (getsockopt(fd, level, optname, optval, (socklen_t *)optlen) != 0 ) { SetLastError(wsaErrno()); ret = SOCKET_ERROR; } release_sock_fd( s, fd ); return ret; case WS_IPV6_DONTFRAG: return get_dont_fragment(s, IPPROTO_IPV6, (BOOL *)optval) ? 0 : SOCKET_ERROR; } FIXME("Unknown IPPROTO_IPV6 optname 0x%08x\n", optname); return SOCKET_ERROR; default: WARN("Unknown level: 0x%08x\n", level); SetLastError(WSAEINVAL); return SOCKET_ERROR; } /* end switch(level) */ } /*********************************************************************** * htonl (WS2_32.8) */ WS_u_long WINAPI WS_htonl(WS_u_long hostlong) { return htonl(hostlong); } /*********************************************************************** * htons (WS2_32.9) */ WS_u_short WINAPI WS_htons(WS_u_short hostshort) { return htons(hostshort); } /*********************************************************************** * WSAHtonl (WS2_32.46) * From MSDN description of error codes, this function should also * check if WinSock has been initialized and the socket is a valid * socket. But why? This function only translates a host byte order * u_long into a network byte order u_long... */ int WINAPI WSAHtonl(SOCKET s, WS_u_long hostlong, WS_u_long *lpnetlong) { if (lpnetlong) { *lpnetlong = htonl(hostlong); return 0; } SetLastError(WSAEFAULT); return SOCKET_ERROR; } /*********************************************************************** * WSAHtons (WS2_32.47) * From MSDN description of error codes, this function should also * check if WinSock has been initialized and the socket is a valid * socket. But why? This function only translates a host byte order * u_short into a network byte order u_short... */ int WINAPI WSAHtons(SOCKET s, WS_u_short hostshort, WS_u_short *lpnetshort) { if (lpnetshort) { *lpnetshort = htons(hostshort); return 0; } SetLastError(WSAEFAULT); return SOCKET_ERROR; } /*********************************************************************** * inet_addr (WS2_32.11) */ WS_u_long WINAPI WS_inet_addr(const char *cp) { if (!cp) return INADDR_NONE; return inet_addr(cp); } /*********************************************************************** * ntohl (WS2_32.14) */ WS_u_long WINAPI WS_ntohl(WS_u_long netlong) { return ntohl(netlong); } /*********************************************************************** * ntohs (WS2_32.15) */ WS_u_short WINAPI WS_ntohs(WS_u_short netshort) { return ntohs(netshort); } /*********************************************************************** * inet_ntoa (WS2_32.12) */ char* WINAPI WS_inet_ntoa(struct WS_in_addr in) { unsigned int long_ip = ntohl(in.WS_s_addr); struct per_thread_data *data = get_per_thread_data(); sprintf( data->ntoa_buffer, "%u.%u.%u.%u", (long_ip >> 24) & 0xff, (long_ip >> 16) & 0xff, (long_ip >> 8) & 0xff, long_ip & 0xff); return data->ntoa_buffer; } static const char *debugstr_wsaioctl(DWORD code) { const char *name = NULL, *buf_type, *family; #define IOCTL_NAME(x) case x: name = #x; break switch (code) { IOCTL_NAME(WS_FIONBIO); IOCTL_NAME(WS_FIONREAD); IOCTL_NAME(WS_SIOCATMARK); /* IOCTL_NAME(WS_SIO_ACQUIRE_PORT_RESERVATION); */ IOCTL_NAME(WS_SIO_ADDRESS_LIST_CHANGE); IOCTL_NAME(WS_SIO_ADDRESS_LIST_QUERY); IOCTL_NAME(WS_SIO_ASSOCIATE_HANDLE); /* IOCTL_NAME(WS_SIO_ASSOCIATE_PORT_RESERVATION); IOCTL_NAME(WS_SIO_BASE_HANDLE); IOCTL_NAME(WS_SIO_BSP_HANDLE); IOCTL_NAME(WS_SIO_BSP_HANDLE_SELECT); IOCTL_NAME(WS_SIO_BSP_HANDLE_POLL); IOCTL_NAME(WS_SIO_CHK_QOS); */ IOCTL_NAME(WS_SIO_ENABLE_CIRCULAR_QUEUEING); IOCTL_NAME(WS_SIO_FIND_ROUTE); IOCTL_NAME(WS_SIO_FLUSH); IOCTL_NAME(WS_SIO_GET_BROADCAST_ADDRESS); IOCTL_NAME(WS_SIO_GET_EXTENSION_FUNCTION_POINTER); IOCTL_NAME(WS_SIO_GET_GROUP_QOS); IOCTL_NAME(WS_SIO_GET_INTERFACE_LIST); /* IOCTL_NAME(WS_SIO_GET_INTERFACE_LIST_EX); */ IOCTL_NAME(WS_SIO_GET_QOS); IOCTL_NAME(WS_SIO_IDEAL_SEND_BACKLOG_CHANGE); IOCTL_NAME(WS_SIO_IDEAL_SEND_BACKLOG_QUERY); IOCTL_NAME(WS_SIO_KEEPALIVE_VALS); IOCTL_NAME(WS_SIO_MULTIPOINT_LOOPBACK); IOCTL_NAME(WS_SIO_MULTICAST_SCOPE); /* IOCTL_NAME(WS_SIO_QUERY_RSS_SCALABILITY_INFO); IOCTL_NAME(WS_SIO_QUERY_WFP_ALE_ENDPOINT_HANDLE); */ IOCTL_NAME(WS_SIO_RCVALL); IOCTL_NAME(WS_SIO_RCVALL_IGMPMCAST); IOCTL_NAME(WS_SIO_RCVALL_MCAST); /* IOCTL_NAME(WS_SIO_RELEASE_PORT_RESERVATION); */ IOCTL_NAME(WS_SIO_ROUTING_INTERFACE_CHANGE); IOCTL_NAME(WS_SIO_ROUTING_INTERFACE_QUERY); IOCTL_NAME(WS_SIO_SET_COMPATIBILITY_MODE); IOCTL_NAME(WS_SIO_SET_GROUP_QOS); IOCTL_NAME(WS_SIO_SET_QOS); IOCTL_NAME(WS_SIO_TRANSLATE_HANDLE); IOCTL_NAME(WS_SIO_UDP_CONNRESET); } #undef IOCTL_NAME if (name) return name + 3; /* If this is not a known code split its bits */ switch(code & 0x18000000) { case WS_IOC_WS2: family = "IOC_WS2"; break; case WS_IOC_PROTOCOL: family = "IOC_PROTOCOL"; break; case WS_IOC_VENDOR: family = "IOC_VENDOR"; break; default: /* WS_IOC_UNIX */ { BYTE size = (code >> 16) & WS_IOCPARM_MASK; char x = (code & 0xff00) >> 8; BYTE y = code & 0xff; char args[14]; switch (code & (WS_IOC_VOID|WS_IOC_INOUT)) { case WS_IOC_VOID: buf_type = "_IO"; sprintf(args, "%d, %d", x, y); break; case WS_IOC_IN: buf_type = "_IOW"; sprintf(args, "'%c', %d, %d", x, y, size); break; case WS_IOC_OUT: buf_type = "_IOR"; sprintf(args, "'%c', %d, %d", x, y, size); break; default: buf_type = "?"; sprintf(args, "'%c', %d, %d", x, y, size); break; } return wine_dbg_sprintf("%s(%s)", buf_type, args); } } /* We are different from WS_IOC_UNIX. */ switch (code & (WS_IOC_VOID|WS_IOC_INOUT)) { case WS_IOC_VOID: buf_type = "_WSAIO"; break; case WS_IOC_INOUT: buf_type = "_WSAIORW"; break; case WS_IOC_IN: buf_type = "_WSAIOW"; break; case WS_IOC_OUT: buf_type = "_WSAIOR"; break; default: buf_type = "?"; break; } return wine_dbg_sprintf("%s(%s, %d)", buf_type, family, (USHORT)(code & 0xffff)); } /* do an ioctl call through the server */ static DWORD server_ioctl_sock( SOCKET s, DWORD code, LPVOID in_buff, DWORD in_size, LPVOID out_buff, DWORD out_size, LPDWORD ret_size, LPWSAOVERLAPPED overlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE completion ) { HANDLE event = overlapped ? overlapped->hEvent : 0; HANDLE handle = SOCKET2HANDLE( s ); struct ws2_async *wsa = NULL; IO_STATUS_BLOCK *io = (PIO_STATUS_BLOCK)overlapped, iosb; void *cvalue = NULL; NTSTATUS status; if (completion) { if (!(wsa = (struct ws2_async *)alloc_async_io( sizeof(*wsa), NULL ))) return WSA_NOT_ENOUGH_MEMORY; wsa->hSocket = handle; wsa->user_overlapped = overlapped; wsa->completion_func = completion; if (!io) io = &wsa->local_iosb; cvalue = wsa; } else if (!io) io = &iosb; else if (!((ULONG_PTR)overlapped->hEvent & 1)) cvalue = overlapped; status = NtDeviceIoControlFile( handle, event, wsa ? ws2_async_apc : NULL, cvalue, io, code, in_buff, in_size, out_buff, out_size ); if (status == STATUS_NOT_SUPPORTED) { FIXME("Unsupported ioctl %x (device=%x access=%x func=%x method=%x)\n", code, code >> 16, (code >> 14) & 3, (code >> 2) & 0xfff, code & 3); } else if (status == STATUS_SUCCESS) *ret_size = io->Information; /* "Information" is the size written to the output buffer */ if (status != STATUS_PENDING) RtlFreeHeap( GetProcessHeap(), 0, wsa ); return NtStatusToWSAError( status ); } static DWORD get_interface_list(SOCKET s, void *out_buff, DWORD out_size, DWORD *ret_size, DWORD *total_bytes) { DWORD size, interface_count = 0, ret; INTERFACE_INFO *info = out_buff; PMIB_IPADDRTABLE table = NULL; struct if_nameindex *if_ni; DWORD status = 0; int fd; if (!out_buff || !ret_size) return WSAEFAULT; if ((fd = get_sock_fd(s, 0, NULL)) == -1) return SOCKET_ERROR; if ((ret = GetIpAddrTable(NULL, &size, TRUE)) != ERROR_INSUFFICIENT_BUFFER) { if (ret != ERROR_NO_DATA) { ERR("Unable to get ip address table.\n"); status = WSAEINVAL; } goto done; } if (!(table = heap_alloc(size))) { ERR("No memory.\n"); status = WSAEINVAL; goto done; } if (GetIpAddrTable(table, &size, TRUE) != NO_ERROR) { ERR("Unable to get interface table.\n"); status = WSAEINVAL; goto done; } if (table->dwNumEntries * sizeof(INTERFACE_INFO) > out_size) { WARN("Buffer too small, dwNumEntries %u, out_size = %u.\n", table->dwNumEntries, out_size); *ret_size = 0; status = WSAEFAULT; goto done; } if (!(if_ni = if_nameindex())) { ERR("Unable to get interface name index.\n"); status = WSAEINVAL; goto done; } for (; interface_count < table->dwNumEntries; ++interface_count, ++info) { unsigned int addr, mask; struct ifreq if_info; unsigned int i; memset(info, 0, sizeof(*info)); for (i = 0; if_ni[i].if_index || if_ni[i].if_name; ++i) if (if_ni[i].if_index == table->table[interface_count].dwIndex) break; if (!if_ni[i].if_name) { ERR("Error obtaining interface name for ifindex %u.\n", table->table[interface_count].dwIndex); status = WSAEINVAL; break; } lstrcpynA(if_info.ifr_name, if_ni[i].if_name, IFNAMSIZ); if (ioctl(fd, SIOCGIFFLAGS, &if_info) < 0) { ERR("Error obtaining status flags for socket.\n"); status = WSAEINVAL; break; } if (if_info.ifr_flags & IFF_BROADCAST) info->iiFlags |= WS_IFF_BROADCAST; #ifdef IFF_POINTOPOINT if (if_info.ifr_flags & IFF_POINTOPOINT) info->iiFlags |= WS_IFF_POINTTOPOINT; #endif if (if_info.ifr_flags & IFF_LOOPBACK) info->iiFlags |= WS_IFF_LOOPBACK; if (if_info.ifr_flags & IFF_UP) info->iiFlags |= WS_IFF_UP; if (if_info.ifr_flags & IFF_MULTICAST) info->iiFlags |= WS_IFF_MULTICAST; addr = table->table[interface_count].dwAddr; mask = table->table[interface_count].dwMask; info->iiAddress.AddressIn.sin_family = WS_AF_INET; info->iiAddress.AddressIn.sin_port = 0; info->iiAddress.AddressIn.sin_addr.WS_s_addr = addr; info->iiNetmask.AddressIn.sin_family = WS_AF_INET; info->iiNetmask.AddressIn.sin_port = 0; info->iiNetmask.AddressIn.sin_addr.WS_s_addr = mask; if (if_info.ifr_flags & IFF_BROADCAST) { info->iiBroadcastAddress.AddressIn.sin_family = WS_AF_INET; info->iiBroadcastAddress.AddressIn.sin_port = 0; info->iiBroadcastAddress.AddressIn.sin_addr.WS_s_addr = addr | ~mask; } } if_freenameindex(if_ni); done: heap_free(table); *total_bytes = sizeof(INTERFACE_INFO) * interface_count; release_sock_fd(s, fd); return status; } /********************************************************************** * WSAIoctl (WS2_32.50) * */ INT WINAPI WSAIoctl(SOCKET s, DWORD code, LPVOID in_buff, DWORD in_size, LPVOID out_buff, DWORD out_size, LPDWORD ret_size, LPWSAOVERLAPPED overlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE completion ) { int fd; DWORD status = 0, total = 0; TRACE("%04lx, %s, %p, %d, %p, %d, %p, %p, %p\n", s, debugstr_wsaioctl(code), in_buff, in_size, out_buff, out_size, ret_size, overlapped, completion); switch (code) { case WS_FIONBIO: if (in_size != sizeof(WS_u_long) || IS_INTRESOURCE(in_buff)) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } TRACE("-> FIONBIO (%x)\n", *(WS_u_long*)in_buff); if (_get_sock_mask(s)) { /* AsyncSelect()'ed sockets are always nonblocking */ if (!*(WS_u_long *)in_buff) status = WSAEINVAL; break; } if (*(WS_u_long *)in_buff) _enable_event(SOCKET2HANDLE(s), 0, FD_WINE_NONBLOCKING, 0); else _enable_event(SOCKET2HANDLE(s), 0, 0, FD_WINE_NONBLOCKING); break; case WS_FIONREAD: { #if defined(linux) int listening = 0; socklen_t len = sizeof(listening); #endif if (out_size != sizeof(WS_u_long) || IS_INTRESOURCE(out_buff)) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; #if defined(linux) /* On Linux, FIONREAD on listening socket always fails (see tcp(7)). However, it succeeds on native. */ if (!getsockopt( fd, SOL_SOCKET, SO_ACCEPTCONN, &listening, &len ) && listening) (*(WS_u_long *) out_buff) = 0; else #endif if (ioctl(fd, FIONREAD, out_buff ) == -1) status = wsaErrno(); release_sock_fd( s, fd ); break; } case WS_SIOCATMARK: { unsigned int oob = 0, atmark = 0; socklen_t oobsize = sizeof(int); if (out_size != sizeof(WS_u_long) || IS_INTRESOURCE(out_buff)) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if ((fd = get_sock_fd( s, 0, NULL )) == -1) return SOCKET_ERROR; /* SO_OOBINLINE sockets must always return TRUE to SIOCATMARK */ if ((getsockopt(fd, SOL_SOCKET, SO_OOBINLINE, &oob, &oobsize ) == -1) || (!oob && ioctl(fd, SIOCATMARK, &atmark ) == -1)) status = wsaErrno(); else { /* The SIOCATMARK value read from ioctl() is reversed * because BSD returns TRUE if it's in the OOB mark * while Windows returns TRUE if there are NO OOB bytes. */ (*(WS_u_long *) out_buff) = oob || !atmark; } release_sock_fd( s, fd ); break; } case WS_FIOASYNC: WARN("Warning: WS1.1 shouldn't be using async I/O\n"); SetLastError(WSAEINVAL); return SOCKET_ERROR; case WS_SIO_GET_INTERFACE_LIST: { TRACE("-> SIO_GET_INTERFACE_LIST request\n"); status = get_interface_list(s, out_buff, out_size, ret_size, &total); break; } case WS_SIO_ADDRESS_LIST_QUERY: { DWORD size; TRACE("-> SIO_ADDRESS_LIST_QUERY request\n"); if (!ret_size) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if (out_size && out_size < FIELD_OFFSET(SOCKET_ADDRESS_LIST, Address[0])) { *ret_size = 0; SetLastError(WSAEINVAL); return SOCKET_ERROR; } if (GetAdaptersInfo(NULL, &size) == ERROR_BUFFER_OVERFLOW) { IP_ADAPTER_INFO *p, *table = HeapAlloc(GetProcessHeap(), 0, size); SOCKET_ADDRESS_LIST *sa_list; SOCKADDR_IN *sockaddr; SOCKET_ADDRESS *sa; unsigned int i; DWORD num; if (!table || GetAdaptersInfo(table, &size)) { HeapFree(GetProcessHeap(), 0, table); status = WSAEINVAL; break; } for (p = table, num = 0; p; p = p->Next) if (p->IpAddressList.IpAddress.String[0]) num++; total = FIELD_OFFSET(SOCKET_ADDRESS_LIST, Address[num]) + num * sizeof(*sockaddr); if (total > out_size || !out_buff) { *ret_size = total; HeapFree(GetProcessHeap(), 0, table); status = WSAEFAULT; break; } sa_list = out_buff; sa = sa_list->Address; sockaddr = (SOCKADDR_IN *)&sa[num]; sa_list->iAddressCount = num; for (p = table, i = 0; p; p = p->Next) { if (!p->IpAddressList.IpAddress.String[0]) continue; sa[i].lpSockaddr = (SOCKADDR *)&sockaddr[i]; sa[i].iSockaddrLength = sizeof(SOCKADDR); sockaddr[i].sin_family = WS_AF_INET; sockaddr[i].sin_port = 0; sockaddr[i].sin_addr.WS_s_addr = inet_addr(p->IpAddressList.IpAddress.String); i++; } HeapFree(GetProcessHeap(), 0, table); } else { WARN("unable to get IP address list\n"); status = WSAEINVAL; } break; } case WS_SIO_FLUSH: FIXME("SIO_FLUSH: stub.\n"); break; case WS_SIO_GET_EXTENSION_FUNCTION_POINTER: { #define EXTENSION_FUNCTION(x, y) { x, y, #y }, static const struct { GUID guid; void *func_ptr; const char *name; } guid_funcs[] = { EXTENSION_FUNCTION(WSAID_CONNECTEX, WS2_ConnectEx) EXTENSION_FUNCTION(WSAID_DISCONNECTEX, WS2_DisconnectEx) EXTENSION_FUNCTION(WSAID_ACCEPTEX, WS2_AcceptEx) EXTENSION_FUNCTION(WSAID_GETACCEPTEXSOCKADDRS, WS2_GetAcceptExSockaddrs) EXTENSION_FUNCTION(WSAID_TRANSMITFILE, WS2_TransmitFile) /* EXTENSION_FUNCTION(WSAID_TRANSMITPACKETS, WS2_TransmitPackets) */ EXTENSION_FUNCTION(WSAID_WSARECVMSG, WS2_WSARecvMsg) EXTENSION_FUNCTION(WSAID_WSASENDMSG, WSASendMsg) }; #undef EXTENSION_FUNCTION BOOL found = FALSE; unsigned int i; for (i = 0; i < ARRAY_SIZE(guid_funcs); i++) { if (IsEqualGUID(&guid_funcs[i].guid, in_buff)) { found = TRUE; break; } } if (found) { TRACE("-> got %s\n", guid_funcs[i].name); *(void **)out_buff = guid_funcs[i].func_ptr; total = sizeof(void *); break; } FIXME("SIO_GET_EXTENSION_FUNCTION_POINTER %s: stub\n", debugstr_guid(in_buff)); status = WSAEOPNOTSUPP; break; } case WS_SIO_KEEPALIVE_VALS: { struct tcp_keepalive *k; int keepalive, keepidle, keepintvl; if (!in_buff || in_size < sizeof(struct tcp_keepalive)) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } k = in_buff; keepalive = k->onoff ? 1 : 0; keepidle = max( 1, (k->keepalivetime + 500) / 1000 ); keepintvl = max( 1, (k->keepaliveinterval + 500) / 1000 ); TRACE("onoff: %d, keepalivetime: %d, keepaliveinterval: %d\n", keepalive, keepidle, keepintvl); fd = get_sock_fd(s, 0, NULL); if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, (void *)&keepalive, sizeof(int)) == -1) status = WSAEINVAL; #if defined(TCP_KEEPIDLE) || defined(TCP_KEEPINTVL) /* these values need to be set only if SO_KEEPALIVE is enabled */ else if(keepalive) { #ifndef TCP_KEEPIDLE FIXME("ignoring keepalive timeout\n"); #else if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, (void *)&keepidle, sizeof(int)) == -1) status = WSAEINVAL; else #endif #ifdef TCP_KEEPINTVL if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPINTVL, (void *)&keepintvl, sizeof(int)) == -1) status = WSAEINVAL; #else FIXME("ignoring keepalive interval\n"); #endif } #else else FIXME("ignoring keepalive interval and timeout\n"); #endif release_sock_fd(s, fd); break; } case WS_SIO_ROUTING_INTERFACE_QUERY: { struct WS_sockaddr *daddr = (struct WS_sockaddr *)in_buff; struct WS_sockaddr_in *daddr_in = (struct WS_sockaddr_in *)daddr; struct WS_sockaddr_in *saddr_in = out_buff; MIB_IPFORWARDROW row; PMIB_IPADDRTABLE ipAddrTable = NULL; DWORD size, i, found_index; TRACE("-> WS_SIO_ROUTING_INTERFACE_QUERY request\n"); if (!in_buff || in_size < sizeof(struct WS_sockaddr) || !out_buff || out_size < sizeof(struct WS_sockaddr_in) || !ret_size) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if (daddr->sa_family != WS_AF_INET) { FIXME("unsupported address family %d\n", daddr->sa_family); status = WSAEAFNOSUPPORT; break; } if (GetBestRoute(daddr_in->sin_addr.S_un.S_addr, 0, &row) != NOERROR || GetIpAddrTable(NULL, &size, FALSE) != ERROR_INSUFFICIENT_BUFFER) { status = WSAEFAULT; break; } ipAddrTable = HeapAlloc(GetProcessHeap(), 0, size); if (GetIpAddrTable(ipAddrTable, &size, FALSE)) { HeapFree(GetProcessHeap(), 0, ipAddrTable); status = WSAEFAULT; break; } for (i = 0, found_index = ipAddrTable->dwNumEntries; i < ipAddrTable->dwNumEntries; i++) { if (ipAddrTable->table[i].dwIndex == row.dwForwardIfIndex) found_index = i; } if (found_index == ipAddrTable->dwNumEntries) { ERR("no matching IP address for interface %d\n", row.dwForwardIfIndex); HeapFree(GetProcessHeap(), 0, ipAddrTable); status = WSAEFAULT; break; } saddr_in->sin_family = WS_AF_INET; saddr_in->sin_addr.S_un.S_addr = ipAddrTable->table[found_index].dwAddr; saddr_in->sin_port = 0; total = sizeof(struct WS_sockaddr_in); HeapFree(GetProcessHeap(), 0, ipAddrTable); break; } case WS_SIO_SET_COMPATIBILITY_MODE: TRACE("WS_SIO_SET_COMPATIBILITY_MODE ignored\n"); status = WSAEOPNOTSUPP; break; case WS_SIO_UDP_CONNRESET: FIXME("WS_SIO_UDP_CONNRESET stub\n"); break; case 0x667e: /* Netscape tries hard to use bogus ioctl 0x667e */ SetLastError(WSAEOPNOTSUPP); return SOCKET_ERROR; case WS_SIO_ADDRESS_LIST_CHANGE: code = IOCTL_AFD_ADDRESS_LIST_CHANGE; status = WSAEOPNOTSUPP; break; default: status = WSAEOPNOTSUPP; break; } if (status == WSAEOPNOTSUPP) { status = server_ioctl_sock(s, code, in_buff, in_size, out_buff, out_size, &total, overlapped, completion); if (status != WSAEOPNOTSUPP) { if (status == 0 || status == WSA_IO_PENDING || status == WSAEWOULDBLOCK) TRACE("-> %s request\n", debugstr_wsaioctl(code)); else ERR("-> %s request failed with status 0x%x\n", debugstr_wsaioctl(code), status); /* overlapped and completion operations will be handled by the server */ completion = NULL; overlapped = NULL; } else FIXME("unsupported WS_IOCTL cmd (%s)\n", debugstr_wsaioctl(code)); } if (completion) { FIXME( "completion routine %p not supported\n", completion ); } else if (overlapped) { ULONG_PTR cvalue = (overlapped && ((ULONG_PTR)overlapped->hEvent & 1) == 0) ? (ULONG_PTR)overlapped : 0; overlapped->Internal = sock_error_to_ntstatus( status ); overlapped->InternalHigh = total; if (cvalue) WS_AddCompletion( HANDLE2SOCKET(s), cvalue, overlapped->Internal, total, FALSE ); if (overlapped->hEvent) NtSetEvent( overlapped->hEvent, NULL ); } if (!status) { if (ret_size) *ret_size = total; return 0; } SetLastError( status ); return SOCKET_ERROR; } /*********************************************************************** * ioctlsocket (WS2_32.10) */ int WINAPI WS_ioctlsocket(SOCKET s, LONG cmd, WS_u_long *argp) { DWORD ret_size; return WSAIoctl( s, cmd, argp, sizeof(WS_u_long), argp, sizeof(WS_u_long), &ret_size, NULL, NULL ); } /*********************************************************************** * listen (WS2_32.13) */ int WINAPI WS_listen(SOCKET s, int backlog) { int fd = get_sock_fd( s, FILE_READ_DATA, NULL ), ret = SOCKET_ERROR; TRACE("socket %04lx, backlog %d\n", s, backlog); if (fd != -1) { int bound = is_fd_bound(fd, NULL, NULL); if (bound <= 0) { SetLastError(bound == -1 ? wsaErrno() : WSAEINVAL); } else if (listen(fd, backlog) == 0) { _enable_event(SOCKET2HANDLE(s), FD_ACCEPT, FD_WINE_LISTENING, FD_CONNECT|FD_WINE_CONNECTED); ret = 0; } else SetLastError(wsaErrno()); release_sock_fd( s, fd ); } return ret; } /*********************************************************************** * recv (WS2_32.16) */ int WINAPI WS_recv(SOCKET s, char *buf, int len, int flags) { DWORD n, dwFlags = flags; WSABUF wsabuf; wsabuf.len = len; wsabuf.buf = buf; if ( WS2_recv_base(s, &wsabuf, 1, &n, &dwFlags, NULL, NULL, NULL, NULL, NULL) == SOCKET_ERROR ) return SOCKET_ERROR; else return n; } /*********************************************************************** * recvfrom (WS2_32.17) */ int WINAPI WS_recvfrom(SOCKET s, char *buf, INT len, int flags, struct WS_sockaddr *from, int *fromlen) { DWORD n, dwFlags = flags; WSABUF wsabuf; wsabuf.len = len; wsabuf.buf = buf; if ( WS2_recv_base(s, &wsabuf, 1, &n, &dwFlags, from, fromlen, NULL, NULL, NULL) == SOCKET_ERROR ) return SOCKET_ERROR; else return n; } /* allocate a poll array for the corresponding fd sets */ static struct pollfd *fd_sets_to_poll( const WS_fd_set *readfds, const WS_fd_set *writefds, const WS_fd_set *exceptfds, int *count_ptr ) { unsigned int i, j = 0, count = 0; struct pollfd *fds; struct per_thread_data *ptb = get_per_thread_data(); if (readfds) count += readfds->fd_count; if (writefds) count += writefds->fd_count; if (exceptfds) count += exceptfds->fd_count; *count_ptr = count; if (!count) { SetLastError(WSAEINVAL); return NULL; } /* check if the cache can hold all descriptors, if not do the resizing */ if (ptb->fd_count < count) { if (!(fds = HeapAlloc(GetProcessHeap(), 0, count * sizeof(fds[0])))) { SetLastError( ERROR_NOT_ENOUGH_MEMORY ); return NULL; } HeapFree(GetProcessHeap(), 0, ptb->fd_cache); ptb->fd_cache = fds; ptb->fd_count = count; } else fds = ptb->fd_cache; if (readfds) for (i = 0; i < readfds->fd_count; i++, j++) { fds[j].fd = get_sock_fd( readfds->fd_array[i], FILE_READ_DATA, NULL ); if (fds[j].fd == -1) goto failed; fds[j].revents = 0; if (is_fd_bound(fds[j].fd, NULL, NULL) == 1) { fds[j].events = POLLIN; } else { release_sock_fd( readfds->fd_array[i], fds[j].fd ); fds[j].fd = -1; fds[j].events = 0; } } if (writefds) for (i = 0; i < writefds->fd_count; i++, j++) { fds[j].fd = get_sock_fd( writefds->fd_array[i], FILE_WRITE_DATA, NULL ); if (fds[j].fd == -1) goto failed; fds[j].revents = 0; if (is_fd_bound(fds[j].fd, NULL, NULL) == 1 || _get_fd_type(fds[j].fd) == SOCK_DGRAM) { fds[j].events = POLLOUT; } else { release_sock_fd( writefds->fd_array[i], fds[j].fd ); fds[j].fd = -1; fds[j].events = 0; } } if (exceptfds) for (i = 0; i < exceptfds->fd_count; i++, j++) { fds[j].fd = get_sock_fd( exceptfds->fd_array[i], 0, NULL ); if (fds[j].fd == -1) goto failed; fds[j].revents = 0; if (is_fd_bound(fds[j].fd, NULL, NULL) == 1) { int oob_inlined = 0; socklen_t olen = sizeof(oob_inlined); fds[j].events = POLLHUP; /* Check if we need to test for urgent data or not */ getsockopt(fds[j].fd, SOL_SOCKET, SO_OOBINLINE, (char*) &oob_inlined, &olen); if (!oob_inlined) fds[j].events |= POLLPRI; } else { release_sock_fd( exceptfds->fd_array[i], fds[j].fd ); fds[j].fd = -1; fds[j].events = 0; } } return fds; failed: count = j; j = 0; if (readfds) for (i = 0; i < readfds->fd_count && j < count; i++, j++) if (fds[j].fd != -1) release_sock_fd( readfds->fd_array[i], fds[j].fd ); if (writefds) for (i = 0; i < writefds->fd_count && j < count; i++, j++) if (fds[j].fd != -1) release_sock_fd( writefds->fd_array[i], fds[j].fd ); if (exceptfds) for (i = 0; i < exceptfds->fd_count && j < count; i++, j++) if (fds[j].fd != -1) release_sock_fd( exceptfds->fd_array[i], fds[j].fd ); return NULL; } /* release the file descriptor obtained in fd_sets_to_poll */ /* must be called with the original fd_set arrays, before calling get_poll_results */ static void release_poll_fds( const WS_fd_set *readfds, const WS_fd_set *writefds, const WS_fd_set *exceptfds, struct pollfd *fds ) { unsigned int i, j = 0; if (readfds) { for (i = 0; i < readfds->fd_count; i++, j++) if (fds[j].fd != -1) release_sock_fd( readfds->fd_array[i], fds[j].fd ); } if (writefds) { for (i = 0; i < writefds->fd_count; i++, j++) if (fds[j].fd != -1) release_sock_fd( writefds->fd_array[i], fds[j].fd ); } if (exceptfds) { for (i = 0; i < exceptfds->fd_count; i++, j++) { if (fds[j].fd == -1) continue; release_sock_fd( exceptfds->fd_array[i], fds[j].fd ); if (fds[j].revents & POLLHUP) { int fd = get_sock_fd( exceptfds->fd_array[i], 0, NULL ); if (fd != -1) release_sock_fd( exceptfds->fd_array[i], fd ); else fds[j].revents = 0; } } } } static int do_poll(struct pollfd *pollfds, int count, int timeout) { struct timeval tv1, tv2; int ret, torig = timeout; if (timeout > 0) gettimeofday( &tv1, 0 ); while ((ret = poll( pollfds, count, timeout )) < 0) { if (errno != EINTR) break; if (timeout < 0) continue; if (timeout == 0) return 0; gettimeofday( &tv2, 0 ); tv2.tv_sec -= tv1.tv_sec; tv2.tv_usec -= tv1.tv_usec; if (tv2.tv_usec < 0) { tv2.tv_usec += 1000000; tv2.tv_sec -= 1; } timeout = torig - (tv2.tv_sec * 1000) - (tv2.tv_usec + 999) / 1000; if (timeout <= 0) return 0; } return ret; } /* map the poll results back into the Windows fd sets */ static int get_poll_results( WS_fd_set *readfds, WS_fd_set *writefds, WS_fd_set *exceptfds, const struct pollfd *fds ) { const struct pollfd *poll_writefds = fds + (readfds ? readfds->fd_count : 0); const struct pollfd *poll_exceptfds = poll_writefds + (writefds ? writefds->fd_count : 0); unsigned int i, k, total = 0; if (readfds) { for (i = k = 0; i < readfds->fd_count; i++) { if (fds[i].revents || (readfds == writefds && (poll_writefds[i].revents & POLLOUT) && !(poll_writefds[i].revents & POLLHUP)) || (readfds == exceptfds && poll_exceptfds[i].revents)) readfds->fd_array[k++] = readfds->fd_array[i]; } readfds->fd_count = k; total += k; } if (writefds && writefds != readfds) { for (i = k = 0; i < writefds->fd_count; i++) { if (((poll_writefds[i].revents & POLLOUT) && !(poll_writefds[i].revents & POLLHUP)) || (writefds == exceptfds && poll_exceptfds[i].revents)) writefds->fd_array[k++] = writefds->fd_array[i]; } writefds->fd_count = k; total += k; } if (exceptfds && exceptfds != readfds && exceptfds != writefds) { for (i = k = 0; i < exceptfds->fd_count; i++) if (poll_exceptfds[i].revents) exceptfds->fd_array[k++] = exceptfds->fd_array[i]; exceptfds->fd_count = k; total += k; } return total; } /*********************************************************************** * select (WS2_32.18) */ int WINAPI WS_select(int nfds, WS_fd_set *ws_readfds, WS_fd_set *ws_writefds, WS_fd_set *ws_exceptfds, const struct WS_timeval* ws_timeout) { struct pollfd *pollfds; int count, ret, timeout = -1; TRACE("read %p, write %p, excp %p timeout %p\n", ws_readfds, ws_writefds, ws_exceptfds, ws_timeout); if (!(pollfds = fd_sets_to_poll( ws_readfds, ws_writefds, ws_exceptfds, &count ))) return SOCKET_ERROR; if (ws_timeout) timeout = (ws_timeout->tv_sec * 1000) + (ws_timeout->tv_usec + 999) / 1000; ret = do_poll(pollfds, count, timeout); release_poll_fds( ws_readfds, ws_writefds, ws_exceptfds, pollfds ); if (ret == -1) SetLastError(wsaErrno()); else ret = get_poll_results( ws_readfds, ws_writefds, ws_exceptfds, pollfds ); return ret; } /*********************************************************************** * WSAPoll */ int WINAPI WSAPoll(WSAPOLLFD *wfds, ULONG count, int timeout) { int i, ret; struct pollfd *ufds; if (!count) { SetLastError(WSAEINVAL); return SOCKET_ERROR; } if (!wfds) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if (!(ufds = HeapAlloc(GetProcessHeap(), 0, count * sizeof(ufds[0])))) { SetLastError(WSAENOBUFS); return SOCKET_ERROR; } for (i = 0; i < count; i++) { ufds[i].fd = get_sock_fd(wfds[i].fd, 0, NULL); ufds[i].events = convert_poll_w2u(wfds[i].events); ufds[i].revents = 0; } ret = do_poll(ufds, count, timeout); for (i = 0; i < count; i++) { if (ufds[i].fd != -1) { release_sock_fd(wfds[i].fd, ufds[i].fd); if (ufds[i].revents & POLLHUP) { /* Check if the socket still exists */ int fd = get_sock_fd(wfds[i].fd, 0, NULL); if (fd != -1) { wfds[i].revents = WS_POLLHUP; release_sock_fd(wfds[i].fd, fd); } else wfds[i].revents = WS_POLLNVAL; } else wfds[i].revents = convert_poll_u2w(ufds[i].revents); } else wfds[i].revents = WS_POLLNVAL; } HeapFree(GetProcessHeap(), 0, ufds); return ret; } /* helper to send completion messages for client-only i/o operation case */ static void WS_AddCompletion( SOCKET sock, ULONG_PTR CompletionValue, NTSTATUS CompletionStatus, ULONG Information, BOOL async ) { SERVER_START_REQ( add_fd_completion ) { req->handle = wine_server_obj_handle( SOCKET2HANDLE(sock) ); req->cvalue = CompletionValue; req->status = CompletionStatus; req->information = Information; req->async = async; wine_server_call( req ); } SERVER_END_REQ; } /*********************************************************************** * send (WS2_32.19) */ int WINAPI WS_send(SOCKET s, const char *buf, int len, int flags) { DWORD n; WSABUF wsabuf; wsabuf.len = len; wsabuf.buf = (char*) buf; if ( WS2_sendto( s, &wsabuf, 1, &n, flags, NULL, 0, NULL, NULL) == SOCKET_ERROR ) return SOCKET_ERROR; else return n; } /*********************************************************************** * WSASend (WS2_32.72) */ INT WINAPI WSASend( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount, LPDWORD lpNumberOfBytesSent, DWORD dwFlags, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine ) { return WS2_sendto( s, lpBuffers, dwBufferCount, lpNumberOfBytesSent, dwFlags, NULL, 0, lpOverlapped, lpCompletionRoutine ); } /*********************************************************************** * WSASendDisconnect (WS2_32.73) */ INT WINAPI WSASendDisconnect( SOCKET s, LPWSABUF lpBuffers ) { return WS_shutdown( s, SD_SEND ); } static int WS2_sendto( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount, LPDWORD lpNumberOfBytesSent, DWORD dwFlags, const struct WS_sockaddr *to, int tolen, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine ) { unsigned int i, options; int n, fd, err, overlapped, flags; struct ws2_async *wsa = NULL, localwsa; int totalLength = 0; DWORD bytes_sent; BOOL is_blocking; TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, to %p, tolen %d, ovl %p, func %p\n", s, lpBuffers, dwBufferCount, dwFlags, to, tolen, lpOverlapped, lpCompletionRoutine); fd = get_sock_fd( s, FILE_WRITE_DATA, &options ); TRACE( "fd=%d, options=%x\n", fd, options ); if ( fd == -1 ) return SOCKET_ERROR; if (!lpOverlapped && !lpNumberOfBytesSent) { err = WSAEFAULT; goto error; } overlapped = (lpOverlapped || lpCompletionRoutine) && !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)); if (overlapped || dwBufferCount > 1) { if (!(wsa = (struct ws2_async *)alloc_async_io( offsetof(struct ws2_async, iovec[dwBufferCount]), WS2_async_send ))) { err = WSAEFAULT; goto error; } } else wsa = &localwsa; wsa->hSocket = SOCKET2HANDLE(s); wsa->addr = (struct WS_sockaddr *)to; wsa->addrlen.val = tolen; wsa->flags = dwFlags; wsa->lpFlags = &wsa->flags; wsa->control = NULL; wsa->n_iovecs = dwBufferCount; wsa->first_iovec = 0; for ( i = 0; i < dwBufferCount; i++ ) { wsa->iovec[i].iov_base = lpBuffers[i].buf; wsa->iovec[i].iov_len = lpBuffers[i].len; totalLength += lpBuffers[i].len; } flags = convert_flags(dwFlags); n = WS2_send( fd, wsa, flags ); if (n == -1 && errno != EAGAIN) { err = wsaErrno(); goto error; } if (overlapped) { IO_STATUS_BLOCK *iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb; ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0; wsa->user_overlapped = lpOverlapped; wsa->completion_func = lpCompletionRoutine; release_sock_fd( s, fd ); if (n == -1 || n < totalLength) { iosb->u.Status = STATUS_PENDING; iosb->Information = n == -1 ? 0 : n; if (wsa->completion_func) err = register_async( ASYNC_TYPE_WRITE, wsa->hSocket, &wsa->io, NULL, ws2_async_apc, wsa, iosb ); else err = register_async( ASYNC_TYPE_WRITE, wsa->hSocket, &wsa->io, lpOverlapped->hEvent, NULL, (void *)cvalue, iosb ); /* Enable the event only after starting the async. The server will deliver it as soon as the async is done. */ _enable_event(SOCKET2HANDLE(s), FD_WRITE, 0, 0); if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa ); SetLastError(NtStatusToWSAError( err )); return SOCKET_ERROR; } iosb->u.Status = STATUS_SUCCESS; iosb->Information = n; if (lpNumberOfBytesSent) *lpNumberOfBytesSent = n; if (!wsa->completion_func) { if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n, FALSE ); if (lpOverlapped->hEvent) SetEvent( lpOverlapped->hEvent ); HeapFree( GetProcessHeap(), 0, wsa ); } else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc, (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 ); SetLastError(ERROR_SUCCESS); return 0; } if ((err = sock_is_blocking( s, &is_blocking ))) goto error; if ( is_blocking ) { /* On a blocking non-overlapped stream socket, * sending blocks until the entire buffer is sent. */ DWORD timeout_start = GetTickCount(); bytes_sent = n == -1 ? 0 : n; while (wsa->first_iovec < wsa->n_iovecs) { struct pollfd pfd; int poll_timeout = -1; INT64 timeout = get_rcvsnd_timeo(fd, FALSE); if (timeout) { timeout -= GetTickCount() - timeout_start; if (timeout < 0) poll_timeout = 0; else poll_timeout = timeout <= INT_MAX ? timeout : INT_MAX; } pfd.fd = fd; pfd.events = POLLOUT; if (!poll_timeout || !poll( &pfd, 1, poll_timeout )) { err = WSAETIMEDOUT; goto error; /* msdn says a timeout in send is fatal */ } n = WS2_send( fd, wsa, flags ); if (n == -1 && errno != EAGAIN) { err = wsaErrno(); goto error; } if (n >= 0) bytes_sent += n; } } else /* non-blocking */ { if (n < totalLength) _enable_event(SOCKET2HANDLE(s), FD_WRITE, 0, 0); if (n == -1) { err = WSAEWOULDBLOCK; goto error; } bytes_sent = n; } TRACE(" -> %i bytes\n", bytes_sent); if (lpNumberOfBytesSent) *lpNumberOfBytesSent = bytes_sent; if (wsa != &localwsa) HeapFree( GetProcessHeap(), 0, wsa ); release_sock_fd( s, fd ); SetLastError(ERROR_SUCCESS); return 0; error: if (wsa != &localwsa) HeapFree( GetProcessHeap(), 0, wsa ); release_sock_fd( s, fd ); WARN(" -> ERROR %d\n", err); SetLastError(err); return SOCKET_ERROR; } /*********************************************************************** * WSASendTo (WS2_32.74) */ INT WINAPI WSASendTo( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount, LPDWORD lpNumberOfBytesSent, DWORD dwFlags, const struct WS_sockaddr *to, int tolen, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine ) { return WS2_sendto( s, lpBuffers, dwBufferCount, lpNumberOfBytesSent, dwFlags, to, tolen, lpOverlapped, lpCompletionRoutine ); } /*********************************************************************** * sendto (WS2_32.20) */ int WINAPI WS_sendto(SOCKET s, const char *buf, int len, int flags, const struct WS_sockaddr *to, int tolen) { DWORD n; WSABUF wsabuf; wsabuf.len = len; wsabuf.buf = (char*) buf; if ( WS2_sendto(s, &wsabuf, 1, &n, flags, to, tolen, NULL, NULL) == SOCKET_ERROR ) return SOCKET_ERROR; else return n; } /*********************************************************************** * setsockopt (WS2_32.21) */ int WINAPI WS_setsockopt(SOCKET s, int level, int optname, const char *optval, int optlen) { int fd; int woptval; struct linger linger; struct timeval tval; struct ip_mreq_source mreq_source; TRACE("(socket %04lx, %s, optval %s, optlen %d)\n", s, debugstr_sockopt(level, optname), debugstr_optval(optval, optlen), optlen); /* some broken apps pass the value directly instead of a pointer to it */ if(optlen && IS_INTRESOURCE(optval)) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } switch(level) { case WS_SOL_SOCKET: switch(optname) { /* Some options need some conversion before they can be sent to * setsockopt. The conversions are done here, then they will fall through * to the general case. Special options that are not passed to * setsockopt follow below that.*/ case WS_SO_DONTLINGER: if (!optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } linger.l_onoff = *(const int*)optval == 0; linger.l_linger = 0; level = SOL_SOCKET; optname = SO_LINGER; optval = (char*)&linger; optlen = sizeof(struct linger); break; case WS_SO_LINGER: if (!optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } linger.l_onoff = ((LINGER*)optval)->l_onoff; linger.l_linger = ((LINGER*)optval)->l_linger; level = SOL_SOCKET; optname = SO_LINGER; optval = (char*)&linger; optlen = sizeof(struct linger); break; case WS_SO_SNDBUF: if (!*(const int *)optval) { FIXME("SO_SNDBUF ignoring request to disable send buffering\n"); #ifdef __APPLE__ return 0; #endif } convert_sockopt(&level, &optname); break; case WS_SO_RCVBUF: if (*(const int*)optval < 2048) { WARN("SO_RCVBF for %d bytes is too small: ignored\n", *(const int*)optval ); return 0; } /* Fall through */ /* The options listed here don't need any special handling. Thanks to * the conversion happening above, options from there will fall through * to this, too.*/ case WS_SO_ACCEPTCONN: case WS_SO_BROADCAST: case WS_SO_ERROR: case WS_SO_KEEPALIVE: case WS_SO_OOBINLINE: /* BSD socket SO_REUSEADDR is not 100% compatible to winsock semantics. * however, using it the BSD way fixes bug 8513 and seems to be what * most programmers assume, anyway */ case WS_SO_REUSEADDR: case WS_SO_TYPE: convert_sockopt(&level, &optname); break; /* SO_DEBUG is a privileged operation, ignore it. */ case WS_SO_DEBUG: TRACE("Ignoring SO_DEBUG\n"); return 0; /* For some reason the game GrandPrixLegends does set SO_DONTROUTE on its * socket. According to MSDN, this option is silently ignored.*/ case WS_SO_DONTROUTE: TRACE("Ignoring SO_DONTROUTE\n"); return 0; /* Stops two sockets from being bound to the same port. Always happens * on unix systems, so just drop it. */ case WS_SO_EXCLUSIVEADDRUSE: TRACE("Ignoring SO_EXCLUSIVEADDRUSE, is always set.\n"); return 0; /* After a ConnectEx call succeeds, the socket can't be used with half of the * normal winsock functions on windows. We don't have that problem. */ case WS_SO_UPDATE_CONNECT_CONTEXT: TRACE("Ignoring SO_UPDATE_CONNECT_CONTEXT, since our sockets are normal\n"); return 0; /* After a AcceptEx call succeeds, the socket can't be used with half of the * normal winsock functions on windows. We don't have that problem. */ case WS_SO_UPDATE_ACCEPT_CONTEXT: TRACE("Ignoring SO_UPDATE_ACCEPT_CONTEXT, since our sockets are normal\n"); return 0; /* SO_OPENTYPE does not require a valid socket handle. */ case WS_SO_OPENTYPE: if (!optlen || optlen < sizeof(int) || !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } get_per_thread_data()->opentype = *(const int *)optval; TRACE("setting global SO_OPENTYPE = 0x%x\n", *((const int*)optval) ); return 0; #ifdef SO_RCVTIMEO case WS_SO_RCVTIMEO: #endif #ifdef SO_SNDTIMEO case WS_SO_SNDTIMEO: #endif #if defined(SO_RCVTIMEO) || defined(SO_SNDTIMEO) if (optval && optlen == sizeof(UINT32)) { /* WinSock passes milliseconds instead of struct timeval */ tval.tv_usec = (*(const UINT32*)optval % 1000) * 1000; tval.tv_sec = *(const UINT32*)optval / 1000; /* min of 500 milliseconds */ if (tval.tv_sec == 0 && tval.tv_usec && tval.tv_usec < 500000) tval.tv_usec = 500000; optlen = sizeof(struct timeval); optval = (char*)&tval; } else if (optlen == sizeof(struct timeval)) { WARN("SO_SND/RCVTIMEO for %d bytes: assuming unixism\n", optlen); } else { WARN("SO_SND/RCVTIMEO for %d bytes is weird: ignored\n", optlen); return 0; } convert_sockopt(&level, &optname); break; #endif case WS_SO_RANDOMIZE_PORT: FIXME("Ignoring WS_SO_RANDOMIZE_PORT\n"); return 0; case WS_SO_PORT_SCALABILITY: FIXME("Ignoring WS_SO_PORT_SCALABILITY\n"); return 0; case WS_SO_REUSE_UNICASTPORT: FIXME("Ignoring WS_SO_REUSE_UNICASTPORT\n"); return 0; case WS_SO_REUSE_MULTICASTPORT: FIXME("Ignoring WS_SO_REUSE_MULTICASTPORT\n"); return 0; default: TRACE("Unknown SOL_SOCKET optname: 0x%08x\n", optname); SetLastError(WSAENOPROTOOPT); return SOCKET_ERROR; } break; /* case WS_SOL_SOCKET */ #ifdef HAS_IPX case WS_NSPROTO_IPX: switch(optname) { case WS_IPX_PTYPE: return set_ipx_packettype(s, *(int*)optval); case WS_IPX_FILTERPTYPE: /* Sets the receive filter packet type, at the moment we don't support it */ FIXME("IPX_FILTERPTYPE: %x\n", *optval); /* Returning 0 is better for now than returning a SOCKET_ERROR */ return 0; default: FIXME("opt_name:%x\n", optname); return SOCKET_ERROR; } break; /* case WS_NSPROTO_IPX */ #endif /* Levels WS_IPPROTO_TCP and WS_IPPROTO_IP convert directly */ case WS_IPPROTO_TCP: switch(optname) { case WS_TCP_NODELAY: convert_sockopt(&level, &optname); break; default: FIXME("Unknown IPPROTO_TCP optname 0x%08x\n", optname); return SOCKET_ERROR; } break; case WS_IPPROTO_IP: switch(optname) { case WS_IP_ADD_SOURCE_MEMBERSHIP: case WS_IP_DROP_SOURCE_MEMBERSHIP: case WS_IP_BLOCK_SOURCE: case WS_IP_UNBLOCK_SOURCE: { WS_IP_MREQ_SOURCE* val = (void*)optval; mreq_source.imr_interface.s_addr = val->imr_interface.S_un.S_addr; mreq_source.imr_multiaddr.s_addr = val->imr_multiaddr.S_un.S_addr; mreq_source.imr_sourceaddr.s_addr = val->imr_sourceaddr.S_un.S_addr; optval = (char*)&mreq_source; optlen = sizeof(mreq_source); convert_sockopt(&level, &optname); break; } case WS_IP_ADD_MEMBERSHIP: case WS_IP_DROP_MEMBERSHIP: #ifdef IP_HDRINCL case WS_IP_HDRINCL: #endif case WS_IP_MULTICAST_IF: case WS_IP_MULTICAST_LOOP: case WS_IP_MULTICAST_TTL: case WS_IP_OPTIONS: #if defined(IP_PKTINFO) || defined(IP_RECVDSTADDR) case WS_IP_PKTINFO: #endif case WS_IP_TOS: case WS_IP_TTL: #ifdef IP_UNICAST_IF case WS_IP_UNICAST_IF: #endif convert_sockopt(&level, &optname); break; case WS_IP_DONTFRAGMENT: return set_dont_fragment(s, IPPROTO_IP, *(BOOL *)optval) ? 0 : SOCKET_ERROR; default: FIXME("Unknown IPPROTO_IP optname 0x%08x\n", optname); return SOCKET_ERROR; } break; case WS_IPPROTO_IPV6: switch(optname) { #ifdef IPV6_ADD_MEMBERSHIP case WS_IPV6_ADD_MEMBERSHIP: #endif #ifdef IPV6_DROP_MEMBERSHIP case WS_IPV6_DROP_MEMBERSHIP: #endif case WS_IPV6_MULTICAST_IF: case WS_IPV6_MULTICAST_HOPS: case WS_IPV6_MULTICAST_LOOP: case WS_IPV6_UNICAST_HOPS: #ifdef IPV6_UNICAST_IF case WS_IPV6_UNICAST_IF: #endif convert_sockopt(&level, &optname); break; case WS_IPV6_DONTFRAG: return set_dont_fragment(s, IPPROTO_IPV6, *(BOOL *)optval) ? 0 : SOCKET_ERROR; case WS_IPV6_PROTECTION_LEVEL: FIXME("IPV6_PROTECTION_LEVEL is ignored!\n"); return 0; case WS_IPV6_V6ONLY: { union generic_unix_sockaddr uaddr; socklen_t uaddrlen; int bound; fd = get_sock_fd( s, 0, NULL ); if (fd == -1) return SOCKET_ERROR; bound = is_fd_bound(fd, &uaddr, &uaddrlen); release_sock_fd( s, fd ); if (bound == 0 && uaddr.addr.sa_family == AF_INET) { /* Changing IPV6_V6ONLY succeeds on AF_INET (IPv4) socket * on Windows (with IPv6 support) if the socket is unbound. * It is essentially a noop, though Windows does store the value */ WARN("Silently ignoring IPPROTO_IPV6+IPV6_V6ONLY on AF_INET socket\n"); return 0; } level = IPPROTO_IPV6; optname = IPV6_V6ONLY; break; } default: FIXME("Unknown IPPROTO_IPV6 optname 0x%08x\n", optname); return SOCKET_ERROR; } break; default: WARN("Unknown level: 0x%08x\n", level); SetLastError(WSAEINVAL); return SOCKET_ERROR; } /* end switch(level) */ /* avoid endianness issues if argument is a 16-bit int */ if (optval && optlen < sizeof(int)) { woptval= *((const INT16 *) optval); optval= (char*) &woptval; woptval&= (1 << optlen * 8) - 1; optlen=sizeof(int); } fd = get_sock_fd( s, 0, NULL ); if (fd == -1) return SOCKET_ERROR; if (setsockopt(fd, level, optname, optval, optlen) == 0) { #ifdef __APPLE__ if (level == SOL_SOCKET && optname == SO_REUSEADDR && setsockopt(fd, level, SO_REUSEPORT, optval, optlen) != 0) { SetLastError(wsaErrno()); release_sock_fd( s, fd ); return SOCKET_ERROR; } #endif release_sock_fd( s, fd ); return 0; } TRACE("Setting socket error, %d\n", wsaErrno()); SetLastError(wsaErrno()); release_sock_fd( s, fd ); return SOCKET_ERROR; } /*********************************************************************** * shutdown (WS2_32.22) */ int WINAPI WS_shutdown(SOCKET s, int how) { int fd, err = WSAENOTSOCK; unsigned int options = 0, clear_flags = 0; fd = get_sock_fd( s, 0, &options ); TRACE("socket %04lx, how 0x%x, options 0x%x\n", s, how, options ); if (fd == -1) return SOCKET_ERROR; switch( how ) { case SD_RECEIVE: /* drop receives */ clear_flags |= FD_READ; break; case SD_SEND: /* drop sends */ clear_flags |= FD_WRITE; break; case SD_BOTH: /* drop all */ clear_flags |= FD_READ|FD_WRITE; /*fall through */ default: clear_flags |= FD_WINE_LISTENING; } if (!(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT))) { switch ( how ) { case SD_RECEIVE: err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ ); break; case SD_SEND: err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE ); break; case SD_BOTH: default: err = WS2_register_async_shutdown( s, ASYNC_TYPE_READ ); if (!err) err = WS2_register_async_shutdown( s, ASYNC_TYPE_WRITE ); break; } if (err) goto error; } else /* non-overlapped mode */ { if ( shutdown( fd, how ) ) { err = wsaErrno(); goto error; } } release_sock_fd( s, fd ); _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags ); if ( how > 1) WSAAsyncSelect( s, 0, 0, 0 ); return 0; error: release_sock_fd( s, fd ); _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags ); SetLastError( err ); return SOCKET_ERROR; } /*********************************************************************** * socket (WS2_32.23) */ SOCKET WINAPI WS_socket(int af, int type, int protocol) { TRACE("af=%d type=%d protocol=%d\n", af, type, protocol); return WSASocketW( af, type, protocol, NULL, 0, get_per_thread_data()->opentype ? 0 : WSA_FLAG_OVERLAPPED ); } /*********************************************************************** * gethostbyaddr (WS2_32.51) */ struct WS_hostent* WINAPI WS_gethostbyaddr(const char *addr, int len, int type) { struct WS_hostent *retval = NULL; struct hostent* host; int unixtype = convert_af_w2u(type); const char *paddr = addr; unsigned long loopback; #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6 char *extrabuf; int ebufsize = 1024; struct hostent hostentry; int locerr = ENOBUFS; #endif /* convert back the magic loopback address if necessary */ if (unixtype == AF_INET && len == 4 && !memcmp(addr, magic_loopback_addr, 4)) { loopback = htonl(INADDR_LOOPBACK); paddr = (char*) &loopback; } #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6 host = NULL; extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ; while(extrabuf) { int res = gethostbyaddr_r(paddr, len, unixtype, &hostentry, extrabuf, ebufsize, &host, &locerr); if (res != ERANGE) break; ebufsize *=2; extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ; } if (host) retval = WS_dup_he(host); else SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr)); HeapFree(GetProcessHeap(),0,extrabuf); #else EnterCriticalSection( &csWSgetXXXbyYYY ); host = gethostbyaddr(paddr, len, unixtype); if (host) retval = WS_dup_he(host); else SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno)); LeaveCriticalSection( &csWSgetXXXbyYYY ); #endif TRACE("ptr %p, len %d, type %d ret %p\n", addr, len, type, retval); return retval; } /*********************************************************************** * WS_compare_routes_by_metric_asc (INTERNAL) * * Comparison function for qsort(), for sorting two routes (struct route) * by metric in ascending order. */ static int WS_compare_routes_by_metric_asc(const void *left, const void *right) { const struct route *a = left, *b = right; if (a->default_route && b->default_route) return a->default_route - b->default_route; if (a->default_route && !b->default_route) return -1; if (b->default_route && !a->default_route) return 1; return a->metric - b->metric; } /*********************************************************************** * WS_get_local_ips (INTERNAL) * * Returns the list of local IP addresses by going through the network * adapters and using the local routing table to sort the addresses * from highest routing priority to lowest routing priority. This * functionality is inferred from the description for obtaining local * IP addresses given in the Knowledge Base Article Q160215. * * Please note that the returned hostent is only freed when the thread * closes and is replaced if another hostent is requested. */ static struct WS_hostent* WS_get_local_ips( char *hostname ) { int numroutes = 0, i, j, default_routes = 0; DWORD n; PIP_ADAPTER_INFO adapters = NULL, k; struct WS_hostent *hostlist = NULL; PMIB_IPFORWARDTABLE routes = NULL; struct route *route_addrs = NULL; DWORD adap_size, route_size; /* Obtain the size of the adapter list and routing table, also allocate memory */ if (GetAdaptersInfo(NULL, &adap_size) != ERROR_BUFFER_OVERFLOW) return NULL; if (GetIpForwardTable(NULL, &route_size, FALSE) != ERROR_INSUFFICIENT_BUFFER) return NULL; adapters = HeapAlloc(GetProcessHeap(), 0, adap_size); routes = HeapAlloc(GetProcessHeap(), 0, route_size); if (adapters == NULL || routes == NULL) goto cleanup; /* Obtain the adapter list and the full routing table */ if (GetAdaptersInfo(adapters, &adap_size) != NO_ERROR) goto cleanup; if (GetIpForwardTable(routes, &route_size, FALSE) != NO_ERROR) goto cleanup; /* Store the interface associated with each route */ for (n = 0; n < routes->dwNumEntries; n++) { IF_INDEX ifindex; DWORD ifmetric, ifdefault = 0; BOOL exists = FALSE; /* Check if this is a default route (there may be more than one) */ if (!routes->table[n].dwForwardDest) ifdefault = ++default_routes; else if (routes->table[n].u1.ForwardType != MIB_IPROUTE_TYPE_DIRECT) continue; ifindex = routes->table[n].dwForwardIfIndex; ifmetric = routes->table[n].dwForwardMetric1; /* Only store the lowest valued metric for an interface */ for (j = 0; j < numroutes; j++) { if (route_addrs[j].interface == ifindex) { if (route_addrs[j].metric > ifmetric) route_addrs[j].metric = ifmetric; exists = TRUE; } } if (exists) continue; route_addrs = heap_realloc(route_addrs, (numroutes+1)*sizeof(struct route)); if (route_addrs == NULL) goto cleanup; /* Memory allocation error, fail gracefully */ route_addrs[numroutes].interface = ifindex; route_addrs[numroutes].metric = ifmetric; route_addrs[numroutes].default_route = ifdefault; /* If no IP is found in the next step (for whatever reason) * then fall back to the magic loopback address. */ memcpy(&(route_addrs[numroutes].addr.s_addr), magic_loopback_addr, 4); numroutes++; } if (numroutes == 0) goto cleanup; /* No routes, fall back to the Magic IP */ /* Find the IP address associated with each found interface */ for (i = 0; i < numroutes; i++) { for (k = adapters; k != NULL; k = k->Next) { char *ip = k->IpAddressList.IpAddress.String; if (route_addrs[i].interface == k->Index) route_addrs[i].addr.s_addr = (in_addr_t) inet_addr(ip); } } /* Allocate a hostent and enough memory for all the IPs, * including the NULL at the end of the list. */ hostlist = WS_create_he(hostname, 1, 0, numroutes+1, sizeof(struct in_addr)); if (hostlist == NULL) goto cleanup; /* Failed to allocate a hostent for the list of IPs */ hostlist->h_addr_list[numroutes] = NULL; /* NULL-terminate the address list */ hostlist->h_aliases[0] = NULL; /* NULL-terminate the alias list */ hostlist->h_addrtype = AF_INET; hostlist->h_length = sizeof(struct in_addr); /* = 4 */ /* Reorder the entries before placing them in the host list. Windows expects * the IP list in order from highest priority to lowest (the critical thing * is that most applications expect the first IP to be the default route). */ if (numroutes > 1) qsort(route_addrs, numroutes, sizeof(struct route), WS_compare_routes_by_metric_asc); for (i = 0; i < numroutes; i++) (*(struct in_addr *) hostlist->h_addr_list[i]) = route_addrs[i].addr; /* Cleanup all allocated memory except the address list, * the address list is used by the calling app. */ cleanup: HeapFree(GetProcessHeap(), 0, route_addrs); HeapFree(GetProcessHeap(), 0, adapters); HeapFree(GetProcessHeap(), 0, routes); return hostlist; } /*********************************************************************** * gethostbyname (WS2_32.52) */ struct WS_hostent* WINAPI WS_gethostbyname(const char* name) { struct WS_hostent *retval = NULL; struct hostent* host; #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6 char *extrabuf; int ebufsize=1024; struct hostent hostentry; int locerr = ENOBUFS; #endif char hostname[100]; if(!num_startup) { SetLastError(WSANOTINITIALISED); return NULL; } if( gethostname( hostname, 100) == -1) { SetLastError(WSAENOBUFS); /* appropriate ? */ return retval; } if( !name || !name[0]) { name = hostname; } /* If the hostname of the local machine is requested then return the * complete list of local IP addresses */ if(strcmp(name, hostname) == 0) retval = WS_get_local_ips(hostname); /* If any other hostname was requested (or the routing table lookup failed) * then return the IP found by the host OS */ if(retval == NULL) { #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6 host = NULL; extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ; while(extrabuf) { int res = gethostbyname_r(name, &hostentry, extrabuf, ebufsize, &host, &locerr); if( res != ERANGE) break; ebufsize *=2; extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ; } if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr)); #else EnterCriticalSection( &csWSgetXXXbyYYY ); host = gethostbyname(name); if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno)); #endif if (host) retval = WS_dup_he(host); #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6 HeapFree(GetProcessHeap(),0,extrabuf); #else LeaveCriticalSection( &csWSgetXXXbyYYY ); #endif } if (retval && retval->h_addr_list[0][0] == 127 && strcmp(name, "localhost") != 0) { /* hostname != "localhost" but has loopback address. replace by our * special address.*/ memcpy(retval->h_addr_list[0], magic_loopback_addr, 4); } TRACE( "%s ret %p\n", debugstr_a(name), retval ); return retval; } static const struct { int prot; const char *names[3]; } protocols[] = { { 0, { "ip", "IP" }}, { 1, { "icmp", "ICMP" }}, { 3, { "ggp", "GGP" }}, { 6, { "tcp", "TCP" }}, { 8, { "egp", "EGP" }}, { 12, { "pup", "PUP" }}, { 17, { "udp", "UDP" }}, { 20, { "hmp", "HMP" }}, { 22, { "xns-idp", "XNS-IDP" }}, { 27, { "rdp", "RDP" }}, { 41, { "ipv6", "IPv6" }}, { 43, { "ipv6-route", "IPv6-Route" }}, { 44, { "ipv6-frag", "IPv6-Frag" }}, { 50, { "esp", "ESP" }}, { 51, { "ah", "AH" }}, { 58, { "ipv6-icmp", "IPv6-ICMP" }}, { 59, { "ipv6-nonxt", "IPv6-NoNxt" }}, { 60, { "ipv6-opts", "IPv6-Opts" }}, { 66, { "rvd", "RVD" }}, }; /*********************************************************************** * getprotobyname (WS2_32.53) */ struct WS_protoent* WINAPI WS_getprotobyname(const char* name) { struct WS_protoent* retval = NULL; unsigned int i; for (i = 0; i < ARRAY_SIZE(protocols); i++) { if (_strnicmp( protocols[i].names[0], name, -1 )) continue; retval = WS_create_pe( protocols[i].names[0], (char **)protocols[i].names + 1, protocols[i].prot ); break; } if (!retval) { WARN( "protocol %s not found\n", debugstr_a(name) ); SetLastError(WSANO_DATA); } TRACE( "%s ret %p\n", debugstr_a(name), retval ); return retval; } /*********************************************************************** * getprotobynumber (WS2_32.54) */ struct WS_protoent* WINAPI WS_getprotobynumber(int number) { struct WS_protoent* retval = NULL; unsigned int i; for (i = 0; i < ARRAY_SIZE(protocols); i++) { if (protocols[i].prot != number) continue; retval = WS_create_pe( protocols[i].names[0], (char **)protocols[i].names + 1, protocols[i].prot ); break; } if (!retval) { WARN( "protocol %d not found\n", number ); SetLastError(WSANO_DATA); } TRACE("%i ret %p\n", number, retval); return retval; } /*********************************************************************** * getservbyname (WS2_32.55) */ struct WS_servent* WINAPI WS_getservbyname(const char *name, const char *proto) { struct WS_servent* retval = NULL; struct servent* serv; char *name_str; char *proto_str = NULL; if (!(name_str = strdup_lower(name))) return NULL; if (proto && *proto) { if (!(proto_str = strdup_lower(proto))) { HeapFree( GetProcessHeap(), 0, name_str ); return NULL; } } EnterCriticalSection( &csWSgetXXXbyYYY ); serv = getservbyname(name_str, proto_str); if( serv != NULL ) { retval = WS_dup_se(serv); } else SetLastError(WSANO_DATA); LeaveCriticalSection( &csWSgetXXXbyYYY ); HeapFree( GetProcessHeap(), 0, proto_str ); HeapFree( GetProcessHeap(), 0, name_str ); TRACE( "%s, %s ret %p\n", debugstr_a(name), debugstr_a(proto), retval ); return retval; } /*********************************************************************** * freeaddrinfo (WS2_32.@) */ void WINAPI WS_freeaddrinfo(struct WS_addrinfo *res) { while (res) { struct WS_addrinfo *next; HeapFree(GetProcessHeap(),0,res->ai_canonname); HeapFree(GetProcessHeap(),0,res->ai_addr); next = res->ai_next; HeapFree(GetProcessHeap(),0,res); res = next; } } /* helper functions for getaddrinfo()/getnameinfo() */ static int convert_aiflag_w2u(int winflags) { unsigned int i; int unixflags = 0; for (i = 0; i < ARRAY_SIZE(ws_aiflag_map); i++) if (ws_aiflag_map[i][0] & winflags) { unixflags |= ws_aiflag_map[i][1]; winflags &= ~ws_aiflag_map[i][0]; } if (winflags) FIXME("Unhandled windows AI_xxx flags 0x%x\n", winflags); return unixflags; } static int convert_niflag_w2u(int winflags) { unsigned int i; int unixflags = 0; for (i = 0; i < ARRAY_SIZE(ws_niflag_map); i++) if (ws_niflag_map[i][0] & winflags) { unixflags |= ws_niflag_map[i][1]; winflags &= ~ws_niflag_map[i][0]; } if (winflags) FIXME("Unhandled windows NI_xxx flags 0x%x\n", winflags); return unixflags; } static int convert_aiflag_u2w(int unixflags) { unsigned int i; int winflags = 0; for (i = 0; i < ARRAY_SIZE(ws_aiflag_map); i++) if (ws_aiflag_map[i][1] & unixflags) { winflags |= ws_aiflag_map[i][0]; unixflags &= ~ws_aiflag_map[i][1]; } if (unixflags) WARN("Unhandled UNIX AI_xxx flags 0x%x\n", unixflags); return winflags; } static int convert_eai_u2w(int unixret) { int i; if (!unixret) return 0; for (i=0;ws_eai_map[i][0];i++) if (ws_eai_map[i][1] == unixret) return ws_eai_map[i][0]; if (unixret == EAI_SYSTEM) /* There are broken versions of glibc which return EAI_SYSTEM * and set errno to 0 instead of returning EAI_NONAME. */ return errno ? sock_get_error( errno ) : WS_EAI_NONAME; FIXME("Unhandled unix EAI_xxx ret %d\n", unixret); return unixret; } static char *get_fqdn(void) { char *ret; DWORD size = 0; GetComputerNameExA( ComputerNamePhysicalDnsFullyQualified, NULL, &size ); if (GetLastError() != ERROR_MORE_DATA) return NULL; if (!(ret = HeapAlloc( GetProcessHeap(), 0, size ))) return NULL; if (!GetComputerNameExA( ComputerNamePhysicalDnsFullyQualified, ret, &size )) { HeapFree( GetProcessHeap(), 0, ret ); return NULL; } return ret; } static BOOL addrinfo_in_list( const struct WS_addrinfo *list, const struct WS_addrinfo *ai ) { const struct WS_addrinfo *cursor = list; while (cursor) { if (ai->ai_flags == cursor->ai_flags && ai->ai_family == cursor->ai_family && ai->ai_socktype == cursor->ai_socktype && ai->ai_protocol == cursor->ai_protocol && ai->ai_addrlen == cursor->ai_addrlen && !memcmp(ai->ai_addr, cursor->ai_addr, ai->ai_addrlen) && ((ai->ai_canonname && cursor->ai_canonname && !strcmp(ai->ai_canonname, cursor->ai_canonname)) || (!ai->ai_canonname && !cursor->ai_canonname))) return TRUE; cursor = cursor->ai_next; } return FALSE; } /*********************************************************************** * getaddrinfo (WS2_32.@) */ int WINAPI WS_getaddrinfo(LPCSTR nodename, LPCSTR servname, const struct WS_addrinfo *hints, struct WS_addrinfo **res) { #ifdef HAVE_GETADDRINFO struct addrinfo *unixaires = NULL; int result; struct addrinfo unixhints, *punixhints = NULL; char *nodeV6 = NULL, *fqdn = NULL; const char *node; *res = NULL; if (!nodename && !servname) { SetLastError(WSAHOST_NOT_FOUND); return WSAHOST_NOT_FOUND; } if (!nodename) node = NULL; else if (!nodename[0]) { if (!(fqdn = get_fqdn())) return WSA_NOT_ENOUGH_MEMORY; node = fqdn; } else { node = nodename; /* Check for [ipv6] or [ipv6]:portnumber, which are supported by Windows */ if (!hints || hints->ai_family == WS_AF_UNSPEC || hints->ai_family == WS_AF_INET6) { char *close_bracket; if (node[0] == '[' && (close_bracket = strchr(node + 1, ']'))) { nodeV6 = HeapAlloc(GetProcessHeap(), 0, close_bracket - node); if (!nodeV6) return WSA_NOT_ENOUGH_MEMORY; lstrcpynA(nodeV6, node + 1, close_bracket - node); node = nodeV6; } } } /* servname tweak required by OSX and BSD kernels */ if (servname && !servname[0]) servname = "0"; if (hints) { punixhints = &unixhints; memset(&unixhints, 0, sizeof(unixhints)); punixhints->ai_flags = convert_aiflag_w2u(hints->ai_flags); /* zero is a wildcard, no need to convert */ if (hints->ai_family) punixhints->ai_family = convert_af_w2u(hints->ai_family); if (hints->ai_socktype) punixhints->ai_socktype = convert_socktype_w2u(hints->ai_socktype); if (hints->ai_protocol) punixhints->ai_protocol = max(convert_proto_w2u(hints->ai_protocol), 0); if (punixhints->ai_socktype < 0) { SetLastError(WSAESOCKTNOSUPPORT); HeapFree(GetProcessHeap(), 0, fqdn); HeapFree(GetProcessHeap(), 0, nodeV6); return SOCKET_ERROR; } /* windows allows invalid combinations of socket type and protocol, unix does not. * fix the parameters here to make getaddrinfo call always work */ if (punixhints->ai_protocol == IPPROTO_TCP && punixhints->ai_socktype != SOCK_STREAM && punixhints->ai_socktype != SOCK_SEQPACKET) punixhints->ai_socktype = 0; else if (punixhints->ai_protocol == IPPROTO_UDP && punixhints->ai_socktype != SOCK_DGRAM) punixhints->ai_socktype = 0; else if (IS_IPX_PROTO(punixhints->ai_protocol) && punixhints->ai_socktype != SOCK_DGRAM) punixhints->ai_socktype = 0; else if (punixhints->ai_protocol == IPPROTO_IPV6) punixhints->ai_protocol = 0; } /* getaddrinfo(3) is thread safe, no need to wrap in CS */ result = getaddrinfo(node, servname, punixhints, &unixaires); if (result && (!hints || !(hints->ai_flags & WS_AI_NUMERICHOST)) && node) { if (!fqdn && !(fqdn = get_fqdn())) { HeapFree(GetProcessHeap(), 0, nodeV6); return WSA_NOT_ENOUGH_MEMORY; } if (!strcmp(fqdn, node) || (!strncmp(fqdn, node, strlen(node)) && fqdn[strlen(node)] == '.')) { /* If it didn't work it means the host name IP is not in /etc/hosts, try again * by sending a NULL host and avoid sending a NULL servname too because that * is invalid */ ERR_(winediag)("Failed to resolve your host name IP\n"); result = getaddrinfo(NULL, servname ? servname : "0", punixhints, &unixaires); if (!result && punixhints && (punixhints->ai_flags & AI_CANONNAME) && unixaires && !unixaires->ai_canonname) { freeaddrinfo(unixaires); result = EAI_NONAME; } } } TRACE("%s, %s %p -> %p %d\n", debugstr_a(nodename), debugstr_a(servname), hints, res, result); HeapFree(GetProcessHeap(), 0, fqdn); HeapFree(GetProcessHeap(), 0, nodeV6); if (!result) { struct addrinfo *xuai = unixaires; struct WS_addrinfo **xai = res; *xai = NULL; while (xuai) { struct WS_addrinfo *ai = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY, sizeof(struct WS_addrinfo)); SIZE_T len; if (!ai) goto outofmem; ai->ai_flags = convert_aiflag_u2w(xuai->ai_flags); ai->ai_family = convert_af_u2w(xuai->ai_family); /* copy whatever was sent in the hints */ if(hints) { ai->ai_socktype = hints->ai_socktype; ai->ai_protocol = hints->ai_protocol; } else { ai->ai_socktype = convert_socktype_u2w(xuai->ai_socktype); ai->ai_protocol = convert_proto_u2w(xuai->ai_protocol); } if (xuai->ai_canonname) { TRACE("canon name - %s\n",debugstr_a(xuai->ai_canonname)); ai->ai_canonname = HeapAlloc(GetProcessHeap(),0,strlen(xuai->ai_canonname)+1); if (!ai->ai_canonname) goto outofmem; strcpy(ai->ai_canonname,xuai->ai_canonname); } len = xuai->ai_addrlen; ai->ai_addr = HeapAlloc(GetProcessHeap(),0,len); if (!ai->ai_addr) goto outofmem; ai->ai_addrlen = len; do { int winlen = ai->ai_addrlen; if (!ws_sockaddr_u2ws(xuai->ai_addr, ai->ai_addr, &winlen)) { ai->ai_addrlen = winlen; break; } len = 2*len; ai->ai_addr = HeapReAlloc(GetProcessHeap(),0,ai->ai_addr,len); if (!ai->ai_addr) goto outofmem; ai->ai_addrlen = len; } while (1); if (addrinfo_in_list(*res, ai)) { HeapFree(GetProcessHeap(), 0, ai->ai_canonname); HeapFree(GetProcessHeap(), 0, ai->ai_addr); HeapFree(GetProcessHeap(), 0, ai); } else { *xai = ai; xai = &ai->ai_next; } xuai = xuai->ai_next; } freeaddrinfo(unixaires); if (TRACE_ON(winsock)) { struct WS_addrinfo *ai = *res; while (ai) { TRACE("=> %p, flags %#x, family %d, type %d, protocol %d, len %ld, name %s, addr %s\n", ai, ai->ai_flags, ai->ai_family, ai->ai_socktype, ai->ai_protocol, ai->ai_addrlen, ai->ai_canonname, debugstr_sockaddr(ai->ai_addr)); ai = ai->ai_next; } } } else result = convert_eai_u2w(result); SetLastError(result); return result; outofmem: if (*res) WS_freeaddrinfo(*res); if (unixaires) freeaddrinfo(unixaires); return WSA_NOT_ENOUGH_MEMORY; #else FIXME("getaddrinfo() failed, not found during buildtime.\n"); return EAI_FAIL; #endif } static ADDRINFOEXW *addrinfo_AtoW(const struct WS_addrinfo *ai) { ADDRINFOEXW *ret; if (!(ret = HeapAlloc(GetProcessHeap(), 0, sizeof(ADDRINFOEXW)))) return NULL; ret->ai_flags = ai->ai_flags; ret->ai_family = ai->ai_family; ret->ai_socktype = ai->ai_socktype; ret->ai_protocol = ai->ai_protocol; ret->ai_addrlen = ai->ai_addrlen; ret->ai_canonname = NULL; ret->ai_addr = NULL; ret->ai_blob = NULL; ret->ai_bloblen = 0; ret->ai_provider = NULL; ret->ai_next = NULL; if (ai->ai_canonname) { int len = MultiByteToWideChar(CP_ACP, 0, ai->ai_canonname, -1, NULL, 0); if (!(ret->ai_canonname = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR)))) { HeapFree(GetProcessHeap(), 0, ret); return NULL; } MultiByteToWideChar(CP_ACP, 0, ai->ai_canonname, -1, ret->ai_canonname, len); } if (ai->ai_addr) { if (!(ret->ai_addr = HeapAlloc(GetProcessHeap(), 0, ai->ai_addrlen))) { HeapFree(GetProcessHeap(), 0, ret->ai_canonname); HeapFree(GetProcessHeap(), 0, ret); return NULL; } memcpy(ret->ai_addr, ai->ai_addr, ai->ai_addrlen); } return ret; } static ADDRINFOEXW *addrinfo_list_AtoW(const struct WS_addrinfo *info) { ADDRINFOEXW *ret, *infoW; if (!(ret = infoW = addrinfo_AtoW(info))) return NULL; while (info->ai_next) { if (!(infoW->ai_next = addrinfo_AtoW(info->ai_next))) { FreeAddrInfoExW(ret); return NULL; } infoW = infoW->ai_next; info = info->ai_next; } return ret; } static struct WS_addrinfo *addrinfo_WtoA(const struct WS_addrinfoW *ai) { struct WS_addrinfo *ret; if (!(ret = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_addrinfo)))) return NULL; ret->ai_flags = ai->ai_flags; ret->ai_family = ai->ai_family; ret->ai_socktype = ai->ai_socktype; ret->ai_protocol = ai->ai_protocol; ret->ai_addrlen = ai->ai_addrlen; ret->ai_canonname = NULL; ret->ai_addr = NULL; ret->ai_next = NULL; if (ai->ai_canonname) { int len = WideCharToMultiByte(CP_ACP, 0, ai->ai_canonname, -1, NULL, 0, NULL, NULL); if (!(ret->ai_canonname = HeapAlloc(GetProcessHeap(), 0, len))) { HeapFree(GetProcessHeap(), 0, ret); return NULL; } WideCharToMultiByte(CP_ACP, 0, ai->ai_canonname, -1, ret->ai_canonname, len, NULL, NULL); } if (ai->ai_addr) { if (!(ret->ai_addr = HeapAlloc(GetProcessHeap(), 0, sizeof(struct WS_sockaddr)))) { HeapFree(GetProcessHeap(), 0, ret->ai_canonname); HeapFree(GetProcessHeap(), 0, ret); return NULL; } memcpy(ret->ai_addr, ai->ai_addr, sizeof(struct WS_sockaddr)); } return ret; } struct getaddrinfo_args { OVERLAPPED *overlapped; LPLOOKUPSERVICE_COMPLETION_ROUTINE completion_routine; ADDRINFOEXW **result; char *nodename; char *servname; }; static void WINAPI getaddrinfo_callback(TP_CALLBACK_INSTANCE *instance, void *context) { struct getaddrinfo_args *args = context; OVERLAPPED *overlapped = args->overlapped; HANDLE event = overlapped->hEvent; LPLOOKUPSERVICE_COMPLETION_ROUTINE completion_routine = args->completion_routine; struct WS_addrinfo *res; int ret; ret = WS_getaddrinfo(args->nodename, args->servname, NULL, &res); if (res) { *args->result = addrinfo_list_AtoW(res); overlapped->u.Pointer = args->result; WS_freeaddrinfo(res); } HeapFree(GetProcessHeap(), 0, args->nodename); HeapFree(GetProcessHeap(), 0, args->servname); HeapFree(GetProcessHeap(), 0, args); overlapped->Internal = ret; if (completion_routine) completion_routine(ret, 0, overlapped); if (event) SetEvent(event); } static int WS_getaddrinfoW(const WCHAR *nodename, const WCHAR *servname, const struct WS_addrinfo *hints, ADDRINFOEXW **res, OVERLAPPED *overlapped, LPLOOKUPSERVICE_COMPLETION_ROUTINE completion_routine) { int ret = EAI_MEMORY, len, i; char *nodenameA = NULL, *servnameA = NULL; struct WS_addrinfo *resA; WCHAR *local_nodenameW = (WCHAR *)nodename; *res = NULL; if (nodename) { /* Is this an IDN? Most likely if any char is above the Ascii table, this * is the simplest validation possible, further validation will be done by * the native getaddrinfo() */ for (i = 0; nodename[i]; i++) { if (nodename[i] > 'z') break; } if (nodename[i]) { if (hints && (hints->ai_flags & WS_AI_DISABLE_IDN_ENCODING)) { /* Name requires conversion but it was disabled */ ret = WSAHOST_NOT_FOUND; WSASetLastError(ret); goto end; } len = IdnToAscii(0, nodename, -1, NULL, 0); if (!len) { ERR("Failed to convert %s to punycode\n", debugstr_w(nodename)); ret = EAI_FAIL; goto end; } if (!(local_nodenameW = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR)))) goto end; IdnToAscii(0, nodename, -1, local_nodenameW, len); } } if (local_nodenameW) { len = WideCharToMultiByte(CP_ACP, 0, local_nodenameW, -1, NULL, 0, NULL, NULL); if (!(nodenameA = HeapAlloc(GetProcessHeap(), 0, len))) goto end; WideCharToMultiByte(CP_ACP, 0, local_nodenameW, -1, nodenameA, len, NULL, NULL); } if (servname) { len = WideCharToMultiByte(CP_ACP, 0, servname, -1, NULL, 0, NULL, NULL); if (!(servnameA = HeapAlloc(GetProcessHeap(), 0, len))) goto end; WideCharToMultiByte(CP_ACP, 0, servname, -1, servnameA, len, NULL, NULL); } if (overlapped) { struct getaddrinfo_args *args; if (overlapped->hEvent && completion_routine) { ret = WSAEINVAL; goto end; } if (!(args = HeapAlloc(GetProcessHeap(), 0, sizeof(*args)))) goto end; args->overlapped = overlapped; args->completion_routine = completion_routine; args->result = res; args->nodename = nodenameA; args->servname = servnameA; overlapped->Internal = WSAEINPROGRESS; if (!TrySubmitThreadpoolCallback(getaddrinfo_callback, args, NULL)) { HeapFree(GetProcessHeap(), 0, args); ret = GetLastError(); goto end; } if (local_nodenameW != nodename) HeapFree(GetProcessHeap(), 0, local_nodenameW); WSASetLastError(ERROR_IO_PENDING); return ERROR_IO_PENDING; } ret = WS_getaddrinfo(nodenameA, servnameA, hints, &resA); if (!ret) { *res = addrinfo_list_AtoW(resA); WS_freeaddrinfo(resA); } end: if (local_nodenameW != nodename) HeapFree(GetProcessHeap(), 0, local_nodenameW); HeapFree(GetProcessHeap(), 0, nodenameA); HeapFree(GetProcessHeap(), 0, servnameA); return ret; } /*********************************************************************** * GetAddrInfoExW (WS2_32.@) */ int WINAPI GetAddrInfoExW(const WCHAR *name, const WCHAR *servname, DWORD namespace, GUID *namespace_id, const ADDRINFOEXW *hints, ADDRINFOEXW **result, struct WS_timeval *timeout, OVERLAPPED *overlapped, LPLOOKUPSERVICE_COMPLETION_ROUTINE completion_routine, HANDLE *handle) { int ret; TRACE("(%s %s %x %s %p %p %p %p %p %p)\n", debugstr_w(name), debugstr_w(servname), namespace, debugstr_guid(namespace_id), hints, result, timeout, overlapped, completion_routine, handle); if (namespace != NS_DNS) FIXME("Unsupported namespace %u\n", namespace); if (namespace_id) FIXME("Unsupported naemspace_id %s\n", debugstr_guid(namespace_id)); if (hints) FIXME("Unsupported hints\n"); if (timeout) FIXME("Unsupported timeout\n"); if (handle) FIXME("Unsupported cancel handle\n"); ret = WS_getaddrinfoW(name, servname, NULL, result, overlapped, completion_routine); if (ret) return ret; if (handle) *handle = (HANDLE)0xdeadbeef; return 0; } /*********************************************************************** * GetAddrInfoExOverlappedResult (WS2_32.@) */ int WINAPI GetAddrInfoExOverlappedResult(OVERLAPPED *overlapped) { TRACE("(%p)\n", overlapped); return overlapped->Internal; } /*********************************************************************** * GetAddrInfoExCancel (WS2_32.@) */ int WINAPI GetAddrInfoExCancel(HANDLE *handle) { FIXME("(%p)\n", handle); return WSA_INVALID_HANDLE; } /*********************************************************************** * GetAddrInfoW (WS2_32.@) */ int WINAPI GetAddrInfoW(LPCWSTR nodename, LPCWSTR servname, const ADDRINFOW *hints, PADDRINFOW *res) { struct WS_addrinfo *hintsA = NULL; ADDRINFOEXW *resex; int ret = EAI_MEMORY; TRACE("nodename %s, servname %s, hints %p, result %p\n", debugstr_w(nodename), debugstr_w(servname), hints, res); *res = NULL; if (hints) hintsA = addrinfo_WtoA(hints); ret = WS_getaddrinfoW(nodename, servname, hintsA, &resex, NULL, NULL); WS_freeaddrinfo(hintsA); if (ret) return ret; if (resex) { /* ADDRINFOEXW has layout compatible with ADDRINFOW except for ai_next field, * so we may convert it in place */ *res = (ADDRINFOW*)resex; do { ((ADDRINFOW*)resex)->ai_next = (ADDRINFOW*)resex->ai_next; resex = resex->ai_next; } while (resex); } return 0; } /*********************************************************************** * FreeAddrInfoW (WS2_32.@) */ void WINAPI FreeAddrInfoW(PADDRINFOW ai) { while (ai) { ADDRINFOW *next; HeapFree(GetProcessHeap(), 0, ai->ai_canonname); HeapFree(GetProcessHeap(), 0, ai->ai_addr); next = ai->ai_next; HeapFree(GetProcessHeap(), 0, ai); ai = next; } } /*********************************************************************** * FreeAddrInfoEx (WS2_32.@) */ void WINAPI FreeAddrInfoEx(ADDRINFOEXA *ai) { TRACE("(%p)\n", ai); while (ai) { ADDRINFOEXA *next; HeapFree(GetProcessHeap(), 0, ai->ai_canonname); HeapFree(GetProcessHeap(), 0, ai->ai_addr); next = ai->ai_next; HeapFree(GetProcessHeap(), 0, ai); ai = next; } } /*********************************************************************** * FreeAddrInfoExW (WS2_32.@) */ void WINAPI FreeAddrInfoExW(ADDRINFOEXW *ai) { TRACE("(%p)\n", ai); while (ai) { ADDRINFOEXW *next; HeapFree(GetProcessHeap(), 0, ai->ai_canonname); HeapFree(GetProcessHeap(), 0, ai->ai_addr); next = ai->ai_next; HeapFree(GetProcessHeap(), 0, ai); ai = next; } } int WINAPI WS_getnameinfo(const SOCKADDR *sa, WS_socklen_t salen, PCHAR host, DWORD hostlen, PCHAR serv, DWORD servlen, INT flags) { #ifdef HAVE_GETNAMEINFO int ret; union generic_unix_sockaddr sa_u; unsigned int size; TRACE("%s %d %p %d %p %d %d\n", debugstr_sockaddr(sa), salen, host, hostlen, serv, servlen, flags); size = ws_sockaddr_ws2u(sa, salen, &sa_u); if (!size) { SetLastError(WSAEFAULT); return WSA_NOT_ENOUGH_MEMORY; } ret = getnameinfo(&sa_u.addr, size, host, hostlen, serv, servlen, convert_niflag_w2u(flags)); return convert_eai_u2w(ret); #else FIXME("getnameinfo() failed, not found during buildtime.\n"); return EAI_FAIL; #endif } int WINAPI GetNameInfoW(const SOCKADDR *sa, WS_socklen_t salen, PWCHAR host, DWORD hostlen, PWCHAR serv, DWORD servlen, INT flags) { int ret; char *hostA = NULL, *servA = NULL; if (host && (!(hostA = HeapAlloc(GetProcessHeap(), 0, hostlen)))) return EAI_MEMORY; if (serv && (!(servA = HeapAlloc(GetProcessHeap(), 0, servlen)))) { HeapFree(GetProcessHeap(), 0, hostA); return EAI_MEMORY; } ret = WS_getnameinfo(sa, salen, hostA, hostlen, servA, servlen, flags); if (!ret) { if (host) MultiByteToWideChar(CP_ACP, 0, hostA, -1, host, hostlen); if (serv) MultiByteToWideChar(CP_ACP, 0, servA, -1, serv, servlen); } HeapFree(GetProcessHeap(), 0, hostA); HeapFree(GetProcessHeap(), 0, servA); return ret; } /*********************************************************************** * getservbyport (WS2_32.56) */ struct WS_servent* WINAPI WS_getservbyport(int port, const char *proto) { struct WS_servent* retval = NULL; #ifdef HAVE_GETSERVBYPORT struct servent* serv; char *proto_str = NULL; if (proto && *proto) { if (!(proto_str = strdup_lower(proto))) return NULL; } EnterCriticalSection( &csWSgetXXXbyYYY ); if( (serv = getservbyport(port, proto_str)) != NULL ) { retval = WS_dup_se(serv); } else SetLastError(WSANO_DATA); LeaveCriticalSection( &csWSgetXXXbyYYY ); HeapFree( GetProcessHeap(), 0, proto_str ); #endif TRACE("%d (i.e. port %d), %s ret %p\n", port, (int)ntohl(port), debugstr_a(proto), retval); return retval; } /*********************************************************************** * gethostname (WS2_32.57) */ int WINAPI WS_gethostname(char *name, int namelen) { char buf[256]; int len; TRACE("name %p, len %d\n", name, namelen); if (!name) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } if (gethostname(buf, sizeof(buf)) != 0) { SetLastError(wsaErrno()); return SOCKET_ERROR; } TRACE("<- '%s'\n", buf); len = strlen(buf); if (len > 15) WARN("Windows supports NetBIOS name length up to 15 bytes!\n"); if (namelen <= len) { SetLastError(WSAEFAULT); WARN("<- not enough space for hostname, required %d, got %d!\n", len + 1, namelen); return SOCKET_ERROR; } strcpy(name, buf); return 0; } /* ------------------------------------- Windows sockets extensions -- * * * * ------------------------------------------------------------------- */ /*********************************************************************** * WSAEnumNetworkEvents (WS2_32.36) */ int WINAPI WSAEnumNetworkEvents(SOCKET s, WSAEVENT hEvent, LPWSANETWORKEVENTS lpEvent) { int ret; int i; int errors[FD_MAX_EVENTS]; TRACE("%04lx, hEvent %p, lpEvent %p\n", s, hEvent, lpEvent ); SERVER_START_REQ( get_socket_event ) { req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) ); req->service = TRUE; req->c_event = wine_server_obj_handle( hEvent ); wine_server_set_reply( req, errors, sizeof(errors) ); if (!(ret = wine_server_call(req))) lpEvent->lNetworkEvents = reply->pmask & reply->mask; } SERVER_END_REQ; if (!ret) { for (i = 0; i < FD_MAX_EVENTS; i++) { if (lpEvent->lNetworkEvents & (1 << i)) lpEvent->iErrorCode[i] = errors[i]; } return 0; } SetLastError(WSAEINVAL); return SOCKET_ERROR; } /*********************************************************************** * WSAEventSelect (WS2_32.39) */ int WINAPI WSAEventSelect(SOCKET s, WSAEVENT hEvent, LONG lEvent) { int ret; TRACE("%04lx, hEvent %p, event %08x\n", s, hEvent, lEvent); SERVER_START_REQ( set_socket_event ) { req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) ); req->mask = lEvent; req->event = wine_server_obj_handle( hEvent ); req->window = 0; req->msg = 0; ret = wine_server_call( req ); } SERVER_END_REQ; if (!ret) return 0; SetLastError(WSAEINVAL); return SOCKET_ERROR; } /********************************************************************** * WSAGetOverlappedResult (WS2_32.40) */ BOOL WINAPI WSAGetOverlappedResult( SOCKET s, LPWSAOVERLAPPED lpOverlapped, LPDWORD lpcbTransfer, BOOL fWait, LPDWORD lpdwFlags ) { NTSTATUS status; TRACE( "socket %04lx ovl %p trans %p, wait %d flags %p\n", s, lpOverlapped, lpcbTransfer, fWait, lpdwFlags ); if ( lpOverlapped == NULL ) { ERR( "Invalid pointer\n" ); SetLastError(WSA_INVALID_PARAMETER); return FALSE; } status = lpOverlapped->Internal; if (status == STATUS_PENDING) { if (!fWait) { SetLastError( WSA_IO_INCOMPLETE ); return FALSE; } if (WaitForSingleObject( lpOverlapped->hEvent ? lpOverlapped->hEvent : SOCKET2HANDLE(s), INFINITE ) == WAIT_FAILED) return FALSE; status = lpOverlapped->Internal; } if ( lpcbTransfer ) *lpcbTransfer = lpOverlapped->InternalHigh; if ( lpdwFlags ) *lpdwFlags = lpOverlapped->u.s.Offset; if (status) SetLastError( RtlNtStatusToDosError(status) ); return !status; } /*********************************************************************** * WSAAsyncSelect (WS2_32.101) */ INT WINAPI WSAAsyncSelect(SOCKET s, HWND hWnd, UINT uMsg, LONG lEvent) { int ret; TRACE("%04lx, hWnd %p, uMsg %08x, event %08x\n", s, hWnd, uMsg, lEvent); SERVER_START_REQ( set_socket_event ) { req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) ); req->mask = lEvent; req->event = 0; req->window = wine_server_user_handle( hWnd ); req->msg = uMsg; ret = wine_server_call( req ); } SERVER_END_REQ; if (!ret) return 0; SetLastError(WSAEINVAL); return SOCKET_ERROR; } /*********************************************************************** * WSACreateEvent (WS2_32.31) * */ WSAEVENT WINAPI WSACreateEvent(void) { /* Create a manual-reset event, with initial state: unsignaled */ TRACE("\n"); return CreateEventW(NULL, TRUE, FALSE, NULL); } /*********************************************************************** * WSACloseEvent (WS2_32.29) * */ BOOL WINAPI WSACloseEvent(WSAEVENT event) { TRACE ("event=%p\n", event); return CloseHandle(event); } /*********************************************************************** * WSASocketA (WS2_32.78) * */ SOCKET WINAPI WSASocketA(int af, int type, int protocol, LPWSAPROTOCOL_INFOA lpProtocolInfo, GROUP g, DWORD dwFlags) { INT len; WSAPROTOCOL_INFOW info; TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%x\n", af, type, protocol, lpProtocolInfo, g, dwFlags); if (!lpProtocolInfo) return WSASocketW(af, type, protocol, NULL, g, dwFlags); memcpy(&info, lpProtocolInfo, FIELD_OFFSET(WSAPROTOCOL_INFOW, szProtocol)); len = MultiByteToWideChar(CP_ACP, 0, lpProtocolInfo->szProtocol, -1, info.szProtocol, WSAPROTOCOL_LEN + 1); if (!len) { SetLastError(WSAEINVAL); return SOCKET_ERROR; } return WSASocketW(af, type, protocol, &info, g, dwFlags); } /*********************************************************************** * WSASocketW (WS2_32.79) * */ SOCKET WINAPI WSASocketW(int af, int type, int protocol, LPWSAPROTOCOL_INFOW lpProtocolInfo, GROUP g, DWORD flags) { static const WCHAR afdW[] = {'\\','D','e','v','i','c','e','\\','A','f','d',0}; struct afd_create_params create_params; OBJECT_ATTRIBUTES attr; UNICODE_STRING string; IO_STATUS_BLOCK io; NTSTATUS status; HANDLE handle; SOCKET ret; DWORD err; /* FIXME: The "advanced" parameters of WSASocketW (lpProtocolInfo, g, dwFlags except WSA_FLAG_OVERLAPPED) are ignored. */ TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%x\n", af, type, protocol, lpProtocolInfo, g, flags ); if (!num_startup) { err = WSANOTINITIALISED; goto done; } /* hack for WSADuplicateSocket */ if (lpProtocolInfo && lpProtocolInfo->dwServiceFlags4 == 0xff00ff00) { ret = lpProtocolInfo->dwServiceFlags3; TRACE("\tgot duplicate %04lx\n", ret); if (!socket_list_add(ret)) { CloseHandle(SOCKET2HANDLE(ret)); return INVALID_SOCKET; } return ret; } if (lpProtocolInfo) { if (af == FROM_PROTOCOL_INFO || !af) af = lpProtocolInfo->iAddressFamily; if (type == FROM_PROTOCOL_INFO || !type) type = lpProtocolInfo->iSocketType; if (protocol == FROM_PROTOCOL_INFO || !protocol) protocol = lpProtocolInfo->iProtocol; } if (!af && !protocol) { WSASetLastError(WSAEINVAL); return INVALID_SOCKET; } if (!af && lpProtocolInfo) { WSASetLastError(WSAEAFNOSUPPORT); return INVALID_SOCKET; } if (!af || !type || !protocol) { unsigned int i; for (i = 0; i < ARRAY_SIZE(supported_protocols); ++i) { const WSAPROTOCOL_INFOW *info = &supported_protocols[i]; if (af && af != info->iAddressFamily) continue; if (type && type != info->iSocketType) continue; if (protocol && (protocol < info->iProtocol || protocol > info->iProtocol + info->iProtocolMaxOffset)) continue; if (!protocol && !(info->dwProviderFlags & PFL_MATCHES_PROTOCOL_ZERO)) continue; if (!af) af = supported_protocols[i].iAddressFamily; if (!type) type = supported_protocols[i].iSocketType; if (!protocol) protocol = supported_protocols[i].iProtocol; } } RtlInitUnicodeString(&string, afdW); InitializeObjectAttributes(&attr, &string, (flags & WSA_FLAG_NO_HANDLE_INHERIT) ? 0 : OBJ_INHERIT, NULL, NULL); if ((status = NtOpenFile(&handle, GENERIC_READ | GENERIC_WRITE | SYNCHRONIZE, &attr, &io, 0, (flags & WSA_FLAG_OVERLAPPED) ? 0 : FILE_SYNCHRONOUS_IO_NONALERT))) { WARN("Failed to create socket, status %#x.\n", status); WSASetLastError(NtStatusToWSAError(status)); return INVALID_SOCKET; } create_params.family = af; create_params.type = type; create_params.protocol = protocol; create_params.flags = flags & ~(WSA_FLAG_NO_HANDLE_INHERIT | WSA_FLAG_OVERLAPPED); if ((status = NtDeviceIoControlFile(handle, NULL, NULL, NULL, &io, IOCTL_AFD_CREATE, &create_params, sizeof(create_params), NULL, 0))) { WARN("Failed to initialize socket, status %#x.\n", status); err = NtStatusToWSAError(status); if (err == WSAEACCES) /* raw socket denied */ { if (type == SOCK_RAW) ERR_(winediag)("Failed to create a socket of type SOCK_RAW, this requires special permissions.\n"); else ERR_(winediag)("Failed to create socket, this requires special permissions.\n"); } WSASetLastError(err); NtClose(handle); return INVALID_SOCKET; } ret = HANDLE2SOCKET(handle); TRACE("\tcreated %04lx\n", ret ); if (!socket_list_add(ret)) { CloseHandle(handle); return INVALID_SOCKET; } return ret; done: WARN("\t\tfailed, error %d!\n", err); SetLastError(err); return INVALID_SOCKET; } /*********************************************************************** * WSAJoinLeaf (WS2_32.58) * */ SOCKET WINAPI WSAJoinLeaf( SOCKET s, const struct WS_sockaddr *addr, int addrlen, LPWSABUF lpCallerData, LPWSABUF lpCalleeData, LPQOS lpSQOS, LPQOS lpGQOS, DWORD dwFlags) { FIXME("stub.\n"); return INVALID_SOCKET; } /*********************************************************************** * __WSAFDIsSet (WS2_32.151) */ int WINAPI __WSAFDIsSet(SOCKET s, WS_fd_set *set) { int i = set->fd_count, ret = 0; while (i--) if (set->fd_array[i] == s) { ret = 1; break; } TRACE("(socket %04lx, fd_set %p, count %i) <- %d\n", s, set, set->fd_count, ret); return ret; } /*********************************************************************** * WSAIsBlocking (WS2_32.114) */ BOOL WINAPI WSAIsBlocking(void) { /* By default WinSock should set all its sockets to non-blocking mode * and poll in PeekMessage loop when processing "blocking" ones. This * function is supposed to tell if the program is in this loop. Our * blocking calls are truly blocking so we always return FALSE. * * Note: It is allowed to call this function without prior WSAStartup(). */ TRACE("\n"); return FALSE; } /*********************************************************************** * WSACancelBlockingCall (WS2_32.113) */ INT WINAPI WSACancelBlockingCall(void) { TRACE("\n"); return 0; } static INT WINAPI WSA_DefaultBlockingHook( FARPROC x ) { FIXME("How was this called?\n"); return x(); } /*********************************************************************** * WSASetBlockingHook (WS2_32.109) */ FARPROC WINAPI WSASetBlockingHook(FARPROC lpBlockFunc) { FARPROC prev = blocking_hook; blocking_hook = lpBlockFunc; TRACE("hook %p\n", lpBlockFunc); return prev; } /*********************************************************************** * WSAUnhookBlockingHook (WS2_32.110) */ INT WINAPI WSAUnhookBlockingHook(void) { blocking_hook = (FARPROC)WSA_DefaultBlockingHook; return 0; } /* ----------------------------------- end of API stuff */ /* ----------------------------------- helper functions - * * TODO: Merge WS_dup_..() stuff into one function that * would operate with a generic structure containing internal * pointers (via a template of some kind). */ static int list_size(char** l, int item_size) { int i,j = 0; if(l) { for(i=0;l[i];i++) j += (item_size) ? item_size : strlen(l[i]) + 1; j += (i + 1) * sizeof(char*); } return j; } static int list_dup(char** l_src, char** l_to, int item_size) { char *p; int i; for (i = 0; l_src[i]; i++) ; p = (char *)(l_to + i + 1); for (i = 0; l_src[i]; i++) { int count = ( item_size ) ? item_size : strlen(l_src[i]) + 1; memcpy(p, l_src[i], count); l_to[i] = p; p += count; } l_to[i] = NULL; return p - (char *)l_to; } /* ----- hostent */ /* create a hostent entry * * Creates the entry with enough memory for the name, aliases * addresses, and the address pointers. Also copies the name * and sets up all the pointers. * * NOTE: The alias and address lists must be allocated with room * for the NULL item terminating the list. This is true even if * the list has no items ("aliases" and "addresses" must be * at least "1", a truly empty list is invalid). */ static struct WS_hostent *WS_create_he(char *name, int aliases, int aliases_size, int addresses, int address_length) { struct WS_hostent *p_to; char *p; int size = (sizeof(struct WS_hostent) + strlen(name) + 1 + sizeof(char *) * aliases + aliases_size + sizeof(char *) * addresses + address_length * (addresses - 1)), i; if (!(p_to = check_buffer_he(size))) return NULL; memset(p_to, 0, size); /* Use the memory in the same way winsock does. * First set the pointer for aliases, second set the pointers for addresses. * Third fill the addresses indexes, fourth jump aliases names size. * Fifth fill the hostname. * NOTE: This method is valid for OS versions >= XP. */ p = (char *)(p_to + 1); p_to->h_aliases = (char **)p; p += sizeof(char *)*aliases; p_to->h_addr_list = (char **)p; p += sizeof(char *)*addresses; for (i = 0, addresses--; i < addresses; i++, p += address_length) p_to->h_addr_list[i] = p; /* NOTE: h_aliases must be filled in manually because we don't know each string * size, leave these pointers NULL (already set to NULL by memset earlier). */ p += aliases_size; p_to->h_name = p; strcpy(p, name); return p_to; } /* duplicate hostent entry * and handle all Win16/Win32 dependent things (struct size, ...) *correctly*. * Ditto for protoent and servent. */ static struct WS_hostent *WS_dup_he(const struct hostent* p_he) { int i, addresses = 0, alias_size = 0; struct WS_hostent *p_to; char *p; for( i = 0; p_he->h_aliases[i]; i++) alias_size += strlen(p_he->h_aliases[i]) + 1; while (p_he->h_addr_list[addresses]) addresses++; p_to = WS_create_he(p_he->h_name, i + 1, alias_size, addresses + 1, p_he->h_length); if (!p_to) return NULL; p_to->h_addrtype = convert_af_u2w(p_he->h_addrtype); p_to->h_length = p_he->h_length; for(i = 0, p = p_to->h_addr_list[0]; p_he->h_addr_list[i]; i++, p += p_to->h_length) memcpy(p, p_he->h_addr_list[i], p_to->h_length); /* Fill the aliases after the IP data */ for(i = 0; p_he->h_aliases[i]; i++) { p_to->h_aliases[i] = p; strcpy(p, p_he->h_aliases[i]); p += strlen(p) + 1; } return p_to; } /* ----- protoent */ static struct WS_protoent *WS_create_pe( const char *name, char **aliases, int prot ) { struct WS_protoent *ret; unsigned int size = sizeof(*ret) + strlen(name) + sizeof(char *) + list_size(aliases, 0); if (!(ret = check_buffer_pe( size ))) return NULL; ret->p_proto = prot; ret->p_name = (char *)(ret + 1); strcpy( ret->p_name, name ); ret->p_aliases = (char **)ret->p_name + strlen(name) / sizeof(char *) + 1; list_dup( aliases, ret->p_aliases, 0 ); return ret; } /* ----- servent */ static struct WS_servent *WS_dup_se(const struct servent* p_se) { char *p; struct WS_servent *p_to; int size = (sizeof(*p_se) + strlen(p_se->s_proto) + 1 + strlen(p_se->s_name) + 1 + list_size(p_se->s_aliases, 0)); if (!(p_to = check_buffer_se(size))) return NULL; p_to->s_port = p_se->s_port; p = (char *)(p_to + 1); p_to->s_name = p; strcpy(p, p_se->s_name); p += strlen(p) + 1; p_to->s_proto = p; strcpy(p, p_se->s_proto); p += strlen(p) + 1; p_to->s_aliases = (char **)p; list_dup(p_se->s_aliases, p_to->s_aliases, 0); return p_to; } /*********************************************************************** * WSARecv (WS2_32.67) */ int WINAPI WSARecv(SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount, LPDWORD NumberOfBytesReceived, LPDWORD lpFlags, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine) { return WS2_recv_base(s, lpBuffers, dwBufferCount, NumberOfBytesReceived, lpFlags, NULL, NULL, lpOverlapped, lpCompletionRoutine, NULL); } static int WS2_recv_base( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount, LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags, struct WS_sockaddr *lpFrom, LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine, LPWSABUF lpControlBuffer ) { unsigned int i, options; int n, fd, err, overlapped, flags; struct ws2_async *wsa = NULL, localwsa; BOOL is_blocking; DWORD timeout_start = GetTickCount(); ULONG_PTR cvalue = (lpOverlapped && ((ULONG_PTR)lpOverlapped->hEvent & 1) == 0) ? (ULONG_PTR)lpOverlapped : 0; TRACE("socket %04lx, wsabuf %p, nbufs %d, flags %d, from %p, fromlen %d, ovl %p, func %p\n", s, lpBuffers, dwBufferCount, *lpFlags, lpFrom, (lpFromlen ? *lpFromlen : -1), lpOverlapped, lpCompletionRoutine); fd = get_sock_fd( s, FILE_READ_DATA, &options ); TRACE( "fd=%d, options=%x\n", fd, options ); if (fd == -1) return SOCKET_ERROR; if (*lpFlags & WS_MSG_OOB) { /* It's invalid to receive OOB data from an OOBINLINED socket * as OOB data is turned into normal data. */ socklen_t len = sizeof(n); if (!getsockopt(fd, SOL_SOCKET, SO_OOBINLINE, (char*) &n, &len) && n) { err = WSAEINVAL; goto error; } } overlapped = (lpOverlapped || lpCompletionRoutine) && !(options & (FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT)); if (overlapped || dwBufferCount > 1) { if (!(wsa = (struct ws2_async *)alloc_async_io( offsetof(struct ws2_async, iovec[dwBufferCount]), WS2_async_recv ))) { err = WSAEFAULT; goto error; } } else wsa = &localwsa; wsa->hSocket = SOCKET2HANDLE(s); wsa->flags = *lpFlags; wsa->lpFlags = lpFlags; wsa->addr = lpFrom; wsa->addrlen.ptr = lpFromlen; wsa->control = lpControlBuffer; wsa->n_iovecs = dwBufferCount; wsa->first_iovec = 0; for (i = 0; i < dwBufferCount; i++) { /* check buffer first to trigger write watches */ if (IsBadWritePtr( lpBuffers[i].buf, lpBuffers[i].len )) { err = WSAEFAULT; goto error; } wsa->iovec[i].iov_base = lpBuffers[i].buf; wsa->iovec[i].iov_len = lpBuffers[i].len; } flags = convert_flags(wsa->flags); for (;;) { n = WS2_recv( fd, wsa, flags ); if (n == -1) { /* Unix-like systems return EINVAL when attempting to read OOB data from * an empty socket buffer, convert that to a Windows expected return. */ if ((flags & MSG_OOB) && errno == EINVAL) errno = EWOULDBLOCK; if (errno != EAGAIN) { err = wsaErrno(); goto error; } } else if (lpNumberOfBytesRecvd) *lpNumberOfBytesRecvd = n; if (overlapped) { IO_STATUS_BLOCK *iosb = lpOverlapped ? (IO_STATUS_BLOCK *)lpOverlapped : &wsa->local_iosb; wsa->user_overlapped = lpOverlapped; wsa->completion_func = lpCompletionRoutine; release_sock_fd( s, fd ); if (n == -1) { iosb->u.Status = STATUS_PENDING; iosb->Information = 0; if (wsa->completion_func) err = register_async( ASYNC_TYPE_READ, wsa->hSocket, &wsa->io, NULL, ws2_async_apc, wsa, iosb ); else err = register_async( ASYNC_TYPE_READ, wsa->hSocket, &wsa->io, lpOverlapped->hEvent, NULL, (void *)cvalue, iosb ); if (err != STATUS_PENDING) HeapFree( GetProcessHeap(), 0, wsa ); SetLastError(NtStatusToWSAError( err )); return SOCKET_ERROR; } iosb->u.Status = STATUS_SUCCESS; iosb->Information = n; if (!wsa->completion_func) { if (cvalue) WS_AddCompletion( s, cvalue, STATUS_SUCCESS, n, FALSE ); if (lpOverlapped->hEvent) SetEvent( lpOverlapped->hEvent ); HeapFree( GetProcessHeap(), 0, wsa ); } else NtQueueApcThread( GetCurrentThread(), (PNTAPCFUNC)ws2_async_apc, (ULONG_PTR)wsa, (ULONG_PTR)iosb, 0 ); _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0); return 0; } if (n != -1) break; if ((err = sock_is_blocking( s, &is_blocking ))) goto error; if ( is_blocking ) { struct pollfd pfd; int poll_timeout = -1; INT64 timeout = get_rcvsnd_timeo(fd, TRUE); if (timeout) { timeout -= GetTickCount() - timeout_start; if (timeout < 0) poll_timeout = 0; else poll_timeout = timeout <= INT_MAX ? timeout : INT_MAX; } pfd.fd = fd; pfd.events = POLLIN; if (*lpFlags & WS_MSG_OOB) pfd.events |= POLLPRI; if (!poll_timeout || !poll( &pfd, 1, poll_timeout )) { err = WSAETIMEDOUT; /* a timeout is not fatal */ _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0); goto error; } } else { _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0); err = WSAEWOULDBLOCK; goto error; } } TRACE(" -> %i bytes\n", n); if (wsa != &localwsa) HeapFree( GetProcessHeap(), 0, wsa ); release_sock_fd( s, fd ); _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0); SetLastError(ERROR_SUCCESS); return 0; error: if (wsa != &localwsa) HeapFree( GetProcessHeap(), 0, wsa ); release_sock_fd( s, fd ); WARN(" -> ERROR %d\n", err); SetLastError( err ); return SOCKET_ERROR; } /*********************************************************************** * WSARecvFrom (WS2_32.69) */ INT WINAPI WSARecvFrom( SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount, LPDWORD lpNumberOfBytesRecvd, LPDWORD lpFlags, struct WS_sockaddr *lpFrom, LPINT lpFromlen, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine ) { return WS2_recv_base( s, lpBuffers, dwBufferCount, lpNumberOfBytesRecvd, lpFlags, lpFrom, lpFromlen, lpOverlapped, lpCompletionRoutine, NULL ); } /*********************************************************************** * WSCInstallProvider (WS2_32.88) */ INT WINAPI WSCInstallProvider( const LPGUID lpProviderId, LPCWSTR lpszProviderDllPath, const LPWSAPROTOCOL_INFOW lpProtocolInfoList, DWORD dwNumberOfEntries, LPINT lpErrno ) { FIXME("(%s, %s, %p, %d, %p): stub !\n", debugstr_guid(lpProviderId), debugstr_w(lpszProviderDllPath), lpProtocolInfoList, dwNumberOfEntries, lpErrno); *lpErrno = 0; return 0; } /*********************************************************************** * WSCDeinstallProvider (WS2_32.83) */ INT WINAPI WSCDeinstallProvider(LPGUID lpProviderId, LPINT lpErrno) { FIXME("(%s, %p): stub !\n", debugstr_guid(lpProviderId), lpErrno); *lpErrno = 0; return 0; } /*********************************************************************** * WSAAccept (WS2_32.26) */ SOCKET WINAPI WSAAccept( SOCKET s, struct WS_sockaddr *addr, LPINT addrlen, LPCONDITIONPROC lpfnCondition, DWORD_PTR dwCallbackData) { int ret = 0, size; WSABUF CallerId, CallerData, CalleeId, CalleeData; /* QOS SQOS, GQOS; */ GROUP g; SOCKET cs; SOCKADDR src_addr, dst_addr; TRACE("socket %04lx, sockaddr %p, addrlen %p, fnCondition %p, dwCallbackData %ld\n", s, addr, addrlen, lpfnCondition, dwCallbackData); cs = WS_accept(s, addr, addrlen); if (cs == SOCKET_ERROR) return SOCKET_ERROR; if (!lpfnCondition) return cs; if (addr && addrlen) { CallerId.buf = (char *)addr; CallerId.len = *addrlen; } else { size = sizeof(src_addr); WS_getpeername(cs, &src_addr, &size); CallerId.buf = (char *)&src_addr; CallerId.len = size; } CallerData.buf = NULL; CallerData.len = 0; size = sizeof(dst_addr); WS_getsockname(cs, &dst_addr, &size); CalleeId.buf = (char *)&dst_addr; CalleeId.len = sizeof(dst_addr); ret = (*lpfnCondition)(&CallerId, &CallerData, NULL, NULL, &CalleeId, &CalleeData, &g, dwCallbackData); switch (ret) { case CF_ACCEPT: return cs; case CF_DEFER: SERVER_START_REQ( set_socket_deferred ) { req->handle = wine_server_obj_handle( SOCKET2HANDLE(s) ); req->deferred = wine_server_obj_handle( SOCKET2HANDLE(cs) ); if ( !wine_server_call_err ( req ) ) { SetLastError( WSATRY_AGAIN ); WS_closesocket( cs ); } } SERVER_END_REQ; return SOCKET_ERROR; case CF_REJECT: WS_closesocket(cs); SetLastError(WSAECONNREFUSED); return SOCKET_ERROR; default: FIXME("Unknown return type from Condition function\n"); SetLastError(WSAENOTSOCK); return SOCKET_ERROR; } } /*********************************************************************** * WSADuplicateSocketA (WS2_32.32) */ int WINAPI WSADuplicateSocketA( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOA lpProtocolInfo ) { return WS_DuplicateSocket(FALSE, s, dwProcessId, (LPWSAPROTOCOL_INFOW) lpProtocolInfo); } /*********************************************************************** * WSADuplicateSocketW (WS2_32.33) */ int WINAPI WSADuplicateSocketW( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOW lpProtocolInfo ) { return WS_DuplicateSocket(TRUE, s, dwProcessId, lpProtocolInfo); } /*********************************************************************** * WSAInstallServiceClassA (WS2_32.48) */ int WINAPI WSAInstallServiceClassA(LPWSASERVICECLASSINFOA info) { FIXME("Request to install service %s\n",debugstr_a(info->lpszServiceClassName)); SetLastError(WSAEACCES); return SOCKET_ERROR; } /*********************************************************************** * WSAInstallServiceClassW (WS2_32.49) */ int WINAPI WSAInstallServiceClassW(LPWSASERVICECLASSINFOW info) { FIXME("Request to install service %s\n",debugstr_w(info->lpszServiceClassName)); SetLastError(WSAEACCES); return SOCKET_ERROR; } /*********************************************************************** * WSARemoveServiceClass (WS2_32.70) */ int WINAPI WSARemoveServiceClass(LPGUID info) { FIXME("Request to remove service %s\n", debugstr_guid(info)); SetLastError(WSATYPE_NOT_FOUND); return SOCKET_ERROR; } /*********************************************************************** * inet_ntop (WS2_32.@) */ PCSTR WINAPI WS_inet_ntop( INT family, PVOID addr, PSTR buffer, SIZE_T len ) { NTSTATUS status; ULONG size = min( len, (ULONG)-1 ); TRACE("family %d, addr (%p), buffer (%p), len %ld\n", family, addr, buffer, len); if (!buffer) { SetLastError( STATUS_INVALID_PARAMETER ); return NULL; } switch (family) { case WS_AF_INET: { status = RtlIpv4AddressToStringExA( (IN_ADDR *)addr, 0, buffer, &size ); break; } case WS_AF_INET6: { status = RtlIpv6AddressToStringExA( (IN6_ADDR *)addr, 0, 0, buffer, &size ); break; } default: SetLastError( WSAEAFNOSUPPORT ); return NULL; } if (status == STATUS_SUCCESS) return buffer; SetLastError( STATUS_INVALID_PARAMETER ); return NULL; } /*********************************************************************** * inet_pton (WS2_32.@) */ INT WINAPI WS_inet_pton(INT family, const char *addr, void *buffer) { NTSTATUS status; const char *terminator; TRACE("family %d, addr %s, buffer (%p)\n", family, debugstr_a(addr), buffer); if (!addr || !buffer) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } switch (family) { case WS_AF_INET: status = RtlIpv4StringToAddressA(addr, TRUE, &terminator, buffer); break; case WS_AF_INET6: status = RtlIpv6StringToAddressA(addr, &terminator, buffer); break; default: SetLastError(WSAEAFNOSUPPORT); return SOCKET_ERROR; } return (status == STATUS_SUCCESS && *terminator == 0); } /*********************************************************************** * InetPtonW (WS2_32.@) */ INT WINAPI InetPtonW(INT family, PCWSTR addr, PVOID buffer) { char *addrA; int len; INT ret; TRACE("family %d, addr %s, buffer (%p)\n", family, debugstr_w(addr), buffer); if (!addr) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } len = WideCharToMultiByte(CP_ACP, 0, addr, -1, NULL, 0, NULL, NULL); if (!(addrA = HeapAlloc(GetProcessHeap(), 0, len))) { SetLastError(WSA_NOT_ENOUGH_MEMORY); return SOCKET_ERROR; } WideCharToMultiByte(CP_ACP, 0, addr, -1, addrA, len, NULL, NULL); ret = WS_inet_pton(family, addrA, buffer); if (!ret) SetLastError(WSAEINVAL); HeapFree(GetProcessHeap(), 0, addrA); return ret; } /*********************************************************************** * InetNtopW (WS2_32.@) */ PCWSTR WINAPI InetNtopW(INT family, PVOID addr, PWSTR buffer, SIZE_T len) { char bufferA[WS_INET6_ADDRSTRLEN]; PWSTR ret = NULL; TRACE("family %d, addr (%p), buffer (%p), len %ld\n", family, addr, buffer, len); if (WS_inet_ntop(family, addr, bufferA, sizeof(bufferA))) { if (MultiByteToWideChar(CP_ACP, 0, bufferA, -1, buffer, len)) ret = buffer; else SetLastError(ERROR_INVALID_PARAMETER); } return ret; } /*********************************************************************** * WSAStringToAddressA (WS2_32.80) */ INT WINAPI WSAStringToAddressA(LPSTR AddressString, INT AddressFamily, LPWSAPROTOCOL_INFOA lpProtocolInfo, LPSOCKADDR lpAddress, LPINT lpAddressLength) { INT res=0; NTSTATUS status; TRACE( "(%s, %x, %p, %p, %p)\n", debugstr_a(AddressString), AddressFamily, lpProtocolInfo, lpAddress, lpAddressLength ); if (!lpAddressLength || !lpAddress) return SOCKET_ERROR; if (!AddressString) { SetLastError(WSAEINVAL); return SOCKET_ERROR; } if (lpProtocolInfo) FIXME("ProtocolInfo not implemented.\n"); switch(AddressFamily) { case WS_AF_INET: { SOCKADDR_IN *addr4 = (SOCKADDR_IN *)lpAddress; /* If lpAddressLength is too small, tell caller the size we need */ if (*lpAddressLength < sizeof(SOCKADDR_IN)) { *lpAddressLength = sizeof(SOCKADDR_IN); res = WSAEFAULT; break; } memset(lpAddress, 0, sizeof(SOCKADDR_IN)); status = RtlIpv4StringToAddressExA(AddressString, FALSE, &addr4->sin_addr, &addr4->sin_port); if (status != STATUS_SUCCESS) { res = WSAEINVAL; break; } addr4->sin_family = WS_AF_INET; *lpAddressLength = sizeof(SOCKADDR_IN); break; } case WS_AF_INET6: { SOCKADDR_IN6 *addr6 = (SOCKADDR_IN6 *)lpAddress; /* If lpAddressLength is too small, tell caller the size we need */ if (*lpAddressLength < sizeof(SOCKADDR_IN6)) { *lpAddressLength = sizeof(SOCKADDR_IN6); res = WSAEFAULT; break; } memset(lpAddress, 0, sizeof(SOCKADDR_IN6)); status = RtlIpv6StringToAddressExA(AddressString, &addr6->sin6_addr, &addr6->sin6_scope_id, &addr6->sin6_port); if (status != STATUS_SUCCESS) { res = WSAEINVAL; break; } addr6->sin6_family = WS_AF_INET6; *lpAddressLength = sizeof(SOCKADDR_IN6); break; } default: /* According to MSDN, only AF_INET and AF_INET6 are supported. */ TRACE("Unsupported address family specified: %d.\n", AddressFamily); res = WSAEINVAL; } if (!res) return 0; SetLastError(res); return SOCKET_ERROR; } /*********************************************************************** * WSAStringToAddressW (WS2_32.81) * * FIXME: Does anybody know if this function allows using Hebrew/Arabic/Chinese... digits? * If this should be the case, it would be required to map these digits * to Unicode digits (0-9) using FoldString first. */ INT WINAPI WSAStringToAddressW(LPWSTR AddressString, INT AddressFamily, LPWSAPROTOCOL_INFOW lpProtocolInfo, LPSOCKADDR lpAddress, LPINT lpAddressLength) { INT sBuffer,res=0; LPSTR workBuffer=NULL; WSAPROTOCOL_INFOA infoA; LPWSAPROTOCOL_INFOA lpProtoInfoA = NULL; TRACE( "(%s, %x, %p, %p, %p)\n", debugstr_w(AddressString), AddressFamily, lpProtocolInfo, lpAddress, lpAddressLength ); if (!lpAddressLength || !lpAddress) return SOCKET_ERROR; /* if ProtocolInfo is available - convert to ANSI variant */ if (lpProtocolInfo) { lpProtoInfoA = &infoA; memcpy( lpProtoInfoA, lpProtocolInfo, FIELD_OFFSET( WSAPROTOCOL_INFOA, szProtocol ) ); if (!WideCharToMultiByte( CP_ACP, 0, lpProtocolInfo->szProtocol, -1, lpProtoInfoA->szProtocol, WSAPROTOCOL_LEN+1, NULL, NULL )) { SetLastError(WSAEINVAL); return SOCKET_ERROR; } } if (AddressString) { /* Translate AddressString to ANSI code page - assumes that only standard digits 0-9 are used with this API call */ sBuffer = WideCharToMultiByte( CP_ACP, 0, AddressString, -1, NULL, 0, NULL, NULL ); workBuffer = HeapAlloc( GetProcessHeap(), 0, sBuffer ); if (workBuffer) { WideCharToMultiByte( CP_ACP, 0, AddressString, -1, workBuffer, sBuffer, NULL, NULL ); res = WSAStringToAddressA(workBuffer,AddressFamily,lpProtoInfoA, lpAddress,lpAddressLength); HeapFree( GetProcessHeap(), 0, workBuffer ); return res; } else res = WSA_NOT_ENOUGH_MEMORY; } else res = WSAEINVAL; SetLastError(res); return SOCKET_ERROR; } /*********************************************************************** * WSAAddressToStringA (WS2_32.27) * * See WSAAddressToStringW */ INT WINAPI WSAAddressToStringA( LPSOCKADDR sockaddr, DWORD len, LPWSAPROTOCOL_INFOA info, LPSTR string, LPDWORD lenstr ) { DWORD size; CHAR buffer[54]; /* 32 digits + 7':' + '[' + '%" + 5 digits + ']:' + 5 digits + '\0' */ CHAR *p; TRACE( "(%p, %d, %p, %p, %p)\n", sockaddr, len, info, string, lenstr ); if (!sockaddr) return SOCKET_ERROR; if (!string || !lenstr) return SOCKET_ERROR; switch(sockaddr->sa_family) { case WS_AF_INET: { unsigned int long_ip = ntohl(((SOCKADDR_IN *)sockaddr)->sin_addr.WS_s_addr); if (len < sizeof(SOCKADDR_IN)) return SOCKET_ERROR; sprintf( buffer, "%u.%u.%u.%u:%u", (long_ip >> 24) & 0xff, (long_ip >> 16) & 0xff, (long_ip >> 8) & 0xff, long_ip & 0xff, ntohs( ((SOCKADDR_IN *)sockaddr)->sin_port ) ); p = strchr( buffer, ':' ); if (!((SOCKADDR_IN *)sockaddr)->sin_port) *p = 0; break; } case WS_AF_INET6: { struct WS_sockaddr_in6 *sockaddr6 = (LPSOCKADDR_IN6) sockaddr; size_t slen; buffer[0] = 0; if (len < sizeof(SOCKADDR_IN6)) return SOCKET_ERROR; if ((sockaddr6->sin6_port)) strcpy(buffer, "["); slen = strlen(buffer); if (!WS_inet_ntop(WS_AF_INET6, &sockaddr6->sin6_addr, &buffer[slen], sizeof(buffer) - slen)) { SetLastError(WSAEINVAL); return SOCKET_ERROR; } if ((sockaddr6->sin6_scope_id)) sprintf(buffer+strlen(buffer), "%%%u", sockaddr6->sin6_scope_id); if ((sockaddr6->sin6_port)) sprintf(buffer+strlen(buffer), "]:%u", ntohs(sockaddr6->sin6_port)); break; } default: SetLastError(WSAEINVAL); return SOCKET_ERROR; } size = strlen( buffer ) + 1; if (*lenstr < size) { *lenstr = size; SetLastError(WSAEFAULT); return SOCKET_ERROR; } TRACE("=> %s,%u bytes\n", debugstr_a(buffer), size); *lenstr = size; strcpy( string, buffer ); return 0; } /*********************************************************************** * WSAAddressToStringW (WS2_32.28) * * Convert a sockaddr address into a readable address string. * * PARAMS * sockaddr [I] Pointer to a sockaddr structure. * len [I] Size of the sockaddr structure. * info [I] Pointer to a WSAPROTOCOL_INFOW structure (optional). * string [I/O] Pointer to a buffer to receive the address string. * lenstr [I/O] Size of the receive buffer in WCHARs. * * RETURNS * Success: 0 * Failure: SOCKET_ERROR * * NOTES * The 'info' parameter is ignored. */ INT WINAPI WSAAddressToStringW( LPSOCKADDR sockaddr, DWORD len, LPWSAPROTOCOL_INFOW info, LPWSTR string, LPDWORD lenstr ) { INT ret; char buf[54]; /* 32 digits + 7':' + '[' + '%" + 5 digits + ']:' + 5 digits + '\0' */ TRACE( "(%p, %d, %p, %p, %p)\n", sockaddr, len, info, string, lenstr ); if ((ret = WSAAddressToStringA(sockaddr, len, NULL, buf, lenstr))) return ret; MultiByteToWideChar(CP_ACP, 0, buf, *lenstr, string, *lenstr); TRACE("=> %s,%u chars\n", debugstr_w(string), *lenstr); return 0; } /*********************************************************************** * WSAEnumNameSpaceProvidersA (WS2_32.34) */ INT WINAPI WSAEnumNameSpaceProvidersA( LPDWORD len, LPWSANAMESPACE_INFOA buffer ) { FIXME( "(%p %p) Stub!\n", len, buffer ); return 0; } /*********************************************************************** * WSAEnumNameSpaceProvidersW (WS2_32.35) */ INT WINAPI WSAEnumNameSpaceProvidersW( LPDWORD len, LPWSANAMESPACE_INFOW buffer ) { FIXME( "(%p %p) Stub!\n", len, buffer ); return 0; } /*********************************************************************** * WSAGetQOSByName (WS2_32.41) */ BOOL WINAPI WSAGetQOSByName( SOCKET s, LPWSABUF lpQOSName, LPQOS lpQOS ) { FIXME( "(0x%04lx %p %p) Stub!\n", s, lpQOSName, lpQOS ); return FALSE; } /*********************************************************************** * WSAGetServiceClassInfoA (WS2_32.42) */ INT WINAPI WSAGetServiceClassInfoA( LPGUID provider, LPGUID service, LPDWORD len, LPWSASERVICECLASSINFOA info ) { FIXME( "(%s %s %p %p) Stub!\n", debugstr_guid(provider), debugstr_guid(service), len, info ); SetLastError(WSA_NOT_ENOUGH_MEMORY); return SOCKET_ERROR; } /*********************************************************************** * WSAGetServiceClassInfoW (WS2_32.43) */ INT WINAPI WSAGetServiceClassInfoW( LPGUID provider, LPGUID service, LPDWORD len, LPWSASERVICECLASSINFOW info ) { FIXME( "(%s %s %p %p) Stub!\n", debugstr_guid(provider), debugstr_guid(service), len, info ); SetLastError(WSA_NOT_ENOUGH_MEMORY); return SOCKET_ERROR; } /*********************************************************************** * WSAGetServiceClassNameByClassIdA (WS2_32.44) */ INT WINAPI WSAGetServiceClassNameByClassIdA( LPGUID class, LPSTR service, LPDWORD len ) { FIXME( "(%s %p %p) Stub!\n", debugstr_guid(class), service, len ); SetLastError(WSA_NOT_ENOUGH_MEMORY); return SOCKET_ERROR; } /*********************************************************************** * WSAGetServiceClassNameByClassIdW (WS2_32.45) */ INT WINAPI WSAGetServiceClassNameByClassIdW( LPGUID class, LPWSTR service, LPDWORD len ) { FIXME( "(%s %p %p) Stub!\n", debugstr_guid(class), service, len ); SetLastError(WSA_NOT_ENOUGH_MEMORY); return SOCKET_ERROR; } /*********************************************************************** * WSALookupServiceBeginA (WS2_32.59) */ INT WINAPI WSALookupServiceBeginA( LPWSAQUERYSETA lpqsRestrictions, DWORD dwControlFlags, LPHANDLE lphLookup) { FIXME("(%p 0x%08x %p) Stub!\n", lpqsRestrictions, dwControlFlags, lphLookup); SetLastError(WSA_NOT_ENOUGH_MEMORY); return SOCKET_ERROR; } /*********************************************************************** * WSALookupServiceBeginW (WS2_32.60) */ INT WINAPI WSALookupServiceBeginW( LPWSAQUERYSETW lpqsRestrictions, DWORD dwControlFlags, LPHANDLE lphLookup) { FIXME("(%p 0x%08x %p) Stub!\n", lpqsRestrictions, dwControlFlags, lphLookup); SetLastError(WSA_NOT_ENOUGH_MEMORY); return SOCKET_ERROR; } /*********************************************************************** * WSALookupServiceEnd (WS2_32.61) */ INT WINAPI WSALookupServiceEnd( HANDLE lookup ) { FIXME("(%p) Stub!\n", lookup ); return 0; } /*********************************************************************** * WSALookupServiceNextA (WS2_32.62) */ INT WINAPI WSALookupServiceNextA( HANDLE lookup, DWORD flags, LPDWORD len, LPWSAQUERYSETA results ) { FIXME( "(%p 0x%08x %p %p) Stub!\n", lookup, flags, len, results ); SetLastError(WSA_E_NO_MORE); return SOCKET_ERROR; } /*********************************************************************** * WSALookupServiceNextW (WS2_32.63) */ INT WINAPI WSALookupServiceNextW( HANDLE lookup, DWORD flags, LPDWORD len, LPWSAQUERYSETW results ) { FIXME( "(%p 0x%08x %p %p) Stub!\n", lookup, flags, len, results ); SetLastError(WSA_E_NO_MORE); return SOCKET_ERROR; } /*********************************************************************** * WSANtohl (WS2_32.64) */ INT WINAPI WSANtohl( SOCKET s, WS_u_long netlong, WS_u_long* lphostlong ) { TRACE( "(%04lx 0x%08x %p)\n", s, netlong, lphostlong ); if (!lphostlong) return WSAEFAULT; *lphostlong = ntohl( netlong ); return 0; } /*********************************************************************** * WSANtohs (WS2_32.65) */ INT WINAPI WSANtohs( SOCKET s, WS_u_short netshort, WS_u_short* lphostshort ) { TRACE( "(%04lx 0x%08x %p)\n", s, netshort, lphostshort ); if (!lphostshort) return WSAEFAULT; *lphostshort = ntohs( netshort ); return 0; } /*********************************************************************** * WSAProviderConfigChange (WS2_32.66) */ INT WINAPI WSAProviderConfigChange( LPHANDLE handle, LPWSAOVERLAPPED overlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE completion ) { FIXME( "(%p %p %p) Stub!\n", handle, overlapped, completion ); return SOCKET_ERROR; } /*********************************************************************** * WSARecvDisconnect (WS2_32.68) */ INT WINAPI WSARecvDisconnect( SOCKET s, LPWSABUF disconnectdata ) { TRACE( "(%04lx %p)\n", s, disconnectdata ); return WS_shutdown( s, SD_RECEIVE ); } /*********************************************************************** * WSASetServiceA (WS2_32.76) */ INT WINAPI WSASetServiceA( LPWSAQUERYSETA query, WSAESETSERVICEOP operation, DWORD flags ) { FIXME( "(%p 0x%08x 0x%08x) Stub!\n", query, operation, flags ); return 0; } /*********************************************************************** * WSASetServiceW (WS2_32.77) */ INT WINAPI WSASetServiceW( LPWSAQUERYSETW query, WSAESETSERVICEOP operation, DWORD flags ) { FIXME( "(%p 0x%08x 0x%08x) Stub!\n", query, operation, flags ); return 0; } /*********************************************************************** * WSCEnableNSProvider (WS2_32.84) */ INT WINAPI WSCEnableNSProvider( LPGUID provider, BOOL enable ) { FIXME( "(%s 0x%08x) Stub!\n", debugstr_guid(provider), enable ); return 0; } /*********************************************************************** * WSCGetProviderInfo */ INT WINAPI WSCGetProviderInfo( LPGUID provider, WSC_PROVIDER_INFO_TYPE info_type, PBYTE info, size_t* len, DWORD flags, LPINT errcode ) { FIXME( "(%s 0x%08x %p %p 0x%08x %p) Stub!\n", debugstr_guid(provider), info_type, info, len, flags, errcode ); if (!errcode) return SOCKET_ERROR; if (!provider) { *errcode = WSAEFAULT; return SOCKET_ERROR; } *errcode = WSANO_RECOVERY; return SOCKET_ERROR; } /*********************************************************************** * WSCGetProviderPath (WS2_32.86) */ INT WINAPI WSCGetProviderPath( LPGUID provider, LPWSTR path, LPINT len, LPINT errcode ) { FIXME( "(%s %p %p %p) Stub!\n", debugstr_guid(provider), path, len, errcode ); if (!provider || !len) { if (errcode) *errcode = WSAEFAULT; return SOCKET_ERROR; } if (*len <= 0) { if (errcode) *errcode = WSAEINVAL; return SOCKET_ERROR; } return 0; } /*********************************************************************** * WSCInstallNameSpace (WS2_32.87) */ INT WINAPI WSCInstallNameSpace( LPWSTR identifier, LPWSTR path, DWORD namespace, DWORD version, LPGUID provider ) { FIXME( "(%s %s 0x%08x 0x%08x %s) Stub!\n", debugstr_w(identifier), debugstr_w(path), namespace, version, debugstr_guid(provider) ); return 0; } /*********************************************************************** * WSCUnInstallNameSpace (WS2_32.89) */ INT WINAPI WSCUnInstallNameSpace( LPGUID lpProviderId ) { FIXME("(%s) Stub!\n", debugstr_guid(lpProviderId)); return NO_ERROR; } /*********************************************************************** * WSCWriteProviderOrder (WS2_32.91) */ INT WINAPI WSCWriteProviderOrder( LPDWORD entry, DWORD number ) { FIXME("(%p 0x%08x) Stub!\n", entry, number); return 0; } /*********************************************************************** * WSCSetApplicationCategory (WS2_32.@) */ INT WINAPI WSCSetApplicationCategory( LPCWSTR path, DWORD len, LPCWSTR extra, DWORD extralen, DWORD lspcat, DWORD *prev_lspcat, LPINT err ) { FIXME("(%s %d %s %d %d %p) Stub!\n", debugstr_w(path), len, debugstr_w(extra), extralen, lspcat, prev_lspcat); return 0; } /*********************************************************************** * WSANSPIoctl (WS2_32.91) */ INT WINAPI WSANSPIoctl( HANDLE hLookup, DWORD dwControlCode, LPVOID lpvInBuffer, DWORD cbInBuffer, LPVOID lpvOutBuffer, DWORD cbOutBuffer, LPDWORD lpcbBytesReturned, LPWSACOMPLETION lpCompletion ) { FIXME("(%p, 0x%08x, %p, 0x%08x, %p, 0x%08x, %p, %p) Stub!\n", hLookup, dwControlCode, lpvInBuffer, cbInBuffer, lpvOutBuffer, cbOutBuffer, lpcbBytesReturned, lpCompletion); SetLastError(WSA_NOT_ENOUGH_MEMORY); return SOCKET_ERROR; } static BOOL protocol_matches_filter( const int *filter, int protocol ) { if (!filter) return TRUE; while (*filter) { if (protocol == *filter++) return TRUE; } return FALSE; } /***************************************************************************** * WSAEnumProtocolsA [WS2_32.@] * * see function WSAEnumProtocolsW */ int WINAPI WSAEnumProtocolsA( int *filter, WSAPROTOCOL_INFOA *protocols, DWORD *size ) { DWORD i, count = 0; TRACE("filter %p, protocols %p, size %p\n", filter, protocols, size); for (i = 0; i < ARRAY_SIZE(supported_protocols); ++i) { if (protocol_matches_filter( filter, supported_protocols[i].iProtocol )) ++count; } if (!protocols || *size < count * sizeof(WSAPROTOCOL_INFOA)) { *size = count * sizeof(WSAPROTOCOL_INFOA); WSASetLastError( WSAENOBUFS ); return SOCKET_ERROR; } count = 0; for (i = 0; i < ARRAY_SIZE(supported_protocols); ++i) { if (protocol_matches_filter( filter, supported_protocols[i].iProtocol )) { memcpy( &protocols[count], &supported_protocols[i], offsetof( WSAPROTOCOL_INFOW, szProtocol ) ); WideCharToMultiByte( CP_ACP, 0, supported_protocols[i].szProtocol, -1, protocols[count].szProtocol, sizeof(protocols[count].szProtocol), NULL, NULL ); ++count; } } return count; } /***************************************************************************** * WSAEnumProtocolsW [WS2_32.@] * * Retrieves information about specified set of active network protocols. * * PARAMS * protocols [I] Pointer to null-terminated array of protocol id's. NULL * retrieves information on all available protocols. * buffer [I] Pointer to a buffer to be filled with WSAPROTOCOL_INFO * structures. * len [I/O] Pointer to a variable specifying buffer size. On output * the variable holds the number of bytes needed when the * specified size is too small. * * RETURNS * Success: number of WSAPROTOCOL_INFO structures in buffer. * Failure: SOCKET_ERROR * * NOTES * NT4SP5 does not return SPX if protocols == NULL * * BUGS * - NT4SP5 returns in addition these list of NETBIOS protocols * (address family 17), each entry two times one for socket type 2 and 5 * * iProtocol szProtocol * 0x80000000 \Device\NwlnkNb * 0xfffffffa \Device\NetBT_CBENT7 * 0xfffffffb \Device\Nbf_CBENT7 * 0xfffffffc \Device\NetBT_NdisWan5 * 0xfffffffd \Device\NetBT_El9202 * 0xfffffffe \Device\Nbf_El9202 * 0xffffffff \Device\Nbf_NdisWan4 * * - there is no check that the operating system supports the returned * protocols */ int WINAPI WSAEnumProtocolsW( int *filter, WSAPROTOCOL_INFOW *protocols, DWORD *size ) { DWORD i, count = 0; TRACE("filter %p, protocols %p, size %p\n", filter, protocols, size); for (i = 0; i < ARRAY_SIZE(supported_protocols); ++i) { if (protocol_matches_filter( filter, supported_protocols[i].iProtocol )) ++count; } if (!protocols || *size < count * sizeof(WSAPROTOCOL_INFOW)) { *size = count * sizeof(WSAPROTOCOL_INFOW); WSASetLastError( WSAENOBUFS ); return SOCKET_ERROR; } count = 0; for (i = 0; i < ARRAY_SIZE(supported_protocols); ++i) { if (protocol_matches_filter( filter, supported_protocols[i].iProtocol )) protocols[count++] = supported_protocols[i]; } return count; } /***************************************************************************** * WSCEnumProtocols [WS2_32.@] * * PARAMS * protocols [I] Null-terminated array of iProtocol values. * buffer [O] Buffer of WSAPROTOCOL_INFOW structures. * len [I/O] Size of buffer on input/output. * errno [O] Error code. * * RETURNS * Success: number of protocols to be reported on. * Failure: SOCKET_ERROR. error is in errno. * * BUGS * Doesn't supply info on layered protocols. * */ INT WINAPI WSCEnumProtocols( LPINT protocols, LPWSAPROTOCOL_INFOW buffer, LPDWORD len, LPINT err ) { INT ret = WSAEnumProtocolsW( protocols, buffer, len ); if (ret == SOCKET_ERROR) *err = WSAENOBUFS; return ret; }