blob: 9f6cdc7ca6004944f4d8a86eaa124e94a87ca437 [file] [log] [blame]
Bram Moolenaare0874f82016-01-24 20:36:41 +01001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 */
8
9/*
10 * Implements communication through a socket or any file handle.
11 */
12
13#include "vim.h"
14
15#if defined(FEAT_CHANNEL) || defined(PROTO)
16
Bram Moolenaard04a0202016-01-26 23:30:18 +010017/* TRUE when netbeans is running with a GUI. */
18#ifdef FEAT_GUI
19# define CH_HAS_GUI (gui.in_use || gui.starting)
20#endif
21
22/* Note: when making changes here also adjust configure.in. */
23#ifdef WIN32
24/* WinSock API is separated from C API, thus we can't use read(), write(),
25 * errno... */
26# define SOCK_ERRNO errno = WSAGetLastError()
27# undef ECONNREFUSED
28# define ECONNREFUSED WSAECONNREFUSED
Bram Moolenaar4d919d72016-02-05 22:36:41 +010029# undef EWOULDBLOCK
30# define EWOULDBLOCK WSAEWOULDBLOCK
Bram Moolenaard04a0202016-01-26 23:30:18 +010031# ifdef EINTR
32# undef EINTR
33# endif
34# define EINTR WSAEINTR
Bram Moolenaard8070362016-02-15 21:56:54 +010035# define sock_write(sd, buf, len) send((SOCKET)sd, buf, len, 0)
36# define sock_read(sd, buf, len) recv((SOCKET)sd, buf, len, 0)
37# define sock_close(sd) closesocket((SOCKET)sd)
Bram Moolenaard04a0202016-01-26 23:30:18 +010038#else
39# include <netdb.h>
40# include <netinet/in.h>
41
42# include <sys/socket.h>
43# ifdef HAVE_LIBGEN_H
44# include <libgen.h>
45# endif
46# define SOCK_ERRNO
47# define sock_write(sd, buf, len) write(sd, buf, len)
48# define sock_read(sd, buf, len) read(sd, buf, len)
49# define sock_close(sd) close(sd)
Bram Moolenaar0943a092016-02-16 13:11:17 +010050# define fd_read(fd, buf, len) read(fd, buf, len)
Bram Moolenaard8070362016-02-15 21:56:54 +010051# define fd_write(sd, buf, len) write(sd, buf, len)
52# define fd_close(sd) close(sd)
Bram Moolenaard04a0202016-01-26 23:30:18 +010053#endif
54
Bram Moolenaard8070362016-02-15 21:56:54 +010055#ifdef WIN32
56 static int
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +010057fd_read(sock_T fd, char *buf, size_t len)
Bram Moolenaard8070362016-02-15 21:56:54 +010058{
59 HANDLE h = (HANDLE)fd;
60 DWORD nread;
61
62 if (!ReadFile(h, buf, (DWORD)len, &nread, NULL))
63 return -1;
64 return (int)nread;
65}
66
67 static int
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +010068fd_write(sock_T fd, char *buf, size_t len)
Bram Moolenaard8070362016-02-15 21:56:54 +010069{
70 HANDLE h = (HANDLE)fd;
71 DWORD nwrite;
72
73 if (!WriteFile(h, buf, (DWORD)len, &nwrite, NULL))
74 return -1;
75 return (int)nwrite;
76}
77
78 static void
79fd_close(sock_T fd)
80{
81 HANDLE h = (HANDLE)fd;
82
83 CloseHandle(h);
84}
85#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +010086
Bram Moolenaar6463ca22016-02-13 17:04:46 +010087/* Log file opened with ch_logfile(). */
88static FILE *log_fd = NULL;
Bram Moolenaar81661fb2016-02-18 22:23:34 +010089#ifdef FEAT_RELTIME
90static proftime_T log_start;
91#endif
Bram Moolenaar6463ca22016-02-13 17:04:46 +010092
93 void
94ch_logfile(FILE *file)
95{
96 if (log_fd != NULL)
97 fclose(log_fd);
98 log_fd = file;
99 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100100 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100101 fprintf(log_fd, "==== start log session ====\n");
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100102#ifdef FEAT_RELTIME
103 profile_start(&log_start);
104#endif
105 }
106}
107
108 int
109ch_log_active()
110{
111 return log_fd != NULL;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100112}
113
114 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100115ch_log_lead(char *what, channel_T *ch)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100116{
117 if (log_fd != NULL)
118 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100119#ifdef FEAT_RELTIME
120 proftime_T log_now;
121
122 profile_start(&log_now);
123 profile_sub(&log_now, &log_start);
124 fprintf(log_fd, "%s ", profile_msg(&log_now));
125#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100126 if (ch != NULL)
127 fprintf(log_fd, "%son %d: ", what, ch->ch_id);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100128 else
129 fprintf(log_fd, "%s: ", what);
130 }
131}
132
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100133 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100134ch_log(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100135{
136 if (log_fd != NULL)
137 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100138 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100139 fputs(msg, log_fd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100140 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100141 fflush(log_fd);
142 }
143}
144
145 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100146ch_logn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100147{
148 if (log_fd != NULL)
149 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100150 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100151 fprintf(log_fd, msg, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100152 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100153 fflush(log_fd);
154 }
155}
156
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100157 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100158ch_logs(channel_T *ch, char *msg, char *name)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100159{
160 if (log_fd != NULL)
161 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100162 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100163 fprintf(log_fd, msg, name);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100164 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100165 fflush(log_fd);
166 }
167}
168
169 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100170ch_logsn(channel_T *ch, char *msg, char *name, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100171{
172 if (log_fd != NULL)
173 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100174 ch_log_lead("", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100175 fprintf(log_fd, msg, name, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100176 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100177 fflush(log_fd);
178 }
179}
180
181 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100182ch_error(channel_T *ch, char *msg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100183{
184 if (log_fd != NULL)
185 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100186 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100187 fputs(msg, log_fd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100188 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100189 fflush(log_fd);
190 }
191}
192
193 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100194ch_errorn(channel_T *ch, char *msg, int nr)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100195{
196 if (log_fd != NULL)
197 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100198 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100199 fprintf(log_fd, msg, nr);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100200 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100201 fflush(log_fd);
202 }
203}
204
205 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100206ch_errors(channel_T *ch, char *msg, char *arg)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100207{
208 if (log_fd != NULL)
209 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100210 ch_log_lead("ERR ", ch);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100211 fprintf(log_fd, msg, arg);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100212 fputc('\n', log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100213 fflush(log_fd);
214 }
215}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100216
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100217#ifdef _WIN32
218# undef PERROR
219# define PERROR(msg) (void)emsg3((char_u *)"%s: %s", \
220 (char_u *)msg, (char_u *)strerror_win32(errno))
221
222 static char *
223strerror_win32(int eno)
224{
225 static LPVOID msgbuf = NULL;
226 char_u *ptr;
227
228 if (msgbuf)
229 LocalFree(msgbuf);
230 FormatMessage(
231 FORMAT_MESSAGE_ALLOCATE_BUFFER |
232 FORMAT_MESSAGE_FROM_SYSTEM |
233 FORMAT_MESSAGE_IGNORE_INSERTS,
234 NULL,
235 eno,
236 MAKELANGID(LANG_ENGLISH, SUBLANG_DEFAULT),
237 (LPTSTR) &msgbuf,
238 0,
239 NULL);
240 /* chomp \r or \n */
241 for (ptr = (char_u *)msgbuf; *ptr; ptr++)
242 switch (*ptr)
243 {
244 case '\r':
245 STRMOVE(ptr, ptr + 1);
246 ptr--;
247 break;
248 case '\n':
249 if (*(ptr + 1) == '\0')
250 *ptr = '\0';
251 else
252 *ptr = ' ';
253 break;
254 }
255 return msgbuf;
256}
257#endif
258
Bram Moolenaar77073442016-02-13 23:23:53 +0100259/*
260 * The list of all allocated channels.
261 */
262static channel_T *first_channel = NULL;
263static int next_ch_id = 0;
264
265/*
266 * Allocate a new channel. The refcount is set to 1.
267 * The channel isn't actually used until it is opened.
268 * Returns NULL if out of memory.
269 */
270 channel_T *
271add_channel(void)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100272{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100273 int part;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100274 channel_T *channel = (channel_T *)alloc_clear((int)sizeof(channel_T));
Bram Moolenaare0874f82016-01-24 20:36:41 +0100275
Bram Moolenaar77073442016-02-13 23:23:53 +0100276 if (channel == NULL)
277 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100278
Bram Moolenaar77073442016-02-13 23:23:53 +0100279 channel->ch_id = next_ch_id++;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100280 ch_log(channel, "Created channel");
Bram Moolenaar77073442016-02-13 23:23:53 +0100281
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100282#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100283 for (part = PART_SOCK; part <= PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100284#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100285 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100286#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100287 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100288 channel->ch_part[part].ch_fd = INVALID_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100289#ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100290 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100291#endif
292#ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100293 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100294#endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100295 channel->ch_part[part].ch_timeout = 2000;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100296 }
Bram Moolenaare0874f82016-01-24 20:36:41 +0100297
Bram Moolenaar77073442016-02-13 23:23:53 +0100298 if (first_channel != NULL)
299 {
300 first_channel->ch_prev = channel;
301 channel->ch_next = first_channel;
302 }
303 first_channel = channel;
304
305 channel->ch_refcount = 1;
306 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100307}
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100308
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100309/*
Bram Moolenaar46c85432016-02-26 11:17:46 +0100310 * Return TRUE if "channel" has a callback and the associated job wasn't
311 * killed.
Bram Moolenaarc8dcbb12016-02-25 23:10:17 +0100312 */
313 static int
Bram Moolenaar46c85432016-02-26 11:17:46 +0100314channel_still_useful(channel_T *channel)
Bram Moolenaarc8dcbb12016-02-25 23:10:17 +0100315{
Bram Moolenaar46c85432016-02-26 11:17:46 +0100316 if (channel->ch_job_killed && channel->ch_job == NULL)
317 return FALSE;
Bram Moolenaarc8dcbb12016-02-25 23:10:17 +0100318 return channel->ch_callback != NULL
319#ifdef CHANNEL_PIPES
320 || channel->ch_part[PART_OUT].ch_callback != NULL
321 || channel->ch_part[PART_ERR].ch_callback != NULL
322#endif
323 || channel->ch_close_cb != NULL;
324}
325
326/*
327 * Close a channel and free all its resources if there is no further action
Bram Moolenaar46c85432016-02-26 11:17:46 +0100328 * possible, there is no callback to be invoked or the associated job was
329 * killed.
Bram Moolenaarc8dcbb12016-02-25 23:10:17 +0100330 */
331 void
332channel_may_free(channel_T *channel)
333{
Bram Moolenaar46c85432016-02-26 11:17:46 +0100334 if (!channel_still_useful(channel))
Bram Moolenaarc8dcbb12016-02-25 23:10:17 +0100335 channel_free(channel);
336}
337
338/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100339 * Close a channel and free all its resources.
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100340 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100341 void
342channel_free(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100343{
Bram Moolenaar8b374212016-02-24 20:43:06 +0100344 channel_close(channel, TRUE);
Bram Moolenaar77073442016-02-13 23:23:53 +0100345 if (channel->ch_next != NULL)
346 channel->ch_next->ch_prev = channel->ch_prev;
347 if (channel->ch_prev == NULL)
348 first_channel = channel->ch_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100349 else
Bram Moolenaar77073442016-02-13 23:23:53 +0100350 channel->ch_prev->ch_next = channel->ch_next;
351 vim_free(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100352}
353
Bram Moolenaard04a0202016-01-26 23:30:18 +0100354#if defined(FEAT_GUI) || defined(PROTO)
Bram Moolenaar77073442016-02-13 23:23:53 +0100355
356#if defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK)
357 static void
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100358channel_read_fd(int fd)
Bram Moolenaar77073442016-02-13 23:23:53 +0100359{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100360 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100361 int part;
Bram Moolenaar77073442016-02-13 23:23:53 +0100362
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100363 channel = channel_fd2channel(fd, &part);
Bram Moolenaar77073442016-02-13 23:23:53 +0100364 if (channel == NULL)
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100365 ch_errorn(NULL, "Channel for fd %d not found", fd);
Bram Moolenaar77073442016-02-13 23:23:53 +0100366 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100367 channel_read(channel, part, "messageFromNetbeans");
Bram Moolenaar77073442016-02-13 23:23:53 +0100368}
369#endif
370
Bram Moolenaare0874f82016-01-24 20:36:41 +0100371/*
Bram Moolenaard04a0202016-01-26 23:30:18 +0100372 * Read a command from netbeans.
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100373 * TODO: instead of channel ID use the FD.
Bram Moolenaare0874f82016-01-24 20:36:41 +0100374 */
Bram Moolenaard04a0202016-01-26 23:30:18 +0100375#ifdef FEAT_GUI_X11
376 static void
377messageFromNetbeans(XtPointer clientData,
378 int *unused1 UNUSED,
379 XtInputId *unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100380{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100381 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100382}
Bram Moolenaard04a0202016-01-26 23:30:18 +0100383#endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100384
Bram Moolenaard04a0202016-01-26 23:30:18 +0100385#ifdef FEAT_GUI_GTK
Bram Moolenaar98921892016-02-23 17:14:37 +0100386# if GTK_CHECK_VERSION(3,0,0)
387 static gboolean
388messageFromNetbeans(GIOChannel *unused1 UNUSED,
389 GIOCondition unused2 UNUSED,
390 gpointer clientData)
391{
392 channel_read_fd(GPOINTER_TO_INT(clientData));
393 return TRUE; /* Return FALSE instead in case the event source is to
394 * be removed after this function returns. */
395}
396# else
Bram Moolenaard04a0202016-01-26 23:30:18 +0100397 static void
398messageFromNetbeans(gpointer clientData,
399 gint unused1 UNUSED,
400 GdkInputCondition unused2 UNUSED)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100401{
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100402 channel_read_fd((int)(long)clientData);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100403}
Bram Moolenaar98921892016-02-23 17:14:37 +0100404# endif
Bram Moolenaare0874f82016-01-24 20:36:41 +0100405#endif
406
407 static void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100408channel_gui_register_one(channel_T *channel, int part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100409{
410# ifdef FEAT_GUI_X11
411 /* Tell notifier we are interested in being called
412 * when there is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100413 if (channel->ch_part[part].ch_inputHandler == (XtInputId)NULL)
414 channel->ch_part[part].ch_inputHandler = XtAppAddInput(
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100415 (XtAppContext)app_context,
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100416 channel->ch_part[part].ch_fd,
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100417 (XtPointer)(XtInputReadMask + XtInputExceptMask),
418 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100419 (XtPointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100420# else
421# ifdef FEAT_GUI_GTK
422 /* Tell gdk we are interested in being called when there
423 * is input on the editor connection socket. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100424 if (channel->ch_part[part].ch_inputHandler == 0)
Bram Moolenaar98921892016-02-23 17:14:37 +0100425# if GTK_CHECK_VERSION(3,0,0)
426 {
427 GIOChannel *chnnl = g_io_channel_unix_new(
428 (gint)channel->ch_part[part].ch_fd);
429
430 channel->ch_part[part].ch_inputHandler = g_io_add_watch(
431 chnnl,
432 G_IO_IN|G_IO_HUP|G_IO_ERR|G_IO_PRI,
433 messageFromNetbeans,
434 GINT_TO_POINTER(channel->ch_part[part].ch_fd));
435
436 g_io_channel_unref(chnnl);
437 }
438# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100439 channel->ch_part[part].ch_inputHandler = gdk_input_add(
440 (gint)channel->ch_part[part].ch_fd,
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100441 (GdkInputCondition)
442 ((int)GDK_INPUT_READ + (int)GDK_INPUT_EXCEPTION),
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100443 messageFromNetbeans,
Bram Moolenaarfffd5562016-02-20 18:44:39 +0100444 (gpointer)(long)channel->ch_part[part].ch_fd);
Bram Moolenaar98921892016-02-23 17:14:37 +0100445# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100446# endif
447# endif
448}
449
450 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100451channel_gui_register(channel_T *channel)
Bram Moolenaare0874f82016-01-24 20:36:41 +0100452{
Bram Moolenaard04a0202016-01-26 23:30:18 +0100453 if (!CH_HAS_GUI)
454 return;
455
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100456 if (channel->CH_SOCK_FD != INVALID_FD)
457 channel_gui_register_one(channel, PART_SOCK);
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100458# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100459 if (channel->CH_OUT_FD != INVALID_FD)
460 channel_gui_register_one(channel, PART_OUT);
461 if (channel->CH_ERR_FD != INVALID_FD)
462 channel_gui_register_one(channel, PART_ERR);
Bram Moolenaare0874f82016-01-24 20:36:41 +0100463# endif
Bram Moolenaard04a0202016-01-26 23:30:18 +0100464}
465
466/*
467 * Register any of our file descriptors with the GUI event handling system.
468 * Called when the GUI has started.
469 */
470 void
471channel_gui_register_all(void)
472{
Bram Moolenaar77073442016-02-13 23:23:53 +0100473 channel_T *channel;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100474
Bram Moolenaar77073442016-02-13 23:23:53 +0100475 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100476 channel_gui_register(channel);
477}
478
479 static void
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100480channel_gui_unregister(channel_T *channel)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100481{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100482 int part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100483
484#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100485 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100486#else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100487 part = PART_SOCK;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100488#endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100489 {
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100490# ifdef FEAT_GUI_X11
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100491 if (channel->ch_part[part].ch_inputHandler != (XtInputId)NULL)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100492 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100493 XtRemoveInput(channel->ch_part[part].ch_inputHandler);
494 channel->ch_part[part].ch_inputHandler = (XtInputId)NULL;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100495 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100496# else
497# ifdef FEAT_GUI_GTK
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100498 if (channel->ch_part[part].ch_inputHandler != 0)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100499 {
Bram Moolenaar98921892016-02-23 17:14:37 +0100500# if GTK_CHECK_VERSION(3,0,0)
501 g_source_remove(channel->ch_part[part].ch_inputHandler);
502# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100503 gdk_input_remove(channel->ch_part[part].ch_inputHandler);
Bram Moolenaar98921892016-02-23 17:14:37 +0100504# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100505 channel->ch_part[part].ch_inputHandler = 0;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100506 }
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100507# endif
508# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +0100509 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100510}
511
512#endif
513
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100514static char *e_cannot_connect = N_("E902: Cannot connect to port");
515
Bram Moolenaard04a0202016-01-26 23:30:18 +0100516/*
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100517 * Open a socket channel to "hostname":"port".
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100518 * "waittime" is the time in msec to wait for the connection.
519 * When negative wait forever.
Bram Moolenaar77073442016-02-13 23:23:53 +0100520 * Returns the channel for success.
521 * Returns NULL for failure.
Bram Moolenaard04a0202016-01-26 23:30:18 +0100522 */
Bram Moolenaar77073442016-02-13 23:23:53 +0100523 channel_T *
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100524channel_open(
525 char *hostname,
526 int port_in,
527 int waittime,
528 void (*nb_close_cb)(void))
Bram Moolenaard04a0202016-01-26 23:30:18 +0100529{
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100530 int sd = -1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100531 struct sockaddr_in server;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100532 struct hostent *host;
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100533#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100534 u_short port = port_in;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100535 u_long val = 1;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100536#else
537 int port = port_in;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100538 struct timeval start_tv;
539 int so_error;
540 socklen_t so_error_len = sizeof(so_error);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100541#endif
Bram Moolenaar77073442016-02-13 23:23:53 +0100542 channel_T *channel;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100543 int ret;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100544
Bram Moolenaarf12d9832016-01-29 21:11:25 +0100545#ifdef WIN32
Bram Moolenaard04a0202016-01-26 23:30:18 +0100546 channel_init_winsock();
547#endif
548
Bram Moolenaar77073442016-02-13 23:23:53 +0100549 channel = add_channel();
550 if (channel == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100551 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100552 ch_error(NULL, "Cannot allocate channel.");
Bram Moolenaar77073442016-02-13 23:23:53 +0100553 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100554 }
555
556 /* Get the server internet address and put into addr structure */
557 /* fill in the socket address structure and connect to server */
558 vim_memset((char *)&server, 0, sizeof(server));
559 server.sin_family = AF_INET;
560 server.sin_port = htons(port);
561 if ((host = gethostbyname(hostname)) == NULL)
562 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100563 ch_error(channel, "in gethostbyname() in channel_open()");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100564 PERROR("E901: gethostbyname() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100565 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100566 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100567 }
568 memcpy((char *)&server.sin_addr, host->h_addr, host->h_length);
569
Bram Moolenaar254e00d2016-02-19 23:23:12 +0100570 /* On Mac and Solaris a zero timeout almost never works. At least wait
571 * one millisecond. Let's do it for all systems, because we don't know why
572 * this is needed. */
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100573 if (waittime == 0)
574 waittime = 1;
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100575
576 /*
577 * For Unix we need to call connect() again after connect() failed.
578 * On Win32 one time is sufficient.
579 */
580 while (TRUE)
581 {
582 if (sd >= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +0100583 sock_close(sd);
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100584 sd = socket(AF_INET, SOCK_STREAM, 0);
585 if (sd == -1)
586 {
587 ch_error(channel, "in socket() in channel_open().");
588 PERROR("E898: socket() in channel_open()");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +0100589 channel_free(channel);
Bram Moolenaar77073442016-02-13 23:23:53 +0100590 return NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100591 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100592
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100593 if (waittime >= 0)
594 {
595 /* Make connect() non-blocking. */
596 if (
597#ifdef _WIN32
598 ioctlsocket(sd, FIONBIO, &val) < 0
599#else
600 fcntl(sd, F_SETFL, O_NONBLOCK) < 0
601#endif
602 )
603 {
604 SOCK_ERRNO;
605 ch_errorn(channel,
606 "channel_open: Connect failed with errno %d", errno);
607 sock_close(sd);
608 channel_free(channel);
609 return NULL;
610 }
611 }
612
613 /* Try connecting to the server. */
614 ch_logsn(channel, "Connecting to %s port %d", hostname, port);
615 ret = connect(sd, (struct sockaddr *)&server, sizeof(server));
616
617 SOCK_ERRNO;
618 if (ret < 0)
619 {
620 if (errno != EWOULDBLOCK
621 && errno != ECONNREFUSED
622
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100623#ifdef EINPROGRESS
624 && errno != EINPROGRESS
625#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100626 )
627 {
628 ch_errorn(channel,
629 "channel_open: Connect failed with errno %d", errno);
630 PERROR(_(e_cannot_connect));
631 sock_close(sd);
632 channel_free(channel);
633 return NULL;
634 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100635 }
Bram Moolenaard04a0202016-01-26 23:30:18 +0100636
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100637 /* If we don't block and connect() failed then try using select() to
638 * wait for the connection to be made. */
639 if (waittime >= 0 && ret < 0)
640 {
641 struct timeval tv;
642 fd_set wfds;
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100643#if defined(__APPLE__) && __APPLE__ == 1
644# define PASS_RFDS
Bram Moolenaar98921892016-02-23 17:14:37 +0100645 fd_set rfds;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100646
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100647 FD_ZERO(&rfds);
648 FD_SET(sd, &rfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100649#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100650 FD_ZERO(&wfds);
651 FD_SET(sd, &wfds);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100652
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100653 tv.tv_sec = waittime / 1000;
654 tv.tv_usec = (waittime % 1000) * 1000;
655#ifndef WIN32
656 gettimeofday(&start_tv, NULL);
657#endif
658 ch_logn(channel,
659 "Waiting for connection (waittime %d msec)...", waittime);
660 ret = select((int)sd + 1,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100661#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100662 &rfds,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100663#else
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100664 NULL,
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100665#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100666 &wfds, NULL, &tv);
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100667
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100668 if (ret < 0)
669 {
670 SOCK_ERRNO;
671 ch_errorn(channel,
672 "channel_open: Connect failed with errno %d", errno);
673 PERROR(_(e_cannot_connect));
674 sock_close(sd);
675 channel_free(channel);
676 return NULL;
677 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100678#ifdef PASS_RFDS
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100679 if (ret == 0 && FD_ISSET(sd, &rfds) && FD_ISSET(sd, &wfds))
680 {
681 /* For OS X, this implies error. See tcp(4). */
682 ch_error(channel, "channel_open: Connect failed");
683 EMSG(_(e_cannot_connect));
684 sock_close(sd);
685 channel_free(channel);
686 return NULL;
687 }
Bram Moolenaare74e8e72016-02-16 22:01:30 +0100688#endif
Bram Moolenaar81661fb2016-02-18 22:23:34 +0100689#ifdef WIN32
690 /* On Win32 select() is expected to work and wait for up to the
691 * waittime for the socket to be open. */
692 if (!FD_ISSET(sd, &wfds) || ret == 0)
693#else
694 /* See socket(7) for the behavior on Linux-like systems:
695 * After putting the socket in non-blocking mode, connect() will
696 * return EINPROGRESS, select() will not wait (as if writing is
697 * possible), need to use getsockopt() to check if the socket is
698 * actually open. */
699 getsockopt(sd, SOL_SOCKET, SO_ERROR, &so_error, &so_error_len);
700 if (!FD_ISSET(sd, &wfds) || ret == 0 || so_error != 0)
701#endif
702 {
703#ifndef WIN32
704 struct timeval end_tv;
705 long elapsed_msec;
706
707 gettimeofday(&end_tv, NULL);
708 elapsed_msec = (end_tv.tv_sec - start_tv.tv_sec) * 1000
709 + (end_tv.tv_usec - start_tv.tv_usec) / 1000;
710 if (waittime > 1 && elapsed_msec < waittime)
711 {
712 /* The port isn't ready but we also didn't get an error.
713 * This happens when the server didn't open the socket
714 * yet. Wait a bit and try again. */
715 mch_delay(waittime < 50 ? (long)waittime : 50L, TRUE);
716 ui_breakcheck();
717 if (!got_int)
718 {
719 /* reduce the waittime by the elapsed time and the 50
720 * msec delay (or a bit more) */
721 waittime -= elapsed_msec;
722 if (waittime > 50)
723 waittime -= 50;
724 else
725 waittime = 1;
726 continue;
727 }
728 /* we were interrupted, behave as if timed out */
729 }
730#endif
731 /* We timed out. */
732 ch_error(channel, "Connection timed out");
733 sock_close(sd);
734 channel_free(channel);
735 return NULL;
736 }
737
738 ch_log(channel, "Connection made");
739 break;
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100740 }
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100741 }
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100742
Bram Moolenaar7a84dbe2016-02-07 21:29:00 +0100743 if (waittime >= 0)
744 {
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100745#ifdef _WIN32
746 val = 0;
747 ioctlsocket(sd, FIONBIO, &val);
748#else
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100749 (void)fcntl(sd, F_SETFL, 0);
Bram Moolenaar4d919d72016-02-05 22:36:41 +0100750#endif
751 }
752
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100753 channel->CH_SOCK_FD = (sock_T)sd;
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100754 channel->ch_nb_close_cb = nb_close_cb;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100755
756#ifdef FEAT_GUI
Bram Moolenaar77073442016-02-13 23:23:53 +0100757 channel_gui_register(channel);
Bram Moolenaard04a0202016-01-26 23:30:18 +0100758#endif
759
Bram Moolenaar77073442016-02-13 23:23:53 +0100760 return channel;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100761}
762
763#if defined(CHANNEL_PIPES) || defined(PROTO)
764 void
Bram Moolenaard8070362016-02-15 21:56:54 +0100765channel_set_pipes(channel_T *channel, sock_T in, sock_T out, sock_T err)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100766{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100767 channel->CH_IN_FD = in;
768 channel->CH_OUT_FD = out;
769 channel->CH_ERR_FD = err;
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100770}
771#endif
772
773 void
Bram Moolenaar77073442016-02-13 23:23:53 +0100774channel_set_job(channel_T *channel, job_T *job)
Bram Moolenaar6463ca22016-02-13 17:04:46 +0100775{
Bram Moolenaar77073442016-02-13 23:23:53 +0100776 channel->ch_job = job;
Bram Moolenaard04a0202016-01-26 23:30:18 +0100777}
778
779/*
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100780 * Set various properties from an "opt" argument.
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100781 */
782 void
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100783channel_set_options(channel_T *channel, jobopt_T *opt)
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100784{
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100785 int part;
786 char_u **cbp;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100787
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100788 if (opt->jo_set & JO_MODE)
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100789 for (part = PART_SOCK; part <= PART_IN; ++part)
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100790 channel->ch_part[part].ch_mode = opt->jo_mode;
791 if (opt->jo_set & JO_IN_MODE)
792 channel->ch_part[PART_IN].ch_mode = opt->jo_in_mode;
793 if (opt->jo_set & JO_OUT_MODE)
794 channel->ch_part[PART_OUT].ch_mode = opt->jo_out_mode;
795 if (opt->jo_set & JO_ERR_MODE)
796 channel->ch_part[PART_ERR].ch_mode = opt->jo_err_mode;
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100797
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100798 if (opt->jo_set & JO_TIMEOUT)
799 for (part = PART_SOCK; part <= PART_IN; ++part)
800 channel->ch_part[part].ch_timeout = opt->jo_timeout;
801 if (opt->jo_set & JO_OUT_TIMEOUT)
802 channel->ch_part[PART_OUT].ch_timeout = opt->jo_out_timeout;
803 if (opt->jo_set & JO_ERR_TIMEOUT)
804 channel->ch_part[PART_ERR].ch_timeout = opt->jo_err_timeout;
805
806 if (opt->jo_set & JO_CALLBACK)
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100807 {
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100808 cbp = &channel->ch_callback;
809 vim_free(*cbp);
810 if (opt->jo_callback != NULL && *opt->jo_callback != NUL)
811 *cbp = vim_strsave(opt->jo_callback);
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100812 else
Bram Moolenaarb6b52522016-02-20 23:30:07 +0100813 *cbp = NULL;
814 }
815 if (opt->jo_set & JO_OUT_CALLBACK)
816 {
817 cbp = &channel->ch_part[PART_OUT].ch_callback;
818 vim_free(*cbp);
819 if (opt->jo_out_cb != NULL && *opt->jo_out_cb != NUL)
820 *cbp = vim_strsave(opt->jo_out_cb);
821 else
822 *cbp = NULL;
823 }
824 if (opt->jo_set & JO_ERR_CALLBACK)
825 {
826 cbp = &channel->ch_part[PART_ERR].ch_callback;
827 vim_free(*cbp);
828 if (opt->jo_err_cb != NULL && *opt->jo_err_cb != NUL)
829 *cbp = vim_strsave(opt->jo_err_cb);
830 else
831 *cbp = NULL;
Bram Moolenaar0ba75a92016-02-19 23:21:26 +0100832 }
Bram Moolenaar4e221c92016-02-23 13:20:22 +0100833 if (opt->jo_set & JO_CLOSE_CALLBACK)
834 {
835 cbp = &channel->ch_close_cb;
836 vim_free(*cbp);
837 if (opt->jo_close_cb != NULL && *opt->jo_close_cb != NUL)
838 *cbp = vim_strsave(opt->jo_close_cb);
839 else
840 *cbp = NULL;
841 }
Bram Moolenaar910b8aa2016-02-16 21:03:07 +0100842}
843
844/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100845 * Set the callback for "channel"/"part" for the response with "id".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100846 */
847 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100848channel_set_req_callback(
849 channel_T *channel,
850 int part,
851 char_u *callback,
852 int id)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100853{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100854 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100855 cbq_T *item = (cbq_T *)alloc((int)sizeof(cbq_T));
856
857 if (item != NULL)
858 {
Bram Moolenaar77073442016-02-13 23:23:53 +0100859 item->cq_callback = vim_strsave(callback);
860 item->cq_seq_nr = id;
861 item->cq_prev = head->cq_prev;
862 head->cq_prev = item;
863 item->cq_next = NULL;
864 if (item->cq_prev == NULL)
865 head->cq_next = item;
866 else
867 item->cq_prev->cq_next = item;
Bram Moolenaara07fec92016-02-05 21:04:08 +0100868 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100869}
870
871/*
Bram Moolenaar77073442016-02-13 23:23:53 +0100872 * Invoke the "callback" on channel "channel".
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100873 */
874 static void
Bram Moolenaar77073442016-02-13 23:23:53 +0100875invoke_callback(channel_T *channel, char_u *callback, typval_T *argv)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100876{
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100877 typval_T rettv;
878 int dummy;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100879
Bram Moolenaar77073442016-02-13 23:23:53 +0100880 argv[0].v_type = VAR_CHANNEL;
881 argv[0].vval.v_channel = channel;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +0100882
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100883 call_func(callback, (int)STRLEN(callback),
884 &rettv, 2, argv, 0L, 0L, &dummy, TRUE, NULL);
Bram Moolenaaree1cffc2016-02-21 19:14:41 +0100885 clear_tv(&rettv);
886
Bram Moolenaar20fb9f32016-01-30 23:20:33 +0100887 /* If an echo command was used the cursor needs to be put back where
888 * it belongs. */
889 setcursor();
890 cursor_on();
891 out_flush();
892}
893
Bram Moolenaarfb1f6262016-01-31 20:24:32 +0100894/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100895 * Return the first buffer from channel "channel"/"part" and remove it.
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100896 * The caller must free it.
897 * Returns NULL if there is nothing.
898 */
899 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100900channel_get(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100901{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100902 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100903 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100904 char_u *p;
905
Bram Moolenaar77073442016-02-13 23:23:53 +0100906 if (node == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100907 return NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100908 /* dispose of the node but keep the buffer */
Bram Moolenaar77073442016-02-13 23:23:53 +0100909 p = node->rq_buffer;
910 head->rq_next = node->rq_next;
911 if (node->rq_next == NULL)
912 head->rq_prev = NULL;
913 else
914 node->rq_next->rq_prev = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100915 vim_free(node);
916 return p;
917}
918
919/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100920 * Returns the whole buffer contents concatenated for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100921 */
922 static char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100923channel_get_all(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100924{
925 /* Concatenate everything into one buffer.
926 * TODO: avoid multiple allocations. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100927 while (channel_collapse(channel, part) == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100928 ;
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100929 return channel_get(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100930}
931
932/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100933 * Collapses the first and second buffer for "channel"/"part".
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100934 * Returns FAIL if that is not possible.
935 */
936 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100937channel_collapse(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100938{
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100939 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaar77073442016-02-13 23:23:53 +0100940 readq_T *node = head->rq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100941 char_u *p;
942
Bram Moolenaar77073442016-02-13 23:23:53 +0100943 if (node == NULL || node->rq_next == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100944 return FAIL;
945
Bram Moolenaar77073442016-02-13 23:23:53 +0100946 p = alloc((unsigned)(STRLEN(node->rq_buffer)
947 + STRLEN(node->rq_next->rq_buffer) + 1));
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100948 if (p == NULL)
949 return FAIL; /* out of memory */
Bram Moolenaar77073442016-02-13 23:23:53 +0100950 STRCPY(p, node->rq_buffer);
951 STRCAT(p, node->rq_next->rq_buffer);
952 vim_free(node->rq_next->rq_buffer);
953 node->rq_next->rq_buffer = p;
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100954
Bram Moolenaar77073442016-02-13 23:23:53 +0100955 /* dispose of the node and its buffer */
956 head->rq_next = node->rq_next;
957 head->rq_next->rq_prev = NULL;
958 vim_free(node->rq_buffer);
Bram Moolenaar19d2f152016-02-01 21:38:19 +0100959 vim_free(node);
960 return OK;
961}
962
963/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +0100964 * Store "buf[len]" on "channel"/"part".
965 * Returns OK or FAIL.
966 */
967 static int
968channel_save(channel_T *channel, int part, char_u *buf, int len)
969{
970 readq_T *node;
971 readq_T *head = &channel->ch_part[part].ch_head;
972 char_u *p;
973 int i;
974
975 node = (readq_T *)alloc(sizeof(readq_T));
976 if (node == NULL)
977 return FAIL; /* out of memory */
978 node->rq_buffer = alloc(len + 1);
979 if (node->rq_buffer == NULL)
980 {
981 vim_free(node);
982 return FAIL; /* out of memory */
983 }
984
985 if (channel->ch_part[part].ch_mode == MODE_NL)
986 {
987 /* Drop any CR before a NL. */
988 p = node->rq_buffer;
989 for (i = 0; i < len; ++i)
990 if (buf[i] != CAR || i + 1 >= len || buf[i + 1] != NL)
991 *p++ = buf[i];
992 *p = NUL;
993 }
994 else
995 {
996 mch_memmove(node->rq_buffer, buf, len);
997 node->rq_buffer[len] = NUL;
998 }
999
1000 /* append node to the tail of the queue */
1001 node->rq_next = NULL;
1002 node->rq_prev = head->rq_prev;
1003 if (head->rq_prev == NULL)
1004 head->rq_next = node;
1005 else
1006 head->rq_prev->rq_next = node;
1007 head->rq_prev = node;
1008
1009 if (log_fd != NULL)
1010 {
1011 ch_log_lead("RECV ", channel);
1012 fprintf(log_fd, "'");
1013 if (fwrite(buf, len, 1, log_fd) != 1)
1014 return FAIL;
1015 fprintf(log_fd, "'\n");
1016 }
1017 return OK;
1018}
1019
1020/*
1021 * Use the read buffer of "channel"/"part" and parse a JSON messages that is
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001022 * complete. The messages are added to the queue.
Bram Moolenaard7ece102016-02-02 23:23:02 +01001023 * Return TRUE if there is more to read.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001024 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001025 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001026channel_parse_json(channel_T *channel, int part)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001027{
1028 js_read_T reader;
1029 typval_T listtv;
1030 jsonq_T *item;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001031 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001032 int ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001033
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001034 if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001035 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001036
1037 /* TODO: make reader work properly */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001038 /* reader.js_buf = channel_peek(channel, part); */
1039 reader.js_buf = channel_get_all(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001040 reader.js_used = 0;
Bram Moolenaar56ead342016-02-02 18:20:08 +01001041 reader.js_fill = NULL;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001042 /* reader.js_fill = channel_fill; */
Bram Moolenaar77073442016-02-13 23:23:53 +01001043 reader.js_cookie = channel;
Bram Moolenaar595e64e2016-02-07 19:19:53 +01001044 ret = json_decode(&reader, &listtv,
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001045 channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001046 if (ret == OK)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001047 {
Bram Moolenaar6076fe12016-02-05 22:49:56 +01001048 /* Only accept the response when it is a list with at least two
1049 * items. */
1050 if (listtv.v_type != VAR_LIST || listtv.vval.v_list->lv_len < 2)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001051 {
1052 /* TODO: give error */
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001053 clear_tv(&listtv);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001054 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001055 else
1056 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001057 item = (jsonq_T *)alloc((unsigned)sizeof(jsonq_T));
1058 if (item == NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001059 clear_tv(&listtv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001060 else
1061 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001062 item->jq_value = alloc_tv();
1063 if (item->jq_value == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001064 {
1065 vim_free(item);
1066 clear_tv(&listtv);
1067 }
1068 else
1069 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001070 *item->jq_value = listtv;
1071 item->jq_prev = head->jq_prev;
1072 head->jq_prev = item;
1073 item->jq_next = NULL;
1074 if (item->jq_prev == NULL)
1075 head->jq_next = item;
1076 else
1077 item->jq_prev->jq_next = item;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001078 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001079 }
1080 }
1081 }
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001082
1083 /* Put the unread part back into the channel.
1084 * TODO: insert in front */
1085 if (reader.js_buf[reader.js_used] != NUL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001086 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001087 channel_save(channel, part, reader.js_buf + reader.js_used,
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001088 (int)(reader.js_end - reader.js_buf) - reader.js_used);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001089 ret = TRUE;
1090 }
1091 else
1092 ret = FALSE;
1093
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001094 vim_free(reader.js_buf);
Bram Moolenaard7ece102016-02-02 23:23:02 +01001095 return ret;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001096}
1097
1098/*
Bram Moolenaard46ae142016-02-16 13:33:52 +01001099 * Remove "node" from the queue that it is in. Does not free it.
Bram Moolenaara07fec92016-02-05 21:04:08 +01001100 */
1101 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001102remove_cb_node(cbq_T *head, cbq_T *node)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001103{
Bram Moolenaar77073442016-02-13 23:23:53 +01001104 if (node->cq_prev == NULL)
1105 head->cq_next = node->cq_next;
1106 else
1107 node->cq_prev->cq_next = node->cq_next;
1108 if (node->cq_next == NULL)
1109 head->cq_prev = node->cq_prev;
1110 else
1111 node->cq_next->cq_prev = node->cq_prev;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001112}
1113
1114/*
1115 * Remove "node" from the queue that it is in and free it.
Bram Moolenaar77073442016-02-13 23:23:53 +01001116 * Caller should have freed or used node->jq_value.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001117 */
1118 static void
Bram Moolenaar77073442016-02-13 23:23:53 +01001119remove_json_node(jsonq_T *head, jsonq_T *node)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001120{
Bram Moolenaar77073442016-02-13 23:23:53 +01001121 if (node->jq_prev == NULL)
1122 head->jq_next = node->jq_next;
1123 else
1124 node->jq_prev->jq_next = node->jq_next;
1125 if (node->jq_next == NULL)
1126 head->jq_prev = node->jq_prev;
1127 else
1128 node->jq_next->jq_prev = node->jq_prev;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001129 vim_free(node);
1130}
1131
1132/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001133 * Get a message from the JSON queue for channel "channel".
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001134 * When "id" is positive it must match the first number in the list.
Bram Moolenaare56bf152016-02-08 23:23:42 +01001135 * When "id" is zero or negative jut get the first message. But not the one
1136 * with id ch_block_id.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001137 * Return OK when found and return the value in "rettv".
1138 * Return FAIL otherwise.
1139 */
1140 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001141channel_get_json(channel_T *channel, int part, int id, typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001142{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001143 jsonq_T *head = &channel->ch_part[part].ch_json_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001144 jsonq_T *item = head->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001145
Bram Moolenaar77073442016-02-13 23:23:53 +01001146 while (item != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001147 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001148 list_T *l = item->jq_value->vval.v_list;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001149 typval_T *tv = &l->lv_first->li_tv;
1150
1151 if ((id > 0 && tv->v_type == VAR_NUMBER && tv->vval.v_number == id)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001152 || (id <= 0 && (tv->v_type != VAR_NUMBER
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001153 || tv->vval.v_number == 0
1154 || tv->vval.v_number != channel->ch_part[part].ch_block_id)))
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001155 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001156 *rettv = item->jq_value;
1157 remove_json_node(head, item);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001158 return OK;
1159 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001160 item = item->jq_next;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001161 }
1162 return FAIL;
1163}
1164
Bram Moolenaarece61b02016-02-20 21:39:05 +01001165#define CH_JSON_MAX_ARGS 4
1166
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001167/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001168 * Execute a command received over "channel"/"part"
Bram Moolenaarece61b02016-02-20 21:39:05 +01001169 * "argv[0]" is the command string.
1170 * "argv[1]" etc. have further arguments, type is VAR_UNKNOWN if missing.
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001171 */
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001172 static void
Bram Moolenaarece61b02016-02-20 21:39:05 +01001173channel_exe_cmd(channel_T *channel, int part, typval_T *argv)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001174{
Bram Moolenaarece61b02016-02-20 21:39:05 +01001175 char_u *cmd = argv[0].vval.v_string;
1176 char_u *arg;
1177 int options = channel->ch_part[part].ch_mode == MODE_JS ? JSON_JS : 0;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001178
Bram Moolenaarece61b02016-02-20 21:39:05 +01001179 if (argv[1].v_type != VAR_STRING)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001180 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001181 ch_error(channel, "received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001182 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001183 EMSG("E903: received command with non-string argument");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001184 return;
1185 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001186 arg = argv[1].vval.v_string;
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001187 if (arg == NULL)
1188 arg = (char_u *)"";
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001189
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001190 if (STRCMP(cmd, "ex") == 0)
1191 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001192 do_cmdline_cmd(arg);
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001193 }
1194 else if (STRCMP(cmd, "normal") == 0)
1195 {
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001196 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001197
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001198 ea.arg = arg;
1199 ea.addr_count = 0;
1200 ea.forceit = TRUE; /* no mapping */
1201 ex_normal(&ea);
1202 }
1203 else if (STRCMP(cmd, "redraw") == 0)
1204 {
1205 exarg_T ea;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001206
Bram Moolenaar14ad6112016-02-01 21:47:13 +01001207 ea.forceit = *arg != NUL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001208 ex_redraw(&ea);
1209 showruler(FALSE);
1210 setcursor();
1211 out_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001212#ifdef FEAT_GUI
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001213 if (gui.in_use)
1214 {
1215 gui_update_cursor(FALSE, FALSE);
1216 gui_mch_flush();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001217 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001218#endif
1219 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001220 else if (STRCMP(cmd, "expr") == 0 || STRCMP(cmd, "call") == 0)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001221 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001222 int is_call = cmd[0] == 'c';
1223 int id_idx = is_call ? 3 : 2;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001224
Bram Moolenaarece61b02016-02-20 21:39:05 +01001225 if (argv[id_idx].v_type != VAR_UNKNOWN
1226 && argv[id_idx].v_type != VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001227 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001228 ch_error(channel, "last argument for expr/call must be a number");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001229 if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001230 EMSG("E904: last argument for expr/call must be a number");
1231 }
1232 else if (is_call && argv[2].v_type != VAR_LIST)
1233 {
1234 ch_error(channel, "third argument for call must be a list");
1235 if (p_verbose > 2)
1236 EMSG("E904: third argument for call must be a list");
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001237 }
1238 else
1239 {
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001240 typval_T *tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001241 typval_T res_tv;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001242 typval_T err_tv;
Bram Moolenaar55fab432016-02-07 16:53:13 +01001243 char_u *json = NULL;
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001244
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001245 /* Don't pollute the display with errors. */
1246 ++emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001247 if (!is_call)
1248 tv = eval_expr(arg, NULL);
1249 else if (func_call(arg, &argv[2], NULL, &res_tv) == OK)
1250 tv = &res_tv;
1251 else
1252 tv = NULL;
1253
1254 if (argv[id_idx].v_type == VAR_NUMBER)
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001255 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001256 int id = argv[id_idx].vval.v_number;
1257
Bram Moolenaar55fab432016-02-07 16:53:13 +01001258 if (tv != NULL)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001259 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001260 if (tv == NULL || (json != NULL && *json == NUL))
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001261 {
Bram Moolenaar55fab432016-02-07 16:53:13 +01001262 /* If evaluation failed or the result can't be encoded
1263 * then return the string "ERROR". */
Bram Moolenaar77073442016-02-13 23:23:53 +01001264 vim_free(json);
1265 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001266 err_tv.v_type = VAR_STRING;
1267 err_tv.vval.v_string = (char_u *)"ERROR";
1268 tv = &err_tv;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001269 json = json_encode_nr_expr(id, tv, options);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001270 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001271 if (json != NULL)
1272 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001273 channel_send(channel,
1274 part == PART_SOCK ? PART_SOCK : PART_IN,
1275 json, (char *)cmd);
Bram Moolenaar55fab432016-02-07 16:53:13 +01001276 vim_free(json);
1277 }
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001278 }
Bram Moolenaar55fab432016-02-07 16:53:13 +01001279 --emsg_skip;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001280 if (tv == &res_tv)
1281 clear_tv(tv);
1282 else if (tv != &err_tv)
Bram Moolenaarfcb1e3d2016-02-03 21:32:46 +01001283 free_tv(tv);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001284 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001285 }
1286 else if (p_verbose > 2)
Bram Moolenaarece61b02016-02-20 21:39:05 +01001287 {
1288 ch_errors(channel, "Receved unknown command: %s", (char *)cmd);
Bram Moolenaarfb1f6262016-01-31 20:24:32 +01001289 EMSG2("E905: received unknown command: %s", cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001290 }
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001291}
1292
1293/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001294 * Invoke a callback for "channel"/"part" if needed.
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001295 * Return TRUE when a message was handled, there might be another one.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001296 */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001297 static int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001298may_invoke_callback(channel_T *channel, int part)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001299{
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001300 char_u *msg = NULL;
1301 typval_T *listtv = NULL;
Bram Moolenaarece61b02016-02-20 21:39:05 +01001302 typval_T argv[CH_JSON_MAX_ARGS];
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001303 int seq_nr = -1;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001304 ch_mode_T ch_mode = channel->ch_part[part].ch_mode;
1305 char_u *callback = NULL;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001306
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001307 if (channel->ch_nb_close_cb != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001308 /* this channel is handled elsewhere (netbeans) */
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001309 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001310
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001311 if (channel->ch_part[part].ch_callback != NULL)
1312 callback = channel->ch_part[part].ch_callback;
1313 else
1314 callback = channel->ch_callback;
1315
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001316 if (ch_mode == MODE_JSON || ch_mode == MODE_JS)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001317 {
Bram Moolenaarece61b02016-02-20 21:39:05 +01001318 listitem_T *item;
1319 int argc = 0;
1320
Bram Moolenaard7ece102016-02-02 23:23:02 +01001321 /* Get any json message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001322 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001323 {
Bram Moolenaard7ece102016-02-02 23:23:02 +01001324 /* Parse readahead, return when there is still no message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001325 channel_parse_json(channel, part);
1326 if (channel_get_json(channel, part, -1, &listtv) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001327 return FALSE;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001328 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001329
Bram Moolenaarece61b02016-02-20 21:39:05 +01001330 for (item = listtv->vval.v_list->lv_first;
1331 item != NULL && argc < CH_JSON_MAX_ARGS;
1332 item = item->li_next)
1333 argv[argc++] = item->li_tv;
1334 while (argc < CH_JSON_MAX_ARGS)
1335 argv[argc++].v_type = VAR_UNKNOWN;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001336
Bram Moolenaarece61b02016-02-20 21:39:05 +01001337 if (argv[0].v_type == VAR_STRING)
1338 {
1339 char_u *cmd = argv[0].vval.v_string;
1340
1341 /* ["cmd", arg] or ["cmd", arg, arg] or ["cmd", arg, arg, arg] */
Bram Moolenaar77073442016-02-13 23:23:53 +01001342 ch_logs(channel, "Executing %s command", (char *)cmd);
Bram Moolenaarece61b02016-02-20 21:39:05 +01001343 channel_exe_cmd(channel, part, argv);
Bram Moolenaar77073442016-02-13 23:23:53 +01001344 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001345 return TRUE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001346 }
1347
Bram Moolenaarece61b02016-02-20 21:39:05 +01001348 if (argv[0].v_type != VAR_NUMBER)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001349 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001350 ch_error(channel,
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001351 "Dropping message with invalid sequence number type");
Bram Moolenaar77073442016-02-13 23:23:53 +01001352 free_tv(listtv);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001353 return FALSE;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001354 }
Bram Moolenaarece61b02016-02-20 21:39:05 +01001355 seq_nr = argv[0].vval.v_number;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001356 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001357 else if (channel_peek(channel, part) == NULL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001358 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001359 /* nothing to read on RAW or NL channel */
Bram Moolenaard7ece102016-02-02 23:23:02 +01001360 return FALSE;
1361 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001362 else
1363 {
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001364 /* If there is no callback drop the message. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001365 if (callback == NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001366 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001367 while ((msg = channel_get(channel, part)) != NULL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001368 vim_free(msg);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001369 return FALSE;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001370 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001371
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001372 if (ch_mode == MODE_NL)
1373 {
1374 char_u *nl;
1375 char_u *buf;
1376
1377 /* See if we have a message ending in NL in the first buffer. If
1378 * not try to concatenate the first and the second buffer. */
1379 while (TRUE)
1380 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001381 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001382 nl = vim_strchr(buf, NL);
1383 if (nl != NULL)
1384 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001385 if (channel_collapse(channel, part) == FAIL)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001386 return FALSE; /* incomplete message */
1387 }
1388 if (nl[1] == NUL)
1389 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001390 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001391 else
1392 {
1393 /* Copy the message into allocated memory and remove it from
1394 * the buffer. */
1395 msg = vim_strnsave(buf, (int)(nl - buf));
1396 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1397 }
1398 }
1399 else
1400 /* For a raw channel we don't know where the message ends, just
1401 * get everything we have. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001402 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001403
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001404 argv[1].v_type = VAR_STRING;
1405 argv[1].vval.v_string = msg;
1406 }
1407
Bram Moolenaara07fec92016-02-05 21:04:08 +01001408 if (seq_nr > 0)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001409 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001410 cbq_T *head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaar77073442016-02-13 23:23:53 +01001411 cbq_T *item = head->cq_next;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001412 int done = FALSE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001413
1414 /* invoke the one-time callback with the matching nr */
Bram Moolenaar77073442016-02-13 23:23:53 +01001415 while (item != NULL)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001416 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001417 if (item->cq_seq_nr == seq_nr)
Bram Moolenaara07fec92016-02-05 21:04:08 +01001418 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001419 ch_log(channel, "Invoking one-time callback");
Bram Moolenaard46ae142016-02-16 13:33:52 +01001420 /* Remove the item from the list first, if the callback
1421 * invokes ch_close() the list will be cleared. */
Bram Moolenaar77073442016-02-13 23:23:53 +01001422 remove_cb_node(head, item);
Bram Moolenaard46ae142016-02-16 13:33:52 +01001423 invoke_callback(channel, item->cq_callback, argv);
1424 vim_free(item->cq_callback);
1425 vim_free(item);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001426 done = TRUE;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001427 break;
1428 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001429 item = item->cq_next;
Bram Moolenaara07fec92016-02-05 21:04:08 +01001430 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001431 if (!done)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001432 ch_log(channel, "Dropping message without callback");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001433 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001434 else if (callback != NULL)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001435 {
1436 /* invoke the channel callback */
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001437 ch_log(channel, "Invoking channel callback");
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001438 invoke_callback(channel, callback, argv);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001439 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001440 else
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001441 ch_log(channel, "Dropping message");
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01001442
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001443 if (listtv != NULL)
Bram Moolenaar77073442016-02-13 23:23:53 +01001444 free_tv(listtv);
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001445 vim_free(msg);
Bram Moolenaardf5b27b2016-02-02 18:43:17 +01001446
1447 return TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001448}
1449
1450/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001451 * Return TRUE when channel "channel" is open for writing to.
1452 * Also returns FALSE or invalid "channel".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001453 */
1454 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001455channel_can_write_to(channel_T *channel)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001456{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001457 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001458#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001459 || channel->CH_IN_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001460#endif
1461 );
Bram Moolenaard04a0202016-01-26 23:30:18 +01001462}
1463
1464/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001465 * Return TRUE when channel "channel" is open for reading or writing.
1466 * Also returns FALSE for invalid "channel".
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001467 */
1468 int
Bram Moolenaar77073442016-02-13 23:23:53 +01001469channel_is_open(channel_T *channel)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001470{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001471 return channel != NULL && (channel->CH_SOCK_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001472#ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001473 || channel->CH_IN_FD != INVALID_FD
1474 || channel->CH_OUT_FD != INVALID_FD
1475 || channel->CH_ERR_FD != INVALID_FD
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001476#endif
1477 );
1478}
1479
1480/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001481 * Return a string indicating the status of the channel.
1482 */
1483 char *
1484channel_status(channel_T *channel)
1485{
1486 if (channel == NULL)
1487 return "fail";
1488 if (channel_is_open(channel))
1489 return "open";
1490 return "closed";
1491}
1492
1493/*
1494 * Close channel "channel".
Bram Moolenaarc8dcbb12016-02-25 23:10:17 +01001495 * Trigger the close callback if "invoke_close_cb" is TRUE.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001496 */
1497 void
Bram Moolenaar8b374212016-02-24 20:43:06 +01001498channel_close(channel_T *channel, int invoke_close_cb)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001499{
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001500 ch_log(channel, "Closing channel");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001501
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001502#ifdef FEAT_GUI
1503 channel_gui_unregister(channel);
1504#endif
1505
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001506 if (channel->CH_SOCK_FD != INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001507 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001508 sock_close(channel->CH_SOCK_FD);
1509 channel->CH_SOCK_FD = INVALID_FD;
Bram Moolenaare0874f82016-01-24 20:36:41 +01001510 }
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001511#if defined(CHANNEL_PIPES)
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001512 if (channel->CH_IN_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001513 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001514 fd_close(channel->CH_IN_FD);
1515 channel->CH_IN_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001516 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001517 if (channel->CH_OUT_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001518 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001519 fd_close(channel->CH_OUT_FD);
1520 channel->CH_OUT_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001521 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001522 if (channel->CH_ERR_FD != INVALID_FD)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001523 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001524 fd_close(channel->CH_ERR_FD);
1525 channel->CH_ERR_FD = INVALID_FD;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001526 }
1527#endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001528
Bram Moolenaar8b374212016-02-24 20:43:06 +01001529 if (invoke_close_cb && channel->ch_close_cb != NULL)
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001530 {
1531 typval_T argv[1];
1532 typval_T rettv;
1533 int dummy;
1534
1535 /* invoke the close callback; increment the refcount to avoid it
1536 * being freed halfway */
1537 argv[0].v_type = VAR_CHANNEL;
1538 argv[0].vval.v_channel = channel;
1539 ++channel->ch_refcount;
1540 call_func(channel->ch_close_cb, (int)STRLEN(channel->ch_close_cb),
1541 &rettv, 1, argv, 0L, 0L, &dummy, TRUE, NULL);
1542 clear_tv(&rettv);
1543 --channel->ch_refcount;
1544
1545 /* the callback is only called once */
1546 vim_free(channel->ch_close_cb);
1547 channel->ch_close_cb = NULL;
1548 }
1549
1550 channel->ch_nb_close_cb = NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001551 channel_clear(channel);
Bram Moolenaare0874f82016-01-24 20:36:41 +01001552}
1553
Bram Moolenaard04a0202016-01-26 23:30:18 +01001554/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001555 * Return the first buffer from "channel"/"part" without removing it.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001556 * Returns NULL if there is nothing.
1557 */
1558 char_u *
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001559channel_peek(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001560{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001561 readq_T *head = &channel->ch_part[part].ch_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001562
Bram Moolenaar77073442016-02-13 23:23:53 +01001563 if (head->rq_next == NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001564 return NULL;
Bram Moolenaar77073442016-02-13 23:23:53 +01001565 return head->rq_next->rq_buffer;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001566}
1567
1568/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001569 * Clear the read buffer on "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001570 */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001571 static void
1572channel_clear_one(channel_T *channel, int part)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001573{
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001574 jsonq_T *json_head = &channel->ch_part[part].ch_json_head;
1575 cbq_T *cb_head = &channel->ch_part[part].ch_cb_head;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001576
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001577 while (channel_peek(channel, part) != NULL)
1578 vim_free(channel_get(channel, part));
Bram Moolenaar77073442016-02-13 23:23:53 +01001579
1580 while (cb_head->cq_next != NULL)
Bram Moolenaard46ae142016-02-16 13:33:52 +01001581 {
1582 cbq_T *node = cb_head->cq_next;
1583
1584 remove_cb_node(cb_head, node);
1585 vim_free(node->cq_callback);
1586 vim_free(node);
1587 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001588
1589 while (json_head->jq_next != NULL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001590 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001591 free_tv(json_head->jq_next->jq_value);
1592 remove_json_node(json_head, json_head->jq_next);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001593 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001594
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001595 vim_free(channel->ch_part[part].ch_callback);
1596 channel->ch_part[part].ch_callback = NULL;
1597}
1598
1599/*
1600 * Clear all the read buffers on "channel".
1601 */
1602 void
1603channel_clear(channel_T *channel)
1604{
1605 channel_clear_one(channel, PART_SOCK);
1606#ifdef CHANNEL_PIPES
1607 channel_clear_one(channel, PART_OUT);
1608 channel_clear_one(channel, PART_ERR);
1609#endif
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01001610 vim_free(channel->ch_callback);
1611 channel->ch_callback = NULL;
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001612 vim_free(channel->ch_close_cb);
1613 channel->ch_close_cb = NULL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001614}
1615
Bram Moolenaar77073442016-02-13 23:23:53 +01001616#if defined(EXITFREE) || defined(PROTO)
1617 void
1618channel_free_all(void)
1619{
1620 channel_T *channel;
1621
1622 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1623 channel_clear(channel);
1624}
1625#endif
1626
1627
Bram Moolenaard04a0202016-01-26 23:30:18 +01001628/* Sent when the channel is found closed when reading. */
Bram Moolenaareed284a2016-02-22 23:13:33 +01001629#define DETACH_MSG_RAW "DETACH\n"
1630#define DETACH_MSG_JSON "\"DETACH\"\n"
Bram Moolenaard04a0202016-01-26 23:30:18 +01001631
1632/* Buffer size for reading incoming messages. */
1633#define MAXMSGSIZE 4096
1634
1635/*
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001636 * Check for reading from "fd" with "timeout" msec.
1637 * Return FAIL when there is nothing to read.
1638 */
1639 static int
Bram Moolenaard8070362016-02-15 21:56:54 +01001640channel_wait(channel_T *channel, sock_T fd, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001641{
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001642 if (timeout > 0)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001643 ch_logn(channel, "Waiting for up to %d msec", timeout);
Bram Moolenaard8070362016-02-15 21:56:54 +01001644
Bram Moolenaard8070362016-02-15 21:56:54 +01001645# ifdef WIN32
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001646 if (fd != channel->CH_SOCK_FD)
Bram Moolenaard8070362016-02-15 21:56:54 +01001647 {
1648 DWORD nread;
1649 int diff;
1650 DWORD deadline = GetTickCount() + timeout;
1651
1652 /* reading from a pipe, not a socket */
1653 while (TRUE)
1654 {
Bram Moolenaare74e8e72016-02-16 22:01:30 +01001655 if (PeekNamedPipe((HANDLE)fd, NULL, 0, NULL, &nread, NULL)
1656 && nread > 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001657 return OK;
1658 diff = deadline - GetTickCount();
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001659 if (diff <= 0)
Bram Moolenaard8070362016-02-15 21:56:54 +01001660 break;
1661 /* Wait for 5 msec.
1662 * TODO: increase the sleep time when looping more often */
1663 Sleep(5);
1664 }
Bram Moolenaard8070362016-02-15 21:56:54 +01001665 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001666 else
Bram Moolenaard8070362016-02-15 21:56:54 +01001667#endif
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001668 {
Bram Moolenaar9186a272016-02-23 19:34:01 +01001669#if defined(HAVE_SELECT)
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001670 struct timeval tval;
1671 fd_set rfds;
1672 int ret;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001673
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001674 FD_ZERO(&rfds);
1675 FD_SET((int)fd, &rfds);
1676 tval.tv_sec = timeout / 1000;
1677 tval.tv_usec = (timeout % 1000) * 1000;
1678 for (;;)
1679 {
1680 ret = select((int)fd + 1, &rfds, NULL, NULL, &tval);
Bram Moolenaar9186a272016-02-23 19:34:01 +01001681# ifdef EINTR
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001682 SOCK_ERRNO;
1683 if (ret == -1 && errno == EINTR)
1684 continue;
Bram Moolenaar9186a272016-02-23 19:34:01 +01001685# endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001686 if (ret > 0)
1687 return OK;
1688 break;
1689 }
Bram Moolenaar9186a272016-02-23 19:34:01 +01001690#else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001691 struct pollfd fds;
1692
1693 fds.fd = fd;
1694 fds.events = POLLIN;
1695 if (poll(&fds, 1, timeout) > 0)
1696 return OK;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001697#endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001698 }
1699 ch_log(channel, "Nothing to read");
1700 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001701}
1702
1703/*
1704 * Return a unique ID to be used in a message.
1705 */
1706 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001707channel_get_id(void)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001708{
1709 static int next_id = 1;
1710
1711 return next_id++;
1712}
1713
1714/*
Bram Moolenaar77073442016-02-13 23:23:53 +01001715 * Read from channel "channel" for as long as there is something to read.
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001716 * "part" is PART_SOCK, PART_OUT or PART_ERR.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001717 * The data is put in the read queue.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001718 */
1719 void
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001720channel_read(channel_T *channel, int part, char *func)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001721{
1722 static char_u *buf = NULL;
1723 int len = 0;
1724 int readlen = 0;
Bram Moolenaard8070362016-02-15 21:56:54 +01001725 sock_T fd;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001726 int use_socket = FALSE;
Bram Moolenaareed284a2016-02-22 23:13:33 +01001727 char *msg;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001728
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001729 fd = channel->ch_part[part].ch_fd;
1730 if (fd == INVALID_FD)
1731 {
1732 ch_error(channel, "channel_read() called while socket is closed");
Bram Moolenaard04a0202016-01-26 23:30:18 +01001733 return;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001734 }
1735 use_socket = fd == channel->CH_SOCK_FD;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001736
1737 /* Allocate a buffer to read into. */
1738 if (buf == NULL)
1739 {
1740 buf = alloc(MAXMSGSIZE);
1741 if (buf == NULL)
1742 return; /* out of memory! */
1743 }
1744
1745 /* Keep on reading for as long as there is something to read.
1746 * Use select() or poll() to avoid blocking on a message that is exactly
1747 * MAXMSGSIZE long. */
1748 for (;;)
1749 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001750 if (channel_wait(channel, fd, 0) == FAIL)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001751 break;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001752 if (use_socket)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001753 len = sock_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001754 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01001755 len = fd_read(fd, (char *)buf, MAXMSGSIZE);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001756 if (len <= 0)
1757 break; /* error or nothing more to read */
1758
1759 /* Store the read message in the queue. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001760 channel_save(channel, part, buf, len);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001761 readlen += len;
1762 if (len < MAXMSGSIZE)
1763 break; /* did read everything that's available */
1764 }
1765
Bram Moolenaar4cafa6d2016-02-26 11:52:39 +01001766 /* Reading a disconnection (readlen == 0), or an error. */
Bram Moolenaarbd73ae12016-02-22 22:19:22 +01001767 if (readlen <= 0)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001768 {
Bram Moolenaar4cafa6d2016-02-26 11:52:39 +01001769 /* Do not give an error message, most likely the other end just
1770 * exited. */
1771 ch_errors(channel, "%s(): Cannot read from channel", func);
1772
Bram Moolenaard04a0202016-01-26 23:30:18 +01001773 /* Queue a "DETACH" netbeans message in the command queue in order to
1774 * terminate the netbeans session later. Do not end the session here
1775 * directly as we may be running in the context of a call to
1776 * netbeans_parse_messages():
1777 * netbeans_parse_messages
1778 * -> autocmd triggered while processing the netbeans cmd
1779 * -> ui_breakcheck
1780 * -> gui event loop or select loop
1781 * -> channel_read()
1782 */
Bram Moolenaareed284a2016-02-22 23:13:33 +01001783 msg = channel->ch_part[part].ch_mode == MODE_RAW
1784 || channel->ch_part[part].ch_mode == MODE_NL
1785 ? DETACH_MSG_RAW : DETACH_MSG_JSON;
1786 channel_save(channel, part, (char_u *)msg, (int)STRLEN(msg));
Bram Moolenaard04a0202016-01-26 23:30:18 +01001787
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001788 /* TODO: When reading from stdout is not possible, should we try to
1789 * keep stdin and stderr open? Probably not, assume the other side
1790 * has died. */
Bram Moolenaar8b374212016-02-24 20:43:06 +01001791 channel_close(channel, TRUE);
Bram Moolenaar4e221c92016-02-23 13:20:22 +01001792 if (channel->ch_nb_close_cb != NULL)
1793 (*channel->ch_nb_close_cb)();
Bram Moolenaard04a0202016-01-26 23:30:18 +01001794 }
1795
1796#if defined(CH_HAS_GUI) && defined(FEAT_GUI_GTK)
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001797 /* signal the main loop that there is something to read */
Bram Moolenaard04a0202016-01-26 23:30:18 +01001798 if (CH_HAS_GUI && gtk_main_level() > 0)
1799 gtk_main_quit();
1800#endif
1801}
1802
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001803/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001804 * Read from RAW or NL "channel"/"part". Blocks until there is something to
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001805 * read or the timeout expires.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001806 * Returns what was read in allocated memory.
1807 * Returns NULL in case of error or timeout.
1808 */
1809 char_u *
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001810channel_read_block(channel_T *channel, int part, int timeout)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001811{
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001812 char_u *buf;
1813 char_u *msg;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001814 ch_mode_T mode = channel->ch_part[part].ch_mode;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001815 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001816 char_u *nl;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001817
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001818 ch_logsn(channel, "Blocking %s read, timeout: %d msec",
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001819 mode == MODE_RAW ? "RAW" : "NL", timeout);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001820
1821 while (TRUE)
1822 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001823 buf = channel_peek(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001824 if (buf != NULL && (mode == MODE_RAW
1825 || (mode == MODE_NL && vim_strchr(buf, NL) != NULL)))
1826 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001827 if (buf != NULL && channel_collapse(channel, part) == OK)
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001828 continue;
1829
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001830 /* Wait for up to the channel timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001831 if (fd == INVALID_FD
1832 || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001833 return NULL;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001834 channel_read(channel, part, "channel_read_block");
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001835 }
1836
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001837 if (mode == MODE_RAW)
1838 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001839 msg = channel_get_all(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001840 }
1841 else
1842 {
1843 nl = vim_strchr(buf, NL);
1844 if (nl[1] == NUL)
1845 {
1846 /* get the whole buffer */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001847 msg = channel_get(channel, part);
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001848 *nl = NUL;
1849 }
1850 else
1851 {
1852 /* Copy the message into allocated memory and remove it from the
1853 * buffer. */
1854 msg = vim_strnsave(buf, (int)(nl - buf));
1855 mch_memmove(buf, nl + 1, STRLEN(nl + 1) + 1);
1856 }
1857 }
1858 if (log_fd != NULL)
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001859 ch_logn(channel, "Returning %d bytes", (int)STRLEN(msg));
Bram Moolenaar9a6e33a2016-02-16 19:25:12 +01001860 return msg;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001861}
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001862
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001863/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001864 * Read one JSON message with ID "id" from "channel"/"part" and store the
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001865 * result in "rettv".
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001866 * When "id" is -1 accept any message;
Bram Moolenaar4d919d72016-02-05 22:36:41 +01001867 * Blocks until the message is received or the timeout is reached.
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001868 */
1869 int
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001870channel_read_json_block(
1871 channel_T *channel,
1872 int part,
1873 int timeout,
1874 int id,
1875 typval_T **rettv)
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001876{
Bram Moolenaare56bf152016-02-08 23:23:42 +01001877 int more;
Bram Moolenaard8070362016-02-15 21:56:54 +01001878 sock_T fd;
Bram Moolenaard7ece102016-02-02 23:23:02 +01001879
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001880 ch_log(channel, "Reading JSON");
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001881 if (id != -1)
1882 channel->ch_part[part].ch_block_id = id;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001883 for (;;)
1884 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001885 more = channel_parse_json(channel, part);
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001886
1887 /* search for messsage "id" */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001888 if (channel_get_json(channel, part, id, rettv) == OK)
Bram Moolenaare56bf152016-02-08 23:23:42 +01001889 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001890 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001891 return OK;
Bram Moolenaare56bf152016-02-08 23:23:42 +01001892 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001893
Bram Moolenaard7ece102016-02-02 23:23:02 +01001894 if (!more)
1895 {
1896 /* Handle any other messages in the queue. If done some more
1897 * messages may have arrived. */
1898 if (channel_parse_messages())
1899 continue;
1900
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001901 /* Wait for up to the timeout. */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001902 fd = channel->ch_part[part].ch_fd;
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01001903 if (fd == INVALID_FD || channel_wait(channel, fd, timeout) == FAIL)
Bram Moolenaard7ece102016-02-02 23:23:02 +01001904 break;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001905 channel_read(channel, part, "channel_read_json_block");
Bram Moolenaard7ece102016-02-02 23:23:02 +01001906 }
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001907 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001908 channel->ch_part[part].ch_block_id = 0;
Bram Moolenaar19d2f152016-02-01 21:38:19 +01001909 return FAIL;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001910}
1911
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001912# if defined(WIN32) || defined(FEAT_GUI_X11) || defined(FEAT_GUI_GTK) \
1913 || defined(PROTO)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001914/*
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001915 * Lookup the channel from the socket. Set "partp" to the fd index.
Bram Moolenaar77073442016-02-13 23:23:53 +01001916 * Returns NULL when the socket isn't found.
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001917 */
Bram Moolenaar77073442016-02-13 23:23:53 +01001918 channel_T *
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001919channel_fd2channel(sock_T fd, int *partp)
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001920{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001921 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001922 int part;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001923
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001924 if (fd != INVALID_FD)
Bram Moolenaar77073442016-02-13 23:23:53 +01001925 for (channel = first_channel; channel != NULL;
1926 channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001927 {
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001928# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001929 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001930# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001931 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001932# endif
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001933 if (channel->ch_part[part].ch_fd == fd)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001934 {
Bram Moolenaarfffd5562016-02-20 18:44:39 +01001935 *partp = part;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001936 return channel;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01001937 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01001938 }
Bram Moolenaar77073442016-02-13 23:23:53 +01001939 return NULL;
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001940}
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001941# endif
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001942
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001943# if defined(WIN32) || defined(PROTO)
1944/*
1945 * Check the channels for anything that is ready to be read.
1946 * The data is put in the read queue.
1947 */
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001948 void
1949channel_handle_events(void)
1950{
1951 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001952 int part;
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001953 sock_T fd;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001954
1955 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
1956 {
Bram Moolenaar9186a272016-02-23 19:34:01 +01001957# ifdef CHANNEL_PIPES
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001958 /* check the socket and pipes */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001959 for (part = PART_SOCK; part <= PART_ERR; ++part)
Bram Moolenaar9186a272016-02-23 19:34:01 +01001960# else
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001961 /* only check the socket */
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001962 part = PART_SOCK;
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001963# endif
Bram Moolenaarb7522a22016-02-21 17:20:55 +01001964 {
1965 fd = channel->ch_part[part].ch_fd;
1966 if (fd != INVALID_FD && channel_wait(channel, fd, 0) == OK)
1967 channel_read(channel, part, "channel_handle_events");
1968 }
Bram Moolenaared5a78e2016-02-19 21:05:03 +01001969 }
1970}
Bram Moolenaar85be35f2016-01-27 21:08:18 +01001971# endif
1972
Bram Moolenaard04a0202016-01-26 23:30:18 +01001973/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001974 * Write "buf" (NUL terminated string) to "channel"/"part".
Bram Moolenaard04a0202016-01-26 23:30:18 +01001975 * When "fun" is not NULL an error message might be given.
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001976 * Return FAIL or OK.
Bram Moolenaard04a0202016-01-26 23:30:18 +01001977 */
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001978 int
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001979channel_send(channel_T *channel, int part, char_u *buf, char *fun)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001980{
Bram Moolenaard04a0202016-01-26 23:30:18 +01001981 int len = (int)STRLEN(buf);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001982 int res;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001983 sock_T fd;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001984
Bram Moolenaar42d38a22016-02-20 18:18:59 +01001985 fd = channel->ch_part[part].ch_fd;
1986 if (fd == INVALID_FD)
Bram Moolenaard04a0202016-01-26 23:30:18 +01001987 {
1988 if (!channel->ch_error && fun != NULL)
1989 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01001990 ch_errors(channel, "%s(): write while not connected", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01001991 EMSG2("E630: %s(): write while not connected", fun);
1992 }
1993 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001994 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01001995 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01001996
Bram Moolenaar6463ca22016-02-13 17:04:46 +01001997 if (log_fd != NULL)
1998 {
Bram Moolenaar77073442016-02-13 23:23:53 +01001999 ch_log_lead("SEND ", channel);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002000 fprintf(log_fd, "'");
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002001 ignored = (int)fwrite(buf, len, 1, log_fd);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002002 fprintf(log_fd, "'\n");
2003 fflush(log_fd);
2004 }
2005
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002006 if (part == PART_SOCK)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002007 res = sock_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002008 else
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01002009 res = fd_write(fd, (char *)buf, len);
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002010 if (res != len)
Bram Moolenaard04a0202016-01-26 23:30:18 +01002011 {
2012 if (!channel->ch_error && fun != NULL)
2013 {
Bram Moolenaar81661fb2016-02-18 22:23:34 +01002014 ch_errors(channel, "%s(): write failed", fun);
Bram Moolenaard04a0202016-01-26 23:30:18 +01002015 EMSG2("E631: %s(): write failed", fun);
2016 }
2017 channel->ch_error = TRUE;
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01002018 return FAIL;
Bram Moolenaard04a0202016-01-26 23:30:18 +01002019 }
Bram Moolenaar3b5f9292016-01-28 22:37:01 +01002020
2021 channel->ch_error = FALSE;
2022 return OK;
Bram Moolenaard04a0202016-01-26 23:30:18 +01002023}
2024
2025# if (defined(UNIX) && !defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002026/*
2027 * Add open channels to the poll struct.
2028 * Return the adjusted struct index.
2029 * The type of "fds" is hidden to avoid problems with the function proto.
2030 */
2031 int
2032channel_poll_setup(int nfd_in, void *fds_in)
2033{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002034 int nfd = nfd_in;
2035 channel_T *channel;
2036 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002037 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002038
Bram Moolenaar77073442016-02-13 23:23:53 +01002039 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002040 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002041# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002042 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002043# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002044 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002045# endif
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002046 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002047 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002048 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002049 channel->ch_part[part].ch_poll_idx = nfd;
2050 fds[nfd].fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002051 fds[nfd].events = POLLIN;
2052 nfd++;
2053 }
2054 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002055 channel->ch_part[part].ch_poll_idx = -1;
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002056 }
2057 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002058
2059 return nfd;
2060}
2061
2062/*
2063 * The type of "fds" is hidden to avoid problems with the function proto.
2064 */
2065 int
2066channel_poll_check(int ret_in, void *fds_in)
2067{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002068 int ret = ret_in;
2069 channel_T *channel;
2070 struct pollfd *fds = fds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002071 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002072
Bram Moolenaar77073442016-02-13 23:23:53 +01002073 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002074 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002075# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002076 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002077# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002078 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002079# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002080 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002081 int idx = channel->ch_part[part].ch_poll_idx;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002082
2083 if (ret > 0 && idx != -1 && fds[idx].revents & POLLIN)
2084 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002085 channel_read(channel, part, "channel_poll_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002086 --ret;
2087 }
2088 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002089 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002090
2091 return ret;
2092}
Bram Moolenaard04a0202016-01-26 23:30:18 +01002093# endif /* UNIX && !HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002094
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002095# if (!defined(WIN32) && defined(HAVE_SELECT)) || defined(PROTO)
Bram Moolenaare0874f82016-01-24 20:36:41 +01002096/*
2097 * The type of "rfds" is hidden to avoid problems with the function proto.
2098 */
2099 int
2100channel_select_setup(int maxfd_in, void *rfds_in)
2101{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002102 int maxfd = maxfd_in;
2103 channel_T *channel;
2104 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002105 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002106
Bram Moolenaar77073442016-02-13 23:23:53 +01002107 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002108 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002109# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002110 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002111# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002112 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002113# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002114 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002115 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002116
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002117 if (fd != INVALID_FD)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002118 {
Bram Moolenaard8070362016-02-15 21:56:54 +01002119 FD_SET((int)fd, rfds);
2120 if (maxfd < (int)fd)
2121 maxfd = (int)fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002122 }
2123 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002124 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002125
2126 return maxfd;
2127}
2128
2129/*
2130 * The type of "rfds" is hidden to avoid problems with the function proto.
2131 */
2132 int
2133channel_select_check(int ret_in, void *rfds_in)
2134{
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002135 int ret = ret_in;
2136 channel_T *channel;
2137 fd_set *rfds = rfds_in;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002138 int part;
Bram Moolenaare0874f82016-01-24 20:36:41 +01002139
Bram Moolenaar77073442016-02-13 23:23:53 +01002140 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002141 {
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002142# ifdef CHANNEL_PIPES
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002143 for (part = PART_SOCK; part < PART_IN; ++part)
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002144# else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002145 part = PART_SOCK;
Bram Moolenaar6463ca22016-02-13 17:04:46 +01002146# endif
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002147 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002148 sock_T fd = channel->ch_part[part].ch_fd;
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002149
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002150 if (ret > 0 && fd != INVALID_FD && FD_ISSET(fd, rfds))
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002151 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002152 channel_read(channel, part, "channel_select_check");
Bram Moolenaar7b3ca762016-02-14 19:13:43 +01002153 --ret;
2154 }
2155 }
Bram Moolenaar16eb4f82016-02-14 23:02:34 +01002156 }
Bram Moolenaare0874f82016-01-24 20:36:41 +01002157
2158 return ret;
2159}
Bram Moolenaared5a78e2016-02-19 21:05:03 +01002160# endif /* !WIN32 && HAVE_SELECT */
Bram Moolenaare0874f82016-01-24 20:36:41 +01002161
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002162/*
Bram Moolenaard7ece102016-02-02 23:23:02 +01002163 * Execute queued up commands.
2164 * Invoked from the main loop when it's safe to execute received commands.
2165 * Return TRUE when something was done.
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002166 */
Bram Moolenaard7ece102016-02-02 23:23:02 +01002167 int
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002168channel_parse_messages(void)
2169{
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002170 channel_T *channel = first_channel;
2171 int ret = FALSE;
2172 int r;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002173 int part = PART_SOCK;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002174
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002175 while (channel != NULL)
2176 {
Bram Moolenaar46c85432016-02-26 11:17:46 +01002177 if (channel->ch_refcount == 0 && !channel_still_useful(channel))
Bram Moolenaarc8dcbb12016-02-25 23:10:17 +01002178 {
2179 /* channel is no longer useful, free it */
2180 channel_free(channel);
2181 channel = first_channel;
2182 part = PART_SOCK;
2183 continue;
2184 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002185 if (channel->ch_part[part].ch_fd != INVALID_FD)
Bram Moolenaard7ece102016-02-02 23:23:02 +01002186 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002187 /* Increase the refcount, in case the handler causes the channel
2188 * to be unreferenced or closed. */
2189 ++channel->ch_refcount;
2190 r = may_invoke_callback(channel, part);
2191 if (r == OK)
2192 ret = TRUE;
2193 if (channel_unref(channel) || r == OK)
2194 {
2195 /* channel was freed or something was done, start over */
2196 channel = first_channel;
2197 part = PART_SOCK;
2198 continue;
2199 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002200 }
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002201#ifdef CHANNEL_PIPES
2202 if (part < PART_ERR)
2203 ++part;
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002204 else
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002205#endif
2206 {
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002207 channel = channel->ch_next;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002208 part = PART_SOCK;
2209 }
Bram Moolenaar3bece9f2016-02-15 20:39:46 +01002210 }
Bram Moolenaard7ece102016-02-02 23:23:02 +01002211 return ret;
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01002212}
2213
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01002214/*
2215 * Mark references to lists used in channels.
2216 */
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002217 int
2218set_ref_in_channel(int copyID)
2219{
Bram Moolenaar77073442016-02-13 23:23:53 +01002220 int abort = FALSE;
2221 channel_T *channel;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002222 int part;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002223
Bram Moolenaar77073442016-02-13 23:23:53 +01002224 for (channel = first_channel; channel != NULL; channel = channel->ch_next)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002225 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002226#ifdef CHANNEL_PIPES
2227 for (part = PART_SOCK; part < PART_IN; ++part)
2228#else
2229 part = PART_SOCK;
2230#endif
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002231 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002232 jsonq_T *head = &channel->ch_part[part].ch_json_head;
2233 jsonq_T *item = head->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002234
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002235 while (item != NULL)
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002236 {
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002237 list_T *l = item->jq_value->vval.v_list;
2238
2239 if (l->lv_copyID != copyID)
2240 {
2241 l->lv_copyID = copyID;
2242 abort = abort || set_ref_in_list(l, copyID, NULL);
2243 }
2244 item = item->jq_next;
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002245 }
Bram Moolenaar4b6a6dc2016-02-04 22:49:49 +01002246 }
2247 }
2248 return abort;
2249}
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002250
2251/*
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002252 * Return the "part" to write to for "channel".
2253 */
2254 int
2255channel_part_send(channel_T *channel)
2256{
2257#ifdef CHANNEL_PIPES
2258 if (channel->CH_SOCK_FD == INVALID_FD)
2259 return PART_IN;
2260#endif
2261 return PART_SOCK;
2262}
2263
2264/*
2265 * Return the default "part" to read from for "channel".
2266 */
2267 int
2268channel_part_read(channel_T *channel)
2269{
2270#ifdef CHANNEL_PIPES
2271 if (channel->CH_SOCK_FD == INVALID_FD)
2272 return PART_OUT;
2273#endif
2274 return PART_SOCK;
2275}
2276
2277/*
2278 * Return the mode of "channel"/"part"
Bram Moolenaar77073442016-02-13 23:23:53 +01002279 * If "channel" is invalid returns MODE_JSON.
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002280 */
2281 ch_mode_T
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002282channel_get_mode(channel_T *channel, int part)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002283{
Bram Moolenaar77073442016-02-13 23:23:53 +01002284 if (channel == NULL)
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002285 return MODE_JSON;
Bram Moolenaar42d38a22016-02-20 18:18:59 +01002286 return channel->ch_part[part].ch_mode;
Bram Moolenaarae8eb3c2016-02-07 21:59:26 +01002287}
2288
Bram Moolenaar6f3a5442016-02-20 19:56:13 +01002289/*
2290 * Return the timeout of "channel"/"part"
2291 */
2292 int
2293channel_get_timeout(channel_T *channel, int part)
2294{
2295 return channel->ch_part[part].ch_timeout;
2296}
2297
Bram Moolenaare0874f82016-01-24 20:36:41 +01002298#endif /* FEAT_CHANNEL */