Bram Moolenaar | edf3f97 | 2016-08-29 22:49:24 +0200 | [diff] [blame] | 1 | /* vi:set ts=8 sts=4 sw=4 noet: |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2 | * |
| 3 | * VIM - Vi IMproved by Bram Moolenaar |
| 4 | * |
| 5 | * Do ":help uganda" in Vim to read copying and usage conditions. |
| 6 | * Do ":help credits" in Vim to see a list of people who contributed. |
| 7 | * See README.txt for an overview of the Vim source code. |
| 8 | */ |
| 9 | |
| 10 | /* |
| 11 | * misc1.c: functions that didn't seem to fit elsewhere |
| 12 | */ |
| 13 | |
| 14 | #include "vim.h" |
| 15 | #include "version.h" |
| 16 | |
Bram Moolenaar | b3f7406 | 2020-02-26 16:16:53 +0100 | [diff] [blame] | 17 | #if defined(__HAIKU__) |
| 18 | # include <storage/FindDirectory.h> |
| 19 | #endif |
| 20 | |
Bram Moolenaar | 0a52df5 | 2019-08-18 22:26:31 +0200 | [diff] [blame] | 21 | #if defined(MSWIN) |
Bram Moolenaar | 828c3d7 | 2018-06-19 18:58:07 +0200 | [diff] [blame] | 22 | # include <lm.h> |
| 23 | #endif |
| 24 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 25 | #define URL_SLASH 1 // path_is_url() has found "://" |
| 26 | #define URL_BACKSLASH 2 // path_is_url() has found ":\\" |
Bram Moolenaar | 5fd0f50 | 2019-02-13 23:13:28 +0100 | [diff] [blame] | 27 | |
Bram Moolenaar | 0a52df5 | 2019-08-18 22:26:31 +0200 | [diff] [blame] | 28 | // All user names (for ~user completion as done by shell). |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 29 | static garray_T ga_users; |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 30 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 31 | /* |
Bram Moolenaar | 2c019c8 | 2013-10-06 17:46:56 +0200 | [diff] [blame] | 32 | * 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 Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 35 | * 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 Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 38 | * If "include_space" is set, include trailing whitespace while calculating the |
| 39 | * length. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 40 | */ |
| 41 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 42 | get_leader_len( |
| 43 | char_u *line, |
| 44 | char_u **flags, |
| 45 | int backward, |
| 46 | int include_space) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 47 | { |
| 48 | int i, j; |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 49 | int result; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 50 | int got_com = FALSE; |
| 51 | int found_one; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 52 | char_u part_buf[COM_MAX_LEN]; // buffer for one option part |
| 53 | char_u *string; // pointer to comment string |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 54 | char_u *list; |
Bram Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 55 | int middle_match_len = 0; |
| 56 | char_u *prev_list; |
Bram Moolenaar | 05da428 | 2011-05-10 14:44:11 +0200 | [diff] [blame] | 57 | char_u *saved_flags = NULL; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 58 | |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 59 | result = i = 0; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 60 | while (VIM_ISWHITE(line[i])) // leading white space is ignored |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 61 | ++i; |
| 62 | |
| 63 | /* |
| 64 | * Repeat to match several nested comment strings. |
| 65 | */ |
Bram Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 66 | while (line[i] != NUL) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 67 | { |
| 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 Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 74 | // Get one option part into part_buf[]. Advance "list" to next |
| 75 | // one. Put "string" at start of string. |
Bram Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 76 | if (!got_com && flags != NULL) |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 77 | *flags = list; // remember where flags started |
Bram Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 78 | prev_list = list; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 79 | (void)copy_option_part(&list, part_buf, COM_MAX_LEN, ","); |
| 80 | string = vim_strchr(part_buf, ':'); |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 81 | if (string == NULL) // missing ':', ignore this part |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 82 | continue; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 83 | *string++ = NUL; // isolate flags from string |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 84 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 85 | // If we found a middle match previously, use that match when this |
| 86 | // is not a middle or end. |
Bram Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 87 | 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 Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 92 | // When we already found a nested comment, only accept further |
| 93 | // nested comments. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 94 | if (got_com && vim_strchr(part_buf, COM_NEST) == NULL) |
| 95 | continue; |
| 96 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 97 | // When 'O' flag present and using "O" command skip this one. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 98 | if (backward && vim_strchr(part_buf, COM_NOBACK) != NULL) |
| 99 | continue; |
| 100 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 101 | // 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 Moolenaar | 1c46544 | 2017-03-12 20:10:05 +0100 | [diff] [blame] | 105 | if (VIM_ISWHITE(string[0])) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 106 | { |
Bram Moolenaar | 1c46544 | 2017-03-12 20:10:05 +0100 | [diff] [blame] | 107 | if (i == 0 || !VIM_ISWHITE(line[i - 1])) |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 108 | continue; // missing white space |
Bram Moolenaar | 1c46544 | 2017-03-12 20:10:05 +0100 | [diff] [blame] | 109 | while (VIM_ISWHITE(string[0])) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 110 | ++string; |
| 111 | } |
| 112 | for (j = 0; string[j] != NUL && string[j] == line[i + j]; ++j) |
| 113 | ; |
| 114 | if (string[j] != NUL) |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 115 | continue; // string doesn't match |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 116 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 117 | // When 'b' flag used, there must be white space or an |
| 118 | // end-of-line after the string in the line. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 119 | if (vim_strchr(part_buf, COM_BLANK) != NULL |
Bram Moolenaar | 1c46544 | 2017-03-12 20:10:05 +0100 | [diff] [blame] | 120 | && !VIM_ISWHITE(line[i + j]) && line[i + j] != NUL) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 121 | continue; |
| 122 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 123 | // 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 Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 128 | 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 Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 138 | // Use this match instead of the middle match, since it's a |
| 139 | // longer thus better match. |
Bram Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 140 | middle_match_len = 0; |
| 141 | |
| 142 | if (middle_match_len == 0) |
| 143 | i += j; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 144 | found_one = TRUE; |
| 145 | break; |
| 146 | } |
| 147 | |
Bram Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 148 | if (middle_match_len != 0) |
| 149 | { |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 150 | // Use the previously found middle match after failing to find a |
| 151 | // match with an end. |
Bram Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 152 | if (!got_com && flags != NULL) |
| 153 | *flags = saved_flags; |
| 154 | i += middle_match_len; |
| 155 | found_one = TRUE; |
| 156 | } |
| 157 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 158 | // No match found, stop scanning. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 159 | if (!found_one) |
| 160 | break; |
| 161 | |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 162 | result = i; |
| 163 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 164 | // Include any trailing white space. |
Bram Moolenaar | 1c46544 | 2017-03-12 20:10:05 +0100 | [diff] [blame] | 165 | while (VIM_ISWHITE(line[i])) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 166 | ++i; |
| 167 | |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 168 | if (include_space) |
| 169 | result = i; |
| 170 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 171 | // If this comment doesn't nest, stop here. |
Bram Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 172 | got_com = TRUE; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 173 | if (vim_strchr(part_buf, COM_NEST) == NULL) |
| 174 | break; |
| 175 | } |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 176 | return result; |
| 177 | } |
Bram Moolenaar | a4271d5 | 2011-05-10 13:38:27 +0200 | [diff] [blame] | 178 | |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 179 | /* |
| 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 Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 187 | get_last_leader_offset(char_u *line, char_u **flags) |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 188 | { |
| 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 Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 197 | char_u part_buf[COM_MAX_LEN]; // buffer for one option part |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 198 | |
| 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 Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 219 | if (string == NULL) // If everything is fine, this cannot actually |
| 220 | // happen. |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 221 | continue; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 222 | *string++ = NUL; // Isolate flags from string. |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 223 | 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 Moolenaar | 1c46544 | 2017-03-12 20:10:05 +0100 | [diff] [blame] | 231 | if (VIM_ISWHITE(string[0])) |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 232 | { |
Bram Moolenaar | 1c46544 | 2017-03-12 20:10:05 +0100 | [diff] [blame] | 233 | if (i == 0 || !VIM_ISWHITE(line[i - 1])) |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 234 | continue; |
Bram Moolenaar | 5393281 | 2018-12-07 21:08:49 +0100 | [diff] [blame] | 235 | while (VIM_ISWHITE(*string)) |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 236 | ++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 Moolenaar | 1c46544 | 2017-03-12 20:10:05 +0100 | [diff] [blame] | 248 | && !VIM_ISWHITE(line[i + j]) && line[i + j] != NUL) |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 249 | continue; |
Bram Moolenaar | 5393281 | 2018-12-07 21:08:49 +0100 | [diff] [blame] | 250 | |
Bram Moolenaar | 4af7259 | 2018-12-09 15:00:52 +0100 | [diff] [blame] | 251 | if (vim_strchr(part_buf, COM_MIDDLE) != NULL) |
Bram Moolenaar | 5393281 | 2018-12-07 21:08:49 +0100 | [diff] [blame] | 252 | { |
Bram Moolenaar | 4af7259 | 2018-12-09 15:00:52 +0100 | [diff] [blame] | 253 | // 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 Moolenaar | 5393281 | 2018-12-07 21:08:49 +0100 | [diff] [blame] | 258 | for (j = 0; VIM_ISWHITE(line[j]) && j <= i; j++) |
| 259 | ; |
| 260 | if (j < i) |
| 261 | continue; |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 262 | } |
| 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 Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 278 | char_u part_buf2[COM_MAX_LEN]; // buffer for one option part |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 279 | 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 Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 290 | // 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 Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 294 | |
Bram Moolenaar | 1c46544 | 2017-03-12 20:10:05 +0100 | [diff] [blame] | 295 | while (VIM_ISWHITE(*com_leader)) |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 296 | ++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 Moolenaar | 1c46544 | 2017-03-12 20:10:05 +0100 | [diff] [blame] | 308 | while (VIM_ISWHITE(*string)) |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 309 | ++string; |
| 310 | len2 = (int)STRLEN(string); |
| 311 | if (len2 == 0) |
| 312 | continue; |
| 313 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 314 | // Now we have to verify whether string ends with a substring |
| 315 | // beginning the com_leader. |
Bram Moolenaar | 8134039 | 2012-06-06 16:12:59 +0200 | [diff] [blame] | 316 | 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 Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 329 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 330 | |
| 331 | /* |
| 332 | * Return the number of window lines occupied by buffer line "lnum". |
Bram Moolenaar | 61c4b04 | 2022-10-18 15:10:11 +0100 | [diff] [blame] | 333 | * Includes any filler lines. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 334 | */ |
| 335 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 336 | plines(linenr_T lnum) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 337 | { |
| 338 | return plines_win(curwin, lnum, TRUE); |
| 339 | } |
| 340 | |
| 341 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 342 | plines_win( |
| 343 | win_T *wp, |
| 344 | linenr_T lnum, |
zeertzjq | bfe377b | 2023-08-17 22:58:53 +0200 | [diff] [blame] | 345 | int limit_winheight) // when TRUE limit to window height |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 346 | { |
| 347 | #if defined(FEAT_DIFF) || defined(PROTO) |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 348 | // Check for filler lines above this buffer line. When folded the result |
| 349 | // is one line anyway. |
zeertzjq | bfe377b | 2023-08-17 22:58:53 +0200 | [diff] [blame] | 350 | return plines_win_nofill(wp, lnum, limit_winheight) |
| 351 | + diff_check_fill(wp, lnum); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 352 | } |
| 353 | |
Bram Moolenaar | 61c4b04 | 2022-10-18 15:10:11 +0100 | [diff] [blame] | 354 | /* |
| 355 | * Return the number of window lines occupied by buffer line "lnum". |
| 356 | * Does not include filler lines. |
| 357 | */ |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 358 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 359 | plines_nofill(linenr_T lnum) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 360 | { |
| 361 | return plines_win_nofill(curwin, lnum, TRUE); |
| 362 | } |
| 363 | |
| 364 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 365 | plines_win_nofill( |
| 366 | win_T *wp, |
| 367 | linenr_T lnum, |
zeertzjq | bfe377b | 2023-08-17 22:58:53 +0200 | [diff] [blame] | 368 | int limit_winheight) // when TRUE limit to window height |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 369 | { |
| 370 | #endif |
| 371 | int lines; |
| 372 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 373 | if (wp->w_width == 0) |
| 374 | return 1; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 375 | |
| 376 | #ifdef FEAT_FOLDING |
Dominique Pelle | af4a61a | 2021-12-27 17:21:41 +0000 | [diff] [blame] | 377 | // Folded lines are handled just like an empty line. |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 378 | // NOTE: Caller must handle lines that are MAYBE folded. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 379 | if (lineFolded(wp, lnum) == TRUE) |
| 380 | return 1; |
| 381 | #endif |
| 382 | |
Bram Moolenaar | 4d91d34 | 2022-08-06 13:48:20 +0100 | [diff] [blame] | 383 | if (!wp->w_p_wrap) |
| 384 | lines = 1 |
| 385 | #ifdef FEAT_PROP_POPUP |
Bram Moolenaar | c9dc03f | 2022-09-12 17:51:07 +0100 | [diff] [blame] | 386 | // add a line for each "above" and "below" aligned text property |
| 387 | + prop_count_above_below(wp->w_buffer, lnum) |
Bram Moolenaar | 4d91d34 | 2022-08-06 13:48:20 +0100 | [diff] [blame] | 388 | #endif |
| 389 | ; |
| 390 | else |
| 391 | lines = plines_win_nofold(wp, lnum); |
| 392 | |
zeertzjq | bfe377b | 2023-08-17 22:58:53 +0200 | [diff] [blame] | 393 | if (limit_winheight && lines > wp->w_height) |
=?UTF-8?q?Dundar=20G=C3=B6c?= | 420fabc | 2022-01-28 15:28:04 +0000 | [diff] [blame] | 394 | return wp->w_height; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 395 | return lines; |
| 396 | } |
| 397 | |
| 398 | /* |
| 399 | * Return number of window lines physical line "lnum" will occupy in window |
| 400 | * "wp". Does not care about folding, 'wrap' or 'diff'. |
| 401 | */ |
| 402 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 403 | plines_win_nofold(win_T *wp, linenr_T lnum) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 404 | { |
| 405 | char_u *s; |
| 406 | long col; |
| 407 | int width; |
Bram Moolenaar | 49a9079 | 2022-08-09 18:25:23 +0100 | [diff] [blame] | 408 | chartabsize_T cts; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 409 | |
| 410 | s = ml_get_buf(wp->w_buffer, lnum, FALSE); |
Bram Moolenaar | 49a9079 | 2022-08-09 18:25:23 +0100 | [diff] [blame] | 411 | init_chartabsize_arg(&cts, wp, lnum, 0, s, s); |
| 412 | if (*s == NUL |
| 413 | #ifdef FEAT_PROP_POPUP |
| 414 | && !cts.cts_has_prop_with_text |
| 415 | #endif |
| 416 | ) |
| 417 | return 1; // be quick for an empty line |
| 418 | win_linetabsize_cts(&cts, (colnr_T)MAXCOL); |
| 419 | clear_chartabsize_arg(&cts); |
| 420 | col = (int)cts.cts_vcol; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 421 | |
Bram Moolenaar | 471c0fa | 2022-08-22 15:19:16 +0100 | [diff] [blame] | 422 | // If list mode is on, then the '$' at the end of the line may take up one |
| 423 | // extra column. |
Bram Moolenaar | eed9d46 | 2021-02-15 20:38:25 +0100 | [diff] [blame] | 424 | if (wp->w_p_list && wp->w_lcs_chars.eol != NUL) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 425 | col += 1; |
| 426 | |
| 427 | /* |
Bram Moolenaar | 6448667 | 2010-05-16 15:46:46 +0200 | [diff] [blame] | 428 | * Add column offset for 'number', 'relativenumber' and 'foldcolumn'. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 429 | */ |
Bram Moolenaar | 0263146 | 2017-09-22 15:20:32 +0200 | [diff] [blame] | 430 | width = wp->w_width - win_col_off(wp); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 431 | if (width <= 0) |
| 432 | return 32000; |
| 433 | if (col <= width) |
| 434 | return 1; |
| 435 | col -= width; |
| 436 | width += win_col_off2(wp); |
| 437 | return (col + (width - 1)) / width + 1; |
| 438 | } |
| 439 | |
| 440 | /* |
| 441 | * Like plines_win(), but only reports the number of physical screen lines |
| 442 | * used from the start of the line to the given column number. |
| 443 | */ |
| 444 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 445 | plines_win_col(win_T *wp, linenr_T lnum, long column) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 446 | { |
| 447 | long col; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 448 | int lines = 0; |
| 449 | int width; |
Bram Moolenaar | 597a422 | 2014-06-25 14:39:50 +0200 | [diff] [blame] | 450 | char_u *line; |
Bram Moolenaar | 7f9969c | 2022-07-25 18:13:54 +0100 | [diff] [blame] | 451 | chartabsize_T cts; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 452 | |
| 453 | #ifdef FEAT_DIFF |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 454 | // Check for filler lines above this buffer line. When folded the result |
| 455 | // is one line anyway. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 456 | lines = diff_check_fill(wp, lnum); |
| 457 | #endif |
| 458 | |
| 459 | if (!wp->w_p_wrap) |
| 460 | return lines + 1; |
| 461 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 462 | if (wp->w_width == 0) |
| 463 | return lines + 1; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 464 | |
Bram Moolenaar | 7f9969c | 2022-07-25 18:13:54 +0100 | [diff] [blame] | 465 | line = ml_get_buf(wp->w_buffer, lnum, FALSE); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 466 | |
Bram Moolenaar | 7f9969c | 2022-07-25 18:13:54 +0100 | [diff] [blame] | 467 | init_chartabsize_arg(&cts, wp, lnum, 0, line, line); |
| 468 | while (*cts.cts_ptr != NUL && --column >= 0) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 469 | { |
Bram Moolenaar | 7f9969c | 2022-07-25 18:13:54 +0100 | [diff] [blame] | 470 | cts.cts_vcol += win_lbr_chartabsize(&cts, NULL); |
| 471 | MB_PTR_ADV(cts.cts_ptr); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 472 | } |
| 473 | |
| 474 | /* |
Bram Moolenaar | 7f9969c | 2022-07-25 18:13:54 +0100 | [diff] [blame] | 475 | * If *cts.cts_ptr is a TAB, and the TAB is not displayed as ^I, and we're |
| 476 | * not in MODE_INSERT state, then col must be adjusted so that it |
| 477 | * represents the last screen position of the TAB. This only fixes an |
| 478 | * error when the TAB wraps from one screen line to the next (when |
| 479 | * 'columns' is not a multiple of 'ts') -- webb. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 480 | */ |
Bram Moolenaar | 7f9969c | 2022-07-25 18:13:54 +0100 | [diff] [blame] | 481 | col = cts.cts_vcol; |
| 482 | if (*cts.cts_ptr == TAB && (State & MODE_NORMAL) |
Bram Moolenaar | 2495910 | 2022-05-07 20:01:16 +0100 | [diff] [blame] | 483 | && (!wp->w_p_list || wp->w_lcs_chars.tab1)) |
Bram Moolenaar | 7f9969c | 2022-07-25 18:13:54 +0100 | [diff] [blame] | 484 | col += win_lbr_chartabsize(&cts, NULL) - 1; |
| 485 | clear_chartabsize_arg(&cts); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 486 | |
| 487 | /* |
Bram Moolenaar | 6448667 | 2010-05-16 15:46:46 +0200 | [diff] [blame] | 488 | * Add column offset for 'number', 'relativenumber', 'foldcolumn', etc. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 489 | */ |
Bram Moolenaar | 0263146 | 2017-09-22 15:20:32 +0200 | [diff] [blame] | 490 | width = wp->w_width - win_col_off(wp); |
Bram Moolenaar | 2647063 | 2006-10-24 19:12:40 +0000 | [diff] [blame] | 491 | if (width <= 0) |
| 492 | return 9999; |
| 493 | |
| 494 | lines += 1; |
| 495 | if (col > width) |
| 496 | lines += (col - width) / (width + win_col_off2(wp)) + 1; |
| 497 | return lines; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 498 | } |
| 499 | |
Luuk van Baal | 32d701f | 2024-04-28 16:24:02 +0200 | [diff] [blame] | 500 | /* |
| 501 | * Return number of window lines the physical line range from "first" until |
| 502 | * "last" will occupy in window "wp". Takes into account folding, 'wrap', |
| 503 | * topfill and filler lines beyond the end of the buffer. Limit to "max" lines. |
| 504 | */ |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 505 | int |
Luuk van Baal | 32d701f | 2024-04-28 16:24:02 +0200 | [diff] [blame] | 506 | plines_m_win(win_T *wp, linenr_T first, linenr_T last, int max) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 507 | { |
| 508 | int count = 0; |
| 509 | |
Luuk van Baal | 32d701f | 2024-04-28 16:24:02 +0200 | [diff] [blame] | 510 | while (first <= last && count < max) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 511 | { |
| 512 | #ifdef FEAT_FOLDING |
| 513 | int x; |
| 514 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 515 | // Check if there are any really folded lines, but also included lines |
| 516 | // that are maybe folded. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 517 | x = foldedCount(wp, first, NULL); |
| 518 | if (x > 0) |
| 519 | { |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 520 | ++count; // count 1 for "+-- folded" line |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 521 | first += x; |
| 522 | } |
| 523 | else |
| 524 | #endif |
| 525 | { |
| 526 | #ifdef FEAT_DIFF |
| 527 | if (first == wp->w_topline) |
Luuk van Baal | 08b0f63 | 2024-04-09 22:43:49 +0200 | [diff] [blame] | 528 | count += plines_win_nofill(wp, first, FALSE) + wp->w_topfill; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 529 | else |
| 530 | #endif |
Luuk van Baal | 08b0f63 | 2024-04-09 22:43:49 +0200 | [diff] [blame] | 531 | count += plines_win(wp, first, FALSE); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 532 | ++first; |
| 533 | } |
| 534 | } |
Luuk van Baal | 08b0f63 | 2024-04-09 22:43:49 +0200 | [diff] [blame] | 535 | #ifdef FEAT_DIFF |
| 536 | if (first == wp->w_buffer->b_ml.ml_line_count + 1) |
| 537 | count += diff_check_fill(wp, first); |
| 538 | #endif |
Luuk van Baal | 32d701f | 2024-04-28 16:24:02 +0200 | [diff] [blame] | 539 | return MIN(max, count); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 540 | } |
| 541 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 542 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 543 | gchar_pos(pos_T *pos) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 544 | { |
Bram Moolenaar | 8ada6aa | 2017-12-19 21:23:21 +0100 | [diff] [blame] | 545 | char_u *ptr; |
Christian Brabandt | c9a1e25 | 2025-01-11 15:25:00 +0100 | [diff] [blame] | 546 | int ptrlen; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 547 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 548 | // When searching columns is sometimes put at the end of a line. |
Bram Moolenaar | 8ada6aa | 2017-12-19 21:23:21 +0100 | [diff] [blame] | 549 | if (pos->col == MAXCOL) |
| 550 | return NUL; |
Christian Brabandt | c9a1e25 | 2025-01-11 15:25:00 +0100 | [diff] [blame] | 551 | ptrlen = ml_get_len(pos->lnum); |
Bram Moolenaar | 8ada6aa | 2017-12-19 21:23:21 +0100 | [diff] [blame] | 552 | ptr = ml_get_pos(pos); |
Christian Brabandt | c9a1e25 | 2025-01-11 15:25:00 +0100 | [diff] [blame] | 553 | if (pos->col > ptrlen) |
| 554 | return NUL; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 555 | if (has_mbyte) |
| 556 | return (*mb_ptr2char)(ptr); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 557 | return (int)*ptr; |
| 558 | } |
| 559 | |
| 560 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 561 | gchar_cursor(void) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 562 | { |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 563 | if (has_mbyte) |
| 564 | return (*mb_ptr2char)(ml_get_cursor()); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 565 | return (int)*ml_get_cursor(); |
| 566 | } |
| 567 | |
| 568 | /* |
| 569 | * Write a character at the current cursor position. |
| 570 | * It is directly written into the block. |
| 571 | */ |
| 572 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 573 | pchar_cursor(int c) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 574 | { |
| 575 | *(ml_get_buf(curbuf, curwin->w_cursor.lnum, TRUE) |
| 576 | + curwin->w_cursor.col) = c; |
| 577 | } |
| 578 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 579 | /* |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 580 | * Skip to next part of an option argument: Skip space and comma. |
| 581 | */ |
| 582 | char_u * |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 583 | skip_to_option_part(char_u *p) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 584 | { |
| 585 | if (*p == ',') |
| 586 | ++p; |
| 587 | while (*p == ' ') |
| 588 | ++p; |
| 589 | return p; |
| 590 | } |
| 591 | |
| 592 | /* |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 593 | * check_status: called when the status bars for the buffer 'buf' |
| 594 | * need to be updated |
| 595 | */ |
| 596 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 597 | check_status(buf_T *buf) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 598 | { |
| 599 | win_T *wp; |
| 600 | |
Bram Moolenaar | 2932359 | 2016-07-24 22:04:11 +0200 | [diff] [blame] | 601 | FOR_ALL_WINDOWS(wp) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 602 | if (wp->w_buffer == buf && wp->w_status_height) |
| 603 | { |
| 604 | wp->w_redr_status = TRUE; |
Bram Moolenaar | 471c0fa | 2022-08-22 15:19:16 +0100 | [diff] [blame] | 605 | set_must_redraw(UPD_VALID); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 606 | } |
| 607 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 608 | |
| 609 | /* |
Bram Moolenaar | 6ed545e | 2022-05-09 20:09:23 +0100 | [diff] [blame] | 610 | * Ask for a reply from the user, a 'y' or a 'n', with prompt "str" (which |
| 611 | * should have been translated already). |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 612 | * No other characters are accepted, the message is repeated until a valid |
| 613 | * reply is entered or CTRL-C is hit. |
| 614 | * If direct is TRUE, don't use vgetc() but ui_inchar(), don't get characters |
| 615 | * from any buffers but directly from the user. |
| 616 | * |
| 617 | * return the 'y' or 'n' |
| 618 | */ |
| 619 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 620 | ask_yesno(char_u *str, int direct) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 621 | { |
| 622 | int r = ' '; |
| 623 | int save_State = State; |
| 624 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 625 | if (exiting) // put terminal in raw mode for this question |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 626 | settmode(TMODE_RAW); |
| 627 | ++no_wait_return; |
| 628 | #ifdef USE_ON_FLY_SCROLL |
Bram Moolenaar | b20b9e1 | 2019-09-21 20:48:04 +0200 | [diff] [blame] | 629 | dont_scroll = TRUE; // disallow scrolling here |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 630 | #endif |
Bram Moolenaar | 2495910 | 2022-05-07 20:01:16 +0100 | [diff] [blame] | 631 | State = MODE_CONFIRM; // mouse behaves like with :confirm |
Bram Moolenaar | b20b9e1 | 2019-09-21 20:48:04 +0200 | [diff] [blame] | 632 | setmouse(); // disables mouse for xterm |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 633 | ++no_mapping; |
Bram Moolenaar | b20b9e1 | 2019-09-21 20:48:04 +0200 | [diff] [blame] | 634 | ++allow_keys; // no mapping here, but recognize keys |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 635 | |
| 636 | while (r != 'y' && r != 'n') |
| 637 | { |
Bram Moolenaar | 13608d8 | 2022-08-29 15:06:50 +0100 | [diff] [blame] | 638 | // same highlighting as for wait_return() |
Bram Moolenaar | f9e3e09 | 2019-01-13 23:38:42 +0100 | [diff] [blame] | 639 | smsg_attr(HL_ATTR(HLF_R), "%s (y/n)?", str); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 640 | if (direct) |
| 641 | r = get_keystroke(); |
| 642 | else |
Bram Moolenaar | 913626c | 2008-01-03 11:43:42 +0000 | [diff] [blame] | 643 | r = plain_vgetc(); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 644 | if (r == Ctrl_C || r == ESC) |
| 645 | r = 'n'; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 646 | msg_putchar(r); // show what you typed |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 647 | out_flush(); |
| 648 | } |
| 649 | --no_wait_return; |
| 650 | State = save_State; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 651 | setmouse(); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 652 | --no_mapping; |
| 653 | --allow_keys; |
| 654 | |
| 655 | return r; |
| 656 | } |
| 657 | |
Bram Moolenaar | 0e57dd8 | 2019-09-16 22:56:03 +0200 | [diff] [blame] | 658 | #if defined(FEAT_EVAL) || defined(PROTO) |
| 659 | |
| 660 | /* |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 661 | * Returns the current mode as a string in "buf[MODE_MAX_LENGTH]", NUL |
| 662 | * terminated. |
| 663 | * The first character represents the major mode, the following ones the minor |
| 664 | * ones. |
| 665 | */ |
| 666 | void |
| 667 | get_mode(char_u *buf) |
| 668 | { |
| 669 | int i = 0; |
| 670 | |
| 671 | if (time_for_testing == 93784) |
| 672 | { |
| 673 | // Testing the two-character code. |
| 674 | buf[i++] = 'x'; |
| 675 | buf[i++] = '!'; |
| 676 | } |
| 677 | #ifdef FEAT_TERMINAL |
| 678 | else if (term_use_loop()) |
h-east | 71ebf3b | 2023-09-03 17:12:55 +0200 | [diff] [blame] | 679 | { |
| 680 | if (State & MODE_CMDLINE) |
| 681 | buf[i++] = 'c'; |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 682 | buf[i++] = 't'; |
h-east | 71ebf3b | 2023-09-03 17:12:55 +0200 | [diff] [blame] | 683 | } |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 684 | #endif |
Bram Moolenaar | 2495910 | 2022-05-07 20:01:16 +0100 | [diff] [blame] | 685 | else if (State == MODE_HITRETURN || State == MODE_ASKMORE |
| 686 | || State == MODE_SETWSIZE |
| 687 | || State == MODE_CONFIRM) |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 688 | { |
| 689 | buf[i++] = 'r'; |
Bram Moolenaar | 2495910 | 2022-05-07 20:01:16 +0100 | [diff] [blame] | 690 | if (State == MODE_ASKMORE) |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 691 | buf[i++] = 'm'; |
Bram Moolenaar | 2495910 | 2022-05-07 20:01:16 +0100 | [diff] [blame] | 692 | else if (State == MODE_CONFIRM) |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 693 | buf[i++] = '?'; |
| 694 | } |
Bram Moolenaar | 2495910 | 2022-05-07 20:01:16 +0100 | [diff] [blame] | 695 | else if (State == MODE_EXTERNCMD) |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 696 | buf[i++] = '!'; |
Bram Moolenaar | 2495910 | 2022-05-07 20:01:16 +0100 | [diff] [blame] | 697 | else if (State & MODE_INSERT) |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 698 | { |
| 699 | if (State & VREPLACE_FLAG) |
| 700 | { |
| 701 | buf[i++] = 'R'; |
| 702 | buf[i++] = 'v'; |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 703 | } |
| 704 | else |
| 705 | { |
| 706 | if (State & REPLACE_FLAG) |
| 707 | buf[i++] = 'R'; |
| 708 | else |
| 709 | buf[i++] = 'i'; |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 710 | } |
zeertzjq | 590f365 | 2022-04-29 11:29:54 +0100 | [diff] [blame] | 711 | |
| 712 | if (ins_compl_active()) |
| 713 | buf[i++] = 'c'; |
| 714 | else if (ctrl_x_mode_not_defined_yet()) |
| 715 | buf[i++] = 'x'; |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 716 | } |
Bram Moolenaar | 2495910 | 2022-05-07 20:01:16 +0100 | [diff] [blame] | 717 | else if ((State & MODE_CMDLINE) || exmode_active) |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 718 | { |
| 719 | buf[i++] = 'c'; |
| 720 | if (exmode_active == EXMODE_VIM) |
| 721 | buf[i++] = 'v'; |
| 722 | else if (exmode_active == EXMODE_NORMAL) |
| 723 | buf[i++] = 'e'; |
Sam-programs | d1c3ef1 | 2023-11-27 22:22:51 +0100 | [diff] [blame] | 724 | if ((State & MODE_CMDLINE) && cmdline_overstrike()) |
| 725 | buf[i++] = 'r'; |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 726 | } |
kuuote | 0fd1cb1 | 2024-08-20 19:53:17 +0200 | [diff] [blame] | 727 | else if (VIsual_active) |
| 728 | { |
| 729 | if (VIsual_select) |
| 730 | buf[i++] = VIsual_mode + 's' - 'v'; |
| 731 | else |
| 732 | { |
| 733 | buf[i++] = VIsual_mode; |
| 734 | if (restart_VIsual_select) |
| 735 | buf[i++] = 's'; |
| 736 | } |
| 737 | } |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 738 | else |
| 739 | { |
| 740 | buf[i++] = 'n'; |
| 741 | if (finish_op) |
| 742 | { |
| 743 | buf[i++] = 'o'; |
| 744 | // to be able to detect force-linewise/blockwise/characterwise |
| 745 | // operations |
| 746 | buf[i++] = motion_force; |
| 747 | } |
| 748 | else if (restart_edit == 'I' || restart_edit == 'R' |
| 749 | || restart_edit == 'V') |
| 750 | { |
| 751 | buf[i++] = 'i'; |
| 752 | buf[i++] = restart_edit; |
| 753 | } |
| 754 | #ifdef FEAT_TERMINAL |
| 755 | else if (term_in_normal_mode()) |
| 756 | buf[i++] = 't'; |
| 757 | #endif |
| 758 | } |
| 759 | |
| 760 | buf[i] = NUL; |
| 761 | } |
| 762 | |
| 763 | /* |
Bram Moolenaar | 0e57dd8 | 2019-09-16 22:56:03 +0200 | [diff] [blame] | 764 | * "mode()" function |
| 765 | */ |
| 766 | void |
| 767 | f_mode(typval_T *argvars, typval_T *rettv) |
| 768 | { |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 769 | char_u buf[MODE_MAX_LENGTH]; |
Bram Moolenaar | 0e57dd8 | 2019-09-16 22:56:03 +0200 | [diff] [blame] | 770 | |
Yegappan Lakshmanan | 4490ec4 | 2021-07-27 22:00:44 +0200 | [diff] [blame] | 771 | if (in_vim9script() && check_for_opt_bool_arg(argvars, 0) == FAIL) |
| 772 | return; |
| 773 | |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 774 | get_mode(buf); |
Bram Moolenaar | 0e57dd8 | 2019-09-16 22:56:03 +0200 | [diff] [blame] | 775 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 776 | // Clear out the minor mode when the argument is not a non-zero number or |
| 777 | // non-empty string. |
Bram Moolenaar | 0e57dd8 | 2019-09-16 22:56:03 +0200 | [diff] [blame] | 778 | if (!non_zero_arg(&argvars[0])) |
| 779 | buf[1] = NUL; |
| 780 | |
| 781 | rettv->vval.v_string = vim_strsave(buf); |
| 782 | rettv->v_type = VAR_STRING; |
| 783 | } |
| 784 | |
| 785 | static void |
| 786 | may_add_state_char(garray_T *gap, char_u *include, int c) |
| 787 | { |
| 788 | if (include == NULL || vim_strchr(include, c) != NULL) |
| 789 | ga_append(gap, c); |
| 790 | } |
| 791 | |
| 792 | /* |
| 793 | * "state()" function |
| 794 | */ |
| 795 | void |
| 796 | f_state(typval_T *argvars, typval_T *rettv) |
| 797 | { |
| 798 | garray_T ga; |
| 799 | char_u *include = NULL; |
| 800 | int i; |
| 801 | |
Yegappan Lakshmanan | 4490ec4 | 2021-07-27 22:00:44 +0200 | [diff] [blame] | 802 | if (in_vim9script() && check_for_opt_string_arg(argvars, 0) == FAIL) |
| 803 | return; |
| 804 | |
Bram Moolenaar | 0e57dd8 | 2019-09-16 22:56:03 +0200 | [diff] [blame] | 805 | ga_init2(&ga, 1, 20); |
| 806 | if (argvars[0].v_type != VAR_UNKNOWN) |
| 807 | include = tv_get_string(&argvars[0]); |
| 808 | |
| 809 | if (!(stuff_empty() && typebuf.tb_len == 0 && scriptin[curscript] == NULL)) |
| 810 | may_add_state_char(&ga, include, 'm'); |
| 811 | if (op_pending()) |
| 812 | may_add_state_char(&ga, include, 'o'); |
| 813 | if (autocmd_busy) |
| 814 | may_add_state_char(&ga, include, 'x'); |
Bram Moolenaar | c258549 | 2019-09-22 21:29:53 +0200 | [diff] [blame] | 815 | if (ins_compl_active()) |
Bram Moolenaar | 0e57dd8 | 2019-09-16 22:56:03 +0200 | [diff] [blame] | 816 | may_add_state_char(&ga, include, 'a'); |
| 817 | |
| 818 | # ifdef FEAT_JOB_CHANNEL |
| 819 | if (channel_in_blocking_wait()) |
| 820 | may_add_state_char(&ga, include, 'w'); |
| 821 | # endif |
Bram Moolenaar | b20b9e1 | 2019-09-21 20:48:04 +0200 | [diff] [blame] | 822 | if (!get_was_safe_state()) |
| 823 | may_add_state_char(&ga, include, 'S'); |
Bram Moolenaar | 0e57dd8 | 2019-09-16 22:56:03 +0200 | [diff] [blame] | 824 | for (i = 0; i < get_callback_depth() && i < 3; ++i) |
| 825 | may_add_state_char(&ga, include, 'c'); |
| 826 | if (msg_scrolled > 0) |
| 827 | may_add_state_char(&ga, include, 's'); |
| 828 | |
| 829 | rettv->v_type = VAR_STRING; |
| 830 | rettv->vval.v_string = ga.ga_data; |
| 831 | } |
| 832 | |
| 833 | #endif // FEAT_EVAL |
| 834 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 835 | /* |
| 836 | * Get a key stroke directly from the user. |
| 837 | * Ignores mouse clicks and scrollbar events, except a click for the left |
| 838 | * button (used at the more prompt). |
| 839 | * Doesn't use vgetc(), because it syncs undo and eats mapped characters. |
| 840 | * Disadvantage: typeahead is ignored. |
| 841 | * Translates the interrupt character for unix to ESC. |
| 842 | */ |
| 843 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 844 | get_keystroke(void) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 845 | { |
Bram Moolenaar | a8c8a68 | 2012-02-05 22:05:48 +0100 | [diff] [blame] | 846 | char_u *buf = NULL; |
| 847 | int buflen = 150; |
| 848 | int maxlen; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 849 | int len = 0; |
| 850 | int n; |
| 851 | int save_mapped_ctrl_c = mapped_ctrl_c; |
Bram Moolenaar | 4395a71 | 2006-09-05 18:57:57 +0000 | [diff] [blame] | 852 | int waited = 0; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 853 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 854 | mapped_ctrl_c = FALSE; // mappings are not used here |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 855 | for (;;) |
| 856 | { |
| 857 | cursor_on(); |
| 858 | out_flush(); |
| 859 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 860 | // Leave some room for check_termcode() to insert a key code into (max |
| 861 | // 5 chars plus NUL). And fix_input_buffer() can triple the number of |
| 862 | // bytes. |
Bram Moolenaar | a8c8a68 | 2012-02-05 22:05:48 +0100 | [diff] [blame] | 863 | maxlen = (buflen - 6 - len) / 3; |
| 864 | if (buf == NULL) |
| 865 | buf = alloc(buflen); |
| 866 | else if (maxlen < 10) |
| 867 | { |
Bram Moolenaar | 9abd5c6 | 2015-02-10 18:34:01 +0100 | [diff] [blame] | 868 | char_u *t_buf = buf; |
| 869 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 870 | // Need some more space. This might happen when receiving a long |
| 871 | // escape sequence. |
Bram Moolenaar | a8c8a68 | 2012-02-05 22:05:48 +0100 | [diff] [blame] | 872 | buflen += 100; |
| 873 | buf = vim_realloc(buf, buflen); |
Bram Moolenaar | 9abd5c6 | 2015-02-10 18:34:01 +0100 | [diff] [blame] | 874 | if (buf == NULL) |
| 875 | vim_free(t_buf); |
Bram Moolenaar | a8c8a68 | 2012-02-05 22:05:48 +0100 | [diff] [blame] | 876 | maxlen = (buflen - 6 - len) / 3; |
| 877 | } |
| 878 | if (buf == NULL) |
| 879 | { |
| 880 | do_outofmem_msg((long_u)buflen); |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 881 | return ESC; // panic! |
Bram Moolenaar | a8c8a68 | 2012-02-05 22:05:48 +0100 | [diff] [blame] | 882 | } |
| 883 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 884 | // First time: blocking wait. Second time: wait up to 100ms for a |
| 885 | // terminal code to complete. |
Bram Moolenaar | a8c8a68 | 2012-02-05 22:05:48 +0100 | [diff] [blame] | 886 | n = ui_inchar(buf + len, maxlen, len == 0 ? -1L : 100L, 0); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 887 | if (n > 0) |
| 888 | { |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 889 | // Replace zero and CSI by a special key code. |
Bram Moolenaar | 6bff02e | 2016-08-16 22:50:55 +0200 | [diff] [blame] | 890 | n = fix_input_buffer(buf + len, n); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 891 | len += n; |
Bram Moolenaar | 4395a71 | 2006-09-05 18:57:57 +0000 | [diff] [blame] | 892 | waited = 0; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 893 | } |
Bram Moolenaar | 4395a71 | 2006-09-05 18:57:57 +0000 | [diff] [blame] | 894 | else if (len > 0) |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 895 | ++waited; // keep track of the waiting time |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 896 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 897 | // Incomplete termcode and not timed out yet: get more characters |
Bram Moolenaar | a8c8a68 | 2012-02-05 22:05:48 +0100 | [diff] [blame] | 898 | if ((n = check_termcode(1, buf, buflen, &len)) < 0 |
Bram Moolenaar | 4395a71 | 2006-09-05 18:57:57 +0000 | [diff] [blame] | 899 | && (!p_ttimeout || waited * 100L < (p_ttm < 0 ? p_tm : p_ttm))) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 900 | continue; |
Bram Moolenaar | 4395a71 | 2006-09-05 18:57:57 +0000 | [diff] [blame] | 901 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 902 | if (n == KEYLEN_REMOVED) // key code removed |
Bram Moolenaar | 6eb634e | 2011-03-03 15:04:08 +0100 | [diff] [blame] | 903 | { |
Bram Moolenaar | 2495910 | 2022-05-07 20:01:16 +0100 | [diff] [blame] | 904 | if (must_redraw != 0 && !need_wait_return && (State |
| 905 | & (MODE_CMDLINE | MODE_HITRETURN | MODE_ASKMORE)) == 0) |
Bram Moolenaar | 6eb634e | 2011-03-03 15:04:08 +0100 | [diff] [blame] | 906 | { |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 907 | // Redrawing was postponed, do it now. |
Bram Moolenaar | 6eb634e | 2011-03-03 15:04:08 +0100 | [diff] [blame] | 908 | update_screen(0); |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 909 | setcursor(); // put cursor back where it belongs |
Bram Moolenaar | 6eb634e | 2011-03-03 15:04:08 +0100 | [diff] [blame] | 910 | } |
Bram Moolenaar | 946ffd4 | 2010-12-30 12:30:31 +0100 | [diff] [blame] | 911 | continue; |
Bram Moolenaar | 6eb634e | 2011-03-03 15:04:08 +0100 | [diff] [blame] | 912 | } |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 913 | if (n > 0) // found a termcode: adjust length |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 914 | len = n; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 915 | if (len == 0) // nothing typed yet |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 916 | continue; |
| 917 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 918 | // Handle modifier and/or special key code. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 919 | n = buf[0]; |
| 920 | if (n == K_SPECIAL) |
| 921 | { |
| 922 | n = TO_SPECIAL(buf[1], buf[2]); |
| 923 | if (buf[1] == KS_MODIFIER |
| 924 | || n == K_IGNORE |
Bram Moolenaar | 2526ef2 | 2013-03-16 14:20:51 +0100 | [diff] [blame] | 925 | || (is_mouse_key(n) && n != K_LEFTMOUSE) |
| 926 | #ifdef FEAT_GUI |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 927 | || n == K_VER_SCROLLBAR |
| 928 | || n == K_HOR_SCROLLBAR |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 929 | #endif |
| 930 | ) |
| 931 | { |
| 932 | if (buf[1] == KS_MODIFIER) |
| 933 | mod_mask = buf[2]; |
| 934 | len -= 3; |
| 935 | if (len > 0) |
| 936 | mch_memmove(buf, buf + 3, (size_t)len); |
| 937 | continue; |
| 938 | } |
Bram Moolenaar | 7fc904b | 2006-04-13 20:37:35 +0000 | [diff] [blame] | 939 | break; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 940 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 941 | if (has_mbyte) |
| 942 | { |
| 943 | if (MB_BYTE2LEN(n) > len) |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 944 | continue; // more bytes to get |
Bram Moolenaar | a8c8a68 | 2012-02-05 22:05:48 +0100 | [diff] [blame] | 945 | buf[len >= buflen ? buflen - 1 : len] = NUL; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 946 | n = (*mb_ptr2char)(buf); |
| 947 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 948 | #ifdef UNIX |
| 949 | if (n == intr_char) |
| 950 | n = ESC; |
| 951 | #endif |
| 952 | break; |
| 953 | } |
Bram Moolenaar | a8c8a68 | 2012-02-05 22:05:48 +0100 | [diff] [blame] | 954 | vim_free(buf); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 955 | |
| 956 | mapped_ctrl_c = save_mapped_ctrl_c; |
| 957 | return n; |
| 958 | } |
| 959 | |
Yegappan Lakshmanan | a04003a | 2024-10-27 21:54:11 +0100 | [diff] [blame] | 960 | // For overflow detection, add a digit safely to an int value. |
| 961 | static int |
| 962 | vim_append_digit_int(int *value, int digit) |
| 963 | { |
| 964 | int x = *value; |
| 965 | if (x > ((INT_MAX - digit) / 10)) |
| 966 | return FAIL; |
| 967 | *value = x * 10 + digit; |
| 968 | return OK; |
| 969 | } |
| 970 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 971 | /* |
Bram Moolenaar | 24bbcfe | 2005-06-28 23:32:02 +0000 | [diff] [blame] | 972 | * Get a number from the user. |
| 973 | * When "mouse_used" is not NULL allow using the mouse. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 974 | */ |
| 975 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 976 | get_number( |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 977 | int colon, // allow colon to abort |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 978 | int *mouse_used) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 979 | { |
| 980 | int n = 0; |
| 981 | int c; |
Bram Moolenaar | 3991dab | 2006-03-27 17:01:56 +0000 | [diff] [blame] | 982 | int typed = 0; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 983 | |
Bram Moolenaar | 24bbcfe | 2005-06-28 23:32:02 +0000 | [diff] [blame] | 984 | if (mouse_used != NULL) |
| 985 | *mouse_used = FALSE; |
| 986 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 987 | // When not printing messages, the user won't know what to type, return a |
| 988 | // zero (as if CR was hit). |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 989 | if (msg_silent != 0) |
| 990 | return 0; |
| 991 | |
| 992 | #ifdef USE_ON_FLY_SCROLL |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 993 | dont_scroll = TRUE; // disallow scrolling here |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 994 | #endif |
| 995 | ++no_mapping; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 996 | ++allow_keys; // no mapping here, but recognize keys |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 997 | for (;;) |
| 998 | { |
| 999 | windgoto(msg_row, msg_col); |
| 1000 | c = safe_vgetc(); |
| 1001 | if (VIM_ISDIGIT(c)) |
| 1002 | { |
Christian Brabandt | 22cbc8a | 2023-11-19 10:47:21 +0100 | [diff] [blame] | 1003 | if (vim_append_digit_int(&n, c - '0') == FAIL) |
Christian Brabandt | 73b2d37 | 2023-11-14 21:58:26 +0100 | [diff] [blame] | 1004 | return 0; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1005 | msg_putchar(c); |
Bram Moolenaar | 3991dab | 2006-03-27 17:01:56 +0000 | [diff] [blame] | 1006 | ++typed; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1007 | } |
| 1008 | else if (c == K_DEL || c == K_KDEL || c == K_BS || c == Ctrl_H) |
| 1009 | { |
Bram Moolenaar | 3991dab | 2006-03-27 17:01:56 +0000 | [diff] [blame] | 1010 | if (typed > 0) |
| 1011 | { |
Bram Moolenaar | 32526b3 | 2019-01-19 17:43:09 +0100 | [diff] [blame] | 1012 | msg_puts("\b \b"); |
Bram Moolenaar | 3991dab | 2006-03-27 17:01:56 +0000 | [diff] [blame] | 1013 | --typed; |
| 1014 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1015 | n /= 10; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1016 | } |
Bram Moolenaar | 24bbcfe | 2005-06-28 23:32:02 +0000 | [diff] [blame] | 1017 | else if (mouse_used != NULL && c == K_LEFTMOUSE) |
| 1018 | { |
| 1019 | *mouse_used = TRUE; |
| 1020 | n = mouse_row + 1; |
| 1021 | break; |
| 1022 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1023 | else if (n == 0 && c == ':' && colon) |
| 1024 | { |
| 1025 | stuffcharReadbuff(':'); |
| 1026 | if (!exmode_active) |
| 1027 | cmdline_row = msg_row; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1028 | skip_redraw = TRUE; // skip redraw once |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1029 | do_redraw = FALSE; |
| 1030 | break; |
| 1031 | } |
=?UTF-8?q?Luka=20Marku=C5=A1i=C4=87?= | 5cf9457 | 2021-05-20 21:14:20 +0200 | [diff] [blame] | 1032 | else if (c == Ctrl_C || c == ESC || c == 'q') |
| 1033 | { |
| 1034 | n = 0; |
| 1035 | break; |
| 1036 | } |
| 1037 | else if (c == CAR || c == NL ) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1038 | break; |
| 1039 | } |
| 1040 | --no_mapping; |
| 1041 | --allow_keys; |
| 1042 | return n; |
| 1043 | } |
| 1044 | |
Bram Moolenaar | 9ba0eb8 | 2005-06-13 22:28:56 +0000 | [diff] [blame] | 1045 | /* |
| 1046 | * Ask the user to enter a number. |
Bram Moolenaar | 24bbcfe | 2005-06-28 23:32:02 +0000 | [diff] [blame] | 1047 | * When "mouse_used" is not NULL allow using the mouse and in that case return |
| 1048 | * the line number. |
Bram Moolenaar | 9ba0eb8 | 2005-06-13 22:28:56 +0000 | [diff] [blame] | 1049 | */ |
| 1050 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1051 | prompt_for_number(int *mouse_used) |
Bram Moolenaar | 9ba0eb8 | 2005-06-13 22:28:56 +0000 | [diff] [blame] | 1052 | { |
| 1053 | int i; |
Bram Moolenaar | d857f0e | 2005-06-21 22:37:39 +0000 | [diff] [blame] | 1054 | int save_cmdline_row; |
| 1055 | int save_State; |
Bram Moolenaar | 9ba0eb8 | 2005-06-13 22:28:56 +0000 | [diff] [blame] | 1056 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1057 | // When using ":silent" assume that <CR> was entered. |
Bram Moolenaar | 42eeac3 | 2005-06-29 22:40:58 +0000 | [diff] [blame] | 1058 | if (mouse_used != NULL) |
Bram Moolenaar | eebd555 | 2020-06-10 15:45:57 +0200 | [diff] [blame] | 1059 | msg_puts(_("Type number and <Enter> or click with the mouse (q or empty cancels): ")); |
Bram Moolenaar | 42eeac3 | 2005-06-29 22:40:58 +0000 | [diff] [blame] | 1060 | else |
Bram Moolenaar | eebd555 | 2020-06-10 15:45:57 +0200 | [diff] [blame] | 1061 | msg_puts(_("Type number and <Enter> (q or empty cancels): ")); |
Bram Moolenaar | d857f0e | 2005-06-21 22:37:39 +0000 | [diff] [blame] | 1062 | |
Bram Moolenaar | 4cbdf15 | 2018-08-26 21:23:07 +0200 | [diff] [blame] | 1063 | // Set the state such that text can be selected/copied/pasted and we still |
| 1064 | // get mouse events. redraw_after_callback() will not redraw if cmdline_row |
| 1065 | // is zero. |
Bram Moolenaar | d857f0e | 2005-06-21 22:37:39 +0000 | [diff] [blame] | 1066 | save_cmdline_row = cmdline_row; |
Bram Moolenaar | 203335e | 2006-09-03 14:35:42 +0000 | [diff] [blame] | 1067 | cmdline_row = 0; |
Bram Moolenaar | d857f0e | 2005-06-21 22:37:39 +0000 | [diff] [blame] | 1068 | save_State = State; |
Bram Moolenaar | 2495910 | 2022-05-07 20:01:16 +0100 | [diff] [blame] | 1069 | State = MODE_CMDLINE; |
Bram Moolenaar | 4cbdf15 | 2018-08-26 21:23:07 +0200 | [diff] [blame] | 1070 | // May show different mouse shape. |
Bram Moolenaar | 7365831 | 2018-04-24 17:41:57 +0200 | [diff] [blame] | 1071 | setmouse(); |
Bram Moolenaar | 7365831 | 2018-04-24 17:41:57 +0200 | [diff] [blame] | 1072 | |
Bram Moolenaar | 24bbcfe | 2005-06-28 23:32:02 +0000 | [diff] [blame] | 1073 | i = get_number(TRUE, mouse_used); |
| 1074 | if (KeyTyped) |
Bram Moolenaar | 9ba0eb8 | 2005-06-13 22:28:56 +0000 | [diff] [blame] | 1075 | { |
Bram Moolenaar | 954bb06 | 2019-06-09 16:40:46 +0200 | [diff] [blame] | 1076 | // don't call wait_return() now |
| 1077 | if (msg_row > 0) |
| 1078 | cmdline_row = msg_row - 1; |
Bram Moolenaar | 9ba0eb8 | 2005-06-13 22:28:56 +0000 | [diff] [blame] | 1079 | need_wait_return = FALSE; |
Bram Moolenaar | dc968e7 | 2019-11-05 21:53:20 +0100 | [diff] [blame] | 1080 | msg_didany = FALSE; |
| 1081 | msg_didout = FALSE; |
Bram Moolenaar | 9ba0eb8 | 2005-06-13 22:28:56 +0000 | [diff] [blame] | 1082 | } |
Bram Moolenaar | d857f0e | 2005-06-21 22:37:39 +0000 | [diff] [blame] | 1083 | else |
| 1084 | cmdline_row = save_cmdline_row; |
| 1085 | State = save_State; |
Bram Moolenaar | 4cbdf15 | 2018-08-26 21:23:07 +0200 | [diff] [blame] | 1086 | // May need to restore mouse shape. |
Bram Moolenaar | 7365831 | 2018-04-24 17:41:57 +0200 | [diff] [blame] | 1087 | setmouse(); |
Bram Moolenaar | d857f0e | 2005-06-21 22:37:39 +0000 | [diff] [blame] | 1088 | |
Bram Moolenaar | 9ba0eb8 | 2005-06-13 22:28:56 +0000 | [diff] [blame] | 1089 | return i; |
| 1090 | } |
| 1091 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1092 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1093 | msgmore(long n) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1094 | { |
| 1095 | long pn; |
| 1096 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1097 | if (global_busy // no messages now, wait until global is finished |
| 1098 | || !messaging()) // 'lazyredraw' set, don't do messages now |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1099 | return; |
| 1100 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1101 | // We don't want to overwrite another important message, but do overwrite |
| 1102 | // a previous "more lines" or "fewer lines" message, so that "5dd" and |
| 1103 | // then "put" reports the last action. |
Bram Moolenaar | 7df2d66 | 2005-01-25 22:18:08 +0000 | [diff] [blame] | 1104 | if (keep_msg != NULL && !keep_msg_more) |
| 1105 | return; |
| 1106 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1107 | if (n > 0) |
| 1108 | pn = n; |
| 1109 | else |
| 1110 | pn = -n; |
| 1111 | |
| 1112 | if (pn > p_report) |
| 1113 | { |
Bram Moolenaar | da6e891 | 2018-08-21 15:12:14 +0200 | [diff] [blame] | 1114 | if (n > 0) |
Bram Moolenaar | 32526b3 | 2019-01-19 17:43:09 +0100 | [diff] [blame] | 1115 | vim_snprintf(msg_buf, MSG_BUF_LEN, |
Bram Moolenaar | da6e891 | 2018-08-21 15:12:14 +0200 | [diff] [blame] | 1116 | NGETTEXT("%ld more line", "%ld more lines", pn), pn); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1117 | else |
Bram Moolenaar | 32526b3 | 2019-01-19 17:43:09 +0100 | [diff] [blame] | 1118 | vim_snprintf(msg_buf, MSG_BUF_LEN, |
Bram Moolenaar | da6e891 | 2018-08-21 15:12:14 +0200 | [diff] [blame] | 1119 | NGETTEXT("%ld line less", "%ld fewer lines", pn), pn); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1120 | if (got_int) |
Bram Moolenaar | 32526b3 | 2019-01-19 17:43:09 +0100 | [diff] [blame] | 1121 | vim_strcat((char_u *)msg_buf, (char_u *)_(" (Interrupted)"), |
| 1122 | MSG_BUF_LEN); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1123 | if (msg(msg_buf)) |
| 1124 | { |
Bram Moolenaar | 32526b3 | 2019-01-19 17:43:09 +0100 | [diff] [blame] | 1125 | set_keep_msg((char_u *)msg_buf, 0); |
Bram Moolenaar | 7df2d66 | 2005-01-25 22:18:08 +0000 | [diff] [blame] | 1126 | keep_msg_more = TRUE; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1127 | } |
| 1128 | } |
| 1129 | } |
| 1130 | |
| 1131 | /* |
| 1132 | * flush map and typeahead buffers and give a warning for an error |
| 1133 | */ |
| 1134 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1135 | beep_flush(void) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1136 | { |
| 1137 | if (emsg_silent == 0) |
| 1138 | { |
Bram Moolenaar | 6a2633b | 2018-10-07 23:16:36 +0200 | [diff] [blame] | 1139 | flush_buffers(FLUSH_MINIMAL); |
Bram Moolenaar | 165bc69 | 2015-07-21 17:53:25 +0200 | [diff] [blame] | 1140 | vim_beep(BO_ERROR); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1141 | } |
| 1142 | } |
| 1143 | |
| 1144 | /* |
Bram Moolenaar | aae9762 | 2022-04-13 14:28:07 +0100 | [diff] [blame] | 1145 | * Give a warning for an error. "val" is one of the BO_ values, e.g., BO_OPER. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1146 | */ |
| 1147 | void |
Bram Moolenaar | aae9762 | 2022-04-13 14:28:07 +0100 | [diff] [blame] | 1148 | vim_beep(unsigned val) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1149 | { |
Bram Moolenaar | b48e96f | 2018-02-13 12:26:14 +0100 | [diff] [blame] | 1150 | #ifdef FEAT_EVAL |
| 1151 | called_vim_beep = TRUE; |
| 1152 | #endif |
| 1153 | |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1154 | if (emsg_silent != 0 || in_assert_fails) |
| 1155 | return; |
| 1156 | |
| 1157 | if (!((bo_flags & val) || (bo_flags & BO_ALL))) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1158 | { |
Bram Moolenaar | 2e147ca | 2017-06-27 17:09:37 +0200 | [diff] [blame] | 1159 | #ifdef ELAPSED_FUNC |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1160 | static int did_init = FALSE; |
| 1161 | static elapsed_T start_tv; |
Bram Moolenaar | 2e147ca | 2017-06-27 17:09:37 +0200 | [diff] [blame] | 1162 | |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1163 | // Only beep once per half a second, otherwise a sequence of beeps |
| 1164 | // would freeze Vim. |
| 1165 | if (!did_init || ELAPSED_FUNC(start_tv) > 500) |
| 1166 | { |
| 1167 | did_init = TRUE; |
| 1168 | ELAPSED_INIT(start_tv); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1169 | #endif |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1170 | if (p_vb |
Bram Moolenaar | 2e147ca | 2017-06-27 17:09:37 +0200 | [diff] [blame] | 1171 | #ifdef FEAT_GUI |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1172 | // While the GUI is starting up the termcap is set for |
| 1173 | // the GUI but the output still goes to a terminal. |
| 1174 | && !(gui.in_use && gui.starting) |
Bram Moolenaar | 2e147ca | 2017-06-27 17:09:37 +0200 | [diff] [blame] | 1175 | #endif |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1176 | ) |
| 1177 | { |
| 1178 | out_str_cf(T_VB); |
Bram Moolenaar | cafafb3 | 2018-02-22 21:07:09 +0100 | [diff] [blame] | 1179 | #ifdef FEAT_VTP |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1180 | // No restore color information, refresh the screen. |
| 1181 | if (has_vtp_working() != 0 |
Bram Moolenaar | cafafb3 | 2018-02-22 21:07:09 +0100 | [diff] [blame] | 1182 | # ifdef FEAT_TERMGUICOLORS |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1183 | && (p_tgc || (!p_tgc && t_colors >= 256)) |
Bram Moolenaar | cafafb3 | 2018-02-22 21:07:09 +0100 | [diff] [blame] | 1184 | # endif |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1185 | ) |
| 1186 | { |
| 1187 | redraw_later(UPD_CLEAR); |
| 1188 | update_screen(0); |
| 1189 | redrawcmd(); |
Bram Moolenaar | cafafb3 | 2018-02-22 21:07:09 +0100 | [diff] [blame] | 1190 | } |
Bram Moolenaar | 2e147ca | 2017-06-27 17:09:37 +0200 | [diff] [blame] | 1191 | #endif |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1192 | } |
| 1193 | else |
| 1194 | out_char(BELL); |
| 1195 | #ifdef ELAPSED_FUNC |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1196 | } |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1197 | #endif |
| 1198 | } |
Bram Moolenaar | 5313dcb | 2005-02-22 08:56:13 +0000 | [diff] [blame] | 1199 | |
Yegappan Lakshmanan | e857598 | 2023-01-14 12:32:28 +0000 | [diff] [blame] | 1200 | // When 'debug' contains "beep" produce a message. If we are sourcing |
| 1201 | // a script or executing a function give the user a hint where the beep |
| 1202 | // comes from. |
| 1203 | if (vim_strchr(p_debug, 'e') != NULL) |
| 1204 | { |
| 1205 | msg_source(HL_ATTR(HLF_W)); |
| 1206 | msg_attr(_("Beep!"), HL_ATTR(HLF_W)); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1207 | } |
| 1208 | } |
| 1209 | |
| 1210 | /* |
| 1211 | * To get the "real" home directory: |
| 1212 | * - get value of $HOME |
| 1213 | * For Unix: |
| 1214 | * - go to that directory |
| 1215 | * - do mch_dirname() to get the real name of that directory. |
| 1216 | * This also works with mounts and links. |
| 1217 | * Don't do this for MS-DOS, it will change the "current dir" for a drive. |
Bram Moolenaar | 25a494c | 2018-11-16 19:39:50 +0100 | [diff] [blame] | 1218 | * For Windows: |
| 1219 | * This code is duplicated in init_homedir() in dosinst.c. Keep in sync! |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1220 | */ |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1221 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1222 | init_homedir(void) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1223 | { |
| 1224 | char_u *var; |
| 1225 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1226 | // In case we are called a second time (when 'encoding' changes). |
Bram Moolenaar | d23a823 | 2018-02-10 18:45:26 +0100 | [diff] [blame] | 1227 | VIM_CLEAR(homedir); |
Bram Moolenaar | 05159a0 | 2005-02-26 23:04:13 +0000 | [diff] [blame] | 1228 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1229 | #ifdef VMS |
| 1230 | var = mch_getenv((char_u *)"SYS$LOGIN"); |
| 1231 | #else |
| 1232 | var = mch_getenv((char_u *)"HOME"); |
| 1233 | #endif |
| 1234 | |
Bram Moolenaar | 4f97475 | 2019-02-17 17:44:42 +0100 | [diff] [blame] | 1235 | #ifdef MSWIN |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1236 | /* |
Bram Moolenaar | 48340b6 | 2017-08-29 22:08:53 +0200 | [diff] [blame] | 1237 | * Typically, $HOME is not defined on Windows, unless the user has |
| 1238 | * specifically defined it for Vim's sake. However, on Windows NT |
| 1239 | * platforms, $HOMEDRIVE and $HOMEPATH are automatically defined for |
| 1240 | * each user. Try constructing $HOME from these. |
| 1241 | */ |
Bram Moolenaar | b47a259 | 2017-08-30 13:22:28 +0200 | [diff] [blame] | 1242 | if (var == NULL || *var == NUL) |
Bram Moolenaar | 48340b6 | 2017-08-29 22:08:53 +0200 | [diff] [blame] | 1243 | { |
| 1244 | char_u *homedrive, *homepath; |
| 1245 | |
| 1246 | homedrive = mch_getenv((char_u *)"HOMEDRIVE"); |
| 1247 | homepath = mch_getenv((char_u *)"HOMEPATH"); |
| 1248 | if (homepath == NULL || *homepath == NUL) |
| 1249 | homepath = (char_u *)"\\"; |
| 1250 | if (homedrive != NULL |
| 1251 | && STRLEN(homedrive) + STRLEN(homepath) < MAXPATHL) |
| 1252 | { |
| 1253 | sprintf((char *)NameBuff, "%s%s", homedrive, homepath); |
| 1254 | if (NameBuff[0] != NUL) |
| 1255 | var = NameBuff; |
| 1256 | } |
| 1257 | } |
| 1258 | |
| 1259 | if (var == NULL) |
| 1260 | var = mch_getenv((char_u *)"USERPROFILE"); |
| 1261 | |
| 1262 | /* |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1263 | * Weird but true: $HOME may contain an indirect reference to another |
| 1264 | * variable, esp. "%USERPROFILE%". Happens when $USERPROFILE isn't set |
| 1265 | * when $HOME is being set. |
| 1266 | */ |
| 1267 | if (var != NULL && *var == '%') |
| 1268 | { |
| 1269 | char_u *p; |
| 1270 | char_u *exp; |
| 1271 | |
| 1272 | p = vim_strchr(var + 1, '%'); |
| 1273 | if (p != NULL) |
| 1274 | { |
Bram Moolenaar | ce0842a | 2005-07-18 21:58:11 +0000 | [diff] [blame] | 1275 | vim_strncpy(NameBuff, var + 1, p - (var + 1)); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1276 | exp = mch_getenv(NameBuff); |
| 1277 | if (exp != NULL && *exp != NUL |
| 1278 | && STRLEN(exp) + STRLEN(p) < MAXPATHL) |
| 1279 | { |
Bram Moolenaar | 555b280 | 2005-05-19 21:08:39 +0000 | [diff] [blame] | 1280 | vim_snprintf((char *)NameBuff, MAXPATHL, "%s%s", exp, p + 1); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1281 | var = NameBuff; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1282 | } |
| 1283 | } |
| 1284 | } |
| 1285 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1286 | if (var != NULL && *var == NUL) // empty is same as not set |
Bram Moolenaar | 48340b6 | 2017-08-29 22:08:53 +0200 | [diff] [blame] | 1287 | var = NULL; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1288 | |
Bram Moolenaar | 05159a0 | 2005-02-26 23:04:13 +0000 | [diff] [blame] | 1289 | if (enc_utf8 && var != NULL) |
| 1290 | { |
| 1291 | int len; |
Bram Moolenaar | b453a53 | 2011-04-28 17:48:44 +0200 | [diff] [blame] | 1292 | char_u *pp = NULL; |
Bram Moolenaar | 05159a0 | 2005-02-26 23:04:13 +0000 | [diff] [blame] | 1293 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1294 | // Convert from active codepage to UTF-8. Other conversions are |
| 1295 | // not done, because they would fail for non-ASCII characters. |
Bram Moolenaar | a93fa7e | 2006-04-17 22:14:47 +0000 | [diff] [blame] | 1296 | acp_to_enc(var, (int)STRLEN(var), &pp, &len); |
Bram Moolenaar | 05159a0 | 2005-02-26 23:04:13 +0000 | [diff] [blame] | 1297 | if (pp != NULL) |
| 1298 | { |
| 1299 | homedir = pp; |
| 1300 | return; |
| 1301 | } |
| 1302 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1303 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1304 | /* |
| 1305 | * Default home dir is C:/ |
| 1306 | * Best assumption we can make in such a situation. |
| 1307 | */ |
| 1308 | if (var == NULL) |
Bram Moolenaar | 6aa2cd4 | 2016-02-16 15:06:59 +0100 | [diff] [blame] | 1309 | var = (char_u *)"C:/"; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1310 | #endif |
Bram Moolenaar | 48340b6 | 2017-08-29 22:08:53 +0200 | [diff] [blame] | 1311 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1312 | if (var != NULL) |
| 1313 | { |
| 1314 | #ifdef UNIX |
| 1315 | /* |
| 1316 | * Change to the directory and get the actual path. This resolves |
| 1317 | * links. Don't do it when we can't return. |
| 1318 | */ |
| 1319 | if (mch_dirname(NameBuff, MAXPATHL) == OK |
| 1320 | && mch_chdir((char *)NameBuff) == 0) |
| 1321 | { |
| 1322 | if (!mch_chdir((char *)var) && mch_dirname(IObuff, IOSIZE) == OK) |
| 1323 | var = IObuff; |
| 1324 | if (mch_chdir((char *)NameBuff) != 0) |
Bram Moolenaar | 460ae5d | 2022-01-01 14:19:49 +0000 | [diff] [blame] | 1325 | emsg(_(e_cannot_go_back_to_previous_directory)); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1326 | } |
| 1327 | #endif |
| 1328 | homedir = vim_strsave(var); |
| 1329 | } |
| 1330 | } |
| 1331 | |
Bram Moolenaar | f461c8e | 2005-06-25 23:04:51 +0000 | [diff] [blame] | 1332 | #if defined(EXITFREE) || defined(PROTO) |
| 1333 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1334 | free_homedir(void) |
Bram Moolenaar | f461c8e | 2005-06-25 23:04:51 +0000 | [diff] [blame] | 1335 | { |
| 1336 | vim_free(homedir); |
| 1337 | } |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 1338 | |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 1339 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1340 | free_users(void) |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 1341 | { |
| 1342 | ga_clear_strings(&ga_users); |
| 1343 | } |
Bram Moolenaar | f461c8e | 2005-06-25 23:04:51 +0000 | [diff] [blame] | 1344 | #endif |
| 1345 | |
K.Takata | ce3189d | 2023-02-15 19:13:43 +0000 | [diff] [blame] | 1346 | #if defined(MSWIN) || defined(PROTO) |
| 1347 | /* |
Yegappan Lakshmanan | a23a11b | 2023-02-21 14:27:41 +0000 | [diff] [blame] | 1348 | * Initialize $VIM and $VIMRUNTIME when 'enc' is updated. |
K.Takata | ce3189d | 2023-02-15 19:13:43 +0000 | [diff] [blame] | 1349 | */ |
| 1350 | void |
| 1351 | init_vimdir(void) |
| 1352 | { |
| 1353 | int mustfree; |
| 1354 | char_u *p; |
| 1355 | |
| 1356 | mch_get_exe_name(); |
| 1357 | |
| 1358 | mustfree = FALSE; |
| 1359 | didset_vim = FALSE; |
| 1360 | p = vim_getenv((char_u *)"VIM", &mustfree); |
| 1361 | if (mustfree) |
| 1362 | vim_free(p); |
| 1363 | |
| 1364 | mustfree = FALSE; |
| 1365 | didset_vimruntime = FALSE; |
| 1366 | p = vim_getenv((char_u *)"VIMRUNTIME", &mustfree); |
| 1367 | if (mustfree) |
| 1368 | vim_free(p); |
| 1369 | } |
| 1370 | #endif |
| 1371 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1372 | /* |
Bram Moolenaar | 9f0545d | 2007-09-26 20:36:32 +0000 | [diff] [blame] | 1373 | * Call expand_env() and store the result in an allocated string. |
| 1374 | * This is not very memory efficient, this expects the result to be freed |
| 1375 | * again soon. |
| 1376 | */ |
| 1377 | char_u * |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1378 | expand_env_save(char_u *src) |
Bram Moolenaar | 9f0545d | 2007-09-26 20:36:32 +0000 | [diff] [blame] | 1379 | { |
| 1380 | return expand_env_save_opt(src, FALSE); |
| 1381 | } |
| 1382 | |
| 1383 | /* |
| 1384 | * Idem, but when "one" is TRUE handle the string as one file name, only |
| 1385 | * expand "~" at the start. |
| 1386 | */ |
| 1387 | char_u * |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1388 | expand_env_save_opt(char_u *src, int one) |
Bram Moolenaar | 9f0545d | 2007-09-26 20:36:32 +0000 | [diff] [blame] | 1389 | { |
| 1390 | char_u *p; |
| 1391 | |
| 1392 | p = alloc(MAXPATHL); |
| 1393 | if (p != NULL) |
| 1394 | expand_env_esc(src, p, MAXPATHL, FALSE, one, NULL); |
| 1395 | return p; |
| 1396 | } |
| 1397 | |
| 1398 | /* |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1399 | * Expand environment variable with path name. |
| 1400 | * "~/" is also expanded, using $HOME. For Unix "~user/" is expanded. |
Bram Moolenaar | 9f0545d | 2007-09-26 20:36:32 +0000 | [diff] [blame] | 1401 | * Skips over "\ ", "\~" and "\$" (not for Win32 though). |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1402 | * If anything fails no expansion is done and dst equals src. |
| 1403 | */ |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1404 | size_t |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1405 | expand_env( |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1406 | char_u *src, // input string e.g. "$HOME/vim.hlp" |
| 1407 | char_u *dst, // where to put the result |
| 1408 | int dstlen) // maximum length of the result |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1409 | { |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1410 | return expand_env_esc(src, dst, dstlen, FALSE, FALSE, NULL); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1411 | } |
| 1412 | |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1413 | size_t |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1414 | expand_env_esc( |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1415 | char_u *srcp, // input string e.g. "$HOME/vim.hlp" |
| 1416 | char_u *dst, // where to put the result |
| 1417 | int dstlen, // maximum length of the result |
| 1418 | int esc, // escape spaces in expanded variables |
| 1419 | int one, // "srcp" is one file name |
| 1420 | char_u *startstr) // start again after this (can be NULL) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1421 | { |
Bram Moolenaar | 24bbcfe | 2005-06-28 23:32:02 +0000 | [diff] [blame] | 1422 | char_u *src; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1423 | char_u *tail; |
| 1424 | int c; |
| 1425 | char_u *var; |
| 1426 | int copy_char; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1427 | int mustfree; // var was allocated, need to free it later |
| 1428 | int at_start = TRUE; // at start of a name |
Bram Moolenaar | 24bbcfe | 2005-06-28 23:32:02 +0000 | [diff] [blame] | 1429 | int startstr_len = 0; |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1430 | char_u *dst_start = dst; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1431 | |
Bram Moolenaar | 24bbcfe | 2005-06-28 23:32:02 +0000 | [diff] [blame] | 1432 | if (startstr != NULL) |
Bram Moolenaar | a93fa7e | 2006-04-17 22:14:47 +0000 | [diff] [blame] | 1433 | startstr_len = (int)STRLEN(startstr); |
Bram Moolenaar | 24bbcfe | 2005-06-28 23:32:02 +0000 | [diff] [blame] | 1434 | |
| 1435 | src = skipwhite(srcp); |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1436 | --dstlen; // leave one char space for "\," |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1437 | while (*src && dstlen > 0) |
| 1438 | { |
Bram Moolenaar | be83b73 | 2015-08-25 14:21:19 +0200 | [diff] [blame] | 1439 | #ifdef FEAT_EVAL |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1440 | // Skip over `=expr`. |
Bram Moolenaar | be83b73 | 2015-08-25 14:21:19 +0200 | [diff] [blame] | 1441 | if (src[0] == '`' && src[1] == '=') |
| 1442 | { |
| 1443 | size_t len; |
| 1444 | |
| 1445 | var = src; |
| 1446 | src += 2; |
Bram Moolenaar | 683581e | 2020-10-22 21:22:58 +0200 | [diff] [blame] | 1447 | (void)skip_expr(&src, NULL); |
Bram Moolenaar | be83b73 | 2015-08-25 14:21:19 +0200 | [diff] [blame] | 1448 | if (*src == '`') |
| 1449 | ++src; |
| 1450 | len = src - var; |
| 1451 | if (len > (size_t)dstlen) |
| 1452 | len = dstlen; |
| 1453 | vim_strncpy(dst, var, len); |
| 1454 | dst += len; |
Bram Moolenaar | 5df1ed2 | 2015-09-01 16:25:34 +0200 | [diff] [blame] | 1455 | dstlen -= (int)len; |
Bram Moolenaar | be83b73 | 2015-08-25 14:21:19 +0200 | [diff] [blame] | 1456 | continue; |
| 1457 | } |
| 1458 | #endif |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1459 | copy_char = TRUE; |
Bram Moolenaar | d4755bb | 2004-09-02 19:12:26 +0000 | [diff] [blame] | 1460 | if ((*src == '$' |
| 1461 | #ifdef VMS |
| 1462 | && at_start |
| 1463 | #endif |
| 1464 | ) |
Bram Moolenaar | 48e330a | 2016-02-23 14:53:34 +0100 | [diff] [blame] | 1465 | #if defined(MSWIN) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1466 | || *src == '%' |
| 1467 | #endif |
| 1468 | || (*src == '~' && at_start)) |
| 1469 | { |
| 1470 | mustfree = FALSE; |
| 1471 | |
| 1472 | /* |
| 1473 | * The variable name is copied into dst temporarily, because it may |
| 1474 | * be a string in read-only memory and a NUL needs to be appended. |
| 1475 | */ |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1476 | if (*src != '~') // environment var |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1477 | { |
| 1478 | tail = src + 1; |
| 1479 | var = dst; |
| 1480 | c = dstlen - 1; |
| 1481 | |
| 1482 | #ifdef UNIX |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1483 | // Unix has ${var-name} type environment vars |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1484 | if (*tail == '{' && !vim_isIDc('{')) |
| 1485 | { |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1486 | tail++; // ignore '{' |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1487 | while (c-- > 0 && *tail && *tail != '}') |
| 1488 | *var++ = *tail++; |
| 1489 | } |
| 1490 | else |
| 1491 | #endif |
| 1492 | { |
| 1493 | while (c-- > 0 && *tail != NUL && ((vim_isIDc(*tail)) |
Bram Moolenaar | 48e330a | 2016-02-23 14:53:34 +0100 | [diff] [blame] | 1494 | #if defined(MSWIN) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1495 | || (*src == '%' && *tail != '%') |
| 1496 | #endif |
| 1497 | )) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1498 | *var++ = *tail++; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1499 | } |
| 1500 | |
Bram Moolenaar | 48e330a | 2016-02-23 14:53:34 +0100 | [diff] [blame] | 1501 | #if defined(MSWIN) || defined(UNIX) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1502 | # ifdef UNIX |
| 1503 | if (src[1] == '{' && *tail != '}') |
| 1504 | # else |
| 1505 | if (*src == '%' && *tail != '%') |
| 1506 | # endif |
| 1507 | var = NULL; |
| 1508 | else |
| 1509 | { |
| 1510 | # ifdef UNIX |
| 1511 | if (src[1] == '{') |
| 1512 | # else |
| 1513 | if (*src == '%') |
| 1514 | #endif |
| 1515 | ++tail; |
| 1516 | #endif |
| 1517 | *var = NUL; |
| 1518 | var = vim_getenv(dst, &mustfree); |
Bram Moolenaar | 48e330a | 2016-02-23 14:53:34 +0100 | [diff] [blame] | 1519 | #if defined(MSWIN) || defined(UNIX) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1520 | } |
| 1521 | #endif |
| 1522 | } |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1523 | // home directory |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1524 | else if ( src[1] == NUL |
| 1525 | || vim_ispathsep(src[1]) |
| 1526 | || vim_strchr((char_u *)" ,\t\n", src[1]) != NULL) |
| 1527 | { |
| 1528 | var = homedir; |
| 1529 | tail = src + 1; |
| 1530 | } |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1531 | else // user directory |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1532 | { |
| 1533 | #if defined(UNIX) || (defined(VMS) && defined(USER_HOME)) |
| 1534 | /* |
| 1535 | * Copy ~user to dst[], so we can put a NUL after it. |
| 1536 | */ |
| 1537 | tail = src; |
| 1538 | var = dst; |
| 1539 | c = dstlen - 1; |
| 1540 | while ( c-- > 0 |
| 1541 | && *tail |
| 1542 | && vim_isfilec(*tail) |
| 1543 | && !vim_ispathsep(*tail)) |
| 1544 | *var++ = *tail++; |
| 1545 | *var = NUL; |
| 1546 | # ifdef UNIX |
| 1547 | /* |
| 1548 | * If the system supports getpwnam(), use it. |
| 1549 | * Otherwise, or if getpwnam() fails, the shell is used to |
| 1550 | * expand ~user. This is slower and may fail if the shell |
| 1551 | * does not support ~user (old versions of /bin/sh). |
| 1552 | */ |
| 1553 | # if defined(HAVE_GETPWNAM) && defined(HAVE_PWD_H) |
| 1554 | { |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1555 | // Note: memory allocated by getpwnam() is never freed. |
| 1556 | // Calling endpwent() apparently doesn't help. |
Bram Moolenaar | 187a4f2 | 2017-02-23 17:07:14 +0100 | [diff] [blame] | 1557 | struct passwd *pw = (*dst == NUL) |
| 1558 | ? NULL : getpwnam((char *)dst + 1); |
| 1559 | |
| 1560 | var = (pw == NULL) ? NULL : (char_u *)pw->pw_dir; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1561 | } |
| 1562 | if (var == NULL) |
| 1563 | # endif |
| 1564 | { |
| 1565 | expand_T xpc; |
| 1566 | |
| 1567 | ExpandInit(&xpc); |
| 1568 | xpc.xp_context = EXPAND_FILES; |
| 1569 | var = ExpandOne(&xpc, dst, NULL, |
| 1570 | WILD_ADD_SLASH|WILD_SILENT, WILD_EXPAND_FREE); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1571 | mustfree = TRUE; |
| 1572 | } |
| 1573 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1574 | # else // !UNIX, thus VMS |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1575 | /* |
| 1576 | * USER_HOME is a comma-separated list of |
| 1577 | * directories to search for the user account in. |
| 1578 | */ |
| 1579 | { |
| 1580 | char_u test[MAXPATHL], paths[MAXPATHL]; |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1581 | size_t testlen; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1582 | char_u *path, *next_path, *ptr; |
Bram Moolenaar | 8767f52 | 2016-07-01 17:17:39 +0200 | [diff] [blame] | 1583 | stat_T st; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1584 | |
| 1585 | STRCPY(paths, USER_HOME); |
| 1586 | next_path = paths; |
| 1587 | while (*next_path) |
| 1588 | { |
| 1589 | for (path = next_path; *next_path && *next_path != ','; |
| 1590 | next_path++); |
| 1591 | if (*next_path) |
| 1592 | *next_path++ = NUL; |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1593 | testlen = vim_snprintf_safelen( |
| 1594 | (char *)test, |
| 1595 | sizeof(test), |
| 1596 | "%s/%s", |
| 1597 | path, |
| 1598 | dst + 1); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1599 | if (mch_stat(test, &st) == 0) |
| 1600 | { |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1601 | var = alloc(testlen + 1); |
| 1602 | if (var != NULL) |
| 1603 | { |
| 1604 | STRCPY(var, test); |
| 1605 | mustfree = TRUE; |
| 1606 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1607 | break; |
| 1608 | } |
| 1609 | } |
| 1610 | } |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1611 | # endif // UNIX |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1612 | #else |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1613 | // cannot expand user's home directory, so don't try |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1614 | var = NULL; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1615 | tail = (char_u *)""; // for gcc |
| 1616 | #endif // UNIX || VMS |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1617 | } |
| 1618 | |
| 1619 | #ifdef BACKSLASH_IN_FILENAME |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1620 | // If 'shellslash' is set change backslashes to forward slashes. |
| 1621 | // Can't use slash_adjust(), p_ssl may be set temporarily. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1622 | if (p_ssl && var != NULL && vim_strchr(var, '\\') != NULL) |
| 1623 | { |
| 1624 | char_u *p = vim_strsave(var); |
| 1625 | |
| 1626 | if (p != NULL) |
| 1627 | { |
| 1628 | if (mustfree) |
| 1629 | vim_free(var); |
| 1630 | var = p; |
| 1631 | mustfree = TRUE; |
| 1632 | forward_slash(var); |
| 1633 | } |
| 1634 | } |
| 1635 | #endif |
| 1636 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1637 | // If "var" contains white space, escape it with a backslash. |
| 1638 | // Required for ":e ~/tt" when $HOME includes a space. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1639 | if (esc && var != NULL && vim_strpbrk(var, (char_u *)" \t") != NULL) |
| 1640 | { |
| 1641 | char_u *p = vim_strsave_escaped(var, (char_u *)" \t"); |
| 1642 | |
| 1643 | if (p != NULL) |
| 1644 | { |
| 1645 | if (mustfree) |
| 1646 | vim_free(var); |
| 1647 | var = p; |
| 1648 | mustfree = TRUE; |
| 1649 | } |
| 1650 | } |
| 1651 | |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1652 | if (var != NULL && *var != NUL) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1653 | { |
Bram Moolenaar | a93fa7e | 2006-04-17 22:14:47 +0000 | [diff] [blame] | 1654 | c = (int)STRLEN(var); |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1655 | |
| 1656 | if (c + STRLEN(tail) + 1 < (unsigned)dstlen) |
| 1657 | { |
| 1658 | STRCPY(dst, var); |
| 1659 | dstlen -= c; |
| 1660 | // if var[] ends in a path separator and tail[] starts |
| 1661 | // with it, skip a character |
| 1662 | if (after_pathsep(dst, dst + c) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1663 | #if defined(BACKSLASH_IN_FILENAME) || defined(AMIGA) |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1664 | && dst[c - 1] != ':' |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1665 | #endif |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1666 | && vim_ispathsep(*tail)) |
| 1667 | ++tail; |
| 1668 | dst += c; |
| 1669 | src = tail; |
| 1670 | copy_char = FALSE; |
| 1671 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1672 | } |
| 1673 | if (mustfree) |
| 1674 | vim_free(var); |
| 1675 | } |
| 1676 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1677 | if (copy_char) // copy at least one char |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1678 | { |
| 1679 | /* |
Bram Moolenaar | 2539402 | 2007-05-10 19:06:20 +0000 | [diff] [blame] | 1680 | * Recognize the start of a new name, for '~'. |
Bram Moolenaar | 9f0545d | 2007-09-26 20:36:32 +0000 | [diff] [blame] | 1681 | * Don't do this when "one" is TRUE, to avoid expanding "~" in |
| 1682 | * ":edit foo ~ foo". |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1683 | */ |
| 1684 | at_start = FALSE; |
| 1685 | if (src[0] == '\\' && src[1] != NUL) |
| 1686 | { |
| 1687 | *dst++ = *src++; |
| 1688 | --dstlen; |
| 1689 | } |
Bram Moolenaar | 9f0545d | 2007-09-26 20:36:32 +0000 | [diff] [blame] | 1690 | else if ((src[0] == ' ' || src[0] == ',') && !one) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1691 | at_start = TRUE; |
Bram Moolenaar | 1c86409 | 2017-08-06 18:15:45 +0200 | [diff] [blame] | 1692 | if (dstlen > 0) |
| 1693 | { |
| 1694 | *dst++ = *src++; |
| 1695 | --dstlen; |
Bram Moolenaar | 24bbcfe | 2005-06-28 23:32:02 +0000 | [diff] [blame] | 1696 | |
Bram Moolenaar | 1c86409 | 2017-08-06 18:15:45 +0200 | [diff] [blame] | 1697 | if (startstr != NULL && src - startstr_len >= srcp |
| 1698 | && STRNCMP(src - startstr_len, startstr, |
| 1699 | startstr_len) == 0) |
| 1700 | at_start = TRUE; |
| 1701 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1702 | } |
Bram Moolenaar | 1c86409 | 2017-08-06 18:15:45 +0200 | [diff] [blame] | 1703 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1704 | } |
| 1705 | *dst = NUL; |
John Marriott | fff0132 | 2025-06-18 18:15:31 +0200 | [diff] [blame] | 1706 | |
| 1707 | return (size_t)(dst - dst_start); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1708 | } |
| 1709 | |
| 1710 | /* |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 1711 | * If the string between "p" and "pend" ends in "name/", return "pend" minus |
| 1712 | * the length of "name/". Otherwise return "pend". |
| 1713 | */ |
| 1714 | static char_u * |
| 1715 | remove_tail(char_u *p, char_u *pend, char_u *name) |
| 1716 | { |
| 1717 | int len = (int)STRLEN(name) + 1; |
| 1718 | char_u *newend = pend - len; |
| 1719 | |
| 1720 | if (newend >= p |
| 1721 | && fnamencmp(newend, name, len - 1) == 0 |
| 1722 | && (newend == p || after_pathsep(p, newend))) |
| 1723 | return newend; |
| 1724 | return pend; |
| 1725 | } |
| 1726 | |
| 1727 | /* |
| 1728 | * Check if the directory "vimdir/<version>" or "vimdir/runtime" exists. |
| 1729 | * Return NULL if not, return its name in allocated memory otherwise. |
| 1730 | */ |
| 1731 | static char_u * |
| 1732 | vim_version_dir(char_u *vimdir) |
| 1733 | { |
| 1734 | char_u *p; |
| 1735 | |
| 1736 | if (vimdir == NULL || *vimdir == NUL) |
| 1737 | return NULL; |
| 1738 | p = concat_fnames(vimdir, (char_u *)VIM_VERSION_NODOT, TRUE); |
| 1739 | if (p != NULL && mch_isdir(p)) |
| 1740 | return p; |
| 1741 | vim_free(p); |
| 1742 | p = concat_fnames(vimdir, (char_u *)RUNTIME_DIRNAME, TRUE); |
| 1743 | if (p != NULL && mch_isdir(p)) |
Bram Moolenaar | 022f9ef | 2022-07-02 17:36:31 +0100 | [diff] [blame] | 1744 | { |
| 1745 | char_u *fname = concat_fnames(p, (char_u *)"defaults.vim", TRUE); |
| 1746 | |
| 1747 | // Check that "defaults.vim" exists in this directory, to avoid picking |
| 1748 | // up a stray "runtime" directory, it would make many tests fail in |
| 1749 | // mysterious ways. |
| 1750 | if (fname != NULL) |
| 1751 | { |
| 1752 | int exists = file_is_readable(fname); |
| 1753 | |
| 1754 | vim_free(fname); |
| 1755 | if (exists) |
| 1756 | return p; |
| 1757 | } |
| 1758 | } |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 1759 | vim_free(p); |
| 1760 | return NULL; |
| 1761 | } |
| 1762 | |
| 1763 | /* |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1764 | * Vim's version of getenv(). |
| 1765 | * Special handling of $HOME, $VIM and $VIMRUNTIME. |
Bram Moolenaar | 2f6b0b8 | 2005-03-08 22:43:10 +0000 | [diff] [blame] | 1766 | * Also does ACP to 'enc' conversion for Win32. |
K.Takata | ce3189d | 2023-02-15 19:13:43 +0000 | [diff] [blame] | 1767 | * "mustfree" is set to TRUE when the returned string is allocated. It must be |
Bram Moolenaar | b453a53 | 2011-04-28 17:48:44 +0200 | [diff] [blame] | 1768 | * initialized to FALSE by the caller. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1769 | */ |
| 1770 | char_u * |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 1771 | vim_getenv(char_u *name, int *mustfree) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1772 | { |
Bram Moolenaar | f0908e6 | 2019-03-30 20:11:50 +0100 | [diff] [blame] | 1773 | char_u *p = NULL; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1774 | char_u *pend; |
| 1775 | int vimruntime; |
Bram Moolenaar | f0908e6 | 2019-03-30 20:11:50 +0100 | [diff] [blame] | 1776 | #ifdef MSWIN |
| 1777 | WCHAR *wn, *wp; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1778 | |
Bram Moolenaar | f0908e6 | 2019-03-30 20:11:50 +0100 | [diff] [blame] | 1779 | // use "C:/" when $HOME is not set |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1780 | if (STRCMP(name, "HOME") == 0) |
| 1781 | return homedir; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1782 | |
Bram Moolenaar | f0908e6 | 2019-03-30 20:11:50 +0100 | [diff] [blame] | 1783 | // Use Wide function |
| 1784 | wn = enc_to_utf16(name, NULL); |
| 1785 | if (wn == NULL) |
| 1786 | return NULL; |
| 1787 | |
| 1788 | wp = _wgetenv(wn); |
| 1789 | vim_free(wn); |
| 1790 | |
| 1791 | if (wp != NULL && *wp == NUL) // empty is the same as not set |
| 1792 | wp = NULL; |
| 1793 | |
| 1794 | if (wp != NULL) |
| 1795 | { |
| 1796 | p = utf16_to_enc(wp, NULL); |
| 1797 | if (p == NULL) |
| 1798 | return NULL; |
| 1799 | |
| 1800 | *mustfree = TRUE; |
| 1801 | return p; |
| 1802 | } |
| 1803 | #else |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1804 | p = mch_getenv(name); |
Bram Moolenaar | f0908e6 | 2019-03-30 20:11:50 +0100 | [diff] [blame] | 1805 | if (p != NULL && *p == NUL) // empty is the same as not set |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1806 | p = NULL; |
| 1807 | |
| 1808 | if (p != NULL) |
| 1809 | return p; |
Bram Moolenaar | 80a8d38 | 2020-05-03 22:57:32 +0200 | [diff] [blame] | 1810 | |
| 1811 | # ifdef __HAIKU__ |
| 1812 | // special handling for user settings directory... |
| 1813 | if (STRCMP(name, "BE_USER_SETTINGS") == 0) |
| 1814 | { |
| 1815 | static char userSettingsPath[MAXPATHL]; |
| 1816 | |
| 1817 | if (find_directory(B_USER_SETTINGS_DIRECTORY, 0, false, |
| 1818 | userSettingsPath, MAXPATHL) == B_OK) |
| 1819 | return (char_u *)userSettingsPath; |
| 1820 | else |
| 1821 | return NULL; |
| 1822 | } |
| 1823 | # endif |
Bram Moolenaar | f0908e6 | 2019-03-30 20:11:50 +0100 | [diff] [blame] | 1824 | #endif |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1825 | |
Bram Moolenaar | f0908e6 | 2019-03-30 20:11:50 +0100 | [diff] [blame] | 1826 | // handling $VIMRUNTIME and $VIM is below, bail out if it's another name. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1827 | vimruntime = (STRCMP(name, "VIMRUNTIME") == 0); |
| 1828 | if (!vimruntime && STRCMP(name, "VIM") != 0) |
| 1829 | return NULL; |
| 1830 | |
| 1831 | /* |
| 1832 | * When expanding $VIMRUNTIME fails, try using $VIM/vim<version> or $VIM. |
| 1833 | * Don't do this when default_vimruntime_dir is non-empty. |
| 1834 | */ |
| 1835 | if (vimruntime |
| 1836 | #ifdef HAVE_PATHDEF |
| 1837 | && *default_vimruntime_dir == NUL |
| 1838 | #endif |
| 1839 | ) |
| 1840 | { |
Bram Moolenaar | f0908e6 | 2019-03-30 20:11:50 +0100 | [diff] [blame] | 1841 | #ifdef MSWIN |
| 1842 | // Use Wide function |
| 1843 | wp = _wgetenv(L"VIM"); |
| 1844 | if (wp != NULL && *wp == NUL) // empty is the same as not set |
| 1845 | wp = NULL; |
| 1846 | if (wp != NULL) |
| 1847 | { |
| 1848 | char_u *q = utf16_to_enc(wp, NULL); |
| 1849 | if (q != NULL) |
| 1850 | { |
| 1851 | p = vim_version_dir(q); |
| 1852 | *mustfree = TRUE; |
| 1853 | if (p == NULL) |
| 1854 | p = q; |
| 1855 | } |
| 1856 | } |
| 1857 | #else |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1858 | p = mch_getenv((char_u *)"VIM"); |
Bram Moolenaar | f0908e6 | 2019-03-30 20:11:50 +0100 | [diff] [blame] | 1859 | if (p != NULL && *p == NUL) // empty is the same as not set |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1860 | p = NULL; |
| 1861 | if (p != NULL) |
| 1862 | { |
| 1863 | p = vim_version_dir(p); |
| 1864 | if (p != NULL) |
| 1865 | *mustfree = TRUE; |
| 1866 | else |
| 1867 | p = mch_getenv((char_u *)"VIM"); |
| 1868 | } |
Bram Moolenaar | f0908e6 | 2019-03-30 20:11:50 +0100 | [diff] [blame] | 1869 | #endif |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1870 | } |
| 1871 | |
| 1872 | /* |
| 1873 | * When expanding $VIM or $VIMRUNTIME fails, try using: |
| 1874 | * - the directory name from 'helpfile' (unless it contains '$') |
| 1875 | * - the executable name from argv[0] |
| 1876 | */ |
| 1877 | if (p == NULL) |
| 1878 | { |
| 1879 | if (p_hf != NULL && vim_strchr(p_hf, '$') == NULL) |
| 1880 | p = p_hf; |
| 1881 | #ifdef USE_EXE_NAME |
| 1882 | /* |
| 1883 | * Use the name of the executable, obtained from argv[0]. |
| 1884 | */ |
| 1885 | else |
| 1886 | p = exe_name; |
| 1887 | #endif |
| 1888 | if (p != NULL) |
| 1889 | { |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1890 | // remove the file name |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1891 | pend = gettail(p); |
| 1892 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1893 | // remove "doc/" from 'helpfile', if present |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1894 | if (p == p_hf) |
| 1895 | pend = remove_tail(p, pend, (char_u *)"doc"); |
| 1896 | |
| 1897 | #ifdef USE_EXE_NAME |
| 1898 | # ifdef MACOS_X |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1899 | // remove "MacOS" from exe_name and add "Resources/vim" |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1900 | if (p == exe_name) |
| 1901 | { |
| 1902 | char_u *pend1; |
Bram Moolenaar | 95e9b49 | 2006-03-15 23:04:43 +0000 | [diff] [blame] | 1903 | char_u *pnew; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1904 | |
Bram Moolenaar | 95e9b49 | 2006-03-15 23:04:43 +0000 | [diff] [blame] | 1905 | pend1 = remove_tail(p, pend, (char_u *)"MacOS"); |
| 1906 | if (pend1 != pend) |
| 1907 | { |
Bram Moolenaar | 964b374 | 2019-05-24 18:54:09 +0200 | [diff] [blame] | 1908 | pnew = alloc(pend1 - p + 15); |
Bram Moolenaar | 95e9b49 | 2006-03-15 23:04:43 +0000 | [diff] [blame] | 1909 | if (pnew != NULL) |
| 1910 | { |
| 1911 | STRNCPY(pnew, p, (pend1 - p)); |
| 1912 | STRCPY(pnew + (pend1 - p), "Resources/vim"); |
| 1913 | p = pnew; |
| 1914 | pend = p + STRLEN(p); |
| 1915 | } |
| 1916 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1917 | } |
| 1918 | # endif |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1919 | // remove "src/" from exe_name, if present |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1920 | if (p == exe_name) |
| 1921 | pend = remove_tail(p, pend, (char_u *)"src"); |
| 1922 | #endif |
| 1923 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1924 | // for $VIM, remove "runtime/" or "vim54/", if present |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1925 | if (!vimruntime) |
| 1926 | { |
| 1927 | pend = remove_tail(p, pend, (char_u *)RUNTIME_DIRNAME); |
| 1928 | pend = remove_tail(p, pend, (char_u *)VIM_VERSION_NODOT); |
| 1929 | } |
| 1930 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1931 | // remove trailing path separator |
Bram Moolenaar | 1cd871b | 2004-12-19 22:46:22 +0000 | [diff] [blame] | 1932 | if (pend > p && after_pathsep(p, pend)) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1933 | --pend; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1934 | |
Bram Moolenaar | 95e9b49 | 2006-03-15 23:04:43 +0000 | [diff] [blame] | 1935 | #ifdef MACOS_X |
| 1936 | if (p == exe_name || p == p_hf) |
| 1937 | #endif |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1938 | // check that the result is a directory name |
Bram Moolenaar | 71ccd03 | 2020-06-12 22:59:11 +0200 | [diff] [blame] | 1939 | p = vim_strnsave(p, pend - p); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1940 | |
| 1941 | if (p != NULL && !mch_isdir(p)) |
Bram Moolenaar | d23a823 | 2018-02-10 18:45:26 +0100 | [diff] [blame] | 1942 | VIM_CLEAR(p); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1943 | else |
| 1944 | { |
| 1945 | #ifdef USE_EXE_NAME |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1946 | // may add "/vim54" or "/runtime" if it exists |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1947 | if (vimruntime && (pend = vim_version_dir(p)) != NULL) |
| 1948 | { |
| 1949 | vim_free(p); |
| 1950 | p = pend; |
| 1951 | } |
| 1952 | #endif |
| 1953 | *mustfree = TRUE; |
| 1954 | } |
| 1955 | } |
| 1956 | } |
| 1957 | |
| 1958 | #ifdef HAVE_PATHDEF |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1959 | // When there is a pathdef.c file we can use default_vim_dir and |
| 1960 | // default_vimruntime_dir |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1961 | if (p == NULL) |
| 1962 | { |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 1963 | // Only use default_vimruntime_dir when it is not empty |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1964 | if (vimruntime && *default_vimruntime_dir != NUL) |
| 1965 | { |
| 1966 | p = default_vimruntime_dir; |
| 1967 | *mustfree = FALSE; |
| 1968 | } |
| 1969 | else if (*default_vim_dir != NUL) |
| 1970 | { |
| 1971 | if (vimruntime && (p = vim_version_dir(default_vim_dir)) != NULL) |
| 1972 | *mustfree = TRUE; |
| 1973 | else |
| 1974 | { |
| 1975 | p = default_vim_dir; |
| 1976 | *mustfree = FALSE; |
| 1977 | } |
| 1978 | } |
| 1979 | } |
| 1980 | #endif |
| 1981 | |
| 1982 | /* |
| 1983 | * Set the environment variable, so that the new value can be found fast |
| 1984 | * next time, and others can also use it (e.g. Perl). |
| 1985 | */ |
| 1986 | if (p != NULL) |
| 1987 | { |
| 1988 | if (vimruntime) |
| 1989 | { |
| 1990 | vim_setenv((char_u *)"VIMRUNTIME", p); |
| 1991 | didset_vimruntime = TRUE; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 1992 | } |
| 1993 | else |
| 1994 | { |
| 1995 | vim_setenv((char_u *)"VIM", p); |
| 1996 | didset_vim = TRUE; |
| 1997 | } |
| 1998 | } |
| 1999 | return p; |
| 2000 | } |
| 2001 | |
Bram Moolenaar | 137374f | 2018-05-13 15:59:50 +0200 | [diff] [blame] | 2002 | void |
| 2003 | vim_unsetenv(char_u *var) |
| 2004 | { |
| 2005 | #ifdef HAVE_UNSETENV |
| 2006 | unsetenv((char *)var); |
| 2007 | #else |
Bram Moolenaar | 1af6a4b | 2018-05-14 22:58:34 +0200 | [diff] [blame] | 2008 | vim_setenv(var, (char_u *)""); |
Bram Moolenaar | 137374f | 2018-05-13 15:59:50 +0200 | [diff] [blame] | 2009 | #endif |
| 2010 | } |
| 2011 | |
LemonBoy | 7714231 | 2022-04-15 20:50:46 +0100 | [diff] [blame] | 2012 | /* |
| 2013 | * Removes environment variable "name" and take care of side effects. |
| 2014 | */ |
| 2015 | void |
| 2016 | vim_unsetenv_ext(char_u *var) |
| 2017 | { |
| 2018 | vim_unsetenv(var); |
| 2019 | |
| 2020 | // "homedir" is not cleared, keep using the old value until $HOME is set. |
| 2021 | if (STRICMP(var, "VIM") == 0) |
| 2022 | didset_vim = FALSE; |
| 2023 | else if (STRICMP(var, "VIMRUNTIME") == 0) |
| 2024 | didset_vimruntime = FALSE; |
| 2025 | } |
Bram Moolenaar | 137374f | 2018-05-13 15:59:50 +0200 | [diff] [blame] | 2026 | |
Bram Moolenaar | 092e09c | 2022-04-15 23:29:23 +0100 | [diff] [blame] | 2027 | #if defined(FEAT_EVAL) || defined(PROTO) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2028 | /* |
Bram Moolenaar | b283a8a | 2020-02-02 22:24:04 +0100 | [diff] [blame] | 2029 | * Set environment variable "name" and take care of side effects. |
| 2030 | */ |
| 2031 | void |
| 2032 | vim_setenv_ext(char_u *name, char_u *val) |
| 2033 | { |
| 2034 | vim_setenv(name, val); |
| 2035 | if (STRICMP(name, "HOME") == 0) |
| 2036 | init_homedir(); |
| 2037 | else if (didset_vim && STRICMP(name, "VIM") == 0) |
| 2038 | didset_vim = FALSE; |
LemonBoy | 7714231 | 2022-04-15 20:50:46 +0100 | [diff] [blame] | 2039 | else if (didset_vimruntime && STRICMP(name, "VIMRUNTIME") == 0) |
Bram Moolenaar | b283a8a | 2020-02-02 22:24:04 +0100 | [diff] [blame] | 2040 | didset_vimruntime = FALSE; |
| 2041 | } |
Dominique Pelle | 748b308 | 2022-01-08 12:41:16 +0000 | [diff] [blame] | 2042 | #endif |
Bram Moolenaar | b283a8a | 2020-02-02 22:24:04 +0100 | [diff] [blame] | 2043 | |
| 2044 | /* |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2045 | * Our portable version of setenv. |
| 2046 | */ |
| 2047 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2048 | vim_setenv(char_u *name, char_u *val) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2049 | { |
| 2050 | #ifdef HAVE_SETENV |
| 2051 | mch_setenv((char *)name, (char *)val, 1); |
| 2052 | #else |
| 2053 | char_u *envbuf; |
| 2054 | |
| 2055 | /* |
| 2056 | * Putenv does not copy the string, it has to remain |
| 2057 | * valid. The allocated memory will never be freed. |
| 2058 | */ |
Bram Moolenaar | 964b374 | 2019-05-24 18:54:09 +0200 | [diff] [blame] | 2059 | envbuf = alloc(STRLEN(name) + STRLEN(val) + 2); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2060 | if (envbuf != NULL) |
| 2061 | { |
| 2062 | sprintf((char *)envbuf, "%s=%s", name, val); |
| 2063 | putenv((char *)envbuf); |
| 2064 | } |
| 2065 | #endif |
Bram Moolenaar | 011a34d | 2012-02-29 13:49:09 +0100 | [diff] [blame] | 2066 | #ifdef FEAT_GETTEXT |
| 2067 | /* |
| 2068 | * When setting $VIMRUNTIME adjust the directory to find message |
| 2069 | * translations to $VIMRUNTIME/lang. |
| 2070 | */ |
| 2071 | if (*val != NUL && STRICMP(name, "VIMRUNTIME") == 0) |
| 2072 | { |
| 2073 | char_u *buf = concat_str(val, (char_u *)"/lang"); |
| 2074 | |
| 2075 | if (buf != NULL) |
| 2076 | { |
| 2077 | bindtextdomain(VIMPACKAGE, (char *)buf); |
| 2078 | vim_free(buf); |
| 2079 | } |
| 2080 | } |
| 2081 | #endif |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2082 | } |
| 2083 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2084 | /* |
| 2085 | * Function given to ExpandGeneric() to obtain an environment variable name. |
| 2086 | */ |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2087 | char_u * |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2088 | get_env_name( |
| 2089 | expand_T *xp UNUSED, |
| 2090 | int idx) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2091 | { |
Bram Moolenaar | 5ff595d | 2022-08-26 22:36:41 +0100 | [diff] [blame] | 2092 | #if defined(AMIGA) |
| 2093 | // No environ[] on the Amiga. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2094 | return NULL; |
Bram Moolenaar | 5ff595d | 2022-08-26 22:36:41 +0100 | [diff] [blame] | 2095 | #else |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2096 | # ifndef __WIN32__ |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2097 | // Borland C++ 5.2 has this in a header file. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2098 | extern char **environ; |
| 2099 | # endif |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2100 | char_u *str; |
| 2101 | int n; |
| 2102 | |
| 2103 | str = (char_u *)environ[idx]; |
| 2104 | if (str == NULL) |
| 2105 | return NULL; |
| 2106 | |
Bram Moolenaar | 5ff595d | 2022-08-26 22:36:41 +0100 | [diff] [blame] | 2107 | for (n = 0; n < EXPAND_BUF_LEN - 1; ++n) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2108 | { |
| 2109 | if (str[n] == '=' || str[n] == NUL) |
| 2110 | break; |
Bram Moolenaar | 5ff595d | 2022-08-26 22:36:41 +0100 | [diff] [blame] | 2111 | xp->xp_buf[n] = str[n]; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2112 | } |
Bram Moolenaar | 5ff595d | 2022-08-26 22:36:41 +0100 | [diff] [blame] | 2113 | xp->xp_buf[n] = NUL; |
| 2114 | return xp->xp_buf; |
| 2115 | #endif |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2116 | } |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2117 | |
| 2118 | /* |
Bram Moolenaar | 6b0b83f | 2018-09-10 19:03:05 +0200 | [diff] [blame] | 2119 | * Add a user name to the list of users in ga_users. |
| 2120 | * Do nothing if user name is NULL or empty. |
| 2121 | */ |
| 2122 | static void |
| 2123 | add_user(char_u *user, int need_copy) |
| 2124 | { |
| 2125 | char_u *user_copy = (user != NULL && need_copy) |
| 2126 | ? vim_strsave(user) : user; |
| 2127 | |
| 2128 | if (user_copy == NULL || *user_copy == NUL || ga_grow(&ga_users, 1) == FAIL) |
| 2129 | { |
| 2130 | if (need_copy) |
Christian Brabandt | 7a2f217 | 2024-03-24 09:50:03 +0100 | [diff] [blame] | 2131 | vim_free(user_copy); |
Bram Moolenaar | 6b0b83f | 2018-09-10 19:03:05 +0200 | [diff] [blame] | 2132 | return; |
| 2133 | } |
| 2134 | ((char_u **)(ga_users.ga_data))[ga_users.ga_len++] = user_copy; |
| 2135 | } |
| 2136 | |
| 2137 | /* |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2138 | * Find all user names for user completion. |
| 2139 | * Done only once and then cached. |
| 2140 | */ |
| 2141 | static void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2142 | init_users(void) |
Bram Moolenaar | 01b626c | 2013-06-16 22:49:14 +0200 | [diff] [blame] | 2143 | { |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2144 | static int lazy_init_done = FALSE; |
| 2145 | |
| 2146 | if (lazy_init_done) |
| 2147 | return; |
| 2148 | |
| 2149 | lazy_init_done = TRUE; |
| 2150 | ga_init2(&ga_users, sizeof(char_u *), 20); |
| 2151 | |
| 2152 | # if defined(HAVE_GETPWENT) && defined(HAVE_PWD_H) |
| 2153 | { |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2154 | struct passwd* pw; |
| 2155 | |
| 2156 | setpwent(); |
| 2157 | while ((pw = getpwent()) != NULL) |
Bram Moolenaar | 6b0b83f | 2018-09-10 19:03:05 +0200 | [diff] [blame] | 2158 | add_user((char_u *)pw->pw_name, TRUE); |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2159 | endpwent(); |
| 2160 | } |
Bram Moolenaar | 4f97475 | 2019-02-17 17:44:42 +0100 | [diff] [blame] | 2161 | # elif defined(MSWIN) |
Bram Moolenaar | 828c3d7 | 2018-06-19 18:58:07 +0200 | [diff] [blame] | 2162 | { |
Bram Moolenaar | 828c3d7 | 2018-06-19 18:58:07 +0200 | [diff] [blame] | 2163 | DWORD nusers = 0, ntotal = 0, i; |
| 2164 | PUSER_INFO_0 uinfo; |
| 2165 | |
| 2166 | if (NetUserEnum(NULL, 0, 0, (LPBYTE *) &uinfo, MAX_PREFERRED_LENGTH, |
| 2167 | &nusers, &ntotal, NULL) == NERR_Success) |
| 2168 | { |
| 2169 | for (i = 0; i < nusers; i++) |
Bram Moolenaar | 6b0b83f | 2018-09-10 19:03:05 +0200 | [diff] [blame] | 2170 | add_user(utf16_to_enc(uinfo[i].usri0_name, NULL), FALSE); |
Bram Moolenaar | 828c3d7 | 2018-06-19 18:58:07 +0200 | [diff] [blame] | 2171 | |
| 2172 | NetApiBufferFree(uinfo); |
| 2173 | } |
| 2174 | } |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2175 | # endif |
Bram Moolenaar | 6b0b83f | 2018-09-10 19:03:05 +0200 | [diff] [blame] | 2176 | # if defined(HAVE_GETPWNAM) |
| 2177 | { |
| 2178 | char_u *user_env = mch_getenv((char_u *)"USER"); |
| 2179 | |
| 2180 | // The $USER environment variable may be a valid remote user name (NIS, |
| 2181 | // LDAP) not already listed by getpwent(), as getpwent() only lists |
| 2182 | // local user names. If $USER is not already listed, check whether it |
| 2183 | // is a valid remote user name using getpwnam() and if it is, add it to |
| 2184 | // the list of user names. |
| 2185 | |
| 2186 | if (user_env != NULL && *user_env != NUL) |
| 2187 | { |
| 2188 | int i; |
| 2189 | |
| 2190 | for (i = 0; i < ga_users.ga_len; i++) |
| 2191 | { |
| 2192 | char_u *local_user = ((char_u **)ga_users.ga_data)[i]; |
| 2193 | |
| 2194 | if (STRCMP(local_user, user_env) == 0) |
| 2195 | break; |
| 2196 | } |
| 2197 | |
| 2198 | if (i == ga_users.ga_len) |
| 2199 | { |
| 2200 | struct passwd *pw = getpwnam((char *)user_env); |
| 2201 | |
| 2202 | if (pw != NULL) |
| 2203 | add_user((char_u *)pw->pw_name, TRUE); |
| 2204 | } |
| 2205 | } |
| 2206 | } |
| 2207 | # endif |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2208 | } |
| 2209 | |
| 2210 | /* |
zeertzjq | c029c13 | 2024-03-28 11:37:26 +0100 | [diff] [blame] | 2211 | * Function given to ExpandGeneric() to obtain user names. |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2212 | */ |
| 2213 | char_u* |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2214 | get_users(expand_T *xp UNUSED, int idx) |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2215 | { |
| 2216 | init_users(); |
| 2217 | if (idx < ga_users.ga_len) |
| 2218 | return ((char_u **)ga_users.ga_data)[idx]; |
| 2219 | return NULL; |
| 2220 | } |
| 2221 | |
| 2222 | /* |
| 2223 | * Check whether name matches a user name. Return: |
| 2224 | * 0 if name does not match any user name. |
| 2225 | * 1 if name partially matches the beginning of a user name. |
| 2226 | * 2 is name fully matches a user name. |
| 2227 | */ |
Bram Moolenaar | 6c5d104 | 2018-07-07 16:41:13 +0200 | [diff] [blame] | 2228 | int |
| 2229 | match_user(char_u *name) |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2230 | { |
| 2231 | int i; |
| 2232 | int n = (int)STRLEN(name); |
| 2233 | int result = 0; |
| 2234 | |
| 2235 | init_users(); |
| 2236 | for (i = 0; i < ga_users.ga_len; i++) |
| 2237 | { |
| 2238 | if (STRCMP(((char_u **)ga_users.ga_data)[i], name) == 0) |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2239 | return 2; // full match |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2240 | if (STRNCMP(((char_u **)ga_users.ga_data)[i], name, n) == 0) |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2241 | result = 1; // partial match |
Bram Moolenaar | 2430586 | 2012-08-15 14:05:05 +0200 | [diff] [blame] | 2242 | } |
| 2243 | return result; |
| 2244 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2245 | |
Bram Moolenaar | 5843f5f | 2019-08-20 20:13:45 +0200 | [diff] [blame] | 2246 | static void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2247 | prepare_to_exit(void) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2248 | { |
Bram Moolenaar | 1cd871b | 2004-12-19 22:46:22 +0000 | [diff] [blame] | 2249 | #if defined(SIGHUP) && defined(SIG_IGN) |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2250 | // Ignore SIGHUP, because a dropped connection causes a read error, which |
| 2251 | // makes Vim exit and then handling SIGHUP causes various reentrance |
| 2252 | // problems. |
ichizok | 378447f | 2023-05-11 22:25:42 +0100 | [diff] [blame] | 2253 | mch_signal(SIGHUP, SIG_IGN); |
Bram Moolenaar | 293ee4d | 2004-12-09 21:34:53 +0000 | [diff] [blame] | 2254 | #endif |
| 2255 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2256 | #ifdef FEAT_GUI |
| 2257 | if (gui.in_use) |
| 2258 | { |
| 2259 | gui.dying = TRUE; |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2260 | out_trash(); // trash any pending output |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2261 | } |
| 2262 | else |
| 2263 | #endif |
| 2264 | { |
| 2265 | windgoto((int)Rows - 1, 0); |
| 2266 | |
| 2267 | /* |
| 2268 | * Switch terminal mode back now, so messages end up on the "normal" |
| 2269 | * screen (if there are two screens). |
| 2270 | */ |
| 2271 | settmode(TMODE_COOK); |
Bram Moolenaar | cea912a | 2016-10-12 14:20:24 +0200 | [diff] [blame] | 2272 | stoptermcap(); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2273 | out_flush(); |
| 2274 | } |
| 2275 | } |
| 2276 | |
| 2277 | /* |
| 2278 | * Preserve files and exit. |
| 2279 | * When called IObuff must contain a message. |
Bram Moolenaar | bec9c20 | 2013-09-05 21:41:39 +0200 | [diff] [blame] | 2280 | * NOTE: This may be called from deathtrap() in a signal handler, avoid unsafe |
| 2281 | * functions, such as allocating memory. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2282 | */ |
| 2283 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2284 | preserve_exit(void) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2285 | { |
| 2286 | buf_T *buf; |
| 2287 | |
| 2288 | prepare_to_exit(); |
| 2289 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2290 | // Setting this will prevent free() calls. That avoids calling free() |
| 2291 | // recursively when free() was invoked with a bad pointer. |
Bram Moolenaar | 4770d09 | 2006-01-12 23:22:24 +0000 | [diff] [blame] | 2292 | really_exiting = TRUE; |
| 2293 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2294 | out_str(IObuff); |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2295 | screen_start(); // don't know where cursor is now |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2296 | out_flush(); |
| 2297 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2298 | ml_close_notmod(); // close all not-modified buffers |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2299 | |
Bram Moolenaar | 2932359 | 2016-07-24 22:04:11 +0200 | [diff] [blame] | 2300 | FOR_ALL_BUFFERS(buf) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2301 | { |
| 2302 | if (buf->b_ml.ml_mfp != NULL && buf->b_ml.ml_mfp->mf_fname != NULL) |
| 2303 | { |
Bram Moolenaar | 69212b1 | 2020-05-10 14:14:03 +0200 | [diff] [blame] | 2304 | OUT_STR("Vim: preserving files...\r\n"); |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2305 | screen_start(); // don't know where cursor is now |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2306 | out_flush(); |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2307 | ml_sync_all(FALSE, FALSE); // preserve all swap files |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2308 | break; |
| 2309 | } |
| 2310 | } |
| 2311 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2312 | ml_close_all(FALSE); // close all memfiles, without deleting |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2313 | |
Bram Moolenaar | 69212b1 | 2020-05-10 14:14:03 +0200 | [diff] [blame] | 2314 | OUT_STR("Vim: Finished.\r\n"); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2315 | |
| 2316 | getout(1); |
| 2317 | } |
| 2318 | |
| 2319 | /* |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2320 | * Check for CTRL-C pressed, but only once in a while. |
| 2321 | * Should be used instead of ui_breakcheck() for functions that check for |
| 2322 | * each line in the file. Calling ui_breakcheck() each time takes too much |
| 2323 | * time, because it can be a system call. |
| 2324 | */ |
| 2325 | |
| 2326 | #ifndef BREAKCHECK_SKIP |
Bram Moolenaar | b4fe0eb | 2019-07-21 14:50:21 +0200 | [diff] [blame] | 2327 | # define BREAKCHECK_SKIP 1000 |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2328 | #endif |
| 2329 | |
| 2330 | static int breakcheck_count = 0; |
| 2331 | |
| 2332 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2333 | line_breakcheck(void) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2334 | { |
| 2335 | if (++breakcheck_count >= BREAKCHECK_SKIP) |
| 2336 | { |
| 2337 | breakcheck_count = 0; |
| 2338 | ui_breakcheck(); |
| 2339 | } |
| 2340 | } |
| 2341 | |
| 2342 | /* |
| 2343 | * Like line_breakcheck() but check 10 times less often. |
| 2344 | */ |
| 2345 | void |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2346 | fast_breakcheck(void) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2347 | { |
| 2348 | if (++breakcheck_count >= BREAKCHECK_SKIP * 10) |
| 2349 | { |
| 2350 | breakcheck_count = 0; |
| 2351 | ui_breakcheck(); |
| 2352 | } |
| 2353 | } |
| 2354 | |
Dominique Pelle | 748b308 | 2022-01-08 12:41:16 +0000 | [diff] [blame] | 2355 | # if defined(FEAT_SPELL) || defined(PROTO) |
Bram Moolenaar | f1ec378 | 2020-03-20 19:37:47 +0100 | [diff] [blame] | 2356 | /* |
| 2357 | * Like line_breakcheck() but check 100 times less often. |
| 2358 | */ |
| 2359 | void |
| 2360 | veryfast_breakcheck(void) |
| 2361 | { |
| 2362 | if (++breakcheck_count >= BREAKCHECK_SKIP * 100) |
| 2363 | { |
| 2364 | breakcheck_count = 0; |
| 2365 | ui_breakcheck(); |
| 2366 | } |
| 2367 | } |
Dominique Pelle | 748b308 | 2022-01-08 12:41:16 +0000 | [diff] [blame] | 2368 | #endif |
Bram Moolenaar | f1ec378 | 2020-03-20 19:37:47 +0100 | [diff] [blame] | 2369 | |
Bram Moolenaar | 0a52df5 | 2019-08-18 22:26:31 +0200 | [diff] [blame] | 2370 | #if defined(VIM_BACKTICK) || defined(FEAT_EVAL) \ |
| 2371 | || (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \ |
| 2372 | || defined(PROTO) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2373 | |
| 2374 | #ifndef SEEK_SET |
| 2375 | # define SEEK_SET 0 |
| 2376 | #endif |
| 2377 | #ifndef SEEK_END |
| 2378 | # define SEEK_END 2 |
| 2379 | #endif |
| 2380 | |
| 2381 | /* |
| 2382 | * Get the stdout of an external command. |
Bram Moolenaar | 39c29ed | 2014-04-05 19:44:40 +0200 | [diff] [blame] | 2383 | * If "ret_len" is NULL replace NUL characters with NL. When "ret_len" is not |
| 2384 | * NULL store the length there. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2385 | * Returns an allocated string, or NULL for error. |
| 2386 | */ |
| 2387 | char_u * |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2388 | get_cmd_output( |
| 2389 | char_u *cmd, |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2390 | char_u *infile, // optional input file name |
| 2391 | int flags, // can be SHELL_SILENT |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2392 | int *ret_len) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2393 | { |
| 2394 | char_u *tempname; |
| 2395 | char_u *command; |
| 2396 | char_u *buffer = NULL; |
| 2397 | int len; |
| 2398 | int i = 0; |
| 2399 | FILE *fd; |
| 2400 | |
| 2401 | if (check_restricted() || check_secure()) |
| 2402 | return NULL; |
| 2403 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2404 | // get a name for the temp file |
Bram Moolenaar | e5c421c | 2015-03-31 13:33:08 +0200 | [diff] [blame] | 2405 | if ((tempname = vim_tempname('o', FALSE)) == NULL) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2406 | { |
Bram Moolenaar | 460ae5d | 2022-01-01 14:19:49 +0000 | [diff] [blame] | 2407 | emsg(_(e_cant_get_temp_file_name)); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2408 | return NULL; |
| 2409 | } |
| 2410 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2411 | // Add the redirection stuff |
Bram Moolenaar | c0197e2 | 2004-09-13 20:26:32 +0000 | [diff] [blame] | 2412 | command = make_filter_cmd(cmd, infile, tempname); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2413 | if (command == NULL) |
| 2414 | goto done; |
| 2415 | |
| 2416 | /* |
| 2417 | * Call the shell to execute the command (errors are ignored). |
| 2418 | * Don't check timestamps here. |
| 2419 | */ |
| 2420 | ++no_check_timestamps; |
| 2421 | call_shell(command, SHELL_DOOUT | SHELL_EXPAND | flags); |
| 2422 | --no_check_timestamps; |
| 2423 | |
| 2424 | vim_free(command); |
| 2425 | |
| 2426 | /* |
| 2427 | * read the names from the file into memory |
| 2428 | */ |
| 2429 | # ifdef VMS |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2430 | // created temporary file is not always readable as binary |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2431 | fd = mch_fopen((char *)tempname, "r"); |
| 2432 | # else |
| 2433 | fd = mch_fopen((char *)tempname, READBIN); |
| 2434 | # endif |
| 2435 | |
Bram Moolenaar | 3df8f6e | 2022-04-17 14:01:51 +0100 | [diff] [blame] | 2436 | // Not being able to seek means we can't read the file. |
| 2437 | if (fd == NULL |
| 2438 | || fseek(fd, 0L, SEEK_END) == -1 |
| 2439 | || (len = ftell(fd)) == -1 // get size of temp file |
| 2440 | || fseek(fd, 0L, SEEK_SET) == -1) // back to the start |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2441 | { |
Bram Moolenaar | a9549c9 | 2022-04-17 14:18:11 +0100 | [diff] [blame] | 2442 | semsg(_(e_cannot_read_from_str_2), tempname); |
Bram Moolenaar | 3df8f6e | 2022-04-17 14:01:51 +0100 | [diff] [blame] | 2443 | if (fd != NULL) |
| 2444 | fclose(fd); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2445 | goto done; |
| 2446 | } |
| 2447 | |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2448 | buffer = alloc(len + 1); |
| 2449 | if (buffer != NULL) |
| 2450 | i = (int)fread((char *)buffer, (size_t)1, (size_t)len, fd); |
| 2451 | fclose(fd); |
| 2452 | mch_remove(tempname); |
| 2453 | if (buffer == NULL) |
| 2454 | goto done; |
| 2455 | #ifdef VMS |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2456 | len = i; // VMS doesn't give us what we asked for... |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2457 | #endif |
| 2458 | if (i != len) |
| 2459 | { |
Bram Moolenaar | 460ae5d | 2022-01-01 14:19:49 +0000 | [diff] [blame] | 2460 | semsg(_(e_cant_read_file_str), tempname); |
Bram Moolenaar | d23a823 | 2018-02-10 18:45:26 +0100 | [diff] [blame] | 2461 | VIM_CLEAR(buffer); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2462 | } |
Bram Moolenaar | 39c29ed | 2014-04-05 19:44:40 +0200 | [diff] [blame] | 2463 | else if (ret_len == NULL) |
Bram Moolenaar | fb332a2 | 2013-08-03 14:10:50 +0200 | [diff] [blame] | 2464 | { |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2465 | // Change NUL into SOH, otherwise the string is truncated. |
Bram Moolenaar | fb332a2 | 2013-08-03 14:10:50 +0200 | [diff] [blame] | 2466 | for (i = 0; i < len; ++i) |
Bram Moolenaar | f40f4ab | 2013-08-03 17:31:28 +0200 | [diff] [blame] | 2467 | if (buffer[i] == NUL) |
| 2468 | buffer[i] = 1; |
Bram Moolenaar | fb332a2 | 2013-08-03 14:10:50 +0200 | [diff] [blame] | 2469 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2470 | buffer[len] = NUL; // make sure the buffer is terminated |
Bram Moolenaar | fb332a2 | 2013-08-03 14:10:50 +0200 | [diff] [blame] | 2471 | } |
Bram Moolenaar | 39c29ed | 2014-04-05 19:44:40 +0200 | [diff] [blame] | 2472 | else |
| 2473 | *ret_len = len; |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2474 | |
| 2475 | done: |
| 2476 | vim_free(tempname); |
| 2477 | return buffer; |
| 2478 | } |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2479 | |
| 2480 | # if defined(FEAT_EVAL) || defined(PROTO) |
| 2481 | |
Bram Moolenaar | 840d16f | 2019-09-10 21:27:18 +0200 | [diff] [blame] | 2482 | static void |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2483 | get_cmd_output_as_rettv( |
| 2484 | typval_T *argvars, |
| 2485 | typval_T *rettv, |
| 2486 | int retlist) |
| 2487 | { |
| 2488 | char_u *res = NULL; |
| 2489 | char_u *p; |
| 2490 | char_u *infile = NULL; |
| 2491 | int err = FALSE; |
| 2492 | FILE *fd; |
| 2493 | list_T *list = NULL; |
| 2494 | int flags = SHELL_SILENT; |
| 2495 | |
| 2496 | rettv->v_type = VAR_STRING; |
| 2497 | rettv->vval.v_string = NULL; |
| 2498 | if (check_restricted() || check_secure()) |
| 2499 | goto errret; |
| 2500 | |
Yegappan Lakshmanan | 0ad871d | 2021-07-23 20:37:56 +0200 | [diff] [blame] | 2501 | if (in_vim9script() |
| 2502 | && (check_for_string_arg(argvars, 0) == FAIL |
Bram Moolenaar | 944cc9c | 2022-06-27 22:17:37 +0100 | [diff] [blame] | 2503 | || check_for_opt_string_or_number_or_list_arg(argvars, 1) |
| 2504 | == FAIL)) |
Yegappan Lakshmanan | 0ad871d | 2021-07-23 20:37:56 +0200 | [diff] [blame] | 2505 | return; |
| 2506 | |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2507 | if (argvars[1].v_type != VAR_UNKNOWN) |
| 2508 | { |
| 2509 | /* |
| 2510 | * Write the text to a temp file, to be used for input of the shell |
| 2511 | * command. |
| 2512 | */ |
| 2513 | if ((infile = vim_tempname('i', TRUE)) == NULL) |
| 2514 | { |
Bram Moolenaar | 460ae5d | 2022-01-01 14:19:49 +0000 | [diff] [blame] | 2515 | emsg(_(e_cant_get_temp_file_name)); |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2516 | goto errret; |
| 2517 | } |
| 2518 | |
| 2519 | fd = mch_fopen((char *)infile, WRITEBIN); |
| 2520 | if (fd == NULL) |
| 2521 | { |
Bram Moolenaar | 460ae5d | 2022-01-01 14:19:49 +0000 | [diff] [blame] | 2522 | semsg(_(e_cant_open_file_str), infile); |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2523 | goto errret; |
| 2524 | } |
| 2525 | if (argvars[1].v_type == VAR_NUMBER) |
| 2526 | { |
| 2527 | linenr_T lnum; |
| 2528 | buf_T *buf; |
| 2529 | |
| 2530 | buf = buflist_findnr(argvars[1].vval.v_number); |
| 2531 | if (buf == NULL) |
| 2532 | { |
Bram Moolenaar | 40bcec1 | 2021-12-05 22:19:27 +0000 | [diff] [blame] | 2533 | semsg(_(e_buffer_nr_does_not_exist), argvars[1].vval.v_number); |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2534 | fclose(fd); |
| 2535 | goto errret; |
| 2536 | } |
| 2537 | |
| 2538 | for (lnum = 1; lnum <= buf->b_ml.ml_line_count; lnum++) |
| 2539 | { |
| 2540 | for (p = ml_get_buf(buf, lnum, FALSE); *p != NUL; ++p) |
| 2541 | if (putc(*p == '\n' ? NUL : *p, fd) == EOF) |
| 2542 | { |
| 2543 | err = TRUE; |
| 2544 | break; |
| 2545 | } |
| 2546 | if (putc(NL, fd) == EOF) |
| 2547 | { |
| 2548 | err = TRUE; |
| 2549 | break; |
| 2550 | } |
| 2551 | } |
| 2552 | } |
| 2553 | else if (argvars[1].v_type == VAR_LIST) |
| 2554 | { |
| 2555 | if (write_list(fd, argvars[1].vval.v_list, TRUE) == FAIL) |
| 2556 | err = TRUE; |
| 2557 | } |
| 2558 | else |
| 2559 | { |
| 2560 | size_t len; |
| 2561 | char_u buf[NUMBUFLEN]; |
| 2562 | |
| 2563 | p = tv_get_string_buf_chk(&argvars[1], buf); |
| 2564 | if (p == NULL) |
| 2565 | { |
| 2566 | fclose(fd); |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2567 | goto errret; // type error; errmsg already given |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2568 | } |
| 2569 | len = STRLEN(p); |
| 2570 | if (len > 0 && fwrite(p, len, 1, fd) != 1) |
| 2571 | err = TRUE; |
| 2572 | } |
| 2573 | if (fclose(fd) != 0) |
| 2574 | err = TRUE; |
| 2575 | if (err) |
| 2576 | { |
Bram Moolenaar | a6f7929 | 2022-01-04 21:30:47 +0000 | [diff] [blame] | 2577 | emsg(_(e_error_writing_temp_file)); |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2578 | goto errret; |
| 2579 | } |
| 2580 | } |
| 2581 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2582 | // Omit SHELL_COOKED when invoked with ":silent". Avoids that the shell |
| 2583 | // echoes typeahead, that messes up the display. |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2584 | if (!msg_silent) |
| 2585 | flags += SHELL_COOKED; |
| 2586 | |
| 2587 | if (retlist) |
| 2588 | { |
| 2589 | int len; |
| 2590 | listitem_T *li; |
| 2591 | char_u *s = NULL; |
| 2592 | char_u *start; |
| 2593 | char_u *end; |
| 2594 | int i; |
| 2595 | |
| 2596 | res = get_cmd_output(tv_get_string(&argvars[0]), infile, flags, &len); |
| 2597 | if (res == NULL) |
| 2598 | goto errret; |
| 2599 | |
| 2600 | list = list_alloc(); |
| 2601 | if (list == NULL) |
| 2602 | goto errret; |
| 2603 | |
| 2604 | for (i = 0; i < len; ++i) |
| 2605 | { |
| 2606 | start = res + i; |
| 2607 | while (i < len && res[i] != NL) |
| 2608 | ++i; |
| 2609 | end = res + i; |
| 2610 | |
| 2611 | s = alloc(end - start + 1); |
| 2612 | if (s == NULL) |
| 2613 | goto errret; |
| 2614 | |
| 2615 | for (p = s; start < end; ++p, ++start) |
| 2616 | *p = *start == NUL ? NL : *start; |
| 2617 | *p = NUL; |
| 2618 | |
| 2619 | li = listitem_alloc(); |
| 2620 | if (li == NULL) |
| 2621 | { |
| 2622 | vim_free(s); |
| 2623 | goto errret; |
| 2624 | } |
| 2625 | li->li_tv.v_type = VAR_STRING; |
| 2626 | li->li_tv.v_lock = 0; |
| 2627 | li->li_tv.vval.v_string = s; |
| 2628 | list_append(list, li); |
| 2629 | } |
| 2630 | |
| 2631 | rettv_list_set(rettv, list); |
| 2632 | list = NULL; |
| 2633 | } |
| 2634 | else |
| 2635 | { |
| 2636 | res = get_cmd_output(tv_get_string(&argvars[0]), infile, flags, NULL); |
| 2637 | #ifdef USE_CRNL |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2638 | // translate <CR><NL> into <NL> |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2639 | if (res != NULL) |
| 2640 | { |
| 2641 | char_u *s, *d; |
| 2642 | |
| 2643 | d = res; |
| 2644 | for (s = res; *s; ++s) |
| 2645 | { |
| 2646 | if (s[0] == CAR && s[1] == NL) |
| 2647 | ++s; |
| 2648 | *d++ = *s; |
| 2649 | } |
| 2650 | *d = NUL; |
| 2651 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2652 | #endif |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2653 | rettv->vval.v_string = res; |
| 2654 | res = NULL; |
| 2655 | } |
| 2656 | |
| 2657 | errret: |
| 2658 | if (infile != NULL) |
| 2659 | { |
| 2660 | mch_remove(infile); |
| 2661 | vim_free(infile); |
| 2662 | } |
| 2663 | if (res != NULL) |
| 2664 | vim_free(res); |
| 2665 | if (list != NULL) |
| 2666 | list_free(list); |
| 2667 | } |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2668 | |
| 2669 | /* |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2670 | * "system()" function |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2671 | */ |
| 2672 | void |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2673 | f_system(typval_T *argvars, typval_T *rettv) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2674 | { |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2675 | get_cmd_output_as_rettv(argvars, rettv, FALSE); |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2676 | } |
| 2677 | |
| 2678 | /* |
Bram Moolenaar | 26262f8 | 2019-09-04 20:59:15 +0200 | [diff] [blame] | 2679 | * "systemlist()" function |
| 2680 | */ |
| 2681 | void |
| 2682 | f_systemlist(typval_T *argvars, typval_T *rettv) |
| 2683 | { |
| 2684 | get_cmd_output_as_rettv(argvars, rettv, TRUE); |
| 2685 | } |
| 2686 | # endif // FEAT_EVAL |
| 2687 | |
| 2688 | #endif |
| 2689 | |
| 2690 | /* |
Bram Moolenaar | a9dc375 | 2010-07-11 20:46:53 +0200 | [diff] [blame] | 2691 | * Return TRUE when need to go to Insert mode because of 'insertmode'. |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2692 | * Don't do this when still processing a command or a mapping. |
| 2693 | * Don't do this when inside a ":normal" command. |
| 2694 | */ |
| 2695 | int |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2696 | goto_im(void) |
Bram Moolenaar | 071d427 | 2004-06-13 20:20:40 +0000 | [diff] [blame] | 2697 | { |
| 2698 | return (p_im && stuff_empty() && typebuf_typed()); |
| 2699 | } |
Bram Moolenaar | 75a8d74 | 2014-05-07 15:10:21 +0200 | [diff] [blame] | 2700 | |
| 2701 | /* |
Bram Moolenaar | 050fe7e | 2014-05-22 14:00:16 +0200 | [diff] [blame] | 2702 | * Returns the isolated name of the shell in allocated memory: |
Bram Moolenaar | 75a8d74 | 2014-05-07 15:10:21 +0200 | [diff] [blame] | 2703 | * - Skip beyond any path. E.g., "/usr/bin/csh -f" -> "csh -f". |
| 2704 | * - Remove any argument. E.g., "csh -f" -> "csh". |
| 2705 | * But don't allow a space in the path, so that this works: |
| 2706 | * "/usr/bin/csh --rcfile ~/.cshrc" |
| 2707 | * But don't do that for Windows, it's common to have a space in the path. |
Bram Moolenaar | 28ee892 | 2020-10-28 20:20:00 +0100 | [diff] [blame] | 2708 | * Returns NULL when out of memory. |
Bram Moolenaar | 75a8d74 | 2014-05-07 15:10:21 +0200 | [diff] [blame] | 2709 | */ |
| 2710 | char_u * |
Bram Moolenaar | 9b57814 | 2016-01-30 19:39:49 +0100 | [diff] [blame] | 2711 | get_isolated_shell_name(void) |
Bram Moolenaar | 75a8d74 | 2014-05-07 15:10:21 +0200 | [diff] [blame] | 2712 | { |
| 2713 | char_u *p; |
| 2714 | |
Bram Moolenaar | 4f97475 | 2019-02-17 17:44:42 +0100 | [diff] [blame] | 2715 | #ifdef MSWIN |
Bram Moolenaar | 75a8d74 | 2014-05-07 15:10:21 +0200 | [diff] [blame] | 2716 | p = gettail(p_sh); |
Bram Moolenaar | 71ccd03 | 2020-06-12 22:59:11 +0200 | [diff] [blame] | 2717 | p = vim_strnsave(p, skiptowhite(p) - p); |
Bram Moolenaar | 75a8d74 | 2014-05-07 15:10:21 +0200 | [diff] [blame] | 2718 | #else |
| 2719 | p = skiptowhite(p_sh); |
| 2720 | if (*p == NUL) |
| 2721 | { |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2722 | // No white space, use the tail. |
Bram Moolenaar | 75a8d74 | 2014-05-07 15:10:21 +0200 | [diff] [blame] | 2723 | p = vim_strsave(gettail(p_sh)); |
| 2724 | } |
| 2725 | else |
| 2726 | { |
| 2727 | char_u *p1, *p2; |
| 2728 | |
Bram Moolenaar | 85a2002 | 2019-12-21 18:25:54 +0100 | [diff] [blame] | 2729 | // Find the last path separator before the space. |
Bram Moolenaar | 75a8d74 | 2014-05-07 15:10:21 +0200 | [diff] [blame] | 2730 | p1 = p_sh; |
Bram Moolenaar | 91acfff | 2017-03-12 19:22:36 +0100 | [diff] [blame] | 2731 | for (p2 = p_sh; p2 < p; MB_PTR_ADV(p2)) |
Bram Moolenaar | 75a8d74 | 2014-05-07 15:10:21 +0200 | [diff] [blame] | 2732 | if (vim_ispathsep(*p2)) |
| 2733 | p1 = p2 + 1; |
Bram Moolenaar | 71ccd03 | 2020-06-12 22:59:11 +0200 | [diff] [blame] | 2734 | p = vim_strnsave(p1, p - p1); |
Bram Moolenaar | 75a8d74 | 2014-05-07 15:10:21 +0200 | [diff] [blame] | 2735 | } |
| 2736 | #endif |
| 2737 | return p; |
| 2738 | } |
Bram Moolenaar | 5fd0f50 | 2019-02-13 23:13:28 +0100 | [diff] [blame] | 2739 | |
| 2740 | /* |
| 2741 | * Check if the "://" of a URL is at the pointer, return URL_SLASH. |
| 2742 | * Also check for ":\\", which MS Internet Explorer accepts, return |
| 2743 | * URL_BACKSLASH. |
| 2744 | */ |
| 2745 | int |
| 2746 | path_is_url(char_u *p) |
| 2747 | { |
| 2748 | if (STRNCMP(p, "://", (size_t)3) == 0) |
| 2749 | return URL_SLASH; |
| 2750 | else if (STRNCMP(p, ":\\\\", (size_t)3) == 0) |
| 2751 | return URL_BACKSLASH; |
| 2752 | return 0; |
| 2753 | } |
| 2754 | |
| 2755 | /* |
Tsuyoshi CHO | 7b7a118 | 2021-07-11 21:51:17 +0200 | [diff] [blame] | 2756 | * Check if "fname" starts with "name://" or "name:\\". |
| 2757 | * Return URL_SLASH for "name://", URL_BACKSLASH for "name:\\". |
Bram Moolenaar | 5fd0f50 | 2019-02-13 23:13:28 +0100 | [diff] [blame] | 2758 | * Return zero otherwise. |
| 2759 | */ |
| 2760 | int |
| 2761 | path_with_url(char_u *fname) |
| 2762 | { |
| 2763 | char_u *p; |
| 2764 | |
Tsuyoshi CHO | 7b7a118 | 2021-07-11 21:51:17 +0200 | [diff] [blame] | 2765 | // We accept alphabetic characters and a dash in scheme part. |
| 2766 | // RFC 3986 allows for more, but it increases the risk of matching |
| 2767 | // non-URL text. |
| 2768 | |
| 2769 | // first character must be alpha |
zeertzjq | 0ef9a5c | 2023-01-17 21:38:25 +0000 | [diff] [blame] | 2770 | if (!ASCII_ISALPHA(*fname)) |
Tsuyoshi CHO | 7b7a118 | 2021-07-11 21:51:17 +0200 | [diff] [blame] | 2771 | return 0; |
| 2772 | |
| 2773 | // check body: alpha or dash |
zeertzjq | 0ef9a5c | 2023-01-17 21:38:25 +0000 | [diff] [blame] | 2774 | for (p = fname + 1; (ASCII_ISALPHA(*p) || (*p == '-')); ++p) |
Bram Moolenaar | 5fd0f50 | 2019-02-13 23:13:28 +0100 | [diff] [blame] | 2775 | ; |
Tsuyoshi CHO | 7b7a118 | 2021-07-11 21:51:17 +0200 | [diff] [blame] | 2776 | |
| 2777 | // check last char is not a dash |
| 2778 | if (p[-1] == '-') |
| 2779 | return 0; |
| 2780 | |
| 2781 | // "://" or ":\\" must follow |
Bram Moolenaar | 5fd0f50 | 2019-02-13 23:13:28 +0100 | [diff] [blame] | 2782 | return path_is_url(p); |
| 2783 | } |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2784 | |
Bram Moolenaar | 3075a45 | 2021-11-17 15:51:52 +0000 | [diff] [blame] | 2785 | #if defined(FEAT_EVAL) || defined(PROTO) |
| 2786 | /* |
| 2787 | * Return the dictionary of v:event. |
| 2788 | * Save and clear the value in case it already has items. |
| 2789 | */ |
| 2790 | dict_T * |
| 2791 | get_v_event(save_v_event_T *sve) |
| 2792 | { |
| 2793 | dict_T *v_event = get_vim_var_dict(VV_EVENT); |
| 2794 | |
| 2795 | if (v_event->dv_hashtab.ht_used > 0) |
| 2796 | { |
| 2797 | // recursive use of v:event, save, make empty and restore later |
| 2798 | sve->sve_did_save = TRUE; |
| 2799 | sve->sve_hashtab = v_event->dv_hashtab; |
| 2800 | hash_init(&v_event->dv_hashtab); |
| 2801 | } |
| 2802 | else |
| 2803 | sve->sve_did_save = FALSE; |
| 2804 | return v_event; |
| 2805 | } |
| 2806 | |
| 2807 | void |
| 2808 | restore_v_event(dict_T *v_event, save_v_event_T *sve) |
| 2809 | { |
| 2810 | dict_free_contents(v_event); |
| 2811 | if (sve->sve_did_save) |
| 2812 | v_event->dv_hashtab = sve->sve_hashtab; |
| 2813 | else |
| 2814 | hash_init(&v_event->dv_hashtab); |
| 2815 | } |
| 2816 | #endif |
| 2817 | |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2818 | /* |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 2819 | * Fires a ModeChanged autocmd event if appropriate. |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2820 | */ |
| 2821 | void |
Yegappan Lakshmanan | a23a11b | 2023-02-21 14:27:41 +0000 | [diff] [blame] | 2822 | may_trigger_modechanged(void) |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2823 | { |
Bram Moolenaar | 3075a45 | 2021-11-17 15:51:52 +0000 | [diff] [blame] | 2824 | #ifdef FEAT_EVAL |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2825 | dict_T *v_event; |
Bram Moolenaar | 3075a45 | 2021-11-17 15:51:52 +0000 | [diff] [blame] | 2826 | save_v_event_T save_v_event; |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 2827 | char_u curr_mode[MODE_MAX_LENGTH]; |
| 2828 | char_u pattern_buf[2 * MODE_MAX_LENGTH]; |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2829 | |
Bram Moolenaar | 61c4b04 | 2022-10-18 15:10:11 +0100 | [diff] [blame] | 2830 | // Skip this when got_int is set, the autocommand will not be executed. |
| 2831 | // Better trigger it next time. |
| 2832 | if (!has_modechanged() || got_int) |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2833 | return; |
| 2834 | |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 2835 | get_mode(curr_mode); |
| 2836 | if (STRCMP(curr_mode, last_mode) == 0) |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | 25def2c | 2021-10-22 18:56:39 +0100 | [diff] [blame] | 2837 | return; |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | 25def2c | 2021-10-22 18:56:39 +0100 | [diff] [blame] | 2838 | |
Bram Moolenaar | 3075a45 | 2021-11-17 15:51:52 +0000 | [diff] [blame] | 2839 | v_event = get_v_event(&save_v_event); |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 2840 | (void)dict_add_string(v_event, "new_mode", curr_mode); |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2841 | (void)dict_add_string(v_event, "old_mode", last_mode); |
| 2842 | dict_set_items_ro(v_event); |
| 2843 | |
| 2844 | // concatenate modes in format "old_mode:new_mode" |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 2845 | vim_snprintf((char *)pattern_buf, sizeof(pattern_buf), "%s:%s", last_mode, |
| 2846 | curr_mode); |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2847 | |
LemonBoy | 2bf52dd | 2022-04-09 18:17:34 +0100 | [diff] [blame] | 2848 | apply_autocmds(EVENT_MODECHANGED, pattern_buf, NULL, FALSE, curbuf); |
| 2849 | STRCPY(last_mode, curr_mode); |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2850 | |
Bram Moolenaar | 3075a45 | 2021-11-17 15:51:52 +0000 | [diff] [blame] | 2851 | restore_v_event(v_event, &save_v_event); |
=?UTF-8?q?Magnus=20Gro=C3=9F?= | f1e8876 | 2021-09-12 13:39:55 +0200 | [diff] [blame] | 2852 | #endif |
| 2853 | } |
Christian Brabandt | 22cbc8a | 2023-11-19 10:47:21 +0100 | [diff] [blame] | 2854 | |
Christian Brabandt | 22cbc8a | 2023-11-19 10:47:21 +0100 | [diff] [blame] | 2855 | // For overflow detection, add a digit safely to a long value. |
| 2856 | int |
| 2857 | vim_append_digit_long(long *value, int digit) |
| 2858 | { |
| 2859 | long x = *value; |
| 2860 | if (x > ((LONG_MAX - (long)digit) / 10)) |
| 2861 | return FAIL; |
| 2862 | *value = x * 10 + (long)digit; |
| 2863 | return OK; |
| 2864 | } |
Ernie Rael | 2b0882f | 2023-11-23 20:21:45 +0100 | [diff] [blame] | 2865 | |
| 2866 | // Return something that fits into an int. |
| 2867 | int |
Ernie Rael | fda700c | 2023-11-30 18:20:00 +0100 | [diff] [blame] | 2868 | trim_to_int(vimlong_T x) |
Ernie Rael | 2b0882f | 2023-11-23 20:21:45 +0100 | [diff] [blame] | 2869 | { |
| 2870 | return x > INT_MAX ? INT_MAX : x < INT_MIN ? INT_MIN : x; |
| 2871 | } |
| 2872 | |