/* Async WINSOCK DNS services * * (C) 1993,1994,1996,1997 John Brezak, Erik Bos, Alex Korobka. * (C) 1999 Marcus Meissner * * 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). * * FIXME: * - Add WSACancel* and correct handle management. (works rather well for * now without it.) * - Verify & Check all calls for correctness * (currently only WSAGetHostByName*, WSAGetServByPort* calls) * - Check error returns. * - mirc/mirc32 Finger @linux.kernel.org sometimes fails in threaded mode. * (not sure why) * - This implementation did ignore the "NOTE:" section above (since the * whole stuff did not work anyway to other changes). */ #include "config.h" #include #include #ifdef HAVE_SYS_IPC_H # include #endif #include #ifdef HAVE_SYS_FILIO_H # include #endif #if defined(__svr4__) #include #ifdef HAVE_SYS_SOCKIO_H # include #endif #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_SOCKET_H #include #endif #ifdef HAVE_NETINET_IN_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 #include #include #ifdef HAVE_ARPA_NAMESER_H # include #endif #ifdef HAVE_RESOLV_H # include #endif #include "wine/winbase16.h" #include "wingdi.h" #include "winuser.h" #include "winsock2.h" #include "wine/winsock16.h" #include "winnt.h" #include "wine/port.h" #include "debugtools.h" DEFAULT_DEBUG_CHANNEL(winsock); /* critical section to protect some non-rentrant net function */ CRITICAL_SECTION csWSgetXXXbyYYY = CRITICAL_SECTION_INIT; /* protoptypes of some functions in socket.c */ UINT16 wsaErrno(void); UINT16 wsaHerrno(int errnr); #define AQ_WIN16 0x00 #define AQ_WIN32 0x04 #define HB_WIN32(hb) (hb->flags & AQ_WIN32) #define AQ_NUMBER 0x00 #define AQ_NAME 0x08 #define AQ_COPYPTR1 0x10 #define AQ_DUPLOWPTR1 0x20 #define AQ_MASKPTR1 0x30 #define AQ_COPYPTR2 0x40 #define AQ_DUPLOWPTR2 0x80 #define AQ_MASKPTR2 0xC0 #define AQ_GETHOST 0 #define AQ_GETPROTO 1 #define AQ_GETSERV 2 #define AQ_GETMASK 3 /* ----------------------------------- helper functions - */ 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* ref, char* base, int item_size) { /* base is either either equal to ref or 0 or SEGPTR */ char* p = ref; char** l_to = (char**)ref; int i,j,k; for(j=0;l_src[j];j++) ; p += (j + 1) * sizeof(char*); for(i=0;ih_name) + 1; size += list_size(p_he->h_aliases, 0); size += list_size(p_he->h_addr_list, p_he->h_length ); } return size; } /* Copy hostent to p_to, fix up inside pointers using p_base (different for * Win16 (linear vs. segmented). Return -neededsize on overrun. */ static int WS_copy_he(char *p_to,char *p_base,int t_size,struct hostent* p_he, int flag) { char* p_name,*p_aliases,*p_addr,*p; struct ws_hostent16 *p_to16 = (struct ws_hostent16*)p_to; struct ws_hostent32 *p_to32 = (struct ws_hostent32*)p_to; int size = hostent_size(p_he) + ( (flag & AQ_WIN16) ? sizeof(struct ws_hostent16) : sizeof(struct ws_hostent32) - sizeof(struct hostent) ); if (t_size < size) return -size; p = p_to; p += (flag & AQ_WIN16) ? sizeof(struct ws_hostent16) : sizeof(struct ws_hostent32); p_name = p; strcpy(p, p_he->h_name); p += strlen(p) + 1; p_aliases = p; p += list_dup(p_he->h_aliases, p, p_base + (p - (char*)p_to), 0); p_addr = p; list_dup(p_he->h_addr_list, p, p_base + (p - (char*)p_to), p_he->h_length); if (flag & AQ_WIN16) { p_to16->h_addrtype = (INT16)p_he->h_addrtype; p_to16->h_length = (INT16)p_he->h_length; p_to16->h_name = (SEGPTR)(p_base + (p_name - p_to)); p_to16->h_aliases = (SEGPTR)(p_base + (p_aliases - p_to)); p_to16->h_addr_list = (SEGPTR)(p_base + (p_addr - p_to)); } else { p_to32->h_addrtype = p_he->h_addrtype; p_to32->h_length = p_he->h_length; p_to32->h_name = (p_base + (p_name - p_to)); p_to32->h_aliases = (char **)(p_base + (p_aliases - p_to)); p_to32->h_addr_list = (char **)(p_base + (p_addr - p_to)); } return size; } /* ----- protoent */ static int protoent_size(struct protoent* p_pe) { int size = 0; if( p_pe ) { size = sizeof(struct protoent); size += strlen(p_pe->p_name) + 1; size += list_size(p_pe->p_aliases, 0); } return size; } /* Copy protoent to p_to, fix up inside pointers using p_base (different for * Win16 (linear vs. segmented). Return -neededsize on overrun. */ static int WS_copy_pe(char *p_to,char *p_base,int t_size,struct protoent* p_pe, int flag) { char* p_name,*p_aliases,*p; struct ws_protoent16 *p_to16 = (struct ws_protoent16*)p_to; struct ws_protoent32 *p_to32 = (struct ws_protoent32*)p_to; int size = protoent_size(p_pe) + ( (flag & AQ_WIN16) ? sizeof(struct ws_protoent16) : sizeof(struct ws_protoent32) - sizeof(struct protoent) ); if (t_size < size) return -size; p = p_to; p += (flag & AQ_WIN16) ? sizeof(struct ws_protoent16) : sizeof(struct ws_protoent32); p_name = p; strcpy(p, p_pe->p_name); p += strlen(p) + 1; p_aliases = p; list_dup(p_pe->p_aliases, p, p_base + (p - (char*)p_to), 0); if (flag & AQ_WIN16) { p_to16->p_proto = (INT16)p_pe->p_proto; p_to16->p_name = (SEGPTR)(p_base) + (p_name - p_to); p_to16->p_aliases = (SEGPTR)((p_base) + (p_aliases - p_to)); } else { p_to32->p_proto = p_pe->p_proto; p_to32->p_name = (p_base) + (p_name - p_to); p_to32->p_aliases = (char **)((p_base) + (p_aliases - p_to)); } return size; } /* ----- servent */ static int servent_size(struct servent* p_se) { int size = 0; if( p_se ) { size += sizeof(struct servent); size += strlen(p_se->s_proto) + strlen(p_se->s_name) + 2; size += list_size(p_se->s_aliases, 0); } return size; } /* Copy servent to p_to, fix up inside pointers using p_base (different for * Win16 (linear vs. segmented). Return -neededsize on overrun. * Take care of different Win16/Win32 servent structs (packing !) */ static int WS_copy_se(char *p_to,char *p_base,int t_size,struct servent* p_se, int flag) { char* p_name,*p_aliases,*p_proto,*p; struct ws_servent16 *p_to16 = (struct ws_servent16*)p_to; struct ws_servent32 *p_to32 = (struct ws_servent32*)p_to; int size = servent_size(p_se) + ( (flag & AQ_WIN16) ? sizeof(struct ws_servent16) : sizeof(struct ws_servent32) - sizeof(struct servent) ); if (t_size < size) return -size; p = p_to; p += (flag & AQ_WIN16) ? sizeof(struct ws_servent16) : sizeof(struct ws_servent32); p_name = p; strcpy(p, p_se->s_name); p += strlen(p) + 1; p_proto = p; strcpy(p, p_se->s_proto); p += strlen(p) + 1; p_aliases = p; list_dup(p_se->s_aliases, p, p_base + (p - p_to), 0); if (flag & AQ_WIN16) { p_to16->s_port = (INT16)p_se->s_port; p_to16->s_name = (SEGPTR)(p_base + (p_name - p_to)); p_to16->s_proto = (SEGPTR)(p_base + (p_proto - p_to)); p_to16->s_aliases = (SEGPTR)(p_base + (p_aliases - p_to)); } else { p_to32->s_port = p_se->s_port; p_to32->s_name = (p_base + (p_name - p_to)); p_to32->s_proto = (p_base + (p_proto - p_to)); p_to32->s_aliases = (char **)(p_base + (p_aliases - p_to)); } return size; } static HANDLE __ws_async_handle = 0xdead; /* Generic async query struct. we use symbolic names for the different queries * for readability. */ typedef struct _async_query { HWND16 hWnd; UINT16 uMsg; LPCSTR ptr1; #define host_name ptr1 #define host_addr ptr1 #define serv_name ptr1 #define proto_name ptr1 LPCSTR ptr2; #define serv_proto ptr2 int int1; #define host_len int1 #define proto_number int1 #define serv_port int1 int int2; #define host_type int2 SEGPTR sbuf; INT16 sbuflen; HANDLE16 async_handle; int flags; int qt; char xbuf[1]; } async_query; /**************************************************************************** * The async query function. * * It is either called as a thread startup routine or directly. It has * to free the passed arg from the process heap and PostMessageA the async * result or the error code. * * FIXME: * - errorhandling not verified. */ static DWORD WINAPI _async_queryfun(LPVOID arg) { async_query *aq = (async_query*)arg; int size = 0; WORD fail = 0; char *targetptr = (HB_WIN32(aq)?(char*)aq->sbuf:(char*)MapSL(aq->sbuf)); switch (aq->flags & AQ_GETMASK) { case AQ_GETHOST: { struct hostent *he; char *copy_hostent = targetptr; #if HAVE_LINUX_GETHOSTBYNAME_R_6 char *extrabuf; int ebufsize=1024; struct hostent hostentry; int locerr = ENOBUFS; he = NULL; extrabuf=HeapAlloc(GetProcessHeap(),0,ebufsize) ; while(extrabuf) { int res = (aq->flags & AQ_NAME) ? gethostbyname_r(aq->host_name, &hostentry, extrabuf, ebufsize, &he, &locerr): gethostbyaddr_r(aq->host_addr,aq->host_len,aq->host_type, &hostentry, extrabuf, ebufsize, &he, &locerr); if( res != ERANGE) break; ebufsize *=2; extrabuf=HeapReAlloc(GetProcessHeap(),0,extrabuf,ebufsize) ; } if (!he) fail = ((locerr < 0) ? wsaErrno() : wsaHerrno(locerr)); #else EnterCriticalSection( &csWSgetXXXbyYYY ); he = (aq->flags & AQ_NAME) ? gethostbyname(aq->host_name): gethostbyaddr(aq->host_addr,aq->host_len,aq->host_type); if (!he) fail = ((h_errno < 0) ? wsaErrno() : wsaHerrno(h_errno)); #endif if (he) { size = WS_copy_he(copy_hostent,(char*)aq->sbuf,aq->sbuflen,he,aq->flags); if (size < 0) { fail = WSAENOBUFS; size = -size; } } #if HAVE_LINUX_GETHOSTBYNAME_R_6 HeapFree(GetProcessHeap(),0,extrabuf); #else LeaveCriticalSection( &csWSgetXXXbyYYY ); #endif } break; case AQ_GETPROTO: { struct protoent *pe; char *copy_protoent = targetptr; EnterCriticalSection( &csWSgetXXXbyYYY ); pe = (aq->flags & AQ_NAME)? getprotobyname(aq->proto_name) : getprotobynumber(aq->proto_number); if (pe) { size = WS_copy_pe(copy_protoent,(char*)aq->sbuf,aq->sbuflen,pe,aq->flags); if (size < 0) { fail = WSAENOBUFS; size = -size; } } else { if (aq->flags & AQ_NAME) MESSAGE("protocol %s not found; You might want to add " "this to /etc/protocols\n", debugstr_a(aq->proto_name) ); else MESSAGE("protocol number %d not found; You might want to add " "this to /etc/protocols\n", aq->proto_number ); fail = WSANO_DATA; } LeaveCriticalSection( &csWSgetXXXbyYYY ); } break; case AQ_GETSERV: { struct servent *se; char *copy_servent = targetptr; EnterCriticalSection( &csWSgetXXXbyYYY ); se = (aq->flags & AQ_NAME)? getservbyname(aq->serv_name,aq->serv_proto) : getservbyport(aq->serv_port,aq->serv_proto); if (se) { size = WS_copy_se(copy_servent,(char*)aq->sbuf,aq->sbuflen,se,aq->flags); if (size < 0) { fail = WSAENOBUFS; size = -size; } } else { if (aq->flags & AQ_NAME) MESSAGE("service %s protocol %s not found; You might want to add " "this to /etc/services\n", debugstr_a(aq->serv_name) , aq->serv_proto ? debugstr_a(aq->serv_proto ):"*"); else MESSAGE("service on port %d protocol %s not found; You might want to add " "this to /etc/services\n", aq->serv_port, aq->serv_proto ? debugstr_a(aq->serv_proto ):"*"); fail = WSANO_DATA; } LeaveCriticalSection( &csWSgetXXXbyYYY ); } break; } PostMessageA(aq->hWnd,aq->uMsg,aq->async_handle,size|(fail<<16)); HeapFree(GetProcessHeap(),0,arg); return 0; } /**************************************************************************** * The main async help function. * * It either starts a thread or just calls the function directly for platforms * with no thread support. This relies on the fact that PostMessage() does * not actually call the windowproc before the function returns. */ static HANDLE16 __WSAsyncDBQuery( HWND hWnd, UINT uMsg,INT int1,LPCSTR ptr1, INT int2, LPCSTR ptr2, void *sbuf, INT sbuflen, UINT flags ) { async_query* aq; char* pto; LPCSTR pfm; int xbuflen = 0; /* allocate buffer to copy protocol- and service name to */ /* note: this is done in the calling thread so we can return */ /* a decent error code if the Alloc fails */ switch (flags & AQ_MASKPTR1) { case 0: break; case AQ_COPYPTR1: xbuflen += int1; break; case AQ_DUPLOWPTR1: xbuflen += strlen(ptr1) + 1; break; } switch (flags & AQ_MASKPTR2) { case 0: break; case AQ_COPYPTR2: xbuflen += int2; break; case AQ_DUPLOWPTR2: xbuflen += strlen(ptr2) + 1; break; } if(!(aq = HeapAlloc(GetProcessHeap(),0,sizeof(async_query) + xbuflen))) { SetLastError(WSAEWOULDBLOCK); /* insufficient resources */ return 0; } pto = aq->xbuf; if (ptr1) switch (flags & AQ_MASKPTR1) { case 0: break; case AQ_COPYPTR1: memcpy(pto, ptr1, int1); ptr1 = pto; pto += int1; break; case AQ_DUPLOWPTR1: pfm = ptr1; ptr1 = pto; do *pto++ = tolower(*pfm); while (*pfm++); break; } if (ptr2) switch (flags & AQ_MASKPTR2) { case 0: break; case AQ_COPYPTR2: memcpy(pto, ptr2, int2); ptr2 = pto; pto += int2; break; case AQ_DUPLOWPTR2: pfm = ptr2; ptr2 = pto; do *pto++ = tolower(*pfm); while (*pfm++); break; } aq->hWnd = hWnd; aq->uMsg = uMsg; aq->int1 = int1; aq->ptr1 = ptr1; aq->int2 = int2; aq->ptr2 = ptr2; aq->async_handle = ++__ws_async_handle; aq->flags = flags; aq->sbuf = (SEGPTR)sbuf; aq->sbuflen = sbuflen; #if 1 if (CreateThread(NULL,0,_async_queryfun,aq,0,NULL) == INVALID_HANDLE_VALUE) #endif _async_queryfun(aq); return __ws_async_handle; } /*********************************************************************** * WSAAsyncGetHostByAddr (WINSOCK.102) */ HANDLE16 WINAPI WSAAsyncGetHostByAddr16(HWND16 hWnd, UINT16 uMsg, LPCSTR addr, INT16 len, INT16 type, SEGPTR sbuf, INT16 buflen) { TRACE("hwnd %04x, msg %04x, addr %08x[%i]\n", hWnd, uMsg, (unsigned)addr , len ); return __WSAsyncDBQuery(hWnd,uMsg,len,addr,type,NULL,(void*)sbuf,buflen, AQ_NUMBER|AQ_COPYPTR1|AQ_WIN16|AQ_GETHOST); } /*********************************************************************** * WSAAsyncGetHostByAddr (WS2_32.102) */ HANDLE WINAPI WSAAsyncGetHostByAddr(HWND hWnd, UINT uMsg, LPCSTR addr, INT len, INT type, LPSTR sbuf, INT buflen) { TRACE("hwnd %04x, msg %04x, addr %08x[%i]\n", hWnd, uMsg, (unsigned)addr , len ); return __WSAsyncDBQuery(hWnd,uMsg,len,addr,type,NULL,sbuf,buflen, AQ_NUMBER|AQ_COPYPTR1|AQ_WIN32|AQ_GETHOST); } /*********************************************************************** * WSAAsyncGetHostByName (WINSOCK.103) */ HANDLE16 WINAPI WSAAsyncGetHostByName16(HWND16 hWnd, UINT16 uMsg, LPCSTR name, SEGPTR sbuf, INT16 buflen) { TRACE("hwnd %04x, msg %04x, host %s, buffer %i\n", hWnd, uMsg, (name)?name:"", (int)buflen ); return __WSAsyncDBQuery(hWnd,uMsg,0,name,0,NULL,(void*)sbuf,buflen, AQ_NAME|AQ_DUPLOWPTR1|AQ_WIN16|AQ_GETHOST); } /*********************************************************************** * WSAAsyncGetHostByName (WS2_32.103) */ HANDLE WINAPI WSAAsyncGetHostByName(HWND hWnd, UINT uMsg, LPCSTR name, LPSTR sbuf, INT buflen) { TRACE("hwnd %04x, msg %08x, host %s, buffer %i\n", (HWND16)hWnd, uMsg, (name)?name:"", (int)buflen ); return __WSAsyncDBQuery(hWnd,uMsg,0,name,0,NULL,sbuf,buflen, AQ_NAME|AQ_DUPLOWPTR1|AQ_WIN32|AQ_GETHOST); } /*********************************************************************** * WSAAsyncGetProtoByName (WINSOCK.105) */ HANDLE16 WINAPI WSAAsyncGetProtoByName16(HWND16 hWnd, UINT16 uMsg, LPCSTR name, SEGPTR sbuf, INT16 buflen) { TRACE("hwnd %04x, msg %08x, protocol %s\n", (HWND16)hWnd, uMsg, (name)?name:"" ); return __WSAsyncDBQuery(hWnd,uMsg,0,name,0,NULL,(void*)sbuf,buflen, AQ_NAME|AQ_DUPLOWPTR1|AQ_WIN16|AQ_GETPROTO); } /*********************************************************************** * WSAAsyncGetProtoByName (WS2_32.105) */ HANDLE WINAPI WSAAsyncGetProtoByName(HWND hWnd, UINT uMsg, LPCSTR name, LPSTR sbuf, INT buflen) { TRACE("hwnd %04x, msg %08x, protocol %s\n", (HWND16)hWnd, uMsg, (name)?name:"" ); return __WSAsyncDBQuery(hWnd,uMsg,0,name,0,NULL,sbuf,buflen, AQ_NAME|AQ_DUPLOWPTR1|AQ_WIN32|AQ_GETPROTO); } /*********************************************************************** * WSAAsyncGetProtoByNumber (WINSOCK.104) */ HANDLE16 WINAPI WSAAsyncGetProtoByNumber16(HWND16 hWnd,UINT16 uMsg,INT16 number, SEGPTR sbuf, INT16 buflen) { TRACE("hwnd %04x, msg %04x, num %i\n", hWnd, uMsg, number ); return __WSAsyncDBQuery(hWnd,uMsg,number,NULL,0,NULL,(void*)sbuf,buflen, AQ_GETPROTO|AQ_NUMBER|AQ_WIN16); } /*********************************************************************** * WSAAsyncGetProtoByNumber (WS2_32.104) */ HANDLE WINAPI WSAAsyncGetProtoByNumber(HWND hWnd, UINT uMsg, INT number, LPSTR sbuf, INT buflen) { TRACE("hwnd %04x, msg %04x, num %i\n", hWnd, uMsg, number ); return __WSAsyncDBQuery(hWnd,uMsg,number,NULL,0,NULL,sbuf,buflen, AQ_GETPROTO|AQ_NUMBER|AQ_WIN32); } /*********************************************************************** * WSAAsyncGetServByName (WINSOCK.107) */ HANDLE16 WINAPI WSAAsyncGetServByName16(HWND16 hWnd, UINT16 uMsg, LPCSTR name, LPCSTR proto, SEGPTR sbuf, INT16 buflen) { TRACE("hwnd %04x, msg %04x, name %s, proto %s\n", hWnd, uMsg, (name)?name:"", (proto)?proto:""); return __WSAsyncDBQuery(hWnd,uMsg,0,name,0,proto,(void*)sbuf,buflen, AQ_GETSERV|AQ_NAME|AQ_DUPLOWPTR1|AQ_DUPLOWPTR2|AQ_WIN16); } /*********************************************************************** * WSAAsyncGetServByName (WS2_32.107) */ HANDLE WINAPI WSAAsyncGetServByName(HWND hWnd, UINT uMsg, LPCSTR name, LPCSTR proto, LPSTR sbuf, INT buflen) { TRACE("hwnd %04x, msg %04x, name %s, proto %s\n", hWnd, uMsg, (name)?name:"", (proto)?proto:""); return __WSAsyncDBQuery(hWnd,uMsg,0,name,0,proto,sbuf,buflen, AQ_GETSERV|AQ_NAME|AQ_DUPLOWPTR1|AQ_DUPLOWPTR2|AQ_WIN32); } /*********************************************************************** * WSAAsyncGetServByPort (WINSOCK.106) */ HANDLE16 WINAPI WSAAsyncGetServByPort16(HWND16 hWnd, UINT16 uMsg, INT16 port, LPCSTR proto, SEGPTR sbuf, INT16 buflen) { TRACE("hwnd %04x, msg %04x, port %i, proto %s\n", hWnd, uMsg, port, (proto)?proto:"" ); return __WSAsyncDBQuery(hWnd,uMsg,port,NULL,0,proto,(void*)sbuf,buflen, AQ_GETSERV|AQ_NUMBER|AQ_DUPLOWPTR2|AQ_WIN16); } /*********************************************************************** * WSAAsyncGetServByPort (WS2_32.106) */ HANDLE WINAPI WSAAsyncGetServByPort(HWND hWnd, UINT uMsg, INT port, LPCSTR proto, LPSTR sbuf, INT buflen) { TRACE("hwnd %04x, msg %04x, port %i, proto %s\n", hWnd, uMsg, port, (proto)?proto:"" ); return __WSAsyncDBQuery(hWnd,uMsg,port,NULL,0,proto,sbuf,buflen, AQ_GETSERV|AQ_NUMBER|AQ_DUPLOWPTR2|AQ_WIN32); } /*********************************************************************** * WSACancelAsyncRequest (WS2_32.108) */ INT WINAPI WSACancelAsyncRequest(HANDLE hAsyncTaskHandle) { FIXME("(%08x),stub\n", hAsyncTaskHandle); return 0; } /*********************************************************************** * WSACancelAsyncRequest (WINSOCK.108) */ INT16 WINAPI WSACancelAsyncRequest16(HANDLE16 hAsyncTaskHandle) { return (HANDLE16)WSACancelAsyncRequest((HANDLE)hAsyncTaskHandle); }