blob: aa97088b403f54bdadf71073e9ac56f4025711bf [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
364 static void
365messageFromNetbeans(gpointer clientData,
366 gint unused1 UNUSED,
367 GdkInputCondition unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100368{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100369 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100370}
371#endif
372
373 static void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100374channel_gui_register_one(channel_T *channel, int part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100375{
376# ifdef FEAT_GUI_X11
377 /* Tell notifier we are interested in being called
378 * when there is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100379 if (channel->ch_part[part].ch_inputHandler == (XtInputId)NULL)
380 channel->ch_part[part].ch_inputHandler = XtAppAddInput(
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100381 (XtAppContext)app_context,
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100382 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100383 (XtPointer)(XtInputReadMask + XtInputExceptMask),
384 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100385 (XtPointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100386# else
387# ifdef FEAT_GUI_GTK
388 /* Tell gdk we are interested in being called when there
389 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100390 if (channel->ch_part[part].ch_inputHandler == 0)
391 channel->ch_part[part].ch_inputHandler = gdk_input_add(
392 (gint)channel->ch_part[part].ch_fd,
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100393 (GdkInputCondition)
394 ((int)GDK_INPUT_READ + (int)GDK_INPUT_EXCEPTION),
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100395 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100396 (gpointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100397# else
398# ifdef FEAT_GUI_W32
399 /* Tell Windows we are interested in receiving message when there
400 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100401 if (channel->ch_part[part].ch_inputHandler == -1)
402 channel->ch_part[part].ch_inputHandler = WSAAsyncSelect(
403 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100404 s_hwnd, WM_NETBEANS, FD_READ);
405# endif
406# endif
407# endif
408}
409
410 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100411channel_gui_register(channel_T *channel)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100412{
Bram Moolenaard04a0202016-01-26 23:30:18 +0100413 if (!CH_HAS_GUI)
414 return;
415
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100416 if (channel->CH_SOCK_FD != INVALID_FD)
417 channel_gui_register_one(channel, PART_SOCK);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100418# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100419 if (channel->CH_OUT_FD != INVALID_FD)
420 channel_gui_register_one(channel, PART_OUT);
421 if (channel->CH_ERR_FD != INVALID_FD)
422 channel_gui_register_one(channel, PART_ERR);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100423# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100424}
425
426/*
427 * Register any of our file descriptors with the GUI event handling system.
428 * Called when the GUI has started.
429 */
430 void
431channel_gui_register_all(void)
432{
Bram Moolenaar77073442016-02-13 23:23:53 +0100433 channel_T *channel;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100434
Bram Moolenaar77073442016-02-13 23:23:53 +0100435 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100436 channel_gui_register(channel);
437}
438
439 static void
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100440channel_gui_unregister(channel_T *channel)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100441{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100442 int part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100443
444#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100445 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100446#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100447 part = PART_SOCK;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100448#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100449 {
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100450# ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100451 if (channel->ch_part[part].ch_inputHandler != (XtInputId)NULL)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100452 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100453 XtRemoveInput(channel->ch_part[part].ch_inputHandler);
454 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100455 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100456# else
457# ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100458 if (channel->ch_part[part].ch_inputHandler != 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100459 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100460 gdk_input_remove(channel->ch_part[part].ch_inputHandler);
461 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100462 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100463# else
464# ifdef FEAT_GUI_W32
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100465 if (channel->ch_part[part].ch_inputHandler == 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100466 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100467 WSAAsyncSelect(channel->ch_part[part].ch_fd, s_hwnd, 0, 0);
468 channel->ch_part[part].ch_inputHandler = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100469 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100470# endif
471# endif
472# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100473 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100474}
475
476#endif
477
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100478static char *e_cannot_connect = N_("E902: Cannot connect to port");
479
Bram Moolenaard04a0202016-01-26 23:30:18 +0100480/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100481 * Open a socket channel to "hostname":"port".
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100482 * "waittime" is the time in msec to wait for the connection.
483 * When negative wait forever.
Bram Moolenaar77073442016-02-13 23:23:53 +0100484 * Returns the channel for success.
485 * Returns NULL for failure.
Bram Moolenaard04a0202016-01-26 23:30:18 +0100486 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100487 channel_T *
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100488channel_open(char *hostname, int port_in, int waittime, void (*close_cb)(void))
Bram Moolenaard04a0202016-01-26 23:30:18 +0100489{
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100490 int sd = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100491 struct sockaddr_in server;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100492 struct hostent *host;
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100493#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100494 u_short port = port_in;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100495 u_long val = 1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100496#else
497 int port = port_in;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100498 struct timeval start_tv;
499 int so_error;
500 socklen_t so_error_len = sizeof(so_error);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100501#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100502 channel_T *channel;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100503 int ret;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100504
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100505#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100506 channel_init_winsock();
507#endif
508
Bram Moolenaar77073442016-02-13 23:23:53 +0100509 channel = add_channel();
510 if (channel == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100511 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100512 ch_error(NULL, "Cannot allocate channel.");
Bram Moolenaar77073442016-02-13 23:23:53 +0100513 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100514 }
515
516 /* Get the server internet address and put into addr structure */
517 /* fill in the socket address structure and connect to server */
518 vim_memset((char *)&server, 0, sizeof(server));
519 server.sin_family = AF_INET;
520 server.sin_port = htons(port);
521 if ((host = gethostbyname(hostname)) == NULL)
522 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100523 ch_error(channel, "in gethostbyname() in channel_open()");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100524 PERROR("E901: gethostbyname() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100525 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100526 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100527 }
528 memcpy((char *)&server.sin_addr, host->h_addr, host->h_length);
529
Bram Moolenaar254e00d2016-02-19 23:23:12 +0100530 /* On Mac and Solaris a zero timeout almost never works. At least wait
531 * one millisecond. Let's do it for all systems, because we don't know why
532 * this is needed. */
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100533 if (waittime == 0)
534 waittime = 1;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100535
536 /*
537 * For Unix we need to call connect() again after connect() failed.
538 * On Win32 one time is sufficient.
539 */
540 while (TRUE)
541 {
542 if (sd >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100543 sock_close(sd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100544 sd = socket(AF_INET, SOCK_STREAM, 0);
545 if (sd == -1)
546 {
547 ch_error(channel, "in socket() in channel_open().");
548 PERROR("E898: socket() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100549 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100550 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100551 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100552
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100553 if (waittime >= 0)
554 {
555 /* Make connect() non-blocking. */
556 if (
557#ifdef _WIN32
558 ioctlsocket(sd, FIONBIO, &val) < 0
559#else
560 fcntl(sd, F_SETFL, O_NONBLOCK) < 0
561#endif
562 )
563 {
564 SOCK_ERRNO;
565 ch_errorn(channel,
566 "channel_open: Connect failed with errno %d", errno);
567 sock_close(sd);
568 channel_free(channel);
569 return NULL;
570 }
571 }
572
573 /* Try connecting to the server. */
574 ch_logsn(channel, "Connecting to %s port %d", hostname, port);
575 ret = connect(sd, (struct sockaddr *)&server, sizeof(server));
576
577 SOCK_ERRNO;
578 if (ret < 0)
579 {
580 if (errno != EWOULDBLOCK
581 && errno != ECONNREFUSED
582
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100583#ifdef EINPROGRESS
584 && errno != EINPROGRESS
585#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100586 )
587 {
588 ch_errorn(channel,
589 "channel_open: Connect failed with errno %d", errno);
590 PERROR(_(e_cannot_connect));
591 sock_close(sd);
592 channel_free(channel);
593 return NULL;
594 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100595 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100596
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100597 /* If we don't block and connect() failed then try using select() to
598 * wait for the connection to be made. */
599 if (waittime >= 0 && ret < 0)
600 {
601 struct timeval tv;
602 fd_set wfds;
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100603#if defined(__APPLE__) && __APPLE__ == 1
604# define PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100605 fd_set rfds;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100606
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100607 FD_ZERO(&rfds);
608 FD_SET(sd, &rfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100609#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100610 FD_ZERO(&wfds);
611 FD_SET(sd, &wfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100612
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100613 tv.tv_sec = waittime / 1000;
614 tv.tv_usec = (waittime % 1000) * 1000;
615#ifndef WIN32
616 gettimeofday(&start_tv, NULL);
617#endif
618 ch_logn(channel,
619 "Waiting for connection (waittime %d msec)...", waittime);
620 ret = select((int)sd + 1,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100621#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100622 &rfds,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100623#else
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100624 NULL,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100625#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100626 &wfds, NULL, &tv);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100627
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100628 if (ret < 0)
629 {
630 SOCK_ERRNO;
631 ch_errorn(channel,
632 "channel_open: Connect failed with errno %d", errno);
633 PERROR(_(e_cannot_connect));
634 sock_close(sd);
635 channel_free(channel);
636 return NULL;
637 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100638#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100639 if (ret == 0 && FD_ISSET(sd, &rfds) && FD_ISSET(sd, &wfds))
640 {
641 /* For OS X, this implies error. See tcp(4). */
642 ch_error(channel, "channel_open: Connect failed");
643 EMSG(_(e_cannot_connect));
644 sock_close(sd);
645 channel_free(channel);
646 return NULL;
647 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100648#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100649#ifdef WIN32
650 /* On Win32 select() is expected to work and wait for up to the
651 * waittime for the socket to be open. */
652 if (!FD_ISSET(sd, &wfds) || ret == 0)
653#else
654 /* See socket(7) for the behavior on Linux-like systems:
655 * After putting the socket in non-blocking mode, connect() will
656 * return EINPROGRESS, select() will not wait (as if writing is
657 * possible), need to use getsockopt() to check if the socket is
658 * actually open. */
659 getsockopt(sd, SOL_SOCKET, SO_ERROR, &so_error, &so_error_len);
660 if (!FD_ISSET(sd, &wfds) || ret == 0 || so_error != 0)
661#endif
662 {
663#ifndef WIN32
664 struct timeval end_tv;
665 long elapsed_msec;
666
667 gettimeofday(&end_tv, NULL);
668 elapsed_msec = (end_tv.tv_sec - start_tv.tv_sec) * 1000
669 + (end_tv.tv_usec - start_tv.tv_usec) / 1000;
670 if (waittime > 1 && elapsed_msec < waittime)
671 {
672 /* The port isn't ready but we also didn't get an error.
673 * This happens when the server didn't open the socket
674 * yet. Wait a bit and try again. */
675 mch_delay(waittime < 50 ? (long)waittime : 50L, TRUE);
676 ui_breakcheck();
677 if (!got_int)
678 {
679 /* reduce the waittime by the elapsed time and the 50
680 * msec delay (or a bit more) */
681 waittime -= elapsed_msec;
682 if (waittime > 50)
683 waittime -= 50;
684 else
685 waittime = 1;
686 continue;
687 }
688 /* we were interrupted, behave as if timed out */
689 }
690#endif
691 /* We timed out. */
692 ch_error(channel, "Connection timed out");
693 sock_close(sd);
694 channel_free(channel);
695 return NULL;
696 }
697
698 ch_log(channel, "Connection made");
699 break;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100700 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100701 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100702
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100703 if (waittime >= 0)
704 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100705#ifdef _WIN32
706 val = 0;
707 ioctlsocket(sd, FIONBIO, &val);
708#else
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100709 (void)fcntl(sd, F_SETFL, 0);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100710#endif
711 }
712
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100713 channel->CH_SOCK_FD = (sock_T)sd;
Bram Moolenaar77073442016-02-13 23:23:53 +0100714 channel->ch_close_cb = close_cb;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100715
716#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +0100717 channel_gui_register(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100718#endif
719
Bram Moolenaar77073442016-02-13 23:23:53 +0100720 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100721}
722
723#if defined(CHANNEL_PIPES) || defined(PROTO)
724 void
Bram Moolenaard8070362016-02-15 21:56:54 +0100725channel_set_pipes(channel_T *channel, sock_T in, sock_T out, sock_T err)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100726{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100727 channel->CH_IN_FD = in;
728 channel->CH_OUT_FD = out;
729 channel->CH_ERR_FD = err;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100730}
731#endif
732
733 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100734channel_set_job(channel_T *channel, job_T *job)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100735{
Bram Moolenaar77073442016-02-13 23:23:53 +0100736 channel->ch_job = job;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100737}
738
739/*
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100740 * Set various properties from an "opt" argument.
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100741 */
742 void
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100743channel_set_options(channel_T *channel, jobopt_T *opt)
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100744{
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100745 int part;
746 char_u **cbp;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100747
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100748 if (opt->jo_set & JO_MODE)
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100749 for (part = PART_SOCK; part <= PART_IN; ++part)
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100750 channel->ch_part[part].ch_mode = opt->jo_mode;
751 if (opt->jo_set & JO_IN_MODE)
752 channel->ch_part[PART_IN].ch_mode = opt->jo_in_mode;
753 if (opt->jo_set & JO_OUT_MODE)
754 channel->ch_part[PART_OUT].ch_mode = opt->jo_out_mode;
755 if (opt->jo_set & JO_ERR_MODE)
756 channel->ch_part[PART_ERR].ch_mode = opt->jo_err_mode;
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100757
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100758 if (opt->jo_set & JO_TIMEOUT)
759 for (part = PART_SOCK; part <= PART_IN; ++part)
760 channel->ch_part[part].ch_timeout = opt->jo_timeout;
761 if (opt->jo_set & JO_OUT_TIMEOUT)
762 channel->ch_part[PART_OUT].ch_timeout = opt->jo_out_timeout;
763 if (opt->jo_set & JO_ERR_TIMEOUT)
764 channel->ch_part[PART_ERR].ch_timeout = opt->jo_err_timeout;
765
766 if (opt->jo_set & JO_CALLBACK)
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100767 {
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100768 cbp = &channel->ch_callback;
769 vim_free(*cbp);
770 if (opt->jo_callback != NULL && *opt->jo_callback != NUL)
771 *cbp = vim_strsave(opt->jo_callback);
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100772 else
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100773 *cbp = NULL;
774 }
775 if (opt->jo_set & JO_OUT_CALLBACK)
776 {
777 cbp = &channel->ch_part[PART_OUT].ch_callback;
778 vim_free(*cbp);
779 if (opt->jo_out_cb != NULL && *opt->jo_out_cb != NUL)
780 *cbp = vim_strsave(opt->jo_out_cb);
781 else
782 *cbp = NULL;
783 }
784 if (opt->jo_set & JO_ERR_CALLBACK)
785 {
786 cbp = &channel->ch_part[PART_ERR].ch_callback;
787 vim_free(*cbp);
788 if (opt->jo_err_cb != NULL && *opt->jo_err_cb != NUL)
789 *cbp = vim_strsave(opt->jo_err_cb);
790 else
791 *cbp = NULL;
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100792 }
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100793}
794
795/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100796 * Set the callback for "channel"/"part" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100797 */
798 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100799channel_set_req_callback(
800 channel_T *channel,
801 int part,
802 char_u *callback,
803 int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100804{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100805 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100806 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
807
808 if (item != NULL)
809 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100810 item->cq_callback = vim_strsave(callback);
811 item->cq_seq_nr = id;
812 item->cq_prev = head->cq_prev;
813 head->cq_prev = item;
814 item->cq_next = NULL;
815 if (item->cq_prev == NULL)
816 head->cq_next = item;
817 else
818 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100819 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100820}
821
822/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100823 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100824 */
825 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100826invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100827{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100828 typval_T rettv;
829 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100830
Bram Moolenaar77073442016-02-13 23:23:53 +0100831 argv[0].v_type = VAR_CHANNEL;
832 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100833
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100834 call_func(callback, (int)STRLEN(callback),
835 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
Bram Moolenaaree1cffc2016-02-21 19:14:41 +0100836 clear_tv(&rettv);
837
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100838 /* If an echo command was used the cursor needs to be put back where
839 * it belongs. */
840 setcursor();
841 cursor_on();
842 out_flush();
843}
844
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100845/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100846 * Return the first buffer from channel "channel"/"part" and remove it.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100847 * The caller must free it.
848 * Returns NULL if there is nothing.
849 */
850 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100851channel_get(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100852{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100853 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100854 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100855 char_u *p;
856
Bram Moolenaar77073442016-02-13 23:23:53 +0100857 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100858 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100859 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100860 p = node->rq_buffer;
861 head->rq_next = node->rq_next;
862 if (node->rq_next == NULL)
863 head->rq_prev = NULL;
864 else
865 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100866 vim_free(node);
867 return p;
868}
869
870/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100871 * Returns the whole buffer contents concatenated for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100872 */
873 static char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100874channel_get_all(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100875{
876 /* Concatenate everything into one buffer.
877 * TODO: avoid multiple allocations. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100878 while (channel_collapse(channel, part) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100879 ;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100880 return channel_get(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100881}
882
883/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100884 * Collapses the first and second buffer for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100885 * Returns FAIL if that is not possible.
886 */
887 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100888channel_collapse(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100889{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100890 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100891 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100892 char_u *p;
893
Bram Moolenaar77073442016-02-13 23:23:53 +0100894 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100895 return FAIL;
896
Bram Moolenaar77073442016-02-13 23:23:53 +0100897 p = alloc((unsigned)(STRLEN(node->rq_buffer)
898 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100899 if (p == NULL)
900 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100901 STRCPY(p, node->rq_buffer);
902 STRCAT(p, node->rq_next->rq_buffer);
903 vim_free(node->rq_next->rq_buffer);
904 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100905
Bram Moolenaar77073442016-02-13 23:23:53 +0100906 /* dispose of the node and its buffer */
907 head->rq_next = node->rq_next;
908 head->rq_next->rq_prev = NULL;
909 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100910 vim_free(node);
911 return OK;
912}
913
914/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100915 * Store "buf[len]" on "channel"/"part".
916 * Returns OK or FAIL.
917 */
918 static int
919channel_save(channel_T *channel, int part, char_u *buf, int len)
920{
921 readq_T *node;
922 readq_T *head = &channel->ch_part[part].ch_head;
923 char_u *p;
924 int i;
925
926 node = (readq_T *)alloc(sizeof(readq_T));
927 if (node == NULL)
928 return FAIL; /* out of memory */
929 node->rq_buffer = alloc(len + 1);
930 if (node->rq_buffer == NULL)
931 {
932 vim_free(node);
933 return FAIL; /* out of memory */
934 }
935
936 if (channel->ch_part[part].ch_mode == MODE_NL)
937 {
938 /* Drop any CR before a NL. */
939 p = node->rq_buffer;
940 for (i = 0; i < len; ++i)
941 if (buf[i] != CAR || i + 1 >= len || buf[i + 1] != NL)
942 *p++ = buf[i];
943 *p = NUL;
944 }
945 else
946 {
947 mch_memmove(node->rq_buffer, buf, len);
948 node->rq_buffer[len] = NUL;
949 }
950
951 /* append node to the tail of the queue */
952 node->rq_next = NULL;
953 node->rq_prev = head->rq_prev;
954 if (head->rq_prev == NULL)
955 head->rq_next = node;
956 else
957 head->rq_prev->rq_next = node;
958 head->rq_prev = node;
959
960 if (log_fd != NULL)
961 {
962 ch_log_lead("RECV ", channel);
963 fprintf(log_fd, "'");
964 if (fwrite(buf, len, 1, log_fd) != 1)
965 return FAIL;
966 fprintf(log_fd, "'\n");
967 }
968 return OK;
969}
970
971/*
972 * Use the read buffer of "channel"/"part" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100973 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +0100974 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100975 */
Bram Moolenaard7ece102016-02-02 23:23:02 +0100976 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100977channel_parse_json(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100978{
979 js_read_T reader;
980 typval_T listtv;
981 jsonq_T *item;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100982 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100983 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100984
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100985 if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100986 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100987
988 /* TODO: make reader work properly */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100989 /* reader.js_buf = channel_peek(channel, part); */
990 reader.js_buf = channel_get_all(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100991 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +0100992 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100993 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +0100994 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100995 ret = json_decode(&reader, &listtv,
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100996 channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100997 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100998 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +0100999 /* Only accept the response when it is a list with at least two
1000 * items. */
1001 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001002 {
1003 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001004 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001005 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001006 else
1007 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001008 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
1009 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001010 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001011 else
1012 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001013 item->jq_value = alloc_tv();
1014 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001015 {
1016 vim_free(item);
1017 clear_tv(&listtv);
1018 }
1019 else
1020 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001021 *item->jq_value = listtv;
1022 item->jq_prev = head->jq_prev;
1023 head->jq_prev = item;
1024 item->jq_next = NULL;
1025 if (item->jq_prev == NULL)
1026 head->jq_next = item;
1027 else
1028 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001029 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001030 }
1031 }
1032 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001033
1034 /* Put the unread part back into the channel.
1035 * TODO: insert in front */
1036 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001037 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001038 channel_save(channel, part, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001039 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001040 ret = TRUE;
1041 }
1042 else
1043 ret = FALSE;
1044
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001045 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001046 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001047}
1048
1049/*
Bram Moolenaard46ae142016-02-16 13:33:52 +01001050 * Remove "node" from the queue that it is in. Does not free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +01001051 */
1052 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001053remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001054{
Bram Moolenaar77073442016-02-13 23:23:53 +01001055 if (node->cq_prev == NULL)
1056 head->cq_next = node->cq_next;
1057 else
1058 node->cq_prev->cq_next = node->cq_next;
1059 if (node->cq_next == NULL)
1060 head->cq_prev = node->cq_prev;
1061 else
1062 node->cq_next->cq_prev = node->cq_prev;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001063}
1064
1065/*
1066 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +01001067 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001068 */
1069 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001070remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001071{
Bram Moolenaar77073442016-02-13 23:23:53 +01001072 if (node->jq_prev == NULL)
1073 head->jq_next = node->jq_next;
1074 else
1075 node->jq_prev->jq_next = node->jq_next;
1076 if (node->jq_next == NULL)
1077 head->jq_prev = node->jq_prev;
1078 else
1079 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001080 vim_free(node);
1081}
1082
1083/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001084 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001085 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +01001086 * When "id" is zero or negative jut get the first message. But not the one
1087 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001088 * Return OK when found and return the value in "rettv".
1089 * Return FAIL otherwise.
1090 */
1091 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001092channel_get_json(channel_T *channel, int part, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001093{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001094 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001095 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001096
Bram Moolenaar77073442016-02-13 23:23:53 +01001097 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001098 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001099 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001100 typval_T *tv = &l->lv_first->li_tv;
1101
1102 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001103 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001104 || tv->vval.v_number == 0
1105 || tv->vval.v_number != channel->ch_part[part].ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001106 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001107 *rettv = item->jq_value;
1108 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001109 return OK;
1110 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001111 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001112 }
1113 return FAIL;
1114}
1115
Bram Moolenaarece61b02016-02-20 21:39:05 +01001116#define CH_JSON_MAX_ARGS 4
1117
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001118/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001119 * Execute a command received over "channel"/"part"
Bram Moolenaarece61b02016-02-20 21:39:05 +01001120 * "argv[0]" is the command string.
1121 * "argv[1]" etc. have further arguments, type is VAR_UNKNOWN if missing.
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001122 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001123 static void
Bram Moolenaarece61b02016-02-20 21:39:05 +01001124channel_exe_cmd(channel_T *channel, int part, typval_T *argv)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001125{
Bram Moolenaarece61b02016-02-20 21:39:05 +01001126 char_u *cmd = argv[0].vval.v_string;
1127 char_u *arg;
1128 int options = channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001129
Bram Moolenaarece61b02016-02-20 21:39:05 +01001130 if (argv[1].v_type != VAR_STRING)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001131 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001132 ch_error(channel, "received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001133 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001134 EMSG("E903: received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001135 return;
1136 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001137 arg = argv[1].vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001138 if (arg == NULL)
1139 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001140
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001141 if (STRCMP(cmd, "ex") == 0)
1142 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001143 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001144 }
1145 else if (STRCMP(cmd, "normal") == 0)
1146 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001147 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001148
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001149 ea.arg = arg;
1150 ea.addr_count = 0;
1151 ea.forceit = TRUE; /* no mapping */
1152 ex_normal(&ea);
1153 }
1154 else if (STRCMP(cmd, "redraw") == 0)
1155 {
1156 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001157
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001158 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001159 ex_redraw(&ea);
1160 showruler(FALSE);
1161 setcursor();
1162 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001163#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001164 if (gui.in_use)
1165 {
1166 gui_update_cursor(FALSE, FALSE);
1167 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001168 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001169#endif
1170 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001171 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "call") == 0)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001172 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001173 int is_call = cmd[0] == 'c';
1174 int id_idx = is_call ? 3 : 2;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001175
Bram Moolenaarece61b02016-02-20 21:39:05 +01001176 if (argv[id_idx].v_type != VAR_UNKNOWN
1177 && argv[id_idx].v_type != VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001178 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001179 ch_error(channel, "last argument for expr/call must be a number");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001180 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001181 EMSG("E904: last argument for expr/call must be a number");
1182 }
1183 else if (is_call && argv[2].v_type != VAR_LIST)
1184 {
1185 ch_error(channel, "third argument for call must be a list");
1186 if (p_verbose > 2)
1187 EMSG("E904: third argument for call must be a list");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001188 }
1189 else
1190 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001191 typval_T *tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001192 typval_T res_tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001193 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +01001194 char_u *json = NULL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001195
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001196 /* Don't pollute the display with errors. */
1197 ++emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001198 if (!is_call)
1199 tv = eval_expr(arg, NULL);
1200 else if (func_call(arg, &argv[2], NULL, &res_tv) == OK)
1201 tv = &res_tv;
1202 else
1203 tv = NULL;
1204
1205 if (argv[id_idx].v_type == VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001206 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001207 int id = argv[id_idx].vval.v_number;
1208
Bram Moolenaar55fab432016-02-07 16:53:13 +01001209 if (tv != NULL)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001210 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001211 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001212 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001213 /* If evaluation failed or the result can't be encoded
1214 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001215 vim_free(json);
1216 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001217 err_tv.v_type = VAR_STRING;
1218 err_tv.vval.v_string = (char_u *)"ERROR";
1219 tv = &err_tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001220 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001221 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001222 if (json != NULL)
1223 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001224 channel_send(channel,
1225 part == PART_SOCK ? PART_SOCK : PART_IN,
1226 json, (char *)cmd);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001227 vim_free(json);
1228 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001229 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001230 --emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001231 if (tv == &res_tv)
1232 clear_tv(tv);
1233 else if (tv != &err_tv)
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001234 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001235 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001236 }
1237 else if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001238 {
1239 ch_errors(channel, "Receved unknown command: %s", (char *)cmd);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001240 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001241 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001242}
1243
1244/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001245 * Invoke a callback for "channel"/"part" if needed.
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001246 * Return TRUE when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001247 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001248 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001249may_invoke_callback(channel_T *channel, int part)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001250{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001251 char_u *msg = NULL;
1252 typval_T *listtv = NULL;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001253 typval_T argv[CH_JSON_MAX_ARGS];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001254 int seq_nr = -1;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001255 ch_mode_T ch_mode = channel->ch_part[part].ch_mode;
1256 char_u *callback = NULL;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001257
Bram Moolenaara07fec92016-02-05 21:04:08 +01001258 if (channel->ch_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001259 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001260 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001261
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001262 if (channel->ch_part[part].ch_callback != NULL)
1263 callback = channel->ch_part[part].ch_callback;
1264 else
1265 callback = channel->ch_callback;
1266
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001267 if (ch_mode == MODE_JSON || ch_mode == MODE_JS)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001268 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001269 listitem_T *item;
1270 int argc = 0;
1271
Bram Moolenaard7ece102016-02-02 23:23:02 +01001272 /* Get any json message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001273 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001274 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001275 /* Parse readahead, return when there is still no message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001276 channel_parse_json(channel, part);
1277 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001278 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001279 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001280
Bram Moolenaarece61b02016-02-20 21:39:05 +01001281 for (item = listtv->vval.v_list->lv_first;
1282 item != NULL && argc < CH_JSON_MAX_ARGS;
1283 item = item->li_next)
1284 argv[argc++] = item->li_tv;
1285 while (argc < CH_JSON_MAX_ARGS)
1286 argv[argc++].v_type = VAR_UNKNOWN;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001287
Bram Moolenaarece61b02016-02-20 21:39:05 +01001288 if (argv[0].v_type == VAR_STRING)
1289 {
1290 char_u *cmd = argv[0].vval.v_string;
1291
1292 /* ["cmd", arg] or ["cmd", arg, arg] or ["cmd", arg, arg, arg] */
Bram Moolenaar77073442016-02-13 23:23:53 +01001293 ch_logs(channel, "Executing %s command", (char *)cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001294 channel_exe_cmd(channel, part, argv);
Bram Moolenaar77073442016-02-13 23:23:53 +01001295 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001296 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001297 }
1298
Bram Moolenaarece61b02016-02-20 21:39:05 +01001299 if (argv[0].v_type != VAR_NUMBER)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001300 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001301 ch_error(channel,
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001302 "Dropping message with invalid sequence number type");
Bram Moolenaar77073442016-02-13 23:23:53 +01001303 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001304 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001305 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001306 seq_nr = argv[0].vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001307 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001308 else if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001309 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001310 /* nothing to read on RAW or NL channel */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001311 return FALSE;
1312 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001313 else
1314 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001315 /* If there is no callback drop the message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001316 if (callback == NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001317 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001318 while ((msg = channel_get(channel, part)) != NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001319 vim_free(msg);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001320 return FALSE;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001321 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001322
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001323 if (ch_mode == MODE_NL)
1324 {
1325 char_u *nl;
1326 char_u *buf;
1327
1328 /* See if we have a message ending in NL in the first buffer. If
1329 * not try to concatenate the first and the second buffer. */
1330 while (TRUE)
1331 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001332 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001333 nl = vim_strchr(buf, NL);
1334 if (nl != NULL)
1335 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001336 if (channel_collapse(channel, part) == FAIL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001337 return FALSE; /* incomplete message */
1338 }
1339 if (nl[1] == NUL)
1340 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001341 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001342 else
1343 {
1344 /* Copy the message into allocated memory and remove it from
1345 * the buffer. */
1346 msg = vim_strnsave(buf, (int)(nl - buf));
1347 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1348 }
1349 }
1350 else
1351 /* For a raw channel we don't know where the message ends, just
1352 * get everything we have. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001353 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001354
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001355 argv[1].v_type = VAR_STRING;
1356 argv[1].vval.v_string = msg;
1357 }
1358
Bram Moolenaara07fec92016-02-05 21:04:08 +01001359 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001360 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001361 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001362 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001363 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001364
1365 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001366 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001367 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001368 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001369 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001370 ch_log(channel, "Invoking one-time callback");
Bram Moolenaard46ae142016-02-16 13:33:52 +01001371 /* Remove the item from the list first, if the callback
1372 * invokes ch_close() the list will be cleared. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001373 remove_cb_node(head, item);
Bram Moolenaard46ae142016-02-16 13:33:52 +01001374 invoke_callback(channel, item->cq_callback, argv);
1375 vim_free(item->cq_callback);
1376 vim_free(item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001377 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001378 break;
1379 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001380 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001381 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001382 if (!done)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001383 ch_log(channel, "Dropping message without callback");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001384 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001385 else if (callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001386 {
1387 /* invoke the channel callback */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001388 ch_log(channel, "Invoking channel callback");
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001389 invoke_callback(channel, callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001390 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001391 else
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001392 ch_log(channel, "Dropping message");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001393
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001394 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001395 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001396 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001397
1398 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001399}
1400
1401/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001402 * Return TRUE when channel "channel" is open for writing to.
1403 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001404 */
1405 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001406channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001407{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001408 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001409#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001410 || channel->CH_IN_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001411#endif
1412 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001413}
1414
1415/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001416 * Return TRUE when channel "channel" is open for reading or writing.
1417 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001418 */
1419 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001420channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001421{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001422 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001423#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001424 || channel->CH_IN_FD != INVALID_FD
1425 || channel->CH_OUT_FD != INVALID_FD
1426 || channel->CH_ERR_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001427#endif
1428 );
1429}
1430
1431/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001432 * Return a string indicating the status of the channel.
1433 */
1434 char *
1435channel_status(channel_T *channel)
1436{
1437 if (channel == NULL)
1438 return "fail";
1439 if (channel_is_open(channel))
1440 return "open";
1441 return "closed";
1442}
1443
1444/*
1445 * Close channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001446 * This does not trigger the close callback.
1447 */
1448 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001449channel_close(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001450{
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001451 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001452
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001453#ifdef FEAT_GUI
1454 channel_gui_unregister(channel);
1455#endif
1456
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001457 if (channel->CH_SOCK_FD != INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001458 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001459 sock_close(channel->CH_SOCK_FD);
1460 channel->CH_SOCK_FD = INVALID_FD;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001461 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001462#if defined(CHANNEL_PIPES)
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001463 if (channel->CH_IN_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001464 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001465 fd_close(channel->CH_IN_FD);
1466 channel->CH_IN_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001467 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001468 if (channel->CH_OUT_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001469 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001470 fd_close(channel->CH_OUT_FD);
1471 channel->CH_OUT_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001472 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001473 if (channel->CH_ERR_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001474 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001475 fd_close(channel->CH_ERR_FD);
1476 channel->CH_ERR_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001477 }
1478#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001479
1480 channel->ch_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001481 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001482}
1483
Bram Moolenaard04a0202016-01-26 23:30:18 +01001484/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001485 * Return the first buffer from "channel"/"part" without removing it.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001486 * Returns NULL if there is nothing.
1487 */
1488 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001489channel_peek(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001490{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001491 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001492
Bram Moolenaar77073442016-02-13 23:23:53 +01001493 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001494 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001495 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001496}
1497
1498/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001499 * Clear the read buffer on "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001500 */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001501 static void
1502channel_clear_one(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001503{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001504 jsonq_T *json_head = &channel->ch_part[part].ch_json_head;
1505 cbq_T *cb_head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001506
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001507 while (channel_peek(channel, part) != NULL)
1508 vim_free(channel_get(channel, part));
Bram Moolenaar77073442016-02-13 23:23:53 +01001509
1510 while (cb_head->cq_next != NULL)
Bram Moolenaard46ae142016-02-16 13:33:52 +01001511 {
1512 cbq_T *node = cb_head->cq_next;
1513
1514 remove_cb_node(cb_head, node);
1515 vim_free(node->cq_callback);
1516 vim_free(node);
1517 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001518
1519 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001520 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001521 free_tv(json_head->jq_next->jq_value);
1522 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001523 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001524
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001525 vim_free(channel->ch_part[part].ch_callback);
1526 channel->ch_part[part].ch_callback = NULL;
1527}
1528
1529/*
1530 * Clear all the read buffers on "channel".
1531 */
1532 void
1533channel_clear(channel_T *channel)
1534{
1535 channel_clear_one(channel, PART_SOCK);
1536#ifdef CHANNEL_PIPES
1537 channel_clear_one(channel, PART_OUT);
1538 channel_clear_one(channel, PART_ERR);
1539#endif
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001540 vim_free(channel->ch_callback);
1541 channel->ch_callback = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001542}
1543
Bram Moolenaar77073442016-02-13 23:23:53 +01001544#if defined(EXITFREE) || defined(PROTO)
1545 void
1546channel_free_all(void)
1547{
1548 channel_T *channel;
1549
1550 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1551 channel_clear(channel);
1552}
1553#endif
1554
1555
Bram Moolenaard04a0202016-01-26 23:30:18 +01001556/* Sent when the channel is found closed when reading. */
1557#define DETACH_MSG "\"DETACH\"\n"
1558
1559/* Buffer size for reading incoming messages. */
1560#define MAXMSGSIZE 4096
1561
1562/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001563 * Check for reading from "fd" with "timeout" msec.
1564 * Return FAIL when there is nothing to read.
Bram Moolenaara8343c12016-02-04 22:09:48 +01001565 * Always returns OK for FEAT_GUI_W32.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001566 */
1567 static int
Bram Moolenaard8070362016-02-15 21:56:54 +01001568channel_wait(channel_T *channel, sock_T fd, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001569{
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001570 if (timeout > 0)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001571 ch_logn(channel, "Waiting for up to %d msec", timeout);
Bram Moolenaard8070362016-02-15 21:56:54 +01001572
Bram Moolenaard8070362016-02-15 21:56:54 +01001573# ifdef WIN32
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001574 if (fd != channel->CH_SOCK_FD)
Bram Moolenaard8070362016-02-15 21:56:54 +01001575 {
1576 DWORD nread;
1577 int diff;
1578 DWORD deadline = GetTickCount() + timeout;
1579
1580 /* reading from a pipe, not a socket */
1581 while (TRUE)
1582 {
Bram Moolenaare74e8e72016-02-16 22:01:30 +01001583 if (PeekNamedPipe((HANDLE)fd, NULL, 0, NULL, &nread, NULL)
1584 && nread > 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001585 return OK;
1586 diff = deadline - GetTickCount();
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001587 if (diff <= 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001588 break;
1589 /* Wait for 5 msec.
1590 * TODO: increase the sleep time when looping more often */
1591 Sleep(5);
1592 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001593 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001594 else
Bram Moolenaard8070362016-02-15 21:56:54 +01001595#endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001596 {
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001597#if defined(FEAT_GUI_W32)
1598 /* Can't check socket for Win32 GUI, always return OK. */
1599 ch_log(channel, "Can't check, assuming there is something to read");
1600 return OK;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001601#else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001602# if defined(HAVE_SELECT)
1603 struct timeval tval;
1604 fd_set rfds;
1605 int ret;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001606
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001607 FD_ZERO(&rfds);
1608 FD_SET((int)fd, &rfds);
1609 tval.tv_sec = timeout / 1000;
1610 tval.tv_usec = (timeout % 1000) * 1000;
1611 for (;;)
1612 {
1613 ret = select((int)fd + 1, &rfds, NULL, NULL, &tval);
1614# ifdef EINTR
1615 SOCK_ERRNO;
1616 if (ret == -1 && errno == EINTR)
1617 continue;
1618# endif
1619 if (ret > 0)
1620 return OK;
1621 break;
1622 }
1623# else
1624 struct pollfd fds;
1625
1626 fds.fd = fd;
1627 fds.events = POLLIN;
1628 if (poll(&fds, 1, timeout) > 0)
1629 return OK;
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001630# endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001631#endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001632 }
1633 ch_log(channel, "Nothing to read");
1634 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001635}
1636
1637/*
1638 * Return a unique ID to be used in a message.
1639 */
1640 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001641channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001642{
1643 static int next_id = 1;
1644
1645 return next_id++;
1646}
1647
1648/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001649 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001650 * "part" is PART_SOCK, PART_OUT or PART_ERR.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001651 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001652 */
1653 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001654channel_read(channel_T *channel, int part, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001655{
1656 static char_u *buf = NULL;
1657 int len = 0;
1658 int readlen = 0;
Bram Moolenaard8070362016-02-15 21:56:54 +01001659 sock_T fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001660 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001661
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001662 fd = channel->ch_part[part].ch_fd;
1663 if (fd == INVALID_FD)
1664 {
1665 ch_error(channel, "channel_read() called while socket is closed");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001666 return;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001667 }
1668 use_socket = fd == channel->CH_SOCK_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001669
1670 /* Allocate a buffer to read into. */
1671 if (buf == NULL)
1672 {
1673 buf = alloc(MAXMSGSIZE);
1674 if (buf == NULL)
1675 return; /* out of memory! */
1676 }
1677
1678 /* Keep on reading for as long as there is something to read.
1679 * Use select() or poll() to avoid blocking on a message that is exactly
1680 * MAXMSGSIZE long. */
1681 for (;;)
1682 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001683 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001684 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001685 if (use_socket)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001686 len = sock_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001687 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001688 len = fd_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001689 if (len <= 0)
1690 break; /* error or nothing more to read */
1691
1692 /* Store the read message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001693 channel_save(channel, part, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001694 readlen += len;
1695 if (len < MAXMSGSIZE)
1696 break; /* did read everything that's available */
1697 }
Bram Moolenaara8343c12016-02-04 22:09:48 +01001698#ifdef FEAT_GUI_W32
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001699 if (use_socket && len == SOCKET_ERROR)
Bram Moolenaara8343c12016-02-04 22:09:48 +01001700 {
1701 /* For Win32 GUI channel_wait() always returns OK and we handle the
1702 * situation that there is nothing to read here.
1703 * TODO: how about a timeout? */
1704 if (WSAGetLastError() == WSAEWOULDBLOCK)
1705 return;
1706 }
1707#endif
Bram Moolenaard04a0202016-01-26 23:30:18 +01001708
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001709 /* Reading a socket disconnection (readlen == 0), or a socket error.
1710 * TODO: call error callback. */
1711 if (readlen <= 0 && channel->ch_job == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001712 {
1713 /* Queue a "DETACH" netbeans message in the command queue in order to
1714 * terminate the netbeans session later. Do not end the session here
1715 * directly as we may be running in the context of a call to
1716 * netbeans_parse_messages():
1717 * netbeans_parse_messages
1718 * -> autocmd triggered while processing the netbeans cmd
1719 * -> ui_breakcheck
1720 * -> gui event loop or select loop
1721 * -> channel_read()
1722 */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001723 ch_errors(channel, "%s(): Cannot read", func);
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001724 channel_save(channel, part,
1725 (char_u *)DETACH_MSG, (int)STRLEN(DETACH_MSG));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001726
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001727 /* TODO: When reading from stdout is not possible, should we try to
1728 * keep stdin and stderr open? Probably not, assume the other side
1729 * has died. */
1730 channel_close(channel);
1731 if (channel->ch_close_cb != NULL)
1732 (*channel->ch_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001733
1734 if (len < 0)
1735 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001736 ch_error(channel, "channel_read(): cannot read from channel");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001737 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001738 }
1739 }
1740
1741#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001742 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001743 if (CH_HAS_GUI && gtk_main_level() > 0)
1744 gtk_main_quit();
1745#endif
1746}
1747
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001748/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001749 * Read from RAW or NL "channel"/"part". Blocks until there is something to
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001750 * read or the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001751 * Returns what was read in allocated memory.
1752 * Returns NULL in case of error or timeout.
1753 */
1754 char_u *
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001755channel_read_block(channel_T *channel, int part, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001756{
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001757 char_u *buf;
1758 char_u *msg;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001759 ch_mode_T mode = channel->ch_part[part].ch_mode;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001760 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001761 char_u *nl;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001762
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001763 ch_logsn(channel, "Blocking %s read, timeout: %d msec",
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001764 mode == MODE_RAW ? "RAW" : "NL", timeout);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001765
1766 while (TRUE)
1767 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001768 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001769 if (buf != NULL && (mode == MODE_RAW
1770 || (mode == MODE_NL && vim_strchr(buf, NL) != NULL)))
1771 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001772 if (buf != NULL && channel_collapse(channel, part) == OK)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001773 continue;
1774
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001775 /* Wait for up to the channel timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001776 if (fd == INVALID_FD
1777 || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001778 return NULL;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001779 channel_read(channel, part, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001780 }
1781
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001782 if (mode == MODE_RAW)
1783 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001784 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001785 }
1786 else
1787 {
1788 nl = vim_strchr(buf, NL);
1789 if (nl[1] == NUL)
1790 {
1791 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001792 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001793 *nl = NUL;
1794 }
1795 else
1796 {
1797 /* Copy the message into allocated memory and remove it from the
1798 * buffer. */
1799 msg = vim_strnsave(buf, (int)(nl - buf));
1800 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1801 }
1802 }
1803 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001804 ch_logn(channel, "Returning %d bytes", (int)STRLEN(msg));
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001805 return msg;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001806}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001807
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001808/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001809 * Read one JSON message with ID "id" from "channel"/"part" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001810 * result in "rettv".
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001811 * When "id" is -1 accept any message;
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001812 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001813 */
1814 int
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001815channel_read_json_block(
1816 channel_T *channel,
1817 int part,
1818 int timeout,
1819 int id,
1820 typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001821{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001822 int more;
Bram Moolenaard8070362016-02-15 21:56:54 +01001823 sock_T fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001824
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001825 ch_log(channel, "Reading JSON");
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001826 if (id != -1)
1827 channel->ch_part[part].ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001828 for (;;)
1829 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001830 more = channel_parse_json(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001831
1832 /* search for messsage "id" */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001833 if (channel_get_json(channel, part, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001834 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001835 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001836 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001837 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001838
Bram Moolenaard7ece102016-02-02 23:23:02 +01001839 if (!more)
1840 {
1841 /* Handle any other messages in the queue. If done some more
1842 * messages may have arrived. */
1843 if (channel_parse_messages())
1844 continue;
1845
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001846 /* Wait for up to the timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001847 fd = channel->ch_part[part].ch_fd;
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001848 if (fd == INVALID_FD || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001849 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001850 channel_read(channel, part, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001851 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001852 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001853 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001854 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001855}
1856
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001857# if defined(WIN32) || defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK) \
1858 || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001859/*
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001860 * Lookup the channel from the socket. Set "partp" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001861 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001862 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001863 channel_T *
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001864channel_fd2channel(sock_T fd, int *partp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001865{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001866 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001867 int part;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001868
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001869 if (fd != INVALID_FD)
Bram Moolenaar77073442016-02-13 23:23:53 +01001870 for (channel = first_channel; channel != NULL;
1871 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001872 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001873# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001874 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001875# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001876 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001877# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001878 if (channel->ch_part[part].ch_fd == fd)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001879 {
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001880 *partp = part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001881 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001882 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001883 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001884 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001885}
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001886# endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001887
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001888# if defined(WIN32) || defined(PROTO)
1889/*
1890 * Check the channels for anything that is ready to be read.
1891 * The data is put in the read queue.
1892 */
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001893 void
1894channel_handle_events(void)
1895{
1896 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001897 int part;
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001898 sock_T fd;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001899
1900 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1901 {
1902# ifdef FEAT_GUI_W32
1903 /* only check the pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001904 for (part = PART_OUT; part <= PART_ERR; ++part)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001905# else
1906# ifdef CHANNEL_PIPES
1907 /* check the socket and pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001908 for (part = PART_SOCK; part <= PART_ERR; ++part)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001909# else
1910 /* only check the socket */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001911 part = PART_SOCK;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001912# endif
1913# endif
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001914 {
1915 fd = channel->ch_part[part].ch_fd;
1916 if (fd != INVALID_FD && channel_wait(channel, fd, 0) == OK)
1917 channel_read(channel, part, "channel_handle_events");
1918 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001919 }
1920}
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001921# endif
1922
Bram Moolenaard04a0202016-01-26 23:30:18 +01001923/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001924 * Write "buf" (NUL terminated string) to "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001925 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001926 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001927 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001928 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001929channel_send(channel_T *channel, int part, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001930{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001931 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001932 int res;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001933 sock_T fd;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001934
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001935 fd = channel->ch_part[part].ch_fd;
1936 if (fd == INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001937 {
1938 if (!channel->ch_error && fun != NULL)
1939 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001940 ch_errors(channel, "%s(): write while not connected", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001941 EMSG2("E630: %s(): write while not connected", fun);
1942 }
1943 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001944 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001945 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001946
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001947 if (log_fd != NULL)
1948 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001949 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001950 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001951 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001952 fprintf(log_fd, "'\n");
1953 fflush(log_fd);
1954 }
1955
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001956 if (part == PART_SOCK)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001957 res = sock_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001958 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001959 res = fd_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001960 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001961 {
1962 if (!channel->ch_error && fun != NULL)
1963 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001964 ch_errors(channel, "%s(): write failed", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001965 EMSG2("E631: %s(): write failed", fun);
1966 }
1967 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001968 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001969 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001970
1971 channel->ch_error = FALSE;
1972 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001973}
1974
1975# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001976/*
1977 * Add open channels to the poll struct.
1978 * Return the adjusted struct index.
1979 * The type of "fds" is hidden to avoid problems with the function proto.
1980 */
1981 int
1982channel_poll_setup(int nfd_in, void *fds_in)
1983{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001984 int nfd = nfd_in;
1985 channel_T *channel;
1986 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001987 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001988
Bram Moolenaar77073442016-02-13 23:23:53 +01001989 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001990 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001991# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001992 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001993# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001994 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001995# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001996 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001997 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001998 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001999 channel->ch_part[part].ch_poll_idx = nfd;
2000 fds[nfd].fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002001 fds[nfd].events = POLLIN;
2002 nfd++;
2003 }
2004 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002005 channel->ch_part[part].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002006 }
2007 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002008
2009 return nfd;
2010}
2011
2012/*
2013 * The type of "fds" is hidden to avoid problems with the function proto.
2014 */
2015 int
2016channel_poll_check(int ret_in, void *fds_in)
2017{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002018 int ret = ret_in;
2019 channel_T *channel;
2020 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002021 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002022
Bram Moolenaar77073442016-02-13 23:23:53 +01002023 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002024 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002025# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002026 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002027# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002028 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002029# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002030 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002031 int idx = channel->ch_part[part].ch_poll_idx;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002032
2033 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
2034 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002035 channel_read(channel, part, "channel_poll_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002036 --ret;
2037 }
2038 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002039 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002040
2041 return ret;
2042}
Bram Moolenaard04a0202016-01-26 23:30:18 +01002043# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002044
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002045# if (!defined(WIN32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002046/*
2047 * The type of "rfds" is hidden to avoid problems with the function proto.
2048 */
2049 int
2050channel_select_setup(int maxfd_in, void *rfds_in)
2051{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002052 int maxfd = maxfd_in;
2053 channel_T *channel;
2054 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002055 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002056
Bram Moolenaar77073442016-02-13 23:23:53 +01002057 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002058 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002059# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002060 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002061# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002062 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002063# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002064 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002065 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002066
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002067 if (fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002068 {
Bram Moolenaard8070362016-02-15 21:56:54 +01002069 FD_SET((int)fd, rfds);
2070 if (maxfd < (int)fd)
2071 maxfd = (int)fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002072 }
2073 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002074 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002075
2076 return maxfd;
2077}
2078
2079/*
2080 * The type of "rfds" is hidden to avoid problems with the function proto.
2081 */
2082 int
2083channel_select_check(int ret_in, void *rfds_in)
2084{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002085 int ret = ret_in;
2086 channel_T *channel;
2087 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002088 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002089
Bram Moolenaar77073442016-02-13 23:23:53 +01002090 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002091 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002092# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002093 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002094# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002095 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002096# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002097 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002098 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002099
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002100 if (ret > 0 && fd != INVALID_FD && FD_ISSET(fd, rfds))
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002101 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002102 channel_read(channel, part, "channel_select_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002103 --ret;
2104 }
2105 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002106 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002107
2108 return ret;
2109}
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002110# endif /* !WIN32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002111
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002112/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01002113 * Execute queued up commands.
2114 * Invoked from the main loop when it's safe to execute received commands.
2115 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002116 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01002117 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002118channel_parse_messages(void)
2119{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002120 channel_T *channel = first_channel;
2121 int ret = FALSE;
2122 int r;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002123 int part = PART_SOCK;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002124
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002125 while (channel != NULL)
2126 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002127 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaard7ece102016-02-02 23:23:02 +01002128 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002129 /* Increase the refcount, in case the handler causes the channel
2130 * to be unreferenced or closed. */
2131 ++channel->ch_refcount;
2132 r = may_invoke_callback(channel, part);
2133 if (r == OK)
2134 ret = TRUE;
2135 if (channel_unref(channel) || r == OK)
2136 {
2137 /* channel was freed or something was done, start over */
2138 channel = first_channel;
2139 part = PART_SOCK;
2140 continue;
2141 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002142 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002143#ifdef CHANNEL_PIPES
2144 if (part < PART_ERR)
2145 ++part;
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002146 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002147#endif
2148 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002149 channel = channel->ch_next;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002150 part = PART_SOCK;
2151 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002152 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002153 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002154}
2155
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01002156/*
2157 * Mark references to lists used in channels.
2158 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002159 int
2160set_ref_in_channel(int copyID)
2161{
Bram Moolenaar77073442016-02-13 23:23:53 +01002162 int abort = FALSE;
2163 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002164 int part;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002165
Bram Moolenaar77073442016-02-13 23:23:53 +01002166 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002167 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002168#ifdef CHANNEL_PIPES
2169 for (part = PART_SOCK; part < PART_IN; ++part)
2170#else
2171 part = PART_SOCK;
2172#endif
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002173 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002174 jsonq_T *head = &channel->ch_part[part].ch_json_head;
2175 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002176
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002177 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002178 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002179 list_T *l = item->jq_value->vval.v_list;
2180
2181 if (l->lv_copyID != copyID)
2182 {
2183 l->lv_copyID = copyID;
2184 abort = abort || set_ref_in_list(l, copyID, NULL);
2185 }
2186 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002187 }
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002188 }
2189 }
2190 return abort;
2191}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002192
2193/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002194 * Return the "part" to write to for "channel".
2195 */
2196 int
2197channel_part_send(channel_T *channel)
2198{
2199#ifdef CHANNEL_PIPES
2200 if (channel->CH_SOCK_FD == INVALID_FD)
2201 return PART_IN;
2202#endif
2203 return PART_SOCK;
2204}
2205
2206/*
2207 * Return the default "part" to read from for "channel".
2208 */
2209 int
2210channel_part_read(channel_T *channel)
2211{
2212#ifdef CHANNEL_PIPES
2213 if (channel->CH_SOCK_FD == INVALID_FD)
2214 return PART_OUT;
2215#endif
2216 return PART_SOCK;
2217}
2218
2219/*
2220 * Return the mode of "channel"/"part"
Bram Moolenaar77073442016-02-13 23:23:53 +01002221 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002222 */
2223 ch_mode_T
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002224channel_get_mode(channel_T *channel, int part)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002225{
Bram Moolenaar77073442016-02-13 23:23:53 +01002226 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002227 return MODE_JSON;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002228 return channel->ch_part[part].ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002229}
2230
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01002231/*
2232 * Return the timeout of "channel"/"part"
2233 */
2234 int
2235channel_get_timeout(channel_T *channel, int part)
2236{
2237 return channel->ch_part[part].ch_timeout;
2238}
2239
Bram Moolenaare0874f82016-01-24 20:36:41 +01002240#endif /* FEAT_CHANNEL */