| /* vi:set ts=8 sts=4 sw=4 noet: |
| * |
| * VIM - Vi IMproved by Bram Moolenaar |
| * |
| * Do ":help uganda" in Vim to read copying and usage conditions. |
| * Do ":help credits" in Vim to see a list of people who contributed. |
| * See README.txt for an overview of the Vim source code. |
| */ |
| |
| /* |
| * misc1.c: functions that didn't seem to fit elsewhere |
| */ |
| |
| #include "vim.h" |
| #include "version.h" |
| |
| #if defined(__HAIKU__) |
| # include <storage/FindDirectory.h> |
| #endif |
| |
| #if defined(MSWIN) |
| # include <lm.h> |
| #endif |
| |
| #define URL_SLASH 1 // path_is_url() has found "://" |
| #define URL_BACKSLASH 2 // path_is_url() has found ":\\" |
| |
| // All user names (for ~user completion as done by shell). |
| static garray_T ga_users; |
| |
| /* |
| * get_leader_len() returns the length in bytes of the prefix of the given |
| * string which introduces a comment. If this string is not a comment then |
| * 0 is returned. |
| * When "flags" is not NULL, it is set to point to the flags of the recognized |
| * comment leader. |
| * "backward" must be true for the "O" command. |
| * If "include_space" is set, include trailing whitespace while calculating the |
| * length. |
| */ |
| int |
| get_leader_len( |
| char_u *line, |
| char_u **flags, |
| int backward, |
| int include_space) |
| { |
| int i, j; |
| int result; |
| int got_com = FALSE; |
| int found_one; |
| char_u part_buf[COM_MAX_LEN]; // buffer for one option part |
| char_u *string; // pointer to comment string |
| char_u *list; |
| int middle_match_len = 0; |
| char_u *prev_list; |
| char_u *saved_flags = NULL; |
| |
| result = i = 0; |
| while (VIM_ISWHITE(line[i])) // leading white space is ignored |
| ++i; |
| |
| /* |
| * Repeat to match several nested comment strings. |
| */ |
| while (line[i] != NUL) |
| { |
| /* |
| * scan through the 'comments' option for a match |
| */ |
| found_one = FALSE; |
| for (list = curbuf->b_p_com; *list; ) |
| { |
| // Get one option part into part_buf[]. Advance "list" to next |
| // one. Put "string" at start of string. |
| if (!got_com && flags != NULL) |
| *flags = list; // remember where flags started |
| prev_list = list; |
| (void)copy_option_part(&list, part_buf, COM_MAX_LEN, ","); |
| string = vim_strchr(part_buf, ':'); |
| if (string == NULL) // missing ':', ignore this part |
| continue; |
| *string++ = NUL; // isolate flags from string |
| |
| // If we found a middle match previously, use that match when this |
| // is not a middle or end. |
| if (middle_match_len != 0 |
| && vim_strchr(part_buf, COM_MIDDLE) == NULL |
| && vim_strchr(part_buf, COM_END) == NULL) |
| break; |
| |
| // When we already found a nested comment, only accept further |
| // nested comments. |
| if (got_com && vim_strchr(part_buf, COM_NEST) == NULL) |
| continue; |
| |
| // When 'O' flag present and using "O" command skip this one. |
| if (backward && vim_strchr(part_buf, COM_NOBACK) != NULL) |
| continue; |
| |
| // Line contents and string must match. |
| // When string starts with white space, must have some white space |
| // (but the amount does not need to match, there might be a mix of |
| // TABs and spaces). |
| if (VIM_ISWHITE(string[0])) |
| { |
| if (i == 0 || !VIM_ISWHITE(line[i - 1])) |
| continue; // missing white space |
| while (VIM_ISWHITE(string[0])) |
| ++string; |
| } |
| for (j = 0; string[j] != NUL && string[j] == line[i + j]; ++j) |
| ; |
| if (string[j] != NUL) |
| continue; // string doesn't match |
| |
| // When 'b' flag used, there must be white space or an |
| // end-of-line after the string in the line. |
| if (vim_strchr(part_buf, COM_BLANK) != NULL |
| && !VIM_ISWHITE(line[i + j]) && line[i + j] != NUL) |
| continue; |
| |
| // We have found a match, stop searching unless this is a middle |
| // comment. The middle comment can be a substring of the end |
| // comment in which case it's better to return the length of the |
| // end comment and its flags. Thus we keep searching with middle |
| // and end matches and use an end match if it matches better. |
| if (vim_strchr(part_buf, COM_MIDDLE) != NULL) |
| { |
| if (middle_match_len == 0) |
| { |
| middle_match_len = j; |
| saved_flags = prev_list; |
| } |
| continue; |
| } |
| if (middle_match_len != 0 && j > middle_match_len) |
| // Use this match instead of the middle match, since it's a |
| // longer thus better match. |
| middle_match_len = 0; |
| |
| if (middle_match_len == 0) |
| i += j; |
| found_one = TRUE; |
| break; |
| } |
| |
| if (middle_match_len != 0) |
| { |
| // Use the previously found middle match after failing to find a |
| // match with an end. |
| if (!got_com && flags != NULL) |
| *flags = saved_flags; |
| i += middle_match_len; |
| found_one = TRUE; |
| } |
| |
| // No match found, stop scanning. |
| if (!found_one) |
| break; |
| |
| result = i; |
| |
| // Include any trailing white space. |
| while (VIM_ISWHITE(line[i])) |
| ++i; |
| |
| if (include_space) |
| result = i; |
| |
| // If this comment doesn't nest, stop here. |
| got_com = TRUE; |
| if (vim_strchr(part_buf, COM_NEST) == NULL) |
| break; |
| } |
| return result; |
| } |
| |
| /* |
| * Return the offset at which the last comment in line starts. If there is no |
| * comment in the whole line, -1 is returned. |
| * |
| * When "flags" is not null, it is set to point to the flags describing the |
| * recognized comment leader. |
| */ |
| int |
| get_last_leader_offset(char_u *line, char_u **flags) |
| { |
| int result = -1; |
| int i, j; |
| int lower_check_bound = 0; |
| char_u *string; |
| char_u *com_leader; |
| char_u *com_flags; |
| char_u *list; |
| int found_one; |
| char_u part_buf[COM_MAX_LEN]; // buffer for one option part |
| |
| /* |
| * Repeat to match several nested comment strings. |
| */ |
| i = (int)STRLEN(line); |
| while (--i >= lower_check_bound) |
| { |
| /* |
| * scan through the 'comments' option for a match |
| */ |
| found_one = FALSE; |
| for (list = curbuf->b_p_com; *list; ) |
| { |
| char_u *flags_save = list; |
| |
| /* |
| * Get one option part into part_buf[]. Advance list to next one. |
| * put string at start of string. |
| */ |
| (void)copy_option_part(&list, part_buf, COM_MAX_LEN, ","); |
| string = vim_strchr(part_buf, ':'); |
| if (string == NULL) // If everything is fine, this cannot actually |
| // happen. |
| continue; |
| *string++ = NUL; // Isolate flags from string. |
| com_leader = string; |
| |
| /* |
| * Line contents and string must match. |
| * When string starts with white space, must have some white space |
| * (but the amount does not need to match, there might be a mix of |
| * TABs and spaces). |
| */ |
| if (VIM_ISWHITE(string[0])) |
| { |
| if (i == 0 || !VIM_ISWHITE(line[i - 1])) |
| continue; |
| while (VIM_ISWHITE(*string)) |
| ++string; |
| } |
| for (j = 0; string[j] != NUL && string[j] == line[i + j]; ++j) |
| /* do nothing */; |
| if (string[j] != NUL) |
| continue; |
| |
| /* |
| * When 'b' flag used, there must be white space or an |
| * end-of-line after the string in the line. |
| */ |
| if (vim_strchr(part_buf, COM_BLANK) != NULL |
| && !VIM_ISWHITE(line[i + j]) && line[i + j] != NUL) |
| continue; |
| |
| if (vim_strchr(part_buf, COM_MIDDLE) != NULL) |
| { |
| // For a middlepart comment, only consider it to match if |
| // everything before the current position in the line is |
| // whitespace. Otherwise we would think we are inside a |
| // comment if the middle part appears somewhere in the middle |
| // of the line. E.g. for C the "*" appears often. |
| for (j = 0; VIM_ISWHITE(line[j]) && j <= i; j++) |
| ; |
| if (j < i) |
| continue; |
| } |
| |
| /* |
| * We have found a match, stop searching. |
| */ |
| found_one = TRUE; |
| |
| if (flags) |
| *flags = flags_save; |
| com_flags = flags_save; |
| |
| break; |
| } |
| |
| if (found_one) |
| { |
| char_u part_buf2[COM_MAX_LEN]; // buffer for one option part |
| int len1, len2, off; |
| |
| result = i; |
| /* |
| * If this comment nests, continue searching. |
| */ |
| if (vim_strchr(part_buf, COM_NEST) != NULL) |
| continue; |
| |
| lower_check_bound = i; |
| |
| // Let's verify whether the comment leader found is a substring |
| // of other comment leaders. If it is, let's adjust the |
| // lower_check_bound so that we make sure that we have determined |
| // the comment leader correctly. |
| |
| while (VIM_ISWHITE(*com_leader)) |
| ++com_leader; |
| len1 = (int)STRLEN(com_leader); |
| |
| for (list = curbuf->b_p_com; *list; ) |
| { |
| char_u *flags_save = list; |
| |
| (void)copy_option_part(&list, part_buf2, COM_MAX_LEN, ","); |
| if (flags_save == com_flags) |
| continue; |
| string = vim_strchr(part_buf2, ':'); |
| ++string; |
| while (VIM_ISWHITE(*string)) |
| ++string; |
| len2 = (int)STRLEN(string); |
| if (len2 == 0) |
| continue; |
| |
| // Now we have to verify whether string ends with a substring |
| // beginning the com_leader. |
| for (off = (len2 > i ? i : len2); off > 0 && off + len1 > len2;) |
| { |
| --off; |
| if (!STRNCMP(string + off, com_leader, len2 - off)) |
| { |
| if (i - off < lower_check_bound) |
| lower_check_bound = i - off; |
| } |
| } |
| } |
| } |
| } |
| return result; |
| } |
| |
| /* |
| * Return the number of window lines occupied by buffer line "lnum". |
| * Includes any filler lines. |
| */ |
| int |
| plines(linenr_T lnum) |
| { |
| return plines_win(curwin, lnum, TRUE); |
| } |
| |
| int |
| plines_win( |
| win_T *wp, |
| linenr_T lnum, |
| int limit_winheight) // when TRUE limit to window height |
| { |
| #if defined(FEAT_DIFF) || defined(PROTO) |
| // Check for filler lines above this buffer line. When folded the result |
| // is one line anyway. |
| return plines_win_nofill(wp, lnum, limit_winheight) |
| + diff_check_fill(wp, lnum); |
| } |
| |
| /* |
| * Return the number of window lines occupied by buffer line "lnum". |
| * Does not include filler lines. |
| */ |
| int |
| plines_nofill(linenr_T lnum) |
| { |
| return plines_win_nofill(curwin, lnum, TRUE); |
| } |
| |
| int |
| plines_win_nofill( |
| win_T *wp, |
| linenr_T lnum, |
| int limit_winheight) // when TRUE limit to window height |
| { |
| #endif |
| int lines; |
| |
| if (wp->w_width == 0) |
| return 1; |
| |
| #ifdef FEAT_FOLDING |
| // Folded lines are handled just like an empty line. |
| // NOTE: Caller must handle lines that are MAYBE folded. |
| if (lineFolded(wp, lnum) == TRUE) |
| return 1; |
| #endif |
| |
| if (!wp->w_p_wrap) |
| lines = 1 |
| #ifdef FEAT_PROP_POPUP |
| // add a line for each "above" and "below" aligned text property |
| + prop_count_above_below(wp->w_buffer, lnum) |
| #endif |
| ; |
| else |
| lines = plines_win_nofold(wp, lnum); |
| |
| if (limit_winheight && lines > wp->w_height) |
| return wp->w_height; |
| return lines; |
| } |
| |
| /* |
| * Return number of window lines physical line "lnum" will occupy in window |
| * "wp". Does not care about folding, 'wrap' or 'diff'. |
| */ |
| int |
| plines_win_nofold(win_T *wp, linenr_T lnum) |
| { |
| char_u *s; |
| long col; |
| int width; |
| chartabsize_T cts; |
| |
| s = ml_get_buf(wp->w_buffer, lnum, FALSE); |
| init_chartabsize_arg(&cts, wp, lnum, 0, s, s); |
| if (*s == NUL |
| #ifdef FEAT_PROP_POPUP |
| && !cts.cts_has_prop_with_text |
| #endif |
| ) |
| return 1; // be quick for an empty line |
| win_linetabsize_cts(&cts, (colnr_T)MAXCOL); |
| clear_chartabsize_arg(&cts); |
| col = (int)cts.cts_vcol; |
| |
| // If list mode is on, then the '$' at the end of the line may take up one |
| // extra column. |
| if (wp->w_p_list && wp->w_lcs_chars.eol != NUL) |
| col += 1; |
| |
| /* |
| * Add column offset for 'number', 'relativenumber' and 'foldcolumn'. |
| */ |
| width = wp->w_width - win_col_off(wp); |
| if (width <= 0) |
| return 32000; |
| if (col <= width) |
| return 1; |
| col -= width; |
| width += win_col_off2(wp); |
| return (col + (width - 1)) / width + 1; |
| } |
| |
| /* |
| * Like plines_win(), but only reports the number of physical screen lines |
| * used from the start of the line to the given column number. |
| */ |
| int |
| plines_win_col(win_T *wp, linenr_T lnum, long column) |
| { |
| long col; |
| int lines = 0; |
| int width; |
| char_u *line; |
| chartabsize_T cts; |
| |
| #ifdef FEAT_DIFF |
| // Check for filler lines above this buffer line. When folded the result |
| // is one line anyway. |
| lines = diff_check_fill(wp, lnum); |
| #endif |
| |
| if (!wp->w_p_wrap) |
| return lines + 1; |
| |
| if (wp->w_width == 0) |
| return lines + 1; |
| |
| line = ml_get_buf(wp->w_buffer, lnum, FALSE); |
| |
| init_chartabsize_arg(&cts, wp, lnum, 0, line, line); |
| while (*cts.cts_ptr != NUL && --column >= 0) |
| { |
| cts.cts_vcol += win_lbr_chartabsize(&cts, NULL); |
| MB_PTR_ADV(cts.cts_ptr); |
| } |
| |
| /* |
| * If *cts.cts_ptr is a TAB, and the TAB is not displayed as ^I, and we're |
| * not in MODE_INSERT state, then col must be adjusted so that it |
| * represents the last screen position of the TAB. This only fixes an |
| * error when the TAB wraps from one screen line to the next (when |
| * 'columns' is not a multiple of 'ts') -- webb. |
| */ |
| col = cts.cts_vcol; |
| if (*cts.cts_ptr == TAB && (State & MODE_NORMAL) |
| && (!wp->w_p_list || wp->w_lcs_chars.tab1)) |
| col += win_lbr_chartabsize(&cts, NULL) - 1; |
| clear_chartabsize_arg(&cts); |
| |
| /* |
| * Add column offset for 'number', 'relativenumber', 'foldcolumn', etc. |
| */ |
| width = wp->w_width - win_col_off(wp); |
| if (width <= 0) |
| return 9999; |
| |
| lines += 1; |
| if (col > width) |
| lines += (col - width) / (width + win_col_off2(wp)) + 1; |
| return lines; |
| } |
| |
| /* |
| * Return number of window lines the physical line range from "first" until |
| * "last" will occupy in window "wp". Takes into account folding, 'wrap', |
| * topfill and filler lines beyond the end of the buffer. Limit to "max" lines. |
| */ |
| int |
| plines_m_win(win_T *wp, linenr_T first, linenr_T last, int max) |
| { |
| int count = 0; |
| |
| while (first <= last && count < max) |
| { |
| #ifdef FEAT_FOLDING |
| int x; |
| |
| // Check if there are any really folded lines, but also included lines |
| // that are maybe folded. |
| x = foldedCount(wp, first, NULL); |
| if (x > 0) |
| { |
| ++count; // count 1 for "+-- folded" line |
| first += x; |
| } |
| else |
| #endif |
| { |
| #ifdef FEAT_DIFF |
| if (first == wp->w_topline) |
| count += plines_win_nofill(wp, first, FALSE) + wp->w_topfill; |
| else |
| #endif |
| count += plines_win(wp, first, FALSE); |
| ++first; |
| } |
| } |
| #ifdef FEAT_DIFF |
| if (first == wp->w_buffer->b_ml.ml_line_count + 1) |
| count += diff_check_fill(wp, first); |
| #endif |
| return MIN(max, count); |
| } |
| |
| int |
| gchar_pos(pos_T *pos) |
| { |
| char_u *ptr; |
| |
| // When searching columns is sometimes put at the end of a line. |
| if (pos->col == MAXCOL) |
| return NUL; |
| ptr = ml_get_pos(pos); |
| if (has_mbyte) |
| return (*mb_ptr2char)(ptr); |
| return (int)*ptr; |
| } |
| |
| int |
| gchar_cursor(void) |
| { |
| if (has_mbyte) |
| return (*mb_ptr2char)(ml_get_cursor()); |
| return (int)*ml_get_cursor(); |
| } |
| |
| /* |
| * Write a character at the current cursor position. |
| * It is directly written into the block. |
| */ |
| void |
| pchar_cursor(int c) |
| { |
| *(ml_get_buf(curbuf, curwin->w_cursor.lnum, TRUE) |
| + curwin->w_cursor.col) = c; |
| } |
| |
| /* |
| * Skip to next part of an option argument: Skip space and comma. |
| */ |
| char_u * |
| skip_to_option_part(char_u *p) |
| { |
| if (*p == ',') |
| ++p; |
| while (*p == ' ') |
| ++p; |
| return p; |
| } |
| |
| /* |
| * check_status: called when the status bars for the buffer 'buf' |
| * need to be updated |
| */ |
| void |
| check_status(buf_T *buf) |
| { |
| win_T *wp; |
| |
| FOR_ALL_WINDOWS(wp) |
| if (wp->w_buffer == buf && wp->w_status_height) |
| { |
| wp->w_redr_status = TRUE; |
| set_must_redraw(UPD_VALID); |
| } |
| } |
| |
| /* |
| * Ask for a reply from the user, a 'y' or a 'n', with prompt "str" (which |
| * should have been translated already). |
| * No other characters are accepted, the message is repeated until a valid |
| * reply is entered or CTRL-C is hit. |
| * If direct is TRUE, don't use vgetc() but ui_inchar(), don't get characters |
| * from any buffers but directly from the user. |
| * |
| * return the 'y' or 'n' |
| */ |
| int |
| ask_yesno(char_u *str, int direct) |
| { |
| int r = ' '; |
| int save_State = State; |
| |
| if (exiting) // put terminal in raw mode for this question |
| settmode(TMODE_RAW); |
| ++no_wait_return; |
| #ifdef USE_ON_FLY_SCROLL |
| dont_scroll = TRUE; // disallow scrolling here |
| #endif |
| State = MODE_CONFIRM; // mouse behaves like with :confirm |
| setmouse(); // disables mouse for xterm |
| ++no_mapping; |
| ++allow_keys; // no mapping here, but recognize keys |
| |
| while (r != 'y' && r != 'n') |
| { |
| // same highlighting as for wait_return() |
| smsg_attr(HL_ATTR(HLF_R), "%s (y/n)?", str); |
| if (direct) |
| r = get_keystroke(); |
| else |
| r = plain_vgetc(); |
| if (r == Ctrl_C || r == ESC) |
| r = 'n'; |
| msg_putchar(r); // show what you typed |
| out_flush(); |
| } |
| --no_wait_return; |
| State = save_State; |
| setmouse(); |
| --no_mapping; |
| --allow_keys; |
| |
| return r; |
| } |
| |
| #if defined(FEAT_EVAL) || defined(PROTO) |
| |
| /* |
| * Returns the current mode as a string in "buf[MODE_MAX_LENGTH]", NUL |
| * terminated. |
| * The first character represents the major mode, the following ones the minor |
| * ones. |
| */ |
| void |
| get_mode(char_u *buf) |
| { |
| int i = 0; |
| |
| if (time_for_testing == 93784) |
| { |
| // Testing the two-character code. |
| buf[i++] = 'x'; |
| buf[i++] = '!'; |
| } |
| #ifdef FEAT_TERMINAL |
| else if (term_use_loop()) |
| { |
| if (State & MODE_CMDLINE) |
| buf[i++] = 'c'; |
| buf[i++] = 't'; |
| } |
| #endif |
| else if (VIsual_active) |
| { |
| if (VIsual_select) |
| buf[i++] = VIsual_mode + 's' - 'v'; |
| else |
| { |
| buf[i++] = VIsual_mode; |
| if (restart_VIsual_select) |
| buf[i++] = 's'; |
| } |
| } |
| else if (State == MODE_HITRETURN || State == MODE_ASKMORE |
| || State == MODE_SETWSIZE |
| || State == MODE_CONFIRM) |
| { |
| buf[i++] = 'r'; |
| if (State == MODE_ASKMORE) |
| buf[i++] = 'm'; |
| else if (State == MODE_CONFIRM) |
| buf[i++] = '?'; |
| } |
| else if (State == MODE_EXTERNCMD) |
| buf[i++] = '!'; |
| else if (State & MODE_INSERT) |
| { |
| if (State & VREPLACE_FLAG) |
| { |
| buf[i++] = 'R'; |
| buf[i++] = 'v'; |
| } |
| else |
| { |
| if (State & REPLACE_FLAG) |
| buf[i++] = 'R'; |
| else |
| buf[i++] = 'i'; |
| } |
| |
| if (ins_compl_active()) |
| buf[i++] = 'c'; |
| else if (ctrl_x_mode_not_defined_yet()) |
| buf[i++] = 'x'; |
| } |
| else if ((State & MODE_CMDLINE) || exmode_active) |
| { |
| buf[i++] = 'c'; |
| if (exmode_active == EXMODE_VIM) |
| buf[i++] = 'v'; |
| else if (exmode_active == EXMODE_NORMAL) |
| buf[i++] = 'e'; |
| if ((State & MODE_CMDLINE) && cmdline_overstrike()) |
| buf[i++] = 'r'; |
| } |
| else |
| { |
| buf[i++] = 'n'; |
| if (finish_op) |
| { |
| buf[i++] = 'o'; |
| // to be able to detect force-linewise/blockwise/characterwise |
| // operations |
| buf[i++] = motion_force; |
| } |
| else if (restart_edit == 'I' || restart_edit == 'R' |
| || restart_edit == 'V') |
| { |
| buf[i++] = 'i'; |
| buf[i++] = restart_edit; |
| } |
| #ifdef FEAT_TERMINAL |
| else if (term_in_normal_mode()) |
| buf[i++] = 't'; |
| #endif |
| } |
| |
| buf[i] = NUL; |
| } |
| |
| /* |
| * "mode()" function |
| */ |
| void |
| f_mode(typval_T *argvars, typval_T *rettv) |
| { |
| char_u buf[MODE_MAX_LENGTH]; |
| |
| if (in_vim9script() && check_for_opt_bool_arg(argvars, 0) == FAIL) |
| return; |
| |
| get_mode(buf); |
| |
| // Clear out the minor mode when the argument is not a non-zero number or |
| // non-empty string. |
| if (!non_zero_arg(&argvars[0])) |
| buf[1] = NUL; |
| |
| rettv->vval.v_string = vim_strsave(buf); |
| rettv->v_type = VAR_STRING; |
| } |
| |
| static void |
| may_add_state_char(garray_T *gap, char_u *include, int c) |
| { |
| if (include == NULL || vim_strchr(include, c) != NULL) |
| ga_append(gap, c); |
| } |
| |
| /* |
| * "state()" function |
| */ |
| void |
| f_state(typval_T *argvars, typval_T *rettv) |
| { |
| garray_T ga; |
| char_u *include = NULL; |
| int i; |
| |
| if (in_vim9script() && check_for_opt_string_arg(argvars, 0) == FAIL) |
| return; |
| |
| ga_init2(&ga, 1, 20); |
| if (argvars[0].v_type != VAR_UNKNOWN) |
| include = tv_get_string(&argvars[0]); |
| |
| if (!(stuff_empty() && typebuf.tb_len == 0 && scriptin[curscript] == NULL)) |
| may_add_state_char(&ga, include, 'm'); |
| if (op_pending()) |
| may_add_state_char(&ga, include, 'o'); |
| if (autocmd_busy) |
| may_add_state_char(&ga, include, 'x'); |
| if (ins_compl_active()) |
| may_add_state_char(&ga, include, 'a'); |
| |
| # ifdef FEAT_JOB_CHANNEL |
| if (channel_in_blocking_wait()) |
| may_add_state_char(&ga, include, 'w'); |
| # endif |
| if (!get_was_safe_state()) |
| may_add_state_char(&ga, include, 'S'); |
| for (i = 0; i < get_callback_depth() && i < 3; ++i) |
| may_add_state_char(&ga, include, 'c'); |
| if (msg_scrolled > 0) |
| may_add_state_char(&ga, include, 's'); |
| |
| rettv->v_type = VAR_STRING; |
| rettv->vval.v_string = ga.ga_data; |
| } |
| |
| #endif // FEAT_EVAL |
| |
| /* |
| * Get a key stroke directly from the user. |
| * Ignores mouse clicks and scrollbar events, except a click for the left |
| * button (used at the more prompt). |
| * Doesn't use vgetc(), because it syncs undo and eats mapped characters. |
| * Disadvantage: typeahead is ignored. |
| * Translates the interrupt character for unix to ESC. |
| */ |
| int |
| get_keystroke(void) |
| { |
| char_u *buf = NULL; |
| int buflen = 150; |
| int maxlen; |
| int len = 0; |
| int n; |
| int save_mapped_ctrl_c = mapped_ctrl_c; |
| int waited = 0; |
| |
| mapped_ctrl_c = FALSE; // mappings are not used here |
| for (;;) |
| { |
| cursor_on(); |
| out_flush(); |
| |
| // Leave some room for check_termcode() to insert a key code into (max |
| // 5 chars plus NUL). And fix_input_buffer() can triple the number of |
| // bytes. |
| maxlen = (buflen - 6 - len) / 3; |
| if (buf == NULL) |
| buf = alloc(buflen); |
| else if (maxlen < 10) |
| { |
| char_u *t_buf = buf; |
| |
| // Need some more space. This might happen when receiving a long |
| // escape sequence. |
| buflen += 100; |
| buf = vim_realloc(buf, buflen); |
| if (buf == NULL) |
| vim_free(t_buf); |
| maxlen = (buflen - 6 - len) / 3; |
| } |
| if (buf == NULL) |
| { |
| do_outofmem_msg((long_u)buflen); |
| return ESC; // panic! |
| } |
| |
| // First time: blocking wait. Second time: wait up to 100ms for a |
| // terminal code to complete. |
| n = ui_inchar(buf + len, maxlen, len == 0 ? -1L : 100L, 0); |
| if (n > 0) |
| { |
| // Replace zero and CSI by a special key code. |
| n = fix_input_buffer(buf + len, n); |
| len += n; |
| waited = 0; |
| } |
| else if (len > 0) |
| ++waited; // keep track of the waiting time |
| |
| // Incomplete termcode and not timed out yet: get more characters |
| if ((n = check_termcode(1, buf, buflen, &len)) < 0 |
| && (!p_ttimeout || waited * 100L < (p_ttm < 0 ? p_tm : p_ttm))) |
| continue; |
| |
| if (n == KEYLEN_REMOVED) // key code removed |
| { |
| if (must_redraw != 0 && !need_wait_return && (State |
| & (MODE_CMDLINE | MODE_HITRETURN | MODE_ASKMORE)) == 0) |
| { |
| // Redrawing was postponed, do it now. |
| update_screen(0); |
| setcursor(); // put cursor back where it belongs |
| } |
| continue; |
| } |
| if (n > 0) // found a termcode: adjust length |
| len = n; |
| if (len == 0) // nothing typed yet |
| continue; |
| |
| // Handle modifier and/or special key code. |
| n = buf[0]; |
| if (n == K_SPECIAL) |
| { |
| n = TO_SPECIAL(buf[1], buf[2]); |
| if (buf[1] == KS_MODIFIER |
| || n == K_IGNORE |
| || (is_mouse_key(n) && n != K_LEFTMOUSE) |
| #ifdef FEAT_GUI |
| || n == K_VER_SCROLLBAR |
| || n == K_HOR_SCROLLBAR |
| #endif |
| ) |
| { |
| if (buf[1] == KS_MODIFIER) |
| mod_mask = buf[2]; |
| len -= 3; |
| if (len > 0) |
| mch_memmove(buf, buf + 3, (size_t)len); |
| continue; |
| } |
| break; |
| } |
| if (has_mbyte) |
| { |
| if (MB_BYTE2LEN(n) > len) |
| continue; // more bytes to get |
| buf[len >= buflen ? buflen - 1 : len] = NUL; |
| n = (*mb_ptr2char)(buf); |
| } |
| #ifdef UNIX |
| if (n == intr_char) |
| n = ESC; |
| #endif |
| break; |
| } |
| vim_free(buf); |
| |
| mapped_ctrl_c = save_mapped_ctrl_c; |
| return n; |
| } |
| |
| /* |
| * Get a number from the user. |
| * When "mouse_used" is not NULL allow using the mouse. |
| */ |
| int |
| get_number( |
| int colon, // allow colon to abort |
| int *mouse_used) |
| { |
| int n = 0; |
| int c; |
| int typed = 0; |
| |
| if (mouse_used != NULL) |
| *mouse_used = FALSE; |
| |
| // When not printing messages, the user won't know what to type, return a |
| // zero (as if CR was hit). |
| if (msg_silent != 0) |
| return 0; |
| |
| #ifdef USE_ON_FLY_SCROLL |
| dont_scroll = TRUE; // disallow scrolling here |
| #endif |
| ++no_mapping; |
| ++allow_keys; // no mapping here, but recognize keys |
| for (;;) |
| { |
| windgoto(msg_row, msg_col); |
| c = safe_vgetc(); |
| if (VIM_ISDIGIT(c)) |
| { |
| if (vim_append_digit_int(&n, c - '0') == FAIL) |
| return 0; |
| msg_putchar(c); |
| ++typed; |
| } |
| else if (c == K_DEL || c == K_KDEL || c == K_BS || c == Ctrl_H) |
| { |
| if (typed > 0) |
| { |
| msg_puts("\b \b"); |
| --typed; |
| } |
| n /= 10; |
| } |
| else if (mouse_used != NULL && c == K_LEFTMOUSE) |
| { |
| *mouse_used = TRUE; |
| n = mouse_row + 1; |
| break; |
| } |
| else if (n == 0 && c == ':' && colon) |
| { |
| stuffcharReadbuff(':'); |
| if (!exmode_active) |
| cmdline_row = msg_row; |
| skip_redraw = TRUE; // skip redraw once |
| do_redraw = FALSE; |
| break; |
| } |
| else if (c == Ctrl_C || c == ESC || c == 'q') |
| { |
| n = 0; |
| break; |
| } |
| else if (c == CAR || c == NL ) |
| break; |
| } |
| --no_mapping; |
| --allow_keys; |
| return n; |
| } |
| |
| /* |
| * Ask the user to enter a number. |
| * When "mouse_used" is not NULL allow using the mouse and in that case return |
| * the line number. |
| */ |
| int |
| prompt_for_number(int *mouse_used) |
| { |
| int i; |
| int save_cmdline_row; |
| int save_State; |
| |
| // When using ":silent" assume that <CR> was entered. |
| if (mouse_used != NULL) |
| msg_puts(_("Type number and <Enter> or click with the mouse (q or empty cancels): ")); |
| else |
| msg_puts(_("Type number and <Enter> (q or empty cancels): ")); |
| |
| // Set the state such that text can be selected/copied/pasted and we still |
| // get mouse events. redraw_after_callback() will not redraw if cmdline_row |
| // is zero. |
| save_cmdline_row = cmdline_row; |
| cmdline_row = 0; |
| save_State = State; |
| State = MODE_CMDLINE; |
| // May show different mouse shape. |
| setmouse(); |
| |
| i = get_number(TRUE, mouse_used); |
| if (KeyTyped) |
| { |
| // don't call wait_return() now |
| if (msg_row > 0) |
| cmdline_row = msg_row - 1; |
| need_wait_return = FALSE; |
| msg_didany = FALSE; |
| msg_didout = FALSE; |
| } |
| else |
| cmdline_row = save_cmdline_row; |
| State = save_State; |
| // May need to restore mouse shape. |
| setmouse(); |
| |
| return i; |
| } |
| |
| void |
| msgmore(long n) |
| { |
| long pn; |
| |
| if (global_busy // no messages now, wait until global is finished |
| || !messaging()) // 'lazyredraw' set, don't do messages now |
| return; |
| |
| // We don't want to overwrite another important message, but do overwrite |
| // a previous "more lines" or "fewer lines" message, so that "5dd" and |
| // then "put" reports the last action. |
| if (keep_msg != NULL && !keep_msg_more) |
| return; |
| |
| if (n > 0) |
| pn = n; |
| else |
| pn = -n; |
| |
| if (pn > p_report) |
| { |
| if (n > 0) |
| vim_snprintf(msg_buf, MSG_BUF_LEN, |
| NGETTEXT("%ld more line", "%ld more lines", pn), pn); |
| else |
| vim_snprintf(msg_buf, MSG_BUF_LEN, |
| NGETTEXT("%ld line less", "%ld fewer lines", pn), pn); |
| if (got_int) |
| vim_strcat((char_u *)msg_buf, (char_u *)_(" (Interrupted)"), |
| MSG_BUF_LEN); |
| if (msg(msg_buf)) |
| { |
| set_keep_msg((char_u *)msg_buf, 0); |
| keep_msg_more = TRUE; |
| } |
| } |
| } |
| |
| /* |
| * flush map and typeahead buffers and give a warning for an error |
| */ |
| void |
| beep_flush(void) |
| { |
| if (emsg_silent == 0) |
| { |
| flush_buffers(FLUSH_MINIMAL); |
| vim_beep(BO_ERROR); |
| } |
| } |
| |
| /* |
| * Give a warning for an error. "val" is one of the BO_ values, e.g., BO_OPER. |
| */ |
| void |
| vim_beep(unsigned val) |
| { |
| #ifdef FEAT_EVAL |
| called_vim_beep = TRUE; |
| #endif |
| |
| if (emsg_silent != 0 || in_assert_fails) |
| return; |
| |
| if (!((bo_flags & val) || (bo_flags & BO_ALL))) |
| { |
| #ifdef ELAPSED_FUNC |
| static int did_init = FALSE; |
| static elapsed_T start_tv; |
| |
| // Only beep once per half a second, otherwise a sequence of beeps |
| // would freeze Vim. |
| if (!did_init || ELAPSED_FUNC(start_tv) > 500) |
| { |
| did_init = TRUE; |
| ELAPSED_INIT(start_tv); |
| #endif |
| if (p_vb |
| #ifdef FEAT_GUI |
| // While the GUI is starting up the termcap is set for |
| // the GUI but the output still goes to a terminal. |
| && !(gui.in_use && gui.starting) |
| #endif |
| ) |
| { |
| out_str_cf(T_VB); |
| #ifdef FEAT_VTP |
| // No restore color information, refresh the screen. |
| if (has_vtp_working() != 0 |
| # ifdef FEAT_TERMGUICOLORS |
| && (p_tgc || (!p_tgc && t_colors >= 256)) |
| # endif |
| ) |
| { |
| redraw_later(UPD_CLEAR); |
| update_screen(0); |
| redrawcmd(); |
| } |
| #endif |
| } |
| else |
| out_char(BELL); |
| #ifdef ELAPSED_FUNC |
| } |
| #endif |
| } |
| |
| // When 'debug' contains "beep" produce a message. If we are sourcing |
| // a script or executing a function give the user a hint where the beep |
| // comes from. |
| if (vim_strchr(p_debug, 'e') != NULL) |
| { |
| msg_source(HL_ATTR(HLF_W)); |
| msg_attr(_("Beep!"), HL_ATTR(HLF_W)); |
| } |
| } |
| |
| /* |
| * To get the "real" home directory: |
| * - get value of $HOME |
| * For Unix: |
| * - go to that directory |
| * - do mch_dirname() to get the real name of that directory. |
| * This also works with mounts and links. |
| * Don't do this for MS-DOS, it will change the "current dir" for a drive. |
| * For Windows: |
| * This code is duplicated in init_homedir() in dosinst.c. Keep in sync! |
| */ |
| void |
| init_homedir(void) |
| { |
| char_u *var; |
| |
| // In case we are called a second time (when 'encoding' changes). |
| VIM_CLEAR(homedir); |
| |
| #ifdef VMS |
| var = mch_getenv((char_u *)"SYS$LOGIN"); |
| #else |
| var = mch_getenv((char_u *)"HOME"); |
| #endif |
| |
| #ifdef MSWIN |
| /* |
| * Typically, $HOME is not defined on Windows, unless the user has |
| * specifically defined it for Vim's sake. However, on Windows NT |
| * platforms, $HOMEDRIVE and $HOMEPATH are automatically defined for |
| * each user. Try constructing $HOME from these. |
| */ |
| if (var == NULL || *var == NUL) |
| { |
| char_u *homedrive, *homepath; |
| |
| homedrive = mch_getenv((char_u *)"HOMEDRIVE"); |
| homepath = mch_getenv((char_u *)"HOMEPATH"); |
| if (homepath == NULL || *homepath == NUL) |
| homepath = (char_u *)"\\"; |
| if (homedrive != NULL |
| && STRLEN(homedrive) + STRLEN(homepath) < MAXPATHL) |
| { |
| sprintf((char *)NameBuff, "%s%s", homedrive, homepath); |
| if (NameBuff[0] != NUL) |
| var = NameBuff; |
| } |
| } |
| |
| if (var == NULL) |
| var = mch_getenv((char_u *)"USERPROFILE"); |
| |
| /* |
| * Weird but true: $HOME may contain an indirect reference to another |
| * variable, esp. "%USERPROFILE%". Happens when $USERPROFILE isn't set |
| * when $HOME is being set. |
| */ |
| if (var != NULL && *var == '%') |
| { |
| char_u *p; |
| char_u *exp; |
| |
| p = vim_strchr(var + 1, '%'); |
| if (p != NULL) |
| { |
| vim_strncpy(NameBuff, var + 1, p - (var + 1)); |
| exp = mch_getenv(NameBuff); |
| if (exp != NULL && *exp != NUL |
| && STRLEN(exp) + STRLEN(p) < MAXPATHL) |
| { |
| vim_snprintf((char *)NameBuff, MAXPATHL, "%s%s", exp, p + 1); |
| var = NameBuff; |
| } |
| } |
| } |
| |
| if (var != NULL && *var == NUL) // empty is same as not set |
| var = NULL; |
| |
| if (enc_utf8 && var != NULL) |
| { |
| int len; |
| char_u *pp = NULL; |
| |
| // Convert from active codepage to UTF-8. Other conversions are |
| // not done, because they would fail for non-ASCII characters. |
| acp_to_enc(var, (int)STRLEN(var), &pp, &len); |
| if (pp != NULL) |
| { |
| homedir = pp; |
| return; |
| } |
| } |
| |
| /* |
| * Default home dir is C:/ |
| * Best assumption we can make in such a situation. |
| */ |
| if (var == NULL) |
| var = (char_u *)"C:/"; |
| #endif |
| |
| if (var != NULL) |
| { |
| #ifdef UNIX |
| /* |
| * Change to the directory and get the actual path. This resolves |
| * links. Don't do it when we can't return. |
| */ |
| if (mch_dirname(NameBuff, MAXPATHL) == OK |
| && mch_chdir((char *)NameBuff) == 0) |
| { |
| if (!mch_chdir((char *)var) && mch_dirname(IObuff, IOSIZE) == OK) |
| var = IObuff; |
| if (mch_chdir((char *)NameBuff) != 0) |
| emsg(_(e_cannot_go_back_to_previous_directory)); |
| } |
| #endif |
| homedir = vim_strsave(var); |
| } |
| } |
| |
| #if defined(EXITFREE) || defined(PROTO) |
| void |
| free_homedir(void) |
| { |
| vim_free(homedir); |
| } |
| |
| void |
| free_users(void) |
| { |
| ga_clear_strings(&ga_users); |
| } |
| #endif |
| |
| #if defined(MSWIN) || defined(PROTO) |
| /* |
| * Initialize $VIM and $VIMRUNTIME when 'enc' is updated. |
| */ |
| void |
| init_vimdir(void) |
| { |
| int mustfree; |
| char_u *p; |
| |
| mch_get_exe_name(); |
| |
| mustfree = FALSE; |
| didset_vim = FALSE; |
| p = vim_getenv((char_u *)"VIM", &mustfree); |
| if (mustfree) |
| vim_free(p); |
| |
| mustfree = FALSE; |
| didset_vimruntime = FALSE; |
| p = vim_getenv((char_u *)"VIMRUNTIME", &mustfree); |
| if (mustfree) |
| vim_free(p); |
| } |
| #endif |
| |
| /* |
| * Call expand_env() and store the result in an allocated string. |
| * This is not very memory efficient, this expects the result to be freed |
| * again soon. |
| */ |
| char_u * |
| expand_env_save(char_u *src) |
| { |
| return expand_env_save_opt(src, FALSE); |
| } |
| |
| /* |
| * Idem, but when "one" is TRUE handle the string as one file name, only |
| * expand "~" at the start. |
| */ |
| char_u * |
| expand_env_save_opt(char_u *src, int one) |
| { |
| char_u *p; |
| |
| p = alloc(MAXPATHL); |
| if (p != NULL) |
| expand_env_esc(src, p, MAXPATHL, FALSE, one, NULL); |
| return p; |
| } |
| |
| /* |
| * Expand environment variable with path name. |
| * "~/" is also expanded, using $HOME. For Unix "~user/" is expanded. |
| * Skips over "\ ", "\~" and "\$" (not for Win32 though). |
| * If anything fails no expansion is done and dst equals src. |
| */ |
| void |
| expand_env( |
| char_u *src, // input string e.g. "$HOME/vim.hlp" |
| char_u *dst, // where to put the result |
| int dstlen) // maximum length of the result |
| { |
| expand_env_esc(src, dst, dstlen, FALSE, FALSE, NULL); |
| } |
| |
| void |
| expand_env_esc( |
| char_u *srcp, // input string e.g. "$HOME/vim.hlp" |
| char_u *dst, // where to put the result |
| int dstlen, // maximum length of the result |
| int esc, // escape spaces in expanded variables |
| int one, // "srcp" is one file name |
| char_u *startstr) // start again after this (can be NULL) |
| { |
| char_u *src; |
| char_u *tail; |
| int c; |
| char_u *var; |
| int copy_char; |
| int mustfree; // var was allocated, need to free it later |
| int at_start = TRUE; // at start of a name |
| int startstr_len = 0; |
| |
| if (startstr != NULL) |
| startstr_len = (int)STRLEN(startstr); |
| |
| src = skipwhite(srcp); |
| --dstlen; // leave one char space for "\," |
| while (*src && dstlen > 0) |
| { |
| #ifdef FEAT_EVAL |
| // Skip over `=expr`. |
| if (src[0] == '`' && src[1] == '=') |
| { |
| size_t len; |
| |
| var = src; |
| src += 2; |
| (void)skip_expr(&src, NULL); |
| if (*src == '`') |
| ++src; |
| len = src - var; |
| if (len > (size_t)dstlen) |
| len = dstlen; |
| vim_strncpy(dst, var, len); |
| dst += len; |
| dstlen -= (int)len; |
| continue; |
| } |
| #endif |
| copy_char = TRUE; |
| if ((*src == '$' |
| #ifdef VMS |
| && at_start |
| #endif |
| ) |
| #if defined(MSWIN) |
| || *src == '%' |
| #endif |
| || (*src == '~' && at_start)) |
| { |
| mustfree = FALSE; |
| |
| /* |
| * The variable name is copied into dst temporarily, because it may |
| * be a string in read-only memory and a NUL needs to be appended. |
| */ |
| if (*src != '~') // environment var |
| { |
| tail = src + 1; |
| var = dst; |
| c = dstlen - 1; |
| |
| #ifdef UNIX |
| // Unix has ${var-name} type environment vars |
| if (*tail == '{' && !vim_isIDc('{')) |
| { |
| tail++; // ignore '{' |
| while (c-- > 0 && *tail && *tail != '}') |
| *var++ = *tail++; |
| } |
| else |
| #endif |
| { |
| while (c-- > 0 && *tail != NUL && ((vim_isIDc(*tail)) |
| #if defined(MSWIN) |
| || (*src == '%' && *tail != '%') |
| #endif |
| )) |
| *var++ = *tail++; |
| } |
| |
| #if defined(MSWIN) || defined(UNIX) |
| # ifdef UNIX |
| if (src[1] == '{' && *tail != '}') |
| # else |
| if (*src == '%' && *tail != '%') |
| # endif |
| var = NULL; |
| else |
| { |
| # ifdef UNIX |
| if (src[1] == '{') |
| # else |
| if (*src == '%') |
| #endif |
| ++tail; |
| #endif |
| *var = NUL; |
| var = vim_getenv(dst, &mustfree); |
| #if defined(MSWIN) || defined(UNIX) |
| } |
| #endif |
| } |
| // home directory |
| else if ( src[1] == NUL |
| || vim_ispathsep(src[1]) |
| || vim_strchr((char_u *)" ,\t\n", src[1]) != NULL) |
| { |
| var = homedir; |
| tail = src + 1; |
| } |
| else // user directory |
| { |
| #if defined(UNIX) || (defined(VMS) && defined(USER_HOME)) |
| /* |
| * Copy ~user to dst[], so we can put a NUL after it. |
| */ |
| tail = src; |
| var = dst; |
| c = dstlen - 1; |
| while ( c-- > 0 |
| && *tail |
| && vim_isfilec(*tail) |
| && !vim_ispathsep(*tail)) |
| *var++ = *tail++; |
| *var = NUL; |
| # ifdef UNIX |
| /* |
| * If the system supports getpwnam(), use it. |
| * Otherwise, or if getpwnam() fails, the shell is used to |
| * expand ~user. This is slower and may fail if the shell |
| * does not support ~user (old versions of /bin/sh). |
| */ |
| # if defined(HAVE_GETPWNAM) && defined(HAVE_PWD_H) |
| { |
| // Note: memory allocated by getpwnam() is never freed. |
| // Calling endpwent() apparently doesn't help. |
| struct passwd *pw = (*dst == NUL) |
| ? NULL : getpwnam((char *)dst + 1); |
| |
| var = (pw == NULL) ? NULL : (char_u *)pw->pw_dir; |
| } |
| if (var == NULL) |
| # endif |
| { |
| expand_T xpc; |
| |
| ExpandInit(&xpc); |
| xpc.xp_context = EXPAND_FILES; |
| var = ExpandOne(&xpc, dst, NULL, |
| WILD_ADD_SLASH|WILD_SILENT, WILD_EXPAND_FREE); |
| mustfree = TRUE; |
| } |
| |
| # else // !UNIX, thus VMS |
| /* |
| * USER_HOME is a comma-separated list of |
| * directories to search for the user account in. |
| */ |
| { |
| char_u test[MAXPATHL], paths[MAXPATHL]; |
| char_u *path, *next_path, *ptr; |
| stat_T st; |
| |
| STRCPY(paths, USER_HOME); |
| next_path = paths; |
| while (*next_path) |
| { |
| for (path = next_path; *next_path && *next_path != ','; |
| next_path++); |
| if (*next_path) |
| *next_path++ = NUL; |
| STRCPY(test, path); |
| STRCAT(test, "/"); |
| STRCAT(test, dst + 1); |
| if (mch_stat(test, &st) == 0) |
| { |
| var = alloc(STRLEN(test) + 1); |
| STRCPY(var, test); |
| mustfree = TRUE; |
| break; |
| } |
| } |
| } |
| # endif // UNIX |
| #else |
| // cannot expand user's home directory, so don't try |
| var = NULL; |
| tail = (char_u *)""; // for gcc |
| #endif // UNIX || VMS |
| } |
| |
| #ifdef BACKSLASH_IN_FILENAME |
| // If 'shellslash' is set change backslashes to forward slashes. |
| // Can't use slash_adjust(), p_ssl may be set temporarily. |
| if (p_ssl && var != NULL && vim_strchr(var, '\\') != NULL) |
| { |
| char_u *p = vim_strsave(var); |
| |
| if (p != NULL) |
| { |
| if (mustfree) |
| vim_free(var); |
| var = p; |
| mustfree = TRUE; |
| forward_slash(var); |
| } |
| } |
| #endif |
| |
| // If "var" contains white space, escape it with a backslash. |
| // Required for ":e ~/tt" when $HOME includes a space. |
| if (esc && var != NULL && vim_strpbrk(var, (char_u *)" \t") != NULL) |
| { |
| char_u *p = vim_strsave_escaped(var, (char_u *)" \t"); |
| |
| if (p != NULL) |
| { |
| if (mustfree) |
| vim_free(var); |
| var = p; |
| mustfree = TRUE; |
| } |
| } |
| |
| if (var != NULL && *var != NUL |
| && (STRLEN(var) + STRLEN(tail) + 1 < (unsigned)dstlen)) |
| { |
| STRCPY(dst, var); |
| dstlen -= (int)STRLEN(var); |
| c = (int)STRLEN(var); |
| // if var[] ends in a path separator and tail[] starts |
| // with it, skip a character |
| if (after_pathsep(dst, dst + c) |
| #if defined(BACKSLASH_IN_FILENAME) || defined(AMIGA) |
| && dst[c - 1] != ':' |
| #endif |
| && vim_ispathsep(*tail)) |
| ++tail; |
| dst += c; |
| src = tail; |
| copy_char = FALSE; |
| } |
| if (mustfree) |
| vim_free(var); |
| } |
| |
| if (copy_char) // copy at least one char |
| { |
| /* |
| * Recognize the start of a new name, for '~'. |
| * Don't do this when "one" is TRUE, to avoid expanding "~" in |
| * ":edit foo ~ foo". |
| */ |
| at_start = FALSE; |
| if (src[0] == '\\' && src[1] != NUL) |
| { |
| *dst++ = *src++; |
| --dstlen; |
| } |
| else if ((src[0] == ' ' || src[0] == ',') && !one) |
| at_start = TRUE; |
| if (dstlen > 0) |
| { |
| *dst++ = *src++; |
| --dstlen; |
| |
| if (startstr != NULL && src - startstr_len >= srcp |
| && STRNCMP(src - startstr_len, startstr, |
| startstr_len) == 0) |
| at_start = TRUE; |
| } |
| } |
| |
| } |
| *dst = NUL; |
| } |
| |
| /* |
| * If the string between "p" and "pend" ends in "name/", return "pend" minus |
| * the length of "name/". Otherwise return "pend". |
| */ |
| static char_u * |
| remove_tail(char_u *p, char_u *pend, char_u *name) |
| { |
| int len = (int)STRLEN(name) + 1; |
| char_u *newend = pend - len; |
| |
| if (newend >= p |
| && fnamencmp(newend, name, len - 1) == 0 |
| && (newend == p || after_pathsep(p, newend))) |
| return newend; |
| return pend; |
| } |
| |
| /* |
| * Check if the directory "vimdir/<version>" or "vimdir/runtime" exists. |
| * Return NULL if not, return its name in allocated memory otherwise. |
| */ |
| static char_u * |
| vim_version_dir(char_u *vimdir) |
| { |
| char_u *p; |
| |
| if (vimdir == NULL || *vimdir == NUL) |
| return NULL; |
| p = concat_fnames(vimdir, (char_u *)VIM_VERSION_NODOT, TRUE); |
| if (p != NULL && mch_isdir(p)) |
| return p; |
| vim_free(p); |
| p = concat_fnames(vimdir, (char_u *)RUNTIME_DIRNAME, TRUE); |
| if (p != NULL && mch_isdir(p)) |
| { |
| char_u *fname = concat_fnames(p, (char_u *)"defaults.vim", TRUE); |
| |
| // Check that "defaults.vim" exists in this directory, to avoid picking |
| // up a stray "runtime" directory, it would make many tests fail in |
| // mysterious ways. |
| if (fname != NULL) |
| { |
| int exists = file_is_readable(fname); |
| |
| vim_free(fname); |
| if (exists) |
| return p; |
| } |
| } |
| vim_free(p); |
| return NULL; |
| } |
| |
| /* |
| * Vim's version of getenv(). |
| * Special handling of $HOME, $VIM and $VIMRUNTIME. |
| * Also does ACP to 'enc' conversion for Win32. |
| * "mustfree" is set to TRUE when the returned string is allocated. It must be |
| * initialized to FALSE by the caller. |
| */ |
| char_u * |
| vim_getenv(char_u *name, int *mustfree) |
| { |
| char_u *p = NULL; |
| char_u *pend; |
| int vimruntime; |
| #ifdef MSWIN |
| WCHAR *wn, *wp; |
| |
| // use "C:/" when $HOME is not set |
| if (STRCMP(name, "HOME") == 0) |
| return homedir; |
| |
| // Use Wide function |
| wn = enc_to_utf16(name, NULL); |
| if (wn == NULL) |
| return NULL; |
| |
| wp = _wgetenv(wn); |
| vim_free(wn); |
| |
| if (wp != NULL && *wp == NUL) // empty is the same as not set |
| wp = NULL; |
| |
| if (wp != NULL) |
| { |
| p = utf16_to_enc(wp, NULL); |
| if (p == NULL) |
| return NULL; |
| |
| *mustfree = TRUE; |
| return p; |
| } |
| #else |
| p = mch_getenv(name); |
| if (p != NULL && *p == NUL) // empty is the same as not set |
| p = NULL; |
| |
| if (p != NULL) |
| return p; |
| |
| # ifdef __HAIKU__ |
| // special handling for user settings directory... |
| if (STRCMP(name, "BE_USER_SETTINGS") == 0) |
| { |
| static char userSettingsPath[MAXPATHL]; |
| |
| if (find_directory(B_USER_SETTINGS_DIRECTORY, 0, false, |
| userSettingsPath, MAXPATHL) == B_OK) |
| return (char_u *)userSettingsPath; |
| else |
| return NULL; |
| } |
| # endif |
| #endif |
| |
| // handling $VIMRUNTIME and $VIM is below, bail out if it's another name. |
| vimruntime = (STRCMP(name, "VIMRUNTIME") == 0); |
| if (!vimruntime && STRCMP(name, "VIM") != 0) |
| return NULL; |
| |
| /* |
| * When expanding $VIMRUNTIME fails, try using $VIM/vim<version> or $VIM. |
| * Don't do this when default_vimruntime_dir is non-empty. |
| */ |
| if (vimruntime |
| #ifdef HAVE_PATHDEF |
| && *default_vimruntime_dir == NUL |
| #endif |
| ) |
| { |
| #ifdef MSWIN |
| // Use Wide function |
| wp = _wgetenv(L"VIM"); |
| if (wp != NULL && *wp == NUL) // empty is the same as not set |
| wp = NULL; |
| if (wp != NULL) |
| { |
| char_u *q = utf16_to_enc(wp, NULL); |
| if (q != NULL) |
| { |
| p = vim_version_dir(q); |
| *mustfree = TRUE; |
| if (p == NULL) |
| p = q; |
| } |
| } |
| #else |
| p = mch_getenv((char_u *)"VIM"); |
| if (p != NULL && *p == NUL) // empty is the same as not set |
| p = NULL; |
| if (p != NULL) |
| { |
| p = vim_version_dir(p); |
| if (p != NULL) |
| *mustfree = TRUE; |
| else |
| p = mch_getenv((char_u *)"VIM"); |
| } |
| #endif |
| } |
| |
| /* |
| * When expanding $VIM or $VIMRUNTIME fails, try using: |
| * - the directory name from 'helpfile' (unless it contains '$') |
| * - the executable name from argv[0] |
| */ |
| if (p == NULL) |
| { |
| if (p_hf != NULL && vim_strchr(p_hf, '$') == NULL) |
| p = p_hf; |
| #ifdef USE_EXE_NAME |
| /* |
| * Use the name of the executable, obtained from argv[0]. |
| */ |
| else |
| p = exe_name; |
| #endif |
| if (p != NULL) |
| { |
| // remove the file name |
| pend = gettail(p); |
| |
| // remove "doc/" from 'helpfile', if present |
| if (p == p_hf) |
| pend = remove_tail(p, pend, (char_u *)"doc"); |
| |
| #ifdef USE_EXE_NAME |
| # ifdef MACOS_X |
| // remove "MacOS" from exe_name and add "Resources/vim" |
| if (p == exe_name) |
| { |
| char_u *pend1; |
| char_u *pnew; |
| |
| pend1 = remove_tail(p, pend, (char_u *)"MacOS"); |
| if (pend1 != pend) |
| { |
| pnew = alloc(pend1 - p + 15); |
| if (pnew != NULL) |
| { |
| STRNCPY(pnew, p, (pend1 - p)); |
| STRCPY(pnew + (pend1 - p), "Resources/vim"); |
| p = pnew; |
| pend = p + STRLEN(p); |
| } |
| } |
| } |
| # endif |
| // remove "src/" from exe_name, if present |
| if (p == exe_name) |
| pend = remove_tail(p, pend, (char_u *)"src"); |
| #endif |
| |
| // for $VIM, remove "runtime/" or "vim54/", if present |
| if (!vimruntime) |
| { |
| pend = remove_tail(p, pend, (char_u *)RUNTIME_DIRNAME); |
| pend = remove_tail(p, pend, (char_u *)VIM_VERSION_NODOT); |
| } |
| |
| // remove trailing path separator |
| if (pend > p && after_pathsep(p, pend)) |
| --pend; |
| |
| #ifdef MACOS_X |
| if (p == exe_name || p == p_hf) |
| #endif |
| // check that the result is a directory name |
| p = vim_strnsave(p, pend - p); |
| |
| if (p != NULL && !mch_isdir(p)) |
| VIM_CLEAR(p); |
| else |
| { |
| #ifdef USE_EXE_NAME |
| // may add "/vim54" or "/runtime" if it exists |
| if (vimruntime && (pend = vim_version_dir(p)) != NULL) |
| { |
| vim_free(p); |
| p = pend; |
| } |
| #endif |
| *mustfree = TRUE; |
| } |
| } |
| } |
| |
| #ifdef HAVE_PATHDEF |
| // When there is a pathdef.c file we can use default_vim_dir and |
| // default_vimruntime_dir |
| if (p == NULL) |
| { |
| // Only use default_vimruntime_dir when it is not empty |
| if (vimruntime && *default_vimruntime_dir != NUL) |
| { |
| p = default_vimruntime_dir; |
| *mustfree = FALSE; |
| } |
| else if (*default_vim_dir != NUL) |
| { |
| if (vimruntime && (p = vim_version_dir(default_vim_dir)) != NULL) |
| *mustfree = TRUE; |
| else |
| { |
| p = default_vim_dir; |
| *mustfree = FALSE; |
| } |
| } |
| } |
| #endif |
| |
| /* |
| * Set the environment variable, so that the new value can be found fast |
| * next time, and others can also use it (e.g. Perl). |
| */ |
| if (p != NULL) |
| { |
| if (vimruntime) |
| { |
| vim_setenv((char_u *)"VIMRUNTIME", p); |
| didset_vimruntime = TRUE; |
| } |
| else |
| { |
| vim_setenv((char_u *)"VIM", p); |
| didset_vim = TRUE; |
| } |
| } |
| return p; |
| } |
| |
| void |
| vim_unsetenv(char_u *var) |
| { |
| #ifdef HAVE_UNSETENV |
| unsetenv((char *)var); |
| #else |
| vim_setenv(var, (char_u *)""); |
| #endif |
| } |
| |
| /* |
| * Removes environment variable "name" and take care of side effects. |
| */ |
| void |
| vim_unsetenv_ext(char_u *var) |
| { |
| vim_unsetenv(var); |
| |
| // "homedir" is not cleared, keep using the old value until $HOME is set. |
| if (STRICMP(var, "VIM") == 0) |
| didset_vim = FALSE; |
| else if (STRICMP(var, "VIMRUNTIME") == 0) |
| didset_vimruntime = FALSE; |
| } |
| |
| #if defined(FEAT_EVAL) || defined(PROTO) |
| /* |
| * Set environment variable "name" and take care of side effects. |
| */ |
| void |
| vim_setenv_ext(char_u *name, char_u *val) |
| { |
| vim_setenv(name, val); |
| if (STRICMP(name, "HOME") == 0) |
| init_homedir(); |
| else if (didset_vim && STRICMP(name, "VIM") == 0) |
| didset_vim = FALSE; |
| else if (didset_vimruntime && STRICMP(name, "VIMRUNTIME") == 0) |
| didset_vimruntime = FALSE; |
| } |
| #endif |
| |
| /* |
| * Our portable version of setenv. |
| */ |
| void |
| vim_setenv(char_u *name, char_u *val) |
| { |
| #ifdef HAVE_SETENV |
| mch_setenv((char *)name, (char *)val, 1); |
| #else |
| char_u *envbuf; |
| |
| /* |
| * Putenv does not copy the string, it has to remain |
| * valid. The allocated memory will never be freed. |
| */ |
| envbuf = alloc(STRLEN(name) + STRLEN(val) + 2); |
| if (envbuf != NULL) |
| { |
| sprintf((char *)envbuf, "%s=%s", name, val); |
| putenv((char *)envbuf); |
| } |
| #endif |
| #ifdef FEAT_GETTEXT |
| /* |
| * When setting $VIMRUNTIME adjust the directory to find message |
| * translations to $VIMRUNTIME/lang. |
| */ |
| if (*val != NUL && STRICMP(name, "VIMRUNTIME") == 0) |
| { |
| char_u *buf = concat_str(val, (char_u *)"/lang"); |
| |
| if (buf != NULL) |
| { |
| bindtextdomain(VIMPACKAGE, (char *)buf); |
| vim_free(buf); |
| } |
| } |
| #endif |
| } |
| |
| /* |
| * Function given to ExpandGeneric() to obtain an environment variable name. |
| */ |
| char_u * |
| get_env_name( |
| expand_T *xp UNUSED, |
| int idx) |
| { |
| #if defined(AMIGA) |
| // No environ[] on the Amiga. |
| return NULL; |
| #else |
| # ifndef __WIN32__ |
| // Borland C++ 5.2 has this in a header file. |
| extern char **environ; |
| # endif |
| char_u *str; |
| int n; |
| |
| str = (char_u *)environ[idx]; |
| if (str == NULL) |
| return NULL; |
| |
| for (n = 0; n < EXPAND_BUF_LEN - 1; ++n) |
| { |
| if (str[n] == '=' || str[n] == NUL) |
| break; |
| xp->xp_buf[n] = str[n]; |
| } |
| xp->xp_buf[n] = NUL; |
| return xp->xp_buf; |
| #endif |
| } |
| |
| /* |
| * Add a user name to the list of users in ga_users. |
| * Do nothing if user name is NULL or empty. |
| */ |
| static void |
| add_user(char_u *user, int need_copy) |
| { |
| char_u *user_copy = (user != NULL && need_copy) |
| ? vim_strsave(user) : user; |
| |
| if (user_copy == NULL || *user_copy == NUL || ga_grow(&ga_users, 1) == FAIL) |
| { |
| if (need_copy) |
| vim_free(user_copy); |
| return; |
| } |
| ((char_u **)(ga_users.ga_data))[ga_users.ga_len++] = user_copy; |
| } |
| |
| /* |
| * Find all user names for user completion. |
| * Done only once and then cached. |
| */ |
| static void |
| init_users(void) |
| { |
| static int lazy_init_done = FALSE; |
| |
| if (lazy_init_done) |
| return; |
| |
| lazy_init_done = TRUE; |
| ga_init2(&ga_users, sizeof(char_u *), 20); |
| |
| # if defined(HAVE_GETPWENT) && defined(HAVE_PWD_H) |
| { |
| struct passwd* pw; |
| |
| setpwent(); |
| while ((pw = getpwent()) != NULL) |
| add_user((char_u *)pw->pw_name, TRUE); |
| endpwent(); |
| } |
| # elif defined(MSWIN) |
| { |
| DWORD nusers = 0, ntotal = 0, i; |
| PUSER_INFO_0 uinfo; |
| |
| if (NetUserEnum(NULL, 0, 0, (LPBYTE *) &uinfo, MAX_PREFERRED_LENGTH, |
| &nusers, &ntotal, NULL) == NERR_Success) |
| { |
| for (i = 0; i < nusers; i++) |
| add_user(utf16_to_enc(uinfo[i].usri0_name, NULL), FALSE); |
| |
| NetApiBufferFree(uinfo); |
| } |
| } |
| # endif |
| # if defined(HAVE_GETPWNAM) |
| { |
| char_u *user_env = mch_getenv((char_u *)"USER"); |
| |
| // The $USER environment variable may be a valid remote user name (NIS, |
| // LDAP) not already listed by getpwent(), as getpwent() only lists |
| // local user names. If $USER is not already listed, check whether it |
| // is a valid remote user name using getpwnam() and if it is, add it to |
| // the list of user names. |
| |
| if (user_env != NULL && *user_env != NUL) |
| { |
| int i; |
| |
| for (i = 0; i < ga_users.ga_len; i++) |
| { |
| char_u *local_user = ((char_u **)ga_users.ga_data)[i]; |
| |
| if (STRCMP(local_user, user_env) == 0) |
| break; |
| } |
| |
| if (i == ga_users.ga_len) |
| { |
| struct passwd *pw = getpwnam((char *)user_env); |
| |
| if (pw != NULL) |
| add_user((char_u *)pw->pw_name, TRUE); |
| } |
| } |
| } |
| # endif |
| } |
| |
| /* |
| * Function given to ExpandGeneric() to obtain user names. |
| */ |
| char_u* |
| get_users(expand_T *xp UNUSED, int idx) |
| { |
| init_users(); |
| if (idx < ga_users.ga_len) |
| return ((char_u **)ga_users.ga_data)[idx]; |
| return NULL; |
| } |
| |
| /* |
| * Check whether name matches a user name. Return: |
| * 0 if name does not match any user name. |
| * 1 if name partially matches the beginning of a user name. |
| * 2 is name fully matches a user name. |
| */ |
| int |
| match_user(char_u *name) |
| { |
| int i; |
| int n = (int)STRLEN(name); |
| int result = 0; |
| |
| init_users(); |
| for (i = 0; i < ga_users.ga_len; i++) |
| { |
| if (STRCMP(((char_u **)ga_users.ga_data)[i], name) == 0) |
| return 2; // full match |
| if (STRNCMP(((char_u **)ga_users.ga_data)[i], name, n) == 0) |
| result = 1; // partial match |
| } |
| return result; |
| } |
| |
| static void |
| prepare_to_exit(void) |
| { |
| #if defined(SIGHUP) && defined(SIG_IGN) |
| // Ignore SIGHUP, because a dropped connection causes a read error, which |
| // makes Vim exit and then handling SIGHUP causes various reentrance |
| // problems. |
| mch_signal(SIGHUP, SIG_IGN); |
| #endif |
| |
| #ifdef FEAT_GUI |
| if (gui.in_use) |
| { |
| gui.dying = TRUE; |
| out_trash(); // trash any pending output |
| } |
| else |
| #endif |
| { |
| windgoto((int)Rows - 1, 0); |
| |
| /* |
| * Switch terminal mode back now, so messages end up on the "normal" |
| * screen (if there are two screens). |
| */ |
| settmode(TMODE_COOK); |
| stoptermcap(); |
| out_flush(); |
| } |
| } |
| |
| /* |
| * Preserve files and exit. |
| * When called IObuff must contain a message. |
| * NOTE: This may be called from deathtrap() in a signal handler, avoid unsafe |
| * functions, such as allocating memory. |
| */ |
| void |
| preserve_exit(void) |
| { |
| buf_T *buf; |
| |
| prepare_to_exit(); |
| |
| // Setting this will prevent free() calls. That avoids calling free() |
| // recursively when free() was invoked with a bad pointer. |
| really_exiting = TRUE; |
| |
| out_str(IObuff); |
| screen_start(); // don't know where cursor is now |
| out_flush(); |
| |
| ml_close_notmod(); // close all not-modified buffers |
| |
| FOR_ALL_BUFFERS(buf) |
| { |
| if (buf->b_ml.ml_mfp != NULL && buf->b_ml.ml_mfp->mf_fname != NULL) |
| { |
| OUT_STR("Vim: preserving files...\r\n"); |
| screen_start(); // don't know where cursor is now |
| out_flush(); |
| ml_sync_all(FALSE, FALSE); // preserve all swap files |
| break; |
| } |
| } |
| |
| ml_close_all(FALSE); // close all memfiles, without deleting |
| |
| OUT_STR("Vim: Finished.\r\n"); |
| |
| getout(1); |
| } |
| |
| /* |
| * Check for CTRL-C pressed, but only once in a while. |
| * Should be used instead of ui_breakcheck() for functions that check for |
| * each line in the file. Calling ui_breakcheck() each time takes too much |
| * time, because it can be a system call. |
| */ |
| |
| #ifndef BREAKCHECK_SKIP |
| # define BREAKCHECK_SKIP 1000 |
| #endif |
| |
| static int breakcheck_count = 0; |
| |
| void |
| line_breakcheck(void) |
| { |
| if (++breakcheck_count >= BREAKCHECK_SKIP) |
| { |
| breakcheck_count = 0; |
| ui_breakcheck(); |
| } |
| } |
| |
| /* |
| * Like line_breakcheck() but check 10 times less often. |
| */ |
| void |
| fast_breakcheck(void) |
| { |
| if (++breakcheck_count >= BREAKCHECK_SKIP * 10) |
| { |
| breakcheck_count = 0; |
| ui_breakcheck(); |
| } |
| } |
| |
| # if defined(FEAT_SPELL) || defined(PROTO) |
| /* |
| * Like line_breakcheck() but check 100 times less often. |
| */ |
| void |
| veryfast_breakcheck(void) |
| { |
| if (++breakcheck_count >= BREAKCHECK_SKIP * 100) |
| { |
| breakcheck_count = 0; |
| ui_breakcheck(); |
| } |
| } |
| #endif |
| |
| #if defined(VIM_BACKTICK) || defined(FEAT_EVAL) \ |
| || (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \ |
| || defined(PROTO) |
| |
| #ifndef SEEK_SET |
| # define SEEK_SET 0 |
| #endif |
| #ifndef SEEK_END |
| # define SEEK_END 2 |
| #endif |
| |
| /* |
| * Get the stdout of an external command. |
| * If "ret_len" is NULL replace NUL characters with NL. When "ret_len" is not |
| * NULL store the length there. |
| * Returns an allocated string, or NULL for error. |
| */ |
| char_u * |
| get_cmd_output( |
| char_u *cmd, |
| char_u *infile, // optional input file name |
| int flags, // can be SHELL_SILENT |
| int *ret_len) |
| { |
| char_u *tempname; |
| char_u *command; |
| char_u *buffer = NULL; |
| int len; |
| int i = 0; |
| FILE *fd; |
| |
| if (check_restricted() || check_secure()) |
| return NULL; |
| |
| // get a name for the temp file |
| if ((tempname = vim_tempname('o', FALSE)) == NULL) |
| { |
| emsg(_(e_cant_get_temp_file_name)); |
| return NULL; |
| } |
| |
| // Add the redirection stuff |
| command = make_filter_cmd(cmd, infile, tempname); |
| if (command == NULL) |
| goto done; |
| |
| /* |
| * Call the shell to execute the command (errors are ignored). |
| * Don't check timestamps here. |
| */ |
| ++no_check_timestamps; |
| call_shell(command, SHELL_DOOUT | SHELL_EXPAND | flags); |
| --no_check_timestamps; |
| |
| vim_free(command); |
| |
| /* |
| * read the names from the file into memory |
| */ |
| # ifdef VMS |
| // created temporary file is not always readable as binary |
| fd = mch_fopen((char *)tempname, "r"); |
| # else |
| fd = mch_fopen((char *)tempname, READBIN); |
| # endif |
| |
| // Not being able to seek means we can't read the file. |
| if (fd == NULL |
| || fseek(fd, 0L, SEEK_END) == -1 |
| || (len = ftell(fd)) == -1 // get size of temp file |
| || fseek(fd, 0L, SEEK_SET) == -1) // back to the start |
| { |
| semsg(_(e_cannot_read_from_str_2), tempname); |
| if (fd != NULL) |
| fclose(fd); |
| goto done; |
| } |
| |
| buffer = alloc(len + 1); |
| if (buffer != NULL) |
| i = (int)fread((char *)buffer, (size_t)1, (size_t)len, fd); |
| fclose(fd); |
| mch_remove(tempname); |
| if (buffer == NULL) |
| goto done; |
| #ifdef VMS |
| len = i; // VMS doesn't give us what we asked for... |
| #endif |
| if (i != len) |
| { |
| semsg(_(e_cant_read_file_str), tempname); |
| VIM_CLEAR(buffer); |
| } |
| else if (ret_len == NULL) |
| { |
| // Change NUL into SOH, otherwise the string is truncated. |
| for (i = 0; i < len; ++i) |
| if (buffer[i] == NUL) |
| buffer[i] = 1; |
| |
| buffer[len] = NUL; // make sure the buffer is terminated |
| } |
| else |
| *ret_len = len; |
| |
| done: |
| vim_free(tempname); |
| return buffer; |
| } |
| |
| # if defined(FEAT_EVAL) || defined(PROTO) |
| |
| static void |
| get_cmd_output_as_rettv( |
| typval_T *argvars, |
| typval_T *rettv, |
| int retlist) |
| { |
| char_u *res = NULL; |
| char_u *p; |
| char_u *infile = NULL; |
| int err = FALSE; |
| FILE *fd; |
| list_T *list = NULL; |
| int flags = SHELL_SILENT; |
| |
| rettv->v_type = VAR_STRING; |
| rettv->vval.v_string = NULL; |
| if (check_restricted() || check_secure()) |
| goto errret; |
| |
| if (in_vim9script() |
| && (check_for_string_arg(argvars, 0) == FAIL |
| || check_for_opt_string_or_number_or_list_arg(argvars, 1) |
| == FAIL)) |
| return; |
| |
| if (argvars[1].v_type != VAR_UNKNOWN) |
| { |
| /* |
| * Write the text to a temp file, to be used for input of the shell |
| * command. |
| */ |
| if ((infile = vim_tempname('i', TRUE)) == NULL) |
| { |
| emsg(_(e_cant_get_temp_file_name)); |
| goto errret; |
| } |
| |
| fd = mch_fopen((char *)infile, WRITEBIN); |
| if (fd == NULL) |
| { |
| semsg(_(e_cant_open_file_str), infile); |
| goto errret; |
| } |
| if (argvars[1].v_type == VAR_NUMBER) |
| { |
| linenr_T lnum; |
| buf_T *buf; |
| |
| buf = buflist_findnr(argvars[1].vval.v_number); |
| if (buf == NULL) |
| { |
| semsg(_(e_buffer_nr_does_not_exist), argvars[1].vval.v_number); |
| fclose(fd); |
| goto errret; |
| } |
| |
| for (lnum = 1; lnum <= buf->b_ml.ml_line_count; lnum++) |
| { |
| for (p = ml_get_buf(buf, lnum, FALSE); *p != NUL; ++p) |
| if (putc(*p == '\n' ? NUL : *p, fd) == EOF) |
| { |
| err = TRUE; |
| break; |
| } |
| if (putc(NL, fd) == EOF) |
| { |
| err = TRUE; |
| break; |
| } |
| } |
| } |
| else if (argvars[1].v_type == VAR_LIST) |
| { |
| if (write_list(fd, argvars[1].vval.v_list, TRUE) == FAIL) |
| err = TRUE; |
| } |
| else |
| { |
| size_t len; |
| char_u buf[NUMBUFLEN]; |
| |
| p = tv_get_string_buf_chk(&argvars[1], buf); |
| if (p == NULL) |
| { |
| fclose(fd); |
| goto errret; // type error; errmsg already given |
| } |
| len = STRLEN(p); |
| if (len > 0 && fwrite(p, len, 1, fd) != 1) |
| err = TRUE; |
| } |
| if (fclose(fd) != 0) |
| err = TRUE; |
| if (err) |
| { |
| emsg(_(e_error_writing_temp_file)); |
| goto errret; |
| } |
| } |
| |
| // Omit SHELL_COOKED when invoked with ":silent". Avoids that the shell |
| // echoes typeahead, that messes up the display. |
| if (!msg_silent) |
| flags += SHELL_COOKED; |
| |
| if (retlist) |
| { |
| int len; |
| listitem_T *li; |
| char_u *s = NULL; |
| char_u *start; |
| char_u *end; |
| int i; |
| |
| res = get_cmd_output(tv_get_string(&argvars[0]), infile, flags, &len); |
| if (res == NULL) |
| goto errret; |
| |
| list = list_alloc(); |
| if (list == NULL) |
| goto errret; |
| |
| for (i = 0; i < len; ++i) |
| { |
| start = res + i; |
| while (i < len && res[i] != NL) |
| ++i; |
| end = res + i; |
| |
| s = alloc(end - start + 1); |
| if (s == NULL) |
| goto errret; |
| |
| for (p = s; start < end; ++p, ++start) |
| *p = *start == NUL ? NL : *start; |
| *p = NUL; |
| |
| li = listitem_alloc(); |
| if (li == NULL) |
| { |
| vim_free(s); |
| goto errret; |
| } |
| li->li_tv.v_type = VAR_STRING; |
| li->li_tv.v_lock = 0; |
| li->li_tv.vval.v_string = s; |
| list_append(list, li); |
| } |
| |
| rettv_list_set(rettv, list); |
| list = NULL; |
| } |
| else |
| { |
| res = get_cmd_output(tv_get_string(&argvars[0]), infile, flags, NULL); |
| #ifdef USE_CRNL |
| // translate <CR><NL> into <NL> |
| if (res != NULL) |
| { |
| char_u *s, *d; |
| |
| d = res; |
| for (s = res; *s; ++s) |
| { |
| if (s[0] == CAR && s[1] == NL) |
| ++s; |
| *d++ = *s; |
| } |
| *d = NUL; |
| } |
| #endif |
| rettv->vval.v_string = res; |
| res = NULL; |
| } |
| |
| errret: |
| if (infile != NULL) |
| { |
| mch_remove(infile); |
| vim_free(infile); |
| } |
| if (res != NULL) |
| vim_free(res); |
| if (list != NULL) |
| list_free(list); |
| } |
| |
| /* |
| * "system()" function |
| */ |
| void |
| f_system(typval_T *argvars, typval_T *rettv) |
| { |
| get_cmd_output_as_rettv(argvars, rettv, FALSE); |
| } |
| |
| /* |
| * "systemlist()" function |
| */ |
| void |
| f_systemlist(typval_T *argvars, typval_T *rettv) |
| { |
| get_cmd_output_as_rettv(argvars, rettv, TRUE); |
| } |
| # endif // FEAT_EVAL |
| |
| #endif |
| |
| /* |
| * Return TRUE when need to go to Insert mode because of 'insertmode'. |
| * Don't do this when still processing a command or a mapping. |
| * Don't do this when inside a ":normal" command. |
| */ |
| int |
| goto_im(void) |
| { |
| return (p_im && stuff_empty() && typebuf_typed()); |
| } |
| |
| /* |
| * Returns the isolated name of the shell in allocated memory: |
| * - Skip beyond any path. E.g., "/usr/bin/csh -f" -> "csh -f". |
| * - Remove any argument. E.g., "csh -f" -> "csh". |
| * But don't allow a space in the path, so that this works: |
| * "/usr/bin/csh --rcfile ~/.cshrc" |
| * But don't do that for Windows, it's common to have a space in the path. |
| * Returns NULL when out of memory. |
| */ |
| char_u * |
| get_isolated_shell_name(void) |
| { |
| char_u *p; |
| |
| #ifdef MSWIN |
| p = gettail(p_sh); |
| p = vim_strnsave(p, skiptowhite(p) - p); |
| #else |
| p = skiptowhite(p_sh); |
| if (*p == NUL) |
| { |
| // No white space, use the tail. |
| p = vim_strsave(gettail(p_sh)); |
| } |
| else |
| { |
| char_u *p1, *p2; |
| |
| // Find the last path separator before the space. |
| p1 = p_sh; |
| for (p2 = p_sh; p2 < p; MB_PTR_ADV(p2)) |
| if (vim_ispathsep(*p2)) |
| p1 = p2 + 1; |
| p = vim_strnsave(p1, p - p1); |
| } |
| #endif |
| return p; |
| } |
| |
| /* |
| * Check if the "://" of a URL is at the pointer, return URL_SLASH. |
| * Also check for ":\\", which MS Internet Explorer accepts, return |
| * URL_BACKSLASH. |
| */ |
| int |
| path_is_url(char_u *p) |
| { |
| if (STRNCMP(p, "://", (size_t)3) == 0) |
| return URL_SLASH; |
| else if (STRNCMP(p, ":\\\\", (size_t)3) == 0) |
| return URL_BACKSLASH; |
| return 0; |
| } |
| |
| /* |
| * Check if "fname" starts with "name://" or "name:\\". |
| * Return URL_SLASH for "name://", URL_BACKSLASH for "name:\\". |
| * Return zero otherwise. |
| */ |
| int |
| path_with_url(char_u *fname) |
| { |
| char_u *p; |
| |
| // We accept alphabetic characters and a dash in scheme part. |
| // RFC 3986 allows for more, but it increases the risk of matching |
| // non-URL text. |
| |
| // first character must be alpha |
| if (!ASCII_ISALPHA(*fname)) |
| return 0; |
| |
| // check body: alpha or dash |
| for (p = fname + 1; (ASCII_ISALPHA(*p) || (*p == '-')); ++p) |
| ; |
| |
| // check last char is not a dash |
| if (p[-1] == '-') |
| return 0; |
| |
| // "://" or ":\\" must follow |
| return path_is_url(p); |
| } |
| |
| #if defined(FEAT_EVAL) || defined(PROTO) |
| /* |
| * Return the dictionary of v:event. |
| * Save and clear the value in case it already has items. |
| */ |
| dict_T * |
| get_v_event(save_v_event_T *sve) |
| { |
| dict_T *v_event = get_vim_var_dict(VV_EVENT); |
| |
| if (v_event->dv_hashtab.ht_used > 0) |
| { |
| // recursive use of v:event, save, make empty and restore later |
| sve->sve_did_save = TRUE; |
| sve->sve_hashtab = v_event->dv_hashtab; |
| hash_init(&v_event->dv_hashtab); |
| } |
| else |
| sve->sve_did_save = FALSE; |
| return v_event; |
| } |
| |
| void |
| restore_v_event(dict_T *v_event, save_v_event_T *sve) |
| { |
| dict_free_contents(v_event); |
| if (sve->sve_did_save) |
| v_event->dv_hashtab = sve->sve_hashtab; |
| else |
| hash_init(&v_event->dv_hashtab); |
| } |
| #endif |
| |
| /* |
| * Fires a ModeChanged autocmd event if appropriate. |
| */ |
| void |
| may_trigger_modechanged(void) |
| { |
| #ifdef FEAT_EVAL |
| dict_T *v_event; |
| save_v_event_T save_v_event; |
| char_u curr_mode[MODE_MAX_LENGTH]; |
| char_u pattern_buf[2 * MODE_MAX_LENGTH]; |
| |
| // Skip this when got_int is set, the autocommand will not be executed. |
| // Better trigger it next time. |
| if (!has_modechanged() || got_int) |
| return; |
| |
| get_mode(curr_mode); |
| if (STRCMP(curr_mode, last_mode) == 0) |
| return; |
| |
| v_event = get_v_event(&save_v_event); |
| (void)dict_add_string(v_event, "new_mode", curr_mode); |
| (void)dict_add_string(v_event, "old_mode", last_mode); |
| dict_set_items_ro(v_event); |
| |
| // concatenate modes in format "old_mode:new_mode" |
| vim_snprintf((char *)pattern_buf, sizeof(pattern_buf), "%s:%s", last_mode, |
| curr_mode); |
| |
| apply_autocmds(EVENT_MODECHANGED, pattern_buf, NULL, FALSE, curbuf); |
| STRCPY(last_mode, curr_mode); |
| |
| restore_v_event(v_event, &save_v_event); |
| #endif |
| } |
| |
| // For overflow detection, add a digit safely to an int value. |
| int |
| vim_append_digit_int(int *value, int digit) |
| { |
| int x = *value; |
| if (x > ((INT_MAX - digit) / 10)) |
| return FAIL; |
| *value = x * 10 + digit; |
| return OK; |
| } |
| |
| // For overflow detection, add a digit safely to a long value. |
| int |
| vim_append_digit_long(long *value, int digit) |
| { |
| long x = *value; |
| if (x > ((LONG_MAX - (long)digit) / 10)) |
| return FAIL; |
| *value = x * 10 + (long)digit; |
| return OK; |
| } |
| |
| // Return something that fits into an int. |
| int |
| trim_to_int(vimlong_T x) |
| { |
| return x > INT_MAX ? INT_MAX : x < INT_MIN ? INT_MIN : x; |
| } |
| |