Commit d2bb24e5 authored by Alex Elder's avatar Alex Elder

rbd: use sizeof (object) instead of sizeof (type)

Fix a few spots in rbd_header_from_disk() to use sizeof (object)
rather than sizeof (type).  Use a local variable to record sizes
to shorten some lines and improve readability.
Signed-off-by: default avatarAlex Elder <elder@inktank.com>
Reviewed-by: default avatarJosh Durgin <josh.durgin@inktank.com>
parent d78fd7ae
...@@ -494,17 +494,21 @@ static int rbd_header_from_disk(struct rbd_image_header *header, ...@@ -494,17 +494,21 @@ static int rbd_header_from_disk(struct rbd_image_header *header,
u32 allocated_snaps) u32 allocated_snaps)
{ {
u32 snap_count; u32 snap_count;
size_t size;
if (!rbd_dev_ondisk_valid(ondisk)) if (!rbd_dev_ondisk_valid(ondisk))
return -ENXIO; return -ENXIO;
snap_count = le32_to_cpu(ondisk->snap_count); snap_count = le32_to_cpu(ondisk->snap_count);
if (snap_count > (SIZE_MAX - sizeof(struct ceph_snap_context))
/ sizeof (u64)) /* Make sure we don't overflow below */
size = SIZE_MAX - sizeof (struct ceph_snap_context);
if (snap_count > size / sizeof (header->snapc->snaps[0]))
return -EINVAL; return -EINVAL;
header->snapc = kmalloc(sizeof(struct ceph_snap_context) +
snap_count * sizeof(u64), size = sizeof (struct ceph_snap_context);
GFP_KERNEL); size += snap_count * sizeof (header->snapc->snaps[0]);
header->snapc = kmalloc(size, GFP_KERNEL);
if (!header->snapc) if (!header->snapc)
return -ENOMEM; return -ENOMEM;
...@@ -515,8 +519,8 @@ static int rbd_header_from_disk(struct rbd_image_header *header, ...@@ -515,8 +519,8 @@ static int rbd_header_from_disk(struct rbd_image_header *header,
GFP_KERNEL); GFP_KERNEL);
if (!header->snap_names) if (!header->snap_names)
goto err_snapc; goto err_snapc;
header->snap_sizes = kmalloc(snap_count * sizeof(u64), size = snap_count * sizeof (*header->snap_sizes);
GFP_KERNEL); header->snap_sizes = kmalloc(size, GFP_KERNEL);
if (!header->snap_sizes) if (!header->snap_sizes)
goto err_names; goto err_names;
} else { } else {
...@@ -526,14 +530,12 @@ static int rbd_header_from_disk(struct rbd_image_header *header, ...@@ -526,14 +530,12 @@ static int rbd_header_from_disk(struct rbd_image_header *header,
header->snap_sizes = NULL; header->snap_sizes = NULL;
} }
header->object_prefix = kmalloc(sizeof (ondisk->block_name) + 1, size = sizeof (ondisk->block_name) + 1;
GFP_KERNEL); header->object_prefix = kmalloc(size, GFP_KERNEL);
if (!header->object_prefix) if (!header->object_prefix)
goto err_sizes; goto err_sizes;
memcpy(header->object_prefix, ondisk->block_name, size - 1);
memcpy(header->object_prefix, ondisk->block_name, header->object_prefix[size - 1] = '\0';
sizeof(ondisk->block_name));
header->object_prefix[sizeof (ondisk->block_name)] = '\0';
header->image_size = le64_to_cpu(ondisk->image_size); header->image_size = le64_to_cpu(ondisk->image_size);
header->obj_order = ondisk->options.order; header->obj_order = ondisk->options.order;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment