This patch implements loading files from /etc/cgconfig.d/. It uses 
already implemented functions. To hold list of files to be read for 
template reloading, new static variable for this list (list of template 
files) has been added to config.c. It is the same list as file list read 
with cgconfigparser (cgconfig.c file), i.e. the same functions are 
called. Because parser reads only one file, it is called for each file 
in template file list. Thus initialization, resp. template duplication 
has to be called before, resp. after each file parsing. Thus new 
functions has been presented, thus modifying config.[h|c] api and 
libcgroup.map. Again parsing more files for templates is analogy to 
parsing files for groups in cgconfig.c.

cgroup_reload_cached_templates_from_file and 
cgroup_init_templates_cache_from_files are modification of 
cgroup_reload_cached_templates and cgroup_init_templates_cache. Only for 
loop and auxiliary variables added.

Signed-off-by: Jan Chaloupka<jchal...@redhat.com>
---
  include/libcgroup/config.h |  18 ++++
  src/config.c               | 203 
++++++++++++++++++++++++++++++++++++++++++++-
  src/libcgroup.map          |   6 ++
  3 files changed, 224 insertions(+), 3 deletions(-)

diff --git a/include/libcgroup/config.h b/include/libcgroup/config.h
index 43568e1..cf52922 100644
--- a/include/libcgroup/config.h
+++ b/include/libcgroup/config.h
@@ -84,6 +84,24 @@ int cgroup_init_templates_cache(char *pathname);
  int cgroup_reload_cached_templates(char *pathname);

  /**
+ * Initializes the templates cache from files.
+ */
+struct cgroup_string_list;
+int cgroup_init_templates_cache_from_files(int * file_index);
+
+/**
+ * Reloads the templates list, using the given configuration files.
+ */
+int cgroup_reload_cached_templates_from_file(int * file_index);
+
+/**
+ * Setting source files of templates. This function has to be called before
+ * any call of cgroup_init_templates_cache_from_files and
+ * cgroup_reload_cached_templates_from_file.
+ */
+void cgroup_templates_cache_set_source_files(struct cgroup_string_list 
* tmpl_files);
+
+/**
   * Physically create a new control group in kernel, based on given control
   * group template and configuration file. If given template is not set in
   * configuration file, then the procedure works create the control group
diff --git a/src/config.c b/src/config.c
index e1ee0a8..c6096b2 100644
--- a/src/config.c
+++ b/src/config.c
@@ -41,6 +41,8 @@
  #include <sys/stat.h>
  #include <sys/types.h>

+#include "tools/tools-common.h"
+
  unsigned int MAX_CGROUPS = 64;    /* NOTE: This value changes 
dynamically */
  unsigned int MAX_TEMPLATES = 64;
                  /* NOTE: This value changes dynamically */
@@ -89,6 +91,7 @@ static int config_template_table_index;
   */
  static struct cgroup *template_table;
  static int template_table_index;
