| | | 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.Diagnostics; |
| | | 5 | | using System.Runtime.CompilerServices; |
| | | 6 | | using System.Runtime.InteropServices; |
| | | 7 | | using System.Text; |
| | | 8 | | #if NET |
| | | 9 | | using System.Runtime.Intrinsics.Arm; |
| | | 10 | | using System.Runtime.Intrinsics.X86; |
| | | 11 | | using System.Runtime.Intrinsics; |
| | | 12 | | #endif |
| | | 13 | | |
| | | 14 | | namespace System.Buffers.Text |
| | | 15 | | { |
| | | 16 | | // AVX2 version based on https://github.com/aklomp/base64/tree/e516d769a2a432c08404f1981e73b431566057be/lib/arch/avx |
| | | 17 | | // Vector128 version based on https://github.com/aklomp/base64/tree/e516d769a2a432c08404f1981e73b431566057be/lib/arc |
| | | 18 | | internal static partial class Base64Helper |
| | | 19 | | { |
| | | 20 | | internal static unsafe OperationStatus DecodeFrom<TBase64Decoder, T>(TBase64Decoder decoder, ReadOnlySpan<T> sou |
| | | 21 | | out int bytesConsumed, out int bytesWritten, bool isFinalBlock, bool ignoreWhiteSpace) |
| | | 22 | | where TBase64Decoder : IBase64Decoder<T> |
| | | 23 | | where T : unmanaged |
| | | 24 | | { |
| | | 25 | | if (source.IsEmpty) |
| | | 26 | | { |
| | 33 | 27 | | bytesConsumed = 0; |
| | 33 | 28 | | bytesWritten = 0; |
| | 33 | 29 | | return OperationStatus.Done; |
| | | 30 | | } |
| | | 31 | | |
| | 352564 | 32 | | fixed (T* srcBytes = &MemoryMarshal.GetReference(source)) |
| | 352564 | 33 | | fixed (byte* destBytes = &MemoryMarshal.GetReference(bytes)) |
| | | 34 | | { |
| | 352564 | 35 | | int srcLength = decoder.SrcLength(isFinalBlock, source.Length); |
| | 352564 | 36 | | int destLength = bytes.Length; |
| | 352564 | 37 | | int maxSrcLength = srcLength; |
| | 352564 | 38 | | int decodedLength = decoder.GetMaxDecodedLength(srcLength); |
| | | 39 | | |
| | | 40 | | // max. 2 padding chars |
| | 352564 | 41 | | if (destLength < decodedLength - 2) |
| | | 42 | | { |
| | | 43 | | // For overflow see comment below |
| | 0 | 44 | | maxSrcLength = destLength / 3 * 4; |
| | | 45 | | } |
| | | 46 | | |
| | 352564 | 47 | | T* src = srcBytes; |
| | 352564 | 48 | | byte* dest = destBytes; |
| | 352564 | 49 | | T* srcEnd = srcBytes + (uint)srcLength; |
| | 352564 | 50 | | T* srcMax = srcBytes + (uint)maxSrcLength; |
| | | 51 | | |
| | | 52 | | #if NET |
| | 352564 | 53 | | if (maxSrcLength >= 24) |
| | | 54 | | { |
| | 136347 | 55 | | T* end = srcMax - 88; |
| | 136347 | 56 | | if (Vector512.IsHardwareAccelerated && Avx512Vbmi.IsSupported && (end >= src)) |
| | | 57 | | { |
| | 0 | 58 | | Avx512Decode(decoder, ref src, ref dest, end, maxSrcLength, destLength, srcBytes, destBytes); |
| | | 59 | | |
| | 0 | 60 | | if (src == srcEnd) |
| | | 61 | | { |
| | | 62 | | goto DoneExit; |
| | | 63 | | } |
| | | 64 | | } |
| | | 65 | | |
| | 136347 | 66 | | end = srcMax - 45; |
| | 136347 | 67 | | if (Avx2.IsSupported && (end >= src)) |
| | | 68 | | { |
| | 132374 | 69 | | Avx2Decode(decoder, ref src, ref dest, end, maxSrcLength, destLength, srcBytes, destBytes); |
| | | 70 | | |
| | 132374 | 71 | | if (src == srcEnd) |
| | | 72 | | { |
| | | 73 | | goto DoneExit; |
| | | 74 | | } |
| | | 75 | | } |
| | | 76 | | |
| | 136347 | 77 | | end = srcMax - 66; |
| | | 78 | | if (AdvSimd.Arm64.IsSupported && (end >= src)) |
| | | 79 | | { |
| | | 80 | | AdvSimdDecode(decoder, ref src, ref dest, end, maxSrcLength, destLength, srcBytes, destBytes); |
| | | 81 | | |
| | | 82 | | if (src == srcEnd) |
| | | 83 | | { |
| | | 84 | | goto DoneExit; |
| | | 85 | | } |
| | | 86 | | } |
| | | 87 | | |
| | 136347 | 88 | | end = srcMax - 24; |
| | 136347 | 89 | | if ((Ssse3.IsSupported || AdvSimd.Arm64.IsSupported) && BitConverter.IsLittleEndian && (end >= src)) |
| | | 90 | | { |
| | 134537 | 91 | | Vector128Decode(decoder, ref src, ref dest, end, maxSrcLength, destLength, srcBytes, destBytes); |
| | | 92 | | |
| | 134537 | 93 | | if (src == srcEnd) |
| | | 94 | | { |
| | | 95 | | goto DoneExit; |
| | | 96 | | } |
| | | 97 | | } |
| | | 98 | | } |
| | | 99 | | #endif |
| | | 100 | | |
| | | 101 | | // Last bytes could have padding characters, so process them separately and treat them as valid only if |
| | | 102 | | // if isFinalBlock is false, padding characters are considered invalid |
| | 352564 | 103 | | int skipLastChunk = isFinalBlock ? 4 : 0; |
| | | 104 | | |
| | 352564 | 105 | | if (destLength >= decodedLength) |
| | | 106 | | { |
| | 352564 | 107 | | maxSrcLength = srcLength - skipLastChunk; |
| | | 108 | | } |
| | | 109 | | else |
| | | 110 | | { |
| | | 111 | | // This should never overflow since destLength here is less than int.MaxValue / 4 * 3 (i.e. 16106127 |
| | | 112 | | // Therefore, (destLength / 3) * 4 will always be less than 2147483641 |
| | 0 | 113 | | Debug.Assert(destLength < (int.MaxValue / 4 * 3)); |
| | | 114 | | #if NET |
| | 0 | 115 | | (maxSrcLength, int remainder) = int.DivRem(destLength, 3); |
| | 0 | 116 | | maxSrcLength *= 4; |
| | | 117 | | #else |
| | | 118 | | maxSrcLength = (destLength / 3) * 4; |
| | | 119 | | int remainder = (int)((uint)destLength % 3); |
| | | 120 | | #endif |
| | 0 | 121 | | if (isFinalBlock && remainder > 0) |
| | | 122 | | { |
| | 0 | 123 | | srcLength &= ~0x3; // In case of Base64UrlDecoder source can be not a multiple of 4, round down |
| | | 124 | | } |
| | | 125 | | } |
| | | 126 | | |
| | 352564 | 127 | | ref sbyte decodingMap = ref MemoryMarshal.GetReference(decoder.DecodingMap); |
| | 352564 | 128 | | srcMax = srcBytes + maxSrcLength; |
| | | 129 | | |
| | 748587 | 130 | | while (src < srcMax) |
| | | 131 | | { |
| | 527208 | 132 | | int result = decoder.DecodeFourElements(src, ref decodingMap); |
| | | 133 | | |
| | 527208 | 134 | | if (result < 0) |
| | | 135 | | { |
| | | 136 | | goto InvalidDataExit; |
| | | 137 | | } |
| | | 138 | | |
| | 396023 | 139 | | WriteThreeLowOrderBytes(dest, result); |
| | 396023 | 140 | | src += 4; |
| | 396023 | 141 | | dest += 3; |
| | | 142 | | } |
| | | 143 | | |
| | 221379 | 144 | | if (maxSrcLength != srcLength - skipLastChunk) |
| | | 145 | | { |
| | | 146 | | goto DestinationTooSmallExit; |
| | | 147 | | } |
| | | 148 | | |
| | 221379 | 149 | | if (src == srcEnd) |
| | | 150 | | { |
| | 212904 | 151 | | if (isFinalBlock) |
| | | 152 | | { |
| | | 153 | | goto InvalidDataExit; |
| | | 154 | | } |
| | | 155 | | |
| | 212904 | 156 | | if (src == srcBytes + source.Length) |
| | | 157 | | { |
| | 212904 | 158 | | goto DoneExit; |
| | | 159 | | } |
| | | 160 | | |
| | | 161 | | goto NeedMoreDataExit; |
| | | 162 | | } |
| | | 163 | | |
| | | 164 | | // if isFinalBlock is false, we will never reach this point |
| | | 165 | | // Handle remaining bytes, for Base64 its always 4 bytes, for Base64Url up to 8 bytes left. |
| | | 166 | | // If more than 4 bytes remained it will end up in DestinationTooSmallExit or InvalidDataExit (might suc |
| | 8475 | 167 | | long remaining = srcEnd - src; |
| | 8475 | 168 | | Debug.Assert(typeof(TBase64Decoder) == typeof(Base64DecoderByte) ? remaining == 4 : remaining < 8); |
| | 8475 | 169 | | int i0 = decoder.DecodeRemaining(srcEnd, ref decodingMap, remaining, out uint t2, out uint t3); |
| | | 170 | | |
| | 8475 | 171 | | byte* destMax = destBytes + (uint)destLength; |
| | | 172 | | |
| | 8475 | 173 | | if (!decoder.IsValidPadding(t3)) |
| | | 174 | | { |
| | 4014 | 175 | | int i2 = Unsafe.Add(ref decodingMap, (IntPtr)t2); |
| | 4014 | 176 | | int i3 = Unsafe.Add(ref decodingMap, (IntPtr)t3); |
| | | 177 | | |
| | 4014 | 178 | | i2 <<= 6; |
| | | 179 | | |
| | 4014 | 180 | | i0 |= i3; |
| | 4014 | 181 | | i0 |= i2; |
| | | 182 | | |
| | 4014 | 183 | | if (i0 < 0) |
| | | 184 | | { |
| | | 185 | | goto InvalidDataExit; |
| | | 186 | | } |
| | 3866 | 187 | | if (dest + 3 > destMax) |
| | | 188 | | { |
| | | 189 | | goto DestinationTooSmallExit; |
| | | 190 | | } |
| | | 191 | | |
| | 3866 | 192 | | WriteThreeLowOrderBytes(dest, i0); |
| | 3866 | 193 | | dest += 3; |
| | 3866 | 194 | | src += 4; |
| | | 195 | | } |
| | 4461 | 196 | | else if (!decoder.IsValidPadding(t2)) |
| | | 197 | | { |
| | 1855 | 198 | | int i2 = Unsafe.Add(ref decodingMap, (IntPtr)t2); |
| | | 199 | | |
| | 1855 | 200 | | i2 <<= 6; |
| | | 201 | | |
| | 1855 | 202 | | i0 |= i2; |
| | | 203 | | |
| | 1855 | 204 | | if ((i0 & 0x800000c0) != 0) // if negative or 2 unused bits are not 0. |
| | | 205 | | { |
| | | 206 | | goto InvalidDataExit; |
| | | 207 | | } |
| | 1559 | 208 | | if (dest + 2 > destMax) |
| | | 209 | | { |
| | | 210 | | goto DestinationTooSmallExit; |
| | | 211 | | } |
| | | 212 | | |
| | 1559 | 213 | | dest[0] = (byte)(i0 >> 16); |
| | 1559 | 214 | | dest[1] = (byte)(i0 >> 8); |
| | 1559 | 215 | | dest += 2; |
| | 1559 | 216 | | src += remaining; |
| | | 217 | | } |
| | | 218 | | else |
| | | 219 | | { |
| | 2606 | 220 | | if ((i0 & 0x8000F000) != 0) // if negative or 4 unused bits are not 0. |
| | | 221 | | { |
| | | 222 | | goto InvalidDataExit; |
| | | 223 | | } |
| | 1752 | 224 | | if (dest + 1 > destMax) |
| | | 225 | | { |
| | | 226 | | goto DestinationTooSmallExit; |
| | | 227 | | } |
| | | 228 | | |
| | 1752 | 229 | | dest[0] = (byte)(i0 >> 16); |
| | 1752 | 230 | | dest += 1; |
| | 1752 | 231 | | src += remaining; |
| | | 232 | | } |
| | | 233 | | |
| | 7177 | 234 | | if (srcLength != source.Length) |
| | | 235 | | { |
| | | 236 | | goto InvalidDataExit; |
| | | 237 | | } |
| | | 238 | | |
| | | 239 | | DoneExit: |
| | 220081 | 240 | | bytesConsumed = (int)(src - srcBytes); |
| | 220081 | 241 | | bytesWritten = (int)(dest - destBytes); |
| | 220081 | 242 | | return OperationStatus.Done; |
| | | 243 | | |
| | | 244 | | DestinationTooSmallExit: |
| | 0 | 245 | | if (srcLength != source.Length && isFinalBlock) |
| | | 246 | | { |
| | | 247 | | goto InvalidDataExit; // if input is not a multiple of 4, and there is no more data, return invalid |
| | | 248 | | } |
| | | 249 | | |
| | 0 | 250 | | if (ignoreWhiteSpace) |
| | | 251 | | { |
| | | 252 | | // Fall through to InvalidDataFallback which strips whitespace and re-evaluates destination size req |
| | | 253 | | goto InvalidDataExit; |
| | | 254 | | } |
| | | 255 | | |
| | 0 | 256 | | bytesConsumed = (int)(src - srcBytes); |
| | 0 | 257 | | bytesWritten = (int)(dest - destBytes); |
| | 0 | 258 | | return OperationStatus.DestinationTooSmall; |
| | | 259 | | |
| | | 260 | | NeedMoreDataExit: |
| | 0 | 261 | | bytesConsumed = (int)(src - srcBytes); |
| | 0 | 262 | | bytesWritten = (int)(dest - destBytes); |
| | 0 | 263 | | return OperationStatus.NeedMoreData; |
| | | 264 | | |
| | | 265 | | InvalidDataExit: |
| | 132483 | 266 | | bytesConsumed = (int)(src - srcBytes); |
| | 132483 | 267 | | bytesWritten = (int)(dest - destBytes); |
| | 132483 | 268 | | return ignoreWhiteSpace ? |
| | 132483 | 269 | | InvalidDataFallback(decoder, source, bytes, ref bytesConsumed, ref bytesWritten, isFinalBlock) : |
| | 132483 | 270 | | OperationStatus.InvalidData; |
| | | 271 | | } |
| | | 272 | | |
| | | 273 | | static OperationStatus InvalidDataFallback(TBase64Decoder decoder, ReadOnlySpan<T> source, Span<byte> bytes, |
| | | 274 | | { |
| | 5467 | 275 | | source = source.Slice(bytesConsumed); |
| | 5467 | 276 | | bytes = bytes.Slice(bytesWritten); |
| | | 277 | | |
| | | 278 | | OperationStatus status; |
| | | 279 | | do |
| | | 280 | | { |
| | 131498 | 281 | | int localConsumed = decoder.IndexOfAnyExceptWhiteSpace(source); |
| | 131498 | 282 | | if (localConsumed < 0) |
| | | 283 | | { |
| | | 284 | | // The remainder of the input is all whitespace. Mark it all as having been consumed, |
| | | 285 | | // and mark the operation as being done. |
| | 428 | 286 | | bytesConsumed += source.Length; |
| | 428 | 287 | | status = OperationStatus.Done; |
| | 428 | 288 | | break; |
| | | 289 | | } |
| | | 290 | | |
| | 131070 | 291 | | if (localConsumed == 0) |
| | | 292 | | { |
| | | 293 | | // Non-whitespace was found at the beginning of the input. Since it wasn't consumed |
| | | 294 | | // by the previous call to DecodeFromUtf8, it must be part of a Base64 sequence |
| | | 295 | | // that was interrupted by whitespace or something else considered invalid. |
| | | 296 | | // Fall back to block-wise decoding. This is very slow, but it's also very non-standard |
| | | 297 | | // formatting of the input; whitespace is typically only found between blocks, such as |
| | | 298 | | // when Convert.ToBase64String inserts a line break every 76 output characters. |
| | 3177 | 299 | | return decoder.DecodeWithWhiteSpaceBlockwiseWrapper(decoder, source, bytes, ref bytesConsumed, r |
| | | 300 | | } |
| | | 301 | | |
| | | 302 | | // Skip over the starting whitespace and continue. |
| | 127893 | 303 | | bytesConsumed += localConsumed; |
| | 127893 | 304 | | source = source.Slice(localConsumed); |
| | | 305 | | |
| | | 306 | | // Try again after consumed whitespace |
| | 127893 | 307 | | status = DecodeFrom(decoder, source, bytes, out localConsumed, out int localWritten, isFinalBlock, i |
| | 127893 | 308 | | bytesConsumed += localConsumed; |
| | 127893 | 309 | | bytesWritten += localWritten; |
| | | 310 | | |
| | 127893 | 311 | | if (status is OperationStatus.Done or OperationStatus.NeedMoreData) |
| | | 312 | | { |
| | | 313 | | break; |
| | | 314 | | } |
| | | 315 | | |
| | | 316 | | // The DecodeFrom helper will return DestinationTooSmall if the destination is too small, |
| | | 317 | | // regardless of whether it's actually too small once you skip whitespace characters. |
| | | 318 | | // In that case we loop again and fall back to block-wise decoding if we can't make progress. |
| | | 319 | | |
| | 126031 | 320 | | source = source.Slice(localConsumed); |
| | 126031 | 321 | | bytes = bytes.Slice(localWritten); |
| | | 322 | | } |
| | 126031 | 323 | | while (!source.IsEmpty); |
| | | 324 | | |
| | 2290 | 325 | | return status; |
| | | 326 | | } |
| | | 327 | | } |
| | | 328 | | |
| | | 329 | | internal static unsafe OperationStatus DecodeFromUtf8InPlace<TBase64Decoder>(TBase64Decoder decoder, Span<byte> |
| | | 330 | | where TBase64Decoder : IBase64Decoder<byte> |
| | | 331 | | { |
| | 495843 | 332 | | if (buffer.IsEmpty) |
| | | 333 | | { |
| | 0 | 334 | | bytesWritten = 0; |
| | 0 | 335 | | return OperationStatus.Done; |
| | | 336 | | } |
| | | 337 | | |
| | 495843 | 338 | | fixed (byte* bufferBytes = &MemoryMarshal.GetReference(buffer)) |
| | | 339 | | { |
| | 495843 | 340 | | uint bufferLength = (uint)buffer.Length; |
| | 495843 | 341 | | uint sourceIndex = 0; |
| | 495843 | 342 | | uint destIndex = 0; |
| | | 343 | | |
| | 495843 | 344 | | if (decoder.IsInvalidLength(buffer.Length)) |
| | | 345 | | { |
| | | 346 | | goto InvalidExit; |
| | | 347 | | } |
| | | 348 | | |
| | 494991 | 349 | | ref sbyte decodingMap = ref MemoryMarshal.GetReference(decoder.DecodingMap); |
| | | 350 | | |
| | 494991 | 351 | | if (bufferLength > 4) |
| | | 352 | | { |
| | 779718 | 353 | | while (sourceIndex < bufferLength - 4) |
| | | 354 | | { |
| | 776444 | 355 | | int result = decoder.DecodeFourElements(bufferBytes + sourceIndex, ref decodingMap); |
| | 776444 | 356 | | if (result < 0) |
| | | 357 | | { |
| | | 358 | | goto InvalidExit; |
| | | 359 | | } |
| | | 360 | | |
| | 774112 | 361 | | WriteThreeLowOrderBytes(bufferBytes + destIndex, result); |
| | 774112 | 362 | | destIndex += 3; |
| | 774112 | 363 | | sourceIndex += 4; |
| | | 364 | | } |
| | | 365 | | } |
| | | 366 | | |
| | | 367 | | uint t0; |
| | | 368 | | uint t1; |
| | | 369 | | uint t2; |
| | | 370 | | uint t3; |
| | | 371 | | |
| | 492659 | 372 | | switch (bufferLength - sourceIndex) |
| | | 373 | | { |
| | | 374 | | case 2: |
| | 1238 | 375 | | t0 = bufferBytes[bufferLength - 2]; |
| | 1238 | 376 | | t1 = bufferBytes[bufferLength - 1]; |
| | 1238 | 377 | | t2 = EncodingPad; |
| | 1238 | 378 | | t3 = EncodingPad; |
| | 1238 | 379 | | break; |
| | | 380 | | case 3: |
| | 1132 | 381 | | t0 = bufferBytes[bufferLength - 3]; |
| | 1132 | 382 | | t1 = bufferBytes[bufferLength - 2]; |
| | 1132 | 383 | | t2 = bufferBytes[bufferLength - 1]; |
| | 1132 | 384 | | t3 = EncodingPad; |
| | 1132 | 385 | | break; |
| | | 386 | | case 4: |
| | 490289 | 387 | | t0 = bufferBytes[bufferLength - 4]; |
| | 490289 | 388 | | t1 = bufferBytes[bufferLength - 3]; |
| | 490289 | 389 | | t2 = bufferBytes[bufferLength - 2]; |
| | 490289 | 390 | | t3 = bufferBytes[bufferLength - 1]; |
| | | 391 | | break; |
| | | 392 | | default: |
| | | 393 | | goto InvalidExit; |
| | | 394 | | } |
| | | 395 | | |
| | 492659 | 396 | | int i0 = Unsafe.Add(ref decodingMap, (int)t0); |
| | 492659 | 397 | | int i1 = Unsafe.Add(ref decodingMap, (int)t1); |
| | | 398 | | |
| | 492659 | 399 | | i0 <<= 18; |
| | 492659 | 400 | | i1 <<= 12; |
| | | 401 | | |
| | 492659 | 402 | | i0 |= i1; |
| | | 403 | | |
| | 492659 | 404 | | if (!decoder.IsValidPadding(t3)) |
| | | 405 | | { |
| | 489329 | 406 | | int i2 = Unsafe.Add(ref decodingMap, (int)t2); |
| | 489329 | 407 | | int i3 = Unsafe.Add(ref decodingMap, (int)t3); |
| | | 408 | | |
| | 489329 | 409 | | i2 <<= 6; |
| | | 410 | | |
| | 489329 | 411 | | i0 |= i3; |
| | 489329 | 412 | | i0 |= i2; |
| | | 413 | | |
| | 489329 | 414 | | if (i0 < 0) |
| | | 415 | | { |
| | | 416 | | goto InvalidExit; |
| | | 417 | | } |
| | | 418 | | |
| | 488877 | 419 | | WriteThreeLowOrderBytes(bufferBytes + destIndex, i0); |
| | 488877 | 420 | | destIndex += 3; |
| | | 421 | | } |
| | 3330 | 422 | | else if (!decoder.IsValidPadding(t2)) |
| | | 423 | | { |
| | 1573 | 424 | | int i2 = Unsafe.Add(ref decodingMap, (int)t2); |
| | | 425 | | |
| | 1573 | 426 | | i2 <<= 6; |
| | | 427 | | |
| | 1573 | 428 | | i0 |= i2; |
| | | 429 | | |
| | 1573 | 430 | | if ((i0 & 0x800000c0) != 0) // if negative or 2 unused bits are not 0. |
| | | 431 | | { |
| | | 432 | | goto InvalidExit; |
| | | 433 | | } |
| | | 434 | | |
| | 1341 | 435 | | bufferBytes[destIndex] = (byte)(i0 >> 16); |
| | 1341 | 436 | | bufferBytes[destIndex + 1] = (byte)(i0 >> 8); |
| | 1341 | 437 | | destIndex += 2; |
| | | 438 | | } |
| | | 439 | | else |
| | | 440 | | { |
| | 1757 | 441 | | if ((i0 & 0x8000F000) != 0) // if negative or 4 unused bits are not 0. |
| | | 442 | | { |
| | | 443 | | goto InvalidExit; |
| | | 444 | | } |
| | | 445 | | |
| | 1471 | 446 | | bufferBytes[destIndex] = (byte)(i0 >> 16); |
| | 1471 | 447 | | destIndex += 1; |
| | | 448 | | } |
| | | 449 | | |
| | 491689 | 450 | | bytesWritten = (int)destIndex; |
| | 491689 | 451 | | return OperationStatus.Done; |
| | | 452 | | |
| | | 453 | | InvalidExit: |
| | 4154 | 454 | | bytesWritten = (int)destIndex; |
| | 4154 | 455 | | return ignoreWhiteSpace ? |
| | 4154 | 456 | | DecodeWithWhiteSpaceFromUtf8InPlace<TBase64Decoder>(decoder, buffer, ref bytesWritten, sourceIndex) |
| | 4154 | 457 | | OperationStatus.InvalidData; |
| | | 458 | | } |
| | | 459 | | } |
| | | 460 | | |
| | | 461 | | internal static OperationStatus DecodeWithWhiteSpaceBlockwise<TBase64Decoder>(TBase64Decoder decoder, ReadOnlySp |
| | | 462 | | where TBase64Decoder : IBase64Decoder<byte> |
| | | 463 | | { |
| | | 464 | | const int BlockSize = 4; |
| | 3177 | 465 | | Span<byte> buffer = stackalloc byte[BlockSize]; |
| | 3177 | 466 | | OperationStatus status = OperationStatus.Done; |
| | | 467 | | |
| | 259590 | 468 | | while (!source.IsEmpty) |
| | | 469 | | { |
| | | 470 | | // Skip over any leading whitespace |
| | 259360 | 471 | | if (IsWhiteSpace(source[0])) |
| | | 472 | | { |
| | 46799 | 473 | | source = source.Slice(1); |
| | 46799 | 474 | | bytesConsumed++; |
| | 46799 | 475 | | continue; |
| | | 476 | | } |
| | | 477 | | |
| | 212561 | 478 | | int encodedIdx = 0; |
| | 212561 | 479 | | int bufferIdx = 0; |
| | 212561 | 480 | | int skipped = 0; |
| | | 481 | | |
| | 2170479 | 482 | | for (; encodedIdx < source.Length && (uint)bufferIdx < (uint)buffer.Length; ++encodedIdx) |
| | | 483 | | { |
| | 978959 | 484 | | if (IsWhiteSpace(source[encodedIdx])) |
| | | 485 | | { |
| | 130965 | 486 | | skipped++; |
| | | 487 | | } |
| | | 488 | | else |
| | | 489 | | { |
| | 847994 | 490 | | buffer[bufferIdx] = source[encodedIdx]; |
| | 847994 | 491 | | bufferIdx++; |
| | | 492 | | } |
| | | 493 | | } |
| | | 494 | | |
| | 212561 | 495 | | source = source.Slice(encodedIdx); |
| | 212561 | 496 | | Debug.Assert(bufferIdx > 0); |
| | | 497 | | |
| | | 498 | | bool hasAnotherBlock; |
| | | 499 | | |
| | 212561 | 500 | | if (typeof(TBase64Decoder) == typeof(Base64DecoderByte)) |
| | | 501 | | { |
| | 0 | 502 | | hasAnotherBlock = source.Length >= BlockSize; |
| | | 503 | | } |
| | | 504 | | else |
| | | 505 | | { |
| | 212561 | 506 | | hasAnotherBlock = source.Length > 1; |
| | | 507 | | } |
| | | 508 | | |
| | 212561 | 509 | | bool localIsFinalBlock = !hasAnotherBlock; |
| | | 510 | | |
| | | 511 | | // If this block contains padding and there's another block, then only whitespace may follow for being v |
| | 212561 | 512 | | if (hasAnotherBlock) |
| | | 513 | | { |
| | 210204 | 514 | | int paddingCount = GetPaddingCount(decoder, ref buffer[BlockSize - 1]); |
| | 210204 | 515 | | if (paddingCount > 0) |
| | | 516 | | { |
| | 239 | 517 | | hasAnotherBlock = false; |
| | 239 | 518 | | localIsFinalBlock = true; |
| | | 519 | | } |
| | | 520 | | } |
| | | 521 | | |
| | 212561 | 522 | | if (localIsFinalBlock && !isFinalBlock) |
| | | 523 | | { |
| | 0 | 524 | | localIsFinalBlock = false; |
| | | 525 | | } |
| | | 526 | | |
| | 212561 | 527 | | status = DecodeFrom<TBase64Decoder, byte>(decoder, buffer.Slice(0, bufferIdx), bytes, out int localConsu |
| | | 528 | | |
| | 212561 | 529 | | if (status != OperationStatus.Done) |
| | | 530 | | { |
| | 985 | 531 | | Debug.Assert(localConsumed == 0 && localWritten == 0, "On failure, should not have consumed or writt |
| | 985 | 532 | | return status; |
| | | 533 | | } |
| | | 534 | | |
| | 211576 | 535 | | bytesConsumed += skipped; |
| | 211576 | 536 | | bytesConsumed += localConsumed; |
| | 211576 | 537 | | bytesWritten += localWritten; |
| | | 538 | | |
| | | 539 | | // The remaining data must all be whitespace in order to be valid. |
| | 211576 | 540 | | if (!hasAnotherBlock) |
| | | 541 | | { |
| | 14822 | 542 | | for (int i = 0; i < source.Length; ++i) |
| | | 543 | | { |
| | 5613 | 544 | | if (!IsWhiteSpace(source[i])) |
| | | 545 | | { |
| | | 546 | | // Revert previous dest increment, since an invalid state followed. |
| | 164 | 547 | | bytesConsumed -= localConsumed; |
| | 164 | 548 | | bytesWritten -= localWritten; |
| | | 549 | | |
| | 164 | 550 | | return OperationStatus.InvalidData; |
| | | 551 | | } |
| | | 552 | | |
| | 5449 | 553 | | bytesConsumed++; |
| | | 554 | | } |
| | | 555 | | |
| | 1798 | 556 | | break; |
| | | 557 | | } |
| | | 558 | | |
| | 209614 | 559 | | bytes = bytes.Slice(localWritten); |
| | 209614 | 560 | | Debug.Assert(!source.IsEmpty); |
| | | 561 | | } |
| | | 562 | | |
| | 2028 | 563 | | return status; |
| | | 564 | | } |
| | | 565 | | |
| | | 566 | | internal static OperationStatus DecodeWithWhiteSpaceBlockwise<TBase64Decoder>(TBase64Decoder decoder, ReadOnlySp |
| | | 567 | | where TBase64Decoder : IBase64Decoder<ushort> |
| | | 568 | | { |
| | | 569 | | const int BlockSize = 4; |
| | 0 | 570 | | Span<ushort> buffer = stackalloc ushort[BlockSize]; |
| | 0 | 571 | | OperationStatus status = OperationStatus.Done; |
| | | 572 | | |
| | 0 | 573 | | while (!source.IsEmpty) |
| | | 574 | | { |
| | | 575 | | // Skip over any leading whitespace |
| | 0 | 576 | | if (IsWhiteSpace(source[0])) |
| | | 577 | | { |
| | 0 | 578 | | source = source.Slice(1); |
| | 0 | 579 | | bytesConsumed++; |
| | 0 | 580 | | continue; |
| | | 581 | | } |
| | | 582 | | |
| | 0 | 583 | | int encodedIdx = 0; |
| | 0 | 584 | | int bufferIdx = 0; |
| | 0 | 585 | | int skipped = 0; |
| | | 586 | | |
| | 0 | 587 | | for (; encodedIdx < source.Length && (uint)bufferIdx < (uint)buffer.Length; ++encodedIdx) |
| | | 588 | | { |
| | 0 | 589 | | if (IsWhiteSpace(source[encodedIdx])) |
| | | 590 | | { |
| | 0 | 591 | | skipped++; |
| | | 592 | | } |
| | | 593 | | else |
| | | 594 | | { |
| | 0 | 595 | | buffer[bufferIdx] = source[encodedIdx]; |
| | 0 | 596 | | bufferIdx++; |
| | | 597 | | } |
| | | 598 | | } |
| | | 599 | | |
| | 0 | 600 | | source = source.Slice(encodedIdx); |
| | 0 | 601 | | Debug.Assert(bufferIdx > 0); |
| | | 602 | | |
| | | 603 | | bool hasAnotherBlock; |
| | | 604 | | |
| | 0 | 605 | | if (decoder is Base64DecoderByte) |
| | | 606 | | { |
| | 0 | 607 | | hasAnotherBlock = source.Length >= BlockSize; |
| | | 608 | | } |
| | | 609 | | else |
| | | 610 | | { |
| | 0 | 611 | | hasAnotherBlock = source.Length > 1; |
| | | 612 | | } |
| | | 613 | | |
| | 0 | 614 | | bool localIsFinalBlock = !hasAnotherBlock; |
| | | 615 | | |
| | | 616 | | // If this block contains padding and there's another block, then only whitespace may follow for being v |
| | 0 | 617 | | if (hasAnotherBlock) |
| | | 618 | | { |
| | 0 | 619 | | int paddingCount = GetPaddingCount(decoder, ref buffer[BlockSize - 1]); |
| | 0 | 620 | | if (paddingCount > 0) |
| | | 621 | | { |
| | 0 | 622 | | hasAnotherBlock = false; |
| | 0 | 623 | | localIsFinalBlock = true; |
| | | 624 | | } |
| | | 625 | | } |
| | | 626 | | |
| | 0 | 627 | | if (localIsFinalBlock && !isFinalBlock) |
| | | 628 | | { |
| | 0 | 629 | | localIsFinalBlock = false; |
| | | 630 | | } |
| | | 631 | | |
| | 0 | 632 | | status = DecodeFrom(decoder, buffer.Slice(0, bufferIdx), bytes, out int localConsumed, out int localWrit |
| | | 633 | | |
| | 0 | 634 | | if (status != OperationStatus.Done) |
| | | 635 | | { |
| | 0 | 636 | | Debug.Assert(localConsumed == 0 && localWritten == 0, "On failure, should not have consumed or writt |
| | 0 | 637 | | return status; |
| | | 638 | | } |
| | | 639 | | |
| | 0 | 640 | | bytesConsumed += skipped; |
| | 0 | 641 | | bytesConsumed += localConsumed; |
| | 0 | 642 | | bytesWritten += localWritten; |
| | | 643 | | |
| | | 644 | | // The remaining data must all be whitespace in order to be valid. |
| | 0 | 645 | | if (!hasAnotherBlock) |
| | | 646 | | { |
| | 0 | 647 | | for (int i = 0; i < source.Length; ++i) |
| | | 648 | | { |
| | 0 | 649 | | if (!IsWhiteSpace(source[i])) |
| | | 650 | | { |
| | | 651 | | // Revert previous dest increment, since an invalid state followed. |
| | 0 | 652 | | bytesConsumed -= localConsumed; |
| | 0 | 653 | | bytesWritten -= localWritten; |
| | | 654 | | |
| | 0 | 655 | | return OperationStatus.InvalidData; |
| | | 656 | | } |
| | | 657 | | } |
| | | 658 | | |
| | 0 | 659 | | bytesConsumed += source.Length; |
| | 0 | 660 | | break; |
| | | 661 | | } |
| | | 662 | | |
| | 0 | 663 | | bytes = bytes.Slice(localWritten); |
| | 0 | 664 | | Debug.Assert(!source.IsEmpty); |
| | | 665 | | } |
| | | 666 | | |
| | 0 | 667 | | return status; |
| | | 668 | | } |
| | | 669 | | |
| | | 670 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 671 | | private static int GetPaddingCount<TBase64Decoder>(TBase64Decoder decoder, ref byte ptrToLastElement) |
| | | 672 | | where TBase64Decoder : IBase64Decoder<byte> |
| | | 673 | | { |
| | 210204 | 674 | | int padding = 0; |
| | | 675 | | |
| | 210204 | 676 | | if (decoder.IsValidPadding(ptrToLastElement)) |
| | | 677 | | { |
| | 233 | 678 | | padding++; |
| | | 679 | | } |
| | | 680 | | |
| | 210204 | 681 | | if (decoder.IsValidPadding(Unsafe.Subtract(ref ptrToLastElement, 1))) |
| | | 682 | | { |
| | 63 | 683 | | padding++; |
| | | 684 | | } |
| | | 685 | | |
| | 210204 | 686 | | return padding; |
| | | 687 | | } |
| | | 688 | | |
| | | 689 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 690 | | private static int GetPaddingCount<TBase64Decoder>(TBase64Decoder decoder, ref ushort ptrToLastElement) |
| | | 691 | | where TBase64Decoder : IBase64Decoder<ushort> |
| | | 692 | | { |
| | 0 | 693 | | int padding = 0; |
| | | 694 | | |
| | 0 | 695 | | if (decoder.IsValidPadding(ptrToLastElement)) |
| | | 696 | | { |
| | 0 | 697 | | padding++; |
| | | 698 | | } |
| | | 699 | | |
| | 0 | 700 | | if (decoder.IsValidPadding(Unsafe.Subtract(ref ptrToLastElement, 1))) |
| | | 701 | | { |
| | 0 | 702 | | padding++; |
| | | 703 | | } |
| | | 704 | | |
| | 0 | 705 | | return padding; |
| | | 706 | | } |
| | | 707 | | |
| | | 708 | | private static OperationStatus DecodeWithWhiteSpaceFromUtf8InPlace<TBase64Decoder>(TBase64Decoder decoder, Span< |
| | | 709 | | where TBase64Decoder : IBase64Decoder<byte> |
| | | 710 | | { |
| | 3308 | 711 | | int BlockSize = Math.Min(source.Length - (int)sourceIndex, 4); |
| | 3308 | 712 | | Span<byte> buffer = stackalloc byte[BlockSize]; |
| | | 713 | | |
| | 3308 | 714 | | OperationStatus status = OperationStatus.Done; |
| | 3308 | 715 | | int localDestIndex = destIndex; |
| | 3308 | 716 | | bool hasPaddingBeenProcessed = false; |
| | 3308 | 717 | | int localBytesWritten = 0; |
| | | 718 | | |
| | 492105 | 719 | | while (sourceIndex < (uint)source.Length) |
| | | 720 | | { |
| | 489946 | 721 | | int bufferIdx = 0; |
| | | 722 | | |
| | 2749894 | 723 | | while (bufferIdx < BlockSize && sourceIndex < (uint)source.Length) |
| | | 724 | | { |
| | 2259948 | 725 | | if (!IsWhiteSpace(source[(int)sourceIndex])) |
| | | 726 | | { |
| | 1955882 | 727 | | buffer[bufferIdx] = source[(int)sourceIndex]; |
| | 1955882 | 728 | | bufferIdx++; |
| | | 729 | | } |
| | | 730 | | |
| | 2259948 | 731 | | sourceIndex++; |
| | | 732 | | } |
| | | 733 | | |
| | 489946 | 734 | | if (bufferIdx == 0) |
| | | 735 | | { |
| | | 736 | | continue; |
| | | 737 | | } |
| | | 738 | | |
| | 489500 | 739 | | if (bufferIdx != 4) |
| | | 740 | | { |
| | | 741 | | // Base64 require 4 bytes, for Base64Url it can be less than 4 bytes but not 1 byte. |
| | 1110 | 742 | | if (decoder is Base64DecoderByte || bufferIdx == 1) |
| | | 743 | | { |
| | 262 | 744 | | status = OperationStatus.InvalidData; |
| | 262 | 745 | | break; |
| | | 746 | | } |
| | | 747 | | else // For Base64Url fill empty slots in last block with padding |
| | | 748 | | { |
| | 2108 | 749 | | while (bufferIdx < BlockSize) // Can happen only for last block |
| | | 750 | | { |
| | 1260 | 751 | | Debug.Assert(source.Length == sourceIndex); |
| | 1260 | 752 | | buffer[bufferIdx++] = (byte)EncodingPad; |
| | | 753 | | } |
| | | 754 | | } |
| | | 755 | | } |
| | | 756 | | |
| | 489238 | 757 | | if (hasPaddingBeenProcessed) |
| | | 758 | | { |
| | | 759 | | // Padding has already been processed, a new valid block cannot be processed. |
| | | 760 | | // Revert previous dest increment, since an invalid state followed. |
| | 41 | 761 | | localDestIndex -= localBytesWritten; |
| | 41 | 762 | | status = OperationStatus.InvalidData; |
| | 41 | 763 | | break; |
| | | 764 | | } |
| | | 765 | | |
| | 489197 | 766 | | status = DecodeFromUtf8InPlace<TBase64Decoder>(decoder, buffer, out localBytesWritten, ignoreWhiteSpace: |
| | 489197 | 767 | | localDestIndex += localBytesWritten; |
| | 489197 | 768 | | hasPaddingBeenProcessed = localBytesWritten < 3; |
| | | 769 | | |
| | 489197 | 770 | | if (status != OperationStatus.Done) |
| | | 771 | | { |
| | | 772 | | break; |
| | | 773 | | } |
| | | 774 | | |
| | | 775 | | // Write result to source span in place. |
| | 3905114 | 776 | | for (int i = 0; i < localBytesWritten; i++) |
| | | 777 | | { |
| | 1464206 | 778 | | source[localDestIndex - localBytesWritten + i] = buffer[i]; |
| | | 779 | | } |
| | | 780 | | } |
| | | 781 | | |
| | 3308 | 782 | | destIndex = localDestIndex; |
| | 3308 | 783 | | return status; |
| | | 784 | | } |
| | | 785 | | |
| | | 786 | | #if NET |
| | | 787 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 788 | | [CompExactlyDependsOn(typeof(Avx512BW))] |
| | | 789 | | [CompExactlyDependsOn(typeof(Avx512Vbmi))] |
| | | 790 | | private static unsafe void Avx512Decode<TBase64Decoder, T>(TBase64Decoder decoder, ref T* srcBytes, ref byte* de |
| | | 791 | | where TBase64Decoder : IBase64Decoder<T> |
| | | 792 | | where T : unmanaged |
| | | 793 | | { |
| | | 794 | | // Reference for VBMI implementation : https://github.com/WojciechMula/base64simd/tree/master/decode |
| | | 795 | | // If we have AVX512 support, pick off 64 bytes at a time for as long as we can, |
| | | 796 | | // but make sure that we quit before seeing any == markers at the end of the |
| | | 797 | | // string. Also, because we write 16 zeroes at the end of the output, ensure |
| | | 798 | | // that there are at least 22 valid bytes of input data remaining to close the |
| | | 799 | | // gap. 64 + 2 + 22 = 88 bytes. |
| | 0 | 800 | | T* src = srcBytes; |
| | 0 | 801 | | byte* dest = destBytes; |
| | | 802 | | |
| | | 803 | | // The JIT won't hoist these "constants", so help it |
| | 0 | 804 | | Vector512<sbyte> vbmiLookup0 = Vector512.Create(decoder.VbmiLookup0).AsSByte(); |
| | 0 | 805 | | Vector512<sbyte> vbmiLookup1 = Vector512.Create(decoder.VbmiLookup1).AsSByte(); |
| | 0 | 806 | | Vector512<byte> vbmiPackedLanesControl = Vector512.Create( |
| | 0 | 807 | | 0x06000102, 0x090a0405, 0x0c0d0e08, 0x16101112, |
| | 0 | 808 | | 0x191a1415, 0x1c1d1e18, 0x26202122, 0x292a2425, |
| | 0 | 809 | | 0x2c2d2e28, 0x36303132, 0x393a3435, 0x3c3d3e38, |
| | 0 | 810 | | 0x00000000, 0x00000000, 0x00000000, 0x00000000).AsByte(); |
| | | 811 | | |
| | 0 | 812 | | Vector512<sbyte> mergeConstant0 = Vector512.Create(0x01400140).AsSByte(); |
| | 0 | 813 | | Vector512<short> mergeConstant1 = Vector512.Create(0x00011000).AsInt16(); |
| | | 814 | | |
| | | 815 | | // This algorithm requires AVX512VBMI support. |
| | | 816 | | // Vbmi was first introduced in CannonLake and is available from IceLake on. |
| | | 817 | | do |
| | | 818 | | { |
| | 0 | 819 | | if (!decoder.TryLoadVector512(src, srcStart, sourceLength, out Vector512<sbyte> str)) |
| | | 820 | | { |
| | | 821 | | break; |
| | | 822 | | } |
| | | 823 | | |
| | | 824 | | // Step 1: Translate encoded Base64 input to their original indices |
| | | 825 | | // This step also checks for invalid inputs and exits. |
| | | 826 | | // After this, we have indices which are verified to have upper 2 bits set to 0 in each byte. |
| | | 827 | | // origIndex = [...|00dddddd|00cccccc|00bbbbbb|00aaaaaa] |
| | 0 | 828 | | Vector512<sbyte> origIndex = Avx512Vbmi.PermuteVar64x8x2(vbmiLookup0, str, vbmiLookup1); |
| | 0 | 829 | | Vector512<sbyte> errorVec = (origIndex.AsInt32() | str.AsInt32()).AsSByte(); |
| | 0 | 830 | | if (errorVec.ExtractMostSignificantBits() != 0) |
| | | 831 | | { |
| | | 832 | | break; |
| | | 833 | | } |
| | | 834 | | |
| | | 835 | | // Step 2: Now we need to reshuffle bits to remove the 0 bits. |
| | | 836 | | // multiAdd1: [...|0000cccc|ccdddddd|0000aaaa|aabbbbbb] |
| | 0 | 837 | | Vector512<short> multiAdd1 = Avx512BW.MultiplyAddAdjacent(origIndex.AsByte(), mergeConstant0); |
| | | 838 | | // multiAdd1: [...|00000000|aaaaaabb|bbbbcccc|ccdddddd] |
| | 0 | 839 | | Vector512<int> multiAdd2 = Avx512BW.MultiplyAddAdjacent(multiAdd1, mergeConstant1); |
| | | 840 | | |
| | | 841 | | // Step 3: Pack 48 bytes |
| | 0 | 842 | | str = Avx512Vbmi.PermuteVar64x8(multiAdd2.AsByte(), vbmiPackedLanesControl).AsSByte(); |
| | | 843 | | |
| | 0 | 844 | | AssertWrite<Vector512<sbyte>>(dest, destStart, destLength); |
| | 0 | 845 | | str.Store((sbyte*)dest); |
| | 0 | 846 | | src += 64; |
| | 0 | 847 | | dest += 48; |
| | | 848 | | } |
| | 0 | 849 | | while (src <= srcEnd); |
| | | 850 | | |
| | 0 | 851 | | srcBytes = src; |
| | 0 | 852 | | destBytes = dest; |
| | 0 | 853 | | } |
| | | 854 | | |
| | | 855 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 856 | | [CompExactlyDependsOn(typeof(Avx2))] |
| | | 857 | | private static unsafe void Avx2Decode<TBase64Decoder, T>(TBase64Decoder decoder, ref T* srcBytes, ref byte* dest |
| | | 858 | | where TBase64Decoder : IBase64Decoder<T> |
| | | 859 | | where T : unmanaged |
| | | 860 | | { |
| | | 861 | | // If we have AVX2 support, pick off 32 bytes at a time for as long as we can, |
| | | 862 | | // but make sure that we quit before seeing any == markers at the end of the |
| | | 863 | | // string. Also, because we write 8 zeroes at the end of the output, ensure |
| | | 864 | | // that there are at least 11 valid bytes of input data remaining to close the |
| | | 865 | | // gap. 32 + 2 + 11 = 45 bytes. |
| | | 866 | | |
| | | 867 | | // See SSSE3-version below for an explanation of how the code works. |
| | | 868 | | |
| | | 869 | | // The JIT won't hoist these "constants", so help it |
| | 132374 | 870 | | Vector256<sbyte> lutHi = Vector256.Create(decoder.Avx2LutHigh); |
| | | 871 | | |
| | 132374 | 872 | | Vector256<sbyte> lutLo = Vector256.Create(decoder.Avx2LutLow); |
| | | 873 | | |
| | 132374 | 874 | | Vector256<sbyte> lutShift = Vector256.Create(decoder.Avx2LutShift); |
| | | 875 | | |
| | 132374 | 876 | | Vector256<sbyte> packBytesInLaneMask = Vector256.Create( |
| | 132374 | 877 | | 2, 1, 0, 6, |
| | 132374 | 878 | | 5, 4, 10, 9, |
| | 132374 | 879 | | 8, 14, 13, 12, |
| | 132374 | 880 | | -1, -1, -1, -1, |
| | 132374 | 881 | | 2, 1, 0, 6, |
| | 132374 | 882 | | 5, 4, 10, 9, |
| | 132374 | 883 | | 8, 14, 13, 12, |
| | 132374 | 884 | | -1, -1, -1, -1); |
| | | 885 | | |
| | 132374 | 886 | | Vector256<int> packLanesControl = Vector256.Create( |
| | 132374 | 887 | | 0, 0, 0, 0, |
| | 132374 | 888 | | 1, 0, 0, 0, |
| | 132374 | 889 | | 2, 0, 0, 0, |
| | 132374 | 890 | | 4, 0, 0, 0, |
| | 132374 | 891 | | 5, 0, 0, 0, |
| | 132374 | 892 | | 6, 0, 0, 0, |
| | 132374 | 893 | | -1, -1, -1, -1, |
| | 132374 | 894 | | -1, -1, -1, -1).AsInt32(); |
| | | 895 | | |
| | 132374 | 896 | | Vector256<sbyte> maskSlashOrUnderscore = Vector256.Create((sbyte)decoder.MaskSlashOrUnderscore); |
| | 132374 | 897 | | Vector256<sbyte> shiftForUnderscore = Vector256.Create((sbyte)33); |
| | 132374 | 898 | | Vector256<sbyte> mergeConstant0 = Vector256.Create(0x01400140).AsSByte(); |
| | 132374 | 899 | | Vector256<short> mergeConstant1 = Vector256.Create(0x00011000).AsInt16(); |
| | | 900 | | |
| | 132374 | 901 | | T* src = srcBytes; |
| | 132374 | 902 | | byte* dest = destBytes; |
| | | 903 | | |
| | | 904 | | //while (remaining >= 45) |
| | | 905 | | do |
| | | 906 | | { |
| | 337401 | 907 | | if (!decoder.TryLoadAvxVector256(src, srcStart, sourceLength, out Vector256<sbyte> str)) |
| | | 908 | | { |
| | | 909 | | break; |
| | | 910 | | } |
| | | 911 | | |
| | 337401 | 912 | | Vector256<sbyte> hiNibbles = ((str.AsInt32()) >>> 4).AsSByte() & maskSlashOrUnderscore; |
| | | 913 | | |
| | 337401 | 914 | | if (!decoder.TryDecode256Core(str, hiNibbles, maskSlashOrUnderscore, lutLo, lutHi, lutShift, shiftForUnd |
| | | 915 | | { |
| | | 916 | | break; |
| | | 917 | | } |
| | | 918 | | |
| | | 919 | | // in, lower lane, bits, upper case are most significant bits, lower case are least significant bits: |
| | | 920 | | // 00llllll 00kkkkLL 00jjKKKK 00JJJJJJ |
| | | 921 | | // 00iiiiii 00hhhhII 00ggHHHH 00GGGGGG |
| | | 922 | | // 00ffffff 00eeeeFF 00ddEEEE 00DDDDDD |
| | | 923 | | // 00cccccc 00bbbbCC 00aaBBBB 00AAAAAA |
| | | 924 | | |
| | 210768 | 925 | | Vector256<short> merge_ab_and_bc = Avx2.MultiplyAddAdjacent(str.AsByte(), mergeConstant0); |
| | | 926 | | // 0000kkkk LLllllll 0000JJJJ JJjjKKKK |
| | | 927 | | // 0000hhhh IIiiiiii 0000GGGG GGggHHHH |
| | | 928 | | // 0000eeee FFffffff 0000DDDD DDddEEEE |
| | | 929 | | // 0000bbbb CCcccccc 0000AAAA AAaaBBBB |
| | | 930 | | |
| | 210768 | 931 | | Vector256<int> output = Avx2.MultiplyAddAdjacent(merge_ab_and_bc, mergeConstant1); |
| | | 932 | | // 00000000 JJJJJJjj KKKKkkkk LLllllll |
| | | 933 | | // 00000000 GGGGGGgg HHHHhhhh IIiiiiii |
| | | 934 | | // 00000000 DDDDDDdd EEEEeeee FFffffff |
| | | 935 | | // 00000000 AAAAAAaa BBBBbbbb CCcccccc |
| | | 936 | | |
| | | 937 | | // Pack bytes together in each lane: |
| | 210768 | 938 | | output = Avx2.Shuffle(output.AsSByte(), packBytesInLaneMask).AsInt32(); |
| | | 939 | | // 00000000 00000000 00000000 00000000 |
| | | 940 | | // LLllllll KKKKkkkk JJJJJJjj IIiiiiii |
| | | 941 | | // HHHHhhhh GGGGGGgg FFffffff EEEEeeee |
| | | 942 | | // DDDDDDdd CCcccccc BBBBbbbb AAAAAAaa |
| | | 943 | | |
| | | 944 | | // Pack lanes |
| | 210768 | 945 | | str = Avx2.PermuteVar8x32(output, packLanesControl).AsSByte(); |
| | | 946 | | |
| | 210768 | 947 | | AssertWrite<Vector256<sbyte>>(dest, destStart, destLength); |
| | 210768 | 948 | | Avx.Store(dest, str.AsByte()); |
| | | 949 | | |
| | 210768 | 950 | | src += 32; |
| | 210768 | 951 | | dest += 24; |
| | | 952 | | } |
| | 210768 | 953 | | while (src <= srcEnd); |
| | | 954 | | |
| | 132374 | 955 | | srcBytes = src; |
| | 132374 | 956 | | destBytes = dest; |
| | 132374 | 957 | | } |
| | | 958 | | |
| | | 959 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 960 | | [CompExactlyDependsOn(typeof(Ssse3))] |
| | | 961 | | [CompExactlyDependsOn(typeof(AdvSimd.Arm64))] |
| | | 962 | | internal static Vector128<byte> SimdShuffle(Vector128<byte> left, Vector128<byte> right, Vector128<byte> mask8F) |
| | | 963 | | { |
| | 582044 | 964 | | Debug.Assert((Ssse3.IsSupported || AdvSimd.Arm64.IsSupported) && BitConverter.IsLittleEndian); |
| | | 965 | | |
| | 582044 | 966 | | if (Ssse3.IsSupported) |
| | | 967 | | { |
| | 582044 | 968 | | return Ssse3.Shuffle(left, right); |
| | | 969 | | } |
| | | 970 | | else |
| | | 971 | | { |
| | 0 | 972 | | return AdvSimd.Arm64.VectorTableLookup(left, right & mask8F); |
| | | 973 | | } |
| | | 974 | | } |
| | | 975 | | |
| | | 976 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 977 | | [CompExactlyDependsOn(typeof(AdvSimd.Arm64))] |
| | | 978 | | private static unsafe void AdvSimdDecode<TBase64Decoder, T>(TBase64Decoder decoder, ref T* srcBytes, ref byte* d |
| | | 979 | | where TBase64Decoder : IBase64Decoder<T> |
| | | 980 | | where T : unmanaged |
| | | 981 | | { |
| | | 982 | | // C# implementation of https://github.com/aklomp/base64/blob/3a5add8652076612a8407627a42c768736a4263f/lib/a |
| | | 983 | | // If we have AdvSimd support, pick off 64 bytes at a time for as long as we can, |
| | | 984 | | // but make sure that we quit before seeing any == markers at the end of the |
| | | 985 | | // string. 64 + 2 = 66 bytes. |
| | | 986 | | |
| | | 987 | | // In the decoding process, we want to map each byte, representing a Base64 value, to its 6-bit (0-63) repre |
| | | 988 | | // It uses the following mapping. Values outside the following groups are invalid and, we abort decoding whe |
| | | 989 | | // |
| | | 990 | | // # From To Char |
| | | 991 | | // 1 [43] [62] + |
| | | 992 | | // 2 [47] [63] / |
| | | 993 | | // 3 [48..57] [52..61] 0..9 |
| | | 994 | | // 4 [65..90] [0..25] A..Z |
| | | 995 | | // 5 [97..122] [26..51] a..z |
| | | 996 | | // |
| | | 997 | | // To map an input value to its Base64 representation, we use look-up tables 'decLutOne' and 'decLutTwo'. |
| | | 998 | | // 'decLutOne' helps to map groups 1, 2 and 3 while 'decLutTwo' maps groups 4 and 5 in the above list. |
| | | 999 | | // After mapping, each value falls between 0-63. Consequently, the last six bits of each byte now hold a val |
| | | 1000 | | // We then compress four such bytes (with valid 4 * 6 = 24 bits) to three UTF8 bytes (3 * 8 = 24 bits). |
| | | 1001 | | // For faster decoding, we use SIMD operations that allow the processing of multiple bytes together. |
| | | 1002 | | // However, the compress operation on adjacent values of a vector could be slower. Thus, we de-interleave wh |
| | | 1003 | | // the input bytes that store adjacent bytes in separate vectors. This later simplifies the compress step wi |
| | | 1004 | | // of logical operations. This requires interleaving while storing the decoded result. |
| | | 1005 | | |
| | | 1006 | | // Values in 'decLutOne' maps input values from 0 to 63. |
| | | 1007 | | // 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 |
| | | 1008 | | // 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 |
| | | 1009 | | // 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 62, 255, 255, 255, 63 |
| | | 1010 | | // 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 255, 255, 255, 255, 255, 255 |
| | | 1011 | | var decLutOne = (Vector128<byte>.AllBitsSet, |
| | | 1012 | | Vector128<byte>.AllBitsSet, |
| | | 1013 | | Vector128.Create(decoder.AdvSimdLutOne3).AsByte(), |
| | | 1014 | | Vector128.Create(0x37363534, 0x3B3A3938, 0xFFFF3D3C, 0xFFFFFFFF).AsByte()); |
| | | 1015 | | |
| | | 1016 | | // Values in 'decLutTwo' maps input values from 63 to 127. |
| | | 1017 | | // 0, 255, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13 |
| | | 1018 | | // 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 255, 255, 255, 255 |
| | | 1019 | | // 255, 255, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39 |
| | | 1020 | | // 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 255, 255, 255, 255 |
| | | 1021 | | var decLutTwo = (Vector128.Create(0x0100FF00, 0x05040302, 0x09080706, 0x0D0C0B0A).AsByte(), |
| | | 1022 | | Vector128.Create(0x11100F0E, 0x15141312, 0x19181716, 0xFFFFFFFF).AsByte(), |
| | | 1023 | | Vector128.Create(decoder.AdvSimdLutTwo3Uint1, 0x1F1E1D1C, 0x23222120, 0x27262524).AsByte(), |
| | | 1024 | | Vector128.Create(0x2B2A2928, 0x2F2E2D2C, 0x33323130, 0xFFFFFFFF).AsByte()); |
| | | 1025 | | |
| | | 1026 | | T* src = srcBytes; |
| | | 1027 | | byte* dest = destBytes; |
| | | 1028 | | Vector128<byte> offset = Vector128.Create<byte>(63); |
| | | 1029 | | |
| | | 1030 | | do |
| | | 1031 | | { |
| | | 1032 | | // Step 1: Load 64 bytes and de-interleave. |
| | | 1033 | | if (!decoder.TryLoadArmVector128x4(src, srcStart, sourceLength, |
| | | 1034 | | out Vector128<byte> str1, out Vector128<byte> str2, out Vector128<byte> str3, out Vector128<byte> st |
| | | 1035 | | { |
| | | 1036 | | break; |
| | | 1037 | | } |
| | | 1038 | | |
| | | 1039 | | // Step 2: Map each valid input to its Base64 value. |
| | | 1040 | | // We use two look-ups to compute partial results and combine them later. |
| | | 1041 | | |
| | | 1042 | | // Step 2.1: Detect valid Base64 values from the first three groups. Maps input as, |
| | | 1043 | | // 0 to 63 (Invalid) => 255 |
| | | 1044 | | // 0 to 63 (Valid) => Their Base64 equivalent |
| | | 1045 | | // 64 to 255 => 0 |
| | | 1046 | | |
| | | 1047 | | // Each input value acts as an index in the look-up table 'decLutOne'. |
| | | 1048 | | // e.g., for group 1: index 43 maps to 62 (Base64 '+'). |
| | | 1049 | | // Group 4 and 5 values are out-of-range (>64), so they are mapped to zero. |
| | | 1050 | | // Other valid indices but invalid values are mapped to 255. |
| | | 1051 | | Vector128<byte> decOne1 = AdvSimd.Arm64.VectorTableLookup(decLutOne, str1); |
| | | 1052 | | Vector128<byte> decOne2 = AdvSimd.Arm64.VectorTableLookup(decLutOne, str2); |
| | | 1053 | | Vector128<byte> decOne3 = AdvSimd.Arm64.VectorTableLookup(decLutOne, str3); |
| | | 1054 | | Vector128<byte> decOne4 = AdvSimd.Arm64.VectorTableLookup(decLutOne, str4); |
| | | 1055 | | |
| | | 1056 | | // Step 2.2: Detect valid Base64 values from groups 4 and 5. Maps input as, |
| | | 1057 | | // 0 to 63 => 0 |
| | | 1058 | | // 64 to 122 (Valid) => Their Base64 equivalent |
| | | 1059 | | // 64 to 122 (Invalid) => 255 |
| | | 1060 | | // 123 to 255 => Remains unchanged |
| | | 1061 | | |
| | | 1062 | | // Subtract/offset each input value by 63 so that it can be used as a valid offset. |
| | | 1063 | | // Subtract saturate makes values from the first three groups set to zero that are |
| | | 1064 | | // then mapped to zero in the subsequent look-up. |
| | | 1065 | | Vector128<byte> decTwo1 = AdvSimd.SubtractSaturate(str1, offset); |
| | | 1066 | | Vector128<byte> decTwo2 = AdvSimd.SubtractSaturate(str2, offset); |
| | | 1067 | | Vector128<byte> decTwo3 = AdvSimd.SubtractSaturate(str3, offset); |
| | | 1068 | | Vector128<byte> decTwo4 = AdvSimd.SubtractSaturate(str4, offset); |
| | | 1069 | | |
| | | 1070 | | // We use VTBX to map values where out-of-range indices are unchanged. |
| | | 1071 | | decTwo1 = AdvSimd.Arm64.VectorTableLookupExtension(decTwo1, decLutTwo, decTwo1); |
| | | 1072 | | decTwo2 = AdvSimd.Arm64.VectorTableLookupExtension(decTwo2, decLutTwo, decTwo2); |
| | | 1073 | | decTwo3 = AdvSimd.Arm64.VectorTableLookupExtension(decTwo3, decLutTwo, decTwo3); |
| | | 1074 | | decTwo4 = AdvSimd.Arm64.VectorTableLookupExtension(decTwo4, decLutTwo, decTwo4); |
| | | 1075 | | |
| | | 1076 | | // Step 3: Combine the partial result. |
| | | 1077 | | // Each look-up above maps valid values to their Base64 equivalent or zero. |
| | | 1078 | | // Thus the intermediate results 'decOne' and 'decTwo' could be OR-ed to get final values. |
| | | 1079 | | str1 = (decOne1 | decTwo1); |
| | | 1080 | | str2 = (decOne2 | decTwo2); |
| | | 1081 | | str3 = (decOne3 | decTwo3); |
| | | 1082 | | str4 = (decOne4 | decTwo4); |
| | | 1083 | | |
| | | 1084 | | // Step 4: Detect an invalid input value. |
| | | 1085 | | // Invalid values < 122 are set to 255 while the ones above 122 are unchanged. |
| | | 1086 | | // Check for invalid input, any value larger than 63. |
| | | 1087 | | Vector128<byte> classified = (Vector128.GreaterThan(str1, offset) |
| | | 1088 | | | Vector128.GreaterThan(str2, offset) |
| | | 1089 | | | Vector128.GreaterThan(str3, offset) |
| | | 1090 | | | Vector128.GreaterThan(str4, offset)); |
| | | 1091 | | |
| | | 1092 | | // Check that all bits are zero. |
| | | 1093 | | if (classified != Vector128<byte>.Zero) |
| | | 1094 | | { |
| | | 1095 | | break; |
| | | 1096 | | } |
| | | 1097 | | |
| | | 1098 | | // Step 5: Compress four bytes into three. |
| | | 1099 | | Vector128<byte> res1 = ((str1 << 2) | (str2 >> 4)); |
| | | 1100 | | Vector128<byte> res2 = ((str2 << 4) | (str3 >> 2)); |
| | | 1101 | | Vector128<byte> res3 = ((str3 << 6) | str4); |
| | | 1102 | | |
| | | 1103 | | // Step 6: Interleave and store decoded results. |
| | | 1104 | | AssertWrite<Vector128<byte>>(dest, destStart, destLength); |
| | | 1105 | | AdvSimd.Arm64.StoreVectorAndZip(dest, (res1, res2, res3)); |
| | | 1106 | | |
| | | 1107 | | src += 64; |
| | | 1108 | | dest += 48; |
| | | 1109 | | } |
| | | 1110 | | while (src <= srcEnd); |
| | | 1111 | | |
| | | 1112 | | srcBytes = src; |
| | | 1113 | | destBytes = dest; |
| | | 1114 | | } |
| | | 1115 | | |
| | | 1116 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1117 | | [CompExactlyDependsOn(typeof(AdvSimd.Arm64))] |
| | | 1118 | | [CompExactlyDependsOn(typeof(Ssse3))] |
| | | 1119 | | private static unsafe void Vector128Decode<TBase64Decoder, T>(TBase64Decoder decoder, ref T* srcBytes, ref byte* |
| | | 1120 | | where TBase64Decoder : IBase64Decoder<T> |
| | | 1121 | | where T : unmanaged |
| | | 1122 | | { |
| | | 1123 | | Debug.Assert((Ssse3.IsSupported || AdvSimd.Arm64.IsSupported) && BitConverter.IsLittleEndian); |
| | | 1124 | | |
| | | 1125 | | // If we have Vector128 support, pick off 16 bytes at a time for as long as we can, |
| | | 1126 | | // but make sure that we quit before seeing any == markers at the end of the |
| | | 1127 | | // string. Also, because we write four zeroes at the end of the output, ensure |
| | | 1128 | | // that there are at least 6 valid bytes of input data remaining to close the |
| | | 1129 | | // gap. 16 + 2 + 6 = 24 bytes. |
| | | 1130 | | |
| | | 1131 | | // The input consists of six character sets in the Base64 alphabet, |
| | | 1132 | | // which we need to map back to the 6-bit values they represent. |
| | | 1133 | | // There are three ranges, two singles, and then there's the rest. |
| | | 1134 | | // |
| | | 1135 | | // # From To Add Characters |
| | | 1136 | | // 1 [43] [62] +19 + |
| | | 1137 | | // 2 [47] [63] +16 / |
| | | 1138 | | // 3 [48..57] [52..61] +4 0..9 |
| | | 1139 | | // 4 [65..90] [0..25] -65 A..Z |
| | | 1140 | | // 5 [97..122] [26..51] -71 a..z |
| | | 1141 | | // (6) Everything else => invalid input |
| | | 1142 | | |
| | | 1143 | | // We will use LUTS for character validation & offset computation |
| | | 1144 | | // Remember that 0x2X and 0x0X are the same index for _mm_shuffle_epi8, |
| | | 1145 | | // this allows to mask with 0x2F instead of 0x0F and thus save one constant declaration (register and/or mem |
| | | 1146 | | |
| | | 1147 | | // For offsets: |
| | | 1148 | | // Perfect hash for lut = ((src>>4)&0x2F)+((src==0x2F)?0xFF:0x00) |
| | | 1149 | | // 0000 = garbage |
| | | 1150 | | // 0001 = / |
| | | 1151 | | // 0010 = + |
| | | 1152 | | // 0011 = 0-9 |
| | | 1153 | | // 0100 = A-Z |
| | | 1154 | | // 0101 = A-Z |
| | | 1155 | | // 0110 = a-z |
| | | 1156 | | // 0111 = a-z |
| | | 1157 | | // 1000 >= garbage |
| | | 1158 | | |
| | | 1159 | | // For validation, here's the table. |
| | | 1160 | | // A character is valid if and only if the AND of the 2 lookups equals 0: |
| | | 1161 | | |
| | | 1162 | | // hi \ lo 0000 0001 0010 0011 0100 0101 0110 0111 1000 1001 1010 1011 1100 1101 1110 1111 |
| | | 1163 | | // LUT 0x15 0x11 0x11 0x11 0x11 0x11 0x11 0x11 0x11 0x11 0x13 0x1A 0x1B 0x1B 0x1B 0x1A |
| | | 1164 | | |
| | | 1165 | | // 0000 0X10 char NUL SOH STX ETX EOT ENQ ACK BEL BS HT LF VT FF CR SO SI |
| | | 1166 | | // andlut 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 |
| | | 1167 | | |
| | | 1168 | | // 0001 0x10 char DLE DC1 DC2 DC3 DC4 NAK SYN ETB CAN EM SUB ESC FS GS RS US |
| | | 1169 | | // andlut 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 |
| | | 1170 | | |
| | | 1171 | | // 0010 0x01 char ! " # $ % & ' ( ) * + , - . / |
| | | 1172 | | // andlut 0x01 0x01 0x01 0x01 0x01 0x01 0x01 0x01 0x01 0x01 0x01 0x00 0x01 0x01 0x01 0x00 |
| | | 1173 | | |
| | | 1174 | | // 0011 0x02 char 0 1 2 3 4 5 6 7 8 9 : ; < = > ? |
| | | 1175 | | // andlut 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x02 0x02 0x02 0x02 0x02 0x02 |
| | | 1176 | | |
| | | 1177 | | // 0100 0x04 char @ A B C D E F G H I J K L M N 0 |
| | | 1178 | | // andlut 0x04 0x00 0x00 0x00 0X00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 |
| | | 1179 | | |
| | | 1180 | | // 0101 0x08 char P Q R S T U V W X Y Z [ \ ] ^ _ |
| | | 1181 | | // andlut 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x08 0x08 0x08 0x08 0x08 |
| | | 1182 | | |
| | | 1183 | | // 0110 0x04 char ` a b c d e f g h i j k l m n o |
| | | 1184 | | // andlut 0x04 0x00 0x00 0x00 0X00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 |
| | | 1185 | | // 0111 0X08 char p q r s t u v w x y z { | } ~ |
| | | 1186 | | // andlut 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x00 0x08 0x08 0x08 0x08 0x08 |
| | | 1187 | | |
| | | 1188 | | // 1000 0x10 andlut 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 |
| | | 1189 | | // 1001 0x10 andlut 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 |
| | | 1190 | | // 1010 0x10 andlut 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 |
| | | 1191 | | // 1011 0x10 andlut 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 |
| | | 1192 | | // 1100 0x10 andlut 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 |
| | | 1193 | | // 1101 0x10 andlut 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 |
| | | 1194 | | // 1110 0x10 andlut 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 |
| | | 1195 | | // 1111 0x10 andlut 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 0x10 |
| | | 1196 | | |
| | | 1197 | | // The JIT won't hoist these "constants", so help it |
| | 134537 | 1198 | | Vector128<byte> lutHi = Vector128.Create(decoder.Vector128LutHigh).AsByte(); |
| | 134537 | 1199 | | Vector128<byte> lutLo = Vector128.Create(decoder.Vector128LutLow).AsByte(); |
| | 134537 | 1200 | | Vector128<sbyte> lutShift = Vector128.Create(decoder.Vector128LutShift).AsSByte(); |
| | 134537 | 1201 | | Vector128<sbyte> packBytesMask = Vector128.Create(0x06000102, 0x090A0405, 0x0C0D0E08, 0xffffffff).AsSByte(); |
| | 134537 | 1202 | | Vector128<byte> mergeConstant0 = Vector128.Create(0x01400140).AsByte(); |
| | 134537 | 1203 | | Vector128<short> mergeConstant1 = Vector128.Create(0x00011000).AsInt16(); |
| | 134537 | 1204 | | Vector128<byte> one = Vector128<byte>.One; |
| | 134537 | 1205 | | Vector128<byte> mask2F = Vector128.Create(decoder.MaskSlashOrUnderscore); |
| | 134537 | 1206 | | Vector128<byte> mask8F = Vector128.Create((byte)0x8F); |
| | 134537 | 1207 | | Vector128<byte> shiftForUnderscore = Vector128.Create((byte)33); |
| | 134537 | 1208 | | T* src = srcBytes; |
| | 134537 | 1209 | | byte* dest = destBytes; |
| | | 1210 | | |
| | | 1211 | | //while (remaining >= 24) |
| | | 1212 | | do |
| | | 1213 | | { |
| | 207394 | 1214 | | if (!decoder.TryLoadVector128(src, srcStart, sourceLength, out Vector128<byte> str)) |
| | | 1215 | | { |
| | | 1216 | | break; |
| | | 1217 | | } |
| | | 1218 | | |
| | | 1219 | | // lookup |
| | 207394 | 1220 | | Vector128<byte> hiNibbles = Vector128.ShiftRightLogical(str.AsInt32(), 4).AsByte() & mask2F; |
| | | 1221 | | |
| | 207394 | 1222 | | if (!decoder.TryDecode128Core(str, hiNibbles, mask2F, mask8F, lutLo, lutHi, lutShift, shiftForUnderscore |
| | | 1223 | | { |
| | | 1224 | | break; |
| | | 1225 | | } |
| | | 1226 | | |
| | | 1227 | | // in, bits, upper case are most significant bits, lower case are least significant bits |
| | | 1228 | | // 00llllll 00kkkkLL 00jjKKKK 00JJJJJJ |
| | | 1229 | | // 00iiiiii 00hhhhII 00ggHHHH 00GGGGGG |
| | | 1230 | | // 00ffffff 00eeeeFF 00ddEEEE 00DDDDDD |
| | | 1231 | | // 00cccccc 00bbbbCC 00aaBBBB 00AAAAAA |
| | | 1232 | | |
| | | 1233 | | Vector128<short> merge_ab_and_bc; |
| | 78840 | 1234 | | if (Ssse3.IsSupported) |
| | | 1235 | | { |
| | 78840 | 1236 | | merge_ab_and_bc = Ssse3.MultiplyAddAdjacent(str.AsByte(), mergeConstant0.AsSByte()); |
| | | 1237 | | } |
| | | 1238 | | else if (AdvSimd.Arm64.IsSupported) |
| | | 1239 | | { |
| | | 1240 | | Vector128<ushort> evens = AdvSimd.ShiftLeftLogicalWideningLower(AdvSimd.Arm64.UnzipEven(str, one).Ge |
| | | 1241 | | Vector128<ushort> odds = AdvSimd.Arm64.TransposeOdd(str, Vector128<byte>.Zero).AsUInt16(); |
| | | 1242 | | merge_ab_and_bc = Vector128.Add(evens, odds).AsInt16(); |
| | | 1243 | | } |
| | | 1244 | | else |
| | | 1245 | | { |
| | | 1246 | | // We explicitly recheck each IsSupported query to ensure that the trimmer can see which paths are l |
| | 0 | 1247 | | ThrowUnreachableException(); |
| | | 1248 | | merge_ab_and_bc = default; |
| | | 1249 | | } |
| | | 1250 | | // 0000kkkk LLllllll 0000JJJJ JJjjKKKK |
| | | 1251 | | // 0000hhhh IIiiiiii 0000GGGG GGggHHHH |
| | | 1252 | | // 0000eeee FFffffff 0000DDDD DDddEEEE |
| | | 1253 | | // 0000bbbb CCcccccc 0000AAAA AAaaBBBB |
| | | 1254 | | |
| | | 1255 | | Vector128<int> output; |
| | 78840 | 1256 | | if (Ssse3.IsSupported) |
| | | 1257 | | { |
| | 78840 | 1258 | | output = Sse2.MultiplyAddAdjacent(merge_ab_and_bc, mergeConstant1); |
| | | 1259 | | } |
| | | 1260 | | else if (AdvSimd.Arm64.IsSupported) |
| | | 1261 | | { |
| | | 1262 | | Vector128<int> ievens = AdvSimd.ShiftLeftLogicalWideningLower(AdvSimd.Arm64.UnzipEven(merge_ab_and_b |
| | | 1263 | | Vector128<int> iodds = AdvSimd.Arm64.TransposeOdd(merge_ab_and_bc, Vector128<short>.Zero).AsInt32(); |
| | | 1264 | | output = Vector128.Add(ievens, iodds).AsInt32(); |
| | | 1265 | | } |
| | | 1266 | | else |
| | | 1267 | | { |
| | | 1268 | | // We explicitly recheck each IsSupported query to ensure that the trimmer can see which paths are l |
| | 0 | 1269 | | ThrowUnreachableException(); |
| | | 1270 | | output = default; |
| | | 1271 | | } |
| | | 1272 | | // 00000000 JJJJJJjj KKKKkkkk LLllllll |
| | | 1273 | | // 00000000 GGGGGGgg HHHHhhhh IIiiiiii |
| | | 1274 | | // 00000000 DDDDDDdd EEEEeeee FFffffff |
| | | 1275 | | // 00000000 AAAAAAaa BBBBbbbb CCcccccc |
| | | 1276 | | |
| | | 1277 | | // Pack bytes together: |
| | 78840 | 1278 | | str = SimdShuffle(output.AsByte(), packBytesMask.AsByte(), mask8F); |
| | | 1279 | | // 00000000 00000000 00000000 00000000 |
| | | 1280 | | // LLllllll KKKKkkkk JJJJJJjj IIiiiiii |
| | | 1281 | | // HHHHhhhh GGGGGGgg FFffffff EEEEeeee |
| | | 1282 | | // DDDDDDdd CCcccccc BBBBbbbb AAAAAAaa |
| | | 1283 | | |
| | 78840 | 1284 | | AssertWrite<Vector128<sbyte>>(dest, destStart, destLength); |
| | 78840 | 1285 | | str.Store(dest); |
| | | 1286 | | |
| | 78840 | 1287 | | src += 16; |
| | 78840 | 1288 | | dest += 12; |
| | | 1289 | | } |
| | 78840 | 1290 | | while (src <= srcEnd); |
| | | 1291 | | |
| | 134537 | 1292 | | srcBytes = src; |
| | 134537 | 1293 | | destBytes = dest; |
| | 134537 | 1294 | | } |
| | | 1295 | | #endif |
| | | 1296 | | |
| | | 1297 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1298 | | private static unsafe void WriteThreeLowOrderBytes(byte* destination, int value) |
| | | 1299 | | { |
| | 1662878 | 1300 | | destination[0] = (byte)(value >> 16); |
| | 1662878 | 1301 | | destination[1] = (byte)(value >> 8); |
| | 1662878 | 1302 | | destination[2] = (byte)value; |
| | 1662878 | 1303 | | } |
| | | 1304 | | |
| | | 1305 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1306 | | internal static bool IsWhiteSpace(int value) |
| | | 1307 | | { |
| | 4402147 | 1308 | | Debug.Assert(value >= 0 && value <= ushort.MaxValue); |
| | | 1309 | | uint charMinusLowUInt32; |
| | 4402147 | 1310 | | return (int)((0xC8000100U << (short)(charMinusLowUInt32 = (ushort)(value - '\t'))) & (charMinusLowUInt32 - 3 |
| | | 1311 | | } |
| | | 1312 | | |
| | | 1313 | | internal readonly struct Base64DecoderByte : IBase64Decoder<byte> |
| | | 1314 | | { |
| | | 1315 | | // Pre-computing this table using a custom string(s_characters) and GenerateDecodingMapAndVerify (found in t |
| | | 1316 | | public ReadOnlySpan<sbyte> DecodingMap => |
| | 0 | 1317 | | [ |
| | 0 | 1318 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, |
| | 0 | 1319 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, |
| | 0 | 1320 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 62, -1, -1, -1, 63, //62 is placed at index 43 ( |
| | 0 | 1321 | | 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -1, -1, -1, //52-61 are placed at index |
| | 0 | 1322 | | -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, |
| | 0 | 1323 | | 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1, //0-25 are placed at index 6 |
| | 0 | 1324 | | -1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, |
| | 0 | 1325 | | 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, -1, -1, -1, -1, -1, //26-51 are placed at index |
| | 0 | 1326 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // Bytes over 122 ('z') are |
| | 0 | 1327 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // Hence, padding the map wi |
| | 0 | 1328 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, |
| | 0 | 1329 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, |
| | 0 | 1330 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, |
| | 0 | 1331 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, |
| | 0 | 1332 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, |
| | 0 | 1333 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, |
| | 0 | 1334 | | ]; |
| | | 1335 | | |
| | | 1336 | | public ReadOnlySpan<uint> VbmiLookup0 => |
| | 0 | 1337 | | [ |
| | 0 | 1338 | | 0x80808080, 0x80808080, 0x80808080, 0x80808080, |
| | 0 | 1339 | | 0x80808080, 0x80808080, 0x80808080, 0x80808080, |
| | 0 | 1340 | | 0x80808080, 0x80808080, 0x3e808080, 0x3f808080, |
| | 0 | 1341 | | 0x37363534, 0x3b3a3938, 0x80803d3c, 0x80808080 |
| | 0 | 1342 | | ]; |
| | | 1343 | | |
| | | 1344 | | public ReadOnlySpan<uint> VbmiLookup1 => |
| | 0 | 1345 | | [ |
| | 0 | 1346 | | 0x02010080, 0x06050403, 0x0a090807, 0x0e0d0c0b, |
| | 0 | 1347 | | 0x1211100f, 0x16151413, 0x80191817, 0x80808080, |
| | 0 | 1348 | | 0x1c1b1a80, 0x201f1e1d, 0x24232221, 0x28272625, |
| | 0 | 1349 | | 0x2c2b2a29, 0x302f2e2d, 0x80333231, 0x80808080 |
| | 0 | 1350 | | ]; |
| | | 1351 | | |
| | | 1352 | | public ReadOnlySpan<sbyte> Avx2LutHigh => |
| | 0 | 1353 | | [ |
| | 0 | 1354 | | 0x10, 0x10, 0x01, 0x02, |
| | 0 | 1355 | | 0x04, 0x08, 0x04, 0x08, |
| | 0 | 1356 | | 0x10, 0x10, 0x10, 0x10, |
| | 0 | 1357 | | 0x10, 0x10, 0x10, 0x10, |
| | 0 | 1358 | | 0x10, 0x10, 0x01, 0x02, |
| | 0 | 1359 | | 0x04, 0x08, 0x04, 0x08, |
| | 0 | 1360 | | 0x10, 0x10, 0x10, 0x10, |
| | 0 | 1361 | | 0x10, 0x10, 0x10, 0x10 |
| | 0 | 1362 | | ]; |
| | | 1363 | | |
| | | 1364 | | public ReadOnlySpan<sbyte> Avx2LutLow => |
| | 0 | 1365 | | [ |
| | 0 | 1366 | | 0x15, 0x11, 0x11, 0x11, |
| | 0 | 1367 | | 0x11, 0x11, 0x11, 0x11, |
| | 0 | 1368 | | 0x11, 0x11, 0x13, 0x1A, |
| | 0 | 1369 | | 0x1B, 0x1B, 0x1B, 0x1A, |
| | 0 | 1370 | | 0x15, 0x11, 0x11, 0x11, |
| | 0 | 1371 | | 0x11, 0x11, 0x11, 0x11, |
| | 0 | 1372 | | 0x11, 0x11, 0x13, 0x1A, |
| | 0 | 1373 | | 0x1B, 0x1B, 0x1B, 0x1A |
| | 0 | 1374 | | ]; |
| | | 1375 | | |
| | | 1376 | | public ReadOnlySpan<sbyte> Avx2LutShift => |
| | 0 | 1377 | | [ |
| | 0 | 1378 | | 0, 16, 19, 4, |
| | 0 | 1379 | | -65, -65, -71, -71, |
| | 0 | 1380 | | 0, 0, 0, 0, |
| | 0 | 1381 | | 0, 0, 0, 0, |
| | 0 | 1382 | | 0, 16, 19, 4, |
| | 0 | 1383 | | -65, -65, -71, -71, |
| | 0 | 1384 | | 0, 0, 0, 0, |
| | 0 | 1385 | | 0, 0, 0, 0 |
| | 0 | 1386 | | ]; |
| | | 1387 | | |
| | 0 | 1388 | | public byte MaskSlashOrUnderscore => (byte)'/'; |
| | | 1389 | | |
| | 0 | 1390 | | public ReadOnlySpan<int> Vector128LutHigh => [0x02011010, 0x08040804, 0x10101010, 0x10101010]; |
| | | 1391 | | |
| | 0 | 1392 | | public ReadOnlySpan<int> Vector128LutLow => [0x11111115, 0x11111111, 0x1A131111, 0x1A1B1B1B]; |
| | | 1393 | | |
| | 0 | 1394 | | public ReadOnlySpan<uint> Vector128LutShift => [0x04131000, 0xb9b9bfbf, 0x00000000, 0x00000000]; |
| | | 1395 | | |
| | 0 | 1396 | | public ReadOnlySpan<uint> AdvSimdLutOne3 => [0xFFFFFFFF, 0xFFFFFFFF, 0x3EFFFFFF, 0x3FFFFFFF]; |
| | | 1397 | | |
| | 0 | 1398 | | public uint AdvSimdLutTwo3Uint1 => 0x1B1AFFFF; |
| | | 1399 | | |
| | 0 | 1400 | | public int GetMaxDecodedLength(int utf8Length) => Base64.GetMaxDecodedFromUtf8Length(utf8Length); |
| | | 1401 | | |
| | 0 | 1402 | | public bool IsInvalidLength(int bufferLength) => bufferLength % 4 != 0; // only decode input if it is a mult |
| | | 1403 | | |
| | 0 | 1404 | | public bool IsValidPadding(uint padChar) => padChar == EncodingPad; |
| | | 1405 | | |
| | 0 | 1406 | | public int SrcLength(bool _, int utf8Length) => utf8Length & ~0x3; // only decode input up to the closest m |
| | | 1407 | | |
| | | 1408 | | #if NET |
| | | 1409 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1410 | | [CompExactlyDependsOn(typeof(AdvSimd.Arm64))] |
| | | 1411 | | [CompExactlyDependsOn(typeof(Ssse3))] |
| | | 1412 | | public bool TryDecode128Core( |
| | | 1413 | | Vector128<byte> str, |
| | | 1414 | | Vector128<byte> hiNibbles, |
| | | 1415 | | Vector128<byte> maskSlashOrUnderscore, |
| | | 1416 | | Vector128<byte> mask8F, |
| | | 1417 | | Vector128<byte> lutLow, |
| | | 1418 | | Vector128<byte> lutHigh, |
| | | 1419 | | Vector128<sbyte> lutShift, |
| | | 1420 | | Vector128<byte> _, |
| | | 1421 | | out Vector128<byte> result) |
| | | 1422 | | { |
| | 0 | 1423 | | Vector128<byte> loNibbles = str & maskSlashOrUnderscore; |
| | 0 | 1424 | | Vector128<byte> hi = SimdShuffle(lutHigh, hiNibbles, mask8F); |
| | 0 | 1425 | | Vector128<byte> lo = SimdShuffle(lutLow, loNibbles, mask8F); |
| | | 1426 | | |
| | | 1427 | | // Check for invalid input: if any "and" values from lo and hi are not zero, |
| | | 1428 | | // fall back on bytewise code to do error checking and reporting: |
| | 0 | 1429 | | if ((lo & hi) != Vector128<byte>.Zero) |
| | | 1430 | | { |
| | 0 | 1431 | | result = default; |
| | 0 | 1432 | | return false; |
| | | 1433 | | } |
| | | 1434 | | |
| | 0 | 1435 | | Vector128<byte> eq2F = Vector128.Equals(str, maskSlashOrUnderscore); |
| | 0 | 1436 | | Vector128<byte> shift = SimdShuffle(lutShift.AsByte(), (eq2F + hiNibbles), mask8F); |
| | | 1437 | | |
| | | 1438 | | // Now simply add the delta values to the input: |
| | 0 | 1439 | | result = str + shift; |
| | | 1440 | | |
| | 0 | 1441 | | return true; |
| | | 1442 | | } |
| | | 1443 | | |
| | | 1444 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1445 | | [CompExactlyDependsOn(typeof(Avx2))] |
| | | 1446 | | public bool TryDecode256Core( |
| | | 1447 | | Vector256<sbyte> str, |
| | | 1448 | | Vector256<sbyte> hiNibbles, |
| | | 1449 | | Vector256<sbyte> maskSlashOrUnderscore, |
| | | 1450 | | Vector256<sbyte> lutLow, |
| | | 1451 | | Vector256<sbyte> lutHigh, |
| | | 1452 | | Vector256<sbyte> lutShift, |
| | | 1453 | | Vector256<sbyte> _, |
| | | 1454 | | out Vector256<sbyte> result) |
| | | 1455 | | { |
| | 0 | 1456 | | Vector256<sbyte> loNibbles = str & maskSlashOrUnderscore; |
| | 0 | 1457 | | Vector256<sbyte> hi = Avx2.Shuffle(lutHigh, hiNibbles); |
| | 0 | 1458 | | Vector256<sbyte> lo = Avx2.Shuffle(lutLow, loNibbles); |
| | | 1459 | | |
| | 0 | 1460 | | if ((lo & hi) != Vector256<sbyte>.Zero) |
| | | 1461 | | { |
| | 0 | 1462 | | result = default; |
| | 0 | 1463 | | return false; |
| | | 1464 | | } |
| | | 1465 | | |
| | 0 | 1466 | | Vector256<sbyte> eq2F = Avx2.CompareEqual(str, maskSlashOrUnderscore); |
| | 0 | 1467 | | Vector256<sbyte> shift = Avx2.Shuffle(lutShift, eq2F + hiNibbles); |
| | | 1468 | | |
| | 0 | 1469 | | result = str + shift; |
| | | 1470 | | |
| | 0 | 1471 | | return true; |
| | | 1472 | | } |
| | | 1473 | | |
| | | 1474 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1475 | | public unsafe bool TryLoadVector512(byte* src, byte* srcStart, int sourceLength, out Vector512<sbyte> str) |
| | | 1476 | | { |
| | 0 | 1477 | | AssertRead<Vector512<sbyte>>(src, srcStart, sourceLength); |
| | 0 | 1478 | | str = Vector512.Load(src).AsSByte(); |
| | 0 | 1479 | | return true; |
| | | 1480 | | } |
| | | 1481 | | |
| | | 1482 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1483 | | [CompExactlyDependsOn(typeof(Avx2))] |
| | | 1484 | | public unsafe bool TryLoadAvxVector256(byte* src, byte* srcStart, int sourceLength, out Vector256<sbyte> str |
| | | 1485 | | { |
| | 149280 | 1486 | | AssertRead<Vector256<sbyte>>(src, srcStart, sourceLength); |
| | 149280 | 1487 | | str = Avx.LoadVector256(src).AsSByte(); |
| | 149280 | 1488 | | return true; |
| | | 1489 | | } |
| | | 1490 | | |
| | | 1491 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1492 | | public unsafe bool TryLoadVector128(byte* src, byte* srcStart, int sourceLength, out Vector128<byte> str) |
| | | 1493 | | { |
| | 201902 | 1494 | | AssertRead<Vector128<sbyte>>(src, srcStart, sourceLength); |
| | 201902 | 1495 | | str = Vector128.LoadUnsafe(ref *src); |
| | 201902 | 1496 | | return true; |
| | | 1497 | | } |
| | | 1498 | | |
| | | 1499 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1500 | | [CompExactlyDependsOn(typeof(AdvSimd.Arm64))] |
| | | 1501 | | public unsafe bool TryLoadArmVector128x4(byte* src, byte* srcStart, int sourceLength, |
| | | 1502 | | out Vector128<byte> str1, out Vector128<byte> str2, out Vector128<byte> str3, out Vector128<byte> str4) |
| | | 1503 | | { |
| | 0 | 1504 | | AssertRead<Vector128<byte>>(src, srcStart, sourceLength); |
| | 0 | 1505 | | (str1, str2, str3, str4) = AdvSimd.Arm64.Load4xVector128AndUnzip(src); |
| | | 1506 | | |
| | 0 | 1507 | | return true; |
| | | 1508 | | } |
| | | 1509 | | #endif // NET |
| | | 1510 | | |
| | | 1511 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1512 | | public unsafe int DecodeFourElements(byte* source, ref sbyte decodingMap) |
| | | 1513 | | { |
| | | 1514 | | // The 'source' span expected to have at least 4 elements, and the 'decodingMap' consists 256 sbytes |
| | 1282589 | 1515 | | uint t0 = source[0]; |
| | 1282589 | 1516 | | uint t1 = source[1]; |
| | 1282589 | 1517 | | uint t2 = source[2]; |
| | 1282589 | 1518 | | uint t3 = source[3]; |
| | | 1519 | | |
| | 1282589 | 1520 | | int i0 = Unsafe.Add(ref decodingMap, (int)t0); |
| | 1282589 | 1521 | | int i1 = Unsafe.Add(ref decodingMap, (int)t1); |
| | 1282589 | 1522 | | int i2 = Unsafe.Add(ref decodingMap, (int)t2); |
| | 1282589 | 1523 | | int i3 = Unsafe.Add(ref decodingMap, (int)t3); |
| | | 1524 | | |
| | 1282589 | 1525 | | i0 <<= 18; |
| | 1282589 | 1526 | | i1 <<= 12; |
| | 1282589 | 1527 | | i2 <<= 6; |
| | | 1528 | | |
| | 1282589 | 1529 | | i0 |= i3; |
| | 1282589 | 1530 | | i1 |= i2; |
| | | 1531 | | |
| | 1282589 | 1532 | | i0 |= i1; |
| | 1282589 | 1533 | | return i0; |
| | | 1534 | | } |
| | | 1535 | | |
| | | 1536 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1537 | | public unsafe int DecodeRemaining(byte* srcEnd, ref sbyte decodingMap, long remaining, out uint t2, out uint |
| | | 1538 | | { |
| | | 1539 | | uint t0; |
| | | 1540 | | uint t1; |
| | 5152 | 1541 | | t2 = EncodingPad; |
| | 5152 | 1542 | | t3 = EncodingPad; |
| | | 1543 | | switch (remaining) |
| | | 1544 | | { |
| | | 1545 | | case 2: |
| | 1016 | 1546 | | t0 = srcEnd[-2]; |
| | 1016 | 1547 | | t1 = srcEnd[-1]; |
| | 1016 | 1548 | | break; |
| | | 1549 | | case 3: |
| | 640 | 1550 | | t0 = srcEnd[-3]; |
| | 640 | 1551 | | t1 = srcEnd[-2]; |
| | 640 | 1552 | | t2 = srcEnd[-1]; |
| | 640 | 1553 | | break; |
| | | 1554 | | case 4: |
| | 3193 | 1555 | | t0 = srcEnd[-4]; |
| | 3193 | 1556 | | t1 = srcEnd[-3]; |
| | 3193 | 1557 | | t2 = srcEnd[-2]; |
| | 3193 | 1558 | | t3 = srcEnd[-1]; |
| | 3193 | 1559 | | break; |
| | | 1560 | | default: |
| | 303 | 1561 | | return -1; |
| | | 1562 | | } |
| | | 1563 | | |
| | 4849 | 1564 | | int i0 = Unsafe.Add(ref decodingMap, (IntPtr)t0); |
| | 4849 | 1565 | | int i1 = Unsafe.Add(ref decodingMap, (IntPtr)t1); |
| | | 1566 | | |
| | 4849 | 1567 | | i0 <<= 18; |
| | 4849 | 1568 | | i1 <<= 12; |
| | | 1569 | | |
| | 4849 | 1570 | | i0 |= i1; |
| | 4849 | 1571 | | return i0; |
| | | 1572 | | } |
| | | 1573 | | |
| | | 1574 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1575 | | public int IndexOfAnyExceptWhiteSpace(ReadOnlySpan<byte> span) |
| | | 1576 | | { |
| | 907264 | 1577 | | for (int i = 0; i < span.Length; i++) |
| | | 1578 | | { |
| | 453204 | 1579 | | if (!IsWhiteSpace(span[i])) |
| | | 1580 | | { |
| | 131070 | 1581 | | return i; |
| | | 1582 | | } |
| | | 1583 | | } |
| | | 1584 | | |
| | 428 | 1585 | | return -1; |
| | | 1586 | | } |
| | | 1587 | | |
| | | 1588 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1589 | | public OperationStatus DecodeWithWhiteSpaceBlockwiseWrapper<TBase64Decoder>(TBase64Decoder decoder, ReadOnly |
| | | 1590 | | Span<byte> bytes, ref int bytesConsumed, ref int bytesWritten, bool isFinalBlock = true) |
| | | 1591 | | where TBase64Decoder : IBase64Decoder<byte> => |
| | 0 | 1592 | | DecodeWithWhiteSpaceBlockwise(decoder, utf8, bytes, ref bytesConsumed, ref bytesWritten, isFinalBlock); |
| | | 1593 | | } |
| | | 1594 | | |
| | | 1595 | | internal readonly struct Base64DecoderChar : IBase64Decoder<ushort> |
| | | 1596 | | { |
| | 0 | 1597 | | public ReadOnlySpan<sbyte> DecodingMap => default(Base64DecoderByte).DecodingMap; |
| | | 1598 | | |
| | 0 | 1599 | | public ReadOnlySpan<uint> VbmiLookup0 => default(Base64DecoderByte).VbmiLookup0; |
| | | 1600 | | |
| | 0 | 1601 | | public ReadOnlySpan<uint> VbmiLookup1 => default(Base64DecoderByte).VbmiLookup1; |
| | | 1602 | | |
| | 0 | 1603 | | public ReadOnlySpan<sbyte> Avx2LutHigh => default(Base64DecoderByte).Avx2LutHigh; |
| | | 1604 | | |
| | 0 | 1605 | | public ReadOnlySpan<sbyte> Avx2LutLow => default(Base64DecoderByte).Avx2LutLow; |
| | | 1606 | | |
| | 0 | 1607 | | public ReadOnlySpan<sbyte> Avx2LutShift => default(Base64DecoderByte).Avx2LutShift; |
| | | 1608 | | |
| | 0 | 1609 | | public byte MaskSlashOrUnderscore => default(Base64DecoderByte).MaskSlashOrUnderscore; |
| | | 1610 | | |
| | 0 | 1611 | | public ReadOnlySpan<int> Vector128LutHigh => default(Base64DecoderByte).Vector128LutHigh; |
| | | 1612 | | |
| | 0 | 1613 | | public ReadOnlySpan<int> Vector128LutLow => default(Base64DecoderByte).Vector128LutLow; |
| | | 1614 | | |
| | 0 | 1615 | | public ReadOnlySpan<uint> Vector128LutShift => default(Base64DecoderByte).Vector128LutShift; |
| | | 1616 | | |
| | 0 | 1617 | | public ReadOnlySpan<uint> AdvSimdLutOne3 => default(Base64DecoderByte).AdvSimdLutOne3; |
| | | 1618 | | |
| | 0 | 1619 | | public uint AdvSimdLutTwo3Uint1 => default(Base64DecoderByte).AdvSimdLutTwo3Uint1; |
| | | 1620 | | |
| | 0 | 1621 | | public int GetMaxDecodedLength(int sourceLength) => Base64.GetMaxDecodedFromUtf8Length(sourceLength); |
| | | 1622 | | |
| | 0 | 1623 | | public bool IsInvalidLength(int bufferLength) => bufferLength % 4 != 0; |
| | | 1624 | | |
| | 0 | 1625 | | public bool IsValidPadding(uint padChar) => padChar == EncodingPad; |
| | | 1626 | | |
| | 0 | 1627 | | public int SrcLength(bool _, int sourceLength) => sourceLength & ~0x3; |
| | | 1628 | | |
| | | 1629 | | #if NET |
| | | 1630 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1631 | | [CompExactlyDependsOn(typeof(AdvSimd.Arm64))] |
| | | 1632 | | [CompExactlyDependsOn(typeof(Ssse3))] |
| | | 1633 | | public bool TryDecode128Core(Vector128<byte> str, Vector128<byte> hiNibbles, Vector128<byte> maskSlashOrUnde |
| | | 1634 | | Vector128<byte> lutLow, Vector128<byte> lutHigh, Vector128<sbyte> lutShift, Vector128<byte> shiftForUnde |
| | 0 | 1635 | | default(Base64DecoderByte).TryDecode128Core(str, hiNibbles, maskSlashOrUnderscore, mask8F, lutLow, lutHi |
| | | 1636 | | |
| | | 1637 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1638 | | [CompExactlyDependsOn(typeof(Avx2))] |
| | | 1639 | | public bool TryDecode256Core(Vector256<sbyte> str, Vector256<sbyte> hiNibbles, Vector256<sbyte> maskSlashOrU |
| | | 1640 | | Vector256<sbyte> lutHigh, Vector256<sbyte> lutShift, Vector256<sbyte> shiftForUnderscore, out Vector256< |
| | 0 | 1641 | | default(Base64DecoderByte).TryDecode256Core(str, hiNibbles, maskSlashOrUnderscore, lutLow, lutHigh, lutS |
| | | 1642 | | |
| | | 1643 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1644 | | public unsafe bool TryLoadVector512(ushort* src, ushort* srcStart, int sourceLength, out Vector512<sbyte> st |
| | | 1645 | | { |
| | 0 | 1646 | | AssertRead<Vector512<ushort>>(src, srcStart, sourceLength); |
| | 0 | 1647 | | Vector512<ushort> utf16VectorLower = Vector512.Load(src); |
| | 0 | 1648 | | Vector512<ushort> utf16VectorUpper = Vector512.Load(src + 32); |
| | 0 | 1649 | | if (Ascii.VectorContainsNonAsciiChar(utf16VectorLower | utf16VectorUpper)) |
| | | 1650 | | { |
| | 0 | 1651 | | str = default; |
| | 0 | 1652 | | return false; |
| | | 1653 | | } |
| | | 1654 | | |
| | 0 | 1655 | | str = Ascii.ExtractAsciiVector(utf16VectorLower, utf16VectorUpper).AsSByte(); |
| | 0 | 1656 | | return true; |
| | | 1657 | | } |
| | | 1658 | | |
| | | 1659 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1660 | | [CompExactlyDependsOn(typeof(Avx2))] |
| | | 1661 | | public unsafe bool TryLoadAvxVector256(ushort* src, ushort* srcStart, int sourceLength, out Vector256<sbyte> |
| | | 1662 | | { |
| | 188121 | 1663 | | AssertRead<Vector256<sbyte>>(src, srcStart, sourceLength); |
| | 188121 | 1664 | | Vector256<ushort> utf16VectorLower = Avx.LoadVector256(src); |
| | 188121 | 1665 | | Vector256<ushort> utf16VectorUpper = Avx.LoadVector256(src + 16); |
| | | 1666 | | |
| | 188121 | 1667 | | if (Ascii.VectorContainsNonAsciiChar(utf16VectorLower | utf16VectorUpper)) |
| | | 1668 | | { |
| | 0 | 1669 | | str = default; |
| | 0 | 1670 | | return false; |
| | | 1671 | | } |
| | | 1672 | | |
| | 188121 | 1673 | | str = Ascii.ExtractAsciiVector(utf16VectorLower, utf16VectorUpper).AsSByte(); |
| | 188121 | 1674 | | return true; |
| | | 1675 | | } |
| | | 1676 | | |
| | | 1677 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1678 | | public unsafe bool TryLoadVector128(ushort* src, ushort* srcStart, int sourceLength, out Vector128<byte> str |
| | | 1679 | | { |
| | 5492 | 1680 | | AssertRead<Vector128<sbyte>>(src, srcStart, sourceLength); |
| | 5492 | 1681 | | Vector128<ushort> utf16VectorLower = Vector128.LoadUnsafe(ref *src); |
| | 5492 | 1682 | | Vector128<ushort> utf16VectorUpper = Vector128.LoadUnsafe(ref *src, 8); |
| | 5492 | 1683 | | if (Ascii.VectorContainsNonAsciiChar(utf16VectorLower | utf16VectorUpper)) |
| | | 1684 | | { |
| | 0 | 1685 | | str = default; |
| | 0 | 1686 | | return false; |
| | | 1687 | | } |
| | | 1688 | | |
| | 5492 | 1689 | | str = Ascii.ExtractAsciiVector(utf16VectorLower, utf16VectorUpper); |
| | 5492 | 1690 | | return true; |
| | | 1691 | | } |
| | | 1692 | | |
| | | 1693 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1694 | | [CompExactlyDependsOn(typeof(AdvSimd.Arm64))] |
| | | 1695 | | public unsafe bool TryLoadArmVector128x4(ushort* src, ushort* srcStart, int sourceLength, |
| | | 1696 | | out Vector128<byte> str1, out Vector128<byte> str2, out Vector128<byte> str3, out Vector128<byte> str4) |
| | | 1697 | | { |
| | 0 | 1698 | | AssertRead<Vector128<sbyte>>(src, srcStart, sourceLength); |
| | 0 | 1699 | | var (s11, s12, s21, s22) = AdvSimd.Arm64.Load4xVector128AndUnzip(src); |
| | 0 | 1700 | | var (s31, s32, s41, s42) = AdvSimd.Arm64.Load4xVector128AndUnzip(src + 32); |
| | | 1701 | | |
| | 0 | 1702 | | if (Ascii.VectorContainsNonAsciiChar(s11 | s12 | s21 | s22 | s31 | s32 | s41 | s42)) |
| | | 1703 | | { |
| | 0 | 1704 | | str1 = str2 = str3 = str4 = default; |
| | 0 | 1705 | | return false; |
| | | 1706 | | } |
| | | 1707 | | |
| | 0 | 1708 | | str1 = Ascii.ExtractAsciiVector(s11, s31); |
| | 0 | 1709 | | str2 = Ascii.ExtractAsciiVector(s12, s32); |
| | 0 | 1710 | | str3 = Ascii.ExtractAsciiVector(s21, s41); |
| | 0 | 1711 | | str4 = Ascii.ExtractAsciiVector(s22, s42); |
| | | 1712 | | |
| | 0 | 1713 | | return true; |
| | | 1714 | | } |
| | | 1715 | | #endif // NET |
| | | 1716 | | |
| | | 1717 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1718 | | public unsafe int DecodeFourElements(ushort* source, ref sbyte decodingMap) |
| | | 1719 | | { |
| | | 1720 | | // The 'source' span expected to have at least 4 elements, and the 'decodingMap' consists 256 sbytes |
| | 21063 | 1721 | | uint t0 = source[0]; |
| | 21063 | 1722 | | uint t1 = source[1]; |
| | 21063 | 1723 | | uint t2 = source[2]; |
| | 21063 | 1724 | | uint t3 = source[3]; |
| | | 1725 | | |
| | 21063 | 1726 | | if (((t0 | t1 | t2 | t3) & 0xffffff00) != 0) |
| | | 1727 | | { |
| | 0 | 1728 | | return -1; // One or more chars falls outside the 00..ff range, invalid Base64 character. |
| | | 1729 | | } |
| | | 1730 | | |
| | 21063 | 1731 | | int i0 = Unsafe.Add(ref decodingMap, (int)t0); |
| | 21063 | 1732 | | int i1 = Unsafe.Add(ref decodingMap, (int)t1); |
| | 21063 | 1733 | | int i2 = Unsafe.Add(ref decodingMap, (int)t2); |
| | 21063 | 1734 | | int i3 = Unsafe.Add(ref decodingMap, (int)t3); |
| | | 1735 | | |
| | 21063 | 1736 | | i0 <<= 18; |
| | 21063 | 1737 | | i1 <<= 12; |
| | 21063 | 1738 | | i2 <<= 6; |
| | | 1739 | | |
| | 21063 | 1740 | | i0 |= i3; |
| | 21063 | 1741 | | i1 |= i2; |
| | | 1742 | | |
| | 21063 | 1743 | | i0 |= i1; |
| | 21063 | 1744 | | return i0; |
| | | 1745 | | } |
| | | 1746 | | |
| | | 1747 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1748 | | public unsafe int DecodeRemaining(ushort* srcEnd, ref sbyte decodingMap, long remaining, out uint t2, out ui |
| | | 1749 | | { |
| | | 1750 | | uint t0; |
| | | 1751 | | uint t1; |
| | 3323 | 1752 | | t2 = EncodingPad; |
| | 3323 | 1753 | | t3 = EncodingPad; |
| | | 1754 | | switch (remaining) |
| | | 1755 | | { |
| | | 1756 | | case 2: |
| | 1184 | 1757 | | t0 = srcEnd[-2]; |
| | 1184 | 1758 | | t1 = srcEnd[-1]; |
| | 1184 | 1759 | | break; |
| | | 1760 | | case 3: |
| | 1052 | 1761 | | t0 = srcEnd[-3]; |
| | 1052 | 1762 | | t1 = srcEnd[-2]; |
| | 1052 | 1763 | | t2 = srcEnd[-1]; |
| | 1052 | 1764 | | break; |
| | | 1765 | | case 4: |
| | 1087 | 1766 | | t0 = srcEnd[-4]; |
| | 1087 | 1767 | | t1 = srcEnd[-3]; |
| | 1087 | 1768 | | t2 = srcEnd[-2]; |
| | 1087 | 1769 | | t3 = srcEnd[-1]; |
| | 1087 | 1770 | | break; |
| | | 1771 | | default: |
| | 0 | 1772 | | return -1; |
| | | 1773 | | } |
| | | 1774 | | |
| | 3323 | 1775 | | if (((t0 | t1 | t2 | t3) & 0xffffff00) != 0) |
| | | 1776 | | { |
| | 0 | 1777 | | return -1; |
| | | 1778 | | } |
| | | 1779 | | |
| | 3323 | 1780 | | int i0 = Unsafe.Add(ref decodingMap, (IntPtr)t0); |
| | 3323 | 1781 | | int i1 = Unsafe.Add(ref decodingMap, (IntPtr)t1); |
| | | 1782 | | |
| | 3323 | 1783 | | i0 <<= 18; |
| | 3323 | 1784 | | i1 <<= 12; |
| | | 1785 | | |
| | 3323 | 1786 | | i0 |= i1; |
| | 3323 | 1787 | | return i0; |
| | | 1788 | | } |
| | | 1789 | | |
| | | 1790 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1791 | | public int IndexOfAnyExceptWhiteSpace(ReadOnlySpan<ushort> span) |
| | | 1792 | | { |
| | 0 | 1793 | | for (int i = 0; i < span.Length; i++) |
| | | 1794 | | { |
| | 0 | 1795 | | if (!IsWhiteSpace(span[i])) |
| | | 1796 | | { |
| | 0 | 1797 | | return i; |
| | | 1798 | | } |
| | | 1799 | | } |
| | | 1800 | | |
| | 0 | 1801 | | return -1; |
| | | 1802 | | } |
| | | 1803 | | |
| | | 1804 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 1805 | | public OperationStatus DecodeWithWhiteSpaceBlockwiseWrapper<TBase64Decoder>(TBase64Decoder decoder, ReadOnly |
| | | 1806 | | Span<byte> bytes, ref int bytesConsumed, ref int bytesWritten, bool isFinalBlock = true) where TBase64De |
| | 0 | 1807 | | DecodeWithWhiteSpaceBlockwise(default(Base64DecoderChar), source, bytes, ref bytesConsumed, ref bytesWri |
| | | 1808 | | } |
| | | 1809 | | } |
| | | 1810 | | } |