Add a cache around getpwuid/getgrgid to avoid repeated calls.

[cumulus.git] / util.cc

diff --git a/util.cc b/util.cc
index 22aab3b..0aa5efb 100644 (file)
--- a/util.cc
+++ b/util.cc
@@ -22,6 +22,9 @@
  * later, for parsing them back, perhaps). */
 
 #include <stdio.h>
+#include <stdlib.h>
+#include <fcntl.h>
+#include <unistd.h>
 #include <uuid/uuid.h>
 
 #include <iostream>
@@ -114,3 +117,21 @@ long long parse_int(const string &s)
 {
     return strtoll(s.c_str(), NULL, 0);
 }
+
+/* Mark a file descriptor as close-on-exec. */
+void cloexec(int fd)
+{
+    long flags = fcntl(fd, F_GETFD);
+
+    if (flags < 0)
+        return;
+
+    fcntl(fd, F_SETFD, flags | FD_CLOEXEC);
+}
+
+/* Report a fatal error and exit. */
+void fatal(string msg)
+{
+    fprintf(stderr, "FATAL: %s\n", msg.c_str());
+    exit(1);
+}