|
 |
2a5d721 |
/* Start/stop KSM, for systemd.
|
|
|
2a5d721 |
* Copyright (C) 2009, 2011 Red Hat, Inc.
|
|
|
2a5d721 |
* Written by Paolo Bonzini <pbonzini@redhat.com>.
|
|
|
2a5d721 |
* Based on the original sysvinit script by Dan Kenigsberg <danken@redhat.com>
|
|
|
2a5d721 |
* This file is distributed under the GNU General Public License, version 2
|
|
|
2a5d721 |
* or later. */
|
|
|
2a5d721 |
|
|
|
2a5d721 |
#include <unistd.h>
|
|
|
2a5d721 |
#include <stdio.h>
|
|
|
2a5d721 |
#include <limits.h>
|
|
|
2a5d721 |
#include <stdint.h>
|
|
|
2a5d721 |
#include <stdlib.h>
|
|
|
2a5d721 |
#include <string.h>
|
|
|
2a5d721 |
|
|
|
2a5d721 |
#define KSM_MAX_KERNEL_PAGES_FILE "/sys/kernel/mm/ksm/max_kernel_pages"
|
|
|
2a5d721 |
#define KSM_RUN_FILE "/sys/kernel/mm/ksm/run"
|
|
|
2a5d721 |
|
|
|
2a5d721 |
char *program_name;
|
|
|
2a5d721 |
|
|
|
2a5d721 |
int usage(void)
|
|
|
2a5d721 |
{
|
|
|
2a5d721 |
fprintf(stderr, "Usage: %s {start|stop}\n", program_name);
|
|
|
2a5d721 |
return 1;
|
|
|
2a5d721 |
}
|
|
|
2a5d721 |
|
|
|
2a5d721 |
int write_value(uint64_t value, char *filename)
|
|
|
2a5d721 |
{
|
|
|
2a5d721 |
FILE *fp;
|
|
|
2a5d721 |
if (!(fp = fopen(filename, "w")) ||
|
|
|
2a5d721 |
fprintf(fp, "%llu\n", (unsigned long long) value) == EOF ||
|
|
|
2a5d721 |
fflush(fp) == EOF ||
|
|
|
2a5d721 |
fclose(fp) == EOF)
|
|
|
2a5d721 |
return 1;
|
|
|
2a5d721 |
|
|
|
2a5d721 |
return 0;
|
|
|
2a5d721 |
}
|
|
|
2a5d721 |
|
|
|
2a5d721 |
uint64_t ksm_max_kernel_pages()
|
|
|
2a5d721 |
{
|
|
|
2a5d721 |
char *var = getenv("KSM_MAX_KERNEL_PAGES");
|
|
|
2a5d721 |
char *endptr;
|
|
|
2a5d721 |
uint64_t value;
|
|
|
2a5d721 |
if (var && *var) {
|
|
|
2a5d721 |
value = strtoll(var, &endptr, 0);
|
|
|
2a5d721 |
if (value < LLONG_MAX && !*endptr)
|
|
|
2a5d721 |
return value;
|
|
|
2a5d721 |
}
|
|
|
2a5d721 |
/* Unless KSM_MAX_KERNEL_PAGES is set, let KSM munch up to half of
|
|
|
2a5d721 |
* total memory. */
|
|
|
2a5d721 |
return sysconf(_SC_PHYS_PAGES) / 2;
|
|
|
2a5d721 |
}
|
|
|
2a5d721 |
|
|
|
2a5d721 |
int start(void)
|
|
|
2a5d721 |
{
|
|
|
2a5d721 |
if (access(KSM_MAX_KERNEL_PAGES_FILE, R_OK) >= 0)
|
|
|
2a5d721 |
write_value(ksm_max_kernel_pages(), KSM_MAX_KERNEL_PAGES_FILE);
|
|
|
2a5d721 |
return write_value(1, KSM_RUN_FILE);
|
|
|
2a5d721 |
}
|
|
|
2a5d721 |
|
|
|
2a5d721 |
int stop(void)
|
|
|
2a5d721 |
{
|
|
|
2a5d721 |
return write_value(0, KSM_RUN_FILE);
|
|
|
2a5d721 |
}
|
|
|
2a5d721 |
|
|
|
2a5d721 |
int main(int argc, char **argv)
|
|
|
2a5d721 |
{
|
|
|
2a5d721 |
program_name = argv[0];
|
|
|
2a5d721 |
if (argc < 2) {
|
|
|
2a5d721 |
return usage();
|
|
|
2a5d721 |
} else if (!strcmp(argv[1], "start")) {
|
|
|
2a5d721 |
return start();
|
|
|
2a5d721 |
} else if (!strcmp(argv[1], "stop")) {
|
|
|
2a5d721 |
return stop();
|
|
|
2a5d721 |
} else {
|
|
|
2a5d721 |
return usage();
|
|
|
2a5d721 |
}
|
|
|
2a5d721 |
}
|