linux-imx

// SPDX-License-Identifier: GPL-2.0+

/*

 * test_ida.c: Test the IDA API

 * Copyright (c) 2016-2018 Microsoft Corporation

 * Copyright (c) 2018 Oracle Corporation

 * Author: Matthew Wilcox <willy@infradead.org>

 */

​

#include <linux/idr.h>

#include <linux/module.h>

​

static unsigned int tests_run;

static unsigned int tests_passed;

​

#ifdef __KERNEL__

void ida_dump(struct ida *ida) { }

#endif

#define IDA_BUG_ON(ida, x) do {                     \

    tests_run++;                            \

    if (x) {                            \

        ida_dump(ida);                      \

        dump_stack();                       \

    } else {                            \

        tests_passed++;                     \

    }                               \

} while (0)

​

/*

 * Straightforward checks that allocating and freeing IDs work.

 */

static void ida_check_alloc(struct ida *ida)

{

    int i, id;

​

    for (i = 0; i < 10000; i++)

        IDA_BUG_ON(ida, ida_alloc(ida, GFP_KERNEL) != i);

​

    ida_free(ida, 20);

    ida_free(ida, 21);

    for (i = 0; i < 3; i++) {

        id = ida_alloc(ida, GFP_KERNEL);

        IDA_BUG_ON(ida, id < 0);

        if (i == 2)

            IDA_BUG_ON(ida, id != 10000);

    }

​

    for (i = 0; i < 5000; i++)

        ida_free(ida, i);

​

    IDA_BUG_ON(ida, ida_alloc_min(ida, 5000, GFP_KERNEL) != 10001);

    ida_destroy(ida);

​

    IDA_BUG_ON(ida, !ida_is_empty(ida));

}

​

/* Destroy an IDA with a single entry at @base */

static void ida_check_destroy_1(struct ida *ida, unsigned int base)

{

    IDA_BUG_ON(ida, ida_alloc_min(ida, base, GFP_KERNEL) != base);

    IDA_BUG_ON(ida, ida_is_empty(ida));

    ida_destroy(ida);

    IDA_BUG_ON(ida, !ida_is_empty(ida));

}