-
Notifications
You must be signed in to change notification settings - Fork 4.8k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Update DllImportGenerator to the latest version (#59117)
- Loading branch information
1 parent
5f0f7a1
commit 5b8ce63
Showing
30 changed files
with
274 additions
and
17 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
225 changes: 225 additions & 0 deletions
225
src/libraries/Common/src/System/Runtime/InteropServices/ArrayMarshaller.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,225 @@ | ||
| // Licensed to the .NET Foundation under one or more agreements. | ||
| // The .NET Foundation licenses this file to you under the MIT license. | ||
|
|
||
| // | ||
| // Types in this file are used for generated p/invokes (docs/design/features/source-generator-pinvokes.md). | ||
| // See the DllImportGenerator experiment in https://github.com/dotnet/runtimelab. | ||
| // | ||
| #if DLLIMPORTGENERATOR_INTERNALUNSAFE | ||
| using Internal.Runtime.CompilerServices; | ||
| #else | ||
| using System.Runtime.CompilerServices; | ||
| #endif | ||
| using System.Diagnostics; | ||
|
|
||
| namespace System.Runtime.InteropServices.GeneratedMarshalling | ||
| { | ||
| internal unsafe ref struct ArrayMarshaller<T> | ||
| { | ||
| private T[]? _managedArray; | ||
| private readonly int _sizeOfNativeElement; | ||
| private IntPtr _allocatedMemory; | ||
|
|
||
| public ArrayMarshaller(int sizeOfNativeElement) | ||
| :this() | ||
| { | ||
| _sizeOfNativeElement = sizeOfNativeElement; | ||
| } | ||
|
|
||
| public ArrayMarshaller(T[]? managed, int sizeOfNativeElement) | ||
| { | ||
| _allocatedMemory = default; | ||
| _sizeOfNativeElement = sizeOfNativeElement; | ||
| if (managed is null) | ||
| { | ||
| _managedArray = null; | ||
| NativeValueStorage = default; | ||
| return; | ||
| } | ||
| _managedArray = managed; | ||
| // Always allocate at least one byte when the array is zero-length. | ||
| int spaceToAllocate = Math.Max(managed.Length * _sizeOfNativeElement, 1); | ||
| _allocatedMemory = Marshal.AllocCoTaskMem(spaceToAllocate); | ||
| NativeValueStorage = new Span<byte>((void*)_allocatedMemory, spaceToAllocate); | ||
| } | ||
|
|
||
| public ArrayMarshaller(T[]? managed, Span<byte> stackSpace, int sizeOfNativeElement) | ||
| { | ||
| _allocatedMemory = default; | ||
| _sizeOfNativeElement = sizeOfNativeElement; | ||
| if (managed is null) | ||
| { | ||
| _managedArray = null; | ||
| NativeValueStorage = default; | ||
| return; | ||
| } | ||
| _managedArray = managed; | ||
| // Always allocate at least one byte when the array is zero-length. | ||
| int spaceToAllocate = Math.Max(managed.Length * _sizeOfNativeElement, 1); | ||
| if (spaceToAllocate <= stackSpace.Length) | ||
| { | ||
| NativeValueStorage = stackSpace[0..spaceToAllocate]; | ||
| } | ||
| else | ||
| { | ||
| _allocatedMemory = Marshal.AllocCoTaskMem(spaceToAllocate); | ||
| NativeValueStorage = new Span<byte>((void*)_allocatedMemory, spaceToAllocate); | ||
| } | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// Stack-alloc threshold set to 256 bytes to enable small arrays to be passed on the stack. | ||
| /// Number kept small to ensure that P/Invokes with a lot of array parameters doesn't | ||
| /// blow the stack since this is a new optimization in the code-generated interop. | ||
| /// </summary> | ||
| public const int StackBufferSize = 0x200; | ||
|
|
||
| public Span<T> ManagedValues => _managedArray; | ||
|
|
||
| public Span<byte> NativeValueStorage { get; private set; } | ||
|
|
||
| public ref byte GetPinnableReference() => ref MemoryMarshal.GetReference(NativeValueStorage); | ||
|
|
||
| public void SetUnmarshalledCollectionLength(int length) | ||
| { | ||
| _managedArray = new T[length]; | ||
| } | ||
|
|
||
| public byte* Value | ||
| { | ||
| get | ||
| { | ||
| Debug.Assert(_managedArray is null || _allocatedMemory != IntPtr.Zero); | ||
| return (byte*)_allocatedMemory; | ||
| } | ||
| set | ||
| { | ||
| if (value == null) | ||
| { | ||
| _managedArray = null; | ||
| NativeValueStorage = default; | ||
| } | ||
| else | ||
| { | ||
| _allocatedMemory = (IntPtr)value; | ||
| NativeValueStorage = new Span<byte>(value, (_managedArray?.Length ?? 0) * _sizeOfNativeElement); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| public T[]? ToManaged() => _managedArray; | ||
|
|
||
| public void FreeNative() | ||
| { | ||
| if (_allocatedMemory != IntPtr.Zero) | ||
| { | ||
| Marshal.FreeCoTaskMem(_allocatedMemory); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| internal unsafe ref struct PtrArrayMarshaller<T> where T : unmanaged | ||
| { | ||
| private T*[]? _managedArray; | ||
| private readonly int _sizeOfNativeElement; | ||
| private IntPtr _allocatedMemory; | ||
|
|
||
| public PtrArrayMarshaller(int sizeOfNativeElement) | ||
| : this() | ||
| { | ||
| _sizeOfNativeElement = sizeOfNativeElement; | ||
| } | ||
|
|
||
| public PtrArrayMarshaller(T*[]? managed, int sizeOfNativeElement) | ||
| { | ||
| _allocatedMemory = default; | ||
| _sizeOfNativeElement = sizeOfNativeElement; | ||
| if (managed is null) | ||
| { | ||
| _managedArray = null; | ||
| NativeValueStorage = default; | ||
| return; | ||
| } | ||
| _managedArray = managed; | ||
| // Always allocate at least one byte when the array is zero-length. | ||
| int spaceToAllocate = Math.Max(managed.Length * _sizeOfNativeElement, 1); | ||
| _allocatedMemory = Marshal.AllocCoTaskMem(spaceToAllocate); | ||
| NativeValueStorage = new Span<byte>((void*)_allocatedMemory, spaceToAllocate); | ||
| } | ||
|
|
||
| public PtrArrayMarshaller(T*[]? managed, Span<byte> stackSpace, int sizeOfNativeElement) | ||
| { | ||
| _allocatedMemory = default; | ||
| _sizeOfNativeElement = sizeOfNativeElement; | ||
| if (managed is null) | ||
| { | ||
| _managedArray = null; | ||
| NativeValueStorage = default; | ||
| return; | ||
| } | ||
| _managedArray = managed; | ||
| // Always allocate at least one byte when the array is zero-length. | ||
| int spaceToAllocate = Math.Max(managed.Length * _sizeOfNativeElement, 1); | ||
| if (spaceToAllocate <= stackSpace.Length) | ||
| { | ||
| NativeValueStorage = stackSpace[0..spaceToAllocate]; | ||
| } | ||
| else | ||
| { | ||
| _allocatedMemory = Marshal.AllocCoTaskMem(spaceToAllocate); | ||
| NativeValueStorage = new Span<byte>((void*)_allocatedMemory, spaceToAllocate); | ||
| } | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// Stack-alloc threshold set to 256 bytes to enable small arrays to be passed on the stack. | ||
| /// Number kept small to ensure that P/Invokes with a lot of array parameters doesn't | ||
| /// blow the stack since this is a new optimization in the code-generated interop. | ||
| /// </summary> | ||
| public const int StackBufferSize = 0x200; | ||
|
|
||
| public Span<IntPtr> ManagedValues => Unsafe.As<IntPtr[]>(_managedArray); | ||
|
|
||
| public Span<byte> NativeValueStorage { get; private set; } | ||
|
|
||
| public ref byte GetPinnableReference() => ref MemoryMarshal.GetReference(NativeValueStorage); | ||
|
|
||
| public void SetUnmarshalledCollectionLength(int length) | ||
| { | ||
| _managedArray = new T*[length]; | ||
| } | ||
|
|
||
| public byte* Value | ||
| { | ||
| get | ||
| { | ||
| Debug.Assert(_managedArray is null || _allocatedMemory != IntPtr.Zero); | ||
| return (byte*)_allocatedMemory; | ||
| } | ||
| set | ||
| { | ||
| if (value == null) | ||
| { | ||
| _managedArray = null; | ||
| NativeValueStorage = default; | ||
| } | ||
| else | ||
| { | ||
| _allocatedMemory = (IntPtr)value; | ||
| NativeValueStorage = new Span<byte>(value, (_managedArray?.Length ?? 0) * _sizeOfNativeElement); | ||
| } | ||
|
|
||
| } | ||
| } | ||
|
|
||
| public T*[]? ToManaged() => _managedArray; | ||
|
|
||
| public void FreeNative() | ||
| { | ||
| if (_allocatedMemory != IntPtr.Zero) | ||
| { | ||
| Marshal.FreeCoTaskMem(_allocatedMemory); | ||
| } | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.