up

Zigrin-Security · Jun 14, 2023 · 9e6747c · 9e6747c
1 parent 5372af2
commit 9e6747c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/cakefuzzer/instrumentation/__init__.py b/cakefuzzer/instrumentation/__init__.py
@@ -28,15 +28,13 @@ class InstrumentationError(CakeFuzzerError):
 
 
 async def apply(*args: Instrumentation) -> List[Instrumentation]:
-    _, unapplied = await check(*args)
-    await asyncio.gather(*[p.apply() for p in unapplied])
-    return unapplied
+    await asyncio.gather(*[p.apply() for p in args])
+    return args
 
 
 async def revert(*args: Instrumentation) -> List[Instrumentation]:
-    applied, _ = await check(*args)
-    await asyncio.gather(*[p.revert() for p in applied])
-    return applied
+    await asyncio.gather(*[p.revert() for p in args])
+    return args
 
 
 async def check(

diff --git a/cakefuzzer/instrumentation/instrumentator.py b/cakefuzzer/instrumentation/instrumentator.py
@@ -175,11 +175,11 @@ async def revert(self) -> None:
 
         applied, unapplied = await check(*annotation_removal)
         await revert(*applied, *unapplied)
-        print("Annotations Reverted", len(applied))
+        print("Annotations Reverted", len(applied) + len(unapplied))
 
         applied, unapplied = await check(*frenames)
         await revert(*applied, *unapplied)
-        print("FunctionCall Renames Reverted", len(applied))
+        print("FunctionCall Renames Reverted", len(applied) + len(unapplied))
 
         applied, _ = await check(*patches)
         await revert(*applied)