blob: 0d46c587e148b3a9c7977a61008e47b3b5d1673e [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
Bram Moolenaard8070362016-02-15 21:56:54 +010055#ifdef WIN32
56 static int
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +010057fd_read(sock_T fd, char *buf, size_t len)
Bram Moolenaard8070362016-02-15 21:56:54 +010058{
59 HANDLE h = (HANDLE)fd;
60 DWORD nread;
61
62 if (!ReadFile(h, buf, (DWORD)len, &nread, NULL))
63 return -1;
64 return (int)nread;
65}
66
67 static int
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +010068fd_write(sock_T fd, char *buf, size_t len)
Bram Moolenaard8070362016-02-15 21:56:54 +010069{
70 HANDLE h = (HANDLE)fd;
71 DWORD nwrite;
72
73 if (!WriteFile(h, buf, (DWORD)len, &nwrite, NULL))
74 return -1;
75 return (int)nwrite;
76}
77
78 static void
79fd_close(sock_T fd)
80{
81 HANDLE h = (HANDLE)fd;
82
83 CloseHandle(h);
84}
85#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +010086
Bram Moolenaar6463ca22016-02-13 17:04:46 +010087/* Log file opened with ch_logfile(). */
88static FILE *log_fd = NULL;
Bram Moolenaar81661fb2016-02-18 22:23:34 +010089#ifdef FEAT_RELTIME
90static proftime_T log_start;
91#endif
Bram Moolenaar6463ca22016-02-13 17:04:46 +010092
93 void
94ch_logfile(FILE *file)
95{
96 if (log_fd != NULL)
97 fclose(log_fd);
98 log_fd = file;
99 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100100 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100101 fprintf(log_fd, "==== start log session ====\n");
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100102#ifdef FEAT_RELTIME
103 profile_start(&log_start);
104#endif
105 }
106}
107
108 int
109ch_log_active()
110{
111 return log_fd != NULL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100112}
113
114 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100115ch_log_lead(char *what, channel_T *ch)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100116{
117 if (log_fd != NULL)
118 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100119#ifdef FEAT_RELTIME
120 proftime_T log_now;
121
122 profile_start(&log_now);
123 profile_sub(&log_now, &log_start);
124 fprintf(log_fd, "%s ", profile_msg(&log_now));
125#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100126 if (ch != NULL)
127 fprintf(log_fd, "%son %d: ", what, ch->ch_id);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100128 else
129 fprintf(log_fd, "%s: ", what);
130 }
131}
132
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100133 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100134ch_log(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100135{
136 if (log_fd != NULL)
137 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100138 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100139 fputs(msg, log_fd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100140 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100141 fflush(log_fd);
142 }
143}
144
145 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100146ch_logn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100147{
148 if (log_fd != NULL)
149 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100150 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100151 fprintf(log_fd, msg, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100152 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100153 fflush(log_fd);
154 }
155}
156
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100157 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100158ch_logs(channel_T *ch, char *msg, char *name)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100159{
160 if (log_fd != NULL)
161 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100162 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100163 fprintf(log_fd, msg, name);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100164 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100165 fflush(log_fd);
166 }
167}
168
169 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100170ch_logsn(channel_T *ch, char *msg, char *name, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100171{
172 if (log_fd != NULL)
173 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100174 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100175 fprintf(log_fd, msg, name, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100176 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100177 fflush(log_fd);
178 }
179}
180
181 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100182ch_error(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100183{
184 if (log_fd != NULL)
185 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100186 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100187 fputs(msg, log_fd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100188 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100189 fflush(log_fd);
190 }
191}
192
193 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100194ch_errorn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100195{
196 if (log_fd != NULL)
197 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100198 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100199 fprintf(log_fd, msg, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100200 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100201 fflush(log_fd);
202 }
203}
204
205 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100206ch_errors(channel_T *ch, char *msg, char *arg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100207{
208 if (log_fd != NULL)
209 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100210 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100211 fprintf(log_fd, msg, arg);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100212 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100213 fflush(log_fd);
214 }
215}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100216
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100217#ifdef _WIN32
218# undef PERROR
219# define PERROR(msg) (void)emsg3((char_u *)"%s: %s", \
220 (char_u *)msg, (char_u *)strerror_win32(errno))
221
222 static char *
223strerror_win32(int eno)
224{
225 static LPVOID msgbuf = NULL;
226 char_u *ptr;
227
228 if (msgbuf)
229 LocalFree(msgbuf);
230 FormatMessage(
231 FORMAT_MESSAGE_ALLOCATE_BUFFER |
232 FORMAT_MESSAGE_FROM_SYSTEM |
233 FORMAT_MESSAGE_IGNORE_INSERTS,
234 NULL,
235 eno,
236 MAKELANGID(LANG_ENGLISH, SUBLANG_DEFAULT),
237 (LPTSTR) &msgbuf,
238 0,
239 NULL);
240 /* chomp \r or \n */
241 for (ptr = (char_u *)msgbuf; *ptr; ptr++)
242 switch (*ptr)
243 {
244 case '\r':
245 STRMOVE(ptr, ptr + 1);
246 ptr--;
247 break;
248 case '\n':
249 if (*(ptr + 1) == '\0')
250 *ptr = '\0';
251 else
252 *ptr = ' ';
253 break;
254 }
255 return msgbuf;
256}
257#endif
258
Bram Moolenaar77073442016-02-13 23:23:53 +0100259/*
260 * The list of all allocated channels.
261 */
262static channel_T *first_channel = NULL;
263static int next_ch_id = 0;
264
265/*
266 * Allocate a new channel. The refcount is set to 1.
267 * The channel isn't actually used until it is opened.
268 * Returns NULL if out of memory.
269 */
270 channel_T *
271add_channel(void)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100272{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100273 int part;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100274 channel_T *channel = (channel_T *)alloc_clear((int)sizeof(channel_T));
Bram Moolenaare0874f82016-01-24 20:36:41 +0100275
Bram Moolenaar77073442016-02-13 23:23:53 +0100276 if (channel == NULL)
277 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100278
Bram Moolenaar77073442016-02-13 23:23:53 +0100279 channel->ch_id = next_ch_id++;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100280 ch_log(channel, "Created channel");
Bram Moolenaar77073442016-02-13 23:23:53 +0100281
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100282#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100283 for (part = PART_SOCK; part <= PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100284#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100285 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100286#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100287 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100288 channel->ch_part[part].ch_fd = INVALID_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100289#ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100290 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100291#endif
292#ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100293 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100294#endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100295 channel->ch_part[part].ch_timeout = 2000;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100296 }
Bram Moolenaare0874f82016-01-24 20:36:41 +0100297
Bram Moolenaar77073442016-02-13 23:23:53 +0100298 if (first_channel != NULL)
299 {
300 first_channel->ch_prev = channel;
301 channel->ch_next = first_channel;
302 }
303 first_channel = channel;
304
305 channel->ch_refcount = 1;
306 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100307}
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100308
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100309/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100310 * Close a channel and free all its resources.
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100311 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100312 void
313channel_free(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100314{
Bram Moolenaar77073442016-02-13 23:23:53 +0100315 channel_close(channel);
316 if (channel->ch_next != NULL)
317 channel->ch_next->ch_prev = channel->ch_prev;
318 if (channel->ch_prev == NULL)
319 first_channel = channel->ch_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100320 else
Bram Moolenaar77073442016-02-13 23:23:53 +0100321 channel->ch_prev->ch_next = channel->ch_next;
322 vim_free(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100323}
324
Bram Moolenaard04a0202016-01-26 23:30:18 +0100325#if defined(FEAT_GUI) || defined(PROTO)
Bram Moolenaar77073442016-02-13 23:23:53 +0100326
327#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
328 static void
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100329channel_read_fd(int fd)
Bram Moolenaar77073442016-02-13 23:23:53 +0100330{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100331 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100332 int part;
Bram Moolenaar77073442016-02-13 23:23:53 +0100333
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100334 channel = channel_fd2channel(fd, &part);
Bram Moolenaar77073442016-02-13 23:23:53 +0100335 if (channel == NULL)
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100336 ch_errorn(NULL, "Channel for fd %d not found", fd);
Bram Moolenaar77073442016-02-13 23:23:53 +0100337 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100338 channel_read(channel, part, "messageFromNetbeans");
Bram Moolenaar77073442016-02-13 23:23:53 +0100339}
340#endif
341
Bram Moolenaare0874f82016-01-24 20:36:41 +0100342/*
Bram Moolenaard04a0202016-01-26 23:30:18 +0100343 * Read a command from netbeans.
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100344 * TODO: instead of channel ID use the FD.
Bram Moolenaare0874f82016-01-24 20:36:41 +0100345 */
Bram Moolenaard04a0202016-01-26 23:30:18 +0100346#ifdef FEAT_GUI_X11
347 static void
348messageFromNetbeans(XtPointer clientData,
349 int *unused1 UNUSED,
350 XtInputId *unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100351{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100352 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100353}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100354#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100355
Bram Moolenaard04a0202016-01-26 23:30:18 +0100356#ifdef FEAT_GUI_GTK
Bram Moolenaar98921892016-02-23 17:14:37 +0100357# if GTK_CHECK_VERSION(3,0,0)
358 static gboolean
359messageFromNetbeans(GIOChannel *unused1 UNUSED,
360 GIOCondition unused2 UNUSED,
361 gpointer clientData)
362{
363 channel_read_fd(GPOINTER_TO_INT(clientData));
364 return TRUE; /* Return FALSE instead in case the event source is to
365 * be removed after this function returns. */
366}
367# else
Bram Moolenaard04a0202016-01-26 23:30:18 +0100368 static void
369messageFromNetbeans(gpointer clientData,
370 gint unused1 UNUSED,
371 GdkInputCondition unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100372{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100373 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100374}
Bram Moolenaar98921892016-02-23 17:14:37 +0100375# endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100376#endif
377
378 static void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100379channel_gui_register_one(channel_T *channel, int part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100380{
381# ifdef FEAT_GUI_X11
382 /* Tell notifier we are interested in being called
383 * when there is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100384 if (channel->ch_part[part].ch_inputHandler == (XtInputId)NULL)
385 channel->ch_part[part].ch_inputHandler = XtAppAddInput(
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100386 (XtAppContext)app_context,
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100387 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100388 (XtPointer)(XtInputReadMask + XtInputExceptMask),
389 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100390 (XtPointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100391# else
392# ifdef FEAT_GUI_GTK
393 /* Tell gdk we are interested in being called when there
394 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100395 if (channel->ch_part[part].ch_inputHandler == 0)
Bram Moolenaar98921892016-02-23 17:14:37 +0100396# if GTK_CHECK_VERSION(3,0,0)
397 {
398 GIOChannel *chnnl = g_io_channel_unix_new(
399 (gint)channel->ch_part[part].ch_fd);
400
401 channel->ch_part[part].ch_inputHandler = g_io_add_watch(
402 chnnl,
403 G_IO_IN|G_IO_HUP|G_IO_ERR|G_IO_PRI,
404 messageFromNetbeans,
405 GINT_TO_POINTER(channel->ch_part[part].ch_fd));
406
407 g_io_channel_unref(chnnl);
408 }
409# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100410 channel->ch_part[part].ch_inputHandler = gdk_input_add(
411 (gint)channel->ch_part[part].ch_fd,
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100412 (GdkInputCondition)
413 ((int)GDK_INPUT_READ + (int)GDK_INPUT_EXCEPTION),
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100414 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100415 (gpointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar98921892016-02-23 17:14:37 +0100416# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100417# endif
418# endif
419}
420
421 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100422channel_gui_register(channel_T *channel)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100423{
Bram Moolenaard04a0202016-01-26 23:30:18 +0100424 if (!CH_HAS_GUI)
425 return;
426
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100427 if (channel->CH_SOCK_FD != INVALID_FD)
428 channel_gui_register_one(channel, PART_SOCK);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100429# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100430 if (channel->CH_OUT_FD != INVALID_FD)
431 channel_gui_register_one(channel, PART_OUT);
432 if (channel->CH_ERR_FD != INVALID_FD)
433 channel_gui_register_one(channel, PART_ERR);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100434# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100435}
436
437/*
438 * Register any of our file descriptors with the GUI event handling system.
439 * Called when the GUI has started.
440 */
441 void
442channel_gui_register_all(void)
443{
Bram Moolenaar77073442016-02-13 23:23:53 +0100444 channel_T *channel;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100445
Bram Moolenaar77073442016-02-13 23:23:53 +0100446 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100447 channel_gui_register(channel);
448}
449
450 static void
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100451channel_gui_unregister(channel_T *channel)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100452{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100453 int part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100454
455#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100456 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100457#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100458 part = PART_SOCK;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100459#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100460 {
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100461# ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100462 if (channel->ch_part[part].ch_inputHandler != (XtInputId)NULL)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100463 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100464 XtRemoveInput(channel->ch_part[part].ch_inputHandler);
465 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100466 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100467# else
468# ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100469 if (channel->ch_part[part].ch_inputHandler != 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100470 {
Bram Moolenaar98921892016-02-23 17:14:37 +0100471# if GTK_CHECK_VERSION(3,0,0)
472 g_source_remove(channel->ch_part[part].ch_inputHandler);
473# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100474 gdk_input_remove(channel->ch_part[part].ch_inputHandler);
Bram Moolenaar98921892016-02-23 17:14:37 +0100475# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100476 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100477 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100478# endif
479# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100480 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100481}
482
483#endif
484
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100485static char *e_cannot_connect = N_("E902: Cannot connect to port");
486
Bram Moolenaard04a0202016-01-26 23:30:18 +0100487/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100488 * Open a socket channel to "hostname":"port".
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100489 * "waittime" is the time in msec to wait for the connection.
490 * When negative wait forever.
Bram Moolenaar77073442016-02-13 23:23:53 +0100491 * Returns the channel for success.
492 * Returns NULL for failure.
Bram Moolenaard04a0202016-01-26 23:30:18 +0100493 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100494 channel_T *
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100495channel_open(
496 char *hostname,
497 int port_in,
498 int waittime,
499 void (*nb_close_cb)(void))
Bram Moolenaard04a0202016-01-26 23:30:18 +0100500{
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100501 int sd = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100502 struct sockaddr_in server;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100503 struct hostent *host;
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100504#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100505 u_short port = port_in;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100506 u_long val = 1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100507#else
508 int port = port_in;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100509 struct timeval start_tv;
510 int so_error;
511 socklen_t so_error_len = sizeof(so_error);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100512#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100513 channel_T *channel;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100514 int ret;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100515
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100516#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100517 channel_init_winsock();
518#endif
519
Bram Moolenaar77073442016-02-13 23:23:53 +0100520 channel = add_channel();
521 if (channel == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100522 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100523 ch_error(NULL, "Cannot allocate channel.");
Bram Moolenaar77073442016-02-13 23:23:53 +0100524 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100525 }
526
527 /* Get the server internet address and put into addr structure */
528 /* fill in the socket address structure and connect to server */
529 vim_memset((char *)&server, 0, sizeof(server));
530 server.sin_family = AF_INET;
531 server.sin_port = htons(port);
532 if ((host = gethostbyname(hostname)) == NULL)
533 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100534 ch_error(channel, "in gethostbyname() in channel_open()");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100535 PERROR("E901: gethostbyname() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100536 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100537 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100538 }
539 memcpy((char *)&server.sin_addr, host->h_addr, host->h_length);
540
Bram Moolenaar254e00d2016-02-19 23:23:12 +0100541 /* On Mac and Solaris a zero timeout almost never works. At least wait
542 * one millisecond. Let's do it for all systems, because we don't know why
543 * this is needed. */
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100544 if (waittime == 0)
545 waittime = 1;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100546
547 /*
548 * For Unix we need to call connect() again after connect() failed.
549 * On Win32 one time is sufficient.
550 */
551 while (TRUE)
552 {
553 if (sd >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100554 sock_close(sd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100555 sd = socket(AF_INET, SOCK_STREAM, 0);
556 if (sd == -1)
557 {
558 ch_error(channel, "in socket() in channel_open().");
559 PERROR("E898: socket() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100560 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100561 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100562 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100563
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100564 if (waittime >= 0)
565 {
566 /* Make connect() non-blocking. */
567 if (
568#ifdef _WIN32
569 ioctlsocket(sd, FIONBIO, &val) < 0
570#else
571 fcntl(sd, F_SETFL, O_NONBLOCK) < 0
572#endif
573 )
574 {
575 SOCK_ERRNO;
576 ch_errorn(channel,
577 "channel_open: Connect failed with errno %d", errno);
578 sock_close(sd);
579 channel_free(channel);
580 return NULL;
581 }
582 }
583
584 /* Try connecting to the server. */
585 ch_logsn(channel, "Connecting to %s port %d", hostname, port);
586 ret = connect(sd, (struct sockaddr *)&server, sizeof(server));
587
588 SOCK_ERRNO;
589 if (ret < 0)
590 {
591 if (errno != EWOULDBLOCK
592 && errno != ECONNREFUSED
593
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100594#ifdef EINPROGRESS
595 && errno != EINPROGRESS
596#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100597 )
598 {
599 ch_errorn(channel,
600 "channel_open: Connect failed with errno %d", errno);
601 PERROR(_(e_cannot_connect));
602 sock_close(sd);
603 channel_free(channel);
604 return NULL;
605 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100606 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100607
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100608 /* If we don't block and connect() failed then try using select() to
609 * wait for the connection to be made. */
610 if (waittime >= 0 && ret < 0)
611 {
612 struct timeval tv;
613 fd_set wfds;
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100614#if defined(__APPLE__) && __APPLE__ == 1
615# define PASS_RFDS
Bram Moolenaar98921892016-02-23 17:14:37 +0100616 fd_set rfds;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100617
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100618 FD_ZERO(&rfds);
619 FD_SET(sd, &rfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100620#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100621 FD_ZERO(&wfds);
622 FD_SET(sd, &wfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100623
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100624 tv.tv_sec = waittime / 1000;
625 tv.tv_usec = (waittime % 1000) * 1000;
626#ifndef WIN32
627 gettimeofday(&start_tv, NULL);
628#endif
629 ch_logn(channel,
630 "Waiting for connection (waittime %d msec)...", waittime);
631 ret = select((int)sd + 1,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100632#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100633 &rfds,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100634#else
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100635 NULL,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100636#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100637 &wfds, NULL, &tv);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100638
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100639 if (ret < 0)
640 {
641 SOCK_ERRNO;
642 ch_errorn(channel,
643 "channel_open: Connect failed with errno %d", errno);
644 PERROR(_(e_cannot_connect));
645 sock_close(sd);
646 channel_free(channel);
647 return NULL;
648 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100649#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100650 if (ret == 0 && FD_ISSET(sd, &rfds) && FD_ISSET(sd, &wfds))
651 {
652 /* For OS X, this implies error. See tcp(4). */
653 ch_error(channel, "channel_open: Connect failed");
654 EMSG(_(e_cannot_connect));
655 sock_close(sd);
656 channel_free(channel);
657 return NULL;
658 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100659#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100660#ifdef WIN32
661 /* On Win32 select() is expected to work and wait for up to the
662 * waittime for the socket to be open. */
663 if (!FD_ISSET(sd, &wfds) || ret == 0)
664#else
665 /* See socket(7) for the behavior on Linux-like systems:
666 * After putting the socket in non-blocking mode, connect() will
667 * return EINPROGRESS, select() will not wait (as if writing is
668 * possible), need to use getsockopt() to check if the socket is
669 * actually open. */
670 getsockopt(sd, SOL_SOCKET, SO_ERROR, &so_error, &so_error_len);
671 if (!FD_ISSET(sd, &wfds) || ret == 0 || so_error != 0)
672#endif
673 {
674#ifndef WIN32
675 struct timeval end_tv;
676 long elapsed_msec;
677
678 gettimeofday(&end_tv, NULL);
679 elapsed_msec = (end_tv.tv_sec - start_tv.tv_sec) * 1000
680 + (end_tv.tv_usec - start_tv.tv_usec) / 1000;
681 if (waittime > 1 && elapsed_msec < waittime)
682 {
683 /* The port isn't ready but we also didn't get an error.
684 * This happens when the server didn't open the socket
685 * yet. Wait a bit and try again. */
686 mch_delay(waittime < 50 ? (long)waittime : 50L, TRUE);
687 ui_breakcheck();
688 if (!got_int)
689 {
690 /* reduce the waittime by the elapsed time and the 50
691 * msec delay (or a bit more) */
692 waittime -= elapsed_msec;
693 if (waittime > 50)
694 waittime -= 50;
695 else
696 waittime = 1;
697 continue;
698 }
699 /* we were interrupted, behave as if timed out */
700 }
701#endif
702 /* We timed out. */
703 ch_error(channel, "Connection timed out");
704 sock_close(sd);
705 channel_free(channel);
706 return NULL;
707 }
708
709 ch_log(channel, "Connection made");
710 break;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100711 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100712 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100713
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100714 if (waittime >= 0)
715 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100716#ifdef _WIN32
717 val = 0;
718 ioctlsocket(sd, FIONBIO, &val);
719#else
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100720 (void)fcntl(sd, F_SETFL, 0);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100721#endif
722 }
723
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100724 channel->CH_SOCK_FD = (sock_T)sd;
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100725 channel->ch_nb_close_cb = nb_close_cb;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100726
727#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +0100728 channel_gui_register(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100729#endif
730
Bram Moolenaar77073442016-02-13 23:23:53 +0100731 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100732}
733
734#if defined(CHANNEL_PIPES) || defined(PROTO)
735 void
Bram Moolenaard8070362016-02-15 21:56:54 +0100736channel_set_pipes(channel_T *channel, sock_T in, sock_T out, sock_T err)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100737{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100738 channel->CH_IN_FD = in;
739 channel->CH_OUT_FD = out;
740 channel->CH_ERR_FD = err;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100741}
742#endif
743
744 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100745channel_set_job(channel_T *channel, job_T *job)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100746{
Bram Moolenaar77073442016-02-13 23:23:53 +0100747 channel->ch_job = job;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100748}
749
750/*
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100751 * Set various properties from an "opt" argument.
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100752 */
753 void
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100754channel_set_options(channel_T *channel, jobopt_T *opt)
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100755{
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100756 int part;
757 char_u **cbp;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100758
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100759 if (opt->jo_set & JO_MODE)
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100760 for (part = PART_SOCK; part <= PART_IN; ++part)
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100761 channel->ch_part[part].ch_mode = opt->jo_mode;
762 if (opt->jo_set & JO_IN_MODE)
763 channel->ch_part[PART_IN].ch_mode = opt->jo_in_mode;
764 if (opt->jo_set & JO_OUT_MODE)
765 channel->ch_part[PART_OUT].ch_mode = opt->jo_out_mode;
766 if (opt->jo_set & JO_ERR_MODE)
767 channel->ch_part[PART_ERR].ch_mode = opt->jo_err_mode;
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100768
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100769 if (opt->jo_set & JO_TIMEOUT)
770 for (part = PART_SOCK; part <= PART_IN; ++part)
771 channel->ch_part[part].ch_timeout = opt->jo_timeout;
772 if (opt->jo_set & JO_OUT_TIMEOUT)
773 channel->ch_part[PART_OUT].ch_timeout = opt->jo_out_timeout;
774 if (opt->jo_set & JO_ERR_TIMEOUT)
775 channel->ch_part[PART_ERR].ch_timeout = opt->jo_err_timeout;
776
777 if (opt->jo_set & JO_CALLBACK)
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100778 {
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100779 cbp = &channel->ch_callback;
780 vim_free(*cbp);
781 if (opt->jo_callback != NULL && *opt->jo_callback != NUL)
782 *cbp = vim_strsave(opt->jo_callback);
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100783 else
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100784 *cbp = NULL;
785 }
786 if (opt->jo_set & JO_OUT_CALLBACK)
787 {
788 cbp = &channel->ch_part[PART_OUT].ch_callback;
789 vim_free(*cbp);
790 if (opt->jo_out_cb != NULL && *opt->jo_out_cb != NUL)
791 *cbp = vim_strsave(opt->jo_out_cb);
792 else
793 *cbp = NULL;
794 }
795 if (opt->jo_set & JO_ERR_CALLBACK)
796 {
797 cbp = &channel->ch_part[PART_ERR].ch_callback;
798 vim_free(*cbp);
799 if (opt->jo_err_cb != NULL && *opt->jo_err_cb != NUL)
800 *cbp = vim_strsave(opt->jo_err_cb);
801 else
802 *cbp = NULL;
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100803 }
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100804 if (opt->jo_set & JO_CLOSE_CALLBACK)
805 {
806 cbp = &channel->ch_close_cb;
807 vim_free(*cbp);
808 if (opt->jo_close_cb != NULL && *opt->jo_close_cb != NUL)
809 *cbp = vim_strsave(opt->jo_close_cb);
810 else
811 *cbp = NULL;
812 }
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100813}
814
815/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100816 * Set the callback for "channel"/"part" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100817 */
818 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100819channel_set_req_callback(
820 channel_T *channel,
821 int part,
822 char_u *callback,
823 int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100824{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100825 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100826 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
827
828 if (item != NULL)
829 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100830 item->cq_callback = vim_strsave(callback);
831 item->cq_seq_nr = id;
832 item->cq_prev = head->cq_prev;
833 head->cq_prev = item;
834 item->cq_next = NULL;
835 if (item->cq_prev == NULL)
836 head->cq_next = item;
837 else
838 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100839 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100840}
841
842/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100843 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100844 */
845 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100846invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100847{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100848 typval_T rettv;
849 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100850
Bram Moolenaar77073442016-02-13 23:23:53 +0100851 argv[0].v_type = VAR_CHANNEL;
852 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100853
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100854 call_func(callback, (int)STRLEN(callback),
855 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
Bram Moolenaaree1cffc2016-02-21 19:14:41 +0100856 clear_tv(&rettv);
857
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100858 /* If an echo command was used the cursor needs to be put back where
859 * it belongs. */
860 setcursor();
861 cursor_on();
862 out_flush();
863}
864
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100865/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100866 * Return the first buffer from channel "channel"/"part" and remove it.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100867 * The caller must free it.
868 * Returns NULL if there is nothing.
869 */
870 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100871channel_get(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100872{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100873 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100874 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100875 char_u *p;
876
Bram Moolenaar77073442016-02-13 23:23:53 +0100877 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100878 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100879 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100880 p = node->rq_buffer;
881 head->rq_next = node->rq_next;
882 if (node->rq_next == NULL)
883 head->rq_prev = NULL;
884 else
885 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100886 vim_free(node);
887 return p;
888}
889
890/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100891 * Returns the whole buffer contents concatenated for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100892 */
893 static char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100894channel_get_all(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100895{
896 /* Concatenate everything into one buffer.
897 * TODO: avoid multiple allocations. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100898 while (channel_collapse(channel, part) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100899 ;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100900 return channel_get(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100901}
902
903/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100904 * Collapses the first and second buffer for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100905 * Returns FAIL if that is not possible.
906 */
907 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100908channel_collapse(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100909{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100910 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100911 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100912 char_u *p;
913
Bram Moolenaar77073442016-02-13 23:23:53 +0100914 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100915 return FAIL;
916
Bram Moolenaar77073442016-02-13 23:23:53 +0100917 p = alloc((unsigned)(STRLEN(node->rq_buffer)
918 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100919 if (p == NULL)
920 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100921 STRCPY(p, node->rq_buffer);
922 STRCAT(p, node->rq_next->rq_buffer);
923 vim_free(node->rq_next->rq_buffer);
924 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100925
Bram Moolenaar77073442016-02-13 23:23:53 +0100926 /* dispose of the node and its buffer */
927 head->rq_next = node->rq_next;
928 head->rq_next->rq_prev = NULL;
929 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100930 vim_free(node);
931 return OK;
932}
933
934/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100935 * Store "buf[len]" on "channel"/"part".
936 * Returns OK or FAIL.
937 */
938 static int
939channel_save(channel_T *channel, int part, char_u *buf, int len)
940{
941 readq_T *node;
942 readq_T *head = &channel->ch_part[part].ch_head;
943 char_u *p;
944 int i;
945
946 node = (readq_T *)alloc(sizeof(readq_T));
947 if (node == NULL)
948 return FAIL; /* out of memory */
949 node->rq_buffer = alloc(len + 1);
950 if (node->rq_buffer == NULL)
951 {
952 vim_free(node);
953 return FAIL; /* out of memory */
954 }
955
956 if (channel->ch_part[part].ch_mode == MODE_NL)
957 {
958 /* Drop any CR before a NL. */
959 p = node->rq_buffer;
960 for (i = 0; i < len; ++i)
961 if (buf[i] != CAR || i + 1 >= len || buf[i + 1] != NL)
962 *p++ = buf[i];
963 *p = NUL;
964 }
965 else
966 {
967 mch_memmove(node->rq_buffer, buf, len);
968 node->rq_buffer[len] = NUL;
969 }
970
971 /* append node to the tail of the queue */
972 node->rq_next = NULL;
973 node->rq_prev = head->rq_prev;
974 if (head->rq_prev == NULL)
975 head->rq_next = node;
976 else
977 head->rq_prev->rq_next = node;
978 head->rq_prev = node;
979
980 if (log_fd != NULL)
981 {
982 ch_log_lead("RECV ", channel);
983 fprintf(log_fd, "'");
984 if (fwrite(buf, len, 1, log_fd) != 1)
985 return FAIL;
986 fprintf(log_fd, "'\n");
987 }
988 return OK;
989}
990
991/*
992 * Use the read buffer of "channel"/"part" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100993 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +0100994 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100995 */
Bram Moolenaard7ece102016-02-02 23:23:02 +0100996 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100997channel_parse_json(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100998{
999 js_read_T reader;
1000 typval_T listtv;
1001 jsonq_T *item;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001002 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001003 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001004
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001005 if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001006 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001007
1008 /* TODO: make reader work properly */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001009 /* reader.js_buf = channel_peek(channel, part); */
1010 reader.js_buf = channel_get_all(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001011 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +01001012 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001013 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +01001014 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001015 ret = json_decode(&reader, &listtv,
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001016 channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001017 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001018 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +01001019 /* Only accept the response when it is a list with at least two
1020 * items. */
1021 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001022 {
1023 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001024 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001025 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001026 else
1027 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001028 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
1029 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001030 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001031 else
1032 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001033 item->jq_value = alloc_tv();
1034 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001035 {
1036 vim_free(item);
1037 clear_tv(&listtv);
1038 }
1039 else
1040 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001041 *item->jq_value = listtv;
1042 item->jq_prev = head->jq_prev;
1043 head->jq_prev = item;
1044 item->jq_next = NULL;
1045 if (item->jq_prev == NULL)
1046 head->jq_next = item;
1047 else
1048 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001049 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001050 }
1051 }
1052 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001053
1054 /* Put the unread part back into the channel.
1055 * TODO: insert in front */
1056 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001057 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001058 channel_save(channel, part, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001059 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001060 ret = TRUE;
1061 }
1062 else
1063 ret = FALSE;
1064
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001065 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001066 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001067}
1068
1069/*
Bram Moolenaard46ae142016-02-16 13:33:52 +01001070 * Remove "node" from the queue that it is in. Does not free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +01001071 */
1072 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001073remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001074{
Bram Moolenaar77073442016-02-13 23:23:53 +01001075 if (node->cq_prev == NULL)
1076 head->cq_next = node->cq_next;
1077 else
1078 node->cq_prev->cq_next = node->cq_next;
1079 if (node->cq_next == NULL)
1080 head->cq_prev = node->cq_prev;
1081 else
1082 node->cq_next->cq_prev = node->cq_prev;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001083}
1084
1085/*
1086 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +01001087 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001088 */
1089 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001090remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001091{
Bram Moolenaar77073442016-02-13 23:23:53 +01001092 if (node->jq_prev == NULL)
1093 head->jq_next = node->jq_next;
1094 else
1095 node->jq_prev->jq_next = node->jq_next;
1096 if (node->jq_next == NULL)
1097 head->jq_prev = node->jq_prev;
1098 else
1099 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001100 vim_free(node);
1101}
1102
1103/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001104 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001105 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +01001106 * When "id" is zero or negative jut get the first message. But not the one
1107 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001108 * Return OK when found and return the value in "rettv".
1109 * Return FAIL otherwise.
1110 */
1111 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001112channel_get_json(channel_T *channel, int part, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001113{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001114 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001115 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001116
Bram Moolenaar77073442016-02-13 23:23:53 +01001117 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001118 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001119 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001120 typval_T *tv = &l->lv_first->li_tv;
1121
1122 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001123 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001124 || tv->vval.v_number == 0
1125 || tv->vval.v_number != channel->ch_part[part].ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001126 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001127 *rettv = item->jq_value;
1128 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001129 return OK;
1130 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001131 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001132 }
1133 return FAIL;
1134}
1135
Bram Moolenaarece61b02016-02-20 21:39:05 +01001136#define CH_JSON_MAX_ARGS 4
1137
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001138/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001139 * Execute a command received over "channel"/"part"
Bram Moolenaarece61b02016-02-20 21:39:05 +01001140 * "argv[0]" is the command string.
1141 * "argv[1]" etc. have further arguments, type is VAR_UNKNOWN if missing.
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001142 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001143 static void
Bram Moolenaarece61b02016-02-20 21:39:05 +01001144channel_exe_cmd(channel_T *channel, int part, typval_T *argv)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001145{
Bram Moolenaarece61b02016-02-20 21:39:05 +01001146 char_u *cmd = argv[0].vval.v_string;
1147 char_u *arg;
1148 int options = channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001149
Bram Moolenaarece61b02016-02-20 21:39:05 +01001150 if (argv[1].v_type != VAR_STRING)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001151 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001152 ch_error(channel, "received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001153 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001154 EMSG("E903: received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001155 return;
1156 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001157 arg = argv[1].vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001158 if (arg == NULL)
1159 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001160
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001161 if (STRCMP(cmd, "ex") == 0)
1162 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001163 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001164 }
1165 else if (STRCMP(cmd, "normal") == 0)
1166 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001167 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001168
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001169 ea.arg = arg;
1170 ea.addr_count = 0;
1171 ea.forceit = TRUE; /* no mapping */
1172 ex_normal(&ea);
1173 }
1174 else if (STRCMP(cmd, "redraw") == 0)
1175 {
1176 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001177
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001178 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001179 ex_redraw(&ea);
1180 showruler(FALSE);
1181 setcursor();
1182 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001183#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001184 if (gui.in_use)
1185 {
1186 gui_update_cursor(FALSE, FALSE);
1187 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001188 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001189#endif
1190 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001191 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "call") == 0)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001192 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001193 int is_call = cmd[0] == 'c';
1194 int id_idx = is_call ? 3 : 2;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001195
Bram Moolenaarece61b02016-02-20 21:39:05 +01001196 if (argv[id_idx].v_type != VAR_UNKNOWN
1197 && argv[id_idx].v_type != VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001198 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001199 ch_error(channel, "last argument for expr/call must be a number");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001200 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001201 EMSG("E904: last argument for expr/call must be a number");
1202 }
1203 else if (is_call && argv[2].v_type != VAR_LIST)
1204 {
1205 ch_error(channel, "third argument for call must be a list");
1206 if (p_verbose > 2)
1207 EMSG("E904: third argument for call must be a list");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001208 }
1209 else
1210 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001211 typval_T *tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001212 typval_T res_tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001213 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +01001214 char_u *json = NULL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001215
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001216 /* Don't pollute the display with errors. */
1217 ++emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001218 if (!is_call)
1219 tv = eval_expr(arg, NULL);
1220 else if (func_call(arg, &argv[2], NULL, &res_tv) == OK)
1221 tv = &res_tv;
1222 else
1223 tv = NULL;
1224
1225 if (argv[id_idx].v_type == VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001226 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001227 int id = argv[id_idx].vval.v_number;
1228
Bram Moolenaar55fab432016-02-07 16:53:13 +01001229 if (tv != NULL)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001230 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001231 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001232 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001233 /* If evaluation failed or the result can't be encoded
1234 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001235 vim_free(json);
1236 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001237 err_tv.v_type = VAR_STRING;
1238 err_tv.vval.v_string = (char_u *)"ERROR";
1239 tv = &err_tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001240 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001241 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001242 if (json != NULL)
1243 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001244 channel_send(channel,
1245 part == PART_SOCK ? PART_SOCK : PART_IN,
1246 json, (char *)cmd);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001247 vim_free(json);
1248 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001249 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001250 --emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001251 if (tv == &res_tv)
1252 clear_tv(tv);
1253 else if (tv != &err_tv)
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001254 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001255 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001256 }
1257 else if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001258 {
1259 ch_errors(channel, "Receved unknown command: %s", (char *)cmd);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001260 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001261 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001262}
1263
1264/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001265 * Invoke a callback for "channel"/"part" if needed.
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001266 * Return TRUE when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001267 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001268 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001269may_invoke_callback(channel_T *channel, int part)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001270{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001271 char_u *msg = NULL;
1272 typval_T *listtv = NULL;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001273 typval_T argv[CH_JSON_MAX_ARGS];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001274 int seq_nr = -1;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001275 ch_mode_T ch_mode = channel->ch_part[part].ch_mode;
1276 char_u *callback = NULL;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001277
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001278 if (channel->ch_nb_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001279 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001280 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001281
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001282 if (channel->ch_part[part].ch_callback != NULL)
1283 callback = channel->ch_part[part].ch_callback;
1284 else
1285 callback = channel->ch_callback;
1286
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001287 if (ch_mode == MODE_JSON || ch_mode == MODE_JS)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001288 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001289 listitem_T *item;
1290 int argc = 0;
1291
Bram Moolenaard7ece102016-02-02 23:23:02 +01001292 /* Get any json message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001293 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001294 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001295 /* Parse readahead, return when there is still no message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001296 channel_parse_json(channel, part);
1297 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001298 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001299 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001300
Bram Moolenaarece61b02016-02-20 21:39:05 +01001301 for (item = listtv->vval.v_list->lv_first;
1302 item != NULL && argc < CH_JSON_MAX_ARGS;
1303 item = item->li_next)
1304 argv[argc++] = item->li_tv;
1305 while (argc < CH_JSON_MAX_ARGS)
1306 argv[argc++].v_type = VAR_UNKNOWN;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001307
Bram Moolenaarece61b02016-02-20 21:39:05 +01001308 if (argv[0].v_type == VAR_STRING)
1309 {
1310 char_u *cmd = argv[0].vval.v_string;
1311
1312 /* ["cmd", arg] or ["cmd", arg, arg] or ["cmd", arg, arg, arg] */
Bram Moolenaar77073442016-02-13 23:23:53 +01001313 ch_logs(channel, "Executing %s command", (char *)cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001314 channel_exe_cmd(channel, part, argv);
Bram Moolenaar77073442016-02-13 23:23:53 +01001315 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001316 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001317 }
1318
Bram Moolenaarece61b02016-02-20 21:39:05 +01001319 if (argv[0].v_type != VAR_NUMBER)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001320 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001321 ch_error(channel,
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001322 "Dropping message with invalid sequence number type");
Bram Moolenaar77073442016-02-13 23:23:53 +01001323 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001324 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001325 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001326 seq_nr = argv[0].vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001327 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001328 else if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001329 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001330 /* nothing to read on RAW or NL channel */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001331 return FALSE;
1332 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001333 else
1334 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001335 /* If there is no callback drop the message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001336 if (callback == NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001337 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001338 while ((msg = channel_get(channel, part)) != NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001339 vim_free(msg);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001340 return FALSE;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001341 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001342
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001343 if (ch_mode == MODE_NL)
1344 {
1345 char_u *nl;
1346 char_u *buf;
1347
1348 /* See if we have a message ending in NL in the first buffer. If
1349 * not try to concatenate the first and the second buffer. */
1350 while (TRUE)
1351 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001352 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001353 nl = vim_strchr(buf, NL);
1354 if (nl != NULL)
1355 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001356 if (channel_collapse(channel, part) == FAIL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001357 return FALSE; /* incomplete message */
1358 }
1359 if (nl[1] == NUL)
1360 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001361 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001362 else
1363 {
1364 /* Copy the message into allocated memory and remove it from
1365 * the buffer. */
1366 msg = vim_strnsave(buf, (int)(nl - buf));
1367 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1368 }
1369 }
1370 else
1371 /* For a raw channel we don't know where the message ends, just
1372 * get everything we have. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001373 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001374
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001375 argv[1].v_type = VAR_STRING;
1376 argv[1].vval.v_string = msg;
1377 }
1378
Bram Moolenaara07fec92016-02-05 21:04:08 +01001379 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001380 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001381 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001382 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001383 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001384
1385 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001386 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001387 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001388 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001389 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001390 ch_log(channel, "Invoking one-time callback");
Bram Moolenaard46ae142016-02-16 13:33:52 +01001391 /* Remove the item from the list first, if the callback
1392 * invokes ch_close() the list will be cleared. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001393 remove_cb_node(head, item);
Bram Moolenaard46ae142016-02-16 13:33:52 +01001394 invoke_callback(channel, item->cq_callback, argv);
1395 vim_free(item->cq_callback);
1396 vim_free(item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001397 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001398 break;
1399 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001400 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001401 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001402 if (!done)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001403 ch_log(channel, "Dropping message without callback");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001404 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001405 else if (callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001406 {
1407 /* invoke the channel callback */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001408 ch_log(channel, "Invoking channel callback");
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001409 invoke_callback(channel, callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001410 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001411 else
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001412 ch_log(channel, "Dropping message");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001413
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001414 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001415 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001416 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001417
1418 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001419}
1420
1421/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001422 * Return TRUE when channel "channel" is open for writing to.
1423 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001424 */
1425 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001426channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001427{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001428 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001429#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001430 || channel->CH_IN_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001431#endif
1432 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001433}
1434
1435/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001436 * Return TRUE when channel "channel" is open for reading or writing.
1437 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001438 */
1439 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001440channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001441{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001442 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001443#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001444 || channel->CH_IN_FD != INVALID_FD
1445 || channel->CH_OUT_FD != INVALID_FD
1446 || channel->CH_ERR_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001447#endif
1448 );
1449}
1450
1451/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001452 * Return a string indicating the status of the channel.
1453 */
1454 char *
1455channel_status(channel_T *channel)
1456{
1457 if (channel == NULL)
1458 return "fail";
1459 if (channel_is_open(channel))
1460 return "open";
1461 return "closed";
1462}
1463
1464/*
1465 * Close channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001466 * This does not trigger the close callback.
1467 */
1468 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001469channel_close(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001470{
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001471 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001472
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001473#ifdef FEAT_GUI
1474 channel_gui_unregister(channel);
1475#endif
1476
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001477 if (channel->CH_SOCK_FD != INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001478 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001479 sock_close(channel->CH_SOCK_FD);
1480 channel->CH_SOCK_FD = INVALID_FD;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001481 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001482#if defined(CHANNEL_PIPES)
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001483 if (channel->CH_IN_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001484 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001485 fd_close(channel->CH_IN_FD);
1486 channel->CH_IN_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001487 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001488 if (channel->CH_OUT_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001489 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001490 fd_close(channel->CH_OUT_FD);
1491 channel->CH_OUT_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001492 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001493 if (channel->CH_ERR_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001494 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001495 fd_close(channel->CH_ERR_FD);
1496 channel->CH_ERR_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001497 }
1498#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001499
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001500 if (channel->ch_close_cb != NULL)
1501 {
1502 typval_T argv[1];
1503 typval_T rettv;
1504 int dummy;
1505
1506 /* invoke the close callback; increment the refcount to avoid it
1507 * being freed halfway */
1508 argv[0].v_type = VAR_CHANNEL;
1509 argv[0].vval.v_channel = channel;
1510 ++channel->ch_refcount;
1511 call_func(channel->ch_close_cb, (int)STRLEN(channel->ch_close_cb),
1512 &rettv, 1, argv, 0L, 0L, &dummy, TRUE, NULL);
1513 clear_tv(&rettv);
1514 --channel->ch_refcount;
1515
1516 /* the callback is only called once */
1517 vim_free(channel->ch_close_cb);
1518 channel->ch_close_cb = NULL;
1519 }
1520
1521 channel->ch_nb_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001522 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001523}
1524
Bram Moolenaard04a0202016-01-26 23:30:18 +01001525/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001526 * Return the first buffer from "channel"/"part" without removing it.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001527 * Returns NULL if there is nothing.
1528 */
1529 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001530channel_peek(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001531{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001532 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001533
Bram Moolenaar77073442016-02-13 23:23:53 +01001534 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001535 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001536 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001537}
1538
1539/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001540 * Clear the read buffer on "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001541 */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001542 static void
1543channel_clear_one(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001544{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001545 jsonq_T *json_head = &channel->ch_part[part].ch_json_head;
1546 cbq_T *cb_head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001547
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001548 while (channel_peek(channel, part) != NULL)
1549 vim_free(channel_get(channel, part));
Bram Moolenaar77073442016-02-13 23:23:53 +01001550
1551 while (cb_head->cq_next != NULL)
Bram Moolenaard46ae142016-02-16 13:33:52 +01001552 {
1553 cbq_T *node = cb_head->cq_next;
1554
1555 remove_cb_node(cb_head, node);
1556 vim_free(node->cq_callback);
1557 vim_free(node);
1558 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001559
1560 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001561 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001562 free_tv(json_head->jq_next->jq_value);
1563 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001564 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001565
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001566 vim_free(channel->ch_part[part].ch_callback);
1567 channel->ch_part[part].ch_callback = NULL;
1568}
1569
1570/*
1571 * Clear all the read buffers on "channel".
1572 */
1573 void
1574channel_clear(channel_T *channel)
1575{
1576 channel_clear_one(channel, PART_SOCK);
1577#ifdef CHANNEL_PIPES
1578 channel_clear_one(channel, PART_OUT);
1579 channel_clear_one(channel, PART_ERR);
1580#endif
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001581 vim_free(channel->ch_callback);
1582 channel->ch_callback = NULL;
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001583 vim_free(channel->ch_close_cb);
1584 channel->ch_close_cb = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001585}
1586
Bram Moolenaar77073442016-02-13 23:23:53 +01001587#if defined(EXITFREE) || defined(PROTO)
1588 void
1589channel_free_all(void)
1590{
1591 channel_T *channel;
1592
1593 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1594 channel_clear(channel);
1595}
1596#endif
1597
1598
Bram Moolenaard04a0202016-01-26 23:30:18 +01001599/* Sent when the channel is found closed when reading. */
Bram Moolenaareed284a2016-02-22 23:13:33 +01001600#define DETACH_MSG_RAW "DETACH\n"
1601#define DETACH_MSG_JSON "\"DETACH\"\n"
Bram Moolenaard04a0202016-01-26 23:30:18 +01001602
1603/* Buffer size for reading incoming messages. */
1604#define MAXMSGSIZE 4096
1605
1606/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001607 * Check for reading from "fd" with "timeout" msec.
1608 * Return FAIL when there is nothing to read.
1609 */
1610 static int
Bram Moolenaard8070362016-02-15 21:56:54 +01001611channel_wait(channel_T *channel, sock_T fd, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001612{
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001613 if (timeout > 0)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001614 ch_logn(channel, "Waiting for up to %d msec", timeout);
Bram Moolenaard8070362016-02-15 21:56:54 +01001615
Bram Moolenaard8070362016-02-15 21:56:54 +01001616# ifdef WIN32
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001617 if (fd != channel->CH_SOCK_FD)
Bram Moolenaard8070362016-02-15 21:56:54 +01001618 {
1619 DWORD nread;
1620 int diff;
1621 DWORD deadline = GetTickCount() + timeout;
1622
1623 /* reading from a pipe, not a socket */
1624 while (TRUE)
1625 {
Bram Moolenaare74e8e72016-02-16 22:01:30 +01001626 if (PeekNamedPipe((HANDLE)fd, NULL, 0, NULL, &nread, NULL)
1627 && nread > 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001628 return OK;
1629 diff = deadline - GetTickCount();
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001630 if (diff <= 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001631 break;
1632 /* Wait for 5 msec.
1633 * TODO: increase the sleep time when looping more often */
1634 Sleep(5);
1635 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001636 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001637 else
Bram Moolenaard8070362016-02-15 21:56:54 +01001638#endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001639 {
Bram Moolenaar9186a272016-02-23 19:34:01 +01001640#if defined(HAVE_SELECT)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001641 struct timeval tval;
1642 fd_set rfds;
1643 int ret;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001644
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001645 FD_ZERO(&rfds);
1646 FD_SET((int)fd, &rfds);
1647 tval.tv_sec = timeout / 1000;
1648 tval.tv_usec = (timeout % 1000) * 1000;
1649 for (;;)
1650 {
1651 ret = select((int)fd + 1, &rfds, NULL, NULL, &tval);
Bram Moolenaar9186a272016-02-23 19:34:01 +01001652# ifdef EINTR
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001653 SOCK_ERRNO;
1654 if (ret == -1 && errno == EINTR)
1655 continue;
Bram Moolenaar9186a272016-02-23 19:34:01 +01001656# endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001657 if (ret > 0)
1658 return OK;
1659 break;
1660 }
Bram Moolenaar9186a272016-02-23 19:34:01 +01001661#else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001662 struct pollfd fds;
1663
1664 fds.fd = fd;
1665 fds.events = POLLIN;
1666 if (poll(&fds, 1, timeout) > 0)
1667 return OK;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001668#endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001669 }
1670 ch_log(channel, "Nothing to read");
1671 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001672}
1673
1674/*
1675 * Return a unique ID to be used in a message.
1676 */
1677 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001678channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001679{
1680 static int next_id = 1;
1681
1682 return next_id++;
1683}
1684
1685/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001686 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001687 * "part" is PART_SOCK, PART_OUT or PART_ERR.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001688 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001689 */
1690 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001691channel_read(channel_T *channel, int part, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001692{
1693 static char_u *buf = NULL;
1694 int len = 0;
1695 int readlen = 0;
Bram Moolenaard8070362016-02-15 21:56:54 +01001696 sock_T fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001697 int use_socket = FALSE;
Bram Moolenaareed284a2016-02-22 23:13:33 +01001698 char *msg;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001699
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001700 fd = channel->ch_part[part].ch_fd;
1701 if (fd == INVALID_FD)
1702 {
1703 ch_error(channel, "channel_read() called while socket is closed");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001704 return;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001705 }
1706 use_socket = fd == channel->CH_SOCK_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001707
1708 /* Allocate a buffer to read into. */
1709 if (buf == NULL)
1710 {
1711 buf = alloc(MAXMSGSIZE);
1712 if (buf == NULL)
1713 return; /* out of memory! */
1714 }
1715
1716 /* Keep on reading for as long as there is something to read.
1717 * Use select() or poll() to avoid blocking on a message that is exactly
1718 * MAXMSGSIZE long. */
1719 for (;;)
1720 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001721 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001722 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001723 if (use_socket)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001724 len = sock_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001725 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001726 len = fd_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001727 if (len <= 0)
1728 break; /* error or nothing more to read */
1729
1730 /* Store the read message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001731 channel_save(channel, part, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001732 readlen += len;
1733 if (len < MAXMSGSIZE)
1734 break; /* did read everything that's available */
1735 }
1736
Bram Moolenaarbd73ae12016-02-22 22:19:22 +01001737 /* Reading a disconnection (readlen == 0), or an error.
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001738 * TODO: call error callback. */
Bram Moolenaarbd73ae12016-02-22 22:19:22 +01001739 if (readlen <= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001740 {
1741 /* Queue a "DETACH" netbeans message in the command queue in order to
1742 * terminate the netbeans session later. Do not end the session here
1743 * directly as we may be running in the context of a call to
1744 * netbeans_parse_messages():
1745 * netbeans_parse_messages
1746 * -> autocmd triggered while processing the netbeans cmd
1747 * -> ui_breakcheck
1748 * -> gui event loop or select loop
1749 * -> channel_read()
1750 */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001751 ch_errors(channel, "%s(): Cannot read", func);
Bram Moolenaareed284a2016-02-22 23:13:33 +01001752 msg = channel->ch_part[part].ch_mode == MODE_RAW
1753 || channel->ch_part[part].ch_mode == MODE_NL
1754 ? DETACH_MSG_RAW : DETACH_MSG_JSON;
1755 channel_save(channel, part, (char_u *)msg, (int)STRLEN(msg));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001756
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001757 /* TODO: When reading from stdout is not possible, should we try to
1758 * keep stdin and stderr open? Probably not, assume the other side
1759 * has died. */
1760 channel_close(channel);
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001761 if (channel->ch_nb_close_cb != NULL)
1762 (*channel->ch_nb_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001763
1764 if (len < 0)
1765 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001766 ch_error(channel, "channel_read(): cannot read from channel");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001767 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001768 }
1769 }
1770
1771#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001772 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001773 if (CH_HAS_GUI && gtk_main_level() > 0)
1774 gtk_main_quit();
1775#endif
1776}
1777
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001778/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001779 * Read from RAW or NL "channel"/"part". Blocks until there is something to
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001780 * read or the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001781 * Returns what was read in allocated memory.
1782 * Returns NULL in case of error or timeout.
1783 */
1784 char_u *
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001785channel_read_block(channel_T *channel, int part, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001786{
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001787 char_u *buf;
1788 char_u *msg;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001789 ch_mode_T mode = channel->ch_part[part].ch_mode;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001790 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001791 char_u *nl;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001792
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001793 ch_logsn(channel, "Blocking %s read, timeout: %d msec",
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001794 mode == MODE_RAW ? "RAW" : "NL", timeout);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001795
1796 while (TRUE)
1797 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001798 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001799 if (buf != NULL && (mode == MODE_RAW
1800 || (mode == MODE_NL && vim_strchr(buf, NL) != NULL)))
1801 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001802 if (buf != NULL && channel_collapse(channel, part) == OK)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001803 continue;
1804
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001805 /* Wait for up to the channel timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001806 if (fd == INVALID_FD
1807 || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001808 return NULL;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001809 channel_read(channel, part, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001810 }
1811
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001812 if (mode == MODE_RAW)
1813 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001814 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001815 }
1816 else
1817 {
1818 nl = vim_strchr(buf, NL);
1819 if (nl[1] == NUL)
1820 {
1821 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001822 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001823 *nl = NUL;
1824 }
1825 else
1826 {
1827 /* Copy the message into allocated memory and remove it from the
1828 * buffer. */
1829 msg = vim_strnsave(buf, (int)(nl - buf));
1830 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1831 }
1832 }
1833 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001834 ch_logn(channel, "Returning %d bytes", (int)STRLEN(msg));
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001835 return msg;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001836}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001837
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001838/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001839 * Read one JSON message with ID "id" from "channel"/"part" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001840 * result in "rettv".
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001841 * When "id" is -1 accept any message;
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001842 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001843 */
1844 int
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001845channel_read_json_block(
1846 channel_T *channel,
1847 int part,
1848 int timeout,
1849 int id,
1850 typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001851{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001852 int more;
Bram Moolenaard8070362016-02-15 21:56:54 +01001853 sock_T fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001854
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001855 ch_log(channel, "Reading JSON");
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001856 if (id != -1)
1857 channel->ch_part[part].ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001858 for (;;)
1859 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001860 more = channel_parse_json(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001861
1862 /* search for messsage "id" */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001863 if (channel_get_json(channel, part, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001864 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001865 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001866 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001867 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001868
Bram Moolenaard7ece102016-02-02 23:23:02 +01001869 if (!more)
1870 {
1871 /* Handle any other messages in the queue. If done some more
1872 * messages may have arrived. */
1873 if (channel_parse_messages())
1874 continue;
1875
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001876 /* Wait for up to the timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001877 fd = channel->ch_part[part].ch_fd;
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001878 if (fd == INVALID_FD || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001879 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001880 channel_read(channel, part, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001881 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001882 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001883 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001884 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001885}
1886
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001887# if defined(WIN32) || defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK) \
1888 || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001889/*
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001890 * Lookup the channel from the socket. Set "partp" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001891 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001892 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001893 channel_T *
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001894channel_fd2channel(sock_T fd, int *partp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001895{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001896 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001897 int part;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001898
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001899 if (fd != INVALID_FD)
Bram Moolenaar77073442016-02-13 23:23:53 +01001900 for (channel = first_channel; channel != NULL;
1901 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001902 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001903# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001904 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001905# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001906 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001907# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001908 if (channel->ch_part[part].ch_fd == fd)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001909 {
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001910 *partp = part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001911 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001912 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001913 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001914 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001915}
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001916# endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001917
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001918# if defined(WIN32) || defined(PROTO)
1919/*
1920 * Check the channels for anything that is ready to be read.
1921 * The data is put in the read queue.
1922 */
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001923 void
1924channel_handle_events(void)
1925{
1926 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001927 int part;
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001928 sock_T fd;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001929
1930 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1931 {
Bram Moolenaar9186a272016-02-23 19:34:01 +01001932# ifdef CHANNEL_PIPES
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001933 /* check the socket and pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001934 for (part = PART_SOCK; part <= PART_ERR; ++part)
Bram Moolenaar9186a272016-02-23 19:34:01 +01001935# else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001936 /* only check the socket */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001937 part = PART_SOCK;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001938# endif
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001939 {
1940 fd = channel->ch_part[part].ch_fd;
1941 if (fd != INVALID_FD && channel_wait(channel, fd, 0) == OK)
1942 channel_read(channel, part, "channel_handle_events");
1943 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001944 }
1945}
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001946# endif
1947
Bram Moolenaard04a0202016-01-26 23:30:18 +01001948/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001949 * Write "buf" (NUL terminated string) to "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001950 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001951 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001952 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001953 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001954channel_send(channel_T *channel, int part, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001955{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001956 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001957 int res;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001958 sock_T fd;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001959
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001960 fd = channel->ch_part[part].ch_fd;
1961 if (fd == INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001962 {
1963 if (!channel->ch_error && fun != NULL)
1964 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001965 ch_errors(channel, "%s(): write while not connected", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001966 EMSG2("E630: %s(): write while not connected", fun);
1967 }
1968 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001969 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001970 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001971
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001972 if (log_fd != NULL)
1973 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001974 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001975 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001976 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001977 fprintf(log_fd, "'\n");
1978 fflush(log_fd);
1979 }
1980
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001981 if (part == PART_SOCK)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001982 res = sock_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001983 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001984 res = fd_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001985 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001986 {
1987 if (!channel->ch_error && fun != NULL)
1988 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001989 ch_errors(channel, "%s(): write failed", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001990 EMSG2("E631: %s(): write failed", fun);
1991 }
1992 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001993 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001994 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001995
1996 channel->ch_error = FALSE;
1997 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001998}
1999
2000# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002001/*
2002 * Add open channels to the poll struct.
2003 * Return the adjusted struct index.
2004 * The type of "fds" is hidden to avoid problems with the function proto.
2005 */
2006 int
2007channel_poll_setup(int nfd_in, void *fds_in)
2008{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002009 int nfd = nfd_in;
2010 channel_T *channel;
2011 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002012 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002013
Bram Moolenaar77073442016-02-13 23:23:53 +01002014 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002015 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002016# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002017 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002018# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002019 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002020# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002021 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002022 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002023 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002024 channel->ch_part[part].ch_poll_idx = nfd;
2025 fds[nfd].fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002026 fds[nfd].events = POLLIN;
2027 nfd++;
2028 }
2029 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002030 channel->ch_part[part].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002031 }
2032 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002033
2034 return nfd;
2035}
2036
2037/*
2038 * The type of "fds" is hidden to avoid problems with the function proto.
2039 */
2040 int
2041channel_poll_check(int ret_in, void *fds_in)
2042{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002043 int ret = ret_in;
2044 channel_T *channel;
2045 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002046 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002047
Bram Moolenaar77073442016-02-13 23:23:53 +01002048 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002049 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002050# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002051 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002052# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002053 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002054# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002055 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002056 int idx = channel->ch_part[part].ch_poll_idx;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002057
2058 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
2059 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002060 channel_read(channel, part, "channel_poll_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002061 --ret;
2062 }
2063 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002064 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002065
2066 return ret;
2067}
Bram Moolenaard04a0202016-01-26 23:30:18 +01002068# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002069
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002070# if (!defined(WIN32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002071/*
2072 * The type of "rfds" is hidden to avoid problems with the function proto.
2073 */
2074 int
2075channel_select_setup(int maxfd_in, void *rfds_in)
2076{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002077 int maxfd = maxfd_in;
2078 channel_T *channel;
2079 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002080 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002081
Bram Moolenaar77073442016-02-13 23:23:53 +01002082 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002083 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002084# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002085 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002086# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002087 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002088# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002089 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002090 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002091
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002092 if (fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002093 {
Bram Moolenaard8070362016-02-15 21:56:54 +01002094 FD_SET((int)fd, rfds);
2095 if (maxfd < (int)fd)
2096 maxfd = (int)fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002097 }
2098 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002099 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002100
2101 return maxfd;
2102}
2103
2104/*
2105 * The type of "rfds" is hidden to avoid problems with the function proto.
2106 */
2107 int
2108channel_select_check(int ret_in, void *rfds_in)
2109{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002110 int ret = ret_in;
2111 channel_T *channel;
2112 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002113 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002114
Bram Moolenaar77073442016-02-13 23:23:53 +01002115 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002116 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002117# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002118 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002119# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002120 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002121# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002122 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002123 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002124
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002125 if (ret > 0 && fd != INVALID_FD && FD_ISSET(fd, rfds))
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002126 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002127 channel_read(channel, part, "channel_select_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002128 --ret;
2129 }
2130 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002131 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002132
2133 return ret;
2134}
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002135# endif /* !WIN32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002136
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002137/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01002138 * Execute queued up commands.
2139 * Invoked from the main loop when it's safe to execute received commands.
2140 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002141 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01002142 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002143channel_parse_messages(void)
2144{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002145 channel_T *channel = first_channel;
2146 int ret = FALSE;
2147 int r;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002148 int part = PART_SOCK;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002149
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002150 while (channel != NULL)
2151 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002152 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaard7ece102016-02-02 23:23:02 +01002153 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002154 /* Increase the refcount, in case the handler causes the channel
2155 * to be unreferenced or closed. */
2156 ++channel->ch_refcount;
2157 r = may_invoke_callback(channel, part);
2158 if (r == OK)
2159 ret = TRUE;
2160 if (channel_unref(channel) || r == OK)
2161 {
2162 /* channel was freed or something was done, start over */
2163 channel = first_channel;
2164 part = PART_SOCK;
2165 continue;
2166 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002167 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002168#ifdef CHANNEL_PIPES
2169 if (part < PART_ERR)
2170 ++part;
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002171 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002172#endif
2173 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002174 channel = channel->ch_next;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002175 part = PART_SOCK;
2176 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002177 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002178 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002179}
2180
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01002181/*
2182 * Mark references to lists used in channels.
2183 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002184 int
2185set_ref_in_channel(int copyID)
2186{
Bram Moolenaar77073442016-02-13 23:23:53 +01002187 int abort = FALSE;
2188 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002189 int part;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002190
Bram Moolenaar77073442016-02-13 23:23:53 +01002191 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002192 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002193#ifdef CHANNEL_PIPES
2194 for (part = PART_SOCK; part < PART_IN; ++part)
2195#else
2196 part = PART_SOCK;
2197#endif
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002198 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002199 jsonq_T *head = &channel->ch_part[part].ch_json_head;
2200 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002201
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002202 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002203 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002204 list_T *l = item->jq_value->vval.v_list;
2205
2206 if (l->lv_copyID != copyID)
2207 {
2208 l->lv_copyID = copyID;
2209 abort = abort || set_ref_in_list(l, copyID, NULL);
2210 }
2211 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002212 }
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002213 }
2214 }
2215 return abort;
2216}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002217
2218/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002219 * Return the "part" to write to for "channel".
2220 */
2221 int
2222channel_part_send(channel_T *channel)
2223{
2224#ifdef CHANNEL_PIPES
2225 if (channel->CH_SOCK_FD == INVALID_FD)
2226 return PART_IN;
2227#endif
2228 return PART_SOCK;
2229}
2230
2231/*
2232 * Return the default "part" to read from for "channel".
2233 */
2234 int
2235channel_part_read(channel_T *channel)
2236{
2237#ifdef CHANNEL_PIPES
2238 if (channel->CH_SOCK_FD == INVALID_FD)
2239 return PART_OUT;
2240#endif
2241 return PART_SOCK;
2242}
2243
2244/*
2245 * Return the mode of "channel"/"part"
Bram Moolenaar77073442016-02-13 23:23:53 +01002246 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002247 */
2248 ch_mode_T
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002249channel_get_mode(channel_T *channel, int part)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002250{
Bram Moolenaar77073442016-02-13 23:23:53 +01002251 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002252 return MODE_JSON;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002253 return channel->ch_part[part].ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002254}
2255
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01002256/*
2257 * Return the timeout of "channel"/"part"
2258 */
2259 int
2260channel_get_timeout(channel_T *channel, int part)
2261{
2262 return channel->ch_part[part].ch_timeout;
2263}
2264
Bram Moolenaare0874f82016-01-24 20:36:41 +01002265#endif /* FEAT_CHANNEL */