blob: f4a313d4164ea48e39a741398a32d737b02706c1 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
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 * misc1.c: functions that didn't seem to fit elsewhere
12 */
13
14#include "vim.h"
15#include "version.h"
16
Bram Moolenaarb3f74062020-02-26 16:16:53 +010017#if defined(__HAIKU__)
18# include <storage/FindDirectory.h>
19#endif
20
Bram Moolenaar0a52df52019-08-18 22:26:31 +020021#if defined(MSWIN)
Bram Moolenaar828c3d72018-06-19 18:58:07 +020022# include <lm.h>
23#endif
24
Bram Moolenaar85a20022019-12-21 18:25:54 +010025#define URL_SLASH 1 // path_is_url() has found "://"
26#define URL_BACKSLASH 2 // path_is_url() has found ":\\"
Bram Moolenaar5fd0f502019-02-13 23:13:28 +010027
Bram Moolenaar0a52df52019-08-18 22:26:31 +020028// All user names (for ~user completion as done by shell).
Bram Moolenaar24305862012-08-15 14:05:05 +020029static garray_T ga_users;
Bram Moolenaar24305862012-08-15 14:05:05 +020030
Bram Moolenaar071d4272004-06-13 20:20:40 +000031/*
Bram Moolenaar2c019c82013-10-06 17:46:56 +020032 * get_leader_len() returns the length in bytes of the prefix of the given
33 * string which introduces a comment. If this string is not a comment then
34 * 0 is returned.
Bram Moolenaar071d4272004-06-13 20:20:40 +000035 * When "flags" is not NULL, it is set to point to the flags of the recognized
36 * comment leader.
37 * "backward" must be true for the "O" command.
Bram Moolenaar81340392012-06-06 16:12:59 +020038 * If "include_space" is set, include trailing whitespace while calculating the
39 * length.
Bram Moolenaar071d4272004-06-13 20:20:40 +000040 */
41 int
Bram Moolenaar9b578142016-01-30 19:39:49 +010042get_leader_len(
43 char_u *line,
44 char_u **flags,
45 int backward,
46 int include_space)
Bram Moolenaar071d4272004-06-13 20:20:40 +000047{
48 int i, j;
Bram Moolenaar81340392012-06-06 16:12:59 +020049 int result;
Bram Moolenaar071d4272004-06-13 20:20:40 +000050 int got_com = FALSE;
51 int found_one;
Bram Moolenaar85a20022019-12-21 18:25:54 +010052 char_u part_buf[COM_MAX_LEN]; // buffer for one option part
53 char_u *string; // pointer to comment string
Bram Moolenaar071d4272004-06-13 20:20:40 +000054 char_u *list;
Bram Moolenaara4271d52011-05-10 13:38:27 +020055 int middle_match_len = 0;
56 char_u *prev_list;
Bram Moolenaar05da4282011-05-10 14:44:11 +020057 char_u *saved_flags = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +000058
Bram Moolenaar81340392012-06-06 16:12:59 +020059 result = i = 0;
Bram Moolenaar85a20022019-12-21 18:25:54 +010060 while (VIM_ISWHITE(line[i])) // leading white space is ignored
Bram Moolenaar071d4272004-06-13 20:20:40 +000061 ++i;
62
63 /*
64 * Repeat to match several nested comment strings.
65 */
Bram Moolenaara4271d52011-05-10 13:38:27 +020066 while (line[i] != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +000067 {
68 /*
69 * scan through the 'comments' option for a match
70 */
71 found_one = FALSE;
72 for (list = curbuf->b_p_com; *list; )
73 {
Bram Moolenaar85a20022019-12-21 18:25:54 +010074 // Get one option part into part_buf[]. Advance "list" to next
75 // one. Put "string" at start of string.
Bram Moolenaara4271d52011-05-10 13:38:27 +020076 if (!got_com && flags != NULL)
Bram Moolenaar85a20022019-12-21 18:25:54 +010077 *flags = list; // remember where flags started
Bram Moolenaara4271d52011-05-10 13:38:27 +020078 prev_list = list;
Bram Moolenaar071d4272004-06-13 20:20:40 +000079 (void)copy_option_part(&list, part_buf, COM_MAX_LEN, ",");
80 string = vim_strchr(part_buf, ':');
Bram Moolenaar85a20022019-12-21 18:25:54 +010081 if (string == NULL) // missing ':', ignore this part
Bram Moolenaar071d4272004-06-13 20:20:40 +000082 continue;
Bram Moolenaar85a20022019-12-21 18:25:54 +010083 *string++ = NUL; // isolate flags from string
Bram Moolenaar071d4272004-06-13 20:20:40 +000084
Bram Moolenaar85a20022019-12-21 18:25:54 +010085 // If we found a middle match previously, use that match when this
86 // is not a middle or end.
Bram Moolenaara4271d52011-05-10 13:38:27 +020087 if (middle_match_len != 0
88 && vim_strchr(part_buf, COM_MIDDLE) == NULL
89 && vim_strchr(part_buf, COM_END) == NULL)
90 break;
91
Bram Moolenaar85a20022019-12-21 18:25:54 +010092 // When we already found a nested comment, only accept further
93 // nested comments.
Bram Moolenaar071d4272004-06-13 20:20:40 +000094 if (got_com && vim_strchr(part_buf, COM_NEST) == NULL)
95 continue;
96
Bram Moolenaar85a20022019-12-21 18:25:54 +010097 // When 'O' flag present and using "O" command skip this one.
Bram Moolenaar071d4272004-06-13 20:20:40 +000098 if (backward && vim_strchr(part_buf, COM_NOBACK) != NULL)
99 continue;
100
Bram Moolenaar85a20022019-12-21 18:25:54 +0100101 // Line contents and string must match.
102 // When string starts with white space, must have some white space
103 // (but the amount does not need to match, there might be a mix of
104 // TABs and spaces).
Bram Moolenaar1c465442017-03-12 20:10:05 +0100105 if (VIM_ISWHITE(string[0]))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000106 {
Bram Moolenaar1c465442017-03-12 20:10:05 +0100107 if (i == 0 || !VIM_ISWHITE(line[i - 1]))
Bram Moolenaar85a20022019-12-21 18:25:54 +0100108 continue; // missing white space
Bram Moolenaar1c465442017-03-12 20:10:05 +0100109 while (VIM_ISWHITE(string[0]))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000110 ++string;
111 }
112 for (j = 0; string[j] != NUL && string[j] == line[i + j]; ++j)
113 ;
114 if (string[j] != NUL)
Bram Moolenaar85a20022019-12-21 18:25:54 +0100115 continue; // string doesn't match
Bram Moolenaar071d4272004-06-13 20:20:40 +0000116
Bram Moolenaar85a20022019-12-21 18:25:54 +0100117 // When 'b' flag used, there must be white space or an
118 // end-of-line after the string in the line.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000119 if (vim_strchr(part_buf, COM_BLANK) != NULL
Bram Moolenaar1c465442017-03-12 20:10:05 +0100120 && !VIM_ISWHITE(line[i + j]) && line[i + j] != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000121 continue;
122
Bram Moolenaar85a20022019-12-21 18:25:54 +0100123 // We have found a match, stop searching unless this is a middle
124 // comment. The middle comment can be a substring of the end
125 // comment in which case it's better to return the length of the
126 // end comment and its flags. Thus we keep searching with middle
127 // and end matches and use an end match if it matches better.
Bram Moolenaara4271d52011-05-10 13:38:27 +0200128 if (vim_strchr(part_buf, COM_MIDDLE) != NULL)
129 {
130 if (middle_match_len == 0)
131 {
132 middle_match_len = j;
133 saved_flags = prev_list;
134 }
135 continue;
136 }
137 if (middle_match_len != 0 && j > middle_match_len)
Bram Moolenaar85a20022019-12-21 18:25:54 +0100138 // Use this match instead of the middle match, since it's a
139 // longer thus better match.
Bram Moolenaara4271d52011-05-10 13:38:27 +0200140 middle_match_len = 0;
141
142 if (middle_match_len == 0)
143 i += j;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000144 found_one = TRUE;
145 break;
146 }
147
Bram Moolenaara4271d52011-05-10 13:38:27 +0200148 if (middle_match_len != 0)
149 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100150 // Use the previously found middle match after failing to find a
151 // match with an end.
Bram Moolenaara4271d52011-05-10 13:38:27 +0200152 if (!got_com && flags != NULL)
153 *flags = saved_flags;
154 i += middle_match_len;
155 found_one = TRUE;
156 }
157
Bram Moolenaar85a20022019-12-21 18:25:54 +0100158 // No match found, stop scanning.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000159 if (!found_one)
160 break;
161
Bram Moolenaar81340392012-06-06 16:12:59 +0200162 result = i;
163
Bram Moolenaar85a20022019-12-21 18:25:54 +0100164 // Include any trailing white space.
Bram Moolenaar1c465442017-03-12 20:10:05 +0100165 while (VIM_ISWHITE(line[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000166 ++i;
167
Bram Moolenaar81340392012-06-06 16:12:59 +0200168 if (include_space)
169 result = i;
170
Bram Moolenaar85a20022019-12-21 18:25:54 +0100171 // If this comment doesn't nest, stop here.
Bram Moolenaara4271d52011-05-10 13:38:27 +0200172 got_com = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000173 if (vim_strchr(part_buf, COM_NEST) == NULL)
174 break;
175 }
Bram Moolenaar81340392012-06-06 16:12:59 +0200176 return result;
177}
Bram Moolenaara4271d52011-05-10 13:38:27 +0200178
Bram Moolenaar81340392012-06-06 16:12:59 +0200179/*
180 * Return the offset at which the last comment in line starts. If there is no
181 * comment in the whole line, -1 is returned.
182 *
183 * When "flags" is not null, it is set to point to the flags describing the
184 * recognized comment leader.
185 */
186 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100187get_last_leader_offset(char_u *line, char_u **flags)
Bram Moolenaar81340392012-06-06 16:12:59 +0200188{
189 int result = -1;
190 int i, j;
191 int lower_check_bound = 0;
192 char_u *string;
193 char_u *com_leader;
194 char_u *com_flags;
195 char_u *list;
196 int found_one;
Bram Moolenaar85a20022019-12-21 18:25:54 +0100197 char_u part_buf[COM_MAX_LEN]; // buffer for one option part
Bram Moolenaar81340392012-06-06 16:12:59 +0200198
199 /*
200 * Repeat to match several nested comment strings.
201 */
202 i = (int)STRLEN(line);
203 while (--i >= lower_check_bound)
204 {
205 /*
206 * scan through the 'comments' option for a match
207 */
208 found_one = FALSE;
209 for (list = curbuf->b_p_com; *list; )
210 {
211 char_u *flags_save = list;
212
213 /*
214 * Get one option part into part_buf[]. Advance list to next one.
215 * put string at start of string.
216 */
217 (void)copy_option_part(&list, part_buf, COM_MAX_LEN, ",");
218 string = vim_strchr(part_buf, ':');
Bram Moolenaar85a20022019-12-21 18:25:54 +0100219 if (string == NULL) // If everything is fine, this cannot actually
220 // happen.
Bram Moolenaar81340392012-06-06 16:12:59 +0200221 continue;
Bram Moolenaar85a20022019-12-21 18:25:54 +0100222 *string++ = NUL; // Isolate flags from string.
Bram Moolenaar81340392012-06-06 16:12:59 +0200223 com_leader = string;
224
225 /*
226 * Line contents and string must match.
227 * When string starts with white space, must have some white space
228 * (but the amount does not need to match, there might be a mix of
229 * TABs and spaces).
230 */
Bram Moolenaar1c465442017-03-12 20:10:05 +0100231 if (VIM_ISWHITE(string[0]))
Bram Moolenaar81340392012-06-06 16:12:59 +0200232 {
Bram Moolenaar1c465442017-03-12 20:10:05 +0100233 if (i == 0 || !VIM_ISWHITE(line[i - 1]))
Bram Moolenaar81340392012-06-06 16:12:59 +0200234 continue;
Bram Moolenaar53932812018-12-07 21:08:49 +0100235 while (VIM_ISWHITE(*string))
Bram Moolenaar81340392012-06-06 16:12:59 +0200236 ++string;
237 }
238 for (j = 0; string[j] != NUL && string[j] == line[i + j]; ++j)
239 /* do nothing */;
240 if (string[j] != NUL)
241 continue;
242
243 /*
244 * When 'b' flag used, there must be white space or an
245 * end-of-line after the string in the line.
246 */
247 if (vim_strchr(part_buf, COM_BLANK) != NULL
Bram Moolenaar1c465442017-03-12 20:10:05 +0100248 && !VIM_ISWHITE(line[i + j]) && line[i + j] != NUL)
Bram Moolenaar81340392012-06-06 16:12:59 +0200249 continue;
Bram Moolenaar53932812018-12-07 21:08:49 +0100250
Bram Moolenaar4af72592018-12-09 15:00:52 +0100251 if (vim_strchr(part_buf, COM_MIDDLE) != NULL)
Bram Moolenaar53932812018-12-07 21:08:49 +0100252 {
Bram Moolenaar4af72592018-12-09 15:00:52 +0100253 // For a middlepart comment, only consider it to match if
254 // everything before the current position in the line is
255 // whitespace. Otherwise we would think we are inside a
256 // comment if the middle part appears somewhere in the middle
257 // of the line. E.g. for C the "*" appears often.
Bram Moolenaar53932812018-12-07 21:08:49 +0100258 for (j = 0; VIM_ISWHITE(line[j]) && j <= i; j++)
259 ;
260 if (j < i)
261 continue;
Bram Moolenaar81340392012-06-06 16:12:59 +0200262 }
263
264 /*
265 * We have found a match, stop searching.
266 */
267 found_one = TRUE;
268
269 if (flags)
270 *flags = flags_save;
271 com_flags = flags_save;
272
273 break;
274 }
275
276 if (found_one)
277 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100278 char_u part_buf2[COM_MAX_LEN]; // buffer for one option part
Bram Moolenaar81340392012-06-06 16:12:59 +0200279 int len1, len2, off;
280
281 result = i;
282 /*
283 * If this comment nests, continue searching.
284 */
285 if (vim_strchr(part_buf, COM_NEST) != NULL)
286 continue;
287
288 lower_check_bound = i;
289
Bram Moolenaar85a20022019-12-21 18:25:54 +0100290 // Let's verify whether the comment leader found is a substring
291 // of other comment leaders. If it is, let's adjust the
292 // lower_check_bound so that we make sure that we have determined
293 // the comment leader correctly.
Bram Moolenaar81340392012-06-06 16:12:59 +0200294
Bram Moolenaar1c465442017-03-12 20:10:05 +0100295 while (VIM_ISWHITE(*com_leader))
Bram Moolenaar81340392012-06-06 16:12:59 +0200296 ++com_leader;
297 len1 = (int)STRLEN(com_leader);
298
299 for (list = curbuf->b_p_com; *list; )
300 {
301 char_u *flags_save = list;
302
303 (void)copy_option_part(&list, part_buf2, COM_MAX_LEN, ",");
304 if (flags_save == com_flags)
305 continue;
306 string = vim_strchr(part_buf2, ':');
307 ++string;
Bram Moolenaar1c465442017-03-12 20:10:05 +0100308 while (VIM_ISWHITE(*string))
Bram Moolenaar81340392012-06-06 16:12:59 +0200309 ++string;
310 len2 = (int)STRLEN(string);
311 if (len2 == 0)
312 continue;
313
Bram Moolenaar85a20022019-12-21 18:25:54 +0100314 // Now we have to verify whether string ends with a substring
315 // beginning the com_leader.
Bram Moolenaar81340392012-06-06 16:12:59 +0200316 for (off = (len2 > i ? i : len2); off > 0 && off + len1 > len2;)
317 {
318 --off;
319 if (!STRNCMP(string + off, com_leader, len2 - off))
320 {
321 if (i - off < lower_check_bound)
322 lower_check_bound = i - off;
323 }
324 }
325 }
326 }
327 }
328 return result;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000329}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000330
331/*
332 * Return the number of window lines occupied by buffer line "lnum".
333 */
334 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100335plines(linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000336{
337 return plines_win(curwin, lnum, TRUE);
338}
339
340 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100341plines_win(
342 win_T *wp,
343 linenr_T lnum,
Bram Moolenaar85a20022019-12-21 18:25:54 +0100344 int winheight) // when TRUE limit to window height
Bram Moolenaar071d4272004-06-13 20:20:40 +0000345{
346#if defined(FEAT_DIFF) || defined(PROTO)
Bram Moolenaar85a20022019-12-21 18:25:54 +0100347 // Check for filler lines above this buffer line. When folded the result
348 // is one line anyway.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000349 return plines_win_nofill(wp, lnum, winheight) + diff_check_fill(wp, lnum);
350}
351
352 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100353plines_nofill(linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000354{
355 return plines_win_nofill(curwin, lnum, TRUE);
356}
357
358 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100359plines_win_nofill(
360 win_T *wp,
361 linenr_T lnum,
Bram Moolenaar85a20022019-12-21 18:25:54 +0100362 int winheight) // when TRUE limit to window height
Bram Moolenaar071d4272004-06-13 20:20:40 +0000363{
364#endif
365 int lines;
366
Bram Moolenaar071d4272004-06-13 20:20:40 +0000367 if (wp->w_width == 0)
368 return 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000369
370#ifdef FEAT_FOLDING
Dominique Pelleaf4a61a2021-12-27 17:21:41 +0000371 // Folded lines are handled just like an empty line.
Bram Moolenaar85a20022019-12-21 18:25:54 +0100372 // NOTE: Caller must handle lines that are MAYBE folded.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000373 if (lineFolded(wp, lnum) == TRUE)
374 return 1;
375#endif
376
Bram Moolenaar4d91d342022-08-06 13:48:20 +0100377 if (!wp->w_p_wrap)
378 lines = 1
379#ifdef FEAT_PROP_POPUP
380 // add a line for each "below" aligned text property
381 + prop_count_below(wp->w_buffer, lnum)
382#endif
383 ;
384 else
385 lines = plines_win_nofold(wp, lnum);
386
Bram Moolenaar071d4272004-06-13 20:20:40 +0000387 if (winheight > 0 && lines > wp->w_height)
=?UTF-8?q?Dundar=20G=C3=B6c?=420fabc2022-01-28 15:28:04 +0000388 return wp->w_height;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000389 return lines;
390}
391
392/*
393 * Return number of window lines physical line "lnum" will occupy in window
394 * "wp". Does not care about folding, 'wrap' or 'diff'.
395 */
396 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100397plines_win_nofold(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000398{
399 char_u *s;
400 long col;
401 int width;
Bram Moolenaar49a90792022-08-09 18:25:23 +0100402 chartabsize_T cts;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000403
404 s = ml_get_buf(wp->w_buffer, lnum, FALSE);
Bram Moolenaar49a90792022-08-09 18:25:23 +0100405 init_chartabsize_arg(&cts, wp, lnum, 0, s, s);
406 if (*s == NUL
407#ifdef FEAT_PROP_POPUP
408 && !cts.cts_has_prop_with_text
409#endif
410 )
411 return 1; // be quick for an empty line
412 win_linetabsize_cts(&cts, (colnr_T)MAXCOL);
413 clear_chartabsize_arg(&cts);
414 col = (int)cts.cts_vcol;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000415
416 /*
417 * If list mode is on, then the '$' at the end of the line may take up one
418 * extra column.
419 */
Bram Moolenaareed9d462021-02-15 20:38:25 +0100420 if (wp->w_p_list && wp->w_lcs_chars.eol != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000421 col += 1;
422
423 /*
Bram Moolenaar64486672010-05-16 15:46:46 +0200424 * Add column offset for 'number', 'relativenumber' and 'foldcolumn'.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000425 */
Bram Moolenaar02631462017-09-22 15:20:32 +0200426 width = wp->w_width - win_col_off(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000427 if (width <= 0)
428 return 32000;
429 if (col <= width)
430 return 1;
431 col -= width;
432 width += win_col_off2(wp);
433 return (col + (width - 1)) / width + 1;
434}
435
436/*
437 * Like plines_win(), but only reports the number of physical screen lines
438 * used from the start of the line to the given column number.
439 */
440 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100441plines_win_col(win_T *wp, linenr_T lnum, long column)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000442{
443 long col;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000444 int lines = 0;
445 int width;
Bram Moolenaar597a4222014-06-25 14:39:50 +0200446 char_u *line;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100447 chartabsize_T cts;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000448
449#ifdef FEAT_DIFF
Bram Moolenaar85a20022019-12-21 18:25:54 +0100450 // Check for filler lines above this buffer line. When folded the result
451 // is one line anyway.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000452 lines = diff_check_fill(wp, lnum);
453#endif
454
455 if (!wp->w_p_wrap)
456 return lines + 1;
457
Bram Moolenaar071d4272004-06-13 20:20:40 +0000458 if (wp->w_width == 0)
459 return lines + 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000460
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100461 line = ml_get_buf(wp->w_buffer, lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000462
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100463 init_chartabsize_arg(&cts, wp, lnum, 0, line, line);
464 while (*cts.cts_ptr != NUL && --column >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000465 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100466 cts.cts_vcol += win_lbr_chartabsize(&cts, NULL);
467 MB_PTR_ADV(cts.cts_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000468 }
469
470 /*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100471 * If *cts.cts_ptr is a TAB, and the TAB is not displayed as ^I, and we're
472 * not in MODE_INSERT state, then col must be adjusted so that it
473 * represents the last screen position of the TAB. This only fixes an
474 * error when the TAB wraps from one screen line to the next (when
475 * 'columns' is not a multiple of 'ts') -- webb.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000476 */
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100477 col = cts.cts_vcol;
478 if (*cts.cts_ptr == TAB && (State & MODE_NORMAL)
Bram Moolenaar24959102022-05-07 20:01:16 +0100479 && (!wp->w_p_list || wp->w_lcs_chars.tab1))
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100480 col += win_lbr_chartabsize(&cts, NULL) - 1;
481 clear_chartabsize_arg(&cts);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000482
483 /*
Bram Moolenaar64486672010-05-16 15:46:46 +0200484 * Add column offset for 'number', 'relativenumber', 'foldcolumn', etc.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000485 */
Bram Moolenaar02631462017-09-22 15:20:32 +0200486 width = wp->w_width - win_col_off(wp);
Bram Moolenaar26470632006-10-24 19:12:40 +0000487 if (width <= 0)
488 return 9999;
489
490 lines += 1;
491 if (col > width)
492 lines += (col - width) / (width + win_col_off2(wp)) + 1;
493 return lines;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000494}
495
496 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100497plines_m_win(win_T *wp, linenr_T first, linenr_T last)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000498{
499 int count = 0;
500
501 while (first <= last)
502 {
503#ifdef FEAT_FOLDING
504 int x;
505
Bram Moolenaar85a20022019-12-21 18:25:54 +0100506 // Check if there are any really folded lines, but also included lines
507 // that are maybe folded.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000508 x = foldedCount(wp, first, NULL);
509 if (x > 0)
510 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100511 ++count; // count 1 for "+-- folded" line
Bram Moolenaar071d4272004-06-13 20:20:40 +0000512 first += x;
513 }
514 else
515#endif
516 {
517#ifdef FEAT_DIFF
518 if (first == wp->w_topline)
519 count += plines_win_nofill(wp, first, TRUE) + wp->w_topfill;
520 else
521#endif
522 count += plines_win(wp, first, TRUE);
523 ++first;
524 }
525 }
526 return (count);
527}
528
Bram Moolenaar071d4272004-06-13 20:20:40 +0000529 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100530gchar_pos(pos_T *pos)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000531{
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100532 char_u *ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000533
Bram Moolenaar85a20022019-12-21 18:25:54 +0100534 // When searching columns is sometimes put at the end of a line.
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100535 if (pos->col == MAXCOL)
536 return NUL;
537 ptr = ml_get_pos(pos);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000538 if (has_mbyte)
539 return (*mb_ptr2char)(ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000540 return (int)*ptr;
541}
542
543 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100544gchar_cursor(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000545{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000546 if (has_mbyte)
547 return (*mb_ptr2char)(ml_get_cursor());
Bram Moolenaar071d4272004-06-13 20:20:40 +0000548 return (int)*ml_get_cursor();
549}
550
551/*
552 * Write a character at the current cursor position.
553 * It is directly written into the block.
554 */
555 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100556pchar_cursor(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000557{
558 *(ml_get_buf(curbuf, curwin->w_cursor.lnum, TRUE)
559 + curwin->w_cursor.col) = c;
560}
561
Bram Moolenaar071d4272004-06-13 20:20:40 +0000562/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000563 * Skip to next part of an option argument: Skip space and comma.
564 */
565 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +0100566skip_to_option_part(char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000567{
568 if (*p == ',')
569 ++p;
570 while (*p == ' ')
571 ++p;
572 return p;
573}
574
575/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000576 * check_status: called when the status bars for the buffer 'buf'
577 * need to be updated
578 */
579 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100580check_status(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000581{
582 win_T *wp;
583
Bram Moolenaar29323592016-07-24 22:04:11 +0200584 FOR_ALL_WINDOWS(wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000585 if (wp->w_buffer == buf && wp->w_status_height)
586 {
587 wp->w_redr_status = TRUE;
588 if (must_redraw < VALID)
589 must_redraw = VALID;
590 }
591}
Bram Moolenaar071d4272004-06-13 20:20:40 +0000592
593/*
Bram Moolenaar6ed545e2022-05-09 20:09:23 +0100594 * Ask for a reply from the user, a 'y' or a 'n', with prompt "str" (which
595 * should have been translated already).
Bram Moolenaar071d4272004-06-13 20:20:40 +0000596 * No other characters are accepted, the message is repeated until a valid
597 * reply is entered or CTRL-C is hit.
598 * If direct is TRUE, don't use vgetc() but ui_inchar(), don't get characters
599 * from any buffers but directly from the user.
600 *
601 * return the 'y' or 'n'
602 */
603 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100604ask_yesno(char_u *str, int direct)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000605{
606 int r = ' ';
607 int save_State = State;
608
Bram Moolenaar85a20022019-12-21 18:25:54 +0100609 if (exiting) // put terminal in raw mode for this question
Bram Moolenaar071d4272004-06-13 20:20:40 +0000610 settmode(TMODE_RAW);
611 ++no_wait_return;
612#ifdef USE_ON_FLY_SCROLL
Bram Moolenaarb20b9e12019-09-21 20:48:04 +0200613 dont_scroll = TRUE; // disallow scrolling here
Bram Moolenaar071d4272004-06-13 20:20:40 +0000614#endif
Bram Moolenaar24959102022-05-07 20:01:16 +0100615 State = MODE_CONFIRM; // mouse behaves like with :confirm
Bram Moolenaarb20b9e12019-09-21 20:48:04 +0200616 setmouse(); // disables mouse for xterm
Bram Moolenaar071d4272004-06-13 20:20:40 +0000617 ++no_mapping;
Bram Moolenaarb20b9e12019-09-21 20:48:04 +0200618 ++allow_keys; // no mapping here, but recognize keys
Bram Moolenaar071d4272004-06-13 20:20:40 +0000619
620 while (r != 'y' && r != 'n')
621 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100622 // same highlighting as for wait_return
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100623 smsg_attr(HL_ATTR(HLF_R), "%s (y/n)?", str);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000624 if (direct)
625 r = get_keystroke();
626 else
Bram Moolenaar913626c2008-01-03 11:43:42 +0000627 r = plain_vgetc();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000628 if (r == Ctrl_C || r == ESC)
629 r = 'n';
Bram Moolenaar85a20022019-12-21 18:25:54 +0100630 msg_putchar(r); // show what you typed
Bram Moolenaar071d4272004-06-13 20:20:40 +0000631 out_flush();
632 }
633 --no_wait_return;
634 State = save_State;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000635 setmouse();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000636 --no_mapping;
637 --allow_keys;
638
639 return r;
640}
641
Bram Moolenaar0e57dd82019-09-16 22:56:03 +0200642#if defined(FEAT_EVAL) || defined(PROTO)
643
644/*
LemonBoy2bf52dd2022-04-09 18:17:34 +0100645 * Returns the current mode as a string in "buf[MODE_MAX_LENGTH]", NUL
646 * terminated.
647 * The first character represents the major mode, the following ones the minor
648 * ones.
649 */
650 void
651get_mode(char_u *buf)
652{
653 int i = 0;
654
655 if (time_for_testing == 93784)
656 {
657 // Testing the two-character code.
658 buf[i++] = 'x';
659 buf[i++] = '!';
660 }
661#ifdef FEAT_TERMINAL
662 else if (term_use_loop())
663 buf[i++] = 't';
664#endif
665 else if (VIsual_active)
666 {
667 if (VIsual_select)
668 buf[i++] = VIsual_mode + 's' - 'v';
669 else
670 {
671 buf[i++] = VIsual_mode;
672 if (restart_VIsual_select)
Bram Moolenaar6ed545e2022-05-09 20:09:23 +0100673 buf[i++] = 's';
LemonBoy2bf52dd2022-04-09 18:17:34 +0100674 }
675 }
Bram Moolenaar24959102022-05-07 20:01:16 +0100676 else if (State == MODE_HITRETURN || State == MODE_ASKMORE
677 || State == MODE_SETWSIZE
678 || State == MODE_CONFIRM)
LemonBoy2bf52dd2022-04-09 18:17:34 +0100679 {
680 buf[i++] = 'r';
Bram Moolenaar24959102022-05-07 20:01:16 +0100681 if (State == MODE_ASKMORE)
LemonBoy2bf52dd2022-04-09 18:17:34 +0100682 buf[i++] = 'm';
Bram Moolenaar24959102022-05-07 20:01:16 +0100683 else if (State == MODE_CONFIRM)
LemonBoy2bf52dd2022-04-09 18:17:34 +0100684 buf[i++] = '?';
685 }
Bram Moolenaar24959102022-05-07 20:01:16 +0100686 else if (State == MODE_EXTERNCMD)
LemonBoy2bf52dd2022-04-09 18:17:34 +0100687 buf[i++] = '!';
Bram Moolenaar24959102022-05-07 20:01:16 +0100688 else if (State & MODE_INSERT)
LemonBoy2bf52dd2022-04-09 18:17:34 +0100689 {
690 if (State & VREPLACE_FLAG)
691 {
692 buf[i++] = 'R';
693 buf[i++] = 'v';
LemonBoy2bf52dd2022-04-09 18:17:34 +0100694 }
695 else
696 {
697 if (State & REPLACE_FLAG)
698 buf[i++] = 'R';
699 else
700 buf[i++] = 'i';
LemonBoy2bf52dd2022-04-09 18:17:34 +0100701 }
zeertzjq590f3652022-04-29 11:29:54 +0100702
703 if (ins_compl_active())
704 buf[i++] = 'c';
705 else if (ctrl_x_mode_not_defined_yet())
706 buf[i++] = 'x';
LemonBoy2bf52dd2022-04-09 18:17:34 +0100707 }
Bram Moolenaar24959102022-05-07 20:01:16 +0100708 else if ((State & MODE_CMDLINE) || exmode_active)
LemonBoy2bf52dd2022-04-09 18:17:34 +0100709 {
710 buf[i++] = 'c';
711 if (exmode_active == EXMODE_VIM)
712 buf[i++] = 'v';
713 else if (exmode_active == EXMODE_NORMAL)
714 buf[i++] = 'e';
715 }
716 else
717 {
718 buf[i++] = 'n';
719 if (finish_op)
720 {
721 buf[i++] = 'o';
722 // to be able to detect force-linewise/blockwise/characterwise
723 // operations
724 buf[i++] = motion_force;
725 }
726 else if (restart_edit == 'I' || restart_edit == 'R'
727 || restart_edit == 'V')
728 {
729 buf[i++] = 'i';
730 buf[i++] = restart_edit;
731 }
732#ifdef FEAT_TERMINAL
733 else if (term_in_normal_mode())
734 buf[i++] = 't';
735#endif
736 }
737
738 buf[i] = NUL;
739}
740
741/*
Bram Moolenaar0e57dd82019-09-16 22:56:03 +0200742 * "mode()" function
743 */
744 void
745f_mode(typval_T *argvars, typval_T *rettv)
746{
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +0200747 char_u buf[MODE_MAX_LENGTH];
Bram Moolenaar0e57dd82019-09-16 22:56:03 +0200748
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200749 if (in_vim9script() && check_for_opt_bool_arg(argvars, 0) == FAIL)
750 return;
751
LemonBoy2bf52dd2022-04-09 18:17:34 +0100752 get_mode(buf);
Bram Moolenaar0e57dd82019-09-16 22:56:03 +0200753
Bram Moolenaar85a20022019-12-21 18:25:54 +0100754 // Clear out the minor mode when the argument is not a non-zero number or
755 // non-empty string.
Bram Moolenaar0e57dd82019-09-16 22:56:03 +0200756 if (!non_zero_arg(&argvars[0]))
757 buf[1] = NUL;
758
759 rettv->vval.v_string = vim_strsave(buf);
760 rettv->v_type = VAR_STRING;
761}
762
763 static void
764may_add_state_char(garray_T *gap, char_u *include, int c)
765{
766 if (include == NULL || vim_strchr(include, c) != NULL)
767 ga_append(gap, c);
768}
769
770/*
771 * "state()" function
772 */
773 void
774f_state(typval_T *argvars, typval_T *rettv)
775{
776 garray_T ga;
777 char_u *include = NULL;
778 int i;
779
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200780 if (in_vim9script() && check_for_opt_string_arg(argvars, 0) == FAIL)
781 return;
782
Bram Moolenaar0e57dd82019-09-16 22:56:03 +0200783 ga_init2(&ga, 1, 20);
784 if (argvars[0].v_type != VAR_UNKNOWN)
785 include = tv_get_string(&argvars[0]);
786
787 if (!(stuff_empty() && typebuf.tb_len == 0 && scriptin[curscript] == NULL))
788 may_add_state_char(&ga, include, 'm');
789 if (op_pending())
790 may_add_state_char(&ga, include, 'o');
791 if (autocmd_busy)
792 may_add_state_char(&ga, include, 'x');
Bram Moolenaarc2585492019-09-22 21:29:53 +0200793 if (ins_compl_active())
Bram Moolenaar0e57dd82019-09-16 22:56:03 +0200794 may_add_state_char(&ga, include, 'a');
795
796# ifdef FEAT_JOB_CHANNEL
797 if (channel_in_blocking_wait())
798 may_add_state_char(&ga, include, 'w');
799# endif
Bram Moolenaarb20b9e12019-09-21 20:48:04 +0200800 if (!get_was_safe_state())
801 may_add_state_char(&ga, include, 'S');
Bram Moolenaar0e57dd82019-09-16 22:56:03 +0200802 for (i = 0; i < get_callback_depth() && i < 3; ++i)
803 may_add_state_char(&ga, include, 'c');
804 if (msg_scrolled > 0)
805 may_add_state_char(&ga, include, 's');
806
807 rettv->v_type = VAR_STRING;
808 rettv->vval.v_string = ga.ga_data;
809}
810
811#endif // FEAT_EVAL
812
Bram Moolenaar071d4272004-06-13 20:20:40 +0000813/*
814 * Get a key stroke directly from the user.
815 * Ignores mouse clicks and scrollbar events, except a click for the left
816 * button (used at the more prompt).
817 * Doesn't use vgetc(), because it syncs undo and eats mapped characters.
818 * Disadvantage: typeahead is ignored.
819 * Translates the interrupt character for unix to ESC.
820 */
821 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100822get_keystroke(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000823{
Bram Moolenaara8c8a682012-02-05 22:05:48 +0100824 char_u *buf = NULL;
825 int buflen = 150;
826 int maxlen;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000827 int len = 0;
828 int n;
829 int save_mapped_ctrl_c = mapped_ctrl_c;
Bram Moolenaar4395a712006-09-05 18:57:57 +0000830 int waited = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000831
Bram Moolenaar85a20022019-12-21 18:25:54 +0100832 mapped_ctrl_c = FALSE; // mappings are not used here
Bram Moolenaar071d4272004-06-13 20:20:40 +0000833 for (;;)
834 {
835 cursor_on();
836 out_flush();
837
Bram Moolenaar85a20022019-12-21 18:25:54 +0100838 // Leave some room for check_termcode() to insert a key code into (max
839 // 5 chars plus NUL). And fix_input_buffer() can triple the number of
840 // bytes.
Bram Moolenaara8c8a682012-02-05 22:05:48 +0100841 maxlen = (buflen - 6 - len) / 3;
842 if (buf == NULL)
843 buf = alloc(buflen);
844 else if (maxlen < 10)
845 {
Bram Moolenaar9abd5c62015-02-10 18:34:01 +0100846 char_u *t_buf = buf;
847
Bram Moolenaar85a20022019-12-21 18:25:54 +0100848 // Need some more space. This might happen when receiving a long
849 // escape sequence.
Bram Moolenaara8c8a682012-02-05 22:05:48 +0100850 buflen += 100;
851 buf = vim_realloc(buf, buflen);
Bram Moolenaar9abd5c62015-02-10 18:34:01 +0100852 if (buf == NULL)
853 vim_free(t_buf);
Bram Moolenaara8c8a682012-02-05 22:05:48 +0100854 maxlen = (buflen - 6 - len) / 3;
855 }
856 if (buf == NULL)
857 {
858 do_outofmem_msg((long_u)buflen);
Bram Moolenaar85a20022019-12-21 18:25:54 +0100859 return ESC; // panic!
Bram Moolenaara8c8a682012-02-05 22:05:48 +0100860 }
861
Bram Moolenaar85a20022019-12-21 18:25:54 +0100862 // First time: blocking wait. Second time: wait up to 100ms for a
863 // terminal code to complete.
Bram Moolenaara8c8a682012-02-05 22:05:48 +0100864 n = ui_inchar(buf + len, maxlen, len == 0 ? -1L : 100L, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000865 if (n > 0)
866 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100867 // Replace zero and CSI by a special key code.
Bram Moolenaar6bff02e2016-08-16 22:50:55 +0200868 n = fix_input_buffer(buf + len, n);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000869 len += n;
Bram Moolenaar4395a712006-09-05 18:57:57 +0000870 waited = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000871 }
Bram Moolenaar4395a712006-09-05 18:57:57 +0000872 else if (len > 0)
Bram Moolenaar85a20022019-12-21 18:25:54 +0100873 ++waited; // keep track of the waiting time
Bram Moolenaar071d4272004-06-13 20:20:40 +0000874
Bram Moolenaar85a20022019-12-21 18:25:54 +0100875 // Incomplete termcode and not timed out yet: get more characters
Bram Moolenaara8c8a682012-02-05 22:05:48 +0100876 if ((n = check_termcode(1, buf, buflen, &len)) < 0
Bram Moolenaar4395a712006-09-05 18:57:57 +0000877 && (!p_ttimeout || waited * 100L < (p_ttm < 0 ? p_tm : p_ttm)))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000878 continue;
Bram Moolenaar4395a712006-09-05 18:57:57 +0000879
Bram Moolenaar85a20022019-12-21 18:25:54 +0100880 if (n == KEYLEN_REMOVED) // key code removed
Bram Moolenaar6eb634e2011-03-03 15:04:08 +0100881 {
Bram Moolenaar24959102022-05-07 20:01:16 +0100882 if (must_redraw != 0 && !need_wait_return && (State
883 & (MODE_CMDLINE | MODE_HITRETURN | MODE_ASKMORE)) == 0)
Bram Moolenaar6eb634e2011-03-03 15:04:08 +0100884 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100885 // Redrawing was postponed, do it now.
Bram Moolenaar6eb634e2011-03-03 15:04:08 +0100886 update_screen(0);
Bram Moolenaar85a20022019-12-21 18:25:54 +0100887 setcursor(); // put cursor back where it belongs
Bram Moolenaar6eb634e2011-03-03 15:04:08 +0100888 }
Bram Moolenaar946ffd42010-12-30 12:30:31 +0100889 continue;
Bram Moolenaar6eb634e2011-03-03 15:04:08 +0100890 }
Bram Moolenaar85a20022019-12-21 18:25:54 +0100891 if (n > 0) // found a termcode: adjust length
Bram Moolenaar071d4272004-06-13 20:20:40 +0000892 len = n;
Bram Moolenaar85a20022019-12-21 18:25:54 +0100893 if (len == 0) // nothing typed yet
Bram Moolenaar071d4272004-06-13 20:20:40 +0000894 continue;
895
Bram Moolenaar85a20022019-12-21 18:25:54 +0100896 // Handle modifier and/or special key code.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000897 n = buf[0];
898 if (n == K_SPECIAL)
899 {
900 n = TO_SPECIAL(buf[1], buf[2]);
901 if (buf[1] == KS_MODIFIER
902 || n == K_IGNORE
Bram Moolenaar2526ef22013-03-16 14:20:51 +0100903 || (is_mouse_key(n) && n != K_LEFTMOUSE)
904#ifdef FEAT_GUI
Bram Moolenaar071d4272004-06-13 20:20:40 +0000905 || n == K_VER_SCROLLBAR
906 || n == K_HOR_SCROLLBAR
Bram Moolenaar071d4272004-06-13 20:20:40 +0000907#endif
908 )
909 {
910 if (buf[1] == KS_MODIFIER)
911 mod_mask = buf[2];
912 len -= 3;
913 if (len > 0)
914 mch_memmove(buf, buf + 3, (size_t)len);
915 continue;
916 }
Bram Moolenaar7fc904b2006-04-13 20:37:35 +0000917 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000918 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000919 if (has_mbyte)
920 {
921 if (MB_BYTE2LEN(n) > len)
Bram Moolenaar85a20022019-12-21 18:25:54 +0100922 continue; // more bytes to get
Bram Moolenaara8c8a682012-02-05 22:05:48 +0100923 buf[len >= buflen ? buflen - 1 : len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000924 n = (*mb_ptr2char)(buf);
925 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000926#ifdef UNIX
927 if (n == intr_char)
928 n = ESC;
929#endif
930 break;
931 }
Bram Moolenaara8c8a682012-02-05 22:05:48 +0100932 vim_free(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000933
934 mapped_ctrl_c = save_mapped_ctrl_c;
935 return n;
936}
937
938/*
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +0000939 * Get a number from the user.
940 * When "mouse_used" is not NULL allow using the mouse.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000941 */
942 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100943get_number(
Bram Moolenaar85a20022019-12-21 18:25:54 +0100944 int colon, // allow colon to abort
Bram Moolenaar9b578142016-01-30 19:39:49 +0100945 int *mouse_used)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000946{
947 int n = 0;
948 int c;
Bram Moolenaar3991dab2006-03-27 17:01:56 +0000949 int typed = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000950
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +0000951 if (mouse_used != NULL)
952 *mouse_used = FALSE;
953
Bram Moolenaar85a20022019-12-21 18:25:54 +0100954 // When not printing messages, the user won't know what to type, return a
955 // zero (as if CR was hit).
Bram Moolenaar071d4272004-06-13 20:20:40 +0000956 if (msg_silent != 0)
957 return 0;
958
959#ifdef USE_ON_FLY_SCROLL
Bram Moolenaar85a20022019-12-21 18:25:54 +0100960 dont_scroll = TRUE; // disallow scrolling here
Bram Moolenaar071d4272004-06-13 20:20:40 +0000961#endif
962 ++no_mapping;
Bram Moolenaar85a20022019-12-21 18:25:54 +0100963 ++allow_keys; // no mapping here, but recognize keys
Bram Moolenaar071d4272004-06-13 20:20:40 +0000964 for (;;)
965 {
966 windgoto(msg_row, msg_col);
967 c = safe_vgetc();
968 if (VIM_ISDIGIT(c))
969 {
970 n = n * 10 + c - '0';
971 msg_putchar(c);
Bram Moolenaar3991dab2006-03-27 17:01:56 +0000972 ++typed;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000973 }
974 else if (c == K_DEL || c == K_KDEL || c == K_BS || c == Ctrl_H)
975 {
Bram Moolenaar3991dab2006-03-27 17:01:56 +0000976 if (typed > 0)
977 {
Bram Moolenaar32526b32019-01-19 17:43:09 +0100978 msg_puts("\b \b");
Bram Moolenaar3991dab2006-03-27 17:01:56 +0000979 --typed;
980 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000981 n /= 10;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000982 }
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +0000983 else if (mouse_used != NULL && c == K_LEFTMOUSE)
984 {
985 *mouse_used = TRUE;
986 n = mouse_row + 1;
987 break;
988 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000989 else if (n == 0 && c == ':' && colon)
990 {
991 stuffcharReadbuff(':');
992 if (!exmode_active)
993 cmdline_row = msg_row;
Bram Moolenaar85a20022019-12-21 18:25:54 +0100994 skip_redraw = TRUE; // skip redraw once
Bram Moolenaar071d4272004-06-13 20:20:40 +0000995 do_redraw = FALSE;
996 break;
997 }
=?UTF-8?q?Luka=20Marku=C5=A1i=C4=87?=5cf94572021-05-20 21:14:20 +0200998 else if (c == Ctrl_C || c == ESC || c == 'q')
999 {
1000 n = 0;
1001 break;
1002 }
1003 else if (c == CAR || c == NL )
Bram Moolenaar071d4272004-06-13 20:20:40 +00001004 break;
1005 }
1006 --no_mapping;
1007 --allow_keys;
1008 return n;
1009}
1010
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001011/*
1012 * Ask the user to enter a number.
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +00001013 * When "mouse_used" is not NULL allow using the mouse and in that case return
1014 * the line number.
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001015 */
1016 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01001017prompt_for_number(int *mouse_used)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001018{
1019 int i;
Bram Moolenaard857f0e2005-06-21 22:37:39 +00001020 int save_cmdline_row;
1021 int save_State;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001022
Bram Moolenaar85a20022019-12-21 18:25:54 +01001023 // When using ":silent" assume that <CR> was entered.
Bram Moolenaar42eeac32005-06-29 22:40:58 +00001024 if (mouse_used != NULL)
Bram Moolenaareebd5552020-06-10 15:45:57 +02001025 msg_puts(_("Type number and <Enter> or click with the mouse (q or empty cancels): "));
Bram Moolenaar42eeac32005-06-29 22:40:58 +00001026 else
Bram Moolenaareebd5552020-06-10 15:45:57 +02001027 msg_puts(_("Type number and <Enter> (q or empty cancels): "));
Bram Moolenaard857f0e2005-06-21 22:37:39 +00001028
Bram Moolenaar4cbdf152018-08-26 21:23:07 +02001029 // Set the state such that text can be selected/copied/pasted and we still
1030 // get mouse events. redraw_after_callback() will not redraw if cmdline_row
1031 // is zero.
Bram Moolenaard857f0e2005-06-21 22:37:39 +00001032 save_cmdline_row = cmdline_row;
Bram Moolenaar203335e2006-09-03 14:35:42 +00001033 cmdline_row = 0;
Bram Moolenaard857f0e2005-06-21 22:37:39 +00001034 save_State = State;
Bram Moolenaar24959102022-05-07 20:01:16 +01001035 State = MODE_CMDLINE;
Bram Moolenaar4cbdf152018-08-26 21:23:07 +02001036 // May show different mouse shape.
Bram Moolenaar73658312018-04-24 17:41:57 +02001037 setmouse();
Bram Moolenaar73658312018-04-24 17:41:57 +02001038
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +00001039 i = get_number(TRUE, mouse_used);
1040 if (KeyTyped)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001041 {
Bram Moolenaar954bb062019-06-09 16:40:46 +02001042 // don't call wait_return() now
1043 if (msg_row > 0)
1044 cmdline_row = msg_row - 1;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001045 need_wait_return = FALSE;
Bram Moolenaardc968e72019-11-05 21:53:20 +01001046 msg_didany = FALSE;
1047 msg_didout = FALSE;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001048 }
Bram Moolenaard857f0e2005-06-21 22:37:39 +00001049 else
1050 cmdline_row = save_cmdline_row;
1051 State = save_State;
Bram Moolenaar4cbdf152018-08-26 21:23:07 +02001052 // May need to restore mouse shape.
Bram Moolenaar73658312018-04-24 17:41:57 +02001053 setmouse();
Bram Moolenaard857f0e2005-06-21 22:37:39 +00001054
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001055 return i;
1056}
1057
Bram Moolenaar071d4272004-06-13 20:20:40 +00001058 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001059msgmore(long n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001060{
1061 long pn;
1062
Bram Moolenaar85a20022019-12-21 18:25:54 +01001063 if (global_busy // no messages now, wait until global is finished
1064 || !messaging()) // 'lazyredraw' set, don't do messages now
Bram Moolenaar071d4272004-06-13 20:20:40 +00001065 return;
1066
Bram Moolenaar85a20022019-12-21 18:25:54 +01001067 // We don't want to overwrite another important message, but do overwrite
1068 // a previous "more lines" or "fewer lines" message, so that "5dd" and
1069 // then "put" reports the last action.
Bram Moolenaar7df2d662005-01-25 22:18:08 +00001070 if (keep_msg != NULL && !keep_msg_more)
1071 return;
1072
Bram Moolenaar071d4272004-06-13 20:20:40 +00001073 if (n > 0)
1074 pn = n;
1075 else
1076 pn = -n;
1077
1078 if (pn > p_report)
1079 {
Bram Moolenaarda6e8912018-08-21 15:12:14 +02001080 if (n > 0)
Bram Moolenaar32526b32019-01-19 17:43:09 +01001081 vim_snprintf(msg_buf, MSG_BUF_LEN,
Bram Moolenaarda6e8912018-08-21 15:12:14 +02001082 NGETTEXT("%ld more line", "%ld more lines", pn), pn);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001083 else
Bram Moolenaar32526b32019-01-19 17:43:09 +01001084 vim_snprintf(msg_buf, MSG_BUF_LEN,
Bram Moolenaarda6e8912018-08-21 15:12:14 +02001085 NGETTEXT("%ld line less", "%ld fewer lines", pn), pn);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001086 if (got_int)
Bram Moolenaar32526b32019-01-19 17:43:09 +01001087 vim_strcat((char_u *)msg_buf, (char_u *)_(" (Interrupted)"),
1088 MSG_BUF_LEN);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001089 if (msg(msg_buf))
1090 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01001091 set_keep_msg((char_u *)msg_buf, 0);
Bram Moolenaar7df2d662005-01-25 22:18:08 +00001092 keep_msg_more = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001093 }
1094 }
1095}
1096
1097/*
1098 * flush map and typeahead buffers and give a warning for an error
1099 */
1100 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001101beep_flush(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001102{
1103 if (emsg_silent == 0)
1104 {
Bram Moolenaar6a2633b2018-10-07 23:16:36 +02001105 flush_buffers(FLUSH_MINIMAL);
Bram Moolenaar165bc692015-07-21 17:53:25 +02001106 vim_beep(BO_ERROR);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001107 }
1108}
1109
1110/*
Bram Moolenaaraae97622022-04-13 14:28:07 +01001111 * Give a warning for an error. "val" is one of the BO_ values, e.g., BO_OPER.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001112 */
1113 void
Bram Moolenaaraae97622022-04-13 14:28:07 +01001114vim_beep(unsigned val)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001115{
Bram Moolenaarb48e96f2018-02-13 12:26:14 +01001116#ifdef FEAT_EVAL
1117 called_vim_beep = TRUE;
1118#endif
1119
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001120 if (emsg_silent == 0 && !in_assert_fails)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001121 {
Bram Moolenaar165bc692015-07-21 17:53:25 +02001122 if (!((bo_flags & val) || (bo_flags & BO_ALL)))
1123 {
Bram Moolenaar2e147ca2017-06-27 17:09:37 +02001124#ifdef ELAPSED_FUNC
1125 static int did_init = FALSE;
Bram Moolenaar1ac56c22019-01-17 22:28:22 +01001126 static elapsed_T start_tv;
Bram Moolenaar2e147ca2017-06-27 17:09:37 +02001127
Bram Moolenaar85a20022019-12-21 18:25:54 +01001128 // Only beep once per half a second, otherwise a sequence of beeps
1129 // would freeze Vim.
Bram Moolenaar2e147ca2017-06-27 17:09:37 +02001130 if (!did_init || ELAPSED_FUNC(start_tv) > 500)
1131 {
1132 did_init = TRUE;
1133 ELAPSED_INIT(start_tv);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001134#endif
Bram Moolenaar2e147ca2017-06-27 17:09:37 +02001135 if (p_vb
1136#ifdef FEAT_GUI
Bram Moolenaar85a20022019-12-21 18:25:54 +01001137 // While the GUI is starting up the termcap is set for
1138 // the GUI but the output still goes to a terminal.
Bram Moolenaar2e147ca2017-06-27 17:09:37 +02001139 && !(gui.in_use && gui.starting)
1140#endif
1141 )
Bram Moolenaarcafafb32018-02-22 21:07:09 +01001142 {
Bram Moolenaar2e147ca2017-06-27 17:09:37 +02001143 out_str_cf(T_VB);
Bram Moolenaarcafafb32018-02-22 21:07:09 +01001144#ifdef FEAT_VTP
Bram Moolenaar85a20022019-12-21 18:25:54 +01001145 // No restore color information, refresh the screen.
Bram Moolenaarcafafb32018-02-22 21:07:09 +01001146 if (has_vtp_working() != 0
1147# ifdef FEAT_TERMGUICOLORS
Bram Moolenaarc5cd8852018-05-01 15:47:38 +02001148 && (p_tgc || (!p_tgc && t_colors >= 256))
Bram Moolenaarcafafb32018-02-22 21:07:09 +01001149# endif
1150 )
1151 {
1152 redraw_later(CLEAR);
1153 update_screen(0);
1154 redrawcmd();
1155 }
1156#endif
1157 }
Bram Moolenaar2e147ca2017-06-27 17:09:37 +02001158 else
1159 out_char(BELL);
1160#ifdef ELAPSED_FUNC
1161 }
1162#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001163 }
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00001164
Bram Moolenaar85a20022019-12-21 18:25:54 +01001165 // When 'debug' contains "beep" produce a message. If we are sourcing
1166 // a script or executing a function give the user a hint where the beep
1167 // comes from.
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00001168 if (vim_strchr(p_debug, 'e') != NULL)
1169 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01001170 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar32526b32019-01-19 17:43:09 +01001171 msg_attr(_("Beep!"), HL_ATTR(HLF_W));
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00001172 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001173 }
1174}
1175
1176/*
1177 * To get the "real" home directory:
1178 * - get value of $HOME
1179 * For Unix:
1180 * - go to that directory
1181 * - do mch_dirname() to get the real name of that directory.
1182 * This also works with mounts and links.
1183 * Don't do this for MS-DOS, it will change the "current dir" for a drive.
Bram Moolenaar25a494c2018-11-16 19:39:50 +01001184 * For Windows:
1185 * This code is duplicated in init_homedir() in dosinst.c. Keep in sync!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001186 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001187 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001188init_homedir(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001189{
1190 char_u *var;
1191
Bram Moolenaar85a20022019-12-21 18:25:54 +01001192 // In case we are called a second time (when 'encoding' changes).
Bram Moolenaard23a8232018-02-10 18:45:26 +01001193 VIM_CLEAR(homedir);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001194
Bram Moolenaar071d4272004-06-13 20:20:40 +00001195#ifdef VMS
1196 var = mch_getenv((char_u *)"SYS$LOGIN");
1197#else
1198 var = mch_getenv((char_u *)"HOME");
1199#endif
1200
Bram Moolenaar4f974752019-02-17 17:44:42 +01001201#ifdef MSWIN
Bram Moolenaar071d4272004-06-13 20:20:40 +00001202 /*
Bram Moolenaar48340b62017-08-29 22:08:53 +02001203 * Typically, $HOME is not defined on Windows, unless the user has
1204 * specifically defined it for Vim's sake. However, on Windows NT
1205 * platforms, $HOMEDRIVE and $HOMEPATH are automatically defined for
1206 * each user. Try constructing $HOME from these.
1207 */
Bram Moolenaarb47a2592017-08-30 13:22:28 +02001208 if (var == NULL || *var == NUL)
Bram Moolenaar48340b62017-08-29 22:08:53 +02001209 {
1210 char_u *homedrive, *homepath;
1211
1212 homedrive = mch_getenv((char_u *)"HOMEDRIVE");
1213 homepath = mch_getenv((char_u *)"HOMEPATH");
1214 if (homepath == NULL || *homepath == NUL)
1215 homepath = (char_u *)"\\";
1216 if (homedrive != NULL
1217 && STRLEN(homedrive) + STRLEN(homepath) < MAXPATHL)
1218 {
1219 sprintf((char *)NameBuff, "%s%s", homedrive, homepath);
1220 if (NameBuff[0] != NUL)
1221 var = NameBuff;
1222 }
1223 }
1224
1225 if (var == NULL)
1226 var = mch_getenv((char_u *)"USERPROFILE");
1227
1228 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001229 * Weird but true: $HOME may contain an indirect reference to another
1230 * variable, esp. "%USERPROFILE%". Happens when $USERPROFILE isn't set
1231 * when $HOME is being set.
1232 */
1233 if (var != NULL && *var == '%')
1234 {
1235 char_u *p;
1236 char_u *exp;
1237
1238 p = vim_strchr(var + 1, '%');
1239 if (p != NULL)
1240 {
Bram Moolenaarce0842a2005-07-18 21:58:11 +00001241 vim_strncpy(NameBuff, var + 1, p - (var + 1));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001242 exp = mch_getenv(NameBuff);
1243 if (exp != NULL && *exp != NUL
1244 && STRLEN(exp) + STRLEN(p) < MAXPATHL)
1245 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00001246 vim_snprintf((char *)NameBuff, MAXPATHL, "%s%s", exp, p + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001247 var = NameBuff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001248 }
1249 }
1250 }
1251
Bram Moolenaar85a20022019-12-21 18:25:54 +01001252 if (var != NULL && *var == NUL) // empty is same as not set
Bram Moolenaar48340b62017-08-29 22:08:53 +02001253 var = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001254
Bram Moolenaar05159a02005-02-26 23:04:13 +00001255 if (enc_utf8 && var != NULL)
1256 {
1257 int len;
Bram Moolenaarb453a532011-04-28 17:48:44 +02001258 char_u *pp = NULL;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001259
Bram Moolenaar85a20022019-12-21 18:25:54 +01001260 // Convert from active codepage to UTF-8. Other conversions are
1261 // not done, because they would fail for non-ASCII characters.
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001262 acp_to_enc(var, (int)STRLEN(var), &pp, &len);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001263 if (pp != NULL)
1264 {
1265 homedir = pp;
1266 return;
1267 }
1268 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001269
Bram Moolenaar071d4272004-06-13 20:20:40 +00001270 /*
1271 * Default home dir is C:/
1272 * Best assumption we can make in such a situation.
1273 */
1274 if (var == NULL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001275 var = (char_u *)"C:/";
Bram Moolenaar071d4272004-06-13 20:20:40 +00001276#endif
Bram Moolenaar48340b62017-08-29 22:08:53 +02001277
Bram Moolenaar071d4272004-06-13 20:20:40 +00001278 if (var != NULL)
1279 {
1280#ifdef UNIX
1281 /*
1282 * Change to the directory and get the actual path. This resolves
1283 * links. Don't do it when we can't return.
1284 */
1285 if (mch_dirname(NameBuff, MAXPATHL) == OK
1286 && mch_chdir((char *)NameBuff) == 0)
1287 {
1288 if (!mch_chdir((char *)var) && mch_dirname(IObuff, IOSIZE) == OK)
1289 var = IObuff;
1290 if (mch_chdir((char *)NameBuff) != 0)
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001291 emsg(_(e_cannot_go_back_to_previous_directory));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001292 }
1293#endif
1294 homedir = vim_strsave(var);
1295 }
1296}
1297
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001298#if defined(EXITFREE) || defined(PROTO)
1299 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001300free_homedir(void)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001301{
1302 vim_free(homedir);
1303}
Bram Moolenaar24305862012-08-15 14:05:05 +02001304
Bram Moolenaar24305862012-08-15 14:05:05 +02001305 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001306free_users(void)
Bram Moolenaar24305862012-08-15 14:05:05 +02001307{
1308 ga_clear_strings(&ga_users);
1309}
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001310#endif
1311
Bram Moolenaar071d4272004-06-13 20:20:40 +00001312/*
Bram Moolenaar9f0545d2007-09-26 20:36:32 +00001313 * Call expand_env() and store the result in an allocated string.
1314 * This is not very memory efficient, this expects the result to be freed
1315 * again soon.
1316 */
1317 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001318expand_env_save(char_u *src)
Bram Moolenaar9f0545d2007-09-26 20:36:32 +00001319{
1320 return expand_env_save_opt(src, FALSE);
1321}
1322
1323/*
1324 * Idem, but when "one" is TRUE handle the string as one file name, only
1325 * expand "~" at the start.
1326 */
1327 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001328expand_env_save_opt(char_u *src, int one)
Bram Moolenaar9f0545d2007-09-26 20:36:32 +00001329{
1330 char_u *p;
1331
1332 p = alloc(MAXPATHL);
1333 if (p != NULL)
1334 expand_env_esc(src, p, MAXPATHL, FALSE, one, NULL);
1335 return p;
1336}
1337
1338/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001339 * Expand environment variable with path name.
1340 * "~/" is also expanded, using $HOME. For Unix "~user/" is expanded.
Bram Moolenaar9f0545d2007-09-26 20:36:32 +00001341 * Skips over "\ ", "\~" and "\$" (not for Win32 though).
Bram Moolenaar071d4272004-06-13 20:20:40 +00001342 * If anything fails no expansion is done and dst equals src.
1343 */
1344 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001345expand_env(
Bram Moolenaar85a20022019-12-21 18:25:54 +01001346 char_u *src, // input string e.g. "$HOME/vim.hlp"
1347 char_u *dst, // where to put the result
1348 int dstlen) // maximum length of the result
Bram Moolenaar071d4272004-06-13 20:20:40 +00001349{
Bram Moolenaar9f0545d2007-09-26 20:36:32 +00001350 expand_env_esc(src, dst, dstlen, FALSE, FALSE, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001351}
1352
1353 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001354expand_env_esc(
Bram Moolenaar85a20022019-12-21 18:25:54 +01001355 char_u *srcp, // input string e.g. "$HOME/vim.hlp"
1356 char_u *dst, // where to put the result
1357 int dstlen, // maximum length of the result
1358 int esc, // escape spaces in expanded variables
1359 int one, // "srcp" is one file name
1360 char_u *startstr) // start again after this (can be NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001361{
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +00001362 char_u *src;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001363 char_u *tail;
1364 int c;
1365 char_u *var;
1366 int copy_char;
Bram Moolenaar85a20022019-12-21 18:25:54 +01001367 int mustfree; // var was allocated, need to free it later
1368 int at_start = TRUE; // at start of a name
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +00001369 int startstr_len = 0;
Yegappan Lakshmanana34b4462022-06-11 10:43:26 +01001370#if defined(BACKSLASH_IN_FILENAME) || defined(AMIGA)
1371 char_u *save_dst = dst;
1372#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001373
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +00001374 if (startstr != NULL)
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001375 startstr_len = (int)STRLEN(startstr);
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +00001376
1377 src = skipwhite(srcp);
Bram Moolenaar85a20022019-12-21 18:25:54 +01001378 --dstlen; // leave one char space for "\,"
Bram Moolenaar071d4272004-06-13 20:20:40 +00001379 while (*src && dstlen > 0)
1380 {
Bram Moolenaarbe83b732015-08-25 14:21:19 +02001381#ifdef FEAT_EVAL
Bram Moolenaar85a20022019-12-21 18:25:54 +01001382 // Skip over `=expr`.
Bram Moolenaarbe83b732015-08-25 14:21:19 +02001383 if (src[0] == '`' && src[1] == '=')
1384 {
1385 size_t len;
1386
1387 var = src;
1388 src += 2;
Bram Moolenaar683581e2020-10-22 21:22:58 +02001389 (void)skip_expr(&src, NULL);
Bram Moolenaarbe83b732015-08-25 14:21:19 +02001390 if (*src == '`')
1391 ++src;
1392 len = src - var;
1393 if (len > (size_t)dstlen)
1394 len = dstlen;
1395 vim_strncpy(dst, var, len);
1396 dst += len;
Bram Moolenaar5df1ed22015-09-01 16:25:34 +02001397 dstlen -= (int)len;
Bram Moolenaarbe83b732015-08-25 14:21:19 +02001398 continue;
1399 }
1400#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001401 copy_char = TRUE;
Bram Moolenaard4755bb2004-09-02 19:12:26 +00001402 if ((*src == '$'
1403#ifdef VMS
1404 && at_start
1405#endif
1406 )
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001407#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001408 || *src == '%'
1409#endif
1410 || (*src == '~' && at_start))
1411 {
1412 mustfree = FALSE;
1413
1414 /*
1415 * The variable name is copied into dst temporarily, because it may
1416 * be a string in read-only memory and a NUL needs to be appended.
1417 */
Bram Moolenaar85a20022019-12-21 18:25:54 +01001418 if (*src != '~') // environment var
Bram Moolenaar071d4272004-06-13 20:20:40 +00001419 {
1420 tail = src + 1;
1421 var = dst;
1422 c = dstlen - 1;
1423
1424#ifdef UNIX
Bram Moolenaar85a20022019-12-21 18:25:54 +01001425 // Unix has ${var-name} type environment vars
Bram Moolenaar071d4272004-06-13 20:20:40 +00001426 if (*tail == '{' && !vim_isIDc('{'))
1427 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001428 tail++; // ignore '{'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001429 while (c-- > 0 && *tail && *tail != '}')
1430 *var++ = *tail++;
1431 }
1432 else
1433#endif
1434 {
1435 while (c-- > 0 && *tail != NUL && ((vim_isIDc(*tail))
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001436#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001437 || (*src == '%' && *tail != '%')
1438#endif
1439 ))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001440 *var++ = *tail++;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001441 }
1442
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001443#if defined(MSWIN) || defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001444# ifdef UNIX
1445 if (src[1] == '{' && *tail != '}')
1446# else
1447 if (*src == '%' && *tail != '%')
1448# endif
1449 var = NULL;
1450 else
1451 {
1452# ifdef UNIX
1453 if (src[1] == '{')
1454# else
1455 if (*src == '%')
1456#endif
1457 ++tail;
1458#endif
1459 *var = NUL;
1460 var = vim_getenv(dst, &mustfree);
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001461#if defined(MSWIN) || defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001462 }
1463#endif
1464 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01001465 // home directory
Bram Moolenaar071d4272004-06-13 20:20:40 +00001466 else if ( src[1] == NUL
1467 || vim_ispathsep(src[1])
1468 || vim_strchr((char_u *)" ,\t\n", src[1]) != NULL)
1469 {
1470 var = homedir;
1471 tail = src + 1;
1472 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01001473 else // user directory
Bram Moolenaar071d4272004-06-13 20:20:40 +00001474 {
1475#if defined(UNIX) || (defined(VMS) && defined(USER_HOME))
1476 /*
1477 * Copy ~user to dst[], so we can put a NUL after it.
1478 */
1479 tail = src;
1480 var = dst;
1481 c = dstlen - 1;
1482 while ( c-- > 0
1483 && *tail
1484 && vim_isfilec(*tail)
1485 && !vim_ispathsep(*tail))
1486 *var++ = *tail++;
1487 *var = NUL;
1488# ifdef UNIX
1489 /*
1490 * If the system supports getpwnam(), use it.
1491 * Otherwise, or if getpwnam() fails, the shell is used to
1492 * expand ~user. This is slower and may fail if the shell
1493 * does not support ~user (old versions of /bin/sh).
1494 */
1495# if defined(HAVE_GETPWNAM) && defined(HAVE_PWD_H)
1496 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001497 // Note: memory allocated by getpwnam() is never freed.
1498 // Calling endpwent() apparently doesn't help.
Bram Moolenaar187a4f22017-02-23 17:07:14 +01001499 struct passwd *pw = (*dst == NUL)
1500 ? NULL : getpwnam((char *)dst + 1);
1501
1502 var = (pw == NULL) ? NULL : (char_u *)pw->pw_dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001503 }
1504 if (var == NULL)
1505# endif
1506 {
1507 expand_T xpc;
1508
1509 ExpandInit(&xpc);
1510 xpc.xp_context = EXPAND_FILES;
1511 var = ExpandOne(&xpc, dst, NULL,
1512 WILD_ADD_SLASH|WILD_SILENT, WILD_EXPAND_FREE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001513 mustfree = TRUE;
1514 }
1515
Bram Moolenaar85a20022019-12-21 18:25:54 +01001516# else // !UNIX, thus VMS
Bram Moolenaar071d4272004-06-13 20:20:40 +00001517 /*
1518 * USER_HOME is a comma-separated list of
1519 * directories to search for the user account in.
1520 */
1521 {
1522 char_u test[MAXPATHL], paths[MAXPATHL];
1523 char_u *path, *next_path, *ptr;
Bram Moolenaar8767f522016-07-01 17:17:39 +02001524 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001525
1526 STRCPY(paths, USER_HOME);
1527 next_path = paths;
1528 while (*next_path)
1529 {
1530 for (path = next_path; *next_path && *next_path != ',';
1531 next_path++);
1532 if (*next_path)
1533 *next_path++ = NUL;
1534 STRCPY(test, path);
1535 STRCAT(test, "/");
1536 STRCAT(test, dst + 1);
1537 if (mch_stat(test, &st) == 0)
1538 {
1539 var = alloc(STRLEN(test) + 1);
1540 STRCPY(var, test);
1541 mustfree = TRUE;
1542 break;
1543 }
1544 }
1545 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01001546# endif // UNIX
Bram Moolenaar071d4272004-06-13 20:20:40 +00001547#else
Bram Moolenaar85a20022019-12-21 18:25:54 +01001548 // cannot expand user's home directory, so don't try
Bram Moolenaar071d4272004-06-13 20:20:40 +00001549 var = NULL;
Bram Moolenaar85a20022019-12-21 18:25:54 +01001550 tail = (char_u *)""; // for gcc
1551#endif // UNIX || VMS
Bram Moolenaar071d4272004-06-13 20:20:40 +00001552 }
1553
1554#ifdef BACKSLASH_IN_FILENAME
Bram Moolenaar85a20022019-12-21 18:25:54 +01001555 // If 'shellslash' is set change backslashes to forward slashes.
1556 // Can't use slash_adjust(), p_ssl may be set temporarily.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001557 if (p_ssl && var != NULL && vim_strchr(var, '\\') != NULL)
1558 {
1559 char_u *p = vim_strsave(var);
1560
1561 if (p != NULL)
1562 {
1563 if (mustfree)
1564 vim_free(var);
1565 var = p;
1566 mustfree = TRUE;
1567 forward_slash(var);
1568 }
1569 }
1570#endif
1571
Bram Moolenaar85a20022019-12-21 18:25:54 +01001572 // If "var" contains white space, escape it with a backslash.
1573 // Required for ":e ~/tt" when $HOME includes a space.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001574 if (esc && var != NULL && vim_strpbrk(var, (char_u *)" \t") != NULL)
1575 {
1576 char_u *p = vim_strsave_escaped(var, (char_u *)" \t");
1577
1578 if (p != NULL)
1579 {
1580 if (mustfree)
1581 vim_free(var);
1582 var = p;
1583 mustfree = TRUE;
1584 }
1585 }
1586
1587 if (var != NULL && *var != NUL
1588 && (STRLEN(var) + STRLEN(tail) + 1 < (unsigned)dstlen))
1589 {
1590 STRCPY(dst, var);
1591 dstlen -= (int)STRLEN(var);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001592 c = (int)STRLEN(var);
Bram Moolenaar85a20022019-12-21 18:25:54 +01001593 // if var[] ends in a path separator and tail[] starts
1594 // with it, skip a character
=?UTF-8?q?Dundar=20G=C3=B6c?=b8366582022-04-14 20:43:56 +01001595 if (after_pathsep(dst, dst + c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001596#if defined(BACKSLASH_IN_FILENAME) || defined(AMIGA)
Yegappan Lakshmanana34b4462022-06-11 10:43:26 +01001597 && (dst == save_dst || dst[-1] != ':')
Bram Moolenaar071d4272004-06-13 20:20:40 +00001598#endif
1599 && vim_ispathsep(*tail))
1600 ++tail;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001601 dst += c;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001602 src = tail;
1603 copy_char = FALSE;
1604 }
1605 if (mustfree)
1606 vim_free(var);
1607 }
1608
Bram Moolenaar85a20022019-12-21 18:25:54 +01001609 if (copy_char) // copy at least one char
Bram Moolenaar071d4272004-06-13 20:20:40 +00001610 {
1611 /*
Bram Moolenaar25394022007-05-10 19:06:20 +00001612 * Recognize the start of a new name, for '~'.
Bram Moolenaar9f0545d2007-09-26 20:36:32 +00001613 * Don't do this when "one" is TRUE, to avoid expanding "~" in
1614 * ":edit foo ~ foo".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001615 */
1616 at_start = FALSE;
1617 if (src[0] == '\\' && src[1] != NUL)
1618 {
1619 *dst++ = *src++;
1620 --dstlen;
1621 }
Bram Moolenaar9f0545d2007-09-26 20:36:32 +00001622 else if ((src[0] == ' ' || src[0] == ',') && !one)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001623 at_start = TRUE;
Bram Moolenaar1c864092017-08-06 18:15:45 +02001624 if (dstlen > 0)
1625 {
1626 *dst++ = *src++;
1627 --dstlen;
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +00001628
Bram Moolenaar1c864092017-08-06 18:15:45 +02001629 if (startstr != NULL && src - startstr_len >= srcp
1630 && STRNCMP(src - startstr_len, startstr,
1631 startstr_len) == 0)
1632 at_start = TRUE;
1633 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001634 }
Bram Moolenaar1c864092017-08-06 18:15:45 +02001635
Bram Moolenaar071d4272004-06-13 20:20:40 +00001636 }
1637 *dst = NUL;
1638}
1639
1640/*
Bram Moolenaar26262f82019-09-04 20:59:15 +02001641 * If the string between "p" and "pend" ends in "name/", return "pend" minus
1642 * the length of "name/". Otherwise return "pend".
1643 */
1644 static char_u *
1645remove_tail(char_u *p, char_u *pend, char_u *name)
1646{
1647 int len = (int)STRLEN(name) + 1;
1648 char_u *newend = pend - len;
1649
1650 if (newend >= p
1651 && fnamencmp(newend, name, len - 1) == 0
1652 && (newend == p || after_pathsep(p, newend)))
1653 return newend;
1654 return pend;
1655}
1656
1657/*
1658 * Check if the directory "vimdir/<version>" or "vimdir/runtime" exists.
1659 * Return NULL if not, return its name in allocated memory otherwise.
1660 */
1661 static char_u *
1662vim_version_dir(char_u *vimdir)
1663{
1664 char_u *p;
1665
1666 if (vimdir == NULL || *vimdir == NUL)
1667 return NULL;
1668 p = concat_fnames(vimdir, (char_u *)VIM_VERSION_NODOT, TRUE);
1669 if (p != NULL && mch_isdir(p))
1670 return p;
1671 vim_free(p);
1672 p = concat_fnames(vimdir, (char_u *)RUNTIME_DIRNAME, TRUE);
1673 if (p != NULL && mch_isdir(p))
Bram Moolenaar022f9ef2022-07-02 17:36:31 +01001674 {
1675 char_u *fname = concat_fnames(p, (char_u *)"defaults.vim", TRUE);
1676
1677 // Check that "defaults.vim" exists in this directory, to avoid picking
1678 // up a stray "runtime" directory, it would make many tests fail in
1679 // mysterious ways.
1680 if (fname != NULL)
1681 {
1682 int exists = file_is_readable(fname);
1683
1684 vim_free(fname);
1685 if (exists)
1686 return p;
1687 }
1688 }
Bram Moolenaar26262f82019-09-04 20:59:15 +02001689 vim_free(p);
1690 return NULL;
1691}
1692
1693/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001694 * Vim's version of getenv().
1695 * Special handling of $HOME, $VIM and $VIMRUNTIME.
Bram Moolenaar2f6b0b82005-03-08 22:43:10 +00001696 * Also does ACP to 'enc' conversion for Win32.
Bram Moolenaarb453a532011-04-28 17:48:44 +02001697 * "mustfree" is set to TRUE when returned is allocated, it must be
1698 * initialized to FALSE by the caller.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001699 */
1700 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001701vim_getenv(char_u *name, int *mustfree)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001702{
Bram Moolenaarf0908e62019-03-30 20:11:50 +01001703 char_u *p = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001704 char_u *pend;
1705 int vimruntime;
Bram Moolenaarf0908e62019-03-30 20:11:50 +01001706#ifdef MSWIN
1707 WCHAR *wn, *wp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001708
Bram Moolenaarf0908e62019-03-30 20:11:50 +01001709 // use "C:/" when $HOME is not set
Bram Moolenaar071d4272004-06-13 20:20:40 +00001710 if (STRCMP(name, "HOME") == 0)
1711 return homedir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001712
Bram Moolenaarf0908e62019-03-30 20:11:50 +01001713 // Use Wide function
1714 wn = enc_to_utf16(name, NULL);
1715 if (wn == NULL)
1716 return NULL;
1717
1718 wp = _wgetenv(wn);
1719 vim_free(wn);
1720
1721 if (wp != NULL && *wp == NUL) // empty is the same as not set
1722 wp = NULL;
1723
1724 if (wp != NULL)
1725 {
1726 p = utf16_to_enc(wp, NULL);
1727 if (p == NULL)
1728 return NULL;
1729
1730 *mustfree = TRUE;
1731 return p;
1732 }
1733#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001734 p = mch_getenv(name);
Bram Moolenaarf0908e62019-03-30 20:11:50 +01001735 if (p != NULL && *p == NUL) // empty is the same as not set
Bram Moolenaar071d4272004-06-13 20:20:40 +00001736 p = NULL;
1737
1738 if (p != NULL)
1739 return p;
Bram Moolenaar80a8d382020-05-03 22:57:32 +02001740
1741# ifdef __HAIKU__
1742 // special handling for user settings directory...
1743 if (STRCMP(name, "BE_USER_SETTINGS") == 0)
1744 {
1745 static char userSettingsPath[MAXPATHL];
1746
1747 if (find_directory(B_USER_SETTINGS_DIRECTORY, 0, false,
1748 userSettingsPath, MAXPATHL) == B_OK)
1749 return (char_u *)userSettingsPath;
1750 else
1751 return NULL;
1752 }
1753# endif
Bram Moolenaarf0908e62019-03-30 20:11:50 +01001754#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001755
Bram Moolenaarf0908e62019-03-30 20:11:50 +01001756 // handling $VIMRUNTIME and $VIM is below, bail out if it's another name.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001757 vimruntime = (STRCMP(name, "VIMRUNTIME") == 0);
1758 if (!vimruntime && STRCMP(name, "VIM") != 0)
1759 return NULL;
1760
1761 /*
1762 * When expanding $VIMRUNTIME fails, try using $VIM/vim<version> or $VIM.
1763 * Don't do this when default_vimruntime_dir is non-empty.
1764 */
1765 if (vimruntime
1766#ifdef HAVE_PATHDEF
1767 && *default_vimruntime_dir == NUL
1768#endif
1769 )
1770 {
Bram Moolenaarf0908e62019-03-30 20:11:50 +01001771#ifdef MSWIN
1772 // Use Wide function
1773 wp = _wgetenv(L"VIM");
1774 if (wp != NULL && *wp == NUL) // empty is the same as not set
1775 wp = NULL;
1776 if (wp != NULL)
1777 {
1778 char_u *q = utf16_to_enc(wp, NULL);
1779 if (q != NULL)
1780 {
1781 p = vim_version_dir(q);
1782 *mustfree = TRUE;
1783 if (p == NULL)
1784 p = q;
1785 }
1786 }
1787#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001788 p = mch_getenv((char_u *)"VIM");
Bram Moolenaarf0908e62019-03-30 20:11:50 +01001789 if (p != NULL && *p == NUL) // empty is the same as not set
Bram Moolenaar071d4272004-06-13 20:20:40 +00001790 p = NULL;
1791 if (p != NULL)
1792 {
1793 p = vim_version_dir(p);
1794 if (p != NULL)
1795 *mustfree = TRUE;
1796 else
1797 p = mch_getenv((char_u *)"VIM");
1798 }
Bram Moolenaarf0908e62019-03-30 20:11:50 +01001799#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001800 }
1801
1802 /*
1803 * When expanding $VIM or $VIMRUNTIME fails, try using:
1804 * - the directory name from 'helpfile' (unless it contains '$')
1805 * - the executable name from argv[0]
1806 */
1807 if (p == NULL)
1808 {
1809 if (p_hf != NULL && vim_strchr(p_hf, '$') == NULL)
1810 p = p_hf;
1811#ifdef USE_EXE_NAME
1812 /*
1813 * Use the name of the executable, obtained from argv[0].
1814 */
1815 else
1816 p = exe_name;
1817#endif
1818 if (p != NULL)
1819 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001820 // remove the file name
Bram Moolenaar071d4272004-06-13 20:20:40 +00001821 pend = gettail(p);
1822
Bram Moolenaar85a20022019-12-21 18:25:54 +01001823 // remove "doc/" from 'helpfile', if present
Bram Moolenaar071d4272004-06-13 20:20:40 +00001824 if (p == p_hf)
1825 pend = remove_tail(p, pend, (char_u *)"doc");
1826
1827#ifdef USE_EXE_NAME
1828# ifdef MACOS_X
Bram Moolenaar85a20022019-12-21 18:25:54 +01001829 // remove "MacOS" from exe_name and add "Resources/vim"
Bram Moolenaar071d4272004-06-13 20:20:40 +00001830 if (p == exe_name)
1831 {
1832 char_u *pend1;
Bram Moolenaar95e9b492006-03-15 23:04:43 +00001833 char_u *pnew;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001834
Bram Moolenaar95e9b492006-03-15 23:04:43 +00001835 pend1 = remove_tail(p, pend, (char_u *)"MacOS");
1836 if (pend1 != pend)
1837 {
Bram Moolenaar964b3742019-05-24 18:54:09 +02001838 pnew = alloc(pend1 - p + 15);
Bram Moolenaar95e9b492006-03-15 23:04:43 +00001839 if (pnew != NULL)
1840 {
1841 STRNCPY(pnew, p, (pend1 - p));
1842 STRCPY(pnew + (pend1 - p), "Resources/vim");
1843 p = pnew;
1844 pend = p + STRLEN(p);
1845 }
1846 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001847 }
1848# endif
Bram Moolenaar85a20022019-12-21 18:25:54 +01001849 // remove "src/" from exe_name, if present
Bram Moolenaar071d4272004-06-13 20:20:40 +00001850 if (p == exe_name)
1851 pend = remove_tail(p, pend, (char_u *)"src");
1852#endif
1853
Bram Moolenaar85a20022019-12-21 18:25:54 +01001854 // for $VIM, remove "runtime/" or "vim54/", if present
Bram Moolenaar071d4272004-06-13 20:20:40 +00001855 if (!vimruntime)
1856 {
1857 pend = remove_tail(p, pend, (char_u *)RUNTIME_DIRNAME);
1858 pend = remove_tail(p, pend, (char_u *)VIM_VERSION_NODOT);
1859 }
1860
Bram Moolenaar85a20022019-12-21 18:25:54 +01001861 // remove trailing path separator
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001862 if (pend > p && after_pathsep(p, pend))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001863 --pend;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001864
Bram Moolenaar95e9b492006-03-15 23:04:43 +00001865#ifdef MACOS_X
1866 if (p == exe_name || p == p_hf)
1867#endif
Bram Moolenaar85a20022019-12-21 18:25:54 +01001868 // check that the result is a directory name
Bram Moolenaar71ccd032020-06-12 22:59:11 +02001869 p = vim_strnsave(p, pend - p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001870
1871 if (p != NULL && !mch_isdir(p))
Bram Moolenaard23a8232018-02-10 18:45:26 +01001872 VIM_CLEAR(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001873 else
1874 {
1875#ifdef USE_EXE_NAME
Bram Moolenaar85a20022019-12-21 18:25:54 +01001876 // may add "/vim54" or "/runtime" if it exists
Bram Moolenaar071d4272004-06-13 20:20:40 +00001877 if (vimruntime && (pend = vim_version_dir(p)) != NULL)
1878 {
1879 vim_free(p);
1880 p = pend;
1881 }
1882#endif
1883 *mustfree = TRUE;
1884 }
1885 }
1886 }
1887
1888#ifdef HAVE_PATHDEF
Bram Moolenaar85a20022019-12-21 18:25:54 +01001889 // When there is a pathdef.c file we can use default_vim_dir and
1890 // default_vimruntime_dir
Bram Moolenaar071d4272004-06-13 20:20:40 +00001891 if (p == NULL)
1892 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001893 // Only use default_vimruntime_dir when it is not empty
Bram Moolenaar071d4272004-06-13 20:20:40 +00001894 if (vimruntime && *default_vimruntime_dir != NUL)
1895 {
1896 p = default_vimruntime_dir;
1897 *mustfree = FALSE;
1898 }
1899 else if (*default_vim_dir != NUL)
1900 {
1901 if (vimruntime && (p = vim_version_dir(default_vim_dir)) != NULL)
1902 *mustfree = TRUE;
1903 else
1904 {
1905 p = default_vim_dir;
1906 *mustfree = FALSE;
1907 }
1908 }
1909 }
1910#endif
1911
1912 /*
1913 * Set the environment variable, so that the new value can be found fast
1914 * next time, and others can also use it (e.g. Perl).
1915 */
1916 if (p != NULL)
1917 {
1918 if (vimruntime)
1919 {
1920 vim_setenv((char_u *)"VIMRUNTIME", p);
1921 didset_vimruntime = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001922 }
1923 else
1924 {
1925 vim_setenv((char_u *)"VIM", p);
1926 didset_vim = TRUE;
1927 }
1928 }
1929 return p;
1930}
1931
Bram Moolenaar137374f2018-05-13 15:59:50 +02001932 void
1933vim_unsetenv(char_u *var)
1934{
1935#ifdef HAVE_UNSETENV
1936 unsetenv((char *)var);
1937#else
Bram Moolenaar1af6a4b2018-05-14 22:58:34 +02001938 vim_setenv(var, (char_u *)"");
Bram Moolenaar137374f2018-05-13 15:59:50 +02001939#endif
1940}
1941
LemonBoy77142312022-04-15 20:50:46 +01001942/*
1943 * Removes environment variable "name" and take care of side effects.
1944 */
1945 void
1946vim_unsetenv_ext(char_u *var)
1947{
1948 vim_unsetenv(var);
1949
1950 // "homedir" is not cleared, keep using the old value until $HOME is set.
1951 if (STRICMP(var, "VIM") == 0)
1952 didset_vim = FALSE;
1953 else if (STRICMP(var, "VIMRUNTIME") == 0)
1954 didset_vimruntime = FALSE;
1955}
Bram Moolenaar137374f2018-05-13 15:59:50 +02001956
Bram Moolenaar092e09c2022-04-15 23:29:23 +01001957#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001958/*
Bram Moolenaarb283a8a2020-02-02 22:24:04 +01001959 * Set environment variable "name" and take care of side effects.
1960 */
1961 void
1962vim_setenv_ext(char_u *name, char_u *val)
1963{
1964 vim_setenv(name, val);
1965 if (STRICMP(name, "HOME") == 0)
1966 init_homedir();
1967 else if (didset_vim && STRICMP(name, "VIM") == 0)
1968 didset_vim = FALSE;
LemonBoy77142312022-04-15 20:50:46 +01001969 else if (didset_vimruntime && STRICMP(name, "VIMRUNTIME") == 0)
Bram Moolenaarb283a8a2020-02-02 22:24:04 +01001970 didset_vimruntime = FALSE;
1971}
Dominique Pelle748b3082022-01-08 12:41:16 +00001972#endif
Bram Moolenaarb283a8a2020-02-02 22:24:04 +01001973
1974/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001975 * Our portable version of setenv.
1976 */
1977 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001978vim_setenv(char_u *name, char_u *val)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001979{
1980#ifdef HAVE_SETENV
1981 mch_setenv((char *)name, (char *)val, 1);
1982#else
1983 char_u *envbuf;
1984
1985 /*
1986 * Putenv does not copy the string, it has to remain
1987 * valid. The allocated memory will never be freed.
1988 */
Bram Moolenaar964b3742019-05-24 18:54:09 +02001989 envbuf = alloc(STRLEN(name) + STRLEN(val) + 2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001990 if (envbuf != NULL)
1991 {
1992 sprintf((char *)envbuf, "%s=%s", name, val);
1993 putenv((char *)envbuf);
1994 }
1995#endif
Bram Moolenaar011a34d2012-02-29 13:49:09 +01001996#ifdef FEAT_GETTEXT
1997 /*
1998 * When setting $VIMRUNTIME adjust the directory to find message
1999 * translations to $VIMRUNTIME/lang.
2000 */
2001 if (*val != NUL && STRICMP(name, "VIMRUNTIME") == 0)
2002 {
2003 char_u *buf = concat_str(val, (char_u *)"/lang");
2004
2005 if (buf != NULL)
2006 {
2007 bindtextdomain(VIMPACKAGE, (char *)buf);
2008 vim_free(buf);
2009 }
2010 }
2011#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002012}
2013
Bram Moolenaar071d4272004-06-13 20:20:40 +00002014/*
2015 * Function given to ExpandGeneric() to obtain an environment variable name.
2016 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002017 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01002018get_env_name(
2019 expand_T *xp UNUSED,
2020 int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002021{
Bram Moolenaard0573012017-10-28 21:11:06 +02002022# if defined(AMIGA)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002023 /*
Bram Moolenaard0573012017-10-28 21:11:06 +02002024 * No environ[] on the Amiga.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002025 */
2026 return NULL;
2027# else
2028# ifndef __WIN32__
Bram Moolenaar85a20022019-12-21 18:25:54 +01002029 // Borland C++ 5.2 has this in a header file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002030 extern char **environ;
2031# endif
Bram Moolenaar21cf8232004-07-16 20:18:37 +00002032# define ENVNAMELEN 100
2033 static char_u name[ENVNAMELEN];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002034 char_u *str;
2035 int n;
2036
2037 str = (char_u *)environ[idx];
2038 if (str == NULL)
2039 return NULL;
2040
Bram Moolenaar21cf8232004-07-16 20:18:37 +00002041 for (n = 0; n < ENVNAMELEN - 1; ++n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002042 {
2043 if (str[n] == '=' || str[n] == NUL)
2044 break;
2045 name[n] = str[n];
2046 }
2047 name[n] = NUL;
2048 return name;
2049# endif
2050}
Bram Moolenaar24305862012-08-15 14:05:05 +02002051
2052/*
Bram Moolenaar6b0b83f2018-09-10 19:03:05 +02002053 * Add a user name to the list of users in ga_users.
2054 * Do nothing if user name is NULL or empty.
2055 */
2056 static void
2057add_user(char_u *user, int need_copy)
2058{
2059 char_u *user_copy = (user != NULL && need_copy)
2060 ? vim_strsave(user) : user;
2061
2062 if (user_copy == NULL || *user_copy == NUL || ga_grow(&ga_users, 1) == FAIL)
2063 {
2064 if (need_copy)
2065 vim_free(user);
2066 return;
2067 }
2068 ((char_u **)(ga_users.ga_data))[ga_users.ga_len++] = user_copy;
2069}
2070
2071/*
Bram Moolenaar24305862012-08-15 14:05:05 +02002072 * Find all user names for user completion.
2073 * Done only once and then cached.
2074 */
2075 static void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002076init_users(void)
Bram Moolenaar01b626c2013-06-16 22:49:14 +02002077{
Bram Moolenaar24305862012-08-15 14:05:05 +02002078 static int lazy_init_done = FALSE;
2079
2080 if (lazy_init_done)
2081 return;
2082
2083 lazy_init_done = TRUE;
2084 ga_init2(&ga_users, sizeof(char_u *), 20);
2085
2086# if defined(HAVE_GETPWENT) && defined(HAVE_PWD_H)
2087 {
Bram Moolenaar24305862012-08-15 14:05:05 +02002088 struct passwd* pw;
2089
2090 setpwent();
2091 while ((pw = getpwent()) != NULL)
Bram Moolenaar6b0b83f2018-09-10 19:03:05 +02002092 add_user((char_u *)pw->pw_name, TRUE);
Bram Moolenaar24305862012-08-15 14:05:05 +02002093 endpwent();
2094 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01002095# elif defined(MSWIN)
Bram Moolenaar828c3d72018-06-19 18:58:07 +02002096 {
Bram Moolenaar828c3d72018-06-19 18:58:07 +02002097 DWORD nusers = 0, ntotal = 0, i;
2098 PUSER_INFO_0 uinfo;
2099
2100 if (NetUserEnum(NULL, 0, 0, (LPBYTE *) &uinfo, MAX_PREFERRED_LENGTH,
2101 &nusers, &ntotal, NULL) == NERR_Success)
2102 {
2103 for (i = 0; i < nusers; i++)
Bram Moolenaar6b0b83f2018-09-10 19:03:05 +02002104 add_user(utf16_to_enc(uinfo[i].usri0_name, NULL), FALSE);
Bram Moolenaar828c3d72018-06-19 18:58:07 +02002105
2106 NetApiBufferFree(uinfo);
2107 }
2108 }
Bram Moolenaar24305862012-08-15 14:05:05 +02002109# endif
Bram Moolenaar6b0b83f2018-09-10 19:03:05 +02002110# if defined(HAVE_GETPWNAM)
2111 {
2112 char_u *user_env = mch_getenv((char_u *)"USER");
2113
2114 // The $USER environment variable may be a valid remote user name (NIS,
2115 // LDAP) not already listed by getpwent(), as getpwent() only lists
2116 // local user names. If $USER is not already listed, check whether it
2117 // is a valid remote user name using getpwnam() and if it is, add it to
2118 // the list of user names.
2119
2120 if (user_env != NULL && *user_env != NUL)
2121 {
2122 int i;
2123
2124 for (i = 0; i < ga_users.ga_len; i++)
2125 {
2126 char_u *local_user = ((char_u **)ga_users.ga_data)[i];
2127
2128 if (STRCMP(local_user, user_env) == 0)
2129 break;
2130 }
2131
2132 if (i == ga_users.ga_len)
2133 {
2134 struct passwd *pw = getpwnam((char *)user_env);
2135
2136 if (pw != NULL)
2137 add_user((char_u *)pw->pw_name, TRUE);
2138 }
2139 }
2140 }
2141# endif
Bram Moolenaar24305862012-08-15 14:05:05 +02002142}
2143
2144/*
2145 * Function given to ExpandGeneric() to obtain an user names.
2146 */
2147 char_u*
Bram Moolenaar9b578142016-01-30 19:39:49 +01002148get_users(expand_T *xp UNUSED, int idx)
Bram Moolenaar24305862012-08-15 14:05:05 +02002149{
2150 init_users();
2151 if (idx < ga_users.ga_len)
2152 return ((char_u **)ga_users.ga_data)[idx];
2153 return NULL;
2154}
2155
2156/*
2157 * Check whether name matches a user name. Return:
2158 * 0 if name does not match any user name.
2159 * 1 if name partially matches the beginning of a user name.
2160 * 2 is name fully matches a user name.
2161 */
Bram Moolenaar6c5d1042018-07-07 16:41:13 +02002162 int
2163match_user(char_u *name)
Bram Moolenaar24305862012-08-15 14:05:05 +02002164{
2165 int i;
2166 int n = (int)STRLEN(name);
2167 int result = 0;
2168
2169 init_users();
2170 for (i = 0; i < ga_users.ga_len; i++)
2171 {
2172 if (STRCMP(((char_u **)ga_users.ga_data)[i], name) == 0)
Bram Moolenaar85a20022019-12-21 18:25:54 +01002173 return 2; // full match
Bram Moolenaar24305862012-08-15 14:05:05 +02002174 if (STRNCMP(((char_u **)ga_users.ga_data)[i], name, n) == 0)
Bram Moolenaar85a20022019-12-21 18:25:54 +01002175 result = 1; // partial match
Bram Moolenaar24305862012-08-15 14:05:05 +02002176 }
2177 return result;
2178}
Bram Moolenaar071d4272004-06-13 20:20:40 +00002179
Bram Moolenaar5843f5f2019-08-20 20:13:45 +02002180 static void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002181prepare_to_exit(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002182{
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002183#if defined(SIGHUP) && defined(SIG_IGN)
Bram Moolenaar85a20022019-12-21 18:25:54 +01002184 // Ignore SIGHUP, because a dropped connection causes a read error, which
2185 // makes Vim exit and then handling SIGHUP causes various reentrance
2186 // problems.
Bram Moolenaar293ee4d2004-12-09 21:34:53 +00002187 signal(SIGHUP, SIG_IGN);
2188#endif
2189
Bram Moolenaar071d4272004-06-13 20:20:40 +00002190#ifdef FEAT_GUI
2191 if (gui.in_use)
2192 {
2193 gui.dying = TRUE;
Bram Moolenaar85a20022019-12-21 18:25:54 +01002194 out_trash(); // trash any pending output
Bram Moolenaar071d4272004-06-13 20:20:40 +00002195 }
2196 else
2197#endif
2198 {
2199 windgoto((int)Rows - 1, 0);
2200
2201 /*
2202 * Switch terminal mode back now, so messages end up on the "normal"
2203 * screen (if there are two screens).
2204 */
2205 settmode(TMODE_COOK);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02002206 stoptermcap();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002207 out_flush();
2208 }
2209}
2210
2211/*
2212 * Preserve files and exit.
2213 * When called IObuff must contain a message.
Bram Moolenaarbec9c202013-09-05 21:41:39 +02002214 * NOTE: This may be called from deathtrap() in a signal handler, avoid unsafe
2215 * functions, such as allocating memory.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002216 */
2217 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002218preserve_exit(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002219{
2220 buf_T *buf;
2221
2222 prepare_to_exit();
2223
Bram Moolenaar85a20022019-12-21 18:25:54 +01002224 // Setting this will prevent free() calls. That avoids calling free()
2225 // recursively when free() was invoked with a bad pointer.
Bram Moolenaar4770d092006-01-12 23:22:24 +00002226 really_exiting = TRUE;
2227
Bram Moolenaar071d4272004-06-13 20:20:40 +00002228 out_str(IObuff);
Bram Moolenaar85a20022019-12-21 18:25:54 +01002229 screen_start(); // don't know where cursor is now
Bram Moolenaar071d4272004-06-13 20:20:40 +00002230 out_flush();
2231
Bram Moolenaar85a20022019-12-21 18:25:54 +01002232 ml_close_notmod(); // close all not-modified buffers
Bram Moolenaar071d4272004-06-13 20:20:40 +00002233
Bram Moolenaar29323592016-07-24 22:04:11 +02002234 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002235 {
2236 if (buf->b_ml.ml_mfp != NULL && buf->b_ml.ml_mfp->mf_fname != NULL)
2237 {
Bram Moolenaar69212b12020-05-10 14:14:03 +02002238 OUT_STR("Vim: preserving files...\r\n");
Bram Moolenaar85a20022019-12-21 18:25:54 +01002239 screen_start(); // don't know where cursor is now
Bram Moolenaar071d4272004-06-13 20:20:40 +00002240 out_flush();
Bram Moolenaar85a20022019-12-21 18:25:54 +01002241 ml_sync_all(FALSE, FALSE); // preserve all swap files
Bram Moolenaar071d4272004-06-13 20:20:40 +00002242 break;
2243 }
2244 }
2245
Bram Moolenaar85a20022019-12-21 18:25:54 +01002246 ml_close_all(FALSE); // close all memfiles, without deleting
Bram Moolenaar071d4272004-06-13 20:20:40 +00002247
Bram Moolenaar69212b12020-05-10 14:14:03 +02002248 OUT_STR("Vim: Finished.\r\n");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002249
2250 getout(1);
2251}
2252
2253/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002254 * Check for CTRL-C pressed, but only once in a while.
2255 * Should be used instead of ui_breakcheck() for functions that check for
2256 * each line in the file. Calling ui_breakcheck() each time takes too much
2257 * time, because it can be a system call.
2258 */
2259
2260#ifndef BREAKCHECK_SKIP
Bram Moolenaarb4fe0eb2019-07-21 14:50:21 +02002261# define BREAKCHECK_SKIP 1000
Bram Moolenaar071d4272004-06-13 20:20:40 +00002262#endif
2263
2264static int breakcheck_count = 0;
2265
2266 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002267line_breakcheck(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002268{
2269 if (++breakcheck_count >= BREAKCHECK_SKIP)
2270 {
2271 breakcheck_count = 0;
2272 ui_breakcheck();
2273 }
2274}
2275
2276/*
2277 * Like line_breakcheck() but check 10 times less often.
2278 */
2279 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002280fast_breakcheck(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002281{
2282 if (++breakcheck_count >= BREAKCHECK_SKIP * 10)
2283 {
2284 breakcheck_count = 0;
2285 ui_breakcheck();
2286 }
2287}
2288
Dominique Pelle748b3082022-01-08 12:41:16 +00002289# if defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaarf1ec3782020-03-20 19:37:47 +01002290/*
2291 * Like line_breakcheck() but check 100 times less often.
2292 */
2293 void
2294veryfast_breakcheck(void)
2295{
2296 if (++breakcheck_count >= BREAKCHECK_SKIP * 100)
2297 {
2298 breakcheck_count = 0;
2299 ui_breakcheck();
2300 }
2301}
Dominique Pelle748b3082022-01-08 12:41:16 +00002302#endif
Bram Moolenaarf1ec3782020-03-20 19:37:47 +01002303
Bram Moolenaar0a52df52019-08-18 22:26:31 +02002304#if defined(VIM_BACKTICK) || defined(FEAT_EVAL) \
2305 || (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
2306 || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002307
2308#ifndef SEEK_SET
2309# define SEEK_SET 0
2310#endif
2311#ifndef SEEK_END
2312# define SEEK_END 2
2313#endif
2314
2315/*
2316 * Get the stdout of an external command.
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02002317 * If "ret_len" is NULL replace NUL characters with NL. When "ret_len" is not
2318 * NULL store the length there.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002319 * Returns an allocated string, or NULL for error.
2320 */
2321 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01002322get_cmd_output(
2323 char_u *cmd,
Bram Moolenaar85a20022019-12-21 18:25:54 +01002324 char_u *infile, // optional input file name
2325 int flags, // can be SHELL_SILENT
Bram Moolenaar9b578142016-01-30 19:39:49 +01002326 int *ret_len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002327{
2328 char_u *tempname;
2329 char_u *command;
2330 char_u *buffer = NULL;
2331 int len;
2332 int i = 0;
2333 FILE *fd;
2334
2335 if (check_restricted() || check_secure())
2336 return NULL;
2337
Bram Moolenaar85a20022019-12-21 18:25:54 +01002338 // get a name for the temp file
Bram Moolenaare5c421c2015-03-31 13:33:08 +02002339 if ((tempname = vim_tempname('o', FALSE)) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002340 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00002341 emsg(_(e_cant_get_temp_file_name));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002342 return NULL;
2343 }
2344
Bram Moolenaar85a20022019-12-21 18:25:54 +01002345 // Add the redirection stuff
Bram Moolenaarc0197e22004-09-13 20:26:32 +00002346 command = make_filter_cmd(cmd, infile, tempname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002347 if (command == NULL)
2348 goto done;
2349
2350 /*
2351 * Call the shell to execute the command (errors are ignored).
2352 * Don't check timestamps here.
2353 */
2354 ++no_check_timestamps;
2355 call_shell(command, SHELL_DOOUT | SHELL_EXPAND | flags);
2356 --no_check_timestamps;
2357
2358 vim_free(command);
2359
2360 /*
2361 * read the names from the file into memory
2362 */
2363# ifdef VMS
Bram Moolenaar85a20022019-12-21 18:25:54 +01002364 // created temporary file is not always readable as binary
Bram Moolenaar071d4272004-06-13 20:20:40 +00002365 fd = mch_fopen((char *)tempname, "r");
2366# else
2367 fd = mch_fopen((char *)tempname, READBIN);
2368# endif
2369
Bram Moolenaar3df8f6e2022-04-17 14:01:51 +01002370 // Not being able to seek means we can't read the file.
2371 if (fd == NULL
2372 || fseek(fd, 0L, SEEK_END) == -1
2373 || (len = ftell(fd)) == -1 // get size of temp file
2374 || fseek(fd, 0L, SEEK_SET) == -1) // back to the start
Bram Moolenaar071d4272004-06-13 20:20:40 +00002375 {
Bram Moolenaara9549c92022-04-17 14:18:11 +01002376 semsg(_(e_cannot_read_from_str_2), tempname);
Bram Moolenaar3df8f6e2022-04-17 14:01:51 +01002377 if (fd != NULL)
2378 fclose(fd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002379 goto done;
2380 }
2381
Bram Moolenaar071d4272004-06-13 20:20:40 +00002382 buffer = alloc(len + 1);
2383 if (buffer != NULL)
2384 i = (int)fread((char *)buffer, (size_t)1, (size_t)len, fd);
2385 fclose(fd);
2386 mch_remove(tempname);
2387 if (buffer == NULL)
2388 goto done;
2389#ifdef VMS
Bram Moolenaar85a20022019-12-21 18:25:54 +01002390 len = i; // VMS doesn't give us what we asked for...
Bram Moolenaar071d4272004-06-13 20:20:40 +00002391#endif
2392 if (i != len)
2393 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00002394 semsg(_(e_cant_read_file_str), tempname);
Bram Moolenaard23a8232018-02-10 18:45:26 +01002395 VIM_CLEAR(buffer);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002396 }
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02002397 else if (ret_len == NULL)
Bram Moolenaarfb332a22013-08-03 14:10:50 +02002398 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01002399 // Change NUL into SOH, otherwise the string is truncated.
Bram Moolenaarfb332a22013-08-03 14:10:50 +02002400 for (i = 0; i < len; ++i)
Bram Moolenaarf40f4ab2013-08-03 17:31:28 +02002401 if (buffer[i] == NUL)
2402 buffer[i] = 1;
Bram Moolenaarfb332a22013-08-03 14:10:50 +02002403
Bram Moolenaar85a20022019-12-21 18:25:54 +01002404 buffer[len] = NUL; // make sure the buffer is terminated
Bram Moolenaarfb332a22013-08-03 14:10:50 +02002405 }
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02002406 else
2407 *ret_len = len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002408
2409done:
2410 vim_free(tempname);
2411 return buffer;
2412}
Bram Moolenaar26262f82019-09-04 20:59:15 +02002413
2414# if defined(FEAT_EVAL) || defined(PROTO)
2415
Bram Moolenaar840d16f2019-09-10 21:27:18 +02002416 static void
Bram Moolenaar26262f82019-09-04 20:59:15 +02002417get_cmd_output_as_rettv(
2418 typval_T *argvars,
2419 typval_T *rettv,
2420 int retlist)
2421{
2422 char_u *res = NULL;
2423 char_u *p;
2424 char_u *infile = NULL;
2425 int err = FALSE;
2426 FILE *fd;
2427 list_T *list = NULL;
2428 int flags = SHELL_SILENT;
2429
2430 rettv->v_type = VAR_STRING;
2431 rettv->vval.v_string = NULL;
2432 if (check_restricted() || check_secure())
2433 goto errret;
2434
Yegappan Lakshmanan0ad871d2021-07-23 20:37:56 +02002435 if (in_vim9script()
2436 && (check_for_string_arg(argvars, 0) == FAIL
Bram Moolenaar944cc9c2022-06-27 22:17:37 +01002437 || check_for_opt_string_or_number_or_list_arg(argvars, 1)
2438 == FAIL))
Yegappan Lakshmanan0ad871d2021-07-23 20:37:56 +02002439 return;
2440
Bram Moolenaar26262f82019-09-04 20:59:15 +02002441 if (argvars[1].v_type != VAR_UNKNOWN)
2442 {
2443 /*
2444 * Write the text to a temp file, to be used for input of the shell
2445 * command.
2446 */
2447 if ((infile = vim_tempname('i', TRUE)) == NULL)
2448 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00002449 emsg(_(e_cant_get_temp_file_name));
Bram Moolenaar26262f82019-09-04 20:59:15 +02002450 goto errret;
2451 }
2452
2453 fd = mch_fopen((char *)infile, WRITEBIN);
2454 if (fd == NULL)
2455 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00002456 semsg(_(e_cant_open_file_str), infile);
Bram Moolenaar26262f82019-09-04 20:59:15 +02002457 goto errret;
2458 }
2459 if (argvars[1].v_type == VAR_NUMBER)
2460 {
2461 linenr_T lnum;
2462 buf_T *buf;
2463
2464 buf = buflist_findnr(argvars[1].vval.v_number);
2465 if (buf == NULL)
2466 {
Bram Moolenaar40bcec12021-12-05 22:19:27 +00002467 semsg(_(e_buffer_nr_does_not_exist), argvars[1].vval.v_number);
Bram Moolenaar26262f82019-09-04 20:59:15 +02002468 fclose(fd);
2469 goto errret;
2470 }
2471
2472 for (lnum = 1; lnum <= buf->b_ml.ml_line_count; lnum++)
2473 {
2474 for (p = ml_get_buf(buf, lnum, FALSE); *p != NUL; ++p)
2475 if (putc(*p == '\n' ? NUL : *p, fd) == EOF)
2476 {
2477 err = TRUE;
2478 break;
2479 }
2480 if (putc(NL, fd) == EOF)
2481 {
2482 err = TRUE;
2483 break;
2484 }
2485 }
2486 }
2487 else if (argvars[1].v_type == VAR_LIST)
2488 {
2489 if (write_list(fd, argvars[1].vval.v_list, TRUE) == FAIL)
2490 err = TRUE;
2491 }
2492 else
2493 {
2494 size_t len;
2495 char_u buf[NUMBUFLEN];
2496
2497 p = tv_get_string_buf_chk(&argvars[1], buf);
2498 if (p == NULL)
2499 {
2500 fclose(fd);
Bram Moolenaar85a20022019-12-21 18:25:54 +01002501 goto errret; // type error; errmsg already given
Bram Moolenaar26262f82019-09-04 20:59:15 +02002502 }
2503 len = STRLEN(p);
2504 if (len > 0 && fwrite(p, len, 1, fd) != 1)
2505 err = TRUE;
2506 }
2507 if (fclose(fd) != 0)
2508 err = TRUE;
2509 if (err)
2510 {
Bram Moolenaara6f79292022-01-04 21:30:47 +00002511 emsg(_(e_error_writing_temp_file));
Bram Moolenaar26262f82019-09-04 20:59:15 +02002512 goto errret;
2513 }
2514 }
2515
Bram Moolenaar85a20022019-12-21 18:25:54 +01002516 // Omit SHELL_COOKED when invoked with ":silent". Avoids that the shell
2517 // echoes typeahead, that messes up the display.
Bram Moolenaar26262f82019-09-04 20:59:15 +02002518 if (!msg_silent)
2519 flags += SHELL_COOKED;
2520
2521 if (retlist)
2522 {
2523 int len;
2524 listitem_T *li;
2525 char_u *s = NULL;
2526 char_u *start;
2527 char_u *end;
2528 int i;
2529
2530 res = get_cmd_output(tv_get_string(&argvars[0]), infile, flags, &len);
2531 if (res == NULL)
2532 goto errret;
2533
2534 list = list_alloc();
2535 if (list == NULL)
2536 goto errret;
2537
2538 for (i = 0; i < len; ++i)
2539 {
2540 start = res + i;
2541 while (i < len && res[i] != NL)
2542 ++i;
2543 end = res + i;
2544
2545 s = alloc(end - start + 1);
2546 if (s == NULL)
2547 goto errret;
2548
2549 for (p = s; start < end; ++p, ++start)
2550 *p = *start == NUL ? NL : *start;
2551 *p = NUL;
2552
2553 li = listitem_alloc();
2554 if (li == NULL)
2555 {
2556 vim_free(s);
2557 goto errret;
2558 }
2559 li->li_tv.v_type = VAR_STRING;
2560 li->li_tv.v_lock = 0;
2561 li->li_tv.vval.v_string = s;
2562 list_append(list, li);
2563 }
2564
2565 rettv_list_set(rettv, list);
2566 list = NULL;
2567 }
2568 else
2569 {
2570 res = get_cmd_output(tv_get_string(&argvars[0]), infile, flags, NULL);
2571#ifdef USE_CRNL
Bram Moolenaar85a20022019-12-21 18:25:54 +01002572 // translate <CR><NL> into <NL>
Bram Moolenaar26262f82019-09-04 20:59:15 +02002573 if (res != NULL)
2574 {
2575 char_u *s, *d;
2576
2577 d = res;
2578 for (s = res; *s; ++s)
2579 {
2580 if (s[0] == CAR && s[1] == NL)
2581 ++s;
2582 *d++ = *s;
2583 }
2584 *d = NUL;
2585 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002586#endif
Bram Moolenaar26262f82019-09-04 20:59:15 +02002587 rettv->vval.v_string = res;
2588 res = NULL;
2589 }
2590
2591errret:
2592 if (infile != NULL)
2593 {
2594 mch_remove(infile);
2595 vim_free(infile);
2596 }
2597 if (res != NULL)
2598 vim_free(res);
2599 if (list != NULL)
2600 list_free(list);
2601}
Bram Moolenaar071d4272004-06-13 20:20:40 +00002602
2603/*
Bram Moolenaar26262f82019-09-04 20:59:15 +02002604 * "system()" function
Bram Moolenaar071d4272004-06-13 20:20:40 +00002605 */
2606 void
Bram Moolenaar26262f82019-09-04 20:59:15 +02002607f_system(typval_T *argvars, typval_T *rettv)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002608{
Bram Moolenaar26262f82019-09-04 20:59:15 +02002609 get_cmd_output_as_rettv(argvars, rettv, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002610}
2611
2612/*
Bram Moolenaar26262f82019-09-04 20:59:15 +02002613 * "systemlist()" function
2614 */
2615 void
2616f_systemlist(typval_T *argvars, typval_T *rettv)
2617{
2618 get_cmd_output_as_rettv(argvars, rettv, TRUE);
2619}
2620# endif // FEAT_EVAL
2621
2622#endif
2623
2624/*
Bram Moolenaara9dc3752010-07-11 20:46:53 +02002625 * Return TRUE when need to go to Insert mode because of 'insertmode'.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002626 * Don't do this when still processing a command or a mapping.
2627 * Don't do this when inside a ":normal" command.
2628 */
2629 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002630goto_im(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002631{
2632 return (p_im && stuff_empty() && typebuf_typed());
2633}
Bram Moolenaar75a8d742014-05-07 15:10:21 +02002634
2635/*
Bram Moolenaar050fe7e2014-05-22 14:00:16 +02002636 * Returns the isolated name of the shell in allocated memory:
Bram Moolenaar75a8d742014-05-07 15:10:21 +02002637 * - Skip beyond any path. E.g., "/usr/bin/csh -f" -> "csh -f".
2638 * - Remove any argument. E.g., "csh -f" -> "csh".
2639 * But don't allow a space in the path, so that this works:
2640 * "/usr/bin/csh --rcfile ~/.cshrc"
2641 * But don't do that for Windows, it's common to have a space in the path.
Bram Moolenaar28ee8922020-10-28 20:20:00 +01002642 * Returns NULL when out of memory.
Bram Moolenaar75a8d742014-05-07 15:10:21 +02002643 */
2644 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01002645get_isolated_shell_name(void)
Bram Moolenaar75a8d742014-05-07 15:10:21 +02002646{
2647 char_u *p;
2648
Bram Moolenaar4f974752019-02-17 17:44:42 +01002649#ifdef MSWIN
Bram Moolenaar75a8d742014-05-07 15:10:21 +02002650 p = gettail(p_sh);
Bram Moolenaar71ccd032020-06-12 22:59:11 +02002651 p = vim_strnsave(p, skiptowhite(p) - p);
Bram Moolenaar75a8d742014-05-07 15:10:21 +02002652#else
2653 p = skiptowhite(p_sh);
2654 if (*p == NUL)
2655 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01002656 // No white space, use the tail.
Bram Moolenaar75a8d742014-05-07 15:10:21 +02002657 p = vim_strsave(gettail(p_sh));
2658 }
2659 else
2660 {
2661 char_u *p1, *p2;
2662
Bram Moolenaar85a20022019-12-21 18:25:54 +01002663 // Find the last path separator before the space.
Bram Moolenaar75a8d742014-05-07 15:10:21 +02002664 p1 = p_sh;
Bram Moolenaar91acfff2017-03-12 19:22:36 +01002665 for (p2 = p_sh; p2 < p; MB_PTR_ADV(p2))
Bram Moolenaar75a8d742014-05-07 15:10:21 +02002666 if (vim_ispathsep(*p2))
2667 p1 = p2 + 1;
Bram Moolenaar71ccd032020-06-12 22:59:11 +02002668 p = vim_strnsave(p1, p - p1);
Bram Moolenaar75a8d742014-05-07 15:10:21 +02002669 }
2670#endif
2671 return p;
2672}
Bram Moolenaar5fd0f502019-02-13 23:13:28 +01002673
2674/*
2675 * Check if the "://" of a URL is at the pointer, return URL_SLASH.
2676 * Also check for ":\\", which MS Internet Explorer accepts, return
2677 * URL_BACKSLASH.
2678 */
2679 int
2680path_is_url(char_u *p)
2681{
2682 if (STRNCMP(p, "://", (size_t)3) == 0)
2683 return URL_SLASH;
2684 else if (STRNCMP(p, ":\\\\", (size_t)3) == 0)
2685 return URL_BACKSLASH;
2686 return 0;
2687}
2688
2689/*
Tsuyoshi CHO7b7a1182021-07-11 21:51:17 +02002690 * Check if "fname" starts with "name://" or "name:\\".
2691 * Return URL_SLASH for "name://", URL_BACKSLASH for "name:\\".
Bram Moolenaar5fd0f502019-02-13 23:13:28 +01002692 * Return zero otherwise.
2693 */
2694 int
2695path_with_url(char_u *fname)
2696{
2697 char_u *p;
2698
Tsuyoshi CHO7b7a1182021-07-11 21:51:17 +02002699 // We accept alphabetic characters and a dash in scheme part.
2700 // RFC 3986 allows for more, but it increases the risk of matching
2701 // non-URL text.
2702
2703 // first character must be alpha
2704 if (!isalpha(*fname))
2705 return 0;
2706
2707 // check body: alpha or dash
itchyny94e7d342021-10-21 18:01:13 +01002708 for (p = fname + 1; (isalpha(*p) || (*p == '-')); ++p)
Bram Moolenaar5fd0f502019-02-13 23:13:28 +01002709 ;
Tsuyoshi CHO7b7a1182021-07-11 21:51:17 +02002710
2711 // check last char is not a dash
2712 if (p[-1] == '-')
2713 return 0;
2714
2715 // "://" or ":\\" must follow
Bram Moolenaar5fd0f502019-02-13 23:13:28 +01002716 return path_is_url(p);
2717}
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +02002718
Bram Moolenaar3075a452021-11-17 15:51:52 +00002719#if defined(FEAT_EVAL) || defined(PROTO)
2720/*
2721 * Return the dictionary of v:event.
2722 * Save and clear the value in case it already has items.
2723 */
2724 dict_T *
2725get_v_event(save_v_event_T *sve)
2726{
2727 dict_T *v_event = get_vim_var_dict(VV_EVENT);
2728
2729 if (v_event->dv_hashtab.ht_used > 0)
2730 {
2731 // recursive use of v:event, save, make empty and restore later
2732 sve->sve_did_save = TRUE;
2733 sve->sve_hashtab = v_event->dv_hashtab;
2734 hash_init(&v_event->dv_hashtab);
2735 }
2736 else
2737 sve->sve_did_save = FALSE;
2738 return v_event;
2739}
2740
2741 void
2742restore_v_event(dict_T *v_event, save_v_event_T *sve)
2743{
2744 dict_free_contents(v_event);
2745 if (sve->sve_did_save)
2746 v_event->dv_hashtab = sve->sve_hashtab;
2747 else
2748 hash_init(&v_event->dv_hashtab);
2749}
2750#endif
2751
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +02002752/*
LemonBoy2bf52dd2022-04-09 18:17:34 +01002753 * Fires a ModeChanged autocmd event if appropriate.
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +02002754 */
2755 void
LemonBoy2bf52dd2022-04-09 18:17:34 +01002756may_trigger_modechanged()
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +02002757{
Bram Moolenaar3075a452021-11-17 15:51:52 +00002758#ifdef FEAT_EVAL
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +02002759 dict_T *v_event;
Bram Moolenaar3075a452021-11-17 15:51:52 +00002760 save_v_event_T save_v_event;
LemonBoy2bf52dd2022-04-09 18:17:34 +01002761 char_u curr_mode[MODE_MAX_LENGTH];
2762 char_u pattern_buf[2 * MODE_MAX_LENGTH];
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +02002763
2764 if (!has_modechanged())
2765 return;
2766
LemonBoy2bf52dd2022-04-09 18:17:34 +01002767 get_mode(curr_mode);
2768 if (STRCMP(curr_mode, last_mode) == 0)
=?UTF-8?q?Magnus=20Gro=C3=9F?=25def2c2021-10-22 18:56:39 +01002769 return;
=?UTF-8?q?Magnus=20Gro=C3=9F?=25def2c2021-10-22 18:56:39 +01002770
Bram Moolenaar3075a452021-11-17 15:51:52 +00002771 v_event = get_v_event(&save_v_event);
LemonBoy2bf52dd2022-04-09 18:17:34 +01002772 (void)dict_add_string(v_event, "new_mode", curr_mode);
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +02002773 (void)dict_add_string(v_event, "old_mode", last_mode);
2774 dict_set_items_ro(v_event);
2775
2776 // concatenate modes in format "old_mode:new_mode"
LemonBoy2bf52dd2022-04-09 18:17:34 +01002777 vim_snprintf((char *)pattern_buf, sizeof(pattern_buf), "%s:%s", last_mode,
2778 curr_mode);
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +02002779
LemonBoy2bf52dd2022-04-09 18:17:34 +01002780 apply_autocmds(EVENT_MODECHANGED, pattern_buf, NULL, FALSE, curbuf);
2781 STRCPY(last_mode, curr_mode);
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +02002782
Bram Moolenaar3075a452021-11-17 15:51:52 +00002783 restore_v_event(v_event, &save_v_event);
=?UTF-8?q?Magnus=20Gro=C3=9F?=f1e88762021-09-12 13:39:55 +02002784#endif
2785}