// SPDX-License-Identifier: GPL-2.0

#ifndef pr_fmt
#define pr_fmt(fmt)	"alt: " fmt
#endif

#include <linux/uaccess.h>
#include <linux/printk.h>
#include <asm/nospec-branch.h>
#include <asm/abs_lowcore.h>
#include <asm/alternative.h>
#include <asm/facility.h>
#include <asm/sections.h>
#include <asm/machine.h>

#ifndef a_debug
#define a_debug		pr_debug
#endif

#ifndef __kernel_va
#define __kernel_va(x)	(void *)(x)
#endif

unsigned long __bootdata_preserved(machine_features[1]);

struct alt_debug {
	unsigned long facilities[MAX_FACILITY_BIT / BITS_PER_LONG];
	unsigned long mfeatures[MAX_MFEATURE_BIT / BITS_PER_LONG];
	int spec;
};

static struct alt_debug __bootdata_preserved(alt_debug);

static void alternative_dump(u8 *old, u8 *new, unsigned int len, unsigned int type, unsigned int data)
{
	char oinsn[33], ninsn[33];
	unsigned long kptr;
	unsigned int pos;

	for (pos = 0; pos < len && 2 * pos < sizeof(oinsn) - 3; pos++)
		hex_byte_pack(&oinsn[2 * pos], old[pos]);
	oinsn[2 * pos] = 0;
	for (pos = 0; pos < len && 2 * pos < sizeof(ninsn) - 3; pos++)
		hex_byte_pack(&ninsn[2 * pos], new[pos]);
	ninsn[2 * pos] = 0;
	kptr = (unsigned long)__kernel_va(old);
	a_debug("[%d/%3d] %016lx: %s -> %s\n", type, data, kptr, oinsn, ninsn);
}

void __apply_alternatives(struct alt_instr *start, struct alt_instr *end, unsigned int ctx)
{
	struct alt_debug *d;
	struct alt_instr *a;
	bool debug, replace;
	u8 *old, *new;

	/*
	 * The scan order should be from start to end. A later scanned
	 * alternative code can overwrite previously scanned alternative code.
	 */
	d = &alt_debug;
	for (a = start; a < end; a++) {
		if (!(a->ctx & ctx))
			continue;
		switch (a->type) {
		case ALT_TYPE_FACILITY:
			replace = test_facility(a->data);
			debug = __test_facility(a->data, d->facilities);
			break;
		case ALT_TYPE_FEATURE:
			replace = test_machine_feature(a->data);
			debug = __test_machine_feature(a->data, d->mfeatures);
			break;
		case ALT_TYPE_SPEC:
			replace = nobp_enabled();
			debug = d->spec;
			break;
		default:
			replace = false;
			debug = false;
		}
		if (!replace)
			continue;
		old = (u8 *)&a->instr_offset + a->instr_offset;
		new = (u8 *)&a->repl_offset + a->repl_offset;
		if (debug)
			alternative_dump(old, new, a->instrlen, a->type, a->data);
		s390_kernel_write(old, new, a->instrlen);
	}
}