do memory card write test in a separate thread

2004-04-07 01:18:11 +00:00
parent 7135f8cc1d
commit d93f758024
9 changed files with 340 additions and 20 deletions
@@ -0,0 +1,128 @@
+#include "global.h"
+#include "MemoryCardDriverThreaded_Windows.h"
+#include "RageUtil.h"
+#include <io.h>
+#include <fcntl.h>
+#include "RageFileManager.h"
+#include "RageLog.h"
+#include <windows.h>
+
+MemoryCardDriverThreaded_Windows::MemoryCardDriverThreaded_Windows()
+{
+	StartThread();
+}
+
+typedef const CString& CCStringRef;
+
+bool TestWrite( CCStringRef sDrive )
+{
+	// TODO: Use RageFileDirect here to detect ready state?
+	TCHAR szVolumeNameBuffer[MAX_PATH];
+	DWORD dwVolumeSerialNumber;
+	DWORD dwMaximumComponentLength;
+	DWORD lpFileSystemFlags;
+	TCHAR szFileSystemNameBuffer[MAX_PATH];
+	BOOL bReady = GetVolumeInformation( 
+		sDrive,
+		szVolumeNameBuffer,
+		sizeof(szVolumeNameBuffer),
+		&dwVolumeSerialNumber,
+		&dwMaximumComponentLength,
+		&lpFileSystemFlags,
+		szFileSystemNameBuffer,
+		sizeof(szFileSystemNameBuffer) );
+	if( !bReady )
+		return false;
+
+	// Try to write a file.
+	// TODO: Can we use RageFile for this?
+	CString sFile = sDrive + "temp";
+	FILE* fp = fopen( sFile, "w" );
+	if( fp == NULL )
+		return false;
+	fclose( fp );
+	remove( sFile );
+
+	return true;
+}
+
+void MemoryCardDriverThreaded_Windows::MountThreadMain()
+{
+	DWORD dwLastLogicalDrives = 0;
+	
+	while( !m_bShutdown )
+	{
+		DWORD dwNewLogicalDrives = ::GetLogicalDrives();
+		if( dwNewLogicalDrives != dwLastLogicalDrives )
+		{
+			vector<UsbStorageDeviceEx> vNewStorageDevices;
+
+			const int MAX_DRIVES = 26;
+			for( int i=2; i<MAX_DRIVES; i++ )	// skip 'a:" and "b:"
+			{
+				DWORD mask = (1 << i);
+				if( dwNewLogicalDrives & mask ) // drive letter is valid
+				{
+					CString sDrive = ssprintf( "%c:\\", 'a'+i%26 );
+
+					LOG->Warn( "Found drive %s", sDrive.c_str() );
+
+					if( GetDriveType(sDrive) == DRIVE_REMOVABLE )	// is a removable drive
+					{
+						UsbStorageDeviceEx usbd;
+						usbd.sOsMountDir = sDrive;
+						usbd.bWriteTestSucceeded = TestWrite( sDrive );
+						vNewStorageDevices.push_back( usbd );
+					}
+				}
+			}
+
+			{
+				LockMut( m_mutexStorageDevices );
+				m_bStorageDevicesChanged = true;
+				m_vStorageDevices = vNewStorageDevices;
+			}
+	    }
+		dwLastLogicalDrives = dwNewLogicalDrives;
+
+		::Sleep( 100 );
+	}
+}
+
+void MemoryCardDriverThreaded_Windows::Mount( UsbStorageDevice* pDevice, CString sMountPoint )
+{
+	ASSERT( !pDevice->sOsMountDir.empty() );
+
+	/* Unmount any previous mounts for this mountpoint. */
+	vector<RageFileManager::DriverLocation> Mounts;
+	FILEMAN->GetLoadedDrivers( Mounts );
+	for( unsigned i = 0; i < Mounts.size(); ++i )
+	{
+		if( Mounts[i].Type.CompareNoCase( "dir" ) )
+				continue; // wrong type
+		if( Mounts[i].Root.CompareNoCase( pDevice->sOsMountDir ) )
+				continue; // wrong root
+		FILEMAN->Unmount( Mounts[i].Type, Mounts[i].Root, Mounts[i].MountPoint );
+	}
+
+	FILEMAN->Mount( "dir", pDevice->sOsMountDir, sMountPoint.c_str() );
+	LOG->Trace( "FILEMAN->Mount %s %s", pDevice->sOsMountDir.c_str(), sMountPoint.c_str() );
+}
+
+void MemoryCardDriverThreaded_Windows::Unmount( UsbStorageDevice* pDevice, CString sMountPoint )
+{
+
+		// nothing to do here...
+}
+
+void MemoryCardDriverThreaded_Windows::Flush( UsbStorageDevice* pDevice )
+{
+	// Do we need anything here?  I don't lose data if ejecting 
+	// soon after a write.  From the activity LED, it looks like 
+	// Windows flushes automatically every ~2 seconds. -Chris
+}
+
+/*
+ * Copyright (c) 2003 by the person(s) listed below.  All rights reserved.
+ *	Chris Danford
+ */