blob: d5d7ffb49b3b81e9423ea18c02a492cc39cd3fce [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 Moolenaard8070362016-02-15 21:56:54 +010050# define fd_read(fd, buf, len, timeout) read(fd, buf, len)
51# 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
61fd_read(sock_T fd, char_u *buf, size_t len, int timeout)
62{
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 Moolenaar4d919d72016-02-05 22:36:41 +0100637 PERROR(_("E899: Cannot connect to port after retry2"));
638 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/*
891 * Remove "node" from the queue that it is in and free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +0100892 * Also frees the contained callback name.
893 */
894 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100895remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +0100896{
Bram Moolenaar77073442016-02-13 23:23:53 +0100897 if (node->cq_prev == NULL)
898 head->cq_next = node->cq_next;
899 else
900 node->cq_prev->cq_next = node->cq_next;
901 if (node->cq_next == NULL)
902 head->cq_prev = node->cq_prev;
903 else
904 node->cq_next->cq_prev = node->cq_prev;
905 vim_free(node->cq_callback);
Bram Moolenaara07fec92016-02-05 21:04:08 +0100906 vim_free(node);
907}
908
909/*
910 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +0100911 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100912 */
913 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100914remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100915{
Bram Moolenaar77073442016-02-13 23:23:53 +0100916 if (node->jq_prev == NULL)
917 head->jq_next = node->jq_next;
918 else
919 node->jq_prev->jq_next = node->jq_next;
920 if (node->jq_next == NULL)
921 head->jq_prev = node->jq_prev;
922 else
923 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100924 vim_free(node);
925}
926
927/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100928 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100929 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +0100930 * When "id" is zero or negative jut get the first message. But not the one
931 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100932 * Return OK when found and return the value in "rettv".
933 * Return FAIL otherwise.
934 */
935 static int
Bram Moolenaar77073442016-02-13 23:23:53 +0100936channel_get_json(channel_T *channel, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100937{
Bram Moolenaare56bf152016-02-08 23:23:42 +0100938 jsonq_T *head = &channel->ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100939 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100940
Bram Moolenaar77073442016-02-13 23:23:53 +0100941 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100942 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100943 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100944 typval_T *tv = &l->lv_first->li_tv;
945
946 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +0100947 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaarf6157282016-02-10 21:07:14 +0100948 || tv->vval.v_number == 0
Bram Moolenaare56bf152016-02-08 23:23:42 +0100949 || tv->vval.v_number != channel->ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100950 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100951 *rettv = item->jq_value;
952 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100953 return OK;
954 }
Bram Moolenaar77073442016-02-13 23:23:53 +0100955 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100956 }
957 return FAIL;
958}
959
960/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100961 * Execute a command received over channel "channel".
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100962 * "cmd" is the command string, "arg2" the second argument.
963 * "arg3" is the third argument, NULL if missing.
964 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100965 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100966channel_exe_cmd(channel_T *channel, char_u *cmd, typval_T *arg2, typval_T *arg3)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100967{
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100968 char_u *arg;
969
970 if (arg2->v_type != VAR_STRING)
971 {
972 if (p_verbose > 2)
973 EMSG("E903: received ex command with non-string argument");
974 return;
975 }
976 arg = arg2->vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +0100977 if (arg == NULL)
978 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100979
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100980 if (STRCMP(cmd, "ex") == 0)
981 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100982 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100983 }
984 else if (STRCMP(cmd, "normal") == 0)
985 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100986 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100987
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100988 ea.arg = arg;
989 ea.addr_count = 0;
990 ea.forceit = TRUE; /* no mapping */
991 ex_normal(&ea);
992 }
993 else if (STRCMP(cmd, "redraw") == 0)
994 {
995 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100996
Bram Moolenaar14ad6112016-02-01 21:47:13 +0100997 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100998 ex_redraw(&ea);
999 showruler(FALSE);
1000 setcursor();
1001 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001002#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001003 if (gui.in_use)
1004 {
1005 gui_update_cursor(FALSE, FALSE);
1006 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001007 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001008#endif
1009 }
1010 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "eval") == 0)
1011 {
1012 int is_eval = cmd[1] == 'v';
1013
Bram Moolenaar66624ff2016-02-03 23:59:43 +01001014 if (is_eval && (arg3 == NULL || arg3->v_type != VAR_NUMBER))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001015 {
1016 if (p_verbose > 2)
1017 EMSG("E904: third argument for eval must be a number");
1018 }
1019 else
1020 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001021 typval_T *tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001022 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +01001023 char_u *json = NULL;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001024 int options = channel->ch_mode == MODE_JS ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001025
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001026 /* Don't pollute the display with errors. */
1027 ++emsg_skip;
1028 tv = eval_expr(arg, NULL);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001029 if (is_eval)
1030 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001031 if (tv != NULL)
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001032 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1033 options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001034 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001035 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001036 /* If evaluation failed or the result can't be encoded
1037 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001038 vim_free(json);
1039 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001040 err_tv.v_type = VAR_STRING;
1041 err_tv.vval.v_string = (char_u *)"ERROR";
1042 tv = &err_tv;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001043 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1044 options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001045 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001046 if (json != NULL)
1047 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001048 channel_send(channel, json, "eval");
Bram Moolenaar55fab432016-02-07 16:53:13 +01001049 vim_free(json);
1050 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001051 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001052 --emsg_skip;
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001053 if (tv != &err_tv)
1054 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001055 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001056 }
1057 else if (p_verbose > 2)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001058 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001059}
1060
1061/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001062 * Invoke a callback for channel "channel" if needed.
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001063 * Return OK when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001064 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001065 static int
Bram Moolenaar77073442016-02-13 23:23:53 +01001066may_invoke_callback(channel_T *channel)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001067{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001068 char_u *msg = NULL;
1069 typval_T *listtv = NULL;
1070 list_T *list;
1071 typval_T *typetv;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001072 typval_T argv[3];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001073 int seq_nr = -1;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001074 ch_mode_T ch_mode = channel->ch_mode;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001075
Bram Moolenaara07fec92016-02-05 21:04:08 +01001076 if (channel->ch_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001077 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001078 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001079
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001080 if (ch_mode != MODE_RAW)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001081 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001082 /* Get any json message in the queue. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001083 if (channel_get_json(channel, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001084 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001085 /* Parse readahead, return when there is still no message. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001086 channel_parse_json(channel);
1087 if (channel_get_json(channel, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001088 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001089 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001090
1091 list = listtv->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001092 argv[1] = list->lv_first->li_next->li_tv;
1093 typetv = &list->lv_first->li_tv;
1094 if (typetv->v_type == VAR_STRING)
1095 {
1096 typval_T *arg3 = NULL;
1097 char_u *cmd = typetv->vval.v_string;
1098
Bram Moolenaar66624ff2016-02-03 23:59:43 +01001099 /* ["cmd", arg] or ["cmd", arg, arg] */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001100 if (list->lv_len == 3)
1101 arg3 = &list->lv_last->li_tv;
Bram Moolenaar77073442016-02-13 23:23:53 +01001102 ch_logs(channel, "Executing %s command", (char *)cmd);
1103 channel_exe_cmd(channel, cmd, &argv[1], arg3);
1104 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001105 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001106 }
1107
1108 if (typetv->v_type != VAR_NUMBER)
1109 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001110 ch_error(channel,
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001111 "Dropping message with invalid sequence number type\n");
Bram Moolenaar77073442016-02-13 23:23:53 +01001112 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001113 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001114 }
1115 seq_nr = typetv->vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001116 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001117 else if (channel_peek(channel) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001118 {
1119 /* nothing to read on raw channel */
1120 return FALSE;
1121 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001122 else
1123 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001124 /* If there is no callback, don't do anything. */
1125 if (channel->ch_callback == NULL)
1126 return FALSE;
1127
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001128 /* For a raw channel we don't know where the message ends, just get
1129 * everything. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001130 msg = channel_get_all(channel);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001131 argv[1].v_type = VAR_STRING;
1132 argv[1].vval.v_string = msg;
1133 }
1134
Bram Moolenaara07fec92016-02-05 21:04:08 +01001135 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001136 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001137 cbq_T *head = &channel->ch_cb_head;
1138 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001139 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001140
1141 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001142 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001143 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001144 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001145 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001146 ch_log(channel, "Invoking one-time callback\n");
1147 invoke_callback(channel, item->cq_callback, argv);
1148 remove_cb_node(head, item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001149 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001150 break;
1151 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001152 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001153 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001154 if (!done)
Bram Moolenaar77073442016-02-13 23:23:53 +01001155 ch_log(channel, "Dropping message without callback\n");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001156 }
Bram Moolenaara07fec92016-02-05 21:04:08 +01001157 else if (channel->ch_callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001158 {
1159 /* invoke the channel callback */
Bram Moolenaar77073442016-02-13 23:23:53 +01001160 ch_log(channel, "Invoking channel callback\n");
1161 invoke_callback(channel, channel->ch_callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001162 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001163 else
Bram Moolenaar77073442016-02-13 23:23:53 +01001164 ch_log(channel, "Dropping message\n");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001165
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001166 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001167 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001168 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001169
1170 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001171}
1172
1173/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001174 * Return TRUE when channel "channel" is open for writing to.
1175 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001176 */
1177 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001178channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001179{
Bram Moolenaard8070362016-02-15 21:56:54 +01001180 return channel != NULL && (channel->CH_SOCK != CHAN_FD_INVALID
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001181#ifdef CHANNEL_PIPES
Bram Moolenaard8070362016-02-15 21:56:54 +01001182 || channel->CH_IN != CHAN_FD_INVALID
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001183#endif
1184 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001185}
1186
1187/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001188 * Return TRUE when channel "channel" is open for reading or writing.
1189 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001190 */
1191 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001192channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001193{
Bram Moolenaard8070362016-02-15 21:56:54 +01001194 return channel != NULL && (channel->CH_SOCK != CHAN_FD_INVALID
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001195#ifdef CHANNEL_PIPES
Bram Moolenaard8070362016-02-15 21:56:54 +01001196 || channel->CH_IN != CHAN_FD_INVALID
1197 || channel->CH_OUT != CHAN_FD_INVALID
1198 || channel->CH_ERR != CHAN_FD_INVALID
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001199#endif
1200 );
1201}
1202
1203/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001204 * Return a string indicating the status of the channel.
1205 */
1206 char *
1207channel_status(channel_T *channel)
1208{
1209 if (channel == NULL)
1210 return "fail";
1211 if (channel_is_open(channel))
1212 return "open";
1213 return "closed";
1214}
1215
1216/*
1217 * Close channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001218 * This does not trigger the close callback.
1219 */
1220 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001221channel_close(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001222{
Bram Moolenaar77073442016-02-13 23:23:53 +01001223 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001224
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001225#ifdef FEAT_GUI
1226 channel_gui_unregister(channel);
1227#endif
1228
Bram Moolenaard8070362016-02-15 21:56:54 +01001229 if (channel->CH_SOCK != CHAN_FD_INVALID)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001230 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001231 sock_close(channel->CH_SOCK);
Bram Moolenaard8070362016-02-15 21:56:54 +01001232 channel->CH_SOCK = CHAN_FD_INVALID;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001233 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001234#if defined(CHANNEL_PIPES)
Bram Moolenaard8070362016-02-15 21:56:54 +01001235 if (channel->CH_IN != CHAN_FD_INVALID)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001236 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001237 fd_close(channel->CH_IN);
1238 channel->CH_IN = CHAN_FD_INVALID;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001239 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001240 if (channel->CH_OUT != CHAN_FD_INVALID)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001241 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001242 fd_close(channel->CH_OUT);
1243 channel->CH_OUT = CHAN_FD_INVALID;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001244 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001245 if (channel->CH_ERR != CHAN_FD_INVALID)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001246 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001247 fd_close(channel->CH_ERR);
1248 channel->CH_ERR = CHAN_FD_INVALID;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001249 }
1250#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001251
1252 channel->ch_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001253 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001254}
1255
Bram Moolenaard04a0202016-01-26 23:30:18 +01001256/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001257 * Store "buf[len]" on channel "channel".
Bram Moolenaar83162462016-01-28 23:10:07 +01001258 * Returns OK or FAIL.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001259 */
Bram Moolenaar83162462016-01-28 23:10:07 +01001260 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001261channel_save(channel_T *channel, char_u *buf, int len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001262{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001263 readq_T *node;
Bram Moolenaar77073442016-02-13 23:23:53 +01001264 readq_T *head = &channel->ch_head;
Bram Moolenaar23689172016-02-15 22:37:37 +01001265 char_u *p;
1266 int i;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001267
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001268 node = (readq_T *)alloc(sizeof(readq_T));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001269 if (node == NULL)
Bram Moolenaar83162462016-01-28 23:10:07 +01001270 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +01001271 node->rq_buffer = alloc(len + 1);
1272 if (node->rq_buffer == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001273 {
1274 vim_free(node);
Bram Moolenaar83162462016-01-28 23:10:07 +01001275 return FAIL; /* out of memory */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001276 }
Bram Moolenaar23689172016-02-15 22:37:37 +01001277
1278 /* TODO: don't strip CR when channel is in raw mode */
1279 p = node->rq_buffer;
1280 for (i = 0; i < len; ++i)
1281 if (buf[i] != CAR || i + 1 >= len || buf[i + 1] != NL)
1282 *p++ = buf[i];
1283 *p = NUL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001284
Bram Moolenaar77073442016-02-13 23:23:53 +01001285 /* append node to the tail of the queue */
1286 node->rq_next = NULL;
1287 node->rq_prev = head->rq_prev;
1288 if (head->rq_prev == NULL)
1289 head->rq_next = node;
1290 else
1291 head->rq_prev->rq_next = node;
1292 head->rq_prev = node;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001293
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001294 if (log_fd != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001295 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001296 ch_log_lead("RECV ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001297 fprintf(log_fd, "'");
1298 if (fwrite(buf, len, 1, log_fd) != 1)
Bram Moolenaar83162462016-01-28 23:10:07 +01001299 return FAIL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001300 fprintf(log_fd, "'\n");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001301 }
Bram Moolenaar83162462016-01-28 23:10:07 +01001302 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001303}
1304
1305/*
1306 * Return the first buffer from the channel without removing it.
1307 * Returns NULL if there is nothing.
1308 */
1309 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +01001310channel_peek(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001311{
Bram Moolenaar77073442016-02-13 23:23:53 +01001312 readq_T *head = &channel->ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001313
Bram Moolenaar77073442016-02-13 23:23:53 +01001314 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001315 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001316 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001317}
1318
1319/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001320 * Clear the read buffer on channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001321 */
1322 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001323channel_clear(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001324{
Bram Moolenaar77073442016-02-13 23:23:53 +01001325 jsonq_T *json_head = &channel->ch_json_head;
1326 cbq_T *cb_head = &channel->ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001327
Bram Moolenaar77073442016-02-13 23:23:53 +01001328 while (channel_peek(channel) != NULL)
1329 vim_free(channel_get(channel));
1330
1331 while (cb_head->cq_next != NULL)
1332 remove_cb_node(cb_head, cb_head->cq_next);
1333
1334 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001335 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001336 free_tv(json_head->jq_next->jq_value);
1337 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001338 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001339
1340 vim_free(channel->ch_callback);
1341 channel->ch_callback = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001342}
1343
Bram Moolenaar77073442016-02-13 23:23:53 +01001344#if defined(EXITFREE) || defined(PROTO)
1345 void
1346channel_free_all(void)
1347{
1348 channel_T *channel;
1349
1350 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1351 channel_clear(channel);
1352}
1353#endif
1354
1355
Bram Moolenaard04a0202016-01-26 23:30:18 +01001356/* Sent when the channel is found closed when reading. */
1357#define DETACH_MSG "\"DETACH\"\n"
1358
1359/* Buffer size for reading incoming messages. */
1360#define MAXMSGSIZE 4096
1361
1362/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001363 * Check for reading from "fd" with "timeout" msec.
1364 * Return FAIL when there is nothing to read.
Bram Moolenaara8343c12016-02-04 22:09:48 +01001365 * Always returns OK for FEAT_GUI_W32.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001366 */
1367 static int
Bram Moolenaard8070362016-02-15 21:56:54 +01001368channel_wait(channel_T *channel, sock_T fd, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001369{
Bram Moolenaara8343c12016-02-04 22:09:48 +01001370#if defined(HAVE_SELECT) && !defined(FEAT_GUI_W32)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001371 struct timeval tval;
1372 fd_set rfds;
1373 int ret;
1374
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001375 if (timeout > 0)
Bram Moolenaar77073442016-02-13 23:23:53 +01001376 ch_logn(channel, "Waiting for %d msec\n", timeout);
Bram Moolenaard8070362016-02-15 21:56:54 +01001377
1378
1379# ifdef WIN32
1380 if (channel->CH_SOCK == CHAN_FD_INVALID)
1381 {
1382 DWORD nread;
1383 int diff;
1384 DWORD deadline = GetTickCount() + timeout;
1385
1386 /* reading from a pipe, not a socket */
1387 while (TRUE)
1388 {
1389 if (PeekNamedPipe(fd, NULL, 0, NULL, &nread, NULL) && nread > 0)
1390 return OK;
1391 diff = deadline - GetTickCount();
1392 if (diff < 0)
1393 break;
1394 /* Wait for 5 msec.
1395 * TODO: increase the sleep time when looping more often */
1396 Sleep(5);
1397 }
1398 return FAIL;
1399 }
1400#endif
1401
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001402 FD_ZERO(&rfds);
Bram Moolenaard8070362016-02-15 21:56:54 +01001403 FD_SET((int)fd, &rfds);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001404 tval.tv_sec = timeout / 1000;
1405 tval.tv_usec = (timeout % 1000) * 1000;
1406 for (;;)
1407 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001408 ret = select((int)fd + 1, &rfds, NULL, NULL, &tval);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001409# ifdef EINTR
1410 if (ret == -1 && errno == EINTR)
1411 continue;
1412# endif
1413 if (ret <= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001414 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001415 ch_log(channel, "Nothing to read\n");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001416 return FAIL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001417 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001418 break;
1419 }
1420#else
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001421# ifdef HAVE_POLL
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001422 struct pollfd fds;
1423
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001424 if (timeout > 0)
Bram Moolenaar77073442016-02-13 23:23:53 +01001425 ch_logn(channel, "Waiting for %d msec\n", timeout);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001426 fds.fd = fd;
1427 fds.events = POLLIN;
1428 if (poll(&fds, 1, timeout) <= 0)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001429 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001430 ch_log(channel, "Nothing to read\n");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001431 return FAIL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001432 }
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001433# endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001434#endif
1435 return OK;
1436}
1437
1438/*
1439 * Return a unique ID to be used in a message.
1440 */
1441 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001442channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001443{
1444 static int next_id = 1;
1445
1446 return next_id++;
1447}
1448
1449/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001450 * Get the file descriptor to read from, either the socket or stdout.
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001451 * TODO: should have a way to read stderr.
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001452 */
Bram Moolenaard8070362016-02-15 21:56:54 +01001453 static sock_T
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001454get_read_fd(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001455{
Bram Moolenaard8070362016-02-15 21:56:54 +01001456 if (channel->CH_SOCK != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001457 return channel->CH_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001458#if defined(CHANNEL_PIPES)
Bram Moolenaard8070362016-02-15 21:56:54 +01001459 if (channel->CH_OUT != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001460 return channel->CH_OUT;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001461#endif
Bram Moolenaar77073442016-02-13 23:23:53 +01001462 ch_error(channel, "channel_read() called while socket is closed\n");
Bram Moolenaard8070362016-02-15 21:56:54 +01001463 return CHAN_FD_INVALID;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001464}
1465
1466/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001467 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001468 * "which" is CHAN_SOCK, CHAN_OUT or CHAN_ERR. When -1 use CHAN_SOCK or
1469 * CHAN_OUT, the one that is open.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001470 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001471 */
1472 void
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001473channel_read(channel_T *channel, int which, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001474{
1475 static char_u *buf = NULL;
1476 int len = 0;
1477 int readlen = 0;
Bram Moolenaard8070362016-02-15 21:56:54 +01001478 sock_T fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001479 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001480
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001481 if (which < 0)
1482 fd = get_read_fd(channel);
1483 else
1484 fd = channel->ch_pfd[which].ch_fd;
Bram Moolenaard8070362016-02-15 21:56:54 +01001485 if (fd == CHAN_FD_INVALID)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001486 return;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001487 use_socket = fd == channel->CH_SOCK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001488
1489 /* Allocate a buffer to read into. */
1490 if (buf == NULL)
1491 {
1492 buf = alloc(MAXMSGSIZE);
1493 if (buf == NULL)
1494 return; /* out of memory! */
1495 }
1496
1497 /* Keep on reading for as long as there is something to read.
1498 * Use select() or poll() to avoid blocking on a message that is exactly
1499 * MAXMSGSIZE long. */
1500 for (;;)
1501 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001502 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001503 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001504 if (use_socket)
1505 len = sock_read(fd, buf, MAXMSGSIZE);
1506 else
Bram Moolenaard8070362016-02-15 21:56:54 +01001507 len = fd_read(fd, buf, MAXMSGSIZE, channel->ch_timeout);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001508 if (len <= 0)
1509 break; /* error or nothing more to read */
1510
1511 /* Store the read message in the queue. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001512 channel_save(channel, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001513 readlen += len;
1514 if (len < MAXMSGSIZE)
1515 break; /* did read everything that's available */
1516 }
Bram Moolenaara8343c12016-02-04 22:09:48 +01001517#ifdef FEAT_GUI_W32
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001518 if (use_socket && len == SOCKET_ERROR)
Bram Moolenaara8343c12016-02-04 22:09:48 +01001519 {
1520 /* For Win32 GUI channel_wait() always returns OK and we handle the
1521 * situation that there is nothing to read here.
1522 * TODO: how about a timeout? */
1523 if (WSAGetLastError() == WSAEWOULDBLOCK)
1524 return;
1525 }
1526#endif
Bram Moolenaard04a0202016-01-26 23:30:18 +01001527
1528 /* Reading a socket disconnection (readlen == 0), or a socket error. */
1529 if (readlen <= 0)
1530 {
1531 /* Queue a "DETACH" netbeans message in the command queue in order to
1532 * terminate the netbeans session later. Do not end the session here
1533 * directly as we may be running in the context of a call to
1534 * netbeans_parse_messages():
1535 * netbeans_parse_messages
1536 * -> autocmd triggered while processing the netbeans cmd
1537 * -> ui_breakcheck
1538 * -> gui event loop or select loop
1539 * -> channel_read()
1540 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001541 ch_errors(channel, "%s(): Cannot read\n", func);
1542 channel_save(channel, (char_u *)DETACH_MSG, (int)STRLEN(DETACH_MSG));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001543
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001544 /* TODO: When reading from stdout is not possible, should we try to
1545 * keep stdin and stderr open? Probably not, assume the other side
1546 * has died. */
1547 channel_close(channel);
1548 if (channel->ch_close_cb != NULL)
1549 (*channel->ch_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001550
1551 if (len < 0)
1552 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001553 ch_error(channel, "channel_read(): cannot read from channel\n");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001554 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001555 }
1556 }
1557
1558#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001559 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001560 if (CH_HAS_GUI && gtk_main_level() > 0)
1561 gtk_main_quit();
1562#endif
1563}
1564
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001565/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001566 * Read from raw channel "channel". Blocks until there is something to read or
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001567 * the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001568 * Returns what was read in allocated memory.
1569 * Returns NULL in case of error or timeout.
1570 */
1571 char_u *
Bram Moolenaar77073442016-02-13 23:23:53 +01001572channel_read_block(channel_T *channel)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001573{
Bram Moolenaar77073442016-02-13 23:23:53 +01001574 ch_log(channel, "Reading raw\n");
1575 if (channel_peek(channel) == NULL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001576 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001577 sock_T fd = get_read_fd(channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001578
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001579 /* TODO: read both out and err if they are different */
Bram Moolenaar77073442016-02-13 23:23:53 +01001580 ch_log(channel, "No readahead\n");
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001581 /* Wait for up to the channel timeout. */
Bram Moolenaard8070362016-02-15 21:56:54 +01001582 if (fd == CHAN_FD_INVALID
1583 || channel_wait(channel, fd, channel->ch_timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001584 return NULL;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001585 channel_read(channel, -1, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001586 }
1587
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001588 /* TODO: only get the first message */
Bram Moolenaar77073442016-02-13 23:23:53 +01001589 ch_log(channel, "Returning readahead\n");
1590 return channel_get_all(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001591}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001592
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001593/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001594 * Read one JSON message with ID "id" from channel "channel" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001595 * result in "rettv".
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001596 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001597 */
1598 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001599channel_read_json_block(channel_T *channel, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001600{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001601 int more;
Bram Moolenaard8070362016-02-15 21:56:54 +01001602 sock_T fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001603
Bram Moolenaar77073442016-02-13 23:23:53 +01001604 ch_log(channel, "Reading JSON\n");
Bram Moolenaare56bf152016-02-08 23:23:42 +01001605 channel->ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001606 for (;;)
1607 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001608 more = channel_parse_json(channel);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001609
1610 /* search for messsage "id" */
Bram Moolenaar77073442016-02-13 23:23:53 +01001611 if (channel_get_json(channel, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001612 {
1613 channel->ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001614 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001615 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001616
Bram Moolenaard7ece102016-02-02 23:23:02 +01001617 if (!more)
1618 {
1619 /* Handle any other messages in the queue. If done some more
1620 * messages may have arrived. */
1621 if (channel_parse_messages())
1622 continue;
1623
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001624 /* Wait for up to the channel timeout. */
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001625 fd = get_read_fd(channel);
Bram Moolenaard8070362016-02-15 21:56:54 +01001626 if (fd == CHAN_FD_INVALID
1627 || channel_wait(channel, fd, channel->ch_timeout) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001628 break;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001629 channel_read(channel, -1, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001630 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001631 }
Bram Moolenaare56bf152016-02-08 23:23:42 +01001632 channel->ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001633 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001634}
1635
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001636# if defined(WIN32) || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001637/*
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001638 * Lookup the channel from the socket. Set "which" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001639 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001640 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001641 channel_T *
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001642channel_fd2channel(sock_T fd, int *whichp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001643{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001644 channel_T *channel;
1645 int i;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001646
Bram Moolenaard8070362016-02-15 21:56:54 +01001647 if (fd != CHAN_FD_INVALID)
Bram Moolenaar77073442016-02-13 23:23:53 +01001648 for (channel = first_channel; channel != NULL;
1649 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001650 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001651# ifdef CHANNEL_PIPES
1652 for (i = CHAN_SOCK; i < CHAN_IN; ++i)
1653# else
1654 i = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001655# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001656 if (channel->ch_pfd[i].ch_fd == fd)
1657 {
1658 *whichp = i;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001659 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001660 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001661 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001662 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001663}
1664# endif
1665
Bram Moolenaard04a0202016-01-26 23:30:18 +01001666/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001667 * Write "buf" (NUL terminated string) to channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001668 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001669 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001670 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001671 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001672channel_send(channel_T *channel, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001673{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001674 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001675 int res;
Bram Moolenaard8070362016-02-15 21:56:54 +01001676 sock_T fd = CHAN_FD_INVALID;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001677 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001678
Bram Moolenaard8070362016-02-15 21:56:54 +01001679 if (channel->CH_SOCK != CHAN_FD_INVALID)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001680 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001681 fd = channel->CH_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001682 use_socket = TRUE;
1683 }
1684#if defined(CHANNEL_PIPES)
Bram Moolenaard8070362016-02-15 21:56:54 +01001685 else if (channel->CH_IN != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001686 fd = channel->CH_IN;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001687#endif
Bram Moolenaard8070362016-02-15 21:56:54 +01001688 if (fd == CHAN_FD_INVALID)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001689 {
1690 if (!channel->ch_error && fun != NULL)
1691 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001692 ch_errors(channel, "%s(): write while not connected\n", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001693 EMSG2("E630: %s(): write while not connected", fun);
1694 }
1695 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001696 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001697 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001698
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001699 if (log_fd != NULL)
1700 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001701 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001702 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001703 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001704 fprintf(log_fd, "'\n");
1705 fflush(log_fd);
1706 }
1707
1708 if (use_socket)
1709 res = sock_write(fd, buf, len);
1710 else
Bram Moolenaard8070362016-02-15 21:56:54 +01001711 res = fd_write(fd, buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001712 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001713 {
1714 if (!channel->ch_error && fun != NULL)
1715 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001716 ch_errors(channel, "%s(): write failed\n", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001717 EMSG2("E631: %s(): write failed", fun);
1718 }
1719 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001720 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001721 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001722
1723 channel->ch_error = FALSE;
1724 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001725}
1726
1727# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001728/*
1729 * Add open channels to the poll struct.
1730 * Return the adjusted struct index.
1731 * The type of "fds" is hidden to avoid problems with the function proto.
1732 */
1733 int
1734channel_poll_setup(int nfd_in, void *fds_in)
1735{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001736 int nfd = nfd_in;
1737 channel_T *channel;
1738 struct pollfd *fds = fds_in;
1739 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001740
Bram Moolenaar77073442016-02-13 23:23:53 +01001741 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001742 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001743# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001744 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1745# else
1746 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001747# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001748 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001749 if (channel->ch_pfd[which].ch_fd != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001750 {
1751 channel->ch_pfd[which].ch_poll_idx = nfd;
1752 fds[nfd].fd = channel->ch_pfd[which].ch_fd;
1753 fds[nfd].events = POLLIN;
1754 nfd++;
1755 }
1756 else
1757 channel->ch_pfd[which].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001758 }
1759 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001760
1761 return nfd;
1762}
1763
1764/*
1765 * The type of "fds" is hidden to avoid problems with the function proto.
1766 */
1767 int
1768channel_poll_check(int ret_in, void *fds_in)
1769{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001770 int ret = ret_in;
1771 channel_T *channel;
1772 struct pollfd *fds = fds_in;
1773 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001774
Bram Moolenaar77073442016-02-13 23:23:53 +01001775 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001776 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001777# ifdef CHANNEL_PIPES
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001778 for (which = CHAN_SOCK; which < CH_IN; ++which)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001779# else
1780 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001781# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001782 {
1783 int idx = channel->ch_pfd[which].ch_poll_idx;
1784
1785 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
1786 {
1787 channel_read(channel, which, "channel_poll_check");
1788 --ret;
1789 }
1790 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001791 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001792
1793 return ret;
1794}
Bram Moolenaard04a0202016-01-26 23:30:18 +01001795# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01001796
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001797# if (!defined(FEAT_GUI_W32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001798/*
1799 * The type of "rfds" is hidden to avoid problems with the function proto.
1800 */
1801 int
1802channel_select_setup(int maxfd_in, void *rfds_in)
1803{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001804 int maxfd = maxfd_in;
1805 channel_T *channel;
1806 fd_set *rfds = rfds_in;
1807 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001808
Bram Moolenaar77073442016-02-13 23:23:53 +01001809 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001810 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001811# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001812 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1813# else
1814 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001815# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001816 {
1817 sock_T fd = channel->ch_pfd[which].ch_fd;
1818
Bram Moolenaard8070362016-02-15 21:56:54 +01001819 if (fd != CHAN_FD_INVALID)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001820 {
Bram Moolenaard8070362016-02-15 21:56:54 +01001821 FD_SET((int)fd, rfds);
1822 if (maxfd < (int)fd)
1823 maxfd = (int)fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001824 }
1825 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001826 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001827
1828 return maxfd;
1829}
1830
1831/*
1832 * The type of "rfds" is hidden to avoid problems with the function proto.
1833 */
1834 int
1835channel_select_check(int ret_in, void *rfds_in)
1836{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001837 int ret = ret_in;
1838 channel_T *channel;
1839 fd_set *rfds = rfds_in;
1840 int which;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001841
Bram Moolenaar77073442016-02-13 23:23:53 +01001842 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001843 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001844# ifdef CHANNEL_PIPES
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001845 for (which = CHAN_SOCK; which < CHAN_IN; ++which)
1846# else
1847 which = CHAN_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001848# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001849 {
1850 sock_T fd = channel->ch_pfd[which].ch_fd;
1851
Bram Moolenaard8070362016-02-15 21:56:54 +01001852 if (ret > 0 && fd != CHAN_FD_INVALID && FD_ISSET(fd, rfds))
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001853 {
1854 channel_read(channel, which, "channel_select_check");
1855 --ret;
1856 }
1857 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001858 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001859
1860 return ret;
1861}
Bram Moolenaarf12d9832016-01-29 21:11:25 +01001862# endif /* !FEAT_GUI_W32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01001863
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001864/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01001865 * Execute queued up commands.
1866 * Invoked from the main loop when it's safe to execute received commands.
1867 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001868 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001869 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001870channel_parse_messages(void)
1871{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001872 channel_T *channel = first_channel;
1873 int ret = FALSE;
1874 int r;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001875
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001876 while (channel != NULL)
1877 {
1878 /* Increase the refcount, in case the handler causes the channel to be
1879 * unreferenced or closed. */
1880 ++channel->ch_refcount;
1881 r = may_invoke_callback(channel);
1882 if (channel_unref(channel))
1883 /* channel was freed, start over */
1884 channel = first_channel;
1885
1886 if (r == OK)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001887 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001888 channel = first_channel; /* something was done, start over */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001889 ret = TRUE;
1890 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001891 else
1892 channel = channel->ch_next;
1893 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01001894 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001895}
1896
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01001897/*
1898 * Mark references to lists used in channels.
1899 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001900 int
1901set_ref_in_channel(int copyID)
1902{
Bram Moolenaar77073442016-02-13 23:23:53 +01001903 int abort = FALSE;
1904 channel_T *channel;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001905
Bram Moolenaar77073442016-02-13 23:23:53 +01001906 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001907 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001908 jsonq_T *head = &channel->ch_json_head;
1909 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001910
Bram Moolenaar77073442016-02-13 23:23:53 +01001911 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001912 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001913 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001914
1915 if (l->lv_copyID != copyID)
1916 {
1917 l->lv_copyID = copyID;
1918 abort = abort || set_ref_in_list(l, copyID, NULL);
1919 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001920 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01001921 }
1922 }
1923 return abort;
1924}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001925
1926/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001927 * Return the mode of channel "channel".
1928 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001929 */
1930 ch_mode_T
Bram Moolenaar77073442016-02-13 23:23:53 +01001931channel_get_mode(channel_T *channel)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001932{
Bram Moolenaar77073442016-02-13 23:23:53 +01001933 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001934 return MODE_JSON;
Bram Moolenaar77073442016-02-13 23:23:53 +01001935 return channel->ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01001936}
1937
Bram Moolenaare0874f82016-01-24 20:36:41 +01001938#endif /* FEAT_CHANNEL */