Introduce sstrndup wrapper.

This commit is contained in:
Michael Hofmann 2015-05-06 16:28:29 +02:00
parent 37bee99538
commit 0319bda1d4
4 changed files with 16 additions and 2 deletions

View File

@ -106,7 +106,7 @@ static int workspaces_string_cb(void *params_, const unsigned char *val, size_t
if (!strcmp(params->cur_key, "name")) {
const char *ws_name = (const char *)val;
params->workspaces_walk->canonical_name = strndup(ws_name, len);
params->workspaces_walk->canonical_name = sstrndup(ws_name, len);
if (config.strip_ws_numbers && params->workspaces_walk->num >= 0) {
/* Special case: strip off the workspace number */

View File

@ -127,6 +127,13 @@ void *srealloc(void *ptr, size_t size);
*/
char *sstrdup(const char *str);
/**
* Safe-wrapper around strndup which exits if strndup returns NULL (meaning that
* there is no more memory available)
*
*/
char *sstrndup(const char *str, size_t size);
/**
* Safe-wrapper around asprintf which exits if it returns -1 (meaning that
* there is no more memory available)

View File

@ -23,7 +23,7 @@ char *resolve_tilde(const char *path) {
char *head, *tail, *result;
tail = strchr(path, '/');
head = strndup(path, tail ? (size_t)(tail - path) : strlen(path));
head = sstrndup(path, tail ? (size_t)(tail - path) : strlen(path));
int res = glob(head, GLOB_TILDE, NULL, &globbuf);
free(head);

View File

@ -48,6 +48,13 @@ char *sstrdup(const char *str) {
return result;
}
char *sstrndup(const char *str, size_t size) {
char *result = strndup(str, size);
if (result == NULL)
err(EXIT_FAILURE, "strndup()");
return result;
}
int sasprintf(char **strp, const char *fmt, ...) {
va_list args;
int result;