rbutil/ipodpatcher/ipodio-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  * This program is free software; you can redistribute it and/or
  19  * modify it under the terms of the GNU General Public License
  20  * as published by the Free Software Foundation; either version 2
  21  * of the License, or (at your option) any later version.
  22  *
  23  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  24  * KIND, either express or implied.
  25  *
  26  ****************************************************************************/
  27
  28 #include <stdio.h>
  29 #include <unistd.h>
  30 #include <fcntl.h>
  31 #include <string.h>
  32 #include <stdlib.h>
  33 #include <sys/types.h>
  34 #include <sys/stat.h>
  35 #include <windows.h>
  36 #include <stddef.h>
  37 #include <winioctl.h>
  38
  39 #include "ipodio.h"
  40
  41 static int lock_volume(HANDLE hDisk)
  42 {
  43   DWORD dummy;
  44
  45   return DeviceIoControl(hDisk, FSCTL_LOCK_VOLUME, NULL, 0, NULL, 0,
  46                          &dummy, NULL);
  47 }
  48
  49 static int unlock_volume(HANDLE hDisk)
  50 {
  51   DWORD dummy;
  52
  53   return DeviceIoControl(hDisk, FSCTL_UNLOCK_VOLUME, NULL, 0, NULL, 0,
  54                          &dummy, NULL);
  55 }
  56
  57 void ipod_print_error(char* msg)
  58 {
  59     LPSTR pMsgBuf = NULL;
  60
  61     printf(msg);
  62     FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM |
  63                   FORMAT_MESSAGE_IGNORE_INSERTS, NULL, GetLastError(),
  64                   MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), pMsgBuf,
  65                   0, NULL);
  66     printf(pMsgBuf);
  67     LocalFree(pMsgBuf);
  68 }
  69
  70 int ipod_open(struct ipod_t* ipod, int silent)
  71 {
  72     DISK_GEOMETRY_EX diskgeometry_ex;
  73     DISK_GEOMETRY diskgeometry;
  74     unsigned long n;
  75
  76     ipod->dh = CreateFileA(ipod->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 (ipod->dh == INVALID_HANDLE_VALUE) {
  81         if (!silent) ipod_print_error(" Error opening disk: ");
  82         if(GetLastError() == ERROR_ACCESS_DENIED)
  83             return -2;
  84         else
  85             return -1;
  86     }
  87
  88     if (!lock_volume(ipod->dh)) {
  89         if (!silent) ipod_print_error(" Error locking disk: ");
  90         return -1;
  91     }
  92
  93     /* Defaults */
  94     ipod->num_heads = 0;
  95     ipod->sectors_per_track = 0;
  96
  97     if (!DeviceIoControl(ipod->dh,
  98                          IOCTL_DISK_GET_DRIVE_GEOMETRY_EX,
  99                          NULL,
 100                          0,
 101                          &diskgeometry_ex,
 102                          sizeof(diskgeometry_ex),
 103                          &n,
 104                          NULL)) {
 105         if (!DeviceIoControl(ipod->dh,
 106                              IOCTL_DISK_GET_DRIVE_GEOMETRY,
 107                              NULL,
 108                              0,
 109                              &diskgeometry,
 110                              sizeof(diskgeometry),
 111                              &n,
 112                              NULL)) {
 113             if (!silent) ipod_print_error(" Error reading disk geometry: ");
 114             return -1;
 115         } else {
 116             ipod->sector_size = diskgeometry.BytesPerSector;
 117             ipod->num_heads = diskgeometry.TracksPerCylinder;
 118             ipod->sectors_per_track = diskgeometry.SectorsPerTrack;
 119         }
 120     } else {
 121         ipod->sector_size = diskgeometry_ex.Geometry.BytesPerSector;
 122         ipod->num_heads = diskgeometry_ex.Geometry.TracksPerCylinder;
 123         ipod->sectors_per_track = diskgeometry_ex.Geometry.SectorsPerTrack;
 124     }
 125
 126     return 0;
 127 }
 128
 129 int ipod_reopen_rw(struct ipod_t* ipod)
 130 {
 131     /* Close existing file and re-open for writing */
 132     unlock_volume(ipod->dh);
 133     CloseHandle(ipod->dh);
 134
 135     ipod->dh = CreateFileA(ipod->diskname, GENERIC_READ | GENERIC_WRITE,
 136                      FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
 137                      FILE_FLAG_WRITE_THROUGH | FILE_FLAG_NO_BUFFERING, NULL);
 138
 139     if (ipod->dh == INVALID_HANDLE_VALUE) {
 140         ipod_print_error(" Error opening disk: ");
 141         return -1;
 142     }
 143
 144     if (!lock_volume(ipod->dh)) {
 145         ipod_print_error(" Error locking disk: ");
 146         return -1;
 147     }
 148
 149     return 0;
 150 }
 151
 152 int ipod_close(struct ipod_t* ipod)
 153 {
 154     unlock_volume(ipod->dh);
 155     CloseHandle(ipod->dh);
 156     return 0;
 157 }
 158
 159 int ipod_alloc_buffer(unsigned char** sectorbuf, int bufsize)
 160 {
 161     /* The ReadFile function requires a memory buffer aligned to a multiple of
 162        the disk sector size. */
 163     *sectorbuf = (unsigned char*)VirtualAlloc(NULL, bufsize, MEM_COMMIT, PAGE_READWRITE);
 164     if (*sectorbuf == NULL) {
 165         ipod_print_error(" Error allocating a buffer: ");
 166         return -1;
 167     }
 168     return 0;
 169 }
 170
 171 int ipod_seek(struct ipod_t* ipod, unsigned long pos)
 172 {
 173     if (SetFilePointer(ipod->dh, pos, NULL, FILE_BEGIN)==0xffffffff) {
 174         ipod_print_error(" Seek error ");
 175         return -1;
 176     }
 177     return 0;
 178 }
 179
 180 ssize_t ipod_read(struct ipod_t* ipod, unsigned char* buf, int nbytes)
 181 {
 182     unsigned long count;
 183
 184     if (!ReadFile(ipod->dh, buf, nbytes, &count, NULL)) {
 185         ipod_print_error(" Error reading from disk: ");
 186         return -1;
 187     }
 188
 189     return count;
 190 }
 191
 192 ssize_t ipod_write(struct ipod_t* ipod, unsigned char* buf, int nbytes)
 193 {
 194     unsigned long count;
 195
 196     if (!WriteFile(ipod->dh, buf, nbytes, &count, NULL)) {
 197         ipod_print_error(" Error writing to disk: ");
 198         return -1;
 199     }
 200
 201     return count;
 202 }
 203