1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
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;
}
|
