X-Git-Url: https://wimlib.net/git/?a=blobdiff_plain;f=include%2Fwimlib%2Fsecurity.h;h=5c864ddd7754e4fbe853d90a1a6bb31d9df1584f;hb=846d3ae386f6c12953816108c04de244533e3a4a;hp=9d338a15faa3b6cf108af21e69d495de48a8d3d7;hpb=67432c2a306fa9cbc31753ae03f6f1ebd0d49220;p=wimlib

diff --git a/include/wimlib/security.h b/include/wimlib/security.h
index 9d338a15..5c864ddd 100644
--- a/include/wimlib/security.h
+++ b/include/wimlib/security.h
@@ -1,16 +1,18 @@
 #ifndef _WIMLIB_SECURITY_H
 #define _WIMLIB_SECURITY_H
 
-#include "wimlib/rbtree.h"
 #include "wimlib/types.h"
 
-/* Red-black tree that maps SHA1 message digests of security descriptors to
- * security IDs, which are themselves indices into the table of security
- * descriptors in the 'struct wim_security_data'. */
+struct wim_security_data;
+struct avl_tree_node;
+
+/* Map from SHA1 message digests of security descriptors to security IDs, which
+ * are themselves indices into the table of security descriptors in the 'struct
+ * wim_security_data'. */
 struct wim_sd_set {
 	struct wim_security_data *sd;
-	struct rb_root rb_root;
-	int32_t orig_num_entries;
+	struct avl_tree_node *root;
+	s32 orig_num_entries;
 };
 
 /* Table of security descriptors for a WIM image. */
@@ -34,12 +36,12 @@ struct wim_security_data {
 };
 
 extern void
-destroy_sd_set(struct wim_sd_set *sd_set, bool rollback);
+rollback_new_security_descriptors(struct wim_sd_set *sd_set);
 
-extern int
-lookup_sd(struct wim_sd_set *set, const u8 hash[]);
+extern void
+destroy_sd_set(struct wim_sd_set *sd_set);
 
-extern int
+extern s32
 sd_set_add_sd(struct wim_sd_set *sd_set, const char descriptor[],
 	      size_t size);
 
@@ -50,12 +52,12 @@ extern struct wim_security_data *
 new_wim_security_data(void);
 
 extern int
-read_wim_security_data(const u8 metadata_resource[], size_t
-		       metadata_resource_len, struct wim_security_data **sd_p);
+read_wim_security_data(const u8 *buf, size_t buf_len,
+		       struct wim_security_data **sd_ret);
 
 extern u8 *
-write_wim_security_data(const struct wim_security_data * restrict sd, u8 *
-			restrict p);
+write_wim_security_data(const struct wim_security_data * restrict sd,
+			u8 * restrict p);
 
 extern void
 print_wim_security_data(const struct wim_security_data *sd);