]> git.sesse.net Git - vlc/blob - src/extras/dirent.c
Support for UDP-Lite (with full checksum coverage only atm)
[vlc] / src / extras / dirent.c
1 /*
2  * dirent.c
3  *
4  * Derived from DIRLIB.C by Matt J. Weinstein 
5  * This note appears in the DIRLIB.H
6  * DIRLIB.H by M. J. Weinstein   Released to public domain 1-Jan-89
7  *
8  * Updated by Jeremy Bettis <jeremy@hksys.com>
9  * Significantly revised and rewinddir, seekdir and telldir added by Colin
10  * Peters <colin@fu.is.saga-u.ac.jp>
11  *      
12  * $Revision: 1.6 $
13  * $Author: sam $
14  * $Date: 2002/11/13 20:51:04 $
15  *
16  */
17
18 #include "config.h"
19
20 #include <stdlib.h>
21 #include <stdio.h>
22
23 #ifdef HAVE_ERRNO_H
24 #   include <errno.h>
25 #else
26     static int errno;
27     /* FIXME: anything clever to put here? */
28 #   define EFAULT 12
29 #   define ENOTDIR 12
30 #   define ENOENT 12
31 #   define ENOMEM 12
32 #   define EINVAL 12
33 #endif
34 #include <string.h>
35 #ifndef UNDER_CE
36 #   include <io.h>
37 #   include <direct.h>
38 #else
39 #   define FILENAME_MAX (260)
40 #endif
41
42 #define WIN32_LEAN_AND_MEAN
43 #include <windows.h> /* for GetFileAttributes */
44
45 #include <tchar.h>
46 #define SUFFIX  "*"
47 #define SLASH   "\\"
48
49 struct dirent
50 {
51         long            d_ino;          /* Always zero. */
52         unsigned short  d_reclen;       /* Always zero. */
53         unsigned short  d_namlen;       /* Length of name in d_name. */
54         char            d_name[FILENAME_MAX]; /* File name. */
55 };
56
57 typedef struct
58 {
59         /* disk transfer area for this dir */
60         WIN32_FIND_DATA         dd_dta;
61
62         /* dirent struct to return from dir (NOTE: this makes this thread
63          * safe as long as only one thread uses a particular DIR struct at
64          * a time) */
65         struct dirent           dd_dir;
66
67         /* findnext handle */
68         HANDLE                  dd_handle;
69
70         /*
71          * Status of search:
72          *   0 = not started yet (next entry to read is first entry)
73          *  -1 = off the end
74          *   positive = 0 based index of next entry
75          */
76         int                     dd_stat;
77
78         /* given path for dir with search pattern (struct is extended) */
79         char                    dd_name[1];
80 } DIR;
81
82 /*
83  * opendir
84  *
85  * Returns a pointer to a DIR structure appropriately filled in to begin
86  * searching a directory.
87  */
88 DIR * 
89 vlc_opendir (const CHAR *szPath)
90 {
91   DIR *nd;
92   unsigned int rc;
93   CHAR szFullPath[MAX_PATH];
94
95   errno = 0;
96
97   if (!szPath)
98     {
99       errno = EFAULT;
100       return (DIR *) 0;
101     }
102
103   if (szPath[0] == '\0')
104     {
105       errno = ENOTDIR;
106       return (DIR *) 0;
107     }
108
109   /* Attempt to determine if the given path really is a directory. */
110 #ifdef UNICODE
111   {
112     wchar_t szPathTmp[MAX_PATH];
113     mbstowcs( szPathTmp, szPath, MAX_PATH );
114     szPathTmp[MAX_PATH-1] = 0;
115     rc = GetFileAttributes (szPathTmp);
116   }
117 #else
118   rc = GetFileAttributes (szPath);
119 #endif
120   if (rc == (unsigned int)-1)
121     {
122       /* call GetLastError for more error info */
123       errno = ENOENT;
124       return (DIR *) 0;
125     }
126   if (!(rc & FILE_ATTRIBUTE_DIRECTORY))
127     {
128       /* Error, entry exists but not a directory. */
129       errno = ENOTDIR;
130       return (DIR *) 0;
131     }
132
133   /* Make an absolute pathname.  */
134 #if defined( UNDER_CE )
135   if (szPath[0] == '\\' || szPath[0] == '/')
136     {
137       sprintf (szFullPath, "%s", szPath);
138       szFullPath[0] = '\\';
139     }
140   else
141     {
142       wchar_t szFullPathTmp[MAX_PATH];
143       if (GetModuleFileName( NULL, szFullPathTmp, MAX_PATH ) )
144         {
145           wcstombs( szFullPath, szFullPathTmp, MAX_PATH );
146           szFullPath[MAX_PATH-1] = 0;
147         }
148       else
149         {
150           /* FIXME: if I wasn't lazy, I'd check for overflows here. */
151           sprintf (szFullPath, "\\%s", szPath );
152         }
153     }
154 #else
155   _fullpath (szFullPath, szPath, MAX_PATH);
156 #endif
157
158   /* Allocate enough space to store DIR structure and the complete
159    * directory path given. */
160   nd = (DIR *) malloc (sizeof (DIR) + strlen (szFullPath) + sizeof (SLASH) +
161                        sizeof (SUFFIX));
162
163   if (!nd)
164     {
165       /* Error, out of memory. */
166       errno = ENOMEM;
167       return (DIR *) 0;
168     }
169
170   /* Create the search expression. */
171   strcpy (nd->dd_name, szFullPath);
172
173   /* Add on a slash if the path does not end with one. */
174   if (nd->dd_name[0] != '\0' &&
175       nd->dd_name[strlen (nd->dd_name) - 1] != '/' &&
176       nd->dd_name[strlen (nd->dd_name) - 1] != '\\')
177     {
178       strcat (nd->dd_name, SLASH);
179     }
180
181   /* Add on the search pattern */
182   strcat (nd->dd_name, SUFFIX);
183
184   /* Initialize handle so that a premature closedir doesn't try
185    * to call FindClose on it. */
186   nd->dd_handle = INVALID_HANDLE_VALUE;
187
188   /* Initialize the status. */
189   nd->dd_stat = 0;
190
191   /* Initialize the dirent structure. ino and reclen are invalid under
192    * Win32, and name simply points at the appropriate part of the
193    * findfirst_t structure. */
194   nd->dd_dir.d_ino = 0;
195   nd->dd_dir.d_reclen = 0;
196   nd->dd_dir.d_namlen = 0;
197   memset (nd->dd_dir.d_name, 0, FILENAME_MAX);
198
199   return nd;
200 }
201
202
203 /*
204  * readdir
205  *
206  * Return a pointer to a dirent structure filled with the information on the
207  * next entry in the directory.
208  */
209 struct dirent *
210 vlc_readdir (DIR * dirp)
211 {
212   errno = 0;
213
214   /* Check for valid DIR struct. */
215   if (!dirp)
216     {
217       errno = EFAULT;
218       return (struct dirent *) 0;
219     }
220
221   if (dirp->dd_stat < 0)
222     {
223       /* We have already returned all files in the directory
224        * (or the structure has an invalid dd_stat). */
225       return (struct dirent *) 0;
226     }
227   else if (dirp->dd_stat == 0)
228     {
229 #ifdef UNICODE
230         wchar_t dd_name[MAX_PATH];
231         mbstowcs( dd_name, dirp->dd_name, MAX_PATH );
232         dd_name[MAX_PATH-1] = 0;
233 #else
234         char *dd_name = dirp->dd_name;
235 #endif
236       /* We haven't started the search yet. */
237       /* Start the search */
238       dirp->dd_handle = FindFirstFile (dd_name, &(dirp->dd_dta));
239
240           if (dirp->dd_handle == INVALID_HANDLE_VALUE)
241         {
242           /* Whoops! Seems there are no files in that
243            * directory. */
244           dirp->dd_stat = -1;
245         }
246       else
247         {
248           dirp->dd_stat = 1;
249         }
250     }
251   else
252     {
253       /* Get the next search entry. */
254       if (!FindNextFile ((HANDLE)dirp->dd_handle, &(dirp->dd_dta)))
255         {
256           /* We are off the end or otherwise error. */
257           FindClose ((HANDLE)dirp->dd_handle);
258           dirp->dd_handle = INVALID_HANDLE_VALUE;
259           dirp->dd_stat = -1;
260         }
261       else
262         {
263           /* Update the status to indicate the correct
264            * number. */
265           dirp->dd_stat++;
266         }
267     }
268
269   if (dirp->dd_stat > 0)
270     {
271       /* Successfully got an entry */
272
273 #ifdef UNICODE
274       char d_name[MAX_PATH];
275       wcstombs( d_name, dirp->dd_dta.cFileName, MAX_PATH );
276       d_name[MAX_PATH-1] = 0;
277 #else
278       char *d_name = dirp->dd_dta.cFileName;
279 #endif
280
281       strcpy (dirp->dd_dir.d_name, d_name);
282       dirp->dd_dir.d_namlen = strlen (dirp->dd_dir.d_name);
283       return &dirp->dd_dir;
284     }
285
286   return (struct dirent *) 0;
287 }
288
289
290 /*
291  * closedir
292  *
293  * Frees up resources allocated by opendir.
294  */
295 int
296 vlc_closedir (DIR * dirp)
297 {
298   int rc;
299
300   errno = 0;
301   rc = 0;
302
303   if (!dirp)
304     {
305       errno = EFAULT;
306       return -1;
307     }
308
309   if (dirp->dd_handle != INVALID_HANDLE_VALUE)
310     {
311       rc = FindClose ((HANDLE)dirp->dd_handle);
312     }
313
314   /* Delete the dir structure. */
315   free (dirp);
316
317   return rc;
318 }
319
320 /*
321  * rewinddir
322  *
323  * Return to the beginning of the directory "stream". We simply call findclose
324  * and then reset things like an opendir.
325  */
326 void
327 vlc_rewinddir (DIR * dirp)
328 {
329   errno = 0;
330
331   if (!dirp)
332     {
333       errno = EFAULT;
334       return;
335     }
336
337   if (dirp->dd_handle != INVALID_HANDLE_VALUE)
338     {
339       FindClose ((HANDLE)dirp->dd_handle);
340     }
341
342   dirp->dd_handle = INVALID_HANDLE_VALUE;
343   dirp->dd_stat = 0;
344 }
345
346 /*
347  * telldir
348  *
349  * Returns the "position" in the "directory stream" which can be used with
350  * seekdir to go back to an old entry. We simply return the value in stat.
351  */
352 long
353 vlc_telldir (DIR * dirp)
354 {
355   errno = 0;
356
357   if (!dirp)
358     {
359       errno = EFAULT;
360       return -1;
361     }
362   return dirp->dd_stat;
363 }
364
365 /*
366  * seekdir
367  *
368  * Seek to an entry previously returned by telldir. We rewind the directory
369  * and call readdir repeatedly until either dd_stat is the position number
370  * or -1 (off the end). This is not perfect, in that the directory may
371  * have changed while we weren't looking. But that is probably the case with
372  * any such system.
373  */
374 void
375 vlc_seekdir (DIR * dirp, long lPos)
376 {
377   errno = 0;
378
379   if (!dirp)
380     {
381       errno = EFAULT;
382       return;
383     }
384
385   if (lPos < -1)
386     {
387       /* Seeking to an invalid position. */
388       errno = EINVAL;
389       return;
390     }
391   else if (lPos == -1)
392     {
393       /* Seek past end. */
394       if (dirp->dd_handle != INVALID_HANDLE_VALUE)
395         {
396           FindClose ((HANDLE)dirp->dd_handle);
397         }
398       dirp->dd_handle = INVALID_HANDLE_VALUE;
399       dirp->dd_stat = -1;
400     }
401   else
402     {
403       /* Rewind and read forward to the appropriate index. */
404       vlc_rewinddir (dirp);
405
406       while ((dirp->dd_stat < lPos) && vlc_readdir (dirp))
407         ;
408     }
409 }