Function GetVolumeHandleWithDir

FSInject/Test.c:24–76

EFI_STATUS
EFIAPI
GetVolumeHandleWithDir(CHAR16 *SearchDir, OUT EFI_HANDLE *Handle)
{
	EFI_STATUS		Status;
	UINTN			HandlesSize;
	UINTN			Idx;
	EFI_HANDLE		*Handles;
	EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *FS = NULL;
	EFI_FILE_PROTOCOL				*FP = NULL;
	EFI_FILE_PROTOCOL				*FP2 = NULL;
	
	// get all handles with EFI_SIMPLE_FILE_SYSTEM_PROTOCOL
	Status = gBS->LocateHandleBuffer(ByProtocol, &gEfiSimpleFileSystemProtocolGuid, NULL, &HandlesSize, &Handles);
	if (EFI_ERROR(Status)) {
		Print(L"LocateHandleBuffer: %r\n", Status);
		return Status;
	}
	Print(L"LocateHandleBuffer: %r, got %d handles\n", Status, HandlesSize);
	
	// find handle that contains SearchDir 
	for (Idx = 0; Idx < HandlesSize; Idx++) {
		// get EFI_SIMPLE_FILE_SYSTEM_PROTOCOL first
		Print(L"Trying handle: %p\n", Handles[Idx]);
		Status = gBS->OpenProtocol(Handles[Idx], &gEfiSimpleFileSystemProtocolGuid, (void **)&FS, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);
		if (EFI_ERROR(Status)) {
			Print(L"OpenProtocol(gEfiSimpleFileSystemProtocolGuid): %r\n", Status);
			continue;
		}
		// open volume
		Status = FS->OpenVolume(FS, &FP);
		if (EFI_ERROR(Status)) {
			Print(L"OpenVolume(): %r\n", Status);
			gBS->CloseProtocol(Handles[Idx], &gEfiSimpleFileSystemProtocolGuid, gImageHandle, NULL);
			continue;
		}
		// try to open injection dir
		Status = FP->Open(FP, &FP2, SearchDir, EFI_FILE_MODE_READ, 0);
		if (EFI_ERROR(Status)) {
			Print(L"Open(%s): %r\n", SearchDir, Status);
			FP->Close(FP);
			gBS->CloseProtocol(Handles[Idx], &gEfiSimpleFileSystemProtocolGuid, gImageHandle, NULL);
			continue;
		}
		// we have found it - close it and return handle
		FP2->Close(FP2);
		FP->Close(FP);
		*Handle = Handles[Idx];
		return EFI_SUCCESS;
	}
	// not found
	return EFI_NOT_FOUND;
}

EFI_STATUS
EFIAPI