/* Mutex to hold while enabling or disabling RoCE */
 static DEFINE_MUTEX(mlx5_roce_en_lock);
 
-static int _mlx5_query_vport_state(struct mlx5_core_dev *mdev, u8 opmod,
-                                  u16 vport, u32 *out, int outlen)
+u8 mlx5_query_vport_state(struct mlx5_core_dev *mdev, u8 opmod, u16 vport)
 {
-       u32 in[MLX5_ST_SZ_DW(query_vport_state_in)] = {0};
+       u32 out[MLX5_ST_SZ_DW(query_vport_state_out)] = {};
+       u32 in[MLX5_ST_SZ_DW(query_vport_state_in)] = {};
+       int err;
 
        MLX5_SET(query_vport_state_in, in, opcode,
                 MLX5_CMD_OP_QUERY_VPORT_STATE);
        if (vport)
                MLX5_SET(query_vport_state_in, in, other_vport, 1);
 
-       return mlx5_cmd_exec(mdev, in, sizeof(in), out, outlen);
-}
-
-u8 mlx5_query_vport_state(struct mlx5_core_dev *mdev, u8 opmod, u16 vport)
-{
-       u32 out[MLX5_ST_SZ_DW(query_vport_state_out)] = {0};
-
-       _mlx5_query_vport_state(mdev, opmod, vport, out, sizeof(out));
+       err = mlx5_cmd_exec_inout(mdev, query_vport_state, in, out);
+       if (err)
+               return 0;
 
        return MLX5_GET(query_vport_state_out, out, state);
 }
 int mlx5_modify_vport_admin_state(struct mlx5_core_dev *mdev, u8 opmod,
                                  u16 vport, u8 other_vport, u8 state)
 {
-       u32 in[MLX5_ST_SZ_DW(modify_vport_state_in)]   = {0};
-       u32 out[MLX5_ST_SZ_DW(modify_vport_state_out)] = {0};
+       u32 in[MLX5_ST_SZ_DW(modify_vport_state_in)] = {};
 
        MLX5_SET(modify_vport_state_in, in, opcode,
                 MLX5_CMD_OP_MODIFY_VPORT_STATE);
        MLX5_SET(modify_vport_state_in, in, other_vport, other_vport);
        MLX5_SET(modify_vport_state_in, in, admin_state, state);
 
-       return mlx5_cmd_exec(mdev, in, sizeof(in), out, sizeof(out));
+       return mlx5_cmd_exec_in(mdev, modify_vport_state, in);
 }
 
 static int mlx5_query_nic_vport_context(struct mlx5_core_dev *mdev, u16 vport,
-                                       u32 *out, int outlen)
+                                       u32 *out)
 {
-       u32 in[MLX5_ST_SZ_DW(query_nic_vport_context_in)] = {0};
+       u32 in[MLX5_ST_SZ_DW(query_nic_vport_context_in)] = {};
 
        MLX5_SET(query_nic_vport_context_in, in, opcode,
                 MLX5_CMD_OP_QUERY_NIC_VPORT_CONTEXT);
        if (vport)
                MLX5_SET(query_nic_vport_context_in, in, other_vport, 1);
 
-       return mlx5_cmd_exec(mdev, in, sizeof(in), out, outlen);
-}
-
-static int mlx5_modify_nic_vport_context(struct mlx5_core_dev *mdev, void *in,
-                                        int inlen)
-{
-       u32 out[MLX5_ST_SZ_DW(modify_nic_vport_context_out)] = {0};
-
-       MLX5_SET(modify_nic_vport_context_in, in, opcode,
-                MLX5_CMD_OP_MODIFY_NIC_VPORT_CONTEXT);
-       return mlx5_cmd_exec(mdev, in, inlen, out, sizeof(out));
+       return mlx5_cmd_exec_inout(mdev, query_nic_vport_context, in, out);
 }
 
 int mlx5_query_nic_vport_min_inline(struct mlx5_core_dev *mdev,
                                    u16 vport, u8 *min_inline)
 {
-       u32 out[MLX5_ST_SZ_DW(query_nic_vport_context_out)] = {0};
+       u32 out[MLX5_ST_SZ_DW(query_nic_vport_context_out)] = {};
        int err;
 
-       err = mlx5_query_nic_vport_context(mdev, vport, out, sizeof(out));
+       err = mlx5_query_nic_vport_context(mdev, vport, out);
        if (!err)
                *min_inline = MLX5_GET(query_nic_vport_context_out, out,
                                       nic_vport_context.min_wqe_inline_mode);
 int mlx5_modify_nic_vport_min_inline(struct mlx5_core_dev *mdev,
                                     u16 vport, u8 min_inline)
 {
-       u32 in[MLX5_ST_SZ_DW(modify_nic_vport_context_in)] = {0};
-       int inlen = MLX5_ST_SZ_BYTES(modify_nic_vport_context_in);
+       u32 in[MLX5_ST_SZ_DW(modify_nic_vport_context_in)] = {};
        void *nic_vport_ctx;
 
        MLX5_SET(modify_nic_vport_context_in, in,
                                     in, nic_vport_context);
        MLX5_SET(nic_vport_context, nic_vport_ctx,
                 min_wqe_inline_mode, min_inline);
+       MLX5_SET(modify_nic_vport_context_in, in, opcode,
+                MLX5_CMD_OP_MODIFY_NIC_VPORT_CONTEXT);
 
-       return mlx5_modify_nic_vport_context(mdev, in, inlen);
+       return mlx5_cmd_exec_in(mdev, modify_nic_vport_context, in);
 }
 
 int mlx5_query_nic_vport_mac_address(struct mlx5_core_dev *mdev,
                                     u16 vport, bool other, u8 *addr)
 {
-       int outlen = MLX5_ST_SZ_BYTES(query_nic_vport_context_out);
+       u32 out[MLX5_ST_SZ_DW(query_nic_vport_context_out)] = {};
        u32 in[MLX5_ST_SZ_DW(query_nic_vport_context_in)] = {};
        u8 *out_addr;
-       u32 *out;
        int err;
 
-       out = kvzalloc(outlen, GFP_KERNEL);
-       if (!out)
-               return -ENOMEM;
-
        out_addr = MLX5_ADDR_OF(query_nic_vport_context_out, out,
                                nic_vport_context.permanent_address);
 
        MLX5_SET(query_nic_vport_context_in, in, vport_number, vport);
        MLX5_SET(query_nic_vport_context_in, in, other_vport, other);
 
-       err = mlx5_cmd_exec(mdev, in, sizeof(in), out, outlen);
+       err = mlx5_cmd_exec_inout(mdev, query_nic_vport_context, in, out);
        if (!err)
                ether_addr_copy(addr, &out_addr[2]);
 
-       kvfree(out);
        return err;
 }
 EXPORT_SYMBOL_GPL(mlx5_query_nic_vport_mac_address);
                                permanent_address);
 
        ether_addr_copy(&perm_mac[2], addr);
