/*
 * Copyright 2005-2009, Haiku, Inc.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Axel Dörfler, axeld@pinc-software.de
 */


#include "amd.h"
#include "generic_x86.h"

#include <OS.h>


static void
amd_init_mtrrs(void)
{
	generic_init_mtrrs(generic_count_mtrrs());
}


static status_t
amd_init(void)
{
	system_info info;
	if (get_system_info(&info) != B_OK)
		return B_ERROR;

	if ((info.cpu_type & B_CPU_x86_VENDOR_MASK) != B_CPU_AMD_x86)
		return B_ERROR;

	// The K6-2 doesn't seem to support write-combining (before model 9),
	// so we ignore anything before that one.
	if (info.cpu_type <= B_CPU_AMD_K6_2)
		return B_ERROR;

	generic_mtrr_compute_physical_mask();
	generic_dump_mtrrs(generic_count_mtrrs());

	return B_OK;
}


static void
amd_set_mtrrs(const x86_mtrr_info* infos, uint32 count)
{
	generic_set_mtrrs(infos, count, generic_count_mtrrs());
}


static status_t
amd_stdops(int32 op, ...)
{
	switch (op) {
		case B_MODULE_INIT:
			return amd_init();
		case B_MODULE_UNINIT:
			return B_OK;
	}

	return B_ERROR;
}


x86_cpu_module_info gAMDModule = {
	{
		"cpu/generic_x86/amd/v1",
		0,
		amd_stdops,
	},

	generic_count_mtrrs,
	amd_init_mtrrs,

	generic_set_mtrr,
	generic_get_mtrr,
	amd_set_mtrrs
};