1 # Copyright (C) 2016 Nippon Telegraph and Telephone Corporation.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
16 from __future__ import print_function
23 from nose.tools import eq_
24 from nose.tools import ok_
26 from ryu.lib import pcaplib
27 from ryu.lib.packet import gre
28 from ryu.lib.packet import packet
29 from ryu.utils import binary_str
30 from ryu.lib.packet.ether_types import ETH_TYPE_IP, ETH_TYPE_TEB
32 LOG = logging.getLogger(__name__)
34 GENEVE_DATA_DIR = os.path.join(
35 os.path.dirname(sys.modules[__name__].__file__),
36 '../../packet_data/pcap/')
39 class Test_gre(unittest.TestCase):
41 Test case gre for ryu.lib.packet.gre.
45 gre_proto = ETH_TYPE_IP
46 nvgre_proto = ETH_TYPE_TEB
53 gre = gre.gre(version=version, protocol=gre_proto, checksum=checksum,
54 key=key, seq_number=seq_number)
56 def test_key_setter(self):
57 self.gre.key = self.key
58 eq_(self.gre._key, self.key)
59 eq_(self.gre._vsid, self.vsid)
60 eq_(self.gre._flow_id, self.flow_id)
62 def test_key_setter_none(self):
64 eq_(self.gre._key, None)
65 eq_(self.gre._vsid, None)
66 eq_(self.gre._flow_id, None)
68 self.gre.key = self.key
70 def test_vsid_setter(self):
71 self.gre.vsid = self.vsid
72 eq_(self.gre._key, self.key)
73 eq_(self.gre._vsid, self.vsid)
74 eq_(self.gre._flow_id, self.flow_id)
76 def test_flowid_setter(self):
77 self.gre.flow_id = self.flow_id
78 eq_(self.gre._key, self.key)
79 eq_(self.gre._vsid, self.vsid)
80 eq_(self.gre._flow_id, self.flow_id)
82 def test_nvgre_init(self):
83 nvgre = gre.nvgre(version=self.version, vsid=self.vsid,
86 eq_(nvgre.version, self.version)
87 eq_(nvgre.protocol, self.nvgre_proto)
88 eq_(nvgre.checksum, None)
89 eq_(nvgre.seq_number, None)
90 eq_(nvgre._key, self.key)
91 eq_(nvgre._vsid, self.vsid)
92 eq_(nvgre._flow_id, self.flow_id)
94 def test_parser(self):
102 # print('*** testing %s ...' % f)
103 for _, buf in pcaplib.Reader(
104 open(GENEVE_DATA_DIR + f + '.pcap', 'rb')):
105 # Checks if message can be parsed as expected.
106 pkt = packet.Packet(buf)
107 gre_pkt = pkt.get_protocol(gre.gre)
108 ok_(isinstance(gre_pkt, gre.gre),
109 'Failed to parse Gre message: %s' % pkt)
111 # Checks if message can be serialized as expected.
115 "b'%s' != b'%s'" % (binary_str(buf), binary_str(pkt.data)))