X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Fsecurity.c;h=788426c25dd7446d8afefb4a903ae3b2e01f9538;hp=4cd898b5b4a3fa9e0cb86d3aa6f7af224f37e9fe;hb=12d5058b6fdcefbaef9d8d4a199d8bf5df3d5f30;hpb=55491147fce2bc03ffb602a3985e7fd4e32169a3

diff --git a/src/security.c b/src/security.c
index 4cd898b5..788426c2 100644
--- a/src/security.c
+++ b/src/security.c
@@ -28,7 +28,7 @@
 #endif
 
 #include "wimlib/assert.h"
-#include "wimlib/buffer_io.h"
+#include "wimlib/endianness.h"
 #include "wimlib/error.h"
 #include "wimlib/security.h"
 #include "wimlib/sha1.h"
@@ -146,7 +146,7 @@ struct wim_security_data_disk {
  * the validation in libntfs-3g.
  */
 static void
-empty_sacl_fixup(SECURITY_DESCRIPTOR_RELATIVE *descr, size_t *size_p)
+empty_sacl_fixup(SECURITY_DESCRIPTOR_RELATIVE *descr, u64 *size_p)
 {
 	/* No-op if no NTFS-3g support, or if NTFS-3g is version 2013 or later
 	 * */
@@ -271,10 +271,9 @@ read_wim_security_data(const u8 metadata_resource[], size_t metadata_resource_le
 		total_len += sd->sizes[i];
 		if (total_len > (u64)sd->total_length)
 			goto out_invalid_sd;
-		sd->descriptors[i] = MALLOC(sd->sizes[i]);
+		sd->descriptors[i] = memdup(p, sd->sizes[i]);
 		if (!sd->descriptors[i])
 			goto out_of_memory;
-		memcpy(sd->descriptors[i], p, sd->sizes[i]);
 		p += sd->sizes[i];
 		empty_sacl_fixup((SECURITY_DESCRIPTOR_RELATIVE*)sd->descriptors[i],
 				 &sd->sizes[i]);
@@ -569,13 +568,12 @@ sd_set_add_sd(struct wim_sd_set *sd_set, const char *descriptor, size_t size)
 	new = MALLOC(sizeof(*new));
 	if (!new)
 		goto out;
-	descr_copy = MALLOC(size);
+
+	descr_copy = memdup(descriptor, size);
 	if (!descr_copy)
 		goto out_free_node;
 
 	sd = sd_set->sd;
-
-	memcpy(descr_copy, descriptor, size);
 	new->security_id = sd->num_entries;
 	copy_hash(new->hash, hash);