On 12/05/2014 04:22 PM, Tyler Hicks wrote:
> This patch moves the logic that sets up the policy into a new function
> in policy_cache.c
> 
> Signed-off-by: Tyler Hicks <[email protected]>
Acked-by: John Johansen <[email protected]>

> ---
>  parser/Makefile       |  2 +-
>  parser/parser_main.c  | 48 ++----------------------------------------------
>  parser/policy_cache.c | 49 +++++++++++++++++++++++++++++++++++++++++++++++++
>  parser/policy_cache.h |  2 ++
>  4 files changed, 54 insertions(+), 47 deletions(-)
> 
> diff --git a/parser/Makefile b/parser/Makefile
> index b95b9da..c50398f 100644
> --- a/parser/Makefile
> +++ b/parser/Makefile
> @@ -246,7 +246,7 @@ common_optarg.o: common_optarg.c common_optarg.h parser.h 
> libapparmor_re/apparmo
>  features.o: features.c features.h parser.h libapparmor_re/apparmor_re.h
>       $(CXX) $(EXTRA_CFLAGS) -c -o $@ $<
>  
> -policy_cache.o: policy_cache.c policy_cache.h parser.h
> +policy_cache.o: policy_cache.c policy_cache.h parser.h features.h
>       $(CXX) $(EXTRA_CFLAGS) -c -o $@ $<
>  
>  kernel_interface.o: kernel_interface.c kernel_interface.h
> diff --git a/parser/parser_main.c b/parser/parser_main.c
> index d3c9701..b11d042 100644
> --- a/parser/parser_main.c
> +++ b/parser/parser_main.c
> @@ -79,8 +79,6 @@ int mru_skip_cache = 1;
>  int debug_cache = 0;
>  struct timespec mru_tstamp;
>  
> -char *cacheloc = NULL;
> -
>  /* Make sure to update BOTH the short and long_options */
>  static const char *short_options = 
> "adf:h::rRVvI:b:BCD:NSm:M:qQn:XKTWkL:O:po:";
>  struct option long_options[] = {
> @@ -837,9 +835,6 @@ static int binary_dir_cb(DIR *dir unused, const char 
> *name, struct stat *st,
>  
>  static void setup_flags(void)
>  {
> -     autofree char *cache_features_path = NULL;
> -     autofree char *cache_flags = NULL;
> -
>       /* Get the match string to determine type of regex support needed */
>       set_supported_features();
>  
> @@ -852,36 +847,6 @@ static void setup_flags(void)
>               skip_read_cache = 1;
>               return;
>       }
> -
> -
> -     /*
> -         * Deal with cache directory versioning:
> -         *  - If cache/.features is missing, create it if --write-cache.
> -         *  - If cache/.features exists, and does not match features_string,
> -         *    force cache reading/writing off.
> -         */
> -     if (asprintf(&cache_features_path, "%s/.features", cacheloc) == -1) {
> -             PERROR(_("Memory allocation error."));
> -             exit(1);
> -     }
> -
> -     cache_flags = load_features_file(cache_features_path);
> -     if (cache_flags) {
> -             if (strcmp(features_string, cache_flags) != 0) {
> -                     if (write_cache && cond_clear_cache) {
> -                             if (create_cache(cacheloc, cache_features_path,
> -                                              features_string))
> -                                     skip_read_cache = 1;
> -                     } else {
> -                             if (show_cache)
> -                                     PERROR("Cache read/write disabled: %s 
> does not match %s\n", FEATURES_FILE, cache_features_path);
> -                             write_cache = 0;
> -                             skip_read_cache = 1;
> -                     }
> -             }
> -     } else if (write_cache) {
> -             create_cache(cacheloc, cache_features_path, features_string);
> -     }
>  }
>  
>  int main(int argc, char *argv[])
> @@ -908,17 +873,6 @@ int main(int argc, char *argv[])
>               return retval;
>       }
>  
> -     /* create the cacheloc once and use it everywhere */
> -     if (!cacheloc) {
> -             if (asprintf(&cacheloc, "%s/cache", basedir) == -1) {
> -                     PERROR(_("Memory allocation error."));
> -                     exit(1);
> -             }
> -     }
> -
> -     if (force_clear_cache) 
> -             exit(clear_cache_files(cacheloc));
> -
>       /* Check to make sure there is an interface to load policy */
>       if (!(UNPRIVILEGED_OPS) && (subdomainbase == NULL) &&
>           !find_subdomainfs_mountpoint()) {
> @@ -929,6 +883,8 @@ int main(int argc, char *argv[])
>  
>       setup_flags();
>  
> +     setup_cache();
> +
>       retval = last_error = 0;
>       for (i = optind; i <= argc; i++) {
>               struct stat stat_file;
> diff --git a/parser/policy_cache.c b/parser/policy_cache.c
> index 7d9818a..aa1455c 100644
> --- a/parser/policy_cache.c
> +++ b/parser/policy_cache.c
> @@ -30,9 +30,12 @@
>  #define _(s) gettext(s)
>  
>  #include "lib.h"
> +#include "features.h"
>  #include "parser.h"
>  #include "policy_cache.h"
>  
> +char *cacheloc = NULL;
> +
>  #define le16_to_cpu(x) ((uint16_t)(le16toh (*(uint16_t *) x)))
>  
>  const char header_string[] = "\004\010\000version\000\002";
> @@ -226,3 +229,49 @@ void install_cache(const char *cachetmpname, const char 
> *cachename)
>               }
>       }
>  }
> +
> +void setup_cache(void)
> +{
> +     autofree char *cache_features_path = NULL;
> +     autofree char *cache_flags = NULL;
> +
> +     /* create the cacheloc once and use it everywhere */
> +     if (!cacheloc) {
> +             if (asprintf(&cacheloc, "%s/cache", basedir) == -1) {
> +                     PERROR(_("Memory allocation error."));
> +                     exit(1);
> +             }
> +     }
> +
> +     if (force_clear_cache)
> +             exit(clear_cache_files(cacheloc));
> +
> +     /*
> +         * Deal with cache directory versioning:
> +         *  - If cache/.features is missing, create it if --write-cache.
> +         *  - If cache/.features exists, and does not match features_string,
> +         *    force cache reading/writing off.
> +         */
> +     if (asprintf(&cache_features_path, "%s/.features", cacheloc) == -1) {
> +             PERROR(_("Memory allocation error."));
> +             exit(1);
> +     }
> +
> +     cache_flags = load_features_file(cache_features_path);
> +     if (cache_flags) {
> +             if (strcmp(features_string, cache_flags) != 0) {
> +                     if (write_cache && cond_clear_cache) {
> +                             if (create_cache(cacheloc, cache_features_path,
> +                                              features_string))
> +                                     skip_read_cache = 1;
> +                     } else {
> +                             if (show_cache)
> +                                     PERROR("Cache read/write disabled: %s 
> does not match %s\n", FEATURES_FILE, cache_features_path);
> +                             write_cache = 0;
> +                             skip_read_cache = 1;
> +                     }
> +             }
> +     } else if (write_cache) {
> +             create_cache(cacheloc, cache_features_path, features_string);
> +     }
> +}
> diff --git a/parser/policy_cache.h b/parser/policy_cache.h
> index 8572ff6..05b05bd 100644
> --- a/parser/policy_cache.h
> +++ b/parser/policy_cache.h
> @@ -35,6 +35,7 @@ extern int force_clear_cache;               /* force 
> clearing regargless of state */
>  extern int create_cache_dir;         /* create the cache dir if missing? */
>  extern int mru_skip_cache;
>  extern int debug_cache;
> +extern char *cacheloc;
>  
>  void set_mru_tstamp(struct timespec t);
>  void update_mru_tstamp(FILE *file, const char *path);
> @@ -46,5 +47,6 @@ void valid_read_cache(const char *cachename);
>  int cache_hit(const char *cachename);
>  int setup_cache_tmp(const char **cachetmpname, const char *cachename);
>  void install_cache(const char *cachetmpname, const char *cachename);
> +void setup_cache(void);
>  
>  #endif /* __AA_POLICY_CACHE_H */
> 



-- 
AppArmor mailing list
[email protected]
Modify settings or unsubscribe at: 
https://lists.ubuntu.com/mailman/listinfo/apparmor

Reply via email to