IpTest.java
3.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
/*
* Copyright 2016-present Open Networking Laboratory
*
* 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.
*/
package org.onlab.packet;
import org.junit.Before;
import org.junit.Test;
import java.nio.ByteBuffer;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.instanceOf;
import static org.hamcrest.Matchers.is;
/**
* Unit tests for IP class.
*/
public class IpTest {
private byte v4Version = 4;
private byte v4HeaderLength = 6;
private byte badVersion = 5;
private byte badHeaderLength = 6;
private static Data data;
private static UDP udp;
private byte[] v4HeaderBytes;
private byte[] v6HeaderBytes;
private byte[] badHeaderBytes;
@Before
public void setUp() throws Exception {
ByteBuffer v4bb = ByteBuffer.allocate(v4HeaderLength * 4);
v4bb.put((byte) ((v4Version & 0xf) << 4));
v4HeaderBytes = v4bb.array();
ByteBuffer badBb = ByteBuffer.allocate(badHeaderLength * 4);
badBb.put((byte) ((badVersion & 0xf) << 4));
badHeaderBytes = badBb.array();
data = new Data();
data.setData("testSerialize".getBytes());
udp = new UDP();
udp.setPayload(data);
byte[] bytePayload = udp.serialize();
byte[] byteHeader = {
(byte) 0x69, (byte) 0x31, (byte) 0x35, (byte) 0x79,
(byte) (bytePayload.length >> 8 & 0xff), (byte) (bytePayload.length & 0xff),
(byte) 0x11, (byte) 0x20,
(byte) 0x20, (byte) 0x01, (byte) 0x0f, (byte) 0x18, (byte) 0x01, (byte) 0x13, (byte) 0x02, (byte) 0x15,
(byte) 0xca, (byte) 0x2a, (byte) 0x14, (byte) 0xff, (byte) 0xfe, (byte) 0x35, (byte) 0x26, (byte) 0xce,
(byte) 0x20, (byte) 0x01, (byte) 0x0f, (byte) 0x18, (byte) 0x01, (byte) 0x13, (byte) 0x02, (byte) 0x15,
(byte) 0xe6, (byte) 0xce, (byte) 0x8f, (byte) 0xff, (byte) 0xfe, (byte) 0x54, (byte) 0x37, (byte) 0xc8,
};
v6HeaderBytes = new byte[byteHeader.length + bytePayload.length];
System.arraycopy(byteHeader, 0, v6HeaderBytes, 0, byteHeader.length);
System.arraycopy(bytePayload, 0, v6HeaderBytes, byteHeader.length, bytePayload.length);
}
@Test
public void testDeserialize() throws Exception {
Deserializer ipDeserializer = IP.deserializer();
IPacket v4Packet = ipDeserializer.deserialize(v4HeaderBytes, 0, v4HeaderLength * 4);
IPacket v6Packet = ipDeserializer.deserialize(v6HeaderBytes, 0, v6HeaderBytes.length);
assertThat(v4Packet, is(instanceOf(IPv4.class)));
assertThat(v6Packet, is(instanceOf(IPv6.class)));
IPv6 ipv6 = (IPv6) v6Packet;
assertThat(ipv6.getVersion(), is((byte) 6));
assertThat(ipv6.getTrafficClass(), is((byte) 0x93));
assertThat(ipv6.getFlowLabel(), is(0x13579));
assertThat(ipv6.getNextHeader(), is(IPv6.PROTOCOL_UDP));
assertThat(ipv6.getHopLimit(), is((byte) 32));
}
@Test(expected = DeserializationException.class)
public void testBadIpVersion() throws Exception {
Deserializer ipDeserializer = IP.deserializer();
ipDeserializer.deserialize(badHeaderBytes, 0, badHeaderLength * 4);
}
}