From 885158c887384c6da247cc061458f2e53367e6b5 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Sun, 16 Oct 2011 15:21:58 +0200 Subject: Move av_tempfile() into libavutil, it is a generically usefull thing and its small. Signed-off-by: Michael Niedermayer --- libavutil/file.c | 31 +++++++++++++++++++++++++++++++ libavutil/file.h | 9 +++++++++ 2 files changed, 40 insertions(+) (limited to 'libavutil') diff --git a/libavutil/file.c b/libavutil/file.c index 31a3b7564f..4d14292ba1 100644 --- a/libavutil/file.c +++ b/libavutil/file.c @@ -130,6 +130,37 @@ void av_file_unmap(uint8_t *bufptr, size_t size) #endif } +int av_tempfile(const char *prefix, char **filename) { + int fd=-1; +#if !HAVE_MKSTEMP + *filename = tempnam(".", prefix); +#else + size_t len = strlen(prefix) + 12; /* room for "/tmp/" and "XXXXXX\0" */ + *filename = av_malloc(len); +#endif + /* -----common section-----*/ + if (*filename == NULL) { + av_log(NULL, AV_LOG_ERROR, "ff_tempfile: Cannot allocate file name\n"); + return -1; + } +#if !HAVE_MKSTEMP + fd = open(*filename, O_RDWR | O_BINARY | O_CREAT, 0444); +#else + snprintf(*filename, len, "/tmp/%sXXXXXX", prefix); + fd = mkstemp(*filename); + if (fd < 0) { + snprintf(*filename, len, "./%sXXXXXX", prefix); + fd = mkstemp(*filename); + } +#endif + /* -----common section-----*/ + if (fd < 0) { + av_log(NULL, AV_LOG_ERROR, "ff_tempfile: Cannot open temporary file %s\n", *filename); + return -1; + } + return fd; /* success */ +} + #ifdef TEST #undef printf diff --git a/libavutil/file.h b/libavutil/file.h index f28627c9d8..c6d2692d52 100644 --- a/libavutil/file.h +++ b/libavutil/file.h @@ -49,4 +49,13 @@ int av_file_map(const char *filename, uint8_t **bufptr, size_t *size, */ void av_file_unmap(uint8_t *bufptr, size_t size); +/** + * Wrapper to work around the lack of mkstemp() on mingw. + * Also, tries to create file in /tmp first, if possible. + * *prefix can be a character constant; *filename will be allocated internally. + * @return file descriptor of opened file (or -1 on error) + * and opened file name in **filename. + */ +int av_tempfile(const char *prefix, char **filename); + #endif /* AVUTIL_FILE_H */ -- cgit v1.2.3