TCP/UDP checksum in imported PCAP fix

Author: EgorBlagov

src/main/java/com/exalttech/trex/ui/views/importPcap/PacketUpdater.java

@@ -73,11 +73,32 @@ public Packet updatePacketSrcDst(Packet packet) {
             }
             packet = updateSrcAddress(packet);
             packet = updateDstAddress(packet);
+            packet = fixL3L4Checksums(packet);
         }
 
         return packet;
     }
 
+    private Packet fixL3L4Checksums(Packet packet) {
+        Packet.Builder builder = packet.getBuilder();
+        builder.get(IpV4Packet.Builder.class).correctChecksumAtBuild(true);
+
+        TcpPacket.Builder tcpBuilder = builder.get(TcpPacket.Builder.class);
+        if (tcpBuilder != null) {
+            tcpBuilder.srcAddr(packet.get(IpV4Packet.class).getHeader().getSrcAddr());
+            tcpBuilder.dstAddr(packet.get(IpV4Packet.class).getHeader().getDstAddr());
+            tcpBuilder.correctChecksumAtBuild(true);
+        }
+
+        UdpPacket.Builder udpBuilder = builder.get(UdpPacket.Builder.class);
+        if (udpBuilder != null) {
+            udpBuilder.srcAddr(packet.get(IpV4Packet.class).getHeader().getSrcAddr());
+            udpBuilder.dstAddr(packet.get(IpV4Packet.class).getHeader().getDstAddr());
+            udpBuilder.correctChecksumAtBuild(true);
+        }
+        return builder.build();
+    }
+
     /**
      * Initialize default source/destination address according to first packet
      *
@@ -164,7 +185,6 @@ private Packet updateSrcAddress(Packet packet) {
                 } else {
                     builder.get(IpV4Packet.Builder.class).dstAddr(modifiedAddress);
                 }
-                builder.get(IpV4Packet.Builder.class).correctChecksumAtBuild(true);
                 packet = builder.build();
             }
 
@@ -187,7 +207,6 @@ private Packet updateDstAddress(Packet packet) {
                 } else {
                     builder.get(IpV4Packet.Builder.class).srcAddr(modifiedAddress);
                 }
-                builder.get(IpV4Packet.Builder.class).correctChecksumAtBuild(true);
                 packet = builder.build();
             }
         } catch (Exception ex) {

src/main/java/com/exalttech/trex/ui/views/streams/builder/VMInstructionBuilder.java

@@ -26,7 +26,20 @@
  * @author Georgekh
  */
 public class VMInstructionBuilder {
+    public enum ChecksumFixHwCs {
+        L4_TYPE_UDP(11),
+        L4_TYPE_TCP(13),
+        L4_TYPE_IP(17);
+
+        private int code;
+        ChecksumFixHwCs(int code) {
+            this.code = code;
+        }
 
+        public int getCode() {
+            return this.code;
+        }
+    }
     boolean isAddFixIPV4Checksum = false;
     String splitByVar = "";
     int vmCacheSize = 0;
@@ -142,8 +155,12 @@ private void updateVMCacheSize(String count) {
     public List<Object> addChecksumInstruction() {
         ArrayList<Object> vmInstructionList = new ArrayList<>();
         LinkedHashMap<String, Object> checksumInstruction = new LinkedHashMap<>();
-        checksumInstruction.put("pkt_offset", isTaggedVlan ? 18 : 14);
-        checksumInstruction.put("type", "fix_checksum_ipv4");
+        int l3Offset = isTaggedVlan ? 18 : 14;
+        int l4Offset = l3Offset + 20;
+        checksumInstruction.put("type", "fix_checksum_hw");
+        checksumInstruction.put("l2_len", l3Offset);
+        checksumInstruction.put("l3_len", l4Offset);
+        checksumInstruction.put("l4_type", ChecksumFixHwCs.L4_TYPE_IP.getCode());
         vmInstructionList.add(checksumInstruction);
 
         return vmInstructionList;