blob: 4a2b20c442ca334245a6c39c1643e1069bcbc695 [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 Moolenaarc8dcbb12016-02-25 23:10:17 +0100310 * Return TRUE if "channel" has a callback.
311 */
312 static int
313channel_has_callback(channel_T *channel)
314{
315 return channel->ch_callback != NULL
316#ifdef CHANNEL_PIPES
317 || channel->ch_part[PART_OUT].ch_callback != NULL
318 || channel->ch_part[PART_ERR].ch_callback != NULL
319#endif
320 || channel->ch_close_cb != NULL;
321}
322
323/*
324 * Close a channel and free all its resources if there is no further action
325 * possible, there is no callback to be invoked.
326 */
327 void
328channel_may_free(channel_T *channel)
329{
330 if (!channel_has_callback(channel))
331 channel_free(channel);
332}
333
334/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100335 * Close a channel and free all its resources.
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100336 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100337 void
338channel_free(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100339{
Bram Moolenaar8b374212016-02-24 20:43:06 +0100340 channel_close(channel, TRUE);
Bram Moolenaar77073442016-02-13 23:23:53 +0100341 if (channel->ch_next != NULL)
342 channel->ch_next->ch_prev = channel->ch_prev;
343 if (channel->ch_prev == NULL)
344 first_channel = channel->ch_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100345 else
Bram Moolenaar77073442016-02-13 23:23:53 +0100346 channel->ch_prev->ch_next = channel->ch_next;
347 vim_free(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100348}
349
Bram Moolenaard04a0202016-01-26 23:30:18 +0100350#if defined(FEAT_GUI) || defined(PROTO)
Bram Moolenaar77073442016-02-13 23:23:53 +0100351
352#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
353 static void
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100354channel_read_fd(int fd)
Bram Moolenaar77073442016-02-13 23:23:53 +0100355{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100356 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100357 int part;
Bram Moolenaar77073442016-02-13 23:23:53 +0100358
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100359 channel = channel_fd2channel(fd, &part);
Bram Moolenaar77073442016-02-13 23:23:53 +0100360 if (channel == NULL)
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100361 ch_errorn(NULL, "Channel for fd %d not found", fd);
Bram Moolenaar77073442016-02-13 23:23:53 +0100362 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100363 channel_read(channel, part, "messageFromNetbeans");
Bram Moolenaar77073442016-02-13 23:23:53 +0100364}
365#endif
366
Bram Moolenaare0874f82016-01-24 20:36:41 +0100367/*
Bram Moolenaard04a0202016-01-26 23:30:18 +0100368 * Read a command from netbeans.
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100369 * TODO: instead of channel ID use the FD.
Bram Moolenaare0874f82016-01-24 20:36:41 +0100370 */
Bram Moolenaard04a0202016-01-26 23:30:18 +0100371#ifdef FEAT_GUI_X11
372 static void
373messageFromNetbeans(XtPointer clientData,
374 int *unused1 UNUSED,
375 XtInputId *unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100376{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100377 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100378}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100379#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100380
Bram Moolenaard04a0202016-01-26 23:30:18 +0100381#ifdef FEAT_GUI_GTK
Bram Moolenaar98921892016-02-23 17:14:37 +0100382# if GTK_CHECK_VERSION(3,0,0)
383 static gboolean
384messageFromNetbeans(GIOChannel *unused1 UNUSED,
385 GIOCondition unused2 UNUSED,
386 gpointer clientData)
387{
388 channel_read_fd(GPOINTER_TO_INT(clientData));
389 return TRUE; /* Return FALSE instead in case the event source is to
390 * be removed after this function returns. */
391}
392# else
Bram Moolenaard04a0202016-01-26 23:30:18 +0100393 static void
394messageFromNetbeans(gpointer clientData,
395 gint unused1 UNUSED,
396 GdkInputCondition unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100397{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100398 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100399}
Bram Moolenaar98921892016-02-23 17:14:37 +0100400# endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100401#endif
402
403 static void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100404channel_gui_register_one(channel_T *channel, int part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100405{
406# ifdef FEAT_GUI_X11
407 /* Tell notifier we are interested in being called
408 * when there is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100409 if (channel->ch_part[part].ch_inputHandler == (XtInputId)NULL)
410 channel->ch_part[part].ch_inputHandler = XtAppAddInput(
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100411 (XtAppContext)app_context,
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100412 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100413 (XtPointer)(XtInputReadMask + XtInputExceptMask),
414 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100415 (XtPointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100416# else
417# ifdef FEAT_GUI_GTK
418 /* Tell gdk we are interested in being called when there
419 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100420 if (channel->ch_part[part].ch_inputHandler == 0)
Bram Moolenaar98921892016-02-23 17:14:37 +0100421# if GTK_CHECK_VERSION(3,0,0)
422 {
423 GIOChannel *chnnl = g_io_channel_unix_new(
424 (gint)channel->ch_part[part].ch_fd);
425
426 channel->ch_part[part].ch_inputHandler = g_io_add_watch(
427 chnnl,
428 G_IO_IN|G_IO_HUP|G_IO_ERR|G_IO_PRI,
429 messageFromNetbeans,
430 GINT_TO_POINTER(channel->ch_part[part].ch_fd));
431
432 g_io_channel_unref(chnnl);
433 }
434# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100435 channel->ch_part[part].ch_inputHandler = gdk_input_add(
436 (gint)channel->ch_part[part].ch_fd,
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100437 (GdkInputCondition)
438 ((int)GDK_INPUT_READ + (int)GDK_INPUT_EXCEPTION),
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100439 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100440 (gpointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar98921892016-02-23 17:14:37 +0100441# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100442# endif
443# endif
444}
445
446 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100447channel_gui_register(channel_T *channel)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100448{
Bram Moolenaard04a0202016-01-26 23:30:18 +0100449 if (!CH_HAS_GUI)
450 return;
451
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100452 if (channel->CH_SOCK_FD != INVALID_FD)
453 channel_gui_register_one(channel, PART_SOCK);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100454# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100455 if (channel->CH_OUT_FD != INVALID_FD)
456 channel_gui_register_one(channel, PART_OUT);
457 if (channel->CH_ERR_FD != INVALID_FD)
458 channel_gui_register_one(channel, PART_ERR);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100459# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100460}
461
462/*
463 * Register any of our file descriptors with the GUI event handling system.
464 * Called when the GUI has started.
465 */
466 void
467channel_gui_register_all(void)
468{
Bram Moolenaar77073442016-02-13 23:23:53 +0100469 channel_T *channel;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100470
Bram Moolenaar77073442016-02-13 23:23:53 +0100471 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100472 channel_gui_register(channel);
473}
474
475 static void
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100476channel_gui_unregister(channel_T *channel)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100477{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100478 int part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100479
480#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100481 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100482#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100483 part = PART_SOCK;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100484#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100485 {
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100486# ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100487 if (channel->ch_part[part].ch_inputHandler != (XtInputId)NULL)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100488 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100489 XtRemoveInput(channel->ch_part[part].ch_inputHandler);
490 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100491 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100492# else
493# ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100494 if (channel->ch_part[part].ch_inputHandler != 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100495 {
Bram Moolenaar98921892016-02-23 17:14:37 +0100496# if GTK_CHECK_VERSION(3,0,0)
497 g_source_remove(channel->ch_part[part].ch_inputHandler);
498# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100499 gdk_input_remove(channel->ch_part[part].ch_inputHandler);
Bram Moolenaar98921892016-02-23 17:14:37 +0100500# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100501 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100502 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100503# endif
504# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100505 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100506}
507
508#endif
509
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100510static char *e_cannot_connect = N_("E902: Cannot connect to port");
511
Bram Moolenaard04a0202016-01-26 23:30:18 +0100512/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100513 * Open a socket channel to "hostname":"port".
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100514 * "waittime" is the time in msec to wait for the connection.
515 * When negative wait forever.
Bram Moolenaar77073442016-02-13 23:23:53 +0100516 * Returns the channel for success.
517 * Returns NULL for failure.
Bram Moolenaard04a0202016-01-26 23:30:18 +0100518 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100519 channel_T *
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100520channel_open(
521 char *hostname,
522 int port_in,
523 int waittime,
524 void (*nb_close_cb)(void))
Bram Moolenaard04a0202016-01-26 23:30:18 +0100525{
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100526 int sd = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100527 struct sockaddr_in server;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100528 struct hostent *host;
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100529#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100530 u_short port = port_in;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100531 u_long val = 1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100532#else
533 int port = port_in;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100534 struct timeval start_tv;
535 int so_error;
536 socklen_t so_error_len = sizeof(so_error);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100537#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100538 channel_T *channel;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100539 int ret;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100540
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100541#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100542 channel_init_winsock();
543#endif
544
Bram Moolenaar77073442016-02-13 23:23:53 +0100545 channel = add_channel();
546 if (channel == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100547 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100548 ch_error(NULL, "Cannot allocate channel.");
Bram Moolenaar77073442016-02-13 23:23:53 +0100549 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100550 }
551
552 /* Get the server internet address and put into addr structure */
553 /* fill in the socket address structure and connect to server */
554 vim_memset((char *)&server, 0, sizeof(server));
555 server.sin_family = AF_INET;
556 server.sin_port = htons(port);
557 if ((host = gethostbyname(hostname)) == NULL)
558 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100559 ch_error(channel, "in gethostbyname() in channel_open()");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100560 PERROR("E901: gethostbyname() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100561 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100562 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100563 }
564 memcpy((char *)&server.sin_addr, host->h_addr, host->h_length);
565
Bram Moolenaar254e00d2016-02-19 23:23:12 +0100566 /* On Mac and Solaris a zero timeout almost never works. At least wait
567 * one millisecond. Let's do it for all systems, because we don't know why
568 * this is needed. */
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100569 if (waittime == 0)
570 waittime = 1;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100571
572 /*
573 * For Unix we need to call connect() again after connect() failed.
574 * On Win32 one time is sufficient.
575 */
576 while (TRUE)
577 {
578 if (sd >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100579 sock_close(sd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100580 sd = socket(AF_INET, SOCK_STREAM, 0);
581 if (sd == -1)
582 {
583 ch_error(channel, "in socket() in channel_open().");
584 PERROR("E898: socket() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100585 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100586 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100587 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100588
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100589 if (waittime >= 0)
590 {
591 /* Make connect() non-blocking. */
592 if (
593#ifdef _WIN32
594 ioctlsocket(sd, FIONBIO, &val) < 0
595#else
596 fcntl(sd, F_SETFL, O_NONBLOCK) < 0
597#endif
598 )
599 {
600 SOCK_ERRNO;
601 ch_errorn(channel,
602 "channel_open: Connect failed with errno %d", errno);
603 sock_close(sd);
604 channel_free(channel);
605 return NULL;
606 }
607 }
608
609 /* Try connecting to the server. */
610 ch_logsn(channel, "Connecting to %s port %d", hostname, port);
611 ret = connect(sd, (struct sockaddr *)&server, sizeof(server));
612
613 SOCK_ERRNO;
614 if (ret < 0)
615 {
616 if (errno != EWOULDBLOCK
617 && errno != ECONNREFUSED
618
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100619#ifdef EINPROGRESS
620 && errno != EINPROGRESS
621#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100622 )
623 {
624 ch_errorn(channel,
625 "channel_open: Connect failed with errno %d", errno);
626 PERROR(_(e_cannot_connect));
627 sock_close(sd);
628 channel_free(channel);
629 return NULL;
630 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100631 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100632
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100633 /* If we don't block and connect() failed then try using select() to
634 * wait for the connection to be made. */
635 if (waittime >= 0 && ret < 0)
636 {
637 struct timeval tv;
638 fd_set wfds;
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100639#if defined(__APPLE__) && __APPLE__ == 1
640# define PASS_RFDS
Bram Moolenaar98921892016-02-23 17:14:37 +0100641 fd_set rfds;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100642
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100643 FD_ZERO(&rfds);
644 FD_SET(sd, &rfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100645#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100646 FD_ZERO(&wfds);
647 FD_SET(sd, &wfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100648
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100649 tv.tv_sec = waittime / 1000;
650 tv.tv_usec = (waittime % 1000) * 1000;
651#ifndef WIN32
652 gettimeofday(&start_tv, NULL);
653#endif
654 ch_logn(channel,
655 "Waiting for connection (waittime %d msec)...", waittime);
656 ret = select((int)sd + 1,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100657#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100658 &rfds,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100659#else
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100660 NULL,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100661#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100662 &wfds, NULL, &tv);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100663
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100664 if (ret < 0)
665 {
666 SOCK_ERRNO;
667 ch_errorn(channel,
668 "channel_open: Connect failed with errno %d", errno);
669 PERROR(_(e_cannot_connect));
670 sock_close(sd);
671 channel_free(channel);
672 return NULL;
673 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100674#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100675 if (ret == 0 && FD_ISSET(sd, &rfds) && FD_ISSET(sd, &wfds))
676 {
677 /* For OS X, this implies error. See tcp(4). */
678 ch_error(channel, "channel_open: Connect failed");
679 EMSG(_(e_cannot_connect));
680 sock_close(sd);
681 channel_free(channel);
682 return NULL;
683 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100684#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100685#ifdef WIN32
686 /* On Win32 select() is expected to work and wait for up to the
687 * waittime for the socket to be open. */
688 if (!FD_ISSET(sd, &wfds) || ret == 0)
689#else
690 /* See socket(7) for the behavior on Linux-like systems:
691 * After putting the socket in non-blocking mode, connect() will
692 * return EINPROGRESS, select() will not wait (as if writing is
693 * possible), need to use getsockopt() to check if the socket is
694 * actually open. */
695 getsockopt(sd, SOL_SOCKET, SO_ERROR, &so_error, &so_error_len);
696 if (!FD_ISSET(sd, &wfds) || ret == 0 || so_error != 0)
697#endif
698 {
699#ifndef WIN32
700 struct timeval end_tv;
701 long elapsed_msec;
702
703 gettimeofday(&end_tv, NULL);
704 elapsed_msec = (end_tv.tv_sec - start_tv.tv_sec) * 1000
705 + (end_tv.tv_usec - start_tv.tv_usec) / 1000;
706 if (waittime > 1 && elapsed_msec < waittime)
707 {
708 /* The port isn't ready but we also didn't get an error.
709 * This happens when the server didn't open the socket
710 * yet. Wait a bit and try again. */
711 mch_delay(waittime < 50 ? (long)waittime : 50L, TRUE);
712 ui_breakcheck();
713 if (!got_int)
714 {
715 /* reduce the waittime by the elapsed time and the 50
716 * msec delay (or a bit more) */
717 waittime -= elapsed_msec;
718 if (waittime > 50)
719 waittime -= 50;
720 else
721 waittime = 1;
722 continue;
723 }
724 /* we were interrupted, behave as if timed out */
725 }
726#endif
727 /* We timed out. */
728 ch_error(channel, "Connection timed out");
729 sock_close(sd);
730 channel_free(channel);
731 return NULL;
732 }
733
734 ch_log(channel, "Connection made");
735 break;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100736 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100737 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100738
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100739 if (waittime >= 0)
740 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100741#ifdef _WIN32
742 val = 0;
743 ioctlsocket(sd, FIONBIO, &val);
744#else
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100745 (void)fcntl(sd, F_SETFL, 0);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100746#endif
747 }
748
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100749 channel->CH_SOCK_FD = (sock_T)sd;
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100750 channel->ch_nb_close_cb = nb_close_cb;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100751
752#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +0100753 channel_gui_register(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100754#endif
755
Bram Moolenaar77073442016-02-13 23:23:53 +0100756 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100757}
758
759#if defined(CHANNEL_PIPES) || defined(PROTO)
760 void
Bram Moolenaard8070362016-02-15 21:56:54 +0100761channel_set_pipes(channel_T *channel, sock_T in, sock_T out, sock_T err)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100762{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100763 channel->CH_IN_FD = in;
764 channel->CH_OUT_FD = out;
765 channel->CH_ERR_FD = err;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100766}
767#endif
768
769 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100770channel_set_job(channel_T *channel, job_T *job)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100771{
Bram Moolenaar77073442016-02-13 23:23:53 +0100772 channel->ch_job = job;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100773}
774
775/*
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100776 * Set various properties from an "opt" argument.
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100777 */
778 void
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100779channel_set_options(channel_T *channel, jobopt_T *opt)
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100780{
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100781 int part;
782 char_u **cbp;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100783
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100784 if (opt->jo_set & JO_MODE)
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100785 for (part = PART_SOCK; part <= PART_IN; ++part)
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100786 channel->ch_part[part].ch_mode = opt->jo_mode;
787 if (opt->jo_set & JO_IN_MODE)
788 channel->ch_part[PART_IN].ch_mode = opt->jo_in_mode;
789 if (opt->jo_set & JO_OUT_MODE)
790 channel->ch_part[PART_OUT].ch_mode = opt->jo_out_mode;
791 if (opt->jo_set & JO_ERR_MODE)
792 channel->ch_part[PART_ERR].ch_mode = opt->jo_err_mode;
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100793
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100794 if (opt->jo_set & JO_TIMEOUT)
795 for (part = PART_SOCK; part <= PART_IN; ++part)
796 channel->ch_part[part].ch_timeout = opt->jo_timeout;
797 if (opt->jo_set & JO_OUT_TIMEOUT)
798 channel->ch_part[PART_OUT].ch_timeout = opt->jo_out_timeout;
799 if (opt->jo_set & JO_ERR_TIMEOUT)
800 channel->ch_part[PART_ERR].ch_timeout = opt->jo_err_timeout;
801
802 if (opt->jo_set & JO_CALLBACK)
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100803 {
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100804 cbp = &channel->ch_callback;
805 vim_free(*cbp);
806 if (opt->jo_callback != NULL && *opt->jo_callback != NUL)
807 *cbp = vim_strsave(opt->jo_callback);
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100808 else
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100809 *cbp = NULL;
810 }
811 if (opt->jo_set & JO_OUT_CALLBACK)
812 {
813 cbp = &channel->ch_part[PART_OUT].ch_callback;
814 vim_free(*cbp);
815 if (opt->jo_out_cb != NULL && *opt->jo_out_cb != NUL)
816 *cbp = vim_strsave(opt->jo_out_cb);
817 else
818 *cbp = NULL;
819 }
820 if (opt->jo_set & JO_ERR_CALLBACK)
821 {
822 cbp = &channel->ch_part[PART_ERR].ch_callback;
823 vim_free(*cbp);
824 if (opt->jo_err_cb != NULL && *opt->jo_err_cb != NUL)
825 *cbp = vim_strsave(opt->jo_err_cb);
826 else
827 *cbp = NULL;
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100828 }
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100829 if (opt->jo_set & JO_CLOSE_CALLBACK)
830 {
831 cbp = &channel->ch_close_cb;
832 vim_free(*cbp);
833 if (opt->jo_close_cb != NULL && *opt->jo_close_cb != NUL)
834 *cbp = vim_strsave(opt->jo_close_cb);
835 else
836 *cbp = NULL;
837 }
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100838}
839
840/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100841 * Set the callback for "channel"/"part" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100842 */
843 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100844channel_set_req_callback(
845 channel_T *channel,
846 int part,
847 char_u *callback,
848 int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100849{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100850 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100851 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
852
853 if (item != NULL)
854 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100855 item->cq_callback = vim_strsave(callback);
856 item->cq_seq_nr = id;
857 item->cq_prev = head->cq_prev;
858 head->cq_prev = item;
859 item->cq_next = NULL;
860 if (item->cq_prev == NULL)
861 head->cq_next = item;
862 else
863 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100864 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100865}
866
867/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100868 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100869 */
870 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100871invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100872{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100873 typval_T rettv;
874 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100875
Bram Moolenaar77073442016-02-13 23:23:53 +0100876 argv[0].v_type = VAR_CHANNEL;
877 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100878
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100879 call_func(callback, (int)STRLEN(callback),
880 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
Bram Moolenaaree1cffc2016-02-21 19:14:41 +0100881 clear_tv(&rettv);
882
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100883 /* If an echo command was used the cursor needs to be put back where
884 * it belongs. */
885 setcursor();
886 cursor_on();
887 out_flush();
888}
889
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100890/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100891 * Return the first buffer from channel "channel"/"part" and remove it.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100892 * The caller must free it.
893 * Returns NULL if there is nothing.
894 */
895 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100896channel_get(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100897{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100898 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100899 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100900 char_u *p;
901
Bram Moolenaar77073442016-02-13 23:23:53 +0100902 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100903 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100904 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100905 p = node->rq_buffer;
906 head->rq_next = node->rq_next;
907 if (node->rq_next == NULL)
908 head->rq_prev = NULL;
909 else
910 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100911 vim_free(node);
912 return p;
913}
914
915/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100916 * Returns the whole buffer contents concatenated for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100917 */
918 static char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100919channel_get_all(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100920{
921 /* Concatenate everything into one buffer.
922 * TODO: avoid multiple allocations. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100923 while (channel_collapse(channel, part) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100924 ;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100925 return channel_get(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100926}
927
928/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100929 * Collapses the first and second buffer for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100930 * Returns FAIL if that is not possible.
931 */
932 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100933channel_collapse(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100934{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100935 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100936 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100937 char_u *p;
938
Bram Moolenaar77073442016-02-13 23:23:53 +0100939 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100940 return FAIL;
941
Bram Moolenaar77073442016-02-13 23:23:53 +0100942 p = alloc((unsigned)(STRLEN(node->rq_buffer)
943 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100944 if (p == NULL)
945 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100946 STRCPY(p, node->rq_buffer);
947 STRCAT(p, node->rq_next->rq_buffer);
948 vim_free(node->rq_next->rq_buffer);
949 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100950
Bram Moolenaar77073442016-02-13 23:23:53 +0100951 /* dispose of the node and its buffer */
952 head->rq_next = node->rq_next;
953 head->rq_next->rq_prev = NULL;
954 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100955 vim_free(node);
956 return OK;
957}
958
959/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100960 * Store "buf[len]" on "channel"/"part".
961 * Returns OK or FAIL.
962 */
963 static int
964channel_save(channel_T *channel, int part, char_u *buf, int len)
965{
966 readq_T *node;
967 readq_T *head = &channel->ch_part[part].ch_head;
968 char_u *p;
969 int i;
970
971 node = (readq_T *)alloc(sizeof(readq_T));
972 if (node == NULL)
973 return FAIL; /* out of memory */
974 node->rq_buffer = alloc(len + 1);
975 if (node->rq_buffer == NULL)
976 {
977 vim_free(node);
978 return FAIL; /* out of memory */
979 }
980
981 if (channel->ch_part[part].ch_mode == MODE_NL)
982 {
983 /* Drop any CR before a NL. */
984 p = node->rq_buffer;
985 for (i = 0; i < len; ++i)
986 if (buf[i] != CAR || i + 1 >= len || buf[i + 1] != NL)
987 *p++ = buf[i];
988 *p = NUL;
989 }
990 else
991 {
992 mch_memmove(node->rq_buffer, buf, len);
993 node->rq_buffer[len] = NUL;
994 }
995
996 /* append node to the tail of the queue */
997 node->rq_next = NULL;
998 node->rq_prev = head->rq_prev;
999 if (head->rq_prev == NULL)
1000 head->rq_next = node;
1001 else
1002 head->rq_prev->rq_next = node;
1003 head->rq_prev = node;
1004
1005 if (log_fd != NULL)
1006 {
1007 ch_log_lead("RECV ", channel);
1008 fprintf(log_fd, "'");
1009 if (fwrite(buf, len, 1, log_fd) != 1)
1010 return FAIL;
1011 fprintf(log_fd, "'\n");
1012 }
1013 return OK;
1014}
1015
1016/*
1017 * Use the read buffer of "channel"/"part" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001018 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +01001019 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001020 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001021 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001022channel_parse_json(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001023{
1024 js_read_T reader;
1025 typval_T listtv;
1026 jsonq_T *item;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001027 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001028 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001029
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001030 if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001031 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001032
1033 /* TODO: make reader work properly */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001034 /* reader.js_buf = channel_peek(channel, part); */
1035 reader.js_buf = channel_get_all(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001036 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +01001037 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001038 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +01001039 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001040 ret = json_decode(&reader, &listtv,
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001041 channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001042 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001043 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +01001044 /* Only accept the response when it is a list with at least two
1045 * items. */
1046 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001047 {
1048 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001049 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001050 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001051 else
1052 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001053 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
1054 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001055 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001056 else
1057 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001058 item->jq_value = alloc_tv();
1059 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001060 {
1061 vim_free(item);
1062 clear_tv(&listtv);
1063 }
1064 else
1065 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001066 *item->jq_value = listtv;
1067 item->jq_prev = head->jq_prev;
1068 head->jq_prev = item;
1069 item->jq_next = NULL;
1070 if (item->jq_prev == NULL)
1071 head->jq_next = item;
1072 else
1073 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001074 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001075 }
1076 }
1077 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001078
1079 /* Put the unread part back into the channel.
1080 * TODO: insert in front */
1081 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001082 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001083 channel_save(channel, part, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001084 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001085 ret = TRUE;
1086 }
1087 else
1088 ret = FALSE;
1089
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001090 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001091 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001092}
1093
1094/*
Bram Moolenaard46ae142016-02-16 13:33:52 +01001095 * Remove "node" from the queue that it is in. Does not free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +01001096 */
1097 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001098remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001099{
Bram Moolenaar77073442016-02-13 23:23:53 +01001100 if (node->cq_prev == NULL)
1101 head->cq_next = node->cq_next;
1102 else
1103 node->cq_prev->cq_next = node->cq_next;
1104 if (node->cq_next == NULL)
1105 head->cq_prev = node->cq_prev;
1106 else
1107 node->cq_next->cq_prev = node->cq_prev;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001108}
1109
1110/*
1111 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +01001112 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001113 */
1114 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001115remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001116{
Bram Moolenaar77073442016-02-13 23:23:53 +01001117 if (node->jq_prev == NULL)
1118 head->jq_next = node->jq_next;
1119 else
1120 node->jq_prev->jq_next = node->jq_next;
1121 if (node->jq_next == NULL)
1122 head->jq_prev = node->jq_prev;
1123 else
1124 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001125 vim_free(node);
1126}
1127
1128/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001129 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001130 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +01001131 * When "id" is zero or negative jut get the first message. But not the one
1132 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001133 * Return OK when found and return the value in "rettv".
1134 * Return FAIL otherwise.
1135 */
1136 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001137channel_get_json(channel_T *channel, int part, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001138{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001139 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001140 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001141
Bram Moolenaar77073442016-02-13 23:23:53 +01001142 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001143 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001144 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001145 typval_T *tv = &l->lv_first->li_tv;
1146
1147 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001148 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001149 || tv->vval.v_number == 0
1150 || tv->vval.v_number != channel->ch_part[part].ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001151 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001152 *rettv = item->jq_value;
1153 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001154 return OK;
1155 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001156 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001157 }
1158 return FAIL;
1159}
1160
Bram Moolenaarece61b02016-02-20 21:39:05 +01001161#define CH_JSON_MAX_ARGS 4
1162
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001163/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001164 * Execute a command received over "channel"/"part"
Bram Moolenaarece61b02016-02-20 21:39:05 +01001165 * "argv[0]" is the command string.
1166 * "argv[1]" etc. have further arguments, type is VAR_UNKNOWN if missing.
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001167 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001168 static void
Bram Moolenaarece61b02016-02-20 21:39:05 +01001169channel_exe_cmd(channel_T *channel, int part, typval_T *argv)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001170{
Bram Moolenaarece61b02016-02-20 21:39:05 +01001171 char_u *cmd = argv[0].vval.v_string;
1172 char_u *arg;
1173 int options = channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001174
Bram Moolenaarece61b02016-02-20 21:39:05 +01001175 if (argv[1].v_type != VAR_STRING)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001176 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001177 ch_error(channel, "received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001178 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001179 EMSG("E903: received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001180 return;
1181 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001182 arg = argv[1].vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001183 if (arg == NULL)
1184 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001185
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001186 if (STRCMP(cmd, "ex") == 0)
1187 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001188 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001189 }
1190 else if (STRCMP(cmd, "normal") == 0)
1191 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001192 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001193
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001194 ea.arg = arg;
1195 ea.addr_count = 0;
1196 ea.forceit = TRUE; /* no mapping */
1197 ex_normal(&ea);
1198 }
1199 else if (STRCMP(cmd, "redraw") == 0)
1200 {
1201 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001202
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001203 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001204 ex_redraw(&ea);
1205 showruler(FALSE);
1206 setcursor();
1207 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001208#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001209 if (gui.in_use)
1210 {
1211 gui_update_cursor(FALSE, FALSE);
1212 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001213 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001214#endif
1215 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001216 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "call") == 0)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001217 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001218 int is_call = cmd[0] == 'c';
1219 int id_idx = is_call ? 3 : 2;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001220
Bram Moolenaarece61b02016-02-20 21:39:05 +01001221 if (argv[id_idx].v_type != VAR_UNKNOWN
1222 && argv[id_idx].v_type != VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001223 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001224 ch_error(channel, "last argument for expr/call must be a number");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001225 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001226 EMSG("E904: last argument for expr/call must be a number");
1227 }
1228 else if (is_call && argv[2].v_type != VAR_LIST)
1229 {
1230 ch_error(channel, "third argument for call must be a list");
1231 if (p_verbose > 2)
1232 EMSG("E904: third argument for call must be a list");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001233 }
1234 else
1235 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001236 typval_T *tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001237 typval_T res_tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001238 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +01001239 char_u *json = NULL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001240
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001241 /* Don't pollute the display with errors. */
1242 ++emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001243 if (!is_call)
1244 tv = eval_expr(arg, NULL);
1245 else if (func_call(arg, &argv[2], NULL, &res_tv) == OK)
1246 tv = &res_tv;
1247 else
1248 tv = NULL;
1249
1250 if (argv[id_idx].v_type == VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001251 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001252 int id = argv[id_idx].vval.v_number;
1253
Bram Moolenaar55fab432016-02-07 16:53:13 +01001254 if (tv != NULL)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001255 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001256 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001257 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001258 /* If evaluation failed or the result can't be encoded
1259 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001260 vim_free(json);
1261 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001262 err_tv.v_type = VAR_STRING;
1263 err_tv.vval.v_string = (char_u *)"ERROR";
1264 tv = &err_tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001265 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001266 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001267 if (json != NULL)
1268 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001269 channel_send(channel,
1270 part == PART_SOCK ? PART_SOCK : PART_IN,
1271 json, (char *)cmd);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001272 vim_free(json);
1273 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001274 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001275 --emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001276 if (tv == &res_tv)
1277 clear_tv(tv);
1278 else if (tv != &err_tv)
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001279 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001280 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001281 }
1282 else if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001283 {
1284 ch_errors(channel, "Receved unknown command: %s", (char *)cmd);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001285 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001286 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001287}
1288
1289/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001290 * Invoke a callback for "channel"/"part" if needed.
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001291 * Return TRUE when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001292 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001293 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001294may_invoke_callback(channel_T *channel, int part)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001295{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001296 char_u *msg = NULL;
1297 typval_T *listtv = NULL;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001298 typval_T argv[CH_JSON_MAX_ARGS];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001299 int seq_nr = -1;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001300 ch_mode_T ch_mode = channel->ch_part[part].ch_mode;
1301 char_u *callback = NULL;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001302
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001303 if (channel->ch_nb_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001304 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001305 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001306
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001307 if (channel->ch_part[part].ch_callback != NULL)
1308 callback = channel->ch_part[part].ch_callback;
1309 else
1310 callback = channel->ch_callback;
1311
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001312 if (ch_mode == MODE_JSON || ch_mode == MODE_JS)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001313 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001314 listitem_T *item;
1315 int argc = 0;
1316
Bram Moolenaard7ece102016-02-02 23:23:02 +01001317 /* Get any json message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001318 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001319 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001320 /* Parse readahead, return when there is still no message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001321 channel_parse_json(channel, part);
1322 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001323 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001324 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001325
Bram Moolenaarece61b02016-02-20 21:39:05 +01001326 for (item = listtv->vval.v_list->lv_first;
1327 item != NULL && argc < CH_JSON_MAX_ARGS;
1328 item = item->li_next)
1329 argv[argc++] = item->li_tv;
1330 while (argc < CH_JSON_MAX_ARGS)
1331 argv[argc++].v_type = VAR_UNKNOWN;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001332
Bram Moolenaarece61b02016-02-20 21:39:05 +01001333 if (argv[0].v_type == VAR_STRING)
1334 {
1335 char_u *cmd = argv[0].vval.v_string;
1336
1337 /* ["cmd", arg] or ["cmd", arg, arg] or ["cmd", arg, arg, arg] */
Bram Moolenaar77073442016-02-13 23:23:53 +01001338 ch_logs(channel, "Executing %s command", (char *)cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001339 channel_exe_cmd(channel, part, argv);
Bram Moolenaar77073442016-02-13 23:23:53 +01001340 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001341 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001342 }
1343
Bram Moolenaarece61b02016-02-20 21:39:05 +01001344 if (argv[0].v_type != VAR_NUMBER)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001345 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001346 ch_error(channel,
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001347 "Dropping message with invalid sequence number type");
Bram Moolenaar77073442016-02-13 23:23:53 +01001348 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001349 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001350 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001351 seq_nr = argv[0].vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001352 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001353 else if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001354 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001355 /* nothing to read on RAW or NL channel */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001356 return FALSE;
1357 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001358 else
1359 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001360 /* If there is no callback drop the message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001361 if (callback == NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001362 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001363 while ((msg = channel_get(channel, part)) != NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001364 vim_free(msg);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001365 return FALSE;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001366 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001367
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001368 if (ch_mode == MODE_NL)
1369 {
1370 char_u *nl;
1371 char_u *buf;
1372
1373 /* See if we have a message ending in NL in the first buffer. If
1374 * not try to concatenate the first and the second buffer. */
1375 while (TRUE)
1376 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001377 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001378 nl = vim_strchr(buf, NL);
1379 if (nl != NULL)
1380 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001381 if (channel_collapse(channel, part) == FAIL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001382 return FALSE; /* incomplete message */
1383 }
1384 if (nl[1] == NUL)
1385 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001386 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001387 else
1388 {
1389 /* Copy the message into allocated memory and remove it from
1390 * the buffer. */
1391 msg = vim_strnsave(buf, (int)(nl - buf));
1392 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1393 }
1394 }
1395 else
1396 /* For a raw channel we don't know where the message ends, just
1397 * get everything we have. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001398 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001399
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001400 argv[1].v_type = VAR_STRING;
1401 argv[1].vval.v_string = msg;
1402 }
1403
Bram Moolenaara07fec92016-02-05 21:04:08 +01001404 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001405 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001406 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001407 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001408 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001409
1410 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001411 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001412 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001413 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001414 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001415 ch_log(channel, "Invoking one-time callback");
Bram Moolenaard46ae142016-02-16 13:33:52 +01001416 /* Remove the item from the list first, if the callback
1417 * invokes ch_close() the list will be cleared. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001418 remove_cb_node(head, item);
Bram Moolenaard46ae142016-02-16 13:33:52 +01001419 invoke_callback(channel, item->cq_callback, argv);
1420 vim_free(item->cq_callback);
1421 vim_free(item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001422 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001423 break;
1424 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001425 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001426 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001427 if (!done)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001428 ch_log(channel, "Dropping message without callback");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001429 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001430 else if (callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001431 {
1432 /* invoke the channel callback */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001433 ch_log(channel, "Invoking channel callback");
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001434 invoke_callback(channel, callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001435 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001436 else
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001437 ch_log(channel, "Dropping message");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001438
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001439 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001440 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001441 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001442
1443 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001444}
1445
1446/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001447 * Return TRUE when channel "channel" is open for writing to.
1448 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001449 */
1450 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001451channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001452{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001453 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001454#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001455 || channel->CH_IN_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001456#endif
1457 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001458}
1459
1460/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001461 * Return TRUE when channel "channel" is open for reading or writing.
1462 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001463 */
1464 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001465channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001466{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001467 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001468#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001469 || channel->CH_IN_FD != INVALID_FD
1470 || channel->CH_OUT_FD != INVALID_FD
1471 || channel->CH_ERR_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001472#endif
1473 );
1474}
1475
1476/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001477 * Return a string indicating the status of the channel.
1478 */
1479 char *
1480channel_status(channel_T *channel)
1481{
1482 if (channel == NULL)
1483 return "fail";
1484 if (channel_is_open(channel))
1485 return "open";
1486 return "closed";
1487}
1488
1489/*
1490 * Close channel "channel".
Bram Moolenaarc8dcbb12016-02-25 23:10:17 +01001491 * Trigger the close callback if "invoke_close_cb" is TRUE.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001492 */
1493 void
Bram Moolenaar8b374212016-02-24 20:43:06 +01001494channel_close(channel_T *channel, int invoke_close_cb)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001495{
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001496 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001497
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001498#ifdef FEAT_GUI
1499 channel_gui_unregister(channel);
1500#endif
1501
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001502 if (channel->CH_SOCK_FD != INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001503 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001504 sock_close(channel->CH_SOCK_FD);
1505 channel->CH_SOCK_FD = INVALID_FD;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001506 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001507#if defined(CHANNEL_PIPES)
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001508 if (channel->CH_IN_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001509 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001510 fd_close(channel->CH_IN_FD);
1511 channel->CH_IN_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001512 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001513 if (channel->CH_OUT_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001514 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001515 fd_close(channel->CH_OUT_FD);
1516 channel->CH_OUT_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001517 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001518 if (channel->CH_ERR_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001519 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001520 fd_close(channel->CH_ERR_FD);
1521 channel->CH_ERR_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001522 }
1523#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001524
Bram Moolenaar8b374212016-02-24 20:43:06 +01001525 if (invoke_close_cb && channel->ch_close_cb != NULL)
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001526 {
1527 typval_T argv[1];
1528 typval_T rettv;
1529 int dummy;
1530
1531 /* invoke the close callback; increment the refcount to avoid it
1532 * being freed halfway */
1533 argv[0].v_type = VAR_CHANNEL;
1534 argv[0].vval.v_channel = channel;
1535 ++channel->ch_refcount;
1536 call_func(channel->ch_close_cb, (int)STRLEN(channel->ch_close_cb),
1537 &rettv, 1, argv, 0L, 0L, &dummy, TRUE, NULL);
1538 clear_tv(&rettv);
1539 --channel->ch_refcount;
1540
1541 /* the callback is only called once */
1542 vim_free(channel->ch_close_cb);
1543 channel->ch_close_cb = NULL;
1544 }
1545
1546 channel->ch_nb_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001547 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001548}
1549
Bram Moolenaard04a0202016-01-26 23:30:18 +01001550/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001551 * Return the first buffer from "channel"/"part" without removing it.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001552 * Returns NULL if there is nothing.
1553 */
1554 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001555channel_peek(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001556{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001557 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001558
Bram Moolenaar77073442016-02-13 23:23:53 +01001559 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001560 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001561 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001562}
1563
1564/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001565 * Clear the read buffer on "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001566 */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001567 static void
1568channel_clear_one(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001569{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001570 jsonq_T *json_head = &channel->ch_part[part].ch_json_head;
1571 cbq_T *cb_head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001572
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001573 while (channel_peek(channel, part) != NULL)
1574 vim_free(channel_get(channel, part));
Bram Moolenaar77073442016-02-13 23:23:53 +01001575
1576 while (cb_head->cq_next != NULL)
Bram Moolenaard46ae142016-02-16 13:33:52 +01001577 {
1578 cbq_T *node = cb_head->cq_next;
1579
1580 remove_cb_node(cb_head, node);
1581 vim_free(node->cq_callback);
1582 vim_free(node);
1583 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001584
1585 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001586 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001587 free_tv(json_head->jq_next->jq_value);
1588 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001589 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001590
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001591 vim_free(channel->ch_part[part].ch_callback);
1592 channel->ch_part[part].ch_callback = NULL;
1593}
1594
1595/*
1596 * Clear all the read buffers on "channel".
1597 */
1598 void
1599channel_clear(channel_T *channel)
1600{
1601 channel_clear_one(channel, PART_SOCK);
1602#ifdef CHANNEL_PIPES
1603 channel_clear_one(channel, PART_OUT);
1604 channel_clear_one(channel, PART_ERR);
1605#endif
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001606 vim_free(channel->ch_callback);
1607 channel->ch_callback = NULL;
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001608 vim_free(channel->ch_close_cb);
1609 channel->ch_close_cb = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001610}
1611
Bram Moolenaar77073442016-02-13 23:23:53 +01001612#if defined(EXITFREE) || defined(PROTO)
1613 void
1614channel_free_all(void)
1615{
1616 channel_T *channel;
1617
1618 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1619 channel_clear(channel);
1620}
1621#endif
1622
1623
Bram Moolenaard04a0202016-01-26 23:30:18 +01001624/* Sent when the channel is found closed when reading. */
Bram Moolenaareed284a2016-02-22 23:13:33 +01001625#define DETACH_MSG_RAW "DETACH\n"
1626#define DETACH_MSG_JSON "\"DETACH\"\n"
Bram Moolenaard04a0202016-01-26 23:30:18 +01001627
1628/* Buffer size for reading incoming messages. */
1629#define MAXMSGSIZE 4096
1630
1631/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001632 * Check for reading from "fd" with "timeout" msec.
1633 * Return FAIL when there is nothing to read.
1634 */
1635 static int
Bram Moolenaard8070362016-02-15 21:56:54 +01001636channel_wait(channel_T *channel, sock_T fd, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001637{
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001638 if (timeout > 0)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001639 ch_logn(channel, "Waiting for up to %d msec", timeout);
Bram Moolenaard8070362016-02-15 21:56:54 +01001640
Bram Moolenaard8070362016-02-15 21:56:54 +01001641# ifdef WIN32
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001642 if (fd != channel->CH_SOCK_FD)
Bram Moolenaard8070362016-02-15 21:56:54 +01001643 {
1644 DWORD nread;
1645 int diff;
1646 DWORD deadline = GetTickCount() + timeout;
1647
1648 /* reading from a pipe, not a socket */
1649 while (TRUE)
1650 {
Bram Moolenaare74e8e72016-02-16 22:01:30 +01001651 if (PeekNamedPipe((HANDLE)fd, NULL, 0, NULL, &nread, NULL)
1652 && nread > 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001653 return OK;
1654 diff = deadline - GetTickCount();
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001655 if (diff <= 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001656 break;
1657 /* Wait for 5 msec.
1658 * TODO: increase the sleep time when looping more often */
1659 Sleep(5);
1660 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001661 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001662 else
Bram Moolenaard8070362016-02-15 21:56:54 +01001663#endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001664 {
Bram Moolenaar9186a272016-02-23 19:34:01 +01001665#if defined(HAVE_SELECT)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001666 struct timeval tval;
1667 fd_set rfds;
1668 int ret;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001669
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001670 FD_ZERO(&rfds);
1671 FD_SET((int)fd, &rfds);
1672 tval.tv_sec = timeout / 1000;
1673 tval.tv_usec = (timeout % 1000) * 1000;
1674 for (;;)
1675 {
1676 ret = select((int)fd + 1, &rfds, NULL, NULL, &tval);
Bram Moolenaar9186a272016-02-23 19:34:01 +01001677# ifdef EINTR
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001678 SOCK_ERRNO;
1679 if (ret == -1 && errno == EINTR)
1680 continue;
Bram Moolenaar9186a272016-02-23 19:34:01 +01001681# endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001682 if (ret > 0)
1683 return OK;
1684 break;
1685 }
Bram Moolenaar9186a272016-02-23 19:34:01 +01001686#else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001687 struct pollfd fds;
1688
1689 fds.fd = fd;
1690 fds.events = POLLIN;
1691 if (poll(&fds, 1, timeout) > 0)
1692 return OK;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001693#endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001694 }
1695 ch_log(channel, "Nothing to read");
1696 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001697}
1698
1699/*
1700 * Return a unique ID to be used in a message.
1701 */
1702 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001703channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001704{
1705 static int next_id = 1;
1706
1707 return next_id++;
1708}
1709
1710/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001711 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001712 * "part" is PART_SOCK, PART_OUT or PART_ERR.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001713 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001714 */
1715 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001716channel_read(channel_T *channel, int part, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001717{
1718 static char_u *buf = NULL;
1719 int len = 0;
1720 int readlen = 0;
Bram Moolenaard8070362016-02-15 21:56:54 +01001721 sock_T fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001722 int use_socket = FALSE;
Bram Moolenaareed284a2016-02-22 23:13:33 +01001723 char *msg;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001724
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001725 fd = channel->ch_part[part].ch_fd;
1726 if (fd == INVALID_FD)
1727 {
1728 ch_error(channel, "channel_read() called while socket is closed");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001729 return;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001730 }
1731 use_socket = fd == channel->CH_SOCK_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001732
1733 /* Allocate a buffer to read into. */
1734 if (buf == NULL)
1735 {
1736 buf = alloc(MAXMSGSIZE);
1737 if (buf == NULL)
1738 return; /* out of memory! */
1739 }
1740
1741 /* Keep on reading for as long as there is something to read.
1742 * Use select() or poll() to avoid blocking on a message that is exactly
1743 * MAXMSGSIZE long. */
1744 for (;;)
1745 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001746 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001747 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001748 if (use_socket)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001749 len = sock_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001750 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001751 len = fd_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001752 if (len <= 0)
1753 break; /* error or nothing more to read */
1754
1755 /* Store the read message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001756 channel_save(channel, part, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001757 readlen += len;
1758 if (len < MAXMSGSIZE)
1759 break; /* did read everything that's available */
1760 }
1761
Bram Moolenaarbd73ae12016-02-22 22:19:22 +01001762 /* Reading a disconnection (readlen == 0), or an error.
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001763 * TODO: call error callback. */
Bram Moolenaarbd73ae12016-02-22 22:19:22 +01001764 if (readlen <= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001765 {
1766 /* Queue a "DETACH" netbeans message in the command queue in order to
1767 * terminate the netbeans session later. Do not end the session here
1768 * directly as we may be running in the context of a call to
1769 * netbeans_parse_messages():
1770 * netbeans_parse_messages
1771 * -> autocmd triggered while processing the netbeans cmd
1772 * -> ui_breakcheck
1773 * -> gui event loop or select loop
1774 * -> channel_read()
1775 */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001776 ch_errors(channel, "%s(): Cannot read", func);
Bram Moolenaareed284a2016-02-22 23:13:33 +01001777 msg = channel->ch_part[part].ch_mode == MODE_RAW
1778 || channel->ch_part[part].ch_mode == MODE_NL
1779 ? DETACH_MSG_RAW : DETACH_MSG_JSON;
1780 channel_save(channel, part, (char_u *)msg, (int)STRLEN(msg));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001781
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001782 /* TODO: When reading from stdout is not possible, should we try to
1783 * keep stdin and stderr open? Probably not, assume the other side
1784 * has died. */
Bram Moolenaar8b374212016-02-24 20:43:06 +01001785 channel_close(channel, TRUE);
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001786 if (channel->ch_nb_close_cb != NULL)
1787 (*channel->ch_nb_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001788
1789 if (len < 0)
1790 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001791 ch_error(channel, "channel_read(): cannot read from channel");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001792 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001793 }
1794 }
1795
1796#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001797 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001798 if (CH_HAS_GUI && gtk_main_level() > 0)
1799 gtk_main_quit();
1800#endif
1801}
1802
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001803/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001804 * Read from RAW or NL "channel"/"part". Blocks until there is something to
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001805 * read or the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001806 * Returns what was read in allocated memory.
1807 * Returns NULL in case of error or timeout.
1808 */
1809 char_u *
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001810channel_read_block(channel_T *channel, int part, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001811{
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001812 char_u *buf;
1813 char_u *msg;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001814 ch_mode_T mode = channel->ch_part[part].ch_mode;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001815 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001816 char_u *nl;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001817
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001818 ch_logsn(channel, "Blocking %s read, timeout: %d msec",
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001819 mode == MODE_RAW ? "RAW" : "NL", timeout);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001820
1821 while (TRUE)
1822 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001823 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001824 if (buf != NULL && (mode == MODE_RAW
1825 || (mode == MODE_NL && vim_strchr(buf, NL) != NULL)))
1826 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001827 if (buf != NULL && channel_collapse(channel, part) == OK)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001828 continue;
1829
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001830 /* Wait for up to the channel timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001831 if (fd == INVALID_FD
1832 || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001833 return NULL;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001834 channel_read(channel, part, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001835 }
1836
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001837 if (mode == MODE_RAW)
1838 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001839 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001840 }
1841 else
1842 {
1843 nl = vim_strchr(buf, NL);
1844 if (nl[1] == NUL)
1845 {
1846 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001847 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001848 *nl = NUL;
1849 }
1850 else
1851 {
1852 /* Copy the message into allocated memory and remove it from the
1853 * buffer. */
1854 msg = vim_strnsave(buf, (int)(nl - buf));
1855 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1856 }
1857 }
1858 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001859 ch_logn(channel, "Returning %d bytes", (int)STRLEN(msg));
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001860 return msg;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001861}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001862
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001863/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001864 * Read one JSON message with ID "id" from "channel"/"part" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001865 * result in "rettv".
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001866 * When "id" is -1 accept any message;
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001867 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001868 */
1869 int
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001870channel_read_json_block(
1871 channel_T *channel,
1872 int part,
1873 int timeout,
1874 int id,
1875 typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001876{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001877 int more;
Bram Moolenaard8070362016-02-15 21:56:54 +01001878 sock_T fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001879
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001880 ch_log(channel, "Reading JSON");
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001881 if (id != -1)
1882 channel->ch_part[part].ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001883 for (;;)
1884 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001885 more = channel_parse_json(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001886
1887 /* search for messsage "id" */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001888 if (channel_get_json(channel, part, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001889 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001890 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001891 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001892 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001893
Bram Moolenaard7ece102016-02-02 23:23:02 +01001894 if (!more)
1895 {
1896 /* Handle any other messages in the queue. If done some more
1897 * messages may have arrived. */
1898 if (channel_parse_messages())
1899 continue;
1900
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001901 /* Wait for up to the timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001902 fd = channel->ch_part[part].ch_fd;
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001903 if (fd == INVALID_FD || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001904 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001905 channel_read(channel, part, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001906 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001907 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001908 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001909 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001910}
1911
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001912# if defined(WIN32) || defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK) \
1913 || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001914/*
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001915 * Lookup the channel from the socket. Set "partp" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001916 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001917 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001918 channel_T *
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001919channel_fd2channel(sock_T fd, int *partp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001920{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001921 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001922 int part;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001923
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001924 if (fd != INVALID_FD)
Bram Moolenaar77073442016-02-13 23:23:53 +01001925 for (channel = first_channel; channel != NULL;
1926 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001927 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001928# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001929 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001930# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001931 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001932# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001933 if (channel->ch_part[part].ch_fd == fd)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001934 {
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001935 *partp = part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001936 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001937 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001938 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001939 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001940}
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001941# endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001942
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001943# if defined(WIN32) || defined(PROTO)
1944/*
1945 * Check the channels for anything that is ready to be read.
1946 * The data is put in the read queue.
1947 */
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001948 void
1949channel_handle_events(void)
1950{
1951 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001952 int part;
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001953 sock_T fd;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001954
1955 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1956 {
Bram Moolenaar9186a272016-02-23 19:34:01 +01001957# ifdef CHANNEL_PIPES
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001958 /* check the socket and pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001959 for (part = PART_SOCK; part <= PART_ERR; ++part)
Bram Moolenaar9186a272016-02-23 19:34:01 +01001960# else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001961 /* only check the socket */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001962 part = PART_SOCK;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001963# endif
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001964 {
1965 fd = channel->ch_part[part].ch_fd;
1966 if (fd != INVALID_FD && channel_wait(channel, fd, 0) == OK)
1967 channel_read(channel, part, "channel_handle_events");
1968 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001969 }
1970}
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001971# endif
1972
Bram Moolenaard04a0202016-01-26 23:30:18 +01001973/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001974 * Write "buf" (NUL terminated string) to "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001975 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001976 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001977 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001978 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001979channel_send(channel_T *channel, int part, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001980{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001981 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001982 int res;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001983 sock_T fd;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001984
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001985 fd = channel->ch_part[part].ch_fd;
1986 if (fd == INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001987 {
1988 if (!channel->ch_error && fun != NULL)
1989 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001990 ch_errors(channel, "%s(): write while not connected", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001991 EMSG2("E630: %s(): write while not connected", fun);
1992 }
1993 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001994 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001995 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001996
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001997 if (log_fd != NULL)
1998 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001999 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002000 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002001 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002002 fprintf(log_fd, "'\n");
2003 fflush(log_fd);
2004 }
2005
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002006 if (part == PART_SOCK)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002007 res = sock_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002008 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002009 res = fd_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002010 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01002011 {
2012 if (!channel->ch_error && fun != NULL)
2013 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01002014 ch_errors(channel, "%s(): write failed", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01002015 EMSG2("E631: %s(): write failed", fun);
2016 }
2017 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01002018 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01002019 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01002020
2021 channel->ch_error = FALSE;
2022 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01002023}
2024
2025# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002026/*
2027 * Add open channels to the poll struct.
2028 * Return the adjusted struct index.
2029 * The type of "fds" is hidden to avoid problems with the function proto.
2030 */
2031 int
2032channel_poll_setup(int nfd_in, void *fds_in)
2033{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002034 int nfd = nfd_in;
2035 channel_T *channel;
2036 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002037 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002038
Bram Moolenaar77073442016-02-13 23:23:53 +01002039 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002040 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002041# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002042 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002043# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002044 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002045# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002046 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002047 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002048 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002049 channel->ch_part[part].ch_poll_idx = nfd;
2050 fds[nfd].fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002051 fds[nfd].events = POLLIN;
2052 nfd++;
2053 }
2054 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002055 channel->ch_part[part].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002056 }
2057 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002058
2059 return nfd;
2060}
2061
2062/*
2063 * The type of "fds" is hidden to avoid problems with the function proto.
2064 */
2065 int
2066channel_poll_check(int ret_in, void *fds_in)
2067{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002068 int ret = ret_in;
2069 channel_T *channel;
2070 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002071 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002072
Bram Moolenaar77073442016-02-13 23:23:53 +01002073 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002074 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002075# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002076 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002077# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002078 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002079# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002080 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002081 int idx = channel->ch_part[part].ch_poll_idx;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002082
2083 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
2084 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002085 channel_read(channel, part, "channel_poll_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002086 --ret;
2087 }
2088 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002089 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002090
2091 return ret;
2092}
Bram Moolenaard04a0202016-01-26 23:30:18 +01002093# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002094
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002095# if (!defined(WIN32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002096/*
2097 * The type of "rfds" is hidden to avoid problems with the function proto.
2098 */
2099 int
2100channel_select_setup(int maxfd_in, void *rfds_in)
2101{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002102 int maxfd = maxfd_in;
2103 channel_T *channel;
2104 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002105 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002106
Bram Moolenaar77073442016-02-13 23:23:53 +01002107 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002108 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002109# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002110 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002111# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002112 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002113# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002114 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002115 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002116
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002117 if (fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002118 {
Bram Moolenaard8070362016-02-15 21:56:54 +01002119 FD_SET((int)fd, rfds);
2120 if (maxfd < (int)fd)
2121 maxfd = (int)fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002122 }
2123 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002124 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002125
2126 return maxfd;
2127}
2128
2129/*
2130 * The type of "rfds" is hidden to avoid problems with the function proto.
2131 */
2132 int
2133channel_select_check(int ret_in, void *rfds_in)
2134{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002135 int ret = ret_in;
2136 channel_T *channel;
2137 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002138 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002139
Bram Moolenaar77073442016-02-13 23:23:53 +01002140 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002141 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002142# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002143 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002144# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002145 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002146# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002147 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002148 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002149
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002150 if (ret > 0 && fd != INVALID_FD && FD_ISSET(fd, rfds))
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002151 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002152 channel_read(channel, part, "channel_select_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002153 --ret;
2154 }
2155 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002156 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002157
2158 return ret;
2159}
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002160# endif /* !WIN32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002161
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002162/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01002163 * Execute queued up commands.
2164 * Invoked from the main loop when it's safe to execute received commands.
2165 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002166 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01002167 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002168channel_parse_messages(void)
2169{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002170 channel_T *channel = first_channel;
2171 int ret = FALSE;
2172 int r;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002173 int part = PART_SOCK;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002174
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002175 while (channel != NULL)
2176 {
Bram Moolenaarc8dcbb12016-02-25 23:10:17 +01002177 if (channel->ch_refcount == 0 && !channel_has_callback(channel))
2178 {
2179 /* channel is no longer useful, free it */
2180 channel_free(channel);
2181 channel = first_channel;
2182 part = PART_SOCK;
2183 continue;
2184 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002185 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaard7ece102016-02-02 23:23:02 +01002186 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002187 /* Increase the refcount, in case the handler causes the channel
2188 * to be unreferenced or closed. */
2189 ++channel->ch_refcount;
2190 r = may_invoke_callback(channel, part);
2191 if (r == OK)
2192 ret = TRUE;
2193 if (channel_unref(channel) || r == OK)
2194 {
2195 /* channel was freed or something was done, start over */
2196 channel = first_channel;
2197 part = PART_SOCK;
2198 continue;
2199 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002200 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002201#ifdef CHANNEL_PIPES
2202 if (part < PART_ERR)
2203 ++part;
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002204 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002205#endif
2206 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002207 channel = channel->ch_next;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002208 part = PART_SOCK;
2209 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002210 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002211 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002212}
2213
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01002214/*
2215 * Mark references to lists used in channels.
2216 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002217 int
2218set_ref_in_channel(int copyID)
2219{
Bram Moolenaar77073442016-02-13 23:23:53 +01002220 int abort = FALSE;
2221 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002222 int part;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002223
Bram Moolenaar77073442016-02-13 23:23:53 +01002224 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002225 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002226#ifdef CHANNEL_PIPES
2227 for (part = PART_SOCK; part < PART_IN; ++part)
2228#else
2229 part = PART_SOCK;
2230#endif
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002231 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002232 jsonq_T *head = &channel->ch_part[part].ch_json_head;
2233 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002234
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002235 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002236 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002237 list_T *l = item->jq_value->vval.v_list;
2238
2239 if (l->lv_copyID != copyID)
2240 {
2241 l->lv_copyID = copyID;
2242 abort = abort || set_ref_in_list(l, copyID, NULL);
2243 }
2244 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002245 }
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002246 }
2247 }
2248 return abort;
2249}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002250
2251/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002252 * Return the "part" to write to for "channel".
2253 */
2254 int
2255channel_part_send(channel_T *channel)
2256{
2257#ifdef CHANNEL_PIPES
2258 if (channel->CH_SOCK_FD == INVALID_FD)
2259 return PART_IN;
2260#endif
2261 return PART_SOCK;
2262}
2263
2264/*
2265 * Return the default "part" to read from for "channel".
2266 */
2267 int
2268channel_part_read(channel_T *channel)
2269{
2270#ifdef CHANNEL_PIPES
2271 if (channel->CH_SOCK_FD == INVALID_FD)
2272 return PART_OUT;
2273#endif
2274 return PART_SOCK;
2275}
2276
2277/*
2278 * Return the mode of "channel"/"part"
Bram Moolenaar77073442016-02-13 23:23:53 +01002279 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002280 */
2281 ch_mode_T
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002282channel_get_mode(channel_T *channel, int part)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002283{
Bram Moolenaar77073442016-02-13 23:23:53 +01002284 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002285 return MODE_JSON;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002286 return channel->ch_part[part].ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002287}
2288
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01002289/*
2290 * Return the timeout of "channel"/"part"
2291 */
2292 int
2293channel_get_timeout(channel_T *channel, int part)
2294{
2295 return channel->ch_part[part].ch_timeout;
2296}
2297
Bram Moolenaare0874f82016-01-24 20:36:41 +01002298#endif /* FEAT_CHANNEL */