blob: cad5a1bc1f93e147fbd6f3f14075bd18bcfac4f9 [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(
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100342 (gint)channel->ch_pfd[which].ch_fd,
343 (GdkInputCondition)
344 ((int)GDK_INPUT_READ + (int)GDK_INPUT_EXCEPTION),
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100345 messageFromNetbeans,
346 (gpointer)(long)channel->ch_id);
347# else
348# ifdef FEAT_GUI_W32
349 /* Tell Windows we are interested in receiving message when there
350 * is input on the editor connection socket. */
351 if (channel->ch_pfd[which].ch_inputHandler == -1)
352 channel->ch_pfd[which].ch_inputHandler = WSAAsyncSelect(
353 channel->ch_pfd[which].ch_fd,
354 s_hwnd, WM_NETBEANS, FD_READ);
355# endif
356# endif
357# endif
358}
359
360 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100361channel_gui_register(channel_T *channel)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100362{
Bram Moolenaard04a0202016-01-26 23:30:18 +0100363 if (!CH_HAS_GUI)
364 return;
365
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100366 if (channel->CH_SOCK >= 0)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100367 channel_gui_register_one(channel, CHAN_SOCK);
368# ifdef CHANNEL_PIPES
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100369 if (channel->CH_OUT >= 0)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100370 channel_gui_register_one(channel, CHAN_OUT);
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100371 if (channel->CH_ERR >= 0)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100372 channel_gui_register_one(channel, CHAN_ERR);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100373# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100374}
375
376/*
377 * Register any of our file descriptors with the GUI event handling system.
378 * Called when the GUI has started.
379 */
380 void
381channel_gui_register_all(void)
382{
Bram Moolenaar77073442016-02-13 23:23:53 +0100383 channel_T *channel;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100384
Bram Moolenaar77073442016-02-13 23:23:53 +0100385 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100386 channel_gui_register(channel);
387}
388
389 static void
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100390channel_gui_unregister(channel_T *channel)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100391{
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100392 int which;
393
394#ifdef CHANNEL_PIPES
395 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
396#else
397 which = CHAN_SOCK;
398#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100399 {
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100400# ifdef FEAT_GUI_X11
401 if (channel->ch_pfd[which].ch_inputHandler != (XtInputId)NULL)
402 {
403 XtRemoveInput(channel->ch_pfd[which].ch_inputHandler);
404 channel->ch_pfd[which].ch_inputHandler = (XtInputId)NULL;
405 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100406# else
407# ifdef FEAT_GUI_GTK
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100408 if (channel->ch_pfd[which].ch_inputHandler != 0)
409 {
410 gdk_input_remove(channel->ch_pfd[which].ch_inputHandler);
411 channel->ch_pfd[which].ch_inputHandler = 0;
412 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100413# else
414# ifdef FEAT_GUI_W32
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100415 if (channel->ch_pfd[which].ch_inputHandler == 0)
416 {
417 WSAAsyncSelect(channel->ch_pfd[which].ch_fd, s_hwnd, 0, 0);
418 channel->ch_pfd[which].ch_inputHandler = -1;
419 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100420# endif
421# endif
422# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100423 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100424}
425
426#endif
427
428/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100429 * Open a socket channel to "hostname":"port".
Bram Moolenaar77073442016-02-13 23:23:53 +0100430 * Returns the channel for success.
431 * Returns NULL for failure.
Bram Moolenaard04a0202016-01-26 23:30:18 +0100432 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100433 channel_T *
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100434channel_open(char *hostname, int port_in, int waittime, void (*close_cb)(void))
Bram Moolenaard04a0202016-01-26 23:30:18 +0100435{
436 int sd;
437 struct sockaddr_in server;
438 struct hostent * host;
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100439#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100440 u_short port = port_in;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100441 u_long val = 1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100442#else
443 int port = port_in;
444#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100445 channel_T *channel;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100446 int ret;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100447
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100448#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100449 channel_init_winsock();
450#endif
451
Bram Moolenaar77073442016-02-13 23:23:53 +0100452 channel = add_channel();
453 if (channel == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100454 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100455 ch_error(NULL, "Cannot allocate channel.\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100456 EMSG(_("E897: All channels are in use"));
Bram Moolenaar77073442016-02-13 23:23:53 +0100457 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100458 }
459
460 if ((sd = (sock_T)socket(AF_INET, SOCK_STREAM, 0)) == (sock_T)-1)
461 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100462 ch_error(NULL, "in socket() in channel_open().\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100463 PERROR("E898: socket() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100464 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100465 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100466 }
467
468 /* Get the server internet address and put into addr structure */
469 /* fill in the socket address structure and connect to server */
470 vim_memset((char *)&server, 0, sizeof(server));
471 server.sin_family = AF_INET;
472 server.sin_port = htons(port);
473 if ((host = gethostbyname(hostname)) == NULL)
474 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100475 ch_error(NULL, "in gethostbyname() in channel_open()\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100476 PERROR("E901: gethostbyname() in channel_open()");
Bram Moolenaard04a0202016-01-26 23:30:18 +0100477 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100478 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100479 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100480 }
481 memcpy((char *)&server.sin_addr, host->h_addr, host->h_length);
482
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100483 if (waittime >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100484 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100485 /* Make connect non-blocking. */
486 if (
487#ifdef _WIN32
488 ioctlsocket(sd, FIONBIO, &val) < 0
489#else
490 fcntl(sd, F_SETFL, O_NONBLOCK) < 0
491#endif
492 )
Bram Moolenaard04a0202016-01-26 23:30:18 +0100493 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100494 SOCK_ERRNO;
Bram Moolenaar77073442016-02-13 23:23:53 +0100495 ch_errorn(NULL, "channel_open: Connect failed with errno %d\n",
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100496 errno);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100497 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100498 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100499 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100500 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100501 }
502
503 /* Try connecting to the server. */
Bram Moolenaar77073442016-02-13 23:23:53 +0100504 ch_logsn(NULL, "Connecting to %s port %d", hostname, port);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100505 ret = connect(sd, (struct sockaddr *)&server, sizeof(server));
506 SOCK_ERRNO;
507 if (ret < 0)
508 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100509 if (errno != EWOULDBLOCK
510#ifdef EINPROGRESS
511 && errno != EINPROGRESS
512#endif
513 )
Bram Moolenaard04a0202016-01-26 23:30:18 +0100514 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100515 ch_errorn(NULL, "channel_open: Connect failed with errno %d\n",
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100516 errno);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100517 PERROR(_("E902: Cannot connect to port"));
Bram Moolenaard04a0202016-01-26 23:30:18 +0100518 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100519 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100520 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100521 }
522 }
523
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100524 if (waittime >= 0 && ret < 0)
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100525 {
526 struct timeval tv;
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100527 fd_set wfds;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100528
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100529 FD_ZERO(&wfds);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100530 FD_SET(sd, &wfds);
Bram Moolenaar26dfc412016-02-06 18:18:54 +0100531 tv.tv_sec = waittime / 1000;
532 tv.tv_usec = (waittime % 1000) * 1000;
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100533 ret = select((int)sd + 1, NULL, &wfds, NULL, &tv);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100534 if (ret < 0)
535 {
536 SOCK_ERRNO;
Bram Moolenaar77073442016-02-13 23:23:53 +0100537 ch_errorn(NULL, "channel_open: Connect failed with errno %d\n",
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100538 errno);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100539 PERROR(_("E902: Cannot connect to port"));
540 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100541 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100542 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100543 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100544 if (!FD_ISSET(sd, &wfds))
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100545 {
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100546 /* don't give an error, we just timed out. */
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100547 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100548 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100549 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100550 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100551 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100552
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100553 if (waittime >= 0)
554 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100555#ifdef _WIN32
556 val = 0;
557 ioctlsocket(sd, FIONBIO, &val);
558#else
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100559 (void)fcntl(sd, F_SETFL, 0);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100560#endif
561 }
562
Bram Moolenaar0fa98e72016-02-07 22:21:19 +0100563 /* Only retry for netbeans. TODO: can we use a waittime instead? */
564 if (errno == ECONNREFUSED && close_cb != NULL)
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100565 {
566 sock_close(sd);
567 if ((sd = (sock_T)socket(AF_INET, SOCK_STREAM, 0)) == (sock_T)-1)
568 {
569 SOCK_ERRNO;
Bram Moolenaar77073442016-02-13 23:23:53 +0100570 ch_log(NULL, "socket() retry in channel_open()\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100571 PERROR("E900: socket() retry in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100572 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100573 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100574 }
575 if (connect(sd, (struct sockaddr *)&server, sizeof(server)))
576 {
577 int retries = 36;
578 int success = FALSE;
579
580 SOCK_ERRNO;
581 while (retries-- && ((errno == ECONNREFUSED)
582 || (errno == EINTR)))
583 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100584 ch_log(NULL, "retrying...\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100585 mch_delay(3000L, TRUE);
586 ui_breakcheck();
587 if (got_int)
588 {
589 errno = EINTR;
590 break;
591 }
592 if (connect(sd, (struct sockaddr *)&server,
593 sizeof(server)) == 0)
594 {
595 success = TRUE;
596 break;
597 }
598 SOCK_ERRNO;
599 }
600 if (!success)
601 {
602 /* Get here when the server can't be found. */
Bram Moolenaar77073442016-02-13 23:23:53 +0100603 ch_error(NULL, "Cannot connect to port after retry\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100604 PERROR(_("E899: Cannot connect to port after retry2"));
605 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100606 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100607 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100608 }
609 }
610 }
611
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100612 channel->CH_SOCK = sd;
Bram Moolenaar77073442016-02-13 23:23:53 +0100613 channel->ch_close_cb = close_cb;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100614
615#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +0100616 channel_gui_register(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100617#endif
618
Bram Moolenaar77073442016-02-13 23:23:53 +0100619 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100620}
621
622#if defined(CHANNEL_PIPES) || defined(PROTO)
623 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100624channel_set_pipes(channel_T *channel, int in, int out, int err)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100625{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100626 channel->CH_IN = in;
627 channel->CH_OUT = out;
628 channel->CH_ERR = err;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100629}
630#endif
631
632 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100633channel_set_job(channel_T *channel, job_T *job)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100634{
Bram Moolenaar77073442016-02-13 23:23:53 +0100635 channel->ch_job = job;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100636}
637
638/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100639 * Set the json mode of channel "channel" to "ch_mode".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100640 */
641 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100642channel_set_json_mode(channel_T *channel, ch_mode_T ch_mode)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100643{
Bram Moolenaar77073442016-02-13 23:23:53 +0100644 channel->ch_mode = ch_mode;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100645}
646
647/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100648 * Set the read timeout of channel "channel".
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100649 */
650 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100651channel_set_timeout(channel_T *channel, int timeout)
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100652{
Bram Moolenaar77073442016-02-13 23:23:53 +0100653 channel->ch_timeout = timeout;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100654}
655
656/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100657 * Set the callback for channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100658 */
659 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100660channel_set_callback(channel_T *channel, char_u *callback)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100661{
Bram Moolenaar77073442016-02-13 23:23:53 +0100662 vim_free(channel->ch_callback);
663 channel->ch_callback = vim_strsave(callback);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100664}
665
666/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100667 * Set the callback for channel "channel" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100668 */
669 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100670channel_set_req_callback(channel_T *channel, char_u *callback, int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100671{
Bram Moolenaar77073442016-02-13 23:23:53 +0100672 cbq_T *head = &channel->ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100673 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
674
675 if (item != NULL)
676 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100677 item->cq_callback = vim_strsave(callback);
678 item->cq_seq_nr = id;
679 item->cq_prev = head->cq_prev;
680 head->cq_prev = item;
681 item->cq_next = NULL;
682 if (item->cq_prev == NULL)
683 head->cq_next = item;
684 else
685 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100686 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100687}
688
689/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100690 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100691 */
692 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100693invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100694{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100695 typval_T rettv;
696 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100697
Bram Moolenaar77073442016-02-13 23:23:53 +0100698 argv[0].v_type = VAR_CHANNEL;
699 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100700
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100701 call_func(callback, (int)STRLEN(callback),
702 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
703 /* If an echo command was used the cursor needs to be put back where
704 * it belongs. */
705 setcursor();
706 cursor_on();
707 out_flush();
708}
709
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100710/*
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100711 * Return the first buffer from the channel and remove it.
712 * The caller must free it.
713 * Returns NULL if there is nothing.
714 */
715 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +0100716channel_get(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100717{
Bram Moolenaar77073442016-02-13 23:23:53 +0100718 readq_T *head = &channel->ch_head;
719 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100720 char_u *p;
721
Bram Moolenaar77073442016-02-13 23:23:53 +0100722 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100723 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100724 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100725 p = node->rq_buffer;
726 head->rq_next = node->rq_next;
727 if (node->rq_next == NULL)
728 head->rq_prev = NULL;
729 else
730 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100731 vim_free(node);
732 return p;
733}
734
735/*
736 * Returns the whole buffer contents concatenated.
737 */
738 static char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +0100739channel_get_all(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100740{
741 /* Concatenate everything into one buffer.
742 * TODO: avoid multiple allocations. */
Bram Moolenaar77073442016-02-13 23:23:53 +0100743 while (channel_collapse(channel) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100744 ;
Bram Moolenaar77073442016-02-13 23:23:53 +0100745 return channel_get(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100746}
747
748/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100749 * Collapses the first and second buffer in the channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100750 * Returns FAIL if that is not possible.
751 */
752 int
Bram Moolenaar77073442016-02-13 23:23:53 +0100753channel_collapse(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100754{
Bram Moolenaar77073442016-02-13 23:23:53 +0100755 readq_T *head = &channel->ch_head;
756 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100757 char_u *p;
758
Bram Moolenaar77073442016-02-13 23:23:53 +0100759 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100760 return FAIL;
761
Bram Moolenaar77073442016-02-13 23:23:53 +0100762 p = alloc((unsigned)(STRLEN(node->rq_buffer)
763 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100764 if (p == NULL)
765 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100766 STRCPY(p, node->rq_buffer);
767 STRCAT(p, node->rq_next->rq_buffer);
768 vim_free(node->rq_next->rq_buffer);
769 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100770
Bram Moolenaar77073442016-02-13 23:23:53 +0100771 /* dispose of the node and its buffer */
772 head->rq_next = node->rq_next;
773 head->rq_next->rq_prev = NULL;
774 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100775 vim_free(node);
776 return OK;
777}
778
779/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100780 * Use the read buffer of channel "channel" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100781 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +0100782 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100783 */
Bram Moolenaard7ece102016-02-02 23:23:02 +0100784 static int
Bram Moolenaar77073442016-02-13 23:23:53 +0100785channel_parse_json(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100786{
787 js_read_T reader;
788 typval_T listtv;
789 jsonq_T *item;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100790 jsonq_T *head = &channel->ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100791 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100792
Bram Moolenaar77073442016-02-13 23:23:53 +0100793 if (channel_peek(channel) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100794 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100795
796 /* TODO: make reader work properly */
Bram Moolenaar77073442016-02-13 23:23:53 +0100797 /* reader.js_buf = channel_peek(channel); */
798 reader.js_buf = channel_get_all(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100799 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +0100800 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100801 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +0100802 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100803 ret = json_decode(&reader, &listtv,
804 channel->ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100805 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100806 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +0100807 /* Only accept the response when it is a list with at least two
808 * items. */
809 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100810 {
811 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100812 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100813 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100814 else
815 {
Bram Moolenaard7ece102016-02-02 23:23:02 +0100816 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
817 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100818 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100819 else
820 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100821 item->jq_value = alloc_tv();
822 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100823 {
824 vim_free(item);
825 clear_tv(&listtv);
826 }
827 else
828 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100829 *item->jq_value = listtv;
830 item->jq_prev = head->jq_prev;
831 head->jq_prev = item;
832 item->jq_next = NULL;
833 if (item->jq_prev == NULL)
834 head->jq_next = item;
835 else
836 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100837 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100838 }
839 }
840 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +0100841
842 /* Put the unread part back into the channel.
843 * TODO: insert in front */
844 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100845 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100846 channel_save(channel, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +0100847 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100848 ret = TRUE;
849 }
850 else
851 ret = FALSE;
852
Bram Moolenaardf5b27b2016-02-02 18:43:17 +0100853 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100854 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100855}
856
857/*
858 * Remove "node" from the queue that it is in and free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +0100859 * Also frees the contained callback name.
860 */
861 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100862remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +0100863{
Bram Moolenaar77073442016-02-13 23:23:53 +0100864 if (node->cq_prev == NULL)
865 head->cq_next = node->cq_next;
866 else
867 node->cq_prev->cq_next = node->cq_next;
868 if (node->cq_next == NULL)
869 head->cq_prev = node->cq_prev;
870 else
871 node->cq_next->cq_prev = node->cq_prev;
872 vim_free(node->cq_callback);
Bram Moolenaara07fec92016-02-05 21:04:08 +0100873 vim_free(node);
874}
875
876/*
877 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +0100878 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100879 */
880 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100881remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100882{
Bram Moolenaar77073442016-02-13 23:23:53 +0100883 if (node->jq_prev == NULL)
884 head->jq_next = node->jq_next;
885 else
886 node->jq_prev->jq_next = node->jq_next;
887 if (node->jq_next == NULL)
888 head->jq_prev = node->jq_prev;
889 else
890 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100891 vim_free(node);
892}
893
894/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100895 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100896 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +0100897 * When "id" is zero or negative jut get the first message. But not the one
898 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100899 * Return OK when found and return the value in "rettv".
900 * Return FAIL otherwise.
901 */
902 static int
Bram Moolenaar77073442016-02-13 23:23:53 +0100903channel_get_json(channel_T *channel, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100904{
Bram Moolenaare56bf152016-02-08 23:23:42 +0100905 jsonq_T *head = &channel->ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100906 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100907
Bram Moolenaar77073442016-02-13 23:23:53 +0100908 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100909 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100910 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100911 typval_T *tv = &l->lv_first->li_tv;
912
913 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +0100914 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaarf6157282016-02-10 21:07:14 +0100915 || tv->vval.v_number == 0
Bram Moolenaare56bf152016-02-08 23:23:42 +0100916 || tv->vval.v_number != channel->ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100917 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100918 *rettv = item->jq_value;
919 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100920 return OK;
921 }
Bram Moolenaar77073442016-02-13 23:23:53 +0100922 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100923 }
924 return FAIL;
925}
926
927/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100928 * Execute a command received over channel "channel".
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100929 * "cmd" is the command string, "arg2" the second argument.
930 * "arg3" is the third argument, NULL if missing.
931 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100932 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100933channel_exe_cmd(channel_T *channel, char_u *cmd, typval_T *arg2, typval_T *arg3)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100934{
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100935 char_u *arg;
936
937 if (arg2->v_type != VAR_STRING)
938 {
939 if (p_verbose > 2)
940 EMSG("E903: received ex command with non-string argument");
941 return;
942 }
943 arg = arg2->vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +0100944 if (arg == NULL)
945 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100946
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100947 if (STRCMP(cmd, "ex") == 0)
948 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100949 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100950 }
951 else if (STRCMP(cmd, "normal") == 0)
952 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100953 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100954
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100955 ea.arg = arg;
956 ea.addr_count = 0;
957 ea.forceit = TRUE; /* no mapping */
958 ex_normal(&ea);
959 }
960 else if (STRCMP(cmd, "redraw") == 0)
961 {
962 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100963
Bram Moolenaar14ad6112016-02-01 21:47:13 +0100964 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100965 ex_redraw(&ea);
966 showruler(FALSE);
967 setcursor();
968 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100969#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100970 if (gui.in_use)
971 {
972 gui_update_cursor(FALSE, FALSE);
973 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100974 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100975#endif
976 }
977 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "eval") == 0)
978 {
979 int is_eval = cmd[1] == 'v';
980
Bram Moolenaar66624ff2016-02-03 23:59:43 +0100981 if (is_eval && (arg3 == NULL || arg3->v_type != VAR_NUMBER))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100982 {
983 if (p_verbose > 2)
984 EMSG("E904: third argument for eval must be a number");
985 }
986 else
987 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +0100988 typval_T *tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100989 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +0100990 char_u *json = NULL;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100991 int options = channel->ch_mode == MODE_JS ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100992
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +0100993 /* Don't pollute the display with errors. */
994 ++emsg_skip;
995 tv = eval_expr(arg, NULL);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100996 if (is_eval)
997 {
Bram Moolenaar55fab432016-02-07 16:53:13 +0100998 if (tv != NULL)
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100999 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1000 options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001001 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001002 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001003 /* If evaluation failed or the result can't be encoded
1004 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001005 vim_free(json);
1006 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001007 err_tv.v_type = VAR_STRING;
1008 err_tv.vval.v_string = (char_u *)"ERROR";
1009 tv = &err_tv;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001010 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1011 options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001012 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001013 if (json != NULL)
1014 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001015 channel_send(channel, json, "eval");
Bram Moolenaar55fab432016-02-07 16:53:13 +01001016 vim_free(json);
1017 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001018 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001019 --emsg_skip;
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001020 if (tv != &err_tv)
1021 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001022 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001023 }
1024 else if (p_verbose > 2)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001025 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001026}
1027
1028/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001029 * Invoke a callback for channel "channel" if needed.
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001030 * Return OK when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001031 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001032 static int
Bram Moolenaar77073442016-02-13 23:23:53 +01001033may_invoke_callback(channel_T *channel)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001034{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001035 char_u *msg = NULL;
1036 typval_T *listtv = NULL;
1037 list_T *list;
1038 typval_T *typetv;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001039 typval_T argv[3];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001040 int seq_nr = -1;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001041 ch_mode_T ch_mode = channel->ch_mode;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001042
Bram Moolenaara07fec92016-02-05 21:04:08 +01001043 if (channel->ch_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001044 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001045 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001046
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001047 if (ch_mode != MODE_RAW)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001048 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001049 /* Get any json message in the queue. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001050 if (channel_get_json(channel, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001051 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001052 /* Parse readahead, return when there is still no message. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001053 channel_parse_json(channel);
1054 if (channel_get_json(channel, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001055 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001056 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001057
1058 list = listtv->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001059 argv[1] = list->lv_first->li_next->li_tv;
1060 typetv = &list->lv_first->li_tv;
1061 if (typetv->v_type == VAR_STRING)
1062 {
1063 typval_T *arg3 = NULL;
1064 char_u *cmd = typetv->vval.v_string;
1065
Bram Moolenaar66624ff2016-02-03 23:59:43 +01001066 /* ["cmd", arg] or ["cmd", arg, arg] */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001067 if (list->lv_len == 3)
1068 arg3 = &list->lv_last->li_tv;
Bram Moolenaar77073442016-02-13 23:23:53 +01001069 ch_logs(channel, "Executing %s command", (char *)cmd);
1070 channel_exe_cmd(channel, cmd, &argv[1], arg3);
1071 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001072 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001073 }
1074
1075 if (typetv->v_type != VAR_NUMBER)
1076 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001077 ch_error(channel,
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001078 "Dropping message with invalid sequence number type\n");
Bram Moolenaar77073442016-02-13 23:23:53 +01001079 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001080 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001081 }
1082 seq_nr = typetv->vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001083 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001084 else if (channel_peek(channel) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001085 {
1086 /* nothing to read on raw channel */
1087 return FALSE;
1088 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001089 else
1090 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001091 /* If there is no callback, don't do anything. */
1092 if (channel->ch_callback == NULL)
1093 return FALSE;
1094
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001095 /* For a raw channel we don't know where the message ends, just get
1096 * everything. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001097 msg = channel_get_all(channel);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001098 argv[1].v_type = VAR_STRING;
1099 argv[1].vval.v_string = msg;
1100 }
1101
Bram Moolenaara07fec92016-02-05 21:04:08 +01001102 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001103 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001104 cbq_T *head = &channel->ch_cb_head;
1105 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001106 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001107
1108 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001109 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001110 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001111 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001112 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001113 ch_log(channel, "Invoking one-time callback\n");
1114 invoke_callback(channel, item->cq_callback, argv);
1115 remove_cb_node(head, item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001116 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001117 break;
1118 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001119 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001120 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001121 if (!done)
Bram Moolenaar77073442016-02-13 23:23:53 +01001122 ch_log(channel, "Dropping message without callback\n");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001123 }
Bram Moolenaara07fec92016-02-05 21:04:08 +01001124 else if (channel->ch_callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001125 {
1126 /* invoke the channel callback */
Bram Moolenaar77073442016-02-13 23:23:53 +01001127 ch_log(channel, "Invoking channel callback\n");
1128 invoke_callback(channel, channel->ch_callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001129 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001130 else
Bram Moolenaar77073442016-02-13 23:23:53 +01001131 ch_log(channel, "Dropping message\n");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001132
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001133 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001134 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001135 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001136
1137 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001138}
1139
1140/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001141 * Return TRUE when channel "channel" is open for writing to.
1142 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001143 */
1144 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001145channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001146{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001147 return channel != NULL && (channel->CH_SOCK >= 0
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001148#ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001149 || channel->CH_IN >= 0
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001150#endif
1151 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001152}
1153
1154/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001155 * Return TRUE when channel "channel" is open for reading or writing.
1156 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001157 */
1158 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001159channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001160{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001161 return channel != NULL && (channel->CH_SOCK >= 0
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001162#ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001163 || channel->CH_IN >= 0
1164 || channel->CH_OUT >= 0
1165 || channel->CH_ERR >= 0
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001166#endif
1167 );
1168}
1169
1170/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001171 * Return a string indicating the status of the channel.
1172 */
1173 char *
1174channel_status(channel_T *channel)
1175{
1176 if (channel == NULL)
1177 return "fail";
1178 if (channel_is_open(channel))
1179 return "open";
1180 return "closed";
1181}
1182
1183/*
1184 * Close channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001185 * This does not trigger the close callback.
1186 */
1187 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001188channel_close(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001189{
Bram Moolenaar77073442016-02-13 23:23:53 +01001190 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001191
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001192#ifdef FEAT_GUI
1193 channel_gui_unregister(channel);
1194#endif
1195
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001196 if (channel->CH_SOCK >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001197 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001198 sock_close(channel->CH_SOCK);
1199 channel->CH_SOCK = -1;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001200 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001201#if defined(CHANNEL_PIPES)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001202 if (channel->CH_IN >= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001203 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001204 close(channel->CH_IN);
1205 channel->CH_IN = -1;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001206 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001207 if (channel->CH_OUT >= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001208 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001209 close(channel->CH_OUT);
1210 channel->CH_OUT = -1;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001211 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001212 if (channel->CH_ERR >= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001213 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001214 close(channel->CH_ERR);
1215 channel->CH_ERR = -1;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001216 }
1217#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001218
1219 channel->ch_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001220 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001221}
1222
Bram Moolenaard04a0202016-01-26 23:30:18 +01001223/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001224 * Store "buf[len]" on channel "channel".
Bram Moolenaar83162462016-01-28 23:10:07 +01001225 * Returns OK or FAIL.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001226 */
Bram Moolenaar83162462016-01-28 23:10:07 +01001227 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001228channel_save(channel_T *channel, char_u *buf, int len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001229{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001230 readq_T *node;
Bram Moolenaar77073442016-02-13 23:23:53 +01001231 readq_T *head = &channel->ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001232
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001233 node = (readq_T *)alloc(sizeof(readq_T));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001234 if (node == NULL)
Bram Moolenaar83162462016-01-28 23:10:07 +01001235 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +01001236 node->rq_buffer = alloc(len + 1);
1237 if (node->rq_buffer == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001238 {
1239 vim_free(node);
Bram Moolenaar83162462016-01-28 23:10:07 +01001240 return FAIL; /* out of memory */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001241 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001242 mch_memmove(node->rq_buffer, buf, (size_t)len);
1243 node->rq_buffer[len] = NUL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001244
Bram Moolenaar77073442016-02-13 23:23:53 +01001245 /* append node to the tail of the queue */
1246 node->rq_next = NULL;
1247 node->rq_prev = head->rq_prev;
1248 if (head->rq_prev == NULL)
1249 head->rq_next = node;
1250 else
1251 head->rq_prev->rq_next = node;
1252 head->rq_prev = node;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001253
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001254 if (log_fd != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001255 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001256 ch_log_lead("RECV ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001257 fprintf(log_fd, "'");
1258 if (fwrite(buf, len, 1, log_fd) != 1)
Bram Moolenaar83162462016-01-28 23:10:07 +01001259 return FAIL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001260 fprintf(log_fd, "'\n");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001261 }
Bram Moolenaar83162462016-01-28 23:10:07 +01001262 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001263}
1264
1265/*
1266 * Return the first buffer from the channel without removing it.
1267 * Returns NULL if there is nothing.
1268 */
1269 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +01001270channel_peek(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001271{
Bram Moolenaar77073442016-02-13 23:23:53 +01001272 readq_T *head = &channel->ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001273
Bram Moolenaar77073442016-02-13 23:23:53 +01001274 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001275 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001276 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001277}
1278
1279/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001280 * Clear the read buffer on channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001281 */
1282 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001283channel_clear(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001284{
Bram Moolenaar77073442016-02-13 23:23:53 +01001285 jsonq_T *json_head = &channel->ch_json_head;
1286 cbq_T *cb_head = &channel->ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001287
Bram Moolenaar77073442016-02-13 23:23:53 +01001288 while (channel_peek(channel) != NULL)
1289 vim_free(channel_get(channel));
1290
1291 while (cb_head->cq_next != NULL)
1292 remove_cb_node(cb_head, cb_head->cq_next);
1293
1294 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001295 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001296 free_tv(json_head->jq_next->jq_value);
1297 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001298 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001299
1300 vim_free(channel->ch_callback);
1301 channel->ch_callback = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001302}
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 Moolenaar16eb4f82016-02-14 23:02:34 +01001386 * TODO: should have a way to read 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 Moolenaar16eb4f82016-02-14 23:02:34 +01001403 * "which" is CHAN_SOCK, CHAN_OUT or CHAN_ERR. When -1 use CHAN_SOCK or
1404 * CHAN_OUT, the one that is open.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001405 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001406 */
1407 void
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001408channel_read(channel_T *channel, int which, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001409{
1410 static char_u *buf = NULL;
1411 int len = 0;
1412 int readlen = 0;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001413 int fd;
1414 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001415
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001416 if (which < 0)
1417 fd = get_read_fd(channel);
1418 else
1419 fd = channel->ch_pfd[which].ch_fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001420 if (fd < 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001421 return;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001422 use_socket = fd == channel->CH_SOCK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001423
1424 /* Allocate a buffer to read into. */
1425 if (buf == NULL)
1426 {
1427 buf = alloc(MAXMSGSIZE);
1428 if (buf == NULL)
1429 return; /* out of memory! */
1430 }
1431
1432 /* Keep on reading for as long as there is something to read.
1433 * Use select() or poll() to avoid blocking on a message that is exactly
1434 * MAXMSGSIZE long. */
1435 for (;;)
1436 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001437 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001438 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001439 if (use_socket)
1440 len = sock_read(fd, buf, MAXMSGSIZE);
1441 else
1442 len = read(fd, buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001443 if (len <= 0)
1444 break; /* error or nothing more to read */
1445
1446 /* Store the read message in the queue. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001447 channel_save(channel, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001448 readlen += len;
1449 if (len < MAXMSGSIZE)
1450 break; /* did read everything that's available */
1451 }
Bram Moolenaara8343c12016-02-04 22:09:48 +01001452#ifdef FEAT_GUI_W32
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001453 if (use_socket && len == SOCKET_ERROR)
Bram Moolenaara8343c12016-02-04 22:09:48 +01001454 {
1455 /* For Win32 GUI channel_wait() always returns OK and we handle the
1456 * situation that there is nothing to read here.
1457 * TODO: how about a timeout? */
1458 if (WSAGetLastError() == WSAEWOULDBLOCK)
1459 return;
1460 }
1461#endif
Bram Moolenaard04a0202016-01-26 23:30:18 +01001462
1463 /* Reading a socket disconnection (readlen == 0), or a socket error. */
1464 if (readlen <= 0)
1465 {
1466 /* Queue a "DETACH" netbeans message in the command queue in order to
1467 * terminate the netbeans session later. Do not end the session here
1468 * directly as we may be running in the context of a call to
1469 * netbeans_parse_messages():
1470 * netbeans_parse_messages
1471 * -> autocmd triggered while processing the netbeans cmd
1472 * -> ui_breakcheck
1473 * -> gui event loop or select loop
1474 * -> channel_read()
1475 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001476 ch_errors(channel, "%s(): Cannot read\n", func);
1477 channel_save(channel, (char_u *)DETACH_MSG, (int)STRLEN(DETACH_MSG));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001478
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001479 /* TODO: When reading from stdout is not possible, should we try to
1480 * keep stdin and stderr open? Probably not, assume the other side
1481 * has died. */
1482 channel_close(channel);
1483 if (channel->ch_close_cb != NULL)
1484 (*channel->ch_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001485
1486 if (len < 0)
1487 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001488 ch_error(channel, "channel_read(): cannot read from channel\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001489 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001490 }
1491 }
1492
1493#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001494 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001495 if (CH_HAS_GUI && gtk_main_level() > 0)
1496 gtk_main_quit();
1497#endif
1498}
1499
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001500/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001501 * Read from raw channel "channel". Blocks until there is something to read or
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001502 * the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001503 * Returns what was read in allocated memory.
1504 * Returns NULL in case of error or timeout.
1505 */
1506 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +01001507channel_read_block(channel_T *channel)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001508{
Bram Moolenaar77073442016-02-13 23:23:53 +01001509 ch_log(channel, "Reading raw\n");
1510 if (channel_peek(channel) == NULL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001511 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001512 int fd = get_read_fd(channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001513
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001514 /* TODO: read both out and err if they are different */
Bram Moolenaar77073442016-02-13 23:23:53 +01001515 ch_log(channel, "No readahead\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001516 /* Wait for up to the channel timeout. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001517 if (fd < 0 || channel_wait(channel, fd, channel->ch_timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001518 return NULL;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001519 channel_read(channel, -1, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001520 }
1521
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001522 /* TODO: only get the first message */
Bram Moolenaar77073442016-02-13 23:23:53 +01001523 ch_log(channel, "Returning readahead\n");
1524 return channel_get_all(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001525}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001526
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001527/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001528 * Read one JSON message with ID "id" from channel "channel" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001529 * result in "rettv".
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001530 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001531 */
1532 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001533channel_read_json_block(channel_T *channel, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001534{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001535 int more;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001536 int fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001537
Bram Moolenaar77073442016-02-13 23:23:53 +01001538 ch_log(channel, "Reading JSON\n");
Bram Moolenaare56bf152016-02-08 23:23:42 +01001539 channel->ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001540 for (;;)
1541 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001542 more = channel_parse_json(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001543
1544 /* search for messsage "id" */
Bram Moolenaar77073442016-02-13 23:23:53 +01001545 if (channel_get_json(channel, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001546 {
1547 channel->ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001548 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001549 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001550
Bram Moolenaard7ece102016-02-02 23:23:02 +01001551 if (!more)
1552 {
1553 /* Handle any other messages in the queue. If done some more
1554 * messages may have arrived. */
1555 if (channel_parse_messages())
1556 continue;
1557
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001558 /* Wait for up to the channel timeout. */
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001559 fd = get_read_fd(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +01001560 if (fd < 0 || channel_wait(channel, fd, channel->ch_timeout)
1561 == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001562 break;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001563 channel_read(channel, -1, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001564 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001565 }
Bram Moolenaare56bf152016-02-08 23:23:42 +01001566 channel->ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001567 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001568}
1569
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001570# if defined(WIN32) || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001571/*
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001572 * Lookup the channel from the socket. Set "which" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001573 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001574 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001575 channel_T *
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001576channel_fd2channel(sock_T fd, int *whichp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001577{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001578 channel_T *channel;
1579 int i;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001580
1581 if (fd >= 0)
Bram Moolenaar77073442016-02-13 23:23:53 +01001582 for (channel = first_channel; channel != NULL;
1583 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001584 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001585# ifdef CHANNEL_PIPES
1586 for (i = CHAN_SOCK; i < CHAN_IN; ++i)
1587# else
1588 i = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001589# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001590 if (channel->ch_pfd[i].ch_fd == fd)
1591 {
1592 *whichp = i;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001593 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001594 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001595 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001596 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001597}
1598# endif
1599
Bram Moolenaard04a0202016-01-26 23:30:18 +01001600/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001601 * Write "buf" (NUL terminated string) to channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001602 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001603 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001604 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001605 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001606channel_send(channel_T *channel, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001607{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001608 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001609 int res;
Bram Moolenaare516c392016-02-13 18:50:38 +01001610 int fd = -1;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001611 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001612
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001613 if (channel->CH_SOCK >= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001614 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001615 fd = channel->CH_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001616 use_socket = TRUE;
1617 }
1618#if defined(CHANNEL_PIPES)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001619 else if (channel->CH_IN >= 0)
1620 fd = channel->CH_IN;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001621#endif
1622 if (fd < 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001623 {
1624 if (!channel->ch_error && fun != NULL)
1625 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001626 ch_errors(channel, "%s(): write while not connected\n", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001627 EMSG2("E630: %s(): write while not connected", fun);
1628 }
1629 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001630 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001631 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001632
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001633 if (log_fd != NULL)
1634 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001635 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001636 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001637 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001638 fprintf(log_fd, "'\n");
1639 fflush(log_fd);
1640 }
1641
1642 if (use_socket)
1643 res = sock_write(fd, buf, len);
1644 else
1645 res = write(fd, buf, len);
1646 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001647 {
1648 if (!channel->ch_error && fun != NULL)
1649 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001650 ch_errors(channel, "%s(): write failed\n", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001651 EMSG2("E631: %s(): write failed", fun);
1652 }
1653 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001654 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001655 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001656
1657 channel->ch_error = FALSE;
1658 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001659}
1660
1661# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001662/*
1663 * Add open channels to the poll struct.
1664 * Return the adjusted struct index.
1665 * The type of "fds" is hidden to avoid problems with the function proto.
1666 */
1667 int
1668channel_poll_setup(int nfd_in, void *fds_in)
1669{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001670 int nfd = nfd_in;
1671 channel_T *channel;
1672 struct pollfd *fds = fds_in;
1673 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001674
Bram Moolenaar77073442016-02-13 23:23:53 +01001675 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001676 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001677# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001678 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1679# else
1680 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001681# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001682 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001683 if (channel->ch_pfd[which].ch_fd >= 0)
1684 {
1685 channel->ch_pfd[which].ch_poll_idx = nfd;
1686 fds[nfd].fd = channel->ch_pfd[which].ch_fd;
1687 fds[nfd].events = POLLIN;
1688 nfd++;
1689 }
1690 else
1691 channel->ch_pfd[which].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001692 }
1693 }
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 Moolenaar16eb4f82016-02-14 23:02:34 +01001710 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001711# ifdef CHANNEL_PIPES
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001712 for (which = CHAN_SOCK; which < CH_IN; ++which)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001713# else
1714 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001715# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001716 {
1717 int idx = channel->ch_pfd[which].ch_poll_idx;
1718
1719 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
1720 {
1721 channel_read(channel, which, "channel_poll_check");
1722 --ret;
1723 }
1724 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001725 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001726
1727 return ret;
1728}
Bram Moolenaard04a0202016-01-26 23:30:18 +01001729# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01001730
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001731# if (!defined(FEAT_GUI_W32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001732/*
1733 * The type of "rfds" is hidden to avoid problems with the function proto.
1734 */
1735 int
1736channel_select_setup(int maxfd_in, void *rfds_in)
1737{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001738 int maxfd = maxfd_in;
1739 channel_T *channel;
1740 fd_set *rfds = rfds_in;
1741 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001742
Bram Moolenaar77073442016-02-13 23:23:53 +01001743 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001744 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001745# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001746 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1747# else
1748 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001749# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001750 {
1751 sock_T fd = channel->ch_pfd[which].ch_fd;
1752
1753 if (fd >= 0)
1754 {
1755 FD_SET(fd, rfds);
1756 if (maxfd < fd)
1757 maxfd = fd;
1758 }
1759 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001760 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001761
1762 return maxfd;
1763}
1764
1765/*
1766 * The type of "rfds" is hidden to avoid problems with the function proto.
1767 */
1768 int
1769channel_select_check(int ret_in, void *rfds_in)
1770{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001771 int ret = ret_in;
1772 channel_T *channel;
1773 fd_set *rfds = rfds_in;
1774 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001775
Bram Moolenaar77073442016-02-13 23:23:53 +01001776 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001777 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001778# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001779 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1780# else
1781 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001782# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001783 {
1784 sock_T fd = channel->ch_pfd[which].ch_fd;
1785
1786 if (ret > 0 && fd >= 0 && FD_ISSET(fd, rfds))
1787 {
1788 channel_read(channel, which, "channel_select_check");
1789 --ret;
1790 }
1791 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001792 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001793
1794 return ret;
1795}
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001796# endif /* !FEAT_GUI_W32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01001797
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001798/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01001799 * Execute queued up commands.
1800 * Invoked from the main loop when it's safe to execute received commands.
1801 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001802 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001803 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001804channel_parse_messages(void)
1805{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001806 channel_T *channel = first_channel;
1807 int ret = FALSE;
1808 int r;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001809
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001810 while (channel != NULL)
1811 {
1812 /* Increase the refcount, in case the handler causes the channel to be
1813 * unreferenced or closed. */
1814 ++channel->ch_refcount;
1815 r = may_invoke_callback(channel);
1816 if (channel_unref(channel))
1817 /* channel was freed, start over */
1818 channel = first_channel;
1819
1820 if (r == OK)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001821 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001822 channel = first_channel; /* something was done, start over */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001823 ret = TRUE;
1824 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001825 else
1826 channel = channel->ch_next;
1827 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01001828 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001829}
1830
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01001831/*
1832 * Mark references to lists used in channels.
1833 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001834 int
1835set_ref_in_channel(int copyID)
1836{
Bram Moolenaar77073442016-02-13 23:23:53 +01001837 int abort = FALSE;
1838 channel_T *channel;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001839
Bram Moolenaar77073442016-02-13 23:23:53 +01001840 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001841 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001842 jsonq_T *head = &channel->ch_json_head;
1843 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001844
Bram Moolenaar77073442016-02-13 23:23:53 +01001845 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001846 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001847 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001848
1849 if (l->lv_copyID != copyID)
1850 {
1851 l->lv_copyID = copyID;
1852 abort = abort || set_ref_in_list(l, copyID, NULL);
1853 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001854 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001855 }
1856 }
1857 return abort;
1858}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001859
1860/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001861 * Return the mode of channel "channel".
1862 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001863 */
1864 ch_mode_T
Bram Moolenaar77073442016-02-13 23:23:53 +01001865channel_get_mode(channel_T *channel)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001866{
Bram Moolenaar77073442016-02-13 23:23:53 +01001867 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001868 return MODE_JSON;
Bram Moolenaar77073442016-02-13 23:23:53 +01001869 return channel->ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001870}
1871
Bram Moolenaare0874f82016-01-24 20:36:41 +01001872#endif /* FEAT_CHANNEL */