1 files changed, 22 insertions, 14 deletions

diff --git a/extra.h b/extra.h
index 9da536d..a4a3d9c 100644
--- a/extra.h
+++ b/extra.h
@@ -403,12 +403,12 @@ static void _rt_write_to_file() {
     }
     fclose(fp);
 out:
-#if LITMUS
-    if (task_mode(BACKGROUND_TASK) != 0) {
-        perror("Unable to become a real-time task");
-        exit(1);
-    }
-#endif /* LITMUS */
+// #if LITMUS
+//     if (task_mode(BACKGROUND_TASK) != 0) {
+//         perror("Unable to become a real-time task");
+//         exit(1);
+//     }
+// #endif /* LITMUS */
 #ifdef PAIRED
     munmap(_rt_barrier, 2);
     sem_unlink(_rt_sem1_name);
@@ -428,7 +428,8 @@ out:
 // Start a job
 static void _rt_start_loop() {
     // Let the first job run before synchrnous release --- make sure we have no further page faults
-    if(first_job){
+    if(first_job == 1){
+        // FLUSH_CACHES
         return;
     }
 #if LITMUS
@@ -442,7 +443,7 @@ static void _rt_start_loop() {
     else{
         if (sleep_next_period() != 0) {
             perror("Unable to sleep for next period");
-        }
+        }   
     }
     
     // if (last == 0) {
@@ -479,7 +480,7 @@ static void _rt_start_loop() {
 static void _rt_stop_loop() {
     clock_gettime(CLOCK_MONOTONIC, &_rt_end);
     // Set first_job == 0
-    if(first_job){
+    if(first_job == 1){
         first_job = 0;
         return;
     }
@@ -493,13 +494,20 @@ static void _rt_stop_loop() {
     long long end = _rt_end_time[_rt_jobs_complete];
     if (end > _rt_cp->deadline) {
         fprintf(stderr, "(%s/%d:%lu) ", _rt_our_prog_name, gettid(), _rt_cp->job_index);
-        fprintf(stderr, " tardy %llu ns, "
-               "relative tardiness %f\n",
-               end - _rt_cp->deadline,
-               (end - _rt_cp->deadline) / (double)_rt_deadline);
+        fprintf(stderr, "tardy %llu ns, relative tardiness %f\n", end - _rt_cp->deadline, (end - _rt_cp->deadline) / (double)_rt_deadline);
     }
 #endif
-    _rt_jobs_complete++;
+     _rt_jobs_complete++;
+#if LITMUS
+    if(_rt_jobs_complete == _rt_max_jobs){
+        // fprintf(stderr, "complete %s: %d\n" , _rt_our_prog_name, _rt_jobs_complete);
+        if (task_mode(BACKGROUND_TASK) != 0) {
+            perror("Unable to become background task");
+            exit(1);
+        }
+    }
+#endif
+    
 }
 
 /****** New API ******