diff options
Diffstat (limited to 'shared')
| -rw-r--r-- | shared/file-util.c | 126 | ||||
| -rw-r--r-- | shared/file-util.h | 43 | ||||
| -rw-r--r-- | shared/xalloc.c | 50 | ||||
| -rw-r--r-- | shared/xalloc.h | 51 |
4 files changed, 270 insertions, 0 deletions
diff --git a/shared/file-util.c b/shared/file-util.c new file mode 100644 index 0000000..4170771 --- /dev/null +++ b/shared/file-util.c @@ -0,0 +1,126 @@ +/* + * Copyright © 2015 Collabora, Ltd. + * + * Permission is hereby granted, free of charge, to any person obtaining + * a copy of this software and associated documentation files (the + * "Software"), to deal in the Software without restriction, including + * without limitation the rights to use, copy, modify, merge, publish, + * distribute, sublicense, and/or sell copies of the Software, and to + * permit persons to whom the Software is furnished to do so, subject to + * the following conditions: + * + * The above copyright notice and this permission notice (including the + * next paragraph) shall be included in all copies or substantial + * portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, + * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND + * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS + * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN + * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN + * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ + +#include <stdio.h> +#include <errno.h> +#include <time.h> +#include <sys/types.h> +#include <sys/stat.h> +#include <fcntl.h> +#include <stdlib.h> + +#include "file-util.h" + +static int +current_time_str(char *str, size_t len, const char *fmt) +{ + time_t t; + struct tm *t_local; + int ret; + + t = time(NULL); + t_local = localtime(&t); + if (!t_local) { + errno = ETIME; + return -1; + } + + ret = strftime(str, len, fmt, t_local); + if (ret == 0) { + errno = ETIME; + return -1; + } + + return ret; +} + +static int +create_file_excl(const char *fname) +{ + return open(fname, O_RDWR | O_CLOEXEC | O_CREAT | O_EXCL, 00666); +} + +/** Create a unique file with date and time in the name + * + * \param path File path + * \param prefix File name prefix. + * \param suffix File name suffix. + * \param[out] name_out Buffer for the resulting file name. + * \param name_len Number of bytes usable in name_out. + * \return stdio FILE pointer, or NULL on failure. + * + * Create and open a new file with the name concatenated from + * path/prefix, date and time, and suffix. If a file with this name + * already exists, an counter number is added to the end of the + * date and time sub-string. The counter is increased until a free file + * name is found. + * + * Once creating the file succeeds, the name of the file is in name_out. + * On failure, the contents of name_out are undefined and errno is set. + */ +FILE * +file_create_dated(const char *path, const char *prefix, const char *suffix, + char *name_out, size_t name_len) +{ + char timestr[128]; + int ret; + int fd; + int cnt = 0; + int with_path; + + with_path = path && path[0]; + + if (current_time_str(timestr, sizeof(timestr), "%F_%H-%M-%S") < 0) + return NULL; + + ret = snprintf(name_out, name_len, "%s%s%s%s%s", + with_path ? path : "", with_path ? "/" : "", + prefix, timestr, suffix); + if (ret < 0 || (size_t)ret >= name_len) { + errno = ENOBUFS; + return NULL; + } + + fd = create_file_excl(name_out); + + while (fd == -1 && errno == EEXIST) { + cnt++; + + ret = snprintf(name_out, name_len, "%s%s%s%s-%d%s", + with_path ? path : "", with_path ? "/" : "", + prefix, timestr, cnt, suffix); + if (ret < 0 || (size_t)ret >= name_len) { + errno = ENOBUFS; + return NULL; + } + + fd = create_file_excl(name_out); + } + + if (fd == -1) + return NULL; + + return fdopen(fd, "w"); +} diff --git a/shared/file-util.h b/shared/file-util.h new file mode 100644 index 0000000..e272dde --- /dev/null +++ b/shared/file-util.h @@ -0,0 +1,43 @@ +/* + * Copyright © 2015 Collabora, Ltd. + * + * Permission is hereby granted, free of charge, to any person obtaining + * a copy of this software and associated documentation files (the + * "Software"), to deal in the Software without restriction, including + * without limitation the rights to use, copy, modify, merge, publish, + * distribute, sublicense, and/or sell copies of the Software, and to + * permit persons to whom the Software is furnished to do so, subject to + * the following conditions: + * + * The above copyright notice and this permission notice (including the + * next paragraph) shall be included in all copies or substantial + * portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, + * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND + * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS + * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN + * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN + * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ + +#ifndef WESTON_FILE_UTIL_H +#define WESTON_FILE_UTIL_H + +#ifdef __cplusplus +extern "C" { +#endif + +#include <stdio.h> + +FILE * +file_create_dated(const char *path, const char *prefix, const char *suffix, + char *name_out, size_t name_len); + +#ifdef __cplusplus +} +#endif + +#endif /* WESTON_FILE_UTIL_H */ diff --git a/shared/xalloc.c b/shared/xalloc.c new file mode 100644 index 0000000..1cc5c12 --- /dev/null +++ b/shared/xalloc.c @@ -0,0 +1,50 @@ +/* + * Copyright © 2008 Kristian Høgsberg + * + * Permission is hereby granted, free of charge, to any person obtaining + * a copy of this software and associated documentation files (the + * "Software"), to deal in the Software without restriction, including + * without limitation the rights to use, copy, modify, merge, publish, + * distribute, sublicense, and/or sell copies of the Software, and to + * permit persons to whom the Software is furnished to do so, subject to + * the following conditions: + * + * The above copyright notice and this permission notice (including the + * next paragraph) shall be included in all copies or substantial + * portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, + * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND + * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS + * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN + * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN + * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ + +#include "config.h" + +#include <errno.h> +#include <stdint.h> +#include <stdio.h> +#include <stdlib.h> + +#include "xalloc.h" + +void * +fail_on_null(void *p, size_t size, char *file, int32_t line) +{ + if (p == NULL) { + fprintf(stderr, "[%s] ", program_invocation_short_name); + if (file) + fprintf(stderr, "%s:%d: ", file, line); + fprintf(stderr, "out of memory"); + if (size) + fprintf(stderr, " (%zd)", size); + fprintf(stderr, "\n"); + exit(EXIT_FAILURE); + } + + return p; +} diff --git a/shared/xalloc.h b/shared/xalloc.h new file mode 100644 index 0000000..cd39dd8 --- /dev/null +++ b/shared/xalloc.h @@ -0,0 +1,51 @@ +/* + * Copyright © 2008 Kristian Høgsberg + * + * Permission is hereby granted, free of charge, to any person obtaining + * a copy of this software and associated documentation files (the + * "Software"), to deal in the Software without restriction, including + * without limitation the rights to use, copy, modify, merge, publish, + * distribute, sublicense, and/or sell copies of the Software, and to + * permit persons to whom the Software is furnished to do so, subject to + * the following conditions: + * + * The above copyright notice and this permission notice (including the + * next paragraph) shall be included in all copies or substantial + * portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, + * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND + * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS + * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN + * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN + * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ + +#ifndef WESTON_XALLOC_H +#define WESTON_XALLOC_H + +#ifdef __cplusplus +extern "C" { +#endif + +#include <stdint.h> +#include <stdlib.h> +#include <string.h> + +#include <libweston/zalloc.h> + +void * +fail_on_null(void *p, size_t size, char *file, int32_t line); + +#define xmalloc(s) (fail_on_null(malloc(s), (s), __FILE__, __LINE__)) +#define xzalloc(s) (fail_on_null(zalloc(s), (s), __FILE__, __LINE__)) +#define xstrdup(s) (fail_on_null(strdup(s), 0, __FILE__, __LINE__)) +#define xrealloc(p, s) (fail_on_null(realloc(p, s), (s), __FILE__, __LINE__)) + +#ifdef __cplusplus +} +#endif + +#endif /* WESTON_XALLOC_H */ |