change all file offsets from long to gint64 so we can - theoretically - handle files...
[obnox/wireshark/wip.git] / capture_loop.c
1 /* capture_loop.c
2  * The actual capturing loop, getting packets and storing it
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
26 /** @file
27  *
28  * Capture loop (internal interface).
29  *
30  * It will open the input and output files, capture the packets,
31  * change ringbuffer output files while capturing and close all files again.
32  *
33  * The input file can be a network interface or capture pipe (unix only).
34  * The output file can be a single or a ringbuffer file handled by wiretap.
35  *
36  */
37
38 #ifdef HAVE_CONFIG_H
39 # include "config.h"
40 #endif
41
42 #ifdef HAVE_LIBPCAP
43
44 #include <string.h>
45
46 #ifdef HAVE_FCNTL_H
47 #include <fcntl.h>
48 #endif
49
50 #ifdef HAVE_UNISTD_H
51 #include <unistd.h>
52 #endif
53
54 #ifdef HAVE_SYS_TYPES_H
55 # include <sys/types.h>
56 #endif
57
58 #ifdef HAVE_SYS_STAT_H
59 # include <sys/stat.h>
60 #endif
61
62 #include <signal.h>
63 #include <errno.h>
64 #include <setjmp.h>
65
66
67 #include <glib.h>
68
69 #include <pcap.h>
70
71 #include "pcapio.h"
72
73 #include "capture-pcap-util.h"
74
75 #include "capture.h"
76 #include "capture_sync.h"
77
78 #include "conditions.h"
79 #include "capture_stop_conditions.h"
80 #include "ringbuffer.h"
81
82 #include "simple_dialog.h"
83 #include "tempfile.h"
84 #include "log.h"
85 #include "file_util.h"
86
87 #include "epan/unicode-utils.h"
88
89 #include "capture_loop.h"
90
91 /*
92  * Standard secondary message for unexpected errors.
93  */
94 static const char please_report[] =
95     "Please report this to the Wireshark developers.  (This is not a crash;\n"
96     "please do not report it as such.)";
97
98 /*
99  * This needs to be static, so that the SIGUSR1 handler can clear the "go"
100  * flag.
101  */
102 static loop_data   ld;
103
104
105 /*
106  * Timeout, in milliseconds, for reads from the stream of captured packets.
107  */
108 #define CAP_READ_TIMEOUT        250
109 static char *cap_pipe_err_str;
110
111 static void capture_loop_packet_cb(u_char *user, const struct pcap_pkthdr *phdr,
112   const u_char *pd);
113 static void capture_loop_get_errmsg(char *errmsg, int errmsglen, const char *fname,
114                           int err, gboolean is_close);
115
116
117
118 /* Take care of byte order in the libpcap headers read from pipes.
119  * (function taken from wiretap/libpcap.c) */
120 static void
121 cap_pipe_adjust_header(gboolean byte_swapped, struct pcap_hdr *hdr, struct pcaprec_hdr *rechdr)
122 {
123   if (byte_swapped) {
124     /* Byte-swap the record header fields. */
125     rechdr->ts_sec = BSWAP32(rechdr->ts_sec);
126     rechdr->ts_usec = BSWAP32(rechdr->ts_usec);
127     rechdr->incl_len = BSWAP32(rechdr->incl_len);
128     rechdr->orig_len = BSWAP32(rechdr->orig_len);
129   }
130
131   /* In file format version 2.3, the "incl_len" and "orig_len" fields were
132      swapped, in order to match the BPF header layout.
133
134      Unfortunately, some files were, according to a comment in the "libpcap"
135      source, written with version 2.3 in their headers but without the
136      interchanged fields, so if "incl_len" is greater than "orig_len" - which
137      would make no sense - we assume that we need to swap them.  */
138   if (hdr->version_major == 2 &&
139       (hdr->version_minor < 3 ||
140        (hdr->version_minor == 3 && rechdr->incl_len > rechdr->orig_len))) {
141     guint32 temp;
142
143     temp = rechdr->orig_len;
144     rechdr->orig_len = rechdr->incl_len;
145     rechdr->incl_len = temp;
146   }
147 }
148
149 /* Provide select() functionality for a single file descriptor
150  * on both UNIX/POSIX and Windows.
151  *
152  * The Windows version calls WaitForSingleObject instead of
153  * select().
154  *
155  * Returns the same values as select.  If an error is returned,
156  * the string cap_pipe_err_str should be used instead of errno.
157  */
158 static int
159 cap_pipe_select(int pipe_fd, gboolean wait_forever) {
160 #ifndef _WIN32
161   fd_set      rfds;
162   struct timeval timeout, *pto;
163   int sel_ret;
164
165   cap_pipe_err_str = "Unknown error";
166
167   FD_ZERO(&rfds);
168   FD_SET(pipe_fd, &rfds);
169   if (wait_forever) {
170     pto = NULL;
171   } else {
172     timeout.tv_sec = 0;
173     timeout.tv_usec = CAP_READ_TIMEOUT * 1000;
174     pto = &timeout;
175   }
176   sel_ret = select(pipe_fd+1, &rfds, NULL, NULL, pto);
177   if (sel_ret < 0)
178     cap_pipe_err_str = strerror(errno);
179   return sel_ret;
180 }
181 #else
182   /* XXX - Should we just use file handles exclusively under Windows?
183    * Otherwise we have to convert between file handles and file descriptors
184    * here and when we open a named pipe.
185    */
186   HANDLE hPipe = (HANDLE) _get_osfhandle(pipe_fd);
187   wchar_t *err_str;
188   DWORD timeout = wait_forever ? INFINITE : CAP_READ_TIMEOUT * 1000;
189   DWORD wait_ret;
190
191   if (hPipe == INVALID_HANDLE_VALUE) {
192     cap_pipe_err_str = "Could not open standard input";
193     return -1;
194   }
195
196   cap_pipe_err_str = "Unknown error";
197
198   wait_ret = WaitForSingleObject(hPipe, timeout);
199   switch (wait_ret) {
200     /* XXX - This probably isn't correct */
201     case WAIT_ABANDONED:
202       errno = EINTR;
203       return -1;
204     case WAIT_OBJECT_0:
205       return 1;
206     case WAIT_TIMEOUT:
207       return 0;
208     case WAIT_FAILED:
209       FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
210         NULL, GetLastError(), 0, (LPTSTR) &err_str, 0, NULL);
211       cap_pipe_err_str = utf_16to8(err_str);
212       LocalFree(err_str);
213       return -1;
214     default:
215       g_assert_not_reached();
216       return -1;
217   }
218 }
219 #endif
220
221
222 /* Mimic pcap_open_live() for pipe captures
223  * We check if "pipename" is "-" (stdin) or a FIFO, open it, and read the
224  * header.
225  * N.B. : we can't read the libpcap formats used in RedHat 6.1 or SuSE 6.3
226  * because we can't seek on pipes (see wiretap/libpcap.c for details) */
227 static int
228 cap_pipe_open_live(char *pipename, struct pcap_hdr *hdr, loop_data *ld,
229                  char *errmsg, int errmsgl)
230 {
231 #ifndef _WIN32
232   struct stat pipe_stat;
233 #else
234 #if 0
235   char *pncopy, *pos;
236   wchar_t *err_str;
237 #endif
238   HANDLE hPipe = NULL;
239 #endif
240   int          sel_ret;
241   int          fd;
242   int          b;
243   guint32       magic;
244   unsigned int bytes_read;
245
246   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "cap_pipe_open_live: %s", pipename);
247   
248   if(pipename == NULL){
249     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_WARNING, "cap_pipe_open_live: pipe name is NULL");
250     g_snprintf(errmsg, errmsgl,
251                "The capture session could not be initiated because the pipe is not valid, maybe lack of privileges?");
252     return -1;
253   }
254
255   /*
256    * XXX (T)Wireshark blocks until we return
257    */
258   if (strcmp(pipename, "-") == 0) {
259     fd = 0; /* read from stdin */
260 #ifdef _WIN32
261     /*
262      * This is needed to set the stdin pipe into binary mode, otherwise
263      * CR/LF are mangled...
264      */
265     _setmode(0, _O_BINARY);
266 #endif  /* _WIN32 */
267   } else {
268 #ifndef _WIN32
269     if (eth_stat(pipename, &pipe_stat) < 0) {
270       if (errno == ENOENT || errno == ENOTDIR)
271         ld->cap_pipe_err = PIPNEXIST;
272       else {
273         g_snprintf(errmsg, errmsgl,
274           "The capture session could not be initiated "
275           "due to error on pipe: %s", strerror(errno));
276         ld->cap_pipe_err = PIPERR;
277       }
278       return -1;
279     }
280     if (! S_ISFIFO(pipe_stat.st_mode)) {
281       if (S_ISCHR(pipe_stat.st_mode)) {
282         /*
283          * Assume the user specified an interface on a system where
284          * interfaces are in /dev.  Pretend we haven't seen it.
285          */
286          ld->cap_pipe_err = PIPNEXIST;
287       } else
288       {
289         g_snprintf(errmsg, errmsgl,
290             "The capture session could not be initiated because\n"
291             "\"%s\" is neither an interface nor a pipe", pipename);
292         ld->cap_pipe_err = PIPERR;
293       }
294       return -1;
295     }
296     fd = eth_open(pipename, O_RDONLY | O_NONBLOCK, 0000 /* no creation so don't matter */);
297     if (fd == -1) {
298       g_snprintf(errmsg, errmsgl,
299           "The capture session could not be initiated "
300           "due to error on pipe open: %s", strerror(errno));
301       ld->cap_pipe_err = PIPERR;
302       return -1;
303     }
304 #else /* _WIN32 */
305 #if 0 /* Enable/disable Windows named pipes */
306 #define PIPE_STR "\\pipe\\"
307     /* Under Windows, named pipes _must_ have the form
308      * "\\<server>\pipe\<pipename>".  <server> may be "." for localhost.
309      */
310     pncopy = g_strdup(pipename);
311     if (strstr(pncopy, "\\\\") == pncopy) {
312       pos = strchr(pncopy + 3, '\\');
313       if (pos && g_strncasecmp(pos, PIPE_STR, strlen(PIPE_STR)) != 0)
314         pos = NULL;
315     }
316
317     g_free(pncopy);
318
319     if (!pos) {
320       g_snprintf(errmsg, errmsgl,
321           "The capture session could not be initiated because\n"
322           "\"%s\" is neither an interface nor a pipe", pipename);
323       ld->cap_pipe_err = PIPNEXIST;
324       return -1;
325     }
326
327     /* Wait for the pipe to appear */
328     while (1) {
329       hPipe = CreateFile(utf_8to16(pipename), GENERIC_READ, 0, NULL,
330           OPEN_EXISTING, 0, NULL);
331
332       if (hPipe != INVALID_HANDLE_VALUE)
333         break;
334
335       if (GetLastError() != ERROR_PIPE_BUSY) {
336         FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
337           NULL, GetLastError(), 0, (LPTSTR) &err_str, 0, NULL);
338         g_snprintf(errmsg, errmsgl,
339             "The capture session on \"%s\" could not be initiated "
340             "due to error on pipe open: pipe busy: %s (error %d)",
341             pipename, utf_16to8(err_str), GetLastError());
342         LocalFree(err_str);
343         ld->cap_pipe_err = PIPERR;
344         return -1;
345       }
346
347       if (!WaitNamedPipe(utf_8to16(pipename), 30 * 1000)) {
348         FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
349           NULL, GetLastError(), 0, (LPTSTR) &err_str, 0, NULL);
350         g_snprintf(errmsg, errmsgl,
351             "The capture session could not be initiated "
352             "due to error on pipe open: %s (error %d)",
353             utf_16to8(err_str), GetLastError());
354         LocalFree(err_str);
355         ld->cap_pipe_err = PIPERR;
356         return -1;
357       }
358     }
359
360     fd = _open_osfhandle((long) hPipe, _O_RDONLY);
361     if (fd == -1) {
362       g_snprintf(errmsg, errmsgl,
363           "The capture session could not be initiated "
364           "due to error on pipe open: %s", strerror(errno));
365       ld->cap_pipe_err = PIPERR;
366       return -1;
367     }
368 #else /* Enable/disable Windows named pipes */
369     /* On Windows, we don't support capturing on pipes, so we give up. */
370
371     g_snprintf(errmsg, errmsgl,
372 "The capture session could not be initiated.  Unable to open interface.");
373     return -1;
374 #endif /* Enable/disable Windows named pipes */
375 #endif /* _WIN32 */
376   }
377
378   ld->from_cap_pipe = TRUE;
379
380   /* read the pcap header */
381   bytes_read = 0;
382   while (bytes_read < sizeof magic) {
383     sel_ret = cap_pipe_select(fd, FALSE);
384     if (sel_ret < 0) {
385       g_snprintf(errmsg, errmsgl,
386         "Unexpected error from select: %s", strerror(errno));
387       goto error;
388     } else if (sel_ret > 0) {
389       b = read(fd, ((char *)&magic)+bytes_read, sizeof magic-bytes_read);
390       if (b <= 0) {
391         if (b == 0)
392           g_snprintf(errmsg, errmsgl, "End of file on pipe during open");
393         else
394           g_snprintf(errmsg, errmsgl, "Error on pipe during open: %s",
395             strerror(errno));
396         goto error;
397       }
398       bytes_read += b;
399     }
400   }
401
402   switch (magic) {
403   case PCAP_MAGIC:
404     /* Host that wrote it has our byte order, and was running
405        a program using either standard or ss990417 libpcap. */
406     ld->cap_pipe_byte_swapped = FALSE;
407     ld->cap_pipe_modified = FALSE;
408     break;
409   case PCAP_MODIFIED_MAGIC:
410     /* Host that wrote it has our byte order, but was running
411        a program using either ss990915 or ss991029 libpcap. */
412     ld->cap_pipe_byte_swapped = FALSE;
413     ld->cap_pipe_modified = TRUE;
414     break;
415   case PCAP_SWAPPED_MAGIC:
416     /* Host that wrote it has a byte order opposite to ours,
417        and was running a program using either standard or
418        ss990417 libpcap. */
419     ld->cap_pipe_byte_swapped = TRUE;
420     ld->cap_pipe_modified = FALSE;
421     break;
422   case PCAP_SWAPPED_MODIFIED_MAGIC:
423     /* Host that wrote it out has a byte order opposite to
424        ours, and was running a program using either ss990915
425        or ss991029 libpcap. */
426     ld->cap_pipe_byte_swapped = TRUE;
427     ld->cap_pipe_modified = TRUE;
428     break;
429   default:
430     /* Not a "libpcap" type we know about. */
431     g_snprintf(errmsg, errmsgl, "Unrecognized libpcap format");
432     goto error;
433   }
434
435   /* Read the rest of the header */
436   bytes_read = 0;
437   while (bytes_read < sizeof(struct pcap_hdr)) {
438     sel_ret = cap_pipe_select(fd, FALSE);
439     if (sel_ret < 0) {
440       g_snprintf(errmsg, errmsgl,
441         "Unexpected error from select: %s", strerror(errno));
442       goto error;
443     } else if (sel_ret > 0) {
444       b = read(fd, ((char *)hdr)+bytes_read,
445             sizeof(struct pcap_hdr) - bytes_read);
446       if (b <= 0) {
447         if (b == 0)
448           g_snprintf(errmsg, errmsgl, "End of file on pipe during open");
449         else
450           g_snprintf(errmsg, errmsgl, "Error on pipe during open: %s",
451             strerror(errno));
452         goto error;
453       }
454       bytes_read += b;
455     }
456   }
457
458   if (ld->cap_pipe_byte_swapped) {
459     /* Byte-swap the header fields about which we care. */
460     hdr->version_major = BSWAP16(hdr->version_major);
461     hdr->version_minor = BSWAP16(hdr->version_minor);
462     hdr->snaplen = BSWAP32(hdr->snaplen);
463     hdr->network = BSWAP32(hdr->network);
464   }
465   ld->linktype = hdr->network;
466
467   if (hdr->version_major < 2) {
468     g_snprintf(errmsg, errmsgl, "Unable to read old libpcap format");
469     goto error;
470   }
471
472   ld->cap_pipe_state = STATE_EXPECT_REC_HDR;
473   ld->cap_pipe_err = PIPOK;
474   return fd;
475
476 error:
477   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "cap_pipe_open_live: error %s", errmsg);
478   ld->cap_pipe_err = PIPERR;
479   eth_close(fd);
480   return -1;
481
482 }
483
484
485 /* We read one record from the pipe, take care of byte order in the record
486  * header, write the record to the capture file, and update capture statistics. */
487 static int
488 cap_pipe_dispatch(loop_data *ld, guchar *data, char *errmsg, int errmsgl)
489 {
490   struct pcap_pkthdr phdr;
491   int b;
492   enum { PD_REC_HDR_READ, PD_DATA_READ, PD_PIPE_EOF, PD_PIPE_ERR,
493           PD_ERR } result;
494
495
496 #ifdef LOG_CAPTURE_VERBOSE
497   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "cap_pipe_dispatch");
498 #endif
499
500   switch (ld->cap_pipe_state) {
501
502   case STATE_EXPECT_REC_HDR:
503     ld->cap_pipe_bytes_to_read = ld->cap_pipe_modified ?
504       sizeof(struct pcaprec_modified_hdr) : sizeof(struct pcaprec_hdr);
505     ld->cap_pipe_bytes_read = 0;
506     ld->cap_pipe_state = STATE_READ_REC_HDR;
507     /* Fall through */
508
509   case STATE_READ_REC_HDR:
510     b = read(ld->cap_pipe_fd, ((char *)&ld->cap_pipe_rechdr)+ld->cap_pipe_bytes_read,
511              ld->cap_pipe_bytes_to_read - ld->cap_pipe_bytes_read);
512     if (b <= 0) {
513       if (b == 0)
514         result = PD_PIPE_EOF;
515       else
516         result = PD_PIPE_ERR;
517       break;
518     }
519     if ((ld->cap_pipe_bytes_read += b) < ld->cap_pipe_bytes_to_read)
520         return 0;
521     result = PD_REC_HDR_READ;
522     break;
523
524   case STATE_EXPECT_DATA:
525     ld->cap_pipe_bytes_read = 0;
526     ld->cap_pipe_state = STATE_READ_DATA;
527     /* Fall through */
528
529   case STATE_READ_DATA:
530     b = read(ld->cap_pipe_fd, data+ld->cap_pipe_bytes_read,
531              ld->cap_pipe_rechdr.hdr.incl_len - ld->cap_pipe_bytes_read);
532     if (b <= 0) {
533       if (b == 0)
534         result = PD_PIPE_EOF;
535       else
536         result = PD_PIPE_ERR;
537       break;
538     }
539     if ((ld->cap_pipe_bytes_read += b) < ld->cap_pipe_rechdr.hdr.incl_len)
540       return 0;
541     result = PD_DATA_READ;
542     break;
543
544   default:
545     g_snprintf(errmsg, errmsgl, "cap_pipe_dispatch: invalid state");
546     result = PD_ERR;
547
548   } /* switch (ld->cap_pipe_state) */
549
550   /*
551    * We've now read as much data as we were expecting, so process it.
552    */
553   switch (result) {
554
555   case PD_REC_HDR_READ:
556     /* We've read the header. Take care of byte order. */
557     cap_pipe_adjust_header(ld->cap_pipe_byte_swapped, &ld->cap_pipe_hdr,
558                            &ld->cap_pipe_rechdr.hdr);
559     if (ld->cap_pipe_rechdr.hdr.incl_len > WTAP_MAX_PACKET_SIZE) {
560       g_snprintf(errmsg, errmsgl, "Frame %u too long (%d bytes)",
561         ld->packet_count+1, ld->cap_pipe_rechdr.hdr.incl_len);
562       break;
563     }
564     ld->cap_pipe_state = STATE_EXPECT_DATA;
565     return 0;
566
567   case PD_DATA_READ:
568     /* Fill in a "struct pcap_pkthdr", and process the packet. */
569     phdr.ts.tv_sec = ld->cap_pipe_rechdr.hdr.ts_sec;
570     phdr.ts.tv_usec = ld->cap_pipe_rechdr.hdr.ts_usec;
571     phdr.caplen = ld->cap_pipe_rechdr.hdr.incl_len;
572     phdr.len = ld->cap_pipe_rechdr.hdr.orig_len;
573
574     ld->packet_cb((u_char *)ld, &phdr, data);
575
576     ld->cap_pipe_state = STATE_EXPECT_REC_HDR;
577     return 1;
578
579   case PD_PIPE_EOF:
580     ld->cap_pipe_err = PIPEOF;
581     return -1;
582
583   case PD_PIPE_ERR:
584     g_snprintf(errmsg, errmsgl, "Error reading from pipe: %s",
585       strerror(errno));
586     /* Fall through */
587   case PD_ERR:
588     break;
589   }
590
591   ld->cap_pipe_err = PIPERR;
592   /* Return here rather than inside the switch to prevent GCC warning */
593   return -1;
594 }
595
596
597 /* open the capture input file (pcap or capture pipe) */
598 gboolean
599 capture_loop_open_input(capture_options *capture_opts, loop_data *ld,
600                         char *errmsg, size_t errmsg_len,
601                         char *secondary_errmsg, size_t secondary_errmsg_len)
602 {
603   gchar       open_err_str[PCAP_ERRBUF_SIZE];
604   gchar      *sync_msg_str;
605   static const char ppamsg[] = "can't find PPA for ";
606   const char *set_linktype_err_str;
607   const char  *libpcap_warn;
608 #ifdef _WIN32
609   gchar      *sync_secondary_msg_str;
610   int         err;
611   WORD        wVersionRequested;
612   WSADATA     wsaData;
613 #endif
614
615
616   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_open_input : %s", capture_opts->iface);
617
618
619 /* XXX - opening Winsock on tshark? */
620
621   /* Initialize Windows Socket if we are in a WIN32 OS
622      This needs to be done before querying the interface for network/netmask */
623 #ifdef _WIN32
624   /* XXX - do we really require 1.1 or earlier?
625      Are there any versions that support only 2.0 or higher? */
626   wVersionRequested = MAKEWORD(1, 1);
627   err = WSAStartup(wVersionRequested, &wsaData);
628   if (err != 0) {
629     switch (err) {
630
631     case WSASYSNOTREADY:
632       g_snprintf(errmsg, errmsg_len,
633         "Couldn't initialize Windows Sockets: Network system not ready for network communication");
634       break;
635
636     case WSAVERNOTSUPPORTED:
637       g_snprintf(errmsg, errmsg_len,
638         "Couldn't initialize Windows Sockets: Windows Sockets version %u.%u not supported",
639         LOBYTE(wVersionRequested), HIBYTE(wVersionRequested));
640       break;
641
642     case WSAEINPROGRESS:
643       g_snprintf(errmsg, errmsg_len,
644         "Couldn't initialize Windows Sockets: Blocking operation is in progress");
645       break;
646
647     case WSAEPROCLIM:
648       g_snprintf(errmsg, errmsg_len,
649         "Couldn't initialize Windows Sockets: Limit on the number of tasks supported by this WinSock implementation has been reached");
650       break;
651
652     case WSAEFAULT:
653       g_snprintf(errmsg, errmsg_len,
654         "Couldn't initialize Windows Sockets: Bad pointer passed to WSAStartup");
655       break;
656
657     default:
658       g_snprintf(errmsg, errmsg_len,
659         "Couldn't initialize Windows Sockets: error %d", err);
660       break;
661     }
662     g_snprintf(secondary_errmsg, secondary_errmsg_len, please_report);
663     return FALSE;
664   }
665 #endif
666
667   /* Open the network interface to capture from it.
668      Some versions of libpcap may put warnings into the error buffer
669      if they succeed; to tell if that's happened, we have to clear
670      the error buffer, and check if it's still a null string.  */
671   open_err_str[0] = '\0';
672   ld->pcap_h = pcap_open_live(capture_opts->iface,
673                        capture_opts->has_snaplen ? capture_opts->snaplen :
674                                                   WTAP_MAX_PACKET_SIZE,
675                        capture_opts->promisc_mode, CAP_READ_TIMEOUT,
676                        open_err_str);
677
678   if (ld->pcap_h != NULL) {
679     /* we've opened "iface" as a network device */
680 #ifdef _WIN32
681     /* try to set the capture buffer size */
682     if (pcap_setbuff(ld->pcap_h, capture_opts->buffer_size * 1024 * 1024) != 0) {
683         sync_secondary_msg_str = g_strdup_printf(
684           "The capture buffer size of %luMB seems to be too high for your machine,\n"
685           "the default of 1MB will be used.\n"
686           "\n"
687           "Nonetheless, the capture is started.\n",
688           capture_opts->buffer_size);
689         report_capture_error("Couldn't set the capture buffer size!",
690                                    sync_secondary_msg_str);
691         g_free(sync_secondary_msg_str);
692     }
693 #endif
694
695     /* setting the data link type only works on real interfaces */
696     if (capture_opts->linktype != -1) {
697       set_linktype_err_str = set_pcap_linktype(ld->pcap_h, capture_opts->iface,
698         capture_opts->linktype);
699       if (set_linktype_err_str != NULL) {
700         g_snprintf(errmsg, errmsg_len, "Unable to set data link type (%s).",
701                    set_linktype_err_str);
702         g_snprintf(secondary_errmsg, secondary_errmsg_len, please_report);
703         return FALSE;
704       }
705     }
706     ld->linktype = get_pcap_linktype(ld->pcap_h, capture_opts->iface);
707   } else {
708     /* We couldn't open "iface" as a network device. */
709     /* Try to open it as a pipe */
710     ld->cap_pipe_fd = cap_pipe_open_live(capture_opts->iface, &ld->cap_pipe_hdr, ld, errmsg, errmsg_len);
711
712     if (ld->cap_pipe_fd == -1) {
713
714       if (ld->cap_pipe_err == PIPNEXIST) {
715         /* Pipe doesn't exist, so output message for interface */
716
717         /* If we got a "can't find PPA for X" message, warn the user (who
718            is running (T)Wireshark on HP-UX) that they don't have a version
719            of libpcap that properly handles HP-UX (libpcap 0.6.x and later
720            versions, which properly handle HP-UX, say "can't find /dev/dlpi
721            PPA for X" rather than "can't find PPA for X"). */
722         if (strncmp(open_err_str, ppamsg, sizeof ppamsg - 1) == 0)
723           libpcap_warn =
724             "\n\n"
725             "You are running (T)Wireshark with a version of the libpcap library\n"
726             "that doesn't handle HP-UX network devices well; this means that\n"
727             "(T)Wireshark may not be able to capture packets.\n"
728             "\n"
729             "To fix this, you should install libpcap 0.6.2, or a later version\n"
730             "of libpcap, rather than libpcap 0.4 or 0.5.x.  It is available in\n"
731             "packaged binary form from the Software Porting And Archive Centre\n"
732             "for HP-UX; the Centre is at http://hpux.connect.org.uk/ - the page\n"
733             "at the URL lists a number of mirror sites.";
734         else
735           libpcap_warn = "";
736         g_snprintf(errmsg, errmsg_len,
737           "The capture session could not be initiated (%s).", open_err_str);
738 #ifndef _WIN32
739         g_snprintf(secondary_errmsg, secondary_errmsg_len,
740 "Please check to make sure you have sufficient permissions, and that you have\n"
741 "the proper interface or pipe specified.%s", libpcap_warn);
742 #else
743     g_snprintf(secondary_errmsg, secondary_errmsg_len,
744 "\n"
745 "Please check that \"%s\" is the proper interface.\n"
746 "\n"
747 "\n"
748 "Help can be found at:\n"
749 "\n"
750 "       http://wiki.wireshark.org/CaptureSetup\n"
751 "\n"
752 "64-bit Windows:\n"
753 "WinPcap does not support 64-bit Windows; you will have to use some other\n"
754 "tool to capture traffic, such as netcap.\n"
755 "For netcap details see: http://support.microsoft.com/?id=310875\n"
756 "\n"
757 "Modem (PPP/WAN):\n"
758 "Note that version 3.0 of WinPcap, and earlier versions of WinPcap, don't\n"
759 "support capturing on PPP/WAN interfaces on Windows NT 4.0 / 2000 / XP /\n"
760 "Server 2003.\n"
761 "WinPcap 3.1 has support for it on Windows 2000 / XP / Server 2003, but has no\n"
762 "support for it on Windows NT 4.0 or Windows Vista (Beta 1).",
763     capture_opts->iface);
764 #endif /* _WIN32 */
765       }
766       /*
767        * Else pipe (or file) does exist and cap_pipe_open_live() has
768        * filled in errmsg
769        */
770       return FALSE;
771     } else
772       /* cap_pipe_open_live() succeeded; don't want
773          error message from pcap_open_live() */
774       open_err_str[0] = '\0';
775   }
776
777 /* XXX - will this work for tshark? */
778 #ifdef MUST_DO_SELECT
779   if (!ld->from_cap_pipe) {
780 #ifdef HAVE_PCAP_GET_SELECTABLE_FD
781     ld->pcap_fd = pcap_get_selectable_fd(ld->pcap_h);
782 #else
783     ld->pcap_fd = pcap_fileno(ld->pcap_h);
784 #endif
785   }
786 #endif
787
788   /* Does "open_err_str" contain a non-empty string?  If so, "pcap_open_live()"
789      returned a warning; print it, but keep capturing. */
790   if (open_err_str[0] != '\0') {
791     sync_msg_str = g_strdup_printf("%s.", open_err_str);
792     report_capture_error(sync_msg_str, "");
793     g_free(sync_msg_str);
794   }
795
796   return TRUE;
797 }
798
799
800 /* close the capture input file (pcap or capture pipe) */
801 static void capture_loop_close_input(loop_data *ld) {
802
803   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_close_input");
804
805   /* if open, close the capture pipe "input file" */
806   if (ld->cap_pipe_fd >= 0) {
807     g_assert(ld->from_cap_pipe);
808     eth_close(ld->cap_pipe_fd);
809     ld->cap_pipe_fd = 0;
810   }
811
812   /* if open, close the pcap "input file" */
813   if(ld->pcap_h != NULL) {
814     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_close_input: closing %p", ld->pcap_h);
815     g_assert(!ld->from_cap_pipe);
816     pcap_close(ld->pcap_h);
817     ld->pcap_h = NULL;
818   }
819
820   ld->go = FALSE;
821   
822 #ifdef _WIN32
823   /* Shut down windows sockets */
824   WSACleanup();
825 #endif
826 }
827
828
829 /* init the capture filter */
830 initfilter_status_t capture_loop_init_filter(pcap_t *pcap_h, gboolean from_cap_pipe, const gchar * iface, gchar * cfilter) {
831   bpf_u_int32 netnum, netmask;
832   gchar       lookup_net_err_str[PCAP_ERRBUF_SIZE];
833   struct bpf_program fcode;
834
835
836   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_init_filter: %s", cfilter);
837
838   /* capture filters only work on real interfaces */
839   if (cfilter && !from_cap_pipe) {
840     /* A capture filter was specified; set it up. */
841     if (pcap_lookupnet(iface, &netnum, &netmask, lookup_net_err_str) < 0) {
842       /*
843        * Well, we can't get the netmask for this interface; it's used
844        * only for filters that check for broadcast IP addresses, so
845        * we just punt and use 0.  It might be nice to warn the user,
846        * but that's a pain in a GUI application, as it'd involve popping
847        * up a message box, and it's not clear how often this would make
848        * a difference (only filters that check for IP broadcast addresses
849        * use the netmask).
850        */
851       /*cmdarg_err(
852         "Warning:  Couldn't obtain netmask info (%s).", lookup_net_err_str);*/
853       netmask = 0;
854     }
855     if (pcap_compile(pcap_h, &fcode, cfilter, 1, netmask) < 0) {
856       /* Treat this specially - our caller might try to compile this
857          as a display filter and, if that succeeds, warn the user that
858          the display and capture filter syntaxes are different. */
859       return INITFILTER_BAD_FILTER;
860     }
861     if (pcap_setfilter(pcap_h, &fcode) < 0) {
862 #ifdef HAVE_PCAP_FREECODE
863       pcap_freecode(&fcode);
864 #endif
865       return INITFILTER_OTHER_ERROR;
866     }
867 #ifdef HAVE_PCAP_FREECODE
868     pcap_freecode(&fcode);
869 #endif
870   }
871
872   return INITFILTER_NO_ERROR;
873 }
874
875
876 /* set up to write to the already-opened capture output file/files */
877 gboolean capture_loop_init_output(capture_options *capture_opts, int save_file_fd, loop_data *ld, char *errmsg, int errmsg_len) {
878   int         file_snaplen;
879   int         err;
880
881
882   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_init_output");
883
884   /* get snaplen */
885   if (ld->from_cap_pipe) {
886     file_snaplen = ld->cap_pipe_hdr.snaplen;
887   } else
888   {
889     file_snaplen = pcap_snapshot(ld->pcap_h);
890   }
891
892   /* Set up to write to the capture file. */
893   if (capture_opts->multi_files_on) {
894     ld->pdh = ringbuf_init_libpcap_fdopen(ld->linktype, file_snaplen,
895                                           &ld->bytes_written, &err);
896   } else {
897     ld->pdh = libpcap_fdopen(save_file_fd, ld->linktype, file_snaplen,
898                              &ld->bytes_written, &err);
899   }
900
901   if (ld->pdh == NULL) {
902     /* We couldn't set up to write to the capture file. */
903     /* XXX - use cf_open_error_message from tshark instead? */
904     switch (err) {
905
906     case WTAP_ERR_CANT_OPEN:
907       strcpy(errmsg, "The file to which the capture would be saved"
908                " couldn't be created for some unknown reason.");
909       break;
910
911     case WTAP_ERR_SHORT_WRITE:
912       strcpy(errmsg, "A full header couldn't be written to the file"
913                " to which the capture would be saved.");
914       break;
915
916     default:
917       if (err < 0) {
918         g_snprintf(errmsg, errmsg_len,
919                      "The file to which the capture would be"
920                      " saved (\"%s\") could not be opened: Error %d.",
921                         capture_opts->save_file, err);
922       } else {
923         g_snprintf(errmsg, errmsg_len,
924                      "The file to which the capture would be"
925                      " saved (\"%s\") could not be opened: %s.",
926                         capture_opts->save_file, strerror(err));
927       }
928       break;
929     }
930
931     return FALSE;
932   }
933
934   return TRUE;
935 }
936
937 gboolean capture_loop_close_output(capture_options *capture_opts, loop_data *ld, int *err_close) {
938
939   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_close_output");
940
941   if (capture_opts->multi_files_on) {
942     return ringbuf_libpcap_dump_close(&capture_opts->save_file, err_close);
943   } else {
944     return libpcap_dump_close(ld->pdh, err_close);
945   }
946 }
947
948 /* dispatch incoming packets (pcap or capture pipe) */
949 int
950 capture_loop_dispatch(capture_options *capture_opts _U_, loop_data *ld,
951                       char *errmsg, int errmsg_len) {
952   int       inpkts;
953   int         sel_ret;
954   guchar pcap_data[WTAP_MAX_PACKET_SIZE];
955
956   if (ld->from_cap_pipe) {
957     /* dispatch from capture pipe */
958 #ifdef LOG_CAPTURE_VERBOSE
959     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from capture pipe");
960 #endif
961     sel_ret = cap_pipe_select(ld->cap_pipe_fd, FALSE);
962     if (sel_ret <= 0) {
963       inpkts = 0;
964       if (sel_ret < 0 && errno != EINTR) {
965         g_snprintf(errmsg, errmsg_len,
966           "Unexpected error from select: %s", strerror(errno));
967         report_capture_error(errmsg, please_report);
968         ld->go = FALSE;
969       }
970     } else {
971       /*
972        * "select()" says we can read from the pipe without blocking
973        */
974       inpkts = cap_pipe_dispatch(ld, pcap_data, errmsg, errmsg_len);
975       if (inpkts < 0) {
976         ld->go = FALSE;
977       }
978     }
979   }
980   else
981   {
982     /* dispatch from pcap */
983 #ifdef MUST_DO_SELECT
984     /*
985      * If we have "pcap_get_selectable_fd()", we use it to get the
986      * descriptor on which to select; if that's -1, it means there
987      * is no descriptor on which you can do a "select()" (perhaps
988      * because you're capturing on a special device, and that device's
989      * driver unfortunately doesn't support "select()", in which case
990      * we don't do the select - which means it might not be possible
991      * to stop a capture until a packet arrives.  If that's unacceptable,
992      * plead with whoever supplies the software for that device to add
993      * "select()" support, or upgrade to libpcap 0.8.1 or later, and
994      * rebuild Wireshark or get a version built with libpcap 0.8.1 or
995      * later, so it can use pcap_breakloop().
996      */
997 #ifdef LOG_CAPTURE_VERBOSE
998     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_dispatch with select");
999 #endif
1000     if (ld->pcap_fd != -1) {
1001       sel_ret = cap_pipe_select(ld->pcap_fd, TRUE);
1002       if (sel_ret > 0) {
1003         /*
1004          * "select()" says we can read from it without blocking; go for
1005          * it.
1006          *
1007          * We don't have pcap_breakloop(), so we only process one packet
1008          * per pcap_dispatch() call, to allow a signal to stop the
1009          * processing immediately, rather than processing all packets
1010          * in a batch before quitting.
1011          */
1012         inpkts = pcap_dispatch(ld->pcap_h, 1, ld->packet_cb, (u_char *)ld);
1013         if (inpkts < 0) {
1014           ld->pcap_err = TRUE;
1015           ld->go = FALSE; /* error or pcap_breakloop() - stop capturing */
1016         }
1017       } else {
1018         inpkts = 0;
1019         if (sel_ret < 0 && errno != EINTR) {
1020           g_snprintf(errmsg, errmsg_len,
1021             "Unexpected error from select: %s", strerror(errno));
1022           report_capture_error(errmsg, please_report);
1023           ld->go = FALSE;
1024         }
1025       }
1026     }
1027     else
1028 #endif /* MUST_DO_SELECT */
1029     {
1030       /* dispatch from pcap without select */
1031 #if 1
1032 #ifdef LOG_CAPTURE_VERBOSE
1033       g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_dispatch");
1034 #endif
1035       if(ld->pcap_h){
1036        /* libpcap crashed when the pcap handle is NULL!!*/
1037 #ifdef _WIN32
1038       /*
1039        * On Windows, we don't support asynchronously telling a process to
1040        * stop capturing; instead, we check for an indication on a pipe
1041        * after processing packets.  We therefore process only one packet
1042        * at a time, so that we can check the pipe after every packet.
1043        */
1044       inpkts = pcap_dispatch(ld->pcap_h, 1, ld->packet_cb, (u_char *) ld);
1045 #else
1046       inpkts = pcap_dispatch(ld->pcap_h, -1, ld->packet_cb, (u_char *) ld);
1047 #endif
1048      } else {
1049         inpkts = -1;
1050       }
1051       if (inpkts < 0) {
1052         if (inpkts == -1) {
1053           /* Error, rather than pcap_breakloop(). */
1054           ld->pcap_err = TRUE;
1055         }
1056         ld->go = FALSE; /* error or pcap_breakloop() - stop capturing */
1057       }
1058 #else /* pcap_next_ex */
1059 #ifdef LOG_CAPTURE_VERBOSE
1060       g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: from pcap_next_ex");
1061 #endif
1062       /* XXX - this is currently unused, as there is some confusion with pcap_next_ex() vs. pcap_dispatch() */
1063
1064       /*
1065        * WinPcap's remote capturing feature doesn't work with pcap_dispatch(),
1066        * see http://wiki.wireshark.org/CaptureSetup_2fWinPcapRemote
1067        * This should be fixed in the WinPcap 4.0 alpha release.
1068        *
1069        * For reference, an example remote interface:
1070        * rpcap://[1.2.3.4]/\Device\NPF_{39993D68-7C9B-4439-A329-F2D888DA7C5C}
1071        */
1072
1073       /* emulate dispatch from pcap */
1074       {
1075         int in;
1076         struct pcap_pkthdr *pkt_header;
1077         u_char *pkt_data;
1078
1079         inpkts = 0;
1080         in = 0;
1081         while(ld->go &&
1082               (in = pcap_next_ex(ld->pcap_h, &pkt_header, &pkt_data)) == 1) {
1083           ld->packet_cb( (u_char *) ld, pkt_header, pkt_data);
1084           inpkts++;
1085         }
1086
1087         if(in < 0) {
1088           ld->pcap_err = TRUE;
1089           ld->go = FALSE;
1090           inpkts = in;
1091         }
1092       }
1093 #endif /* pcap_next_ex */
1094     }
1095   }
1096
1097 #ifdef LOG_CAPTURE_VERBOSE
1098   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_dispatch: %d new packet%s", inpkts, plurality(inpkts, "", "s"));
1099 #endif
1100
1101   return inpkts;
1102 }
1103
1104
1105 /* open the output file (temporary/specified name/ringbuffer/named pipe/stdout) */
1106 /* Returns TRUE if the file opened successfully, FALSE otherwise. */
1107 gboolean
1108 capture_loop_open_output(capture_options *capture_opts, int *save_file_fd,
1109                       char *errmsg, int errmsg_len) {
1110
1111   char tmpname[128+1];
1112   gchar *capfile_name;
1113   gboolean is_tempfile;
1114
1115
1116   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "capture_loop_open_output: %s",
1117       (capture_opts->save_file) ? capture_opts->save_file : "");
1118
1119   if (capture_opts->save_file != NULL) {
1120     /* We return to the caller while the capture is in progress.
1121      * Therefore we need to take a copy of save_file in
1122      * case the caller destroys it after we return.
1123      */
1124     capfile_name = g_strdup(capture_opts->save_file);
1125     if (strcmp(capfile_name, "-") == 0) {
1126       /* Write to the standard output. */
1127       if (capture_opts->multi_files_on) {
1128         /* ringbuffer is enabled; that doesn't work with standard output */
1129         g_snprintf(errmsg, errmsg_len,
1130             "Ring buffer requested, but capture is being written to the standard output.");
1131         g_free(capfile_name);
1132         return FALSE;
1133       } else {
1134         *save_file_fd = 1;
1135       }
1136     } else {
1137       if (capture_opts->multi_files_on) {
1138         /* ringbuffer is enabled */
1139         *save_file_fd = ringbuf_init(capfile_name,
1140             (capture_opts->has_ring_num_files) ? capture_opts->ring_num_files : 0);
1141
1142         /* we need the ringbuf name */
1143         if(*save_file_fd != -1) {
1144             g_free(capfile_name);
1145             capfile_name = g_strdup(ringbuf_current_filename());
1146         }
1147       } else {
1148         /* Try to open/create the specified file for use as a capture buffer. */
1149         *save_file_fd = open(capfile_name, O_RDWR|O_BINARY|O_TRUNC|O_CREAT,
1150                              0600);
1151       }
1152     }
1153     is_tempfile = FALSE;
1154   } else {
1155     /* Choose a random name for the temporary capture buffer */
1156     *save_file_fd = create_tempfile(tmpname, sizeof tmpname, "ether");
1157     capfile_name = g_strdup(tmpname);
1158     is_tempfile = TRUE;
1159   }
1160
1161   /* did we fail to open the output file? */
1162   if (*save_file_fd == -1) {
1163     if (is_tempfile) {
1164       g_snprintf(errmsg, errmsg_len,
1165         "The temporary file to which the capture would be saved (\"%s\") "
1166         "could not be opened: %s.", capfile_name, strerror(errno));
1167     } else {
1168       if (capture_opts->multi_files_on) {
1169         ringbuf_error_cleanup();
1170       }
1171
1172       g_snprintf(errmsg, errmsg_len,
1173             "The file to which the capture would be saved (\"%s\") "
1174         "could not be opened: %s.", capfile_name,
1175         strerror(errno));
1176     }
1177     g_free(capfile_name);
1178     return FALSE;
1179   }
1180
1181   if(capture_opts->save_file != NULL) {
1182     g_free(capture_opts->save_file);
1183   }
1184   capture_opts->save_file = capfile_name;
1185   /* capture_opts.save_file is "g_free"ed later, which is equivalent to
1186      "g_free(capfile_name)". */
1187
1188   return TRUE;
1189 }
1190
1191
1192 static void
1193 capture_loop_stop_signal_handler(int signo _U_)
1194 {
1195   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Signal: Stop capture");
1196   capture_loop_stop();
1197 }
1198
1199 #ifdef _WIN32
1200 #define TIME_GET() GetTickCount()
1201 #else
1202 #define TIME_GET() time(NULL)
1203 #endif
1204
1205 /* Do the low-level work of a capture.
1206    Returns TRUE if it succeeds, FALSE otherwise. */
1207 int
1208 capture_loop_start(capture_options *capture_opts, gboolean *stats_known, struct pcap_stat *stats)
1209 {
1210 #ifndef _WIN32
1211   struct sigaction act;
1212 #endif
1213   time_t      upd_time, cur_time;
1214   time_t      start_time;
1215   int         err_close;
1216   int         inpkts;
1217   gint        inpkts_to_sync_pipe = 0;     /* packets not already send out to the sync_pipe */
1218   condition  *cnd_file_duration = NULL;
1219   condition  *cnd_autostop_files = NULL;
1220   condition  *cnd_autostop_size = NULL;
1221   condition  *cnd_autostop_duration = NULL;
1222   guint32     autostop_files = 0;
1223   gboolean    write_ok;
1224   gboolean    close_ok;
1225   gboolean    cfilter_error = FALSE;
1226 #define MSG_MAX_LENGTH 4096+1
1227   char        errmsg[MSG_MAX_LENGTH] = "";
1228   char        secondary_errmsg[MSG_MAX_LENGTH] = "";
1229   int         save_file_fd = -1;
1230
1231   if(capture_opts == NULL){
1232     return FALSE;
1233   }
1234
1235   /* init the loop data */
1236   ld.go                 = TRUE;
1237   ld.packet_count       = 0;
1238   if (capture_opts->has_autostop_packets)
1239     ld.packet_max       = capture_opts->autostop_packets;
1240   else
1241     ld.packet_max       = 0;    /* no limit */
1242   ld.err                = 0;    /* no error seen yet */
1243   ld.wtap_linktype      = WTAP_ENCAP_UNKNOWN;
1244   ld.pcap_err           = FALSE;
1245   ld.from_cap_pipe      = FALSE;
1246   ld.pdh                = NULL;
1247   ld.cap_pipe_fd        = -1;
1248 #ifdef MUST_DO_SELECT
1249   ld.pcap_fd            = 0;
1250 #endif
1251   ld.packet_cb          = capture_loop_packet_cb;
1252
1253
1254   /* We haven't yet gotten the capture statistics. */
1255   *stats_known      = FALSE;
1256
1257 #ifndef _WIN32
1258   /*
1259    * Catch SIGUSR1, so that we exit cleanly if the parent process
1260    * kills us with it due to the user selecting "Capture->Stop".
1261    */
1262   act.sa_handler = capture_loop_stop_signal_handler;
1263   /*
1264    * Arrange that system calls not get restarted, because when
1265    * our signal handler returns we don't want to restart
1266    * a call that was waiting for packets to arrive.
1267    */
1268   act.sa_flags = 0;
1269   sigemptyset(&act.sa_mask);
1270   sigaction(SIGUSR1, &act, NULL);
1271 #endif /* _WIN32 */
1272
1273   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop starting ...");
1274   capture_opts_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, capture_opts);
1275
1276   /* open the "input file" from network interface or capture pipe */
1277   if (!capture_loop_open_input(capture_opts, &ld, errmsg, sizeof(errmsg),
1278                                secondary_errmsg, sizeof(secondary_errmsg))) {
1279     goto error;
1280   }
1281
1282   /* init the input filter from the network interface (capture pipe will do nothing) */
1283   switch (capture_loop_init_filter(ld.pcap_h, ld.from_cap_pipe, capture_opts->iface, capture_opts->cfilter)) {
1284
1285   case INITFILTER_NO_ERROR:
1286     break;
1287
1288   case INITFILTER_BAD_FILTER:
1289     cfilter_error = TRUE;
1290     g_snprintf(errmsg, sizeof(errmsg), "%s", pcap_geterr(ld.pcap_h));
1291     *secondary_errmsg = '\0';
1292     goto error;
1293
1294   case INITFILTER_OTHER_ERROR:
1295     g_snprintf(errmsg, sizeof(errmsg), "Can't install filter (%s).",
1296                pcap_geterr(ld.pcap_h));
1297     g_snprintf(secondary_errmsg, sizeof(secondary_errmsg), "%s", please_report);
1298     goto error;
1299   }
1300
1301   /* If we're supposed to write to a capture file, open it for output
1302      (temporary/specified name/ringbuffer) */
1303   if (capture_opts->saving_to_file) {
1304     if (!capture_loop_open_output(capture_opts, &save_file_fd, errmsg, sizeof(errmsg))) {
1305       *secondary_errmsg = '\0';
1306       goto error;
1307     }
1308
1309     /* set up to write to the already-opened capture output file/files */
1310     if (!capture_loop_init_output(capture_opts, save_file_fd, &ld, errmsg, sizeof(errmsg))) {
1311       *secondary_errmsg = '\0';
1312       goto error;
1313     }
1314
1315   /* XXX - capture SIGTERM and close the capture, in case we're on a
1316      Linux 2.0[.x] system and you have to explicitly close the capture
1317      stream in order to turn promiscuous mode off?  We need to do that
1318      in other places as well - and I don't think that works all the
1319      time in any case, due to libpcap bugs. */
1320
1321     /* Well, we should be able to start capturing.
1322
1323        Sync out the capture file, so the header makes it to the file system,
1324        and send a "capture started successfully and capture file created"
1325        message to our parent so that they'll open the capture file and
1326        update its windows to indicate that we have a live capture in
1327        progress. */
1328     libpcap_dump_flush(ld.pdh, NULL);
1329     report_new_capture_file(capture_opts->save_file);
1330   }
1331
1332   /* initialize capture stop (and alike) conditions */
1333   init_capture_stop_conditions();
1334   /* create stop conditions */
1335   if (capture_opts->has_autostop_filesize)
1336     cnd_autostop_size =
1337         cnd_new(CND_CLASS_CAPTURESIZE,(long)capture_opts->autostop_filesize * 1024);
1338   if (capture_opts->has_autostop_duration)
1339     cnd_autostop_duration =
1340         cnd_new(CND_CLASS_TIMEOUT,(gint32)capture_opts->autostop_duration);
1341
1342   if (capture_opts->multi_files_on) {
1343       if (capture_opts->has_file_duration)
1344         cnd_file_duration =
1345             cnd_new(CND_CLASS_TIMEOUT, capture_opts->file_duration);
1346
1347       if (capture_opts->has_autostop_files)
1348         cnd_autostop_files =
1349             cnd_new(CND_CLASS_CAPTURESIZE, capture_opts->autostop_files);
1350   }
1351
1352   /* init the time values */
1353   start_time = TIME_GET();
1354   upd_time = TIME_GET();
1355
1356   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop running!");
1357
1358   /* WOW, everything is prepared! */
1359   /* please fasten your seat belts, we will enter now the actual capture loop */
1360   while (ld.go) {
1361     /* dispatch incoming packets */
1362     inpkts = capture_loop_dispatch(capture_opts, &ld, errmsg, sizeof(errmsg));
1363
1364 #ifdef _WIN32
1365     /* any news from our parent (signal pipe)? -> just stop the capture */
1366     if (!signal_pipe_check_running()) {
1367       ld.go = FALSE;
1368     }
1369 #endif
1370
1371     if (inpkts > 0) {
1372       inpkts_to_sync_pipe += inpkts;
1373
1374       /* check capture size condition */
1375       if (cnd_autostop_size != NULL &&
1376           cnd_eval(cnd_autostop_size, (guint32)ld.bytes_written)){
1377         /* Capture size limit reached, do we have another file? */
1378         if (capture_opts->multi_files_on) {
1379           if (cnd_autostop_files != NULL &&
1380               cnd_eval(cnd_autostop_files, ++autostop_files)) {
1381              /* no files left: stop here */
1382             ld.go = FALSE;
1383             continue;
1384           }
1385
1386           /* Switch to the next ringbuffer file */
1387           if (ringbuf_switch_file(&ld.pdh, &capture_opts->save_file,
1388               &save_file_fd, &ld.bytes_written, &ld.err)) {
1389             /* File switch succeeded: reset the conditions */
1390             cnd_reset(cnd_autostop_size);
1391             if (cnd_file_duration) {
1392               cnd_reset(cnd_file_duration);
1393             }
1394             libpcap_dump_flush(ld.pdh, NULL);
1395             report_packet_count(inpkts_to_sync_pipe);
1396             inpkts_to_sync_pipe = 0;
1397             report_new_capture_file(capture_opts->save_file);
1398           } else {
1399             /* File switch failed: stop here */
1400             ld.go = FALSE;
1401             continue;
1402           }
1403         } else {
1404           /* single file, stop now */
1405           ld.go = FALSE;
1406           continue;
1407         }
1408       } /* cnd_autostop_size */
1409       if (capture_opts->output_to_pipe) {
1410         libpcap_dump_flush(ld.pdh, NULL);
1411       }
1412     } /* inpkts */
1413
1414     /* Only update once a second (Win32: 500ms) so as not to overload slow displays */
1415     cur_time = TIME_GET();
1416 #ifdef _WIN32
1417     if ( (cur_time - upd_time) > 500) {
1418 #else
1419     if (cur_time - upd_time > 0) {
1420 #endif
1421         upd_time = cur_time;
1422
1423       /*if (pcap_stats(pch, stats) >= 0) {
1424         *stats_known = TRUE;
1425       }*/
1426
1427       /* Let the parent process know. */
1428       if (inpkts_to_sync_pipe) {
1429         /* do sync here */
1430         libpcap_dump_flush(ld.pdh, NULL);
1431
1432         /* Send our parent a message saying we've written out "inpkts_to_sync_pipe"
1433            packets to the capture file. */
1434         report_packet_count(inpkts_to_sync_pipe);
1435
1436         inpkts_to_sync_pipe = 0;
1437       }
1438
1439       /* check capture duration condition */
1440       if (cnd_autostop_duration != NULL && cnd_eval(cnd_autostop_duration)) {
1441         /* The maximum capture time has elapsed; stop the capture. */
1442         ld.go = FALSE;
1443         continue;
1444       }
1445
1446       /* check capture file duration condition */
1447       if (cnd_file_duration != NULL && cnd_eval(cnd_file_duration)) {
1448         /* duration limit reached, do we have another file? */
1449         if (capture_opts->multi_files_on) {
1450           if (cnd_autostop_files != NULL &&
1451               cnd_eval(cnd_autostop_files, ++autostop_files)) {
1452             /* no files left: stop here */
1453             ld.go = FALSE;
1454             continue;
1455           }
1456
1457           /* Switch to the next ringbuffer file */
1458           if (ringbuf_switch_file(&ld.pdh, &capture_opts->save_file,
1459                                   &save_file_fd, &ld.bytes_written, &ld.err)) {
1460             /* file switch succeeded: reset the conditions */
1461             cnd_reset(cnd_file_duration);
1462             if(cnd_autostop_size)
1463               cnd_reset(cnd_autostop_size);
1464             libpcap_dump_flush(ld.pdh, NULL);
1465             report_packet_count(inpkts_to_sync_pipe);
1466             inpkts_to_sync_pipe = 0;
1467             report_new_capture_file(capture_opts->save_file);
1468           } else {
1469             /* File switch failed: stop here */
1470             ld.go = FALSE;
1471             continue;
1472           }
1473         } else {
1474           /* single file, stop now */
1475           ld.go = FALSE;
1476           continue;
1477         }
1478       } /* cnd_file_duration */
1479     }
1480
1481   } /* while (ld.go) */
1482
1483   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopping ...");
1484
1485   /* delete stop conditions */
1486   if (cnd_file_duration != NULL)
1487     cnd_delete(cnd_file_duration);
1488   if (cnd_autostop_files != NULL)
1489     cnd_delete(cnd_autostop_files);
1490   if (cnd_autostop_size != NULL)
1491     cnd_delete(cnd_autostop_size);
1492   if (cnd_autostop_duration != NULL)
1493     cnd_delete(cnd_autostop_duration);
1494
1495   /* did we had a pcap (input) error? */
1496   if (ld.pcap_err) {
1497     g_snprintf(errmsg, sizeof(errmsg), "Error while capturing packets: %s",
1498       pcap_geterr(ld.pcap_h));
1499     report_capture_error(errmsg, please_report);
1500   }
1501     else if (ld.from_cap_pipe && ld.cap_pipe_err == PIPERR)
1502       report_capture_error(errmsg, "");
1503
1504   /* did we had an error while capturing? */
1505   if (ld.err == 0) {
1506     write_ok = TRUE;
1507   } else {
1508     capture_loop_get_errmsg(errmsg, sizeof(errmsg), capture_opts->save_file, ld.err,
1509                               FALSE);
1510     report_capture_error(errmsg, please_report);
1511     write_ok = FALSE;
1512   }
1513
1514   if (capture_opts->saving_to_file) {
1515     /* close the wiretap (output) file */
1516     close_ok = capture_loop_close_output(capture_opts, &ld, &err_close);
1517   } else
1518     close_ok = TRUE;
1519
1520   /* there might be packets not yet notified to the parent */
1521   /* (do this after closing the file, so all packets are already flushed) */
1522   if(inpkts_to_sync_pipe) {
1523     report_packet_count(inpkts_to_sync_pipe);
1524     inpkts_to_sync_pipe = 0;
1525   }
1526
1527   /* If we've displayed a message about a write error, there's no point
1528      in displaying another message about an error on close. */
1529   if (!close_ok && write_ok) {
1530     capture_loop_get_errmsg(errmsg, sizeof(errmsg), capture_opts->save_file, err_close,
1531                 TRUE);
1532     report_capture_error(errmsg, "");
1533   }
1534
1535   /*
1536    * XXX We exhibit different behaviour between normal mode and sync mode
1537    * when the pipe is stdin and not already at EOF.  If we're a child, the
1538    * parent's stdin isn't closed, so if the user starts another capture,
1539    * cap_pipe_open_live() will very likely not see the expected magic bytes and
1540    * will say "Unrecognized libpcap format".  On the other hand, in normal
1541    * mode, cap_pipe_open_live() will say "End of file on pipe during open".
1542    */
1543
1544   /* get packet drop statistics from pcap */
1545   if(ld.pcap_h != NULL) {
1546     g_assert(!ld.from_cap_pipe);
1547     /* Get the capture statistics, so we know how many packets were
1548        dropped. */
1549     if (pcap_stats(ld.pcap_h, stats) >= 0) {
1550       *stats_known = TRUE;
1551       /* Let the parent process know. */
1552       report_packet_drops(stats->ps_drop);
1553     } else {
1554       g_snprintf(errmsg, sizeof(errmsg),
1555                 "Can't get packet-drop statistics: %s",
1556                 pcap_geterr(ld.pcap_h));
1557       report_capture_error(errmsg, please_report);
1558     }
1559   }
1560
1561   /* close the input file (pcap or capture pipe) */
1562   capture_loop_close_input(&ld);
1563
1564   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopped!");
1565
1566   /* ok, if the write and the close were successful. */
1567   return write_ok && close_ok;
1568
1569 error:
1570   if (capture_opts->multi_files_on) {
1571     /* cleanup ringbuffer */
1572     ringbuf_error_cleanup();
1573   } else {
1574     /* We can't use the save file, and we have no FILE * for the stream
1575        to close in order to close it, so close the FD directly. */
1576     if(save_file_fd != -1) {
1577       eth_close(save_file_fd);
1578     }
1579
1580     /* We couldn't even start the capture, so get rid of the capture
1581        file. */
1582     if(capture_opts->save_file != NULL) {
1583       eth_unlink(capture_opts->save_file);
1584       g_free(capture_opts->save_file);
1585           capture_opts->save_file = NULL;
1586     }
1587   }
1588   capture_opts->save_file = NULL;
1589   if (cfilter_error)
1590     report_cfilter_error(capture_opts->cfilter, errmsg);
1591   else
1592     report_capture_error(errmsg, secondary_errmsg);
1593
1594   /* close the input file (pcap or cap_pipe) */
1595   capture_loop_close_input(&ld);
1596
1597   g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_INFO, "Capture loop stopped with error");
1598
1599   return FALSE;
1600 }
1601
1602
1603 void capture_loop_stop(void)
1604 {
1605 #ifdef HAVE_PCAP_BREAKLOOP
1606   if(ld.pcap_h != NULL)
1607     pcap_breakloop(ld.pcap_h);
1608 #endif
1609   ld.go = FALSE;
1610 }
1611
1612
1613 static void
1614 capture_loop_get_errmsg(char *errmsg, int errmsglen, const char *fname,
1615                           int err, gboolean is_close)
1616 {
1617   switch (err) {
1618
1619   case ENOSPC:
1620     g_snprintf(errmsg, errmsglen,
1621                 "Not all the packets could be written to the file"
1622                 " to which the capture was being saved\n"
1623                 "(\"%s\") because there is no space left on the file system\n"
1624                 "on which that file resides.",
1625                 fname);
1626     break;
1627
1628 #ifdef EDQUOT
1629   case EDQUOT:
1630     g_snprintf(errmsg, errmsglen,
1631                 "Not all the packets could be written to the file"
1632                 " to which the capture was being saved\n"
1633                 "(\"%s\") because you are too close to, or over,"
1634                 " your disk quota\n"
1635                 "on the file system on which that file resides.",
1636                 fname);
1637   break;
1638 #endif
1639
1640   case WTAP_ERR_CANT_CLOSE:
1641     g_snprintf(errmsg, errmsglen,
1642                 "The file to which the capture was being saved"
1643                 " couldn't be closed for some unknown reason.");
1644     break;
1645
1646   case WTAP_ERR_SHORT_WRITE:
1647     g_snprintf(errmsg, errmsglen,
1648                 "Not all the packets could be written to the file"
1649                 " to which the capture was being saved\n"
1650                 "(\"%s\").",
1651                 fname);
1652     break;
1653
1654   default:
1655     if (is_close) {
1656       g_snprintf(errmsg, errmsglen,
1657                 "The file to which the capture was being saved\n"
1658                 "(\"%s\") could not be closed: %s.",
1659                 fname, wtap_strerror(err));
1660     } else {
1661       g_snprintf(errmsg, errmsglen,
1662                 "An error occurred while writing to the file"
1663                 " to which the capture was being saved\n"
1664                 "(\"%s\"): %s.",
1665                 fname, wtap_strerror(err));
1666     }
1667     break;
1668   }
1669 }
1670
1671
1672 /* one packet was captured, process it */
1673 static void
1674 capture_loop_packet_cb(u_char *user, const struct pcap_pkthdr *phdr,
1675   const u_char *pd)
1676 {
1677   loop_data *ld = (loop_data *) user;
1678   int err;
1679
1680   /* if the user told us to stop after x packets, do we have enough? */
1681   ld->packet_count++;
1682   if ((ld->packet_max > 0) && (ld->packet_count >= ld->packet_max))
1683   {
1684      ld->go = FALSE;
1685   }
1686
1687   if (ld->pdh) {
1688     /* We're supposed to write the packet to a file; do so.
1689        If this fails, set "ld->go" to FALSE, to stop the capture, and set
1690        "ld->err" to the error. */
1691     if (!libpcap_write_packet(ld->pdh, phdr, pd, &ld->bytes_written, &err)) {
1692       ld->go = FALSE;
1693       ld->err = err;
1694     }
1695   }
1696 }
1697
1698 #endif /* HAVE_LIBPCAP */