1 files changed, 13 insertions, 9 deletions

diff --git a/baseline/source/rijndael_enc/rijndael_enc.c b/baseline/source/rijndael_enc/rijndael_enc.c
index 6c85eee..f0582c9 100644
--- a/baseline/source/rijndael_enc/rijndael_enc.c
+++ b/baseline/source/rijndael_enc/rijndael_enc.c
@@ -68,6 +68,7 @@ void rijndael_enc_init( void )
   rijndael_enc_fin.data = rijndael_enc_data;
   rijndael_enc_fin.size = 31369;
   rijndael_enc_fin.cur_pos = 0;
+  rijndael_enc_checksum = 0;
 
   unsigned i;
   volatile int x = 0;
@@ -225,14 +226,17 @@ void _Pragma( "entrypoint" ) rijndael_enc_main( void )
 
 int main( int argc, char** argv )
 {
-        SET_UP
-        for (jobsComplete=0; jobsComplete<maxJobs; jobsComplete++){
-                START_LOOP
-                rijndael_enc_init();
-                rijndael_enc_main();
-                STOP_LOOP
-        }
-        WRITE_TO_FILE
-  return ( rijndael_enc_return() );
+  int exit_code = -1;
+  SET_UP
+  for_each_job {
+    rijndael_enc_init();
+    rijndael_enc_main();
+    // This checks the checksum is as expected.
+    // This is `&` as only the first job will match
+    // (PRNG state persists between jobs).
+    exit_code &= rijndael_enc_return();
+  }
+  WRITE_TO_FILE
+  return ( exit_code );
 }