rbutil/sansapatcher/sansaio-win32.c

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  * $Id$
   9  *
  10  * Copyright (C) 2006-2007 Dave Chapman
  11  *
  12  * error(), lock_volume() and unlock_volume() functions and inspiration taken
  13  * from:
  14  *       RawDisk - Direct Disk Read/Write Access for NT/2000/XP
  15  *       Copyright (c) 2003 Jan Kiszka
  16  *       http://www.stud.uni-hannover.de/user/73174/RawDisk/
  17  *
  18  * All files in this archive are subject to the GNU General Public License.
  19  * See the file COPYING in the source tree root for full license agreement.
  20  *
  21  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  22  * KIND, either express or implied.
  23  *
  24  ****************************************************************************/
  25
  26 #include <stdio.h>
  27 #include <unistd.h>
  28 #include <fcntl.h>
  29 #include <string.h>
  30 #include <stdlib.h>
  31 #include <sys/types.h>
  32 #include <sys/stat.h>
  33 #ifdef __WIN32__
  34 #include <windows.h>
  35 #include <winioctl.h>
  36 #endif
  37
  38 #include "sansaio.h"
  39
  40 static int lock_volume(HANDLE hDisk)
  41 {
  42   DWORD dummy;
  43
  44   return DeviceIoControl(hDisk, FSCTL_LOCK_VOLUME, NULL, 0, NULL, 0,
  45                          &dummy, NULL);
  46 }
  47
  48 static int unlock_volume(HANDLE hDisk)
  49 {
  50   DWORD dummy;
  51
  52   return DeviceIoControl(hDisk, FSCTL_UNLOCK_VOLUME, NULL, 0, NULL, 0,
  53                          &dummy, NULL);
  54 }
  55
  56 #ifndef RBUTIL
  57 void print_error(char* msg)
  58 {
  59     char* pMsgBuf;
  60
  61     printf(msg);
  62     FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM |
  63                   FORMAT_MESSAGE_IGNORE_INSERTS, NULL, GetLastError(),
  64                   MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&pMsgBuf,
  65                   0, NULL);
  66     printf(pMsgBuf);
  67     LocalFree(pMsgBuf);
  68 }
  69 #endif
  70 int sansa_open(struct sansa_t* sansa, int silent)
  71 {
  72     DISK_GEOMETRY_EX diskgeometry_ex;
  73     DISK_GEOMETRY diskgeometry;
  74     unsigned long n;
  75
  76     sansa->dh = CreateFileA(sansa->diskname, GENERIC_READ,
  77                     FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
  78                     FILE_FLAG_WRITE_THROUGH | FILE_FLAG_NO_BUFFERING, NULL);
  79
  80     if (sansa->dh == INVALID_HANDLE_VALUE) {
  81         if (!silent) print_error(" Error opening disk: ");
  82         return -1;
  83     }
  84
  85     if (!lock_volume(sansa->dh)) {
  86         if (!silent) print_error(" Error locking disk: ");
  87         return -1;
  88     }
  89
  90     if (!DeviceIoControl(sansa->dh,
  91                          IOCTL_DISK_GET_DRIVE_GEOMETRY_EX,
  92                          NULL,
  93                          0,
  94                          &diskgeometry_ex,
  95                          sizeof(diskgeometry_ex),
  96                          &n,
  97                          NULL)) {
  98         if (!DeviceIoControl(sansa->dh,
  99                              IOCTL_DISK_GET_DRIVE_GEOMETRY,
 100                              NULL,
 101                              0,
 102                              &diskgeometry,
 103                              sizeof(diskgeometry),
 104                              &n,
 105                              NULL)) {
 106             if (!silent) print_error(" Error reading disk geometry: ");
 107             return -1;
 108         } else {
 109             sansa->sector_size=diskgeometry.BytesPerSector;
 110         }
 111     } else {
 112         sansa->sector_size=diskgeometry_ex.Geometry.BytesPerSector;
 113     }
 114
 115     return 0;
 116 }
 117
 118 int sansa_reopen_rw(struct sansa_t* sansa)
 119 {
 120     /* Close existing file and re-open for writing */
 121     unlock_volume(sansa->dh);
 122     CloseHandle(sansa->dh);
 123
 124     sansa->dh = CreateFileA(sansa->diskname, GENERIC_READ | GENERIC_WRITE,
 125                      FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
 126                      FILE_FLAG_WRITE_THROUGH | FILE_FLAG_NO_BUFFERING, NULL);
 127
 128     if (sansa->dh == INVALID_HANDLE_VALUE) {
 129         print_error(" Error opening disk: ");
 130         return -1;
 131     }
 132
 133     if (!lock_volume(sansa->dh)) {
 134         print_error(" Error locking disk: ");
 135         return -1;
 136     }
 137
 138     return 0;
 139 }
 140
 141 int sansa_close(struct sansa_t* sansa)
 142 {
 143     unlock_volume(sansa->dh);
 144     CloseHandle(sansa->dh);
 145     return 0;
 146 }
 147
 148 int sansa_alloc_buffer(unsigned char** sectorbuf, int bufsize)
 149 {
 150     /* The ReadFile function requires a memory buffer aligned to a multiple of
 151        the disk sector size. */
 152     *sectorbuf = (unsigned char*)VirtualAlloc(NULL, bufsize, MEM_COMMIT, PAGE_READWRITE);
 153     if (*sectorbuf == NULL) {
 154         print_error(" Error allocating a buffer: ");
 155         return -1;
 156     }
 157     return 0;
 158 }
 159
 160 int sansa_seek(struct sansa_t* sansa, loff_t pos)
 161 {
 162     LARGE_INTEGER li;
 163
 164     li.QuadPart = pos;
 165
 166     li.LowPart = SetFilePointer (sansa->dh, li.LowPart, &li.HighPart, FILE_BEGIN);
 167
 168     if (li.LowPart == INVALID_SET_FILE_POINTER && GetLastError() != NO_ERROR) {
 169         print_error(" Seek error ");
 170         return -1;
 171     }
 172     return 0;
 173 }
 174
 175 int sansa_read(struct sansa_t* sansa, unsigned char* buf, int nbytes)
 176 {
 177     unsigned long count;
 178
 179     if (!ReadFile(sansa->dh, buf, nbytes, &count, NULL)) {
 180         print_error(" Error reading from disk: ");
 181         return -1;
 182     }
 183
 184     return count;
 185 }
 186
 187 int sansa_write(struct sansa_t* sansa, unsigned char* buf, int nbytes)
 188 {
 189     unsigned long count;
 190
 191     if (!WriteFile(sansa->dh, buf, nbytes, &count, NULL)) {
 192         print_error(" Error writing to disk: ");
 193         return -1;
 194     }
 195
 196     return count;
 197 }