linux-imx

// SPDX-License-Identifier: GPL-2.0

/*

 * Range add and subtract

 */

#include <linux/kernel.h>

#include <linux/init.h>

#include <linux/sort.h>

#include <linux/string.h>

#include <linux/range.h>

​

int add_range(struct range *range, int az, int nr_range, u64 start, u64 end)

{

    if (start >= end)

        return nr_range;

​

    /* Out of slots: */

    if (nr_range >= az)

        return nr_range;

​

    range[nr_range].start = start;

    range[nr_range].end = end;

​

    nr_range++;

​

    return nr_range;

}

​

int add_range_with_merge(struct range *range, int az, int nr_range,

             u64 start, u64 end)

{

    int i;

​

    if (start >= end)

        return nr_range;

​

    /* get new start/end: */

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

        u64 common_start, common_end;

​

        if (!range[i].end)

            continue;

​

        common_start = max(range[i].start, start);

        common_end = min(range[i].end, end);

        if (common_start > common_end)

            continue;

​

        /* new start/end, will add it back at last */

        start = min(range[i].start, start);

        end = max(range[i].end, end);

​

        memmove(&range[i], &range[i + 1],

            (nr_range - (i + 1)) * sizeof(range[i]));

        range[nr_range - 1].start = 0;

        range[nr_range - 1].end   = 0;

        nr_range--;

        i--;

    }

​

    /* Need to add it: */

    return add_range(range, az, nr_range, start, end);

}

​