Add type-safe policy_handle_create/find

1 files changed, 12 insertions, 0 deletions

diff --git a/source3/include/proto.h b/source3/include/proto.h
index 6156a475a4..8eb5c46fbd 100644
--- a/source3/include/proto.h
+++ b/source3/include/proto.h
@@ -5884,6 +5884,18 @@ bool close_policy_hnd(pipes_struct *p, struct policy_handle *hnd);
 void close_policy_by_pipe(pipes_struct *p);
 bool pipe_access_check(pipes_struct *p);
 
+NTSTATUS _policy_handle_create(struct pipes_struct *p, struct policy_handle *hnd,
+			       void *pdata, size_t size, const char *name);
+#define policy_handle_create(_p, _hnd, _ptr, _type) \
+	_policy_handle_create((_p), (_hnd), (_ptr), sizeof(_type), #_type)
+
+void *_policy_handle_find(struct pipes_struct *p,
+			  const struct policy_handle *hnd,
+			  const char *type);
+#define policy_handle_find(_p, _hnd, _type) \
+	(_type *)_policy_handle_find((_p), (_hnd), #_type)
+
+
 /* The following definitions come from rpc_server/srv_pipe.c  */
 
 bool create_next_pdu(pipes_struct *p);