| | | 1 | | // Licensed to the .NET Foundation under one or more agreements. |
| | | 2 | | // The .NET Foundation licenses this file to you under the MIT license. |
| | | 3 | | |
| | | 4 | | using System.Collections; |
| | | 5 | | using System.Collections.Generic; |
| | | 6 | | using System.Diagnostics; |
| | | 7 | | using System.Diagnostics.CodeAnalysis; |
| | | 8 | | using System.Runtime.CompilerServices; |
| | | 9 | | using System.Runtime.InteropServices; |
| | | 10 | | using System.Text; |
| | | 11 | | using System.Threading; |
| | | 12 | | |
| | | 13 | | namespace System.Net.Http.Headers |
| | | 14 | | { |
| | | 15 | | /// <summary> |
| | | 16 | | /// Key/value pairs of headers. The value is either a raw <see cref="string"/> or a <see cref="HttpHeaders.HeaderSto |
| | | 17 | | /// We're using a custom type instead of <see cref="KeyValuePair{TKey, TValue}"/> because we need ref access to fiel |
| | | 18 | | /// </summary> |
| | | 19 | | internal struct HeaderEntry |
| | | 20 | | { |
| | | 21 | | public HeaderDescriptor Key; |
| | | 22 | | public object Value; |
| | | 23 | | |
| | | 24 | | public HeaderEntry(HeaderDescriptor key, object value) |
| | | 25 | | { |
| | | 26 | | Key = key; |
| | | 27 | | Value = value; |
| | | 28 | | } |
| | | 29 | | } |
| | | 30 | | |
| | | 31 | | public abstract class HttpHeaders : IEnumerable<KeyValuePair<string, IEnumerable<string>>> |
| | | 32 | | { |
| | | 33 | | // This type is used to store a collection of headers in 'headerStore': |
| | | 34 | | // - A header can have multiple values. |
| | | 35 | | // - A header can have an associated parser which is able to parse the raw string value into a strongly typed ob |
| | | 36 | | // - If a header has an associated parser and the provided raw value can't be parsed, the value is considered |
| | | 37 | | // invalid. Invalid values are stored if added using TryAddWithoutValidation(). If the value was added using A |
| | | 38 | | // Add() will throw FormatException. |
| | | 39 | | // - Since parsing header values is expensive and users usually only care about a few headers, header values are |
| | | 40 | | // lazily initialized. |
| | | 41 | | // |
| | | 42 | | // Given the properties above, a header value can have three states: |
| | | 43 | | // - 'raw': The header value was added using TryAddWithoutValidation() and it wasn't parsed yet. |
| | | 44 | | // - 'parsed': The header value was successfully parsed. It was either added using Add() where the value was par |
| | | 45 | | // immediately, or if added using TryAddWithoutValidation() a user already accessed a property/method triggeri |
| | | 46 | | // value to be parsed. |
| | | 47 | | // - 'invalid': The header value was parsed, but parsing failed because the value is invalid. Storing invalid va |
| | | 48 | | // allows users to still retrieve the value (by calling GetValues()), but it will not be exposed as strongly t |
| | | 49 | | // object. E.g. the client receives a response with the following header: 'Via: 1.1 proxy, invalid' |
| | | 50 | | // - HttpHeaders.GetValues() will return "1.1 proxy", "invalid" |
| | | 51 | | // - HttpResponseHeaders.Via collection will only contain one ViaHeaderValue object with value "1.1 proxy" |
| | | 52 | | |
| | | 53 | | /// <summary>Either a <see cref="HeaderEntry"/> array or a Dictionary<<see cref="HeaderDescriptor"/>, <see cr |
| | | 54 | | private object? _headerStore; |
| | | 55 | | private int _count; |
| | | 56 | | |
| | | 57 | | private readonly HttpHeaderType _allowedHeaderTypes; |
| | | 58 | | private readonly HttpHeaderType _treatAsCustomHeaderTypes; |
| | | 59 | | |
| | | 60 | | protected HttpHeaders() |
| | 0 | 61 | | : this(HttpHeaderType.All, HttpHeaderType.None) |
| | 0 | 62 | | { |
| | 0 | 63 | | } |
| | | 64 | | |
| | 3 | 65 | | internal HttpHeaders(HttpHeaderType allowedHeaderTypes, HttpHeaderType treatAsCustomHeaderTypes) |
| | 3 | 66 | | { |
| | | 67 | | // Should be no overlap |
| | 3 | 68 | | Debug.Assert((allowedHeaderTypes & treatAsCustomHeaderTypes) == 0); |
| | | 69 | | |
| | 3 | 70 | | _allowedHeaderTypes = allowedHeaderTypes & ~HttpHeaderType.NonTrailing; |
| | 3 | 71 | | _treatAsCustomHeaderTypes = treatAsCustomHeaderTypes & ~HttpHeaderType.NonTrailing; |
| | 3 | 72 | | } |
| | | 73 | | |
| | | 74 | | /// <summary>Gets a view of the contents of this headers collection that does not parse nor validate the data up |
| | 25107 | 75 | | public HttpHeadersNonValidated NonValidated => new HttpHeadersNonValidated(this); |
| | | 76 | | |
| | 38969 | 77 | | public void Add(string name, string? value) => Add(GetHeaderDescriptor(name), value); |
| | | 78 | | |
| | | 79 | | internal void Add(HeaderDescriptor descriptor, string? value) |
| | 38969 | 80 | | { |
| | 38969 | 81 | | if (descriptor.Parser is null) |
| | 14111 | 82 | | { |
| | | 83 | | // If the header has no parser, we only have to check for new lines or null. |
| | 14111 | 84 | | CheckIsAllowedHeaderName(descriptor); |
| | 14111 | 85 | | CheckContainsNewLineOrNull(value); |
| | 10406 | 86 | | TryAddWithoutValidation(descriptor, value); |
| | 10406 | 87 | | return; |
| | | 88 | | } |
| | | 89 | | |
| | | 90 | | // We don't use GetOrCreateHeaderInfo() here, since this would create a new header in the store. If parsing |
| | | 91 | | // the value then throws, we would have to remove the header from the store again. So just get a |
| | | 92 | | // HeaderStoreItemInfo object and try to parse the value. If it works, we'll add the header. |
| | 24858 | 93 | | PrepareHeaderInfoForAdd(descriptor, out HeaderStoreItemInfo info, out bool addToStore); |
| | 24858 | 94 | | ParseAndAddValue(descriptor, info, value); |
| | | 95 | | |
| | | 96 | | // If we get here, then the value could be parsed correctly. If we created a new HeaderStoreItemInfo, add |
| | | 97 | | // it to the store if we added at least one value. |
| | 16899 | 98 | | if (addToStore && (info.ParsedAndInvalidValues != null)) |
| | 8541 | 99 | | { |
| | 8541 | 100 | | info.AssertContainsNoInvalidValues(); |
| | 8541 | 101 | | Debug.Assert(!Contains(descriptor)); |
| | 8541 | 102 | | AddEntryToStore(new HeaderEntry(descriptor, info)); |
| | 8541 | 103 | | } |
| | 27305 | 104 | | } |
| | | 105 | | |
| | 0 | 106 | | public void Add(string name, IEnumerable<string?> values) => Add(GetHeaderDescriptor(name), values); |
| | | 107 | | |
| | | 108 | | internal void Add(HeaderDescriptor descriptor, IEnumerable<string?> values) |
| | 0 | 109 | | { |
| | 0 | 110 | | ArgumentNullException.ThrowIfNull(values); |
| | | 111 | | |
| | | 112 | | // It's relatively common to only add a single value with this overload, especially when copying |
| | | 113 | | // between HttpHeaders collections. Avoid boxing the enumerator and possibly a HeaderStoreItemInfo |
| | | 114 | | // allocation by deferring to the overload for a single value instead. |
| | 0 | 115 | | if (values is IList<string?> { Count: 1 } valuesList) |
| | 0 | 116 | | { |
| | 0 | 117 | | Add(descriptor, valuesList[0]); |
| | 0 | 118 | | return; |
| | | 119 | | } |
| | | 120 | | |
| | 0 | 121 | | PrepareHeaderInfoForAdd(descriptor, out HeaderStoreItemInfo info, out bool addToStore); |
| | | 122 | | |
| | | 123 | | try |
| | 0 | 124 | | { |
| | | 125 | | // Note that if the first couple of values are valid followed by an invalid value, the valid values |
| | | 126 | | // will be added to the store before the exception for the invalid value is thrown. |
| | 0 | 127 | | if (descriptor.Parser is null) |
| | 0 | 128 | | { |
| | 0 | 129 | | foreach (string? value in values) |
| | 0 | 130 | | { |
| | | 131 | | // If the header has no parser, we only have to check for new lines or null. |
| | 0 | 132 | | CheckContainsNewLineOrNull(value); |
| | 0 | 133 | | AddParsedValue(info, value ?? string.Empty); |
| | 0 | 134 | | } |
| | 0 | 135 | | } |
| | | 136 | | else |
| | 0 | 137 | | { |
| | 0 | 138 | | foreach (string? value in values) |
| | 0 | 139 | | { |
| | 0 | 140 | | ParseAndAddValue(descriptor, info, value); |
| | 0 | 141 | | } |
| | 0 | 142 | | } |
| | 0 | 143 | | } |
| | | 144 | | finally |
| | 0 | 145 | | { |
| | | 146 | | // Even if one of the values was invalid, make sure we add the header for the valid ones. We need to be |
| | | 147 | | // consistent here: If values get added to an _existing_ header, then all values until the invalid one |
| | | 148 | | // get added. Same here: If multiple values get added to a _new_ header, make sure the header gets added |
| | | 149 | | // with the valid values. |
| | | 150 | | // However, if all values for a _new_ header were invalid, then don't add the header. |
| | 0 | 151 | | if (addToStore && (info.ParsedAndInvalidValues != null)) |
| | 0 | 152 | | { |
| | 0 | 153 | | info.AssertContainsNoInvalidValues(); |
| | 0 | 154 | | Debug.Assert(!Contains(descriptor)); |
| | 0 | 155 | | AddEntryToStore(new HeaderEntry(descriptor, info)); |
| | 0 | 156 | | } |
| | 0 | 157 | | } |
| | 0 | 158 | | } |
| | | 159 | | |
| | | 160 | | public bool TryAddWithoutValidation(string name, string? value) => |
| | 37662 | 161 | | TryGetHeaderDescriptor(name, out HeaderDescriptor descriptor) && |
| | 37662 | 162 | | TryAddWithoutValidation(descriptor, value); |
| | | 163 | | |
| | | 164 | | internal bool TryAddWithoutValidation(HeaderDescriptor descriptor, string? value) |
| | 35513 | 165 | | { |
| | | 166 | | // Normalize null values to be empty values, which are allowed. If the user adds multiple |
| | | 167 | | // null/empty values, all of them are added to the collection. This will result in delimiter-only |
| | | 168 | | // values, e.g. adding two null-strings (or empty, or whitespace-only) results in "My-Header: ,". |
| | 35513 | 169 | | value ??= string.Empty; |
| | | 170 | | |
| | 35513 | 171 | | ref object? storeValueRef = ref GetValueRefOrAddDefault(descriptor); |
| | 35513 | 172 | | object? currentValue = storeValueRef; |
| | | 173 | | |
| | 35513 | 174 | | if (currentValue is null) |
| | 30310 | 175 | | { |
| | 30310 | 176 | | storeValueRef = value; |
| | 30310 | 177 | | } |
| | | 178 | | else |
| | 5203 | 179 | | { |
| | 5203 | 180 | | if (currentValue is not HeaderStoreItemInfo info) |
| | 5203 | 181 | | { |
| | | 182 | | // The header store contained a single raw string value, so promote it |
| | | 183 | | // to being a HeaderStoreItemInfo and add to it. |
| | 5203 | 184 | | Debug.Assert(currentValue is string); |
| | 5203 | 185 | | storeValueRef = info = new HeaderStoreItemInfo() { RawValue = currentValue }; |
| | 5203 | 186 | | } |
| | | 187 | | |
| | 5203 | 188 | | AddRawValue(info, value); |
| | 5203 | 189 | | } |
| | | 190 | | |
| | 35513 | 191 | | return true; |
| | 35513 | 192 | | } |
| | | 193 | | |
| | | 194 | | public bool TryAddWithoutValidation(string name, IEnumerable<string?> values) => |
| | 0 | 195 | | TryGetHeaderDescriptor(name, out HeaderDescriptor descriptor) && |
| | 0 | 196 | | TryAddWithoutValidation(descriptor, values); |
| | | 197 | | |
| | | 198 | | internal bool TryAddWithoutValidation(HeaderDescriptor descriptor, IEnumerable<string?> values) |
| | 0 | 199 | | { |
| | 0 | 200 | | ArgumentNullException.ThrowIfNull(values); |
| | | 201 | | |
| | 0 | 202 | | if (values is IList<string?> valuesList) |
| | 0 | 203 | | { |
| | 0 | 204 | | int count = valuesList.Count; |
| | | 205 | | |
| | 0 | 206 | | if (count > 0) |
| | 0 | 207 | | { |
| | | 208 | | // The store value is either a string (a single unparsed value) or a HeaderStoreItemInfo. |
| | | 209 | | // The RawValue on HeaderStoreItemInfo can likewise be either a single string or a List<string>. |
| | | 210 | | |
| | 0 | 211 | | ref object? storeValueRef = ref GetValueRefOrAddDefault(descriptor); |
| | 0 | 212 | | object? storeValue = storeValueRef; |
| | | 213 | | |
| | | 214 | | // If the storeValue was already set or we're adding more than 1 value, |
| | | 215 | | // we'll have to store the values in a List<string> on HeaderStoreItemInfo. |
| | 0 | 216 | | if (storeValue is not null || count > 1) |
| | 0 | 217 | | { |
| | 0 | 218 | | if (storeValue is not HeaderStoreItemInfo info) |
| | 0 | 219 | | { |
| | 0 | 220 | | storeValueRef = info = new HeaderStoreItemInfo { RawValue = storeValue }; |
| | 0 | 221 | | } |
| | | 222 | | |
| | 0 | 223 | | object? rawValue = info.RawValue; |
| | 0 | 224 | | if (rawValue is not List<string> rawValues) |
| | 0 | 225 | | { |
| | 0 | 226 | | info.RawValue = rawValues = new List<string>(); |
| | | 227 | | |
| | 0 | 228 | | if (rawValue != null) |
| | 0 | 229 | | { |
| | 0 | 230 | | rawValues.EnsureCapacity(count + 1); |
| | 0 | 231 | | rawValues.Add((string)rawValue); |
| | 0 | 232 | | } |
| | 0 | 233 | | } |
| | | 234 | | |
| | 0 | 235 | | rawValues.EnsureCapacity(rawValues.Count + count); |
| | | 236 | | |
| | 0 | 237 | | for (int i = 0; i < count; i++) |
| | 0 | 238 | | { |
| | 0 | 239 | | rawValues.Add(valuesList[i] ?? string.Empty); |
| | 0 | 240 | | } |
| | 0 | 241 | | } |
| | | 242 | | else |
| | 0 | 243 | | { |
| | | 244 | | // We're adding a single value to a new header entry. We can store the unparsed value as-is. |
| | 0 | 245 | | storeValueRef = valuesList[0] ?? string.Empty; |
| | 0 | 246 | | } |
| | 0 | 247 | | } |
| | 0 | 248 | | } |
| | | 249 | | else |
| | 0 | 250 | | { |
| | 0 | 251 | | foreach (string? value in values) |
| | 0 | 252 | | { |
| | 0 | 253 | | TryAddWithoutValidation(descriptor, value ?? string.Empty); |
| | 0 | 254 | | } |
| | 0 | 255 | | } |
| | | 256 | | |
| | 0 | 257 | | return true; |
| | 0 | 258 | | } |
| | | 259 | | |
| | 0 | 260 | | public IEnumerable<string> GetValues(string name) => GetValues(GetHeaderDescriptor(name)); |
| | | 261 | | |
| | | 262 | | internal IEnumerable<string> GetValues(HeaderDescriptor descriptor) |
| | 0 | 263 | | { |
| | 0 | 264 | | if (TryGetValues(descriptor, out IEnumerable<string>? values)) |
| | 0 | 265 | | { |
| | 0 | 266 | | return values; |
| | | 267 | | } |
| | | 268 | | |
| | 0 | 269 | | throw new InvalidOperationException(SR.net_http_headers_not_found); |
| | 0 | 270 | | } |
| | | 271 | | |
| | | 272 | | public bool TryGetValues(string name, [NotNullWhen(true)] out IEnumerable<string>? values) |
| | 0 | 273 | | { |
| | 0 | 274 | | if (TryGetHeaderDescriptor(name, out HeaderDescriptor descriptor)) |
| | 0 | 275 | | { |
| | 0 | 276 | | return TryGetValues(descriptor, out values); |
| | | 277 | | } |
| | | 278 | | |
| | 0 | 279 | | values = null; |
| | 0 | 280 | | return false; |
| | 0 | 281 | | } |
| | | 282 | | |
| | | 283 | | internal bool TryGetValues(HeaderDescriptor descriptor, [NotNullWhen(true)] out IEnumerable<string>? values) |
| | 0 | 284 | | { |
| | 0 | 285 | | ref object storeValueRef = ref GetValueRefOrNullRef(descriptor); |
| | 0 | 286 | | if (!Unsafe.IsNullRef(ref storeValueRef)) |
| | 0 | 287 | | { |
| | 0 | 288 | | object value = storeValueRef; |
| | | 289 | | |
| | 0 | 290 | | if (value is not HeaderStoreItemInfo info) |
| | 0 | 291 | | { |
| | 0 | 292 | | if (descriptor.Parser is null) |
| | 0 | 293 | | { |
| | | 294 | | // This is a custom header without a known parser, so unparsed values won't change. |
| | | 295 | | // Avoid allocating the HeaderStoreItemInfo and just return the raw value as-is. |
| | 0 | 296 | | values = new string[] { (string)value }; |
| | 0 | 297 | | return true; |
| | | 298 | | } |
| | | 299 | | |
| | 0 | 300 | | info = ReplaceWithHeaderStoreItemInfo(ref storeValueRef, value); |
| | 0 | 301 | | } |
| | | 302 | | |
| | 0 | 303 | | ParseRawHeaderValues(descriptor, info); |
| | 0 | 304 | | values = GetStoreValuesAsStringArray(descriptor, info); |
| | 0 | 305 | | return true; |
| | | 306 | | } |
| | | 307 | | |
| | 0 | 308 | | values = null; |
| | 0 | 309 | | return false; |
| | 0 | 310 | | } |
| | | 311 | | |
| | 0 | 312 | | public bool Contains(string name) => Contains(GetHeaderDescriptor(name)); |
| | | 313 | | |
| | | 314 | | public override string ToString() |
| | 0 | 315 | | { |
| | 0 | 316 | | var vsb = new ValueStringBuilder(stackalloc char[512]); |
| | 0 | 317 | | Dump(ref vsb, indentLines: false); |
| | 0 | 318 | | return vsb.ToString(); |
| | 0 | 319 | | } |
| | | 320 | | |
| | | 321 | | internal void Dump(ref ValueStringBuilder builder, bool indentLines) |
| | 0 | 322 | | { |
| | | 323 | | // Return all headers as string similar to: |
| | | 324 | | // HeaderName1: Value1, Value2 |
| | | 325 | | // HeaderName2: Value1 |
| | | 326 | | // ... |
| | | 327 | | |
| | 0 | 328 | | foreach (HeaderEntry entry in GetEntries()) |
| | 0 | 329 | | { |
| | 0 | 330 | | if (indentLines) |
| | 0 | 331 | | { |
| | 0 | 332 | | builder.Append(" "); |
| | 0 | 333 | | } |
| | | 334 | | |
| | 0 | 335 | | builder.Append(entry.Key.Name); |
| | 0 | 336 | | builder.Append(": "); |
| | | 337 | | |
| | 0 | 338 | | GetStoreValuesAsStringOrStringArray(entry.Key, entry.Value, out string? singleValue, out string[]? multi |
| | 0 | 339 | | Debug.Assert(singleValue is not null ^ multiValue is not null); |
| | | 340 | | |
| | 0 | 341 | | if (singleValue is not null) |
| | 0 | 342 | | { |
| | 0 | 343 | | builder.Append(singleValue); |
| | 0 | 344 | | } |
| | | 345 | | else |
| | 0 | 346 | | { |
| | | 347 | | // Note that if we get multiple values for a header that doesn't support multiple values, we'll |
| | | 348 | | // just separate the values using a comma (default separator). |
| | 0 | 349 | | string separator = entry.Key.Separator; |
| | | 350 | | |
| | 0 | 351 | | Debug.Assert(multiValue is not null && multiValue.Length > 0); |
| | 0 | 352 | | builder.Append(multiValue[0]); |
| | 0 | 353 | | for (int i = 1; i < multiValue.Length; i++) |
| | 0 | 354 | | { |
| | 0 | 355 | | builder.Append(separator); |
| | 0 | 356 | | builder.Append(multiValue[i]); |
| | 0 | 357 | | } |
| | 0 | 358 | | } |
| | | 359 | | |
| | 0 | 360 | | builder.Append(Environment.NewLine); |
| | 0 | 361 | | } |
| | 0 | 362 | | } |
| | | 363 | | |
| | | 364 | | internal string GetHeaderString(HeaderDescriptor descriptor) |
| | 0 | 365 | | { |
| | 0 | 366 | | if (TryGetHeaderValue(descriptor, out object? info)) |
| | 0 | 367 | | { |
| | 0 | 368 | | GetStoreValuesAsStringOrStringArray(descriptor, info, out string? singleValue, out string[]? multiValue) |
| | 0 | 369 | | Debug.Assert(singleValue is not null ^ multiValue is not null); |
| | | 370 | | |
| | 0 | 371 | | if (singleValue is not null) |
| | 0 | 372 | | { |
| | 0 | 373 | | return singleValue; |
| | | 374 | | } |
| | | 375 | | |
| | | 376 | | // Note that if we get multiple values for a header that doesn't support multiple values, we'll |
| | | 377 | | // just separate the values using a comma (default separator). |
| | 0 | 378 | | return string.Join(descriptor.Separator, multiValue!); |
| | | 379 | | } |
| | | 380 | | |
| | 0 | 381 | | return string.Empty; |
| | 0 | 382 | | } |
| | | 383 | | |
| | | 384 | | #region IEnumerable<KeyValuePair<string, IEnumerable<string>>> Members |
| | | 385 | | |
| | 50214 | 386 | | public IEnumerator<KeyValuePair<string, IEnumerable<string>>> GetEnumerator() => _count == 0 ? |
| | 50214 | 387 | | ((IEnumerable<KeyValuePair<string, IEnumerable<string>>>)Array.Empty<KeyValuePair<string, IEnumerable<st |
| | 50214 | 388 | | GetEnumeratorCore(); |
| | | 389 | | |
| | | 390 | | private IEnumerator<KeyValuePair<string, IEnumerable<string>>> GetEnumeratorCore() |
| | 38851 | 391 | | { |
| | 38851 | 392 | | Debug.Assert(_headerStore is not null); |
| | | 393 | | |
| | 38851 | 394 | | HeaderEntry[]? entries = GetEntriesArray(); |
| | 38851 | 395 | | Debug.Assert(_count != 0 && entries is not null, "Caller should have validated the collection is not empty") |
| | | 396 | | |
| | 155404 | 397 | | for (int i = 0; i < _count; i++) |
| | 38851 | 398 | | { |
| | 38851 | 399 | | HeaderEntry entry = entries[i]; |
| | | 400 | | |
| | 38851 | 401 | | if (entry.Value is not HeaderStoreItemInfo info) |
| | 25107 | 402 | | { |
| | 25107 | 403 | | if (entry.Key.Parser is null) |
| | 8908 | 404 | | { |
| | | 405 | | // This is a custom header without a known parser, so unparsed values won't change. |
| | | 406 | | // Avoid allocating the HeaderStoreItemInfo and just return the raw value as-is. |
| | 8908 | 407 | | yield return new KeyValuePair<string, IEnumerable<string>>(entry.Key.Name, new string[] { (strin |
| | 8908 | 408 | | continue; |
| | | 409 | | } |
| | | 410 | | |
| | | 411 | | // To retain consistent semantics, we need to upgrade a raw string to a HeaderStoreItemInfo |
| | | 412 | | // during enumeration so that we can parse the raw value in order to a) return |
| | | 413 | | // the correct set of parsed values, and b) update the instance for subsequent enumerations |
| | | 414 | | // to reflect that parsing. |
| | 16199 | 415 | | ref object storeValueRef = ref EntriesAreLiveView |
| | 16199 | 416 | | ? ref entries[i].Value |
| | 16199 | 417 | | : ref CollectionsMarshal.GetValueRefOrNullRef((Dictionary<HeaderDescriptor, object>)_headerStore |
| | | 418 | | |
| | 16199 | 419 | | info = ReplaceWithHeaderStoreItemInfo(ref storeValueRef, entry.Value); |
| | 16199 | 420 | | } |
| | | 421 | | |
| | | 422 | | // Make sure we parse all raw values before returning the result. Note that this has to be |
| | | 423 | | // done before we calculate the array length (next line): A raw value may contain a list of |
| | | 424 | | // values. |
| | 29943 | 425 | | ParseRawHeaderValues(entry.Key, info); |
| | | 426 | | |
| | 29943 | 427 | | string[] values = GetStoreValuesAsStringArray(entry.Key, info); |
| | 29943 | 428 | | yield return new KeyValuePair<string, IEnumerable<string>>(entry.Key.Name, values); |
| | 29943 | 429 | | } |
| | 38851 | 430 | | } |
| | | 431 | | |
| | | 432 | | #endregion |
| | | 433 | | |
| | | 434 | | #region IEnumerable Members |
| | | 435 | | |
| | 0 | 436 | | Collections.IEnumerator Collections.IEnumerable.GetEnumerator() => GetEnumerator(); |
| | | 437 | | |
| | | 438 | | #endregion |
| | | 439 | | |
| | | 440 | | internal void AddParsedValue(HeaderDescriptor descriptor, object value) |
| | 0 | 441 | | { |
| | 0 | 442 | | Debug.Assert(value != null); |
| | 0 | 443 | | Debug.Assert(descriptor.Parser != null, "Can't add parsed value if there is no parser available."); |
| | | 444 | | |
| | 0 | 445 | | HeaderStoreItemInfo info = GetOrCreateHeaderInfo(descriptor); |
| | | 446 | | |
| | | 447 | | // If the current header has only one value, we can't add another value. The strongly typed property |
| | | 448 | | // must not call AddParsedValue(), but SetParsedValue(). E.g. for headers like 'Date', 'Host'. |
| | 0 | 449 | | Debug.Assert(descriptor.Parser.SupportsMultipleValues, $"Header '{descriptor.Name}' doesn't support multiple |
| | | 450 | | |
| | 0 | 451 | | AddParsedValue(info, value); |
| | 0 | 452 | | } |
| | | 453 | | |
| | | 454 | | internal void SetParsedValue(HeaderDescriptor descriptor, object value) |
| | 0 | 455 | | { |
| | 0 | 456 | | Debug.Assert(value != null); |
| | 0 | 457 | | Debug.Assert(descriptor.Parser != null, "Can't add parsed value if there is no parser available."); |
| | | 458 | | |
| | | 459 | | // This method will first clear all values. This is used e.g. when setting the 'Date' or 'Host' header. |
| | | 460 | | // i.e. headers not supporting collections. |
| | 0 | 461 | | HeaderStoreItemInfo info = GetOrCreateHeaderInfo(descriptor); |
| | | 462 | | |
| | 0 | 463 | | info.ParsedAndInvalidValues = null; |
| | 0 | 464 | | info.RawValue = null; |
| | | 465 | | |
| | 0 | 466 | | AddParsedValue(info, value); |
| | 0 | 467 | | } |
| | | 468 | | |
| | | 469 | | internal void SetOrRemoveParsedValue(HeaderDescriptor descriptor, object? value) |
| | 0 | 470 | | { |
| | 0 | 471 | | if (value == null) |
| | 0 | 472 | | { |
| | 0 | 473 | | Remove(descriptor); |
| | 0 | 474 | | } |
| | | 475 | | else |
| | 0 | 476 | | { |
| | 0 | 477 | | SetParsedValue(descriptor, value); |
| | 0 | 478 | | } |
| | 0 | 479 | | } |
| | | 480 | | |
| | 0 | 481 | | public bool Remove(string name) => Remove(GetHeaderDescriptor(name)); |
| | | 482 | | |
| | | 483 | | internal bool RemoveParsedValue(HeaderDescriptor descriptor, object value, bool removeAll = false) |
| | 0 | 484 | | { |
| | 0 | 485 | | Debug.Assert(value != null); |
| | | 486 | | |
| | | 487 | | // If we have a value for this header, then verify if we have a single value. If so, compare that |
| | | 488 | | // value with 'item'. If we have a list of values, then remove 'item' from the list. |
| | 0 | 489 | | if (TryGetAndParseHeaderInfo(descriptor, out HeaderStoreItemInfo? info)) |
| | 0 | 490 | | { |
| | 0 | 491 | | Debug.Assert(descriptor.Parser != null, "Can't add parsed value if there is no parser available."); |
| | 0 | 492 | | Debug.Assert(descriptor.Parser.SupportsMultipleValues, |
| | 0 | 493 | | "This method should not be used for single-value headers. Use Remove(string) instead."); |
| | | 494 | | |
| | | 495 | | // If there is no entry, just return. |
| | 0 | 496 | | var parsedValue = info.ParsedAndInvalidValues; |
| | 0 | 497 | | if (parsedValue == null) |
| | 0 | 498 | | { |
| | 0 | 499 | | return false; |
| | | 500 | | } |
| | | 501 | | |
| | 0 | 502 | | bool result = false; |
| | 0 | 503 | | IEqualityComparer? comparer = descriptor.Parser.Comparer; |
| | | 504 | | |
| | 0 | 505 | | List<object>? parsedValues = parsedValue as List<object>; |
| | 0 | 506 | | if (parsedValues == null) |
| | 0 | 507 | | { |
| | 0 | 508 | | if (parsedValue is not InvalidValue) |
| | 0 | 509 | | { |
| | 0 | 510 | | Debug.Assert(parsedValue.GetType() == value.GetType(), |
| | 0 | 511 | | "Stored value does not have the same type as 'value'."); |
| | | 512 | | |
| | 0 | 513 | | if (AreEqual(value, parsedValue, comparer)) |
| | 0 | 514 | | { |
| | 0 | 515 | | info.ParsedAndInvalidValues = null; |
| | 0 | 516 | | result = true; |
| | 0 | 517 | | } |
| | 0 | 518 | | } |
| | 0 | 519 | | } |
| | | 520 | | else |
| | 0 | 521 | | { |
| | 0 | 522 | | for (int i = 0; i < parsedValues.Count; i++) |
| | 0 | 523 | | { |
| | 0 | 524 | | object item = parsedValues[i]; |
| | 0 | 525 | | if (item is not InvalidValue) |
| | 0 | 526 | | { |
| | 0 | 527 | | Debug.Assert(item.GetType() == value.GetType(), |
| | 0 | 528 | | "One of the stored values does not have the same type as 'value'."); |
| | | 529 | | |
| | 0 | 530 | | if (AreEqual(value, item, comparer)) |
| | 0 | 531 | | { |
| | 0 | 532 | | parsedValues.RemoveAt(i); |
| | 0 | 533 | | i--; |
| | | 534 | | |
| | 0 | 535 | | if (!result) |
| | 0 | 536 | | { |
| | 0 | 537 | | result = true; |
| | | 538 | | |
| | 0 | 539 | | if (!removeAll) |
| | 0 | 540 | | { |
| | 0 | 541 | | break; |
| | | 542 | | } |
| | 0 | 543 | | } |
| | | 544 | | else |
| | 0 | 545 | | { |
| | | 546 | | // We've removed a second item. Fallback to RemoveAll in case there are more to main |
| | | 547 | | // Create a copy of the locals to avoid the capture allocation in the common case. |
| | 0 | 548 | | object valueLocal = value; |
| | 0 | 549 | | IEqualityComparer? comparerLocal = comparer; |
| | 0 | 550 | | parsedValues.RemoveAll(item => item is not InvalidValue && AreEqual(valueLocal, item |
| | 0 | 551 | | break; |
| | | 552 | | } |
| | 0 | 553 | | } |
| | 0 | 554 | | } |
| | 0 | 555 | | } |
| | | 556 | | |
| | | 557 | | // If we removed the last item in a list, remove the list. |
| | 0 | 558 | | if (parsedValues.Count == 0) |
| | 0 | 559 | | { |
| | 0 | 560 | | info.AssertContainsNoInvalidValues(); |
| | 0 | 561 | | info.ParsedAndInvalidValues = null; |
| | 0 | 562 | | } |
| | 0 | 563 | | } |
| | | 564 | | |
| | | 565 | | // If there is no value for the header left, remove the header. |
| | 0 | 566 | | if (info.IsEmpty) |
| | 0 | 567 | | { |
| | 0 | 568 | | bool headerRemoved = Remove(descriptor); |
| | 0 | 569 | | Debug.Assert(headerRemoved, $"Existing header '{descriptor.Name}' couldn't be removed."); |
| | 0 | 570 | | } |
| | | 571 | | |
| | 0 | 572 | | return result; |
| | | 573 | | } |
| | | 574 | | |
| | 0 | 575 | | return false; |
| | 0 | 576 | | } |
| | | 577 | | |
| | | 578 | | internal bool ContainsParsedValue(HeaderDescriptor descriptor, object value) |
| | 0 | 579 | | { |
| | 0 | 580 | | Debug.Assert(value != null); |
| | | 581 | | |
| | | 582 | | // If we have a value for this header, then verify if we have a single value. If so, compare that |
| | | 583 | | // value with 'item'. If we have a list of values, then compare each item in the list with 'item'. |
| | 0 | 584 | | if (TryGetAndParseHeaderInfo(descriptor, out HeaderStoreItemInfo? info)) |
| | 0 | 585 | | { |
| | 0 | 586 | | Debug.Assert(descriptor.Parser != null, "Can't add parsed value if there is no parser available."); |
| | 0 | 587 | | Debug.Assert(descriptor.Parser.SupportsMultipleValues, |
| | 0 | 588 | | "This method should not be used for single-value headers. Use equality comparer instead."); |
| | | 589 | | |
| | | 590 | | // If there is no entry, just return. |
| | 0 | 591 | | var parsedValue = info.ParsedAndInvalidValues; |
| | 0 | 592 | | if (parsedValue == null) |
| | 0 | 593 | | { |
| | 0 | 594 | | return false; |
| | | 595 | | } |
| | | 596 | | |
| | 0 | 597 | | List<object>? parsedValues = parsedValue as List<object>; |
| | | 598 | | |
| | 0 | 599 | | IEqualityComparer? comparer = descriptor.Parser.Comparer; |
| | | 600 | | |
| | 0 | 601 | | if (parsedValues == null) |
| | 0 | 602 | | { |
| | 0 | 603 | | if (parsedValue is not InvalidValue) |
| | 0 | 604 | | { |
| | 0 | 605 | | Debug.Assert(parsedValue.GetType() == value.GetType(), |
| | 0 | 606 | | "Stored value does not have the same type as 'value'."); |
| | | 607 | | |
| | 0 | 608 | | return AreEqual(value, parsedValue, comparer); |
| | | 609 | | } |
| | 0 | 610 | | } |
| | | 611 | | else |
| | 0 | 612 | | { |
| | 0 | 613 | | foreach (object item in parsedValues) |
| | 0 | 614 | | { |
| | 0 | 615 | | if (item is not InvalidValue) |
| | 0 | 616 | | { |
| | 0 | 617 | | Debug.Assert(item.GetType() == value.GetType(), |
| | 0 | 618 | | "One of the stored values does not have the same type as 'value'."); |
| | | 619 | | |
| | 0 | 620 | | if (AreEqual(value, item, comparer)) |
| | 0 | 621 | | { |
| | 0 | 622 | | return true; |
| | | 623 | | } |
| | 0 | 624 | | } |
| | 0 | 625 | | } |
| | | 626 | | |
| | 0 | 627 | | return false; |
| | | 628 | | } |
| | 0 | 629 | | } |
| | | 630 | | |
| | 0 | 631 | | return false; |
| | 0 | 632 | | } |
| | | 633 | | |
| | | 634 | | internal virtual void AddHeaders(HttpHeaders sourceHeaders) |
| | 0 | 635 | | { |
| | 0 | 636 | | Debug.Assert(sourceHeaders != null); |
| | 0 | 637 | | Debug.Assert(GetType() == sourceHeaders.GetType(), "Can only copy headers from an instance of the same type. |
| | | 638 | | |
| | | 639 | | // Only add header values if they're not already set on the message. Note that we don't merge |
| | | 640 | | // collections: If both the default headers and the message have set some values for a certain |
| | | 641 | | // header, then we don't try to merge the values. |
| | 0 | 642 | | if (_count == 0 && sourceHeaders._headerStore is HeaderEntry[] sourceEntries) |
| | 0 | 643 | | { |
| | | 644 | | // If the target collection is empty, we don't have to search for existing values |
| | 0 | 645 | | _count = sourceHeaders._count; |
| | 0 | 646 | | if (_headerStore is not HeaderEntry[] entries || entries.Length < _count) |
| | 0 | 647 | | { |
| | 0 | 648 | | entries = new HeaderEntry[sourceEntries.Length]; |
| | 0 | 649 | | _headerStore = entries; |
| | 0 | 650 | | } |
| | | 651 | | |
| | 0 | 652 | | for (int i = 0; i < _count && i < sourceEntries.Length; i++) |
| | 0 | 653 | | { |
| | 0 | 654 | | HeaderEntry entry = sourceEntries[i]; |
| | 0 | 655 | | if (entry.Value is HeaderStoreItemInfo info) |
| | 0 | 656 | | { |
| | 0 | 657 | | entry.Value = CloneHeaderInfo(entry.Key, info); |
| | 0 | 658 | | } |
| | 0 | 659 | | entries[i] = entry; |
| | 0 | 660 | | } |
| | 0 | 661 | | } |
| | | 662 | | else |
| | 0 | 663 | | { |
| | 0 | 664 | | foreach (HeaderEntry entry in sourceHeaders.GetEntries()) |
| | 0 | 665 | | { |
| | 0 | 666 | | ref object? storeValueRef = ref GetValueRefOrAddDefault(entry.Key); |
| | 0 | 667 | | if (storeValueRef is null) |
| | 0 | 668 | | { |
| | 0 | 669 | | object sourceValue = entry.Value; |
| | 0 | 670 | | if (sourceValue is HeaderStoreItemInfo info) |
| | 0 | 671 | | { |
| | 0 | 672 | | storeValueRef = CloneHeaderInfo(entry.Key, info); |
| | 0 | 673 | | } |
| | | 674 | | else |
| | 0 | 675 | | { |
| | 0 | 676 | | Debug.Assert(sourceValue is string); |
| | 0 | 677 | | storeValueRef = sourceValue; |
| | 0 | 678 | | } |
| | 0 | 679 | | } |
| | 0 | 680 | | } |
| | 0 | 681 | | } |
| | 0 | 682 | | } |
| | | 683 | | |
| | | 684 | | private static HeaderStoreItemInfo CloneHeaderInfo(HeaderDescriptor descriptor, HeaderStoreItemInfo sourceInfo) |
| | 0 | 685 | | { |
| | 0 | 686 | | lock (sourceInfo) |
| | 0 | 687 | | { |
| | 0 | 688 | | var destinationInfo = new HeaderStoreItemInfo |
| | 0 | 689 | | { |
| | 0 | 690 | | // Always copy raw values |
| | 0 | 691 | | RawValue = CloneStringHeaderInfoValues(sourceInfo.RawValue) |
| | 0 | 692 | | }; |
| | | 693 | | |
| | 0 | 694 | | if (descriptor.Parser == null) |
| | 0 | 695 | | { |
| | 0 | 696 | | sourceInfo.AssertContainsNoInvalidValues(); |
| | 0 | 697 | | destinationInfo.ParsedAndInvalidValues = CloneStringHeaderInfoValues(sourceInfo.ParsedAndInvalidValu |
| | 0 | 698 | | } |
| | | 699 | | else |
| | 0 | 700 | | { |
| | | 701 | | // We have a parser, so we also have to clone invalid values and parsed values. |
| | 0 | 702 | | if (sourceInfo.ParsedAndInvalidValues != null) |
| | 0 | 703 | | { |
| | 0 | 704 | | List<object>? sourceValues = sourceInfo.ParsedAndInvalidValues as List<object>; |
| | 0 | 705 | | if (sourceValues == null) |
| | 0 | 706 | | { |
| | 0 | 707 | | CloneAndAddValue(destinationInfo, sourceInfo.ParsedAndInvalidValues); |
| | 0 | 708 | | } |
| | | 709 | | else |
| | 0 | 710 | | { |
| | 0 | 711 | | foreach (object item in sourceValues) |
| | 0 | 712 | | { |
| | 0 | 713 | | CloneAndAddValue(destinationInfo, item); |
| | 0 | 714 | | } |
| | 0 | 715 | | } |
| | 0 | 716 | | } |
| | 0 | 717 | | } |
| | | 718 | | |
| | 0 | 719 | | return destinationInfo; |
| | | 720 | | } |
| | 0 | 721 | | } |
| | | 722 | | |
| | | 723 | | private static void CloneAndAddValue(HeaderStoreItemInfo destinationInfo, object source) |
| | 0 | 724 | | { |
| | | 725 | | // We only have one value. Clone it and assign it to the store. |
| | 0 | 726 | | if (source is ICloneable cloneableValue) |
| | 0 | 727 | | { |
| | 0 | 728 | | Debug.Assert(source is not InvalidValue); |
| | 0 | 729 | | AddParsedValue(destinationInfo, cloneableValue.Clone()); |
| | 0 | 730 | | } |
| | | 731 | | else |
| | 0 | 732 | | { |
| | | 733 | | // If it doesn't implement ICloneable, it's a value type or an immutable type like String/Uri. |
| | 0 | 734 | | AddParsedValue(destinationInfo, source); |
| | 0 | 735 | | } |
| | 0 | 736 | | } |
| | | 737 | | |
| | | 738 | | [return: NotNullIfNotNull(nameof(source))] |
| | | 739 | | private static object? CloneStringHeaderInfoValues(object? source) |
| | 0 | 740 | | { |
| | 0 | 741 | | if (source == null) |
| | 0 | 742 | | { |
| | 0 | 743 | | return null; |
| | | 744 | | } |
| | | 745 | | |
| | 0 | 746 | | List<object>? sourceValues = source as List<object>; |
| | 0 | 747 | | if (sourceValues == null) |
| | 0 | 748 | | { |
| | | 749 | | // If we just have one value, return the reference to the string (strings are immutable so it's OK |
| | | 750 | | // to use the reference). |
| | 0 | 751 | | return source; |
| | | 752 | | } |
| | | 753 | | else |
| | 0 | 754 | | { |
| | | 755 | | // If we have a list of strings, create a new list and copy all strings to the new list. |
| | 0 | 756 | | return new List<object>(sourceValues); |
| | | 757 | | } |
| | 0 | 758 | | } |
| | | 759 | | |
| | | 760 | | private HeaderStoreItemInfo GetOrCreateHeaderInfo(HeaderDescriptor descriptor) |
| | 0 | 761 | | { |
| | 0 | 762 | | if (TryGetAndParseHeaderInfo(descriptor, out HeaderStoreItemInfo? info)) |
| | 0 | 763 | | { |
| | 0 | 764 | | return info; |
| | | 765 | | } |
| | | 766 | | else |
| | 0 | 767 | | { |
| | 0 | 768 | | return CreateAndAddHeaderToStore(descriptor); |
| | | 769 | | } |
| | 0 | 770 | | } |
| | | 771 | | |
| | | 772 | | private HeaderStoreItemInfo CreateAndAddHeaderToStore(HeaderDescriptor descriptor) |
| | 0 | 773 | | { |
| | 0 | 774 | | Debug.Assert(!Contains(descriptor)); |
| | | 775 | | |
| | | 776 | | // If we don't have the header in the store yet, add it now. |
| | 0 | 777 | | HeaderStoreItemInfo result = new HeaderStoreItemInfo(); |
| | | 778 | | |
| | | 779 | | // If the descriptor header type is in _treatAsCustomHeaderTypes, it must be converted to a custom header be |
| | 0 | 780 | | Debug.Assert((descriptor.HeaderType & _treatAsCustomHeaderTypes) == 0); |
| | | 781 | | |
| | 0 | 782 | | AddEntryToStore(new HeaderEntry(descriptor, result)); |
| | | 783 | | |
| | 0 | 784 | | return result; |
| | 0 | 785 | | } |
| | | 786 | | |
| | | 787 | | internal bool TryGetHeaderValue(HeaderDescriptor descriptor, [NotNullWhen(true)] out object? value) |
| | 0 | 788 | | { |
| | 0 | 789 | | ref object storeValueRef = ref GetValueRefOrNullRef(descriptor); |
| | 0 | 790 | | if (Unsafe.IsNullRef(ref storeValueRef)) |
| | 0 | 791 | | { |
| | 0 | 792 | | value = null; |
| | 0 | 793 | | return false; |
| | | 794 | | } |
| | | 795 | | else |
| | 0 | 796 | | { |
| | 0 | 797 | | value = storeValueRef; |
| | 0 | 798 | | return true; |
| | | 799 | | } |
| | 0 | 800 | | } |
| | | 801 | | |
| | | 802 | | private bool TryGetAndParseHeaderInfo(HeaderDescriptor key, [NotNullWhen(true)] out HeaderStoreItemInfo? info) |
| | 24858 | 803 | | { |
| | 24858 | 804 | | ref object storeValueRef = ref GetValueRefOrNullRef(key); |
| | 24858 | 805 | | if (!Unsafe.IsNullRef(ref storeValueRef)) |
| | 8541 | 806 | | { |
| | 8541 | 807 | | object value = storeValueRef; |
| | | 808 | | |
| | 8541 | 809 | | info = value is HeaderStoreItemInfo hsi |
| | 8541 | 810 | | ? hsi |
| | 8541 | 811 | | : ReplaceWithHeaderStoreItemInfo(ref storeValueRef, value); |
| | | 812 | | |
| | 8541 | 813 | | ParseRawHeaderValues(key, info); |
| | 8541 | 814 | | return true; |
| | | 815 | | } |
| | | 816 | | |
| | 16317 | 817 | | info = null; |
| | 16317 | 818 | | return false; |
| | 24858 | 819 | | } |
| | | 820 | | |
| | | 821 | | /// <summary> |
| | | 822 | | /// Replaces <paramref name="storeValueRef"/> with a new <see cref="HeaderStoreItemInfo"/>, |
| | | 823 | | /// or returns the existing <see cref="HeaderStoreItemInfo"/> if a different thread beat us to it. |
| | | 824 | | /// </summary> |
| | | 825 | | /// <remarks> |
| | | 826 | | /// This helper should be used any time we're upgrading a storage slot from an unparsed string to a HeaderStoreI |
| | | 827 | | /// Concurrent writes to the header collection are UB, so we don't need to worry about race conditions when doin |
| | | 828 | | /// </remarks> |
| | | 829 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 830 | | private static HeaderStoreItemInfo ReplaceWithHeaderStoreItemInfo(ref object storeValueRef, object value) |
| | 16199 | 831 | | { |
| | 16199 | 832 | | Debug.Assert(value is string); |
| | | 833 | | |
| | 16199 | 834 | | var info = new HeaderStoreItemInfo() { RawValue = value }; |
| | 16199 | 835 | | object previousValue = Interlocked.CompareExchange(ref storeValueRef, info, value); |
| | | 836 | | |
| | 16199 | 837 | | if (ReferenceEquals(previousValue, value)) |
| | 16199 | 838 | | { |
| | 16199 | 839 | | return info; |
| | | 840 | | } |
| | | 841 | | |
| | | 842 | | // Rare race condition: Another thread replaced the value with a HeaderStoreItemInfo. |
| | 0 | 843 | | return (HeaderStoreItemInfo)previousValue; |
| | 16199 | 844 | | } |
| | | 845 | | |
| | | 846 | | private static void ParseRawHeaderValues(HeaderDescriptor descriptor, HeaderStoreItemInfo info) |
| | 38484 | 847 | | { |
| | | 848 | | // Unlike TryGetHeaderInfo() this method tries to parse all non-validated header values (if any) |
| | | 849 | | // before returning to the caller. |
| | 38484 | 850 | | lock (info) |
| | 38484 | 851 | | { |
| | 38484 | 852 | | Debug.Assert(!info.IsEmpty); |
| | 38484 | 853 | | if (info.RawValue != null) |
| | 21402 | 854 | | { |
| | 21402 | 855 | | if (info.RawValue is List<string> rawValues) |
| | 5203 | 856 | | { |
| | 36421 | 857 | | foreach (string rawValue in rawValues) |
| | 10406 | 858 | | { |
| | 10406 | 859 | | ParseSingleRawHeaderValue(info, descriptor, rawValue); |
| | 10406 | 860 | | } |
| | 5203 | 861 | | } |
| | | 862 | | else |
| | 16199 | 863 | | { |
| | 16199 | 864 | | string? rawValue = info.RawValue as string; |
| | 16199 | 865 | | Debug.Assert(rawValue is not null); |
| | 16199 | 866 | | ParseSingleRawHeaderValue(info, descriptor, rawValue); |
| | 16199 | 867 | | } |
| | | 868 | | |
| | | 869 | | // At this point all values are either in info.ParsedValue, info.InvalidValue. Reset RawValue. |
| | 21402 | 870 | | Debug.Assert(info.ParsedAndInvalidValues is not null); |
| | 21402 | 871 | | info.RawValue = null; |
| | 21402 | 872 | | } |
| | 38484 | 873 | | } |
| | 38484 | 874 | | } |
| | | 875 | | |
| | | 876 | | private static void ParseSingleRawHeaderValue(HeaderStoreItemInfo info, HeaderDescriptor descriptor, string rawV |
| | 26605 | 877 | | { |
| | 26605 | 878 | | Debug.Assert(Monitor.IsEntered(info)); |
| | 26605 | 879 | | if (descriptor.Parser == null) |
| | 10406 | 880 | | { |
| | 10406 | 881 | | if (HttpRuleParser.ContainsNewLineOrNull(rawValue)) |
| | 0 | 882 | | { |
| | 0 | 883 | | if (NetEventSource.Log.IsEnabled()) NetEventSource.Error(null, SR.Format(SR.net_http_log_headers_no_ |
| | 0 | 884 | | AddInvalidValue(info, rawValue); |
| | 0 | 885 | | } |
| | | 886 | | else |
| | 10406 | 887 | | { |
| | 10406 | 888 | | AddParsedValue(info, rawValue); |
| | 10406 | 889 | | } |
| | 10406 | 890 | | } |
| | | 891 | | else |
| | 16199 | 892 | | { |
| | 16199 | 893 | | if (!TryParseAndAddRawHeaderValue(descriptor, info, rawValue, true)) |
| | 7540 | 894 | | { |
| | 7540 | 895 | | if (NetEventSource.Log.IsEnabled()) NetEventSource.Log.HeadersInvalidValue(descriptor.Name, rawValue |
| | 7540 | 896 | | } |
| | 16199 | 897 | | } |
| | 26605 | 898 | | } |
| | | 899 | | |
| | | 900 | | // See Add(name, string) |
| | | 901 | | internal bool TryParseAndAddValue(HeaderDescriptor descriptor, string? value) |
| | 0 | 902 | | { |
| | | 903 | | // We don't use GetOrCreateHeaderInfo() here, since this would create a new header in the store. If parsing |
| | | 904 | | // the value then throws, we would have to remove the header from the store again. So just get a |
| | | 905 | | // HeaderStoreItemInfo object and try to parse the value. If it works, we'll add the header. |
| | | 906 | | HeaderStoreItemInfo info; |
| | | 907 | | bool addToStore; |
| | 0 | 908 | | PrepareHeaderInfoForAdd(descriptor, out info, out addToStore); |
| | | 909 | | |
| | 0 | 910 | | bool result = TryParseAndAddRawHeaderValue(descriptor, info, value, false); |
| | | 911 | | |
| | 0 | 912 | | if (result && addToStore && (info.ParsedAndInvalidValues != null)) |
| | 0 | 913 | | { |
| | 0 | 914 | | info.AssertContainsNoInvalidValues(); |
| | | 915 | | // If we get here, then the value could be parsed correctly. If we created a new HeaderStoreItemInfo, ad |
| | | 916 | | // it to the store if we added at least one value. |
| | 0 | 917 | | Debug.Assert(!Contains(descriptor)); |
| | 0 | 918 | | AddEntryToStore(new HeaderEntry(descriptor, info)); |
| | 0 | 919 | | } |
| | | 920 | | |
| | 0 | 921 | | return result; |
| | 0 | 922 | | } |
| | | 923 | | |
| | | 924 | | // See ParseAndAddValue |
| | | 925 | | private static bool TryParseAndAddRawHeaderValue(HeaderDescriptor descriptor, HeaderStoreItemInfo info, string? |
| | 16199 | 926 | | { |
| | 16199 | 927 | | Debug.Assert(info != null); |
| | 16199 | 928 | | Debug.Assert(descriptor.Parser != null); |
| | | 929 | | |
| | | 930 | | // Values are added as 'invalid' if we either can't parse the value OR if we already have a value |
| | | 931 | | // and the current header doesn't support multiple values: e.g. trying to add a date/time value |
| | | 932 | | // to the 'Date' header if we already have a date/time value will result in the second value being |
| | | 933 | | // added to the 'invalid' header values. |
| | 16199 | 934 | | if (!info.CanAddParsedValue(descriptor.Parser)) |
| | 0 | 935 | | { |
| | 0 | 936 | | if (addWhenInvalid) |
| | 0 | 937 | | { |
| | 0 | 938 | | AddInvalidValue(info, value ?? string.Empty); |
| | 0 | 939 | | } |
| | 0 | 940 | | return false; |
| | | 941 | | } |
| | | 942 | | |
| | 16199 | 943 | | int index = 0; |
| | | 944 | | |
| | 16199 | 945 | | if (descriptor.Parser.TryParseValue(value, info.ParsedAndInvalidValues, ref index, out object? parsedValue)) |
| | 10093 | 946 | | { |
| | | 947 | | // The raw string only represented one value (which was successfully parsed). Add the value and return. |
| | 10093 | 948 | | if ((value == null) || (index == value.Length)) |
| | 2925 | 949 | | { |
| | 2925 | 950 | | if (parsedValue != null) |
| | 2807 | 951 | | { |
| | 2807 | 952 | | AddParsedValue(info, parsedValue); |
| | 2807 | 953 | | } |
| | 118 | 954 | | else if (addWhenInvalid && info.ParsedAndInvalidValues is null) |
| | 118 | 955 | | { |
| | 118 | 956 | | AddInvalidValue(info, value ?? string.Empty); |
| | 118 | 957 | | } |
| | 2925 | 958 | | return true; |
| | | 959 | | } |
| | 7168 | 960 | | Debug.Assert(index < value.Length, "Parser must return an index value within the string length."); |
| | | 961 | | |
| | | 962 | | // If we successfully parsed a value, but there are more left to read, store the results in a temp |
| | | 963 | | // list. Only when all values are parsed successfully write the list to the store. |
| | 7168 | 964 | | List<object> parsedValues = new List<object>(); |
| | 7168 | 965 | | if (parsedValue != null) |
| | 7168 | 966 | | { |
| | 7168 | 967 | | parsedValues.Add(parsedValue); |
| | 7168 | 968 | | } |
| | | 969 | | |
| | 261837 | 970 | | while (index < value.Length) |
| | 256103 | 971 | | { |
| | 256103 | 972 | | if (descriptor.Parser.TryParseValue(value, info.ParsedAndInvalidValues, ref index, out parsedValue)) |
| | 254669 | 973 | | { |
| | 254669 | 974 | | if (parsedValue != null) |
| | 254669 | 975 | | { |
| | 254669 | 976 | | parsedValues.Add(parsedValue); |
| | 254669 | 977 | | } |
| | 254669 | 978 | | } |
| | | 979 | | else |
| | 1434 | 980 | | { |
| | 1434 | 981 | | if (addWhenInvalid) |
| | 1434 | 982 | | { |
| | 1434 | 983 | | AddInvalidValue(info, value); |
| | 1434 | 984 | | } |
| | 1434 | 985 | | return false; |
| | | 986 | | } |
| | 254669 | 987 | | } |
| | | 988 | | |
| | | 989 | | // All values were parsed correctly. Copy results to the store. |
| | 499862 | 990 | | foreach (object item in parsedValues) |
| | 241330 | 991 | | { |
| | 241330 | 992 | | AddParsedValue(info, item); |
| | 241330 | 993 | | } |
| | | 994 | | |
| | 5734 | 995 | | if (parsedValues.Count == 0 && addWhenInvalid && info.ParsedAndInvalidValues is null) |
| | 0 | 996 | | { |
| | 0 | 997 | | AddInvalidValue(info, value); |
| | 0 | 998 | | } |
| | | 999 | | |
| | 5734 | 1000 | | return true; |
| | | 1001 | | } |
| | | 1002 | | |
| | 6106 | 1003 | | Debug.Assert(value != null); |
| | 6106 | 1004 | | if (addWhenInvalid) |
| | 6106 | 1005 | | { |
| | 6106 | 1006 | | AddInvalidValue(info, value ?? string.Empty); |
| | 6106 | 1007 | | } |
| | 6106 | 1008 | | return false; |
| | 16199 | 1009 | | } |
| | | 1010 | | |
| | | 1011 | | private static void AddParsedValue(HeaderStoreItemInfo info, object value) |
| | 740732 | 1012 | | { |
| | 740732 | 1013 | | Debug.Assert(!(value is List<object>), |
| | 740732 | 1014 | | "Header value types must not derive from List<object> since this type is used internally to store " + |
| | 740732 | 1015 | | "lists of values. So we would not be able to distinguish between a single value and a list of values."); |
| | | 1016 | | |
| | 740732 | 1017 | | AddValueToStoreValue<object>(value, ref info.ParsedAndInvalidValues); |
| | 740732 | 1018 | | } |
| | | 1019 | | |
| | | 1020 | | private static void AddInvalidValue(HeaderStoreItemInfo info, string value) |
| | 7658 | 1021 | | { |
| | 7658 | 1022 | | AddValueToStoreValue<object>(new InvalidValue(value), ref info.ParsedAndInvalidValues); |
| | 7658 | 1023 | | } |
| | | 1024 | | |
| | | 1025 | | private static void AddRawValue(HeaderStoreItemInfo info, string value) |
| | 5203 | 1026 | | { |
| | 5203 | 1027 | | AddValueToStoreValue<string>(value, ref info.RawValue); |
| | 5203 | 1028 | | } |
| | | 1029 | | |
| | | 1030 | | private static void AddValueToStoreValue<T>(T value, ref object? currentStoreValue) where T : class |
| | 753593 | 1031 | | { |
| | | 1032 | | // If there is no value set yet, then add current item as value (we don't create a list |
| | | 1033 | | // if not required). If 'info.Value' is already assigned then make sure 'info.Value' is a |
| | | 1034 | | // List<T> and append 'item' to the list. |
| | 753593 | 1035 | | if (currentStoreValue == null) |
| | 29943 | 1036 | | { |
| | 29943 | 1037 | | currentStoreValue = value; |
| | 29943 | 1038 | | } |
| | | 1039 | | else |
| | 723650 | 1040 | | { |
| | 723650 | 1041 | | List<T>? storeValues = currentStoreValue as List<T>; |
| | | 1042 | | |
| | 723650 | 1043 | | if (storeValues == null) |
| | 22596 | 1044 | | { |
| | 22596 | 1045 | | storeValues = new List<T>(2); |
| | 22596 | 1046 | | Debug.Assert(currentStoreValue is T); |
| | 22596 | 1047 | | storeValues.Add((T)currentStoreValue); |
| | 22596 | 1048 | | currentStoreValue = storeValues; |
| | 22596 | 1049 | | } |
| | 723650 | 1050 | | Debug.Assert(value is T); |
| | 723650 | 1051 | | storeValues.Add((T)value); |
| | 723650 | 1052 | | } |
| | 753593 | 1053 | | } |
| | | 1054 | | |
| | | 1055 | | internal object? GetSingleParsedValue(HeaderDescriptor descriptor) |
| | 0 | 1056 | | { |
| | 0 | 1057 | | if (!TryGetAndParseHeaderInfo(descriptor, out HeaderStoreItemInfo? info)) |
| | 0 | 1058 | | { |
| | 0 | 1059 | | return null; |
| | | 1060 | | } |
| | | 1061 | | |
| | 0 | 1062 | | return info.GetSingleParsedValue(); |
| | 0 | 1063 | | } |
| | | 1064 | | |
| | | 1065 | | internal object? GetParsedAndInvalidValues(HeaderDescriptor descriptor) |
| | 0 | 1066 | | { |
| | 0 | 1067 | | if (!TryGetAndParseHeaderInfo(descriptor, out HeaderStoreItemInfo? info)) |
| | 0 | 1068 | | { |
| | 0 | 1069 | | return null; |
| | | 1070 | | } |
| | | 1071 | | |
| | 0 | 1072 | | return info.ParsedAndInvalidValues; |
| | 0 | 1073 | | } |
| | | 1074 | | |
| | 19846 | 1075 | | internal virtual bool IsAllowedHeaderName(HeaderDescriptor descriptor) => true; |
| | | 1076 | | |
| | | 1077 | | private void CheckIsAllowedHeaderName(HeaderDescriptor descriptor) |
| | 38969 | 1078 | | { |
| | 38969 | 1079 | | if (!IsAllowedHeaderName(descriptor)) |
| | 0 | 1080 | | { |
| | 0 | 1081 | | throw new InvalidOperationException(SR.Format(SR.net_http_headers_not_allowed_header_name, descriptor.Na |
| | | 1082 | | } |
| | 38969 | 1083 | | } |
| | | 1084 | | |
| | | 1085 | | private void PrepareHeaderInfoForAdd(HeaderDescriptor descriptor, out HeaderStoreItemInfo info, out bool addToSt |
| | 24858 | 1086 | | { |
| | 24858 | 1087 | | CheckIsAllowedHeaderName(descriptor); |
| | | 1088 | | |
| | 24858 | 1089 | | addToStore = false; |
| | 24858 | 1090 | | if (!TryGetAndParseHeaderInfo(descriptor, out info!)) |
| | 16317 | 1091 | | { |
| | 16317 | 1092 | | info = new HeaderStoreItemInfo(); |
| | 16317 | 1093 | | addToStore = true; |
| | 16317 | 1094 | | } |
| | 24858 | 1095 | | } |
| | | 1096 | | |
| | | 1097 | | private static void ParseAndAddValue(HeaderDescriptor descriptor, HeaderStoreItemInfo info, string? value) |
| | 24858 | 1098 | | { |
| | 24858 | 1099 | | Debug.Assert(info != null); |
| | 24858 | 1100 | | Debug.Assert(descriptor.Parser != null); |
| | | 1101 | | |
| | | 1102 | | // If the header only supports 1 value, we can add the current value only if there is no |
| | | 1103 | | // value already set. |
| | 24858 | 1104 | | if (!info.CanAddParsedValue(descriptor.Parser)) |
| | 419 | 1105 | | { |
| | 419 | 1106 | | throw new FormatException(SR.Format(System.Globalization.CultureInfo.InvariantCulture, SR.net_http_heade |
| | | 1107 | | } |
| | | 1108 | | |
| | 24439 | 1109 | | int index = 0; |
| | 24439 | 1110 | | object parsedValue = descriptor.Parser.ParseValue(value, info.ParsedAndInvalidValues, ref index); |
| | | 1111 | | |
| | | 1112 | | // The raw string only represented one value (which was successfully parsed). Add the value and return. |
| | | 1113 | | // If value is null we still have to first call ParseValue() to allow the parser to decide whether null is |
| | | 1114 | | // a valid value. If it is (i.e. no exception thrown), we set the parsed value (if any) and return. |
| | 18333 | 1115 | | if ((value == null) || (index == value.Length)) |
| | 5431 | 1116 | | { |
| | | 1117 | | // If the returned value is null, then it means the header accepts empty values. i.e. we don't throw |
| | | 1118 | | // but we don't add 'null' to the store either. |
| | 5431 | 1119 | | if (parsedValue != null) |
| | 3529 | 1120 | | { |
| | 3529 | 1121 | | AddParsedValue(info, parsedValue); |
| | 3529 | 1122 | | } |
| | 5431 | 1123 | | return; |
| | | 1124 | | } |
| | 12902 | 1125 | | Debug.Assert(index < value.Length, "Parser must return an index value within the string length."); |
| | | 1126 | | |
| | | 1127 | | // If we successfully parsed a value, but there are more left to read, store the results in a temp |
| | | 1128 | | // list. Only when all values are parsed successfully write the list to the store. |
| | 12902 | 1129 | | List<object> parsedValues = new List<object>(); |
| | 12902 | 1130 | | if (parsedValue != null) |
| | 12902 | 1131 | | { |
| | 12902 | 1132 | | parsedValues.Add(parsedValue); |
| | 12902 | 1133 | | } |
| | | 1134 | | |
| | 503167 | 1135 | | while (index < value.Length) |
| | 491699 | 1136 | | { |
| | 491699 | 1137 | | parsedValue = descriptor.Parser.ParseValue(value, info.ParsedAndInvalidValues, ref index); |
| | 490265 | 1138 | | if (parsedValue != null) |
| | 490265 | 1139 | | { |
| | 490265 | 1140 | | parsedValues.Add(parsedValue); |
| | 490265 | 1141 | | } |
| | 490265 | 1142 | | } |
| | | 1143 | | |
| | | 1144 | | // All values were parsed correctly. Copy results to the store. |
| | 999724 | 1145 | | foreach (object item in parsedValues) |
| | 482660 | 1146 | | { |
| | 482660 | 1147 | | AddParsedValue(info, item); |
| | 482660 | 1148 | | } |
| | 16899 | 1149 | | } |
| | | 1150 | | |
| | | 1151 | | internal HeaderDescriptor GetHeaderDescriptor(string name) |
| | 38969 | 1152 | | { |
| | 38969 | 1153 | | ArgumentException.ThrowIfNullOrEmpty(name); |
| | | 1154 | | |
| | 38969 | 1155 | | if (!HeaderDescriptor.TryGet(name, out HeaderDescriptor descriptor)) |
| | 0 | 1156 | | { |
| | 0 | 1157 | | throw new FormatException(SR.Format(SR.net_http_headers_invalid_header_name, name)); |
| | | 1158 | | } |
| | | 1159 | | |
| | 38969 | 1160 | | if ((descriptor.HeaderType & _allowedHeaderTypes) != 0) |
| | 26119 | 1161 | | { |
| | 26119 | 1162 | | return descriptor; |
| | | 1163 | | } |
| | 12850 | 1164 | | else if ((descriptor.HeaderType & _treatAsCustomHeaderTypes) != 0) |
| | 12850 | 1165 | | { |
| | 12850 | 1166 | | return descriptor.AsCustomHeader(); |
| | | 1167 | | } |
| | | 1168 | | |
| | 0 | 1169 | | throw new InvalidOperationException(SR.Format(SR.net_http_headers_not_allowed_header_name, name)); |
| | 38969 | 1170 | | } |
| | | 1171 | | |
| | | 1172 | | internal bool TryGetHeaderDescriptor(string name, out HeaderDescriptor descriptor) |
| | 37662 | 1173 | | { |
| | 37662 | 1174 | | if (string.IsNullOrEmpty(name)) |
| | 0 | 1175 | | { |
| | 0 | 1176 | | descriptor = default; |
| | 0 | 1177 | | return false; |
| | | 1178 | | } |
| | | 1179 | | |
| | 37662 | 1180 | | if (HeaderDescriptor.TryGet(name, out descriptor)) |
| | 37662 | 1181 | | { |
| | 37662 | 1182 | | HttpHeaderType headerType = descriptor.HeaderType; |
| | | 1183 | | |
| | 37662 | 1184 | | if ((headerType & _allowedHeaderTypes) != 0) |
| | 17076 | 1185 | | { |
| | 17076 | 1186 | | return true; |
| | | 1187 | | } |
| | | 1188 | | |
| | 20586 | 1189 | | if ((headerType & _treatAsCustomHeaderTypes) != 0) |
| | 8031 | 1190 | | { |
| | 8031 | 1191 | | descriptor = descriptor.AsCustomHeader(); |
| | 8031 | 1192 | | return true; |
| | | 1193 | | } |
| | 12555 | 1194 | | } |
| | | 1195 | | |
| | 12555 | 1196 | | return false; |
| | 37662 | 1197 | | } |
| | | 1198 | | |
| | | 1199 | | internal static void CheckContainsNewLineOrNull(string? value) |
| | 123636 | 1200 | | { |
| | 123636 | 1201 | | if (value == null) |
| | 99518 | 1202 | | { |
| | 99518 | 1203 | | return; |
| | | 1204 | | } |
| | | 1205 | | |
| | 24118 | 1206 | | if (HttpRuleParser.ContainsNewLineOrNull(value)) |
| | 3705 | 1207 | | { |
| | 3705 | 1208 | | throw new FormatException(SR.net_http_headers_no_newlines_no_nul); |
| | | 1209 | | } |
| | 119931 | 1210 | | } |
| | | 1211 | | |
| | | 1212 | | internal static string[] GetStoreValuesAsStringArray(HeaderDescriptor descriptor, HeaderStoreItemInfo info) |
| | 29943 | 1213 | | { |
| | 29943 | 1214 | | GetStoreValuesAsStringOrStringArray(descriptor, info, out string? singleValue, out string[]? multiValue); |
| | 29943 | 1215 | | Debug.Assert(singleValue is not null ^ multiValue is not null); |
| | 29943 | 1216 | | return multiValue ?? new[] { singleValue! }; |
| | 29943 | 1217 | | } |
| | | 1218 | | |
| | | 1219 | | internal static void GetStoreValuesAsStringOrStringArray(HeaderDescriptor descriptor, object sourceValues, out s |
| | 43687 | 1220 | | { |
| | 43687 | 1221 | | HeaderStoreItemInfo? info = sourceValues as HeaderStoreItemInfo; |
| | 43687 | 1222 | | if (info is null) |
| | 0 | 1223 | | { |
| | 0 | 1224 | | Debug.Assert(sourceValues is string); |
| | 0 | 1225 | | singleValue = (string)sourceValues; |
| | 0 | 1226 | | multiValue = null; |
| | 0 | 1227 | | return; |
| | | 1228 | | } |
| | | 1229 | | |
| | 43687 | 1230 | | lock (info) |
| | 43687 | 1231 | | { |
| | 43687 | 1232 | | int length = GetValueCount(info); |
| | | 1233 | | |
| | | 1234 | | scoped Span<string?> values; |
| | 43687 | 1235 | | singleValue = null; |
| | 43687 | 1236 | | if (length == 1) |
| | 14635 | 1237 | | { |
| | 14635 | 1238 | | multiValue = null; |
| | 14635 | 1239 | | values = new Span<string?>(ref singleValue); |
| | 14635 | 1240 | | } |
| | | 1241 | | else |
| | 29052 | 1242 | | { |
| | 29052 | 1243 | | Debug.Assert(length > 1, "The header should have been removed when it became empty"); |
| | 29052 | 1244 | | values = (multiValue = new string[length])!; |
| | 29052 | 1245 | | } |
| | | 1246 | | |
| | 43687 | 1247 | | int currentIndex = 0; |
| | 43687 | 1248 | | ReadStoreValues<object?>(values, info.ParsedAndInvalidValues, descriptor.Parser, ref currentIndex); |
| | 43687 | 1249 | | ReadStoreValues<string?>(values, info.RawValue, null, ref currentIndex); |
| | | 1250 | | |
| | 43687 | 1251 | | Debug.Assert(currentIndex == length); |
| | 43687 | 1252 | | } |
| | 43687 | 1253 | | } |
| | | 1254 | | |
| | | 1255 | | internal static int GetStoreValuesIntoStringArray(HeaderDescriptor descriptor, object sourceValues, [NotNull] re |
| | 0 | 1256 | | { |
| | 0 | 1257 | | values ??= Array.Empty<string>(); |
| | | 1258 | | |
| | 0 | 1259 | | HeaderStoreItemInfo? info = sourceValues as HeaderStoreItemInfo; |
| | 0 | 1260 | | if (info is null) |
| | 0 | 1261 | | { |
| | 0 | 1262 | | Debug.Assert(sourceValues is string); |
| | | 1263 | | |
| | 0 | 1264 | | if (values.Length == 0) |
| | 0 | 1265 | | { |
| | 0 | 1266 | | values = new string[1]; |
| | 0 | 1267 | | } |
| | | 1268 | | |
| | 0 | 1269 | | values[0] = (string)sourceValues; |
| | 0 | 1270 | | return 1; |
| | | 1271 | | } |
| | | 1272 | | |
| | 0 | 1273 | | lock (info) |
| | 0 | 1274 | | { |
| | 0 | 1275 | | int length = GetValueCount(info); |
| | 0 | 1276 | | Debug.Assert(length > 0); |
| | | 1277 | | |
| | 0 | 1278 | | if (values.Length < length) |
| | 0 | 1279 | | { |
| | 0 | 1280 | | values = new string[length]; |
| | 0 | 1281 | | } |
| | | 1282 | | |
| | 0 | 1283 | | int currentIndex = 0; |
| | 0 | 1284 | | ReadStoreValues<object?>(values!, info.ParsedAndInvalidValues, descriptor.Parser, ref currentIndex); |
| | 0 | 1285 | | ReadStoreValues<string?>(values!, info.RawValue, null, ref currentIndex); |
| | 0 | 1286 | | Debug.Assert(currentIndex == length); |
| | | 1287 | | |
| | 0 | 1288 | | return length; |
| | | 1289 | | } |
| | 0 | 1290 | | } |
| | | 1291 | | |
| | | 1292 | | private static int GetValueCount(HeaderStoreItemInfo info) |
| | 43687 | 1293 | | { |
| | 43687 | 1294 | | Debug.Assert(info != null); |
| | 43687 | 1295 | | Debug.Assert(Monitor.IsEntered(info)); |
| | | 1296 | | |
| | 43687 | 1297 | | return Count<object>(info.ParsedAndInvalidValues) + Count<string>(info.RawValue); |
| | | 1298 | | |
| | | 1299 | | static int Count<T>(object? valueStore) => |
| | 87374 | 1300 | | valueStore is null ? 0 : |
| | 87374 | 1301 | | valueStore is List<T> list ? list.Count : |
| | 87374 | 1302 | | 1; |
| | 43687 | 1303 | | } |
| | | 1304 | | |
| | | 1305 | | private static void ReadStoreValues<T>(Span<string?> values, object? storeValue, HttpHeaderParser? parser, ref i |
| | 87374 | 1306 | | { |
| | 87374 | 1307 | | if (storeValue != null) |
| | 43687 | 1308 | | { |
| | 43687 | 1309 | | List<T>? storeValues = storeValue as List<T>; |
| | | 1310 | | |
| | 43687 | 1311 | | if (storeValues == null) |
| | 14635 | 1312 | | { |
| | 14635 | 1313 | | values[currentIndex] = parser == null || storeValue is InvalidValue ? storeValue.ToString() : parser |
| | 14635 | 1314 | | currentIndex++; |
| | 14635 | 1315 | | } |
| | | 1316 | | else |
| | 29052 | 1317 | | { |
| | 2547856 | 1318 | | foreach (object? item in storeValues) |
| | 1230350 | 1319 | | { |
| | 1230350 | 1320 | | Debug.Assert(item != null); |
| | 1230350 | 1321 | | values[currentIndex] = parser == null || item is InvalidValue ? item.ToString() : parser.ToStrin |
| | 1230350 | 1322 | | currentIndex++; |
| | 1230350 | 1323 | | } |
| | 29052 | 1324 | | } |
| | 43687 | 1325 | | } |
| | 87374 | 1326 | | } |
| | | 1327 | | |
| | | 1328 | | private static bool AreEqual(object value, object? storeValue, IEqualityComparer? comparer) |
| | 0 | 1329 | | { |
| | 0 | 1330 | | Debug.Assert(value != null); |
| | | 1331 | | |
| | 0 | 1332 | | if (comparer != null) |
| | 0 | 1333 | | { |
| | 0 | 1334 | | return comparer.Equals(value, storeValue); |
| | | 1335 | | } |
| | | 1336 | | |
| | | 1337 | | // We don't have a comparer, so use the Equals() method. |
| | 0 | 1338 | | return value.Equals(storeValue); |
| | 0 | 1339 | | } |
| | | 1340 | | |
| | | 1341 | | internal sealed class InvalidValue |
| | | 1342 | | { |
| | | 1343 | | private readonly string _value; |
| | | 1344 | | |
| | 7658 | 1345 | | public InvalidValue(string value) |
| | 7658 | 1346 | | { |
| | 7658 | 1347 | | Debug.Assert(value is not null); |
| | 7658 | 1348 | | _value = value; |
| | 7658 | 1349 | | } |
| | | 1350 | | |
| | 7658 | 1351 | | public override string ToString() => _value; |
| | | 1352 | | } |
| | | 1353 | | |
| | | 1354 | | internal sealed class HeaderStoreItemInfo |
| | | 1355 | | { |
| | 113157 | 1356 | | internal HeaderStoreItemInfo() { } |
| | | 1357 | | |
| | | 1358 | | internal object? RawValue; |
| | | 1359 | | internal object? ParsedAndInvalidValues; |
| | | 1360 | | |
| | | 1361 | | public bool CanAddParsedValue(HttpHeaderParser parser) |
| | 41057 | 1362 | | { |
| | 41057 | 1363 | | Debug.Assert(parser != null, "There should be no reason to call CanAddValue if there is no parser for th |
| | | 1364 | | |
| | | 1365 | | // If the header only supports one value, and we have already a value set, then we can't add |
| | | 1366 | | // another value. E.g. the 'Date' header only supports one value. We can't add multiple timestamps |
| | | 1367 | | // to 'Date'. |
| | | 1368 | | // So if this is a known header, ask the parser if it supports multiple values and check whether |
| | | 1369 | | // we already have a (valid or invalid) value. |
| | | 1370 | | // Note that we ignore the rawValue by purpose: E.g. we are parsing 2 raw values for a header only |
| | | 1371 | | // supporting 1 value. When the first value gets parsed, CanAddValue returns true and we add the |
| | | 1372 | | // parsed value to ParsedValue. When the second value is parsed, CanAddValue returns false, because |
| | | 1373 | | // we have already a parsed value. |
| | 41057 | 1374 | | return parser.SupportsMultipleValues || ParsedAndInvalidValues is null; |
| | 41057 | 1375 | | } |
| | | 1376 | | |
| | | 1377 | | [Conditional("DEBUG")] |
| | | 1378 | | public void AssertContainsNoInvalidValues() |
| | 8541 | 1379 | | { |
| | 8541 | 1380 | | if (ParsedAndInvalidValues is not null) |
| | 8541 | 1381 | | { |
| | 8541 | 1382 | | if (ParsedAndInvalidValues is List<object> list) |
| | 5734 | 1383 | | { |
| | 499862 | 1384 | | foreach (object item in list) |
| | 241330 | 1385 | | { |
| | 241330 | 1386 | | Debug.Assert(item is not InvalidValue); |
| | 241330 | 1387 | | } |
| | 5734 | 1388 | | } |
| | | 1389 | | else |
| | 2807 | 1390 | | { |
| | 2807 | 1391 | | Debug.Assert(ParsedAndInvalidValues is not InvalidValue); |
| | 2807 | 1392 | | } |
| | 8541 | 1393 | | } |
| | 8541 | 1394 | | } |
| | | 1395 | | |
| | | 1396 | | public object? GetSingleParsedValue() |
| | 0 | 1397 | | { |
| | 0 | 1398 | | if (ParsedAndInvalidValues is not null) |
| | 0 | 1399 | | { |
| | 0 | 1400 | | if (ParsedAndInvalidValues is List<object> list) |
| | 0 | 1401 | | { |
| | 0 | 1402 | | AssertContainsSingleParsedValue(list); |
| | 0 | 1403 | | foreach (object item in list) |
| | 0 | 1404 | | { |
| | 0 | 1405 | | if (item is not InvalidValue) |
| | 0 | 1406 | | { |
| | 0 | 1407 | | return item; |
| | | 1408 | | } |
| | 0 | 1409 | | } |
| | 0 | 1410 | | } |
| | | 1411 | | else |
| | 0 | 1412 | | { |
| | 0 | 1413 | | if (ParsedAndInvalidValues is not InvalidValue) |
| | 0 | 1414 | | { |
| | 0 | 1415 | | return ParsedAndInvalidValues; |
| | | 1416 | | } |
| | 0 | 1417 | | } |
| | 0 | 1418 | | } |
| | | 1419 | | |
| | 0 | 1420 | | return null; |
| | 0 | 1421 | | } |
| | | 1422 | | |
| | | 1423 | | [Conditional("DEBUG")] |
| | | 1424 | | private static void AssertContainsSingleParsedValue(List<object> list) |
| | 0 | 1425 | | { |
| | 0 | 1426 | | int count = 0; |
| | 0 | 1427 | | foreach (object item in list) |
| | 0 | 1428 | | { |
| | 0 | 1429 | | if (item is not InvalidValue) |
| | 0 | 1430 | | { |
| | 0 | 1431 | | count++; |
| | 0 | 1432 | | } |
| | 0 | 1433 | | } |
| | | 1434 | | |
| | 0 | 1435 | | Debug.Assert(count == 1, "Only a single parsed value should be stored for this parser"); |
| | 0 | 1436 | | } |
| | | 1437 | | |
| | 38484 | 1438 | | public bool IsEmpty => RawValue == null && ParsedAndInvalidValues == null; |
| | | 1439 | | } |
| | | 1440 | | |
| | | 1441 | | |
| | | 1442 | | #region Low-level implementation details that work with _headerStore directly |
| | | 1443 | | |
| | | 1444 | | private const int InitialCapacity = 4; |
| | | 1445 | | internal const int ArrayThreshold = 64; // Above this threshold, header ordering will not be preserved |
| | | 1446 | | |
| | | 1447 | | internal HeaderEntry[]? GetEntriesArray() |
| | 63958 | 1448 | | { |
| | 63958 | 1449 | | object? store = _headerStore; |
| | 63958 | 1450 | | if (store is null) |
| | 0 | 1451 | | { |
| | 0 | 1452 | | return null; |
| | | 1453 | | } |
| | 63958 | 1454 | | else if (store is HeaderEntry[] entries) |
| | 63958 | 1455 | | { |
| | 63958 | 1456 | | return entries; |
| | | 1457 | | } |
| | | 1458 | | else |
| | 0 | 1459 | | { |
| | 0 | 1460 | | return GetEntriesFromDictionary(); |
| | | 1461 | | } |
| | | 1462 | | |
| | | 1463 | | HeaderEntry[] GetEntriesFromDictionary() |
| | 0 | 1464 | | { |
| | 0 | 1465 | | var dictionary = (Dictionary<HeaderDescriptor, object>)_headerStore!; |
| | 0 | 1466 | | var entries = new HeaderEntry[dictionary.Count]; |
| | 0 | 1467 | | int i = 0; |
| | 0 | 1468 | | foreach (KeyValuePair<HeaderDescriptor, object> entry in dictionary) |
| | 0 | 1469 | | { |
| | 0 | 1470 | | entries[i++] = new HeaderEntry |
| | 0 | 1471 | | { |
| | 0 | 1472 | | Key = entry.Key, |
| | 0 | 1473 | | Value = entry.Value |
| | 0 | 1474 | | }; |
| | 0 | 1475 | | } |
| | 0 | 1476 | | return entries; |
| | 0 | 1477 | | } |
| | 63958 | 1478 | | } |
| | | 1479 | | |
| | | 1480 | | internal ReadOnlySpan<HeaderEntry> GetEntries() |
| | 0 | 1481 | | { |
| | 0 | 1482 | | return new ReadOnlySpan<HeaderEntry>(GetEntriesArray(), 0, _count); |
| | 0 | 1483 | | } |
| | | 1484 | | |
| | 25107 | 1485 | | internal int Count => _count; |
| | | 1486 | | |
| | 16199 | 1487 | | private bool EntriesAreLiveView => _headerStore is HeaderEntry[]; |
| | | 1488 | | |
| | | 1489 | | private ref object GetValueRefOrNullRef(HeaderDescriptor key) |
| | 41940 | 1490 | | { |
| | 41940 | 1491 | | ref object valueRef = ref Unsafe.NullRef<object>(); |
| | | 1492 | | |
| | 41940 | 1493 | | object? store = _headerStore; |
| | 41940 | 1494 | | if (store is HeaderEntry[] entries) |
| | 41940 | 1495 | | { |
| | 83880 | 1496 | | for (int i = 0; i < _count && i < entries.Length; i++) |
| | 8541 | 1497 | | { |
| | 8541 | 1498 | | if (key.Equals(entries[i].Key)) |
| | 8541 | 1499 | | { |
| | 8541 | 1500 | | valueRef = ref entries[i].Value; |
| | 8541 | 1501 | | break; |
| | | 1502 | | } |
| | 0 | 1503 | | } |
| | 41940 | 1504 | | } |
| | 0 | 1505 | | else if (store is not null) |
| | 0 | 1506 | | { |
| | 0 | 1507 | | valueRef = ref CollectionsMarshal.GetValueRefOrNullRef((Dictionary<HeaderDescriptor, object>)store, key) |
| | 0 | 1508 | | } |
| | | 1509 | | |
| | 41940 | 1510 | | return ref valueRef; |
| | 41940 | 1511 | | } |
| | | 1512 | | |
| | | 1513 | | private ref object? GetValueRefOrAddDefault(HeaderDescriptor key) |
| | 35513 | 1514 | | { |
| | 35513 | 1515 | | object? store = _headerStore; |
| | 35513 | 1516 | | if (store is HeaderEntry[] entries) |
| | 35510 | 1517 | | { |
| | 71020 | 1518 | | for (int i = 0; i < _count && i < entries.Length; i++) |
| | 5203 | 1519 | | { |
| | 5203 | 1520 | | if (key.Equals(entries[i].Key)) |
| | 5203 | 1521 | | { |
| | 5203 | 1522 | | return ref entries[i].Value!; |
| | | 1523 | | } |
| | 0 | 1524 | | } |
| | | 1525 | | |
| | 30307 | 1526 | | int count = _count; |
| | 30307 | 1527 | | _count++; |
| | 30307 | 1528 | | if ((uint)count < (uint)entries.Length) |
| | 30307 | 1529 | | { |
| | 30307 | 1530 | | entries[count].Key = key; |
| | 30307 | 1531 | | return ref entries[count].Value!; |
| | | 1532 | | } |
| | | 1533 | | |
| | 0 | 1534 | | return ref GrowEntriesAndAddDefault(key); |
| | | 1535 | | } |
| | 3 | 1536 | | else if (store is null) |
| | 3 | 1537 | | { |
| | 3 | 1538 | | _count++; |
| | 3 | 1539 | | entries = new HeaderEntry[InitialCapacity]; |
| | 3 | 1540 | | _headerStore = entries; |
| | 3 | 1541 | | entries[0].Key = key; |
| | 3 | 1542 | | return ref entries[0].Value!; |
| | | 1543 | | } |
| | | 1544 | | else |
| | 0 | 1545 | | { |
| | 0 | 1546 | | return ref DictionaryGetValueRefOrAddDefault(key); |
| | | 1547 | | } |
| | | 1548 | | |
| | | 1549 | | ref object? GrowEntriesAndAddDefault(HeaderDescriptor key) |
| | 0 | 1550 | | { |
| | 0 | 1551 | | var entries = (HeaderEntry[])_headerStore!; |
| | 0 | 1552 | | if (entries.Length == ArrayThreshold) |
| | 0 | 1553 | | { |
| | 0 | 1554 | | return ref ConvertToDictionaryAndAddDefault(key); |
| | | 1555 | | } |
| | | 1556 | | else |
| | 0 | 1557 | | { |
| | 0 | 1558 | | Array.Resize(ref entries, entries.Length << 1); |
| | 0 | 1559 | | _headerStore = entries; |
| | 0 | 1560 | | ref HeaderEntry firstNewEntry = ref entries[entries.Length >> 1]; |
| | 0 | 1561 | | firstNewEntry.Key = key; |
| | 0 | 1562 | | return ref firstNewEntry.Value!; |
| | | 1563 | | } |
| | 0 | 1564 | | } |
| | | 1565 | | |
| | | 1566 | | ref object? ConvertToDictionaryAndAddDefault(HeaderDescriptor key) |
| | 0 | 1567 | | { |
| | 0 | 1568 | | var entries = (HeaderEntry[])_headerStore!; |
| | 0 | 1569 | | var dictionary = new Dictionary<HeaderDescriptor, object>(ArrayThreshold); |
| | 0 | 1570 | | _headerStore = dictionary; |
| | 0 | 1571 | | foreach (HeaderEntry entry in entries) |
| | 0 | 1572 | | { |
| | 0 | 1573 | | dictionary.Add(entry.Key, entry.Value); |
| | 0 | 1574 | | } |
| | 0 | 1575 | | Debug.Assert(dictionary.Count == _count - 1); |
| | 0 | 1576 | | return ref CollectionsMarshal.GetValueRefOrAddDefault(dictionary, key, out _); |
| | 0 | 1577 | | } |
| | | 1578 | | |
| | | 1579 | | ref object? DictionaryGetValueRefOrAddDefault(HeaderDescriptor key) |
| | 0 | 1580 | | { |
| | 0 | 1581 | | var dictionary = (Dictionary<HeaderDescriptor, object>)_headerStore!; |
| | 0 | 1582 | | ref object? value = ref CollectionsMarshal.GetValueRefOrAddDefault(dictionary, key, out _); |
| | 0 | 1583 | | if (value is null) |
| | 0 | 1584 | | { |
| | 0 | 1585 | | _count++; |
| | 0 | 1586 | | } |
| | 0 | 1587 | | return ref value; |
| | 0 | 1588 | | } |
| | 35513 | 1589 | | } |
| | | 1590 | | |
| | | 1591 | | private void AddEntryToStore(HeaderEntry entry) |
| | 8541 | 1592 | | { |
| | 8541 | 1593 | | Debug.Assert(!Contains(entry.Key)); |
| | | 1594 | | |
| | 8541 | 1595 | | if (_headerStore is HeaderEntry[] entries) |
| | 8541 | 1596 | | { |
| | 8541 | 1597 | | int count = _count; |
| | 8541 | 1598 | | if ((uint)count < (uint)entries.Length) |
| | 8541 | 1599 | | { |
| | 8541 | 1600 | | entries[count] = entry; |
| | 8541 | 1601 | | _count++; |
| | 8541 | 1602 | | return; |
| | | 1603 | | } |
| | 0 | 1604 | | } |
| | | 1605 | | |
| | 0 | 1606 | | GetValueRefOrAddDefault(entry.Key) = entry.Value; |
| | 8541 | 1607 | | } |
| | | 1608 | | |
| | | 1609 | | internal bool Contains(HeaderDescriptor key) |
| | 17082 | 1610 | | { |
| | 17082 | 1611 | | return !Unsafe.IsNullRef(ref GetValueRefOrNullRef(key)); |
| | 17082 | 1612 | | } |
| | | 1613 | | |
| | | 1614 | | public void Clear() |
| | 62769 | 1615 | | { |
| | 62769 | 1616 | | if (_headerStore is HeaderEntry[] entries) |
| | 62722 | 1617 | | { |
| | 62722 | 1618 | | Array.Clear(entries, 0, _count); |
| | 62722 | 1619 | | } |
| | | 1620 | | else |
| | 47 | 1621 | | { |
| | 47 | 1622 | | _headerStore = null; |
| | 47 | 1623 | | } |
| | 62769 | 1624 | | _count = 0; |
| | 62769 | 1625 | | } |
| | | 1626 | | |
| | | 1627 | | internal bool Remove(HeaderDescriptor key) |
| | 0 | 1628 | | { |
| | 0 | 1629 | | bool removed = false; |
| | | 1630 | | |
| | 0 | 1631 | | object? store = _headerStore; |
| | 0 | 1632 | | if (store is HeaderEntry[] entries) |
| | 0 | 1633 | | { |
| | 0 | 1634 | | for (int i = 0; i < _count && i < entries.Length; i++) |
| | 0 | 1635 | | { |
| | 0 | 1636 | | if (key.Equals(entries[i].Key)) |
| | 0 | 1637 | | { |
| | 0 | 1638 | | while (i + 1 < _count && (uint)(i + 1) < (uint)entries.Length) |
| | 0 | 1639 | | { |
| | 0 | 1640 | | entries[i] = entries[i + 1]; |
| | 0 | 1641 | | i++; |
| | 0 | 1642 | | } |
| | 0 | 1643 | | entries[i] = default; |
| | 0 | 1644 | | removed = true; |
| | 0 | 1645 | | break; |
| | | 1646 | | } |
| | 0 | 1647 | | } |
| | 0 | 1648 | | } |
| | 0 | 1649 | | else if (store is not null) |
| | 0 | 1650 | | { |
| | 0 | 1651 | | removed = ((Dictionary<HeaderDescriptor, object>)store).Remove(key); |
| | 0 | 1652 | | } |
| | | 1653 | | |
| | 0 | 1654 | | if (removed) |
| | 0 | 1655 | | { |
| | 0 | 1656 | | _count--; |
| | 0 | 1657 | | } |
| | | 1658 | | |
| | 0 | 1659 | | return removed; |
| | 0 | 1660 | | } |
| | | 1661 | | |
| | | 1662 | | #endregion // _headerStore implementation |
| | | 1663 | | } |
| | | 1664 | | } |