oliverbooth/VpSharp
1
0
Fork 0
mirror of https://github.com/oliverbooth/VpSharp synced 2024-11-23 00:28:48 +00:00
Code Issues Wiki

stackalloc sufficient length for byte span

Browse Source
This commit is contained in:
Oliver Booth 2022-12-04 13:57:29 +00:00
parent 0c0866f54e
commit 20e5eebf47
No known key found for this signature in database
GPG Key ID: 32A00B35503AF634
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
VpSharp/src/Internal/Utf8StringToNative.cs
View File

@ -5,6 +5,7 @@ namespace VpSharp.Internal;
internal sealed class Utf8StringToNative : ICustomMarshaler internal sealed class Utf8StringToNative : ICustomMarshaler
{ {
private static readonly Encoding Utf8Encoding = Encoding.UTF8;
private static Utf8StringToNative? s_instance; private static Utf8StringToNative? s_instance;
public static ICustomMarshaler GetInstance(string cookie) public static ICustomMarshaler GetInstance(string cookie)
@ -29,14 +30,15 @@ internal sealed class Utf8StringToNative : ICustomMarshaler
public unsafe nint MarshalManagedToNative(object managedObj) public unsafe nint MarshalManagedToNative(object managedObj)
{ {
var managedString = (string)managedObj; var managedString = (string)managedObj;
Span<byte> utf8Bytes = stackalloc byte[managedString.Length]; int byteCount = Utf8Encoding.GetByteCount(managedString);
Encoding.UTF8.GetBytes(managedString, utf8Bytes); Span<byte> utf8Bytes = stackalloc byte[byteCount];
Utf8Encoding.GetBytes(managedString, utf8Bytes);
fixed (byte* data = &MemoryMarshal.GetReference(utf8Bytes)) fixed (byte* data = &MemoryMarshal.GetReference(utf8Bytes))
{ {
nint buffer = Marshal.AllocHGlobal(utf8Bytes.Length + 1); nint buffer = Marshal.AllocHGlobal(byteCount + 1);
Buffer.MemoryCopy(data, (void*)buffer, utf8Bytes.Length, utf8Bytes.Length); Buffer.MemoryCopy(data, (void*)buffer, byteCount, byteCount);
Marshal.WriteByte(buffer, utf8Bytes.Length, 0); Marshal.WriteByte(buffer, byteCount, 0);
return buffer; return buffer;
} }
} }