chibicitiberiu/rainmeter-studio
1
0
Fork 0
mirror of https://github.com/chibicitiberiu/rainmeter-studio.git synced 2024-02-24 04:33:31 +00:00
Code Issues Releases Wiki Activity

Additional change to r1183.

Browse Source
This commit is contained in:
Birunthan Mohanathas 2012-02-13 19:01:17 +00:00
parent afa1c7cc20
commit 4d814bbd58
1 changed files with 29 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
Plugins/PluginRecycleManager/RecycleManager.cpp
View File

@ -34,7 +34,7 @@ struct MeasureData
struct BinData struct BinData
{ {
FILETIME lastAccess; FILETIME lastWrite;
HANDLE directory; HANDLE directory;
WCHAR drive; WCHAR drive;
}; };
@ -47,6 +47,7 @@ std::vector<BinData> g_BinData;
double g_BinCount = 0; double g_BinCount = 0;
double g_BinSize = 0; double g_BinSize = 0;
UINT g_UpdateCount = 0;
UINT g_InstanceCount = 0; UINT g_InstanceCount = 0;
HANDLE g_Thread = NULL; HANDLE g_Thread = NULL;
@ -83,8 +84,11 @@ PLUGIN_EXPORT double Update(void* data)
{ {
MeasureData* measure = (MeasureData*)data; MeasureData* measure = (MeasureData*)data;
if (!g_Thread) ++g_UpdateCount;
if (g_UpdateCount > g_InstanceCount && !g_Thread)
{ {
g_UpdateCount = 0;
WCHAR buffer[128]; WCHAR buffer[128];
DWORD len = GetLogicalDriveStrings(128, buffer); DWORD len = GetLogicalDriveStrings(128, buffer);
@ -110,16 +114,16 @@ PLUGIN_EXPORT double Update(void* data)
*pos = DRIVE_HANDLED; *pos = DRIVE_HANDLED;
++iter; ++iter;
if (data.lastAccess.dwHighDateTime != 0xFFFFFFFF && if (data.lastWrite.dwHighDateTime != 0xFFFFFFFF &&
data.lastAccess.dwLowDateTime != 0xFFFFFFFF) data.lastWrite.dwLowDateTime != 0xFFFFFFFF)
{ {
FILETIME lastAccess; FILETIME lastWrite;
GetFileTime(data.directory, NULL, &lastAccess, NULL); GetFileTime(data.directory, NULL, NULL, &lastWrite);
if (data.lastAccess.dwHighDateTime != lastAccess.dwHighDateTime || if (data.lastWrite.dwHighDateTime != lastWrite.dwHighDateTime ||
data.lastAccess.dwLowDateTime != lastAccess.dwLowDateTime) data.lastWrite.dwLowDateTime != lastWrite.dwLowDateTime)
{ {
data.lastAccess.dwHighDateTime = lastAccess.dwHighDateTime; data.lastWrite.dwHighDateTime = lastWrite.dwHighDateTime;
data.lastAccess.dwLowDateTime = lastAccess.dwLowDateTime; data.lastWrite.dwLowDateTime = lastWrite.dwLowDateTime;
changed = true; changed = true;
} }
} }
@ -142,20 +146,26 @@ PLUGIN_EXPORT double Update(void* data)
{ {
// New drive // New drive
BinData data = {0}; BinData data = {0};
data.directory = GetRecycleBinHandle(buffer[i]);
data.drive = buffer[i]; data.drive = buffer[i];
data.lastAccess.dwHighDateTime =
data.lastAccess.dwLowDateTime = data.directory ? 0 : 0xFFFFFFFF; WCHAR drive[] = L"\0:\\";
drive[0] = buffer[i];
if (GetDriveType(drive) == DRIVE_FIXED)
{
data.directory = GetRecycleBinHandle(buffer[i]);
}
data.lastWrite.dwHighDateTime =
data.lastWrite.dwLowDateTime = data.directory ? 0 : 0xFFFFFFFF;
g_BinData.push_back(data); g_BinData.push_back(data);
} }
} }
if (changed) if (changed)
{ {
g_Thread = (HANDLE)_beginthreadex(NULL, 64 * 1024, QueryRecycleBinThreadProc, NULL, 0, NULL); g_Thread = (HANDLE)_beginthreadex(NULL, 0, QueryRecycleBinThreadProc, NULL, 0, NULL);
} }
} }
return measure->count ? g_BinCount : g_BinSize; return measure->count ? g_BinCount : g_BinSize;
@ -265,7 +275,9 @@ HANDLE GetRecycleBinHandle(WCHAR drive)
{ {
do do
{ {
if (!(fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) || (fd.cFileName[0] == L'.')) if (!(fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ||
(fd.cFileName[0] == L'.') ||
(wcsnlen(fd.cFileName, 10) < 10))
{ {
continue; continue;
} }