git.sesse.net Git - bcachefs-tools-debian/blob - linux/shrinker.c

   1
   2 #include <stdio.h>
   3
   4 #include <linux/list.h>
   5 #include <linux/mutex.h>
   6 #include <linux/shrinker.h>
   7
   8 #include "tools-util.h"
   9
  10 static LIST_HEAD(shrinker_list);
  11 static DEFINE_MUTEX(shrinker_lock);
  12
  13 int register_shrinker(struct shrinker *shrinker)
  14 {
  15         mutex_lock(&shrinker_lock);
  16         list_add_tail(&shrinker->list, &shrinker_list);
  17         mutex_unlock(&shrinker_lock);
  18         return 0;
  19 }
  20
  21 void unregister_shrinker(struct shrinker *shrinker)
  22 {
  23         mutex_lock(&shrinker_lock);
  24         list_del(&shrinker->list);
  25         mutex_unlock(&shrinker_lock);
  26 }
  27
  28 struct meminfo {
  29         u64             total;
  30         u64             available;
  31
  32 };
  33
  34 static u64 parse_meminfo_line(const char *line)
  35 {
  36         u64 v;
  37
  38         if (sscanf(line, " %llu kB", &v) < 1)
  39                 die("sscanf error");
  40         return v << 10;
  41 }
  42
  43 static struct meminfo read_meminfo(void)
  44 {
  45         struct meminfo ret = { 0 };
  46         size_t len, n = 0;
  47         char *line = NULL;
  48         const char *v;
  49         FILE *f;
  50
  51         f = fopen("/proc/meminfo", "r");
  52         if (!f)
  53                 die("error opening /proc/meminfo: %m");
  54
  55         while ((len = getline(&line, &n, f)) != -1) {
  56                 if ((v = strcmp_prefix(line, "MemTotal:")))
  57                         ret.total = parse_meminfo_line(v);
  58
  59                 if ((v = strcmp_prefix(line, "MemAvailable:")))
  60                         ret.available = parse_meminfo_line(v);
  61         }
  62
  63         fclose(f);
  64         free(line);
  65
  66         return ret;
  67 }
  68
  69 void run_shrinkers(void)
  70 {
  71         struct shrinker *shrinker;
  72         struct meminfo info = read_meminfo();
  73         s64 want_shrink = (info.total >> 2) - info.available;
  74
  75         if (want_shrink <= 0)
  76                 return;
  77
  78         mutex_lock(&shrinker_lock);
  79         list_for_each_entry(shrinker, &shrinker_list, list) {
  80                 struct shrink_control sc = {
  81                         .nr_to_scan = want_shrink >> PAGE_SHIFT
  82                 };
  83
  84                 shrinker->scan_objects(shrinker, &sc);
  85         }
  86         mutex_unlock(&shrinker_lock);
  87 }