example: provide an example of both the full and partial identify commands

This change adds a --partial argument to the mi-mctp command, so we have
an example of both the full and partial identify APIs.

Signed-off-by: Jeremy Kerr <[email protected]>

Author: jk-ozlabs

examples/mi-mctp.c

@@ -182,9 +182,10 @@ int do_identify(nvme_mi_ep_t ep, int argc, char **argv)
 	struct nvme_id_ctrl id;
 	uint16_t ctrl_id;
 	char buf[41];
+	bool partial;
 	int rc, tmp;
 
-	if (argc != 2) {
+	if (argc < 2) {
 		fprintf(stderr, "no controller ID specified\n");
 		return -1;
 	}
@@ -197,6 +198,8 @@ int do_identify(nvme_mi_ep_t ep, int argc, char **argv)
 
 	ctrl_id = tmp & 0xffff;
 
+	partial = argc > 2 && !strcmp(argv[2], "--partial");
+
 	ctrl = nvme_mi_init_ctrl(ep, tmp);
 	if (!ctrl) {
 		warn("can't create controller");
@@ -210,9 +213,18 @@ int do_identify(nvme_mi_ep_t ep, int argc, char **argv)
 	id_args.cntid = ctrl_id;
 	id_args.csi = NVME_CSI_NVM;
 
-	/* we only use the fields before rab; just request partial ID data */
-	rc = nvme_mi_admin_identify_partial(ctrl, &id_args, 0,
+	/* for this example code, we can either do a full or partial identify;
+	 * since we're only printing the fields before the 'rab' member,
+	 * these will be equivalent, aside from the size of the MI
+	 * response.
+	 */
+	if (partial) {
+		rc = nvme_mi_admin_identify_partial(ctrl, &id_args, 0,
 					    offsetof(struct nvme_id_ctrl, rab));
+	} else {
+		rc = nvme_mi_admin_identify(ctrl, &id_args);
+	}
+
 	if (rc) {
 		warn("can't perform Admin Identify command");
 		return -1;
@@ -249,7 +261,7 @@ int main(int argc, char **argv)
 		fprintf(stderr, "where action is:\n"
 			"  info\n"
 			"  controllers\n"
-			"  identify <controller-id>\n");
+			"  identify <controller-id> [--partial]\n");
 		return EXIT_FAILURE;
 	}