/* * based on Windows Sockets 1.1 specs * (ftp.microsoft.com:/Advsys/winsock/spec11/WINSOCK.TXT) * * Copyright (C) 1993,1994,1996,1997 John Brezak, Erik Bos, Alex Korobka. * * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * * NOTE: If you make any changes to fix a particular app, make sure * they don't break something else like Netscape or telnet and ftp * clients and servers (www.winsite.com got a lot of those). */ #include "config.h" #include "wine/port.h" #include #include #include #include #ifdef HAVE_SYS_IPC_H # include #endif #ifdef HAVE_SYS_IOCTL_H # include #endif #ifdef HAVE_SYS_FILIO_H # include #endif #ifdef HAVE_SYS_SOCKIO_H # include #endif #if defined(__EMX__) # include #endif #ifdef HAVE_SYS_PARAM_H # include #endif #ifdef HAVE_SYS_MSG_H # include #endif #ifdef HAVE_SYS_WAIT_H # include #endif #ifdef HAVE_SYS_UIO_H # include #endif #ifdef HAVE_SYS_SOCKET_H #include #endif #ifdef HAVE_NETINET_IN_H # include #endif #ifdef HAVE_NETINET_TCP_H # include #endif #ifdef HAVE_ARPA_INET_H # include #endif #include #include #include #ifdef HAVE_SYS_ERRNO_H #include #endif #ifdef HAVE_NETDB_H #include #endif #ifdef HAVE_UNISTD_H # include #endif #include #ifdef HAVE_ARPA_NAMESER_H # include #endif #ifdef HAVE_RESOLV_H # include #endif #ifdef HAVE_NET_IF_H # include #endif #ifdef HAVE_IPX_GNU # include # define HAVE_IPX #endif #ifdef HAVE_IPX_LINUX # include # include # define HAVE_IPX #endif #ifdef HAVE_SYS_POLL_H # include #endif #ifdef HAVE_SYS_TIME_H # include #endif #define NONAMELESSUNION #define NONAMELESSSTRUCT #include "windef.h" #include "winbase.h" #include "wingdi.h" #include "winuser.h" #include "winerror.h" #include "winsock2.h" #include "ws2tcpip.h" #include "wsipx.h" #include "winnt.h" #include "iphlpapi.h" #include "thread.h" #include "wine/server.h" #include "wine/debug.h" #ifdef __FreeBSD__ # define sipx_network sipx_addr.x_net # define sipx_node sipx_addr.x_host.c_host #endif /* __FreeBSD__ */ WINE_DEFAULT_DEBUG_CHANNEL(winsock); /* critical section to protect some non-rentrant net function */ extern CRITICAL_SECTION csWSgetXXXbyYYY; inline static const char *debugstr_sockaddr( const struct WS_sockaddr *a ) { if (!a) return "(nil)"; return wine_dbg_sprintf("{ family %d, address %s, port %d }", ((struct sockaddr_in *)a)->sin_family, inet_ntoa(((struct sockaddr_in *)a)->sin_addr), ntohs(((struct sockaddr_in *)a)->sin_port)); } /* HANDLE<->SOCKET conversion (SOCKET is UINT_PTR). */ #define SOCKET2HANDLE(s) ((HANDLE)(s)) #define HANDLE2SOCKET(h) ((SOCKET)(h)) /**************************************************************** * Async IO declarations ****************************************************************/ #include "async.h" static DWORD ws2_async_get_count (const struct async_private *ovp); static void CALLBACK ws2_async_call_completion (ULONG_PTR data); static void ws2_async_cleanup ( struct async_private *ovp ); static struct async_ops ws2_async_ops = { ws2_async_get_count, ws2_async_call_completion, ws2_async_cleanup }; static struct async_ops ws2_nocomp_async_ops = { ws2_async_get_count, NULL, /* call_completion */ ws2_async_cleanup }; typedef struct ws2_async { async_private async; LPWSAOVERLAPPED user_overlapped; LPWSAOVERLAPPED_COMPLETION_ROUTINE completion_func; struct iovec *iovec; int n_iovecs; struct WS_sockaddr *addr; union { int val; /* for send operations */ int *ptr; /* for recv operations */ } addrlen; DWORD flags; } ws2_async; /****************************************************************/ /* ----------------------------------- 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 ); static struct WS_hostent *he_buffer; /* typecast for Win32 ws_hostent */ static struct WS_servent *se_buffer; /* typecast for Win32 ws_servent */ static struct WS_protoent *pe_buffer; /* typecast for Win32 ws_protoent */ static INT num_startup; /* reference counter */ static FARPROC blocking_hook = WSA_DefaultBlockingHook; /* function prototypes */ static struct WS_hostent *WS_dup_he(const struct hostent* p_he); static struct WS_protoent *WS_dup_pe(const struct protoent* p_pe); static struct WS_servent *WS_dup_se(const struct servent* p_se); int WSAIOCTL_GetInterfaceCount(void); int WSAIOCTL_GetInterfaceName(int intNumber, char *intName); UINT wsaErrno(void); UINT wsaHerrno(int errnr); #define MAP_OPTION(opt) { WS_##opt, opt } static const int ws_sock_map[][2] = { MAP_OPTION( SO_DEBUG ), 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 { 0, 0 } }; static const int ws_tcp_map[][2] = { #ifdef TCP_NODELAY MAP_OPTION( TCP_NODELAY ), #endif { 0, 0 } }; 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_OPTIONS ), #ifdef IP_HDRINCL MAP_OPTION( IP_HDRINCL ), #endif MAP_OPTION( IP_TOS ), MAP_OPTION( IP_TTL ), { 0, 0 } }; static DWORD opentype_tls_index = TLS_OUT_OF_INDEXES; /* TLS index for SO_OPENTYPE flag */ inline static DWORD NtStatusToWSAError ( const DWORD status ) { /* We only need to cover the status codes set by server async request handling */ DWORD wserr; switch ( status ) { case STATUS_SUCCESS: wserr = 0; break; case STATUS_PENDING: wserr = WSA_IO_PENDING; break; case STATUS_INVALID_HANDLE: wserr = WSAENOTSOCK; break; /* WSAEBADF ? */ case STATUS_INVALID_PARAMETER: wserr = WSAEINVAL; break; case STATUS_PIPE_DISCONNECTED: wserr = WSAESHUTDOWN; break; case STATUS_CANCELLED: wserr = WSA_OPERATION_ABORTED; break; case STATUS_TIMEOUT: wserr = WSAETIMEDOUT; break; case STATUS_NO_MEMORY: wserr = WSAEFAULT; break; default: if ( status >= WSABASEERR && status <= WSABASEERR+1004 ) /* It is not a NT status code but a winsock error */ wserr = status; else { wserr = RtlNtStatusToDosError( status ); FIXME ( "Status code %08lx converted to DOS error code %lx\n", status, wserr ); } } return wserr; } /* set last error code from NT status without mapping WSA errors */ inline static unsigned int set_error( unsigned int err ) { if (err) { err = NtStatusToWSAError ( err ); SetLastError( err ); } return err; } inline static int get_sock_fd( SOCKET s, DWORD access, int *flags ) { int fd; if (set_error( wine_server_handle_to_fd( SOCKET2HANDLE(s), access, &fd, NULL, flags ) )) return -1; return fd; } inline static void release_sock_fd( SOCKET s, int fd ) { wine_server_release_fd( SOCKET2HANDLE(s), fd ); } static void _enable_event( HANDLE s, unsigned int event, unsigned int sstate, unsigned int cstate ) { SERVER_START_REQ( enable_socket_event ) { req->handle = s; req->mask = event; req->sstate = sstate; req->cstate = cstate; wine_server_call( req ); } SERVER_END_REQ; } static int _is_blocking(SOCKET s) { int ret; SERVER_START_REQ( get_socket_event ) { req->handle = SOCKET2HANDLE(s); req->service = FALSE; req->c_event = 0; wine_server_call( req ); ret = (reply->state & FD_WINE_NONBLOCKING) == 0; } SERVER_END_REQ; return ret; } static unsigned int _get_sock_mask(SOCKET s) { unsigned int ret; SERVER_START_REQ( get_socket_event ) { req->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) { /* do a dummy wineserver request in order to let the wineserver run through its select loop once */ (void)_is_blocking(s); } static int _get_sock_error(SOCKET s, unsigned int bit) { int events[FD_MAX_EVENTS]; SERVER_START_REQ( get_socket_event ) { req->handle = SOCKET2HANDLE(s); req->service = FALSE; req->c_event = 0; wine_server_set_reply( req, events, sizeof(events) ); wine_server_call( req ); } SERVER_END_REQ; return events[bit]; } static void WINSOCK_DeleteIData(void) { /* delete scratch buffers */ if (he_buffer) HeapFree( GetProcessHeap(), 0, he_buffer ); if (se_buffer) HeapFree( GetProcessHeap(), 0, se_buffer ); if (pe_buffer) HeapFree( GetProcessHeap(), 0, pe_buffer ); he_buffer = NULL; se_buffer = NULL; pe_buffer = NULL; num_startup = 0; } /*********************************************************************** * DllMain (WS2_32.init) */ BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID fImpLoad) { TRACE("%p 0x%lx %p\n", hInstDLL, fdwReason, fImpLoad); switch (fdwReason) { case DLL_PROCESS_ATTACH: DisableThreadLibraryCalls(hInstDLL); opentype_tls_index = TlsAlloc(); break; case DLL_PROCESS_DETACH: TlsFree( opentype_tls_index ); WINSOCK_DeleteIData(); break; } return TRUE; } /*********************************************************************** * convert_sockopt() * * Converts socket flags from Windows format. * Return 1 if converted, 0 if not (error). */ static int convert_sockopt(INT *level, INT *optname) { int i; switch (*level) { case WS_SOL_SOCKET: *level = SOL_SOCKET; for(i=0; ws_sock_map[i][0]; 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; ws_tcp_map[i][0]; 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; ws_ip_map[i][0]; 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; default: FIXME("Unimplemented or unknown socket level\n"); } return 0; } static inline BOOL is_timeout_option( int optname ) { #ifdef SO_RCVTIMEO if (optname == SO_RCVTIMEO) return TRUE; #endif #ifdef SO_SNDTIMEO if (optname == SO_SNDTIMEO) return TRUE; #endif return FALSE; } /* ----------------------------------- Per-thread info (or per-process?) */ static char *strdup_lower(const char *str) { int i; char *ret = HeapAlloc( GetProcessHeap(), 0, strlen(str) + 1 ); if (ret) { for (i = 0; str[i]; i++) ret[i] = tolower(str[i]); ret[i] = 0; } else SetLastError(WSAENOBUFS); return ret; } static fd_set* fd_set_import( fd_set* fds, const WS_fd_set* wsfds, int access, int* highfd, int lfd[] ) { /* translate Winsock fd set into local fd set */ if( wsfds ) { int i; FD_ZERO(fds); for( i = 0; i < wsfds->fd_count; i++ ) { int s = wsfds->fd_array[i]; int fd = get_sock_fd( s, access, NULL ); if (fd != -1) { lfd[ i ] = fd; if( fd > *highfd ) *highfd = fd; FD_SET(fd, fds); } else lfd[ i ] = -1; } return fds; } return NULL; } inline static int sock_error_p(int s) { unsigned int optval, optlen; optlen = sizeof(optval); getsockopt(s, SOL_SOCKET, SO_ERROR, (void *) &optval, &optlen); if (optval) WARN("\t[%i] error: %d\n", s, optval); return optval != 0; } static int fd_set_export( const fd_set* fds, fd_set* exceptfds, WS_fd_set* wsfds, int lfd[] ) { int num_err = 0; /* translate local fd set into Winsock fd set, adding * errors to exceptfds (only if app requested it) */ if( wsfds ) { int i, j, count = wsfds->fd_count; for( i = 0, j = 0; i < count; i++ ) { int fd = lfd[i]; SOCKET s = wsfds->fd_array[i]; if (fd == -1) continue; if( FD_ISSET(fd, fds) ) { if ( exceptfds && sock_error_p(fd) ) { FD_SET(fd, exceptfds); num_err++; } else wsfds->fd_array[j++] = s; } release_sock_fd( s, fd ); } wsfds->fd_count = j; } return num_err; } static void fd_set_unimport( WS_fd_set* wsfds, int lfd[] ) { if ( wsfds ) { int i; for( i = 0; i < wsfds->fd_count; i++ ) if ( lfd[i] >= 0 ) release_sock_fd( wsfds->fd_array[i], lfd[i] ); wsfds->fd_count = 0; } } /* utility: given an fd, will block until one of the events occurs */ static inline int do_block( int fd, int events ) { struct pollfd pfd; pfd.fd = fd; pfd.events = events; poll(&pfd, 1, -1); return pfd.revents; } /* ----------------------------------- 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\n"); return 0; } /*********************************************************************** * WSACleanup (WS2_32.116) */ INT WINAPI WSACleanup(void) { if (num_startup) { if (--num_startup > 0) return 0; WINSOCK_DeleteIData(); return 0; } SetLastError(WSANOTINITIALISED); return SOCKET_ERROR; } /*********************************************************************** * WSAGetLastError (WINSOCK.111) * WSAGetLastError (WS2_32.111) */ INT WINAPI WSAGetLastError(void) { return GetLastError(); } /*********************************************************************** * WSASetLastError (WS2_32.112) */ void WINAPI WSASetLastError(INT iError) { SetLastError(iError); } static struct WS_hostent *check_buffer_he(int size) { static int he_len; if (he_buffer) { if (he_len >= size ) return he_buffer; HeapFree( GetProcessHeap(), 0, he_buffer ); } he_buffer = HeapAlloc( GetProcessHeap(), 0, (he_len = size) ); if (!he_buffer) SetLastError(WSAENOBUFS); return he_buffer; } static struct WS_servent *check_buffer_se(int size) { static int se_len; if (se_buffer) { if (se_len >= size ) return se_buffer; HeapFree( GetProcessHeap(), 0, se_buffer ); } se_buffer = HeapAlloc( GetProcessHeap(), 0, (se_len = size) ); if (!se_buffer) SetLastError(WSAENOBUFS); return se_buffer; } static struct WS_protoent *check_buffer_pe(int size) { static int pe_len; if (pe_buffer) { if (pe_len >= size ) return pe_buffer; HeapFree( GetProcessHeap(), 0, pe_buffer ); } pe_buffer = HeapAlloc( GetProcessHeap(), 0, (pe_len = size) ); if (!pe_buffer) SetLastError(WSAENOBUFS); return pe_buffer; } /* ----------------------------------- i/o APIs */ #ifdef HAVE_IPX #define SUPPORTED_PF(pf) ((pf)==WS_AF_INET || (pf)== WS_AF_IPX) #else #define SUPPORTED_PF(pf) ((pf)==WS_AF_INET) #endif /**********************************************************************/ /* Returns the converted address if successful, NULL if it was too small to * start with. Note that the returned pointer may be the original pointer * if no conversion is necessary. */ static const struct sockaddr* ws_sockaddr_ws2u(const struct WS_sockaddr* wsaddr, int wsaddrlen, int *uaddrlen) { switch (wsaddr->sa_family) { #ifdef HAVE_IPX case WS_AF_IPX: { struct WS_sockaddr_ipx* wsipx=(struct WS_sockaddr_ipx*)wsaddr; struct sockaddr_ipx* uipx; 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 return (const struct sockaddr*)uipx; } #endif default: if (wsaddrlensa_family) { #ifdef HAVE_IPX case AF_IPX: { struct sockaddr_ipx* uipx=(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=uaddrlen; 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 default: /* No conversion needed */ memcpy(wsaddr,uaddr,*wsaddrlen); if (*wsaddrleniosb->Information; } static void ws2_async_cleanup ( struct async_private *ap ) { struct ws2_async *as = (struct ws2_async*) ap; TRACE ( "as: %p uovl %p ovl %p\n", as, as->user_overlapped, as->async.iosb ); if ( !as->user_overlapped ) { #if 0 /* FIXME: I don't think this is really used */ if ( as->overlapped->hEvent != INVALID_HANDLE_VALUE ) WSACloseEvent ( as->overlapped->hEvent ); #endif HeapFree ( GetProcessHeap(), 0, as->async.iosb ); } if ( as->iovec ) HeapFree ( GetProcessHeap(), 0, as->iovec ); HeapFree ( GetProcessHeap(), 0, as ); } static void CALLBACK ws2_async_call_completion (ULONG_PTR data) { ws2_async* as = (ws2_async*) data; TRACE ("data: %p\n", as); as->completion_func ( NtStatusToWSAError (as->async.iosb->u.Status), as->async.iosb->Information, as->user_overlapped, as->flags ); ws2_async_cleanup ( &as->async ); } /*********************************************************************** * WS2_make_async (INTERNAL) */ static void WS2_async_recv (async_private *as); static void WS2_async_send (async_private *as); inline static struct ws2_async* WS2_make_async (SOCKET s, int fd, int type, struct iovec *iovec, DWORD dwBufferCount, LPDWORD lpFlags, struct WS_sockaddr *addr, LPINT addrlen, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine) { struct ws2_async *wsa = HeapAlloc ( GetProcessHeap(), 0, sizeof ( ws2_async ) ); TRACE ( "wsa %p\n", wsa ); if (!wsa) return NULL; wsa->async.ops = ( lpCompletionRoutine ? &ws2_async_ops : &ws2_nocomp_async_ops ); wsa->async.handle = (HANDLE) s; wsa->async.fd = fd; wsa->async.type = type; switch (type) { case ASYNC_TYPE_READ: wsa->flags = *lpFlags; wsa->async.func = WS2_async_recv; wsa->addrlen.ptr = addrlen; break; case ASYNC_TYPE_WRITE: wsa->flags = 0; wsa->async.func = WS2_async_send; wsa->addrlen.val = *addrlen; break; default: ERR ("Invalid async type: %d\n", type); } wsa->user_overlapped = lpOverlapped; wsa->completion_func = lpCompletionRoutine; wsa->iovec = iovec; wsa->n_iovecs = dwBufferCount; wsa->addr = addr; if ( lpOverlapped ) { wsa->async.iosb = (IO_STATUS_BLOCK*)lpOverlapped; wsa->async.event = ( lpCompletionRoutine ? INVALID_HANDLE_VALUE : lpOverlapped->hEvent ); } else { wsa->async.iosb = HeapAlloc ( GetProcessHeap(), 0, sizeof (IO_STATUS_BLOCK) ); if ( !wsa->async.iosb ) goto error; wsa->async.event = INVALID_HANDLE_VALUE; } wsa->async.iosb->Information = 0; TRACE ( "wsa %p, ops %p, h %p, ev %p, fd %d, func %p, iosb %p, uov %p, cfunc %p\n", wsa, wsa->async.ops, wsa->async.handle, wsa->async.event, wsa->async.fd, wsa->async.func, wsa->async.iosb, wsa->user_overlapped, wsa->completion_func ); return wsa; error: TRACE ("Error\n"); HeapFree ( GetProcessHeap(), 0, wsa ); return NULL; } /*********************************************************************** * WS2_recv (INTERNAL) * * Work horse for both synchronous and asynchronous recv() operations. */ static int WS2_recv ( int fd, struct iovec* iov, int count, struct WS_sockaddr *lpFrom, LPINT lpFromlen, LPDWORD lpFlags ) { struct msghdr hdr; int n; TRACE ( "fd %d, iovec %p, count %d addr %s, len %p, flags %lx\n", fd, iov, count, debugstr_sockaddr(lpFrom), lpFromlen, *lpFlags); hdr.msg_name = NULL; if ( lpFrom ) { hdr.msg_namelen = *lpFromlen; hdr.msg_name = ws_sockaddr_alloc ( lpFrom, lpFromlen, &hdr.msg_namelen ); if ( !hdr.msg_name ) { WSASetLastError ( WSAEFAULT ); n = -1; goto out; } } else hdr.msg_namelen = 0; hdr.msg_iov = iov; hdr.msg_iovlen = count; #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 if ( (n = recvmsg (fd, &hdr, *lpFlags)) == -1 ) { TRACE ( "recvmsg error %d\n", errno); goto out; } if ( lpFrom && ws_sockaddr_u2ws ( hdr.msg_name, hdr.msg_namelen, lpFrom, lpFromlen ) != 0 ) { /* The from buffer was too small, but we read the data * anyway. Is that really bad? */ WSASetLastError ( WSAEFAULT ); WARN ( "Address buffer too small\n" ); } out: ws_sockaddr_free ( hdr.msg_name, lpFrom ); TRACE ("-> %d\n", n); return n; } /*********************************************************************** * WS2_async_recv (INTERNAL) * * Handler for overlapped recv() operations. */ static void WS2_async_recv ( async_private *as ) { ws2_async* wsa = (ws2_async*) as; int result, err; TRACE ( "async %p\n", wsa ); if ( wsa->async.iosb->u.Status != STATUS_PENDING ) { TRACE ( "status: %ld\n", wsa->async.iosb->u.Status ); return; } result = WS2_recv ( wsa->async.fd, wsa->iovec, wsa->n_iovecs, wsa->addr, wsa->addrlen.ptr, &wsa->flags ); if (result >= 0) { wsa->async.iosb->u.Status = STATUS_SUCCESS; wsa->async.iosb->Information = result; TRACE ( "received %d bytes\n", result ); _enable_event ( wsa->async.handle, FD_READ, 0, 0 ); return; } err = wsaErrno (); if ( err == WSAEINTR || err == WSAEWOULDBLOCK ) /* errno: EINTR / EAGAIN */ { wsa->async.iosb->u.Status = STATUS_PENDING; _enable_event ( wsa->async.handle, FD_READ, 0, 0 ); TRACE ( "still pending\n" ); } else { wsa->async.iosb->u.Status = err; TRACE ( "Error: %x\n", err ); } } /*********************************************************************** * WS2_send (INTERNAL) * * Work horse for both synchronous and asynchronous send() operations. */ static int WS2_send ( int fd, struct iovec* iov, int count, const struct WS_sockaddr *to, INT tolen, DWORD dwFlags ) { struct msghdr hdr; int n = -1; TRACE ( "fd %d, iovec %p, count %d addr %s, len %d, flags %lx\n", fd, iov, count, debugstr_sockaddr(to), tolen, dwFlags); hdr.msg_name = NULL; if ( to ) { hdr.msg_name = (struct sockaddr*) ws_sockaddr_ws2u ( to, tolen, &hdr.msg_namelen ); if ( !hdr.msg_name ) { WSASetLastError ( WSAEFAULT ); goto out; } } else hdr.msg_namelen = 0; hdr.msg_iov = iov; hdr.msg_iovlen = count; #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 n = sendmsg (fd, &hdr, dwFlags); out: ws_sockaddr_free ( hdr.msg_name, to ); return n; } /*********************************************************************** * WS2_async_send (INTERNAL) * * Handler for overlapped send() operations. */ static void WS2_async_send ( async_private *as ) { ws2_async* wsa = (ws2_async*) as; int result, err; TRACE ( "async %p\n", wsa ); if ( wsa->async.iosb->u.Status != STATUS_PENDING ) { TRACE ( "status: %ld\n", wsa->async.iosb->u.Status ); return; } result = WS2_send ( wsa->async.fd, wsa->iovec, wsa->n_iovecs, wsa->addr, wsa->addrlen.val, wsa->flags ); if (result >= 0) { wsa->async.iosb->u.Status = STATUS_SUCCESS; wsa->async.iosb->Information = result; TRACE ( "sent %d bytes\n", result ); _enable_event ( wsa->async.handle, FD_WRITE, 0, 0 ); return; } err = wsaErrno (); if ( err == WSAEINTR ) { wsa->async.iosb->u.Status = STATUS_PENDING; _enable_event ( wsa->async.handle, FD_WRITE, 0, 0 ); TRACE ( "still pending\n" ); } else { /* We set the status to a winsock error code and check for that later in NtStatusToWSAError () */ wsa->async.iosb->u.Status = err; TRACE ( "Error: %x\n", err ); } } /*********************************************************************** * WS2_async_shutdown (INTERNAL) * * Handler for shutdown() operations on overlapped sockets. */ static void WS2_async_shutdown ( async_private *as ) { ws2_async* wsa = (ws2_async*) as; int err = 1; TRACE ( "async %p %d\n", wsa, wsa->async.type ); switch ( wsa->async.type ) { case ASYNC_TYPE_READ: err = shutdown ( wsa->async.fd, 0 ); break; case ASYNC_TYPE_WRITE: err = shutdown ( wsa->async.fd, 1 ); break; default: ERR ("invalid type: %d\n", wsa->async.type ); } if ( err ) wsa->async.iosb->u.Status = wsaErrno (); else wsa->async.iosb->u.Status = STATUS_SUCCESS; } /*********************************************************************** * WS2_register_async_shutdown (INTERNAL) * * Helper function for WS_shutdown() on overlapped sockets. */ static int WS2_register_async_shutdown ( SOCKET s, int fd, int type ) { struct ws2_async *wsa; int ret, err = WSAEFAULT; DWORD dwflags = 0; int len = 0; LPWSAOVERLAPPED ovl = HeapAlloc (GetProcessHeap(), 0, sizeof ( WSAOVERLAPPED )); TRACE ("s %d fd %d type %d\n", s, fd, type); if (!ovl) goto out; ovl->hEvent = WSACreateEvent (); if ( ovl->hEvent == WSA_INVALID_EVENT ) goto out_free; wsa = WS2_make_async ( s, fd, type, NULL, 0, &dwflags, NULL, &len, ovl, NULL ); if ( !wsa ) goto out_close; /* Hack: this will cause ws2_async_cleanup() to free the overlapped structure */ wsa->user_overlapped = NULL; wsa->async.func = WS2_async_shutdown; if ( (ret = register_new_async ( &wsa->async )) ) { err = NtStatusToWSAError ( ret ); goto out; } /* Try immediate completion */ while ( WaitForSingleObjectEx ( ovl->hEvent, 0, TRUE ) == STATUS_USER_APC ); return 0; out_close: WSACloseEvent ( ovl->hEvent ); out_free: HeapFree ( GetProcessHeap(), 0, ovl ); out: return err; } /*********************************************************************** * accept (WS2_32.1) */ SOCKET WINAPI WS_accept(SOCKET s, struct WS_sockaddr *addr, int *addrlen32) { SOCKET as; TRACE("socket %04x\n", s ); if (_is_blocking(s)) { int fd = get_sock_fd( s, GENERIC_READ, NULL ); if (fd == -1) return INVALID_SOCKET; /* block here */ do_block(fd, POLLIN); _sync_sock_state(s); /* let wineserver notice connection */ release_sock_fd( s, fd ); /* retrieve any error codes from it */ SetLastError(_get_sock_error(s, FD_ACCEPT_BIT)); /* FIXME: care about the error? */ } SERVER_START_REQ( accept_socket ) { req->lhandle = SOCKET2HANDLE(s); req->access = GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE; req->inherit = TRUE; set_error( wine_server_call( req ) ); as = HANDLE2SOCKET( reply->handle ); } SERVER_END_REQ; if (as) { if (addr) WS_getpeername(as, addr, addrlen32); return as; } return INVALID_SOCKET; } /*********************************************************************** * 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 %04x, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen); if (fd != -1) { if (!name || !SUPPORTED_PF(name->sa_family)) { SetLastError(WSAEAFNOSUPPORT); } else { const struct sockaddr* uaddr; int uaddrlen; uaddr=ws_sockaddr_ws2u(name,namelen,&uaddrlen); if (uaddr == NULL) { SetLastError(WSAEFAULT); } else { int on = 1; /* The game GrandPrixLegends binds more than one time, but does * not do a SO_REUSEADDR - Stevens says this is ok */ TRACE( "Setting WS_SO_REUSEADDR on socket before we bind it\n"); WS_setsockopt( s, WS_SOL_SOCKET, WS_SO_REUSEADDR, (char*)&on, sizeof(on) ); if (bind(fd, uaddr, uaddrlen) < 0) { int loc_errno = errno; WARN("\tfailure - errno = %i\n", errno); errno = loc_errno; switch (errno) { case EBADF: SetLastError(WSAENOTSOCK); break; case EADDRNOTAVAIL: SetLastError(WSAEINVAL); break; default: SetLastError(wsaErrno()); break; } } else { res=0; /* success */ } ws_sockaddr_free(uaddr,name); } } release_sock_fd( s, fd ); } return res; } /*********************************************************************** * closesocket (WS2_32.3) */ int WINAPI WS_closesocket(SOCKET s) { TRACE("socket %08x\n", s); if (CloseHandle(SOCKET2HANDLE(s))) return 0; return SOCKET_ERROR; } /*********************************************************************** * connect (WS2_32.4) */ int WINAPI WS_connect(SOCKET s, const struct WS_sockaddr* name, int namelen) { int fd = get_sock_fd( s, GENERIC_READ, NULL ); TRACE("socket %04x, ptr %p %s, length %d\n", s, name, debugstr_sockaddr(name), namelen); if (fd != -1) { const struct sockaddr* uaddr; int uaddrlen; uaddr=ws_sockaddr_ws2u(name,namelen,&uaddrlen); if (uaddr == NULL) { SetLastError(WSAEFAULT); } else { int rc; rc=connect(fd, uaddr, uaddrlen); ws_sockaddr_free(uaddr,name); if (rc == 0) goto connect_success; } if (errno == EINPROGRESS) { /* tell wineserver that a connection is in progress */ _enable_event(SOCKET2HANDLE(s), FD_CONNECT|FD_READ|FD_WRITE, FD_CONNECT|FD_READ|FD_WRITE, FD_WINE_CONNECTED|FD_WINE_LISTENING); if (_is_blocking(s)) { int result; /* block here */ do_block(fd, POLLIN | POLLOUT ); _sync_sock_state(s); /* let wineserver notice connection */ /* retrieve any error codes from it */ result = _get_sock_error(s, FD_CONNECT_BIT); if (result) SetLastError(result); else { goto connect_success; } } else { SetLastError(WSAEWOULDBLOCK); } } else { SetLastError(wsaErrno()); } release_sock_fd( s, fd ); } 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); 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 ); } /*********************************************************************** * getpeername (WS2_32.5) */ int WINAPI WS_getpeername(SOCKET s, struct WS_sockaddr *name, int *namelen) { int fd; int res; TRACE("socket: %04x, ptr %p, len %8x\n", s, name, *namelen); /* 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) { struct sockaddr* uaddr; int uaddrlen; uaddr=ws_sockaddr_alloc(name,namelen,&uaddrlen); if (getpeername(fd, uaddr, &uaddrlen) != 0) { SetLastError(wsaErrno()); } else if (ws_sockaddr_u2ws(uaddr,uaddrlen,name,namelen) != 0) { /* The buffer was too small */ SetLastError(WSAEFAULT); } else { res=0; } ws_sockaddr_free(uaddr,name); release_sock_fd( s, fd ); } return res; } /*********************************************************************** * getsockname (WS2_32.6) */ int WINAPI WS_getsockname(SOCKET s, struct WS_sockaddr *name, int *namelen) { int fd; int res; TRACE("socket: %04x, ptr %p, len %8x\n", s, name, *namelen); /* 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) { struct sockaddr* uaddr; int uaddrlen; uaddr=ws_sockaddr_alloc(name,namelen,&uaddrlen); if (getsockname(fd, uaddr, &uaddrlen) != 0) { SetLastError(wsaErrno()); } else if (ws_sockaddr_u2ws(uaddr,uaddrlen,name,namelen) != 0) { /* The buffer was too small */ SetLastError(WSAEFAULT); } else { res=0; } 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; TRACE("socket: %04x, level 0x%x, name 0x%x, ptr %8x, len %d\n", s, level, (int) optname, (int) optval, (int) *optlen); /* SO_OPENTYPE does not require a valid socket handle. */ if (level == WS_SOL_SOCKET && optname == WS_SO_OPENTYPE) { if (!optlen || *optlen < sizeof(int) || !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } *(int *)optval = (int)TlsGetValue( opentype_tls_index ); *optlen = sizeof(int); TRACE("getting global SO_OPENTYPE = 0x%x\n", *((int*)optval) ); return 0; } fd = get_sock_fd( s, 0, NULL ); if (fd != -1) { if (!convert_sockopt(&level, &optname)) { SetLastError(WSAENOPROTOOPT); /* Unknown option */ } else { if (getsockopt(fd, (int) level, optname, optval, optlen) == 0 ) { release_sock_fd( s, fd ); return 0; } SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno()); } release_sock_fd( s, fd ); } return SOCKET_ERROR; } /*********************************************************************** * htonl (WINSOCK.8) * htonl (WS2_32.8) */ u_long WINAPI WS_htonl(u_long hostlong) { return htonl(hostlong); } /*********************************************************************** * htons (WINSOCK.9) * htons (WS2_32.9) */ u_short WINAPI WS_htons(u_short hostshort) { return htons(hostshort); } /*********************************************************************** * WSAHtonl (WS2_32.46) */ int WINAPI WSAHtonl(SOCKET s, u_long hostlong, u_long *lpnetlong) { FIXME("stub.\n"); return INVALID_SOCKET; } /*********************************************************************** * WSAHtons (WS2_32.47) */ int WINAPI WSAHtons(SOCKET s, u_short hostshort, u_short *lpnetshort) { FIXME("stub.\n"); return INVALID_SOCKET; } /*********************************************************************** * inet_addr (WINSOCK.10) * inet_addr (WS2_32.11) */ u_long WINAPI WS_inet_addr(const char *cp) { return inet_addr(cp); } /*********************************************************************** * ntohl (WINSOCK.14) * ntohl (WS2_32.14) */ u_long WINAPI WS_ntohl(u_long netlong) { return ntohl(netlong); } /*********************************************************************** * ntohs (WINSOCK.15) * ntohs (WS2_32.15) */ u_short WINAPI WS_ntohs(u_short netshort) { return ntohs(netshort); } /*********************************************************************** * inet_ntoa (WS2_32.12) */ char* WINAPI WS_inet_ntoa(struct WS_in_addr in) { /* use "buffer for dummies" here because some applications have * propensity to decode addresses in ws_hostent structure without * saving them first... */ static char dbuffer[16]; /* Yes, 16: 4*3 digits + 3 '.' + 1 '\0' */ char* s = inet_ntoa(*((struct in_addr*)&in)); if( s ) { strcpy(dbuffer, s); return dbuffer; } SetLastError(wsaErrno()); return NULL; } /********************************************************************** * WSAIoctl (WS2_32.50) * * * FIXME: Only SIO_GET_INTERFACE_LIST option implemented. */ INT WINAPI WSAIoctl (SOCKET s, DWORD dwIoControlCode, LPVOID lpvInBuffer, DWORD cbInBuffer, LPVOID lpbOutBuffer, DWORD cbOutBuffer, LPDWORD lpcbBytesReturned, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine) { int fd = get_sock_fd( s, 0, NULL ); if (fd == -1) return SOCKET_ERROR; switch( dwIoControlCode ) { case SIO_GET_INTERFACE_LIST: { INTERFACE_INFO* intArray = (INTERFACE_INFO*)lpbOutBuffer; DWORD size, numInt, apiReturn; TRACE ("-> SIO_GET_INTERFACE_LIST request\n"); if (!lpbOutBuffer) { release_sock_fd( s, fd ); WSASetLastError(WSAEFAULT); return SOCKET_ERROR; } if (!lpcbBytesReturned) { release_sock_fd( s, fd ); WSASetLastError(WSAEFAULT); return SOCKET_ERROR; } apiReturn = GetAdaptersInfo(NULL, &size); if (apiReturn == ERROR_NO_DATA) { numInt = 0; } else if (apiReturn == ERROR_BUFFER_OVERFLOW) { PIP_ADAPTER_INFO table = (PIP_ADAPTER_INFO)HeapAlloc(GetProcessHeap(),0,size); if (table) { if (GetAdaptersInfo(table, &size) == NO_ERROR) { PIP_ADAPTER_INFO ptr; if (size > cbOutBuffer) { HeapFree(GetProcessHeap(),0,table); release_sock_fd( s, fd ); WSASetLastError(WSAEFAULT); return (SOCKET_ERROR); } for (ptr = table, numInt = 0; ptr; ptr = ptr->Next, intArray++, numInt++) { unsigned int addr, mask, bcast; struct ifreq ifInfo; /* Socket Status Flags */ strncpy(ifInfo.ifr_name, ptr->AdapterName, IFNAMSIZ); ifInfo.ifr_name[IFNAMSIZ-1] = '\0'; if (ioctl(fd, SIOCGIFFLAGS, &ifInfo) < 0) { ERR ("Error obtaining status flags for socket!\n"); HeapFree(GetProcessHeap(),0,table); release_sock_fd( s, fd ); WSASetLastError(WSAEINVAL); return (SOCKET_ERROR); } else { /* set flags; the values of IFF_* are not the same under Linux and Windows, therefore must generate new flags */ intArray->iiFlags = 0; if (ifInfo.ifr_flags & IFF_BROADCAST) intArray->iiFlags |= WS_IFF_BROADCAST; #ifdef IFF_POINTOPOINT if (ifInfo.ifr_flags & IFF_POINTOPOINT) intArray->iiFlags |= WS_IFF_POINTTOPOINT; #endif if (ifInfo.ifr_flags & IFF_LOOPBACK) intArray->iiFlags |= WS_IFF_LOOPBACK; if (ifInfo.ifr_flags & IFF_UP) intArray->iiFlags |= WS_IFF_UP; } addr = inet_addr(ptr->IpAddressList.IpAddress.String); mask = inet_addr(ptr->IpAddressList.IpMask.String); bcast = addr | (addr & !mask); intArray->iiAddress.AddressIn.sin_family = AF_INET; intArray->iiAddress.AddressIn.sin_port = 0; intArray->iiAddress.AddressIn.sin_addr.WS_s_addr = addr; intArray->iiNetmask.AddressIn.sin_family = AF_INET; intArray->iiNetmask.AddressIn.sin_port = 0; intArray->iiNetmask.AddressIn.sin_addr.WS_s_addr = mask; intArray->iiBroadcastAddress.AddressIn.sin_family = AF_INET; intArray->iiBroadcastAddress.AddressIn.sin_port = 0; intArray->iiBroadcastAddress.AddressIn.sin_addr. WS_s_addr = bcast; } } else { ERR ("Unable to get interface table!\n"); release_sock_fd( s, fd ); HeapFree(GetProcessHeap(),0,table); WSASetLastError(WSAEINVAL); return (SOCKET_ERROR); } HeapFree(GetProcessHeap(),0,table); } else { release_sock_fd( s, fd ); WSASetLastError(WSAEINVAL); return (SOCKET_ERROR); } } else { ERR ("Unable to get interface table!\n"); release_sock_fd( s, fd ); WSASetLastError(WSAEINVAL); return (SOCKET_ERROR); } /* Calculate the size of the array being returned */ *lpcbBytesReturned = sizeof(INTERFACE_INFO) * numInt; break; } default: WARN("\tunsupported WS_IOCTL cmd (%08lx)\n", dwIoControlCode); release_sock_fd( s, fd ); WSASetLastError(WSAEOPNOTSUPP); return (SOCKET_ERROR); } /* Function executed with no errors */ release_sock_fd( s, fd ); return (0); } /*********************************************************************** * ioctlsocket (WS2_32.10) */ int WINAPI WS_ioctlsocket(SOCKET s, long cmd, u_long *argp) { int fd; long newcmd = cmd; TRACE("socket %04x, cmd %08lx, ptr %p\n", s, cmd, argp); switch( cmd ) { case WS_FIONREAD: newcmd=FIONREAD; break; case WS_FIONBIO: newcmd=FIONBIO; if( _get_sock_mask(s) ) { /* AsyncSelect()'ed sockets are always nonblocking */ if (*argp) return 0; SetLastError(WSAEINVAL); return SOCKET_ERROR; } if (*argp) _enable_event(SOCKET2HANDLE(s), 0, FD_WINE_NONBLOCKING, 0); else _enable_event(SOCKET2HANDLE(s), 0, 0, FD_WINE_NONBLOCKING); return 0; case WS_SIOCATMARK: newcmd=SIOCATMARK; break; case WS__IOW('f',125,u_long): WARN("Warning: WS1.1 shouldn't be using async I/O\n"); SetLastError(WSAEINVAL); return SOCKET_ERROR; case SIOCGIFBRDADDR: case SIOCGIFNETMASK: case SIOCGIFADDR: /* These don't need any special handling. They are used by WsControl, and are here to suppress an unecessary warning. */ break; default: /* Netscape tries hard to use bogus ioctl 0x667e */ WARN("\tunknown WS_IOCTL cmd (%08lx)\n", cmd); break; } fd = get_sock_fd( s, 0, NULL ); if (fd != -1) { if( ioctl(fd, newcmd, (char*)argp ) == 0 ) { release_sock_fd( s, fd ); return 0; } SetLastError((errno == EBADF) ? WSAENOTSOCK : wsaErrno()); release_sock_fd( s, fd ); } return SOCKET_ERROR; } /*********************************************************************** * listen (WS2_32.13) */ int WINAPI WS_listen(SOCKET s, int backlog) { int fd = get_sock_fd( s, GENERIC_READ, NULL ); TRACE("socket %04x, backlog %d\n", s, backlog); if (fd != -1) { if (listen(fd, backlog) == 0) { release_sock_fd( s, fd ); _enable_event(SOCKET2HANDLE(s), FD_ACCEPT, FD_WINE_LISTENING, FD_CONNECT|FD_WINE_CONNECTED); return 0; } SetLastError(wsaErrno()); release_sock_fd( s, fd ); } return SOCKET_ERROR; } /*********************************************************************** * 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 ( WSARecvFrom (s, &wsabuf, 1, &n, &dwFlags, 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 ( WSARecvFrom (s, &wsabuf, 1, &n, &dwFlags, from, fromlen, NULL, NULL) == SOCKET_ERROR ) return SOCKET_ERROR; else return n; } /*********************************************************************** * 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) { int highfd = 0; fd_set readfds, writefds, exceptfds; fd_set *p_read, *p_write, *p_except; int readfd[FD_SETSIZE], writefd[FD_SETSIZE], exceptfd[FD_SETSIZE]; struct timeval timeout, *timeoutaddr = NULL; TRACE("read %p, write %p, excp %p timeout %p\n", ws_readfds, ws_writefds, ws_exceptfds, ws_timeout); p_read = fd_set_import(&readfds, ws_readfds, GENERIC_READ, &highfd, readfd); p_write = fd_set_import(&writefds, ws_writefds, GENERIC_WRITE, &highfd, writefd); p_except = fd_set_import(&exceptfds, ws_exceptfds, 0, &highfd, exceptfd); if (ws_timeout) { timeoutaddr = &timeout; timeout.tv_sec=ws_timeout->tv_sec; timeout.tv_usec=ws_timeout->tv_usec; } if( (highfd = select(highfd + 1, p_read, p_write, p_except, timeoutaddr)) > 0 ) { fd_set_export(&readfds, p_except, ws_readfds, readfd); fd_set_export(&writefds, p_except, ws_writefds, writefd); if (p_except && ws_exceptfds) { int i, j; for (i = j = 0; i < ws_exceptfds->fd_count; i++) { int fd = exceptfd[i]; SOCKET s = ws_exceptfds->fd_array[i]; if (fd == -1) continue; if (FD_ISSET(fd, &exceptfds)) ws_exceptfds->fd_array[j++] = s; release_sock_fd( s, fd ); } ws_exceptfds->fd_count = j; } return highfd; } fd_set_unimport(ws_readfds, readfd); fd_set_unimport(ws_writefds, writefd); fd_set_unimport(ws_exceptfds, exceptfd); if( highfd == 0 ) return 0; SetLastError(wsaErrno()); return SOCKET_ERROR; } /*********************************************************************** * 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 ( WSASendTo ( 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 WSASendTo ( 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 ); } /*********************************************************************** * 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 ) { int i, n, fd, err = WSAENOTSOCK, flags, ret; struct iovec* iovec; struct ws2_async *wsa; TRACE ("socket %04x, wsabuf %p, nbufs %ld, flags %ld, to %p, tolen %d, ovl %p, func %p\n", s, lpBuffers, dwBufferCount, dwFlags, to, tolen, lpOverlapped, lpCompletionRoutine); fd = get_sock_fd( s, GENERIC_WRITE, &flags ); TRACE ( "fd=%d, flags=%x\n", fd, flags ); if ( fd == -1 ) return SOCKET_ERROR; if (flags & FD_FLAG_SEND_SHUTDOWN) { WSASetLastError ( WSAESHUTDOWN ); goto err_close; } if ( !lpNumberOfBytesSent ) { err = WSAEFAULT; goto err_close; } iovec = HeapAlloc (GetProcessHeap(), 0, dwBufferCount * sizeof (struct iovec) ); if ( !iovec ) { err = WSAEFAULT; goto err_close; } for ( i = 0; i < dwBufferCount; i++ ) { iovec[i].iov_base = lpBuffers[i].buf; iovec[i].iov_len = lpBuffers[i].len; } if ( (lpOverlapped || lpCompletionRoutine) && flags & FD_FLAG_OVERLAPPED ) { wsa = WS2_make_async ( s, fd, ASYNC_TYPE_WRITE, iovec, dwBufferCount, &dwFlags, (struct WS_sockaddr*) to, &tolen, lpOverlapped, lpCompletionRoutine ); if ( !wsa ) { err = WSAEFAULT; goto err_free; } if ( ( ret = register_new_async ( &wsa->async )) ) { err = NtStatusToWSAError ( ret ); if ( !lpOverlapped ) HeapFree ( GetProcessHeap(), 0, wsa->async.iosb ); HeapFree ( GetProcessHeap(), 0, wsa ); goto err_free; } /* Try immediate completion */ if ( lpOverlapped && !NtResetEvent( lpOverlapped->hEvent, NULL ) ) { if ( WSAGetOverlappedResult ( s, lpOverlapped, lpNumberOfBytesSent, FALSE, &dwFlags) ) return 0; if ( (err = WSAGetLastError ()) != WSA_IO_INCOMPLETE ) goto error; } WSASetLastError ( WSA_IO_PENDING ); return SOCKET_ERROR; } if (_is_blocking(s)) { /* FIXME: exceptfds? */ do_block(fd, POLLOUT); } n = WS2_send ( fd, iovec, dwBufferCount, to, tolen, dwFlags ); if ( n == -1 ) { err = wsaErrno(); if ( err == WSAEWOULDBLOCK ) _enable_event (SOCKET2HANDLE(s), FD_WRITE, 0, 0); goto err_free; } TRACE(" -> %i bytes\n", n); *lpNumberOfBytesSent = n; HeapFree ( GetProcessHeap(), 0, iovec ); release_sock_fd( s, fd ); return 0; err_free: HeapFree ( GetProcessHeap(), 0, iovec ); err_close: release_sock_fd( s, fd ); error: WARN (" -> ERROR %d\n", err); WSASetLastError (err); return SOCKET_ERROR; } /*********************************************************************** * 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 ( WSASendTo (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; TRACE("socket: %04x, level %d, name %d, ptr %p, len %d\n", s, level, optname, optval, optlen); /* SO_OPENTYPE does not require a valid socket handle. */ if (level == WS_SOL_SOCKET && optname == WS_SO_OPENTYPE) { if (optlen < sizeof(int) || !optval) { SetLastError(WSAEFAULT); return SOCKET_ERROR; } TlsSetValue( opentype_tls_index, (LPVOID)*(int *)optval ); TRACE("setting global SO_OPENTYPE to 0x%x\n", *(int *)optval ); return 0; } /* For some reason the game GrandPrixLegends does set SO_DONTROUTE on its * socket. This will either not happen under windows or it is ignored in * windows (but it works in linux and therefor prevents the game to find * games outsite the current network) */ if ( level==WS_SOL_SOCKET && optname==WS_SO_DONTROUTE ) { FIXME("Does windows ignore SO_DONTROUTE?\n"); return 0; } /* Is a privileged and useless operation, so we don't. */ if ((optname == WS_SO_DEBUG) && (level == WS_SOL_SOCKET)) { FIXME("(%d,SOL_SOCKET,SO_DEBUG,%p(%ld)) attempted (is privileged). Ignoring.\n",s,optval,*(DWORD*)optval); return 0; } if(optname == WS_SO_DONTLINGER && level == WS_SOL_SOCKET) { /* This is unique to WinSock and takes special conversion */ linger.l_onoff = *((int*)optval) ? 0: 1; linger.l_linger = 0; optname=SO_LINGER; optval = (char*)&linger; optlen = sizeof(struct linger); level = SOL_SOCKET; } else { if (!convert_sockopt(&level, &optname)) { ERR("Invalid level (%d) or optname (%d)\n", level, optname); SetLastError(WSAENOPROTOOPT); return SOCKET_ERROR; } if (optname == SO_LINGER && optval) { /* yes, uses unsigned short in both win16/win32 */ linger.l_onoff = ((UINT16*)optval)[0]; linger.l_linger = ((UINT16*)optval)[1]; /* FIXME: what is documented behavior if SO_LINGER optval is null?? */ optval = (char*)&linger; optlen = sizeof(struct linger); } else if (optval && optlen < sizeof(int)) { woptval= *((INT16 *) optval); optval= (char*) &woptval; optlen=sizeof(int); } if (level == SOL_SOCKET && is_timeout_option(optname)) { if (optlen == sizeof(UINT32)) { /* WinSock passes miliseconds instead of struct timeval */ tval.tv_usec = *(PUINT32)optval % 1000; tval.tv_sec = *(PUINT32)optval / 1000; /* min of 500 milisec */ if (tval.tv_sec == 0 && tval.tv_usec < 500) tval.tv_usec = 500; 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; } } if (level == SOL_SOCKET && optname == SO_RCVBUF && *(int*)optval < 2048) { WARN("SO_RCVBF for %d bytes is too small: ignored\n", *(int*)optval ); return 0; } } fd = get_sock_fd( s, 0, NULL ); if (fd == -1) return SOCKET_ERROR; if (setsockopt(fd, level, optname, optval, optlen) == 0) { 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, fd0 = -1, fd1 = -1, flags, err = WSAENOTSOCK; unsigned int clear_flags = 0; fd = get_sock_fd( s, 0, &flags ); TRACE("socket %04x, how %i %x\n", s, how, flags ); if (fd == -1) return SOCKET_ERROR; switch( how ) { case 0: /* drop receives */ clear_flags |= FD_READ; break; case 1: /* drop sends */ clear_flags |= FD_WRITE; break; case 2: /* drop all */ clear_flags |= FD_READ|FD_WRITE; default: clear_flags |= FD_WINE_LISTENING; } if ( flags & FD_FLAG_OVERLAPPED ) { switch ( how ) { case SD_RECEIVE: fd0 = fd; break; case SD_SEND: fd1 = fd; break; case SD_BOTH: default: fd0 = fd; fd1 = get_sock_fd ( s, 0, NULL ); break; } if ( fd0 != -1 ) { err = WS2_register_async_shutdown ( s, fd0, ASYNC_TYPE_READ ); if ( err ) { release_sock_fd( s, fd0 ); goto error; } } if ( fd1 != -1 ) { err = WS2_register_async_shutdown ( s, fd1, ASYNC_TYPE_WRITE ); if ( err ) { release_sock_fd( s, fd1 ); goto error; } } } else /* non-overlapped mode */ { if ( shutdown( fd, how ) ) { err = wsaErrno (); release_sock_fd( s, fd ); 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: _enable_event( SOCKET2HANDLE(s), 0, 0, clear_flags ); WSASetLastError ( 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 WSASocketA ( af, type, protocol, NULL, 0, (TlsGetValue(opentype_tls_index) ? 0 : WSA_FLAG_OVERLAPPED) ); } /*********************************************************************** * gethostbyaddr (WS2_32.51) */ struct WS_hostent* WINAPI WS_gethostbyaddr(const char *addr, int len, int type) { struct WS_hostent *retval = NULL; struct hostent* host; #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6 char *extrabuf; int ebufsize=1024; struct hostent hostentry; int locerr=ENOBUFS; host = NULL; extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ; while(extrabuf) { int res = gethostbyaddr_r(addr, len, type, &hostentry, extrabuf, ebufsize, &host, &locerr); if( res != ERANGE) break; ebufsize *=2; extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ; } if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr)); #else EnterCriticalSection( &csWSgetXXXbyYYY ); host = gethostbyaddr(addr, len, type); if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno)); #endif if( host != NULL ) retval = WS_dup_he(host); #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6 HeapFree(GetProcessHeap(),0,extrabuf); #else LeaveCriticalSection( &csWSgetXXXbyYYY ); #endif TRACE("ptr %p, len %d, type %d ret %p\n", addr, len, type, retval); return retval; } /*********************************************************************** * gethostbyname (WS2_32.52) */ struct WS_hostent* WINAPI WS_gethostbyname(const char* name) { struct WS_hostent *retval = NULL; struct hostent* host; #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6 char *extrabuf; int ebufsize=1024; struct hostent hostentry; int locerr = ENOBUFS; #endif char buf[100]; if( !name) { name = buf; if( gethostname( buf, 100) == -1) { SetLastError( WSAENOBUFS); /* appropriate ? */ return retval; } } #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6 host = NULL; extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ; while(extrabuf) { int res = gethostbyname_r(name, &hostentry, extrabuf, ebufsize, &host, &locerr); if( res != ERANGE) break; ebufsize *=2; extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ; } if (!host) SetLastError((locerr < 0) ? wsaErrno() : wsaHerrno(locerr)); #else EnterCriticalSection( &csWSgetXXXbyYYY ); host = gethostbyname(name); if (!host) SetLastError((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno)); #endif if (host) retval = WS_dup_he(host); #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6 HeapFree(GetProcessHeap(),0,extrabuf); #else LeaveCriticalSection( &csWSgetXXXbyYYY ); #endif TRACE( "%s ret %p\n", debugstr_a(name), retval ); return retval; } /*********************************************************************** * getprotobyname (WS2_32.53) */ struct WS_protoent* WINAPI WS_getprotobyname(const char* name) { struct WS_protoent* retval = NULL; #ifdef HAVE_GETPROTOBYNAME struct protoent* proto; EnterCriticalSection( &csWSgetXXXbyYYY ); if( (proto = getprotobyname(name)) != NULL ) { retval = WS_dup_pe(proto); } else { MESSAGE("protocol %s not found; You might want to add " "this to /etc/protocols\n", debugstr_a(name) ); SetLastError(WSANO_DATA); } LeaveCriticalSection( &csWSgetXXXbyYYY ); #endif TRACE( "%s ret %p\n", debugstr_a(name), retval ); return retval; } /*********************************************************************** * getprotobynumber (WS2_32.54) */ struct WS_protoent* WINAPI WS_getprotobynumber(int number) { struct WS_protoent* retval = NULL; #ifdef HAVE_GETPROTOBYNUMBER struct protoent* proto; EnterCriticalSection( &csWSgetXXXbyYYY ); if( (proto = getprotobynumber(number)) != NULL ) { retval = WS_dup_pe(proto); } else { MESSAGE("protocol number %d not found; You might want to add " "this to /etc/protocols\n", number ); SetLastError(WSANO_DATA); } LeaveCriticalSection( &csWSgetXXXbyYYY ); #endif TRACE("%i ret %p\n", number, retval); return retval; } /*********************************************************************** * getservbyname (WS2_32.55) */ struct WS_servent* WINAPI WS_getservbyname(const char *name, const char *proto) { struct WS_servent* retval = NULL; struct servent* serv; char *name_str; char *proto_str = NULL; if (!(name_str = strdup_lower(name))) return NULL; if (proto && *proto) { if (!(proto_str = strdup_lower(proto))) { HeapFree( GetProcessHeap(), 0, name_str ); return NULL; } } EnterCriticalSection( &csWSgetXXXbyYYY ); serv = getservbyname(name_str, proto_str); if( serv != NULL ) { retval = WS_dup_se(serv); } else SetLastError(WSANO_DATA); LeaveCriticalSection( &csWSgetXXXbyYYY ); if (proto_str) HeapFree( GetProcessHeap(), 0, proto_str ); HeapFree( GetProcessHeap(), 0, name_str ); TRACE( "%s, %s ret %p\n", debugstr_a(name), debugstr_a(proto), retval ); return retval; } /*********************************************************************** * getservbyport (WS2_32.56) */ struct WS_servent* WINAPI WS_getservbyport(int port, const char *proto) { struct WS_servent* retval = NULL; #ifdef HAVE_GETSERVBYPORT struct servent* serv; char *proto_str = NULL; if (proto && *proto) { if (!(proto_str = strdup_lower(proto))) return NULL; } EnterCriticalSection( &csWSgetXXXbyYYY ); if( (serv = getservbyport(port, proto_str)) != NULL ) { retval = WS_dup_se(serv); } else SetLastError(WSANO_DATA); LeaveCriticalSection( &csWSgetXXXbyYYY ); if (proto_str) HeapFree( GetProcessHeap(), 0, proto_str ); #endif TRACE("%d (i.e. port %d), %s ret %p\n", port, (int)ntohl(port), debugstr_a(proto), retval); return retval; } /*********************************************************************** * gethostname (WS2_32.57) */ int WINAPI WS_gethostname(char *name, int namelen) { TRACE("name %p, len %d\n", name, namelen); if (gethostname(name, namelen) == 0) { TRACE("<- '%s'\n", name); return 0; } SetLastError((errno == EINVAL) ? WSAEFAULT : wsaErrno()); TRACE("<- ERROR !\n"); return SOCKET_ERROR; } /* ------------------------------------- Windows sockets extensions -- * * * * ------------------------------------------------------------------- */ /*********************************************************************** * WSAEnumNetworkEvents (WS2_32.36) */ int WINAPI WSAEnumNetworkEvents(SOCKET s, WSAEVENT hEvent, LPWSANETWORKEVENTS lpEvent) { int ret; TRACE("%08x, hEvent %p, lpEvent %08x\n", s, hEvent, (unsigned)lpEvent ); SERVER_START_REQ( get_socket_event ) { req->handle = SOCKET2HANDLE(s); req->service = TRUE; req->c_event = hEvent; wine_server_set_reply( req, lpEvent->iErrorCode, sizeof(lpEvent->iErrorCode) ); if (!(ret = wine_server_call(req))) lpEvent->lNetworkEvents = reply->pmask & reply->mask; } SERVER_END_REQ; if (!ret) return 0; SetLastError(WSAEINVAL); return SOCKET_ERROR; } /*********************************************************************** * WSAEventSelect (WS2_32.39) */ int WINAPI WSAEventSelect(SOCKET s, WSAEVENT hEvent, LONG lEvent) { int ret; TRACE("%08x, hEvent %p, event %08x\n", s, hEvent, (unsigned)lEvent ); SERVER_START_REQ( set_socket_event ) { req->handle = SOCKET2HANDLE(s); req->mask = lEvent; req->event = 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 ) { DWORD r; TRACE ( "socket %d ovl %p trans %p, wait %d flags %p\n", s, lpOverlapped, lpcbTransfer, fWait, lpdwFlags ); if ( !(lpOverlapped && lpOverlapped->hEvent) ) { ERR ( "Invalid pointer\n" ); WSASetLastError (WSA_INVALID_PARAMETER); return FALSE; } if ( fWait ) { while ( WaitForSingleObjectEx (lpOverlapped->hEvent, INFINITE, TRUE) == STATUS_USER_APC ); } else if ( lpOverlapped->Internal == STATUS_PENDING ) { /* Wait in order to give APCs a chance to run. */ /* This is cheating, so we must set the event again in case of success - it may be a non-manual reset event. */ while ( (r = WaitForSingleObjectEx (lpOverlapped->hEvent, 0, TRUE)) == STATUS_USER_APC ); if ( r == WAIT_OBJECT_0 ) NtSetEvent ( lpOverlapped->hEvent, NULL ); } if ( lpcbTransfer ) *lpcbTransfer = lpOverlapped->InternalHigh; if ( lpdwFlags ) *lpdwFlags = lpOverlapped->Offset; switch ( lpOverlapped->Internal ) { case STATUS_SUCCESS: return TRUE; case STATUS_PENDING: WSASetLastError ( WSA_IO_INCOMPLETE ); if (fWait) ERR ("PENDING status after waiting!\n"); return FALSE; default: WSASetLastError ( NtStatusToWSAError ( lpOverlapped->Internal )); return FALSE; } } /*********************************************************************** * WSAAsyncSelect (WS2_32.101) */ INT WINAPI WSAAsyncSelect(SOCKET s, HWND hWnd, UINT uMsg, LONG lEvent) { int ret; TRACE("%x, hWnd %p, uMsg %08x, event %08lx\n", s, hWnd, uMsg, lEvent ); SERVER_START_REQ( set_socket_event ) { req->handle = SOCKET2HANDLE(s); req->mask = lEvent; req->event = 0; req->window = 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: unsignealed */ TRACE("\n"); return CreateEventA(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) { SOCKET ret; /* FIXME: The "advanced" parameters of WSASocketA (lpProtocolInfo, g, dwFlags except WSA_FLAG_OVERLAPPED) are ignored. */ TRACE("af=%d type=%d protocol=%d protocol_info=%p group=%d flags=0x%lx\n", af, type, protocol, lpProtocolInfo, g, dwFlags ); /* hack for WSADuplicateSocket */ if (lpProtocolInfo && lpProtocolInfo->dwServiceFlags4 == 0xff00ff00) { ret = lpProtocolInfo->dwCatalogEntryId; TRACE("\tgot duplicate %04x\n", ret); return ret; } /* check the socket family */ switch(af) { #ifdef HAVE_IPX case WS_AF_IPX: af = AF_IPX; #endif case AF_INET: case AF_UNSPEC: break; default: SetLastError(WSAEAFNOSUPPORT); return INVALID_SOCKET; } /* check the socket type */ switch(type) { case WS_SOCK_STREAM: type=SOCK_STREAM; break; case WS_SOCK_DGRAM: type=SOCK_DGRAM; break; case WS_SOCK_RAW: type=SOCK_RAW; break; default: SetLastError(WSAESOCKTNOSUPPORT); return INVALID_SOCKET; } /* check the protocol type */ if ( protocol < 0 ) /* don't support negative values */ { SetLastError(WSAEPROTONOSUPPORT); return INVALID_SOCKET; } if ( af == AF_UNSPEC) /* did they not specify the address family? */ switch(protocol) { case IPPROTO_TCP: if (type == SOCK_STREAM) { af = AF_INET; break; } case IPPROTO_UDP: if (type == SOCK_DGRAM) { af = AF_INET; break; } default: SetLastError(WSAEPROTOTYPE); return INVALID_SOCKET; } SERVER_START_REQ( create_socket ) { req->family = af; req->type = type; req->protocol = protocol; req->access = GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE; req->flags = dwFlags; req->inherit = TRUE; set_error( wine_server_call( req ) ); ret = HANDLE2SOCKET( reply->handle ); } SERVER_END_REQ; if (ret) { TRACE("\tcreated %04x\n", ret ); return ret; } if (GetLastError() == WSAEACCES) /* raw socket denied */ { if (type == SOCK_RAW) MESSAGE("WARNING: Trying to create a socket of type SOCK_RAW, will fail unless running as root\n"); else MESSAGE("WS_SOCKET: not enough privileges to create socket, try running as root\n"); SetLastError(WSAESOCKTNOSUPPORT); } WARN("\t\tfailed!\n"); 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; TRACE("(%d,%8lx(%i))\n", s,(unsigned long)set, i); while (i--) if (set->fd_array[i] == s) return 1; return 0; } /*********************************************************************** * WSAIsBlocking (WINSOCK.114) * 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 (WINSOCK.113) * 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 = WSA_DefaultBlockingHook; return 0; } /* ----------------------------------- end of API stuff */ /* ----------------------------------- helper functions - * * TODO: Merge WS_dup_..() stuff into one function that * would operate with a generic structure containing internal * pointers (via a template of some kind). */ static int list_size(char** l, int item_size) { int i,j = 0; if(l) { for(i=0;l[i];i++) j += (item_size) ? item_size : strlen(l[i]) + 1; j += (i + 1) * sizeof(char*); } return j; } static int list_dup(char** l_src, char** l_to, int item_size) { char *p; int i; for (i = 0; l_src[i]; i++) ; p = (char *)(l_to + i + 1); for (i = 0; l_src[i]; i++) { int count = ( item_size ) ? item_size : strlen(l_src[i]) + 1; memcpy(p, l_src[i], count); l_to[i] = p; p += count; } l_to[i] = NULL; return (p - (char *)l_to); } /* ----- hostent */ /* duplicate hostent entry * and handle all Win16/Win32 dependent things (struct size, ...) *correctly*. * Dito for protoent and servent. */ static struct WS_hostent *WS_dup_he(const struct hostent* p_he) { char *p; struct WS_hostent *p_to; int size = (sizeof(*p_he) + strlen(p_he->h_name) + 1 + list_size(p_he->h_aliases, 0) + list_size(p_he->h_addr_list, p_he->h_length)); if (!(p_to = check_buffer_he(size))) return NULL; p_to->h_addrtype = p_he->h_addrtype; p_to->h_length = p_he->h_length; p = (char *)(p_to + 1); p_to->h_name = p; strcpy(p, p_he->h_name); p += strlen(p) + 1; p_to->h_aliases = (char **)p; p += list_dup(p_he->h_aliases, p_to->h_aliases, 0); p_to->h_addr_list = (char **)p; list_dup(p_he->h_addr_list, p_to->h_addr_list, p_he->h_length); return p_to; } /* ----- protoent */ static struct WS_protoent *WS_dup_pe(const struct protoent* p_pe) { char *p; struct WS_protoent *p_to; int size = (sizeof(*p_pe) + strlen(p_pe->p_name) + 1 + list_size(p_pe->p_aliases, 0)); if (!(p_to = check_buffer_pe(size))) return NULL; p_to->p_proto = p_pe->p_proto; p = (char *)(p_to + 1); p_to->p_name = p; strcpy(p, p_pe->p_name); p += strlen(p) + 1; p_to->p_aliases = (char **)p; list_dup(p_pe->p_aliases, p_to->p_aliases, 0); return p_to; } /* ----- servent */ static struct WS_servent *WS_dup_se(const struct servent* p_se) { char *p; struct WS_servent *p_to; int size = (sizeof(*p_se) + strlen(p_se->s_proto) + 1 + strlen(p_se->s_name) + 1 + list_size(p_se->s_aliases, 0)); if (!(p_to = check_buffer_se(size))) return NULL; p_to->s_port = p_se->s_port; p = (char *)(p_to + 1); p_to->s_name = p; strcpy(p, p_se->s_name); p += strlen(p) + 1; p_to->s_proto = p; strcpy(p, p_se->s_proto); p += strlen(p) + 1; p_to->s_aliases = (char **)p; list_dup(p_se->s_aliases, p_to->s_aliases, 0); return p_to; } /* ----------------------------------- error handling */ UINT wsaErrno(void) { int loc_errno = errno; WARN("errno %d, (%s).\n", loc_errno, strerror(loc_errno)); switch(loc_errno) { case EINTR: return WSAEINTR; case EBADF: return WSAEBADF; case EPERM: case EACCES: return WSAEACCES; case EFAULT: return WSAEFAULT; case EINVAL: return WSAEINVAL; case EMFILE: return WSAEMFILE; case EWOULDBLOCK: return WSAEWOULDBLOCK; case EINPROGRESS: return WSAEINPROGRESS; case EALREADY: return WSAEALREADY; case 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", loc_errno); return WSAEOPNOTSUPP; } } UINT wsaHerrno(int loc_errno) { WARN("h_errno %d.\n", loc_errno); switch(loc_errno) { case HOST_NOT_FOUND: return WSAHOST_NOT_FOUND; case TRY_AGAIN: return WSATRY_AGAIN; case NO_RECOVERY: return WSANO_RECOVERY; case NO_DATA: return WSANO_DATA; case ENOBUFS: return WSAENOBUFS; case 0: return 0; default: WARN("Unknown h_errno %d!\n", loc_errno); return WSAEOPNOTSUPP; } } /*********************************************************************** * WSARecv (WS2_32.67) */ int WINAPI WSARecv (SOCKET s, LPWSABUF lpBuffers, DWORD dwBufferCount, LPDWORD NumberOfBytesReceived, LPDWORD lpFlags, LPWSAOVERLAPPED lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine) { return WSARecvFrom (s, lpBuffers, dwBufferCount, NumberOfBytesReceived, lpFlags, NULL, NULL, lpOverlapped, lpCompletionRoutine); } /*********************************************************************** * 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 ) { int i, n, fd, err = WSAENOTSOCK, flags, ret; struct iovec* iovec; struct ws2_async *wsa; TRACE("socket %04x, wsabuf %p, nbufs %ld, flags %ld, from %p, fromlen %ld, ovl %p, func %p\n", s, lpBuffers, dwBufferCount, *lpFlags, lpFrom, (lpFromlen ? *lpFromlen : -1L), lpOverlapped, lpCompletionRoutine); fd = get_sock_fd( s, GENERIC_READ, &flags ); TRACE ( "fd=%d, flags=%x\n", fd, flags ); if (fd == -1) return SOCKET_ERROR; if (flags & FD_FLAG_RECV_SHUTDOWN) { WSASetLastError ( WSAESHUTDOWN ); goto err_close; } iovec = HeapAlloc ( GetProcessHeap(), 0, dwBufferCount * sizeof (struct iovec) ); if ( !iovec ) { err = WSAEFAULT; goto err_close; } for (i = 0; i < dwBufferCount; i++) { iovec[i].iov_base = lpBuffers[i].buf; iovec[i].iov_len = lpBuffers[i].len; } if ( (lpOverlapped || lpCompletionRoutine) && flags & FD_FLAG_OVERLAPPED ) { wsa = WS2_make_async ( s, fd, ASYNC_TYPE_READ, iovec, dwBufferCount, lpFlags, lpFrom, lpFromlen, lpOverlapped, lpCompletionRoutine ); if ( !wsa ) { err = WSAEFAULT; goto err_free; } if ( ( ret = register_new_async ( &wsa->async )) ) { err = NtStatusToWSAError ( ret ); if ( !lpOverlapped ) HeapFree ( GetProcessHeap(), 0, wsa->async.iosb ); HeapFree ( GetProcessHeap(), 0, wsa ); goto err_free; } /* Try immediate completion */ if ( lpOverlapped && !NtResetEvent( lpOverlapped->hEvent, NULL ) ) { if ( WSAGetOverlappedResult ( s, lpOverlapped, lpNumberOfBytesRecvd, FALSE, lpFlags) ) return 0; if ( (err = WSAGetLastError ()) != WSA_IO_INCOMPLETE ) goto error; } WSASetLastError ( WSA_IO_PENDING ); return SOCKET_ERROR; } if ( _is_blocking(s) ) { /* block here */ /* FIXME: OOB and exceptfds? */ do_block(fd, POLLIN); } n = WS2_recv ( fd, iovec, dwBufferCount, lpFrom, lpFromlen, lpFlags ); if ( n == -1 ) { err = wsaErrno(); goto err_free; } TRACE(" -> %i bytes\n", n); *lpNumberOfBytesRecvd = n; HeapFree (GetProcessHeap(), 0, iovec); release_sock_fd( s, fd ); _enable_event(SOCKET2HANDLE(s), FD_READ, 0, 0); return 0; err_free: HeapFree (GetProcessHeap(), 0, iovec); err_close: release_sock_fd( s, fd ); error: WARN(" -> ERROR %d\n", err); WSASetLastError ( err ); return SOCKET_ERROR; } /*********************************************************************** * WSCInstallProvider (WS2_32.88) */ INT WINAPI WSCInstallProvider( const LPGUID lpProviderId, LPCWSTR lpszProviderDllPath, const LPWSAPROTOCOL_INFOW lpProtocolInfoList, DWORD dwNumberOfEntries, LPINT lpErrno ) { FIXME("(%s, %s, %p, %ld, %p): stub !\n", debugstr_guid(lpProviderId), debugstr_w(lpszProviderDllPath), lpProtocolInfoList, dwNumberOfEntries, lpErrno); *lpErrno = 0; return 0; } /*********************************************************************** * WSCDeinstallProvider (WS2_32.83) */ INT WINAPI WSCDeinstallProvider(LPGUID lpProviderId, LPINT lpErrno) { FIXME("(%s, %p): stub !\n", debugstr_guid(lpProviderId), lpErrno); *lpErrno = 0; return 0; } /*********************************************************************** * WSAAccept (WS2_32.26) */ SOCKET WINAPI WSAAccept( SOCKET s, struct WS_sockaddr *addr, LPINT addrlen, LPCONDITIONPROC lpfnCondition, DWORD dwCallbackData) { int ret = 0, size = 0; WSABUF CallerId, CallerData, CalleeId, CalleeData; /* QOS SQOS, GQOS; */ GROUP g; SOCKET cs; SOCKADDR src_addr, dst_addr; TRACE("Socket %u, sockaddr %p, addrlen %p, fnCondition %p, dwCallbackData %ld\n", s, addr, addrlen, lpfnCondition, dwCallbackData); size = sizeof(src_addr); cs = WS_accept(s, &src_addr, &size); if (cs == SOCKET_ERROR) return SOCKET_ERROR; CallerId.buf = (char *)&src_addr; CallerId.len = sizeof(src_addr); CallerData.buf = NULL; CallerData.len = (ULONG)NULL; 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: if (addr && addrlen) addr = memcpy(addr, &src_addr, (*addrlen > size) ? size : *addrlen ); return cs; case CF_DEFER: SERVER_START_REQ ( set_socket_deferred ) { req->handle = SOCKET2HANDLE (s); req->deferred = 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; } } /*********************************************************************** * WSAEnumProtocolsA (WS2_32.37) */ int WINAPI WSAEnumProtocolsA(LPINT lpiProtocols, LPWSAPROTOCOL_INFOA lpProtocolBuffer, LPDWORD lpdwBufferLength) { FIXME("(%p,%p,%p): stub\n", lpiProtocols,lpProtocolBuffer, lpdwBufferLength); return 0; } /*********************************************************************** * WSAEnumProtocolsW (WS2_32.38) */ int WINAPI WSAEnumProtocolsW(LPINT lpiProtocols, LPWSAPROTOCOL_INFOW lpProtocolBuffer, LPDWORD lpdwBufferLength) { FIXME("(%p,%p,%p): stub\n", lpiProtocols,lpProtocolBuffer, lpdwBufferLength); return 0; } /*********************************************************************** * WSADuplicateSocketA (WS2_32.32) */ int WINAPI WSADuplicateSocketA( SOCKET s, DWORD dwProcessId, LPWSAPROTOCOL_INFOA lpProtocolInfo ) { HANDLE hProcess; TRACE("(%d,%lx,%p)\n", s, dwProcessId, lpProtocolInfo); memset(lpProtocolInfo, 0, sizeof(*lpProtocolInfo)); /* FIXME: WS_getsockopt(s, WS_SOL_SOCKET, SO_PROTOCOL_INFO, lpProtocolInfo, sizeof(*lpProtocolInfo)); */ /* 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 */ hProcess = OpenProcess(PROCESS_DUP_HANDLE, FALSE, dwProcessId); DuplicateHandle(GetCurrentProcess(), SOCKET2HANDLE(s), hProcess, (LPHANDLE)&lpProtocolInfo->dwCatalogEntryId, 0, FALSE, DUPLICATE_SAME_ACCESS); CloseHandle(hProcess); lpProtocolInfo->dwServiceFlags4 = 0xff00ff00; /* magic */ return 0; } /*********************************************************************** * WSAInstallServiceClassA (WS2_32.48) */ int WINAPI WSAInstallServiceClassA(LPWSASERVICECLASSINFOA info) { FIXME("Request to install service %s\n",debugstr_a(info->lpszServiceClassName)); WSASetLastError(WSAEACCES); return SOCKET_ERROR; } /*********************************************************************** * WSAInstallServiceClassW (WS2_32.49) */ int WINAPI WSAInstallServiceClassW(LPWSASERVICECLASSINFOW info) { FIXME("Request to install service %s\n",debugstr_w(info->lpszServiceClassName)); WSASetLastError(WSAEACCES); return SOCKET_ERROR; } /*********************************************************************** * WSARemoveServiceClass (WS2_32.70) */ int WINAPI WSARemoveServiceClass(LPGUID info) { FIXME("Request to remove service %p\n",info); WSASetLastError(WSATYPE_NOT_FOUND); return SOCKET_ERROR; }