256890b2b94f02c7131d467954957a7fd4f2bd40
[ffmpeg.git] / libavformat / os_support.h
1 /*
2  * various OS-feature replacement utilities
3  * copyright (c) 2000, 2001, 2002 Fabrice Bellard
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #ifndef AVFORMAT_OS_SUPPORT_H
23 #define AVFORMAT_OS_SUPPORT_H
24
25 /**
26  * @file
27  * miscellaneous OS support macros and functions.
28  */
29
30 #include "config.h"
31
32 #include <sys/stat.h>
33
34 #ifdef _WIN32
35 #if HAVE_DIRECT_H
36 #include <direct.h>
37 #endif
38 #if HAVE_IO_H
39 #include <io.h>
40 #endif
41 #endif
42
43 #if defined(_WIN32) && !defined(__MINGW32CE__)
44 #  include <fcntl.h>
45 #  ifdef lseek
46 #   undef lseek
47 #  endif
48 #  define lseek(f,p,w) _lseeki64((f), (p), (w))
49 #  ifdef stat
50 #   undef stat
51 #  endif
52 #  define stat _stati64
53 #  ifdef fstat
54 #   undef fstat
55 #  endif
56 #  define fstat(f,s) _fstati64((f), (s))
57 #endif /* defined(_WIN32) && !defined(__MINGW32CE__) */
58
59
60 #ifdef __ANDROID__
61 #  if HAVE_UNISTD_H
62 #    include <unistd.h>
63 #  endif
64 #  ifdef lseek
65 #   undef lseek
66 #  endif
67 #  define lseek(f,p,w) lseek64((f), (p), (w))
68 #endif
69
70 static inline int is_dos_path(const char *path)
71 {
72 #if HAVE_DOS_PATHS
73     if (path[0] && path[1] == ':')
74         return 1;
75 #endif
76     return 0;
77 }
78
79 #if defined(__OS2__) || defined(__Plan9__)
80 #define SHUT_RD 0
81 #define SHUT_WR 1
82 #define SHUT_RDWR 2
83 #endif
84
85 #if defined(_WIN32)
86 #define SHUT_RD SD_RECEIVE
87 #define SHUT_WR SD_SEND
88 #define SHUT_RDWR SD_BOTH
89
90 #ifndef S_IRUSR
91 #define S_IRUSR S_IREAD
92 #endif
93 #ifndef S_IWUSR
94 #define S_IWUSR S_IWRITE
95 #endif
96 #endif
97
98 #if CONFIG_NETWORK
99 #if !HAVE_SOCKLEN_T
100 typedef int socklen_t;
101 #endif
102
103 /* most of the time closing a socket is just closing an fd */
104 #if !HAVE_CLOSESOCKET
105 #define closesocket close
106 #endif
107
108 #if !HAVE_POLL_H
109 typedef unsigned long nfds_t;
110
111 #if HAVE_WINSOCK2_H
112 #include <winsock2.h>
113 #endif
114 #if !HAVE_STRUCT_POLLFD
115 struct pollfd {
116     int fd;
117     short events;  /* events to look for */
118     short revents; /* events that occurred */
119 };
120
121 /* events & revents */
122 #define POLLIN     0x0001  /* any readable data available */
123 #define POLLOUT    0x0002  /* file descriptor is writeable */
124 #define POLLRDNORM POLLIN
125 #define POLLWRNORM POLLOUT
126 #define POLLRDBAND 0x0008  /* priority readable data */
127 #define POLLWRBAND 0x0010  /* priority data can be written */
128 #define POLLPRI    0x0020  /* high priority readable data */
129
130 /* revents only */
131 #define POLLERR    0x0004  /* errors pending */
132 #define POLLHUP    0x0080  /* disconnected */
133 #define POLLNVAL   0x1000  /* invalid file descriptor */
134 #endif
135
136
137 int ff_poll(struct pollfd *fds, nfds_t numfds, int timeout);
138 #define poll ff_poll
139 #endif /* HAVE_POLL_H */
140 #endif /* CONFIG_NETWORK */
141
142 #if defined(__MINGW32CE__)
143 #define mkdir(a, b) _mkdir(a)
144 #elif defined(_WIN32)
145 #include <stdio.h>
146 #include <windows.h>
147 #include "libavutil/mem.h"
148
149 static inline int utf8towchar(const char *filename_utf8, wchar_t **filename_w)
150 {
151     int num_chars;
152     num_chars = MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, filename_utf8, -1, NULL, 0);
153     if (num_chars <= 0) {
154         *filename_w = NULL;
155         return 0;
156     }
157     *filename_w = av_mallocz(sizeof(wchar_t) * num_chars);
158     if (!*filename_w) {
159         errno = ENOMEM;
160         return -1;
161     }
162     MultiByteToWideChar(CP_UTF8, 0, filename_utf8, -1, *filename_w, num_chars);
163     return 0;
164 }
165
166 #define DEF_FS_FUNCTION(name, wfunc, afunc)               \
167 static inline int win32_##name(const char *filename_utf8) \
168 {                                                         \
169     wchar_t *filename_w;                                  \
170     int ret;                                              \
171                                                           \
172     if (utf8towchar(filename_utf8, &filename_w))          \
173         return -1;                                        \
174     if (!filename_w)                                      \
175         goto fallback;                                    \
176                                                           \
177     ret = wfunc(filename_w);                              \
178     av_free(filename_w);                                  \
179     return ret;                                           \
180                                                           \
181 fallback:                                                 \
182     /* filename may be be in CP_ACP */                    \
183     return afunc(filename_utf8);                          \
184 }
185
186 DEF_FS_FUNCTION(unlink, _wunlink, _unlink)
187 DEF_FS_FUNCTION(mkdir,  _wmkdir,  _mkdir)
188 DEF_FS_FUNCTION(rmdir,  _wrmdir , _rmdir)
189
190 static inline int win32_rename(const char *src_utf8, const char *dest_utf8)
191 {
192     wchar_t *src_w, *dest_w;
193     int ret;
194
195     if (utf8towchar(src_utf8, &src_w))
196         return -1;
197     if (utf8towchar(dest_utf8, &dest_w)) {
198         av_free(src_w);
199         return -1;
200     }
201     if (!src_w || !dest_w) {
202         av_free(src_w);
203         av_free(dest_w);
204         goto fallback;
205     }
206
207     ret = _wrename(src_w, dest_w);
208     av_free(src_w);
209     av_free(dest_w);
210     return ret;
211
212 fallback:
213     /* filename may be be in CP_ACP */
214     return rename(src_utf8, dest_utf8);
215 }
216
217 #define mkdir(a, b) win32_mkdir(a)
218 #define rename      win32_rename
219 #define rmdir       win32_rmdir
220 #define unlink      win32_unlink
221
222 #endif
223
224 #endif /* AVFORMAT_OS_SUPPORT_H */