+static struct cgroup_string_list * template_files = NULL;


  /*
@@ -1572,6 +1575,199 @@ int cgroup_init_templates_cache(char *pathname)

  }

+/**
+ * Setting source files of templates. This function has to be called before
+ * any call of cgroup_init_templates_cache_from_files and
+ * cgroup_reload_cached_templates_from_file.
+ * @param tmpl_files
+ */
+void cgroup_templates_cache_set_source_files(struct cgroup_string_list 
* tmpl_files) {
+    template_files = tmpl_files;
+}
+
+/**
+ * Initializes the templates cache from files.
+ *    @return 0 on success, > 0 on error
+ */
+int cgroup_init_templates_cache_from_files(int * file_index) {
+        int ret = 0;
+    int i, j;
+
+        if (!template_files) {
+            /* source files has not been set */
+            cgroup_dbg("Template source files have not been set\n");
+            ret = ECGOTHER;
+            *file_index = -1;
+            return ret;
+        }
+
+    if (template_table) {
+        /* template structures have to be free */
+        for (i = 0; i < template_table_index; i++)
+            cgroup_free_controllers(&template_table[i]);
+        free(template_table);
+        template_table = NULL;
+    }
+    template_table_index = 0;
+
+    if ((config_template_table_index != 0) || (config_table_index != 0)) {
+        /* config structures have to be clean */
+        cgroup_free_config();
+    }
+
+        char * pathname;
+        int template_table_last_index = 0;
+
+        for (j = 0; j < template_files->count; j++) {
+            pathname = template_files->items[j];
+
+            cgroup_dbg("Loading cached templates from %s.\n", pathname);
+            /* Attempt to read the configuration file and cache the 
rules. */
+            ret = cgroup_parse_config(pathname);
+            if (ret) {
+                    cgroup_dbg("Could not initialize rule cache, error 
was: %d\n",
+                            ret);
+                    *file_index = j;
+                    return ret;
+            }
+
+            /* copy template data to templates cache structures */
+            if (config_template_table_index > 0) {
+                template_table_last_index = template_table_index;
+                template_table_index += config_template_table_index;
+                template_table = realloc(template_table, 
template_table_index*sizeof(struct cgroup));
+
+                if (template_table == NULL) {
+                        ret = ECGOTHER;
+                        return ret;
+                }
+
+                for (i = 0; i < config_template_table_index; i++) {
+                    template_table[i + template_table_last_index].index 
= 0;
+                }
+            }
+
+            cgroup_dbg("Init cache: copying templates to template table 
from %s.\n", pathname);
+            for (i = 0; i < config_template_table_index; i++) {
+                    int ti = i + template_table_last_index;
+                    cgroup_copy_cgroup(&template_table[ti],
+                            &config_template_table[i]);
+                    strcpy((template_table[ti]).name,
+                            (config_template_table[i]).name);
+                    template_table[ti].tasks_uid =
+                            config_template_table[i].tasks_uid;
+                    template_table[ti].tasks_gid =
+                            config_template_table[i].tasks_gid;
+                    template_table[ti].task_fperm =
+                            config_template_table[i].task_fperm;
+                    template_table[ti].control_uid =
+                            config_template_table[i].control_uid;
+                    template_table[ti].control_gid =
+                            config_template_table[i].control_gid;
+                    template_table[ti].control_fperm =
+                            config_template_table[i].control_fperm;
+                    template_table[ti].control_dperm =
+ config_template_table[i].control_dperm;
+            }
+            cgroup_dbg("Init cache: Templates to template table copied\n");
+        }
+
+    return ret;
+}
+
+/**
+ * Reloads the templates list, using the given configuration files.
+ *    @return 0 on success, > 0 on failure
+ */
+int cgroup_reload_cached_templates_from_file(int * file_index)
+{
+    int i, j;
+    int ret = 0;
+
+        if (!template_files) {
+            /* source files has not been set */
+            cgroup_dbg("Template source files have not been set\n");
+            ret = ECGOTHER;
+            *file_index = -1;
+            return ret;
+        }
+
+    if (template_table) {
+        /* template structures have to be free */
+        for (i = 0; i < template_table_index; i++)
+            cgroup_free_controllers(&template_table[i]);
+        free(template_table);
+        template_table = NULL;
+    }
+    template_table_index = 0;
+
+    if ((config_template_table_index != 0) || (config_table_index != 0)) {
+        /* config template structures have to be free as well*/
+        cgroup_free_config();
+    }
+
+
+        char * pathname;
+        int template_table_last_index = 0;
+
+        for (j = 0; j < template_files->count; j++) {
+            pathname = template_files->items[j];
+
+            /* reloading data to config template structures */
+            cgroup_dbg("Reloading cached templates from %s.\n", pathname);
+            ret = cgroup_parse_config(pathname);
+            if (ret) {
+                    cgroup_dbg("Could not reload template cache, error 
was: %d\n",
+                            ret);
+                    *file_index = j;
+                    return ret;
+            }
+
+            /* copy data to templates cache structures */
+            if (config_template_table_index > 0) {
+                template_table_last_index = template_table_index;
+                template_table_index += config_template_table_index;
+                template_table = realloc(template_table, 
template_table_index*sizeof(struct cgroup));
+
+                if (template_table == NULL) {
+                        ret = ECGOTHER;
+                        return ret;
+                }
+
+                /* clear cgroups of new templates */
+                for (i = 0; i < config_template_table_index; i++) {
+                    template_table[i + template_table_last_index].index 
= 0;
+                }
+            }
+
+            cgroup_dbg("Reload cache: copying templates to template 
table from %s.\n", pathname);
+            for (i = 0; i < config_template_table_index; i++) {
+                    int ti = i + template_table_last_index;
+                    cgroup_copy_cgroup(&template_table[ti],
+                            &config_template_table[i]);
+                    strcpy((template_table[ti]).name,
+                            (config_template_table[i]).name);
+                    template_table[ti].tasks_uid =
+                            config_template_table[i].tasks_uid;
+                    template_table[ti].tasks_gid =
+                            config_template_table[i].tasks_gid;
+                    template_table[ti].task_fperm =
+                            config_template_table[i].task_fperm;
+                    template_table[ti].control_uid =
+                            config_template_table[i].control_uid;
+                    template_table[ti].control_gid =
+                            config_template_table[i].control_gid;
+                    template_table[ti].control_fperm =
+                            config_template_table[i].control_fperm;
+                    template_table[ti].control_dperm =
+                            config_template_table[i].control_dperm;
+            }
+            cgroup_dbg("Reload cache: Templates to template table 
copied\n");
+        }
+
+    return ret;
+}
+
  /*
   * Create a given cgroup, based on template configuration if it is present
   * if the template is not present cgroup is creted using 
cgroup_create_cgroup
@@ -1593,13 +1789,14 @@ int cgroup_config_create_template_group(struct 
cgroup *cgroup,
       * use CGCONFIG_CONF_FILE by default
       */
      if (!(flags & CGFLAG_USE_TEMPLATE_CACHE)) {
+                int fileindex;
          if (template_table_index == 0)
              /* the rules cache is empty */
-            ret = cgroup_init_templates_cache(CGCONFIG_CONF_FILE);
+                        ret = 
cgroup_init_templates_cache_from_files(&fileindex);
+
          else
              /* cache is not empty */
-            ret = cgroup_reload_cached_templates(
-                CGCONFIG_CONF_FILE);
+                        ret = 
cgroup_reload_cached_templates_from_file(&fileindex);
          if (ret != 0) {
              cgroup_dbg("Failed initialize templates cache.\n");
              return ret;
diff --git a/src/libcgroup.map b/src/libcgroup.map
index b0c162c..f853af0 100644
--- a/src/libcgroup.map
+++ b/src/libcgroup.map
@@ -117,3 +117,9 @@ CGROUP_0.39 {
      cgroup_log;
      cgroup_parse_log_level_str;
  } CGROUP_0.38;
+
+CGROUP_0.40 {
+    cgroup_init_templates_cache_from_files;
+        cgroup_reload_cached_templates_from_file;
+        cgroup_templates_cache_set_source_files;
+} CGROUP_0.39;
-- 
1.9.0

------------------------------------------------------------------------------
Learn Graph Databases - Download FREE O'Reilly Book
"Graph Databases" is the definitive new guide to graph databases and their
applications. Written by three acclaimed leaders in the field,
this first edition is now available. Download your free book today!
http://p.sf.net/sfu/NeoTech
_______________________________________________
Libcg-devel mailing list
Libcg-devel@lists.sourceforge.net
https://lists.sourceforge.net/lists/listinfo/libcg-devel

Reply via email to