From c876d3bbefe0dc00c27ca0c12d29da5874946962 Mon Sep 17 00:00:00 2001 From: Dominik Riebeling Date: Wed, 15 Dec 2021 21:04:28 +0100 Subject: rbutil: Merge rbutil with utils folder. rbutil uses several components from the utils folder, and can be considered part of utils too. Having it in a separate folder is an arbitrary split that doesn't help anymore these days, so merge them. This also allows other utils to easily use libtools.make without the need to navigate to a different folder. Change-Id: I3fc2f4de19e3e776553efb5dea5f779dfec0dc21 --- utils/ipodpatcher/ipodio-win32.c | 226 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 226 insertions(+) create mode 100644 utils/ipodpatcher/ipodio-win32.c (limited to 'utils/ipodpatcher/ipodio-win32.c') diff --git a/utils/ipodpatcher/ipodio-win32.c b/utils/ipodpatcher/ipodio-win32.c new file mode 100644 index 0000000000..cea218774a --- /dev/null +++ b/utils/ipodpatcher/ipodio-win32.c @@ -0,0 +1,226 @@ +/*************************************************************************** + * __________ __ ___. + * 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 +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#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); + 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 + -- cgit v1.2.3