github.com/gopacket/gopacket@v1.1.0/layers/dhcpv6_test.go (about)

     1  // Copyright 2018, 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  	"bytes"
    11  	"testing"
    12  
    13  	"github.com/gopacket/gopacket"
    14  )
    15  
    16  func TestDHCPv6EncodeRequest(t *testing.T) {
    17  	dhcpv6 := &DHCPv6{MsgType: DHCPv6MsgTypeRequest, HopCount: 0, TransactionID: []byte{87, 25, 88}}
    18  
    19  	client := &DHCPv6DUID{Type: DHCPv6DUIDTypeLLT, HardwareType: []byte{0, 1}, Time: []byte{28, 56, 38, 45}, LinkLayerAddress: []byte{8, 0, 39, 254, 143, 149}}
    20  	dhcpv6.Options = append(dhcpv6.Options, NewDHCPv6Option(DHCPv6OptClientID, client.Encode()))
    21  
    22  	server := &DHCPv6DUID{Type: DHCPv6DUIDTypeLLT, HardwareType: []byte{0, 1}, Time: []byte{28, 56, 37, 232}, LinkLayerAddress: []byte{8, 0, 39, 212, 16, 187}}
    23  	dhcpv6.Options = append(dhcpv6.Options, NewDHCPv6Option(DHCPv6OptServerID, server.Encode()))
    24  
    25  	buf := gopacket.NewSerializeBuffer()
    26  	opts := gopacket.SerializeOptions{FixLengths: true}
    27  	err := gopacket.SerializeLayers(buf, opts, dhcpv6)
    28  	if err != nil {
    29  		t.Fatal(err)
    30  	}
    31  
    32  	p2 := gopacket.NewPacket(buf.Bytes(), LayerTypeDHCPv6, testDecodeOptions)
    33  	dhcpv62 := p2.Layer(LayerTypeDHCPv6).(*DHCPv6)
    34  	testDHCPv6Equal(t, dhcpv6, dhcpv62)
    35  }
    36  
    37  func TestDHCPv6EncodeReply(t *testing.T) {
    38  	dhcpv6 := &DHCPv6{MsgType: DHCPv6MsgTypeReply, HopCount: 0, TransactionID: []byte{87, 25, 88}}
    39  
    40  	client := &DHCPv6DUID{Type: DHCPv6DUIDTypeLLT, HardwareType: []byte{0, 1}, Time: []byte{28, 56, 38, 45}, LinkLayerAddress: []byte{8, 0, 39, 254, 143, 149}}
    41  	dhcpv6.Options = append(dhcpv6.Options, NewDHCPv6Option(DHCPv6OptClientID, client.Encode()))
    42  
    43  	server := &DHCPv6DUID{Type: DHCPv6DUIDTypeLLT, HardwareType: []byte{0, 1}, Time: []byte{28, 56, 37, 232}, LinkLayerAddress: []byte{8, 0, 39, 212, 16, 187}}
    44  	dhcpv6.Options = append(dhcpv6.Options, NewDHCPv6Option(DHCPv6OptServerID, server.Encode()))
    45  
    46  	buf := gopacket.NewSerializeBuffer()
    47  	opts := gopacket.SerializeOptions{FixLengths: true}
    48  	err := gopacket.SerializeLayers(buf, opts, dhcpv6)
    49  	if err != nil {
    50  		t.Fatal(err)
    51  	}
    52  
    53  	p2 := gopacket.NewPacket(buf.Bytes(), LayerTypeDHCPv6, testDecodeOptions)
    54  	dhcpv62 := p2.Layer(LayerTypeDHCPv6).(*DHCPv6)
    55  	testDHCPv6Equal(t, dhcpv6, dhcpv62)
    56  }
    57  
    58  func testDHCPv6Equal(t *testing.T, d1, d2 *DHCPv6) {
    59  	if d1.MsgType != d2.MsgType {
    60  		t.Errorf("expected MsgType=%s, got %s", d1.MsgType, d2.MsgType)
    61  	}
    62  	if d1.HopCount != d2.HopCount {
    63  		t.Errorf("expected HopCount=%d, got %d", d1.HopCount, d2.HopCount)
    64  	}
    65  	if !d1.LinkAddr.Equal(d2.LinkAddr) {
    66  		t.Errorf("expected LinkAddr=%v, got %v", d1.LinkAddr, d2.LinkAddr)
    67  	}
    68  	if !d1.PeerAddr.Equal(d2.PeerAddr) {
    69  		t.Errorf("expected PeerAddr=%v, got %v", d1.PeerAddr, d2.PeerAddr)
    70  	}
    71  	if !bytes.Equal(d1.TransactionID, d2.TransactionID) {
    72  		t.Errorf("expected TransactionID=%v, got %v", d1.TransactionID, d2.TransactionID)
    73  	}
    74  	if len(d1.Options) != len(d2.Options) {
    75  		t.Errorf("expected %d options, got %d", len(d1.Options), len(d2.Options))
    76  	}
    77  
    78  	for i, o := range d1.Options {
    79  		testDHCPv6OptionEqual(t, i, o, d2.Options[i])
    80  	}
    81  }
    82  
    83  func testDHCPv6OptionEqual(t *testing.T, idx int, d1, d2 DHCPv6Option) {
    84  	if d1.Code != d2.Code {
    85  		t.Errorf("expection Options[%d].Code = %s, got %s", idx, d1.Code, d2.Code)
    86  	}
    87  	if d1.Length != d2.Length {
    88  		t.Errorf("expection Options[%d].Length = %d, got %d", idx, d1.Length, d2.Length)
    89  	}
    90  	if !bytes.Equal(d1.Data, d2.Data) {
    91  		t.Errorf("expection Options[%d].Data to be = %v, got %v", idx, d1.Data, d2.Data)
    92  	}
    93  }