+       MLX5_SET(modify_nic_vport_context_in, in, opcode,
+                MLX5_CMD_OP_MODIFY_NIC_VPORT_CONTEXT);
 
-       err = mlx5_modify_nic_vport_context(mdev, in, inlen);
+       err = mlx5_cmd_exec_in(mdev, modify_nic_vport_context, in);
 
        kvfree(in);
 
        if (!out)
                return -ENOMEM;
 
-       err = mlx5_query_nic_vport_context(mdev, 0, out, outlen);
+       err = mlx5_query_nic_vport_context(mdev, 0, out);
        if (!err)
                *mtu = MLX5_GET(query_nic_vport_context_out, out,
                                nic_vport_context.mtu);
 
        MLX5_SET(modify_nic_vport_context_in, in, field_select.mtu, 1);
        MLX5_SET(modify_nic_vport_context_in, in, nic_vport_context.mtu, mtu);
+       MLX5_SET(modify_nic_vport_context_in, in, opcode,
+                MLX5_CMD_OP_MODIFY_NIC_VPORT_CONTEXT);
 
-       err = mlx5_modify_nic_vport_context(mdev, in, inlen);
+       err = mlx5_cmd_exec_in(mdev, modify_nic_vport_context, in);
 
        kvfree(in);
        return err;
                req_list_size = max_list_size;
        }
 
