#include <linux/netdevice.h>
 #include <linux/types.h>
 #include <linux/errno.h>
+#include <linux/rtnetlink.h>
+#include <net/dcbnl.h>
 
 #include "bnx2x.h"
 #include "bnx2x_cmn.h"
 #include "bnx2x_dcb.h"
 
-#ifdef BCM_DCBNL
-#include <linux/rtnetlink.h>
-#endif
-
 /* forward declarations of dcbx related functions */
 static int bnx2x_dcbx_stop_hw_tx(struct bnx2x *bp);
 static void bnx2x_pfc_set_pfc(struct bnx2x *bp);
        }
 }
 
+/* maps unmapped priorities to to the same COS as L2 */
+static void bnx2x_dcbx_map_nw(struct bnx2x *bp)
+{
+       int i;
+       u32 unmapped = (1 << MAX_PFC_PRIORITIES) - 1; /* all ones */
+       u32 *ttp = bp->dcbx_port_params.app.traffic_type_priority;
+       u32 nw_prio = 1 << ttp[LLFC_TRAFFIC_TYPE_NW];
+       struct bnx2x_dcbx_cos_params *cos_params =
+                       bp->dcbx_port_params.ets.cos_params;
+
+       /* get unmapped priorities by clearing mapped bits */
+       for (i = 0; i < LLFC_DRIVER_TRAFFIC_TYPE_MAX; i++)
+               unmapped &= ~(1 << ttp[i]);
+
+       /* find cos for nw prio and extend it with unmapped */
+       for (i = 0; i < ARRAY_SIZE(bp->dcbx_port_params.ets.cos_params); i++) {
+               if (cos_params[i].pri_bitmask & nw_prio) {
+                       /* extend the bitmask with unmapped */
+                       DP(NETIF_MSG_LINK,
+                          "cos %d extended with 0x%08x", i, unmapped);
+                       cos_params[i].pri_bitmask |= unmapped;
+                       break;
+               }
+       }
+}
+
 static void bnx2x_get_dcbx_drv_param(struct bnx2x *bp,
                                     struct dcbx_features *features,
                                     u32 error)
        bnx2x_dcbx_get_pfc_feature(bp, &features->pfc, error);
 
        bnx2x_dcbx_get_ets_feature(bp, &features->ets, error);
+
+       bnx2x_dcbx_map_nw(bp);
 }
 
 #define DCBX_LOCAL_MIB_MAX_TRY_READ            (100)
                        if (bp->dcbx_port_params.ets.cos_params[cos].pri_bitmask
                            & (1 << prio)) {
                                bp->prio_to_cos[prio] = cos;
+                               DP(NETIF_MSG_LINK,
+                                  "tx_mapping %d --> %d\n", prio, cos);
                        }
                }
        }
                DP(NETIF_MSG_LINK, "BNX2X_DCBX_STATE_TX_RELEASED\n");
                bnx2x_fw_command(bp, DRV_MSG_CODE_DCBX_PMF_DRV_OK, 0);
 #ifdef BCM_DCBNL
-               /**
+               /*
                 * Send a notification for the new negotiated parameters
                 */
                dcbnl_cee_notify(bp->dev, RTM_GETDCB, DCB_CMD_CEE_GET, 0, 0);
                                                          pri_join_mask,
                                                          num_of_dif_pri);
 
-
        for (i = 0; i < cos_data.num_of_cos ; i++) {
                struct bnx2x_dcbx_cos_params *p =
                        &bp->dcbx_port_params.ets.cos_params[i];