vendor/github.com/google/gopacket/layers/geneve_test.go

   1 // Copyright 2016 Google, Inc. All rights reserved.
   2 //
   3 // Use of this source code is governed by a BSD-style license
   4 // that can be found in the LICENSE file in the root of the source
   5 // tree.
   6
   7 package layers
   8
   9 import (
  10         "reflect"
  11         "testing"
  12
  13         "github.com/google/gopacket"
  14 )
  15
  16 var testPacketGeneve1 = []byte{
  17         0x00, 0x04, 0x00, 0x01, 0x00, 0x06, 0xfa, 0x16, 0x3e, 0x23, 0xd3, 0x42,
  18         0x00, 0x00, 0x08, 0x00, 0x45, 0x00, 0x00, 0x86, 0x87, 0x39, 0x40, 0x00,
  19         0x40, 0x11, 0x31, 0x35, 0xc0, 0xa8, 0x00, 0x53, 0xc0, 0xa8, 0x00, 0x55,
  20         0x31, 0x57, 0x17, 0xc1, 0x00, 0x72, 0x00, 0x00, 0x00, 0x00, 0x65, 0x58,
  21         0x00, 0x00, 0x00, 0x00, 0xba, 0x09, 0x60, 0x5f, 0xa0, 0x91, 0xa2, 0xfe,
  22         0x54, 0x48, 0x88, 0x51, 0x08, 0x00, 0x45, 0x00, 0x00, 0x54, 0x01, 0xf6,
  23         0x40, 0x00, 0x40, 0x01, 0xb7, 0x5f, 0xc0, 0xa8, 0x00, 0x01, 0xc0, 0xa8,
  24         0x00, 0x02, 0x08, 0x00, 0x79, 0xdf, 0x0c, 0xfa, 0x63, 0xc4, 0x03, 0x0b,
  25         0x50, 0x58, 0x00, 0x00, 0x00, 0x00, 0xee, 0x2b, 0x0d, 0x00, 0x00, 0x00,
  26         0x00, 0x00, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19,
  27         0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f, 0x20, 0x21, 0x22, 0x23, 0x24, 0x25,
  28         0x26, 0x27, 0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f, 0x30, 0x31,
  29         0x32, 0x33, 0x34, 0x35, 0x36, 0x37,
  30 }
  31
  32 var testPacketGeneve2 = []byte{
  33         0x12, 0xbe, 0x4e, 0xb6, 0xa7, 0xc7, 0x02, 0x88, 0x0a, 0x81, 0xbd, 0x6d,
  34         0x08, 0x00, 0x45, 0x00, 0x00, 0x86, 0x20, 0xf2, 0x00, 0x00, 0x40, 0x11,
  35         0x01, 0x52, 0xac, 0x10, 0x00, 0x01, 0xac, 0x10, 0x00, 0x02, 0x40, 0xa6,
  36         0x17, 0xc1, 0x00, 0x72, 0x00, 0x00, 0x00, 0x00, 0x65, 0x58, 0x00, 0x00,
  37         0x0a, 0x00, 0xd2, 0x8c, 0xdb, 0x12, 0x53, 0xd5, 0x8e, 0xab, 0xa2, 0xa5,
  38         0x02, 0xf7, 0x08, 0x00, 0x45, 0x00, 0x00, 0x54, 0x38, 0x1a, 0x40, 0x00,
  39         0x40, 0x01, 0x81, 0x3b, 0xc0, 0xa8, 0x00, 0x01, 0xc0, 0xa8, 0x00, 0x02,
  40         0x08, 0x00, 0xdd, 0x9d, 0x7e, 0xde, 0x02, 0xc3, 0xcb, 0x07, 0x51, 0x58,
  41         0x00, 0x00, 0x00, 0x00, 0xba, 0x8d, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00,
  42         0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19, 0x1a, 0x1b,
  43         0x1c, 0x1d, 0x1e, 0x1f, 0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27,
  44         0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f, 0x30, 0x31, 0x32, 0x33,
  45         0x34, 0x35, 0x36, 0x37,
  46 }
  47
  48 func TestDecodeGeneve1(t *testing.T) {
  49         p := gopacket.NewPacket(testPacketGeneve1, LinkTypeLinuxSLL, gopacket.Default)
  50         if p.ErrorLayer() != nil {
  51                 t.Error("Failed to decode packet:", p.ErrorLayer().Error())
  52         }
  53         checkLayers(p, []gopacket.LayerType{
  54                 LayerTypeLinuxSLL, LayerTypeIPv4, LayerTypeUDP, LayerTypeGeneve,
  55                 LayerTypeEthernet, LayerTypeIPv4, LayerTypeICMPv4, gopacket.LayerTypePayload,
  56         }, t)
  57         if got, ok := p.Layer(LayerTypeGeneve).(*Geneve); ok {
  58                 want := &Geneve{
  59                         BaseLayer: BaseLayer{
  60                                 Contents: testPacketGeneve1[44:52],
  61                                 Payload:  testPacketGeneve1[52:150],
  62                         },
  63                         Version:        0x0,
  64                         OptionsLength:  0x0,
  65                         OAMPacket:      false,
  66                         CriticalOption: false,
  67                         Protocol:       EthernetTypeTransparentEthernetBridging,
  68                         VNI:            0x0,
  69                 }
  70                 if !reflect.DeepEqual(want, got) {
  71                         t.Errorf("Geneve layer mismatch, \nwant %#v\ngot  %#v\n", want, got)
  72                 }
  73         }
  74 }
  75
  76 func TestDecodeGeneve2(t *testing.T) {
  77         p := gopacket.NewPacket(testPacketGeneve2, LinkTypeEthernet, gopacket.Default)
  78         if p.ErrorLayer() != nil {
  79                 t.Error("Failed to decode packet:", p.ErrorLayer().Error())
  80         }
  81         checkLayers(p, []gopacket.LayerType{
  82                 LayerTypeEthernet, LayerTypeIPv4, LayerTypeUDP, LayerTypeGeneve,
  83                 LayerTypeEthernet, LayerTypeIPv4, LayerTypeICMPv4, gopacket.LayerTypePayload,
  84         }, t)
  85         if got, ok := p.Layer(LayerTypeGeneve).(*Geneve); ok {
  86                 want := &Geneve{
  87                         BaseLayer: BaseLayer{
  88                                 Contents: testPacketGeneve2[42:50],
  89                                 Payload:  testPacketGeneve2[50:148],
  90                         },
  91                         Version:        0x0,
  92                         OptionsLength:  0x0,
  93                         OAMPacket:      false,
  94                         CriticalOption: false,
  95                         Protocol:       EthernetTypeTransparentEthernetBridging,
  96                         VNI:            0xa,
  97                 }
  98                 if !reflect.DeepEqual(want, got) {
  99                         t.Errorf("Geneve layer mismatch, \nwant %#v\ngot  %#v\n", want, got)
 100                 }
 101         }
 102 }
 103
 104 func BenchmarkDecodeGeneve1(b *testing.B) {
 105         for i := 0; i < b.N; i++ {
 106                 gopacket.NewPacket(testPacketGeneve1, LinkTypeEthernet, gopacket.NoCopy)
 107         }
 108 }