rewrite code to allow for file/directory retry on fails

source/ngc/dvd.cpp

@@ -23,10 +23,11 @@ extern "C" {
 }
 #endif
 
+#include "snes9xGX.h"
 #include "menu.h"
 #include "gcunzip.h"
 #include "filebrowser.h"
-#include "snes9xGX.h"
+#include "fileop.h"
 
 #define MAXDVDFILES 2000
 
@@ -502,7 +503,7 @@ getentry (int entrycount, unsigned char dvdbuffer[])
  * The return value is number of files collected, or -1 on failure.
  ***************************************************************************/
 int
-ParseDVDdirectory ()
+ParseDVDdirectory (bool change)
 {
 	int pdlength;
 	u64 pdoffset;
@@ -514,6 +515,9 @@ ParseDVDdirectory ()
 	// reset browser
 	ResetBrowser();
 	
+	if(change && !ChangeInterface(METHOD_DVD, NOTSILENT))
+		return 0;
+
 	pdoffset = rdoffset = dvddir;
 	pdlength = dvddirlength;
 	filecount = 0;
@@ -601,7 +605,7 @@ static bool SwitchDVDFolderR(char * dir, int maxDepth)
 		if(browserList[dirindex].isdir) // only parse directories
 		{
 			UpdateDirName(METHOD_DVD);
-			ParseDVDdirectory();
+			ParseDVDdirectory(false);
 		}
 		else
 		{
@@ -636,7 +640,7 @@ bool SwitchDVDFolder(char origdir[])
 	dvddir = dvdrootdir;
 	dvddirlength = dvdrootlength;
 	browser.dir[0] = 0;
-	ParseDVDdirectory();
+	ParseDVDdirectory(true);
 
 	return SwitchDVDFolderR(dirptr, 0);
 }

source/ngc/dvd.h

@@ -14,7 +14,7 @@
 #define _NGCDVD_
 
 bool MountDVD(bool silent);
-int ParseDVDdirectory();
+int ParseDVDdirectory(bool change);
 void SetDVDdirectory(u64 dir, int length);
 bool SwitchDVDFolder(char dir[]);
 

source/ngc/filebrowser.cpp

@@ -501,11 +501,11 @@ int BrowserChangeFolder(int method)
 	switch (method)
 	{
 		case METHOD_DVD:
-			ParseDVDdirectory();
+			ParseDVDdirectory(true);
 			break;
 
 		default:
-			ParseDirectory();
+			ParseDirectory(method);
 			break;
 	}
 
@@ -530,25 +530,18 @@ OpenGameList ()
 	if(method == METHOD_AUTO)
 		method = autoLoadMethod();
 
-	if(ChangeInterface(method, NOTSILENT))
-	{
 	// change current dir to roms directory
 	switch(method)
 	{
 		case METHOD_DVD:
 			browser.dir[0] = 0;
-				ParseDVDdirectory(); // Parse root directory
+			if(ParseDVDdirectory(true)) // Parse root directory
 				SwitchDVDFolder(GCSettings.LoadFolder); // switch to ROM folder
 			break;
 		default:
 			sprintf(browser.dir, "/%s", GCSettings.LoadFolder);
-				ParseDirectory(); // Parse root directory
+			ParseDirectory(method); // Parse root directory
 			break;
 	}
-	}
-	else
-	{
-		ResetBrowser();
-	}
 	return browser.numEntries;
 }

source/ngc/fileop.cpp

@@ -182,6 +182,7 @@ bool MountFAT(int method)
 		unmountRequired[method] = false;
 		fatUnmount(rootdir);
 		disc->shutdown();
