ivampiresp/Spdk
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

examples/nvme/reserve: add malloc checks and frees

Browse Source 
Change-Id: I576ac640c3c8a94237e5437e7e7f029d8526c071
Signed-off-by: Daniel Verkamp <daniel.verkamp@intel.com>
This commit is contained in:
Daniel Verkamp 2016-03-15 14:31:53 -07:00
parent e3eba25d92
commit 4fc355ba68
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/nvme/reserve/reservation.c
View File

@ -116,10 +116,15 @@ get_host_identifier(struct spdk_nvme_ctrlr *ctrlr)
outstanding_commands = 0; outstanding_commands = 0;
host_id = rte_malloc(NULL, 8, 0); host_id = rte_malloc(NULL, 8, 0);
if (host_id == NULL) {
fprintf(stderr, "host_id allocation failed\n");
return -1;
}
ret = spdk_nvme_ctrlr_cmd_admin_raw(ctrlr, &cmd, host_id, 8, ret = spdk_nvme_ctrlr_cmd_admin_raw(ctrlr, &cmd, host_id, 8,
get_feature_completion, &features[SPDK_NVME_FEAT_HOST_IDENTIFIER]); get_feature_completion, &features[SPDK_NVME_FEAT_HOST_IDENTIFIER]);
if (ret) { if (ret) {
fprintf(stdout, "Get Feature: Failed\n"); fprintf(stdout, "Get Feature: Failed\n");
rte_free(host_id);
return -1; return -1;
} }
@ -133,6 +138,7 @@ get_host_identifier(struct spdk_nvme_ctrlr *ctrlr)
fprintf(stdout, "Get Feature: Host Identifier 0x%"PRIx64"\n", *host_id); fprintf(stdout, "Get Feature: Host Identifier 0x%"PRIx64"\n", *host_id);
} }
rte_free(host_id);
return 0; return 0;
} }
@ -147,6 +153,10 @@ set_host_identifier(struct spdk_nvme_ctrlr *ctrlr)
cmd.cdw10 = SPDK_NVME_FEAT_HOST_IDENTIFIER; cmd.cdw10 = SPDK_NVME_FEAT_HOST_IDENTIFIER;
host_id = rte_malloc(NULL, 8, 0); host_id = rte_malloc(NULL, 8, 0);
if (host_id == NULL) {
fprintf(stderr, "host_id allocation failed\n");
return -1;
}
*host_id = HOST_ID; *host_id = HOST_ID;
outstanding_commands = 0; outstanding_commands = 0;