wined3d: Fix NULL dereference in generator function.
[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 #endif
35 #ifdef HAVE_RESOLV_H
36 # include <resolv.h>
37 #endif
38
39 #include "windef.h"
40 #include "winbase.h"
41 #include "winerror.h"
42 #include "winnls.h"
43 #include "windns.h"
44
45 #include "dnsapi.h"
46
47 WINE_DEFAULT_DEBUG_CHANNEL(dnsapi);
48
49 /******************************************************************************
50  * DnsNameCompare_A               [DNSAPI.@]
51  *
52  */
53 BOOL WINAPI DnsNameCompare_A( LPSTR name1, LPSTR name2 )
54 {
55     BOOL ret;
56     LPWSTR name1W, name2W;
57
58     TRACE( "(%s,%s)\n", debugstr_a(name1), debugstr_a(name2) );
59
60     name1W = dns_strdup_aw( name1 );
61     name2W = dns_strdup_aw( name2 );
62
63     ret = DnsNameCompare_W( name1W, name2W );
64
65     dns_free( name1W );
66     dns_free( name2W );
67
68     return ret;
69 }
70
71 /******************************************************************************
72  * DnsNameCompare_W               [DNSAPI.@]
73  *
74  */
75 BOOL WINAPI DnsNameCompare_W( LPWSTR name1, LPWSTR name2 )
76 {
77     WCHAR *p, *q;
78
79     TRACE( "(%s,%s)\n", debugstr_w(name1), debugstr_w(name2) );
80
81     if (!name1 && !name2) return TRUE;
82     if (!name1 || !name2) return FALSE;
83  
84     p = name1 + lstrlenW( name1 ) - 1;
85     q = name2 + lstrlenW( name2 ) - 1;
86
87     while (*p == '.' && p >= name1) p--;
88     while (*q == '.' && q >= name2) q--;
89
90     if (p - name1 != q - name2) return FALSE;
91
92     while (name1 <= p)
93     {
94         if (toupperW( *name1 ) != toupperW( *name2 ))
95             return FALSE;
96
97         name1++;
98         name2++;
99     }
100     return TRUE;
101 }
102
103 /******************************************************************************
104  * DnsValidateName_A              [DNSAPI.@]
105  *
106  */
107 DNS_STATUS WINAPI DnsValidateName_A( LPCSTR name, DNS_NAME_FORMAT format )
108 {
109     LPWSTR nameW;
110     DNS_STATUS ret;
111
112     TRACE( "(%s, %d)\n", debugstr_a(name), format );
113
114     nameW = dns_strdup_aw( name );
115     ret = DnsValidateName_W( nameW, format );
116
117     dns_free( nameW );
118     return ret;
119 }
120
121 /******************************************************************************
122  * DnsValidateName_UTF8           [DNSAPI.@]
123  *
124  */
125 DNS_STATUS WINAPI DnsValidateName_UTF8( LPCSTR name, DNS_NAME_FORMAT format )
126 {
127     LPWSTR nameW;
128     DNS_STATUS ret;
129
130     TRACE( "(%s, %d)\n", debugstr_a(name), format );
131
132     nameW = dns_strdup_uw( name );
133     ret = DnsValidateName_W( nameW, format );
134
135     dns_free( nameW );
136     return ret;
137 }
138
139 #define HAS_EXTENDED        0x0001
140 #define HAS_NUMERIC         0x0002
141 #define HAS_NON_NUMERIC     0x0004
142 #define HAS_DOT             0x0008
143 #define HAS_DOT_DOT         0x0010
144 #define HAS_SPACE           0x0020
145 #define HAS_INVALID         0x0040
146 #define HAS_ASTERISK        0x0080
147 #define HAS_UNDERSCORE      0x0100
148 #define HAS_LONG_LABEL      0x0200
149
150 /******************************************************************************
151  * DnsValidateName_W              [DNSAPI.@]
152  *
153  */
154 DNS_STATUS WINAPI DnsValidateName_W( LPCWSTR name, DNS_NAME_FORMAT format )
155 {
156     const WCHAR *p;
157     unsigned int i, j, state = 0;
158     static const WCHAR invalid[] = {
159         '{','|','}','~','[','\\',']','^','\'',':',';','<','=','>',
160         '?','@','!','\"','#','$','%','^','`','(',')','+','/',',',0 };
161
162     TRACE( "(%s, %d)\n", debugstr_w(name), format );
163
164     if (!name) return ERROR_INVALID_NAME;
165
166     for (p = name, i = 0, j = 0; *p; p++, i++, j++)
167     {
168         if (*p == '.')
169         {
170             j = 0;
171             state |= HAS_DOT;
172             if (p[1] == '.') state |= HAS_DOT_DOT;
173         }
174         else if (*p < '0' || *p > '9') state |= HAS_NON_NUMERIC;
175         else state |= HAS_NUMERIC;
176
177         if (j > 62) state |= HAS_LONG_LABEL;
178
179         if (strchrW( invalid, *p )) state |= HAS_INVALID;
180         else if ((unsigned)*p > 127) state |= HAS_EXTENDED;
181         else if (*p == ' ') state |= HAS_SPACE;
182         else if (*p == '_') state |= HAS_UNDERSCORE;
183         else if (*p == '*') state |= HAS_ASTERISK;
184     }
185
186     if (i == 0 || i > 255 ||
187         (state & HAS_LONG_LABEL) ||
188         (state & HAS_DOT_DOT) ||
189         (name[0] == '.' && name[1])) return ERROR_INVALID_NAME;
190
191     switch (format)
192     {
193     case DnsNameDomain:
194     {
195         if (!(state & HAS_NON_NUMERIC) && (state & HAS_NUMERIC))
196             return DNS_ERROR_NUMERIC_NAME;
197         if ((state & HAS_EXTENDED) || (state & HAS_UNDERSCORE))
198             return DNS_ERROR_NON_RFC_NAME;
199         if ((state & HAS_SPACE) ||
200             (state & HAS_INVALID) ||
201             (state & HAS_ASTERISK)) return DNS_ERROR_INVALID_NAME_CHAR;
202         break;
203     }
204     case DnsNameDomainLabel:
205     {
206         if (state & HAS_DOT) return ERROR_INVALID_NAME;
207         if ((state & HAS_EXTENDED) || (state & HAS_UNDERSCORE))
208             return DNS_ERROR_NON_RFC_NAME;
209         if ((state & HAS_SPACE) ||
210             (state & HAS_INVALID) ||
211             (state & HAS_ASTERISK)) return DNS_ERROR_INVALID_NAME_CHAR;
212         break;
213     }
214     case DnsNameHostnameFull:
215     {
216         if (!(state & HAS_NON_NUMERIC) && (state & HAS_NUMERIC))
217             return DNS_ERROR_NUMERIC_NAME;
218         if ((state & HAS_EXTENDED) || (state & HAS_UNDERSCORE))
219             return DNS_ERROR_NON_RFC_NAME;
220         if ((state & HAS_SPACE) ||
221             (state & HAS_INVALID) ||
222             (state & HAS_ASTERISK)) return DNS_ERROR_INVALID_NAME_CHAR;
223         break;
224     }
225     case DnsNameHostnameLabel:
226     {
227         if (state & HAS_DOT) return ERROR_INVALID_NAME;
228         if (!(state & HAS_NON_NUMERIC) && (state & HAS_NUMERIC))
229             return DNS_ERROR_NUMERIC_NAME;
230         if ((state & HAS_EXTENDED) || (state & HAS_UNDERSCORE))
231             return DNS_ERROR_NON_RFC_NAME;
232         if ((state & HAS_SPACE) ||
233             (state & HAS_INVALID) ||
234             (state & HAS_ASTERISK)) return DNS_ERROR_INVALID_NAME_CHAR;
235         break;
236     }
237     case DnsNameWildcard:
238     {
239         if (!(state & HAS_NON_NUMERIC) && (state & HAS_NUMERIC))
240             return ERROR_INVALID_NAME;
241         if (name[0] != '*') return ERROR_INVALID_NAME;
242         if (name[1] && name[1] != '.')
243             return DNS_ERROR_INVALID_NAME_CHAR;
244         if ((state & HAS_EXTENDED) ||
245             (state & HAS_SPACE) ||
246             (state & HAS_INVALID)) return ERROR_INVALID_NAME;
247         break;
248     }
249     case DnsNameSrvRecord:
250     {
251         if (!(state & HAS_NON_NUMERIC) && (state & HAS_NUMERIC))
252             return ERROR_INVALID_NAME;
253         if (name[0] != '_') return ERROR_INVALID_NAME;
254         if ((state & HAS_UNDERSCORE) && !name[1])
255             return DNS_ERROR_NON_RFC_NAME;
256         if ((state & HAS_EXTENDED) ||
257             (state & HAS_SPACE) ||
258             (state & HAS_INVALID)) return ERROR_INVALID_NAME;
259         break;
260     }
261     default:
262         WARN( "unknown format: %d\n", format );
263         break;
264     }
265     return ERROR_SUCCESS;
266 }