| 1 |  | 
|---|
| 2 | /*********************************************************************** | 
|---|
| 3 |  | 
|---|
| 4 | $Id: $ | 
|---|
| 5 |  | 
|---|
| 6 | Path handling utility functions | 
|---|
| 7 |  | 
|---|
| 8 | Copyright (c) 1993-98 M. Kimes | 
|---|
| 9 | Copyright (c) 2001, 2008 Steven H. Levine | 
|---|
| 10 |  | 
|---|
| 11 | 05 Jan 08 SHL Move from arccnrs.c and comp.c to here | 
|---|
| 12 | 06 Jan 08 GKY Add NormalizeCmdLine to check program strings on entry | 
|---|
| 13 | 29 Feb 08 GKY Changes to enable user settable command line length | 
|---|
| 14 |  | 
|---|
| 15 | ***********************************************************************/ | 
|---|
| 16 |  | 
|---|
| 17 | #include <stdlib.h> | 
|---|
| 18 | #include <string.h> | 
|---|
| 19 |  | 
|---|
| 20 | #define INCL_WIN | 
|---|
| 21 | #define INCL_DOS | 
|---|
| 22 | #define INCL_LONGLONG | 
|---|
| 23 |  | 
|---|
| 24 | #include "fm3dll.h"                     // needs_quoting | 
|---|
| 25 | #include "fm3str.h" | 
|---|
| 26 | #include "pathutil.h" | 
|---|
| 27 | #include "strips.h"                     // remove_first_occurence_of_character | 
|---|
| 28 | #include "valid.h"                      // needs_quoting | 
|---|
| 29 | #include "errutil.h"                    // Dos_Error... | 
|---|
| 30 | #include "strutil.h"                    // GetPString | 
|---|
| 31 | #include "wrappers.h"                   // xmalloc | 
|---|
| 32 | #include "fortify.h" | 
|---|
| 33 |  | 
|---|
| 34 | static PSZ pszSrcFile = __FILE__; | 
|---|
| 35 |  | 
|---|
| 36 | // #pragma data_seg(DATA1) | 
|---|
| 37 |  | 
|---|
| 38 | /** | 
|---|
| 39 | * Build full path name in callers buffer given directory | 
|---|
| 40 | * name and filename | 
|---|
| 41 | * @param pszPathName points to drive/directory if not NULL | 
|---|
| 42 | * @returns pointer to full path name in caller's buffer | 
|---|
| 43 | * @note OK for pszFullPathName and pszPathName to point to same buffer | 
|---|
| 44 | * | 
|---|
| 45 | */ | 
|---|
| 46 |  | 
|---|
| 47 | PSZ BldFullPathName(PSZ pszFullPathName, PSZ pszPathName, PSZ pszFileName) | 
|---|
| 48 | { | 
|---|
| 49 | UINT c = pszPathName ? strlen(pszPathName) : 0; | 
|---|
| 50 | if (c > 0) { | 
|---|
| 51 | memcpy(pszFullPathName, pszPathName, c); | 
|---|
| 52 | if (pszFullPathName[c - 1] != '\\') | 
|---|
| 53 | pszFullPathName[c++] = '\\'; | 
|---|
| 54 | } | 
|---|
| 55 | strcpy(pszFullPathName + c, pszFileName); | 
|---|
| 56 | return pszFullPathName; | 
|---|
| 57 | } | 
|---|
| 58 |  | 
|---|
| 59 | /** | 
|---|
| 60 | * Build quoted full path name in callers buffer given | 
|---|
| 61 | * directory name and filename | 
|---|
| 62 | * @param pszPathName points to drive/directory if not NULL | 
|---|
| 63 | * @returns pointer to quoted path name in caller's buffer | 
|---|
| 64 | */ | 
|---|
| 65 |  | 
|---|
| 66 | PSZ BldQuotedFullPathName(PSZ pszFullPathName, PSZ pszPathName, PSZ pszFileName) | 
|---|
| 67 | { | 
|---|
| 68 | UINT c = pszPathName ? strlen(pszPathName) : 0; | 
|---|
| 69 | BOOL q = needs_quoting(pszPathName) || needs_quoting(pszFileName); | 
|---|
| 70 | PSZ psz = pszFullPathName; | 
|---|
| 71 |  | 
|---|
| 72 | if (q) | 
|---|
| 73 | *psz++ = '"'; | 
|---|
| 74 | if (c > 0) { | 
|---|
| 75 | memcpy(psz, pszPathName, c); | 
|---|
| 76 | psz += c; | 
|---|
| 77 | if (*(psz - 1) != '\\') | 
|---|
| 78 | *psz++ = '\\'; | 
|---|
| 79 | } | 
|---|
| 80 | strcpy(psz, pszFileName); | 
|---|
| 81 | if (q) { | 
|---|
| 82 | psz += strlen(psz); | 
|---|
| 83 | *psz++ = '"'; | 
|---|
| 84 | *psz = 0; | 
|---|
| 85 | } | 
|---|
| 86 | return pszFullPathName; | 
|---|
| 87 | } | 
|---|
| 88 |  | 
|---|
| 89 | /** | 
|---|
| 90 | * Build quoted full path name in callers buffer given a filename | 
|---|
| 91 | * @returns pointer to quoted file name in caller's buffer | 
|---|
| 92 | */ | 
|---|
| 93 |  | 
|---|
| 94 | PSZ BldQuotedFileName(PSZ pszQuotedFileName, PSZ pszFileName) | 
|---|
| 95 | { | 
|---|
| 96 | BOOL q = needs_quoting(pszFileName); | 
|---|
| 97 | PSZ psz = pszQuotedFileName; | 
|---|
| 98 |  | 
|---|
| 99 | if (q) | 
|---|
| 100 | *psz++ = '"'; | 
|---|
| 101 | strcpy(psz, pszFileName); | 
|---|
| 102 | if (q) { | 
|---|
| 103 | psz += strlen(psz); | 
|---|
| 104 | *psz++ = '"'; | 
|---|
| 105 | *psz = 0; | 
|---|
| 106 | } | 
|---|
| 107 | return pszQuotedFileName; | 
|---|
| 108 | } | 
|---|
| 109 |  | 
|---|
| 110 | /** NormalizeCmdLine | 
|---|
| 111 | * Checks a command line for common errors (missing quotes, missing extension, | 
|---|
| 112 | * no space between exe and args etc) | 
|---|
| 113 | * Command line passed as pszCmdLine_ | 
|---|
| 114 | * A pointer to a buffer of the size MaxComLineStrg should be supplied in | 
|---|
| 115 | * pszWorkBuf. This is where the quoted etc as necessary command | 
|---|
| 116 | * line string will be returned. | 
|---|
| 117 | */ | 
|---|
| 118 |  | 
|---|
| 119 | PCSZ NormalizeCmdLine(PSZ pszWorkBuf, PSZ pszCmdLine_) | 
|---|
| 120 | { | 
|---|
| 121 | char *szCmdLine, *szArgs; | 
|---|
| 122 | char *offset = '\0', *offsetexe, *offsetcom, *offsetcmd, *offsetbtm, *offsetbat; | 
|---|
| 123 | APIRET ret; | 
|---|
| 124 | ULONG ulAppType; | 
|---|
| 125 | char *pszChar; | 
|---|
| 126 | FILEFINDBUF3 FindBuffer; | 
|---|
| 127 | ULONG ulResultBufLen = sizeof(FILEFINDBUF3); | 
|---|
| 128 | HDIR hdirFindHandle = HDIR_CREATE; | 
|---|
| 129 | ULONG ulFindCount = 1; | 
|---|
| 130 | PSZ pszNewCmdLine = pszWorkBuf; | 
|---|
| 131 |  | 
|---|
| 132 | szCmdLine = xmalloc(MaxComLineStrg, pszSrcFile, __LINE__); | 
|---|
| 133 | if (!szCmdLine) | 
|---|
| 134 | return pszCmdLine_; //already complained | 
|---|
| 135 | szArgs = xmalloc(MaxComLineStrg, pszSrcFile, __LINE__); | 
|---|
| 136 | if (!szArgs) { | 
|---|
| 137 | free(szCmdLine); | 
|---|
| 138 | return pszCmdLine_; //already complained | 
|---|
| 139 | } | 
|---|
| 140 | bstrip(pszCmdLine_); | 
|---|
| 141 | memset(pszWorkBuf, 0, MaxComLineStrg); | 
|---|
| 142 | strcpy(szCmdLine, pszCmdLine_); | 
|---|
| 143 | if (szCmdLine[0] != '\0') { | 
|---|
| 144 | offsetexe = strstr(strlwr(pszCmdLine_), ".exe"); | 
|---|
| 145 | offsetcmd = strstr(strlwr(pszCmdLine_), ".cmd"); | 
|---|
| 146 | offsetcom = strstr(strlwr(pszCmdLine_), ".com"); | 
|---|
| 147 | offsetbtm = strstr(strlwr(pszCmdLine_), ".btm"); | 
|---|
| 148 | offsetbat = strstr(strlwr(pszCmdLine_), ".bat"); | 
|---|
| 149 | if (offsetexe) | 
|---|
| 150 | offset = offsetexe; | 
|---|
| 151 | else { | 
|---|
| 152 | if (offsetcom) | 
|---|
| 153 | offset = offsetcom; | 
|---|
| 154 | else { | 
|---|
| 155 | if (offsetcmd) | 
|---|
| 156 | offset = offsetcmd; | 
|---|
| 157 | else { | 
|---|
| 158 | if (offsetbtm) | 
|---|
| 159 | offset = offsetbtm; | 
|---|
| 160 | else { | 
|---|
| 161 | if (offsetbat) | 
|---|
| 162 | offset = offsetexe; | 
|---|
| 163 | } | 
|---|
| 164 | } | 
|---|
| 165 | } | 
|---|
| 166 | } | 
|---|
| 167 | if (offset) { | 
|---|
| 168 | szCmdLine[offset + 4 - pszCmdLine_] = '\0'; | 
|---|
| 169 | strcpy(szArgs, &pszCmdLine_[offset + 4 - pszCmdLine_]); | 
|---|
| 170 | while (strchr(szCmdLine, '\"')) | 
|---|
| 171 | remove_first_occurence_of_character("\"", szCmdLine); | 
|---|
| 172 | if ((szArgs[0] == '\"' && szArgs[1] == ' ') || | 
|---|
| 173 | !strstr(pszCmdLine_, "\\:")|| | 
|---|
| 174 | strchr(szArgs, '\"') == strrchr(szArgs, '\"')) | 
|---|
| 175 | remove_first_occurence_of_character("\"", szArgs); | 
|---|
| 176 | if (strchr(szArgs, '\"') != strrchr(szArgs, '\"')) | 
|---|
| 177 | saymsg(MB_OK, HWND_DESKTOP, | 
|---|
| 178 | NullStr, | 
|---|
| 179 | GetPString(IDS_QUOTESINARGSTEXT), | 
|---|
| 180 | pszCmdLine_); | 
|---|
| 181 | if (!offsetexe) { | 
|---|
| 182 | ret = DosFindFirst(szCmdLine, &hdirFindHandle, FILE_NORMAL, &FindBuffer, | 
|---|
| 183 | ulResultBufLen, &ulFindCount, FIL_STANDARD); | 
|---|
| 184 | if (ret) { | 
|---|
| 185 | pszChar = szCmdLine; | 
|---|
| 186 | while (pszChar) { | 
|---|
| 187 | if (*pszChar == ' ') { | 
|---|
| 188 | *pszChar = '\0'; | 
|---|
| 189 | strcat(szCmdLine, ".exe"); | 
|---|
| 190 | ret = DosQueryAppType(szCmdLine, &ulAppType); | 
|---|
| 191 | //printf("%d %s\n", ret, szCmdLine); fflush(stdout); | 
|---|
| 192 | if (!ret) { | 
|---|
| 193 | strcpy(szArgs, pszCmdLine_ + strlen(szCmdLine) - 3); | 
|---|
| 194 | break; | 
|---|
| 195 | } | 
|---|
| 196 | } | 
|---|
| 197 | strcpy(szCmdLine, pszCmdLine_); | 
|---|
| 198 | pszChar++; | 
|---|
| 199 | } | 
|---|
| 200 | } | 
|---|
| 201 | } | 
|---|
| 202 | else | 
|---|
| 203 | ret = DosQueryAppType(szCmdLine, &ulAppType); | 
|---|
| 204 | BldQuotedFileName(pszNewCmdLine, szCmdLine); | 
|---|
| 205 | //printf("%d A", ret); fflush(stdout); | 
|---|
| 206 | if (ret) { | 
|---|
| 207 | ret = saymsg(MB_YESNO, | 
|---|
| 208 | HWND_DESKTOP, | 
|---|
| 209 | NullStr, | 
|---|
| 210 | GetPString(IDS_PROGRAMNOTFOUNDTEXT), | 
|---|
| 211 | pszCmdLine_); | 
|---|
| 212 | if (ret == MBID_YES){ | 
|---|
| 213 | if (szArgs[0] != ' ') | 
|---|
| 214 | strcat(pszNewCmdLine, " "); | 
|---|
| 215 | strcat(pszNewCmdLine, szArgs); | 
|---|
| 216 | } | 
|---|
| 217 | else{ | 
|---|
| 218 | fCancelAction = TRUE; | 
|---|
| 219 | pszNewCmdLine = pszCmdLine_; | 
|---|
| 220 | } | 
|---|
| 221 | } | 
|---|
| 222 | else{ | 
|---|
| 223 | if (szArgs[0] != ' ') | 
|---|
| 224 | strcat(pszNewCmdLine, " "); | 
|---|
| 225 | strcat(pszNewCmdLine, szArgs); | 
|---|
| 226 | } | 
|---|
| 227 |  | 
|---|
| 228 | } | 
|---|
| 229 | else if (szCmdLine && (!strchr(szCmdLine, '.') || | 
|---|
| 230 | strrchr(szCmdLine, '.' ) < strrchr(szCmdLine, '\\'))) { | 
|---|
| 231 | if (!strchr(szCmdLine, ' ')) { | 
|---|
| 232 | while (strchr(szCmdLine, '\"')) | 
|---|
| 233 | remove_first_occurence_of_character("\"", szCmdLine); | 
|---|
| 234 | strcat(szCmdLine, ".exe"); | 
|---|
| 235 | ret = DosFindFirst(szCmdLine, &hdirFindHandle, FILE_NORMAL, &FindBuffer, | 
|---|
| 236 | ulResultBufLen, &ulFindCount, FIL_STANDARD); | 
|---|
| 237 | //printf("%d", ret); fflush(stdout); | 
|---|
| 238 | } | 
|---|
| 239 | else { | 
|---|
| 240 | pszChar = szCmdLine; | 
|---|
| 241 | while (pszChar) { | 
|---|
| 242 | while (strchr(szCmdLine, '\"')) | 
|---|
| 243 | remove_first_occurence_of_character("\"", szCmdLine); | 
|---|
| 244 | if (*pszChar == ' ') { | 
|---|
| 245 | *pszChar = '\0'; | 
|---|
| 246 | strcat(szCmdLine, ".exe"); | 
|---|
| 247 | ret = DosQueryAppType(szCmdLine, &ulAppType); | 
|---|
| 248 | //printf("%d %s\n", ret, szCmdLine); fflush(stdout); | 
|---|
| 249 | if (!ret) { | 
|---|
| 250 | break; | 
|---|
| 251 | } | 
|---|
| 252 | } | 
|---|
| 253 | strcpy(szCmdLine, pszCmdLine_); | 
|---|
| 254 | pszChar++; | 
|---|
| 255 | } | 
|---|
| 256 | } | 
|---|
| 257 | if (!ret){ | 
|---|
| 258 | BldQuotedFileName(pszNewCmdLine, szCmdLine); | 
|---|
| 259 | strcpy(szArgs, pszCmdLine_ + strlen(szCmdLine) - 3); | 
|---|
| 260 | if ((szArgs[0] == '\"' && szArgs[1] == ' ') || | 
|---|
| 261 | !strstr(pszCmdLine_, "\\:" ) || | 
|---|
| 262 | strchr(szArgs, '\"') == strrchr(szArgs, '\"')) | 
|---|
| 263 | remove_first_occurence_of_character("\"", szArgs); | 
|---|
| 264 | if (strchr(szArgs, '\"') != strrchr(szArgs, '\"')) | 
|---|
| 265 | saymsg(MB_OK, HWND_DESKTOP, | 
|---|
| 266 | NullStr, | 
|---|
| 267 | GetPString(IDS_QUOTESINARGSTEXT), | 
|---|
| 268 | pszCmdLine_); | 
|---|
| 269 | if (szArgs[0] != ' ') | 
|---|
| 270 | strcat(pszNewCmdLine, " "); | 
|---|
| 271 | strcat(pszNewCmdLine, szArgs); | 
|---|
| 272 | } | 
|---|
| 273 | else { | 
|---|
| 274 | ret = saymsg(MB_OK, | 
|---|
| 275 | HWND_DESKTOP, | 
|---|
| 276 | NullStr, | 
|---|
| 277 | GetPString(IDS_PROGRAMNOTEXE2TEXT), | 
|---|
| 278 | pszCmdLine_); | 
|---|
| 279 | fCancelAction = TRUE; | 
|---|
| 280 | pszNewCmdLine = pszCmdLine_; | 
|---|
| 281 | } | 
|---|
| 282 | } | 
|---|
| 283 | else { | 
|---|
| 284 | pszChar = strrchr(szCmdLine, '.'); | 
|---|
| 285 | while (pszChar && *pszChar !=' ') { | 
|---|
| 286 | pszChar++; | 
|---|
| 287 | } | 
|---|
| 288 | *pszChar = '\0'; | 
|---|
| 289 | strcpy (szArgs, pszCmdLine_ + strlen(szCmdLine)); | 
|---|
| 290 | while (strchr(szCmdLine, '\"')) | 
|---|
| 291 | remove_first_occurence_of_character("\"", szCmdLine); | 
|---|
| 292 | if ((szArgs[0] == '\"' && szArgs[1] == ' ') || | 
|---|
| 293 | !strstr(pszCmdLine_, "\\:")|| | 
|---|
| 294 | strchr(szArgs, '\"') == strrchr(szArgs, '\"')) | 
|---|
| 295 | remove_first_occurence_of_character("\"", szArgs); | 
|---|
| 296 | if (strchr(szArgs, '\"') != strrchr(szArgs, '\"')) | 
|---|
| 297 | saymsg(MB_OK, HWND_DESKTOP, | 
|---|
| 298 | NullStr, | 
|---|
| 299 | GetPString(IDS_QUOTESINARGSTEXT), | 
|---|
| 300 | pszCmdLine_); | 
|---|
| 301 | ret = DosFindFirst(szCmdLine, &hdirFindHandle, FILE_NORMAL, &FindBuffer, | 
|---|
| 302 | ulResultBufLen, &ulFindCount, FIL_STANDARD); | 
|---|
| 303 |  | 
|---|
| 304 | BldQuotedFileName(pszNewCmdLine, szCmdLine); | 
|---|
| 305 | //printf("%d %s ", ret, szCmdLine); fflush(stdout); | 
|---|
| 306 | if (ret) { | 
|---|
| 307 | ret = saymsg(MB_YESNO, | 
|---|
| 308 | HWND_DESKTOP, | 
|---|
| 309 | NullStr, | 
|---|
| 310 | GetPString(IDS_PROGRAMNOTFOUNDTEXT), | 
|---|
| 311 | pszCmdLine_); | 
|---|
| 312 | if (ret == MBID_YES) { | 
|---|
| 313 | if (szArgs[0] != ' ') | 
|---|
| 314 | strcat(pszNewCmdLine, " "); | 
|---|
| 315 | strcat(pszNewCmdLine, szArgs); | 
|---|
| 316 | } | 
|---|
| 317 | else { | 
|---|
| 318 | fCancelAction = TRUE; | 
|---|
| 319 | pszWorkBuf = pszCmdLine_; | 
|---|
| 320 | } | 
|---|
| 321 | } | 
|---|
| 322 | ret = saymsg(MB_YESNOCANCEL, | 
|---|
| 323 | HWND_DESKTOP, | 
|---|
| 324 | NullStr, | 
|---|
| 325 | GetPString(IDS_PROGRAMNOTEXE3TEXT), | 
|---|
| 326 | pszCmdLine_, pszNewCmdLine); | 
|---|
| 327 | if (ret == MBID_YES){ | 
|---|
| 328 | if (szArgs[0] != ' ') | 
|---|
| 329 | strcat(pszNewCmdLine, " "); | 
|---|
| 330 | strcat(pszNewCmdLine, szArgs); | 
|---|
| 331 | } | 
|---|
| 332 | if (ret == MBID_CANCEL){ | 
|---|
| 333 | fCancelAction = TRUE; | 
|---|
| 334 | pszNewCmdLine = pszCmdLine_; | 
|---|
| 335 | } | 
|---|
| 336 | } | 
|---|
| 337 | } | 
|---|
| 338 | free(szArgs); | 
|---|
| 339 | free(szCmdLine); | 
|---|
| 340 | return pszWorkBuf; | 
|---|
| 341 | } | 
|---|
| 342 |  | 
|---|
| 343 | #pragma alloc_text(PATHUTIL,BldFullPathName) | 
|---|
| 344 | #pragma alloc_text(PATHUTIL,BldQuotedFileName) | 
|---|
| 345 | #pragma alloc_text(PATHUTIL,BldQuotedFullPathName) | 
|---|
| 346 | #pragma alloc_text(PATHUTIL,NormalizeCmdLine) | 
|---|