blob: 1e26eae97a40489e935f5089e6ae56635c2e0c95 [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 Moolenaar77073442016-02-13 23:23:53 +0100332#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
333 static channel_T *
334channel_from_id(int id)
335{
336 channel_T *channel;
337
338 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
339 if (channel->ch_id == id)
340 return channel;
341 return NULL;
342}
343#endif
344
Bram Moolenaard04a0202016-01-26 23:30:18 +0100345#if defined(FEAT_GUI) || defined(PROTO)
Bram Moolenaar77073442016-02-13 23:23:53 +0100346
347#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
348 static void
349channel_read_netbeans(int id)
350{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100351 channel_T *channel = channel_from_id(id);
352 int part;
Bram Moolenaar77073442016-02-13 23:23:53 +0100353
354 if (channel == NULL)
355 ch_errorn(NULL, "Channel %d not found", id);
356 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100357 {
358 /* TODO: check stderr */
359 if (channel->CH_SOCK_FD != INVALID_FD)
360 part = PART_SOCK;
361 else
362 part = PART_OUT;
363 channel_read(channel, part, "messageFromNetbeans");
364 }
Bram Moolenaar77073442016-02-13 23:23:53 +0100365}
366#endif
367
Bram Moolenaare0874f82016-01-24 20:36:41 +0100368/*
Bram Moolenaard04a0202016-01-26 23:30:18 +0100369 * Read a command from netbeans.
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100370 * TODO: instead of channel ID use the FD.
Bram Moolenaare0874f82016-01-24 20:36:41 +0100371 */
Bram Moolenaard04a0202016-01-26 23:30:18 +0100372#ifdef FEAT_GUI_X11
373 static void
374messageFromNetbeans(XtPointer clientData,
375 int *unused1 UNUSED,
376 XtInputId *unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100377{
Bram Moolenaar77073442016-02-13 23:23:53 +0100378 channel_read_netbeans((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100379}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100380#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100381
Bram Moolenaard04a0202016-01-26 23:30:18 +0100382#ifdef FEAT_GUI_GTK
383 static void
384messageFromNetbeans(gpointer clientData,
385 gint unused1 UNUSED,
386 GdkInputCondition unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100387{
Bram Moolenaar77073442016-02-13 23:23:53 +0100388 channel_read_netbeans((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100389}
390#endif
391
392 static void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100393channel_gui_register_one(channel_T *channel, int part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100394{
395# ifdef FEAT_GUI_X11
396 /* Tell notifier we are interested in being called
397 * when there is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100398 if (channel->ch_part[part].ch_inputHandler == (XtInputId)NULL)
399 channel->ch_part[part].ch_inputHandler = XtAppAddInput(
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100400 (XtAppContext)app_context,
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100401 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100402 (XtPointer)(XtInputReadMask + XtInputExceptMask),
403 messageFromNetbeans,
404 (XtPointer)(long)channel->ch_id);
405# else
406# ifdef FEAT_GUI_GTK
407 /* Tell gdk we are interested in being called when there
408 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100409 if (channel->ch_part[part].ch_inputHandler == 0)
410 channel->ch_part[part].ch_inputHandler = gdk_input_add(
411 (gint)channel->ch_part[part].ch_fd,
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100412 (GdkInputCondition)
413 ((int)GDK_INPUT_READ + (int)GDK_INPUT_EXCEPTION),
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100414 messageFromNetbeans,
415 (gpointer)(long)channel->ch_id);
416# else
417# ifdef FEAT_GUI_W32
418 /* Tell Windows we are interested in receiving message when there
419 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100420 if (channel->ch_part[part].ch_inputHandler == -1)
421 channel->ch_part[part].ch_inputHandler = WSAAsyncSelect(
422 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100423 s_hwnd, WM_NETBEANS, FD_READ);
424# endif
425# endif
426# endif
427}
428
429 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100430channel_gui_register(channel_T *channel)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100431{
Bram Moolenaard04a0202016-01-26 23:30:18 +0100432 if (!CH_HAS_GUI)
433 return;
434
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100435 if (channel->CH_SOCK_FD != INVALID_FD)
436 channel_gui_register_one(channel, PART_SOCK);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100437# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100438 if (channel->CH_OUT_FD != INVALID_FD)
439 channel_gui_register_one(channel, PART_OUT);
440 if (channel->CH_ERR_FD != INVALID_FD)
441 channel_gui_register_one(channel, PART_ERR);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100442# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100443}
444
445/*
446 * Register any of our file descriptors with the GUI event handling system.
447 * Called when the GUI has started.
448 */
449 void
450channel_gui_register_all(void)
451{
Bram Moolenaar77073442016-02-13 23:23:53 +0100452 channel_T *channel;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100453
Bram Moolenaar77073442016-02-13 23:23:53 +0100454 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100455 channel_gui_register(channel);
456}
457
458 static void
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100459channel_gui_unregister(channel_T *channel)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100460{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100461 int part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100462
463#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100464 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100465#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100466 part = PART_SOCK;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100467#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100468 {
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100469# ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100470 if (channel->ch_part[part].ch_inputHandler != (XtInputId)NULL)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100471 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100472 XtRemoveInput(channel->ch_part[part].ch_inputHandler);
473 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100474 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100475# else
476# ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100477 if (channel->ch_part[part].ch_inputHandler != 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100478 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100479 gdk_input_remove(channel->ch_part[part].ch_inputHandler);
480 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100481 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100482# else
483# ifdef FEAT_GUI_W32
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100484 if (channel->ch_part[part].ch_inputHandler == 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100485 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100486 WSAAsyncSelect(channel->ch_part[part].ch_fd, s_hwnd, 0, 0);
487 channel->ch_part[part].ch_inputHandler = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100488 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100489# endif
490# endif
491# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100492 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100493}
494
495#endif
496
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100497static char *e_cannot_connect = N_("E902: Cannot connect to port");
498
Bram Moolenaard04a0202016-01-26 23:30:18 +0100499/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100500 * Open a socket channel to "hostname":"port".
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100501 * "waittime" is the time in msec to wait for the connection.
502 * When negative wait forever.
Bram Moolenaar77073442016-02-13 23:23:53 +0100503 * Returns the channel for success.
504 * Returns NULL for failure.
Bram Moolenaard04a0202016-01-26 23:30:18 +0100505 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100506 channel_T *
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100507channel_open(char *hostname, int port_in, int waittime, void (*close_cb)(void))
Bram Moolenaard04a0202016-01-26 23:30:18 +0100508{
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100509 int sd = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100510 struct sockaddr_in server;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100511 struct hostent *host;
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100512#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100513 u_short port = port_in;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100514 u_long val = 1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100515#else
516 int port = port_in;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100517 struct timeval start_tv;
518 int so_error;
519 socklen_t so_error_len = sizeof(so_error);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100520#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100521 channel_T *channel;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100522 int ret;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100523
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100524#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100525 channel_init_winsock();
526#endif
527
Bram Moolenaar77073442016-02-13 23:23:53 +0100528 channel = add_channel();
529 if (channel == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100530 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100531 ch_error(NULL, "Cannot allocate channel.");
Bram Moolenaar77073442016-02-13 23:23:53 +0100532 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100533 }
534
535 /* Get the server internet address and put into addr structure */
536 /* fill in the socket address structure and connect to server */
537 vim_memset((char *)&server, 0, sizeof(server));
538 server.sin_family = AF_INET;
539 server.sin_port = htons(port);
540 if ((host = gethostbyname(hostname)) == NULL)
541 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100542 ch_error(channel, "in gethostbyname() in channel_open()");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100543 PERROR("E901: gethostbyname() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100544 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100545 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100546 }
547 memcpy((char *)&server.sin_addr, host->h_addr, host->h_length);
548
Bram Moolenaar254e00d2016-02-19 23:23:12 +0100549 /* On Mac and Solaris a zero timeout almost never works. At least wait
550 * one millisecond. Let's do it for all systems, because we don't know why
551 * this is needed. */
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100552 if (waittime == 0)
553 waittime = 1;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100554
555 /*
556 * For Unix we need to call connect() again after connect() failed.
557 * On Win32 one time is sufficient.
558 */
559 while (TRUE)
560 {
561 if (sd >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100562 sock_close(sd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100563 sd = socket(AF_INET, SOCK_STREAM, 0);
564 if (sd == -1)
565 {
566 ch_error(channel, "in socket() in channel_open().");
567 PERROR("E898: socket() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100568 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100569 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100570 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100571
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100572 if (waittime >= 0)
573 {
574 /* Make connect() non-blocking. */
575 if (
576#ifdef _WIN32
577 ioctlsocket(sd, FIONBIO, &val) < 0
578#else
579 fcntl(sd, F_SETFL, O_NONBLOCK) < 0
580#endif
581 )
582 {
583 SOCK_ERRNO;
584 ch_errorn(channel,
585 "channel_open: Connect failed with errno %d", errno);
586 sock_close(sd);
587 channel_free(channel);
588 return NULL;
589 }
590 }
591
592 /* Try connecting to the server. */
593 ch_logsn(channel, "Connecting to %s port %d", hostname, port);
594 ret = connect(sd, (struct sockaddr *)&server, sizeof(server));
595
596 SOCK_ERRNO;
597 if (ret < 0)
598 {
599 if (errno != EWOULDBLOCK
600 && errno != ECONNREFUSED
601
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100602#ifdef EINPROGRESS
603 && errno != EINPROGRESS
604#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100605 )
606 {
607 ch_errorn(channel,
608 "channel_open: Connect failed with errno %d", errno);
609 PERROR(_(e_cannot_connect));
610 sock_close(sd);
611 channel_free(channel);
612 return NULL;
613 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100614 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100615
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100616 /* If we don't block and connect() failed then try using select() to
617 * wait for the connection to be made. */
618 if (waittime >= 0 && ret < 0)
619 {
620 struct timeval tv;
621 fd_set wfds;
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100622#if defined(__APPLE__) && __APPLE__ == 1
623# define PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100624 fd_set rfds;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100625
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100626 FD_ZERO(&rfds);
627 FD_SET(sd, &rfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100628#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100629 FD_ZERO(&wfds);
630 FD_SET(sd, &wfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100631
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100632 tv.tv_sec = waittime / 1000;
633 tv.tv_usec = (waittime % 1000) * 1000;
634#ifndef WIN32
635 gettimeofday(&start_tv, NULL);
636#endif
637 ch_logn(channel,
638 "Waiting for connection (waittime %d msec)...", waittime);
639 ret = select((int)sd + 1,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100640#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100641 &rfds,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100642#else
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100643 NULL,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100644#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100645 &wfds, NULL, &tv);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100646
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100647 if (ret < 0)
648 {
649 SOCK_ERRNO;
650 ch_errorn(channel,
651 "channel_open: Connect failed with errno %d", errno);
652 PERROR(_(e_cannot_connect));
653 sock_close(sd);
654 channel_free(channel);
655 return NULL;
656 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100657#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100658 if (ret == 0 && FD_ISSET(sd, &rfds) && FD_ISSET(sd, &wfds))
659 {
660 /* For OS X, this implies error. See tcp(4). */
661 ch_error(channel, "channel_open: Connect failed");
662 EMSG(_(e_cannot_connect));
663 sock_close(sd);
664 channel_free(channel);
665 return NULL;
666 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100667#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100668#ifdef WIN32
669 /* On Win32 select() is expected to work and wait for up to the
670 * waittime for the socket to be open. */
671 if (!FD_ISSET(sd, &wfds) || ret == 0)
672#else
673 /* See socket(7) for the behavior on Linux-like systems:
674 * After putting the socket in non-blocking mode, connect() will
675 * return EINPROGRESS, select() will not wait (as if writing is
676 * possible), need to use getsockopt() to check if the socket is
677 * actually open. */
678 getsockopt(sd, SOL_SOCKET, SO_ERROR, &so_error, &so_error_len);
679 if (!FD_ISSET(sd, &wfds) || ret == 0 || so_error != 0)
680#endif
681 {
682#ifndef WIN32
683 struct timeval end_tv;
684 long elapsed_msec;
685
686 gettimeofday(&end_tv, NULL);
687 elapsed_msec = (end_tv.tv_sec - start_tv.tv_sec) * 1000
688 + (end_tv.tv_usec - start_tv.tv_usec) / 1000;
689 if (waittime > 1 && elapsed_msec < waittime)
690 {
691 /* The port isn't ready but we also didn't get an error.
692 * This happens when the server didn't open the socket
693 * yet. Wait a bit and try again. */
694 mch_delay(waittime < 50 ? (long)waittime : 50L, TRUE);
695 ui_breakcheck();
696 if (!got_int)
697 {
698 /* reduce the waittime by the elapsed time and the 50
699 * msec delay (or a bit more) */
700 waittime -= elapsed_msec;
701 if (waittime > 50)
702 waittime -= 50;
703 else
704 waittime = 1;
705 continue;
706 }
707 /* we were interrupted, behave as if timed out */
708 }
709#endif
710 /* We timed out. */
711 ch_error(channel, "Connection timed out");
712 sock_close(sd);
713 channel_free(channel);
714 return NULL;
715 }
716
717 ch_log(channel, "Connection made");
718 break;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100719 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100720 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100721
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100722 if (waittime >= 0)
723 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100724#ifdef _WIN32
725 val = 0;
726 ioctlsocket(sd, FIONBIO, &val);
727#else
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100728 (void)fcntl(sd, F_SETFL, 0);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100729#endif
730 }
731
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100732 channel->CH_SOCK_FD = (sock_T)sd;
Bram Moolenaar77073442016-02-13 23:23:53 +0100733 channel->ch_close_cb = close_cb;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100734
735#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +0100736 channel_gui_register(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100737#endif
738
Bram Moolenaar77073442016-02-13 23:23:53 +0100739 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100740}
741
742#if defined(CHANNEL_PIPES) || defined(PROTO)
743 void
Bram Moolenaard8070362016-02-15 21:56:54 +0100744channel_set_pipes(channel_T *channel, sock_T in, sock_T out, sock_T err)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100745{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100746 channel->CH_IN_FD = in;
747 channel->CH_OUT_FD = out;
748 channel->CH_ERR_FD = err;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100749}
750#endif
751
752 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100753channel_set_job(channel_T *channel, job_T *job)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100754{
Bram Moolenaar77073442016-02-13 23:23:53 +0100755 channel->ch_job = job;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100756}
757
758/*
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100759 * Set various properties from an "options" argument.
760 */
761 void
762channel_set_options(channel_T *channel, jobopt_T *options)
763{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100764 int part;
765
Bram Moolenaar40ea1da2016-02-19 22:33:35 +0100766 if (options->jo_set & JO_MODE)
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100767 for (part = PART_SOCK; part <= PART_IN; ++part)
768 channel->ch_part[part].ch_mode = options->jo_mode;
Bram Moolenaar40ea1da2016-02-19 22:33:35 +0100769 if (options->jo_set & JO_TIMEOUT)
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100770 for (part = PART_SOCK; part <= PART_IN; ++part)
771 channel->ch_part[part].ch_timeout = options->jo_timeout;
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100772
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100773 if (options->jo_set & JO_CALLBACK)
774 {
775 vim_free(channel->ch_callback);
776 if (options->jo_callback != NULL && *options->jo_callback != NUL)
777 channel->ch_callback = vim_strsave(options->jo_callback);
778 else
779 channel->ch_callback = NULL;
780 }
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100781}
782
783/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100784 * Set the callback for "channel"/"part" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100785 */
786 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100787channel_set_req_callback(
788 channel_T *channel,
789 int part,
790 char_u *callback,
791 int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100792{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100793 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100794 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
795
796 if (item != NULL)
797 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100798 item->cq_callback = vim_strsave(callback);
799 item->cq_seq_nr = id;
800 item->cq_prev = head->cq_prev;
801 head->cq_prev = item;
802 item->cq_next = NULL;
803 if (item->cq_prev == NULL)
804 head->cq_next = item;
805 else
806 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100807 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100808}
809
810/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100811 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100812 */
813 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100814invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100815{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100816 typval_T rettv;
817 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100818
Bram Moolenaar77073442016-02-13 23:23:53 +0100819 argv[0].v_type = VAR_CHANNEL;
820 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100821
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100822 call_func(callback, (int)STRLEN(callback),
823 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
824 /* If an echo command was used the cursor needs to be put back where
825 * it belongs. */
826 setcursor();
827 cursor_on();
828 out_flush();
829}
830
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100831/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100832 * Return the first buffer from channel "channel"/"part" and remove it.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100833 * The caller must free it.
834 * Returns NULL if there is nothing.
835 */
836 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100837channel_get(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100838{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100839 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100840 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100841 char_u *p;
842
Bram Moolenaar77073442016-02-13 23:23:53 +0100843 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100844 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100845 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100846 p = node->rq_buffer;
847 head->rq_next = node->rq_next;
848 if (node->rq_next == NULL)
849 head->rq_prev = NULL;
850 else
851 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100852 vim_free(node);
853 return p;
854}
855
856/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100857 * Returns the whole buffer contents concatenated for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100858 */
859 static char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100860channel_get_all(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100861{
862 /* Concatenate everything into one buffer.
863 * TODO: avoid multiple allocations. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100864 while (channel_collapse(channel, part) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100865 ;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100866 return channel_get(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100867}
868
869/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100870 * Collapses the first and second buffer for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100871 * Returns FAIL if that is not possible.
872 */
873 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100874channel_collapse(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100875{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100876 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100877 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100878 char_u *p;
879
Bram Moolenaar77073442016-02-13 23:23:53 +0100880 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100881 return FAIL;
882
Bram Moolenaar77073442016-02-13 23:23:53 +0100883 p = alloc((unsigned)(STRLEN(node->rq_buffer)
884 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100885 if (p == NULL)
886 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100887 STRCPY(p, node->rq_buffer);
888 STRCAT(p, node->rq_next->rq_buffer);
889 vim_free(node->rq_next->rq_buffer);
890 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100891
Bram Moolenaar77073442016-02-13 23:23:53 +0100892 /* dispose of the node and its buffer */
893 head->rq_next = node->rq_next;
894 head->rq_next->rq_prev = NULL;
895 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100896 vim_free(node);
897 return OK;
898}
899
900/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100901 * Store "buf[len]" on "channel"/"part".
902 * Returns OK or FAIL.
903 */
904 static int
905channel_save(channel_T *channel, int part, char_u *buf, int len)
906{
907 readq_T *node;
908 readq_T *head = &channel->ch_part[part].ch_head;
909 char_u *p;
910 int i;
911
912 node = (readq_T *)alloc(sizeof(readq_T));
913 if (node == NULL)
914 return FAIL; /* out of memory */
915 node->rq_buffer = alloc(len + 1);
916 if (node->rq_buffer == NULL)
917 {
918 vim_free(node);
919 return FAIL; /* out of memory */
920 }
921
922 if (channel->ch_part[part].ch_mode == MODE_NL)
923 {
924 /* Drop any CR before a NL. */
925 p = node->rq_buffer;
926 for (i = 0; i < len; ++i)
927 if (buf[i] != CAR || i + 1 >= len || buf[i + 1] != NL)
928 *p++ = buf[i];
929 *p = NUL;
930 }
931 else
932 {
933 mch_memmove(node->rq_buffer, buf, len);
934 node->rq_buffer[len] = NUL;
935 }
936
937 /* append node to the tail of the queue */
938 node->rq_next = NULL;
939 node->rq_prev = head->rq_prev;
940 if (head->rq_prev == NULL)
941 head->rq_next = node;
942 else
943 head->rq_prev->rq_next = node;
944 head->rq_prev = node;
945
946 if (log_fd != NULL)
947 {
948 ch_log_lead("RECV ", channel);
949 fprintf(log_fd, "'");
950 if (fwrite(buf, len, 1, log_fd) != 1)
951 return FAIL;
952 fprintf(log_fd, "'\n");
953 }
954 return OK;
955}
956
957/*
958 * Use the read buffer of "channel"/"part" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100959 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +0100960 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100961 */
Bram Moolenaard7ece102016-02-02 23:23:02 +0100962 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100963channel_parse_json(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100964{
965 js_read_T reader;
966 typval_T listtv;
967 jsonq_T *item;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100968 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100969 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100970
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100971 if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100972 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100973
974 /* TODO: make reader work properly */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100975 /* reader.js_buf = channel_peek(channel, part); */
976 reader.js_buf = channel_get_all(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100977 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +0100978 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100979 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +0100980 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100981 ret = json_decode(&reader, &listtv,
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100982 channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100983 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100984 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +0100985 /* Only accept the response when it is a list with at least two
986 * items. */
987 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100988 {
989 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100990 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100991 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100992 else
993 {
Bram Moolenaard7ece102016-02-02 23:23:02 +0100994 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
995 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100996 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100997 else
998 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100999 item->jq_value = alloc_tv();
1000 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001001 {
1002 vim_free(item);
1003 clear_tv(&listtv);
1004 }
1005 else
1006 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001007 *item->jq_value = listtv;
1008 item->jq_prev = head->jq_prev;
1009 head->jq_prev = item;
1010 item->jq_next = NULL;
1011 if (item->jq_prev == NULL)
1012 head->jq_next = item;
1013 else
1014 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001015 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001016 }
1017 }
1018 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001019
1020 /* Put the unread part back into the channel.
1021 * TODO: insert in front */
1022 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001023 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001024 channel_save(channel, part, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001025 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001026 ret = TRUE;
1027 }
1028 else
1029 ret = FALSE;
1030
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001031 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001032 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001033}
1034
1035/*
Bram Moolenaard46ae142016-02-16 13:33:52 +01001036 * Remove "node" from the queue that it is in. Does not free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +01001037 */
1038 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001039remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001040{
Bram Moolenaar77073442016-02-13 23:23:53 +01001041 if (node->cq_prev == NULL)
1042 head->cq_next = node->cq_next;
1043 else
1044 node->cq_prev->cq_next = node->cq_next;
1045 if (node->cq_next == NULL)
1046 head->cq_prev = node->cq_prev;
1047 else
1048 node->cq_next->cq_prev = node->cq_prev;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001049}
1050
1051/*
1052 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +01001053 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001054 */
1055 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001056remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001057{
Bram Moolenaar77073442016-02-13 23:23:53 +01001058 if (node->jq_prev == NULL)
1059 head->jq_next = node->jq_next;
1060 else
1061 node->jq_prev->jq_next = node->jq_next;
1062 if (node->jq_next == NULL)
1063 head->jq_prev = node->jq_prev;
1064 else
1065 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001066 vim_free(node);
1067}
1068
1069/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001070 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001071 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +01001072 * When "id" is zero or negative jut get the first message. But not the one
1073 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001074 * Return OK when found and return the value in "rettv".
1075 * Return FAIL otherwise.
1076 */
1077 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001078channel_get_json(channel_T *channel, int part, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001079{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001080 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001081 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001082
Bram Moolenaar77073442016-02-13 23:23:53 +01001083 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001084 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001085 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001086 typval_T *tv = &l->lv_first->li_tv;
1087
1088 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001089 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001090 || tv->vval.v_number == 0
1091 || tv->vval.v_number != channel->ch_part[part].ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001092 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001093 *rettv = item->jq_value;
1094 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001095 return OK;
1096 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001097 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001098 }
1099 return FAIL;
1100}
1101
1102/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001103 * Execute a command received over "channel"/"part"
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001104 * "cmd" is the command string, "arg2" the second argument.
1105 * "arg3" is the third argument, NULL if missing.
1106 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001107 static void
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001108channel_exe_cmd(
1109 channel_T *channel,
1110 int part,
1111 char_u *cmd,
1112 typval_T *arg2,
1113 typval_T *arg3)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001114{
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001115 char_u *arg;
1116
1117 if (arg2->v_type != VAR_STRING)
1118 {
1119 if (p_verbose > 2)
1120 EMSG("E903: received ex command with non-string argument");
1121 return;
1122 }
1123 arg = arg2->vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001124 if (arg == NULL)
1125 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001126
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001127 if (STRCMP(cmd, "ex") == 0)
1128 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001129 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001130 }
1131 else if (STRCMP(cmd, "normal") == 0)
1132 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001133 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001134
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001135 ea.arg = arg;
1136 ea.addr_count = 0;
1137 ea.forceit = TRUE; /* no mapping */
1138 ex_normal(&ea);
1139 }
1140 else if (STRCMP(cmd, "redraw") == 0)
1141 {
1142 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001143
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001144 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001145 ex_redraw(&ea);
1146 showruler(FALSE);
1147 setcursor();
1148 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001149#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001150 if (gui.in_use)
1151 {
1152 gui_update_cursor(FALSE, FALSE);
1153 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001154 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001155#endif
1156 }
1157 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "eval") == 0)
1158 {
1159 int is_eval = cmd[1] == 'v';
1160
Bram Moolenaar66624ff2016-02-03 23:59:43 +01001161 if (is_eval && (arg3 == NULL || arg3->v_type != VAR_NUMBER))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001162 {
1163 if (p_verbose > 2)
1164 EMSG("E904: third argument for eval must be a number");
1165 }
1166 else
1167 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001168 typval_T *tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001169 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +01001170 char_u *json = NULL;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001171 int options = channel->ch_part[part].ch_mode == MODE_JS
1172 ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001173
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001174 /* Don't pollute the display with errors. */
1175 ++emsg_skip;
1176 tv = eval_expr(arg, NULL);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001177 if (is_eval)
1178 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001179 if (tv != NULL)
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001180 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1181 options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001182 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001183 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001184 /* If evaluation failed or the result can't be encoded
1185 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001186 vim_free(json);
1187 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001188 err_tv.v_type = VAR_STRING;
1189 err_tv.vval.v_string = (char_u *)"ERROR";
1190 tv = &err_tv;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001191 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1192 options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001193 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001194 if (json != NULL)
1195 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001196 channel_send(channel, part, json, "eval");
Bram Moolenaar55fab432016-02-07 16:53:13 +01001197 vim_free(json);
1198 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001199 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001200 --emsg_skip;
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001201 if (tv != &err_tv)
1202 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001203 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001204 }
1205 else if (p_verbose > 2)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001206 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001207}
1208
1209/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001210 * Invoke a callback for "channel"/"part" if needed.
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001211 * Return TRUE when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001212 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001213 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001214may_invoke_callback(channel_T *channel, int part)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001215{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001216 char_u *msg = NULL;
1217 typval_T *listtv = NULL;
1218 list_T *list;
1219 typval_T *typetv;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001220 typval_T argv[3];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001221 int seq_nr = -1;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001222 ch_mode_T ch_mode = channel->ch_part[part].ch_mode;
1223 char_u *callback = NULL;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001224
Bram Moolenaara07fec92016-02-05 21:04:08 +01001225 if (channel->ch_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001226 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001227 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001228
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001229 if (channel->ch_part[part].ch_callback != NULL)
1230 callback = channel->ch_part[part].ch_callback;
1231 else
1232 callback = channel->ch_callback;
1233
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001234 if (ch_mode == MODE_JSON || ch_mode == MODE_JS)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001235 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001236 /* Get any json message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001237 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001238 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001239 /* Parse readahead, return when there is still no message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001240 channel_parse_json(channel, part);
1241 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001242 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001243 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001244
1245 list = listtv->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001246 argv[1] = list->lv_first->li_next->li_tv;
1247 typetv = &list->lv_first->li_tv;
1248 if (typetv->v_type == VAR_STRING)
1249 {
1250 typval_T *arg3 = NULL;
1251 char_u *cmd = typetv->vval.v_string;
1252
Bram Moolenaar66624ff2016-02-03 23:59:43 +01001253 /* ["cmd", arg] or ["cmd", arg, arg] */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001254 if (list->lv_len == 3)
1255 arg3 = &list->lv_last->li_tv;
Bram Moolenaar77073442016-02-13 23:23:53 +01001256 ch_logs(channel, "Executing %s command", (char *)cmd);
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001257 channel_exe_cmd(channel, part, cmd, &argv[1], arg3);
Bram Moolenaar77073442016-02-13 23:23:53 +01001258 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001259 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001260 }
1261
1262 if (typetv->v_type != VAR_NUMBER)
1263 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001264 ch_error(channel,
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001265 "Dropping message with invalid sequence number type");
Bram Moolenaar77073442016-02-13 23:23:53 +01001266 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001267 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001268 }
1269 seq_nr = typetv->vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001270 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001271 else if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001272 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001273 /* nothing to read on RAW or NL channel */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001274 return FALSE;
1275 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001276 else
1277 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001278 /* If there is no callback drop the message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001279 if (callback == NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001280 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001281 while ((msg = channel_get(channel, part)) != NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001282 vim_free(msg);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001283 return FALSE;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001284 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001285
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001286 if (ch_mode == MODE_NL)
1287 {
1288 char_u *nl;
1289 char_u *buf;
1290
1291 /* See if we have a message ending in NL in the first buffer. If
1292 * not try to concatenate the first and the second buffer. */
1293 while (TRUE)
1294 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001295 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001296 nl = vim_strchr(buf, NL);
1297 if (nl != NULL)
1298 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001299 if (channel_collapse(channel, part) == FAIL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001300 return FALSE; /* incomplete message */
1301 }
1302 if (nl[1] == NUL)
1303 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001304 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001305 else
1306 {
1307 /* Copy the message into allocated memory and remove it from
1308 * the buffer. */
1309 msg = vim_strnsave(buf, (int)(nl - buf));
1310 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1311 }
1312 }
1313 else
1314 /* For a raw channel we don't know where the message ends, just
1315 * get everything we have. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001316 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001317
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001318 argv[1].v_type = VAR_STRING;
1319 argv[1].vval.v_string = msg;
1320 }
1321
Bram Moolenaara07fec92016-02-05 21:04:08 +01001322 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001323 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001324 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001325 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001326 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001327
1328 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001329 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001330 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001331 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001332 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001333 ch_log(channel, "Invoking one-time callback");
Bram Moolenaard46ae142016-02-16 13:33:52 +01001334 /* Remove the item from the list first, if the callback
1335 * invokes ch_close() the list will be cleared. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001336 remove_cb_node(head, item);
Bram Moolenaard46ae142016-02-16 13:33:52 +01001337 invoke_callback(channel, item->cq_callback, argv);
1338 vim_free(item->cq_callback);
1339 vim_free(item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001340 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001341 break;
1342 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001343 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001344 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001345 if (!done)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001346 ch_log(channel, "Dropping message without callback");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001347 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001348 else if (callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001349 {
1350 /* invoke the channel callback */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001351 ch_log(channel, "Invoking channel callback");
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001352 invoke_callback(channel, callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001353 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001354 else
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001355 ch_log(channel, "Dropping message");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001356
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001357 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001358 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001359 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001360
1361 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001362}
1363
1364/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001365 * Return TRUE when channel "channel" is open for writing to.
1366 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001367 */
1368 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001369channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001370{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001371 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001372#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001373 || channel->CH_IN_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001374#endif
1375 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001376}
1377
1378/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001379 * Return TRUE when channel "channel" is open for reading or writing.
1380 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001381 */
1382 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001383channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001384{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001385 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001386#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001387 || channel->CH_IN_FD != INVALID_FD
1388 || channel->CH_OUT_FD != INVALID_FD
1389 || channel->CH_ERR_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001390#endif
1391 );
1392}
1393
1394/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001395 * Return a string indicating the status of the channel.
1396 */
1397 char *
1398channel_status(channel_T *channel)
1399{
1400 if (channel == NULL)
1401 return "fail";
1402 if (channel_is_open(channel))
1403 return "open";
1404 return "closed";
1405}
1406
1407/*
1408 * Close channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001409 * This does not trigger the close callback.
1410 */
1411 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001412channel_close(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001413{
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001414 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001415
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001416#ifdef FEAT_GUI
1417 channel_gui_unregister(channel);
1418#endif
1419
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001420 if (channel->CH_SOCK_FD != INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001421 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001422 sock_close(channel->CH_SOCK_FD);
1423 channel->CH_SOCK_FD = INVALID_FD;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001424 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001425#if defined(CHANNEL_PIPES)
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001426 if (channel->CH_IN_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001427 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001428 fd_close(channel->CH_IN_FD);
1429 channel->CH_IN_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001430 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001431 if (channel->CH_OUT_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001432 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001433 fd_close(channel->CH_OUT_FD);
1434 channel->CH_OUT_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001435 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001436 if (channel->CH_ERR_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001437 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001438 fd_close(channel->CH_ERR_FD);
1439 channel->CH_ERR_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001440 }
1441#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001442
1443 channel->ch_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001444 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001445}
1446
Bram Moolenaard04a0202016-01-26 23:30:18 +01001447/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001448 * Return the first buffer from "channel"/"part" without removing it.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001449 * Returns NULL if there is nothing.
1450 */
1451 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001452channel_peek(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001453{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001454 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001455
Bram Moolenaar77073442016-02-13 23:23:53 +01001456 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001457 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001458 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001459}
1460
1461/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001462 * Clear the read buffer on "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001463 */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001464 static void
1465channel_clear_one(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001466{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001467 jsonq_T *json_head = &channel->ch_part[part].ch_json_head;
1468 cbq_T *cb_head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001469
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001470 while (channel_peek(channel, part) != NULL)
1471 vim_free(channel_get(channel, part));
Bram Moolenaar77073442016-02-13 23:23:53 +01001472
1473 while (cb_head->cq_next != NULL)
Bram Moolenaard46ae142016-02-16 13:33:52 +01001474 {
1475 cbq_T *node = cb_head->cq_next;
1476
1477 remove_cb_node(cb_head, node);
1478 vim_free(node->cq_callback);
1479 vim_free(node);
1480 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001481
1482 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001483 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001484 free_tv(json_head->jq_next->jq_value);
1485 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001486 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001487
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001488 vim_free(channel->ch_part[part].ch_callback);
1489 channel->ch_part[part].ch_callback = NULL;
1490}
1491
1492/*
1493 * Clear all the read buffers on "channel".
1494 */
1495 void
1496channel_clear(channel_T *channel)
1497{
1498 channel_clear_one(channel, PART_SOCK);
1499#ifdef CHANNEL_PIPES
1500 channel_clear_one(channel, PART_OUT);
1501 channel_clear_one(channel, PART_ERR);
1502#endif
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001503 vim_free(channel->ch_callback);
1504 channel->ch_callback = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001505}
1506
Bram Moolenaar77073442016-02-13 23:23:53 +01001507#if defined(EXITFREE) || defined(PROTO)
1508 void
1509channel_free_all(void)
1510{
1511 channel_T *channel;
1512
1513 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1514 channel_clear(channel);
1515}
1516#endif
1517
1518
Bram Moolenaard04a0202016-01-26 23:30:18 +01001519/* Sent when the channel is found closed when reading. */
1520#define DETACH_MSG "\"DETACH\"\n"
1521
1522/* Buffer size for reading incoming messages. */
1523#define MAXMSGSIZE 4096
1524
1525/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001526 * Check for reading from "fd" with "timeout" msec.
1527 * Return FAIL when there is nothing to read.
Bram Moolenaara8343c12016-02-04 22:09:48 +01001528 * Always returns OK for FEAT_GUI_W32.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001529 */
1530 static int
Bram Moolenaard8070362016-02-15 21:56:54 +01001531channel_wait(channel_T *channel, sock_T fd, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001532{
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001533 if (timeout > 0)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001534 ch_logn(channel, "Waiting for up to %d msec", timeout);
Bram Moolenaard8070362016-02-15 21:56:54 +01001535
Bram Moolenaard8070362016-02-15 21:56:54 +01001536# ifdef WIN32
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001537 if (fd != channel->CH_SOCK_FD)
Bram Moolenaard8070362016-02-15 21:56:54 +01001538 {
1539 DWORD nread;
1540 int diff;
1541 DWORD deadline = GetTickCount() + timeout;
1542
1543 /* reading from a pipe, not a socket */
1544 while (TRUE)
1545 {
Bram Moolenaare74e8e72016-02-16 22:01:30 +01001546 if (PeekNamedPipe((HANDLE)fd, NULL, 0, NULL, &nread, NULL)
1547 && nread > 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001548 return OK;
1549 diff = deadline - GetTickCount();
1550 if (diff < 0)
1551 break;
1552 /* Wait for 5 msec.
1553 * TODO: increase the sleep time when looping more often */
1554 Sleep(5);
1555 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001556 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001557 else
Bram Moolenaard8070362016-02-15 21:56:54 +01001558#endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001559 {
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001560#if defined(FEAT_GUI_W32)
1561 /* Can't check socket for Win32 GUI, always return OK. */
1562 ch_log(channel, "Can't check, assuming there is something to read");
1563 return OK;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001564#else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001565# if defined(HAVE_SELECT)
1566 struct timeval tval;
1567 fd_set rfds;
1568 int ret;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001569
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001570 FD_ZERO(&rfds);
1571 FD_SET((int)fd, &rfds);
1572 tval.tv_sec = timeout / 1000;
1573 tval.tv_usec = (timeout % 1000) * 1000;
1574 for (;;)
1575 {
1576 ret = select((int)fd + 1, &rfds, NULL, NULL, &tval);
1577# ifdef EINTR
1578 SOCK_ERRNO;
1579 if (ret == -1 && errno == EINTR)
1580 continue;
1581# endif
1582 if (ret > 0)
1583 return OK;
1584 break;
1585 }
1586# else
1587 struct pollfd fds;
1588
1589 fds.fd = fd;
1590 fds.events = POLLIN;
1591 if (poll(&fds, 1, timeout) > 0)
1592 return OK;
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001593# endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001594#endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001595 }
1596 ch_log(channel, "Nothing to read");
1597 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001598}
1599
1600/*
1601 * Return a unique ID to be used in a message.
1602 */
1603 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001604channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001605{
1606 static int next_id = 1;
1607
1608 return next_id++;
1609}
1610
1611/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001612 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001613 * "part" is PART_SOCK, PART_OUT or PART_ERR.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001614 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001615 */
1616 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001617channel_read(channel_T *channel, int part, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001618{
1619 static char_u *buf = NULL;
1620 int len = 0;
1621 int readlen = 0;
Bram Moolenaard8070362016-02-15 21:56:54 +01001622 sock_T fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001623 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001624
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001625 fd = channel->ch_part[part].ch_fd;
1626 if (fd == INVALID_FD)
1627 {
1628 ch_error(channel, "channel_read() called while socket is closed");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001629 return;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001630 }
1631 use_socket = fd == channel->CH_SOCK_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001632
1633 /* Allocate a buffer to read into. */
1634 if (buf == NULL)
1635 {
1636 buf = alloc(MAXMSGSIZE);
1637 if (buf == NULL)
1638 return; /* out of memory! */
1639 }
1640
1641 /* Keep on reading for as long as there is something to read.
1642 * Use select() or poll() to avoid blocking on a message that is exactly
1643 * MAXMSGSIZE long. */
1644 for (;;)
1645 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001646 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001647 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001648 if (use_socket)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001649 len = sock_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001650 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001651 len = fd_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001652 if (len <= 0)
1653 break; /* error or nothing more to read */
1654
1655 /* Store the read message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001656 channel_save(channel, part, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001657 readlen += len;
1658 if (len < MAXMSGSIZE)
1659 break; /* did read everything that's available */
1660 }
Bram Moolenaara8343c12016-02-04 22:09:48 +01001661#ifdef FEAT_GUI_W32
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001662 if (use_socket && len == SOCKET_ERROR)
Bram Moolenaara8343c12016-02-04 22:09:48 +01001663 {
1664 /* For Win32 GUI channel_wait() always returns OK and we handle the
1665 * situation that there is nothing to read here.
1666 * TODO: how about a timeout? */
1667 if (WSAGetLastError() == WSAEWOULDBLOCK)
1668 return;
1669 }
1670#endif
Bram Moolenaard04a0202016-01-26 23:30:18 +01001671
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001672 /* Reading a socket disconnection (readlen == 0), or a socket error.
1673 * TODO: call error callback. */
1674 if (readlen <= 0 && channel->ch_job == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001675 {
1676 /* Queue a "DETACH" netbeans message in the command queue in order to
1677 * terminate the netbeans session later. Do not end the session here
1678 * directly as we may be running in the context of a call to
1679 * netbeans_parse_messages():
1680 * netbeans_parse_messages
1681 * -> autocmd triggered while processing the netbeans cmd
1682 * -> ui_breakcheck
1683 * -> gui event loop or select loop
1684 * -> channel_read()
1685 */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001686 ch_errors(channel, "%s(): Cannot read", func);
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001687 channel_save(channel, part,
1688 (char_u *)DETACH_MSG, (int)STRLEN(DETACH_MSG));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001689
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001690 /* TODO: When reading from stdout is not possible, should we try to
1691 * keep stdin and stderr open? Probably not, assume the other side
1692 * has died. */
1693 channel_close(channel);
1694 if (channel->ch_close_cb != NULL)
1695 (*channel->ch_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001696
1697 if (len < 0)
1698 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001699 ch_error(channel, "channel_read(): cannot read from channel");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001700 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001701 }
1702 }
1703
1704#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001705 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001706 if (CH_HAS_GUI && gtk_main_level() > 0)
1707 gtk_main_quit();
1708#endif
1709}
1710
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001711/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001712 * Read from RAW or NL "channel"/"part". Blocks until there is something to
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001713 * read or the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001714 * Returns what was read in allocated memory.
1715 * Returns NULL in case of error or timeout.
1716 */
1717 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001718channel_read_block(channel_T *channel, int part)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001719{
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001720 char_u *buf;
1721 char_u *msg;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001722 ch_mode_T mode = channel->ch_part[part].ch_mode;
1723 int timeout = channel->ch_part[part].ch_timeout;
1724 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001725 char_u *nl;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001726
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001727 ch_logsn(channel, "Blocking %s read, timeout: %d msec",
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001728 mode == MODE_RAW ? "RAW" : "NL", timeout);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001729
1730 while (TRUE)
1731 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001732 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001733 if (buf != NULL && (mode == MODE_RAW
1734 || (mode == MODE_NL && vim_strchr(buf, NL) != NULL)))
1735 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001736 if (buf != NULL && channel_collapse(channel, part) == OK)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001737 continue;
1738
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001739 /* Wait for up to the channel timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001740 if (fd == INVALID_FD
1741 || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001742 return NULL;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001743 channel_read(channel, part, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001744 }
1745
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001746 if (mode == MODE_RAW)
1747 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001748 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001749 }
1750 else
1751 {
1752 nl = vim_strchr(buf, NL);
1753 if (nl[1] == NUL)
1754 {
1755 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001756 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001757 *nl = NUL;
1758 }
1759 else
1760 {
1761 /* Copy the message into allocated memory and remove it from the
1762 * buffer. */
1763 msg = vim_strnsave(buf, (int)(nl - buf));
1764 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1765 }
1766 }
1767 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001768 ch_logn(channel, "Returning %d bytes", (int)STRLEN(msg));
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001769 return msg;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001770}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001771
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001772/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001773 * Read one JSON message with ID "id" from "channel"/"part" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001774 * result in "rettv".
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001775 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001776 */
1777 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001778channel_read_json_block(channel_T *channel, int part, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001779{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001780 int more;
Bram Moolenaard8070362016-02-15 21:56:54 +01001781 sock_T fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001782
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001783 ch_log(channel, "Reading JSON");
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001784 channel->ch_part[part].ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001785 for (;;)
1786 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001787 more = channel_parse_json(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001788
1789 /* search for messsage "id" */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001790 if (channel_get_json(channel, part, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001791 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001792 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001793 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001794 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001795
Bram Moolenaard7ece102016-02-02 23:23:02 +01001796 if (!more)
1797 {
1798 /* Handle any other messages in the queue. If done some more
1799 * messages may have arrived. */
1800 if (channel_parse_messages())
1801 continue;
1802
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001803 /* Wait for up to the channel timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001804 fd = channel->ch_part[part].ch_fd;
1805 if (fd == INVALID_FD || channel_wait(channel, fd,
1806 channel->ch_part[part].ch_timeout) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001807 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001808 channel_read(channel, part, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001809 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001810 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001811 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001812 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001813}
1814
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001815# if defined(WIN32) || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001816/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001817 * Lookup the channel from the socket. Set "part" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001818 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001819 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001820 channel_T *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001821channel_fd2channel(sock_T fd, int *part)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001822{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001823 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001824 int part;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001825
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001826 if (fd != INVALID_FD)
Bram Moolenaar77073442016-02-13 23:23:53 +01001827 for (channel = first_channel; channel != NULL;
1828 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001829 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001830# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001831 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001832# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001833 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001834# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001835 if (channel->ch_part[part].ch_fd == fd)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001836 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001837 *part = part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001838 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001839 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001840 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001841 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001842}
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001843
1844 void
1845channel_handle_events(void)
1846{
1847 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001848 int part;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001849 static int loop = 0;
1850
1851 /* Skip heavily polling */
1852 if (loop++ % 2)
1853 return;
1854
1855 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1856 {
1857# ifdef FEAT_GUI_W32
1858 /* only check the pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001859 for (part = PART_OUT; part <= PART_ERR; ++part)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001860# else
1861# ifdef CHANNEL_PIPES
1862 /* check the socket and pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001863 for (part = PART_SOCK; part <= PART_ERR; ++part)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001864# else
1865 /* only check the socket */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001866 part = PART_SOCK;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001867# endif
1868# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001869 channel_read(channel, part, "channel_handle_events");
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001870 }
1871}
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001872# endif
1873
Bram Moolenaard04a0202016-01-26 23:30:18 +01001874/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001875 * Write "buf" (NUL terminated string) to "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001876 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001877 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001878 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001879 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001880channel_send(channel_T *channel, int part, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001881{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001882 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001883 int res;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001884 sock_T fd;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001885
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001886 fd = channel->ch_part[part].ch_fd;
1887 if (fd == INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001888 {
1889 if (!channel->ch_error && fun != NULL)
1890 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001891 ch_errors(channel, "%s(): write while not connected", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001892 EMSG2("E630: %s(): write while not connected", fun);
1893 }
1894 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001895 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001896 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001897
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001898 if (log_fd != NULL)
1899 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001900 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001901 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001902 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001903 fprintf(log_fd, "'\n");
1904 fflush(log_fd);
1905 }
1906
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001907 if (part == PART_SOCK)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001908 res = sock_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001909 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001910 res = fd_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001911 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001912 {
1913 if (!channel->ch_error && fun != NULL)
1914 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001915 ch_errors(channel, "%s(): write failed", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001916 EMSG2("E631: %s(): write failed", fun);
1917 }
1918 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001919 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001920 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001921
1922 channel->ch_error = FALSE;
1923 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001924}
1925
1926# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001927/*
1928 * Add open channels to the poll struct.
1929 * Return the adjusted struct index.
1930 * The type of "fds" is hidden to avoid problems with the function proto.
1931 */
1932 int
1933channel_poll_setup(int nfd_in, void *fds_in)
1934{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001935 int nfd = nfd_in;
1936 channel_T *channel;
1937 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001938 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001939
Bram Moolenaar77073442016-02-13 23:23:53 +01001940 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001941 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001942# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001943 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001944# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001945 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001946# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001947 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001948 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001949 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001950 channel->ch_part[part].ch_poll_idx = nfd;
1951 fds[nfd].fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001952 fds[nfd].events = POLLIN;
1953 nfd++;
1954 }
1955 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001956 channel->ch_part[part].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001957 }
1958 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001959
1960 return nfd;
1961}
1962
1963/*
1964 * The type of "fds" is hidden to avoid problems with the function proto.
1965 */
1966 int
1967channel_poll_check(int ret_in, void *fds_in)
1968{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001969 int ret = ret_in;
1970 channel_T *channel;
1971 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001972 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001973
Bram Moolenaar77073442016-02-13 23:23:53 +01001974 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001975 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001976# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001977 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001978# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001979 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001980# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001981 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001982 int idx = channel->ch_part[part].ch_poll_idx;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001983
1984 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
1985 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001986 channel_read(channel, part, "channel_poll_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001987 --ret;
1988 }
1989 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001990 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001991
1992 return ret;
1993}
Bram Moolenaard04a0202016-01-26 23:30:18 +01001994# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01001995
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001996# if (!defined(WIN32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001997/*
1998 * The type of "rfds" is hidden to avoid problems with the function proto.
1999 */
2000 int
2001channel_select_setup(int maxfd_in, void *rfds_in)
2002{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002003 int maxfd = maxfd_in;
2004 channel_T *channel;
2005 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002006 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002007
Bram Moolenaar77073442016-02-13 23:23:53 +01002008 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002009 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002010# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002011 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002012# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002013 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002014# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002015 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002016 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002017
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002018 if (fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002019 {
Bram Moolenaard8070362016-02-15 21:56:54 +01002020 FD_SET((int)fd, rfds);
2021 if (maxfd < (int)fd)
2022 maxfd = (int)fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002023 }
2024 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002025 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002026
2027 return maxfd;
2028}
2029
2030/*
2031 * The type of "rfds" is hidden to avoid problems with the function proto.
2032 */
2033 int
2034channel_select_check(int ret_in, void *rfds_in)
2035{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002036 int ret = ret_in;
2037 channel_T *channel;
2038 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002039 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002040
Bram Moolenaar77073442016-02-13 23:23:53 +01002041 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002042 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002043# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002044 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002045# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002046 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002047# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002048 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002049 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002050
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002051 if (ret > 0 && fd != INVALID_FD && FD_ISSET(fd, rfds))
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002052 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002053 channel_read(channel, part, "channel_select_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002054 --ret;
2055 }
2056 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002057 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002058
2059 return ret;
2060}
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002061# endif /* !WIN32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002062
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002063/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01002064 * Execute queued up commands.
2065 * Invoked from the main loop when it's safe to execute received commands.
2066 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002067 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01002068 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002069channel_parse_messages(void)
2070{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002071 channel_T *channel = first_channel;
2072 int ret = FALSE;
2073 int r;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002074 int part = PART_SOCK;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002075
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002076 while (channel != NULL)
2077 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002078 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaard7ece102016-02-02 23:23:02 +01002079 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002080 /* Increase the refcount, in case the handler causes the channel
2081 * to be unreferenced or closed. */
2082 ++channel->ch_refcount;
2083 r = may_invoke_callback(channel, part);
2084 if (r == OK)
2085 ret = TRUE;
2086 if (channel_unref(channel) || r == OK)
2087 {
2088 /* channel was freed or something was done, start over */
2089 channel = first_channel;
2090 part = PART_SOCK;
2091 continue;
2092 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002093 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002094#ifdef CHANNEL_PIPES
2095 if (part < PART_ERR)
2096 ++part;
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002097 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002098#endif
2099 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002100 channel = channel->ch_next;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002101 part = PART_SOCK;
2102 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002103 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002104 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002105}
2106
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01002107/*
2108 * Mark references to lists used in channels.
2109 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002110 int
2111set_ref_in_channel(int copyID)
2112{
Bram Moolenaar77073442016-02-13 23:23:53 +01002113 int abort = FALSE;
2114 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002115 int part;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002116
Bram Moolenaar77073442016-02-13 23:23:53 +01002117 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002118 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002119#ifdef CHANNEL_PIPES
2120 for (part = PART_SOCK; part < PART_IN; ++part)
2121#else
2122 part = PART_SOCK;
2123#endif
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002124 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002125 jsonq_T *head = &channel->ch_part[part].ch_json_head;
2126 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002127
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002128 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002129 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002130 list_T *l = item->jq_value->vval.v_list;
2131
2132 if (l->lv_copyID != copyID)
2133 {
2134 l->lv_copyID = copyID;
2135 abort = abort || set_ref_in_list(l, copyID, NULL);
2136 }
2137 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002138 }
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002139 }
2140 }
2141 return abort;
2142}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002143
2144/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002145 * Return the "part" to write to for "channel".
2146 */
2147 int
2148channel_part_send(channel_T *channel)
2149{
2150#ifdef CHANNEL_PIPES
2151 if (channel->CH_SOCK_FD == INVALID_FD)
2152 return PART_IN;
2153#endif
2154 return PART_SOCK;
2155}
2156
2157/*
2158 * Return the default "part" to read from for "channel".
2159 */
2160 int
2161channel_part_read(channel_T *channel)
2162{
2163#ifdef CHANNEL_PIPES
2164 if (channel->CH_SOCK_FD == INVALID_FD)
2165 return PART_OUT;
2166#endif
2167 return PART_SOCK;
2168}
2169
2170/*
2171 * Return the mode of "channel"/"part"
Bram Moolenaar77073442016-02-13 23:23:53 +01002172 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002173 */
2174 ch_mode_T
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002175channel_get_mode(channel_T *channel, int part)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002176{
Bram Moolenaar77073442016-02-13 23:23:53 +01002177 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002178 return MODE_JSON;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002179 return channel->ch_part[part].ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002180}
2181
Bram Moolenaare0874f82016-01-24 20:36:41 +01002182#endif /* FEAT_CHANNEL */