-       out_sz = MLX5_ST_SZ_BYTES(modify_nic_vport_context_in) +
+       out_sz = MLX5_ST_SZ_BYTES(query_nic_vport_context_in) +
                        req_list_size * MLX5_ST_SZ_BYTES(mac_address_layout);
 
        out = kzalloc(out_sz, GFP_KERNEL);
                                   u8 addr_list[][ETH_ALEN],
                                   int list_size)
 {
-       u32 out[MLX5_ST_SZ_DW(modify_nic_vport_context_out)];
+       u32 out[MLX5_ST_SZ_DW(modify_nic_vport_context_out)] = {};
        void *nic_vport_ctx;
        int max_list_size;
        int in_sz;
        in_sz = MLX5_ST_SZ_BYTES(modify_nic_vport_context_in) +
                list_size * MLX5_ST_SZ_BYTES(mac_address_layout);
 
-       memset(out, 0, sizeof(out));
        in = kzalloc(in_sz, GFP_KERNEL);
        if (!in)
                return -ENOMEM;
        if (!out)
                return -ENOMEM;
 
-       mlx5_query_nic_vport_context(mdev, 0, out, outlen);
+       mlx5_query_nic_vport_context(mdev, 0, out);
 
        *system_image_guid = MLX5_GET64(query_nic_vport_context_out, out,
                                        nic_vport_context.system_image_guid);
        if (!out)
                return -ENOMEM;
 
-       mlx5_query_nic_vport_context(mdev, 0, out, outlen);
+       mlx5_query_nic_vport_context(mdev, 0, out);
 
        *node_guid = MLX5_GET64(query_nic_vport_context_out, out,
                                nic_vport_context.node_guid);
        nic_vport_context = MLX5_ADDR_OF(modify_nic_vport_context_in,
                                         in, nic_vport_context);
        MLX5_SET64(nic_vport_context, nic_vport_context, node_guid, node_guid);
+       MLX5_SET(modify_nic_vport_context_in, in, opcode,
+                MLX5_CMD_OP_MODIFY_NIC_VPORT_CONTEXT);
 
-       err = mlx5_modify_nic_vport_context(mdev, in, inlen);
+       err = mlx5_cmd_exec_in(mdev, modify_nic_vport_context, in);
 
        kvfree(in);
 
        if (!out)
                return -ENOMEM;
 
-       mlx5_query_nic_vport_context(mdev, 0, out, outlen);
+       mlx5_query_nic_vport_context(mdev, 0, out);
 
        *qkey_viol_cntr = MLX5_GET(query_nic_vport_context_out, out,
                                   nic_vport_context.qkey_violation_counter);
                                 struct mlx5_hca_vport_context *rep)
 {
        int out_sz = MLX5_ST_SZ_BYTES(query_hca_vport_context_out);
-       int in[MLX5_ST_SZ_DW(query_hca_vport_context_in)] = {0};
+       int in[MLX5_ST_SZ_DW(query_hca_vport_context_in)] = {};
        int is_group_manager;
        void *out;
        void *ctx;
        if (MLX5_CAP_GEN(dev, num_ports) == 2)
                MLX5_SET(query_hca_vport_context_in, in, port_num, port_num);
 
-       err = mlx5_cmd_exec(dev, in, sizeof(in), out,  out_sz);
+       err = mlx5_cmd_exec_inout(dev, query_hca_vport_context, in, out);
        if (err)
                goto ex;
 
        if (!out)
                return -ENOMEM;
 
-       err = mlx5_query_nic_vport_context(mdev, vport, out, outlen);
+       err = mlx5_query_nic_vport_context(mdev, vport, out);
        if (err)
                goto out;
 
                 nic_vport_context.promisc_mc, promisc_mc);
        MLX5_SET(modify_nic_vport_context_in, in,
                 nic_vport_context.promisc_all, promisc_all);
+       MLX5_SET(modify_nic_vport_context_in, in, opcode,
+                MLX5_CMD_OP_MODIFY_NIC_VPORT_CONTEXT);
 
-       err = mlx5_modify_nic_vport_context(mdev, in, inlen);
+       err = mlx5_cmd_exec_in(mdev, modify_nic_vport_context, in);
 
        kvfree(in);
 
        if (MLX5_CAP_GEN(mdev, disable_local_lb_uc))
                MLX5_SET(modify_nic_vport_context_in, in,
                         field_select.disable_uc_local_lb, 1);
+       MLX5_SET(modify_nic_vport_context_in, in, opcode,
+                MLX5_CMD_OP_MODIFY_NIC_VPORT_CONTEXT);
 
-       err = mlx5_modify_nic_vport_context(mdev, in, inlen);
+       err = mlx5_cmd_exec_in(mdev, modify_nic_vport_context, in);
 
        if (!err)
                mlx5_core_dbg(mdev, "%s local_lb\n",
        if (!out)
                return -ENOMEM;
 
-       err = mlx5_query_nic_vport_context(mdev, 0, out, outlen);
+       err = mlx5_query_nic_vport_context(mdev, 0, out);
        if (err)
                goto out;
 
        MLX5_SET(modify_nic_vport_context_in, in, field_select.roce_en, 1);
        MLX5_SET(modify_nic_vport_context_in, in, nic_vport_context.roce_en,
                 state);
+       MLX5_SET(modify_nic_vport_context_in, in, opcode,
+                MLX5_CMD_OP_MODIFY_NIC_VPORT_CONTEXT);
 
-       err = mlx5_modify_nic_vport_context(mdev, in, inlen);
+       err = mlx5_cmd_exec_in(mdev, modify_nic_vport_context, in);
 
        kvfree(in);
 
        mutex_unlock(&mlx5_roce_en_lock);
        return err;
 }
