top of page

Memory Efficiency: ref and unsafe Feature in C# 13

Our problem revolves around the limitations imposed on using ref and unsafe constructs in certain contexts. Specifically, async and iterator methods (those employing yield return) had restrictions that hindered their expressive power.

Memory Efficiency: ref and unsafe Feature in C# 13

Before C# 13

In earlier versions of C#, developers faced limitations when using certain language features within async methods and iterator methods:


Iterator Methods (Using yield return): Iterator methods faced two significant limitations:

  • No Local ref Variables: Iterator methods couldn’t declare local ref variables. This restriction prevented developers from using references to data within the method.

  • Unsafe Contexts Not Allowed: Unsafe contexts (which involve pointers and other low-level operations) were disallowed within iterator methods. Consequently, developers couldn’t perform unsafe operations within these methods.


Async Methods: Similarly, async methods encountered similar restrictions:

  • No Local ref Variables: Developers couldn’t declare local ref variables within async methods.

  • Unsafe Contexts Prohibited: Unsafe contexts were also off-limits in async methods.


Motivation for Change:

The issue stemmed from the interaction between await and ref variables. When using await, the compiler couldn’t guarantee that the reference held by a ref variable would remain valid after the await. However, this restriction seemed overly cautious in certain cases (such as using ref variables only between two await calls).



What Changed in C# 13:

Async Methods:

These ref variables can be used within the method, but cannot cross an await boundary. This ensures safety while still providing flexibility.

async Task<int> ComputeAsync(ref int value)
{
    // Some computation
    await Task.Delay(100);
    value++; // Valid usage of ref variable
    return value;
}

Iterator Methods:

However, there’s a caveat: all yield return and yield break statements must still be in safe contexts. This means you can use unsafe constructs within the method body, but the actual yielding of values must be safe.

unsafe IEnumerable<int> GenerateNumbers()
{
    int* ptr = stackalloc int[10];
    for (int i = 0; i < 10; i++)
    {
        ptr[i] = i * i;
        yield return ptr[i]; // Safe context for yield return
    }
}

Developers can now safely use ref local variables and ref struct types (such as Span<T> or ReadOnlySpan<T>) in more places. The compiler enforces safety rules, ensuring that these constructs are used correctly. This enhancement allows for more expressive code in async and iterator methods while maintaining safety.



Ref Local Variables in Async Methods

Ref local variables allow you to create references to existing variables. Unlike regular local variables, which store a copy of the value, ref local variables directly reference the original data. This can be particularly useful when working with large data structures or to modify the original value within a method.


For example, consider the following code:

void ModifyValue(ref int value)
{
    value *= 2;
}

void Main()
{
    int originalValue = 10;
    ref int reference = ref originalValue;

    ModifyValue(ref reference);

    Console.WriteLine(originalValue); // Outputs 20
}

In this example, reference is a ref local variable that points to the original originalValue. When we modify reference, the original value is updated as well.


Older Restriction in Async Methods

Earlier, async methods couldn’t declare ref local variables. This limitation affected scenarios where you needed to work with references within asynchronous code. For instance, if you wanted to manipulate a large buffer or perform other operations that required direct memory access, you were out of luck within an async method.


New

C# 13 brings a welcome change: async methods can now declare ref local variables. This enhancement allows you to use references within async methods while maintaining safety.


For example:

async Task ModifyAsync(ref int value)
{
    await Task.Delay(100);
    value *= 2;
}

async Task MainAsync()
{
    int originalValue = 10;
    await ModifyAsync(ref originalValue);

    Console.WriteLine(originalValue); // Outputs 20
}

In this updated code, ModifyAsync is an async method that accepts a ref local variable (value). It awaits a delay (simulating some asynchronous work) and then modifies the value. The change is reflected in the original variable outside the method.


Note:

Remember that ref local variables in async methods cannot cross an await boundary. In other words, you can use them within the method, but any modifications won’t persist across asynchronous operations. This ensures safety and prevents potential issues related to the lifetime of references.


