Merge pull request #995 from SirCmpwn/memory-use

Handle allocation failures
This commit is contained in:
Drew DeVault 2016-12-17 13:23:44 -05:00 committed by GitHub
commit f04ee0e68d
30 changed files with 399 additions and 101 deletions

View file

@ -52,10 +52,18 @@ struct ipc_response *ipc_recv_response(int socketfd) {
} }
struct ipc_response *response = malloc(sizeof(struct ipc_response)); struct ipc_response *response = malloc(sizeof(struct ipc_response));
if (!response) {
goto error_1;
}
total = 0; total = 0;
response->size = data32[0]; response->size = data32[0];
response->type = data32[1]; response->type = data32[1];
char *payload = malloc(response->size + 1); char *payload = malloc(response->size + 1);
if (!payload) {
goto error_2;
}
while (total < response->size) { while (total < response->size) {
ssize_t received = recv(socketfd, payload + total, response->size - total, 0); ssize_t received = recv(socketfd, payload + total, response->size - total, 0);
if (received < 0) { if (received < 0) {
@ -67,6 +75,11 @@ struct ipc_response *ipc_recv_response(int socketfd) {
response->payload = payload; response->payload = payload;
return response; return response;
error_2:
free(response);
error_1:
sway_log(L_ERROR, "Unable to allocate memory for IPC response");
return NULL;
} }
void free_ipc_response(struct ipc_response *response) { void free_ipc_response(struct ipc_response *response) {

View file

@ -5,6 +5,9 @@
list_t *create_list(void) { list_t *create_list(void) {
list_t *list = malloc(sizeof(list_t)); list_t *list = malloc(sizeof(list_t));
if (!list) {
return NULL;
}
list->capacity = 10; list->capacity = 10;
list->length = 0; list->length = 0;
list->items = malloc(sizeof(void*) * list->capacity); list->items = malloc(sizeof(void*) * list->capacity);

View file

@ -88,9 +88,14 @@ void _sway_log(const char *filename, int line, log_importance_t verbosity, const
} }
if (filename && line) { if (filename && line) {
char *file = strdup(filename); const char *file = filename + strlen(filename);
fprintf(stderr, "[%s:%d] ", basename(file), line); while (file != filename && *file != '/') {
free(file); --file;
}
if (*file == '/') {
++file;
}
fprintf(stderr, "[%s:%d] ", file, line);
} }
va_list args; va_list args;

View file

@ -1,4 +1,5 @@
#include "readline.h" #include "readline.h"
#include "log.h"
#include <stdlib.h> #include <stdlib.h>
#include <stdio.h> #include <stdio.h>
@ -7,6 +8,7 @@ char *read_line(FILE *file) {
char *string = malloc(size); char *string = malloc(size);
char lastChar = '\0'; char lastChar = '\0';
if (!string) { if (!string) {
sway_log(L_ERROR, "Unable to allocate memory for read_line");
return NULL; return NULL;
} }
while (1) { while (1) {
@ -27,6 +29,7 @@ char *read_line(FILE *file) {
char *new_string = realloc(string, size *= 2); char *new_string = realloc(string, size *= 2);
if (!new_string) { if (!new_string) {
free(string); free(string);
sway_log(L_ERROR, "Unable to allocate memory for read_line");
return NULL; return NULL;
} }
string = new_string; string = new_string;

View file

@ -25,13 +25,8 @@ bool _sway_assert(bool condition, const char* format, ...) __attribute__((format
void _sway_log(const char *filename, int line, log_importance_t verbosity, const char* format, ...) __attribute__((format(printf,4,5))); void _sway_log(const char *filename, int line, log_importance_t verbosity, const char* format, ...) __attribute__((format(printf,4,5)));
#ifndef NDEBUG
#define sway_log(VERBOSITY, FMT, ...) \ #define sway_log(VERBOSITY, FMT, ...) \
_sway_log(__FILE__, __LINE__, VERBOSITY, FMT, ##__VA_ARGS__) _sway_log(__FILE__, __LINE__, VERBOSITY, FMT, ##__VA_ARGS__)
#else
#define sway_log(VERBOSITY, FMT, ...) \
_sway_log(NULL, 0, VERBOSITY, FMT, ##__VA_ARGS__)
#endif
void error_handler(int sig); void error_handler(int sig);

View file

@ -29,20 +29,24 @@ void border_clear(struct border *border) {
static cairo_t *create_border_buffer(swayc_t *view, struct wlc_geometry g, cairo_surface_t **surface) { static cairo_t *create_border_buffer(swayc_t *view, struct wlc_geometry g, cairo_surface_t **surface) {
if (view->border == NULL) { if (view->border == NULL) {
view->border = malloc(sizeof(struct border)); view->border = malloc(sizeof(struct border));
if (!view->border) {
sway_log(L_ERROR, "Unable to allocate window border information");
return NULL;
}
} }
cairo_t *cr; cairo_t *cr;
int stride = cairo_format_stride_for_width(CAIRO_FORMAT_ARGB32, g.size.w); int stride = cairo_format_stride_for_width(CAIRO_FORMAT_ARGB32, g.size.w);
view->border->buffer = calloc(stride * g.size.h, sizeof(unsigned char)); view->border->buffer = calloc(stride * g.size.h, sizeof(unsigned char));
view->border->geometry = g; view->border->geometry = g;
if (!view->border->buffer) { if (!view->border->buffer) {
sway_log(L_DEBUG, "Unable to allocate buffer"); sway_log(L_ERROR, "Unable to allocate window border buffer");
return NULL; return NULL;
} }
*surface = cairo_image_surface_create_for_data(view->border->buffer, *surface = cairo_image_surface_create_for_data(view->border->buffer,
CAIRO_FORMAT_ARGB32, g.size.w, g.size.h, stride); CAIRO_FORMAT_ARGB32, g.size.w, g.size.h, stride);
if (cairo_surface_status(*surface) != CAIRO_STATUS_SUCCESS) { if (cairo_surface_status(*surface) != CAIRO_STATUS_SUCCESS) {
border_clear(view->border); border_clear(view->border);
sway_log(L_DEBUG, "Unable to allocate surface"); sway_log(L_ERROR, "Unable to allocate window border surface");
return NULL; return NULL;
} }
cr = cairo_create(*surface); cr = cairo_create(*surface);
@ -50,7 +54,7 @@ static cairo_t *create_border_buffer(swayc_t *view, struct wlc_geometry g, cairo
if (cairo_status(cr) != CAIRO_STATUS_SUCCESS) { if (cairo_status(cr) != CAIRO_STATUS_SUCCESS) {
cairo_surface_destroy(*surface); cairo_surface_destroy(*surface);
border_clear(view->border); border_clear(view->border);
sway_log(L_DEBUG, "Unable to create cairo context"); sway_log(L_ERROR, "Unable to create cairo context");
return NULL; return NULL;
} }
return cr; return cr;
@ -238,6 +242,10 @@ static char *generate_container_title(swayc_t *container) {
} }
int len = 9; int len = 9;
name = malloc(len * sizeof(char)); name = malloc(len * sizeof(char));
if (!name) {
sway_log(L_ERROR, "Unable to allocate container title");
return NULL;
}
snprintf(name, len, "sway: %c[", layout); snprintf(name, len, "sway: %c[", layout);
int i; int i;
@ -257,6 +265,11 @@ static char *generate_container_title(swayc_t *container) {
} }
name = malloc(len * sizeof(char)); name = malloc(len * sizeof(char));
if (!name) {
free(prev_name);
sway_log(L_ERROR, "Unable to allocate container title");
return NULL;
}
if (i < container->children->length-1) { if (i < container->children->length-1) {
snprintf(name, len, "%s%s ", prev_name, title); snprintf(name, len, "%s%s ", prev_name, title);
} else { } else {
@ -268,6 +281,11 @@ static char *generate_container_title(swayc_t *container) {
prev_name = name; prev_name = name;
len = strlen(name) + 2; len = strlen(name) + 2;
name = malloc(len * sizeof(char)); name = malloc(len * sizeof(char));
if (!name) {
free(prev_name);
sway_log(L_ERROR, "Unable to allocate container title");
return NULL;
}
snprintf(name, len, "%s]", prev_name); snprintf(name, len, "%s]", prev_name);
free(prev_name); free(prev_name);
free(container->name); free(container->name);
@ -347,6 +365,9 @@ static void update_view_border(swayc_t *view) {
} }
}; };
cr = create_border_buffer(view, g, &surface); cr = create_border_buffer(view, g, &surface);
if (!cr) {
goto cleanup;
}
bool render_top = !should_hide_top_border(view, view->y); bool render_top = !should_hide_top_border(view, view->y);
if (view == focused || is_child_of_focused) { if (view == focused || is_child_of_focused) {
@ -408,6 +429,8 @@ static void update_view_border(swayc_t *view) {
} }
} }
cleanup:
if (surface) { if (surface) {
cairo_surface_flush(surface); cairo_surface_flush(surface);
cairo_surface_destroy(surface); cairo_surface_destroy(surface);

View file

@ -121,6 +121,9 @@ void input_cmd_apply(struct input_config *input) {
for (int i = 0; i < input_devices->length; ++i) { for (int i = 0; i < input_devices->length; ++i) {
device = input_devices->items[i]; device = input_devices->items[i];
char* dev_identifier = libinput_dev_unique_id(device); char* dev_identifier = libinput_dev_unique_id(device);
if (!dev_identifier) {
break;
}
int match = dev_identifier && strcmp(dev_identifier, input->identifier) == 0; int match = dev_identifier && strcmp(dev_identifier, input->identifier) == 0;
free(dev_identifier); free(dev_identifier);
if (match) { if (match) {
@ -386,7 +389,11 @@ struct cmd_results *handle_command(char *_exec, enum command_context context) {
if (!results) { if (!results) {
int len = strlen(criteria) + strlen(head) + 4; int len = strlen(criteria) + strlen(head) + 4;
char *tmp = malloc(len); char *tmp = malloc(len);
snprintf(tmp, len, "[%s] %s", criteria, head); if (tmp) {
snprintf(tmp, len, "[%s] %s", criteria, head);
} else {
sway_log(L_DEBUG, "Unable to allocate criteria string for cmd result");
}
results = cmd_results_new(CMD_INVALID, tmp, results = cmd_results_new(CMD_INVALID, tmp,
"Can't handle criteria string: Refusing to execute command"); "Can't handle criteria string: Refusing to execute command");
free(tmp); free(tmp);
@ -568,6 +575,9 @@ struct cmd_results *config_commands_command(char *exec) {
} }
if (!policy) { if (!policy) {
policy = alloc_command_policy(cmd); policy = alloc_command_policy(cmd);
if (!policy) {
sway_abort("Unable to allocate security policy");
}
list_add(config->command_policies, policy); list_add(config->command_policies, policy);
} }
policy->context = context; policy->context = context;
@ -584,6 +594,10 @@ cleanup:
struct cmd_results *cmd_results_new(enum cmd_status status, const char* input, const char *format, ...) { struct cmd_results *cmd_results_new(enum cmd_status status, const char* input, const char *format, ...) {
struct cmd_results *results = malloc(sizeof(struct cmd_results)); struct cmd_results *results = malloc(sizeof(struct cmd_results));
if (!results) {
sway_log(L_ERROR, "Unable to allocate command results");
return NULL;
}
results->status = status; results->status = status;
if (input) { if (input) {
results->input = strdup(input); // input is the command name results->input = strdup(input); // input is the command name
@ -594,7 +608,9 @@ struct cmd_results *cmd_results_new(enum cmd_status status, const char* input, c
char *error = malloc(256); char *error = malloc(256);
va_list args; va_list args;
va_start(args, format); va_start(args, format);
vsnprintf(error, 256, format, args); if (error) {
vsnprintf(error, 256, format, args);
}
va_end(args); va_end(args);
results->error = error; results->error = error;
} else { } else {

View file

@ -23,10 +23,16 @@ struct cmd_results *cmd_assign(int argc, char **argv) {
char *movecmd = "move container to workspace "; char *movecmd = "move container to workspace ";
int arglen = strlen(movecmd) + strlen(*argv) + 1; int arglen = strlen(movecmd) + strlen(*argv) + 1;
char *cmdlist = calloc(1, arglen); char *cmdlist = calloc(1, arglen);
if (!cmdlist) {
return cmd_results_new(CMD_FAILURE, "assign", "Unable to allocate command list");
}
snprintf(cmdlist, arglen, "%s%s", movecmd, *argv); snprintf(cmdlist, arglen, "%s%s", movecmd, *argv);
struct criteria *crit = malloc(sizeof(struct criteria)); struct criteria *crit = malloc(sizeof(struct criteria));
if (!crit) {
free(cmdlist);
return cmd_results_new(CMD_FAILURE, "assign", "Unable to allocate criteria");
}
crit->crit_raw = strdup(criteria); crit->crit_raw = strdup(criteria);
crit->cmdlist = cmdlist; crit->cmdlist = cmdlist;
crit->tokens = create_list(); crit->tokens = create_list();

View file

@ -32,6 +32,9 @@ struct cmd_results *cmd_bar(int argc, char **argv) {
// Create new bar with default values // Create new bar with default values
struct bar_config *bar = default_bar_config(); struct bar_config *bar = default_bar_config();
if (!bar) {
return cmd_results_new(CMD_FAILURE, "bar", "Unable to allocate bar state");
}
// set bar id // set bar id
int i; int i;
@ -39,7 +42,11 @@ struct cmd_results *cmd_bar(int argc, char **argv) {
if (bar == config->bars->items[i]) { if (bar == config->bars->items[i]) {
const int len = 5 + numlen(i); // "bar-" + i + \0 const int len = 5 + numlen(i); // "bar-" + i + \0
bar->id = malloc(len * sizeof(char)); bar->id = malloc(len * sizeof(char));
snprintf(bar->id, len, "bar-%d", i); if (bar->id) {
snprintf(bar->id, len, "bar-%d", i);
} else {
return cmd_results_new(CMD_FAILURE, "bar", "Unable to allocate bar ID");
}
break; break;
} }
} }

View file

@ -26,6 +26,9 @@ struct cmd_results *bar_cmd_bindsym(int argc, char **argv) {
return cmd_results_new(CMD_INVALID, "bindsym", "Invalid mouse binding %s", argv[1]); return cmd_results_new(CMD_INVALID, "bindsym", "Invalid mouse binding %s", argv[1]);
} }
struct sway_mouse_binding *binding = malloc(sizeof(struct sway_mouse_binding)); struct sway_mouse_binding *binding = malloc(sizeof(struct sway_mouse_binding));
if (!binding) {
return cmd_results_new(CMD_FAILURE, "bindsym", "Unable to allocate binding");
}
binding->button = numbutton; binding->button = numbutton;
binding->command = join_args(argv + 1, argc - 1); binding->command = join_args(argv + 1, argc - 1);

View file

@ -9,6 +9,9 @@ static struct cmd_results *parse_single_color(char **color, const char *cmd_name
if (!*color) { if (!*color) {
*color = malloc(10); *color = malloc(10);
if (!*color) {
return cmd_results_new(CMD_FAILURE, cmd_name, "Unable to allocate color");
}
} }
error = add_color(cmd_name, *color, argv[0]); error = add_color(cmd_name, *color, argv[0]);
@ -29,6 +32,9 @@ static struct cmd_results *parse_three_colors(char ***colors, const char *cmd_na
for (i = 0; i < 3; i++) { for (i = 0; i < 3; i++) {
if (!*colors[i]) { if (!*colors[i]) {
*(colors[i]) = malloc(10); *(colors[i]) = malloc(10);
if (!*(colors[i])) {
return cmd_results_new(CMD_FAILURE, cmd_name, "Unable to allocate color");
}
} }
error = add_color(cmd_name, *(colors[i]), argv[i]); error = add_color(cmd_name, *(colors[i]), argv[i]);
if (error) { if (error) {

View file

@ -16,6 +16,10 @@ struct cmd_results *cmd_bindsym(int argc, char **argv) {
} }
struct sway_binding *binding = malloc(sizeof(struct sway_binding)); struct sway_binding *binding = malloc(sizeof(struct sway_binding));
if (!binding) {
return cmd_results_new(CMD_FAILURE, "bindsym",
"Unable to allocate binding");
}
binding->keys = create_list(); binding->keys = create_list();
binding->modifiers = 0; binding->modifiers = 0;
binding->release = false; binding->release = false;
@ -46,14 +50,21 @@ struct cmd_results *cmd_bindsym(int argc, char **argv) {
continue; continue;
} }
// Check for xkb key // Check for xkb key
xkb_keysym_t sym = xkb_keysym_from_name(split->items[i], XKB_KEYSYM_CASE_INSENSITIVE); xkb_keysym_t sym = xkb_keysym_from_name(split->items[i],
XKB_KEYSYM_CASE_INSENSITIVE);
if (!sym) { if (!sym) {
error = cmd_results_new(CMD_INVALID, "bindsym", "Unknown key '%s'", (char *)split->items[i]);
free_sway_binding(binding); free_sway_binding(binding);
list_free(split); free_flat_list(split);
return error; return cmd_results_new(CMD_INVALID, "bindsym", "Unknown key '%s'",
(char *)split->items[i]);
} }
xkb_keysym_t *key = malloc(sizeof(xkb_keysym_t)); xkb_keysym_t *key = malloc(sizeof(xkb_keysym_t));
if (!key) {
free_sway_binding(binding);
free_flat_list(split);
return cmd_results_new(CMD_FAILURE, "bindsym",
"Unable to allocate binding");
}
*key = sym; *key = sym;
list_add(binding->keys, key); list_add(binding->keys, key);
} }
@ -82,6 +93,10 @@ struct cmd_results *cmd_bindcode(int argc, char **argv) {
} }
struct sway_binding *binding = malloc(sizeof(struct sway_binding)); struct sway_binding *binding = malloc(sizeof(struct sway_binding));
if (!binding) {
return cmd_results_new(CMD_FAILURE, "bindsym",
"Unable to allocate binding");
}
binding->keys = create_list(); binding->keys = create_list();
binding->modifiers = 0; binding->modifiers = 0;
binding->release = false; binding->release = false;

View file

@ -39,6 +39,9 @@ struct cmd_results *cmd_exec_always(int argc, char **argv) {
pid_t pid; pid_t pid;
pid_t *child = malloc(sizeof(pid_t)); // malloc'd so that Linux can avoid copying the process space pid_t *child = malloc(sizeof(pid_t)); // malloc'd so that Linux can avoid copying the process space
if (!child) {
return cmd_results_new(CMD_FAILURE, "exec_always", "Unable to allocate child pid");
}
// Fork process // Fork process
if ((pid = fork()) == 0) { if ((pid = fork()) == 0) {
// Fork child process again // Fork child process again
@ -56,7 +59,7 @@ struct cmd_results *cmd_exec_always(int argc, char **argv) {
_exit(0); // Close child process _exit(0); // Close child process
} else if (pid < 0) { } else if (pid < 0) {
free(child); free(child);
return cmd_results_new(CMD_FAILURE, "exec_always", "Command failed (sway could not fork)."); return cmd_results_new(CMD_FAILURE, "exec_always", "fork() failed");
} }
close(fd[1]); // close write close(fd[1]); // close write
ssize_t s = 0; ssize_t s = 0;
@ -73,8 +76,6 @@ struct cmd_results *cmd_exec_always(int argc, char **argv) {
pw->pid = child; pw->pid = child;
pw->workspace = strdup(ws->name); pw->workspace = strdup(ws->name);
pid_workspace_add(pw); pid_workspace_add(pw);
// TODO: keep track of this pid and open the corresponding view on the current workspace
// blocked pending feature in wlc
} else { } else {
free(child); free(child);
} }

View file

@ -14,6 +14,9 @@ struct cmd_results *cmd_for_window(int argc, char **argv) {
char *criteria = argv[0], *cmdlist = join_args(argv + 1, argc - 1); char *criteria = argv[0], *cmdlist = join_args(argv + 1, argc - 1);
struct criteria *crit = malloc(sizeof(struct criteria)); struct criteria *crit = malloc(sizeof(struct criteria));
if (!crit) {
return cmd_results_new(CMD_FAILURE, "for_window", "Unable to allocate criteria");
}
crit->crit_raw = strdup(criteria); crit->crit_raw = strdup(criteria);
crit->cmdlist = cmdlist; crit->cmdlist = cmdlist;
crit->tokens = create_list(); crit->tokens = create_list();

View file

@ -30,7 +30,10 @@ struct cmd_results *cmd_mode(int argc, char **argv) {
} }
// Create mode if it doesn't exist // Create mode if it doesn't exist
if (!mode && mode_make) { if (!mode && mode_make) {
mode = malloc(sizeof*mode); mode = malloc(sizeof(struct sway_mode));
if (!mode) {
return cmd_results_new(CMD_FAILURE, "mode", "Unable to allocate mode");
}
mode->name = strdup(mode_name); mode->name = strdup(mode_name);
mode->bindings = create_list(); mode->bindings = create_list();
list_add(config->modes, mode); list_add(config->modes, mode);

View file

@ -26,6 +26,9 @@ struct cmd_results *cmd_output(int argc, char **argv) {
const char *name = argv[0]; const char *name = argv[0];
struct output_config *output = calloc(1, sizeof(struct output_config)); struct output_config *output = calloc(1, sizeof(struct output_config));
if (!output) {
return cmd_results_new(CMD_FAILURE, "output", "Unable to allocate output config");
}
output->x = output->y = output->width = output->height = -1; output->x = output->y = output->width = output->height = -1;
output->name = strdup(name); output->name = strdup(name);
output->enabled = -1; output->enabled = -1;
@ -113,12 +116,20 @@ struct cmd_results *cmd_output(int argc, char **argv) {
src = p.we_wordv[0]; src = p.we_wordv[0];
if (config->reading && *src != '/') { if (config->reading && *src != '/') {
char *conf = strdup(config->current_config); char *conf = strdup(config->current_config);
char *conf_path = dirname(conf); if (conf) {
src = malloc(strlen(conf_path) + strlen(src) + 2); char *conf_path = dirname(conf);
sprintf(src, "%s/%s", conf_path, p.we_wordv[0]); src = malloc(strlen(conf_path) + strlen(src) + 2);
free(conf); if (src) {
sprintf(src, "%s/%s", conf_path, p.we_wordv[0]);
} else {
sway_log(L_ERROR, "Unable to allocate background source");
}
free(conf);
} else {
sway_log(L_ERROR, "Unable to allocate background source");
}
} }
if (access(src, F_OK) == -1) { if (!src || access(src, F_OK) == -1) {
return cmd_results_new(CMD_INVALID, "output", "Background file unreadable (%s)", src); return cmd_results_new(CMD_INVALID, "output", "Background file unreadable (%s)", src);
} }
for (char *m = mode; *m; ++m) *m = tolower(*m); for (char *m = mode; *m; ++m) *m = tolower(*m);

View file

@ -50,6 +50,9 @@ static struct feature_policy *get_policy(const char *name) {
} }
if (!policy) { if (!policy) {
policy = alloc_feature_policy(name); policy = alloc_feature_policy(name);
if (!policy) {
sway_abort("Unable to allocate security policy");
}
list_add(config->feature_policies, policy); list_add(config->feature_policies, policy);
} }
return policy; return policy;

View file

@ -47,6 +47,9 @@ struct cmd_results *cmd_set(int argc, char **argv) {
free(var->value); free(var->value);
} else { } else {
var = malloc(sizeof(struct sway_variable)); var = malloc(sizeof(struct sway_variable));
if (!var) {
return cmd_results_new(CMD_FAILURE, "set", "Unable to allocate variable");
}
var->name = strdup(argv[0]); var->name = strdup(argv[0]);
list_add(config->symbols, var); list_add(config->symbols, var);
list_qsort(config->symbols, compare_set_qsort); list_qsort(config->symbols, compare_set_qsort);

View file

@ -61,6 +61,10 @@ struct cmd_results *cmd_workspace(int argc, char **argv) {
return error; return error;
} }
struct workspace_output *wso = calloc(1, sizeof(struct workspace_output)); struct workspace_output *wso = calloc(1, sizeof(struct workspace_output));
if (!wso) {
return cmd_results_new(CMD_FAILURE, "workspace output",
"Unable to allocate workspace output");
}
wso->workspace = strdup(argv[0]); wso->workspace = strdup(argv[0]);
wso->output = strdup(argv[2]); wso->output = strdup(argv[2]);
int i = -1; int i = -1;

View file

@ -29,21 +29,30 @@ struct sway_config *config = NULL;
static void terminate_swaybar(pid_t pid); static void terminate_swaybar(pid_t pid);
static void free_variable(struct sway_variable *var) { static void free_variable(struct sway_variable *var) {
if (!var) {
return;
}
free(var->name); free(var->name);
free(var->value); free(var->value);
free(var); free(var);
} }
static void free_binding(struct sway_binding *bind) { static void free_binding(struct sway_binding *bind) {
if (!bind) {
return;
}
free_flat_list(bind->keys); free_flat_list(bind->keys);
free(bind->command); free(bind->command);
free(bind); free(bind);
} }
static void free_mode(struct sway_mode *mode) { static void free_mode(struct sway_mode *mode) {
if (!mode) {
return;
}
free(mode->name); free(mode->name);
int i; int i;
for (i = 0; i < mode->bindings->length; ++i) { for (i = 0; mode->bindings && i < mode->bindings->length; ++i) {
free_binding(mode->bindings->items[i]); free_binding(mode->bindings->items[i]);
} }
list_free(mode->bindings); list_free(mode->bindings);
@ -51,13 +60,16 @@ static void free_mode(struct sway_mode *mode) {
} }
static void free_bar(struct bar_config *bar) { static void free_bar(struct bar_config *bar) {
if (!bar) {
return;
}
free(bar->mode); free(bar->mode);
free(bar->hidden_state); free(bar->hidden_state);
free(bar->status_command); free(bar->status_command);
free(bar->font); free(bar->font);
free(bar->separator_symbol); free(bar->separator_symbol);
int i; int i;
for (i = 0; i < bar->bindings->length; ++i) { for (i = 0; bar->bindings && i < bar->bindings->length; ++i) {
free_sway_mouse_binding(bar->bindings->items[i]); free_sway_mouse_binding(bar->bindings->items[i]);
} }
list_free(bar->bindings); list_free(bar->bindings);
@ -96,16 +108,25 @@ static void free_bar(struct bar_config *bar) {
} }
void free_input_config(struct input_config *ic) { void free_input_config(struct input_config *ic) {
if (!ic) {
return;
}
free(ic->identifier); free(ic->identifier);
free(ic); free(ic);
} }
void free_output_config(struct output_config *oc) { void free_output_config(struct output_config *oc) {
if (!oc) {
return;
}
free(oc->name); free(oc->name);
free(oc); free(oc);
} }
static void free_workspace_output(struct workspace_output *wo) { static void free_workspace_output(struct workspace_output *wo) {
if (!wo) {
return;
}
free(wo->output); free(wo->output);
free(wo->workspace); free(wo->workspace);
free(wo); free(wo);
@ -134,6 +155,10 @@ void pid_workspace_add(struct pid_workspace *pw) {
struct pid_workspace *list_pw = NULL; struct pid_workspace *list_pw = NULL;
struct timespec ts; struct timespec ts;
time_t *now = malloc(sizeof(time_t)); time_t *now = malloc(sizeof(time_t));
if (!now) {
sway_log(L_ERROR, "Allocating time for pid_workspace failed");
return;
}
pid_workspace_cleanup(); pid_workspace_cleanup();
@ -168,65 +193,74 @@ void free_pid_workspace(struct pid_workspace *pw) {
} }
void free_command_policy(struct command_policy *policy) { void free_command_policy(struct command_policy *policy) {
if (!policy) {
return;
}
free(policy->command); free(policy->command);
free(policy); free(policy);
} }
void free_feature_policy(struct feature_policy *policy) { void free_feature_policy(struct feature_policy *policy) {
if (!policy) {
return;
}
free(policy->program); free(policy->program);
free(policy); free(policy);
} }
void free_config(struct sway_config *config) { void free_config(struct sway_config *config) {
if (!config) {
return;
}
int i; int i;
for (i = 0; i < config->symbols->length; ++i) { for (i = 0; config->symbols && i < config->symbols->length; ++i) {
free_variable(config->symbols->items[i]); free_variable(config->symbols->items[i]);
} }
list_free(config->symbols); list_free(config->symbols);
for (i = 0; i < config->modes->length; ++i) { for (i = 0; config->modes && i < config->modes->length; ++i) {
free_mode(config->modes->items[i]); free_mode(config->modes->items[i]);
} }
list_free(config->modes); list_free(config->modes);
for (i = 0; i < config->bars->length; ++i) { for (i = 0; config->bars && i < config->bars->length; ++i) {
free_bar(config->bars->items[i]); free_bar(config->bars->items[i]);
} }
list_free(config->bars); list_free(config->bars);
free_flat_list(config->cmd_queue); free_flat_list(config->cmd_queue);
for (i = 0; i < config->workspace_outputs->length; ++i) { for (i = 0; config->workspace_outputs && i < config->workspace_outputs->length; ++i) {
free_workspace_output(config->workspace_outputs->items[i]); free_workspace_output(config->workspace_outputs->items[i]);
} }
list_free(config->workspace_outputs); list_free(config->workspace_outputs);
for (i = 0; i < config->pid_workspaces->length; ++i) { for (i = 0; config->pid_workspaces && i < config->pid_workspaces->length; ++i) {
free_pid_workspace(config->pid_workspaces->items[i]); free_pid_workspace(config->pid_workspaces->items[i]);
} }
list_free(config->pid_workspaces); list_free(config->pid_workspaces);
for (i = 0; i < config->criteria->length; ++i) { for (i = 0; config->criteria && i < config->criteria->length; ++i) {
free_criteria(config->criteria->items[i]); free_criteria(config->criteria->items[i]);
} }
list_free(config->criteria); list_free(config->criteria);
for (i = 0; i < config->input_configs->length; ++i) { for (i = 0; config->input_configs && i < config->input_configs->length; ++i) {
free_input_config(config->input_configs->items[i]); free_input_config(config->input_configs->items[i]);
} }
list_free(config->input_configs); list_free(config->input_configs);
for (i = 0; i < config->output_configs->length; ++i) { for (i = 0; config->output_configs && i < config->output_configs->length; ++i) {
free_output_config(config->output_configs->items[i]); free_output_config(config->output_configs->items[i]);
} }
list_free(config->output_configs); list_free(config->output_configs);
for (i = 0; i < config->command_policies->length; ++i) { for (i = 0; config->command_policies && i < config->command_policies->length; ++i) {
free_command_policy(config->command_policies->items[i]); free_command_policy(config->command_policies->items[i]);
} }
list_free(config->command_policies); list_free(config->command_policies);
for (i = 0; i < config->feature_policies->length; ++i) { for (i = 0; config->feature_policies && i < config->feature_policies->length; ++i) {
free_feature_policy(config->feature_policies->items[i]); free_feature_policy(config->feature_policies->items[i]);
} }
list_free(config->feature_policies); list_free(config->feature_policies);
@ -243,37 +277,37 @@ void free_config(struct sway_config *config) {
static bool file_exists(const char *path) { static bool file_exists(const char *path) {
return access(path, R_OK) != -1; return path && access(path, R_OK) != -1;
} }
static void config_defaults(struct sway_config *config) { static void config_defaults(struct sway_config *config) {
config->symbols = create_list(); if (!(config->symbols = create_list())) goto cleanup;
config->modes = create_list(); if (!(config->modes = create_list())) goto cleanup;
config->bars = create_list(); if (!(config->bars = create_list())) goto cleanup;
config->workspace_outputs = create_list(); if (!(config->workspace_outputs = create_list())) goto cleanup;
config->pid_workspaces = create_list(); if (!(config->pid_workspaces = create_list())) goto cleanup;
config->criteria = create_list(); if (!(config->criteria = create_list())) goto cleanup;
config->input_configs = create_list(); if (!(config->input_configs = create_list())) goto cleanup;
config->output_configs = create_list(); if (!(config->output_configs = create_list())) goto cleanup;
config->cmd_queue = create_list(); if (!(config->cmd_queue = create_list())) goto cleanup;
config->current_mode = malloc(sizeof(struct sway_mode)); if (!(config->current_mode = malloc(sizeof(struct sway_mode)))) goto cleanup;
config->current_mode->name = malloc(sizeof("default")); if (!(config->current_mode->name = malloc(sizeof("default")))) goto cleanup;
strcpy(config->current_mode->name, "default"); strcpy(config->current_mode->name, "default");
config->current_mode->bindings = create_list(); if (!(config->current_mode->bindings = create_list())) goto cleanup;
list_add(config->modes, config->current_mode); list_add(config->modes, config->current_mode);
config->floating_mod = 0; config->floating_mod = 0;
config->dragging_key = M_LEFT_CLICK; config->dragging_key = M_LEFT_CLICK;
config->resizing_key = M_RIGHT_CLICK; config->resizing_key = M_RIGHT_CLICK;
config->floating_scroll_up_cmd = strdup(""); if (!(config->floating_scroll_up_cmd = strdup(""))) goto cleanup;
config->floating_scroll_down_cmd = strdup(""); if (!(config->floating_scroll_down_cmd = strdup(""))) goto cleanup;
config->floating_scroll_left_cmd = strdup(""); if (!(config->floating_scroll_left_cmd = strdup(""))) goto cleanup;
config->floating_scroll_right_cmd = strdup(""); if (!(config->floating_scroll_right_cmd = strdup(""))) goto cleanup;
config->default_layout = L_NONE; config->default_layout = L_NONE;
config->default_orientation = L_NONE; config->default_orientation = L_NONE;
config->font = strdup("monospace 10"); if (!(config->font = strdup("monospace 10"))) goto cleanup;
config->font_height = get_font_text_height(config->font); config->font_height = get_font_text_height(config->font);
// floating view // floating view
@ -297,9 +331,9 @@ static void config_defaults(struct sway_config *config) {
config->gaps_inner = 0; config->gaps_inner = 0;
config->gaps_outer = 0; config->gaps_outer = 0;
config->active_bar_modifiers = create_list(); if (!(config->active_bar_modifiers = create_list())) goto cleanup;
config->config_chain = create_list(); if (!(config->config_chain = create_list())) goto cleanup;
config->current_config = NULL; config->current_config = NULL;
// borders // borders
@ -343,9 +377,13 @@ static void config_defaults(struct sway_config *config) {
config->border_colors.background = 0xFFFFFFFF; config->border_colors.background = 0xFFFFFFFF;
// Security // Security
config->command_policies = create_list(); if (!(config->command_policies = create_list())) goto cleanup;
config->feature_policies = create_list(); if (!(config->feature_policies = create_list())) goto cleanup;
config->ipc_policy = UINT32_MAX; config->ipc_policy = UINT32_MAX;
return;
cleanup:
sway_abort("Unable to allocate config structures");
} }
static int compare_modifiers(const void *left, const void *right) { static int compare_modifiers(const void *left, const void *right) {
@ -386,11 +424,15 @@ static char *get_config_path(void) {
if (!getenv("XDG_CONFIG_HOME")) { if (!getenv("XDG_CONFIG_HOME")) {
char *home = getenv("HOME"); char *home = getenv("HOME");
char *config_home = malloc(strlen(home) + strlen("/.config") + 1); char *config_home = malloc(strlen(home) + strlen("/.config") + 1);
strcpy(config_home, home); if (!config_home) {
strcat(config_home, "/.config"); sway_log(L_ERROR, "Unable to allocate $HOME/.config");
setenv("XDG_CONFIG_HOME", config_home, 1); } else {
sway_log(L_DEBUG, "Set XDG_CONFIG_HOME to %s", config_home); strcpy(config_home, home);
free(config_home); strcat(config_home, "/.config");
setenv("XDG_CONFIG_HOME", config_home, 1);
sway_log(L_DEBUG, "Set XDG_CONFIG_HOME to %s", config_home);
free(config_home);
}
} }
wordexp_t p; wordexp_t p;
@ -452,6 +494,9 @@ bool load_main_config(const char *file, bool is_active) {
struct sway_config *old_config = config; struct sway_config *old_config = config;
config = calloc(1, sizeof(struct sway_config)); config = calloc(1, sizeof(struct sway_config));
if (!config) {
sway_abort("Unable to allocate config");
}
config_defaults(config); config_defaults(config);
if (is_active) { if (is_active) {
@ -491,6 +536,10 @@ static bool load_include_config(const char *path, const char *parent_dir, struct
if (len >= 1 && path[0] != '/') { if (len >= 1 && path[0] != '/') {
len = len + strlen(parent_dir) + 2; len = len + strlen(parent_dir) + 2;
full_path = malloc(len * sizeof(char)); full_path = malloc(len * sizeof(char));
if (!full_path) {
sway_log(L_ERROR, "Unable to allocate full path to included config");
return false;
}
snprintf(full_path, len, "%s/%s", parent_dir, path); snprintf(full_path, len, "%s/%s", parent_dir, path);
} }
@ -575,6 +624,9 @@ bool read_config(FILE *file, struct sway_config *config) {
char *line; char *line;
while (!feof(file)) { while (!feof(file)) {
line = read_line(file); line = read_line(file);
if (!line) {
continue;
}
line_number++; line_number++;
line = strip_whitespace(line); line = strip_whitespace(line);
if (line[0] == '#') { if (line[0] == '#') {
@ -816,6 +868,10 @@ static void invoke_swaybar(struct bar_config *bar) {
// run custom swaybar // run custom swaybar
int len = strlen(bar->swaybar_command) + strlen(bar->id) + 5; int len = strlen(bar->swaybar_command) + strlen(bar->id) + 5;
char *command = malloc(len * sizeof(char)); char *command = malloc(len * sizeof(char));
if (!command) {
sway_log(L_ERROR, "Unable to allocate swaybar command string");
return;
}
snprintf(command, len, "%s -b %s", bar->swaybar_command, bar->id); snprintf(command, len, "%s -b %s", bar->swaybar_command, bar->id);
char *const cmd[] = { char *const cmd[] = {
@ -1052,6 +1108,11 @@ char *do_var_replacement(char *str) {
if (strncmp(find, var->name, vnlen) == 0) { if (strncmp(find, var->name, vnlen) == 0) {
int vvlen = strlen(var->value); int vvlen = strlen(var->value);
char *newstr = malloc(strlen(str) - vnlen + vvlen + 1); char *newstr = malloc(strlen(str) - vnlen + vvlen + 1);
if (!newstr) {
sway_log(L_ERROR,
"Unable to allocate replacement during variable expansion");
break;
}
char *newptr = newstr; char *newptr = newstr;
int offset = find - str; int offset = find - str;
strncpy(newptr, str, offset); strncpy(newptr, str, offset);
@ -1185,6 +1246,9 @@ void free_sway_mouse_binding(struct sway_mouse_binding *binding) {
struct sway_binding *sway_binding_dup(struct sway_binding *sb) { struct sway_binding *sway_binding_dup(struct sway_binding *sb) {
struct sway_binding *new_sb = malloc(sizeof(struct sway_binding)); struct sway_binding *new_sb = malloc(sizeof(struct sway_binding));
if (!new_sb) {
return NULL;
}
new_sb->order = sb->order; new_sb->order = sb->order;
new_sb->modifiers = sb->modifiers; new_sb->modifiers = sb->modifiers;
@ -1194,6 +1258,10 @@ struct sway_binding *sway_binding_dup(struct sway_binding *sb) {
int i; int i;
for (i = 0; i < sb->keys->length; ++i) { for (i = 0; i < sb->keys->length; ++i) {
xkb_keysym_t *key = malloc(sizeof(xkb_keysym_t)); xkb_keysym_t *key = malloc(sizeof(xkb_keysym_t));
if (!key) {
free_sway_binding(new_sb);
return NULL;
}
*key = *(xkb_keysym_t *)sb->keys->items[i]; *key = *(xkb_keysym_t *)sb->keys->items[i];
list_add(new_sb->keys, key); list_add(new_sb->keys, key);
} }
@ -1204,13 +1272,16 @@ struct sway_binding *sway_binding_dup(struct sway_binding *sb) {
struct bar_config *default_bar_config(void) { struct bar_config *default_bar_config(void) {
struct bar_config *bar = NULL; struct bar_config *bar = NULL;
bar = malloc(sizeof(struct bar_config)); bar = malloc(sizeof(struct bar_config));
bar->mode = strdup("dock"); if (!bar) {
bar->hidden_state = strdup("hide"); return NULL;
}
if (!(bar->mode = strdup("dock"))) goto cleanup;
if (!(bar->hidden_state = strdup("hide"))) goto cleanup;
bar->modifier = WLC_BIT_MOD_LOGO; bar->modifier = WLC_BIT_MOD_LOGO;
bar->outputs = NULL; bar->outputs = NULL;
bar->position = DESKTOP_SHELL_PANEL_POSITION_BOTTOM; bar->position = DESKTOP_SHELL_PANEL_POSITION_BOTTOM;
bar->bindings = create_list(); if (!(bar->bindings = create_list())) goto cleanup;
bar->status_command = strdup("while :; do date +'%Y-%m-%d %l:%M:%S %p' && sleep 1; done"); if (!(bar->status_command = strdup("while :; do date +'%Y-%m-%d %l:%M:%S %p' && sleep 1; done"))) goto cleanup;
bar->pango_markup = false; bar->pango_markup = false;
bar->swaybar_command = NULL; bar->swaybar_command = NULL;
bar->font = NULL; bar->font = NULL;
@ -1224,21 +1295,21 @@ struct bar_config *default_bar_config(void) {
bar->verbose = false; bar->verbose = false;
bar->pid = 0; bar->pid = 0;
// set default colors // set default colors
bar->colors.background = strndup("#000000ff", 9); if (!(bar->colors.background = strndup("#000000ff", 9))) goto cleanup;
bar->colors.statusline = strndup("#ffffffff", 9); if (!(bar->colors.statusline = strndup("#ffffffff", 9))) goto cleanup;
bar->colors.separator = strndup("#666666ff", 9); if (!(bar->colors.separator = strndup("#666666ff", 9))) goto cleanup;
bar->colors.focused_workspace_border = strndup("#4c7899ff", 9); if (!(bar->colors.focused_workspace_border = strndup("#4c7899ff", 9))) goto cleanup;
bar->colors.focused_workspace_bg = strndup("#285577ff", 9); if (!(bar->colors.focused_workspace_bg = strndup("#285577ff", 9))) goto cleanup;
bar->colors.focused_workspace_text = strndup("#ffffffff", 9); if (!(bar->colors.focused_workspace_text = strndup("#ffffffff", 9))) goto cleanup;
bar->colors.active_workspace_border = strndup("#333333ff", 9); if (!(bar->colors.active_workspace_border = strndup("#333333ff", 9))) goto cleanup;
bar->colors.active_workspace_bg = strndup("#5f676aff", 9); if (!(bar->colors.active_workspace_bg = strndup("#5f676aff", 9))) goto cleanup;
bar->colors.active_workspace_text = strndup("#ffffffff", 9); if (!(bar->colors.active_workspace_text = strndup("#ffffffff", 9))) goto cleanup;
bar->colors.inactive_workspace_border = strndup("#333333ff", 9); if (!(bar->colors.inactive_workspace_border = strndup("#333333ff", 9))) goto cleanup;
bar->colors.inactive_workspace_bg = strndup("#222222ff", 9); if (!(bar->colors.inactive_workspace_bg = strndup("#222222ff", 9))) goto cleanup;
bar->colors.inactive_workspace_text = strndup("#888888ff", 9); if (!(bar->colors.inactive_workspace_text = strndup("#888888ff", 9))) goto cleanup;
bar->colors.urgent_workspace_border = strndup("#2f343aff", 9); if (!(bar->colors.urgent_workspace_border = strndup("#2f343aff", 9))) goto cleanup;
bar->colors.urgent_workspace_bg = strndup("#900000ff", 9); if (!(bar->colors.urgent_workspace_bg = strndup("#900000ff", 9))) goto cleanup;
bar->colors.urgent_workspace_text = strndup("#ffffffff", 9); if (!(bar->colors.urgent_workspace_text = strndup("#ffffffff", 9))) goto cleanup;
// if the following colors stay undefined, they fall back to background, // if the following colors stay undefined, they fall back to background,
// statusline, separator and urgent_workspace_*. // statusline, separator and urgent_workspace_*.
bar->colors.focused_background = NULL; bar->colors.focused_background = NULL;
@ -1251,4 +1322,8 @@ struct bar_config *default_bar_config(void) {
list_add(config->bars, bar); list_add(config->bars, bar);
return bar; return bar;
cleanup:
free_bar(bar);
return NULL;
} }

View file

@ -23,6 +23,9 @@ static swayc_t *new_swayc(enum swayc_types type) {
// next id starts at 1 because 0 is assigned to root_container in layout.c // next id starts at 1 because 0 is assigned to root_container in layout.c
static size_t next_id = 1; static size_t next_id = 1;
swayc_t *c = calloc(1, sizeof(swayc_t)); swayc_t *c = calloc(1, sizeof(swayc_t));
if (!c) {
return NULL;
}
c->id = next_id++; c->id = next_id++;
c->handle = -1; c->handle = -1;
c->gaps = -1; c->gaps = -1;
@ -217,7 +220,7 @@ swayc_t *new_workspace(swayc_t *output, const char *name) {
workspace->y = output->y; workspace->y = output->y;
workspace->width = output->width; workspace->width = output->width;
workspace->height = output->height; workspace->height = output->height;
workspace->name = strdup(name); workspace->name = !name ? NULL : strdup(name);
workspace->visible = false; workspace->visible = false;
workspace->floating = create_list(); workspace->floating = create_list();

View file

@ -23,6 +23,10 @@ static struct panel_config *find_or_create_panel_config(struct wl_resource *reso
} }
sway_log(L_DEBUG, "Creating panel config for resource %p", resource); sway_log(L_DEBUG, "Creating panel config for resource %p", resource);
struct panel_config *config = calloc(1, sizeof(struct panel_config)); struct panel_config *config = calloc(1, sizeof(struct panel_config));
if (!config) {
sway_log(L_ERROR, "Unable to create panel config");
return NULL;
}
list_add(desktop_shell.panels, config); list_add(desktop_shell.panels, config);
config->wl_resource = resource; config->wl_resource = resource;
return config; return config;
@ -81,6 +85,10 @@ static void set_background(struct wl_client *client, struct wl_resource *resourc
} }
sway_log(L_DEBUG, "Setting surface %p as background for output %d", surface, (int)output); sway_log(L_DEBUG, "Setting surface %p as background for output %d", surface, (int)output);
struct background_config *config = malloc(sizeof(struct background_config)); struct background_config *config = malloc(sizeof(struct background_config));
if (!config) {
sway_log(L_ERROR, "Unable to allocate background config");
return;
}
config->client = client; config->client = client;
config->output = output; config->output = output;
config->surface = wlc_resource_from_wl_surface_resource(surface); config->surface = wlc_resource_from_wl_surface_resource(surface);

View file

@ -123,6 +123,11 @@ static void update_background_geometries(wlc_handle output) {
static bool handle_input_created(struct libinput_device *device) { static bool handle_input_created(struct libinput_device *device) {
const char *identifier = libinput_dev_unique_id(device); const char *identifier = libinput_dev_unique_id(device);
if (!identifier) {
sway_log(L_ERROR, "Unable to allocate unique name for input device %p",
device);
return true;
}
sway_log(L_INFO, "Found input device (%s)", identifier); sway_log(L_INFO, "Found input device (%s)", identifier);
list_add(input_devices, device); list_add(input_devices, device);
@ -402,6 +407,10 @@ static bool handle_view_created(wlc_handle handle) {
} else { } else {
swayc_t *output = swayc_parent_by_type(focused, C_OUTPUT); swayc_t *output = swayc_parent_by_type(focused, C_OUTPUT);
wlc_handle *h = malloc(sizeof(wlc_handle)); wlc_handle *h = malloc(sizeof(wlc_handle));
if (!h) {
sway_log(L_ERROR, "Unable to allocate window handle, view handler bailing out");
return true;
}
*h = handle; *h = handle;
sway_log(L_DEBUG, "Adding unmanaged window %p to %p", h, output->unmanaged); sway_log(L_DEBUG, "Adding unmanaged window %p to %p", h, output->unmanaged);
list_add(output->unmanaged, h); list_add(output->unmanaged, h);
@ -571,6 +580,10 @@ static void handle_binding_command(struct sway_binding *binding) {
// binding since it will be gone after the reload has completed. // binding since it will be gone after the reload has completed.
if (strcasecmp(binding->command, "reload") == 0) { if (strcasecmp(binding->command, "reload") == 0) {
binding_copy = sway_binding_dup(binding); binding_copy = sway_binding_dup(binding);
if (!binding_copy) {
sway_log(L_ERROR, "Unable to duplicate binding during reload");
return;
}
reload = true; reload = true;
} }

View file

@ -11,8 +11,16 @@
struct input_config *new_input_config(const char* identifier) { struct input_config *new_input_config(const char* identifier) {
struct input_config *input = calloc(1, sizeof(struct input_config)); struct input_config *input = calloc(1, sizeof(struct input_config));
if (!input) {
sway_log(L_DEBUG, "Unable to allocate input config");
return NULL;
}
sway_log(L_DEBUG, "new_input_config(%s)", identifier); sway_log(L_DEBUG, "new_input_config(%s)", identifier);
input->identifier = strdup(identifier); if (!(input->identifier = strdup(identifier))) {
free(input);
sway_log(L_DEBUG, "Unable to allocate input config");
return NULL;
}
input->tap = INT_MIN; input->tap = INT_MIN;
input->drag_lock = INT_MIN; input->drag_lock = INT_MIN;
@ -45,6 +53,10 @@ char *libinput_dev_unique_id(struct libinput_device *device) {
int len = strlen(name) + sizeof(char) * 6; int len = strlen(name) + sizeof(char) * 6;
char *identifier = malloc(len); char *identifier = malloc(len);
if (!identifier) {
sway_log(L_ERROR, "Unable to allocate unique input device name");
return NULL;
}
const char *fmt = "%d:%d:%s"; const char *fmt = "%d:%d:%s";
snprintf(identifier, len, fmt, vendor, product, name); snprintf(identifier, len, fmt, vendor, product, name);

View file

@ -245,6 +245,15 @@ json_object *ipc_json_get_version() {
#if defined SWAY_GIT_VERSION && defined SWAY_GIT_BRANCH && defined SWAY_VERSION_DATE #if defined SWAY_GIT_VERSION && defined SWAY_GIT_BRANCH && defined SWAY_VERSION_DATE
char *full_version = calloc(strlen(SWAY_GIT_VERSION) + strlen(SWAY_GIT_BRANCH) + strlen(SWAY_VERSION_DATE) + 20, 1); char *full_version = calloc(strlen(SWAY_GIT_VERSION) + strlen(SWAY_GIT_BRANCH) + strlen(SWAY_VERSION_DATE) + 20, 1);
if (!full_version) {
json_object_object_add(version, "human_readable",
json_object_new_string("Allocating version string failed"));
// TODO: it's stupid that we allocate this in the first place
json_object_object_add(version, "major", json_object_new_int(0));
json_object_object_add(version, "minor", json_object_new_int(0));
json_object_object_add(version, "patch", json_object_new_int(0));
return version;
}
strcat(full_version, SWAY_GIT_VERSION); strcat(full_version, SWAY_GIT_VERSION);
strcat(full_version, " ("); strcat(full_version, " (");
strcat(full_version, SWAY_VERSION_DATE); strcat(full_version, SWAY_VERSION_DATE);

View file

@ -106,7 +106,7 @@ void ipc_terminate(void) {
struct sockaddr_un *ipc_user_sockaddr(void) { struct sockaddr_un *ipc_user_sockaddr(void) {
struct sockaddr_un *ipc_sockaddr = malloc(sizeof(struct sockaddr_un)); struct sockaddr_un *ipc_sockaddr = malloc(sizeof(struct sockaddr_un));
if (ipc_sockaddr == NULL) { if (ipc_sockaddr == NULL) {
sway_abort("can't malloc ipc_sockaddr"); sway_abort("Can't allocate ipc_sockaddr");
} }
ipc_sockaddr->sun_family = AF_UNIX; ipc_sockaddr->sun_family = AF_UNIX;
@ -119,7 +119,7 @@ struct sockaddr_un *ipc_user_sockaddr(void) {
} }
if (path_size <= snprintf(ipc_sockaddr->sun_path, path_size, if (path_size <= snprintf(ipc_sockaddr->sun_path, path_size,
"%s/sway-ipc.%i.%i.sock", dir, getuid(), getpid())) { "%s/sway-ipc.%i.%i.sock", dir, getuid(), getpid())) {
sway_abort("socket path won't fit into ipc_sockaddr->sun_path"); sway_abort("Socket path won't fit into ipc_sockaddr->sun_path");
} }
return ipc_sockaddr; return ipc_sockaddr;
@ -148,13 +148,13 @@ int ipc_handle_connection(int fd, uint32_t mask, void *data) {
int client_fd = accept(ipc_socket, NULL, NULL); int client_fd = accept(ipc_socket, NULL, NULL);
if (client_fd == -1) { if (client_fd == -1) {
sway_log_errno(L_INFO, "Unable to accept IPC client connection"); sway_log_errno(L_ERROR, "Unable to accept IPC client connection");
return 0; return 0;
} }
int flags; int flags;
if ((flags=fcntl(client_fd, F_GETFD)) == -1 || fcntl(client_fd, F_SETFD, flags|FD_CLOEXEC) == -1) { if ((flags=fcntl(client_fd, F_GETFD)) == -1 || fcntl(client_fd, F_SETFD, flags|FD_CLOEXEC) == -1) {
sway_log_errno(L_INFO, "Unable to set CLOEXEC on IPC client socket"); sway_log_errno(L_ERROR, "Unable to set CLOEXEC on IPC client socket");
close(client_fd); close(client_fd);
return 0; return 0;
} }
@ -171,6 +171,11 @@ int ipc_handle_connection(int fd, uint32_t mask, void *data) {
} }
struct ipc_client* client = malloc(sizeof(struct ipc_client)); struct ipc_client* client = malloc(sizeof(struct ipc_client));
if (!client) {
sway_log(L_ERROR, "Unable to allocate ipc client");
close(client_fd);
return 0;
}
client->payload_length = 0; client->payload_length = 0;
client->fd = client_fd; client->fd = client_fd;
client->subscribed_events = 0; client->subscribed_events = 0;
@ -187,7 +192,7 @@ int ipc_client_handle_readable(int client_fd, uint32_t mask, void *data) {
struct ipc_client *client = data; struct ipc_client *client = data;
if (mask & WLC_EVENT_ERROR) { if (mask & WLC_EVENT_ERROR) {
sway_log(L_INFO, "IPC Client socket error, removing client"); sway_log(L_ERROR, "IPC Client socket error, removing client");
client->fd = -1; client->fd = -1;
ipc_client_disconnect(client); ipc_client_disconnect(client);
return 0; return 0;
@ -291,6 +296,12 @@ void ipc_get_pixels(wlc_handle output) {
char response_header[9]; char response_header[9];
memset(response_header, 0, sizeof(response_header)); memset(response_header, 0, sizeof(response_header));
char *data = malloc(sizeof(response_header) + size->w * size->h * 4); char *data = malloc(sizeof(response_header) + size->w * size->h * 4);
if (!data) {
sway_log(L_ERROR, "Unable to allocate pixels for get_pixels");
ipc_client_disconnect(req->client);
free(req);
continue;
}
wlc_pixels_read(WLC_RGBA8888, &req->geo, &g_out, data + sizeof(response_header)); wlc_pixels_read(WLC_RGBA8888, &req->geo, &g_out, data + sizeof(response_header));
response_header[0] = 1; response_header[0] = 1;
@ -318,7 +329,7 @@ void ipc_client_handle_command(struct ipc_client *client) {
char *buf = malloc(client->payload_length + 1); char *buf = malloc(client->payload_length + 1);
if (!buf) { if (!buf) {
sway_log_errno(L_INFO, "Out of memory"); sway_log_errno(L_INFO, "Unable to allocate IPC payload");
ipc_client_disconnect(client); ipc_client_disconnect(client);
return; return;
} }
@ -414,7 +425,11 @@ void ipc_client_handle_command(struct ipc_client *client) {
struct libinput_device *device = input_devices->items[i]; struct libinput_device *device = input_devices->items[i];
char* identifier = libinput_dev_unique_id(device); char* identifier = libinput_dev_unique_id(device);
json_object *device_object = json_object_new_object(); json_object *device_object = json_object_new_object();
json_object_object_add(device_object, "identifier", json_object_new_string(identifier)); if (!identifier) {
json_object_object_add(device_object, "identifier", NULL);
} else {
json_object_object_add(device_object, "identifier", json_object_new_string(identifier));
}
json_object_array_add(inputs, device_object); json_object_array_add(inputs, device_object);
free(identifier); free(identifier);
} }
@ -493,6 +508,10 @@ void ipc_client_handle_command(struct ipc_client *client) {
goto exit_cleanup; goto exit_cleanup;
} }
struct get_pixels_request *req = malloc(sizeof(struct get_pixels_request)); struct get_pixels_request *req = malloc(sizeof(struct get_pixels_request));
if (!req) {
sway_log(L_ERROR, "Unable to allocate get_pixels request");
goto exit_cleanup;
}
req->client = client; req->client = client;
req->output = output->handle; req->output = output->handle;
req->geo = g; req->geo = g;

View file

@ -53,7 +53,10 @@ void detect_proprietary() {
return; return;
} }
while (!feof(f)) { while (!feof(f)) {
char *line = read_line(f); char *line;
if (!(line = read_line(f))) {
break;
}
if (strstr(line, "nvidia")) { if (strstr(line, "nvidia")) {
fprintf(stderr, "\x1B[1;31mWarning: Proprietary nvidia drivers do NOT support Wayland. Use nouveau.\x1B[0m\n"); fprintf(stderr, "\x1B[1;31mWarning: Proprietary nvidia drivers do NOT support Wayland. Use nouveau.\x1B[0m\n");
fprintf(stderr, "\x1B[1;31mYes, they STILL don't work with the newly announced wayland \"support\".\x1B[0m\n"); fprintf(stderr, "\x1B[1;31mYes, they STILL don't work with the newly announced wayland \"support\".\x1B[0m\n");
@ -118,7 +121,10 @@ static void log_distro() {
if (f) { if (f) {
sway_log(L_INFO, "Contents of %s:", paths[i]); sway_log(L_INFO, "Contents of %s:", paths[i]);
while (!feof(f)) { while (!feof(f)) {
char *line = read_line(f); char *line;
if (!(line = read_line(f))) {
break;
}
if (*line) { if (*line) {
sway_log(L_INFO, "%s", line); sway_log(L_INFO, "%s", line);
} }
@ -136,7 +142,10 @@ static void log_kernel() {
return; return;
} }
while (!feof(f)) { while (!feof(f)) {
char *line = read_line(f); char *line;
if (!(line = read_line(f))) {
break;
}
if (*line) { if (*line) {
sway_log(L_INFO, "%s", line); sway_log(L_INFO, "%s", line);
} }

View file

@ -15,14 +15,28 @@ struct feature_policy *alloc_feature_policy(const char *program) {
} }
struct feature_policy *policy = malloc(sizeof(struct feature_policy)); struct feature_policy *policy = malloc(sizeof(struct feature_policy));
if (!policy) {
return NULL;
}
policy->program = strdup(program); policy->program = strdup(program);
if (!policy->program) {
free(policy);
return NULL;
}
policy->features = default_policy; policy->features = default_policy;
return policy; return policy;
} }
struct command_policy *alloc_command_policy(const char *command) { struct command_policy *alloc_command_policy(const char *command) {
struct command_policy *policy = malloc(sizeof(struct command_policy)); struct command_policy *policy = malloc(sizeof(struct command_policy));
if (!policy) {
return NULL;
}
policy->command = strdup(command); policy->command = strdup(command);
if (!policy->command) {
free(policy);
return NULL;
}
policy->context = 0; policy->context = 0;
return policy; return policy;
} }
@ -35,12 +49,14 @@ enum secure_feature get_feature_policy(pid_t pid) {
#endif #endif
int pathlen = snprintf(NULL, 0, fmt, pid); int pathlen = snprintf(NULL, 0, fmt, pid);
char *path = malloc(pathlen + 1); char *path = malloc(pathlen + 1);
snprintf(path, pathlen + 1, fmt, pid); if (path) {
snprintf(path, pathlen + 1, fmt, pid);
}
static char link[2048]; static char link[2048];
uint32_t default_policy = 0; uint32_t default_policy = 0;
ssize_t len = readlink(path, link, sizeof(link)); ssize_t len = !path ? -1 : readlink(path, link, sizeof(link));
if (len < 0) { if (len < 0) {
sway_log(L_INFO, sway_log(L_INFO,
"WARNING: unable to read %s for security check. Using default policy.", "WARNING: unable to read %s for security check. Using default policy.",

View file

@ -121,6 +121,10 @@ char *workspace_next_name(const char *output_name) {
l = 3; l = 3;
} }
char *name = malloc(l + 1); char *name = malloc(l + 1);
if (!name) {
sway_log(L_ERROR, "Could not allocate workspace name");
return NULL;
}
sprintf(name, "%d", ws_num++); sprintf(name, "%d", ws_num++);
return name; return name;
} }
@ -278,7 +282,11 @@ bool workspace_switch(swayc_t *workspace) {
|| (strcmp(prev_workspace_name, active_ws->name) || (strcmp(prev_workspace_name, active_ws->name)
&& active_ws != workspace)) { && active_ws != workspace)) {
free(prev_workspace_name); free(prev_workspace_name);
prev_workspace_name = malloc(strlen(active_ws->name)+1); prev_workspace_name = malloc(strlen(active_ws->name) + 1);
if (!prev_workspace_name) {
sway_log(L_ERROR, "Unable to allocate previous workspace name");
return false;
}
strcpy(prev_workspace_name, active_ws->name); strcpy(prev_workspace_name, active_ws->name);
} }

View file

@ -331,6 +331,9 @@ void ipc_bar_init(struct bar *bar, const char *bar_id) {
bool handle_ipc_event(struct bar *bar) { bool handle_ipc_event(struct bar *bar) {
struct ipc_response *resp = ipc_recv_response(bar->ipc_event_socketfd); struct ipc_response *resp = ipc_recv_response(bar->ipc_event_socketfd);
if (!resp) {
return false;
}
switch (resp->type) { switch (resp->type) {
case IPC_EVENT_WORKSPACE: case IPC_EVENT_WORKSPACE:
ipc_update_workspaces(bar); ipc_update_workspaces(bar);