Home | History | Annotate | Download | only in libpcap
      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 /*
     39  * sockutils.h may include <crtdbg.h> on Windows, and pcap-int.h will
     40  * include portability.h, and portability.h, on Windows, expects that
     41  * <crtdbg.h> has already been included, so include sockutils.h first.
     42  */
     43 #include "sockutils.h"
     44 #include "pcap-int.h"	// for the details of the pcap_t structure
     45 #include "pcap-rpcap.h"
     46 #include "rpcap-protocol.h"
     47 #include <errno.h>		// for the errno variable
     48 #include <stdlib.h>		// for malloc(), free(), ...
     49 #include <string.h>		// for strstr, etc
     50 
     51 #ifndef _WIN32
     52 #include <dirent.h>		// for readdir
     53 #endif
     54 
     55 /* String identifier to be used in the pcap_findalldevs_ex() */
     56 #define PCAP_TEXT_SOURCE_FILE "File"
     57 /* String identifier to be used in the pcap_findalldevs_ex() */
     58 #define PCAP_TEXT_SOURCE_ADAPTER "Network adapter"
     59 
     60 /* String identifier to be used in the pcap_findalldevs_ex() */
     61 #define PCAP_TEXT_SOURCE_ON_LOCAL_HOST "on local host"
     62 
     63 /****************************************************
     64  *                                                  *
     65  * Function bodies                                  *
     66  *                                                  *
     67  ****************************************************/
     68 
     69 int pcap_findalldevs_ex(char *source, struct pcap_rmtauth *auth, pcap_if_t **alldevs, char *errbuf)
     70 {
     71 	int type;
     72 	char name[PCAP_BUF_SIZE], path[PCAP_BUF_SIZE], filename[PCAP_BUF_SIZE];
     73 	pcap_t *fp;
     74 	char tmpstring[PCAP_BUF_SIZE + 1];		/* Needed to convert names and descriptions from 'old' syntax to the 'new' one */
     75 	pcap_if_t *lastdev;	/* Last device in the pcap_if_t list */
     76 	pcap_if_t *dev;		/* Device we're adding to the pcap_if_t list */
     77 
     78 	/* List starts out empty. */
     79 	(*alldevs) = NULL;
     80 	lastdev = NULL;
     81 
     82 	if (strlen(source) > PCAP_BUF_SIZE)
     83 	{
     84 		pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "The source string is too long. Cannot handle it correctly.");
     85 		return -1;
     86 	}
     87 
     88 	/*
     89 	 * Determine the type of the source (file, local, remote)
     90 	 * There are some differences if pcap_findalldevs_ex() is called to list files and remote adapters.
     91 	 * In the first case, the name of the directory we have to look into must be present (therefore
     92 	 * the 'name' parameter of the pcap_parsesrcstr() is present).
     93 	 * In the second case, the name of the adapter is not required (we need just the host). So, we have
     94 	 * to use a first time this function to get the source type, and a second time to get the appropriate
     95 	 * info, which depends on the source type.
     96 	 */
     97 	if (pcap_parsesrcstr(source, &type, NULL, NULL, NULL, errbuf) == -1)
     98 		return -1;
     99 
    100 	switch (type)
    101 	{
    102 	case PCAP_SRC_IFLOCAL:
    103 		if (pcap_parsesrcstr(source, &type, NULL, NULL, NULL, errbuf) == -1)
    104 			return -1;
    105 
    106 		/* Initialize temporary string */
    107 		tmpstring[PCAP_BUF_SIZE] = 0;
    108 
    109 		/* The user wants to retrieve adapters from a local host */
    110 		if (pcap_findalldevs(alldevs, errbuf) == -1)
    111 			return -1;
    112 
    113 		if (*alldevs == NULL)
    114 		{
    115 			pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE,
    116 				"No interfaces found! Make sure libpcap/WinPcap is properly installed"
    117 				" on the local machine.");
    118 			return -1;
    119 		}
    120 
    121 		/* Scan all the interfaces and modify name and description */
    122 		/* This is a trick in order to avoid the re-implementation of the pcap_findalldevs here */
    123 		dev = *alldevs;
    124 		while (dev)
    125 		{
    126 			/* Create the new device identifier */
    127 			if (pcap_createsrcstr(tmpstring, PCAP_SRC_IFLOCAL, NULL, NULL, dev->name, errbuf) == -1)
    128 				return -1;
    129 
    130 			/* Delete the old pointer */
    131 			free(dev->name);
    132 
    133 			/* Make a copy of the new device identifier */
    134 			dev->name = strdup(tmpstring);
    135 			if (dev->name == NULL)
    136 			{
    137 				pcap_fmt_errmsg_for_errno(errbuf,
    138 				    PCAP_ERRBUF_SIZE, errno,
    139 				    "malloc() failed");
    140 				pcap_freealldevs(*alldevs);
    141 				return -1;
    142 			}
    143 
    144 			/* Create the new device description */
    145 			if ((dev->description == NULL) || (dev->description[0] == 0))
    146 				pcap_snprintf(tmpstring, sizeof(tmpstring) - 1, "%s '%s' %s", PCAP_TEXT_SOURCE_ADAPTER,
    147 				dev->name, PCAP_TEXT_SOURCE_ON_LOCAL_HOST);
    148 			else
    149 				pcap_snprintf(tmpstring, sizeof(tmpstring) - 1, "%s '%s' %s", PCAP_TEXT_SOURCE_ADAPTER,
    150 				dev->description, PCAP_TEXT_SOURCE_ON_LOCAL_HOST);
    151 
    152 			/* Delete the old pointer */
    153 			free(dev->description);
    154 
    155 			/* Make a copy of the description */
    156 			dev->description = strdup(tmpstring);
    157 			if (dev->description == NULL)
    158 			{
    159 				pcap_fmt_errmsg_for_errno(errbuf,
    160 				    PCAP_ERRBUF_SIZE, errno,
    161 				    "malloc() failed");
    162 				pcap_freealldevs(*alldevs);
    163 				return -1;
    164 			}
    165 
    166 			dev = dev->next;
    167 		}
    168 
    169 		return 0;
    170 
    171 	case PCAP_SRC_FILE:
    172 	{
    173 		size_t stringlen;
    174 #ifdef _WIN32
    175 		WIN32_FIND_DATA filedata;
    176 		HANDLE filehandle;
    177 #else
    178 		struct dirent *filedata;
    179 		DIR *unixdir;
    180 #endif
    181 
    182 		if (pcap_parsesrcstr(source, &type, NULL, NULL, name, errbuf) == -1)
    183 			return -1;
    184 
    185 		/* Check that the filename is correct */
    186 		stringlen = strlen(name);
    187 
    188 		/* The directory must end with '\' in Win32 and '/' in UNIX */
    189 #ifdef _WIN32
    190 #define ENDING_CHAR '\\'
    191 #else
    192 #define ENDING_CHAR '/'
    193 #endif
    194 
    195 		if (name[stringlen - 1] != ENDING_CHAR)
    196 		{
    197 			name[stringlen] = ENDING_CHAR;
    198 			name[stringlen + 1] = 0;
    199 
    200 			stringlen++;
    201 		}
    202 
    203 		/* Save the path for future reference */
    204 		pcap_snprintf(path, sizeof(path), "%s", name);
    205 
    206 #ifdef _WIN32
    207 		/* To perform directory listing, Win32 must have an 'asterisk' as ending char */
    208 		if (name[stringlen - 1] != '*')
    209 		{
    210 			name[stringlen] = '*';
    211 			name[stringlen + 1] = 0;
    212 		}
    213 
    214 		filehandle = FindFirstFile(name, &filedata);
    215 
    216 		if (filehandle == INVALID_HANDLE_VALUE)
    217 		{
    218 			pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Error when listing files: does folder '%s' exist?", path);
    219 			return -1;
    220 		}
    221 
    222 #else
    223 		/* opening the folder */
    224 		unixdir= opendir(path);
    225 
    226 		/* get the first file into it */
    227 		filedata= readdir(unixdir);
    228 
    229 		if (filedata == NULL)
    230 		{
    231 			pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Error when listing files: does folder '%s' exist?", path);
    232 			return -1;
    233 		}
    234 #endif
    235 
    236 		/* Add all files we find to the list. */
    237 		do
    238 		{
    239 
    240 #ifdef _WIN32
    241 			pcap_snprintf(filename, sizeof(filename), "%s%s", path, filedata.cFileName);
    242 #else
    243 			pcap_snprintf(filename, sizeof(filename), "%s%s", path, filedata->d_name);
    244 #endif
    245 
    246 			fp = pcap_open_offline(filename, errbuf);
    247 
    248 			if (fp)
    249 			{
    250 				/* allocate the main structure */
    251 				dev = (pcap_if_t *)malloc(sizeof(pcap_if_t));
    252 				if (dev == NULL)
    253 				{
    254 					pcap_fmt_errmsg_for_errno(errbuf,
    255 					    PCAP_ERRBUF_SIZE, errno,
    256 					    "malloc() failed");
    257 					pcap_freealldevs(*alldevs);
    258 					return -1;
    259 				}
    260 
    261 				/* Initialize the structure to 'zero' */
    262 				memset(dev, 0, sizeof(pcap_if_t));
    263 
    264 				/* Append it to the list. */
    265 				if (lastdev == NULL)
    266 				{
    267 					/*
    268 					 * List is empty, so it's also
    269 					 * the first device.
    270 					 */
    271 					*alldevs = dev;
    272 				}
    273 				else
    274 				{
    275 					/*
    276 					 * Append after the last device.
    277 					 */
    278 					lastdev->next = dev;
    279 				}
    280 				/* It's now the last device. */
    281 				lastdev = dev;
    282 
    283 				/* Create the new source identifier */
    284 				if (pcap_createsrcstr(tmpstring, PCAP_SRC_FILE, NULL, NULL, filename, errbuf) == -1)
    285 				{
    286 					pcap_freealldevs(*alldevs);
    287 					return -1;
    288 				}
    289 
    290 				stringlen = strlen(tmpstring);
    291 
    292 				dev->name = (char *)malloc(stringlen + 1);
    293 				if (dev->name == NULL)
    294 				{
    295 					pcap_fmt_errmsg_for_errno(errbuf,
    296 					    PCAP_ERRBUF_SIZE, errno,
    297 					    "malloc() failed");
    298 					pcap_freealldevs(*alldevs);
    299 					return -1;
    300 				}
    301 
    302 				strlcpy(dev->name, tmpstring, stringlen);
    303 
    304 				dev->name[stringlen] = 0;
    305 
    306 				/* Create the description */
    307 				pcap_snprintf(tmpstring, sizeof(tmpstring) - 1, "%s '%s' %s", PCAP_TEXT_SOURCE_FILE,
    308 					filename, PCAP_TEXT_SOURCE_ON_LOCAL_HOST);
    309 
    310 				stringlen = strlen(tmpstring);
    311 
    312 				dev->description = (char *)malloc(stringlen + 1);
    313 
    314 				if (dev->description == NULL)
    315 				{
    316 					pcap_fmt_errmsg_for_errno(errbuf,
    317 					    PCAP_ERRBUF_SIZE, errno,
    318 					    "malloc() failed");
    319 					pcap_freealldevs(*alldevs);
    320 					return -1;
    321 				}
    322 
    323 				/* Copy the new device description into the correct memory location */
    324 				strlcpy(dev->description, tmpstring, stringlen + 1);
    325 
    326 				pcap_close(fp);
    327 			}
    328 		}
    329 #ifdef _WIN32
    330 		while (FindNextFile(filehandle, &filedata) != 0);
    331 #else
    332 		while ( (filedata= readdir(unixdir)) != NULL);
    333 #endif
    334 
    335 
    336 #ifdef _WIN32
    337 		/* Close the search handle. */
    338 		FindClose(filehandle);
    339 #endif
    340 
    341 		return 0;
    342 	}
    343 
    344 	case PCAP_SRC_IFREMOTE:
    345 		return pcap_findalldevs_ex_remote(source, auth, alldevs, errbuf);
    346 
    347 	default:
    348 		strlcpy(errbuf, "Source type not supported", PCAP_ERRBUF_SIZE);
    349 		return -1;
    350 	}
    351 }
    352 
    353 pcap_t *pcap_open(const char *source, int snaplen, int flags, int read_timeout, struct pcap_rmtauth *auth, char *errbuf)
    354 {
    355 	char name[PCAP_BUF_SIZE];
    356 	int type;
    357 	pcap_t *fp;
    358 	int status;
    359 
    360 	if (strlen(source) > PCAP_BUF_SIZE)
    361 	{
    362 		pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "The source string is too long. Cannot handle it correctly.");
    363 		return NULL;
    364 	}
    365 
    366 	/*
    367 	 * Determine the type of the source (file, local, remote) and,
    368 	 * if it's file or local, the name of the file or capture device.
    369 	 */
    370 	if (pcap_parsesrcstr(source, &type, NULL, NULL, name, errbuf) == -1)
    371 		return NULL;
    372 
    373 	switch (type)
    374 	{
    375 	case PCAP_SRC_FILE:
    376 		return pcap_open_offline(name, errbuf);
    377 
    378 	case PCAP_SRC_IFLOCAL:
    379 		fp = pcap_create(name, errbuf);
    380 		break;
    381 
    382 	case PCAP_SRC_IFREMOTE:
    383 		/*
    384 		 * Although we already have host, port and iface, we prefer
    385 		 * to pass only 'source' to pcap_open_rpcap(), so that it
    386 		 * has to call pcap_parsesrcstr() again.
    387 		 * This is less optimized, but much clearer.
    388 		 */
    389 		return pcap_open_rpcap(source, snaplen, flags, read_timeout, auth, errbuf);
    390 
    391 	default:
    392 		strlcpy(errbuf, "Source type not supported", PCAP_ERRBUF_SIZE);
    393 		return NULL;
    394 	}
    395 
    396 	if (fp == NULL)
    397 		return (NULL);
    398 	status = pcap_set_snaplen(fp, snaplen);
    399 	if (status < 0)
    400 		goto fail;
    401 	if (flags & PCAP_OPENFLAG_PROMISCUOUS)
    402 	{
    403 		status = pcap_set_promisc(fp, 1);
    404 		if (status < 0)
    405 			goto fail;
    406 	}
    407 	if (flags & PCAP_OPENFLAG_MAX_RESPONSIVENESS)
    408 	{
    409 		status = pcap_set_immediate_mode(fp, 1);
    410 		if (status < 0)
    411 			goto fail;
    412 	}
    413 #ifdef _WIN32
    414 	/*
    415 	 * This flag is supported on Windows only.
    416 	 * XXX - is there a way to support it with
    417 	 * the capture mechanisms on UN*X?  It's not
    418 	 * exactly a "set direction" operation; I
    419 	 * think it means "do not capture packets
    420 	 * injected with pcap_sendpacket() or
    421 	 * pcap_inject()".
    422 	 */
    423 	/* disable loopback capture if requested */
    424 	if (flags & PCAP_OPENFLAG_NOCAPTURE_LOCAL)
    425 		fp->opt.nocapture_local = 1;
    426 #endif /* _WIN32 */
    427 	status = pcap_set_timeout(fp, read_timeout);
    428 	if (status < 0)
    429 		goto fail;
    430 	status = pcap_activate(fp);
    431 	if (status < 0)
    432 		goto fail;
    433 	return fp;
    434 
    435 fail:
    436 	if (status == PCAP_ERROR)
    437 		pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "%s: %s",
    438 		    name, fp->errbuf);
    439 	else if (status == PCAP_ERROR_NO_SUCH_DEVICE ||
    440 	    status == PCAP_ERROR_PERM_DENIED ||
    441 	    status == PCAP_ERROR_PROMISC_PERM_DENIED)
    442 		pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "%s: %s (%s)",
    443 		    name, pcap_statustostr(status), fp->errbuf);
    444 	else
    445 		pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "%s: %s",
    446 		    name, pcap_statustostr(status));
    447 	pcap_close(fp);
    448 	return NULL;
    449 }
    450 
    451 struct pcap_samp *pcap_setsampling(pcap_t *p)
    452 {
    453 	return &p->rmt_samp;
    454 }
    455