From dca184cdd4100860fd6238699098e7321a263aab Mon Sep 17 00:00:00 2001
From: Michal Sieron <msieron@antmicro.com>
Date: Mon, 6 Feb 2023 17:28:48 +0100
Subject: [PATCH] Add --no-attack-time option to rowhammer scripts

Signed-off-by: Michal Sieron <msieron@antmicro.com>
---
 docs/source/usage.md                     | 10 +++++
 rowhammer_tester/scripts/hw_rowhammer.py | 25 ++++++------
 rowhammer_tester/scripts/rowhammer.py    | 50 ++++++++++++++++++++----
 3 files changed, 66 insertions(+), 19 deletions(-)

diff --git a/docs/source/usage.md b/docs/source/usage.md
index 1a283fc5..f0f4eac4 100644
--- a/docs/source/usage.md
+++ b/docs/source/usage.md
@@ -206,6 +206,16 @@ Different attack and row selection modes can be used, but only one of them can b
   (venv) $ python hw_rowhammer.py --row-pairs random --start-row 4 --nrows 10
   ```
 
+- `--no-attack-time <time>`
+
+  Instead of performing a rowhammer attack, the script will load the RAM with selected pattern and sleep for `time` nanoseconds.
+  After this time, it will check for any bitflips that could have happened.
+  This option does not imply `--no-refresh`!
+
+  ```sh
+  (venv) $ python hw_rowhammer.py --no-attack-time 100e9 --no-refresh
+  ```
+
 ### Patterns
 
 User can choose a pattern that memory will be initially filled with:
diff --git a/rowhammer_tester/scripts/hw_rowhammer.py b/rowhammer_tester/scripts/hw_rowhammer.py
index bf8f6108..9534bcc6 100755
--- a/rowhammer_tester/scripts/hw_rowhammer.py
+++ b/rowhammer_tester/scripts/hw_rowhammer.py
@@ -113,17 +113,20 @@ def run(self, row_pairs, pattern_generator, read_count, row_progress=16, verify_
             print('\nDisabling refresh ...')
             self.wb.regs.controller_settings_refresh.write(0)
 
-        print('\nRunning Rowhammer attacks ...')
-        for i, row_tuple in enumerate(row_pairs, start=1):
-            s = 'Iter {:{n}} / {:{n}}'.format(i, len(row_pairs), n=len(str(len(row_pairs))))
-            if self.payload_executor:
-                self.payload_executor_attack(read_count=read_count, row_tuple=row_tuple)
-            else:
-                if len(row_tuple) & (len(row_tuple) - 1) != 0:
-                    print("ERROR: BIST only supports power of 2 rows\n")
-                    return
-
-                self.attack(row_tuple, read_count=read_count, progress_header=s)
+        if self.no_attack_time is not None:
+            self.no_attack_sleep()
+        else:
+            print('\nRunning Rowhammer attacks ...')
+            for i, row_tuple in enumerate(row_pairs, start=1):
+                s = 'Iter {:{n}} / {:{n}}'.format(i, len(row_pairs), n=len(str(len(row_pairs))))
+                if self.payload_executor:
+                    self.payload_executor_attack(read_count=read_count, row_tuple=row_tuple)
+                else:
+                    if len(row_tuple) & (len(row_tuple) - 1) != 0:
+                        print("ERROR: BIST only supports power of 2 rows\n")
+                        return
+
+                    self.attack(row_tuple, read_count=read_count, progress_header=s)
 
         if self.no_refresh:
             print('\nReenabling refresh ...')
diff --git a/rowhammer_tester/scripts/rowhammer.py b/rowhammer_tester/scripts/rowhammer.py
index 72e7de6e..23d284c1 100755
--- a/rowhammer_tester/scripts/rowhammer.py
+++ b/rowhammer_tester/scripts/rowhammer.py
@@ -9,9 +9,10 @@
 import os
 
 from pathlib import Path
+
 from rowhammer_tester.scripts.utils import (
     memfill, memcheck, memwrite, DRAMAddressConverter, litex_server, RemoteClient,
-    get_litedram_settings, get_generated_defs, execute_payload, read_ident)
+    get_litedram_settings, get_generated_defs, execute_payload, read_ident, _progress)
 from rowhammer_tester.scripts.playbook.lib import (generate_payload_from_row_list)
 
 ################################################################################
@@ -37,6 +38,7 @@ def __init__(
             no_refresh=False,
             verbose=False,
             payload_executor=False,
+            no_attack_time=None,
             data_inversion=False):
         for name, val in locals().items():
             setattr(self, name, val)
@@ -204,6 +206,16 @@ def display_errors(self, row_errors, read_count, do_error_summary=False):
         if do_error_summary:
             return err_dict
 
+    def no_attack_sleep(self):
+        sleep_time = self.no_attack_time / 1e9
+        print(f"\nSleeping for {sleep_time} s as `--no-attack-time` flag was used ...")
+
+        for slept in range(int(10 * sleep_time)):  # refresh every 0.1 s
+            time.sleep(0.1)
+            _progress((slept + 1) / 10, sleep_time, last=False, opt=f"Sleeping for {sleep_time} s")
+
+        print()
+
     def run(self, row_pairs, pattern_generator, read_count, row_progress=16, verify_initial=False):
         """
         Main part of the script.
@@ -244,13 +256,16 @@ def run(self, row_pairs, pattern_generator, read_count, row_progress=16, verify_
             print('\nDisabling refresh ...')
             self.wb.regs.controller_settings_refresh.write(0)
 
-        print('\nRunning Rowhammer attacks ...')
-        for i, row_tuple in enumerate(row_pairs, start=1):
-            s = 'Iter {:{n}} / {:{n}}'.format(i, len(row_pairs), n=len(str(len(row_pairs))))
-            if self.payload_executor:
-                self.payload_executor_attack(read_count=read_count, row_tuple=row_tuple)
-            else:
-                self.attack(row_tuple, read_count=read_count, progress_header=s)
+        if self.no_attack_time is not None:
+            self.no_attack_sleep()
+        else:
+            print('\nRunning Rowhammer attacks ...')
+            for i, row_tuple in enumerate(row_pairs, start=1):
+                s = 'Iter {:{n}} / {:{n}}'.format(i, len(row_pairs), n=len(str(len(row_pairs))))
+                if self.payload_executor:
+                    self.payload_executor_attack(read_count=read_count, row_tuple=row_tuple)
+                else:
+                    self.attack(row_tuple, read_count=read_count, progress_header=s)
 
         if self.no_refresh:
             print('\nReenabling refresh ...')
@@ -349,6 +364,10 @@ def main(row_hammer_cls):
         '--all-rows',
         action='store_true',
         help='Run whole test sequence on all rows. Optionally, set --row-jump and --start-row')
+    row_selector_group.add_argument(
+        '--no-attack-time',
+        type=float,
+        help='Don\'t attack. Instead sleep for provided nanoseconds')
     parser.add_argument(
         '--row-pair-distance',
         type=int,
@@ -436,6 +455,9 @@ def rand_row():
             return rng.randint(args.start_row, args.start_row + args.nrows)
 
         row_pairs = [(rand_row(), rand_row()) for i in range(args.nrows)]
+    elif args.no_attack_time is not None:
+        if args.no_attack_time < 0:
+            parser.error("No attack time can't be negative")
     else:
         parser.error("No operation specified")
 
@@ -462,6 +484,7 @@ def rand_row():
         no_refresh=args.no_refresh,
         payload_executor=args.payload_executor,
         data_inversion=args.data_inversion,
+        no_attack_time=args.no_attack_time,
     )
 
     if args.log_dir:
@@ -503,6 +526,17 @@ def rand_row():
                     "hammer_row_2": pair[1],
                     "errors_in_rows": err_in_rows
                 }
+        elif args.no_attack_time is not None:
+            err_in_rows = row_hammer.run( # dummy row pair and read count
+                row_pairs=[(0, 0)], read_count=0, pattern_generator=pattern)
+
+            row_hammer.err_summary = {
+                "no_attack_time": {
+                    str(args.no_attack_time): {
+                        "errors_in_rows": err_in_rows,
+                    },
+                },
+            }
         else:
             err_in_rows = row_hammer.run(
                 row_pairs=row_pairs, read_count=count, pattern_generator=pattern)