blob: 74ee2ec15a458f875834aaedd81cdcc2415d6b2d [file] [log] [blame]
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001/* vi:set ts=8 sts=4 sw=4 noet:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
Bram Moolenaar9810cfb2019-12-11 21:23:00 +010011 * filepath.c: dealing with file names and paths.
Bram Moolenaarb005cd82019-09-04 15:54:55 +020012 */
13
14#include "vim.h"
15
16#ifdef MSWIN
17/*
18 * Functions for ":8" filename modifier: get 8.3 version of a filename.
19 */
20
21/*
22 * Get the short path (8.3) for the filename in "fnamep".
23 * Only works for a valid file name.
24 * When the path gets longer "fnamep" is changed and the allocated buffer
25 * is put in "bufp".
26 * *fnamelen is the length of "fnamep" and set to 0 for a nonexistent path.
27 * Returns OK on success, FAIL on failure.
28 */
29 static int
30get_short_pathname(char_u **fnamep, char_u **bufp, int *fnamelen)
31{
32 int l, len;
Bram Moolenaar3f396972019-10-30 04:10:06 +010033 WCHAR *newbuf;
34 WCHAR *wfname;
Bram Moolenaarb005cd82019-09-04 15:54:55 +020035
Bram Moolenaar3f396972019-10-30 04:10:06 +010036 len = MAXPATHL;
37 newbuf = malloc(len * sizeof(*newbuf));
38 if (newbuf == NULL)
39 return FAIL;
40
41 wfname = enc_to_utf16(*fnamep, NULL);
42 if (wfname == NULL)
43 {
44 vim_free(newbuf);
45 return FAIL;
46 }
47
48 l = GetShortPathNameW(wfname, newbuf, len);
Bram Moolenaarb005cd82019-09-04 15:54:55 +020049 if (l > len - 1)
50 {
Bram Moolenaar26262f82019-09-04 20:59:15 +020051 // If that doesn't work (not enough space), then save the string
52 // and try again with a new buffer big enough.
Bram Moolenaar3f396972019-10-30 04:10:06 +010053 WCHAR *newbuf_t = newbuf;
54 newbuf = vim_realloc(newbuf, (l + 1) * sizeof(*newbuf));
Bram Moolenaarb005cd82019-09-04 15:54:55 +020055 if (newbuf == NULL)
Bram Moolenaar3f396972019-10-30 04:10:06 +010056 {
57 vim_free(wfname);
58 vim_free(newbuf_t);
Bram Moolenaarb005cd82019-09-04 15:54:55 +020059 return FAIL;
Bram Moolenaar3f396972019-10-30 04:10:06 +010060 }
Bram Moolenaar26262f82019-09-04 20:59:15 +020061 // Really should always succeed, as the buffer is big enough.
Bram Moolenaar3f396972019-10-30 04:10:06 +010062 l = GetShortPathNameW(wfname, newbuf, l+1);
Bram Moolenaarb005cd82019-09-04 15:54:55 +020063 }
Bram Moolenaar3f396972019-10-30 04:10:06 +010064 if (l != 0)
65 {
66 char_u *p = utf16_to_enc(newbuf, NULL);
Bram Moolenaarc74fbfe2020-04-06 22:56:28 +020067
Bram Moolenaar3f396972019-10-30 04:10:06 +010068 if (p != NULL)
69 {
70 vim_free(*bufp);
71 *fnamep = *bufp = p;
72 }
73 else
74 {
75 vim_free(wfname);
76 vim_free(newbuf);
77 return FAIL;
78 }
79 }
80 vim_free(wfname);
81 vim_free(newbuf);
Bram Moolenaarb005cd82019-09-04 15:54:55 +020082
Bram Moolenaar2ade7142019-11-04 20:36:50 +010083 *fnamelen = l == 0 ? l : (int)STRLEN(*bufp);
Bram Moolenaarb005cd82019-09-04 15:54:55 +020084 return OK;
85}
86
87/*
88 * Get the short path (8.3) for the filename in "fname". The converted
89 * path is returned in "bufp".
90 *
91 * Some of the directories specified in "fname" may not exist. This function
92 * will shorten the existing directories at the beginning of the path and then
93 * append the remaining non-existing path.
94 *
95 * fname - Pointer to the filename to shorten. On return, contains the
96 * pointer to the shortened pathname
97 * bufp - Pointer to an allocated buffer for the filename.
98 * fnamelen - Length of the filename pointed to by fname
99 *
100 * Returns OK on success (or nothing done) and FAIL on failure (out of memory).
101 */
102 static int
103shortpath_for_invalid_fname(
104 char_u **fname,
105 char_u **bufp,
106 int *fnamelen)
107{
108 char_u *short_fname, *save_fname, *pbuf_unused;
109 char_u *endp, *save_endp;
110 char_u ch;
111 int old_len, len;
112 int new_len, sfx_len;
113 int retval = OK;
114
Bram Moolenaar26262f82019-09-04 20:59:15 +0200115 // Make a copy
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200116 old_len = *fnamelen;
117 save_fname = vim_strnsave(*fname, old_len);
118 pbuf_unused = NULL;
119 short_fname = NULL;
120
Bram Moolenaar26262f82019-09-04 20:59:15 +0200121 endp = save_fname + old_len - 1; // Find the end of the copy
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200122 save_endp = endp;
123
124 /*
125 * Try shortening the supplied path till it succeeds by removing one
126 * directory at a time from the tail of the path.
127 */
128 len = 0;
129 for (;;)
130 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200131 // go back one path-separator
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200132 while (endp > save_fname && !after_pathsep(save_fname, endp + 1))
133 --endp;
134 if (endp <= save_fname)
Bram Moolenaar26262f82019-09-04 20:59:15 +0200135 break; // processed the complete path
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200136
137 /*
138 * Replace the path separator with a NUL and try to shorten the
139 * resulting path.
140 */
141 ch = *endp;
142 *endp = 0;
143 short_fname = save_fname;
144 len = (int)STRLEN(short_fname) + 1;
145 if (get_short_pathname(&short_fname, &pbuf_unused, &len) == FAIL)
146 {
147 retval = FAIL;
148 goto theend;
149 }
Bram Moolenaar26262f82019-09-04 20:59:15 +0200150 *endp = ch; // preserve the string
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200151
152 if (len > 0)
Bram Moolenaar26262f82019-09-04 20:59:15 +0200153 break; // successfully shortened the path
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200154
Bram Moolenaar26262f82019-09-04 20:59:15 +0200155 // failed to shorten the path. Skip the path separator
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200156 --endp;
157 }
158
159 if (len > 0)
160 {
161 /*
162 * Succeeded in shortening the path. Now concatenate the shortened
163 * path with the remaining path at the tail.
164 */
165
Bram Moolenaar217e1b82019-12-01 21:41:28 +0100166 // Compute the length of the new path.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200167 sfx_len = (int)(save_endp - endp) + 1;
168 new_len = len + sfx_len;
169
170 *fnamelen = new_len;
171 vim_free(*bufp);
172 if (new_len > old_len)
173 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200174 // There is not enough space in the currently allocated string,
175 // copy it to a buffer big enough.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200176 *fname = *bufp = vim_strnsave(short_fname, new_len);
177 if (*fname == NULL)
178 {
179 retval = FAIL;
180 goto theend;
181 }
182 }
183 else
184 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200185 // Transfer short_fname to the main buffer (it's big enough),
186 // unless get_short_pathname() did its work in-place.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200187 *fname = *bufp = save_fname;
188 if (short_fname != save_fname)
189 vim_strncpy(save_fname, short_fname, len);
190 save_fname = NULL;
191 }
192
Bram Moolenaar26262f82019-09-04 20:59:15 +0200193 // concat the not-shortened part of the path
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200194 vim_strncpy(*fname + len, endp, sfx_len);
195 (*fname)[new_len] = NUL;
196 }
197
198theend:
199 vim_free(pbuf_unused);
200 vim_free(save_fname);
201
202 return retval;
203}
204
205/*
206 * Get a pathname for a partial path.
207 * Returns OK for success, FAIL for failure.
208 */
209 static int
210shortpath_for_partial(
211 char_u **fnamep,
212 char_u **bufp,
213 int *fnamelen)
214{
215 int sepcount, len, tflen;
216 char_u *p;
217 char_u *pbuf, *tfname;
218 int hasTilde;
219
Bram Moolenaar26262f82019-09-04 20:59:15 +0200220 // Count up the path separators from the RHS.. so we know which part
221 // of the path to return.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200222 sepcount = 0;
223 for (p = *fnamep; p < *fnamep + *fnamelen; MB_PTR_ADV(p))
224 if (vim_ispathsep(*p))
225 ++sepcount;
226
Bram Moolenaar26262f82019-09-04 20:59:15 +0200227 // Need full path first (use expand_env() to remove a "~/")
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200228 hasTilde = (**fnamep == '~');
229 if (hasTilde)
230 pbuf = tfname = expand_env_save(*fnamep);
231 else
232 pbuf = tfname = FullName_save(*fnamep, FALSE);
233
234 len = tflen = (int)STRLEN(tfname);
235
236 if (get_short_pathname(&tfname, &pbuf, &len) == FAIL)
237 return FAIL;
238
239 if (len == 0)
240 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200241 // Don't have a valid filename, so shorten the rest of the
242 // path if we can. This CAN give us invalid 8.3 filenames, but
243 // there's not a lot of point in guessing what it might be.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200244 len = tflen;
245 if (shortpath_for_invalid_fname(&tfname, &pbuf, &len) == FAIL)
246 return FAIL;
247 }
248
Bram Moolenaar26262f82019-09-04 20:59:15 +0200249 // Count the paths backward to find the beginning of the desired string.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200250 for (p = tfname + len - 1; p >= tfname; --p)
251 {
252 if (has_mbyte)
253 p -= mb_head_off(tfname, p);
254 if (vim_ispathsep(*p))
255 {
256 if (sepcount == 0 || (hasTilde && sepcount == 1))
257 break;
258 else
259 sepcount --;
260 }
261 }
262 if (hasTilde)
263 {
264 --p;
265 if (p >= tfname)
266 *p = '~';
267 else
268 return FAIL;
269 }
270 else
271 ++p;
272
Bram Moolenaar26262f82019-09-04 20:59:15 +0200273 // Copy in the string - p indexes into tfname - allocated at pbuf
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200274 vim_free(*bufp);
275 *fnamelen = (int)STRLEN(p);
276 *bufp = pbuf;
277 *fnamep = p;
278
279 return OK;
280}
281#endif // MSWIN
282
283/*
284 * Adjust a filename, according to a string of modifiers.
285 * *fnamep must be NUL terminated when called. When returning, the length is
286 * determined by *fnamelen.
287 * Returns VALID_ flags or -1 for failure.
288 * When there is an error, *fnamep is set to NULL.
289 */
290 int
291modify_fname(
292 char_u *src, // string with modifiers
293 int tilde_file, // "~" is a file name, not $HOME
294 int *usedlen, // characters after src that are used
295 char_u **fnamep, // file name so far
296 char_u **bufp, // buffer for allocated file name or NULL
297 int *fnamelen) // length of fnamep
298{
299 int valid = 0;
300 char_u *tail;
301 char_u *s, *p, *pbuf;
302 char_u dirname[MAXPATHL];
303 int c;
304 int has_fullname = 0;
Bram Moolenaard816cd92020-02-04 22:23:09 +0100305 int has_homerelative = 0;
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200306#ifdef MSWIN
307 char_u *fname_start = *fnamep;
308 int has_shortname = 0;
309#endif
310
311repeat:
Bram Moolenaar26262f82019-09-04 20:59:15 +0200312 // ":p" - full path/file_name
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200313 if (src[*usedlen] == ':' && src[*usedlen + 1] == 'p')
314 {
315 has_fullname = 1;
316
317 valid |= VALID_PATH;
318 *usedlen += 2;
319
Bram Moolenaar26262f82019-09-04 20:59:15 +0200320 // Expand "~/path" for all systems and "~user/path" for Unix and VMS
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200321 if ((*fnamep)[0] == '~'
322#if !defined(UNIX) && !(defined(VMS) && defined(USER_HOME))
323 && ((*fnamep)[1] == '/'
324# ifdef BACKSLASH_IN_FILENAME
325 || (*fnamep)[1] == '\\'
326# endif
327 || (*fnamep)[1] == NUL)
328#endif
329 && !(tilde_file && (*fnamep)[1] == NUL)
330 )
331 {
332 *fnamep = expand_env_save(*fnamep);
Bram Moolenaar26262f82019-09-04 20:59:15 +0200333 vim_free(*bufp); // free any allocated file name
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200334 *bufp = *fnamep;
335 if (*fnamep == NULL)
336 return -1;
337 }
338
Bram Moolenaar26262f82019-09-04 20:59:15 +0200339 // When "/." or "/.." is used: force expansion to get rid of it.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200340 for (p = *fnamep; *p != NUL; MB_PTR_ADV(p))
341 {
342 if (vim_ispathsep(*p)
343 && p[1] == '.'
344 && (p[2] == NUL
345 || vim_ispathsep(p[2])
346 || (p[2] == '.'
347 && (p[3] == NUL || vim_ispathsep(p[3])))))
348 break;
349 }
350
Bram Moolenaar26262f82019-09-04 20:59:15 +0200351 // FullName_save() is slow, don't use it when not needed.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200352 if (*p != NUL || !vim_isAbsName(*fnamep))
353 {
354 *fnamep = FullName_save(*fnamep, *p != NUL);
Bram Moolenaar26262f82019-09-04 20:59:15 +0200355 vim_free(*bufp); // free any allocated file name
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200356 *bufp = *fnamep;
357 if (*fnamep == NULL)
358 return -1;
359 }
360
361#ifdef MSWIN
362# if _WIN32_WINNT >= 0x0500
363 if (vim_strchr(*fnamep, '~') != NULL)
364 {
365 // Expand 8.3 filename to full path. Needed to make sure the same
366 // file does not have two different names.
367 // Note: problem does not occur if _WIN32_WINNT < 0x0500.
368 WCHAR *wfname = enc_to_utf16(*fnamep, NULL);
369 WCHAR buf[_MAX_PATH];
370
371 if (wfname != NULL)
372 {
373 if (GetLongPathNameW(wfname, buf, _MAX_PATH))
374 {
375 char_u *p = utf16_to_enc(buf, NULL);
376
377 if (p != NULL)
378 {
379 vim_free(*bufp); // free any allocated file name
380 *bufp = *fnamep = p;
381 }
382 }
383 vim_free(wfname);
384 }
385 }
386# endif
387#endif
Bram Moolenaar26262f82019-09-04 20:59:15 +0200388 // Append a path separator to a directory.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200389 if (mch_isdir(*fnamep))
390 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200391 // Make room for one or two extra characters.
Bram Moolenaar71ccd032020-06-12 22:59:11 +0200392 *fnamep = vim_strnsave(*fnamep, STRLEN(*fnamep) + 2);
Bram Moolenaar26262f82019-09-04 20:59:15 +0200393 vim_free(*bufp); // free any allocated file name
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200394 *bufp = *fnamep;
395 if (*fnamep == NULL)
396 return -1;
397 add_pathsep(*fnamep);
398 }
399 }
400
Bram Moolenaar26262f82019-09-04 20:59:15 +0200401 // ":." - path relative to the current directory
402 // ":~" - path relative to the home directory
403 // ":8" - shortname path - postponed till after
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200404 while (src[*usedlen] == ':'
405 && ((c = src[*usedlen + 1]) == '.' || c == '~' || c == '8'))
406 {
407 *usedlen += 2;
408 if (c == '8')
409 {
410#ifdef MSWIN
Bram Moolenaar26262f82019-09-04 20:59:15 +0200411 has_shortname = 1; // Postpone this.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200412#endif
413 continue;
414 }
415 pbuf = NULL;
Bram Moolenaar26262f82019-09-04 20:59:15 +0200416 // Need full path first (use expand_env() to remove a "~/")
Bram Moolenaard816cd92020-02-04 22:23:09 +0100417 if (!has_fullname && !has_homerelative)
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200418 {
Bram Moolenaar0e390f42020-06-10 13:12:28 +0200419 if ((c == '.' || c == '~') && **fnamep == '~')
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200420 p = pbuf = expand_env_save(*fnamep);
421 else
422 p = pbuf = FullName_save(*fnamep, FALSE);
423 }
424 else
425 p = *fnamep;
426
427 has_fullname = 0;
428
429 if (p != NULL)
430 {
431 if (c == '.')
432 {
Bram Moolenaard816cd92020-02-04 22:23:09 +0100433 size_t namelen;
434
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200435 mch_dirname(dirname, MAXPATHL);
Bram Moolenaard816cd92020-02-04 22:23:09 +0100436 if (has_homerelative)
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200437 {
Bram Moolenaard816cd92020-02-04 22:23:09 +0100438 s = vim_strsave(dirname);
439 if (s != NULL)
440 {
441 home_replace(NULL, s, dirname, MAXPATHL, TRUE);
442 vim_free(s);
443 }
444 }
445 namelen = STRLEN(dirname);
446
447 // Do not call shorten_fname() here since it removes the prefix
448 // even though the path does not have a prefix.
449 if (fnamencmp(p, dirname, namelen) == 0)
450 {
451 p += namelen;
Bram Moolenaara78e9c62020-02-05 21:14:00 +0100452 if (vim_ispathsep(*p))
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200453 {
Bram Moolenaara78e9c62020-02-05 21:14:00 +0100454 while (*p && vim_ispathsep(*p))
455 ++p;
456 *fnamep = p;
457 if (pbuf != NULL)
458 {
459 // free any allocated file name
460 vim_free(*bufp);
461 *bufp = pbuf;
462 pbuf = NULL;
463 }
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200464 }
465 }
466 }
467 else
468 {
469 home_replace(NULL, p, dirname, MAXPATHL, TRUE);
Bram Moolenaar26262f82019-09-04 20:59:15 +0200470 // Only replace it when it starts with '~'
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200471 if (*dirname == '~')
472 {
473 s = vim_strsave(dirname);
474 if (s != NULL)
475 {
476 *fnamep = s;
477 vim_free(*bufp);
478 *bufp = s;
Bram Moolenaard816cd92020-02-04 22:23:09 +0100479 has_homerelative = TRUE;
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200480 }
481 }
482 }
483 vim_free(pbuf);
484 }
485 }
486
487 tail = gettail(*fnamep);
488 *fnamelen = (int)STRLEN(*fnamep);
489
Bram Moolenaar26262f82019-09-04 20:59:15 +0200490 // ":h" - head, remove "/file_name", can be repeated
491 // Don't remove the first "/" or "c:\"
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200492 while (src[*usedlen] == ':' && src[*usedlen + 1] == 'h')
493 {
494 valid |= VALID_HEAD;
495 *usedlen += 2;
496 s = get_past_head(*fnamep);
497 while (tail > s && after_pathsep(s, tail))
498 MB_PTR_BACK(*fnamep, tail);
499 *fnamelen = (int)(tail - *fnamep);
500#ifdef VMS
501 if (*fnamelen > 0)
Bram Moolenaar26262f82019-09-04 20:59:15 +0200502 *fnamelen += 1; // the path separator is part of the path
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200503#endif
504 if (*fnamelen == 0)
505 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200506 // Result is empty. Turn it into "." to make ":cd %:h" work.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200507 p = vim_strsave((char_u *)".");
508 if (p == NULL)
509 return -1;
510 vim_free(*bufp);
511 *bufp = *fnamep = tail = p;
512 *fnamelen = 1;
513 }
514 else
515 {
516 while (tail > s && !after_pathsep(s, tail))
517 MB_PTR_BACK(*fnamep, tail);
518 }
519 }
520
Bram Moolenaar26262f82019-09-04 20:59:15 +0200521 // ":8" - shortname
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200522 if (src[*usedlen] == ':' && src[*usedlen + 1] == '8')
523 {
524 *usedlen += 2;
525#ifdef MSWIN
526 has_shortname = 1;
527#endif
528 }
529
530#ifdef MSWIN
531 /*
532 * Handle ":8" after we have done 'heads' and before we do 'tails'.
533 */
534 if (has_shortname)
535 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200536 // Copy the string if it is shortened by :h and when it wasn't copied
537 // yet, because we are going to change it in place. Avoids changing
538 // the buffer name for "%:8".
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200539 if (*fnamelen < (int)STRLEN(*fnamep) || *fnamep == fname_start)
540 {
541 p = vim_strnsave(*fnamep, *fnamelen);
542 if (p == NULL)
543 return -1;
544 vim_free(*bufp);
545 *bufp = *fnamep = p;
546 }
547
Bram Moolenaar26262f82019-09-04 20:59:15 +0200548 // Split into two implementations - makes it easier. First is where
549 // there isn't a full name already, second is where there is.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200550 if (!has_fullname && !vim_isAbsName(*fnamep))
551 {
552 if (shortpath_for_partial(fnamep, bufp, fnamelen) == FAIL)
553 return -1;
554 }
555 else
556 {
557 int l = *fnamelen;
558
Bram Moolenaar26262f82019-09-04 20:59:15 +0200559 // Simple case, already have the full-name.
560 // Nearly always shorter, so try first time.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200561 if (get_short_pathname(fnamep, bufp, &l) == FAIL)
562 return -1;
563
564 if (l == 0)
565 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200566 // Couldn't find the filename, search the paths.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200567 l = *fnamelen;
568 if (shortpath_for_invalid_fname(fnamep, bufp, &l) == FAIL)
569 return -1;
570 }
571 *fnamelen = l;
572 }
573 }
574#endif // MSWIN
575
Bram Moolenaar26262f82019-09-04 20:59:15 +0200576 // ":t" - tail, just the basename
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200577 if (src[*usedlen] == ':' && src[*usedlen + 1] == 't')
578 {
579 *usedlen += 2;
580 *fnamelen -= (int)(tail - *fnamep);
581 *fnamep = tail;
582 }
583
Bram Moolenaar26262f82019-09-04 20:59:15 +0200584 // ":e" - extension, can be repeated
585 // ":r" - root, without extension, can be repeated
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200586 while (src[*usedlen] == ':'
587 && (src[*usedlen + 1] == 'e' || src[*usedlen + 1] == 'r'))
588 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200589 // find a '.' in the tail:
590 // - for second :e: before the current fname
591 // - otherwise: The last '.'
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200592 if (src[*usedlen + 1] == 'e' && *fnamep > tail)
593 s = *fnamep - 2;
594 else
595 s = *fnamep + *fnamelen - 1;
596 for ( ; s > tail; --s)
597 if (s[0] == '.')
598 break;
Bram Moolenaar26262f82019-09-04 20:59:15 +0200599 if (src[*usedlen + 1] == 'e') // :e
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200600 {
601 if (s > tail)
602 {
603 *fnamelen += (int)(*fnamep - (s + 1));
604 *fnamep = s + 1;
605#ifdef VMS
Bram Moolenaar26262f82019-09-04 20:59:15 +0200606 // cut version from the extension
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200607 s = *fnamep + *fnamelen - 1;
608 for ( ; s > *fnamep; --s)
609 if (s[0] == ';')
610 break;
611 if (s > *fnamep)
612 *fnamelen = s - *fnamep;
613#endif
614 }
615 else if (*fnamep <= tail)
616 *fnamelen = 0;
617 }
Bram Moolenaar26262f82019-09-04 20:59:15 +0200618 else // :r
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200619 {
Bram Moolenaarb1892952019-10-08 23:26:50 +0200620 char_u *limit = *fnamep;
621
622 if (limit < tail)
623 limit = tail;
624 if (s > limit) // remove one extension
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200625 *fnamelen = (int)(s - *fnamep);
626 }
627 *usedlen += 2;
628 }
629
Bram Moolenaar26262f82019-09-04 20:59:15 +0200630 // ":s?pat?foo?" - substitute
631 // ":gs?pat?foo?" - global substitute
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200632 if (src[*usedlen] == ':'
633 && (src[*usedlen + 1] == 's'
634 || (src[*usedlen + 1] == 'g' && src[*usedlen + 2] == 's')))
635 {
636 char_u *str;
637 char_u *pat;
638 char_u *sub;
639 int sep;
640 char_u *flags;
641 int didit = FALSE;
642
643 flags = (char_u *)"";
644 s = src + *usedlen + 2;
645 if (src[*usedlen + 1] == 'g')
646 {
647 flags = (char_u *)"g";
648 ++s;
649 }
650
651 sep = *s++;
652 if (sep)
653 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200654 // find end of pattern
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200655 p = vim_strchr(s, sep);
656 if (p != NULL)
657 {
Bram Moolenaar71ccd032020-06-12 22:59:11 +0200658 pat = vim_strnsave(s, p - s);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200659 if (pat != NULL)
660 {
661 s = p + 1;
Bram Moolenaar26262f82019-09-04 20:59:15 +0200662 // find end of substitution
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200663 p = vim_strchr(s, sep);
664 if (p != NULL)
665 {
Bram Moolenaar71ccd032020-06-12 22:59:11 +0200666 sub = vim_strnsave(s, p - s);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200667 str = vim_strnsave(*fnamep, *fnamelen);
668 if (sub != NULL && str != NULL)
669 {
670 *usedlen = (int)(p + 1 - src);
671 s = do_string_sub(str, pat, sub, NULL, flags);
672 if (s != NULL)
673 {
674 *fnamep = s;
675 *fnamelen = (int)STRLEN(s);
676 vim_free(*bufp);
677 *bufp = s;
678 didit = TRUE;
679 }
680 }
681 vim_free(sub);
682 vim_free(str);
683 }
684 vim_free(pat);
685 }
686 }
Bram Moolenaar26262f82019-09-04 20:59:15 +0200687 // after using ":s", repeat all the modifiers
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200688 if (didit)
689 goto repeat;
690 }
691 }
692
693 if (src[*usedlen] == ':' && src[*usedlen + 1] == 'S')
694 {
Bram Moolenaar26262f82019-09-04 20:59:15 +0200695 // vim_strsave_shellescape() needs a NUL terminated string.
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200696 c = (*fnamep)[*fnamelen];
697 if (c != NUL)
698 (*fnamep)[*fnamelen] = NUL;
699 p = vim_strsave_shellescape(*fnamep, FALSE, FALSE);
700 if (c != NUL)
701 (*fnamep)[*fnamelen] = c;
702 if (p == NULL)
703 return -1;
704 vim_free(*bufp);
705 *bufp = *fnamep = p;
706 *fnamelen = (int)STRLEN(p);
707 *usedlen += 2;
708 }
709
710 return valid;
711}
712
Bram Moolenaar273af492020-09-25 23:49:01 +0200713/*
714 * Shorten the path of a file from "~/foo/../.bar/fname" to "~/f/../.b/fname"
715 * "trim_len" specifies how many characters to keep for each directory.
716 * Must be 1 or more.
717 * It's done in-place.
718 */
719 static void
720shorten_dir_len(char_u *str, int trim_len)
721{
722 char_u *tail, *s, *d;
723 int skip = FALSE;
724 int dirchunk_len = 0;
725
726 tail = gettail(str);
727 d = str;
728 for (s = str; ; ++s)
729 {
730 if (s >= tail) // copy the whole tail
731 {
732 *d++ = *s;
733 if (*s == NUL)
734 break;
735 }
736 else if (vim_ispathsep(*s)) // copy '/' and next char
737 {
738 *d++ = *s;
739 skip = FALSE;
740 dirchunk_len = 0;
741 }
742 else if (!skip)
743 {
744 *d++ = *s; // copy next char
745 if (*s != '~' && *s != '.') // and leading "~" and "."
746 {
747 ++dirchunk_len; // only count word chars for the size
748
749 // keep copying chars until we have our preferred length (or
750 // until the above if/else branches move us along)
751 if (dirchunk_len >= trim_len)
752 skip = TRUE;
753 }
754
755 if (has_mbyte)
756 {
757 int l = mb_ptr2len(s);
758
759 while (--l > 0)
760 *d++ = *++s;
761 }
762 }
763 }
764}
765
766/*
767 * Shorten the path of a file from "~/foo/../.bar/fname" to "~/f/../.b/fname"
768 * It's done in-place.
769 */
770 void
771shorten_dir(char_u *str)
772{
773 shorten_dir_len(str, 1);
774}
775
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200776#if defined(FEAT_EVAL) || defined(PROTO)
777
778/*
779 * "chdir(dir)" function
780 */
781 void
782f_chdir(typval_T *argvars, typval_T *rettv)
783{
784 char_u *cwd;
785 cdscope_T scope = CDSCOPE_GLOBAL;
786
787 rettv->v_type = VAR_STRING;
788 rettv->vval.v_string = NULL;
789
790 if (argvars[0].v_type != VAR_STRING)
Bram Moolenaarc5809432021-03-27 21:23:30 +0100791 {
Bram Moolenaard816cd92020-02-04 22:23:09 +0100792 // Returning an empty string means it failed.
Bram Moolenaar002bc792020-06-05 22:33:42 +0200793 // No error message, for historic reasons.
Bram Moolenaarc5809432021-03-27 21:23:30 +0100794 if (in_vim9script())
795 (void) check_for_string_arg(argvars, 0);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200796 return;
Bram Moolenaarc5809432021-03-27 21:23:30 +0100797 }
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200798
799 // Return the current directory
800 cwd = alloc(MAXPATHL);
801 if (cwd != NULL)
802 {
803 if (mch_dirname(cwd, MAXPATHL) != FAIL)
804 {
805#ifdef BACKSLASH_IN_FILENAME
806 slash_adjust(cwd);
807#endif
808 rettv->vval.v_string = vim_strsave(cwd);
809 }
810 vim_free(cwd);
811 }
812
813 if (curwin->w_localdir != NULL)
814 scope = CDSCOPE_WINDOW;
815 else if (curtab->tp_localdir != NULL)
816 scope = CDSCOPE_TABPAGE;
817
818 if (!changedir_func(argvars[0].vval.v_string, TRUE, scope))
819 // Directory change failed
820 VIM_CLEAR(rettv->vval.v_string);
821}
822
823/*
824 * "delete()" function
825 */
826 void
827f_delete(typval_T *argvars, typval_T *rettv)
828{
829 char_u nbuf[NUMBUFLEN];
830 char_u *name;
831 char_u *flags;
832
833 rettv->vval.v_number = -1;
834 if (check_restricted() || check_secure())
835 return;
836
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200837 if (in_vim9script()
838 && (check_for_string_arg(argvars, 0) == FAIL
839 || check_for_opt_string_arg(argvars, 1) == FAIL))
840 return;
841
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200842 name = tv_get_string(&argvars[0]);
843 if (name == NULL || *name == NUL)
844 {
845 emsg(_(e_invarg));
846 return;
847 }
848
849 if (argvars[1].v_type != VAR_UNKNOWN)
850 flags = tv_get_string_buf(&argvars[1], nbuf);
851 else
852 flags = (char_u *)"";
853
854 if (*flags == NUL)
Bram Moolenaar26262f82019-09-04 20:59:15 +0200855 // delete a file
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200856 rettv->vval.v_number = mch_remove(name) == 0 ? 0 : -1;
857 else if (STRCMP(flags, "d") == 0)
Bram Moolenaar26262f82019-09-04 20:59:15 +0200858 // delete an empty directory
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200859 rettv->vval.v_number = mch_rmdir(name) == 0 ? 0 : -1;
860 else if (STRCMP(flags, "rf") == 0)
Bram Moolenaar26262f82019-09-04 20:59:15 +0200861 // delete a directory recursively
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200862 rettv->vval.v_number = delete_recursive(name);
863 else
Bram Moolenaar108010a2021-06-27 22:03:33 +0200864 semsg(_(e_invalid_expression_str), flags);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200865}
866
867/*
868 * "executable()" function
869 */
870 void
871f_executable(typval_T *argvars, typval_T *rettv)
872{
Bram Moolenaar32105ae2021-03-27 18:59:25 +0100873 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100874 return;
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200875
Bram Moolenaar26262f82019-09-04 20:59:15 +0200876 // Check in $PATH and also check directly if there is a directory name.
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100877 rettv->vval.v_number = mch_can_exe(tv_get_string(&argvars[0]), NULL, TRUE);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200878}
879
880/*
881 * "exepath()" function
882 */
883 void
884f_exepath(typval_T *argvars, typval_T *rettv)
885{
886 char_u *p = NULL;
887
Bram Moolenaar32105ae2021-03-27 18:59:25 +0100888 if (in_vim9script() && check_for_nonempty_string_arg(argvars, 0) == FAIL)
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100889 return;
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200890 (void)mch_can_exe(tv_get_string(&argvars[0]), &p, TRUE);
891 rettv->v_type = VAR_STRING;
892 rettv->vval.v_string = p;
893}
894
895/*
896 * "filereadable()" function
897 */
898 void
899f_filereadable(typval_T *argvars, typval_T *rettv)
900{
901 int fd;
902 char_u *p;
903 int n;
904
Bram Moolenaar32105ae2021-03-27 18:59:25 +0100905 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100906 return;
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200907
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200908#ifndef O_NONBLOCK
909# define O_NONBLOCK 0
910#endif
911 p = tv_get_string(&argvars[0]);
912 if (*p && !mch_isdir(p) && (fd = mch_open((char *)p,
913 O_RDONLY | O_NONBLOCK, 0)) >= 0)
914 {
915 n = TRUE;
916 close(fd);
917 }
918 else
919 n = FALSE;
920
921 rettv->vval.v_number = n;
922}
923
924/*
925 * Return 0 for not writable, 1 for writable file, 2 for a dir which we have
926 * rights to write into.
927 */
928 void
929f_filewritable(typval_T *argvars, typval_T *rettv)
930{
Bram Moolenaar32105ae2021-03-27 18:59:25 +0100931 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100932 return;
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200933 rettv->vval.v_number = filewritable(tv_get_string(&argvars[0]));
934}
935
Bram Moolenaar840d16f2019-09-10 21:27:18 +0200936 static void
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200937findfilendir(
938 typval_T *argvars UNUSED,
939 typval_T *rettv,
940 int find_what UNUSED)
941{
942#ifdef FEAT_SEARCHPATH
943 char_u *fname;
944 char_u *fresult = NULL;
945 char_u *path = *curbuf->b_p_path == NUL ? p_path : curbuf->b_p_path;
946 char_u *p;
947 char_u pathbuf[NUMBUFLEN];
948 int count = 1;
949 int first = TRUE;
950 int error = FALSE;
951#endif
952
953 rettv->vval.v_string = NULL;
954 rettv->v_type = VAR_STRING;
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200955 if (in_vim9script()
956 && (check_for_nonempty_string_arg(argvars, 0) == FAIL
957 || check_for_opt_string_arg(argvars, 1) == FAIL
958 || (argvars[1].v_type != VAR_UNKNOWN
959 && check_for_opt_number_arg(argvars, 2) == FAIL)))
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100960 return;
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200961
962#ifdef FEAT_SEARCHPATH
963 fname = tv_get_string(&argvars[0]);
964
965 if (argvars[1].v_type != VAR_UNKNOWN)
966 {
967 p = tv_get_string_buf_chk(&argvars[1], pathbuf);
968 if (p == NULL)
969 error = TRUE;
970 else
971 {
972 if (*p != NUL)
973 path = p;
974
975 if (argvars[2].v_type != VAR_UNKNOWN)
976 count = (int)tv_get_number_chk(&argvars[2], &error);
977 }
978 }
979
980 if (count < 0 && rettv_list_alloc(rettv) == FAIL)
981 error = TRUE;
982
983 if (*fname != NUL && !error)
984 {
985 do
986 {
987 if (rettv->v_type == VAR_STRING || rettv->v_type == VAR_LIST)
988 vim_free(fresult);
989 fresult = find_file_in_path_option(first ? fname : NULL,
990 first ? (int)STRLEN(fname) : 0,
991 0, first, path,
992 find_what,
993 curbuf->b_ffname,
994 find_what == FINDFILE_DIR
995 ? (char_u *)"" : curbuf->b_p_sua);
996 first = FALSE;
997
998 if (fresult != NULL && rettv->v_type == VAR_LIST)
999 list_append_string(rettv->vval.v_list, fresult, -1);
1000
1001 } while ((rettv->v_type == VAR_LIST || --count > 0) && fresult != NULL);
1002 }
1003
1004 if (rettv->v_type == VAR_STRING)
1005 rettv->vval.v_string = fresult;
1006#endif
1007}
1008
1009/*
1010 * "finddir({fname}[, {path}[, {count}]])" function
1011 */
1012 void
1013f_finddir(typval_T *argvars, typval_T *rettv)
1014{
1015 findfilendir(argvars, rettv, FINDFILE_DIR);
1016}
1017
1018/*
1019 * "findfile({fname}[, {path}[, {count}]])" function
1020 */
1021 void
1022f_findfile(typval_T *argvars, typval_T *rettv)
1023{
1024 findfilendir(argvars, rettv, FINDFILE_FILE);
1025}
1026
1027/*
1028 * "fnamemodify({fname}, {mods})" function
1029 */
1030 void
1031f_fnamemodify(typval_T *argvars, typval_T *rettv)
1032{
1033 char_u *fname;
1034 char_u *mods;
1035 int usedlen = 0;
Bram Moolenaarc5308522020-12-13 12:25:35 +01001036 int len = 0;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001037 char_u *fbuf = NULL;
1038 char_u buf[NUMBUFLEN];
1039
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001040 if (in_vim9script()
1041 && (check_for_string_arg(argvars, 0) == FAIL
1042 || check_for_string_arg(argvars, 1) == FAIL))
Bram Moolenaar7bb4e742020-12-09 12:41:50 +01001043 return;
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001044
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001045 fname = tv_get_string_chk(&argvars[0]);
1046 mods = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaarc5308522020-12-13 12:25:35 +01001047 if (mods == NULL || fname == NULL)
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001048 fname = NULL;
Bram Moolenaarc5308522020-12-13 12:25:35 +01001049 else
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001050 {
1051 len = (int)STRLEN(fname);
Bram Moolenaarc5308522020-12-13 12:25:35 +01001052 if (mods != NULL && *mods != NUL)
1053 (void)modify_fname(mods, FALSE, &usedlen, &fname, &fbuf, &len);
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001054 }
1055
1056 rettv->v_type = VAR_STRING;
1057 if (fname == NULL)
1058 rettv->vval.v_string = NULL;
1059 else
1060 rettv->vval.v_string = vim_strnsave(fname, len);
1061 vim_free(fbuf);
1062}
1063
1064/*
1065 * "getcwd()" function
1066 *
1067 * Return the current working directory of a window in a tab page.
1068 * First optional argument 'winnr' is the window number or -1 and the second
1069 * optional argument 'tabnr' is the tab page number.
1070 *
1071 * If no arguments are supplied, then return the directory of the current
1072 * window.
1073 * If only 'winnr' is specified and is not -1 or 0 then return the directory of
1074 * the specified window.
1075 * If 'winnr' is 0 then return the directory of the current window.
1076 * If both 'winnr and 'tabnr' are specified and 'winnr' is -1 then return the
1077 * directory of the specified tab page. Otherwise return the directory of the
1078 * specified window in the specified tab page.
1079 * If the window or the tab page doesn't exist then return NULL.
1080 */
1081 void
1082f_getcwd(typval_T *argvars, typval_T *rettv)
1083{
1084 win_T *wp = NULL;
1085 tabpage_T *tp = NULL;
1086 char_u *cwd;
1087 int global = FALSE;
1088
1089 rettv->v_type = VAR_STRING;
1090 rettv->vval.v_string = NULL;
1091
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001092 if (in_vim9script()
1093 && (check_for_opt_number_arg(argvars, 0) == FAIL
1094 || (argvars[0].v_type != VAR_UNKNOWN
1095 && check_for_opt_number_arg(argvars, 1) == FAIL)))
1096 return;
1097
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001098 if (argvars[0].v_type == VAR_NUMBER
1099 && argvars[0].vval.v_number == -1
1100 && argvars[1].v_type == VAR_UNKNOWN)
1101 global = TRUE;
1102 else
1103 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
1104
1105 if (wp != NULL && wp->w_localdir != NULL)
1106 rettv->vval.v_string = vim_strsave(wp->w_localdir);
1107 else if (tp != NULL && tp->tp_localdir != NULL)
1108 rettv->vval.v_string = vim_strsave(tp->tp_localdir);
1109 else if (wp != NULL || tp != NULL || global)
1110 {
1111 if (globaldir != NULL)
1112 rettv->vval.v_string = vim_strsave(globaldir);
1113 else
1114 {
1115 cwd = alloc(MAXPATHL);
1116 if (cwd != NULL)
1117 {
1118 if (mch_dirname(cwd, MAXPATHL) != FAIL)
1119 rettv->vval.v_string = vim_strsave(cwd);
1120 vim_free(cwd);
1121 }
1122 }
1123 }
1124#ifdef BACKSLASH_IN_FILENAME
1125 if (rettv->vval.v_string != NULL)
1126 slash_adjust(rettv->vval.v_string);
1127#endif
1128}
1129
1130/*
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001131 * Convert "st" to file permission string.
1132 */
1133 char_u *
1134getfpermst(stat_T *st, char_u *perm)
1135{
1136 char_u flags[] = "rwx";
1137 int i;
1138
1139 for (i = 0; i < 9; i++)
1140 {
1141 if (st->st_mode & (1 << (8 - i)))
1142 perm[i] = flags[i % 3];
1143 else
1144 perm[i] = '-';
1145 }
1146 return perm;
1147}
1148
1149/*
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001150 * "getfperm({fname})" function
1151 */
1152 void
1153f_getfperm(typval_T *argvars, typval_T *rettv)
1154{
1155 char_u *fname;
1156 stat_T st;
1157 char_u *perm = NULL;
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001158 char_u permbuf[] = "---------";
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001159
Bram Moolenaar32105ae2021-03-27 18:59:25 +01001160 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
Bram Moolenaar7bb4e742020-12-09 12:41:50 +01001161 return;
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001162
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001163 fname = tv_get_string(&argvars[0]);
1164
1165 rettv->v_type = VAR_STRING;
1166 if (mch_stat((char *)fname, &st) >= 0)
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001167 perm = vim_strsave(getfpermst(&st, permbuf));
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001168 rettv->vval.v_string = perm;
1169}
1170
1171/*
1172 * "getfsize({fname})" function
1173 */
1174 void
1175f_getfsize(typval_T *argvars, typval_T *rettv)
1176{
1177 char_u *fname;
1178 stat_T st;
1179
Bram Moolenaar32105ae2021-03-27 18:59:25 +01001180 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
Bram Moolenaar7bb4e742020-12-09 12:41:50 +01001181 return;
1182
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001183 fname = tv_get_string(&argvars[0]);
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001184 if (mch_stat((char *)fname, &st) >= 0)
1185 {
1186 if (mch_isdir(fname))
1187 rettv->vval.v_number = 0;
1188 else
1189 {
1190 rettv->vval.v_number = (varnumber_T)st.st_size;
1191
Bram Moolenaar26262f82019-09-04 20:59:15 +02001192 // non-perfect check for overflow
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001193 if ((off_T)rettv->vval.v_number != (off_T)st.st_size)
1194 rettv->vval.v_number = -2;
1195 }
1196 }
1197 else
1198 rettv->vval.v_number = -1;
1199}
1200
1201/*
1202 * "getftime({fname})" function
1203 */
1204 void
1205f_getftime(typval_T *argvars, typval_T *rettv)
1206{
1207 char_u *fname;
1208 stat_T st;
1209
Bram Moolenaar32105ae2021-03-27 18:59:25 +01001210 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
Bram Moolenaar7bb4e742020-12-09 12:41:50 +01001211 return;
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001212
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001213 fname = tv_get_string(&argvars[0]);
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001214 if (mch_stat((char *)fname, &st) >= 0)
1215 rettv->vval.v_number = (varnumber_T)st.st_mtime;
1216 else
1217 rettv->vval.v_number = -1;
1218}
1219
1220/*
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001221 * Convert "st" to file type string.
1222 */
1223 char_u *
1224getftypest(stat_T *st)
1225{
1226 char *t;
1227
1228 if (S_ISREG(st->st_mode))
1229 t = "file";
1230 else if (S_ISDIR(st->st_mode))
1231 t = "dir";
1232 else if (S_ISLNK(st->st_mode))
1233 t = "link";
1234 else if (S_ISBLK(st->st_mode))
1235 t = "bdev";
1236 else if (S_ISCHR(st->st_mode))
1237 t = "cdev";
1238 else if (S_ISFIFO(st->st_mode))
1239 t = "fifo";
1240 else if (S_ISSOCK(st->st_mode))
1241 t = "socket";
1242 else
1243 t = "other";
1244 return (char_u*)t;
1245}
1246
1247/*
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001248 * "getftype({fname})" function
1249 */
1250 void
1251f_getftype(typval_T *argvars, typval_T *rettv)
1252{
1253 char_u *fname;
1254 stat_T st;
1255 char_u *type = NULL;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001256
Bram Moolenaar32105ae2021-03-27 18:59:25 +01001257 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
Bram Moolenaar7bb4e742020-12-09 12:41:50 +01001258 return;
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001259
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001260 fname = tv_get_string(&argvars[0]);
1261
1262 rettv->v_type = VAR_STRING;
1263 if (mch_lstat((char *)fname, &st) >= 0)
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001264 type = vim_strsave(getftypest(&st));
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001265 rettv->vval.v_string = type;
1266}
1267
1268/*
1269 * "glob()" function
1270 */
1271 void
1272f_glob(typval_T *argvars, typval_T *rettv)
1273{
1274 int options = WILD_SILENT|WILD_USE_NL;
1275 expand_T xpc;
1276 int error = FALSE;
1277
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001278 if (in_vim9script()
1279 && (check_for_string_arg(argvars, 0) == FAIL
1280 || check_for_opt_bool_arg(argvars, 1) == FAIL
1281 || (argvars[1].v_type != VAR_UNKNOWN
1282 && (check_for_opt_bool_arg(argvars, 2) == FAIL
1283 || (argvars[2].v_type != VAR_UNKNOWN
1284 && check_for_opt_bool_arg(argvars, 3) == FAIL)))))
1285 return;
1286
Bram Moolenaar26262f82019-09-04 20:59:15 +02001287 // When the optional second argument is non-zero, don't remove matches
1288 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001289 rettv->v_type = VAR_STRING;
1290 if (argvars[1].v_type != VAR_UNKNOWN)
1291 {
Bram Moolenaar5892ea12020-09-02 21:53:11 +02001292 if (tv_get_bool_chk(&argvars[1], &error))
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001293 options |= WILD_KEEP_ALL;
1294 if (argvars[2].v_type != VAR_UNKNOWN)
1295 {
Bram Moolenaar5892ea12020-09-02 21:53:11 +02001296 if (tv_get_bool_chk(&argvars[2], &error))
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001297 rettv_list_set(rettv, NULL);
1298 if (argvars[3].v_type != VAR_UNKNOWN
Bram Moolenaar5892ea12020-09-02 21:53:11 +02001299 && tv_get_bool_chk(&argvars[3], &error))
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001300 options |= WILD_ALLLINKS;
1301 }
1302 }
1303 if (!error)
1304 {
1305 ExpandInit(&xpc);
1306 xpc.xp_context = EXPAND_FILES;
1307 if (p_wic)
1308 options += WILD_ICASE;
1309 if (rettv->v_type == VAR_STRING)
1310 rettv->vval.v_string = ExpandOne(&xpc, tv_get_string(&argvars[0]),
1311 NULL, options, WILD_ALL);
1312 else if (rettv_list_alloc(rettv) != FAIL)
1313 {
1314 int i;
1315
1316 ExpandOne(&xpc, tv_get_string(&argvars[0]),
1317 NULL, options, WILD_ALL_KEEP);
1318 for (i = 0; i < xpc.xp_numfiles; i++)
1319 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
1320
1321 ExpandCleanup(&xpc);
1322 }
1323 }
1324 else
1325 rettv->vval.v_string = NULL;
1326}
1327
1328/*
1329 * "glob2regpat()" function
1330 */
1331 void
1332f_glob2regpat(typval_T *argvars, typval_T *rettv)
1333{
Bram Moolenaar3cfa5b12021-06-06 14:14:39 +02001334 char_u buf[NUMBUFLEN];
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001335 char_u *pat;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001336
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001337 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
1338 return;
1339
1340 pat = tv_get_string_buf_chk_strict(&argvars[0], buf, in_vim9script());
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001341 rettv->v_type = VAR_STRING;
1342 rettv->vval.v_string = (pat == NULL)
1343 ? NULL : file_pat_to_reg_pat(pat, NULL, NULL, FALSE);
1344}
1345
1346/*
1347 * "globpath()" function
1348 */
1349 void
1350f_globpath(typval_T *argvars, typval_T *rettv)
1351{
1352 int flags = WILD_IGNORE_COMPLETESLASH;
1353 char_u buf1[NUMBUFLEN];
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001354 char_u *file;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001355 int error = FALSE;
1356 garray_T ga;
1357 int i;
1358
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001359 if (in_vim9script()
1360 && (check_for_string_arg(argvars, 0) == FAIL
1361 || check_for_string_arg(argvars, 1) == FAIL
1362 || check_for_opt_bool_arg(argvars, 2) == FAIL
1363 || (argvars[2].v_type != VAR_UNKNOWN
1364 && (check_for_opt_bool_arg(argvars, 3) == FAIL
1365 || (argvars[3].v_type != VAR_UNKNOWN
1366 && check_for_opt_bool_arg(argvars, 4) == FAIL)))))
1367 return;
1368
1369 file = tv_get_string_buf_chk(&argvars[1], buf1);
1370
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001371 // When the optional second argument is non-zero, don't remove matches
1372 // for 'wildignore' and don't put matches for 'suffixes' at the end.
1373 rettv->v_type = VAR_STRING;
1374 if (argvars[2].v_type != VAR_UNKNOWN)
1375 {
Bram Moolenaarf966ce52020-09-02 21:57:07 +02001376 if (tv_get_bool_chk(&argvars[2], &error))
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001377 flags |= WILD_KEEP_ALL;
1378 if (argvars[3].v_type != VAR_UNKNOWN)
1379 {
Bram Moolenaarf966ce52020-09-02 21:57:07 +02001380 if (tv_get_bool_chk(&argvars[3], &error))
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001381 rettv_list_set(rettv, NULL);
1382 if (argvars[4].v_type != VAR_UNKNOWN
Bram Moolenaarf966ce52020-09-02 21:57:07 +02001383 && tv_get_bool_chk(&argvars[4], &error))
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001384 flags |= WILD_ALLLINKS;
1385 }
1386 }
1387 if (file != NULL && !error)
1388 {
1389 ga_init2(&ga, (int)sizeof(char_u *), 10);
1390 globpath(tv_get_string(&argvars[0]), file, &ga, flags);
1391 if (rettv->v_type == VAR_STRING)
1392 rettv->vval.v_string = ga_concat_strings(&ga, "\n");
1393 else if (rettv_list_alloc(rettv) != FAIL)
1394 for (i = 0; i < ga.ga_len; ++i)
1395 list_append_string(rettv->vval.v_list,
1396 ((char_u **)(ga.ga_data))[i], -1);
1397 ga_clear_strings(&ga);
1398 }
1399 else
1400 rettv->vval.v_string = NULL;
1401}
1402
1403/*
1404 * "isdirectory()" function
1405 */
1406 void
1407f_isdirectory(typval_T *argvars, typval_T *rettv)
1408{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001409 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
1410 return;
1411
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001412 rettv->vval.v_number = mch_isdir(tv_get_string(&argvars[0]));
1413}
1414
1415/*
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001416 * Create the directory in which "dir" is located, and higher levels when
1417 * needed.
1418 * Return OK or FAIL.
1419 */
1420 static int
1421mkdir_recurse(char_u *dir, int prot)
1422{
1423 char_u *p;
1424 char_u *updir;
1425 int r = FAIL;
1426
Bram Moolenaar26262f82019-09-04 20:59:15 +02001427 // Get end of directory name in "dir".
1428 // We're done when it's "/" or "c:/".
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001429 p = gettail_sep(dir);
1430 if (p <= get_past_head(dir))
1431 return OK;
1432
Bram Moolenaar26262f82019-09-04 20:59:15 +02001433 // If the directory exists we're done. Otherwise: create it.
Bram Moolenaar71ccd032020-06-12 22:59:11 +02001434 updir = vim_strnsave(dir, p - dir);
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001435 if (updir == NULL)
1436 return FAIL;
1437 if (mch_isdir(updir))
1438 r = OK;
1439 else if (mkdir_recurse(updir, prot) == OK)
1440 r = vim_mkdir_emsg(updir, prot);
1441 vim_free(updir);
1442 return r;
1443}
1444
1445/*
1446 * "mkdir()" function
1447 */
1448 void
1449f_mkdir(typval_T *argvars, typval_T *rettv)
1450{
1451 char_u *dir;
1452 char_u buf[NUMBUFLEN];
1453 int prot = 0755;
1454
1455 rettv->vval.v_number = FAIL;
1456 if (check_restricted() || check_secure())
1457 return;
1458
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001459 if (in_vim9script()
1460 && (check_for_nonempty_string_arg(argvars, 0) == FAIL
1461 || check_for_opt_string_arg(argvars, 1) == FAIL
1462 || (argvars[1].v_type != VAR_UNKNOWN
1463 && check_for_opt_number_arg(argvars, 2) == FAIL)))
1464 return;
1465
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001466 dir = tv_get_string_buf(&argvars[0], buf);
1467 if (*dir == NUL)
1468 return;
1469
1470 if (*gettail(dir) == NUL)
Bram Moolenaar26262f82019-09-04 20:59:15 +02001471 // remove trailing slashes
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001472 *gettail_sep(dir) = NUL;
1473
1474 if (argvars[1].v_type != VAR_UNKNOWN)
1475 {
1476 if (argvars[2].v_type != VAR_UNKNOWN)
1477 {
1478 prot = (int)tv_get_number_chk(&argvars[2], NULL);
1479 if (prot == -1)
1480 return;
1481 }
1482 if (STRCMP(tv_get_string(&argvars[1]), "p") == 0)
1483 {
1484 if (mch_isdir(dir))
1485 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02001486 // With the "p" flag it's OK if the dir already exists.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001487 rettv->vval.v_number = OK;
1488 return;
1489 }
1490 mkdir_recurse(dir, prot);
1491 }
1492 }
1493 rettv->vval.v_number = vim_mkdir_emsg(dir, prot);
1494}
1495
1496/*
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02001497 * "pathshorten()" function
1498 */
1499 void
1500f_pathshorten(typval_T *argvars, typval_T *rettv)
1501{
1502 char_u *p;
Bram Moolenaar6a33ef02020-09-25 22:42:48 +02001503 int trim_len = 1;
1504
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001505 if (in_vim9script()
1506 && (check_for_string_arg(argvars, 0) == FAIL
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001507 || check_for_opt_number_arg(argvars, 1) == FAIL))
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001508 return;
1509
Bram Moolenaar6a33ef02020-09-25 22:42:48 +02001510 if (argvars[1].v_type != VAR_UNKNOWN)
1511 {
1512 trim_len = (int)tv_get_number(&argvars[1]);
1513 if (trim_len < 1)
1514 trim_len = 1;
1515 }
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02001516
1517 rettv->v_type = VAR_STRING;
1518 p = tv_get_string_chk(&argvars[0]);
Bram Moolenaar6a33ef02020-09-25 22:42:48 +02001519
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02001520 if (p == NULL)
1521 rettv->vval.v_string = NULL;
1522 else
1523 {
1524 p = vim_strsave(p);
1525 rettv->vval.v_string = p;
1526 if (p != NULL)
Bram Moolenaar6a33ef02020-09-25 22:42:48 +02001527 shorten_dir_len(p, trim_len);
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02001528 }
1529}
1530
1531/*
Bram Moolenaarf8abbf32020-08-19 16:00:06 +02001532 * Common code for readdir_checkitem() and readdirex_checkitem().
1533 * Either "name" or "dict" is NULL.
Bram Moolenaar80147dd2020-02-04 22:32:59 +01001534 */
1535 static int
Bram Moolenaarf8abbf32020-08-19 16:00:06 +02001536checkitem_common(void *context, char_u *name, dict_T *dict)
Bram Moolenaar80147dd2020-02-04 22:32:59 +01001537{
1538 typval_T *expr = (typval_T *)context;
1539 typval_T save_val;
1540 typval_T rettv;
1541 typval_T argv[2];
1542 int retval = 0;
1543 int error = FALSE;
Bram Moolenaar80147dd2020-02-04 22:32:59 +01001544
1545 prepare_vimvar(VV_VAL, &save_val);
Bram Moolenaarf8abbf32020-08-19 16:00:06 +02001546 if (name != NULL)
1547 {
1548 set_vim_var_string(VV_VAL, name, -1);
1549 argv[0].v_type = VAR_STRING;
1550 argv[0].vval.v_string = name;
1551 }
1552 else
1553 {
1554 set_vim_var_dict(VV_VAL, dict);
1555 argv[0].v_type = VAR_DICT;
1556 argv[0].vval.v_dict = dict;
1557 }
Bram Moolenaar80147dd2020-02-04 22:32:59 +01001558
1559 if (eval_expr_typval(expr, argv, 1, &rettv) == FAIL)
1560 goto theend;
1561
Bram Moolenaarf8abbf32020-08-19 16:00:06 +02001562 // We want to use -1, but also true/false should be allowed.
1563 if (rettv.v_type == VAR_SPECIAL || rettv.v_type == VAR_BOOL)
1564 {
1565 rettv.v_type = VAR_NUMBER;
1566 rettv.vval.v_number = rettv.vval.v_number == VVAL_TRUE;
1567 }
Bram Moolenaar80147dd2020-02-04 22:32:59 +01001568 retval = tv_get_number_chk(&rettv, &error);
1569 if (error)
1570 retval = -1;
1571 clear_tv(&rettv);
1572
1573theend:
Bram Moolenaarf8abbf32020-08-19 16:00:06 +02001574 if (name != NULL)
1575 set_vim_var_string(VV_VAL, NULL, 0);
1576 else
1577 set_vim_var_dict(VV_VAL, NULL);
Bram Moolenaar80147dd2020-02-04 22:32:59 +01001578 restore_vimvar(VV_VAL, &save_val);
1579 return retval;
1580}
1581
Bram Moolenaarf8abbf32020-08-19 16:00:06 +02001582/*
1583 * Evaluate "expr" (= "context") for readdir().
1584 */
1585 static int
1586readdir_checkitem(void *context, void *item)
1587{
1588 char_u *name = (char_u *)item;
1589
1590 return checkitem_common(context, name, NULL);
1591}
1592
Bram Moolenaar84cf6bd2020-06-16 20:03:43 +02001593 static int
1594readdirex_dict_arg(typval_T *tv, int *cmp)
1595{
1596 char_u *compare;
1597
1598 if (tv->v_type != VAR_DICT)
1599 {
1600 emsg(_(e_dictreq));
1601 return FAIL;
1602 }
1603
1604 if (dict_find(tv->vval.v_dict, (char_u *)"sort", -1) != NULL)
1605 compare = dict_get_string(tv->vval.v_dict, (char_u *)"sort", FALSE);
1606 else
1607 {
1608 semsg(_(e_no_dict_key), "sort");
1609 return FAIL;
1610 }
1611
1612 if (STRCMP(compare, (char_u *) "none") == 0)
1613 *cmp = READDIR_SORT_NONE;
1614 else if (STRCMP(compare, (char_u *) "case") == 0)
1615 *cmp = READDIR_SORT_BYTE;
1616 else if (STRCMP(compare, (char_u *) "icase") == 0)
1617 *cmp = READDIR_SORT_IC;
1618 else if (STRCMP(compare, (char_u *) "collate") == 0)
1619 *cmp = READDIR_SORT_COLLATE;
1620 return OK;
1621}
1622
Bram Moolenaar80147dd2020-02-04 22:32:59 +01001623/*
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001624 * "readdir()" function
1625 */
1626 void
1627f_readdir(typval_T *argvars, typval_T *rettv)
1628{
1629 typval_T *expr;
1630 int ret;
1631 char_u *path;
1632 char_u *p;
1633 garray_T ga;
1634 int i;
Bram Moolenaar84cf6bd2020-06-16 20:03:43 +02001635 int sort = READDIR_SORT_BYTE;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001636
1637 if (rettv_list_alloc(rettv) == FAIL)
1638 return;
Yegappan Lakshmanan5bca9062021-07-24 21:33:26 +02001639
1640 if (in_vim9script()
1641 && (check_for_string_arg(argvars, 0) == FAIL
1642 || (argvars[1].v_type != VAR_UNKNOWN
1643 && check_for_opt_dict_arg(argvars, 2) == FAIL)))
1644 return;
1645
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001646 path = tv_get_string(&argvars[0]);
1647 expr = &argvars[1];
1648
Bram Moolenaar84cf6bd2020-06-16 20:03:43 +02001649 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN &&
1650 readdirex_dict_arg(&argvars[2], &sort) == FAIL)
1651 return;
1652
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001653 ret = readdir_core(&ga, path, FALSE, (void *)expr,
Bram Moolenaar84cf6bd2020-06-16 20:03:43 +02001654 (expr->v_type == VAR_UNKNOWN) ? NULL : readdir_checkitem, sort);
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001655 if (ret == OK)
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001656 {
1657 for (i = 0; i < ga.ga_len; i++)
1658 {
1659 p = ((char_u **)ga.ga_data)[i];
1660 list_append_string(rettv->vval.v_list, p, -1);
1661 }
1662 }
1663 ga_clear_strings(&ga);
1664}
1665
1666/*
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001667 * Evaluate "expr" (= "context") for readdirex().
1668 */
1669 static int
1670readdirex_checkitem(void *context, void *item)
1671{
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001672 dict_T *dict = (dict_T*)item;
1673
Bram Moolenaarf8abbf32020-08-19 16:00:06 +02001674 return checkitem_common(context, NULL, dict);
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001675}
1676
1677/*
1678 * "readdirex()" function
1679 */
1680 void
1681f_readdirex(typval_T *argvars, typval_T *rettv)
1682{
1683 typval_T *expr;
1684 int ret;
1685 char_u *path;
1686 garray_T ga;
1687 int i;
Bram Moolenaar84cf6bd2020-06-16 20:03:43 +02001688 int sort = READDIR_SORT_BYTE;
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001689
1690 if (rettv_list_alloc(rettv) == FAIL)
1691 return;
Yegappan Lakshmanan5bca9062021-07-24 21:33:26 +02001692
1693 if (in_vim9script()
1694 && (check_for_string_arg(argvars, 0) == FAIL
1695 || (argvars[1].v_type != VAR_UNKNOWN
1696 && check_for_opt_dict_arg(argvars, 2) == FAIL)))
1697 return;
1698
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001699 path = tv_get_string(&argvars[0]);
1700 expr = &argvars[1];
1701
Bram Moolenaar84cf6bd2020-06-16 20:03:43 +02001702 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN &&
1703 readdirex_dict_arg(&argvars[2], &sort) == FAIL)
1704 return;
1705
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001706 ret = readdir_core(&ga, path, TRUE, (void *)expr,
Bram Moolenaar84cf6bd2020-06-16 20:03:43 +02001707 (expr->v_type == VAR_UNKNOWN) ? NULL : readdirex_checkitem, sort);
Bram Moolenaar6c9ba042020-06-01 16:09:41 +02001708 if (ret == OK)
1709 {
1710 for (i = 0; i < ga.ga_len; i++)
1711 {
1712 dict_T *dict = ((dict_T**)ga.ga_data)[i];
1713 list_append_dict(rettv->vval.v_list, dict);
1714 dict_unref(dict);
1715 }
1716 }
1717 ga_clear(&ga);
1718}
1719
1720/*
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001721 * "readfile()" function
1722 */
Bram Moolenaarc423ad72021-01-13 20:38:03 +01001723 static void
1724read_file_or_blob(typval_T *argvars, typval_T *rettv, int always_blob)
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001725{
1726 int binary = FALSE;
Bram Moolenaarc423ad72021-01-13 20:38:03 +01001727 int blob = always_blob;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001728 int failed = FALSE;
1729 char_u *fname;
1730 FILE *fd;
Bram Moolenaar26262f82019-09-04 20:59:15 +02001731 char_u buf[(IOSIZE/256)*256]; // rounded to avoid odd + 1
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001732 int io_size = sizeof(buf);
Bram Moolenaar26262f82019-09-04 20:59:15 +02001733 int readlen; // size of last fread()
1734 char_u *prev = NULL; // previously read bytes, if any
1735 long prevlen = 0; // length of data in prev
1736 long prevsize = 0; // size of prev buffer
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001737 long maxline = MAXLNUM;
1738 long cnt = 0;
Bram Moolenaar26262f82019-09-04 20:59:15 +02001739 char_u *p; // position in buf
1740 char_u *start; // start of current line
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001741
1742 if (argvars[1].v_type != VAR_UNKNOWN)
1743 {
1744 if (STRCMP(tv_get_string(&argvars[1]), "b") == 0)
1745 binary = TRUE;
1746 if (STRCMP(tv_get_string(&argvars[1]), "B") == 0)
1747 blob = TRUE;
1748
1749 if (argvars[2].v_type != VAR_UNKNOWN)
1750 maxline = (long)tv_get_number(&argvars[2]);
1751 }
1752
Bram Moolenaar15352dc2020-04-06 21:12:42 +02001753 if ((blob ? rettv_blob_alloc(rettv) : rettv_list_alloc(rettv)) == FAIL)
1754 return;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001755
Bram Moolenaar26262f82019-09-04 20:59:15 +02001756 // Always open the file in binary mode, library functions have a mind of
1757 // their own about CR-LF conversion.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001758 fname = tv_get_string(&argvars[0]);
Bram Moolenaar15352dc2020-04-06 21:12:42 +02001759
1760 if (mch_isdir(fname))
1761 {
Bram Moolenaar108010a2021-06-27 22:03:33 +02001762 semsg(_(e_src_is_directory), fname);
Bram Moolenaar15352dc2020-04-06 21:12:42 +02001763 return;
1764 }
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001765 if (*fname == NUL || (fd = mch_fopen((char *)fname, READBIN)) == NULL)
1766 {
1767 semsg(_(e_notopen), *fname == NUL ? (char_u *)_("<empty>") : fname);
1768 return;
1769 }
1770
1771 if (blob)
1772 {
1773 if (read_blob(fd, rettv->vval.v_blob) == FAIL)
1774 {
Bram Moolenaar15352dc2020-04-06 21:12:42 +02001775 semsg(_(e_notread), fname);
1776 // An empty blob is returned on error.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001777 blob_free(rettv->vval.v_blob);
Bram Moolenaar15352dc2020-04-06 21:12:42 +02001778 rettv->vval.v_blob = NULL;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001779 }
1780 fclose(fd);
1781 return;
1782 }
1783
1784 while (cnt < maxline || maxline < 0)
1785 {
1786 readlen = (int)fread(buf, 1, io_size, fd);
1787
Bram Moolenaar26262f82019-09-04 20:59:15 +02001788 // This for loop processes what was read, but is also entered at end
1789 // of file so that either:
1790 // - an incomplete line gets written
1791 // - a "binary" file gets an empty line at the end if it ends in a
1792 // newline.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001793 for (p = buf, start = buf;
1794 p < buf + readlen || (readlen <= 0 && (prevlen > 0 || binary));
1795 ++p)
1796 {
1797 if (*p == '\n' || readlen <= 0)
1798 {
1799 listitem_T *li;
1800 char_u *s = NULL;
1801 long_u len = p - start;
1802
Bram Moolenaar26262f82019-09-04 20:59:15 +02001803 // Finished a line. Remove CRs before NL.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001804 if (readlen > 0 && !binary)
1805 {
1806 while (len > 0 && start[len - 1] == '\r')
1807 --len;
Bram Moolenaar26262f82019-09-04 20:59:15 +02001808 // removal may cross back to the "prev" string
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001809 if (len == 0)
1810 while (prevlen > 0 && prev[prevlen - 1] == '\r')
1811 --prevlen;
1812 }
1813 if (prevlen == 0)
Bram Moolenaar71ccd032020-06-12 22:59:11 +02001814 s = vim_strnsave(start, len);
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001815 else
1816 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02001817 // Change "prev" buffer to be the right size. This way
1818 // the bytes are only copied once, and very long lines are
1819 // allocated only once.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001820 if ((s = vim_realloc(prev, prevlen + len + 1)) != NULL)
1821 {
1822 mch_memmove(s + prevlen, start, len);
1823 s[prevlen + len] = NUL;
Bram Moolenaar26262f82019-09-04 20:59:15 +02001824 prev = NULL; // the list will own the string
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001825 prevlen = prevsize = 0;
1826 }
1827 }
1828 if (s == NULL)
1829 {
1830 do_outofmem_msg((long_u) prevlen + len + 1);
1831 failed = TRUE;
1832 break;
1833 }
1834
1835 if ((li = listitem_alloc()) == NULL)
1836 {
1837 vim_free(s);
1838 failed = TRUE;
1839 break;
1840 }
1841 li->li_tv.v_type = VAR_STRING;
1842 li->li_tv.v_lock = 0;
1843 li->li_tv.vval.v_string = s;
1844 list_append(rettv->vval.v_list, li);
1845
Bram Moolenaar26262f82019-09-04 20:59:15 +02001846 start = p + 1; // step over newline
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001847 if ((++cnt >= maxline && maxline >= 0) || readlen <= 0)
1848 break;
1849 }
1850 else if (*p == NUL)
1851 *p = '\n';
Bram Moolenaar26262f82019-09-04 20:59:15 +02001852 // Check for utf8 "bom"; U+FEFF is encoded as EF BB BF. Do this
1853 // when finding the BF and check the previous two bytes.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001854 else if (*p == 0xbf && enc_utf8 && !binary)
1855 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02001856 // Find the two bytes before the 0xbf. If p is at buf, or buf
1857 // + 1, these may be in the "prev" string.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001858 char_u back1 = p >= buf + 1 ? p[-1]
1859 : prevlen >= 1 ? prev[prevlen - 1] : NUL;
1860 char_u back2 = p >= buf + 2 ? p[-2]
1861 : p == buf + 1 && prevlen >= 1 ? prev[prevlen - 1]
1862 : prevlen >= 2 ? prev[prevlen - 2] : NUL;
1863
1864 if (back2 == 0xef && back1 == 0xbb)
1865 {
1866 char_u *dest = p - 2;
1867
Bram Moolenaar26262f82019-09-04 20:59:15 +02001868 // Usually a BOM is at the beginning of a file, and so at
1869 // the beginning of a line; then we can just step over it.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001870 if (start == dest)
1871 start = p + 1;
1872 else
1873 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02001874 // have to shuffle buf to close gap
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001875 int adjust_prevlen = 0;
1876
1877 if (dest < buf)
1878 {
Bram Moolenaarc423ad72021-01-13 20:38:03 +01001879 // must be 1 or 2
1880 adjust_prevlen = (int)(buf - dest);
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001881 dest = buf;
1882 }
1883 if (readlen > p - buf + 1)
1884 mch_memmove(dest, p + 1, readlen - (p - buf) - 1);
1885 readlen -= 3 - adjust_prevlen;
1886 prevlen -= adjust_prevlen;
1887 p = dest - 1;
1888 }
1889 }
1890 }
Bram Moolenaar26262f82019-09-04 20:59:15 +02001891 } // for
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001892
1893 if (failed || (cnt >= maxline && maxline >= 0) || readlen <= 0)
1894 break;
1895 if (start < p)
1896 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02001897 // There's part of a line in buf, store it in "prev".
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001898 if (p - start + prevlen >= prevsize)
1899 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02001900 // need bigger "prev" buffer
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001901 char_u *newprev;
1902
Bram Moolenaar26262f82019-09-04 20:59:15 +02001903 // A common use case is ordinary text files and "prev" gets a
1904 // fragment of a line, so the first allocation is made
1905 // small, to avoid repeatedly 'allocing' large and
1906 // 'reallocing' small.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001907 if (prevsize == 0)
1908 prevsize = (long)(p - start);
1909 else
1910 {
1911 long grow50pc = (prevsize * 3) / 2;
1912 long growmin = (long)((p - start) * 2 + prevlen);
1913 prevsize = grow50pc > growmin ? grow50pc : growmin;
1914 }
1915 newprev = vim_realloc(prev, prevsize);
1916 if (newprev == NULL)
1917 {
1918 do_outofmem_msg((long_u)prevsize);
1919 failed = TRUE;
1920 break;
1921 }
1922 prev = newprev;
1923 }
Bram Moolenaar26262f82019-09-04 20:59:15 +02001924 // Add the line part to end of "prev".
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001925 mch_memmove(prev + prevlen, start, p - start);
1926 prevlen += (long)(p - start);
1927 }
Bram Moolenaar26262f82019-09-04 20:59:15 +02001928 } // while
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001929
Bram Moolenaar26262f82019-09-04 20:59:15 +02001930 // For a negative line count use only the lines at the end of the file,
1931 // free the rest.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001932 if (!failed && maxline < 0)
1933 while (cnt > -maxline)
1934 {
1935 listitem_remove(rettv->vval.v_list, rettv->vval.v_list->lv_first);
1936 --cnt;
1937 }
1938
1939 if (failed)
1940 {
1941 // an empty list is returned on error
1942 list_free(rettv->vval.v_list);
1943 rettv_list_alloc(rettv);
1944 }
1945
1946 vim_free(prev);
1947 fclose(fd);
1948}
1949
1950/*
Bram Moolenaarc423ad72021-01-13 20:38:03 +01001951 * "readblob()" function
1952 */
1953 void
1954f_readblob(typval_T *argvars, typval_T *rettv)
1955{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001956 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
1957 return;
1958
Bram Moolenaarc423ad72021-01-13 20:38:03 +01001959 read_file_or_blob(argvars, rettv, TRUE);
1960}
1961
1962/*
1963 * "readfile()" function
1964 */
1965 void
1966f_readfile(typval_T *argvars, typval_T *rettv)
1967{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001968 if (in_vim9script()
1969 && (check_for_nonempty_string_arg(argvars, 0) == FAIL
1970 || check_for_opt_string_arg(argvars, 1) == FAIL
1971 || (argvars[1].v_type != VAR_UNKNOWN
1972 && check_for_opt_number_arg(argvars, 2) == FAIL)))
1973 return;
1974
Bram Moolenaarc423ad72021-01-13 20:38:03 +01001975 read_file_or_blob(argvars, rettv, FALSE);
1976}
1977
1978/*
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001979 * "resolve()" function
1980 */
1981 void
1982f_resolve(typval_T *argvars, typval_T *rettv)
1983{
1984 char_u *p;
1985#ifdef HAVE_READLINK
1986 char_u *buf = NULL;
1987#endif
1988
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001989 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
1990 return;
1991
Bram Moolenaarb005cd82019-09-04 15:54:55 +02001992 p = tv_get_string(&argvars[0]);
1993#ifdef FEAT_SHORTCUT
1994 {
1995 char_u *v = NULL;
1996
1997 v = mch_resolve_path(p, TRUE);
1998 if (v != NULL)
1999 rettv->vval.v_string = v;
2000 else
2001 rettv->vval.v_string = vim_strsave(p);
2002 }
2003#else
2004# ifdef HAVE_READLINK
2005 {
2006 char_u *cpy;
2007 int len;
2008 char_u *remain = NULL;
2009 char_u *q;
2010 int is_relative_to_current = FALSE;
2011 int has_trailing_pathsep = FALSE;
2012 int limit = 100;
2013
2014 p = vim_strsave(p);
Bram Moolenaar70188f52019-12-23 18:18:52 +01002015 if (p == NULL)
2016 goto fail;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002017 if (p[0] == '.' && (vim_ispathsep(p[1])
2018 || (p[1] == '.' && (vim_ispathsep(p[2])))))
2019 is_relative_to_current = TRUE;
2020
2021 len = STRLEN(p);
Bram Moolenaar50c4e9e2020-10-05 20:38:06 +02002022 if (len > 1 && after_pathsep(p, p + len))
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002023 {
2024 has_trailing_pathsep = TRUE;
Bram Moolenaar26262f82019-09-04 20:59:15 +02002025 p[len - 1] = NUL; // the trailing slash breaks readlink()
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002026 }
2027
2028 q = getnextcomp(p);
2029 if (*q != NUL)
2030 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02002031 // Separate the first path component in "p", and keep the
2032 // remainder (beginning with the path separator).
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002033 remain = vim_strsave(q - 1);
2034 q[-1] = NUL;
2035 }
2036
2037 buf = alloc(MAXPATHL + 1);
2038 if (buf == NULL)
Bram Moolenaar70188f52019-12-23 18:18:52 +01002039 {
2040 vim_free(p);
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002041 goto fail;
Bram Moolenaar70188f52019-12-23 18:18:52 +01002042 }
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002043
2044 for (;;)
2045 {
2046 for (;;)
2047 {
2048 len = readlink((char *)p, (char *)buf, MAXPATHL);
2049 if (len <= 0)
2050 break;
2051 buf[len] = NUL;
2052
2053 if (limit-- == 0)
2054 {
2055 vim_free(p);
2056 vim_free(remain);
2057 emsg(_("E655: Too many symbolic links (cycle?)"));
2058 rettv->vval.v_string = NULL;
2059 goto fail;
2060 }
2061
Bram Moolenaar26262f82019-09-04 20:59:15 +02002062 // Ensure that the result will have a trailing path separator
2063 // if the argument has one.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002064 if (remain == NULL && has_trailing_pathsep)
2065 add_pathsep(buf);
2066
Bram Moolenaar26262f82019-09-04 20:59:15 +02002067 // Separate the first path component in the link value and
2068 // concatenate the remainders.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002069 q = getnextcomp(vim_ispathsep(*buf) ? buf + 1 : buf);
2070 if (*q != NUL)
2071 {
2072 if (remain == NULL)
2073 remain = vim_strsave(q - 1);
2074 else
2075 {
2076 cpy = concat_str(q - 1, remain);
2077 if (cpy != NULL)
2078 {
2079 vim_free(remain);
2080 remain = cpy;
2081 }
2082 }
2083 q[-1] = NUL;
2084 }
2085
2086 q = gettail(p);
2087 if (q > p && *q == NUL)
2088 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02002089 // Ignore trailing path separator.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002090 q[-1] = NUL;
2091 q = gettail(p);
2092 }
2093 if (q > p && !mch_isFullName(buf))
2094 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02002095 // symlink is relative to directory of argument
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002096 cpy = alloc(STRLEN(p) + STRLEN(buf) + 1);
2097 if (cpy != NULL)
2098 {
2099 STRCPY(cpy, p);
2100 STRCPY(gettail(cpy), buf);
2101 vim_free(p);
2102 p = cpy;
2103 }
2104 }
2105 else
2106 {
2107 vim_free(p);
2108 p = vim_strsave(buf);
2109 }
2110 }
2111
2112 if (remain == NULL)
2113 break;
2114
Bram Moolenaar26262f82019-09-04 20:59:15 +02002115 // Append the first path component of "remain" to "p".
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002116 q = getnextcomp(remain + 1);
2117 len = q - remain - (*q != NUL);
2118 cpy = vim_strnsave(p, STRLEN(p) + len);
2119 if (cpy != NULL)
2120 {
2121 STRNCAT(cpy, remain, len);
2122 vim_free(p);
2123 p = cpy;
2124 }
Bram Moolenaar26262f82019-09-04 20:59:15 +02002125 // Shorten "remain".
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002126 if (*q != NUL)
2127 STRMOVE(remain, q - 1);
2128 else
2129 VIM_CLEAR(remain);
2130 }
2131
Bram Moolenaar26262f82019-09-04 20:59:15 +02002132 // If the result is a relative path name, make it explicitly relative to
2133 // the current directory if and only if the argument had this form.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002134 if (!vim_ispathsep(*p))
2135 {
2136 if (is_relative_to_current
2137 && *p != NUL
2138 && !(p[0] == '.'
2139 && (p[1] == NUL
2140 || vim_ispathsep(p[1])
2141 || (p[1] == '.'
2142 && (p[2] == NUL
2143 || vim_ispathsep(p[2]))))))
2144 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02002145 // Prepend "./".
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002146 cpy = concat_str((char_u *)"./", p);
2147 if (cpy != NULL)
2148 {
2149 vim_free(p);
2150 p = cpy;
2151 }
2152 }
2153 else if (!is_relative_to_current)
2154 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02002155 // Strip leading "./".
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002156 q = p;
2157 while (q[0] == '.' && vim_ispathsep(q[1]))
2158 q += 2;
2159 if (q > p)
2160 STRMOVE(p, p + 2);
2161 }
2162 }
2163
Bram Moolenaar26262f82019-09-04 20:59:15 +02002164 // Ensure that the result will have no trailing path separator
2165 // if the argument had none. But keep "/" or "//".
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002166 if (!has_trailing_pathsep)
2167 {
2168 q = p + STRLEN(p);
2169 if (after_pathsep(p, q))
2170 *gettail_sep(p) = NUL;
2171 }
2172
2173 rettv->vval.v_string = p;
2174 }
2175# else
2176 rettv->vval.v_string = vim_strsave(p);
2177# endif
2178#endif
2179
2180 simplify_filename(rettv->vval.v_string);
2181
2182#ifdef HAVE_READLINK
2183fail:
2184 vim_free(buf);
2185#endif
2186 rettv->v_type = VAR_STRING;
2187}
2188
2189/*
2190 * "tempname()" function
2191 */
2192 void
2193f_tempname(typval_T *argvars UNUSED, typval_T *rettv)
2194{
2195 static int x = 'A';
2196
2197 rettv->v_type = VAR_STRING;
2198 rettv->vval.v_string = vim_tempname(x, FALSE);
2199
Bram Moolenaar26262f82019-09-04 20:59:15 +02002200 // Advance 'x' to use A-Z and 0-9, so that there are at least 34 different
2201 // names. Skip 'I' and 'O', they are used for shell redirection.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002202 do
2203 {
2204 if (x == 'Z')
2205 x = '0';
2206 else if (x == '9')
2207 x = 'A';
2208 else
2209 {
2210#ifdef EBCDIC
2211 if (x == 'I')
2212 x = 'J';
2213 else if (x == 'R')
2214 x = 'S';
2215 else
2216#endif
2217 ++x;
2218 }
2219 } while (x == 'I' || x == 'O');
2220}
2221
2222/*
2223 * "writefile()" function
2224 */
2225 void
2226f_writefile(typval_T *argvars, typval_T *rettv)
2227{
2228 int binary = FALSE;
2229 int append = FALSE;
2230#ifdef HAVE_FSYNC
2231 int do_fsync = p_fs;
2232#endif
2233 char_u *fname;
2234 FILE *fd;
2235 int ret = 0;
2236 listitem_T *li;
2237 list_T *list = NULL;
2238 blob_T *blob = NULL;
2239
2240 rettv->vval.v_number = -1;
2241 if (check_secure())
2242 return;
2243
Yegappan Lakshmanan0ad871d2021-07-23 20:37:56 +02002244 if (in_vim9script()
2245 && (check_for_list_or_blob_arg(argvars, 0) == FAIL
2246 || check_for_string_arg(argvars, 1) == FAIL
2247 || check_for_opt_string_arg(argvars, 2) == FAIL))
2248 return;
2249
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002250 if (argvars[0].v_type == VAR_LIST)
2251 {
2252 list = argvars[0].vval.v_list;
2253 if (list == NULL)
2254 return;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02002255 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002256 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002257 if (tv_get_string_chk(&li->li_tv) == NULL)
2258 return;
2259 }
2260 else if (argvars[0].v_type == VAR_BLOB)
2261 {
2262 blob = argvars[0].vval.v_blob;
2263 if (blob == NULL)
2264 return;
2265 }
2266 else
2267 {
Bram Moolenaar18a2b872020-03-19 13:08:45 +01002268 semsg(_(e_invarg2),
2269 _("writefile() first argument must be a List or a Blob"));
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002270 return;
2271 }
2272
2273 if (argvars[2].v_type != VAR_UNKNOWN)
2274 {
2275 char_u *arg2 = tv_get_string_chk(&argvars[2]);
2276
2277 if (arg2 == NULL)
2278 return;
2279 if (vim_strchr(arg2, 'b') != NULL)
2280 binary = TRUE;
2281 if (vim_strchr(arg2, 'a') != NULL)
2282 append = TRUE;
2283#ifdef HAVE_FSYNC
2284 if (vim_strchr(arg2, 's') != NULL)
2285 do_fsync = TRUE;
2286 else if (vim_strchr(arg2, 'S') != NULL)
2287 do_fsync = FALSE;
2288#endif
2289 }
2290
2291 fname = tv_get_string_chk(&argvars[1]);
2292 if (fname == NULL)
2293 return;
2294
Bram Moolenaar26262f82019-09-04 20:59:15 +02002295 // Always open the file in binary mode, library functions have a mind of
2296 // their own about CR-LF conversion.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002297 if (*fname == NUL || (fd = mch_fopen((char *)fname,
2298 append ? APPENDBIN : WRITEBIN)) == NULL)
2299 {
2300 semsg(_(e_notcreate), *fname == NUL ? (char_u *)_("<empty>") : fname);
2301 ret = -1;
2302 }
2303 else if (blob)
2304 {
2305 if (write_blob(fd, blob) == FAIL)
2306 ret = -1;
2307#ifdef HAVE_FSYNC
2308 else if (do_fsync)
2309 // Ignore the error, the user wouldn't know what to do about it.
2310 // May happen for a device.
2311 vim_ignored = vim_fsync(fileno(fd));
2312#endif
2313 fclose(fd);
2314 }
2315 else
2316 {
2317 if (write_list(fd, list, binary) == FAIL)
2318 ret = -1;
2319#ifdef HAVE_FSYNC
2320 else if (do_fsync)
Bram Moolenaar26262f82019-09-04 20:59:15 +02002321 // Ignore the error, the user wouldn't know what to do about it.
2322 // May happen for a device.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002323 vim_ignored = vim_fsync(fileno(fd));
2324#endif
2325 fclose(fd);
2326 }
2327
2328 rettv->vval.v_number = ret;
2329}
2330
2331#endif // FEAT_EVAL
2332
2333#if defined(FEAT_BROWSE) || defined(PROTO)
2334/*
2335 * Generic browse function. Calls gui_mch_browse() when possible.
2336 * Later this may pop-up a non-GUI file selector (external command?).
2337 */
2338 char_u *
2339do_browse(
Bram Moolenaar26262f82019-09-04 20:59:15 +02002340 int flags, // BROWSE_SAVE and BROWSE_DIR
2341 char_u *title, // title for the window
2342 char_u *dflt, // default file name (may include directory)
2343 char_u *ext, // extension added
2344 char_u *initdir, // initial directory, NULL for current dir or
2345 // when using path from "dflt"
2346 char_u *filter, // file name filter
2347 buf_T *buf) // buffer to read/write for
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002348{
2349 char_u *fname;
Bram Moolenaar26262f82019-09-04 20:59:15 +02002350 static char_u *last_dir = NULL; // last used directory
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002351 char_u *tofree = NULL;
Bram Moolenaare1004402020-10-24 20:49:43 +02002352 int save_cmod_flags = cmdmod.cmod_flags;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002353
Bram Moolenaar26262f82019-09-04 20:59:15 +02002354 // Must turn off browse to avoid that autocommands will get the
2355 // flag too!
Bram Moolenaare1004402020-10-24 20:49:43 +02002356 cmdmod.cmod_flags &= ~CMOD_BROWSE;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002357
2358 if (title == NULL || *title == NUL)
2359 {
2360 if (flags & BROWSE_DIR)
2361 title = (char_u *)_("Select Directory dialog");
2362 else if (flags & BROWSE_SAVE)
2363 title = (char_u *)_("Save File dialog");
2364 else
2365 title = (char_u *)_("Open File dialog");
2366 }
2367
Bram Moolenaar26262f82019-09-04 20:59:15 +02002368 // When no directory specified, use default file name, default dir, buffer
2369 // dir, last dir or current dir
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002370 if ((initdir == NULL || *initdir == NUL) && dflt != NULL && *dflt != NUL)
2371 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02002372 if (mch_isdir(dflt)) // default file name is a directory
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002373 {
2374 initdir = dflt;
2375 dflt = NULL;
2376 }
Bram Moolenaar26262f82019-09-04 20:59:15 +02002377 else if (gettail(dflt) != dflt) // default file name includes a path
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002378 {
2379 tofree = vim_strsave(dflt);
2380 if (tofree != NULL)
2381 {
2382 initdir = tofree;
2383 *gettail(initdir) = NUL;
2384 dflt = gettail(dflt);
2385 }
2386 }
2387 }
2388
2389 if (initdir == NULL || *initdir == NUL)
2390 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02002391 // When 'browsedir' is a directory, use it
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002392 if (STRCMP(p_bsdir, "last") != 0
2393 && STRCMP(p_bsdir, "buffer") != 0
2394 && STRCMP(p_bsdir, "current") != 0
2395 && mch_isdir(p_bsdir))
2396 initdir = p_bsdir;
Bram Moolenaar26262f82019-09-04 20:59:15 +02002397 // When saving or 'browsedir' is "buffer", use buffer fname
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002398 else if (((flags & BROWSE_SAVE) || *p_bsdir == 'b')
2399 && buf != NULL && buf->b_ffname != NULL)
2400 {
2401 if (dflt == NULL || *dflt == NUL)
2402 dflt = gettail(curbuf->b_ffname);
2403 tofree = vim_strsave(curbuf->b_ffname);
2404 if (tofree != NULL)
2405 {
2406 initdir = tofree;
2407 *gettail(initdir) = NUL;
2408 }
2409 }
Bram Moolenaar26262f82019-09-04 20:59:15 +02002410 // When 'browsedir' is "last", use dir from last browse
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002411 else if (*p_bsdir == 'l')
2412 initdir = last_dir;
Bram Moolenaar26262f82019-09-04 20:59:15 +02002413 // When 'browsedir is "current", use current directory. This is the
2414 // default already, leave initdir empty.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002415 }
2416
2417# ifdef FEAT_GUI
Bram Moolenaar26262f82019-09-04 20:59:15 +02002418 if (gui.in_use) // when this changes, also adjust f_has()!
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002419 {
2420 if (filter == NULL
2421# ifdef FEAT_EVAL
2422 && (filter = get_var_value((char_u *)"b:browsefilter")) == NULL
2423 && (filter = get_var_value((char_u *)"g:browsefilter")) == NULL
2424# endif
2425 )
2426 filter = BROWSE_FILTER_DEFAULT;
2427 if (flags & BROWSE_DIR)
2428 {
2429# if defined(FEAT_GUI_GTK) || defined(MSWIN)
Bram Moolenaar26262f82019-09-04 20:59:15 +02002430 // For systems that have a directory dialog.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002431 fname = gui_mch_browsedir(title, initdir);
2432# else
Bram Moolenaar26262f82019-09-04 20:59:15 +02002433 // Generic solution for selecting a directory: select a file and
2434 // remove the file name.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002435 fname = gui_mch_browse(0, title, dflt, ext, initdir, (char_u *)"");
2436# endif
2437# if !defined(FEAT_GUI_GTK)
Bram Moolenaar26262f82019-09-04 20:59:15 +02002438 // Win32 adds a dummy file name, others return an arbitrary file
2439 // name. GTK+ 2 returns only the directory,
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002440 if (fname != NULL && *fname != NUL && !mch_isdir(fname))
2441 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02002442 // Remove the file name.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002443 char_u *tail = gettail_sep(fname);
2444
2445 if (tail == fname)
Bram Moolenaar26262f82019-09-04 20:59:15 +02002446 *tail++ = '.'; // use current dir
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002447 *tail = NUL;
2448 }
2449# endif
2450 }
2451 else
2452 fname = gui_mch_browse(flags & BROWSE_SAVE,
2453 title, dflt, ext, initdir, (char_u *)_(filter));
2454
Bram Moolenaar26262f82019-09-04 20:59:15 +02002455 // We hang around in the dialog for a while, the user might do some
2456 // things to our files. The Win32 dialog allows deleting or renaming
2457 // a file, check timestamps.
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002458 need_check_timestamps = TRUE;
2459 did_check_timestamps = FALSE;
2460 }
2461 else
2462# endif
2463 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02002464 // TODO: non-GUI file selector here
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002465 emsg(_("E338: Sorry, no file browser in console mode"));
2466 fname = NULL;
2467 }
2468
Bram Moolenaar26262f82019-09-04 20:59:15 +02002469 // keep the directory for next time
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002470 if (fname != NULL)
2471 {
2472 vim_free(last_dir);
2473 last_dir = vim_strsave(fname);
2474 if (last_dir != NULL && !(flags & BROWSE_DIR))
2475 {
2476 *gettail(last_dir) = NUL;
2477 if (*last_dir == NUL)
2478 {
Bram Moolenaar26262f82019-09-04 20:59:15 +02002479 // filename only returned, must be in current dir
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002480 vim_free(last_dir);
2481 last_dir = alloc(MAXPATHL);
2482 if (last_dir != NULL)
2483 mch_dirname(last_dir, MAXPATHL);
2484 }
2485 }
2486 }
2487
2488 vim_free(tofree);
Bram Moolenaare1004402020-10-24 20:49:43 +02002489 cmdmod.cmod_flags = save_cmod_flags;
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002490
2491 return fname;
2492}
2493#endif
2494
2495#if defined(FEAT_EVAL) || defined(PROTO)
2496
2497/*
2498 * "browse(save, title, initdir, default)" function
2499 */
2500 void
2501f_browse(typval_T *argvars UNUSED, typval_T *rettv)
2502{
2503# ifdef FEAT_BROWSE
2504 int save;
2505 char_u *title;
2506 char_u *initdir;
2507 char_u *defname;
2508 char_u buf[NUMBUFLEN];
2509 char_u buf2[NUMBUFLEN];
2510 int error = FALSE;
2511
Bram Moolenaarf28f2ac2021-03-22 22:21:26 +01002512 if (in_vim9script()
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02002513 && (check_for_bool_arg(argvars, 0) == FAIL
2514 || check_for_string_arg(argvars, 1) == FAIL
Bram Moolenaar32105ae2021-03-27 18:59:25 +01002515 || check_for_string_arg(argvars, 2) == FAIL
2516 || check_for_string_arg(argvars, 3) == FAIL))
Bram Moolenaarf28f2ac2021-03-22 22:21:26 +01002517 return;
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02002518
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002519 save = (int)tv_get_number_chk(&argvars[0], &error);
2520 title = tv_get_string_chk(&argvars[1]);
2521 initdir = tv_get_string_buf_chk(&argvars[2], buf);
2522 defname = tv_get_string_buf_chk(&argvars[3], buf2);
2523
2524 if (error || title == NULL || initdir == NULL || defname == NULL)
2525 rettv->vval.v_string = NULL;
2526 else
2527 rettv->vval.v_string =
2528 do_browse(save ? BROWSE_SAVE : 0,
2529 title, defname, NULL, initdir, NULL, curbuf);
2530# else
2531 rettv->vval.v_string = NULL;
2532# endif
2533 rettv->v_type = VAR_STRING;
2534}
2535
2536/*
2537 * "browsedir(title, initdir)" function
2538 */
2539 void
2540f_browsedir(typval_T *argvars UNUSED, typval_T *rettv)
2541{
2542# ifdef FEAT_BROWSE
2543 char_u *title;
2544 char_u *initdir;
2545 char_u buf[NUMBUFLEN];
2546
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02002547 if (in_vim9script()
2548 && (check_for_string_arg(argvars, 0) == FAIL
2549 || check_for_string_arg(argvars, 1) == FAIL))
2550 return;
2551
Bram Moolenaarb005cd82019-09-04 15:54:55 +02002552 title = tv_get_string_chk(&argvars[0]);
2553 initdir = tv_get_string_buf_chk(&argvars[1], buf);
2554
2555 if (title == NULL || initdir == NULL)
2556 rettv->vval.v_string = NULL;
2557 else
2558 rettv->vval.v_string = do_browse(BROWSE_DIR,
2559 title, NULL, NULL, initdir, NULL, curbuf);
2560# else
2561 rettv->vval.v_string = NULL;
2562# endif
2563 rettv->v_type = VAR_STRING;
2564}
2565
2566#endif // FEAT_EVAL
Bram Moolenaar26262f82019-09-04 20:59:15 +02002567
2568/*
2569 * Replace home directory by "~" in each space or comma separated file name in
2570 * 'src'.
2571 * If anything fails (except when out of space) dst equals src.
2572 */
2573 void
2574home_replace(
2575 buf_T *buf, // when not NULL, check for help files
2576 char_u *src, // input file name
2577 char_u *dst, // where to put the result
2578 int dstlen, // maximum length of the result
2579 int one) // if TRUE, only replace one file name, include
2580 // spaces and commas in the file name.
2581{
2582 size_t dirlen = 0, envlen = 0;
2583 size_t len;
2584 char_u *homedir_env, *homedir_env_orig;
2585 char_u *p;
2586
2587 if (src == NULL)
2588 {
2589 *dst = NUL;
2590 return;
2591 }
2592
2593 /*
2594 * If the file is a help file, remove the path completely.
2595 */
2596 if (buf != NULL && buf->b_help)
2597 {
2598 vim_snprintf((char *)dst, dstlen, "%s", gettail(src));
2599 return;
2600 }
2601
2602 /*
2603 * We check both the value of the $HOME environment variable and the
2604 * "real" home directory.
2605 */
2606 if (homedir != NULL)
2607 dirlen = STRLEN(homedir);
2608
2609#ifdef VMS
2610 homedir_env_orig = homedir_env = mch_getenv((char_u *)"SYS$LOGIN");
2611#else
2612 homedir_env_orig = homedir_env = mch_getenv((char_u *)"HOME");
2613#endif
2614#ifdef MSWIN
2615 if (homedir_env == NULL)
2616 homedir_env_orig = homedir_env = mch_getenv((char_u *)"USERPROFILE");
2617#endif
2618 // Empty is the same as not set.
2619 if (homedir_env != NULL && *homedir_env == NUL)
2620 homedir_env = NULL;
2621
2622 if (homedir_env != NULL && *homedir_env == '~')
2623 {
2624 int usedlen = 0;
2625 int flen;
2626 char_u *fbuf = NULL;
2627
2628 flen = (int)STRLEN(homedir_env);
2629 (void)modify_fname((char_u *)":p", FALSE, &usedlen,
2630 &homedir_env, &fbuf, &flen);
2631 flen = (int)STRLEN(homedir_env);
2632 if (flen > 0 && vim_ispathsep(homedir_env[flen - 1]))
2633 // Remove the trailing / that is added to a directory.
2634 homedir_env[flen - 1] = NUL;
2635 }
2636
2637 if (homedir_env != NULL)
2638 envlen = STRLEN(homedir_env);
2639
2640 if (!one)
2641 src = skipwhite(src);
2642 while (*src && dstlen > 0)
2643 {
2644 /*
2645 * Here we are at the beginning of a file name.
2646 * First, check to see if the beginning of the file name matches
2647 * $HOME or the "real" home directory. Check that there is a '/'
2648 * after the match (so that if e.g. the file is "/home/pieter/bla",
2649 * and the home directory is "/home/piet", the file does not end up
2650 * as "~er/bla" (which would seem to indicate the file "bla" in user
2651 * er's home directory)).
2652 */
2653 p = homedir;
2654 len = dirlen;
2655 for (;;)
2656 {
2657 if ( len
2658 && fnamencmp(src, p, len) == 0
2659 && (vim_ispathsep(src[len])
2660 || (!one && (src[len] == ',' || src[len] == ' '))
2661 || src[len] == NUL))
2662 {
2663 src += len;
2664 if (--dstlen > 0)
2665 *dst++ = '~';
2666
Bram Moolenaar0e390f42020-06-10 13:12:28 +02002667 // Do not add directory separator into dst, because dst is
2668 // expected to just return the directory name without the
2669 // directory separator '/'.
Bram Moolenaar26262f82019-09-04 20:59:15 +02002670 break;
2671 }
2672 if (p == homedir_env)
2673 break;
2674 p = homedir_env;
2675 len = envlen;
2676 }
2677
2678 // if (!one) skip to separator: space or comma
2679 while (*src && (one || (*src != ',' && *src != ' ')) && --dstlen > 0)
2680 *dst++ = *src++;
2681 // skip separator
2682 while ((*src == ' ' || *src == ',') && --dstlen > 0)
2683 *dst++ = *src++;
2684 }
2685 // if (dstlen == 0) out of space, what to do???
2686
2687 *dst = NUL;
2688
2689 if (homedir_env != homedir_env_orig)
2690 vim_free(homedir_env);
2691}
2692
2693/*
2694 * Like home_replace, store the replaced string in allocated memory.
2695 * When something fails, NULL is returned.
2696 */
2697 char_u *
2698home_replace_save(
2699 buf_T *buf, // when not NULL, check for help files
2700 char_u *src) // input file name
2701{
2702 char_u *dst;
2703 unsigned len;
2704
2705 len = 3; // space for "~/" and trailing NUL
2706 if (src != NULL) // just in case
2707 len += (unsigned)STRLEN(src);
2708 dst = alloc(len);
2709 if (dst != NULL)
2710 home_replace(buf, src, dst, len, TRUE);
2711 return dst;
2712}
2713
2714/*
2715 * Compare two file names and return:
2716 * FPC_SAME if they both exist and are the same file.
2717 * FPC_SAMEX if they both don't exist and have the same file name.
2718 * FPC_DIFF if they both exist and are different files.
2719 * FPC_NOTX if they both don't exist.
2720 * FPC_DIFFX if one of them doesn't exist.
2721 * For the first name environment variables are expanded if "expandenv" is
2722 * TRUE.
2723 */
2724 int
2725fullpathcmp(
2726 char_u *s1,
2727 char_u *s2,
2728 int checkname, // when both don't exist, check file names
2729 int expandenv)
2730{
2731#ifdef UNIX
2732 char_u exp1[MAXPATHL];
2733 char_u full1[MAXPATHL];
2734 char_u full2[MAXPATHL];
2735 stat_T st1, st2;
2736 int r1, r2;
2737
2738 if (expandenv)
2739 expand_env(s1, exp1, MAXPATHL);
2740 else
2741 vim_strncpy(exp1, s1, MAXPATHL - 1);
2742 r1 = mch_stat((char *)exp1, &st1);
2743 r2 = mch_stat((char *)s2, &st2);
2744 if (r1 != 0 && r2 != 0)
2745 {
Bram Moolenaar217e1b82019-12-01 21:41:28 +01002746 // if mch_stat() doesn't work, may compare the names
Bram Moolenaar26262f82019-09-04 20:59:15 +02002747 if (checkname)
2748 {
2749 if (fnamecmp(exp1, s2) == 0)
2750 return FPC_SAMEX;
2751 r1 = vim_FullName(exp1, full1, MAXPATHL, FALSE);
2752 r2 = vim_FullName(s2, full2, MAXPATHL, FALSE);
2753 if (r1 == OK && r2 == OK && fnamecmp(full1, full2) == 0)
2754 return FPC_SAMEX;
2755 }
2756 return FPC_NOTX;
2757 }
2758 if (r1 != 0 || r2 != 0)
2759 return FPC_DIFFX;
2760 if (st1.st_dev == st2.st_dev && st1.st_ino == st2.st_ino)
2761 return FPC_SAME;
2762 return FPC_DIFF;
2763#else
2764 char_u *exp1; // expanded s1
2765 char_u *full1; // full path of s1
2766 char_u *full2; // full path of s2
2767 int retval = FPC_DIFF;
2768 int r1, r2;
2769
2770 // allocate one buffer to store three paths (alloc()/free() is slow!)
2771 if ((exp1 = alloc(MAXPATHL * 3)) != NULL)
2772 {
2773 full1 = exp1 + MAXPATHL;
2774 full2 = full1 + MAXPATHL;
2775
2776 if (expandenv)
2777 expand_env(s1, exp1, MAXPATHL);
2778 else
2779 vim_strncpy(exp1, s1, MAXPATHL - 1);
2780 r1 = vim_FullName(exp1, full1, MAXPATHL, FALSE);
2781 r2 = vim_FullName(s2, full2, MAXPATHL, FALSE);
2782
2783 // If vim_FullName() fails, the file probably doesn't exist.
2784 if (r1 != OK && r2 != OK)
2785 {
2786 if (checkname && fnamecmp(exp1, s2) == 0)
2787 retval = FPC_SAMEX;
2788 else
2789 retval = FPC_NOTX;
2790 }
2791 else if (r1 != OK || r2 != OK)
2792 retval = FPC_DIFFX;
2793 else if (fnamecmp(full1, full2))
2794 retval = FPC_DIFF;
2795 else
2796 retval = FPC_SAME;
2797 vim_free(exp1);
2798 }
2799 return retval;
2800#endif
2801}
2802
2803/*
2804 * Get the tail of a path: the file name.
2805 * When the path ends in a path separator the tail is the NUL after it.
2806 * Fail safe: never returns NULL.
2807 */
2808 char_u *
2809gettail(char_u *fname)
2810{
2811 char_u *p1, *p2;
2812
2813 if (fname == NULL)
2814 return (char_u *)"";
2815 for (p1 = p2 = get_past_head(fname); *p2; ) // find last part of path
2816 {
2817 if (vim_ispathsep_nocolon(*p2))
2818 p1 = p2 + 1;
2819 MB_PTR_ADV(p2);
2820 }
2821 return p1;
2822}
2823
2824/*
2825 * Get pointer to tail of "fname", including path separators. Putting a NUL
2826 * here leaves the directory name. Takes care of "c:/" and "//".
2827 * Always returns a valid pointer.
2828 */
2829 char_u *
2830gettail_sep(char_u *fname)
2831{
2832 char_u *p;
2833 char_u *t;
2834
2835 p = get_past_head(fname); // don't remove the '/' from "c:/file"
2836 t = gettail(fname);
2837 while (t > p && after_pathsep(fname, t))
2838 --t;
2839#ifdef VMS
2840 // path separator is part of the path
2841 ++t;
2842#endif
2843 return t;
2844}
2845
2846/*
2847 * get the next path component (just after the next path separator).
2848 */
2849 char_u *
2850getnextcomp(char_u *fname)
2851{
2852 while (*fname && !vim_ispathsep(*fname))
2853 MB_PTR_ADV(fname);
2854 if (*fname)
2855 ++fname;
2856 return fname;
2857}
2858
2859/*
2860 * Get a pointer to one character past the head of a path name.
2861 * Unix: after "/"; DOS: after "c:\"; Amiga: after "disk:/"; Mac: no head.
2862 * If there is no head, path is returned.
2863 */
2864 char_u *
2865get_past_head(char_u *path)
2866{
2867 char_u *retval;
2868
2869#if defined(MSWIN)
2870 // may skip "c:"
2871 if (isalpha(path[0]) && path[1] == ':')
2872 retval = path + 2;
2873 else
2874 retval = path;
2875#else
2876# if defined(AMIGA)
2877 // may skip "label:"
2878 retval = vim_strchr(path, ':');
2879 if (retval == NULL)
2880 retval = path;
2881# else // Unix
2882 retval = path;
2883# endif
2884#endif
2885
2886 while (vim_ispathsep(*retval))
2887 ++retval;
2888
2889 return retval;
2890}
2891
2892/*
2893 * Return TRUE if 'c' is a path separator.
2894 * Note that for MS-Windows this includes the colon.
2895 */
2896 int
2897vim_ispathsep(int c)
2898{
2899#ifdef UNIX
2900 return (c == '/'); // UNIX has ':' inside file names
2901#else
2902# ifdef BACKSLASH_IN_FILENAME
2903 return (c == ':' || c == '/' || c == '\\');
2904# else
2905# ifdef VMS
2906 // server"user passwd"::device:[full.path.name]fname.extension;version"
2907 return (c == ':' || c == '[' || c == ']' || c == '/'
2908 || c == '<' || c == '>' || c == '"' );
2909# else
2910 return (c == ':' || c == '/');
2911# endif // VMS
2912# endif
2913#endif
2914}
2915
2916/*
2917 * Like vim_ispathsep(c), but exclude the colon for MS-Windows.
2918 */
2919 int
2920vim_ispathsep_nocolon(int c)
2921{
2922 return vim_ispathsep(c)
2923#ifdef BACKSLASH_IN_FILENAME
2924 && c != ':'
2925#endif
2926 ;
2927}
2928
2929/*
Bram Moolenaar26262f82019-09-04 20:59:15 +02002930 * Return TRUE if the directory of "fname" exists, FALSE otherwise.
2931 * Also returns TRUE if there is no directory name.
2932 * "fname" must be writable!.
2933 */
2934 int
2935dir_of_file_exists(char_u *fname)
2936{
2937 char_u *p;
2938 int c;
2939 int retval;
2940
2941 p = gettail_sep(fname);
2942 if (p == fname)
2943 return TRUE;
2944 c = *p;
2945 *p = NUL;
2946 retval = mch_isdir(fname);
2947 *p = c;
2948 return retval;
2949}
2950
2951/*
2952 * Versions of fnamecmp() and fnamencmp() that handle '/' and '\' equally
2953 * and deal with 'fileignorecase'.
2954 */
2955 int
2956vim_fnamecmp(char_u *x, char_u *y)
2957{
2958#ifdef BACKSLASH_IN_FILENAME
2959 return vim_fnamencmp(x, y, MAXPATHL);
2960#else
2961 if (p_fic)
2962 return MB_STRICMP(x, y);
2963 return STRCMP(x, y);
2964#endif
2965}
2966
2967 int
2968vim_fnamencmp(char_u *x, char_u *y, size_t len)
2969{
2970#ifdef BACKSLASH_IN_FILENAME
2971 char_u *px = x;
2972 char_u *py = y;
2973 int cx = NUL;
2974 int cy = NUL;
2975
2976 while (len > 0)
2977 {
2978 cx = PTR2CHAR(px);
2979 cy = PTR2CHAR(py);
2980 if (cx == NUL || cy == NUL
2981 || ((p_fic ? MB_TOLOWER(cx) != MB_TOLOWER(cy) : cx != cy)
2982 && !(cx == '/' && cy == '\\')
2983 && !(cx == '\\' && cy == '/')))
2984 break;
Bram Moolenaar1614a142019-10-06 22:00:13 +02002985 len -= mb_ptr2len(px);
2986 px += mb_ptr2len(px);
2987 py += mb_ptr2len(py);
Bram Moolenaar26262f82019-09-04 20:59:15 +02002988 }
2989 if (len == 0)
2990 return 0;
2991 return (cx - cy);
2992#else
2993 if (p_fic)
2994 return MB_STRNICMP(x, y, len);
2995 return STRNCMP(x, y, len);
2996#endif
2997}
2998
2999/*
3000 * Concatenate file names fname1 and fname2 into allocated memory.
3001 * Only add a '/' or '\\' when 'sep' is TRUE and it is necessary.
3002 */
3003 char_u *
3004concat_fnames(char_u *fname1, char_u *fname2, int sep)
3005{
3006 char_u *dest;
3007
3008 dest = alloc(STRLEN(fname1) + STRLEN(fname2) + 3);
3009 if (dest != NULL)
3010 {
3011 STRCPY(dest, fname1);
3012 if (sep)
3013 add_pathsep(dest);
3014 STRCAT(dest, fname2);
3015 }
3016 return dest;
3017}
3018
3019/*
3020 * Add a path separator to a file name, unless it already ends in a path
3021 * separator.
3022 */
3023 void
3024add_pathsep(char_u *p)
3025{
3026 if (*p != NUL && !after_pathsep(p, p + STRLEN(p)))
3027 STRCAT(p, PATHSEPSTR);
3028}
3029
3030/*
3031 * FullName_save - Make an allocated copy of a full file name.
3032 * Returns NULL when out of memory.
3033 */
3034 char_u *
3035FullName_save(
3036 char_u *fname,
3037 int force) // force expansion, even when it already looks
3038 // like a full path name
3039{
3040 char_u *buf;
3041 char_u *new_fname = NULL;
3042
3043 if (fname == NULL)
3044 return NULL;
3045
3046 buf = alloc(MAXPATHL);
3047 if (buf != NULL)
3048 {
3049 if (vim_FullName(fname, buf, MAXPATHL, force) != FAIL)
3050 new_fname = vim_strsave(buf);
3051 else
3052 new_fname = vim_strsave(fname);
3053 vim_free(buf);
3054 }
3055 return new_fname;
3056}
3057
3058/*
3059 * return TRUE if "fname" exists.
3060 */
3061 int
3062vim_fexists(char_u *fname)
3063{
3064 stat_T st;
3065
3066 if (mch_stat((char *)fname, &st))
3067 return FALSE;
3068 return TRUE;
3069}
3070
3071/*
3072 * Invoke expand_wildcards() for one pattern.
3073 * Expand items like "%:h" before the expansion.
3074 * Returns OK or FAIL.
3075 */
3076 int
3077expand_wildcards_eval(
3078 char_u **pat, // pointer to input pattern
3079 int *num_file, // resulting number of files
3080 char_u ***file, // array of resulting files
3081 int flags) // EW_DIR, etc.
3082{
3083 int ret = FAIL;
3084 char_u *eval_pat = NULL;
3085 char_u *exp_pat = *pat;
3086 char *ignored_msg;
3087 int usedlen;
3088
3089 if (*exp_pat == '%' || *exp_pat == '#' || *exp_pat == '<')
3090 {
3091 ++emsg_off;
3092 eval_pat = eval_vars(exp_pat, exp_pat, &usedlen,
3093 NULL, &ignored_msg, NULL);
3094 --emsg_off;
3095 if (eval_pat != NULL)
3096 exp_pat = concat_str(eval_pat, exp_pat + usedlen);
3097 }
3098
3099 if (exp_pat != NULL)
3100 ret = expand_wildcards(1, &exp_pat, num_file, file, flags);
3101
3102 if (eval_pat != NULL)
3103 {
3104 vim_free(exp_pat);
3105 vim_free(eval_pat);
3106 }
3107
3108 return ret;
3109}
3110
3111/*
3112 * Expand wildcards. Calls gen_expand_wildcards() and removes files matching
3113 * 'wildignore'.
3114 * Returns OK or FAIL. When FAIL then "num_files" won't be set.
3115 */
3116 int
3117expand_wildcards(
3118 int num_pat, // number of input patterns
3119 char_u **pat, // array of input patterns
3120 int *num_files, // resulting number of files
3121 char_u ***files, // array of resulting files
3122 int flags) // EW_DIR, etc.
3123{
3124 int retval;
3125 int i, j;
3126 char_u *p;
3127 int non_suf_match; // number without matching suffix
3128
3129 retval = gen_expand_wildcards(num_pat, pat, num_files, files, flags);
3130
3131 // When keeping all matches, return here
3132 if ((flags & EW_KEEPALL) || retval == FAIL)
3133 return retval;
3134
3135#ifdef FEAT_WILDIGN
3136 /*
3137 * Remove names that match 'wildignore'.
3138 */
3139 if (*p_wig)
3140 {
3141 char_u *ffname;
3142
3143 // check all files in (*files)[]
3144 for (i = 0; i < *num_files; ++i)
3145 {
3146 ffname = FullName_save((*files)[i], FALSE);
3147 if (ffname == NULL) // out of memory
3148 break;
3149# ifdef VMS
3150 vms_remove_version(ffname);
3151# endif
3152 if (match_file_list(p_wig, (*files)[i], ffname))
3153 {
3154 // remove this matching file from the list
3155 vim_free((*files)[i]);
3156 for (j = i; j + 1 < *num_files; ++j)
3157 (*files)[j] = (*files)[j + 1];
3158 --*num_files;
3159 --i;
3160 }
3161 vim_free(ffname);
3162 }
3163
3164 // If the number of matches is now zero, we fail.
3165 if (*num_files == 0)
3166 {
3167 VIM_CLEAR(*files);
3168 return FAIL;
3169 }
3170 }
3171#endif
3172
3173 /*
3174 * Move the names where 'suffixes' match to the end.
3175 */
3176 if (*num_files > 1)
3177 {
3178 non_suf_match = 0;
3179 for (i = 0; i < *num_files; ++i)
3180 {
3181 if (!match_suffix((*files)[i]))
3182 {
3183 /*
3184 * Move the name without matching suffix to the front
3185 * of the list.
3186 */
3187 p = (*files)[i];
3188 for (j = i; j > non_suf_match; --j)
3189 (*files)[j] = (*files)[j - 1];
3190 (*files)[non_suf_match++] = p;
3191 }
3192 }
3193 }
3194
3195 return retval;
3196}
3197
3198/*
3199 * Return TRUE if "fname" matches with an entry in 'suffixes'.
3200 */
3201 int
3202match_suffix(char_u *fname)
3203{
3204 int fnamelen, setsuflen;
3205 char_u *setsuf;
3206#define MAXSUFLEN 30 // maximum length of a file suffix
3207 char_u suf_buf[MAXSUFLEN];
3208
3209 fnamelen = (int)STRLEN(fname);
3210 setsuflen = 0;
3211 for (setsuf = p_su; *setsuf; )
3212 {
3213 setsuflen = copy_option_part(&setsuf, suf_buf, MAXSUFLEN, ".,");
3214 if (setsuflen == 0)
3215 {
3216 char_u *tail = gettail(fname);
3217
3218 // empty entry: match name without a '.'
3219 if (vim_strchr(tail, '.') == NULL)
3220 {
3221 setsuflen = 1;
3222 break;
3223 }
3224 }
3225 else
3226 {
3227 if (fnamelen >= setsuflen
3228 && fnamencmp(suf_buf, fname + fnamelen - setsuflen,
3229 (size_t)setsuflen) == 0)
3230 break;
3231 setsuflen = 0;
3232 }
3233 }
3234 return (setsuflen != 0);
3235}
3236
3237#ifdef VIM_BACKTICK
3238
3239/*
3240 * Return TRUE if we can expand this backtick thing here.
3241 */
3242 static int
3243vim_backtick(char_u *p)
3244{
3245 return (*p == '`' && *(p + 1) != NUL && *(p + STRLEN(p) - 1) == '`');
3246}
3247
3248/*
3249 * Expand an item in `backticks` by executing it as a command.
3250 * Currently only works when pat[] starts and ends with a `.
3251 * Returns number of file names found, -1 if an error is encountered.
3252 */
3253 static int
3254expand_backtick(
3255 garray_T *gap,
3256 char_u *pat,
3257 int flags) // EW_* flags
3258{
3259 char_u *p;
3260 char_u *cmd;
3261 char_u *buffer;
3262 int cnt = 0;
3263 int i;
3264
3265 // Create the command: lop off the backticks.
Bram Moolenaar71ccd032020-06-12 22:59:11 +02003266 cmd = vim_strnsave(pat + 1, STRLEN(pat) - 2);
Bram Moolenaar26262f82019-09-04 20:59:15 +02003267 if (cmd == NULL)
3268 return -1;
3269
3270#ifdef FEAT_EVAL
3271 if (*cmd == '=') // `={expr}`: Expand expression
Bram Moolenaarb171fb12020-06-24 20:34:03 +02003272 buffer = eval_to_string(cmd + 1, TRUE);
Bram Moolenaar26262f82019-09-04 20:59:15 +02003273 else
3274#endif
3275 buffer = get_cmd_output(cmd, NULL,
3276 (flags & EW_SILENT) ? SHELL_SILENT : 0, NULL);
3277 vim_free(cmd);
3278 if (buffer == NULL)
3279 return -1;
3280
3281 cmd = buffer;
3282 while (*cmd != NUL)
3283 {
3284 cmd = skipwhite(cmd); // skip over white space
3285 p = cmd;
3286 while (*p != NUL && *p != '\r' && *p != '\n') // skip over entry
3287 ++p;
3288 // add an entry if it is not empty
3289 if (p > cmd)
3290 {
3291 i = *p;
3292 *p = NUL;
3293 addfile(gap, cmd, flags);
3294 *p = i;
3295 ++cnt;
3296 }
3297 cmd = p;
3298 while (*cmd != NUL && (*cmd == '\r' || *cmd == '\n'))
3299 ++cmd;
3300 }
3301
3302 vim_free(buffer);
3303 return cnt;
3304}
3305#endif // VIM_BACKTICK
3306
3307#if defined(MSWIN)
3308/*
3309 * File name expansion code for MS-DOS, Win16 and Win32. It's here because
3310 * it's shared between these systems.
3311 */
3312
3313/*
3314 * comparison function for qsort in dos_expandpath()
3315 */
3316 static int
3317pstrcmp(const void *a, const void *b)
3318{
3319 return (pathcmp(*(char **)a, *(char **)b, -1));
3320}
3321
3322/*
3323 * Recursively expand one path component into all matching files and/or
3324 * directories. Adds matches to "gap". Handles "*", "?", "[a-z]", "**", etc.
3325 * Return the number of matches found.
3326 * "path" has backslashes before chars that are not to be expanded, starting
3327 * at "path[wildoff]".
3328 * Return the number of matches found.
3329 * NOTE: much of this is identical to unix_expandpath(), keep in sync!
3330 */
3331 static int
3332dos_expandpath(
3333 garray_T *gap,
3334 char_u *path,
3335 int wildoff,
3336 int flags, // EW_* flags
3337 int didstar) // expanded "**" once already
3338{
3339 char_u *buf;
3340 char_u *path_end;
3341 char_u *p, *s, *e;
3342 int start_len = gap->ga_len;
3343 char_u *pat;
3344 regmatch_T regmatch;
3345 int starts_with_dot;
3346 int matches;
3347 int len;
3348 int starstar = FALSE;
3349 static int stardepth = 0; // depth for "**" expansion
3350 HANDLE hFind = INVALID_HANDLE_VALUE;
3351 WIN32_FIND_DATAW wfb;
3352 WCHAR *wn = NULL; // UCS-2 name, NULL when not used.
3353 char_u *matchname;
3354 int ok;
Bram Moolenaarc74fbfe2020-04-06 22:56:28 +02003355 char_u *p_alt;
Bram Moolenaar26262f82019-09-04 20:59:15 +02003356
3357 // Expanding "**" may take a long time, check for CTRL-C.
3358 if (stardepth > 0)
3359 {
3360 ui_breakcheck();
3361 if (got_int)
3362 return 0;
3363 }
3364
3365 // Make room for file name. When doing encoding conversion the actual
3366 // length may be quite a bit longer, thus use the maximum possible length.
3367 buf = alloc(MAXPATHL);
3368 if (buf == NULL)
3369 return 0;
3370
3371 /*
3372 * Find the first part in the path name that contains a wildcard or a ~1.
3373 * Copy it into buf, including the preceding characters.
3374 */
3375 p = buf;
3376 s = buf;
3377 e = NULL;
3378 path_end = path;
3379 while (*path_end != NUL)
3380 {
3381 // May ignore a wildcard that has a backslash before it; it will
3382 // be removed by rem_backslash() or file_pat_to_reg_pat() below.
3383 if (path_end >= path + wildoff && rem_backslash(path_end))
3384 *p++ = *path_end++;
3385 else if (*path_end == '\\' || *path_end == ':' || *path_end == '/')
3386 {
3387 if (e != NULL)
3388 break;
3389 s = p + 1;
3390 }
3391 else if (path_end >= path + wildoff
3392 && vim_strchr((char_u *)"*?[~", *path_end) != NULL)
3393 e = p;
3394 if (has_mbyte)
3395 {
3396 len = (*mb_ptr2len)(path_end);
3397 STRNCPY(p, path_end, len);
3398 p += len;
3399 path_end += len;
3400 }
3401 else
3402 *p++ = *path_end++;
3403 }
3404 e = p;
3405 *e = NUL;
3406
3407 // now we have one wildcard component between s and e
3408 // Remove backslashes between "wildoff" and the start of the wildcard
3409 // component.
3410 for (p = buf + wildoff; p < s; ++p)
3411 if (rem_backslash(p))
3412 {
3413 STRMOVE(p, p + 1);
3414 --e;
3415 --s;
3416 }
3417
3418 // Check for "**" between "s" and "e".
3419 for (p = s; p < e; ++p)
3420 if (p[0] == '*' && p[1] == '*')
3421 starstar = TRUE;
3422
3423 starts_with_dot = *s == '.';
3424 pat = file_pat_to_reg_pat(s, e, NULL, FALSE);
3425 if (pat == NULL)
3426 {
3427 vim_free(buf);
3428 return 0;
3429 }
3430
3431 // compile the regexp into a program
3432 if (flags & (EW_NOERROR | EW_NOTWILD))
3433 ++emsg_silent;
3434 regmatch.rm_ic = TRUE; // Always ignore case
3435 regmatch.regprog = vim_regcomp(pat, RE_MAGIC);
3436 if (flags & (EW_NOERROR | EW_NOTWILD))
3437 --emsg_silent;
3438 vim_free(pat);
3439
3440 if (regmatch.regprog == NULL && (flags & EW_NOTWILD) == 0)
3441 {
3442 vim_free(buf);
3443 return 0;
3444 }
3445
3446 // remember the pattern or file name being looked for
3447 matchname = vim_strsave(s);
3448
3449 // If "**" is by itself, this is the first time we encounter it and more
3450 // is following then find matches without any directory.
3451 if (!didstar && stardepth < 100 && starstar && e - s == 2
3452 && *path_end == '/')
3453 {
3454 STRCPY(s, path_end + 1);
3455 ++stardepth;
3456 (void)dos_expandpath(gap, buf, (int)(s - buf), flags, TRUE);
3457 --stardepth;
3458 }
3459
3460 // Scan all files in the directory with "dir/ *.*"
3461 STRCPY(s, "*.*");
3462 wn = enc_to_utf16(buf, NULL);
3463 if (wn != NULL)
3464 hFind = FindFirstFileW(wn, &wfb);
3465 ok = (hFind != INVALID_HANDLE_VALUE);
3466
3467 while (ok)
3468 {
3469 p = utf16_to_enc(wfb.cFileName, NULL); // p is allocated here
Bram Moolenaarc74fbfe2020-04-06 22:56:28 +02003470
Bram Moolenaar26262f82019-09-04 20:59:15 +02003471 if (p == NULL)
3472 break; // out of memory
3473
Bram Moolenaar0dc5f602021-02-07 14:01:35 +01003474 // Do not use the alternate filename when the file name ends in '~',
3475 // because it picks up backup files: short name for "foo.vim~" is
3476 // "foo~1.vim", which matches "*.vim".
3477 if (*wfb.cAlternateFileName == NUL || p[STRLEN(p) - 1] == '~')
Bram Moolenaar40655d52020-04-06 23:49:50 +02003478 p_alt = NULL;
Bram Moolenaarc74fbfe2020-04-06 22:56:28 +02003479 else
3480 p_alt = utf16_to_enc(wfb.cAlternateFileName, NULL);
3481
Bram Moolenaar26262f82019-09-04 20:59:15 +02003482 // Ignore entries starting with a dot, unless when asked for. Accept
3483 // all entries found with "matchname".
3484 if ((p[0] != '.' || starts_with_dot
3485 || ((flags & EW_DODOT)
3486 && p[1] != NUL && (p[1] != '.' || p[2] != NUL)))
3487 && (matchname == NULL
3488 || (regmatch.regprog != NULL
Bram Moolenaar40655d52020-04-06 23:49:50 +02003489 && (vim_regexec(&regmatch, p, (colnr_T)0)
Bram Moolenaarc74fbfe2020-04-06 22:56:28 +02003490 || (p_alt != NULL
Bram Moolenaar40655d52020-04-06 23:49:50 +02003491 && vim_regexec(&regmatch, p_alt, (colnr_T)0))))
Bram Moolenaar26262f82019-09-04 20:59:15 +02003492 || ((flags & EW_NOTWILD)
3493 && fnamencmp(path + (s - buf), p, e - s) == 0)))
3494 {
3495 STRCPY(s, p);
3496 len = (int)STRLEN(buf);
3497
Bram Moolenaarc74fbfe2020-04-06 22:56:28 +02003498 if (starstar && stardepth < 100
3499 && (wfb.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
Bram Moolenaar26262f82019-09-04 20:59:15 +02003500 {
3501 // For "**" in the pattern first go deeper in the tree to
3502 // find matches.
3503 STRCPY(buf + len, "/**");
3504 STRCPY(buf + len + 3, path_end);
3505 ++stardepth;
3506 (void)dos_expandpath(gap, buf, len + 1, flags, TRUE);
3507 --stardepth;
3508 }
3509
3510 STRCPY(buf + len, path_end);
3511 if (mch_has_exp_wildcard(path_end))
3512 {
3513 // need to expand another component of the path
3514 // remove backslashes for the remaining components only
3515 (void)dos_expandpath(gap, buf, len + 1, flags, FALSE);
3516 }
3517 else
3518 {
3519 // no more wildcards, check if there is a match
3520 // remove backslashes for the remaining components only
3521 if (*path_end != 0)
3522 backslash_halve(buf + len + 1);
3523 if (mch_getperm(buf) >= 0) // add existing file
3524 addfile(gap, buf, flags);
3525 }
3526 }
3527
Bram Moolenaarc74fbfe2020-04-06 22:56:28 +02003528 vim_free(p_alt);
Bram Moolenaar26262f82019-09-04 20:59:15 +02003529 vim_free(p);
3530 ok = FindNextFileW(hFind, &wfb);
Bram Moolenaar26262f82019-09-04 20:59:15 +02003531 }
3532
3533 FindClose(hFind);
3534 vim_free(wn);
3535 vim_free(buf);
3536 vim_regfree(regmatch.regprog);
3537 vim_free(matchname);
3538
3539 matches = gap->ga_len - start_len;
3540 if (matches > 0)
3541 qsort(((char_u **)gap->ga_data) + start_len, (size_t)matches,
3542 sizeof(char_u *), pstrcmp);
3543 return matches;
3544}
3545
3546 int
3547mch_expandpath(
3548 garray_T *gap,
3549 char_u *path,
3550 int flags) // EW_* flags
3551{
3552 return dos_expandpath(gap, path, 0, flags, FALSE);
3553}
3554#endif // MSWIN
3555
3556#if (defined(UNIX) && !defined(VMS)) || defined(USE_UNIXFILENAME) \
3557 || defined(PROTO)
3558/*
3559 * Unix style wildcard expansion code.
3560 * It's here because it's used both for Unix and Mac.
3561 */
3562 static int
3563pstrcmp(const void *a, const void *b)
3564{
3565 return (pathcmp(*(char **)a, *(char **)b, -1));
3566}
3567
3568/*
3569 * Recursively expand one path component into all matching files and/or
3570 * directories. Adds matches to "gap". Handles "*", "?", "[a-z]", "**", etc.
3571 * "path" has backslashes before chars that are not to be expanded, starting
3572 * at "path + wildoff".
3573 * Return the number of matches found.
3574 * NOTE: much of this is identical to dos_expandpath(), keep in sync!
3575 */
3576 int
3577unix_expandpath(
3578 garray_T *gap,
3579 char_u *path,
3580 int wildoff,
3581 int flags, // EW_* flags
3582 int didstar) // expanded "**" once already
3583{
3584 char_u *buf;
3585 char_u *path_end;
3586 char_u *p, *s, *e;
3587 int start_len = gap->ga_len;
3588 char_u *pat;
3589 regmatch_T regmatch;
3590 int starts_with_dot;
3591 int matches;
3592 int len;
3593 int starstar = FALSE;
3594 static int stardepth = 0; // depth for "**" expansion
3595
3596 DIR *dirp;
3597 struct dirent *dp;
3598
3599 // Expanding "**" may take a long time, check for CTRL-C.
3600 if (stardepth > 0)
3601 {
3602 ui_breakcheck();
3603 if (got_int)
3604 return 0;
3605 }
3606
3607 // make room for file name
3608 buf = alloc(STRLEN(path) + BASENAMELEN + 5);
3609 if (buf == NULL)
3610 return 0;
3611
3612 /*
3613 * Find the first part in the path name that contains a wildcard.
3614 * When EW_ICASE is set every letter is considered to be a wildcard.
3615 * Copy it into "buf", including the preceding characters.
3616 */
3617 p = buf;
3618 s = buf;
3619 e = NULL;
3620 path_end = path;
3621 while (*path_end != NUL)
3622 {
3623 // May ignore a wildcard that has a backslash before it; it will
3624 // be removed by rem_backslash() or file_pat_to_reg_pat() below.
3625 if (path_end >= path + wildoff && rem_backslash(path_end))
3626 *p++ = *path_end++;
3627 else if (*path_end == '/')
3628 {
3629 if (e != NULL)
3630 break;
3631 s = p + 1;
3632 }
3633 else if (path_end >= path + wildoff
3634 && (vim_strchr((char_u *)"*?[{~$", *path_end) != NULL
3635 || (!p_fic && (flags & EW_ICASE)
3636 && isalpha(PTR2CHAR(path_end)))))
3637 e = p;
3638 if (has_mbyte)
3639 {
3640 len = (*mb_ptr2len)(path_end);
3641 STRNCPY(p, path_end, len);
3642 p += len;
3643 path_end += len;
3644 }
3645 else
3646 *p++ = *path_end++;
3647 }
3648 e = p;
3649 *e = NUL;
3650
3651 // Now we have one wildcard component between "s" and "e".
3652 // Remove backslashes between "wildoff" and the start of the wildcard
3653 // component.
3654 for (p = buf + wildoff; p < s; ++p)
3655 if (rem_backslash(p))
3656 {
3657 STRMOVE(p, p + 1);
3658 --e;
3659 --s;
3660 }
3661
3662 // Check for "**" between "s" and "e".
3663 for (p = s; p < e; ++p)
3664 if (p[0] == '*' && p[1] == '*')
3665 starstar = TRUE;
3666
3667 // convert the file pattern to a regexp pattern
3668 starts_with_dot = *s == '.';
3669 pat = file_pat_to_reg_pat(s, e, NULL, FALSE);
3670 if (pat == NULL)
3671 {
3672 vim_free(buf);
3673 return 0;
3674 }
3675
3676 // compile the regexp into a program
3677 if (flags & EW_ICASE)
3678 regmatch.rm_ic = TRUE; // 'wildignorecase' set
3679 else
3680 regmatch.rm_ic = p_fic; // ignore case when 'fileignorecase' is set
3681 if (flags & (EW_NOERROR | EW_NOTWILD))
3682 ++emsg_silent;
3683 regmatch.regprog = vim_regcomp(pat, RE_MAGIC);
3684 if (flags & (EW_NOERROR | EW_NOTWILD))
3685 --emsg_silent;
3686 vim_free(pat);
3687
3688 if (regmatch.regprog == NULL && (flags & EW_NOTWILD) == 0)
3689 {
3690 vim_free(buf);
3691 return 0;
3692 }
3693
3694 // If "**" is by itself, this is the first time we encounter it and more
3695 // is following then find matches without any directory.
3696 if (!didstar && stardepth < 100 && starstar && e - s == 2
3697 && *path_end == '/')
3698 {
3699 STRCPY(s, path_end + 1);
3700 ++stardepth;
3701 (void)unix_expandpath(gap, buf, (int)(s - buf), flags, TRUE);
3702 --stardepth;
3703 }
3704
3705 // open the directory for scanning
3706 *s = NUL;
3707 dirp = opendir(*buf == NUL ? "." : (char *)buf);
3708
3709 // Find all matching entries
3710 if (dirp != NULL)
3711 {
3712 for (;;)
3713 {
3714 dp = readdir(dirp);
3715 if (dp == NULL)
3716 break;
3717 if ((dp->d_name[0] != '.' || starts_with_dot
3718 || ((flags & EW_DODOT)
3719 && dp->d_name[1] != NUL
3720 && (dp->d_name[1] != '.' || dp->d_name[2] != NUL)))
3721 && ((regmatch.regprog != NULL && vim_regexec(&regmatch,
3722 (char_u *)dp->d_name, (colnr_T)0))
3723 || ((flags & EW_NOTWILD)
3724 && fnamencmp(path + (s - buf), dp->d_name, e - s) == 0)))
3725 {
3726 STRCPY(s, dp->d_name);
3727 len = STRLEN(buf);
3728
3729 if (starstar && stardepth < 100)
3730 {
3731 // For "**" in the pattern first go deeper in the tree to
3732 // find matches.
3733 STRCPY(buf + len, "/**");
3734 STRCPY(buf + len + 3, path_end);
3735 ++stardepth;
3736 (void)unix_expandpath(gap, buf, len + 1, flags, TRUE);
3737 --stardepth;
3738 }
3739
3740 STRCPY(buf + len, path_end);
3741 if (mch_has_exp_wildcard(path_end)) // handle more wildcards
3742 {
3743 // need to expand another component of the path
3744 // remove backslashes for the remaining components only
3745 (void)unix_expandpath(gap, buf, len + 1, flags, FALSE);
3746 }
3747 else
3748 {
3749 stat_T sb;
3750
3751 // no more wildcards, check if there is a match
3752 // remove backslashes for the remaining components only
3753 if (*path_end != NUL)
3754 backslash_halve(buf + len + 1);
3755 // add existing file or symbolic link
3756 if ((flags & EW_ALLLINKS) ? mch_lstat((char *)buf, &sb) >= 0
3757 : mch_getperm(buf) >= 0)
3758 {
3759#ifdef MACOS_CONVERT
3760 size_t precomp_len = STRLEN(buf)+1;
3761 char_u *precomp_buf =
3762 mac_precompose_path(buf, precomp_len, &precomp_len);
3763
3764 if (precomp_buf)
3765 {
3766 mch_memmove(buf, precomp_buf, precomp_len);
3767 vim_free(precomp_buf);
3768 }
3769#endif
3770 addfile(gap, buf, flags);
3771 }
3772 }
3773 }
3774 }
3775
3776 closedir(dirp);
3777 }
3778
3779 vim_free(buf);
3780 vim_regfree(regmatch.regprog);
3781
3782 matches = gap->ga_len - start_len;
3783 if (matches > 0)
3784 qsort(((char_u **)gap->ga_data) + start_len, matches,
3785 sizeof(char_u *), pstrcmp);
3786 return matches;
3787}
3788#endif
3789
3790/*
3791 * Return TRUE if "p" contains what looks like an environment variable.
3792 * Allowing for escaping.
3793 */
3794 static int
3795has_env_var(char_u *p)
3796{
3797 for ( ; *p; MB_PTR_ADV(p))
3798 {
3799 if (*p == '\\' && p[1] != NUL)
3800 ++p;
3801 else if (vim_strchr((char_u *)
3802#if defined(MSWIN)
3803 "$%"
3804#else
3805 "$"
3806#endif
3807 , *p) != NULL)
3808 return TRUE;
3809 }
3810 return FALSE;
3811}
3812
3813#ifdef SPECIAL_WILDCHAR
3814/*
3815 * Return TRUE if "p" contains a special wildcard character, one that Vim
3816 * cannot expand, requires using a shell.
3817 */
3818 static int
3819has_special_wildchar(char_u *p)
3820{
3821 for ( ; *p; MB_PTR_ADV(p))
3822 {
3823 // Disallow line break characters.
3824 if (*p == '\r' || *p == '\n')
3825 break;
3826 // Allow for escaping.
3827 if (*p == '\\' && p[1] != NUL && p[1] != '\r' && p[1] != '\n')
3828 ++p;
3829 else if (vim_strchr((char_u *)SPECIAL_WILDCHAR, *p) != NULL)
3830 {
3831 // A { must be followed by a matching }.
3832 if (*p == '{' && vim_strchr(p, '}') == NULL)
3833 continue;
3834 // A quote and backtick must be followed by another one.
3835 if ((*p == '`' || *p == '\'') && vim_strchr(p, *p) == NULL)
3836 continue;
3837 return TRUE;
3838 }
3839 }
3840 return FALSE;
3841}
3842#endif
3843
3844/*
3845 * Generic wildcard expansion code.
3846 *
3847 * Characters in "pat" that should not be expanded must be preceded with a
3848 * backslash. E.g., "/path\ with\ spaces/my\*star*"
3849 *
3850 * Return FAIL when no single file was found. In this case "num_file" is not
3851 * set, and "file" may contain an error message.
3852 * Return OK when some files found. "num_file" is set to the number of
3853 * matches, "file" to the array of matches. Call FreeWild() later.
3854 */
3855 int
3856gen_expand_wildcards(
3857 int num_pat, // number of input patterns
3858 char_u **pat, // array of input patterns
3859 int *num_file, // resulting number of files
3860 char_u ***file, // array of resulting files
3861 int flags) // EW_* flags
3862{
3863 int i;
3864 garray_T ga;
3865 char_u *p;
3866 static int recursive = FALSE;
3867 int add_pat;
3868 int retval = OK;
3869#if defined(FEAT_SEARCHPATH)
3870 int did_expand_in_path = FALSE;
3871#endif
3872
3873 /*
3874 * expand_env() is called to expand things like "~user". If this fails,
3875 * it calls ExpandOne(), which brings us back here. In this case, always
3876 * call the machine specific expansion function, if possible. Otherwise,
3877 * return FAIL.
3878 */
3879 if (recursive)
3880#ifdef SPECIAL_WILDCHAR
3881 return mch_expand_wildcards(num_pat, pat, num_file, file, flags);
3882#else
3883 return FAIL;
3884#endif
3885
3886#ifdef SPECIAL_WILDCHAR
3887 /*
3888 * If there are any special wildcard characters which we cannot handle
3889 * here, call machine specific function for all the expansion. This
3890 * avoids starting the shell for each argument separately.
3891 * For `=expr` do use the internal function.
3892 */
3893 for (i = 0; i < num_pat; i++)
3894 {
3895 if (has_special_wildchar(pat[i])
3896# ifdef VIM_BACKTICK
3897 && !(vim_backtick(pat[i]) && pat[i][1] == '=')
3898# endif
3899 )
3900 return mch_expand_wildcards(num_pat, pat, num_file, file, flags);
3901 }
3902#endif
3903
3904 recursive = TRUE;
3905
3906 /*
3907 * The matching file names are stored in a growarray. Init it empty.
3908 */
3909 ga_init2(&ga, (int)sizeof(char_u *), 30);
3910
3911 for (i = 0; i < num_pat; ++i)
3912 {
3913 add_pat = -1;
3914 p = pat[i];
3915
3916#ifdef VIM_BACKTICK
3917 if (vim_backtick(p))
3918 {
3919 add_pat = expand_backtick(&ga, p, flags);
3920 if (add_pat == -1)
3921 retval = FAIL;
3922 }
3923 else
3924#endif
3925 {
3926 /*
3927 * First expand environment variables, "~/" and "~user/".
3928 */
3929 if ((has_env_var(p) && !(flags & EW_NOTENV)) || *p == '~')
3930 {
3931 p = expand_env_save_opt(p, TRUE);
3932 if (p == NULL)
3933 p = pat[i];
3934#ifdef UNIX
3935 /*
3936 * On Unix, if expand_env() can't expand an environment
3937 * variable, use the shell to do that. Discard previously
3938 * found file names and start all over again.
3939 */
3940 else if (has_env_var(p) || *p == '~')
3941 {
3942 vim_free(p);
3943 ga_clear_strings(&ga);
3944 i = mch_expand_wildcards(num_pat, pat, num_file, file,
3945 flags|EW_KEEPDOLLAR);
3946 recursive = FALSE;
3947 return i;
3948 }
3949#endif
3950 }
3951
3952 /*
3953 * If there are wildcards: Expand file names and add each match to
3954 * the list. If there is no match, and EW_NOTFOUND is given, add
3955 * the pattern.
3956 * If there are no wildcards: Add the file name if it exists or
3957 * when EW_NOTFOUND is given.
3958 */
3959 if (mch_has_exp_wildcard(p))
3960 {
3961#if defined(FEAT_SEARCHPATH)
3962 if ((flags & EW_PATH)
3963 && !mch_isFullName(p)
3964 && !(p[0] == '.'
3965 && (vim_ispathsep(p[1])
3966 || (p[1] == '.' && vim_ispathsep(p[2]))))
3967 )
3968 {
3969 // :find completion where 'path' is used.
3970 // Recursiveness is OK here.
3971 recursive = FALSE;
3972 add_pat = expand_in_path(&ga, p, flags);
3973 recursive = TRUE;
3974 did_expand_in_path = TRUE;
3975 }
3976 else
3977#endif
3978 add_pat = mch_expandpath(&ga, p, flags);
3979 }
3980 }
3981
3982 if (add_pat == -1 || (add_pat == 0 && (flags & EW_NOTFOUND)))
3983 {
3984 char_u *t = backslash_halve_save(p);
3985
3986 // When EW_NOTFOUND is used, always add files and dirs. Makes
3987 // "vim c:/" work.
3988 if (flags & EW_NOTFOUND)
3989 addfile(&ga, t, flags | EW_DIR | EW_FILE);
3990 else
3991 addfile(&ga, t, flags);
3992
3993 if (t != p)
3994 vim_free(t);
3995 }
3996
3997#if defined(FEAT_SEARCHPATH)
3998 if (did_expand_in_path && ga.ga_len > 0 && (flags & EW_PATH))
3999 uniquefy_paths(&ga, p);
4000#endif
4001 if (p != pat[i])
4002 vim_free(p);
4003 }
4004
Bram Moolenaar566cc8c2020-06-29 21:14:51 +02004005 // When returning FAIL the array must be freed here.
4006 if (retval == FAIL)
4007 ga_clear(&ga);
4008
Bram Moolenaar26262f82019-09-04 20:59:15 +02004009 *num_file = ga.ga_len;
Bram Moolenaar566cc8c2020-06-29 21:14:51 +02004010 *file = (ga.ga_data != NULL) ? (char_u **)ga.ga_data
4011 : (char_u **)_("no matches");
Bram Moolenaar26262f82019-09-04 20:59:15 +02004012
4013 recursive = FALSE;
4014
4015 return ((flags & EW_EMPTYOK) || ga.ga_data != NULL) ? retval : FAIL;
4016}
4017
4018/*
4019 * Add a file to a file list. Accepted flags:
4020 * EW_DIR add directories
4021 * EW_FILE add files
4022 * EW_EXEC add executable files
4023 * EW_NOTFOUND add even when it doesn't exist
4024 * EW_ADDSLASH add slash after directory name
4025 * EW_ALLLINKS add symlink also when the referred file does not exist
4026 */
4027 void
4028addfile(
4029 garray_T *gap,
Bram Moolenaar217e1b82019-12-01 21:41:28 +01004030 char_u *f, // filename
Bram Moolenaar26262f82019-09-04 20:59:15 +02004031 int flags)
4032{
4033 char_u *p;
4034 int isdir;
4035 stat_T sb;
4036
4037 // if the file/dir/link doesn't exist, may not add it
4038 if (!(flags & EW_NOTFOUND) && ((flags & EW_ALLLINKS)
4039 ? mch_lstat((char *)f, &sb) < 0 : mch_getperm(f) < 0))
4040 return;
4041
4042#ifdef FNAME_ILLEGAL
4043 // if the file/dir contains illegal characters, don't add it
4044 if (vim_strpbrk(f, (char_u *)FNAME_ILLEGAL) != NULL)
4045 return;
4046#endif
4047
4048 isdir = mch_isdir(f);
4049 if ((isdir && !(flags & EW_DIR)) || (!isdir && !(flags & EW_FILE)))
4050 return;
4051
4052 // If the file isn't executable, may not add it. Do accept directories.
4053 // When invoked from expand_shellcmd() do not use $PATH.
4054 if (!isdir && (flags & EW_EXEC)
4055 && !mch_can_exe(f, NULL, !(flags & EW_SHELLCMD)))
4056 return;
4057
4058 // Make room for another item in the file list.
4059 if (ga_grow(gap, 1) == FAIL)
4060 return;
4061
4062 p = alloc(STRLEN(f) + 1 + isdir);
4063 if (p == NULL)
4064 return;
4065
4066 STRCPY(p, f);
4067#ifdef BACKSLASH_IN_FILENAME
4068 slash_adjust(p);
4069#endif
4070 /*
4071 * Append a slash or backslash after directory names if none is present.
4072 */
4073#ifndef DONT_ADD_PATHSEP_TO_DIR
4074 if (isdir && (flags & EW_ADDSLASH))
4075 add_pathsep(p);
4076#endif
4077 ((char_u **)gap->ga_data)[gap->ga_len++] = p;
4078}
4079
4080/*
4081 * Free the list of files returned by expand_wildcards() or other expansion
4082 * functions.
4083 */
4084 void
4085FreeWild(int count, char_u **files)
4086{
4087 if (count <= 0 || files == NULL)
4088 return;
4089 while (count--)
4090 vim_free(files[count]);
4091 vim_free(files);
4092}
4093
4094/*
4095 * Compare path "p[]" to "q[]".
4096 * If "maxlen" >= 0 compare "p[maxlen]" to "q[maxlen]"
4097 * Return value like strcmp(p, q), but consider path separators.
4098 */
4099 int
4100pathcmp(const char *p, const char *q, int maxlen)
4101{
4102 int i, j;
4103 int c1, c2;
4104 const char *s = NULL;
4105
4106 for (i = 0, j = 0; maxlen < 0 || (i < maxlen && j < maxlen);)
4107 {
4108 c1 = PTR2CHAR((char_u *)p + i);
4109 c2 = PTR2CHAR((char_u *)q + j);
4110
4111 // End of "p": check if "q" also ends or just has a slash.
4112 if (c1 == NUL)
4113 {
4114 if (c2 == NUL) // full match
4115 return 0;
4116 s = q;
4117 i = j;
4118 break;
4119 }
4120
4121 // End of "q": check if "p" just has a slash.
4122 if (c2 == NUL)
4123 {
4124 s = p;
4125 break;
4126 }
4127
4128 if ((p_fic ? MB_TOUPPER(c1) != MB_TOUPPER(c2) : c1 != c2)
4129#ifdef BACKSLASH_IN_FILENAME
4130 // consider '/' and '\\' to be equal
4131 && !((c1 == '/' && c2 == '\\')
4132 || (c1 == '\\' && c2 == '/'))
4133#endif
4134 )
4135 {
4136 if (vim_ispathsep(c1))
4137 return -1;
4138 if (vim_ispathsep(c2))
4139 return 1;
4140 return p_fic ? MB_TOUPPER(c1) - MB_TOUPPER(c2)
4141 : c1 - c2; // no match
4142 }
4143
Bram Moolenaar1614a142019-10-06 22:00:13 +02004144 i += mb_ptr2len((char_u *)p + i);
4145 j += mb_ptr2len((char_u *)q + j);
Bram Moolenaar26262f82019-09-04 20:59:15 +02004146 }
4147 if (s == NULL) // "i" or "j" ran into "maxlen"
4148 return 0;
4149
4150 c1 = PTR2CHAR((char_u *)s + i);
Bram Moolenaar1614a142019-10-06 22:00:13 +02004151 c2 = PTR2CHAR((char_u *)s + i + mb_ptr2len((char_u *)s + i));
Bram Moolenaar26262f82019-09-04 20:59:15 +02004152 // ignore a trailing slash, but not "//" or ":/"
4153 if (c2 == NUL
4154 && i > 0
4155 && !after_pathsep((char_u *)s, (char_u *)s + i)
4156#ifdef BACKSLASH_IN_FILENAME
4157 && (c1 == '/' || c1 == '\\')
4158#else
4159 && c1 == '/'
4160#endif
4161 )
4162 return 0; // match with trailing slash
4163 if (s == q)
4164 return -1; // no match
4165 return 1;
4166}
4167
4168/*
4169 * Return TRUE if "name" is a full (absolute) path name or URL.
4170 */
4171 int
4172vim_isAbsName(char_u *name)
4173{
4174 return (path_with_url(name) != 0 || mch_isFullName(name));
4175}
4176
4177/*
4178 * Get absolute file name into buffer "buf[len]".
4179 *
4180 * return FAIL for failure, OK otherwise
4181 */
4182 int
4183vim_FullName(
4184 char_u *fname,
4185 char_u *buf,
4186 int len,
4187 int force) // force expansion even when already absolute
4188{
4189 int retval = OK;
4190 int url;
4191
4192 *buf = NUL;
4193 if (fname == NULL)
4194 return FAIL;
4195
4196 url = path_with_url(fname);
4197 if (!url)
4198 retval = mch_FullName(fname, buf, len, force);
4199 if (url || retval == FAIL)
4200 {
4201 // something failed; use the file name (truncate when too long)
4202 vim_strncpy(buf, fname, len - 1);
4203 }
4204#if defined(MSWIN)
4205 slash_adjust(buf);
4206#endif
4207 return retval;
4208}