Added a preference to capture in Pcap-NG from GUI.
[obnox/wireshark/wip.git] / capture_sync.c
1 /* capture_sync.c
2  * Synchronisation between Wireshark capture parent and child instances
3  *
4  * $Id$
5  *
6  * Wireshark - Network traffic analyzer
7  * By Gerald Combs <gerald@wireshark.org>
8  * Copyright 1998 Gerald Combs
9  *
10  * This program is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU General Public License
12  * as published by the Free Software Foundation; either version 2
13  * of the License, or (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
23  */
24
25 #ifdef HAVE_CONFIG_H
26 # include "config.h"
27 #endif
28
29 #ifdef HAVE_LIBPCAP
30
31 #include <glib.h>
32 #include <stdio.h>
33 #include <ctype.h>
34 #include <string.h>
35
36 #ifdef HAVE_UNISTD_H
37 #include <unistd.h>
38 #endif
39
40 #ifdef HAVE_FCNTL_H
41 #include <fcntl.h>
42 #endif
43
44 #include <signal.h>
45
46 #ifdef _WIN32
47 #include <wsutil/unicode-utils.h>
48 #endif
49
50 #ifdef HAVE_SYS_WAIT_H
51 # include <sys/wait.h>
52 #endif
53
54 #include "capture-pcap-util.h"
55
56 #ifndef _WIN32
57 /*
58  * Define various POSIX macros (and, in the case of WCOREDUMP, non-POSIX
59  * macros) on UNIX systems that don't have them.
60  */
61 #ifndef WIFEXITED
62 # define WIFEXITED(status)      (((status) & 0177) == 0)
63 #endif
64 #ifndef WIFSTOPPED
65 # define WIFSTOPPED(status)     (((status) & 0177) == 0177)
66 #endif
67 #ifndef WIFSIGNALED
68 # define WIFSIGNALED(status)    (!WIFSTOPPED(status) && !WIFEXITED(status))
69 #endif
70 #ifndef WEXITSTATUS
71 # define WEXITSTATUS(status)    ((status) >> 8)
72 #endif
73 #ifndef WTERMSIG
74 # define WTERMSIG(status)       ((status) & 0177)
75 #endif
76 #ifndef WCOREDUMP
77 # define WCOREDUMP(status)      ((status) & 0200)
78 #endif
79 #ifndef WSTOPSIG
80 # define WSTOPSIG(status)       ((status) >> 8)
81 #endif
82 #endif /* _WIN32 */
83
84 #include <epan/packet.h>
85 #include <epan/prefs.h>
86
87 #include "globals.h"
88 #include "file.h"
89 #include <epan/filesystem.h>
90 #include <epan/report_err.h>
91
92 #include "capture.h"
93 #include "capture_sync.h"
94
95 #include "sync_pipe.h"
96
97 #ifdef _WIN32
98 #include "capture-wpcap.h"
99 #endif
100 #include "ui_util.h"
101 #include <wsutil/file_util.h>
102 #include "log.h"
103
104 #ifdef _WIN32
105 #include <process.h>    /* For spawning child process */
106 #endif
107
108
109
110 #ifndef _WIN32
111 static const char *sync_pipe_signame(int);
112 #endif
113
114
115 static gboolean sync_pipe_input_cb(gint source, gpointer user_data);
116 static void sync_pipe_wait_for_child(capture_options *capture_opts);
117
118
119
120 /* Append an arg (realloc) to an argc/argv array */
121 /* (add a string pointer to a NULL-terminated array of string pointers) */
122 static const char **
123 sync_pipe_add_arg(const char **args, int *argc, const char *arg)
124 {
125   /* Grow the array; "*argc" currently contains the number of string
126      pointers, *not* counting the NULL pointer at the end, so we have
127      to add 2 in order to get the new size of the array, including the
128      new pointer and the terminating NULL pointer. */
129   args = g_realloc( (gpointer) args, (*argc + 2) * sizeof (char *));
130
131   /* Stuff the pointer into the penultimate element of the array, which
132      is the one at the index specified by "*argc". */
133   args[*argc] = arg;
134
135   /* Now bump the count. */
136   (*argc)++;
137
138   /* We overwrite the NULL pointer; put it back right after the
139      element we added. */
140   args[*argc] = NULL;
141
142   return args;
143 }
144
145
146
147 #ifdef _WIN32
148 /* Quote the argument element if necessary, so that it will get
149  * reconstructed correctly in the C runtime startup code.  Note that
150  * the unquoting algorithm in the C runtime is really weird, and
151  * rather different than what Unix shells do. See stdargv.c in the C
152  * runtime sources (in the Platform SDK, in src/crt).
153  *
154  * Stolen from GLib's protect_argv(), an internal routine that quotes
155  * string in an argument list so that they arguments will be handled
156  * correctly in the command-line string passed to CreateProcess()
157  * if that string is constructed by gluing those strings together.
158  */
159 static gchar *
160 protect_arg (const gchar *argv)
161 {
162     gchar *new_arg;
163     const gchar *p = argv;
164     gchar *q;
165     gint len = 0;
166     gboolean need_dblquotes = FALSE;
167
168     while (*p) {
169         if (*p == ' ' || *p == '\t')
170             need_dblquotes = TRUE;
171         else if (*p == '"')
172             len++;
173         else if (*p == '\\') {
174             const gchar *pp = p;
175
176             while (*pp && *pp == '\\')
177                 pp++;
178             if (*pp == '"')
179                 len++;
180         }
181         len++;
182         p++;
183     }
184
185     q = new_arg = g_malloc (len + need_dblquotes*2 + 1);
186     p = argv;
187
188     if (need_dblquotes)
189         *q++ = '"';
190
191     while (*p) {
192         if (*p == '"')
193             *q++ = '\\';
194         else if (*p == '\\') {
195             const gchar *pp = p;
196
197             while (*pp && *pp == '\\')
198                 pp++;
199             if (*pp == '"')
200                 *q++ = '\\';
201         }
202         *q++ = *p;
203         p++;
204     }
205
206     if (need_dblquotes)
207         *q++ = '"';
208     *q++ = '\0';
209
210     return new_arg;
211 }
212 #endif
213
214 /* Initialize an argument list and add dumpcap to it. */
215 static const char **
216 init_pipe_args(int *argc) {
217     const char **argv;
218     const char *progfile_dir;
219     char *exename;
220
221     progfile_dir = get_progfile_dir();
222     if (progfile_dir == NULL) {
223       return NULL;
224     }
225
226     /* Allocate the string pointer array with enough space for the
227        terminating NULL pointer. */
228     *argc = 0;
229     argv = g_malloc(sizeof (char *));
230     *argv = NULL;
231
232     /* take Wireshark's absolute program path and replace "Wireshark" with "dumpcap" */
233     exename = g_strdup_printf("%s" G_DIR_SEPARATOR_S "dumpcap", progfile_dir);
234
235     /* Make that the first argument in the argument list (argv[0]). */
236     argv = sync_pipe_add_arg(argv, argc, exename);
237
238     return argv;
239 }
240
241 #define ARGV_NUMBER_LEN 24
242 /* a new capture run: start a new dumpcap task and hand over parameters through command line */
243 gboolean
244 sync_pipe_start(capture_options *capture_opts) {
245     char ssnap[ARGV_NUMBER_LEN];
246     char sdlt[ARGV_NUMBER_LEN];
247     char scount[ARGV_NUMBER_LEN];
248     char sfilesize[ARGV_NUMBER_LEN];
249     char sfile_duration[ARGV_NUMBER_LEN];
250     char sring_num_files[ARGV_NUMBER_LEN];
251     char sautostop_files[ARGV_NUMBER_LEN];
252     char sautostop_filesize[ARGV_NUMBER_LEN];
253     char sautostop_duration[ARGV_NUMBER_LEN];
254 #ifdef HAVE_PCAP_REMOTE
255     char sauth[256];
256 #endif
257 #ifdef HAVE_PCAP_SETSAMPLING
258     char ssampling[ARGV_NUMBER_LEN];
259 #endif
260 #ifdef _WIN32
261     char buffer_size[ARGV_NUMBER_LEN];
262     HANDLE sync_pipe_read;                  /* pipe used to send messages from child to parent */
263     HANDLE sync_pipe_write;                 /* pipe used to send messages from child to parent */
264     HANDLE signal_pipe;                     /* named pipe used to send messages from parent to child (currently only stop) */
265     GString *args = g_string_sized_new(200);
266     gchar *quoted_arg;
267     SECURITY_ATTRIBUTES sa;
268     STARTUPINFO si;
269     PROCESS_INFORMATION pi;
270     int i;
271     char control_id[ARGV_NUMBER_LEN];
272     gchar *signal_pipe_name;
273 #else
274     char errmsg[1024+1];
275     int sync_pipe[2];                       /* pipe used to send messages from child to parent */
276     enum PIPES { PIPE_READ, PIPE_WRITE };   /* Constants 0 and 1 for PIPE_READ and PIPE_WRITE */
277 #endif
278     int sync_pipe_read_fd;
279     int argc;
280     const char **argv;
281
282
283     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "sync_pipe_start");
284     capture_opts_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, capture_opts);
285
286     capture_opts->fork_child = -1;
287
288     argv = init_pipe_args(&argc);
289     if (!argv) {
290         /* We don't know where to find dumpcap. */
291         report_failure("We don't know where to find dumpcap.");
292         return FALSE;
293     }
294
295     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "argv[0]: %s", argv[0]);
296
297     argv = sync_pipe_add_arg(argv, &argc, "-i");
298     argv = sync_pipe_add_arg(argv, &argc, capture_opts->iface);
299
300     if (capture_opts->has_snaplen) {
301       argv = sync_pipe_add_arg(argv, &argc, "-s");
302       g_snprintf(ssnap, ARGV_NUMBER_LEN, "%d",capture_opts->snaplen);
303       argv = sync_pipe_add_arg(argv, &argc, ssnap);
304     }
305
306     if (capture_opts->linktype != -1) {
307       argv = sync_pipe_add_arg(argv, &argc, "-y");
308       g_snprintf(sdlt, ARGV_NUMBER_LEN, "%s",linktype_val_to_name(capture_opts->linktype));
309       argv = sync_pipe_add_arg(argv, &argc, sdlt);
310     }
311
312     if(capture_opts->multi_files_on) {
313       if (capture_opts->has_autostop_filesize) {
314         argv = sync_pipe_add_arg(argv, &argc, "-b");
315         g_snprintf(sfilesize, ARGV_NUMBER_LEN, "filesize:%d",capture_opts->autostop_filesize);
316         argv = sync_pipe_add_arg(argv, &argc, sfilesize);
317       }
318
319       if (capture_opts->has_file_duration) {
320         argv = sync_pipe_add_arg(argv, &argc, "-b");
321         g_snprintf(sfile_duration, ARGV_NUMBER_LEN, "duration:%d",capture_opts->file_duration);
322         argv = sync_pipe_add_arg(argv, &argc, sfile_duration);
323       }
324
325       if (capture_opts->has_ring_num_files) {
326         argv = sync_pipe_add_arg(argv, &argc, "-b");
327         g_snprintf(sring_num_files, ARGV_NUMBER_LEN, "files:%d",capture_opts->ring_num_files);
328         argv = sync_pipe_add_arg(argv, &argc, sring_num_files);
329       }
330
331       if (capture_opts->has_autostop_files) {
332         argv = sync_pipe_add_arg(argv, &argc, "-a");
333         g_snprintf(sautostop_files, ARGV_NUMBER_LEN, "files:%d",capture_opts->autostop_files);
334         argv = sync_pipe_add_arg(argv, &argc, sautostop_files);
335       }
336     } else {
337         if (capture_opts->has_autostop_filesize) {
338           argv = sync_pipe_add_arg(argv, &argc, "-a");
339           g_snprintf(sautostop_filesize, ARGV_NUMBER_LEN, "filesize:%d",capture_opts->autostop_filesize);
340           argv = sync_pipe_add_arg(argv, &argc, sautostop_filesize);
341         }
342     }
343
344     if (capture_opts->has_autostop_packets) {
345       argv = sync_pipe_add_arg(argv, &argc, "-c");
346       g_snprintf(scount, ARGV_NUMBER_LEN, "%d",capture_opts->autostop_packets);
347       argv = sync_pipe_add_arg(argv, &argc, scount);
348     }
349
350     if (capture_opts->has_autostop_duration) {
351       argv = sync_pipe_add_arg(argv, &argc, "-a");
352       g_snprintf(sautostop_duration, ARGV_NUMBER_LEN, "duration:%d",capture_opts->autostop_duration);
353       argv = sync_pipe_add_arg(argv, &argc, sautostop_duration);
354     }
355
356     if (!capture_opts->promisc_mode)
357       argv = sync_pipe_add_arg(argv, &argc, "-p");
358     if (capture_opts->use_pcapng)
359       argv = sync_pipe_add_arg(argv, &argc, "-n");
360 #ifdef HAVE_PCAP_REMOTE
361     if (capture_opts->datatx_udp)
362       argv = sync_pipe_add_arg(argv, &argc, "-u");
363
364     if (!capture_opts->nocap_rpcap)
365       argv = sync_pipe_add_arg(argv, &argc, "-r");
366
367     if (capture_opts->auth_type == CAPTURE_AUTH_PWD)
368     {
369         argv = sync_pipe_add_arg(argv, &argc, "-A");
370         g_snprintf(sauth, sizeof(sauth), "%s:%s", capture_opts->auth_username,
371                    capture_opts->auth_password);
372         argv = sync_pipe_add_arg(argv, &argc, sauth);
373     }
374 #endif
375 #ifdef HAVE_PCAP_SETSAMPLING
376     if (capture_opts->sampling_method != CAPTURE_SAMP_NONE)
377     {
378         argv = sync_pipe_add_arg(argv, &argc, "-m");
379         g_snprintf(ssampling, ARGV_NUMBER_LEN, "%s:%d",
380              capture_opts->sampling_method == CAPTURE_SAMP_BY_COUNT ? "count" :
381              capture_opts->sampling_method == CAPTURE_SAMP_BY_TIMER ? "timer" :
382              "undef",
383              capture_opts->sampling_param);
384         argv = sync_pipe_add_arg(argv, &argc, ssampling);
385     }
386 #endif
387
388     /* dumpcap should be running in capture child mode (hidden feature) */
389 #ifndef DEBUG_CHILD
390     argv = sync_pipe_add_arg(argv, &argc, "-Z");
391 #ifdef _WIN32
392     g_snprintf(control_id, ARGV_NUMBER_LEN, "%d", GetCurrentProcessId());
393     argv = sync_pipe_add_arg(argv, &argc, control_id);
394 #else
395     argv = sync_pipe_add_arg(argv, &argc, SIGNAL_PIPE_CTRL_ID_NONE);
396 #endif
397 #endif
398
399 #ifdef _WIN32
400     argv = sync_pipe_add_arg(argv, &argc, "-B");
401 #ifdef HAVE_PCAP_REMOTE
402     if (capture_opts->src_type == CAPTURE_IFREMOTE)
403       /* No buffer size when using remote interfaces */
404       g_snprintf(buffer_size, ARGV_NUMBER_LEN, "%d", 1);
405     else
406 #endif
407     g_snprintf(buffer_size, ARGV_NUMBER_LEN, "%d",capture_opts->buffer_size);
408     argv = sync_pipe_add_arg(argv, &argc, buffer_size);
409 #endif
410
411     if (capture_opts->cfilter != NULL && strlen(capture_opts->cfilter) != 0) {
412       argv = sync_pipe_add_arg(argv, &argc, "-f");
413       argv = sync_pipe_add_arg(argv, &argc, capture_opts->cfilter);
414     }
415
416     if(capture_opts->save_file) {
417       argv = sync_pipe_add_arg(argv, &argc, "-w");
418       argv = sync_pipe_add_arg(argv, &argc, capture_opts->save_file);
419     }
420
421 #ifdef _WIN32
422     /* init SECURITY_ATTRIBUTES */
423     sa.nLength = sizeof(SECURITY_ATTRIBUTES);
424     sa.bInheritHandle = TRUE;
425     sa.lpSecurityDescriptor = NULL;
426
427     /* Create a pipe for the child process */
428     /* (increase this value if you have trouble while fast capture file switches) */
429     if (! CreatePipe(&sync_pipe_read, &sync_pipe_write, &sa, 5120)) {
430       /* Couldn't create the pipe between parent and child. */
431       report_failure("Couldn't create sync pipe: %s", strerror(errno));
432       g_free( (gpointer) argv[0]);
433       g_free( (gpointer) argv);
434       return FALSE;
435     }
436
437     /* Create the signal pipe */
438     signal_pipe_name = g_strdup_printf(SIGNAL_PIPE_FORMAT, control_id);
439     signal_pipe = CreateNamedPipe(utf_8to16(signal_pipe_name),
440       PIPE_ACCESS_OUTBOUND, PIPE_TYPE_BYTE, 1, 65535, 65535, 0, NULL);
441     g_free(signal_pipe_name);
442
443     if (signal_pipe == INVALID_HANDLE_VALUE) {
444       /* Couldn't create the signal pipe between parent and child. */
445       report_failure("Couldn't create signal pipe: %s", strerror(errno));
446       g_free( (gpointer) argv[0]);
447       g_free( (gpointer) argv);
448       return FALSE;
449     }
450
451     /* init STARTUPINFO */
452     memset(&si, 0, sizeof(si));
453     si.cb           = sizeof(si);
454 #ifdef DEBUG_CHILD
455     si.dwFlags = STARTF_USESHOWWINDOW;
456     si.wShowWindow  = SW_SHOW;
457 #else
458     si.dwFlags = STARTF_USESTDHANDLES|STARTF_USESHOWWINDOW;
459     si.wShowWindow  = SW_HIDE;  /* this hides the console window */
460     si.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
461     si.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE);
462     si.hStdError = sync_pipe_write;
463     /*si.hStdError = (HANDLE) _get_osfhandle(2);*/
464 #endif
465
466     /* convert args array into a single string */
467     /* XXX - could change sync_pipe_add_arg() instead */
468     /* there is a drawback here: the length is internally limited to 1024 bytes */
469     for(i=0; argv[i] != 0; i++) {
470         if(i != 0) g_string_append_c(args, ' ');    /* don't prepend a space before the path!!! */
471         quoted_arg = protect_arg(argv[i]);
472         g_string_append(args, quoted_arg);
473         g_free(quoted_arg);
474     }
475
476     /* call dumpcap */
477     if(!CreateProcess(NULL, utf_8to16(args->str), NULL, NULL, TRUE,
478                       CREATE_NEW_CONSOLE, NULL, NULL, &si, &pi)) {
479       report_failure("Couldn't run %s in child process: error %u",
480                      args->str, GetLastError());
481       CloseHandle(sync_pipe_read);
482       CloseHandle(sync_pipe_write);
483       g_free( (gpointer) argv[0]);
484       g_free( (gpointer) argv);
485       return FALSE;
486     }
487     capture_opts->fork_child = (int) pi.hProcess;
488     g_string_free(args, TRUE);
489
490     /* associate the operating system filehandle to a C run-time file handle */
491     /* (good file handle infos at: http://www.flounder.com/handles.htm) */
492     sync_pipe_read_fd = _open_osfhandle( (long) sync_pipe_read, _O_BINARY);
493
494     /* associate the operating system filehandle to a C run-time file handle */
495     capture_opts->signal_pipe_write_fd = _open_osfhandle( (long) signal_pipe, _O_BINARY);
496
497 #else /* _WIN32 */
498     if (pipe(sync_pipe) < 0) {
499       /* Couldn't create the pipe between parent and child. */
500       report_failure("Couldn't create sync pipe: %s", strerror(errno));
501       g_free( (gpointer) argv[0]);
502       g_free(argv);
503       return FALSE;
504     }
505
506     if ((capture_opts->fork_child = fork()) == 0) {
507       /*
508        * Child process - run dumpcap with the right arguments to make
509        * it just capture with the specified capture parameters
510        */
511       dup2(sync_pipe[PIPE_WRITE], 2);
512       ws_close(sync_pipe[PIPE_READ]);
513       execv(argv[0], (gpointer)argv);
514       g_snprintf(errmsg, sizeof errmsg, "Couldn't run %s in child process: %s",
515                 argv[0], strerror(errno));
516       sync_pipe_errmsg_to_parent(2, errmsg, "");
517
518       /* Exit with "_exit()", so that we don't close the connection
519          to the X server (and cause stuff buffered up by our parent but
520          not yet sent to be sent, as that stuff should only be sent by
521          our parent).  We've sent an error message to the parent, so
522          we exit with an exit status of 1 (any exit status other than
523          0 or 1 will cause an additional message to report that exit
524          status, over and above the error message we sent to the parent). */
525       _exit(1);
526     }
527
528     sync_pipe_read_fd = sync_pipe[PIPE_READ];
529 #endif
530
531     g_free( (gpointer) argv[0]);  /* exename */
532
533     /* Parent process - read messages from the child process over the
534        sync pipe. */
535     g_free( (gpointer) argv);   /* free up arg array */
536
537     /* Close the write side of the pipe, so that only the child has it
538        open, and thus it completely closes, and thus returns to us
539        an EOF indication, if the child closes it (either deliberately
540        or by exiting abnormally). */
541 #ifdef _WIN32
542     CloseHandle(sync_pipe_write);
543 #else
544     ws_close(sync_pipe[PIPE_WRITE]);
545 #endif
546
547     if (capture_opts->fork_child == -1) {
548       /* We couldn't even create the child process. */
549       report_failure("Couldn't create child process: %s", strerror(errno));
550       ws_close(sync_pipe_read_fd);
551 #ifdef _WIN32
552       ws_close(capture_opts->signal_pipe_write_fd);
553 #endif
554       return FALSE;
555     }
556
557     /* we might wait for a moment till child is ready, so update screen now */
558     main_window_update();
559
560     /* We were able to set up to read the capture file;
561        arrange that our callback be called whenever it's possible
562        to read from the sync pipe, so that it's called when
563        the child process wants to tell us something. */
564
565     /* we have a running capture, now wait for the real capture filename */
566     pipe_input_set_handler(sync_pipe_read_fd, (gpointer) capture_opts,
567         &capture_opts->fork_child, sync_pipe_input_cb);
568
569     return TRUE;
570 }
571
572 /*
573  * Open dumpcap with the supplied arguments.  On success, msg points to
574  * a buffer containing the dumpcap output and returns 0.  read_fd and
575  * fork_child point to the pipe's file descriptor and child PID/handle,
576  * respectively.  On failure, msg points to the error message returned by
577  * dumpcap, and returns dumpcap's exit value.  In either case, msg must be
578  * freed with g_free().
579  */
580 /* XXX - This duplicates a lot of code in sync_pipe_start() */
581 #define PIPE_BUF_SIZE 5120
582 static int
583 sync_pipe_open_command(const char** argv, int *read_fd, int *fork_child, gchar **msg) {
584 #ifdef _WIN32
585     HANDLE sync_pipe_read;                  /* pipe used to send messages from child to parent */
586     HANDLE sync_pipe_write;                 /* pipe used to send messages from parent to child */
587     GString *args = g_string_sized_new(200);
588     gchar *quoted_arg;
589     SECURITY_ATTRIBUTES sa;
590     STARTUPINFO si;
591     PROCESS_INFORMATION pi;
592     int i;
593 #else
594     char errmsg[1024+1];
595     int sync_pipe[2];                       /* pipe used to send messages from child to parent */
596     enum PIPES { PIPE_READ, PIPE_WRITE };   /* Constants 0 and 1 for PIPE_READ and PIPE_WRITE */
597 #endif
598
599     *fork_child = -1;
600     *read_fd = -1;
601     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "sync_pipe_run_command");
602
603     if (!msg) {
604         /* We can't return anything */
605 #ifdef _WIN32
606         g_string_free(args, TRUE);
607 #endif
608         return -1;
609     }
610
611 #ifdef _WIN32
612     /* init SECURITY_ATTRIBUTES */
613     sa.nLength = sizeof(SECURITY_ATTRIBUTES);
614     sa.bInheritHandle = TRUE;
615     sa.lpSecurityDescriptor = NULL;
616
617     /* Create a pipe for the child process */
618     /* (inrease this value if you have trouble while fast capture file switches) */
619     if (! CreatePipe(&sync_pipe_read, &sync_pipe_write, &sa, 5120)) {
620         /* Couldn't create the pipe between parent and child. */
621         *msg = g_strdup_printf("Couldn't create sync pipe: %s", strerror(errno));
622         g_free( (gpointer) argv[0]);
623         g_free( (gpointer) argv);
624         return CANT_RUN_DUMPCAP;
625     }
626
627     /* init STARTUPINFO */
628     memset(&si, 0, sizeof(si));
629     si.cb           = sizeof(si);
630 #ifdef DEBUG_CHILD
631     si.dwFlags = STARTF_USESHOWWINDOW;
632     si.wShowWindow  = SW_SHOW;
633 #else
634     si.dwFlags = STARTF_USESTDHANDLES|STARTF_USESHOWWINDOW;
635     si.wShowWindow  = SW_HIDE;  /* this hides the console window */
636     si.hStdInput = NULL;
637     si.hStdOutput = sync_pipe_write;
638     si.hStdError = sync_pipe_write;
639     /*si.hStdError = (HANDLE) _get_osfhandle(2);*/
640 #endif
641
642     /* convert args array into a single string */
643     /* XXX - could change sync_pipe_add_arg() instead */
644     /* there is a drawback here: the length is internally limited to 1024 bytes */
645     for(i=0; argv[i] != 0; i++) {
646         if(i != 0) g_string_append_c(args, ' ');    /* don't prepend a space before the path!!! */
647         quoted_arg = protect_arg(argv[i]);
648         g_string_append(args, quoted_arg);
649         g_free(quoted_arg);
650     }
651
652     /* call dumpcap */
653     if(!CreateProcess(NULL, utf_8to16(args->str), NULL, NULL, TRUE,
654                       CREATE_NEW_CONSOLE, NULL, NULL, &si, &pi)) {
655         *msg = g_strdup_printf("Couldn't run %s in child process: error %u",
656                         args->str, GetLastError());
657         CloseHandle(sync_pipe_read);
658         CloseHandle(sync_pipe_write);
659         g_free( (gpointer) argv[0]);
660         g_free( (gpointer) argv);
661         return CANT_RUN_DUMPCAP;
662     }
663     *fork_child = (int) pi.hProcess;
664     g_string_free(args, TRUE);
665
666     /* associate the operating system filehandle to a C run-time file handle */
667     /* (good file handle infos at: http://www.flounder.com/handles.htm) */
668     *read_fd = _open_osfhandle( (long) sync_pipe_read, _O_BINARY);
669
670 #else /* _WIN32 */
671     if (pipe(sync_pipe) < 0) {
672         /* Couldn't create the pipe between parent and child. */
673         *msg = g_strdup_printf("Couldn't create sync pipe: %s", strerror(errno));
674         g_free( (gpointer) argv[0]);
675         g_free(argv);
676         return CANT_RUN_DUMPCAP;
677     }
678
679     if ((*fork_child = fork()) == 0) {
680         /*
681          * Child process - run dumpcap with the right arguments to make
682          * it just capture with the specified capture parameters
683          */
684         dup2(sync_pipe[PIPE_WRITE], 1);
685         ws_close(sync_pipe[PIPE_READ]);
686         execv(argv[0], (gpointer)argv);
687         g_snprintf(errmsg, sizeof errmsg, "Couldn't run %s in child process: %s",
688                    argv[0], strerror(errno));
689         sync_pipe_errmsg_to_parent(1, errmsg, "");
690
691         /* Exit with "_exit()", so that we don't close the connection
692            to the X server (and cause stuff buffered up by our parent but
693            not yet sent to be sent, as that stuff should only be sent by
694            our parent).  We've sent an error message to the parent, so
695            we exit with an exit status of 1 (any exit status other than
696            0 or 1 will cause an additional message to report that exit
697            status, over and above the error message we sent to the parent). */
698         _exit(1);
699     }
700
701     *read_fd = sync_pipe[PIPE_READ];
702 #endif
703
704     g_free( (gpointer) argv[0]);  /* exename */
705
706     /* Parent process - read messages from the child process over the
707        sync pipe. */
708     g_free( (gpointer) argv);   /* free up arg array */
709
710     /* Close the write side of the pipe, so that only the child has it
711        open, and thus it completely closes, and thus returns to us
712        an EOF indication, if the child closes it (either deliberately
713        or by exiting abnormally). */
714 #ifdef _WIN32
715     CloseHandle(sync_pipe_write);
716 #else
717     ws_close(sync_pipe[PIPE_WRITE]);
718 #endif
719
720     if (*fork_child == -1) {
721         /* We couldn't even create the child process. */
722         *msg = g_strdup_printf("Couldn't create child process: %s", strerror(errno));
723         ws_close(*read_fd);
724         return CANT_RUN_DUMPCAP;
725     }
726
727     /* we might wait for a moment till child is ready, so update screen now */
728     main_window_update();
729     return 0;
730 }
731
732 static int
733 #ifdef _WIN32
734 sync_pipe_close_command(int *read_fd, int *fork_child, gchar **msg) {
735 #else
736 sync_pipe_close_command(int *read_fd, gchar **msg) {
737 #endif
738     int fork_child_status;
739
740     ws_close(*read_fd);
741
742     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "sync_pipe_close_command: wait till child closed");
743
744 #ifdef _WIN32
745     /* XXX - Should we signal the child somehow? */
746     sync_pipe_kill(*fork_child);
747     if (_cwait(&fork_child_status, *fork_child, _WAIT_CHILD) == -1) {
748         *msg = g_strdup_printf("Child capture process stopped unexpectedly "
749             "(errno:%u)", errno);
750         return CANT_RUN_DUMPCAP;
751     }
752 #else
753     if (wait(&fork_child_status) != -1) {
754         if (WIFEXITED(fork_child_status)) {
755             /* The child exited. */
756             fork_child_status = WEXITSTATUS(fork_child_status);
757         } else {
758             if (WIFSTOPPED(fork_child_status)) {
759                 /* It stopped, rather than exiting.  "Should not happen." */
760                 *msg = g_strdup_printf("Child capture process stopped: %s",
761                     sync_pipe_signame(WSTOPSIG(fork_child_status)));
762             } else if (WIFSIGNALED(fork_child_status)) {
763                 /* It died with a signal. */
764                 *msg = g_strdup_printf("Child capture process died: %s%s",
765                     sync_pipe_signame(WTERMSIG(fork_child_status)),
766                     WCOREDUMP(fork_child_status) ? " - core dumped" : "");
767             } else {
768                 /* What?  It had to either have exited, or stopped, or died with
769                    a signal; what happened here? */
770                 *msg = g_strdup_printf("Child capture process died: wait status %#o",
771                     fork_child_status);
772             }
773             return CANT_RUN_DUMPCAP;
774         }
775     } else {
776       *msg = g_strdup_printf("Child capture process stopped unexpectedly "
777         "(errno:%u)", errno);
778       return CANT_RUN_DUMPCAP;
779     }
780 #endif
781     return 0;
782 }
783
784 /*
785  * Run dumpcap with the supplied arguments.  On success, msg points to
786  * a buffer containing the dumpcap output and returns 0.  On failure, msg
787  * points to the error message returned by dumpcap, and returns dumpcap's
788  * exit value.  In either case, msg must be freed with g_free().
789  */
790 /* XXX - This duplicates a lot of code in sync_pipe_start() */
791 #define PIPE_BUF_SIZE 5120
792 static int
793 sync_pipe_run_command(const char** argv, gchar **msg) {
794     int sync_pipe_read_fd, fork_child, ret;
795     gchar buf[PIPE_BUF_SIZE+1];
796     GString *msg_buf = NULL;
797     int count;
798
799     ret = sync_pipe_open_command(argv, &sync_pipe_read_fd, &fork_child, msg);
800
801     if (ret)
802         return ret;
803
804     /* We were able to set up to read dumpcap's output.  Do so and
805        return its exit value. */
806     msg_buf = g_string_new("");
807     while ((count = ws_read(sync_pipe_read_fd, buf, PIPE_BUF_SIZE)) > 0) {
808         buf[count] = '\0';
809         g_string_append(msg_buf, buf);
810     }
811
812 #ifdef _WIN32
813     ret = sync_pipe_close_command(&sync_pipe_read_fd, &fork_child, msg);
814 #else
815     ret = sync_pipe_close_command(&sync_pipe_read_fd, msg);
816 #endif
817
818     if (ret) {
819         g_string_free(msg_buf, TRUE);
820         return ret;
821     }
822
823     *msg = msg_buf->str;
824     g_string_free(msg_buf, FALSE);
825     return 0;
826 }
827
828 /*
829  * Get an interface list using dumpcap.  On success, msg points to
830  * a buffer containing the dumpcap output and returns 0.  On failure, msg
831  * points to the error message returned by dumpcap, and returns dumpcap's
832  * exit value.  In either case, msg must be freed with g_free().
833  */
834 int
835 sync_interface_list_open(gchar **msg) {
836     int argc;
837     const char **argv;
838
839     if (!msg) {
840         /* We can't return anything */
841         return -1;
842     }
843
844     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "sync_interface_list_open");
845
846     argv = init_pipe_args(&argc);
847
848     if (!argv) {
849         *msg = g_strdup_printf("We don't know where to find dumpcap.");
850         return CANT_RUN_DUMPCAP;
851     }
852
853     /* Ask for the interface list */
854     argv = sync_pipe_add_arg(argv, &argc, "-D");
855     argv = sync_pipe_add_arg(argv, &argc, "-M");
856
857 #if 0
858     /* dumpcap should be running in capture child mode (hidden feature)                   */
859     /* XXX: Actually: don't run dumpcap in capture_child_mode.                            */
860     /*     Instead run dumpcap in 'normal' mode so that dumpcap err msgs are sent to      */
861     /*     stderr in normal format and are then sent to whereever our stderr goes.        */
862     /*     Note: Using 'dumpcap -D -M -Z' (capture_child mode) changes only the format of */
863     /*           dumpcap err msgs. That is: dumpcap in capture_child mode outputs err     */
864     /*           msgs to stderr in a special type/len/string format which would then      */
865     /*           currently be sent as is to stderr resulting in garbled output.           */
866     /*     ToDo: Revise this code to be similar to sync_pipe_start so that 'dumpcap -Z'   */
867     /*     special format error messages to stderr are captured and returned to caller    */
868     /*     (eg: so can be processed and displayed in a pop-up box).                       */
869 #ifndef DEBUG_CHILD
870     argv = sync_pipe_add_arg(argv, &argc, "-Z");
871     argv = sync_pipe_add_arg(argv, &argc, SIGNAL_PIPE_CTRL_ID_NONE);
872 #endif
873 #endif
874
875     return sync_pipe_run_command(argv, msg);
876 }
877
878 /*
879  * Get an linktype list using dumpcap.  On success, msg points to
880  * a buffer containing the dumpcap output and returns 0.  On failure, msg
881  * points to the error message returned by dumpcap, and returns dumpcap's
882  * exit value.  In either case, msg must be freed with g_free().
883  */
884 int
885 sync_linktype_list_open(const gchar *ifname, gchar **msg) {
886     int argc;
887     const char **argv;
888
889     if (!msg) {
890         /* We can't return anything */
891         return -1;
892     }
893
894     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "sync_linktype_list_open");
895
896     argv = init_pipe_args(&argc);
897
898     if (!argv) {
899         *msg = g_strdup_printf("We don't know where to find dumpcap.");
900         return CANT_RUN_DUMPCAP;
901     }
902
903     /* Ask for the linktype list */
904     argv = sync_pipe_add_arg(argv, &argc, "-i");
905     argv = sync_pipe_add_arg(argv, &argc, ifname);
906     argv = sync_pipe_add_arg(argv, &argc, "-L");
907     argv = sync_pipe_add_arg(argv, &argc, "-M");
908
909 #if 0
910     /* dumpcap should be running in capture child mode (hidden feature)                   */
911     /* XXX: Actually: don't run dumpcap in capture_child_mode.                            */
912     /*     Instead run dumpcap in 'normal' mode so that dumpcap err msgs are sent to      */
913     /*     stderr in normal format and are then sent to whereever our stderr goes.        */
914     /*     Note: Using 'dumpcap -L -M -Z' (capture_child mode) changes only the format of */
915     /*           dumpcap err msgs. That is: dumpcap in capture_child mode outputs err     */
916     /*           msgs to stderr in a special type/len/string format which would then      */
917     /*           currently be sent as is to stderr resulting in garbled output.           */
918     /*     ToDo: Revise this code to be similar to sync_pipe_start so that 'dumpcap -Z'   */
919     /*     special format error messages to stderr are captured and returned to caller    */
920     /*     (eg: so can be processed and displayed in a pop-up box).                       */
921 #ifndef DEBUG_CHILD
922     argv = sync_pipe_add_arg(argv, &argc, "-Z");
923     argv = sync_pipe_add_arg(argv, &argc, SIGNAL_PIPE_CTRL_ID_NONE);
924 #endif
925 #endif
926     return sync_pipe_run_command(argv, msg);
927 }
928
929 /*
930  * Start getting interface statistics using dumpcap.  On success, read_fd
931  * contains the file descriptor for the pipe's stdout, msg is unchanged,
932  * and zero is returned.  On failure, msg will point to an error message
933  * that must be g_free()d and a nonzero error value will be returned.
934  */
935 int
936 sync_interface_stats_open(int *read_fd, int *fork_child, gchar **msg) {
937     int argc;
938     const char **argv;
939
940     if (!msg) {
941         /* We can't return anything */
942         return -1;
943     }
944
945     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "sync_interface_stats_open");
946
947     argv = init_pipe_args(&argc);
948
949     if (!argv) {
950         *msg = g_strdup_printf("We don't know where to find dumpcap.");
951         return CANT_RUN_DUMPCAP;
952     }
953
954     /* Ask for the interface statistics */
955     argv = sync_pipe_add_arg(argv, &argc, "-S");
956     argv = sync_pipe_add_arg(argv, &argc, "-M");
957
958 #if 0
959     /* dumpcap should be running in capture child mode (hidden feature)                   */
960     /* XXX: Actually: don't run dumpcap in capture_child_mode.                            */
961     /*     Instead run dumpcap in 'normal' mode so that dumpcap err msgs are sent to      */
962     /*     stderr in normal format and are then sent to whereever our stderr goes.        */
963     /*     Note: Using 'dumpcap -S -M -Z' (capture_child mode) changes only the format of */
964     /*           dumpcap err msgs. That is: dumpcap in capture_child mode outputs err     */
965     /*           msgs to stderr in a special type/len/string format which would then      */
966     /*           currently be sent as is to stderr resulting in garbled output.           */
967     /*     ToDo: Revise this code to be similar to sync_pipe_start so that 'dumpcap -Z'   */
968     /*     special format error messages to stderr are captured and returned to caller    */
969     /*     (eg: so can be processed and displayed in a pop-up box).                       */
970 #ifndef DEBUG_CHILD
971     argv = sync_pipe_add_arg(argv, &argc, "-Z");
972     argv = sync_pipe_add_arg(argv, &argc, SIGNAL_PIPE_CTRL_ID_NONE);
973 #endif
974 #endif
975     return sync_pipe_open_command(argv, read_fd, fork_child, msg);
976 }
977
978 /* Close down the stats process */
979 int
980 sync_interface_stats_close(int *read_fd, int *fork_child
981 #ifndef _WIN32
982 _U_
983 #endif
984 , gchar **msg) {
985 #ifdef _WIN32
986     return sync_pipe_close_command(read_fd, fork_child, msg);
987 #else
988     return sync_pipe_close_command(read_fd, msg);
989 #endif
990 }
991
992 /* read a number of bytes from a pipe */
993 /* (blocks until enough bytes read or an error occurs) */
994 static int
995 pipe_read_bytes(int pipe, char *bytes, int required) {
996     int newly;
997     int offset = 0;
998
999     while(required) {
1000         newly = read(pipe, &bytes[offset], required);
1001         if (newly == 0) {
1002             /* EOF */
1003             g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG,
1004                   "read from pipe %d: EOF (capture closed?)", pipe);
1005             return offset;
1006         }
1007         if (newly < 0) {
1008             /* error */
1009             g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG,
1010                   "read from pipe %d: error(%u): %s", pipe, errno, strerror(errno));
1011             return newly;
1012         }
1013
1014         required -= newly;
1015         offset += newly;
1016     }
1017
1018     return offset;
1019 }
1020
1021 static gboolean pipe_data_available(int pipe) {
1022 #ifdef _WIN32 /* PeekNamedPipe */
1023     HANDLE hPipe = (HANDLE) _get_osfhandle(pipe);
1024     DWORD bytes_avail;
1025
1026     if (hPipe == INVALID_HANDLE_VALUE)
1027         return FALSE;
1028
1029     if (! PeekNamedPipe(hPipe, NULL, 0, NULL, &bytes_avail, NULL))
1030         return FALSE;
1031
1032     if (bytes_avail > 0)
1033         return TRUE;
1034     return FALSE;
1035 #else /* select */
1036     fd_set rfds;
1037     struct timeval timeout;
1038
1039     FD_ZERO(&rfds);
1040     FD_SET(pipe, &rfds);
1041     timeout.tv_sec = 0;
1042     timeout.tv_usec = 0;
1043
1044     if (select(pipe+1, &rfds, NULL, NULL, &timeout) > 0)
1045         return TRUE;
1046
1047     return FALSE;
1048 #endif
1049 }
1050
1051 /* Read a line from a pipe, similar to fgets */
1052 int
1053 sync_pipe_gets_nonblock(int pipe_fd, char *bytes, int max) {
1054     int newly;
1055     int offset = -1;
1056
1057     while(offset < max - 1) {
1058         offset++;
1059         if (! pipe_data_available(pipe_fd))
1060             break;
1061         newly = read(pipe_fd, &bytes[offset], 1);
1062         if (newly == 0) {
1063             /* EOF - not necessarily an error */
1064             break;
1065         } else if (newly < 0) {
1066             /* error */
1067             g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG,
1068                   "read from pipe %d: error(%u): %s", pipe_fd, errno, strerror(errno));
1069             return newly;
1070         } else if (bytes[offset] == '\n') {
1071             break;
1072         }
1073     }
1074
1075     if (offset >= 0)
1076         bytes[offset] = '\0';
1077
1078     return offset;
1079 }
1080
1081
1082 /* convert header values (indicator and 4-byte length) */
1083 static void
1084 pipe_convert_header(const guchar *header, int header_len, char *indicator, int *block_len) {
1085
1086     g_assert(header_len == 4);
1087
1088     /* convert header values */
1089     *indicator = header[0];
1090     *block_len = header[1]<<16 | header[2]<<8 | header[3];
1091 }
1092
1093 /* read a message from the sending pipe in the standard format
1094    (1-byte message indicator, 3-byte message length (excluding length
1095    and indicator field), and the rest is the message) */
1096 static int
1097 pipe_read_block(int pipe, char *indicator, int len, char *msg) {
1098     int required;
1099     int newly;
1100     guchar header[4];
1101
1102
1103     /* read header (indicator and 3-byte length) */
1104     newly = pipe_read_bytes(pipe, header, 4);
1105     if(newly != 4) {
1106         g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG,
1107               "read %d failed to read header: %u", pipe, newly);
1108         return -1;
1109     }
1110
1111     /* convert header values */
1112     pipe_convert_header(header, 4, indicator, &required);
1113
1114     /* only indicator with no value? */
1115     if(required == 0) {
1116         g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG,
1117               "read %d indicator: %c empty value", pipe, *indicator);
1118         return 4;
1119     }
1120
1121     /* does the data fit into the given buffer? */
1122     if(required > len) {
1123         g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG,
1124               "read %d length error, required %d > len %d, indicator: %u",
1125               pipe, required, len, *indicator);
1126
1127         /* we have a problem here, try to read some more bytes from the pipe to debug where the problem really is */
1128         memcpy(msg, header, sizeof(header));
1129         newly = read(pipe, &msg[sizeof(header)], len-sizeof(header));
1130         g_warning("Unknown message from dumpcap, try to show it as a string: %s", msg);
1131         return -1;
1132     }
1133     len = required;
1134
1135     /* read the actual block data */
1136     newly = pipe_read_bytes(pipe, msg, required);
1137     if(newly != required) {
1138         g_warning("Unknown message from dumpcap, try to show it as a string: %s", msg);
1139         return -1;
1140     }
1141
1142     /* XXX If message is "2part", the msg probably won't be sent to debug log correctly */
1143     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG,
1144           "read %d ok indicator: %c len: %u msg: %s", pipe, *indicator,
1145           len, msg);
1146     return newly + 4;
1147 }
1148
1149
1150 /* There's stuff to read from the sync pipe, meaning the child has sent
1151    us a message, or the sync pipe has closed, meaning the child has
1152    closed it (perhaps because it exited). */
1153 static gboolean
1154 sync_pipe_input_cb(gint source, gpointer user_data)
1155 {
1156   capture_options *capture_opts = (capture_options *)user_data;
1157   char buffer[SP_MAX_MSG_LEN+1];
1158   int  nread;
1159   char indicator;
1160   int  primary_len;
1161   char * primary_msg;
1162   int  secondary_len;
1163   char * secondary_msg;
1164
1165
1166   nread = pipe_read_block(source, &indicator, SP_MAX_MSG_LEN, buffer);
1167   if(nread <= 0) {
1168     if (nread == 0)
1169       g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG,
1170             "sync_pipe_input_cb: child has closed sync_pipe");
1171     else
1172       g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG,
1173             "sync_pipe_input_cb: error reading from sync pipe");
1174
1175     /* The child has closed the sync pipe, meaning it's not going to be
1176        capturing any more packets.  Pick up its exit status, and
1177        complain if it did anything other than exit with status 0.
1178
1179        We don't have to worry about killing the child, if the sync pipe
1180        returned an error. Usually this error is caused as the child killed itself
1181        while going down. Even in the rare cases that this isn't the case,
1182        the child will get an error when writing to the broken pipe the next time,
1183        cleaning itself up then. */
1184     sync_pipe_wait_for_child(capture_opts);
1185
1186 #ifdef _WIN32
1187     ws_close(capture_opts->signal_pipe_write_fd);
1188 #endif
1189     capture_input_closed(capture_opts);
1190     return FALSE;
1191   }
1192
1193   /* we got a valid message block from the child, process it */
1194   switch(indicator) {
1195   case SP_FILE:
1196     if(!capture_input_new_file(capture_opts, buffer)) {
1197       g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "sync_pipe_input_cb: file failed, closing capture");
1198
1199       /* We weren't able to open the new capture file; user has been
1200          alerted. Close the sync pipe. */
1201       ws_close(source);
1202
1203       /* the child has send us a filename which we couldn't open.
1204          this probably means, the child is creating files faster than we can handle it.
1205          this should only be the case for very fast file switches
1206          we can't do much more than telling the child to stop
1207          (this is the "emergency brake" if user e.g. wants to switch files every second) */
1208       sync_pipe_stop(capture_opts);
1209     }
1210     break;
1211   case SP_PACKET_COUNT:
1212     nread = atoi(buffer);
1213     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "sync_pipe_input_cb: new packets %u", nread);
1214     capture_input_new_packets(capture_opts, nread);
1215     break;
1216   case SP_ERROR_MSG:
1217     /* convert primary message */
1218     pipe_convert_header(buffer, 4, &indicator, &primary_len);
1219     primary_msg = buffer+4;
1220     /* convert secondary message */
1221     pipe_convert_header(primary_msg + primary_len, 4, &indicator, &secondary_len);
1222     secondary_msg = primary_msg + primary_len + 4;
1223     /* message output */
1224     capture_input_error_message(capture_opts, primary_msg, secondary_msg);
1225     /* the capture child will close the sync_pipe, nothing to do for now */
1226     /* (an error message doesn't mean we have to stop capturing) */
1227     break;
1228   case SP_BAD_FILTER:
1229     capture_input_cfilter_error_message(capture_opts, buffer);
1230     /* the capture child will close the sync_pipe, nothing to do for now */
1231     break;
1232   case SP_DROPS:
1233     capture_input_drops(capture_opts, (guint32)strtoul(buffer, NULL, 10));
1234     break;
1235   default:
1236     g_assert_not_reached();
1237   }
1238
1239   return TRUE;
1240 }
1241
1242
1243
1244 /* the child process is going down, wait until it's completely terminated */
1245 static void
1246 sync_pipe_wait_for_child(capture_options *capture_opts)
1247 {
1248   int  wstatus;
1249
1250
1251   g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "sync_pipe_wait_for_child: wait till child closed");
1252   g_assert(capture_opts->fork_child != -1);
1253
1254 #ifdef _WIN32
1255   if (_cwait(&wstatus, capture_opts->fork_child, _WAIT_CHILD) == -1) {
1256     report_failure("Child capture process stopped unexpectedly (errno:%u)",
1257                    errno);
1258   }
1259 #else
1260   if (wait(&wstatus) != -1) {
1261     if (WIFEXITED(wstatus)) {
1262       /* The child exited; display its exit status, if it seems uncommon (0=ok, 1=error) */
1263       /* the child will inform us about errors through the sync_pipe, which will popup */
1264       /* an error message, so don't popup another one */
1265
1266       /* If there are situations where the child won't send us such an error message, */
1267       /* this should be fixed in the child and not here! */
1268       if (WEXITSTATUS(wstatus) != 0 && WEXITSTATUS(wstatus) != 1) {
1269         report_failure("Child capture process exited: exit status %d",
1270                        WEXITSTATUS(wstatus));
1271       }
1272     } else if (WIFSTOPPED(wstatus)) {
1273       /* It stopped, rather than exiting.  "Should not happen." */
1274       report_failure("Child capture process stopped: %s",
1275                      sync_pipe_signame(WSTOPSIG(wstatus)));
1276     } else if (WIFSIGNALED(wstatus)) {
1277       /* It died with a signal. */
1278       report_failure("Child capture process died: %s%s",
1279                      sync_pipe_signame(WTERMSIG(wstatus)),
1280                      WCOREDUMP(wstatus) ? " - core dumped" : "");
1281     } else {
1282       /* What?  It had to either have exited, or stopped, or died with
1283          a signal; what happened here? */
1284       report_failure("Child capture process died: wait status %#o", wstatus);
1285     }
1286   }
1287 #endif
1288
1289   /* No more child process. */
1290   capture_opts->fork_child = -1;
1291
1292   g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "sync_pipe_wait_for_child: capture child closed");
1293 }
1294
1295
1296 #ifndef _WIN32
1297 /* convert signal to corresponding name */
1298 static const char *
1299 sync_pipe_signame(int sig)
1300 {
1301   const char *sigmsg;
1302   static char sigmsg_buf[6+1+3+1];
1303
1304   switch (sig) {
1305
1306   case SIGHUP:
1307     sigmsg = "Hangup";
1308     break;
1309
1310   case SIGINT:
1311     sigmsg = "Interrupted";
1312     break;
1313
1314   case SIGQUIT:
1315     sigmsg = "Quit";
1316     break;
1317
1318   case SIGILL:
1319     sigmsg = "Illegal instruction";
1320     break;
1321
1322   case SIGTRAP:
1323     sigmsg = "Trace trap";
1324     break;
1325
1326   case SIGABRT:
1327     sigmsg = "Abort";
1328     break;
1329
1330   case SIGFPE:
1331     sigmsg = "Arithmetic exception";
1332     break;
1333
1334   case SIGKILL:
1335     sigmsg = "Killed";
1336     break;
1337
1338   case SIGBUS:
1339     sigmsg = "Bus error";
1340     break;
1341
1342   case SIGSEGV:
1343     sigmsg = "Segmentation violation";
1344     break;
1345
1346   /* http://metalab.unc.edu/pub/Linux/docs/HOWTO/GCC-HOWTO
1347      Linux is POSIX compliant.  These are not POSIX-defined signals ---
1348      ISO/IEC 9945-1:1990 (IEEE Std 1003.1-1990), paragraph B.3.3.1.1 sez:
1349
1350         ``The signals SIGBUS, SIGEMT, SIGIOT, SIGTRAP, and SIGSYS
1351         were omitted from POSIX.1 because their behavior is
1352         implementation dependent and could not be adequately catego-
1353         rized.  Conforming implementations may deliver these sig-
1354         nals, but must document the circumstances under which they
1355         are delivered and note any restrictions concerning their
1356         delivery.''
1357
1358      So we only check for SIGSYS on those systems that happen to
1359      implement them (a system can be POSIX-compliant and implement
1360      them, it's just that POSIX doesn't *require* a POSIX-compliant
1361      system to implement them).
1362    */
1363
1364 #ifdef SIGSYS
1365   case SIGSYS:
1366     sigmsg = "Bad system call";
1367     break;
1368 #endif
1369
1370   case SIGPIPE:
1371     sigmsg = "Broken pipe";
1372     break;
1373
1374   case SIGALRM:
1375     sigmsg = "Alarm clock";
1376     break;
1377
1378   case SIGTERM:
1379     sigmsg = "Terminated";
1380     break;
1381
1382   default:
1383         /* Returning a static buffer is ok in the context we use it here */
1384     g_snprintf(sigmsg_buf, sizeof sigmsg_buf, "Signal %d", sig);
1385     sigmsg = sigmsg_buf;
1386     break;
1387   }
1388   return sigmsg;
1389 }
1390 #endif
1391
1392
1393 #ifdef _WIN32
1394 /* tell the child through the signal pipe that we want to quit the capture */
1395 static void
1396 signal_pipe_capquit_to_child(capture_options *capture_opts)
1397 {
1398     const char quit_msg[] = "QUIT";
1399     int ret;
1400
1401
1402     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_DEBUG, "signal_pipe_capquit_to_child");
1403
1404     /* it doesn't matter *what* we send here, the first byte will stop the capture */
1405     /* simply sending a "QUIT" string */
1406     /*pipe_write_block(capture_opts->signal_pipe_write_fd, SP_QUIT, quit_msg);*/
1407     ret = write(capture_opts->signal_pipe_write_fd, quit_msg, sizeof quit_msg);
1408     if(ret == -1) {
1409         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_WARNING,
1410               "signal_pipe_capquit_to_child: %d header: error %s", capture_opts->signal_pipe_write_fd, strerror(errno));
1411     }
1412 }
1413 #endif
1414
1415
1416 /* user wants to stop the capture run */
1417 void
1418 sync_pipe_stop(capture_options *capture_opts)
1419 {
1420 #ifdef _WIN32
1421   int count;
1422   DWORD childstatus;
1423   gboolean terminate = TRUE;
1424 #endif
1425
1426   if (capture_opts->fork_child != -1) {
1427 #ifndef _WIN32
1428     /* send the SIGUSR1 signal to close the capture child gracefully. */
1429     int sts = kill(capture_opts->fork_child, SIGUSR1);
1430     if (sts != 0) {
1431         g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_WARNING,
1432               "Sending SIGUSR1 to child failed: %s\n", strerror(errno));
1433     }
1434 #else
1435 #define STOP_SLEEP_TIME 500 /* ms */
1436 #define STOP_CHECK_TIME 50
1437     /* First, use the special signal pipe to try to close the capture child
1438      * gracefully.
1439      */
1440     signal_pipe_capquit_to_child(capture_opts);
1441
1442     /* Next, wait for the process to exit on its own */
1443     for (count = 0; count < STOP_SLEEP_TIME / STOP_CHECK_TIME; count++) {
1444       if (GetExitCodeProcess((HANDLE) capture_opts->fork_child, &childstatus) &&
1445               childstatus != STILL_ACTIVE) {
1446         terminate = FALSE;
1447         break;
1448       }
1449       Sleep(STOP_CHECK_TIME);
1450     }
1451
1452     /* Force the issue. */
1453     if (terminate) {
1454       g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_WARNING,
1455             "sync_pipe_stop: forcing child to exit");
1456       sync_pipe_kill(capture_opts->fork_child);
1457     }
1458 #endif
1459   }
1460 }
1461
1462
1463 /* Wireshark has to exit, force the capture child to close */
1464 void
1465 sync_pipe_kill(int fork_child)
1466 {
1467     if (fork_child != -1) {
1468 #ifndef _WIN32
1469         int sts = kill(fork_child, SIGTERM);    /* SIGTERM so it can clean up if necessary */
1470         if (sts != 0) {
1471             g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_WARNING,
1472                   "Sending SIGTERM to child failed: %s\n", strerror(errno));
1473         }
1474 #else
1475       /* Remark: This is not the preferred method of closing a process!
1476        * the clean way would be getting the process id of the child process,
1477        * then getting window handle hWnd of that process (using EnumChildWindows),
1478        * and then do a SendMessage(hWnd, WM_CLOSE, 0, 0)
1479        *
1480        * Unfortunately, I don't know how to get the process id from the
1481        * handle.  OpenProcess will get an handle (not a window handle)
1482        * from the process ID; it will not get a window handle from the
1483        * process ID.  (How could it?  A process can have more than one
1484        * window.  For that matter, a process might have *no* windows,
1485        * as a process running dumpcap, the normal child process program,
1486        * probably does.)
1487        *
1488        * Hint: GenerateConsoleCtrlEvent() will only work if both processes are
1489        * running in the same console; that's not necessarily the case for
1490        * us, as we might not be running in a console.
1491        * And this also will require to have the process id.
1492        */
1493         TerminateProcess((HANDLE) (fork_child), 0);
1494 #endif
1495     }
1496 }
1497
1498 #endif /* HAVE_LIBPCAP */