blob: 169db9cd36ad54a992b1fb055d5c17aad8e38cc7 [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 Moolenaar910b8aa2016-02-16 21:03:07 +0100740 * Set various properties from an "options" argument.
741 */
742 void
743channel_set_options(channel_T *channel, jobopt_T *options)
744{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100745 int part;
746
Bram Moolenaar40ea1da2016-02-19 22:33:35 +0100747 if (options->jo_set & JO_MODE)
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100748 for (part = PART_SOCK; part <= PART_IN; ++part)
749 channel->ch_part[part].ch_mode = options->jo_mode;
Bram Moolenaar40ea1da2016-02-19 22:33:35 +0100750 if (options->jo_set & JO_TIMEOUT)
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100751 for (part = PART_SOCK; part <= PART_IN; ++part)
752 channel->ch_part[part].ch_timeout = options->jo_timeout;
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100753
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100754 if (options->jo_set & JO_CALLBACK)
755 {
756 vim_free(channel->ch_callback);
757 if (options->jo_callback != NULL && *options->jo_callback != NUL)
758 channel->ch_callback = vim_strsave(options->jo_callback);
759 else
760 channel->ch_callback = NULL;
761 }
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100762}
763
764/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100765 * Set the callback for "channel"/"part" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100766 */
767 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100768channel_set_req_callback(
769 channel_T *channel,
770 int part,
771 char_u *callback,
772 int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100773{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100774 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100775 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
776
777 if (item != NULL)
778 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100779 item->cq_callback = vim_strsave(callback);
780 item->cq_seq_nr = id;
781 item->cq_prev = head->cq_prev;
782 head->cq_prev = item;
783 item->cq_next = NULL;
784 if (item->cq_prev == NULL)
785 head->cq_next = item;
786 else
787 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100788 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100789}
790
791/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100792 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100793 */
794 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100795invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100796{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100797 typval_T rettv;
798 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100799
Bram Moolenaar77073442016-02-13 23:23:53 +0100800 argv[0].v_type = VAR_CHANNEL;
801 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100802
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100803 call_func(callback, (int)STRLEN(callback),
804 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
805 /* If an echo command was used the cursor needs to be put back where
806 * it belongs. */
807 setcursor();
808 cursor_on();
809 out_flush();
810}
811
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100812/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100813 * Return the first buffer from channel "channel"/"part" and remove it.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100814 * The caller must free it.
815 * Returns NULL if there is nothing.
816 */
817 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100818channel_get(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100819{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100820 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100821 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100822 char_u *p;
823
Bram Moolenaar77073442016-02-13 23:23:53 +0100824 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100825 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100826 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100827 p = node->rq_buffer;
828 head->rq_next = node->rq_next;
829 if (node->rq_next == NULL)
830 head->rq_prev = NULL;
831 else
832 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100833 vim_free(node);
834 return p;
835}
836
837/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100838 * Returns the whole buffer contents concatenated for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100839 */
840 static char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100841channel_get_all(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100842{
843 /* Concatenate everything into one buffer.
844 * TODO: avoid multiple allocations. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100845 while (channel_collapse(channel, part) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100846 ;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100847 return channel_get(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100848}
849
850/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100851 * Collapses the first and second buffer for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100852 * Returns FAIL if that is not possible.
853 */
854 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100855channel_collapse(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100856{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100857 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100858 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100859 char_u *p;
860
Bram Moolenaar77073442016-02-13 23:23:53 +0100861 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100862 return FAIL;
863
Bram Moolenaar77073442016-02-13 23:23:53 +0100864 p = alloc((unsigned)(STRLEN(node->rq_buffer)
865 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100866 if (p == NULL)
867 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100868 STRCPY(p, node->rq_buffer);
869 STRCAT(p, node->rq_next->rq_buffer);
870 vim_free(node->rq_next->rq_buffer);
871 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100872
Bram Moolenaar77073442016-02-13 23:23:53 +0100873 /* dispose of the node and its buffer */
874 head->rq_next = node->rq_next;
875 head->rq_next->rq_prev = NULL;
876 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100877 vim_free(node);
878 return OK;
879}
880
881/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100882 * Store "buf[len]" on "channel"/"part".
883 * Returns OK or FAIL.
884 */
885 static int
886channel_save(channel_T *channel, int part, char_u *buf, int len)
887{
888 readq_T *node;
889 readq_T *head = &channel->ch_part[part].ch_head;
890 char_u *p;
891 int i;
892
893 node = (readq_T *)alloc(sizeof(readq_T));
894 if (node == NULL)
895 return FAIL; /* out of memory */
896 node->rq_buffer = alloc(len + 1);
897 if (node->rq_buffer == NULL)
898 {
899 vim_free(node);
900 return FAIL; /* out of memory */
901 }
902
903 if (channel->ch_part[part].ch_mode == MODE_NL)
904 {
905 /* Drop any CR before a NL. */
906 p = node->rq_buffer;
907 for (i = 0; i < len; ++i)
908 if (buf[i] != CAR || i + 1 >= len || buf[i + 1] != NL)
909 *p++ = buf[i];
910 *p = NUL;
911 }
912 else
913 {
914 mch_memmove(node->rq_buffer, buf, len);
915 node->rq_buffer[len] = NUL;
916 }
917
918 /* append node to the tail of the queue */
919 node->rq_next = NULL;
920 node->rq_prev = head->rq_prev;
921 if (head->rq_prev == NULL)
922 head->rq_next = node;
923 else
924 head->rq_prev->rq_next = node;
925 head->rq_prev = node;
926
927 if (log_fd != NULL)
928 {
929 ch_log_lead("RECV ", channel);
930 fprintf(log_fd, "'");
931 if (fwrite(buf, len, 1, log_fd) != 1)
932 return FAIL;
933 fprintf(log_fd, "'\n");
934 }
935 return OK;
936}
937
938/*
939 * Use the read buffer of "channel"/"part" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100940 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +0100941 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100942 */
Bram Moolenaard7ece102016-02-02 23:23:02 +0100943 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100944channel_parse_json(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100945{
946 js_read_T reader;
947 typval_T listtv;
948 jsonq_T *item;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100949 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100950 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100951
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100952 if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100953 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100954
955 /* TODO: make reader work properly */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100956 /* reader.js_buf = channel_peek(channel, part); */
957 reader.js_buf = channel_get_all(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100958 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +0100959 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100960 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +0100961 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +0100962 ret = json_decode(&reader, &listtv,
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100963 channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100964 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100965 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +0100966 /* Only accept the response when it is a list with at least two
967 * items. */
968 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100969 {
970 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100971 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +0100972 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100973 else
974 {
Bram Moolenaard7ece102016-02-02 23:23:02 +0100975 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
976 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100977 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100978 else
979 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100980 item->jq_value = alloc_tv();
981 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +0100982 {
983 vim_free(item);
984 clear_tv(&listtv);
985 }
986 else
987 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100988 *item->jq_value = listtv;
989 item->jq_prev = head->jq_prev;
990 head->jq_prev = item;
991 item->jq_next = NULL;
992 if (item->jq_prev == NULL)
993 head->jq_next = item;
994 else
995 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +0100996 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100997 }
998 }
999 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001000
1001 /* Put the unread part back into the channel.
1002 * TODO: insert in front */
1003 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001004 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001005 channel_save(channel, part, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001006 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001007 ret = TRUE;
1008 }
1009 else
1010 ret = FALSE;
1011
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001012 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001013 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001014}
1015
1016/*
Bram Moolenaard46ae142016-02-16 13:33:52 +01001017 * Remove "node" from the queue that it is in. Does not free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +01001018 */
1019 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001020remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001021{
Bram Moolenaar77073442016-02-13 23:23:53 +01001022 if (node->cq_prev == NULL)
1023 head->cq_next = node->cq_next;
1024 else
1025 node->cq_prev->cq_next = node->cq_next;
1026 if (node->cq_next == NULL)
1027 head->cq_prev = node->cq_prev;
1028 else
1029 node->cq_next->cq_prev = node->cq_prev;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001030}
1031
1032/*
1033 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +01001034 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001035 */
1036 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001037remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001038{
Bram Moolenaar77073442016-02-13 23:23:53 +01001039 if (node->jq_prev == NULL)
1040 head->jq_next = node->jq_next;
1041 else
1042 node->jq_prev->jq_next = node->jq_next;
1043 if (node->jq_next == NULL)
1044 head->jq_prev = node->jq_prev;
1045 else
1046 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001047 vim_free(node);
1048}
1049
1050/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001051 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001052 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +01001053 * When "id" is zero or negative jut get the first message. But not the one
1054 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001055 * Return OK when found and return the value in "rettv".
1056 * Return FAIL otherwise.
1057 */
1058 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001059channel_get_json(channel_T *channel, int part, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001060{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001061 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001062 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001063
Bram Moolenaar77073442016-02-13 23:23:53 +01001064 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001065 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001066 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001067 typval_T *tv = &l->lv_first->li_tv;
1068
1069 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001070 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001071 || tv->vval.v_number == 0
1072 || tv->vval.v_number != channel->ch_part[part].ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001073 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001074 *rettv = item->jq_value;
1075 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001076 return OK;
1077 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001078 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001079 }
1080 return FAIL;
1081}
1082
1083/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001084 * Execute a command received over "channel"/"part"
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001085 * "cmd" is the command string, "arg2" the second argument.
1086 * "arg3" is the third argument, NULL if missing.
1087 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001088 static void
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001089channel_exe_cmd(
1090 channel_T *channel,
1091 int part,
1092 char_u *cmd,
1093 typval_T *arg2,
1094 typval_T *arg3)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001095{
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001096 char_u *arg;
1097
1098 if (arg2->v_type != VAR_STRING)
1099 {
1100 if (p_verbose > 2)
1101 EMSG("E903: received ex command with non-string argument");
1102 return;
1103 }
1104 arg = arg2->vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001105 if (arg == NULL)
1106 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001107
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001108 if (STRCMP(cmd, "ex") == 0)
1109 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001110 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001111 }
1112 else if (STRCMP(cmd, "normal") == 0)
1113 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001114 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001115
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001116 ea.arg = arg;
1117 ea.addr_count = 0;
1118 ea.forceit = TRUE; /* no mapping */
1119 ex_normal(&ea);
1120 }
1121 else if (STRCMP(cmd, "redraw") == 0)
1122 {
1123 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001124
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001125 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001126 ex_redraw(&ea);
1127 showruler(FALSE);
1128 setcursor();
1129 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001130#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001131 if (gui.in_use)
1132 {
1133 gui_update_cursor(FALSE, FALSE);
1134 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001135 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001136#endif
1137 }
1138 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "eval") == 0)
1139 {
1140 int is_eval = cmd[1] == 'v';
1141
Bram Moolenaar66624ff2016-02-03 23:59:43 +01001142 if (is_eval && (arg3 == NULL || arg3->v_type != VAR_NUMBER))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001143 {
1144 if (p_verbose > 2)
1145 EMSG("E904: third argument for eval must be a number");
1146 }
1147 else
1148 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001149 typval_T *tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001150 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +01001151 char_u *json = NULL;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001152 int options = channel->ch_part[part].ch_mode == MODE_JS
1153 ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001154
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001155 /* Don't pollute the display with errors. */
1156 ++emsg_skip;
1157 tv = eval_expr(arg, NULL);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001158 if (is_eval)
1159 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001160 if (tv != NULL)
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001161 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1162 options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001163 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001164 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001165 /* If evaluation failed or the result can't be encoded
1166 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001167 vim_free(json);
1168 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001169 err_tv.v_type = VAR_STRING;
1170 err_tv.vval.v_string = (char_u *)"ERROR";
1171 tv = &err_tv;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001172 json = json_encode_nr_expr(arg3->vval.v_number, tv,
1173 options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001174 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001175 if (json != NULL)
1176 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001177 channel_send(channel, part, json, "eval");
Bram Moolenaar55fab432016-02-07 16:53:13 +01001178 vim_free(json);
1179 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001180 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001181 --emsg_skip;
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001182 if (tv != &err_tv)
1183 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001184 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001185 }
1186 else if (p_verbose > 2)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001187 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001188}
1189
1190/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001191 * Invoke a callback for "channel"/"part" if needed.
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001192 * Return TRUE when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001193 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001194 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001195may_invoke_callback(channel_T *channel, int part)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001196{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001197 char_u *msg = NULL;
1198 typval_T *listtv = NULL;
1199 list_T *list;
1200 typval_T *typetv;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001201 typval_T argv[3];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001202 int seq_nr = -1;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001203 ch_mode_T ch_mode = channel->ch_part[part].ch_mode;
1204 char_u *callback = NULL;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001205
Bram Moolenaara07fec92016-02-05 21:04:08 +01001206 if (channel->ch_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001207 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001208 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001209
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001210 if (channel->ch_part[part].ch_callback != NULL)
1211 callback = channel->ch_part[part].ch_callback;
1212 else
1213 callback = channel->ch_callback;
1214
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001215 if (ch_mode == MODE_JSON || ch_mode == MODE_JS)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001216 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001217 /* Get any json message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001218 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001219 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001220 /* Parse readahead, return when there is still no message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001221 channel_parse_json(channel, part);
1222 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001223 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001224 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001225
1226 list = listtv->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001227 argv[1] = list->lv_first->li_next->li_tv;
1228 typetv = &list->lv_first->li_tv;
1229 if (typetv->v_type == VAR_STRING)
1230 {
1231 typval_T *arg3 = NULL;
1232 char_u *cmd = typetv->vval.v_string;
1233
Bram Moolenaar66624ff2016-02-03 23:59:43 +01001234 /* ["cmd", arg] or ["cmd", arg, arg] */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001235 if (list->lv_len == 3)
1236 arg3 = &list->lv_last->li_tv;
Bram Moolenaar77073442016-02-13 23:23:53 +01001237 ch_logs(channel, "Executing %s command", (char *)cmd);
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001238 channel_exe_cmd(channel, part, cmd, &argv[1], arg3);
Bram Moolenaar77073442016-02-13 23:23:53 +01001239 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001240 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001241 }
1242
1243 if (typetv->v_type != VAR_NUMBER)
1244 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001245 ch_error(channel,
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001246 "Dropping message with invalid sequence number type");
Bram Moolenaar77073442016-02-13 23:23:53 +01001247 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001248 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001249 }
1250 seq_nr = typetv->vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001251 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001252 else if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001253 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001254 /* nothing to read on RAW or NL channel */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001255 return FALSE;
1256 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001257 else
1258 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001259 /* If there is no callback drop the message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001260 if (callback == NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001261 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001262 while ((msg = channel_get(channel, part)) != NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001263 vim_free(msg);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001264 return FALSE;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001265 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001266
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001267 if (ch_mode == MODE_NL)
1268 {
1269 char_u *nl;
1270 char_u *buf;
1271
1272 /* See if we have a message ending in NL in the first buffer. If
1273 * not try to concatenate the first and the second buffer. */
1274 while (TRUE)
1275 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001276 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001277 nl = vim_strchr(buf, NL);
1278 if (nl != NULL)
1279 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001280 if (channel_collapse(channel, part) == FAIL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001281 return FALSE; /* incomplete message */
1282 }
1283 if (nl[1] == NUL)
1284 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001285 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001286 else
1287 {
1288 /* Copy the message into allocated memory and remove it from
1289 * the buffer. */
1290 msg = vim_strnsave(buf, (int)(nl - buf));
1291 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1292 }
1293 }
1294 else
1295 /* For a raw channel we don't know where the message ends, just
1296 * get everything we have. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001297 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001298
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001299 argv[1].v_type = VAR_STRING;
1300 argv[1].vval.v_string = msg;
1301 }
1302
Bram Moolenaara07fec92016-02-05 21:04:08 +01001303 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001304 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001305 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001306 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001307 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001308
1309 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001310 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001311 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001312 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001313 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001314 ch_log(channel, "Invoking one-time callback");
Bram Moolenaard46ae142016-02-16 13:33:52 +01001315 /* Remove the item from the list first, if the callback
1316 * invokes ch_close() the list will be cleared. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001317 remove_cb_node(head, item);
Bram Moolenaard46ae142016-02-16 13:33:52 +01001318 invoke_callback(channel, item->cq_callback, argv);
1319 vim_free(item->cq_callback);
1320 vim_free(item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001321 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001322 break;
1323 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001324 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001325 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001326 if (!done)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001327 ch_log(channel, "Dropping message without callback");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001328 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001329 else if (callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001330 {
1331 /* invoke the channel callback */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001332 ch_log(channel, "Invoking channel callback");
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001333 invoke_callback(channel, callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001334 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001335 else
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001336 ch_log(channel, "Dropping message");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001337
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001338 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001339 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001340 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001341
1342 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001343}
1344
1345/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001346 * Return TRUE when channel "channel" is open for writing to.
1347 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001348 */
1349 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001350channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001351{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001352 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001353#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001354 || channel->CH_IN_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001355#endif
1356 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001357}
1358
1359/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001360 * Return TRUE when channel "channel" is open for reading or writing.
1361 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001362 */
1363 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001364channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001365{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001366 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001367#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001368 || channel->CH_IN_FD != INVALID_FD
1369 || channel->CH_OUT_FD != INVALID_FD
1370 || channel->CH_ERR_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001371#endif
1372 );
1373}
1374
1375/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001376 * Return a string indicating the status of the channel.
1377 */
1378 char *
1379channel_status(channel_T *channel)
1380{
1381 if (channel == NULL)
1382 return "fail";
1383 if (channel_is_open(channel))
1384 return "open";
1385 return "closed";
1386}
1387
1388/*
1389 * Close channel "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001390 * This does not trigger the close callback.
1391 */
1392 void
Bram Moolenaar77073442016-02-13 23:23:53 +01001393channel_close(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001394{
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001395 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001396
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001397#ifdef FEAT_GUI
1398 channel_gui_unregister(channel);
1399#endif
1400
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001401 if (channel->CH_SOCK_FD != INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001402 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001403 sock_close(channel->CH_SOCK_FD);
1404 channel->CH_SOCK_FD = INVALID_FD;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001405 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001406#if defined(CHANNEL_PIPES)
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001407 if (channel->CH_IN_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001408 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001409 fd_close(channel->CH_IN_FD);
1410 channel->CH_IN_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001411 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001412 if (channel->CH_OUT_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001413 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001414 fd_close(channel->CH_OUT_FD);
1415 channel->CH_OUT_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001416 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001417 if (channel->CH_ERR_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001418 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001419 fd_close(channel->CH_ERR_FD);
1420 channel->CH_ERR_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001421 }
1422#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001423
1424 channel->ch_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001425 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001426}
1427
Bram Moolenaard04a0202016-01-26 23:30:18 +01001428/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001429 * Return the first buffer from "channel"/"part" without removing it.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001430 * Returns NULL if there is nothing.
1431 */
1432 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001433channel_peek(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001434{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001435 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001436
Bram Moolenaar77073442016-02-13 23:23:53 +01001437 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001438 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001439 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001440}
1441
1442/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001443 * Clear the read buffer on "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001444 */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001445 static void
1446channel_clear_one(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001447{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001448 jsonq_T *json_head = &channel->ch_part[part].ch_json_head;
1449 cbq_T *cb_head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001450
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001451 while (channel_peek(channel, part) != NULL)
1452 vim_free(channel_get(channel, part));
Bram Moolenaar77073442016-02-13 23:23:53 +01001453
1454 while (cb_head->cq_next != NULL)
Bram Moolenaard46ae142016-02-16 13:33:52 +01001455 {
1456 cbq_T *node = cb_head->cq_next;
1457
1458 remove_cb_node(cb_head, node);
1459 vim_free(node->cq_callback);
1460 vim_free(node);
1461 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001462
1463 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001464 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001465 free_tv(json_head->jq_next->jq_value);
1466 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001467 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001468
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001469 vim_free(channel->ch_part[part].ch_callback);
1470 channel->ch_part[part].ch_callback = NULL;
1471}
1472
1473/*
1474 * Clear all the read buffers on "channel".
1475 */
1476 void
1477channel_clear(channel_T *channel)
1478{
1479 channel_clear_one(channel, PART_SOCK);
1480#ifdef CHANNEL_PIPES
1481 channel_clear_one(channel, PART_OUT);
1482 channel_clear_one(channel, PART_ERR);
1483#endif
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001484 vim_free(channel->ch_callback);
1485 channel->ch_callback = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001486}
1487
Bram Moolenaar77073442016-02-13 23:23:53 +01001488#if defined(EXITFREE) || defined(PROTO)
1489 void
1490channel_free_all(void)
1491{
1492 channel_T *channel;
1493
1494 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1495 channel_clear(channel);
1496}
1497#endif
1498
1499
Bram Moolenaard04a0202016-01-26 23:30:18 +01001500/* Sent when the channel is found closed when reading. */
1501#define DETACH_MSG "\"DETACH\"\n"
1502
1503/* Buffer size for reading incoming messages. */
1504#define MAXMSGSIZE 4096
1505
1506/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001507 * Check for reading from "fd" with "timeout" msec.
1508 * Return FAIL when there is nothing to read.
Bram Moolenaara8343c12016-02-04 22:09:48 +01001509 * Always returns OK for FEAT_GUI_W32.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001510 */
1511 static int
Bram Moolenaard8070362016-02-15 21:56:54 +01001512channel_wait(channel_T *channel, sock_T fd, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001513{
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001514 if (timeout > 0)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001515 ch_logn(channel, "Waiting for up to %d msec", timeout);
Bram Moolenaard8070362016-02-15 21:56:54 +01001516
Bram Moolenaard8070362016-02-15 21:56:54 +01001517# ifdef WIN32
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001518 if (fd != channel->CH_SOCK_FD)
Bram Moolenaard8070362016-02-15 21:56:54 +01001519 {
1520 DWORD nread;
1521 int diff;
1522 DWORD deadline = GetTickCount() + timeout;
1523
1524 /* reading from a pipe, not a socket */
1525 while (TRUE)
1526 {
Bram Moolenaare74e8e72016-02-16 22:01:30 +01001527 if (PeekNamedPipe((HANDLE)fd, NULL, 0, NULL, &nread, NULL)
1528 && nread > 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001529 return OK;
1530 diff = deadline - GetTickCount();
1531 if (diff < 0)
1532 break;
1533 /* Wait for 5 msec.
1534 * TODO: increase the sleep time when looping more often */
1535 Sleep(5);
1536 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001537 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001538 else
Bram Moolenaard8070362016-02-15 21:56:54 +01001539#endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001540 {
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001541#if defined(FEAT_GUI_W32)
1542 /* Can't check socket for Win32 GUI, always return OK. */
1543 ch_log(channel, "Can't check, assuming there is something to read");
1544 return OK;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001545#else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001546# if defined(HAVE_SELECT)
1547 struct timeval tval;
1548 fd_set rfds;
1549 int ret;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001550
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001551 FD_ZERO(&rfds);
1552 FD_SET((int)fd, &rfds);
1553 tval.tv_sec = timeout / 1000;
1554 tval.tv_usec = (timeout % 1000) * 1000;
1555 for (;;)
1556 {
1557 ret = select((int)fd + 1, &rfds, NULL, NULL, &tval);
1558# ifdef EINTR
1559 SOCK_ERRNO;
1560 if (ret == -1 && errno == EINTR)
1561 continue;
1562# endif
1563 if (ret > 0)
1564 return OK;
1565 break;
1566 }
1567# else
1568 struct pollfd fds;
1569
1570 fds.fd = fd;
1571 fds.events = POLLIN;
1572 if (poll(&fds, 1, timeout) > 0)
1573 return OK;
Bram Moolenaarb8b65112016-01-28 23:01:49 +01001574# endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001575#endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001576 }
1577 ch_log(channel, "Nothing to read");
1578 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001579}
1580
1581/*
1582 * Return a unique ID to be used in a message.
1583 */
1584 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001585channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001586{
1587 static int next_id = 1;
1588
1589 return next_id++;
1590}
1591
1592/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001593 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001594 * "part" is PART_SOCK, PART_OUT or PART_ERR.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001595 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001596 */
1597 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001598channel_read(channel_T *channel, int part, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001599{
1600 static char_u *buf = NULL;
1601 int len = 0;
1602 int readlen = 0;
Bram Moolenaard8070362016-02-15 21:56:54 +01001603 sock_T fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001604 int use_socket = FALSE;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001605
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001606 fd = channel->ch_part[part].ch_fd;
1607 if (fd == INVALID_FD)
1608 {
1609 ch_error(channel, "channel_read() called while socket is closed");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001610 return;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001611 }
1612 use_socket = fd == channel->CH_SOCK_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001613
1614 /* Allocate a buffer to read into. */
1615 if (buf == NULL)
1616 {
1617 buf = alloc(MAXMSGSIZE);
1618 if (buf == NULL)
1619 return; /* out of memory! */
1620 }
1621
1622 /* Keep on reading for as long as there is something to read.
1623 * Use select() or poll() to avoid blocking on a message that is exactly
1624 * MAXMSGSIZE long. */
1625 for (;;)
1626 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001627 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001628 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001629 if (use_socket)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001630 len = sock_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001631 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001632 len = fd_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001633 if (len <= 0)
1634 break; /* error or nothing more to read */
1635
1636 /* Store the read message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001637 channel_save(channel, part, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001638 readlen += len;
1639 if (len < MAXMSGSIZE)
1640 break; /* did read everything that's available */
1641 }
Bram Moolenaara8343c12016-02-04 22:09:48 +01001642#ifdef FEAT_GUI_W32
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001643 if (use_socket && len == SOCKET_ERROR)
Bram Moolenaara8343c12016-02-04 22:09:48 +01001644 {
1645 /* For Win32 GUI channel_wait() always returns OK and we handle the
1646 * situation that there is nothing to read here.
1647 * TODO: how about a timeout? */
1648 if (WSAGetLastError() == WSAEWOULDBLOCK)
1649 return;
1650 }
1651#endif
Bram Moolenaard04a0202016-01-26 23:30:18 +01001652
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001653 /* Reading a socket disconnection (readlen == 0), or a socket error.
1654 * TODO: call error callback. */
1655 if (readlen <= 0 && channel->ch_job == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001656 {
1657 /* Queue a "DETACH" netbeans message in the command queue in order to
1658 * terminate the netbeans session later. Do not end the session here
1659 * directly as we may be running in the context of a call to
1660 * netbeans_parse_messages():
1661 * netbeans_parse_messages
1662 * -> autocmd triggered while processing the netbeans cmd
1663 * -> ui_breakcheck
1664 * -> gui event loop or select loop
1665 * -> channel_read()
1666 */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001667 ch_errors(channel, "%s(): Cannot read", func);
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001668 channel_save(channel, part,
1669 (char_u *)DETACH_MSG, (int)STRLEN(DETACH_MSG));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001670
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001671 /* TODO: When reading from stdout is not possible, should we try to
1672 * keep stdin and stderr open? Probably not, assume the other side
1673 * has died. */
1674 channel_close(channel);
1675 if (channel->ch_close_cb != NULL)
1676 (*channel->ch_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001677
1678 if (len < 0)
1679 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001680 ch_error(channel, "channel_read(): cannot read from channel");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001681 PERROR(_("E896: read from channel"));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001682 }
1683 }
1684
1685#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001686 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001687 if (CH_HAS_GUI && gtk_main_level() > 0)
1688 gtk_main_quit();
1689#endif
1690}
1691
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001692/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001693 * Read from RAW or NL "channel"/"part". Blocks until there is something to
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001694 * read or the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001695 * Returns what was read in allocated memory.
1696 * Returns NULL in case of error or timeout.
1697 */
1698 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001699channel_read_block(channel_T *channel, int part)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001700{
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001701 char_u *buf;
1702 char_u *msg;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001703 ch_mode_T mode = channel->ch_part[part].ch_mode;
1704 int timeout = channel->ch_part[part].ch_timeout;
1705 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001706 char_u *nl;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001707
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001708 ch_logsn(channel, "Blocking %s read, timeout: %d msec",
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001709 mode == MODE_RAW ? "RAW" : "NL", timeout);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001710
1711 while (TRUE)
1712 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001713 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001714 if (buf != NULL && (mode == MODE_RAW
1715 || (mode == MODE_NL && vim_strchr(buf, NL) != NULL)))
1716 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001717 if (buf != NULL && channel_collapse(channel, part) == OK)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001718 continue;
1719
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001720 /* Wait for up to the channel timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001721 if (fd == INVALID_FD
1722 || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001723 return NULL;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001724 channel_read(channel, part, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001725 }
1726
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001727 if (mode == MODE_RAW)
1728 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001729 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001730 }
1731 else
1732 {
1733 nl = vim_strchr(buf, NL);
1734 if (nl[1] == NUL)
1735 {
1736 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001737 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001738 *nl = NUL;
1739 }
1740 else
1741 {
1742 /* Copy the message into allocated memory and remove it from the
1743 * buffer. */
1744 msg = vim_strnsave(buf, (int)(nl - buf));
1745 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1746 }
1747 }
1748 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001749 ch_logn(channel, "Returning %d bytes", (int)STRLEN(msg));
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001750 return msg;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001751}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001752
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001753/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001754 * Read one JSON message with ID "id" from "channel"/"part" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001755 * result in "rettv".
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001756 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001757 */
1758 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001759channel_read_json_block(channel_T *channel, int part, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001760{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001761 int more;
Bram Moolenaard8070362016-02-15 21:56:54 +01001762 sock_T fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001763
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001764 ch_log(channel, "Reading JSON");
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001765 channel->ch_part[part].ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001766 for (;;)
1767 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001768 more = channel_parse_json(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001769
1770 /* search for messsage "id" */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001771 if (channel_get_json(channel, part, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001772 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001773 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001774 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001775 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001776
Bram Moolenaard7ece102016-02-02 23:23:02 +01001777 if (!more)
1778 {
1779 /* Handle any other messages in the queue. If done some more
1780 * messages may have arrived. */
1781 if (channel_parse_messages())
1782 continue;
1783
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001784 /* Wait for up to the channel timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001785 fd = channel->ch_part[part].ch_fd;
1786 if (fd == INVALID_FD || channel_wait(channel, fd,
1787 channel->ch_part[part].ch_timeout) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001788 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001789 channel_read(channel, part, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001790 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001791 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001792 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001793 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001794}
1795
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001796# if defined(WIN32) || defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK) \
1797 || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001798/*
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001799 * Lookup the channel from the socket. Set "partp" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001800 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001801 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001802 channel_T *
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001803channel_fd2channel(sock_T fd, int *partp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001804{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001805 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001806 int part;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001807
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001808 if (fd != INVALID_FD)
Bram Moolenaar77073442016-02-13 23:23:53 +01001809 for (channel = first_channel; channel != NULL;
1810 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001811 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001812# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001813 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001814# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001815 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001816# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001817 if (channel->ch_part[part].ch_fd == fd)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001818 {
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001819 *partp = part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001820 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001821 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001822 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001823 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001824}
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001825
1826 void
1827channel_handle_events(void)
1828{
1829 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001830 int part;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001831 static int loop = 0;
1832
1833 /* Skip heavily polling */
1834 if (loop++ % 2)
1835 return;
1836
1837 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1838 {
1839# ifdef FEAT_GUI_W32
1840 /* only check the pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001841 for (part = PART_OUT; part <= PART_ERR; ++part)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001842# else
1843# ifdef CHANNEL_PIPES
1844 /* check the socket and pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001845 for (part = PART_SOCK; part <= PART_ERR; ++part)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001846# else
1847 /* only check the socket */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001848 part = PART_SOCK;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001849# endif
1850# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001851 channel_read(channel, part, "channel_handle_events");
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001852 }
1853}
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001854# endif
1855
Bram Moolenaard04a0202016-01-26 23:30:18 +01001856/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001857 * Write "buf" (NUL terminated string) to "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001858 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001859 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001860 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001861 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001862channel_send(channel_T *channel, int part, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001863{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001864 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001865 int res;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001866 sock_T fd;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001867
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001868 fd = channel->ch_part[part].ch_fd;
1869 if (fd == INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001870 {
1871 if (!channel->ch_error && fun != NULL)
1872 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001873 ch_errors(channel, "%s(): write while not connected", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001874 EMSG2("E630: %s(): write while not connected", fun);
1875 }
1876 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001877 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001878 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001879
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001880 if (log_fd != NULL)
1881 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001882 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001883 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001884 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001885 fprintf(log_fd, "'\n");
1886 fflush(log_fd);
1887 }
1888
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001889 if (part == PART_SOCK)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001890 res = sock_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001891 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001892 res = fd_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001893 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001894 {
1895 if (!channel->ch_error && fun != NULL)
1896 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001897 ch_errors(channel, "%s(): write failed", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001898 EMSG2("E631: %s(): write failed", fun);
1899 }
1900 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001901 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001902 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001903
1904 channel->ch_error = FALSE;
1905 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001906}
1907
1908# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001909/*
1910 * Add open channels to the poll struct.
1911 * Return the adjusted struct index.
1912 * The type of "fds" is hidden to avoid problems with the function proto.
1913 */
1914 int
1915channel_poll_setup(int nfd_in, void *fds_in)
1916{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001917 int nfd = nfd_in;
1918 channel_T *channel;
1919 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001920 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001921
Bram Moolenaar77073442016-02-13 23:23:53 +01001922 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001923 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001924# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001925 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001926# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001927 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001928# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001929 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001930 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001931 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001932 channel->ch_part[part].ch_poll_idx = nfd;
1933 fds[nfd].fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001934 fds[nfd].events = POLLIN;
1935 nfd++;
1936 }
1937 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001938 channel->ch_part[part].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001939 }
1940 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001941
1942 return nfd;
1943}
1944
1945/*
1946 * The type of "fds" is hidden to avoid problems with the function proto.
1947 */
1948 int
1949channel_poll_check(int ret_in, void *fds_in)
1950{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001951 int ret = ret_in;
1952 channel_T *channel;
1953 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001954 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001955
Bram Moolenaar77073442016-02-13 23:23:53 +01001956 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001957 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001958# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001959 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001960# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001961 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001962# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001963 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001964 int idx = channel->ch_part[part].ch_poll_idx;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001965
1966 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
1967 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001968 channel_read(channel, part, "channel_poll_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001969 --ret;
1970 }
1971 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001972 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01001973
1974 return ret;
1975}
Bram Moolenaard04a0202016-01-26 23:30:18 +01001976# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01001977
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001978# if (!defined(WIN32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01001979/*
1980 * The type of "rfds" is hidden to avoid problems with the function proto.
1981 */
1982 int
1983channel_select_setup(int maxfd_in, void *rfds_in)
1984{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001985 int maxfd = maxfd_in;
1986 channel_T *channel;
1987 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001988 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001989
Bram Moolenaar77073442016-02-13 23:23:53 +01001990 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001991 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001992# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001993 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001994# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001995 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001996# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001997 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001998 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001999
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002000 if (fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002001 {
Bram Moolenaard8070362016-02-15 21:56:54 +01002002 FD_SET((int)fd, rfds);
2003 if (maxfd < (int)fd)
2004 maxfd = (int)fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002005 }
2006 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002007 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002008
2009 return maxfd;
2010}
2011
2012/*
2013 * The type of "rfds" is hidden to avoid problems with the function proto.
2014 */
2015 int
2016channel_select_check(int ret_in, void *rfds_in)
2017{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002018 int ret = ret_in;
2019 channel_T *channel;
2020 fd_set *rfds = rfds_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 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002032
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002033 if (ret > 0 && fd != INVALID_FD && FD_ISSET(fd, rfds))
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002034 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002035 channel_read(channel, part, "channel_select_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 Moolenaared5a78e2016-02-19 21:05:03 +01002043# endif /* !WIN32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002044
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002045/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01002046 * Execute queued up commands.
2047 * Invoked from the main loop when it's safe to execute received commands.
2048 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002049 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01002050 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002051channel_parse_messages(void)
2052{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002053 channel_T *channel = first_channel;
2054 int ret = FALSE;
2055 int r;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002056 int part = PART_SOCK;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002057
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002058 while (channel != NULL)
2059 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002060 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaard7ece102016-02-02 23:23:02 +01002061 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002062 /* Increase the refcount, in case the handler causes the channel
2063 * to be unreferenced or closed. */
2064 ++channel->ch_refcount;
2065 r = may_invoke_callback(channel, part);
2066 if (r == OK)
2067 ret = TRUE;
2068 if (channel_unref(channel) || r == OK)
2069 {
2070 /* channel was freed or something was done, start over */
2071 channel = first_channel;
2072 part = PART_SOCK;
2073 continue;
2074 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002075 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002076#ifdef CHANNEL_PIPES
2077 if (part < PART_ERR)
2078 ++part;
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002079 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002080#endif
2081 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002082 channel = channel->ch_next;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002083 part = PART_SOCK;
2084 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002085 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002086 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002087}
2088
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01002089/*
2090 * Mark references to lists used in channels.
2091 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002092 int
2093set_ref_in_channel(int copyID)
2094{
Bram Moolenaar77073442016-02-13 23:23:53 +01002095 int abort = FALSE;
2096 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002097 int part;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002098
Bram Moolenaar77073442016-02-13 23:23:53 +01002099 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002100 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002101#ifdef CHANNEL_PIPES
2102 for (part = PART_SOCK; part < PART_IN; ++part)
2103#else
2104 part = PART_SOCK;
2105#endif
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002106 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002107 jsonq_T *head = &channel->ch_part[part].ch_json_head;
2108 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002109
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002110 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002111 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002112 list_T *l = item->jq_value->vval.v_list;
2113
2114 if (l->lv_copyID != copyID)
2115 {
2116 l->lv_copyID = copyID;
2117 abort = abort || set_ref_in_list(l, copyID, NULL);
2118 }
2119 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002120 }
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002121 }
2122 }
2123 return abort;
2124}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002125
2126/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002127 * Return the "part" to write to for "channel".
2128 */
2129 int
2130channel_part_send(channel_T *channel)
2131{
2132#ifdef CHANNEL_PIPES
2133 if (channel->CH_SOCK_FD == INVALID_FD)
2134 return PART_IN;
2135#endif
2136 return PART_SOCK;
2137}
2138
2139/*
2140 * Return the default "part" to read from for "channel".
2141 */
2142 int
2143channel_part_read(channel_T *channel)
2144{
2145#ifdef CHANNEL_PIPES
2146 if (channel->CH_SOCK_FD == INVALID_FD)
2147 return PART_OUT;
2148#endif
2149 return PART_SOCK;
2150}
2151
2152/*
2153 * Return the mode of "channel"/"part"
Bram Moolenaar77073442016-02-13 23:23:53 +01002154 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002155 */
2156 ch_mode_T
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002157channel_get_mode(channel_T *channel, int part)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002158{
Bram Moolenaar77073442016-02-13 23:23:53 +01002159 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002160 return MODE_JSON;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002161 return channel->ch_part[part].ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002162}
2163
Bram Moolenaare0874f82016-01-24 20:36:41 +01002164#endif /* FEAT_CHANNEL */