cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

tracking.c (1460B)

      1// SPDX-License-Identifier: GPL-2.0-or-later
      2/*
      3 * Module taint unload tracking support
      4 *
      5 * Copyright (C) 2022 Aaron Tomlin
      6 */
      7
      8#include <linux/module.h>
      9#include <linux/string.h>
     10#include <linux/printk.h>
     11#include <linux/slab.h>
     12#include <linux/list.h>
     13#include <linux/rculist.h>
     14#include "internal.h"
     15
     16static LIST_HEAD(unloaded_tainted_modules);
     17
     18int try_add_tainted_module(struct module *mod)
     19{
     20	struct mod_unload_taint *mod_taint;
     21
     22	module_assert_mutex_or_preempt();
     23
     24	list_for_each_entry_rcu(mod_taint, &unloaded_tainted_modules, list,
     25				lockdep_is_held(&module_mutex)) {
     26		if (!strcmp(mod_taint->name, mod->name) &&
     27		    mod_taint->taints & mod->taints) {
     28			mod_taint->count++;
     29			goto out;
     30		}
     31	}
     32
     33	mod_taint = kmalloc(sizeof(*mod_taint), GFP_KERNEL);
     34	if (unlikely(!mod_taint))
     35		return -ENOMEM;
     36	strscpy(mod_taint->name, mod->name, MODULE_NAME_LEN);
     37	mod_taint->taints = mod->taints;
     38	list_add_rcu(&mod_taint->list, &unloaded_tainted_modules);
     39	mod_taint->count = 1;
     40out:
     41	return 0;
     42}
     43
     44void print_unloaded_tainted_modules(void)
     45{
     46	struct mod_unload_taint *mod_taint;
     47	char buf[MODULE_FLAGS_BUF_SIZE];
     48
     49	if (!list_empty(&unloaded_tainted_modules)) {
     50		printk(KERN_DEFAULT "Unloaded tainted modules:");
     51		list_for_each_entry_rcu(mod_taint, &unloaded_tainted_modules,
     52					list) {
     53			size_t l;
     54
     55			l = module_flags_taint(mod_taint->taints, buf);
     56			buf[l++] = '\0';
     57			pr_cont(" %s(%s):%llu", mod_taint->name, buf,
     58				mod_taint->count);
     59		}
     60	}
     61}