From ff9c5cea4f021485bee51c8ae7fd5b5f3b5b93d8 Mon Sep 17 00:00:00 2001 From: Donghyun Kim Date: Mon, 28 Mar 2022 03:31:29 +0900 Subject: [PATCH] Modify return type hint of method for get driver's and device's --- .../framework/symbols/windows/extensions/__init__.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/volatility3/framework/symbols/windows/extensions/__init__.py b/volatility3/framework/symbols/windows/extensions/__init__.py index c1972cb7f..2266305fa 100755 --- a/volatility3/framework/symbols/windows/extensions/__init__.py +++ b/volatility3/framework/symbols/windows/extensions/__init__.py @@ -7,9 +7,10 @@ import datetime import functools import logging import math -from typing import Iterable, Iterator, List, Optional, Tuple, Union +from typing import Generator, Iterable, Iterator, List, Optional, Tuple, Union from volatility3.framework import constants, exceptions, interfaces, objects, renderers, symbols +from volatility3.framework.interfaces.objects import ObjectInterface from volatility3.framework.layers import intel from volatility3.framework.renderers import conversion from volatility3.framework.symbols import generic @@ -359,7 +360,7 @@ class DEVICE_OBJECT(objects.StructType, pool.ExecutiveObject): except(ValueError): return renderers.UnparsableValue() - def get_attached_devices(self) -> interfaces.objects.ObjectInterface: + def get_attached_devices(self) -> Generator[ObjectInterface, None, None]: """Enumerate the device's attaches""" device = self.AttachedDevice.dereference() while device: @@ -377,7 +378,7 @@ class DRIVER_OBJECT(objects.StructType, pool.ExecutiveObject): except(ValueError): return renderers.UnparsableValue() - def get_devices(self) -> interfaces.objects.ObjectInterface: + def get_devices(self) -> Generator[ObjectInterface, None, None]: """Enumerate the driver's device objects""" device = self.DeviceObject.dereference() while device: