From 7d1a3d82cc8d0cb82d206010b0944a86e575f617 Mon Sep 17 00:00:00 2001
From: Pete Batard <pete@akeo.ie>
Date: Tue, 4 Feb 2014 20:01:28 +0000
Subject: [PATCH] [syslinux] improve handling of 4.x C32 files

* Download the files from files/syslinux-4.07 on the server
* Save local copy in rufus_files/syslinux-4.07
* Move menu.c32/vesamenu.c32 check to BootCheck()
---
 src/iso.c    |  13 ++++--
 src/rufus.c  | 128 ++++++++++++++++++++++++++++-----------------------
 src/rufus.h  |   1 -
 src/rufus.rc |  12 ++---
 4 files changed, 84 insertions(+), 70 deletions(-)

diff --git a/src/iso.c b/src/iso.c
index 246b0fd4..acd89e2a 100644
--- a/src/iso.c
+++ b/src/iso.c
@@ -69,7 +69,7 @@ static const char* pe_dirname[] = { "/i386", "/minint" };
 static const char* pe_file[] = { "ntdetect.com", "setupldr.bin", "txtsetup.sif" };
 static const char* reactos_name = "setupldr.sys"; // TODO: freeldr.sys doesn't seem to work
 static const char* autorun_name = "autorun.inf";
-static const char* old_c32_name[NB_OLD_C32] = OLD_C32_NAMES;
+const char* old_c32_name[NB_OLD_C32] = OLD_C32_NAMES;
 static const int64_t old_c32_threshold[NB_OLD_C32] = OLD_C32_THRESHOLD;
 static uint8_t i_joliet_level = 0;
 static uint64_t total_blocks, nb_blocks;
