cbits/dirUtils.c

   1 /*
   2  * (c) The University of Glasgow 2002
   3  *
   4  * Directory Runtime Support
   5  */
   6
   7 /* needed only for solaris2_HOST_OS */
   8 #include "ghcconfig.h"
   9
  10 // The following is required on Solaris to force the POSIX versions of
  11 // the various _r functions instead of the Solaris versions.
  12 #ifdef solaris2_HOST_OS
  13 #define _POSIX_PTHREAD_SEMANTICS
  14 #endif
  15
  16 #include "HsBase.h"
  17
  18 #if defined(_MSC_VER) || defined(__MINGW32__) || defined(_WIN32)
  19 #include <windows.h>
  20
  21 static
  22 int
  23 toErrno(DWORD rc)
  24 {
  25     switch (rc) {
  26     case ERROR_FILE_NOT_FOUND:    return ENOENT;
  27     case ERROR_PATH_NOT_FOUND:    return ENOENT;
  28     case ERROR_TOO_MANY_OPEN_FILES: return EMFILE;
  29     case ERROR_ACCESS_DENIED:     return EACCES;
  30     case ERROR_INVALID_HANDLE:    return EBADF; /* kinda sorta */
  31     case ERROR_NOT_ENOUGH_MEMORY: return ENOMEM;
  32     case ERROR_INVALID_ACCESS:    return EINVAL;
  33     case ERROR_INVALID_DATA:      return EINVAL;
  34     case ERROR_OUTOFMEMORY:       return ENOMEM;
  35     case ERROR_SHARING_VIOLATION: return EACCES;
  36     case ERROR_LOCK_VIOLATION:    return EACCES;
  37     case ERROR_ALREADY_EXISTS:    return EEXIST;
  38     case ERROR_BUSY:              return EBUSY;
  39     case ERROR_BROKEN_PIPE:       return EPIPE;
  40     case ERROR_PIPE_CONNECTED:    return EBUSY;
  41     case ERROR_PIPE_LISTENING:    return EBUSY;
  42     case ERROR_NOT_CONNECTED:     return EINVAL;
  43
  44     case ERROR_NOT_OWNER:         return EPERM;
  45     case ERROR_DIRECTORY:         return ENOTDIR;
  46     case ERROR_FILE_INVALID:      return EACCES;
  47     case ERROR_FILE_EXISTS:       return EEXIST;
  48
  49     default:
  50         return rc;
  51     }
  52 }
  53 #endif
  54
  55
  56 /*
  57  * read an entry from the directory stream; opt for the
  58  * re-entrant friendly way of doing this, if available.
  59  */
  60 int
  61 __hscore_readdir( DIR *dirPtr, struct dirent **pDirEnt )
  62 {
  63 #if HAVE_READDIR_R
  64   struct dirent* p;
  65   int res;
  66   static unsigned int nm_max = (unsigned int)-1;
  67
  68   if (pDirEnt == NULL) {
  69     return -1;
  70   }
  71   if (nm_max == (unsigned int)-1) {
  72 #ifdef NAME_MAX
  73     nm_max = NAME_MAX + 1;
  74 #else
  75     nm_max = pathconf(".", _PC_NAME_MAX);
  76     if (nm_max == -1) { nm_max = 255; }
  77     nm_max++;
  78 #endif
  79   }
  80   p = (struct dirent*)malloc(sizeof(struct dirent) + nm_max);
  81   if (p == NULL) return -1;
  82   res = readdir_r(dirPtr, p, pDirEnt);
  83   if (res != 0) {
  84       *pDirEnt = NULL;
  85       free(p);
  86   }
  87   else if (*pDirEnt == NULL) {
  88     // end of stream
  89     free(p);
  90   }
  91   return res;
  92 #else
  93
  94   if (pDirEnt == NULL) {
  95     return -1;
  96   }
  97
  98   *pDirEnt = readdir(dirPtr);
  99   if (*pDirEnt == NULL) {
 100     return -1;
 101   } else {
 102     return 0;
 103   }
 104 #endif
 105 }
 106
 107 /*
 108  * Function: __hscore_renameFile()
 109  *
 110  * Provide Haskell98's semantics for renaming files and directories.
 111  * It mirrors that of POSIX.1's behaviour for rename() by overwriting
 112  * the target if it exists (the MS CRT implementation of rename() returns
 113  * an error
 114  *
 115  */
 116 int
 117 __hscore_renameFile( char *src, char *dest)
 118 {
 119 #if defined(_MSC_VER) || defined(__MINGW32__) || defined(_WIN32)
 120     static int forNT = -1;
 121
 122     /* ToDo: propagate error codes back */
 123     if (MoveFileA(src, dest)) {
 124         return 0;
 125     } else {
 126         ;
 127     }
 128
 129     /* Failed...it could be because the target already existed. */
 130     if ( !GetFileAttributes(dest) ) {
 131         /* No, it's not there - just fail. */
 132         errno = toErrno(GetLastError());
 133         return (-1);
 134     }
 135
 136     if (forNT == -1) {
 137         OSVERSIONINFO ovi;
 138         ovi.dwOSVersionInfoSize = sizeof(ovi);
 139         if ( !GetVersionEx(&ovi) ) {
 140             errno = toErrno(GetLastError());
 141             return (-1);
 142         }
 143         forNT = ((ovi.dwPlatformId & VER_PLATFORM_WIN32_NT) != 0);
 144     }
 145
 146     if (forNT) {
 147         /* Easy, go for MoveFileEx() */
 148         if ( MoveFileExA(src, dest, MOVEFILE_REPLACE_EXISTING) ) {
 149             return 0;
 150         } else {
 151             errno = toErrno(GetLastError());
 152             return (-1);
 153         }
 154     }
 155
 156     /* No MoveFileEx() for Win9x, try deleting the target. */
 157     /* Similarly, if the MoveFile*() ops didn't work out under NT */
 158     if (DeleteFileA(dest)) {
 159         if (MoveFileA(src,dest)) {
 160             return 0;
 161         } else {
 162             errno = toErrno(GetLastError());
 163             return (-1);
 164         }
 165     } else {
 166         errno = toErrno(GetLastError());
 167         return (-1);
 168     }
 169 #else
 170     return rename(src,dest);
 171 #endif
 172 }
 173
 174 /*
 175  * Function: __hscore_getFolderPath()
 176  *
 177  * Late-bound version of SHGetFolderPath(), coping with OS versions
 178  * that have shell32's lacking that particular API.
 179  *
 180  */
 181 #if defined(_MSC_VER) || defined(__MINGW32__) || defined(_WIN32)
 182 typedef HRESULT (*HSCORE_GETAPPFOLDERFUNTY)(HWND,int,HANDLE,DWORD,char*);
 183 int
 184 __hscore_getFolderPath(HWND hwndOwner,
 185                        int nFolder,
 186                        HANDLE hToken,
 187                        DWORD dwFlags,
 188                        char*  pszPath)
 189 {
 190     static int loaded_dll = 0;
 191     static HMODULE hMod = (HMODULE)NULL;
 192     static HSCORE_GETAPPFOLDERFUNTY funcPtr = NULL;
 193     /* The DLLs to try loading entry point from */
 194     char* dlls[] = { "shell32.dll", "shfolder.dll" };
 195
 196     if (loaded_dll < 0) {
 197         return (-1);
 198     } else if (loaded_dll == 0) {
 199         int i;
 200         for(i=0;i < sizeof(dlls); i++) {
 201             hMod = LoadLibrary(dlls[i]);
 202             if ( hMod != NULL &&
 203                  (funcPtr = (HSCORE_GETAPPFOLDERFUNTY)GetProcAddress(hMod, "SHGetFolderPathA")) ) {
 204                 loaded_dll = 1;
 205                 break;
 206             }
 207         }
 208         if (loaded_dll == 0) {
 209             loaded_dll = (-1);
 210             return (-1);
 211         }
 212     }
 213     /* OK, if we got this far the function has been bound */
 214     return (int)funcPtr(hwndOwner,nFolder,hToken,dwFlags,pszPath);
 215     /* ToDo: unload the DLL on shutdown? */
 216 }
 217 #endif