Apply Partial RELRO
[platform/upstream/libpcap.git] / pcap-new.c
1 /*
2  * Copyright (c) 2002 - 2005 NetGroup, Politecnico di Torino (Italy)
3  * Copyright (c) 2005 - 2008 CACE Technologies, Davis (California)
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  *
10  * 1. Redistributions of source code must retain the above copyright
11  * notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  * notice, this list of conditions and the following disclaimer in the
14  * documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the Politecnico di Torino, CACE Technologies
16  * nor the names of its contributors may be used to endorse or promote
17  * products derived from this software without specific prior written
18  * permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
26  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31  *
32  */
33
34 #ifdef HAVE_CONFIG_H
35 #include <config.h>
36 #endif
37
38 #include "ftmacros.h"
39
40 /*
41  * sockutils.h may include <crtdbg.h> on Windows, and pcap-int.h will
42  * include portability.h, and portability.h, on Windows, expects that
43  * <crtdbg.h> has already been included, so include sockutils.h first.
44  */
45 #include "sockutils.h"
46 #include "pcap-int.h"   // for the details of the pcap_t structure
47 #include "pcap-rpcap.h"
48 #include "rpcap-protocol.h"
49 #include <errno.h>              // for the errno variable
50 #include <stdlib.h>             // for malloc(), free(), ...
51 #include <string.h>             // for strstr, etc
52
53 #ifndef _WIN32
54 #include <dirent.h>             // for readdir
55 #endif
56
57 /* String identifier to be used in the pcap_findalldevs_ex() */
58 #define PCAP_TEXT_SOURCE_FILE "File"
59 #define PCAP_TEXT_SOURCE_FILE_LEN (sizeof PCAP_TEXT_SOURCE_FILE - 1)
60 /* String identifier to be used in the pcap_findalldevs_ex() */
61 #define PCAP_TEXT_SOURCE_ADAPTER "Network adapter"
62 #define PCAP_TEXT_SOURCE_ADAPTER_LEN (sizeof "Network adapter" - 1)
63
64 /* String identifier to be used in the pcap_findalldevs_ex() */
65 #define PCAP_TEXT_SOURCE_ON_LOCAL_HOST "on local host"
66 #define PCAP_TEXT_SOURCE_ON_LOCAL_HOST_LEN (sizeof PCAP_TEXT_SOURCE_ON_LOCAL_HOST + 1)
67
68 /****************************************************
69  *                                                  *
70  * Function bodies                                  *
71  *                                                  *
72  ****************************************************/
73
74 int pcap_findalldevs_ex(const char *source, struct pcap_rmtauth *auth, pcap_if_t **alldevs, char *errbuf)
75 {
76         int type;
77         char name[PCAP_BUF_SIZE], path[PCAP_BUF_SIZE], filename[PCAP_BUF_SIZE];
78         size_t pathlen;
79         size_t stringlen;
80         pcap_t *fp;
81         char tmpstring[PCAP_BUF_SIZE + 1];              /* Needed to convert names and descriptions from 'old' syntax to the 'new' one */
82         pcap_if_t *lastdev;     /* Last device in the pcap_if_t list */
83         pcap_if_t *dev;         /* Device we're adding to the pcap_if_t list */
84
85         /* List starts out empty. */
86         (*alldevs) = NULL;
87         lastdev = NULL;
88
89         if (strlen(source) > PCAP_BUF_SIZE)
90         {
91                 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "The source string is too long. Cannot handle it correctly.");
92                 return -1;
93         }
94
95         /*
96          * Determine the type of the source (file, local, remote)
97          * There are some differences if pcap_findalldevs_ex() is called to list files and remote adapters.
98          * In the first case, the name of the directory we have to look into must be present (therefore
99          * the 'name' parameter of the pcap_parsesrcstr() is present).
100          * In the second case, the name of the adapter is not required (we need just the host). So, we have
101          * to use a first time this function to get the source type, and a second time to get the appropriate
102          * info, which depends on the source type.
103          */
104         if (pcap_parsesrcstr(source, &type, NULL, NULL, NULL, errbuf) == -1)
105                 return -1;
106
107         switch (type)
108         {
109         case PCAP_SRC_IFLOCAL:
110                 if (pcap_parsesrcstr(source, &type, NULL, NULL, NULL, errbuf) == -1)
111                         return -1;
112
113                 /* Initialize temporary string */
114                 tmpstring[PCAP_BUF_SIZE] = 0;
115
116                 /* The user wants to retrieve adapters from a local host */
117                 if (pcap_findalldevs(alldevs, errbuf) == -1)
118                         return -1;
119
120                 if (*alldevs == NULL)
121                 {
122                         pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE,
123                                 "No interfaces found! Make sure libpcap/Npcap is properly installed"
124                                 " on the local machine.");
125                         return -1;
126                 }
127
128                 /* Scan all the interfaces and modify name and description */
129                 /* This is a trick in order to avoid the re-implementation of the pcap_findalldevs here */
130                 dev = *alldevs;
131                 while (dev)
132                 {
133                         char *localdesc, *desc;
134
135                         /* Create the new device identifier */
136                         if (pcap_createsrcstr(tmpstring, PCAP_SRC_IFLOCAL, NULL, NULL, dev->name, errbuf) == -1)
137                                 return -1;
138
139                         /* Delete the old pointer */
140                         free(dev->name);
141
142                         /* Make a copy of the new device identifier */
143                         dev->name = strdup(tmpstring);
144                         if (dev->name == NULL)
145                         {
146                                 pcap_fmt_errmsg_for_errno(errbuf,
147                                     PCAP_ERRBUF_SIZE, errno,
148                                     "malloc() failed");
149                                 pcap_freealldevs(*alldevs);
150                                 return -1;
151                         }
152
153                         /*
154                          * Create the description.
155                          */
156                         if ((dev->description == NULL) || (dev->description[0] == 0))
157                                 localdesc = dev->name;
158                         else
159                                 localdesc = dev->description;
160                         if (pcap_asprintf(&desc, "%s '%s' %s",
161                             PCAP_TEXT_SOURCE_ADAPTER, localdesc,
162                             PCAP_TEXT_SOURCE_ON_LOCAL_HOST) == -1)
163                         {
164                                 pcap_fmt_errmsg_for_errno(errbuf,
165                                     PCAP_ERRBUF_SIZE, errno,
166                                     "malloc() failed");
167                                 pcap_freealldevs(*alldevs);
168                                 return -1;
169                         }
170
171                         /* Now overwrite the description */
172                         free(dev->description);
173                         dev->description = desc;
174
175                         dev = dev->next;
176                 }
177
178                 return 0;
179
180         case PCAP_SRC_FILE:
181         {
182 #ifdef _WIN32
183                 WIN32_FIND_DATA filedata;
184                 HANDLE filehandle;
185 #else
186                 struct dirent *filedata;
187                 DIR *unixdir;
188 #endif
189
190                 if (pcap_parsesrcstr(source, &type, NULL, NULL, name, errbuf) == -1)
191                         return -1;
192
193                 /* Check that the filename is correct */
194                 stringlen = strlen(name);
195
196                 /* The directory must end with '\' in Win32 and '/' in UNIX */
197 #ifdef _WIN32
198 #define ENDING_CHAR '\\'
199 #else
200 #define ENDING_CHAR '/'
201 #endif
202
203                 if (name[stringlen - 1] != ENDING_CHAR)
204                 {
205                         name[stringlen] = ENDING_CHAR;
206                         name[stringlen + 1] = 0;
207
208                         stringlen++;
209                 }
210
211                 /* Save the path for future reference */
212                 pcap_snprintf(path, sizeof(path), "%s", name);
213                 pathlen = strlen(path);
214
215 #ifdef _WIN32
216                 /* To perform directory listing, Win32 must have an 'asterisk' as ending char */
217                 if (name[stringlen - 1] != '*')
218                 {
219                         name[stringlen] = '*';
220                         name[stringlen + 1] = 0;
221                 }
222
223                 filehandle = FindFirstFile(name, &filedata);
224
225                 if (filehandle == INVALID_HANDLE_VALUE)
226                 {
227                         pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Error when listing files: does folder '%s' exist?", path);
228                         return -1;
229                 }
230
231 #else
232                 /* opening the folder */
233                 unixdir= opendir(path);
234
235                 /* get the first file into it */
236                 filedata= readdir(unixdir);
237
238                 if (filedata == NULL)
239                 {
240                         pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Error when listing files: does folder '%s' exist?", path);
241                         return -1;
242                 }
243 #endif
244
245                 /* Add all files we find to the list. */
246                 do
247                 {
248 #ifdef _WIN32
249                         /* Skip the file if the pathname won't fit in the buffer */
250                         if (pathlen + strlen(filedata.cFileName) >= sizeof(filename))
251                                 continue;
252                         pcap_snprintf(filename, sizeof(filename), "%s%s", path, filedata.cFileName);
253 #else
254                         if (pathlen + strlen(filedata->d_name) >= sizeof(filename))
255                                 continue;
256                         pcap_snprintf(filename, sizeof(filename), "%s%s", path, filedata->d_name);
257 #endif
258
259                         fp = pcap_open_offline(filename, errbuf);
260
261                         if (fp)
262                         {
263                                 /* allocate the main structure */
264                                 dev = (pcap_if_t *)malloc(sizeof(pcap_if_t));
265                                 if (dev == NULL)
266                                 {
267                                         pcap_fmt_errmsg_for_errno(errbuf,
268                                             PCAP_ERRBUF_SIZE, errno,
269                                             "malloc() failed");
270                                         pcap_freealldevs(*alldevs);
271                                         return -1;
272                                 }
273
274                                 /* Initialize the structure to 'zero' */
275                                 memset(dev, 0, sizeof(pcap_if_t));
276
277                                 /* Append it to the list. */
278                                 if (lastdev == NULL)
279                                 {
280                                         /*
281                                          * List is empty, so it's also
282                                          * the first device.
283                                          */
284                                         *alldevs = dev;
285                                 }
286                                 else
287                                 {
288                                         /*
289                                          * Append after the last device.
290                                          */
291                                         lastdev->next = dev;
292                                 }
293                                 /* It's now the last device. */
294                                 lastdev = dev;
295
296                                 /* Create the new source identifier */
297                                 if (pcap_createsrcstr(tmpstring, PCAP_SRC_FILE, NULL, NULL, filename, errbuf) == -1)
298                                 {
299                                         pcap_freealldevs(*alldevs);
300                                         return -1;
301                                 }
302
303                                 dev->name = strdup(tmpstring);
304                                 if (dev->name == NULL)
305                                 {
306                                         pcap_fmt_errmsg_for_errno(errbuf,
307                                             PCAP_ERRBUF_SIZE, errno,
308                                             "malloc() failed");
309                                         pcap_freealldevs(*alldevs);
310                                         return -1;
311                                 }
312
313                                 /*
314                                  * Create the description.
315                                  */
316                                 if (pcap_asprintf(&dev->description,
317                                     "%s '%s' %s", PCAP_TEXT_SOURCE_FILE,
318                                     filename, PCAP_TEXT_SOURCE_ON_LOCAL_HOST) == -1)
319                                 {
320                                         pcap_fmt_errmsg_for_errno(errbuf,
321                                             PCAP_ERRBUF_SIZE, errno,
322                                             "malloc() failed");
323                                         pcap_freealldevs(*alldevs);
324                                         return -1;
325                                 }
326
327                                 pcap_close(fp);
328                         }
329                 }
330 #ifdef _WIN32
331                 while (FindNextFile(filehandle, &filedata) != 0);
332 #else
333                 while ( (filedata= readdir(unixdir)) != NULL);
334 #endif
335
336
337 #ifdef _WIN32
338                 /* Close the search handle. */
339                 FindClose(filehandle);
340 #endif
341
342                 return 0;
343         }
344
345         case PCAP_SRC_IFREMOTE:
346                 return pcap_findalldevs_ex_remote(source, auth, alldevs, errbuf);
347
348         default:
349                 pcap_strlcpy(errbuf, "Source type not supported", PCAP_ERRBUF_SIZE);
350                 return -1;
351         }
352 }
353
354 pcap_t *pcap_open(const char *source, int snaplen, int flags, int read_timeout, struct pcap_rmtauth *auth, char *errbuf)
355 {
356         char name[PCAP_BUF_SIZE];
357         int type;
358         pcap_t *fp;
359         int status;
360
361         /*
362          * A null device name is equivalent to the "any" device -
363          * which might not be supported on this platform, but
364          * this means that you'll get a "not supported" error
365          * rather than, say, a crash when we try to dereference
366          * the null pointer.
367          */
368         if (source == NULL)
369                 source = "any";
370
371         if (strlen(source) > PCAP_BUF_SIZE)
372         {
373                 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "The source string is too long. Cannot handle it correctly.");
374                 return NULL;
375         }
376
377         /*
378          * Determine the type of the source (file, local, remote) and,
379          * if it's file or local, the name of the file or capture device.
380          */
381         if (pcap_parsesrcstr(source, &type, NULL, NULL, name, errbuf) == -1)
382                 return NULL;
383
384         switch (type)
385         {
386         case PCAP_SRC_FILE:
387                 return pcap_open_offline(name, errbuf);
388
389         case PCAP_SRC_IFLOCAL:
390                 fp = pcap_create(name, errbuf);
391                 break;
392
393         case PCAP_SRC_IFREMOTE:
394                 /*
395                  * Although we already have host, port and iface, we prefer
396                  * to pass only 'source' to pcap_open_rpcap(), so that it
397                  * has to call pcap_parsesrcstr() again.
398                  * This is less optimized, but much clearer.
399                  */
400                 return pcap_open_rpcap(source, snaplen, flags, read_timeout, auth, errbuf);
401
402         default:
403                 pcap_strlcpy(errbuf, "Source type not supported", PCAP_ERRBUF_SIZE);
404                 return NULL;
405         }
406
407         if (fp == NULL)
408                 return (NULL);
409         status = pcap_set_snaplen(fp, snaplen);
410         if (status < 0)
411                 goto fail;
412         if (flags & PCAP_OPENFLAG_PROMISCUOUS)
413         {
414                 status = pcap_set_promisc(fp, 1);
415                 if (status < 0)
416                         goto fail;
417         }
418         if (flags & PCAP_OPENFLAG_MAX_RESPONSIVENESS)
419         {
420                 status = pcap_set_immediate_mode(fp, 1);
421                 if (status < 0)
422                         goto fail;
423         }
424 #ifdef _WIN32
425         /*
426          * This flag is supported on Windows only.
427          * XXX - is there a way to support it with
428          * the capture mechanisms on UN*X?  It's not
429          * exactly a "set direction" operation; I
430          * think it means "do not capture packets
431          * injected with pcap_sendpacket() or
432          * pcap_inject()".
433          */
434         /* disable loopback capture if requested */
435         if (flags & PCAP_OPENFLAG_NOCAPTURE_LOCAL)
436                 fp->opt.nocapture_local = 1;
437 #endif /* _WIN32 */
438         status = pcap_set_timeout(fp, read_timeout);
439         if (status < 0)
440                 goto fail;
441         status = pcap_activate(fp);
442         if (status < 0)
443                 goto fail;
444         return fp;
445
446 fail:
447         if (status == PCAP_ERROR)
448                 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "%s: %s",
449                     name, fp->errbuf);
450         else if (status == PCAP_ERROR_NO_SUCH_DEVICE ||
451             status == PCAP_ERROR_PERM_DENIED ||
452             status == PCAP_ERROR_PROMISC_PERM_DENIED)
453                 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "%s: %s (%s)",
454                     name, pcap_statustostr(status), fp->errbuf);
455         else
456                 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "%s: %s",
457                     name, pcap_statustostr(status));
458         pcap_close(fp);
459         return NULL;
460 }
461
462 struct pcap_samp *pcap_setsampling(pcap_t *p)
463 {
464         return &p->rmt_samp;
465 }