blob: d35061fbf10d5ebb020d222eea0b529d62115ac4 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9/*
10 * os_win32.c
11 *
12 * Used for both the console version and the Win32 GUI. A lot of code is for
13 * the console version only, so there is a lot of "#ifndef FEAT_GUI_W32".
14 *
15 * Win32 (Windows NT and Windows 95) system-dependent routines.
16 * Portions lifted from the Win32 SDK samples, the MSDOS-dependent code,
17 * NetHack 3.1.3, GNU Emacs 19.30, and Vile 5.5.
18 *
19 * George V. Reilly <george@reilly.org> wrote most of this.
20 * Roger Knobbe <rogerk@wonderware.com> did the initial port of Vim 3.0.
21 */
22
Bram Moolenaar071d4272004-06-13 20:20:40 +000023#include "vim.h"
24
Bram Moolenaar325b7a22004-07-05 15:58:32 +000025#ifdef FEAT_MZSCHEME
26# include "if_mzsch.h"
27#endif
28
Bram Moolenaar071d4272004-06-13 20:20:40 +000029#include <sys/types.h>
Bram Moolenaar071d4272004-06-13 20:20:40 +000030#include <signal.h>
31#include <limits.h>
Bram Moolenaar82881492012-11-20 16:53:39 +010032
33/* cproto fails on missing include files */
34#ifndef PROTO
35# include <process.h>
36#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000037
38#undef chdir
39#ifdef __GNUC__
40# ifndef __MINGW32__
41# include <dirent.h>
42# endif
43#else
44# include <direct.h>
45#endif
46
Bram Moolenaar82881492012-11-20 16:53:39 +010047#ifndef PROTO
48# if defined(FEAT_TITLE) && !defined(FEAT_GUI_W32)
49# include <shellapi.h>
50# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000051#endif
52
Bram Moolenaarfb630902016-10-29 14:55:00 +020053#ifdef FEAT_JOB_CHANNEL
54# include <tlhelp32.h>
55#endif
56
Bram Moolenaar071d4272004-06-13 20:20:40 +000057#ifdef __MINGW32__
58# ifndef FROM_LEFT_1ST_BUTTON_PRESSED
59# define FROM_LEFT_1ST_BUTTON_PRESSED 0x0001
60# endif
61# ifndef RIGHTMOST_BUTTON_PRESSED
62# define RIGHTMOST_BUTTON_PRESSED 0x0002
63# endif
64# ifndef FROM_LEFT_2ND_BUTTON_PRESSED
65# define FROM_LEFT_2ND_BUTTON_PRESSED 0x0004
66# endif
67# ifndef FROM_LEFT_3RD_BUTTON_PRESSED
68# define FROM_LEFT_3RD_BUTTON_PRESSED 0x0008
69# endif
70# ifndef FROM_LEFT_4TH_BUTTON_PRESSED
71# define FROM_LEFT_4TH_BUTTON_PRESSED 0x0010
72# endif
73
74/*
75 * EventFlags
76 */
77# ifndef MOUSE_MOVED
78# define MOUSE_MOVED 0x0001
79# endif
80# ifndef DOUBLE_CLICK
81# define DOUBLE_CLICK 0x0002
82# endif
83#endif
84
85/* Record all output and all keyboard & mouse input */
86/* #define MCH_WRITE_DUMP */
87
88#ifdef MCH_WRITE_DUMP
89FILE* fdDump = NULL;
90#endif
91
92/*
93 * When generating prototypes for Win32 on Unix, these lines make the syntax
94 * errors disappear. They do not need to be correct.
95 */
96#ifdef PROTO
97#define WINAPI
Bram Moolenaar071d4272004-06-13 20:20:40 +000098typedef char * LPCSTR;
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +000099typedef char * LPWSTR;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000100typedef int ACCESS_MASK;
101typedef int BOOL;
102typedef int COLORREF;
103typedef int CONSOLE_CURSOR_INFO;
104typedef int COORD;
105typedef int DWORD;
106typedef int HANDLE;
Bram Moolenaaref269542016-01-19 13:22:12 +0100107typedef int LPHANDLE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000108typedef int HDC;
109typedef int HFONT;
110typedef int HICON;
111typedef int HINSTANCE;
112typedef int HWND;
113typedef int INPUT_RECORD;
114typedef int KEY_EVENT_RECORD;
115typedef int LOGFONT;
116typedef int LPBOOL;
117typedef int LPCTSTR;
118typedef int LPDWORD;
119typedef int LPSTR;
120typedef int LPTSTR;
121typedef int LPVOID;
122typedef int MOUSE_EVENT_RECORD;
123typedef int PACL;
124typedef int PDWORD;
125typedef int PHANDLE;
126typedef int PRINTDLG;
127typedef int PSECURITY_DESCRIPTOR;
128typedef int PSID;
129typedef int SECURITY_INFORMATION;
130typedef int SHORT;
131typedef int SMALL_RECT;
132typedef int TEXTMETRIC;
133typedef int TOKEN_INFORMATION_CLASS;
134typedef int TRUSTEE;
135typedef int WORD;
136typedef int WCHAR;
137typedef void VOID;
Bram Moolenaar82881492012-11-20 16:53:39 +0100138typedef int BY_HANDLE_FILE_INFORMATION;
Bram Moolenaar32ac8cd2013-07-03 18:49:17 +0200139typedef int SE_OBJECT_TYPE;
140typedef int PSNSECINFO;
141typedef int PSNSECINFOW;
Bram Moolenaarb8e0bdb2014-11-12 16:10:48 +0100142typedef int STARTUPINFO;
143typedef int PROCESS_INFORMATION;
Bram Moolenaard90b6c02016-08-28 18:10:45 +0200144typedef int LPSECURITY_ATTRIBUTES;
145# define __stdcall /* empty */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000146#endif
147
Bram Moolenaar071d4272004-06-13 20:20:40 +0000148#if defined(__BORLANDC__)
149/* Strangely Borland uses a non-standard name. */
150# define wcsicmp(a, b) wcscmpi((a), (b))
151#endif
152
153#ifndef FEAT_GUI_W32
154/* Win32 Console handles for input and output */
155static HANDLE g_hConIn = INVALID_HANDLE_VALUE;
156static HANDLE g_hConOut = INVALID_HANDLE_VALUE;
157
158/* Win32 Screen buffer,coordinate,console I/O information */
159static SMALL_RECT g_srScrollRegion;
160static COORD g_coord; /* 0-based, but external coords are 1-based */
161
162/* The attribute of the screen when the editor was started */
163static WORD g_attrDefault = 7; /* lightgray text on black background */
164static WORD g_attrCurrent;
165
166static int g_fCBrkPressed = FALSE; /* set by ctrl-break interrupt */
167static int g_fCtrlCPressed = FALSE; /* set when ctrl-C or ctrl-break detected */
168static int g_fForceExit = FALSE; /* set when forcefully exiting */
169
170static void termcap_mode_start(void);
171static void termcap_mode_end(void);
172static void clear_chars(COORD coord, DWORD n);
173static void clear_screen(void);
174static void clear_to_end_of_display(void);
175static void clear_to_end_of_line(void);
176static void scroll(unsigned cLines);
177static void set_scroll_region(unsigned left, unsigned top,
178 unsigned right, unsigned bottom);
179static void insert_lines(unsigned cLines);
180static void delete_lines(unsigned cLines);
181static void gotoxy(unsigned x, unsigned y);
182static void normvideo(void);
183static void textattr(WORD wAttr);
184static void textcolor(WORD wAttr);
185static void textbackground(WORD wAttr);
186static void standout(void);
187static void standend(void);
188static void visual_bell(void);
189static void cursor_visible(BOOL fVisible);
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200190static DWORD write_chars(char_u *pchBuf, DWORD cbToWrite);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000191static void create_conin(void);
192static int s_cursor_visible = TRUE;
193static int did_create_conin = FALSE;
194#else
195static int s_dont_use_vimrun = TRUE;
196static int need_vimrun_warning = FALSE;
197static char *vimrun_path = "vimrun ";
198#endif
199
Bram Moolenaar12b559e2013-06-12 22:41:37 +0200200static int win32_getattrs(char_u *name);
201static int win32_setattrs(char_u *name, int attrs);
202static int win32_set_archive(char_u *name);
203
Bram Moolenaar071d4272004-06-13 20:20:40 +0000204#ifndef FEAT_GUI_W32
205static int suppress_winsize = 1; /* don't fiddle with console */
206#endif
207
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200208static char_u *exe_path = NULL;
209
Bram Moolenaarf50eb782014-02-05 13:36:54 +0100210static BOOL win8_or_later = FALSE;
211
Bram Moolenaar418f81b2016-02-16 20:12:02 +0100212#ifndef FEAT_GUI_W32
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100213/*
214 * Version of ReadConsoleInput() that works with IME.
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100215 * Works around problems on Windows 8.
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100216 */
217 static BOOL
218read_console_input(
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100219 HANDLE hInput,
220 INPUT_RECORD *lpBuffer,
221 DWORD nLength,
222 LPDWORD lpEvents)
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100223{
224 enum
225 {
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100226 IRSIZE = 10
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100227 };
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100228 static INPUT_RECORD s_irCache[IRSIZE];
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100229 static DWORD s_dwIndex = 0;
230 static DWORD s_dwMax = 0;
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100231 DWORD dwEvents;
Bram Moolenaardd415a62014-02-05 14:02:27 +0100232 int head;
233 int tail;
234 int i;
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100235
Bram Moolenaarbb86ebb2015-08-04 19:27:05 +0200236 if (nLength == -2)
237 return (s_dwMax > 0) ? TRUE : FALSE;
238
Bram Moolenaarf50eb782014-02-05 13:36:54 +0100239 if (!win8_or_later)
240 {
241 if (nLength == -1)
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200242 return PeekConsoleInputW(hInput, lpBuffer, 1, lpEvents);
243 return ReadConsoleInputW(hInput, lpBuffer, 1, &dwEvents);
Bram Moolenaarf50eb782014-02-05 13:36:54 +0100244 }
245
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100246 if (s_dwMax == 0)
247 {
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100248 if (nLength == -1)
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200249 return PeekConsoleInputW(hInput, lpBuffer, 1, lpEvents);
250 if (!ReadConsoleInputW(hInput, s_irCache, IRSIZE, &dwEvents))
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100251 return FALSE;
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100252 s_dwIndex = 0;
253 s_dwMax = dwEvents;
254 if (dwEvents == 0)
255 {
256 *lpEvents = 0;
257 return TRUE;
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100258 }
Bram Moolenaardd415a62014-02-05 14:02:27 +0100259
260 if (s_dwMax > 1)
261 {
262 head = 0;
263 tail = s_dwMax - 1;
264 while (head != tail)
265 {
266 if (s_irCache[head].EventType == WINDOW_BUFFER_SIZE_EVENT
267 && s_irCache[head + 1].EventType
268 == WINDOW_BUFFER_SIZE_EVENT)
269 {
270 /* Remove duplicate event to avoid flicker. */
271 for (i = head; i < tail; ++i)
272 s_irCache[i] = s_irCache[i + 1];
273 --tail;
274 continue;
275 }
276 head++;
277 }
278 s_dwMax = tail + 1;
279 }
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100280 }
Bram Moolenaardd415a62014-02-05 14:02:27 +0100281
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100282 *lpBuffer = s_irCache[s_dwIndex];
Bram Moolenaarbb86ebb2015-08-04 19:27:05 +0200283 if (!(nLength == -1 || nLength == -2) && ++s_dwIndex >= s_dwMax)
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100284 s_dwMax = 0;
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100285 *lpEvents = 1;
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100286 return TRUE;
287}
288
289/*
290 * Version of PeekConsoleInput() that works with IME.
291 */
292 static BOOL
293peek_console_input(
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100294 HANDLE hInput,
295 INPUT_RECORD *lpBuffer,
296 DWORD nLength,
297 LPDWORD lpEvents)
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100298{
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100299 return read_console_input(hInput, lpBuffer, -1, lpEvents);
Bram Moolenaar3a69e112014-01-10 13:51:42 +0100300}
301
Bram Moolenaar418f81b2016-02-16 20:12:02 +0100302# ifdef FEAT_CLIENTSERVER
Bram Moolenaarbb86ebb2015-08-04 19:27:05 +0200303 static DWORD
304msg_wait_for_multiple_objects(
305 DWORD nCount,
306 LPHANDLE pHandles,
307 BOOL fWaitAll,
308 DWORD dwMilliseconds,
309 DWORD dwWakeMask)
310{
311 if (read_console_input(NULL, NULL, -2, NULL))
312 return WAIT_OBJECT_0;
313 return MsgWaitForMultipleObjects(nCount, pHandles, fWaitAll,
314 dwMilliseconds, dwWakeMask);
315}
Bram Moolenaar418f81b2016-02-16 20:12:02 +0100316# endif
Bram Moolenaarbb86ebb2015-08-04 19:27:05 +0200317
Bram Moolenaar418f81b2016-02-16 20:12:02 +0100318# ifndef FEAT_CLIENTSERVER
Bram Moolenaarbb86ebb2015-08-04 19:27:05 +0200319 static DWORD
320wait_for_single_object(
321 HANDLE hHandle,
322 DWORD dwMilliseconds)
323{
324 if (read_console_input(NULL, NULL, -2, NULL))
325 return WAIT_OBJECT_0;
326 return WaitForSingleObject(hHandle, dwMilliseconds);
327}
Bram Moolenaar418f81b2016-02-16 20:12:02 +0100328# endif
329#endif
Bram Moolenaarbb86ebb2015-08-04 19:27:05 +0200330
Bram Moolenaar071d4272004-06-13 20:20:40 +0000331 static void
332get_exe_name(void)
333{
Bram Moolenaar27d9ece2010-11-10 15:37:05 +0100334 /* Maximum length of $PATH is more than MAXPATHL. 8191 is often mentioned
335 * as the maximum length that works (plus a NUL byte). */
336#define MAX_ENV_PATH_LEN 8192
337 char temp[MAX_ENV_PATH_LEN];
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200338 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000339
340 if (exe_name == NULL)
341 {
342 /* store the name of the executable, may be used for $VIM */
Bram Moolenaar27d9ece2010-11-10 15:37:05 +0100343 GetModuleFileName(NULL, temp, MAX_ENV_PATH_LEN - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000344 if (*temp != NUL)
345 exe_name = FullName_save((char_u *)temp, FALSE);
346 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +0000347
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200348 if (exe_path == NULL && exe_name != NULL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +0000349 {
Bram Moolenaar6b5ef062010-10-27 12:18:00 +0200350 exe_path = vim_strnsave(exe_name,
351 (int)(gettail_sep(exe_name) - exe_name));
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200352 if (exe_path != NULL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +0000353 {
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200354 /* Append our starting directory to $PATH, so that when doing
355 * "!xxd" it's found in our starting directory. Needed because
356 * SearchPath() also looks there. */
357 p = mch_getenv("PATH");
Bram Moolenaar27d9ece2010-11-10 15:37:05 +0100358 if (p == NULL
359 || STRLEN(p) + STRLEN(exe_path) + 2 < MAX_ENV_PATH_LEN)
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200360 {
Bram Moolenaar27d9ece2010-11-10 15:37:05 +0100361 if (p == NULL || *p == NUL)
362 temp[0] = NUL;
363 else
364 {
365 STRCPY(temp, p);
366 STRCAT(temp, ";");
367 }
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200368 STRCAT(temp, exe_path);
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +0100369 vim_setenv((char_u *)"PATH", (char_u *)temp);
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200370 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +0000371 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +0000372 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000373}
374
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200375/*
Bram Moolenaar6b707b42012-02-21 21:22:44 +0100376 * Unescape characters in "p" that appear in "escaped".
377 */
378 static void
379unescape_shellxquote(char_u *p, char_u *escaped)
380{
Bram Moolenaar4336cdf2012-02-29 13:58:47 +0100381 int l = (int)STRLEN(p);
Bram Moolenaar6b707b42012-02-21 21:22:44 +0100382 int n;
383
384 while (*p != NUL)
385 {
386 if (*p == '^' && vim_strchr(escaped, p[1]) != NULL)
387 mch_memmove(p, p + 1, l--);
388#ifdef FEAT_MBYTE
389 n = (*mb_ptr2len)(p);
390#else
391 n = 1;
392#endif
393 p += n;
394 l -= n;
395 }
396}
397
398/*
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200399 * Load library "name".
400 */
401 HINSTANCE
402vimLoadLib(char *name)
403{
Bram Moolenaar17aa8cc2012-10-21 21:38:45 +0200404 HINSTANCE dll = NULL;
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200405
Bram Moolenaarfaca8402012-10-21 02:37:10 +0200406 /* NOTE: Do not use mch_dirname() and mch_chdir() here, they may call
407 * vimLoadLib() recursively, which causes a stack overflow. */
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200408 if (exe_path == NULL)
409 get_exe_name();
Bram Moolenaar17aa8cc2012-10-21 21:38:45 +0200410 if (exe_path != NULL)
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200411 {
Bram Moolenaar17aa8cc2012-10-21 21:38:45 +0200412 WCHAR old_dirw[MAXPATHL];
413
414 if (GetCurrentDirectoryW(MAXPATHL, old_dirw) != 0)
415 {
416 /* Change directory to where the executable is, both to make
417 * sure we find a .dll there and to avoid looking for a .dll
418 * in the current directory. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +0100419 SetCurrentDirectory((LPCSTR)exe_path);
Bram Moolenaar17aa8cc2012-10-21 21:38:45 +0200420 dll = LoadLibrary(name);
421 SetCurrentDirectoryW(old_dirw);
422 return dll;
423 }
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200424 }
425 return dll;
426}
427
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100428#if defined(DYNAMIC_ICONV) || defined(DYNAMIC_GETTEXT) || defined(PROTO)
429/*
430 * Get related information about 'funcname' which is imported by 'hInst'.
431 * If 'info' is 0, return the function address.
432 * If 'info' is 1, return the module name which the function is imported from.
433 */
434 static void *
435get_imported_func_info(HINSTANCE hInst, const char *funcname, int info)
436{
437 PBYTE pImage = (PBYTE)hInst;
438 PIMAGE_DOS_HEADER pDOS = (PIMAGE_DOS_HEADER)hInst;
439 PIMAGE_NT_HEADERS pPE;
440 PIMAGE_IMPORT_DESCRIPTOR pImpDesc;
441 PIMAGE_THUNK_DATA pIAT; /* Import Address Table */
442 PIMAGE_THUNK_DATA pINT; /* Import Name Table */
443 PIMAGE_IMPORT_BY_NAME pImpName;
444
445 if (pDOS->e_magic != IMAGE_DOS_SIGNATURE)
446 return NULL;
447 pPE = (PIMAGE_NT_HEADERS)(pImage + pDOS->e_lfanew);
448 if (pPE->Signature != IMAGE_NT_SIGNATURE)
449 return NULL;
450 pImpDesc = (PIMAGE_IMPORT_DESCRIPTOR)(pImage
451 + pPE->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT]
452 .VirtualAddress);
453 for (; pImpDesc->FirstThunk; ++pImpDesc)
454 {
455 if (!pImpDesc->OriginalFirstThunk)
456 continue;
457 pIAT = (PIMAGE_THUNK_DATA)(pImage + pImpDesc->FirstThunk);
458 pINT = (PIMAGE_THUNK_DATA)(pImage + pImpDesc->OriginalFirstThunk);
459 for (; pIAT->u1.Function; ++pIAT, ++pINT)
460 {
461 if (IMAGE_SNAP_BY_ORDINAL(pINT->u1.Ordinal))
462 continue;
463 pImpName = (PIMAGE_IMPORT_BY_NAME)(pImage
464 + (UINT_PTR)(pINT->u1.AddressOfData));
465 if (strcmp((char *)pImpName->Name, funcname) == 0)
466 {
467 switch (info)
468 {
469 case 0:
470 return (void *)pIAT->u1.Function;
471 case 1:
472 return (void *)(pImage + pImpDesc->Name);
473 default:
474 return NULL;
475 }
476 }
477 }
478 }
479 return NULL;
480}
481
482/*
483 * Get the module handle which 'funcname' in 'hInst' is imported from.
484 */
485 HINSTANCE
486find_imported_module_by_funcname(HINSTANCE hInst, const char *funcname)
487{
488 char *modulename;
489
490 modulename = (char *)get_imported_func_info(hInst, funcname, 1);
491 if (modulename != NULL)
492 return GetModuleHandleA(modulename);
493 return NULL;
494}
495
496/*
497 * Get the address of 'funcname' which is imported by 'hInst' DLL.
498 */
499 void *
500get_dll_import_func(HINSTANCE hInst, const char *funcname)
501{
502 return get_imported_func_info(hInst, funcname, 0);
503}
504#endif
505
Bram Moolenaar071d4272004-06-13 20:20:40 +0000506#if defined(DYNAMIC_GETTEXT) || defined(PROTO)
507# ifndef GETTEXT_DLL
508# define GETTEXT_DLL "libintl.dll"
Bram Moolenaar286eacd2016-01-16 18:05:50 +0100509# define GETTEXT_DLL_ALT "libintl-8.dll"
Bram Moolenaar071d4272004-06-13 20:20:40 +0000510# endif
Bram Moolenaarf6a2b082012-06-29 13:14:03 +0200511/* Dummy functions */
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000512static char *null_libintl_gettext(const char *);
Bram Moolenaaree695f72016-08-03 22:08:45 +0200513static char *null_libintl_ngettext(const char *, const char *, unsigned long n);
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000514static char *null_libintl_textdomain(const char *);
515static char *null_libintl_bindtextdomain(const char *, const char *);
516static char *null_libintl_bind_textdomain_codeset(const char *, const char *);
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100517static int null_libintl_putenv(const char *);
Bram Moolenaar7c23d1d2017-02-01 13:14:16 +0100518static int null_libintl_wputenv(const wchar_t *);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000519
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200520static HINSTANCE hLibintlDLL = NULL;
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000521char *(*dyn_libintl_gettext)(const char *) = null_libintl_gettext;
Bram Moolenaaree695f72016-08-03 22:08:45 +0200522char *(*dyn_libintl_ngettext)(const char *, const char *, unsigned long n)
523 = null_libintl_ngettext;
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000524char *(*dyn_libintl_textdomain)(const char *) = null_libintl_textdomain;
525char *(*dyn_libintl_bindtextdomain)(const char *, const char *)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000526 = null_libintl_bindtextdomain;
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000527char *(*dyn_libintl_bind_textdomain_codeset)(const char *, const char *)
528 = null_libintl_bind_textdomain_codeset;
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100529int (*dyn_libintl_putenv)(const char *) = null_libintl_putenv;
Bram Moolenaar7c23d1d2017-02-01 13:14:16 +0100530int (*dyn_libintl_wputenv)(const wchar_t *) = null_libintl_wputenv;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000531
532 int
Bram Moolenaar05540972016-01-30 20:31:25 +0100533dyn_libintl_init(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534{
535 int i;
536 static struct
537 {
538 char *name;
539 FARPROC *ptr;
540 } libintl_entry[] =
541 {
542 {"gettext", (FARPROC*)&dyn_libintl_gettext},
Bram Moolenaaree695f72016-08-03 22:08:45 +0200543 {"ngettext", (FARPROC*)&dyn_libintl_ngettext},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000544 {"textdomain", (FARPROC*)&dyn_libintl_textdomain},
545 {"bindtextdomain", (FARPROC*)&dyn_libintl_bindtextdomain},
546 {NULL, NULL}
547 };
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100548 HINSTANCE hmsvcrt;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000549
550 /* No need to initialize twice. */
551 if (hLibintlDLL)
552 return 1;
553 /* Load gettext library (libintl.dll) */
Bram Moolenaar923e43b2016-01-28 15:07:38 +0100554 hLibintlDLL = vimLoadLib(GETTEXT_DLL);
Bram Moolenaar938ee832016-01-24 15:36:03 +0100555#ifdef GETTEXT_DLL_ALT
Bram Moolenaar286eacd2016-01-16 18:05:50 +0100556 if (!hLibintlDLL)
557 hLibintlDLL = vimLoadLib(GETTEXT_DLL_ALT);
Bram Moolenaar938ee832016-01-24 15:36:03 +0100558#endif
559 if (!hLibintlDLL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000560 {
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200561 if (p_verbose > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000562 {
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200563 verbose_enter();
564 EMSG2(_(e_loadlib), GETTEXT_DLL);
565 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566 }
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200567 return 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000568 }
569 for (i = 0; libintl_entry[i].name != NULL
570 && libintl_entry[i].ptr != NULL; ++i)
571 {
572 if ((*libintl_entry[i].ptr = (FARPROC)GetProcAddress(hLibintlDLL,
573 libintl_entry[i].name)) == NULL)
574 {
575 dyn_libintl_end();
576 if (p_verbose > 0)
Bram Moolenaara04f10b2005-05-31 22:09:46 +0000577 {
578 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000579 EMSG2(_(e_loadfunc), libintl_entry[i].name);
Bram Moolenaara04f10b2005-05-31 22:09:46 +0000580 verbose_leave();
581 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000582 return 0;
583 }
584 }
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000585
586 /* The bind_textdomain_codeset() function is optional. */
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +0000587 dyn_libintl_bind_textdomain_codeset = (void *)GetProcAddress(hLibintlDLL,
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000588 "bind_textdomain_codeset");
589 if (dyn_libintl_bind_textdomain_codeset == NULL)
590 dyn_libintl_bind_textdomain_codeset =
591 null_libintl_bind_textdomain_codeset;
592
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100593 /* _putenv() function for the libintl.dll is optional. */
594 hmsvcrt = find_imported_module_by_funcname(hLibintlDLL, "getenv");
595 if (hmsvcrt != NULL)
Bram Moolenaar7c23d1d2017-02-01 13:14:16 +0100596 {
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100597 dyn_libintl_putenv = (void *)GetProcAddress(hmsvcrt, "_putenv");
Bram Moolenaar7c23d1d2017-02-01 13:14:16 +0100598 dyn_libintl_wputenv = (void *)GetProcAddress(hmsvcrt, "_wputenv");
599 }
600 if (dyn_libintl_putenv == NULL || dyn_libintl_putenv == _putenv)
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100601 dyn_libintl_putenv = null_libintl_putenv;
Bram Moolenaar7c23d1d2017-02-01 13:14:16 +0100602 if (dyn_libintl_wputenv == NULL || dyn_libintl_wputenv == _wputenv)
603 dyn_libintl_wputenv = null_libintl_wputenv;
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100604
Bram Moolenaar071d4272004-06-13 20:20:40 +0000605 return 1;
606}
607
608 void
Bram Moolenaar05540972016-01-30 20:31:25 +0100609dyn_libintl_end(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000610{
611 if (hLibintlDLL)
612 FreeLibrary(hLibintlDLL);
613 hLibintlDLL = NULL;
614 dyn_libintl_gettext = null_libintl_gettext;
Bram Moolenaaree695f72016-08-03 22:08:45 +0200615 dyn_libintl_ngettext = null_libintl_ngettext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000616 dyn_libintl_textdomain = null_libintl_textdomain;
617 dyn_libintl_bindtextdomain = null_libintl_bindtextdomain;
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000618 dyn_libintl_bind_textdomain_codeset = null_libintl_bind_textdomain_codeset;
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100619 dyn_libintl_putenv = null_libintl_putenv;
Bram Moolenaar7c23d1d2017-02-01 13:14:16 +0100620 dyn_libintl_wputenv = null_libintl_wputenv;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000621}
622
623 static char *
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +0000624null_libintl_gettext(const char *msgid)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000625{
626 return (char*)msgid;
627}
628
629 static char *
Bram Moolenaaree695f72016-08-03 22:08:45 +0200630null_libintl_ngettext(
631 const char *msgid,
632 const char *msgid_plural,
633 unsigned long n)
634{
Bram Moolenaarc90f2ae2016-08-04 22:00:15 +0200635 return (char *)(n == 1 ? msgid : msgid_plural);
Bram Moolenaaree695f72016-08-03 22:08:45 +0200636}
637
Bram Moolenaaree695f72016-08-03 22:08:45 +0200638 static char *
Bram Moolenaar1266d672017-02-01 13:43:36 +0100639null_libintl_bindtextdomain(
640 const char *domainname UNUSED,
641 const char *dirname UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000642{
643 return NULL;
644}
645
646 static char *
Bram Moolenaar1266d672017-02-01 13:43:36 +0100647null_libintl_bind_textdomain_codeset(
648 const char *domainname UNUSED,
649 const char *codeset UNUSED)
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000650{
651 return NULL;
652}
653
654 static char *
Bram Moolenaar1266d672017-02-01 13:43:36 +0100655null_libintl_textdomain(const char *domainname UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000656{
657 return NULL;
658}
659
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100660 int
Bram Moolenaar1266d672017-02-01 13:43:36 +0100661null_libintl_putenv(const char *envstring UNUSED)
Bram Moolenaar972c3b82017-01-12 21:44:49 +0100662{
663 return 0;
664}
665
Bram Moolenaar7c23d1d2017-02-01 13:14:16 +0100666 int
Bram Moolenaar1266d672017-02-01 13:43:36 +0100667null_libintl_wputenv(const wchar_t *envstring UNUSED)
Bram Moolenaar7c23d1d2017-02-01 13:14:16 +0100668{
669 return 0;
670}
671
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672#endif /* DYNAMIC_GETTEXT */
673
674/* This symbol is not defined in older versions of the SDK or Visual C++ */
675
676#ifndef VER_PLATFORM_WIN32_WINDOWS
677# define VER_PLATFORM_WIN32_WINDOWS 1
678#endif
679
680DWORD g_PlatformId;
681
682#ifdef HAVE_ACL
Bram Moolenaar82881492012-11-20 16:53:39 +0100683# ifndef PROTO
684# include <aclapi.h>
685# endif
Bram Moolenaar27515922013-06-29 15:36:26 +0200686# ifndef PROTECTED_DACL_SECURITY_INFORMATION
687# define PROTECTED_DACL_SECURITY_INFORMATION 0x80000000L
688# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000689#endif
690
Bram Moolenaar27515922013-06-29 15:36:26 +0200691#ifdef HAVE_ACL
692/*
693 * Enables or disables the specified privilege.
694 */
695 static BOOL
696win32_enable_privilege(LPTSTR lpszPrivilege, BOOL bEnable)
697{
Bram Moolenaarb0d5c962014-01-12 13:24:51 +0100698 BOOL bResult;
699 LUID luid;
700 HANDLE hToken;
701 TOKEN_PRIVILEGES tokenPrivileges;
Bram Moolenaar27515922013-06-29 15:36:26 +0200702
703 if (!OpenProcessToken(GetCurrentProcess(),
704 TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
705 return FALSE;
706
707 if (!LookupPrivilegeValue(NULL, lpszPrivilege, &luid))
708 {
709 CloseHandle(hToken);
710 return FALSE;
711 }
712
Bram Moolenaar45500912014-07-09 20:51:07 +0200713 tokenPrivileges.PrivilegeCount = 1;
Bram Moolenaar27515922013-06-29 15:36:26 +0200714 tokenPrivileges.Privileges[0].Luid = luid;
715 tokenPrivileges.Privileges[0].Attributes = bEnable ?
716 SE_PRIVILEGE_ENABLED : 0;
717
718 bResult = AdjustTokenPrivileges(hToken, FALSE, &tokenPrivileges,
719 sizeof(TOKEN_PRIVILEGES), NULL, NULL);
720
721 CloseHandle(hToken);
722
723 return bResult && GetLastError() == ERROR_SUCCESS;
724}
725#endif
726
Bram Moolenaar071d4272004-06-13 20:20:40 +0000727/*
728 * Set g_PlatformId to VER_PLATFORM_WIN32_NT (NT) or
729 * VER_PLATFORM_WIN32_WINDOWS (Win95).
730 */
731 void
732PlatformId(void)
733{
734 static int done = FALSE;
735
736 if (!done)
737 {
738 OSVERSIONINFO ovi;
739
740 ovi.dwOSVersionInfoSize = sizeof(ovi);
741 GetVersionEx(&ovi);
742
743 g_PlatformId = ovi.dwPlatformId;
744
Bram Moolenaarf50eb782014-02-05 13:36:54 +0100745 if ((ovi.dwMajorVersion == 6 && ovi.dwMinorVersion >= 2)
746 || ovi.dwMajorVersion > 6)
747 win8_or_later = TRUE;
748
Bram Moolenaar071d4272004-06-13 20:20:40 +0000749#ifdef HAVE_ACL
Bram Moolenaarcea912a2016-10-12 14:20:24 +0200750 /* Enable privilege for getting or setting SACLs. */
751 win32_enable_privilege(SE_SECURITY_NAME, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000752#endif
753 done = TRUE;
754 }
755}
756
Bram Moolenaar071d4272004-06-13 20:20:40 +0000757#ifndef FEAT_GUI_W32
758
759#define SHIFT (SHIFT_PRESSED)
760#define CTRL (RIGHT_CTRL_PRESSED | LEFT_CTRL_PRESSED)
761#define ALT (RIGHT_ALT_PRESSED | LEFT_ALT_PRESSED)
762#define ALT_GR (RIGHT_ALT_PRESSED | LEFT_CTRL_PRESSED)
763
764
765/* When uChar.AsciiChar is 0, then we need to look at wVirtualKeyCode.
766 * We map function keys to their ANSI terminal equivalents, as produced
767 * by ANSI.SYS, for compatibility with the MS-DOS version of Vim. Any
768 * ANSI key with a value >= '\300' is nonstandard, but provided anyway
769 * so that the user can have access to all SHIFT-, CTRL-, and ALT-
770 * combinations of function/arrow/etc keys.
771 */
772
Bram Moolenaard6f676d2005-06-01 21:51:55 +0000773static const struct
Bram Moolenaar071d4272004-06-13 20:20:40 +0000774{
775 WORD wVirtKey;
776 BOOL fAnsiKey;
777 int chAlone;
778 int chShift;
779 int chCtrl;
780 int chAlt;
781} VirtKeyMap[] =
782{
783
784/* Key ANSI alone shift ctrl alt */
785 { VK_ESCAPE,FALSE, ESC, ESC, ESC, ESC, },
786
787 { VK_F1, TRUE, ';', 'T', '^', 'h', },
788 { VK_F2, TRUE, '<', 'U', '_', 'i', },
789 { VK_F3, TRUE, '=', 'V', '`', 'j', },
790 { VK_F4, TRUE, '>', 'W', 'a', 'k', },
791 { VK_F5, TRUE, '?', 'X', 'b', 'l', },
792 { VK_F6, TRUE, '@', 'Y', 'c', 'm', },
793 { VK_F7, TRUE, 'A', 'Z', 'd', 'n', },
794 { VK_F8, TRUE, 'B', '[', 'e', 'o', },
795 { VK_F9, TRUE, 'C', '\\', 'f', 'p', },
796 { VK_F10, TRUE, 'D', ']', 'g', 'q', },
797 { VK_F11, TRUE, '\205', '\207', '\211', '\213', },
798 { VK_F12, TRUE, '\206', '\210', '\212', '\214', },
799
800 { VK_HOME, TRUE, 'G', '\302', 'w', '\303', },
801 { VK_UP, TRUE, 'H', '\304', '\305', '\306', },
802 { VK_PRIOR, TRUE, 'I', '\307', '\204', '\310', }, /*PgUp*/
803 { VK_LEFT, TRUE, 'K', '\311', 's', '\312', },
804 { VK_RIGHT, TRUE, 'M', '\313', 't', '\314', },
805 { VK_END, TRUE, 'O', '\315', 'u', '\316', },
806 { VK_DOWN, TRUE, 'P', '\317', '\320', '\321', },
807 { VK_NEXT, TRUE, 'Q', '\322', 'v', '\323', }, /*PgDn*/
808 { VK_INSERT,TRUE, 'R', '\324', '\325', '\326', },
809 { VK_DELETE,TRUE, 'S', '\327', '\330', '\331', },
810
811 { VK_SNAPSHOT,TRUE, 0, 0, 0, 'r', }, /*PrtScrn*/
812
813#if 0
814 /* Most people don't have F13-F20, but what the hell... */
815 { VK_F13, TRUE, '\332', '\333', '\334', '\335', },
816 { VK_F14, TRUE, '\336', '\337', '\340', '\341', },
817 { VK_F15, TRUE, '\342', '\343', '\344', '\345', },
818 { VK_F16, TRUE, '\346', '\347', '\350', '\351', },
819 { VK_F17, TRUE, '\352', '\353', '\354', '\355', },
820 { VK_F18, TRUE, '\356', '\357', '\360', '\361', },
821 { VK_F19, TRUE, '\362', '\363', '\364', '\365', },
822 { VK_F20, TRUE, '\366', '\367', '\370', '\371', },
823#endif
824 { VK_ADD, TRUE, 'N', 'N', 'N', 'N', }, /* keyp '+' */
825 { VK_SUBTRACT, TRUE,'J', 'J', 'J', 'J', }, /* keyp '-' */
826 /* { VK_DIVIDE, TRUE,'N', 'N', 'N', 'N', }, keyp '/' */
827 { VK_MULTIPLY, TRUE,'7', '7', '7', '7', }, /* keyp '*' */
828
829 { VK_NUMPAD0,TRUE, '\332', '\333', '\334', '\335', },
830 { VK_NUMPAD1,TRUE, '\336', '\337', '\340', '\341', },
831 { VK_NUMPAD2,TRUE, '\342', '\343', '\344', '\345', },
832 { VK_NUMPAD3,TRUE, '\346', '\347', '\350', '\351', },
833 { VK_NUMPAD4,TRUE, '\352', '\353', '\354', '\355', },
834 { VK_NUMPAD5,TRUE, '\356', '\357', '\360', '\361', },
835 { VK_NUMPAD6,TRUE, '\362', '\363', '\364', '\365', },
836 { VK_NUMPAD7,TRUE, '\366', '\367', '\370', '\371', },
837 { VK_NUMPAD8,TRUE, '\372', '\373', '\374', '\375', },
838 /* Sorry, out of number space! <negri>*/
839 { VK_NUMPAD9,TRUE, '\376', '\377', '\377', '\367', },
840
841};
842
843
844#ifdef _MSC_VER
845// The ToAscii bug destroys several registers. Need to turn off optimization
846// or the GetConsoleKeyboardLayoutName hack will fail in non-debug versions
Bram Moolenaar7b5f8322006-03-23 22:47:08 +0000847# pragma warning(push)
848# pragma warning(disable: 4748)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000849# pragma optimize("", off)
850#endif
851
852#if defined(__GNUC__) && !defined(__MINGW32__) && !defined(__CYGWIN__)
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200853# define UChar UnicodeChar
Bram Moolenaar071d4272004-06-13 20:20:40 +0000854#else
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200855# define UChar uChar.UnicodeChar
Bram Moolenaar071d4272004-06-13 20:20:40 +0000856#endif
857
858/* The return code indicates key code size. */
859 static int
860#ifdef __BORLANDC__
861 __stdcall
862#endif
863win32_kbd_patch_key(
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +0000864 KEY_EVENT_RECORD *pker)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000865{
866 UINT uMods = pker->dwControlKeyState;
867 static int s_iIsDead = 0;
868 static WORD awAnsiCode[2];
869 static BYTE abKeystate[256];
870
871
872 if (s_iIsDead == 2)
873 {
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200874 pker->UChar = (WCHAR) awAnsiCode[1];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000875 s_iIsDead = 0;
876 return 1;
877 }
878
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200879 if (pker->UChar != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000880 return 1;
881
Bram Moolenaar7db5fc82010-05-24 11:59:29 +0200882 vim_memset(abKeystate, 0, sizeof (abKeystate));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000883
Bram Moolenaar071d4272004-06-13 20:20:40 +0000884 /* Clear any pending dead keys */
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200885 ToUnicode(VK_SPACE, MapVirtualKey(VK_SPACE, 0), abKeystate, awAnsiCode, 2, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000886
887 if (uMods & SHIFT_PRESSED)
888 abKeystate[VK_SHIFT] = 0x80;
889 if (uMods & CAPSLOCK_ON)
890 abKeystate[VK_CAPITAL] = 1;
891
892 if ((uMods & ALT_GR) == ALT_GR)
893 {
894 abKeystate[VK_CONTROL] = abKeystate[VK_LCONTROL] =
895 abKeystate[VK_MENU] = abKeystate[VK_RMENU] = 0x80;
896 }
897
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200898 s_iIsDead = ToUnicode(pker->wVirtualKeyCode, pker->wVirtualScanCode,
899 abKeystate, awAnsiCode, 2, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000900
901 if (s_iIsDead > 0)
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200902 pker->UChar = (WCHAR) awAnsiCode[0];
Bram Moolenaar071d4272004-06-13 20:20:40 +0000903
904 return s_iIsDead;
905}
906
907#ifdef _MSC_VER
908/* MUST switch optimization on again here, otherwise a call to
909 * decode_key_event() may crash (e.g. when hitting caps-lock) */
910# pragma optimize("", on)
Bram Moolenaar7b5f8322006-03-23 22:47:08 +0000911# pragma warning(pop)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000912
913# if (_MSC_VER < 1100)
914/* MUST turn off global optimisation for this next function, or
915 * pressing ctrl-minus in insert mode crashes Vim when built with
916 * VC4.1. -- negri. */
917# pragma optimize("g", off)
918# endif
919#endif
920
921static BOOL g_fJustGotFocus = FALSE;
922
923/*
924 * Decode a KEY_EVENT into one or two keystrokes
925 */
926 static BOOL
927decode_key_event(
928 KEY_EVENT_RECORD *pker,
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200929 WCHAR *pch,
930 WCHAR *pch2,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000931 int *pmodifiers,
932 BOOL fDoPost)
933{
934 int i;
935 const int nModifs = pker->dwControlKeyState & (SHIFT | ALT | CTRL);
936
937 *pch = *pch2 = NUL;
938 g_fJustGotFocus = FALSE;
939
940 /* ignore key up events */
941 if (!pker->bKeyDown)
942 return FALSE;
943
944 /* ignore some keystrokes */
945 switch (pker->wVirtualKeyCode)
946 {
947 /* modifiers */
948 case VK_SHIFT:
949 case VK_CONTROL:
950 case VK_MENU: /* Alt key */
951 return FALSE;
952
953 default:
954 break;
955 }
956
957 /* special cases */
Bram Moolenaarac360bf2015-09-01 20:31:20 +0200958 if ((nModifs & CTRL) != 0 && (nModifs & ~CTRL) == 0 && pker->UChar == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000959 {
960 /* Ctrl-6 is Ctrl-^ */
961 if (pker->wVirtualKeyCode == '6')
962 {
963 *pch = Ctrl_HAT;
964 return TRUE;
965 }
966 /* Ctrl-2 is Ctrl-@ */
967 else if (pker->wVirtualKeyCode == '2')
968 {
969 *pch = NUL;
970 return TRUE;
971 }
972 /* Ctrl-- is Ctrl-_ */
973 else if (pker->wVirtualKeyCode == 0xBD)
974 {
975 *pch = Ctrl__;
976 return TRUE;
977 }
978 }
979
980 /* Shift-TAB */
981 if (pker->wVirtualKeyCode == VK_TAB && (nModifs & SHIFT_PRESSED))
982 {
983 *pch = K_NUL;
984 *pch2 = '\017';
985 return TRUE;
986 }
987
988 for (i = sizeof(VirtKeyMap) / sizeof(VirtKeyMap[0]); --i >= 0; )
989 {
990 if (VirtKeyMap[i].wVirtKey == pker->wVirtualKeyCode)
991 {
992 if (nModifs == 0)
993 *pch = VirtKeyMap[i].chAlone;
994 else if ((nModifs & SHIFT) != 0 && (nModifs & ~SHIFT) == 0)
995 *pch = VirtKeyMap[i].chShift;
996 else if ((nModifs & CTRL) != 0 && (nModifs & ~CTRL) == 0)
997 *pch = VirtKeyMap[i].chCtrl;
998 else if ((nModifs & ALT) != 0 && (nModifs & ~ALT) == 0)
999 *pch = VirtKeyMap[i].chAlt;
1000
1001 if (*pch != 0)
1002 {
1003 if (VirtKeyMap[i].fAnsiKey)
1004 {
1005 *pch2 = *pch;
1006 *pch = K_NUL;
1007 }
1008
1009 return TRUE;
1010 }
1011 }
1012 }
1013
1014 i = win32_kbd_patch_key(pker);
1015
1016 if (i < 0)
1017 *pch = NUL;
1018 else
1019 {
Bram Moolenaarac360bf2015-09-01 20:31:20 +02001020 *pch = (i > 0) ? pker->UChar : NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001021
1022 if (pmodifiers != NULL)
1023 {
1024 /* Pass on the ALT key as a modifier, but only when not combined
1025 * with CTRL (which is ALTGR). */
1026 if ((nModifs & ALT) != 0 && (nModifs & CTRL) == 0)
1027 *pmodifiers |= MOD_MASK_ALT;
1028
1029 /* Pass on SHIFT only for special keys, because we don't know when
1030 * it's already included with the character. */
1031 if ((nModifs & SHIFT) != 0 && *pch <= 0x20)
1032 *pmodifiers |= MOD_MASK_SHIFT;
1033
1034 /* Pass on CTRL only for non-special keys, because we don't know
1035 * when it's already included with the character. And not when
1036 * combined with ALT (which is ALTGR). */
1037 if ((nModifs & CTRL) != 0 && (nModifs & ALT) == 0
1038 && *pch >= 0x20 && *pch < 0x80)
1039 *pmodifiers |= MOD_MASK_CTRL;
1040 }
1041 }
1042
1043 return (*pch != NUL);
1044}
1045
1046#ifdef _MSC_VER
1047# pragma optimize("", on)
1048#endif
1049
1050#endif /* FEAT_GUI_W32 */
1051
1052
1053#ifdef FEAT_MOUSE
1054
1055/*
1056 * For the GUI the mouse handling is in gui_w32.c.
1057 */
1058# ifdef FEAT_GUI_W32
1059 void
Bram Moolenaar1266d672017-02-01 13:43:36 +01001060mch_setmouse(int on UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001061{
1062}
1063# else
1064static int g_fMouseAvail = FALSE; /* mouse present */
1065static int g_fMouseActive = FALSE; /* mouse enabled */
1066static int g_nMouseClick = -1; /* mouse status */
1067static int g_xMouse; /* mouse x coordinate */
1068static int g_yMouse; /* mouse y coordinate */
1069
1070/*
1071 * Enable or disable mouse input
1072 */
1073 void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00001074mch_setmouse(int on)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001075{
1076 DWORD cmodein;
1077
1078 if (!g_fMouseAvail)
1079 return;
1080
1081 g_fMouseActive = on;
1082 GetConsoleMode(g_hConIn, &cmodein);
1083
1084 if (g_fMouseActive)
1085 cmodein |= ENABLE_MOUSE_INPUT;
1086 else
1087 cmodein &= ~ENABLE_MOUSE_INPUT;
1088
1089 SetConsoleMode(g_hConIn, cmodein);
1090}
1091
Bram Moolenaar071d4272004-06-13 20:20:40 +00001092/*
1093 * Decode a MOUSE_EVENT. If it's a valid event, return MOUSE_LEFT,
1094 * MOUSE_MIDDLE, or MOUSE_RIGHT for a click; MOUSE_DRAG for a mouse
1095 * move with a button held down; and MOUSE_RELEASE after a MOUSE_DRAG
1096 * or a MOUSE_LEFT, _MIDDLE, or _RIGHT. We encode the button type,
1097 * the number of clicks, and the Shift/Ctrl/Alt modifiers in g_nMouseClick,
1098 * and we return the mouse position in g_xMouse and g_yMouse.
1099 *
1100 * Every MOUSE_LEFT, _MIDDLE, or _RIGHT will be followed by zero or more
1101 * MOUSE_DRAGs and one MOUSE_RELEASE. MOUSE_RELEASE will be followed only
1102 * by MOUSE_LEFT, _MIDDLE, or _RIGHT.
1103 *
1104 * For multiple clicks, we send, say, MOUSE_LEFT/1 click, MOUSE_RELEASE,
1105 * MOUSE_LEFT/2 clicks, MOUSE_RELEASE, MOUSE_LEFT/3 clicks, MOUSE_RELEASE, ....
1106 *
1107 * Windows will send us MOUSE_MOVED notifications whenever the mouse
1108 * moves, even if it stays within the same character cell. We ignore
1109 * all MOUSE_MOVED messages if the position hasn't really changed, and
1110 * we ignore all MOUSE_MOVED messages where no button is held down (i.e.,
1111 * we're only interested in MOUSE_DRAG).
1112 *
1113 * All of this is complicated by the code that fakes MOUSE_MIDDLE on
1114 * 2-button mouses by pressing the left & right buttons simultaneously.
1115 * In practice, it's almost impossible to click both at the same time,
1116 * so we need to delay a little. Also, we tend not to get MOUSE_RELEASE
1117 * in such cases, if the user is clicking quickly.
1118 */
1119 static BOOL
1120decode_mouse_event(
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00001121 MOUSE_EVENT_RECORD *pmer)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001122{
1123 static int s_nOldButton = -1;
1124 static int s_nOldMouseClick = -1;
1125 static int s_xOldMouse = -1;
1126 static int s_yOldMouse = -1;
1127 static linenr_T s_old_topline = 0;
1128#ifdef FEAT_DIFF
1129 static int s_old_topfill = 0;
1130#endif
1131 static int s_cClicks = 1;
1132 static BOOL s_fReleased = TRUE;
1133 static DWORD s_dwLastClickTime = 0;
1134 static BOOL s_fNextIsMiddle = FALSE;
1135
1136 static DWORD cButtons = 0; /* number of buttons supported */
1137
1138 const DWORD LEFT = FROM_LEFT_1ST_BUTTON_PRESSED;
1139 const DWORD MIDDLE = FROM_LEFT_2ND_BUTTON_PRESSED;
1140 const DWORD RIGHT = RIGHTMOST_BUTTON_PRESSED;
1141 const DWORD LEFT_RIGHT = LEFT | RIGHT;
1142
1143 int nButton;
1144
1145 if (cButtons == 0 && !GetNumberOfConsoleMouseButtons(&cButtons))
1146 cButtons = 2;
1147
1148 if (!g_fMouseAvail || !g_fMouseActive)
1149 {
1150 g_nMouseClick = -1;
1151 return FALSE;
1152 }
1153
1154 /* get a spurious MOUSE_EVENT immediately after receiving focus; ignore */
1155 if (g_fJustGotFocus)
1156 {
1157 g_fJustGotFocus = FALSE;
1158 return FALSE;
1159 }
1160
1161 /* unprocessed mouse click? */
1162 if (g_nMouseClick != -1)
1163 return TRUE;
1164
1165 nButton = -1;
1166 g_xMouse = pmer->dwMousePosition.X;
1167 g_yMouse = pmer->dwMousePosition.Y;
1168
1169 if (pmer->dwEventFlags == MOUSE_MOVED)
1170 {
1171 /* ignore MOUSE_MOVED events if (x, y) hasn't changed. (We get these
1172 * events even when the mouse moves only within a char cell.) */
1173 if (s_xOldMouse == g_xMouse && s_yOldMouse == g_yMouse)
1174 return FALSE;
1175 }
1176
1177 /* If no buttons are pressed... */
1178 if ((pmer->dwButtonState & ((1 << cButtons) - 1)) == 0)
1179 {
1180 /* If the last thing returned was MOUSE_RELEASE, ignore this */
1181 if (s_fReleased)
1182 return FALSE;
1183
1184 nButton = MOUSE_RELEASE;
1185 s_fReleased = TRUE;
1186 }
1187 else /* one or more buttons pressed */
1188 {
1189 /* on a 2-button mouse, hold down left and right buttons
1190 * simultaneously to get MIDDLE. */
1191
1192 if (cButtons == 2 && s_nOldButton != MOUSE_DRAG)
1193 {
1194 DWORD dwLR = (pmer->dwButtonState & LEFT_RIGHT);
1195
1196 /* if either left or right button only is pressed, see if the
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001197 * next mouse event has both of them pressed */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001198 if (dwLR == LEFT || dwLR == RIGHT)
1199 {
1200 for (;;)
1201 {
1202 /* wait a short time for next input event */
1203 if (WaitForSingleObject(g_hConIn, p_mouset / 3)
1204 != WAIT_OBJECT_0)
1205 break;
1206 else
1207 {
1208 DWORD cRecords = 0;
1209 INPUT_RECORD ir;
1210 MOUSE_EVENT_RECORD* pmer2 = &ir.Event.MouseEvent;
1211
Bram Moolenaar3a69e112014-01-10 13:51:42 +01001212 peek_console_input(g_hConIn, &ir, 1, &cRecords);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001213
1214 if (cRecords == 0 || ir.EventType != MOUSE_EVENT
1215 || !(pmer2->dwButtonState & LEFT_RIGHT))
1216 break;
1217 else
1218 {
1219 if (pmer2->dwEventFlags != MOUSE_MOVED)
1220 {
Bram Moolenaar3a69e112014-01-10 13:51:42 +01001221 read_console_input(g_hConIn, &ir, 1, &cRecords);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001222
1223 return decode_mouse_event(pmer2);
1224 }
1225 else if (s_xOldMouse == pmer2->dwMousePosition.X &&
1226 s_yOldMouse == pmer2->dwMousePosition.Y)
1227 {
1228 /* throw away spurious mouse move */
Bram Moolenaar3a69e112014-01-10 13:51:42 +01001229 read_console_input(g_hConIn, &ir, 1, &cRecords);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001230
1231 /* are there any more mouse events in queue? */
Bram Moolenaar3a69e112014-01-10 13:51:42 +01001232 peek_console_input(g_hConIn, &ir, 1, &cRecords);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001233
1234 if (cRecords==0 || ir.EventType != MOUSE_EVENT)
1235 break;
1236 }
1237 else
1238 break;
1239 }
1240 }
1241 }
1242 }
1243 }
1244
1245 if (s_fNextIsMiddle)
1246 {
1247 nButton = (pmer->dwEventFlags == MOUSE_MOVED)
1248 ? MOUSE_DRAG : MOUSE_MIDDLE;
1249 s_fNextIsMiddle = FALSE;
1250 }
1251 else if (cButtons == 2 &&
1252 ((pmer->dwButtonState & LEFT_RIGHT) == LEFT_RIGHT))
1253 {
1254 nButton = MOUSE_MIDDLE;
1255
1256 if (! s_fReleased && pmer->dwEventFlags != MOUSE_MOVED)
1257 {
1258 s_fNextIsMiddle = TRUE;
1259 nButton = MOUSE_RELEASE;
1260 }
1261 }
1262 else if ((pmer->dwButtonState & LEFT) == LEFT)
1263 nButton = MOUSE_LEFT;
1264 else if ((pmer->dwButtonState & MIDDLE) == MIDDLE)
1265 nButton = MOUSE_MIDDLE;
1266 else if ((pmer->dwButtonState & RIGHT) == RIGHT)
1267 nButton = MOUSE_RIGHT;
1268
1269 if (! s_fReleased && ! s_fNextIsMiddle
1270 && nButton != s_nOldButton && s_nOldButton != MOUSE_DRAG)
1271 return FALSE;
1272
1273 s_fReleased = s_fNextIsMiddle;
1274 }
1275
1276 if (pmer->dwEventFlags == 0 || pmer->dwEventFlags == DOUBLE_CLICK)
1277 {
1278 /* button pressed or released, without mouse moving */
1279 if (nButton != -1 && nButton != MOUSE_RELEASE)
1280 {
1281 DWORD dwCurrentTime = GetTickCount();
1282
1283 if (s_xOldMouse != g_xMouse
1284 || s_yOldMouse != g_yMouse
1285 || s_nOldButton != nButton
1286 || s_old_topline != curwin->w_topline
1287#ifdef FEAT_DIFF
1288 || s_old_topfill != curwin->w_topfill
1289#endif
1290 || (int)(dwCurrentTime - s_dwLastClickTime) > p_mouset)
1291 {
1292 s_cClicks = 1;
1293 }
1294 else if (++s_cClicks > 4)
1295 {
1296 s_cClicks = 1;
1297 }
1298
1299 s_dwLastClickTime = dwCurrentTime;
1300 }
1301 }
1302 else if (pmer->dwEventFlags == MOUSE_MOVED)
1303 {
1304 if (nButton != -1 && nButton != MOUSE_RELEASE)
1305 nButton = MOUSE_DRAG;
1306
1307 s_cClicks = 1;
1308 }
1309
1310 if (nButton == -1)
1311 return FALSE;
1312
1313 if (nButton != MOUSE_RELEASE)
1314 s_nOldButton = nButton;
1315
1316 g_nMouseClick = nButton;
1317
1318 if (pmer->dwControlKeyState & SHIFT_PRESSED)
1319 g_nMouseClick |= MOUSE_SHIFT;
1320 if (pmer->dwControlKeyState & (RIGHT_CTRL_PRESSED | LEFT_CTRL_PRESSED))
1321 g_nMouseClick |= MOUSE_CTRL;
1322 if (pmer->dwControlKeyState & (RIGHT_ALT_PRESSED | LEFT_ALT_PRESSED))
1323 g_nMouseClick |= MOUSE_ALT;
1324
1325 if (nButton != MOUSE_DRAG && nButton != MOUSE_RELEASE)
1326 SET_NUM_MOUSE_CLICKS(g_nMouseClick, s_cClicks);
1327
1328 /* only pass on interesting (i.e., different) mouse events */
1329 if (s_xOldMouse == g_xMouse
1330 && s_yOldMouse == g_yMouse
1331 && s_nOldMouseClick == g_nMouseClick)
1332 {
1333 g_nMouseClick = -1;
1334 return FALSE;
1335 }
1336
1337 s_xOldMouse = g_xMouse;
1338 s_yOldMouse = g_yMouse;
1339 s_old_topline = curwin->w_topline;
1340#ifdef FEAT_DIFF
1341 s_old_topfill = curwin->w_topfill;
1342#endif
1343 s_nOldMouseClick = g_nMouseClick;
1344
1345 return TRUE;
1346}
1347
1348# endif /* FEAT_GUI_W32 */
1349#endif /* FEAT_MOUSE */
1350
1351
1352#ifdef MCH_CURSOR_SHAPE
1353/*
1354 * Set the shape of the cursor.
1355 * 'thickness' can be from 1 (thin) to 99 (block)
1356 */
1357 static void
1358mch_set_cursor_shape(int thickness)
1359{
1360 CONSOLE_CURSOR_INFO ConsoleCursorInfo;
1361 ConsoleCursorInfo.dwSize = thickness;
1362 ConsoleCursorInfo.bVisible = s_cursor_visible;
1363
1364 SetConsoleCursorInfo(g_hConOut, &ConsoleCursorInfo);
1365 if (s_cursor_visible)
1366 SetConsoleCursorPosition(g_hConOut, g_coord);
1367}
1368
1369 void
1370mch_update_cursor(void)
1371{
1372 int idx;
1373 int thickness;
1374
1375 /*
1376 * How the cursor is drawn depends on the current mode.
1377 */
1378 idx = get_shape_idx(FALSE);
1379
1380 if (shape_table[idx].shape == SHAPE_BLOCK)
1381 thickness = 99; /* 100 doesn't work on W95 */
1382 else
1383 thickness = shape_table[idx].percentage;
1384 mch_set_cursor_shape(thickness);
1385}
1386#endif
1387
1388#ifndef FEAT_GUI_W32 /* this isn't used for the GUI */
1389/*
1390 * Handle FOCUS_EVENT.
1391 */
1392 static void
1393handle_focus_event(INPUT_RECORD ir)
1394{
1395 g_fJustGotFocus = ir.Event.FocusEvent.bSetFocus;
1396 ui_focus_change((int)g_fJustGotFocus);
1397}
1398
1399/*
1400 * Wait until console input from keyboard or mouse is available,
1401 * or the time is up.
1402 * Return TRUE if something is available FALSE if not.
1403 */
1404 static int
1405WaitForChar(long msec)
1406{
1407 DWORD dwNow = 0, dwEndTime = 0;
1408 INPUT_RECORD ir;
1409 DWORD cRecords;
Bram Moolenaarac360bf2015-09-01 20:31:20 +02001410 WCHAR ch, ch2;
Bram Moolenaar4445f7e2016-04-20 20:55:56 +02001411#ifdef FEAT_TIMERS
Bram Moolenaar40b1b542016-04-20 20:18:23 +02001412 int tb_change_cnt = typebuf.tb_change_cnt;
Bram Moolenaar4445f7e2016-04-20 20:55:56 +02001413#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001414
1415 if (msec > 0)
1416 /* Wait until the specified time has elapsed. */
1417 dwEndTime = GetTickCount() + msec;
1418 else if (msec < 0)
1419 /* Wait forever. */
1420 dwEndTime = INFINITE;
1421
1422 /* We need to loop until the end of the time period, because
1423 * we might get multiple unusable mouse events in that time.
1424 */
1425 for (;;)
1426 {
Bram Moolenaarca568ae2016-02-01 21:32:58 +01001427#ifdef MESSAGE_QUEUE
1428 parse_queued_messages();
1429#endif
Bram Moolenaar325b7a22004-07-05 15:58:32 +00001430#ifdef FEAT_MZSCHEME
1431 mzvim_check_threads();
1432#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001433#ifdef FEAT_CLIENTSERVER
1434 serverProcessPendingMessages();
1435#endif
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001436
Bram Moolenaar071d4272004-06-13 20:20:40 +00001437 if (0
1438#ifdef FEAT_MOUSE
1439 || g_nMouseClick != -1
1440#endif
1441#ifdef FEAT_CLIENTSERVER
1442 || input_available()
1443#endif
1444 )
1445 return TRUE;
1446
1447 if (msec > 0)
1448 {
Bram Moolenaarb7512b72013-08-10 12:45:09 +02001449 /* If the specified wait time has passed, return. Beware that
1450 * GetTickCount() may wrap around (overflow). */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001451 dwNow = GetTickCount();
Bram Moolenaarb7512b72013-08-10 12:45:09 +02001452 if ((int)(dwNow - dwEndTime) >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001453 break;
1454 }
1455 if (msec != 0)
1456 {
Bram Moolenaar325b7a22004-07-05 15:58:32 +00001457 DWORD dwWaitTime = dwEndTime - dwNow;
1458
Bram Moolenaar509ce2a2016-03-11 22:52:15 +01001459#ifdef FEAT_JOB_CHANNEL
Bram Moolenaar8a8199e2016-11-26 15:13:33 +01001460 /* Check channel while waiting for input. */
Bram Moolenaar9186a272016-02-23 19:34:01 +01001461 if (dwWaitTime > 100)
Bram Moolenaar8a8199e2016-11-26 15:13:33 +01001462 {
Bram Moolenaar9186a272016-02-23 19:34:01 +01001463 dwWaitTime = 100;
Bram Moolenaar8a8199e2016-11-26 15:13:33 +01001464 /* If there is readahead then parse_queued_messages() timed out
1465 * and we should call it again soon. */
1466 if (channel_any_readahead())
1467 dwWaitTime = 10;
1468 }
Bram Moolenaar9186a272016-02-23 19:34:01 +01001469#endif
Bram Moolenaar325b7a22004-07-05 15:58:32 +00001470#ifdef FEAT_MZSCHEME
1471 if (mzthreads_allowed() && p_mzq > 0
1472 && (msec < 0 || (long)dwWaitTime > p_mzq))
1473 dwWaitTime = p_mzq; /* don't wait longer than 'mzquantum' */
1474#endif
Bram Moolenaar0e0b3dd2016-03-17 17:58:56 +01001475#ifdef FEAT_TIMERS
1476 {
1477 long due_time;
1478
1479 /* When waiting very briefly don't trigger timers. */
1480 if (dwWaitTime > 10)
1481 {
1482 /* Trigger timers and then get the time in msec until the
1483 * next one is due. Wait up to that time. */
1484 due_time = check_due_timer();
Bram Moolenaar40b1b542016-04-20 20:18:23 +02001485 if (typebuf.tb_change_cnt != tb_change_cnt)
1486 {
1487 /* timer may have used feedkeys() */
1488 return FALSE;
1489 }
Bram Moolenaar0e0b3dd2016-03-17 17:58:56 +01001490 if (due_time > 0 && dwWaitTime > (DWORD)due_time)
1491 dwWaitTime = due_time;
1492 }
1493 }
1494#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001495#ifdef FEAT_CLIENTSERVER
1496 /* Wait for either an event on the console input or a message in
1497 * the client-server window. */
Bram Moolenaarbb86ebb2015-08-04 19:27:05 +02001498 if (msg_wait_for_multiple_objects(1, &g_hConIn, FALSE,
Bram Moolenaar325b7a22004-07-05 15:58:32 +00001499 dwWaitTime, QS_SENDMESSAGE) != WAIT_OBJECT_0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001500#else
Bram Moolenaarbb86ebb2015-08-04 19:27:05 +02001501 if (wait_for_single_object(g_hConIn, dwWaitTime) != WAIT_OBJECT_0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001502#endif
1503 continue;
1504 }
1505
1506 cRecords = 0;
Bram Moolenaar3a69e112014-01-10 13:51:42 +01001507 peek_console_input(g_hConIn, &ir, 1, &cRecords);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001508
1509#ifdef FEAT_MBYTE_IME
1510 if (State & CMDLINE && msg_row == Rows - 1)
1511 {
1512 CONSOLE_SCREEN_BUFFER_INFO csbi;
1513
1514 if (GetConsoleScreenBufferInfo(g_hConOut, &csbi))
1515 {
1516 if (csbi.dwCursorPosition.Y != msg_row)
1517 {
1518 /* The screen is now messed up, must redraw the
1519 * command line and later all the windows. */
1520 redraw_all_later(CLEAR);
1521 cmdline_row -= (msg_row - csbi.dwCursorPosition.Y);
1522 redrawcmd();
1523 }
1524 }
1525 }
1526#endif
1527
1528 if (cRecords > 0)
1529 {
1530 if (ir.EventType == KEY_EVENT && ir.Event.KeyEvent.bKeyDown)
1531 {
1532#ifdef FEAT_MBYTE_IME
1533 /* Windows IME sends two '\n's with only one 'ENTER'. First:
1534 * wVirtualKeyCode == 13. second: wVirtualKeyCode == 0 */
Bram Moolenaarac360bf2015-09-01 20:31:20 +02001535 if (ir.Event.KeyEvent.UChar == 0
Bram Moolenaar071d4272004-06-13 20:20:40 +00001536 && ir.Event.KeyEvent.wVirtualKeyCode == 13)
1537 {
Bram Moolenaar3a69e112014-01-10 13:51:42 +01001538 read_console_input(g_hConIn, &ir, 1, &cRecords);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001539 continue;
1540 }
1541#endif
1542 if (decode_key_event(&ir.Event.KeyEvent, &ch, &ch2,
1543 NULL, FALSE))
1544 return TRUE;
1545 }
1546
Bram Moolenaar3a69e112014-01-10 13:51:42 +01001547 read_console_input(g_hConIn, &ir, 1, &cRecords);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001548
1549 if (ir.EventType == FOCUS_EVENT)
1550 handle_focus_event(ir);
1551 else if (ir.EventType == WINDOW_BUFFER_SIZE_EVENT)
1552 shell_resized();
1553#ifdef FEAT_MOUSE
1554 else if (ir.EventType == MOUSE_EVENT
1555 && decode_mouse_event(&ir.Event.MouseEvent))
1556 return TRUE;
1557#endif
1558 }
1559 else if (msec == 0)
1560 break;
1561 }
1562
1563#ifdef FEAT_CLIENTSERVER
1564 /* Something might have been received while we were waiting. */
1565 if (input_available())
1566 return TRUE;
1567#endif
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001568
Bram Moolenaar071d4272004-06-13 20:20:40 +00001569 return FALSE;
1570}
1571
1572#ifndef FEAT_GUI_MSWIN
1573/*
1574 * return non-zero if a character is available
1575 */
1576 int
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00001577mch_char_avail(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001578{
1579 return WaitForChar(0L);
1580}
1581#endif
1582
1583/*
1584 * Create the console input. Used when reading stdin doesn't work.
1585 */
1586 static void
1587create_conin(void)
1588{
1589 g_hConIn = CreateFile("CONIN$", GENERIC_READ|GENERIC_WRITE,
1590 FILE_SHARE_READ|FILE_SHARE_WRITE,
1591 (LPSECURITY_ATTRIBUTES) NULL,
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001592 OPEN_EXISTING, 0, (HANDLE)NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001593 did_create_conin = TRUE;
1594}
1595
1596/*
Bram Moolenaar0e0b3dd2016-03-17 17:58:56 +01001597 * Get a keystroke or a mouse event, use a blocking wait.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001598 */
Bram Moolenaarac360bf2015-09-01 20:31:20 +02001599 static WCHAR
1600tgetch(int *pmodifiers, WCHAR *pch2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001601{
Bram Moolenaarac360bf2015-09-01 20:31:20 +02001602 WCHAR ch;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001603
1604 for (;;)
1605 {
1606 INPUT_RECORD ir;
1607 DWORD cRecords = 0;
1608
1609#ifdef FEAT_CLIENTSERVER
1610 (void)WaitForChar(-1L);
1611 if (input_available())
1612 return 0;
1613# ifdef FEAT_MOUSE
1614 if (g_nMouseClick != -1)
1615 return 0;
1616# endif
1617#endif
Bram Moolenaar3a69e112014-01-10 13:51:42 +01001618 if (read_console_input(g_hConIn, &ir, 1, &cRecords) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001619 {
1620 if (did_create_conin)
1621 read_error_exit();
1622 create_conin();
1623 continue;
1624 }
1625
1626 if (ir.EventType == KEY_EVENT)
1627 {
1628 if (decode_key_event(&ir.Event.KeyEvent, &ch, pch2,
1629 pmodifiers, TRUE))
1630 return ch;
1631 }
1632 else if (ir.EventType == FOCUS_EVENT)
1633 handle_focus_event(ir);
1634 else if (ir.EventType == WINDOW_BUFFER_SIZE_EVENT)
1635 shell_resized();
1636#ifdef FEAT_MOUSE
1637 else if (ir.EventType == MOUSE_EVENT)
1638 {
1639 if (decode_mouse_event(&ir.Event.MouseEvent))
1640 return 0;
1641 }
1642#endif
1643 }
1644}
1645#endif /* !FEAT_GUI_W32 */
1646
1647
1648/*
Bram Moolenaarf6a2b082012-06-29 13:14:03 +02001649 * mch_inchar(): low-level input function.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001650 * Get one or more characters from the keyboard or the mouse.
1651 * If time == 0, do not wait for characters.
1652 * If time == n, wait a short time for characters.
1653 * If time == -1, wait forever for characters.
1654 * Returns the number of characters read into buf.
1655 */
1656 int
1657mch_inchar(
Bram Moolenaar1266d672017-02-01 13:43:36 +01001658 char_u *buf UNUSED,
1659 int maxlen UNUSED,
1660 long time UNUSED,
1661 int tb_change_cnt UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001662{
1663#ifndef FEAT_GUI_W32 /* this isn't used for the GUI */
1664
1665 int len;
1666 int c;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001667#define TYPEAHEADLEN 20
1668 static char_u typeahead[TYPEAHEADLEN]; /* previously typed bytes. */
1669 static int typeaheadlen = 0;
1670
1671 /* First use any typeahead that was kept because "buf" was too small. */
1672 if (typeaheadlen > 0)
1673 goto theend;
1674
Bram Moolenaar071d4272004-06-13 20:20:40 +00001675 if (time >= 0)
1676 {
1677 if (!WaitForChar(time)) /* no character available */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001678 return 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001679 }
1680 else /* time == -1, wait forever */
1681 {
1682 mch_set_winsize_now(); /* Allow winsize changes from now on */
1683
Bram Moolenaar3918c952005-03-15 22:34:55 +00001684 /*
1685 * If there is no character available within 2 seconds (default)
1686 * write the autoscript file to disk. Or cause the CursorHold event
1687 * to be triggered.
1688 */
1689 if (!WaitForChar(p_ut))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001690 {
1691#ifdef FEAT_AUTOCMD
Bram Moolenaard35f9712005-12-18 22:02:33 +00001692 if (trigger_cursorhold() && maxlen >= 3)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001693 {
Bram Moolenaar3918c952005-03-15 22:34:55 +00001694 buf[0] = K_SPECIAL;
1695 buf[1] = KS_EXTRA;
1696 buf[2] = (int)KE_CURSORHOLD;
1697 return 3;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001698 }
1699#endif
Bram Moolenaar702517d2005-06-27 22:34:07 +00001700 before_blocking();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001701 }
1702 }
1703
1704 /*
1705 * Try to read as many characters as there are, until the buffer is full.
1706 */
1707
1708 /* we will get at least one key. Get more if they are available. */
1709 g_fCBrkPressed = FALSE;
1710
1711#ifdef MCH_WRITE_DUMP
1712 if (fdDump)
1713 fputc('[', fdDump);
1714#endif
1715
1716 /* Keep looping until there is something in the typeahead buffer and more
1717 * to get and still room in the buffer (up to two bytes for a char and
1718 * three bytes for a modifier). */
1719 while ((typeaheadlen == 0 || WaitForChar(0L))
1720 && typeaheadlen + 5 <= TYPEAHEADLEN)
1721 {
1722 if (typebuf_changed(tb_change_cnt))
1723 {
1724 /* "buf" may be invalid now if a client put something in the
1725 * typeahead buffer and "buf" is in the typeahead buffer. */
1726 typeaheadlen = 0;
1727 break;
1728 }
1729#ifdef FEAT_MOUSE
1730 if (g_nMouseClick != -1)
1731 {
1732# ifdef MCH_WRITE_DUMP
1733 if (fdDump)
1734 fprintf(fdDump, "{%02x @ %d, %d}",
1735 g_nMouseClick, g_xMouse, g_yMouse);
1736# endif
1737 typeahead[typeaheadlen++] = ESC + 128;
1738 typeahead[typeaheadlen++] = 'M';
1739 typeahead[typeaheadlen++] = g_nMouseClick;
1740 typeahead[typeaheadlen++] = g_xMouse + '!';
1741 typeahead[typeaheadlen++] = g_yMouse + '!';
1742 g_nMouseClick = -1;
1743 }
1744 else
1745#endif
1746 {
Bram Moolenaarac360bf2015-09-01 20:31:20 +02001747 WCHAR ch2 = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001748 int modifiers = 0;
1749
1750 c = tgetch(&modifiers, &ch2);
1751
1752 if (typebuf_changed(tb_change_cnt))
1753 {
1754 /* "buf" may be invalid now if a client put something in the
1755 * typeahead buffer and "buf" is in the typeahead buffer. */
1756 typeaheadlen = 0;
1757 break;
1758 }
1759
1760 if (c == Ctrl_C && ctrl_c_interrupts)
1761 {
1762#if defined(FEAT_CLIENTSERVER)
1763 trash_input_buf();
1764#endif
1765 got_int = TRUE;
1766 }
1767
1768#ifdef FEAT_MOUSE
1769 if (g_nMouseClick == -1)
1770#endif
1771 {
1772 int n = 1;
Bram Moolenaar45500912014-07-09 20:51:07 +02001773 int conv = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001774
Bram Moolenaarac360bf2015-09-01 20:31:20 +02001775#ifdef FEAT_MBYTE
1776 if (ch2 == NUL)
1777 {
1778 int i;
1779 char_u *p;
1780 WCHAR ch[2];
1781
1782 ch[0] = c;
1783 if (c >= 0xD800 && c <= 0xDBFF) /* High surrogate */
1784 {
1785 ch[1] = tgetch(&modifiers, &ch2);
1786 n++;
1787 }
1788 p = utf16_to_enc(ch, &n);
1789 if (p != NULL)
1790 {
1791 for (i = 0; i < n; i++)
1792 typeahead[typeaheadlen + i] = p[i];
1793 vim_free(p);
1794 }
1795 }
1796 else
1797#endif
1798 typeahead[typeaheadlen] = c;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001799 if (ch2 != NUL)
1800 {
Bram Moolenaarac360bf2015-09-01 20:31:20 +02001801 typeahead[typeaheadlen + n] = 3;
1802 typeahead[typeaheadlen + n + 1] = (char_u)ch2;
Bram Moolenaar45500912014-07-09 20:51:07 +02001803 n += 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001804 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001805
Bram Moolenaar45500912014-07-09 20:51:07 +02001806 if (conv)
1807 {
1808 char_u *p = typeahead + typeaheadlen;
Bram Moolenaar45500912014-07-09 20:51:07 +02001809
Bram Moolenaar1ec4dd42015-01-20 19:39:35 +01001810 if (*p != K_NUL)
Bram Moolenaar45500912014-07-09 20:51:07 +02001811 {
Bram Moolenaar1ec4dd42015-01-20 19:39:35 +01001812 char_u *e = typeahead + TYPEAHEADLEN;
1813
1814 while (*p && p < e)
Bram Moolenaar45500912014-07-09 20:51:07 +02001815 {
Bram Moolenaar1ec4dd42015-01-20 19:39:35 +01001816 if (*p == K_NUL)
1817 {
1818 ++p;
1819 mch_memmove(p + 1, p, ((size_t)(e - p)) - 1);
1820 *p = 3;
1821 ++n;
1822 }
Bram Moolenaar45500912014-07-09 20:51:07 +02001823 ++p;
Bram Moolenaar45500912014-07-09 20:51:07 +02001824 }
Bram Moolenaar45500912014-07-09 20:51:07 +02001825 }
1826 }
1827
Bram Moolenaar071d4272004-06-13 20:20:40 +00001828 /* Use the ALT key to set the 8th bit of the character
1829 * when it's one byte, the 8th bit isn't set yet and not
1830 * using a double-byte encoding (would become a lead
1831 * byte). */
1832 if ((modifiers & MOD_MASK_ALT)
1833 && n == 1
1834 && (typeahead[typeaheadlen] & 0x80) == 0
1835#ifdef FEAT_MBYTE
1836 && !enc_dbcs
1837#endif
1838 )
1839 {
Bram Moolenaar85a3e5c2007-11-20 16:22:16 +00001840#ifdef FEAT_MBYTE
1841 n = (*mb_char2bytes)(typeahead[typeaheadlen] | 0x80,
1842 typeahead + typeaheadlen);
1843#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001844 typeahead[typeaheadlen] |= 0x80;
Bram Moolenaar85a3e5c2007-11-20 16:22:16 +00001845#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001846 modifiers &= ~MOD_MASK_ALT;
1847 }
1848
1849 if (modifiers != 0)
1850 {
1851 /* Prepend modifiers to the character. */
1852 mch_memmove(typeahead + typeaheadlen + 3,
1853 typeahead + typeaheadlen, n);
1854 typeahead[typeaheadlen++] = K_SPECIAL;
1855 typeahead[typeaheadlen++] = (char_u)KS_MODIFIER;
1856 typeahead[typeaheadlen++] = modifiers;
1857 }
1858
1859 typeaheadlen += n;
1860
1861#ifdef MCH_WRITE_DUMP
1862 if (fdDump)
1863 fputc(c, fdDump);
1864#endif
1865 }
1866 }
1867 }
1868
1869#ifdef MCH_WRITE_DUMP
1870 if (fdDump)
1871 {
1872 fputs("]\n", fdDump);
1873 fflush(fdDump);
1874 }
1875#endif
1876
Bram Moolenaar071d4272004-06-13 20:20:40 +00001877theend:
1878 /* Move typeahead to "buf", as much as fits. */
1879 len = 0;
1880 while (len < maxlen && typeaheadlen > 0)
1881 {
1882 buf[len++] = typeahead[0];
1883 mch_memmove(typeahead, typeahead + 1, --typeaheadlen);
1884 }
1885 return len;
1886
1887#else /* FEAT_GUI_W32 */
1888 return 0;
1889#endif /* FEAT_GUI_W32 */
1890}
1891
Bram Moolenaar82881492012-11-20 16:53:39 +01001892#ifndef PROTO
1893# ifndef __MINGW32__
1894# include <shellapi.h> /* required for FindExecutable() */
1895# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001896#endif
1897
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001898/*
1899 * Return TRUE if "name" is in $PATH.
Bram Moolenaarf4b8e572004-06-24 15:53:16 +00001900 * TODO: Should somehow check if it's really executable.
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001901 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001902 static int
Bram Moolenaarc7f02552014-04-01 21:00:59 +02001903executable_exists(char *name, char_u **path)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001904{
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001905 char *dum;
1906 char fname[_MAX_PATH];
Bram Moolenaarc40bdee2014-08-29 17:45:32 +02001907 char *curpath, *newpath;
1908 long n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001909
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001910#ifdef FEAT_MBYTE
1911 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001912 {
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001913 WCHAR *p = enc_to_utf16((char_u *)name, NULL);
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001914 WCHAR fnamew[_MAX_PATH];
1915 WCHAR *dumw;
Bram Moolenaarc40bdee2014-08-29 17:45:32 +02001916 WCHAR *wcurpath, *wnewpath;
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001917
1918 if (p != NULL)
1919 {
Bram Moolenaarc40bdee2014-08-29 17:45:32 +02001920 wcurpath = _wgetenv(L"PATH");
1921 wnewpath = (WCHAR*)alloc((unsigned)(wcslen(wcurpath) + 3)
1922 * sizeof(WCHAR));
1923 if (wnewpath == NULL)
1924 return FALSE;
1925 wcscpy(wnewpath, L".;");
1926 wcscat(wnewpath, wcurpath);
1927 n = (long)SearchPathW(wnewpath, p, NULL, _MAX_PATH, fnamew, &dumw);
1928 vim_free(wnewpath);
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001929 vim_free(p);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02001930 if (n == 0)
1931 return FALSE;
1932 if (GetFileAttributesW(fnamew) & FILE_ATTRIBUTE_DIRECTORY)
1933 return FALSE;
1934 if (path != NULL)
1935 *path = utf16_to_enc(fnamew, NULL);
1936 return TRUE;
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001937 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001938 }
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001939#endif
Bram Moolenaarc40bdee2014-08-29 17:45:32 +02001940
1941 curpath = getenv("PATH");
1942 newpath = (char*)alloc((unsigned)(STRLEN(curpath) + 3));
1943 if (newpath == NULL)
1944 return FALSE;
1945 STRCPY(newpath, ".;");
1946 STRCAT(newpath, curpath);
1947 n = (long)SearchPath(newpath, name, NULL, _MAX_PATH, fname, &dum);
1948 vim_free(newpath);
1949 if (n == 0)
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001950 return FALSE;
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001951 if (mch_isdir((char_u *)fname))
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001952 return FALSE;
Bram Moolenaarc7f02552014-04-01 21:00:59 +02001953 if (path != NULL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001954 *path = vim_strsave((char_u *)fname);
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00001955 return TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001956}
1957
Bram Moolenaard32a99a2010-09-21 17:29:23 +02001958#if ((defined(__MINGW32__) || defined (__CYGWIN32__)) && \
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02001959 __MSVCRT_VERSION__ >= 0x800) || (defined(_MSC_VER) && _MSC_VER >= 1400)
Bram Moolenaard32a99a2010-09-21 17:29:23 +02001960/*
1961 * Bad parameter handler.
1962 *
1963 * Certain MS CRT functions will intentionally crash when passed invalid
1964 * parameters to highlight possible security holes. Setting this function as
1965 * the bad parameter handler will prevent the crash.
1966 *
1967 * In debug builds the parameters contain CRT information that might help track
1968 * down the source of a problem, but in non-debug builds the arguments are all
1969 * NULL/0. Debug builds will also produce assert dialogs from the CRT, it is
1970 * worth allowing these to make debugging of issues easier.
1971 */
1972 static void
1973bad_param_handler(const wchar_t *expression,
1974 const wchar_t *function,
1975 const wchar_t *file,
1976 unsigned int line,
1977 uintptr_t pReserved)
1978{
1979}
1980
1981# define SET_INVALID_PARAM_HANDLER \
1982 ((void)_set_invalid_parameter_handler(bad_param_handler))
1983#else
1984# define SET_INVALID_PARAM_HANDLER
1985#endif
1986
Bram Moolenaar071d4272004-06-13 20:20:40 +00001987#ifdef FEAT_GUI_W32
1988
1989/*
1990 * GUI version of mch_init().
1991 */
1992 void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00001993mch_init(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001994{
1995#ifndef __MINGW32__
1996 extern int _fmode;
1997#endif
1998
Bram Moolenaard32a99a2010-09-21 17:29:23 +02001999 /* Silently handle invalid parameters to CRT functions */
2000 SET_INVALID_PARAM_HANDLER;
2001
Bram Moolenaar071d4272004-06-13 20:20:40 +00002002 /* Let critical errors result in a failure, not in a dialog box. Required
2003 * for the timestamp test to work on removed floppies. */
2004 SetErrorMode(SEM_FAILCRITICALERRORS);
2005
2006 _fmode = O_BINARY; /* we do our own CR-LF translation */
2007
2008 /* Specify window size. Is there a place to get the default from? */
2009 Rows = 25;
2010 Columns = 80;
2011
2012 /* Look for 'vimrun' */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002013 {
2014 char_u vimrun_location[_MAX_PATH + 4];
2015
2016 /* First try in same directory as gvim.exe */
2017 STRCPY(vimrun_location, exe_name);
2018 STRCPY(gettail(vimrun_location), "vimrun.exe");
2019 if (mch_getperm(vimrun_location) >= 0)
2020 {
2021 if (*skiptowhite(vimrun_location) != NUL)
2022 {
2023 /* Enclose path with white space in double quotes. */
2024 mch_memmove(vimrun_location + 1, vimrun_location,
2025 STRLEN(vimrun_location) + 1);
2026 *vimrun_location = '"';
2027 STRCPY(gettail(vimrun_location), "vimrun\" ");
2028 }
2029 else
2030 STRCPY(gettail(vimrun_location), "vimrun ");
2031
2032 vimrun_path = (char *)vim_strsave(vimrun_location);
2033 s_dont_use_vimrun = FALSE;
2034 }
Bram Moolenaarc7f02552014-04-01 21:00:59 +02002035 else if (executable_exists("vimrun.exe", NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002036 s_dont_use_vimrun = FALSE;
2037
2038 /* Don't give the warning for a missing vimrun.exe right now, but only
2039 * when vimrun was supposed to be used. Don't bother people that do
2040 * not need vimrun.exe. */
2041 if (s_dont_use_vimrun)
2042 need_vimrun_warning = TRUE;
2043 }
2044
2045 /*
2046 * If "finstr.exe" doesn't exist, use "grep -n" for 'grepprg'.
2047 * Otherwise the default "findstr /n" is used.
2048 */
Bram Moolenaarc7f02552014-04-01 21:00:59 +02002049 if (!executable_exists("findstr.exe", NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002050 set_option_value((char_u *)"grepprg", 0, (char_u *)"grep -n", 0);
2051
2052#ifdef FEAT_CLIPBOARD
Bram Moolenaar693e40c2013-02-26 14:56:42 +01002053 win_clip_init();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002054#endif
2055}
2056
2057
2058#else /* FEAT_GUI_W32 */
2059
2060#define SRWIDTH(sr) ((sr).Right - (sr).Left + 1)
2061#define SRHEIGHT(sr) ((sr).Bottom - (sr).Top + 1)
2062
2063/*
2064 * ClearConsoleBuffer()
2065 * Description:
2066 * Clears the entire contents of the console screen buffer, using the
2067 * specified attribute.
2068 * Returns:
2069 * TRUE on success
2070 */
2071 static BOOL
2072ClearConsoleBuffer(WORD wAttribute)
2073{
2074 CONSOLE_SCREEN_BUFFER_INFO csbi;
2075 COORD coord;
2076 DWORD NumCells, dummy;
2077
2078 if (!GetConsoleScreenBufferInfo(g_hConOut, &csbi))
2079 return FALSE;
2080
2081 NumCells = csbi.dwSize.X * csbi.dwSize.Y;
2082 coord.X = 0;
2083 coord.Y = 0;
2084 if (!FillConsoleOutputCharacter(g_hConOut, ' ', NumCells,
2085 coord, &dummy))
2086 {
2087 return FALSE;
2088 }
2089 if (!FillConsoleOutputAttribute(g_hConOut, wAttribute, NumCells,
2090 coord, &dummy))
2091 {
2092 return FALSE;
2093 }
2094
2095 return TRUE;
2096}
2097
2098/*
2099 * FitConsoleWindow()
2100 * Description:
2101 * Checks if the console window will fit within given buffer dimensions.
2102 * Also, if requested, will shrink the window to fit.
2103 * Returns:
2104 * TRUE on success
2105 */
2106 static BOOL
2107FitConsoleWindow(
2108 COORD dwBufferSize,
2109 BOOL WantAdjust)
2110{
2111 CONSOLE_SCREEN_BUFFER_INFO csbi;
2112 COORD dwWindowSize;
2113 BOOL NeedAdjust = FALSE;
2114
2115 if (GetConsoleScreenBufferInfo(g_hConOut, &csbi))
2116 {
2117 /*
2118 * A buffer resize will fail if the current console window does
2119 * not lie completely within that buffer. To avoid this, we might
2120 * have to move and possibly shrink the window.
2121 */
2122 if (csbi.srWindow.Right >= dwBufferSize.X)
2123 {
2124 dwWindowSize.X = SRWIDTH(csbi.srWindow);
2125 if (dwWindowSize.X > dwBufferSize.X)
2126 dwWindowSize.X = dwBufferSize.X;
2127 csbi.srWindow.Right = dwBufferSize.X - 1;
2128 csbi.srWindow.Left = dwBufferSize.X - dwWindowSize.X;
2129 NeedAdjust = TRUE;
2130 }
2131 if (csbi.srWindow.Bottom >= dwBufferSize.Y)
2132 {
2133 dwWindowSize.Y = SRHEIGHT(csbi.srWindow);
2134 if (dwWindowSize.Y > dwBufferSize.Y)
2135 dwWindowSize.Y = dwBufferSize.Y;
2136 csbi.srWindow.Bottom = dwBufferSize.Y - 1;
2137 csbi.srWindow.Top = dwBufferSize.Y - dwWindowSize.Y;
2138 NeedAdjust = TRUE;
2139 }
2140 if (NeedAdjust && WantAdjust)
2141 {
2142 if (!SetConsoleWindowInfo(g_hConOut, TRUE, &csbi.srWindow))
2143 return FALSE;
2144 }
2145 return TRUE;
2146 }
2147
2148 return FALSE;
2149}
2150
2151typedef struct ConsoleBufferStruct
2152{
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00002153 BOOL IsValid;
2154 CONSOLE_SCREEN_BUFFER_INFO Info;
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002155 PCHAR_INFO Buffer;
2156 COORD BufferSize;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002157} ConsoleBuffer;
2158
2159/*
2160 * SaveConsoleBuffer()
2161 * Description:
2162 * Saves important information about the console buffer, including the
2163 * actual buffer contents. The saved information is suitable for later
2164 * restoration by RestoreConsoleBuffer().
2165 * Returns:
2166 * TRUE if all information was saved; FALSE otherwise
2167 * If FALSE, still sets cb->IsValid if buffer characteristics were saved.
2168 */
2169 static BOOL
2170SaveConsoleBuffer(
2171 ConsoleBuffer *cb)
2172{
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002173 DWORD NumCells;
2174 COORD BufferCoord;
2175 SMALL_RECT ReadRegion;
2176 WORD Y, Y_incr;
2177
Bram Moolenaar071d4272004-06-13 20:20:40 +00002178 if (cb == NULL)
2179 return FALSE;
2180
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002181 if (!GetConsoleScreenBufferInfo(g_hConOut, &cb->Info))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002182 {
2183 cb->IsValid = FALSE;
2184 return FALSE;
2185 }
2186 cb->IsValid = TRUE;
2187
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002188 /*
2189 * Allocate a buffer large enough to hold the entire console screen
2190 * buffer. If this ConsoleBuffer structure has already been initialized
2191 * with a buffer of the correct size, then just use that one.
2192 */
2193 if (!cb->IsValid || cb->Buffer == NULL ||
2194 cb->BufferSize.X != cb->Info.dwSize.X ||
2195 cb->BufferSize.Y != cb->Info.dwSize.Y)
2196 {
2197 cb->BufferSize.X = cb->Info.dwSize.X;
2198 cb->BufferSize.Y = cb->Info.dwSize.Y;
2199 NumCells = cb->BufferSize.X * cb->BufferSize.Y;
2200 vim_free(cb->Buffer);
2201 cb->Buffer = (PCHAR_INFO)alloc(NumCells * sizeof(CHAR_INFO));
2202 if (cb->Buffer == NULL)
2203 return FALSE;
2204 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002205
2206 /*
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002207 * We will now copy the console screen buffer into our buffer.
2208 * ReadConsoleOutput() seems to be limited as far as how much you
2209 * can read at a time. Empirically, this number seems to be about
2210 * 12000 cells (rows * columns). Start at position (0, 0) and copy
2211 * in chunks until it is all copied. The chunks will all have the
2212 * same horizontal characteristics, so initialize them now. The
2213 * height of each chunk will be (12000 / width).
Bram Moolenaar071d4272004-06-13 20:20:40 +00002214 */
Bram Moolenaar61594242015-09-01 20:23:37 +02002215 BufferCoord.X = 0;
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002216 ReadRegion.Left = 0;
2217 ReadRegion.Right = cb->Info.dwSize.X - 1;
2218 Y_incr = 12000 / cb->Info.dwSize.X;
2219 for (Y = 0; Y < cb->BufferSize.Y; Y += Y_incr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002220 {
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002221 /*
2222 * Read into position (0, Y) in our buffer.
2223 */
2224 BufferCoord.Y = Y;
2225 /*
2226 * Read the region whose top left corner is (0, Y) and whose bottom
2227 * right corner is (width - 1, Y + Y_incr - 1). This should define
2228 * a region of size width by Y_incr. Don't worry if this region is
2229 * too large for the remaining buffer; it will be cropped.
2230 */
2231 ReadRegion.Top = Y;
2232 ReadRegion.Bottom = Y + Y_incr - 1;
2233 if (!ReadConsoleOutput(g_hConOut, /* output handle */
2234 cb->Buffer, /* our buffer */
2235 cb->BufferSize, /* dimensions of our buffer */
2236 BufferCoord, /* offset in our buffer */
2237 &ReadRegion)) /* region to save */
2238 {
2239 vim_free(cb->Buffer);
2240 cb->Buffer = NULL;
2241 return FALSE;
2242 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002243 }
2244
2245 return TRUE;
2246}
2247
2248/*
2249 * RestoreConsoleBuffer()
2250 * Description:
2251 * Restores important information about the console buffer, including the
2252 * actual buffer contents, if desired. The information to restore is in
2253 * the same format used by SaveConsoleBuffer().
2254 * Returns:
2255 * TRUE on success
2256 */
2257 static BOOL
2258RestoreConsoleBuffer(
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00002259 ConsoleBuffer *cb,
2260 BOOL RestoreScreen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002261{
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002262 COORD BufferCoord;
2263 SMALL_RECT WriteRegion;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002264
2265 if (cb == NULL || !cb->IsValid)
2266 return FALSE;
2267
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002268 /*
2269 * Before restoring the buffer contents, clear the current buffer, and
2270 * restore the cursor position and window information. Doing this now
2271 * prevents old buffer contents from "flashing" onto the screen.
2272 */
2273 if (RestoreScreen)
2274 ClearConsoleBuffer(cb->Info.wAttributes);
2275
2276 FitConsoleWindow(cb->Info.dwSize, TRUE);
2277 if (!SetConsoleScreenBufferSize(g_hConOut, cb->Info.dwSize))
2278 return FALSE;
2279 if (!SetConsoleTextAttribute(g_hConOut, cb->Info.wAttributes))
2280 return FALSE;
2281
2282 if (!RestoreScreen)
2283 {
2284 /*
2285 * No need to restore the screen buffer contents, so we're done.
2286 */
2287 return TRUE;
2288 }
2289
2290 if (!SetConsoleCursorPosition(g_hConOut, cb->Info.dwCursorPosition))
2291 return FALSE;
2292 if (!SetConsoleWindowInfo(g_hConOut, TRUE, &cb->Info.srWindow))
2293 return FALSE;
2294
2295 /*
2296 * Restore the screen buffer contents.
2297 */
2298 if (cb->Buffer != NULL)
2299 {
2300 BufferCoord.X = 0;
2301 BufferCoord.Y = 0;
2302 WriteRegion.Left = 0;
2303 WriteRegion.Top = 0;
2304 WriteRegion.Right = cb->Info.dwSize.X - 1;
2305 WriteRegion.Bottom = cb->Info.dwSize.Y - 1;
2306 if (!WriteConsoleOutput(g_hConOut, /* output handle */
2307 cb->Buffer, /* our buffer */
2308 cb->BufferSize, /* dimensions of our buffer */
2309 BufferCoord, /* offset in our buffer */
2310 &WriteRegion)) /* region to restore */
2311 {
2312 return FALSE;
2313 }
2314 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002315
2316 return TRUE;
2317}
2318
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002319#define FEAT_RESTORE_ORIG_SCREEN
2320#ifdef FEAT_RESTORE_ORIG_SCREEN
2321static ConsoleBuffer g_cbOrig = { 0 };
2322#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002323static ConsoleBuffer g_cbNonTermcap = { 0 };
2324static ConsoleBuffer g_cbTermcap = { 0 };
2325
2326#ifdef FEAT_TITLE
2327#ifdef __BORLANDC__
2328typedef HWND (__stdcall *GETCONSOLEWINDOWPROC)(VOID);
2329#else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002330typedef HWND (WINAPI *GETCONSOLEWINDOWPROC)(VOID);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002331#endif
2332char g_szOrigTitle[256] = { 0 };
2333HWND g_hWnd = NULL; /* also used in os_mswin.c */
2334static HICON g_hOrigIconSmall = NULL;
2335static HICON g_hOrigIcon = NULL;
2336static HICON g_hVimIcon = NULL;
2337static BOOL g_fCanChangeIcon = FALSE;
2338
2339/* ICON* are not defined in VC++ 4.0 */
2340#ifndef ICON_SMALL
2341#define ICON_SMALL 0
2342#endif
2343#ifndef ICON_BIG
2344#define ICON_BIG 1
2345#endif
2346/*
2347 * GetConsoleIcon()
2348 * Description:
2349 * Attempts to retrieve the small icon and/or the big icon currently in
2350 * use by a given window.
2351 * Returns:
2352 * TRUE on success
2353 */
2354 static BOOL
2355GetConsoleIcon(
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00002356 HWND hWnd,
2357 HICON *phIconSmall,
2358 HICON *phIcon)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002359{
2360 if (hWnd == NULL)
2361 return FALSE;
2362
2363 if (phIconSmall != NULL)
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00002364 *phIconSmall = (HICON)SendMessage(hWnd, WM_GETICON,
2365 (WPARAM)ICON_SMALL, (LPARAM)0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002366 if (phIcon != NULL)
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00002367 *phIcon = (HICON)SendMessage(hWnd, WM_GETICON,
2368 (WPARAM)ICON_BIG, (LPARAM)0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002369 return TRUE;
2370}
2371
2372/*
2373 * SetConsoleIcon()
2374 * Description:
2375 * Attempts to change the small icon and/or the big icon currently in
2376 * use by a given window.
2377 * Returns:
2378 * TRUE on success
2379 */
2380 static BOOL
2381SetConsoleIcon(
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00002382 HWND hWnd,
2383 HICON hIconSmall,
2384 HICON hIcon)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002385{
Bram Moolenaar071d4272004-06-13 20:20:40 +00002386 if (hWnd == NULL)
2387 return FALSE;
2388
2389 if (hIconSmall != NULL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002390 SendMessage(hWnd, WM_SETICON,
2391 (WPARAM)ICON_SMALL, (LPARAM)hIconSmall);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002392 if (hIcon != NULL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002393 SendMessage(hWnd, WM_SETICON,
2394 (WPARAM)ICON_BIG, (LPARAM) hIcon);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002395 return TRUE;
2396}
2397
2398/*
2399 * SaveConsoleTitleAndIcon()
2400 * Description:
2401 * Saves the current console window title in g_szOrigTitle, for later
2402 * restoration. Also, attempts to obtain a handle to the console window,
2403 * and use it to save the small and big icons currently in use by the
2404 * console window. This is not always possible on some versions of Windows;
2405 * nor is it possible when running Vim remotely using Telnet (since the
2406 * console window the user sees is owned by a remote process).
2407 */
2408 static void
2409SaveConsoleTitleAndIcon(void)
2410{
Bram Moolenaar071d4272004-06-13 20:20:40 +00002411 /* Save the original title. */
2412 if (!GetConsoleTitle(g_szOrigTitle, sizeof(g_szOrigTitle)))
2413 return;
2414
2415 /*
2416 * Obtain a handle to the console window using GetConsoleWindow() from
2417 * KERNEL32.DLL; we need to handle in order to change the window icon.
2418 * This function only exists on NT-based Windows, starting with Windows
2419 * 2000. On older operating systems, we can't change the window icon
2420 * anyway.
2421 */
Bram Moolenaarcea912a2016-10-12 14:20:24 +02002422 g_hWnd = GetConsoleWindow();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002423 if (g_hWnd == NULL)
2424 return;
2425
2426 /* Save the original console window icon. */
2427 GetConsoleIcon(g_hWnd, &g_hOrigIconSmall, &g_hOrigIcon);
2428 if (g_hOrigIconSmall == NULL || g_hOrigIcon == NULL)
2429 return;
2430
2431 /* Extract the first icon contained in the Vim executable. */
Bram Moolenaarcddc91c2014-09-23 21:53:41 +02002432 if (mch_icon_load((HANDLE *)&g_hVimIcon) == FAIL || g_hVimIcon == NULL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002433 g_hVimIcon = ExtractIcon(NULL, (LPCSTR)exe_name, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002434 if (g_hVimIcon != NULL)
2435 g_fCanChangeIcon = TRUE;
2436}
2437#endif
2438
2439static int g_fWindInitCalled = FALSE;
2440static int g_fTermcapMode = FALSE;
2441static CONSOLE_CURSOR_INFO g_cci;
2442static DWORD g_cmodein = 0;
2443static DWORD g_cmodeout = 0;
2444
2445/*
2446 * non-GUI version of mch_init().
2447 */
2448 void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00002449mch_init(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002450{
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002451#ifndef FEAT_RESTORE_ORIG_SCREEN
2452 CONSOLE_SCREEN_BUFFER_INFO csbi;
2453#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002454#ifndef __MINGW32__
2455 extern int _fmode;
2456#endif
2457
Bram Moolenaard32a99a2010-09-21 17:29:23 +02002458 /* Silently handle invalid parameters to CRT functions */
2459 SET_INVALID_PARAM_HANDLER;
2460
Bram Moolenaar071d4272004-06-13 20:20:40 +00002461 /* Let critical errors result in a failure, not in a dialog box. Required
2462 * for the timestamp test to work on removed floppies. */
2463 SetErrorMode(SEM_FAILCRITICALERRORS);
2464
2465 _fmode = O_BINARY; /* we do our own CR-LF translation */
2466 out_flush();
2467
2468 /* Obtain handles for the standard Console I/O devices */
2469 if (read_cmd_fd == 0)
2470 g_hConIn = GetStdHandle(STD_INPUT_HANDLE);
2471 else
2472 create_conin();
2473 g_hConOut = GetStdHandle(STD_OUTPUT_HANDLE);
2474
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002475#ifdef FEAT_RESTORE_ORIG_SCREEN
2476 /* Save the initial console buffer for later restoration */
2477 SaveConsoleBuffer(&g_cbOrig);
2478 g_attrCurrent = g_attrDefault = g_cbOrig.Info.wAttributes;
2479#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002480 /* Get current text attributes */
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01002481 GetConsoleScreenBufferInfo(g_hConOut, &csbi);
2482 g_attrCurrent = g_attrDefault = csbi.wAttributes;
2483#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002484 if (cterm_normal_fg_color == 0)
2485 cterm_normal_fg_color = (g_attrCurrent & 0xf) + 1;
2486 if (cterm_normal_bg_color == 0)
2487 cterm_normal_bg_color = ((g_attrCurrent >> 4) & 0xf) + 1;
2488
2489 /* set termcap codes to current text attributes */
2490 update_tcap(g_attrCurrent);
2491
2492 GetConsoleCursorInfo(g_hConOut, &g_cci);
2493 GetConsoleMode(g_hConIn, &g_cmodein);
2494 GetConsoleMode(g_hConOut, &g_cmodeout);
2495
2496#ifdef FEAT_TITLE
2497 SaveConsoleTitleAndIcon();
2498 /*
2499 * Set both the small and big icons of the console window to Vim's icon.
2500 * Note that Vim presently only has one size of icon (32x32), but it
2501 * automatically gets scaled down to 16x16 when setting the small icon.
2502 */
2503 if (g_fCanChangeIcon)
2504 SetConsoleIcon(g_hWnd, g_hVimIcon, g_hVimIcon);
2505#endif
2506
2507 ui_get_shellsize();
2508
2509#ifdef MCH_WRITE_DUMP
2510 fdDump = fopen("dump", "wt");
2511
2512 if (fdDump)
2513 {
2514 time_t t;
2515
2516 time(&t);
2517 fputs(ctime(&t), fdDump);
2518 fflush(fdDump);
2519 }
2520#endif
2521
2522 g_fWindInitCalled = TRUE;
2523
2524#ifdef FEAT_MOUSE
2525 g_fMouseAvail = GetSystemMetrics(SM_MOUSEPRESENT);
2526#endif
2527
2528#ifdef FEAT_CLIPBOARD
Bram Moolenaar693e40c2013-02-26 14:56:42 +01002529 win_clip_init();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002530#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002531}
2532
2533/*
2534 * non-GUI version of mch_exit().
2535 * Shut down and exit with status `r'
2536 * Careful: mch_exit() may be called before mch_init()!
2537 */
2538 void
2539mch_exit(int r)
2540{
2541 stoptermcap();
2542
2543 if (g_fWindInitCalled)
2544 settmode(TMODE_COOK);
2545
2546 ml_close_all(TRUE); /* remove all memfiles */
2547
2548 if (g_fWindInitCalled)
2549 {
2550#ifdef FEAT_TITLE
2551 mch_restore_title(3);
2552 /*
2553 * Restore both the small and big icons of the console window to
2554 * what they were at startup. Don't do this when the window is
2555 * closed, Vim would hang here.
2556 */
2557 if (g_fCanChangeIcon && !g_fForceExit)
2558 SetConsoleIcon(g_hWnd, g_hOrigIconSmall, g_hOrigIcon);
2559#endif
2560
2561#ifdef MCH_WRITE_DUMP
2562 if (fdDump)
2563 {
2564 time_t t;
2565
2566 time(&t);
2567 fputs(ctime(&t), fdDump);
2568 fclose(fdDump);
2569 }
2570 fdDump = NULL;
2571#endif
2572 }
2573
2574 SetConsoleCursorInfo(g_hConOut, &g_cci);
2575 SetConsoleMode(g_hConIn, g_cmodein);
2576 SetConsoleMode(g_hConOut, g_cmodeout);
2577
2578#ifdef DYNAMIC_GETTEXT
2579 dyn_libintl_end();
2580#endif
2581
2582 exit(r);
2583}
2584#endif /* !FEAT_GUI_W32 */
2585
Bram Moolenaar071d4272004-06-13 20:20:40 +00002586/*
2587 * Do we have an interactive window?
2588 */
2589 int
2590mch_check_win(
Bram Moolenaar1266d672017-02-01 13:43:36 +01002591 int argc UNUSED,
2592 char **argv UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002593{
2594 get_exe_name();
2595
2596#ifdef FEAT_GUI_W32
2597 return OK; /* GUI always has a tty */
2598#else
2599 if (isatty(1))
2600 return OK;
2601 return FAIL;
2602#endif
2603}
2604
2605
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002606#ifdef FEAT_MBYTE
2607/*
2608 * fname_casew(): Wide version of fname_case(). Set the case of the file name,
2609 * if it already exists. When "len" is > 0, also expand short to long
2610 * filenames.
2611 * Return FAIL if wide functions are not available, OK otherwise.
2612 * NOTE: much of this is identical to fname_case(), keep in sync!
2613 */
2614 static int
2615fname_casew(
2616 WCHAR *name,
2617 int len)
2618{
2619 WCHAR szTrueName[_MAX_PATH + 2];
2620 WCHAR szTrueNameTemp[_MAX_PATH + 2];
2621 WCHAR *ptrue, *ptruePrev;
2622 WCHAR *porig, *porigPrev;
2623 int flen;
2624 WIN32_FIND_DATAW fb;
Bram Moolenaar73c61632013-12-07 14:48:10 +01002625 HANDLE hFind = INVALID_HANDLE_VALUE;
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002626 int c;
2627 int slen;
2628
2629 flen = (int)wcslen(name);
2630 if (flen > _MAX_PATH)
2631 return OK;
2632
2633 /* slash_adjust(name) not needed, already adjusted by fname_case(). */
2634
2635 /* Build the new name in szTrueName[] one component at a time. */
2636 porig = name;
2637 ptrue = szTrueName;
2638
2639 if (iswalpha(porig[0]) && porig[1] == L':')
2640 {
2641 /* copy leading drive letter */
2642 *ptrue++ = *porig++;
2643 *ptrue++ = *porig++;
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002644 }
Bram Moolenaar73c61632013-12-07 14:48:10 +01002645 *ptrue = NUL; /* in case nothing follows */
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002646
2647 while (*porig != NUL)
2648 {
2649 /* copy \ characters */
2650 while (*porig == psepc)
2651 *ptrue++ = *porig++;
2652
2653 ptruePrev = ptrue;
2654 porigPrev = porig;
2655 while (*porig != NUL && *porig != psepc)
2656 {
2657 *ptrue++ = *porig++;
2658 }
2659 *ptrue = NUL;
2660
2661 /* To avoid a slow failure append "\*" when searching a directory,
2662 * server or network share. */
2663 wcscpy(szTrueNameTemp, szTrueName);
2664 slen = (int)wcslen(szTrueNameTemp);
2665 if (*porig == psepc && slen + 2 < _MAX_PATH)
2666 wcscpy(szTrueNameTemp + slen, L"\\*");
2667
2668 /* Skip "", "." and "..". */
2669 if (ptrue > ptruePrev
2670 && (ptruePrev[0] != L'.'
2671 || (ptruePrev[1] != NUL
2672 && (ptruePrev[1] != L'.' || ptruePrev[2] != NUL)))
2673 && (hFind = FindFirstFileW(szTrueNameTemp, &fb))
2674 != INVALID_HANDLE_VALUE)
2675 {
2676 c = *porig;
2677 *porig = NUL;
2678
2679 /* Only use the match when it's the same name (ignoring case) or
2680 * expansion is allowed and there is a match with the short name
2681 * and there is enough room. */
2682 if (_wcsicoll(porigPrev, fb.cFileName) == 0
2683 || (len > 0
2684 && (_wcsicoll(porigPrev, fb.cAlternateFileName) == 0
2685 && (int)(ptruePrev - szTrueName)
2686 + (int)wcslen(fb.cFileName) < len)))
2687 {
2688 wcscpy(ptruePrev, fb.cFileName);
2689
2690 /* Look for exact match and prefer it if found. Must be a
2691 * long name, otherwise there would be only one match. */
2692 while (FindNextFileW(hFind, &fb))
2693 {
2694 if (*fb.cAlternateFileName != NUL
2695 && (wcscoll(porigPrev, fb.cFileName) == 0
2696 || (len > 0
2697 && (_wcsicoll(porigPrev,
2698 fb.cAlternateFileName) == 0
2699 && (int)(ptruePrev - szTrueName)
2700 + (int)wcslen(fb.cFileName) < len))))
2701 {
2702 wcscpy(ptruePrev, fb.cFileName);
2703 break;
2704 }
2705 }
2706 }
2707 FindClose(hFind);
2708 *porig = c;
2709 ptrue = ptruePrev + wcslen(ptruePrev);
2710 }
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002711 }
2712
2713 wcscpy(name, szTrueName);
2714 return OK;
2715}
2716#endif
2717
Bram Moolenaar071d4272004-06-13 20:20:40 +00002718/*
2719 * fname_case(): Set the case of the file name, if it already exists.
2720 * When "len" is > 0, also expand short to long filenames.
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002721 * NOTE: much of this is identical to fname_casew(), keep in sync!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002722 */
2723 void
2724fname_case(
2725 char_u *name,
2726 int len)
2727{
2728 char szTrueName[_MAX_PATH + 2];
Bram Moolenaar464c9252010-10-13 20:37:41 +02002729 char szTrueNameTemp[_MAX_PATH + 2];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002730 char *ptrue, *ptruePrev;
2731 char *porig, *porigPrev;
2732 int flen;
2733 WIN32_FIND_DATA fb;
2734 HANDLE hFind;
2735 int c;
Bram Moolenaar464c9252010-10-13 20:37:41 +02002736 int slen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002737
Bram Moolenaara3ffd9c2005-07-21 21:03:15 +00002738 flen = (int)STRLEN(name);
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002739 if (flen == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002740 return;
2741
2742 slash_adjust(name);
2743
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002744#ifdef FEAT_MBYTE
2745 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
2746 {
2747 WCHAR *p = enc_to_utf16(name, NULL);
2748
2749 if (p != NULL)
2750 {
2751 char_u *q;
Bram Moolenaar21d89b62014-10-07 10:38:40 +02002752 WCHAR buf[_MAX_PATH + 1];
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002753
Bram Moolenaar21d89b62014-10-07 10:38:40 +02002754 wcsncpy(buf, p, _MAX_PATH);
2755 buf[_MAX_PATH] = L'\0';
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002756 vim_free(p);
2757
2758 if (fname_casew(buf, (len > 0) ? _MAX_PATH : 0) == OK)
2759 {
2760 q = utf16_to_enc(buf, NULL);
2761 if (q != NULL)
2762 {
2763 vim_strncpy(name, q, (len > 0) ? len - 1 : flen);
2764 vim_free(q);
2765 return;
2766 }
2767 }
2768 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02002769 return;
Bram Moolenaar65f04f62013-08-30 17:29:16 +02002770 }
2771#endif
2772
2773 /* If 'enc' is utf-8, flen can be larger than _MAX_PATH.
2774 * So we should check this after calling wide function. */
2775 if (flen > _MAX_PATH)
2776 return;
2777
Bram Moolenaar071d4272004-06-13 20:20:40 +00002778 /* Build the new name in szTrueName[] one component at a time. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002779 porig = (char *)name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002780 ptrue = szTrueName;
2781
2782 if (isalpha(porig[0]) && porig[1] == ':')
2783 {
2784 /* copy leading drive letter */
2785 *ptrue++ = *porig++;
2786 *ptrue++ = *porig++;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002787 }
Bram Moolenaar73c61632013-12-07 14:48:10 +01002788 *ptrue = NUL; /* in case nothing follows */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002789
2790 while (*porig != NUL)
2791 {
2792 /* copy \ characters */
2793 while (*porig == psepc)
2794 *ptrue++ = *porig++;
2795
2796 ptruePrev = ptrue;
2797 porigPrev = porig;
2798 while (*porig != NUL && *porig != psepc)
2799 {
2800#ifdef FEAT_MBYTE
2801 int l;
2802
2803 if (enc_dbcs)
2804 {
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002805 l = (*mb_ptr2len)((char_u *)porig);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002806 while (--l >= 0)
2807 *ptrue++ = *porig++;
2808 }
2809 else
2810#endif
2811 *ptrue++ = *porig++;
2812 }
2813 *ptrue = NUL;
2814
Bram Moolenaar464c9252010-10-13 20:37:41 +02002815 /* To avoid a slow failure append "\*" when searching a directory,
2816 * server or network share. */
2817 STRCPY(szTrueNameTemp, szTrueName);
Bram Moolenaar6b5ef062010-10-27 12:18:00 +02002818 slen = (int)strlen(szTrueNameTemp);
Bram Moolenaar464c9252010-10-13 20:37:41 +02002819 if (*porig == psepc && slen + 2 < _MAX_PATH)
2820 STRCPY(szTrueNameTemp + slen, "\\*");
2821
Bram Moolenaar071d4272004-06-13 20:20:40 +00002822 /* Skip "", "." and "..". */
2823 if (ptrue > ptruePrev
2824 && (ptruePrev[0] != '.'
2825 || (ptruePrev[1] != NUL
2826 && (ptruePrev[1] != '.' || ptruePrev[2] != NUL)))
Bram Moolenaar464c9252010-10-13 20:37:41 +02002827 && (hFind = FindFirstFile(szTrueNameTemp, &fb))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002828 != INVALID_HANDLE_VALUE)
2829 {
2830 c = *porig;
2831 *porig = NUL;
2832
2833 /* Only use the match when it's the same name (ignoring case) or
2834 * expansion is allowed and there is a match with the short name
2835 * and there is enough room. */
2836 if (_stricoll(porigPrev, fb.cFileName) == 0
2837 || (len > 0
2838 && (_stricoll(porigPrev, fb.cAlternateFileName) == 0
2839 && (int)(ptruePrev - szTrueName)
2840 + (int)strlen(fb.cFileName) < len)))
2841 {
2842 STRCPY(ptruePrev, fb.cFileName);
2843
2844 /* Look for exact match and prefer it if found. Must be a
2845 * long name, otherwise there would be only one match. */
2846 while (FindNextFile(hFind, &fb))
2847 {
2848 if (*fb.cAlternateFileName != NUL
2849 && (strcoll(porigPrev, fb.cFileName) == 0
2850 || (len > 0
2851 && (_stricoll(porigPrev,
2852 fb.cAlternateFileName) == 0
2853 && (int)(ptruePrev - szTrueName)
2854 + (int)strlen(fb.cFileName) < len))))
2855 {
2856 STRCPY(ptruePrev, fb.cFileName);
2857 break;
2858 }
2859 }
2860 }
2861 FindClose(hFind);
2862 *porig = c;
2863 ptrue = ptruePrev + strlen(ptruePrev);
2864 }
2865 }
2866
2867 STRCPY(name, szTrueName);
2868}
2869
2870
2871/*
2872 * Insert user name in s[len].
2873 */
2874 int
2875mch_get_user_name(
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00002876 char_u *s,
2877 int len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002878{
Bram Moolenaar41a09032007-10-01 18:34:34 +00002879 char szUserName[256 + 1]; /* UNLEN is 256 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002880 DWORD cch = sizeof szUserName;
2881
Bram Moolenaarc8020ee2013-12-11 18:18:06 +01002882#ifdef FEAT_MBYTE
2883 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
2884 {
2885 WCHAR wszUserName[256 + 1]; /* UNLEN is 256 */
2886 DWORD wcch = sizeof(wszUserName) / sizeof(WCHAR);
2887
2888 if (GetUserNameW(wszUserName, &wcch))
2889 {
2890 char_u *p = utf16_to_enc(wszUserName, NULL);
2891
2892 if (p != NULL)
2893 {
2894 vim_strncpy(s, p, len - 1);
2895 vim_free(p);
2896 return OK;
2897 }
2898 }
Bram Moolenaarc8020ee2013-12-11 18:18:06 +01002899 }
2900#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002901 if (GetUserName(szUserName, &cch))
2902 {
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002903 vim_strncpy(s, (char_u *)szUserName, len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002904 return OK;
2905 }
2906 s[0] = NUL;
2907 return FAIL;
2908}
2909
2910
2911/*
2912 * Insert host name in s[len].
2913 */
2914 void
2915mch_get_host_name(
2916 char_u *s,
2917 int len)
2918{
2919 DWORD cch = len;
2920
Bram Moolenaar2cc87382013-12-11 18:21:45 +01002921#ifdef FEAT_MBYTE
2922 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
2923 {
2924 WCHAR wszHostName[256 + 1];
2925 DWORD wcch = sizeof(wszHostName) / sizeof(WCHAR);
2926
2927 if (GetComputerNameW(wszHostName, &wcch))
2928 {
2929 char_u *p = utf16_to_enc(wszHostName, NULL);
2930
2931 if (p != NULL)
2932 {
2933 vim_strncpy(s, p, len - 1);
2934 vim_free(p);
2935 return;
2936 }
2937 }
Bram Moolenaar2cc87382013-12-11 18:21:45 +01002938 }
2939#endif
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002940 if (!GetComputerName((LPSTR)s, &cch))
2941 vim_strncpy(s, (char_u *)"PC (Win32 Vim)", len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002942}
2943
2944
2945/*
2946 * return process ID
2947 */
2948 long
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00002949mch_get_pid(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002950{
2951 return (long)GetCurrentProcessId();
2952}
2953
2954
2955/*
2956 * Get name of current directory into buffer 'buf' of length 'len' bytes.
2957 * Return OK for success, FAIL for failure.
2958 */
2959 int
2960mch_dirname(
2961 char_u *buf,
2962 int len)
2963{
2964 /*
2965 * Originally this was:
2966 * return (getcwd(buf, len) != NULL ? OK : FAIL);
2967 * But the Win32s known bug list says that getcwd() doesn't work
2968 * so use the Win32 system call instead. <Negri>
2969 */
2970#ifdef FEAT_MBYTE
2971 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
2972 {
2973 WCHAR wbuf[_MAX_PATH + 1];
2974
2975 if (GetCurrentDirectoryW(_MAX_PATH, wbuf) != 0)
2976 {
Bram Moolenaar36f692d2008-11-20 16:10:17 +00002977 char_u *p = utf16_to_enc(wbuf, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002978
2979 if (p != NULL)
2980 {
Bram Moolenaarfe3ca8d2005-07-18 21:43:02 +00002981 vim_strncpy(buf, p, len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002982 vim_free(p);
2983 return OK;
2984 }
2985 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02002986 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002987 }
2988#endif
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002989 return (GetCurrentDirectory(len, (LPSTR)buf) != 0 ? OK : FAIL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002990}
2991
2992/*
Bram Moolenaarffa22202013-11-21 12:34:11 +01002993 * Get file permissions for "name".
2994 * Return mode_t or -1 for error.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002995 */
2996 long
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00002997mch_getperm(char_u *name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002998{
Bram Moolenaar8767f522016-07-01 17:17:39 +02002999 stat_T st;
Bram Moolenaarffa22202013-11-21 12:34:11 +01003000 int n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003001
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01003002 n = mch_stat((char *)name, &st);
Bram Moolenaar78cf3f02014-01-10 18:16:07 +01003003 return n == 0 ? (long)(unsigned short)st.st_mode : -1L;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003004}
3005
3006
3007/*
Bram Moolenaare24a9c02013-07-24 13:49:22 +02003008 * Set file permission for "name" to "perm".
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003009 *
Bram Moolenaare24a9c02013-07-24 13:49:22 +02003010 * Return FAIL for failure, OK otherwise.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003011 */
3012 int
Bram Moolenaare24a9c02013-07-24 13:49:22 +02003013mch_setperm(char_u *name, long perm)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003014{
Bram Moolenaare24a9c02013-07-24 13:49:22 +02003015 long n = -1;
3016
Bram Moolenaar071d4272004-06-13 20:20:40 +00003017#ifdef FEAT_MBYTE
3018 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
3019 {
Bram Moolenaare24a9c02013-07-24 13:49:22 +02003020 WCHAR *p = enc_to_utf16(name, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003021
3022 if (p != NULL)
3023 {
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003024 n = _wchmod(p, perm);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003025 vim_free(p);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003026 if (n == -1)
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003027 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003028 }
3029 }
Bram Moolenaare24a9c02013-07-24 13:49:22 +02003030 if (n == -1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003031#endif
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01003032 n = _chmod((const char *)name, perm);
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003033 if (n == -1)
3034 return FAIL;
3035
3036 win32_set_archive(name);
3037
3038 return OK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003039}
3040
3041/*
3042 * Set hidden flag for "name".
3043 */
3044 void
3045mch_hide(char_u *name)
3046{
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003047 int attrs = win32_getattrs(name);
3048 if (attrs == -1)
3049 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003050
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003051 attrs |= FILE_ATTRIBUTE_HIDDEN;
3052 win32_setattrs(name, attrs);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003053}
3054
3055/*
Bram Moolenaar8a52ba72015-11-02 14:45:56 +01003056 * Return TRUE if file "name" exists and is hidden.
3057 */
3058 int
3059mch_ishidden(char_u *name)
3060{
3061 int f = win32_getattrs(name);
3062
3063 if (f == -1)
3064 return FALSE; /* file does not exist at all */
3065
3066 return (f & FILE_ATTRIBUTE_HIDDEN) != 0;
3067}
3068
3069/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00003070 * return TRUE if "name" is a directory
3071 * return FALSE if "name" is not a directory or upon error
3072 */
3073 int
3074mch_isdir(char_u *name)
3075{
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003076 int f = win32_getattrs(name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003077
3078 if (f == -1)
3079 return FALSE; /* file does not exist at all */
3080
3081 return (f & FILE_ATTRIBUTE_DIRECTORY) != 0;
3082}
3083
3084/*
Bram Moolenaar203258c2016-01-17 22:15:16 +01003085 * return TRUE if "name" is a directory, NOT a symlink to a directory
3086 * return FALSE if "name" is not a directory
3087 * return FALSE for error
3088 */
3089 int
3090mch_isrealdir(char_u *name)
3091{
3092 return mch_isdir(name) && !mch_is_symbolic_link(name);
3093}
3094
3095/*
Bram Moolenaar3c9c99c2011-05-05 18:31:59 +02003096 * Create directory "name".
3097 * Return 0 on success, -1 on error.
3098 */
3099 int
3100mch_mkdir(char_u *name)
3101{
3102#ifdef FEAT_MBYTE
3103 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
3104 {
3105 WCHAR *p;
3106 int retval;
3107
3108 p = enc_to_utf16(name, NULL);
3109 if (p == NULL)
3110 return -1;
3111 retval = _wmkdir(p);
3112 vim_free(p);
3113 return retval;
3114 }
3115#endif
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01003116 return _mkdir((const char *)name);
Bram Moolenaar3c9c99c2011-05-05 18:31:59 +02003117}
3118
3119/*
Bram Moolenaar4cf76792016-01-16 22:02:57 +01003120 * Delete directory "name".
3121 * Return 0 on success, -1 on error.
3122 */
3123 int
3124mch_rmdir(char_u *name)
3125{
3126#ifdef FEAT_MBYTE
3127 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
3128 {
3129 WCHAR *p;
3130 int retval;
3131
3132 p = enc_to_utf16(name, NULL);
3133 if (p == NULL)
3134 return -1;
3135 retval = _wrmdir(p);
3136 vim_free(p);
3137 return retval;
3138 }
3139#endif
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01003140 return _rmdir((const char *)name);
Bram Moolenaar4cf76792016-01-16 22:02:57 +01003141}
3142
3143/*
Bram Moolenaar03f48552006-02-28 23:52:23 +00003144 * Return TRUE if file "fname" has more than one link.
3145 */
3146 int
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003147mch_is_hard_link(char_u *fname)
Bram Moolenaar03f48552006-02-28 23:52:23 +00003148{
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003149 BY_HANDLE_FILE_INFORMATION info;
3150
3151 return win32_fileinfo(fname, &info) == FILEINFO_OK
3152 && info.nNumberOfLinks > 1;
3153}
3154
3155/*
Bram Moolenaar203258c2016-01-17 22:15:16 +01003156 * Return TRUE if "name" is a symbolic link (or a junction).
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003157 */
3158 int
Bram Moolenaar203258c2016-01-17 22:15:16 +01003159mch_is_symbolic_link(char_u *name)
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003160{
3161 HANDLE hFind;
3162 int res = FALSE;
3163 WIN32_FIND_DATAA findDataA;
3164 DWORD fileFlags = 0, reparseTag = 0;
3165#ifdef FEAT_MBYTE
3166 WCHAR *wn = NULL;
3167 WIN32_FIND_DATAW findDataW;
3168
3169 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
Bram Moolenaar203258c2016-01-17 22:15:16 +01003170 wn = enc_to_utf16(name, NULL);
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003171 if (wn != NULL)
3172 {
3173 hFind = FindFirstFileW(wn, &findDataW);
3174 vim_free(wn);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003175 if (hFind != INVALID_HANDLE_VALUE)
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003176 {
3177 fileFlags = findDataW.dwFileAttributes;
3178 reparseTag = findDataW.dwReserved0;
3179 }
3180 }
Bram Moolenaar03e114b2013-06-16 16:34:56 +02003181 else
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003182#endif
Bram Moolenaar03e114b2013-06-16 16:34:56 +02003183 {
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01003184 hFind = FindFirstFile((LPCSTR)name, &findDataA);
Bram Moolenaar03e114b2013-06-16 16:34:56 +02003185 if (hFind != INVALID_HANDLE_VALUE)
3186 {
3187 fileFlags = findDataA.dwFileAttributes;
3188 reparseTag = findDataA.dwReserved0;
3189 }
3190 }
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003191
3192 if (hFind != INVALID_HANDLE_VALUE)
3193 FindClose(hFind);
3194
3195 if ((fileFlags & FILE_ATTRIBUTE_REPARSE_POINT)
Bram Moolenaar203258c2016-01-17 22:15:16 +01003196 && (reparseTag == IO_REPARSE_TAG_SYMLINK
3197 || reparseTag == IO_REPARSE_TAG_MOUNT_POINT))
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003198 res = TRUE;
3199
3200 return res;
3201}
3202
3203/*
3204 * Return TRUE if file "fname" has more than one link or if it is a symbolic
3205 * link.
3206 */
3207 int
3208mch_is_linked(char_u *fname)
3209{
3210 if (mch_is_hard_link(fname) || mch_is_symbolic_link(fname))
3211 return TRUE;
3212 return FALSE;
3213}
3214
3215/*
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003216 * Get the by-handle-file-information for "fname".
3217 * Returns FILEINFO_OK when OK.
3218 * returns FILEINFO_ENC_FAIL when enc_to_utf16() failed.
3219 * Returns FILEINFO_READ_FAIL when CreateFile() failed.
3220 * Returns FILEINFO_INFO_FAIL when GetFileInformationByHandle() failed.
3221 */
3222 int
3223win32_fileinfo(char_u *fname, BY_HANDLE_FILE_INFORMATION *info)
3224{
Bram Moolenaar03f48552006-02-28 23:52:23 +00003225 HANDLE hFile;
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003226 int res = FILEINFO_READ_FAIL;
Bram Moolenaar03f48552006-02-28 23:52:23 +00003227#ifdef FEAT_MBYTE
3228 WCHAR *wn = NULL;
3229
3230 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003231 {
Bram Moolenaar36f692d2008-11-20 16:10:17 +00003232 wn = enc_to_utf16(fname, NULL);
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003233 if (wn == NULL)
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003234 return FILEINFO_ENC_FAIL;
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003235 }
Bram Moolenaar03f48552006-02-28 23:52:23 +00003236 if (wn != NULL)
3237 {
3238 hFile = CreateFileW(wn, /* file name */
3239 GENERIC_READ, /* access mode */
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003240 FILE_SHARE_READ | FILE_SHARE_WRITE, /* share mode */
Bram Moolenaar03f48552006-02-28 23:52:23 +00003241 NULL, /* security descriptor */
3242 OPEN_EXISTING, /* creation disposition */
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003243 FILE_FLAG_BACKUP_SEMANTICS, /* file attributes */
Bram Moolenaar03f48552006-02-28 23:52:23 +00003244 NULL); /* handle to template file */
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003245 vim_free(wn);
Bram Moolenaar03f48552006-02-28 23:52:23 +00003246 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003247 else
Bram Moolenaar03f48552006-02-28 23:52:23 +00003248#endif
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01003249 hFile = CreateFile((LPCSTR)fname, /* file name */
3250 GENERIC_READ, /* access mode */
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003251 FILE_SHARE_READ | FILE_SHARE_WRITE, /* share mode */
Bram Moolenaar03f48552006-02-28 23:52:23 +00003252 NULL, /* security descriptor */
3253 OPEN_EXISTING, /* creation disposition */
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003254 FILE_FLAG_BACKUP_SEMANTICS, /* file attributes */
Bram Moolenaar03f48552006-02-28 23:52:23 +00003255 NULL); /* handle to template file */
3256
3257 if (hFile != INVALID_HANDLE_VALUE)
3258 {
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02003259 if (GetFileInformationByHandle(hFile, info) != 0)
3260 res = FILEINFO_OK;
3261 else
3262 res = FILEINFO_INFO_FAIL;
Bram Moolenaar03f48552006-02-28 23:52:23 +00003263 CloseHandle(hFile);
3264 }
3265
Bram Moolenaar03f48552006-02-28 23:52:23 +00003266 return res;
3267}
3268
3269/*
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003270 * get file attributes for `name'
3271 * -1 : error
3272 * else FILE_ATTRIBUTE_* defined in winnt.h
3273 */
Bram Moolenaarffa22202013-11-21 12:34:11 +01003274 static int
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003275win32_getattrs(char_u *name)
3276{
3277 int attr;
3278#ifdef FEAT_MBYTE
3279 WCHAR *p = NULL;
3280
3281 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
3282 p = enc_to_utf16(name, NULL);
3283
3284 if (p != NULL)
3285 {
3286 attr = GetFileAttributesW(p);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003287 vim_free(p);
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003288 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003289 else
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003290#endif
3291 attr = GetFileAttributes((char *)name);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003292
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003293 return attr;
3294}
3295
3296/*
3297 * set file attributes for `name' to `attrs'
3298 *
3299 * return -1 for failure, 0 otherwise
3300 */
3301 static
3302 int
3303win32_setattrs(char_u *name, int attrs)
3304{
3305 int res;
3306#ifdef FEAT_MBYTE
3307 WCHAR *p = NULL;
3308
3309 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
3310 p = enc_to_utf16(name, NULL);
3311
3312 if (p != NULL)
3313 {
3314 res = SetFileAttributesW(p, attrs);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003315 vim_free(p);
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003316 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003317 else
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003318#endif
3319 res = SetFileAttributes((char *)name, attrs);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003320
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003321 return res ? 0 : -1;
3322}
3323
3324/*
3325 * Set archive flag for "name".
3326 */
3327 static
3328 int
3329win32_set_archive(char_u *name)
3330{
3331 int attrs = win32_getattrs(name);
3332 if (attrs == -1)
3333 return -1;
3334
3335 attrs |= FILE_ATTRIBUTE_ARCHIVE;
3336 return win32_setattrs(name, attrs);
3337}
3338
3339/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00003340 * Return TRUE if file or directory "name" is writable (not readonly).
3341 * Strange semantics of Win32: a readonly directory is writable, but you can't
3342 * delete a file. Let's say this means it is writable.
3343 */
3344 int
3345mch_writable(char_u *name)
3346{
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003347 int attrs = win32_getattrs(name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003348
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003349 return (attrs != -1 && (!(attrs & FILE_ATTRIBUTE_READONLY)
3350 || (attrs & FILE_ATTRIBUTE_DIRECTORY)));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003351}
3352
Bram Moolenaar071d4272004-06-13 20:20:40 +00003353/*
3354 * Return 1 if "name" can be executed, 0 if not.
Bram Moolenaar77b77102015-03-21 22:18:41 +01003355 * If "use_path" is FALSE only check if "name" is executable.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003356 * Return -1 if unknown.
3357 */
3358 int
Bram Moolenaar77b77102015-03-21 22:18:41 +01003359mch_can_exe(char_u *name, char_u **path, int use_path)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003360{
Bram Moolenaarf4b8e572004-06-24 15:53:16 +00003361 char_u buf[_MAX_PATH];
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00003362 int len = (int)STRLEN(name);
Bram Moolenaarf4b8e572004-06-24 15:53:16 +00003363 char_u *p;
3364
3365 if (len >= _MAX_PATH) /* safety check */
3366 return FALSE;
Bram Moolenaar77b77102015-03-21 22:18:41 +01003367 if (!use_path)
3368 {
3369 /* TODO: check if file is really executable. */
3370 return mch_getperm(name) != -1 && !mch_isdir(name);
3371 }
Bram Moolenaarf4b8e572004-06-24 15:53:16 +00003372
3373 /* If there already is an extension try using the name directly. Also do
3374 * this with a Unix-shell like 'shell'. */
3375 if (vim_strchr(gettail(name), '.') != NULL
3376 || strstr((char *)gettail(p_sh), "sh") != NULL)
Bram Moolenaarc7f02552014-04-01 21:00:59 +02003377 if (executable_exists((char *)name, path))
Bram Moolenaarf4b8e572004-06-24 15:53:16 +00003378 return TRUE;
3379
3380 /*
3381 * Loop over all extensions in $PATHEXT.
3382 */
Bram Moolenaarfe3ca8d2005-07-18 21:43:02 +00003383 vim_strncpy(buf, name, _MAX_PATH - 1);
Bram Moolenaarf4b8e572004-06-24 15:53:16 +00003384 p = mch_getenv("PATHEXT");
3385 if (p == NULL)
3386 p = (char_u *)".com;.exe;.bat;.cmd";
3387 while (*p)
3388 {
3389 if (p[0] == '.' && (p[1] == NUL || p[1] == ';'))
3390 {
3391 /* A single "." means no extension is added. */
3392 buf[len] = NUL;
3393 ++p;
3394 if (*p)
3395 ++p;
3396 }
3397 else
3398 copy_option_part(&p, buf + len, _MAX_PATH - len, ";");
Bram Moolenaarc7f02552014-04-01 21:00:59 +02003399 if (executable_exists((char *)buf, path))
Bram Moolenaarf4b8e572004-06-24 15:53:16 +00003400 return TRUE;
3401 }
3402 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003403}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003404
3405/*
3406 * Check what "name" is:
3407 * NODE_NORMAL: file or directory (or doesn't exist)
3408 * NODE_WRITABLE: writable device, socket, fifo, etc.
3409 * NODE_OTHER: non-writable things
3410 */
3411 int
3412mch_nodetype(char_u *name)
3413{
3414 HANDLE hFile;
3415 int type;
Bram Moolenaar4dee1bb2013-08-30 17:11:33 +02003416#ifdef FEAT_MBYTE
3417 WCHAR *wn = NULL;
3418#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003419
Bram Moolenaar043545e2006-10-10 16:44:07 +00003420 /* We can't open a file with a name "\\.\con" or "\\.\prn" and trying to
3421 * read from it later will cause Vim to hang. Thus return NODE_WRITABLE
3422 * here. */
3423 if (STRNCMP(name, "\\\\.\\", 4) == 0)
3424 return NODE_WRITABLE;
3425
Bram Moolenaar4dee1bb2013-08-30 17:11:33 +02003426#ifdef FEAT_MBYTE
3427 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
Bram Moolenaar4dee1bb2013-08-30 17:11:33 +02003428 wn = enc_to_utf16(name, NULL);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003429
3430 if (wn != NULL)
3431 {
3432 hFile = CreateFileW(wn, /* file name */
3433 GENERIC_WRITE, /* access mode */
3434 0, /* share mode */
3435 NULL, /* security descriptor */
3436 OPEN_EXISTING, /* creation disposition */
3437 0, /* file attributes */
3438 NULL); /* handle to template file */
3439 vim_free(wn);
Bram Moolenaar4dee1bb2013-08-30 17:11:33 +02003440 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003441 else
Bram Moolenaar4dee1bb2013-08-30 17:11:33 +02003442#endif
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01003443 hFile = CreateFile((LPCSTR)name, /* file name */
3444 GENERIC_WRITE, /* access mode */
3445 0, /* share mode */
3446 NULL, /* security descriptor */
3447 OPEN_EXISTING, /* creation disposition */
3448 0, /* file attributes */
3449 NULL); /* handle to template file */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003450
3451 if (hFile == INVALID_HANDLE_VALUE)
3452 return NODE_NORMAL;
3453
3454 type = GetFileType(hFile);
3455 CloseHandle(hFile);
3456 if (type == FILE_TYPE_CHAR)
3457 return NODE_WRITABLE;
3458 if (type == FILE_TYPE_DISK)
3459 return NODE_NORMAL;
3460 return NODE_OTHER;
3461}
3462
3463#ifdef HAVE_ACL
3464struct my_acl
3465{
3466 PSECURITY_DESCRIPTOR pSecurityDescriptor;
3467 PSID pSidOwner;
3468 PSID pSidGroup;
3469 PACL pDacl;
3470 PACL pSacl;
3471};
3472#endif
3473
3474/*
3475 * Return a pointer to the ACL of file "fname" in allocated memory.
3476 * Return NULL if the ACL is not available for whatever reason.
3477 */
3478 vim_acl_T
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00003479mch_get_acl(char_u *fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003480{
3481#ifndef HAVE_ACL
3482 return (vim_acl_T)NULL;
3483#else
3484 struct my_acl *p = NULL;
Bram Moolenaar27515922013-06-29 15:36:26 +02003485 DWORD err;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003486
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003487 p = (struct my_acl *)alloc_clear((unsigned)sizeof(struct my_acl));
3488 if (p != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003489 {
Bram Moolenaar27515922013-06-29 15:36:26 +02003490# ifdef FEAT_MBYTE
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003491 WCHAR *wn = NULL;
Bram Moolenaar27515922013-06-29 15:36:26 +02003492
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003493 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
3494 wn = enc_to_utf16(fname, NULL);
3495 if (wn != NULL)
3496 {
3497 /* Try to retrieve the entire security descriptor. */
3498 err = GetNamedSecurityInfoW(
3499 wn, // Abstract filename
3500 SE_FILE_OBJECT, // File Object
3501 OWNER_SECURITY_INFORMATION |
3502 GROUP_SECURITY_INFORMATION |
3503 DACL_SECURITY_INFORMATION |
3504 SACL_SECURITY_INFORMATION,
3505 &p->pSidOwner, // Ownership information.
3506 &p->pSidGroup, // Group membership.
3507 &p->pDacl, // Discretionary information.
3508 &p->pSacl, // For auditing purposes.
3509 &p->pSecurityDescriptor);
3510 if (err == ERROR_ACCESS_DENIED ||
3511 err == ERROR_PRIVILEGE_NOT_HELD)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003512 {
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003513 /* Retrieve only DACL. */
3514 (void)GetNamedSecurityInfoW(
3515 wn,
3516 SE_FILE_OBJECT,
3517 DACL_SECURITY_INFORMATION,
3518 NULL,
3519 NULL,
3520 &p->pDacl,
3521 NULL,
3522 &p->pSecurityDescriptor);
Bram Moolenaar27515922013-06-29 15:36:26 +02003523 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003524 if (p->pSecurityDescriptor == NULL)
Bram Moolenaar27515922013-06-29 15:36:26 +02003525 {
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003526 mch_free_acl((vim_acl_T)p);
3527 p = NULL;
3528 }
3529 vim_free(wn);
3530 }
3531 else
3532# endif
3533 {
3534 /* Try to retrieve the entire security descriptor. */
3535 err = GetNamedSecurityInfo(
3536 (LPSTR)fname, // Abstract filename
3537 SE_FILE_OBJECT, // File Object
3538 OWNER_SECURITY_INFORMATION |
3539 GROUP_SECURITY_INFORMATION |
3540 DACL_SECURITY_INFORMATION |
3541 SACL_SECURITY_INFORMATION,
3542 &p->pSidOwner, // Ownership information.
3543 &p->pSidGroup, // Group membership.
3544 &p->pDacl, // Discretionary information.
3545 &p->pSacl, // For auditing purposes.
3546 &p->pSecurityDescriptor);
3547 if (err == ERROR_ACCESS_DENIED ||
3548 err == ERROR_PRIVILEGE_NOT_HELD)
3549 {
3550 /* Retrieve only DACL. */
3551 (void)GetNamedSecurityInfo(
3552 (LPSTR)fname,
3553 SE_FILE_OBJECT,
3554 DACL_SECURITY_INFORMATION,
3555 NULL,
3556 NULL,
3557 &p->pDacl,
3558 NULL,
3559 &p->pSecurityDescriptor);
3560 }
3561 if (p->pSecurityDescriptor == NULL)
3562 {
3563 mch_free_acl((vim_acl_T)p);
3564 p = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003565 }
3566 }
3567 }
3568
3569 return (vim_acl_T)p;
3570#endif
3571}
3572
Bram Moolenaar27515922013-06-29 15:36:26 +02003573#ifdef HAVE_ACL
3574/*
3575 * Check if "acl" contains inherited ACE.
3576 */
3577 static BOOL
3578is_acl_inherited(PACL acl)
3579{
3580 DWORD i;
3581 ACL_SIZE_INFORMATION acl_info;
3582 PACCESS_ALLOWED_ACE ace;
3583
3584 acl_info.AceCount = 0;
3585 GetAclInformation(acl, &acl_info, sizeof(acl_info), AclSizeInformation);
3586 for (i = 0; i < acl_info.AceCount; i++)
3587 {
3588 GetAce(acl, i, (LPVOID *)&ace);
3589 if (ace->Header.AceFlags & INHERITED_ACE)
3590 return TRUE;
3591 }
3592 return FALSE;
3593}
3594#endif
3595
Bram Moolenaar071d4272004-06-13 20:20:40 +00003596/*
3597 * Set the ACL of file "fname" to "acl" (unless it's NULL).
3598 * Errors are ignored.
3599 * This must only be called with "acl" equal to what mch_get_acl() returned.
3600 */
3601 void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00003602mch_set_acl(char_u *fname, vim_acl_T acl)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003603{
3604#ifdef HAVE_ACL
3605 struct my_acl *p = (struct my_acl *)acl;
Bram Moolenaar27515922013-06-29 15:36:26 +02003606 SECURITY_INFORMATION sec_info = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003607
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003608 if (p != NULL)
Bram Moolenaar27515922013-06-29 15:36:26 +02003609 {
3610# ifdef FEAT_MBYTE
3611 WCHAR *wn = NULL;
3612# endif
3613
3614 /* Set security flags */
3615 if (p->pSidOwner)
3616 sec_info |= OWNER_SECURITY_INFORMATION;
3617 if (p->pSidGroup)
3618 sec_info |= GROUP_SECURITY_INFORMATION;
3619 if (p->pDacl)
3620 {
3621 sec_info |= DACL_SECURITY_INFORMATION;
3622 /* Do not inherit its parent's DACL.
3623 * If the DACL is inherited, Cygwin permissions would be changed.
3624 */
3625 if (!is_acl_inherited(p->pDacl))
3626 sec_info |= PROTECTED_DACL_SECURITY_INFORMATION;
3627 }
3628 if (p->pSacl)
3629 sec_info |= SACL_SECURITY_INFORMATION;
3630
3631# ifdef FEAT_MBYTE
3632 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
3633 wn = enc_to_utf16(fname, NULL);
3634 if (wn != NULL)
3635 {
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003636 (void)SetNamedSecurityInfoW(
Bram Moolenaar27515922013-06-29 15:36:26 +02003637 wn, // Abstract filename
3638 SE_FILE_OBJECT, // File Object
3639 sec_info,
3640 p->pSidOwner, // Ownership information.
3641 p->pSidGroup, // Group membership.
3642 p->pDacl, // Discretionary information.
3643 p->pSacl // For auditing purposes.
3644 );
3645 vim_free(wn);
3646 }
3647 else
3648# endif
3649 {
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003650 (void)SetNamedSecurityInfo(
Bram Moolenaar27515922013-06-29 15:36:26 +02003651 (LPSTR)fname, // Abstract filename
3652 SE_FILE_OBJECT, // File Object
3653 sec_info,
3654 p->pSidOwner, // Ownership information.
3655 p->pSidGroup, // Group membership.
3656 p->pDacl, // Discretionary information.
3657 p->pSacl // For auditing purposes.
3658 );
3659 }
3660 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003661#endif
3662}
3663
3664 void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00003665mch_free_acl(vim_acl_T acl)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003666{
3667#ifdef HAVE_ACL
3668 struct my_acl *p = (struct my_acl *)acl;
3669
3670 if (p != NULL)
3671 {
3672 LocalFree(p->pSecurityDescriptor); // Free the memory just in case
3673 vim_free(p);
3674 }
3675#endif
3676}
3677
3678#ifndef FEAT_GUI_W32
3679
3680/*
3681 * handler for ctrl-break, ctrl-c interrupts, and fatal events.
3682 */
3683 static BOOL WINAPI
3684handler_routine(
3685 DWORD dwCtrlType)
3686{
3687 switch (dwCtrlType)
3688 {
3689 case CTRL_C_EVENT:
3690 if (ctrl_c_interrupts)
3691 g_fCtrlCPressed = TRUE;
3692 return TRUE;
3693
3694 case CTRL_BREAK_EVENT:
3695 g_fCBrkPressed = TRUE;
3696 return TRUE;
3697
3698 /* fatal events: shut down gracefully */
3699 case CTRL_CLOSE_EVENT:
3700 case CTRL_LOGOFF_EVENT:
3701 case CTRL_SHUTDOWN_EVENT:
3702 windgoto((int)Rows - 1, 0);
3703 g_fForceExit = TRUE;
3704
Bram Moolenaar0fde2902008-03-16 13:54:13 +00003705 vim_snprintf((char *)IObuff, IOSIZE, _("Vim: Caught %s event\n"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00003706 (dwCtrlType == CTRL_CLOSE_EVENT
3707 ? _("close")
3708 : dwCtrlType == CTRL_LOGOFF_EVENT
3709 ? _("logoff")
3710 : _("shutdown")));
3711#ifdef DEBUG
3712 OutputDebugString(IObuff);
3713#endif
3714
3715 preserve_exit(); /* output IObuff, preserve files and exit */
3716
3717 return TRUE; /* not reached */
3718
3719 default:
3720 return FALSE;
3721 }
3722}
3723
3724
3725/*
3726 * set the tty in (raw) ? "raw" : "cooked" mode
3727 */
3728 void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00003729mch_settmode(int tmode)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003730{
3731 DWORD cmodein;
3732 DWORD cmodeout;
3733 BOOL bEnableHandler;
3734
3735 GetConsoleMode(g_hConIn, &cmodein);
3736 GetConsoleMode(g_hConOut, &cmodeout);
3737 if (tmode == TMODE_RAW)
3738 {
3739 cmodein &= ~(ENABLE_LINE_INPUT | ENABLE_PROCESSED_INPUT |
3740 ENABLE_ECHO_INPUT);
3741#ifdef FEAT_MOUSE
3742 if (g_fMouseActive)
3743 cmodein |= ENABLE_MOUSE_INPUT;
3744#endif
3745 cmodeout &= ~(ENABLE_PROCESSED_OUTPUT | ENABLE_WRAP_AT_EOL_OUTPUT);
3746 bEnableHandler = TRUE;
3747 }
3748 else /* cooked */
3749 {
3750 cmodein |= (ENABLE_LINE_INPUT | ENABLE_PROCESSED_INPUT |
3751 ENABLE_ECHO_INPUT);
3752 cmodeout |= (ENABLE_PROCESSED_OUTPUT | ENABLE_WRAP_AT_EOL_OUTPUT);
3753 bEnableHandler = FALSE;
3754 }
3755 SetConsoleMode(g_hConIn, cmodein);
3756 SetConsoleMode(g_hConOut, cmodeout);
3757 SetConsoleCtrlHandler(handler_routine, bEnableHandler);
3758
3759#ifdef MCH_WRITE_DUMP
3760 if (fdDump)
3761 {
3762 fprintf(fdDump, "mch_settmode(%s, in = %x, out = %x)\n",
3763 tmode == TMODE_RAW ? "raw" :
3764 tmode == TMODE_COOK ? "cooked" : "normal",
3765 cmodein, cmodeout);
3766 fflush(fdDump);
3767 }
3768#endif
3769}
3770
3771
3772/*
3773 * Get the size of the current window in `Rows' and `Columns'
3774 * Return OK when size could be determined, FAIL otherwise.
3775 */
3776 int
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00003777mch_get_shellsize(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003778{
3779 CONSOLE_SCREEN_BUFFER_INFO csbi;
3780
3781 if (!g_fTermcapMode && g_cbTermcap.IsValid)
3782 {
3783 /*
3784 * For some reason, we are trying to get the screen dimensions
3785 * even though we are not in termcap mode. The 'Rows' and 'Columns'
3786 * variables are really intended to mean the size of Vim screen
3787 * while in termcap mode.
3788 */
3789 Rows = g_cbTermcap.Info.dwSize.Y;
3790 Columns = g_cbTermcap.Info.dwSize.X;
3791 }
3792 else if (GetConsoleScreenBufferInfo(g_hConOut, &csbi))
3793 {
3794 Rows = csbi.srWindow.Bottom - csbi.srWindow.Top + 1;
3795 Columns = csbi.srWindow.Right - csbi.srWindow.Left + 1;
3796 }
3797 else
3798 {
3799 Rows = 25;
3800 Columns = 80;
3801 }
3802 return OK;
3803}
3804
3805/*
3806 * Set a console window to `xSize' * `ySize'
3807 */
3808 static void
3809ResizeConBufAndWindow(
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00003810 HANDLE hConsole,
3811 int xSize,
3812 int ySize)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003813{
3814 CONSOLE_SCREEN_BUFFER_INFO csbi; /* hold current console buffer info */
3815 SMALL_RECT srWindowRect; /* hold the new console size */
3816 COORD coordScreen;
3817
3818#ifdef MCH_WRITE_DUMP
3819 if (fdDump)
3820 {
3821 fprintf(fdDump, "ResizeConBufAndWindow(%d, %d)\n", xSize, ySize);
3822 fflush(fdDump);
3823 }
3824#endif
3825
3826 /* get the largest size we can size the console window to */
3827 coordScreen = GetLargestConsoleWindowSize(hConsole);
3828
3829 /* define the new console window size and scroll position */
3830 srWindowRect.Left = srWindowRect.Top = (SHORT) 0;
3831 srWindowRect.Right = (SHORT) (min(xSize, coordScreen.X) - 1);
3832 srWindowRect.Bottom = (SHORT) (min(ySize, coordScreen.Y) - 1);
3833
3834 if (GetConsoleScreenBufferInfo(g_hConOut, &csbi))
3835 {
3836 int sx, sy;
3837
3838 sx = csbi.srWindow.Right - csbi.srWindow.Left + 1;
3839 sy = csbi.srWindow.Bottom - csbi.srWindow.Top + 1;
3840 if (sy < ySize || sx < xSize)
3841 {
3842 /*
3843 * Increasing number of lines/columns, do buffer first.
3844 * Use the maximal size in x and y direction.
3845 */
3846 if (sy < ySize)
3847 coordScreen.Y = ySize;
3848 else
3849 coordScreen.Y = sy;
3850 if (sx < xSize)
3851 coordScreen.X = xSize;
3852 else
3853 coordScreen.X = sx;
3854 SetConsoleScreenBufferSize(hConsole, coordScreen);
3855 }
3856 }
3857
3858 if (!SetConsoleWindowInfo(g_hConOut, TRUE, &srWindowRect))
3859 {
3860#ifdef MCH_WRITE_DUMP
3861 if (fdDump)
3862 {
3863 fprintf(fdDump, "SetConsoleWindowInfo failed: %lx\n",
3864 GetLastError());
3865 fflush(fdDump);
3866 }
3867#endif
3868 }
3869
3870 /* define the new console buffer size */
3871 coordScreen.X = xSize;
3872 coordScreen.Y = ySize;
3873
3874 if (!SetConsoleScreenBufferSize(hConsole, coordScreen))
3875 {
3876#ifdef MCH_WRITE_DUMP
3877 if (fdDump)
3878 {
3879 fprintf(fdDump, "SetConsoleScreenBufferSize failed: %lx\n",
3880 GetLastError());
3881 fflush(fdDump);
3882 }
3883#endif
3884 }
3885}
3886
3887
3888/*
3889 * Set the console window to `Rows' * `Columns'
3890 */
3891 void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00003892mch_set_shellsize(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003893{
3894 COORD coordScreen;
3895
3896 /* Don't change window size while still starting up */
3897 if (suppress_winsize != 0)
3898 {
3899 suppress_winsize = 2;
3900 return;
3901 }
3902
3903 if (term_console)
3904 {
3905 coordScreen = GetLargestConsoleWindowSize(g_hConOut);
3906
3907 /* Clamp Rows and Columns to reasonable values */
3908 if (Rows > coordScreen.Y)
3909 Rows = coordScreen.Y;
3910 if (Columns > coordScreen.X)
3911 Columns = coordScreen.X;
3912
3913 ResizeConBufAndWindow(g_hConOut, Columns, Rows);
3914 }
3915}
3916
3917/*
3918 * Rows and/or Columns has changed.
3919 */
3920 void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00003921mch_new_shellsize(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003922{
3923 set_scroll_region(0, 0, Columns - 1, Rows - 1);
3924}
3925
3926
3927/*
3928 * Called when started up, to set the winsize that was delayed.
3929 */
3930 void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00003931mch_set_winsize_now(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003932{
3933 if (suppress_winsize == 2)
3934 {
3935 suppress_winsize = 0;
3936 mch_set_shellsize();
3937 shell_resized();
3938 }
3939 suppress_winsize = 0;
3940}
3941#endif /* FEAT_GUI_W32 */
3942
Bram Moolenaar910cffb2013-12-11 17:58:35 +01003943 static BOOL
3944vim_create_process(
Bram Moolenaar36c85b22013-12-12 20:25:44 +01003945 char *cmd,
Bram Moolenaar910cffb2013-12-11 17:58:35 +01003946 BOOL inherit_handles,
Bram Moolenaar3f1138e2014-01-05 13:29:26 +01003947 DWORD flags,
Bram Moolenaar910cffb2013-12-11 17:58:35 +01003948 STARTUPINFO *si,
3949 PROCESS_INFORMATION *pi)
3950{
Bram Moolenaarcea912a2016-10-12 14:20:24 +02003951#ifdef FEAT_MBYTE
Bram Moolenaar910cffb2013-12-11 17:58:35 +01003952 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
3953 {
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01003954 WCHAR *wcmd = enc_to_utf16((char_u *)cmd, NULL);
Bram Moolenaar910cffb2013-12-11 17:58:35 +01003955
3956 if (wcmd != NULL)
3957 {
3958 BOOL ret;
3959 ret = CreateProcessW(
3960 NULL, /* Executable name */
3961 wcmd, /* Command to execute */
3962 NULL, /* Process security attributes */
3963 NULL, /* Thread security attributes */
3964 inherit_handles, /* Inherit handles */
3965 flags, /* Creation flags */
3966 NULL, /* Environment */
3967 NULL, /* Current directory */
Bram Moolenaar36c85b22013-12-12 20:25:44 +01003968 (LPSTARTUPINFOW)si, /* Startup information */
Bram Moolenaar910cffb2013-12-11 17:58:35 +01003969 pi); /* Process information */
3970 vim_free(wcmd);
3971 return ret;
3972 }
3973 }
3974#endif
3975 return CreateProcess(
3976 NULL, /* Executable name */
3977 cmd, /* Command to execute */
3978 NULL, /* Process security attributes */
3979 NULL, /* Thread security attributes */
3980 inherit_handles, /* Inherit handles */
3981 flags, /* Creation flags */
3982 NULL, /* Environment */
3983 NULL, /* Current directory */
3984 si, /* Startup information */
3985 pi); /* Process information */
3986}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003987
3988
3989#if defined(FEAT_GUI_W32) || defined(PROTO)
3990
3991/*
3992 * Specialised version of system() for Win32 GUI mode.
3993 * This version proceeds as follows:
3994 * 1. Create a console window for use by the subprocess
3995 * 2. Run the subprocess (it gets the allocated console by default)
3996 * 3. Wait for the subprocess to terminate and get its exit code
3997 * 4. Prompt the user to press a key to close the console window
3998 */
3999 static int
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004000mch_system_classic(char *cmd, int options)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004001{
4002 STARTUPINFO si;
4003 PROCESS_INFORMATION pi;
4004 DWORD ret = 0;
4005 HWND hwnd = GetFocus();
4006
4007 si.cb = sizeof(si);
4008 si.lpReserved = NULL;
4009 si.lpDesktop = NULL;
4010 si.lpTitle = NULL;
4011 si.dwFlags = STARTF_USESHOWWINDOW;
4012 /*
Bram Moolenaarcea912a2016-10-12 14:20:24 +02004013 * It's nicer to run a filter command in a minimized window.
Bram Moolenaar96e5cee2010-11-24 12:35:21 +01004014 * Don't activate the window to keep focus on Vim.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004015 */
Bram Moolenaarcea912a2016-10-12 14:20:24 +02004016 if (options & SHELL_DOOUT)
Bram Moolenaar96e5cee2010-11-24 12:35:21 +01004017 si.wShowWindow = SW_SHOWMINNOACTIVE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004018 else
4019 si.wShowWindow = SW_SHOWNORMAL;
4020 si.cbReserved2 = 0;
4021 si.lpReserved2 = NULL;
4022
Bram Moolenaar071d4272004-06-13 20:20:40 +00004023 /* Now, run the command */
Bram Moolenaar910cffb2013-12-11 17:58:35 +01004024 vim_create_process(cmd, FALSE,
4025 CREATE_DEFAULT_ERROR_MODE | CREATE_NEW_CONSOLE, &si, &pi);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004026
4027 /* Wait for the command to terminate before continuing */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004028 {
4029#ifdef FEAT_GUI
4030 int delay = 1;
4031
4032 /* Keep updating the window while waiting for the shell to finish. */
4033 for (;;)
4034 {
4035 MSG msg;
4036
Bram Moolenaar8c85fa32011-08-10 17:08:03 +02004037 if (pPeekMessage(&msg, (HWND)NULL, 0, 0, PM_REMOVE))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004038 {
4039 TranslateMessage(&msg);
Bram Moolenaar8c85fa32011-08-10 17:08:03 +02004040 pDispatchMessage(&msg);
Bram Moolenaare4195c52012-08-02 12:31:44 +02004041 delay = 1;
4042 continue;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004043 }
4044 if (WaitForSingleObject(pi.hProcess, delay) != WAIT_TIMEOUT)
4045 break;
4046
4047 /* We start waiting for a very short time and then increase it, so
4048 * that we respond quickly when the process is quick, and don't
4049 * consume too much overhead when it's slow. */
4050 if (delay < 50)
4051 delay += 10;
4052 }
4053#else
4054 WaitForSingleObject(pi.hProcess, INFINITE);
4055#endif
4056
4057 /* Get the command exit code */
4058 GetExitCodeProcess(pi.hProcess, &ret);
4059 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004060
4061 /* Close the handles to the subprocess, so that it goes away */
4062 CloseHandle(pi.hThread);
4063 CloseHandle(pi.hProcess);
4064
4065 /* Try to get input focus back. Doesn't always work though. */
4066 PostMessage(hwnd, WM_SETFOCUS, 0, 0);
4067
4068 return ret;
4069}
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004070
4071/*
4072 * Thread launched by the gui to send the current buffer data to the
4073 * process. This way avoid to hang up vim totally if the children
4074 * process take a long time to process the lines.
4075 */
Bram Moolenaar4c38d662016-08-03 20:54:57 +02004076 static unsigned int __stdcall
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004077sub_process_writer(LPVOID param)
4078{
4079 HANDLE g_hChildStd_IN_Wr = param;
4080 linenr_T lnum = curbuf->b_op_start.lnum;
4081 DWORD len = 0;
4082 DWORD l;
4083 char_u *lp = ml_get(lnum);
4084 char_u *s;
4085 int written = 0;
4086
4087 for (;;)
4088 {
4089 l = (DWORD)STRLEN(lp + written);
4090 if (l == 0)
4091 len = 0;
4092 else if (lp[written] == NL)
4093 {
4094 /* NL -> NUL translation */
4095 WriteFile(g_hChildStd_IN_Wr, "", 1, &len, NULL);
4096 }
4097 else
4098 {
4099 s = vim_strchr(lp + written, NL);
4100 WriteFile(g_hChildStd_IN_Wr, (char *)lp + written,
4101 s == NULL ? l : (DWORD)(s - (lp + written)),
4102 &len, NULL);
4103 }
4104 if (len == (int)l)
4105 {
4106 /* Finished a line, add a NL, unless this line should not have
4107 * one. */
4108 if (lnum != curbuf->b_op_end.lnum
Bram Moolenaar34d72d42015-07-17 14:18:08 +02004109 || (!curbuf->b_p_bin
4110 && curbuf->b_p_fixeol)
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004111 || (lnum != curbuf->b_no_eol_lnum
4112 && (lnum != curbuf->b_ml.ml_line_count
4113 || curbuf->b_p_eol)))
4114 {
Bram Moolenaaraf62ff32013-03-19 14:48:29 +01004115 WriteFile(g_hChildStd_IN_Wr, "\n", 1, (LPDWORD)&ignored, NULL);
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004116 }
4117
4118 ++lnum;
4119 if (lnum > curbuf->b_op_end.lnum)
4120 break;
4121
4122 lp = ml_get(lnum);
4123 written = 0;
4124 }
4125 else if (len > 0)
4126 written += len;
4127 }
4128
4129 /* finished all the lines, close pipe */
4130 CloseHandle(g_hChildStd_IN_Wr);
Bram Moolenaar86f2cd52016-08-02 21:55:17 +02004131 return 0;
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004132}
4133
4134
4135# define BUFLEN 100 /* length for buffer, stolen from unix version */
4136
4137/*
4138 * This function read from the children's stdout and write the
4139 * data on screen or in the buffer accordingly.
4140 */
4141 static void
4142dump_pipe(int options,
4143 HANDLE g_hChildStd_OUT_Rd,
4144 garray_T *ga,
4145 char_u buffer[],
4146 DWORD *buffer_off)
4147{
4148 DWORD availableBytes = 0;
4149 DWORD i;
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004150 int ret;
4151 DWORD len;
4152 DWORD toRead;
4153 int repeatCount;
4154
4155 /* we query the pipe to see if there is any data to read
4156 * to avoid to perform a blocking read */
4157 ret = PeekNamedPipe(g_hChildStd_OUT_Rd, /* pipe to query */
4158 NULL, /* optional buffer */
Bram Moolenaarf6a2b082012-06-29 13:14:03 +02004159 0, /* buffer size */
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004160 NULL, /* number of read bytes */
4161 &availableBytes, /* available bytes total */
4162 NULL); /* byteLeft */
4163
4164 repeatCount = 0;
4165 /* We got real data in the pipe, read it */
Bram Moolenaarf6a2b082012-06-29 13:14:03 +02004166 while (ret != 0 && availableBytes > 0)
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004167 {
4168 repeatCount++;
4169 toRead =
4170# ifdef FEAT_MBYTE
4171 (DWORD)(BUFLEN - *buffer_off);
4172# else
4173 (DWORD)BUFLEN;
4174# endif
4175 toRead = availableBytes < toRead ? availableBytes : toRead;
4176 ReadFile(g_hChildStd_OUT_Rd, buffer
4177# ifdef FEAT_MBYTE
4178 + *buffer_off, toRead
4179# else
4180 , toRead
4181# endif
4182 , &len, NULL);
4183
4184 /* If we haven't read anything, there is a problem */
4185 if (len == 0)
4186 break;
4187
4188 availableBytes -= len;
4189
4190 if (options & SHELL_READ)
4191 {
4192 /* Do NUL -> NL translation, append NL separated
4193 * lines to the current buffer. */
4194 for (i = 0; i < len; ++i)
4195 {
4196 if (buffer[i] == NL)
4197 append_ga_line(ga);
4198 else if (buffer[i] == NUL)
4199 ga_append(ga, NL);
4200 else
4201 ga_append(ga, buffer[i]);
4202 }
4203 }
4204# ifdef FEAT_MBYTE
4205 else if (has_mbyte)
4206 {
4207 int l;
Bram Moolenaar2eba1822011-08-27 15:10:04 +02004208 int c;
4209 char_u *p;
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004210
4211 len += *buffer_off;
4212 buffer[len] = NUL;
4213
4214 /* Check if the last character in buffer[] is
4215 * incomplete, keep these bytes for the next
4216 * round. */
4217 for (p = buffer; p < buffer + len; p += l)
4218 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02004219 l = MB_CPTR2LEN(p);
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004220 if (l == 0)
4221 l = 1; /* NUL byte? */
4222 else if (MB_BYTE2LEN(*p) != l)
4223 break;
4224 }
4225 if (p == buffer) /* no complete character */
4226 {
4227 /* avoid getting stuck at an illegal byte */
4228 if (len >= 12)
4229 ++p;
4230 else
4231 {
4232 *buffer_off = len;
4233 return;
4234 }
4235 }
4236 c = *p;
4237 *p = NUL;
4238 msg_puts(buffer);
4239 if (p < buffer + len)
4240 {
4241 *p = c;
4242 *buffer_off = (DWORD)((buffer + len) - p);
4243 mch_memmove(buffer, p, *buffer_off);
4244 return;
4245 }
4246 *buffer_off = 0;
4247 }
4248# endif /* FEAT_MBYTE */
4249 else
4250 {
4251 buffer[len] = NUL;
4252 msg_puts(buffer);
4253 }
4254
4255 windgoto(msg_row, msg_col);
4256 cursor_on();
4257 out_flush();
4258 }
4259}
4260
4261/*
4262 * Version of system to use for windows NT > 5.0 (Win2K), use pipe
4263 * for communication and doesn't open any new window.
4264 */
4265 static int
4266mch_system_piped(char *cmd, int options)
4267{
4268 STARTUPINFO si;
4269 PROCESS_INFORMATION pi;
4270 DWORD ret = 0;
4271
4272 HANDLE g_hChildStd_IN_Rd = NULL;
4273 HANDLE g_hChildStd_IN_Wr = NULL;
4274 HANDLE g_hChildStd_OUT_Rd = NULL;
4275 HANDLE g_hChildStd_OUT_Wr = NULL;
4276
4277 char_u buffer[BUFLEN + 1]; /* reading buffer + size */
4278 DWORD len;
4279
4280 /* buffer used to receive keys */
4281 char_u ta_buf[BUFLEN + 1]; /* TypeAHead */
4282 int ta_len = 0; /* valid bytes in ta_buf[] */
4283
4284 DWORD i;
4285 int c;
4286 int noread_cnt = 0;
4287 garray_T ga;
4288 int delay = 1;
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004289 DWORD buffer_off = 0; /* valid bytes in buffer[] */
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004290 char *p = NULL;
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004291
4292 SECURITY_ATTRIBUTES saAttr;
4293
4294 /* Set the bInheritHandle flag so pipe handles are inherited. */
4295 saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
4296 saAttr.bInheritHandle = TRUE;
4297 saAttr.lpSecurityDescriptor = NULL;
4298
4299 if ( ! CreatePipe(&g_hChildStd_OUT_Rd, &g_hChildStd_OUT_Wr, &saAttr, 0)
4300 /* Ensure the read handle to the pipe for STDOUT is not inherited. */
Bram Moolenaarcea912a2016-10-12 14:20:24 +02004301 || ! SetHandleInformation(g_hChildStd_OUT_Rd, HANDLE_FLAG_INHERIT, 0)
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004302 /* Create a pipe for the child process's STDIN. */
4303 || ! CreatePipe(&g_hChildStd_IN_Rd, &g_hChildStd_IN_Wr, &saAttr, 0)
4304 /* Ensure the write handle to the pipe for STDIN is not inherited. */
Bram Moolenaarcea912a2016-10-12 14:20:24 +02004305 || ! SetHandleInformation(g_hChildStd_IN_Wr, HANDLE_FLAG_INHERIT, 0) )
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004306 {
4307 CloseHandle(g_hChildStd_IN_Rd);
4308 CloseHandle(g_hChildStd_IN_Wr);
4309 CloseHandle(g_hChildStd_OUT_Rd);
4310 CloseHandle(g_hChildStd_OUT_Wr);
4311 MSG_PUTS(_("\nCannot create pipes\n"));
4312 }
4313
4314 si.cb = sizeof(si);
4315 si.lpReserved = NULL;
4316 si.lpDesktop = NULL;
4317 si.lpTitle = NULL;
4318 si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;
4319
4320 /* set-up our file redirection */
4321 si.hStdError = g_hChildStd_OUT_Wr;
4322 si.hStdOutput = g_hChildStd_OUT_Wr;
4323 si.hStdInput = g_hChildStd_IN_Rd;
4324 si.wShowWindow = SW_HIDE;
4325 si.cbReserved2 = 0;
4326 si.lpReserved2 = NULL;
4327
4328 if (options & SHELL_READ)
4329 ga_init2(&ga, 1, BUFLEN);
4330
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004331 if (cmd != NULL)
4332 {
4333 p = (char *)vim_strsave((char_u *)cmd);
4334 if (p != NULL)
4335 unescape_shellxquote((char_u *)p, p_sxe);
4336 else
4337 p = cmd;
4338 }
4339
Bram Moolenaar910cffb2013-12-11 17:58:35 +01004340 /* Now, run the command.
4341 * About "Inherit handles" being TRUE: this command can be litigious,
4342 * handle inheritance was deactivated for pending temp file, but, if we
4343 * deactivate it, the pipes don't work for some reason. */
4344 vim_create_process(p, TRUE, CREATE_DEFAULT_ERROR_MODE, &si, &pi);
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004345
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004346 if (p != cmd)
4347 vim_free(p);
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004348
4349 /* Close our unused side of the pipes */
4350 CloseHandle(g_hChildStd_IN_Rd);
4351 CloseHandle(g_hChildStd_OUT_Wr);
4352
4353 if (options & SHELL_WRITE)
4354 {
Bram Moolenaar86f2cd52016-08-02 21:55:17 +02004355 HANDLE thread = (HANDLE)
4356 _beginthreadex(NULL, /* security attributes */
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004357 0, /* default stack size */
4358 sub_process_writer, /* function to be executed */
4359 g_hChildStd_IN_Wr, /* parameter */
4360 0, /* creation flag, start immediately */
4361 NULL); /* we don't care about thread id */
4362 CloseHandle(thread);
4363 g_hChildStd_IN_Wr = NULL;
4364 }
4365
4366 /* Keep updating the window while waiting for the shell to finish. */
4367 for (;;)
4368 {
4369 MSG msg;
4370
Bram Moolenaar175d0702013-12-11 18:36:33 +01004371 if (pPeekMessage(&msg, (HWND)NULL, 0, 0, PM_REMOVE))
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004372 {
4373 TranslateMessage(&msg);
Bram Moolenaar175d0702013-12-11 18:36:33 +01004374 pDispatchMessage(&msg);
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004375 }
4376
4377 /* write pipe information in the window */
4378 if ((options & (SHELL_READ|SHELL_WRITE))
4379# ifdef FEAT_GUI
4380 || gui.in_use
4381# endif
4382 )
4383 {
4384 len = 0;
4385 if (!(options & SHELL_EXPAND)
4386 && ((options &
4387 (SHELL_READ|SHELL_WRITE|SHELL_COOKED))
4388 != (SHELL_READ|SHELL_WRITE|SHELL_COOKED)
4389# ifdef FEAT_GUI
4390 || gui.in_use
4391# endif
4392 )
4393 && (ta_len > 0 || noread_cnt > 4))
4394 {
4395 if (ta_len == 0)
4396 {
4397 /* Get extra characters when we don't have any. Reset the
4398 * counter and timer. */
4399 noread_cnt = 0;
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004400 len = ui_inchar(ta_buf, BUFLEN, 10L, 0);
4401 }
4402 if (ta_len > 0 || len > 0)
4403 {
4404 /*
4405 * For pipes: Check for CTRL-C: send interrupt signal to
4406 * child. Check for CTRL-D: EOF, close pipe to child.
4407 */
4408 if (len == 1 && cmd != NULL)
4409 {
4410 if (ta_buf[ta_len] == Ctrl_C)
4411 {
4412 /* Learn what exit code is expected, for
4413 * now put 9 as SIGKILL */
4414 TerminateProcess(pi.hProcess, 9);
4415 }
4416 if (ta_buf[ta_len] == Ctrl_D)
4417 {
4418 CloseHandle(g_hChildStd_IN_Wr);
4419 g_hChildStd_IN_Wr = NULL;
4420 }
4421 }
4422
4423 /* replace K_BS by <BS> and K_DEL by <DEL> */
4424 for (i = ta_len; i < ta_len + len; ++i)
4425 {
4426 if (ta_buf[i] == CSI && len - i > 2)
4427 {
4428 c = TERMCAP2KEY(ta_buf[i + 1], ta_buf[i + 2]);
4429 if (c == K_DEL || c == K_KDEL || c == K_BS)
4430 {
4431 mch_memmove(ta_buf + i + 1, ta_buf + i + 3,
4432 (size_t)(len - i - 2));
4433 if (c == K_DEL || c == K_KDEL)
4434 ta_buf[i] = DEL;
4435 else
4436 ta_buf[i] = Ctrl_H;
4437 len -= 2;
4438 }
4439 }
4440 else if (ta_buf[i] == '\r')
4441 ta_buf[i] = '\n';
4442# ifdef FEAT_MBYTE
4443 if (has_mbyte)
4444 i += (*mb_ptr2len_len)(ta_buf + i,
4445 ta_len + len - i) - 1;
4446# endif
4447 }
4448
4449 /*
4450 * For pipes: echo the typed characters. For a pty this
4451 * does not seem to work.
4452 */
4453 for (i = ta_len; i < ta_len + len; ++i)
4454 {
4455 if (ta_buf[i] == '\n' || ta_buf[i] == '\b')
4456 msg_putchar(ta_buf[i]);
4457# ifdef FEAT_MBYTE
4458 else if (has_mbyte)
4459 {
4460 int l = (*mb_ptr2len)(ta_buf + i);
4461
4462 msg_outtrans_len(ta_buf + i, l);
4463 i += l - 1;
4464 }
4465# endif
4466 else
4467 msg_outtrans_len(ta_buf + i, 1);
4468 }
4469 windgoto(msg_row, msg_col);
4470 out_flush();
4471
4472 ta_len += len;
4473
4474 /*
4475 * Write the characters to the child, unless EOF has been
4476 * typed for pipes. Write one character at a time, to
4477 * avoid losing too much typeahead. When writing buffer
4478 * lines, drop the typed characters (only check for
4479 * CTRL-C).
4480 */
4481 if (options & SHELL_WRITE)
4482 ta_len = 0;
4483 else if (g_hChildStd_IN_Wr != NULL)
4484 {
4485 WriteFile(g_hChildStd_IN_Wr, (char*)ta_buf,
4486 1, &len, NULL);
4487 // if we are typing in, we want to keep things reactive
4488 delay = 1;
4489 if (len > 0)
4490 {
4491 ta_len -= len;
4492 mch_memmove(ta_buf, ta_buf + len, ta_len);
4493 }
4494 }
4495 }
4496 }
4497 }
4498
4499 if (ta_len)
4500 ui_inchar_undo(ta_buf, ta_len);
4501
4502 if (WaitForSingleObject(pi.hProcess, delay) != WAIT_TIMEOUT)
4503 {
Bram Moolenaar2eba1822011-08-27 15:10:04 +02004504 dump_pipe(options, g_hChildStd_OUT_Rd, &ga, buffer, &buffer_off);
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004505 break;
4506 }
4507
4508 ++noread_cnt;
Bram Moolenaar2eba1822011-08-27 15:10:04 +02004509 dump_pipe(options, g_hChildStd_OUT_Rd, &ga, buffer, &buffer_off);
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004510
4511 /* We start waiting for a very short time and then increase it, so
4512 * that we respond quickly when the process is quick, and don't
4513 * consume too much overhead when it's slow. */
4514 if (delay < 50)
4515 delay += 10;
4516 }
4517
4518 /* Close the pipe */
4519 CloseHandle(g_hChildStd_OUT_Rd);
4520 if (g_hChildStd_IN_Wr != NULL)
4521 CloseHandle(g_hChildStd_IN_Wr);
4522
4523 WaitForSingleObject(pi.hProcess, INFINITE);
4524
4525 /* Get the command exit code */
4526 GetExitCodeProcess(pi.hProcess, &ret);
4527
4528 if (options & SHELL_READ)
4529 {
4530 if (ga.ga_len > 0)
4531 {
4532 append_ga_line(&ga);
4533 /* remember that the NL was missing */
4534 curbuf->b_no_eol_lnum = curwin->w_cursor.lnum;
4535 }
4536 else
4537 curbuf->b_no_eol_lnum = 0;
4538 ga_clear(&ga);
4539 }
4540
4541 /* Close the handles to the subprocess, so that it goes away */
4542 CloseHandle(pi.hThread);
4543 CloseHandle(pi.hProcess);
4544
4545 return ret;
4546}
4547
4548 static int
4549mch_system(char *cmd, int options)
4550{
4551 /* if we can pipe and the shelltemp option is off */
Bram Moolenaarcea912a2016-10-12 14:20:24 +02004552 if (!p_stmp)
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02004553 return mch_system_piped(cmd, options);
4554 else
4555 return mch_system_classic(cmd, options);
4556}
Bram Moolenaar071d4272004-06-13 20:20:40 +00004557#else
4558
Bram Moolenaar910cffb2013-12-11 17:58:35 +01004559# ifdef FEAT_MBYTE
4560 static int
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01004561mch_system(char *cmd, int options)
Bram Moolenaar910cffb2013-12-11 17:58:35 +01004562{
4563 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
4564 {
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004565 WCHAR *wcmd = enc_to_utf16((char_u *)cmd, NULL);
Bram Moolenaar910cffb2013-12-11 17:58:35 +01004566 if (wcmd != NULL)
4567 {
4568 int ret = _wsystem(wcmd);
4569 vim_free(wcmd);
4570 return ret;
4571 }
4572 }
4573 return system(cmd);
4574}
4575# else
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01004576# define mch_system(c, o) system(c)
Bram Moolenaar910cffb2013-12-11 17:58:35 +01004577# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004578
4579#endif
4580
4581/*
4582 * Either execute a command by calling the shell or start a new shell
4583 */
4584 int
4585mch_call_shell(
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00004586 char_u *cmd,
4587 int options) /* SHELL_*, see vim.h */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004588{
4589 int x = 0;
4590 int tmode = cur_tmode;
4591#ifdef FEAT_TITLE
Bram Moolenaar799d6ab2014-10-16 16:16:37 +02004592 char szShellTitle[512];
Bram Moolenaar1df52d72014-10-15 22:50:10 +02004593# ifdef FEAT_MBYTE
Bram Moolenaar799d6ab2014-10-16 16:16:37 +02004594 int did_set_title = FALSE;
4595
Bram Moolenaar1df52d72014-10-15 22:50:10 +02004596 /* Change the title to reflect that we are in a subshell. */
4597 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
4598 {
4599 WCHAR szShellTitle[512];
4600
4601 if (GetConsoleTitleW(szShellTitle,
4602 sizeof(szShellTitle)/sizeof(WCHAR) - 4) > 0)
4603 {
4604 if (cmd == NULL)
4605 wcscat(szShellTitle, L" :sh");
4606 else
4607 {
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004608 WCHAR *wn = enc_to_utf16((char_u *)cmd, NULL);
Bram Moolenaar1df52d72014-10-15 22:50:10 +02004609
4610 if (wn != NULL)
4611 {
4612 wcscat(szShellTitle, L" - !");
4613 if ((wcslen(szShellTitle) + wcslen(wn) <
4614 sizeof(szShellTitle)/sizeof(WCHAR)))
4615 wcscat(szShellTitle, wn);
4616 SetConsoleTitleW(szShellTitle);
4617 vim_free(wn);
Bram Moolenaar799d6ab2014-10-16 16:16:37 +02004618 did_set_title = TRUE;
Bram Moolenaar1df52d72014-10-15 22:50:10 +02004619 }
4620 }
4621 }
4622 }
Bram Moolenaar799d6ab2014-10-16 16:16:37 +02004623 if (!did_set_title)
4624# endif
4625 /* Change the title to reflect that we are in a subshell. */
4626 if (GetConsoleTitle(szShellTitle, sizeof(szShellTitle) - 4) > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004627 {
Bram Moolenaar799d6ab2014-10-16 16:16:37 +02004628 if (cmd == NULL)
4629 strcat(szShellTitle, " :sh");
4630 else
4631 {
4632 strcat(szShellTitle, " - !");
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004633 if ((strlen(szShellTitle) + strlen((char *)cmd)
4634 < sizeof(szShellTitle)))
4635 strcat(szShellTitle, (char *)cmd);
Bram Moolenaar799d6ab2014-10-16 16:16:37 +02004636 }
4637 SetConsoleTitle(szShellTitle);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004638 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004639#endif
4640
4641 out_flush();
4642
4643#ifdef MCH_WRITE_DUMP
4644 if (fdDump)
4645 {
4646 fprintf(fdDump, "mch_call_shell(\"%s\", %d)\n", cmd, options);
4647 fflush(fdDump);
4648 }
4649#endif
4650
4651 /*
4652 * Catch all deadly signals while running the external command, because a
4653 * CTRL-C, Ctrl-Break or illegal instruction might otherwise kill us.
4654 */
4655 signal(SIGINT, SIG_IGN);
4656#if defined(__GNUC__) && !defined(__MINGW32__)
4657 signal(SIGKILL, SIG_IGN);
4658#else
4659 signal(SIGBREAK, SIG_IGN);
4660#endif
4661 signal(SIGILL, SIG_IGN);
4662 signal(SIGFPE, SIG_IGN);
4663 signal(SIGSEGV, SIG_IGN);
4664 signal(SIGTERM, SIG_IGN);
4665 signal(SIGABRT, SIG_IGN);
4666
4667 if (options & SHELL_COOKED)
4668 settmode(TMODE_COOK); /* set to normal mode */
4669
4670 if (cmd == NULL)
4671 {
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004672 x = mch_system((char *)p_sh, options);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004673 }
4674 else
4675 {
4676 /* we use "command" or "cmd" to start the shell; slow but easy */
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004677 char_u *newcmd = NULL;
4678 char_u *cmdbase = cmd;
4679 long_u cmdlen;
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004680
4681 /* Skip a leading ", ( and "(. */
4682 if (*cmdbase == '"' )
4683 ++cmdbase;
4684 if (*cmdbase == '(')
4685 ++cmdbase;
4686
4687 if ((STRNICMP(cmdbase, "start", 5) == 0) && vim_iswhite(cmdbase[5]))
4688 {
4689 STARTUPINFO si;
4690 PROCESS_INFORMATION pi;
4691 DWORD flags = CREATE_NEW_CONSOLE;
4692 char_u *p;
4693
Bram Moolenaarfcc3f462014-01-24 19:55:37 +01004694 ZeroMemory(&si, sizeof(si));
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004695 si.cb = sizeof(si);
4696 si.lpReserved = NULL;
4697 si.lpDesktop = NULL;
4698 si.lpTitle = NULL;
4699 si.dwFlags = 0;
4700 si.cbReserved2 = 0;
4701 si.lpReserved2 = NULL;
4702
4703 cmdbase = skipwhite(cmdbase + 5);
4704 if ((STRNICMP(cmdbase, "/min", 4) == 0)
4705 && vim_iswhite(cmdbase[4]))
4706 {
4707 cmdbase = skipwhite(cmdbase + 4);
4708 si.dwFlags = STARTF_USESHOWWINDOW;
4709 si.wShowWindow = SW_SHOWMINNOACTIVE;
4710 }
4711 else if ((STRNICMP(cmdbase, "/b", 2) == 0)
4712 && vim_iswhite(cmdbase[2]))
4713 {
4714 cmdbase = skipwhite(cmdbase + 2);
4715 flags = CREATE_NO_WINDOW;
4716 si.dwFlags = STARTF_USESTDHANDLES;
4717 si.hStdInput = CreateFile("\\\\.\\NUL", // File name
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004718 GENERIC_READ, // Access flags
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004719 0, // Share flags
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004720 NULL, // Security att.
4721 OPEN_EXISTING, // Open flags
4722 FILE_ATTRIBUTE_NORMAL, // File att.
4723 NULL); // Temp file
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004724 si.hStdOutput = si.hStdInput;
4725 si.hStdError = si.hStdInput;
4726 }
4727
4728 /* Remove a trailing ", ) and )" if they have a match
4729 * at the start of the command. */
4730 if (cmdbase > cmd)
4731 {
4732 p = cmdbase + STRLEN(cmdbase);
4733 if (p > cmdbase && p[-1] == '"' && *cmd == '"')
4734 *--p = NUL;
4735 if (p > cmdbase && p[-1] == ')'
4736 && (*cmd =='(' || cmd[1] == '('))
4737 *--p = NUL;
4738 }
4739
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004740 newcmd = cmdbase;
4741 unescape_shellxquote(cmdbase, p_sxe);
4742
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004743 /*
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004744 * If creating new console, arguments are passed to the
4745 * 'cmd.exe' as-is. If it's not, arguments are not treated
4746 * correctly for current 'cmd.exe'. So unescape characters in
4747 * shellxescape except '|' for avoiding to be treated as
4748 * argument to them. Pass the arguments to sub-shell.
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004749 */
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004750 if (flags != CREATE_NEW_CONSOLE)
4751 {
4752 char_u *subcmd;
Bram Moolenaaree7d1002012-02-22 15:34:08 +01004753 char_u *cmd_shell = mch_getenv("COMSPEC");
4754
4755 if (cmd_shell == NULL || *cmd_shell == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004756 cmd_shell = (char_u *)default_shell();
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004757
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004758 subcmd = vim_strsave_escaped_ext(cmdbase,
4759 (char_u *)"|", '^', FALSE);
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004760 if (subcmd != NULL)
4761 {
4762 /* make "cmd.exe /c arguments" */
4763 cmdlen = STRLEN(cmd_shell) + STRLEN(subcmd) + 5;
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004764 newcmd = lalloc(cmdlen, TRUE);
4765 if (newcmd != NULL)
4766 vim_snprintf((char *)newcmd, cmdlen, "%s /c %s",
Bram Moolenaaree7d1002012-02-22 15:34:08 +01004767 cmd_shell, subcmd);
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004768 else
4769 newcmd = cmdbase;
Bram Moolenaaree7d1002012-02-22 15:34:08 +01004770 vim_free(subcmd);
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004771 }
4772 }
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004773
4774 /*
4775 * Now, start the command as a process, so that it doesn't
4776 * inherit our handles which causes unpleasant dangling swap
4777 * files if we exit before the spawned process
4778 */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004779 if (vim_create_process((char *)newcmd, FALSE, flags, &si, &pi))
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004780 x = 0;
4781 else
4782 {
4783 x = -1;
4784#ifdef FEAT_GUI_W32
4785 EMSG(_("E371: Command not found"));
4786#endif
4787 }
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004788
4789 if (newcmd != cmdbase)
4790 vim_free(newcmd);
4791
Bram Moolenaarfcc3f462014-01-24 19:55:37 +01004792 if (si.dwFlags == STARTF_USESTDHANDLES && si.hStdInput != NULL)
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004793 {
Bram Moolenaarfcc3f462014-01-24 19:55:37 +01004794 /* Close the handle to \\.\NUL created above. */
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004795 CloseHandle(si.hStdInput);
4796 }
4797 /* Close the handles to the subprocess, so that it goes away */
4798 CloseHandle(pi.hThread);
4799 CloseHandle(pi.hProcess);
4800 }
4801 else
4802 {
Bram Moolenaarfb7df7b2012-02-22 13:07:05 +01004803 cmdlen = (
Bram Moolenaar071d4272004-06-13 20:20:40 +00004804#ifdef FEAT_GUI_W32
Bram Moolenaarcea912a2016-10-12 14:20:24 +02004805 (!p_stmp ? 0 : STRLEN(vimrun_path)) +
Bram Moolenaar071d4272004-06-13 20:20:40 +00004806#endif
Bram Moolenaar0fde2902008-03-16 13:54:13 +00004807 STRLEN(p_sh) + STRLEN(p_shcf) + STRLEN(cmd) + 10);
4808
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004809 newcmd = lalloc(cmdlen, TRUE);
4810 if (newcmd != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004811 {
4812#if defined(FEAT_GUI_W32)
4813 if (need_vimrun_warning)
4814 {
Bram Moolenaar63e43442016-11-19 17:28:44 +01004815 char *msg = _("VIMRUN.EXE not found in your $PATH.\n"
4816 "External commands will not pause after completion.\n"
4817 "See :help win32-vimrun for more information.");
4818 char *title = _("Vim Warning");
4819# ifdef FEAT_MBYTE
4820 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
4821 {
4822 WCHAR *wmsg = enc_to_utf16((char_u *)msg, NULL);
4823 WCHAR *wtitle = enc_to_utf16((char_u *)title, NULL);
4824
4825 if (wmsg != NULL && wtitle != NULL)
4826 MessageBoxW(NULL, wmsg, wtitle, MB_ICONWARNING);
4827 vim_free(wmsg);
4828 vim_free(wtitle);
4829 }
4830 else
4831# endif
4832 MessageBox(NULL, msg, title, MB_ICONWARNING);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004833 need_vimrun_warning = FALSE;
4834 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02004835 if (!s_dont_use_vimrun && p_stmp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004836 /* Use vimrun to execute the command. It opens a console
4837 * window, which can be closed without killing Vim. */
Bram Moolenaarcc448b32010-07-14 16:52:17 +02004838 vim_snprintf((char *)newcmd, cmdlen, "%s%s%s %s %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00004839 vimrun_path,
4840 (msg_silent != 0 || (options & SHELL_DOOUT))
4841 ? "-s " : "",
4842 p_sh, p_shcf, cmd);
4843 else
4844#endif
Bram Moolenaarcc448b32010-07-14 16:52:17 +02004845 vim_snprintf((char *)newcmd, cmdlen, "%s %s %s",
Bram Moolenaar0fde2902008-03-16 13:54:13 +00004846 p_sh, p_shcf, cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004847 x = mch_system((char *)newcmd, options);
Bram Moolenaar6b707b42012-02-21 21:22:44 +01004848 vim_free(newcmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004849 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004850 }
4851 }
4852
4853 if (tmode == TMODE_RAW)
4854 settmode(TMODE_RAW); /* set to raw mode */
4855
4856 /* Print the return value, unless "vimrun" was used. */
4857 if (x != 0 && !(options & SHELL_SILENT) && !emsg_silent
4858#if defined(FEAT_GUI_W32)
Bram Moolenaarcea912a2016-10-12 14:20:24 +02004859 && ((options & SHELL_DOOUT) || s_dont_use_vimrun || !p_stmp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004860#endif
4861 )
4862 {
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004863 smsg((char_u *)_("shell returned %d"), x);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004864 msg_putchar('\n');
4865 }
4866#ifdef FEAT_TITLE
4867 resettitle();
4868#endif
4869
4870 signal(SIGINT, SIG_DFL);
4871#if defined(__GNUC__) && !defined(__MINGW32__)
4872 signal(SIGKILL, SIG_DFL);
4873#else
4874 signal(SIGBREAK, SIG_DFL);
4875#endif
4876 signal(SIGILL, SIG_DFL);
4877 signal(SIGFPE, SIG_DFL);
4878 signal(SIGSEGV, SIG_DFL);
4879 signal(SIGTERM, SIG_DFL);
4880 signal(SIGABRT, SIG_DFL);
4881
4882 return x;
4883}
4884
Bram Moolenaar509ce2a2016-03-11 22:52:15 +01004885#if defined(FEAT_JOB_CHANNEL) || defined(PROTO)
Bram Moolenaar7bffaa92016-03-10 21:46:03 +01004886 static HANDLE
4887job_io_file_open(
Bram Moolenaar972c3b82017-01-12 21:44:49 +01004888 char_u *fname,
4889 DWORD dwDesiredAccess,
4890 DWORD dwShareMode,
4891 LPSECURITY_ATTRIBUTES lpSecurityAttributes,
4892 DWORD dwCreationDisposition,
4893 DWORD dwFlagsAndAttributes)
Bram Moolenaar7bffaa92016-03-10 21:46:03 +01004894{
4895 HANDLE h;
4896# ifdef FEAT_MBYTE
4897 WCHAR *wn = NULL;
4898 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
4899 {
Bram Moolenaar972c3b82017-01-12 21:44:49 +01004900 wn = enc_to_utf16(fname, NULL);
4901 if (wn != NULL)
4902 {
4903 h = CreateFileW(wn, dwDesiredAccess, dwShareMode,
4904 lpSecurityAttributes, dwCreationDisposition,
4905 dwFlagsAndAttributes, NULL);
4906 vim_free(wn);
4907 }
Bram Moolenaar7bffaa92016-03-10 21:46:03 +01004908 }
4909 if (wn == NULL)
4910# endif
Bram Moolenaar972c3b82017-01-12 21:44:49 +01004911 h = CreateFile((LPCSTR)fname, dwDesiredAccess, dwShareMode,
4912 lpSecurityAttributes, dwCreationDisposition,
4913 dwFlagsAndAttributes, NULL);
Bram Moolenaar7bffaa92016-03-10 21:46:03 +01004914 return h;
4915}
4916
Bram Moolenaar942d6b22016-02-07 19:57:16 +01004917 void
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01004918mch_start_job(char *cmd, job_T *job, jobopt_T *options)
Bram Moolenaar942d6b22016-02-07 19:57:16 +01004919{
4920 STARTUPINFO si;
4921 PROCESS_INFORMATION pi;
Bram Moolenaar14207f42016-10-27 21:13:10 +02004922 HANDLE jo;
Bram Moolenaard5d3d302016-03-09 20:54:51 +01004923 SECURITY_ATTRIBUTES saAttr;
4924 channel_T *channel = NULL;
Bram Moolenaard8070362016-02-15 21:56:54 +01004925 HANDLE ifd[2];
4926 HANDLE ofd[2];
4927 HANDLE efd[2];
Bram Moolenaard5d3d302016-03-09 20:54:51 +01004928
4929 int use_null_for_in = options->jo_io[PART_IN] == JIO_NULL;
4930 int use_null_for_out = options->jo_io[PART_OUT] == JIO_NULL;
4931 int use_null_for_err = options->jo_io[PART_ERR] == JIO_NULL;
4932 int use_file_for_in = options->jo_io[PART_IN] == JIO_FILE;
4933 int use_file_for_out = options->jo_io[PART_OUT] == JIO_FILE;
4934 int use_file_for_err = options->jo_io[PART_ERR] == JIO_FILE;
4935 int use_out_for_err = options->jo_io[PART_ERR] == JIO_OUT;
4936
4937 if (use_out_for_err && use_null_for_out)
4938 use_null_for_err = TRUE;
Bram Moolenaard8070362016-02-15 21:56:54 +01004939
4940 ifd[0] = INVALID_HANDLE_VALUE;
4941 ifd[1] = INVALID_HANDLE_VALUE;
4942 ofd[0] = INVALID_HANDLE_VALUE;
4943 ofd[1] = INVALID_HANDLE_VALUE;
4944 efd[0] = INVALID_HANDLE_VALUE;
4945 efd[1] = INVALID_HANDLE_VALUE;
Bram Moolenaar942d6b22016-02-07 19:57:16 +01004946
Bram Moolenaar14207f42016-10-27 21:13:10 +02004947 jo = CreateJobObject(NULL, NULL);
4948 if (jo == NULL)
4949 {
4950 job->jv_status = JOB_FAILED;
4951 goto failed;
4952 }
4953
Bram Moolenaar76467df2016-02-12 19:30:26 +01004954 ZeroMemory(&pi, sizeof(pi));
Bram Moolenaar942d6b22016-02-07 19:57:16 +01004955 ZeroMemory(&si, sizeof(si));
4956 si.cb = sizeof(si);
Bram Moolenaard8070362016-02-15 21:56:54 +01004957 si.dwFlags |= STARTF_USESHOWWINDOW;
Bram Moolenaar76467df2016-02-12 19:30:26 +01004958 si.wShowWindow = SW_HIDE;
Bram Moolenaar942d6b22016-02-07 19:57:16 +01004959
Bram Moolenaard8070362016-02-15 21:56:54 +01004960 saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
4961 saAttr.bInheritHandle = TRUE;
4962 saAttr.lpSecurityDescriptor = NULL;
Bram Moolenaar13d6fb12016-03-08 18:40:52 +01004963
Bram Moolenaarb69fccf2016-03-06 23:06:25 +01004964 if (use_file_for_in)
4965 {
4966 char_u *fname = options->jo_io_name[PART_IN];
4967
Bram Moolenaar7bffaa92016-03-10 21:46:03 +01004968 ifd[0] = job_io_file_open(fname, GENERIC_READ,
4969 FILE_SHARE_READ | FILE_SHARE_WRITE,
4970 &saAttr, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL);
4971 if (ifd[0] == INVALID_HANDLE_VALUE)
Bram Moolenaar94d01912016-03-08 13:48:51 +01004972 {
4973 EMSG2(_(e_notopen), fname);
4974 goto failed;
4975 }
Bram Moolenaarb69fccf2016-03-06 23:06:25 +01004976 }
Bram Moolenaard5d3d302016-03-09 20:54:51 +01004977 else if (!use_null_for_in &&
4978 (!CreatePipe(&ifd[0], &ifd[1], &saAttr, 0)
Bram Moolenaarcea912a2016-10-12 14:20:24 +02004979 || !SetHandleInformation(ifd[1], HANDLE_FLAG_INHERIT, 0)))
Bram Moolenaarb69fccf2016-03-06 23:06:25 +01004980 goto failed;
4981
Bram Moolenaar13d6fb12016-03-08 18:40:52 +01004982 if (use_file_for_out)
4983 {
4984 char_u *fname = options->jo_io_name[PART_OUT];
4985
Bram Moolenaar7bffaa92016-03-10 21:46:03 +01004986 ofd[1] = job_io_file_open(fname, GENERIC_WRITE,
4987 FILE_SHARE_READ | FILE_SHARE_WRITE,
4988 &saAttr, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL);
4989 if (ofd[1] == INVALID_HANDLE_VALUE)
Bram Moolenaar13d6fb12016-03-08 18:40:52 +01004990 {
4991 EMSG2(_(e_notopen), fname);
4992 goto failed;
4993 }
4994 }
Bram Moolenaard5d3d302016-03-09 20:54:51 +01004995 else if (!use_null_for_out &&
4996 (!CreatePipe(&ofd[0], &ofd[1], &saAttr, 0)
Bram Moolenaarcea912a2016-10-12 14:20:24 +02004997 || !SetHandleInformation(ofd[0], HANDLE_FLAG_INHERIT, 0)))
Bram Moolenaarb69fccf2016-03-06 23:06:25 +01004998 goto failed;
4999
Bram Moolenaar13d6fb12016-03-08 18:40:52 +01005000 if (use_file_for_err)
5001 {
5002 char_u *fname = options->jo_io_name[PART_ERR];
5003
Bram Moolenaar7bffaa92016-03-10 21:46:03 +01005004 efd[1] = job_io_file_open(fname, GENERIC_WRITE,
5005 FILE_SHARE_READ | FILE_SHARE_WRITE,
5006 &saAttr, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL);
5007 if (efd[1] == INVALID_HANDLE_VALUE)
Bram Moolenaar13d6fb12016-03-08 18:40:52 +01005008 {
5009 EMSG2(_(e_notopen), fname);
5010 goto failed;
5011 }
5012 }
Bram Moolenaard5d3d302016-03-09 20:54:51 +01005013 else if (!use_out_for_err && !use_null_for_err &&
5014 (!CreatePipe(&efd[0], &efd[1], &saAttr, 0)
Bram Moolenaarcea912a2016-10-12 14:20:24 +02005015 || !SetHandleInformation(efd[0], HANDLE_FLAG_INHERIT, 0)))
Bram Moolenaard8070362016-02-15 21:56:54 +01005016 goto failed;
Bram Moolenaar13d6fb12016-03-08 18:40:52 +01005017
Bram Moolenaard8070362016-02-15 21:56:54 +01005018 si.dwFlags |= STARTF_USESTDHANDLES;
5019 si.hStdInput = ifd[0];
Bram Moolenaard5d3d302016-03-09 20:54:51 +01005020 si.hStdOutput = ofd[1];
5021 si.hStdError = use_out_for_err ? ofd[1] : efd[1];
5022
5023 if (!use_null_for_in || !use_null_for_out || !use_null_for_err)
5024 {
Bram Moolenaarde279892016-03-11 22:19:44 +01005025 if (options->jo_set & JO_CHANNEL)
5026 {
5027 channel = options->jo_channel;
5028 if (channel != NULL)
5029 ++channel->ch_refcount;
5030 }
5031 else
5032 channel = add_channel();
Bram Moolenaard5d3d302016-03-09 20:54:51 +01005033 if (channel == NULL)
5034 goto failed;
5035 }
Bram Moolenaard8070362016-02-15 21:56:54 +01005036
5037 if (!vim_create_process(cmd, TRUE,
Bram Moolenaar14207f42016-10-27 21:13:10 +02005038 CREATE_SUSPENDED |
Bram Moolenaar942d6b22016-02-07 19:57:16 +01005039 CREATE_DEFAULT_ERROR_MODE |
5040 CREATE_NEW_PROCESS_GROUP |
Bram Moolenaar76467df2016-02-12 19:30:26 +01005041 CREATE_NEW_CONSOLE,
Bram Moolenaar942d6b22016-02-07 19:57:16 +01005042 &si, &pi))
Bram Moolenaar76467df2016-02-12 19:30:26 +01005043 {
Bram Moolenaar14207f42016-10-27 21:13:10 +02005044 CloseHandle(jo);
Bram Moolenaar942d6b22016-02-07 19:57:16 +01005045 job->jv_status = JOB_FAILED;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01005046 goto failed;
Bram Moolenaar76467df2016-02-12 19:30:26 +01005047 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01005048
Bram Moolenaar14207f42016-10-27 21:13:10 +02005049 if (!AssignProcessToJobObject(jo, pi.hProcess))
5050 {
5051 /* if failing, switch the way to terminate
5052 * process with TerminateProcess. */
5053 CloseHandle(jo);
5054 jo = NULL;
5055 }
5056 ResumeThread(pi.hThread);
Bram Moolenaar75578a32016-03-10 16:33:31 +01005057 CloseHandle(pi.hThread);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01005058 job->jv_proc_info = pi;
Bram Moolenaar14207f42016-10-27 21:13:10 +02005059 job->jv_job_object = jo;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01005060 job->jv_status = JOB_STARTED;
5061
Bram Moolenaar641ad6c2016-09-01 18:32:11 +02005062 CloseHandle(ifd[0]);
5063 CloseHandle(ofd[1]);
5064 if (!use_out_for_err && !use_null_for_err)
Bram Moolenaarc25558b2016-03-03 21:02:23 +01005065 CloseHandle(efd[1]);
Bram Moolenaard8070362016-02-15 21:56:54 +01005066
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01005067 job->jv_channel = channel;
Bram Moolenaard5d3d302016-03-09 20:54:51 +01005068 if (channel != NULL)
5069 {
5070 channel_set_pipes(channel,
5071 use_file_for_in || use_null_for_in
5072 ? INVALID_FD : (sock_T)ifd[1],
5073 use_file_for_out || use_null_for_out
5074 ? INVALID_FD : (sock_T)ofd[0],
5075 use_out_for_err || use_file_for_err || use_null_for_err
5076 ? INVALID_FD : (sock_T)efd[0]);
5077 channel_set_job(channel, job, options);
Bram Moolenaard5d3d302016-03-09 20:54:51 +01005078 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01005079 return;
5080
5081failed:
Bram Moolenaard8070362016-02-15 21:56:54 +01005082 CloseHandle(ifd[0]);
5083 CloseHandle(ofd[0]);
5084 CloseHandle(efd[0]);
5085 CloseHandle(ifd[1]);
5086 CloseHandle(ofd[1]);
5087 CloseHandle(efd[1]);
Bram Moolenaarde279892016-03-11 22:19:44 +01005088 channel_unref(channel);
Bram Moolenaar942d6b22016-02-07 19:57:16 +01005089}
5090
5091 char *
5092mch_job_status(job_T *job)
5093{
5094 DWORD dwExitCode = 0;
5095
Bram Moolenaar76467df2016-02-12 19:30:26 +01005096 if (!GetExitCodeProcess(job->jv_proc_info.hProcess, &dwExitCode)
5097 || dwExitCode != STILL_ACTIVE)
Bram Moolenaar942d6b22016-02-07 19:57:16 +01005098 {
Bram Moolenaareab089d2016-02-21 19:32:02 +01005099 job->jv_exitval = (int)dwExitCode;
Bram Moolenaar7df915d2016-11-17 17:25:32 +01005100 if (job->jv_status < JOB_ENDED)
Bram Moolenaar97792de2016-10-15 18:36:49 +02005101 {
5102 ch_log(job->jv_channel, "Job ended");
5103 job->jv_status = JOB_ENDED;
5104 }
Bram Moolenaar942d6b22016-02-07 19:57:16 +01005105 return "dead";
5106 }
5107 return "run";
5108}
5109
Bram Moolenaar97792de2016-10-15 18:36:49 +02005110 job_T *
5111mch_detect_ended_job(job_T *job_list)
5112{
5113 HANDLE jobHandles[MAXIMUM_WAIT_OBJECTS];
5114 job_T *jobArray[MAXIMUM_WAIT_OBJECTS];
5115 job_T *job = job_list;
5116
5117 while (job != NULL)
5118 {
5119 DWORD n;
5120 DWORD result;
5121
5122 for (n = 0; n < MAXIMUM_WAIT_OBJECTS
5123 && job != NULL; job = job->jv_next)
5124 {
5125 if (job->jv_status == JOB_STARTED)
5126 {
5127 jobHandles[n] = job->jv_proc_info.hProcess;
5128 jobArray[n] = job;
5129 ++n;
5130 }
5131 }
5132 if (n == 0)
5133 continue;
5134 result = WaitForMultipleObjects(n, jobHandles, FALSE, 0);
5135 if (result >= WAIT_OBJECT_0 && result < WAIT_OBJECT_0 + n)
5136 {
5137 job_T *wait_job = jobArray[result - WAIT_OBJECT_0];
5138
5139 if (STRCMP(mch_job_status(wait_job), "dead") == 0)
5140 return wait_job;
5141 }
5142 }
5143 return NULL;
5144}
5145
Bram Moolenaarfb630902016-10-29 14:55:00 +02005146 static BOOL
5147terminate_all(HANDLE process, int code)
5148{
5149 PROCESSENTRY32 pe;
5150 HANDLE h = INVALID_HANDLE_VALUE;
5151 DWORD pid = GetProcessId(process);
5152
5153 if (pid != 0)
5154 {
5155 h = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
5156 if (h != INVALID_HANDLE_VALUE)
5157 {
5158 pe.dwSize = sizeof(PROCESSENTRY32);
5159 if (!Process32First(h, &pe))
5160 goto theend;
5161
5162 do
5163 {
5164 if (pe.th32ParentProcessID == pid)
5165 {
5166 HANDLE ph = OpenProcess(
5167 PROCESS_ALL_ACCESS, FALSE, pe.th32ProcessID);
5168 if (ph != NULL)
5169 {
5170 terminate_all(ph, code);
5171 CloseHandle(ph);
5172 }
5173 }
5174 } while (Process32Next(h, &pe));
5175
5176 CloseHandle(h);
5177 }
5178 }
5179
5180theend:
5181 return TerminateProcess(process, code);
5182}
5183
5184/*
5185 * Send a (deadly) signal to "job".
5186 * Return FAIL if it didn't work.
5187 */
Bram Moolenaar942d6b22016-02-07 19:57:16 +01005188 int
5189mch_stop_job(job_T *job, char_u *how)
5190{
Bram Moolenaar923d9262016-02-25 20:56:01 +01005191 int ret;
Bram Moolenaar76467df2016-02-12 19:30:26 +01005192
Bram Moolenaar923d9262016-02-25 20:56:01 +01005193 if (STRCMP(how, "term") == 0 || STRCMP(how, "kill") == 0 || *how == NUL)
Bram Moolenaar76467df2016-02-12 19:30:26 +01005194 {
Bram Moolenaarfb630902016-10-29 14:55:00 +02005195 /* deadly signal */
Bram Moolenaar14207f42016-10-27 21:13:10 +02005196 if (job->jv_job_object != NULL)
5197 return TerminateJobObject(job->jv_job_object, 0) ? OK : FAIL;
Bram Moolenaarfb630902016-10-29 14:55:00 +02005198 return terminate_all(job->jv_proc_info.hProcess, 0) ? OK : FAIL;
Bram Moolenaar76467df2016-02-12 19:30:26 +01005199 }
5200
5201 if (!AttachConsole(job->jv_proc_info.dwProcessId))
5202 return FAIL;
5203 ret = GenerateConsoleCtrlEvent(
Bram Moolenaar923d9262016-02-25 20:56:01 +01005204 STRCMP(how, "int") == 0 ? CTRL_C_EVENT : CTRL_BREAK_EVENT,
5205 job->jv_proc_info.dwProcessId)
5206 ? OK : FAIL;
Bram Moolenaar76467df2016-02-12 19:30:26 +01005207 FreeConsole();
5208 return ret;
5209}
5210
5211/*
5212 * Clear the data related to "job".
5213 */
5214 void
5215mch_clear_job(job_T *job)
5216{
5217 if (job->jv_status != JOB_FAILED)
5218 {
Bram Moolenaar14207f42016-10-27 21:13:10 +02005219 if (job->jv_job_object != NULL)
5220 CloseHandle(job->jv_job_object);
Bram Moolenaar76467df2016-02-12 19:30:26 +01005221 CloseHandle(job->jv_proc_info.hProcess);
5222 }
Bram Moolenaar942d6b22016-02-07 19:57:16 +01005223}
5224#endif
5225
Bram Moolenaar071d4272004-06-13 20:20:40 +00005226
5227#ifndef FEAT_GUI_W32
5228
5229/*
5230 * Start termcap mode
5231 */
5232 static void
5233termcap_mode_start(void)
5234{
5235 DWORD cmodein;
5236
5237 if (g_fTermcapMode)
5238 return;
5239
5240 SaveConsoleBuffer(&g_cbNonTermcap);
5241
5242 if (g_cbTermcap.IsValid)
5243 {
5244 /*
5245 * We've been in termcap mode before. Restore certain screen
5246 * characteristics, including the buffer size and the window
5247 * size. Since we will be redrawing the screen, we don't need
5248 * to restore the actual contents of the buffer.
5249 */
5250 RestoreConsoleBuffer(&g_cbTermcap, FALSE);
5251 SetConsoleWindowInfo(g_hConOut, TRUE, &g_cbTermcap.Info.srWindow);
5252 Rows = g_cbTermcap.Info.dwSize.Y;
5253 Columns = g_cbTermcap.Info.dwSize.X;
5254 }
5255 else
5256 {
5257 /*
5258 * This is our first time entering termcap mode. Clear the console
5259 * screen buffer, and resize the buffer to match the current window
5260 * size. We will use this as the size of our editing environment.
5261 */
5262 ClearConsoleBuffer(g_attrCurrent);
5263 ResizeConBufAndWindow(g_hConOut, Columns, Rows);
5264 }
5265
5266#ifdef FEAT_TITLE
5267 resettitle();
5268#endif
5269
5270 GetConsoleMode(g_hConIn, &cmodein);
5271#ifdef FEAT_MOUSE
5272 if (g_fMouseActive)
5273 cmodein |= ENABLE_MOUSE_INPUT;
5274 else
5275 cmodein &= ~ENABLE_MOUSE_INPUT;
5276#endif
5277 cmodein |= ENABLE_WINDOW_INPUT;
5278 SetConsoleMode(g_hConIn, cmodein);
5279
5280 redraw_later_clear();
5281 g_fTermcapMode = TRUE;
5282}
5283
5284
5285/*
5286 * End termcap mode
5287 */
5288 static void
5289termcap_mode_end(void)
5290{
5291 DWORD cmodein;
5292 ConsoleBuffer *cb;
5293 COORD coord;
5294 DWORD dwDummy;
5295
5296 if (!g_fTermcapMode)
5297 return;
5298
5299 SaveConsoleBuffer(&g_cbTermcap);
5300
5301 GetConsoleMode(g_hConIn, &cmodein);
5302 cmodein &= ~(ENABLE_MOUSE_INPUT | ENABLE_WINDOW_INPUT);
5303 SetConsoleMode(g_hConIn, cmodein);
5304
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01005305#ifdef FEAT_RESTORE_ORIG_SCREEN
5306 cb = exiting ? &g_cbOrig : &g_cbNonTermcap;
5307#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00005308 cb = &g_cbNonTermcap;
Bram Moolenaar4c0aac52015-10-30 16:46:55 +01005309#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005310 RestoreConsoleBuffer(cb, p_rs);
5311 SetConsoleCursorInfo(g_hConOut, &g_cci);
5312
5313 if (p_rs || exiting)
5314 {
5315 /*
5316 * Clear anything that happens to be on the current line.
5317 */
5318 coord.X = 0;
5319 coord.Y = (SHORT) (p_rs ? cb->Info.dwCursorPosition.Y : (Rows - 1));
5320 FillConsoleOutputCharacter(g_hConOut, ' ',
5321 cb->Info.dwSize.X, coord, &dwDummy);
5322 /*
5323 * The following is just for aesthetics. If we are exiting without
5324 * restoring the screen, then we want to have a prompt string
5325 * appear at the bottom line. However, the command interpreter
5326 * seems to always advance the cursor one line before displaying
5327 * the prompt string, which causes the screen to scroll. To
5328 * counter this, move the cursor up one line before exiting.
5329 */
5330 if (exiting && !p_rs)
5331 coord.Y--;
5332 /*
5333 * Position the cursor at the leftmost column of the desired row.
5334 */
5335 SetConsoleCursorPosition(g_hConOut, coord);
5336 }
5337
5338 g_fTermcapMode = FALSE;
5339}
5340#endif /* FEAT_GUI_W32 */
5341
5342
5343#ifdef FEAT_GUI_W32
5344 void
5345mch_write(
Bram Moolenaar1266d672017-02-01 13:43:36 +01005346 char_u *s UNUSED,
5347 int len UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005348{
5349 /* never used */
5350}
5351
5352#else
5353
5354/*
5355 * clear `n' chars, starting from `coord'
5356 */
5357 static void
5358clear_chars(
5359 COORD coord,
5360 DWORD n)
5361{
5362 DWORD dwDummy;
5363
5364 FillConsoleOutputCharacter(g_hConOut, ' ', n, coord, &dwDummy);
5365 FillConsoleOutputAttribute(g_hConOut, g_attrCurrent, n, coord, &dwDummy);
5366}
5367
5368
5369/*
5370 * Clear the screen
5371 */
5372 static void
5373clear_screen(void)
5374{
5375 g_coord.X = g_coord.Y = 0;
5376 clear_chars(g_coord, Rows * Columns);
5377}
5378
5379
5380/*
5381 * Clear to end of display
5382 */
5383 static void
5384clear_to_end_of_display(void)
5385{
5386 clear_chars(g_coord, (Rows - g_coord.Y - 1)
5387 * Columns + (Columns - g_coord.X));
5388}
5389
5390
5391/*
5392 * Clear to end of line
5393 */
5394 static void
5395clear_to_end_of_line(void)
5396{
5397 clear_chars(g_coord, Columns - g_coord.X);
5398}
5399
5400
5401/*
5402 * Scroll the scroll region up by `cLines' lines
5403 */
5404 static void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005405scroll(unsigned cLines)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005406{
5407 COORD oldcoord = g_coord;
5408
5409 gotoxy(g_srScrollRegion.Left + 1, g_srScrollRegion.Top + 1);
5410 delete_lines(cLines);
5411
5412 g_coord = oldcoord;
5413}
5414
5415
5416/*
5417 * Set the scroll region
5418 */
5419 static void
5420set_scroll_region(
5421 unsigned left,
5422 unsigned top,
5423 unsigned right,
5424 unsigned bottom)
5425{
5426 if (left >= right
5427 || top >= bottom
5428 || right > (unsigned) Columns - 1
5429 || bottom > (unsigned) Rows - 1)
5430 return;
5431
5432 g_srScrollRegion.Left = left;
5433 g_srScrollRegion.Top = top;
5434 g_srScrollRegion.Right = right;
5435 g_srScrollRegion.Bottom = bottom;
5436}
5437
5438
5439/*
5440 * Insert `cLines' lines at the current cursor position
5441 */
5442 static void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005443insert_lines(unsigned cLines)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005444{
5445 SMALL_RECT source;
5446 COORD dest;
5447 CHAR_INFO fill;
5448
5449 dest.X = 0;
5450 dest.Y = g_coord.Y + cLines;
5451
5452 source.Left = 0;
5453 source.Top = g_coord.Y;
5454 source.Right = g_srScrollRegion.Right;
5455 source.Bottom = g_srScrollRegion.Bottom - cLines;
5456
5457 fill.Char.AsciiChar = ' ';
5458 fill.Attributes = g_attrCurrent;
5459
5460 ScrollConsoleScreenBuffer(g_hConOut, &source, NULL, dest, &fill);
5461
5462 /* Here we have to deal with a win32 console flake: If the scroll
5463 * region looks like abc and we scroll c to a and fill with d we get
5464 * cbd... if we scroll block c one line at a time to a, we get cdd...
5465 * vim expects cdd consistently... So we have to deal with that
5466 * here... (this also occurs scrolling the same way in the other
5467 * direction). */
5468
5469 if (source.Bottom < dest.Y)
5470 {
5471 COORD coord;
5472
5473 coord.X = 0;
5474 coord.Y = source.Bottom;
5475 clear_chars(coord, Columns * (dest.Y - source.Bottom));
5476 }
5477}
5478
5479
5480/*
5481 * Delete `cLines' lines at the current cursor position
5482 */
5483 static void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005484delete_lines(unsigned cLines)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005485{
5486 SMALL_RECT source;
5487 COORD dest;
5488 CHAR_INFO fill;
5489 int nb;
5490
5491 dest.X = 0;
5492 dest.Y = g_coord.Y;
5493
5494 source.Left = 0;
5495 source.Top = g_coord.Y + cLines;
5496 source.Right = g_srScrollRegion.Right;
5497 source.Bottom = g_srScrollRegion.Bottom;
5498
5499 fill.Char.AsciiChar = ' ';
5500 fill.Attributes = g_attrCurrent;
5501
5502 ScrollConsoleScreenBuffer(g_hConOut, &source, NULL, dest, &fill);
5503
5504 /* Here we have to deal with a win32 console flake: If the scroll
5505 * region looks like abc and we scroll c to a and fill with d we get
5506 * cbd... if we scroll block c one line at a time to a, we get cdd...
5507 * vim expects cdd consistently... So we have to deal with that
5508 * here... (this also occurs scrolling the same way in the other
5509 * direction). */
5510
5511 nb = dest.Y + (source.Bottom - source.Top) + 1;
5512
5513 if (nb < source.Top)
5514 {
5515 COORD coord;
5516
5517 coord.X = 0;
5518 coord.Y = nb;
5519 clear_chars(coord, Columns * (source.Top - nb));
5520 }
5521}
5522
5523
5524/*
5525 * Set the cursor position
5526 */
5527 static void
5528gotoxy(
5529 unsigned x,
5530 unsigned y)
5531{
5532 if (x < 1 || x > (unsigned)Columns || y < 1 || y > (unsigned)Rows)
5533 return;
5534
5535 /* external cursor coords are 1-based; internal are 0-based */
5536 g_coord.X = x - 1;
5537 g_coord.Y = y - 1;
5538 SetConsoleCursorPosition(g_hConOut, g_coord);
5539}
5540
5541
5542/*
5543 * Set the current text attribute = (foreground | background)
5544 * See ../doc/os_win32.txt for the numbers.
5545 */
5546 static void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005547textattr(WORD wAttr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005548{
Bram Moolenaar6383b922015-03-24 17:12:19 +01005549 g_attrCurrent = wAttr & 0xff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005550
5551 SetConsoleTextAttribute(g_hConOut, wAttr);
5552}
5553
5554
5555 static void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005556textcolor(WORD wAttr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005557{
Bram Moolenaar6383b922015-03-24 17:12:19 +01005558 g_attrCurrent = (g_attrCurrent & 0xf0) + (wAttr & 0x0f);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005559
5560 SetConsoleTextAttribute(g_hConOut, g_attrCurrent);
5561}
5562
5563
5564 static void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005565textbackground(WORD wAttr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005566{
Bram Moolenaar6383b922015-03-24 17:12:19 +01005567 g_attrCurrent = (g_attrCurrent & 0x0f) + ((wAttr & 0x0f) << 4);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005568
5569 SetConsoleTextAttribute(g_hConOut, g_attrCurrent);
5570}
5571
5572
5573/*
5574 * restore the default text attribute (whatever we started with)
5575 */
5576 static void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005577normvideo(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005578{
5579 textattr(g_attrDefault);
5580}
5581
5582
5583static WORD g_attrPreStandout = 0;
5584
5585/*
5586 * Make the text standout, by brightening it
5587 */
5588 static void
5589standout(void)
5590{
5591 g_attrPreStandout = g_attrCurrent;
5592 textattr((WORD) (g_attrCurrent|FOREGROUND_INTENSITY|BACKGROUND_INTENSITY));
5593}
5594
5595
5596/*
5597 * Turn off standout mode
5598 */
5599 static void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005600standend(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005601{
5602 if (g_attrPreStandout)
5603 {
5604 textattr(g_attrPreStandout);
5605 g_attrPreStandout = 0;
5606 }
5607}
5608
5609
5610/*
Bram Moolenaarff1d0d42007-05-10 17:24:16 +00005611 * Set normal fg/bg color, based on T_ME. Called when t_me has been set.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005612 */
5613 void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005614mch_set_normal_colors(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005615{
5616 char_u *p;
5617 int n;
5618
5619 cterm_normal_fg_color = (g_attrDefault & 0xf) + 1;
5620 cterm_normal_bg_color = ((g_attrDefault >> 4) & 0xf) + 1;
5621 if (T_ME[0] == ESC && T_ME[1] == '|')
5622 {
5623 p = T_ME + 2;
5624 n = getdigits(&p);
5625 if (*p == 'm' && n > 0)
5626 {
5627 cterm_normal_fg_color = (n & 0xf) + 1;
5628 cterm_normal_bg_color = ((n >> 4) & 0xf) + 1;
5629 }
5630 }
5631}
5632
5633
5634/*
5635 * visual bell: flash the screen
5636 */
5637 static void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005638visual_bell(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005639{
5640 COORD coordOrigin = {0, 0};
5641 WORD attrFlash = ~g_attrCurrent & 0xff;
5642
5643 DWORD dwDummy;
5644 LPWORD oldattrs = (LPWORD)alloc(Rows * Columns * sizeof(WORD));
5645
5646 if (oldattrs == NULL)
5647 return;
5648 ReadConsoleOutputAttribute(g_hConOut, oldattrs, Rows * Columns,
5649 coordOrigin, &dwDummy);
5650 FillConsoleOutputAttribute(g_hConOut, attrFlash, Rows * Columns,
5651 coordOrigin, &dwDummy);
5652
5653 Sleep(15); /* wait for 15 msec */
5654 WriteConsoleOutputAttribute(g_hConOut, oldattrs, Rows * Columns,
5655 coordOrigin, &dwDummy);
5656 vim_free(oldattrs);
5657}
5658
5659
5660/*
5661 * Make the cursor visible or invisible
5662 */
5663 static void
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00005664cursor_visible(BOOL fVisible)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005665{
5666 s_cursor_visible = fVisible;
5667#ifdef MCH_CURSOR_SHAPE
5668 mch_update_cursor();
5669#endif
5670}
5671
5672
5673/*
Bram Moolenaarac360bf2015-09-01 20:31:20 +02005674 * write `cbToWrite' bytes in `pchBuf' to the screen
5675 * Returns the number of bytes actually written (at least one).
Bram Moolenaar071d4272004-06-13 20:20:40 +00005676 */
Bram Moolenaarac360bf2015-09-01 20:31:20 +02005677 static DWORD
Bram Moolenaar071d4272004-06-13 20:20:40 +00005678write_chars(
Bram Moolenaarac360bf2015-09-01 20:31:20 +02005679 char_u *pchBuf,
5680 DWORD cbToWrite)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005681{
5682 COORD coord = g_coord;
5683 DWORD written;
5684
Bram Moolenaarac360bf2015-09-01 20:31:20 +02005685#ifdef FEAT_MBYTE
5686 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
5687 {
5688 static WCHAR *unicodebuf = NULL;
5689 static int unibuflen = 0;
5690 int length;
5691 DWORD n, cchwritten, cells;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005692
Bram Moolenaarac360bf2015-09-01 20:31:20 +02005693 length = MultiByteToWideChar(CP_UTF8, 0, (LPCSTR)pchBuf, cbToWrite, 0, 0);
5694 if (unicodebuf == NULL || length > unibuflen)
5695 {
5696 vim_free(unicodebuf);
5697 unicodebuf = (WCHAR *)lalloc(length * sizeof(WCHAR), FALSE);
5698 unibuflen = length;
5699 }
5700 MultiByteToWideChar(CP_UTF8, 0, (LPCSTR)pchBuf, cbToWrite,
5701 unicodebuf, unibuflen);
5702
5703 cells = mb_string2cells(pchBuf, cbToWrite);
5704 FillConsoleOutputAttribute(g_hConOut, g_attrCurrent, cells,
5705 coord, &written);
5706 /* When writing fails or didn't write a single character, pretend one
5707 * character was written, otherwise we get stuck. */
5708 if (WriteConsoleOutputCharacterW(g_hConOut, unicodebuf, length,
5709 coord, &cchwritten) == 0
5710 || cchwritten == 0)
5711 cchwritten = 1;
5712
5713 if (cchwritten == length)
5714 {
5715 written = cbToWrite;
5716 g_coord.X += (SHORT)cells;
5717 }
5718 else
5719 {
5720 char_u *p = pchBuf;
5721 for (n = 0; n < cchwritten; n++)
5722 mb_cptr_adv(p);
5723 written = p - pchBuf;
5724 g_coord.X += (SHORT)mb_string2cells(pchBuf, written);
5725 }
5726 }
5727 else
5728#endif
5729 {
5730 FillConsoleOutputAttribute(g_hConOut, g_attrCurrent, cbToWrite,
5731 coord, &written);
5732 /* When writing fails or didn't write a single character, pretend one
5733 * character was written, otherwise we get stuck. */
5734 if (WriteConsoleOutputCharacter(g_hConOut, (LPCSTR)pchBuf, cbToWrite,
5735 coord, &written) == 0
5736 || written == 0)
5737 written = 1;
5738
5739 g_coord.X += (SHORT) written;
5740 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005741
5742 while (g_coord.X > g_srScrollRegion.Right)
5743 {
5744 g_coord.X -= (SHORT) Columns;
5745 if (g_coord.Y < g_srScrollRegion.Bottom)
5746 g_coord.Y++;
5747 }
5748
5749 gotoxy(g_coord.X + 1, g_coord.Y + 1);
5750
5751 return written;
5752}
5753
5754
5755/*
5756 * mch_write(): write the output buffer to the screen, translating ESC
5757 * sequences into calls to console output routines.
5758 */
5759 void
5760mch_write(
5761 char_u *s,
5762 int len)
5763{
5764 s[len] = NUL;
5765
5766 if (!term_console)
5767 {
5768 write(1, s, (unsigned)len);
5769 return;
5770 }
5771
5772 /* translate ESC | sequences into faked bios calls */
5773 while (len--)
5774 {
5775 /* optimization: use one single write_chars for runs of text,
5776 * rather than once per character It ain't curses, but it helps. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005777 DWORD prefix = (DWORD)strcspn((char *)s, "\n\r\b\a\033");
Bram Moolenaar071d4272004-06-13 20:20:40 +00005778
5779 if (p_wd)
5780 {
5781 WaitForChar(p_wd);
5782 if (prefix != 0)
5783 prefix = 1;
5784 }
5785
5786 if (prefix != 0)
5787 {
5788 DWORD nWritten;
5789
5790 nWritten = write_chars(s, prefix);
5791#ifdef MCH_WRITE_DUMP
5792 if (fdDump)
5793 {
5794 fputc('>', fdDump);
5795 fwrite(s, sizeof(char_u), nWritten, fdDump);
5796 fputs("<\n", fdDump);
5797 }
5798#endif
5799 len -= (nWritten - 1);
5800 s += nWritten;
5801 }
5802 else if (s[0] == '\n')
5803 {
5804 /* \n, newline: go to the beginning of the next line or scroll */
5805 if (g_coord.Y == g_srScrollRegion.Bottom)
5806 {
5807 scroll(1);
5808 gotoxy(g_srScrollRegion.Left + 1, g_srScrollRegion.Bottom + 1);
5809 }
5810 else
5811 {
5812 gotoxy(g_srScrollRegion.Left + 1, g_coord.Y + 2);
5813 }
5814#ifdef MCH_WRITE_DUMP
5815 if (fdDump)
5816 fputs("\\n\n", fdDump);
5817#endif
5818 s++;
5819 }
5820 else if (s[0] == '\r')
5821 {
5822 /* \r, carriage return: go to beginning of line */
5823 gotoxy(g_srScrollRegion.Left+1, g_coord.Y + 1);
5824#ifdef MCH_WRITE_DUMP
5825 if (fdDump)
5826 fputs("\\r\n", fdDump);
5827#endif
5828 s++;
5829 }
5830 else if (s[0] == '\b')
5831 {
5832 /* \b, backspace: move cursor one position left */
5833 if (g_coord.X > g_srScrollRegion.Left)
5834 g_coord.X--;
5835 else if (g_coord.Y > g_srScrollRegion.Top)
5836 {
5837 g_coord.X = g_srScrollRegion.Right;
5838 g_coord.Y--;
5839 }
5840 gotoxy(g_coord.X + 1, g_coord.Y + 1);
5841#ifdef MCH_WRITE_DUMP
5842 if (fdDump)
5843 fputs("\\b\n", fdDump);
5844#endif
5845 s++;
5846 }
5847 else if (s[0] == '\a')
5848 {
5849 /* \a, bell */
5850 MessageBeep(0xFFFFFFFF);
5851#ifdef MCH_WRITE_DUMP
5852 if (fdDump)
5853 fputs("\\a\n", fdDump);
5854#endif
5855 s++;
5856 }
5857 else if (s[0] == ESC && len >= 3-1 && s[1] == '|')
5858 {
5859#ifdef MCH_WRITE_DUMP
Bram Moolenaarc0197e22004-09-13 20:26:32 +00005860 char_u *old_s = s;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005861#endif
Bram Moolenaarc0197e22004-09-13 20:26:32 +00005862 char_u *p;
5863 int arg1 = 0, arg2 = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005864
5865 switch (s[2])
5866 {
5867 /* one or two numeric arguments, separated by ';' */
5868
5869 case '0': case '1': case '2': case '3': case '4':
5870 case '5': case '6': case '7': case '8': case '9':
5871 p = s + 2;
5872 arg1 = getdigits(&p); /* no check for length! */
5873 if (p > s + len)
5874 break;
5875
5876 if (*p == ';')
5877 {
5878 ++p;
5879 arg2 = getdigits(&p); /* no check for length! */
5880 if (p > s + len)
5881 break;
5882
5883 if (*p == 'H')
5884 gotoxy(arg2, arg1);
5885 else if (*p == 'r')
5886 set_scroll_region(0, arg1 - 1, Columns - 1, arg2 - 1);
5887 }
5888 else if (*p == 'A')
5889 {
5890 /* move cursor up arg1 lines in same column */
5891 gotoxy(g_coord.X + 1,
5892 max(g_srScrollRegion.Top, g_coord.Y - arg1) + 1);
5893 }
5894 else if (*p == 'C')
5895 {
5896 /* move cursor right arg1 columns in same line */
5897 gotoxy(min(g_srScrollRegion.Right, g_coord.X + arg1) + 1,
5898 g_coord.Y + 1);
5899 }
5900 else if (*p == 'H')
5901 {
5902 gotoxy(1, arg1);
5903 }
5904 else if (*p == 'L')
5905 {
5906 insert_lines(arg1);
5907 }
5908 else if (*p == 'm')
5909 {
5910 if (arg1 == 0)
5911 normvideo();
5912 else
5913 textattr((WORD) arg1);
5914 }
5915 else if (*p == 'f')
5916 {
5917 textcolor((WORD) arg1);
5918 }
5919 else if (*p == 'b')
5920 {
5921 textbackground((WORD) arg1);
5922 }
5923 else if (*p == 'M')
5924 {
5925 delete_lines(arg1);
5926 }
5927
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00005928 len -= (int)(p - s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005929 s = p + 1;
5930 break;
5931
5932
5933 /* Three-character escape sequences */
5934
5935 case 'A':
5936 /* move cursor up one line in same column */
5937 gotoxy(g_coord.X + 1,
5938 max(g_srScrollRegion.Top, g_coord.Y - 1) + 1);
5939 goto got3;
5940
5941 case 'B':
5942 visual_bell();
5943 goto got3;
5944
5945 case 'C':
5946 /* move cursor right one column in same line */
5947 gotoxy(min(g_srScrollRegion.Right, g_coord.X + 1) + 1,
5948 g_coord.Y + 1);
5949 goto got3;
5950
5951 case 'E':
5952 termcap_mode_end();
5953 goto got3;
5954
5955 case 'F':
5956 standout();
5957 goto got3;
5958
5959 case 'f':
5960 standend();
5961 goto got3;
5962
5963 case 'H':
5964 gotoxy(1, 1);
5965 goto got3;
5966
5967 case 'j':
5968 clear_to_end_of_display();
5969 goto got3;
5970
5971 case 'J':
5972 clear_screen();
5973 goto got3;
5974
5975 case 'K':
5976 clear_to_end_of_line();
5977 goto got3;
5978
5979 case 'L':
5980 insert_lines(1);
5981 goto got3;
5982
5983 case 'M':
5984 delete_lines(1);
5985 goto got3;
5986
5987 case 'S':
5988 termcap_mode_start();
5989 goto got3;
5990
5991 case 'V':
5992 cursor_visible(TRUE);
5993 goto got3;
5994
5995 case 'v':
5996 cursor_visible(FALSE);
5997 goto got3;
5998
5999 got3:
6000 s += 3;
6001 len -= 2;
6002 }
6003
6004#ifdef MCH_WRITE_DUMP
6005 if (fdDump)
6006 {
6007 fputs("ESC | ", fdDump);
6008 fwrite(old_s + 2, sizeof(char_u), s - old_s - 2, fdDump);
6009 fputc('\n', fdDump);
6010 }
6011#endif
6012 }
6013 else
6014 {
6015 /* Write a single character */
6016 DWORD nWritten;
6017
6018 nWritten = write_chars(s, 1);
6019#ifdef MCH_WRITE_DUMP
6020 if (fdDump)
6021 {
6022 fputc('>', fdDump);
6023 fwrite(s, sizeof(char_u), nWritten, fdDump);
6024 fputs("<\n", fdDump);
6025 }
6026#endif
6027
6028 len -= (nWritten - 1);
6029 s += nWritten;
6030 }
6031 }
6032
6033#ifdef MCH_WRITE_DUMP
6034 if (fdDump)
6035 fflush(fdDump);
6036#endif
6037}
6038
6039#endif /* FEAT_GUI_W32 */
6040
6041
6042/*
Bram Moolenaar0e0b3dd2016-03-17 17:58:56 +01006043 * Delay for "msec" milliseconds.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006044 */
6045 void
6046mch_delay(
6047 long msec,
Bram Moolenaar1266d672017-02-01 13:43:36 +01006048 int ignoreinput UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006049{
6050#ifdef FEAT_GUI_W32
6051 Sleep((int)msec); /* never wait for input */
Bram Moolenaar325b7a22004-07-05 15:58:32 +00006052#else /* Console */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006053 if (ignoreinput)
Bram Moolenaar325b7a22004-07-05 15:58:32 +00006054# ifdef FEAT_MZSCHEME
6055 if (mzthreads_allowed() && p_mzq > 0 && msec > p_mzq)
6056 {
6057 int towait = p_mzq;
6058
6059 /* if msec is large enough, wait by portions in p_mzq */
6060 while (msec > 0)
6061 {
6062 mzvim_check_threads();
6063 if (msec < towait)
6064 towait = msec;
6065 Sleep(towait);
6066 msec -= towait;
6067 }
6068 }
6069 else
6070# endif
6071 Sleep((int)msec);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006072 else
6073 WaitForChar(msec);
6074#endif
6075}
6076
6077
6078/*
Bram Moolenaar203258c2016-01-17 22:15:16 +01006079 * This version of remove is not scared by a readonly (backup) file.
6080 * This can also remove a symbolic link like Unix.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006081 * Return 0 for success, -1 for failure.
6082 */
6083 int
6084mch_remove(char_u *name)
6085{
6086#ifdef FEAT_MBYTE
6087 WCHAR *wn = NULL;
6088 int n;
Bram Moolenaar12b559e2013-06-12 22:41:37 +02006089#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006090
Bram Moolenaar203258c2016-01-17 22:15:16 +01006091 /*
6092 * On Windows, deleting a directory's symbolic link is done by
6093 * RemoveDirectory(): mch_rmdir. It seems unnatural, but it is fact.
6094 */
6095 if (mch_isdir(name) && mch_is_symbolic_link(name))
6096 return mch_rmdir(name);
6097
Bram Moolenaar12b559e2013-06-12 22:41:37 +02006098 win32_setattrs(name, FILE_ATTRIBUTE_NORMAL);
6099
6100#ifdef FEAT_MBYTE
Bram Moolenaar071d4272004-06-13 20:20:40 +00006101 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
6102 {
Bram Moolenaar36f692d2008-11-20 16:10:17 +00006103 wn = enc_to_utf16(name, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006104 if (wn != NULL)
6105 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006106 n = DeleteFileW(wn) ? 0 : -1;
6107 vim_free(wn);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006108 return n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006109 }
6110 }
6111#endif
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01006112 return DeleteFile((LPCSTR)name) ? 0 : -1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006113}
6114
6115
6116/*
Bram Moolenaarb9c31e72016-09-29 15:18:57 +02006117 * Check for an "interrupt signal": CTRL-break or CTRL-C.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006118 */
6119 void
Bram Moolenaarb9c31e72016-09-29 15:18:57 +02006120mch_breakcheck(int force)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006121{
6122#ifndef FEAT_GUI_W32 /* never used */
6123 if (g_fCtrlCPressed || g_fCBrkPressed)
6124 {
6125 g_fCtrlCPressed = g_fCBrkPressed = FALSE;
6126 got_int = TRUE;
6127 }
6128#endif
6129}
6130
Bram Moolenaaree273972016-01-02 21:11:51 +01006131/* physical RAM to leave for the OS */
6132#define WINNT_RESERVE_BYTES (256*1024*1024)
Bram Moolenaaree273972016-01-02 21:11:51 +01006133
6134/*
6135 * How much main memory in KiB that can be used by VIM.
6136 */
Bram Moolenaaree273972016-01-02 21:11:51 +01006137 long_u
Bram Moolenaar1266d672017-02-01 13:43:36 +01006138mch_total_mem(int special UNUSED)
Bram Moolenaaree273972016-01-02 21:11:51 +01006139{
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006140 MEMORYSTATUSEX ms;
6141
Bram Moolenaaree273972016-01-02 21:11:51 +01006142 PlatformId();
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006143 /* Need to use GlobalMemoryStatusEx() when there is more memory than
6144 * what fits in 32 bits. But it's not always available. */
6145 ms.dwLength = sizeof(MEMORYSTATUSEX);
6146 GlobalMemoryStatusEx(&ms);
6147 if (ms.ullAvailVirtual < ms.ullTotalPhys)
Bram Moolenaaree273972016-01-02 21:11:51 +01006148 {
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006149 /* Process address space fits in physical RAM, use all of it. */
6150 return (long_u)(ms.ullAvailVirtual / 1024);
Bram Moolenaaree273972016-01-02 21:11:51 +01006151 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006152 if (ms.ullTotalPhys <= WINNT_RESERVE_BYTES)
Bram Moolenaaree273972016-01-02 21:11:51 +01006153 {
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006154 /* Catch old NT box or perverse hardware setup. */
6155 return (long_u)((ms.ullTotalPhys / 2) / 1024);
Bram Moolenaaree273972016-01-02 21:11:51 +01006156 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006157 /* Use physical RAM less reserve for OS + data. */
6158 return (long_u)((ms.ullTotalPhys - WINNT_RESERVE_BYTES) / 1024);
Bram Moolenaaree273972016-01-02 21:11:51 +01006159}
Bram Moolenaar071d4272004-06-13 20:20:40 +00006160
Bram Moolenaar071d4272004-06-13 20:20:40 +00006161#ifdef FEAT_MBYTE
6162/*
6163 * Same code as below, but with wide functions and no comments.
6164 * Return 0 for success, non-zero for failure.
6165 */
6166 int
6167mch_wrename(WCHAR *wold, WCHAR *wnew)
6168{
6169 WCHAR *p;
6170 int i;
6171 WCHAR szTempFile[_MAX_PATH + 1];
6172 WCHAR szNewPath[_MAX_PATH + 1];
6173 HANDLE hf;
6174
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006175 p = wold;
6176 for (i = 0; wold[i] != NUL; ++i)
6177 if ((wold[i] == '/' || wold[i] == '\\' || wold[i] == ':')
6178 && wold[i + 1] != 0)
6179 p = wold + i + 1;
6180 if ((int)(wold + i - p) < 8 || p[6] != '~')
6181 return (MoveFileW(wold, wnew) == 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006182
6183 if (GetFullPathNameW(wnew, _MAX_PATH, szNewPath, &p) == 0 || p == NULL)
6184 return -1;
6185 *p = NUL;
6186
6187 if (GetTempFileNameW(szNewPath, L"VIM", 0, szTempFile) == 0)
6188 return -2;
6189
6190 if (!DeleteFileW(szTempFile))
6191 return -3;
6192
6193 if (!MoveFileW(wold, szTempFile))
6194 return -4;
6195
6196 if ((hf = CreateFileW(wold, GENERIC_WRITE, 0, NULL, CREATE_NEW,
6197 FILE_ATTRIBUTE_NORMAL, NULL)) == INVALID_HANDLE_VALUE)
6198 return -5;
6199 if (!CloseHandle(hf))
6200 return -6;
6201
6202 if (!MoveFileW(szTempFile, wnew))
6203 {
6204 (void)MoveFileW(szTempFile, wold);
6205 return -7;
6206 }
6207
6208 DeleteFileW(szTempFile);
6209
6210 if (!DeleteFileW(wold))
6211 return -8;
6212
6213 return 0;
6214}
6215#endif
6216
6217
6218/*
6219 * mch_rename() works around a bug in rename (aka MoveFile) in
6220 * Windows 95: rename("foo.bar", "foo.bar~") will generate a
6221 * file whose short file name is "FOO.BAR" (its long file name will
6222 * be correct: "foo.bar~"). Because a file can be accessed by
6223 * either its SFN or its LFN, "foo.bar" has effectively been
6224 * renamed to "foo.bar", which is not at all what was wanted. This
6225 * seems to happen only when renaming files with three-character
6226 * extensions by appending a suffix that does not include ".".
6227 * Windows NT gets it right, however, with an SFN of "FOO~1.BAR".
6228 *
6229 * There is another problem, which isn't really a bug but isn't right either:
6230 * When renaming "abcdef~1.txt" to "abcdef~1.txt~", the short name can be
6231 * "abcdef~1.txt" again. This has been reported on Windows NT 4.0 with
6232 * service pack 6. Doesn't seem to happen on Windows 98.
6233 *
6234 * Like rename(), returns 0 upon success, non-zero upon failure.
6235 * Should probably set errno appropriately when errors occur.
6236 */
6237 int
6238mch_rename(
6239 const char *pszOldFile,
6240 const char *pszNewFile)
6241{
6242 char szTempFile[_MAX_PATH+1];
6243 char szNewPath[_MAX_PATH+1];
6244 char *pszFilePart;
6245 HANDLE hf;
6246#ifdef FEAT_MBYTE
6247 WCHAR *wold = NULL;
6248 WCHAR *wnew = NULL;
6249 int retval = -1;
6250
6251 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
6252 {
Bram Moolenaar36f692d2008-11-20 16:10:17 +00006253 wold = enc_to_utf16((char_u *)pszOldFile, NULL);
6254 wnew = enc_to_utf16((char_u *)pszNewFile, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006255 if (wold != NULL && wnew != NULL)
6256 retval = mch_wrename(wold, wnew);
6257 vim_free(wold);
6258 vim_free(wnew);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006259 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006260 }
6261#endif
6262
6263 /*
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006264 * No need to play tricks unless the file name contains a "~" as the
6265 * seventh character.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006266 */
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006267 pszFilePart = (char *)gettail((char_u *)pszOldFile);
6268 if (STRLEN(pszFilePart) < 8 || pszFilePart[6] != '~')
6269 return rename(pszOldFile, pszNewFile);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006270
6271 /* Get base path of new file name. Undocumented feature: If pszNewFile is
6272 * a directory, no error is returned and pszFilePart will be NULL. */
6273 if (GetFullPathName(pszNewFile, _MAX_PATH, szNewPath, &pszFilePart) == 0
6274 || pszFilePart == NULL)
6275 return -1;
6276 *pszFilePart = NUL;
6277
6278 /* Get (and create) a unique temporary file name in directory of new file */
6279 if (GetTempFileName(szNewPath, "VIM", 0, szTempFile) == 0)
6280 return -2;
6281
6282 /* blow the temp file away */
6283 if (!DeleteFile(szTempFile))
6284 return -3;
6285
6286 /* rename old file to the temp file */
6287 if (!MoveFile(pszOldFile, szTempFile))
6288 return -4;
6289
6290 /* now create an empty file called pszOldFile; this prevents the operating
6291 * system using pszOldFile as an alias (SFN) if we're renaming within the
6292 * same directory. For example, we're editing a file called
6293 * filename.asc.txt by its SFN, filena~1.txt. If we rename filena~1.txt
6294 * to filena~1.txt~ (i.e., we're making a backup while writing it), the
6295 * SFN for filena~1.txt~ will be filena~1.txt, by default, which will
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006296 * cause all sorts of problems later in buf_write(). So, we create an
6297 * empty file called filena~1.txt and the system will have to find some
6298 * other SFN for filena~1.txt~, such as filena~2.txt
Bram Moolenaar071d4272004-06-13 20:20:40 +00006299 */
6300 if ((hf = CreateFile(pszOldFile, GENERIC_WRITE, 0, NULL, CREATE_NEW,
6301 FILE_ATTRIBUTE_NORMAL, NULL)) == INVALID_HANDLE_VALUE)
6302 return -5;
6303 if (!CloseHandle(hf))
6304 return -6;
6305
6306 /* rename the temp file to the new file */
6307 if (!MoveFile(szTempFile, pszNewFile))
6308 {
6309 /* Renaming failed. Rename the file back to its old name, so that it
6310 * looks like nothing happened. */
6311 (void)MoveFile(szTempFile, pszOldFile);
6312
6313 return -7;
6314 }
6315
6316 /* Seems to be left around on Novell filesystems */
6317 DeleteFile(szTempFile);
6318
6319 /* finally, remove the empty old file */
6320 if (!DeleteFile(pszOldFile))
6321 return -8;
6322
6323 return 0; /* success */
6324}
6325
6326/*
6327 * Get the default shell for the current hardware platform
6328 */
6329 char *
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006330default_shell(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006331{
Bram Moolenaar071d4272004-06-13 20:20:40 +00006332 PlatformId();
6333
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006334 return "cmd.exe";
Bram Moolenaar071d4272004-06-13 20:20:40 +00006335}
6336
6337/*
6338 * mch_access() extends access() to do more detailed check on network drives.
6339 * Returns 0 if file "n" has access rights according to "p", -1 otherwise.
6340 */
6341 int
6342mch_access(char *n, int p)
6343{
6344 HANDLE hFile;
6345 DWORD am;
6346 int retval = -1; /* default: fail */
6347#ifdef FEAT_MBYTE
6348 WCHAR *wn = NULL;
6349
6350 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01006351 wn = enc_to_utf16((char_u *)n, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006352#endif
6353
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01006354 if (mch_isdir((char_u *)n))
Bram Moolenaar071d4272004-06-13 20:20:40 +00006355 {
6356 char TempName[_MAX_PATH + 16] = "";
6357#ifdef FEAT_MBYTE
6358 WCHAR TempNameW[_MAX_PATH + 16] = L"";
6359#endif
6360
6361 if (p & R_OK)
6362 {
6363 /* Read check is performed by seeing if we can do a find file on
6364 * the directory for any file. */
6365#ifdef FEAT_MBYTE
6366 if (wn != NULL)
6367 {
6368 int i;
6369 WIN32_FIND_DATAW d;
6370
6371 for (i = 0; i < _MAX_PATH && wn[i] != 0; ++i)
6372 TempNameW[i] = wn[i];
6373 if (TempNameW[i - 1] != '\\' && TempNameW[i - 1] != '/')
6374 TempNameW[i++] = '\\';
6375 TempNameW[i++] = '*';
6376 TempNameW[i++] = 0;
6377
6378 hFile = FindFirstFileW(TempNameW, &d);
6379 if (hFile == INVALID_HANDLE_VALUE)
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006380 goto getout;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006381 else
6382 (void)FindClose(hFile);
6383 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006384 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00006385#endif
6386 {
6387 char *pch;
6388 WIN32_FIND_DATA d;
6389
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01006390 vim_strncpy((char_u *)TempName, (char_u *)n, _MAX_PATH);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006391 pch = TempName + STRLEN(TempName) - 1;
6392 if (*pch != '\\' && *pch != '/')
6393 *++pch = '\\';
6394 *++pch = '*';
6395 *++pch = NUL;
6396
6397 hFile = FindFirstFile(TempName, &d);
6398 if (hFile == INVALID_HANDLE_VALUE)
6399 goto getout;
6400 (void)FindClose(hFile);
6401 }
6402 }
6403
6404 if (p & W_OK)
6405 {
6406 /* Trying to create a temporary file in the directory should catch
6407 * directories on read-only network shares. However, in
6408 * directories whose ACL allows writes but denies deletes will end
6409 * up keeping the temporary file :-(. */
6410#ifdef FEAT_MBYTE
6411 if (wn != NULL)
6412 {
6413 if (!GetTempFileNameW(wn, L"VIM", 0, TempNameW))
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006414 goto getout;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006415 else
6416 DeleteFileW(TempNameW);
6417 }
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006418 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00006419#endif
6420 {
6421 if (!GetTempFileName(n, "VIM", 0, TempName))
6422 goto getout;
6423 mch_remove((char_u *)TempName);
6424 }
6425 }
6426 }
6427 else
6428 {
6429 /* Trying to open the file for the required access does ACL, read-only
6430 * network share, and file attribute checks. */
6431 am = ((p & W_OK) ? GENERIC_WRITE : 0)
6432 | ((p & R_OK) ? GENERIC_READ : 0);
6433#ifdef FEAT_MBYTE
6434 if (wn != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006435 hFile = CreateFileW(wn, am, 0, NULL, OPEN_EXISTING, 0, NULL);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006436 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00006437#endif
6438 hFile = CreateFile(n, am, 0, NULL, OPEN_EXISTING, 0, NULL);
6439 if (hFile == INVALID_HANDLE_VALUE)
6440 goto getout;
6441 CloseHandle(hFile);
6442 }
6443
6444 retval = 0; /* success */
6445getout:
6446#ifdef FEAT_MBYTE
6447 vim_free(wn);
6448#endif
6449 return retval;
6450}
6451
6452#if defined(FEAT_MBYTE) || defined(PROTO)
6453/*
Bram Moolenaar36f692d2008-11-20 16:10:17 +00006454 * Version of open() that may use UTF-16 file name.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006455 */
6456 int
6457mch_open(char *name, int flags, int mode)
6458{
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006459 /* _wopen() does not work with Borland C 5.5: creates a read-only file. */
6460# ifndef __BORLANDC__
Bram Moolenaar071d4272004-06-13 20:20:40 +00006461 WCHAR *wn;
6462 int f;
6463
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006464 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006465 {
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01006466 wn = enc_to_utf16((char_u *)name, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006467 if (wn != NULL)
6468 {
6469 f = _wopen(wn, flags, mode);
6470 vim_free(wn);
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006471 return f;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006472 }
6473 }
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006474# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006475
Bram Moolenaarf9e6c3b2014-11-05 18:36:03 +01006476 /* open() can open a file which name is longer than _MAX_PATH bytes
6477 * and shorter than _MAX_PATH characters successfully, but sometimes it
6478 * causes unexpected error in another part. We make it an error explicitly
6479 * here. */
6480 if (strlen(name) >= _MAX_PATH)
6481 return -1;
6482
Bram Moolenaar071d4272004-06-13 20:20:40 +00006483 return open(name, flags, mode);
6484}
6485
6486/*
Bram Moolenaar36f692d2008-11-20 16:10:17 +00006487 * Version of fopen() that may use UTF-16 file name.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006488 */
6489 FILE *
6490mch_fopen(char *name, char *mode)
6491{
6492 WCHAR *wn, *wm;
6493 FILE *f = NULL;
6494
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006495 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006496 {
Bram Moolenaare6a91fd2008-07-24 18:51:11 +00006497# if defined(DEBUG) && _MSC_VER >= 1400
Bram Moolenaar0fde2902008-03-16 13:54:13 +00006498 /* Work around an annoying assertion in the Microsoft debug CRT
6499 * when mode's text/binary setting doesn't match _get_fmode(). */
6500 char newMode = mode[strlen(mode) - 1];
6501 int oldMode = 0;
6502
6503 _get_fmode(&oldMode);
6504 if (newMode == 't')
6505 _set_fmode(_O_TEXT);
6506 else if (newMode == 'b')
6507 _set_fmode(_O_BINARY);
6508# endif
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01006509 wn = enc_to_utf16((char_u *)name, NULL);
6510 wm = enc_to_utf16((char_u *)mode, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006511 if (wn != NULL && wm != NULL)
6512 f = _wfopen(wn, wm);
6513 vim_free(wn);
6514 vim_free(wm);
Bram Moolenaar0fde2902008-03-16 13:54:13 +00006515
Bram Moolenaare6a91fd2008-07-24 18:51:11 +00006516# if defined(DEBUG) && _MSC_VER >= 1400
Bram Moolenaar0fde2902008-03-16 13:54:13 +00006517 _set_fmode(oldMode);
6518# endif
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006519 return f;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006520 }
6521
Bram Moolenaarf9e6c3b2014-11-05 18:36:03 +01006522 /* fopen() can open a file which name is longer than _MAX_PATH bytes
6523 * and shorter than _MAX_PATH characters successfully, but sometimes it
6524 * causes unexpected error in another part. We make it an error explicitly
6525 * here. */
6526 if (strlen(name) >= _MAX_PATH)
6527 return NULL;
6528
Bram Moolenaar071d4272004-06-13 20:20:40 +00006529 return fopen(name, mode);
6530}
6531#endif
6532
6533#ifdef FEAT_MBYTE
6534/*
6535 * SUB STREAM (aka info stream) handling:
6536 *
6537 * NTFS can have sub streams for each file. Normal contents of file is
6538 * stored in the main stream, and extra contents (author information and
6539 * title and so on) can be stored in sub stream. After Windows 2000, user
6540 * can access and store those informations in sub streams via explorer's
6541 * property menuitem in right click menu. Those informations in sub streams
6542 * were lost when copying only the main stream. So we have to copy sub
6543 * streams.
6544 *
6545 * Incomplete explanation:
6546 * http://msdn.microsoft.com/library/en-us/dnw2k/html/ntfs5.asp
6547 * More useful info and an example:
6548 * http://www.sysinternals.com/ntw2k/source/misc.shtml#streams
6549 */
6550
6551/*
6552 * Copy info stream data "substream". Read from the file with BackupRead(sh)
6553 * and write to stream "substream" of file "to".
6554 * Errors are ignored.
6555 */
6556 static void
6557copy_substream(HANDLE sh, void *context, WCHAR *to, WCHAR *substream, long len)
6558{
6559 HANDLE hTo;
6560 WCHAR *to_name;
6561
6562 to_name = malloc((wcslen(to) + wcslen(substream) + 1) * sizeof(WCHAR));
6563 wcscpy(to_name, to);
6564 wcscat(to_name, substream);
6565
6566 hTo = CreateFileW(to_name, GENERIC_WRITE, 0, NULL, OPEN_ALWAYS,
6567 FILE_ATTRIBUTE_NORMAL, NULL);
6568 if (hTo != INVALID_HANDLE_VALUE)
6569 {
6570 long done;
6571 DWORD todo;
6572 DWORD readcnt, written;
6573 char buf[4096];
6574
6575 /* Copy block of bytes at a time. Abort when something goes wrong. */
6576 for (done = 0; done < len; done += written)
6577 {
6578 /* (size_t) cast for Borland C 5.5 */
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00006579 todo = (DWORD)((size_t)(len - done) > sizeof(buf) ? sizeof(buf)
6580 : (size_t)(len - done));
Bram Moolenaar071d4272004-06-13 20:20:40 +00006581 if (!BackupRead(sh, (LPBYTE)buf, todo, &readcnt,
6582 FALSE, FALSE, context)
6583 || readcnt != todo
6584 || !WriteFile(hTo, buf, todo, &written, NULL)
6585 || written != todo)
6586 break;
6587 }
6588 CloseHandle(hTo);
6589 }
6590
6591 free(to_name);
6592}
6593
6594/*
6595 * Copy info streams from file "from" to file "to".
6596 */
6597 static void
6598copy_infostreams(char_u *from, char_u *to)
6599{
6600 WCHAR *fromw;
6601 WCHAR *tow;
6602 HANDLE sh;
6603 WIN32_STREAM_ID sid;
6604 int headersize;
6605 WCHAR streamname[_MAX_PATH];
6606 DWORD readcount;
6607 void *context = NULL;
6608 DWORD lo, hi;
6609 int len;
6610
6611 /* Convert the file names to wide characters. */
Bram Moolenaar36f692d2008-11-20 16:10:17 +00006612 fromw = enc_to_utf16(from, NULL);
6613 tow = enc_to_utf16(to, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006614 if (fromw != NULL && tow != NULL)
6615 {
6616 /* Open the file for reading. */
6617 sh = CreateFileW(fromw, GENERIC_READ, FILE_SHARE_READ, NULL,
6618 OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
6619 if (sh != INVALID_HANDLE_VALUE)
6620 {
6621 /* Use BackupRead() to find the info streams. Repeat until we
6622 * have done them all.*/
6623 for (;;)
6624 {
6625 /* Get the header to find the length of the stream name. If
6626 * the "readcount" is zero we have done all info streams. */
6627 ZeroMemory(&sid, sizeof(WIN32_STREAM_ID));
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00006628 headersize = (int)((char *)&sid.cStreamName - (char *)&sid.dwStreamId);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006629 if (!BackupRead(sh, (LPBYTE)&sid, headersize,
6630 &readcount, FALSE, FALSE, &context)
6631 || readcount == 0)
6632 break;
6633
6634 /* We only deal with streams that have a name. The normal
6635 * file data appears to be without a name, even though docs
6636 * suggest it is called "::$DATA". */
6637 if (sid.dwStreamNameSize > 0)
6638 {
6639 /* Read the stream name. */
6640 if (!BackupRead(sh, (LPBYTE)streamname,
6641 sid.dwStreamNameSize,
6642 &readcount, FALSE, FALSE, &context))
6643 break;
6644
6645 /* Copy an info stream with a name ":anything:$DATA".
6646 * Skip "::$DATA", it has no stream name (examples suggest
6647 * it might be used for the normal file contents).
6648 * Note that BackupRead() counts bytes, but the name is in
6649 * wide characters. */
6650 len = readcount / sizeof(WCHAR);
6651 streamname[len] = 0;
6652 if (len > 7 && wcsicmp(streamname + len - 6,
6653 L":$DATA") == 0)
6654 {
6655 streamname[len - 6] = 0;
6656 copy_substream(sh, &context, tow, streamname,
Bram Moolenaarf4b8e572004-06-24 15:53:16 +00006657 (long)sid.Size.u.LowPart);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006658 }
6659 }
6660
6661 /* Advance to the next stream. We might try seeking too far,
6662 * but BackupSeek() doesn't skip over stream borders, thus
6663 * that's OK. */
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00006664 (void)BackupSeek(sh, sid.Size.u.LowPart, sid.Size.u.HighPart,
Bram Moolenaar071d4272004-06-13 20:20:40 +00006665 &lo, &hi, &context);
6666 }
6667
6668 /* Clear the context. */
6669 (void)BackupRead(sh, NULL, 0, &readcount, TRUE, FALSE, &context);
6670
6671 CloseHandle(sh);
6672 }
6673 }
6674 vim_free(fromw);
6675 vim_free(tow);
6676}
6677#endif
6678
6679/*
6680 * Copy file attributes from file "from" to file "to".
6681 * For Windows NT and later we copy info streams.
6682 * Always returns zero, errors are ignored.
6683 */
6684 int
6685mch_copy_file_attribute(char_u *from, char_u *to)
6686{
6687#ifdef FEAT_MBYTE
6688 /* File streams only work on Windows NT and later. */
6689 PlatformId();
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006690 copy_infostreams(from, to);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006691#endif
6692 return 0;
6693}
6694
6695#if defined(MYRESETSTKOFLW) || defined(PROTO)
6696/*
6697 * Recreate a destroyed stack guard page in win32.
6698 * Written by Benjamin Peterson.
6699 */
6700
6701/* These magic numbers are from the MS header files */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006702#define MIN_STACK_WINNT 2
6703
6704/*
6705 * This function does the same thing as _resetstkoflw(), which is only
6706 * available in DevStudio .net and later.
6707 * Returns 0 for failure, 1 for success.
6708 */
6709 int
6710myresetstkoflw(void)
6711{
6712 BYTE *pStackPtr;
6713 BYTE *pGuardPage;
6714 BYTE *pStackBase;
6715 BYTE *pLowestPossiblePage;
6716 MEMORY_BASIC_INFORMATION mbi;
6717 SYSTEM_INFO si;
6718 DWORD nPageSize;
6719 DWORD dummy;
6720
Bram Moolenaar071d4272004-06-13 20:20:40 +00006721 PlatformId();
Bram Moolenaar071d4272004-06-13 20:20:40 +00006722
6723 /* We need to know the system page size. */
6724 GetSystemInfo(&si);
6725 nPageSize = si.dwPageSize;
6726
6727 /* ...and the current stack pointer */
6728 pStackPtr = (BYTE*)_alloca(1);
6729
6730 /* ...and the base of the stack. */
6731 if (VirtualQuery(pStackPtr, &mbi, sizeof mbi) == 0)
6732 return 0;
6733 pStackBase = (BYTE*)mbi.AllocationBase;
6734
6735 /* ...and the page thats min_stack_req pages away from stack base; this is
6736 * the lowest page we could use. */
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006737 pLowestPossiblePage = pStackBase + MIN_STACK_WINNT * nPageSize;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006738
Bram Moolenaar071d4272004-06-13 20:20:40 +00006739 {
Bram Moolenaarcea912a2016-10-12 14:20:24 +02006740 /* We want the first committed page in the stack Start at the stack
6741 * base and move forward through memory until we find a committed block.
6742 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006743 BYTE *pBlock = pStackBase;
6744
Bram Moolenaara466c992005-07-09 21:03:22 +00006745 for (;;)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006746 {
6747 if (VirtualQuery(pBlock, &mbi, sizeof mbi) == 0)
6748 return 0;
6749
6750 pBlock += mbi.RegionSize;
6751
6752 if (mbi.State & MEM_COMMIT)
6753 break;
6754 }
6755
6756 /* mbi now describes the first committed block in the stack. */
6757 if (mbi.Protect & PAGE_GUARD)
6758 return 1;
6759
6760 /* decide where the guard page should start */
6761 if ((long_u)(mbi.BaseAddress) < (long_u)pLowestPossiblePage)
6762 pGuardPage = pLowestPossiblePage;
6763 else
6764 pGuardPage = (BYTE*)mbi.BaseAddress;
6765
6766 /* allocate the guard page */
6767 if (!VirtualAlloc(pGuardPage, nPageSize, MEM_COMMIT, PAGE_READWRITE))
6768 return 0;
6769
6770 /* apply the guard attribute to the page */
6771 if (!VirtualProtect(pGuardPage, nPageSize, PAGE_READWRITE | PAGE_GUARD,
6772 &dummy))
6773 return 0;
6774 }
6775
6776 return 1;
6777}
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006778#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006779
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006780
6781#if defined(FEAT_MBYTE) || defined(PROTO)
6782/*
6783 * The command line arguments in UCS2
6784 */
Bram Moolenaard857f0e2005-06-21 22:37:39 +00006785static int nArgsW = 0;
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006786static LPWSTR *ArglistW = NULL;
6787static int global_argc = 0;
6788static char **global_argv;
6789
6790static int used_file_argc = 0; /* last argument in global_argv[] used
6791 for the argument list. */
6792static int *used_file_indexes = NULL; /* indexes in global_argv[] for
6793 command line arguments added to
6794 the argument list */
6795static int used_file_count = 0; /* nr of entries in used_file_indexes */
6796static int used_file_literal = FALSE; /* take file names literally */
6797static int used_file_full_path = FALSE; /* file name was full path */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00006798static int used_file_diff_mode = FALSE; /* file name was with diff mode */
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006799static int used_alist_count = 0;
6800
6801
6802/*
6803 * Get the command line arguments. Unicode version.
6804 * Returns argc. Zero when something fails.
6805 */
6806 int
6807get_cmd_argsW(char ***argvp)
6808{
6809 char **argv = NULL;
6810 int argc = 0;
6811 int i;
6812
Bram Moolenaar14993322014-09-09 12:25:33 +02006813 free_cmd_argsW();
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006814 ArglistW = CommandLineToArgvW(GetCommandLineW(), &nArgsW);
6815 if (ArglistW != NULL)
6816 {
6817 argv = malloc((nArgsW + 1) * sizeof(char *));
6818 if (argv != NULL)
6819 {
6820 argc = nArgsW;
6821 argv[argc] = NULL;
6822 for (i = 0; i < argc; ++i)
6823 {
6824 int len;
6825
6826 /* Convert each Unicode argument to the current codepage. */
6827 WideCharToMultiByte_alloc(GetACP(), 0,
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00006828 ArglistW[i], (int)wcslen(ArglistW[i]) + 1,
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006829 (LPSTR *)&argv[i], &len, 0, 0);
6830 if (argv[i] == NULL)
6831 {
6832 /* Out of memory, clear everything. */
6833 while (i > 0)
6834 free(argv[--i]);
6835 free(argv);
Bram Moolenaar73c61632013-12-07 14:48:10 +01006836 argv = NULL;
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006837 argc = 0;
6838 }
6839 }
6840 }
6841 }
6842
6843 global_argc = argc;
6844 global_argv = argv;
6845 if (argc > 0)
Bram Moolenaar14993322014-09-09 12:25:33 +02006846 {
6847 if (used_file_indexes != NULL)
6848 free(used_file_indexes);
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006849 used_file_indexes = malloc(argc * sizeof(int));
Bram Moolenaar14993322014-09-09 12:25:33 +02006850 }
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006851
6852 if (argvp != NULL)
6853 *argvp = argv;
6854 return argc;
6855}
6856
6857 void
6858free_cmd_argsW(void)
6859{
6860 if (ArglistW != NULL)
6861 {
6862 GlobalFree(ArglistW);
6863 ArglistW = NULL;
6864 }
6865}
6866
6867/*
6868 * Remember "name" is an argument that was added to the argument list.
6869 * This avoids that we have to re-parse the argument list when fix_arg_enc()
6870 * is called.
6871 */
6872 void
Bram Moolenaar910f66f2006-04-05 20:41:53 +00006873used_file_arg(char *name, int literal, int full_path, int diff_mode)
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006874{
6875 int i;
6876
6877 if (used_file_indexes == NULL)
6878 return;
6879 for (i = used_file_argc + 1; i < global_argc; ++i)
6880 if (STRCMP(global_argv[i], name) == 0)
6881 {
6882 used_file_argc = i;
6883 used_file_indexes[used_file_count++] = i;
6884 break;
6885 }
6886 used_file_literal = literal;
6887 used_file_full_path = full_path;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00006888 used_file_diff_mode = diff_mode;
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006889}
6890
6891/*
6892 * Remember the length of the argument list as it was. If it changes then we
6893 * leave it alone when 'encoding' is set.
6894 */
6895 void
6896set_alist_count(void)
6897{
6898 used_alist_count = GARGCOUNT;
6899}
6900
6901/*
6902 * Fix the encoding of the command line arguments. Invoked when 'encoding'
6903 * has been changed while starting up. Use the UCS-2 command line arguments
6904 * and convert them to 'encoding'.
6905 */
6906 void
6907fix_arg_enc(void)
6908{
6909 int i;
6910 int idx;
6911 char_u *str;
Bram Moolenaar86b68352004-12-27 21:59:20 +00006912 int *fnum_list;
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006913
6914 /* Safety checks:
6915 * - if argument count differs between the wide and non-wide argument
6916 * list, something must be wrong.
6917 * - the file name arguments must have been located.
6918 * - the length of the argument list wasn't changed by the user.
6919 */
Bram Moolenaard857f0e2005-06-21 22:37:39 +00006920 if (global_argc != nArgsW
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006921 || ArglistW == NULL
6922 || used_file_indexes == NULL
6923 || used_file_count == 0
6924 || used_alist_count != GARGCOUNT)
6925 return;
6926
Bram Moolenaar86b68352004-12-27 21:59:20 +00006927 /* Remember the buffer numbers for the arguments. */
6928 fnum_list = (int *)alloc((int)sizeof(int) * GARGCOUNT);
6929 if (fnum_list == NULL)
6930 return; /* out of memory */
6931 for (i = 0; i < GARGCOUNT; ++i)
6932 fnum_list[i] = GARGLIST[i].ae_fnum;
6933
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006934 /* Clear the argument list. Make room for the new arguments. */
6935 alist_clear(&global_alist);
6936 if (ga_grow(&global_alist.al_ga, used_file_count) == FAIL)
Bram Moolenaar86b68352004-12-27 21:59:20 +00006937 return; /* out of memory */
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006938
6939 for (i = 0; i < used_file_count; ++i)
6940 {
6941 idx = used_file_indexes[i];
Bram Moolenaar36f692d2008-11-20 16:10:17 +00006942 str = utf16_to_enc(ArglistW[idx], NULL);
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006943 if (str != NULL)
Bram Moolenaar86b68352004-12-27 21:59:20 +00006944 {
Bram Moolenaar910f66f2006-04-05 20:41:53 +00006945#ifdef FEAT_DIFF
6946 /* When using diff mode may need to concatenate file name to
6947 * directory name. Just like it's done in main(). */
6948 if (used_file_diff_mode && mch_isdir(str) && GARGCOUNT > 0
6949 && !mch_isdir(alist_name(&GARGLIST[0])))
6950 {
6951 char_u *r;
6952
6953 r = concat_fnames(str, gettail(alist_name(&GARGLIST[0])), TRUE);
6954 if (r != NULL)
6955 {
6956 vim_free(str);
6957 str = r;
6958 }
6959 }
6960#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00006961 /* Re-use the old buffer by renaming it. When not using literal
6962 * names it's done by alist_expand() below. */
6963 if (used_file_literal)
6964 buf_set_name(fnum_list[i], str);
6965
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006966 alist_add(&global_alist, str, used_file_literal ? 2 : 0);
Bram Moolenaar86b68352004-12-27 21:59:20 +00006967 }
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006968 }
6969
6970 if (!used_file_literal)
6971 {
6972 /* Now expand wildcards in the arguments. */
6973 /* Temporarily add '(' and ')' to 'isfname'. These are valid
6974 * filename characters but are excluded from 'isfname' to make
6975 * "gf" work on a file name in parenthesis (e.g.: see vim.h). */
6976 do_cmdline_cmd((char_u *)":let SaVe_ISF = &isf|set isf+=(,)");
Bram Moolenaar86b68352004-12-27 21:59:20 +00006977 alist_expand(fnum_list, used_alist_count);
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006978 do_cmdline_cmd((char_u *)":let &isf = SaVe_ISF|unlet SaVe_ISF");
6979 }
6980
6981 /* If wildcard expansion failed, we are editing the first file of the
6982 * arglist and there is no file name: Edit the first argument now. */
6983 if (curwin->w_arg_idx == 0 && curbuf->b_fname == NULL)
6984 {
6985 do_cmdline_cmd((char_u *)":rewind");
6986 if (GARGCOUNT == 1 && used_file_full_path)
6987 (void)vim_chdirfile(alist_name(&GARGLIST[0]));
6988 }
Bram Moolenaar86b68352004-12-27 21:59:20 +00006989
6990 set_alist_count();
Bram Moolenaar3fdfa4a2004-10-07 21:02:47 +00006991}
Bram Moolenaar071d4272004-06-13 20:20:40 +00006992#endif
Bram Moolenaar7c23d1d2017-02-01 13:14:16 +01006993
6994 int
6995mch_setenv(char *var, char *value, int x)
6996{
6997 char_u *envbuf;
6998
6999 envbuf = alloc((unsigned)(STRLEN(var) + STRLEN(value) + 2));
7000 if (envbuf == NULL)
7001 return -1;
7002
7003 sprintf((char *)envbuf, "%s=%s", var, value);
7004
7005#ifdef FEAT_MBYTE
7006 if (enc_codepage >= 0 && (int)GetACP() != enc_codepage)
7007 {
7008 WCHAR *p = enc_to_utf16(envbuf, NULL);
7009
7010 vim_free(envbuf);
7011 if (p == NULL)
7012 return -1;
7013 _wputenv(p);
7014# ifdef libintl_wputenv
7015 libintl_wputenv(p);
7016# endif
7017 /* Unlike Un*x systems, we can free the string for _wputenv(). */
7018 vim_free(p);
7019 }
7020 else
7021#endif
7022 {
7023 _putenv((char *)envbuf);
7024# ifdef libintl_putenv
7025 libintl_putenv((char *)envbuf);
7026# endif
7027 /* Unlike Un*x systems, we can free the string for _putenv(). */
7028 vim_free(envbuf);
7029 }
7030
7031 return 0;
7032}