src/vnet/unix/pcap.h

   1 /*
   2  * Copyright (c) 2015 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15 /*
  16  * pcap.h: libpcap packet capture format
  17  *
  18  * Copyright (c) 2008 Eliot Dresselhaus
  19  *
  20  * Permission is hereby granted, free of charge, to any person obtaining
  21  * a copy of this software and associated documentation files (the
  22  * "Software"), to deal in the Software without restriction, including
  23  * without limitation the rights to use, copy, modify, merge, publish,
  24  * distribute, sublicense, and/or sell copies of the Software, and to
  25  * permit persons to whom the Software is furnished to do so, subject to
  26  * the following conditions:
  27  *
  28  * The above copyright notice and this permission notice shall be
  29  * included in all copies or substantial portions of the Software.
  30  *
  31  *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  32  *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  33  *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  34  *  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  35  *  LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  36  *  OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  37  *  WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  38  */
  39 /**
  40  * @file
  41  * @brief PCAP utility definitions
  42  */
  43 #ifndef included_vnet_pcap_h
  44 #define included_vnet_pcap_h
  45
  46 #include <vlib/vlib.h>
  47
  48 /**
  49  * @brief Packet types supported by PCAP
  50  *
  51  * null 0
  52  * ethernet 1
  53  * ppp 9
  54  * ip 12
  55  * hdlc 104
  56  */
  57 #define foreach_vnet_pcap_packet_type           \
  58   _ (null, 0)                                   \
  59   _ (ethernet, 1)                               \
  60   _ (ppp, 9)                                    \
  61   _ (ip, 12)                                    \
  62   _ (hdlc, 104)
  63
  64 typedef enum
  65 {
  66 #define _(f,n) PCAP_PACKET_TYPE_##f = (n),
  67   foreach_vnet_pcap_packet_type
  68 #undef _
  69 } pcap_packet_type_t;
  70
  71 #define foreach_pcap_file_header                        \
  72   /** 0xa1b2c3d4 host byte order.                       \
  73      0xd4c3b2a1 => need to byte swap everything. */     \
  74   _ (u32, magic)                                        \
  75                                                         \
  76   /** Currently major 2 minor 4. */                     \
  77   _ (u16, major_version)                                \
  78   _ (u16, minor_version)                                \
  79                                                         \
  80   /** 0 for GMT. */                                     \
  81   _ (u32, time_zone)                                    \
  82                                                         \
  83   /** Accuracy of timestamps.  Typically set to 0. */   \
  84   _ (u32, sigfigs)                                      \
  85                                                         \
  86   /** Size of largest packet in file. */                \
  87   _ (u32, max_packet_size_in_bytes)                     \
  88                                                         \
  89   /** One of vnet_pcap_packet_type_t. */                \
  90   _ (u32, packet_type)
  91
  92 /** File header struct */
  93 typedef struct
  94 {
  95 #define _(t, f) t f;
  96   foreach_pcap_file_header
  97 #undef _
  98 } pcap_file_header_t;
  99
 100 #define foreach_pcap_packet_header                                      \
 101   /** Time stamp in seconds  */                                         \
 102   _ (u32, time_in_sec)                                                  \
 103   /** Time stamp in microseconds. */                                    \
 104   _ (u32, time_in_usec)                                                 \
 105                                                                         \
 106   /** Number of bytes stored in file. */                                \
 107   _ (u32, n_packet_bytes_stored_in_file)                                \
 108   /** Number of bytes in actual packet. */                              \
 109   _ (u32, n_bytes_in_packet)
 110
 111 /** Packet header. */
 112 typedef struct
 113 {
 114 #define _(t, f) t f;
 115   foreach_pcap_packet_header
 116 #undef _
 117   /** Packet data follows. */
 118   u8 data[0];
 119 } pcap_packet_header_t;
 120
 121 /**
 122  * @brief PCAP main state data structure
 123  */
 124 typedef struct
 125 {
 126   /** File name of pcap output. */
 127   char *file_name;
 128
 129   /** Number of packets to capture. */
 130   u32 n_packets_to_capture;
 131
 132   /** Packet type */
 133   pcap_packet_type_t packet_type;
 134
 135   /** Number of packets currently captured. */
 136   u32 n_packets_captured;
 137
 138   /** flags */
 139   u32 flags;
 140 #define PCAP_MAIN_INIT_DONE (1 << 0)
 141
 142   /** File descriptor for reading/writing. */
 143   int file_descriptor;
 144
 145   /** Bytes written */
 146   u32 n_pcap_data_written;
 147
 148   /** Vector of pcap data. */
 149   u8 *pcap_data;
 150
 151   /** Packets read from file. */
 152   u8 **packets_read;
 153
 154   /** Timestamps */
 155   u64 *timestamps;
 156
 157   /** Min/Max Packet bytes */
 158   u32 min_packet_bytes, max_packet_bytes;
 159 } pcap_main_t;
 160
 161 /** Write out data to output file. */
 162 clib_error_t *pcap_write (pcap_main_t * pm);
 163
 164 /** Read data from file. */
 165 clib_error_t *pcap_read (pcap_main_t * pm);
 166
 167 /**
 168  * @brief Add packet
 169  *
 170  * @param *pm - pcap_main_t
 171  * @param time_now - f64
 172  * @param n_bytes_in_trace - u32
 173  * @param n_bytes_in_packet - u32
 174  *
 175  * @return Packet Data
 176  *
 177  */
 178 static inline void *
 179 pcap_add_packet (pcap_main_t * pm,
 180                  f64 time_now, u32 n_bytes_in_trace, u32 n_bytes_in_packet)
 181 {
 182   pcap_packet_header_t *h;
 183   u8 *d;
 184
 185   vec_add2 (pm->pcap_data, d, sizeof (h[0]) + n_bytes_in_trace);
 186   h = (void *) (d);
 187   h->time_in_sec = time_now;
 188   h->time_in_usec = 1e6 * (time_now - h->time_in_sec);
 189   h->n_packet_bytes_stored_in_file = n_bytes_in_trace;
 190   h->n_bytes_in_packet = n_bytes_in_packet;
 191   pm->n_packets_captured++;
 192   return h->data;
 193 }
 194
 195 /**
 196  * @brief Add buffer (vlib_buffer_t) to the trace
 197  *
 198  * @param *pm - pcap_main_t
 199  * @param *vm - vlib_main_t
 200  * @param buffer_index - u32
 201  * @param n_bytes_in_trace - u32
 202  *
 203  */
 204 static inline void
 205 pcap_add_buffer (pcap_main_t * pm,
 206                  vlib_main_t * vm, u32 buffer_index, u32 n_bytes_in_trace)
 207 {
 208   vlib_buffer_t *b = vlib_get_buffer (vm, buffer_index);
 209   u32 n = vlib_buffer_length_in_chain (vm, b);
 210   i32 n_left = clib_min (n_bytes_in_trace, n);
 211   f64 time_now = vlib_time_now (vm);
 212   void *d;
 213
 214   if (PREDICT_TRUE (pm->n_packets_captured < pm->n_packets_to_capture))
 215     {
 216       d = pcap_add_packet (pm, time_now, n_left, n);
 217       while (1)
 218         {
 219           u32 copy_length = clib_min ((u32) n_left, b->current_length);
 220           clib_memcpy (d, b->data + b->current_data, copy_length);
 221           n_left -= b->current_length;
 222           if (n_left <= 0)
 223             break;
 224           d += b->current_length;
 225           ASSERT (b->flags & VLIB_BUFFER_NEXT_PRESENT);
 226           b = vlib_get_buffer (vm, b->next_buffer);
 227         }
 228     }
 229 }
 230
 231 #endif /* included_vnet_pcap_h */
 232
 233 /*
 234  * fd.io coding-style-patch-verification: ON
 235  *
 236  * Local Variables:
 237  * eval: (c-set-style "gnu")
 238  * End:
 239  */