1a0ee43a1SRui Paulo #ifdef HAVE_CONFIG_H 2a0ee43a1SRui Paulo #include "config.h" 3a0ee43a1SRui Paulo #endif 4a0ee43a1SRui Paulo 5a0ee43a1SRui Paulo #include <sys/param.h> 6a0ee43a1SRui Paulo 7a0ee43a1SRui Paulo #include <stdlib.h> 8a0ee43a1SRui Paulo #include <string.h> 9a0ee43a1SRui Paulo #include <errno.h> 10a0ee43a1SRui Paulo 11a0ee43a1SRui Paulo #include <ctype.h> 12a0ee43a1SRui Paulo #include <netinet/in.h> 13a0ee43a1SRui Paulo #include <sys/mman.h> 14a0ee43a1SRui Paulo #include <sys/socket.h> 15a0ee43a1SRui Paulo #include <sys/types.h> 16a0ee43a1SRui Paulo #include <unistd.h> 17a0ee43a1SRui Paulo 18a0ee43a1SRui Paulo #include "snf.h" 19a0ee43a1SRui Paulo #include "pcap-int.h" 20a0ee43a1SRui Paulo 21a0ee43a1SRui Paulo #ifdef SNF_ONLY 22a0ee43a1SRui Paulo #define snf_create pcap_create 23a0ee43a1SRui Paulo #define snf_platform_finddevs pcap_platform_finddevs 24a0ee43a1SRui Paulo #endif 25a0ee43a1SRui Paulo 26a0ee43a1SRui Paulo static int 27a0ee43a1SRui Paulo snf_set_datalink(pcap_t *p, int dlt) 28a0ee43a1SRui Paulo { 29a0ee43a1SRui Paulo p->linktype = dlt; 30a0ee43a1SRui Paulo return (0); 31a0ee43a1SRui Paulo } 32a0ee43a1SRui Paulo 33a0ee43a1SRui Paulo static int 34a0ee43a1SRui Paulo snf_pcap_stats(pcap_t *p, struct pcap_stat *ps) 35a0ee43a1SRui Paulo { 36a0ee43a1SRui Paulo struct snf_ring_stats stats; 37a0ee43a1SRui Paulo int rc; 38a0ee43a1SRui Paulo 39a0ee43a1SRui Paulo if ((rc = snf_ring_getstats(p->md.snf_ring, &stats))) { 40a0ee43a1SRui Paulo snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "snf_get_stats: %s", 41a0ee43a1SRui Paulo pcap_strerror(rc)); 42a0ee43a1SRui Paulo return -1; 43a0ee43a1SRui Paulo } 44a0ee43a1SRui Paulo ps->ps_recv = stats.ring_pkt_recv + stats.ring_pkt_overflow; 45a0ee43a1SRui Paulo ps->ps_drop = stats.ring_pkt_overflow; 46a0ee43a1SRui Paulo ps->ps_ifdrop = stats.nic_pkt_overflow + stats.nic_pkt_bad; 47a0ee43a1SRui Paulo return 0; 48a0ee43a1SRui Paulo } 49a0ee43a1SRui Paulo 50a0ee43a1SRui Paulo static void 51a0ee43a1SRui Paulo snf_platform_cleanup(pcap_t *p) 52a0ee43a1SRui Paulo { 53a0ee43a1SRui Paulo if (p == NULL) 54a0ee43a1SRui Paulo return; 55a0ee43a1SRui Paulo 56a0ee43a1SRui Paulo snf_ring_close(p->md.snf_ring); 57a0ee43a1SRui Paulo snf_close(p->md.snf_handle); 58a0ee43a1SRui Paulo pcap_cleanup_live_common(p); 59a0ee43a1SRui Paulo } 60a0ee43a1SRui Paulo 61a0ee43a1SRui Paulo static int 62a0ee43a1SRui Paulo snf_getnonblock(pcap_t *p, char *errbuf) 63a0ee43a1SRui Paulo { 64a0ee43a1SRui Paulo return (p->md.snf_timeout == 0); 65a0ee43a1SRui Paulo } 66a0ee43a1SRui Paulo 67a0ee43a1SRui Paulo static int 68a0ee43a1SRui Paulo snf_setnonblock(pcap_t *p, int nonblock, char *errbuf) 69a0ee43a1SRui Paulo { 70a0ee43a1SRui Paulo if (nonblock) 71a0ee43a1SRui Paulo p->md.snf_timeout = 0; 72a0ee43a1SRui Paulo else { 73a0ee43a1SRui Paulo if (p->md.timeout <= 0) 74a0ee43a1SRui Paulo p->md.snf_timeout = -1; /* forever */ 75a0ee43a1SRui Paulo else 76a0ee43a1SRui Paulo p->md.snf_timeout = p->md.timeout; 77a0ee43a1SRui Paulo } 78a0ee43a1SRui Paulo return (0); 79a0ee43a1SRui Paulo } 80a0ee43a1SRui Paulo 81a0ee43a1SRui Paulo #define _NSEC_PER_SEC 1000000000 82a0ee43a1SRui Paulo 83a0ee43a1SRui Paulo static inline 84a0ee43a1SRui Paulo struct timeval 85a0ee43a1SRui Paulo snf_timestamp_to_timeval(const int64_t ts_nanosec) 86a0ee43a1SRui Paulo { 87a0ee43a1SRui Paulo struct timeval tv; 88a0ee43a1SRui Paulo int32_t rem; 89a0ee43a1SRui Paulo if (ts_nanosec == 0) 90a0ee43a1SRui Paulo return (struct timeval) { 0, 0 }; 91a0ee43a1SRui Paulo tv.tv_sec = ts_nanosec / _NSEC_PER_SEC; 92a0ee43a1SRui Paulo tv.tv_usec = (ts_nanosec % _NSEC_PER_SEC) / 1000; 93a0ee43a1SRui Paulo return tv; 94a0ee43a1SRui Paulo } 95a0ee43a1SRui Paulo 96a0ee43a1SRui Paulo static int 97a0ee43a1SRui Paulo snf_read(pcap_t *p, int cnt, pcap_handler callback, u_char *user) 98a0ee43a1SRui Paulo { 99a0ee43a1SRui Paulo struct pcap_pkthdr hdr; 100a0ee43a1SRui Paulo int i, flags, err, caplen, n; 101a0ee43a1SRui Paulo struct snf_recv_req req; 102a0ee43a1SRui Paulo 103a0ee43a1SRui Paulo if (!p || cnt == 0) 104a0ee43a1SRui Paulo return -1; 105a0ee43a1SRui Paulo 106a0ee43a1SRui Paulo n = 0; 107a0ee43a1SRui Paulo while (n < cnt || cnt < 0) { 108a0ee43a1SRui Paulo /* 109a0ee43a1SRui Paulo * Has "pcap_breakloop()" been called? 110a0ee43a1SRui Paulo */ 111a0ee43a1SRui Paulo if (p->break_loop) { 112a0ee43a1SRui Paulo if (n == 0) { 113a0ee43a1SRui Paulo p->break_loop = 0; 114a0ee43a1SRui Paulo return (-2); 115a0ee43a1SRui Paulo } else { 116a0ee43a1SRui Paulo return (n); 117a0ee43a1SRui Paulo } 118a0ee43a1SRui Paulo } 119a0ee43a1SRui Paulo 120a0ee43a1SRui Paulo err = snf_ring_recv(p->md.snf_ring, p->md.snf_timeout, &req); 121a0ee43a1SRui Paulo 122a0ee43a1SRui Paulo if (err) { 123a0ee43a1SRui Paulo if (err == EBUSY || err == EAGAIN) 124a0ee43a1SRui Paulo return (0); 125a0ee43a1SRui Paulo if (err == EINTR) 126a0ee43a1SRui Paulo continue; 127a0ee43a1SRui Paulo if (err != 0) { 128a0ee43a1SRui Paulo snprintf(p->errbuf, PCAP_ERRBUF_SIZE, "snf_read: %s", 129a0ee43a1SRui Paulo pcap_strerror(err)); 130a0ee43a1SRui Paulo return -1; 131a0ee43a1SRui Paulo } 132a0ee43a1SRui Paulo } 133a0ee43a1SRui Paulo 134a0ee43a1SRui Paulo caplen = req.length; 135a0ee43a1SRui Paulo if (caplen > p->snapshot) 136a0ee43a1SRui Paulo caplen = p->snapshot; 137a0ee43a1SRui Paulo 138a0ee43a1SRui Paulo if ((p->fcode.bf_insns == NULL) || 139a0ee43a1SRui Paulo bpf_filter(p->fcode.bf_insns, req.pkt_addr, req.length, caplen)) { 140a0ee43a1SRui Paulo hdr.ts = snf_timestamp_to_timeval(req.timestamp); 141a0ee43a1SRui Paulo hdr.caplen = caplen; 142a0ee43a1SRui Paulo hdr.len = req.length; 143a0ee43a1SRui Paulo callback(user, &hdr, req.pkt_addr); 144a0ee43a1SRui Paulo } 145a0ee43a1SRui Paulo n++; 146a0ee43a1SRui Paulo } 147a0ee43a1SRui Paulo return (n); 148a0ee43a1SRui Paulo } 149a0ee43a1SRui Paulo 150a0ee43a1SRui Paulo static int 151a0ee43a1SRui Paulo snf_setfilter(pcap_t *p, struct bpf_program *fp) 152a0ee43a1SRui Paulo { 153a0ee43a1SRui Paulo if (!p) 154a0ee43a1SRui Paulo return -1; 155a0ee43a1SRui Paulo if (!fp) { 156a0ee43a1SRui Paulo strncpy(p->errbuf, "setfilter: No filter specified", 157a0ee43a1SRui Paulo sizeof(p->errbuf)); 158a0ee43a1SRui Paulo return -1; 159a0ee43a1SRui Paulo } 160a0ee43a1SRui Paulo 161a0ee43a1SRui Paulo /* Make our private copy of the filter */ 162a0ee43a1SRui Paulo 163a0ee43a1SRui Paulo if (install_bpf_program(p, fp) < 0) 164a0ee43a1SRui Paulo return -1; 165a0ee43a1SRui Paulo 166a0ee43a1SRui Paulo p->md.use_bpf = 0; 167a0ee43a1SRui Paulo 168a0ee43a1SRui Paulo return (0); 169a0ee43a1SRui Paulo } 170a0ee43a1SRui Paulo 171a0ee43a1SRui Paulo static int 172a0ee43a1SRui Paulo snf_inject(pcap_t *p, const void *buf _U_, size_t size _U_) 173a0ee43a1SRui Paulo { 174a0ee43a1SRui Paulo strlcpy(p->errbuf, "Sending packets isn't supported with snf", 175a0ee43a1SRui Paulo PCAP_ERRBUF_SIZE); 176a0ee43a1SRui Paulo return (-1); 177a0ee43a1SRui Paulo } 178a0ee43a1SRui Paulo 179a0ee43a1SRui Paulo static int 180a0ee43a1SRui Paulo snf_activate(pcap_t* p) 181a0ee43a1SRui Paulo { 182a0ee43a1SRui Paulo char *device = p->opt.source; 183a0ee43a1SRui Paulo const char *nr = NULL; 184a0ee43a1SRui Paulo int err; 185a0ee43a1SRui Paulo int flags = 0; 186a0ee43a1SRui Paulo 187a0ee43a1SRui Paulo if (device == NULL) { 188a0ee43a1SRui Paulo snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 189a0ee43a1SRui Paulo "device is NULL: %s", pcap_strerror(errno)); 190a0ee43a1SRui Paulo return -1; 191a0ee43a1SRui Paulo } 192a0ee43a1SRui Paulo 193a0ee43a1SRui Paulo /* In Libpcap, we set pshared by default if NUM_RINGS is set to > 1. 194a0ee43a1SRui Paulo * Since libpcap isn't thread-safe */ 195a0ee43a1SRui Paulo if ((nr = getenv("SNF_NUM_RINGS")) && *nr && atoi(nr) > 1) 196a0ee43a1SRui Paulo flags |= SNF_F_PSHARED; 197a0ee43a1SRui Paulo else 198a0ee43a1SRui Paulo nr = NULL; 199a0ee43a1SRui Paulo 200a0ee43a1SRui Paulo err = snf_open(p->md.snf_boardnum, 201a0ee43a1SRui Paulo 0, /* let SNF API parse SNF_NUM_RINGS, if set */ 202a0ee43a1SRui Paulo NULL, /* default RSS, or use SNF_RSS_FLAGS env */ 203a0ee43a1SRui Paulo 0, /* default to SNF_DATARING_SIZE from env */ 204a0ee43a1SRui Paulo flags, /* may want pshared */ 205a0ee43a1SRui Paulo &p->md.snf_handle); 206a0ee43a1SRui Paulo if (err != 0) { 207a0ee43a1SRui Paulo snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 208a0ee43a1SRui Paulo "snf_open failed: %s", pcap_strerror(err)); 209a0ee43a1SRui Paulo return -1; 210a0ee43a1SRui Paulo } 211a0ee43a1SRui Paulo 212a0ee43a1SRui Paulo err = snf_ring_open(p->md.snf_handle, &p->md.snf_ring); 213a0ee43a1SRui Paulo if (err != 0) { 214a0ee43a1SRui Paulo snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 215a0ee43a1SRui Paulo "snf_ring_open failed: %s", pcap_strerror(err)); 216a0ee43a1SRui Paulo return -1; 217a0ee43a1SRui Paulo } 218a0ee43a1SRui Paulo 219a0ee43a1SRui Paulo if (p->md.timeout <= 0) 220a0ee43a1SRui Paulo p->md.snf_timeout = -1; 221a0ee43a1SRui Paulo else 222a0ee43a1SRui Paulo p->md.snf_timeout = p->md.timeout; 223a0ee43a1SRui Paulo 224a0ee43a1SRui Paulo err = snf_start(p->md.snf_handle); 225a0ee43a1SRui Paulo if (err != 0) { 226a0ee43a1SRui Paulo snprintf(p->errbuf, PCAP_ERRBUF_SIZE, 227a0ee43a1SRui Paulo "snf_start failed: %s", pcap_strerror(err)); 228a0ee43a1SRui Paulo return -1; 229a0ee43a1SRui Paulo } 230a0ee43a1SRui Paulo 231a0ee43a1SRui Paulo /* 232a0ee43a1SRui Paulo * "select()" and "poll()" don't work on snf descriptors. 233a0ee43a1SRui Paulo */ 234a0ee43a1SRui Paulo p->selectable_fd = -1; 235a0ee43a1SRui Paulo p->linktype = DLT_EN10MB; 236a0ee43a1SRui Paulo p->read_op = snf_read; 237a0ee43a1SRui Paulo p->inject_op = snf_inject; 238a0ee43a1SRui Paulo p->setfilter_op = snf_setfilter; 239a0ee43a1SRui Paulo p->setdirection_op = NULL; /* Not implemented.*/ 240a0ee43a1SRui Paulo p->set_datalink_op = snf_set_datalink; 241a0ee43a1SRui Paulo p->getnonblock_op = snf_getnonblock; 242a0ee43a1SRui Paulo p->setnonblock_op = snf_setnonblock; 243a0ee43a1SRui Paulo p->stats_op = snf_pcap_stats; 244a0ee43a1SRui Paulo p->cleanup_op = snf_platform_cleanup; 245a0ee43a1SRui Paulo p->md.stat.ps_recv = 0; 246a0ee43a1SRui Paulo p->md.stat.ps_drop = 0; 247a0ee43a1SRui Paulo p->md.stat.ps_ifdrop = 0; 248a0ee43a1SRui Paulo return 0; 249a0ee43a1SRui Paulo } 250a0ee43a1SRui Paulo 251a0ee43a1SRui Paulo int 252a0ee43a1SRui Paulo snf_platform_finddevs(pcap_if_t **devlistp, char *errbuf) 253a0ee43a1SRui Paulo { 254a0ee43a1SRui Paulo /* 255a0ee43a1SRui Paulo * There are no platform-specific devices since each device 256a0ee43a1SRui Paulo * exists as a regular Ethernet device. 257a0ee43a1SRui Paulo */ 258a0ee43a1SRui Paulo return 0; 259a0ee43a1SRui Paulo } 260a0ee43a1SRui Paulo 261a0ee43a1SRui Paulo pcap_t * 262a0ee43a1SRui Paulo snf_create(const char *device, char *ebuf) 263a0ee43a1SRui Paulo { 264a0ee43a1SRui Paulo pcap_t *p; 265a0ee43a1SRui Paulo int boardnum = -1; 266a0ee43a1SRui Paulo struct snf_ifaddrs *ifaddrs, *ifa; 267a0ee43a1SRui Paulo size_t devlen; 268a0ee43a1SRui Paulo 269a0ee43a1SRui Paulo if (snf_init(SNF_VERSION_API)) 270a0ee43a1SRui Paulo return NULL; 271a0ee43a1SRui Paulo 272a0ee43a1SRui Paulo /* 273a0ee43a1SRui Paulo * Match a given interface name to our list of interface names, from 274a0ee43a1SRui Paulo * which we can obtain the intended board number 275a0ee43a1SRui Paulo */ 276a0ee43a1SRui Paulo if (snf_getifaddrs(&ifaddrs) || ifaddrs == NULL) 277a0ee43a1SRui Paulo return NULL; 278a0ee43a1SRui Paulo devlen = strlen(device) + 1; 279a0ee43a1SRui Paulo ifa = ifaddrs; 280a0ee43a1SRui Paulo while (ifa) { 281a0ee43a1SRui Paulo if (!strncmp(device, ifa->snf_ifa_name, devlen)) { 282a0ee43a1SRui Paulo boardnum = ifa->snf_ifa_boardnum; 283a0ee43a1SRui Paulo break; 284a0ee43a1SRui Paulo } 285a0ee43a1SRui Paulo ifa = ifa->snf_ifa_next; 286a0ee43a1SRui Paulo } 287a0ee43a1SRui Paulo snf_freeifaddrs(ifaddrs); 288a0ee43a1SRui Paulo 289a0ee43a1SRui Paulo if (ifa == NULL) { 290a0ee43a1SRui Paulo /* 291a0ee43a1SRui Paulo * If we can't find the device by name, support the name "snfX" 292a0ee43a1SRui Paulo * and "snf10gX" where X is the board number. 293a0ee43a1SRui Paulo */ 294a0ee43a1SRui Paulo if (sscanf(device, "snf10g%d", &boardnum) != 1 && 295a0ee43a1SRui Paulo sscanf(device, "snf%d", &boardnum) != 1) 296a0ee43a1SRui Paulo return NULL; 297a0ee43a1SRui Paulo } 298a0ee43a1SRui Paulo 299a0ee43a1SRui Paulo p = pcap_create_common(device, ebuf); 300a0ee43a1SRui Paulo if (p == NULL) 301a0ee43a1SRui Paulo return NULL; 302a0ee43a1SRui Paulo 303a0ee43a1SRui Paulo p->activate_op = snf_activate; 304a0ee43a1SRui Paulo p->md.snf_boardnum = boardnum; 305a0ee43a1SRui Paulo return p; 306a0ee43a1SRui Paulo } 307