urlmon: Implemented a parser for URI query strings.
[wine] / dlls / urlmon / uri.c
1 /*
2  * Copyright 2010 Jacek Caban for CodeWeavers
3  * Copyright 2010 Thomas Mullaly
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2.1 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this library; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
18  */
19
20 #include "urlmon_main.h"
21 #include "wine/debug.h"
22
23 #define NO_SHLWAPI_REG
24 #include "shlwapi.h"
25
26 #define UINT_MAX 0xffffffff
27 #define USHORT_MAX 0xffff
28
29 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
30
31 typedef struct {
32     const IUriVtbl  *lpIUriVtbl;
33     LONG ref;
34
35     BSTR            raw_uri;
36
37     /* Information about the canonicalized URI's buffer. */
38     WCHAR           *canon_uri;
39     DWORD           canon_size;
40     DWORD           canon_len;
41
42     INT             scheme_start;
43     DWORD           scheme_len;
44     URL_SCHEME      scheme_type;
45
46     INT             userinfo_start;
47     DWORD           userinfo_len;
48     INT             userinfo_split;
49
50     INT             host_start;
51     DWORD           host_len;
52     Uri_HOST_TYPE   host_type;
53
54     USHORT          port;
55     BOOL            has_port;
56
57     INT             authority_start;
58     DWORD           authority_len;
59
60     INT             domain_offset;
61
62     INT             path_start;
63     DWORD           path_len;
64     INT             extension_offset;
65 } Uri;
66
67 typedef struct {
68     const IUriBuilderVtbl  *lpIUriBuilderVtbl;
69     LONG ref;
70 } UriBuilder;
71
72 typedef struct {
73     const WCHAR *str;
74     DWORD       len;
75 } h16;
76
77 typedef struct {
78     /* IPv6 addresses can hold up to 8 h16 components. */
79     h16         components[8];
80     DWORD       h16_count;
81
82     /* An IPv6 can have 1 elision ("::"). */
83     const WCHAR *elision;
84
85     /* An IPv6 can contain 1 IPv4 address as the last 32bits of the address. */
86     const WCHAR *ipv4;
87     DWORD       ipv4_len;
88
89     INT         components_size;
90     INT         elision_size;
91 } ipv6_address;
92
93 typedef struct {
94     BSTR            uri;
95
96     BOOL            is_relative;
97     BOOL            is_opaque;
98     BOOL            has_implicit_scheme;
99     BOOL            has_implicit_ip;
100     UINT            implicit_ipv4;
101
102     const WCHAR     *scheme;
103     DWORD           scheme_len;
104     URL_SCHEME      scheme_type;
105
106     const WCHAR     *userinfo;
107     DWORD           userinfo_len;
108     INT             userinfo_split;
109
110     const WCHAR     *host;
111     DWORD           host_len;
112     Uri_HOST_TYPE   host_type;
113
114     BOOL            has_ipv6;
115     ipv6_address    ipv6_address;
116
117     const WCHAR     *port;
118     DWORD           port_len;
119     USHORT          port_value;
120
121     const WCHAR     *path;
122     DWORD           path_len;
123
124     const WCHAR     *query;
125     DWORD           query_len;
126 } parse_data;
127
128 static const CHAR hexDigits[] = "0123456789ABCDEF";
129
130 /* List of scheme types/scheme names that are recognized by the IUri interface as of IE 7. */
131 static const struct {
132     URL_SCHEME  scheme;
133     WCHAR       scheme_name[16];
134 } recognized_schemes[] = {
135     {URL_SCHEME_FTP,            {'f','t','p',0}},
136     {URL_SCHEME_HTTP,           {'h','t','t','p',0}},
137     {URL_SCHEME_GOPHER,         {'g','o','p','h','e','r',0}},
138     {URL_SCHEME_MAILTO,         {'m','a','i','l','t','o',0}},
139     {URL_SCHEME_NEWS,           {'n','e','w','s',0}},
140     {URL_SCHEME_NNTP,           {'n','n','t','p',0}},
141     {URL_SCHEME_TELNET,         {'t','e','l','n','e','t',0}},
142     {URL_SCHEME_WAIS,           {'w','a','i','s',0}},
143     {URL_SCHEME_FILE,           {'f','i','l','e',0}},
144     {URL_SCHEME_MK,             {'m','k',0}},
145     {URL_SCHEME_HTTPS,          {'h','t','t','p','s',0}},
146     {URL_SCHEME_SHELL,          {'s','h','e','l','l',0}},
147     {URL_SCHEME_SNEWS,          {'s','n','e','w','s',0}},
148     {URL_SCHEME_LOCAL,          {'l','o','c','a','l',0}},
149     {URL_SCHEME_JAVASCRIPT,     {'j','a','v','a','s','c','r','i','p','t',0}},
150     {URL_SCHEME_VBSCRIPT,       {'v','b','s','c','r','i','p','t',0}},
151     {URL_SCHEME_ABOUT,          {'a','b','o','u','t',0}},
152     {URL_SCHEME_RES,            {'r','e','s',0}},
153     {URL_SCHEME_MSSHELLROOTED,  {'m','s','-','s','h','e','l','l','-','r','o','o','t','e','d',0}},
154     {URL_SCHEME_MSSHELLIDLIST,  {'m','s','-','s','h','e','l','l','-','i','d','l','i','s','t',0}},
155     {URL_SCHEME_MSHELP,         {'h','c','p',0}},
156     {URL_SCHEME_WILDCARD,       {'*',0}}
157 };
158
159 /* List of default ports Windows recognizes. */
160 static const struct {
161     URL_SCHEME  scheme;
162     USHORT      port;
163 } default_ports[] = {
164     {URL_SCHEME_FTP,    21},
165     {URL_SCHEME_HTTP,   80},
166     {URL_SCHEME_GOPHER, 70},
167     {URL_SCHEME_NNTP,   119},
168     {URL_SCHEME_TELNET, 23},
169     {URL_SCHEME_WAIS,   210},
170     {URL_SCHEME_HTTPS,  443},
171 };
172
173 /* List of 3 character top level domain names Windows seems to recognize.
174  * There might be more, but, these are the only ones I've found so far.
175  */
176 static const struct {
177     WCHAR tld_name[4];
178 } recognized_tlds[] = {
179     {{'c','o','m',0}},
180     {{'e','d','u',0}},
181     {{'g','o','v',0}},
182     {{'i','n','t',0}},
183     {{'m','i','l',0}},
184     {{'n','e','t',0}},
185     {{'o','r','g',0}}
186 };
187
188 static inline BOOL is_alpha(WCHAR val) {
189         return ((val >= 'a' && val <= 'z') || (val >= 'A' && val <= 'Z'));
190 }
191
192 static inline BOOL is_num(WCHAR val) {
193         return (val >= '0' && val <= '9');
194 }
195
196 /* A URI is implicitly a file path if it begins with
197  * a drive letter (eg X:) or starts with "\\" (UNC path).
198  */
199 static inline BOOL is_implicit_file_path(const WCHAR *str) {
200     if(is_alpha(str[0]) && str[1] == ':')
201         return TRUE;
202     else if(str[0] == '\\' && str[1] == '\\')
203         return TRUE;
204
205     return FALSE;
206 }
207
208 /* Checks if the URI is a hierarchical URI. A hierarchical
209  * URI is one that has "//" after the scheme.
210  */
211 static BOOL check_hierarchical(const WCHAR **ptr) {
212     const WCHAR *start = *ptr;
213
214     if(**ptr != '/')
215         return FALSE;
216
217     ++(*ptr);
218     if(**ptr != '/') {
219         *ptr = start;
220         return FALSE;
221     }
222
223     ++(*ptr);
224     return TRUE;
225 }
226
227 /* unreserved    = ALPHA / DIGIT / "-" / "." / "_" / "~" */
228 static inline BOOL is_unreserved(WCHAR val) {
229     return (is_alpha(val) || is_num(val) || val == '-' || val == '.' ||
230             val == '_' || val == '~');
231 }
232
233 /* sub-delims    = "!" / "$" / "&" / "'" / "(" / ")"
234  *               / "*" / "+" / "," / ";" / "="
235  */
236 static inline BOOL is_subdelim(WCHAR val) {
237     return (val == '!' || val == '$' || val == '&' ||
238             val == '\'' || val == '(' || val == ')' ||
239             val == '*' || val == '+' || val == ',' ||
240             val == ';' || val == '=');
241 }
242
243 /* gen-delims  = ":" / "/" / "?" / "#" / "[" / "]" / "@" */
244 static inline BOOL is_gendelim(WCHAR val) {
245     return (val == ':' || val == '/' || val == '?' ||
246             val == '#' || val == '[' || val == ']' ||
247             val == '@');
248 }
249
250 /* Characters that delimit the end of the authority
251  * section of a URI. Sometimes a '\\' is considered
252  * an authority delimeter.
253  */
254 static inline BOOL is_auth_delim(WCHAR val, BOOL acceptSlash) {
255     return (val == '#' || val == '/' || val == '?' ||
256             val == '\0' || (acceptSlash && val == '\\'));
257 }
258
259 /* reserved = gen-delims / sub-delims */
260 static inline BOOL is_reserved(WCHAR val) {
261     return (is_subdelim(val) || is_gendelim(val));
262 }
263
264 static inline BOOL is_hexdigit(WCHAR val) {
265     return ((val >= 'a' && val <= 'f') ||
266             (val >= 'A' && val <= 'F') ||
267             (val >= '0' && val <= '9'));
268 }
269
270 static inline BOOL is_path_delim(WCHAR val) {
271     return (!val || val == '#' || val == '?');
272 }
273
274 /* Computes the size of the given IPv6 address.
275  * Each h16 component is 16bits, if there is an IPv4 address, it's
276  * 32bits. If there's an elision it can be 16bits to 128bits, depending
277  * on the number of other components.
278  *
279  * Modeled after google-url's CheckIPv6ComponentsSize function
280  */
281 static void compute_ipv6_comps_size(ipv6_address *address) {
282     address->components_size = address->h16_count * 2;
283
284     if(address->ipv4)
285         /* IPv4 address is 4 bytes. */
286         address->components_size += 4;
287
288     if(address->elision) {
289         /* An elision can be anywhere from 2 bytes up to 16 bytes.
290          * It size depends on the size of the h16 and IPv4 components.
291          */
292         address->elision_size = 16 - address->components_size;
293         if(address->elision_size < 2)
294             address->elision_size = 2;
295     } else
296         address->elision_size = 0;
297 }
298
299 /* Taken from dlls/jscript/lex.c */
300 static int hex_to_int(WCHAR val) {
301     if(val >= '0' && val <= '9')
302         return val - '0';
303     else if(val >= 'a' && val <= 'f')
304         return val - 'a' + 10;
305     else if(val >= 'A' && val <= 'F')
306         return val - 'A' + 10;
307
308     return -1;
309 }
310
311 /* Helper function for converting a percent encoded string
312  * representation of a WCHAR value into its actual WCHAR value. If
313  * the two characters following the '%' aren't valid hex values then
314  * this function returns the NULL character.
315  *
316  * Eg.
317  *  "%2E" will result in '.' being returned by this function.
318  */
319 static WCHAR decode_pct_val(const WCHAR *ptr) {
320     WCHAR ret = '\0';
321
322     if(*ptr == '%' && is_hexdigit(*(ptr + 1)) && is_hexdigit(*(ptr + 2))) {
323         INT a = hex_to_int(*(ptr + 1));
324         INT b = hex_to_int(*(ptr + 2));
325
326         ret = a << 4;
327         ret += b;
328     }
329
330     return ret;
331 }
332
333 /* Helper function for percent encoding a given character
334  * and storing the encoded value into a given buffer (dest).
335  *
336  * It's up to the calling function to ensure that there is
337  * at least enough space in 'dest' for the percent encoded
338  * value to be stored (so dest + 3 spaces available).
339  */
340 static inline void pct_encode_val(WCHAR val, WCHAR *dest) {
341     dest[0] = '%';
342     dest[1] = hexDigits[(val >> 4) & 0xf];
343     dest[2] = hexDigits[val & 0xf];
344 }
345
346 /* Scans the range of characters [str, end] and returns the last occurence
347  * of 'ch' or returns NULL.
348  */
349 static const WCHAR *str_last_of(const WCHAR *str, const WCHAR *end, WCHAR ch) {
350     const WCHAR *ptr = end;
351
352     while(ptr >= str) {
353         if(*ptr == ch)
354             return ptr;
355         --ptr;
356     }
357
358     return NULL;
359 }
360
361 /* Attempts to parse the domain name from the host.
362  *
363  * This function also includes the Top-level Domain (TLD) name
364  * of the host when it tries to find the domain name. If it finds
365  * a valid domain name it will assign 'domain_start' the offset
366  * into 'host' where the domain name starts.
367  *
368  * It's implied that if a domain name its range is implied to be
369  * [host+domain_start, host+host_len).
370  */
371 static void find_domain_name(const WCHAR *host, DWORD host_len,
372                              INT *domain_start) {
373     const WCHAR *last_tld, *sec_last_tld, *end;
374
375     end = host+host_len-1;
376
377     *domain_start = -1;
378
379     /* There has to be at least enough room for a '.' followed by a
380      * 3 character TLD for a domain to even exist in the host name.
381      */
382     if(host_len < 4)
383         return;
384
385     last_tld = str_last_of(host, end, '.');
386     if(!last_tld)
387         /* http://hostname -> has no domain name. */
388         return;
389
390     sec_last_tld = str_last_of(host, last_tld-1, '.');
391     if(!sec_last_tld) {
392         /* If the '.' is at the beginning of the host there
393          * has to be at least 3 characters in the TLD for it
394          * to be valid.
395          *  Ex: .com -> .com as the domain name.
396          *      .co  -> has no domain name.
397          */
398         if(last_tld-host == 0) {
399             if(end-(last_tld-1) < 3)
400                 return;
401         } else if(last_tld-host == 3) {
402             DWORD i;
403
404             /* If there's three characters in front of last_tld and
405              * they are on the list of recognized TLDs, then this
406              * host doesn't have a domain (since the host only contains
407              * a TLD name.
408              *  Ex: edu.uk -> has no domain name.
409              *      foo.uk -> foo.uk as the domain name.
410              */
411             for(i = 0; i < sizeof(recognized_tlds)/sizeof(recognized_tlds[0]); ++i) {
412                 if(!StrCmpNIW(host, recognized_tlds[i].tld_name, 3))
413                     return;
414             }
415         } else if(last_tld-host < 3)
416             /* Anything less then 3 characters is considered part
417              * of the TLD name.
418              *  Ex: ak.uk -> Has no domain name.
419              */
420             return;
421
422         /* Otherwise the domain name is the whole host name. */
423         *domain_start = 0;
424     } else if(end+1-last_tld > 3) {
425         /* If the last_tld has more then 3 characters then it's automatically
426          * considered the TLD of the domain name.
427          *  Ex: www.winehq.org.uk.test -> uk.test as the domain name.
428          */
429         *domain_start = (sec_last_tld+1)-host;
430     } else if(last_tld - (sec_last_tld+1) < 4) {
431         DWORD i;
432         /* If the sec_last_tld is 3 characters long it HAS to be on the list of
433          * recognized to still be considered part of the TLD name, otherwise
434          * its considered the domain name.
435          *  Ex: www.google.com.uk -> google.com.uk as the domain name.
436          *      www.google.foo.uk -> foo.uk as the domain name.
437          */
438         if(last_tld - (sec_last_tld+1) == 3) {
439             for(i = 0; i < sizeof(recognized_tlds)/sizeof(recognized_tlds[0]); ++i) {
440                 if(!StrCmpNIW(sec_last_tld+1, recognized_tlds[i].tld_name, 3)) {
441                     const WCHAR *domain = str_last_of(host, sec_last_tld-1, '.');
442
443                     if(!domain)
444                         *domain_start = 0;
445                     else
446                         *domain_start = (domain+1) - host;
447                     TRACE("Found domain name %s\n", debugstr_wn(host+*domain_start,
448                                                         (host+host_len)-(host+*domain_start)));
449                     return;
450                 }
451             }
452
453             *domain_start = (sec_last_tld+1)-host;
454         } else {
455             /* Since the sec_last_tld is less then 3 characters it's considered
456              * part of the TLD.
457              *  Ex: www.google.fo.uk -> google.fo.uk as the domain name.
458              */
459             const WCHAR *domain = str_last_of(host, sec_last_tld-1, '.');
460
461             if(!domain)
462                 *domain_start = 0;
463             else
464                 *domain_start = (domain+1) - host;
465         }
466     } else {
467         /* The second to last TLD has more then 3 characters making it
468          * the domain name.
469          *  Ex: www.google.test.us -> test.us as the domain name.
470          */
471         *domain_start = (sec_last_tld+1)-host;
472     }
473
474     TRACE("Found domain name %s\n", debugstr_wn(host+*domain_start,
475                                         (host+host_len)-(host+*domain_start)));
476 }
477
478 /* Removes the dot segments from a heirarchical URIs path component. This
479  * function performs the removal in place.
480  *
481  * This is a modified version of Qt's QUrl function "removeDotsFromPath".
482  *
483  * This function returns the new length of the path string.
484  */
485 static DWORD remove_dot_segments(WCHAR *path, DWORD path_len) {
486     WCHAR *out = path;
487     const WCHAR *in = out;
488     const WCHAR *end = out + path_len;
489     DWORD len;
490
491     while(in < end) {
492         /* A.  if the input buffer begins with a prefix of "/./" or "/.",
493          *     where "." is a complete path segment, then replace that
494          *     prefix with "/" in the input buffer; otherwise,
495          */
496         if(in <= end - 3 && in[0] == '/' && in[1] == '.' && in[2] == '/') {
497             in += 2;
498             continue;
499         } else if(in == end - 2 && in[0] == '/' && in[1] == '.') {
500             *out++ = '/';
501             in += 2;
502             break;
503         }
504
505         /* B.  if the input buffer begins with a prefix of "/../" or "/..",
506          *     where ".." is a complete path segment, then replace that
507          *     prefix with "/" in the input buffer and remove the last
508          *     segment and its preceding "/" (if any) from the output
509          *     buffer; otherwise,
510          */
511         if(in <= end - 4 && in[0] == '/' && in[1] == '.' && in[2] == '.' && in[3] == '/') {
512             while(out > path && *(--out) != '/');
513
514             in += 3;
515             continue;
516         } else if(in == end - 3 && in[0] == '/' && in[1] == '.' && in[2] == '.') {
517             while(out > path && *(--out) != '/');
518
519             if(*out == '/')
520                 ++out;
521
522             in += 3;
523             break;
524         }
525
526         /* C.  move the first path segment in the input buffer to the end of
527          *     the output buffer, including the initial "/" character (if
528          *     any) and any subsequent characters up to, but not including,
529          *     the next "/" character or the end of the input buffer.
530          */
531         *out++ = *in++;
532         while(in < end && *in != '/')
533             *out++ = *in++;
534     }
535
536     len = out - path;
537     TRACE("(%p %d): Path after dot segments removed %s len=%d\n", path, path_len,
538         debugstr_wn(path, len), len);
539     return len;
540 }
541
542 /* Attempts to find the file extension in a given path. */
543 static INT find_file_extension(const WCHAR *path, DWORD path_len) {
544     const WCHAR *end;
545
546     for(end = path+path_len-1; end >= path && *end != '/' && *end != '\\'; --end) {
547         if(*end == '.')
548             return end-path;
549     }
550
551     return -1;
552 }
553
554 /* Computes the location where the elision should occur in the IPv6
555  * address using the numerical values of each component stored in
556  * 'values'. If the address shouldn't contain an elision then 'index'
557  * is assigned -1 as it's value. Otherwise 'index' will contain the
558  * starting index (into values) where the elision should be, and 'count'
559  * will contain the number of cells the elision covers.
560  *
561  * NOTES:
562  *  Windows will expand an elision if the elision only represents 1 h16
563  *  component of the URI.
564  *
565  *  Ex: [1::2:3:4:5:6:7] -> [1:0:2:3:4:5:6:7]
566  *
567  *  If the IPv6 address contains an IPv4 address, the IPv4 address is also
568  *  considered for being included as part of an elision if all it's components
569  *  are zeros.
570  *
571  *  Ex: [1:2:3:4:5:6:0.0.0.0] -> [1:2:3:4:5:6::]
572  */
573 static void compute_elision_location(const ipv6_address *address, const USHORT values[8],
574                                      INT *index, DWORD *count) {
575     DWORD i, max_len, cur_len;
576     INT max_index, cur_index;
577
578     max_len = cur_len = 0;
579     max_index = cur_index = -1;
580     for(i = 0; i < 8; ++i) {
581         BOOL check_ipv4 = (address->ipv4 && i == 6);
582         BOOL is_end = (check_ipv4 || i == 7);
583
584         if(check_ipv4) {
585             /* Check if the IPv4 address contains only zeros. */
586             if(values[i] == 0 && values[i+1] == 0) {
587                 if(cur_index == -1)
588                     cur_index = i;
589
590                 cur_len += 2;
591                 ++i;
592             }
593         } else if(values[i] == 0) {
594             if(cur_index == -1)
595                 cur_index = i;
596
597             ++cur_len;
598         }
599
600         if(is_end || values[i] != 0) {
601             /* We only consider it for an elision if it's
602              * more then 1 component long.
603              */
604             if(cur_len > 1 && cur_len > max_len) {
605                 /* Found the new elision location. */
606                 max_len = cur_len;
607                 max_index = cur_index;
608             }
609
610             /* Reset the current range for the next range of zeros. */
611             cur_index = -1;
612             cur_len = 0;
613         }
614     }
615
616     *index = max_index;
617     *count = max_len;
618 }
619
620 /* Converts the specified IPv4 address into an uint value.
621  *
622  * This function assumes that the IPv4 address has already been validated.
623  */
624 static UINT ipv4toui(const WCHAR *ip, DWORD len) {
625     UINT ret = 0;
626     DWORD comp_value = 0;
627     const WCHAR *ptr;
628
629     for(ptr = ip; ptr < ip+len; ++ptr) {
630         if(*ptr == '.') {
631             ret <<= 8;
632             ret += comp_value;
633             comp_value = 0;
634         } else
635             comp_value = comp_value*10 + (*ptr-'0');
636     }
637
638     ret <<= 8;
639     ret += comp_value;
640
641     return ret;
642 }
643
644 /* Converts an IPv4 address in numerical form into it's fully qualified
645  * string form. This function returns the number of characters written
646  * to 'dest'. If 'dest' is NULL this function will return the number of
647  * characters that would have been written.
648  *
649  * It's up to the caller to ensure there's enough space in 'dest' for the
650  * address.
651  */
652 static DWORD ui2ipv4(WCHAR *dest, UINT address) {
653     static const WCHAR formatW[] =
654         {'%','u','.','%','u','.','%','u','.','%','u',0};
655     DWORD ret = 0;
656     UCHAR digits[4];
657
658     digits[0] = (address >> 24) & 0xff;
659     digits[1] = (address >> 16) & 0xff;
660     digits[2] = (address >> 8) & 0xff;
661     digits[3] = address & 0xff;
662
663     if(!dest) {
664         WCHAR tmp[16];
665         ret = sprintfW(tmp, formatW, digits[0], digits[1], digits[2], digits[3]);
666     } else
667         ret = sprintfW(dest, formatW, digits[0], digits[1], digits[2], digits[3]);
668
669     return ret;
670 }
671
672 /* Converts an h16 component (from an IPv6 address) into it's
673  * numerical value.
674  *
675  * This function assumes that the h16 component has already been validated.
676  */
677 static USHORT h16tous(h16 component) {
678     DWORD i;
679     USHORT ret = 0;
680
681     for(i = 0; i < component.len; ++i) {
682         ret <<= 4;
683         ret += hex_to_int(component.str[i]);
684     }
685
686     return ret;
687 }
688
689 /* Converts an IPv6 address into it's 128 bits (16 bytes) numerical value.
690  *
691  * This function assumes that the ipv6_address has already been validated.
692  */
693 static BOOL ipv6_to_number(const ipv6_address *address, USHORT number[8]) {
694     DWORD i, cur_component = 0;
695     BOOL already_passed_elision = FALSE;
696
697     for(i = 0; i < address->h16_count; ++i) {
698         if(address->elision) {
699             if(address->components[i].str > address->elision && !already_passed_elision) {
700                 /* Means we just passed the elision and need to add it's values to
701                  * 'number' before we do anything else.
702                  */
703                 DWORD j = 0;
704                 for(j = 0; j < address->elision_size; j+=2)
705                     number[cur_component++] = 0;
706
707                 already_passed_elision = TRUE;
708             }
709         }
710
711         number[cur_component++] = h16tous(address->components[i]);
712     }
713
714     /* Case when the elision appears after the h16 components. */
715     if(!already_passed_elision && address->elision) {
716         for(i = 0; i < address->elision_size; i+=2)
717             number[cur_component++] = 0;
718         already_passed_elision = TRUE;
719     }
720
721     if(address->ipv4) {
722         UINT value = ipv4toui(address->ipv4, address->ipv4_len);
723
724         if(cur_component != 6) {
725             ERR("(%p %p): Failed sanity check with %d\n", address, number, cur_component);
726             return FALSE;
727         }
728
729         number[cur_component++] = (value >> 16) & 0xffff;
730         number[cur_component] = value & 0xffff;
731     }
732
733     return TRUE;
734 }
735
736 /* Checks if the characters pointed to by 'ptr' are
737  * a percent encoded data octet.
738  *
739  * pct-encoded = "%" HEXDIG HEXDIG
740  */
741 static BOOL check_pct_encoded(const WCHAR **ptr) {
742     const WCHAR *start = *ptr;
743
744     if(**ptr != '%')
745         return FALSE;
746
747     ++(*ptr);
748     if(!is_hexdigit(**ptr)) {
749         *ptr = start;
750         return FALSE;
751     }
752
753     ++(*ptr);
754     if(!is_hexdigit(**ptr)) {
755         *ptr = start;
756         return FALSE;
757     }
758
759     ++(*ptr);
760     return TRUE;
761 }
762
763 /* dec-octet   = DIGIT                 ; 0-9
764  *             / %x31-39 DIGIT         ; 10-99
765  *             / "1" 2DIGIT            ; 100-199
766  *             / "2" %x30-34 DIGIT     ; 200-249
767  *             / "25" %x30-35          ; 250-255
768  */
769 static BOOL check_dec_octet(const WCHAR **ptr) {
770     const WCHAR *c1, *c2, *c3;
771
772     c1 = *ptr;
773     /* A dec-octet must be at least 1 digit long. */
774     if(*c1 < '0' || *c1 > '9')
775         return FALSE;
776
777     ++(*ptr);
778
779     c2 = *ptr;
780     /* Since the 1 digit requirment was meet, it doesn't
781      * matter if this is a DIGIT value, it's considered a
782      * dec-octet.
783      */
784     if(*c2 < '0' || *c2 > '9')
785         return TRUE;
786
787     ++(*ptr);
788
789     c3 = *ptr;
790     /* Same explanation as above. */
791     if(*c3 < '0' || *c3 > '9')
792         return TRUE;
793
794     /* Anything > 255 isn't a valid IP dec-octet. */
795     if(*c1 >= '2' && *c2 >= '5' && *c3 >= '5') {
796         *ptr = c1;
797         return FALSE;
798     }
799
800     ++(*ptr);
801     return TRUE;
802 }
803
804 /* Checks if there is an implicit IPv4 address in the host component of the URI.
805  * The max value of an implicit IPv4 address is UINT_MAX.
806  *
807  *  Ex:
808  *      "234567" would be considered an implicit IPv4 address.
809  */
810 static BOOL check_implicit_ipv4(const WCHAR **ptr, UINT *val) {
811     const WCHAR *start = *ptr;
812     ULONGLONG ret = 0;
813     *val = 0;
814
815     while(is_num(**ptr)) {
816         ret = ret*10 + (**ptr - '0');
817
818         if(ret > UINT_MAX) {
819             *ptr = start;
820             return FALSE;
821         }
822         ++(*ptr);
823     }
824
825     if(*ptr == start)
826         return FALSE;
827
828     *val = ret;
829     return TRUE;
830 }
831
832 /* Checks if the string contains an IPv4 address.
833  *
834  * This function has a strict mode or a non-strict mode of operation
835  * When 'strict' is set to FALSE this function will return TRUE if
836  * the string contains at least 'dec-octet "." dec-octet' since partial
837  * IPv4 addresses will be normalized out into full IPv4 addresses. When
838  * 'strict' is set this function expects there to be a full IPv4 address.
839  *
840  * IPv4address = dec-octet "." dec-octet "." dec-octet "." dec-octet
841  */
842 static BOOL check_ipv4address(const WCHAR **ptr, BOOL strict) {
843     const WCHAR *start = *ptr;
844
845     if(!check_dec_octet(ptr)) {
846         *ptr = start;
847         return FALSE;
848     }
849
850     if(**ptr != '.') {
851         *ptr = start;
852         return FALSE;
853     }
854
855     ++(*ptr);
856     if(!check_dec_octet(ptr)) {
857         *ptr = start;
858         return FALSE;
859     }
860
861     if(**ptr != '.') {
862         if(strict) {
863             *ptr = start;
864             return FALSE;
865         } else
866             return TRUE;
867     }
868
869     ++(*ptr);
870     if(!check_dec_octet(ptr)) {
871         *ptr = start;
872         return FALSE;
873     }
874
875     if(**ptr != '.') {
876         if(strict) {
877             *ptr = start;
878             return FALSE;
879         } else
880             return TRUE;
881     }
882
883     ++(*ptr);
884     if(!check_dec_octet(ptr)) {
885         *ptr = start;
886         return FALSE;
887     }
888
889     /* Found a four digit ip address. */
890     return TRUE;
891 }
892 /* Tries to parse the scheme name of the URI.
893  *
894  * scheme = ALPHA *(ALPHA | NUM | '+' | '-' | '.') as defined by RFC 3896.
895  * NOTE: Windows accepts a number as the first character of a scheme.
896  */
897 static BOOL parse_scheme_name(const WCHAR **ptr, parse_data *data) {
898     const WCHAR *start = *ptr;
899
900     data->scheme = NULL;
901     data->scheme_len = 0;
902
903     while(**ptr) {
904         if(**ptr == '*' && *ptr == start) {
905             /* Might have found a wildcard scheme. If it is the next
906              * char has to be a ':' for it to be a valid URI
907              */
908             ++(*ptr);
909             break;
910         } else if(!is_num(**ptr) && !is_alpha(**ptr) && **ptr != '+' &&
911            **ptr != '-' && **ptr != '.')
912             break;
913
914         (*ptr)++;
915     }
916
917     if(*ptr == start)
918         return FALSE;
919
920     /* Schemes must end with a ':' */
921     if(**ptr != ':') {
922         *ptr = start;
923         return FALSE;
924     }
925
926     data->scheme = start;
927     data->scheme_len = *ptr - start;
928
929     ++(*ptr);
930     return TRUE;
931 }
932
933 /* Tries to deduce the corresponding URL_SCHEME for the given URI. Stores
934  * the deduced URL_SCHEME in data->scheme_type.
935  */
936 static BOOL parse_scheme_type(parse_data *data) {
937     /* If there's scheme data then see if it's a recognized scheme. */
938     if(data->scheme && data->scheme_len) {
939         DWORD i;
940
941         for(i = 0; i < sizeof(recognized_schemes)/sizeof(recognized_schemes[0]); ++i) {
942             if(lstrlenW(recognized_schemes[i].scheme_name) == data->scheme_len) {
943                 /* Has to be a case insensitive compare. */
944                 if(!StrCmpNIW(recognized_schemes[i].scheme_name, data->scheme, data->scheme_len)) {
945                     data->scheme_type = recognized_schemes[i].scheme;
946                     return TRUE;
947                 }
948             }
949         }
950
951         /* If we get here it means it's not a recognized scheme. */
952         data->scheme_type = URL_SCHEME_UNKNOWN;
953         return TRUE;
954     } else if(data->is_relative) {
955         /* Relative URI's have no scheme. */
956         data->scheme_type = URL_SCHEME_UNKNOWN;
957         return TRUE;
958     } else {
959         /* Should never reach here! what happened... */
960         FIXME("(%p): Unable to determine scheme type for URI %s\n", data, debugstr_w(data->uri));
961         return FALSE;
962     }
963 }
964
965 /* Tries to parse (or deduce) the scheme_name of a URI. If it can't
966  * parse a scheme from the URI it will try to deduce the scheme_name and scheme_type
967  * using the flags specified in 'flags' (if any). Flags that affect how this function
968  * operates are the Uri_CREATE_ALLOW_* flags.
969  *
970  * All parsed/deduced information will be stored in 'data' when the function returns.
971  *
972  * Returns TRUE if it was able to successfully parse the information.
973  */
974 static BOOL parse_scheme(const WCHAR **ptr, parse_data *data, DWORD flags) {
975     static const WCHAR fileW[] = {'f','i','l','e',0};
976     static const WCHAR wildcardW[] = {'*',0};
977
978     /* First check to see if the uri could implicitly be a file path. */
979     if(is_implicit_file_path(*ptr)) {
980         if(flags & Uri_CREATE_ALLOW_IMPLICIT_FILE_SCHEME) {
981             data->scheme = fileW;
982             data->scheme_len = lstrlenW(fileW);
983             data->has_implicit_scheme = TRUE;
984
985             TRACE("(%p %p %x): URI is an implicit file path.\n", ptr, data, flags);
986         } else {
987             /* Window's does not consider anything that can implicitly be a file
988              * path to be a valid URI if the ALLOW_IMPLICIT_FILE_SCHEME flag is not set...
989              */
990             TRACE("(%p %p %x): URI is implicitly a file path, but, the ALLOW_IMPLICIT_FILE_SCHEME flag wasn't set.\n",
991                     ptr, data, flags);
992             return FALSE;
993         }
994     } else if(!parse_scheme_name(ptr, data)) {
995         /* No Scheme was found, this means it could be:
996          *      a) an implicit Wildcard scheme
997          *      b) a relative URI
998          *      c) a invalid URI.
999          */
1000         if(flags & Uri_CREATE_ALLOW_IMPLICIT_WILDCARD_SCHEME) {
1001             data->scheme = wildcardW;
1002             data->scheme_len = lstrlenW(wildcardW);
1003             data->has_implicit_scheme = TRUE;
1004
1005             TRACE("(%p %p %x): URI is an implicit wildcard scheme.\n", ptr, data, flags);
1006         } else if (flags & Uri_CREATE_ALLOW_RELATIVE) {
1007             data->is_relative = TRUE;
1008             TRACE("(%p %p %x): URI is relative.\n", ptr, data, flags);
1009         } else {
1010             TRACE("(%p %p %x): Malformed URI found. Unable to deduce scheme name.\n", ptr, data, flags);
1011             return FALSE;
1012         }
1013     }
1014
1015     if(!data->is_relative)
1016         TRACE("(%p %p %x): Found scheme=%s scheme_len=%d\n", ptr, data, flags,
1017                 debugstr_wn(data->scheme, data->scheme_len), data->scheme_len);
1018
1019     if(!parse_scheme_type(data))
1020         return FALSE;
1021
1022     TRACE("(%p %p %x): Assigned %d as the URL_SCHEME.\n", ptr, data, flags, data->scheme_type);
1023     return TRUE;
1024 }
1025
1026 /* Parses the userinfo part of the URI (if it exists). The userinfo field of
1027  * a URI can consist of "username:password@", or just "username@".
1028  *
1029  * RFC def:
1030  * userinfo    = *( unreserved / pct-encoded / sub-delims / ":" )
1031  *
1032  * NOTES:
1033  *  1)  If there is more than one ':' in the userinfo part of the URI Windows
1034  *      uses the first occurence of ':' to delimit the username and password
1035  *      components.
1036  *
1037  *      ex:
1038  *          ftp://user:pass:word@winehq.org
1039  *
1040  *      Would yield, "user" as the username and "pass:word" as the password.
1041  *
1042  *  2)  Windows allows any character to appear in the "userinfo" part of
1043  *      a URI, as long as it's not an authority delimeter character set.
1044  */
1045 static void parse_userinfo(const WCHAR **ptr, parse_data *data, DWORD flags) {
1046     data->userinfo = *ptr;
1047     data->userinfo_split = -1;
1048
1049     while(**ptr != '@') {
1050         if(**ptr == ':' && data->userinfo_split == -1)
1051             data->userinfo_split = *ptr - data->userinfo;
1052         else if(**ptr == '%') {
1053             /* If it's a known scheme type, it has to be a valid percent
1054              * encoded value.
1055              */
1056             if(!check_pct_encoded(ptr)) {
1057                 if(data->scheme_type != URL_SCHEME_UNKNOWN) {
1058                     *ptr = data->userinfo;
1059                     data->userinfo = NULL;
1060                     data->userinfo_split = -1;
1061
1062                     TRACE("(%p %p %x): URI contained no userinfo.\n", ptr, data, flags);
1063                     return;
1064                 }
1065             } else
1066                 continue;
1067         } else if(is_auth_delim(**ptr, data->scheme_type != URL_SCHEME_UNKNOWN))
1068             break;
1069
1070         ++(*ptr);
1071     }
1072
1073     if(**ptr != '@') {
1074         *ptr = data->userinfo;
1075         data->userinfo = NULL;
1076         data->userinfo_split = -1;
1077
1078         TRACE("(%p %p %x): URI contained no userinfo.\n", ptr, data, flags);
1079         return;
1080     }
1081
1082     data->userinfo_len = *ptr - data->userinfo;
1083     TRACE("(%p %p %x): Found userinfo=%s userinfo_len=%d split=%d.\n", ptr, data, flags,
1084             debugstr_wn(data->userinfo, data->userinfo_len), data->userinfo_len, data->userinfo_split);
1085     ++(*ptr);
1086 }
1087
1088 /* Attempts to parse a port from the URI.
1089  *
1090  * NOTES:
1091  *  Windows seems to have a cap on what the maximum value
1092  *  for a port can be. The max value is USHORT_MAX.
1093  *
1094  * port = *DIGIT
1095  */
1096 static BOOL parse_port(const WCHAR **ptr, parse_data *data, DWORD flags) {
1097     UINT port = 0;
1098     data->port = *ptr;
1099
1100     while(!is_auth_delim(**ptr, data->scheme_type != URL_SCHEME_UNKNOWN)) {
1101         if(!is_num(**ptr)) {
1102             *ptr = data->port;
1103             data->port = NULL;
1104             return FALSE;
1105         }
1106
1107         port = port*10 + (**ptr-'0');
1108
1109         if(port > USHORT_MAX) {
1110             *ptr = data->port;
1111             data->port = NULL;
1112             return FALSE;
1113         }
1114
1115         ++(*ptr);
1116     }
1117
1118     data->port_value = port;
1119     data->port_len = *ptr - data->port;
1120
1121     TRACE("(%p %p %x): Found port %s len=%d value=%u\n", ptr, data, flags,
1122         debugstr_wn(data->port, data->port_len), data->port_len, data->port_value);
1123     return TRUE;
1124 }
1125
1126 /* Attempts to parse a IPv4 address from the URI.
1127  *
1128  * NOTES:
1129  *  Window's normalizes IPv4 addresses, This means there's three
1130  *  possibilities for the URI to contain an IPv4 address.
1131  *      1)  A well formed address (ex. 192.2.2.2).
1132  *      2)  A partially formed address. For example "192.0" would
1133  *          normalize to "192.0.0.0" during canonicalization.
1134  *      3)  An implicit IPv4 address. For example "256" would
1135  *          normalize to "0.0.1.0" during canonicalization. Also
1136  *          note that the maximum value for an implicit IP address
1137  *          is UINT_MAX, if the value in the URI exceeds this then
1138  *          it is not considered an IPv4 address.
1139  */
1140 static BOOL parse_ipv4address(const WCHAR **ptr, parse_data *data, DWORD flags) {
1141     const BOOL is_unknown = data->scheme_type == URL_SCHEME_UNKNOWN;
1142     data->host = *ptr;
1143
1144     if(!check_ipv4address(ptr, FALSE)) {
1145         if(!check_implicit_ipv4(ptr, &data->implicit_ipv4)) {
1146             TRACE("(%p %p %x): URI didn't contain anything looking like an IPv4 address.\n",
1147                 ptr, data, flags);
1148             *ptr = data->host;
1149             data->host = NULL;
1150             return FALSE;
1151         } else
1152             data->has_implicit_ip = TRUE;
1153     }
1154
1155     /* Check if what we found is the only part of the host name (if it isn't
1156      * we don't have an IPv4 address).
1157      */
1158     if(**ptr == ':') {
1159         ++(*ptr);
1160         if(!parse_port(ptr, data, flags)) {
1161             *ptr = data->host;
1162             data->host = NULL;
1163             return FALSE;
1164         }
1165     } else if(!is_auth_delim(**ptr, !is_unknown)) {
1166         /* Found more data which belongs the host, so this isn't an IPv4. */
1167         *ptr = data->host;
1168         data->host = NULL;
1169         data->has_implicit_ip = FALSE;
1170         return FALSE;
1171     }
1172
1173     data->host_len = *ptr - data->host;
1174     data->host_type = Uri_HOST_IPV4;
1175
1176     TRACE("(%p %p %x): IPv4 address found. host=%s host_len=%d host_type=%d\n",
1177         ptr, data, flags, debugstr_wn(data->host, data->host_len),
1178         data->host_len, data->host_type);
1179     return TRUE;
1180 }
1181
1182 /* Attempts to parse the reg-name from the URI.
1183  *
1184  * Because of the way Windows handles ':' this function also
1185  * handles parsing the port.
1186  *
1187  * reg-name = *( unreserved / pct-encoded / sub-delims )
1188  *
1189  * NOTE:
1190  *  Windows allows everything, but, the characters in "auth_delims" and ':'
1191  *  to appear in a reg-name, unless it's an unknown scheme type then ':' is
1192  *  allowed to appear (even if a valid port isn't after it).
1193  *
1194  *  Windows doesn't like host names which start with '[' and end with ']'
1195  *  and don't contain a valid IP literal address in between them.
1196  *
1197  *  On Windows if an '[' is encountered in the host name the ':' no longer
1198  *  counts as a delimiter until you reach the next ']' or an "authority delimeter".
1199  *
1200  *  A reg-name CAN be empty.
1201  */
1202 static BOOL parse_reg_name(const WCHAR **ptr, parse_data *data, DWORD flags) {
1203     const BOOL has_start_bracket = **ptr == '[';
1204     const BOOL known_scheme = data->scheme_type != URL_SCHEME_UNKNOWN;
1205     BOOL inside_brackets = has_start_bracket;
1206     BOOL ignore_col = FALSE;
1207
1208     /* We have to be careful with file schemes. */
1209     if(data->scheme_type == URL_SCHEME_FILE) {
1210         /* This is because an implicit file scheme could be "C:\\test" and it
1211          * would trick this function into thinking the host is "C", when after
1212          * canonicalization the host would end up being an empty string.
1213          */
1214         if(is_alpha(**ptr) && *(*ptr+1) == ':') {
1215             /* Regular old drive paths don't have a host type (or host name). */
1216             data->host_type = Uri_HOST_UNKNOWN;
1217             data->host = *ptr;
1218             data->host_len = 0;
1219             return TRUE;
1220         } else if(**ptr == '\\' && *(*ptr+1) == '\\')
1221             /* Skip past the "\\" of a UNC path. */
1222             *ptr += 2;
1223     }
1224
1225     data->host = *ptr;
1226
1227     while(!is_auth_delim(**ptr, known_scheme)) {
1228         if(**ptr == ':' && !ignore_col) {
1229             /* We can ignore ':' if were inside brackets.*/
1230             if(!inside_brackets) {
1231                 const WCHAR *tmp = (*ptr)++;
1232
1233                 /* Attempt to parse the port. */
1234                 if(!parse_port(ptr, data, flags)) {
1235                     /* Windows expects there to be a valid port for known scheme types. */
1236                     if(data->scheme_type != URL_SCHEME_UNKNOWN) {
1237                         *ptr = data->host;
1238                         data->host = NULL;
1239                         TRACE("(%p %p %x): Expected valid port\n", ptr, data, flags);
1240                         return FALSE;
1241                     } else
1242                         /* Windows gives up on trying to parse a port when it
1243                          * encounters 1 invalid port.
1244                          */
1245                         ignore_col = TRUE;
1246                 } else {
1247                     data->host_len = tmp - data->host;
1248                     break;
1249                 }
1250             }
1251         } else if(**ptr == '%' && known_scheme) {
1252             /* Has to be a legit % encoded value. */
1253             if(!check_pct_encoded(ptr)) {
1254                 *ptr = data->host;
1255                 data->host = NULL;
1256                 return FALSE;
1257             } else
1258                 continue;
1259         } else if(**ptr == ']')
1260             inside_brackets = FALSE;
1261         else if(**ptr == '[')
1262             inside_brackets = TRUE;
1263
1264         ++(*ptr);
1265     }
1266
1267     if(has_start_bracket) {
1268         /* Make sure the last character of the host wasn't a ']'. */
1269         if(*(*ptr-1) == ']') {
1270             TRACE("(%p %p %x): Expected an IP literal inside of the host\n",
1271                 ptr, data, flags);
1272             *ptr = data->host;
1273             data->host = NULL;
1274             return FALSE;
1275         }
1276     }
1277
1278     /* Don't overwrite our length if we found a port earlier. */
1279     if(!data->port)
1280         data->host_len = *ptr - data->host;
1281
1282     /* If the host is empty, then it's an unknown host type. */
1283     if(data->host_len == 0)
1284         data->host_type = Uri_HOST_UNKNOWN;
1285     else
1286         data->host_type = Uri_HOST_DNS;
1287
1288     TRACE("(%p %p %x): Parsed reg-name. host=%s len=%d\n", ptr, data, flags,
1289         debugstr_wn(data->host, data->host_len), data->host_len);
1290     return TRUE;
1291 }
1292
1293 /* Attempts to parse an IPv6 address out of the URI.
1294  *
1295  * IPv6address =                               6( h16 ":" ) ls32
1296  *                /                       "::" 5( h16 ":" ) ls32
1297  *                / [               h16 ] "::" 4( h16 ":" ) ls32
1298  *                / [ *1( h16 ":" ) h16 ] "::" 3( h16 ":" ) ls32
1299  *                / [ *2( h16 ":" ) h16 ] "::" 2( h16 ":" ) ls32
1300  *                / [ *3( h16 ":" ) h16 ] "::"    h16 ":"   ls32
1301  *                / [ *4( h16 ":" ) h16 ] "::"              ls32
1302  *                / [ *5( h16 ":" ) h16 ] "::"              h16
1303  *                / [ *6( h16 ":" ) h16 ] "::"
1304  *
1305  * ls32        = ( h16 ":" h16 ) / IPv4address
1306  *             ; least-significant 32 bits of address.
1307  *
1308  * h16         = 1*4HEXDIG
1309  *             ; 16 bits of address represented in hexadecimal.
1310  *
1311  * Modeled after google-url's 'DoParseIPv6' function.
1312  */
1313 static BOOL parse_ipv6address(const WCHAR **ptr, parse_data *data, DWORD flags) {
1314     const WCHAR *start, *cur_start;
1315     ipv6_address ip;
1316
1317     start = cur_start = *ptr;
1318     memset(&ip, 0, sizeof(ipv6_address));
1319
1320     for(;; ++(*ptr)) {
1321         /* Check if we're on the last character of the host. */
1322         BOOL is_end = (is_auth_delim(**ptr, data->scheme_type != URL_SCHEME_UNKNOWN)
1323                         || **ptr == ']');
1324
1325         BOOL is_split = (**ptr == ':');
1326         BOOL is_elision = (is_split && !is_end && *(*ptr+1) == ':');
1327
1328         /* Check if we're at the end of of the a component, or
1329          * if we're at the end of the IPv6 address.
1330          */
1331         if(is_split || is_end) {
1332             DWORD cur_len = 0;
1333
1334             cur_len = *ptr - cur_start;
1335
1336             /* h16 can't have a length > 4. */
1337             if(cur_len > 4) {
1338                 *ptr = start;
1339
1340                 TRACE("(%p %p %x): h16 component to long.\n",
1341                     ptr, data, flags);
1342                 return FALSE;
1343             }
1344
1345             if(cur_len == 0) {
1346                 /* An h16 component can't have the length of 0 unless
1347                  * the elision is at the beginning of the address, or
1348                  * at the end of the address.
1349                  */
1350                 if(!((*ptr == start && is_elision) ||
1351                     (is_end && (*ptr-2) == ip.elision))) {
1352                     *ptr = start;
1353                     TRACE("(%p %p %x): IPv6 component can not have a length of 0.\n",
1354                         ptr, data, flags);
1355                     return FALSE;
1356                 }
1357             }
1358
1359             if(cur_len > 0) {
1360                 /* An IPv6 address can have no more than 8 h16 components. */
1361                 if(ip.h16_count >= 8) {
1362                     *ptr = start;
1363                     TRACE("(%p %p %x): Not a IPv6 address, to many h16 components.\n",
1364                         ptr, data, flags);
1365                     return FALSE;
1366                 }
1367
1368                 ip.components[ip.h16_count].str = cur_start;
1369                 ip.components[ip.h16_count].len = cur_len;
1370
1371                 TRACE("(%p %p %x): Found h16 component %s, len=%d, h16_count=%d\n",
1372                     ptr, data, flags, debugstr_wn(cur_start, cur_len), cur_len,
1373                     ip.h16_count);
1374                 ++ip.h16_count;
1375             }
1376         }
1377
1378         if(is_end)
1379             break;
1380
1381         if(is_elision) {
1382             /* A IPv6 address can only have 1 elision ('::'). */
1383             if(ip.elision) {
1384                 *ptr = start;
1385
1386                 TRACE("(%p %p %x): IPv6 address cannot have 2 elisions.\n",
1387                     ptr, data, flags);
1388                 return FALSE;
1389             }
1390
1391             ip.elision = *ptr;
1392             ++(*ptr);
1393         }
1394
1395         if(is_split)
1396             cur_start = *ptr+1;
1397         else {
1398             if(!check_ipv4address(ptr, TRUE)) {
1399                 if(!is_hexdigit(**ptr)) {
1400                     /* Not a valid character for an IPv6 address. */
1401                     *ptr = start;
1402                     return FALSE;
1403                 }
1404             } else {
1405                 /* Found an IPv4 address. */
1406                 ip.ipv4 = cur_start;
1407                 ip.ipv4_len = *ptr - cur_start;
1408
1409                 TRACE("(%p %p %x): Found an attached IPv4 address %s len=%d.\n",
1410                     ptr, data, flags, debugstr_wn(ip.ipv4, ip.ipv4_len),
1411                     ip.ipv4_len);
1412
1413                 /* IPv4 addresses can only appear at the end of a IPv6. */
1414                 break;
1415             }
1416         }
1417     }
1418
1419     compute_ipv6_comps_size(&ip);
1420
1421     /* Make sure the IPv6 address adds up to 16 bytes. */
1422     if(ip.components_size + ip.elision_size != 16) {
1423         *ptr = start;
1424         TRACE("(%p %p %x): Invalid IPv6 address, did not add up to 16 bytes.\n",
1425             ptr, data, flags);
1426         return FALSE;
1427     }
1428
1429     if(ip.elision_size == 2) {
1430         /* For some reason on Windows if an elision that represents
1431          * only 1 h16 component is encountered at the very begin or
1432          * end of an IPv6 address, Windows does not consider it a
1433          * valid IPv6 address.
1434          *
1435          *  Ex: [::2:3:4:5:6:7] is not valid, even though the sum
1436          *      of all the components == 128bits.
1437          */
1438          if(ip.elision < ip.components[0].str ||
1439             ip.elision > ip.components[ip.h16_count-1].str) {
1440             *ptr = start;
1441             TRACE("(%p %p %x): Invalid IPv6 address. Detected elision of 2 bytes at the beginning or end of the address.\n",
1442                 ptr, data, flags);
1443             return FALSE;
1444         }
1445     }
1446
1447     data->host_type = Uri_HOST_IPV6;
1448     data->has_ipv6 = TRUE;
1449     data->ipv6_address = ip;
1450
1451     TRACE("(%p %p %x): Found valid IPv6 literal %s len=%d\n",
1452         ptr, data, flags, debugstr_wn(start, *ptr-start),
1453         *ptr-start);
1454     return TRUE;
1455 }
1456
1457 /*  IPvFuture  = "v" 1*HEXDIG "." 1*( unreserved / sub-delims / ":" ) */
1458 static BOOL parse_ipvfuture(const WCHAR **ptr, parse_data *data, DWORD flags) {
1459     const WCHAR *start = *ptr;
1460
1461     /* IPvFuture has to start with a 'v' or 'V'. */
1462     if(**ptr != 'v' && **ptr != 'V')
1463         return FALSE;
1464
1465     /* Following the v their must be atleast 1 hexdigit. */
1466     ++(*ptr);
1467     if(!is_hexdigit(**ptr)) {
1468         *ptr = start;
1469         return FALSE;
1470     }
1471
1472     ++(*ptr);
1473     while(is_hexdigit(**ptr))
1474         ++(*ptr);
1475
1476     /* End of the hexdigit sequence must be a '.' */
1477     if(**ptr != '.') {
1478         *ptr = start;
1479         return FALSE;
1480     }
1481
1482     ++(*ptr);
1483     if(!is_unreserved(**ptr) && !is_subdelim(**ptr) && **ptr != ':') {
1484         *ptr = start;
1485         return FALSE;
1486     }
1487
1488     ++(*ptr);
1489     while(is_unreserved(**ptr) || is_subdelim(**ptr) || **ptr == ':')
1490         ++(*ptr);
1491
1492     data->host_type = Uri_HOST_UNKNOWN;
1493
1494     TRACE("(%p %p %x): Parsed IPvFuture address %s len=%d\n", ptr, data, flags,
1495         debugstr_wn(start, *ptr-start), *ptr-start);
1496
1497     return TRUE;
1498 }
1499
1500 /* IP-literal = "[" ( IPv6address / IPvFuture  ) "]" */
1501 static BOOL parse_ip_literal(const WCHAR **ptr, parse_data *data, DWORD flags) {
1502     data->host = *ptr;
1503
1504     if(**ptr != '[') {
1505         data->host = NULL;
1506         return FALSE;
1507     }
1508
1509     ++(*ptr);
1510     if(!parse_ipv6address(ptr, data, flags)) {
1511         if(!parse_ipvfuture(ptr, data, flags)) {
1512             *ptr = data->host;
1513             data->host = NULL;
1514             return FALSE;
1515         }
1516     }
1517
1518     if(**ptr != ']') {
1519         *ptr = data->host;
1520         data->host = NULL;
1521         return FALSE;
1522     }
1523
1524     ++(*ptr);
1525     if(**ptr == ':') {
1526         ++(*ptr);
1527         /* If a valid port is not found, then let it trickle down to
1528          * parse_reg_name.
1529          */
1530         if(!parse_port(ptr, data, flags)) {
1531             *ptr = data->host;
1532             data->host = NULL;
1533             return FALSE;
1534         }
1535     } else
1536         data->host_len = *ptr - data->host;
1537
1538     return TRUE;
1539 }
1540
1541 /* Parses the host information from the URI.
1542  *
1543  * host = IP-literal / IPv4address / reg-name
1544  */
1545 static BOOL parse_host(const WCHAR **ptr, parse_data *data, DWORD flags) {
1546     if(!parse_ip_literal(ptr, data, flags)) {
1547         if(!parse_ipv4address(ptr, data, flags)) {
1548             if(!parse_reg_name(ptr, data, flags)) {
1549                 TRACE("(%p %p %x): Malformed URI, Unknown host type.\n",
1550                     ptr, data, flags);
1551                 return FALSE;
1552             }
1553         }
1554     }
1555
1556     return TRUE;
1557 }
1558
1559 /* Parses the authority information from the URI.
1560  *
1561  * authority   = [ userinfo "@" ] host [ ":" port ]
1562  */
1563 static BOOL parse_authority(const WCHAR **ptr, parse_data *data, DWORD flags) {
1564     parse_userinfo(ptr, data, flags);
1565
1566     /* Parsing the port will happen during one of the host parsing
1567      * routines (if the URI has a port).
1568      */
1569     if(!parse_host(ptr, data, flags))
1570         return FALSE;
1571
1572     return TRUE;
1573 }
1574
1575 /* Attempts to parse the path information of a hierarchical URI. */
1576 static BOOL parse_path_hierarchical(const WCHAR **ptr, parse_data *data, DWORD flags) {
1577     const WCHAR *start = *ptr;
1578     static const WCHAR slash[] = {'/',0};
1579
1580     if(is_path_delim(**ptr)) {
1581         if(data->scheme_type == URL_SCHEME_WILDCARD) {
1582             /* Wildcard schemes don't get a '/' attached if their path is
1583              * empty.
1584              */
1585             data->path = NULL;
1586             data->path_len = 0;
1587         } else if(!(flags & Uri_CREATE_NO_CANONICALIZE)) {
1588             /* If the path component is empty, then a '/' is added. */
1589             data->path = slash;
1590             data->path_len = 1;
1591         }
1592     } else {
1593         while(!is_path_delim(**ptr)) {
1594             if(**ptr == '%' && data->scheme_type != URL_SCHEME_UNKNOWN &&
1595                data->scheme_type != URL_SCHEME_FILE) {
1596                 if(!check_pct_encoded(ptr)) {
1597                     *ptr = start;
1598                     return FALSE;
1599                 } else
1600                     continue;
1601             } else if(**ptr == '\\') {
1602                 /* Not allowed to have a backslash if NO_CANONICALIZE is set
1603                  * and the scheme is known type (but not a file scheme).
1604                  */
1605                 if(flags & Uri_CREATE_NO_CANONICALIZE) {
1606                     if(data->scheme_type != URL_SCHEME_FILE &&
1607                        data->scheme_type != URL_SCHEME_UNKNOWN) {
1608                         *ptr = start;
1609                         return FALSE;
1610                     }
1611                 }
1612             }
1613
1614             ++(*ptr);
1615         }
1616
1617         /* The only time a URI doesn't have a path is when
1618          * the NO_CANONICALIZE flag is set and the raw URI
1619          * didn't contain one.
1620          */
1621         if(*ptr == start) {
1622             data->path = NULL;
1623             data->path_len = 0;
1624         } else {
1625             data->path = start;
1626             data->path_len = *ptr - start;
1627         }
1628     }
1629
1630     if(data->path)
1631         TRACE("(%p %p %x): Parsed path %s len=%d\n", ptr, data, flags,
1632             debugstr_wn(data->path, data->path_len), data->path_len);
1633     else
1634         TRACE("(%p %p %x): The URI contained no path\n", ptr, data, flags);
1635
1636     return TRUE;
1637 }
1638
1639 /* Parses the path of a opaque URI (much less strict then the parser
1640  * for a hierarchical URI).
1641  *
1642  * NOTE:
1643  *  Windows allows invalid % encoded data to appear in opaque URI paths
1644  *  for unknown scheme types.
1645  */
1646 static BOOL parse_path_opaque(const WCHAR **ptr, parse_data *data, DWORD flags) {
1647     const BOOL known_scheme = data->scheme_type != URL_SCHEME_UNKNOWN;
1648
1649     data->path = *ptr;
1650
1651     while(!is_path_delim(**ptr)) {
1652         if(**ptr == '%' && known_scheme) {
1653             if(!check_pct_encoded(ptr)) {
1654                 *ptr = data->path;
1655                 data->path = NULL;
1656                 return FALSE;
1657             } else
1658                 continue;
1659         }
1660
1661         ++(*ptr);
1662     }
1663
1664     data->path_len = *ptr - data->path;
1665     TRACE("(%p %p %x): Parsed opaque URI path %s len=%d\n", ptr, data, flags,
1666         debugstr_wn(data->path, data->path_len), data->path_len);
1667     return TRUE;
1668 }
1669
1670 /* Determines how the URI should be parsed after the scheme information.
1671  *
1672  * If the scheme is followed, by "//" then, it is treated as an hierarchical URI
1673  * which then the authority and path information will be parsed out. Otherwise, the
1674  * URI will be treated as an opaque URI which the authority information is not parsed
1675  * out.
1676  *
1677  * RFC 3896 definition of hier-part:
1678  *
1679  * hier-part   = "//" authority path-abempty
1680  *                 / path-absolute
1681  *                 / path-rootless
1682  *                 / path-empty
1683  *
1684  * MSDN opaque URI definition:
1685  *  scheme ":" path [ "#" fragment ]
1686  *
1687  * NOTES:
1688  *  If the URI is of an unknown scheme type and has a "//" following the scheme then it
1689  *  is treated as a hierarchical URI, but, if the CREATE_NO_CRACK_UNKNOWN_SCHEMES flag is
1690  *  set then it is considered an opaque URI reguardless of what follows the scheme information
1691  *  (per MSDN documentation).
1692  */
1693 static BOOL parse_hierpart(const WCHAR **ptr, parse_data *data, DWORD flags) {
1694     const WCHAR *start = *ptr;
1695
1696     /* Checks if the authority information needs to be parsed.
1697      *
1698      * Relative URI's aren't hierarchical URI's, but, they could trick
1699      * "check_hierarchical" into thinking it is, so we need to explicitly
1700      * make sure it's not relative. Also, if the URI is an implicit file
1701      * scheme it might not contain a "//", but, it's considered hierarchical
1702      * anyways. Wildcard Schemes are always considered hierarchical
1703      */
1704     if(data->scheme_type == URL_SCHEME_WILDCARD ||
1705        data->scheme_type == URL_SCHEME_FILE ||
1706        (!data->is_relative && check_hierarchical(ptr))) {
1707         /* Only treat it as a hierarchical URI if the scheme_type is known or
1708          * the Uri_CREATE_NO_CRACK_UNKNOWN_SCHEMES flag is not set.
1709          */
1710         if(data->scheme_type != URL_SCHEME_UNKNOWN ||
1711            !(flags & Uri_CREATE_NO_CRACK_UNKNOWN_SCHEMES)) {
1712             TRACE("(%p %p %x): Treating URI as an hierarchical URI.\n", ptr, data, flags);
1713             data->is_opaque = FALSE;
1714
1715             if(data->scheme_type == URL_SCHEME_FILE)
1716                 /* Skip past the "//" after the scheme (if any). */
1717                 check_hierarchical(ptr);
1718
1719             /* TODO: Handle hierarchical URI's, parse authority then parse the path. */
1720             if(!parse_authority(ptr, data, flags))
1721                 return FALSE;
1722
1723             return parse_path_hierarchical(ptr, data, flags);
1724         } else
1725             /* Reset ptr to it's starting position so opaque path parsing
1726              * begins at the correct location.
1727              */
1728             *ptr = start;
1729     }
1730
1731     /* If it reaches here, then the URI will be treated as an opaque
1732      * URI.
1733      */
1734
1735     TRACE("(%p %p %x): Treating URI as an opaque URI.\n", ptr, data, flags);
1736
1737     data->is_opaque = TRUE;
1738     if(!parse_path_opaque(ptr, data, flags))
1739         return FALSE;
1740
1741     return TRUE;
1742 }
1743
1744 /* Attempts to parse the query string from the URI.
1745  *
1746  * NOTES:
1747  *  If NO_DECODE_EXTRA_INFO flag is set, then invalid percent encoded
1748  *  data is allowed appear in the query string. For unknown scheme types
1749  *  invalid percent encoded data is allowed to appear reguardless.
1750  */
1751 static BOOL parse_query(const WCHAR **ptr, parse_data *data, DWORD flags) {
1752     const BOOL known_scheme = data->scheme_type != URL_SCHEME_UNKNOWN;
1753
1754     if(**ptr != '?') {
1755         TRACE("(%p %p %x): URI didn't contain a query string.\n", ptr, data, flags);
1756         return TRUE;
1757     }
1758
1759     data->query = *ptr;
1760
1761     ++(*ptr);
1762     while(**ptr && **ptr != '#') {
1763         if(**ptr == '%' && known_scheme &&
1764            !(flags & Uri_CREATE_NO_DECODE_EXTRA_INFO)) {
1765             if(!check_pct_encoded(ptr)) {
1766                 *ptr = data->query;
1767                 data->query = NULL;
1768                 return FALSE;
1769             } else
1770                 continue;
1771         }
1772
1773         ++(*ptr);
1774     }
1775
1776     data->query_len = *ptr - data->query;
1777
1778     TRACE("(%p %p %x): Parsed query string %s len=%d\n", ptr, data, flags,
1779         debugstr_wn(data->query, data->query_len), data->query_len);
1780     return TRUE;
1781 }
1782
1783 /* Parses and validates the components of the specified by data->uri
1784  * and stores the information it parses into 'data'.
1785  *
1786  * Returns TRUE if it successfully parsed the URI. False otherwise.
1787  */
1788 static BOOL parse_uri(parse_data *data, DWORD flags) {
1789     const WCHAR *ptr;
1790     const WCHAR **pptr;
1791
1792     ptr = data->uri;
1793     pptr = &ptr;
1794
1795     TRACE("(%p %x): BEGINNING TO PARSE URI %s.\n", data, flags, debugstr_w(data->uri));
1796
1797     if(!parse_scheme(pptr, data, flags))
1798         return FALSE;
1799
1800     if(!parse_hierpart(pptr, data, flags))
1801         return FALSE;
1802
1803     if(!parse_query(pptr, data, flags))
1804         return FALSE;
1805
1806     /* TODO: Parse fragment (if the URI has one). */
1807
1808     TRACE("(%p %x): FINISHED PARSING URI.\n", data, flags);
1809     return TRUE;
1810 }
1811
1812 /* Canonicalizes the userinfo of the URI represented by the parse_data.
1813  *
1814  * Canonicalization of the userinfo is a simple process. If there are any percent
1815  * encoded characters that fall in the "unreserved" character set, they are decoded
1816  * to their actual value. If a character is not in the "unreserved" or "reserved" sets
1817  * then it is percent encoded. Other than that the characters are copied over without
1818  * change.
1819  */
1820 static BOOL canonicalize_userinfo(const parse_data *data, Uri *uri, DWORD flags, BOOL computeOnly) {
1821     DWORD i = 0;
1822
1823     uri->userinfo_start = uri->userinfo_split = -1;
1824     uri->userinfo_len = 0;
1825
1826     if(!data->userinfo)
1827         /* URI doesn't have userinfo, so nothing to do here. */
1828         return TRUE;
1829
1830     uri->userinfo_start = uri->canon_len;
1831
1832     while(i < data->userinfo_len) {
1833         if(data->userinfo[i] == ':' && uri->userinfo_split == -1)
1834             /* Windows only considers the first ':' as the delimiter. */
1835             uri->userinfo_split = uri->canon_len - uri->userinfo_start;
1836         else if(data->userinfo[i] == '%') {
1837             /* Only decode % encoded values for known scheme types. */
1838             if(data->scheme_type != URL_SCHEME_UNKNOWN) {
1839                 /* See if the value really needs decoded. */
1840                 WCHAR val = decode_pct_val(data->userinfo + i);
1841                 if(is_unreserved(val)) {
1842                     if(!computeOnly)
1843                         uri->canon_uri[uri->canon_len] = val;
1844
1845                     ++uri->canon_len;
1846
1847                     /* Move pass the hex characters. */
1848                     i += 3;
1849                     continue;
1850                 }
1851             }
1852         } else if(!is_reserved(data->userinfo[i]) && !is_unreserved(data->userinfo[i]) &&
1853                   data->userinfo[i] != '\\') {
1854             /* Only percent encode forbidden characters if the NO_ENCODE_FORBIDDEN_CHARACTERS flag
1855              * is NOT set.
1856              */
1857             if(!(flags & Uri_CREATE_NO_ENCODE_FORBIDDEN_CHARACTERS)) {
1858                 if(!computeOnly)
1859                     pct_encode_val(data->userinfo[i], uri->canon_uri + uri->canon_len);
1860
1861                 uri->canon_len += 3;
1862                 ++i;
1863                 continue;
1864             }
1865         }
1866
1867         if(!computeOnly)
1868             /* Nothing special, so just copy the character over. */
1869             uri->canon_uri[uri->canon_len] = data->userinfo[i];
1870
1871         ++uri->canon_len;
1872         ++i;
1873     }
1874
1875     uri->userinfo_len = uri->canon_len - uri->userinfo_start;
1876     if(!computeOnly)
1877         TRACE("(%p %p %x %d): Canonicalized userinfo, userinfo_start=%d, userinfo=%s, userinfo_split=%d userinfo_len=%d.\n",
1878                 data, uri, flags, computeOnly, uri->userinfo_start, debugstr_wn(uri->canon_uri + uri->userinfo_start, uri->userinfo_len),
1879                 uri->userinfo_split, uri->userinfo_len);
1880
1881     /* Now insert the '@' after the userinfo. */
1882     if(!computeOnly)
1883         uri->canon_uri[uri->canon_len] = '@';
1884
1885     ++uri->canon_len;
1886     return TRUE;
1887 }
1888
1889 /* Attempts to canonicalize a reg_name.
1890  *
1891  * Things that happen:
1892  *  1)  If Uri_CREATE_NO_CANONICALIZE flag is not set, then the reg_name is
1893  *      lower cased. Unless it's an unknown scheme type, which case it's
1894  *      no lower cased reguardless.
1895  *
1896  *  2)  Unreserved % encoded characters are decoded for known
1897  *      scheme types.
1898  *
1899  *  3)  Forbidden characters are % encoded as long as
1900  *      Uri_CREATE_NO_ENCODE_FORBIDDEN_CHARACTERS flag is not set and
1901  *      it isn't an unknown scheme type.
1902  *
1903  *  4)  If it's a file scheme and the host is "localhost" it's removed.
1904  */
1905 static BOOL canonicalize_reg_name(const parse_data *data, Uri *uri,
1906                                   DWORD flags, BOOL computeOnly) {
1907     static const WCHAR localhostW[] =
1908             {'l','o','c','a','l','h','o','s','t',0};
1909     const WCHAR *ptr;
1910     const BOOL known_scheme = data->scheme_type != URL_SCHEME_UNKNOWN;
1911
1912     uri->host_start = uri->canon_len;
1913
1914     if(data->scheme_type == URL_SCHEME_FILE &&
1915        data->host_len == lstrlenW(localhostW)) {
1916         if(!StrCmpNIW(data->host, localhostW, data->host_len)) {
1917             uri->host_start = -1;
1918             uri->host_len = 0;
1919             uri->host_type = Uri_HOST_UNKNOWN;
1920             return TRUE;
1921         }
1922     }
1923
1924     for(ptr = data->host; ptr < data->host+data->host_len; ++ptr) {
1925         if(*ptr == '%' && known_scheme) {
1926             WCHAR val = decode_pct_val(ptr);
1927             if(is_unreserved(val)) {
1928                 /* If NO_CANONICALZE is not set, then windows lower cases the
1929                  * decoded value.
1930                  */
1931                 if(!(flags & Uri_CREATE_NO_CANONICALIZE) && isupperW(val)) {
1932                     if(!computeOnly)
1933                         uri->canon_uri[uri->canon_len] = tolowerW(val);
1934                 } else {
1935                     if(!computeOnly)
1936                         uri->canon_uri[uri->canon_len] = val;
1937                 }
1938                 ++uri->canon_len;
1939
1940                 /* Skip past the % encoded character. */
1941                 ptr += 2;
1942                 continue;
1943             } else {
1944                 /* Just copy the % over. */
1945                 if(!computeOnly)
1946                     uri->canon_uri[uri->canon_len] = *ptr;
1947                 ++uri->canon_len;
1948             }
1949         } else if(*ptr == '\\') {
1950             /* Only unknown scheme types could have made it here with a '\\' in the host name. */
1951             if(!computeOnly)
1952                 uri->canon_uri[uri->canon_len] = *ptr;
1953             ++uri->canon_len;
1954         } else if(!(flags & Uri_CREATE_NO_ENCODE_FORBIDDEN_CHARACTERS) &&
1955                   !is_unreserved(*ptr) && !is_reserved(*ptr) && known_scheme) {
1956             if(!computeOnly) {
1957                 pct_encode_val(*ptr, uri->canon_uri+uri->canon_len);
1958
1959                 /* The percent encoded value gets lower cased also. */
1960                 if(!(flags & Uri_CREATE_NO_CANONICALIZE)) {
1961                     uri->canon_uri[uri->canon_len+1] = tolowerW(uri->canon_uri[uri->canon_len+1]);
1962                     uri->canon_uri[uri->canon_len+2] = tolowerW(uri->canon_uri[uri->canon_len+2]);
1963                 }
1964             }
1965
1966             uri->canon_len += 3;
1967         } else {
1968             if(!computeOnly) {
1969                 if(!(flags & Uri_CREATE_NO_CANONICALIZE) && known_scheme)
1970                     uri->canon_uri[uri->canon_len] = tolowerW(*ptr);
1971                 else
1972                     uri->canon_uri[uri->canon_len] = *ptr;
1973             }
1974
1975             ++uri->canon_len;
1976         }
1977     }
1978
1979     uri->host_len = uri->canon_len - uri->host_start;
1980
1981     if(!computeOnly)
1982         TRACE("(%p %p %x %d): Canonicalize reg_name=%s len=%d\n", data, uri, flags,
1983             computeOnly, debugstr_wn(uri->canon_uri+uri->host_start, uri->host_len),
1984             uri->host_len);
1985
1986     if(!computeOnly)
1987         find_domain_name(uri->canon_uri+uri->host_start, uri->host_len,
1988             &(uri->domain_offset));
1989
1990     return TRUE;
1991 }
1992
1993 /* Attempts to canonicalize an implicit IPv4 address. */
1994 static BOOL canonicalize_implicit_ipv4address(const parse_data *data, Uri *uri, DWORD flags, BOOL computeOnly) {
1995     uri->host_start = uri->canon_len;
1996
1997     TRACE("%u\n", data->implicit_ipv4);
1998     /* For unknown scheme types Window's doesn't convert
1999      * the value into an IP address, but, it still considers
2000      * it an IPv4 address.
2001      */
2002     if(data->scheme_type == URL_SCHEME_UNKNOWN) {
2003         if(!computeOnly)
2004             memcpy(uri->canon_uri+uri->canon_len, data->host, data->host_len*sizeof(WCHAR));
2005         uri->canon_len += data->host_len;
2006     } else {
2007         if(!computeOnly)
2008             uri->canon_len += ui2ipv4(uri->canon_uri+uri->canon_len, data->implicit_ipv4);
2009         else
2010             uri->canon_len += ui2ipv4(NULL, data->implicit_ipv4);
2011     }
2012
2013     uri->host_len = uri->canon_len - uri->host_start;
2014     uri->host_type = Uri_HOST_IPV4;
2015
2016     if(!computeOnly)
2017         TRACE("%p %p %x %d): Canonicalized implicit IP address=%s len=%d\n",
2018             data, uri, flags, computeOnly,
2019             debugstr_wn(uri->canon_uri+uri->host_start, uri->host_len),
2020             uri->host_len);
2021
2022     return TRUE;
2023 }
2024
2025 /* Attempts to canonicalize an IPv4 address.
2026  *
2027  * If the parse_data represents a URI that has an implicit IPv4 address
2028  * (ex. http://256/, this function will convert 256 into 0.0.1.0). If
2029  * the implicit IP address exceeds the value of UINT_MAX (maximum value
2030  * for an IPv4 address) it's canonicalized as if were a reg-name.
2031  *
2032  * If the parse_data contains a partial or full IPv4 address it normalizes it.
2033  * A partial IPv4 address is something like "192.0" and would be normalized to
2034  * "192.0.0.0". With a full (or partial) IPv4 address like "192.002.01.003" would
2035  * be normalized to "192.2.1.3".
2036  *
2037  * NOTES:
2038  *  Window's ONLY normalizes IPv4 address for known scheme types (one that isn't
2039  *  URL_SCHEME_UNKNOWN). For unknown scheme types, it simply copies the data from
2040  *  the original URI into the canonicalized URI, but, it still recognizes URI's
2041  *  host type as HOST_IPV4.
2042  */
2043 static BOOL canonicalize_ipv4address(const parse_data *data, Uri *uri, DWORD flags, BOOL computeOnly) {
2044     if(data->has_implicit_ip)
2045         return canonicalize_implicit_ipv4address(data, uri, flags, computeOnly);
2046     else {
2047         uri->host_start = uri->canon_len;
2048
2049         /* Windows only normalizes for known scheme types. */
2050         if(data->scheme_type != URL_SCHEME_UNKNOWN) {
2051             /* parse_data contains a partial or full IPv4 address, so normalize it. */
2052             DWORD i, octetDigitCount = 0, octetCount = 0;
2053             BOOL octetHasDigit = FALSE;
2054
2055             for(i = 0; i < data->host_len; ++i) {
2056                 if(data->host[i] == '0' && !octetHasDigit) {
2057                     /* Can ignore leading zeros if:
2058                      *  1) It isn't the last digit of the octet.
2059                      *  2) i+1 != data->host_len
2060                      *  3) i+1 != '.'
2061                      */
2062                     if(octetDigitCount == 2 ||
2063                        i+1 == data->host_len ||
2064                        data->host[i+1] == '.') {
2065                         if(!computeOnly)
2066                             uri->canon_uri[uri->canon_len] = data->host[i];
2067                         ++uri->canon_len;
2068                         TRACE("Adding zero\n");
2069                     }
2070                 } else if(data->host[i] == '.') {
2071                     if(!computeOnly)
2072                         uri->canon_uri[uri->canon_len] = data->host[i];
2073                     ++uri->canon_len;
2074
2075                     octetDigitCount = 0;
2076                     octetHasDigit = FALSE;
2077                     ++octetCount;
2078                 } else {
2079                     if(!computeOnly)
2080                         uri->canon_uri[uri->canon_len] = data->host[i];
2081                     ++uri->canon_len;
2082
2083                     ++octetDigitCount;
2084                     octetHasDigit = TRUE;
2085                 }
2086             }
2087
2088             /* Make sure the canonicalized IP address has 4 dec-octets.
2089              * If doesn't add "0" ones until there is 4;
2090              */
2091             for( ; octetCount < 3; ++octetCount) {
2092                 if(!computeOnly) {
2093                     uri->canon_uri[uri->canon_len] = '.';
2094                     uri->canon_uri[uri->canon_len+1] = '0';
2095                 }
2096
2097                 uri->canon_len += 2;
2098             }
2099         } else {
2100             /* Windows doesn't normalize addresses in unknown schemes. */
2101             if(!computeOnly)
2102                 memcpy(uri->canon_uri+uri->canon_len, data->host, data->host_len*sizeof(WCHAR));
2103             uri->canon_len += data->host_len;
2104         }
2105
2106         uri->host_len = uri->canon_len - uri->host_start;
2107         if(!computeOnly)
2108             TRACE("(%p %p %x %d): Canonicalized IPv4 address, ip=%s len=%d\n",
2109                 data, uri, flags, computeOnly,
2110                 debugstr_wn(uri->canon_uri+uri->host_start, uri->host_len),
2111                 uri->host_len);
2112     }
2113
2114     return TRUE;
2115 }
2116
2117 /* Attempts to canonicalize the IPv6 address of the URI.
2118  *
2119  * Multiple things happen during the canonicalization of an IPv6 address:
2120  *  1)  Any leading zero's in an h16 component are removed.
2121  *      Ex: [0001:0022::] -> [1:22::]
2122  *
2123  *  2)  The longest sequence of zero h16 components are compressed
2124  *      into a "::" (elision). If there's a tie, the first is choosen.
2125  *
2126  *      Ex: [0:0:0:0:1:6:7:8]   -> [::1:6:7:8]
2127  *          [0:0:0:0:1:2::]     -> [::1:2:0:0]
2128  *          [0:0:1:2:0:0:7:8]   -> [::1:2:0:0:7:8]
2129  *
2130  *  3)  If an IPv4 address is attached to the IPv6 address, it's
2131  *      also normalized.
2132  *      Ex: [::001.002.022.000] -> [::1.2.22.0]
2133  *
2134  *  4)  If an elision is present, but, only represents 1 h16 component
2135  *      it's expanded.
2136  *
2137  *      Ex: [1::2:3:4:5:6:7] -> [1:0:2:3:4:5:6:7]
2138  *
2139  *  5)  If the IPv6 address contains an IPv4 address and there exists
2140  *      at least 1 non-zero h16 component the IPv4 address is converted
2141  *      into two h16 components, otherwise it's normalized and kept as is.
2142  *
2143  *      Ex: [::192.200.003.4]       -> [::192.200.3.4]
2144  *          [ffff::192.200.003.4]   -> [ffff::c0c8:3041]
2145  *
2146  * NOTE:
2147  *  For unknown scheme types Windows simply copies the address over without any
2148  *  changes.
2149  *
2150  *  IPv4 address can be included in an elision if all its components are 0's.
2151  */
2152 static BOOL canonicalize_ipv6address(const parse_data *data, Uri *uri,
2153                                      DWORD flags, BOOL computeOnly) {
2154     uri->host_start = uri->canon_len;
2155
2156     if(data->scheme_type == URL_SCHEME_UNKNOWN) {
2157         if(!computeOnly)
2158             memcpy(uri->canon_uri+uri->canon_len, data->host, data->host_len*sizeof(WCHAR));
2159         uri->canon_len += data->host_len;
2160     } else {
2161         USHORT values[8];
2162         INT elision_start;
2163         DWORD i, elision_len;
2164
2165         if(!ipv6_to_number(&(data->ipv6_address), values)) {
2166             TRACE("(%p %p %x %d): Failed to compute numerical value for IPv6 address.\n",
2167                 data, uri, flags, computeOnly);
2168             return FALSE;
2169         }
2170
2171         if(!computeOnly)
2172             uri->canon_uri[uri->canon_len] = '[';
2173         ++uri->canon_len;
2174
2175         /* Find where the elision should occur (if any). */
2176         compute_elision_location(&(data->ipv6_address), values, &elision_start, &elision_len);
2177
2178         TRACE("%p %p %x %d): Elision starts at %d, len=%u\n", data, uri, flags,
2179             computeOnly, elision_start, elision_len);
2180
2181         for(i = 0; i < 8; ++i) {
2182             BOOL in_elision = (elision_start > -1 && i >= elision_start &&
2183                                i < elision_start+elision_len);
2184             BOOL do_ipv4 = (i == 6 && data->ipv6_address.ipv4 && !in_elision &&
2185                             data->ipv6_address.h16_count == 0);
2186
2187             if(i == elision_start) {
2188                 if(!computeOnly) {
2189                     uri->canon_uri[uri->canon_len] = ':';
2190                     uri->canon_uri[uri->canon_len+1] = ':';
2191                 }
2192                 uri->canon_len += 2;
2193             }
2194
2195             /* We can ignore the current component if we're in the elision. */
2196             if(in_elision)
2197                 continue;
2198
2199             /* We only add a ':' if we're not at i == 0, or when we're at
2200              * the very end of elision range since the ':' colon was handled
2201              * earlier. Otherwise we would end up with ":::" after elision.
2202              */
2203             if(i != 0 && !(elision_start > -1 && i == elision_start+elision_len)) {
2204                 if(!computeOnly)
2205                     uri->canon_uri[uri->canon_len] = ':';
2206                 ++uri->canon_len;
2207             }
2208
2209             if(do_ipv4) {
2210                 UINT val;
2211                 DWORD len;
2212
2213                 /* Combine the two parts of the IPv4 address values. */
2214                 val = values[i];
2215                 val <<= 16;
2216                 val += values[i+1];
2217
2218                 if(!computeOnly)
2219                     len = ui2ipv4(uri->canon_uri+uri->canon_len, val);
2220                 else
2221                     len = ui2ipv4(NULL, val);
2222
2223                 uri->canon_len += len;
2224                 ++i;
2225             } else {
2226                 /* Write a regular h16 component to the URI. */
2227
2228                 /* Short circuit for the trivial case. */
2229                 if(values[i] == 0) {
2230                     if(!computeOnly)
2231                         uri->canon_uri[uri->canon_len] = '0';
2232                     ++uri->canon_len;
2233                 } else {
2234                     static const WCHAR formatW[] = {'%','x',0};
2235
2236                     if(!computeOnly)
2237                         uri->canon_len += sprintfW(uri->canon_uri+uri->canon_len,
2238                                             formatW, values[i]);
2239                     else {
2240                         WCHAR tmp[5];
2241                         uri->canon_len += sprintfW(tmp, formatW, values[i]);
2242                     }
2243                 }
2244             }
2245         }
2246
2247         /* Add the closing ']'. */
2248         if(!computeOnly)
2249             uri->canon_uri[uri->canon_len] = ']';
2250         ++uri->canon_len;
2251     }
2252
2253     uri->host_len = uri->canon_len - uri->host_start;
2254
2255     if(!computeOnly)
2256         TRACE("(%p %p %x %d): Canonicalized IPv6 address %s, len=%d\n", data, uri, flags,
2257             computeOnly, debugstr_wn(uri->canon_uri+uri->host_start, uri->host_len),
2258             uri->host_len);
2259
2260     return TRUE;
2261 }
2262
2263 /* Attempts to canonicalize the host of the URI (if any). */
2264 static BOOL canonicalize_host(const parse_data *data, Uri *uri, DWORD flags, BOOL computeOnly) {
2265     uri->host_start = -1;
2266     uri->host_len = 0;
2267     uri->domain_offset = -1;
2268
2269     if(data->host) {
2270         switch(data->host_type) {
2271         case Uri_HOST_DNS:
2272             uri->host_type = Uri_HOST_DNS;
2273             if(!canonicalize_reg_name(data, uri, flags, computeOnly))
2274                 return FALSE;
2275
2276             break;
2277         case Uri_HOST_IPV4:
2278             uri->host_type = Uri_HOST_IPV4;
2279             if(!canonicalize_ipv4address(data, uri, flags, computeOnly))
2280                 return FALSE;
2281
2282             break;
2283         case Uri_HOST_IPV6:
2284             if(!canonicalize_ipv6address(data, uri, flags, computeOnly))
2285                 return FALSE;
2286
2287             uri->host_type = Uri_HOST_IPV6;
2288             break;
2289         case Uri_HOST_UNKNOWN:
2290             if(data->host_len > 0 || data->scheme_type != URL_SCHEME_FILE) {
2291                 uri->host_start = uri->canon_len;
2292
2293                 /* Nothing happens to unknown host types. */
2294                 if(!computeOnly)
2295                     memcpy(uri->canon_uri+uri->canon_len, data->host, data->host_len*sizeof(WCHAR));
2296                 uri->canon_len += data->host_len;
2297                 uri->host_len = data->host_len;
2298             }
2299
2300             uri->host_type = Uri_HOST_UNKNOWN;
2301             break;
2302         default:
2303             FIXME("(%p %p %x %d): Canonicalization for host type %d not supported.\n", data,
2304                     uri, flags, computeOnly, data->host_type);
2305             return FALSE;
2306        }
2307    }
2308
2309    return TRUE;
2310 }
2311
2312 static BOOL canonicalize_port(const parse_data *data, Uri *uri, DWORD flags, BOOL computeOnly) {
2313     BOOL has_default_port = FALSE;
2314     USHORT default_port = 0;
2315     DWORD i;
2316
2317     uri->has_port = FALSE;
2318
2319     /* Check if the scheme has a default port. */
2320     for(i = 0; i < sizeof(default_ports)/sizeof(default_ports[0]); ++i) {
2321         if(default_ports[i].scheme == data->scheme_type) {
2322             has_default_port = TRUE;
2323             default_port = default_ports[i].port;
2324             break;
2325         }
2326     }
2327
2328     if(data->port || has_default_port)
2329         uri->has_port = TRUE;
2330
2331     /* Possible cases:
2332      *  1)  Has a port which is the default port.
2333      *  2)  Has a port (not the default).
2334      *  3)  Doesn't have a port, but, scheme has a default port.
2335      *  4)  No port.
2336      */
2337     if(has_default_port && data->port && data->port_value == default_port) {
2338         /* If it's the default port and this flag isn't set, don't do anything. */
2339         if(flags & Uri_CREATE_NO_CANONICALIZE) {
2340             /* Copy the original port over. */
2341             if(!computeOnly) {
2342                 uri->canon_uri[uri->canon_len] = ':';
2343                 memcpy(uri->canon_uri+uri->canon_len+1, data->port, data->port_len*sizeof(WCHAR));
2344             }
2345             uri->canon_len += data->port_len+1;
2346         }
2347
2348         uri->port = default_port;
2349     } else if(data->port) {
2350         if(!computeOnly)
2351             uri->canon_uri[uri->canon_len] = ':';
2352         ++uri->canon_len;
2353
2354         if(flags & Uri_CREATE_NO_CANONICALIZE) {
2355             /* Copy the original over without changes. */
2356             if(!computeOnly)
2357                 memcpy(uri->canon_uri+uri->canon_len, data->port, data->port_len*sizeof(WCHAR));
2358             uri->canon_len += data->port_len;
2359         } else {
2360             const WCHAR formatW[] = {'%','u',0};
2361             INT len = 0;
2362             if(!computeOnly)
2363                 len = sprintfW(uri->canon_uri+uri->canon_len, formatW, data->port_value);
2364             else {
2365                 WCHAR tmp[6];
2366                 len = sprintfW(tmp, formatW, data->port_value);
2367             }
2368             uri->canon_len += len;
2369         }
2370
2371         uri->port = data->port_value;
2372     } else if(has_default_port)
2373         uri->port = default_port;
2374
2375     return TRUE;
2376 }
2377
2378 /* Canonicalizes the authority of the URI represented by the parse_data. */
2379 static BOOL canonicalize_authority(const parse_data *data, Uri *uri, DWORD flags, BOOL computeOnly) {
2380     uri->authority_start = uri->canon_len;
2381     uri->authority_len = 0;
2382
2383     if(!canonicalize_userinfo(data, uri, flags, computeOnly))
2384         return FALSE;
2385
2386     if(!canonicalize_host(data, uri, flags, computeOnly))
2387         return FALSE;
2388
2389     if(!canonicalize_port(data, uri, flags, computeOnly))
2390         return FALSE;
2391
2392     if(uri->host_start != -1)
2393         uri->authority_len = uri->canon_len - uri->authority_start;
2394     else
2395         uri->authority_start = -1;
2396
2397     return TRUE;
2398 }
2399
2400 /* Attempts to canonicalize the path of a hierarchical URI.
2401  *
2402  * Things that happen:
2403  *  1). Forbidden characters are percent encoded, unless the NO_ENCODE_FORBIDDEN
2404  *      flag is set or it's a file URI. Forbidden characters are always encoded
2405  *      for file schemes reguardless and forbidden characters are never encoded
2406  *      for unknown scheme types.
2407  *
2408  *  2). For known scheme types '\\' are changed to '/'.
2409  *
2410  *  3). Percent encoded, unreserved characters are decoded to their actual values.
2411  *      Unless the scheme type is unknown. For file schemes any percent encoded
2412  *      character in the unreserved or reserved set is decoded.
2413  *
2414  *  4). For File schemes if the path is starts with a drive letter and doesn't
2415  *      start with a '/' then one is appended.
2416  *      Ex: file://c:/test.mp3 -> file:///c:/test.mp3
2417  *
2418  *  5). Dot segments are removed from the path for all scheme types
2419  *      unless NO_CANONICALIZE flag is set. Dot segments aren't removed
2420  *      for wildcard scheme types.
2421  *
2422  * NOTES:
2423  *      file://c:/test%20test   -> file:///c:/test%2520test
2424  *      file://c:/test%3Etest   -> file:///c:/test%253Etest
2425  *      file:///c:/test%20test  -> file:///c:/test%20test
2426  *      file:///c:/test%test    -> file:///c:/test%25test
2427  */
2428 static BOOL canonicalize_path_hierarchical(const parse_data *data, Uri *uri,
2429                                            DWORD flags, BOOL computeOnly) {
2430     const WCHAR *ptr;
2431     const BOOL known_scheme = data->scheme_type != URL_SCHEME_UNKNOWN;
2432     const BOOL is_file = data->scheme_type == URL_SCHEME_FILE;
2433
2434     BOOL escape_pct = FALSE;
2435
2436     if(!data->path) {
2437         uri->path_start = -1;
2438         uri->path_len = 0;
2439         return TRUE;
2440     }
2441
2442     uri->path_start = uri->canon_len;
2443
2444     /* Check if a '/' needs to be appended for the file scheme. */
2445     if(is_file) {
2446         if(data->path_len > 1 && is_alpha(*(data->path)) &&
2447            *(data->path+1) == ':') {
2448             if(!computeOnly)
2449                 uri->canon_uri[uri->canon_len] = '/';
2450             uri->canon_len++;
2451             escape_pct = TRUE;
2452         }
2453     }
2454
2455     for(ptr = data->path; ptr < data->path+data->path_len; ++ptr) {
2456         if(*ptr == '%') {
2457             const WCHAR *tmp = ptr;
2458             WCHAR val;
2459
2460             /* Check if the % represents a valid encoded char, or if it needs encoded. */
2461             BOOL force_encode = !check_pct_encoded(&tmp) && is_file;
2462             val = decode_pct_val(ptr);
2463
2464             if(force_encode || escape_pct) {
2465                 /* Escape the percent sign in the file URI. */
2466                 if(!computeOnly)
2467                     pct_encode_val(*ptr, uri->canon_uri+uri->canon_len);
2468                 uri->canon_len += 3;
2469             } else if((is_unreserved(val) && known_scheme) ||
2470                       (is_file && (is_unreserved(val) || is_reserved(val)))) {
2471                 if(!computeOnly)
2472                     uri->canon_uri[uri->canon_len] = val;
2473                 ++uri->canon_len;
2474
2475                 ptr += 2;
2476                 continue;
2477             } else {
2478                 if(!computeOnly)
2479                     uri->canon_uri[uri->canon_len] = *ptr;
2480                 ++uri->canon_len;
2481             }
2482         } else if(*ptr == '\\' && known_scheme) {
2483             if(!computeOnly)
2484                 uri->canon_uri[uri->canon_len] = '/';
2485             ++uri->canon_len;
2486         } else if(known_scheme && !is_unreserved(*ptr) && !is_reserved(*ptr) &&
2487                   (!(flags & Uri_CREATE_NO_ENCODE_FORBIDDEN_CHARACTERS) || is_file)) {
2488             /* Escape the forbidden character. */
2489             if(!computeOnly)
2490                 pct_encode_val(*ptr, uri->canon_uri+uri->canon_len);
2491             uri->canon_len += 3;
2492         } else {
2493             if(!computeOnly)
2494                 uri->canon_uri[uri->canon_len] = *ptr;
2495             ++uri->canon_len;
2496         }
2497     }
2498
2499     uri->path_len = uri->canon_len - uri->path_start;
2500
2501     /* Removing the dot segments only happens when it's not in
2502      * computeOnly mode and it's not a wildcard scheme.
2503      */
2504     if(!computeOnly && data->scheme_type != URL_SCHEME_WILDCARD) {
2505         if(!(flags & Uri_CREATE_NO_CANONICALIZE)) {
2506             /* Remove the dot segments (if any) and reset everything to the new
2507              * correct length.
2508              */
2509             DWORD new_len = remove_dot_segments(uri->canon_uri+uri->path_start, uri->path_len);
2510             uri->canon_len -= uri->path_len-new_len;
2511             uri->path_len = new_len;
2512         }
2513     }
2514
2515     if(!computeOnly)
2516         TRACE("Canonicalized path %s len=%d\n",
2517             debugstr_wn(uri->canon_uri+uri->path_start, uri->path_len),
2518             uri->path_len);
2519
2520     return TRUE;
2521 }
2522
2523 /* Attempts to canonicalize the path for an opaque URI.
2524  *
2525  * For known scheme types:
2526  *  1)  forbidden characters are percent encoded if
2527  *      NO_ENCODE_FORBIDDEN_CHARACTERS isn't set.
2528  *
2529  *  2)  Percent encoded, unreserved characters are decoded
2530  *      to their actual values, for known scheme types.
2531  *
2532  *  3)  '\\' are changed to '/' for known scheme types
2533  *      except for mailto schemes.
2534  */
2535 static BOOL canonicalize_path_opaque(const parse_data *data, Uri *uri, DWORD flags, BOOL computeOnly) {
2536     const WCHAR *ptr;
2537     const BOOL known_scheme = data->scheme_type != URL_SCHEME_UNKNOWN;
2538
2539     if(!data->path) {
2540         uri->path_start = -1;
2541         uri->path_len = 0;
2542         return TRUE;
2543     }
2544
2545     uri->path_start = uri->canon_len;
2546
2547     /* Windows doesn't allow a "//" to appear after the scheme
2548      * of a URI, if it's an opaque URI.
2549      */
2550     if(data->scheme && *(data->path) == '/' && *(data->path+1) == '/') {
2551         /* So it inserts a "/." before the "//" if it exists. */
2552         if(!computeOnly) {
2553             uri->canon_uri[uri->canon_len] = '/';
2554             uri->canon_uri[uri->canon_len+1] = '.';
2555         }
2556
2557         uri->canon_len += 2;
2558     }
2559
2560     for(ptr = data->path; ptr < data->path+data->path_len; ++ptr) {
2561         if(*ptr == '%' && known_scheme) {
2562             WCHAR val = decode_pct_val(ptr);
2563
2564             if(is_unreserved(val)) {
2565                 if(!computeOnly)
2566                     uri->canon_uri[uri->canon_len] = val;
2567                 ++uri->canon_len;
2568
2569                 ptr += 2;
2570                 continue;
2571             } else {
2572                 if(!computeOnly)
2573                     uri->canon_uri[uri->canon_len] = *ptr;
2574                 ++uri->canon_len;
2575             }
2576         } else if(known_scheme && !is_unreserved(*ptr) && !is_reserved(*ptr) &&
2577                   !(flags & Uri_CREATE_NO_ENCODE_FORBIDDEN_CHARACTERS)) {
2578             if(!computeOnly)
2579                 pct_encode_val(*ptr, uri->canon_uri+uri->canon_len);
2580             uri->canon_len += 3;
2581         } else {
2582             if(!computeOnly)
2583                 uri->canon_uri[uri->canon_len] = *ptr;
2584             ++uri->canon_len;
2585         }
2586     }
2587
2588     uri->path_len = uri->canon_len - uri->path_start;
2589
2590     TRACE("(%p %p %x %d): Canonicalized opaque URI path %s len=%d\n", data, uri, flags, computeOnly,
2591         debugstr_wn(uri->canon_uri+uri->path_start, uri->path_len), uri->path_len);
2592     return TRUE;
2593 }
2594
2595 /* Determines how the URI represented by the parse_data should be canonicalized.
2596  *
2597  * Essentially, if the parse_data represents an hierarchical URI then it calls
2598  * canonicalize_authority and the canonicalization functions for the path. If the
2599  * URI is opaque it canonicalizes the path of the URI.
2600  */
2601 static BOOL canonicalize_hierpart(const parse_data *data, Uri *uri, DWORD flags, BOOL computeOnly) {
2602     if(!data->is_opaque) {
2603         /* "//" is only added for non-wildcard scheme types. */
2604         if(data->scheme_type != URL_SCHEME_WILDCARD) {
2605             if(!computeOnly) {
2606                 INT pos = uri->canon_len;
2607
2608                 uri->canon_uri[pos] = '/';
2609                 uri->canon_uri[pos+1] = '/';
2610            }
2611            uri->canon_len += 2;
2612         }
2613
2614         if(!canonicalize_authority(data, uri, flags, computeOnly))
2615             return FALSE;
2616
2617         /* TODO: Canonicalize the path of the URI. */
2618         if(!canonicalize_path_hierarchical(data, uri, flags, computeOnly))
2619             return FALSE;
2620
2621     } else {
2622         /* Opaque URI's don't have an authority. */
2623         uri->userinfo_start = uri->userinfo_split = -1;
2624         uri->userinfo_len = 0;
2625         uri->host_start = -1;
2626         uri->host_len = 0;
2627         uri->host_type = Uri_HOST_UNKNOWN;
2628         uri->has_port = FALSE;
2629         uri->authority_start = -1;
2630         uri->authority_len = 0;
2631         uri->domain_offset = -1;
2632
2633         if(!canonicalize_path_opaque(data, uri, flags, computeOnly))
2634             return FALSE;
2635     }
2636
2637     if(uri->path_start > -1 && !computeOnly)
2638         /* Finding file extensions happens for both types of URIs. */
2639         uri->extension_offset = find_file_extension(uri->canon_uri+uri->path_start, uri->path_len);
2640     else
2641         uri->extension_offset = -1;
2642
2643     return TRUE;
2644 }
2645
2646 /* Canonicalizes the scheme information specified in the parse_data using the specified flags. */
2647 static BOOL canonicalize_scheme(const parse_data *data, Uri *uri, DWORD flags, BOOL computeOnly) {
2648     uri->scheme_start = -1;
2649     uri->scheme_len = 0;
2650
2651     if(!data->scheme) {
2652         /* The only type of URI that doesn't have to have a scheme is a relative
2653          * URI.
2654          */
2655         if(!data->is_relative) {
2656             FIXME("(%p %p %x): Unable to determine the scheme type of %s.\n", data,
2657                     uri, flags, debugstr_w(data->uri));
2658             return FALSE;
2659         }
2660     } else {
2661         if(!computeOnly) {
2662             DWORD i;
2663             INT pos = uri->canon_len;
2664
2665             for(i = 0; i < data->scheme_len; ++i) {
2666                 /* Scheme name must be lower case after canonicalization. */
2667                 uri->canon_uri[i + pos] = tolowerW(data->scheme[i]);
2668             }
2669
2670             uri->canon_uri[i + pos] = ':';
2671             uri->scheme_start = pos;
2672
2673             TRACE("(%p %p %x): Canonicalized scheme=%s, len=%d.\n", data, uri, flags,
2674                     debugstr_wn(uri->canon_uri,  uri->scheme_len), data->scheme_len);
2675         }
2676
2677         /* This happens in both computation modes. */
2678         uri->canon_len += data->scheme_len + 1;
2679         uri->scheme_len = data->scheme_len;
2680     }
2681     return TRUE;
2682 }
2683
2684 /* Compute's what the length of the URI specified by the parse_data will be
2685  * after canonicalization occurs using the specified flags.
2686  *
2687  * This function will return a non-zero value indicating the length of the canonicalized
2688  * URI, or -1 on error.
2689  */
2690 static int compute_canonicalized_length(const parse_data *data, DWORD flags) {
2691     Uri uri;
2692
2693     memset(&uri, 0, sizeof(Uri));
2694
2695     TRACE("(%p %x): Beginning to compute canonicalized length for URI %s\n", data, flags,
2696             debugstr_w(data->uri));
2697
2698     if(!canonicalize_scheme(data, &uri, flags, TRUE)) {
2699         ERR("(%p %x): Failed to compute URI scheme length.\n", data, flags);
2700         return -1;
2701     }
2702
2703     if(!canonicalize_hierpart(data, &uri, flags, TRUE)) {
2704         ERR("(%p %x): Failed to compute URI hierpart length.\n", data, flags);
2705         return -1;
2706     }
2707
2708     TRACE("(%p %x): Finished computing canonicalized URI length. length=%d\n", data, flags, uri.canon_len);
2709
2710     return uri.canon_len;
2711 }
2712
2713 /* Canonicalizes the URI data specified in the parse_data, using the given flags. If the
2714  * canonicalization succeededs it will store all the canonicalization information
2715  * in the pointer to the Uri.
2716  *
2717  * To canonicalize a URI this function first computes what the length of the URI
2718  * specified by the parse_data will be. Once this is done it will then perfom the actual
2719  * canonicalization of the URI.
2720  */
2721 static HRESULT canonicalize_uri(const parse_data *data, Uri *uri, DWORD flags) {
2722     INT len;
2723
2724     uri->canon_uri = NULL;
2725     len = uri->canon_size = uri->canon_len = 0;
2726
2727     TRACE("(%p %p %x): beginning to canonicalize URI %s.\n", data, uri, flags, debugstr_w(data->uri));
2728
2729     /* First try to compute the length of the URI. */
2730     len = compute_canonicalized_length(data, flags);
2731     if(len == -1) {
2732         ERR("(%p %p %x): Could not compute the canonicalized length of %s.\n", data, uri, flags,
2733                 debugstr_w(data->uri));
2734         return E_INVALIDARG;
2735     }
2736
2737     uri->canon_uri = heap_alloc((len+1)*sizeof(WCHAR));
2738     if(!uri->canon_uri)
2739         return E_OUTOFMEMORY;
2740
2741     uri->canon_size = len;
2742     if(!canonicalize_scheme(data, uri, flags, FALSE)) {
2743         ERR("(%p %p %x): Unable to canonicalize the scheme of the URI.\n", data, uri, flags);
2744         heap_free(uri->canon_uri);
2745         return E_INVALIDARG;
2746     }
2747     uri->scheme_type = data->scheme_type;
2748
2749     if(!canonicalize_hierpart(data, uri, flags, FALSE)) {
2750         ERR("(%p %p %x): Unable to canonicalize the heirpart of the URI\n", data, uri, flags);
2751         heap_free(uri->canon_uri);
2752         return E_INVALIDARG;
2753     }
2754
2755     /* There's a possibility we didn't use all the space we allocated
2756      * earlier.
2757      */
2758     if(uri->canon_len < uri->canon_size) {
2759         /* This happens if the URI is hierarchical and dot
2760          * segments were removed from it's path.
2761          */
2762         WCHAR *tmp = heap_realloc(uri->canon_uri, (uri->canon_len+1)*sizeof(WCHAR));
2763         if(!tmp)
2764             return E_OUTOFMEMORY;
2765
2766         uri->canon_uri = tmp;
2767         uri->canon_size = uri->canon_len;
2768     }
2769
2770     uri->canon_uri[uri->canon_len] = '\0';
2771     TRACE("(%p %p %x): finished canonicalizing the URI. uri=%s\n", data, uri, flags, debugstr_w(uri->canon_uri));
2772
2773     return S_OK;
2774 }
2775
2776 #define URI(x)         ((IUri*)  &(x)->lpIUriVtbl)
2777 #define URIBUILDER(x)  ((IUriBuilder*)  &(x)->lpIUriBuilderVtbl)
2778
2779 #define URI_THIS(iface) DEFINE_THIS(Uri, IUri, iface)
2780
2781 static HRESULT WINAPI Uri_QueryInterface(IUri *iface, REFIID riid, void **ppv)
2782 {
2783     Uri *This = URI_THIS(iface);
2784
2785     if(IsEqualGUID(&IID_IUnknown, riid)) {
2786         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
2787         *ppv = URI(This);
2788     }else if(IsEqualGUID(&IID_IUri, riid)) {
2789         TRACE("(%p)->(IID_IUri %p)\n", This, ppv);
2790         *ppv = URI(This);
2791     }else {
2792         TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
2793         *ppv = NULL;
2794         return E_NOINTERFACE;
2795     }
2796
2797     IUnknown_AddRef((IUnknown*)*ppv);
2798     return S_OK;
2799 }
2800
2801 static ULONG WINAPI Uri_AddRef(IUri *iface)
2802 {
2803     Uri *This = URI_THIS(iface);
2804     LONG ref = InterlockedIncrement(&This->ref);
2805
2806     TRACE("(%p) ref=%d\n", This, ref);
2807
2808     return ref;
2809 }
2810
2811 static ULONG WINAPI Uri_Release(IUri *iface)
2812 {
2813     Uri *This = URI_THIS(iface);
2814     LONG ref = InterlockedDecrement(&This->ref);
2815
2816     TRACE("(%p) ref=%d\n", This, ref);
2817
2818     if(!ref) {
2819         SysFreeString(This->raw_uri);
2820         heap_free(This->canon_uri);
2821         heap_free(This);
2822     }
2823
2824     return ref;
2825 }
2826
2827 static HRESULT WINAPI Uri_GetPropertyBSTR(IUri *iface, Uri_PROPERTY uriProp, BSTR *pbstrProperty, DWORD dwFlags)
2828 {
2829     Uri *This = URI_THIS(iface);
2830     HRESULT hres;
2831     TRACE("(%p)->(%d %p %x)\n", This, uriProp, pbstrProperty, dwFlags);
2832
2833     if(!pbstrProperty)
2834         return E_POINTER;
2835
2836     if(uriProp > Uri_PROPERTY_STRING_LAST) {
2837         /* Windows allocates an empty BSTR for invalid Uri_PROPERTY's. */
2838         *pbstrProperty = SysAllocStringLen(NULL, 0);
2839         if(!(*pbstrProperty))
2840             return E_OUTOFMEMORY;
2841
2842         /* It only returns S_FALSE for the ZONE property... */
2843         if(uriProp == Uri_PROPERTY_ZONE)
2844             return S_FALSE;
2845         else
2846             return S_OK;
2847     }
2848
2849     /* Don't have support for flags yet. */
2850     if(dwFlags) {
2851         FIXME("(%p)->(%d %p %x)\n", This, uriProp, pbstrProperty, dwFlags);
2852         return E_NOTIMPL;
2853     }
2854
2855     switch(uriProp) {
2856     case Uri_PROPERTY_AUTHORITY:
2857         if(This->authority_start > -1) {
2858             *pbstrProperty = SysAllocStringLen(This->canon_uri+This->authority_start, This->authority_len);
2859             hres = S_OK;
2860         } else {
2861             *pbstrProperty = SysAllocStringLen(NULL, 0);
2862             hres = S_FALSE;
2863         }
2864
2865         if(!(*pbstrProperty))
2866             hres = E_OUTOFMEMORY;
2867
2868         break;
2869     case Uri_PROPERTY_DOMAIN:
2870         if(This->domain_offset > -1) {
2871             *pbstrProperty = SysAllocStringLen(This->canon_uri+This->host_start+This->domain_offset,
2872                                                This->host_len-This->domain_offset);
2873             hres = S_OK;
2874         } else {
2875             *pbstrProperty = SysAllocStringLen(NULL, 0);
2876             hres = S_FALSE;
2877         }
2878
2879         if(!(*pbstrProperty))
2880             hres = E_OUTOFMEMORY;
2881
2882         break;
2883     case Uri_PROPERTY_EXTENSION:
2884         if(This->extension_offset > -1) {
2885             *pbstrProperty = SysAllocStringLen(This->canon_uri+This->path_start+This->extension_offset,
2886                                                This->path_len-This->extension_offset);
2887             hres = S_OK;
2888         } else {
2889             *pbstrProperty = SysAllocStringLen(NULL, 0);
2890             hres = S_FALSE;
2891         }
2892
2893         if(!(*pbstrProperty))
2894             hres = E_OUTOFMEMORY;
2895
2896         break;
2897     case Uri_PROPERTY_HOST:
2898         if(This->host_start > -1) {
2899             /* The '[' and ']' aren't included for IPv6 addresses. */
2900             if(This->host_type == Uri_HOST_IPV6)
2901                 *pbstrProperty = SysAllocStringLen(This->canon_uri+This->host_start+1, This->host_len-2);
2902             else
2903                 *pbstrProperty = SysAllocStringLen(This->canon_uri+This->host_start, This->host_len);
2904
2905             hres = S_OK;
2906         } else {
2907             *pbstrProperty = SysAllocStringLen(NULL, 0);
2908             hres = S_FALSE;
2909         }
2910
2911         if(!(*pbstrProperty))
2912             hres = E_OUTOFMEMORY;
2913
2914         break;
2915     case Uri_PROPERTY_PASSWORD:
2916         if(This->userinfo_split > -1) {
2917             *pbstrProperty = SysAllocStringLen(
2918                 This->canon_uri+This->userinfo_start+This->userinfo_split+1,
2919                 This->userinfo_len-This->userinfo_split-1);
2920             hres = S_OK;
2921         } else {
2922             *pbstrProperty = SysAllocStringLen(NULL, 0);
2923             hres = S_FALSE;
2924         }
2925
2926         if(!(*pbstrProperty))
2927             return E_OUTOFMEMORY;
2928
2929         break;
2930     case Uri_PROPERTY_PATH:
2931         if(This->path_start > -1) {
2932             *pbstrProperty = SysAllocStringLen(This->canon_uri+This->path_start, This->path_len);
2933             hres = S_OK;
2934         } else {
2935             *pbstrProperty = SysAllocStringLen(NULL, 0);
2936             hres = S_FALSE;
2937         }
2938
2939         if(!(*pbstrProperty))
2940             hres = E_OUTOFMEMORY;
2941
2942         break;
2943     case Uri_PROPERTY_RAW_URI:
2944         *pbstrProperty = SysAllocString(This->raw_uri);
2945         if(!(*pbstrProperty))
2946             hres = E_OUTOFMEMORY;
2947         else
2948             hres = S_OK;
2949         break;
2950     case Uri_PROPERTY_SCHEME_NAME:
2951         if(This->scheme_start > -1) {
2952             *pbstrProperty = SysAllocStringLen(This->canon_uri + This->scheme_start, This->scheme_len);
2953             hres = S_OK;
2954         } else {
2955             *pbstrProperty = SysAllocStringLen(NULL, 0);
2956             hres = S_FALSE;
2957         }
2958
2959         if(!(*pbstrProperty))
2960             hres = E_OUTOFMEMORY;
2961
2962         break;
2963     case Uri_PROPERTY_USER_INFO:
2964         if(This->userinfo_start > -1) {
2965             *pbstrProperty = SysAllocStringLen(This->canon_uri+This->userinfo_start, This->userinfo_len);
2966             hres = S_OK;
2967         } else {
2968             *pbstrProperty = SysAllocStringLen(NULL, 0);
2969             hres = S_FALSE;
2970         }
2971
2972         if(!(*pbstrProperty))
2973             hres = E_OUTOFMEMORY;
2974
2975         break;
2976     case Uri_PROPERTY_USER_NAME:
2977         if(This->userinfo_start > -1) {
2978             /* If userinfo_split is set, that means a password exists
2979              * so the username is only from userinfo_start to userinfo_split.
2980              */
2981             if(This->userinfo_split > -1) {
2982                 *pbstrProperty = SysAllocStringLen(This->canon_uri + This->userinfo_start, This->userinfo_split);
2983                 hres = S_OK;
2984             } else {
2985                 *pbstrProperty = SysAllocStringLen(This->canon_uri + This->userinfo_start, This->userinfo_len);
2986                 hres = S_OK;
2987             }
2988         } else {
2989             *pbstrProperty = SysAllocStringLen(NULL, 0);
2990             hres = S_FALSE;
2991         }
2992
2993         if(!(*pbstrProperty))
2994             return E_OUTOFMEMORY;
2995
2996         break;
2997     default:
2998         FIXME("(%p)->(%d %p %x)\n", This, uriProp, pbstrProperty, dwFlags);
2999         hres = E_NOTIMPL;
3000     }
3001
3002     return hres;
3003 }
3004
3005 static HRESULT WINAPI Uri_GetPropertyLength(IUri *iface, Uri_PROPERTY uriProp, DWORD *pcchProperty, DWORD dwFlags)
3006 {
3007     Uri *This = URI_THIS(iface);
3008     HRESULT hres;
3009     TRACE("(%p)->(%d %p %x)\n", This, uriProp, pcchProperty, dwFlags);
3010
3011     if(!pcchProperty)
3012         return E_INVALIDARG;
3013
3014     /* Can only return a length for a property if it's a string. */
3015     if(uriProp > Uri_PROPERTY_STRING_LAST)
3016         return E_INVALIDARG;
3017
3018     /* Don't have support for flags yet. */
3019     if(dwFlags) {
3020         FIXME("(%p)->(%d %p %x)\n", This, uriProp, pcchProperty, dwFlags);
3021         return E_NOTIMPL;
3022     }
3023
3024     switch(uriProp) {
3025     case Uri_PROPERTY_AUTHORITY:
3026         *pcchProperty = This->authority_len;
3027         hres = (This->authority_start > -1) ? S_OK : S_FALSE;
3028         break;
3029     case Uri_PROPERTY_DOMAIN:
3030         if(This->domain_offset > -1)
3031             *pcchProperty = This->host_len - This->domain_offset;
3032         else
3033             *pcchProperty = 0;
3034
3035         hres = (This->domain_offset > -1) ? S_OK : S_FALSE;
3036         break;
3037     case Uri_PROPERTY_EXTENSION:
3038         if(This->extension_offset > -1) {
3039             *pcchProperty = This->path_len - This->extension_offset;
3040             hres = S_OK;
3041         } else {
3042             *pcchProperty = 0;
3043             hres = S_FALSE;
3044         }
3045
3046         break;
3047     case Uri_PROPERTY_HOST:
3048         *pcchProperty = This->host_len;
3049
3050         /* '[' and ']' aren't included in the length. */
3051         if(This->host_type == Uri_HOST_IPV6)
3052             *pcchProperty -= 2;
3053
3054         hres = (This->host_start > -1) ? S_OK : S_FALSE;
3055         break;
3056     case Uri_PROPERTY_PASSWORD:
3057         *pcchProperty = (This->userinfo_split > -1) ? This->userinfo_len-This->userinfo_split-1 : 0;
3058         hres = (This->userinfo_split > -1) ? S_OK : S_FALSE;
3059         break;
3060     case Uri_PROPERTY_PATH:
3061         *pcchProperty = This->path_len;
3062         hres = (This->path_start > -1) ? S_OK : S_FALSE;
3063         break;
3064     case Uri_PROPERTY_RAW_URI:
3065         *pcchProperty = SysStringLen(This->raw_uri);
3066         hres = S_OK;
3067         break;
3068     case Uri_PROPERTY_SCHEME_NAME:
3069         *pcchProperty = This->scheme_len;
3070         hres = (This->scheme_start > -1) ? S_OK : S_FALSE;
3071         break;
3072     case Uri_PROPERTY_USER_INFO:
3073         *pcchProperty = This->userinfo_len;
3074         hres = (This->userinfo_start > -1) ? S_OK : S_FALSE;
3075         break;
3076     case Uri_PROPERTY_USER_NAME:
3077         *pcchProperty = (This->userinfo_split > -1) ? This->userinfo_split : This->userinfo_len;
3078         hres = (This->userinfo_start > -1) ? S_OK : S_FALSE;
3079         break;
3080     default:
3081         FIXME("(%p)->(%d %p %x)\n", This, uriProp, pcchProperty, dwFlags);
3082         hres = E_NOTIMPL;
3083     }
3084
3085     return hres;
3086 }
3087
3088 static HRESULT WINAPI Uri_GetPropertyDWORD(IUri *iface, Uri_PROPERTY uriProp, DWORD *pcchProperty, DWORD dwFlags)
3089 {
3090     Uri *This = URI_THIS(iface);
3091     HRESULT hres;
3092
3093     TRACE("(%p)->(%d %p %x)\n", This, uriProp, pcchProperty, dwFlags);
3094
3095     if(!pcchProperty)
3096         return E_INVALIDARG;
3097
3098     /* Microsoft's implementation for the ZONE property of a URI seems to be lacking...
3099      * From what I can tell, instead of checking which URLZONE the URI belongs to it
3100      * simply assigns URLZONE_INVALID and returns E_NOTIMPL. This also applies to the GetZone
3101      * function.
3102      */
3103     if(uriProp == Uri_PROPERTY_ZONE) {
3104         *pcchProperty = URLZONE_INVALID;
3105         return E_NOTIMPL;
3106     }
3107
3108     if(uriProp < Uri_PROPERTY_DWORD_START) {
3109         *pcchProperty = 0;
3110         return E_INVALIDARG;
3111     }
3112
3113     switch(uriProp) {
3114     case Uri_PROPERTY_HOST_TYPE:
3115         *pcchProperty = This->host_type;
3116         hres = S_OK;
3117         break;
3118     case Uri_PROPERTY_PORT:
3119         if(!This->has_port) {
3120             *pcchProperty = 0;
3121             hres = S_FALSE;
3122         } else {
3123             *pcchProperty = This->port;
3124             hres = S_OK;
3125         }
3126
3127         break;
3128     case Uri_PROPERTY_SCHEME:
3129         *pcchProperty = This->scheme_type;
3130         hres = S_OK;
3131         break;
3132     default:
3133         FIXME("(%p)->(%d %p %x)\n", This, uriProp, pcchProperty, dwFlags);
3134         hres = E_NOTIMPL;
3135     }
3136
3137     return hres;
3138 }
3139
3140 static HRESULT WINAPI Uri_HasProperty(IUri *iface, Uri_PROPERTY uriProp, BOOL *pfHasProperty)
3141 {
3142     Uri *This = URI_THIS(iface);
3143     FIXME("(%p)->(%d %p)\n", This, uriProp, pfHasProperty);
3144
3145     if(!pfHasProperty)
3146         return E_INVALIDARG;
3147
3148     return E_NOTIMPL;
3149 }
3150
3151 static HRESULT WINAPI Uri_GetAbsoluteUri(IUri *iface, BSTR *pstrAbsoluteUri)
3152 {
3153     Uri *This = URI_THIS(iface);
3154     FIXME("(%p)->(%p)\n", This, pstrAbsoluteUri);
3155
3156     if(!pstrAbsoluteUri)
3157         return E_POINTER;
3158
3159     return E_NOTIMPL;
3160 }
3161
3162 static HRESULT WINAPI Uri_GetAuthority(IUri *iface, BSTR *pstrAuthority)
3163 {
3164     TRACE("(%p)->(%p)\n", iface, pstrAuthority);
3165     return Uri_GetPropertyBSTR(iface, Uri_PROPERTY_AUTHORITY, pstrAuthority, 0);
3166 }
3167
3168 static HRESULT WINAPI Uri_GetDisplayUri(IUri *iface, BSTR *pstrDisplayUri)
3169 {
3170     Uri *This = URI_THIS(iface);
3171     FIXME("(%p)->(%p)\n", This, pstrDisplayUri);
3172
3173     if(!pstrDisplayUri)
3174         return E_POINTER;
3175
3176     return E_NOTIMPL;
3177 }
3178
3179 static HRESULT WINAPI Uri_GetDomain(IUri *iface, BSTR *pstrDomain)
3180 {
3181     TRACE("(%p)->(%p)\n", iface, pstrDomain);
3182     return Uri_GetPropertyBSTR(iface, Uri_PROPERTY_DOMAIN, pstrDomain, 0);
3183 }
3184
3185 static HRESULT WINAPI Uri_GetExtension(IUri *iface, BSTR *pstrExtension)
3186 {
3187     TRACE("(%p)->(%p)\n", iface, pstrExtension);
3188     return Uri_GetPropertyBSTR(iface, Uri_PROPERTY_EXTENSION, pstrExtension, 0);
3189 }
3190
3191 static HRESULT WINAPI Uri_GetFragment(IUri *iface, BSTR *pstrFragment)
3192 {
3193     Uri *This = URI_THIS(iface);
3194     FIXME("(%p)->(%p)\n", This, pstrFragment);
3195
3196     if(!pstrFragment)
3197         return E_POINTER;
3198
3199     return E_NOTIMPL;
3200 }
3201
3202 static HRESULT WINAPI Uri_GetHost(IUri *iface, BSTR *pstrHost)
3203 {
3204     TRACE("(%p)->(%p)\n", iface, pstrHost);
3205     return Uri_GetPropertyBSTR(iface, Uri_PROPERTY_HOST, pstrHost, 0);
3206 }
3207
3208 static HRESULT WINAPI Uri_GetPassword(IUri *iface, BSTR *pstrPassword)
3209 {
3210     TRACE("(%p)->(%p)\n", iface, pstrPassword);
3211     return Uri_GetPropertyBSTR(iface, Uri_PROPERTY_PASSWORD, pstrPassword, 0);
3212 }
3213
3214 static HRESULT WINAPI Uri_GetPath(IUri *iface, BSTR *pstrPath)
3215 {
3216     TRACE("(%p)->(%p)\n", iface, pstrPath);
3217     return Uri_GetPropertyBSTR(iface, Uri_PROPERTY_PATH, pstrPath, 0);
3218 }
3219
3220 static HRESULT WINAPI Uri_GetPathAndQuery(IUri *iface, BSTR *pstrPathAndQuery)
3221 {
3222     Uri *This = URI_THIS(iface);
3223     FIXME("(%p)->(%p)\n", This, pstrPathAndQuery);
3224
3225     if(!pstrPathAndQuery)
3226         return E_POINTER;
3227
3228     return E_NOTIMPL;
3229 }
3230
3231 static HRESULT WINAPI Uri_GetQuery(IUri *iface, BSTR *pstrQuery)
3232 {
3233     Uri *This = URI_THIS(iface);
3234     FIXME("(%p)->(%p)\n", This, pstrQuery);
3235
3236     if(!pstrQuery)
3237         return E_POINTER;
3238
3239     return E_NOTIMPL;
3240 }
3241
3242 static HRESULT WINAPI Uri_GetRawUri(IUri *iface, BSTR *pstrRawUri)
3243 {
3244     Uri *This = URI_THIS(iface);
3245     TRACE("(%p)->(%p)\n", This, pstrRawUri);
3246
3247     /* Just forward the call to GetPropertyBSTR. */
3248     return Uri_GetPropertyBSTR(iface, Uri_PROPERTY_RAW_URI, pstrRawUri, 0);
3249 }
3250
3251 static HRESULT WINAPI Uri_GetSchemeName(IUri *iface, BSTR *pstrSchemeName)
3252 {
3253     Uri *This = URI_THIS(iface);
3254     TRACE("(%p)->(%p)\n", This, pstrSchemeName);
3255     return Uri_GetPropertyBSTR(iface, Uri_PROPERTY_SCHEME_NAME, pstrSchemeName, 0);
3256 }
3257
3258 static HRESULT WINAPI Uri_GetUserInfo(IUri *iface, BSTR *pstrUserInfo)
3259 {
3260     TRACE("(%p)->(%p)\n", iface, pstrUserInfo);
3261     return Uri_GetPropertyBSTR(iface, Uri_PROPERTY_USER_INFO, pstrUserInfo, 0);
3262 }
3263
3264 static HRESULT WINAPI Uri_GetUserName(IUri *iface, BSTR *pstrUserName)
3265 {
3266     TRACE("(%p)->(%p)\n", iface, pstrUserName);
3267     return Uri_GetPropertyBSTR(iface, Uri_PROPERTY_USER_NAME, pstrUserName, 0);
3268 }
3269
3270 static HRESULT WINAPI Uri_GetHostType(IUri *iface, DWORD *pdwHostType)
3271 {
3272     TRACE("(%p)->(%p)\n", iface, pdwHostType);
3273     return Uri_GetPropertyDWORD(iface, Uri_PROPERTY_HOST_TYPE, pdwHostType, 0);
3274 }
3275
3276 static HRESULT WINAPI Uri_GetPort(IUri *iface, DWORD *pdwPort)
3277 {
3278     TRACE("(%p)->(%p)\n", iface, pdwPort);
3279     return Uri_GetPropertyDWORD(iface, Uri_PROPERTY_PORT, pdwPort, 0);
3280 }
3281
3282 static HRESULT WINAPI Uri_GetScheme(IUri *iface, DWORD *pdwScheme)
3283 {
3284     Uri *This = URI_THIS(iface);
3285     TRACE("(%p)->(%p)\n", This, pdwScheme);
3286     return Uri_GetPropertyDWORD(iface, Uri_PROPERTY_SCHEME, pdwScheme, 0);
3287 }
3288
3289 static HRESULT WINAPI Uri_GetZone(IUri *iface, DWORD *pdwZone)
3290 {
3291     TRACE("(%p)->(%p)\n", iface, pdwZone);
3292     return Uri_GetPropertyDWORD(iface, Uri_PROPERTY_ZONE,pdwZone, 0);
3293 }
3294
3295 static HRESULT WINAPI Uri_GetProperties(IUri *iface, DWORD *pdwProperties)
3296 {
3297     Uri *This = URI_THIS(iface);
3298     FIXME("(%p)->(%p)\n", This, pdwProperties);
3299
3300     if(!pdwProperties)
3301         return E_INVALIDARG;
3302
3303     return E_NOTIMPL;
3304 }
3305
3306 static HRESULT WINAPI Uri_IsEqual(IUri *iface, IUri *pUri, BOOL *pfEqual)
3307 {
3308     Uri *This = URI_THIS(iface);
3309     TRACE("(%p)->(%p %p)\n", This, pUri, pfEqual);
3310
3311     if(!pfEqual)
3312         return E_POINTER;
3313
3314     if(!pUri) {
3315         *pfEqual = FALSE;
3316
3317         /* For some reason Windows returns S_OK here... */
3318         return S_OK;
3319     }
3320
3321     FIXME("(%p)->(%p %p)\n", This, pUri, pfEqual);
3322     return E_NOTIMPL;
3323 }
3324
3325 #undef URI_THIS
3326
3327 static const IUriVtbl UriVtbl = {
3328     Uri_QueryInterface,
3329     Uri_AddRef,
3330     Uri_Release,
3331     Uri_GetPropertyBSTR,
3332     Uri_GetPropertyLength,
3333     Uri_GetPropertyDWORD,
3334     Uri_HasProperty,
3335     Uri_GetAbsoluteUri,
3336     Uri_GetAuthority,
3337     Uri_GetDisplayUri,
3338     Uri_GetDomain,
3339     Uri_GetExtension,
3340     Uri_GetFragment,
3341     Uri_GetHost,
3342     Uri_GetPassword,
3343     Uri_GetPath,
3344     Uri_GetPathAndQuery,
3345     Uri_GetQuery,
3346     Uri_GetRawUri,
3347     Uri_GetSchemeName,
3348     Uri_GetUserInfo,
3349     Uri_GetUserName,
3350     Uri_GetHostType,
3351     Uri_GetPort,
3352     Uri_GetScheme,
3353     Uri_GetZone,
3354     Uri_GetProperties,
3355     Uri_IsEqual
3356 };
3357
3358 /***********************************************************************
3359  *           CreateUri (urlmon.@)
3360  */
3361 HRESULT WINAPI CreateUri(LPCWSTR pwzURI, DWORD dwFlags, DWORD_PTR dwReserved, IUri **ppURI)
3362 {
3363     Uri *ret;
3364     HRESULT hr;
3365     parse_data data;
3366
3367     TRACE("(%s %x %x %p)\n", debugstr_w(pwzURI), dwFlags, (DWORD)dwReserved, ppURI);
3368
3369     if(!ppURI)
3370         return E_INVALIDARG;
3371
3372     if(!pwzURI) {
3373         *ppURI = NULL;
3374         return E_INVALIDARG;
3375     }
3376
3377     ret = heap_alloc(sizeof(Uri));
3378     if(!ret)
3379         return E_OUTOFMEMORY;
3380
3381     ret->lpIUriVtbl = &UriVtbl;
3382     ret->ref = 1;
3383
3384     /* Create a copy of pwzURI and store it as the raw_uri. */
3385     ret->raw_uri = SysAllocString(pwzURI);
3386     if(!ret->raw_uri) {
3387         heap_free(ret);
3388         return E_OUTOFMEMORY;
3389     }
3390
3391     memset(&data, 0, sizeof(parse_data));
3392     data.uri = ret->raw_uri;
3393
3394     /* Validate and parse the URI into it's components. */
3395     if(!parse_uri(&data, dwFlags)) {
3396         /* Encountered an unsupported or invalid URI */
3397         SysFreeString(ret->raw_uri);
3398         heap_free(ret);
3399         *ppURI = NULL;
3400         return E_INVALIDARG;
3401     }
3402
3403     /* Canonicalize the URI. */
3404     hr = canonicalize_uri(&data, ret, dwFlags);
3405     if(FAILED(hr)) {
3406         SysFreeString(ret->raw_uri);
3407         heap_free(ret);
3408         *ppURI = NULL;
3409         return hr;
3410     }
3411
3412     *ppURI = URI(ret);
3413     return S_OK;
3414 }
3415
3416 #define URIBUILDER_THIS(iface) DEFINE_THIS(UriBuilder, IUriBuilder, iface)
3417
3418 static HRESULT WINAPI UriBuilder_QueryInterface(IUriBuilder *iface, REFIID riid, void **ppv)
3419 {
3420     UriBuilder *This = URIBUILDER_THIS(iface);
3421
3422     if(IsEqualGUID(&IID_IUnknown, riid)) {
3423         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
3424         *ppv = URIBUILDER(This);
3425     }else if(IsEqualGUID(&IID_IUriBuilder, riid)) {
3426         TRACE("(%p)->(IID_IUri %p)\n", This, ppv);
3427         *ppv = URIBUILDER(This);
3428     }else {
3429         TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
3430         *ppv = NULL;
3431         return E_NOINTERFACE;
3432     }
3433
3434     IUnknown_AddRef((IUnknown*)*ppv);
3435     return S_OK;
3436 }
3437
3438 static ULONG WINAPI UriBuilder_AddRef(IUriBuilder *iface)
3439 {
3440     UriBuilder *This = URIBUILDER_THIS(iface);
3441     LONG ref = InterlockedIncrement(&This->ref);
3442
3443     TRACE("(%p) ref=%d\n", This, ref);
3444
3445     return ref;
3446 }
3447
3448 static ULONG WINAPI UriBuilder_Release(IUriBuilder *iface)
3449 {
3450     UriBuilder *This = URIBUILDER_THIS(iface);
3451     LONG ref = InterlockedDecrement(&This->ref);
3452
3453     TRACE("(%p) ref=%d\n", This, ref);
3454
3455     if(!ref)
3456         heap_free(This);
3457
3458     return ref;
3459 }
3460
3461 static HRESULT WINAPI UriBuilder_CreateUriSimple(IUriBuilder *iface,
3462                                                  DWORD        dwAllowEncodingPropertyMask,
3463                                                  DWORD_PTR    dwReserved,
3464                                                  IUri       **ppIUri)
3465 {
3466     UriBuilder *This = URIBUILDER_THIS(iface);
3467     FIXME("(%p)->(%d %d %p)\n", This, dwAllowEncodingPropertyMask, (DWORD)dwReserved, ppIUri);
3468     return E_NOTIMPL;
3469 }
3470
3471 static HRESULT WINAPI UriBuilder_CreateUri(IUriBuilder *iface,
3472                                            DWORD        dwCreateFlags,
3473                                            DWORD        dwAllowEncodingPropertyMask,
3474                                            DWORD_PTR    dwReserved,
3475                                            IUri       **ppIUri)
3476 {
3477     UriBuilder *This = URIBUILDER_THIS(iface);
3478     FIXME("(%p)->(0x%08x %d %d %p)\n", This, dwCreateFlags, dwAllowEncodingPropertyMask, (DWORD)dwReserved, ppIUri);
3479     return E_NOTIMPL;
3480 }
3481
3482 static HRESULT WINAPI UriBuilder_CreateUriWithFlags(IUriBuilder *iface,
3483                                          DWORD        dwCreateFlags,
3484                                          DWORD        dwUriBuilderFlags,
3485                                          DWORD        dwAllowEncodingPropertyMask,
3486                                          DWORD_PTR    dwReserved,
3487                                          IUri       **ppIUri)
3488 {
3489     UriBuilder *This = URIBUILDER_THIS(iface);
3490     FIXME("(%p)->(0x%08x 0x%08x %d %d %p)\n", This, dwCreateFlags, dwUriBuilderFlags,
3491         dwAllowEncodingPropertyMask, (DWORD)dwReserved, ppIUri);
3492     return E_NOTIMPL;
3493 }
3494
3495 static HRESULT WINAPI  UriBuilder_GetIUri(IUriBuilder *iface, IUri **ppIUri)
3496 {
3497     UriBuilder *This = URIBUILDER_THIS(iface);
3498     FIXME("(%p)->(%p)\n", This, ppIUri);
3499     return E_NOTIMPL;
3500 }
3501
3502 static HRESULT WINAPI UriBuilder_SetIUri(IUriBuilder *iface, IUri *pIUri)
3503 {
3504     UriBuilder *This = URIBUILDER_THIS(iface);
3505     FIXME("(%p)->(%p)\n", This, pIUri);
3506     return E_NOTIMPL;
3507 }
3508
3509 static HRESULT WINAPI UriBuilder_GetFragment(IUriBuilder *iface, DWORD *pcchFragment, LPCWSTR *ppwzFragment)
3510 {
3511     UriBuilder *This = URIBUILDER_THIS(iface);
3512     FIXME("(%p)->(%p %p)\n", This, pcchFragment, ppwzFragment);
3513     return E_NOTIMPL;
3514 }
3515
3516 static HRESULT WINAPI UriBuilder_GetHost(IUriBuilder *iface, DWORD *pcchHost, LPCWSTR *ppwzHost)
3517 {
3518     UriBuilder *This = URIBUILDER_THIS(iface);
3519     FIXME("(%p)->(%p %p)\n", This, pcchHost, ppwzHost);
3520     return E_NOTIMPL;
3521 }
3522
3523 static HRESULT WINAPI UriBuilder_GetPassword(IUriBuilder *iface, DWORD *pcchPassword, LPCWSTR *ppwzPassword)
3524 {
3525     UriBuilder *This = URIBUILDER_THIS(iface);
3526     FIXME("(%p)->(%p %p)\n", This, pcchPassword, ppwzPassword);
3527     return E_NOTIMPL;
3528 }
3529
3530 static HRESULT WINAPI UriBuilder_GetPath(IUriBuilder *iface, DWORD *pcchPath, LPCWSTR *ppwzPath)
3531 {
3532     UriBuilder *This = URIBUILDER_THIS(iface);
3533     FIXME("(%p)->(%p %p)\n", This, pcchPath, ppwzPath);
3534     return E_NOTIMPL;
3535 }
3536
3537 static HRESULT WINAPI UriBuilder_GetPort(IUriBuilder *iface, BOOL *pfHasPort, DWORD *pdwPort)
3538 {
3539     UriBuilder *This = URIBUILDER_THIS(iface);
3540     FIXME("(%p)->(%p %p)\n", This, pfHasPort, pdwPort);
3541     return E_NOTIMPL;
3542 }
3543
3544 static HRESULT WINAPI UriBuilder_GetQuery(IUriBuilder *iface, DWORD *pcchQuery, LPCWSTR *ppwzQuery)
3545 {
3546     UriBuilder *This = URIBUILDER_THIS(iface);
3547     FIXME("(%p)->(%p %p)\n", This, pcchQuery, ppwzQuery);
3548     return E_NOTIMPL;
3549 }
3550
3551 static HRESULT WINAPI UriBuilder_GetSchemeName(IUriBuilder *iface, DWORD *pcchSchemeName, LPCWSTR *ppwzSchemeName)
3552 {
3553     UriBuilder *This = URIBUILDER_THIS(iface);
3554     FIXME("(%p)->(%p %p)\n", This, pcchSchemeName, ppwzSchemeName);
3555     return E_NOTIMPL;
3556 }
3557
3558 static HRESULT WINAPI UriBuilder_GetUserName(IUriBuilder *iface, DWORD *pcchUserName, LPCWSTR *ppwzUserName)
3559 {
3560     UriBuilder *This = URIBUILDER_THIS(iface);
3561     FIXME("(%p)->(%p %p)\n", This, pcchUserName, ppwzUserName);
3562     return E_NOTIMPL;
3563 }
3564
3565 static HRESULT WINAPI UriBuilder_SetFragment(IUriBuilder *iface, LPCWSTR pwzNewValue)
3566 {
3567     UriBuilder *This = URIBUILDER_THIS(iface);
3568     FIXME("(%p)->(%s)\n", This, debugstr_w(pwzNewValue));
3569     return E_NOTIMPL;
3570 }
3571
3572 static HRESULT WINAPI UriBuilder_SetHost(IUriBuilder *iface, LPCWSTR pwzNewValue)
3573 {
3574     UriBuilder *This = URIBUILDER_THIS(iface);
3575     FIXME("(%p)->(%s)\n", This, debugstr_w(pwzNewValue));
3576     return E_NOTIMPL;
3577 }
3578
3579 static HRESULT WINAPI UriBuilder_SetPassword(IUriBuilder *iface, LPCWSTR pwzNewValue)
3580 {
3581     UriBuilder *This = URIBUILDER_THIS(iface);
3582     FIXME("(%p)->(%s)\n", This, debugstr_w(pwzNewValue));
3583     return E_NOTIMPL;
3584 }
3585
3586 static HRESULT WINAPI UriBuilder_SetPath(IUriBuilder *iface, LPCWSTR pwzNewValue)
3587 {
3588     UriBuilder *This = URIBUILDER_THIS(iface);
3589     FIXME("(%p)->(%s)\n", This, debugstr_w(pwzNewValue));
3590     return E_NOTIMPL;
3591 }
3592
3593 static HRESULT WINAPI UriBuilder_SetPort(IUriBuilder *iface, BOOL fHasPort, DWORD dwNewValue)
3594 {
3595     UriBuilder *This = URIBUILDER_THIS(iface);
3596     FIXME("(%p)->(%d %d)\n", This, fHasPort, dwNewValue);
3597     return E_NOTIMPL;
3598 }
3599
3600 static HRESULT WINAPI UriBuilder_SetQuery(IUriBuilder *iface, LPCWSTR pwzNewValue)
3601 {
3602     UriBuilder *This = URIBUILDER_THIS(iface);
3603     FIXME("(%p)->(%s)\n", This, debugstr_w(pwzNewValue));
3604     return E_NOTIMPL;
3605 }
3606
3607 static HRESULT WINAPI UriBuilder_SetSchemeName(IUriBuilder *iface, LPCWSTR pwzNewValue)
3608 {
3609     UriBuilder *This = URIBUILDER_THIS(iface);
3610     FIXME("(%p)->(%s)\n", This, debugstr_w(pwzNewValue));
3611     return E_NOTIMPL;
3612 }
3613
3614 static HRESULT WINAPI UriBuilder_SetUserName(IUriBuilder *iface, LPCWSTR pwzNewValue)
3615 {
3616     UriBuilder *This = URIBUILDER_THIS(iface);
3617     FIXME("(%p)->(%s)\n", This, debugstr_w(pwzNewValue));
3618     return E_NOTIMPL;
3619 }
3620
3621 static HRESULT WINAPI UriBuilder_RemoveProperties(IUriBuilder *iface, DWORD dwPropertyMask)
3622 {
3623     UriBuilder *This = URIBUILDER_THIS(iface);
3624     FIXME("(%p)->(0x%08x)\n", This, dwPropertyMask);
3625     return E_NOTIMPL;
3626 }
3627
3628 static HRESULT WINAPI UriBuilder_HasBeenModified(IUriBuilder *iface, BOOL *pfModified)
3629 {
3630     UriBuilder *This = URIBUILDER_THIS(iface);
3631     FIXME("(%p)->(%p)\n", This, pfModified);
3632     return E_NOTIMPL;
3633 }
3634
3635 #undef URIBUILDER_THIS
3636
3637 static const IUriBuilderVtbl UriBuilderVtbl = {
3638     UriBuilder_QueryInterface,
3639     UriBuilder_AddRef,
3640     UriBuilder_Release,
3641     UriBuilder_CreateUriSimple,
3642     UriBuilder_CreateUri,
3643     UriBuilder_CreateUriWithFlags,
3644     UriBuilder_GetIUri,
3645     UriBuilder_SetIUri,
3646     UriBuilder_GetFragment,
3647     UriBuilder_GetHost,
3648     UriBuilder_GetPassword,
3649     UriBuilder_GetPath,
3650     UriBuilder_GetPort,
3651     UriBuilder_GetQuery,
3652     UriBuilder_GetSchemeName,
3653     UriBuilder_GetUserName,
3654     UriBuilder_SetFragment,
3655     UriBuilder_SetHost,
3656     UriBuilder_SetPassword,
3657     UriBuilder_SetPath,
3658     UriBuilder_SetPort,
3659     UriBuilder_SetQuery,
3660     UriBuilder_SetSchemeName,
3661     UriBuilder_SetUserName,
3662     UriBuilder_RemoveProperties,
3663     UriBuilder_HasBeenModified,
3664 };
3665
3666 /***********************************************************************
3667  *           CreateIUriBuilder (urlmon.@)
3668  */
3669 HRESULT WINAPI CreateIUriBuilder(IUri *pIUri, DWORD dwFlags, DWORD_PTR dwReserved, IUriBuilder **ppIUriBuilder)
3670 {
3671     UriBuilder *ret;
3672
3673     TRACE("(%p %x %x %p)\n", pIUri, dwFlags, (DWORD)dwReserved, ppIUriBuilder);
3674
3675     ret = heap_alloc(sizeof(UriBuilder));
3676     if(!ret)
3677         return E_OUTOFMEMORY;
3678
3679     ret->lpIUriBuilderVtbl = &UriBuilderVtbl;
3680     ret->ref = 1;
3681
3682     *ppIUriBuilder = URIBUILDER(ret);
3683     return S_OK;
3684 }