+		isMounted[method] = false;
 	}
 	if(!isMounted[method])
 	{
@@ -269,37 +270,45 @@ bool ChangeInterface(int method, bool silent)
  * Browse subdirectories
  **************************************************************************/
 int
-ParseDirectory()
+ParseDirectory(int method)
 {
-	DIR_ITER *dir;
+	DIR_ITER *dir = NULL;
 	char fulldir[MAXPATHLEN];
 	char filename[MAXPATHLEN];
 	char tmpname[MAXPATHLEN];
 	struct stat filestat;
 	struct tm * timeinfo;
 	char msg[128];
-
+	int retry = 1;
-	ShowAction("Loading...");
 
 	// reset browser
 	ResetBrowser();
 
-	// add device to path
+	ShowAction("Loading...");
-	sprintf(fulldir, "%s%s", rootdir, browser.dir);
 
 	// open the directory
+	while(dir == NULL && retry == 1)
+	{
+		if(ChangeInterface(method, NOTSILENT))
+		{
+			sprintf(fulldir, "%s%s", rootdir, browser.dir); // add device to path
 			dir = diropen(fulldir);
-
 			if(dir == NULL)
 			{
+				unmountRequired[method] = true;
 				sprintf(msg, "Error opening %s", fulldir);
-		ErrorPrompt(msg);
+				retry = ErrorPromptRetry(msg);
+			}
+		}
+	}
 
-		// if we can't open the dir, open root dir
+	// if we can't open the dir, try opening the root dir
+	if (dir == NULL)
+	{
+		if(ChangeInterface(method, SILENT))
+		{
 			sprintf(browser.dir,"/");
-
 			dir = diropen(rootdir);
-
 			if (dir == NULL)
 			{
 				sprintf(msg, "Error opening %s", rootdir);
@@ -307,6 +316,7 @@ ParseDirectory()
 				return -1;
 			}
 		}
+	}
 
 	// index files/folders
 	int entryNum = 0;
@@ -430,9 +440,8 @@ LoadFile (char * rbuffer, char *filepath, u32 length, int method, bool silent)
 	char zipbuffer[2048];
 	u32 size = 0;
 	u32 readsize = 0;
-
+	char fullpath[MAXPATHLEN];
-	if(!ChangeInterface(method, silent))
+	int retry = 1;
-		return 0;
 
 	switch(method)
 	{
@@ -451,10 +460,12 @@ LoadFile (char * rbuffer, char *filepath, u32 length, int method, bool silent)
 	// since we're loading a file
 	LWP_SuspendThread (devicethread);
 
-	// add device to filepath
+	// open the file
-	char fullpath[1024];
+	while(!size && retry == 1)
-	sprintf(fullpath, "%s%s", rootdir, filepath);
+	{
-
+		if(ChangeInterface(method, silent))
+		{
+			sprintf(fullpath, "%s%s", rootdir, filepath); // add device to filepath
 			file = fopen (fullpath, "rb");
 
 			if(file > 0)
@@ -488,9 +499,9 @@ LoadFile (char * rbuffer, char *filepath, u32 length, int method, bool silent)
 								if(size - offset > 1024*512) nextread = 1024*512;
 								else nextread = size-offset;
 								ShowProgress ("Loading...", offset, size);
-						readsize = fread (rbuffer + offset, 1, nextread, file); // read in 512K chunks
+								readsize = fread (rbuffer + offset, 1, nextread, file); // read in next chunk
 
-						if(readsize <= 0 || readsize > (1024*512))
+								if(readsize <= 0 || readsize > nextread)
 									break; // read failure
 
 								if(readsize > 0)
@@ -505,10 +516,19 @@ LoadFile (char * rbuffer, char *filepath, u32 length, int method, bool silent)
 				}
 				fclose (file);
 			}
-	if(!size && !silent)
+		}
+		if(!size)
+		{
+			if(!silent)
 			{
 				unmountRequired[method] = true;
-		ErrorPrompt("Error loading file!");
+				retry = ErrorPromptRetry("Error loading file!");
+			}
+			else
+			{
+				retry = 0;
+			}
+		}
 	}
 
 	// go back to checking if devices were inserted/removed
@@ -529,12 +549,11 @@ u32 LoadFile(char * filepath, int method, bool silent)
 u32
 SaveFile (char * buffer, char *filepath, u32 datasize, int method, bool silent)
 {
-	if(datasize == 0)
+	char fullpath[MAXPATHLEN];
-		return 0;
-
 	u32 written = 0;
+	int retry = 1;
 
-	if(!ChangeInterface(method, silent))
+	if(datasize == 0)
 		return 0;
 
 	ShowAction("Saving...");
@@ -542,38 +561,41 @@ SaveFile (char * buffer, char *filepath, u32 datasize, int method, bool silent)
 	if(method == METHOD_MC_SLOTA || method == METHOD_MC_SLOTB)
 	{
 		if(method == METHOD_MC_SLOTA)
-			written = SaveMCFile (buffer, CARD_SLOTA, filepath, datasize, silent);
+			return SaveMCFile (buffer, CARD_SLOTA, filepath, datasize, silent);
 		else
-			written = SaveMCFile (buffer, CARD_SLOTB, filepath, datasize, silent);
+			return SaveMCFile (buffer, CARD_SLOTB, filepath, datasize, silent);
 	}
-	else
+
-    {
 	// stop checking if devices were removed/inserted
 	// since we're saving a file
 	LWP_SuspendThread (devicethread);
 
-		// add device to filepath
+	while(!written && retry == 1)
-		char fullpath[1024];
+	{
-		sprintf(fullpath, "%s%s", rootdir, filepath);
+		if(ChangeInterface(method, silent))
-
+		{
-		// open file for writing
+			sprintf(fullpath, "%s%s", rootdir, filepath); // add device to filepath
 			file = fopen (fullpath, "wb");
 
 			if (file > 0)
 			{
 				written = fwrite (savebuffer, 1, datasize, file);
+				if(written < datasize) written = 0;
 				fclose (file);
 			}
-
+		}
 		if(!written)
+		{
 			unmountRequired[method] = true;
+			if(!silent)
+				retry = ErrorPromptRetry("Error saving file!");
+			else
+				retry = 0;
+		}
+	}
 
 	// go back to checking if devices were inserted/removed
 	LWP_ResumeThread (devicethread);
-    }
-
-	if(!written && !silent)
-		ErrorPrompt("Error saving file!");
 
 	CancelAction();
     return written;

source/ngc/fileop.h

@@ -27,7 +27,7 @@ void InitDeviceThread();
 void MountAllFAT();
 void UnmountAllFAT();
 bool ChangeInterface(int method, bool silent);
-int ParseDirectory();
+int ParseDirectory(int method);
 void AllocSaveBuffer();
 void FreeSaveBuffer();
 u32 LoadFile(char * rbuffer, char *filepath, u32 length, int method, bool silent);

source/ngc/menu.cpp

@@ -179,6 +179,7 @@ WindowPrompt(const char *title, const char *msg, const char *btn1Label, const ch
 		promptWindow.Append(&btn2);
 
 	promptWindow.SetEffect(EFFECT_SLIDE_TOP | EFFECT_SLIDE_IN, 50);
+	CancelAction();
 	HaltGui();
 	mainWindow->SetState(STATE_DISABLED);
 	mainWindow->Append(&promptWindow);
@@ -465,13 +466,16 @@ ShowAction (const char *msg)
 
 void ErrorPrompt(const char *msg)
 {
-	CancelAction();
 	WindowPrompt("Error", msg, "OK", NULL);
 }
 
+int ErrorPromptRetry(const char *msg)
+{
+	return WindowPrompt("Error", msg, "Retry", "Cancel");
+}
+
 void InfoPrompt(const char *msg)
 {
-	CancelAction();
 	WindowPrompt("Information", msg, "OK", NULL);
 }
 
@@ -1435,7 +1439,7 @@ static int MenuGameSaves(int action)
 	else
 	{
 		strncpy(browser.dir, GCSettings.SaveFolder, 200);
-		ParseDirectory();
+		ParseDirectory(GCSettings.SaveMethod);
 	}
 
 	for(i=0; i < browser.numEntries; i++)
@@ -1485,6 +1489,17 @@ static int MenuGameSaves(int action)
 						{
 							saves.previewImg[j] = new GuiImageData(savebuffer);
 						}
+						/*char scrfile2[1024];
+						sprintf(scrfile, "%s/%s.png", GCSettings.SaveFolder, tmp);
+						sprintf(scrfile2, "%s/resave/%s.png", GCSettings.SaveFolder, tmp);
+
+						AllocSaveBuffer();
+						int scrsize = LoadFile(scrfile, GCSettings.SaveMethod, SILENT);
+						if(scrsize > 0)
+						{
+							//saves.previewImg[j] = new GuiImageData(savebuffer);
+							SaveFile(scrfile2, scrsize, GCSettings.SaveMethod, SILENT);
+						}*/
 						FreeSaveBuffer();
 					}
 					strftime(saves.date[j], 20, "%a %b %d", &browserList[j].mtime);

source/ngc/menu.h

@@ -16,6 +16,7 @@
 void InitGUIThreads();
 void MainMenu (int menuitem);
 void ErrorPrompt(const char * msg);
+int ErrorPromptRetry(const char * msg);
 void InfoPrompt(const char * msg);
 void ShowAction (const char *msg);
 void CancelAction();