@@ -218,7 +218,7 @@ static int udf_extract_files(udf_t *p_udf, udf_dirent_t *p_udf_dirent, const cha
 	BOOL r, is_syslinux_cfg, is_old_c32[NB_OLD_C32];
 	int i_length;
 	size_t i, nul_pos;
-	char* psz_fullpath = NULL;
+	char tmp[128], *psz_fullpath = NULL;
 	const char* psz_basename;
 	udf_dirent_t *p_udf_dirent2;
 	uint8_t buf[UDF_BLOCKSIZE];
@@ -266,7 +266,8 @@ static int udf_extract_files(udf_t *p_udf, udf_dirent_t *p_udf_dirent, const cha
 			psz_fullpath[nul_pos] = 0;
 			for (i=0; i<NB_OLD_C32; i++) {
 				if (is_old_c32[i] && use_own_c32[i]) {
-					if (CopyFileA(old_c32_name[i], psz_fullpath, FALSE)) {
+					static_sprintf(tmp, "%s/syslinux-%s/%s", FILES_DIR, embedded_sl_version_str[0], old_c32_name[i]);
+					if (CopyFileA(tmp, psz_fullpath, FALSE)) {
 						uprintf("  Replaced with local version\n");
 						break;
 					}
@@ -346,7 +347,7 @@ static int iso_extract_files(iso9660_t* p_iso, const char *psz_path)
 	DWORD buf_size, wr_size, err;
 	BOOL s, is_syslinux_cfg, is_old_c32[NB_OLD_C32];
 	int i_length, r = 1;
-	char psz_fullpath[1024], *psz_basename;
+	char tmp[128], psz_fullpath[1024], *psz_basename;
 	const char *psz_iso_name = &psz_fullpath[strlen(psz_extract_dir)];
 	unsigned char buf[ISO_BLOCKSIZE];
 	CdioListNode_t* p_entnode;
@@ -406,7 +407,8 @@ static int iso_extract_files(iso9660_t* p_iso, const char *psz_path)
 			psz_fullpath[nul_pos] = 0;
 			for (i=0; i<NB_OLD_C32; i++) {
 				if (is_old_c32[i] && use_own_c32[i]) {
-					if (CopyFileA(old_c32_name[i], psz_fullpath, FALSE)) {
+					static_sprintf(tmp, "%s/syslinux-%s/%s", FILES_DIR, embedded_sl_version_str[0], old_c32_name[i]);
+					if (CopyFileA(tmp, psz_fullpath, FALSE)) {
 						uprintf("  Replaced with local version\n");
 						break;
 					}
@@ -510,6 +512,7 @@ BOOL ExtractISO(const char* src_iso, const char* dest_dir, BOOL scan)
 		SendMessage(hISOProgressBar, PBM_SETMARQUEE, TRUE, 0);
 	} else {
 		uprintf("Extracting files...\n");
+		IGNORE_RETVAL(_chdirU(app_dir));
 		SetWindowTextU(hISOProgressDlg, lmprintf(MSG_231));
 		if (total_blocks == 0) {
 			uprintf("Error: ISO has not been properly scanned.\n");
diff --git a/src/rufus.c b/src/rufus.c
index 7dbc97a0..a5f597b2 100644
--- a/src/rufus.c
+++ b/src/rufus.c
@@ -118,6 +118,7 @@ static BOOL iso_provided = FALSE;
 static BOOL user_notified = FALSE;
 static BOOL relaunch = FALSE;
 extern BOOL force_large_fat32, enable_joliet, enable_rockridge, enable_ntfs_compression;
+extern const char* old_c32_name[NB_OLD_C32];
 static int selection_default;
 static loc_cmd* selected_locale = NULL;
 static UINT_PTR UM_LANGUAGE_MENU_MAX = UM_LANGUAGE_MENU;
@@ -1184,9 +1185,6 @@ DWORD WINAPI ISOScanThread(LPVOID param)
 {
 	int i;
 	BOOL r;
-	FILE* fd;
-	const char* old_c32_name[NB_OLD_C32] = OLD_C32_NAMES;
-	const char* new_c32_url[NB_OLD_C32] = NEW_C32_URL;
 	char isolinux_str[16] = "No";
 
 	if (iso_path == NULL)
@@ -1222,30 +1220,6 @@ DWORD WINAPI ISOScanThread(LPVOID param)
 		safe_free(iso_path);
 		SetMBRProps();
 	} else if (HAS_SYSLINUX(iso_report)) {
-		if (SL_MAJOR(iso_report.sl_version) < 5) {
-			_chdirU(app_dir);
-			for (i=0; i<NB_OLD_C32; i++) {
-				if (iso_report.has_old_c32[i]) {
-					fd = fopen(old_c32_name[i], "rb");
-					if (fd != NULL) {
-						// If a file already exists in the current directory, use that one
-						uprintf("Will replace obsolete '%s' from ISO with the one found in current directory\n", old_c32_name[i]);
-						fclose(fd);
-						use_own_c32[i] = TRUE;
-					} else {
-						PrintStatus(0, FALSE, MSG_204, old_c32_name[i]);
-						if (MessageBoxU(hMainDialog, lmprintf(MSG_084, old_c32_name[i], old_c32_name[i]),
-							 lmprintf(MSG_083, old_c32_name[i]), MB_YESNO|MB_ICONWARNING|MB_IS_RTL) == IDYES) {
-							SetWindowTextU(hISOProgressDlg, lmprintf(MSG_085, old_c32_name[i]));
-							SetWindowTextU(hISOFileName, new_c32_url[i]);
-							if (DownloadFile(new_c32_url[i], old_c32_name[i], hISOProgressDlg))
-								use_own_c32[i] = TRUE;
-						}
-					}
-				}
-			}
-		}
-
 		// Enable bootable and set Target System and FS accordingly
 		CheckDlgButton(hMainDialog, IDC_BOOT, BST_CHECKED);
 		SetTargetSystem();
@@ -1342,6 +1316,7 @@ static BOOL BootCheck(void)
 {
 	int i, fs, bt, dt, r;
 	FILE* fd;
+	DWORD len;
 	const char* ldlinux = "ldlinux";
 	const char* syslinux = "syslinux";
 	const char* ldlinux_ext[3] = { "sys", "bss", "c32" };
@@ -1393,40 +1368,77 @@ static BOOL BootCheck(void)
 			MessageBoxU(hMainDialog, lmprintf(MSG_100), lmprintf(MSG_099), MB_OK|MB_ICONERROR|MB_IS_RTL);
 			return FALSE;
 		}
-		if ((SL_MAJOR(iso_report.sl_version) >= 5) && (iso_report.sl_version != embedded_sl_version[1])) {
-			// Unlike what was the case for v4 and earlier, Syslinux v5+ versions are INCOMPATIBLE with one another!
-			IGNORE_RETVAL(_chdirU(app_dir));
-			IGNORE_RETVAL(_mkdir(FILES_DIR));
-			IGNORE_RETVAL(_chdir(FILES_DIR));
-			for (i=0; i<2; i++) {
-				// Check if we already have the relevant ldlinux_v#.##.sys & ldlinux_v#.##.bss files
-				static_sprintf(tmp, "%s-%s/%s.%s", syslinux, iso_report.sl_version_str, ldlinux, ldlinux_ext[i]);
-				fd = fopen(tmp, "rb");
-				if (fd != NULL) {
-					fseek(fd, 0, SEEK_END);
-					syslinux_ldlinux_len[i] = (DWORD)ftell(fd);
-					fclose(fd);
+
+		if (HAS_SYSLINUX(iso_report)) {
+			if (SL_MAJOR(iso_report.sl_version) < 5) {
+				IGNORE_RETVAL(_chdirU(app_dir));
+				for (i=0; i<NB_OLD_C32; i++) {
+					if (iso_report.has_old_c32[i]) {
+						if (i==0) {
+							IGNORE_RETVAL(_mkdir(FILES_DIR));
+							IGNORE_RETVAL(_chdir(FILES_DIR));
+						}
+						static_sprintf(tmp, "%s-%s/%s", syslinux, embedded_sl_version_str[0], old_c32_name[i]);
+						fd = fopen(tmp, "rb");
+						if (fd != NULL) {
+							// If a file already exists in the current directory, use that one
+							uprintf("Will replace obsolete '%s' from ISO with the one found in './%s'\n", old_c32_name[i], tmp);
+							fclose(fd);
+							use_own_c32[i] = TRUE;
+						} else {
+							PrintStatus(0, FALSE, MSG_204, old_c32_name[i]);
+							if (MessageBoxU(hMainDialog, lmprintf(MSG_084, old_c32_name[i], old_c32_name[i]),
+									lmprintf(MSG_083, old_c32_name[i]), MB_YESNO|MB_ICONWARNING|MB_IS_RTL) == IDYES) {
+								static_sprintf(tmp, "%s-%s", syslinux, embedded_sl_version_str[0]);
+								IGNORE_RETVAL(_mkdir(tmp));
+								SetWindowTextU(hISOProgressDlg, lmprintf(MSG_085, old_c32_name[i]));
+								static_sprintf(tmp, "%s/%s-%s/%s", FILES_URL, syslinux, embedded_sl_version_str[0], old_c32_name[i]);
+								SetWindowTextU(hISOFileName, tmp);
+								len = DownloadFile(tmp, &tmp[sizeof(FILES_URL)], hISOProgressDlg);
+								if (len == 0) {
+									uprintf("Couldn't download the files - cancelling\n");
+									return FALSE;
+								}
+								use_own_c32[i] = TRUE;
+							}
+						}
+					}
 				}
-			}
-			if ((syslinux_ldlinux_len[0] != 0) && (syslinux_ldlinux_len[1] != 0)) {
-				uprintf("Will reuse '%s.%s' and '%s.%s' from './%s/%s-%s/' for Syslinux installation\n",
-					ldlinux, ldlinux_ext[0], ldlinux, ldlinux_ext[1], FILES_DIR, syslinux, iso_report.sl_version_str);
-			} else {
-				r = MessageBoxU(hMainDialog, lmprintf(MSG_114, iso_report.sl_version_str, embedded_sl_version_str[1]),
-					lmprintf(MSG_115), MB_YESNO|MB_ICONWARNING|MB_IS_RTL);
-				if (r != IDYES)
-					return FALSE;
+			} else if (iso_report.sl_version != embedded_sl_version[1]) {
+				// Unlike what was the case for v4 and earlier, Syslinux v5+ versions are INCOMPATIBLE with one another!
+				IGNORE_RETVAL(_chdirU(app_dir));
+				IGNORE_RETVAL(_mkdir(FILES_DIR));
+				IGNORE_RETVAL(_chdir(FILES_DIR));
 				for (i=0; i<2; i++) {
-					static_sprintf(tmp, "%s-%s", syslinux, iso_report.sl_version_str);
-					IGNORE_RETVAL(_mkdir(tmp));
-					static_sprintf(tmp, "%s.%s %s", ldlinux, ldlinux_ext[i], iso_report.sl_version_str);
-					SetWindowTextU(hISOProgressDlg, lmprintf(MSG_085, tmp));
-					static_sprintf(tmp, "%s/%s-%s/%s.%s", FILES_URL, syslinux, iso_report.sl_version_str, ldlinux, ldlinux_ext[i]);
-					SetWindowTextU(hISOFileName, tmp);
-					syslinux_ldlinux_len[i] = DownloadFile(tmp, &tmp[sizeof(FILES_URL)], hISOProgressDlg);
-					if (syslinux_ldlinux_len[i] == 0) {
-						uprintf("Couldn't download the files - cancelling\n");
+					// Check if we already have the relevant ldlinux_v#.##.sys & ldlinux_v#.##.bss files
+					static_sprintf(tmp, "%s-%s/%s.%s", syslinux, iso_report.sl_version_str, ldlinux, ldlinux_ext[i]);
+					fd = fopen(tmp, "rb");
+					if (fd != NULL) {
+						fseek(fd, 0, SEEK_END);
+						syslinux_ldlinux_len[i] = (DWORD)ftell(fd);
+						fclose(fd);
+					}
+				}
+				if ((syslinux_ldlinux_len[0] != 0) && (syslinux_ldlinux_len[1] != 0)) {
+					uprintf("Will reuse '%s.%s' and '%s.%s' from './%s/%s-%s/' for Syslinux installation\n",
+						ldlinux, ldlinux_ext[0], ldlinux, ldlinux_ext[1], FILES_DIR, syslinux, iso_report.sl_version_str);
+				} else {
+					r = MessageBoxU(hMainDialog, lmprintf(MSG_114, iso_report.sl_version_str, embedded_sl_version_str[1]),
+						lmprintf(MSG_115), MB_YESNO|MB_ICONWARNING|MB_IS_RTL);
+					if (r != IDYES)
 						return FALSE;
+					for (i=0; i<2; i++) {
+						static_sprintf(tmp, "%s-%s", syslinux, iso_report.sl_version_str);
+						IGNORE_RETVAL(_mkdir(tmp));
+						static_sprintf(tmp, "%s.%s %s", ldlinux, ldlinux_ext[i], iso_report.sl_version_str);
+						SetWindowTextU(hISOProgressDlg, lmprintf(MSG_085, tmp));
+						static_sprintf(tmp, "%s/%s-%s/%s.%s", FILES_URL, syslinux, iso_report.sl_version_str, ldlinux, ldlinux_ext[i]);
+						SetWindowTextU(hISOFileName, tmp);
+						syslinux_ldlinux_len[i] = DownloadFile(tmp, &tmp[sizeof(FILES_URL)], hISOProgressDlg);
+						if (syslinux_ldlinux_len[i] == 0) {
+							uprintf("Couldn't download the files - cancelling\n");
+							return FALSE;
+						}
 					}
 				}
 			}
diff --git a/src/rufus.h b/src/rufus.h
index 39bf17da..5590ef27 100644
--- a/src/rufus.h
+++ b/src/rufus.h
@@ -214,7 +214,6 @@ typedef struct {
 #define NB_OLD_C32          2
 #define OLD_C32_NAMES       { "menu.c32", "vesamenu.c32" }
 #define OLD_C32_THRESHOLD   { 53500, 148000 }
-#define NEW_C32_URL         { DOWNLOAD_URL "/menu.c32", DOWNLOAD_URL "/vesamenu.c32" }
 
 /* ISO details that the application may want */
 #define WINPE_MININT    0x2A
diff --git a/src/rufus.rc b/src/rufus.rc
index c52cccf6..66021134 100644
--- a/src/rufus.rc
+++ b/src/rufus.rc
@@ -32,7 +32,7 @@ LANGUAGE LANG_NEUTRAL, SUBLANG_NEUTRAL
 
 IDD_DIALOG DIALOGEX 12, 12, 206, 329
 STYLE DS_SETFONT | DS_MODALFRAME | DS_FIXEDSYS | DS_CENTER | WS_POPUP | WS_CAPTION | WS_SYSMENU
-CAPTION "Rufus 1.4.4.402"
+CAPTION "Rufus 1.4.4.403"
 FONT 8, "MS Shell Dlg", 400, 0, 0x1
 BEGIN
     DEFPUSHBUTTON   "Start",IDC_START,94,291,50,14
@@ -165,7 +165,7 @@ END
 RTL_IDD_DIALOG DIALOGEX 12, 12, 206, 329
 STYLE DS_SETFONT | DS_MODALFRAME | DS_FIXEDSYS | DS_CENTER | WS_POPUP | WS_CAPTION | WS_SYSMENU
 EXSTYLE WS_EX_RTLREADING | WS_EX_APPWINDOW | WS_EX_LAYOUTRTL
-CAPTION "Rufus 1.4.4.402"
+CAPTION "Rufus 1.4.4.403"
 FONT 8, "MS Shell Dlg", 400, 0, 0x1
 BEGIN
     DEFPUSHBUTTON   "Start",IDC_START,94,291,50,14
@@ -427,8 +427,8 @@ END
 //
 
 VS_VERSION_INFO VERSIONINFO
- FILEVERSION 1,4,4,402
- PRODUCTVERSION 1,4,4,402
+ FILEVERSION 1,4,4,403
+ PRODUCTVERSION 1,4,4,403
  FILEFLAGSMASK 0x3fL
 #ifdef _DEBUG
  FILEFLAGS 0x1L
@@ -445,13 +445,13 @@ BEGIN
         BEGIN
             VALUE "CompanyName", "Akeo Consulting (http://akeo.ie)"
             VALUE "FileDescription", "Rufus"
-            VALUE "FileVersion", "1.4.4.402"
+            VALUE "FileVersion", "1.4.4.403"
             VALUE "InternalName", "Rufus"
             VALUE "LegalCopyright", "© 2011-2014 Pete Batard (GPL v3)"
             VALUE "LegalTrademarks", "http://www.gnu.org/copyleft/gpl.html"
             VALUE "OriginalFilename", "rufus.exe"
             VALUE "ProductName", "Rufus"
-            VALUE "ProductVersion", "1.4.4.402"
+            VALUE "ProductVersion", "1.4.4.403"
         END
     END
     BLOCK "VarFileInfo"