blob: daf1c34caf6efe9a315e6b4f525f88dd1ad91b10 [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
35# define sock_write(sd, buf, len) send(sd, buf, len, 0)
36# define sock_read(sd, buf, len) recv(sd, buf, len, 0)
37# define sock_close(sd) closesocket(sd)
38# define sleep(t) Sleep(t*1000) /* WinAPI Sleep() accepts milliseconds */
39#else
40# include <netdb.h>
41# include <netinet/in.h>
42
43# include <sys/socket.h>
44# ifdef HAVE_LIBGEN_H
45# include <libgen.h>
46# endif
47# define SOCK_ERRNO
48# define sock_write(sd, buf, len) write(sd, buf, len)
49# define sock_read(sd, buf, len) read(sd, buf, len)
50# define sock_close(sd) close(sd)
51#endif
52
53#ifdef FEAT_GUI_W32
54extern HWND s_hwnd; /* Gvim's Window handle */
55#endif
56
Bram Moolenaare0874f82016-01-24 20:36:41 +010057
Bram Moolenaar6463ca22016-02-13 17:04:46 +010058/* Log file opened with ch_logfile(). */
59static FILE *log_fd = NULL;
60
61 void
62ch_logfile(FILE *file)
63{
64 if (log_fd != NULL)
65 fclose(log_fd);
66 log_fd = file;
67 if (log_fd != NULL)
68 fprintf(log_fd, "==== start log session ====\n");
69}
70
71 static void
Bram Moolenaar77073442016-02-13 23:23:53 +010072ch_log_lead(char *what, channel_T *ch)
Bram Moolenaar6463ca22016-02-13 17:04:46 +010073{
74 if (log_fd != NULL)
75 {
Bram Moolenaar77073442016-02-13 23:23:53 +010076 if (ch != NULL)
77 fprintf(log_fd, "%son %d: ", what, ch->ch_id);
Bram Moolenaar6463ca22016-02-13 17:04:46 +010078 else
79 fprintf(log_fd, "%s: ", what);
80 }
81}
82
83 static void
Bram Moolenaar77073442016-02-13 23:23:53 +010084ch_log(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +010085{
86 if (log_fd != NULL)
87 {
Bram Moolenaar77073442016-02-13 23:23:53 +010088 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +010089 fputs(msg, log_fd);
90 fflush(log_fd);
91 }
92}
93
94 static void
Bram Moolenaar77073442016-02-13 23:23:53 +010095ch_logn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +010096{
97 if (log_fd != NULL)
98 {
Bram Moolenaar77073442016-02-13 23:23:53 +010099 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100100 fprintf(log_fd, msg, nr);
101 fflush(log_fd);
102 }
103}
104
105 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100106ch_logs(channel_T *ch, char *msg, char *name)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100107{
108 if (log_fd != NULL)
109 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100110 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100111 fprintf(log_fd, msg, name);
112 fflush(log_fd);
113 }
114}
115
116 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100117ch_logsn(channel_T *ch, char *msg, char *name, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100118{
119 if (log_fd != NULL)
120 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100121 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100122 fprintf(log_fd, msg, name, nr);
123 fflush(log_fd);
124 }
125}
126
127 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100128ch_error(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100129{
130 if (log_fd != NULL)
131 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100132 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100133 fputs(msg, log_fd);
134 fflush(log_fd);
135 }
136}
137
138 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100139ch_errorn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100140{
141 if (log_fd != NULL)
142 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100143 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100144 fprintf(log_fd, msg, nr);
145 fflush(log_fd);
146 }
147}
148
149 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100150ch_errors(channel_T *ch, char *msg, char *arg)
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("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100155 fprintf(log_fd, msg, arg);
156 fflush(log_fd);
157 }
158}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100159
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100160#ifdef _WIN32
161# undef PERROR
162# define PERROR(msg) (void)emsg3((char_u *)"%s: %s", \
163 (char_u *)msg, (char_u *)strerror_win32(errno))
164
165 static char *
166strerror_win32(int eno)
167{
168 static LPVOID msgbuf = NULL;
169 char_u *ptr;
170
171 if (msgbuf)
172 LocalFree(msgbuf);
173 FormatMessage(
174 FORMAT_MESSAGE_ALLOCATE_BUFFER |
175 FORMAT_MESSAGE_FROM_SYSTEM |
176 FORMAT_MESSAGE_IGNORE_INSERTS,
177 NULL,
178 eno,
179 MAKELANGID(LANG_ENGLISH, SUBLANG_DEFAULT),
180 (LPTSTR) &msgbuf,
181 0,
182 NULL);
183 /* chomp \r or \n */
184 for (ptr = (char_u *)msgbuf; *ptr; ptr++)
185 switch (*ptr)
186 {
187 case '\r':
188 STRMOVE(ptr, ptr + 1);
189 ptr--;
190 break;
191 case '\n':
192 if (*(ptr + 1) == '\0')
193 *ptr = '\0';
194 else
195 *ptr = ' ';
196 break;
197 }
198 return msgbuf;
199}
200#endif
201
Bram Moolenaar77073442016-02-13 23:23:53 +0100202/*
203 * The list of all allocated channels.
204 */
205static channel_T *first_channel = NULL;
206static int next_ch_id = 0;
207
208/*
209 * Allocate a new channel. The refcount is set to 1.
210 * The channel isn't actually used until it is opened.
211 * Returns NULL if out of memory.
212 */
213 channel_T *
214add_channel(void)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100215{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100216 int which;
217 channel_T *channel = (channel_T *)alloc_clear((int)sizeof(channel_T));
Bram Moolenaare0874f82016-01-24 20:36:41 +0100218
Bram Moolenaar77073442016-02-13 23:23:53 +0100219 if (channel == NULL)
220 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100221
Bram Moolenaar77073442016-02-13 23:23:53 +0100222 channel->ch_id = next_ch_id++;
223 ch_log(channel, "Opening channel\n");
224
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100225#ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100226 for (which = CHAN_SOCK; which <= CHAN_IN; ++which)
227#else
228 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100229#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100230 {
231 channel->ch_pfd[which].ch_fd = (sock_T)-1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100232#ifdef FEAT_GUI_X11
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100233 channel->ch_pfd[which].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100234#endif
235#ifdef FEAT_GUI_GTK
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100236 channel->ch_pfd[which].ch_inputHandler = 0;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100237#endif
238#ifdef FEAT_GUI_W32
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100239 channel->ch_pfd[which].ch_inputHandler = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100240#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100241 }
Bram Moolenaare0874f82016-01-24 20:36:41 +0100242
Bram Moolenaar77073442016-02-13 23:23:53 +0100243 channel->ch_timeout = 2000;
244
245 if (first_channel != NULL)
246 {
247 first_channel->ch_prev = channel;
248 channel->ch_next = first_channel;
249 }
250 first_channel = channel;
251
252 channel->ch_refcount = 1;
253 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100254}
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100255
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100256/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100257 * Close a channel and free all its resources.
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100258 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100259 void
260channel_free(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100261{
Bram Moolenaar77073442016-02-13 23:23:53 +0100262 channel_close(channel);
263 if (channel->ch_next != NULL)
264 channel->ch_next->ch_prev = channel->ch_prev;
265 if (channel->ch_prev == NULL)
266 first_channel = channel->ch_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100267 else
Bram Moolenaar77073442016-02-13 23:23:53 +0100268 channel->ch_prev->ch_next = channel->ch_next;
269 vim_free(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100270}
271
Bram Moolenaar77073442016-02-13 23:23:53 +0100272#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
273 static channel_T *
274channel_from_id(int id)
275{
276 channel_T *channel;
277
278 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
279 if (channel->ch_id == id)
280 return channel;
281 return NULL;
282}
283#endif
284
Bram Moolenaard04a0202016-01-26 23:30:18 +0100285#if defined(FEAT_GUI) || defined(PROTO)
Bram Moolenaar77073442016-02-13 23:23:53 +0100286
287#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
288 static void
289channel_read_netbeans(int id)
290{
291 channel_T *channel = channel_from_id(id);
292
293 if (channel == NULL)
294 ch_errorn(NULL, "Channel %d not found", id);
295 else
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100296 channel_read(channel, -1, "messageFromNetbeans");
Bram Moolenaar77073442016-02-13 23:23:53 +0100297}
298#endif
299
Bram Moolenaare0874f82016-01-24 20:36:41 +0100300/*
Bram Moolenaard04a0202016-01-26 23:30:18 +0100301 * Read a command from netbeans.
Bram Moolenaare0874f82016-01-24 20:36:41 +0100302 */
Bram Moolenaard04a0202016-01-26 23:30:18 +0100303#ifdef FEAT_GUI_X11
304 static void
305messageFromNetbeans(XtPointer clientData,
306 int *unused1 UNUSED,
307 XtInputId *unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100308{
Bram Moolenaar77073442016-02-13 23:23:53 +0100309 channel_read_netbeans((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100310}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100311#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100312
Bram Moolenaard04a0202016-01-26 23:30:18 +0100313#ifdef FEAT_GUI_GTK
314 static void
315messageFromNetbeans(gpointer clientData,
316 gint unused1 UNUSED,
317 GdkInputCondition unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100318{
Bram Moolenaar77073442016-02-13 23:23:53 +0100319 channel_read_netbeans((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100320}
321#endif
322
323 static void
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100324channel_gui_register_one(channel_T *channel, int which)
325{
326# ifdef FEAT_GUI_X11
327 /* Tell notifier we are interested in being called
328 * when there is input on the editor connection socket. */
329 if (channel->ch_pfd[which].ch_inputHandler == (XtInputId)NULL)
330 channel->ch_pfd[which].ch_inputHandler = XtAppAddInput(
331 (XtAppContext)app_context,
332 channel->ch_pfd[which].ch_fd,
333 (XtPointer)(XtInputReadMask + XtInputExceptMask),
334 messageFromNetbeans,
335 (XtPointer)(long)channel->ch_id);
336# else
337# ifdef FEAT_GUI_GTK
338 /* Tell gdk we are interested in being called when there
339 * is input on the editor connection socket. */
340 if (channel->ch_pfd[which].ch_inputHandler == 0)
341 channel->ch_pfd[which].ch_inputHandler = gdk_input_add(
342 (gint)channel->ch_pfd[which].ch_fd, (GdkInputCondition)
343 ((int)GDK_INPUT_READ + (int)GDK_INPUT_EXCEPTION),
344 messageFromNetbeans,
345 (gpointer)(long)channel->ch_id);
346# else
347# ifdef FEAT_GUI_W32
348 /* Tell Windows we are interested in receiving message when there
349 * is input on the editor connection socket. */
350 if (channel->ch_pfd[which].ch_inputHandler == -1)
351 channel->ch_pfd[which].ch_inputHandler = WSAAsyncSelect(
352 channel->ch_pfd[which].ch_fd,
353 s_hwnd, WM_NETBEANS, FD_READ);
354# endif
355# endif
356# endif
357}
358
359 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100360channel_gui_register(channel_T *channel)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100361{
Bram Moolenaard04a0202016-01-26 23:30:18 +0100362 if (!CH_HAS_GUI)
363 return;
364
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100365 if (channel->ch_pfd[CHAN_SOCK].ch_fd >= 0)
366 channel_gui_register_one(channel, CHAN_SOCK);
367# ifdef CHANNEL_PIPES
368 if (channel->ch_pfd[CHAN_OUT].ch_fd >= 0)
369 channel_gui_register_one(channel, CHAN_OUT);
370 if (channel->ch_pfd[CHAN_ERR].ch_fd >= 0)
371 channel_gui_register_one(channel, CHAN_ERR);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100372# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100373}
374
375/*
376 * Register any of our file descriptors with the GUI event handling system.
377 * Called when the GUI has started.
378 */
379 void
380channel_gui_register_all(void)
381{
Bram Moolenaar77073442016-02-13 23:23:53 +0100382 channel_T *channel;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100383
Bram Moolenaar77073442016-02-13 23:23:53 +0100384 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100385 channel_gui_register(channel);
386}
387
388 static void
389channel_gui_unregister_one(channel_T *channel, int which)
390{
391# ifdef FEAT_GUI_X11
392 if (channel->ch_pfd[which].ch_inputHandler != (XtInputId)NULL)
393 {
394 XtRemoveInput(channel->ch_pfd[which].ch_inputHandler);
395 channel->ch_pfd[which].ch_inputHandler = (XtInputId)NULL;
396 }
397# else
398# ifdef FEAT_GUI_GTK
399 if (channel->ch_pfd[which].ch_inputHandler != 0)
400 {
401 gdk_input_remove(channel->ch_pfd[which].ch_inputHandler);
402 channel->ch_pfd[which].ch_inputHandler = 0;
403 }
404# else
405# ifdef FEAT_GUI_W32
406 if (channel->ch_pfd[which].ch_inputHandler == 0)
407 {
408 WSAAsyncSelect(channel->ch_pfd[which].ch_fd, s_hwnd, 0, 0);
409 channel->ch_pfd[which].ch_inputHandler = -1;
410 }
411# endif
412# endif
413# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100414}
415
416 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100417channel_gui_unregister(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100418{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100419 if (channel->ch_pfd[CHAN_SOCK].ch_fd >= 0)
420 channel_gui_unregister_one(channel, CHAN_SOCK);
421# ifdef CHANNEL_PIPES
422 if (channel->ch_pfd[CHAN_OUT].ch_fd >= 0)
423 channel_gui_unregister_one(channel, CHAN_OUT);
424 if (channel->ch_pfd[CHAN_ERR].ch_fd >= 0)
425 channel_gui_unregister_one(channel, CHAN_ERR);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100426# endif
427}
428
429#endif
430
431/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100432 * Open a socket channel to "hostname":"port".
Bram Moolenaar77073442016-02-13 23:23:53 +0100433 * Returns the channel for success.
434 * Returns NULL for failure.
Bram Moolenaard04a0202016-01-26 23:30:18 +0100435 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100436 channel_T *
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100437channel_open(char *hostname, int port_in, int waittime, void (*close_cb)(void))
Bram Moolenaard04a0202016-01-26 23:30:18 +0100438{
439 int sd;
440 struct sockaddr_in server;
441 struct hostent * host;
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100442#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100443 u_short port = port_in;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100444 u_long val = 1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100445#else
446 int port = port_in;
447#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100448 channel_T *channel;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100449 int ret;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100450
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100451#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100452 channel_init_winsock();
453#endif
454
Bram Moolenaar77073442016-02-13 23:23:53 +0100455 channel = add_channel();
456 if (channel == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100457 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100458 ch_error(NULL, "Cannot allocate channel.\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100459 EMSG(_("E897: All channels are in use"));
Bram Moolenaar77073442016-02-13 23:23:53 +0100460 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100461 }
462
463 if ((sd = (sock_T)socket(AF_INET, SOCK_STREAM, 0)) == (sock_T)-1)
464 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100465 ch_error(NULL, "in socket() in channel_open().\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100466 PERROR("E898: socket() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100467 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100468 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100469 }
470
471 /* Get the server internet address and put into addr structure */
472 /* fill in the socket address structure and connect to server */
473 vim_memset((char *)&server, 0, sizeof(server));
474 server.sin_family = AF_INET;
475 server.sin_port = htons(port);
476 if ((host = gethostbyname(hostname)) == NULL)
477 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100478 ch_error(NULL, "in gethostbyname() in channel_open()\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100479 PERROR("E901: gethostbyname() in channel_open()");
Bram Moolenaard04a0202016-01-26 23:30:18 +0100480 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100481 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100482 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100483 }
484 memcpy((char *)&server.sin_addr, host->h_addr, host->h_length);
485
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100486 if (waittime >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100487 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100488 /* Make connect non-blocking. */
489 if (
490#ifdef _WIN32
491 ioctlsocket(sd, FIONBIO, &val) < 0
492#else
493 fcntl(sd, F_SETFL, O_NONBLOCK) < 0
494#endif
495 )
Bram Moolenaard04a0202016-01-26 23:30:18 +0100496 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100497 SOCK_ERRNO;
Bram Moolenaar77073442016-02-13 23:23:53 +0100498 ch_errorn(NULL, "channel_open: Connect failed with errno %d\n",
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100499 errno);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100500 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100501 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100502 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100503 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100504 }
505
506 /* Try connecting to the server. */
Bram Moolenaar77073442016-02-13 23:23:53 +0100507 ch_logsn(NULL, "Connecting to %s port %d", hostname, port);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100508 ret = connect(sd, (struct sockaddr *)&server, sizeof(server));
509 SOCK_ERRNO;
510 if (ret < 0)
511 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100512 if (errno != EWOULDBLOCK
513#ifdef EINPROGRESS
514 && errno != EINPROGRESS
515#endif
516 )
Bram Moolenaard04a0202016-01-26 23:30:18 +0100517 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100518 ch_errorn(NULL, "channel_open: Connect failed with errno %d\n",
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100519 errno);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100520 PERROR(_("E902: Cannot connect to port"));
Bram Moolenaard04a0202016-01-26 23:30:18 +0100521 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100522 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100523 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100524 }
525 }
526
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100527 if (waittime >= 0 && ret < 0)
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100528 {
529 struct timeval tv;
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100530 fd_set wfds;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100531
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100532 FD_ZERO(&wfds);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100533 FD_SET(sd, &wfds);
Bram Moolenaar26dfc412016-02-06 18:18:54 +0100534 tv.tv_sec = waittime / 1000;
535 tv.tv_usec = (waittime % 1000) * 1000;
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100536 ret = select((int)sd + 1, NULL, &wfds, NULL, &tv);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100537 if (ret < 0)
538 {
539 SOCK_ERRNO;
Bram Moolenaar77073442016-02-13 23:23:53 +0100540 ch_errorn(NULL, "channel_open: Connect failed with errno %d\n",
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100541 errno);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100542 PERROR(_("E902: Cannot connect to port"));
543 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100544 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100545 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100546 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100547 if (!FD_ISSET(sd, &wfds))
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100548 {
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100549 /* don't give an error, we just timed out. */
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100550 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100551 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100552 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100553 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100554 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100555
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100556 if (waittime >= 0)
557 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100558#ifdef _WIN32
559 val = 0;
560 ioctlsocket(sd, FIONBIO, &val);
561#else
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100562 (void)fcntl(sd, F_SETFL, 0);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100563#endif
564 }
565
Bram Moolenaar0fa98e72016-02-07 22:21:19 +0100566 /* Only retry for netbeans. TODO: can we use a waittime instead? */
567 if (errno == ECONNREFUSED && close_cb != NULL)
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100568 {
569 sock_close(sd);
570 if ((sd = (sock_T)socket(AF_INET, SOCK_STREAM, 0)) == (sock_T)-1)
571 {
572 SOCK_ERRNO;
Bram Moolenaar77073442016-02-13 23:23:53 +0100573 ch_log(NULL, "socket() retry in channel_open()\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100574 PERROR("E900: socket() retry in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100575 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100576 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100577 }
578 if (connect(sd, (struct sockaddr *)&server, sizeof(server)))
579 {
580 int retries = 36;
581 int success = FALSE;
582
583 SOCK_ERRNO;
584 while (retries-- && ((errno == ECONNREFUSED)
585 || (errno == EINTR)))
586 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100587 ch_log(NULL, "retrying...\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100588 mch_delay(3000L, TRUE);
589 ui_breakcheck();
590 if (got_int)
591 {
592 errno = EINTR;
593 break;
594 }
595 if (connect(sd, (struct sockaddr *)&server,
596 sizeof(server)) == 0)
597 {
598 success = TRUE;
599 break;
600 }
601 SOCK_ERRNO;
602 }
603 if (!success)
604 {
605 /* Get here when the server can't be found. */
Bram Moolenaar77073442016-02-13 23:23:53 +0100606 ch_error(NULL, "Cannot connect to port after retry\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100607 PERROR(_("E899: Cannot connect to port after retry2"));
608 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100609 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100610 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100611 }
612 }
613 }
614
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100615 channel->CH_SOCK = sd;
Bram Moolenaar77073442016-02-13 23:23:53 +0100616 channel->ch_close_cb = close_cb;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100617
618#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +0100619 channel_gui_register(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100620#endif
621
Bram Moolenaar77073442016-02-13 23:23:53 +0100622 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100623}
624
625#if defined(CHANNEL_PIPES) || defined(PROTO)
626 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100627channel_set_pipes(channel_T *channel, int in, int out, int err)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100628{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100629 channel->CH_IN = in;
630 channel->CH_OUT = out;
631 channel->CH_ERR = err;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100632}
633#endif
634
635 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100636channel_set_job(channel_T *channel, job_T *job)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100637{
Bram Moolenaar77073442016-02-13 23:23:53 +0100638 channel->ch_job = job;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100639}
640
641/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100642 * Set the json mode of channel "channel" to "ch_mode".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100643 */
644 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100645channel_set_json_mode(channel_T *channel, ch_mode_T ch_mode)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100646{
Bram Moolenaar77073442016-02-13 23:23:53 +0100647 channel->ch_mode = ch_mode;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100648}
649
650/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100651 * Set the read timeout of channel "channel".
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100652 */
653 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100654channel_set_timeout(channel_T *channel, int timeout)
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100655{
Bram Moolenaar77073442016-02-13 23:23:53 +0100656 channel->ch_timeout = timeout;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100657}
658
659/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100660 * Set the callback for channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100661 */
662 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100663channel_set_callback(channel_T *channel, char_u *callback)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100664{
Bram Moolenaar77073442016-02-13 23:23:53 +0100665 vim_free(channel->ch_callback);
666 channel->ch_callback = vim_strsave(callback);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100667}
668
669/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100670 * Set the callback for channel "channel" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100671 */
672 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100673channel_set_req_callback(channel_T *channel, char_u *callback, int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100674{
Bram Moolenaar77073442016-02-13 23:23:53 +0100675 cbq_T *head = &channel->ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100676 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
677
678 if (item != NULL)
679 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100680 item->cq_callback = vim_strsave(callback);
681 item->cq_seq_nr = id;
682 item->cq_prev = head->cq_prev;
683 head->cq_prev = item;
684 item->cq_next = NULL;
685 if (item->cq_prev == NULL)
686 head->cq_next = item;
687 else
688 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100689 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100690}
691
692/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100693 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100694 */
695 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100696invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100697{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100698 typval_T rettv;
699 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100700
Bram Moolenaar77073442016-02-13 23:23:53 +0100701 argv[0].v_type = VAR_CHANNEL;
702 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100703
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100704 call_func(callback, (int)STRLEN(callback),
705 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
706 /* If an echo command was used the cursor needs to be put back where
707 * it belongs. */
708 setcursor();
709 cursor_on();
710 out_flush();
711}
712
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100713/*
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100714 * Return the first buffer from the channel and remove it.
715 * The caller must free it.
716 * Returns NULL if there is nothing.
717 */
718 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +0100719channel_get(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100720{
Bram Moolenaar77073442016-02-13 23:23:53 +0100721 readq_T *head = &channel->ch_head;
722 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100723 char_u *p;
724
Bram Moolenaar77073442016-02-13 23:23:53 +0100725 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100726 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100727 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100728 p = node->rq_buffer;
729 head->rq_next = node->rq_next;
730 if (node->rq_next == NULL)
731 head->rq_prev = NULL;
732 else
733 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100734 vim_free(node);
735 return p;
736}
737
738/*
739 * Returns the whole buffer contents concatenated.
740 */
741 static char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +0100742channel_get_all(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100743{
744 /* Concatenate everything into one buffer.
745 * TODO: avoid multiple allocations. */
Bram Moolenaar77073442016-02-13 23:23:53 +0100746 while (channel_collapse(channel) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100747 ;
Bram Moolenaar77073442016-02-13 23:23:53 +0100748 return channel_get(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100749}
750
751/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100752 * Collapses the first and second buffer in the channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100753 * Returns FAIL if that is not possible.
754 */
755 int
Bram Moolenaar77073442016-02-13 23:23:53 +0100756channel_collapse(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100757{
Bram Moolenaar77073442016-02-13 23:23:53 +0100758 readq_T *head = &channel->ch_head;
759 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100760 char_u *p;
761
Bram Moolenaar77073442016-02-13 23:23:53 +0100762 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100763 return FAIL;
764
Bram Moolenaar77073442016-02-13 23:23:53 +0100765 p = alloc((unsigned)(STRLEN(node->rq_buffer)
766 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100767 if (p == NULL)
768 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100769 STRCPY(p, node->rq_buffer);
770 STRCAT(p, node->rq_next->rq_buffer);
771 vim_free(node->rq_next->rq_buffer);
772 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100773
Bram Moolenaar77073442016-02-13 23:23:53 +0100774 /* dispose of the node and its buffer */
775 head->rq_next = node->rq_next;
776 head->rq_next->rq_prev = NULL;
777 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100778 vim_free(node);
779 return OK;
780}
781
782/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100783 * Use the read buffer of channel "channel" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100784 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +0100785 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100786 */
Bram Moolenaard7ece102016-02-02 23:23:02 +0100787 static int
Bram Moolenaar77073442016-02-13 23:23:53 +0100788channel_parse_json(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100789{
790 js_read_T reader;
791 typval_T listtv;
792 jsonq_T *item;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100793 jsonq_T *head = &channel->ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100794 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100795
Bram Moolenaar77073442016-02-13 23:23:53 +0100796 if (channel_peek(channel) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100797 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100798
799 /* TODO: make reader work properly */
Bram Moolenaar77073442016-02-13 23:23:53 +0100800 /* reader.js_buf = channel_peek(channel); */
801 reader.js_buf = channel_get_all(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100802 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +0100803 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100804 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +0100805 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100806 ret = json_decode(&reader, &listtv,
807 channel->ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100808 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100809 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +0100810 /* Only accept the response when it is a list with at least two
811 * items. */
812 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100813 {
814 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100815 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100816 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100817 else
818 {
Bram Moolenaard7ece102016-02-02 23:23:02 +0100819 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
820 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100821 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100822 else
823 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100824 item->jq_value = alloc_tv();
825 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100826 {
827 vim_free(item);
828 clear_tv(&listtv);
829 }
830 else
831 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100832 *item->jq_value = listtv;
833 item->jq_prev = head->jq_prev;
834 head->jq_prev = item;
835 item->jq_next = NULL;
836 if (item->jq_prev == NULL)
837 head->jq_next = item;
838 else
839 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100840 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100841 }
842 }
843 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +0100844
845 /* Put the unread part back into the channel.
846 * TODO: insert in front */
847 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100848 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100849 channel_save(channel, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +0100850 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100851 ret = TRUE;
852 }
853 else
854 ret = FALSE;
855
Bram Moolenaardf5b27b2016-02-02 18:43:17 +0100856 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100857 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100858}
859
860/*
861 * Remove "node" from the queue that it is in and free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +0100862 * Also frees the contained callback name.
863 */
864 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100865remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +0100866{
Bram Moolenaar77073442016-02-13 23:23:53 +0100867 if (node->cq_prev == NULL)
868 head->cq_next = node->cq_next;
869 else
870 node->cq_prev->cq_next = node->cq_next;
871 if (node->cq_next == NULL)
872 head->cq_prev = node->cq_prev;
873 else
874 node->cq_next->cq_prev = node->cq_prev;
875 vim_free(node->cq_callback);
Bram Moolenaara07fec92016-02-05 21:04:08 +0100876 vim_free(node);
877}
878
879/*
880 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +0100881 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100882 */
883 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100884remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100885{
Bram Moolenaar77073442016-02-13 23:23:53 +0100886 if (node->jq_prev == NULL)
887 head->jq_next = node->jq_next;
888 else
889 node->jq_prev->jq_next = node->jq_next;
890 if (node->jq_next == NULL)
891 head->jq_prev = node->jq_prev;
892 else
893 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100894 vim_free(node);
895}
896
897/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100898 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100899 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +0100900 * When "id" is zero or negative jut get the first message. But not the one
901 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100902 * Return OK when found and return the value in "rettv".
903 * Return FAIL otherwise.
904 */
905 static int
Bram Moolenaar77073442016-02-13 23:23:53 +0100906channel_get_json(channel_T *channel, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100907{
Bram Moolenaare56bf152016-02-08 23:23:42 +0100908 jsonq_T *head = &channel->ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100909 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100910
Bram Moolenaar77073442016-02-13 23:23:53 +0100911 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100912 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100913 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100914 typval_T *tv = &l->lv_first->li_tv;
915
916 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +0100917 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaarf6157282016-02-10 21:07:14 +0100918 || tv->vval.v_number == 0
Bram Moolenaare56bf152016-02-08 23:23:42 +0100919 || tv->vval.v_number != channel->ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100920 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100921 *rettv = item->jq_value;
922 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100923 return OK;
924 }
Bram Moolenaar77073442016-02-13 23:23:53 +0100925 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100926 }
927 return FAIL;
928}
929
930/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100931 * Execute a command received over channel "channel".
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100932 * "cmd" is the command string, "arg2" the second argument.
933 * "arg3" is the third argument, NULL if missing.
934 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100935 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100936channel_exe_cmd(channel_T *channel, char_u *cmd, typval_T *arg2, typval_T *arg3)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100937{
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100938 char_u *arg;
939
940 if (arg2->v_type != VAR_STRING)
941 {
942 if (p_verbose > 2)
943 EMSG("E903: received ex command with non-string argument");
944 return;
945 }
946 arg = arg2->vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +0100947 if (arg == NULL)
948 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100949
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100950 if (STRCMP(cmd, "ex") == 0)
951 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100952 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100953 }
954 else if (STRCMP(cmd, "normal") == 0)
955 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100956 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100957
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100958 ea.arg = arg;
959 ea.addr_count = 0;
960 ea.forceit = TRUE; /* no mapping */
961 ex_normal(&ea);
962 }
963 else if (STRCMP(cmd, "redraw") == 0)
964 {
965 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100966
Bram Moolenaar14ad6112016-02-01 21:47:13 +0100967 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100968 ex_redraw(&ea);
969 showruler(FALSE);
970 setcursor();
971 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100972#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100973 if (gui.in_use)
974 {
975 gui_update_cursor(FALSE, FALSE);
976 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100977 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100978#endif
979 }
980 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "eval") == 0)
981 {
982 int is_eval = cmd[1] == 'v';
983
Bram Moolenaar66624ff2016-02-03 23:59:43 +0100984 if (is_eval && (arg3 == NULL || arg3->v_type != VAR_NUMBER))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100985 {
986 if (p_verbose > 2)
987 EMSG("E904: third argument for eval must be a number");
988 }
989 else
990 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +0100991 typval_T *tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100992 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +0100993 char_u *json = NULL;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100994 int options = channel->ch_mode == MODE_JS ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100995
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +0100996 /* Don't pollute the display with errors. */
997 ++emsg_skip;
998 tv = eval_expr(arg, NULL);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100999 if (is_eval)
1000 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001001 if (tv != NULL)
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001002 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1003 options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001004 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001005 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001006 /* If evaluation failed or the result can't be encoded
1007 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001008 vim_free(json);
1009 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001010 err_tv.v_type = VAR_STRING;
1011 err_tv.vval.v_string = (char_u *)"ERROR";
1012 tv = &err_tv;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001013 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1014 options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001015 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001016 if (json != NULL)
1017 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001018 channel_send(channel, json, "eval");
Bram Moolenaar55fab432016-02-07 16:53:13 +01001019 vim_free(json);
1020 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001021 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001022 --emsg_skip;
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001023 if (tv != &err_tv)
1024 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001025 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001026 }
1027 else if (p_verbose > 2)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001028 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001029}
1030
1031/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001032 * Invoke a callback for channel "channel" if needed.
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001033 * Return OK when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001034 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001035 static int
Bram Moolenaar77073442016-02-13 23:23:53 +01001036may_invoke_callback(channel_T *channel)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001037{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001038 char_u *msg = NULL;
1039 typval_T *listtv = NULL;
1040 list_T *list;
1041 typval_T *typetv;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001042 typval_T argv[3];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001043 int seq_nr = -1;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001044 ch_mode_T ch_mode = channel->ch_mode;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001045
Bram Moolenaara07fec92016-02-05 21:04:08 +01001046 if (channel->ch_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001047 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001048 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001049
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001050 if (ch_mode != MODE_RAW)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001051 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001052 /* Get any json message in the queue. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001053 if (channel_get_json(channel, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001054 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001055 /* Parse readahead, return when there is still no message. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001056 channel_parse_json(channel);
1057 if (channel_get_json(channel, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001058 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001059 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001060
1061 list = listtv->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001062 argv[1] = list->lv_first->li_next->li_tv;
1063 typetv = &list->lv_first->li_tv;
1064 if (typetv->v_type == VAR_STRING)
1065 {
1066 typval_T *arg3 = NULL;
1067 char_u *cmd = typetv->vval.v_string;
1068
Bram Moolenaar66624ff2016-02-03 23:59:43 +01001069 /* ["cmd", arg] or ["cmd", arg, arg] */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001070 if (list->lv_len == 3)
1071 arg3 = &list->lv_last->li_tv;
Bram Moolenaar77073442016-02-13 23:23:53 +01001072 ch_logs(channel, "Executing %s command", (char *)cmd);
1073 channel_exe_cmd(channel, cmd, &argv[1], arg3);
1074 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001075 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001076 }
1077
1078 if (typetv->v_type != VAR_NUMBER)
1079 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001080 ch_error(channel,
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001081 "Dropping message with invalid sequence number type\n");
Bram Moolenaar77073442016-02-13 23:23:53 +01001082 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001083 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001084 }
1085 seq_nr = typetv->vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001086 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001087 else if (channel_peek(channel) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001088 {
1089 /* nothing to read on raw channel */
1090 return FALSE;
1091 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001092 else
1093 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001094 /* If there is no callback, don't do anything. */
1095 if (channel->ch_callback == NULL)
1096 return FALSE;
1097
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001098 /* For a raw channel we don't know where the message ends, just get
1099 * everything. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001100 msg = channel_get_all(channel);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001101 argv[1].v_type = VAR_STRING;
1102 argv[1].vval.v_string = msg;
1103 }
1104
Bram Moolenaara07fec92016-02-05 21:04:08 +01001105 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001106 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001107 cbq_T *head = &channel->ch_cb_head;
1108 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001109 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001110
1111 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001112 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001113 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001114 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001115 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001116 ch_log(channel, "Invoking one-time callback\n");
1117 invoke_callback(channel, item->cq_callback, argv);
1118 remove_cb_node(head, item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001119 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001120 break;
1121 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001122 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001123 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001124 if (!done)
Bram Moolenaar77073442016-02-13 23:23:53 +01001125 ch_log(channel, "Dropping message without callback\n");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001126 }
Bram Moolenaara07fec92016-02-05 21:04:08 +01001127 else if (channel->ch_callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001128 {
1129 /* invoke the channel callback */
Bram Moolenaar77073442016-02-13 23:23:53 +01001130 ch_log(channel, "Invoking channel callback\n");
1131 invoke_callback(channel, channel->ch_callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001132 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001133 else
Bram Moolenaar77073442016-02-13 23:23:53 +01001134 ch_log(channel, "Dropping message\n");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001135
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001136 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001137 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001138 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001139
1140 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001141}
1142
1143/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001144 * Return TRUE when channel "channel" is open for writing to.
1145 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001146 */
1147 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001148channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001149{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001150 return channel != NULL && (channel->CH_SOCK >= 0
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001151#ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001152 || channel->CH_IN >= 0
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001153#endif
1154 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001155}
1156
1157/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001158 * Return TRUE when channel "channel" is open for reading or writing.
1159 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001160 */
1161 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001162channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001163{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001164 return channel != NULL && (channel->CH_SOCK >= 0
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001165#ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001166 || channel->CH_IN >= 0
1167 || channel->CH_OUT >= 0
1168 || channel->CH_ERR >= 0
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001169#endif
1170 );
1171}
1172
1173/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001174 * Return a string indicating the status of the channel.
1175 */
1176 char *
1177channel_status(channel_T *channel)
1178{
1179 if (channel == NULL)
1180 return "fail";
1181 if (channel_is_open(channel))
1182 return "open";
1183 return "closed";
1184}
1185
1186/*
1187 * Close channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001188 * This does not trigger the close callback.
1189 */
1190 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001191channel_close(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001192{
Bram Moolenaar77073442016-02-13 23:23:53 +01001193 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001194
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001195 if (channel->CH_SOCK >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001196 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001197 sock_close(channel->CH_SOCK);
1198 channel->CH_SOCK = -1;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001199 channel->ch_close_cb = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001200#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +01001201 channel_gui_unregister(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001202#endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001203 vim_free(channel->ch_callback);
1204 channel->ch_callback = NULL;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001205 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001206#if defined(CHANNEL_PIPES)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001207 if (channel->CH_IN >= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001208 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001209 close(channel->CH_IN);
1210 channel->CH_IN = -1;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001211 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001212 if (channel->CH_OUT >= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001213 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001214 close(channel->CH_OUT);
1215 channel->CH_OUT = -1;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001216 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001217 if (channel->CH_ERR >= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001218 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001219 close(channel->CH_ERR);
1220 channel->CH_ERR = -1;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001221 }
1222#endif
Bram Moolenaar77073442016-02-13 23:23:53 +01001223 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001224}
1225
Bram Moolenaard04a0202016-01-26 23:30:18 +01001226/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001227 * Store "buf[len]" on channel "channel".
Bram Moolenaar83162462016-01-28 23:10:07 +01001228 * Returns OK or FAIL.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001229 */
Bram Moolenaar83162462016-01-28 23:10:07 +01001230 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001231channel_save(channel_T *channel, char_u *buf, int len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001232{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001233 readq_T *node;
Bram Moolenaar77073442016-02-13 23:23:53 +01001234 readq_T *head = &channel->ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001235
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001236 node = (readq_T *)alloc(sizeof(readq_T));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001237 if (node == NULL)
Bram Moolenaar83162462016-01-28 23:10:07 +01001238 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +01001239 node->rq_buffer = alloc(len + 1);
1240 if (node->rq_buffer == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001241 {
1242 vim_free(node);
Bram Moolenaar83162462016-01-28 23:10:07 +01001243 return FAIL; /* out of memory */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001244 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001245 mch_memmove(node->rq_buffer, buf, (size_t)len);
1246 node->rq_buffer[len] = NUL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001247
Bram Moolenaar77073442016-02-13 23:23:53 +01001248 /* append node to the tail of the queue */
1249 node->rq_next = NULL;
1250 node->rq_prev = head->rq_prev;
1251 if (head->rq_prev == NULL)
1252 head->rq_next = node;
1253 else
1254 head->rq_prev->rq_next = node;
1255 head->rq_prev = node;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001256
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001257 if (log_fd != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001258 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001259 ch_log_lead("RECV ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001260 fprintf(log_fd, "'");
1261 if (fwrite(buf, len, 1, log_fd) != 1)
Bram Moolenaar83162462016-01-28 23:10:07 +01001262 return FAIL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001263 fprintf(log_fd, "'\n");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001264 }
Bram Moolenaar83162462016-01-28 23:10:07 +01001265 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001266}
1267
1268/*
1269 * Return the first buffer from the channel without removing it.
1270 * Returns NULL if there is nothing.
1271 */
1272 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +01001273channel_peek(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001274{
Bram Moolenaar77073442016-02-13 23:23:53 +01001275 readq_T *head = &channel->ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001276
Bram Moolenaar77073442016-02-13 23:23:53 +01001277 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001278 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001279 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001280}
1281
1282/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001283 * Clear the read buffer on channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001284 */
1285 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001286channel_clear(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001287{
Bram Moolenaar77073442016-02-13 23:23:53 +01001288 jsonq_T *json_head = &channel->ch_json_head;
1289 cbq_T *cb_head = &channel->ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001290
Bram Moolenaar77073442016-02-13 23:23:53 +01001291 while (channel_peek(channel) != NULL)
1292 vim_free(channel_get(channel));
1293
1294 while (cb_head->cq_next != NULL)
1295 remove_cb_node(cb_head, cb_head->cq_next);
1296
1297 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001298 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001299 free_tv(json_head->jq_next->jq_value);
1300 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001301 }
1302}
1303
Bram Moolenaar77073442016-02-13 23:23:53 +01001304#if defined(EXITFREE) || defined(PROTO)
1305 void
1306channel_free_all(void)
1307{
1308 channel_T *channel;
1309
1310 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1311 channel_clear(channel);
1312}
1313#endif
1314
1315
Bram Moolenaard04a0202016-01-26 23:30:18 +01001316/* Sent when the channel is found closed when reading. */
1317#define DETACH_MSG "\"DETACH\"\n"
1318
1319/* Buffer size for reading incoming messages. */
1320#define MAXMSGSIZE 4096
1321
1322/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001323 * Check for reading from "fd" with "timeout" msec.
1324 * Return FAIL when there is nothing to read.
Bram Moolenaara8343c12016-02-04 22:09:48 +01001325 * Always returns OK for FEAT_GUI_W32.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001326 */
1327 static int
Bram Moolenaar77073442016-02-13 23:23:53 +01001328channel_wait(channel_T *channel, int fd, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001329{
Bram Moolenaara8343c12016-02-04 22:09:48 +01001330#if defined(HAVE_SELECT) && !defined(FEAT_GUI_W32)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001331 struct timeval tval;
1332 fd_set rfds;
1333 int ret;
1334
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001335 if (timeout > 0)
Bram Moolenaar77073442016-02-13 23:23:53 +01001336 ch_logn(channel, "Waiting for %d msec\n", timeout);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001337 FD_ZERO(&rfds);
1338 FD_SET(fd, &rfds);
1339 tval.tv_sec = timeout / 1000;
1340 tval.tv_usec = (timeout % 1000) * 1000;
1341 for (;;)
1342 {
1343 ret = select(fd + 1, &rfds, NULL, NULL, &tval);
1344# ifdef EINTR
1345 if (ret == -1 && errno == EINTR)
1346 continue;
1347# endif
1348 if (ret <= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001349 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001350 ch_log(channel, "Nothing to read\n");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001351 return FAIL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001352 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001353 break;
1354 }
1355#else
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001356# ifdef HAVE_POLL
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001357 struct pollfd fds;
1358
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001359 if (timeout > 0)
Bram Moolenaar77073442016-02-13 23:23:53 +01001360 ch_logn(channel, "Waiting for %d msec\n", timeout);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001361 fds.fd = fd;
1362 fds.events = POLLIN;
1363 if (poll(&fds, 1, timeout) <= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001364 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001365 ch_log(channel, "Nothing to read\n");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001366 return FAIL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001367 }
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001368# endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001369#endif
1370 return OK;
1371}
1372
1373/*
1374 * Return a unique ID to be used in a message.
1375 */
1376 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001377channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001378{
1379 static int next_id = 1;
1380
1381 return next_id++;
1382}
1383
1384/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001385 * Get the file descriptor to read from, either the socket or stdout.
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001386 * TODO: never gets stderr.
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001387 */
1388 static int
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001389get_read_fd(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001390{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001391 if (channel->CH_SOCK >= 0)
1392 return channel->CH_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001393#if defined(CHANNEL_PIPES)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001394 if (channel->CH_OUT >= 0)
1395 return channel->CH_OUT;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001396#endif
Bram Moolenaar77073442016-02-13 23:23:53 +01001397 ch_error(channel, "channel_read() called while socket is closed\n");
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001398 return -1;
1399}
1400
1401/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001402 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001403 * "which" is CHAN_SOCK, CHAN_OUT or CHAN_ERR. When -1 guess.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001404 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001405 */
1406 void
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001407channel_read(channel_T *channel, int which, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001408{
1409 static char_u *buf = NULL;
1410 int len = 0;
1411 int readlen = 0;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001412 int fd;
1413 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001414
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001415 if (which < 0)
1416 fd = get_read_fd(channel);
1417 else
1418 fd = channel->ch_pfd[which].ch_fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001419 if (fd < 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001420 return;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001421 use_socket = fd == channel->CH_SOCK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001422
1423 /* Allocate a buffer to read into. */
1424 if (buf == NULL)
1425 {
1426 buf = alloc(MAXMSGSIZE);
1427 if (buf == NULL)
1428 return; /* out of memory! */
1429 }
1430
1431 /* Keep on reading for as long as there is something to read.
1432 * Use select() or poll() to avoid blocking on a message that is exactly
1433 * MAXMSGSIZE long. */
1434 for (;;)
1435 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001436 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001437 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001438 if (use_socket)
1439 len = sock_read(fd, buf, MAXMSGSIZE);
1440 else
1441 len = read(fd, buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001442 if (len <= 0)
1443 break; /* error or nothing more to read */
1444
1445 /* Store the read message in the queue. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001446 channel_save(channel, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001447 readlen += len;
1448 if (len < MAXMSGSIZE)
1449 break; /* did read everything that's available */
1450 }
Bram Moolenaara8343c12016-02-04 22:09:48 +01001451#ifdef FEAT_GUI_W32
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001452 if (use_socket && len == SOCKET_ERROR)
Bram Moolenaara8343c12016-02-04 22:09:48 +01001453 {
1454 /* For Win32 GUI channel_wait() always returns OK and we handle the
1455 * situation that there is nothing to read here.
1456 * TODO: how about a timeout? */
1457 if (WSAGetLastError() == WSAEWOULDBLOCK)
1458 return;
1459 }
1460#endif
Bram Moolenaard04a0202016-01-26 23:30:18 +01001461
1462 /* Reading a socket disconnection (readlen == 0), or a socket error. */
1463 if (readlen <= 0)
1464 {
1465 /* Queue a "DETACH" netbeans message in the command queue in order to
1466 * terminate the netbeans session later. Do not end the session here
1467 * directly as we may be running in the context of a call to
1468 * netbeans_parse_messages():
1469 * netbeans_parse_messages
1470 * -> autocmd triggered while processing the netbeans cmd
1471 * -> ui_breakcheck
1472 * -> gui event loop or select loop
1473 * -> channel_read()
1474 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001475 ch_errors(channel, "%s(): Cannot read\n", func);
1476 channel_save(channel, (char_u *)DETACH_MSG, (int)STRLEN(DETACH_MSG));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001477
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001478 if (use_socket)
1479 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001480 channel_close(channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001481 if (channel->ch_close_cb != NULL)
1482 (*channel->ch_close_cb)();
1483 }
1484#if defined(CHANNEL_PIPES)
1485 else
1486 {
1487 close(fd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001488 channel->CH_OUT = -1;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001489 }
1490#endif
Bram Moolenaard04a0202016-01-26 23:30:18 +01001491
1492 if (len < 0)
1493 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001494 ch_error(channel, "channel_read(): cannot read from channel\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001495 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001496 }
1497 }
1498
1499#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001500 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001501 if (CH_HAS_GUI && gtk_main_level() > 0)
1502 gtk_main_quit();
1503#endif
1504}
1505
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001506/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001507 * Read from raw channel "channel". Blocks until there is something to read or
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001508 * the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001509 * Returns what was read in allocated memory.
1510 * Returns NULL in case of error or timeout.
1511 */
1512 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +01001513channel_read_block(channel_T *channel)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001514{
Bram Moolenaar77073442016-02-13 23:23:53 +01001515 ch_log(channel, "Reading raw\n");
1516 if (channel_peek(channel) == NULL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001517 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001518 int fd = get_read_fd(channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001519
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001520 /* TODO: read both out and err if they are different */
Bram Moolenaar77073442016-02-13 23:23:53 +01001521 ch_log(channel, "No readahead\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001522 /* Wait for up to the channel timeout. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001523 if (fd < 0 || channel_wait(channel, fd, channel->ch_timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001524 return NULL;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001525 channel_read(channel, -1, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001526 }
1527
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001528 /* TODO: only get the first message */
Bram Moolenaar77073442016-02-13 23:23:53 +01001529 ch_log(channel, "Returning readahead\n");
1530 return channel_get_all(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001531}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001532
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001533/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001534 * Read one JSON message with ID "id" from channel "channel" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001535 * result in "rettv".
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001536 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001537 */
1538 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001539channel_read_json_block(channel_T *channel, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001540{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001541 int more;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001542 int fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001543
Bram Moolenaar77073442016-02-13 23:23:53 +01001544 ch_log(channel, "Reading JSON\n");
Bram Moolenaare56bf152016-02-08 23:23:42 +01001545 channel->ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001546 for (;;)
1547 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001548 more = channel_parse_json(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001549
1550 /* search for messsage "id" */
Bram Moolenaar77073442016-02-13 23:23:53 +01001551 if (channel_get_json(channel, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001552 {
1553 channel->ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001554 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001555 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001556
Bram Moolenaard7ece102016-02-02 23:23:02 +01001557 if (!more)
1558 {
1559 /* Handle any other messages in the queue. If done some more
1560 * messages may have arrived. */
1561 if (channel_parse_messages())
1562 continue;
1563
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001564 /* Wait for up to the channel timeout. */
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001565 fd = get_read_fd(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +01001566 if (fd < 0 || channel_wait(channel, fd, channel->ch_timeout)
1567 == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001568 break;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001569 channel_read(channel, -1, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001570 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001571 }
Bram Moolenaare56bf152016-02-08 23:23:42 +01001572 channel->ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001573 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001574}
1575
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001576# if defined(WIN32) || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001577/*
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001578 * Lookup the channel from the socket. Set "which" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001579 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001580 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001581 channel_T *
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001582channel_fd2channel(sock_T fd, int *whichp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001583{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001584 channel_T *channel;
1585 int i;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001586
1587 if (fd >= 0)
Bram Moolenaar77073442016-02-13 23:23:53 +01001588 for (channel = first_channel; channel != NULL;
1589 channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001590# ifdef CHANNEL_PIPES
1591 for (i = CHAN_SOCK; i < CHAN_IN; ++i)
1592# else
1593 i = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001594# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001595 if (channel->ch_pfd[i].ch_fd == fd)
1596 {
1597 *whichp = i;
1598 return channel
1599 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001600 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001601}
1602# endif
1603
Bram Moolenaard04a0202016-01-26 23:30:18 +01001604/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001605 * Write "buf" (NUL terminated string) to channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001606 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001607 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001608 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001609 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001610channel_send(channel_T *channel, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001611{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001612 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001613 int res;
Bram Moolenaare516c392016-02-13 18:50:38 +01001614 int fd = -1;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001615 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001616
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001617 if (channel->CH_SOCK >= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001618 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001619 fd = channel->CH_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001620 use_socket = TRUE;
1621 }
1622#if defined(CHANNEL_PIPES)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001623 else if (channel->CH_IN >= 0)
1624 fd = channel->CH_IN;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001625#endif
1626 if (fd < 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001627 {
1628 if (!channel->ch_error && fun != NULL)
1629 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001630 ch_errors(channel, "%s(): write while not connected\n", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001631 EMSG2("E630: %s(): write while not connected", fun);
1632 }
1633 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001634 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001635 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001636
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001637 if (log_fd != NULL)
1638 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001639 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001640 fprintf(log_fd, "'");
1641 ignored = fwrite(buf, len, 1, log_fd);
1642 fprintf(log_fd, "'\n");
1643 fflush(log_fd);
1644 }
1645
1646 if (use_socket)
1647 res = sock_write(fd, buf, len);
1648 else
1649 res = write(fd, buf, len);
1650 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001651 {
1652 if (!channel->ch_error && fun != NULL)
1653 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001654 ch_errors(channel, "%s(): write failed\n", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001655 EMSG2("E631: %s(): write failed", fun);
1656 }
1657 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001658 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001659 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001660
1661 channel->ch_error = FALSE;
1662 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001663}
1664
1665# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001666/*
1667 * Add open channels to the poll struct.
1668 * Return the adjusted struct index.
1669 * The type of "fds" is hidden to avoid problems with the function proto.
1670 */
1671 int
1672channel_poll_setup(int nfd_in, void *fds_in)
1673{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001674 int nfd = nfd_in;
1675 channel_T *channel;
1676 struct pollfd *fds = fds_in;
1677 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001678
Bram Moolenaar77073442016-02-13 23:23:53 +01001679 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001680# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001681 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1682# else
1683 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001684# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001685 if (channel->ch_pfd[which].ch_fd >= 0)
1686 {
1687 channel->ch_pfd[which].ch_poll_idx = nfd;
1688 fds[nfd].fd = channel->ch_pfd[which].ch_fd;
1689 fds[nfd].events = POLLIN;
1690 nfd++;
1691 }
1692 else
1693 channel->ch_pfd[which].ch_poll_idx = -1;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001694
1695 return nfd;
1696}
1697
1698/*
1699 * The type of "fds" is hidden to avoid problems with the function proto.
1700 */
1701 int
1702channel_poll_check(int ret_in, void *fds_in)
1703{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001704 int ret = ret_in;
1705 channel_T *channel;
1706 struct pollfd *fds = fds_in;
1707 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001708
Bram Moolenaar77073442016-02-13 23:23:53 +01001709 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001710# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001711 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1712# else
1713 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001714# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001715 {
1716 int idx = channel->ch_pfd[which].ch_poll_idx;
1717
1718 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
1719 {
1720 channel_read(channel, which, "channel_poll_check");
1721 --ret;
1722 }
1723 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001724
1725 return ret;
1726}
Bram Moolenaard04a0202016-01-26 23:30:18 +01001727# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01001728
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001729# if (!defined(FEAT_GUI_W32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001730/*
1731 * The type of "rfds" is hidden to avoid problems with the function proto.
1732 */
1733 int
1734channel_select_setup(int maxfd_in, void *rfds_in)
1735{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001736 int maxfd = maxfd_in;
1737 channel_T *channel;
1738 fd_set *rfds = rfds_in;
1739 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001740
Bram Moolenaar77073442016-02-13 23:23:53 +01001741 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001742# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001743 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1744# else
1745 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001746# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001747 {
1748 sock_T fd = channel->ch_pfd[which].ch_fd;
1749
1750 if (fd >= 0)
1751 {
1752 FD_SET(fd, rfds);
1753 if (maxfd < fd)
1754 maxfd = fd;
1755 }
1756 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001757
1758 return maxfd;
1759}
1760
1761/*
1762 * The type of "rfds" is hidden to avoid problems with the function proto.
1763 */
1764 int
1765channel_select_check(int ret_in, void *rfds_in)
1766{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001767 int ret = ret_in;
1768 channel_T *channel;
1769 fd_set *rfds = rfds_in;
1770 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001771
Bram Moolenaar77073442016-02-13 23:23:53 +01001772 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001773# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001774 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1775# else
1776 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001777# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001778 {
1779 sock_T fd = channel->ch_pfd[which].ch_fd;
1780
1781 if (ret > 0 && fd >= 0 && FD_ISSET(fd, rfds))
1782 {
1783 channel_read(channel, which, "channel_select_check");
1784 --ret;
1785 }
1786 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001787
1788 return ret;
1789}
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001790# endif /* !FEAT_GUI_W32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01001791
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001792/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01001793 * Execute queued up commands.
1794 * Invoked from the main loop when it's safe to execute received commands.
1795 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001796 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001797 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001798channel_parse_messages(void)
1799{
Bram Moolenaar77073442016-02-13 23:23:53 +01001800 channel_T *channel;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001801 int ret = FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001802
Bram Moolenaar77073442016-02-13 23:23:53 +01001803 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1804 while (may_invoke_callback(channel) == OK)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001805 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001806 channel = first_channel; /* start over */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001807 ret = TRUE;
1808 }
1809 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001810}
1811
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01001812/*
1813 * Mark references to lists used in channels.
1814 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001815 int
1816set_ref_in_channel(int copyID)
1817{
Bram Moolenaar77073442016-02-13 23:23:53 +01001818 int abort = FALSE;
1819 channel_T *channel;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001820
Bram Moolenaar77073442016-02-13 23:23:53 +01001821 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001822 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001823 jsonq_T *head = &channel->ch_json_head;
1824 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001825
Bram Moolenaar77073442016-02-13 23:23:53 +01001826 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001827 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001828 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001829
1830 if (l->lv_copyID != copyID)
1831 {
1832 l->lv_copyID = copyID;
1833 abort = abort || set_ref_in_list(l, copyID, NULL);
1834 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001835 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001836 }
1837 }
1838 return abort;
1839}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001840
1841/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001842 * Return the mode of channel "channel".
1843 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001844 */
1845 ch_mode_T
Bram Moolenaar77073442016-02-13 23:23:53 +01001846channel_get_mode(channel_T *channel)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001847{
Bram Moolenaar77073442016-02-13 23:23:53 +01001848 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001849 return MODE_JSON;
Bram Moolenaar77073442016-02-13 23:23:53 +01001850 return channel->ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001851}
1852
Bram Moolenaare0874f82016-01-24 20:36:41 +01001853#endif /* FEAT_CHANNEL */