// SPDX-License-Identifier: GPL-2.0
#define boot_fmt(fmt)   "alt: " fmt
#include "boot.h"

#define a_debug         boot_debug

#include "../kernel/alternative.c"

static void alt_debug_all(int type)
{
        int i;

        switch (type) {
        case ALT_TYPE_FACILITY:
                for (i = 0; i < ARRAY_SIZE(alt_debug.facilities); i++)
                        alt_debug.facilities[i] = -1UL;
                break;
        case ALT_TYPE_FEATURE:
                for (i = 0; i < ARRAY_SIZE(alt_debug.mfeatures); i++)
                        alt_debug.mfeatures[i] = -1UL;
                break;
        case ALT_TYPE_SPEC:
                alt_debug.spec = 1;
                break;
        }
}

static void alt_debug_modify(int type, unsigned int nr, bool clear)
{
        switch (type) {
        case ALT_TYPE_FACILITY:
                if (clear)
                        __clear_facility(nr, alt_debug.facilities);
                else
                        __set_facility(nr, alt_debug.facilities);
                break;
        case ALT_TYPE_FEATURE:
                if (clear)
                        __clear_machine_feature(nr, alt_debug.mfeatures);
                else
                        __set_machine_feature(nr, alt_debug.mfeatures);
                break;
        }
}

static char *alt_debug_parse(int type, char *str)
{
        unsigned long val, endval;
        char *endp;
        bool clear;
        int i;

        if (*str == ':') {
                str++;
        } else {
                alt_debug_all(type);
                return str;
        }
        clear = false;
        if (*str == '!') {
                alt_debug_all(type);
                clear = true;
                str++;
        }
        while (*str) {
                val = simple_strtoull(str, &endp, 0);
                if (str == endp)
                        break;
                str = endp;
                if (*str == '-') {
                        str++;
                        endval = simple_strtoull(str, &endp, 0);
                        if (str == endp)
                                break;
                        str = endp;
                        while (val <= endval) {
                                alt_debug_modify(type, val, clear);
                                val++;
                        }
                } else {
                        alt_debug_modify(type, val, clear);
                }
                if (*str != ',')
                        break;
                str++;
        }
        return str;
}

/*
 * Use debug-alternative command line parameter for debugging:
 * "debug-alternative"
 *  -> print debug message for every single alternative
 *
 * "debug-alternative=0;2"
 * -> print debug message for all alternatives with type 0 and 2
 *
 * "debug-alternative=0:0-7"
 * -> print debug message for all alternatives with type 0 and with
 *    facility numbers within the range of 0-7
 *    (if type 0 is ALT_TYPE_FACILITY)
 *
 * "debug-alternative=0:!8;1"
 * -> print debug message for all alternatives with type 0, for all
 *    facility number, except facility 8, and in addition print all
 *    alternatives with type 1
 */
void alt_debug_setup(char *str)
{
        unsigned long type;
        char *endp;
        int i;

        if (!str) {
                alt_debug_all(ALT_TYPE_FACILITY);
                alt_debug_all(ALT_TYPE_FEATURE);
                alt_debug_all(ALT_TYPE_SPEC);
                return;
        }
        while (*str) {
                type = simple_strtoull(str, &endp, 0);
                if (str == endp)
                        break;
                str = endp;
                switch (type) {
                case ALT_TYPE_FACILITY:
                case ALT_TYPE_FEATURE:
                        str = alt_debug_parse(type, str);
                        break;
                case ALT_TYPE_SPEC:
                        alt_debug_all(ALT_TYPE_SPEC);
                        break;
                }
                if (*str != ';')
                        break;
                str++;
        }
}