appneta/tcpreplay

View on GitHub
src/fragroute/pkt.h

Summary

Maintainability
Test Coverage
/*
 * pkt.h
 *
 * Copyright (c) 2001 Dug Song <dugsong@monkey.org>
 *
 * $Id$
 */

#pragma once

#include "defines.h"
#include "config.h"
#include "lib/queue.h"
#include <sys/time.h>

#ifdef HAVE_LIBDNET
/* need to undef these which are pulled in via defines.h, prior to importing dnet.h */
#undef icmp_id
#undef icmp_seq
#undef icmp_data
#undef icmp_mask
#ifdef HAVE_DNET_H
#include <dnet.h>
#endif
#ifdef HAVE_DUMBNET_H
#include <dumbnet.h>
#endif
#endif

#define PKT_BUF_LEN (ETH_HDR_LEN + ETH_MTU)
#define PKT_BUF_ALIGN 2

struct pkt {
    struct timeval pkt_ts;

    struct eth_hdr *pkt_eth;
    union {
        u_char *eth_data;
        struct ip_hdr *ip;
        struct ip6_hdr *ip6;
    } pkt_n_hdr_u;
    union {
        u_char *ip_data;
        struct icmp_hdr *icmp;
        struct tcp_hdr *tcp;
        struct udp_hdr *udp;
    } pkt_t_hdr_u;
    union {
        u_char *t_data;
        union icmp_msg *icmp;
    } pkt_t_data_u;

    u_char *pkt_buf;
    size_t pkt_buf_size;
    u_char *pkt_data;
    u_char *pkt_end;

    TAILQ_ENTRY(pkt) pkt_next;
};
#define pkt_ip pkt_n_hdr_u.ip
#define pkt_ip6 pkt_n_hdr_u.ip6
#define pkt_eth_data pkt_n_hdr_u.eth_data

#define pkt_icmp pkt_t_hdr_u.icmp
#define pkt_tcp pkt_t_hdr_u.tcp
#define pkt_udp pkt_t_hdr_u.udp
#define pkt_ip_data pkt_t_hdr_u.ip_data

#define pkt_tcp_data pkt_t_data_u.t_data
#define pkt_udp_data pkt_t_data_u.t_data
#define pkt_icmp_msg pkt_t_data_u.icmp

TAILQ_HEAD(pktq, pkt);

void pkt_init(int size);
void pkt_close(void);

struct pkt *pkt_new(size_t len);
struct pkt *pkt_dup(struct pkt *);
void pkt_decorate(struct pkt *pkt);
void pkt_free(struct pkt *pkt);

void pktq_reverse(struct pktq *pktq);
void pktq_shuffle(rand_t *r, struct pktq *pktq);
struct pkt *pktq_random(rand_t *r, struct pktq *pktq);