2011-11-13 20:53:23 +00:00
|
|
|
/*
|
2011-12-05 11:36:02 +00:00
|
|
|
* Rufus: The Reliable USB Formatting Utility
|
2011-11-13 20:53:23 +00:00
|
|
|
* Copyright (c) 2011 Pete Batard <pete@akeo.ie>
|
|
|
|
*
|
|
|
|
* 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 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
2011-12-01 17:20:52 +00:00
|
|
|
#include <windows.h>
|
2011-12-11 23:38:16 +00:00
|
|
|
#include <winioctl.h> // for DISK_GEOMETRY
|
2011-11-13 20:53:23 +00:00
|
|
|
|
|
|
|
#pragma once
|
|
|
|
|
2011-12-11 23:38:16 +00:00
|
|
|
/* Program options */
|
|
|
|
#define RUFUS_DEBUG // print debug info to Debug facility (use debugview to consult)
|
|
|
|
#define DISABLE_AUTORUN // disable new USB drive notification from explorer when application is running
|
2011-11-13 20:53:23 +00:00
|
|
|
|
2011-11-24 02:24:50 +00:00
|
|
|
#define STR_NO_LABEL "NO_LABEL"
|
2011-11-28 12:37:58 +00:00
|
|
|
#define RUFUS_CANCELBOX_TITLE "Rufus - Cancellation"
|
2011-11-23 12:27:51 +00:00
|
|
|
#define DRIVE_INDEX_MIN 0x80
|
|
|
|
#define DRIVE_INDEX_MAX 0xC0
|
2011-11-24 02:24:50 +00:00
|
|
|
#define MAX_DRIVES 16
|
2011-11-21 17:06:17 +00:00
|
|
|
#define MAX_TOOLTIPS 16
|
2011-12-09 01:39:13 +00:00
|
|
|
#define MAX_PROGRESS (0xFFFF-1) // leave room for 1 more for insta-progress workaround
|
2011-11-24 20:02:14 +00:00
|
|
|
#define PROPOSEDLABEL_TOLERANCE 0.10
|
2011-11-24 23:49:42 +00:00
|
|
|
#define FS_DEFAULT FS_FAT32
|
2011-11-21 17:06:17 +00:00
|
|
|
#define WHITE RGB(255,255,255)
|
|
|
|
#define SEPARATOR_GREY RGB(223,223,223)
|
2011-12-09 22:47:44 +00:00
|
|
|
#define RUFUS_URL "http://rufus.akeo.ie"
|
2011-11-21 17:06:17 +00:00
|
|
|
#define BUG_URL "https://github.com/pbatard/rufus/issues"
|
2011-11-17 01:43:06 +00:00
|
|
|
#define IGNORE_RETVAL(expr) do { (void)(expr); } while(0)
|
2011-11-18 21:46:34 +00:00
|
|
|
#ifndef ARRAYSIZE
|
|
|
|
#define ARRAYSIZE(A) (sizeof(A)/sizeof((A)[0]))
|
|
|
|
#endif
|
2011-11-27 23:40:28 +00:00
|
|
|
#define IsChecked(CheckBox_ID) (IsDlgButtonChecked(hMainDialog, CheckBox_ID) == BST_CHECKED)
|
2011-11-13 20:53:23 +00:00
|
|
|
|
|
|
|
#define safe_free(p) do {free((void*)p); p = NULL;} while(0)
|
2011-11-18 01:58:08 +00:00
|
|
|
#define safe_closehandle(h) do {if (h != INVALID_HANDLE_VALUE) {CloseHandle(h); h = INVALID_HANDLE_VALUE;}} while(0)
|
2011-11-13 20:53:23 +00:00
|
|
|
#define safe_min(a, b) min((size_t)(a), (size_t)(b))
|
|
|
|
#define safe_strcp(dst, dst_max, src, count) do {memcpy(dst, src, safe_min(count, dst_max)); \
|
|
|
|
((char*)dst)[safe_min(count, dst_max)-1] = 0;} while(0)
|
|
|
|
#define safe_strcpy(dst, dst_max, src) safe_strcp(dst, dst_max, src, safe_strlen(src)+1)
|
|
|
|
#define safe_strncat(dst, dst_max, src, count) strncat(dst, src, safe_min(count, dst_max - safe_strlen(dst) - 1))
|
|
|
|
#define safe_strcat(dst, dst_max, src) safe_strncat(dst, dst_max, src, safe_strlen(src)+1)
|
|
|
|
#define safe_strcmp(str1, str2) strcmp(((str1==NULL)?"<NULL>":str1), ((str2==NULL)?"<NULL>":str2))
|
|
|
|
#define safe_stricmp(str1, str2) _stricmp(((str1==NULL)?"<NULL>":str1), ((str2==NULL)?"<NULL>":str2))
|
|
|
|
#define safe_strncmp(str1, str2, count) strncmp(((str1==NULL)?"<NULL>":str1), ((str2==NULL)?"<NULL>":str2), count)
|
|
|
|
#define safe_closehandle(h) do {if (h != INVALID_HANDLE_VALUE) {CloseHandle(h); h = INVALID_HANDLE_VALUE;}} while(0)
|
2011-11-29 20:18:23 +00:00
|
|
|
#define safe_unlockclose(h) do {if (h != INVALID_HANDLE_VALUE) {UnlockDrive(h); CloseHandle(h); h = INVALID_HANDLE_VALUE;}} while(0)
|
2011-11-13 20:53:23 +00:00
|
|
|
#define safe_sprintf _snprintf
|
|
|
|
#define safe_strlen(str) ((((char*)str)==NULL)?0:strlen(str))
|
|
|
|
#define safe_strdup _strdup
|
2011-11-17 01:43:06 +00:00
|
|
|
#if defined(_MSC_VER)
|
|
|
|
#define safe_vsnprintf(buf, size, format, arg) _vsnprintf_s(buf, size, _TRUNCATE, format, arg)
|
|
|
|
#else
|
|
|
|
#define safe_vsnprintf vsnprintf
|
|
|
|
#endif
|
|
|
|
|
2011-11-21 17:06:17 +00:00
|
|
|
#ifdef RUFUS_DEBUG
|
|
|
|
extern void _uprintf(const char *format, ...);
|
|
|
|
#define uprintf(...) _uprintf(__VA_ARGS__)
|
|
|
|
#else
|
|
|
|
#define uprintf(...)
|
2011-11-18 01:58:08 +00:00
|
|
|
#endif
|
|
|
|
|
2011-11-26 00:25:04 +00:00
|
|
|
/* Custom Windows messages */
|
|
|
|
enum user_message_type {
|
2011-12-09 01:39:13 +00:00
|
|
|
UM_FORMAT_COMPLETED = WM_APP
|
2011-11-26 00:25:04 +00:00
|
|
|
};
|
|
|
|
|
2011-11-21 17:06:17 +00:00
|
|
|
/* Custom notifications */
|
2011-12-08 00:22:13 +00:00
|
|
|
enum notification_type {
|
2011-11-21 17:06:17 +00:00
|
|
|
MSG_INFO,
|
|
|
|
MSG_WARNING,
|
|
|
|
MSG_ERROR
|
|
|
|
};
|
|
|
|
|
2011-12-08 00:22:13 +00:00
|
|
|
/* Timers used throughout the program */
|
2011-12-09 01:39:13 +00:00
|
|
|
enum timer_type {
|
|
|
|
TID_MESSAGE = 0x1000,
|
|
|
|
TID_BADBLOCKS_UPDATE,
|
|
|
|
TID_APP_TIMER
|
|
|
|
};
|
|
|
|
|
|
|
|
/* Action type, for progress bar breakdown */
|
|
|
|
enum action_type {
|
|
|
|
OP_BADBLOCKS,
|
|
|
|
OP_ZERO_MBR,
|
|
|
|
OP_PARTITION,
|
2011-12-09 12:57:32 +00:00
|
|
|
OP_FORMAT,
|
|
|
|
OP_CREATE_FS,
|
2011-12-09 01:39:13 +00:00
|
|
|
OP_FIX_MBR,
|
|
|
|
OP_DOS,
|
|
|
|
OP_MAX
|
2011-12-08 00:22:13 +00:00
|
|
|
};
|
|
|
|
|
2011-11-24 23:49:42 +00:00
|
|
|
/* File system indexes in our FS combobox */
|
2011-11-26 00:25:04 +00:00
|
|
|
// TODO: FormatEx should support "NTFS", "FAT", "FAT32", "UDF", and "EXFAT" as per
|
|
|
|
// http://msdn.microsoft.com/en-us/library/windows/desktop/aa819439.aspx
|
2011-12-01 15:01:10 +00:00
|
|
|
|
|
|
|
enum {
|
|
|
|
FS_UNKNOWN = -1,
|
2011-11-24 23:49:42 +00:00
|
|
|
FS_FAT16 = 0,
|
|
|
|
FS_FAT32,
|
|
|
|
FS_NTFS,
|
2011-12-01 02:56:44 +00:00
|
|
|
FS_EXFAT,
|
2011-11-24 23:49:42 +00:00
|
|
|
FS_MAX
|
|
|
|
};
|
|
|
|
|
2011-12-15 00:46:47 +00:00
|
|
|
enum dos_type {
|
|
|
|
DT_FREEDOS = 0,
|
|
|
|
DT_WINME,
|
|
|
|
DT_MAX
|
|
|
|
};
|
|
|
|
|
2011-12-01 17:20:52 +00:00
|
|
|
/* Current drive info */
|
2011-11-20 22:49:55 +00:00
|
|
|
typedef struct {
|
2011-12-01 17:20:52 +00:00
|
|
|
DWORD DeviceNumber;
|
|
|
|
LONGLONG DiskSize;
|
|
|
|
DISK_GEOMETRY Geometry;
|
|
|
|
DWORD FirstSector;
|
|
|
|
int FSType;
|
|
|
|
struct {
|
|
|
|
ULONG Allowed;
|
|
|
|
ULONG Default;
|
|
|
|
} ClusterSize[FS_MAX];
|
|
|
|
} RUFUS_DRIVE_INFO;
|
2011-11-26 00:25:04 +00:00
|
|
|
|
|
|
|
/*
|
2011-12-01 17:20:52 +00:00
|
|
|
* Globals
|
|
|
|
*/
|
|
|
|
extern HINSTANCE hMainInstance;
|
|
|
|
extern HWND hMainDialog, hStatus, hDeviceList, hCapacity;
|
2011-12-15 00:46:47 +00:00
|
|
|
extern HWND hFileSystem, hClusterSize, hLabel, hDOSType;
|
2011-12-01 17:20:52 +00:00
|
|
|
extern float fScale;
|
|
|
|
extern char szFolderPath[MAX_PATH];
|
|
|
|
extern DWORD FormatStatus;
|
|
|
|
extern RUFUS_DRIVE_INFO SelectedDrive;
|
2011-12-09 01:39:13 +00:00
|
|
|
extern const int nb_steps[FS_MAX];
|
2011-12-01 17:20:52 +00:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Shared prototypes
|
2011-11-26 00:25:04 +00:00
|
|
|
*/
|
2011-12-04 19:47:27 +00:00
|
|
|
extern const char *WindowsErrorString(void);
|
2011-12-01 17:20:52 +00:00
|
|
|
extern void DumpBufferHex(void *buf, size_t size);
|
2011-12-08 00:22:13 +00:00
|
|
|
extern void PrintStatus(unsigned int duration, const char *format, ...);
|
2011-12-09 01:39:13 +00:00
|
|
|
extern void UpdateProgress(int op, float percent);
|
2011-12-04 19:47:27 +00:00
|
|
|
extern const char* StrError(DWORD error_code);
|
2011-12-01 17:20:52 +00:00
|
|
|
extern void CenterDialog(HWND hDlg);
|
|
|
|
extern void CreateStatusBar(void);
|
|
|
|
extern INT_PTR CreateAboutBox(void);
|
|
|
|
extern HWND CreateTooltip(HWND hControl, const char* message, int duration);
|
|
|
|
extern void DestroyTooltip(HWND hWnd);
|
|
|
|
extern void DestroyAllTooltips(void);
|
2011-12-04 19:47:27 +00:00
|
|
|
extern BOOL Notification(int type, char* title, char* format, ...);
|
2011-12-15 00:46:47 +00:00
|
|
|
extern BOOL ExtractDOS(const char* path, int dos_type);
|
2011-12-01 17:20:52 +00:00
|
|
|
extern void __cdecl FormatThread(void* param);
|
|
|
|
extern BOOL CreatePartition(HANDLE hDrive);
|
|
|
|
extern HANDLE GetDriveHandle(DWORD DriveIndex, char* DriveLetter, BOOL bWriteAccess, BOOL bLockDrive);
|
2011-12-01 17:54:35 +00:00
|
|
|
extern BOOL GetDriveLabel(DWORD DriveIndex, char* letter, char** label);
|
2011-12-11 02:19:38 +00:00
|
|
|
extern BOOL UnmountDrive(HANDLE hDrive);
|
2011-12-01 17:20:52 +00:00
|
|
|
|
|
|
|
__inline static BOOL UnlockDrive(HANDLE hDrive)
|
2011-11-26 00:25:04 +00:00
|
|
|
{
|
2011-12-01 17:20:52 +00:00
|
|
|
DWORD size;
|
|
|
|
return DeviceIoControl(hDrive, FSCTL_UNLOCK_VOLUME, NULL, 0, NULL, 0, &size, NULL);
|
2011-11-26 00:25:04 +00:00
|
|
|
}
|
|
|
|
|
2011-12-01 17:20:52 +00:00
|
|
|
/* Basic String Array */
|
|
|
|
typedef struct {
|
|
|
|
char** Table;
|
|
|
|
size_t Size;
|
|
|
|
size_t Index;
|
|
|
|
size_t Max;
|
|
|
|
} StrArray;
|
|
|
|
extern void StrArrayCreate(StrArray* arr, size_t initial_size);
|
|
|
|
extern void StrArrayAdd(StrArray* arr, const char* str);
|
|
|
|
extern void StrArrayClear(StrArray* arr);
|
|
|
|
extern void StrArrayDestroy(StrArray* arr);
|
|
|
|
|
2011-12-01 17:54:35 +00:00
|
|
|
/* Clang/MinGW32 has an issue with intptr_t */
|
|
|
|
#ifndef _UINTPTR_T_DEFINED
|
|
|
|
#define _UINTPTR_T_DEFINED
|
|
|
|
#ifdef _WIN64
|
|
|
|
typedef unsigned __int64 uintptr_t;
|
|
|
|
#else
|
|
|
|
typedef unsigned int uintptr_t;
|
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
|
2011-12-01 17:20:52 +00:00
|
|
|
/* We need a redef of this MS structure */
|
2011-11-26 00:25:04 +00:00
|
|
|
typedef struct {
|
2011-12-01 17:20:52 +00:00
|
|
|
DWORD DeviceType;
|
|
|
|
ULONG DeviceNumber;
|
|
|
|
ULONG PartitionNumber;
|
|
|
|
} STORAGE_DEVICE_NUMBER_REDEF;
|
|
|
|
|
2011-11-28 12:37:58 +00:00
|
|
|
|
|
|
|
/* Custom application errors */
|
|
|
|
#define FAC(f) (f<<16)
|
2011-12-04 19:47:27 +00:00
|
|
|
#define APPERR(err) (APPLICATION_ERROR_MASK|err)
|
|
|
|
#define ERROR_INCOMPATIBLE_FS 0x1201
|
|
|
|
#define ERROR_CANT_QUICK_FORMAT 0x1202
|
|
|
|
#define ERROR_INVALID_CLUSTER_SIZE 0x1203
|
|
|
|
#define ERROR_INVALID_VOLUME_SIZE 0x1204
|
|
|
|
#define ERROR_CANT_START_THREAD 0x1205
|
2011-12-06 23:35:55 +00:00
|
|
|
#define ERROR_BADBLOCKS_FAILURE 0x1206
|