1 files changed, 14 insertions, 15 deletions

diff --git a/bin/rtspin_edfos.c b/bin/rtspin_edfos.c
index e39a8c0..3ea41c7 100644
--- a/bin/rtspin_edfos.c
+++ b/bin/rtspin_edfos.c
@@ -141,10 +141,10 @@ int main(int argc, char** argv)
         lt_t wcet;
         lt_t period;
         /* [num,den] */
-        lt_t fracs[NR_CPUS][2];
+        lt_t fracs[NR_CPUS_EDF_OS][2];
         double wcet_ms, period_ms;
-        int migrate = 0;
-        int cpu = 0;
+        int cpu = -1;
+        int ccpu;
         int opt;
         int wait = 0;
         int test_loop = 0;
@@ -156,9 +156,9 @@ int main(int argc, char** argv)
 
         progname = argv[0];
 
-        for (i = 0; i < NR_CPUS; i++) {
+        for (i = 0; i < NR_CPUS_EDF_OS; i++) {
                 fracs[i][0] = 0;
-                fracs[i][1] = 1;
+                fracs[i][1] = 0;
         }
 
         while ((opt = getopt(argc, argv, OPTSTR)) != -1) {
@@ -170,7 +170,6 @@ int main(int argc, char** argv)
                         cpu = atoi(optarg);
                         fracs[cpu][0] = 1;
                         fracs[cpu][1] = 1;
-                        migrate = 1;
                         break;
                 case 'c':
                         class = str2class(optarg);
@@ -215,10 +214,12 @@ int main(int argc, char** argv)
         duration  = atof(argv[optind + 2]);
         /* frac num, den = 0 means fixed task */
         for (i = optind + 3; i < argc; i += 3) {
-                cpu = atof(argv[i]);
-                fracs[cpu][0] = argv[i+1];
-                fracs[cpu][1] = argv[i+2];
+                ccpu = atof(argv[i]);
+                fracs[ccpu][0] = argv[i+1];
+                fracs[ccpu][1] = argv[i+2];
         }
+        if (cpu == -1)
+                cpu = compute_initial_cpu_edfos(fracs);
         wcet   = wcet_ms * __NS_PER_MS;
         period = period_ms * __NS_PER_MS;
         if (wcet <= 0)
@@ -231,13 +232,11 @@ int main(int argc, char** argv)
                       "exceed the period.");
         }
 
-        if (migrate) {
-                ret = be_migrate_to(cpu);
-                if (ret < 0)
-                        bail_out("could not migrate to target partition");
-        }
+        ret = be_migrate_to(cpu);
+        if (ret < 0)
+                bail_out("could not migrate to target partition");
 
-        ret = sporadic_task_ns_edfos(wcet, period, 0, fracs, class,
+        ret = sporadic_task_ns_edfos(wcet, period, 0, cpu, fracs, class,
                                 NO_ENFORCEMENT);
 
         if (ret < 0)