refactor: refactoring StdoutRedirector

LiteLDev · Aug 11, 2024 · eac1156 · eac1156
1 parent 72320b2
commit eac1156
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src-client/ll/core/main.cpp b/src-client/ll/core/main.cpp
@@ -1 +1,7 @@
+#include "windows.h"
 
+BOOL APIENTRY DllMain(HMODULE hModule, DWORD ulReasonForCall, LPVOID lpReserved) {
+    if (ulReasonForCall != DLL_PROCESS_ATTACH) return TRUE;
+
+    return TRUE;
+}
diff --git a/src/ll/api/io/StdoutRedirector.cpp b/src/ll/api/io/StdoutRedirector.cpp
@@ -70,10 +70,9 @@ std::string StdoutRedirector::read() {
 
         fSuccess = ReadFile(hRead, buf, sizeof(buf), &cbRead, NULL);
 
-        result.append(buf, cbRead);
+        if (cbRead > 0) result.append(buf, cbRead);
 
-        if (!fSuccess && GetLastError() != ERROR_MORE_DATA) break;
-    } while (!fSuccess); // repeat loop if ERROR_MORE_DATA
+    } while (GetLastError() == ERROR_MORE_DATA); // repeat loop if ERROR_MORE_DATA
     return result;
 }
 } // namespace ll::io