#include "rvu_npc_fs.h"
 #include "rvu_npc_hash.h"
 
-#define NPC_BYTESM             GENMASK_ULL(19, 16)
-#define NPC_HDR_OFFSET         GENMASK_ULL(15, 8)
-#define NPC_KEY_OFFSET         GENMASK_ULL(5, 0)
-#define NPC_LDATA_EN           BIT_ULL(7)
-
 static const char * const npc_flow_names[] = {
        [NPC_DMAC]      = "dmac",
        [NPC_SMAC]      = "smac",
 static void npc_scan_ldata(struct rvu *rvu, int blkaddr, u8 lid,
                           u8 lt, u64 cfg, u8 intf)
 {
+       struct npc_mcam_kex_hash *mkex_hash = rvu->kpu.mkex_hash;
        struct npc_mcam *mcam = &rvu->hw->mcam;
        u8 hdr, key, nr_bytes, bit_offset;
        u8 la_ltype, la_start;
        NPC_SCAN_HDR(NPC_SIP_IPV4, NPC_LID_LC, NPC_LT_LC_IP, 12, 4);
        NPC_SCAN_HDR(NPC_DIP_IPV4, NPC_LID_LC, NPC_LT_LC_IP, 16, 4);
        NPC_SCAN_HDR(NPC_IPFRAG_IPV6, NPC_LID_LC, NPC_LT_LC_IP6_EXT, 6, 1);
-       NPC_SCAN_HDR(NPC_SIP_IPV6, NPC_LID_LC, NPC_LT_LC_IP6, 8, 16);
-       NPC_SCAN_HDR(NPC_DIP_IPV6, NPC_LID_LC, NPC_LT_LC_IP6, 24, 16);
+       if (rvu->hw->cap.npc_hash_extract) {
+               if (mkex_hash->lid_lt_ld_hash_en[intf][lid][lt][0])
+                       NPC_SCAN_HDR(NPC_SIP_IPV6, NPC_LID_LC, NPC_LT_LC_IP6, 8, 4);
+               else
+                       NPC_SCAN_HDR(NPC_SIP_IPV6, NPC_LID_LC, NPC_LT_LC_IP6, 8, 16);
+
+               if (mkex_hash->lid_lt_ld_hash_en[intf][lid][lt][1])
+                       NPC_SCAN_HDR(NPC_DIP_IPV6, NPC_LID_LC, NPC_LT_LC_IP6, 24, 4);
+               else
+                       NPC_SCAN_HDR(NPC_DIP_IPV6, NPC_LID_LC, NPC_LT_LC_IP6, 24, 16);
+       } else {
+               NPC_SCAN_HDR(NPC_SIP_IPV6, NPC_LID_LC, NPC_LT_LC_IP6, 8, 16);
+               NPC_SCAN_HDR(NPC_DIP_IPV6, NPC_LID_LC, NPC_LT_LC_IP6, 24, 16);
+       }
+
        NPC_SCAN_HDR(NPC_SPORT_UDP, NPC_LID_LD, NPC_LT_LD_UDP, 0, 2);
        NPC_SCAN_HDR(NPC_DPORT_UDP, NPC_LID_LD, NPC_LT_LD_UDP, 2, 2);
        NPC_SCAN_HDR(NPC_SPORT_TCP, NPC_LID_LD, NPC_LT_LD_TCP, 0, 2);
 
        return hash_out;
 }
 
-u32 npc_field_hash_calc(u64 *ldata, struct npc_mcam_kex_hash *mkex_hash,
-                       u64 *secret_key, u8 intf, u8 hash_idx)
+u32 npc_field_hash_calc(u64 *ldata, struct npc_get_field_hash_info_rsp rsp,
+                       u8 intf, u8 hash_idx)
 {
        u64 hash_key[3];
        u64 data_padded[2];
        u32 field_hash;
 
-       hash_key[0] = secret_key[1] << 31;
-       hash_key[0] |= secret_key[2];
-       hash_key[1] = secret_key[1] >> 33;
-       hash_key[1] |= secret_key[0] << 31;
-       hash_key[2] = secret_key[0] >> 33;
+       hash_key[0] = rsp.secret_key[1] << 31;
+       hash_key[0] |= rsp.secret_key[2];
+       hash_key[1] = rsp.secret_key[1] >> 33;
+       hash_key[1] |= rsp.secret_key[0] << 31;
+       hash_key[2] = rsp.secret_key[0] >> 33;
 
-       data_padded[0] = mkex_hash->hash_mask[intf][hash_idx][0] & ldata[0];
-       data_padded[1] = mkex_hash->hash_mask[intf][hash_idx][1] & ldata[1];
+       data_padded[0] = rsp.hash_mask[intf][hash_idx][0] & ldata[0];
+       data_padded[1] = rsp.hash_mask[intf][hash_idx][1] & ldata[1];
        field_hash = rvu_npc_toeplitz_hash(data_padded, hash_key, 128, 159);
 
-       field_hash &= mkex_hash->hash_ctrl[intf][hash_idx] >> 32;
-       field_hash |= mkex_hash->hash_ctrl[intf][hash_idx];
+       field_hash &= FIELD_GET(GENMASK(63, 32), rsp.hash_ctrl[intf][hash_idx]);
+       field_hash += FIELD_GET(GENMASK(31, 0), rsp.hash_ctrl[intf][hash_idx]);
        return field_hash;
 }
 
-static u64 npc_update_use_hash(int lt, int ld)
+static u64 npc_update_use_hash(struct rvu *rvu, int blkaddr,
+                              u8 intf, int lid, int lt, int ld)
 {
-       u64 cfg = 0;
-
-       switch (lt) {
-       case NPC_LT_LC_IP6:
-               /* Update use_hash(bit-20) and bytesm1 (bit-16:19)
-                * in KEX_LD_CFG
-                */
-               cfg = KEX_LD_CFG_USE_HASH(0x1, 0x03,
-                                         ld ? 0x18 : 0x8,
-                                         0x1, 0x0, ld ? 0x14 : 0x10);
-               break;
-       }
+       u8 hdr, key;
+       u64 cfg;
+
+       cfg = rvu_read64(rvu, blkaddr, NPC_AF_INTFX_LIDX_LTX_LDX_CFG(intf, lid, lt, ld));
+       hdr = FIELD_GET(NPC_HDR_OFFSET, cfg);
+       key = FIELD_GET(NPC_KEY_OFFSET, cfg);
+
+       /* Update use_hash(bit-20) to 'true' and
+        * bytesm1(bit-16:19) to '0x3' in KEX_LD_CFG
+        */
+       cfg = KEX_LD_CFG_USE_HASH(0x1, 0x03,
+                                 hdr, 0x1, 0x0, key);
 
        return cfg;
 }
                for (lt = 0; lt < NPC_MAX_LT; lt++) {
                        for (ld = 0; ld < NPC_MAX_LD; ld++) {
                                if (mkex_hash->lid_lt_ld_hash_en[intf][lid][lt][ld]) {
-                                       u64 cfg = npc_update_use_hash(lt, ld);
+                                       u64 cfg;
 
                                        if (hash_cnt == NPC_MAX_HASH)
                                                return;
 
+                                       cfg = npc_update_use_hash(rvu, blkaddr,
+                                                                 intf, lid, lt, ld);
                                        /* Set updated KEX configuration */
                                        SET_KEX_LD(intf, lid, lt, ld, cfg);
                                        /* Set HASH configuration */
                for (lt = 0; lt < NPC_MAX_LT; lt++) {
                        for (ld = 0; ld < NPC_MAX_LD; ld++)
                                if (mkex_hash->lid_lt_ld_hash_en[intf][lid][lt][ld]) {
-                                       u64 cfg = npc_update_use_hash(lt, ld);
+                                       u64 cfg;
 
                                        if (hash_cnt == NPC_MAX_HASH)
                                                return;
 
+                                       cfg = npc_update_use_hash(rvu, blkaddr,
+                                                                 intf, lid, lt, ld);
                                        /* Set updated KEX configuration */
                                        SET_KEX_LD(intf, lid, lt, ld, cfg);
                                        /* Set HASH configuration */
                                 * is hashed to 32 bit value.
                                 */
                                case NPC_LT_LC_IP6:
-                                       if (features & BIT_ULL(NPC_SIP_IPV6)) {
+                                       /* ld[0] == hash_idx[0] == Source IPv6
+                                        * ld[1] == hash_idx[1] == Destination IPv6
+                                        */
+                                       if ((features & BIT_ULL(NPC_SIP_IPV6)) && !hash_idx) {
                                                u32 src_ip[IPV6_WORDS];
 
                                                be32_to_cpu_array(src_ip, pkt->ip6src, IPV6_WORDS);
-                                               ldata[0] = (u64)src_ip[0] << 32 | src_ip[1];
-                                               ldata[1] = (u64)src_ip[2] << 32 | src_ip[3];
+                                               ldata[1] = (u64)src_ip[0] << 32 | src_ip[1];
+                                               ldata[0] = (u64)src_ip[2] << 32 | src_ip[3];
                                                field_hash = npc_field_hash_calc(ldata,
-                                                                                mkex_hash,
-                                                                                rsp.secret_key,
+                                                                                rsp,
                                                                                 intf,
                                                                                 hash_idx);
                                                npc_update_entry(rvu, NPC_SIP_IPV6, entry,
-                                                                field_hash, 0, 32, 0, intf);
+                                                                field_hash, 0,
+                                                                GENMASK(31, 0), 0, intf);
                                                memcpy(&opkt->ip6src, &pkt->ip6src,
                                                       sizeof(pkt->ip6src));
                                                memcpy(&omask->ip6src, &mask->ip6src,
                                                       sizeof(mask->ip6src));
-                                               break;
-                                       }
-
-                                       if (features & BIT_ULL(NPC_DIP_IPV6)) {
+                                       } else if ((features & BIT_ULL(NPC_DIP_IPV6)) && hash_idx) {
                                                u32 dst_ip[IPV6_WORDS];
 
                                                be32_to_cpu_array(dst_ip, pkt->ip6dst, IPV6_WORDS);
-                                               ldata[0] = (u64)dst_ip[0] << 32 | dst_ip[1];
-                                               ldata[1] = (u64)dst_ip[2] << 32 | dst_ip[3];
+                                               ldata[1] = (u64)dst_ip[0] << 32 | dst_ip[1];
+                                               ldata[0] = (u64)dst_ip[2] << 32 | dst_ip[3];
                                                field_hash = npc_field_hash_calc(ldata,
-                                                                                mkex_hash,
-                                                                                rsp.secret_key,
+                                                                                rsp,
                                                                                 intf,
                                                                                 hash_idx);
                                                npc_update_entry(rvu, NPC_DIP_IPV6, entry,
-                                                                field_hash, 0, 32, 0, intf);
+                                                                field_hash, 0,
+                                                                GENMASK(31, 0), 0, intf);
                                                memcpy(&opkt->ip6dst, &pkt->ip6dst,
                                                       sizeof(pkt->ip6dst));
                                                memcpy(&omask->ip6dst, &mask->ip6dst,
                                                       sizeof(mask->ip6dst));
                                        }
+
                                        break;
                                }
                        }