1/*
  2 * (C)opyright MMVI Anselm R. Garbe <garbeam at gmail dot com>
  3 * See LICENSE file for license details.
  4 */
  5
  6#include <stdarg.h>
  7#include <stdio.h>
  8#include <stdlib.h>
  9#include <string.h>
 10#include <sys/types.h>
 11#include <sys/wait.h>
 12#include <unistd.h>
 13#include <X11/Xatom.h>
 14
 15#include "util.h"
 16
 17void
 18error(char *errstr, ...) {
 19	va_list ap;
 20	va_start(ap, errstr);
 21	vfprintf(stderr, errstr, ap);
 22	va_end(ap);
 23	exit(1);
 24}
 25
 26static void
 27bad_malloc(unsigned int size)
 28{
 29	fprintf(stderr, "fatal: could not malloc() %d bytes\n",
 30			(int) size);
 31	exit(1);
 32}
 33
 34void *
 35emallocz(unsigned int size)
 36{
 37	void *res = calloc(1, size);
 38	if(!res)
 39		bad_malloc(size);
 40	return res;
 41}
 42
 43void *
 44emalloc(unsigned int size)
 45{
 46	void *res = malloc(size);
 47	if(!res)
 48		bad_malloc(size);
 49	return res;
 50}
 51
 52void *
 53erealloc(void *ptr, unsigned int size)
 54{
 55	void *res = realloc(ptr, size);
 56	if(!res)
 57		bad_malloc(size);
 58	return res;
 59}
 60
 61char *
 62estrdup(const char *str)
 63{
 64	void *res = strdup(str);
 65	if(!res)
 66		bad_malloc(strlen(str));
 67	return res;
 68}
 69
 70void
 71failed_assert(char *a, char *file, int line)
 72{
 73	fprintf(stderr, "Assertion \"%s\" failed at %s:%d\n", a, file, line);
 74	abort();
 75}
 76
 77void
 78swap(void **p1, void **p2)
 79{
 80	void *tmp = *p1;
 81	*p1 = *p2;
 82	*p2 = tmp;
 83}
 84
 85void
 86spawn(Display *dpy, char *argv[])
 87{
 88	if(!argv || !argv[0])
 89		return;
 90	if(fork() == 0) {
 91		if(fork() == 0) {
 92			if(dpy)
 93				close(ConnectionNumber(dpy));
 94			setsid();
 95			execvp(argv[0], argv);
 96			fprintf(stderr, "gridwm: execvp %s", argv[0]);
 97			perror(" failed");
 98		}
 99		exit (0);
100	}
101	wait(0);
102}
103
104void
105pipe_spawn(char *buf, unsigned int len, Display *dpy, char *argv[])
106{
107	unsigned int l, n;
108	int pfd[2];
109
110	if(!argv || !argv[0])
111		return;
112
113	if(pipe(pfd) == -1) {
114		perror("pipe");
115		exit(1);
116	}
117
118	if(fork() == 0) {
119		if(dpy)
120			close(ConnectionNumber(dpy));
121		setsid();
122		dup2(pfd[1], STDOUT_FILENO);
123		close(pfd[0]);
124		close(pfd[1]);
125		execvp(argv[0], argv);
126		fprintf(stderr, "gridwm: execvp %s", argv[0]);
127		perror(" failed");
128	}
129	else {
130		n = 0;
131		close(pfd[1]);
132		while(l > n) {
133			if((l = read(pfd[0], buf + n, len - n)) < 1)
134				break;
135			n += l;
136		}
137		close(pfd[0]);
138		buf[n - 1] = 0;
139	}
140	wait(0);
141}
142
143
144unsigned char *
145getselection(unsigned long offset, unsigned long *len, unsigned long *remain)
146{
147	Display *dpy;
148	Atom xa_clip_string;
149	Window w;
150	XEvent ev;
151	Atom typeret;
152	int format;
153	unsigned char *data;
154	unsigned char *result = NULL;
155
156	dpy = XOpenDisplay(0);
157	if(!dpy)
158		return NULL;
159	xa_clip_string = XInternAtom(dpy, "_SEL_STRING", False);
160	w = XCreateSimpleWindow(dpy, DefaultRootWindow(dpy), 10, 10, 200, 200,
161			1, CopyFromParent, CopyFromParent);
162	XConvertSelection(dpy, XA_PRIMARY, XA_STRING, xa_clip_string,
163			w, CurrentTime);
164	XFlush(dpy);
165	XNextEvent(dpy, &ev);
166	if(ev.type == SelectionNotify && ev.xselection.property != None) {
167		XGetWindowProperty(dpy, w, ev.xselection.property, offset, 4096L, False,
168				AnyPropertyType, &typeret, &format, len, remain, &data);
169		if(*len) {
170			result = emalloc(sizeof(unsigned char) * *len);
171			memcpy(result, data, *len);
172		}
173		XDeleteProperty(dpy, w, ev.xselection.property);
174	}
175	XDestroyWindow(dpy, w);
176	XCloseDisplay(dpy);
177	return result;
178}