aboutsummaryrefslogtreecommitdiff
path: root/tests/migration/stress.c
diff options
context:
space:
mode:
Diffstat (limited to 'tests/migration/stress.c')
-rw-r--r--tests/migration/stress.c328
1 files changed, 0 insertions, 328 deletions
diff --git a/tests/migration/stress.c b/tests/migration/stress.c
deleted file mode 100644
index 88acf8d..0000000
--- a/tests/migration/stress.c
+++ /dev/null
@@ -1,328 +0,0 @@
-/*
- * Migration stress workload
- *
- * Copyright (c) 2016 Red Hat, Inc.
- *
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 2.1 of the License, or (at your option) any later version.
- *
- * This library is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with this library; if not, see <http://www.gnu.org/licenses/>.
- */
-
-#include "qemu/osdep.h"
-#include <getopt.h>
-#include <sys/reboot.h>
-#include <sys/syscall.h>
-#include <linux/random.h>
-#include <pthread.h>
-#include <sys/mount.h>
-
-const char *argv0;
-
-#define RAM_PAGE_SIZE 4096
-
-#ifndef CONFIG_GETTID
-static int gettid(void)
-{
- return syscall(SYS_gettid);
-}
-#endif
-
-static __attribute__((noreturn)) void exit_failure(void)
-{
- if (getpid() == 1) {
- sync();
- reboot(RB_POWER_OFF);
- fprintf(stderr, "%s (%05d): ERROR: cannot reboot: %s\n",
- argv0, gettid(), strerror(errno));
- abort();
- } else {
- exit(1);
- }
-}
-
-static int get_command_arg_str(const char *name,
- char **val)
-{
- static char line[1024];
- FILE *fp = fopen("/proc/cmdline", "r");
- char *start, *end;
-
- if (fp == NULL) {
- fprintf(stderr, "%s (%05d): ERROR: cannot open /proc/cmdline: %s\n",
- argv0, gettid(), strerror(errno));
- return -1;
- }
-
- if (!fgets(line, sizeof line, fp)) {
- fprintf(stderr, "%s (%05d): ERROR: cannot read /proc/cmdline: %s\n",
- argv0, gettid(), strerror(errno));
- fclose(fp);
- return -1;
- }
- fclose(fp);
-
- start = strstr(line, name);
- if (!start)
- return 0;
-
- start += strlen(name);
-
- if (*start != '=') {
- fprintf(stderr, "%s (%05d): ERROR: no value provided for '%s' in /proc/cmdline\n",
- argv0, gettid(), name);
- }
- start++;
-
- end = strstr(start, " ");
- if (!end)
- end = strstr(start, "\n");
-
- if (end == start) {
- fprintf(stderr, "%s (%05d): ERROR: no value provided for '%s' in /proc/cmdline\n",
- argv0, gettid(), name);
- return -1;
- }
-
- if (end)
- *val = g_strndup(start, end - start);
- else
- *val = g_strdup(start);
- return 1;
-}
-
-
-static int get_command_arg_ull(const char *name,
- unsigned long long *val)
-{
- char *valstr;
- char *end;
-
- int ret = get_command_arg_str(name, &valstr);
- if (ret <= 0)
- return ret;
-
- errno = 0;
- *val = strtoll(valstr, &end, 10);
- if (errno || *end) {
- fprintf(stderr, "%s (%05d): ERROR: cannot parse %s value %s\n",
- argv0, gettid(), name, valstr);
- g_free(valstr);
- return -1;
- }
- g_free(valstr);
- return 0;
-}
-
-
-static int random_bytes(char *buf, size_t len)
-{
- int fd;
-
- fd = open("/dev/urandom", O_RDONLY);
- if (fd < 0) {
- fprintf(stderr, "%s (%05d): ERROR: cannot open /dev/urandom: %s\n",
- argv0, gettid(), strerror(errno));
- return -1;
- }
-
- if (read(fd, buf, len) != len) {
- fprintf(stderr, "%s (%05d): ERROR: cannot read /dev/urandom: %s\n",
- argv0, gettid(), strerror(errno));
- close(fd);
- return -1;
- }
-
- close(fd);
-
- return 0;
-}
-
-
-static unsigned long long now(void)
-{
- struct timeval tv;
-
- gettimeofday(&tv, NULL);
-
- return (tv.tv_sec * 1000ull) + (tv.tv_usec / 1000ull);
-}
-
-static void stressone(unsigned long long ramsizeMB)
-{
- size_t pagesPerMB = 1024 * 1024 / RAM_PAGE_SIZE;
- g_autofree char *ram = g_malloc(ramsizeMB * 1024 * 1024);
- char *ramptr;
- size_t i, j, k;
- g_autofree char *data = g_malloc(RAM_PAGE_SIZE);
- char *dataptr;
- size_t nMB = 0;
- unsigned long long before, after;
-
- /* We don't care about initial state, but we do want
- * to fault it all into RAM, otherwise the first iter
- * of the loop below will be quite slow. We can't use
- * 0x0 as the byte as gcc optimizes that away into a
- * calloc instead :-) */
- memset(ram, 0xfe, ramsizeMB * 1024 * 1024);
-
- if (random_bytes(data, RAM_PAGE_SIZE) < 0) {
- return;
- }
-
- before = now();
-
- while (1) {
-
- ramptr = ram;
- for (i = 0; i < ramsizeMB; i++, nMB++) {
- for (j = 0; j < pagesPerMB; j++) {
- dataptr = data;
- for (k = 0; k < RAM_PAGE_SIZE; k += sizeof(long long)) {
- ramptr += sizeof(long long);
- dataptr += sizeof(long long);
- *(unsigned long long *)ramptr ^= *(unsigned long long *)dataptr;
- }
- }
-
- if (nMB == 1024) {
- after = now();
- fprintf(stderr, "%s (%05d): INFO: %06llums copied 1 GB in %05llums\n",
- argv0, gettid(), after, after - before);
- before = now();
- nMB = 0;
- }
- }
- }
-}
-
-
-static void *stressthread(void *arg)
-{
- unsigned long long ramsizeMB = *(unsigned long long *)arg;
-
- stressone(ramsizeMB);
-
- return NULL;
-}
-
-static void stress(unsigned long long ramsizeGB, int ncpus)
-{
- size_t i;
- unsigned long long ramsizeMB = ramsizeGB * 1024 / ncpus;
- ncpus--;
-
- for (i = 0; i < ncpus; i++) {
- pthread_t thr;
- pthread_create(&thr, NULL,
- stressthread, &ramsizeMB);
- }
-
- stressone(ramsizeMB);
-}
-
-
-static int mount_misc(const char *fstype, const char *dir)
-{
- if (g_mkdir_with_parents(dir, 0755) < 0 && errno != EEXIST) {
- fprintf(stderr, "%s (%05d): ERROR: cannot create %s: %s\n",
- argv0, gettid(), dir, strerror(errno));
- return -1;
- }
-
- if (mount("none", dir, fstype, 0, NULL) < 0) {
- fprintf(stderr, "%s (%05d): ERROR: cannot mount %s: %s\n",
- argv0, gettid(), dir, strerror(errno));
- return -1;
- }
-
- return 0;
-}
-
-static int mount_all(void)
-{
- if (mount_misc("proc", "/proc") < 0 ||
- mount_misc("sysfs", "/sys") < 0 ||
- mount_misc("tmpfs", "/dev") < 0)
- return -1;
-
- mknod("/dev/urandom", 0777 | S_IFCHR, makedev(1, 9));
- mknod("/dev/random", 0777 | S_IFCHR, makedev(1, 8));
-
- return 0;
-}
-
-int main(int argc, char **argv)
-{
- unsigned long long ramsizeGB = 1;
- char *end;
- int ch;
- int opt_ind = 0;
- const char *sopt = "hr:c:";
- struct option lopt[] = {
- { "help", no_argument, NULL, 'h' },
- { "ramsize", required_argument, NULL, 'r' },
- { "cpus", required_argument, NULL, 'c' },
- { NULL, 0, NULL, 0 }
- };
- int ret;
- int ncpus = 0;
-
- argv0 = argv[0];
-
- while ((ch = getopt_long(argc, argv, sopt, lopt, &opt_ind)) != -1) {
- switch (ch) {
- case 'r':
- errno = 0;
- ramsizeGB = strtoll(optarg, &end, 10);
- if (errno != 0 || *end) {
- fprintf(stderr, "%s (%05d): ERROR: Cannot parse RAM size %s\n",
- argv0, gettid(), optarg);
- exit_failure();
- }
- break;
-
- case 'c':
- errno = 0;
- ncpus = strtoll(optarg, &end, 10);
- if (errno != 0 || *end) {
- fprintf(stderr, "%s (%05d): ERROR: Cannot parse CPU count %s\n",
- argv0, gettid(), optarg);
- exit_failure();
- }
- break;
-
- case '?':
- case 'h':
- fprintf(stderr, "%s: [--help][--ramsize GB][--cpus N]\n", argv0);
- exit_failure();
- }
- }
-
- if (getpid() == 1) {
- if (mount_all() < 0)
- exit_failure();
-
- ret = get_command_arg_ull("ramsize", &ramsizeGB);
- if (ret < 0)
- exit_failure();
- }
-
- if (ncpus == 0)
- ncpus = sysconf(_SC_NPROCESSORS_ONLN);
-
- fprintf(stdout, "%s (%05d): INFO: RAM %llu GiB across %d CPUs\n",
- argv0, gettid(), ramsizeGB, ncpus);
-
- stress(ramsizeGB, ncpus);
-
- exit_failure();
-}
generated by cgit v1.1 at 2025-07-05 00:04:38 +0000