/*************************************************************************** * __________ __ ___. * Open \______ \ ____ ____ | | _\_ |__ _______ ___ * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ / * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < < * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \ * \/ \/ \/ \/ \/ * $Id$ * * Copyright (C) 2006-2007 Dave Chapman * * error(), lock_volume() and unlock_volume() functions and inspiration taken * from: * RawDisk - Direct Disk Read/Write Access for NT/2000/XP * Copyright (c) 2003 Jan Kiszka * http://www.stud.uni-hannover.de/user/73174/RawDisk/ * * All files in this archive are subject to the GNU General Public License. * See the file COPYING in the source tree root for full license agreement. * * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY * KIND, either express or implied. * ****************************************************************************/ #include #include #include #include #include #include #include #ifdef __WIN32__ #include #include #endif #include "ipodio.h" static int lock_volume(HANDLE hDisk) { DWORD dummy; return DeviceIoControl(hDisk, FSCTL_LOCK_VOLUME, NULL, 0, NULL, 0, &dummy, NULL); } static int unlock_volume(HANDLE hDisk) { DWORD dummy; return DeviceIoControl(hDisk, FSCTL_UNLOCK_VOLUME, NULL, 0, NULL, 0, &dummy, NULL); } void print_error(char* msg) { char* pMsgBuf; printf(msg); FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, NULL, GetLastError(), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&pMsgBuf, 0, NULL); printf(pMsgBuf); LocalFree(pMsgBuf); } int ipod_open(struct ipod_t* ipod, int silent) { DISK_GEOMETRY_EX diskgeometry_ex; DISK_GEOMETRY diskgeometry; unsigned long n; ipod->dh = CreateFileA(ipod->diskname, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_FLAG_WRITE_THROUGH | FILE_FLAG_NO_BUFFERING, NULL); if (ipod->dh == INVALID_HANDLE_VALUE) { if (!silent) print_error(" Error opening disk: "); if(GetLastError() == ERROR_ACCESS_DENIED) return -2; else return -1; } if (!lock_volume(ipod->dh)) { if (!silent) print_error(" Error locking disk: "); return -1; } /* Defaults */ ipod->num_heads = 0; ipod->sectors_per_track = 0; if (!DeviceIoControl(ipod->dh, IOCTL_DISK_GET_DRIVE_GEOMETRY_EX, NULL, 0, &diskgeometry_ex, sizeof(diskgeometry_ex), &n, NULL)) { if (!DeviceIoControl(ipod->dh, IOCTL_DISK_GET_DRIVE_GEOMETRY, NULL, 0, &diskgeometry, sizeof(diskgeometry), &n, NULL)) { if (!silent) print_error(" Error reading disk geometry: "); return -1; } else { ipod->sector_size = diskgeometry.BytesPerSector; ipod->num_heads = diskgeometry.TracksPerCylinder; ipod->sectors_per_track = diskgeometry.SectorsPerTrack; } } else { ipod->sector_size = diskgeometry_ex.Geometry.BytesPerSector; ipod->num_heads = diskgeometry_ex.Geometry.TracksPerCylinder; ipod->sectors_per_track = diskgeometry_ex.Geometry.SectorsPerTrack; } return 0; } int ipod_reopen_rw(struct ipod_t* ipod) { /* Close existing file and re-open for writing */ unlock_volume(ipod->dh); CloseHandle(ipod->dh); ipod->dh = CreateFileA(ipod->diskname, GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_FLAG_WRITE_THROUGH | FILE_FLAG_NO_BUFFERING, NULL); if (ipod->dh == INVALID_HANDLE_VALUE) { print_error(" Error opening disk: "); return -1; } if (!lock_volume(ipod->dh)) { print_error(" Error locking disk: "); return -1; } return 0; } int ipod_close(struct ipod_t* ipod) { unlock_volume(ipod->dh); CloseHandle(ipod->dh); return 0; } int ipod_alloc_buffer(unsigned char** sectorbuf, int bufsize) { /* The ReadFile function requires a memory buffer aligned to a multiple of the disk sector size. */ *sectorbuf = (unsigned char*)VirtualAlloc(NULL, bufsize, MEM_COMMIT, PAGE_READWRITE); if (*sectorbuf == NULL) { print_error(" Error allocating a buffer: "); return -1; } return 0; } int ipod_seek(struct ipod_t* ipod, unsigned long pos) { if (SetFilePointer(ipod->dh, pos, NULL, FILE_BEGIN)==0xffffffff) { print_error(" Seek error "); return -1; } return 0; } ssize_t ipod_read(struct ipod_t* ipod, unsigned char* buf, int nbytes) { unsigned long count; if (!ReadFile(ipod->dh, buf, nbytes, &count, NULL)) { print_error(" Error reading from disk: "); return -1; } return count; } ssize_t ipod_write(struct ipod_t* ipod, unsigned char* buf, int nbytes) { unsigned long count; if (!WriteFile(ipod->dh, buf, nbytes, &count, NULL)) { print_error(" Error writing to disk: "); return -1; } return count; }