blob: 9174265997eeb51c72a143be4d5079be2c361047 [file] [log] [blame]
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001/* vi:set ts=8 sts=4 sw=4 noet:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * strings.c: string manipulation functions
12 */
13
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +020014#define USING_FLOAT_STUFF
Yegappan Lakshmanana2438132021-07-10 21:29:18 +020015#include "vim.h"
16
17/*
18 * Copy "string" into newly allocated memory.
19 */
20 char_u *
21vim_strsave(char_u *string)
22{
23 char_u *p;
24 size_t len;
25
26 len = STRLEN(string) + 1;
27 p = alloc(len);
28 if (p != NULL)
29 mch_memmove(p, string, len);
30 return p;
31}
32
33/*
34 * Copy up to "len" bytes of "string" into newly allocated memory and
35 * terminate with a NUL.
36 * The allocated memory always has size "len + 1", also when "string" is
37 * shorter.
38 */
39 char_u *
40vim_strnsave(char_u *string, size_t len)
41{
42 char_u *p;
43
44 p = alloc(len + 1);
45 if (p != NULL)
46 {
47 STRNCPY(p, string, len);
48 p[len] = NUL;
49 }
50 return p;
51}
52
53/*
54 * Same as vim_strsave(), but any characters found in esc_chars are preceded
55 * by a backslash.
56 */
57 char_u *
58vim_strsave_escaped(char_u *string, char_u *esc_chars)
59{
60 return vim_strsave_escaped_ext(string, esc_chars, '\\', FALSE);
61}
62
63/*
64 * Same as vim_strsave_escaped(), but when "bsl" is TRUE also escape
65 * characters where rem_backslash() would remove the backslash.
66 * Escape the characters with "cc".
67 */
68 char_u *
69vim_strsave_escaped_ext(
70 char_u *string,
71 char_u *esc_chars,
72 int cc,
73 int bsl)
74{
75 char_u *p;
76 char_u *p2;
77 char_u *escaped_string;
78 unsigned length;
79 int l;
80
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +020081 // First count the number of backslashes required.
82 // Then allocate the memory and insert them.
Yegappan Lakshmanana2438132021-07-10 21:29:18 +020083 length = 1; // count the trailing NUL
84 for (p = string; *p; p++)
85 {
86 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
87 {
88 length += l; // count a multibyte char
89 p += l - 1;
90 continue;
91 }
92 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
93 ++length; // count a backslash
94 ++length; // count an ordinary char
95 }
96 escaped_string = alloc(length);
97 if (escaped_string != NULL)
98 {
99 p2 = escaped_string;
100 for (p = string; *p; p++)
101 {
102 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
103 {
104 mch_memmove(p2, p, (size_t)l);
105 p2 += l;
106 p += l - 1; // skip multibyte char
107 continue;
108 }
109 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
110 *p2++ = cc;
111 *p2++ = *p;
112 }
113 *p2 = NUL;
114 }
115 return escaped_string;
116}
117
118/*
119 * Return TRUE when 'shell' has "csh" in the tail.
120 */
121 int
122csh_like_shell(void)
123{
124 return (strstr((char *)gettail(p_sh), "csh") != NULL);
125}
126
127/*
Jason Cox6e823512021-08-29 12:36:49 +0200128 * Return TRUE when 'shell' has "fish" in the tail.
129 */
Dominique Pellede05ae72021-08-30 19:57:34 +0200130 static int
Jason Cox6e823512021-08-29 12:36:49 +0200131fish_like_shell(void)
132{
133 return (strstr((char *)gettail(p_sh), "fish") != NULL);
134}
135
136/*
Yegappan Lakshmanana2438132021-07-10 21:29:18 +0200137 * Escape "string" for use as a shell argument with system().
138 * This uses single quotes, except when we know we need to use double quotes
139 * (MS-DOS and MS-Windows not using PowerShell and without 'shellslash' set).
140 * PowerShell also uses a novel escaping for enclosed single quotes - double
141 * them up.
142 * Escape a newline, depending on the 'shell' option.
143 * When "do_special" is TRUE also replace "!", "%", "#" and things starting
144 * with "<" like "<cfile>".
145 * When "do_newline" is FALSE do not escape newline unless it is csh shell.
146 * Returns the result in allocated memory, NULL if we have run out.
147 */
148 char_u *
149vim_strsave_shellescape(char_u *string, int do_special, int do_newline)
150{
151 unsigned length;
152 char_u *p;
153 char_u *d;
154 char_u *escaped_string;
155 int l;
156 int csh_like;
Jason Cox6e823512021-08-29 12:36:49 +0200157 int fish_like;
Yegappan Lakshmanana2438132021-07-10 21:29:18 +0200158 char_u *shname;
159 int powershell;
160# ifdef MSWIN
161 int double_quotes;
162# endif
163
164 // Only csh and similar shells expand '!' within single quotes. For sh and
165 // the like we must not put a backslash before it, it will be taken
166 // literally. If do_special is set the '!' will be escaped twice.
167 // Csh also needs to have "\n" escaped twice when do_special is set.
168 csh_like = csh_like_shell();
169
Jason Cox6e823512021-08-29 12:36:49 +0200170 // Fish shell uses '\' as an escape character within single quotes, so '\'
171 // itself must be escaped to get a literal '\'.
172 fish_like = fish_like_shell();
173
Dominique Pelleaf4a61a2021-12-27 17:21:41 +0000174 // PowerShell uses its own version for quoting single quotes
Yegappan Lakshmanana2438132021-07-10 21:29:18 +0200175 shname = gettail(p_sh);
176 powershell = strstr((char *)shname, "pwsh") != NULL;
177# ifdef MSWIN
178 powershell = powershell || strstr((char *)shname, "powershell") != NULL;
179 // PowerShell only accepts single quotes so override shellslash.
180 double_quotes = !powershell && !p_ssl;
181# endif
182
183 // First count the number of extra bytes required.
184 length = (unsigned)STRLEN(string) + 3; // two quotes and a trailing NUL
185 for (p = string; *p != NUL; MB_PTR_ADV(p))
186 {
187# ifdef MSWIN
188 if (double_quotes)
189 {
190 if (*p == '"')
191 ++length; // " -> ""
192 }
193 else
194# endif
195 if (*p == '\'')
196 {
197 if (powershell)
198 length +=2; // ' => ''
199 else
200 length += 3; // ' => '\''
201 }
202 if ((*p == '\n' && (csh_like || do_newline))
203 || (*p == '!' && (csh_like || do_special)))
204 {
205 ++length; // insert backslash
206 if (csh_like && do_special)
207 ++length; // insert backslash
208 }
209 if (do_special && find_cmdline_var(p, &l) >= 0)
210 {
211 ++length; // insert backslash
212 p += l - 1;
213 }
Jason Cox6e823512021-08-29 12:36:49 +0200214 if (*p == '\\' && fish_like)
215 ++length; // insert backslash
Yegappan Lakshmanana2438132021-07-10 21:29:18 +0200216 }
217
218 // Allocate memory for the result and fill it.
219 escaped_string = alloc(length);
220 if (escaped_string != NULL)
221 {
222 d = escaped_string;
223
224 // add opening quote
225# ifdef MSWIN
226 if (double_quotes)
227 *d++ = '"';
228 else
229# endif
230 *d++ = '\'';
231
232 for (p = string; *p != NUL; )
233 {
234# ifdef MSWIN
235 if (double_quotes)
236 {
237 if (*p == '"')
238 {
239 *d++ = '"';
240 *d++ = '"';
241 ++p;
242 continue;
243 }
244 }
245 else
246# endif
247 if (*p == '\'')
248 {
249 if (powershell)
250 {
251 *d++ = '\'';
252 *d++ = '\'';
253 }
254 else
255 {
256 *d++ = '\'';
257 *d++ = '\\';
258 *d++ = '\'';
259 *d++ = '\'';
260 }
261 ++p;
262 continue;
263 }
264 if ((*p == '\n' && (csh_like || do_newline))
265 || (*p == '!' && (csh_like || do_special)))
266 {
267 *d++ = '\\';
268 if (csh_like && do_special)
269 *d++ = '\\';
270 *d++ = *p++;
271 continue;
272 }
273 if (do_special && find_cmdline_var(p, &l) >= 0)
274 {
275 *d++ = '\\'; // insert backslash
276 while (--l >= 0) // copy the var
277 *d++ = *p++;
278 continue;
279 }
Jason Cox6e823512021-08-29 12:36:49 +0200280 if (*p == '\\' && fish_like)
281 {
282 *d++ = '\\';
283 *d++ = *p++;
Bram Moolenaar66315972021-09-01 14:31:51 +0200284 continue;
Jason Cox6e823512021-08-29 12:36:49 +0200285 }
Yegappan Lakshmanana2438132021-07-10 21:29:18 +0200286
287 MB_COPY_CHAR(p, d);
288 }
289
290 // add terminating quote and finish with a NUL
291# ifdef MSWIN
292 if (double_quotes)
293 *d++ = '"';
294 else
295# endif
296 *d++ = '\'';
297 *d = NUL;
298 }
299
300 return escaped_string;
301}
302
303/*
304 * Like vim_strsave(), but make all characters uppercase.
305 * This uses ASCII lower-to-upper case translation, language independent.
306 */
307 char_u *
308vim_strsave_up(char_u *string)
309{
310 char_u *p1;
311
312 p1 = vim_strsave(string);
313 vim_strup(p1);
314 return p1;
315}
316
317/*
318 * Like vim_strnsave(), but make all characters uppercase.
319 * This uses ASCII lower-to-upper case translation, language independent.
320 */
321 char_u *
322vim_strnsave_up(char_u *string, size_t len)
323{
324 char_u *p1;
325
326 p1 = vim_strnsave(string, len);
327 vim_strup(p1);
328 return p1;
329}
330
331/*
332 * ASCII lower-to-upper case translation, language independent.
333 */
334 void
335vim_strup(
336 char_u *p)
337{
338 char_u *p2;
339 int c;
340
341 if (p != NULL)
342 {
343 p2 = p;
344 while ((c = *p2) != NUL)
Yegappan Lakshmanana2438132021-07-10 21:29:18 +0200345 *p2++ = (c < 'a' || c > 'z') ? c : (c - 0x20);
Yegappan Lakshmanana2438132021-07-10 21:29:18 +0200346 }
347}
348
349#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
350/*
351 * Make string "s" all upper-case and return it in allocated memory.
352 * Handles multi-byte characters as well as possible.
353 * Returns NULL when out of memory.
354 */
355 static char_u *
356strup_save(char_u *orig)
357{
358 char_u *p;
359 char_u *res;
360
361 res = p = vim_strsave(orig);
362
363 if (res != NULL)
364 while (*p != NUL)
365 {
366 int l;
367
368 if (enc_utf8)
369 {
370 int c, uc;
371 int newl;
372 char_u *s;
373
374 c = utf_ptr2char(p);
375 l = utf_ptr2len(p);
376 if (c == 0)
377 {
378 // overlong sequence, use only the first byte
379 c = *p;
380 l = 1;
381 }
382 uc = utf_toupper(c);
383
384 // Reallocate string when byte count changes. This is rare,
385 // thus it's OK to do another malloc()/free().
386 newl = utf_char2len(uc);
387 if (newl != l)
388 {
389 s = alloc(STRLEN(res) + 1 + newl - l);
390 if (s == NULL)
391 {
392 vim_free(res);
393 return NULL;
394 }
395 mch_memmove(s, res, p - res);
396 STRCPY(s + (p - res) + newl, p + l);
397 p = s + (p - res);
398 vim_free(res);
399 res = s;
400 }
401
402 utf_char2bytes(uc, p);
403 p += newl;
404 }
405 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
406 p += l; // skip multi-byte character
407 else
408 {
409 *p = TOUPPER_LOC(*p); // note that toupper() can be a macro
410 p++;
411 }
412 }
413
414 return res;
415}
416
417/*
418 * Make string "s" all lower-case and return it in allocated memory.
419 * Handles multi-byte characters as well as possible.
420 * Returns NULL when out of memory.
421 */
422 char_u *
423strlow_save(char_u *orig)
424{
425 char_u *p;
426 char_u *res;
427
428 res = p = vim_strsave(orig);
429
430 if (res != NULL)
431 while (*p != NUL)
432 {
433 int l;
434
435 if (enc_utf8)
436 {
437 int c, lc;
438 int newl;
439 char_u *s;
440
441 c = utf_ptr2char(p);
442 l = utf_ptr2len(p);
443 if (c == 0)
444 {
445 // overlong sequence, use only the first byte
446 c = *p;
447 l = 1;
448 }
449 lc = utf_tolower(c);
450
451 // Reallocate string when byte count changes. This is rare,
452 // thus it's OK to do another malloc()/free().
453 newl = utf_char2len(lc);
454 if (newl != l)
455 {
456 s = alloc(STRLEN(res) + 1 + newl - l);
457 if (s == NULL)
458 {
459 vim_free(res);
460 return NULL;
461 }
462 mch_memmove(s, res, p - res);
463 STRCPY(s + (p - res) + newl, p + l);
464 p = s + (p - res);
465 vim_free(res);
466 res = s;
467 }
468
469 utf_char2bytes(lc, p);
470 p += newl;
471 }
472 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
473 p += l; // skip multi-byte character
474 else
475 {
476 *p = TOLOWER_LOC(*p); // note that tolower() can be a macro
477 p++;
478 }
479 }
480
481 return res;
482}
483#endif
484
485/*
486 * delete spaces at the end of a string
487 */
488 void
489del_trailing_spaces(char_u *ptr)
490{
491 char_u *q;
492
493 q = ptr + STRLEN(ptr);
494 while (--q > ptr && VIM_ISWHITE(q[0]) && q[-1] != '\\' && q[-1] != Ctrl_V)
495 *q = NUL;
496}
497
498/*
499 * Like strncpy(), but always terminate the result with one NUL.
500 * "to" must be "len + 1" long!
501 */
502 void
503vim_strncpy(char_u *to, char_u *from, size_t len)
504{
505 STRNCPY(to, from, len);
506 to[len] = NUL;
507}
508
509/*
510 * Like strcat(), but make sure the result fits in "tosize" bytes and is
511 * always NUL terminated. "from" and "to" may overlap.
512 */
513 void
514vim_strcat(char_u *to, char_u *from, size_t tosize)
515{
516 size_t tolen = STRLEN(to);
517 size_t fromlen = STRLEN(from);
518
519 if (tolen + fromlen + 1 > tosize)
520 {
521 mch_memmove(to + tolen, from, tosize - tolen - 1);
522 to[tosize - 1] = NUL;
523 }
524 else
525 mch_memmove(to + tolen, from, fromlen + 1);
526}
527
Bram Moolenaarc32949b2023-01-04 15:56:51 +0000528/*
529 * A version of strlen() that has a maximum length.
530 */
531 size_t
532vim_strlen_maxlen(char *s, size_t maxlen)
533{
534 size_t i;
535 for (i = 0; i < maxlen; ++i)
536 if (s[i] == NUL)
537 break;
538 return i;
539}
540
Yegappan Lakshmanana2438132021-07-10 21:29:18 +0200541#if (!defined(HAVE_STRCASECMP) && !defined(HAVE_STRICMP)) || defined(PROTO)
542/*
543 * Compare two strings, ignoring case, using current locale.
544 * Doesn't work for multi-byte characters.
545 * return 0 for match, < 0 for smaller, > 0 for bigger
546 */
547 int
548vim_stricmp(char *s1, char *s2)
549{
550 int i;
551
552 for (;;)
553 {
554 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
555 if (i != 0)
556 return i; // this character different
557 if (*s1 == NUL)
558 break; // strings match until NUL
559 ++s1;
560 ++s2;
561 }
562 return 0; // strings match
563}
564#endif
565
566#if (!defined(HAVE_STRNCASECMP) && !defined(HAVE_STRNICMP)) || defined(PROTO)
567/*
568 * Compare two strings, for length "len", ignoring case, using current locale.
569 * Doesn't work for multi-byte characters.
570 * return 0 for match, < 0 for smaller, > 0 for bigger
571 */
572 int
573vim_strnicmp(char *s1, char *s2, size_t len)
574{
575 int i;
576
577 while (len > 0)
578 {
579 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
580 if (i != 0)
581 return i; // this character different
582 if (*s1 == NUL)
583 break; // strings match until NUL
584 ++s1;
585 ++s2;
586 --len;
587 }
588 return 0; // strings match
589}
590#endif
591
592/*
593 * Search for first occurrence of "c" in "string".
594 * Version of strchr() that handles unsigned char strings with characters from
595 * 128 to 255 correctly. It also doesn't return a pointer to the NUL at the
596 * end of the string.
597 */
Bram Moolenaarc32949b2023-01-04 15:56:51 +0000598 char_u *
Yegappan Lakshmanana2438132021-07-10 21:29:18 +0200599vim_strchr(char_u *string, int c)
600{
601 char_u *p;
602 int b;
603
604 p = string;
605 if (enc_utf8 && c >= 0x80)
606 {
607 while (*p != NUL)
608 {
609 int l = utfc_ptr2len(p);
610
611 // Avoid matching an illegal byte here.
612 if (utf_ptr2char(p) == c && l > 1)
613 return p;
614 p += l;
615 }
616 return NULL;
617 }
618 if (enc_dbcs != 0 && c > 255)
619 {
620 int n2 = c & 0xff;
621
622 c = ((unsigned)c >> 8) & 0xff;
623 while ((b = *p) != NUL)
624 {
625 if (b == c && p[1] == n2)
626 return p;
627 p += (*mb_ptr2len)(p);
628 }
629 return NULL;
630 }
631 if (has_mbyte)
632 {
633 while ((b = *p) != NUL)
634 {
635 if (b == c)
636 return p;
637 p += (*mb_ptr2len)(p);
638 }
639 return NULL;
640 }
641 while ((b = *p) != NUL)
642 {
643 if (b == c)
644 return p;
645 ++p;
646 }
647 return NULL;
648}
649
650/*
651 * Version of strchr() that only works for bytes and handles unsigned char
652 * strings with characters above 128 correctly. It also doesn't return a
653 * pointer to the NUL at the end of the string.
654 */
655 char_u *
656vim_strbyte(char_u *string, int c)
657{
658 char_u *p = string;
659
660 while (*p != NUL)
661 {
662 if (*p == c)
663 return p;
664 ++p;
665 }
666 return NULL;
667}
668
669/*
670 * Search for last occurrence of "c" in "string".
671 * Version of strrchr() that handles unsigned char strings with characters from
672 * 128 to 255 correctly. It also doesn't return a pointer to the NUL at the
673 * end of the string.
674 * Return NULL if not found.
675 * Does not handle multi-byte char for "c"!
676 */
677 char_u *
678vim_strrchr(char_u *string, int c)
679{
680 char_u *retval = NULL;
681 char_u *p = string;
682
683 while (*p)
684 {
685 if (*p == c)
686 retval = p;
687 MB_PTR_ADV(p);
688 }
689 return retval;
690}
691
692/*
693 * Vim's version of strpbrk(), in case it's missing.
694 * Don't generate a prototype for this, causes problems when it's not used.
695 */
696#ifndef PROTO
697# ifndef HAVE_STRPBRK
698# ifdef vim_strpbrk
699# undef vim_strpbrk
700# endif
701 char_u *
702vim_strpbrk(char_u *s, char_u *charset)
703{
704 while (*s)
705 {
706 if (vim_strchr(charset, *s) != NULL)
707 return s;
708 MB_PTR_ADV(s);
709 }
710 return NULL;
711}
712# endif
713#endif
714
715/*
716 * Sort an array of strings.
717 */
718static int sort_compare(const void *s1, const void *s2);
719
720 static int
721sort_compare(const void *s1, const void *s2)
722{
723 return STRCMP(*(char **)s1, *(char **)s2);
724}
725
726 void
727sort_strings(
728 char_u **files,
729 int count)
730{
731 qsort((void *)files, (size_t)count, sizeof(char_u *), sort_compare);
732}
733
734#if defined(FEAT_QUICKFIX) || defined(FEAT_SPELL) || defined(PROTO)
735/*
736 * Return TRUE if string "s" contains a non-ASCII character (128 or higher).
737 * When "s" is NULL FALSE is returned.
738 */
739 int
740has_non_ascii(char_u *s)
741{
742 char_u *p;
743
744 if (s != NULL)
745 for (p = s; *p != NUL; ++p)
746 if (*p >= 128)
747 return TRUE;
748 return FALSE;
749}
750#endif
751
752/*
753 * Concatenate two strings and return the result in allocated memory.
754 * Returns NULL when out of memory.
755 */
756 char_u *
757concat_str(char_u *str1, char_u *str2)
758{
759 char_u *dest;
760 size_t l = str1 == NULL ? 0 : STRLEN(str1);
761
762 dest = alloc(l + (str2 == NULL ? 0 : STRLEN(str2)) + 1L);
763 if (dest != NULL)
764 {
765 if (str1 == NULL)
766 *dest = NUL;
767 else
768 STRCPY(dest, str1);
769 if (str2 != NULL)
770 STRCPY(dest + l, str2);
771 }
772 return dest;
773}
774
775#if defined(FEAT_EVAL) || defined(PROTO)
Yegappan Lakshmanana2438132021-07-10 21:29:18 +0200776/*
777 * Return string "str" in ' quotes, doubling ' characters.
778 * If "str" is NULL an empty string is assumed.
779 * If "function" is TRUE make it function('string').
780 */
781 char_u *
782string_quote(char_u *str, int function)
783{
784 unsigned len;
785 char_u *p, *r, *s;
786
787 len = (function ? 13 : 3);
788 if (str != NULL)
789 {
790 len += (unsigned)STRLEN(str);
791 for (p = str; *p != NUL; MB_PTR_ADV(p))
792 if (*p == '\'')
793 ++len;
794 }
795 s = r = alloc(len);
796 if (r != NULL)
797 {
798 if (function)
799 {
800 STRCPY(r, "function('");
801 r += 10;
802 }
803 else
804 *r++ = '\'';
805 if (str != NULL)
806 for (p = str; *p != NUL; )
807 {
808 if (*p == '\'')
809 *r++ = '\'';
810 MB_COPY_CHAR(p, r);
811 }
812 *r++ = '\'';
813 if (function)
814 *r++ = ')';
815 *r++ = NUL;
816 }
817 return s;
818}
819
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000820/*
821 * Count the number of times "needle" occurs in string "haystack". Case is
822 * ignored if "ic" is TRUE.
823 */
824 long
825string_count(char_u *haystack, char_u *needle, int ic)
826{
827 long n = 0;
828 char_u *p = haystack;
829 char_u *next;
830
831 if (p == NULL || needle == NULL || *needle == NUL)
832 return 0;
833
834 if (ic)
835 {
836 size_t len = STRLEN(needle);
837
838 while (*p != NUL)
839 {
840 if (MB_STRNICMP(p, needle, len) == 0)
841 {
842 ++n;
843 p += len;
844 }
845 else
846 MB_PTR_ADV(p);
847 }
848 }
849 else
850 while ((next = (char_u *)strstr((char *)p, (char *)needle)) != NULL)
851 {
852 ++n;
853 p = next + STRLEN(needle);
854 }
855
856 return n;
857}
858
859/*
860 * Make a typval_T of the first character of "input" and store it in "output".
861 * Return OK or FAIL.
862 */
863 static int
864copy_first_char_to_tv(char_u *input, typval_T *output)
865{
866 char_u buf[MB_MAXBYTES + 1];
867 int len;
868
869 if (input == NULL || output == NULL)
870 return FAIL;
871
872 len = has_mbyte ? mb_ptr2len(input) : 1;
873 STRNCPY(buf, input, len);
874 buf[len] = NUL;
875 output->v_type = VAR_STRING;
876 output->vval.v_string = vim_strsave(buf);
877
878 return output->vval.v_string == NULL ? FAIL : OK;
879}
880
881/*
882 * Implementation of map() and filter() for a String. Apply "expr" to every
883 * character in string "str" and return the result in "rettv".
884 */
885 void
886string_filter_map(
887 char_u *str,
888 filtermap_T filtermap,
889 typval_T *expr,
890 typval_T *rettv)
891{
892 char_u *p;
893 typval_T tv;
894 garray_T ga;
895 int len = 0;
896 int idx = 0;
897 int rem;
Bram Moolenaar82418262022-09-28 16:16:15 +0100898 typval_T newtv;
899 funccall_T *fc;
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000900
901 rettv->v_type = VAR_STRING;
902 rettv->vval.v_string = NULL;
903
904 // set_vim_var_nr() doesn't set the type
905 set_vim_var_type(VV_KEY, VAR_NUMBER);
906
Bram Moolenaar82418262022-09-28 16:16:15 +0100907 // Create one funccal_T for all eval_expr_typval() calls.
908 fc = eval_expr_get_funccal(expr, &newtv);
909
Bram Moolenaar04935fb2022-01-08 16:19:22 +0000910 ga_init2(&ga, sizeof(char), 80);
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000911 for (p = str; *p != NUL; p += len)
912 {
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000913 if (copy_first_char_to_tv(p, &tv) == FAIL)
914 break;
915 len = (int)STRLEN(tv.vval.v_string);
916
Bram Moolenaardd7eff02022-05-06 11:02:05 +0100917 newtv.v_type = VAR_UNKNOWN;
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000918 set_vim_var_nr(VV_KEY, idx);
Bram Moolenaar82418262022-09-28 16:16:15 +0100919 if (filter_map_one(&tv, expr, filtermap, fc, &newtv, &rem) == FAIL
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000920 || did_emsg)
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000921 {
922 clear_tv(&newtv);
923 clear_tv(&tv);
924 break;
925 }
926 else if (filtermap != FILTERMAP_FILTER)
927 {
928 if (newtv.v_type != VAR_STRING)
929 {
930 clear_tv(&newtv);
931 clear_tv(&tv);
Bram Moolenaare70cec92022-01-01 14:25:55 +0000932 emsg(_(e_string_required));
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000933 break;
934 }
935 else
936 ga_concat(&ga, newtv.vval.v_string);
937 }
938 else if (!rem)
939 ga_concat(&ga, tv.vval.v_string);
940
941 clear_tv(&newtv);
942 clear_tv(&tv);
943
944 ++idx;
945 }
946 ga_append(&ga, NUL);
947 rettv->vval.v_string = ga.ga_data;
Bram Moolenaar82418262022-09-28 16:16:15 +0100948 if (fc != NULL)
949 remove_funccal();
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000950}
951
952/*
Bram Moolenaarf1c60d42022-09-22 17:07:00 +0100953 * Implementation of reduce() for String "argvars[0]" using the function "expr"
954 * starting with the optional initial value "argvars[2]" and return the result
955 * in "rettv".
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000956 */
957 void
958string_reduce(
959 typval_T *argvars,
Bram Moolenaarf1c60d42022-09-22 17:07:00 +0100960 typval_T *expr,
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000961 typval_T *rettv)
962{
963 char_u *p = tv_get_string(&argvars[0]);
964 int len;
965 typval_T argv[3];
966 int r;
967 int called_emsg_start = called_emsg;
Bram Moolenaar82418262022-09-28 16:16:15 +0100968 funccall_T *fc;
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000969
970 if (argvars[2].v_type == VAR_UNKNOWN)
971 {
972 if (*p == NUL)
973 {
Bram Moolenaare70cec92022-01-01 14:25:55 +0000974 semsg(_(e_reduce_of_an_empty_str_with_no_initial_value), "String");
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000975 return;
976 }
977 if (copy_first_char_to_tv(p, rettv) == FAIL)
978 return;
979 p += STRLEN(rettv->vval.v_string);
980 }
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +0100981 else if (check_for_string_arg(argvars, 2) == FAIL)
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000982 return;
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000983 else
984 copy_tv(&argvars[2], rettv);
985
Bram Moolenaar82418262022-09-28 16:16:15 +0100986 // Create one funccal_T for all eval_expr_typval() calls.
987 fc = eval_expr_get_funccal(expr, rettv);
988
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000989 for ( ; *p != NUL; p += len)
990 {
991 argv[0] = *rettv;
992 if (copy_first_char_to_tv(p, &argv[1]) == FAIL)
993 break;
994 len = (int)STRLEN(argv[1].vval.v_string);
Bram Moolenaarf1c60d42022-09-22 17:07:00 +0100995
Bram Moolenaar82418262022-09-28 16:16:15 +0100996 r = eval_expr_typval(expr, argv, 2, fc, rettv);
Bram Moolenaarf1c60d42022-09-22 17:07:00 +0100997
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +0000998 clear_tv(&argv[0]);
999 clear_tv(&argv[1]);
1000 if (r == FAIL || called_emsg != called_emsg_start)
1001 return;
1002 }
Bram Moolenaar82418262022-09-28 16:16:15 +01001003
1004 if (fc != NULL)
1005 remove_funccal();
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +00001006}
1007
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001008 static void
1009byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
1010{
1011 char_u *t;
1012 char_u *str;
1013 varnumber_T idx;
1014
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001015 rettv->vval.v_number = -1;
1016
1017 if (in_vim9script()
1018 && (check_for_string_arg(argvars, 0) == FAIL
1019 || check_for_number_arg(argvars, 1) == FAIL))
1020 return;
1021
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001022 str = tv_get_string_chk(&argvars[0]);
1023 idx = tv_get_number_chk(&argvars[1], NULL);
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001024 if (str == NULL || idx < 0)
1025 return;
1026
1027 t = str;
1028 for ( ; idx > 0; idx--)
1029 {
1030 if (*t == NUL) // EOL reached
1031 return;
1032 if (enc_utf8 && comp)
1033 t += utf_ptr2len(t);
1034 else
1035 t += (*mb_ptr2len)(t);
1036 }
1037 rettv->vval.v_number = (varnumber_T)(t - str);
1038}
1039
1040/*
1041 * "byteidx()" function
1042 */
1043 void
1044f_byteidx(typval_T *argvars, typval_T *rettv)
1045{
1046 byteidx(argvars, rettv, FALSE);
1047}
1048
1049/*
1050 * "byteidxcomp()" function
1051 */
1052 void
1053f_byteidxcomp(typval_T *argvars, typval_T *rettv)
1054{
1055 byteidx(argvars, rettv, TRUE);
1056}
1057
1058/*
1059 * "charidx()" function
1060 */
1061 void
1062f_charidx(typval_T *argvars, typval_T *rettv)
1063{
1064 char_u *str;
1065 varnumber_T idx;
1066 varnumber_T countcc = FALSE;
1067 char_u *p;
1068 int len;
1069 int (*ptr2len)(char_u *);
1070
1071 rettv->vval.v_number = -1;
1072
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001073 if ((check_for_string_arg(argvars, 0) == FAIL
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001074 || check_for_number_arg(argvars, 1) == FAIL
1075 || check_for_opt_bool_arg(argvars, 2) == FAIL))
1076 return;
1077
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001078 str = tv_get_string_chk(&argvars[0]);
1079 idx = tv_get_number_chk(&argvars[1], NULL);
1080 if (str == NULL || idx < 0)
1081 return;
1082
1083 if (argvars[2].v_type != VAR_UNKNOWN)
1084 countcc = tv_get_bool(&argvars[2]);
1085 if (countcc < 0 || countcc > 1)
1086 {
1087 semsg(_(e_using_number_as_bool_nr), countcc);
1088 return;
1089 }
1090
1091 if (enc_utf8 && countcc)
1092 ptr2len = utf_ptr2len;
1093 else
1094 ptr2len = mb_ptr2len;
1095
1096 for (p = str, len = 0; p <= str + idx; len++)
1097 {
1098 if (*p == NUL)
1099 return;
1100 p += ptr2len(p);
1101 }
1102
1103 rettv->vval.v_number = len > 0 ? len - 1 : 0;
1104}
1105
1106/*
1107 * "str2list()" function
1108 */
1109 void
1110f_str2list(typval_T *argvars, typval_T *rettv)
1111{
1112 char_u *p;
1113 int utf8 = FALSE;
1114
1115 if (rettv_list_alloc(rettv) == FAIL)
1116 return;
1117
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02001118 if (in_vim9script()
1119 && (check_for_string_arg(argvars, 0) == FAIL
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001120 || check_for_opt_bool_arg(argvars, 1) == FAIL))
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02001121 return;
1122
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001123 if (argvars[1].v_type != VAR_UNKNOWN)
1124 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
1125
1126 p = tv_get_string(&argvars[0]);
1127
1128 if (has_mbyte || utf8)
1129 {
1130 int (*ptr2len)(char_u *);
1131 int (*ptr2char)(char_u *);
1132
1133 if (utf8 || enc_utf8)
1134 {
1135 ptr2len = utf_ptr2len;
1136 ptr2char = utf_ptr2char;
1137 }
1138 else
1139 {
1140 ptr2len = mb_ptr2len;
1141 ptr2char = mb_ptr2char;
1142 }
1143
1144 for ( ; *p != NUL; p += (*ptr2len)(p))
1145 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
1146 }
1147 else
1148 for ( ; *p != NUL; ++p)
1149 list_append_number(rettv->vval.v_list, *p);
1150}
1151
1152/*
1153 * "str2nr()" function
1154 */
1155 void
1156f_str2nr(typval_T *argvars, typval_T *rettv)
1157{
1158 int base = 10;
1159 char_u *p;
1160 varnumber_T n;
1161 int what = 0;
1162 int isneg;
1163
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001164 if (in_vim9script()
1165 && (check_for_string_arg(argvars, 0) == FAIL
1166 || check_for_opt_number_arg(argvars, 1) == FAIL
1167 || (argvars[1].v_type != VAR_UNKNOWN
1168 && check_for_opt_bool_arg(argvars, 2) == FAIL)))
1169 return;
1170
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001171 if (argvars[1].v_type != VAR_UNKNOWN)
1172 {
1173 base = (int)tv_get_number(&argvars[1]);
1174 if (base != 2 && base != 8 && base != 10 && base != 16)
1175 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001176 emsg(_(e_invalid_argument));
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001177 return;
1178 }
1179 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[2]))
1180 what |= STR2NR_QUOTE;
1181 }
1182
1183 p = skipwhite(tv_get_string_strict(&argvars[0]));
1184 isneg = (*p == '-');
1185 if (*p == '+' || *p == '-')
1186 p = skipwhite(p + 1);
1187 switch (base)
1188 {
1189 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
1190 case 8: what |= STR2NR_OCT + STR2NR_OOCT + STR2NR_FORCE; break;
1191 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
1192 }
1193 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
1194 // Text after the number is silently ignored.
1195 if (isneg)
1196 rettv->vval.v_number = -n;
1197 else
1198 rettv->vval.v_number = n;
1199
1200}
1201
1202/*
1203 * "strgetchar()" function
1204 */
1205 void
1206f_strgetchar(typval_T *argvars, typval_T *rettv)
1207{
1208 char_u *str;
1209 int len;
1210 int error = FALSE;
1211 int charidx;
1212 int byteidx = 0;
1213
1214 rettv->vval.v_number = -1;
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001215
1216 if (in_vim9script()
1217 && (check_for_string_arg(argvars, 0) == FAIL
1218 || check_for_number_arg(argvars, 1) == FAIL))
1219 return;
1220
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001221 str = tv_get_string_chk(&argvars[0]);
1222 if (str == NULL)
1223 return;
1224 len = (int)STRLEN(str);
1225 charidx = (int)tv_get_number_chk(&argvars[1], &error);
1226 if (error)
1227 return;
1228
1229 while (charidx >= 0 && byteidx < len)
1230 {
1231 if (charidx == 0)
1232 {
1233 rettv->vval.v_number = mb_ptr2char(str + byteidx);
1234 break;
1235 }
1236 --charidx;
1237 byteidx += MB_CPTR2LEN(str + byteidx);
1238 }
1239}
1240
1241/*
1242 * "stridx()" function
1243 */
1244 void
1245f_stridx(typval_T *argvars, typval_T *rettv)
1246{
1247 char_u buf[NUMBUFLEN];
1248 char_u *needle;
1249 char_u *haystack;
1250 char_u *save_haystack;
1251 char_u *pos;
1252 int start_idx;
1253
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001254 if (in_vim9script()
1255 && (check_for_string_arg(argvars, 0) == FAIL
1256 || check_for_string_arg(argvars, 1) == FAIL
1257 || check_for_opt_number_arg(argvars, 2) == FAIL))
1258 return;
1259
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001260 needle = tv_get_string_chk(&argvars[1]);
1261 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
1262 rettv->vval.v_number = -1;
1263 if (needle == NULL || haystack == NULL)
1264 return; // type error; errmsg already given
1265
1266 if (argvars[2].v_type != VAR_UNKNOWN)
1267 {
1268 int error = FALSE;
1269
1270 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
1271 if (error || start_idx >= (int)STRLEN(haystack))
1272 return;
1273 if (start_idx >= 0)
1274 haystack += start_idx;
1275 }
1276
1277 pos = (char_u *)strstr((char *)haystack, (char *)needle);
1278 if (pos != NULL)
1279 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
1280}
1281
1282/*
1283 * "string()" function
1284 */
1285 void
1286f_string(typval_T *argvars, typval_T *rettv)
1287{
1288 char_u *tofree;
1289 char_u numbuf[NUMBUFLEN];
1290
1291 rettv->v_type = VAR_STRING;
1292 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
1293 get_copyID());
1294 // Make a copy if we have a value but it's not in allocated memory.
1295 if (rettv->vval.v_string != NULL && tofree == NULL)
1296 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
1297}
1298
1299/*
1300 * "strlen()" function
1301 */
1302 void
1303f_strlen(typval_T *argvars, typval_T *rettv)
1304{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001305 if (in_vim9script()
1306 && check_for_string_or_number_arg(argvars, 0) == FAIL)
1307 return;
1308
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001309 rettv->vval.v_number = (varnumber_T)(STRLEN(
1310 tv_get_string(&argvars[0])));
1311}
1312
1313 static void
1314strchar_common(typval_T *argvars, typval_T *rettv, int skipcc)
1315{
1316 char_u *s = tv_get_string(&argvars[0]);
1317 varnumber_T len = 0;
1318 int (*func_mb_ptr2char_adv)(char_u **pp);
1319
1320 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
1321 while (*s != NUL)
1322 {
1323 func_mb_ptr2char_adv(&s);
1324 ++len;
1325 }
1326 rettv->vval.v_number = len;
1327}
1328
1329/*
1330 * "strcharlen()" function
1331 */
1332 void
1333f_strcharlen(typval_T *argvars, typval_T *rettv)
1334{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001335 if (in_vim9script()
1336 && check_for_string_or_number_arg(argvars, 0) == FAIL)
1337 return;
1338
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001339 strchar_common(argvars, rettv, TRUE);
1340}
1341
1342/*
1343 * "strchars()" function
1344 */
1345 void
1346f_strchars(typval_T *argvars, typval_T *rettv)
1347{
1348 varnumber_T skipcc = FALSE;
1349
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02001350 if (in_vim9script()
1351 && (check_for_string_arg(argvars, 0) == FAIL
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001352 || check_for_opt_bool_arg(argvars, 1) == FAIL))
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02001353 return;
1354
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001355 if (argvars[1].v_type != VAR_UNKNOWN)
1356 skipcc = tv_get_bool(&argvars[1]);
1357 if (skipcc < 0 || skipcc > 1)
1358 semsg(_(e_using_number_as_bool_nr), skipcc);
1359 else
1360 strchar_common(argvars, rettv, skipcc);
1361}
1362
1363/*
1364 * "strdisplaywidth()" function
1365 */
1366 void
1367f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
1368{
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001369 char_u *s;
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001370 int col = 0;
1371
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001372 rettv->vval.v_number = -1;
1373
1374 if (in_vim9script()
1375 && (check_for_string_arg(argvars, 0) == FAIL
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001376 || check_for_opt_number_arg(argvars, 1) == FAIL))
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001377 return;
1378
1379 s = tv_get_string(&argvars[0]);
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001380 if (argvars[1].v_type != VAR_UNKNOWN)
1381 col = (int)tv_get_number(&argvars[1]);
1382
1383 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
1384}
1385
1386/*
1387 * "strwidth()" function
1388 */
1389 void
1390f_strwidth(typval_T *argvars, typval_T *rettv)
1391{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001392 char_u *s;
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001393
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001394 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
1395 return;
1396
1397 s = tv_get_string_strict(&argvars[0]);
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001398 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
1399}
1400
1401/*
1402 * "strcharpart()" function
1403 */
1404 void
1405f_strcharpart(typval_T *argvars, typval_T *rettv)
1406{
1407 char_u *p;
1408 int nchar;
1409 int nbyte = 0;
1410 int charlen;
1411 int skipcc = FALSE;
1412 int len = 0;
1413 int slen;
1414 int error = FALSE;
1415
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001416 if (in_vim9script()
1417 && (check_for_string_arg(argvars, 0) == FAIL
1418 || check_for_number_arg(argvars, 1) == FAIL
1419 || check_for_opt_number_arg(argvars, 2) == FAIL
1420 || (argvars[2].v_type != VAR_UNKNOWN
1421 && check_for_opt_bool_arg(argvars, 3) == FAIL)))
1422 return;
1423
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001424 p = tv_get_string(&argvars[0]);
1425 slen = (int)STRLEN(p);
1426
1427 nchar = (int)tv_get_number_chk(&argvars[1], &error);
1428 if (!error)
1429 {
1430 if (argvars[2].v_type != VAR_UNKNOWN
1431 && argvars[3].v_type != VAR_UNKNOWN)
1432 {
1433 skipcc = tv_get_bool(&argvars[3]);
1434 if (skipcc < 0 || skipcc > 1)
1435 {
1436 semsg(_(e_using_number_as_bool_nr), skipcc);
1437 return;
1438 }
1439 }
1440
1441 if (nchar > 0)
1442 while (nchar > 0 && nbyte < slen)
1443 {
1444 if (skipcc)
1445 nbyte += mb_ptr2len(p + nbyte);
1446 else
1447 nbyte += MB_CPTR2LEN(p + nbyte);
1448 --nchar;
1449 }
1450 else
1451 nbyte = nchar;
1452 if (argvars[2].v_type != VAR_UNKNOWN)
1453 {
1454 charlen = (int)tv_get_number(&argvars[2]);
1455 while (charlen > 0 && nbyte + len < slen)
1456 {
1457 int off = nbyte + len;
1458
1459 if (off < 0)
1460 len += 1;
1461 else
1462 {
1463 if (skipcc)
1464 len += mb_ptr2len(p + off);
1465 else
1466 len += MB_CPTR2LEN(p + off);
1467 }
1468 --charlen;
1469 }
1470 }
1471 else
1472 len = slen - nbyte; // default: all bytes that are available.
1473 }
1474
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02001475 // Only return the overlap between the specified part and the actual
1476 // string.
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001477 if (nbyte < 0)
1478 {
1479 len += nbyte;
1480 nbyte = 0;
1481 }
1482 else if (nbyte > slen)
1483 nbyte = slen;
1484 if (len < 0)
1485 len = 0;
1486 else if (nbyte + len > slen)
1487 len = slen - nbyte;
1488
1489 rettv->v_type = VAR_STRING;
1490 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
1491}
1492
1493/*
1494 * "strpart()" function
1495 */
1496 void
1497f_strpart(typval_T *argvars, typval_T *rettv)
1498{
1499 char_u *p;
1500 int n;
1501 int len;
1502 int slen;
1503 int error = FALSE;
1504
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001505 if (in_vim9script()
1506 && (check_for_string_arg(argvars, 0) == FAIL
1507 || check_for_number_arg(argvars, 1) == FAIL
1508 || check_for_opt_number_arg(argvars, 2) == FAIL
1509 || (argvars[2].v_type != VAR_UNKNOWN
1510 && check_for_opt_bool_arg(argvars, 3) == FAIL)))
1511 return;
1512
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001513 p = tv_get_string(&argvars[0]);
1514 slen = (int)STRLEN(p);
1515
1516 n = (int)tv_get_number_chk(&argvars[1], &error);
1517 if (error)
1518 len = 0;
1519 else if (argvars[2].v_type != VAR_UNKNOWN)
1520 len = (int)tv_get_number(&argvars[2]);
1521 else
1522 len = slen - n; // default len: all bytes that are available.
1523
1524 // Only return the overlap between the specified part and the actual
1525 // string.
1526 if (n < 0)
1527 {
1528 len += n;
1529 n = 0;
1530 }
1531 else if (n > slen)
1532 n = slen;
1533 if (len < 0)
1534 len = 0;
1535 else if (n + len > slen)
1536 len = slen - n;
1537
1538 if (argvars[2].v_type != VAR_UNKNOWN && argvars[3].v_type != VAR_UNKNOWN)
1539 {
1540 int off;
1541
1542 // length in characters
1543 for (off = n; off < slen && len > 0; --len)
1544 off += mb_ptr2len(p + off);
1545 len = off - n;
1546 }
1547
1548 rettv->v_type = VAR_STRING;
1549 rettv->vval.v_string = vim_strnsave(p + n, len);
1550}
1551
1552/*
1553 * "strridx()" function
1554 */
1555 void
1556f_strridx(typval_T *argvars, typval_T *rettv)
1557{
1558 char_u buf[NUMBUFLEN];
1559 char_u *needle;
1560 char_u *haystack;
1561 char_u *rest;
1562 char_u *lastmatch = NULL;
1563 int haystack_len, end_idx;
1564
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001565 if (in_vim9script()
1566 && (check_for_string_arg(argvars, 0) == FAIL
1567 || check_for_string_arg(argvars, 1) == FAIL
1568 || check_for_opt_number_arg(argvars, 2) == FAIL))
1569 return;
1570
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001571 needle = tv_get_string_chk(&argvars[1]);
1572 haystack = tv_get_string_buf_chk(&argvars[0], buf);
1573
1574 rettv->vval.v_number = -1;
1575 if (needle == NULL || haystack == NULL)
1576 return; // type error; errmsg already given
1577
1578 haystack_len = (int)STRLEN(haystack);
1579 if (argvars[2].v_type != VAR_UNKNOWN)
1580 {
1581 // Third argument: upper limit for index
1582 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
1583 if (end_idx < 0)
1584 return; // can never find a match
1585 }
1586 else
1587 end_idx = haystack_len;
1588
1589 if (*needle == NUL)
1590 {
1591 // Empty string matches past the end.
1592 lastmatch = haystack + end_idx;
1593 }
1594 else
1595 {
1596 for (rest = haystack; *rest != '\0'; ++rest)
1597 {
1598 rest = (char_u *)strstr((char *)rest, (char *)needle);
1599 if (rest == NULL || rest > haystack + end_idx)
1600 break;
1601 lastmatch = rest;
1602 }
1603 }
1604
1605 if (lastmatch == NULL)
1606 rettv->vval.v_number = -1;
1607 else
1608 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
1609}
1610
1611/*
1612 * "strtrans()" function
1613 */
1614 void
1615f_strtrans(typval_T *argvars, typval_T *rettv)
1616{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001617 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
1618 return;
1619
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001620 rettv->v_type = VAR_STRING;
1621 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
1622}
1623
1624/*
1625 * "tolower(string)" function
1626 */
1627 void
1628f_tolower(typval_T *argvars, typval_T *rettv)
1629{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001630 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
1631 return;
1632
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001633 rettv->v_type = VAR_STRING;
1634 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
1635}
1636
1637/*
1638 * "toupper(string)" function
1639 */
1640 void
1641f_toupper(typval_T *argvars, typval_T *rettv)
1642{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001643 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
1644 return;
1645
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001646 rettv->v_type = VAR_STRING;
1647 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
1648}
1649
1650/*
1651 * "tr(string, fromstr, tostr)" function
1652 */
1653 void
1654f_tr(typval_T *argvars, typval_T *rettv)
1655{
1656 char_u *in_str;
1657 char_u *fromstr;
1658 char_u *tostr;
1659 char_u *p;
1660 int inlen;
1661 int fromlen;
1662 int tolen;
1663 int idx;
1664 char_u *cpstr;
1665 int cplen;
1666 int first = TRUE;
1667 char_u buf[NUMBUFLEN];
1668 char_u buf2[NUMBUFLEN];
1669 garray_T ga;
1670
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001671 if (in_vim9script()
1672 && (check_for_string_arg(argvars, 0) == FAIL
1673 || check_for_string_arg(argvars, 1) == FAIL
1674 || check_for_string_arg(argvars, 2) == FAIL))
1675 return;
1676
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001677 in_str = tv_get_string(&argvars[0]);
1678 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
1679 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
1680
1681 // Default return value: empty string.
1682 rettv->v_type = VAR_STRING;
1683 rettv->vval.v_string = NULL;
1684 if (fromstr == NULL || tostr == NULL)
1685 return; // type error; errmsg already given
Bram Moolenaar04935fb2022-01-08 16:19:22 +00001686 ga_init2(&ga, sizeof(char), 80);
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001687
1688 if (!has_mbyte)
1689 // not multi-byte: fromstr and tostr must be the same length
1690 if (STRLEN(fromstr) != STRLEN(tostr))
1691 {
1692error:
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001693 semsg(_(e_invalid_argument_str), fromstr);
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001694 ga_clear(&ga);
1695 return;
1696 }
1697
1698 // fromstr and tostr have to contain the same number of chars
1699 while (*in_str != NUL)
1700 {
1701 if (has_mbyte)
1702 {
1703 inlen = (*mb_ptr2len)(in_str);
1704 cpstr = in_str;
1705 cplen = inlen;
1706 idx = 0;
1707 for (p = fromstr; *p != NUL; p += fromlen)
1708 {
1709 fromlen = (*mb_ptr2len)(p);
1710 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
1711 {
1712 for (p = tostr; *p != NUL; p += tolen)
1713 {
1714 tolen = (*mb_ptr2len)(p);
1715 if (idx-- == 0)
1716 {
1717 cplen = tolen;
1718 cpstr = p;
1719 break;
1720 }
1721 }
1722 if (*p == NUL) // tostr is shorter than fromstr
1723 goto error;
1724 break;
1725 }
1726 ++idx;
1727 }
1728
1729 if (first && cpstr == in_str)
1730 {
1731 // Check that fromstr and tostr have the same number of
1732 // (multi-byte) characters. Done only once when a character
1733 // of in_str doesn't appear in fromstr.
1734 first = FALSE;
1735 for (p = tostr; *p != NUL; p += tolen)
1736 {
1737 tolen = (*mb_ptr2len)(p);
1738 --idx;
1739 }
1740 if (idx != 0)
1741 goto error;
1742 }
1743
1744 (void)ga_grow(&ga, cplen);
1745 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
1746 ga.ga_len += cplen;
1747
1748 in_str += inlen;
1749 }
1750 else
1751 {
1752 // When not using multi-byte chars we can do it faster.
1753 p = vim_strchr(fromstr, *in_str);
1754 if (p != NULL)
1755 ga_append(&ga, tostr[p - fromstr]);
1756 else
1757 ga_append(&ga, *in_str);
1758 ++in_str;
1759 }
1760 }
1761
1762 // add a terminating NUL
1763 (void)ga_grow(&ga, 1);
1764 ga_append(&ga, NUL);
1765
1766 rettv->vval.v_string = ga.ga_data;
1767}
1768
1769/*
1770 * "trim({expr})" function
1771 */
1772 void
1773f_trim(typval_T *argvars, typval_T *rettv)
1774{
1775 char_u buf1[NUMBUFLEN];
1776 char_u buf2[NUMBUFLEN];
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001777 char_u *head;
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001778 char_u *mask = NULL;
1779 char_u *tail;
1780 char_u *prev;
1781 char_u *p;
1782 int c1;
1783 int dir = 0;
1784
1785 rettv->v_type = VAR_STRING;
1786 rettv->vval.v_string = NULL;
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001787
1788 if (in_vim9script()
1789 && (check_for_string_arg(argvars, 0) == FAIL
1790 || check_for_opt_string_arg(argvars, 1) == FAIL
1791 || (argvars[1].v_type != VAR_UNKNOWN
1792 && check_for_opt_number_arg(argvars, 2) == FAIL)))
1793 return;
1794
1795 head = tv_get_string_buf_chk(&argvars[0], buf1);
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001796 if (head == NULL)
1797 return;
1798
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001799 if (check_for_opt_string_arg(argvars, 1) == FAIL)
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001800 return;
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001801
1802 if (argvars[1].v_type == VAR_STRING)
1803 {
1804 mask = tv_get_string_buf_chk(&argvars[1], buf2);
1805
1806 if (argvars[2].v_type != VAR_UNKNOWN)
1807 {
1808 int error = 0;
1809
1810 // leading or trailing characters to trim
1811 dir = (int)tv_get_number_chk(&argvars[2], &error);
1812 if (error)
1813 return;
1814 if (dir < 0 || dir > 2)
1815 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001816 semsg(_(e_invalid_argument_str), tv_get_string(&argvars[2]));
Yegappan Lakshmanana2438132021-07-10 21:29:18 +02001817 return;
1818 }
1819 }
1820 }
1821
1822 if (dir == 0 || dir == 1)
1823 {
1824 // Trim leading characters
1825 while (*head != NUL)
1826 {
1827 c1 = PTR2CHAR(head);
1828 if (mask == NULL)
1829 {
1830 if (c1 > ' ' && c1 != 0xa0)
1831 break;
1832 }
1833 else
1834 {
1835 for (p = mask; *p != NUL; MB_PTR_ADV(p))
1836 if (c1 == PTR2CHAR(p))
1837 break;
1838 if (*p == NUL)
1839 break;
1840 }
1841 MB_PTR_ADV(head);
1842 }
1843 }
1844
1845 tail = head + STRLEN(head);
1846 if (dir == 0 || dir == 2)
1847 {
1848 // Trim trailing characters
1849 for (; tail > head; tail = prev)
1850 {
1851 prev = tail;
1852 MB_PTR_BACK(head, prev);
1853 c1 = PTR2CHAR(prev);
1854 if (mask == NULL)
1855 {
1856 if (c1 > ' ' && c1 != 0xa0)
1857 break;
1858 }
1859 else
1860 {
1861 for (p = mask; *p != NUL; MB_PTR_ADV(p))
1862 if (c1 == PTR2CHAR(p))
1863 break;
1864 if (*p == NUL)
1865 break;
1866 }
1867 }
1868 }
1869 rettv->vval.v_string = vim_strnsave(head, tail - head);
1870}
1871
Bram Moolenaar677658a2022-01-05 16:09:06 +00001872static char *e_printf = N_(e_insufficient_arguments_for_printf);
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02001873
1874/*
1875 * Get number argument from "idxp" entry in "tvs". First entry is 1.
1876 */
1877 static varnumber_T
1878tv_nr(typval_T *tvs, int *idxp)
1879{
1880 int idx = *idxp - 1;
1881 varnumber_T n = 0;
1882 int err = FALSE;
1883
1884 if (tvs[idx].v_type == VAR_UNKNOWN)
1885 emsg(_(e_printf));
1886 else
1887 {
1888 ++*idxp;
1889 n = tv_get_number_chk(&tvs[idx], &err);
1890 if (err)
1891 n = 0;
1892 }
1893 return n;
1894}
1895
1896/*
1897 * Get string argument from "idxp" entry in "tvs". First entry is 1.
1898 * If "tofree" is NULL tv_get_string_chk() is used. Some types (e.g. List)
1899 * are not converted to a string.
1900 * If "tofree" is not NULL echo_string() is used. All types are converted to
1901 * a string with the same format as ":echo". The caller must free "*tofree".
1902 * Returns NULL for an error.
1903 */
1904 static char *
1905tv_str(typval_T *tvs, int *idxp, char_u **tofree)
1906{
1907 int idx = *idxp - 1;
1908 char *s = NULL;
1909 static char_u numbuf[NUMBUFLEN];
1910
1911 if (tvs[idx].v_type == VAR_UNKNOWN)
1912 emsg(_(e_printf));
1913 else
1914 {
1915 ++*idxp;
1916 if (tofree != NULL)
1917 s = (char *)echo_string(&tvs[idx], tofree, numbuf, get_copyID());
1918 else
1919 s = (char *)tv_get_string_chk(&tvs[idx]);
1920 }
1921 return s;
1922}
1923
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02001924/*
1925 * Get float argument from "idxp" entry in "tvs". First entry is 1.
1926 */
1927 static double
1928tv_float(typval_T *tvs, int *idxp)
1929{
1930 int idx = *idxp - 1;
1931 double f = 0;
1932
1933 if (tvs[idx].v_type == VAR_UNKNOWN)
1934 emsg(_(e_printf));
1935 else
1936 {
1937 ++*idxp;
1938 if (tvs[idx].v_type == VAR_FLOAT)
1939 f = tvs[idx].vval.v_float;
1940 else if (tvs[idx].v_type == VAR_NUMBER)
1941 f = (double)tvs[idx].vval.v_number;
1942 else
Bram Moolenaar9d00e4a2022-01-05 17:49:15 +00001943 emsg(_(e_expected_float_argument_for_printf));
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02001944 }
1945 return f;
1946}
Yegappan Lakshmananf973eeb2021-12-22 18:19:26 +00001947
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02001948#endif
1949
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02001950/*
1951 * Return the representation of infinity for printf() function:
1952 * "-inf", "inf", "+inf", " inf", "-INF", "INF", "+INF" or " INF".
1953 */
1954 static const char *
1955infinity_str(int positive,
1956 char fmt_spec,
1957 int force_sign,
1958 int space_for_positive)
1959{
1960 static const char *table[] =
1961 {
1962 "-inf", "inf", "+inf", " inf",
1963 "-INF", "INF", "+INF", " INF"
1964 };
1965 int idx = positive * (1 + force_sign + force_sign * space_for_positive);
1966
1967 if (ASCII_ISUPPER(fmt_spec))
1968 idx += 4;
1969 return table[idx];
1970}
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02001971
1972/*
1973 * This code was included to provide a portable vsnprintf() and snprintf().
1974 * Some systems may provide their own, but we always use this one for
1975 * consistency.
1976 *
1977 * This code is based on snprintf.c - a portable implementation of snprintf
1978 * by Mark Martinec <mark.martinec@ijs.si>, Version 2.2, 2000-10-06.
1979 * Included with permission. It was heavily modified to fit in Vim.
1980 * The original code, including useful comments, can be found here:
1981 * http://www.ijs.si/software/snprintf/
1982 *
1983 * This snprintf() only supports the following conversion specifiers:
1984 * s, c, d, u, o, x, X, p (and synonyms: i, D, U, O - see below)
1985 * with flags: '-', '+', ' ', '0' and '#'.
1986 * An asterisk is supported for field width as well as precision.
1987 *
1988 * Limited support for floating point was added: 'f', 'F', 'e', 'E', 'g', 'G'.
1989 *
1990 * Length modifiers 'h' (short int) and 'l' (long int) and 'll' (long long int)
1991 * are supported. NOTE: for 'll' the argument is varnumber_T or uvarnumber_T.
1992 *
1993 * The locale is not used, the string is used as a byte string. This is only
1994 * relevant for double-byte encodings where the second byte may be '%'.
1995 *
1996 * It is permitted for "str_m" to be zero, and it is permitted to specify NULL
1997 * pointer for resulting string argument if "str_m" is zero (as per ISO C99).
1998 *
1999 * The return value is the number of characters which would be generated
2000 * for the given input, excluding the trailing NUL. If this value
2001 * is greater or equal to "str_m", not all characters from the result
2002 * have been stored in str, output bytes beyond the ("str_m"-1) -th character
2003 * are discarded. If "str_m" is greater than zero it is guaranteed
2004 * the resulting string will be NUL-terminated.
2005 */
2006
2007/*
2008 * When va_list is not supported we only define vim_snprintf().
2009 *
2010 * vim_vsnprintf_typval() can be invoked with either "va_list" or a list of
2011 * "typval_T". When the latter is not used it must be NULL.
2012 */
2013
2014// When generating prototypes all of this is skipped, cproto doesn't
2015// understand this.
2016#ifndef PROTO
2017
2018// Like vim_vsnprintf() but append to the string.
2019 int
2020vim_snprintf_add(char *str, size_t str_m, const char *fmt, ...)
2021{
2022 va_list ap;
2023 int str_l;
2024 size_t len = STRLEN(str);
2025 size_t space;
2026
2027 if (str_m <= len)
2028 space = 0;
2029 else
2030 space = str_m - len;
2031 va_start(ap, fmt);
2032 str_l = vim_vsnprintf(str + len, space, fmt, ap);
2033 va_end(ap);
2034 return str_l;
2035}
2036
2037 int
2038vim_snprintf(char *str, size_t str_m, const char *fmt, ...)
2039{
2040 va_list ap;
2041 int str_l;
2042
2043 va_start(ap, fmt);
2044 str_l = vim_vsnprintf(str, str_m, fmt, ap);
2045 va_end(ap);
2046 return str_l;
2047}
2048
2049 int
2050vim_vsnprintf(
2051 char *str,
2052 size_t str_m,
2053 const char *fmt,
2054 va_list ap)
2055{
2056 return vim_vsnprintf_typval(str, str_m, fmt, ap, NULL);
2057}
2058
2059 int
2060vim_vsnprintf_typval(
2061 char *str,
2062 size_t str_m,
2063 const char *fmt,
2064 va_list ap,
2065 typval_T *tvs)
2066{
2067 size_t str_l = 0;
2068 const char *p = fmt;
2069 int arg_idx = 1;
2070
2071 if (p == NULL)
2072 p = "";
2073 while (*p != NUL)
2074 {
2075 if (*p != '%')
2076 {
2077 char *q = strchr(p + 1, '%');
2078 size_t n = (q == NULL) ? STRLEN(p) : (size_t)(q - p);
2079
2080 // Copy up to the next '%' or NUL without any changes.
2081 if (str_l < str_m)
2082 {
2083 size_t avail = str_m - str_l;
2084
2085 mch_memmove(str + str_l, p, n > avail ? avail : n);
2086 }
2087 p += n;
2088 str_l += n;
2089 }
2090 else
2091 {
2092 size_t min_field_width = 0, precision = 0;
2093 int zero_padding = 0, precision_specified = 0, justify_left = 0;
2094 int alternate_form = 0, force_sign = 0;
2095
2096 // If both the ' ' and '+' flags appear, the ' ' flag should be
2097 // ignored.
2098 int space_for_positive = 1;
2099
2100 // allowed values: \0, h, l, L
2101 char length_modifier = '\0';
2102
2103 // temporary buffer for simple numeric->string conversion
Bram Moolenaar73e28dc2022-09-17 21:08:33 +01002104# define TMP_LEN 350 // On my system 1e308 is the biggest number possible.
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002105 // That sounds reasonable to use as the maximum
2106 // printable.
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002107 char tmp[TMP_LEN];
2108
2109 // string address in case of string argument
2110 const char *str_arg = NULL;
2111
2112 // natural field width of arg without padding and sign
2113 size_t str_arg_l;
2114
2115 // unsigned char argument value - only defined for c conversion.
2116 // N.B. standard explicitly states the char argument for the c
2117 // conversion is unsigned
2118 unsigned char uchar_arg;
2119
2120 // number of zeros to be inserted for numeric conversions as
2121 // required by the precision or minimal field width
2122 size_t number_of_zeros_to_pad = 0;
2123
2124 // index into tmp where zero padding is to be inserted
2125 size_t zero_padding_insertion_ind = 0;
2126
2127 // current conversion specifier character
2128 char fmt_spec = '\0';
2129
2130 // buffer for 's' and 'S' specs
2131 char_u *tofree = NULL;
2132
2133
2134 p++; // skip '%'
2135
2136 // parse flags
2137 while (*p == '0' || *p == '-' || *p == '+' || *p == ' '
2138 || *p == '#' || *p == '\'')
2139 {
2140 switch (*p)
2141 {
2142 case '0': zero_padding = 1; break;
2143 case '-': justify_left = 1; break;
2144 case '+': force_sign = 1; space_for_positive = 0; break;
2145 case ' ': force_sign = 1;
2146 // If both the ' ' and '+' flags appear, the ' '
2147 // flag should be ignored
2148 break;
2149 case '#': alternate_form = 1; break;
2150 case '\'': break;
2151 }
2152 p++;
2153 }
2154 // If the '0' and '-' flags both appear, the '0' flag should be
2155 // ignored.
2156
2157 // parse field width
2158 if (*p == '*')
2159 {
2160 int j;
2161
2162 p++;
2163 j =
2164# if defined(FEAT_EVAL)
2165 tvs != NULL ? tv_nr(tvs, &arg_idx) :
2166# endif
2167 va_arg(ap, int);
2168 if (j >= 0)
2169 min_field_width = j;
2170 else
2171 {
2172 min_field_width = -j;
2173 justify_left = 1;
2174 }
2175 }
2176 else if (VIM_ISDIGIT((int)(*p)))
2177 {
2178 // size_t could be wider than unsigned int; make sure we treat
2179 // argument like common implementations do
2180 unsigned int uj = *p++ - '0';
2181
2182 while (VIM_ISDIGIT((int)(*p)))
2183 uj = 10 * uj + (unsigned int)(*p++ - '0');
2184 min_field_width = uj;
2185 }
2186
2187 // parse precision
2188 if (*p == '.')
2189 {
2190 p++;
2191 precision_specified = 1;
2192 if (*p == '*')
2193 {
2194 int j;
2195
2196 j =
2197# if defined(FEAT_EVAL)
2198 tvs != NULL ? tv_nr(tvs, &arg_idx) :
2199# endif
2200 va_arg(ap, int);
2201 p++;
2202 if (j >= 0)
2203 precision = j;
2204 else
2205 {
2206 precision_specified = 0;
2207 precision = 0;
2208 }
2209 }
2210 else if (VIM_ISDIGIT((int)(*p)))
2211 {
2212 // size_t could be wider than unsigned int; make sure we
2213 // treat argument like common implementations do
2214 unsigned int uj = *p++ - '0';
2215
2216 while (VIM_ISDIGIT((int)(*p)))
2217 uj = 10 * uj + (unsigned int)(*p++ - '0');
2218 precision = uj;
2219 }
2220 }
2221
2222 // parse 'h', 'l' and 'll' length modifiers
2223 if (*p == 'h' || *p == 'l')
2224 {
2225 length_modifier = *p;
2226 p++;
2227 if (length_modifier == 'l' && *p == 'l')
2228 {
2229 // double l = __int64 / varnumber_T
2230 length_modifier = 'L';
2231 p++;
2232 }
2233 }
2234 fmt_spec = *p;
2235
2236 // common synonyms:
2237 switch (fmt_spec)
2238 {
2239 case 'i': fmt_spec = 'd'; break;
2240 case 'D': fmt_spec = 'd'; length_modifier = 'l'; break;
2241 case 'U': fmt_spec = 'u'; length_modifier = 'l'; break;
2242 case 'O': fmt_spec = 'o'; length_modifier = 'l'; break;
2243 default: break;
2244 }
2245
2246# if defined(FEAT_EVAL)
2247 switch (fmt_spec)
2248 {
2249 case 'd': case 'u': case 'o': case 'x': case 'X':
2250 if (tvs != NULL && length_modifier == '\0')
2251 length_modifier = 'L';
2252 }
2253# endif
2254
2255 // get parameter value, do initial processing
2256 switch (fmt_spec)
2257 {
2258 // '%' and 'c' behave similar to 's' regarding flags and field
2259 // widths
2260 case '%':
2261 case 'c':
2262 case 's':
2263 case 'S':
2264 str_arg_l = 1;
2265 switch (fmt_spec)
2266 {
2267 case '%':
2268 str_arg = p;
2269 break;
2270
2271 case 'c':
2272 {
2273 int j;
2274
2275 j =
2276# if defined(FEAT_EVAL)
2277 tvs != NULL ? tv_nr(tvs, &arg_idx) :
2278# endif
2279 va_arg(ap, int);
2280 // standard demands unsigned char
2281 uchar_arg = (unsigned char)j;
2282 str_arg = (char *)&uchar_arg;
2283 break;
2284 }
2285
2286 case 's':
2287 case 'S':
2288 str_arg =
2289# if defined(FEAT_EVAL)
2290 tvs != NULL ? tv_str(tvs, &arg_idx, &tofree) :
2291# endif
2292 va_arg(ap, char *);
2293 if (str_arg == NULL)
2294 {
2295 str_arg = "[NULL]";
2296 str_arg_l = 6;
2297 }
2298 // make sure not to address string beyond the specified
2299 // precision !!!
2300 else if (!precision_specified)
2301 str_arg_l = strlen(str_arg);
2302 // truncate string if necessary as requested by precision
2303 else if (precision == 0)
2304 str_arg_l = 0;
2305 else
2306 {
2307 // Don't put the #if inside memchr(), it can be a
2308 // macro.
2309 // memchr on HP does not like n > 2^31 !!!
2310 char *q = memchr(str_arg, '\0',
2311 precision <= (size_t)0x7fffffffL ? precision
2312 : (size_t)0x7fffffffL);
presukud85fccd2021-11-20 19:38:31 +00002313
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002314 str_arg_l = (q == NULL) ? precision
2315 : (size_t)(q - str_arg);
2316 }
2317 if (fmt_spec == 'S')
2318 {
presuku1f2453f2021-11-24 15:32:57 +00002319 char_u *p1;
2320 size_t i;
2321 int cell;
presukud85fccd2021-11-20 19:38:31 +00002322
presuku1f2453f2021-11-24 15:32:57 +00002323 for (i = 0, p1 = (char_u *)str_arg; *p1;
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002324 p1 += mb_ptr2len(p1))
presuku1f2453f2021-11-24 15:32:57 +00002325 {
2326 cell = mb_ptr2cells(p1);
2327 if (precision_specified && i + cell > precision)
2328 break;
2329 i += cell;
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002330 }
presuku1f2453f2021-11-24 15:32:57 +00002331
2332 str_arg_l = p1 - (char_u *)str_arg;
presukud85fccd2021-11-20 19:38:31 +00002333 if (min_field_width != 0)
presuku1f2453f2021-11-24 15:32:57 +00002334 min_field_width += str_arg_l - i;
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002335 }
2336 break;
2337
2338 default:
2339 break;
2340 }
2341 break;
2342
2343 case 'd': case 'u':
2344 case 'b': case 'B':
2345 case 'o':
2346 case 'x': case 'X':
2347 case 'p':
2348 {
2349 // NOTE: the u, b, o, x, X and p conversion specifiers
2350 // imply the value is unsigned; d implies a signed
2351 // value
2352
2353 // 0 if numeric argument is zero (or if pointer is
2354 // NULL for 'p'), +1 if greater than zero (or nonzero
2355 // for unsigned arguments), -1 if negative (unsigned
2356 // argument is never negative)
2357 int arg_sign = 0;
2358
2359 // only set for length modifier h, or for no length
2360 // modifiers
2361 int int_arg = 0;
2362 unsigned int uint_arg = 0;
2363
2364 // only set for length modifier l
2365 long int long_arg = 0;
2366 unsigned long int ulong_arg = 0;
2367
2368 // only set for length modifier ll
2369 varnumber_T llong_arg = 0;
2370 uvarnumber_T ullong_arg = 0;
2371
2372 // only set for b conversion
2373 uvarnumber_T bin_arg = 0;
2374
2375 // pointer argument value -only defined for p
2376 // conversion
2377 void *ptr_arg = NULL;
2378
2379 if (fmt_spec == 'p')
2380 {
2381 length_modifier = '\0';
2382 ptr_arg =
2383# if defined(FEAT_EVAL)
2384 tvs != NULL ? (void *)tv_str(tvs, &arg_idx,
2385 NULL) :
2386# endif
2387 va_arg(ap, void *);
2388 if (ptr_arg != NULL)
2389 arg_sign = 1;
2390 }
2391 else if (fmt_spec == 'b' || fmt_spec == 'B')
2392 {
2393 bin_arg =
2394# if defined(FEAT_EVAL)
2395 tvs != NULL ?
2396 (uvarnumber_T)tv_nr(tvs, &arg_idx) :
2397# endif
2398 va_arg(ap, uvarnumber_T);
2399 if (bin_arg != 0)
2400 arg_sign = 1;
2401 }
2402 else if (fmt_spec == 'd')
2403 {
2404 // signed
2405 switch (length_modifier)
2406 {
2407 case '\0':
2408 case 'h':
2409 // char and short arguments are passed as int.
2410 int_arg =
2411# if defined(FEAT_EVAL)
2412 tvs != NULL ? tv_nr(tvs, &arg_idx) :
2413# endif
2414 va_arg(ap, int);
2415 if (int_arg > 0)
2416 arg_sign = 1;
2417 else if (int_arg < 0)
2418 arg_sign = -1;
2419 break;
2420 case 'l':
2421 long_arg =
2422# if defined(FEAT_EVAL)
2423 tvs != NULL ? tv_nr(tvs, &arg_idx) :
2424# endif
2425 va_arg(ap, long int);
2426 if (long_arg > 0)
2427 arg_sign = 1;
2428 else if (long_arg < 0)
2429 arg_sign = -1;
2430 break;
2431 case 'L':
2432 llong_arg =
2433# if defined(FEAT_EVAL)
2434 tvs != NULL ? tv_nr(tvs, &arg_idx) :
2435# endif
2436 va_arg(ap, varnumber_T);
2437 if (llong_arg > 0)
2438 arg_sign = 1;
2439 else if (llong_arg < 0)
2440 arg_sign = -1;
2441 break;
2442 }
2443 }
2444 else
2445 {
2446 // unsigned
2447 switch (length_modifier)
2448 {
2449 case '\0':
2450 case 'h':
2451 uint_arg =
2452# if defined(FEAT_EVAL)
2453 tvs != NULL ? (unsigned)
2454 tv_nr(tvs, &arg_idx) :
2455# endif
2456 va_arg(ap, unsigned int);
2457 if (uint_arg != 0)
2458 arg_sign = 1;
2459 break;
2460 case 'l':
2461 ulong_arg =
2462# if defined(FEAT_EVAL)
2463 tvs != NULL ? (unsigned long)
2464 tv_nr(tvs, &arg_idx) :
2465# endif
2466 va_arg(ap, unsigned long int);
2467 if (ulong_arg != 0)
2468 arg_sign = 1;
2469 break;
2470 case 'L':
2471 ullong_arg =
2472# if defined(FEAT_EVAL)
2473 tvs != NULL ? (uvarnumber_T)
2474 tv_nr(tvs, &arg_idx) :
2475# endif
2476 va_arg(ap, uvarnumber_T);
2477 if (ullong_arg != 0)
2478 arg_sign = 1;
2479 break;
2480 }
2481 }
2482
2483 str_arg = tmp;
2484 str_arg_l = 0;
2485
2486 // NOTE:
2487 // For d, i, u, o, x, and X conversions, if precision is
2488 // specified, the '0' flag should be ignored. This is so
2489 // with Solaris 2.6, Digital UNIX 4.0, HPUX 10, Linux,
2490 // FreeBSD, NetBSD; but not with Perl.
2491 if (precision_specified)
2492 zero_padding = 0;
2493 if (fmt_spec == 'd')
2494 {
2495 if (force_sign && arg_sign >= 0)
2496 tmp[str_arg_l++] = space_for_positive ? ' ' : '+';
2497 // leave negative numbers for sprintf to handle, to
2498 // avoid handling tricky cases like (short int)-32768
2499 }
2500 else if (alternate_form)
2501 {
2502 if (arg_sign != 0
2503 && (fmt_spec == 'b' || fmt_spec == 'B'
2504 || fmt_spec == 'x' || fmt_spec == 'X') )
2505 {
2506 tmp[str_arg_l++] = '0';
2507 tmp[str_arg_l++] = fmt_spec;
2508 }
2509 // alternate form should have no effect for p
2510 // conversion, but ...
2511 }
2512
2513 zero_padding_insertion_ind = str_arg_l;
2514 if (!precision_specified)
2515 precision = 1; // default precision is 1
2516 if (precision == 0 && arg_sign == 0)
2517 {
2518 // When zero value is formatted with an explicit
2519 // precision 0, the resulting formatted string is
2520 // empty (d, i, u, b, B, o, x, X, p).
2521 }
2522 else
2523 {
2524 char f[6];
2525 int f_l = 0;
2526
2527 // construct a simple format string for sprintf
2528 f[f_l++] = '%';
2529 if (!length_modifier)
2530 ;
2531 else if (length_modifier == 'L')
2532 {
2533# ifdef MSWIN
2534 f[f_l++] = 'I';
2535 f[f_l++] = '6';
2536 f[f_l++] = '4';
2537# else
2538 f[f_l++] = 'l';
2539 f[f_l++] = 'l';
2540# endif
2541 }
2542 else
2543 f[f_l++] = length_modifier;
2544 f[f_l++] = fmt_spec;
2545 f[f_l++] = '\0';
2546
2547 if (fmt_spec == 'p')
2548 str_arg_l += sprintf(tmp + str_arg_l, f, ptr_arg);
2549 else if (fmt_spec == 'b' || fmt_spec == 'B')
2550 {
2551 char b[8 * sizeof(uvarnumber_T)];
2552 size_t b_l = 0;
2553 uvarnumber_T bn = bin_arg;
2554
2555 do
2556 {
2557 b[sizeof(b) - ++b_l] = '0' + (bn & 0x1);
2558 bn >>= 1;
2559 }
2560 while (bn != 0);
2561
2562 memcpy(tmp + str_arg_l, b + sizeof(b) - b_l, b_l);
2563 str_arg_l += b_l;
2564 }
2565 else if (fmt_spec == 'd')
2566 {
2567 // signed
2568 switch (length_modifier)
2569 {
2570 case '\0': str_arg_l += sprintf(
2571 tmp + str_arg_l, f,
2572 int_arg);
2573 break;
2574 case 'h': str_arg_l += sprintf(
2575 tmp + str_arg_l, f,
2576 (short)int_arg);
2577 break;
2578 case 'l': str_arg_l += sprintf(
2579 tmp + str_arg_l, f, long_arg);
2580 break;
2581 case 'L': str_arg_l += sprintf(
2582 tmp + str_arg_l, f, llong_arg);
2583 break;
2584 }
2585 }
2586 else
2587 {
2588 // unsigned
2589 switch (length_modifier)
2590 {
2591 case '\0': str_arg_l += sprintf(
2592 tmp + str_arg_l, f,
2593 uint_arg);
2594 break;
2595 case 'h': str_arg_l += sprintf(
2596 tmp + str_arg_l, f,
2597 (unsigned short)uint_arg);
2598 break;
2599 case 'l': str_arg_l += sprintf(
2600 tmp + str_arg_l, f, ulong_arg);
2601 break;
2602 case 'L': str_arg_l += sprintf(
2603 tmp + str_arg_l, f, ullong_arg);
2604 break;
2605 }
2606 }
2607
2608 // include the optional minus sign and possible
2609 // "0x" in the region before the zero padding
2610 // insertion point
2611 if (zero_padding_insertion_ind < str_arg_l
2612 && tmp[zero_padding_insertion_ind] == '-')
2613 zero_padding_insertion_ind++;
2614 if (zero_padding_insertion_ind + 1 < str_arg_l
2615 && tmp[zero_padding_insertion_ind] == '0'
2616 && (tmp[zero_padding_insertion_ind + 1] == 'x'
2617 || tmp[zero_padding_insertion_ind + 1] == 'X'))
2618 zero_padding_insertion_ind += 2;
2619 }
2620
2621 {
2622 size_t num_of_digits = str_arg_l
2623 - zero_padding_insertion_ind;
2624
2625 if (alternate_form && fmt_spec == 'o'
2626 // unless zero is already the first
2627 // character
2628 && !(zero_padding_insertion_ind < str_arg_l
2629 && tmp[zero_padding_insertion_ind] == '0'))
2630 {
2631 // assure leading zero for alternate-form
2632 // octal numbers
2633 if (!precision_specified
2634 || precision < num_of_digits + 1)
2635 {
2636 // precision is increased to force the
2637 // first character to be zero, except if a
2638 // zero value is formatted with an
2639 // explicit precision of zero
2640 precision = num_of_digits + 1;
2641 }
2642 }
2643 // zero padding to specified precision?
2644 if (num_of_digits < precision)
2645 number_of_zeros_to_pad = precision - num_of_digits;
2646 }
2647 // zero padding to specified minimal field width?
2648 if (!justify_left && zero_padding)
2649 {
2650 int n = (int)(min_field_width - (str_arg_l
2651 + number_of_zeros_to_pad));
2652 if (n > 0)
2653 number_of_zeros_to_pad += n;
2654 }
2655 break;
2656 }
2657
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002658 case 'f':
2659 case 'F':
2660 case 'e':
2661 case 'E':
2662 case 'g':
2663 case 'G':
2664 {
2665 // Floating point.
2666 double f;
2667 double abs_f;
2668 char format[40];
2669 int l;
2670 int remove_trailing_zeroes = FALSE;
2671
2672 f =
Bram Moolenaar73e28dc2022-09-17 21:08:33 +01002673# if defined(FEAT_EVAL)
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002674 tvs != NULL ? tv_float(tvs, &arg_idx) :
Bram Moolenaar73e28dc2022-09-17 21:08:33 +01002675# endif
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002676 va_arg(ap, double);
2677 abs_f = f < 0 ? -f : f;
2678
2679 if (fmt_spec == 'g' || fmt_spec == 'G')
2680 {
2681 // Would be nice to use %g directly, but it prints
2682 // "1.0" as "1", we don't want that.
2683 if ((abs_f >= 0.001 && abs_f < 10000000.0)
2684 || abs_f == 0.0)
2685 fmt_spec = ASCII_ISUPPER(fmt_spec) ? 'F' : 'f';
2686 else
2687 fmt_spec = fmt_spec == 'g' ? 'e' : 'E';
2688 remove_trailing_zeroes = TRUE;
2689 }
2690
2691 if ((fmt_spec == 'f' || fmt_spec == 'F') &&
Bram Moolenaar73e28dc2022-09-17 21:08:33 +01002692# ifdef VAX
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002693 abs_f > 1.0e38
Bram Moolenaar73e28dc2022-09-17 21:08:33 +01002694# else
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002695 abs_f > 1.0e307
Bram Moolenaar73e28dc2022-09-17 21:08:33 +01002696# endif
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002697 )
2698 {
2699 // Avoid a buffer overflow
2700 STRCPY(tmp, infinity_str(f > 0.0, fmt_spec,
2701 force_sign, space_for_positive));
2702 str_arg_l = STRLEN(tmp);
2703 zero_padding = 0;
2704 }
2705 else
2706 {
2707 if (isnan(f))
2708 {
2709 // Not a number: nan or NAN
2710 STRCPY(tmp, ASCII_ISUPPER(fmt_spec) ? "NAN"
2711 : "nan");
2712 str_arg_l = 3;
2713 zero_padding = 0;
2714 }
2715 else if (isinf(f))
2716 {
2717 STRCPY(tmp, infinity_str(f > 0.0, fmt_spec,
2718 force_sign, space_for_positive));
2719 str_arg_l = STRLEN(tmp);
2720 zero_padding = 0;
2721 }
2722 else
2723 {
2724 // Regular float number
2725 format[0] = '%';
2726 l = 1;
2727 if (force_sign)
2728 format[l++] = space_for_positive ? ' ' : '+';
2729 if (precision_specified)
2730 {
2731 size_t max_prec = TMP_LEN - 10;
2732
2733 // Make sure we don't get more digits than we
2734 // have room for.
2735 if ((fmt_spec == 'f' || fmt_spec == 'F')
2736 && abs_f > 1.0)
2737 max_prec -= (size_t)log10(abs_f);
2738 if (precision > max_prec)
2739 precision = max_prec;
2740 l += sprintf(format + l, ".%d", (int)precision);
2741 }
2742 format[l] = fmt_spec == 'F' ? 'f' : fmt_spec;
2743 format[l + 1] = NUL;
2744
2745 str_arg_l = sprintf(tmp, format, f);
2746 }
2747
2748 if (remove_trailing_zeroes)
2749 {
2750 int i;
2751 char *tp;
2752
2753 // Using %g or %G: remove superfluous zeroes.
2754 if (fmt_spec == 'f' || fmt_spec == 'F')
2755 tp = tmp + str_arg_l - 1;
2756 else
2757 {
2758 tp = (char *)vim_strchr((char_u *)tmp,
2759 fmt_spec == 'e' ? 'e' : 'E');
2760 if (tp != NULL)
2761 {
2762 // Remove superfluous '+' and leading
2763 // zeroes from the exponent.
2764 if (tp[1] == '+')
2765 {
2766 // Change "1.0e+07" to "1.0e07"
2767 STRMOVE(tp + 1, tp + 2);
2768 --str_arg_l;
2769 }
2770 i = (tp[1] == '-') ? 2 : 1;
2771 while (tp[i] == '0')
2772 {
2773 // Change "1.0e07" to "1.0e7"
2774 STRMOVE(tp + i, tp + i + 1);
2775 --str_arg_l;
2776 }
2777 --tp;
2778 }
2779 }
2780
2781 if (tp != NULL && !precision_specified)
2782 // Remove trailing zeroes, but keep the one
2783 // just after a dot.
2784 while (tp > tmp + 2 && *tp == '0'
2785 && tp[-1] != '.')
2786 {
2787 STRMOVE(tp, tp + 1);
2788 --tp;
2789 --str_arg_l;
2790 }
2791 }
2792 else
2793 {
2794 char *tp;
2795
2796 // Be consistent: some printf("%e") use 1.0e+12
2797 // and some 1.0e+012. Remove one zero in the last
2798 // case.
2799 tp = (char *)vim_strchr((char_u *)tmp,
2800 fmt_spec == 'e' ? 'e' : 'E');
2801 if (tp != NULL && (tp[1] == '+' || tp[1] == '-')
2802 && tp[2] == '0'
2803 && vim_isdigit(tp[3])
2804 && vim_isdigit(tp[4]))
2805 {
2806 STRMOVE(tp + 2, tp + 3);
2807 --str_arg_l;
2808 }
2809 }
2810 }
2811 if (zero_padding && min_field_width > str_arg_l
2812 && (tmp[0] == '-' || force_sign))
2813 {
2814 // padding 0's should be inserted after the sign
2815 number_of_zeros_to_pad = min_field_width - str_arg_l;
2816 zero_padding_insertion_ind = 1;
2817 }
2818 str_arg = tmp;
2819 break;
2820 }
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002821
2822 default:
2823 // unrecognized conversion specifier, keep format string
2824 // as-is
2825 zero_padding = 0; // turn zero padding off for non-numeric
2826 // conversion
2827 justify_left = 1;
2828 min_field_width = 0; // reset flags
2829
2830 // discard the unrecognized conversion, just keep *
2831 // the unrecognized conversion character
2832 str_arg = p;
2833 str_arg_l = 0;
2834 if (*p != NUL)
2835 str_arg_l++; // include invalid conversion specifier
2836 // unchanged if not at end-of-string
2837 break;
2838 }
2839
2840 if (*p != NUL)
2841 p++; // step over the just processed conversion specifier
2842
2843 // insert padding to the left as requested by min_field_width;
2844 // this does not include the zero padding in case of numerical
2845 // conversions
2846 if (!justify_left)
2847 {
2848 // left padding with blank or zero
2849 int pn = (int)(min_field_width - (str_arg_l + number_of_zeros_to_pad));
2850
2851 if (pn > 0)
2852 {
2853 if (str_l < str_m)
2854 {
2855 size_t avail = str_m - str_l;
2856
2857 vim_memset(str + str_l, zero_padding ? '0' : ' ',
2858 (size_t)pn > avail ? avail
2859 : (size_t)pn);
2860 }
2861 str_l += pn;
2862 }
2863 }
2864
2865 // zero padding as requested by the precision or by the minimal
2866 // field width for numeric conversions required?
2867 if (number_of_zeros_to_pad == 0)
2868 {
2869 // will not copy first part of numeric right now, *
2870 // force it to be copied later in its entirety
2871 zero_padding_insertion_ind = 0;
2872 }
2873 else
2874 {
2875 // insert first part of numerics (sign or '0x') before zero
2876 // padding
2877 int zn = (int)zero_padding_insertion_ind;
2878
2879 if (zn > 0)
2880 {
2881 if (str_l < str_m)
2882 {
2883 size_t avail = str_m - str_l;
2884
2885 mch_memmove(str + str_l, str_arg,
2886 (size_t)zn > avail ? avail
2887 : (size_t)zn);
2888 }
2889 str_l += zn;
2890 }
2891
2892 // insert zero padding as requested by the precision or min
2893 // field width
2894 zn = (int)number_of_zeros_to_pad;
2895 if (zn > 0)
2896 {
2897 if (str_l < str_m)
2898 {
2899 size_t avail = str_m - str_l;
2900
2901 vim_memset(str + str_l, '0',
2902 (size_t)zn > avail ? avail
2903 : (size_t)zn);
2904 }
2905 str_l += zn;
2906 }
2907 }
2908
2909 // insert formatted string
2910 // (or as-is conversion specifier for unknown conversions)
2911 {
2912 int sn = (int)(str_arg_l - zero_padding_insertion_ind);
2913
2914 if (sn > 0)
2915 {
2916 if (str_l < str_m)
2917 {
2918 size_t avail = str_m - str_l;
2919
2920 mch_memmove(str + str_l,
2921 str_arg + zero_padding_insertion_ind,
2922 (size_t)sn > avail ? avail : (size_t)sn);
2923 }
2924 str_l += sn;
2925 }
2926 }
2927
2928 // insert right padding
2929 if (justify_left)
2930 {
2931 // right blank padding to the field width
2932 int pn = (int)(min_field_width
2933 - (str_arg_l + number_of_zeros_to_pad));
2934
2935 if (pn > 0)
2936 {
2937 if (str_l < str_m)
2938 {
2939 size_t avail = str_m - str_l;
2940
2941 vim_memset(str + str_l, ' ',
2942 (size_t)pn > avail ? avail
2943 : (size_t)pn);
2944 }
2945 str_l += pn;
2946 }
2947 }
2948 vim_free(tofree);
2949 }
2950 }
2951
2952 if (str_m > 0)
2953 {
2954 // make sure the string is nul-terminated even at the expense of
2955 // overwriting the last character (shouldn't happen, but just in case)
2956 //
2957 str[str_l <= str_m - 1 ? str_l : str_m - 1] = '\0';
2958 }
2959
2960 if (tvs != NULL && tvs[arg_idx - 1].v_type != VAR_UNKNOWN)
Bram Moolenaar677658a2022-01-05 16:09:06 +00002961 emsg(_(e_too_many_arguments_to_printf));
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +02002962
2963 // Return the number of characters formatted (excluding trailing nul
2964 // character), that is, the number of characters that would have been
2965 // written to the buffer if it were large enough.
2966 return (int)str_l;
2967}
2968
2969#endif // PROTO