ICMP fix and Packet class should store and provide access to the raw payload in case...
[controller.git] / opendaylight / sal / api / src / test / java / org / opendaylight / controller / sal / packet / IPv4Test.java
index 43118ec0dd1f642f011915fde396688e5f95ace2..429272f22769dc2ef36a17510cfe09910189a607 100644 (file)
@@ -1,4 +1,3 @@
-
 /*
  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
  *
@@ -9,12 +8,16 @@
 
 package org.opendaylight.controller.sal.packet;
 
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+import java.util.Arrays;
+
 import junit.framework.Assert;
 
 import org.junit.Test;
-import org.opendaylight.controller.sal.packet.ICMP;
-import org.opendaylight.controller.sal.packet.IPv4;
-import org.opendaylight.controller.sal.packet.Packet;
+import org.opendaylight.controller.sal.utils.EtherTypes;
+import org.opendaylight.controller.sal.utils.IPProtocols;
+import org.opendaylight.controller.sal.utils.NetUtils;
 
 public class IPv4Test {
 
@@ -51,7 +54,6 @@ public class IPv4Test {
         byte[] iptotLength = { 3, -24 };
         ip.hdrFieldsMap.put("TotalLength", iptotLength);
         short totalLength = ip.getTotalLength();
-        //System.out.println(totalLength);
         Assert.assertTrue(totalLength == 1000);
     }
 
@@ -91,7 +93,6 @@ public class IPv4Test {
         Assert.assertTrue(protocol == 1);
 
         Class<? extends Packet> clazz = IPv4.protocolClassMap.get(protocol);
-        System.out.printf("clazz = %s\n", clazz.getName());
         Assert.assertTrue(clazz == ICMP.class);
     }
 
@@ -157,6 +158,11 @@ public class IPv4Test {
         byte[] totalLength = ip.hdrFieldsMap.get("TotalLength");
         Assert.assertTrue(totalLength[0] == 3);
         Assert.assertTrue(totalLength[1] == -24);
+
+        ip.setTotalLength((short)84);
+        totalLength = ip.hdrFieldsMap.get("TotalLength");
+        Assert.assertTrue(totalLength[0] == 0);
+        Assert.assertTrue(totalLength[1] == 84);
     }
 
     @Test
@@ -206,7 +212,6 @@ public class IPv4Test {
         Assert.assertTrue(fragmentOffset[1] == -35);
     }
 
-
     @Test
     public void testSetDestinationAddress() {
         IPv4 ip = new IPv4();
@@ -219,4 +224,112 @@ public class IPv4Test {
         Assert.assertTrue(destinationAddress[3] == 110);
     }
 
+    @Test
+    public void testChecksum() {
+        byte header[] = { (byte) 0x45, 00, 00, (byte) 0x3c, (byte) 0x1c,
+                (byte) 0x46, (byte) 0x40, 00, (byte) 0x40, 06, (byte) 0xb1,
+                (byte) 0xe6, (byte) 0xac, (byte) 0x10, (byte) 0x0a,
+                (byte) 0x63, (byte) 0xac, (byte) 0x10, (byte) 0x0a, (byte) 0x0c };
+        byte header2[] = { (byte) 0x45, 00, 00, (byte) 0x73, 00, 00,
+                (byte) 0x40, 00, (byte) 0x40, (byte) 0x11, (byte) 0xb8,
+                (byte) 0x61, (byte) 0xc0, (byte) 0xa8, 00, 01, (byte) 0xc0,
+                (byte) 0xa8, 00, (byte) 0xc7 };
+        byte header3[] = { (byte) 0x45, 00, 00, (byte) 0x47, (byte) 0x73,
+                (byte) 0x88, (byte) 0x40, 00, (byte) 0x40, 06, (byte) 0xA2,
+                (byte) 0xC4, (byte) 0x83, (byte) 0x9F, (byte) 0x0E,
+                (byte) 0x85, (byte) 0x83, (byte) 0x9F, (byte) 0x0E, (byte) 0xA1 };
+        byte header4[] = { (byte) 0x45, 00, 00, (byte) 0x54, 00, 00,
+                (byte) 0x40, 00, (byte) 0x40, 01, (byte) 0xf0, (byte) 0x8e,
+                (byte) 0xc0, (byte) 0xa8, (byte) 0x64, (byte) 0x65,
+                (byte) 0xc0, (byte) 0xa8, (byte) 0x64, (byte) 0x64 };
+        byte header5[] = { (byte) 0x45, 00, 00, (byte) 0x54, 00, 00,
+                (byte) 0x40, 00, (byte) 0x40, 01, (byte) 0xef, (byte) 0x8d,
+                (byte) 0xc0, (byte) 0xa8, (byte) 0x64, (byte) 0x65,
+                (byte) 0xc0, (byte) 0xa8, (byte) 0x65, (byte) 0x65 };
+        byte header6[] = { (byte) 0x45, 00, 00, (byte) 0x54, 00, 00,
+                (byte) 0x40, 00, (byte) 0x40, 01, (byte) 0x0b, (byte) 0x92,
+                (byte) 0xc0, (byte) 0xa8, (byte) 0x64, (byte) 0x65, (byte) 0x9,
+                (byte) 0x9, (byte) 0x1, (byte) 0x1 };
+        byte header7[] = { (byte) 0x45, 00, 00, (byte) 0x54, 00, 00,
+                (byte) 0x40, 00, (byte) 0x40, 01, (byte) 0, (byte) 0,
+                (byte) 0xc0, (byte) 0xa8, (byte) 0x64, (byte) 0x65, (byte) 0x9,
+                (byte) 0x9, (byte) 0x2, (byte) 0x2 };
+
+        IPv4 ip = new IPv4();
+
+        Assert.assertTrue(NetUtils.getUnsignedShort(ip.computeChecksum(header,
+                0)) == 0xB1E6);
+        Assert.assertTrue(NetUtils.getUnsignedShort(ip.computeChecksum(header2,
+                0)) == 0xb861);
+        Assert.assertTrue(NetUtils.getUnsignedShort(ip.computeChecksum(header3,
+                0)) == 0xa2c4);
+        Assert.assertTrue(NetUtils.getUnsignedShort(ip.computeChecksum(header4,
+                0)) == 0xf08e);
+        Assert.assertTrue(NetUtils.getUnsignedShort(ip.computeChecksum(header5,
+                0)) == 0xef8d);
+        Assert.assertTrue(NetUtils.getUnsignedShort(ip.computeChecksum(header6,
+                0)) == 0x0b92);
+        Assert.assertTrue(NetUtils.getUnsignedShort(ip.computeChecksum(header7,
+                0)) == 0x0a91);
+    }
+
+    @Test
+    public void testFullIP() throws UnknownHostException, PacketException {
+        byte[] icmpRawPayload = new byte[] { (byte) 0x38, (byte) 0x26,
+                (byte) 0x9e, (byte) 0x51, (byte) 0x00, (byte) 0x00,
+                (byte) 0x00, (byte) 0x00, (byte) 0x2e, (byte) 0x6a,
+                (byte) 0x08, (byte) 0x00, (byte) 0x00, (byte) 0x00,
+                (byte) 0x00, (byte) 0x00, (byte) 0x10, (byte) 0x11,
+                (byte) 0x12, (byte) 0x13, (byte) 0x14, (byte) 0x15,
+                (byte) 0x16, (byte) 0x17, (byte) 0x18, (byte) 0x19,
+                (byte) 0x1a, (byte) 0x1b, (byte) 0x1c, (byte) 0x1d,
+                (byte) 0x1e, (byte) 0x1f, (byte) 0x20, (byte) 0x21,
+                (byte) 0x22, (byte) 0x23, (byte) 0x24, (byte) 0x25,
+                (byte) 0x26, (byte) 0x27, (byte) 0x28, (byte) 0x29,
+                (byte) 0x2a, (byte) 0x2b, (byte) 0x2c, (byte) 0x2d,
+                (byte) 0x2e, (byte) 0x2f, (byte) 0x30, (byte) 0x31,
+                (byte) 0x32, (byte) 0x33, (byte) 0x34, (byte) 0x35,
+                (byte) 0x36, (byte) 0x37 };
+        ICMP icmp = new ICMP();
+        icmp.setType((byte) 8);
+        icmp.setCode((byte) 0);
+        icmp.setIdentifier((short) 0x46f5);
+        icmp.setSequenceNumber((short) 2);
+        icmp.setRawPayload(icmpRawPayload);
+
+        IPv4 ip = new IPv4();
+        ip.setVersion((byte) 4);
+        ip.setIdentification((short) 5);
+        ip.setDiffServ((byte) 0);
+        ip.setECN((byte) 0);
+        ip.setTotalLength((short) 84);
+        ip.setFlags((byte) 2);
+        ip.setFragmentOffset((short) 0);
+        ip.setTtl((byte) 64);
+        ip.setProtocol(IPProtocols.ICMP.byteValue());
+        ip.setDestinationAddress(InetAddress.getByName("192.168.100.100"));
+        ip.setSourceAddress(InetAddress.getByName("192.168.100.101"));
+        ip.setPayload(icmp);
+
+        Ethernet eth = new Ethernet();
+        eth.setDestinationMACAddress(new byte[] { (byte) 0x98, (byte) 0xfc,
+                (byte) 0x11, (byte) 0x93, (byte) 0x5c, (byte) 0xb8 });
+        eth.setSourceMACAddress(new byte[] { (byte) 0x00, (byte) 0x24,
+                (byte) 0xd7, (byte) 0xa9, (byte) 0xa3, (byte) 0x50 });
+        eth.setEtherType(EtherTypes.IPv4.shortValue());
+        eth.setPayload(ip);
+
+        byte[] stream = eth.serialize();
+
+        Ethernet decEth = new Ethernet();
+        decEth.deserialize(stream, 0, stream.length * NetUtils.NumBitsInAByte);
+
+        IPv4 decIp = (IPv4) decEth.getPayload();
+        Assert.assertFalse(decIp.isCorrupted());
+        Assert.assertTrue(ip.equals(decIp));
+
+        ICMP decIcmp = (ICMP) decIp.getPayload();
+        Assert.assertFalse(decIcmp.isCorrupted());
+        Assert.assertTrue(Arrays.equals(icmpRawPayload, decIcmp.getRawPayload()));
+    }
 }