diff options
Diffstat (limited to 'swaybar/status_line.c')
-rw-r--r-- | swaybar/status_line.c | 42 |
1 files changed, 29 insertions, 13 deletions
diff --git a/swaybar/status_line.c b/swaybar/status_line.c index 48b43248..ed6dc7c8 100644 --- a/swaybar/status_line.c +++ b/swaybar/status_line.c | |||
@@ -1,12 +1,15 @@ | |||
1 | #define _POSIX_C_SOURCE 200809L | 1 | #define _POSIX_C_SOURCE 200809L |
2 | #include <fcntl.h> | 2 | #include <fcntl.h> |
3 | #include <sys/ioctl.h> | ||
3 | #include <json-c/json.h> | 4 | #include <json-c/json.h> |
4 | #include <stdlib.h> | 5 | #include <stdlib.h> |
5 | #include <string.h> | 6 | #include <string.h> |
6 | #include <stdio.h> | 7 | #include <stdio.h> |
7 | #include <unistd.h> | 8 | #include <unistd.h> |
8 | #include <wlr/util/log.h> | 9 | #include <wlr/util/log.h> |
10 | #include "swaybar/bar.h" | ||
9 | #include "swaybar/config.h" | 11 | #include "swaybar/config.h" |
12 | #include "swaybar/i3bar.h" | ||
10 | #include "swaybar/event_loop.h" | 13 | #include "swaybar/event_loop.h" |
11 | #include "swaybar/status_line.h" | 14 | #include "swaybar/status_line.h" |
12 | #include "readline.h" | 15 | #include "readline.h" |
@@ -34,18 +37,35 @@ bool status_handle_readable(struct status_line *status) { | |||
34 | switch (status->protocol) { | 37 | switch (status->protocol) { |
35 | case PROTOCOL_UNDEF: | 38 | case PROTOCOL_UNDEF: |
36 | errno = 0; | 39 | errno = 0; |
37 | read_bytes = getline(&status->buffer, | 40 | int available_bytes; |
38 | &status->buffer_size, status->read); | 41 | if (ioctl(status->read_fd, FIONREAD, &available_bytes) == -1) { |
39 | if (errno == EAGAIN) { | 42 | wlr_log(WLR_ERROR, "Unable to read status command output size"); |
40 | clearerr(status->read); | ||
41 | } else if (errno) { | ||
42 | status_error(status, "[error reading from status command]"); | 43 | status_error(status, "[error reading from status command]"); |
43 | return true; | 44 | return true; |
44 | } | 45 | } |
45 | 46 | ||
47 | if ((size_t)available_bytes + 1 > status->buffer_size) { | ||
48 | // need room for leading '\0' too | ||
49 | status->buffer_size = available_bytes + 1; | ||
50 | status->buffer = realloc(status->buffer, status->buffer_size); | ||
51 | } | ||
52 | if (status->buffer == NULL) { | ||
53 | wlr_log_errno(WLR_ERROR, "Unable to read status line"); | ||
54 | status_error(status, "[error reading from status command]"); | ||
55 | return true; | ||
56 | } | ||
57 | |||
58 | read_bytes = read(status->read_fd, status->buffer, available_bytes); | ||
59 | if (read_bytes != available_bytes) { | ||
60 | status_error(status, "[error reading from status command]"); | ||
61 | return true; | ||
62 | } | ||
63 | status->buffer[available_bytes] = 0; | ||
64 | |||
46 | // the header must be sent completely the first time round | 65 | // the header must be sent completely the first time round |
66 | char *newline = strchr(status->buffer, '\n'); | ||
47 | json_object *header, *version; | 67 | json_object *header, *version; |
48 | if (status->buffer[read_bytes - 1] == '\n' | 68 | if (newline != NULL |
49 | && (header = json_tokener_parse(status->buffer)) | 69 | && (header = json_tokener_parse(status->buffer)) |
50 | && json_object_object_get_ex(header, "version", &version) | 70 | && json_object_object_get_ex(header, "version", &version) |
51 | && json_object_get_int(version) == 1) { | 71 | && json_object_get_int(version) == 1) { |
@@ -67,13 +87,9 @@ bool status_handle_readable(struct status_line *status) { | |||
67 | 87 | ||
68 | wl_list_init(&status->blocks); | 88 | wl_list_init(&status->blocks); |
69 | status->tokener = json_tokener_new(); | 89 | status->tokener = json_tokener_new(); |
70 | read_bytes = getdelim(&status->buffer, &status->buffer_size, EOF, status->read); | 90 | status->buffer_index = strlen(newline + 1); |
71 | if (read_bytes > 0) { | 91 | memmove(status->buffer, newline + 1, status->buffer_index + 1); |
72 | status->buffer_index = read_bytes; | 92 | return i3bar_handle_readable(status); |
73 | return i3bar_handle_readable(status); | ||
74 | } else { | ||
75 | return false; | ||
76 | } | ||
77 | } | 93 | } |
78 | 94 | ||
79 | wlr_log(WLR_DEBUG, "Using text protocol."); | 95 | wlr_log(WLR_DEBUG, "Using text protocol."); |