X-Git-Url: https://git.danieliu.xyz/?p=dwmblocks.git;a=blobdiff_plain;f=dwmblocks.c;h=64ad7a33e2d6b6acf39f28b899e1a74a23be9cee;hp=f7f23cc51080f8f0325105a4ed8319bfdb50e397;hb=b6b0be4f4fe77fea6f1aef07232a44abd0139098;hpb=09a00ab345f1c1c4bce5496f7c7d054476e35392 diff --git a/dwmblocks.c b/dwmblocks.c index f7f23cc..64ad7a3 100644 --- a/dwmblocks.c +++ b/dwmblocks.c @@ -3,8 +3,20 @@ #include #include #include +#ifndef NO_X #include +#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; @@ -12,75 +24,73 @@ typedef struct { unsigned int interval; unsigned int signal; } Block; +#ifndef __OpenBSD__ +void dummysighandler(int num); +#endif void sighandler(int num); -void replace(char *str, char old, char new); void getcmds(int time); -void getsigcmds(int signal); +void getsigcmds(unsigned int signal); void setupsignals(); +void sighandler(int signum); int getstatus(char *str, char *last); -void setroot(); void statusloop(); -void statusinit(); -void sighandler(int signum); -void termhandler(int signum); +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; +#else +static void (*writestatus) () = pstdout; +#endif #include "blocks.h" -static Display *dpy; -static int screen; -static Window root; -static char statusbar[LENGTH(blocks)][50] = {0}; -static char statusstr[2][256]; +static char statusbar[LENGTH(blocks)][CMDLENGTH] = {0}; +static char statusstr[2][STATUSLENGTH]; static int statusContinue = 1; -static void (*writestatus) () = setroot; - -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; -} +static int returnStatus = 0; //opens process *cmd and stores output in *output void getcmd(const Block *block, char *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); - while((c = fgetc(cmdf)) != EOF) - { - output[i++] = c; - if(c == '\n') - break; + fgets(output+i, CMDLENGTH-i-delimLen, cmdf); + i = strlen(output); + if (i == 0)//return if block and command output are both empty + 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); } - if (delim != '\0' && --i) - output[i++] = delim; - output[i++] = '\0'; + 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]); } } -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]); @@ -89,10 +99,15 @@ void getsigcmds(int signal) void setupsignals() { - 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); + signal(SIGMINUS+blocks[i].signal, sighandler); } } @@ -101,27 +116,33 @@ 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]); - } - 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() { @@ -137,43 +158,53 @@ void statusloop() setupsignals(); int i = 0; getcmds(-1); - while(statusContinue) - { - getcmds(i); + while (1) { + getcmds(i++); writestatus(); + if (!statusContinue) + break; sleep(1.0); - i++; } } -void statusinit() +#ifndef __OpenBSD__ +/* this signal handler should do nothing */ +void dummysighandler(int signum) { - statusloop(); + return; } - +#endif void sighandler(int signum) { - getsigcmds(signum-SIGRTMIN); + getsigcmds(signum-SIGPLUS); writestatus(); } -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][0]; - 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); - statusinit(); + statusloop(); +#ifndef NO_X + XCloseDisplay(dpy); +#endif + return 0; }