412 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			412 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| // SPDX-License-Identifier: GPL-2.0-only
 | |
| /*
 | |
|  * The hwprobe interface, for allowing userspace to probe to see which features
 | |
|  * are supported by the hardware.  See Documentation/arch/riscv/hwprobe.rst for
 | |
|  * more details.
 | |
|  */
 | |
| #include <linux/syscalls.h>
 | |
| #include <asm/cacheflush.h>
 | |
| #include <asm/cpufeature.h>
 | |
| #include <asm/hwprobe.h>
 | |
| #include <asm/sbi.h>
 | |
| #include <asm/switch_to.h>
 | |
| #include <asm/uaccess.h>
 | |
| #include <asm/unistd.h>
 | |
| #include <asm/vector.h>
 | |
| #include <vdso/vsyscall.h>
 | |
| 
 | |
| 
 | |
| static void hwprobe_arch_id(struct riscv_hwprobe *pair,
 | |
| 			    const struct cpumask *cpus)
 | |
| {
 | |
| 	u64 id = -1ULL;
 | |
| 	bool first = true;
 | |
| 	int cpu;
 | |
| 
 | |
| 	for_each_cpu(cpu, cpus) {
 | |
| 		u64 cpu_id;
 | |
| 
 | |
| 		switch (pair->key) {
 | |
| 		case RISCV_HWPROBE_KEY_MVENDORID:
 | |
| 			cpu_id = riscv_cached_mvendorid(cpu);
 | |
| 			break;
 | |
| 		case RISCV_HWPROBE_KEY_MIMPID:
 | |
| 			cpu_id = riscv_cached_mimpid(cpu);
 | |
| 			break;
 | |
| 		case RISCV_HWPROBE_KEY_MARCHID:
 | |
| 			cpu_id = riscv_cached_marchid(cpu);
 | |
| 			break;
 | |
| 		}
 | |
| 
 | |
| 		if (first) {
 | |
| 			id = cpu_id;
 | |
| 			first = false;
 | |
| 		}
 | |
| 
 | |
| 		/*
 | |
| 		 * If there's a mismatch for the given set, return -1 in the
 | |
| 		 * value.
 | |
| 		 */
 | |
| 		if (id != cpu_id) {
 | |
| 			id = -1ULL;
 | |
| 			break;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	pair->value = id;
 | |
| }
 | |
| 
 | |
| static void hwprobe_isa_ext0(struct riscv_hwprobe *pair,
 | |
| 			     const struct cpumask *cpus)
 | |
| {
 | |
| 	int cpu;
 | |
| 	u64 missing = 0;
 | |
| 
 | |
| 	pair->value = 0;
 | |
| 	if (has_fpu())
 | |
| 		pair->value |= RISCV_HWPROBE_IMA_FD;
 | |
| 
 | |
| 	if (riscv_isa_extension_available(NULL, c))
 | |
| 		pair->value |= RISCV_HWPROBE_IMA_C;
 | |
| 
 | |
| 	if (has_vector())
 | |
| 		pair->value |= RISCV_HWPROBE_IMA_V;
 | |
| 
 | |
| 	/*
 | |
| 	 * Loop through and record extensions that 1) anyone has, and 2) anyone
 | |
| 	 * doesn't have.
 | |
| 	 */
 | |
| 	for_each_cpu(cpu, cpus) {
 | |
| 		struct riscv_isainfo *isainfo = &hart_isa[cpu];
 | |
| 
 | |
| #define EXT_KEY(ext)									\
 | |
| 	do {										\
 | |
| 		if (__riscv_isa_extension_available(isainfo->isa, RISCV_ISA_EXT_##ext))	\
 | |
| 			pair->value |= RISCV_HWPROBE_EXT_##ext;				\
 | |
| 		else									\
 | |
| 			missing |= RISCV_HWPROBE_EXT_##ext;				\
 | |
| 	} while (false)
 | |
| 
 | |
| 		/*
 | |
| 		 * Only use EXT_KEY() for extensions which can be exposed to userspace,
 | |
| 		 * regardless of the kernel's configuration, as no other checks, besides
 | |
| 		 * presence in the hart_isa bitmap, are made.
 | |
| 		 */
 | |
| 		EXT_KEY(ZBA);
 | |
| 		EXT_KEY(ZBB);
 | |
| 		EXT_KEY(ZBS);
 | |
| 		EXT_KEY(ZICBOZ);
 | |
| 		EXT_KEY(ZBC);
 | |
| 
 | |
| 		EXT_KEY(ZBKB);
 | |
| 		EXT_KEY(ZBKC);
 | |
| 		EXT_KEY(ZBKX);
 | |
| 		EXT_KEY(ZKND);
 | |
| 		EXT_KEY(ZKNE);
 | |
| 		EXT_KEY(ZKNH);
 | |
| 		EXT_KEY(ZKSED);
 | |
| 		EXT_KEY(ZKSH);
 | |
| 		EXT_KEY(ZKT);
 | |
| 		EXT_KEY(ZIHINTNTL);
 | |
| 		EXT_KEY(ZTSO);
 | |
| 		EXT_KEY(ZACAS);
 | |
| 		EXT_KEY(ZICOND);
 | |
| 
 | |
| 		if (has_vector()) {
 | |
| 			EXT_KEY(ZVBB);
 | |
| 			EXT_KEY(ZVBC);
 | |
| 			EXT_KEY(ZVKB);
 | |
| 			EXT_KEY(ZVKG);
 | |
| 			EXT_KEY(ZVKNED);
 | |
| 			EXT_KEY(ZVKNHA);
 | |
| 			EXT_KEY(ZVKNHB);
 | |
| 			EXT_KEY(ZVKSED);
 | |
| 			EXT_KEY(ZVKSH);
 | |
| 			EXT_KEY(ZVKT);
 | |
| 			EXT_KEY(ZVFH);
 | |
| 			EXT_KEY(ZVFHMIN);
 | |
| 		}
 | |
| 
 | |
| 		if (has_fpu()) {
 | |
| 			EXT_KEY(ZFH);
 | |
| 			EXT_KEY(ZFHMIN);
 | |
| 			EXT_KEY(ZFA);
 | |
| 		}
 | |
| #undef EXT_KEY
 | |
| 	}
 | |
| 
 | |
| 	/* Now turn off reporting features if any CPU is missing it. */
 | |
| 	pair->value &= ~missing;
 | |
| }
 | |
| 
 | |
| static bool hwprobe_ext0_has(const struct cpumask *cpus, unsigned long ext)
 | |
| {
 | |
| 	struct riscv_hwprobe pair;
 | |
| 
 | |
| 	hwprobe_isa_ext0(&pair, cpus);
 | |
| 	return (pair.value & ext);
 | |
| }
 | |
| 
 | |
| static u64 hwprobe_misaligned(const struct cpumask *cpus)
 | |
| {
 | |
| 	int cpu;
 | |
| 	u64 perf = -1ULL;
 | |
| 
 | |
| 	for_each_cpu(cpu, cpus) {
 | |
| 		int this_perf = per_cpu(misaligned_access_speed, cpu);
 | |
| 
 | |
| 		if (perf == -1ULL)
 | |
| 			perf = this_perf;
 | |
| 
 | |
| 		if (perf != this_perf) {
 | |
| 			perf = RISCV_HWPROBE_MISALIGNED_UNKNOWN;
 | |
| 			break;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if (perf == -1ULL)
 | |
| 		return RISCV_HWPROBE_MISALIGNED_UNKNOWN;
 | |
| 
 | |
| 	return perf;
 | |
| }
 | |
| 
 | |
| static void hwprobe_one_pair(struct riscv_hwprobe *pair,
 | |
| 			     const struct cpumask *cpus)
 | |
| {
 | |
| 	switch (pair->key) {
 | |
| 	case RISCV_HWPROBE_KEY_MVENDORID:
 | |
| 	case RISCV_HWPROBE_KEY_MARCHID:
 | |
| 	case RISCV_HWPROBE_KEY_MIMPID:
 | |
| 		hwprobe_arch_id(pair, cpus);
 | |
| 		break;
 | |
| 	/*
 | |
| 	 * The kernel already assumes that the base single-letter ISA
 | |
| 	 * extensions are supported on all harts, and only supports the
 | |
| 	 * IMA base, so just cheat a bit here and tell that to
 | |
| 	 * userspace.
 | |
| 	 */
 | |
| 	case RISCV_HWPROBE_KEY_BASE_BEHAVIOR:
 | |
| 		pair->value = RISCV_HWPROBE_BASE_BEHAVIOR_IMA;
 | |
| 		break;
 | |
| 
 | |
| 	case RISCV_HWPROBE_KEY_IMA_EXT_0:
 | |
| 		hwprobe_isa_ext0(pair, cpus);
 | |
| 		break;
 | |
| 
 | |
| 	case RISCV_HWPROBE_KEY_CPUPERF_0:
 | |
| 		pair->value = hwprobe_misaligned(cpus);
 | |
| 		break;
 | |
| 
 | |
| 	case RISCV_HWPROBE_KEY_ZICBOZ_BLOCK_SIZE:
 | |
| 		pair->value = 0;
 | |
| 		if (hwprobe_ext0_has(cpus, RISCV_HWPROBE_EXT_ZICBOZ))
 | |
| 			pair->value = riscv_cboz_block_size;
 | |
| 		break;
 | |
| 
 | |
| 	/*
 | |
| 	 * For forward compatibility, unknown keys don't fail the whole
 | |
| 	 * call, but get their element key set to -1 and value set to 0
 | |
| 	 * indicating they're unrecognized.
 | |
| 	 */
 | |
| 	default:
 | |
| 		pair->key = -1;
 | |
| 		pair->value = 0;
 | |
| 		break;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| static int hwprobe_get_values(struct riscv_hwprobe __user *pairs,
 | |
| 			      size_t pair_count, size_t cpusetsize,
 | |
| 			      unsigned long __user *cpus_user,
 | |
| 			      unsigned int flags)
 | |
| {
 | |
| 	size_t out;
 | |
| 	int ret;
 | |
| 	cpumask_t cpus;
 | |
| 
 | |
| 	/* Check the reserved flags. */
 | |
| 	if (flags != 0)
 | |
| 		return -EINVAL;
 | |
| 
 | |
| 	/*
 | |
| 	 * The interface supports taking in a CPU mask, and returns values that
 | |
| 	 * are consistent across that mask. Allow userspace to specify NULL and
 | |
| 	 * 0 as a shortcut to all online CPUs.
 | |
| 	 */
 | |
| 	cpumask_clear(&cpus);
 | |
| 	if (!cpusetsize && !cpus_user) {
 | |
| 		cpumask_copy(&cpus, cpu_online_mask);
 | |
| 	} else {
 | |
| 		if (cpusetsize > cpumask_size())
 | |
| 			cpusetsize = cpumask_size();
 | |
| 
 | |
| 		ret = copy_from_user(&cpus, cpus_user, cpusetsize);
 | |
| 		if (ret)
 | |
| 			return -EFAULT;
 | |
| 
 | |
| 		/*
 | |
| 		 * Userspace must provide at least one online CPU, without that
 | |
| 		 * there's no way to define what is supported.
 | |
| 		 */
 | |
| 		cpumask_and(&cpus, &cpus, cpu_online_mask);
 | |
| 		if (cpumask_empty(&cpus))
 | |
| 			return -EINVAL;
 | |
| 	}
 | |
| 
 | |
| 	for (out = 0; out < pair_count; out++, pairs++) {
 | |
| 		struct riscv_hwprobe pair;
 | |
| 
 | |
| 		if (get_user(pair.key, &pairs->key))
 | |
| 			return -EFAULT;
 | |
| 
 | |
| 		pair.value = 0;
 | |
| 		hwprobe_one_pair(&pair, &cpus);
 | |
| 		ret = put_user(pair.key, &pairs->key);
 | |
| 		if (ret == 0)
 | |
| 			ret = put_user(pair.value, &pairs->value);
 | |
| 
 | |
| 		if (ret)
 | |
| 			return -EFAULT;
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static int hwprobe_get_cpus(struct riscv_hwprobe __user *pairs,
 | |
| 			    size_t pair_count, size_t cpusetsize,
 | |
| 			    unsigned long __user *cpus_user,
 | |
| 			    unsigned int flags)
 | |
| {
 | |
| 	cpumask_t cpus, one_cpu;
 | |
| 	bool clear_all = false;
 | |
| 	size_t i;
 | |
| 	int ret;
 | |
| 
 | |
| 	if (flags != RISCV_HWPROBE_WHICH_CPUS)
 | |
| 		return -EINVAL;
 | |
| 
 | |
| 	if (!cpusetsize || !cpus_user)
 | |
| 		return -EINVAL;
 | |
| 
 | |
| 	if (cpusetsize > cpumask_size())
 | |
| 		cpusetsize = cpumask_size();
 | |
| 
 | |
| 	ret = copy_from_user(&cpus, cpus_user, cpusetsize);
 | |
| 	if (ret)
 | |
| 		return -EFAULT;
 | |
| 
 | |
| 	if (cpumask_empty(&cpus))
 | |
| 		cpumask_copy(&cpus, cpu_online_mask);
 | |
| 
 | |
| 	cpumask_and(&cpus, &cpus, cpu_online_mask);
 | |
| 
 | |
| 	cpumask_clear(&one_cpu);
 | |
| 
 | |
| 	for (i = 0; i < pair_count; i++) {
 | |
| 		struct riscv_hwprobe pair, tmp;
 | |
| 		int cpu;
 | |
| 
 | |
| 		ret = copy_from_user(&pair, &pairs[i], sizeof(pair));
 | |
| 		if (ret)
 | |
| 			return -EFAULT;
 | |
| 
 | |
| 		if (!riscv_hwprobe_key_is_valid(pair.key)) {
 | |
| 			clear_all = true;
 | |
| 			pair = (struct riscv_hwprobe){ .key = -1, };
 | |
| 			ret = copy_to_user(&pairs[i], &pair, sizeof(pair));
 | |
| 			if (ret)
 | |
| 				return -EFAULT;
 | |
| 		}
 | |
| 
 | |
| 		if (clear_all)
 | |
| 			continue;
 | |
| 
 | |
| 		tmp = (struct riscv_hwprobe){ .key = pair.key, };
 | |
| 
 | |
| 		for_each_cpu(cpu, &cpus) {
 | |
| 			cpumask_set_cpu(cpu, &one_cpu);
 | |
| 
 | |
| 			hwprobe_one_pair(&tmp, &one_cpu);
 | |
| 
 | |
| 			if (!riscv_hwprobe_pair_cmp(&tmp, &pair))
 | |
| 				cpumask_clear_cpu(cpu, &cpus);
 | |
| 
 | |
| 			cpumask_clear_cpu(cpu, &one_cpu);
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if (clear_all)
 | |
| 		cpumask_clear(&cpus);
 | |
| 
 | |
| 	ret = copy_to_user(cpus_user, &cpus, cpusetsize);
 | |
| 	if (ret)
 | |
| 		return -EFAULT;
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static int do_riscv_hwprobe(struct riscv_hwprobe __user *pairs,
 | |
| 			    size_t pair_count, size_t cpusetsize,
 | |
| 			    unsigned long __user *cpus_user,
 | |
| 			    unsigned int flags)
 | |
| {
 | |
| 	if (flags & RISCV_HWPROBE_WHICH_CPUS)
 | |
| 		return hwprobe_get_cpus(pairs, pair_count, cpusetsize,
 | |
| 					cpus_user, flags);
 | |
| 
 | |
| 	return hwprobe_get_values(pairs, pair_count, cpusetsize,
 | |
| 				  cpus_user, flags);
 | |
| }
 | |
| 
 | |
| #ifdef CONFIG_MMU
 | |
| 
 | |
| static int __init init_hwprobe_vdso_data(void)
 | |
| {
 | |
| 	struct vdso_data *vd = __arch_get_k_vdso_data();
 | |
| 	struct arch_vdso_data *avd = &vd->arch_data;
 | |
| 	u64 id_bitsmash = 0;
 | |
| 	struct riscv_hwprobe pair;
 | |
| 	int key;
 | |
| 
 | |
| 	/*
 | |
| 	 * Initialize vDSO data with the answers for the "all CPUs" case, to
 | |
| 	 * save a syscall in the common case.
 | |
| 	 */
 | |
| 	for (key = 0; key <= RISCV_HWPROBE_MAX_KEY; key++) {
 | |
| 		pair.key = key;
 | |
| 		hwprobe_one_pair(&pair, cpu_online_mask);
 | |
| 
 | |
| 		WARN_ON_ONCE(pair.key < 0);
 | |
| 
 | |
| 		avd->all_cpu_hwprobe_values[key] = pair.value;
 | |
| 		/*
 | |
| 		 * Smash together the vendor, arch, and impl IDs to see if
 | |
| 		 * they're all 0 or any negative.
 | |
| 		 */
 | |
| 		if (key <= RISCV_HWPROBE_KEY_MIMPID)
 | |
| 			id_bitsmash |= pair.value;
 | |
| 	}
 | |
| 
 | |
| 	/*
 | |
| 	 * If the arch, vendor, and implementation ID are all the same across
 | |
| 	 * all harts, then assume all CPUs are the same, and allow the vDSO to
 | |
| 	 * answer queries for arbitrary masks. However if all values are 0 (not
 | |
| 	 * populated) or any value returns -1 (varies across CPUs), then the
 | |
| 	 * vDSO should defer to the kernel for exotic cpu masks.
 | |
| 	 */
 | |
| 	avd->homogeneous_cpus = id_bitsmash != 0 && id_bitsmash != -1;
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| arch_initcall_sync(init_hwprobe_vdso_data);
 | |
| 
 | |
| #endif /* CONFIG_MMU */
 | |
| 
 | |
| SYSCALL_DEFINE5(riscv_hwprobe, struct riscv_hwprobe __user *, pairs,
 | |
| 		size_t, pair_count, size_t, cpusetsize, unsigned long __user *,
 | |
| 		cpus, unsigned int, flags)
 | |
| {
 | |
| 	return do_riscv_hwprobe(pairs, pair_count, cpusetsize,
 | |
| 				cpus, flags);
 | |
| }
 |