1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
#include "recycler.h"
#define WIN32_LEAN_AND_MEAN
#include "windows.h"
#include <Shellapi.h> // Included for shell constants such as FO_* values
#include <shobjidl.h> // Required for necessary shell dependencies
#include <algorithm>
#include <string>
#include <cstdio>
#include <comdef.h>
void writeString(const wchar_t* input, wchar_t* output, size_t outputBufferLen)
{
const size_t newSize = min(wcslen(input) + 1, outputBufferLen); //including null-termination
memcpy(output, input, newSize * sizeof(wchar_t));
output[newSize-1] = 0; //if output buffer is too small...
}
std::wstring numberToHexString(const long number)
{
wchar_t result[100];
swprintf(result, 100, L"0x%08x", number);
return std::wstring(result);
}
void writeErrorMsg(const wchar_t* input, HRESULT hr, wchar_t* output, size_t outputBufferLen)
{
std::wstring formattedMsg(input);
formattedMsg += L" (";
formattedMsg += numberToHexString(hr);
formattedMsg += L": ";
formattedMsg += _com_error(hr).ErrorMessage();
formattedMsg += L")";
writeString(formattedMsg.c_str(), output, outputBufferLen);
}
//IShellItem resource management
template <class T>
class ReleaseAtExit
{
public:
ReleaseAtExit(T*& item) : item_(item) {}
~ReleaseAtExit()
{
if (item_ != NULL)
item_->Release();
}
private:
T*& item_;
};
bool Utility::moveToRecycleBin(const wchar_t* fileNames[],
size_t fileNo, //size of fileNames array
wchar_t* errorMessage,
size_t errorBufferLen)
{
HRESULT hr;
// Create the IFileOperation interface
IFileOperation* pfo = NULL;
ReleaseAtExit<IFileOperation> dummy(pfo);
hr = CoCreateInstance(CLSID_FileOperation,
NULL,
CLSCTX_ALL,
IID_PPV_ARGS(&pfo));
if (FAILED(hr))
{
writeErrorMsg(L"Error calling \"CoCreateInstance\".", hr, errorMessage, errorBufferLen);
return false;
}
// Set the operation flags. Turn off all UI
// from being shown to the user during the
// operation. This includes error, confirmation
// and progress dialogs.
hr = pfo->SetOperationFlags(FOF_ALLOWUNDO |
FOF_NOCONFIRMATION |
FOF_SILENT |
FOF_NOERRORUI);
if (FAILED(hr))
{
writeErrorMsg(L"Error calling \"SetOperationFlags\".", hr, errorMessage, errorBufferLen);
return false;
}
for (size_t i = 0; i < fileNo; ++i)
{
//create file/folder item object
IShellItem* psiFile = NULL;
ReleaseAtExit<IShellItem> dummy2(psiFile);
hr = SHCreateItemFromParsingName(fileNames[i],
NULL,
IID_PPV_ARGS(&psiFile));
if (FAILED(hr))
{
std::wstring message(L"Error calling \"SHCreateItemFromParsingName\" for file ");
message += std::wstring(L"\"") + fileNames[i] + L"\".";
writeErrorMsg(message.c_str(), hr, errorMessage, errorBufferLen);
return false;
}
hr = pfo->DeleteItem(psiFile, NULL);
if (FAILED(hr))
{
writeErrorMsg(L"Error calling \"DeleteItem\".", hr, errorMessage, errorBufferLen);
return false;
}
}
//perform actual operations
hr = pfo->PerformOperations();
if (FAILED(hr))
{
writeErrorMsg(L"Error calling \"PerformOperations\".", hr, errorMessage, errorBufferLen);
return false;
}
//check if errors occured: if FOFX_EARLYFAILURE is not used, PerformOperations() can return with success despite errors!
BOOL pfAnyOperationsAborted = FALSE;
hr = pfo->GetAnyOperationsAborted(&pfAnyOperationsAborted);
if (FAILED(hr))
{
writeErrorMsg(L"Error calling \"GetAnyOperationsAborted\".", hr, errorMessage, errorBufferLen);
return false;
}
if (pfAnyOperationsAborted == TRUE)
{
writeString(L"Operation did not complete successfully.", errorMessage, errorBufferLen);
return false;
}
return true;
}
|
