| | | 1 | | // Copyright (c) Microsoft Corporation. All rights reserved. |
| | | 2 | | // Licensed under the MIT License. |
| | | 3 | | |
| | | 4 | | using System; |
| | | 5 | | using System.Collections.Generic; |
| | | 6 | | using System.Globalization; |
| | | 7 | | using System.Net; |
| | | 8 | | using System.Text; |
| | | 9 | | |
| | | 10 | | namespace Azure.Storage |
| | | 11 | | { |
| | | 12 | | /// <summary> |
| | | 13 | | /// Extension methods used to manipulate URIs. |
| | | 14 | | /// </summary> |
| | | 15 | | internal static class UriExtensions |
| | | 16 | | { |
| | | 17 | | /// <summary> |
| | | 18 | | /// Append a segment to a URIs path. |
| | | 19 | | /// </summary> |
| | | 20 | | /// <param name="uri">The URI.</param> |
| | | 21 | | /// <param name="segment">The relative segment to append.</param> |
| | | 22 | | /// <returns>The combined URI.</returns> |
| | | 23 | | public static Uri AppendToPath(this Uri uri, string segment) |
| | | 24 | | { |
| | 0 | 25 | | var builder = new UriBuilder(uri); |
| | 0 | 26 | | var path = builder.Path; |
| | 0 | 27 | | var seperator = (path.Length == 0 || path[path.Length - 1] != '/') ? "/" : ""; |
| | | 28 | | // In URLs, the percent sign is used to encode special characters, so if the segment |
| | | 29 | | // has a percent sign in their URL path, we have to encode it before adding it to the path |
| | 0 | 30 | | segment = segment.Replace(Constants.PercentSign, Constants.EncodedPercentSign); |
| | 0 | 31 | | builder.Path += seperator + segment; |
| | 0 | 32 | | return builder.Uri; |
| | | 33 | | } |
| | | 34 | | |
| | | 35 | | /// <summary> |
| | | 36 | | /// Get the (already encoded) query parameters on a URI. |
| | | 37 | | /// </summary> |
| | | 38 | | /// <param name="uri">The URI.</param> |
| | | 39 | | /// <returns>Dictionary mapping query parameters to values.</returns> |
| | | 40 | | public static IDictionary<string, string> GetQueryParameters(this Uri uri) |
| | | 41 | | { |
| | 4 | 42 | | var parameters = new Dictionary<string, string>(); |
| | 4 | 43 | | var query = uri.Query ?? ""; |
| | 4 | 44 | | if (!string.IsNullOrEmpty(query)) |
| | | 45 | | { |
| | 0 | 46 | | if (query.StartsWith("?", true, CultureInfo.InvariantCulture)) |
| | | 47 | | { |
| | 0 | 48 | | query = query.Substring(1); |
| | | 49 | | } |
| | 0 | 50 | | foreach (var param in query.Split(new[] { '&' }, StringSplitOptions.RemoveEmptyEntries)) |
| | | 51 | | { |
| | 0 | 52 | | var parts = param.Split(new[] { '=' }, 2); |
| | 0 | 53 | | var name = WebUtility.UrlDecode(parts[0]); |
| | 0 | 54 | | if (parts.Length == 1) |
| | | 55 | | { |
| | 0 | 56 | | parameters.Add(name, default); |
| | | 57 | | } |
| | | 58 | | else |
| | | 59 | | { |
| | 0 | 60 | | parameters.Add(name, WebUtility.UrlDecode(parts[1])); |
| | | 61 | | } |
| | | 62 | | } |
| | | 63 | | } |
| | 4 | 64 | | return parameters; |
| | | 65 | | } |
| | | 66 | | |
| | | 67 | | /// <summary> |
| | | 68 | | /// Get the account name from the domain portion of a Uri. |
| | | 69 | | /// </summary> |
| | | 70 | | /// <param name="uri">The Uri.</param> |
| | | 71 | | /// <param name="serviceSubDomain">The service subdomain used to validate that the |
| | | 72 | | /// domain is in the expected format. This should be "blob" for blobs, "file" for files, |
| | | 73 | | /// "queue" for queues, "blob" and "dfs" for datalake.</param> |
| | | 74 | | /// <returns>Account name or null if not able to be parsed.</returns> |
| | | 75 | | public static string GetAccountNameFromDomain(this Uri uri, string serviceSubDomain) => |
| | 0 | 76 | | GetAccountNameFromDomain(uri.Host, serviceSubDomain); |
| | | 77 | | |
| | | 78 | | /// <summary> |
| | | 79 | | /// Get the account name from the host. |
| | | 80 | | /// </summary> |
| | | 81 | | /// <param name="host">Host.</param> |
| | | 82 | | /// <param name="serviceSubDomain">The service subdomain used to validate that the |
| | | 83 | | /// domain is in the expected format. This should be "blob" for blobs, "file" for files, |
| | | 84 | | /// "queue" for queues, "blob" and "dfs" for datalake.</param> |
| | | 85 | | /// <returns>Account name or null if not able to be parsed.</returns> |
| | | 86 | | public static string GetAccountNameFromDomain(string host, string serviceSubDomain) |
| | | 87 | | { |
| | 0 | 88 | | var accountEndIndex = host.IndexOf(".", StringComparison.InvariantCulture); |
| | 0 | 89 | | if (accountEndIndex >= 0) |
| | | 90 | | { |
| | 0 | 91 | | var serviceStartIndex = accountEndIndex + 1; |
| | 0 | 92 | | var serviceEndIndex = host.IndexOf(".", serviceStartIndex, StringComparison.InvariantCulture); |
| | 0 | 93 | | if (serviceEndIndex > serviceStartIndex) |
| | | 94 | | { |
| | 0 | 95 | | var service = host.Substring(serviceStartIndex, serviceEndIndex - serviceStartIndex); |
| | 0 | 96 | | if (service == serviceSubDomain) |
| | | 97 | | { |
| | 0 | 98 | | return host.Substring(0, accountEndIndex); |
| | | 99 | | } |
| | | 100 | | } |
| | | 101 | | } |
| | 0 | 102 | | return null; |
| | | 103 | | } |
| | | 104 | | |
| | | 105 | | /// <summary> |
| | | 106 | | /// If path starts with a slash, remove it |
| | | 107 | | /// </summary> |
| | | 108 | | /// <param name="uri">The Uri.</param> |
| | | 109 | | /// <returns>Sanitized Uri.</returns> |
| | | 110 | | public static string GetPath(this Uri uri) => |
| | 0 | 111 | | (uri.AbsolutePath[0] == '/') ? |
| | 0 | 112 | | uri.AbsolutePath.Substring(1) : |
| | 0 | 113 | | uri.AbsolutePath; |
| | | 114 | | |
| | | 115 | | // See remarks at https://docs.microsoft.com/en-us/dotnet/api/system.net.ipaddress.tryparse?view=netframework-4. |
| | | 116 | | /// <summary> |
| | | 117 | | /// Check to see if Uri is using IP Endpoint style. |
| | | 118 | | /// </summary> |
| | | 119 | | /// <param name="uri">The Uri.</param> |
| | | 120 | | /// <returns>True if using IP Endpoint style.</returns> |
| | | 121 | | public static bool IsHostIPEndPointStyle(this Uri uri) => |
| | 0 | 122 | | !string.IsNullOrEmpty(uri.Host) && |
| | 0 | 123 | | uri.Host.IndexOf(".", StringComparison.InvariantCulture) >= 0 && |
| | 0 | 124 | | IPAddress.TryParse(uri.Host, out _); |
| | | 125 | | |
| | | 126 | | /// <summary> |
| | | 127 | | /// Appends a query parameter to the string builder. |
| | | 128 | | /// </summary> |
| | | 129 | | /// <param name="sb">string builder instance.</param> |
| | | 130 | | /// <param name="key">query parameter key.</param> |
| | | 131 | | /// <param name="value">query parameter value.</param> |
| | | 132 | | internal static void AppendQueryParameter(this StringBuilder sb, string key, string value) => |
| | 1860 | 133 | | sb |
| | 1860 | 134 | | .Append(sb.Length > 0 ? "&" : "") |
| | 1860 | 135 | | .Append(key) |
| | 1860 | 136 | | .Append('=') |
| | 1860 | 137 | | .Append(value); |
| | | 138 | | } |
| | | 139 | | } |