urlmon/tests: Keep track of download state in tests.
[wine] / dlls / dnsapi / name.c
1 /*
2  * DNS support
3  *
4  * Copyright (C) 2006 Matthew Kehrer
5  * Copyright (C) 2006 Hans Leidekker
6  * 
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21
22 #include "config.h"
23 #include "wine/debug.h"
24 #include "wine/unicode.h"
25
26 #include <stdarg.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 # undef NOERROR
35 #endif
36 #ifdef HAVE_RESOLV_H
37 # include <resolv.h>
38 #endif
39
40 #include "windef.h"
41 #include "winbase.h"
42 #include "winerror.h"
43 #include "winnls.h"
44 #include "windns.h"
45
46 #include "dnsapi.h"
47
48 WINE_DEFAULT_DEBUG_CHANNEL(dnsapi);
49
50 /******************************************************************************
51  * DnsNameCompare_A               [DNSAPI.@]
52  *
53  */
54 BOOL WINAPI DnsNameCompare_A( LPSTR name1, LPSTR name2 )
55 {
56     BOOL ret;
57     LPWSTR name1W, name2W;
58
59     TRACE( "(%s,%s)\n", debugstr_a(name1), debugstr_a(name2) );
60
61     name1W = dns_strdup_aw( name1 );
62     name2W = dns_strdup_aw( name2 );
63
64     ret = DnsNameCompare_W( name1W, name2W );
65
66     dns_free( name1W );
67     dns_free( name2W );
68
69     return ret;
70 }
71
72 /******************************************************************************
73  * DnsNameCompare_W               [DNSAPI.@]
74  *
75  */
76 BOOL WINAPI DnsNameCompare_W( LPWSTR name1, LPWSTR name2 )
77 {
78     WCHAR *p, *q;
79
80     TRACE( "(%s,%s)\n", debugstr_w(name1), debugstr_w(name2) );
81
82     if (!name1 && !name2) return TRUE;
83     if (!name1 || !name2) return FALSE;
84  
85     p = name1 + lstrlenW( name1 ) - 1;
86     q = name2 + lstrlenW( name2 ) - 1;
87
88     while (*p == '.' && p >= name1) p--;
89     while (*q == '.' && q >= name2) q--;
90
91     if (p - name1 != q - name2) return FALSE;
92
93     while (name1 <= p)
94     {
95         if (toupperW( *name1 ) != toupperW( *name2 ))
96             return FALSE;
97
98         name1++;
99         name2++;
100     }
101     return TRUE;
102 }
103
104 /******************************************************************************
105  * DnsValidateName_A              [DNSAPI.@]
106  *
107  */
108 DNS_STATUS WINAPI DnsValidateName_A( LPCSTR name, DNS_NAME_FORMAT format )
109 {
110     LPWSTR nameW;
111     DNS_STATUS ret;
112
113     TRACE( "(%s, %d)\n", debugstr_a(name), format );
114
115     nameW = dns_strdup_aw( name );
116     ret = DnsValidateName_W( nameW, format );
117
118     dns_free( nameW );
119     return ret;
120 }
121
122 /******************************************************************************
123  * DnsValidateName_UTF8           [DNSAPI.@]
124  *
125  */
126 DNS_STATUS WINAPI DnsValidateName_UTF8( LPCSTR name, DNS_NAME_FORMAT format )
127 {
128     LPWSTR nameW;
129     DNS_STATUS ret;
130
131     TRACE( "(%s, %d)\n", debugstr_a(name), format );
132
133     nameW = dns_strdup_uw( name );
134     ret = DnsValidateName_W( nameW, format );
135
136     dns_free( nameW );
137     return ret;
138 }
139
140 #define HAS_EXTENDED        0x0001
141 #define HAS_NUMERIC         0x0002
142 #define HAS_NON_NUMERIC     0x0004
143 #define HAS_DOT             0x0008
144 #define HAS_DOT_DOT         0x0010
145 #define HAS_SPACE           0x0020
146 #define HAS_INVALID         0x0040
147 #define HAS_ASTERISK        0x0080
148 #define HAS_UNDERSCORE      0x0100
149 #define HAS_LONG_LABEL      0x0200
150
151 /******************************************************************************
152  * DnsValidateName_W              [DNSAPI.@]
153  *
154  */
155 DNS_STATUS WINAPI DnsValidateName_W( LPCWSTR name, DNS_NAME_FORMAT format )
156 {
157     const WCHAR *p;
158     unsigned int i, j, state = 0;
159     static const WCHAR invalid[] = {
160         '{','|','}','~','[','\\',']','^','\'',':',';','<','=','>',
161         '?','@','!','\"','#','$','%','^','`','(',')','+','/',',',0 };
162
163     TRACE( "(%s, %d)\n", debugstr_w(name), format );
164
165     if (!name) return ERROR_INVALID_NAME;
166
167     for (p = name, i = 0, j = 0; *p; p++, i++, j++)
168     {
169         if (*p == '.')
170         {
171             j = 0;
172             state |= HAS_DOT;
173             if (p[1] == '.') state |= HAS_DOT_DOT;
174         }
175         else if (*p < '0' || *p > '9') state |= HAS_NON_NUMERIC;
176         else state |= HAS_NUMERIC;
177
178         if (j > 62) state |= HAS_LONG_LABEL;
179
180         if (strchrW( invalid, *p )) state |= HAS_INVALID;
181         else if ((unsigned)*p > 127) state |= HAS_EXTENDED;
182         else if (*p == ' ') state |= HAS_SPACE;
183         else if (*p == '_') state |= HAS_UNDERSCORE;
184         else if (*p == '*') state |= HAS_ASTERISK;
185     }
186
187     if (i == 0 || i > 255 ||
188         (state & HAS_LONG_LABEL) ||
189         (state & HAS_DOT_DOT) ||
190         (name[0] == '.' && name[1])) return ERROR_INVALID_NAME;
191
192     switch (format)
193     {
194     case DnsNameDomain:
195     {
196         if (!(state & HAS_NON_NUMERIC) && (state & HAS_NUMERIC))
197             return DNS_ERROR_NUMERIC_NAME;
198         if ((state & HAS_EXTENDED) || (state & HAS_UNDERSCORE))
199             return DNS_ERROR_NON_RFC_NAME;
200         if ((state & HAS_SPACE) ||
201             (state & HAS_INVALID) ||
202             (state & HAS_ASTERISK)) return DNS_ERROR_INVALID_NAME_CHAR;
203         break;
204     }
205     case DnsNameDomainLabel:
206     {
207         if (state & HAS_DOT) return ERROR_INVALID_NAME;
208         if ((state & HAS_EXTENDED) || (state & HAS_UNDERSCORE))
209             return DNS_ERROR_NON_RFC_NAME;
210         if ((state & HAS_SPACE) ||
211             (state & HAS_INVALID) ||
212             (state & HAS_ASTERISK)) return DNS_ERROR_INVALID_NAME_CHAR;
213         break;
214     }
215     case DnsNameHostnameFull:
216     {
217         if (!(state & HAS_NON_NUMERIC) && (state & HAS_NUMERIC))
218             return DNS_ERROR_NUMERIC_NAME;
219         if ((state & HAS_EXTENDED) || (state & HAS_UNDERSCORE))
220             return DNS_ERROR_NON_RFC_NAME;
221         if ((state & HAS_SPACE) ||
222             (state & HAS_INVALID) ||
223             (state & HAS_ASTERISK)) return DNS_ERROR_INVALID_NAME_CHAR;
224         break;
225     }
226     case DnsNameHostnameLabel:
227     {
228         if (state & HAS_DOT) return ERROR_INVALID_NAME;
229         if (!(state & HAS_NON_NUMERIC) && (state & HAS_NUMERIC))
230             return DNS_ERROR_NUMERIC_NAME;
231         if ((state & HAS_EXTENDED) || (state & HAS_UNDERSCORE))
232             return DNS_ERROR_NON_RFC_NAME;
233         if ((state & HAS_SPACE) ||
234             (state & HAS_INVALID) ||
235             (state & HAS_ASTERISK)) return DNS_ERROR_INVALID_NAME_CHAR;
236         break;
237     }
238     case DnsNameWildcard:
239     {
240         if (!(state & HAS_NON_NUMERIC) && (state & HAS_NUMERIC))
241             return ERROR_INVALID_NAME;
242         if (name[0] != '*') return ERROR_INVALID_NAME;
243         if (name[1] && name[1] != '.')
244             return DNS_ERROR_INVALID_NAME_CHAR;
245         if ((state & HAS_EXTENDED) ||
246             (state & HAS_SPACE) ||
247             (state & HAS_INVALID)) return ERROR_INVALID_NAME;
248         break;
249     }
250     case DnsNameSrvRecord:
251     {
252         if (!(state & HAS_NON_NUMERIC) && (state & HAS_NUMERIC))
253             return ERROR_INVALID_NAME;
254         if (name[0] != '_') return ERROR_INVALID_NAME;
255         if ((state & HAS_UNDERSCORE) && !name[1])
256             return DNS_ERROR_NON_RFC_NAME;
257         if ((state & HAS_EXTENDED) ||
258             (state & HAS_SPACE) ||
259             (state & HAS_INVALID)) return ERROR_INVALID_NAME;
260         break;
261     }
262     default:
263         WARN( "unknown format: %d\n", format );
264         break;
265     }
266     return ERROR_SUCCESS;
267 }