+/*
+ * vim:ts=4:sw=4:expandtab
+ *
+ * i3bar - an xcb-based status- and ws-bar for i3
+ * © 2010-2012 Axel Wagner and contributors (see also: LICENSE)
+ *
+ * ipc.c: Communicating with i3
+ *
+ */
+#include <stdlib.h>
#include <stdio.h>
-#include <ev.h>
+#include <unistd.h>
+#include <stdint.h>
+#include <string.h>
+#include <errno.h>
#include <sys/socket.h>
#include <sys/un.h>
-#include <string.h>
-#include <stdint.h>
-#include <stdlib.h>
-#include <unistd.h>
#include <i3/ipc.h>
+#include <ev.h>
#include "common.h"
-#include "ipc.h"
-struct callback_t {
- void (*callback)(char*, void*);
- void* params;
- struct callback_t* next;
-};
+ev_io *i3_connection;
-struct callback_t* outputs_cb_queue;
-struct callback_t* workspaces_cb_queue;
-
-int get_ipc_fd(const char* socket_path) {
- int sockfd = socket(AF_LOCAL, SOCK_STREAM, 0);
- if (sockfd == -1) {
- printf("ERROR: Could not create Socket!\n");
- exit(EXIT_FAILURE);
- }
-
- struct sockaddr_un addr;
- memset(&addr, 0, sizeof(struct sockaddr_un));
- addr.sun_family = AF_LOCAL;
- strcpy(addr.sun_path, socket_path);
- if (connect(sockfd, (const struct sockaddr*) &addr, sizeof(struct sockaddr_un)) < 0) {
- printf("ERROR: Could not connct to i3\n");
- exit(EXIT_FAILURE);
- }
- return sockfd;
-}
+const char *sock_path;
-void get_outputs_cb(struct ev_loop* loop, ev_io *watcher, int revents) {
+typedef void(*handler_t)(char*);
+/*
+ * Called, when we get a reply to a command from i3.
+ * Since i3 does not give us much feedback on commands, we do not much
+ *
+ */
+void got_command_reply(char *reply) {
+ /* TODO: Error handling for command-replies */
}
-void init_i3(const char* socket_path) {
- int sockfd = get_ipc_fd(socket_path);
-
- struct get_outputs_callback* cb = malloc(sizeof(struct get_outputs_callback));
- cb->callback = callback;
- cb->params = params;
-
- ev_io* get_outputs_write = malloc(sizeof(ev_io));
-
- ev_io_init(get_outputs_write, &get_outputs_write_cb, sockfd, EV_WRITE);
- get_outputs_write->data = (void*) cb;
- ev_io_start(main_loop, get_outputs_write);
+/*
+ * Called, when we get a reply with workspaces-data
+ *
+ */
+void got_workspace_reply(char *reply) {
+ DLOG("Got Workspace-Data!\n");
+ parse_workspaces_json(reply);
+ draw_bars();
+}
- ev_io* get_outputs_read = malloc(sizeof(ev_io));
- ev_io_init(get_outputs_read, &get_outputs_read_cb, sockfd, EV_READ);
- get_outputs_read->data = (void*) cb;
- ev_io_start(main_loop, get_outputs_read);
+/*
+ * Called, when we get a reply for a subscription.
+ * Since i3 does not give us much feedback on commands, we do not much
+ *
+ */
+void got_subscribe_reply(char *reply) {
+ DLOG("Got Subscribe Reply: %s\n", reply);
+ /* TODO: Error handling for subscribe-commands */
}
+/*
+ * Called, when we get a reply with outputs-data
+ *
+ */
+void got_output_reply(char *reply) {
+ DLOG("Parsing Outputs-JSON...\n");
+ parse_outputs_json(reply);
+ DLOG("Reconfiguring Windows...\n");
+ realloc_sl_buffer();
+ reconfig_windows();
+
+ i3_output *o_walk;
+ SLIST_FOREACH(o_walk, outputs, slist) {
+ kick_tray_clients(o_walk);
+ }
+
+ draw_bars();
+}
-void get_outputs_write_cb(struct ev_loop* loop, ev_io *watcher, int revents) {
- ev_io_stop(loop, watcher);
-
- int buffer_size = strlen(I3_IPC_MAGIC) + sizeof(uint32_t) + sizeof(uint32_t);
- char msg[buffer_size];
- char *walk = msg;
- uint32_t msg_size = 0;
- uint32_t msg_type = I3_IPC_MESSAGE_TYPE_GET_OUTPUTS;
- int sockfd = watcher->fd;
-
- strcpy(walk, I3_IPC_MAGIC);
- walk += strlen(I3_IPC_MAGIC);
- memcpy(walk, &msg_size, sizeof(uint32_t));
- walk += sizeof(uint32_t);
- memcpy(walk, &msg_type, sizeof(uint32_t));
-
- int sent_bytes = 0;
- int bytes_to_go = buffer_size;
- while (sent_bytes < bytes_to_go) {
- int n = write(sockfd, msg + sent_bytes, bytes_to_go);
- if (n == -1) {
- printf("ERROR: write() failed!\n");
- exit(EXIT_FAILURE);
- }
-
- sent_bytes += n;
- bytes_to_go -= n;
- }
- FREE(watcher);
+/*
+ * Called when we get the configuration for our bar instance
+ *
+ */
+void got_bar_config(char *reply) {
+ DLOG("Received bar config \"%s\"\n", reply);
+ /* We initiate the main-function by requesting infos about the outputs and
+ * workspaces. Everything else (creating the bars, showing the right workspace-
+ * buttons and more) is taken care of by the event-drivenness of the code */
+ i3_send_msg(I3_IPC_MESSAGE_TYPE_GET_OUTPUTS, NULL);
+ parse_config_json(reply);
+
+ /* Now we can actually use 'config', so let's subscribe to the appropriate
+ * events and request the workspaces if necessary. */
+ subscribe_events();
+ if (!config.disable_ws)
+ i3_send_msg(I3_IPC_MESSAGE_TYPE_GET_WORKSPACES, NULL);
+
+ /* Initialize the rest of XCB */
+ init_xcb_late(config.fontname);
+
+ /* Resolve color strings to colorpixels and save them, then free the strings. */
+ init_colors(&(config.colors));
+ free_colors(&(config.colors));
+
+ /* The name of this function is actually misleading. Even if no command is
+ * specified, this function initiates the watchers to listen on stdin and
+ * react accordingly */
+ start_child(config.command);
+ FREE(config.command);
}
-void get_outputs_read_cb(struct ev_loop* loop, ev_io *watcher, int revents) {
- ev_io_stop(loop, watcher);
-
- int to_read = strlen(I3_IPC_MAGIC) + sizeof(uint32_t) + sizeof(uint32_t);
- char msg[to_read];
- char *walk = msg;
- int sockfd = watcher->fd;
- uint8_t *reply;
- struct get_outputs_callback* cb = watcher->data;
-
- uint32_t reply_length;
-
- uint32_t read_bytes = 0;
- while (read_bytes < to_read) {
- int n = read(sockfd, msg + read_bytes, to_read);
- if (n == -1) {
- printf("ERROR: read() failed!\n");
- exit(EXIT_FAILURE);
- }
- if (n == 0) {
- printf("ERROR: No reply!\n");
- exit(EXIT_FAILURE);
- }
-
- read_bytes += n;
- to_read -= n;
- }
-
- if (memcmp(walk, I3_IPC_MAGIC, strlen(I3_IPC_MAGIC)) != 0) {
- printf("ERROR: Wrong magic!\n");
- exit(EXIT_FAILURE);
- }
-
- walk += strlen(I3_IPC_MAGIC);
- reply_length = *((uint32_t*) walk);
- walk += sizeof(uint32_t);
- if (*((uint32_t*) walk) != I3_IPC_MESSAGE_TYPE_GET_OUTPUTS) {
- printf("ERROR: Wrong reply type (%d) expected %d!\n",
- *((uint32_t*) walk),
- I3_IPC_MESSAGE_TYPE_GET_OUTPUTS);
- exit(EXIT_FAILURE);
- }
- walk += sizeof(uint32_t);
-
- reply = malloc(reply_length);
- if (reply == NULL) {
- printf("ERROR: malloc() failed!\n");
- exit(EXIT_FAILURE);
- }
-
- to_read = reply_length;
- read_bytes = 0;
- while (read_bytes < to_read) {
- int n = read(sockfd, reply + read_bytes, to_read);
- if (n == -1) {
- printf("ERROR: read() failed!\n");
- exit(EXIT_FAILURE);
- }
-
- read_bytes += n;
- to_read -= n;
- }
-
- cb->callback((char*) reply, cb->params);
- FREE(cb);
- FREE(watcher);
+/* Data-structure to easily call the reply-handlers later */
+handler_t reply_handlers[] = {
+ &got_command_reply,
+ &got_workspace_reply,
+ &got_subscribe_reply,
+ &got_output_reply,
+ NULL,
+ NULL,
+ &got_bar_config,
+};
+
+/*
+ * Called, when a workspace-event arrives (i.e. the user changed the workspace)
+ *
+ */
+void got_workspace_event(char *event) {
+ DLOG("Got Workspace Event!\n");
+ i3_send_msg(I3_IPC_MESSAGE_TYPE_GET_WORKSPACES, NULL);
}
-void get_outputs_json(void (*callback)(char*, void*), void* params) {
+/*
+ * Called, when an output-event arrives (i.e. the screen-configuration changed)
+ *
+ */
+void got_output_event(char *event) {
+ DLOG("Got Output Event!\n");
+ i3_send_msg(I3_IPC_MESSAGE_TYPE_GET_OUTPUTS, NULL);
+ if (!config.disable_ws) {
+ i3_send_msg(I3_IPC_MESSAGE_TYPE_GET_WORKSPACES, NULL);
+ }
}
+/* Data-structure to easily call the reply-handlers later */
+handler_t event_handlers[] = {
+ &got_workspace_event,
+ &got_output_event
+};
+/*
+ * Called, when we get a message from i3
+ *
+ */
+void got_data(struct ev_loop *loop, ev_io *watcher, int events) {
+ DLOG("Got data!\n");
+ int fd = watcher->fd;
+
+ /* First we only read the header, because we know its length */
+ uint32_t header_len = strlen(I3_IPC_MAGIC) + sizeof(uint32_t)*2;
+ char *header = smalloc(header_len);
+
+ /* We first parse the fixed-length IPC-header, to know, how much data
+ * we have to expect */
+ uint32_t rec = 0;
+ while (rec < header_len) {
+ int n = read(fd, header + rec, header_len - rec);
+ if (n == -1) {
+ ELOG("read() failed: %s\n", strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+ if (n == 0) {
+ /* EOF received. Since i3 will restart i3bar instances as appropriate,
+ * we exit here. */
+ DLOG("EOF received, exiting...\n");
+ clean_xcb();
+ exit(EXIT_SUCCESS);
+ }
+ rec += n;
+ }
+
+ if (strncmp(header, I3_IPC_MAGIC, strlen(I3_IPC_MAGIC))) {
+ ELOG("Wrong magic code: %.*s\n Expected: %s\n",
+ (int) strlen(I3_IPC_MAGIC),
+ header,
+ I3_IPC_MAGIC);
+ exit(EXIT_FAILURE);
+ }
+
+ char *walk = header + strlen(I3_IPC_MAGIC);
+ uint32_t size;
+ memcpy(&size, (uint32_t*)walk, sizeof(uint32_t));
+ walk += sizeof(uint32_t);
+ uint32_t type;
+ memcpy(&type, (uint32_t*)walk, sizeof(uint32_t));
+
+ /* Now that we know, what to expect, we can start read()ing the rest
+ * of the message */
+ char *buffer = smalloc(size + 1);
+ rec = 0;
+
+ while (rec < size) {
+ int n = read(fd, buffer + rec, size - rec);
+ if (n == -1) {
+ ELOG("read() failed: %s\n", strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+ if (n == 0) {
+ ELOG("Nothing to read!\n");
+ exit(EXIT_FAILURE);
+ }
+ rec += n;
+ }
+ buffer[size] = '\0';
+
+ /* And call the callback (indexed by the type) */
+ if (type & (1 << 31)) {
+ type ^= 1 << 31;
+ event_handlers[type](buffer);
+ } else {
+ if (reply_handlers[type])
+ reply_handlers[type](buffer);
+ }
+
+ FREE(header);
+ FREE(buffer);
+}
-struct get_workspaces_callback {
- void (*callback)(char*, void*);
- void* params;
-};
+/*
+ * Sends a Message to i3.
+ * type must be a valid I3_IPC_MESSAGE_TYPE (see i3/ipc.h for further information)
+ *
+ */
+int i3_send_msg(uint32_t type, const char *payload) {
+ uint32_t len = 0;
+ if (payload != NULL) {
+ len = strlen(payload);
+ }
+
+ /* We are a wellbehaved client and send a proper header first */
+ uint32_t to_write = strlen (I3_IPC_MAGIC) + sizeof(uint32_t)*2 + len;
+ /* TODO: I'm not entirely sure if this buffer really has to contain more
+ * than the pure header (why not just write() the payload from *payload?),
+ * but we leave it for now */
+ char *buffer = smalloc(to_write);
+ char *walk = buffer;
+
+ strncpy(buffer, I3_IPC_MAGIC, strlen(I3_IPC_MAGIC));
+ walk += strlen(I3_IPC_MAGIC);
+ memcpy(walk, &len, sizeof(uint32_t));
+ walk += sizeof(uint32_t);
+ memcpy(walk, &type, sizeof(uint32_t));
+ walk += sizeof(uint32_t);
+
+ if (payload != NULL)
+ strncpy(walk, payload, len);
+
+ uint32_t written = 0;
+
+ while (to_write > 0) {
+ int n = write(i3_connection->fd, buffer + written, to_write);
+ if (n == -1) {
+ ELOG("write() failed: %s\n", strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ to_write -= n;
+ written += n;
+ }
+
+ FREE(buffer);
+
+ return 1;
+}
-void get_workspaces_write_cb(struct ev_loop* loop, ev_io *watcher, int revents) {
- ev_io_stop(loop, watcher);
- //FREE(watcher);
-
- int buffer_size = strlen(I3_IPC_MAGIC) + sizeof(uint32_t) + sizeof(uint32_t);
- char msg[buffer_size];
- char *walk = msg;
- uint32_t msg_size = 0;
- uint32_t msg_type = I3_IPC_MESSAGE_TYPE_GET_WORKSPACES;
- int sockfd = watcher->fd;
-
- strcpy(walk, I3_IPC_MAGIC);
- walk += strlen(I3_IPC_MAGIC);
- memcpy(walk, &msg_size, sizeof(uint32_t));
- walk += sizeof(uint32_t);
- memcpy(walk, &msg_type, sizeof(uint32_t));
-
- int sent_bytes = 0;
- int bytes_to_go = buffer_size;
- while (sent_bytes < bytes_to_go) {
- int n = write(sockfd, msg + sent_bytes, bytes_to_go);
- if (n == -1) {
- printf("ERROR: write() failed!\n");
- exit(EXIT_FAILURE);
- }
-
- sent_bytes += n;
- bytes_to_go -= n;
- }
- FREE(watcher);
+/*
+ * Initiate a connection to i3.
+ * socket-path must be a valid path to the ipc_socket of i3
+ *
+ */
+int init_connection(const char *socket_path) {
+ sock_path = socket_path;
+ int sockfd = ipc_connect(socket_path);
+ i3_connection = smalloc(sizeof(ev_io));
+ ev_io_init(i3_connection, &got_data, sockfd, EV_READ);
+ ev_io_start(main_loop, i3_connection);
+ return 1;
}
-void get_workspaces_read_cb(struct ev_loop* loop, ev_io *watcher, int revents) {
- ev_io_stop(loop, watcher);
- //FREE(watcher);
-
- int to_read = strlen(I3_IPC_MAGIC) + sizeof(uint32_t) + sizeof(uint32_t);
- char msg[to_read];
- char *walk = msg;
- int sockfd = watcher->fd;
- uint8_t *reply;
- struct get_workspaces_callback* cb = watcher->data;
-
- uint32_t reply_length;
-
- uint32_t read_bytes = 0;
- while (read_bytes < to_read) {
- int n = read(sockfd, msg + read_bytes, to_read);
- if (n == -1) {
- printf("ERROR: read() failed!\n");
- exit(EXIT_FAILURE);
- }
- if (n == 0) {
- printf("ERROR: No reply!\n");
- exit(EXIT_FAILURE);
- }
-
- read_bytes += n;
- to_read -= n;
- }
-
- if (memcmp(walk, I3_IPC_MAGIC, strlen(I3_IPC_MAGIC)) != 0) {
- printf("ERROR: Wrong magic!\n");
- exit(EXIT_FAILURE);
- }
-
- walk += strlen(I3_IPC_MAGIC);
- reply_length = *((uint32_t*) walk);
- walk += sizeof(uint32_t);
- if (*((uint32_t*) walk) != I3_IPC_MESSAGE_TYPE_GET_WORKSPACES) {
- printf("ERROR: Wrong reply type (%d) expected %d!\n",
- *((uint32_t*) walk),
- I3_IPC_MESSAGE_TYPE_GET_WORKSPACES);
- exit(EXIT_FAILURE);
- }
- walk += sizeof(uint32_t);
-
- reply = malloc(reply_length);
- if (reply == NULL) {
- printf("ERROR: malloc() failed!\n");
- exit(EXIT_FAILURE);
- }
-
- to_read = reply_length;
- read_bytes = 0;
- while (read_bytes < to_read) {
- int n = read(sockfd, reply + read_bytes, to_read);
- if (n == -1) {
- printf("ERROR: read() failed!\n");
- exit(EXIT_FAILURE);
- }
-
- read_bytes += n;
- to_read -= n;
- }
-
- cb->callback((char*) reply, cb->params);
- FREE(cb);
-
- FREE(watcher);
+/*
+ * Destroy the connection to i3.
+ */
+void destroy_connection(void) {
+ close(i3_connection->fd);
+ ev_io_stop(main_loop, i3_connection);
}
-void get_workspaces_json(void (*callback)(char*, void*), void* params) {
- socket_path = "/home/mero/.i3/ipc.sock";
-
- int sockfd = socket(AF_LOCAL, SOCK_STREAM, 0);
- if (sockfd == -1) {
- printf("ERROR: Could not create Socket!\n");
- exit(EXIT_FAILURE);
- }
-
- struct sockaddr_un addr;
- memset(&addr, 0, sizeof(struct sockaddr_un));
- addr.sun_family = AF_LOCAL;
- strcpy(addr.sun_path, socket_path);
- if (connect(sockfd, (const struct sockaddr*) &addr, sizeof(struct sockaddr_un)) < 0) {
- printf("ERROR: Could not connct to i3\n");
- exit(EXIT_FAILURE);
- }
-
- struct get_workspaces_callback* cb = malloc(sizeof(struct get_workspaces_callback));
- cb->callback = callback;
- cb->params = params;
-
- ev_io* get_workspaces_write = malloc(sizeof(ev_io));
-
- ev_io_init(get_workspaces_write, &get_workspaces_write_cb, sockfd, EV_WRITE);
- get_workspaces_write->data = (void*) cb;
- ev_io_start(main_loop, get_workspaces_write);
-
- ev_io* get_workspaces_read = malloc(sizeof(ev_io));
- ev_io_init(get_workspaces_read, &get_workspaces_read_cb, sockfd, EV_READ);
- get_workspaces_read->data = (void*) cb;
- ev_io_start(main_loop, get_workspaces_read);
+/*
+ * Subscribe to all the i3-events, we need
+ *
+ */
+void subscribe_events(void) {
+ if (config.disable_ws) {
+ i3_send_msg(I3_IPC_MESSAGE_TYPE_SUBSCRIBE, "[ \"output\" ]");
+ } else {
+ i3_send_msg(I3_IPC_MESSAGE_TYPE_SUBSCRIBE, "[ \"workspace\", \"output\" ]");
+ }
}