From 59874008ddd8304dd490b5f3493899cc2fd83fba Mon Sep 17 00:00:00 2001
From: Colin Patrick McCabe <cmccabe@alumni.cmu.edu>
Date: Thu, 10 Feb 2011 06:44:03 -0800
Subject: [PATCH] msgr: better error checking for write_pid_file

Use safe_write. Check error code of close.

Fix what looks like a "lack of braces around multi-statement block"
error where we always complain that we're unable to write the pid file,
even if we succeed.

Signed-off-by: Colin McCabe <colin.mccabe@dreamhost.com>
---
 src/msg/SimpleMessenger.cc | 50 +++++++++++++++++++++++++-------------
 1 file changed, 33 insertions(+), 17 deletions(-)

diff --git a/src/msg/SimpleMessenger.cc b/src/msg/SimpleMessenger.cc
index c0ac78f636017..88f1b8a86f28c 100644
--- a/src/msg/SimpleMessenger.cc
+++ b/src/msg/SimpleMessenger.cc
@@ -33,8 +33,9 @@
 #include <iostream>
 #include <fstream>
 
-#include "common/errno.h"
 #include "common/Timer.h"
+#include "common/errno.h"
+#include "common/safe_io.h"
 #include "common/signal.h"
 
 #define DOUT_SUBSYS ms
@@ -2363,25 +2364,40 @@ static void handle_signal(int sig)
 
 int SimpleMessenger::write_pid_file(int pid)
 {
+  int ret, fd;
+
   if (!g_conf.pid_file)
     return 0;
 
-  int r = 0;
-  int fd = ::open(g_conf.pid_file, O_CREAT|O_TRUNC|O_WRONLY, 0644);
-  if (fd >= 0) {
-    char buf[20];
-    int len = snprintf(buf, sizeof(buf), "%d\n", pid);
-    r = ::write(fd, buf, len);
-    if (r < 0)
-      r = -errno;
-    ::close(fd);
-
-    signal(SIGTERM, handle_signal);
-    signal(SIGINT, handle_signal);
-  } else
-    r = -errno;
-  derr << "unable to write pid to " << g_conf.pid_file << ": " << cpp_strerror(errno) << dendl;
-  return r;
+  fd = TEMP_FAILURE_RETRY(::open(g_conf.pid_file,
+				 O_CREAT|O_TRUNC|O_WRONLY, 0644));
+  if (fd < 0) {
+    int err = errno;
+    derr << "SimpleMessenger::write_pid_file: failed to open pid file '"
+	 << g_conf.pid_file << "': " << cpp_strerror(err) << dendl;
+    return err;
+  }
+
+  char buf[20];
+  int len = snprintf(buf, sizeof(buf), "%d\n", pid);
+  ret = safe_write(fd, buf, len);
+  if (ret < 0) {
+    derr << "SimpleMessenger::write_pid_file: failed to write to pid file '"
+	 << g_conf.pid_file << "': " << cpp_strerror(ret) << dendl;
+    TEMP_FAILURE_RETRY(::close(fd));
+    return ret;
+  }
+  if (TEMP_FAILURE_RETRY(::close(fd))) {
+    ret = errno;
+    derr << "SimpleMessenger::write_pid_file: failed to close to pid file '"
+	 << g_conf.pid_file << "': " << cpp_strerror(ret) << dendl;
+    return ret;
+  }
+
+  signal(SIGTERM, handle_signal);
+  signal(SIGINT, handle_signal);
+
+  return 0;
 }
 
 int SimpleMessenger::start(bool nodaemon)
-- 
2.39.5