blob: 28fee0ac91465f315dd4e7c2cd16f053a2ec8b95 [file] [log] [blame]
Bram Moolenaare0874f82016-01-24 20:36:41 +01001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 */
8
9/*
10 * Implements communication through a socket or any file handle.
11 */
12
13#include "vim.h"
14
15#if defined(FEAT_CHANNEL) || defined(PROTO)
16
Bram Moolenaard04a0202016-01-26 23:30:18 +010017/* TRUE when netbeans is running with a GUI. */
18#ifdef FEAT_GUI
19# define CH_HAS_GUI (gui.in_use || gui.starting)
20#endif
21
22/* Note: when making changes here also adjust configure.in. */
23#ifdef WIN32
24/* WinSock API is separated from C API, thus we can't use read(), write(),
25 * errno... */
26# define SOCK_ERRNO errno = WSAGetLastError()
27# undef ECONNREFUSED
28# define ECONNREFUSED WSAECONNREFUSED
Bram Moolenaar4d919d72016-02-05 22:36:41 +010029# undef EWOULDBLOCK
30# define EWOULDBLOCK WSAEWOULDBLOCK
Bram Moolenaard04a0202016-01-26 23:30:18 +010031# ifdef EINTR
32# undef EINTR
33# endif
34# define EINTR WSAEINTR
Bram Moolenaard8070362016-02-15 21:56:54 +010035# define sock_write(sd, buf, len) send((SOCKET)sd, buf, len, 0)
36# define sock_read(sd, buf, len) recv((SOCKET)sd, buf, len, 0)
37# define sock_close(sd) closesocket((SOCKET)sd)
Bram Moolenaard04a0202016-01-26 23:30:18 +010038#else
39# include <netdb.h>
40# include <netinet/in.h>
41
42# include <sys/socket.h>
43# ifdef HAVE_LIBGEN_H
44# include <libgen.h>
45# endif
46# define SOCK_ERRNO
47# define sock_write(sd, buf, len) write(sd, buf, len)
48# define sock_read(sd, buf, len) read(sd, buf, len)
49# define sock_close(sd) close(sd)
Bram Moolenaar0943a092016-02-16 13:11:17 +010050# define fd_read(fd, buf, len) read(fd, buf, len)
Bram Moolenaard8070362016-02-15 21:56:54 +010051# define fd_write(sd, buf, len) write(sd, buf, len)
52# define fd_close(sd) close(sd)
Bram Moolenaard04a0202016-01-26 23:30:18 +010053#endif
54
55#ifdef FEAT_GUI_W32
56extern HWND s_hwnd; /* Gvim's Window handle */
57#endif
58
Bram Moolenaard8070362016-02-15 21:56:54 +010059#ifdef WIN32
60 static int
Bram Moolenaar0943a092016-02-16 13:11:17 +010061fd_read(sock_T fd, char_u *buf, size_t len)
Bram Moolenaard8070362016-02-15 21:56:54 +010062{
63 HANDLE h = (HANDLE)fd;
64 DWORD nread;
65
66 if (!ReadFile(h, buf, (DWORD)len, &nread, NULL))
67 return -1;
68 return (int)nread;
69}
70
71 static int
72fd_write(sock_T fd, char_u *buf, size_t len)
73{
74 HANDLE h = (HANDLE)fd;
75 DWORD nwrite;
76
77 if (!WriteFile(h, buf, (DWORD)len, &nwrite, NULL))
78 return -1;
79 return (int)nwrite;
80}
81
82 static void
83fd_close(sock_T fd)
84{
85 HANDLE h = (HANDLE)fd;
86
87 CloseHandle(h);
88}
89#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +010090
Bram Moolenaar6463ca22016-02-13 17:04:46 +010091/* Log file opened with ch_logfile(). */
92static FILE *log_fd = NULL;
93
94 void
95ch_logfile(FILE *file)
96{
97 if (log_fd != NULL)
98 fclose(log_fd);
99 log_fd = file;
100 if (log_fd != NULL)
101 fprintf(log_fd, "==== start log session ====\n");
102}
103
104 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100105ch_log_lead(char *what, channel_T *ch)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100106{
107 if (log_fd != NULL)
108 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100109 if (ch != NULL)
110 fprintf(log_fd, "%son %d: ", what, ch->ch_id);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100111 else
112 fprintf(log_fd, "%s: ", what);
113 }
114}
115
116 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100117ch_log(channel_T *ch, char *msg)
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 fputs(msg, log_fd);
123 fflush(log_fd);
124 }
125}
126
127 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100128ch_logn(channel_T *ch, char *msg, int nr)
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("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100133 fprintf(log_fd, msg, nr);
134 fflush(log_fd);
135 }
136}
137
138 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100139ch_logs(channel_T *ch, char *msg, char *name)
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("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100144 fprintf(log_fd, msg, name);
145 fflush(log_fd);
146 }
147}
148
149 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100150ch_logsn(channel_T *ch, char *msg, char *name, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100151{
152 if (log_fd != NULL)
153 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100154 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100155 fprintf(log_fd, msg, name, nr);
156 fflush(log_fd);
157 }
158}
159
160 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100161ch_error(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100162{
163 if (log_fd != NULL)
164 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100165 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100166 fputs(msg, log_fd);
167 fflush(log_fd);
168 }
169}
170
171 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100172ch_errorn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100173{
174 if (log_fd != NULL)
175 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100176 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100177 fprintf(log_fd, msg, nr);
178 fflush(log_fd);
179 }
180}
181
182 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100183ch_errors(channel_T *ch, char *msg, char *arg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100184{
185 if (log_fd != NULL)
186 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100187 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100188 fprintf(log_fd, msg, arg);
189 fflush(log_fd);
190 }
191}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100192
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100193#ifdef _WIN32
194# undef PERROR
195# define PERROR(msg) (void)emsg3((char_u *)"%s: %s", \
196 (char_u *)msg, (char_u *)strerror_win32(errno))
197
198 static char *
199strerror_win32(int eno)
200{
201 static LPVOID msgbuf = NULL;
202 char_u *ptr;
203
204 if (msgbuf)
205 LocalFree(msgbuf);
206 FormatMessage(
207 FORMAT_MESSAGE_ALLOCATE_BUFFER |
208 FORMAT_MESSAGE_FROM_SYSTEM |
209 FORMAT_MESSAGE_IGNORE_INSERTS,
210 NULL,
211 eno,
212 MAKELANGID(LANG_ENGLISH, SUBLANG_DEFAULT),
213 (LPTSTR) &msgbuf,
214 0,
215 NULL);
216 /* chomp \r or \n */
217 for (ptr = (char_u *)msgbuf; *ptr; ptr++)
218 switch (*ptr)
219 {
220 case '\r':
221 STRMOVE(ptr, ptr + 1);
222 ptr--;
223 break;
224 case '\n':
225 if (*(ptr + 1) == '\0')
226 *ptr = '\0';
227 else
228 *ptr = ' ';
229 break;
230 }
231 return msgbuf;
232}
233#endif
234
Bram Moolenaar77073442016-02-13 23:23:53 +0100235/*
236 * The list of all allocated channels.
237 */
238static channel_T *first_channel = NULL;
239static int next_ch_id = 0;
240
241/*
242 * Allocate a new channel. The refcount is set to 1.
243 * The channel isn't actually used until it is opened.
244 * Returns NULL if out of memory.
245 */
246 channel_T *
247add_channel(void)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100248{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100249 int which;
250 channel_T *channel = (channel_T *)alloc_clear((int)sizeof(channel_T));
Bram Moolenaare0874f82016-01-24 20:36:41 +0100251
Bram Moolenaar77073442016-02-13 23:23:53 +0100252 if (channel == NULL)
253 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100254
Bram Moolenaar77073442016-02-13 23:23:53 +0100255 channel->ch_id = next_ch_id++;
256 ch_log(channel, "Opening channel\n");
257
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100258#ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100259 for (which = CHAN_SOCK; which <= CHAN_IN; ++which)
260#else
261 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100262#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100263 {
Bram Moolenaard8070362016-02-15 21:56:54 +0100264 channel->ch_pfd[which].ch_fd = CHAN_FD_INVALID;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100265#ifdef FEAT_GUI_X11
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100266 channel->ch_pfd[which].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100267#endif
268#ifdef FEAT_GUI_GTK
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100269 channel->ch_pfd[which].ch_inputHandler = 0;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100270#endif
271#ifdef FEAT_GUI_W32
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100272 channel->ch_pfd[which].ch_inputHandler = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100273#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100274 }
Bram Moolenaare0874f82016-01-24 20:36:41 +0100275
Bram Moolenaar77073442016-02-13 23:23:53 +0100276 channel->ch_timeout = 2000;
277
278 if (first_channel != NULL)
279 {
280 first_channel->ch_prev = channel;
281 channel->ch_next = first_channel;
282 }
283 first_channel = channel;
284
285 channel->ch_refcount = 1;
286 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100287}
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100288
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100289/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100290 * Close a channel and free all its resources.
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100291 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100292 void
293channel_free(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100294{
Bram Moolenaar77073442016-02-13 23:23:53 +0100295 channel_close(channel);
296 if (channel->ch_next != NULL)
297 channel->ch_next->ch_prev = channel->ch_prev;
298 if (channel->ch_prev == NULL)
299 first_channel = channel->ch_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100300 else
Bram Moolenaar77073442016-02-13 23:23:53 +0100301 channel->ch_prev->ch_next = channel->ch_next;
302 vim_free(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100303}
304
Bram Moolenaar77073442016-02-13 23:23:53 +0100305#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
306 static channel_T *
307channel_from_id(int id)
308{
309 channel_T *channel;
310
311 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
312 if (channel->ch_id == id)
313 return channel;
314 return NULL;
315}
316#endif
317
Bram Moolenaard04a0202016-01-26 23:30:18 +0100318#if defined(FEAT_GUI) || defined(PROTO)
Bram Moolenaar77073442016-02-13 23:23:53 +0100319
320#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
321 static void
322channel_read_netbeans(int id)
323{
324 channel_T *channel = channel_from_id(id);
325
326 if (channel == NULL)
327 ch_errorn(NULL, "Channel %d not found", id);
328 else
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100329 channel_read(channel, -1, "messageFromNetbeans");
Bram Moolenaar77073442016-02-13 23:23:53 +0100330}
331#endif
332
Bram Moolenaare0874f82016-01-24 20:36:41 +0100333/*
Bram Moolenaard04a0202016-01-26 23:30:18 +0100334 * Read a command from netbeans.
Bram Moolenaare0874f82016-01-24 20:36:41 +0100335 */
Bram Moolenaard04a0202016-01-26 23:30:18 +0100336#ifdef FEAT_GUI_X11
337 static void
338messageFromNetbeans(XtPointer clientData,
339 int *unused1 UNUSED,
340 XtInputId *unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100341{
Bram Moolenaar77073442016-02-13 23:23:53 +0100342 channel_read_netbeans((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100343}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100344#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100345
Bram Moolenaard04a0202016-01-26 23:30:18 +0100346#ifdef FEAT_GUI_GTK
347 static void
348messageFromNetbeans(gpointer clientData,
349 gint unused1 UNUSED,
350 GdkInputCondition unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100351{
Bram Moolenaar77073442016-02-13 23:23:53 +0100352 channel_read_netbeans((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100353}
354#endif
355
356 static void
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100357channel_gui_register_one(channel_T *channel, int which)
358{
359# ifdef FEAT_GUI_X11
360 /* Tell notifier we are interested in being called
361 * when there is input on the editor connection socket. */
362 if (channel->ch_pfd[which].ch_inputHandler == (XtInputId)NULL)
363 channel->ch_pfd[which].ch_inputHandler = XtAppAddInput(
364 (XtAppContext)app_context,
365 channel->ch_pfd[which].ch_fd,
366 (XtPointer)(XtInputReadMask + XtInputExceptMask),
367 messageFromNetbeans,
368 (XtPointer)(long)channel->ch_id);
369# else
370# ifdef FEAT_GUI_GTK
371 /* Tell gdk we are interested in being called when there
372 * is input on the editor connection socket. */
373 if (channel->ch_pfd[which].ch_inputHandler == 0)
374 channel->ch_pfd[which].ch_inputHandler = gdk_input_add(
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100375 (gint)channel->ch_pfd[which].ch_fd,
376 (GdkInputCondition)
377 ((int)GDK_INPUT_READ + (int)GDK_INPUT_EXCEPTION),
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100378 messageFromNetbeans,
379 (gpointer)(long)channel->ch_id);
380# else
381# ifdef FEAT_GUI_W32
382 /* Tell Windows we are interested in receiving message when there
383 * is input on the editor connection socket. */
384 if (channel->ch_pfd[which].ch_inputHandler == -1)
385 channel->ch_pfd[which].ch_inputHandler = WSAAsyncSelect(
386 channel->ch_pfd[which].ch_fd,
387 s_hwnd, WM_NETBEANS, FD_READ);
388# endif
389# endif
390# endif
391}
392
393 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100394channel_gui_register(channel_T *channel)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100395{
Bram Moolenaard04a0202016-01-26 23:30:18 +0100396 if (!CH_HAS_GUI)
397 return;
398
Bram Moolenaard8070362016-02-15 21:56:54 +0100399 if (channel->CH_SOCK != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100400 channel_gui_register_one(channel, CHAN_SOCK);
401# ifdef CHANNEL_PIPES
Bram Moolenaard8070362016-02-15 21:56:54 +0100402 if (channel->CH_OUT != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100403 channel_gui_register_one(channel, CHAN_OUT);
Bram Moolenaard8070362016-02-15 21:56:54 +0100404 if (channel->CH_ERR != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100405 channel_gui_register_one(channel, CHAN_ERR);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100406# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100407}
408
409/*
410 * Register any of our file descriptors with the GUI event handling system.
411 * Called when the GUI has started.
412 */
413 void
414channel_gui_register_all(void)
415{
Bram Moolenaar77073442016-02-13 23:23:53 +0100416 channel_T *channel;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100417
Bram Moolenaar77073442016-02-13 23:23:53 +0100418 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100419 channel_gui_register(channel);
420}
421
422 static void
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100423channel_gui_unregister(channel_T *channel)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100424{
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100425 int which;
426
427#ifdef CHANNEL_PIPES
428 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
429#else
430 which = CHAN_SOCK;
431#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100432 {
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100433# ifdef FEAT_GUI_X11
434 if (channel->ch_pfd[which].ch_inputHandler != (XtInputId)NULL)
435 {
436 XtRemoveInput(channel->ch_pfd[which].ch_inputHandler);
437 channel->ch_pfd[which].ch_inputHandler = (XtInputId)NULL;
438 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100439# else
440# ifdef FEAT_GUI_GTK
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100441 if (channel->ch_pfd[which].ch_inputHandler != 0)
442 {
443 gdk_input_remove(channel->ch_pfd[which].ch_inputHandler);
444 channel->ch_pfd[which].ch_inputHandler = 0;
445 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100446# else
447# ifdef FEAT_GUI_W32
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100448 if (channel->ch_pfd[which].ch_inputHandler == 0)
449 {
450 WSAAsyncSelect(channel->ch_pfd[which].ch_fd, s_hwnd, 0, 0);
451 channel->ch_pfd[which].ch_inputHandler = -1;
452 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100453# endif
454# endif
455# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100456 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100457}
458
459#endif
460
461/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100462 * Open a socket channel to "hostname":"port".
Bram Moolenaar77073442016-02-13 23:23:53 +0100463 * Returns the channel for success.
464 * Returns NULL for failure.
Bram Moolenaard04a0202016-01-26 23:30:18 +0100465 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100466 channel_T *
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100467channel_open(char *hostname, int port_in, int waittime, void (*close_cb)(void))
Bram Moolenaard04a0202016-01-26 23:30:18 +0100468{
469 int sd;
470 struct sockaddr_in server;
471 struct hostent * host;
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100472#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100473 u_short port = port_in;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100474 u_long val = 1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100475#else
476 int port = port_in;
477#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100478 channel_T *channel;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100479 int ret;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100480
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100481#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100482 channel_init_winsock();
483#endif
484
Bram Moolenaar77073442016-02-13 23:23:53 +0100485 channel = add_channel();
486 if (channel == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100487 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100488 ch_error(NULL, "Cannot allocate channel.\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100489 EMSG(_("E897: All channels are in use"));
Bram Moolenaar77073442016-02-13 23:23:53 +0100490 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100491 }
492
Bram Moolenaard8070362016-02-15 21:56:54 +0100493 if ((sd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100494 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100495 ch_error(NULL, "in socket() in channel_open().\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100496 PERROR("E898: socket() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100497 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100498 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100499 }
500
501 /* Get the server internet address and put into addr structure */
502 /* fill in the socket address structure and connect to server */
503 vim_memset((char *)&server, 0, sizeof(server));
504 server.sin_family = AF_INET;
505 server.sin_port = htons(port);
506 if ((host = gethostbyname(hostname)) == NULL)
507 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100508 ch_error(NULL, "in gethostbyname() in channel_open()\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100509 PERROR("E901: gethostbyname() in channel_open()");
Bram Moolenaard04a0202016-01-26 23:30:18 +0100510 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100511 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100512 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100513 }
514 memcpy((char *)&server.sin_addr, host->h_addr, host->h_length);
515
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100516 if (waittime >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100517 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100518 /* Make connect non-blocking. */
519 if (
520#ifdef _WIN32
521 ioctlsocket(sd, FIONBIO, &val) < 0
522#else
523 fcntl(sd, F_SETFL, O_NONBLOCK) < 0
524#endif
525 )
Bram Moolenaard04a0202016-01-26 23:30:18 +0100526 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100527 SOCK_ERRNO;
Bram Moolenaar77073442016-02-13 23:23:53 +0100528 ch_errorn(NULL, "channel_open: Connect failed with errno %d\n",
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100529 errno);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100530 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100531 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100532 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100533 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100534 }
535
536 /* Try connecting to the server. */
Bram Moolenaar77073442016-02-13 23:23:53 +0100537 ch_logsn(NULL, "Connecting to %s port %d", hostname, port);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100538 ret = connect(sd, (struct sockaddr *)&server, sizeof(server));
539 SOCK_ERRNO;
540 if (ret < 0)
541 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100542 if (errno != EWOULDBLOCK
543#ifdef EINPROGRESS
544 && errno != EINPROGRESS
545#endif
546 )
Bram Moolenaard04a0202016-01-26 23:30:18 +0100547 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100548 ch_errorn(NULL, "channel_open: Connect failed with errno %d\n",
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100549 errno);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100550 PERROR(_("E902: Cannot connect to port"));
Bram Moolenaard04a0202016-01-26 23:30:18 +0100551 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100552 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100553 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100554 }
555 }
556
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100557 if (waittime >= 0 && ret < 0)
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100558 {
559 struct timeval tv;
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100560 fd_set wfds;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100561
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100562 FD_ZERO(&wfds);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100563 FD_SET(sd, &wfds);
Bram Moolenaar26dfc412016-02-06 18:18:54 +0100564 tv.tv_sec = waittime / 1000;
565 tv.tv_usec = (waittime % 1000) * 1000;
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100566 ret = select((int)sd + 1, NULL, &wfds, NULL, &tv);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100567 if (ret < 0)
568 {
569 SOCK_ERRNO;
Bram Moolenaar77073442016-02-13 23:23:53 +0100570 ch_errorn(NULL, "channel_open: Connect failed with errno %d\n",
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100571 errno);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100572 PERROR(_("E902: Cannot connect to port"));
573 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100574 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100575 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100576 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100577 if (!FD_ISSET(sd, &wfds))
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100578 {
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100579 /* don't give an error, we just timed out. */
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100580 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100581 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100582 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100583 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100584 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100585
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100586 if (waittime >= 0)
587 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100588#ifdef _WIN32
589 val = 0;
590 ioctlsocket(sd, FIONBIO, &val);
591#else
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100592 (void)fcntl(sd, F_SETFL, 0);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100593#endif
594 }
595
Bram Moolenaar0fa98e72016-02-07 22:21:19 +0100596 /* Only retry for netbeans. TODO: can we use a waittime instead? */
597 if (errno == ECONNREFUSED && close_cb != NULL)
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100598 {
599 sock_close(sd);
Bram Moolenaard8070362016-02-15 21:56:54 +0100600 if ((sd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100601 {
602 SOCK_ERRNO;
Bram Moolenaar77073442016-02-13 23:23:53 +0100603 ch_log(NULL, "socket() retry in channel_open()\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100604 PERROR("E900: socket() retry in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100605 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100606 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100607 }
608 if (connect(sd, (struct sockaddr *)&server, sizeof(server)))
609 {
610 int retries = 36;
611 int success = FALSE;
612
613 SOCK_ERRNO;
614 while (retries-- && ((errno == ECONNREFUSED)
615 || (errno == EINTR)))
616 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100617 ch_log(NULL, "retrying...\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100618 mch_delay(3000L, TRUE);
619 ui_breakcheck();
620 if (got_int)
621 {
622 errno = EINTR;
623 break;
624 }
625 if (connect(sd, (struct sockaddr *)&server,
626 sizeof(server)) == 0)
627 {
628 success = TRUE;
629 break;
630 }
631 SOCK_ERRNO;
632 }
633 if (!success)
634 {
635 /* Get here when the server can't be found. */
Bram Moolenaar77073442016-02-13 23:23:53 +0100636 ch_error(NULL, "Cannot connect to port after retry\n");
Bram Moolenaar5cefd402016-02-16 12:44:26 +0100637 PERROR(_("E899: Cannot connect to port after retry"));
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100638 sock_close(sd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100639 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100640 return NULL;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100641 }
642 }
643 }
644
Bram Moolenaard8070362016-02-15 21:56:54 +0100645 channel->CH_SOCK = (sock_T)sd;
Bram Moolenaar77073442016-02-13 23:23:53 +0100646 channel->ch_close_cb = close_cb;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100647
648#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +0100649 channel_gui_register(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100650#endif
651
Bram Moolenaar77073442016-02-13 23:23:53 +0100652 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100653}
654
655#if defined(CHANNEL_PIPES) || defined(PROTO)
656 void
Bram Moolenaard8070362016-02-15 21:56:54 +0100657channel_set_pipes(channel_T *channel, sock_T in, sock_T out, sock_T err)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100658{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100659 channel->CH_IN = in;
660 channel->CH_OUT = out;
661 channel->CH_ERR = err;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100662}
663#endif
664
665 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100666channel_set_job(channel_T *channel, job_T *job)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100667{
Bram Moolenaar77073442016-02-13 23:23:53 +0100668 channel->ch_job = job;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100669}
670
671/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100672 * Set the json mode of channel "channel" to "ch_mode".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100673 */
674 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100675channel_set_json_mode(channel_T *channel, ch_mode_T ch_mode)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100676{
Bram Moolenaar77073442016-02-13 23:23:53 +0100677 channel->ch_mode = ch_mode;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100678}
679
680/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100681 * Set the read timeout of channel "channel".
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100682 */
683 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100684channel_set_timeout(channel_T *channel, int timeout)
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100685{
Bram Moolenaar77073442016-02-13 23:23:53 +0100686 channel->ch_timeout = timeout;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100687}
688
689/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100690 * Set the callback for channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100691 */
692 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100693channel_set_callback(channel_T *channel, char_u *callback)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100694{
Bram Moolenaar77073442016-02-13 23:23:53 +0100695 vim_free(channel->ch_callback);
696 channel->ch_callback = vim_strsave(callback);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100697}
698
699/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100700 * Set the callback for channel "channel" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100701 */
702 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100703channel_set_req_callback(channel_T *channel, char_u *callback, int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100704{
Bram Moolenaar77073442016-02-13 23:23:53 +0100705 cbq_T *head = &channel->ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100706 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
707
708 if (item != NULL)
709 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100710 item->cq_callback = vim_strsave(callback);
711 item->cq_seq_nr = id;
712 item->cq_prev = head->cq_prev;
713 head->cq_prev = item;
714 item->cq_next = NULL;
715 if (item->cq_prev == NULL)
716 head->cq_next = item;
717 else
718 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100719 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100720}
721
722/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100723 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100724 */
725 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100726invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100727{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100728 typval_T rettv;
729 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100730
Bram Moolenaar77073442016-02-13 23:23:53 +0100731 argv[0].v_type = VAR_CHANNEL;
732 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100733
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100734 call_func(callback, (int)STRLEN(callback),
735 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
736 /* If an echo command was used the cursor needs to be put back where
737 * it belongs. */
738 setcursor();
739 cursor_on();
740 out_flush();
741}
742
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100743/*
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100744 * Return the first buffer from the channel and remove it.
745 * The caller must free it.
746 * Returns NULL if there is nothing.
747 */
748 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +0100749channel_get(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100750{
Bram Moolenaar77073442016-02-13 23:23:53 +0100751 readq_T *head = &channel->ch_head;
752 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100753 char_u *p;
754
Bram Moolenaar77073442016-02-13 23:23:53 +0100755 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100756 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100757 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100758 p = node->rq_buffer;
759 head->rq_next = node->rq_next;
760 if (node->rq_next == NULL)
761 head->rq_prev = NULL;
762 else
763 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100764 vim_free(node);
765 return p;
766}
767
768/*
769 * Returns the whole buffer contents concatenated.
770 */
771 static char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +0100772channel_get_all(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100773{
774 /* Concatenate everything into one buffer.
775 * TODO: avoid multiple allocations. */
Bram Moolenaar77073442016-02-13 23:23:53 +0100776 while (channel_collapse(channel) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100777 ;
Bram Moolenaar77073442016-02-13 23:23:53 +0100778 return channel_get(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100779}
780
781/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100782 * Collapses the first and second buffer in the channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100783 * Returns FAIL if that is not possible.
784 */
785 int
Bram Moolenaar77073442016-02-13 23:23:53 +0100786channel_collapse(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100787{
Bram Moolenaar77073442016-02-13 23:23:53 +0100788 readq_T *head = &channel->ch_head;
789 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100790 char_u *p;
791
Bram Moolenaar77073442016-02-13 23:23:53 +0100792 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100793 return FAIL;
794
Bram Moolenaar77073442016-02-13 23:23:53 +0100795 p = alloc((unsigned)(STRLEN(node->rq_buffer)
796 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100797 if (p == NULL)
798 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100799 STRCPY(p, node->rq_buffer);
800 STRCAT(p, node->rq_next->rq_buffer);
801 vim_free(node->rq_next->rq_buffer);
802 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100803
Bram Moolenaar77073442016-02-13 23:23:53 +0100804 /* dispose of the node and its buffer */
805 head->rq_next = node->rq_next;
806 head->rq_next->rq_prev = NULL;
807 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100808 vim_free(node);
809 return OK;
810}
811
812/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100813 * Use the read buffer of channel "channel" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100814 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +0100815 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100816 */
Bram Moolenaard7ece102016-02-02 23:23:02 +0100817 static int
Bram Moolenaar77073442016-02-13 23:23:53 +0100818channel_parse_json(channel_T *channel)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100819{
820 js_read_T reader;
821 typval_T listtv;
822 jsonq_T *item;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100823 jsonq_T *head = &channel->ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100824 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100825
Bram Moolenaar77073442016-02-13 23:23:53 +0100826 if (channel_peek(channel) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100827 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100828
829 /* TODO: make reader work properly */
Bram Moolenaar77073442016-02-13 23:23:53 +0100830 /* reader.js_buf = channel_peek(channel); */
831 reader.js_buf = channel_get_all(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100832 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +0100833 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100834 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +0100835 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100836 ret = json_decode(&reader, &listtv,
837 channel->ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100838 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100839 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +0100840 /* Only accept the response when it is a list with at least two
841 * items. */
842 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100843 {
844 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100845 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100846 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100847 else
848 {
Bram Moolenaard7ece102016-02-02 23:23:02 +0100849 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
850 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100851 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100852 else
853 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100854 item->jq_value = alloc_tv();
855 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100856 {
857 vim_free(item);
858 clear_tv(&listtv);
859 }
860 else
861 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100862 *item->jq_value = listtv;
863 item->jq_prev = head->jq_prev;
864 head->jq_prev = item;
865 item->jq_next = NULL;
866 if (item->jq_prev == NULL)
867 head->jq_next = item;
868 else
869 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100870 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100871 }
872 }
873 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +0100874
875 /* Put the unread part back into the channel.
876 * TODO: insert in front */
877 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100878 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100879 channel_save(channel, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +0100880 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100881 ret = TRUE;
882 }
883 else
884 ret = FALSE;
885
Bram Moolenaardf5b27b2016-02-02 18:43:17 +0100886 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100887 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100888}
889
890/*
Bram Moolenaard46ae142016-02-16 13:33:52 +0100891 * Remove "node" from the queue that it is in. Does not free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +0100892 */
893 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100894remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +0100895{
Bram Moolenaar77073442016-02-13 23:23:53 +0100896 if (node->cq_prev == NULL)
897 head->cq_next = node->cq_next;
898 else
899 node->cq_prev->cq_next = node->cq_next;
900 if (node->cq_next == NULL)
901 head->cq_prev = node->cq_prev;
902 else
903 node->cq_next->cq_prev = node->cq_prev;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100904}
905
906/*
907 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +0100908 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100909 */
910 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100911remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100912{
Bram Moolenaar77073442016-02-13 23:23:53 +0100913 if (node->jq_prev == NULL)
914 head->jq_next = node->jq_next;
915 else
916 node->jq_prev->jq_next = node->jq_next;
917 if (node->jq_next == NULL)
918 head->jq_prev = node->jq_prev;
919 else
920 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100921 vim_free(node);
922}
923
924/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100925 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100926 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +0100927 * When "id" is zero or negative jut get the first message. But not the one
928 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100929 * Return OK when found and return the value in "rettv".
930 * Return FAIL otherwise.
931 */
932 static int
Bram Moolenaar77073442016-02-13 23:23:53 +0100933channel_get_json(channel_T *channel, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100934{
Bram Moolenaare56bf152016-02-08 23:23:42 +0100935 jsonq_T *head = &channel->ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100936 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100937
Bram Moolenaar77073442016-02-13 23:23:53 +0100938 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100939 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100940 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100941 typval_T *tv = &l->lv_first->li_tv;
942
943 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +0100944 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaarf6157282016-02-10 21:07:14 +0100945 || tv->vval.v_number == 0
Bram Moolenaare56bf152016-02-08 23:23:42 +0100946 || tv->vval.v_number != channel->ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100947 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100948 *rettv = item->jq_value;
949 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100950 return OK;
951 }
Bram Moolenaar77073442016-02-13 23:23:53 +0100952 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100953 }
954 return FAIL;
955}
956
957/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100958 * Execute a command received over channel "channel".
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100959 * "cmd" is the command string, "arg2" the second argument.
960 * "arg3" is the third argument, NULL if missing.
961 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100962 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100963channel_exe_cmd(channel_T *channel, char_u *cmd, typval_T *arg2, typval_T *arg3)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100964{
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100965 char_u *arg;
966
967 if (arg2->v_type != VAR_STRING)
968 {
969 if (p_verbose > 2)
970 EMSG("E903: received ex command with non-string argument");
971 return;
972 }
973 arg = arg2->vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +0100974 if (arg == NULL)
975 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100976
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100977 if (STRCMP(cmd, "ex") == 0)
978 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100979 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100980 }
981 else if (STRCMP(cmd, "normal") == 0)
982 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100983 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100984
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100985 ea.arg = arg;
986 ea.addr_count = 0;
987 ea.forceit = TRUE; /* no mapping */
988 ex_normal(&ea);
989 }
990 else if (STRCMP(cmd, "redraw") == 0)
991 {
992 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100993
Bram Moolenaar14ad6112016-02-01 21:47:13 +0100994 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100995 ex_redraw(&ea);
996 showruler(FALSE);
997 setcursor();
998 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100999#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001000 if (gui.in_use)
1001 {
1002 gui_update_cursor(FALSE, FALSE);
1003 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001004 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001005#endif
1006 }
1007 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "eval") == 0)
1008 {
1009 int is_eval = cmd[1] == 'v';
1010
Bram Moolenaar66624ff2016-02-03 23:59:43 +01001011 if (is_eval && (arg3 == NULL || arg3->v_type != VAR_NUMBER))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001012 {
1013 if (p_verbose > 2)
1014 EMSG("E904: third argument for eval must be a number");
1015 }
1016 else
1017 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001018 typval_T *tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001019 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +01001020 char_u *json = NULL;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001021 int options = channel->ch_mode == MODE_JS ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001022
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001023 /* Don't pollute the display with errors. */
1024 ++emsg_skip;
1025 tv = eval_expr(arg, NULL);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001026 if (is_eval)
1027 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001028 if (tv != NULL)
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001029 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1030 options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001031 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001032 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001033 /* If evaluation failed or the result can't be encoded
1034 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001035 vim_free(json);
1036 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001037 err_tv.v_type = VAR_STRING;
1038 err_tv.vval.v_string = (char_u *)"ERROR";
1039 tv = &err_tv;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001040 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1041 options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001042 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001043 if (json != NULL)
1044 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001045 channel_send(channel, json, "eval");
Bram Moolenaar55fab432016-02-07 16:53:13 +01001046 vim_free(json);
1047 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001048 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001049 --emsg_skip;
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001050 if (tv != &err_tv)
1051 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001052 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001053 }
1054 else if (p_verbose > 2)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001055 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001056}
1057
1058/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001059 * Invoke a callback for channel "channel" if needed.
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001060 * Return OK when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001061 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001062 static int
Bram Moolenaar77073442016-02-13 23:23:53 +01001063may_invoke_callback(channel_T *channel)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001064{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001065 char_u *msg = NULL;
1066 typval_T *listtv = NULL;
1067 list_T *list;
1068 typval_T *typetv;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001069 typval_T argv[3];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001070 int seq_nr = -1;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001071 ch_mode_T ch_mode = channel->ch_mode;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001072
Bram Moolenaara07fec92016-02-05 21:04:08 +01001073 if (channel->ch_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001074 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001075 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001076
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001077 if (ch_mode != MODE_RAW)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001078 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001079 /* Get any json message in the queue. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001080 if (channel_get_json(channel, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001081 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001082 /* Parse readahead, return when there is still no message. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001083 channel_parse_json(channel);
1084 if (channel_get_json(channel, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001085 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001086 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001087
1088 list = listtv->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001089 argv[1] = list->lv_first->li_next->li_tv;
1090 typetv = &list->lv_first->li_tv;
1091 if (typetv->v_type == VAR_STRING)
1092 {
1093 typval_T *arg3 = NULL;
1094 char_u *cmd = typetv->vval.v_string;
1095
Bram Moolenaar66624ff2016-02-03 23:59:43 +01001096 /* ["cmd", arg] or ["cmd", arg, arg] */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001097 if (list->lv_len == 3)
1098 arg3 = &list->lv_last->li_tv;
Bram Moolenaar77073442016-02-13 23:23:53 +01001099 ch_logs(channel, "Executing %s command", (char *)cmd);
1100 channel_exe_cmd(channel, cmd, &argv[1], arg3);
1101 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001102 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001103 }
1104
1105 if (typetv->v_type != VAR_NUMBER)
1106 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001107 ch_error(channel,
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001108 "Dropping message with invalid sequence number type\n");
Bram Moolenaar77073442016-02-13 23:23:53 +01001109 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001110 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001111 }
1112 seq_nr = typetv->vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001113 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001114 else if (channel_peek(channel) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001115 {
1116 /* nothing to read on raw channel */
1117 return FALSE;
1118 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001119 else
1120 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001121 /* If there is no callback, don't do anything. */
1122 if (channel->ch_callback == NULL)
1123 return FALSE;
1124
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001125 /* For a raw channel we don't know where the message ends, just get
1126 * everything. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001127 msg = channel_get_all(channel);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001128 argv[1].v_type = VAR_STRING;
1129 argv[1].vval.v_string = msg;
1130 }
1131
Bram Moolenaara07fec92016-02-05 21:04:08 +01001132 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001133 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001134 cbq_T *head = &channel->ch_cb_head;
1135 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001136 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001137
1138 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001139 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001140 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001141 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001142 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001143 ch_log(channel, "Invoking one-time callback\n");
Bram Moolenaard46ae142016-02-16 13:33:52 +01001144 /* Remove the item from the list first, if the callback
1145 * invokes ch_close() the list will be cleared. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001146 remove_cb_node(head, item);
Bram Moolenaard46ae142016-02-16 13:33:52 +01001147 invoke_callback(channel, item->cq_callback, argv);
1148 vim_free(item->cq_callback);
1149 vim_free(item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001150 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001151 break;
1152 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001153 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001154 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001155 if (!done)
Bram Moolenaar77073442016-02-13 23:23:53 +01001156 ch_log(channel, "Dropping message without callback\n");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001157 }
Bram Moolenaara07fec92016-02-05 21:04:08 +01001158 else if (channel->ch_callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001159 {
1160 /* invoke the channel callback */
Bram Moolenaar77073442016-02-13 23:23:53 +01001161 ch_log(channel, "Invoking channel callback\n");
1162 invoke_callback(channel, channel->ch_callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001163 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001164 else
Bram Moolenaar77073442016-02-13 23:23:53 +01001165 ch_log(channel, "Dropping message\n");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001166
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001167 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001168 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001169 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001170
1171 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001172}
1173
1174/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001175 * Return TRUE when channel "channel" is open for writing to.
1176 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001177 */
1178 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001179channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001180{
Bram Moolenaard8070362016-02-15 21:56:54 +01001181 return channel != NULL && (channel->CH_SOCK != CHAN_FD_INVALID
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001182#ifdef CHANNEL_PIPES
Bram Moolenaard8070362016-02-15 21:56:54 +01001183 || channel->CH_IN != CHAN_FD_INVALID
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001184#endif
1185 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001186}
1187
1188/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001189 * Return TRUE when channel "channel" is open for reading or writing.
1190 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001191 */
1192 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001193channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001194{
Bram Moolenaard8070362016-02-15 21:56:54 +01001195 return channel != NULL && (channel->CH_SOCK != CHAN_FD_INVALID
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001196#ifdef CHANNEL_PIPES
Bram Moolenaard8070362016-02-15 21:56:54 +01001197 || channel->CH_IN != CHAN_FD_INVALID
1198 || channel->CH_OUT != CHAN_FD_INVALID
1199 || channel->CH_ERR != CHAN_FD_INVALID
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001200#endif
1201 );
1202}
1203
1204/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001205 * Return a string indicating the status of the channel.
1206 */
1207 char *
1208channel_status(channel_T *channel)
1209{
1210 if (channel == NULL)
1211 return "fail";
1212 if (channel_is_open(channel))
1213 return "open";
1214 return "closed";
1215}
1216
1217/*
1218 * Close channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001219 * This does not trigger the close callback.
1220 */
1221 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001222channel_close(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001223{
Bram Moolenaar5cefd402016-02-16 12:44:26 +01001224 ch_log(channel, "Closing channel\n");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001225
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001226#ifdef FEAT_GUI
1227 channel_gui_unregister(channel);
1228#endif
1229
Bram Moolenaard8070362016-02-15 21:56:54 +01001230 if (channel->CH_SOCK != CHAN_FD_INVALID)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001231 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001232 sock_close(channel->CH_SOCK);
Bram Moolenaard8070362016-02-15 21:56:54 +01001233 channel->CH_SOCK = CHAN_FD_INVALID;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001234 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001235#if defined(CHANNEL_PIPES)
Bram Moolenaard8070362016-02-15 21:56:54 +01001236 if (channel->CH_IN != CHAN_FD_INVALID)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001237 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001238 fd_close(channel->CH_IN);
1239 channel->CH_IN = CHAN_FD_INVALID;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001240 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001241 if (channel->CH_OUT != CHAN_FD_INVALID)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001242 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001243 fd_close(channel->CH_OUT);
1244 channel->CH_OUT = CHAN_FD_INVALID;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001245 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001246 if (channel->CH_ERR != CHAN_FD_INVALID)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001247 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001248 fd_close(channel->CH_ERR);
1249 channel->CH_ERR = CHAN_FD_INVALID;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001250 }
1251#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001252
1253 channel->ch_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001254 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001255}
1256
Bram Moolenaard04a0202016-01-26 23:30:18 +01001257/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001258 * Store "buf[len]" on channel "channel".
Bram Moolenaar83162462016-01-28 23:10:07 +01001259 * Returns OK or FAIL.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001260 */
Bram Moolenaar83162462016-01-28 23:10:07 +01001261 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001262channel_save(channel_T *channel, char_u *buf, int len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001263{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001264 readq_T *node;
Bram Moolenaar77073442016-02-13 23:23:53 +01001265 readq_T *head = &channel->ch_head;
Bram Moolenaar23689172016-02-15 22:37:37 +01001266 char_u *p;
1267 int i;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001268
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001269 node = (readq_T *)alloc(sizeof(readq_T));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001270 if (node == NULL)
Bram Moolenaar83162462016-01-28 23:10:07 +01001271 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +01001272 node->rq_buffer = alloc(len + 1);
1273 if (node->rq_buffer == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001274 {
1275 vim_free(node);
Bram Moolenaar83162462016-01-28 23:10:07 +01001276 return FAIL; /* out of memory */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001277 }
Bram Moolenaar23689172016-02-15 22:37:37 +01001278
1279 /* TODO: don't strip CR when channel is in raw mode */
1280 p = node->rq_buffer;
1281 for (i = 0; i < len; ++i)
1282 if (buf[i] != CAR || i + 1 >= len || buf[i + 1] != NL)
1283 *p++ = buf[i];
1284 *p = NUL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001285
Bram Moolenaar77073442016-02-13 23:23:53 +01001286 /* append node to the tail of the queue */
1287 node->rq_next = NULL;
1288 node->rq_prev = head->rq_prev;
1289 if (head->rq_prev == NULL)
1290 head->rq_next = node;
1291 else
1292 head->rq_prev->rq_next = node;
1293 head->rq_prev = node;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001294
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001295 if (log_fd != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001296 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001297 ch_log_lead("RECV ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001298 fprintf(log_fd, "'");
1299 if (fwrite(buf, len, 1, log_fd) != 1)
Bram Moolenaar83162462016-01-28 23:10:07 +01001300 return FAIL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001301 fprintf(log_fd, "'\n");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001302 }
Bram Moolenaar83162462016-01-28 23:10:07 +01001303 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001304}
1305
1306/*
1307 * Return the first buffer from the channel without removing it.
1308 * Returns NULL if there is nothing.
1309 */
1310 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +01001311channel_peek(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001312{
Bram Moolenaar77073442016-02-13 23:23:53 +01001313 readq_T *head = &channel->ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001314
Bram Moolenaar77073442016-02-13 23:23:53 +01001315 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001316 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001317 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001318}
1319
1320/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001321 * Clear the read buffer on channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001322 */
1323 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001324channel_clear(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001325{
Bram Moolenaar77073442016-02-13 23:23:53 +01001326 jsonq_T *json_head = &channel->ch_json_head;
1327 cbq_T *cb_head = &channel->ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001328
Bram Moolenaar77073442016-02-13 23:23:53 +01001329 while (channel_peek(channel) != NULL)
1330 vim_free(channel_get(channel));
1331
1332 while (cb_head->cq_next != NULL)
Bram Moolenaard46ae142016-02-16 13:33:52 +01001333 {
1334 cbq_T *node = cb_head->cq_next;
1335
1336 remove_cb_node(cb_head, node);
1337 vim_free(node->cq_callback);
1338 vim_free(node);
1339 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001340
1341 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001342 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001343 free_tv(json_head->jq_next->jq_value);
1344 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001345 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001346
1347 vim_free(channel->ch_callback);
1348 channel->ch_callback = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001349}
1350
Bram Moolenaar77073442016-02-13 23:23:53 +01001351#if defined(EXITFREE) || defined(PROTO)
1352 void
1353channel_free_all(void)
1354{
1355 channel_T *channel;
1356
1357 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1358 channel_clear(channel);
1359}
1360#endif
1361
1362
Bram Moolenaard04a0202016-01-26 23:30:18 +01001363/* Sent when the channel is found closed when reading. */
1364#define DETACH_MSG "\"DETACH\"\n"
1365
1366/* Buffer size for reading incoming messages. */
1367#define MAXMSGSIZE 4096
1368
1369/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001370 * Check for reading from "fd" with "timeout" msec.
1371 * Return FAIL when there is nothing to read.
Bram Moolenaara8343c12016-02-04 22:09:48 +01001372 * Always returns OK for FEAT_GUI_W32.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001373 */
1374 static int
Bram Moolenaard8070362016-02-15 21:56:54 +01001375channel_wait(channel_T *channel, sock_T fd, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001376{
Bram Moolenaara8343c12016-02-04 22:09:48 +01001377#if defined(HAVE_SELECT) && !defined(FEAT_GUI_W32)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001378 struct timeval tval;
1379 fd_set rfds;
1380 int ret;
1381
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001382 if (timeout > 0)
Bram Moolenaar77073442016-02-13 23:23:53 +01001383 ch_logn(channel, "Waiting for %d msec\n", timeout);
Bram Moolenaard8070362016-02-15 21:56:54 +01001384
1385
1386# ifdef WIN32
1387 if (channel->CH_SOCK == CHAN_FD_INVALID)
1388 {
1389 DWORD nread;
1390 int diff;
1391 DWORD deadline = GetTickCount() + timeout;
1392
1393 /* reading from a pipe, not a socket */
1394 while (TRUE)
1395 {
1396 if (PeekNamedPipe(fd, NULL, 0, NULL, &nread, NULL) && nread > 0)
1397 return OK;
1398 diff = deadline - GetTickCount();
1399 if (diff < 0)
1400 break;
1401 /* Wait for 5 msec.
1402 * TODO: increase the sleep time when looping more often */
1403 Sleep(5);
1404 }
1405 return FAIL;
1406 }
1407#endif
1408
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001409 FD_ZERO(&rfds);
Bram Moolenaard8070362016-02-15 21:56:54 +01001410 FD_SET((int)fd, &rfds);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001411 tval.tv_sec = timeout / 1000;
1412 tval.tv_usec = (timeout % 1000) * 1000;
1413 for (;;)
1414 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001415 ret = select((int)fd + 1, &rfds, NULL, NULL, &tval);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001416# ifdef EINTR
1417 if (ret == -1 && errno == EINTR)
1418 continue;
1419# endif
1420 if (ret <= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001421 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001422 ch_log(channel, "Nothing to read\n");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001423 return FAIL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001424 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001425 break;
1426 }
1427#else
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001428# ifdef HAVE_POLL
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001429 struct pollfd fds;
1430
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001431 if (timeout > 0)
Bram Moolenaar77073442016-02-13 23:23:53 +01001432 ch_logn(channel, "Waiting for %d msec\n", timeout);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001433 fds.fd = fd;
1434 fds.events = POLLIN;
1435 if (poll(&fds, 1, timeout) <= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001436 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001437 ch_log(channel, "Nothing to read\n");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001438 return FAIL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001439 }
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001440# endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001441#endif
1442 return OK;
1443}
1444
1445/*
1446 * Return a unique ID to be used in a message.
1447 */
1448 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001449channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001450{
1451 static int next_id = 1;
1452
1453 return next_id++;
1454}
1455
1456/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001457 * Get the file descriptor to read from, either the socket or stdout.
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001458 * TODO: should have a way to read stderr.
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001459 */
Bram Moolenaard8070362016-02-15 21:56:54 +01001460 static sock_T
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001461get_read_fd(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001462{
Bram Moolenaard8070362016-02-15 21:56:54 +01001463 if (channel->CH_SOCK != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001464 return channel->CH_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001465#if defined(CHANNEL_PIPES)
Bram Moolenaard8070362016-02-15 21:56:54 +01001466 if (channel->CH_OUT != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001467 return channel->CH_OUT;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001468#endif
Bram Moolenaar77073442016-02-13 23:23:53 +01001469 ch_error(channel, "channel_read() called while socket is closed\n");
Bram Moolenaard8070362016-02-15 21:56:54 +01001470 return CHAN_FD_INVALID;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001471}
1472
1473/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001474 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001475 * "which" is CHAN_SOCK, CHAN_OUT or CHAN_ERR. When -1 use CHAN_SOCK or
1476 * CHAN_OUT, the one that is open.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001477 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001478 */
1479 void
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001480channel_read(channel_T *channel, int which, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001481{
1482 static char_u *buf = NULL;
1483 int len = 0;
1484 int readlen = 0;
Bram Moolenaard8070362016-02-15 21:56:54 +01001485 sock_T fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001486 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001487
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001488 if (which < 0)
1489 fd = get_read_fd(channel);
1490 else
1491 fd = channel->ch_pfd[which].ch_fd;
Bram Moolenaard8070362016-02-15 21:56:54 +01001492 if (fd == CHAN_FD_INVALID)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001493 return;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001494 use_socket = fd == channel->CH_SOCK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001495
1496 /* Allocate a buffer to read into. */
1497 if (buf == NULL)
1498 {
1499 buf = alloc(MAXMSGSIZE);
1500 if (buf == NULL)
1501 return; /* out of memory! */
1502 }
1503
1504 /* Keep on reading for as long as there is something to read.
1505 * Use select() or poll() to avoid blocking on a message that is exactly
1506 * MAXMSGSIZE long. */
1507 for (;;)
1508 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001509 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001510 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001511 if (use_socket)
1512 len = sock_read(fd, buf, MAXMSGSIZE);
1513 else
Bram Moolenaar0943a092016-02-16 13:11:17 +01001514 len = fd_read(fd, buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001515 if (len <= 0)
1516 break; /* error or nothing more to read */
1517
1518 /* Store the read message in the queue. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001519 channel_save(channel, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001520 readlen += len;
1521 if (len < MAXMSGSIZE)
1522 break; /* did read everything that's available */
1523 }
Bram Moolenaara8343c12016-02-04 22:09:48 +01001524#ifdef FEAT_GUI_W32
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001525 if (use_socket && len == SOCKET_ERROR)
Bram Moolenaara8343c12016-02-04 22:09:48 +01001526 {
1527 /* For Win32 GUI channel_wait() always returns OK and we handle the
1528 * situation that there is nothing to read here.
1529 * TODO: how about a timeout? */
1530 if (WSAGetLastError() == WSAEWOULDBLOCK)
1531 return;
1532 }
1533#endif
Bram Moolenaard04a0202016-01-26 23:30:18 +01001534
1535 /* Reading a socket disconnection (readlen == 0), or a socket error. */
1536 if (readlen <= 0)
1537 {
1538 /* Queue a "DETACH" netbeans message in the command queue in order to
1539 * terminate the netbeans session later. Do not end the session here
1540 * directly as we may be running in the context of a call to
1541 * netbeans_parse_messages():
1542 * netbeans_parse_messages
1543 * -> autocmd triggered while processing the netbeans cmd
1544 * -> ui_breakcheck
1545 * -> gui event loop or select loop
1546 * -> channel_read()
1547 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001548 ch_errors(channel, "%s(): Cannot read\n", func);
1549 channel_save(channel, (char_u *)DETACH_MSG, (int)STRLEN(DETACH_MSG));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001550
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001551 /* TODO: When reading from stdout is not possible, should we try to
1552 * keep stdin and stderr open? Probably not, assume the other side
1553 * has died. */
1554 channel_close(channel);
1555 if (channel->ch_close_cb != NULL)
1556 (*channel->ch_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001557
1558 if (len < 0)
1559 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001560 ch_error(channel, "channel_read(): cannot read from channel\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001561 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001562 }
1563 }
1564
1565#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001566 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001567 if (CH_HAS_GUI && gtk_main_level() > 0)
1568 gtk_main_quit();
1569#endif
1570}
1571
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001572/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001573 * Read from raw channel "channel". Blocks until there is something to read or
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001574 * the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001575 * Returns what was read in allocated memory.
1576 * Returns NULL in case of error or timeout.
1577 */
1578 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +01001579channel_read_block(channel_T *channel)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001580{
Bram Moolenaar77073442016-02-13 23:23:53 +01001581 ch_log(channel, "Reading raw\n");
1582 if (channel_peek(channel) == NULL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001583 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001584 sock_T fd = get_read_fd(channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001585
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001586 /* TODO: read both out and err if they are different */
Bram Moolenaar77073442016-02-13 23:23:53 +01001587 ch_log(channel, "No readahead\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001588 /* Wait for up to the channel timeout. */
Bram Moolenaard8070362016-02-15 21:56:54 +01001589 if (fd == CHAN_FD_INVALID
1590 || channel_wait(channel, fd, channel->ch_timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001591 return NULL;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001592 channel_read(channel, -1, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001593 }
1594
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001595 /* TODO: only get the first message */
Bram Moolenaar77073442016-02-13 23:23:53 +01001596 ch_log(channel, "Returning readahead\n");
1597 return channel_get_all(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001598}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001599
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001600/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001601 * Read one JSON message with ID "id" from channel "channel" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001602 * result in "rettv".
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001603 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001604 */
1605 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001606channel_read_json_block(channel_T *channel, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001607{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001608 int more;
Bram Moolenaard8070362016-02-15 21:56:54 +01001609 sock_T fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001610
Bram Moolenaar77073442016-02-13 23:23:53 +01001611 ch_log(channel, "Reading JSON\n");
Bram Moolenaare56bf152016-02-08 23:23:42 +01001612 channel->ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001613 for (;;)
1614 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001615 more = channel_parse_json(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001616
1617 /* search for messsage "id" */
Bram Moolenaar77073442016-02-13 23:23:53 +01001618 if (channel_get_json(channel, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001619 {
1620 channel->ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001621 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001622 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001623
Bram Moolenaard7ece102016-02-02 23:23:02 +01001624 if (!more)
1625 {
1626 /* Handle any other messages in the queue. If done some more
1627 * messages may have arrived. */
1628 if (channel_parse_messages())
1629 continue;
1630
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001631 /* Wait for up to the channel timeout. */
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001632 fd = get_read_fd(channel);
Bram Moolenaard8070362016-02-15 21:56:54 +01001633 if (fd == CHAN_FD_INVALID
1634 || channel_wait(channel, fd, channel->ch_timeout) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001635 break;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001636 channel_read(channel, -1, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001637 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001638 }
Bram Moolenaare56bf152016-02-08 23:23:42 +01001639 channel->ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001640 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001641}
1642
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001643# if defined(WIN32) || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001644/*
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001645 * Lookup the channel from the socket. Set "which" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001646 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001647 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001648 channel_T *
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001649channel_fd2channel(sock_T fd, int *whichp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001650{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001651 channel_T *channel;
1652 int i;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001653
Bram Moolenaard8070362016-02-15 21:56:54 +01001654 if (fd != CHAN_FD_INVALID)
Bram Moolenaar77073442016-02-13 23:23:53 +01001655 for (channel = first_channel; channel != NULL;
1656 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001657 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001658# ifdef CHANNEL_PIPES
1659 for (i = CHAN_SOCK; i < CHAN_IN; ++i)
1660# else
1661 i = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001662# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001663 if (channel->ch_pfd[i].ch_fd == fd)
1664 {
1665 *whichp = i;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001666 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001667 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001668 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001669 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001670}
1671# endif
1672
Bram Moolenaard04a0202016-01-26 23:30:18 +01001673/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001674 * Write "buf" (NUL terminated string) to channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001675 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001676 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001677 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001678 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001679channel_send(channel_T *channel, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001680{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001681 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001682 int res;
Bram Moolenaard8070362016-02-15 21:56:54 +01001683 sock_T fd = CHAN_FD_INVALID;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001684 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001685
Bram Moolenaard8070362016-02-15 21:56:54 +01001686 if (channel->CH_SOCK != CHAN_FD_INVALID)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001687 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001688 fd = channel->CH_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001689 use_socket = TRUE;
1690 }
1691#if defined(CHANNEL_PIPES)
Bram Moolenaard8070362016-02-15 21:56:54 +01001692 else if (channel->CH_IN != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001693 fd = channel->CH_IN;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001694#endif
Bram Moolenaard8070362016-02-15 21:56:54 +01001695 if (fd == CHAN_FD_INVALID)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001696 {
1697 if (!channel->ch_error && fun != NULL)
1698 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001699 ch_errors(channel, "%s(): write while not connected\n", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001700 EMSG2("E630: %s(): write while not connected", fun);
1701 }
1702 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001703 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001704 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001705
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001706 if (log_fd != NULL)
1707 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001708 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001709 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001710 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001711 fprintf(log_fd, "'\n");
1712 fflush(log_fd);
1713 }
1714
1715 if (use_socket)
1716 res = sock_write(fd, buf, len);
1717 else
Bram Moolenaard8070362016-02-15 21:56:54 +01001718 res = fd_write(fd, buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001719 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001720 {
1721 if (!channel->ch_error && fun != NULL)
1722 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001723 ch_errors(channel, "%s(): write failed\n", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001724 EMSG2("E631: %s(): write failed", fun);
1725 }
1726 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001727 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001728 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001729
1730 channel->ch_error = FALSE;
1731 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001732}
1733
1734# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001735/*
1736 * Add open channels to the poll struct.
1737 * Return the adjusted struct index.
1738 * The type of "fds" is hidden to avoid problems with the function proto.
1739 */
1740 int
1741channel_poll_setup(int nfd_in, void *fds_in)
1742{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001743 int nfd = nfd_in;
1744 channel_T *channel;
1745 struct pollfd *fds = fds_in;
1746 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001747
Bram Moolenaar77073442016-02-13 23:23:53 +01001748 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001749 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001750# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001751 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1752# else
1753 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001754# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001755 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001756 if (channel->ch_pfd[which].ch_fd != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001757 {
1758 channel->ch_pfd[which].ch_poll_idx = nfd;
1759 fds[nfd].fd = channel->ch_pfd[which].ch_fd;
1760 fds[nfd].events = POLLIN;
1761 nfd++;
1762 }
1763 else
1764 channel->ch_pfd[which].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001765 }
1766 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001767
1768 return nfd;
1769}
1770
1771/*
1772 * The type of "fds" is hidden to avoid problems with the function proto.
1773 */
1774 int
1775channel_poll_check(int ret_in, void *fds_in)
1776{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001777 int ret = ret_in;
1778 channel_T *channel;
1779 struct pollfd *fds = fds_in;
1780 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001781
Bram Moolenaar77073442016-02-13 23:23:53 +01001782 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001783 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001784# ifdef CHANNEL_PIPES
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001785 for (which = CHAN_SOCK; which < CH_IN; ++which)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001786# else
1787 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001788# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001789 {
1790 int idx = channel->ch_pfd[which].ch_poll_idx;
1791
1792 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
1793 {
1794 channel_read(channel, which, "channel_poll_check");
1795 --ret;
1796 }
1797 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001798 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001799
1800 return ret;
1801}
Bram Moolenaard04a0202016-01-26 23:30:18 +01001802# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01001803
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001804# if (!defined(FEAT_GUI_W32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001805/*
1806 * The type of "rfds" is hidden to avoid problems with the function proto.
1807 */
1808 int
1809channel_select_setup(int maxfd_in, void *rfds_in)
1810{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001811 int maxfd = maxfd_in;
1812 channel_T *channel;
1813 fd_set *rfds = rfds_in;
1814 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001815
Bram Moolenaar77073442016-02-13 23:23:53 +01001816 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001817 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001818# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001819 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1820# else
1821 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001822# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001823 {
1824 sock_T fd = channel->ch_pfd[which].ch_fd;
1825
Bram Moolenaard8070362016-02-15 21:56:54 +01001826 if (fd != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001827 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001828 FD_SET((int)fd, rfds);
1829 if (maxfd < (int)fd)
1830 maxfd = (int)fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001831 }
1832 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001833 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001834
1835 return maxfd;
1836}
1837
1838/*
1839 * The type of "rfds" is hidden to avoid problems with the function proto.
1840 */
1841 int
1842channel_select_check(int ret_in, void *rfds_in)
1843{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001844 int ret = ret_in;
1845 channel_T *channel;
1846 fd_set *rfds = rfds_in;
1847 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001848
Bram Moolenaar77073442016-02-13 23:23:53 +01001849 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001850 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001851# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001852 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1853# else
1854 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001855# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001856 {
1857 sock_T fd = channel->ch_pfd[which].ch_fd;
1858
Bram Moolenaard8070362016-02-15 21:56:54 +01001859 if (ret > 0 && fd != CHAN_FD_INVALID && FD_ISSET(fd, rfds))
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001860 {
1861 channel_read(channel, which, "channel_select_check");
1862 --ret;
1863 }
1864 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001865 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001866
1867 return ret;
1868}
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001869# endif /* !FEAT_GUI_W32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01001870
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001871/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01001872 * Execute queued up commands.
1873 * Invoked from the main loop when it's safe to execute received commands.
1874 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001875 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001876 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001877channel_parse_messages(void)
1878{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001879 channel_T *channel = first_channel;
1880 int ret = FALSE;
1881 int r;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001882
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001883 while (channel != NULL)
1884 {
1885 /* Increase the refcount, in case the handler causes the channel to be
1886 * unreferenced or closed. */
1887 ++channel->ch_refcount;
1888 r = may_invoke_callback(channel);
1889 if (channel_unref(channel))
1890 /* channel was freed, start over */
1891 channel = first_channel;
1892
1893 if (r == OK)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001894 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001895 channel = first_channel; /* something was done, start over */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001896 ret = TRUE;
1897 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001898 else
1899 channel = channel->ch_next;
1900 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01001901 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001902}
1903
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01001904/*
1905 * Mark references to lists used in channels.
1906 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001907 int
1908set_ref_in_channel(int copyID)
1909{
Bram Moolenaar77073442016-02-13 23:23:53 +01001910 int abort = FALSE;
1911 channel_T *channel;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001912
Bram Moolenaar77073442016-02-13 23:23:53 +01001913 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001914 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001915 jsonq_T *head = &channel->ch_json_head;
1916 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001917
Bram Moolenaar77073442016-02-13 23:23:53 +01001918 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001919 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001920 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001921
1922 if (l->lv_copyID != copyID)
1923 {
1924 l->lv_copyID = copyID;
1925 abort = abort || set_ref_in_list(l, copyID, NULL);
1926 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001927 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001928 }
1929 }
1930 return abort;
1931}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001932
1933/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001934 * Return the mode of channel "channel".
1935 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001936 */
1937 ch_mode_T
Bram Moolenaar77073442016-02-13 23:23:53 +01001938channel_get_mode(channel_T *channel)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001939{
Bram Moolenaar77073442016-02-13 23:23:53 +01001940 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001941 return MODE_JSON;
Bram Moolenaar77073442016-02-13 23:23:53 +01001942 return channel->ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001943}
1944
Bram Moolenaare0874f82016-01-24 20:36:41 +01001945#endif /* FEAT_CHANNEL */