mirror of
				https://github.com/pbatard/rufus.git
				synced 2024-08-14 23:57:05 +00:00 
			
		
		
		
	[misc] code cleanup
* Also fix WDK compilation issues
This commit is contained in:
		
							parent
							
								
									afec69c428
								
							
						
					
					
						commit
						55b7d5c2da
					
				
					 9 changed files with 39 additions and 34 deletions
				
			
		|  | @ -155,7 +155,7 @@ static __inline void ToUpper(char* str) | |||
| { | ||||
| 	size_t i; | ||||
| 	for (i = 0; i < safe_strlen(str); i++) | ||||
| 		str[i] = toupper(str[i]); | ||||
| 		str[i] = (char)toupper(str[i]); | ||||
| } | ||||
| 
 | ||||
| BOOL GetOpticalMedia(IMG_SAVE* img_save) | ||||
|  |  | |||
|  | @ -363,7 +363,7 @@ iso9660_set_ltime_with_timezone(const struct tm *p_tm, | |||
| 
 | ||||
|   if (!p_tm) return; | ||||
| 
 | ||||
|   snprintf(_pvd_date, 17, | ||||
|   _snprintf(_pvd_date, 17, | ||||
|            "%4.4d%2.2d%2.2d" "%2.2d%2.2d%2.2d" "%2.2d", | ||||
|            p_tm->tm_year + 1900, p_tm->tm_mon + 1, p_tm->tm_mday, | ||||
|            p_tm->tm_hour, p_tm->tm_min, p_tm->tm_sec, | ||||
|  | @ -1111,7 +1111,7 @@ iso9660_pathname_isofy (const char pathname[], uint16_t version) | |||
| 
 | ||||
|   cdio_assert (strlen (pathname) < (sizeof (tmpbuf) - sizeof (";65535"))); | ||||
| 
 | ||||
|   snprintf (tmpbuf, sizeof(tmpbuf), "%s;%d", pathname, version); | ||||
|   _snprintf (tmpbuf, sizeof(tmpbuf), "%s;%d", pathname, version); | ||||
| 
 | ||||
|   return strdup (tmpbuf); | ||||
| } | ||||
|  |  | |||
|  | @ -1437,7 +1437,7 @@ find_lsn_recurse (void *p_image, iso9660_readdir_t iso9660_readdir, | |||
| 
 | ||||
|       if (*ppsz_full_filename != NULL) free(*ppsz_full_filename); | ||||
|       *ppsz_full_filename = calloc(1, len); | ||||
|       snprintf (*ppsz_full_filename, len, "%s%s/", psz_path, psz_filename); | ||||
|       _snprintf (*ppsz_full_filename, len, "%s%s/", psz_path, psz_filename); | ||||
| 
 | ||||
|       if (statbuf->type == _STAT_DIR | ||||
|           && strcmp ((char *) statbuf->filename, ".") | ||||
|  |  | |||
|  | @ -648,7 +648,7 @@ WORD get_language_id(loc_cmd* lcmd) | |||
| 	// Find if the selected language is the user default
 | ||||
| 	for (i = 0; i<lcmd->unum_size; i++) { | ||||
| 		if (lcmd->unum[i] == lang_id) { | ||||
| 			ubpushf("Will use default UI locale 0x%04X", lang_id); | ||||
| 			ubprintf("Will use default UI locale 0x%04X", lang_id); | ||||
| 			return MAKELANGID(lang_id, SUBLANG_DEFAULT); | ||||
| 		} | ||||
| 	} | ||||
|  | @ -662,12 +662,12 @@ WORD get_language_id(loc_cmd* lcmd) | |||
| 		// boolean to tell us that we found what we were after.
 | ||||
| 		EnumUILanguages(EnumUILanguagesProc, 0x4, (LONG_PTR)wlang);	// 0x04 = MUI_LANGUAGE_ID
 | ||||
| 		if (found_lang) { | ||||
| 			ubpushf("Detected installed Windows Language Pack for 0x%04X (%s)", lcmd->unum[i], lcmd->txt[1]); | ||||
| 			ubprintf("Detected installed Windows Language Pack for 0x%04X (%s)", lcmd->unum[i], lcmd->txt[1]); | ||||
| 			return MAKELANGID(lcmd->unum[i], SUBLANG_DEFAULT); | ||||
| 		} | ||||
| 	} | ||||
| 
 | ||||
| 	ubpushf("NOTE: No Windows Language Pack is installed for %s on this system.\r\n" | ||||
| 	ubprintf("NOTE: No Windows Language Pack is installed for %s on this system.\r\n" | ||||
| 		"This means that some controls may still be displayed using the system locale.", lcmd->txt[1]); | ||||
| 	return MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT); | ||||
| } | ||||
|  |  | |||
|  | @ -1743,7 +1743,7 @@ static void InitDialog(HWND hDlg) | |||
| 		embedded_sl_version_str[1], embedded_sl_version_ext[1]); | ||||
| 	uprintf("Grub versions: %s, %s", GRUB4DOS_VERSION, GRUB2_PACKAGE_VERSION); | ||||
| 	uprintf("System locale ID: 0x%04X", GetUserDefaultUILanguage()); | ||||
| 	ubpop(); | ||||
| 	ubflush(); | ||||
| 	if (selected_locale->ctrl_id & LOC_NEEDS_UPDATE) { | ||||
| 		uprintf("NOTE: The %s translation requires an update, but the current translator hasn't submitted " | ||||
| 			"one. Because of this, some messages will only be displayed in English.", selected_locale->txt[1]); | ||||
|  | @ -3174,7 +3174,7 @@ int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine | |||
| 	} | ||||
| 
 | ||||
| relaunch: | ||||
| 	ubpushf("Localization set to '%s'", selected_locale->txt[0]); | ||||
| 	ubprintf("Localization set to '%s'", selected_locale->txt[0]); | ||||
| 	right_to_left_mode = ((selected_locale->ctrl_id) & LOC_RIGHT_TO_LEFT); | ||||
| 	SetProcessDefaultLayout(right_to_left_mode?LAYOUT_RTL:0); | ||||
| 	if (get_loc_data_file(loc_file, selected_locale)) | ||||
|  | @ -3452,7 +3452,7 @@ out: | |||
| 		SetLGP(TRUE, &existing_key, ep_reg, "NoDriveTypeAutorun", 0); | ||||
| 	if ((nWindowsVersion > WINDOWS_XP) && (!automount) && (!SetAutoMount(FALSE))) | ||||
| 		uprintf("Failed to restore AutoMount to disabled"); | ||||
| 	ubpop(); | ||||
| 	ubflush(); | ||||
| 	// Unconditional delete with retry, just in case...
 | ||||
| 	for (i = 0; (!DeleteFileA(cmdline_hogger)) && (i <= 10); i++) | ||||
| 		Sleep(200); | ||||
|  |  | |||
							
								
								
									
										17
									
								
								src/rufus.h
									
										
									
									
									
								
							
							
						
						
									
										17
									
								
								src/rufus.h
									
										
									
									
									
								
							|  | @ -35,7 +35,7 @@ | |||
| #pragma once | ||||
| 
 | ||||
| /* Program options */ | ||||
| #define RUFUS_DEBUG                 // print debug info to Debug facility
 | ||||
| #define RUFUS_LOGGING               // print info to logging facility
 | ||||
| /* Features not ready for prime time and that may *DESTROY* your data - USE AT YOUR OWN RISKS! */ | ||||
| // #define RUFUS_TEST
 | ||||
| 
 | ||||
|  | @ -119,17 +119,17 @@ | |||
| #define safe_vsnprintf vsnprintf | ||||
| #endif | ||||
| 
 | ||||
| #ifdef RUFUS_DEBUG | ||||
| #ifdef RUFUS_LOGGING | ||||
| extern void _uprintf(const char *format, ...); | ||||
| #define uprintf(...) _uprintf(__VA_ARGS__) | ||||
| #define ubpushf(...) do { safe_sprintf(&ubuffer[ubuffer_pos], UBUFFER_SIZE - ubuffer_pos - 2, __VA_ARGS__); \ | ||||
| 	ubuffer_pos = strlen(ubuffer); ubuffer[ubuffer_pos++] = '\r'; ubuffer[ubuffer_pos++] = '\n'; \ | ||||
| 	ubuffer[ubuffer_pos] = 0; } while(0) | ||||
| #define ubpop() do { if (ubuffer_pos) uprintf("%s", ubuffer); ubuffer_pos = 0; } while(0) | ||||
| #define vuprintf(...) if (verbose) _uprintf(__VA_ARGS__) | ||||
| #define vvuprintf(...) if (verbose > 1) _uprintf(__VA_ARGS__) | ||||
| #define suprintf(...) if (!bSilent) _uprintf(__VA_ARGS__) | ||||
| #define uuprintf(...) if (usb_debug) _uprintf(__VA_ARGS__) | ||||
| #define ubprintf(...) do { safe_sprintf(&ubuffer[ubuffer_pos], UBUFFER_SIZE - ubuffer_pos - 2, __VA_ARGS__); \ | ||||
| 	ubuffer_pos = strlen(ubuffer); ubuffer[ubuffer_pos++] = '\r'; ubuffer[ubuffer_pos++] = '\n'; \ | ||||
| 	ubuffer[ubuffer_pos] = 0; } while(0) | ||||
| #define ubflush() do { if (ubuffer_pos) uprintf("%s", ubuffer); ubuffer_pos = 0; } while(0) | ||||
| #ifdef _DEBUG | ||||
| #define duprintf(...) _uprintf(__VA_ARGS__) | ||||
| #else | ||||
|  | @ -141,6 +141,11 @@ extern void _uprintf(const char *format, ...); | |||
| #define vvuprintf(...) | ||||
| #define duprintf(...) | ||||
| #define suprintf(...) | ||||
| #define uuprintf(...) | ||||
| #define duprintf(...) | ||||
| #define ubprintf(...) | ||||
| #define ubflush() | ||||
| #define _uprintf NULL | ||||
| #endif | ||||
| 
 | ||||
