From 7b32ca113321a596fa006d43807a41062dcdd6dc Mon Sep 17 00:00:00 2001
From: Barak Michener <me@barakmich.com>
Date: Sat, 31 Aug 2013 22:19:18 -0400
Subject: [PATCH] convert to cpp

---
 tpm.c  | 214 -----------------------------------------------------------------
 tpm.cc | 214 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 214 insertions(+), 214 deletions(-)
 delete mode 100644 tpm.c
 create mode 100644 tpm.cc

diff --git a/tpm.c b/tpm.c
deleted file mode 100644
index c3c0926..0000000
--- a/tpm.c
+++ /dev/null
@@ -1,214 +0,0 @@
-#include <errno.h>
-#include <fcntl.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <syslog.h>
-#include <sys/socket.h>
-#include <sys/stat.h>
-#include <sys/time.h>
-#include <sys/types.h>
-#include <sys/un.h>
-#include <time.h>
-#include <unistd.h>
-
-const int kDefaultLengthSecs = 25 * 60;
-const char* kDefaultDoneMessage = "";
-
-char* MakeSocketName() {
-  char* username = getenv("USER");
-  char* socket_name =  (char*) calloc(strlen(username) + strlen(P_tmpdir) + 40, sizeof(char));
-  strcat(socket_name, P_tmpdir);
-  if (P_tmpdir[strlen(P_tmpdir) - 1] != '/') {
-    strcat(socket_name, "/");
-  } 
-  strcat(socket_name, ".tpm-");
-  strcat(socket_name, username);
-  return socket_name;
-}
-
-char* MakePostHookPath() {
-  char* home = getenv("HOME");
-  char* post_hook =  (char*) calloc(strlen(home) + 40, sizeof(char));
-  strcat(post_hook, home);
-  strcat(post_hook, "/.tpm-post.sh");
-  return post_hook;
-}
-
-int ClientMain(char* done_message) {
-  int sock_fd;
-  struct sockaddr_un remote;
-  if ((sock_fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
-    perror("socket");
-    exit(1);
-  }
-  remote.sun_family = AF_UNIX;
-  char * socket_name = MakeSocketName();
-  strcpy(remote.sun_path, socket_name);
-  free(socket_name);
-
-  int len = strlen(remote.sun_path) + sizeof(remote.sun_family) + 1;
-  if (connect(sock_fd, (struct sockaddr *)&remote, len) == -1) {
-    printf("%s\n", done_message);
-    return 0;
-  }
-
-  int recv_length;
-  char buf[100];
-  if ((recv_length = recv(sock_fd, buf, 100, 0)) > 0) {
-    buf[recv_length] = '\0';
-    printf("%s\n", buf);
-  }
-  close(sock_fd);
-  return 0;
-}
-
-int DaemonMain(int countdown_time) {
-        /* Our process ID and Session ID */
-        int ok;
-        pid_t pid, sid;
-        struct timespec ts;
-        ts.tv_sec = 0;
-        ts.tv_nsec = 1000 /*micro*/ * 1000 /* milli */ * 50;
-
-        char * socket_name = MakeSocketName();
-    
-        int sock_fd;
-        struct sockaddr_un local, remote;
-
-        if ((sock_fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
-          perror("socket");
-          exit(1);
-        }
-        local.sun_family = AF_UNIX;
-        strcpy(local.sun_path, socket_name);
-        unlink(local.sun_path);
-        free(socket_name);
-        int len = strlen(local.sun_path) + sizeof(local.sun_family) + 1;
-        if (bind(sock_fd, (struct sockaddr *)&local, len) == -1) {
-          perror("bind");
-          exit(1);
-        }
-        int flags = fcntl(sock_fd,F_GETFL,0);
-        fcntl(sock_fd, F_SETFL, flags | O_NONBLOCK);
-
-        /* Fork off the parent process */
-        pid = fork();
-        if (pid < 0) {
-                exit(EXIT_FAILURE);
-        }
-        /* If we got a good PID, then
-           we can exit the parent process. */
-        if (pid > 0) {
-                exit(EXIT_SUCCESS);
-        }
-
-        /* Change the file mode mask */
-        umask(0);
-                
-        /* Open any logs here */        
-                
-        /* Create a new SID for the child process */
-        sid = setsid();
-        if (sid < 0) {
-                /* Log the failure */
-                exit(EXIT_FAILURE);
-        }
-        
-        
-        /* Change the current working directory */
-        if ((chdir("/")) < 0) {
-                /* Log the failure */
-                exit(EXIT_FAILURE);
-        }
-        
-        /* Close out the standard file descriptors */
-        close(STDIN_FILENO);
-        close(STDOUT_FILENO);
-        close(STDERR_FILENO);
-        
-        if (listen(sock_fd, 5) == -1) {
-          exit(1);
-        }
-
-        
-        /* The Big Loop */
-        int wakeup = 0;
-        struct timeval start_time;
-        ok = gettimeofday(&start_time, NULL);
-        int remote_fd = -1;
-
-        while (!wakeup) {
-           struct timeval current_time;
-           struct timeval diff_time;
-           ok = gettimeofday(&current_time, NULL);
-           timersub(&current_time, &start_time, &diff_time);
-           if (diff_time.tv_sec >= countdown_time) {
-             wakeup = 1;
-             continue;
-           }
-           long remaining_time = (long)countdown_time - diff_time.tv_sec;
-           socklen_t t = sizeof(remote);
-           while ((remote_fd = accept(sock_fd, (struct sockaddr *)&remote, &t)) != -1) {
-             char message[100];
-             int message_len = snprintf(message, 100, "%ld:%02ld", remaining_time / 60, remaining_time % 60);
-             if (send(remote_fd, message, message_len, 0) < 0) {
-              exit(1);
-             }
-             close(remote_fd);
-           }
-           // Make sure it was just EWOULDBLOCK
-           if (errno != EAGAIN && errno != EWOULDBLOCK) {
-             exit(1);
-           }
-           ok = nanosleep(&ts, NULL);
-        }
-   close(sock_fd);
-   printf("%s\n", local.sun_path);
-   unlink(local.sun_path);
-   char* post_hook = MakePostHookPath();
-   if (access(post_hook, X_OK) != -1) {
-     execv(post_hook, NULL);
-   }
-   free(post_hook);
-   exit(EXIT_SUCCESS);
-}
-
-int main(int argc, char** argv) {
-    int c;
-    int countdown_time = kDefaultLengthSecs;
-    char* done_message = (char*) kDefaultDoneMessage;
-    int got_positional = 0;
-    char* command = "";
-    while (1) {
-      while ( (c = getopt(argc, argv, "bs:m:d:")) != -1) {
-          switch (c) {
-          case 's':
-              countdown_time = atoi(optarg);
-              break;
-          case 'm':
-              countdown_time = atoi(optarg) * 60;
-              break;
-          case 'd':
-              done_message = optarg;
-              break;
-          }
-      }
-      if (optind < argc && !got_positional) {
-          command = argv[optind];
-          got_positional = 1; 
-      } else {
-        break;
-      }
-      optind++;
-    }
-
-
-    if (strcmp(command, "start") == 0) {
-      DaemonMain(countdown_time);
-    }
-    else {
-      return ClientMain(done_message);
-    }
-    exit(0);   
-}
diff --git a/tpm.cc b/tpm.cc
new file mode 100644
index 0000000..c3c0926
--- /dev/null
+++ b/tpm.cc
@@ -0,0 +1,214 @@
+#include <errno.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <syslog.h>
+#include <sys/socket.h>
+#include <sys/stat.h>
+#include <sys/time.h>
+#include <sys/types.h>
+#include <sys/un.h>
+#include <time.h>
+#include <unistd.h>
+
+const int kDefaultLengthSecs = 25 * 60;
+const char* kDefaultDoneMessage = "";
+
+char* MakeSocketName() {
+  char* username = getenv("USER");
+  char* socket_name =  (char*) calloc(strlen(username) + strlen(P_tmpdir) + 40, sizeof(char));
+  strcat(socket_name, P_tmpdir);
+  if (P_tmpdir[strlen(P_tmpdir) - 1] != '/') {
+    strcat(socket_name, "/");
+  } 
+  strcat(socket_name, ".tpm-");
+  strcat(socket_name, username);
+  return socket_name;
+}
+
+char* MakePostHookPath() {
+  char* home = getenv("HOME");
+  char* post_hook =  (char*) calloc(strlen(home) + 40, sizeof(char));
+  strcat(post_hook, home);
+  strcat(post_hook, "/.tpm-post.sh");
+  return post_hook;
+}
+
+int ClientMain(char* done_message) {
+  int sock_fd;
+  struct sockaddr_un remote;
+  if ((sock_fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
+    perror("socket");
+    exit(1);
+  }
+  remote.sun_family = AF_UNIX;
+  char * socket_name = MakeSocketName();
+  strcpy(remote.sun_path, socket_name);
+  free(socket_name);
+
+  int len = strlen(remote.sun_path) + sizeof(remote.sun_family) + 1;
+  if (connect(sock_fd, (struct sockaddr *)&remote, len) == -1) {
+    printf("%s\n", done_message);
+    return 0;
+  }
+
+  int recv_length;
+  char buf[100];
+  if ((recv_length = recv(sock_fd, buf, 100, 0)) > 0) {
+    buf[recv_length] = '\0';
+    printf("%s\n", buf);
+  }
+  close(sock_fd);
+  return 0;
+}
+
+int DaemonMain(int countdown_time) {
+        /* Our process ID and Session ID */
+        int ok;
+        pid_t pid, sid;
+        struct timespec ts;
+        ts.tv_sec = 0;
+        ts.tv_nsec = 1000 /*micro*/ * 1000 /* milli */ * 50;
+
+        char * socket_name = MakeSocketName();
+    
+        int sock_fd;
+        struct sockaddr_un local, remote;
+
+        if ((sock_fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
+          perror("socket");
+          exit(1);
+        }
+        local.sun_family = AF_UNIX;
+        strcpy(local.sun_path, socket_name);
+        unlink(local.sun_path);
+        free(socket_name);
+        int len = strlen(local.sun_path) + sizeof(local.sun_family) + 1;
+        if (bind(sock_fd, (struct sockaddr *)&local, len) == -1) {
+          perror("bind");
+          exit(1);
+        }
+        int flags = fcntl(sock_fd,F_GETFL,0);
+        fcntl(sock_fd, F_SETFL, flags | O_NONBLOCK);
+
+        /* Fork off the parent process */
+        pid = fork();
+        if (pid < 0) {
+                exit(EXIT_FAILURE);
+        }
+        /* If we got a good PID, then
+           we can exit the parent process. */
+        if (pid > 0) {
+                exit(EXIT_SUCCESS);
+        }
+
+        /* Change the file mode mask */
+        umask(0);
+                
+        /* Open any logs here */        
+                
+        /* Create a new SID for the child process */
+        sid = setsid();
+        if (sid < 0) {
+                /* Log the failure */
+                exit(EXIT_FAILURE);
+        }
+        
+        
+        /* Change the current working directory */
+        if ((chdir("/")) < 0) {
+                /* Log the failure */
+                exit(EXIT_FAILURE);
+        }
+        
+        /* Close out the standard file descriptors */
+        close(STDIN_FILENO);
+        close(STDOUT_FILENO);
+        close(STDERR_FILENO);
+        
+        if (listen(sock_fd, 5) == -1) {
+          exit(1);
+        }
+
+        
+        /* The Big Loop */
+        int wakeup = 0;
+        struct timeval start_time;
+        ok = gettimeofday(&start_time, NULL);
+        int remote_fd = -1;
+
+        while (!wakeup) {
+           struct timeval current_time;
+           struct timeval diff_time;
+           ok = gettimeofday(&current_time, NULL);
+           timersub(&current_time, &start_time, &diff_time);
+           if (diff_time.tv_sec >= countdown_time) {
+             wakeup = 1;
+             continue;
+           }
+           long remaining_time = (long)countdown_time - diff_time.tv_sec;
+           socklen_t t = sizeof(remote);
+           while ((remote_fd = accept(sock_fd, (struct sockaddr *)&remote, &t)) != -1) {
+             char message[100];
+             int message_len = snprintf(message, 100, "%ld:%02ld", remaining_time / 60, remaining_time % 60);
+             if (send(remote_fd, message, message_len, 0) < 0) {
+              exit(1);
+             }
+             close(remote_fd);
+           }
+           // Make sure it was just EWOULDBLOCK
+           if (errno != EAGAIN && errno != EWOULDBLOCK) {
+             exit(1);
+           }
+           ok = nanosleep(&ts, NULL);
+        }
+   close(sock_fd);
+   printf("%s\n", local.sun_path);
+   unlink(local.sun_path);
+   char* post_hook = MakePostHookPath();
+   if (access(post_hook, X_OK) != -1) {
+     execv(post_hook, NULL);
+   }
+   free(post_hook);
+   exit(EXIT_SUCCESS);
+}
+
+int main(int argc, char** argv) {
+    int c;
+    int countdown_time = kDefaultLengthSecs;
+    char* done_message = (char*) kDefaultDoneMessage;
+    int got_positional = 0;
+    char* command = "";
+    while (1) {
+      while ( (c = getopt(argc, argv, "bs:m:d:")) != -1) {
+          switch (c) {
+          case 's':
+              countdown_time = atoi(optarg);
+              break;
+          case 'm':
+              countdown_time = atoi(optarg) * 60;
+              break;
+          case 'd':
+              done_message = optarg;
+              break;
+          }
+      }
+      if (optind < argc && !got_positional) {
+          command = argv[optind];
+          got_positional = 1; 
+      } else {
+        break;
+      }
+      optind++;
+    }
+
+
+    if (strcmp(command, "start") == 0) {
+      DaemonMain(countdown_time);
+    }
+    else {
+      return ClientMain(done_message);
+    }
+    exit(0);   
+}