Merge pull request #415 from orestisfl/resolve-tilde

Resolve tilde
This commit is contained in:
Ingo Bürk 2020-05-01 13:17:09 +02:00 committed by GitHub
commit ef03d7c479
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 62 additions and 67 deletions

View File

@ -23,7 +23,6 @@
#include <getopt.h> #include <getopt.h>
#include <signal.h> #include <signal.h>
#include <confuse.h> #include <confuse.h>
#include <glob.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <sys/types.h> #include <sys/types.h>
#include <time.h> #include <time.h>
@ -35,12 +34,6 @@
#include "i3status.h" #include "i3status.h"
#define exit_if_null(pointer, ...) \
{ \
if (pointer == NULL) \
die(__VA_ARGS__); \
}
#define CFG_CUSTOM_ALIGN_OPT \ #define CFG_CUSTOM_ALIGN_OPT \
CFG_STR_CB("align", NULL, CFGF_NONE, parse_align) CFG_STR_CB("align", NULL, CFGF_NONE, parse_align)
@ -106,21 +99,6 @@ static bool path_exists(const char *path) {
return (stat(path, &buf) == 0); return (stat(path, &buf) == 0);
} }
static void *scalloc(size_t size) {
void *result = calloc(size, 1);
exit_if_null(result, "Error: out of memory (calloc(%zu))\n", size);
return result;
}
char *sstrdup(const char *str) {
if (str == NULL) {
return NULL;
}
char *result = strdup(str);
exit_if_null(result, "Error: out of memory (strdup())\n");
return result;
}
/* /*
* Parses the "align" module option (to validate input). * Parses the "align" module option (to validate input).
*/ */
@ -205,39 +183,6 @@ static int valid_color(const char *value) {
return 1; return 1;
} }
/*
* This function resolves ~ in pathnames.
* It may resolve wildcards in the first part of the path, but if no match
* or multiple matches are found, it just returns a copy of path as given.
*
*/
static char *resolve_tilde(const char *path) {
static glob_t globbuf;
char *head, *tail, *result = NULL;
tail = strchr(path, '/');
head = strndup(path, tail ? (size_t)(tail - path) : strlen(path));
int res = glob(head, GLOB_TILDE, NULL, &globbuf);
free(head);
/* no match, or many wildcard matches are bad */
if (res == GLOB_NOMATCH || globbuf.gl_pathc != 1)
result = sstrdup(path);
else if (res != 0) {
die("glob() failed");
} else {
head = globbuf.gl_pathv[0];
result = scalloc(strlen(head) + (tail ? strlen(tail) : 0) + 1);
strcpy(result, head);
if (tail) {
strcat(result, tail);
}
}
globfree(&globbuf);
return result;
}
static char *get_config_path(void) { static char *get_config_path(void) {
char *xdg_config_home, *xdg_config_dirs, *config_path; char *xdg_config_home, *xdg_config_dirs, *config_path;

View File

@ -183,13 +183,14 @@ struct min_width {
const char *str; const char *str;
}; };
char *sstrdup(const char *str);
/* src/general.c */ /* src/general.c */
char *skip_character(char *input, char character, int amount); char *skip_character(char *input, char character, int amount);
void die(const char *fmt, ...) __attribute__((format(printf, 1, 2), noreturn)); void die(const char *fmt, ...) __attribute__((format(printf, 1, 2), noreturn));
bool slurp(const char *filename, char *destination, int size); bool slurp(const char *filename, char *destination, int size);
char *resolve_tilde(const char *path);
void *scalloc(size_t size);
char *sstrdup(const char *str);
/* src/output.c */ /* src/output.c */
void print_separator(const char *separator); void print_separator(const char *separator);

View File

@ -6,11 +6,18 @@
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <unistd.h> #include <unistd.h>
#include <glob.h>
#include <sys/fcntl.h> #include <sys/fcntl.h>
#include <sys/stat.h> #include <sys/stat.h>
#include "i3status.h" #include "i3status.h"
#define exit_if_null(pointer, ...) \
{ \
if (pointer == NULL) \
die(__VA_ARGS__); \
}
/* /*
* Reads size bytes into the destination buffer from filename. * Reads size bytes into the destination buffer from filename.
* *
@ -33,6 +40,54 @@ bool slurp(const char *filename, char *destination, int size) {
return n != -1; return n != -1;
} }
/*
* This function resolves ~ in pathnames.
* It may resolve wildcards in the first part of the path, but if no match
* or multiple matches are found, it just returns a copy of path as given.
*
*/
char *resolve_tilde(const char *path) {
static glob_t globbuf;
char *head, *tail, *result = NULL;
tail = strchr(path, '/');
head = strndup(path, tail ? (size_t)(tail - path) : strlen(path));
int res = glob(head, GLOB_TILDE, NULL, &globbuf);
free(head);
/* no match, or many wildcard matches are bad */
if (res == GLOB_NOMATCH || globbuf.gl_pathc != 1)
result = sstrdup(path);
else if (res != 0) {
die("glob() failed");
} else {
head = globbuf.gl_pathv[0];
result = scalloc(strlen(head) + (tail ? strlen(tail) : 0) + 1);
strcpy(result, head);
if (tail) {
strcat(result, tail);
}
}
globfree(&globbuf);
return result;
}
char *sstrdup(const char *str) {
if (str == NULL) {
return NULL;
}
char *result = strdup(str);
exit_if_null(result, "Error: out of memory (strdup())\n");
return result;
}
void *scalloc(size_t size) {
void *result = calloc(size, 1);
exit_if_null(result, "Error: out of memory (calloc(%zu))\n", size);
return result;
}
/* /*
* Skip the given character for exactly 'amount' times, returns * Skip the given character for exactly 'amount' times, returns
* a pointer to the first non-'character' character in 'input'. * a pointer to the first non-'character' character in 'input'.

View File

@ -14,24 +14,18 @@
#define STRING_SIZE 10 #define STRING_SIZE 10
static void *scalloc(size_t size) {
void *result = calloc(size, 1);
if (result == NULL) {
die("Error: out of memory (calloc(%zu))\n", size);
}
return result;
}
void print_file_contents(yajl_gen json_gen, char *buffer, const char *title, const char *path, const char *format, const char *format_bad, const int max_chars) { void print_file_contents(yajl_gen json_gen, char *buffer, const char *title, const char *path, const char *format, const char *format_bad, const int max_chars) {
const char *walk = format; const char *walk = format;
char *outwalk = buffer; char *outwalk = buffer;
char *buf = scalloc(max_chars * sizeof(char) + 1); char *buf = scalloc(max_chars * sizeof(char) + 1);
int n = -1; char *abs_path = resolve_tilde(path);
int fd = open(path, O_RDONLY); int fd = open(abs_path, O_RDONLY);
free(abs_path);
INSTANCE(path); INSTANCE(path);
int n = -1;
if (fd > -1) { if (fd > -1) {
n = read(fd, buf, max_chars); n = read(fd, buf, max_chars);
if (n != -1) { if (n != -1) {