Made hRng.c finally use audio entropy with the fallback being the old timing implementation

QoL: Made distributionHelper.ps1 go back to tool dir
Also noted __init__.py to 0.3.0-beta

Author: DefinetlyNotAI

pyCTools/__init__.py

@@ -6,4 +6,4 @@
 if os.name != 'nt':
     raise OSError('This package only supports Windows OS.')
 
-VERSION = "0.2.1-beta"
+VERSION = "0.3.0-beta"

pyCTools/hwrng.py

@@ -142,3 +142,20 @@ def maxrng_threadsafe(self, size: int) -> bytes:
         if not success:
             raise RuntimeError("maxrng_threadsafe failed")
         return bytes(buf)
+
+    def setup_threads(self):
+        """
+        Initialize the RNG for thread-safe operations.
+
+        This method should be called before using `maxrng_threadsafe()`.
+        It prepares the RNG for multithreaded access.
+
+        Raises:
+            RuntimeError: If the initialization fails.
+        """
+        self.dll.maxrng_init()
+        if not self.test_threading_available():
+            raise RuntimeError(
+                "Failed to initialize RNG for threading. "
+                "Ensure that the hRng DLL supports thread-safe operations."
+            )

src/hRng.c

@@ -84,14 +84,44 @@ static void collect_disk_entropy(const BCRYPT_HASH_HANDLE hHash)
 // Audio entropy fallback (simple timing fallback)
 static void collect_audio_entropy(const BCRYPT_HASH_HANDLE hHash)
 {
-    // Simplified: no real audio capture to keep minimal
-    // Just hash QueryPerformanceCounter several times with Sleep
-    for (int i = 0; i < 5; i++)
-    {
-        LARGE_INTEGER counter;
-        QueryPerformanceCounter(&counter);
-        BCryptHashData(hHash, (PUCHAR)&counter, sizeof(counter), 0);
-        Sleep(10);
+    HWAVEIN hWaveIn = NULL;
+    WAVEFORMATEX wfx = {0};
+    wfx.wFormatTag = WAVE_FORMAT_PCM;
+    wfx.nChannels = 1;
+    wfx.nSamplesPerSec = 8000;
+    wfx.wBitsPerSample = 8;
+    wfx.nBlockAlign = 1;
+    wfx.nAvgBytesPerSec = 8000;
+    wfx.cbSize = 0;
+
+    const MMRESULT res = waveInOpen(&hWaveIn, WAVE_MAPPER, &wfx, 0, 0, CALLBACK_NULL);
+    if (res == MMSYSERR_NOERROR && hWaveIn) {
+        WAVEHDR hdr = {0};
+        BYTE buffer[256] = {0};
+        hdr.lpData = (LPSTR)buffer;
+        hdr.dwBufferLength = sizeof(buffer);
+        hdr.dwFlags = 0;
+
+        if (waveInPrepareHeader(hWaveIn, &hdr, sizeof(hdr)) == MMSYSERR_NOERROR) {
+            if (waveInAddBuffer(hWaveIn, &hdr, sizeof(hdr)) == MMSYSERR_NOERROR) {
+                if (waveInStart(hWaveIn) == MMSYSERR_NOERROR) {
+                    Sleep(50); // Let it capture some audio
+                    waveInStop(hWaveIn);
+                    BCryptHashData(hHash, buffer, sizeof(buffer), 0);
+                }
+            }
+            waveInUnprepareHeader(hWaveIn, &hdr, sizeof(hdr));
+        }
+        waveInClose(hWaveIn);
+    } else {
+        // Fallback: Just hash QueryPerformanceCounter several times with Sleep
+        for (int i = 0; i < 5; i++)
+        {
+            LARGE_INTEGER counter;
+            QueryPerformanceCounter(&counter);
+            BCryptHashData(hHash, (PUCHAR)&counter, sizeof(counter), 0);
+            Sleep(10);
+        }
     }
 }
 

tool/distributionHelper.ps1

@@ -140,4 +140,6 @@ Hash      : $($hashObj.Hash)
 } catch {
     Write-CustomError "Unexpected error occurred." $_.Exception.Message
     exit 1
+} finally {
+    Set-Location tool
 }