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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
|
#ifndef NULLSOFT_UTILITY_AUTOWIDEFN_H
#define NULLSOFT_UTILITY_AUTOWIDEFN_H
/* Winamp defines this, but this little block lets us use this thing outside of Winamp */
#ifndef FILENAME_SIZE
#define FILENAME_SIZE (MAX_PATH*4)
#define REMOVE_FILENAME_SIZE
#endif
#include <windows.h>
#include "AutoWide.h"
#include "AutoChar.h"
#include <shlwapi.h>
/*
Tries to find a filename that underwent a destructive Unicode-to-ANSI conversion
*/
#pragma warning(push)
#pragma warning(disable:4995)
class AutoWideFn
{
public:
AutoWideFn(const char *narrowFn)
{
wideFn[0]=0;
if (!narrowFn)
return;
CreateFile_HACK(narrowFn, wideFn);
}
operator wchar_t *() { return wideFn; }
bool unicode_find(/*char *path,*/ char *pattern, wchar_t *out, UINT out_ptr, bool dir, HANDLE *f)
{
WIN32_FIND_DATAW fd = {0};
if (*f == INVALID_HANDLE_VALUE)
{
lstrcpyW(out + out_ptr, L"*");
*f = FindFirstFileW(out, &fd);
out[out_ptr] = 0;
if (*f == INVALID_HANDLE_VALUE) return 0;
}
else
{
if (!FindNextFileW(*f, &fd))
{
FindClose(*f);
*f = INVALID_HANDLE_VALUE;
return 0;
}
}
if (*f == INVALID_HANDLE_VALUE) return 0;
char temp[MAX_PATH*2 + 1] = {0};
do
{
if (dir)
{
if (!(fd.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)) continue;
}
else
{
if (fd.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY) continue;
}
WideCharToMultiByte(CP_ACP, 0, fd.cFileName, -1, temp, sizeof(temp), 0, 0);
//wcstombs(temp,fd.cFileName,sizeof(temp));
if (!_stricmp(temp, pattern))
{ //found
lstrcpyW(out + out_ptr, fd.cFileName);
return 1;
}
WideCharToMultiByte(CP_ACP, 0, fd.cAlternateFileName, -1, temp, sizeof(temp), 0, 0);
if (!_stricmp(temp, pattern))
{ //found
lstrcpyW(out + out_ptr, fd.cFileName);
return 1;
}
}
while (FindNextFileW(*f, &fd));
FindClose(*f);
*f = INVALID_HANDLE_VALUE;
return 0;
}
bool unicode_open_recur(char *path, char *ptr, wchar_t *out, UINT out_ptr)
{
char * next = strchr(ptr, '\\');
if (next)
{ //dig another dir
HANDLE f = INVALID_HANDLE_VALUE;
bool found;
do
{
next[0] = 0;
char * zz = _strdup(ptr);
char bk = *ptr;
*ptr = 0;
found = unicode_find(/*path,*/ zz, out, out_ptr, 1, &f);
free(zz);
*ptr = bk;
next[0] = '\\';
if (found)
{
UINT op_bk = out_ptr;
while (out_ptr < FILENAME_SIZE && out[out_ptr]) out_ptr++;
out[out_ptr++] = '\\';
if (unicode_open_recur(path, next + 1, out, out_ptr))
{
if (f != INVALID_HANDLE_VALUE) FindClose(f);
return 1;
}
out_ptr = op_bk;
out[out_ptr] = 0;
}
} while (found);
}
else
{ //final dir
HANDLE f = INVALID_HANDLE_VALUE;
char * zz = _strdup(ptr);
char bk = *ptr;
*ptr = 0;
bool found = unicode_find(/*path,*/ zz, out, out_ptr, 0, &f);
if (!found)
{
if (f != INVALID_HANDLE_VALUE)
{
FindClose(f);
f = INVALID_HANDLE_VALUE;
}
found = unicode_find(/*path,*/ zz, out, out_ptr, 1, &f);
}
free(zz);
*ptr = bk;
if (f != INVALID_HANDLE_VALUE) FindClose(f);
return found;
}
return 0;
}
void CreateFile_HACK(const char * path, wchar_t out[FILENAME_SIZE])
{
MultiByteToWideChar(CP_ACP, 0, path, -1, out, FILENAME_SIZE);
if (PathIsURLW(out))
return ;
if (!StrChrW(out, L'?'))
return ; // no unconvertables? Great!
// if (PathFileExistsW(out))
// return ; // no unconvertables? Great!
bool found = false;
memset(out, 0, FILENAME_SIZE * sizeof(wchar_t));
char * _p = _strdup(path);
char * t = strchr(_p, '\\');
if (t)
{
char bk = t[1];
t[1] = 0;
UINT o = MultiByteToWideChar(CP_ACP, 0, _p, -1, out, FILENAME_SIZE);
o--;
t[1] = bk;
found = unicode_open_recur(_p, t + 1, out, o);
}
else
found = unicode_open_recur(_p, _p, out, 0);
free(_p);
if (!found)
MultiByteToWideChar(CP_ACP, 0, path, -1, out, FILENAME_SIZE);
}
private:
wchar_t wideFn[FILENAME_SIZE];
};
#pragma warning(pop)
#ifdef REMOVE_FILENAME_SIZE
#undef FILENAME_SIZE
#endif
#endif
|
