mirror of
https://github.com/NickHu/sway
synced 2024-11-16 19:49:56 +01:00
79369681ab
When swaynag is run with the -l/--detailed-message option, a crash may
occur if the detailed message read from stdin is large enough. E.g.:
swaynag -m hello -l < ~/.config/sway/config
The root cause is that the read_from_stdin() function under-allocates
memory for the destination buffer which causes that buffer to be overflowed
when copying line data to it with snprintf().
The repair is to allocate one more byte for the terminating null byte.
N.B. although getline() returns the number of bytes read excluding a
terminating null byte, the line buffer is terminated with a null byte. Thus
we have a guarantee that the line buffer will be null terminated (which is
important when copying with snprintf()).
387 lines
11 KiB
C
387 lines
11 KiB
C
#define _POSIX_C_SOURCE 200809L
|
|
#include <getopt.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <wordexp.h>
|
|
#include <unistd.h>
|
|
#include "log.h"
|
|
#include "list.h"
|
|
#include "swaynag/swaynag.h"
|
|
#include "swaynag/types.h"
|
|
#include "util.h"
|
|
#include "wlr-layer-shell-unstable-v1-client-protocol.h"
|
|
|
|
static char *read_from_stdin(void) {
|
|
char *buffer = NULL;
|
|
size_t buffer_len = 0;
|
|
char *line = NULL;
|
|
size_t line_size = 0;
|
|
ssize_t nread;
|
|
while ((nread = getline(&line, &line_size, stdin)) != -1) {
|
|
buffer = realloc(buffer, buffer_len + nread + 1);
|
|
snprintf(&buffer[buffer_len], nread + 1, "%s", line);
|
|
buffer_len += nread;
|
|
}
|
|
free(line);
|
|
|
|
while (buffer && buffer[buffer_len - 1] == '\n') {
|
|
buffer[--buffer_len] = '\0';
|
|
}
|
|
|
|
return buffer;
|
|
}
|
|
|
|
int swaynag_parse_options(int argc, char **argv, struct swaynag *swaynag,
|
|
list_t *types, struct swaynag_type *type, char **config, bool *debug) {
|
|
enum type_options {
|
|
TO_COLOR_BACKGROUND = 256,
|
|
TO_COLOR_BORDER,
|
|
TO_COLOR_BORDER_BOTTOM,
|
|
TO_COLOR_BUTTON,
|
|
TO_COLOR_TEXT,
|
|
TO_THICK_BAR_BORDER,
|
|
TO_PADDING_MESSAGE,
|
|
TO_THICK_DET_BORDER,
|
|
TO_THICK_BTN_BORDER,
|
|
TO_GAP_BTN,
|
|
TO_GAP_BTN_DISMISS,
|
|
TO_MARGIN_BTN_RIGHT,
|
|
TO_PADDING_BTN,
|
|
};
|
|
|
|
static struct option opts[] = {
|
|
{"button", required_argument, NULL, 'b'},
|
|
{"button-no-terminal", required_argument, NULL, 'B'},
|
|
{"config", required_argument, NULL, 'c'},
|
|
{"debug", no_argument, NULL, 'd'},
|
|
{"edge", required_argument, NULL, 'e'},
|
|
{"font", required_argument, NULL, 'f'},
|
|
{"help", no_argument, NULL, 'h'},
|
|
{"detailed-message", no_argument, NULL, 'l'},
|
|
{"detailed-button", required_argument, NULL, 'L'},
|
|
{"message", required_argument, NULL, 'm'},
|
|
{"output", required_argument, NULL, 'o'},
|
|
{"dismiss-button", required_argument, NULL, 's'},
|
|
{"type", required_argument, NULL, 't'},
|
|
{"version", no_argument, NULL, 'v'},
|
|
|
|
{"background", required_argument, NULL, TO_COLOR_BACKGROUND},
|
|
{"border", required_argument, NULL, TO_COLOR_BORDER},
|
|
{"border-bottom", required_argument, NULL, TO_COLOR_BORDER_BOTTOM},
|
|
{"button-background", required_argument, NULL, TO_COLOR_BUTTON},
|
|
{"text", required_argument, NULL, TO_COLOR_TEXT},
|
|
{"border-bottom-size", required_argument, NULL, TO_THICK_BAR_BORDER},
|
|
{"message-padding", required_argument, NULL, TO_PADDING_MESSAGE},
|
|
{"details-border-size", required_argument, NULL, TO_THICK_DET_BORDER},
|
|
{"button-border-size", required_argument, NULL, TO_THICK_BTN_BORDER},
|
|
{"button-gap", required_argument, NULL, TO_GAP_BTN},
|
|
{"button-dismiss-gap", required_argument, NULL, TO_GAP_BTN_DISMISS},
|
|
{"button-margin-right", required_argument, NULL, TO_MARGIN_BTN_RIGHT},
|
|
{"button-padding", required_argument, NULL, TO_PADDING_BTN},
|
|
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
const char *usage =
|
|
"Usage: swaynag [options...]\n"
|
|
"\n"
|
|
" -b, --button <text> <action> Create a button with text that "
|
|
"executes action in a terminal when pressed. Multiple buttons can "
|
|
"be defined.\n"
|
|
" -B, --button-no-terminal <text> <action> Like --button, but does"
|
|
"not run the action in a terminal.\n"
|
|
" -c, --config <path> Path to config file.\n"
|
|
" -d, --debug Enable debugging.\n"
|
|
" -e, --edge top|bottom Set the edge to use.\n"
|
|
" -f, --font <font> Set the font to use.\n"
|
|
" -h, --help Show help message and quit.\n"
|
|
" -l, --detailed-message Read a detailed message from stdin.\n"
|
|
" -L, --detailed-button <text> Set the text of the detail button.\n"
|
|
" -m, --message <msg> Set the message text.\n"
|
|
" -o, --output <output> Set the output to use.\n"
|
|
" -s, --dismiss-button <text> Set the dismiss button text.\n"
|
|
" -t, --type <type> Set the message type.\n"
|
|
" -v, --version Show the version number and quit.\n"
|
|
"\n"
|
|
"The following appearance options can also be given:\n"
|
|
" --background RRGGBB[AA] Background color.\n"
|
|
" --border RRGGBB[AA] Border color.\n"
|
|
" --border-bottom RRGGBB[AA] Bottom border color.\n"
|
|
" --button-background RRGGBB[AA] Button background color.\n"
|
|
" --text RRGGBB[AA] Text color.\n"
|
|
" --border-bottom-size size Thickness of the bar border.\n"
|
|
" --message-padding padding Padding for the message.\n"
|
|
" --details-border-size size Thickness for the details border.\n"
|
|
" --button-border-size size Thickness for the button border.\n"
|
|
" --button-gap gap Size of the gap between buttons\n"
|
|
" --button-dismiss-gap gap Size of the gap for dismiss button.\n"
|
|
" --button-margin-right margin Margin from dismiss button to edge.\n"
|
|
" --button-padding padding Padding for the button text.\n";
|
|
|
|
optind = 1;
|
|
while (1) {
|
|
int c = getopt_long(argc, argv, "b:B:c:de:f:hlL:m:o:s:t:v", opts, NULL);
|
|
if (c == -1) {
|
|
break;
|
|
}
|
|
switch (c) {
|
|
case 'b': // Button
|
|
case 'B': // Button (No Terminal)
|
|
if (swaynag) {
|
|
if (optind >= argc) {
|
|
fprintf(stderr, "Missing action for button %s\n", optarg);
|
|
return EXIT_FAILURE;
|
|
}
|
|
struct swaynag_button *button;
|
|
button = calloc(sizeof(struct swaynag_button), 1);
|
|
button->text = strdup(optarg);
|
|
button->type = SWAYNAG_ACTION_COMMAND;
|
|
button->action = strdup(argv[optind]);
|
|
button->terminal = c == 'b';
|
|
list_add(swaynag->buttons, button);
|
|
}
|
|
optind++;
|
|
break;
|
|
case 'c': // Config
|
|
if (config) {
|
|
*config = strdup(optarg);
|
|
}
|
|
break;
|
|
case 'd': // Debug
|
|
if (debug) {
|
|
*debug = true;
|
|
}
|
|
break;
|
|
case 'e': // Edge
|
|
if (type) {
|
|
if (strcmp(optarg, "top") == 0) {
|
|
type->anchors = ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP
|
|
| ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT
|
|
| ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT;
|
|
} else if (strcmp(optarg, "bottom") == 0) {
|
|
type->anchors = ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM
|
|
| ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT
|
|
| ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT;
|
|
} else {
|
|
fprintf(stderr, "Invalid edge: %s\n", optarg);
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
break;
|
|
case 'f': // Font
|
|
if (type) {
|
|
free(type->font);
|
|
type->font = strdup(optarg);
|
|
}
|
|
break;
|
|
case 'l': // Detailed Message
|
|
if (swaynag) {
|
|
free(swaynag->details.message);
|
|
swaynag->details.message = read_from_stdin();
|
|
swaynag->details.button_up.text = strdup("▲");
|
|
swaynag->details.button_down.text = strdup("▼");
|
|
}
|
|
break;
|
|
case 'L': // Detailed Button Text
|
|
if (swaynag) {
|
|
free(swaynag->details.button_details->text);
|
|
swaynag->details.button_details->text = strdup(optarg);
|
|
}
|
|
break;
|
|
case 'm': // Message
|
|
if (swaynag) {
|
|
free(swaynag->message);
|
|
swaynag->message = strdup(optarg);
|
|
}
|
|
break;
|
|
case 'o': // Output
|
|
if (type) {
|
|
free(type->output);
|
|
type->output = strdup(optarg);
|
|
}
|
|
break;
|
|
case 's': // Dismiss Button Text
|
|
if (swaynag) {
|
|
struct swaynag_button *button_close;
|
|
button_close = swaynag->buttons->items[0];
|
|
free(button_close->text);
|
|
button_close->text = strdup(optarg);
|
|
}
|
|
break;
|
|
case 't': // Type
|
|
if (swaynag) {
|
|
swaynag->type = swaynag_type_get(types, optarg);
|
|
if (!swaynag->type) {
|
|
fprintf(stderr, "Unknown type %s\n", optarg);
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
break;
|
|
case 'v': // Version
|
|
fprintf(stdout, "swaynag version " SWAY_VERSION "\n");
|
|
return -1;
|
|
case TO_COLOR_BACKGROUND: // Background color
|
|
if (type) {
|
|
type->background = parse_color(optarg);
|
|
}
|
|
break;
|
|
case TO_COLOR_BORDER: // Border color
|
|
if (type) {
|
|
type->border = parse_color(optarg);
|
|
}
|
|
break;
|
|
case TO_COLOR_BORDER_BOTTOM: // Bottom border color
|
|
if (type) {
|
|
type->border_bottom = parse_color(optarg);
|
|
}
|
|
break;
|
|
case TO_COLOR_BUTTON: // Button background color
|
|
if (type) {
|
|
type->button_background = parse_color(optarg);
|
|
}
|
|
break;
|
|
case TO_COLOR_TEXT: // Text color
|
|
if (type) {
|
|
type->text = parse_color(optarg);
|
|
}
|
|
break;
|
|
case TO_THICK_BAR_BORDER: // Bottom border thickness
|
|
if (type) {
|
|
type->bar_border_thickness = strtol(optarg, NULL, 0);
|
|
}
|
|
break;
|
|
case TO_PADDING_MESSAGE: // Message padding
|
|
if (type) {
|
|
type->message_padding = strtol(optarg, NULL, 0);
|
|
}
|
|
break;
|
|
case TO_THICK_DET_BORDER: // Details border thickness
|
|
if (type) {
|
|
type->details_border_thickness = strtol(optarg, NULL, 0);
|
|
}
|
|
break;
|
|
case TO_THICK_BTN_BORDER: // Button border thickness
|
|
if (type) {
|
|
type->button_border_thickness = strtol(optarg, NULL, 0);
|
|
}
|
|
break;
|
|
case TO_GAP_BTN: // Gap between buttons
|
|
if (type) {
|
|
type->button_gap = strtol(optarg, NULL, 0);
|
|
}
|
|
break;
|
|
case TO_GAP_BTN_DISMISS: // Gap between dismiss button
|
|
if (type) {
|
|
type->button_gap_close = strtol(optarg, NULL, 0);
|
|
}
|
|
break;
|
|
case TO_MARGIN_BTN_RIGHT: // Margin on the right side of button area
|
|
if (type) {
|
|
type->button_margin_right = strtol(optarg, NULL, 0);
|
|
}
|
|
break;
|
|
case TO_PADDING_BTN: // Padding for the button text
|
|
if (type) {
|
|
type->button_padding = strtol(optarg, NULL, 0);
|
|
}
|
|
break;
|
|
default: // Help or unknown flag
|
|
fprintf(c == 'h' ? stdout : stderr, "%s", usage);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static bool file_exists(const char *path) {
|
|
return path && access(path, R_OK) != -1;
|
|
}
|
|
|
|
char *swaynag_get_config_path(void) {
|
|
static const char *config_paths[] = {
|
|
"$HOME/.swaynag/config",
|
|
"$XDG_CONFIG_HOME/swaynag/config",
|
|
SYSCONFDIR "/swaynag/config",
|
|
};
|
|
|
|
char *config_home = getenv("XDG_CONFIG_HOME");
|
|
if (!config_home || config_home[0] == '\0') {
|
|
config_paths[1] = "$HOME/.config/swaynag/config";
|
|
}
|
|
|
|
wordexp_t p;
|
|
for (size_t i = 0; i < sizeof(config_paths) / sizeof(char *); ++i) {
|
|
if (wordexp(config_paths[i], &p, 0) == 0) {
|
|
char *path = strdup(p.we_wordv[0]);
|
|
wordfree(&p);
|
|
if (file_exists(path)) {
|
|
return path;
|
|
}
|
|
free(path);
|
|
}
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
int swaynag_load_config(char *path, struct swaynag *swaynag, list_t *types) {
|
|
FILE *config = fopen(path, "r");
|
|
if (!config) {
|
|
fprintf(stderr, "Failed to read config. Running without it.\n");
|
|
return 0;
|
|
}
|
|
|
|
struct swaynag_type *type;
|
|
type = calloc(1, sizeof(struct swaynag_type));
|
|
type->name = strdup("<config>");
|
|
list_add(types, type);
|
|
|
|
char *line = NULL;
|
|
size_t line_size = 0;
|
|
ssize_t nread;
|
|
int line_number = 0;
|
|
int result = 0;
|
|
while ((nread = getline(&line, &line_size, config)) != -1) {
|
|
line_number++;
|
|
if (!*line || line[0] == '\n' || line[0] == '#') {
|
|
continue;
|
|
}
|
|
|
|
if (line[nread - 1] == '\n') {
|
|
line[nread - 1] = '\0';
|
|
}
|
|
|
|
if (line[0] == '[') {
|
|
char *close = strchr(line, ']');
|
|
if (!close) {
|
|
fprintf(stderr, "Closing bracket not found on line %d\n",
|
|
line_number);
|
|
result = 1;
|
|
break;
|
|
}
|
|
char *name = calloc(1, close - line);
|
|
strncat(name, line + 1, close - line - 1);
|
|
type = swaynag_type_get(types, name);
|
|
if (!type) {
|
|
type = calloc(1, sizeof(struct swaynag_type));
|
|
type->name = strdup(name);
|
|
list_add(types, type);
|
|
}
|
|
free(name);
|
|
} else {
|
|
char *flag = malloc(sizeof(char) * (nread + 3));
|
|
sprintf(flag, "--%s", line);
|
|
char *argv[] = {"swaynag", flag};
|
|
result = swaynag_parse_options(2, argv, swaynag, types, type,
|
|
NULL, NULL);
|
|
free(flag);
|
|
if (result != 0) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
free(line);
|
|
fclose(config);
|
|
return result;
|
|
}
|