-
Notifications
You must be signed in to change notification settings - Fork 774
Expand file tree
/
Copy pathStrPtr.cs
More file actions
74 lines (62 loc) · 2.17 KB
/
StrPtr.cs
File metadata and controls
74 lines (62 loc) · 2.17 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
using System;
using System.Runtime.InteropServices;
using System.Text;
namespace Python.Runtime.Native
{
[StructLayout(LayoutKind.Sequential)]
struct StrPtr : IDisposable
{
public IntPtr RawPointer { get; set; }
unsafe byte* Bytes => (byte*)this.RawPointer;
public unsafe StrPtr(string value) : this(value, Encodings.UTF8) {}
public unsafe StrPtr(string value, Encoding encoding)
{
if (value is null) throw new ArgumentNullException(nameof(value));
if (encoding is null) throw new ArgumentNullException(nameof(encoding));
var bytes = encoding.GetBytes(value);
this.RawPointer = Marshal.AllocHGlobal(checked(bytes.Length + 1));
try
{
Marshal.Copy(bytes, 0, this.RawPointer, bytes.Length);
this.Bytes[bytes.Length] = 0;
}
catch
{
this.Dispose();
throw;
}
}
public unsafe string? ToString(Encoding encoding)
{
if (encoding is null) throw new ArgumentNullException(nameof(encoding));
if (this.RawPointer == IntPtr.Zero) return null;
return encoding.GetString((byte*)this.RawPointer, byteCount: checked((int)this.ByteCount));
}
public unsafe nuint ByteCount
{
get
{
if (this.RawPointer == IntPtr.Zero) throw new NullReferenceException();
nuint zeroIndex = 0;
while (this.Bytes[zeroIndex] != 0)
{
zeroIndex++;
}
return zeroIndex;
}
}
public void Dispose()
{
if (this.RawPointer == IntPtr.Zero)
return;
Marshal.FreeHGlobal(this.RawPointer);
this.RawPointer = IntPtr.Zero;
}
internal static Encoding GetEncodingByPythonName(string pyEncodingName)
{
// https://stackoverflow.com/a/7798749/231238
if (pyEncodingName == "mbcs") return Encoding.Default;
return Encoding.GetEncoding(pyEncodingName);
}
}
}