| /* |
| * This file is part of FFmpeg. |
| * |
| * FFmpeg is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * FFmpeg is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with FFmpeg; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| */ |
| |
| #include "config.h" |
| #include "internal.h" |
| #include "mem.h" |
| #include <stdarg.h> |
| #include <fcntl.h> |
| #include <sys/stat.h> |
| #if HAVE_UNISTD_H |
| #include <unistd.h> |
| #endif |
| #if HAVE_IO_H |
| #include <io.h> |
| #endif |
| |
| #ifdef _WIN32 |
| #undef open |
| #undef lseek |
| #undef stat |
| #undef fstat |
| #include <windows.h> |
| #include <share.h> |
| #include <errno.h> |
| #include "wchar_filename.h" |
| |
| static int win32_open(const char *filename_utf8, int oflag, int pmode) |
| { |
| int fd; |
| wchar_t *filename_w; |
| |
| /* convert UTF-8 to wide chars */ |
| if (utf8towchar(filename_utf8, &filename_w)) |
| return -1; |
| if (!filename_w) |
| goto fallback; |
| |
| fd = _wsopen(filename_w, oflag, SH_DENYNO, pmode); |
| av_freep(&filename_w); |
| |
| if (fd != -1 || (oflag & O_CREAT)) |
| return fd; |
| |
| fallback: |
| /* filename may be in CP_ACP */ |
| return _sopen(filename_utf8, oflag, SH_DENYNO, pmode); |
| } |
| #define open win32_open |
| #endif |
| |
| int avpriv_open(const char *filename, int flags, ...) |
| { |
| int fd; |
| unsigned int mode = 0; |
| va_list ap; |
| |
| va_start(ap, flags); |
| if (flags & O_CREAT) |
| mode = va_arg(ap, unsigned int); |
| va_end(ap); |
| |
| #ifdef O_CLOEXEC |
| flags |= O_CLOEXEC; |
| #endif |
| #ifdef O_NOINHERIT |
| flags |= O_NOINHERIT; |
| #endif |
| |
| fd = open(filename, flags, mode); |
| #if HAVE_FCNTL |
| if (fd != -1) { |
| if (fcntl(fd, F_SETFD, FD_CLOEXEC) == -1) |
| av_log(NULL, AV_LOG_DEBUG, "Failed to set close on exec\n"); |
| } |
| #endif |
| |
| return fd; |
| } |
| |
| typedef struct FileLogContext { |
| const AVClass *class; |
| int log_offset; |
| void *log_ctx; |
| } FileLogContext; |
| |
| static const AVClass file_log_ctx_class = { |
| .class_name = "TEMPFILE", |
| .item_name = av_default_item_name, |
| .option = NULL, |
| .version = LIBAVUTIL_VERSION_INT, |
| .log_level_offset_offset = offsetof(FileLogContext, log_offset), |
| .parent_log_context_offset = offsetof(FileLogContext, log_ctx), |
| }; |
| |
| int avpriv_tempfile(const char *prefix, char **filename, int log_offset, void *log_ctx) |
| { |
| FileLogContext file_log_ctx = { &file_log_ctx_class, log_offset, log_ctx }; |
| int fd = -1; |
| #if !HAVE_MKSTEMP |
| void *ptr= tempnam(NULL, prefix); |
| if(!ptr) |
| ptr= tempnam(".", prefix); |
| *filename = av_strdup(ptr); |
| #undef free |
| free(ptr); |
| #else |
| size_t len = strlen(prefix) + 12; /* room for "/tmp/" and "XXXXXX\0" */ |
| *filename = av_malloc(len); |
| #endif |
| /* -----common section-----*/ |
| if (!*filename) { |
| av_log(&file_log_ctx, AV_LOG_ERROR, "ff_tempfile: Cannot allocate file name\n"); |
| return AVERROR(ENOMEM); |
| } |
| #if !HAVE_MKSTEMP |
| # ifndef O_BINARY |
| # define O_BINARY 0 |
| # endif |
| # ifndef O_EXCL |
| # define O_EXCL 0 |
| # endif |
| fd = open(*filename, O_RDWR | O_BINARY | O_CREAT | O_EXCL, 0600); |
| #else |
| snprintf(*filename, len, "/tmp/%sXXXXXX", prefix); |
| fd = mkstemp(*filename); |
| #if defined(_WIN32) || defined (__ANDROID__) || defined(__DJGPP__) |
| if (fd < 0) { |
| snprintf(*filename, len, "./%sXXXXXX", prefix); |
| fd = mkstemp(*filename); |
| } |
| #endif |
| #endif |
| /* -----common section-----*/ |
| if (fd < 0) { |
| int err = AVERROR(errno); |
| av_log(&file_log_ctx, AV_LOG_ERROR, "ff_tempfile: Cannot open temporary file %s\n", *filename); |
| av_freep(filename); |
| return err; |
| } |
| return fd; /* success */ |
| } |
| |
| FILE *av_fopen_utf8(const char *path, const char *mode) |
| { |
| int fd; |
| int access; |
| const char *m = mode; |
| |
| switch (*m++) { |
| case 'r': access = O_RDONLY; break; |
| case 'w': access = O_CREAT|O_WRONLY|O_TRUNC; break; |
| case 'a': access = O_CREAT|O_WRONLY|O_APPEND; break; |
| default : |
| errno = EINVAL; |
| return NULL; |
| } |
| while (*m) { |
| if (*m == '+') { |
| access &= ~(O_RDONLY | O_WRONLY); |
| access |= O_RDWR; |
| } else if (*m == 'b') { |
| #ifdef O_BINARY |
| access |= O_BINARY; |
| #endif |
| } else if (*m) { |
| errno = EINVAL; |
| return NULL; |
| } |
| m++; |
| } |
| fd = avpriv_open(path, access, 0666); |
| if (fd == -1) |
| return NULL; |
| return fdopen(fd, mode); |
| } |