Add sched domain proc support to C-FL

1 files changed, 61 insertions, 1 deletions

diff --git a/litmus/sched_cfl_split.c b/litmus/sched_cfl_split.c
index 8dc72c39dcb2..eef10f46c7e1 100644
--- a/litmus/sched_cfl_split.c
+++ b/litmus/sched_cfl_split.c
@@ -914,6 +914,55 @@ static void cleanup_cflsplit(void)
         }
 }
 
+static struct domain_proc_info cflsplit_domain_proc_info;
+static long cflsplit_get_domain_proc_info(struct domain_proc_info **ret)
+{
+        *ret = &cflsplit_domain_proc_info;
+        return 0;
+}
+
+static void cflsplit_setup_domain_proc(void)
+{
+        int i, cpu, domain;
+#ifdef CONFIG_RELEASE_MASTER
+        int release_master = atomic_read(&release_master_cpu);
+        /* skip over the domain with the release master if cluster size is 1 */
+        int skip_domain = (1 == cluster_size && release_master != NO_CPU) ?
+                        release_master : NO_CPU;
+#else
+        int release_master = NO_CPU;
+        int skip_domain = NO_CPU;
+#endif
+        int num_rt_cpus = num_online_cpus() - (release_master != NO_CPU);
+        int num_rt_domains = num_clusters - (skip_domain != NO_CPU);
+        struct cd_mapping *map;
+
+        memset(&cflsplit_domain_proc_info, sizeof(cflsplit_domain_proc_info), 0);
+        init_domain_proc_info(&cflsplit_domain_proc_info, num_rt_cpus, num_rt_domains);
+        cflsplit_domain_proc_info.num_cpus = num_rt_cpus;
+        cflsplit_domain_proc_info.num_domains = num_rt_domains;
+
+        for (cpu = 0, i = 0; cpu < num_online_cpus(); ++cpu) {
+                if (cpu == release_master)
+                        continue;
+                map = &cflsplit_domain_proc_info.cpu_to_domains[i];
+                /* pointer math to figure out the domain index */
+                domain = remote_cluster(cpu) - cflsplit;
+                map->id = cpu;
+                cpumask_set_cpu(domain, map->mask);
+                ++i;
+        }
+
+        for (domain = 0, i = 0; domain < num_clusters; ++domain) {
+                if (domain == skip_domain)
+                        continue;
+                map = &cflsplit_domain_proc_info.domain_to_cpus[i];
+                map->id = i;
+                cpumask_copy(map->mask, cflsplit[domain].cpu_map);
+                ++i;
+        }
+}
+
 static long cflsplit_activate_plugin(void)
 {
         int i, j, cpu, ccpu, cpu_count;
@@ -1035,8 +1084,17 @@ static long cflsplit_activate_plugin(void)
                 }
         }
 
-        free_cpumask_var(mask);
         clusters_allocated = 1;
+        free_cpumask_var(mask);
+
+        cflsplit_setup_domain_proc();
+
+        return 0;
+}
+
+static long cflsplit_deactivate_plugin(void)
+{
+        destroy_domain_proc_info(&cflsplit_domain_proc_info);
         return 0;
 }
 
@@ -1054,6 +1112,8 @@ static struct sched_plugin cflsplit_plugin __cacheline_aligned_in_smp = {
         .task_block             = cflsplit_task_block,
         .admit_task             = cflsplit_admit_task,
         .activate_plugin        = cflsplit_activate_plugin,
+        .deactivate_plugin  = cflsplit_deactivate_plugin,
+        .get_domain_proc_info   = cflsplit_get_domain_proc_info,
 };
 
 static struct proc_dir_entry *cluster_file = NULL, *cflsplit_dir = NULL;