linux-imx

// SPDX-License-Identifier: GPL-2.0

​

/*

 * Test module for stress and analyze performance of vmalloc allocator.

 * (C) 2018 Uladzislau Rezki (Sony) <urezki@gmail.com>

 */

#include <linux/init.h>

#include <linux/kernel.h>

#include <linux/module.h>

#include <linux/vmalloc.h>

#include <linux/random.h>

#include <linux/kthread.h>

#include <linux/moduleparam.h>

#include <linux/completion.h>

#include <linux/delay.h>

#include <linux/rwsem.h>

#include <linux/mm.h>

​

#define __param(type, name, init, msg)      \

    static type name = init;                \

    module_param(name, type, 0444);         \

    MODULE_PARM_DESC(name, msg)             \

​

__param(bool, single_cpu_test, false,

    "Use single first online CPU to run tests");

​

__param(bool, sequential_test_order, false,

    "Use sequential stress tests order");

​

__param(int, test_repeat_count, 1,

    "Set test repeat counter");

​

__param(int, test_loop_count, 1000000,

    "Set test loop counter");

​

__param(int, run_test_mask, INT_MAX,

    "Set tests specified in the mask.\n\n"

        "\t\tid: 1,   name: fix_size_alloc_test\n"

        "\t\tid: 2,   name: full_fit_alloc_test\n"

        "\t\tid: 4,   name: long_busy_list_alloc_test\n"

        "\t\tid: 8,   name: random_size_alloc_test\n"

        "\t\tid: 16,  name: fix_align_alloc_test\n"

        "\t\tid: 32,  name: random_size_align_alloc_test\n"

        "\t\tid: 64,  name: align_shift_alloc_test\n"

        "\t\tid: 128, name: pcpu_alloc_test\n"

        /* Add a new test case description here. */

);

​

/*

 * Depends on single_cpu_test parameter. If it is true, then

 * use first online CPU to trigger a test on, otherwise go with

 * all online CPUs.

 */

static cpumask_t cpus_run_test_mask = CPU_MASK_NONE;

​

/*

 * Read write semaphore for synchronization of setup

 * phase that is done in main thread and workers.

 */

static DECLARE_RWSEM(prepare_for_test_rwsem);

​

/*

 * Completion tracking for worker threads.