2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS system libraries
4 * FILE: lib/dnsapi/dnsapi/query.c
5 * PURPOSE: DNSAPI functions built on the ADNS library.
6 * PROGRAMER: Art Yerkes
14 #include <internal/windns.h>
16 #define NTOS_MODE_USER
19 /* DnsQuery ****************************
20 * Begin a DNS query, and allow the result to be placed in the application
21 * supplied result pointer. The result can be manipulated with the record
24 * Name -- The DNS object to be queried.
25 * Type -- The type of records to be returned. These are
27 * Options -- Query options. DNS_QUERY_STANDARD is the base
28 * state, and every other option takes precedence.
29 * multiple options can be combined. Listed in
31 * Servers -- List of alternate servers (optional)
32 * QueryResultSet -- Pointer to the result pointer that will be filled
33 * when the response is available.
34 * Reserved -- Response as it appears on the wire. Optional.
37 char *xstrsave(const char *str
) {
40 p
= RtlAllocateHeap( RtlGetProcessHeap(), 0, strlen(str
)+1 );
45 DNS_STATUS WINAPI DnsQuery_A
50 PDNS_RECORD
*QueryResultSet
,
61 adns_error
= adns_init( &astate
,
66 if( adns_error
!= adns_s_ok
) {
67 return DnsIntTranslateAdnsToDNS_STATUS( adns_error
);
70 adns_error
= adns_synchronous( astate
,
76 if( adns_error
!= adns_s_ok
) {
77 adns_finish( astate
);
78 return DnsIntTranslateAdnsToDNS_STATUS( adns_error
);
81 if( answer
&& answer
->rrs
.addr
) {
83 (PDNS_RECORD
)RtlAllocateHeap( RtlGetProcessHeap(), 0,
84 sizeof( DNS_RECORD
) );
85 (*QueryResultSet
)->pNext
= NULL
;
86 (*QueryResultSet
)->wType
= Type
;
87 (*QueryResultSet
)->pName
= xstrsave( Name
);
88 (*QueryResultSet
)->wDataLength
= sizeof(DNS_A_DATA
);
89 (*QueryResultSet
)->Data
.A
.IpAddress
=
90 answer
->rrs
.addr
->addr
.inet
.sin_addr
.s_addr
;
91 adns_finish( astate
);
94 return ERROR_FILE_NOT_FOUND
;
96 return ERROR_OUTOFMEMORY
; /* XXX arty: find a better error code. */
100 static PCHAR
DnsWToC( const WCHAR
*WideString
) {
101 int chars
= wcstombs( NULL
, WideString
, 0 );
102 PCHAR out
= RtlAllocateHeap( RtlGetProcessHeap(), 0, chars
+ 1 );
103 wcstombs( out
, WideString
, chars
+ 1 );
107 static PWCHAR
DnsCToW( const CHAR
*NarrowString
) {
108 int chars
= mbstowcs( NULL
, NarrowString
, 0 );
109 PWCHAR out
= RtlAllocateHeap( RtlGetProcessHeap(), 0,
110 (chars
+ 1) * sizeof(WCHAR
) );
111 mbstowcs( out
, NarrowString
, chars
+ 1 );
115 DNS_STATUS WINAPI DnsQuery_W
120 PDNS_RECORD
*QueryResultSet
,
125 PDNS_RECORD QueryResultWide
;
126 PDNS_RECORD ConvertedRecord
= 0, LastRecord
= 0;
128 Buffer
= DnsWToC( Name
);
130 Status
= DnsQuery_A( Buffer
, Type
, Options
, Servers
, &QueryResultWide
,
133 while( Status
== ERROR_SUCCESS
&& QueryResultWide
) {
134 switch( QueryResultWide
->wType
) {
141 ConvertedRecord
= RtlAllocateHeap( RtlGetProcessHeap(), 0,
142 sizeof(DNS_RECORD
) );
143 ConvertedRecord
->pName
= (PCHAR
)DnsCToW( QueryResultWide
->pName
);
144 ConvertedRecord
->wType
= QueryResultWide
->wType
;
145 ConvertedRecord
->wDataLength
= QueryResultWide
->wDataLength
;
146 memcpy( ConvertedRecord
, QueryResultWide
,
147 QueryResultWide
->wDataLength
);
158 ConvertedRecord
= RtlAllocateHeap( RtlGetProcessHeap(), 0,
159 sizeof(DNS_RECORD
) );
160 ConvertedRecord
->pName
= (PCHAR
)DnsCToW( QueryResultWide
->pName
);
161 ConvertedRecord
->wType
= QueryResultWide
->wType
;
162 ConvertedRecord
->wDataLength
= sizeof(DNS_PTR_DATA
);
163 ConvertedRecord
->Data
.PTR
.pNameHost
=
164 (PCHAR
)DnsCToW( QueryResultWide
->Data
.PTR
.pNameHost
);
170 ConvertedRecord
= RtlAllocateHeap( RtlGetProcessHeap(), 0,
171 sizeof(DNS_RECORD
) );
172 ConvertedRecord
->pName
= (PCHAR
)DnsCToW( QueryResultWide
->pName
);
173 ConvertedRecord
->wType
= QueryResultWide
->wType
;
174 ConvertedRecord
->wDataLength
= sizeof(DNS_MINFO_DATA
);
175 ConvertedRecord
->Data
.MINFO
.pNameMailbox
=
176 (PCHAR
)DnsCToW( QueryResultWide
->Data
.MINFO
.pNameMailbox
);
177 ConvertedRecord
->Data
.MINFO
.pNameErrorsMailbox
=
178 (PCHAR
)DnsCToW( QueryResultWide
->Data
.MINFO
.pNameErrorsMailbox
);
186 ConvertedRecord
= RtlAllocateHeap( RtlGetProcessHeap(), 0,
187 sizeof(DNS_RECORD
) );
188 ConvertedRecord
->pName
= (PCHAR
)DnsCToW( QueryResultWide
->pName
);
189 ConvertedRecord
->wType
= QueryResultWide
->wType
;
190 ConvertedRecord
->wDataLength
= sizeof(DNS_MX_DATA
);
191 ConvertedRecord
->Data
.MX
.pNameExchange
=
192 (PCHAR
)DnsCToW( QueryResultWide
->Data
.MX
.pNameExchange
);
193 ConvertedRecord
->Data
.MX
.wPreference
=
194 QueryResultWide
->Data
.MX
.wPreference
;
202 ConvertedRecord
= RtlAllocateHeap( RtlGetProcessHeap(), 0,
203 sizeof(DNS_TXT_DATA
) +
205 Data
.TXT
.dwStringCount
);
206 ConvertedRecord
->pName
= (PCHAR
)DnsCToW( QueryResultWide
->pName
);
207 ConvertedRecord
->wType
= QueryResultWide
->wType
;
208 ConvertedRecord
->wDataLength
=
209 sizeof(DNS_TXT_DATA
) +
210 (sizeof(PWCHAR
) * QueryResultWide
->Data
.TXT
.dwStringCount
);
211 ConvertedRecord
->Data
.TXT
.dwStringCount
=
212 QueryResultWide
->Data
.TXT
.dwStringCount
;
213 for( i
= 0; i
< ConvertedRecord
->Data
.TXT
.dwStringCount
; i
++ ) {
214 ConvertedRecord
->Data
.TXT
.pStringArray
[i
] =
215 (PCHAR
)DnsCToW( QueryResultWide
->Data
.TXT
.pStringArray
[i
] );
220 ConvertedRecord
= RtlAllocateHeap( RtlGetProcessHeap(), 0,
221 sizeof(DNS_NULL_DATA
) +
223 Data
.Null
.dwByteCount
);
224 ConvertedRecord
->pName
= (PCHAR
)DnsCToW( QueryResultWide
->pName
);
225 ConvertedRecord
->wType
= QueryResultWide
->wType
;
226 ConvertedRecord
->wDataLength
=
227 sizeof(DNS_NULL_DATA
) + QueryResultWide
->Data
.Null
.dwByteCount
;
228 ConvertedRecord
->Data
.Null
.dwByteCount
=
229 QueryResultWide
->Data
.Null
.dwByteCount
;
230 memcpy( &ConvertedRecord
->Data
.Null
.Data
,
231 &QueryResultWide
->Data
.Null
.Data
,
232 QueryResultWide
->Data
.Null
.dwByteCount
);
237 ConvertedRecord
= RtlAllocateHeap( RtlGetProcessHeap(), 0,
238 sizeof(DNS_RECORDA
) );
239 ConvertedRecord
->pName
= DnsCToW( QueryResultWide
->pName
);
240 ConvertedRecord
->wType
= QueryResultWide
->wType
;
241 ConvertedRecord
->wDataLength
= sizeof(DNS_SIG_DATAA
);
242 memcpy( &ConvertedRecord
->Data
.SIG
,
243 &QueryResultWide
->Data
.SIG
,
244 sizeof(QueryResultWide
->Data
.SIG
) );
245 ConvertedRecord
->Data
.SIG
.pNameSigner
=
246 DnsCToW( QueryResultWide
->Data
.SIG
.pNameSigner
);
250 ConvertedRecord
= RtlAllocateHeap( RtlGetProcessHeap(), 0,
251 sizeof(DNS_RECORDA
) );
252 ConvertedRecord
->pName
= DnsCToW( QueryResultWide
->pName
);
253 ConvertedRecord
->wType
= QueryResultWide
->wType
;
254 ConvertedRecord
->wDataLength
= sizeof(DNS_NXT_DATAA
);
255 memcpy( &ConvertedRecord
->Data
.NXT
,
256 &QueryResultWide
->Data
.NXT
,
257 sizeof(QueryResultWide
->Data
.NXT
) );
258 ConvertedRecord
->Data
.NXT
.pNameNext
=
259 DnsCToW( QueryResultWide
->Data
.NXT
.pNameNext
);
263 ConvertedRecord
= RtlAllocateHeap( RtlGetProcessHeap(), 0,
264 sizeof(DNS_RECORDA
) );
265 ConvertedRecord
->pName
= DnsCToW( QueryResultWide
->pName
);
266 ConvertedRecord
->wType
= QueryResultWide
->wType
;
267 ConvertedRecord
->wDataLength
= sizeof(DNS_SRV_DATAA
);
268 memcpy( &ConvertedRecord
->Data
.SRV
,
269 &QueryResultWide
->Data
.SRV
,
270 sizeof(QueryResultWide
->Data
.SRV
) );
271 ConvertedRecord
->Data
.SRV
.pNameTarget
=
272 DnsCToW( QueryResultWide
->Data
.SRV
.pNameTarget
);
278 LastRecord
->pNext
= ConvertedRecord
;
279 LastRecord
= LastRecord
->pNext
;
281 LastRecord
= *QueryResultSet
= ConvertedRecord
;
285 LastRecord
->pNext
= 0;
288 RtlFreeHeap( RtlGetProcessHeap(), 0, Buffer
);
293 DNS_STATUS WINAPI DnsQuery_UTF8
298 PDNS_RECORD
*QueryResultSet
,
300 return DnsQuery_UTF8( Name
, Type
, Options
, Servers
, QueryResultSet
,
304 void DnsIntFreeRecordList( PDNS_RECORD ToDelete
) {
306 PDNS_RECORD next
= 0;
309 if( ToDelete
->pName
)
310 RtlFreeHeap( RtlGetProcessHeap(), 0, ToDelete
->pName
);
311 switch( ToDelete
->wType
) {
320 RtlFreeHeap( RtlGetProcessHeap(), 0, ToDelete
->Data
.PTR
.pNameHost
);
325 RtlFreeHeap( RtlGetProcessHeap(), 0,
326 ToDelete
->Data
.MINFO
.pNameMailbox
);
327 RtlFreeHeap( RtlGetProcessHeap(), 0,
328 ToDelete
->Data
.MINFO
.pNameErrorsMailbox
);
335 RtlFreeHeap( RtlGetProcessHeap(), 0, ToDelete
->Data
.MX
.pNameExchange
);
343 for( i
= 0; i
< ToDelete
->Data
.TXT
.dwStringCount
; i
++ ) {
344 RtlFreeHeap( RtlGetProcessHeap(), 0,
345 ToDelete
->Data
.TXT
.pStringArray
[i
] );
347 RtlFreeHeap( RtlGetProcessHeap(), 0, ToDelete
->Data
.TXT
.pStringArray
);
352 RtlFreeHeap( RtlGetProcessHeap(), 0, ToDelete
->Data
.SIG
.pNameSigner
);
356 RtlFreeHeap( RtlGetProcessHeap(), 0, ToDelete
->Data
.NXT
.pNameNext
);
360 RtlFreeHeap( RtlGetProcessHeap(), 0, ToDelete
->Data
.SRV
.pNameTarget
);
365 next
= ToDelete
->pNext
;
366 RtlFreeHeap( RtlGetProcessHeap(), 0, ToDelete
);