| /* |
| * Copyright 2008 Hans Leidekker for CodeWeavers |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| |
| #include "config.h" |
| #include <stdarg.h> |
| |
| #include "wine/debug.h" |
| |
| #include "windef.h" |
| #include "winbase.h" |
| #include "winreg.h" |
| #include "winhttp.h" |
| #include "shlwapi.h" |
| |
| #include "winhttp_private.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(winhttp); |
| |
| static const WCHAR scheme_http[] = {'h','t','t','p',0}; |
| static const WCHAR scheme_https[] = {'h','t','t','p','s',0}; |
| |
| static BOOL set_component( WCHAR **str, DWORD *str_len, WCHAR *value, DWORD len, DWORD flags ) |
| { |
| if (!*str) |
| { |
| if (len && *str_len && (flags & (ICU_DECODE|ICU_ESCAPE))) |
| { |
| set_last_error( ERROR_INVALID_PARAMETER ); |
| return FALSE; |
| } |
| *str = value; |
| *str_len = len; |
| } |
| else |
| { |
| if (len > (*str_len) - 1) |
| { |
| *str_len = len + 1; |
| set_last_error( ERROR_INSUFFICIENT_BUFFER ); |
| return FALSE; |
| } |
| memcpy( *str, value, len * sizeof(WCHAR) ); |
| (*str)[len] = 0; |
| *str_len = len; |
| } |
| return TRUE; |
| } |
| |
| static WCHAR *decode_url( LPCWSTR url, DWORD *len ) |
| { |
| const WCHAR *p = url; |
| WCHAR hex[3], *q, *ret; |
| |
| if (!(ret = heap_alloc( *len * sizeof(WCHAR) ))) return NULL; |
| q = ret; |
| while (*len > 0) |
| { |
| if (p[0] == '%' && isxdigitW( p[1] ) && isxdigitW( p[2] )) |
| { |
| hex[0] = p[1]; |
| hex[1] = p[2]; |
| hex[2] = 0; |
| *q++ = strtolW( hex, NULL, 16 ); |
| p += 3; |
| *len -= 3; |
| } |
| else |
| { |
| *q++ = *p++; |
| *len -= 1; |
| } |
| } |
| *len = q - ret; |
| return ret; |
| } |
| |
| static BOOL need_escape( WCHAR c ) |
| { |
| if (isalnumW( c )) return FALSE; |
| |
| if (c <= 31 || c >= 127) return TRUE; |
| else |
| { |
| switch (c) |
| { |
| case ' ': |
| case '"': |
| case '#': |
| case '%': |
| case '<': |
| case '>': |
| case ']': |
| case '\\': |
| case '[': |
| case '^': |
| case '`': |
| case '{': |
| case '|': |
| case '}': |
| case '~': |
| return TRUE; |
| default: |
| return FALSE; |
| } |
| } |
| } |
| |
| static DWORD copy_escape( WCHAR *dst, const WCHAR *src, DWORD len ) |
| { |
| static const WCHAR hex[] = {'0','1','2','3','4','5','6','7','8','9','A','B','C','D','E','F'}; |
| DWORD ret = len; |
| unsigned int i; |
| WCHAR *p = dst; |
| |
| for (i = 0; i < len; i++, p++) |
| { |
| if (need_escape( src[i] )) |
| { |
| p[0] = '%'; |
| p[1] = hex[(src[i] >> 4) & 0xf]; |
| p[2] = hex[src[i] & 0xf]; |
| ret += 2; |
| p += 2; |
| } |
| else *p = src[i]; |
| } |
| dst[ret] = 0; |
| return ret; |
| } |
| |
| static WCHAR *escape_url( LPCWSTR url, DWORD *len ) |
| { |
| WCHAR *ret; |
| const WCHAR *p, *q; |
| |
| if ((p = q = strrchrW( url, '/' ))) |
| { |
| while (*q) |
| { |
| if (need_escape( *q )) *len += 2; |
| q++; |
| } |
| } |
| if (!(ret = heap_alloc( (*len + 1) * sizeof(WCHAR) ))) return NULL; |
| if (!p) strcpyW( ret, url ); |
| else |
| { |
| memcpy( ret, url, (p - url) * sizeof(WCHAR) ); |
| copy_escape( ret + (p - url), p, q - p ); |
| } |
| return ret; |
| } |
| |
| /*********************************************************************** |
| * WinHttpCrackUrl (winhttp.@) |
| */ |
| BOOL WINAPI WinHttpCrackUrl( LPCWSTR url, DWORD len, DWORD flags, LPURL_COMPONENTSW uc ) |
| { |
| BOOL ret = FALSE; |
| WCHAR *p, *q, *r, *url_decoded = NULL, *url_escaped = NULL; |
| |
| TRACE("%s, %d, %x, %p\n", debugstr_w(url), len, flags, uc); |
| |
| if (!url || !uc || uc->dwStructSize != sizeof(URL_COMPONENTS)) |
| { |
| set_last_error( ERROR_INVALID_PARAMETER ); |
| return FALSE; |
| } |
| if (!len) len = strlenW( url ); |
| |
| if (flags & ICU_ESCAPE) |
| { |
| if (!(url_escaped = escape_url( url, &len ))) |
| { |
| set_last_error( ERROR_OUTOFMEMORY ); |
| return FALSE; |
| } |
| url = url_escaped; |
| } |
| else if (flags & ICU_DECODE) |
| { |
| if (!(url_decoded = decode_url( url, &len ))) |
| { |
| set_last_error( ERROR_OUTOFMEMORY ); |
| return FALSE; |
| } |
| url = url_decoded; |
| } |
| if (!(p = strchrW( url, ':' ))) |
| { |
| set_last_error( ERROR_WINHTTP_UNRECOGNIZED_SCHEME ); |
| return FALSE; |
| } |
| if (p - url == 4 && !strncmpiW( url, scheme_http, 4 )) uc->nScheme = INTERNET_SCHEME_HTTP; |
| else if (p - url == 5 && !strncmpiW( url, scheme_https, 5 )) uc->nScheme = INTERNET_SCHEME_HTTPS; |
| else |
| { |
| set_last_error( ERROR_WINHTTP_UNRECOGNIZED_SCHEME ); |
| goto exit; |
| } |
| if (!(set_component( &uc->lpszScheme, &uc->dwSchemeLength, (WCHAR *)url, p - url, flags ))) goto exit; |
| |
| p++; /* skip ':' */ |
| if (!p[0] || p[0] != '/' || p[1] != '/') goto exit; |
| p += 2; |
| |
| if (!p[0]) goto exit; |
| if ((q = memchrW( p, '@', len - (p - url) )) && !(memchrW( p, '/', q - p ))) |
| { |
| if ((r = memchrW( p, ':', q - p ))) |
| { |
| if (!(set_component( &uc->lpszUserName, &uc->dwUserNameLength, p, r - p, flags ))) goto exit; |
| r++; |
| if (!(set_component( &uc->lpszPassword, &uc->dwPasswordLength, r, q - r, flags ))) goto exit; |
| } |
| else |
| { |
| if (!(set_component( &uc->lpszUserName, &uc->dwUserNameLength, p, q - p, flags ))) goto exit; |
| if (!(set_component( &uc->lpszPassword, &uc->dwPasswordLength, NULL, 0, flags ))) goto exit; |
| } |
| p = q + 1; |
| } |
| else |
| { |
| if (!(set_component( &uc->lpszUserName, &uc->dwUserNameLength, NULL, 0, flags ))) goto exit; |
| if (!(set_component( &uc->lpszPassword, &uc->dwPasswordLength, NULL, 0, flags ))) goto exit; |
| } |
| if ((q = memchrW( p, '/', len - (p - url) ))) |
| { |
| if ((r = memchrW( p, ':', q - p ))) |
| { |
| if (!(set_component( &uc->lpszHostName, &uc->dwHostNameLength, p, r - p, flags ))) goto exit; |
| r++; |
| uc->nPort = atoiW( r ); |
| } |
| else |
| { |
| if (!(set_component( &uc->lpszHostName, &uc->dwHostNameLength, p, q - p, flags ))) goto exit; |
| if (uc->nScheme == INTERNET_SCHEME_HTTP) uc->nPort = INTERNET_DEFAULT_HTTP_PORT; |
| if (uc->nScheme == INTERNET_SCHEME_HTTPS) uc->nPort = INTERNET_DEFAULT_HTTPS_PORT; |
| } |
| |
| if ((r = memchrW( q, '?', len - (q - url) ))) |
| { |
| if (!(set_component( &uc->lpszUrlPath, &uc->dwUrlPathLength, q, r - q, flags ))) goto exit; |
| if (!(set_component( &uc->lpszExtraInfo, &uc->dwExtraInfoLength, r, len - (r - url), flags ))) goto exit; |
| } |
| else |
| { |
| if (!(set_component( &uc->lpszUrlPath, &uc->dwUrlPathLength, q, len - (q - url), flags ))) goto exit; |
| if (!(set_component( &uc->lpszExtraInfo, &uc->dwExtraInfoLength, (WCHAR *)url + len, 0, flags ))) goto exit; |
| } |
| } |
| else |
| { |
| if ((r = memchrW( p, ':', len - (p - url) ))) |
| { |
| if (!(set_component( &uc->lpszHostName, &uc->dwHostNameLength, p, r - p, flags ))) goto exit; |
| r++; |
| uc->nPort = atoiW( r ); |
| } |
| else |
| { |
| if (!(set_component( &uc->lpszHostName, &uc->dwHostNameLength, p, len - (p - url), flags ))) goto exit; |
| if (uc->nScheme == INTERNET_SCHEME_HTTP) uc->nPort = INTERNET_DEFAULT_HTTP_PORT; |
| if (uc->nScheme == INTERNET_SCHEME_HTTPS) uc->nPort = INTERNET_DEFAULT_HTTPS_PORT; |
| } |
| if (!(set_component( &uc->lpszUrlPath, &uc->dwUrlPathLength, (WCHAR *)url + len, 0, flags ))) goto exit; |
| if (!(set_component( &uc->lpszExtraInfo, &uc->dwExtraInfoLength, (WCHAR *)url + len, 0, flags ))) goto exit; |
| } |
| |
| ret = TRUE; |
| |
| TRACE("scheme(%s) host(%s) port(%d) path(%s) extra(%s)\n", |
| debugstr_wn( uc->lpszScheme, uc->dwSchemeLength ), |
| debugstr_wn( uc->lpszHostName, uc->dwHostNameLength ), |
| uc->nPort, |
| debugstr_wn( uc->lpszUrlPath, uc->dwUrlPathLength ), |
| debugstr_wn( uc->lpszExtraInfo, uc->dwExtraInfoLength )); |
| |
| exit: |
| heap_free( url_decoded ); |
| heap_free( url_escaped ); |
| return ret; |
| } |
| |
| static INTERNET_SCHEME get_scheme( const WCHAR *scheme, DWORD len ) |
| { |
| if (!strncmpW( scheme, scheme_http, len )) return INTERNET_SCHEME_HTTP; |
| if (!strncmpW( scheme, scheme_https, len )) return INTERNET_SCHEME_HTTPS; |
| return 0; |
| } |
| |
| static const WCHAR *get_scheme_string( INTERNET_SCHEME scheme ) |
| { |
| if (scheme == INTERNET_SCHEME_HTTP) return scheme_http; |
| if (scheme == INTERNET_SCHEME_HTTPS) return scheme_https; |
| return NULL; |
| } |
| |
| static BOOL uses_default_port( INTERNET_SCHEME scheme, INTERNET_PORT port ) |
| { |
| if ((scheme == INTERNET_SCHEME_HTTP) && (port == INTERNET_DEFAULT_HTTP_PORT)) return TRUE; |
| if ((scheme == INTERNET_SCHEME_HTTPS) && (port == INTERNET_DEFAULT_HTTPS_PORT)) return TRUE; |
| return FALSE; |
| } |
| |
| static DWORD comp_length( DWORD len, DWORD flags, WCHAR *comp ) |
| { |
| DWORD ret; |
| unsigned int i; |
| |
| ret = len ? len : strlenW( comp ); |
| if (!(flags & ICU_ESCAPE)) return ret; |
| for (i = 0; i < len; i++) if (need_escape( comp[i] )) ret += 2; |
| return ret; |
| } |
| |
| static BOOL calc_length( URL_COMPONENTS *uc, DWORD flags, LPDWORD len ) |
| { |
| static const WCHAR formatW[] = {'%','u',0}; |
| INTERNET_SCHEME scheme; |
| |
| *len = 0; |
| if (uc->lpszScheme) |
| { |
| DWORD scheme_len = comp_length( uc->dwSchemeLength, 0, uc->lpszScheme ); |
| *len += scheme_len; |
| scheme = get_scheme( uc->lpszScheme, scheme_len ); |
| } |
| else |
| { |
| scheme = uc->nScheme; |
| if (!scheme) scheme = INTERNET_SCHEME_HTTP; |
| *len += strlenW( get_scheme_string( scheme ) ); |
| } |
| *len += 1; /* ':' */ |
| if (uc->lpszHostName) *len += 2; /* "//" */ |
| |
| if (uc->lpszUserName) |
| { |
| *len += comp_length( uc->dwUserNameLength, 0, uc->lpszUserName ); |
| *len += 1; /* "@" */ |
| } |
| else |
| { |
| if (uc->lpszPassword) |
| { |
| set_last_error( ERROR_INVALID_PARAMETER ); |
| return FALSE; |
| } |
| } |
| if (uc->lpszPassword) |
| { |
| *len += 1; /* ":" */ |
| *len += comp_length( uc->dwPasswordLength, 0, uc->lpszPassword ); |
| } |
| if (uc->lpszHostName) |
| { |
| *len += comp_length( uc->dwHostNameLength, 0, uc->lpszHostName ); |
| |
| if (!uses_default_port( scheme, uc->nPort )) |
| { |
| WCHAR port[sizeof("65535")]; |
| |
| sprintfW( port, formatW, uc->nPort ); |
| *len += strlenW( port ); |
| *len += 1; /* ":" */ |
| } |
| if (uc->lpszUrlPath && *uc->lpszUrlPath != '/') *len += 1; /* '/' */ |
| } |
| if (uc->lpszUrlPath) *len += comp_length( uc->dwUrlPathLength, flags, uc->lpszUrlPath ); |
| if (uc->lpszExtraInfo) *len += comp_length( uc->dwExtraInfoLength, flags, uc->lpszExtraInfo ); |
| return TRUE; |
| } |
| |
| /*********************************************************************** |
| * WinHttpCreateUrl (winhttp.@) |
| */ |
| BOOL WINAPI WinHttpCreateUrl( LPURL_COMPONENTS uc, DWORD flags, LPWSTR url, LPDWORD required ) |
| { |
| static const WCHAR formatW[] = {'%','u',0}; |
| static const WCHAR twoslashW[] = {'/','/'}; |
| |
| DWORD len; |
| INTERNET_SCHEME scheme; |
| |
| TRACE("%p, 0x%08x, %p, %p\n", uc, flags, url, required); |
| |
| if (!uc || uc->dwStructSize != sizeof(URL_COMPONENTS) || !required || !url) |
| { |
| set_last_error( ERROR_INVALID_PARAMETER ); |
| return FALSE; |
| } |
| |
| if (!calc_length( uc, flags, &len )) return FALSE; |
| |
| if (*required < len) |
| { |
| *required = len + 1; |
| set_last_error( ERROR_INSUFFICIENT_BUFFER ); |
| return FALSE; |
| } |
| |
| url[0] = 0; |
| *required = len; |
| if (uc->lpszScheme) |
| { |
| len = comp_length( uc->dwSchemeLength, 0, uc->lpszScheme ); |
| memcpy( url, uc->lpszScheme, len * sizeof(WCHAR) ); |
| url += len; |
| |
| scheme = get_scheme( uc->lpszScheme, len ); |
| } |
| else |
| { |
| const WCHAR *schemeW; |
| scheme = uc->nScheme; |
| |
| if (!scheme) scheme = INTERNET_SCHEME_HTTP; |
| |
| schemeW = get_scheme_string( scheme ); |
| len = strlenW( schemeW ); |
| memcpy( url, schemeW, len * sizeof(WCHAR) ); |
| url += len; |
| } |
| |
| /* all schemes are followed by at least a colon */ |
| *url = ':'; |
| url++; |
| |
| if (uc->lpszHostName) |
| { |
| memcpy( url, twoslashW, sizeof(twoslashW) ); |
| url += sizeof(twoslashW) / sizeof(twoslashW[0]); |
| } |
| if (uc->lpszUserName) |
| { |
| len = comp_length( uc->dwUserNameLength, 0, uc->lpszUserName ); |
| memcpy( url, uc->lpszUserName, len * sizeof(WCHAR) ); |
| url += len; |
| |
| if (uc->lpszPassword) |
| { |
| *url = ':'; |
| url++; |
| |
| len = comp_length( uc->dwPasswordLength, 0, uc->lpszPassword ); |
| memcpy( url, uc->lpszPassword, len * sizeof(WCHAR) ); |
| url += len; |
| } |
| *url = '@'; |
| url++; |
| } |
| if (uc->lpszHostName) |
| { |
| len = comp_length( uc->dwHostNameLength, 0, uc->lpszHostName ); |
| memcpy( url, uc->lpszHostName, len * sizeof(WCHAR) ); |
| url += len; |
| |
| if (!uses_default_port( scheme, uc->nPort )) |
| { |
| WCHAR port[sizeof("65535")]; |
| |
| sprintfW( port, formatW, uc->nPort ); |
| *url = ':'; |
| url++; |
| |
| len = strlenW( port ); |
| memcpy( url, port, len * sizeof(WCHAR) ); |
| url += len; |
| } |
| |
| /* add slash between hostname and path if necessary */ |
| if (uc->lpszUrlPath && *uc->lpszUrlPath != '/') |
| { |
| *url = '/'; |
| url++; |
| } |
| } |
| if (uc->lpszUrlPath) |
| { |
| len = comp_length( uc->dwUrlPathLength, 0, uc->lpszUrlPath ); |
| if (flags & ICU_ESCAPE) url += copy_escape( url, uc->lpszUrlPath, len ); |
| else |
| { |
| memcpy( url, uc->lpszUrlPath, len * sizeof(WCHAR) ); |
| url += len; |
| } |
| } |
| if (uc->lpszExtraInfo) |
| { |
| len = comp_length( uc->dwExtraInfoLength, 0, uc->lpszExtraInfo ); |
| if (flags & ICU_ESCAPE) url += copy_escape( url, uc->lpszExtraInfo, len ); |
| else |
| { |
| memcpy( url, uc->lpszExtraInfo, len * sizeof(WCHAR) ); |
| url += len; |
| } |
| } |
| *url = 0; |
| return TRUE; |
| } |
