/*************************************************************************** * __________ __ ___. * 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 #ifdef __WIN32__ #include #include #endif #include "sansaio.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 sansa_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 sansa_open(struct sansa_t* sansa, int silent) { DISK_GEOMETRY_EX diskgeometry_ex; DISK_GEOMETRY diskgeometry; unsigned long n; sansa->dh = CreateFileA(sansa->diskname, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_FLAG_WRITE_THROUGH | FILE_FLAG_NO_BUFFERING, NULL); if (sansa->dh == INVALID_HANDLE_VALUE) { if (!silent) sansa_print_error(" Error opening disk: "); if(GetLastError() == ERROR_ACCESS_DENIED) return -2; else return -1; } if (!lock_volume(sansa->dh)) { if (!silent) sansa_print_error(" Error locking disk: "); return -1; } if (!DeviceIoControl(sansa->dh, IOCTL_DISK_GET_DRIVE_GEOMETRY_EX, NULL, 0, &diskgeometry_ex, sizeof(diskgeometry_ex), &n, NULL)) { if (!DeviceIoControl(sansa->dh, IOCTL_DISK_GET_DRIVE_GEOMETRY, NULL, 0, &diskgeometry, sizeof(diskgeometry), &n, NULL)) { if (!silent) sansa_print_error(" Error reading disk geometry: "); return -1; } else { sansa->sector_size=diskgeometry.BytesPerSector; } } else { sansa->sector_size=diskgeometry_ex.Geometry.BytesPerSector; } return 0; } int sansa_reopen_rw(struct sansa_t* sansa) { /* Close existing file and re-open for writing */ unlock_volume(sansa->dh); CloseHandle(sansa->dh); sansa->dh = CreateFileA(sansa->diskname, GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_FLAG_WRITE_THROUGH | FILE_FLAG_NO_BUFFERING, NULL); if (sansa->dh == INVALID_HANDLE_VALUE) { sansa_print_error(" Error opening disk: "); return -1; } if (!lock_volume(sansa->dh)) { sansa_print_error(" Error locking disk: "); return -1; } return 0; } int sansa_close(struct sansa_t* sansa) { unlock_volume(sansa->dh); CloseHandle(sansa->dh); return 0; } int sansa_alloc_buffer(struct sansa_t* sansa, int bufsize) { /* The ReadFile function requires a memory buffer aligned to a multiple of the disk sector size. */ sansa->sectorbuf = (unsigned char*)VirtualAlloc(NULL, bufsize, MEM_COMMIT, PAGE_READWRITE); if (sansa->sectorbuf == NULL) { sansa_print_error(" Error allocating a buffer: "); return -1; } return 0; } int sansa_seek(struct sansa_t* sansa, loff_t pos) { LARGE_INTEGER li; li.QuadPart = pos; li.LowPart = SetFilePointer (sansa->dh, li.LowPart, &li.HighPart, FILE_BEGIN); if (li.LowPart == INVALID_SET_FILE_POINTER && GetLastError() != NO_ERROR) { sansa_print_error(" Seek error "); return -1; } return 0; } int sansa_read(struct sansa_t* sansa, unsigned char* buf, int nbytes) { unsigned long count; if (!ReadFile(sansa->dh, buf, nbytes, &count, NULL)) { sansa_print_error(" Error reading from disk: "); return -1; } return count; } int sansa_write(struct sansa_t* sansa, int nbytes) { unsigned long count; if (!WriteFile(sansa->dh, sansa->sectorbuf, nbytes, &count, NULL)) { sansa_print_error(" Error writing to disk: "); return -1; } return count; } #endif