#include <linux/module.h>
 #include "mlx5_core.h"
 
-int mlx5_cmd_query_adapter(struct mlx5_core_dev *dev)
+int mlx5_cmd_query_adapter(struct mlx5_core_dev *dev, u32 *out, int outlen)
 {
-       struct mlx5_cmd_query_adapter_mbox_out *out;
-       struct mlx5_cmd_query_adapter_mbox_in in;
+       u32 in[MLX5_ST_SZ_DW(query_adapter_in)];
+
+       memset(in, 0, sizeof(in));
+
+       MLX5_SET(query_adapter_in, in, opcode, MLX5_CMD_OP_QUERY_ADAPTER);
+
+       return mlx5_cmd_exec_check_status(dev, in, sizeof(in), out, outlen);
+}
+
+int mlx5_query_board_id(struct mlx5_core_dev *dev)
+{
+       u32 *out;
+       int outlen = MLX5_ST_SZ_BYTES(query_adapter_out);
        int err;
 
-       out = kzalloc(sizeof(*out), GFP_KERNEL);
+       out = kzalloc(outlen, GFP_KERNEL);
        if (!out)
                return -ENOMEM;
 
-       memset(&in, 0, sizeof(in));
-       in.hdr.opcode = cpu_to_be16(MLX5_CMD_OP_QUERY_ADAPTER);
-       err = mlx5_cmd_exec(dev, &in, sizeof(in), out, sizeof(*out));
+       err = mlx5_cmd_query_adapter(dev, out, outlen);
        if (err)
-               goto out_out;
+               goto out;
 
-       if (out->hdr.status) {
-               err = mlx5_cmd_status_to_err(&out->hdr);
-               goto out_out;
-       }
+       memcpy(dev->board_id,
+              MLX5_ADDR_OF(query_adapter_out, out,
+                           query_adapter_struct.vsd_contd_psid),
+              MLX5_FLD_SZ_BYTES(query_adapter_out,
+                                query_adapter_struct.vsd_contd_psid));
 
-       memcpy(dev->board_id, out->vsd_psid, sizeof(out->vsd_psid));
-
-out_out:
+out:
        kfree(out);
+       return err;
+}
+
+int mlx5_core_query_vendor_id(struct mlx5_core_dev *mdev, u32 *vendor_id)
+{
+       u32 *out;
+       int outlen = MLX5_ST_SZ_BYTES(query_adapter_out);
+       int err;
+
+       out = kzalloc(outlen, GFP_KERNEL);
+       if (!out)
+               return -ENOMEM;
 
+       err = mlx5_cmd_query_adapter(mdev, out, outlen);
+       if (err)
+               goto out;
+
+       *vendor_id = MLX5_GET(query_adapter_out, out,
+                             query_adapter_struct.ieee_vendor_id);
+out:
+       kfree(out);
        return err;
 }
+EXPORT_SYMBOL(mlx5_core_query_vendor_id);
 
 int mlx5_query_hca_caps(struct mlx5_core_dev *dev)
 {