summaryrefslogtreecommitdiffstats
path: root/utils/ipodpatcher/ipodio-win32.c
diff options
context:
space:
mode:
Diffstat (limited to 'utils/ipodpatcher/ipodio-win32.c')
-rw-r--r--utils/ipodpatcher/ipodio-win32.c227
1 files changed, 227 insertions, 0 deletions
diff --git a/utils/ipodpatcher/ipodio-win32.c b/utils/ipodpatcher/ipodio-win32.c
new file mode 100644
index 0000000000..2c52a64658
--- /dev/null
+++ b/utils/ipodpatcher/ipodio-win32.c
@@ -0,0 +1,227 @@
+/***************************************************************************
+ * __________ __ ___.
+ * 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/
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
+ * KIND, either express or implied.
+ *
+ ****************************************************************************/
+
+#if defined(_WIN32)
+
+#include <stdio.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <string.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <windows.h>
+#include <stddef.h>
+#include <winioctl.h>
+
+#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 ipod_print_error(char* msg)
+{
+ LPSTR pMsgBuf = NULL;
+
+ printf(msg);
+ FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM |
+ FORMAT_MESSAGE_IGNORE_INSERTS, NULL, GetLastError(),
+ MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), 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) ipod_print_error(" Error opening disk: ");
+ if(GetLastError() == ERROR_ACCESS_DENIED)
+ return -2;
+ else
+ return -1;
+ }
+
+ if (!lock_volume(ipod->dh)) {
+ if (!silent) ipod_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) ipod_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) {
+ ipod_print_error(" Error opening disk: ");
+ return -1;
+ }
+
+ if (!lock_volume(ipod->dh)) {
+ ipod_print_error(" Error locking disk: ");
+ return -1;
+ }
+
+ return 0;
+}
+
+int ipod_close(struct ipod_t* ipod)
+{
+ unlock_volume(ipod->dh);
+ CloseHandle(ipod->dh);
+ ipod->dh = INVALID_HANDLE_VALUE;
+ return 0;
+}
+
+int ipod_alloc_buffer(struct ipod_t* ipod, int bufsize)
+{
+ /* The ReadFile function requires a memory buffer aligned to a multiple of
+ the disk sector size. */
+ ipod->sectorbuf = (unsigned char*)VirtualAlloc(NULL, bufsize, MEM_COMMIT, PAGE_READWRITE);
+ if (ipod->sectorbuf== NULL) {
+ ipod_print_error(" Error allocating a buffer: ");
+ return -1;
+ }
+ return 0;
+}
+
+int ipod_dealloc_buffer(struct ipod_t* ipod)
+{
+ if (ipod->sectorbuf == NULL) {
+ return -1;
+ }
+ if(!VirtualFree(ipod->sectorbuf, 0, MEM_RELEASE)) {
+ ipod_print_error(" Error releasing buffer ");
+ return -1;
+ }
+ ipod->sectorbuf = NULL;
+ return 0;
+}
+
+int ipod_seek(struct ipod_t* ipod, unsigned long pos)
+{
+ if (SetFilePointer(ipod->dh, pos, NULL, FILE_BEGIN)==0xffffffff) {
+ ipod_print_error(" Seek error ");
+ return -1;
+ }
+ return 0;
+}
+
+ssize_t ipod_read(struct ipod_t* ipod, int nbytes)
+{
+ unsigned long count;
+
+ if(ipod->sectorbuf == NULL) {
+ return -1;
+ }
+ if (!ReadFile(ipod->dh, ipod->sectorbuf, nbytes, &count, NULL)) {
+ ipod_print_error(" Error reading from disk: ");
+ return -1;
+ }
+
+ return count;
+}
+
+ssize_t ipod_write(struct ipod_t* ipod, int nbytes)
+{
+ unsigned long count;
+
+ if(ipod->sectorbuf == NULL) {
+ return -1;
+ }
+ if (!WriteFile(ipod->dh, ipod->sectorbuf, nbytes, &count, NULL)) {
+ ipod_print_error(" Error writing to disk: ");
+ return -1;
+ }
+
+ return count;
+}
+
+#endif
+