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 #include <string.h> /* for strlen(), ... */ 41 #include <stdlib.h> /* for malloc(), free(), ... */ 42 #include <stdarg.h> /* for functions with variable number of arguments */ 43 #include <errno.h> /* for the errno variable */ 44 #include "sockutils.h" 45 #include "pcap-int.h" 46 #include "rpcap-protocol.h" 47 #include "pcap-rpcap.h" 48 49 /* 50 * This file contains the pcap module for capturing from a remote machine's 51 * interfaces using the RPCAP protocol. 52 * 53 * WARNING: All the RPCAP functions that are allowed to return a buffer 54 * containing the error description can return max PCAP_ERRBUF_SIZE characters. 55 * However there is no guarantees that the string will be zero-terminated. 56 * Best practice is to define the errbuf variable as a char of size 57 * 'PCAP_ERRBUF_SIZE+1' and to insert manually a NULL character at the end 58 * of the buffer. This will guarantee that no buffer overflows occur even 59 * if we use the printf() to show the error on the screen. 60 * 61 * XXX - actually, null-terminating the error string is part of the 62 * contract for the pcap API; if there's any place in the pcap code 63 * that doesn't guarantee null-termination, even at the expense of 64 * cutting the message short, that's a bug and needs to be fixed. 65 */ 66 67 #define PCAP_STATS_STANDARD 0 /* Used by pcap_stats_rpcap to see if we want standard or extended statistics */ 68 #ifdef _WIN32 69 #define PCAP_STATS_EX 1 /* Used by pcap_stats_rpcap to see if we want standard or extended statistics */ 70 #endif 71 72 /* 73 * \brief Keeps a list of all the opened connections in the active mode. 74 * 75 * This structure defines a linked list of items that are needed to keep the info required to 76 * manage the active mode. 77 * In other words, when a new connection in active mode starts, this structure is updated so that 78 * it reflects the list of active mode connections currently opened. 79 * This structure is required by findalldevs() and open_remote() to see if they have to open a new 80 * control connection toward the host, or they already have a control connection in place. 81 */ 82 struct activehosts 83 { 84 struct sockaddr_storage host; 85 SOCKET sockctrl; 86 uint8 protocol_version; 87 struct activehosts *next; 88 }; 89 90 /* Keeps a list of all the opened connections in the active mode. */ 91 static struct activehosts *activeHosts; 92 93 /* 94 * Keeps the main socket identifier when we want to accept a new remote 95 * connection (active mode only). 96 * See the documentation of pcap_remoteact_accept() and 97 * pcap_remoteact_cleanup() for more details. 98 */ 99 static SOCKET sockmain; 100 101 /* 102 * Private data for capturing remotely using the rpcap protocol. 103 */ 104 struct pcap_rpcap { 105 /* 106 * This is '1' if we're the network client; it is needed by several 107 * functions (such as pcap_setfilter()) to know whether they have 108 * to use the socket or have to open the local adapter. 109 */ 110 int rmt_clientside; 111 112 SOCKET rmt_sockctrl; /* socket ID of the socket used for the control connection */ 113 SOCKET rmt_sockdata; /* socket ID of the socket used for the data connection */ 114 int rmt_flags; /* we have to save flags, since they are passed by the pcap_open_live(), but they are used by the pcap_startcapture() */ 115 int rmt_capstarted; /* 'true' if the capture is already started (needed to knoe if we have to call the pcap_startcapture() */ 116 char *currentfilter; /* Pointer to a buffer (allocated at run-time) that stores the current filter. Needed when flag PCAP_OPENFLAG_NOCAPTURE_RPCAP is turned on. */ 117 118 uint8 protocol_version; /* negotiated protocol version */ 119 120 unsigned int TotNetDrops; /* keeps the number of packets that have been dropped by the network */ 121 122 /* 123 * This keeps the number of packets that have been received by the 124 * application. 125 * 126 * Packets dropped by the kernel buffer are not counted in this 127 * variable. It is always equal to (TotAccepted - TotDrops), 128 * except for the case of remote capture, in which we have also 129 * packets in flight, i.e. that have been transmitted by the remote 130 * host, but that have not been received (yet) from the client. 131 * In this case, (TotAccepted - TotDrops - TotNetDrops) gives a 132 * wrong result, since this number does not corresponds always to 133 * the number of packet received by the application. For this reason, 134 * in the remote capture we need another variable that takes into 135 * account of the number of packets actually received by the 136 * application. 137 */ 138 unsigned int TotCapt; 139 140 struct pcap_stat stat; 141 /* XXX */ 142 struct pcap *next; /* list of open pcaps that need stuff cleared on close */ 143 }; 144 145 /**************************************************** 146 * * 147 * Locally defined functions * 148 * * 149 ****************************************************/ 150 static struct pcap_stat *rpcap_stats_rpcap(pcap_t *p, struct pcap_stat *ps, int mode); 151 static int pcap_pack_bpffilter(pcap_t *fp, char *sendbuf, int *sendbufidx, struct bpf_program *prog); 152 static int pcap_createfilter_norpcappkt(pcap_t *fp, struct bpf_program *prog); 153 static int pcap_updatefilter_remote(pcap_t *fp, struct bpf_program *prog); 154 static void pcap_save_current_filter_rpcap(pcap_t *fp, const char *filter); 155 static int pcap_setfilter_rpcap(pcap_t *fp, struct bpf_program *prog); 156 static int pcap_setsampling_remote(pcap_t *fp); 157 static int pcap_startcapture_remote(pcap_t *fp); 158 static int rpcap_recv_msg_header(SOCKET sock, struct rpcap_header *header, char *errbuf); 159 static int rpcap_check_msg_ver(SOCKET sock, uint8 expected_ver, struct rpcap_header *header, char *errbuf); 160 static int rpcap_check_msg_type(SOCKET sock, uint8 request_type, struct rpcap_header *header, uint16 *errcode, char *errbuf); 161 static int rpcap_process_msg_header(SOCKET sock, uint8 ver, uint8 request_type, struct rpcap_header *header, char *errbuf); 162 static int rpcap_recv(SOCKET sock, void *buffer, size_t toread, uint32 *plen, char *errbuf); 163 static void rpcap_msg_err(SOCKET sockctrl, uint32 plen, char *remote_errbuf); 164 static int rpcap_discard(SOCKET sock, uint32 len, char *errbuf); 165 static int rpcap_read_packet_msg(SOCKET sock, pcap_t *p, size_t size); 166 167 /**************************************************** 168 * * 169 * Function bodies * 170 * * 171 ****************************************************/ 172 173 /* 174 * This function translates (i.e. de-serializes) a 'rpcap_sockaddr' 175 * structure from the network byte order to a 'sockaddr_in" or 176 * 'sockaddr_in6' structure in the host byte order. 177 * 178 * It accepts an 'rpcap_sockaddr' structure as it is received from the 179 * network, and checks the address family field against various values 180 * to see whether it looks like an IPv4 address, an IPv6 address, or 181 * neither of those. It checks for multiple values in order to try 182 * to handle older rpcap daemons that sent the native OS's 'sockaddr_in' 183 * or 'sockaddr_in6' structures over the wire with some members 184 * byte-swapped, and to handle the fact that AF_INET6 has different 185 * values on different OSes. 186 * 187 * For IPv4 addresses, it converts the address family to host byte 188 * order from network byte order and puts it into the structure, 189 * sets the length if a sockaddr structure has a length, converts the 190 * port number to host byte order from network byte order and puts 191 * it into the structure, copies over the IPv4 address, and zeroes 192 * out the zero padding. 193 * 194 * For IPv6 addresses, it converts the address family to host byte 195 * order from network byte order and puts it into the structure, 196 * sets the length if a sockaddr structure has a length, converts the 197 * port number and flow information to host byte order from network 198 * byte order and puts them into the structure, copies over the IPv6 199 * address, and converts the scope ID to host byte order from network 200 * byte order and puts it into the structure. 201 * 202 * The function will allocate the 'sockaddrout' variable according to the 203 * address family in use. In case the address does not belong to the 204 * AF_INET nor AF_INET6 families, 'sockaddrout' is not allocated and a 205 * NULL pointer is returned. This usually happens because that address 206 * does not exist on the other host, or is of an address family other 207 * than AF_INET or AF_INET6, so the RPCAP daemon sent a 'sockaddr_storage' 208 * structure containing all 'zero' values. 209 * 210 * Older RPCAPDs sent the addresses over the wire in the OS's native 211 * structure format. For most OSes, this looks like the over-the-wire 212 * format, but might have a different value for AF_INET6 than the value 213 * on the machine receiving the reply. For OSes with the newer BSD-style 214 * sockaddr structures, this has, instead of a 2-byte address family, 215 * a 1-byte structure length followed by a 1-byte address family. The 216 * RPCAPD code would put the address family in network byte order before 217 * sending it; that would set it to 0 on a little-endian machine, as 218 * htons() of any value between 1 and 255 would result in a value > 255, 219 * with its lower 8 bits zero, so putting that back into a 1-byte field 220 * would set it to 0. 221 * 222 * Therefore, for older RPCAPDs running on an OS with newer BSD-style 223 * sockaddr structures, the family field, if treated as a big-endian 224 * (network byte order) 16-bit field, would be: 225 * 226 * (length << 8) | family if sent by a big-endian machine 227 * (length << 8) if sent by a little-endian machine 228 * 229 * For current RPCAPDs, and for older RPCAPDs running on an OS with 230 * older BSD-style sockaddr structures, the family field, if treated 231 * as a big-endian 16-bit field, would just contain the family. 232 * 233 * \param sockaddrin: a 'rpcap_sockaddr' pointer to the variable that has 234 * to be de-serialized. 235 * 236 * \param sockaddrout: a 'sockaddr_storage' pointer to the variable that will contain 237 * the de-serialized data. The structure returned can be either a 'sockaddr_in' or 'sockaddr_in6'. 238 * This variable will be allocated automatically inside this function. 239 * 240 * \param errbuf: a pointer to a user-allocated buffer (of size PCAP_ERRBUF_SIZE) 241 * that will contain the error message (in case there is one). 242 * 243 * \return '0' if everything is fine, '-1' if some errors occurred. Basically, the error 244 * can be only the fact that the malloc() failed to allocate memory. 245 * The error message is returned in the 'errbuf' variable, while the deserialized address 246 * is returned into the 'sockaddrout' variable. 247 * 248 * \warning This function supports only AF_INET and AF_INET6 address families. 249 * 250 * \warning The sockaddrout (if not NULL) must be deallocated by the user. 251 */ 252 253 /* 254 * Possible IPv4 family values other than the designated over-the-wire value, 255 * which is 2 (because everybody uses 2 for AF_INET4). 256 */ 257 #define SOCKADDR_IN_LEN 16 /* length of struct sockaddr_in */ 258 #define SOCKADDR_IN6_LEN 28 /* length of struct sockaddr_in6 */ 259 #define NEW_BSD_AF_INET_BE ((SOCKADDR_IN_LEN << 8) | 2) 260 #define NEW_BSD_AF_INET_LE (SOCKADDR_IN_LEN << 8) 261 262 /* 263 * Possible IPv6 family values other than the designated over-the-wire value, 264 * which is 23 (because that's what Windows uses, and most RPCAP servers 265 * out there are probably running Windows, as WinPcap includes the server 266 * but few if any UN*Xes build and ship it). 267 * 268 * The new BSD sockaddr structure format was in place before 4.4-Lite, so 269 * all the free-software BSDs use it. 270 */ 271 #define NEW_BSD_AF_INET6_BSD_BE ((SOCKADDR_IN6_LEN << 8) | 24) /* NetBSD, OpenBSD, BSD/OS */ 272 #define NEW_BSD_AF_INET6_FREEBSD_BE ((SOCKADDR_IN6_LEN << 8) | 28) /* FreeBSD, DragonFly BSD */ 273 #define NEW_BSD_AF_INET6_DARWIN_BE ((SOCKADDR_IN6_LEN << 8) | 30) /* macOS, iOS, anything else Darwin-based */ 274 #define NEW_BSD_AF_INET6_LE (SOCKADDR_IN6_LEN << 8) 275 #define LINUX_AF_INET6 10 276 #define HPUX_AF_INET6 22 277 #define AIX_AF_INET6 24 278 #define SOLARIS_AF_INET6 26 279 280 static int 281 rpcap_deseraddr(struct rpcap_sockaddr *sockaddrin, struct sockaddr_storage **sockaddrout, char *errbuf) 282 { 283 /* Warning: we support only AF_INET and AF_INET6 */ 284 switch (ntohs(sockaddrin->family)) 285 { 286 case RPCAP_AF_INET: 287 case NEW_BSD_AF_INET_BE: 288 case NEW_BSD_AF_INET_LE: 289 { 290 struct rpcap_sockaddr_in *sockaddrin_ipv4; 291 struct sockaddr_in *sockaddrout_ipv4; 292 293 (*sockaddrout) = (struct sockaddr_storage *) malloc(sizeof(struct sockaddr_in)); 294 if ((*sockaddrout) == NULL) 295 { 296 pcap_fmt_errmsg_for_errno(errbuf, PCAP_ERRBUF_SIZE, 297 errno, "malloc() failed"); 298 return -1; 299 } 300 sockaddrin_ipv4 = (struct rpcap_sockaddr_in *) sockaddrin; 301 sockaddrout_ipv4 = (struct sockaddr_in *) (*sockaddrout); 302 sockaddrout_ipv4->sin_family = AF_INET; 303 sockaddrout_ipv4->sin_port = ntohs(sockaddrin_ipv4->port); 304 memcpy(&sockaddrout_ipv4->sin_addr, &sockaddrin_ipv4->addr, sizeof(sockaddrout_ipv4->sin_addr)); 305 memset(sockaddrout_ipv4->sin_zero, 0, sizeof(sockaddrout_ipv4->sin_zero)); 306 break; 307 } 308 309 #ifdef AF_INET6 310 case RPCAP_AF_INET6: 311 case NEW_BSD_AF_INET6_BSD_BE: 312 case NEW_BSD_AF_INET6_FREEBSD_BE: 313 case NEW_BSD_AF_INET6_DARWIN_BE: 314 case NEW_BSD_AF_INET6_LE: 315 case LINUX_AF_INET6: 316 case HPUX_AF_INET6: 317 case AIX_AF_INET6: 318 case SOLARIS_AF_INET6: 319 { 320 struct rpcap_sockaddr_in6 *sockaddrin_ipv6; 321 struct sockaddr_in6 *sockaddrout_ipv6; 322 323 (*sockaddrout) = (struct sockaddr_storage *) malloc(sizeof(struct sockaddr_in6)); 324 if ((*sockaddrout) == NULL) 325 { 326 pcap_fmt_errmsg_for_errno(errbuf, PCAP_ERRBUF_SIZE, 327 errno, "malloc() failed"); 328 return -1; 329 } 330 sockaddrin_ipv6 = (struct rpcap_sockaddr_in6 *) sockaddrin; 331 sockaddrout_ipv6 = (struct sockaddr_in6 *) (*sockaddrout); 332 sockaddrout_ipv6->sin6_family = AF_INET6; 333 sockaddrout_ipv6->sin6_port = ntohs(sockaddrin_ipv6->port); 334 sockaddrout_ipv6->sin6_flowinfo = ntohl(sockaddrin_ipv6->flowinfo); 335 memcpy(&sockaddrout_ipv6->sin6_addr, &sockaddrin_ipv6->addr, sizeof(sockaddrout_ipv6->sin6_addr)); 336 sockaddrout_ipv6->sin6_scope_id = ntohl(sockaddrin_ipv6->scope_id); 337 break; 338 } 339 #endif 340 341 default: 342 /* 343 * It is neither AF_INET nor AF_INET6 (or, if the OS doesn't 344 * support AF_INET6, it's not AF_INET). 345 */ 346 *sockaddrout = NULL; 347 break; 348 } 349 return 0; 350 } 351 352 /* 353 * This function reads a packet from the network socket. It does not 354 * deliver the packet to a pcap_dispatch()/pcap_loop() callback (hence 355 * the "nocb" string into its name). 356 * 357 * This function is called by pcap_read_rpcap(). 358 * 359 * WARNING: By choice, this function does not make use of semaphores. A smarter 360 * implementation should put a semaphore into the data thread, and a signal will 361 * be raised as soon as there is data into the socket buffer. 362 * However this is complicated and it does not bring any advantages when reading 363 * from the network, in which network delays can be much more important than 364 * these optimizations. Therefore, we chose the following approach: 365 * - the 'timeout' chosen by the user is split in two (half on the server side, 366 * with the usual meaning, and half on the client side) 367 * - this function checks for packets; if there are no packets, it waits for 368 * timeout/2 and then it checks again. If packets are still missing, it returns, 369 * otherwise it reads packets. 370 */ 371 static int pcap_read_nocb_remote(pcap_t *p, struct pcap_pkthdr *pkt_header, u_char **pkt_data) 372 { 373 struct pcap_rpcap *pr = p->priv; /* structure used when doing a remote live capture */ 374 struct rpcap_header *header; /* general header according to the RPCAP format */ 375 struct rpcap_pkthdr *net_pkt_header; /* header of the packet, from the message */ 376 u_char *net_pkt_data; /* packet data from the message */ 377 uint32 plen; 378 int retval; /* generic return value */ 379 int msglen; 380 381 /* Structures needed for the select() call */ 382 struct timeval tv; /* maximum time the select() can block waiting for data */ 383 fd_set rfds; /* set of socket descriptors we have to check */ 384 385 /* 386 * Define the packet buffer timeout, to be used in the select() 387 * 'timeout', in pcap_t, is in milliseconds; we have to convert it into sec and microsec 388 */ 389 tv.tv_sec = p->opt.timeout / 1000; 390 tv.tv_usec = (p->opt.timeout - tv.tv_sec * 1000) * 1000; 391 392 /* Watch out sockdata to see if it has input */ 393 FD_ZERO(&rfds); 394 395 /* 396 * 'fp->rmt_sockdata' has always to be set before calling the select(), 397 * since it is cleared by the select() 398 */ 399 FD_SET(pr->rmt_sockdata, &rfds); 400 401 retval = select((int) pr->rmt_sockdata + 1, &rfds, NULL, NULL, &tv); 402 if (retval == -1) 403 { 404 #ifndef _WIN32 405 if (errno == EINTR) 406 { 407 /* Interrupted. */ 408 return 0; 409 } 410 #endif 411 sock_geterror("select()", p->errbuf, PCAP_ERRBUF_SIZE); 412 return -1; 413 } 414 415 /* There is no data waiting, so return '0' */ 416 if (retval == 0) 417 return 0; 418 419 /* 420 * We have to define 'header' as a pointer to a larger buffer, 421 * because in case of UDP we have to read all the message within a single call 422 */ 423 header = (struct rpcap_header *) p->buffer; 424 net_pkt_header = (struct rpcap_pkthdr *) ((char *)p->buffer + sizeof(struct rpcap_header)); 425 net_pkt_data = (u_char *)p->buffer + sizeof(struct rpcap_header) + sizeof(struct rpcap_pkthdr); 426 427 if (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP) 428 { 429 /* Read the entire message from the network */ 430 msglen = sock_recv_dgram(pr->rmt_sockdata, p->buffer, 431 p->bufsize, p->errbuf, PCAP_ERRBUF_SIZE); 432 if (msglen == -1) 433 { 434 /* Network error. */ 435 return -1; 436 } 437 if (msglen == -3) 438 { 439 /* Interrupted receive. */ 440 return 0; 441 } 442 if ((size_t)msglen < sizeof(struct rpcap_header)) 443 { 444 /* 445 * Message is shorter than an rpcap header. 446 */ 447 pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 448 "UDP packet message is shorter than an rpcap header"); 449 return -1; 450 } 451 plen = ntohl(header->plen); 452 if ((size_t)msglen < sizeof(struct rpcap_header) + plen) 453 { 454 /* 455 * Message is shorter than the header claims it 456 * is. 457 */ 458 pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 459 "UDP packet message is shorter than its rpcap header claims"); 460 return -1; 461 } 462 } 463 else 464 { 465 int status; 466 467 if ((size_t)p->cc < sizeof(struct rpcap_header)) 468 { 469 /* 470 * We haven't read any of the packet header yet. 471 * The size we should get is the size of the 472 * packet header. 473 */ 474 status = rpcap_read_packet_msg(pr->rmt_sockdata, p, 475 sizeof(struct rpcap_header)); 476 if (status == -1) 477 { 478 /* Network error. */ 479 return -1; 480 } 481 if (status == -3) 482 { 483 /* Interrupted receive. */ 484 return 0; 485 } 486 } 487 488 /* 489 * We have the header, so we know how long the 490 * message payload is. The size we should get 491 * is the size of the packet header plus the 492 * size of the payload. 493 */ 494 plen = ntohl(header->plen); 495 if (plen > p->bufsize - sizeof(struct rpcap_header)) 496 { 497 /* 498 * This is bigger than the largest 499 * record we'd expect. (We do it by 500 * subtracting in order to avoid an 501 * overflow.) 502 */ 503 pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 504 "Server sent us a message larger than the largest expected packet message"); 505 return -1; 506 } 507 status = rpcap_read_packet_msg(pr->rmt_sockdata, p, 508 sizeof(struct rpcap_header) + plen); 509 if (status == -1) 510 { 511 /* Network error. */ 512 return -1; 513 } 514 if (status == -3) 515 { 516 /* Interrupted receive. */ 517 return 0; 518 } 519 520 /* 521 * We have the entire message; reset the buffer pointer 522 * and count, as the next read should start a new 523 * message. 524 */ 525 p->bp = p->buffer; 526 p->cc = 0; 527 } 528 529 /* 530 * We have the entire message. 531 */ 532 header->plen = plen; 533 534 /* 535 * Did the server specify the version we negotiated? 536 */ 537 if (rpcap_check_msg_ver(pr->rmt_sockdata, pr->protocol_version, 538 header, p->errbuf) == -1) 539 { 540 return 0; /* Return 'no packets received' */ 541 } 542 543 /* 544 * Is this a RPCAP_MSG_PACKET message? 545 */ 546 if (header->type != RPCAP_MSG_PACKET) 547 { 548 return 0; /* Return 'no packets received' */ 549 } 550 551 if (ntohl(net_pkt_header->caplen) > plen) 552 { 553 pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 554 "Packet's captured data goes past the end of the received packet message."); 555 return -1; 556 } 557 558 /* Fill in packet header */ 559 pkt_header->caplen = ntohl(net_pkt_header->caplen); 560 pkt_header->len = ntohl(net_pkt_header->len); 561 pkt_header->ts.tv_sec = ntohl(net_pkt_header->timestamp_sec); 562 pkt_header->ts.tv_usec = ntohl(net_pkt_header->timestamp_usec); 563 564 /* Supply a pointer to the beginning of the packet data */ 565 *pkt_data = net_pkt_data; 566 567 /* 568 * I don't update the counter of the packets dropped by the network since we're using TCP, 569 * therefore no packets are dropped. Just update the number of packets received correctly 570 */ 571 pr->TotCapt++; 572 573 if (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP) 574 { 575 unsigned int npkt; 576 577 /* We're using UDP, so we need to update the counter of the packets dropped by the network */ 578 npkt = ntohl(net_pkt_header->npkt); 579 580 if (pr->TotCapt != npkt) 581 { 582 pr->TotNetDrops += (npkt - pr->TotCapt); 583 pr->TotCapt = npkt; 584 } 585 } 586 587 /* Packet read successfully */ 588 return 1; 589 } 590 591 /* 592 * This function reads a packet from the network socket. 593 * 594 * This function relies on the pcap_read_nocb_remote to deliver packets. The 595 * difference, here, is that as soon as a packet is read, it is delivered 596 * to the application by means of a callback function. 597 */ 598 static int pcap_read_rpcap(pcap_t *p, int cnt, pcap_handler callback, u_char *user) 599 { 600 struct pcap_rpcap *pr = p->priv; /* structure used when doing a remote live capture */ 601 struct pcap_pkthdr pkt_header; 602 u_char *pkt_data; 603 int n = 0; 604 int ret; 605 606 /* 607 * If this is client-side, and we haven't already started 608 * the capture, start it now. 609 */ 610 if (pr->rmt_clientside) 611 { 612 /* We are on an remote capture */ 613 if (!pr->rmt_capstarted) 614 { 615 /* 616 * The capture isn't started yet, so try to 617 * start it. 618 */ 619 if (pcap_startcapture_remote(p)) 620 return -1; 621 } 622 } 623 624 while (n < cnt || PACKET_COUNT_IS_UNLIMITED(cnt)) 625 { 626 /* 627 * Has "pcap_breakloop()" been called? 628 */ 629 if (p->break_loop) { 630 /* 631 * Yes - clear the flag that indicates that it 632 * has, and return PCAP_ERROR_BREAK to indicate 633 * that we were told to break out of the loop. 634 */ 635 p->break_loop = 0; 636 return (PCAP_ERROR_BREAK); 637 } 638 639 /* 640 * Read some packets. 641 */ 642 ret = pcap_read_nocb_remote(p, &pkt_header, &pkt_data); 643 if (ret == 1) 644 { 645 /* 646 * We got a packet. Hand it to the callback 647 * and count it so we can return the count. 648 */ 649 (*callback)(user, &pkt_header, pkt_data); 650 n++; 651 } 652 else if (ret == -1) 653 { 654 /* Error. */ 655 return ret; 656 } 657 else 658 { 659 /* 660 * No packet; this could mean that we timed 661 * out, or that we got interrupted, or that 662 * we got a bad packet. 663 * 664 * Were we told to break out of the loop? 665 */ 666 if (p->break_loop) { 667 /* 668 * Yes. 669 */ 670 p->break_loop = 0; 671 return (PCAP_ERROR_BREAK); 672 } 673 /* No - return the number of packets we've processed. */ 674 return n; 675 } 676 } 677 return n; 678 } 679 680 /* 681 * This function sends a CLOSE command to the capture server. 682 * 683 * It is called when the user calls pcap_close(). It sends a command 684 * to our peer that says 'ok, let's stop capturing'. 685 * 686 * WARNING: Since we're closing the connection, we do not check for errors. 687 */ 688 static void pcap_cleanup_rpcap(pcap_t *fp) 689 { 690 struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */ 691 struct rpcap_header header; /* header of the RPCAP packet */ 692 struct activehosts *temp; /* temp var needed to scan the host list chain, to detect if we're in active mode */ 693 int active = 0; /* active mode or not? */ 694 695 /* detect if we're in active mode */ 696 temp = activeHosts; 697 while (temp) 698 { 699 if (temp->sockctrl == pr->rmt_sockctrl) 700 { 701 active = 1; 702 break; 703 } 704 temp = temp->next; 705 } 706 707 if (!active) 708 { 709 rpcap_createhdr(&header, pr->protocol_version, 710 RPCAP_MSG_CLOSE, 0, 0); 711 712 /* 713 * Send the close request; don't report any errors, as 714 * we're closing this pcap_t, and have no place to report 715 * the error. No reply is sent to this message. 716 */ 717 (void)sock_send(pr->rmt_sockctrl, (char *)&header, 718 sizeof(struct rpcap_header), NULL, 0); 719 } 720 else 721 { 722 rpcap_createhdr(&header, pr->protocol_version, 723 RPCAP_MSG_ENDCAP_REQ, 0, 0); 724 725 /* 726 * Send the end capture request; don't report any errors, 727 * as we're closing this pcap_t, and have no place to 728 * report the error. 729 */ 730 if (sock_send(pr->rmt_sockctrl, (char *)&header, 731 sizeof(struct rpcap_header), NULL, 0) == 0) 732 { 733 /* 734 * Wait for the answer; don't report any errors, 735 * as we're closing this pcap_t, and have no 736 * place to report the error. 737 */ 738 if (rpcap_process_msg_header(pr->rmt_sockctrl, 739 pr->protocol_version, RPCAP_MSG_ENDCAP_REQ, 740 &header, NULL) == 0) 741 { 742 (void)rpcap_discard(pr->rmt_sockctrl, 743 header.plen, NULL); 744 } 745 } 746 } 747 748 if (pr->rmt_sockdata) 749 { 750 sock_close(pr->rmt_sockdata, NULL, 0); 751 pr->rmt_sockdata = 0; 752 } 753 754 if ((!active) && (pr->rmt_sockctrl)) 755 sock_close(pr->rmt_sockctrl, NULL, 0); 756 757 pr->rmt_sockctrl = 0; 758 759 if (pr->currentfilter) 760 { 761 free(pr->currentfilter); 762 pr->currentfilter = NULL; 763 } 764 765 pcap_cleanup_live_common(fp); 766 767 /* To avoid inconsistencies in the number of sock_init() */ 768 sock_cleanup(); 769 } 770 771 /* 772 * This function retrieves network statistics from our peer; 773 * it provides only the standard statistics. 774 */ 775 static int pcap_stats_rpcap(pcap_t *p, struct pcap_stat *ps) 776 { 777 struct pcap_stat *retval; 778 779 retval = rpcap_stats_rpcap(p, ps, PCAP_STATS_STANDARD); 780 781 if (retval) 782 return 0; 783 else 784 return -1; 785 } 786 787 #ifdef _WIN32 788 /* 789 * This function retrieves network statistics from our peer; 790 * it provides the additional statistics supported by pcap_stats_ex(). 791 */ 792 static struct pcap_stat *pcap_stats_ex_rpcap(pcap_t *p, int *pcap_stat_size) 793 { 794 *pcap_stat_size = sizeof (p->stat); 795 796 /* PCAP_STATS_EX (third param) means 'extended pcap_stats()' */ 797 return (rpcap_stats_rpcap(p, &(p->stat), PCAP_STATS_EX)); 798 } 799 #endif 800 801 /* 802 * This function retrieves network statistics from our peer. It 803 * is used by the two previous functions. 804 * 805 * It can be called in two modes: 806 * - PCAP_STATS_STANDARD: if we want just standard statistics (i.e., 807 * for pcap_stats()) 808 * - PCAP_STATS_EX: if we want extended statistics (i.e., for 809 * pcap_stats_ex()) 810 * 811 * This 'mode' parameter is needed because in pcap_stats() the variable that 812 * keeps the statistics is allocated by the user. On Windows, this structure 813 * has been extended in order to keep new stats. However, if the user has a 814 * smaller structure and it passes it to pcap_stats(), this function will 815 * try to fill in more data than the size of the structure, so that memory 816 * after the structure will be overwritten. 817 * 818 * So, we need to know it we have to copy just the standard fields, or the 819 * extended fields as well. 820 * 821 * In case we want to copy the extended fields as well, the problem of 822 * memory overflow no longer exists because the structure that's filled 823 * in is part of the pcap_t, so that it can be guaranteed to be large 824 * enough for the additional statistics. 825 * 826 * \param p: the pcap_t structure related to the current instance. 827 * 828 * \param ps: a pointer to a 'pcap_stat' structure, needed for compatibility 829 * with pcap_stat(), where the structure is allocated by the user. In case 830 * of pcap_stats_ex(), this structure and the function return value point 831 * to the same variable. 832 * 833 * \param mode: one of PCAP_STATS_STANDARD or PCAP_STATS_EX. 834 * 835 * \return The structure that keeps the statistics, or NULL in case of error. 836 * The error string is placed in the pcap_t structure. 837 */ 838 static struct pcap_stat *rpcap_stats_rpcap(pcap_t *p, struct pcap_stat *ps, int mode) 839 { 840 struct pcap_rpcap *pr = p->priv; /* structure used when doing a remote live capture */ 841 struct rpcap_header header; /* header of the RPCAP packet */ 842 struct rpcap_stats netstats; /* statistics sent on the network */ 843 uint32 plen; /* data remaining in the message */ 844 845 #ifdef _WIN32 846 if (mode != PCAP_STATS_STANDARD && mode != PCAP_STATS_EX) 847 #else 848 if (mode != PCAP_STATS_STANDARD) 849 #endif 850 { 851 pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 852 "Invalid stats mode %d", mode); 853 return NULL; 854 } 855 856 /* 857 * If the capture has not yet started, we cannot request statistics 858 * for the capture from our peer, so we return 0 for all statistics, 859 * as nothing's been seen yet. 860 */ 861 if (!pr->rmt_capstarted) 862 { 863 ps->ps_drop = 0; 864 ps->ps_ifdrop = 0; 865 ps->ps_recv = 0; 866 #ifdef _WIN32 867 if (mode == PCAP_STATS_EX) 868 { 869 ps->ps_capt = 0; 870 ps->ps_sent = 0; 871 ps->ps_netdrop = 0; 872 } 873 #endif /* _WIN32 */ 874 875 return ps; 876 } 877 878 rpcap_createhdr(&header, pr->protocol_version, 879 RPCAP_MSG_STATS_REQ, 0, 0); 880 881 /* Send the PCAP_STATS command */ 882 if (sock_send(pr->rmt_sockctrl, (char *)&header, 883 sizeof(struct rpcap_header), p->errbuf, PCAP_ERRBUF_SIZE) < 0) 884 return NULL; /* Unrecoverable network error */ 885 886 /* Receive and process the reply message header. */ 887 if (rpcap_process_msg_header(pr->rmt_sockctrl, pr->protocol_version, 888 RPCAP_MSG_STATS_REQ, &header, p->errbuf) == -1) 889 return NULL; /* Error */ 890 891 plen = header.plen; 892 893 /* Read the reply body */ 894 if (rpcap_recv(pr->rmt_sockctrl, (char *)&netstats, 895 sizeof(struct rpcap_stats), &plen, p->errbuf) == -1) 896 goto error; 897 898 ps->ps_drop = ntohl(netstats.krnldrop); 899 ps->ps_ifdrop = ntohl(netstats.ifdrop); 900 ps->ps_recv = ntohl(netstats.ifrecv); 901 #ifdef _WIN32 902 if (mode == PCAP_STATS_EX) 903 { 904 ps->ps_capt = pr->TotCapt; 905 ps->ps_netdrop = pr->TotNetDrops; 906 ps->ps_sent = ntohl(netstats.svrcapt); 907 } 908 #endif /* _WIN32 */ 909 910 /* Discard the rest of the message. */ 911 if (rpcap_discard(pr->rmt_sockctrl, plen, p->errbuf) == -1) 912 goto error_nodiscard; 913 914 return ps; 915 916 error: 917 /* 918 * Discard the rest of the message. 919 * We already reported an error; if this gets an error, just 920 * drive on. 921 */ 922 (void)rpcap_discard(pr->rmt_sockctrl, plen, NULL); 923 924 error_nodiscard: 925 return NULL; 926 } 927 928 /* 929 * This function returns the entry in the list of active hosts for this 930 * active connection (active mode only), or NULL if there is no 931 * active connection or an error occurred. It is just for internal 932 * use. 933 * 934 * \param host: a string that keeps the host name of the host for which we 935 * want to get the socket ID for that active connection. 936 * 937 * \param error: a pointer to an int that is set to 1 if an error occurred 938 * and 0 otherwise. 939 * 940 * \param errbuf: a pointer to a user-allocated buffer (of size 941 * PCAP_ERRBUF_SIZE) that will contain the error message (in case 942 * there is one). 943 * 944 * \return the entry for this host in the list of active connections 945 * if found, NULL if it's not found or there's an error. 946 */ 947 static struct activehosts * 948 rpcap_remoteact_getsock(const char *host, int *error, char *errbuf) 949 { 950 struct activehosts *temp; /* temp var needed to scan the host list chain */ 951 struct addrinfo hints, *addrinfo, *ai_next; /* temp var needed to translate between hostname to its address */ 952 int retval; 953 954 /* retrieve the network address corresponding to 'host' */ 955 addrinfo = NULL; 956 memset(&hints, 0, sizeof(struct addrinfo)); 957 hints.ai_family = PF_UNSPEC; 958 hints.ai_socktype = SOCK_STREAM; 959 960 retval = getaddrinfo(host, "0", &hints, &addrinfo); 961 if (retval != 0) 962 { 963 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "getaddrinfo() %s", 964 gai_strerror(retval)); 965 *error = 1; 966 return NULL; 967 } 968 969 temp = activeHosts; 970 971 while (temp) 972 { 973 ai_next = addrinfo; 974 while (ai_next) 975 { 976 if (sock_cmpaddr(&temp->host, (struct sockaddr_storage *) ai_next->ai_addr) == 0) 977 { 978 *error = 0; 979 freeaddrinfo(addrinfo); 980 return temp; 981 } 982 983 ai_next = ai_next->ai_next; 984 } 985 temp = temp->next; 986 } 987 988 if (addrinfo) 989 freeaddrinfo(addrinfo); 990 991 /* 992 * The host for which you want to get the socket ID does not have an 993 * active connection. 994 */ 995 *error = 0; 996 return NULL; 997 } 998 999 /* 1000 * This function starts a remote capture. 1001 * 1002 * This function is required since the RPCAP protocol decouples the 'open' 1003 * from the 'start capture' functions. 1004 * This function takes all the parameters needed (which have been stored 1005 * into the pcap_t structure) and sends them to the server. 1006 * 1007 * \param fp: the pcap_t descriptor of the device currently open. 1008 * 1009 * \return '0' if everything is fine, '-1' otherwise. The error message 1010 * (if one) is returned into the 'errbuf' field of the pcap_t structure. 1011 */ 1012 static int pcap_startcapture_remote(pcap_t *fp) 1013 { 1014 struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */ 1015 char sendbuf[RPCAP_NETBUF_SIZE]; /* temporary buffer in which data to be sent is buffered */ 1016 int sendbufidx = 0; /* index which keeps the number of bytes currently buffered */ 1017 char portdata[PCAP_BUF_SIZE]; /* temp variable needed to keep the network port for the data connection */ 1018 uint32 plen; 1019 int active = 0; /* '1' if we're in active mode */ 1020 struct activehosts *temp; /* temp var needed to scan the host list chain, to detect if we're in active mode */ 1021 char host[INET6_ADDRSTRLEN + 1]; /* numeric name of the other host */ 1022 1023 /* socket-related variables*/ 1024 struct addrinfo hints; /* temp, needed to open a socket connection */ 1025 struct addrinfo *addrinfo; /* temp, needed to open a socket connection */ 1026 SOCKET sockdata = 0; /* socket descriptor of the data connection */ 1027 struct sockaddr_storage saddr; /* temp, needed to retrieve the network data port chosen on the local machine */ 1028 socklen_t saddrlen; /* temp, needed to retrieve the network data port chosen on the local machine */ 1029 int ai_family; /* temp, keeps the address family used by the control connection */ 1030 1031 /* RPCAP-related variables*/ 1032 struct rpcap_header header; /* header of the RPCAP packet */ 1033 struct rpcap_startcapreq *startcapreq; /* start capture request message */ 1034 struct rpcap_startcapreply startcapreply; /* start capture reply message */ 1035 1036 /* Variables related to the buffer setting */ 1037 int res; 1038 socklen_t itemp; 1039 int sockbufsize = 0; 1040 uint32 server_sockbufsize; 1041 1042 /* 1043 * Let's check if sampling has been required. 1044 * If so, let's set it first 1045 */ 1046 if (pcap_setsampling_remote(fp) != 0) 1047 return -1; 1048 1049 /* detect if we're in active mode */ 1050 temp = activeHosts; 1051 while (temp) 1052 { 1053 if (temp->sockctrl == pr->rmt_sockctrl) 1054 { 1055 active = 1; 1056 break; 1057 } 1058 temp = temp->next; 1059 } 1060 1061 addrinfo = NULL; 1062 1063 /* 1064 * Gets the complete sockaddr structure used in the ctrl connection 1065 * This is needed to get the address family of the control socket 1066 * Tip: I cannot save the ai_family of the ctrl sock in the pcap_t struct, 1067 * since the ctrl socket can already be open in case of active mode; 1068 * so I would have to call getpeername() anyway 1069 */ 1070 saddrlen = sizeof(struct sockaddr_storage); 1071 if (getpeername(pr->rmt_sockctrl, (struct sockaddr *) &saddr, &saddrlen) == -1) 1072 { 1073 sock_geterror("getsockname()", fp->errbuf, PCAP_ERRBUF_SIZE); 1074 goto error_nodiscard; 1075 } 1076 ai_family = ((struct sockaddr_storage *) &saddr)->ss_family; 1077 1078 /* Get the numeric address of the remote host we are connected to */ 1079 if (getnameinfo((struct sockaddr *) &saddr, saddrlen, host, 1080 sizeof(host), NULL, 0, NI_NUMERICHOST)) 1081 { 1082 sock_geterror("getnameinfo()", fp->errbuf, PCAP_ERRBUF_SIZE); 1083 goto error_nodiscard; 1084 } 1085 1086 /* 1087 * Data connection is opened by the server toward the client if: 1088 * - we're using TCP, and the user wants us to be in active mode 1089 * - we're using UDP 1090 */ 1091 if ((active) || (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP)) 1092 { 1093 /* 1094 * We have to create a new socket to receive packets 1095 * We have to do that immediately, since we have to tell the other 1096 * end which network port we picked up 1097 */ 1098 memset(&hints, 0, sizeof(struct addrinfo)); 1099 /* TEMP addrinfo is NULL in case of active */ 1100 hints.ai_family = ai_family; /* Use the same address family of the control socket */ 1101 hints.ai_socktype = (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP) ? SOCK_DGRAM : SOCK_STREAM; 1102 hints.ai_flags = AI_PASSIVE; /* Data connection is opened by the server toward the client */ 1103 1104 /* Let's the server pick up a free network port for us */ 1105 if (sock_initaddress(NULL, "0", &hints, &addrinfo, fp->errbuf, PCAP_ERRBUF_SIZE) == -1) 1106 goto error_nodiscard; 1107 1108 if ((sockdata = sock_open(addrinfo, SOCKOPEN_SERVER, 1109 1 /* max 1 connection in queue */, fp->errbuf, PCAP_ERRBUF_SIZE)) == INVALID_SOCKET) 1110 goto error_nodiscard; 1111 1112 /* addrinfo is no longer used */ 1113 freeaddrinfo(addrinfo); 1114 addrinfo = NULL; 1115 1116 /* get the complete sockaddr structure used in the data connection */ 1117 saddrlen = sizeof(struct sockaddr_storage); 1118 if (getsockname(sockdata, (struct sockaddr *) &saddr, &saddrlen) == -1) 1119 { 1120 sock_geterror("getsockname()", fp->errbuf, PCAP_ERRBUF_SIZE); 1121 goto error_nodiscard; 1122 } 1123 1124 /* Get the local port the system picked up */ 1125 if (getnameinfo((struct sockaddr *) &saddr, saddrlen, NULL, 1126 0, portdata, sizeof(portdata), NI_NUMERICSERV)) 1127 { 1128 sock_geterror("getnameinfo()", fp->errbuf, PCAP_ERRBUF_SIZE); 1129 goto error_nodiscard; 1130 } 1131 } 1132 1133 /* 1134 * Now it's time to start playing with the RPCAP protocol 1135 * RPCAP start capture command: create the request message 1136 */ 1137 if (sock_bufferize(NULL, sizeof(struct rpcap_header), NULL, 1138 &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE)) 1139 goto error_nodiscard; 1140 1141 rpcap_createhdr((struct rpcap_header *) sendbuf, 1142 pr->protocol_version, RPCAP_MSG_STARTCAP_REQ, 0, 1143 sizeof(struct rpcap_startcapreq) + sizeof(struct rpcap_filter) + fp->fcode.bf_len * sizeof(struct rpcap_filterbpf_insn)); 1144 1145 /* Fill the structure needed to open an adapter remotely */ 1146 startcapreq = (struct rpcap_startcapreq *) &sendbuf[sendbufidx]; 1147 1148 if (sock_bufferize(NULL, sizeof(struct rpcap_startcapreq), NULL, 1149 &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE)) 1150 goto error_nodiscard; 1151 1152 memset(startcapreq, 0, sizeof(struct rpcap_startcapreq)); 1153 1154 /* By default, apply half the timeout on one side, half of the other */ 1155 fp->opt.timeout = fp->opt.timeout / 2; 1156 startcapreq->read_timeout = htonl(fp->opt.timeout); 1157 1158 /* portdata on the openreq is meaningful only if we're in active mode */ 1159 if ((active) || (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP)) 1160 { 1161 sscanf(portdata, "%d", (int *)&(startcapreq->portdata)); /* cast to avoid a compiler warning */ 1162 startcapreq->portdata = htons(startcapreq->portdata); 1163 } 1164 1165 startcapreq->snaplen = htonl(fp->snapshot); 1166 startcapreq->flags = 0; 1167 1168 if (pr->rmt_flags & PCAP_OPENFLAG_PROMISCUOUS) 1169 startcapreq->flags |= RPCAP_STARTCAPREQ_FLAG_PROMISC; 1170 if (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP) 1171 startcapreq->flags |= RPCAP_STARTCAPREQ_FLAG_DGRAM; 1172 if (active) 1173 startcapreq->flags |= RPCAP_STARTCAPREQ_FLAG_SERVEROPEN; 1174 1175 startcapreq->flags = htons(startcapreq->flags); 1176 1177 /* Pack the capture filter */ 1178 if (pcap_pack_bpffilter(fp, &sendbuf[sendbufidx], &sendbufidx, &fp->fcode)) 1179 goto error_nodiscard; 1180 1181 if (sock_send(pr->rmt_sockctrl, sendbuf, sendbufidx, fp->errbuf, 1182 PCAP_ERRBUF_SIZE) < 0) 1183 goto error_nodiscard; 1184 1185 /* Receive and process the reply message header. */ 1186 if (rpcap_process_msg_header(pr->rmt_sockctrl, pr->protocol_version, 1187 RPCAP_MSG_STARTCAP_REQ, &header, fp->errbuf) == -1) 1188 goto error_nodiscard; 1189 1190 plen = header.plen; 1191 1192 if (rpcap_recv(pr->rmt_sockctrl, (char *)&startcapreply, 1193 sizeof(struct rpcap_startcapreply), &plen, fp->errbuf) == -1) 1194 goto error; 1195 1196 /* 1197 * In case of UDP data stream, the connection is always opened by the daemon 1198 * So, this case is already covered by the code above. 1199 * Now, we have still to handle TCP connections, because: 1200 * - if we're in active mode, we have to wait for a remote connection 1201 * - if we're in passive more, we have to start a connection 1202 * 1203 * We have to do he job in two steps because in case we're opening a TCP connection, we have 1204 * to tell the port we're using to the remote side; in case we're accepting a TCP 1205 * connection, we have to wait this info from the remote side. 1206 */ 1207 if (!(pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP)) 1208 { 1209 if (!active) 1210 { 1211 memset(&hints, 0, sizeof(struct addrinfo)); 1212 hints.ai_family = ai_family; /* Use the same address family of the control socket */ 1213 hints.ai_socktype = (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP) ? SOCK_DGRAM : SOCK_STREAM; 1214 pcap_snprintf(portdata, PCAP_BUF_SIZE, "%d", ntohs(startcapreply.portdata)); 1215 1216 /* Let's the server pick up a free network port for us */ 1217 if (sock_initaddress(host, portdata, &hints, &addrinfo, fp->errbuf, PCAP_ERRBUF_SIZE) == -1) 1218 goto error; 1219 1220 if ((sockdata = sock_open(addrinfo, SOCKOPEN_CLIENT, 0, fp->errbuf, PCAP_ERRBUF_SIZE)) == INVALID_SOCKET) 1221 goto error; 1222 1223 /* addrinfo is no longer used */ 1224 freeaddrinfo(addrinfo); 1225 addrinfo = NULL; 1226 } 1227 else 1228 { 1229 SOCKET socktemp; /* We need another socket, since we're going to accept() a connection */ 1230 1231 /* Connection creation */ 1232 saddrlen = sizeof(struct sockaddr_storage); 1233 1234 socktemp = accept(sockdata, (struct sockaddr *) &saddr, &saddrlen); 1235 1236 if (socktemp == INVALID_SOCKET) 1237 { 1238 sock_geterror("accept()", fp->errbuf, PCAP_ERRBUF_SIZE); 1239 goto error; 1240 } 1241 1242 /* Now that I accepted the connection, the server socket is no longer needed */ 1243 sock_close(sockdata, fp->errbuf, PCAP_ERRBUF_SIZE); 1244 sockdata = socktemp; 1245 } 1246 } 1247 1248 /* Let's save the socket of the data connection */ 1249 pr->rmt_sockdata = sockdata; 1250 1251 /* 1252 * Set the size of the socket buffer for the data socket. 1253 * It has the same size as the local capture buffer used 1254 * on the other side of the connection. 1255 */ 1256 server_sockbufsize = ntohl(startcapreply.bufsize); 1257 1258 /* Let's get the actual size of the socket buffer */ 1259 itemp = sizeof(sockbufsize); 1260 1261 res = getsockopt(sockdata, SOL_SOCKET, SO_RCVBUF, (char *)&sockbufsize, &itemp); 1262 if (res == -1) 1263 { 1264 sock_geterror("pcap_startcapture_remote(): getsockopt() failed", fp->errbuf, PCAP_ERRBUF_SIZE); 1265 goto error; 1266 } 1267 1268 /* 1269 * Warning: on some kernels (e.g. Linux), the size of the user 1270 * buffer does not take into account the pcap_header and such, 1271 * and it is set equal to the snaplen. 1272 * 1273 * In my view, this is wrong (the meaning of the bufsize became 1274 * a bit strange). So, here bufsize is the whole size of the 1275 * user buffer. In case the bufsize returned is too small, 1276 * let's adjust it accordingly. 1277 */ 1278 if (server_sockbufsize <= (u_int) fp->snapshot) 1279 server_sockbufsize += sizeof(struct pcap_pkthdr); 1280 1281 /* if the current socket buffer is smaller than the desired one */ 1282 if ((u_int) sockbufsize < server_sockbufsize) 1283 { 1284 /* 1285 * Loop until the buffer size is OK or the original 1286 * socket buffer size is larger than this one. 1287 */ 1288 for (;;) 1289 { 1290 res = setsockopt(sockdata, SOL_SOCKET, SO_RCVBUF, 1291 (char *)&(server_sockbufsize), 1292 sizeof(server_sockbufsize)); 1293 1294 if (res == 0) 1295 break; 1296 1297 /* 1298 * If something goes wrong, halve the buffer size 1299 * (checking that it does not become smaller than 1300 * the current one). 1301 */ 1302 server_sockbufsize /= 2; 1303 1304 if ((u_int) sockbufsize >= server_sockbufsize) 1305 { 1306 server_sockbufsize = sockbufsize; 1307 break; 1308 } 1309 } 1310 } 1311 1312 /* 1313 * Let's allocate the packet; this is required in order to put 1314 * the packet somewhere when extracting data from the socket. 1315 * Since buffering has already been done in the socket buffer, 1316 * here we need just a buffer whose size is equal to the 1317 * largest possible packet message for the snapshot size, 1318 * namely the length of the message header plus the length 1319 * of the packet header plus the snapshot length. 1320 */ 1321 fp->bufsize = sizeof(struct rpcap_header) + sizeof(struct rpcap_pkthdr) + fp->snapshot; 1322 1323 fp->buffer = (u_char *)malloc(fp->bufsize); 1324 if (fp->buffer == NULL) 1325 { 1326 pcap_fmt_errmsg_for_errno(fp->errbuf, PCAP_ERRBUF_SIZE, 1327 errno, "malloc"); 1328 goto error; 1329 } 1330 1331 /* 1332 * The buffer is currently empty. 1333 */ 1334 fp->bp = fp->buffer; 1335 fp->cc = 0; 1336 1337 /* Discard the rest of the message. */ 1338 if (rpcap_discard(pr->rmt_sockctrl, plen, fp->errbuf) == -1) 1339 goto error_nodiscard; 1340 1341 /* 1342 * In case the user does not want to capture RPCAP packets, let's update the filter 1343 * We have to update it here (instead of sending it into the 'StartCapture' message 1344 * because when we generate the 'start capture' we do not know (yet) all the ports 1345 * we're currently using. 1346 */ 1347 if (pr->rmt_flags & PCAP_OPENFLAG_NOCAPTURE_RPCAP) 1348 { 1349 struct bpf_program fcode; 1350 1351 if (pcap_createfilter_norpcappkt(fp, &fcode) == -1) 1352 goto error; 1353 1354 /* We cannot use 'pcap_setfilter_rpcap' because formally the capture has not been started yet */ 1355 /* (the 'pr->rmt_capstarted' variable will be updated some lines below) */ 1356 if (pcap_updatefilter_remote(fp, &fcode) == -1) 1357 goto error; 1358 1359 pcap_freecode(&fcode); 1360 } 1361 1362 pr->rmt_capstarted = 1; 1363 return 0; 1364 1365 error: 1366 /* 1367 * When the connection has been established, we have to close it. So, at the 1368 * beginning of this function, if an error occur we return immediately with 1369 * a return NULL; when the connection is established, we have to come here 1370 * ('goto error;') in order to close everything properly. 1371 */ 1372 1373 /* 1374 * Discard the rest of the message. 1375 * We already reported an error; if this gets an error, just 1376 * drive on. 1377 */ 1378 (void)rpcap_discard(pr->rmt_sockctrl, plen, NULL); 1379 1380 error_nodiscard: 1381 if ((sockdata) && (sockdata != -1)) /* we can be here because sockdata said 'error' */ 1382 sock_close(sockdata, NULL, 0); 1383 1384 if (!active) 1385 sock_close(pr->rmt_sockctrl, NULL, 0); 1386 1387 if (addrinfo != NULL) 1388 freeaddrinfo(addrinfo); 1389 1390 /* 1391 * We do not have to call pcap_close() here, because this function is always called 1392 * by the user in case something bad happens 1393 */ 1394 #if 0 1395 if (fp) 1396 { 1397 pcap_close(fp); 1398 fp= NULL; 1399 } 1400 #endif 1401 1402 return -1; 1403 } 1404 1405 /* 1406 * This function takes a bpf program and sends it to the other host. 1407 * 1408 * This function can be called in two cases: 1409 * - pcap_startcapture_remote() is called (we have to send the filter 1410 * along with the 'start capture' command) 1411 * - we want to udpate the filter during a capture (i.e. pcap_setfilter() 1412 * after the capture has been started) 1413 * 1414 * This function serializes the filter into the sending buffer ('sendbuf', 1415 * passed as a parameter) and return back. It does not send anything on 1416 * the network. 1417 * 1418 * \param fp: the pcap_t descriptor of the device currently opened. 1419 * 1420 * \param sendbuf: the buffer on which the serialized data has to copied. 1421 * 1422 * \param sendbufidx: it is used to return the abounf of bytes copied into the buffer. 1423 * 1424 * \param prog: the bpf program we have to copy. 1425 * 1426 * \return '0' if everything is fine, '-1' otherwise. The error message (if one) 1427 * is returned into the 'errbuf' field of the pcap_t structure. 1428 */ 1429 static int pcap_pack_bpffilter(pcap_t *fp, char *sendbuf, int *sendbufidx, struct bpf_program *prog) 1430 { 1431 struct rpcap_filter *filter; 1432 struct rpcap_filterbpf_insn *insn; 1433 struct bpf_insn *bf_insn; 1434 struct bpf_program fake_prog; /* To be used just in case the user forgot to set a filter */ 1435 unsigned int i; 1436 1437 if (prog->bf_len == 0) /* No filters have been specified; so, let's apply a "fake" filter */ 1438 { 1439 if (pcap_compile(fp, &fake_prog, NULL /* buffer */, 1, 0) == -1) 1440 return -1; 1441 1442 prog = &fake_prog; 1443 } 1444 1445 filter = (struct rpcap_filter *) sendbuf; 1446 1447 if (sock_bufferize(NULL, sizeof(struct rpcap_filter), NULL, sendbufidx, 1448 RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE)) 1449 return -1; 1450 1451 filter->filtertype = htons(RPCAP_UPDATEFILTER_BPF); 1452 filter->nitems = htonl((int32)prog->bf_len); 1453 1454 if (sock_bufferize(NULL, prog->bf_len * sizeof(struct rpcap_filterbpf_insn), 1455 NULL, sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE)) 1456 return -1; 1457 1458 insn = (struct rpcap_filterbpf_insn *) (filter + 1); 1459 bf_insn = prog->bf_insns; 1460 1461 for (i = 0; i < prog->bf_len; i++) 1462 { 1463 insn->code = htons(bf_insn->code); 1464 insn->jf = bf_insn->jf; 1465 insn->jt = bf_insn->jt; 1466 insn->k = htonl(bf_insn->k); 1467 1468 insn++; 1469 bf_insn++; 1470 } 1471 1472 return 0; 1473 } 1474 1475 /* 1476 * This function updates a filter on a remote host. 1477 * 1478 * It is called when the user wants to update a filter. 1479 * In case we're capturing from the network, it sends the filter to our 1480 * peer. 1481 * This function is *not* called automatically when the user calls 1482 * pcap_setfilter(). 1483 * There will be two cases: 1484 * - the capture has been started: in this case, pcap_setfilter_rpcap() 1485 * calls pcap_updatefilter_remote() 1486 * - the capture has not started yet: in this case, pcap_setfilter_rpcap() 1487 * stores the filter into the pcap_t structure, and then the filter is 1488 * sent with pcap_startcap(). 1489 * 1490 * WARNING This function *does not* clear the packet currently into the 1491 * buffers. Therefore, the user has to expect to receive some packets 1492 * that are related to the previous filter. If you want to discard all 1493 * the packets before applying a new filter, you have to close the 1494 * current capture session and start a new one. 1495 * 1496 * XXX - we really should have pcap_setfilter() always discard packets 1497 * received with the old filter, and have a separate pcap_setfilter_noflush() 1498 * function that doesn't discard any packets. 1499 */ 1500 static int pcap_updatefilter_remote(pcap_t *fp, struct bpf_program *prog) 1501 { 1502 struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */ 1503 char sendbuf[RPCAP_NETBUF_SIZE]; /* temporary buffer in which data to be sent is buffered */ 1504 int sendbufidx = 0; /* index which keeps the number of bytes currently buffered */ 1505 struct rpcap_header header; /* To keep the reply message */ 1506 1507 if (sock_bufferize(NULL, sizeof(struct rpcap_header), NULL, &sendbufidx, 1508 RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE)) 1509 return -1; 1510 1511 rpcap_createhdr((struct rpcap_header *) sendbuf, 1512 pr->protocol_version, RPCAP_MSG_UPDATEFILTER_REQ, 0, 1513 sizeof(struct rpcap_filter) + prog->bf_len * sizeof(struct rpcap_filterbpf_insn)); 1514 1515 if (pcap_pack_bpffilter(fp, &sendbuf[sendbufidx], &sendbufidx, prog)) 1516 return -1; 1517 1518 if (sock_send(pr->rmt_sockctrl, sendbuf, sendbufidx, fp->errbuf, 1519 PCAP_ERRBUF_SIZE) < 0) 1520 return -1; 1521 1522 /* Receive and process the reply message header. */ 1523 if (rpcap_process_msg_header(pr->rmt_sockctrl, pr->protocol_version, 1524 RPCAP_MSG_UPDATEFILTER_REQ, &header, fp->errbuf) == -1) 1525 return -1; 1526 1527 /* 1528 * It shouldn't have any contents; discard it if it does. 1529 */ 1530 if (rpcap_discard(pr->rmt_sockctrl, header.plen, fp->errbuf) == -1) 1531 return -1; 1532 1533 return 0; 1534 } 1535 1536 static void 1537 pcap_save_current_filter_rpcap(pcap_t *fp, const char *filter) 1538 { 1539 struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */ 1540 1541 /* 1542 * Check if: 1543 * - We are on an remote capture 1544 * - we do not want to capture RPCAP traffic 1545 * 1546 * If so, we have to save the current filter, because we have to 1547 * add some piece of stuff later 1548 */ 1549 if (pr->rmt_clientside && 1550 (pr->rmt_flags & PCAP_OPENFLAG_NOCAPTURE_RPCAP)) 1551 { 1552 if (pr->currentfilter) 1553 free(pr->currentfilter); 1554 1555 if (filter == NULL) 1556 filter = ""; 1557 1558 pr->currentfilter = strdup(filter); 1559 } 1560 } 1561 1562 /* 1563 * This function sends a filter to a remote host. 1564 * 1565 * This function is called when the user wants to set a filter. 1566 * It sends the filter to our peer. 1567 * This function is called automatically when the user calls pcap_setfilter(). 1568 * 1569 * Parameters and return values are exactly the same of pcap_setfilter(). 1570 */ 1571 static int pcap_setfilter_rpcap(pcap_t *fp, struct bpf_program *prog) 1572 { 1573 struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */ 1574 1575 if (!pr->rmt_capstarted) 1576 { 1577 /* copy filter into the pcap_t structure */ 1578 if (install_bpf_program(fp, prog) == -1) 1579 return -1; 1580 return 0; 1581 } 1582 1583 /* we have to update a filter during run-time */ 1584 if (pcap_updatefilter_remote(fp, prog)) 1585 return -1; 1586 1587 return 0; 1588 } 1589 1590 /* 1591 * This function updates the current filter in order not to capture rpcap 1592 * packets. 1593 * 1594 * This function is called *only* when the user wants exclude RPCAP packets 1595 * related to the current session from the captured packets. 1596 * 1597 * \return '0' if everything is fine, '-1' otherwise. The error message (if one) 1598 * is returned into the 'errbuf' field of the pcap_t structure. 1599 */ 1600 static int pcap_createfilter_norpcappkt(pcap_t *fp, struct bpf_program *prog) 1601 { 1602 struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */ 1603 int RetVal = 0; 1604 1605 /* We do not want to capture our RPCAP traffic. So, let's update the filter */ 1606 if (pr->rmt_flags & PCAP_OPENFLAG_NOCAPTURE_RPCAP) 1607 { 1608 struct sockaddr_storage saddr; /* temp, needed to retrieve the network data port chosen on the local machine */ 1609 socklen_t saddrlen; /* temp, needed to retrieve the network data port chosen on the local machine */ 1610 char myaddress[128]; 1611 char myctrlport[128]; 1612 char mydataport[128]; 1613 char peeraddress[128]; 1614 char peerctrlport[128]; 1615 char *newfilter; 1616 1617 /* Get the name/port of our peer */ 1618 saddrlen = sizeof(struct sockaddr_storage); 1619 if (getpeername(pr->rmt_sockctrl, (struct sockaddr *) &saddr, &saddrlen) == -1) 1620 { 1621 sock_geterror("getpeername()", fp->errbuf, PCAP_ERRBUF_SIZE); 1622 return -1; 1623 } 1624 1625 if (getnameinfo((struct sockaddr *) &saddr, saddrlen, peeraddress, 1626 sizeof(peeraddress), peerctrlport, sizeof(peerctrlport), NI_NUMERICHOST | NI_NUMERICSERV)) 1627 { 1628 sock_geterror("getnameinfo()", fp->errbuf, PCAP_ERRBUF_SIZE); 1629 return -1; 1630 } 1631 1632 /* We cannot check the data port, because this is available only in case of TCP sockets */ 1633 /* Get the name/port of the current host */ 1634 if (getsockname(pr->rmt_sockctrl, (struct sockaddr *) &saddr, &saddrlen) == -1) 1635 { 1636 sock_geterror("getsockname()", fp->errbuf, PCAP_ERRBUF_SIZE); 1637 return -1; 1638 } 1639 1640 /* Get the local port the system picked up */ 1641 if (getnameinfo((struct sockaddr *) &saddr, saddrlen, myaddress, 1642 sizeof(myaddress), myctrlport, sizeof(myctrlport), NI_NUMERICHOST | NI_NUMERICSERV)) 1643 { 1644 sock_geterror("getnameinfo()", fp->errbuf, PCAP_ERRBUF_SIZE); 1645 return -1; 1646 } 1647 1648 /* Let's now check the data port */ 1649 if (getsockname(pr->rmt_sockdata, (struct sockaddr *) &saddr, &saddrlen) == -1) 1650 { 1651 sock_geterror("getsockname()", fp->errbuf, PCAP_ERRBUF_SIZE); 1652 return -1; 1653 } 1654 1655 /* Get the local port the system picked up */ 1656 if (getnameinfo((struct sockaddr *) &saddr, saddrlen, NULL, 0, mydataport, sizeof(mydataport), NI_NUMERICSERV)) 1657 { 1658 sock_geterror("getnameinfo()", fp->errbuf, PCAP_ERRBUF_SIZE); 1659 return -1; 1660 } 1661 1662 if (pr->currentfilter && pr->currentfilter[0] != '\0') 1663 { 1664 /* 1665 * We have a current filter; add items to it to 1666 * filter out this rpcap session. 1667 */ 1668 if (pcap_asprintf(&newfilter, 1669 "(%s) and not (host %s and host %s and port %s and port %s) and not (host %s and host %s and port %s)", 1670 pr->currentfilter, myaddress, peeraddress, 1671 myctrlport, peerctrlport, myaddress, peeraddress, 1672 mydataport) == -1) 1673 { 1674 /* Failed. */ 1675 pcap_snprintf(fp->errbuf, PCAP_ERRBUF_SIZE, 1676 "Can't allocate memory for new filter"); 1677 return -1; 1678 } 1679 } 1680 else 1681 { 1682 /* 1683 * We have no current filter; construct a filter to 1684 * filter out this rpcap session. 1685 */ 1686 if (pcap_asprintf(&newfilter, 1687 "not (host %s and host %s and port %s and port %s) and not (host %s and host %s and port %s)", 1688 myaddress, peeraddress, myctrlport, peerctrlport, 1689 myaddress, peeraddress, mydataport) == -1) 1690 { 1691 /* Failed. */ 1692 pcap_snprintf(fp->errbuf, PCAP_ERRBUF_SIZE, 1693 "Can't allocate memory for new filter"); 1694 return -1; 1695 } 1696 } 1697 1698 /* 1699 * This is only an hack to prevent the save_current_filter 1700 * routine, which will be called when we call pcap_compile(), 1701 * from saving the modified filter. 1702 */ 1703 pr->rmt_clientside = 0; 1704 1705 if (pcap_compile(fp, prog, newfilter, 1, 0) == -1) 1706 RetVal = -1; 1707 1708 /* Undo the hack. */ 1709 pr->rmt_clientside = 1; 1710 1711 free(newfilter); 1712 } 1713 1714 return RetVal; 1715 } 1716 1717 /* 1718 * This function sets sampling parameters in the remote host. 1719 * 1720 * It is called when the user wants to set activate sampling on the 1721 * remote host. 1722 * 1723 * Sampling parameters are defined into the 'pcap_t' structure. 1724 * 1725 * \param p: the pcap_t descriptor of the device currently opened. 1726 * 1727 * \return '0' if everything is OK, '-1' is something goes wrong. The 1728 * error message is returned in the 'errbuf' member of the pcap_t structure. 1729 */ 1730 static int pcap_setsampling_remote(pcap_t *fp) 1731 { 1732 struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */ 1733 char sendbuf[RPCAP_NETBUF_SIZE];/* temporary buffer in which data to be sent is buffered */ 1734 int sendbufidx = 0; /* index which keeps the number of bytes currently buffered */ 1735 struct rpcap_header header; /* To keep the reply message */ 1736 struct rpcap_sampling *sampling_pars; /* Structure that is needed to send sampling parameters to the remote host */ 1737 1738 /* If no samping is requested, return 'ok' */ 1739 if (fp->rmt_samp.method == PCAP_SAMP_NOSAMP) 1740 return 0; 1741 1742 /* 1743 * Check for sampling parameters that don't fit in a message. 1744 * We'll let the server complain about invalid parameters 1745 * that do fit into the message. 1746 */ 1747 if (fp->rmt_samp.method < 0 || fp->rmt_samp.method > 255) { 1748 pcap_snprintf(fp->errbuf, PCAP_ERRBUF_SIZE, 1749 "Invalid sampling method %d", fp->rmt_samp.method); 1750 return -1; 1751 } 1752 if (fp->rmt_samp.value < 0 || fp->rmt_samp.value > 65535) { 1753 pcap_snprintf(fp->errbuf, PCAP_ERRBUF_SIZE, 1754 "Invalid sampling value %d", fp->rmt_samp.value); 1755 return -1; 1756 } 1757 1758 if (sock_bufferize(NULL, sizeof(struct rpcap_header), NULL, 1759 &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE)) 1760 return -1; 1761 1762 rpcap_createhdr((struct rpcap_header *) sendbuf, 1763 pr->protocol_version, RPCAP_MSG_SETSAMPLING_REQ, 0, 1764 sizeof(struct rpcap_sampling)); 1765 1766 /* Fill the structure needed to open an adapter remotely */ 1767 sampling_pars = (struct rpcap_sampling *) &sendbuf[sendbufidx]; 1768 1769 if (sock_bufferize(NULL, sizeof(struct rpcap_sampling), NULL, 1770 &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE)) 1771 return -1; 1772 1773 memset(sampling_pars, 0, sizeof(struct rpcap_sampling)); 1774 1775 sampling_pars->method = (uint8)fp->rmt_samp.method; 1776 sampling_pars->value = (uint16)htonl(fp->rmt_samp.value); 1777 1778 if (sock_send(pr->rmt_sockctrl, sendbuf, sendbufidx, fp->errbuf, 1779 PCAP_ERRBUF_SIZE) < 0) 1780 return -1; 1781 1782 /* Receive and process the reply message header. */ 1783 if (rpcap_process_msg_header(pr->rmt_sockctrl, pr->protocol_version, 1784 RPCAP_MSG_SETSAMPLING_REQ, &header, fp->errbuf) == -1) 1785 return -1; 1786 1787 /* 1788 * It shouldn't have any contents; discard it if it does. 1789 */ 1790 if (rpcap_discard(pr->rmt_sockctrl, header.plen, fp->errbuf) == -1) 1791 return -1; 1792 1793 return 0; 1794 } 1795 1796 /********************************************************* 1797 * * 1798 * Miscellaneous functions * 1799 * * 1800 *********************************************************/ 1801 1802 /* 1803 * This function performs authentication and protocol version 1804 * negotiation. It is required in order to open the connection 1805 * with the other end party. 1806 * 1807 * It sends authentication parameters on the control socket and 1808 * reads the reply. If the reply is a success indication, it 1809 * checks whether the reply includes minimum and maximum supported 1810 * versions from the server; if not, it assumes both are 0, as 1811 * that means it's an older server that doesn't return supported 1812 * version numbers in authentication replies, so it only supports 1813 * version 0. It then tries to determine the maximum version 1814 * supported both by us and by the server. If it can find such a 1815 * version, it sets us up to use that version; otherwise, it fails, 1816 * indicating that there is no version supported by us and by the 1817 * server. 1818 * 1819 * \param sock: the socket we are currently using. 1820 * 1821 * \param ver: pointer to variable to which to set the protocol version 1822 * number we selected. 1823 * 1824 * \param auth: authentication parameters that have to be sent. 1825 * 1826 * \param errbuf: a pointer to a user-allocated buffer (of size 1827 * PCAP_ERRBUF_SIZE) that will contain the error message (in case there 1828 * is one). It could be a network problem or the fact that the authorization 1829 * failed. 1830 * 1831 * \return '0' if everything is fine, '-1' for an error. For errors, 1832 * an error message string is returned in the 'errbuf' variable. 1833 */ 1834 static int rpcap_doauth(SOCKET sockctrl, uint8 *ver, struct pcap_rmtauth *auth, char *errbuf) 1835 { 1836 char sendbuf[RPCAP_NETBUF_SIZE]; /* temporary buffer in which data that has to be sent is buffered */ 1837 int sendbufidx = 0; /* index which keeps the number of bytes currently buffered */ 1838 uint16 length; /* length of the payload of this message */ 1839 struct rpcap_auth *rpauth; 1840 uint16 auth_type; 1841 struct rpcap_header header; 1842 size_t str_length; 1843 uint32 plen; 1844 struct rpcap_authreply authreply; /* authentication reply message */ 1845 uint8 ourvers; 1846 1847 if (auth) 1848 { 1849 switch (auth->type) 1850 { 1851 case RPCAP_RMTAUTH_NULL: 1852 length = sizeof(struct rpcap_auth); 1853 break; 1854 1855 case RPCAP_RMTAUTH_PWD: 1856 length = sizeof(struct rpcap_auth); 1857 if (auth->username) 1858 { 1859 str_length = strlen(auth->username); 1860 if (str_length > 65535) 1861 { 1862 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "User name is too long (> 65535 bytes)"); 1863 return -1; 1864 } 1865 length += (uint16)str_length; 1866 } 1867 if (auth->password) 1868 { 1869 str_length = strlen(auth->password); 1870 if (str_length > 65535) 1871 { 1872 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Password is too long (> 65535 bytes)"); 1873 return -1; 1874 } 1875 length += (uint16)str_length; 1876 } 1877 break; 1878 1879 default: 1880 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Authentication type not recognized."); 1881 return -1; 1882 } 1883 1884 auth_type = (uint16)auth->type; 1885 } 1886 else 1887 { 1888 auth_type = RPCAP_RMTAUTH_NULL; 1889 length = sizeof(struct rpcap_auth); 1890 } 1891 1892 if (sock_bufferize(NULL, sizeof(struct rpcap_header), NULL, 1893 &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, errbuf, PCAP_ERRBUF_SIZE)) 1894 return -1; 1895 1896 rpcap_createhdr((struct rpcap_header *) sendbuf, 0, 1897 RPCAP_MSG_AUTH_REQ, 0, length); 1898 1899 rpauth = (struct rpcap_auth *) &sendbuf[sendbufidx]; 1900 1901 if (sock_bufferize(NULL, sizeof(struct rpcap_auth), NULL, 1902 &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, errbuf, PCAP_ERRBUF_SIZE)) 1903 return -1; 1904 1905 memset(rpauth, 0, sizeof(struct rpcap_auth)); 1906 1907 rpauth->type = htons(auth_type); 1908 1909 if (auth_type == RPCAP_RMTAUTH_PWD) 1910 { 1911 if (auth->username) 1912 rpauth->slen1 = (uint16)strlen(auth->username); 1913 else 1914 rpauth->slen1 = 0; 1915 1916 if (sock_bufferize(auth->username, rpauth->slen1, sendbuf, 1917 &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_BUFFERIZE, errbuf, PCAP_ERRBUF_SIZE)) 1918 return -1; 1919 1920 if (auth->password) 1921 rpauth->slen2 = (uint16)strlen(auth->password); 1922 else 1923 rpauth->slen2 = 0; 1924 1925 if (sock_bufferize(auth->password, rpauth->slen2, sendbuf, 1926 &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_BUFFERIZE, errbuf, PCAP_ERRBUF_SIZE)) 1927 return -1; 1928 1929 rpauth->slen1 = htons(rpauth->slen1); 1930 rpauth->slen2 = htons(rpauth->slen2); 1931 } 1932 1933 if (sock_send(sockctrl, sendbuf, sendbufidx, errbuf, 1934 PCAP_ERRBUF_SIZE) < 0) 1935 return -1; 1936 1937 /* Receive and process the reply message header */ 1938 if (rpcap_process_msg_header(sockctrl, 0, RPCAP_MSG_AUTH_REQ, 1939 &header, errbuf) == -1) 1940 return -1; 1941 1942 /* 1943 * OK, it's an authentication reply, so we're logged in. 1944 * 1945 * Did it send any additional information? 1946 */ 1947 plen = header.plen; 1948 if (plen != 0) 1949 { 1950 /* Yes - is it big enough to be version information? */ 1951 if (plen < sizeof(struct rpcap_authreply)) 1952 { 1953 /* No - discard it and fail. */ 1954 (void)rpcap_discard(sockctrl, plen, NULL); 1955 return -1; 1956 } 1957 1958 /* Read the reply body */ 1959 if (rpcap_recv(sockctrl, (char *)&authreply, 1960 sizeof(struct rpcap_authreply), &plen, errbuf) == -1) 1961 { 1962 (void)rpcap_discard(sockctrl, plen, NULL); 1963 return -1; 1964 } 1965 1966 /* Discard the rest of the message, if there is any. */ 1967 if (rpcap_discard(sockctrl, plen, errbuf) == -1) 1968 return -1; 1969 1970 /* 1971 * Check the minimum and maximum versions for sanity; 1972 * the minimum must be <= the maximum. 1973 */ 1974 if (authreply.minvers > authreply.maxvers) 1975 { 1976 /* 1977 * Bogus - give up on this server. 1978 */ 1979 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, 1980 "The server's minimum supported protocol version is greater than its maximum supported protocol version"); 1981 return -1; 1982 } 1983 } 1984 else 1985 { 1986 /* No - it supports only version 0. */ 1987 authreply.minvers = 0; 1988 authreply.maxvers = 0; 1989 } 1990 1991 /* 1992 * OK, let's start with the maximum version the server supports. 1993 */ 1994 ourvers = authreply.maxvers; 1995 1996 #if RPCAP_MIN_VERSION != 0 1997 /* 1998 * If that's less than the minimum version we support, we 1999 * can't communicate. 2000 */ 2001 if (ourvers < RPCAP_MIN_VERSION) 2002 goto novers; 2003 #endif 2004 2005 /* 2006 * If that's greater than the maximum version we support, 2007 * choose the maximum version we support. 2008 */ 2009 if (ourvers > RPCAP_MAX_VERSION) 2010 { 2011 ourvers = RPCAP_MAX_VERSION; 2012 2013 /* 2014 * If that's less than the minimum version they 2015 * support, we can't communicate. 2016 */ 2017 if (ourvers < authreply.minvers) 2018 goto novers; 2019 } 2020 2021 *ver = ourvers; 2022 return 0; 2023 2024 novers: 2025 /* 2026 * There is no version we both support; that is a fatal error. 2027 */ 2028 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, 2029 "The server doesn't support any protocol version that we support"); 2030 return -1; 2031 } 2032 2033 /* We don't currently support non-blocking mode. */ 2034 static int 2035 pcap_getnonblock_rpcap(pcap_t *p) 2036 { 2037 pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 2038 "Non-blocking mode isn't supported for capturing remotely with rpcap"); 2039 return (-1); 2040 } 2041 2042 static int 2043 pcap_setnonblock_rpcap(pcap_t *p, int nonblock _U_) 2044 { 2045 pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 2046 "Non-blocking mode isn't supported for capturing remotely with rpcap"); 2047 return (-1); 2048 } 2049 2050 static int 2051 rpcap_setup_session(const char *source, struct pcap_rmtauth *auth, 2052 int *activep, SOCKET *sockctrlp, uint8 *protocol_versionp, 2053 char *host, char *port, char *iface, char *errbuf) 2054 { 2055 int type; 2056 struct activehosts *activeconn; /* active connection, if there is one */ 2057 int error; /* 1 if rpcap_remoteact_getsock got an error */ 2058 2059 /* 2060 * Determine the type of the source (NULL, file, local, remote). 2061 * You must have a valid source string even if we're in active mode, 2062 * because otherwise the call to the following function will fail. 2063 */ 2064 if (pcap_parsesrcstr(source, &type, host, port, iface, errbuf) == -1) 2065 return -1; 2066 2067 /* 2068 * It must be remote. 2069 */ 2070 if (type != PCAP_SRC_IFREMOTE) 2071 { 2072 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, 2073 "Non-remote interface passed to remote capture routine"); 2074 return -1; 2075 } 2076 2077 /* Warning: this call can be the first one called by the user. */ 2078 /* For this reason, we have to initialize the WinSock support. */ 2079 if (sock_init(errbuf, PCAP_ERRBUF_SIZE) == -1) 2080 return -1; 2081 2082 /* Check for active mode */ 2083 activeconn = rpcap_remoteact_getsock(host, &error, errbuf); 2084 if (activeconn != NULL) 2085 { 2086 *activep = 1; 2087 *sockctrlp = activeconn->sockctrl; 2088 *protocol_versionp = activeconn->protocol_version; 2089 } 2090 else 2091 { 2092 *activep = 0; 2093 struct addrinfo hints; /* temp variable needed to resolve hostnames into to socket representation */ 2094 struct addrinfo *addrinfo; /* temp variable needed to resolve hostnames into to socket representation */ 2095 2096 if (error) 2097 { 2098 /* 2099 * Call failed. 2100 */ 2101 return -1; 2102 } 2103 2104 /* 2105 * We're not in active mode; let's try to open a new 2106 * control connection. 2107 */ 2108 memset(&hints, 0, sizeof(struct addrinfo)); 2109 hints.ai_family = PF_UNSPEC; 2110 hints.ai_socktype = SOCK_STREAM; 2111 2112 if (port[0] == 0) 2113 { 2114 /* the user chose not to specify the port */ 2115 if (sock_initaddress(host, RPCAP_DEFAULT_NETPORT, 2116 &hints, &addrinfo, errbuf, PCAP_ERRBUF_SIZE) == -1) 2117 return -1; 2118 } 2119 else 2120 { 2121 if (sock_initaddress(host, port, &hints, &addrinfo, 2122 errbuf, PCAP_ERRBUF_SIZE) == -1) 2123 return -1; 2124 } 2125 2126 if ((*sockctrlp = sock_open(addrinfo, SOCKOPEN_CLIENT, 0, 2127 errbuf, PCAP_ERRBUF_SIZE)) == INVALID_SOCKET) 2128 { 2129 freeaddrinfo(addrinfo); 2130 return -1; 2131 } 2132 2133 /* addrinfo is no longer used */ 2134 freeaddrinfo(addrinfo); 2135 addrinfo = NULL; 2136 2137 if (rpcap_doauth(*sockctrlp, protocol_versionp, auth, 2138 errbuf) == -1) 2139 { 2140 sock_close(*sockctrlp, NULL, 0); 2141 return -1; 2142 } 2143 } 2144 return 0; 2145 } 2146 2147 /* 2148 * This function opens a remote adapter by opening an RPCAP connection and 2149 * so on. 2150 * 2151 * It does the job of pcap_open_live() for a remote interface; it's called 2152 * by pcap_open() for remote interfaces. 2153 * 2154 * We do not start the capture until pcap_startcapture_remote() is called. 2155 * 2156 * This is because, when doing a remote capture, we cannot start capturing 2157 * data as soon as the 'open adapter' command is sent. Suppose the remote 2158 * adapter is already overloaded; if we start a capture (which, by default, 2159 * has a NULL filter) the new traffic can saturate the network. 2160 * 2161 * Instead, we want to "open" the adapter, then send a "start capture" 2162 * command only when we're ready to start the capture. 2163 * This function does this job: it sends an "open adapter" command 2164 * (according to the RPCAP protocol), but it does not start the capture. 2165 * 2166 * Since the other libpcap functions do not share this way of life, we 2167 * have to do some dirty things in order to make everything work. 2168 * 2169 * \param source: see pcap_open(). 2170 * \param snaplen: see pcap_open(). 2171 * \param flags: see pcap_open(). 2172 * \param read_timeout: see pcap_open(). 2173 * \param auth: see pcap_open(). 2174 * \param errbuf: see pcap_open(). 2175 * 2176 * \return a pcap_t pointer in case of success, NULL otherwise. In case of 2177 * success, the pcap_t pointer can be used as a parameter to the following 2178 * calls (pcap_compile() and so on). In case of problems, errbuf contains 2179 * a text explanation of error. 2180 * 2181 * WARNING: In case we call pcap_compile() and the capture has not yet 2182 * been started, the filter will be saved into the pcap_t structure, 2183 * and it will be sent to the other host later (when 2184 * pcap_startcapture_remote() is called). 2185 */ 2186 pcap_t *pcap_open_rpcap(const char *source, int snaplen, int flags, int read_timeout, struct pcap_rmtauth *auth, char *errbuf) 2187 { 2188 pcap_t *fp; 2189 char *source_str; 2190 struct pcap_rpcap *pr; /* structure used when doing a remote live capture */ 2191 char host[PCAP_BUF_SIZE], ctrlport[PCAP_BUF_SIZE], iface[PCAP_BUF_SIZE]; 2192 SOCKET sockctrl; 2193 uint8 protocol_version; /* negotiated protocol version */ 2194 int active; 2195 uint32 plen; 2196 char sendbuf[RPCAP_NETBUF_SIZE]; /* temporary buffer in which data to be sent is buffered */ 2197 int sendbufidx = 0; /* index which keeps the number of bytes currently buffered */ 2198 2199 /* RPCAP-related variables */ 2200 struct rpcap_header header; /* header of the RPCAP packet */ 2201 struct rpcap_openreply openreply; /* open reply message */ 2202 2203 fp = pcap_create_common(errbuf, sizeof (struct pcap_rpcap)); 2204 if (fp == NULL) 2205 { 2206 return NULL; 2207 } 2208 source_str = strdup(source); 2209 if (source_str == NULL) { 2210 pcap_fmt_errmsg_for_errno(errbuf, PCAP_ERRBUF_SIZE, 2211 errno, "malloc"); 2212 return NULL; 2213 } 2214 2215 /* 2216 * Turn a negative snapshot value (invalid), a snapshot value of 2217 * 0 (unspecified), or a value bigger than the normal maximum 2218 * value, into the maximum allowed value. 2219 * 2220 * If some application really *needs* a bigger snapshot 2221 * length, we should just increase MAXIMUM_SNAPLEN. 2222 * 2223 * XXX - should we leave this up to the remote server to 2224 * do? 2225 */ 2226 if (snaplen <= 0 || snaplen > MAXIMUM_SNAPLEN) 2227 snaplen = MAXIMUM_SNAPLEN; 2228 2229 fp->opt.device = source_str; 2230 fp->snapshot = snaplen; 2231 fp->opt.timeout = read_timeout; 2232 pr = fp->priv; 2233 pr->rmt_flags = flags; 2234 2235 /* 2236 * Attempt to set up the session with the server. 2237 */ 2238 if (rpcap_setup_session(fp->opt.device, auth, &active, &sockctrl, 2239 &protocol_version, host, ctrlport, iface, errbuf) == -1) 2240 { 2241 /* Session setup failed. */ 2242 pcap_close(fp); 2243 return NULL; 2244 } 2245 2246 /* 2247 * Now it's time to start playing with the RPCAP protocol 2248 * RPCAP open command: create the request message 2249 */ 2250 if (sock_bufferize(NULL, sizeof(struct rpcap_header), NULL, 2251 &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, errbuf, PCAP_ERRBUF_SIZE)) 2252 goto error_nodiscard; 2253 2254 rpcap_createhdr((struct rpcap_header *) sendbuf, protocol_version, 2255 RPCAP_MSG_OPEN_REQ, 0, (uint32) strlen(iface)); 2256 2257 if (sock_bufferize(iface, (int) strlen(iface), sendbuf, &sendbufidx, 2258 RPCAP_NETBUF_SIZE, SOCKBUF_BUFFERIZE, errbuf, PCAP_ERRBUF_SIZE)) 2259 goto error_nodiscard; 2260 2261 if (sock_send(sockctrl, sendbuf, sendbufidx, errbuf, 2262 PCAP_ERRBUF_SIZE) < 0) 2263 goto error_nodiscard; 2264 2265 /* Receive and process the reply message header. */ 2266 if (rpcap_process_msg_header(sockctrl, protocol_version, 2267 RPCAP_MSG_OPEN_REQ, &header, errbuf) == -1) 2268 goto error_nodiscard; 2269 plen = header.plen; 2270 2271 /* Read the reply body */ 2272 if (rpcap_recv(sockctrl, (char *)&openreply, 2273 sizeof(struct rpcap_openreply), &plen, errbuf) == -1) 2274 goto error; 2275 2276 /* Discard the rest of the message, if there is any. */ 2277 if (rpcap_discard(sockctrl, plen, errbuf) == -1) 2278 goto error_nodiscard; 2279 2280 /* Set proper fields into the pcap_t struct */ 2281 fp->linktype = ntohl(openreply.linktype); 2282 fp->tzoff = ntohl(openreply.tzoff); 2283 pr->rmt_sockctrl = sockctrl; 2284 pr->protocol_version = protocol_version; 2285 pr->rmt_clientside = 1; 2286 2287 /* This code is duplicated from the end of this function */ 2288 fp->read_op = pcap_read_rpcap; 2289 fp->save_current_filter_op = pcap_save_current_filter_rpcap; 2290 fp->setfilter_op = pcap_setfilter_rpcap; 2291 fp->getnonblock_op = pcap_getnonblock_rpcap; 2292 fp->setnonblock_op = pcap_setnonblock_rpcap; 2293 fp->stats_op = pcap_stats_rpcap; 2294 #ifdef _WIN32 2295 fp->stats_ex_op = pcap_stats_ex_rpcap; 2296 #endif 2297 fp->cleanup_op = pcap_cleanup_rpcap; 2298 2299 fp->activated = 1; 2300 return fp; 2301 2302 error: 2303 /* 2304 * When the connection has been established, we have to close it. So, at the 2305 * beginning of this function, if an error occur we return immediately with 2306 * a return NULL; when the connection is established, we have to come here 2307 * ('goto error;') in order to close everything properly. 2308 */ 2309 2310 /* 2311 * Discard the rest of the message. 2312 * We already reported an error; if this gets an error, just 2313 * drive on. 2314 */ 2315 (void)rpcap_discard(sockctrl, plen, NULL); 2316 2317 error_nodiscard: 2318 if (!active) 2319 sock_close(sockctrl, NULL, 0); 2320 2321 pcap_close(fp); 2322 return NULL; 2323 } 2324 2325 /* String identifier to be used in the pcap_findalldevs_ex() */ 2326 #define PCAP_TEXT_SOURCE_ADAPTER "Network adapter" 2327 #define PCAP_TEXT_SOURCE_ADAPTER_LEN (sizeof PCAP_TEXT_SOURCE_ADAPTER - 1) 2328 /* String identifier to be used in the pcap_findalldevs_ex() */ 2329 #define PCAP_TEXT_SOURCE_ON_REMOTE_HOST "on remote node" 2330 #define PCAP_TEXT_SOURCE_ON_REMOTE_HOST_LEN (sizeof PCAP_TEXT_SOURCE_ON_REMOTE_HOST - 1) 2331 2332 static void 2333 freeaddr(struct pcap_addr *addr) 2334 { 2335 free(addr->addr); 2336 free(addr->netmask); 2337 free(addr->broadaddr); 2338 free(addr->dstaddr); 2339 free(addr); 2340 } 2341 2342 int 2343 pcap_findalldevs_ex_remote(const char *source, struct pcap_rmtauth *auth, pcap_if_t **alldevs, char *errbuf) 2344 { 2345 uint8 protocol_version; /* protocol version */ 2346 SOCKET sockctrl; /* socket descriptor of the control connection */ 2347 uint32 plen; 2348 struct rpcap_header header; /* structure that keeps the general header of the rpcap protocol */ 2349 int i, j; /* temp variables */ 2350 int nif; /* Number of interfaces listed */ 2351 int active; /* 'true' if we the other end-party is in active mode */ 2352 char host[PCAP_BUF_SIZE], port[PCAP_BUF_SIZE]; 2353 char tmpstring[PCAP_BUF_SIZE + 1]; /* Needed to convert names and descriptions from 'old' syntax to the 'new' one */ 2354 pcap_if_t *lastdev; /* Last device in the pcap_if_t list */ 2355 pcap_if_t *dev; /* Device we're adding to the pcap_if_t list */ 2356 2357 /* List starts out empty. */ 2358 (*alldevs) = NULL; 2359 lastdev = NULL; 2360 2361 /* 2362 * Attempt to set up the session with the server. 2363 */ 2364 if (rpcap_setup_session(source, auth, &active, &sockctrl, 2365 &protocol_version, host, port, NULL, errbuf) == -1) 2366 { 2367 /* Session setup failed. */ 2368 return -1; 2369 } 2370 2371 /* RPCAP findalldevs command */ 2372 rpcap_createhdr(&header, protocol_version, RPCAP_MSG_FINDALLIF_REQ, 2373 0, 0); 2374 2375 if (sock_send(sockctrl, (char *)&header, sizeof(struct rpcap_header), 2376 errbuf, PCAP_ERRBUF_SIZE) < 0) 2377 goto error_nodiscard; 2378 2379 /* Receive and process the reply message header. */ 2380 if (rpcap_process_msg_header(sockctrl, protocol_version, 2381 RPCAP_MSG_FINDALLIF_REQ, &header, errbuf) == -1) 2382 goto error_nodiscard; 2383 2384 plen = header.plen; 2385 2386 /* read the number of interfaces */ 2387 nif = ntohs(header.value); 2388 2389 /* loop until all interfaces have been received */ 2390 for (i = 0; i < nif; i++) 2391 { 2392 struct rpcap_findalldevs_if findalldevs_if; 2393 char tmpstring2[PCAP_BUF_SIZE + 1]; /* Needed to convert names and descriptions from 'old' syntax to the 'new' one */ 2394 struct pcap_addr *addr, *prevaddr; 2395 2396 tmpstring2[PCAP_BUF_SIZE] = 0; 2397 2398 /* receive the findalldevs structure from remote host */ 2399 if (rpcap_recv(sockctrl, (char *)&findalldevs_if, 2400 sizeof(struct rpcap_findalldevs_if), &plen, errbuf) == -1) 2401 goto error; 2402 2403 findalldevs_if.namelen = ntohs(findalldevs_if.namelen); 2404 findalldevs_if.desclen = ntohs(findalldevs_if.desclen); 2405 findalldevs_if.naddr = ntohs(findalldevs_if.naddr); 2406 2407 /* allocate the main structure */ 2408 dev = (pcap_if_t *)malloc(sizeof(pcap_if_t)); 2409 if (dev == NULL) 2410 { 2411 pcap_fmt_errmsg_for_errno(errbuf, PCAP_ERRBUF_SIZE, 2412 errno, "malloc() failed"); 2413 goto error; 2414 } 2415 2416 /* Initialize the structure to 'zero' */ 2417 memset(dev, 0, sizeof(pcap_if_t)); 2418 2419 /* Append it to the list. */ 2420 if (lastdev == NULL) 2421 { 2422 /* 2423 * List is empty, so it's also the first device. 2424 */ 2425 *alldevs = dev; 2426 } 2427 else 2428 { 2429 /* 2430 * Append after the last device. 2431 */ 2432 lastdev->next = dev; 2433 } 2434 /* It's now the last device. */ 2435 lastdev = dev; 2436 2437 /* allocate mem for name and description */ 2438 if (findalldevs_if.namelen) 2439 { 2440 2441 if (findalldevs_if.namelen >= sizeof(tmpstring)) 2442 { 2443 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Interface name too long"); 2444 goto error; 2445 } 2446 2447 /* Retrieve adapter name */ 2448 if (rpcap_recv(sockctrl, tmpstring, 2449 findalldevs_if.namelen, &plen, errbuf) == -1) 2450 goto error; 2451 2452 tmpstring[findalldevs_if.namelen] = 0; 2453 2454 /* Create the new device identifier */ 2455 if (pcap_createsrcstr(tmpstring2, PCAP_SRC_IFREMOTE, 2456 host, port, tmpstring, errbuf) == -1) 2457 goto error; 2458 2459 dev->name = strdup(tmpstring2); 2460 if (dev->name == NULL) 2461 { 2462 pcap_fmt_errmsg_for_errno(errbuf, 2463 PCAP_ERRBUF_SIZE, errno, "malloc() failed"); 2464 goto error; 2465 } 2466 } 2467 2468 if (findalldevs_if.desclen) 2469 { 2470 if (findalldevs_if.desclen >= sizeof(tmpstring)) 2471 { 2472 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Interface description too long"); 2473 goto error; 2474 } 2475 2476 /* Retrieve adapter description */ 2477 if (rpcap_recv(sockctrl, tmpstring, 2478 findalldevs_if.desclen, &plen, errbuf) == -1) 2479 goto error; 2480 2481 tmpstring[findalldevs_if.desclen] = 0; 2482 2483 if (pcap_asprintf(&dev->description, 2484 "%s '%s' %s %s", PCAP_TEXT_SOURCE_ADAPTER, 2485 tmpstring, PCAP_TEXT_SOURCE_ON_REMOTE_HOST, host) == -1) 2486 { 2487 pcap_fmt_errmsg_for_errno(errbuf, 2488 PCAP_ERRBUF_SIZE, errno, "malloc() failed"); 2489 goto error; 2490 } 2491 } 2492 2493 dev->flags = ntohl(findalldevs_if.flags); 2494 2495 prevaddr = NULL; 2496 /* loop until all addresses have been received */ 2497 for (j = 0; j < findalldevs_if.naddr; j++) 2498 { 2499 struct rpcap_findalldevs_ifaddr ifaddr; 2500 2501 /* Retrieve the interface addresses */ 2502 if (rpcap_recv(sockctrl, (char *)&ifaddr, 2503 sizeof(struct rpcap_findalldevs_ifaddr), 2504 &plen, errbuf) == -1) 2505 goto error; 2506 2507 /* 2508 * Deserialize all the address components. 2509 */ 2510 addr = (struct pcap_addr *) malloc(sizeof(struct pcap_addr)); 2511 if (addr == NULL) 2512 { 2513 pcap_fmt_errmsg_for_errno(errbuf, 2514 PCAP_ERRBUF_SIZE, errno, "malloc() failed"); 2515 goto error; 2516 } 2517 addr->next = NULL; 2518 addr->addr = NULL; 2519 addr->netmask = NULL; 2520 addr->broadaddr = NULL; 2521 addr->dstaddr = NULL; 2522 2523 if (rpcap_deseraddr(&ifaddr.addr, 2524 (struct sockaddr_storage **) &addr->addr, errbuf) == -1) 2525 { 2526 freeaddr(addr); 2527 goto error; 2528 } 2529 if (rpcap_deseraddr(&ifaddr.netmask, 2530 (struct sockaddr_storage **) &addr->netmask, errbuf) == -1) 2531 { 2532 freeaddr(addr); 2533 goto error; 2534 } 2535 if (rpcap_deseraddr(&ifaddr.broadaddr, 2536 (struct sockaddr_storage **) &addr->broadaddr, errbuf) == -1) 2537 { 2538 freeaddr(addr); 2539 goto error; 2540 } 2541 if (rpcap_deseraddr(&ifaddr.dstaddr, 2542 (struct sockaddr_storage **) &addr->dstaddr, errbuf) == -1) 2543 { 2544 freeaddr(addr); 2545 goto error; 2546 } 2547 2548 if ((addr->addr == NULL) && (addr->netmask == NULL) && 2549 (addr->broadaddr == NULL) && (addr->dstaddr == NULL)) 2550 { 2551 /* 2552 * None of the addresses are IPv4 or IPv6 2553 * addresses, so throw this entry away. 2554 */ 2555 free(addr); 2556 } 2557 else 2558 { 2559 /* 2560 * Add this entry to the list. 2561 */ 2562 if (prevaddr == NULL) 2563 { 2564 dev->addresses = addr; 2565 } 2566 else 2567 { 2568 prevaddr->next = addr; 2569 } 2570 prevaddr = addr; 2571 } 2572 } 2573 } 2574 2575 /* Discard the rest of the message. */ 2576 if (rpcap_discard(sockctrl, plen, errbuf) == 1) 2577 goto error_nodiscard; 2578 2579 /* Control connection has to be closed only in case the remote machine is in passive mode */ 2580 if (!active) 2581 { 2582 /* DO not send RPCAP_CLOSE, since we did not open a pcap_t; no need to free resources */ 2583 if (sock_close(sockctrl, errbuf, PCAP_ERRBUF_SIZE)) 2584 return -1; 2585 } 2586 2587 /* To avoid inconsistencies in the number of sock_init() */ 2588 sock_cleanup(); 2589 2590 return 0; 2591 2592 error: 2593 /* 2594 * In case there has been an error, I don't want to overwrite it with a new one 2595 * if the following call fails. I want to return always the original error. 2596 * 2597 * Take care: this connection can already be closed when we try to close it. 2598 * This happens because a previous error in the rpcapd, which requested to 2599 * closed the connection. In that case, we already recognized that into the 2600 * rpspck_isheaderok() and we already acknowledged the closing. 2601 * In that sense, this call is useless here (however it is needed in case 2602 * the client generates the error). 2603 * 2604 * Checks if all the data has been read; if not, discard the data in excess 2605 */ 2606 (void) rpcap_discard(sockctrl, plen, NULL); 2607 2608 error_nodiscard: 2609 /* Control connection has to be closed only in case the remote machine is in passive mode */ 2610 if (!active) 2611 sock_close(sockctrl, NULL, 0); 2612 2613 /* To avoid inconsistencies in the number of sock_init() */ 2614 sock_cleanup(); 2615 2616 /* Free whatever interfaces we've allocated. */ 2617 pcap_freealldevs(*alldevs); 2618 2619 return -1; 2620 } 2621 2622 /* 2623 * Active mode routines. 2624 * 2625 * The old libpcap API is somewhat ugly, and makes active mode difficult 2626 * to implement; we provide some APIs for it that work only with rpcap. 2627 */ 2628 2629 SOCKET pcap_remoteact_accept(const char *address, const char *port, const char *hostlist, char *connectinghost, struct pcap_rmtauth *auth, char *errbuf) 2630 { 2631 /* socket-related variables */ 2632 struct addrinfo hints; /* temporary struct to keep settings needed to open the new socket */ 2633 struct addrinfo *addrinfo; /* keeps the addrinfo chain; required to open a new socket */ 2634 struct sockaddr_storage from; /* generic sockaddr_storage variable */ 2635 socklen_t fromlen; /* keeps the length of the sockaddr_storage variable */ 2636 SOCKET sockctrl; /* keeps the main socket identifier */ 2637 uint8 protocol_version; /* negotiated protocol version */ 2638 struct activehosts *temp, *prev; /* temp var needed to scan he host list chain */ 2639 2640 *connectinghost = 0; /* just in case */ 2641 2642 /* Prepare to open a new server socket */ 2643 memset(&hints, 0, sizeof(struct addrinfo)); 2644 /* WARNING Currently it supports only ONE socket family among ipv4 and IPv6 */ 2645 hints.ai_family = AF_INET; /* PF_UNSPEC to have both IPv4 and IPv6 server */ 2646 hints.ai_flags = AI_PASSIVE; /* Ready to a bind() socket */ 2647 hints.ai_socktype = SOCK_STREAM; 2648 2649 /* Warning: this call can be the first one called by the user. */ 2650 /* For this reason, we have to initialize the WinSock support. */ 2651 if (sock_init(errbuf, PCAP_ERRBUF_SIZE) == -1) 2652 return (SOCKET)-1; 2653 2654 /* Do the work */ 2655 if ((port == NULL) || (port[0] == 0)) 2656 { 2657 if (sock_initaddress(address, RPCAP_DEFAULT_NETPORT_ACTIVE, &hints, &addrinfo, errbuf, PCAP_ERRBUF_SIZE) == -1) 2658 { 2659 return (SOCKET)-2; 2660 } 2661 } 2662 else 2663 { 2664 if (sock_initaddress(address, port, &hints, &addrinfo, errbuf, PCAP_ERRBUF_SIZE) == -1) 2665 { 2666 return (SOCKET)-2; 2667 } 2668 } 2669 2670 2671 if ((sockmain = sock_open(addrinfo, SOCKOPEN_SERVER, 1, errbuf, PCAP_ERRBUF_SIZE)) == INVALID_SOCKET) 2672 { 2673 freeaddrinfo(addrinfo); 2674 return (SOCKET)-2; 2675 } 2676 freeaddrinfo(addrinfo); 2677 2678 /* Connection creation */ 2679 fromlen = sizeof(struct sockaddr_storage); 2680 2681 sockctrl = accept(sockmain, (struct sockaddr *) &from, &fromlen); 2682 2683 /* We're not using sock_close, since we do not want to send a shutdown */ 2684 /* (which is not allowed on a non-connected socket) */ 2685 closesocket(sockmain); 2686 sockmain = 0; 2687 2688 if (sockctrl == INVALID_SOCKET) 2689 { 2690 sock_geterror("accept()", errbuf, PCAP_ERRBUF_SIZE); 2691 return (SOCKET)-2; 2692 } 2693 2694 /* Get the numeric for of the name of the connecting host */ 2695 if (getnameinfo((struct sockaddr *) &from, fromlen, connectinghost, RPCAP_HOSTLIST_SIZE, NULL, 0, NI_NUMERICHOST)) 2696 { 2697 sock_geterror("getnameinfo()", errbuf, PCAP_ERRBUF_SIZE); 2698 rpcap_senderror(sockctrl, 0, PCAP_ERR_REMOTEACCEPT, errbuf, NULL); 2699 sock_close(sockctrl, NULL, 0); 2700 return (SOCKET)-1; 2701 } 2702 2703 /* checks if the connecting host is among the ones allowed */ 2704 if (sock_check_hostlist((char *)hostlist, RPCAP_HOSTLIST_SEP, &from, errbuf, PCAP_ERRBUF_SIZE) < 0) 2705 { 2706 rpcap_senderror(sockctrl, 0, PCAP_ERR_REMOTEACCEPT, errbuf, NULL); 2707 sock_close(sockctrl, NULL, 0); 2708 return (SOCKET)-1; 2709 } 2710 2711 /* 2712 * Send authentication to the remote machine. 2713 */ 2714 if (rpcap_doauth(sockctrl, &protocol_version, auth, errbuf) == -1) 2715 { 2716 /* Unrecoverable error. */ 2717 rpcap_senderror(sockctrl, 0, PCAP_ERR_REMOTEACCEPT, errbuf, NULL); 2718 sock_close(sockctrl, NULL, 0); 2719 return (SOCKET)-3; 2720 } 2721 2722 /* Checks that this host does not already have a cntrl connection in place */ 2723 2724 /* Initialize pointers */ 2725 temp = activeHosts; 2726 prev = NULL; 2727 2728 while (temp) 2729 { 2730 /* This host already has an active connection in place, so I don't have to update the host list */ 2731 if (sock_cmpaddr(&temp->host, &from) == 0) 2732 return sockctrl; 2733 2734 prev = temp; 2735 temp = temp->next; 2736 } 2737 2738 /* The host does not exist in the list; so I have to update the list */ 2739 if (prev) 2740 { 2741 prev->next = (struct activehosts *) malloc(sizeof(struct activehosts)); 2742 temp = prev->next; 2743 } 2744 else 2745 { 2746 activeHosts = (struct activehosts *) malloc(sizeof(struct activehosts)); 2747 temp = activeHosts; 2748 } 2749 2750 if (temp == NULL) 2751 { 2752 pcap_fmt_errmsg_for_errno(errbuf, PCAP_ERRBUF_SIZE, 2753 errno, "malloc() failed"); 2754 rpcap_senderror(sockctrl, protocol_version, PCAP_ERR_REMOTEACCEPT, errbuf, NULL); 2755 sock_close(sockctrl, NULL, 0); 2756 return (SOCKET)-1; 2757 } 2758 2759 memcpy(&temp->host, &from, fromlen); 2760 temp->sockctrl = sockctrl; 2761 temp->protocol_version = protocol_version; 2762 temp->next = NULL; 2763 2764 return sockctrl; 2765 } 2766 2767 int pcap_remoteact_close(const char *host, char *errbuf) 2768 { 2769 struct activehosts *temp, *prev; /* temp var needed to scan the host list chain */ 2770 struct addrinfo hints, *addrinfo, *ai_next; /* temp var needed to translate between hostname to its address */ 2771 int retval; 2772 2773 temp = activeHosts; 2774 prev = NULL; 2775 2776 /* retrieve the network address corresponding to 'host' */ 2777 addrinfo = NULL; 2778 memset(&hints, 0, sizeof(struct addrinfo)); 2779 hints.ai_family = PF_UNSPEC; 2780 hints.ai_socktype = SOCK_STREAM; 2781 2782 retval = getaddrinfo(host, "0", &hints, &addrinfo); 2783 if (retval != 0) 2784 { 2785 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "getaddrinfo() %s", gai_strerror(retval)); 2786 return -1; 2787 } 2788 2789 while (temp) 2790 { 2791 ai_next = addrinfo; 2792 while (ai_next) 2793 { 2794 if (sock_cmpaddr(&temp->host, (struct sockaddr_storage *) ai_next->ai_addr) == 0) 2795 { 2796 struct rpcap_header header; 2797 int status = 0; 2798 2799 /* Close this connection */ 2800 rpcap_createhdr(&header, temp->protocol_version, 2801 RPCAP_MSG_CLOSE, 0, 0); 2802 2803 /* 2804 * Don't check for errors, since we're 2805 * just cleaning up. 2806 */ 2807 if (sock_send(temp->sockctrl, 2808 (char *)&header, 2809 sizeof(struct rpcap_header), errbuf, 2810 PCAP_ERRBUF_SIZE) < 0) 2811 { 2812 /* 2813 * Let that error be the one we 2814 * report. 2815 */ 2816 (void)sock_close(temp->sockctrl, NULL, 2817 0); 2818 status = -1; 2819 } 2820 else 2821 { 2822 if (sock_close(temp->sockctrl, errbuf, 2823 PCAP_ERRBUF_SIZE) == -1) 2824 status = -1; 2825 } 2826 2827 /* 2828 * Remove the host from the list of active 2829 * hosts. 2830 */ 2831 if (prev) 2832 prev->next = temp->next; 2833 else 2834 activeHosts = temp->next; 2835 2836 freeaddrinfo(addrinfo); 2837 2838 free(temp); 2839 2840 /* To avoid inconsistencies in the number of sock_init() */ 2841 sock_cleanup(); 2842 2843 return status; 2844 } 2845 2846 ai_next = ai_next->ai_next; 2847 } 2848 prev = temp; 2849 temp = temp->next; 2850 } 2851 2852 if (addrinfo) 2853 freeaddrinfo(addrinfo); 2854 2855 /* To avoid inconsistencies in the number of sock_init() */ 2856 sock_cleanup(); 2857 2858 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "The host you want to close the active connection is not known"); 2859 return -1; 2860 } 2861 2862 void pcap_remoteact_cleanup(void) 2863 { 2864 /* Very dirty, but it works */ 2865 if (sockmain) 2866 { 2867 closesocket(sockmain); 2868 2869 /* To avoid inconsistencies in the number of sock_init() */ 2870 sock_cleanup(); 2871 } 2872 2873 } 2874 2875 int pcap_remoteact_list(char *hostlist, char sep, int size, char *errbuf) 2876 { 2877 struct activehosts *temp; /* temp var needed to scan the host list chain */ 2878 size_t len; 2879 char hoststr[RPCAP_HOSTLIST_SIZE + 1]; 2880 2881 temp = activeHosts; 2882 2883 len = 0; 2884 *hostlist = 0; 2885 2886 while (temp) 2887 { 2888 /*int sock_getascii_addrport(const struct sockaddr_storage *sockaddr, char *address, int addrlen, char *port, int portlen, int flags, char *errbuf, int errbuflen) */ 2889 2890 /* Get the numeric form of the name of the connecting host */ 2891 if (sock_getascii_addrport((struct sockaddr_storage *) &temp->host, hoststr, 2892 RPCAP_HOSTLIST_SIZE, NULL, 0, NI_NUMERICHOST, errbuf, PCAP_ERRBUF_SIZE) != -1) 2893 /* if (getnameinfo( (struct sockaddr *) &temp->host, sizeof (struct sockaddr_storage), hoststr, */ 2894 /* RPCAP_HOSTLIST_SIZE, NULL, 0, NI_NUMERICHOST) ) */ 2895 { 2896 /* sock_geterror("getnameinfo()", errbuf, PCAP_ERRBUF_SIZE); */ 2897 return -1; 2898 } 2899 2900 len = len + strlen(hoststr) + 1 /* the separator */; 2901 2902 if ((size < 0) || (len >= (size_t)size)) 2903 { 2904 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "The string you provided is not able to keep " 2905 "the hostnames for all the active connections"); 2906 return -1; 2907 } 2908 2909 pcap_strlcat(hostlist, hoststr, PCAP_ERRBUF_SIZE); 2910 hostlist[len - 1] = sep; 2911 hostlist[len] = 0; 2912 2913 temp = temp->next; 2914 } 2915 2916 return 0; 2917 } 2918 2919 /* 2920 * Receive the header of a message. 2921 */ 2922 static int rpcap_recv_msg_header(SOCKET sock, struct rpcap_header *header, char *errbuf) 2923 { 2924 int nrecv; 2925 2926 nrecv = sock_recv(sock, (char *) header, sizeof(struct rpcap_header), 2927 SOCK_RECEIVEALL_YES|SOCK_EOF_IS_ERROR, errbuf, 2928 PCAP_ERRBUF_SIZE); 2929 if (nrecv == -1) 2930 { 2931 /* Network error. */ 2932 return -1; 2933 } 2934 header->plen = ntohl(header->plen); 2935 return 0; 2936 } 2937 2938 /* 2939 * Make sure the protocol version of a received message is what we were 2940 * expecting. 2941 */ 2942 static int rpcap_check_msg_ver(SOCKET sock, uint8 expected_ver, struct rpcap_header *header, char *errbuf) 2943 { 2944 /* 2945 * Did the server specify the version we negotiated? 2946 */ 2947 if (header->ver != expected_ver) 2948 { 2949 /* 2950 * Discard the rest of the message. 2951 */ 2952 if (rpcap_discard(sock, header->plen, errbuf) == -1) 2953 return -1; 2954 2955 /* 2956 * Tell our caller that it's not the negotiated version. 2957 */ 2958 if (errbuf != NULL) 2959 { 2960 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, 2961 "Server sent us a message with version %u when we were expecting %u", 2962 header->ver, expected_ver); 2963 } 2964 return -1; 2965 } 2966 return 0; 2967 } 2968 2969 /* 2970 * Check the message type of a received message, which should either be 2971 * the expected message type or RPCAP_MSG_ERROR. 2972 */ 2973 static int rpcap_check_msg_type(SOCKET sock, uint8 request_type, struct rpcap_header *header, uint16 *errcode, char *errbuf) 2974 { 2975 const char *request_type_string; 2976 const char *msg_type_string; 2977 2978 /* 2979 * What type of message is it? 2980 */ 2981 if (header->type == RPCAP_MSG_ERROR) 2982 { 2983 /* 2984 * The server reported an error. 2985 * Hand that error back to our caller. 2986 */ 2987 *errcode = ntohs(header->value); 2988 rpcap_msg_err(sock, header->plen, errbuf); 2989 return -1; 2990 } 2991 2992 *errcode = 0; 2993 2994 /* 2995 * For a given request type value, the expected reply type value 2996 * is the request type value with ORed with RPCAP_MSG_IS_REPLY. 2997 */ 2998 if (header->type != (request_type | RPCAP_MSG_IS_REPLY)) 2999 { 3000 /* 3001 * This isn't a reply to the request we sent. 3002 */ 3003 3004 /* 3005 * Discard the rest of the message. 3006 */ 3007 if (rpcap_discard(sock, header->plen, errbuf) == -1) 3008 return -1; 3009 3010 /* 3011 * Tell our caller about it. 3012 */ 3013 request_type_string = rpcap_msg_type_string(request_type); 3014 msg_type_string = rpcap_msg_type_string(header->type); 3015 if (errbuf != NULL) 3016 { 3017 if (request_type_string == NULL) 3018 { 3019 /* This should not happen. */ 3020 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, 3021 "rpcap_check_msg_type called for request message with type %u", 3022 request_type); 3023 return -1; 3024 } 3025 if (msg_type_string != NULL) 3026 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, 3027 "%s message received in response to a %s message", 3028 msg_type_string, request_type_string); 3029 else 3030 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, 3031 "Message of unknown type %u message received in response to a %s request", 3032 header->type, request_type_string); 3033 } 3034 return -1; 3035 } 3036 3037 return 0; 3038 } 3039 3040 /* 3041 * Receive and process the header of a message. 3042 */ 3043 static int rpcap_process_msg_header(SOCKET sock, uint8 expected_ver, uint8 request_type, struct rpcap_header *header, char *errbuf) 3044 { 3045 uint16 errcode; 3046 3047 if (rpcap_recv_msg_header(sock, header, errbuf) == -1) 3048 { 3049 /* Network error. */ 3050 return -1; 3051 } 3052 3053 /* 3054 * Did the server specify the version we negotiated? 3055 */ 3056 if (rpcap_check_msg_ver(sock, expected_ver, header, errbuf) == -1) 3057 return -1; 3058 3059 /* 3060 * Check the message type. 3061 */ 3062 return rpcap_check_msg_type(sock, request_type, header, 3063 &errcode, errbuf); 3064 } 3065 3066 /* 3067 * Read data from a message. 3068 * If we're trying to read more data that remains, puts an error 3069 * message into errmsgbuf and returns -2. Otherwise, tries to read 3070 * the data and, if that succeeds, subtracts the amount read from 3071 * the number of bytes of data that remains. 3072 * Returns 0 on success, logs a message and returns -1 on a network 3073 * error. 3074 */ 3075 static int rpcap_recv(SOCKET sock, void *buffer, size_t toread, uint32 *plen, char *errbuf) 3076 { 3077 int nread; 3078 3079 if (toread > *plen) 3080 { 3081 /* The server sent us a bad message */ 3082 pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Message payload is too short"); 3083 return -1; 3084 } 3085 nread = sock_recv(sock, buffer, toread, 3086 SOCK_RECEIVEALL_YES|SOCK_EOF_IS_ERROR, errbuf, PCAP_ERRBUF_SIZE); 3087 if (nread == -1) 3088 { 3089 return -1; 3090 } 3091 *plen -= nread; 3092 return 0; 3093 } 3094 3095 /* 3096 * This handles the RPCAP_MSG_ERROR message. 3097 */ 3098 static void rpcap_msg_err(SOCKET sockctrl, uint32 plen, char *remote_errbuf) 3099 { 3100 char errbuf[PCAP_ERRBUF_SIZE]; 3101 3102 if (plen >= PCAP_ERRBUF_SIZE) 3103 { 3104 /* 3105 * Message is too long; just read as much of it as we 3106 * can into the buffer provided, and discard the rest. 3107 */ 3108 if (sock_recv(sockctrl, remote_errbuf, PCAP_ERRBUF_SIZE - 1, 3109 SOCK_RECEIVEALL_YES|SOCK_EOF_IS_ERROR, errbuf, 3110 PCAP_ERRBUF_SIZE) == -1) 3111 { 3112 // Network error. 3113 pcap_snprintf(remote_errbuf, PCAP_ERRBUF_SIZE, "Read of error message from client failed: %s", errbuf); 3114 return; 3115 } 3116 3117 /* 3118 * Null-terminate it. 3119 */ 3120 remote_errbuf[PCAP_ERRBUF_SIZE - 1] = '\0'; 3121 3122 /* 3123 * Throw away the rest. 3124 */ 3125 (void)rpcap_discard(sockctrl, plen - (PCAP_ERRBUF_SIZE - 1), remote_errbuf); 3126 } 3127 else if (plen == 0) 3128 { 3129 /* Empty error string. */ 3130 remote_errbuf[0] = '\0'; 3131 } 3132 else 3133 { 3134 if (sock_recv(sockctrl, remote_errbuf, plen, 3135 SOCK_RECEIVEALL_YES|SOCK_EOF_IS_ERROR, errbuf, 3136 PCAP_ERRBUF_SIZE) == -1) 3137 { 3138 // Network error. 3139 pcap_snprintf(remote_errbuf, PCAP_ERRBUF_SIZE, "Read of error message from client failed: %s", errbuf); 3140 return; 3141 } 3142 3143 /* 3144 * Null-terminate it. 3145 */ 3146 remote_errbuf[plen] = '\0'; 3147 } 3148 } 3149 3150 /* 3151 * Discard data from a connection. 3152 * Mostly used to discard wrong-sized messages. 3153 * Returns 0 on success, logs a message and returns -1 on a network 3154 * error. 3155 */ 3156 static int rpcap_discard(SOCKET sock, uint32 len, char *errbuf) 3157 { 3158 if (len != 0) 3159 { 3160 if (sock_discard(sock, len, errbuf, PCAP_ERRBUF_SIZE) == -1) 3161 { 3162 // Network error. 3163 return -1; 3164 } 3165 } 3166 return 0; 3167 } 3168 3169 /* 3170 * Read bytes into the pcap_t's buffer until we have the specified 3171 * number of bytes read or we get an error or interrupt indication. 3172 */ 3173 static int rpcap_read_packet_msg(SOCKET sock, pcap_t *p, size_t size) 3174 { 3175 u_char *bp; 3176 int cc; 3177 int bytes_read; 3178 3179 bp = p->bp; 3180 cc = p->cc; 3181 3182 /* 3183 * Loop until we have the amount of data requested or we get 3184 * an error or interrupt. 3185 */ 3186 while ((size_t)cc < size) 3187 { 3188 /* 3189 * We haven't read all of the packet header yet. 3190 * Read what remains, which could be all of it. 3191 */ 3192 bytes_read = sock_recv(sock, bp, size - cc, 3193 SOCK_RECEIVEALL_NO|SOCK_EOF_IS_ERROR, p->errbuf, 3194 PCAP_ERRBUF_SIZE); 3195 if (bytes_read == -1) 3196 { 3197 /* 3198 * Network error. Update the read pointer and 3199 * byte count, and return an error indication. 3200 */ 3201 p->bp = bp; 3202 p->cc = cc; 3203 return -1; 3204 } 3205 if (bytes_read == -3) 3206 { 3207 /* 3208 * Interrupted receive. Update the read 3209 * pointer and byte count, and return 3210 * an interrupted indication. 3211 */ 3212 p->bp = bp; 3213 p->cc = cc; 3214 return -3; 3215 } 3216 if (bytes_read == 0) 3217 { 3218 /* 3219 * EOF - server terminated the connection. 3220 * Update the read pointer and byte count, and 3221 * return an error indication. 3222 */ 3223 pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 3224 "The server terminated the connection."); 3225 return -1; 3226 } 3227 bp += bytes_read; 3228 cc += bytes_read; 3229 } 3230 p->bp = bp; 3231 p->cc = cc; 3232 return 0; 3233 } 3234