470 lines
14 KiB
C
470 lines
14 KiB
C
/* Async WINSOCK DNS services
|
|
*
|
|
* Copyright (C) 1993,1994,1996,1997 John Brezak, Erik Bos, Alex Korobka.
|
|
* Copyright (C) 1999 Marcus Meissner
|
|
* Copyright (C) 2009 Alexandre Julliard
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*
|
|
* NOTE: If you make any changes to fix a particular app, make sure
|
|
* they don't break something else like Netscape or telnet and ftp
|
|
* clients and servers (www.winsite.com got a lot of those).
|
|
*
|
|
* 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 <stdarg.h>
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "wingdi.h"
|
|
#include "winuser.h"
|
|
#include "winsock2.h"
|
|
#include "ws2spi.h"
|
|
|
|
#include "wine/debug.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(winsock);
|
|
|
|
|
|
struct async_query_header
|
|
{
|
|
LPARAM (*func)( struct async_query_header *query );
|
|
HWND hWnd;
|
|
UINT uMsg;
|
|
void *sbuf;
|
|
INT sbuflen;
|
|
HANDLE handle;
|
|
};
|
|
|
|
struct async_query_gethostbyname
|
|
{
|
|
struct async_query_header query;
|
|
char *host_name;
|
|
};
|
|
|
|
struct async_query_gethostbyaddr
|
|
{
|
|
struct async_query_header query;
|
|
char *host_addr;
|
|
int host_len;
|
|
int host_type;
|
|
};
|
|
|
|
struct async_query_getprotobyname
|
|
{
|
|
struct async_query_header query;
|
|
char *proto_name;
|
|
};
|
|
|
|
struct async_query_getprotobynumber
|
|
{
|
|
struct async_query_header query;
|
|
int proto_number;
|
|
};
|
|
|
|
struct async_query_getservbyname
|
|
{
|
|
struct async_query_header query;
|
|
char *serv_name;
|
|
char *serv_proto;
|
|
};
|
|
|
|
struct async_query_getservbyport
|
|
{
|
|
struct async_query_header query;
|
|
char *serv_proto;
|
|
int serv_port;
|
|
};
|
|
|
|
|
|
/* ----------------------------------- 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, int item_size)
|
|
{
|
|
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;i<j;i++)
|
|
{ l_to[i] = p;
|
|
k = ( item_size ) ? item_size : strlen(l_src[i]) + 1;
|
|
memcpy(p, l_src[i], k); p += k; }
|
|
l_to[i] = NULL;
|
|
return (p - ref);
|
|
}
|
|
|
|
/* ----- hostent */
|
|
|
|
static LPARAM copy_he(void *base, int size, const struct hostent *he)
|
|
{
|
|
char *p;
|
|
int needed;
|
|
struct hostent *to = base;
|
|
|
|
if (!he) return MAKELPARAM( 0, GetLastError() );
|
|
|
|
needed = sizeof(struct hostent) + strlen(he->h_name) + 1 +
|
|
list_size(he->h_aliases, 0) +
|
|
list_size(he->h_addr_list, he->h_length );
|
|
if (size < needed) return MAKELPARAM( needed, WSAENOBUFS );
|
|
|
|
to->h_addrtype = he->h_addrtype;
|
|
to->h_length = he->h_length;
|
|
p = (char *)(to + 1);
|
|
to->h_name = p;
|
|
strcpy(p, he->h_name); p += strlen(p) + 1;
|
|
to->h_aliases = (char **)p;
|
|
p += list_dup(he->h_aliases, p, 0);
|
|
to->h_addr_list = (char **)p;
|
|
list_dup(he->h_addr_list, p, he->h_length);
|
|
return MAKELPARAM( needed, 0 );
|
|
}
|
|
|
|
static LPARAM async_gethostbyname( struct async_query_header *query )
|
|
{
|
|
struct async_query_gethostbyname *aq = CONTAINING_RECORD( query, struct async_query_gethostbyname, query );
|
|
struct hostent *he = gethostbyname( aq->host_name );
|
|
|
|
return copy_he( query->sbuf, query->sbuflen, he );
|
|
}
|
|
|
|
static LPARAM async_gethostbyaddr( struct async_query_header *query )
|
|
{
|
|
struct async_query_gethostbyaddr *aq = CONTAINING_RECORD( query, struct async_query_gethostbyaddr, query );
|
|
struct hostent *he = gethostbyaddr( aq->host_addr, aq->host_len, aq->host_type );
|
|
|
|
return copy_he( query->sbuf, query->sbuflen, he );
|
|
}
|
|
|
|
/* ----- protoent */
|
|
|
|
static LPARAM copy_pe( void *base, int size, const struct protoent *pe )
|
|
{
|
|
char *p;
|
|
int needed;
|
|
struct protoent *to = base;
|
|
|
|
if (!pe) return MAKELPARAM( 0, GetLastError() );
|
|
|
|
needed = sizeof(struct protoent) + strlen( pe->p_name ) + 1 + list_size( pe->p_aliases, 0 );
|
|
if (size < needed) return MAKELPARAM( needed, WSAENOBUFS );
|
|
|
|
to->p_proto = pe->p_proto;
|
|
p = (char *)(to + 1);
|
|
to->p_name = p;
|
|
strcpy(p, pe->p_name); p += strlen(p) + 1;
|
|
to->p_aliases = (char **)p;
|
|
list_dup(pe->p_aliases, p, 0);
|
|
return MAKELPARAM( needed, 0 );
|
|
}
|
|
|
|
static LPARAM async_getprotobyname( struct async_query_header *query )
|
|
{
|
|
struct async_query_getprotobyname *aq = CONTAINING_RECORD( query, struct async_query_getprotobyname, query );
|
|
struct protoent *pe = getprotobyname( aq->proto_name );
|
|
|
|
return copy_pe( query->sbuf, query->sbuflen, pe );
|
|
}
|
|
|
|
static LPARAM async_getprotobynumber( struct async_query_header *query )
|
|
{
|
|
struct async_query_getprotobynumber *aq = CONTAINING_RECORD( query, struct async_query_getprotobynumber, query );
|
|
struct protoent *pe = getprotobynumber( aq->proto_number );
|
|
|
|
return copy_pe( query->sbuf, query->sbuflen, pe );
|
|
}
|
|
|
|
/* ----- servent */
|
|
|
|
static LPARAM copy_se( void *base, int size, const struct servent *se )
|
|
{
|
|
char *p;
|
|
int needed;
|
|
struct servent *to = base;
|
|
|
|
if (!se) return MAKELPARAM( 0, GetLastError() );
|
|
|
|
needed = sizeof(struct servent) + strlen( se->s_proto ) + strlen( se->s_name ) + 2 + list_size( se->s_aliases, 0 );
|
|
if (size < needed) return MAKELPARAM( needed, WSAENOBUFS );
|
|
|
|
to->s_port = se->s_port;
|
|
p = (char *)(to + 1);
|
|
to->s_name = p;
|
|
strcpy(p, se->s_name); p += strlen(p) + 1;
|
|
to->s_proto = p;
|
|
strcpy(p, se->s_proto); p += strlen(p) + 1;
|
|
to->s_aliases = (char **)p;
|
|
list_dup(se->s_aliases, p, 0);
|
|
return MAKELPARAM( needed, 0 );
|
|
}
|
|
|
|
static LPARAM async_getservbyname( struct async_query_header *query )
|
|
{
|
|
struct async_query_getservbyname *aq = CONTAINING_RECORD( query, struct async_query_getservbyname, query );
|
|
struct servent *se = getservbyname( aq->serv_name, aq->serv_proto );
|
|
|
|
return copy_se( query->sbuf, query->sbuflen, se );
|
|
}
|
|
|
|
static LPARAM async_getservbyport( struct async_query_header *query )
|
|
{
|
|
struct async_query_getservbyport *aq = CONTAINING_RECORD( query, struct async_query_getservbyport, query );
|
|
struct servent *se = getservbyport( aq->serv_port, aq->serv_proto );
|
|
|
|
return copy_se( query->sbuf, query->sbuflen, se );
|
|
}
|
|
|
|
|
|
static void WINAPI async_worker( TP_CALLBACK_INSTANCE *instance, void *context )
|
|
{
|
|
struct async_query_header *query = context;
|
|
LPARAM lparam = query->func( query );
|
|
PostMessageW( query->hWnd, query->uMsg, (WPARAM)query->handle, lparam );
|
|
HeapFree( GetProcessHeap(), 0, query );
|
|
}
|
|
|
|
|
|
/****************************************************************************
|
|
* 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 HANDLE run_query( HWND hWnd, UINT uMsg, LPARAM (*func)(struct async_query_header *),
|
|
struct async_query_header *query, void *sbuf, INT sbuflen )
|
|
{
|
|
static LONG next_handle = 0xdead;
|
|
ULONG handle;
|
|
do
|
|
handle = LOWORD( InterlockedIncrement( &next_handle ));
|
|
while (!handle); /* avoid handle 0 */
|
|
|
|
query->func = func;
|
|
query->hWnd = hWnd;
|
|
query->uMsg = uMsg;
|
|
query->handle = UlongToHandle( handle );
|
|
query->sbuf = sbuf;
|
|
query->sbuflen = sbuflen;
|
|
|
|
if (!TrySubmitThreadpoolCallback( async_worker, query, NULL ))
|
|
{
|
|
SetLastError( WSAEWOULDBLOCK );
|
|
HeapFree( GetProcessHeap(), 0, query );
|
|
return 0;
|
|
}
|
|
return UlongToHandle( handle );
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* WSAAsyncGetHostByAddr (WS2_32.102)
|
|
*/
|
|
HANDLE WINAPI WSAAsyncGetHostByAddr(HWND hWnd, UINT uMsg, LPCSTR addr,
|
|
INT len, INT type, LPSTR sbuf, INT buflen)
|
|
{
|
|
struct async_query_gethostbyaddr *aq;
|
|
|
|
TRACE("hwnd %p, msg %04x, addr %p[%i]\n", hWnd, uMsg, addr, len );
|
|
|
|
if (!(aq = HeapAlloc( GetProcessHeap(), 0, sizeof(*aq) + len )))
|
|
{
|
|
SetLastError( WSAEWOULDBLOCK );
|
|
return 0;
|
|
}
|
|
aq->host_addr = (char *)(aq + 1);
|
|
aq->host_len = len;
|
|
aq->host_type = type;
|
|
memcpy( aq->host_addr, addr, len );
|
|
return run_query( hWnd, uMsg, async_gethostbyaddr, &aq->query, sbuf, buflen );
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WSAAsyncGetHostByName (WS2_32.103)
|
|
*/
|
|
HANDLE WINAPI WSAAsyncGetHostByName(HWND hWnd, UINT uMsg, LPCSTR name,
|
|
LPSTR sbuf, INT buflen)
|
|
{
|
|
struct async_query_gethostbyname *aq;
|
|
unsigned int len = strlen(name) + 1;
|
|
|
|
TRACE("hwnd %p, msg %04x, host %s, buffer %i\n", hWnd, uMsg, debugstr_a(name), buflen );
|
|
|
|
if (!(aq = HeapAlloc( GetProcessHeap(), 0, sizeof(*aq) + len )))
|
|
{
|
|
SetLastError( WSAEWOULDBLOCK );
|
|
return 0;
|
|
}
|
|
aq->host_name = (char *)(aq + 1);
|
|
strcpy( aq->host_name, name );
|
|
return run_query( hWnd, uMsg, async_gethostbyname, &aq->query, sbuf, buflen );
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WSAAsyncGetProtoByName (WS2_32.105)
|
|
*/
|
|
HANDLE WINAPI WSAAsyncGetProtoByName(HWND hWnd, UINT uMsg, LPCSTR name,
|
|
LPSTR sbuf, INT buflen)
|
|
{
|
|
struct async_query_getprotobyname *aq;
|
|
unsigned int len = strlen(name) + 1;
|
|
|
|
TRACE("hwnd %p, msg %04x, proto %s, buffer %i\n", hWnd, uMsg, debugstr_a(name), buflen );
|
|
|
|
if (!(aq = HeapAlloc( GetProcessHeap(), 0, sizeof(*aq) + len )))
|
|
{
|
|
SetLastError( WSAEWOULDBLOCK );
|
|
return 0;
|
|
}
|
|
aq->proto_name = (char *)(aq + 1);
|
|
strcpy( aq->proto_name, name );
|
|
return run_query( hWnd, uMsg, async_getprotobyname, &aq->query, sbuf, buflen );
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* WSAAsyncGetProtoByNumber (WS2_32.104)
|
|
*/
|
|
HANDLE WINAPI WSAAsyncGetProtoByNumber(HWND hWnd, UINT uMsg, INT number,
|
|
LPSTR sbuf, INT buflen)
|
|
{
|
|
struct async_query_getprotobynumber *aq;
|
|
|
|
TRACE("hwnd %p, msg %04x, num %i\n", hWnd, uMsg, number );
|
|
|
|
if (!(aq = HeapAlloc( GetProcessHeap(), 0, sizeof(*aq) )))
|
|
{
|
|
SetLastError( WSAEWOULDBLOCK );
|
|
return 0;
|
|
}
|
|
aq->proto_number = number;
|
|
return run_query( hWnd, uMsg, async_getprotobynumber, &aq->query, sbuf, buflen );
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WSAAsyncGetServByName (WS2_32.107)
|
|
*/
|
|
HANDLE WINAPI WSAAsyncGetServByName(HWND hWnd, UINT uMsg, LPCSTR name,
|
|
LPCSTR proto, LPSTR sbuf, INT buflen)
|
|
{
|
|
struct async_query_getservbyname *aq;
|
|
unsigned int len1 = strlen(name) + 1;
|
|
unsigned int len2 = proto ? strlen(proto) + 1 : 0;
|
|
|
|
TRACE("hwnd %p, msg %04x, name %s, proto %s\n", hWnd, uMsg, debugstr_a(name), debugstr_a(proto));
|
|
|
|
if (!(aq = HeapAlloc( GetProcessHeap(), 0, sizeof(*aq) + len1 + len2 )))
|
|
{
|
|
SetLastError( WSAEWOULDBLOCK );
|
|
return 0;
|
|
}
|
|
|
|
aq->serv_name = (char *)(aq + 1);
|
|
strcpy( aq->serv_name, name );
|
|
|
|
if (proto)
|
|
{
|
|
aq->serv_proto = aq->serv_name + len1;
|
|
strcpy( aq->serv_proto, proto );
|
|
}
|
|
else
|
|
aq->serv_proto = NULL;
|
|
|
|
return run_query( hWnd, uMsg, async_getservbyname, &aq->query, sbuf, buflen );
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WSAAsyncGetServByPort (WS2_32.106)
|
|
*/
|
|
HANDLE WINAPI WSAAsyncGetServByPort(HWND hWnd, UINT uMsg, INT port,
|
|
LPCSTR proto, LPSTR sbuf, INT buflen)
|
|
{
|
|
struct async_query_getservbyport *aq;
|
|
unsigned int len = proto ? strlen(proto) + 1 : 0;
|
|
|
|
TRACE("hwnd %p, msg %04x, port %i, proto %s\n", hWnd, uMsg, port, debugstr_a(proto));
|
|
|
|
if (!(aq = HeapAlloc( GetProcessHeap(), 0, sizeof(*aq) + len )))
|
|
{
|
|
SetLastError( WSAEWOULDBLOCK );
|
|
return 0;
|
|
}
|
|
|
|
if (proto)
|
|
{
|
|
aq->serv_proto = (char *)(aq + 1);
|
|
strcpy( aq->serv_proto, proto );
|
|
}
|
|
else
|
|
aq->serv_proto = NULL;
|
|
|
|
aq->serv_port = port;
|
|
|
|
return run_query( hWnd, uMsg, async_getservbyport, &aq->query, sbuf, buflen );
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WSACancelAsyncRequest (WS2_32.108)
|
|
*/
|
|
INT WINAPI WSACancelAsyncRequest(HANDLE hAsyncTaskHandle)
|
|
{
|
|
FIXME("(%p),stub\n", hAsyncTaskHandle);
|
|
return 0;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WSApSetPostRoutine (WS2_32.24)
|
|
*/
|
|
INT WINAPI WSApSetPostRoutine(LPWPUPOSTMESSAGE lpPostRoutine)
|
|
{
|
|
FIXME("(%p), stub !\n", lpPostRoutine);
|
|
return 0;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WPUCompleteOverlappedRequest (WS2_32.25)
|
|
*/
|
|
WSAEVENT WINAPI WPUCompleteOverlappedRequest(SOCKET s, LPWSAOVERLAPPED overlapped,
|
|
DWORD error, DWORD transferred, LPINT errcode)
|
|
{
|
|
FIXME("(0x%08lx,%p,0x%08x,0x%08x,%p), stub !\n", s, overlapped, error, transferred, errcode);
|
|
|
|
if (errcode)
|
|
*errcode = WSAEINVAL;
|
|
|
|
return NULL;
|
|
}
|