Ref Struct Types in Async Methods:

Ref struct types, such as Span<T> or ReadOnlySpan<T>, provide a way to work with memory slices more efficiently and expressively. These types allow you to create views over existing data without copying it. They are used to deal with large buffers, memory-mapped files, or other scenarios where minimizing memory allocations and copying is crucial.

  • Span<T>: Represents a contiguous region of memory that can be used to read or modify data. It provides a view over an array, stack-allocated memory, or other memory sources.

  • ReadOnlySpan<T>: Similar to Span<T>, but read-only. It is useful when you don’t need to modify the data.


C# 13 introduces a significant improvement: async methods can now use local variables of ref struct types. This means you can create and manipulate Span<T> or ReadOnlySpan<T> instances within async methods, just like you would with regular local variables.


For example:

async Task ProcessDataAsync(byte[] data)
{
    // Create a read-only span over the data
    ReadOnlySpan<byte> dataSpan = data;

    // Perform some asynchronous operations
    await Task.Delay(100);

    // Access and process the data within the span
    int sum = 0;
    foreach (byte value in dataSpan)
    {
        sum += value;
    }

    Console.WriteLine($"Sum of data: {sum}");
}

In this code, dataSpan is a local variable of type ReadOnlySpan<byte>. We create a view over the data array and perform some asynchronous work. The span remains valid within the method, allowing efficient data processing.


Ref struct types and regular local variables have distinct characteristics, especially concerning memory usage.

Aspect

Regular Local Variables

Ref Struct Types (Span/ReadOnlySpan)

Memory Allocation

Allocated on the stack.

Stack-allocated or point to existing memory.


Fast and deterministic (automatic reclamation).

Provide views over data without copying.


Limited stack space; large vars can cause overflow.

Efficient access to memory regions (e.g., buffers).

Copying and Views

Copies data when assigned to a local variable.

Avoids unnecessary copying; views into original data.


Expensive for large data structures.

Ideal for working directly with memory.

Mutability

Mutable or immutable based on type.

Mutable or read-only (depending on ref struct type).


Immutable types prevent modification after init.

Span allows both read and write; ReadOnlySpan is read-only.

Lifetime and Safety

Well-defined lifetime within scope.

Requires careful handling (e.g., pointers, safety).


Generally safe; no manual memory management.

Ref struct vars in async methods can’t cross await boundary.

Use Cases

Simple data storage, loop counters, temp values.

Efficient data processing, memory views, no copies.


Suitable where copying is acceptable.

Ideal for large buffers, parsing, memory-mapped files.

Similar to ref local variables, ref struct variables in async methods cannot cross an await boundary. You can use them within the method, but any modifications won’t persist across asynchronous operations. This restriction ensures safety and prevents potential issues related to the lifetime of references.


Ref struct types generally outperform heap-allocated objects in terms of memory usage and execution speed due to their stack allocation and direct memory access. However, their usage requires careful attention to lifetime and safety.

Aspect

Ref Struct Types (Stack-Allocated)

Heap-Allocated Objects (Classes)

Memory Allocation

Allocated on the stack or inline within containing types (based on compiler optimizations).

Managed by the garbage collector and reside on the heap.


Fast and deterministic (automatic reclamation).

Involves more overhead due to memory management.


Ideal for small data structures and temporary values.

Suitable for larger, long-lived objects.

Copying and Views

Avoids unnecessary copying; provides views into existing data (e.g., arrays).

Involves copying references (not the actual data).


Efficient for working directly with memory regions.

Passing heap-allocated objects means passing references.

Mutability

Can be mutable or read-only (e.g., Span<T>).

Mutable or immutable based on their design.


Span<T> allows both read and write operations.

Immutable classes (e.g., string, readonly struct).

Lifetime and Safety

Well-defined lifetime within their scope.