| /* Custom Windows messages */ | ||||
|  |  | |||
							
								
								
									
										10
									
								
								src/rufus.rc
									
										
									
									
									
								
							
							
						
						
									
										10
									
								
								src/rufus.rc
									
										
									
									
									
								
							|  | @ -33,7 +33,7 @@ LANGUAGE LANG_NEUTRAL, SUBLANG_NEUTRAL | |||
| IDD_DIALOG DIALOGEX 12, 12, 242, 376 | ||||
| STYLE DS_SETFONT | DS_MODALFRAME | DS_CENTER | WS_MINIMIZEBOX | WS_POPUP | WS_CAPTION | WS_SYSMENU | ||||
| EXSTYLE WS_EX_ACCEPTFILES | ||||
| CAPTION "Rufus 2.15.1092" | ||||
| CAPTION "Rufus 2.15.1093" | ||||
| FONT 8, "Segoe UI Symbol", 400, 0, 0x0 | ||||
| BEGIN | ||||
|     LTEXT           "Device",IDS_DEVICE_TXT,9,6,200,8 | ||||
|  | @ -334,8 +334,8 @@ END | |||
| // | ||||
| 
 | ||||
| VS_VERSION_INFO VERSIONINFO | ||||
|  FILEVERSION 2,15,1092,0 | ||||
|  PRODUCTVERSION 2,15,1092,0 | ||||
|  FILEVERSION 2,15,1093,0 | ||||
|  PRODUCTVERSION 2,15,1093,0 | ||||
|  FILEFLAGSMASK 0x3fL | ||||
| #ifdef _DEBUG | ||||
|  FILEFLAGS 0x1L | ||||
|  | @ -352,13 +352,13 @@ BEGIN | |||
|         BEGIN | ||||
|             VALUE "CompanyName", "Akeo Consulting (http://akeo.ie)" | ||||
|             VALUE "FileDescription", "Rufus" | ||||
|             VALUE "FileVersion", "2.15.1092" | ||||
|             VALUE "FileVersion", "2.15.1093" | ||||
|             VALUE "InternalName", "Rufus" | ||||
|             VALUE "LegalCopyright", "© 2011-2017 Pete Batard (GPL v3)" | ||||
|             VALUE "LegalTrademarks", "http://www.gnu.org/copyleft/gpl.html" | ||||
|             VALUE "OriginalFilename", "rufus.exe" | ||||
|             VALUE "ProductName", "Rufus" | ||||
|             VALUE "ProductVersion", "2.15.1092" | ||||
|             VALUE "ProductVersion", "2.15.1093" | ||||
|         END | ||||
|     END | ||||
|     BLOCK "VarFileInfo" | ||||
|  |  | |||
							
								
								
									
										24
									
								
								src/stdfn.c
									
										
									
									
									
								
							
							
						
						
									
										24
									
								
								src/stdfn.c
									
										
									
									
									
								
							|  | @ -717,19 +717,19 @@ DWORD WINAPI SetLGPThread(LPVOID param) | |||
