1 | // Licensed to the .NET Foundation under one or more agreements. |
2 | // The .NET Foundation licenses this file to you under the MIT license. |
3 | // See the LICENSE file in the project root for more information. |
4 | |
5 | /*============================================================ |
6 | ** |
7 | ** Source: CreateProcessW/test1/parentprocess.c |
8 | ** |
9 | ** Purpose: Test to ensure CreateProcessW starts a new process. This test |
10 | ** launches a child process, and examines a file written by the child. |
11 | ** This process (the parent process) reads the file created by the child and |
12 | ** compares the value the child wrote to the file. (a const char *) |
13 | ** |
14 | ** Dependencies: GetCurrentDirectory |
15 | ** MultiByteToWideChar |
16 | ** wcslen |
17 | ** strlen |
18 | ** WideCharToMultiByte |
19 | ** WaitForSingleObject |
20 | ** fopen |
21 | ** fclose |
22 | ** Fail |
23 | ** |
24 | |
25 | ** |
26 | **=========================================================*/ |
27 | |
28 | #define UNICODE |
29 | #include <palsuite.h> |
30 | |
31 | const WCHAR szCommonFileW[] = |
32 | {'c','h','i','l','d','d','a','t','a','.','t','m','p','\0'}; |
33 | |
34 | const WCHAR szChildFileW[] = u"paltest_createprocessw_test1_child" ; |
35 | |
36 | const WCHAR szPathDelimW[] = {'\\','\0'}; |
37 | |
38 | const char *szCommonStringA = "058d2d057111a313aa82401c2e856002\0" ; |
39 | |
40 | /* |
41 | * Take two wide strings representing file and directory names |
42 | * (dirName, fileName), join the strings with the appropriate path |
43 | * delimiter and populate a wide character buffer (absPathName) with |
44 | * the resulting string. |
45 | * |
46 | * Returns: The number of wide characters in the resulting string. |
47 | * 0 is returned on Error. |
48 | */ |
49 | int |
50 | mkAbsoluteFilenameW ( |
51 | LPWSTR dirName, |
52 | DWORD dwDirLength, |
53 | LPCWSTR fileName, |
54 | DWORD dwFileLength, |
55 | LPWSTR absPathName ) |
56 | { |
57 | extern const WCHAR szPathDelimW[]; |
58 | |
59 | DWORD sizeDN, sizeFN, sizeAPN; |
60 | |
61 | sizeDN = wcslen( dirName ); |
62 | sizeFN = wcslen( fileName ); |
63 | sizeAPN = (sizeDN + 1 + sizeFN + 1); |
64 | |
65 | /* insure ((dirName + DELIM + fileName + \0) =< _MAX_PATH ) */ |
66 | if ( sizeAPN > _MAX_PATH ) |
67 | { |
68 | return ( 0 ); |
69 | } |
70 | |
71 | wcsncpy(absPathName, dirName, dwDirLength +1); |
72 | wcsncpy(absPathName, szPathDelimW, 2); |
73 | wcsncpy(absPathName, fileName, dwFileLength +1); |
74 | |
75 | return (sizeAPN); |
76 | |
77 | } |
78 | |
79 | int __cdecl main( int argc, char **argv ) |
80 | |
81 | { |
82 | |
83 | STARTUPINFOW si; |
84 | PROCESS_INFORMATION pi; |
85 | |
86 | static FILE * fp; |
87 | |
88 | DWORD dwFileLength; |
89 | DWORD dwDirLength; |
90 | DWORD dwSize; |
91 | |
92 | size_t cslen; |
93 | |
94 | char szReadStringA[256]; |
95 | |
96 | char szAbsPathNameA[_MAX_PATH]; |
97 | WCHAR szDirNameW[_MAX_DIR]; |
98 | WCHAR absPathBuf[_MAX_PATH]; |
99 | WCHAR *szAbsPathNameW; |
100 | |
101 | |
102 | if(0 != (PAL_Initialize(argc, argv))) |
103 | { |
104 | return ( FAIL ); |
105 | } |
106 | |
107 | ZeroMemory ( &si, sizeof(si) ); |
108 | si.cb = sizeof(si); |
109 | ZeroMemory ( &pi, sizeof(pi) ); |
110 | |
111 | szAbsPathNameW=&absPathBuf[0]; |
112 | dwFileLength = wcslen( szChildFileW ); |
113 | |
114 | dwDirLength = GetCurrentDirectory(_MAX_PATH, szDirNameW); |
115 | |
116 | if (0 == dwDirLength) |
117 | { |
118 | Fail ("GetCurrentDirectory call failed. Could not get " |
119 | "current working directory\n. Exiting.\n" ); |
120 | } |
121 | |
122 | dwSize = mkAbsoluteFilenameW( szDirNameW, dwDirLength, szChildFileW, |
123 | dwFileLength, szAbsPathNameW ); |
124 | |
125 | if (0 == dwSize) |
126 | { |
127 | Fail ("Palsuite Code: mkAbsoluteFilename() call failed. Could " |
128 | "not build absolute path name to file\n. Exiting.\n" ); |
129 | } |
130 | |
131 | if ( !CreateProcessW ( NULL, |
132 | szAbsPathNameW, |
133 | NULL, |
134 | NULL, |
135 | FALSE, |
136 | CREATE_NEW_CONSOLE, |
137 | NULL, |
138 | NULL, |
139 | &si, |
140 | &pi ) |
141 | ) |
142 | { |
143 | Fail ( "CreateProcess call failed. GetLastError returned %d\n" , |
144 | GetLastError() ); |
145 | } |
146 | |
147 | WaitForSingleObject ( pi.hProcess, INFINITE ); |
148 | |
149 | szAbsPathNameW=&absPathBuf[0]; |
150 | |
151 | dwFileLength = wcslen( szCommonFileW ); |
152 | |
153 | dwSize = mkAbsoluteFilenameW( szDirNameW, dwDirLength, szCommonFileW, |
154 | dwFileLength, szAbsPathNameW ); |
155 | |
156 | /* set the string length for the open call*/ |
157 | |
158 | if (0 == dwSize) |
159 | { |
160 | Fail ("Palsuite Code: mkAbsoluteFilename() call failed. Could " |
161 | "not build absolute path name to file\n. Exiting.\n" ); |
162 | } |
163 | |
164 | WideCharToMultiByte (CP_ACP, 0, szAbsPathNameW, -1, szAbsPathNameA, |
165 | (dwSize + 1), NULL, NULL); |
166 | |
167 | if ( NULL == ( fp = fopen ( szAbsPathNameA , "r" ) ) ) |
168 | { |
169 | Fail ("%s\nunable to open %s\nfor reading. Exiting.\n" , argv[0], |
170 | szAbsPathNameA ); |
171 | } |
172 | |
173 | cslen = strlen ( szCommonStringA ); |
174 | |
175 | if ( NULL == fgets( szReadStringA, (cslen + 1), fp )) |
176 | { |
177 | /* |
178 | * A return value of NULL indicates an error condition or an |
179 | * EOF condition |
180 | */ |
181 | Fail ("%s\nunable to read file\n%s\nszReadStringA is %s\n" |
182 | "Exiting.\n" , argv[0], szAbsPathNameA, |
183 | szReadStringA ); |
184 | } |
185 | |
186 | if ( 0 != strncmp( szReadStringA, szCommonStringA, cslen )) |
187 | { |
188 | Fail ("string comparison failed.\n szReadStringA is %s and\n" |
189 | "szCommonStringA is %s\n" , szReadStringA, |
190 | szCommonStringA ); |
191 | } |
192 | else |
193 | { |
194 | Trace ("string comparison passed.\n" ); |
195 | } |
196 | |
197 | if (0 != (fclose ( fp ))) |
198 | { |
199 | Trace ("%s unable to close file %s. This may cause a file pointer " |
200 | "leak. Continuing.\n" , argv[0], szAbsPathNameA ); |
201 | } |
202 | |
203 | /* Close process and thread handle */ |
204 | CloseHandle ( pi.hProcess ); |
205 | CloseHandle ( pi.hThread ); |
206 | |
207 | PAL_Terminate(); |
208 | return ( PASS ); |
209 | |
210 | } |
211 | |