Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
225 changes: 177 additions & 48 deletions herbe.c
Original file line number Diff line number Diff line change
@@ -1,13 +1,18 @@
#include <X11/Xlib.h>
#include <X11/Xft/Xft.h>
#include <X11/Xlib.h>
#include <errno.h>
#include <fcntl.h>
#include <fcntl.h>
#include <mqueue.h>
#include <signal.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <unistd.h>
#include <string.h>
#include <stdarg.h>
#include <fcntl.h>
#include <semaphore.h>
#include <sys/file.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <unistd.h>

#include "config.h"

Expand All @@ -16,9 +21,19 @@
#define EXIT_DISMISS 2

Display *display;
XftFont *font;
Window window;
int num_of_lines;
char **lines;
int exit_code = EXIT_DISMISS;

struct mq_object {
pid_t pid;
long timestamp;
char buffer[1024];
};
long lastTimestamp;

static void die(const char *format, ...)
{
va_list ap;
Expand All @@ -29,6 +44,19 @@ static void die(const char *format, ...)
exit(EXIT_FAIL);
}

void read_y_offset(unsigned int **offset, int *id) {
int shm_id = shmget(8432, sizeof(unsigned int), IPC_CREAT | 0660);
if (shm_id == -1) die("shmget failed");

*offset = (unsigned int *)shmat(shm_id, 0, 0);
if (*offset == (unsigned int *)-1) die("shmat failed\n");
*id = shm_id;
}

void free_y_offset(int id) {
shmctl(id, IPC_RMID, NULL);
}

int get_max_len(char *string, XftFont *font, int max_text_width)
{
int eol = strlen(string);
Expand Down Expand Up @@ -70,6 +98,81 @@ int get_max_len(char *string, XftFont *font, int max_text_width)
return ++eol;
}

void freeLines() {
if(lines) {
for (int i = 0; i < num_of_lines; i++)
free(lines[i]);
free(lines);
}
}

void constructLines(char* strList[], int numberOfStrings) {
freeLines();
int max_text_width = width - 2 * padding;
num_of_lines = 0;
int lines_size = 5;
lines = malloc(lines_size * sizeof(char *));
if (!lines)
die("malloc failed");

for (int i = 0; i < numberOfStrings; i++)
{
for (unsigned int eol = get_max_len(strList[i], font, max_text_width); eol; strList[i] += eol, num_of_lines++, eol = get_max_len(strList[i], font, max_text_width))
{
if (lines_size <= num_of_lines)
{
lines = realloc(lines, (lines_size += 5) * sizeof(char *));
if (!lines)
die("realloc failed");
}
lines[num_of_lines] = malloc((eol + 1) * sizeof(char));
if (!lines[num_of_lines])
die("malloc failed");

strncpy(lines[num_of_lines], strList[i], eol);
lines[num_of_lines][eol] = '\0';
}
}
}

void reload(union sigval sv);
void readAllEvents(mqd_t mqd) {

struct sigevent event = {.sigev_notify=SIGEV_THREAD, .sigev_signo=SIGHUP, .sigev_value.sival_int=mqd, .sigev_notify_function=reload};
if(mq_notify(mqd, &event) == -1) {
perror("mq_notify failed");
exit(1);
}
struct mq_object object;
while(1) {
int ret = mq_receive(mqd, (char*)&object, sizeof(object), NULL);
if(ret==-1) {
if(errno == EAGAIN)
return;
perror("mq_receive");
exit(1);
}
if(object.timestamp && lastTimestamp > object.timestamp)
return;
if(object.timestamp)
lastTimestamp = object.timestamp;
char *buffer = object.buffer;

constructLines(&buffer, 1);
kill(object.pid, SIGTERM);
}
}
void reload(union sigval sv) {
// we've already timed out
if(alarm(duration) == 0)
return;
readAllEvents(sv.sival_int);
XEvent event;
event.type = Expose;
XSendEvent(display, window, 0, 0, &event);
XFlush(display);
}

void expire(int sig)
{
XEvent event;
Expand All @@ -79,14 +182,56 @@ void expire(int sig)
XFlush(display);
}


void exitSuccess() {
exit(0);
}

