diff options
Diffstat (limited to 'security/tomoyo/memory.c')
| -rw-r--r-- | security/tomoyo/memory.c | 282 |
1 files changed, 282 insertions, 0 deletions
diff --git a/security/tomoyo/memory.c b/security/tomoyo/memory.c new file mode 100644 index 00000000000..297612669c7 --- /dev/null +++ b/security/tomoyo/memory.c | |||
| @@ -0,0 +1,282 @@ | |||
| 1 | /* | ||
| 2 | * security/tomoyo/memory.c | ||
| 3 | * | ||
| 4 | * Memory management functions for TOMOYO. | ||
| 5 | * | ||
| 6 | * Copyright (C) 2005-2010 NTT DATA CORPORATION | ||
| 7 | */ | ||
| 8 | |||
| 9 | #include <linux/hash.h> | ||
| 10 | #include <linux/slab.h> | ||
| 11 | #include "common.h" | ||
| 12 | |||
| 13 | /** | ||
| 14 | * tomoyo_warn_oom - Print out of memory warning message. | ||
| 15 | * | ||
| 16 | * @function: Function's name. | ||
| 17 | */ | ||
| 18 | void tomoyo_warn_oom(const char *function) | ||
| 19 | { | ||
| 20 | /* Reduce error messages. */ | ||
| 21 | static pid_t tomoyo_last_pid; | ||
| 22 | const pid_t pid = current->pid; | ||
| 23 | if (tomoyo_last_pid != pid) { | ||
| 24 | printk(KERN_WARNING "ERROR: Out of memory at %s.\n", | ||
| 25 | function); | ||
| 26 | tomoyo_last_pid = pid; | ||
| 27 | } | ||
| 28 | if (!tomoyo_policy_loaded) | ||
| 29 | panic("MAC Initialization failed.\n"); | ||
| 30 | } | ||
| 31 | |||
| 32 | /* Memory allocated for policy. */ | ||
| 33 | static atomic_t tomoyo_policy_memory_size; | ||
| 34 | /* Quota for holding policy. */ | ||
| 35 | static unsigned int tomoyo_quota_for_policy; | ||
| 36 | |||
| 37 | /** | ||
| 38 | * tomoyo_memory_ok - Check memory quota. | ||
| 39 | * | ||
| 40 | * @ptr: Pointer to allocated memory. | ||
| 41 | * | ||
| 42 | * Returns true on success, false otherwise. | ||
| 43 | * | ||
| 44 | * Returns true if @ptr is not NULL and quota not exceeded, false otherwise. | ||
| 45 | */ | ||
| 46 | bool tomoyo_memory_ok(void *ptr) | ||
| 47 | { | ||
| 48 | size_t s = ptr ? ksize(ptr) : 0; | ||
| 49 | atomic_add(s, &tomoyo_policy_memory_size); | ||
| 50 | if (ptr && (!tomoyo_quota_for_policy || | ||
| 51 | atomic_read(&tomoyo_policy_memory_size) | ||
| 52 | <= tomoyo_quota_for_policy)) { | ||
| 53 | memset(ptr, 0, s); | ||
| 54 | return true; | ||
| 55 | } | ||
| 56 | atomic_sub(s, &tomoyo_policy_memory_size); | ||
| 57 | tomoyo_warn_oom(__func__); | ||
| 58 | return false; | ||
| 59 | } | ||
| 60 | |||
| 61 | /** | ||
| 62 | * tomoyo_commit_ok - Check memory quota. | ||
| 63 | * | ||
| 64 | * @data: Data to copy from. | ||
| 65 | * @size: Size in byte. | ||
| 66 | * | ||
| 67 | * Returns pointer to allocated memory on success, NULL otherwise. | ||
| 68 | * @data is zero-cleared on success. | ||
| 69 | */ | ||
| 70 | void *tomoyo_commit_ok(void *data, const unsigned int size) | ||
| 71 | { | ||
| 72 | void *ptr = kzalloc(size, GFP_NOFS); | ||
| 73 | if (tomoyo_memory_ok(ptr)) { | ||
| 74 | memmove(ptr, data, size); | ||
| 75 | memset(data, 0, size); | ||
| 76 | return ptr; | ||
| 77 | } | ||
| 78 | return NULL; | ||
| 79 | } | ||
| 80 | |||
| 81 | /** | ||
| 82 | * tomoyo_memory_free - Free memory for elements. | ||
| 83 | * | ||
| 84 | * @ptr: Pointer to allocated memory. | ||
| 85 | */ | ||
| 86 | void tomoyo_memory_free(void *ptr) | ||
| 87 | { | ||
| 88 | atomic_sub(ksize(ptr), &tomoyo_policy_memory_size); | ||
| 89 | kfree(ptr); | ||
| 90 | } | ||
| 91 | |||
| 92 | /** | ||
| 93 | * tomoyo_get_group - Allocate memory for "struct tomoyo_path_group"/"struct tomoyo_number_group". | ||
| 94 | * | ||
| 95 | * @group_name: The name of address group. | ||
| 96 | * @idx: Index number. | ||
| 97 | * | ||
| 98 | * Returns pointer to "struct tomoyo_group" on success, NULL otherwise. | ||
| 99 | */ | ||
| 100 | struct tomoyo_group *tomoyo_get_group(const char *group_name, const u8 idx) | ||
| 101 | { | ||
| 102 | struct tomoyo_group e = { }; | ||
| 103 | struct tomoyo_group *group = NULL; | ||
| 104 | bool found = false; | ||
| 105 | if (!tomoyo_correct_word(group_name) || idx >= TOMOYO_MAX_GROUP) | ||
| 106 | return NULL; | ||
| 107 | e.group_name = tomoyo_get_name(group_name); | ||
| 108 | if (!e.group_name) | ||
| 109 | return NULL; | ||
| 110 | if (mutex_lock_interruptible(&tomoyo_policy_lock)) | ||
| 111 | goto out; | ||
| 112 | list_for_each_entry(group, &tomoyo_group_list[idx], list) { | ||
| 113 | if (e.group_name != group->group_name) | ||
| 114 | continue; | ||
| 115 | atomic_inc(&group->users); | ||
| 116 | found = true; | ||
| 117 | break; | ||
| 118 | } | ||
| 119 | if (!found) { | ||
| 120 | struct tomoyo_group *entry = tomoyo_commit_ok(&e, sizeof(e)); | ||
| 121 | if (entry) { | ||
| 122 | INIT_LIST_HEAD(&entry->member_list); | ||
| 123 | atomic_set(&entry->users, 1); | ||
| 124 | list_add_tail_rcu(&entry->list, | ||
| 125 | &tomoyo_group_list[idx]); | ||
| 126 | group = entry; | ||
| 127 | found = true; | ||
| 128 | } | ||
| 129 | } | ||
| 130 | mutex_unlock(&tomoyo_policy_lock); | ||
| 131 | out: | ||
| 132 | tomoyo_put_name(e.group_name); | ||
| 133 | return found ? group : NULL; | ||
| 134 | } | ||
| 135 | |||
| 136 | /* | ||
| 137 | * tomoyo_name_list is used for holding string data used by TOMOYO. | ||
| 138 | * Since same string data is likely used for multiple times (e.g. | ||
| 139 | * "/lib/libc-2.5.so"), TOMOYO shares string data in the form of | ||
| 140 | * "const struct tomoyo_path_info *". | ||
| 141 | */ | ||
| 142 | struct list_head tomoyo_name_list[TOMOYO_MAX_HASH]; | ||
| 143 | |||
| 144 | /** | ||
| 145 | * tomoyo_get_name - Allocate permanent memory for string data. | ||
| 146 | * | ||
| 147 | * @name: The string to store into the permernent memory. | ||
| 148 | * | ||
| 149 | * Returns pointer to "struct tomoyo_path_info" on success, NULL otherwise. | ||
| 150 | */ | ||
| 151 | const struct tomoyo_path_info *tomoyo_get_name(const char *name) | ||
| 152 | { | ||
| 153 | struct tomoyo_name *ptr; | ||
| 154 | unsigned int hash; | ||
| 155 | int len; | ||
| 156 | int allocated_len; | ||
| 157 | struct list_head *head; | ||
| 158 | |||
| 159 | if (!name) | ||
| 160 | return NULL; | ||
| 161 | len = strlen(name) + 1; | ||
| 162 | hash = full_name_hash((const unsigned char *) name, len - 1); | ||
| 163 | head = &tomoyo_name_list[hash_long(hash, TOMOYO_HASH_BITS)]; | ||
| 164 | if (mutex_lock_interruptible(&tomoyo_policy_lock)) | ||
| 165 | return NULL; | ||
| 166 | list_for_each_entry(ptr, head, list) { | ||
| 167 | if (hash != ptr->entry.hash || strcmp(name, ptr->entry.name)) | ||
| 168 | continue; | ||
| 169 | atomic_inc(&ptr->users); | ||
| 170 | goto out; | ||
| 171 | } | ||
| 172 | ptr = kzalloc(sizeof(*ptr) + len, GFP_NOFS); | ||
| 173 | allocated_len = ptr ? ksize(ptr) : 0; | ||
| 174 | if (!ptr || (tomoyo_quota_for_policy && | ||
| 175 | atomic_read(&tomoyo_policy_memory_size) + allocated_len | ||
| 176 | > tomoyo_quota_for_policy)) { | ||
| 177 | kfree(ptr); | ||
| 178 | ptr = NULL; | ||
| 179 | tomoyo_warn_oom(__func__); | ||
| 180 | goto out; | ||
| 181 | } | ||
| 182 | atomic_add(allocated_len, &tomoyo_policy_memory_size); | ||
| 183 | ptr->entry.name = ((char *) ptr) + sizeof(*ptr); | ||
| 184 | memmove((char *) ptr->entry.name, name, len); | ||
| 185 | atomic_set(&ptr->users, 1); | ||
| 186 | tomoyo_fill_path_info(&ptr->entry); | ||
| 187 | list_add_tail(&ptr->list, head); | ||
| 188 | out: | ||
| 189 | mutex_unlock(&tomoyo_policy_lock); | ||
| 190 | return ptr ? &ptr->entry : NULL; | ||
| 191 | } | ||
| 192 | |||
| 193 | /** | ||
| 194 | * tomoyo_mm_init - Initialize mm related code. | ||
| 195 | */ | ||
| 196 | void __init tomoyo_mm_init(void) | ||
| 197 | { | ||
| 198 | int idx; | ||
| 199 | |||
| 200 | for (idx = 0; idx < TOMOYO_MAX_POLICY; idx++) | ||
| 201 | INIT_LIST_HEAD(&tomoyo_policy_list[idx]); | ||
| 202 | for (idx = 0; idx < TOMOYO_MAX_GROUP; idx++) | ||
| 203 | INIT_LIST_HEAD(&tomoyo_group_list[idx]); | ||
| 204 | for (idx = 0; idx < TOMOYO_MAX_HASH; idx++) | ||
| 205 | INIT_LIST_HEAD(&tomoyo_name_list[idx]); | ||
| 206 | INIT_LIST_HEAD(&tomoyo_kernel_domain.acl_info_list); | ||
| 207 | tomoyo_kernel_domain.domainname = tomoyo_get_name(TOMOYO_ROOT_NAME); | ||
| 208 | list_add_tail_rcu(&tomoyo_kernel_domain.list, &tomoyo_domain_list); | ||
| 209 | idx = tomoyo_read_lock(); | ||
| 210 | if (tomoyo_find_domain(TOMOYO_ROOT_NAME) != &tomoyo_kernel_domain) | ||
| 211 | panic("Can't register tomoyo_kernel_domain"); | ||
| 212 | { | ||
| 213 | /* Load built-in policy. */ | ||
| 214 | tomoyo_write_transition_control("/sbin/hotplug", false, | ||
| 215 | TOMOYO_TRANSITION_CONTROL_INITIALIZE); | ||
| 216 | tomoyo_write_transition_control("/sbin/modprobe", false, | ||
| 217 | TOMOYO_TRANSITION_CONTROL_INITIALIZE); | ||
| 218 | } | ||
| 219 | tomoyo_read_unlock(idx); | ||
| 220 | } | ||
| 221 | |||
| 222 | |||
| 223 | /* Memory allocated for query lists. */ | ||
| 224 | unsigned int tomoyo_query_memory_size; | ||
| 225 | /* Quota for holding query lists. */ | ||
| 226 | unsigned int tomoyo_quota_for_query; | ||
| 227 | |||
| 228 | /** | ||
| 229 | * tomoyo_read_memory_counter - Check for memory usage in bytes. | ||
| 230 | * | ||
| 231 | * @head: Pointer to "struct tomoyo_io_buffer". | ||
| 232 | * | ||
| 233 | * Returns memory usage. | ||
| 234 | */ | ||
| 235 | void tomoyo_read_memory_counter(struct tomoyo_io_buffer *head) | ||
| 236 | { | ||
| 237 | if (!head->r.eof) { | ||
| 238 | const unsigned int policy | ||
| 239 | = atomic_read(&tomoyo_policy_memory_size); | ||
| 240 | const unsigned int query = tomoyo_query_memory_size; | ||
| 241 | char buffer[64]; | ||
| 242 | |||
| 243 | memset(buffer, 0, sizeof(buffer)); | ||
| 244 | if (tomoyo_quota_for_policy) | ||
| 245 | snprintf(buffer, sizeof(buffer) - 1, | ||
| 246 | " (Quota: %10u)", | ||
| 247 | tomoyo_quota_for_policy); | ||
| 248 | else | ||
| 249 | buffer[0] = '\0'; | ||
| 250 | tomoyo_io_printf(head, "Policy: %10u%s\n", policy, | ||
| 251 | buffer); | ||
| 252 | if (tomoyo_quota_for_query) | ||
| 253 | snprintf(buffer, sizeof(buffer) - 1, | ||
| 254 | " (Quota: %10u)", | ||
| 255 | tomoyo_quota_for_query); | ||
| 256 | else | ||
| 257 | buffer[0] = '\0'; | ||
| 258 | tomoyo_io_printf(head, "Query lists: %10u%s\n", query, | ||
| 259 | buffer); | ||
| 260 | tomoyo_io_printf(head, "Total: %10u\n", policy + query); | ||
| 261 | head->r.eof = true; | ||
| 262 | } | ||
| 263 | } | ||
| 264 | |||
| 265 | /** | ||
| 266 | * tomoyo_write_memory_quota - Set memory quota. | ||
| 267 | * | ||
| 268 | * @head: Pointer to "struct tomoyo_io_buffer". | ||
| 269 | * | ||
| 270 | * Returns 0. | ||
| 271 | */ | ||
| 272 | int tomoyo_write_memory_quota(struct tomoyo_io_buffer *head) | ||
| 273 | { | ||
| 274 | char *data = head->write_buf; | ||
| 275 | unsigned int size; | ||
| 276 | |||
| 277 | if (sscanf(data, "Policy: %u", &size) == 1) | ||
| 278 | tomoyo_quota_for_policy = size; | ||
| 279 | else if (sscanf(data, "Query lists: %u", &size) == 1) | ||
| 280 | tomoyo_quota_for_query = size; | ||
| 281 | return 0; | ||
| 282 | } | ||
