X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fsal%2Fapi%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fsal%2Fpacket%2FICMP.java;h=9bdb5d084c57b7fede9fe1557a4934ae31e981e4;hb=ff1b4a79cca00743a00c3b0b1100bd0ab2b2fb31;hp=d1f81c5775f5d9952d9ef1a2fcde2f22248de11f;hpb=42210c03b0a4c54706320ba9f55794c0abd4d201;p=controller.git

diff --git a/opendaylight/sal/api/src/main/java/org/opendaylight/controller/sal/packet/ICMP.java b/opendaylight/sal/api/src/main/java/org/opendaylight/controller/sal/packet/ICMP.java
index d1f81c5775..9bdb5d084c 100644
--- a/opendaylight/sal/api/src/main/java/org/opendaylight/controller/sal/packet/ICMP.java
+++ b/opendaylight/sal/api/src/main/java/org/opendaylight/controller/sal/packet/ICMP.java
@@ -13,34 +13,29 @@ import java.util.HashMap;
 import java.util.LinkedHashMap;
 import java.util.Map;
 
-import org.apache.commons.lang3.builder.EqualsBuilder;
-import org.apache.commons.lang3.builder.HashCodeBuilder;
 import org.apache.commons.lang3.tuple.ImmutablePair;
 import org.apache.commons.lang3.tuple.Pair;
+import org.opendaylight.controller.sal.utils.NetUtils;
 
 /**
  * Class that represents the ICMP packet objects
- *
- *
  */
 
 public class ICMP extends Packet {
-    private static final String TYPECODE = "TypeCode";
+    private static final String TYPE = "Type";
     private static final String CODE = "Code";
-    private static final String HEADERCHECKSUM = "HeaderChecksum";
+    private static final String CHECKSUM = "Checksum";
     private static final String IDENTIFIER = "Identifier";
     private static final String SEQNUMBER = "SequenceNumber";
 
     private static Map<String, Pair<Integer, Integer>> fieldCoordinates = new LinkedHashMap<String, Pair<Integer, Integer>>() {
         private static final long serialVersionUID = 1L;
-
         {
-            put(TYPECODE, new ImmutablePair<Integer, Integer>(0, 8));
+            put(TYPE, new ImmutablePair<Integer, Integer>(0, 8));
             put(CODE, new ImmutablePair<Integer, Integer>(8, 8));
-            put(HEADERCHECKSUM, new ImmutablePair<Integer, Integer>(16, 16));
+            put(CHECKSUM, new ImmutablePair<Integer, Integer>(16, 16));
             put(IDENTIFIER, new ImmutablePair<Integer, Integer>(32, 16));
             put(SEQNUMBER, new ImmutablePair<Integer, Integer>(48, 16));
-
         }
     };
 
@@ -64,7 +59,7 @@ public class ICMP extends Packet {
         hdrFieldsMap = fieldValues;
     }
 
-    private Map<String, byte[]> fieldValues;
+    private final Map<String, byte[]> fieldValues;
 
     @Override
     public void setHeaderField(String headerField, byte[] readValue) {
@@ -72,13 +67,28 @@ public class ICMP extends Packet {
     }
 
     /**
-     * Sets the TypeCode of ICMP  for the current ICMP object instance
-     * @param short - typeCode
-     * @return ICMP
+     * Sets the type for the current ICMP message
+     *
+     * @param type
+     *            The ICMP message type
+     * @return This ICMP object
+     */
+    public ICMP setType(byte type) {
+        byte[] icmpType = BitBufferHelper.toByteArray(type);
+        fieldValues.put(TYPE, icmpType);
+        return this;
+    }
+
+    /**
+     * Sets the ICMP code (type subtype) for the current ICMP object instance
+     *
+     * @param code
+     *            The ICMP message type subtype
+     * @return This ICMP object
      */
-    public ICMP setTypeCode(short typeCode) {
-        byte[] icmpTypeCode = BitBufferHelper.toByteArray(typeCode);
-        fieldValues.put(TYPECODE, icmpTypeCode);
+    public ICMP setCode(byte code) {
+        byte[] icmpCode = BitBufferHelper.toByteArray(code);
+        fieldValues.put(CODE, icmpCode);
         return this;
     }
 
@@ -89,7 +99,7 @@ public class ICMP extends Packet {
      */
     public ICMP setChecksum(short checksum) {
         byte[] icmpChecksum = BitBufferHelper.toByteArray(checksum);
-        fieldValues.put(HEADERCHECKSUM, icmpChecksum);
+        fieldValues.put(CHECKSUM, icmpChecksum);
         return this;
     }
 
@@ -115,13 +125,77 @@ public class ICMP extends Packet {
         return this;
     }
 
+    /**
+     * Gets the header size in bits
+     * @return The ICMP header size in bits
+     */
     @Override
-    public int hashCode() {
-        return HashCodeBuilder.reflectionHashCode(this);
+    public int getHeaderSize() {
+        return 64;
+    }
+
+    /**
+     * Computes the ICMP checksum on the serialized ICMP message
+     *
+     * @param serialized
+     *            The data stream
+     * @param start
+     *            The byte index on the data stream from which the ICMP packet
+     *            starts
+     * @return The checksum
+     */
+    short computeChecksum(byte[] data, int start) {
+        int sum = 0, carry = 0, finalSum = 0;
+        int end = start + this.getHeaderSize() / NetUtils.NumBitsInAByte
+                + rawPayload.length;
+        int checksumStartByte = start + getfieldOffset(CHECKSUM)
+                / NetUtils.NumBitsInAByte;
+
+        for (int i = start; i <= (end - 1); i = i + 2) {
+            // Skip, if the current bytes are checkSum bytes
+            if (i == checksumStartByte) {
+                continue;
+            }
+            StringBuffer sbuffer = new StringBuffer();
+            sbuffer.append(String.format("%02X", data[i]));
+            if (i < (data.length - 1)) {
+                sbuffer.append(String.format("%02X", data[i + 1]));
+            }
+            sum += Integer.valueOf(sbuffer.toString(), 16);
+        }
+        carry = (sum >> 16) & 0xFF;
+        finalSum = (sum & 0xFFFF) + carry;
+        return (short) ~((short) finalSum & 0xFFFF);
+    }
+
+    @Override
+    protected void postSerializeCustomOperation(byte[] serializedBytes)
+            throws PacketException {
+        byte[] checkSum = BitBufferHelper
+                .toByteArray(computeChecksum(serializedBytes, 0));
+        try {
+            BitBufferHelper.setBytes(serializedBytes, checkSum,
+                    getfieldOffset(CHECKSUM), getfieldnumBits(CHECKSUM));
+        } catch (BufferException e) {
+            throw new PacketException(e.getMessage());
+        }
     }
 
     @Override
-    public boolean equals(Object obj) {
-        return EqualsBuilder.reflectionEquals(this, obj);
+    protected void postDeserializeCustomOperation(byte[] data, int endBitOffset) {
+        short computedChecksum = computeChecksum(data, endBitOffset / NetUtils.NumBitsInAByte);
+        short actualChecksum = BitBufferHelper.getShort(fieldValues.get(CHECKSUM));
+
+        if (computedChecksum != actualChecksum) {
+            corrupted = true;
+        }
+    }
+
+    /**
+     * Gets the checksum value stored
+     * @return the checksum
+     */
+    public short getChecksum() {
+        return (BitBufferHelper.getShort(fieldValues.get(CHECKSUM)));
     }
 }