Xinqi Bao's Git

ip: fixed memory leak
[slstatus.git] / slstatus.c
index 7ff323f..3222b02 100644 (file)
@@ -17,7 +17,6 @@ struct arg {
        const char *args;
 };
 
        const char *args;
 };
 
-char *argv0;
 char buf[1024];
 static int done;
 static Display *dpy;
 char buf[1024];
 static int done;
 static Display *dpy;
@@ -43,8 +42,7 @@ difftimespec(struct timespec *res, struct timespec *a, struct timespec *b)
 static void
 usage(void)
 {
 static void
 usage(void)
 {
-       fprintf(stderr, "usage: %s [-s]\n", argv0);
-       exit(1);
+       die("usage: %s [-s]", argv0);
 }
 
 int
 }
 
 int
@@ -80,14 +78,12 @@ main(int argc, char *argv[])
        }
 
        if (!sflag && !(dpy = XOpenDisplay(NULL))) {
        }
 
        if (!sflag && !(dpy = XOpenDisplay(NULL))) {
-               fprintf(stderr, "XOpenDisplay: Failed to open display\n");
-               return 1;
+               die("XOpenDisplay: Failed to open display");
        }
 
        while (!done) {
                if (clock_gettime(CLOCK_MONOTONIC, &start) < 0) {
        }
 
        while (!done) {
                if (clock_gettime(CLOCK_MONOTONIC, &start) < 0) {
-                       fprintf(stderr, "clock_gettime: %s\n", strerror(errno));
-                       return 1;
+                       die("clock_gettime:");
                }
 
                status[0] = '\0';
                }
 
                status[0] = '\0';
@@ -95,13 +91,8 @@ main(int argc, char *argv[])
                        if (!(res = args[i].func(args[i].args))) {
                                res = unknown_str;
                        }
                        if (!(res = args[i].func(args[i].args))) {
                                res = unknown_str;
                        }
-                       if ((ret = snprintf(status + len, sizeof(status) - len,
+                       if ((ret = esnprintf(status + len, sizeof(status) - len,
                                            args[i].fmt, res)) < 0) {
                                            args[i].fmt, res)) < 0) {
-                               fprintf(stderr, "snprintf: %s\n",
-                                       strerror(errno));
-                               break;
-                       } else if ((size_t)ret >= sizeof(status) - len) {
-                               fprintf(stderr, "snprintf: Output truncated\n");
                                break;
                        }
                        len += ret;
                                break;
                        }
                        len += ret;
@@ -111,18 +102,14 @@ main(int argc, char *argv[])
                        printf("%s\n", status);
                } else {
                        if (XStoreName(dpy, DefaultRootWindow(dpy), status) < 0) {
                        printf("%s\n", status);
                } else {
                        if (XStoreName(dpy, DefaultRootWindow(dpy), status) < 0) {
-                               fprintf(stderr,
-                                       "XStoreName: Allocation failed\n");
-                               return 1;
+                               die("XStoreName: Allocation failed");
                        }
                        XFlush(dpy);
                }
 
                if (!done) {
                        if (clock_gettime(CLOCK_MONOTONIC, &current) < 0) {
                        }
                        XFlush(dpy);
                }
 
                if (!done) {
                        if (clock_gettime(CLOCK_MONOTONIC, &current) < 0) {
-                               fprintf(stderr, "clock_gettime: %s\n",
-                                       strerror(errno));
-                               return 1;
+                               die("clock_gettime:");
                        }
                        difftimespec(&diff, &current, &start);
 
                        }
                        difftimespec(&diff, &current, &start);
 
@@ -133,9 +120,7 @@ main(int argc, char *argv[])
                        if (wait.tv_sec >= 0) {
                                if (nanosleep(&wait, NULL) < 0 &&
                                    errno != EINTR) {
                        if (wait.tv_sec >= 0) {
                                if (nanosleep(&wait, NULL) < 0 &&
                                    errno != EINTR) {
-                                       fprintf(stderr, "nanosleep: %s\n",
-                                               strerror(errno));
-                                       return 1;
+                                       die("nanosleep:");
                                }
                        }
                }
                                }
                        }
                }
@@ -144,9 +129,7 @@ main(int argc, char *argv[])
        if (!sflag) {
                XStoreName(dpy, DefaultRootWindow(dpy), NULL);
                if (XCloseDisplay(dpy) < 0) {
        if (!sflag) {
                XStoreName(dpy, DefaultRootWindow(dpy), NULL);
                if (XCloseDisplay(dpy) < 0) {
-                       fprintf(stderr,
-                               "XCloseDisplay: Failed to close display\n");
-                       return 1;
+                       die("XCloseDisplay: Failed to close display");
                }
        }
 
                }
        }