Signal protocol notify events while processing calls to InstallProtocolInterface.

Author: assafcarlsbad

qiling/os/uefi/protocols/common.py

@@ -4,7 +4,7 @@
 #
 
 from qiling.os.uefi.const import EFI_SUCCESS, EFI_NOT_FOUND, EFI_UNSUPPORTED, EFI_BUFFER_TOO_SMALL, EFI_INVALID_PARAMETER
-from qiling.os.uefi.utils import read_int64, write_int64, check_and_notify_protocols
+from qiling.os.uefi.utils import read_int64, write_int64, check_and_notify_protocols, signal_event
 from qiling.os.uefi.UefiSpec import EFI_LOCATE_SEARCH_TYPE
 
 # TODO: get rid of this
@@ -37,6 +37,12 @@ def InstallProtocolInterface(context, params):
 
 	dic[params["Protocol"]] = params["Interface"]
 	context.protocols[handle] = dic
+
+	for (event_id, event_dic) in context.ql.loader.events.items():
+		if event_dic['Guid'] == params['Protocol']:
+			# The event was previously registered by 'RegisterProtocolNotify'.
+			signal_event(context.ql, event_id)	
+
 	check_and_notify_protocols(context.ql)
 	write_int64(context.ql, params["Handle"], handle)