admin/libiscsi
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Update iscsi-dd.c

Browse Source 
add check after malloc to avoid referencing an illegal pointer
This commit is contained in:
sallyjunjun
2022-05-31 14:20:43 +08:00
committed by Bart Van Assche
parent babb0c3d51
commit 8592dc5883
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
examples/iscsi-dd.c
View File

@@ -131,6 +131,11 @@ void read_cb(struct iscsi_context *iscsi, int status, void *command_data, void *
} }
wt = malloc(sizeof(struct write_task)); wt = malloc(sizeof(struct write_task));
if (wt == NULL) {
fprintf(stderr, "failed to alloc write task\n");
exit(10);
}
wt->rt = task; wt->rt = task;
wt->client = client; wt->client = client;
@@ -430,6 +435,11 @@ void cscd_ident_inq(struct iscsi_context *iscsi,
_tgt_desig->designator_type = tgt_desig->designator_type; _tgt_desig->designator_type = tgt_desig->designator_type;
_tgt_desig->designator_length = tgt_desig->designator_length; _tgt_desig->designator_length = tgt_desig->designator_length;
_tgt_desig->designator = malloc(tgt_desig->designator_length); _tgt_desig->designator = malloc(tgt_desig->designator_length);
if (_tgt_desig->designator == NULL) {
fprintf(stderr, "failed to alloc designator\n");
exit(10);
}
memcpy(_tgt_desig->designator, tgt_desig->designator, tgt_desig->designator_length); memcpy(_tgt_desig->designator, tgt_desig->designator, tgt_desig->designator_length);
scsi_free_scsi_task(task); scsi_free_scsi_task(task);