ShuriZma
/
BizHawk
mirror of https://github.com/TASEmulators/BizHawk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Optimise `CWDHacks.Get`

This commit is contained in:
YoshiRulz 2022-02-08 16:13:14 +10:00
parent da567272f3
commit d997ad9871
No known key found for this signature in database
GPG Key ID: C4DE31C245353FB7
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/BizHawk.Common/Win32/CWDHacks.cs
View File

@ -1,11 +1,16 @@
using System;
using System.Runtime.InteropServices;
using System.Text;
namespace BizHawk.Common
{
/// <summary>Gets/Sets the current working directory while bypassing the security checks triggered by the public API (<see cref="Environment.CurrentDirectory"/>).</summary>
public static unsafe class CWDHacks
{
private const uint BUFFER_LEN = 0x200U;
private static readonly byte[] BUFFER = new byte[BUFFER_LEN];
[DllImport("kernel32.dll", SetLastError = true)]
private static extern uint GetCurrentDirectoryW(uint nBufferLength, byte* pBuffer);
@ -14,15 +19,20 @@ namespace BizHawk.Common
public static bool Set(string newCWD)
{
fixed (byte* pstr = &System.Text.Encoding.Unicode.GetBytes($"{newCWD}\0")[0])
fixed (byte* pstr = &Encoding.Unicode.GetBytes($"{newCWD}\0")[0])
return SetCurrentDirectoryW(pstr);
}
public static string Get()
{
var buf = new byte[32768];
fixed (byte* pBuf = &buf[0])
return System.Text.Encoding.Unicode.GetString(buf, 0, 2 * (int) GetCurrentDirectoryW(32767, pBuf));
uint result;
fixed (byte* pBuf = &BUFFER[0]) result = GetCurrentDirectoryW(BUFFER_LEN, pBuf);
if (result <= BUFFER_LEN && result is not 0U) return Encoding.Unicode.GetString(BUFFER, 0, (int) (2U * result));
var buf = new byte[result];
uint result1;
fixed (byte* pBuf = &buf[0]) result1 = GetCurrentDirectoryW(BUFFER_LEN, pBuf);
if (result1 == result) return Encoding.Unicode.GetString(buf, 0, (int) (2U * result));
throw new Exception();
}
}
}