From f5f511147190ca0bdc4cc876dddf8ca5c48ecda3 Mon Sep 17 00:00:00 2001
From: "Justin C. Miller" <justin@devjustinian.com>
Date: Sun, 23 Feb 2020 19:27:32 -0800
Subject: [PATCH] Add handle_protocol and set_mem to boot_services

---
 include/uefi/boot_services.h | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/include/uefi/boot_services.h b/include/uefi/boot_services.h
index 1d9843d..3cbaffe 100644
--- a/include/uefi/boot_services.h
+++ b/include/uefi/boot_services.h
@@ -15,8 +15,10 @@ namespace uefi {
 namespace bs_impl {
 	using allocate_pages = status (*)(allocate_type, memory_type, uint64_t, uintptr_t*);
 	using allocate_pool = status (*)(memory_type, uint64_t, void**);
+	using handle_protocol = status (*)(handle, guid *, void **);
 	using create_event = status (*)(evt, tpl, event_notify, void*, event*);
 	using locate_protocol = status (*)(const guid *, void *, void **);
+	using set_mem = void (*)(void *, uint64_t, uint8_t);
 }
 
 struct boot_services {
@@ -47,7 +49,7 @@ struct boot_services {
 	void *install_protocol_interface;
 	void *reinstall_protocol_interface;
 	void *uninstall_protocol_interface;
-	void *handle_protocol;
+	bs_impl::handle_protocol handle_protocol;
 	void *_reserved;
 	void *register_protocol_notify;
 	void *locate_handle;
@@ -87,7 +89,7 @@ struct boot_services {
 
 	// Miscellaneous Services
 	void *copy_mem;
-	void *set_mem;
+	bs_impl::set_mem set_mem;
 	void *create_event_ex;
 };