Skip to content

Allow reference types for pinned GC.AllocateArray() #89293

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 16 commits into from
Aug 11, 2023
Merged

Allow reference types for pinned GC.AllocateArray() #89293

Show file tree
Hide file tree
Changes from 5 commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
d579879
Relaxing constraints of System.GC.AllocateArray<T>.
jthorborg Jul 21, 2023
5fc9157
Adding coverage of GC.AllocateArray with ref types.
jthorborg Jul 21, 2023
376fb89
Mono/NativeAOT relaxing of System.GC.AllocateArray<T>.
jthorborg Jul 21, 2023
808d582
Merge branch 'dotnet:main' into poh-allocate-array-allow-reference-types
jthorborg Jul 21, 2023
ea6de35
Relaxed GC.AllocateUninitializedArray for ref types as well.
jthorborg Jul 21, 2023
c403266
Simplifiy conditionals in AllocateUninitializedArray.
jthorborg Jul 21, 2023
30fd375
Also simplify path for NativeOAT.
jthorborg Jul 21, 2023
13fe7a1
Simplify pinning paths.
jthorborg Jul 21, 2023
3aeca6b
Revert mistake in prior refactor.
jthorborg Jul 21, 2023
2e502dc
Renaming in tests.
jthorborg Jul 21, 2023
d7dfad7
Update src/coreclr/System.Private.CoreLib/src/System/GC.CoreCLR.cs
jkotas Aug 2, 2023
3a1bca6
Apply suggestions from code review
jkotas Aug 2, 2023
4b805e5
Don't use `var` if type name doesn't exist explicitly on right-hand side
jthorborg Aug 10, 2023
f3aff36
PR feedback for using terse method tables and JIT intrinsics for GC a…
jthorborg Aug 10, 2023
349970a
Changed EmbeddedValueType and added comment.
jthorborg Aug 10, 2023
bfe62ec
Suggestions from PR feedback.
jthorborg Aug 11, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 1 addition & 10 deletions src/coreclr/System.Private.CoreLib/src/System/GC.CoreCLR.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -732,9 +732,6 @@ internal static void UnregisterMemoryLoadChangeNotification(Action notification)
/// <typeparam name="T">Specifies the type of the array element.</typeparam>
/// <param name="length">Specifies the length of the array.</param>
/// <param name="pinned">Specifies whether the allocated array must be pinned.</param>
/// <remarks>
/// If pinned is set to true, <typeparamref name="T"/> must not be a reference type or a type that contains object references.
/// </remarks>
[MethodImpl(MethodImplOptions.AggressiveInlining)] // forced to ensure no perf drop for small memory buffers (hot path)
public static unsafe T[] AllocateUninitializedArray<T>(int length, bool pinned = false) // T[] rather than T?[] to match `new T[length]` behavior
{
Expand All @@ -759,7 +756,7 @@ public static unsafe T[] AllocateUninitializedArray<T>(int length, bool pinned =
}
else if (RuntimeHelpers.IsReferenceOrContainsReferences<T>())
{
ThrowHelper.ThrowInvalidTypeWithPointersNotSupported(typeof(T));
return AllocateArray<T>(length, pinned: true);
}

GC_ALLOC_FLAGS flags = GC_ALLOC_FLAGS.GC_ALLOC_ZEROING_OPTIONAL;
Expand All @@ -775,18 +772,12 @@ public static unsafe T[] AllocateUninitializedArray<T>(int length, bool pinned =
/// <typeparam name="T">Specifies the type of the array element.</typeparam>
/// <param name="length">Specifies the length of the array.</param>
/// <param name="pinned">Specifies whether the allocated array must be pinned.</param>
/// <remarks>
/// If pinned is set to true, <typeparamref name="T"/> must not be a reference type or a type that contains object references.
/// </remarks>
public static T[] AllocateArray<T>(int length, bool pinned = false) // T[] rather than T?[] to match `new T[length]` behavior
{
GC_ALLOC_FLAGS flags = GC_ALLOC_FLAGS.GC_ALLOC_NO_FLAGS;

if (pinned)
{
if (RuntimeHelpers.IsReferenceOrContainsReferences<T>())
ThrowHelper.ThrowInvalidTypeWithPointersNotSupported(typeof(T));

flags = GC_ALLOC_FLAGS.GC_ALLOC_PINNED_OBJECT_HEAP;
}

Expand Down
11 changes: 1 addition & 10 deletions src/coreclr/nativeaot/System.Private.CoreLib/src/System/GC.NativeAot.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -795,9 +795,6 @@ internal static ulong GetSegmentSize()
/// <typeparam name="T">Specifies the type of the array element.</typeparam>
/// <param name="length">Specifies the length of the array.</param>
/// <param name="pinned">Specifies whether the allocated array must be pinned.</param>
/// <remarks>
/// If pinned is set to true, <typeparamref name="T"/> must not be a reference type or a type that contains object references.
/// </remarks>
[MethodImpl(MethodImplOptions.AggressiveInlining)] // forced to ensure no perf drop for small memory buffers (hot path)
public static unsafe T[] AllocateUninitializedArray<T>(int length, bool pinned = false)
{
Expand All @@ -821,7 +818,7 @@ public static unsafe T[] AllocateUninitializedArray<T>(int length, bool pinned =
}
else if (RuntimeHelpers.IsReferenceOrContainsReferences<T>())
{
ThrowHelper.ThrowInvalidTypeWithPointersNotSupported(typeof(T));
return AllocateArray<T>(length, pinned: true);
}

// kept outside of the small arrays hot path to have inlining without big size growth
Expand Down Expand Up @@ -851,18 +848,12 @@ static T[] AllocateNewUninitializedArray(int length, bool pinned)
/// <typeparam name="T">Specifies the type of the array element.</typeparam>
/// <param name="length">Specifies the length of the array.</param>
/// <param name="pinned">Specifies whether the allocated array must be pinned.</param>
/// <remarks>
/// If pinned is set to true, <typeparamref name="T"/> must not be a reference type or a type that contains object references.
/// </remarks>
public static unsafe T[] AllocateArray<T>(int length, bool pinned = false)
{
GC_ALLOC_FLAGS flags = GC_ALLOC_FLAGS.GC_ALLOC_NO_FLAGS;

if (pinned)
{
if (RuntimeHelpers.IsReferenceOrContainsReferences<T>())
ThrowHelper.ThrowInvalidTypeWithPointersNotSupported(typeof(T));

flags = GC_ALLOC_FLAGS.GC_ALLOC_PINNED_OBJECT_HEAP;
}

Expand Down
49 changes: 45 additions & 4 deletions src/libraries/System.Runtime/tests/System/GCTests.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1058,11 +1058,52 @@ private static void AllocateArrayTooLarge()
Assert.Throws<OutOfMemoryException>(() => GC.AllocateUninitializedArray<double>(int.MaxValue, pinned: true));
}

[Fact]
private static void AllocateArrayRefType()
struct EmbeddedValueType<T>
{
unsafe fixed byte _[7];
Copy link
Member

@AaronRobinsonMSFT AaronRobinsonMSFT Jul 22, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is there a specific reason for this? It looks like trying to fiddle with offsets and alignment. I tend to avoid these kind of mechanisms in tests without a comment as to the why.

Copy link
Contributor Author

@jthorborg jthorborg Jul 22, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It was a simple way of increasing the size of the struct, making it less trivially laid out and requiring padding bits - to add a bit of confidence in we're not just overwriting the first set of bytes in a memory location.

We could add a comment or just remove it, it's not required.

Copy link
Member

@jkotas jkotas Aug 7, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think a non-generic struct like this would be sufficient for what you are trying to test here

private struct StructType
{
    internal object _;
    internal string _value;
    internal int _;
}

public T Value;
}

[Theory]
[InlineData(false), InlineData(true)]
private static void AllocateArray_UninitializedOrNot_WithManagedType_DoesNotThrow(bool pinned)
{
void TryType<T>()
{
GC.AllocateUninitializedArray<T>(100, pinned);
GC.AllocateArray<T>(100, pinned);

GC.AllocateArray<EmbeddedValueType<T>>(100, pinned);
GC.AllocateUninitializedArray<EmbeddedValueType<T>>(100, pinned);
}

TryType<string>();
TryType<object>();
}

[Theory]
[InlineData(false), InlineData(true)]
private unsafe static void AllocateArrayPinned_ManagedValueType_CanRoundtripThroughPointer(bool uninitialized)
{
GC.AllocateUninitializedArray<string>(100);
Assert.Throws<ArgumentException>(() => GC.AllocateUninitializedArray<string>(100, pinned: true));
const int k_Length = 100;
var rng = new Random(0xAF);

var array = uninitialized ? GC.AllocateUninitializedArray<EmbeddedValueType<string>>(k_Length, pinned: true) : GC.AllocateArray<EmbeddedValueType<string>>(k_Length, pinned: true);
byte* pointer = (byte*)Unsafe.AsPointer(ref array[0]);
var size = Unsafe.SizeOf<EmbeddedValueType<string>>();

for(int i = 0; i < k_Length; ++i)
{
var idx = rng.Next(k_Length);
ref var evt = ref Unsafe.AsRef<EmbeddedValueType<string>>(pointer + size * idx);

var stringValue = rng.NextSingle().ToString();
evt.Value = stringValue;

Assert.Equal(evt.Value, array[idx].Value);
}

GC.KeepAlive(array);
}

[Fact]
Expand Down
2 changes: 0 additions & 2 deletions src/mono/System.Private.CoreLib/src/System/GC.Mono.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -281,8 +281,6 @@ public static T[] AllocateUninitializedArray<T>(int length, bool pinned = false)
public static T[] AllocateArray<T>(int length, bool pinned = false)
{
if (pinned) {
if (RuntimeHelpers.IsReferenceOrContainsReferences<T>())
ThrowHelper.ThrowInvalidTypeWithPointersNotSupported(typeof(T));
return Unsafe.As<T[]>(AllocPinnedArray(typeof(T[]), length));
}

Expand Down