diff options
Diffstat (limited to 'dwmblocks.c')
| -rw-r--r-- | dwmblocks.c | 207 |
1 files changed, 94 insertions, 113 deletions
diff --git a/dwmblocks.c b/dwmblocks.c index 391ee0d..ded717c 100644 --- a/dwmblocks.c +++ b/dwmblocks.c @@ -3,9 +3,20 @@ #include<string.h> #include<unistd.h> #include<signal.h> +#ifndef NO_X #include<X11/Xlib.h> +#endif +#ifdef __OpenBSD__ +#define SIGPLUS SIGUSR1+1 +#define SIGMINUS SIGUSR1-1 +#else +#define SIGPLUS SIGRTMIN +#define SIGMINUS SIGRTMIN +#endif #define LENGTH(X) (sizeof(X) / sizeof (X[0])) #define CMDLENGTH 50 +#define MIN( a, b ) ( ( a < b) ? a : b ) +#define STATUSLENGTH (LENGTH(blocks) * CMDLENGTH + 1) typedef struct { char* icon; @@ -13,95 +24,76 @@ typedef struct { unsigned int interval; unsigned int signal; } Block; +#ifndef __OpenBSD__ +void dummysighandler(int num); +#endif void sighandler(int num); -void buttonhandler(int sig, siginfo_t *si, void *ucontext); -void replace(char *str, char old, char new); -void remove_all(char *str, char to_remove); void getcmds(int time); -#ifndef __OpenBSD__ -void getsigcmds(int signal); +void getsigcmds(unsigned int signal); void setupsignals(); void sighandler(int signum); -#endif int getstatus(char *str, char *last); -void setroot(); void statusloop(); -void termhandler(int signum); - - -#include "config.h" - +void termhandler(); +void pstdout(); +#ifndef NO_X +void setroot(); +static void (*writestatus) () = setroot; +static int setupX(); static Display *dpy; static int screen; static Window root; -static char statusbar[LENGTH(blocks)][CMDLENGTH] = {0}; -static char statusstr[2][256]; -static int statusContinue = 1; -static void (*writestatus) () = setroot; +#else +static void (*writestatus) () = pstdout; +#endif -void replace(char *str, char old, char new) -{ - int N = strlen(str); - for(int i = 0; i < N; i++) - if(str[i] == old) - str[i] = new; -} -void remove_all(char *str, char to_remove) { - char *read = str; - char *write = str; - while (*read) { - if (*read == to_remove) { - read++; - *write = *read; - } - read++; - write++; - } -} +#include "blocks.h" + +static char statusbar[LENGTH(blocks)][CMDLENGTH] = {0}; +static char statusstr[2][STATUSLENGTH]; +static int statusContinue = 1; +static int returnStatus = 0; //opens process *cmd and stores output in *output void getcmd(const Block *block, char *output) { - if (block->signal) - { - output[0] = block->signal; - output++; - } strcpy(output, block->icon); - char *cmd = block->command; - FILE *cmdf = popen(cmd,"r"); + FILE *cmdf = popen(block->command, "r"); if (!cmdf) return; - char c; int i = strlen(block->icon); - fgets(output+i, CMDLENGTH-(strlen(delim)+1), cmdf); - remove_all(output, '\n'); + fgets(output+i, CMDLENGTH-i-delimLen, cmdf); i = strlen(output); - if ((i > 0 && block != &blocks[LENGTH(blocks) - 1])) - strcat(output, delim); - i+=strlen(delim); - output[i++] = '\0'; + if (i == 0) { + //return if block and command output are both empty + pclose(cmdf); + return; + } + if (delim[0] != '\0') { + //only chop off newline if one is present at the end + i = output[i-1] == '\n' ? i-1 : i; + strncpy(output+i, delim, delimLen); + } + else + output[i++] = '\0'; pclose(cmdf); } void getcmds(int time) { const Block* current; - for(int i = 0; i < LENGTH(blocks); i++) - { + for (unsigned int i = 0; i < LENGTH(blocks); i++) { current = blocks + i; if ((current->interval != 0 && time % current->interval == 0) || time == -1) getcmd(current,statusbar[i]); } } -#ifndef __OpenBSD__ -void getsigcmds(int signal) +void getsigcmds(unsigned int signal) { const Block *current; - for (int i = 0; i < LENGTH(blocks); i++) - { + for (unsigned int i = 0; i < LENGTH(blocks); i++) { current = blocks + i; if (current->signal == signal) getcmd(current,statusbar[i]); @@ -110,49 +102,50 @@ void getsigcmds(int signal) void setupsignals() { - struct sigaction sa; - for(int i = 0; i < LENGTH(blocks); i++) - { +#ifndef __OpenBSD__ + /* initialize all real time signals with dummy handler */ + for (int i = SIGRTMIN; i <= SIGRTMAX; i++) + signal(i, dummysighandler); +#endif + + for (unsigned int i = 0; i < LENGTH(blocks); i++) { if (blocks[i].signal > 0) - { - signal(SIGRTMIN+blocks[i].signal, sighandler); - sigaddset(&sa.sa_mask, SIGRTMIN+blocks[i].signal); - } + signal(SIGMINUS+blocks[i].signal, sighandler); } - sa.sa_sigaction = buttonhandler; - sa.sa_flags = SA_SIGINFO; - sigaction(SIGUSR1, &sa, NULL); - signal(SIGCHLD, SIG_IGN); } -#endif int getstatus(char *str, char *last) { strcpy(last, str); str[0] = '\0'; - for(int i = 0; i < LENGTH(blocks); i++) { + for (unsigned int i = 0; i < LENGTH(blocks); i++) strcat(str, statusbar[i]); - if (i == LENGTH(blocks) - 1) - strcat(str, " "); - } - str[strlen(str)-1] = '\0'; + str[strlen(str)-strlen(delim)] = '\0'; return strcmp(str, last);//0 if they are the same } +#ifndef NO_X void setroot() { if (!getstatus(statusstr[0], statusstr[1]))//Only set root if text has changed. return; - Display *d = XOpenDisplay(NULL); - if (d) { - dpy = d; + XStoreName(dpy, root, statusstr[0]); + XFlush(dpy); +} + +int setupX() +{ + dpy = XOpenDisplay(NULL); + if (!dpy) { + fprintf(stderr, "dwmblocks: Failed to open display\n"); + return 0; } screen = DefaultScreen(dpy); root = RootWindow(dpy, screen); - XStoreName(dpy, root, statusstr[0]); - XCloseDisplay(dpy); + return 1; } +#endif void pstdout() { @@ -165,68 +158,56 @@ void pstdout() void statusloop() { -#ifndef __OpenBSD__ setupsignals(); -#endif int i = 0; getcmds(-1); - while(statusContinue) - { - getcmds(i); + while (1) { + getcmds(i++); writestatus(); + if (!statusContinue) + break; sleep(1.0); - i++; } } #ifndef __OpenBSD__ -void sighandler(int signum) +/* this signal handler should do nothing */ +void dummysighandler(int signum) { - getsigcmds(signum-SIGRTMIN); - writestatus(); + return; } +#endif -void buttonhandler(int sig, siginfo_t *si, void *ucontext) +void sighandler(int signum) { - char button[2] = {'0' + si->si_value.sival_int & 0xff, '\0'}; - sig = si->si_value.sival_int >> 8; - if (fork() == 0) - { - const Block *current; - for (int i = 0; i < LENGTH(blocks); i++) - { - current = blocks + i; - if (current->signal == sig) - break; - } - char *command[] = { "/bin/sh", "-c", current->command, NULL }; - setenv("BLOCK_BUTTON", button, 1); - setsid(); - execvp(command[0], command); - exit(EXIT_SUCCESS); - } - getsigcmds(sig); + getsigcmds(signum-SIGPLUS); writestatus(); } -#endif - -void termhandler(int signum) +void termhandler() { statusContinue = 0; - exit(0); } int main(int argc, char** argv) { - for(int i = 0; i < argc; i++) - { + for (int i = 0; i < argc; i++) {//Handle command line arguments if (!strcmp("-d",argv[i])) - delim = argv[++i]; - else if(!strcmp("-p",argv[i])) + strncpy(delim, argv[++i], delimLen); + else if (!strcmp("-p",argv[i])) writestatus = pstdout; } +#ifndef NO_X + if (!setupX()) + return 1; +#endif + delimLen = MIN(delimLen, strlen(delim)); + delim[delimLen++] = '\0'; signal(SIGTERM, termhandler); signal(SIGINT, termhandler); statusloop(); +#ifndef NO_X + XCloseDisplay(dpy); +#endif + return 0; } |