X-Git-Url: https://wimlib.net/git/?a=blobdiff_plain;f=src%2Ftagged_items.c;h=723773f94be3e678afe6c0f88bae6a3355a961ce;hb=f9894b7e0e2a2a46fffdf7274d8671fb00f98089;hp=1ae227c85f551b6afb65a98aa9c6b694674fb9c3;hpb=fb9de673c23ece7e273d93a46c49e65cbe20a99a;p=wimlib

diff --git a/src/tagged_items.c b/src/tagged_items.c
index 1ae227c8..723773f9 100644
--- a/src/tagged_items.c
+++ b/src/tagged_items.c
@@ -74,8 +74,14 @@ inode_get_tagged_item(const struct wim_inode *inode,
 		      u32 desired_tag, u32 min_data_len)
 {
 	size_t minlen_with_hdr = sizeof(struct tagged_item_header) + min_data_len;
-	size_t len_remaining = inode->i_extra_size;
-	u8 *p = inode->i_extra;
+	size_t len_remaining;
+	u8 *p;
+
+	if (!inode->i_extra)
+		return NULL;
+
+	len_remaining = inode->i_extra->size;
+	p = inode->i_extra->data;
 
 	/* Iterate through the tagged items.  */
 	while (len_remaining >= minlen_with_hdr) {
@@ -85,7 +91,7 @@ inode_get_tagged_item(const struct wim_inode *inode,
 
 		hdr = (struct tagged_item_header *)p;
 		tag = le32_to_cpu(hdr->tag);
-		len = (le32_to_cpu(hdr->length) + 7) & ~7;
+		len = ALIGN(le32_to_cpu(hdr->length), 8);
 
 		/* Length overflow?  */
 		if (unlikely(len > len_remaining - sizeof(struct tagged_item_header)))
@@ -108,29 +114,31 @@ inode_add_tagged_item(struct wim_inode *inode, u32 tag, u32 len)
 {
 	size_t itemsize;
 	size_t newsize;
-	u8 *buf;
+	struct wim_inode_extra *extra;
 	struct tagged_item_header *hdr;
 
 	/* We prepend the item instead of appending it because it's easier.  */
 
-	itemsize = sizeof(struct tagged_item_header) + ((len + 7) & ~7);
-	newsize = itemsize + inode->i_extra_size;
+	itemsize = sizeof(struct tagged_item_header) + ALIGN(len, 8);
+	newsize = itemsize;
+	if (inode->i_extra)
+		newsize += inode->i_extra->size;
 
-	buf = MALLOC(newsize);
-	if (!buf)
+	extra = MALLOC(sizeof(struct wim_inode_extra) + newsize);
+	if (!extra)
 		return NULL;
-
-	if (inode->i_extra_size) {
-		memcpy(buf + itemsize, inode->i_extra, inode->i_extra_size);
+	if (inode->i_extra) {
+		memcpy(&extra->data[itemsize], inode->i_extra->data,
+		       inode->i_extra->size);
 		FREE(inode->i_extra);
 	}
-	inode->i_extra = buf;
-	inode->i_extra_size = newsize;
+	extra->size = newsize;
+	inode->i_extra = extra;
 
-	hdr = (struct tagged_item_header *)buf;
+	hdr = (struct tagged_item_header *)extra->data;
 	hdr->tag = cpu_to_le32(tag);
 	hdr->length = cpu_to_le32(len);
-	return memset(hdr->data, 0, (len + 7) & ~7);
+	return memset(hdr->data, 0, ALIGN(len, 8));
 }
 
 static inline struct wimlib_unix_data_disk *