(feat) add explicit disconnect on gui close

Author: leonjza

pipetap-dll/inject/control_server.cpp

@@ -623,6 +623,10 @@ namespace pipetap::inject {
                 }
                 break;
 
+            case PT_CMD_CLIENT_DISCONNECT:
+                NamedPipeClient::instance().close_all(/*reason=*/0u, /*win32err=*/0u);
+                break;
+
             default:
                 // Ignore unknown control messages
                 break;

pipetap-gui/app/app.cpp

@@ -25,6 +25,13 @@ namespace pipetap {
     using pipetap::win::IntegrityRidToCStr;
     using pipetap::win::RelaunchAsAdministrator;
 
+    namespace {
+        ui::replay::Manager& ReplayManager() { static ui::replay::Manager replay; return replay; }
+        ui::proxy::Manager& ProxyManager() { static ui::proxy::Manager proxy; return proxy; }
+        injection::VM& InjectorVm() { static injection::VM inj; return inj; }
+        ui::pipelist::VM& PipesVm() { static ui::pipelist::VM pipelist; return pipelist; }
+    }
+
     static void MenuBarRightText(const char* txt, const ImVec4* color = nullptr) {
         float w = ImGui::CalcTextSize(txt).x;
         float full = ImGui::GetWindowContentRegionMax().x;
@@ -44,10 +51,10 @@ namespace pipetap {
     }
 
     void UI() {
-        static ui::replay::Manager   replay;
-        static ui::proxy::Manager    proxy;
-        static injection::VM         inj;
-        static ui::pipelist::VM      pipelist;
+        auto& replay = ReplayManager();
+        auto& proxy = ProxyManager();
+        auto& inj = InjectorVm();
+        auto& pipelist = PipesVm();
 
         // visibility for each panel
         static bool show_replay = true;
@@ -222,4 +229,13 @@ namespace pipetap {
         ImGui::End();
         ImGui::PopStyleVar();
     }
+
+    void UIShutdown() {
+        auto& proxy = ProxyManager();
+        for (auto& tab : proxy.tabs) {
+            if (tab && tab->client) {
+                tab->client->Stop();
+            }
+        }
+    }
 } // namespace pipetap

pipetap-gui/app/ctrlclient.cpp

@@ -38,6 +38,12 @@ namespace pipetap::ctrlclient {
         // prevent late UI calls
         { std::lock_guard<std::mutex> lk(cb_mx_); cb_ = nullptr; }
 
+        if (connected.load(std::memory_order_acquire)) {
+            PT_ClientDisconnect bye{};
+            bye.reason = 0;
+            (void)SendControlMessage(PT_CMD_CLIENT_DISCONNECT, &bye, sizeof(bye), nullptr, 0);
+        }
+
         DisconnectHandles(true);
 
         if (reader_.joinable()) {

pipetap-gui/include/app.h

@@ -3,4 +3,5 @@
 namespace pipetap
 {
     void UI();
-}
+    void UIShutdown();
+}

pipetap-gui/main.cpp

@@ -194,6 +194,9 @@ int APIENTRY wWinMain(_In_ HINSTANCE hInstance, _In_opt_ HINSTANCE, _In_ LPWSTR,
         g_SwapChainOccluded = (hr == DXGI_STATUS_OCCLUDED);
     }
 
+    // Gracefully drop any control connections before tearing down ImGui/DirectX
+    pipetap::UIShutdown();
+
     // Cleanup
     ImGui_ImplDX11_Shutdown();
     ImGui_ImplWin32_Shutdown();

pipetap-shared/include/pipetap/controlpipe.h

@@ -71,6 +71,7 @@ enum : uint16_t {
     PT_CMD_PROXY_OPEN = 0x1201, // VALUE: PT_ProxyOpen + name bytes
     PT_CMD_PROXY_SEND = 0x1202, // VALUE: PT_ProxySend + data bytes
     PT_CMD_PROXY_CLOSE = 0x1203, // VALUE: PT_ProxyClose
+    PT_CMD_CLIENT_DISCONNECT = 0x12F0, // VALUE: PT_ClientDisconnect (GUI closing)
 };
 
 // PT_HELLO
@@ -162,6 +163,11 @@ struct PT_ProxyClose {
     uint64_t session_id;
 };
 
+// GUI notifying DLL that the control client is intentionally dropping
+struct PT_ClientDisconnect {
+    uint32_t reason; // 0 = GUI exit
+};
+
 // Closed event (DLL -> GUI)
 struct PT_ProxyClosed {
     uint64_t session_id;