X-Git-Url: https://ruderich.org/simon/gitweb/?a=blobdiff_plain;f=src%2Fwall-notify.c;h=7fb3b33cbc0bdaaf2e7436abf77bd7ca4b8d4fa1;hb=d9e9fb8ccfa208052b89aee8cfd3fc6c38752b96;hp=dd37e897a03ccea27adb731e0466fea59b232f6f;hpb=f03987a4a9b3d22308c08d336258660d21e3ebf0;p=wall-notify%2Fwall-notify.git

diff --git a/src/wall-notify.c b/src/wall-notify.c
index dd37e89..7fb3b33 100644
--- a/src/wall-notify.c
+++ b/src/wall-notify.c
@@ -205,9 +205,6 @@ static void pass_buffer_to_program(const char *buffer, size_t length, char **arg
 
     pid_t pid;
 
-    /* Skip argv[0]. */
-    argv++;
-
     if (pipe(fds) != 0) {
         perror("pipe");
         return;
@@ -294,14 +291,23 @@ static void handle_wall(int fd, char **argv) {
     }
 }
 
+static void usage(const char *argv0) {
+    fprintf(stderr, "usage: %s <cmd args..>\n", argv0);
+    exit(EXIT_FAILURE);
+}
+
 
 int main(int argc, char **argv) {
     int ptm, pts;
     char *name;
+    const char *argv0;
+
+    argv0 = argv[0];
+    /* Don't pass our argv[0] to the notification program. */
+    argv++;
 
     if (argc < 2) {
-        fprintf(stderr, "usage: %s <cmd args..>\n", argv[0]);
-        exit(EXIT_FAILURE);
+        usage(argv0);
     }
 
     ptm = open_tty();