From cff00a871188e16298631f0100efade743c645c4 Mon Sep 17 00:00:00 2001
From: Fangrui Song <i@maskray.me>
Date: Mon, 4 Mar 2019 06:52:07 -0800
Subject: [PATCH] stdin: synthesize an "exit" NotificationMessage in abnormal
 termination

---
 src/pipeline.cc | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/src/pipeline.cc b/src/pipeline.cc
index 8c911a04..4c1070e9 100644
--- a/src/pipeline.cc
+++ b/src/pipeline.cc
@@ -503,13 +503,14 @@ void LaunchStdin() {
     set_thread_name("stdin");
     std::string str;
     const std::string_view kContentLength("Content-Length: ");
+    bool received_exit = false;
     while (true) {
       int len = 0;
       str.clear();
       while (true) {
         int c = getchar();
         if (c == EOF)
-          return;
+          goto quit;
         if (c == '\n') {
           if (str.empty())
             break;
@@ -525,7 +526,7 @@ void LaunchStdin() {
       for (int i = 0; i < len; ++i) {
         int c = getchar();
         if (c == EOF)
-          return;
+          goto quit;
         str[i] = c;
       }
 
@@ -549,16 +550,28 @@ void LaunchStdin() {
         LOG_V(2) << "receive NotificationMessage " << method;
       if (method.empty())
         continue;
-      bool should_exit = method == "exit";
+      received_exit = method == "exit";
       // g_config is not available before "initialize". Use 0 in that case.
       on_request->PushBack(
           {id, std::move(method), std::move(message), std::move(document),
            chrono::steady_clock::now() +
                chrono::milliseconds(g_config ? g_config->request.timeout : 0)});
 
-      if (should_exit)
+      if (received_exit)
         break;
     }
+
+  quit:
+    if (!received_exit) {
+      const std::string_view str("{\"jsonrpc\":\"2.0\",\"method\":\"exit\"}");
+      auto message = std::make_unique<char[]>(str.size());
+      std::copy(str.begin(), str.end(), message.get());
+      auto document = std::make_unique<rapidjson::Document>();
+      document->Parse(message.get(), str.size());
+      on_request->PushBack({RequestId(), std::string("exit"),
+                            std::move(message), std::move(document),
+                            chrono::steady_clock::now()});
+    }
     ThreadLeave();
   }).detach();
 }