-EXPORT_SYMBOL_GPL(mlx5_nic_vport_disable_roce);
+EXPORT_SYMBOL(mlx5_nic_vport_disable_roce);
 
 int mlx5_core_query_vport_counter(struct mlx5_core_dev *dev, u8 other_vport,
-                                 int vf, u8 port_num, void *out,
-                                 size_t out_sz)
+                                 int vf, u8 port_num, void *out)
 {
-       int     in_sz = MLX5_ST_SZ_BYTES(query_vport_counter_in);
-       int     is_group_manager;
-       void   *in;
-       int     err;
+       int in_sz = MLX5_ST_SZ_BYTES(query_vport_counter_in);
+       int is_group_manager;
+       void *in;
+       int err;
 
        is_group_manager = MLX5_CAP_GEN(dev, vport_group_manager);
        in = kvzalloc(in_sz, GFP_KERNEL);
        if (MLX5_CAP_GEN(dev, num_ports) == 2)
                MLX5_SET(query_vport_counter_in, in, port_num, port_num);
 
-       err = mlx5_cmd_exec(dev, in, in_sz, out,  out_sz);
+       err = mlx5_cmd_exec_inout(dev, query_vport_counter, in, out);
 free:
        kvfree(in);
        return err;
                                u8 other_vport, u64 *rx_discard_vport_down,
                                u64 *tx_discard_vport_down)
 {
-       u32 out[MLX5_ST_SZ_DW(query_vnic_env_out)] = {0};
-       u32 in[MLX5_ST_SZ_DW(query_vnic_env_in)] = {0};
+       u32 out[MLX5_ST_SZ_DW(query_vnic_env_out)] = {};
+       u32 in[MLX5_ST_SZ_DW(query_vnic_env_in)] = {};
        int err;
 
        MLX5_SET(query_vnic_env_in, in, opcode,
        MLX5_SET(query_vnic_env_in, in, vport_number, vport);
        MLX5_SET(query_vnic_env_in, in, other_vport, other_vport);
 
-       err = mlx5_cmd_exec(mdev, in, sizeof(in), out, sizeof(out));
+       err = mlx5_cmd_exec_inout(mdev, query_vnic_env, in, out);
        if (err)
                return err;
 
                                       struct mlx5_hca_vport_context *req)
 {
        int in_sz = MLX5_ST_SZ_BYTES(modify_hca_vport_context_in);
-       u8 out[MLX5_ST_SZ_BYTES(modify_hca_vport_context_out)];
        int is_group_manager;
+       void *ctx;
        void *in;
        int err;
-       void *ctx;
 
        mlx5_core_dbg(dev, "vf %d\n", vf);
        is_group_manager = MLX5_CAP_GEN(dev, vport_group_manager);
        if (!in)
                return -ENOMEM;
 
-       memset(out, 0, sizeof(out));
        MLX5_SET(modify_hca_vport_context_in, in, opcode, MLX5_CMD_OP_MODIFY_HCA_VPORT_CONTEXT);
        if (other_vport) {
                if (is_group_manager) {
        MLX5_SET(hca_vport_context, ctx, cap_mask1, req->cap_mask1);
        MLX5_SET(hca_vport_context, ctx, cap_mask1_field_select,
                 req->cap_mask1_perm);
-       err = mlx5_cmd_exec(dev, in, in_sz, out, sizeof(out));
+       err = mlx5_cmd_exec_in(dev, modify_hca_vport_context, in);
 ex:
        kfree(in);
        return err;
        MLX5_SET(modify_nic_vport_context_in, in,
                 nic_vport_context.affiliation_criteria,
                 MLX5_CAP_GEN(port_mdev, affiliate_nic_vport_criteria));
+       MLX5_SET(modify_nic_vport_context_in, in, opcode,
+                MLX5_CMD_OP_MODIFY_NIC_VPORT_CONTEXT);
 
-       err = mlx5_modify_nic_vport_context(port_mdev, in, inlen);
+       err = mlx5_cmd_exec_in(port_mdev, modify_nic_vport_context, in);
        if (err)
                mlx5_nic_vport_disable_roce(port_mdev);
 
                 nic_vport_context.affiliated_vhca_id, 0);
        MLX5_SET(modify_nic_vport_context_in, in,
                 nic_vport_context.affiliation_criteria, 0);
+       MLX5_SET(modify_nic_vport_context_in, in, opcode,
+                MLX5_CMD_OP_MODIFY_NIC_VPORT_CONTEXT);
 
-       err = mlx5_modify_nic_vport_context(port_mdev, in, inlen);
+       err = mlx5_cmd_exec_in(port_mdev, modify_nic_vport_context, in);
        if (!err)
                mlx5_nic_vport_disable_roce(port_mdev);
 
 {
        return MLX5_SPECIAL_VPORTS(dev) + mlx5_core_max_vfs(dev);
 }
-EXPORT_SYMBOL(mlx5_eswitch_get_total_vports);
+EXPORT_SYMBOL_GPL(mlx5_eswitch_get_total_vports);