| 	// We need an IGroupPolicyObject instance to set a Local Group Policy
 | ||||
| 	hr = CoCreateInstance(&my_CLSID_GroupPolicyObject, NULL, CLSCTX_INPROC_SERVER, &my_IID_IGroupPolicyObject, (LPVOID*)&pLGPO); | ||||
| 	if (FAILED(hr)) { | ||||
| 		ubpushf("SetLGP: CoCreateInstance failed; hr = %x", hr); | ||||
| 		ubprintf("SetLGP: CoCreateInstance failed; hr = %x", hr); | ||||
| 		goto error; | ||||
| 	} | ||||
| 
 | ||||
| 	hr = pLGPO->lpVtbl->OpenLocalMachineGPO(pLGPO, GPO_OPEN_LOAD_REGISTRY); | ||||
| 	if (FAILED(hr)) { | ||||
| 		ubpushf("SetLGP: OpenLocalMachineGPO failed - error %x", hr); | ||||
| 		ubprintf("SetLGP: OpenLocalMachineGPO failed - error %x", hr); | ||||
| 		goto error; | ||||
| 	} | ||||
| 
 | ||||
| 	hr = pLGPO->lpVtbl->GetRegistryKey(pLGPO, GPO_SECTION_MACHINE, &path_key); | ||||
| 	if (FAILED(hr)) { | ||||
| 		ubpushf("SetLGP: GetRegistryKey failed - error %x", hr); | ||||
| 		ubprintf("SetLGP: GetRegistryKey failed - error %x", hr); | ||||
| 		goto error; | ||||
| 	} | ||||
| 
 | ||||