Managed by the garbage collector (long-lived).


Generally safe; no manual memory management.

Requires more complex memory management.

Use Cases

Efficient data processing, memory views, no copies.

Complex data structures, long-lived objects, shared state.


Ideal for large buffers, parsing, memory-mapped files.

Suitable for object-oriented designs and reference semantics.

While ref struct types can be used in async methods, they cannot cross await boundaries. This restriction ensures safety and prevents potential issues related to reference lifetimes.


Unsafe Contexts in Iterator Methods

Iterator methods are a powerful feature in C# that allows you to create custom iterators for sequences of data. These methods use the yield return statement to produce a sequence of values lazily. Essentially, they allow you to generate values on-the-fly without loading the entire sequence into memory.


For example, consider a simple iterator method that generates Fibonacci numbers:

public static IEnumerable<int> FibonacciSequence()
{
    int a = 0, b = 1;
    while (true)
    {
        yield return a;
        (a, b) = (b, a + b);
    }
}

In this example, calling FibonacciSequence() will give you an infinite sequence of Fibonacci numbers.


Now, you can use pointers, stack-allocated memory, and other unsafe features within the method body. For example:

public static unsafe IEnumerable<int> GenerateRandomNumbers()
{
    int* buffer = stackalloc int[10];
    for (int i = 0; i < 10; i++)
    {
        buffer[i] = i * i;
        yield return buffer[i]; // Safe context for yield return
    }
}

In this code, we create an iterator method called GenerateRandomNumbers(). It uses stackalloc to allocate memory on the stack for an array of integers. The method then yields each value from the buffer. Note that the yield return statement remains in a safe context.


Performance Impact Unsafe Contexts in Iterator Methods:

  • Pointer Arithmetic Efficiency: Unsafe contexts allow you to use pointers directly, bypassing some of the safety checks imposed by the C# runtime. When working with large data structures (such as arrays), pointer arithmetic can be significantly faster than using the usual array indexing (e.g., arr[i]).

  • Reduced Bounds Checking: In iterator methods, using pointers allows you to perform bounds checking once (at the beginning of the method) rather than on each access. This can improve performance when iterating over large collections.

  • Optimized Memory Access: Unsafe code lets you work directly with memory addresses, which can lead to more efficient memory access patterns. For example, you can read or modify memory in chunks rather than element-by-element.


Limitations and Considerations of Unsafe Contexts in Iterator Methods

  • Safety Trade-Off: Unsafe contexts come with risks. You’re responsible for ensuring memory safety, avoiding null references, and preventing buffer overflows. Incorrect pointer manipulation can lead to crashes or security vulnerabilities.

  • Compiler Constraints: The C# compiler imposes restrictions on unsafe code within iterator methods. For example, you cannot use unsafe constructs directly within a yield return or yield break statement.

  • Verifiability and Security: Iterator blocks must generate verifiable code, even if they’re nested within an unsafe context. This ensures that the generated code doesn’t violate safety rules.

  • JIT Optimization: The .NET Just-In-Time (JIT) compiler may optimize certain bounds checks away, even in safe code. It’s essential to profile and measure performance to determine if unsafe contexts provide significant benefits.


While unsafe contexts can enhance performance in iterator methods, they require careful handling and thorough testing. Developers should weigh the trade-offs between performance gains and safety considerations based on their specific use cases.


Conclusion

In C# 13, exciting enhancements have been introduced to improve the flexibility and expressiveness of code. First, async methods can now declare ref local variables, allowing efficient memory access and direct references within asynchronous code. For example, you can modify values asynchronously while maintaining safety. Second, unsafe contexts (involving pointers and low-level memory operations) are now allowed in both async methods and iterator methods. This change empowers developers to work with pointers, stack-allocated memory, and other unsafe features within these constructs. However, it’s essential to ensure that yield return and yield break statements remain in safe contexts.


Happy Coding!😊

Comments


bottom of page