From 505dad17d2216fab0342afb5794f300a949dd4c8 Mon Sep 17 00:00:00 2001
From: Konstantin Aladyshev <aladyshev22@gmail.com>
Date: Mon, 12 Feb 2024 16:21:34 +0300
Subject: Add protocol installation callbacks lesson

Signed-off-by: Konstantin Aladyshev <aladyshev22@gmail.com>
---
 .../ProtocolEventDriver/ProtocolEventDriver.c      | 99 ++++++++++++++++++++++
 1 file changed, 99 insertions(+)
 create mode 100644 UefiLessonsPkg/ProtocolEventDriver/ProtocolEventDriver.c

(limited to 'UefiLessonsPkg/ProtocolEventDriver/ProtocolEventDriver.c')

diff --git a/UefiLessonsPkg/ProtocolEventDriver/ProtocolEventDriver.c b/UefiLessonsPkg/ProtocolEventDriver/ProtocolEventDriver.c
new file mode 100644
index 0000000..02ed0f9
--- /dev/null
+++ b/UefiLessonsPkg/ProtocolEventDriver/ProtocolEventDriver.c
@@ -0,0 +1,99 @@
+/*
+ * Copyright (c) 2024, Konstantin Aladyshev <aladyshev22@gmail.com>
+ *
+ * SPDX-License-Identifier: MIT
+ */
+
+#include <Library/UefiBootServicesTableLib.h>
+#include <Library/UefiLib.h>
+
+#include <Protocol/SimpleClass.h>
+
+EFI_EVENT Event;
+STATIC VOID  *mRegistrationTracker;
+UINTN NotifyData = 0;
+
+VOID EFIAPI NotifyFunc(EFI_EVENT Event, VOID* Context)
+{
+  if (Context == NULL)
+    return;
+
+  Print(L"\nEvent is signaled! Context = %d\n", *(UINTN*)Context);
+  *(UINTN*)Context += 1;
+
+  SIMPLE_CLASS_PROTOCOL* SimpleClass;
+  EFI_STATUS Status;
+  Status = gBS->LocateProtocol(&gSimpleClassProtocolGuid,
+                               // NULL,
+                               mRegistrationTracker,
+                               (VOID**)&SimpleClass);
+  if (EFI_ERROR(Status)) {
+    Print(L"Error! LocateProtocol returned: %r\n", Status);
+    return;
+  }
+
+  UINTN Number;
+  Status = SimpleClass->GetNumber(&Number);
+  if (!EFI_ERROR(Status)) {
+    Print(L"Current number = %d\n", Number);
+  } else {
+    Print(L"Error! Can't get number: %r\n", Status);
+    return;
+  }
+
+  Status = SimpleClass->SetNumber(Number+5);
+  if (EFI_ERROR(Status)) {
+    Print(L"Error! Can't set number: %r\n", Status);
+    return;
+  }
+}
+
+
+EFI_STATUS
+EFIAPI
+ProtocolEventDriverUnload (
+  EFI_HANDLE ImageHandle
+  )
+{
+  gBS->CloseEvent(Event);
+
+  return EFI_SUCCESS;
+}
+
+EFI_STATUS
+EFIAPI
+ProtocolEventDriverEntryPoint (
+  IN EFI_HANDLE        ImageHandle,
+  IN EFI_SYSTEM_TABLE  *SystemTable
+  )
+{
+
+  EFI_STATUS Status;
+
+  Status = gBS->CreateEvent(EVT_NOTIFY_SIGNAL,
+                            TPL_NOTIFY,
+                            &NotifyFunc,
+                            &NotifyData,
+                            &Event);
+  if (EFI_ERROR(Status)) {
+    Print(L"Error! CreateEvent returned: %r\n", Status);
+    return Status;
+  }
+
+  Status = gBS->RegisterProtocolNotify(&gSimpleClassProtocolGuid,
+                                       Event,
+                                       &mRegistrationTracker);
+  if (EFI_ERROR(Status)) {
+    Print(L"Error! RegisterProtocolNotify returned: %r\n", Status);
+    return Status;
+  }
+
+/*
+  Event = EfiCreateProtocolNotifyEvent(&gSimpleClassProtocolGuid,
+                                       TPL_NOTIFY,
+                                       &NotifyFunc,
+                                       &NotifyData,
+                                       &mRegistrationTracker);
+*/
+  return EFI_SUCCESS;
+}
-- 
cgit v1.2.3-18-g5258