blob: e9068fee5e0b322c166cbc1036b2e49b3cd39295 [file] [log] [blame]
Bram Moolenaare0874f82016-01-24 20:36:41 +01001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 */
8
9/*
10 * Implements communication through a socket or any file handle.
11 */
12
13#include "vim.h"
14
15#if defined(FEAT_CHANNEL) || defined(PROTO)
16
Bram Moolenaard04a0202016-01-26 23:30:18 +010017/* TRUE when netbeans is running with a GUI. */
18#ifdef FEAT_GUI
19# define CH_HAS_GUI (gui.in_use || gui.starting)
20#endif
21
22/* Note: when making changes here also adjust configure.in. */
23#ifdef WIN32
24/* WinSock API is separated from C API, thus we can't use read(), write(),
25 * errno... */
26# define SOCK_ERRNO errno = WSAGetLastError()
27# undef ECONNREFUSED
28# define ECONNREFUSED WSAECONNREFUSED
Bram Moolenaar4d919d72016-02-05 22:36:41 +010029# undef EWOULDBLOCK
30# define EWOULDBLOCK WSAEWOULDBLOCK
Bram Moolenaard04a0202016-01-26 23:30:18 +010031# ifdef EINTR
32# undef EINTR
33# endif
34# define EINTR WSAEINTR
Bram Moolenaard8070362016-02-15 21:56:54 +010035# define sock_write(sd, buf, len) send((SOCKET)sd, buf, len, 0)
36# define sock_read(sd, buf, len) recv((SOCKET)sd, buf, len, 0)
37# define sock_close(sd) closesocket((SOCKET)sd)
Bram Moolenaard04a0202016-01-26 23:30:18 +010038#else
39# include <netdb.h>
40# include <netinet/in.h>
41
42# include <sys/socket.h>
43# ifdef HAVE_LIBGEN_H
44# include <libgen.h>
45# endif
46# define SOCK_ERRNO
47# define sock_write(sd, buf, len) write(sd, buf, len)
48# define sock_read(sd, buf, len) read(sd, buf, len)
49# define sock_close(sd) close(sd)
Bram Moolenaar0943a092016-02-16 13:11:17 +010050# define fd_read(fd, buf, len) read(fd, buf, len)
Bram Moolenaard8070362016-02-15 21:56:54 +010051# define fd_write(sd, buf, len) write(sd, buf, len)
52# define fd_close(sd) close(sd)
Bram Moolenaard04a0202016-01-26 23:30:18 +010053#endif
54
55#ifdef FEAT_GUI_W32
56extern HWND s_hwnd; /* Gvim's Window handle */
57#endif
58
Bram Moolenaard8070362016-02-15 21:56:54 +010059#ifdef WIN32
60 static int
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +010061fd_read(sock_T fd, char *buf, size_t len)
Bram Moolenaard8070362016-02-15 21:56:54 +010062{
63 HANDLE h = (HANDLE)fd;
64 DWORD nread;
65
66 if (!ReadFile(h, buf, (DWORD)len, &nread, NULL))
67 return -1;
68 return (int)nread;
69}
70
71 static int
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +010072fd_write(sock_T fd, char *buf, size_t len)
Bram Moolenaard8070362016-02-15 21:56:54 +010073{
74 HANDLE h = (HANDLE)fd;
75 DWORD nwrite;
76
77 if (!WriteFile(h, buf, (DWORD)len, &nwrite, NULL))
78 return -1;
79 return (int)nwrite;
80}
81
82 static void
83fd_close(sock_T fd)
84{
85 HANDLE h = (HANDLE)fd;
86
87 CloseHandle(h);
88}
89#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +010090
Bram Moolenaar6463ca22016-02-13 17:04:46 +010091/* Log file opened with ch_logfile(). */
92static FILE *log_fd = NULL;
Bram Moolenaar81661fb2016-02-18 22:23:34 +010093#ifdef FEAT_RELTIME
94static proftime_T log_start;
95#endif
Bram Moolenaar6463ca22016-02-13 17:04:46 +010096
97 void
98ch_logfile(FILE *file)
99{
100 if (log_fd != NULL)
101 fclose(log_fd);
102 log_fd = file;
103 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100104 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100105 fprintf(log_fd, "==== start log session ====\n");
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100106#ifdef FEAT_RELTIME
107 profile_start(&log_start);
108#endif
109 }
110}
111
112 int
113ch_log_active()
114{
115 return log_fd != NULL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100116}
117
118 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100119ch_log_lead(char *what, channel_T *ch)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100120{
121 if (log_fd != NULL)
122 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100123#ifdef FEAT_RELTIME
124 proftime_T log_now;
125
126 profile_start(&log_now);
127 profile_sub(&log_now, &log_start);
128 fprintf(log_fd, "%s ", profile_msg(&log_now));
129#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100130 if (ch != NULL)
131 fprintf(log_fd, "%son %d: ", what, ch->ch_id);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100132 else
133 fprintf(log_fd, "%s: ", what);
134 }
135}
136
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100137 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100138ch_log(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100139{
140 if (log_fd != NULL)
141 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100142 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100143 fputs(msg, log_fd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100144 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100145 fflush(log_fd);
146 }
147}
148
149 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100150ch_logn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100151{
152 if (log_fd != NULL)
153 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100154 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100155 fprintf(log_fd, msg, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100156 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100157 fflush(log_fd);
158 }
159}
160
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100161 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100162ch_logs(channel_T *ch, char *msg, char *name)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100163{
164 if (log_fd != NULL)
165 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100166 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100167 fprintf(log_fd, msg, name);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100168 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100169 fflush(log_fd);
170 }
171}
172
173 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100174ch_logsn(channel_T *ch, char *msg, char *name, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100175{
176 if (log_fd != NULL)
177 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100178 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100179 fprintf(log_fd, msg, name, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100180 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100181 fflush(log_fd);
182 }
183}
184
185 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100186ch_error(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100187{
188 if (log_fd != NULL)
189 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100190 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100191 fputs(msg, log_fd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100192 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100193 fflush(log_fd);
194 }
195}
196
197 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100198ch_errorn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100199{
200 if (log_fd != NULL)
201 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100202 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100203 fprintf(log_fd, msg, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100204 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100205 fflush(log_fd);
206 }
207}
208
209 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100210ch_errors(channel_T *ch, char *msg, char *arg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100211{
212 if (log_fd != NULL)
213 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100214 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100215 fprintf(log_fd, msg, arg);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100216 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100217 fflush(log_fd);
218 }
219}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100220
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100221#ifdef _WIN32
222# undef PERROR
223# define PERROR(msg) (void)emsg3((char_u *)"%s: %s", \
224 (char_u *)msg, (char_u *)strerror_win32(errno))
225
226 static char *
227strerror_win32(int eno)
228{
229 static LPVOID msgbuf = NULL;
230 char_u *ptr;
231
232 if (msgbuf)
233 LocalFree(msgbuf);
234 FormatMessage(
235 FORMAT_MESSAGE_ALLOCATE_BUFFER |
236 FORMAT_MESSAGE_FROM_SYSTEM |
237 FORMAT_MESSAGE_IGNORE_INSERTS,
238 NULL,
239 eno,
240 MAKELANGID(LANG_ENGLISH, SUBLANG_DEFAULT),
241 (LPTSTR) &msgbuf,
242 0,
243 NULL);
244 /* chomp \r or \n */
245 for (ptr = (char_u *)msgbuf; *ptr; ptr++)
246 switch (*ptr)
247 {
248 case '\r':
249 STRMOVE(ptr, ptr + 1);
250 ptr--;
251 break;
252 case '\n':
253 if (*(ptr + 1) == '\0')
254 *ptr = '\0';
255 else
256 *ptr = ' ';
257 break;
258 }
259 return msgbuf;
260}
261#endif
262
Bram Moolenaar77073442016-02-13 23:23:53 +0100263/*
264 * The list of all allocated channels.
265 */
266static channel_T *first_channel = NULL;
267static int next_ch_id = 0;
268
269/*
270 * Allocate a new channel. The refcount is set to 1.
271 * The channel isn't actually used until it is opened.
272 * Returns NULL if out of memory.
273 */
274 channel_T *
275add_channel(void)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100276{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100277 int part;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100278 channel_T *channel = (channel_T *)alloc_clear((int)sizeof(channel_T));
Bram Moolenaare0874f82016-01-24 20:36:41 +0100279
Bram Moolenaar77073442016-02-13 23:23:53 +0100280 if (channel == NULL)
281 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100282
Bram Moolenaar77073442016-02-13 23:23:53 +0100283 channel->ch_id = next_ch_id++;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100284 ch_log(channel, "Created channel");
Bram Moolenaar77073442016-02-13 23:23:53 +0100285
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100286#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100287 for (part = PART_SOCK; part <= PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100288#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100289 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100290#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100291 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100292 channel->ch_part[part].ch_fd = INVALID_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100293#ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100294 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100295#endif
296#ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100297 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100298#endif
299#ifdef FEAT_GUI_W32
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100300 channel->ch_part[part].ch_inputHandler = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100301#endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100302 channel->ch_part[part].ch_timeout = 2000;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100303 }
Bram Moolenaare0874f82016-01-24 20:36:41 +0100304
Bram Moolenaar77073442016-02-13 23:23:53 +0100305 if (first_channel != NULL)
306 {
307 first_channel->ch_prev = channel;
308 channel->ch_next = first_channel;
309 }
310 first_channel = channel;
311
312 channel->ch_refcount = 1;
313 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100314}
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100315
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100316/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100317 * Close a channel and free all its resources.
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100318 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100319 void
320channel_free(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100321{
Bram Moolenaar77073442016-02-13 23:23:53 +0100322 channel_close(channel);
323 if (channel->ch_next != NULL)
324 channel->ch_next->ch_prev = channel->ch_prev;
325 if (channel->ch_prev == NULL)
326 first_channel = channel->ch_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100327 else
Bram Moolenaar77073442016-02-13 23:23:53 +0100328 channel->ch_prev->ch_next = channel->ch_next;
329 vim_free(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100330}
331
Bram Moolenaard04a0202016-01-26 23:30:18 +0100332#if defined(FEAT_GUI) || defined(PROTO)
Bram Moolenaar77073442016-02-13 23:23:53 +0100333
334#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
335 static void
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100336channel_read_fd(int fd)
Bram Moolenaar77073442016-02-13 23:23:53 +0100337{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100338 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100339 int part;
Bram Moolenaar77073442016-02-13 23:23:53 +0100340
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100341 channel = channel_fd2channel(fd, &part);
Bram Moolenaar77073442016-02-13 23:23:53 +0100342 if (channel == NULL)
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100343 ch_errorn(NULL, "Channel for fd %d not found", fd);
Bram Moolenaar77073442016-02-13 23:23:53 +0100344 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100345 channel_read(channel, part, "messageFromNetbeans");
Bram Moolenaar77073442016-02-13 23:23:53 +0100346}
347#endif
348
Bram Moolenaare0874f82016-01-24 20:36:41 +0100349/*
Bram Moolenaard04a0202016-01-26 23:30:18 +0100350 * Read a command from netbeans.
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100351 * TODO: instead of channel ID use the FD.
Bram Moolenaare0874f82016-01-24 20:36:41 +0100352 */
Bram Moolenaard04a0202016-01-26 23:30:18 +0100353#ifdef FEAT_GUI_X11
354 static void
355messageFromNetbeans(XtPointer clientData,
356 int *unused1 UNUSED,
357 XtInputId *unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100358{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100359 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100360}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100361#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100362
Bram Moolenaard04a0202016-01-26 23:30:18 +0100363#ifdef FEAT_GUI_GTK
364 static void
365messageFromNetbeans(gpointer clientData,
366 gint unused1 UNUSED,
367 GdkInputCondition unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100368{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100369 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100370}
371#endif
372
373 static void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100374channel_gui_register_one(channel_T *channel, int part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100375{
376# ifdef FEAT_GUI_X11
377 /* Tell notifier we are interested in being called
378 * when there is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100379 if (channel->ch_part[part].ch_inputHandler == (XtInputId)NULL)
380 channel->ch_part[part].ch_inputHandler = XtAppAddInput(
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100381 (XtAppContext)app_context,
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100382 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100383 (XtPointer)(XtInputReadMask + XtInputExceptMask),
384 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100385 (XtPointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100386# else
387# ifdef FEAT_GUI_GTK
388 /* Tell gdk we are interested in being called when there
389 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100390 if (channel->ch_part[part].ch_inputHandler == 0)
391 channel->ch_part[part].ch_inputHandler = gdk_input_add(
392 (gint)channel->ch_part[part].ch_fd,
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100393 (GdkInputCondition)
394 ((int)GDK_INPUT_READ + (int)GDK_INPUT_EXCEPTION),
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100395 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100396 (gpointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100397# else
398# ifdef FEAT_GUI_W32
399 /* Tell Windows we are interested in receiving message when there
400 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100401 if (channel->ch_part[part].ch_inputHandler == -1)
402 channel->ch_part[part].ch_inputHandler = WSAAsyncSelect(
403 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100404 s_hwnd, WM_NETBEANS, FD_READ);
405# endif
406# endif
407# endif
408}
409
410 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100411channel_gui_register(channel_T *channel)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100412{
Bram Moolenaard04a0202016-01-26 23:30:18 +0100413 if (!CH_HAS_GUI)
414 return;
415
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100416 if (channel->CH_SOCK_FD != INVALID_FD)
417 channel_gui_register_one(channel, PART_SOCK);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100418# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100419 if (channel->CH_OUT_FD != INVALID_FD)
420 channel_gui_register_one(channel, PART_OUT);
421 if (channel->CH_ERR_FD != INVALID_FD)
422 channel_gui_register_one(channel, PART_ERR);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100423# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100424}
425
426/*
427 * Register any of our file descriptors with the GUI event handling system.
428 * Called when the GUI has started.
429 */
430 void
431channel_gui_register_all(void)
432{
Bram Moolenaar77073442016-02-13 23:23:53 +0100433 channel_T *channel;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100434
Bram Moolenaar77073442016-02-13 23:23:53 +0100435 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100436 channel_gui_register(channel);
437}
438
439 static void
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100440channel_gui_unregister(channel_T *channel)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100441{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100442 int part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100443
444#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100445 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100446#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100447 part = PART_SOCK;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100448#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100449 {
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100450# ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100451 if (channel->ch_part[part].ch_inputHandler != (XtInputId)NULL)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100452 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100453 XtRemoveInput(channel->ch_part[part].ch_inputHandler);
454 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100455 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100456# else
457# ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100458 if (channel->ch_part[part].ch_inputHandler != 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100459 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100460 gdk_input_remove(channel->ch_part[part].ch_inputHandler);
461 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100462 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100463# else
464# ifdef FEAT_GUI_W32
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100465 if (channel->ch_part[part].ch_inputHandler == 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100466 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100467 WSAAsyncSelect(channel->ch_part[part].ch_fd, s_hwnd, 0, 0);
468 channel->ch_part[part].ch_inputHandler = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100469 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100470# endif
471# endif
472# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100473 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100474}
475
476#endif
477
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100478static char *e_cannot_connect = N_("E902: Cannot connect to port");
479
Bram Moolenaard04a0202016-01-26 23:30:18 +0100480/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100481 * Open a socket channel to "hostname":"port".
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100482 * "waittime" is the time in msec to wait for the connection.
483 * When negative wait forever.
Bram Moolenaar77073442016-02-13 23:23:53 +0100484 * Returns the channel for success.
485 * Returns NULL for failure.
Bram Moolenaard04a0202016-01-26 23:30:18 +0100486 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100487 channel_T *
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100488channel_open(
489 char *hostname,
490 int port_in,
491 int waittime,
492 void (*nb_close_cb)(void))
Bram Moolenaard04a0202016-01-26 23:30:18 +0100493{
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100494 int sd = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100495 struct sockaddr_in server;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100496 struct hostent *host;
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100497#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100498 u_short port = port_in;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100499 u_long val = 1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100500#else
501 int port = port_in;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100502 struct timeval start_tv;
503 int so_error;
504 socklen_t so_error_len = sizeof(so_error);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100505#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100506 channel_T *channel;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100507 int ret;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100508
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100509#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100510 channel_init_winsock();
511#endif
512
Bram Moolenaar77073442016-02-13 23:23:53 +0100513 channel = add_channel();
514 if (channel == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100515 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100516 ch_error(NULL, "Cannot allocate channel.");
Bram Moolenaar77073442016-02-13 23:23:53 +0100517 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100518 }
519
520 /* Get the server internet address and put into addr structure */
521 /* fill in the socket address structure and connect to server */
522 vim_memset((char *)&server, 0, sizeof(server));
523 server.sin_family = AF_INET;
524 server.sin_port = htons(port);
525 if ((host = gethostbyname(hostname)) == NULL)
526 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100527 ch_error(channel, "in gethostbyname() in channel_open()");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100528 PERROR("E901: gethostbyname() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100529 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100530 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100531 }
532 memcpy((char *)&server.sin_addr, host->h_addr, host->h_length);
533
Bram Moolenaar254e00d2016-02-19 23:23:12 +0100534 /* On Mac and Solaris a zero timeout almost never works. At least wait
535 * one millisecond. Let's do it for all systems, because we don't know why
536 * this is needed. */
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100537 if (waittime == 0)
538 waittime = 1;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100539
540 /*
541 * For Unix we need to call connect() again after connect() failed.
542 * On Win32 one time is sufficient.
543 */
544 while (TRUE)
545 {
546 if (sd >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100547 sock_close(sd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100548 sd = socket(AF_INET, SOCK_STREAM, 0);
549 if (sd == -1)
550 {
551 ch_error(channel, "in socket() in channel_open().");
552 PERROR("E898: socket() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100553 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100554 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100555 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100556
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100557 if (waittime >= 0)
558 {
559 /* Make connect() non-blocking. */
560 if (
561#ifdef _WIN32
562 ioctlsocket(sd, FIONBIO, &val) < 0
563#else
564 fcntl(sd, F_SETFL, O_NONBLOCK) < 0
565#endif
566 )
567 {
568 SOCK_ERRNO;
569 ch_errorn(channel,
570 "channel_open: Connect failed with errno %d", errno);
571 sock_close(sd);
572 channel_free(channel);
573 return NULL;
574 }
575 }
576
577 /* Try connecting to the server. */
578 ch_logsn(channel, "Connecting to %s port %d", hostname, port);
579 ret = connect(sd, (struct sockaddr *)&server, sizeof(server));
580
581 SOCK_ERRNO;
582 if (ret < 0)
583 {
584 if (errno != EWOULDBLOCK
585 && errno != ECONNREFUSED
586
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100587#ifdef EINPROGRESS
588 && errno != EINPROGRESS
589#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100590 )
591 {
592 ch_errorn(channel,
593 "channel_open: Connect failed with errno %d", errno);
594 PERROR(_(e_cannot_connect));
595 sock_close(sd);
596 channel_free(channel);
597 return NULL;
598 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100599 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100600
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100601 /* If we don't block and connect() failed then try using select() to
602 * wait for the connection to be made. */
603 if (waittime >= 0 && ret < 0)
604 {
605 struct timeval tv;
606 fd_set wfds;
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100607#if defined(__APPLE__) && __APPLE__ == 1
608# define PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100609 fd_set rfds;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100610
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100611 FD_ZERO(&rfds);
612 FD_SET(sd, &rfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100613#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100614 FD_ZERO(&wfds);
615 FD_SET(sd, &wfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100616
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100617 tv.tv_sec = waittime / 1000;
618 tv.tv_usec = (waittime % 1000) * 1000;
619#ifndef WIN32
620 gettimeofday(&start_tv, NULL);
621#endif
622 ch_logn(channel,
623 "Waiting for connection (waittime %d msec)...", waittime);
624 ret = select((int)sd + 1,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100625#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100626 &rfds,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100627#else
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100628 NULL,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100629#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100630 &wfds, NULL, &tv);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100631
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100632 if (ret < 0)
633 {
634 SOCK_ERRNO;
635 ch_errorn(channel,
636 "channel_open: Connect failed with errno %d", errno);
637 PERROR(_(e_cannot_connect));
638 sock_close(sd);
639 channel_free(channel);
640 return NULL;
641 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100642#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100643 if (ret == 0 && FD_ISSET(sd, &rfds) && FD_ISSET(sd, &wfds))
644 {
645 /* For OS X, this implies error. See tcp(4). */
646 ch_error(channel, "channel_open: Connect failed");
647 EMSG(_(e_cannot_connect));
648 sock_close(sd);
649 channel_free(channel);
650 return NULL;
651 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100652#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100653#ifdef WIN32
654 /* On Win32 select() is expected to work and wait for up to the
655 * waittime for the socket to be open. */
656 if (!FD_ISSET(sd, &wfds) || ret == 0)
657#else
658 /* See socket(7) for the behavior on Linux-like systems:
659 * After putting the socket in non-blocking mode, connect() will
660 * return EINPROGRESS, select() will not wait (as if writing is
661 * possible), need to use getsockopt() to check if the socket is
662 * actually open. */
663 getsockopt(sd, SOL_SOCKET, SO_ERROR, &so_error, &so_error_len);
664 if (!FD_ISSET(sd, &wfds) || ret == 0 || so_error != 0)
665#endif
666 {
667#ifndef WIN32
668 struct timeval end_tv;
669 long elapsed_msec;
670
671 gettimeofday(&end_tv, NULL);
672 elapsed_msec = (end_tv.tv_sec - start_tv.tv_sec) * 1000
673 + (end_tv.tv_usec - start_tv.tv_usec) / 1000;
674 if (waittime > 1 && elapsed_msec < waittime)
675 {
676 /* The port isn't ready but we also didn't get an error.
677 * This happens when the server didn't open the socket
678 * yet. Wait a bit and try again. */
679 mch_delay(waittime < 50 ? (long)waittime : 50L, TRUE);
680 ui_breakcheck();
681 if (!got_int)
682 {
683 /* reduce the waittime by the elapsed time and the 50
684 * msec delay (or a bit more) */
685 waittime -= elapsed_msec;
686 if (waittime > 50)
687 waittime -= 50;
688 else
689 waittime = 1;
690 continue;
691 }
692 /* we were interrupted, behave as if timed out */
693 }
694#endif
695 /* We timed out. */
696 ch_error(channel, "Connection timed out");
697 sock_close(sd);
698 channel_free(channel);
699 return NULL;
700 }
701
702 ch_log(channel, "Connection made");
703 break;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100704 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100705 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100706
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100707 if (waittime >= 0)
708 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100709#ifdef _WIN32
710 val = 0;
711 ioctlsocket(sd, FIONBIO, &val);
712#else
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100713 (void)fcntl(sd, F_SETFL, 0);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100714#endif
715 }
716
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100717 channel->CH_SOCK_FD = (sock_T)sd;
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100718 channel->ch_nb_close_cb = nb_close_cb;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100719
720#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +0100721 channel_gui_register(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100722#endif
723
Bram Moolenaar77073442016-02-13 23:23:53 +0100724 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100725}
726
727#if defined(CHANNEL_PIPES) || defined(PROTO)
728 void
Bram Moolenaard8070362016-02-15 21:56:54 +0100729channel_set_pipes(channel_T *channel, sock_T in, sock_T out, sock_T err)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100730{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100731 channel->CH_IN_FD = in;
732 channel->CH_OUT_FD = out;
733 channel->CH_ERR_FD = err;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100734}
735#endif
736
737 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100738channel_set_job(channel_T *channel, job_T *job)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100739{
Bram Moolenaar77073442016-02-13 23:23:53 +0100740 channel->ch_job = job;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100741}
742
743/*
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100744 * Set various properties from an "opt" argument.
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100745 */
746 void
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100747channel_set_options(channel_T *channel, jobopt_T *opt)
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100748{
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100749 int part;
750 char_u **cbp;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100751
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100752 if (opt->jo_set & JO_MODE)
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100753 for (part = PART_SOCK; part <= PART_IN; ++part)
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100754 channel->ch_part[part].ch_mode = opt->jo_mode;
755 if (opt->jo_set & JO_IN_MODE)
756 channel->ch_part[PART_IN].ch_mode = opt->jo_in_mode;
757 if (opt->jo_set & JO_OUT_MODE)
758 channel->ch_part[PART_OUT].ch_mode = opt->jo_out_mode;
759 if (opt->jo_set & JO_ERR_MODE)
760 channel->ch_part[PART_ERR].ch_mode = opt->jo_err_mode;
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100761
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100762 if (opt->jo_set & JO_TIMEOUT)
763 for (part = PART_SOCK; part <= PART_IN; ++part)
764 channel->ch_part[part].ch_timeout = opt->jo_timeout;
765 if (opt->jo_set & JO_OUT_TIMEOUT)
766 channel->ch_part[PART_OUT].ch_timeout = opt->jo_out_timeout;
767 if (opt->jo_set & JO_ERR_TIMEOUT)
768 channel->ch_part[PART_ERR].ch_timeout = opt->jo_err_timeout;
769
770 if (opt->jo_set & JO_CALLBACK)
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100771 {
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100772 cbp = &channel->ch_callback;
773 vim_free(*cbp);
774 if (opt->jo_callback != NULL && *opt->jo_callback != NUL)
775 *cbp = vim_strsave(opt->jo_callback);
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100776 else
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100777 *cbp = NULL;
778 }
779 if (opt->jo_set & JO_OUT_CALLBACK)
780 {
781 cbp = &channel->ch_part[PART_OUT].ch_callback;
782 vim_free(*cbp);
783 if (opt->jo_out_cb != NULL && *opt->jo_out_cb != NUL)
784 *cbp = vim_strsave(opt->jo_out_cb);
785 else
786 *cbp = NULL;
787 }
788 if (opt->jo_set & JO_ERR_CALLBACK)
789 {
790 cbp = &channel->ch_part[PART_ERR].ch_callback;
791 vim_free(*cbp);
792 if (opt->jo_err_cb != NULL && *opt->jo_err_cb != NUL)
793 *cbp = vim_strsave(opt->jo_err_cb);
794 else
795 *cbp = NULL;
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100796 }
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100797 if (opt->jo_set & JO_CLOSE_CALLBACK)
798 {
799 cbp = &channel->ch_close_cb;
800 vim_free(*cbp);
801 if (opt->jo_close_cb != NULL && *opt->jo_close_cb != NUL)
802 *cbp = vim_strsave(opt->jo_close_cb);
803 else
804 *cbp = NULL;
805 }
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100806}
807
808/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100809 * Set the callback for "channel"/"part" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100810 */
811 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100812channel_set_req_callback(
813 channel_T *channel,
814 int part,
815 char_u *callback,
816 int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100817{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100818 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100819 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
820
821 if (item != NULL)
822 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100823 item->cq_callback = vim_strsave(callback);
824 item->cq_seq_nr = id;
825 item->cq_prev = head->cq_prev;
826 head->cq_prev = item;
827 item->cq_next = NULL;
828 if (item->cq_prev == NULL)
829 head->cq_next = item;
830 else
831 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100832 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100833}
834
835/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100836 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100837 */
838 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100839invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100840{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100841 typval_T rettv;
842 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100843
Bram Moolenaar77073442016-02-13 23:23:53 +0100844 argv[0].v_type = VAR_CHANNEL;
845 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100846
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100847 call_func(callback, (int)STRLEN(callback),
848 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
Bram Moolenaaree1cffc2016-02-21 19:14:41 +0100849 clear_tv(&rettv);
850
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100851 /* If an echo command was used the cursor needs to be put back where
852 * it belongs. */
853 setcursor();
854 cursor_on();
855 out_flush();
856}
857
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100858/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100859 * Return the first buffer from channel "channel"/"part" and remove it.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100860 * The caller must free it.
861 * Returns NULL if there is nothing.
862 */
863 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100864channel_get(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100865{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100866 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100867 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100868 char_u *p;
869
Bram Moolenaar77073442016-02-13 23:23:53 +0100870 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100871 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100872 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100873 p = node->rq_buffer;
874 head->rq_next = node->rq_next;
875 if (node->rq_next == NULL)
876 head->rq_prev = NULL;
877 else
878 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100879 vim_free(node);
880 return p;
881}
882
883/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100884 * Returns the whole buffer contents concatenated for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100885 */
886 static char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100887channel_get_all(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100888{
889 /* Concatenate everything into one buffer.
890 * TODO: avoid multiple allocations. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100891 while (channel_collapse(channel, part) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100892 ;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100893 return channel_get(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100894}
895
896/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100897 * Collapses the first and second buffer for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100898 * Returns FAIL if that is not possible.
899 */
900 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100901channel_collapse(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100902{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100903 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100904 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100905 char_u *p;
906
Bram Moolenaar77073442016-02-13 23:23:53 +0100907 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100908 return FAIL;
909
Bram Moolenaar77073442016-02-13 23:23:53 +0100910 p = alloc((unsigned)(STRLEN(node->rq_buffer)
911 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100912 if (p == NULL)
913 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100914 STRCPY(p, node->rq_buffer);
915 STRCAT(p, node->rq_next->rq_buffer);
916 vim_free(node->rq_next->rq_buffer);
917 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100918
Bram Moolenaar77073442016-02-13 23:23:53 +0100919 /* dispose of the node and its buffer */
920 head->rq_next = node->rq_next;
921 head->rq_next->rq_prev = NULL;
922 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100923 vim_free(node);
924 return OK;
925}
926
927/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100928 * Store "buf[len]" on "channel"/"part".
929 * Returns OK or FAIL.
930 */
931 static int
932channel_save(channel_T *channel, int part, char_u *buf, int len)
933{
934 readq_T *node;
935 readq_T *head = &channel->ch_part[part].ch_head;
936 char_u *p;
937 int i;
938
939 node = (readq_T *)alloc(sizeof(readq_T));
940 if (node == NULL)
941 return FAIL; /* out of memory */
942 node->rq_buffer = alloc(len + 1);
943 if (node->rq_buffer == NULL)
944 {
945 vim_free(node);
946 return FAIL; /* out of memory */
947 }
948
949 if (channel->ch_part[part].ch_mode == MODE_NL)
950 {
951 /* Drop any CR before a NL. */
952 p = node->rq_buffer;
953 for (i = 0; i < len; ++i)
954 if (buf[i] != CAR || i + 1 >= len || buf[i + 1] != NL)
955 *p++ = buf[i];
956 *p = NUL;
957 }
958 else
959 {
960 mch_memmove(node->rq_buffer, buf, len);
961 node->rq_buffer[len] = NUL;
962 }
963
964 /* append node to the tail of the queue */
965 node->rq_next = NULL;
966 node->rq_prev = head->rq_prev;
967 if (head->rq_prev == NULL)
968 head->rq_next = node;
969 else
970 head->rq_prev->rq_next = node;
971 head->rq_prev = node;
972
973 if (log_fd != NULL)
974 {
975 ch_log_lead("RECV ", channel);
976 fprintf(log_fd, "'");
977 if (fwrite(buf, len, 1, log_fd) != 1)
978 return FAIL;
979 fprintf(log_fd, "'\n");
980 }
981 return OK;
982}
983
984/*
985 * Use the read buffer of "channel"/"part" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100986 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +0100987 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100988 */
Bram Moolenaard7ece102016-02-02 23:23:02 +0100989 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100990channel_parse_json(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100991{
992 js_read_T reader;
993 typval_T listtv;
994 jsonq_T *item;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100995 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100996 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100997
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100998 if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100999 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001000
1001 /* TODO: make reader work properly */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001002 /* reader.js_buf = channel_peek(channel, part); */
1003 reader.js_buf = channel_get_all(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001004 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +01001005 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001006 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +01001007 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001008 ret = json_decode(&reader, &listtv,
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001009 channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001010 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001011 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +01001012 /* Only accept the response when it is a list with at least two
1013 * items. */
1014 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001015 {
1016 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001017 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001018 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001019 else
1020 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001021 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
1022 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001023 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001024 else
1025 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001026 item->jq_value = alloc_tv();
1027 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001028 {
1029 vim_free(item);
1030 clear_tv(&listtv);
1031 }
1032 else
1033 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001034 *item->jq_value = listtv;
1035 item->jq_prev = head->jq_prev;
1036 head->jq_prev = item;
1037 item->jq_next = NULL;
1038 if (item->jq_prev == NULL)
1039 head->jq_next = item;
1040 else
1041 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001042 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001043 }
1044 }
1045 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001046
1047 /* Put the unread part back into the channel.
1048 * TODO: insert in front */
1049 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001050 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001051 channel_save(channel, part, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001052 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001053 ret = TRUE;
1054 }
1055 else
1056 ret = FALSE;
1057
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001058 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001059 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001060}
1061
1062/*
Bram Moolenaard46ae142016-02-16 13:33:52 +01001063 * Remove "node" from the queue that it is in. Does not free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +01001064 */
1065 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001066remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001067{
Bram Moolenaar77073442016-02-13 23:23:53 +01001068 if (node->cq_prev == NULL)
1069 head->cq_next = node->cq_next;
1070 else
1071 node->cq_prev->cq_next = node->cq_next;
1072 if (node->cq_next == NULL)
1073 head->cq_prev = node->cq_prev;
1074 else
1075 node->cq_next->cq_prev = node->cq_prev;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001076}
1077
1078/*
1079 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +01001080 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001081 */
1082 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001083remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001084{
Bram Moolenaar77073442016-02-13 23:23:53 +01001085 if (node->jq_prev == NULL)
1086 head->jq_next = node->jq_next;
1087 else
1088 node->jq_prev->jq_next = node->jq_next;
1089 if (node->jq_next == NULL)
1090 head->jq_prev = node->jq_prev;
1091 else
1092 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001093 vim_free(node);
1094}
1095
1096/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001097 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001098 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +01001099 * When "id" is zero or negative jut get the first message. But not the one
1100 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001101 * Return OK when found and return the value in "rettv".
1102 * Return FAIL otherwise.
1103 */
1104 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001105channel_get_json(channel_T *channel, int part, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001106{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001107 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001108 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001109
Bram Moolenaar77073442016-02-13 23:23:53 +01001110 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001111 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001112 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001113 typval_T *tv = &l->lv_first->li_tv;
1114
1115 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001116 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001117 || tv->vval.v_number == 0
1118 || tv->vval.v_number != channel->ch_part[part].ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001119 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001120 *rettv = item->jq_value;
1121 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001122 return OK;
1123 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001124 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001125 }
1126 return FAIL;
1127}
1128
Bram Moolenaarece61b02016-02-20 21:39:05 +01001129#define CH_JSON_MAX_ARGS 4
1130
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001131/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001132 * Execute a command received over "channel"/"part"
Bram Moolenaarece61b02016-02-20 21:39:05 +01001133 * "argv[0]" is the command string.
1134 * "argv[1]" etc. have further arguments, type is VAR_UNKNOWN if missing.
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001135 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001136 static void
Bram Moolenaarece61b02016-02-20 21:39:05 +01001137channel_exe_cmd(channel_T *channel, int part, typval_T *argv)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001138{
Bram Moolenaarece61b02016-02-20 21:39:05 +01001139 char_u *cmd = argv[0].vval.v_string;
1140 char_u *arg;
1141 int options = channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001142
Bram Moolenaarece61b02016-02-20 21:39:05 +01001143 if (argv[1].v_type != VAR_STRING)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001144 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001145 ch_error(channel, "received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001146 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001147 EMSG("E903: received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001148 return;
1149 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001150 arg = argv[1].vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001151 if (arg == NULL)
1152 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001153
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001154 if (STRCMP(cmd, "ex") == 0)
1155 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001156 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001157 }
1158 else if (STRCMP(cmd, "normal") == 0)
1159 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001160 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001161
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001162 ea.arg = arg;
1163 ea.addr_count = 0;
1164 ea.forceit = TRUE; /* no mapping */
1165 ex_normal(&ea);
1166 }
1167 else if (STRCMP(cmd, "redraw") == 0)
1168 {
1169 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001170
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001171 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001172 ex_redraw(&ea);
1173 showruler(FALSE);
1174 setcursor();
1175 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001176#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001177 if (gui.in_use)
1178 {
1179 gui_update_cursor(FALSE, FALSE);
1180 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001181 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001182#endif
1183 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001184 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "call") == 0)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001185 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001186 int is_call = cmd[0] == 'c';
1187 int id_idx = is_call ? 3 : 2;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001188
Bram Moolenaarece61b02016-02-20 21:39:05 +01001189 if (argv[id_idx].v_type != VAR_UNKNOWN
1190 && argv[id_idx].v_type != VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001191 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001192 ch_error(channel, "last argument for expr/call must be a number");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001193 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001194 EMSG("E904: last argument for expr/call must be a number");
1195 }
1196 else if (is_call && argv[2].v_type != VAR_LIST)
1197 {
1198 ch_error(channel, "third argument for call must be a list");
1199 if (p_verbose > 2)
1200 EMSG("E904: third argument for call must be a list");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001201 }
1202 else
1203 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001204 typval_T *tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001205 typval_T res_tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001206 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +01001207 char_u *json = NULL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001208
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001209 /* Don't pollute the display with errors. */
1210 ++emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001211 if (!is_call)
1212 tv = eval_expr(arg, NULL);
1213 else if (func_call(arg, &argv[2], NULL, &res_tv) == OK)
1214 tv = &res_tv;
1215 else
1216 tv = NULL;
1217
1218 if (argv[id_idx].v_type == VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001219 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001220 int id = argv[id_idx].vval.v_number;
1221
Bram Moolenaar55fab432016-02-07 16:53:13 +01001222 if (tv != NULL)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001223 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001224 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001225 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001226 /* If evaluation failed or the result can't be encoded
1227 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001228 vim_free(json);
1229 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001230 err_tv.v_type = VAR_STRING;
1231 err_tv.vval.v_string = (char_u *)"ERROR";
1232 tv = &err_tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001233 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001234 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001235 if (json != NULL)
1236 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001237 channel_send(channel,
1238 part == PART_SOCK ? PART_SOCK : PART_IN,
1239 json, (char *)cmd);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001240 vim_free(json);
1241 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001242 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001243 --emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001244 if (tv == &res_tv)
1245 clear_tv(tv);
1246 else if (tv != &err_tv)
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001247 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001248 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001249 }
1250 else if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001251 {
1252 ch_errors(channel, "Receved unknown command: %s", (char *)cmd);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001253 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001254 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001255}
1256
1257/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001258 * Invoke a callback for "channel"/"part" if needed.
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001259 * Return TRUE when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001260 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001261 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001262may_invoke_callback(channel_T *channel, int part)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001263{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001264 char_u *msg = NULL;
1265 typval_T *listtv = NULL;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001266 typval_T argv[CH_JSON_MAX_ARGS];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001267 int seq_nr = -1;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001268 ch_mode_T ch_mode = channel->ch_part[part].ch_mode;
1269 char_u *callback = NULL;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001270
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001271 if (channel->ch_nb_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001272 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001273 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001274
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001275 if (channel->ch_part[part].ch_callback != NULL)
1276 callback = channel->ch_part[part].ch_callback;
1277 else
1278 callback = channel->ch_callback;
1279
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001280 if (ch_mode == MODE_JSON || ch_mode == MODE_JS)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001281 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001282 listitem_T *item;
1283 int argc = 0;
1284
Bram Moolenaard7ece102016-02-02 23:23:02 +01001285 /* Get any json message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001286 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001287 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001288 /* Parse readahead, return when there is still no message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001289 channel_parse_json(channel, part);
1290 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001291 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001292 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001293
Bram Moolenaarece61b02016-02-20 21:39:05 +01001294 for (item = listtv->vval.v_list->lv_first;
1295 item != NULL && argc < CH_JSON_MAX_ARGS;
1296 item = item->li_next)
1297 argv[argc++] = item->li_tv;
1298 while (argc < CH_JSON_MAX_ARGS)
1299 argv[argc++].v_type = VAR_UNKNOWN;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001300
Bram Moolenaarece61b02016-02-20 21:39:05 +01001301 if (argv[0].v_type == VAR_STRING)
1302 {
1303 char_u *cmd = argv[0].vval.v_string;
1304
1305 /* ["cmd", arg] or ["cmd", arg, arg] or ["cmd", arg, arg, arg] */
Bram Moolenaar77073442016-02-13 23:23:53 +01001306 ch_logs(channel, "Executing %s command", (char *)cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001307 channel_exe_cmd(channel, part, argv);
Bram Moolenaar77073442016-02-13 23:23:53 +01001308 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001309 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001310 }
1311
Bram Moolenaarece61b02016-02-20 21:39:05 +01001312 if (argv[0].v_type != VAR_NUMBER)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001313 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001314 ch_error(channel,
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001315 "Dropping message with invalid sequence number type");
Bram Moolenaar77073442016-02-13 23:23:53 +01001316 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001317 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001318 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001319 seq_nr = argv[0].vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001320 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001321 else if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001322 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001323 /* nothing to read on RAW or NL channel */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001324 return FALSE;
1325 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001326 else
1327 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001328 /* If there is no callback drop the message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001329 if (callback == NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001330 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001331 while ((msg = channel_get(channel, part)) != NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001332 vim_free(msg);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001333 return FALSE;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001334 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001335
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001336 if (ch_mode == MODE_NL)
1337 {
1338 char_u *nl;
1339 char_u *buf;
1340
1341 /* See if we have a message ending in NL in the first buffer. If
1342 * not try to concatenate the first and the second buffer. */
1343 while (TRUE)
1344 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001345 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001346 nl = vim_strchr(buf, NL);
1347 if (nl != NULL)
1348 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001349 if (channel_collapse(channel, part) == FAIL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001350 return FALSE; /* incomplete message */
1351 }
1352 if (nl[1] == NUL)
1353 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001354 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001355 else
1356 {
1357 /* Copy the message into allocated memory and remove it from
1358 * the buffer. */
1359 msg = vim_strnsave(buf, (int)(nl - buf));
1360 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1361 }
1362 }
1363 else
1364 /* For a raw channel we don't know where the message ends, just
1365 * get everything we have. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001366 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001367
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001368 argv[1].v_type = VAR_STRING;
1369 argv[1].vval.v_string = msg;
1370 }
1371
Bram Moolenaara07fec92016-02-05 21:04:08 +01001372 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001373 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001374 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001375 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001376 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001377
1378 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001379 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001380 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001381 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001382 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001383 ch_log(channel, "Invoking one-time callback");
Bram Moolenaard46ae142016-02-16 13:33:52 +01001384 /* Remove the item from the list first, if the callback
1385 * invokes ch_close() the list will be cleared. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001386 remove_cb_node(head, item);
Bram Moolenaard46ae142016-02-16 13:33:52 +01001387 invoke_callback(channel, item->cq_callback, argv);
1388 vim_free(item->cq_callback);
1389 vim_free(item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001390 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001391 break;
1392 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001393 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001394 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001395 if (!done)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001396 ch_log(channel, "Dropping message without callback");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001397 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001398 else if (callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001399 {
1400 /* invoke the channel callback */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001401 ch_log(channel, "Invoking channel callback");
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001402 invoke_callback(channel, callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001403 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001404 else
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001405 ch_log(channel, "Dropping message");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001406
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001407 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001408 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001409 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001410
1411 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001412}
1413
1414/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001415 * Return TRUE when channel "channel" is open for writing to.
1416 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001417 */
1418 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001419channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001420{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001421 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001422#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001423 || channel->CH_IN_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001424#endif
1425 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001426}
1427
1428/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001429 * Return TRUE when channel "channel" is open for reading or writing.
1430 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001431 */
1432 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001433channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001434{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001435 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001436#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001437 || channel->CH_IN_FD != INVALID_FD
1438 || channel->CH_OUT_FD != INVALID_FD
1439 || channel->CH_ERR_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001440#endif
1441 );
1442}
1443
1444/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001445 * Return a string indicating the status of the channel.
1446 */
1447 char *
1448channel_status(channel_T *channel)
1449{
1450 if (channel == NULL)
1451 return "fail";
1452 if (channel_is_open(channel))
1453 return "open";
1454 return "closed";
1455}
1456
1457/*
1458 * Close channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001459 * This does not trigger the close callback.
1460 */
1461 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001462channel_close(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001463{
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001464 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001465
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001466#ifdef FEAT_GUI
1467 channel_gui_unregister(channel);
1468#endif
1469
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001470 if (channel->CH_SOCK_FD != INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001471 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001472 sock_close(channel->CH_SOCK_FD);
1473 channel->CH_SOCK_FD = INVALID_FD;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001474 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001475#if defined(CHANNEL_PIPES)
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001476 if (channel->CH_IN_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001477 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001478 fd_close(channel->CH_IN_FD);
1479 channel->CH_IN_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001480 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001481 if (channel->CH_OUT_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001482 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001483 fd_close(channel->CH_OUT_FD);
1484 channel->CH_OUT_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001485 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001486 if (channel->CH_ERR_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001487 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001488 fd_close(channel->CH_ERR_FD);
1489 channel->CH_ERR_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001490 }
1491#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001492
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001493 if (channel->ch_close_cb != NULL)
1494 {
1495 typval_T argv[1];
1496 typval_T rettv;
1497 int dummy;
1498
1499 /* invoke the close callback; increment the refcount to avoid it
1500 * being freed halfway */
1501 argv[0].v_type = VAR_CHANNEL;
1502 argv[0].vval.v_channel = channel;
1503 ++channel->ch_refcount;
1504 call_func(channel->ch_close_cb, (int)STRLEN(channel->ch_close_cb),
1505 &rettv, 1, argv, 0L, 0L, &dummy, TRUE, NULL);
1506 clear_tv(&rettv);
1507 --channel->ch_refcount;
1508
1509 /* the callback is only called once */
1510 vim_free(channel->ch_close_cb);
1511 channel->ch_close_cb = NULL;
1512 }
1513
1514 channel->ch_nb_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001515 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001516}
1517
Bram Moolenaard04a0202016-01-26 23:30:18 +01001518/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001519 * Return the first buffer from "channel"/"part" without removing it.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001520 * Returns NULL if there is nothing.
1521 */
1522 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001523channel_peek(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001524{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001525 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001526
Bram Moolenaar77073442016-02-13 23:23:53 +01001527 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001528 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001529 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001530}
1531
1532/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001533 * Clear the read buffer on "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001534 */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001535 static void
1536channel_clear_one(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001537{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001538 jsonq_T *json_head = &channel->ch_part[part].ch_json_head;
1539 cbq_T *cb_head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001540
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001541 while (channel_peek(channel, part) != NULL)
1542 vim_free(channel_get(channel, part));
Bram Moolenaar77073442016-02-13 23:23:53 +01001543
1544 while (cb_head->cq_next != NULL)
Bram Moolenaard46ae142016-02-16 13:33:52 +01001545 {
1546 cbq_T *node = cb_head->cq_next;
1547
1548 remove_cb_node(cb_head, node);
1549 vim_free(node->cq_callback);
1550 vim_free(node);
1551 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001552
1553 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001554 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001555 free_tv(json_head->jq_next->jq_value);
1556 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001557 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001558
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001559 vim_free(channel->ch_part[part].ch_callback);
1560 channel->ch_part[part].ch_callback = NULL;
1561}
1562
1563/*
1564 * Clear all the read buffers on "channel".
1565 */
1566 void
1567channel_clear(channel_T *channel)
1568{
1569 channel_clear_one(channel, PART_SOCK);
1570#ifdef CHANNEL_PIPES
1571 channel_clear_one(channel, PART_OUT);
1572 channel_clear_one(channel, PART_ERR);
1573#endif
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001574 vim_free(channel->ch_callback);
1575 channel->ch_callback = NULL;
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001576 vim_free(channel->ch_close_cb);
1577 channel->ch_close_cb = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001578}
1579
Bram Moolenaar77073442016-02-13 23:23:53 +01001580#if defined(EXITFREE) || defined(PROTO)
1581 void
1582channel_free_all(void)
1583{
1584 channel_T *channel;
1585
1586 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1587 channel_clear(channel);
1588}
1589#endif
1590
1591
Bram Moolenaard04a0202016-01-26 23:30:18 +01001592/* Sent when the channel is found closed when reading. */
Bram Moolenaareed284a2016-02-22 23:13:33 +01001593#define DETACH_MSG_RAW "DETACH\n"
1594#define DETACH_MSG_JSON "\"DETACH\"\n"
Bram Moolenaard04a0202016-01-26 23:30:18 +01001595
1596/* Buffer size for reading incoming messages. */
1597#define MAXMSGSIZE 4096
1598
1599/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001600 * Check for reading from "fd" with "timeout" msec.
1601 * Return FAIL when there is nothing to read.
Bram Moolenaara8343c12016-02-04 22:09:48 +01001602 * Always returns OK for FEAT_GUI_W32.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001603 */
1604 static int
Bram Moolenaard8070362016-02-15 21:56:54 +01001605channel_wait(channel_T *channel, sock_T fd, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001606{
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001607 if (timeout > 0)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001608 ch_logn(channel, "Waiting for up to %d msec", timeout);
Bram Moolenaard8070362016-02-15 21:56:54 +01001609
Bram Moolenaard8070362016-02-15 21:56:54 +01001610# ifdef WIN32
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001611 if (fd != channel->CH_SOCK_FD)
Bram Moolenaard8070362016-02-15 21:56:54 +01001612 {
1613 DWORD nread;
1614 int diff;
1615 DWORD deadline = GetTickCount() + timeout;
1616
1617 /* reading from a pipe, not a socket */
1618 while (TRUE)
1619 {
Bram Moolenaare74e8e72016-02-16 22:01:30 +01001620 if (PeekNamedPipe((HANDLE)fd, NULL, 0, NULL, &nread, NULL)
1621 && nread > 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001622 return OK;
1623 diff = deadline - GetTickCount();
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001624 if (diff <= 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001625 break;
1626 /* Wait for 5 msec.
1627 * TODO: increase the sleep time when looping more often */
1628 Sleep(5);
1629 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001630 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001631 else
Bram Moolenaard8070362016-02-15 21:56:54 +01001632#endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001633 {
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001634#if defined(FEAT_GUI_W32)
1635 /* Can't check socket for Win32 GUI, always return OK. */
1636 ch_log(channel, "Can't check, assuming there is something to read");
1637 return OK;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001638#else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001639# if defined(HAVE_SELECT)
1640 struct timeval tval;
1641 fd_set rfds;
1642 int ret;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001643
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001644 FD_ZERO(&rfds);
1645 FD_SET((int)fd, &rfds);
1646 tval.tv_sec = timeout / 1000;
1647 tval.tv_usec = (timeout % 1000) * 1000;
1648 for (;;)
1649 {
1650 ret = select((int)fd + 1, &rfds, NULL, NULL, &tval);
1651# ifdef EINTR
1652 SOCK_ERRNO;
1653 if (ret == -1 && errno == EINTR)
1654 continue;
1655# endif
1656 if (ret > 0)
1657 return OK;
1658 break;
1659 }
1660# else
1661 struct pollfd fds;
1662
1663 fds.fd = fd;
1664 fds.events = POLLIN;
1665 if (poll(&fds, 1, timeout) > 0)
1666 return OK;
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001667# endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001668#endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001669 }
1670 ch_log(channel, "Nothing to read");
1671 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001672}
1673
1674/*
1675 * Return a unique ID to be used in a message.
1676 */
1677 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001678channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001679{
1680 static int next_id = 1;
1681
1682 return next_id++;
1683}
1684
1685/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001686 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001687 * "part" is PART_SOCK, PART_OUT or PART_ERR.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001688 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001689 */
1690 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001691channel_read(channel_T *channel, int part, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001692{
1693 static char_u *buf = NULL;
1694 int len = 0;
1695 int readlen = 0;
Bram Moolenaard8070362016-02-15 21:56:54 +01001696 sock_T fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001697 int use_socket = FALSE;
Bram Moolenaareed284a2016-02-22 23:13:33 +01001698 char *msg;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001699
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001700 fd = channel->ch_part[part].ch_fd;
1701 if (fd == INVALID_FD)
1702 {
1703 ch_error(channel, "channel_read() called while socket is closed");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001704 return;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001705 }
1706 use_socket = fd == channel->CH_SOCK_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001707
1708 /* Allocate a buffer to read into. */
1709 if (buf == NULL)
1710 {
1711 buf = alloc(MAXMSGSIZE);
1712 if (buf == NULL)
1713 return; /* out of memory! */
1714 }
1715
1716 /* Keep on reading for as long as there is something to read.
1717 * Use select() or poll() to avoid blocking on a message that is exactly
1718 * MAXMSGSIZE long. */
1719 for (;;)
1720 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001721 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001722 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001723 if (use_socket)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001724 len = sock_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001725 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001726 len = fd_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001727 if (len <= 0)
1728 break; /* error or nothing more to read */
1729
1730 /* Store the read message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001731 channel_save(channel, part, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001732 readlen += len;
1733 if (len < MAXMSGSIZE)
1734 break; /* did read everything that's available */
1735 }
Bram Moolenaara8343c12016-02-04 22:09:48 +01001736#ifdef FEAT_GUI_W32
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001737 if (use_socket && len == SOCKET_ERROR)
Bram Moolenaara8343c12016-02-04 22:09:48 +01001738 {
1739 /* For Win32 GUI channel_wait() always returns OK and we handle the
1740 * situation that there is nothing to read here.
1741 * TODO: how about a timeout? */
1742 if (WSAGetLastError() == WSAEWOULDBLOCK)
1743 return;
1744 }
1745#endif
Bram Moolenaard04a0202016-01-26 23:30:18 +01001746
Bram Moolenaarbd73ae12016-02-22 22:19:22 +01001747 /* Reading a disconnection (readlen == 0), or an error.
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001748 * TODO: call error callback. */
Bram Moolenaarbd73ae12016-02-22 22:19:22 +01001749 if (readlen <= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001750 {
1751 /* Queue a "DETACH" netbeans message in the command queue in order to
1752 * terminate the netbeans session later. Do not end the session here
1753 * directly as we may be running in the context of a call to
1754 * netbeans_parse_messages():
1755 * netbeans_parse_messages
1756 * -> autocmd triggered while processing the netbeans cmd
1757 * -> ui_breakcheck
1758 * -> gui event loop or select loop
1759 * -> channel_read()
1760 */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001761 ch_errors(channel, "%s(): Cannot read", func);
Bram Moolenaareed284a2016-02-22 23:13:33 +01001762 msg = channel->ch_part[part].ch_mode == MODE_RAW
1763 || channel->ch_part[part].ch_mode == MODE_NL
1764 ? DETACH_MSG_RAW : DETACH_MSG_JSON;
1765 channel_save(channel, part, (char_u *)msg, (int)STRLEN(msg));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001766
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001767 /* TODO: When reading from stdout is not possible, should we try to
1768 * keep stdin and stderr open? Probably not, assume the other side
1769 * has died. */
1770 channel_close(channel);
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001771 if (channel->ch_nb_close_cb != NULL)
1772 (*channel->ch_nb_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001773
1774 if (len < 0)
1775 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001776 ch_error(channel, "channel_read(): cannot read from channel");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001777 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001778 }
1779 }
1780
1781#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001782 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001783 if (CH_HAS_GUI && gtk_main_level() > 0)
1784 gtk_main_quit();
1785#endif
1786}
1787
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001788/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001789 * Read from RAW or NL "channel"/"part". Blocks until there is something to
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001790 * read or the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001791 * Returns what was read in allocated memory.
1792 * Returns NULL in case of error or timeout.
1793 */
1794 char_u *
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001795channel_read_block(channel_T *channel, int part, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001796{
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001797 char_u *buf;
1798 char_u *msg;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001799 ch_mode_T mode = channel->ch_part[part].ch_mode;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001800 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001801 char_u *nl;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001802
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001803 ch_logsn(channel, "Blocking %s read, timeout: %d msec",
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001804 mode == MODE_RAW ? "RAW" : "NL", timeout);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001805
1806 while (TRUE)
1807 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001808 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001809 if (buf != NULL && (mode == MODE_RAW
1810 || (mode == MODE_NL && vim_strchr(buf, NL) != NULL)))
1811 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001812 if (buf != NULL && channel_collapse(channel, part) == OK)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001813 continue;
1814
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001815 /* Wait for up to the channel timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001816 if (fd == INVALID_FD
1817 || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001818 return NULL;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001819 channel_read(channel, part, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001820 }
1821
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001822 if (mode == MODE_RAW)
1823 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001824 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001825 }
1826 else
1827 {
1828 nl = vim_strchr(buf, NL);
1829 if (nl[1] == NUL)
1830 {
1831 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001832 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001833 *nl = NUL;
1834 }
1835 else
1836 {
1837 /* Copy the message into allocated memory and remove it from the
1838 * buffer. */
1839 msg = vim_strnsave(buf, (int)(nl - buf));
1840 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1841 }
1842 }
1843 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001844 ch_logn(channel, "Returning %d bytes", (int)STRLEN(msg));
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001845 return msg;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001846}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001847
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001848/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001849 * Read one JSON message with ID "id" from "channel"/"part" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001850 * result in "rettv".
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001851 * When "id" is -1 accept any message;
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001852 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001853 */
1854 int
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001855channel_read_json_block(
1856 channel_T *channel,
1857 int part,
1858 int timeout,
1859 int id,
1860 typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001861{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001862 int more;
Bram Moolenaard8070362016-02-15 21:56:54 +01001863 sock_T fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001864
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001865 ch_log(channel, "Reading JSON");
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001866 if (id != -1)
1867 channel->ch_part[part].ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001868 for (;;)
1869 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001870 more = channel_parse_json(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001871
1872 /* search for messsage "id" */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001873 if (channel_get_json(channel, part, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001874 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001875 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001876 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001877 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001878
Bram Moolenaard7ece102016-02-02 23:23:02 +01001879 if (!more)
1880 {
1881 /* Handle any other messages in the queue. If done some more
1882 * messages may have arrived. */
1883 if (channel_parse_messages())
1884 continue;
1885
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001886 /* Wait for up to the timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001887 fd = channel->ch_part[part].ch_fd;
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001888 if (fd == INVALID_FD || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001889 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001890 channel_read(channel, part, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001891 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001892 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001893 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001894 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001895}
1896
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001897# if defined(WIN32) || defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK) \
1898 || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001899/*
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001900 * Lookup the channel from the socket. Set "partp" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001901 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001902 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001903 channel_T *
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001904channel_fd2channel(sock_T fd, int *partp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001905{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001906 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001907 int part;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001908
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001909 if (fd != INVALID_FD)
Bram Moolenaar77073442016-02-13 23:23:53 +01001910 for (channel = first_channel; channel != NULL;
1911 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001912 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001913# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001914 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001915# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001916 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001917# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001918 if (channel->ch_part[part].ch_fd == fd)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001919 {
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001920 *partp = part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001921 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001922 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001923 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001924 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001925}
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001926# endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001927
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001928# if defined(WIN32) || defined(PROTO)
1929/*
1930 * Check the channels for anything that is ready to be read.
1931 * The data is put in the read queue.
1932 */
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001933 void
1934channel_handle_events(void)
1935{
1936 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001937 int part;
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001938 sock_T fd;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001939
1940 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1941 {
1942# ifdef FEAT_GUI_W32
1943 /* only check the pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001944 for (part = PART_OUT; part <= PART_ERR; ++part)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001945# else
1946# ifdef CHANNEL_PIPES
1947 /* check the socket and pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001948 for (part = PART_SOCK; part <= PART_ERR; ++part)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001949# else
1950 /* only check the socket */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001951 part = PART_SOCK;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001952# endif
1953# endif
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001954 {
1955 fd = channel->ch_part[part].ch_fd;
1956 if (fd != INVALID_FD && channel_wait(channel, fd, 0) == OK)
1957 channel_read(channel, part, "channel_handle_events");
1958 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001959 }
1960}
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001961# endif
1962
Bram Moolenaard04a0202016-01-26 23:30:18 +01001963/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001964 * Write "buf" (NUL terminated string) to "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001965 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001966 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001967 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001968 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001969channel_send(channel_T *channel, int part, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001970{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001971 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001972 int res;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001973 sock_T fd;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001974
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001975 fd = channel->ch_part[part].ch_fd;
1976 if (fd == INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001977 {
1978 if (!channel->ch_error && fun != NULL)
1979 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001980 ch_errors(channel, "%s(): write while not connected", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001981 EMSG2("E630: %s(): write while not connected", fun);
1982 }
1983 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001984 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001985 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001986
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001987 if (log_fd != NULL)
1988 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001989 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001990 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001991 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001992 fprintf(log_fd, "'\n");
1993 fflush(log_fd);
1994 }
1995
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001996 if (part == PART_SOCK)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001997 res = sock_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001998 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001999 res = fd_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002000 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01002001 {
2002 if (!channel->ch_error && fun != NULL)
2003 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01002004 ch_errors(channel, "%s(): write failed", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01002005 EMSG2("E631: %s(): write failed", fun);
2006 }
2007 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01002008 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01002009 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01002010
2011 channel->ch_error = FALSE;
2012 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01002013}
2014
2015# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002016/*
2017 * Add open channels to the poll struct.
2018 * Return the adjusted struct index.
2019 * The type of "fds" is hidden to avoid problems with the function proto.
2020 */
2021 int
2022channel_poll_setup(int nfd_in, void *fds_in)
2023{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002024 int nfd = nfd_in;
2025 channel_T *channel;
2026 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002027 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002028
Bram Moolenaar77073442016-02-13 23:23:53 +01002029 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002030 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002031# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002032 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002033# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002034 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002035# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002036 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002037 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002038 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002039 channel->ch_part[part].ch_poll_idx = nfd;
2040 fds[nfd].fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002041 fds[nfd].events = POLLIN;
2042 nfd++;
2043 }
2044 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002045 channel->ch_part[part].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002046 }
2047 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002048
2049 return nfd;
2050}
2051
2052/*
2053 * The type of "fds" is hidden to avoid problems with the function proto.
2054 */
2055 int
2056channel_poll_check(int ret_in, void *fds_in)
2057{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002058 int ret = ret_in;
2059 channel_T *channel;
2060 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002061 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002062
Bram Moolenaar77073442016-02-13 23:23:53 +01002063 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002064 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002065# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002066 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002067# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002068 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002069# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002070 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002071 int idx = channel->ch_part[part].ch_poll_idx;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002072
2073 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
2074 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002075 channel_read(channel, part, "channel_poll_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002076 --ret;
2077 }
2078 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002079 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002080
2081 return ret;
2082}
Bram Moolenaard04a0202016-01-26 23:30:18 +01002083# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002084
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002085# if (!defined(WIN32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002086/*
2087 * The type of "rfds" is hidden to avoid problems with the function proto.
2088 */
2089 int
2090channel_select_setup(int maxfd_in, void *rfds_in)
2091{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002092 int maxfd = maxfd_in;
2093 channel_T *channel;
2094 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002095 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002096
Bram Moolenaar77073442016-02-13 23:23:53 +01002097 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002098 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002099# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002100 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002101# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002102 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002103# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002104 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002105 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002106
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002107 if (fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002108 {
Bram Moolenaard8070362016-02-15 21:56:54 +01002109 FD_SET((int)fd, rfds);
2110 if (maxfd < (int)fd)
2111 maxfd = (int)fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002112 }
2113 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002114 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002115
2116 return maxfd;
2117}
2118
2119/*
2120 * The type of "rfds" is hidden to avoid problems with the function proto.
2121 */
2122 int
2123channel_select_check(int ret_in, void *rfds_in)
2124{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002125 int ret = ret_in;
2126 channel_T *channel;
2127 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002128 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002129
Bram Moolenaar77073442016-02-13 23:23:53 +01002130 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002131 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002132# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002133 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002134# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002135 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002136# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002137 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002138 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002139
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002140 if (ret > 0 && fd != INVALID_FD && FD_ISSET(fd, rfds))
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002141 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002142 channel_read(channel, part, "channel_select_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002143 --ret;
2144 }
2145 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002146 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002147
2148 return ret;
2149}
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002150# endif /* !WIN32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002151
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002152/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01002153 * Execute queued up commands.
2154 * Invoked from the main loop when it's safe to execute received commands.
2155 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002156 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01002157 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002158channel_parse_messages(void)
2159{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002160 channel_T *channel = first_channel;
2161 int ret = FALSE;
2162 int r;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002163 int part = PART_SOCK;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002164
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002165 while (channel != NULL)
2166 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002167 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaard7ece102016-02-02 23:23:02 +01002168 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002169 /* Increase the refcount, in case the handler causes the channel
2170 * to be unreferenced or closed. */
2171 ++channel->ch_refcount;
2172 r = may_invoke_callback(channel, part);
2173 if (r == OK)
2174 ret = TRUE;
2175 if (channel_unref(channel) || r == OK)
2176 {
2177 /* channel was freed or something was done, start over */
2178 channel = first_channel;
2179 part = PART_SOCK;
2180 continue;
2181 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002182 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002183#ifdef CHANNEL_PIPES
2184 if (part < PART_ERR)
2185 ++part;
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002186 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002187#endif
2188 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002189 channel = channel->ch_next;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002190 part = PART_SOCK;
2191 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002192 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002193 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002194}
2195
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01002196/*
2197 * Mark references to lists used in channels.
2198 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002199 int
2200set_ref_in_channel(int copyID)
2201{
Bram Moolenaar77073442016-02-13 23:23:53 +01002202 int abort = FALSE;
2203 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002204 int part;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002205
Bram Moolenaar77073442016-02-13 23:23:53 +01002206 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002207 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002208#ifdef CHANNEL_PIPES
2209 for (part = PART_SOCK; part < PART_IN; ++part)
2210#else
2211 part = PART_SOCK;
2212#endif
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002213 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002214 jsonq_T *head = &channel->ch_part[part].ch_json_head;
2215 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002216
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002217 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002218 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002219 list_T *l = item->jq_value->vval.v_list;
2220
2221 if (l->lv_copyID != copyID)
2222 {
2223 l->lv_copyID = copyID;
2224 abort = abort || set_ref_in_list(l, copyID, NULL);
2225 }
2226 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002227 }
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002228 }
2229 }
2230 return abort;
2231}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002232
2233/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002234 * Return the "part" to write to for "channel".
2235 */
2236 int
2237channel_part_send(channel_T *channel)
2238{
2239#ifdef CHANNEL_PIPES
2240 if (channel->CH_SOCK_FD == INVALID_FD)
2241 return PART_IN;
2242#endif
2243 return PART_SOCK;
2244}
2245
2246/*
2247 * Return the default "part" to read from for "channel".
2248 */
2249 int
2250channel_part_read(channel_T *channel)
2251{
2252#ifdef CHANNEL_PIPES
2253 if (channel->CH_SOCK_FD == INVALID_FD)
2254 return PART_OUT;
2255#endif
2256 return PART_SOCK;
2257}
2258
2259/*
2260 * Return the mode of "channel"/"part"
Bram Moolenaar77073442016-02-13 23:23:53 +01002261 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002262 */
2263 ch_mode_T
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002264channel_get_mode(channel_T *channel, int part)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002265{
Bram Moolenaar77073442016-02-13 23:23:53 +01002266 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002267 return MODE_JSON;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002268 return channel->ch_part[part].ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002269}
2270
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01002271/*
2272 * Return the timeout of "channel"/"part"
2273 */
2274 int
2275channel_get_timeout(channel_T *channel, int part)
2276{
2277 return channel->ch_part[part].ch_timeout;
2278}
2279
Bram Moolenaare0874f82016-01-24 20:36:41 +01002280#endif /* FEAT_CHANNEL */