dlls/dnsapi/query.c

   1 /*
   2  * DNS support
   3  *
   4  * Copyright (C) 2006 Hans Leidekker
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23 #include "wine/debug.h"
  24
  25 #include <stdarg.h>
  26 #include <string.h>
  27 #include <sys/types.h>
  28
  29 #ifdef HAVE_NETINET_IN_H
  30 # include <netinet/in.h>
  31 #endif
  32 #ifdef HAVE_ARPA_NAMESER_H
  33 # include <arpa/nameser.h>
  34 #endif
  35 #ifdef HAVE_RESOLV_H
  36 # include <resolv.h>
  37 #endif
  38 #ifdef HAVE_NETDB_H
  39 # include <netdb.h>
  40 #endif
  41
  42 #include "windef.h"
  43 #include "winbase.h"
  44 #include "winerror.h"
  45 #include "winnls.h"
  46 #include "windns.h"
  47
  48 #include "dnsapi.h"
  49
  50 WINE_DEFAULT_DEBUG_CHANNEL(dnsapi);
  51
  52 #ifdef HAVE_RESOLV
  53
  54 static CRITICAL_SECTION resolver_cs;
  55 static CRITICAL_SECTION_DEBUG resolver_cs_debug =
  56 {
  57     0, 0, &resolver_cs,
  58     { &resolver_cs_debug.ProcessLocksList,
  59       &resolver_cs_debug.ProcessLocksList },
  60       0, 0, { (DWORD_PTR)(__FILE__ ": resolver_cs") }
  61 };
  62 static CRITICAL_SECTION resolver_cs = { &resolver_cs_debug, -1, 0, 0, 0, 0 };
  63
  64 #define LOCK_RESOLVER()     do { EnterCriticalSection( &resolver_cs ); } while (0)
  65 #define UNLOCK_RESOLVER()   do { LeaveCriticalSection( &resolver_cs ); } while (0)
  66
  67
  68 /*  the resolver lock must be held and res_init() must have been
  69  *  called before calling this function.
  70  */
  71 static DNS_STATUS dns_get_serverlist( PIP4_ARRAY addrs, PDWORD len )
  72 {
  73     unsigned int i, size;
  74
  75     size = sizeof(IP4_ARRAY) + sizeof(IP4_ADDRESS) * (_res.nscount - 1);
  76     if (!addrs || *len < size)
  77     {
  78         *len = size;
  79         return ERROR_INSUFFICIENT_BUFFER;
  80     }
  81
  82     addrs->AddrCount = _res.nscount;
  83
  84     for (i = 0; i < _res.nscount; i++)
  85         addrs->AddrArray[i] = _res.nsaddr_list[i].sin_addr.s_addr;
  86
  87     return ERROR_SUCCESS;
  88 }
  89
  90 #endif /* HAVE_RESOLV */
  91
  92 static DNS_STATUS dns_get_hostname_a( COMPUTER_NAME_FORMAT format,
  93                                       LPSTR buffer, PDWORD len )
  94 {
  95     char name[256];
  96     DWORD size = sizeof(name);
  97
  98     if (!GetComputerNameExA( format, name, &size ))
  99         return DNS_ERROR_NAME_DOES_NOT_EXIST;
 100
 101     if (!buffer || (size = lstrlenA( name ) + 1) > *len)
 102     {
 103         *len = size;
 104         return ERROR_INSUFFICIENT_BUFFER;
 105     }
 106
 107     lstrcpyA( buffer, name );
 108     return ERROR_SUCCESS;
 109 }
 110
 111 static DNS_STATUS dns_get_hostname_w( COMPUTER_NAME_FORMAT format,
 112                                       LPWSTR buffer, PDWORD len )
 113 {
 114     WCHAR name[256];
 115     DWORD size = sizeof(name);
 116
 117     if (!GetComputerNameExW( format, name, &size ))
 118         return DNS_ERROR_NAME_DOES_NOT_EXIST;
 119
 120     if (!buffer || (size = lstrlenW( name ) + 1) > *len)
 121     {
 122         *len = size;
 123         return ERROR_INSUFFICIENT_BUFFER;
 124     }
 125
 126     lstrcpyW( buffer, name );
 127     return ERROR_SUCCESS;
 128 }
 129
 130 /******************************************************************************
 131  * DnsQueryConfig          [DNSAPI.@]
 132  *
 133  */
 134 DNS_STATUS WINAPI DnsQueryConfig( DNS_CONFIG_TYPE config, DWORD flag, PWSTR adapter,
 135                                   PVOID reserved, PVOID buffer, PDWORD len )
 136 {
 137     DNS_STATUS ret = ERROR_INVALID_PARAMETER;
 138
 139     TRACE( "(%d,0x%08lx,%s,%p,%p,%p)\n", config, flag, debugstr_w(adapter),
 140            reserved, buffer, len );
 141
 142     if (!len) return ERROR_INVALID_PARAMETER;
 143
 144     switch (config)
 145     {
 146     case DnsConfigDnsServerList:
 147     {
 148 #ifdef HAVE_RESOLV
 149         LOCK_RESOLVER();
 150
 151         res_init();
 152         ret = dns_get_serverlist( (IP4_ARRAY *)buffer, len );
 153
 154         UNLOCK_RESOLVER();
 155         break;
 156 #else
 157         WARN( "compiled without resolver support\n" );
 158         break;
 159 #endif
 160     }
 161     case DnsConfigHostName_A:
 162     case DnsConfigHostName_UTF8:
 163         return dns_get_hostname_a( ComputerNameDnsHostname, buffer, len );
 164
 165     case DnsConfigFullHostName_A:
 166     case DnsConfigFullHostName_UTF8:
 167         return dns_get_hostname_a( ComputerNameDnsFullyQualified, buffer, len );
 168
 169     case DnsConfigPrimaryDomainName_A:
 170     case DnsConfigPrimaryDomainName_UTF8:
 171         return dns_get_hostname_a( ComputerNameDnsDomain, buffer, len );
 172
 173     case DnsConfigHostName_W:
 174         return dns_get_hostname_w( ComputerNameDnsHostname, buffer, len );
 175
 176     case DnsConfigFullHostName_W:
 177         return dns_get_hostname_w( ComputerNameDnsFullyQualified, buffer, len );
 178
 179     case DnsConfigPrimaryDomainName_W:
 180         return dns_get_hostname_w( ComputerNameDnsDomain, buffer, len );
 181
 182     case DnsConfigAdapterDomainName_A:
 183     case DnsConfigAdapterDomainName_W:
 184     case DnsConfigAdapterDomainName_UTF8:
 185     case DnsConfigSearchList:
 186     case DnsConfigAdapterInfo:
 187     case DnsConfigPrimaryHostNameRegistrationEnabled:
 188     case DnsConfigAdapterHostNameRegistrationEnabled:
 189     case DnsConfigAddressRegistrationMaxCount:
 190         FIXME( "unimplemented config type %d\n", config );
 191         break;
 192
 193     default:
 194         WARN( "unknown config type: %d\n", config );
 195         break;
 196     }
 197     return ret;
 198 }