1 /* 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 /* 11 * filepath.c: dealing with file names and paths. 12 */ 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 30 get_short_pathname(char_u **fnamep, char_u **bufp, int *fnamelen) 31 { 32 int l, len; 33 WCHAR *newbuf; 34 WCHAR *wfname; 35 36 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); 49 if (l > len - 1) 50 { 51 // If that doesn't work (not enough space), then save the string 52 // and try again with a new buffer big enough. 53 WCHAR *newbuf_t = newbuf; 54 newbuf = vim_realloc(newbuf, (l + 1) * sizeof(*newbuf)); 55 if (newbuf == NULL) 56 { 57 vim_free(wfname); 58 vim_free(newbuf_t); 59 return FAIL; 60 } 61 // Really should always succeed, as the buffer is big enough. 62 l = GetShortPathNameW(wfname, newbuf, l+1); 63 } 64 if (l != 0) 65 { 66 char_u *p = utf16_to_enc(newbuf, NULL); 67 68 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); 82 83 *fnamelen = l == 0 ? l : (int)STRLEN(*bufp); 84 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 103 shortpath_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 115 // Make a copy 116 old_len = *fnamelen; 117 save_fname = vim_strnsave(*fname, old_len); 118 pbuf_unused = NULL; 119 short_fname = NULL; 120 121 endp = save_fname + old_len - 1; // Find the end of the copy 122 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 { 131 // go back one path-separator 132 while (endp > save_fname && !after_pathsep(save_fname, endp + 1)) 133 --endp; 134 if (endp <= save_fname) 135 break; // processed the complete path 136 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 } 150 *endp = ch; // preserve the string 151 152 if (len > 0) 153 break; // successfully shortened the path 154 155 // failed to shorten the path. Skip the path separator 156 --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 166 // Compute the length of the new path. 167 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 { 174 // There is not enough space in the currently allocated string, 175 // copy it to a buffer big enough. 176 *fname = *bufp = vim_strnsave(short_fname, new_len); 177 if (*fname == NULL) 178 { 179 retval = FAIL; 180 goto theend; 181 } 182 } 183 else 184 { 185 // Transfer short_fname to the main buffer (it's big enough), 186 // unless get_short_pathname() did its work in-place. 187 *fname = *bufp = save_fname; 188 if (short_fname != save_fname) 189 vim_strncpy(save_fname, short_fname, len); 190 save_fname = NULL; 191 } 192 193 // concat the not-shortened part of the path 194 vim_strncpy(*fname + len, endp, sfx_len); 195 (*fname)[new_len] = NUL; 196 } 197 198 theend: 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 210 shortpath_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 220 // Count up the path separators from the RHS.. so we know which part 221 // of the path to return. 222 sepcount = 0; 223 for (p = *fnamep; p < *fnamep + *fnamelen; MB_PTR_ADV(p)) 224 if (vim_ispathsep(*p)) 225 ++sepcount; 226 227 // Need full path first (use expand_env() to remove a "~/") 228 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 { 241 // 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. 244 len = tflen; 245 if (shortpath_for_invalid_fname(&tfname, &pbuf, &len) == FAIL) 246 return FAIL; 247 } 248 249 // Count the paths backward to find the beginning of the desired string. 250 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 273 // Copy in the string - p indexes into tfname - allocated at pbuf 274 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 291 modify_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; 305 int has_homerelative = 0; 306 #ifdef MSWIN 307 char_u *fname_start = *fnamep; 308 int has_shortname = 0; 309 #endif 310 311 repeat: 312 // ":p" - full path/file_name 313 if (src[*usedlen] == ':' && src[*usedlen + 1] == 'p') 314 { 315 has_fullname = 1; 316 317 valid |= VALID_PATH; 318 *usedlen += 2; 319 320 // Expand "~/path" for all systems and "~user/path" for Unix and VMS 321 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); 333 vim_free(*bufp); // free any allocated file name 334 *bufp = *fnamep; 335 if (*fnamep == NULL) 336 return -1; 337 } 338 339 // When "/." or "/.." is used: force expansion to get rid of it. 340 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 351 // FullName_save() is slow, don't use it when not needed. 352 if (*p != NUL || !vim_isAbsName(*fnamep)) 353 { 354 *fnamep = FullName_save(*fnamep, *p != NUL); 355 vim_free(*bufp); // free any allocated file name 356 *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 388 // Append a path separator to a directory. 389 if (mch_isdir(*fnamep)) 390 { 391 // Make room for one or two extra characters. 392 *fnamep = vim_strnsave(*fnamep, STRLEN(*fnamep) + 2); 393 vim_free(*bufp); // free any allocated file name 394 *bufp = *fnamep; 395 if (*fnamep == NULL) 396 return -1; 397 add_pathsep(*fnamep); 398 } 399 } 400 401 // ":." - path relative to the current directory 402 // ":~" - path relative to the home directory 403 // ":8" - shortname path - postponed till after 404 while (src[*usedlen] == ':' 405 && ((c = src[*usedlen + 1]) == '.' || c == '~' || c == '8')) 406 { 407 *usedlen += 2; 408 if (c == '8') 409 { 410 #ifdef MSWIN 411 has_shortname = 1; // Postpone this. 412 #endif 413 continue; 414 } 415 pbuf = NULL; 416 // Need full path first (use expand_env() to remove a "~/") 417 if (!has_fullname && !has_homerelative) 418 { 419 if ((c == '.' || c == '~') && **fnamep == '~') 420 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 { 433 size_t namelen; 434 435 mch_dirname(dirname, MAXPATHL); 436 if (has_homerelative) 437 { 438 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; 452 if (vim_ispathsep(*p)) 453 { 454 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 } 464 } 465 } 466 } 467 else 468 { 469 home_replace(NULL, p, dirname, MAXPATHL, TRUE); 470 // Only replace it when it starts with '~' 471 if (*dirname == '~') 472 { 473 s = vim_strsave(dirname); 474 if (s != NULL) 475 { 476 *fnamep = s; 477 vim_free(*bufp); 478 *bufp = s; 479 has_homerelative = TRUE; 480 } 481 } 482 } 483 vim_free(pbuf); 484 } 485 } 486 487 tail = gettail(*fnamep); 488 *fnamelen = (int)STRLEN(*fnamep); 489 490 // ":h" - head, remove "/file_name", can be repeated 491 // Don't remove the first "/" or "c:\" 492 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) 502 *fnamelen += 1; // the path separator is part of the path 503 #endif 504 if (*fnamelen == 0) 505 { 506 // Result is empty. Turn it into "." to make ":cd %:h" work. 507 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 521 // ":8" - shortname 522 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 { 536 // 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". 539 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 548 // Split into two implementations - makes it easier. First is where 549 // there isn't a full name already, second is where there is. 550 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 559 // Simple case, already have the full-name. 560 // Nearly always shorter, so try first time. 561 if (get_short_pathname(fnamep, bufp, &l) == FAIL) 562 return -1; 563 564 if (l == 0) 565 { 566 // Couldn't find the filename, search the paths. 567 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 576 // ":t" - tail, just the basename 577 if (src[*usedlen] == ':' && src[*usedlen + 1] == 't') 578 { 579 *usedlen += 2; 580 *fnamelen -= (int)(tail - *fnamep); 581 *fnamep = tail; 582 } 583 584 // ":e" - extension, can be repeated 585 // ":r" - root, without extension, can be repeated 586 while (src[*usedlen] == ':' 587 && (src[*usedlen + 1] == 'e' || src[*usedlen + 1] == 'r')) 588 { 589 // find a '.' in the tail: 590 // - for second :e: before the current fname 591 // - otherwise: The last '.' 592 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; 599 if (src[*usedlen + 1] == 'e') // :e 600 { 601 if (s > tail) 602 { 603 *fnamelen += (int)(*fnamep - (s + 1)); 604 *fnamep = s + 1; 605 #ifdef VMS 606 // cut version from the extension 607 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 } 618 else // :r 619 { 620 char_u *limit = *fnamep; 621 622 if (limit < tail) 623 limit = tail; 624 if (s > limit) // remove one extension 625 *fnamelen = (int)(s - *fnamep); 626 } 627 *usedlen += 2; 628 } 629 630 // ":s?pat?foo?" - substitute 631 // ":gs?pat?foo?" - global substitute 632 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 { 654 // find end of pattern 655 p = vim_strchr(s, sep); 656 if (p != NULL) 657 { 658 pat = vim_strnsave(s, p - s); 659 if (pat != NULL) 660 { 661 s = p + 1; 662 // find end of substitution 663 p = vim_strchr(s, sep); 664 if (p != NULL) 665 { 666 sub = vim_strnsave(s, p - s); 667 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 } 687 // after using ":s", repeat all the modifiers 688 if (didit) 689 goto repeat; 690 } 691 } 692 693 if (src[*usedlen] == ':' && src[*usedlen + 1] == 'S') 694 { 695 // vim_strsave_shellescape() needs a NUL terminated string. 696 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 713 /* 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 720 shorten_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 771 shorten_dir(char_u *str) 772 { 773 shorten_dir_len(str, 1); 774 } 775 776 #if defined(FEAT_EVAL) || defined(PROTO) 777 778 /* 779 * "chdir(dir)" function 780 */ 781 void 782 f_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) 791 { 792 // Returning an empty string means it failed. 793 // No error message, for historic reasons. 794 if (in_vim9script()) 795 (void) check_for_string_arg(argvars, 0); 796 return; 797 } 798 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 827 f_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 837 if (in_vim9script() 838 && (check_for_string_arg(argvars, 0) == FAIL 839 || check_for_opt_string_arg(argvars, 1) == FAIL)) 840 return; 841 842 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) 855 // delete a file 856 rettv->vval.v_number = mch_remove(name) == 0 ? 0 : -1; 857 else if (STRCMP(flags, "d") == 0) 858 // delete an empty directory 859 rettv->vval.v_number = mch_rmdir(name) == 0 ? 0 : -1; 860 else if (STRCMP(flags, "rf") == 0) 861 // delete a directory recursively 862 rettv->vval.v_number = delete_recursive(name); 863 else 864 semsg(_(e_invalid_expression_str), flags); 865 } 866 867 /* 868 * "executable()" function 869 */ 870 void 871 f_executable(typval_T *argvars, typval_T *rettv) 872 { 873 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL) 874 return; 875 876 // Check in $PATH and also check directly if there is a directory name. 877 rettv->vval.v_number = mch_can_exe(tv_get_string(&argvars[0]), NULL, TRUE); 878 } 879 880 /* 881 * "exepath()" function 882 */ 883 void 884 f_exepath(typval_T *argvars, typval_T *rettv) 885 { 886 char_u *p = NULL; 887 888 if (in_vim9script() && check_for_nonempty_string_arg(argvars, 0) == FAIL) 889 return; 890 (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 899 f_filereadable(typval_T *argvars, typval_T *rettv) 900 { 901 int fd; 902 char_u *p; 903 int n; 904 905 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL) 906 return; 907 908 #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 929 f_filewritable(typval_T *argvars, typval_T *rettv) 930 { 931 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL) 932 return; 933 rettv->vval.v_number = filewritable(tv_get_string(&argvars[0])); 934 } 935 936 static void 937 findfilendir( 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; 955 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))) 960 return; 961 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 1013 f_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 1022 f_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 1031 f_fnamemodify(typval_T *argvars, typval_T *rettv) 1032 { 1033 char_u *fname; 1034 char_u *mods; 1035 int usedlen = 0; 1036 int len = 0; 1037 char_u *fbuf = NULL; 1038 char_u buf[NUMBUFLEN]; 1039 1040 if (in_vim9script() 1041 && (check_for_string_arg(argvars, 0) == FAIL 1042 || check_for_string_arg(argvars, 1) == FAIL)) 1043 return; 1044 1045 fname = tv_get_string_chk(&argvars[0]); 1046 mods = tv_get_string_buf_chk(&argvars[1], buf); 1047 if (mods == NULL || fname == NULL) 1048 fname = NULL; 1049 else 1050 { 1051 len = (int)STRLEN(fname); 1052 if (mods != NULL && *mods != NUL) 1053 (void)modify_fname(mods, FALSE, &usedlen, &fname, &fbuf, &len); 1054 } 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 1082 f_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 1092 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 1098 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 /* 1131 * Convert "st" to file permission string. 1132 */ 1133 char_u * 1134 getfpermst(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 /* 1150 * "getfperm({fname})" function 1151 */ 1152 void 1153 f_getfperm(typval_T *argvars, typval_T *rettv) 1154 { 1155 char_u *fname; 1156 stat_T st; 1157 char_u *perm = NULL; 1158 char_u permbuf[] = "---------"; 1159 1160 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL) 1161 return; 1162 1163 fname = tv_get_string(&argvars[0]); 1164 1165 rettv->v_type = VAR_STRING; 1166 if (mch_stat((char *)fname, &st) >= 0) 1167 perm = vim_strsave(getfpermst(&st, permbuf)); 1168 rettv->vval.v_string = perm; 1169 } 1170 1171 /* 1172 * "getfsize({fname})" function 1173 */ 1174 void 1175 f_getfsize(typval_T *argvars, typval_T *rettv) 1176 { 1177 char_u *fname; 1178 stat_T st; 1179 1180 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL) 1181 return; 1182 1183 fname = tv_get_string(&argvars[0]); 1184 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 1192 // non-perfect check for overflow 1193 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 1205 f_getftime(typval_T *argvars, typval_T *rettv) 1206 { 1207 char_u *fname; 1208 stat_T st; 1209 1210 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL) 1211 return; 1212 1213 fname = tv_get_string(&argvars[0]); 1214 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 /* 1221 * Convert "st" to file type string. 1222 */ 1223 char_u * 1224 getftypest(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 /* 1248 * "getftype({fname})" function 1249 */ 1250 void 1251 f_getftype(typval_T *argvars, typval_T *rettv) 1252 { 1253 char_u *fname; 1254 stat_T st; 1255 char_u *type = NULL; 1256 1257 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL) 1258 return; 1259 1260 fname = tv_get_string(&argvars[0]); 1261 1262 rettv->v_type = VAR_STRING; 1263 if (mch_lstat((char *)fname, &st) >= 0) 1264 type = vim_strsave(getftypest(&st)); 1265 rettv->vval.v_string = type; 1266 } 1267 1268 /* 1269 * "glob()" function 1270 */ 1271 void 1272 f_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 1278 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 1287 // 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. 1289 rettv->v_type = VAR_STRING; 1290 if (argvars[1].v_type != VAR_UNKNOWN) 1291 { 1292 if (tv_get_bool_chk(&argvars[1], &error)) 1293 options |= WILD_KEEP_ALL; 1294 if (argvars[2].v_type != VAR_UNKNOWN) 1295 { 1296 if (tv_get_bool_chk(&argvars[2], &error)) 1297 rettv_list_set(rettv, NULL); 1298 if (argvars[3].v_type != VAR_UNKNOWN 1299 && tv_get_bool_chk(&argvars[3], &error)) 1300 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 1332 f_glob2regpat(typval_T *argvars, typval_T *rettv) 1333 { 1334 char_u buf[NUMBUFLEN]; 1335 char_u *pat; 1336 1337 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()); 1341 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 1350 f_globpath(typval_T *argvars, typval_T *rettv) 1351 { 1352 int flags = WILD_IGNORE_COMPLETESLASH; 1353 char_u buf1[NUMBUFLEN]; 1354 char_u *file; 1355 int error = FALSE; 1356 garray_T ga; 1357 int i; 1358 1359 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 1371 // 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 { 1376 if (tv_get_bool_chk(&argvars[2], &error)) 1377 flags |= WILD_KEEP_ALL; 1378 if (argvars[3].v_type != VAR_UNKNOWN) 1379 { 1380 if (tv_get_bool_chk(&argvars[3], &error)) 1381 rettv_list_set(rettv, NULL); 1382 if (argvars[4].v_type != VAR_UNKNOWN 1383 && tv_get_bool_chk(&argvars[4], &error)) 1384 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 1407 f_isdirectory(typval_T *argvars, typval_T *rettv) 1408 { 1409 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL) 1410 return; 1411 1412 rettv->vval.v_number = mch_isdir(tv_get_string(&argvars[0])); 1413 } 1414 1415 /* 1416 * Create the directory in which "dir" is located, and higher levels when 1417 * needed. 1418 * Return OK or FAIL. 1419 */ 1420 static int 1421 mkdir_recurse(char_u *dir, int prot) 1422 { 1423 char_u *p; 1424 char_u *updir; 1425 int r = FAIL; 1426 1427 // Get end of directory name in "dir". 1428 // We're done when it's "/" or "c:/". 1429 p = gettail_sep(dir); 1430 if (p <= get_past_head(dir)) 1431 return OK; 1432 1433 // If the directory exists we're done. Otherwise: create it. 1434 updir = vim_strnsave(dir, p - dir); 1435 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 1449 f_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 1459 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 1466 dir = tv_get_string_buf(&argvars[0], buf); 1467 if (*dir == NUL) 1468 return; 1469 1470 if (*gettail(dir) == NUL) 1471 // remove trailing slashes 1472 *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 { 1486 // With the "p" flag it's OK if the dir already exists. 1487 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 /* 1497 * "pathshorten()" function 1498 */ 1499 void 1500 f_pathshorten(typval_T *argvars, typval_T *rettv) 1501 { 1502 char_u *p; 1503 int trim_len = 1; 1504 1505 if (in_vim9script() 1506 && (check_for_string_arg(argvars, 0) == FAIL 1507 || check_for_opt_number_arg(argvars, 1) == FAIL)) 1508 return; 1509 1510 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 } 1516 1517 rettv->v_type = VAR_STRING; 1518 p = tv_get_string_chk(&argvars[0]); 1519 1520 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) 1527 shorten_dir_len(p, trim_len); 1528 } 1529 } 1530 1531 /* 1532 * Common code for readdir_checkitem() and readdirex_checkitem(). 1533 * Either "name" or "dict" is NULL. 1534 */ 1535 static int 1536 checkitem_common(void *context, char_u *name, dict_T *dict) 1537 { 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; 1544 1545 prepare_vimvar(VV_VAL, &save_val); 1546 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 } 1558 1559 if (eval_expr_typval(expr, argv, 1, &rettv) == FAIL) 1560 goto theend; 1561 1562 // 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 } 1568 retval = tv_get_number_chk(&rettv, &error); 1569 if (error) 1570 retval = -1; 1571 clear_tv(&rettv); 1572 1573 theend: 1574 if (name != NULL) 1575 set_vim_var_string(VV_VAL, NULL, 0); 1576 else 1577 set_vim_var_dict(VV_VAL, NULL); 1578 restore_vimvar(VV_VAL, &save_val); 1579 return retval; 1580 } 1581 1582 /* 1583 * Evaluate "expr" (= "context") for readdir(). 1584 */ 1585 static int 1586 readdir_checkitem(void *context, void *item) 1587 { 1588 char_u *name = (char_u *)item; 1589 1590 return checkitem_common(context, name, NULL); 1591 } 1592 1593 static int 1594 readdirex_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 1623 /* 1624 * "readdir()" function 1625 */ 1626 void 1627 f_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; 1635 int sort = READDIR_SORT_BYTE; 1636 1637 if (rettv_list_alloc(rettv) == FAIL) 1638 return; 1639 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 1646 path = tv_get_string(&argvars[0]); 1647 expr = &argvars[1]; 1648 1649 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN && 1650 readdirex_dict_arg(&argvars[2], &sort) == FAIL) 1651 return; 1652 1653 ret = readdir_core(&ga, path, FALSE, (void *)expr, 1654 (expr->v_type == VAR_UNKNOWN) ? NULL : readdir_checkitem, sort); 1655 if (ret == OK) 1656 { 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 /* 1667 * Evaluate "expr" (= "context") for readdirex(). 1668 */ 1669 static int 1670 readdirex_checkitem(void *context, void *item) 1671 { 1672 dict_T *dict = (dict_T*)item; 1673 1674 return checkitem_common(context, NULL, dict); 1675 } 1676 1677 /* 1678 * "readdirex()" function 1679 */ 1680 void 1681 f_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; 1688 int sort = READDIR_SORT_BYTE; 1689 1690 if (rettv_list_alloc(rettv) == FAIL) 1691 return; 1692 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 1699 path = tv_get_string(&argvars[0]); 1700 expr = &argvars[1]; 1701 1702 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN && 1703 readdirex_dict_arg(&argvars[2], &sort) == FAIL) 1704 return; 1705 1706 ret = readdir_core(&ga, path, TRUE, (void *)expr, 1707 (expr->v_type == VAR_UNKNOWN) ? NULL : readdirex_checkitem, sort); 1708 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 /* 1721 * "readfile()" function 1722 */ 1723 static void 1724 read_file_or_blob(typval_T *argvars, typval_T *rettv, int always_blob) 1725 { 1726 int binary = FALSE; 1727 int blob = always_blob; 1728 int failed = FALSE; 1729 char_u *fname; 1730 FILE *fd; 1731 char_u buf[(IOSIZE/256)*256]; // rounded to avoid odd + 1 1732 int io_size = sizeof(buf); 1733 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 1737 long maxline = MAXLNUM; 1738 long cnt = 0; 1739 char_u *p; // position in buf 1740 char_u *start; // start of current line 1741 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 1753 if ((blob ? rettv_blob_alloc(rettv) : rettv_list_alloc(rettv)) == FAIL) 1754 return; 1755 1756 // Always open the file in binary mode, library functions have a mind of 1757 // their own about CR-LF conversion. 1758 fname = tv_get_string(&argvars[0]); 1759 1760 if (mch_isdir(fname)) 1761 { 1762 semsg(_(e_src_is_directory), fname); 1763 return; 1764 } 1765 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 { 1775 semsg(_(e_notread), fname); 1776 // An empty blob is returned on error. 1777 blob_free(rettv->vval.v_blob); 1778 rettv->vval.v_blob = NULL; 1779 } 1780 fclose(fd); 1781 return; 1782 } 1783 1784 while (cnt < maxline || maxline < 0) 1785 { 1786 readlen = (int)fread(buf, 1, io_size, fd); 1787 1788 // 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. 1793 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 1803 // Finished a line. Remove CRs before NL. 1804 if (readlen > 0 && !binary) 1805 { 1806 while (len > 0 && start[len - 1] == '\r') 1807 --len; 1808 // removal may cross back to the "prev" string 1809 if (len == 0) 1810 while (prevlen > 0 && prev[prevlen - 1] == '\r') 1811 --prevlen; 1812 } 1813 if (prevlen == 0) 1814 s = vim_strnsave(start, len); 1815 else 1816 { 1817 // 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. 1820 if ((s = vim_realloc(prev, prevlen + len + 1)) != NULL) 1821 { 1822 mch_memmove(s + prevlen, start, len); 1823 s[prevlen + len] = NUL; 1824 prev = NULL; // the list will own the string 1825 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 1846 start = p + 1; // step over newline 1847 if ((++cnt >= maxline && maxline >= 0) || readlen <= 0) 1848 break; 1849 } 1850 else if (*p == NUL) 1851 *p = '\n'; 1852 // 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. 1854 else if (*p == 0xbf && enc_utf8 && !binary) 1855 { 1856 // Find the two bytes before the 0xbf. If p is at buf, or buf 1857 // + 1, these may be in the "prev" string. 1858 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 1868 // 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. 1870 if (start == dest) 1871 start = p + 1; 1872 else 1873 { 1874 // have to shuffle buf to close gap 1875 int adjust_prevlen = 0; 1876 1877 if (dest < buf) 1878 { 1879 // must be 1 or 2 1880 adjust_prevlen = (int)(buf - dest); 1881 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 } 1891 } // for 1892 1893 if (failed || (cnt >= maxline && maxline >= 0) || readlen <= 0) 1894 break; 1895 if (start < p) 1896 { 1897 // There's part of a line in buf, store it in "prev". 1898 if (p - start + prevlen >= prevsize) 1899 { 1900 // need bigger "prev" buffer 1901 char_u *newprev; 1902 1903 // 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. 1907 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 } 1924 // Add the line part to end of "prev". 1925 mch_memmove(prev + prevlen, start, p - start); 1926 prevlen += (long)(p - start); 1927 } 1928 } // while 1929 1930 // For a negative line count use only the lines at the end of the file, 1931 // free the rest. 1932 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 /* 1951 * "readblob()" function 1952 */ 1953 void 1954 f_readblob(typval_T *argvars, typval_T *rettv) 1955 { 1956 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL) 1957 return; 1958 1959 read_file_or_blob(argvars, rettv, TRUE); 1960 } 1961 1962 /* 1963 * "readfile()" function 1964 */ 1965 void 1966 f_readfile(typval_T *argvars, typval_T *rettv) 1967 { 1968 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 1975 read_file_or_blob(argvars, rettv, FALSE); 1976 } 1977 1978 /* 1979 * "resolve()" function 1980 */ 1981 void 1982 f_resolve(typval_T *argvars, typval_T *rettv) 1983 { 1984 char_u *p; 1985 #ifdef HAVE_READLINK 1986 char_u *buf = NULL; 1987 #endif 1988 1989 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL) 1990 return; 1991 1992 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); 2015 if (p == NULL) 2016 goto fail; 2017 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); 2022 if (len > 1 && after_pathsep(p, p + len)) 2023 { 2024 has_trailing_pathsep = TRUE; 2025 p[len - 1] = NUL; // the trailing slash breaks readlink() 2026 } 2027 2028 q = getnextcomp(p); 2029 if (*q != NUL) 2030 { 2031 // Separate the first path component in "p", and keep the 2032 // remainder (beginning with the path separator). 2033 remain = vim_strsave(q - 1); 2034 q[-1] = NUL; 2035 } 2036 2037 buf = alloc(MAXPATHL + 1); 2038 if (buf == NULL) 2039 { 2040 vim_free(p); 2041 goto fail; 2042 } 2043 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 2062 // Ensure that the result will have a trailing path separator 2063 // if the argument has one. 2064 if (remain == NULL && has_trailing_pathsep) 2065 add_pathsep(buf); 2066 2067 // Separate the first path component in the link value and 2068 // concatenate the remainders. 2069 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 { 2089 // Ignore trailing path separator. 2090 q[-1] = NUL; 2091 q = gettail(p); 2092 } 2093 if (q > p && !mch_isFullName(buf)) 2094 { 2095 // symlink is relative to directory of argument 2096 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 2115 // Append the first path component of "remain" to "p". 2116 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 } 2125 // Shorten "remain". 2126 if (*q != NUL) 2127 STRMOVE(remain, q - 1); 2128 else 2129 VIM_CLEAR(remain); 2130 } 2131 2132 // 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. 2134 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 { 2145 // Prepend "./". 2146 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 { 2155 // Strip leading "./". 2156 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 2164 // Ensure that the result will have no trailing path separator 2165 // if the argument had none. But keep "/" or "//". 2166 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 2183 fail: 2184 vim_free(buf); 2185 #endif 2186 rettv->v_type = VAR_STRING; 2187 } 2188 2189 /* 2190 * "tempname()" function 2191 */ 2192 void 2193 f_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 2200 // 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. 2202 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 2226 f_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 2244 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 2250 if (argvars[0].v_type == VAR_LIST) 2251 { 2252 list = argvars[0].vval.v_list; 2253 if (list == NULL) 2254 return; 2255 CHECK_LIST_MATERIALIZE(list); 2256 FOR_ALL_LIST_ITEMS(list, li) 2257 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 { 2268 semsg(_(e_invarg2), 2269 _("writefile() first argument must be a List or a Blob")); 2270 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 2295 // Always open the file in binary mode, library functions have a mind of 2296 // their own about CR-LF conversion. 2297 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) 2321 // Ignore the error, the user wouldn't know what to do about it. 2322 // May happen for a device. 2323 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 * 2339 do_browse( 2340 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 2348 { 2349 char_u *fname; 2350 static char_u *last_dir = NULL; // last used directory 2351 char_u *tofree = NULL; 2352 int save_cmod_flags = cmdmod.cmod_flags; 2353 2354 // Must turn off browse to avoid that autocommands will get the 2355 // flag too! 2356 cmdmod.cmod_flags &= ~CMOD_BROWSE; 2357 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 2368 // When no directory specified, use default file name, default dir, buffer 2369 // dir, last dir or current dir 2370 if ((initdir == NULL || *initdir == NUL) && dflt != NULL && *dflt != NUL) 2371 { 2372 if (mch_isdir(dflt)) // default file name is a directory 2373 { 2374 initdir = dflt; 2375 dflt = NULL; 2376 } 2377 else if (gettail(dflt) != dflt) // default file name includes a path 2378 { 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 { 2391 // When 'browsedir' is a directory, use it 2392 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; 2397 // When saving or 'browsedir' is "buffer", use buffer fname 2398 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 } 2410 // When 'browsedir' is "last", use dir from last browse 2411 else if (*p_bsdir == 'l') 2412 initdir = last_dir; 2413 // When 'browsedir is "current", use current directory. This is the 2414 // default already, leave initdir empty. 2415 } 2416 2417 # ifdef FEAT_GUI 2418 if (gui.in_use) // when this changes, also adjust f_has()! 2419 { 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) 2430 // For systems that have a directory dialog. 2431 fname = gui_mch_browsedir(title, initdir); 2432 # else 2433 // Generic solution for selecting a directory: select a file and 2434 // remove the file name. 2435 fname = gui_mch_browse(0, title, dflt, ext, initdir, (char_u *)""); 2436 # endif 2437 # if !defined(FEAT_GUI_GTK) 2438 // Win32 adds a dummy file name, others return an arbitrary file 2439 // name. GTK+ 2 returns only the directory, 2440 if (fname != NULL && *fname != NUL && !mch_isdir(fname)) 2441 { 2442 // Remove the file name. 2443 char_u *tail = gettail_sep(fname); 2444 2445 if (tail == fname) 2446 *tail++ = '.'; // use current dir 2447 *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 2455 // 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. 2458 need_check_timestamps = TRUE; 2459 did_check_timestamps = FALSE; 2460 } 2461 else 2462 # endif 2463 { 2464 // TODO: non-GUI file selector here 2465 emsg(_("E338: Sorry, no file browser in console mode")); 2466 fname = NULL; 2467 } 2468 2469 // keep the directory for next time 2470 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 { 2479 // filename only returned, must be in current dir 2480 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); 2489 cmdmod.cmod_flags = save_cmod_flags; 2490 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 2501 f_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 2512 if (in_vim9script() 2513 && (check_for_bool_arg(argvars, 0) == FAIL 2514 || check_for_string_arg(argvars, 1) == FAIL 2515 || check_for_string_arg(argvars, 2) == FAIL 2516 || check_for_string_arg(argvars, 3) == FAIL)) 2517 return; 2518 2519 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 2540 f_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 2547 if (in_vim9script() 2548 && (check_for_string_arg(argvars, 0) == FAIL 2549 || check_for_string_arg(argvars, 1) == FAIL)) 2550 return; 2551 2552 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 2567 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 2574 home_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 2667 // Do not add directory separator into dst, because dst is 2668 // expected to just return the directory name without the 2669 // directory separator '/'. 2670 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 * 2698 home_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 2725 fullpathcmp( 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 { 2746 // if mch_stat() doesn't work, may compare the names 2747 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 * 2809 gettail(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 * 2830 gettail_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 * 2850 getnextcomp(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 * 2865 get_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 2897 vim_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 2920 vim_ispathsep_nocolon(int c) 2921 { 2922 return vim_ispathsep(c) 2923 #ifdef BACKSLASH_IN_FILENAME 2924 && c != ':' 2925 #endif 2926 ; 2927 } 2928 2929 /* 2930 * 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 2935 dir_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 2956 vim_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 2968 vim_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; 2985 len -= mb_ptr2len(px); 2986 px += mb_ptr2len(px); 2987 py += mb_ptr2len(py); 2988 } 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 * 3004 concat_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 3024 add_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 * 3035 FullName_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 3062 vim_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 3077 expand_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 3117 expand_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 3202 match_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 3243 vim_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 3254 expand_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. 3266 cmd = vim_strnsave(pat + 1, STRLEN(pat) - 2); 3267 if (cmd == NULL) 3268 return -1; 3269 3270 #ifdef FEAT_EVAL 3271 if (*cmd == '=') // `={expr}`: Expand expression 3272 buffer = eval_to_string(cmd + 1, TRUE); 3273 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 3317 pstrcmp(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 3332 dos_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; 3355 char_u *p_alt; 3356 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 3470 3471 if (p == NULL) 3472 break; // out of memory 3473 3474 // 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] == '~') 3478 p_alt = NULL; 3479 else 3480 p_alt = utf16_to_enc(wfb.cAlternateFileName, NULL); 3481 3482 // 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 3489 && (vim_regexec(®match, p, (colnr_T)0) 3490 || (p_alt != NULL 3491 && vim_regexec(®match, p_alt, (colnr_T)0)))) 3492 || ((flags & EW_NOTWILD) 3493 && fnamencmp(path + (s - buf), p, e - s) == 0))) 3494 { 3495 STRCPY(s, p); 3496 len = (int)STRLEN(buf); 3497 3498 if (starstar && stardepth < 100 3499 && (wfb.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) 3500 { 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 3528 vim_free(p_alt); 3529 vim_free(p); 3530 ok = FindNextFileW(hFind, &wfb); 3531 } 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 3547 mch_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 3563 pstrcmp(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 3577 unix_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(®match, 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 3795 has_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 3819 has_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 3856 gen_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 4005 // When returning FAIL the array must be freed here. 4006 if (retval == FAIL) 4007 ga_clear(&ga); 4008 4009 *num_file = ga.ga_len; 4010 *file = (ga.ga_data != NULL) ? (char_u **)ga.ga_data 4011 : (char_u **)_("no matches"); 4012 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 4028 addfile( 4029 garray_T *gap, 4030 char_u *f, // filename 4031 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 4085 FreeWild(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 4100 pathcmp(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 4144 i += mb_ptr2len((char_u *)p + i); 4145 j += mb_ptr2len((char_u *)q + j); 4146 } 4147 if (s == NULL) // "i" or "j" ran into "maxlen" 4148 return 0; 4149 4150 c1 = PTR2CHAR((char_u *)s + i); 4151 c2 = PTR2CHAR((char_u *)s + i + mb_ptr2len((char_u *)s + i)); 4152 // 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 4172 vim_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 4183 vim_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 } 4209