/* * 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 "ws2_32_private.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__ */ #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}, }, }; #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 ); 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; 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 ); int num_startup; static FARPROC blocking_hook = (FARPROC)WSA_DefaultBlockingHook; /* function prototypes */ 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_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_poll_map[][2] = { MAP_OPTION( POLLERR ), MAP_OPTION( POLLHUP ), MAP_OPTION( POLLNVAL ), MAP_OPTION( POLLWRNORM ), MAP_OPTION( POLLWRBAND ), MAP_OPTION( POLLRDNORM ), { WS_POLLRDBAND, POLLPRI } }; #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_ACCESS_VIOLATION; case EINVAL: return STATUS_INVALID_PARAMETER; case ENFILE: case EMFILE: return STATUS_TOO_MANY_OPENED_FILES; case EINPROGRESS: case EWOULDBLOCK: return STATUS_DEVICE_NOT_READY; 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_SHARING_VIOLATION; case EADDRNOTAVAIL: return STATUS_INVALID_PARAMETER; case ECONNREFUSED: return STATUS_CONNECTION_REFUSED; case ESHUTDOWN: return STATUS_PIPE_DISCONNECTED; case ENOTCONN: return STATUS_INVALID_CONNECTION; 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; } } 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 EINPROGRESS: case EWOULDBLOCK: return WSAEWOULDBLOCK; 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 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_ACCESS_VIOLATION; case WSAEINVAL: return STATUS_INVALID_PARAMETER; case WSAEMFILE: return STATUS_TOO_MANY_OPENED_FILES; case WSAEINPROGRESS: case WSAEWOULDBLOCK: return STATUS_DEVICE_NOT_READY; 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_SHARING_VIOLATION; case WSAEADDRNOTAVAIL: return STATUS_INVALID_PARAMETER; case WSAECONNREFUSED: return STATUS_CONNECTION_REFUSED; case WSAESHUTDOWN: return STATUS_PIPE_DISCONNECTED; case WSAENOTCONN: return STATUS_INVALID_CONNECTION; 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( NTSTATUS status ) { static const struct { NTSTATUS status; DWORD error; } errors[] = { {STATUS_PENDING, ERROR_IO_PENDING}, {STATUS_BUFFER_OVERFLOW, WSAEMSGSIZE}, {STATUS_NOT_IMPLEMENTED, WSAEOPNOTSUPP}, {STATUS_ACCESS_VIOLATION, WSAEFAULT}, {STATUS_PAGEFILE_QUOTA, WSAENOBUFS}, {STATUS_INVALID_HANDLE, WSAENOTSOCK}, {STATUS_NO_SUCH_DEVICE, WSAENETDOWN}, {STATUS_NO_SUCH_FILE, WSAENETDOWN}, {STATUS_NO_MEMORY, WSAENOBUFS}, {STATUS_CONFLICTING_ADDRESSES, WSAENOBUFS}, {STATUS_ACCESS_DENIED, WSAEACCES}, {STATUS_BUFFER_TOO_SMALL, WSAEFAULT}, {STATUS_OBJECT_TYPE_MISMATCH, WSAENOTSOCK}, {STATUS_OBJECT_NAME_NOT_FOUND, WSAENETDOWN}, {STATUS_OBJECT_PATH_NOT_FOUND, WSAENETDOWN}, {STATUS_SHARING_VIOLATION, WSAEADDRINUSE}, {STATUS_QUOTA_EXCEEDED, WSAENOBUFS}, {STATUS_TOO_MANY_PAGING_FILES, WSAENOBUFS}, {STATUS_INSUFFICIENT_RESOURCES, WSAENOBUFS}, {STATUS_WORKING_SET_QUOTA, WSAENOBUFS}, {STATUS_DEVICE_NOT_READY, WSAEWOULDBLOCK}, {STATUS_PIPE_DISCONNECTED, WSAESHUTDOWN}, {STATUS_IO_TIMEOUT, WSAETIMEDOUT}, {STATUS_NOT_SUPPORTED, WSAEOPNOTSUPP}, {STATUS_REMOTE_NOT_LISTENING, WSAECONNREFUSED}, {STATUS_BAD_NETWORK_PATH, WSAENETUNREACH}, {STATUS_NETWORK_BUSY, WSAENETDOWN}, {STATUS_INVALID_NETWORK_RESPONSE, WSAENETDOWN}, {STATUS_UNEXPECTED_NETWORK_ERROR, WSAENETDOWN}, {STATUS_REQUEST_NOT_ACCEPTED, WSAEWOULDBLOCK}, {STATUS_CANCELLED, ERROR_OPERATION_ABORTED}, {STATUS_COMMITMENT_LIMIT, WSAENOBUFS}, {STATUS_LOCAL_DISCONNECT, WSAECONNABORTED}, {STATUS_REMOTE_DISCONNECT, WSAECONNRESET}, {STATUS_REMOTE_RESOURCES, WSAENOBUFS}, {STATUS_LINK_FAILED, WSAECONNRESET}, {STATUS_LINK_TIMEOUT, WSAETIMEDOUT}, {STATUS_INVALID_CONNECTION, WSAENOTCONN}, {STATUS_INVALID_ADDRESS, WSAEADDRNOTAVAIL}, {STATUS_INVALID_BUFFER_SIZE, WSAEMSGSIZE}, {STATUS_INVALID_ADDRESS_COMPONENT, WSAEADDRNOTAVAIL}, {STATUS_TOO_MANY_ADDRESSES, WSAENOBUFS}, {STATUS_ADDRESS_ALREADY_EXISTS, WSAEADDRINUSE}, {STATUS_CONNECTION_DISCONNECTED, WSAECONNRESET}, {STATUS_CONNECTION_RESET, WSAECONNRESET}, {STATUS_TRANSACTION_ABORTED, WSAECONNABORTED}, {STATUS_CONNECTION_REFUSED, WSAECONNREFUSED}, {STATUS_GRACEFUL_DISCONNECT, WSAEDISCON}, {STATUS_CONNECTION_ACTIVE, WSAEISCONN}, {STATUS_NETWORK_UNREACHABLE, WSAENETUNREACH}, {STATUS_HOST_UNREACHABLE, WSAEHOSTUNREACH}, {STATUS_PROTOCOL_UNREACHABLE, WSAENETUNREACH}, {STATUS_PORT_UNREACHABLE, WSAECONNRESET}, {STATUS_REQUEST_ABORTED, WSAEINTR}, {STATUS_CONNECTION_ABORTED, WSAECONNABORTED}, {STATUS_DATATYPE_MISALIGNMENT_ERROR,WSAEFAULT}, {STATUS_HOST_DOWN, WSAEHOSTDOWN}, {0x80070000 | ERROR_IO_INCOMPLETE, ERROR_IO_INCOMPLETE}, {0xc0010000 | ERROR_IO_INCOMPLETE, ERROR_IO_INCOMPLETE}, {0xc0070000 | ERROR_IO_INCOMPLETE, ERROR_IO_INCOMPLETE}, }; unsigned int i; for (i = 0; i < ARRAY_SIZE(errors); ++i) { if (errors[i].status == status) return errors[i].error; } return NT_SUCCESS(status) ? RtlNtStatusToDosErrorNoTeb(status) : WSAEINVAL; } /* 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 ) { close( 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; } 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 instance, DWORD reason, void *reserved ) { if (reason == DLL_THREAD_DETACH) free_per_thread_data(); 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; } /* 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; } 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; } 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 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. */ 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 */ 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) ); close( 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) ); close( 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; close( 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; close( 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 ); close( 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 == WSAEWOULDBLOCK) { /* 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 == WSAEWOULDBLOCK) { struct ws2_async *wsa; DWORD size; 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); size = offsetof( struct ws2_async, iovec[1] ) + sendBufLen; /* Indirectly call WSASend */ if (!(wsa = (struct ws2_async *)alloc_async_io( size, 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 = &wsa->iovec[1]; wsa->iovec[0].iov_len = sendBufLen; if (sendBufLen) memcpy( wsa->iovec[0].iov_base, sendBuf, 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) */ } 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 ); } /*********************************************************************** * 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; SetLastError( NtStatusToWSAError(status) ); return NT_SUCCESS( 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; break; } } 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 = RtlNtStatusToDosError( 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; } /*********************************************************************** * 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 ); } /*********************************************************************** * 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); } /*********************************************************************** * 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; } /*********************************************************************** * WSARecvDisconnect (WS2_32.68) */ INT WINAPI WSARecvDisconnect( SOCKET s, LPWSABUF disconnectdata ) { TRACE( "(%04lx %p)\n", s, disconnectdata ); return WS_shutdown( s, SD_RECEIVE ); } 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; }