[apparmor] [PATCH v2 12/42] parser: Move policy cache initialization code into its own function
Tyler Hicks
tyhicks at canonical.com
Fri Mar 6 21:48:28 UTC 2015
This patch moves the logic that sets up the policy into a new function
in policy_cache.c
Signed-off-by: Tyler Hicks <tyhicks at canonical.com>
Acked-by: John Johansen <john.johansen at canonical.com>
---
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 9735ea4..372136d 100644
--- a/parser/Makefile
+++ b/parser/Makefile
@@ -240,7 +240,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 01ed0e2..ec70bea 100644
--- a/parser/parser_main.c
+++ b/parser/parser_main.c
@@ -80,8 +80,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[] = {
@@ -856,9 +854,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();
@@ -871,36 +866,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[])
@@ -927,17 +892,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()) {
@@ -948,6 +902,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 */
--
2.1.4
More information about the AppArmor
mailing list