/*
 *  $Id: features.c,v 1.26 2004/03/19 14:44:41 davej Exp $
 *  This file is part of x86info
 *  (C) 2001 Dave Jones.
 *
 *  Licensed under the terms of the GNU GPL License version 2.
 *
 *  Feature flag decoding.
 */

#include <stdio.h>
#include "x86info.h"

void decode_feature_flags (struct cpudata *cpu)
{
	unsigned long eax, ebx, ecx, edx;
	unsigned int i;

	/* CPUID 0x00000001 EDX flags */
	const char *generic_cap_flags[] = {
		"fpu", "vme", "de", "pse", "tsc", "msr", "pae", "mce",
		"cx8", "apic", NULL, "sep", "mtrr", "pge", "mca", "cmov",
		"pat", "pse36", "psn", "clflsh", NULL, "ds", "acpi", "mmx",
		"fxsr", "sse", "sse2", "ss", "ht", "tm", NULL, "pbe"
	};
	const char *generic_cap_flags_desc[] = {
		"Onboard FPU",
		"Virtual Mode Extensions",
		"Debugging Extensions",
		"Page Size Extensions",
		"Time Stamp Counter",
		"Model-Specific Registers",
		"Physical Address Extensions",
		"Machine Check Architecture",
		"CMPXCHG8 instruction",
		"Onboard APIC",
		NULL,
		"SYSENTER/SYSEXIT",
		"Memory Type Range Registers",
		"Page Global Enable",
		"Machine Check Architecture",
		"CMOV instruction",
		"Page Attribute Table",
		"36-bit PSEs",
		"Processor serial number",
		"CLFLUSH instruction",
		NULL,
		"Debug Trace Store",
		"ACPI via MSR",
		"MMX support",
		"FXSAVE and FXRESTORE instructions",
		"SSE support",
		"SSE2 support",
		"CPU self snoop",
		"Hyper-Threading",
		"Thermal Monitor",
		NULL,
		"Pending Break Enable"
	};
	/* CPUID 0x00000001 ECX flags */
	const char *intel_cap_generic_ecx_flags[] = {
		"sse3", NULL, NULL, "monitor", "ds-cpl", "vmx", NULL, "est",
		"tm2", "ssse3", "cntx-id", NULL, NULL, "cx16", "xTPR", NULL,
		NULL, NULL, "dca", NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL
	};
	/* CPUID 0x80000001 EDX flags */
	const char *intel_cap_extended_edx_flags[] = {
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, "SYSCALL", NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, "xd", NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, "em64t", NULL, NULL,
	};
	/* CPUID 0x80000001 ECX flags */
	const char *intel_cap_extended_ecx_flags[] = {
		"lahf_lm", NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
	};


	const char *amd_cap_generic_ecx_flags[] = {
		"sse3", NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, "cmpxchg16b", NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL
	};
	const char *amd_cap_extended_edx_flags[] = {
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, "syscall", NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, "mp/sempron", "nx", NULL, "mmxext", NULL,
		NULL, "Fast fxsave/fxrstor", NULL, "rdtscp", NULL, "lm", "3dnowext", "3dnow"
	};
	const char *amd_cap_extended_ecx_flags[] = {
		"lahf/sahf", "CMP legacy", "svm", "ExtApicSpace", "LockMovCr0", NULL, NULL, NULL,
		"3DNowPrefetch", NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL
	};

	const char *centaur_cap_flags[] = {
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, "mmxext", NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, "3dnowext", "3dnow"
	};
	const char *transmeta_cap_flags[] = {
		"recovery", "longrun", NULL, "lrti", NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
		NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL
	};

	cpuid(cpu->number, 0x00000001, &eax, &ebx, &ecx, &edx);
	cpu->flags_ecx = ecx;
	cpu->flags_edx = edx;
	if (cpu->maxei >= 0x80000001) {
		cpuid(cpu->number, 0x80000001, &eax, &ebx, &ecx, &edx);
		cpu->eflags_ecx = ecx;
		cpu->eflags_edx = edx;
	}

	if (silent != 0)
		return;

	printf ("Feature flags:\n");
	for (i = 0; i < 32; i++) {
		if (cpu->flags_edx & (1 << i)) {
			if (verbose) {
				printf ("\t%s\n", generic_cap_flags_desc[i]);
			} else {
				printf (" %s", generic_cap_flags[i]);
			}
		}
	}

	/* Vendor specific extensions. */
	switch (cpu->vendor) {
		case VENDOR_AMD:
			for (i = 0; i < 32; i++) {
				if (cpu->flags_ecx & (1 << i) && amd_cap_generic_ecx_flags[i])
					printf (" %s", amd_cap_generic_ecx_flags[i]);
			}
			printf ("\n");
			if (cpu->maxei < 0x80000001)
				break;
			printf ("Extended feature flags:\n");
			for (i = 0; i < 32; i++) {
				if (cpu->eflags_edx & (1 << i) && amd_cap_extended_edx_flags[i])
					printf (" %s", amd_cap_extended_edx_flags[i]);
			}
			for (i = 0; i < 32; i++) {
				if (cpu->eflags_ecx & (1 << i) && amd_cap_extended_ecx_flags[i])
					printf (" %s", amd_cap_extended_ecx_flags[i]);
			}
			break;

		case VENDOR_CENTAUR:
			printf ("\n");
			printf ("Extended feature flags:\n");
			for (i = 0; i < 32; i++) {
				if (cpu->flags_ecx & (1 << i) && centaur_cap_flags[i])
					printf (" %s", centaur_cap_flags[i]);
			}
			break;

		case VENDOR_TRANSMETA:
			printf ("\n");
			printf ("Extended feature flags:\n");
			for (i = 0; i < 32; i++) {
				if (cpu->flags_ecx & (1 << i) && transmeta_cap_flags[i])
					printf (" %s", transmeta_cap_flags[i]);
			}
			break;

		case VENDOR_CYRIX:
			printf ("\n");
			break;

		case VENDOR_INTEL:
			for (i = 0; i < 32; i++) {
				if (cpu->flags_ecx & (1 << i) && intel_cap_generic_ecx_flags[i])
					printf (" %s", intel_cap_generic_ecx_flags[i]);
			}
			printf ("\n");
			if (cpu->maxei < 0x80000001)
				break;
			printf ("Extended feature flags:\n");
			for (i = 0; i < 32; i++) {
				if (cpu->eflags_edx & (1 << i) && intel_cap_extended_edx_flags[i])
					printf (" %s", intel_cap_extended_edx_flags[i]);
			}
			for (i = 0; i < 32; i++) {
				if (cpu->eflags_ecx & (1 << i) && intel_cap_extended_ecx_flags[i])
					printf (" %s", intel_cap_extended_ecx_flags[i]);
			}
			break;

		default:
			/* Unknown CPU manufacturer or no special handling needed */
			break;
	}

	printf ("\n");
}