#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"
* 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
* */
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]);
"%u bytes, but calculated %u bytes",
sd->total_length, (unsigned)total_len);
}
-out_return_sd:
*sd_ret = sd;
ret = 0;
goto out;
u8 *orig_p = p;
struct wim_security_data_disk *sd_disk = (struct wim_security_data_disk*)p;
+ u32 num_entries = sd->num_entries;
sd_disk->total_length = cpu_to_le32(sd->total_length);
- sd_disk->num_entries = cpu_to_le32(sd->num_entries);
+ sd_disk->num_entries = cpu_to_le32(num_entries);
- for (u32 i = 0; i < sd->num_entries; i++)
+ for (u32 i = 0; i < num_entries; i++)
sd_disk->sizes[i] = cpu_to_le64(sd->sizes[i]);
- p = (u8*)&sd_disk->sizes[sd_disk->num_entries];
+ p = (u8*)&sd_disk->sizes[num_entries];
- for (u32 i = 0; i < sd->num_entries; i++)
+ for (u32 i = 0; i < num_entries; i++)
p = mempcpy(p, sd->descriptors[i], sd->sizes[i]);
- while (p - orig_p < sd->total_length)
+ while ((uintptr_t)p & 7)
*p++ = 0;
wimlib_assert(p - orig_p == sd->total_length);
- wimlib_assert(((uintptr_t)p & 7) == 0);
DEBUG("Successfully wrote security data.");
return p;
struct sd_node *new;
u8 **descriptors;
u64 *sizes;
- char *descr_copy;
+ u8 *descr_copy;
struct wim_security_data *sd;
bool bret;
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);