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
|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
/*=====================================================================
**
** Source: SetCurrentDirectoryW.c (test 1)
**
** Purpose: Tests the PAL implementation of the SetCurrentDirectoryW function.
**
**
**===================================================================*/
#include <palsuite.h>
const char* szFileName = "blah";
const char* szDirName = "testing";
// In order to avoid the "chicken and egg" scenario, this is another
// method of getting the current directory. GetFullPathNameW is called with
// a dummy file name and then the file name is stripped off leaving the
// current working directory
BOOL GetCurrentDir(WCHAR* szwCurrentDir)
{
DWORD dwRc = 0;
WCHAR szwReturnedPath[_MAX_DIR+1];
LPWSTR pPathPtr;
WCHAR* szwFileName = NULL;
WCHAR* szwDirName = NULL;
int nCount = 0;
// use GetFullPathName to to get the current path by stripping
// the file name off the end
memset(szwReturnedPath, 0, sizeof(WCHAR)*(_MAX_DIR+1));
szwFileName = convert((char*)szFileName);
dwRc = GetFullPathNameW(szwFileName,
_MAX_DIR,
szwReturnedPath,
&pPathPtr);
if (dwRc == 0)
{
// GetFullPathName failed
Trace("SetCurrentDirectoryW: ERROR -> GetFullPathNameW failed "
"with error code: %ld.\n", GetLastError());
RemoveDirectoryW(szwDirName);
free(szwFileName);
return(FALSE);
}
// now strip the file name from the full path to get the current path
nCount = lstrlenW(szwReturnedPath) - lstrlenW(szwFileName);
memset(szwCurrentDir, 0, sizeof(WCHAR)*(_MAX_DIR+1));
wcsncpy(szwCurrentDir, szwReturnedPath, nCount - 1);
free(szwFileName);
return(TRUE);
}
int __cdecl main(int argc, char *argv[])
{
WCHAR* szwDirName = NULL;
WCHAR szwNewDir[_MAX_DIR+1];
WCHAR szwBuiltDir[_MAX_DIR+1];
WCHAR szwHomeDir[_MAX_DIR+1];
#if WIN32
WCHAR szwSlash[] = {'\\','\0'};
#else
WCHAR szwSlash[] = {'/','\0'};
#endif
if (0 != PAL_Initialize(argc,argv))
{
return FAIL;
}
// remove the directory just in case a previous run of the test didn't
szwDirName = convert((char*)szDirName);
RemoveDirectoryW(szwDirName);
// create a temp directory off the current directory
if (CreateDirectoryW(szwDirName, NULL) != TRUE)
{
Trace("SetCurrentDirectoryW: ERROR -> CreateDirectoryW failed "
"with error code: %ld.\n", GetLastError());
RemoveDirectoryW(szwDirName);
free(szwDirName);
Fail("");
}
// find out what the current "home" directory is
memset(szwHomeDir, 0, sizeof(WCHAR)*(_MAX_DIR+1));
if(GetCurrentDir(szwHomeDir) != TRUE)
{
RemoveDirectoryW(szwDirName);
free(szwDirName);
PAL_TerminateEx(FAIL);
return FAIL;
}
// set the current directory to the temp directory
if (SetCurrentDirectoryW(szwDirName) != TRUE)
{
Trace("SetCurrentDirectoryW: ERROR -> Unable to set current "
"directory. Failed with error code: %ld.\n", GetLastError());
RemoveDirectoryW(szwDirName);
free(szwDirName);
Fail("");
}
// append the temp name to the "home" directory
memset(szwBuiltDir, 0, sizeof(WCHAR)*(_MAX_DIR+1));
wcscpy(szwBuiltDir, szwHomeDir);
wcscat(szwBuiltDir, szwSlash);
wcscat(szwBuiltDir, szwDirName);
// get the new current directory
memset(szwNewDir, 0, sizeof(WCHAR)*(_MAX_DIR+1));
if(GetCurrentDir(szwNewDir) != TRUE)
{
RemoveDirectoryW(szwDirName);
free(szwDirName);
PAL_TerminateEx(FAIL);
return FAIL;
}
// compare the new current dir to the compiled current dir
if (wcsncmp(szwNewDir, szwBuiltDir, wcslen(szwNewDir)) != 0)
{
RemoveDirectoryW(szwDirName);
free(szwDirName);
Fail("SetCurrentDirectoryW: ERROR -> The set directory does not "
"compare to the built directory.\n");
}
// set the current dir back to the original
if (SetCurrentDirectoryW(szwHomeDir) != TRUE)
{
Trace("SetCurrentDirectoryW: ERROR -> Unable to set current "
"directory. Failed with error code: %ld.\n", GetLastError());
RemoveDirectoryW(szwDirName);
free(szwDirName);
Fail("");
}
// get the new current directory
memset(szwNewDir, 0, sizeof(WCHAR)*(_MAX_DIR+1));
if(GetCurrentDir(szwNewDir) != TRUE)
{
RemoveDirectoryW(szwDirName);
free(szwDirName);
PAL_TerminateEx(FAIL);
return FAIL;
}
// ensure it compares to the "home" directory which is where
// we should be now
if (wcsncmp(szwNewDir, szwHomeDir, wcslen(szwNewDir)) != 0)
{
RemoveDirectoryW(szwDirName);
free(szwDirName);
Fail("SetCurrentDirectoryW: ERROR -> The set directory does not "
"compare to the built directory.\n");
}
RemoveDirectoryW(szwDirName);
free(szwDirName);
PAL_Terminate();
return PASS;
}
|