int main(int argc, char *argv[])
{
if (argc == 1)
{
sem_unlink("/herbe");
die("Usage: %s body", argv[0]);
}

const char* id =getenv("HERBE_ID");
mqd_t mqd=-1;
if(id) {
struct mq_attr attr = { .mq_maxmsg = 10, .mq_msgsize = sizeof(struct mq_object) };
mqd = mq_open(id, O_RDWR|O_CREAT|O_NONBLOCK, 0722, &attr);
if(mqd==-1){
perror("mq_open");
die("mq_open");
}
while (1) {
if(flock(mqd, LOCK_EX|LOCK_NB) == 0) {
// if we get the lock, register for events
break;
}
if(errno != EWOULDBLOCK) {
perror("flock");
exit(1);
}
// someone else is listening for events
char* ts_str = getenv("NOTIFICATION_ID");
lastTimestamp = ts_str?atol(ts_str):0;
struct mq_object object = {getpid(), lastTimestamp, {0}};
char *buffer=object.buffer;
for(int i=1;i<argc;i++) {
strcat(buffer, argv[i]);
strcat(buffer, "\n");
}
signal(SIGTERM, exitSuccess);
if(mq_send(mqd, (char*)&object, sizeof(object), 1)==-1) {
perror("mq_send");
exit(1);
}
signal(SIGALRM, SIG_IGN);
alarm(1);
pause();
}
}

struct sigaction act_expire, act_ignore;

act_expire.sa_handler = expire;
Expand Down Expand Up @@ -121,46 +266,29 @@ int main(int argc, char *argv[])
attributes.background_pixel = color.pixel;
XftColorAllocName(display, visual, colormap, border_color, &color);
attributes.border_pixel = color.pixel;
font = XftFontOpenName(display, screen, font_pattern);

int num_of_lines = 0;
int max_text_width = width - 2 * padding;
int lines_size = 5;
char **lines = malloc(lines_size * sizeof(char *));
if (!lines)
die("malloc failed");
constructLines(argv+1, argc-1);

XftFont *font = XftFontOpenName(display, screen, font_pattern);
//unsigned int x = pos_x;
//unsigned int y = pos_y;
int y_offset_id;
unsigned int *y_offset;
read_y_offset(&y_offset, &y_offset_id);

for (int i = 1; i < argc; i++)
{
for (unsigned int eol = get_max_len(argv[i], font, max_text_width); eol; argv[i] += eol, num_of_lines++, eol = get_max_len(argv[i], font, max_text_width))
{
if (lines_size <= num_of_lines)
{
lines = realloc(lines, (lines_size += 5) * sizeof(char *));
if (!lines)
die("realloc failed");
}

lines[num_of_lines] = malloc((eol + 1) * sizeof(char));
if (!lines[num_of_lines])
die("malloc failed");

strncpy(lines[num_of_lines], argv[i], eol);
lines[num_of_lines][eol] = '\0';
}
}

unsigned int x = pos_x;
unsigned int y = pos_y;
unsigned int text_height = font->ascent - font->descent;
unsigned int height = (num_of_lines - 1) * line_spacing + num_of_lines * text_height + 2 * padding;
unsigned int x = pos_x;
unsigned int y = pos_y + *y_offset;
unsigned int used_y_offset = (*y_offset) += height + padding;

if (corner == TOP_RIGHT || corner == BOTTOM_RIGHT)
x = screen_width - width - border_size * 2 - pos_x;
x = screen_width - width - border_size * 2 - x;
//x = screen_width - width - border_size * 2 - pos_x;

if (corner == BOTTOM_LEFT || corner == BOTTOM_RIGHT)
y = screen_height - height - border_size * 2 - pos_y;
y = screen_height - height - border_size * 2 - y;
//y = screen_height - height - border_size * 2 - pos_y;

window = XCreateWindow(display, RootWindow(display, screen), x, y, width, height, border_size, DefaultDepth(display, screen),
CopyFromParent, visual, CWOverrideRedirect | CWBackPixel | CWBorderPixel, &attributes);
Expand All @@ -171,15 +299,16 @@ int main(int argc, char *argv[])
XSelectInput(display, window, ExposureMask | ButtonPress);
XMapWindow(display, window);

sem_t *mutex = sem_open("/herbe", O_CREAT, 0644, 1);
sem_wait(mutex);

sigaction(SIGUSR1, &act_expire, 0);
sigaction(SIGUSR2, &act_expire, 0);

if (duration != 0)
alarm(duration);

if(id) {
readAllEvents(mqd);
}

for (;;)
{
XEvent event;
Expand All @@ -204,17 +333,17 @@ int main(int argc, char *argv[])
}
}

sem_post(mutex);
sem_close(mutex);

for (int i = 0; i < num_of_lines; i++)
free(lines[i]);
if (used_y_offset == *y_offset) free_y_offset(y_offset_id);
freeLines();

free(lines);
XftDrawDestroy(draw);
XftColorFree(display, visual, colormap, &color);
XftFontClose(display, font);
XCloseDisplay(display);

if(id) {
mq_close(mqd);
}

return exit_code;
}
}