140 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			140 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| // SPDX-License-Identifier: GPL-2.0
 | |
| #include <sys/types.h>
 | |
| #include <errno.h>
 | |
| #include <unistd.h>
 | |
| #include <stdio.h>
 | |
| #include <stdlib.h>
 | |
| #include <string.h>
 | |
| #include <regex.h>
 | |
| 
 | |
| #include "../../../util/debug.h"
 | |
| #include "../../../util/header.h"
 | |
| #include "cpuid.h"
 | |
| 
 | |
| void get_cpuid_0(char *vendor, unsigned int *lvl)
 | |
| {
 | |
| 	unsigned int b, c, d;
 | |
| 
 | |
| 	cpuid(0, 0, lvl, &b, &c, &d);
 | |
| 	strncpy(&vendor[0], (char *)(&b), 4);
 | |
| 	strncpy(&vendor[4], (char *)(&d), 4);
 | |
| 	strncpy(&vendor[8], (char *)(&c), 4);
 | |
| 	vendor[12] = '\0';
 | |
| }
 | |
| 
 | |
| static int
 | |
| __get_cpuid(char *buffer, size_t sz, const char *fmt)
 | |
| {
 | |
| 	unsigned int a, b, c, d, lvl;
 | |
| 	int family = -1, model = -1, step = -1;
 | |
| 	int nb;
 | |
| 	char vendor[16];
 | |
| 
 | |
| 	get_cpuid_0(vendor, &lvl);
 | |
| 
 | |
| 	if (lvl >= 1) {
 | |
| 		cpuid(1, 0, &a, &b, &c, &d);
 | |
| 
 | |
| 		family = (a >> 8) & 0xf;  /* bits 11 - 8 */
 | |
| 		model  = (a >> 4) & 0xf;  /* Bits  7 - 4 */
 | |
| 		step   = a & 0xf;
 | |
| 
 | |
| 		/* extended family */
 | |
| 		if (family == 0xf)
 | |
| 			family += (a >> 20) & 0xff;
 | |
| 
 | |
| 		/* extended model */
 | |
| 		if (family >= 0x6)
 | |
| 			model += ((a >> 16) & 0xf) << 4;
 | |
| 	}
 | |
| 	nb = scnprintf(buffer, sz, fmt, vendor, family, model, step);
 | |
| 
 | |
| 	/* look for end marker to ensure the entire data fit */
 | |
| 	if (strchr(buffer, '$')) {
 | |
| 		buffer[nb-1] = '\0';
 | |
| 		return 0;
 | |
| 	}
 | |
| 	return ENOBUFS;
 | |
| }
 | |
| 
 | |
| int
 | |
| get_cpuid(char *buffer, size_t sz)
 | |
| {
 | |
| 	return __get_cpuid(buffer, sz, "%s,%u,%u,%u$");
 | |
| }
 | |
| 
 | |
| char *
 | |
| get_cpuid_str(struct perf_pmu *pmu __maybe_unused)
 | |
| {
 | |
| 	char *buf = malloc(128);
 | |
| 
 | |
| 	if (buf && __get_cpuid(buf, 128, "%s-%u-%X-%X$") < 0) {
 | |
| 		free(buf);
 | |
| 		return NULL;
 | |
| 	}
 | |
| 	return buf;
 | |
| }
 | |
| 
 | |
| /* Full CPUID format for x86 is vendor-family-model-stepping */
 | |
| static bool is_full_cpuid(const char *id)
 | |
| {
 | |
| 	const char *tmp = id;
 | |
| 	int count = 0;
 | |
| 
 | |
| 	while ((tmp = strchr(tmp, '-')) != NULL) {
 | |
| 		count++;
 | |
| 		tmp++;
 | |
| 	}
 | |
| 
 | |
| 	if (count == 3)
 | |
| 		return true;
 | |
| 
 | |
| 	return false;
 | |
| }
 | |
| 
 | |
| int strcmp_cpuid_str(const char *mapcpuid, const char *id)
 | |
| {
 | |
| 	regex_t re;
 | |
| 	regmatch_t pmatch[1];
 | |
| 	int match;
 | |
| 	bool full_mapcpuid = is_full_cpuid(mapcpuid);
 | |
| 	bool full_cpuid = is_full_cpuid(id);
 | |
| 
 | |
| 	/*
 | |
| 	 * Full CPUID format is required to identify a platform.
 | |
| 	 * Error out if the cpuid string is incomplete.
 | |
| 	 */
 | |
| 	if (full_mapcpuid && !full_cpuid) {
 | |
| 		pr_info("Invalid CPUID %s. Full CPUID is required, "
 | |
| 			"vendor-family-model-stepping\n", id);
 | |
| 		return 1;
 | |
| 	}
 | |
| 
 | |
| 	if (regcomp(&re, mapcpuid, REG_EXTENDED) != 0) {
 | |
| 		/* Warn unable to generate match particular string. */
 | |
| 		pr_info("Invalid regular expression %s\n", mapcpuid);
 | |
| 		return 1;
 | |
| 	}
 | |
| 
 | |
| 	match = !regexec(&re, id, 1, pmatch, 0);
 | |
| 	regfree(&re);
 | |
| 	if (match) {
 | |
| 		size_t match_len = (pmatch[0].rm_eo - pmatch[0].rm_so);
 | |
| 		size_t cpuid_len;
 | |
| 
 | |
| 		/* If the full CPUID format isn't required,
 | |
| 		 * ignoring the stepping.
 | |
| 		 */
 | |
| 		if (!full_mapcpuid && full_cpuid)
 | |
| 			cpuid_len = strrchr(id, '-') - id;
 | |
| 		else
 | |
| 			cpuid_len = strlen(id);
 | |
| 
 | |
| 		/* Verify the entire string matched. */
 | |
| 		if (match_len == cpuid_len)
 | |
| 			return 0;
 | |
| 	}
 | |
| 
 | |
| 	return 1;
 | |
| }
 |