1 /***************************************************************************
3 * Open \______ \ ____ ____ | | _\_ |__ _______ ___
4 * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ /
5 * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < <
6 * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \
10 * Copyright (C) 2006-2007 Dave Chapman
12 * error(), lock_volume() and unlock_volume() functions and inspiration taken
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/
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.
23 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
24 * KIND, either express or implied.
26 ****************************************************************************/
33 #include <sys/types.h>
42 static int lock_volume(HANDLE hDisk
)
46 return DeviceIoControl(hDisk
, FSCTL_LOCK_VOLUME
, NULL
, 0, NULL
, 0,
50 static int unlock_volume(HANDLE hDisk
)
54 return DeviceIoControl(hDisk
, FSCTL_UNLOCK_VOLUME
, NULL
, 0, NULL
, 0,
58 void print_error(char* msg
)
63 FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER
| FORMAT_MESSAGE_FROM_SYSTEM
|
64 FORMAT_MESSAGE_IGNORE_INSERTS
, NULL
, GetLastError(),
65 MAKELANGID(LANG_NEUTRAL
, SUBLANG_DEFAULT
), (LPTSTR
)&pMsgBuf
,
71 int ipod_open(struct ipod_t
* ipod
, int silent
)
73 DISK_GEOMETRY_EX diskgeometry_ex
;
74 DISK_GEOMETRY diskgeometry
;
77 ipod
->dh
= CreateFileA(ipod
->diskname
, GENERIC_READ
,
78 FILE_SHARE_READ
| FILE_SHARE_WRITE
, NULL
, OPEN_EXISTING
,
79 FILE_FLAG_WRITE_THROUGH
| FILE_FLAG_NO_BUFFERING
, NULL
);
81 if (ipod
->dh
== INVALID_HANDLE_VALUE
) {
82 if (!silent
) print_error(" Error opening disk: ");
83 if(GetLastError() == ERROR_ACCESS_DENIED
)
89 if (!lock_volume(ipod
->dh
)) {
90 if (!silent
) print_error(" Error locking disk: ");
96 ipod
->sectors_per_track
= 0;
98 if (!DeviceIoControl(ipod
->dh
,
99 IOCTL_DISK_GET_DRIVE_GEOMETRY_EX
,
103 sizeof(diskgeometry_ex
),
106 if (!DeviceIoControl(ipod
->dh
,
107 IOCTL_DISK_GET_DRIVE_GEOMETRY
,
111 sizeof(diskgeometry
),
114 if (!silent
) print_error(" Error reading disk geometry: ");
117 ipod
->sector_size
= diskgeometry
.BytesPerSector
;
118 ipod
->num_heads
= diskgeometry
.TracksPerCylinder
;
119 ipod
->sectors_per_track
= diskgeometry
.SectorsPerTrack
;
122 ipod
->sector_size
= diskgeometry_ex
.Geometry
.BytesPerSector
;
123 ipod
->num_heads
= diskgeometry_ex
.Geometry
.TracksPerCylinder
;
124 ipod
->sectors_per_track
= diskgeometry_ex
.Geometry
.SectorsPerTrack
;
130 int ipod_reopen_rw(struct ipod_t
* ipod
)
132 /* Close existing file and re-open for writing */
133 unlock_volume(ipod
->dh
);
134 CloseHandle(ipod
->dh
);
136 ipod
->dh
= CreateFileA(ipod
->diskname
, GENERIC_READ
| GENERIC_WRITE
,
137 FILE_SHARE_READ
| FILE_SHARE_WRITE
, NULL
, OPEN_EXISTING
,
138 FILE_FLAG_WRITE_THROUGH
| FILE_FLAG_NO_BUFFERING
, NULL
);
140 if (ipod
->dh
== INVALID_HANDLE_VALUE
) {
141 print_error(" Error opening disk: ");
145 if (!lock_volume(ipod
->dh
)) {
146 print_error(" Error locking disk: ");
153 int ipod_close(struct ipod_t
* ipod
)
155 unlock_volume(ipod
->dh
);
156 CloseHandle(ipod
->dh
);
160 int ipod_alloc_buffer(unsigned char** sectorbuf
, int bufsize
)
162 /* The ReadFile function requires a memory buffer aligned to a multiple of
163 the disk sector size. */
164 *sectorbuf
= (unsigned char*)VirtualAlloc(NULL
, bufsize
, MEM_COMMIT
, PAGE_READWRITE
);
165 if (*sectorbuf
== NULL
) {
166 print_error(" Error allocating a buffer: ");
172 int ipod_seek(struct ipod_t
* ipod
, unsigned long pos
)
174 if (SetFilePointer(ipod
->dh
, pos
, NULL
, FILE_BEGIN
)==0xffffffff) {
175 print_error(" Seek error ");
181 ssize_t
ipod_read(struct ipod_t
* ipod
, unsigned char* buf
, int nbytes
)
185 if (!ReadFile(ipod
->dh
, buf
, nbytes
, &count
, NULL
)) {
186 print_error(" Error reading from disk: ");
193 ssize_t
ipod_write(struct ipod_t
* ipod
, unsigned char* buf
, int nbytes
)
197 if (!WriteFile(ipod
->dh
, buf
, nbytes
, &count
, NULL
)) {
198 print_error(" Error writing to disk: ");