blob: c59758d85304e84aeb173685fb9ebcf716bf7846 [file] [log] [blame]
Bram Moolenaare0874f82016-01-24 20:36:41 +01001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 */
8
9/*
10 * Implements communication through a socket or any file handle.
11 */
12
13#include "vim.h"
14
15#if defined(FEAT_CHANNEL) || defined(PROTO)
16
Bram Moolenaard04a0202016-01-26 23:30:18 +010017/* TRUE when netbeans is running with a GUI. */
18#ifdef FEAT_GUI
19# define CH_HAS_GUI (gui.in_use || gui.starting)
20#endif
21
22/* Note: when making changes here also adjust configure.in. */
23#ifdef WIN32
24/* WinSock API is separated from C API, thus we can't use read(), write(),
25 * errno... */
26# define SOCK_ERRNO errno = WSAGetLastError()
27# undef ECONNREFUSED
28# define ECONNREFUSED WSAECONNREFUSED
Bram Moolenaar4d919d72016-02-05 22:36:41 +010029# undef EWOULDBLOCK
30# define EWOULDBLOCK WSAEWOULDBLOCK
Bram Moolenaard04a0202016-01-26 23:30:18 +010031# ifdef EINTR
32# undef EINTR
33# endif
34# define EINTR WSAEINTR
Bram Moolenaard8070362016-02-15 21:56:54 +010035# define sock_write(sd, buf, len) send((SOCKET)sd, buf, len, 0)
36# define sock_read(sd, buf, len) recv((SOCKET)sd, buf, len, 0)
37# define sock_close(sd) closesocket((SOCKET)sd)
Bram Moolenaard04a0202016-01-26 23:30:18 +010038#else
39# include <netdb.h>
40# include <netinet/in.h>
41
42# include <sys/socket.h>
43# ifdef HAVE_LIBGEN_H
44# include <libgen.h>
45# endif
46# define SOCK_ERRNO
47# define sock_write(sd, buf, len) write(sd, buf, len)
48# define sock_read(sd, buf, len) read(sd, buf, len)
49# define sock_close(sd) close(sd)
Bram Moolenaar0943a092016-02-16 13:11:17 +010050# define fd_read(fd, buf, len) read(fd, buf, len)
Bram Moolenaard8070362016-02-15 21:56:54 +010051# define fd_write(sd, buf, len) write(sd, buf, len)
52# define fd_close(sd) close(sd)
Bram Moolenaard04a0202016-01-26 23:30:18 +010053#endif
54
55#ifdef FEAT_GUI_W32
56extern HWND s_hwnd; /* Gvim's Window handle */
57#endif
58
Bram Moolenaard8070362016-02-15 21:56:54 +010059#ifdef WIN32
60 static int
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +010061fd_read(sock_T fd, char *buf, size_t len)
Bram Moolenaard8070362016-02-15 21:56:54 +010062{
63 HANDLE h = (HANDLE)fd;
64 DWORD nread;
65
66 if (!ReadFile(h, buf, (DWORD)len, &nread, NULL))
67 return -1;
68 return (int)nread;
69}
70
71 static int
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +010072fd_write(sock_T fd, char *buf, size_t len)
Bram Moolenaard8070362016-02-15 21:56:54 +010073{
74 HANDLE h = (HANDLE)fd;
75 DWORD nwrite;
76
77 if (!WriteFile(h, buf, (DWORD)len, &nwrite, NULL))
78 return -1;
79 return (int)nwrite;
80}
81
82 static void
83fd_close(sock_T fd)
84{
85 HANDLE h = (HANDLE)fd;
86
87 CloseHandle(h);
88}
89#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +010090
Bram Moolenaar6463ca22016-02-13 17:04:46 +010091/* Log file opened with ch_logfile(). */
92static FILE *log_fd = NULL;
Bram Moolenaar81661fb2016-02-18 22:23:34 +010093#ifdef FEAT_RELTIME
94static proftime_T log_start;
95#endif
Bram Moolenaar6463ca22016-02-13 17:04:46 +010096
97 void
98ch_logfile(FILE *file)
99{
100 if (log_fd != NULL)
101 fclose(log_fd);
102 log_fd = file;
103 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100104 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100105 fprintf(log_fd, "==== start log session ====\n");
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100106#ifdef FEAT_RELTIME
107 profile_start(&log_start);
108#endif
109 }
110}
111
112 int
113ch_log_active()
114{
115 return log_fd != NULL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100116}
117
118 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100119ch_log_lead(char *what, channel_T *ch)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100120{
121 if (log_fd != NULL)
122 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100123#ifdef FEAT_RELTIME
124 proftime_T log_now;
125
126 profile_start(&log_now);
127 profile_sub(&log_now, &log_start);
128 fprintf(log_fd, "%s ", profile_msg(&log_now));
129#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100130 if (ch != NULL)
131 fprintf(log_fd, "%son %d: ", what, ch->ch_id);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100132 else
133 fprintf(log_fd, "%s: ", what);
134 }
135}
136
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100137 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100138ch_log(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100139{
140 if (log_fd != NULL)
141 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100142 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100143 fputs(msg, log_fd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100144 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100145 fflush(log_fd);
146 }
147}
148
149 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100150ch_logn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100151{
152 if (log_fd != NULL)
153 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100154 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100155 fprintf(log_fd, msg, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100156 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100157 fflush(log_fd);
158 }
159}
160
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100161 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100162ch_logs(channel_T *ch, char *msg, char *name)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100163{
164 if (log_fd != NULL)
165 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100166 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100167 fprintf(log_fd, msg, name);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100168 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100169 fflush(log_fd);
170 }
171}
172
173 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100174ch_logsn(channel_T *ch, char *msg, char *name, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100175{
176 if (log_fd != NULL)
177 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100178 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100179 fprintf(log_fd, msg, name, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100180 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100181 fflush(log_fd);
182 }
183}
184
185 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100186ch_error(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100187{
188 if (log_fd != NULL)
189 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100190 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100191 fputs(msg, log_fd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100192 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100193 fflush(log_fd);
194 }
195}
196
197 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100198ch_errorn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100199{
200 if (log_fd != NULL)
201 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100202 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100203 fprintf(log_fd, msg, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100204 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100205 fflush(log_fd);
206 }
207}
208
209 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100210ch_errors(channel_T *ch, char *msg, char *arg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100211{
212 if (log_fd != NULL)
213 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100214 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100215 fprintf(log_fd, msg, arg);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100216 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100217 fflush(log_fd);
218 }
219}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100220
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100221#ifdef _WIN32
222# undef PERROR
223# define PERROR(msg) (void)emsg3((char_u *)"%s: %s", \
224 (char_u *)msg, (char_u *)strerror_win32(errno))
225
226 static char *
227strerror_win32(int eno)
228{
229 static LPVOID msgbuf = NULL;
230 char_u *ptr;
231
232 if (msgbuf)
233 LocalFree(msgbuf);
234 FormatMessage(
235 FORMAT_MESSAGE_ALLOCATE_BUFFER |
236 FORMAT_MESSAGE_FROM_SYSTEM |
237 FORMAT_MESSAGE_IGNORE_INSERTS,
238 NULL,
239 eno,
240 MAKELANGID(LANG_ENGLISH, SUBLANG_DEFAULT),
241 (LPTSTR) &msgbuf,
242 0,
243 NULL);
244 /* chomp \r or \n */
245 for (ptr = (char_u *)msgbuf; *ptr; ptr++)
246 switch (*ptr)
247 {
248 case '\r':
249 STRMOVE(ptr, ptr + 1);
250 ptr--;
251 break;
252 case '\n':
253 if (*(ptr + 1) == '\0')
254 *ptr = '\0';
255 else
256 *ptr = ' ';
257 break;
258 }
259 return msgbuf;
260}
261#endif
262
Bram Moolenaar77073442016-02-13 23:23:53 +0100263/*
264 * The list of all allocated channels.
265 */
266static channel_T *first_channel = NULL;
267static int next_ch_id = 0;
268
269/*
270 * Allocate a new channel. The refcount is set to 1.
271 * The channel isn't actually used until it is opened.
272 * Returns NULL if out of memory.
273 */
274 channel_T *
275add_channel(void)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100276{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100277 int part;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100278 channel_T *channel = (channel_T *)alloc_clear((int)sizeof(channel_T));
Bram Moolenaare0874f82016-01-24 20:36:41 +0100279
Bram Moolenaar77073442016-02-13 23:23:53 +0100280 if (channel == NULL)
281 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100282
Bram Moolenaar77073442016-02-13 23:23:53 +0100283 channel->ch_id = next_ch_id++;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100284 ch_log(channel, "Created channel");
Bram Moolenaar77073442016-02-13 23:23:53 +0100285
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100286#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100287 for (part = PART_SOCK; part <= PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100288#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100289 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100290#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100291 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100292 channel->ch_part[part].ch_fd = INVALID_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100293#ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100294 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100295#endif
296#ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100297 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100298#endif
299#ifdef FEAT_GUI_W32
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100300 channel->ch_part[part].ch_inputHandler = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100301#endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100302 channel->ch_part[part].ch_timeout = 2000;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100303 }
Bram Moolenaare0874f82016-01-24 20:36:41 +0100304
Bram Moolenaar77073442016-02-13 23:23:53 +0100305 if (first_channel != NULL)
306 {
307 first_channel->ch_prev = channel;
308 channel->ch_next = first_channel;
309 }
310 first_channel = channel;
311
312 channel->ch_refcount = 1;
313 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100314}
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100315
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100316/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100317 * Close a channel and free all its resources.
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100318 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100319 void
320channel_free(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100321{
Bram Moolenaar77073442016-02-13 23:23:53 +0100322 channel_close(channel);
323 if (channel->ch_next != NULL)
324 channel->ch_next->ch_prev = channel->ch_prev;
325 if (channel->ch_prev == NULL)
326 first_channel = channel->ch_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100327 else
Bram Moolenaar77073442016-02-13 23:23:53 +0100328 channel->ch_prev->ch_next = channel->ch_next;
329 vim_free(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100330}
331
Bram Moolenaard04a0202016-01-26 23:30:18 +0100332#if defined(FEAT_GUI) || defined(PROTO)
Bram Moolenaar77073442016-02-13 23:23:53 +0100333
334#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
335 static void
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100336channel_read_fd(int fd)
Bram Moolenaar77073442016-02-13 23:23:53 +0100337{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100338 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100339 int part;
Bram Moolenaar77073442016-02-13 23:23:53 +0100340
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100341 channel = channel_fd2channel(fd, &part);
Bram Moolenaar77073442016-02-13 23:23:53 +0100342 if (channel == NULL)
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100343 ch_errorn(NULL, "Channel for fd %d not found", fd);
Bram Moolenaar77073442016-02-13 23:23:53 +0100344 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100345 channel_read(channel, part, "messageFromNetbeans");
Bram Moolenaar77073442016-02-13 23:23:53 +0100346}
347#endif
348
Bram Moolenaare0874f82016-01-24 20:36:41 +0100349/*
Bram Moolenaard04a0202016-01-26 23:30:18 +0100350 * Read a command from netbeans.
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100351 * TODO: instead of channel ID use the FD.
Bram Moolenaare0874f82016-01-24 20:36:41 +0100352 */
Bram Moolenaard04a0202016-01-26 23:30:18 +0100353#ifdef FEAT_GUI_X11
354 static void
355messageFromNetbeans(XtPointer clientData,
356 int *unused1 UNUSED,
357 XtInputId *unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100358{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100359 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100360}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100361#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100362
Bram Moolenaard04a0202016-01-26 23:30:18 +0100363#ifdef FEAT_GUI_GTK
Bram Moolenaar98921892016-02-23 17:14:37 +0100364# if GTK_CHECK_VERSION(3,0,0)
365 static gboolean
366messageFromNetbeans(GIOChannel *unused1 UNUSED,
367 GIOCondition unused2 UNUSED,
368 gpointer clientData)
369{
370 channel_read_fd(GPOINTER_TO_INT(clientData));
371 return TRUE; /* Return FALSE instead in case the event source is to
372 * be removed after this function returns. */
373}
374# else
Bram Moolenaard04a0202016-01-26 23:30:18 +0100375 static void
376messageFromNetbeans(gpointer clientData,
377 gint unused1 UNUSED,
378 GdkInputCondition unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100379{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100380 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100381}
Bram Moolenaar98921892016-02-23 17:14:37 +0100382# endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100383#endif
384
385 static void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100386channel_gui_register_one(channel_T *channel, int part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100387{
388# ifdef FEAT_GUI_X11
389 /* Tell notifier we are interested in being called
390 * when there is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100391 if (channel->ch_part[part].ch_inputHandler == (XtInputId)NULL)
392 channel->ch_part[part].ch_inputHandler = XtAppAddInput(
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100393 (XtAppContext)app_context,
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100394 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100395 (XtPointer)(XtInputReadMask + XtInputExceptMask),
396 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100397 (XtPointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100398# else
399# ifdef FEAT_GUI_GTK
400 /* Tell gdk we are interested in being called when there
401 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100402 if (channel->ch_part[part].ch_inputHandler == 0)
Bram Moolenaar98921892016-02-23 17:14:37 +0100403# if GTK_CHECK_VERSION(3,0,0)
404 {
405 GIOChannel *chnnl = g_io_channel_unix_new(
406 (gint)channel->ch_part[part].ch_fd);
407
408 channel->ch_part[part].ch_inputHandler = g_io_add_watch(
409 chnnl,
410 G_IO_IN|G_IO_HUP|G_IO_ERR|G_IO_PRI,
411 messageFromNetbeans,
412 GINT_TO_POINTER(channel->ch_part[part].ch_fd));
413
414 g_io_channel_unref(chnnl);
415 }
416# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100417 channel->ch_part[part].ch_inputHandler = gdk_input_add(
418 (gint)channel->ch_part[part].ch_fd,
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100419 (GdkInputCondition)
420 ((int)GDK_INPUT_READ + (int)GDK_INPUT_EXCEPTION),
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100421 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100422 (gpointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar98921892016-02-23 17:14:37 +0100423# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100424# else
425# ifdef FEAT_GUI_W32
426 /* Tell Windows we are interested in receiving message when there
427 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100428 if (channel->ch_part[part].ch_inputHandler == -1)
429 channel->ch_part[part].ch_inputHandler = WSAAsyncSelect(
430 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100431 s_hwnd, WM_NETBEANS, FD_READ);
432# endif
433# endif
434# endif
435}
436
437 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100438channel_gui_register(channel_T *channel)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100439{
Bram Moolenaard04a0202016-01-26 23:30:18 +0100440 if (!CH_HAS_GUI)
441 return;
442
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100443 if (channel->CH_SOCK_FD != INVALID_FD)
444 channel_gui_register_one(channel, PART_SOCK);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100445# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100446 if (channel->CH_OUT_FD != INVALID_FD)
447 channel_gui_register_one(channel, PART_OUT);
448 if (channel->CH_ERR_FD != INVALID_FD)
449 channel_gui_register_one(channel, PART_ERR);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100450# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100451}
452
453/*
454 * Register any of our file descriptors with the GUI event handling system.
455 * Called when the GUI has started.
456 */
457 void
458channel_gui_register_all(void)
459{
Bram Moolenaar77073442016-02-13 23:23:53 +0100460 channel_T *channel;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100461
Bram Moolenaar77073442016-02-13 23:23:53 +0100462 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100463 channel_gui_register(channel);
464}
465
466 static void
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100467channel_gui_unregister(channel_T *channel)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100468{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100469 int part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100470
471#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100472 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100473#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100474 part = PART_SOCK;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100475#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100476 {
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100477# ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100478 if (channel->ch_part[part].ch_inputHandler != (XtInputId)NULL)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100479 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100480 XtRemoveInput(channel->ch_part[part].ch_inputHandler);
481 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100482 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100483# else
484# ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100485 if (channel->ch_part[part].ch_inputHandler != 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100486 {
Bram Moolenaar98921892016-02-23 17:14:37 +0100487# if GTK_CHECK_VERSION(3,0,0)
488 g_source_remove(channel->ch_part[part].ch_inputHandler);
489# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100490 gdk_input_remove(channel->ch_part[part].ch_inputHandler);
Bram Moolenaar98921892016-02-23 17:14:37 +0100491# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100492 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100493 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100494# else
495# ifdef FEAT_GUI_W32
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100496 if (channel->ch_part[part].ch_inputHandler == 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100497 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100498 WSAAsyncSelect(channel->ch_part[part].ch_fd, s_hwnd, 0, 0);
499 channel->ch_part[part].ch_inputHandler = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100500 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100501# endif
502# endif
503# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100504 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100505}
506
507#endif
508
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100509static char *e_cannot_connect = N_("E902: Cannot connect to port");
510
Bram Moolenaard04a0202016-01-26 23:30:18 +0100511/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100512 * Open a socket channel to "hostname":"port".
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100513 * "waittime" is the time in msec to wait for the connection.
514 * When negative wait forever.
Bram Moolenaar77073442016-02-13 23:23:53 +0100515 * Returns the channel for success.
516 * Returns NULL for failure.
Bram Moolenaard04a0202016-01-26 23:30:18 +0100517 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100518 channel_T *
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100519channel_open(
520 char *hostname,
521 int port_in,
522 int waittime,
523 void (*nb_close_cb)(void))
Bram Moolenaard04a0202016-01-26 23:30:18 +0100524{
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100525 int sd = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100526 struct sockaddr_in server;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100527 struct hostent *host;
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100528#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100529 u_short port = port_in;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100530 u_long val = 1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100531#else
532 int port = port_in;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100533 struct timeval start_tv;
534 int so_error;
535 socklen_t so_error_len = sizeof(so_error);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100536#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100537 channel_T *channel;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100538 int ret;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100539
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100540#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100541 channel_init_winsock();
542#endif
543
Bram Moolenaar77073442016-02-13 23:23:53 +0100544 channel = add_channel();
545 if (channel == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100546 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100547 ch_error(NULL, "Cannot allocate channel.");
Bram Moolenaar77073442016-02-13 23:23:53 +0100548 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100549 }
550
551 /* Get the server internet address and put into addr structure */
552 /* fill in the socket address structure and connect to server */
553 vim_memset((char *)&server, 0, sizeof(server));
554 server.sin_family = AF_INET;
555 server.sin_port = htons(port);
556 if ((host = gethostbyname(hostname)) == NULL)
557 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100558 ch_error(channel, "in gethostbyname() in channel_open()");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100559 PERROR("E901: gethostbyname() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100560 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100561 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100562 }
563 memcpy((char *)&server.sin_addr, host->h_addr, host->h_length);
564
Bram Moolenaar254e00d2016-02-19 23:23:12 +0100565 /* On Mac and Solaris a zero timeout almost never works. At least wait
566 * one millisecond. Let's do it for all systems, because we don't know why
567 * this is needed. */
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100568 if (waittime == 0)
569 waittime = 1;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100570
571 /*
572 * For Unix we need to call connect() again after connect() failed.
573 * On Win32 one time is sufficient.
574 */
575 while (TRUE)
576 {
577 if (sd >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100578 sock_close(sd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100579 sd = socket(AF_INET, SOCK_STREAM, 0);
580 if (sd == -1)
581 {
582 ch_error(channel, "in socket() in channel_open().");
583 PERROR("E898: socket() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100584 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100585 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100586 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100587
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100588 if (waittime >= 0)
589 {
590 /* Make connect() non-blocking. */
591 if (
592#ifdef _WIN32
593 ioctlsocket(sd, FIONBIO, &val) < 0
594#else
595 fcntl(sd, F_SETFL, O_NONBLOCK) < 0
596#endif
597 )
598 {
599 SOCK_ERRNO;
600 ch_errorn(channel,
601 "channel_open: Connect failed with errno %d", errno);
602 sock_close(sd);
603 channel_free(channel);
604 return NULL;
605 }
606 }
607
608 /* Try connecting to the server. */
609 ch_logsn(channel, "Connecting to %s port %d", hostname, port);
610 ret = connect(sd, (struct sockaddr *)&server, sizeof(server));
611
612 SOCK_ERRNO;
613 if (ret < 0)
614 {
615 if (errno != EWOULDBLOCK
616 && errno != ECONNREFUSED
617
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100618#ifdef EINPROGRESS
619 && errno != EINPROGRESS
620#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100621 )
622 {
623 ch_errorn(channel,
624 "channel_open: Connect failed with errno %d", errno);
625 PERROR(_(e_cannot_connect));
626 sock_close(sd);
627 channel_free(channel);
628 return NULL;
629 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100630 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100631
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100632 /* If we don't block and connect() failed then try using select() to
633 * wait for the connection to be made. */
634 if (waittime >= 0 && ret < 0)
635 {
636 struct timeval tv;
637 fd_set wfds;
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100638#if defined(__APPLE__) && __APPLE__ == 1
639# define PASS_RFDS
Bram Moolenaar98921892016-02-23 17:14:37 +0100640 fd_set rfds;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100641
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100642 FD_ZERO(&rfds);
643 FD_SET(sd, &rfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100644#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100645 FD_ZERO(&wfds);
646 FD_SET(sd, &wfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100647
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100648 tv.tv_sec = waittime / 1000;
649 tv.tv_usec = (waittime % 1000) * 1000;
650#ifndef WIN32
651 gettimeofday(&start_tv, NULL);
652#endif
653 ch_logn(channel,
654 "Waiting for connection (waittime %d msec)...", waittime);
655 ret = select((int)sd + 1,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100656#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100657 &rfds,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100658#else
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100659 NULL,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100660#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100661 &wfds, NULL, &tv);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100662
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100663 if (ret < 0)
664 {
665 SOCK_ERRNO;
666 ch_errorn(channel,
667 "channel_open: Connect failed with errno %d", errno);
668 PERROR(_(e_cannot_connect));
669 sock_close(sd);
670 channel_free(channel);
671 return NULL;
672 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100673#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100674 if (ret == 0 && FD_ISSET(sd, &rfds) && FD_ISSET(sd, &wfds))
675 {
676 /* For OS X, this implies error. See tcp(4). */
677 ch_error(channel, "channel_open: Connect failed");
678 EMSG(_(e_cannot_connect));
679 sock_close(sd);
680 channel_free(channel);
681 return NULL;
682 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100683#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100684#ifdef WIN32
685 /* On Win32 select() is expected to work and wait for up to the
686 * waittime for the socket to be open. */
687 if (!FD_ISSET(sd, &wfds) || ret == 0)
688#else
689 /* See socket(7) for the behavior on Linux-like systems:
690 * After putting the socket in non-blocking mode, connect() will
691 * return EINPROGRESS, select() will not wait (as if writing is
692 * possible), need to use getsockopt() to check if the socket is
693 * actually open. */
694 getsockopt(sd, SOL_SOCKET, SO_ERROR, &so_error, &so_error_len);
695 if (!FD_ISSET(sd, &wfds) || ret == 0 || so_error != 0)
696#endif
697 {
698#ifndef WIN32
699 struct timeval end_tv;
700 long elapsed_msec;
701
702 gettimeofday(&end_tv, NULL);
703 elapsed_msec = (end_tv.tv_sec - start_tv.tv_sec) * 1000
704 + (end_tv.tv_usec - start_tv.tv_usec) / 1000;
705 if (waittime > 1 && elapsed_msec < waittime)
706 {
707 /* The port isn't ready but we also didn't get an error.
708 * This happens when the server didn't open the socket
709 * yet. Wait a bit and try again. */
710 mch_delay(waittime < 50 ? (long)waittime : 50L, TRUE);
711 ui_breakcheck();
712 if (!got_int)
713 {
714 /* reduce the waittime by the elapsed time and the 50
715 * msec delay (or a bit more) */
716 waittime -= elapsed_msec;
717 if (waittime > 50)
718 waittime -= 50;
719 else
720 waittime = 1;
721 continue;
722 }
723 /* we were interrupted, behave as if timed out */
724 }
725#endif
726 /* We timed out. */
727 ch_error(channel, "Connection timed out");
728 sock_close(sd);
729 channel_free(channel);
730 return NULL;
731 }
732
733 ch_log(channel, "Connection made");
734 break;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100735 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100736 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100737
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100738 if (waittime >= 0)
739 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100740#ifdef _WIN32
741 val = 0;
742 ioctlsocket(sd, FIONBIO, &val);
743#else
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100744 (void)fcntl(sd, F_SETFL, 0);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100745#endif
746 }
747
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100748 channel->CH_SOCK_FD = (sock_T)sd;
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100749 channel->ch_nb_close_cb = nb_close_cb;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100750
751#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +0100752 channel_gui_register(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100753#endif
754
Bram Moolenaar77073442016-02-13 23:23:53 +0100755 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100756}
757
758#if defined(CHANNEL_PIPES) || defined(PROTO)
759 void
Bram Moolenaard8070362016-02-15 21:56:54 +0100760channel_set_pipes(channel_T *channel, sock_T in, sock_T out, sock_T err)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100761{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100762 channel->CH_IN_FD = in;
763 channel->CH_OUT_FD = out;
764 channel->CH_ERR_FD = err;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100765}
766#endif
767
768 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100769channel_set_job(channel_T *channel, job_T *job)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100770{
Bram Moolenaar77073442016-02-13 23:23:53 +0100771 channel->ch_job = job;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100772}
773
774/*
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100775 * Set various properties from an "opt" argument.
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100776 */
777 void
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100778channel_set_options(channel_T *channel, jobopt_T *opt)
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100779{
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100780 int part;
781 char_u **cbp;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100782
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100783 if (opt->jo_set & JO_MODE)
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100784 for (part = PART_SOCK; part <= PART_IN; ++part)
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100785 channel->ch_part[part].ch_mode = opt->jo_mode;
786 if (opt->jo_set & JO_IN_MODE)
787 channel->ch_part[PART_IN].ch_mode = opt->jo_in_mode;
788 if (opt->jo_set & JO_OUT_MODE)
789 channel->ch_part[PART_OUT].ch_mode = opt->jo_out_mode;
790 if (opt->jo_set & JO_ERR_MODE)
791 channel->ch_part[PART_ERR].ch_mode = opt->jo_err_mode;
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100792
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100793 if (opt->jo_set & JO_TIMEOUT)
794 for (part = PART_SOCK; part <= PART_IN; ++part)
795 channel->ch_part[part].ch_timeout = opt->jo_timeout;
796 if (opt->jo_set & JO_OUT_TIMEOUT)
797 channel->ch_part[PART_OUT].ch_timeout = opt->jo_out_timeout;
798 if (opt->jo_set & JO_ERR_TIMEOUT)
799 channel->ch_part[PART_ERR].ch_timeout = opt->jo_err_timeout;
800
801 if (opt->jo_set & JO_CALLBACK)
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100802 {
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100803 cbp = &channel->ch_callback;
804 vim_free(*cbp);
805 if (opt->jo_callback != NULL && *opt->jo_callback != NUL)
806 *cbp = vim_strsave(opt->jo_callback);
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100807 else
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100808 *cbp = NULL;
809 }
810 if (opt->jo_set & JO_OUT_CALLBACK)
811 {
812 cbp = &channel->ch_part[PART_OUT].ch_callback;
813 vim_free(*cbp);
814 if (opt->jo_out_cb != NULL && *opt->jo_out_cb != NUL)
815 *cbp = vim_strsave(opt->jo_out_cb);
816 else
817 *cbp = NULL;
818 }
819 if (opt->jo_set & JO_ERR_CALLBACK)
820 {
821 cbp = &channel->ch_part[PART_ERR].ch_callback;
822 vim_free(*cbp);
823 if (opt->jo_err_cb != NULL && *opt->jo_err_cb != NUL)
824 *cbp = vim_strsave(opt->jo_err_cb);
825 else
826 *cbp = NULL;
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100827 }
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100828 if (opt->jo_set & JO_CLOSE_CALLBACK)
829 {
830 cbp = &channel->ch_close_cb;
831 vim_free(*cbp);
832 if (opt->jo_close_cb != NULL && *opt->jo_close_cb != NUL)
833 *cbp = vim_strsave(opt->jo_close_cb);
834 else
835 *cbp = NULL;
836 }
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100837}
838
839/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100840 * Set the callback for "channel"/"part" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100841 */
842 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100843channel_set_req_callback(
844 channel_T *channel,
845 int part,
846 char_u *callback,
847 int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100848{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100849 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100850 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
851
852 if (item != NULL)
853 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100854 item->cq_callback = vim_strsave(callback);
855 item->cq_seq_nr = id;
856 item->cq_prev = head->cq_prev;
857 head->cq_prev = item;
858 item->cq_next = NULL;
859 if (item->cq_prev == NULL)
860 head->cq_next = item;
861 else
862 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100863 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100864}
865
866/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100867 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100868 */
869 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100870invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100871{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100872 typval_T rettv;
873 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100874
Bram Moolenaar77073442016-02-13 23:23:53 +0100875 argv[0].v_type = VAR_CHANNEL;
876 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100877
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100878 call_func(callback, (int)STRLEN(callback),
879 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
Bram Moolenaaree1cffc2016-02-21 19:14:41 +0100880 clear_tv(&rettv);
881
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100882 /* If an echo command was used the cursor needs to be put back where
883 * it belongs. */
884 setcursor();
885 cursor_on();
886 out_flush();
887}
888
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100889/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100890 * Return the first buffer from channel "channel"/"part" and remove it.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100891 * The caller must free it.
892 * Returns NULL if there is nothing.
893 */
894 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100895channel_get(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100896{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100897 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100898 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100899 char_u *p;
900
Bram Moolenaar77073442016-02-13 23:23:53 +0100901 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100902 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100903 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100904 p = node->rq_buffer;
905 head->rq_next = node->rq_next;
906 if (node->rq_next == NULL)
907 head->rq_prev = NULL;
908 else
909 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100910 vim_free(node);
911 return p;
912}
913
914/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100915 * Returns the whole buffer contents concatenated for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100916 */
917 static char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100918channel_get_all(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100919{
920 /* Concatenate everything into one buffer.
921 * TODO: avoid multiple allocations. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100922 while (channel_collapse(channel, part) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100923 ;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100924 return channel_get(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100925}
926
927/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100928 * Collapses the first and second buffer for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100929 * Returns FAIL if that is not possible.
930 */
931 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100932channel_collapse(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100933{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100934 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100935 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100936 char_u *p;
937
Bram Moolenaar77073442016-02-13 23:23:53 +0100938 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100939 return FAIL;
940
Bram Moolenaar77073442016-02-13 23:23:53 +0100941 p = alloc((unsigned)(STRLEN(node->rq_buffer)
942 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100943 if (p == NULL)
944 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100945 STRCPY(p, node->rq_buffer);
946 STRCAT(p, node->rq_next->rq_buffer);
947 vim_free(node->rq_next->rq_buffer);
948 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100949
Bram Moolenaar77073442016-02-13 23:23:53 +0100950 /* dispose of the node and its buffer */
951 head->rq_next = node->rq_next;
952 head->rq_next->rq_prev = NULL;
953 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100954 vim_free(node);
955 return OK;
956}
957
958/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100959 * Store "buf[len]" on "channel"/"part".
960 * Returns OK or FAIL.
961 */
962 static int
963channel_save(channel_T *channel, int part, char_u *buf, int len)
964{
965 readq_T *node;
966 readq_T *head = &channel->ch_part[part].ch_head;
967 char_u *p;
968 int i;
969
970 node = (readq_T *)alloc(sizeof(readq_T));
971 if (node == NULL)
972 return FAIL; /* out of memory */
973 node->rq_buffer = alloc(len + 1);
974 if (node->rq_buffer == NULL)
975 {
976 vim_free(node);
977 return FAIL; /* out of memory */
978 }
979
980 if (channel->ch_part[part].ch_mode == MODE_NL)
981 {
982 /* Drop any CR before a NL. */
983 p = node->rq_buffer;
984 for (i = 0; i < len; ++i)
985 if (buf[i] != CAR || i + 1 >= len || buf[i + 1] != NL)
986 *p++ = buf[i];
987 *p = NUL;
988 }
989 else
990 {
991 mch_memmove(node->rq_buffer, buf, len);
992 node->rq_buffer[len] = NUL;
993 }
994
995 /* append node to the tail of the queue */
996 node->rq_next = NULL;
997 node->rq_prev = head->rq_prev;
998 if (head->rq_prev == NULL)
999 head->rq_next = node;
1000 else
1001 head->rq_prev->rq_next = node;
1002 head->rq_prev = node;
1003
1004 if (log_fd != NULL)
1005 {
1006 ch_log_lead("RECV ", channel);
1007 fprintf(log_fd, "'");
1008 if (fwrite(buf, len, 1, log_fd) != 1)
1009 return FAIL;
1010 fprintf(log_fd, "'\n");
1011 }
1012 return OK;
1013}
1014
1015/*
1016 * Use the read buffer of "channel"/"part" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001017 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +01001018 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001019 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001020 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001021channel_parse_json(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001022{
1023 js_read_T reader;
1024 typval_T listtv;
1025 jsonq_T *item;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001026 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001027 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001028
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001029 if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001030 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001031
1032 /* TODO: make reader work properly */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001033 /* reader.js_buf = channel_peek(channel, part); */
1034 reader.js_buf = channel_get_all(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001035 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +01001036 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001037 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +01001038 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001039 ret = json_decode(&reader, &listtv,
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001040 channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001041 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001042 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +01001043 /* Only accept the response when it is a list with at least two
1044 * items. */
1045 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001046 {
1047 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001048 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001049 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001050 else
1051 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001052 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
1053 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001054 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001055 else
1056 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001057 item->jq_value = alloc_tv();
1058 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001059 {
1060 vim_free(item);
1061 clear_tv(&listtv);
1062 }
1063 else
1064 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001065 *item->jq_value = listtv;
1066 item->jq_prev = head->jq_prev;
1067 head->jq_prev = item;
1068 item->jq_next = NULL;
1069 if (item->jq_prev == NULL)
1070 head->jq_next = item;
1071 else
1072 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001073 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001074 }
1075 }
1076 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001077
1078 /* Put the unread part back into the channel.
1079 * TODO: insert in front */
1080 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001081 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001082 channel_save(channel, part, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001083 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001084 ret = TRUE;
1085 }
1086 else
1087 ret = FALSE;
1088
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001089 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001090 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001091}
1092
1093/*
Bram Moolenaard46ae142016-02-16 13:33:52 +01001094 * Remove "node" from the queue that it is in. Does not free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +01001095 */
1096 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001097remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001098{
Bram Moolenaar77073442016-02-13 23:23:53 +01001099 if (node->cq_prev == NULL)
1100 head->cq_next = node->cq_next;
1101 else
1102 node->cq_prev->cq_next = node->cq_next;
1103 if (node->cq_next == NULL)
1104 head->cq_prev = node->cq_prev;
1105 else
1106 node->cq_next->cq_prev = node->cq_prev;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001107}
1108
1109/*
1110 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +01001111 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001112 */
1113 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001114remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001115{
Bram Moolenaar77073442016-02-13 23:23:53 +01001116 if (node->jq_prev == NULL)
1117 head->jq_next = node->jq_next;
1118 else
1119 node->jq_prev->jq_next = node->jq_next;
1120 if (node->jq_next == NULL)
1121 head->jq_prev = node->jq_prev;
1122 else
1123 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001124 vim_free(node);
1125}
1126
1127/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001128 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001129 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +01001130 * When "id" is zero or negative jut get the first message. But not the one
1131 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001132 * Return OK when found and return the value in "rettv".
1133 * Return FAIL otherwise.
1134 */
1135 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001136channel_get_json(channel_T *channel, int part, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001137{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001138 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001139 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001140
Bram Moolenaar77073442016-02-13 23:23:53 +01001141 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001142 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001143 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001144 typval_T *tv = &l->lv_first->li_tv;
1145
1146 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001147 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001148 || tv->vval.v_number == 0
1149 || tv->vval.v_number != channel->ch_part[part].ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001150 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001151 *rettv = item->jq_value;
1152 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001153 return OK;
1154 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001155 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001156 }
1157 return FAIL;
1158}
1159
Bram Moolenaarece61b02016-02-20 21:39:05 +01001160#define CH_JSON_MAX_ARGS 4
1161
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001162/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001163 * Execute a command received over "channel"/"part"
Bram Moolenaarece61b02016-02-20 21:39:05 +01001164 * "argv[0]" is the command string.
1165 * "argv[1]" etc. have further arguments, type is VAR_UNKNOWN if missing.
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001166 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001167 static void
Bram Moolenaarece61b02016-02-20 21:39:05 +01001168channel_exe_cmd(channel_T *channel, int part, typval_T *argv)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001169{
Bram Moolenaarece61b02016-02-20 21:39:05 +01001170 char_u *cmd = argv[0].vval.v_string;
1171 char_u *arg;
1172 int options = channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001173
Bram Moolenaarece61b02016-02-20 21:39:05 +01001174 if (argv[1].v_type != VAR_STRING)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001175 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001176 ch_error(channel, "received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001177 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001178 EMSG("E903: received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001179 return;
1180 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001181 arg = argv[1].vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001182 if (arg == NULL)
1183 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001184
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001185 if (STRCMP(cmd, "ex") == 0)
1186 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001187 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001188 }
1189 else if (STRCMP(cmd, "normal") == 0)
1190 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001191 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001192
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001193 ea.arg = arg;
1194 ea.addr_count = 0;
1195 ea.forceit = TRUE; /* no mapping */
1196 ex_normal(&ea);
1197 }
1198 else if (STRCMP(cmd, "redraw") == 0)
1199 {
1200 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001201
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001202 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001203 ex_redraw(&ea);
1204 showruler(FALSE);
1205 setcursor();
1206 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001207#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001208 if (gui.in_use)
1209 {
1210 gui_update_cursor(FALSE, FALSE);
1211 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001212 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001213#endif
1214 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001215 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "call") == 0)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001216 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001217 int is_call = cmd[0] == 'c';
1218 int id_idx = is_call ? 3 : 2;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001219
Bram Moolenaarece61b02016-02-20 21:39:05 +01001220 if (argv[id_idx].v_type != VAR_UNKNOWN
1221 && argv[id_idx].v_type != VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001222 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001223 ch_error(channel, "last argument for expr/call must be a number");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001224 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001225 EMSG("E904: last argument for expr/call must be a number");
1226 }
1227 else if (is_call && argv[2].v_type != VAR_LIST)
1228 {
1229 ch_error(channel, "third argument for call must be a list");
1230 if (p_verbose > 2)
1231 EMSG("E904: third argument for call must be a list");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001232 }
1233 else
1234 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001235 typval_T *tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001236 typval_T res_tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001237 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +01001238 char_u *json = NULL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001239
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001240 /* Don't pollute the display with errors. */
1241 ++emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001242 if (!is_call)
1243 tv = eval_expr(arg, NULL);
1244 else if (func_call(arg, &argv[2], NULL, &res_tv) == OK)
1245 tv = &res_tv;
1246 else
1247 tv = NULL;
1248
1249 if (argv[id_idx].v_type == VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001250 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001251 int id = argv[id_idx].vval.v_number;
1252
Bram Moolenaar55fab432016-02-07 16:53:13 +01001253 if (tv != NULL)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001254 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001255 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001256 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001257 /* If evaluation failed or the result can't be encoded
1258 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001259 vim_free(json);
1260 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001261 err_tv.v_type = VAR_STRING;
1262 err_tv.vval.v_string = (char_u *)"ERROR";
1263 tv = &err_tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001264 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001265 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001266 if (json != NULL)
1267 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001268 channel_send(channel,
1269 part == PART_SOCK ? PART_SOCK : PART_IN,
1270 json, (char *)cmd);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001271 vim_free(json);
1272 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001273 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001274 --emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001275 if (tv == &res_tv)
1276 clear_tv(tv);
1277 else if (tv != &err_tv)
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001278 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001279 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001280 }
1281 else if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001282 {
1283 ch_errors(channel, "Receved unknown command: %s", (char *)cmd);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001284 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001285 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001286}
1287
1288/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001289 * Invoke a callback for "channel"/"part" if needed.
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001290 * Return TRUE when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001291 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001292 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001293may_invoke_callback(channel_T *channel, int part)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001294{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001295 char_u *msg = NULL;
1296 typval_T *listtv = NULL;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001297 typval_T argv[CH_JSON_MAX_ARGS];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001298 int seq_nr = -1;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001299 ch_mode_T ch_mode = channel->ch_part[part].ch_mode;
1300 char_u *callback = NULL;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001301
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001302 if (channel->ch_nb_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001303 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001304 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001305
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001306 if (channel->ch_part[part].ch_callback != NULL)
1307 callback = channel->ch_part[part].ch_callback;
1308 else
1309 callback = channel->ch_callback;
1310
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001311 if (ch_mode == MODE_JSON || ch_mode == MODE_JS)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001312 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001313 listitem_T *item;
1314 int argc = 0;
1315
Bram Moolenaard7ece102016-02-02 23:23:02 +01001316 /* Get any json message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001317 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001318 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001319 /* Parse readahead, return when there is still no message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001320 channel_parse_json(channel, part);
1321 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001322 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001323 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001324
Bram Moolenaarece61b02016-02-20 21:39:05 +01001325 for (item = listtv->vval.v_list->lv_first;
1326 item != NULL && argc < CH_JSON_MAX_ARGS;
1327 item = item->li_next)
1328 argv[argc++] = item->li_tv;
1329 while (argc < CH_JSON_MAX_ARGS)
1330 argv[argc++].v_type = VAR_UNKNOWN;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001331
Bram Moolenaarece61b02016-02-20 21:39:05 +01001332 if (argv[0].v_type == VAR_STRING)
1333 {
1334 char_u *cmd = argv[0].vval.v_string;
1335
1336 /* ["cmd", arg] or ["cmd", arg, arg] or ["cmd", arg, arg, arg] */
Bram Moolenaar77073442016-02-13 23:23:53 +01001337 ch_logs(channel, "Executing %s command", (char *)cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001338 channel_exe_cmd(channel, part, argv);
Bram Moolenaar77073442016-02-13 23:23:53 +01001339 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001340 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001341 }
1342
Bram Moolenaarece61b02016-02-20 21:39:05 +01001343 if (argv[0].v_type != VAR_NUMBER)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001344 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001345 ch_error(channel,
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001346 "Dropping message with invalid sequence number type");
Bram Moolenaar77073442016-02-13 23:23:53 +01001347 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001348 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001349 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001350 seq_nr = argv[0].vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001351 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001352 else if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001353 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001354 /* nothing to read on RAW or NL channel */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001355 return FALSE;
1356 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001357 else
1358 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001359 /* If there is no callback drop the message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001360 if (callback == NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001361 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001362 while ((msg = channel_get(channel, part)) != NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001363 vim_free(msg);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001364 return FALSE;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001365 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001366
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001367 if (ch_mode == MODE_NL)
1368 {
1369 char_u *nl;
1370 char_u *buf;
1371
1372 /* See if we have a message ending in NL in the first buffer. If
1373 * not try to concatenate the first and the second buffer. */
1374 while (TRUE)
1375 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001376 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001377 nl = vim_strchr(buf, NL);
1378 if (nl != NULL)
1379 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001380 if (channel_collapse(channel, part) == FAIL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001381 return FALSE; /* incomplete message */
1382 }
1383 if (nl[1] == NUL)
1384 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001385 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001386 else
1387 {
1388 /* Copy the message into allocated memory and remove it from
1389 * the buffer. */
1390 msg = vim_strnsave(buf, (int)(nl - buf));
1391 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1392 }
1393 }
1394 else
1395 /* For a raw channel we don't know where the message ends, just
1396 * get everything we have. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001397 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001398
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001399 argv[1].v_type = VAR_STRING;
1400 argv[1].vval.v_string = msg;
1401 }
1402
Bram Moolenaara07fec92016-02-05 21:04:08 +01001403 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001404 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001405 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001406 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001407 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001408
1409 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001410 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001411 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001412 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001413 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001414 ch_log(channel, "Invoking one-time callback");
Bram Moolenaard46ae142016-02-16 13:33:52 +01001415 /* Remove the item from the list first, if the callback
1416 * invokes ch_close() the list will be cleared. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001417 remove_cb_node(head, item);
Bram Moolenaard46ae142016-02-16 13:33:52 +01001418 invoke_callback(channel, item->cq_callback, argv);
1419 vim_free(item->cq_callback);
1420 vim_free(item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001421 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001422 break;
1423 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001424 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001425 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001426 if (!done)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001427 ch_log(channel, "Dropping message without callback");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001428 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001429 else if (callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001430 {
1431 /* invoke the channel callback */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001432 ch_log(channel, "Invoking channel callback");
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001433 invoke_callback(channel, callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001434 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001435 else
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001436 ch_log(channel, "Dropping message");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001437
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001438 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001439 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001440 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001441
1442 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001443}
1444
1445/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001446 * Return TRUE when channel "channel" is open for writing to.
1447 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001448 */
1449 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001450channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001451{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001452 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001453#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001454 || channel->CH_IN_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001455#endif
1456 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001457}
1458
1459/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001460 * Return TRUE when channel "channel" is open for reading or writing.
1461 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001462 */
1463 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001464channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001465{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001466 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001467#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001468 || channel->CH_IN_FD != INVALID_FD
1469 || channel->CH_OUT_FD != INVALID_FD
1470 || channel->CH_ERR_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001471#endif
1472 );
1473}
1474
1475/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001476 * Return a string indicating the status of the channel.
1477 */
1478 char *
1479channel_status(channel_T *channel)
1480{
1481 if (channel == NULL)
1482 return "fail";
1483 if (channel_is_open(channel))
1484 return "open";
1485 return "closed";
1486}
1487
1488/*
1489 * Close channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001490 * This does not trigger the close callback.
1491 */
1492 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001493channel_close(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001494{
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001495 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001496
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001497#ifdef FEAT_GUI
1498 channel_gui_unregister(channel);
1499#endif
1500
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001501 if (channel->CH_SOCK_FD != INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001502 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001503 sock_close(channel->CH_SOCK_FD);
1504 channel->CH_SOCK_FD = INVALID_FD;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001505 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001506#if defined(CHANNEL_PIPES)
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001507 if (channel->CH_IN_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001508 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001509 fd_close(channel->CH_IN_FD);
1510 channel->CH_IN_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001511 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001512 if (channel->CH_OUT_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001513 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001514 fd_close(channel->CH_OUT_FD);
1515 channel->CH_OUT_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001516 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001517 if (channel->CH_ERR_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001518 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001519 fd_close(channel->CH_ERR_FD);
1520 channel->CH_ERR_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001521 }
1522#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001523
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001524 if (channel->ch_close_cb != NULL)
1525 {
1526 typval_T argv[1];
1527 typval_T rettv;
1528 int dummy;
1529
1530 /* invoke the close callback; increment the refcount to avoid it
1531 * being freed halfway */
1532 argv[0].v_type = VAR_CHANNEL;
1533 argv[0].vval.v_channel = channel;
1534 ++channel->ch_refcount;
1535 call_func(channel->ch_close_cb, (int)STRLEN(channel->ch_close_cb),
1536 &rettv, 1, argv, 0L, 0L, &dummy, TRUE, NULL);
1537 clear_tv(&rettv);
1538 --channel->ch_refcount;
1539
1540 /* the callback is only called once */
1541 vim_free(channel->ch_close_cb);
1542 channel->ch_close_cb = NULL;
1543 }
1544
1545 channel->ch_nb_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001546 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001547}
1548
Bram Moolenaard04a0202016-01-26 23:30:18 +01001549/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001550 * Return the first buffer from "channel"/"part" without removing it.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001551 * Returns NULL if there is nothing.
1552 */
1553 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001554channel_peek(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001555{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001556 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001557
Bram Moolenaar77073442016-02-13 23:23:53 +01001558 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001559 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001560 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001561}
1562
1563/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001564 * Clear the read buffer on "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001565 */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001566 static void
1567channel_clear_one(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001568{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001569 jsonq_T *json_head = &channel->ch_part[part].ch_json_head;
1570 cbq_T *cb_head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001571
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001572 while (channel_peek(channel, part) != NULL)
1573 vim_free(channel_get(channel, part));
Bram Moolenaar77073442016-02-13 23:23:53 +01001574
1575 while (cb_head->cq_next != NULL)
Bram Moolenaard46ae142016-02-16 13:33:52 +01001576 {
1577 cbq_T *node = cb_head->cq_next;
1578
1579 remove_cb_node(cb_head, node);
1580 vim_free(node->cq_callback);
1581 vim_free(node);
1582 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001583
1584 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001585 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001586 free_tv(json_head->jq_next->jq_value);
1587 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001588 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001589
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001590 vim_free(channel->ch_part[part].ch_callback);
1591 channel->ch_part[part].ch_callback = NULL;
1592}
1593
1594/*
1595 * Clear all the read buffers on "channel".
1596 */
1597 void
1598channel_clear(channel_T *channel)
1599{
1600 channel_clear_one(channel, PART_SOCK);
1601#ifdef CHANNEL_PIPES
1602 channel_clear_one(channel, PART_OUT);
1603 channel_clear_one(channel, PART_ERR);
1604#endif
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001605 vim_free(channel->ch_callback);
1606 channel->ch_callback = NULL;
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001607 vim_free(channel->ch_close_cb);
1608 channel->ch_close_cb = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001609}
1610
Bram Moolenaar77073442016-02-13 23:23:53 +01001611#if defined(EXITFREE) || defined(PROTO)
1612 void
1613channel_free_all(void)
1614{
1615 channel_T *channel;
1616
1617 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1618 channel_clear(channel);
1619}
1620#endif
1621
1622
Bram Moolenaard04a0202016-01-26 23:30:18 +01001623/* Sent when the channel is found closed when reading. */
Bram Moolenaareed284a2016-02-22 23:13:33 +01001624#define DETACH_MSG_RAW "DETACH\n"
1625#define DETACH_MSG_JSON "\"DETACH\"\n"
Bram Moolenaard04a0202016-01-26 23:30:18 +01001626
1627/* Buffer size for reading incoming messages. */
1628#define MAXMSGSIZE 4096
1629
1630/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001631 * Check for reading from "fd" with "timeout" msec.
1632 * Return FAIL when there is nothing to read.
Bram Moolenaara8343c12016-02-04 22:09:48 +01001633 * Always returns OK for FEAT_GUI_W32.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001634 */
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 Moolenaared5a78e2016-02-19 21:05:03 +01001665#if defined(FEAT_GUI_W32)
1666 /* Can't check socket for Win32 GUI, always return OK. */
1667 ch_log(channel, "Can't check, assuming there is something to read");
1668 return OK;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001669#else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001670# if defined(HAVE_SELECT)
1671 struct timeval tval;
1672 fd_set rfds;
1673 int ret;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001674
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001675 FD_ZERO(&rfds);
1676 FD_SET((int)fd, &rfds);
1677 tval.tv_sec = timeout / 1000;
1678 tval.tv_usec = (timeout % 1000) * 1000;
1679 for (;;)
1680 {
1681 ret = select((int)fd + 1, &rfds, NULL, NULL, &tval);
1682# ifdef EINTR
1683 SOCK_ERRNO;
1684 if (ret == -1 && errno == EINTR)
1685 continue;
1686# endif
1687 if (ret > 0)
1688 return OK;
1689 break;
1690 }
1691# else
1692 struct pollfd fds;
1693
1694 fds.fd = fd;
1695 fds.events = POLLIN;
1696 if (poll(&fds, 1, timeout) > 0)
1697 return OK;
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001698# endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001699#endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001700 }
1701 ch_log(channel, "Nothing to read");
1702 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001703}
1704
1705/*
1706 * Return a unique ID to be used in a message.
1707 */
1708 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001709channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001710{
1711 static int next_id = 1;
1712
1713 return next_id++;
1714}
1715
1716/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001717 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001718 * "part" is PART_SOCK, PART_OUT or PART_ERR.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001719 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001720 */
1721 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001722channel_read(channel_T *channel, int part, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001723{
1724 static char_u *buf = NULL;
1725 int len = 0;
1726 int readlen = 0;
Bram Moolenaard8070362016-02-15 21:56:54 +01001727 sock_T fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001728 int use_socket = FALSE;
Bram Moolenaareed284a2016-02-22 23:13:33 +01001729 char *msg;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001730
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001731 fd = channel->ch_part[part].ch_fd;
1732 if (fd == INVALID_FD)
1733 {
1734 ch_error(channel, "channel_read() called while socket is closed");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001735 return;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001736 }
1737 use_socket = fd == channel->CH_SOCK_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001738
1739 /* Allocate a buffer to read into. */
1740 if (buf == NULL)
1741 {
1742 buf = alloc(MAXMSGSIZE);
1743 if (buf == NULL)
1744 return; /* out of memory! */
1745 }
1746
1747 /* Keep on reading for as long as there is something to read.
1748 * Use select() or poll() to avoid blocking on a message that is exactly
1749 * MAXMSGSIZE long. */
1750 for (;;)
1751 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001752 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001753 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001754 if (use_socket)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001755 len = sock_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001756 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001757 len = fd_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001758 if (len <= 0)
1759 break; /* error or nothing more to read */
1760
1761 /* Store the read message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001762 channel_save(channel, part, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001763 readlen += len;
1764 if (len < MAXMSGSIZE)
1765 break; /* did read everything that's available */
1766 }
Bram Moolenaara8343c12016-02-04 22:09:48 +01001767#ifdef FEAT_GUI_W32
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001768 if (use_socket && len == SOCKET_ERROR)
Bram Moolenaara8343c12016-02-04 22:09:48 +01001769 {
1770 /* For Win32 GUI channel_wait() always returns OK and we handle the
1771 * situation that there is nothing to read here.
1772 * TODO: how about a timeout? */
1773 if (WSAGetLastError() == WSAEWOULDBLOCK)
1774 return;
1775 }
1776#endif
Bram Moolenaard04a0202016-01-26 23:30:18 +01001777
Bram Moolenaarbd73ae12016-02-22 22:19:22 +01001778 /* Reading a disconnection (readlen == 0), or an error.
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001779 * TODO: call error callback. */
Bram Moolenaarbd73ae12016-02-22 22:19:22 +01001780 if (readlen <= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001781 {
1782 /* Queue a "DETACH" netbeans message in the command queue in order to
1783 * terminate the netbeans session later. Do not end the session here
1784 * directly as we may be running in the context of a call to
1785 * netbeans_parse_messages():
1786 * netbeans_parse_messages
1787 * -> autocmd triggered while processing the netbeans cmd
1788 * -> ui_breakcheck
1789 * -> gui event loop or select loop
1790 * -> channel_read()
1791 */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001792 ch_errors(channel, "%s(): Cannot read", func);
Bram Moolenaareed284a2016-02-22 23:13:33 +01001793 msg = channel->ch_part[part].ch_mode == MODE_RAW
1794 || channel->ch_part[part].ch_mode == MODE_NL
1795 ? DETACH_MSG_RAW : DETACH_MSG_JSON;
1796 channel_save(channel, part, (char_u *)msg, (int)STRLEN(msg));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001797
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001798 /* TODO: When reading from stdout is not possible, should we try to
1799 * keep stdin and stderr open? Probably not, assume the other side
1800 * has died. */
1801 channel_close(channel);
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001802 if (channel->ch_nb_close_cb != NULL)
1803 (*channel->ch_nb_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001804
1805 if (len < 0)
1806 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001807 ch_error(channel, "channel_read(): cannot read from channel");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001808 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001809 }
1810 }
1811
1812#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001813 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001814 if (CH_HAS_GUI && gtk_main_level() > 0)
1815 gtk_main_quit();
1816#endif
1817}
1818
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001819/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001820 * Read from RAW or NL "channel"/"part". Blocks until there is something to
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001821 * read or the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001822 * Returns what was read in allocated memory.
1823 * Returns NULL in case of error or timeout.
1824 */
1825 char_u *
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001826channel_read_block(channel_T *channel, int part, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001827{
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001828 char_u *buf;
1829 char_u *msg;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001830 ch_mode_T mode = channel->ch_part[part].ch_mode;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001831 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001832 char_u *nl;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001833
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001834 ch_logsn(channel, "Blocking %s read, timeout: %d msec",
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001835 mode == MODE_RAW ? "RAW" : "NL", timeout);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001836
1837 while (TRUE)
1838 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001839 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001840 if (buf != NULL && (mode == MODE_RAW
1841 || (mode == MODE_NL && vim_strchr(buf, NL) != NULL)))
1842 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001843 if (buf != NULL && channel_collapse(channel, part) == OK)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001844 continue;
1845
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001846 /* Wait for up to the channel timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001847 if (fd == INVALID_FD
1848 || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001849 return NULL;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001850 channel_read(channel, part, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001851 }
1852
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001853 if (mode == MODE_RAW)
1854 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001855 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001856 }
1857 else
1858 {
1859 nl = vim_strchr(buf, NL);
1860 if (nl[1] == NUL)
1861 {
1862 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001863 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001864 *nl = NUL;
1865 }
1866 else
1867 {
1868 /* Copy the message into allocated memory and remove it from the
1869 * buffer. */
1870 msg = vim_strnsave(buf, (int)(nl - buf));
1871 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1872 }
1873 }
1874 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001875 ch_logn(channel, "Returning %d bytes", (int)STRLEN(msg));
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001876 return msg;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001877}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001878
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001879/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001880 * Read one JSON message with ID "id" from "channel"/"part" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001881 * result in "rettv".
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001882 * When "id" is -1 accept any message;
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001883 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001884 */
1885 int
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001886channel_read_json_block(
1887 channel_T *channel,
1888 int part,
1889 int timeout,
1890 int id,
1891 typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001892{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001893 int more;
Bram Moolenaard8070362016-02-15 21:56:54 +01001894 sock_T fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001895
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001896 ch_log(channel, "Reading JSON");
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001897 if (id != -1)
1898 channel->ch_part[part].ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001899 for (;;)
1900 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001901 more = channel_parse_json(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001902
1903 /* search for messsage "id" */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001904 if (channel_get_json(channel, part, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001905 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001906 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001907 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001908 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001909
Bram Moolenaard7ece102016-02-02 23:23:02 +01001910 if (!more)
1911 {
1912 /* Handle any other messages in the queue. If done some more
1913 * messages may have arrived. */
1914 if (channel_parse_messages())
1915 continue;
1916
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001917 /* Wait for up to the timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001918 fd = channel->ch_part[part].ch_fd;
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001919 if (fd == INVALID_FD || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001920 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001921 channel_read(channel, part, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001922 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001923 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001924 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001925 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001926}
1927
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001928# if defined(WIN32) || defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK) \
1929 || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001930/*
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001931 * Lookup the channel from the socket. Set "partp" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001932 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001933 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001934 channel_T *
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001935channel_fd2channel(sock_T fd, int *partp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001936{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001937 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001938 int part;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001939
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001940 if (fd != INVALID_FD)
Bram Moolenaar77073442016-02-13 23:23:53 +01001941 for (channel = first_channel; channel != NULL;
1942 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001943 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001944# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001945 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001946# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001947 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001948# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001949 if (channel->ch_part[part].ch_fd == fd)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001950 {
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001951 *partp = part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001952 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001953 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001954 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001955 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001956}
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001957# endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001958
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001959# if defined(WIN32) || defined(PROTO)
1960/*
1961 * Check the channels for anything that is ready to be read.
1962 * The data is put in the read queue.
1963 */
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001964 void
1965channel_handle_events(void)
1966{
1967 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001968 int part;
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001969 sock_T fd;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001970
1971 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1972 {
1973# ifdef FEAT_GUI_W32
1974 /* only check the pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001975 for (part = PART_OUT; part <= PART_ERR; ++part)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001976# else
1977# ifdef CHANNEL_PIPES
1978 /* check the socket and pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001979 for (part = PART_SOCK; part <= PART_ERR; ++part)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001980# else
1981 /* only check the socket */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001982 part = PART_SOCK;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001983# endif
1984# endif
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001985 {
1986 fd = channel->ch_part[part].ch_fd;
1987 if (fd != INVALID_FD && channel_wait(channel, fd, 0) == OK)
1988 channel_read(channel, part, "channel_handle_events");
1989 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001990 }
1991}
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001992# endif
1993
Bram Moolenaard04a0202016-01-26 23:30:18 +01001994/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001995 * Write "buf" (NUL terminated string) to "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001996 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001997 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001998 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001999 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002000channel_send(channel_T *channel, int part, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01002001{
Bram Moolenaard04a0202016-01-26 23:30:18 +01002002 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002003 int res;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002004 sock_T fd;
Bram Moolenaard04a0202016-01-26 23:30:18 +01002005
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002006 fd = channel->ch_part[part].ch_fd;
2007 if (fd == INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01002008 {
2009 if (!channel->ch_error && fun != NULL)
2010 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01002011 ch_errors(channel, "%s(): write while not connected", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01002012 EMSG2("E630: %s(): write while not connected", fun);
2013 }
2014 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01002015 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01002016 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01002017
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002018 if (log_fd != NULL)
2019 {
Bram Moolenaar77073442016-02-13 23:23:53 +01002020 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002021 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002022 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002023 fprintf(log_fd, "'\n");
2024 fflush(log_fd);
2025 }
2026
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002027 if (part == PART_SOCK)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002028 res = sock_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002029 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002030 res = fd_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002031 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01002032 {
2033 if (!channel->ch_error && fun != NULL)
2034 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01002035 ch_errors(channel, "%s(): write failed", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01002036 EMSG2("E631: %s(): write failed", fun);
2037 }
2038 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01002039 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01002040 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01002041
2042 channel->ch_error = FALSE;
2043 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01002044}
2045
2046# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002047/*
2048 * Add open channels to the poll struct.
2049 * Return the adjusted struct index.
2050 * The type of "fds" is hidden to avoid problems with the function proto.
2051 */
2052 int
2053channel_poll_setup(int nfd_in, void *fds_in)
2054{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002055 int nfd = nfd_in;
2056 channel_T *channel;
2057 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002058 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002059
Bram Moolenaar77073442016-02-13 23:23:53 +01002060 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002061 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002062# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002063 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002064# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002065 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002066# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002067 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002068 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002069 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002070 channel->ch_part[part].ch_poll_idx = nfd;
2071 fds[nfd].fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002072 fds[nfd].events = POLLIN;
2073 nfd++;
2074 }
2075 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002076 channel->ch_part[part].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002077 }
2078 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002079
2080 return nfd;
2081}
2082
2083/*
2084 * The type of "fds" is hidden to avoid problems with the function proto.
2085 */
2086 int
2087channel_poll_check(int ret_in, void *fds_in)
2088{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002089 int ret = ret_in;
2090 channel_T *channel;
2091 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002092 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002093
Bram Moolenaar77073442016-02-13 23:23:53 +01002094 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002095 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002096# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002097 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002098# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002099 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002100# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002101 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002102 int idx = channel->ch_part[part].ch_poll_idx;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002103
2104 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
2105 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002106 channel_read(channel, part, "channel_poll_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002107 --ret;
2108 }
2109 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002110 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002111
2112 return ret;
2113}
Bram Moolenaard04a0202016-01-26 23:30:18 +01002114# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002115
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002116# if (!defined(WIN32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002117/*
2118 * The type of "rfds" is hidden to avoid problems with the function proto.
2119 */
2120 int
2121channel_select_setup(int maxfd_in, void *rfds_in)
2122{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002123 int maxfd = maxfd_in;
2124 channel_T *channel;
2125 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002126 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002127
Bram Moolenaar77073442016-02-13 23:23:53 +01002128 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002129 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002130# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002131 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002132# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002133 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002134# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002135 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002136 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002137
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002138 if (fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002139 {
Bram Moolenaard8070362016-02-15 21:56:54 +01002140 FD_SET((int)fd, rfds);
2141 if (maxfd < (int)fd)
2142 maxfd = (int)fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002143 }
2144 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002145 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002146
2147 return maxfd;
2148}
2149
2150/*
2151 * The type of "rfds" is hidden to avoid problems with the function proto.
2152 */
2153 int
2154channel_select_check(int ret_in, void *rfds_in)
2155{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002156 int ret = ret_in;
2157 channel_T *channel;
2158 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002159 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002160
Bram Moolenaar77073442016-02-13 23:23:53 +01002161 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002162 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002163# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002164 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002165# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002166 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002167# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002168 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002169 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002170
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002171 if (ret > 0 && fd != INVALID_FD && FD_ISSET(fd, rfds))
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002172 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002173 channel_read(channel, part, "channel_select_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002174 --ret;
2175 }
2176 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002177 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002178
2179 return ret;
2180}
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002181# endif /* !WIN32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002182
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002183/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01002184 * Execute queued up commands.
2185 * Invoked from the main loop when it's safe to execute received commands.
2186 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002187 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01002188 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002189channel_parse_messages(void)
2190{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002191 channel_T *channel = first_channel;
2192 int ret = FALSE;
2193 int r;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002194 int part = PART_SOCK;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002195
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002196 while (channel != NULL)
2197 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002198 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaard7ece102016-02-02 23:23:02 +01002199 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002200 /* Increase the refcount, in case the handler causes the channel
2201 * to be unreferenced or closed. */
2202 ++channel->ch_refcount;
2203 r = may_invoke_callback(channel, part);
2204 if (r == OK)
2205 ret = TRUE;
2206 if (channel_unref(channel) || r == OK)
2207 {
2208 /* channel was freed or something was done, start over */
2209 channel = first_channel;
2210 part = PART_SOCK;
2211 continue;
2212 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002213 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002214#ifdef CHANNEL_PIPES
2215 if (part < PART_ERR)
2216 ++part;
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002217 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002218#endif
2219 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002220 channel = channel->ch_next;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002221 part = PART_SOCK;
2222 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002223 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002224 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002225}
2226
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01002227/*
2228 * Mark references to lists used in channels.
2229 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002230 int
2231set_ref_in_channel(int copyID)
2232{
Bram Moolenaar77073442016-02-13 23:23:53 +01002233 int abort = FALSE;
2234 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002235 int part;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002236
Bram Moolenaar77073442016-02-13 23:23:53 +01002237 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002238 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002239#ifdef CHANNEL_PIPES
2240 for (part = PART_SOCK; part < PART_IN; ++part)
2241#else
2242 part = PART_SOCK;
2243#endif
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002244 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002245 jsonq_T *head = &channel->ch_part[part].ch_json_head;
2246 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002247
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002248 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002249 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002250 list_T *l = item->jq_value->vval.v_list;
2251
2252 if (l->lv_copyID != copyID)
2253 {
2254 l->lv_copyID = copyID;
2255 abort = abort || set_ref_in_list(l, copyID, NULL);
2256 }
2257 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002258 }
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002259 }
2260 }
2261 return abort;
2262}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002263
2264/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002265 * Return the "part" to write to for "channel".
2266 */
2267 int
2268channel_part_send(channel_T *channel)
2269{
2270#ifdef CHANNEL_PIPES
2271 if (channel->CH_SOCK_FD == INVALID_FD)
2272 return PART_IN;
2273#endif
2274 return PART_SOCK;
2275}
2276
2277/*
2278 * Return the default "part" to read from for "channel".
2279 */
2280 int
2281channel_part_read(channel_T *channel)
2282{
2283#ifdef CHANNEL_PIPES
2284 if (channel->CH_SOCK_FD == INVALID_FD)
2285 return PART_OUT;
2286#endif
2287 return PART_SOCK;
2288}
2289
2290/*
2291 * Return the mode of "channel"/"part"
Bram Moolenaar77073442016-02-13 23:23:53 +01002292 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002293 */
2294 ch_mode_T
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002295channel_get_mode(channel_T *channel, int part)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002296{
Bram Moolenaar77073442016-02-13 23:23:53 +01002297 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002298 return MODE_JSON;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002299 return channel->ch_part[part].ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002300}
2301
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01002302/*
2303 * Return the timeout of "channel"/"part"
2304 */
2305 int
2306channel_get_timeout(channel_T *channel, int part)
2307{
2308 return channel->ch_part[part].ch_timeout;
2309}
2310
Bram Moolenaare0874f82016-01-24 20:36:41 +01002311#endif /* FEAT_CHANNEL */