|  | @ -737,7 +737,7 @@ DWORD WINAPI SetLGPThread(LPVOID param) | |||
| 	r = RegCreateKeyExA(path_key, p->szPath, 0, NULL, 0, KEY_SET_VALUE | KEY_QUERY_VALUE, | ||||
| 		NULL, &policy_key, &disp); | ||||
| 	if (r != ERROR_SUCCESS) { | ||||
| 		ubpushf("SetLGP: Failed to open LGPO path %s - error %x", p->szPath, hr); | ||||
| 		ubprintf("SetLGP: Failed to open LGPO path %s - error %x", p->szPath, hr); | ||||
| 		policy_key = NULL; | ||||
| 		goto error; | ||||
| 	} | ||||
|  | @ -751,7 +751,7 @@ DWORD WINAPI SetLGPThread(LPVOID param) | |||
| 			// The Key exists but not its value, which is OK
 | ||||
| 			*(p->bExistingKey) = FALSE; | ||||
| 		} else if (r != ERROR_SUCCESS) { | ||||
| 			ubpushf("SetLGP: Failed to read original %s policy value - error %x", p->szPolicy, r); | ||||
| 			ubprintf("SetLGP: Failed to read original %s policy value - error %x", p->szPolicy, r); | ||||
| 		} | ||||
| 	} | ||||
| 
 | ||||
