173 lines
4.6 KiB
Python
173 lines
4.6 KiB
Python
# Copyright (C) 2013 Nippon Telegraph and Telephone Corporation.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
|
# implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import logging
|
|
import struct
|
|
import unittest
|
|
|
|
from nose.tools import eq_
|
|
from nose.tools import ok_
|
|
from nose.tools import raises
|
|
from ryu.ofproto import ether
|
|
from ryu.ofproto import inet
|
|
from ryu.lib.packet import ethernet
|
|
from ryu.lib.packet import packet
|
|
from ryu.lib.packet import ipv4
|
|
from ryu.lib.packet import vlan
|
|
from ryu.lib.packet import pbb
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class Test_itag(unittest.TestCase):
|
|
|
|
pcp = 3
|
|
dei = 0
|
|
uca = 1
|
|
sid = 16770000
|
|
data = pcp << 29 | dei << 28 | uca << 27 | sid
|
|
buf = struct.pack(pbb.itag._PACK_STR, data)
|
|
it = pbb.itag(pcp, dei, uca, sid)
|
|
|
|
def setUp(self):
|
|
pass
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def test_init(self):
|
|
eq_(self.pcp, self.it.pcp)
|
|
eq_(self.dei, self.it.dei)
|
|
eq_(self.uca, self.it.uca)
|
|
eq_(self.sid, self.it.sid)
|
|
|
|
def test_parser(self):
|
|
_res = pbb.itag.parser(self.buf)
|
|
if type(_res) is tuple:
|
|
res = _res[0]
|
|
else:
|
|
res = _res
|
|
eq_(res.pcp, self.pcp)
|
|
eq_(res.dei, self.dei)
|
|
eq_(res.uca, self.uca)
|
|
eq_(res.sid, self.sid)
|
|
|
|
def test_serialize(self):
|
|
data = bytearray()
|
|
prev = None
|
|
buf = self.it.serialize(data, prev)
|
|
res = struct.unpack(pbb.itag._PACK_STR, buf)
|
|
eq_(res[0], self.data)
|
|
|
|
def _build_itag(self):
|
|
b_src_mac = '00:07:0d:af:f4:54'
|
|
b_dst_mac = '00:00:00:00:00:00'
|
|
b_ethertype = ether.ETH_TYPE_8021AD
|
|
e1 = ethernet.ethernet(b_dst_mac, b_src_mac, b_ethertype)
|
|
|
|
b_pcp = 0
|
|
b_cfi = 0
|
|
b_vid = 32
|
|
b_ethertype = ether.ETH_TYPE_8021Q
|
|
bt = vlan.svlan(b_pcp, b_cfi, b_vid, b_ethertype)
|
|
|
|
c_src_mac = '11:11:11:11:11:11'
|
|
c_dst_mac = 'aa:aa:aa:aa:aa:aa'
|
|
c_ethertype = ether.ETH_TYPE_8021AD
|
|
e2 = ethernet.ethernet(c_dst_mac, c_src_mac, c_ethertype)
|
|
|
|
s_pcp = 0
|
|
s_cfi = 0
|
|
s_vid = 32
|
|
s_ethertype = ether.ETH_TYPE_8021Q
|
|
st = vlan.svlan(s_pcp, s_cfi, s_vid, s_ethertype)
|
|
|
|
c_pcp = 0
|
|
c_cfi = 0
|
|
c_vid = 32
|
|
c_ethertype = ether.ETH_TYPE_IP
|
|
ct = vlan.vlan(c_pcp, c_cfi, c_vid, c_ethertype)
|
|
|
|
version = 4
|
|
header_length = 20
|
|
tos = 0
|
|
total_length = 24
|
|
identification = 0x8a5d
|
|
flags = 0
|
|
offset = 1480
|
|
ttl = 64
|
|
proto = inet.IPPROTO_ICMP
|
|
csum = 0xa7f2
|
|
src = '131.151.32.21'
|
|
dst = '131.151.32.129'
|
|
option = b'TEST'
|
|
ip = ipv4.ipv4(version, header_length, tos, total_length,
|
|
identification, flags, offset, ttl, proto, csum,
|
|
src, dst, option)
|
|
|
|
p = packet.Packet()
|
|
|
|
p.add_protocol(e1)
|
|
p.add_protocol(bt)
|
|
p.add_protocol(self.it)
|
|
p.add_protocol(e2)
|
|
p.add_protocol(st)
|
|
p.add_protocol(ct)
|
|
p.add_protocol(ip)
|
|
p.serialize()
|
|
|
|
return p
|
|
|
|
def test_build_itag(self):
|
|
p = self._build_itag()
|
|
|
|
e = p.get_protocols(ethernet.ethernet)
|
|
ok_(e)
|
|
ok_(isinstance(e, list))
|
|
eq_(e[0].ethertype, ether.ETH_TYPE_8021AD)
|
|
eq_(e[1].ethertype, ether.ETH_TYPE_8021AD)
|
|
|
|
sv = p.get_protocols(vlan.svlan)
|
|
ok_(sv)
|
|
ok_(isinstance(sv, list))
|
|
eq_(sv[0].ethertype, ether.ETH_TYPE_8021Q)
|
|
eq_(sv[1].ethertype, ether.ETH_TYPE_8021Q)
|
|
|
|
it = p.get_protocol(pbb.itag)
|
|
ok_(it)
|
|
|
|
v = p.get_protocol(vlan.vlan)
|
|
ok_(v)
|
|
eq_(v.ethertype, ether.ETH_TYPE_IP)
|
|
|
|
ip = p.get_protocol(ipv4.ipv4)
|
|
ok_(ip)
|
|
|
|
eq_(it.pcp, self.pcp)
|
|
eq_(it.dei, self.dei)
|
|
eq_(it.uca, self.uca)
|
|
eq_(it.sid, self.sid)
|
|
|
|
@raises(Exception)
|
|
def test_malformed_itag(self):
|
|
m_short_buf = self.buf[1:pbb.itag._MIN_LEN]
|
|
pbb.itag.parser(m_short_buf)
|
|
|
|
def test_json(self):
|
|
jsondict = self.it.to_jsondict()
|
|
it = pbb.itag.from_jsondict(jsondict['itag'])
|
|
eq_(str(self.it), str(it))
|