filesystem.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:12k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * filesystem.c: File system helpers
  3.  *****************************************************************************
  4.  * Copyright (C) 2005-2006 the VideoLAN team
  5.  * Copyright © 2005-2008 Rémi Denis-Courmont
  6.  *
  7.  * Authors: Rémi Denis-Courmont <rem # videolan.org>
  8.  *
  9.  * This program is free software; you can redistribute it and/or modify
  10.  * it under the terms of the GNU General Public License as published by
  11.  * the Free Software Foundation; either version 2 of the License, or
  12.  * (at your option) any later version.
  13.  *
  14.  * This program is distributed in the hope that it will be useful,
  15.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17.  * GNU General Public License for more details.
  18.  *
  19.  * You should have received a copy of the GNU General Public License
  20.  * along with this program; if not, write to the Free Software
  21.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  26. #ifdef HAVE_CONFIG_H
  27. # include "config.h"
  28. #endif
  29. #include <vlc_common.h>
  30. #include <vlc_charset.h>
  31. #include "libvlc.h" /* utf8_mkdir */
  32. #include <vlc_rand.h>
  33. #include <assert.h>
  34. #include <stdio.h>
  35. #include <errno.h>
  36. #include <sys/types.h>
  37. #ifdef HAVE_DIRENT_H
  38. #  include <dirent.h>
  39. #endif
  40. #ifdef UNDER_CE
  41. #  include <tchar.h>
  42. #endif
  43. #ifdef HAVE_SYS_STAT_H
  44. # include <sys/stat.h>
  45. #endif
  46. #ifdef HAVE_FCNTL_H
  47. # include <fcntl.h>
  48. #endif
  49. #ifdef WIN32
  50. # include <io.h>
  51. # ifndef UNDER_CE
  52. #  include <direct.h>
  53. # endif
  54. #else
  55. # include <unistd.h>
  56. #endif
  57. #ifndef HAVE_LSTAT
  58. # define lstat( a, b ) stat(a, b)
  59. #endif
  60. #ifdef WIN32
  61. static int convert_path (const char *restrict path, wchar_t *restrict wpath)
  62. {
  63.     if (!MultiByteToWideChar (CP_UTF8, 0, path, -1, wpath, MAX_PATH))
  64.     {
  65.         errno = ENOENT;
  66.         return -1;
  67.     }
  68.     wpath[MAX_PATH] = L'';
  69.     return 0;
  70. }
  71. # define CONVERT_PATH(path, wpath, err) 
  72.   wchar_t wpath[MAX_PATH+1]; 
  73.   if (convert_path (path, wpath)) 
  74.       return (err)
  75. #endif
  76. /**
  77.  * Opens a system file handle.
  78.  *
  79.  * @param filename file path to open (with UTF-8 encoding)
  80.  * @param flags open() flags, see the C library open() documentation
  81.  * @param mode file permissions if creating a new file
  82.  * @return a file handle on success, -1 on error (see errno).
  83.  */
  84. int utf8_open (const char *filename, int flags, mode_t mode)
  85. {
  86. #ifdef UNDER_CE
  87.     /*_open translates to wchar internally on WinCE*/
  88.     return _open (filename, flags, mode);
  89. #elif defined (WIN32)
  90.     /*
  91.      * open() cannot open files with non-“ANSI” characters on Windows.
  92.      * We use _wopen() instead. Same thing for mkdir() and stat().
  93.      */
  94.     CONVERT_PATH(filename, wpath, -1);
  95.     return _wopen (wpath, flags, mode);
  96. #endif
  97.     const char *local_name = ToLocale (filename);
  98.     if (local_name == NULL)
  99.     {
  100.         errno = ENOENT;
  101.         return -1;
  102.     }
  103.     int fd;
  104. #ifdef O_CLOEXEC
  105.     fd = open (local_name, flags | O_CLOEXEC, mode);
  106.     if (fd == -1 && errno == EINVAL)
  107. #endif
  108.     {
  109.         fd = open (local_name, flags, mode);
  110. #ifdef HAVE_FCNTL
  111.         if (fd != -1)
  112.         {
  113.             int flags = fcntl (fd, F_GETFD);
  114.             fcntl (fd, F_SETFD, FD_CLOEXEC | ((flags != -1) ? flags : 0));
  115.         }
  116. #endif
  117.     }
  118.     LocaleFree (local_name);
  119.     return fd;
  120. }
  121. /**
  122.  * Opens a FILE pointer.
  123.  * @param filename file path, using UTF-8 encoding
  124.  * @param mode fopen file open mode
  125.  * @return NULL on error, an open FILE pointer on success.
  126.  */
  127. FILE *utf8_fopen (const char *filename, const char *mode)
  128. {
  129.     int rwflags = 0, oflags = 0;
  130.     bool append = false;
  131.     for (const char *ptr = mode; *ptr; ptr++)
  132.     {
  133.         switch (*ptr)
  134.         {
  135.             case 'r':
  136.                 rwflags = O_RDONLY;
  137.                 break;
  138.             case 'a':
  139.                 rwflags = O_WRONLY;
  140.                 oflags |= O_CREAT;
  141.                 append = true;
  142.                 break;
  143.             case 'w':
  144.                 rwflags = O_WRONLY;
  145.                 oflags |= O_CREAT | O_TRUNC;
  146.                 break;
  147.             case '+':
  148.                 rwflags = O_RDWR;
  149.                 break;
  150. #ifdef O_TEXT
  151.             case 't':
  152.                 oflags |= O_TEXT;
  153.                 break;
  154. #endif
  155.         }
  156.     }
  157.     int fd = utf8_open (filename, rwflags | oflags, 0666);
  158.     if (fd == -1)
  159.         return NULL;
  160.     if (append && (lseek (fd, 0, SEEK_END) == -1))
  161.     {
  162.         close (fd);
  163.         return NULL;
  164.     }
  165.     FILE *stream = fdopen (fd, mode);
  166.     if (stream == NULL)
  167.         close (fd);
  168.     return stream;
  169. }
  170. /**
  171.  * Creates a directory using UTF-8 paths.
  172.  *
  173.  * @param dirname a UTF-8 string with the name of the directory that you
  174.  *        want to create.
  175.  * @param mode directory permissions
  176.  * @return 0 on success, -1 on error (see errno).
  177.  */
  178. int utf8_mkdir( const char *dirname, mode_t mode )
  179. {
  180. #if defined (UNDER_CE)
  181.     (void) mode;
  182.     /* mkdir converts internally to wchar */
  183.     return _mkdir(dirname);
  184. #elif defined (WIN32)
  185.     (void) mode;
  186.     CONVERT_PATH (dirname, wpath, -1);
  187.     return _wmkdir (wpath);
  188. #else
  189.     char *locname = ToLocale( dirname );
  190.     int res;
  191.     if( locname == NULL )
  192.     {
  193.         errno = ENOENT;
  194.         return -1;
  195.     }
  196.     res = mkdir( locname, mode );
  197.     LocaleFree( locname );
  198.     return res;
  199. #endif
  200. }
  201. /**
  202.  * Opens a DIR pointer.
  203.  *
  204.  * @param dirname UTF-8 representation of the directory name
  205.  * @return a pointer to the DIR struct, or NULL in case of error.
  206.  * Release with standard closedir().
  207.  */
  208. DIR *utf8_opendir( const char *dirname )
  209. {
  210. #ifdef WIN32
  211.     CONVERT_PATH (dirname, wpath, NULL);
  212.     return (DIR *)vlc_wopendir (wpath);
  213. #else
  214.     const char *local_name = ToLocale( dirname );
  215.     if( local_name != NULL )
  216.     {
  217.         DIR *dir = opendir( local_name );
  218.         LocaleFree( local_name );
  219.         return dir;
  220.     }
  221. #endif
  222.     errno = ENOENT;
  223.     return NULL;
  224. }
  225. /**
  226.  * Reads the next file name from an open directory.
  227.  *
  228.  * @param dir The directory that is being read
  229.  *
  230.  * @return a UTF-8 string of the directory entry.
  231.  * Use free() to free this memory.
  232.  */
  233. char *utf8_readdir( DIR *dir )
  234. {
  235. #ifdef WIN32
  236.     struct _wdirent *ent = vlc_wreaddir (dir);
  237.     if (ent == NULL)
  238.         return NULL;
  239.     return FromWide (ent->d_name);
  240. #else
  241.     struct dirent *ent;
  242.     ent = readdir( (DIR *)dir );
  243.     if( ent == NULL )
  244.         return NULL;
  245.     return vlc_fix_readdir( ent->d_name );
  246. #endif
  247. }
  248. static int dummy_select( const char *str )
  249. {
  250.     (void)str;
  251.     return 1;
  252. }
  253. /**
  254.  * Does the same as utf8_scandir(), but takes an open directory pointer
  255.  * instead of a directory path.
  256.  */
  257. int utf8_loaddir( DIR *dir, char ***namelist,
  258.                   int (*select)( const char * ),
  259.                   int (*compar)( const char **, const char ** ) )
  260. {
  261.     if( select == NULL )
  262.         select = dummy_select;
  263.     if( dir == NULL )
  264.         return -1;
  265.     else
  266.     {
  267.         char **tab = NULL;
  268.         char *entry;
  269.         unsigned num = 0;
  270.         rewinddir( dir );
  271.         while( ( entry = utf8_readdir( dir ) ) != NULL )
  272.         {
  273.             char **newtab;
  274.             if( !select( entry ) )
  275.             {
  276.                 free( entry );
  277.                 continue;
  278.             }
  279.             newtab = realloc( tab, sizeof( char * ) * (num + 1) );
  280.             if( newtab == NULL )
  281.             {
  282.                 free( entry );
  283.                 goto error;
  284.             }
  285.             tab = newtab;
  286.             tab[num++] = entry;
  287.         }
  288.         if( compar != NULL )
  289.             qsort( tab, num, sizeof( tab[0] ),
  290.                    (int (*)( const void *, const void *))compar );
  291.         *namelist = tab;
  292.         return num;
  293.     error:{
  294.         unsigned i;
  295.         for( i = 0; i < num; i++ )
  296.             free( tab[i] );
  297.         if( tab != NULL )
  298.             free( tab );
  299.         }
  300.     }
  301.     return -1;
  302. }
  303. /**
  304.  * Selects file entries from a directory, as GNU C scandir().
  305.  *
  306.  * @param dirname UTF-8 diretory path
  307.  * @param pointer [OUT] pointer set, on succesful completion, to the address
  308.  * of a table of UTF-8 filenames. All filenames must be freed with free().
  309.  * The table itself must be freed with free() as well.
  310.  *
  311.  * @return How many file names were selected (possibly 0),
  312.  * or -1 in case of error.
  313.  */
  314. int utf8_scandir( const char *dirname, char ***namelist,
  315.                   int (*select)( const char * ),
  316.                   int (*compar)( const char **, const char ** ) )
  317. {
  318.     DIR *dir = utf8_opendir (dirname);
  319.     int val = -1;
  320.     if (dir != NULL)
  321.     {
  322.         val = utf8_loaddir (dir, namelist, select, compar);
  323.         closedir (dir);
  324.     }
  325.     return val;
  326. }
  327. static int utf8_statEx( const char *filename, struct stat *buf,
  328.                         bool deref )
  329. {
  330. #ifdef UNDER_CE
  331.     /*_stat translates to wchar internally on WinCE*/
  332.     return _stat( filename, buf );
  333. #elif defined (WIN32)
  334.     CONVERT_PATH (filename, wpath, -1);
  335.     return _wstati64 (wpath, buf);
  336. #endif
  337. #ifdef HAVE_SYS_STAT_H
  338.     const char *local_name = ToLocale( filename );
  339.     if( local_name != NULL )
  340.     {
  341.         int res = deref ? stat( local_name, buf )
  342.                        : lstat( local_name, buf );
  343.         LocaleFree( local_name );
  344.         return res;
  345.     }
  346.     errno = ENOENT;
  347. #endif
  348.     return -1;
  349. }
  350. /**
  351.  * Finds file/inode informations, as stat().
  352.  * Consider using fstat() instead, if possible.
  353.  *
  354.  * @param filename UTF-8 file path
  355.  */
  356. int utf8_stat( const char *filename, struct stat *buf)
  357. {
  358.     return utf8_statEx( filename, buf, true );
  359. }
  360. /**
  361.  * Finds file/inode informations, as lstat().
  362.  * Consider using fstat() instead, if possible.
  363.  *
  364.  * @param filename UTF-8 file path
  365.  */
  366. int utf8_lstat( const char *filename, struct stat *buf)
  367. {
  368.     return utf8_statEx( filename, buf, false );
  369. }
  370. /**
  371.  * Removes a file.
  372.  *
  373.  * @param filename a UTF-8 string with the name of the file you want to delete.
  374.  * @return A 0 return value indicates success. A -1 return value indicates an
  375.  *        error, and an error code is stored in errno
  376.  */
  377. int utf8_unlink( const char *filename )
  378. {
  379. #ifdef UNDER_CE
  380.     /*_open translates to wchar internally on WinCE*/
  381.     return _unlink( filename );
  382. #elif defined (WIN32)
  383.     CONVERT_PATH (filename, wpath, -1);
  384.     return _wunlink (wpath);
  385. #endif
  386.     const char *local_name = ToLocale( filename );
  387.     if( local_name == NULL )
  388.     {
  389.         errno = ENOENT;
  390.         return -1;
  391.     }
  392.     int ret = unlink( local_name );
  393.     LocaleFree( local_name );
  394.     return ret;
  395. }
  396. /**
  397.  * Moves a file atomically. This only works within a single file system.
  398.  *
  399.  * @param oldpath path to the file before the move
  400.  * @param newpath intended path to the file after the move
  401.  * @return A 0 return value indicates success. A -1 return value indicates an
  402.  *        error, and an error code is stored in errno
  403.  */
  404. int utf8_rename (const char *oldpath, const char *newpath)
  405. {
  406. #if defined (WIN32)
  407.     CONVERT_PATH (oldpath, wold, -1);
  408.     CONVERT_PATH (newpath, wnew, -1);
  409. # ifdef UNDER_CE
  410.     /* FIXME: errno support */
  411.     if (MoveFileW (wold, wnew))
  412.         return 0;
  413.     else
  414.         return -1;
  415. #else
  416.     return _wrename (wold, wnew);
  417. #endif
  418. #endif
  419.     const char *lo = ToLocale (oldpath);
  420.     if (lo == NULL)
  421.         goto error;
  422.     const char *ln = ToLocale (newpath);
  423.     if (ln == NULL)
  424.     {
  425.         LocaleFree (lo);
  426. error:
  427.         errno = ENOENT;
  428.         return -1;
  429.     }
  430.     int ret = rename (lo, ln);
  431.     LocaleFree (lo);
  432.     LocaleFree (ln);
  433.     return ret;
  434. }
  435. int utf8_mkstemp( char *template )
  436. {
  437.     static const char digits[] = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
  438.     static const int i_digits = sizeof(digits)/sizeof(*digits) - 1;
  439.     /* */
  440.     assert( template );
  441.     /* Check template validity */
  442.     const size_t i_length = strlen( template );
  443.     char *psz_rand = &template[i_length-6];
  444.     if( i_length < 6 || strcmp( psz_rand, "XXXXXX" ) )
  445.     {
  446.         errno = EINVAL;
  447.         return -1;
  448.     }
  449.     /* */
  450.     for( int i = 0; i < 256; i++ )
  451.     {
  452.         /* Create a pseudo random file name */
  453.         uint8_t pi_rand[6];
  454.         vlc_rand_bytes( pi_rand, sizeof(pi_rand) );
  455.         for( int j = 0; j < 6; j++ )
  456.             psz_rand[j] = digits[pi_rand[j] % i_digits];
  457.         /* */
  458.         int fd = utf8_open( template, O_CREAT | O_EXCL | O_RDWR, 0600 );
  459.         if( fd >= 0 )
  460.             return fd;
  461.         if( errno != EEXIST )
  462.             return -1;
  463.     }
  464.     errno = EEXIST;
  465.     return -1;
  466. }