|  | @ -762,7 +762,7 @@ DWORD WINAPI SetLGPThread(LPVOID param) | |||
| 		r = RegDeleteValueA(policy_key, p->szPolicy); | ||||
| 	} | ||||
| 	if (r != ERROR_SUCCESS) { | ||||
| 		ubpushf("SetLGP: RegSetValueEx / RegDeleteValue failed - error %x", r); | ||||
| 		ubprintf("SetLGP: RegSetValueEx / RegDeleteValue failed - error %x", r); | ||||
| 	} | ||||
| 	RegCloseKey(policy_key); | ||||
| 	policy_key = NULL; | ||||
|  | @ -770,13 +770,13 @@ DWORD WINAPI SetLGPThread(LPVOID param) | |||
| 	// Apply policy
 | ||||
| 	hr = pLGPO->lpVtbl->Save(pLGPO, TRUE, (p->bRestore)?FALSE:TRUE, &ext_guid, &snap_guid); | ||||
| 	if (hr != S_OK) { | ||||
| 		ubpushf("SetLGP: Unable to apply %s policy - error %x", p->szPolicy, hr); | ||||
| 		ubprintf("SetLGP: Unable to apply %s policy - error %x", p->szPolicy, hr); | ||||
| 		goto error; | ||||
| 	} else { | ||||
| 		if ((!p->bRestore) || (*(p->bExistingKey))) { | ||||
| 			ubpushf("SetLGP: Successfully %s %s policy to 0x%08X", (p->bRestore)?"restored":"set", p->szPolicy, val); | ||||
| 			ubprintf("SetLGP: Successfully %s %s policy to 0x%08X", (p->bRestore)?"restored":"set", p->szPolicy, val); | ||||
| 		} else { | ||||
| 			ubpushf("SetLGP: Successfully removed %s policy key", p->szPolicy); | ||||
| 			ubprintf("SetLGP: Successfully removed %s policy key", p->szPolicy); | ||||
| 		} | ||||
| 	} | ||||
| 
 | ||||
|  | @ -800,17 +800,17 @@ BOOL SetLGP(BOOL bRestore, BOOL* bExistingKey, const char* szPath, const char* s | |||
| 	HANDLE thread_id; | ||||
| 
 | ||||
| 	if (ReadSettingBool(SETTING_DISABLE_LGP)) { | ||||
| 		ubpushf("LPG handling disabled, per settings"); | ||||
| 		ubprintf("LPG handling disabled, per settings"); | ||||
| 		return FALSE; | ||||
| 	} | ||||
| 
 | ||||
| 	thread_id = CreateThread(NULL, 0, SetLGPThread, (LPVOID)¶ms, 0, NULL); | ||||
| 	if (thread_id == NULL) { | ||||
| 		ubpushf("SetLGP: Unable to start thread"); | ||||
| 		ubprintf("SetLGP: Unable to start thread"); | ||||
| 		return FALSE; | ||||
| 	} | ||||
| 	if (WaitForSingleObject(thread_id, 5000) != WAIT_OBJECT_0) { | ||||
| 		ubpushf("SetLGP: Killing stuck thread!"); | ||||
| 		ubprintf("SetLGP: Killing stuck thread!"); | ||||
| 		TerminateThread(thread_id, 0); | ||||
| 		CloseHandle(thread_id); | ||||
| 		return FALSE; | ||||
|  |  | |||
|  | @ -41,7 +41,7 @@ HWND hStatus; | |||
| size_t ubuffer_pos = 0; | ||||
| char ubuffer[UBUFFER_SIZE];	// Buffer for ubpushf() messages we don't log right away
 | ||||
| 
 | ||||
| #ifdef RUFUS_DEBUG | ||||
| #ifdef RUFUS_LOGGING | ||||
| void _uprintf(const char *format, ...) | ||||
| { | ||||
| 	static char buf[4096]; | ||||
|  |  | |||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue