/*
 * Copyright (c) 1996, by Steve Passe
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. The name of the developer may NOT be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 * $FreeBSD: src/sys/i386/i386/mp_machdep.c,v 1.115.2.15 2003/03/14 21:22:35 jhb Exp $
 */

#include "opt_cpu.h"

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/sysctl.h>
#include <sys/malloc.h>
#include <sys/memrange.h>
#include <sys/cons.h>	/* cngetc() */
#include <sys/machintr.h>
#include <sys/cpu_topology.h>

#include <sys/mplock2.h>

#include <vm/vm.h>
#include <vm/vm_param.h>
#include <vm/pmap.h>
#include <vm/vm_kern.h>
#include <vm/vm_extern.h>
#include <sys/lock.h>
#include <vm/vm_map.h>

#include <machine/smp.h>
#include <machine_base/apic/apicreg.h>
#include <machine/atomic.h>
#include <machine/cpufunc.h>
#include <machine/cputypes.h>
#include <machine_base/apic/lapic.h>
#include <machine_base/apic/ioapic.h>
#include <machine_base/acpica/acpi_md_cpu.h>
#include <machine/psl.h>
#include <machine/segments.h>
#include <machine/tss.h>
#include <machine/specialreg.h>
#include <machine/globaldata.h>
#include <machine/pmap_inval.h>
#include <machine/clock.h>

#include <machine/md_var.h>		/* setidt() */
#include <machine_base/icu/icu.h>	/* IPIs */
#include <machine_base/icu/icu_var.h>
#include <machine_base/apic/ioapic_abi.h>
#include <machine/intr_machdep.h>	/* IPIs */

#define WARMBOOT_TARGET		0
#define WARMBOOT_OFF		(KERNBASE + 0x0467)
#define WARMBOOT_SEG		(KERNBASE + 0x0469)

#define CMOS_REG		(0x70)
#define CMOS_DATA		(0x71)
#define BIOS_RESET		(0x0f)
#define BIOS_WARM		(0x0a)

#define INVLPG_TIMEOUT_DEFAULT	10
#define INVLPG_TIMEOUT_VM	60

/*
 * this code MUST be enabled here and in mpboot.s.
 * it follows the very early stages of AP boot by placing values in CMOS ram.
 * it NORMALLY will never be needed and thus the primitive method for enabling.
 *
 */
#if defined(CHECK_POINTS)
#define CHECK_READ(A)	 (outb(CMOS_REG, (A)), inb(CMOS_DATA))
#define CHECK_WRITE(A,D) (outb(CMOS_REG, (A)), outb(CMOS_DATA, (D)))

#define CHECK_INIT(D);				\
	CHECK_WRITE(0x34, (D));			\
	CHECK_WRITE(0x35, (D));			\
	CHECK_WRITE(0x36, (D));			\
	CHECK_WRITE(0x37, (D));			\
	CHECK_WRITE(0x38, (D));			\
	CHECK_WRITE(0x39, (D));

#define CHECK_PRINT(S);				\
	kprintf("%s: %d, %d, %d, %d, %d, %d\n",	\
	   (S),					\
	   CHECK_READ(0x34),			\
	   CHECK_READ(0x35),			\
	   CHECK_READ(0x36),			\
	   CHECK_READ(0x37),			\
	   CHECK_READ(0x38),			\
	   CHECK_READ(0x39));

#else				/* CHECK_POINTS */

#define CHECK_INIT(D)
#define CHECK_PRINT(S)

#endif				/* CHECK_POINTS */

/*
 * Values to send to the POST hardware.
 */
#define MP_BOOTADDRESS_POST	0x10
#define MP_PROBE_POST		0x11
#define MPTABLE_PASS1_POST	0x12

#define MP_START_POST		0x13
#define MP_ENABLE_POST		0x14
#define MPTABLE_PASS2_POST	0x15

#define START_ALL_APS_POST	0x16
#define INSTALL_AP_TRAMP_POST	0x17
#define START_AP_POST		0x18

#define MP_ANNOUNCE_POST	0x19

/** XXX FIXME: where does this really belong, isa.h/isa.c perhaps? */
int	current_postcode;

/** XXX FIXME: what system files declare these??? */

extern int naps;
extern int _udatasel;

int64_t tsc0_offset;
extern int64_t tsc_offsets[];

/* AP uses this during bootstrap.  Do not staticize.  */
char *bootSTK;
static int bootAP;

struct pcb stoppcbs[MAXCPU];

extern inthand_t IDTVEC(fast_syscall), IDTVEC(fast_syscall32);

/*
 * Local data and functions.
 */

static u_int	boot_address;
static int	mp_finish;
static int	mp_finish_lapic;

static int	start_all_aps(u_int boot_addr);
#if 0
static void	install_ap_tramp(u_int boot_addr);
#endif
static int	start_ap(struct mdglobaldata *gd, u_int boot_addr, int smibest);
static int	smitest(void);
static void	mp_bsp_simple_setup(void);

/* which cpus have been started */
__read_mostly static cpumask_t smp_startup_mask = CPUMASK_INITIALIZER_ONLYONE;
/* which cpus have lapic been inited */
__read_mostly static cpumask_t smp_lapic_mask = CPUMASK_INITIALIZER_ONLYONE;
/* which cpus are ready for IPIs etc? */
__read_mostly cpumask_t smp_active_mask = CPUMASK_INITIALIZER_ONLYONE;
__read_mostly cpumask_t smp_finalize_mask = CPUMASK_INITIALIZER_ONLYONE;

SYSCTL_OPAQUE(_machdep, OID_AUTO, smp_active, CTLFLAG_RD,
	      &smp_active_mask, sizeof(smp_active_mask), "LU", "");
static u_int	bootMP_size;
__read_mostly static u_int	report_invlpg_src;
SYSCTL_INT(_machdep, OID_AUTO, report_invlpg_src, CTLFLAG_RW,
	&report_invlpg_src, 0, "");
__read_mostly static u_int	report_invltlb_src;
SYSCTL_INT(_machdep, OID_AUTO, report_invltlb_src, CTLFLAG_RW,
	&report_invltlb_src, 0, "");
__read_mostly static int	optimized_invltlb;
SYSCTL_INT(_machdep, OID_AUTO, optimized_invltlb, CTLFLAG_RW,
	&optimized_invltlb, 0, "");
__read_mostly static int	all_but_self_ipi_enable = 1;
SYSCTL_INT(_machdep, OID_AUTO, all_but_self_ipi_enable, CTLFLAG_RW,
	&all_but_self_ipi_enable, 0, "");
__read_mostly static int	invlpg_timeout = INVLPG_TIMEOUT_DEFAULT;
SYSCTL_INT(_machdep, OID_AUTO, invlpg_timeout, CTLFLAG_RW,
	&invlpg_timeout, 0, "");

/* Local data for detecting CPU TOPOLOGY */
static int core_bits = 0;
static int logical_CPU_bits = 0;


/*
 * Calculate usable address in base memory for AP trampoline code.
 */
u_int
mp_bootaddress(u_int basemem)
{
	POSTCODE(MP_BOOTADDRESS_POST);

	bootMP_size = mptramp_end - mptramp_start;
	boot_address = trunc_page(basemem * 1024); /* round down to 4k boundary */
	if (((basemem * 1024) - boot_address) < bootMP_size)
		boot_address -= PAGE_SIZE;	/* not enough, lower by 4k */
	/* 3 levels of page table pages */
	mptramp_pagetables = boot_address - (PAGE_SIZE * 3);

	return mptramp_pagetables;
}

/*
 * Print various information about the SMP system hardware and setup.
 */
void
mp_announce(void)
{
	int     x;

	POSTCODE(MP_ANNOUNCE_POST);

	kprintf("DragonFly/MP: Multiprocessor motherboard\n");
	kprintf(" cpu0 (BSP): apic id: %2d\n", CPUID_TO_APICID(0));
	for (x = 1; x <= naps; ++x)
		kprintf(" cpu%d (AP):  apic id: %2d\n", x, CPUID_TO_APICID(x));

	if (!ioapic_enable)
		kprintf(" Warning: APIC I/O disabled\n");
}

/*
 * AP cpu's call this to sync up protected mode.
 *
 * WARNING! %gs is not set up on entry.  This routine sets up %gs.
 */
void
init_secondary(void)
{
	int	gsel_tss;
	int	x, myid = bootAP;
	u_int64_t msr, cr0;
	struct mdglobaldata *md;
	struct privatespace *ps;
	struct user_segment_descriptor *gdt;

	ps = CPU_prvspace[myid];
	gdt = ps->mdglobaldata.gd_gdt;

	gdt_segs[GPROC0_SEL].ssd_base = (long)&ps->common_tss;
	ps->mdglobaldata.mi.gd_prvspace = ps;

	/* We fill the 32-bit segment descriptors */
	for (x = 0; x < NGDT; x++) {
		if (x != GPROC0_SEL && x != (GPROC0_SEL + 1))
			ssdtosd(&gdt_segs[x], &gdt[x]);
	}
	/* And now a 64-bit one */
	ssdtosyssd(&gdt_segs[GPROC0_SEL],
	    (struct system_segment_descriptor *)&gdt[GPROC0_SEL]);

	r_gdt.rd_limit = MAXGDT_LIMIT - 1;
	r_gdt.rd_base = (long)(intptr_t)gdt;
	lgdt(&r_gdt);			/* does magic intra-segment return */

	/* lgdt() destroys the GSBASE value, so we load GSBASE after lgdt() */
	wrmsr(MSR_FSBASE, 0);		/* User value */
	wrmsr(MSR_GSBASE, (u_int64_t)ps);
	wrmsr(MSR_KGSBASE, 0);		/* XXX User value while we're in the kernel */

	lidt(&r_idt_arr[mdcpu->mi.gd_cpuid]);

	load_ds(_udatasel);
	load_es(_udatasel);
	load_fs(_udatasel);

#if 0
	lldt(_default_ldt);
	mdcpu->gd_currentldt = _default_ldt;
#endif

	gsel_tss = GSEL(GPROC0_SEL, SEL_KPL);
	gdt[GPROC0_SEL].sd_type = SDT_SYSTSS;

	md = mdcpu;	/* loaded through %gs:0 (mdglobaldata.mi.gd_prvspace)*/

	/*
	 * TSS entry point for interrupts, traps, and exceptions
	 * (sans NMI).  This will always go to near the top of the pcpu
	 * trampoline area.  Hardware-pushed data will be copied into
	 * the trap-frame on entry, and (if necessary) returned to the
	 * trampoline on exit.
	 *
	 * We store some pcb data for the trampoline code above the
	 * stack the cpu hw pushes into, and arrange things so the
	 * address of tr_pcb_rsp is the same as the desired top of
	 * stack.
	 */
	ps->common_tss.tss_rsp0 = (register_t)&ps->trampoline.tr_pcb_rsp;
	ps->trampoline.tr_pcb_rsp = ps->common_tss.tss_rsp0;
	ps->trampoline.tr_pcb_gs_kernel = (register_t)md;
	ps->trampoline.tr_pcb_cr3 = KPML4phys;	/* adj to user cr3 live */
	ps->dbltramp.tr_pcb_gs_kernel = (register_t)md;
	ps->dbltramp.tr_pcb_cr3 = KPML4phys;
	ps->dbgtramp.tr_pcb_gs_kernel = (register_t)md;
	ps->dbgtramp.tr_pcb_cr3 = KPML4phys;

#if 0 /* JG XXX */
	ps->common_tss.tss_ioopt = (sizeof ps->common_tss) << 16;
#endif
	md->gd_tss_gdt = &gdt[GPROC0_SEL];
	md->gd_common_tssd = *md->gd_tss_gdt;

	/* double fault stack */
	ps->common_tss.tss_ist1 = (register_t)&ps->dbltramp.tr_pcb_rsp;
	ps->common_tss.tss_ist2 = (register_t)&ps->dbgtramp.tr_pcb_rsp;

	ltr(gsel_tss);

	/*
	 * Set to a known state:
	 * Set by mpboot.s: CR0_PG, CR0_PE
	 * Set by cpu_setregs: CR0_NE, CR0_MP, CR0_TS, CR0_WP, CR0_AM
	 */
	cr0 = rcr0();
	cr0 &= ~(CR0_CD | CR0_NW | CR0_EM);
	load_cr0(cr0);

	/* Set up the fast syscall stuff */
	msr = rdmsr(MSR_EFER) | EFER_SCE;
	wrmsr(MSR_EFER, msr);
	wrmsr(MSR_LSTAR, (u_int64_t)IDTVEC(fast_syscall));
	wrmsr(MSR_CSTAR, (u_int64_t)IDTVEC(fast_syscall32));
	msr = ((u_int64_t)GSEL(GCODE_SEL, SEL_KPL) << 32) |
	      ((u_int64_t)GSEL(GUCODE32_SEL, SEL_UPL) << 48);
	wrmsr(MSR_STAR, msr);
	wrmsr(MSR_SF_MASK, PSL_NT|PSL_T|PSL_I|PSL_C|PSL_D|PSL_IOPL|PSL_AC);

	pmap_set_opt();		/* PSE/4MB pages, etc */
	pmap_init_pat();	/* Page Attribute Table */

	/* set up CPU registers and state */
	cpu_setregs();

	/* set up SSE/NX registers */
	initializecpu(myid);

	/* set up FPU state on the AP */
	npxinit();

	/* If BSP is in the X2APIC mode, put the AP into the X2APIC mode. */
	if (x2apic_enable)
		lapic_x2apic_enter(FALSE);

	/* disable the APIC, just to be SURE */
	LAPIC_WRITE(svr, (LAPIC_READ(svr) & ~APIC_SVR_ENABLE));
}

/*******************************************************************
 * local functions and data
 */

/*
 * Start the SMP system
 */
static void
mp_start_aps(void *dummy __unused)
{
	if (lapic_enable) {
		/* start each Application Processor */
		start_all_aps(boot_address);
	} else {
		mp_bsp_simple_setup();
	}
}
SYSINIT(startaps, SI_BOOT2_START_APS, SI_ORDER_FIRST, mp_start_aps, NULL);

/*
 * start each AP in our list
 */
static int
start_all_aps(u_int boot_addr)
{
	vm_offset_t va = boot_address + KERNBASE;
	u_int64_t *pt4, *pt3, *pt2;
	int	pssize;
	int     x, i;
	int	shift;
	int	smicount;
	int	smibest;
	int	smilast;
	u_char  mpbiosreason;
	u_long  mpbioswarmvec;
	struct mdglobaldata *gd;
	struct privatespace *ps;
	size_t ipiq_size;

	POSTCODE(START_ALL_APS_POST);

	/* install the AP 1st level boot code */
	pmap_kenter(va, boot_address);
	cpu_invlpg((void *)va);		/* JG XXX */
	bcopy(mptramp_start, (void *)va, bootMP_size);

	/* Locate the page tables, they'll be below the trampoline */
	pt4 = (u_int64_t *)(uintptr_t)(mptramp_pagetables + KERNBASE);
	pt3 = pt4 + (PAGE_SIZE) / sizeof(u_int64_t);
	pt2 = pt3 + (PAGE_SIZE) / sizeof(u_int64_t);

	/* Create the initial 1GB replicated page tables */
	for (i = 0; i < 512; i++) {
		/* Each slot of the level 4 pages points to the same level 3 page */
		pt4[i] = (u_int64_t)(uintptr_t)(mptramp_pagetables + PAGE_SIZE);
		pt4[i] |= kernel_pmap->pmap_bits[PG_V_IDX] |
		    kernel_pmap->pmap_bits[PG_RW_IDX] |
		    kernel_pmap->pmap_bits[PG_U_IDX];

		/* Each slot of the level 3 pages points to the same level 2 page */
		pt3[i] = (u_int64_t)(uintptr_t)(mptramp_pagetables + (2 * PAGE_SIZE));
		pt3[i] |= kernel_pmap->pmap_bits[PG_V_IDX] |
		    kernel_pmap->pmap_bits[PG_RW_IDX] |
		    kernel_pmap->pmap_bits[PG_U_IDX];

		/* The level 2 page slots are mapped with 2MB pages for 1GB. */
		pt2[i] = i * (2 * 1024 * 1024);
		pt2[i] |= kernel_pmap->pmap_bits[PG_V_IDX] |
		    kernel_pmap->pmap_bits[PG_RW_IDX] |
		    kernel_pmap->pmap_bits[PG_PS_IDX] |
		    kernel_pmap->pmap_bits[PG_U_IDX];
	}

	/* save the current value of the warm-start vector */
	mpbioswarmvec = *((u_int32_t *) WARMBOOT_OFF);
	outb(CMOS_REG, BIOS_RESET);
	mpbiosreason = inb(CMOS_DATA);

	/* setup a vector to our boot code */
	*((volatile u_short *) WARMBOOT_OFF) = WARMBOOT_TARGET;
	*((volatile u_short *) WARMBOOT_SEG) = (boot_address >> 4);
	outb(CMOS_REG, BIOS_RESET);
	outb(CMOS_DATA, BIOS_WARM);	/* 'warm-start' */

	/*
	 * If we have a TSC we can figure out the SMI interrupt rate.
	 * The SMI does not necessarily use a constant rate.  Spend
	 * up to 250ms trying to figure it out.
	 */
	smibest = 0;
	if (cpu_feature & CPUID_TSC) {
		set_apic_timer(275000);
		smilast = read_apic_timer();
		for (x = 0; x < 20 && read_apic_timer(); ++x) {
			smicount = smitest();
			if (smibest == 0 || smilast - smicount < smibest)
				smibest = smilast - smicount;
			smilast = smicount;
		}
		if (smibest > 250000)
			smibest = 0;
	}
	if (smibest)
		kprintf("SMI Frequency (worst case): %d Hz (%d us)\n",
			1000000 / smibest, smibest);

	/*
	 * This is nasty but if we are a guest in a virtual machine,
	 * give the smpinvl synchronization code up to 60 seconds
	 */

	if (vmm_guest != VMM_GUEST_NONE)
		invlpg_timeout = INVLPG_TIMEOUT_VM;

	/* start each AP */
	for (x = 1; x <= naps; ++x) {
		/* This is a bit verbose, it will go away soon.  */

		pssize = sizeof(struct privatespace);
		ps = (void *)
			kmem_alloc3(kernel_map, pssize, VM_SUBSYS_GD,
				    KM_CPU(x));
		bzero(ps, pssize);
		CPU_prvspace[x] = ps;
		gd = &ps->mdglobaldata;
		gd->mi.gd_prvspace = ps;
		gd->gd_gdt = (void *)
			kmem_alloc3(kernel_map, MAXGDT_LIMIT, VM_SUBSYS_GD,
				    KM_CPU(x));
		bzero(gd->gd_gdt, MAXGDT_LIMIT);

#if 0
		kprintf("ps %d %p %d\n", x, ps, pssize);
#endif

		/* prime data page for it to use */
		mi_gdinit(&gd->mi, x);
		cpu_gdinit(gd, x);
		ipiq_size = sizeof(struct lwkt_ipiq) * (naps + 1);
		gd->mi.gd_ipiq = (void *)kmem_alloc3(kernel_map, ipiq_size,
						     VM_SUBSYS_IPIQ, KM_CPU(x));
		bzero(gd->mi.gd_ipiq, ipiq_size);

		gd->gd_acpi_id = CPUID_TO_ACPIID(gd->mi.gd_cpuid);

		/* initialize arc4random. */
		arc4_init_pcpu(x);

		/* setup a vector to our boot code */
		*((volatile u_short *) WARMBOOT_OFF) = WARMBOOT_TARGET;
		*((volatile u_short *) WARMBOOT_SEG) = (boot_addr >> 4);
		outb(CMOS_REG, BIOS_RESET);
		outb(CMOS_DATA, BIOS_WARM);	/* 'warm-start' */

		/*
		 * Setup the AP boot stack
		 */
		bootSTK = &ps->idlestack[UPAGES * PAGE_SIZE - PAGE_SIZE];
		bootAP = x;

		/* attempt to start the Application Processor */
		CHECK_INIT(99);	/* setup checkpoints */
		if (!start_ap(gd, boot_addr, smibest)) {
			kprintf("\nAP #%d (PHY# %d) failed!\n",
				x, CPUID_TO_APICID(x));
			CHECK_PRINT("trace");	/* show checkpoints */
			/* better panic as the AP may be running loose */
			kprintf("panic y/n? [y] ");
			cnpoll(TRUE);
			if (cngetc() != 'n')
				panic("bye-bye");
			cnpoll(FALSE);
		}
		CHECK_PRINT("trace");		/* show checkpoints */
	}

	/* set ncpus to 1 + highest logical cpu.  Not all may have come up */
	ncpus = x;

	for (shift = 0; (1 << shift) <= ncpus; ++shift)
		;
	--shift;

	/* ncpus_fit -- ncpus rounded up to the nearest power of 2 */
	if ((1 << shift) < ncpus)
		++shift;
	ncpus_fit = 1 << shift;
	ncpus_fit_mask = ncpus_fit - 1;

	/* build our map of 'other' CPUs */
	mycpu->gd_other_cpus = smp_startup_mask;
	CPUMASK_NANDBIT(mycpu->gd_other_cpus, mycpu->gd_cpuid);

	malloc_reinit_ncpus();

	gd = (struct mdglobaldata *)mycpu;
	gd->gd_acpi_id = CPUID_TO_ACPIID(mycpu->gd_cpuid);

	ipiq_size = sizeof(struct lwkt_ipiq) * ncpus;
	mycpu->gd_ipiq = (void *)kmem_alloc3(kernel_map, ipiq_size,
					     VM_SUBSYS_IPIQ, KM_CPU(0));
	bzero(mycpu->gd_ipiq, ipiq_size);

	/* initialize arc4random. */
	arc4_init_pcpu(0);

	/* restore the warmstart vector */
	*(u_long *) WARMBOOT_OFF = mpbioswarmvec;
	outb(CMOS_REG, BIOS_RESET);
	outb(CMOS_DATA, mpbiosreason);

	/*
	 * NOTE!  The idlestack for the BSP was setup by locore.  Finish
	 * up, clean out the P==V mapping we did earlier.
	 */
	pmap_set_opt();

	/*
	 * Wait all APs to finish initializing LAPIC
	 */
	if (bootverbose)
		kprintf("SMP: Waiting APs LAPIC initialization\n");
	if (cpu_feature & CPUID_TSC)
		tsc0_offset = rdtsc();
	tsc_offsets[0] = 0;
	mp_finish_lapic = 1;
	rel_mplock();

	while (CPUMASK_CMPMASKNEQ(smp_lapic_mask, smp_startup_mask)) {
		cpu_pause();
		cpu_lfence();
		if (cpu_feature & CPUID_TSC)
			tsc0_offset = rdtsc();
	}
	while (try_mplock() == 0) {
		cpu_pause();
		cpu_lfence();
	}

	/* number of APs actually started */
	return ncpus - 1;
}


/*
 * load the 1st level AP boot code into base memory.
 */

/* targets for relocation */
extern void bigJump(void);
extern void bootCodeSeg(void);
extern void bootDataSeg(void);
extern void MPentry(void);
extern u_int MP_GDT;
extern u_int mp_gdtbase;

#if 0

static void
install_ap_tramp(u_int boot_addr)
{
	int     x;
	int     size = *(int *) ((u_long) & bootMP_size);
	u_char *src = (u_char *) ((u_long) bootMP);
	u_char *dst = (u_char *) boot_addr + KERNBASE;
	u_int   boot_base = (u_int) bootMP;
	u_int8_t *dst8;
	u_int16_t *dst16;
	u_int32_t *dst32;

	POSTCODE(INSTALL_AP_TRAMP_POST);

	for (x = 0; x < size; ++x)
		*dst++ = *src++;

	/*
	 * modify addresses in code we just moved to basemem. unfortunately we
	 * need fairly detailed info about mpboot.s for this to work.  changes
	 * to mpboot.s might require changes here.
	 */

	/* boot code is located in KERNEL space */
	dst = (u_char *) boot_addr + KERNBASE;

	/* modify the lgdt arg */
	dst32 = (u_int32_t *) (dst + ((u_int) & mp_gdtbase - boot_base));
	*dst32 = boot_addr + ((u_int) & MP_GDT - boot_base);

	/* modify the ljmp target for MPentry() */
	dst32 = (u_int32_t *) (dst + ((u_int) bigJump - boot_base) + 1);
	*dst32 = ((u_int) MPentry - KERNBASE);

	/* modify the target for boot code segment */
	dst16 = (u_int16_t *) (dst + ((u_int) bootCodeSeg - boot_base));
	dst8 = (u_int8_t *) (dst16 + 1);
	*dst16 = (u_int) boot_addr & 0xffff;
	*dst8 = ((u_int) boot_addr >> 16) & 0xff;

	/* modify the target for boot data segment */
	dst16 = (u_int16_t *) (dst + ((u_int) bootDataSeg - boot_base));
	dst8 = (u_int8_t *) (dst16 + 1);
	*dst16 = (u_int) boot_addr & 0xffff;
	*dst8 = ((u_int) boot_addr >> 16) & 0xff;
}

#endif

/*
 * This function starts the AP (application processor) identified
 * by the APIC ID 'physicalCpu'.  It does quite a "song and dance"
 * to accomplish this.  This is necessary because of the nuances
 * of the different hardware we might encounter.  It ain't pretty,
 * but it seems to work.
 *
 * NOTE: eventually an AP gets to ap_init(), which is called just 
 * before the AP goes into the LWKT scheduler's idle loop.
 */
static int
start_ap(struct mdglobaldata *gd, u_int boot_addr, int smibest)
{
	int     physical_cpu;
	int     vector;

	POSTCODE(START_AP_POST);

	/* get the PHYSICAL APIC ID# */
	physical_cpu = CPUID_TO_APICID(gd->mi.gd_cpuid);

	/* calculate the vector */
	vector = (boot_addr >> 12) & 0xff;

	/* We don't want anything interfering */
	cpu_disable_intr();

	/* Make sure the target cpu sees everything */
	wbinvd();

	/*
	 * Try to detect when a SMI has occurred, wait up to 200ms.
	 *
	 * If a SMI occurs during an AP reset but before we issue
	 * the STARTUP command, the AP may brick.  To work around
	 * this problem we hold off doing the AP startup until
	 * after we have detected the SMI.  Hopefully another SMI
	 * will not occur before we finish the AP startup.
	 *
	 * Retries don't seem to help.  SMIs have a window of opportunity
	 * and if USB->legacy keyboard emulation is enabled in the BIOS
	 * the interrupt rate can be quite high.
	 *
	 * NOTE: Don't worry about the L1 cache load, it might bloat
	 *	 ldelta a little but ndelta will be so huge when the SMI
	 *	 occurs the detection logic will still work fine.
	 */
	if (smibest) {
		set_apic_timer(200000);
		smitest();
	}

	/*
	 * first we do an INIT/RESET IPI this INIT IPI might be run, reseting
	 * and running the target CPU. OR this INIT IPI might be latched (P5
	 * bug), CPU waiting for STARTUP IPI. OR this INIT IPI might be
	 * ignored.
	 *
	 * see apic/apicreg.h for icr bit definitions.
	 *
	 * TIME CRITICAL CODE, DO NOT DO ANY KPRINTFS IN THE HOT PATH.
	 */

	/*
	 * Do an INIT IPI: assert RESET
	 *
	 * Use edge triggered mode to assert INIT
	 */
	lapic_seticr_sync(physical_cpu,
	    APIC_DESTMODE_PHY |
	    APIC_DEST_DESTFLD |
	    APIC_TRIGMOD_EDGE |
	    APIC_LEVEL_ASSERT |
	    APIC_DELMODE_INIT);

	/*
	 * The spec calls for a 10ms delay but we may have to use a
	 * MUCH lower delay to avoid bricking an AP due to a fast SMI
	 * interrupt.  We have other loops here too and dividing by 2
	 * doesn't seem to be enough even after subtracting 350us,
	 * so we divide by 4.
	 *
	 * Our minimum delay is 150uS, maximum is 10ms.  If no SMI
	 * interrupt was detected we use the full 10ms.
	 */
	if (smibest == 0)
		u_sleep(10000);
	else if (smibest < 150 * 4 + 350)
		u_sleep(150);
	else if ((smibest - 350) / 4 < 10000)
		u_sleep((smibest - 350) / 4);
	else
		u_sleep(10000);

	/*
	 * Do an INIT IPI: deassert RESET
	 *
	 * Use level triggered mode to deassert.  It is unclear
	 * why we need to do this.
	 */
	lapic_seticr_sync(physical_cpu,
	    APIC_DESTMODE_PHY |
	    APIC_DEST_DESTFLD |
	    APIC_TRIGMOD_LEVEL |
	    APIC_LEVEL_DEASSERT |
	    APIC_DELMODE_INIT);
	u_sleep(150);				/* wait 150us */

	/*
	 * Next we do a STARTUP IPI: the previous INIT IPI might still be
	 * latched, (P5 bug) this 1st STARTUP would then terminate
	 * immediately, and the previously started INIT IPI would continue. OR
	 * the previous INIT IPI has already run. and this STARTUP IPI will
	 * run. OR the previous INIT IPI was ignored. and this STARTUP IPI
	 * will run.
	 *
	 * XXX set APIC_LEVEL_ASSERT
	 */
	lapic_seticr_sync(physical_cpu,
	    APIC_DESTMODE_PHY |
	    APIC_DEST_DESTFLD |
	    APIC_DELMODE_STARTUP |
	    vector);
	u_sleep(200);		/* wait ~200uS */

	/*
	 * Finally we do a 2nd STARTUP IPI: this 2nd STARTUP IPI should run IF
	 * the previous STARTUP IPI was cancelled by a latched INIT IPI. OR
	 * this STARTUP IPI will be ignored, as only ONE STARTUP IPI is
	 * recognized after hardware RESET or INIT IPI.
	 *
	 * XXX set APIC_LEVEL_ASSERT
	 */
	lapic_seticr_sync(physical_cpu,
	    APIC_DESTMODE_PHY |
	    APIC_DEST_DESTFLD |
	    APIC_DELMODE_STARTUP |
	    vector);

	/* Resume normal operation */
	cpu_enable_intr();

	/* wait for it to start, see ap_init() */
	set_apic_timer(5000000);/* == 5 seconds */
	while (read_apic_timer()) {
		if (CPUMASK_TESTBIT(smp_startup_mask, gd->mi.gd_cpuid))
			return 1;	/* return SUCCESS */
	}

	return 0;		/* return FAILURE */
}

static
int
smitest(void)
{
	int64_t	ltsc;
	int64_t	ntsc;
	int64_t	ldelta;
	int64_t	ndelta;
	int count;

	ldelta = 0;
	ndelta = 0;
	while (read_apic_timer()) {
		ltsc = rdtsc();
		for (count = 0; count < 100; ++count)
			ntsc = rdtsc();	/* force loop to occur */
		if (ldelta) {
			ndelta = ntsc - ltsc;
			if (ldelta > ndelta)
				ldelta = ndelta;
			if (ndelta > ldelta * 2)
				break;
		} else {
			ldelta = ntsc - ltsc;
		}
	}
	return(read_apic_timer());
}

/*
 * Synchronously flush the TLB on all other CPU's.  The current cpu's
 * TLB is not flushed.  If the caller wishes to flush the current cpu's
 * TLB the caller must call cpu_invltlb() in addition to smp_invltlb().
 *
 * This routine may be called concurrently from multiple cpus.  When this
 * happens, smp_invltlb() can wind up sticking around in the confirmation
 * while() loop at the end as additional cpus are added to the global
 * cpumask, until they are acknowledged by another IPI.
 *
 * NOTE: If for some reason we were unable to start all cpus we cannot
 *	 safely use broadcast IPIs.
 */

cpumask_t smp_smurf_mask;
static cpumask_t smp_invltlb_mask;
#define LOOPRECOVER
#define LOOPMASK_IN
#ifdef LOOPMASK_IN
cpumask_t smp_in_mask;
#endif
cpumask_t smp_invmask;
extern cpumask_t smp_idleinvl_mask;
extern cpumask_t smp_idleinvl_reqs;

/*
 * Atomically OR bits in *mask to smp_smurf_mask.  Adjust *mask to remove
 * bits that do not need to be IPId.  These bits are still part of the command,
 * but the target cpus have already been signalled and do not need to be
 * sigalled again.
 */
#include <sys/spinlock.h>
#include <sys/spinlock2.h>

static __noinline
void
smp_smurf_fetchset(cpumask_t *mask)
{
	cpumask_t omask;
	int i;
	__uint64_t obits;
	__uint64_t nbits;

	i = 0;
	while (i < CPUMASK_ELEMENTS) {
		obits = smp_smurf_mask.ary[i];
		cpu_ccfence();
		nbits = obits | mask->ary[i];
		if (atomic_cmpset_long(&smp_smurf_mask.ary[i], obits, nbits)) {
			omask.ary[i] = obits;
			++i;
		}
	}
	CPUMASK_NANDMASK(*mask, omask);
}

/*
 * This is a mechanism which guarantees that cpu_invltlb() will be executed
 * on idle cpus without having to signal or wake them up.  The invltlb will be
 * executed when they wake up, prior to any scheduling or interrupt thread.
 *
 * (*mask) is modified to remove the cpus we successfully negotiate this
 * function with.  This function may only be used with semi-synchronous
 * commands (typically invltlb's or semi-synchronous invalidations which
 * are usually associated only with kernel memory).
 */
void
smp_smurf_idleinvlclr(cpumask_t *mask)
{
	if (optimized_invltlb) {
		ATOMIC_CPUMASK_ORMASK(smp_idleinvl_reqs, *mask);
		/* cpu_lfence() not needed */
		CPUMASK_NANDMASK(*mask, smp_idleinvl_mask);
	}
}

/*
 * Issue cpu_invltlb() across all cpus except the current cpu.
 *
 * This function will arrange to avoid idle cpus, but still gurantee that
 * invltlb is run on them when they wake up prior to any scheduling or
 * nominal interrupt.
 */
void
smp_invltlb(void)
{
	struct mdglobaldata *md = mdcpu;
	cpumask_t mask;
	unsigned long rflags;
#ifdef LOOPRECOVER
	tsc_uclock_t tsc_base = rdtsc();
	int repeats = 0;
#endif

	if (report_invltlb_src > 0) {
		if (--report_invltlb_src <= 0)
			print_backtrace(8);
	}

	/*
	 * Disallow normal interrupts, set all active cpus except our own
	 * in the global smp_invltlb_mask.
	 */
	++md->mi.gd_cnt.v_smpinvltlb;
	crit_enter_gd(&md->mi);

	/*
	 * Bits we want to set in smp_invltlb_mask.  We do not want to signal
	 * our own cpu.  Also try to remove bits associated with idle cpus
	 * that we can flag for auto-invltlb.
	 */
	mask = smp_active_mask;
	CPUMASK_NANDBIT(mask, md->mi.gd_cpuid);
	smp_smurf_idleinvlclr(&mask);

	rflags = read_rflags();
	cpu_disable_intr();
	ATOMIC_CPUMASK_ORMASK(smp_invltlb_mask, mask);

	/*
	 * IPI non-idle cpus represented by mask.  The omask calculation
	 * removes cpus from the mask which already have a Xinvltlb IPI
	 * pending (avoid double-queueing the IPI).
	 *
	 * We must disable real interrupts when setting the smurf flags or
	 * we might race a XINVLTLB before we manage to send the ipi's for
	 * the bits we set.
	 *
	 * NOTE: We are not signalling ourselves, mask already does NOT
	 * include our own cpu.
	 */
	smp_smurf_fetchset(&mask);

	/*
	 * Issue the IPI.  Note that the XINVLTLB IPI runs regardless of
	 * the critical section count on the target cpus.
	 */
	CPUMASK_ORMASK(mask, md->mi.gd_cpumask);
	if (all_but_self_ipi_enable &&
	    (all_but_self_ipi_enable >= 2 ||
	     CPUMASK_CMPMASKEQ(smp_startup_mask, mask))) {
		all_but_self_ipi(XINVLTLB_OFFSET);
	} else {
		CPUMASK_NANDMASK(mask, md->mi.gd_cpumask);
		selected_apic_ipi(mask, XINVLTLB_OFFSET, APIC_DELMODE_FIXED);
	}

	/*
	 * Wait for acknowledgement by all cpus.  smp_inval_intr() will
	 * temporarily enable interrupts to avoid deadlocking the lapic,
	 * and will also handle running cpu_invltlb() and remote invlpg
	 * command son our cpu if some other cpu requests it of us.
	 *
	 * WARNING! I originally tried to implement this as a hard loop
	 *	    checking only smp_invltlb_mask (and issuing a local
	 *	    cpu_invltlb() if requested), with interrupts enabled
	 *	    and without calling smp_inval_intr().  This DID NOT WORK.
	 *	    It resulted in weird races where smurf bits would get
	 *	    cleared without any action being taken.
	 */
	smp_inval_intr();
	CPUMASK_ASSZERO(mask);
	while (CPUMASK_CMPMASKNEQ(smp_invltlb_mask, mask)) {
		smp_inval_intr();
		cpu_pause();
#ifdef LOOPRECOVER
		if (tsc_frequency && rdtsc() - tsc_base > tsc_frequency) {
			/*
			 * cpuid 	- cpu doing the waiting
			 * invltlb_mask - IPI in progress
			 */
			kprintf("smp_invltlb %2d: WARNING blocked %d sec: "
				"inv=%08jx "
				"smurf=%08jx "
#ifdef LOOPMASK_IN
				"in=%08jx "
#endif
				"idle=%08jx/%08jx\n",
				md->mi.gd_cpuid,
				repeats + 1,
				smp_invltlb_mask.ary[0],
				smp_smurf_mask.ary[0],
#ifdef LOOPMASK_IN
				smp_in_mask.ary[0],
#endif
				smp_idleinvl_mask.ary[0],
				smp_idleinvl_reqs.ary[0]);
			mdcpu->gd_xinvaltlb = 0;
			ATOMIC_CPUMASK_NANDMASK(smp_smurf_mask,
						smp_invltlb_mask);
			smp_invlpg(&smp_active_mask);

			/*
			 * Reload tsc_base for retry, give up after
			 * 10 seconds (60 seconds if in VM).
			 */
			tsc_base = rdtsc();
			if (++repeats > invlpg_timeout) {
				kprintf("smp_invltlb: giving up\n");
				CPUMASK_ASSZERO(smp_invltlb_mask);
			}
		}
#endif
	}
	write_rflags(rflags);
	crit_exit_gd(&md->mi);
}

/*
 * Called from a critical section with interrupts hard-disabled.
 * This function issues an XINVLTLB IPI and then executes any pending
 * command on the current cpu before returning.
 */
void
smp_invlpg(cpumask_t *cmdmask)
{
	struct mdglobaldata *md = mdcpu;
	cpumask_t mask;

	if (report_invlpg_src > 0) {
		if (--report_invlpg_src <= 0)
			print_backtrace(8);
	}

	/*
	 * Disallow normal interrupts, set all active cpus in the pmap,
	 * plus our own for completion processing (it might or might not
	 * be part of the set).
	 */
	mask = smp_active_mask;
	CPUMASK_ANDMASK(mask, *cmdmask);
	CPUMASK_ORMASK(mask, md->mi.gd_cpumask);

	/*
	 * Avoid double-queuing IPIs, which can deadlock us.  We must disable
	 * real interrupts when setting the smurf flags or we might race a
	 * XINVLTLB before we manage to send the ipi's for the bits we set.
	 *
	 * NOTE: We might be including our own cpu in the smurf mask.
	 */
	smp_smurf_fetchset(&mask);

	/*
	 * Issue the IPI.  Note that the XINVLTLB IPI runs regardless of
	 * the critical section count on the target cpus.
	 *
	 * We do not include our own cpu when issuing the IPI.
	 */
	if (all_but_self_ipi_enable &&
	    (all_but_self_ipi_enable >= 2 ||
	     CPUMASK_CMPMASKEQ(smp_startup_mask, mask))) {
		all_but_self_ipi(XINVLTLB_OFFSET);
	} else {
		CPUMASK_NANDMASK(mask, md->mi.gd_cpumask);
		selected_apic_ipi(mask, XINVLTLB_OFFSET, APIC_DELMODE_FIXED);
	}

	/*
	 * This will synchronously wait for our command to complete,
	 * as well as process commands from other cpus.  It also handles
	 * reentrancy.
	 *
	 * (interrupts are disabled and we are in a critical section here)
	 */
	smp_inval_intr();
}

/*
 * Issue rip/rsp sniffs
 */
void
smp_sniff(void)
{
	globaldata_t gd = mycpu;
	int dummy;
	register_t rflags;

	/*
	 * Ignore all_but_self_ipi_enable here and just use it.
	 */
	rflags = read_rflags();
	cpu_disable_intr();
	all_but_self_ipi(XSNIFF_OFFSET);
	gd->gd_sample_pc = smp_sniff;
	gd->gd_sample_sp = &dummy;
	write_rflags(rflags);
}

void
cpu_sniff(int dcpu)
{
	globaldata_t rgd = globaldata_find(dcpu);
	register_t rflags;
	int dummy;

	/*
	 * Ignore all_but_self_ipi_enable here and just use it.
	 */
	rflags = read_rflags();
	cpu_disable_intr();
	single_apic_ipi(dcpu, XSNIFF_OFFSET, APIC_DELMODE_FIXED);
	rgd->gd_sample_pc = cpu_sniff;
	rgd->gd_sample_sp = &dummy;
	write_rflags(rflags);
}

/*
 * Called from Xinvltlb assembly with interrupts hard-disabled and in a
 * critical section.  gd_intr_nesting_level may or may not be bumped
 * depending on entry.
 *
 * THIS CODE IS INTENDED TO EXPLICITLY IGNORE THE CRITICAL SECTION COUNT.
 * THAT IS, THE INTERRUPT IS INTENDED TO FUNCTION EVEN WHEN MAINLINE CODE
 * IS IN A CRITICAL SECTION.
 */
void
smp_inval_intr(void)
{
	struct mdglobaldata *md = mdcpu;
	cpumask_t cpumask;
#ifdef LOOPRECOVER
	tsc_uclock_t tsc_base = rdtsc();
#endif

#if 0
	/*
	 * The idle code is in a critical section, but that doesn't stop
	 * Xinvltlb from executing, so deal with the race which can occur
	 * in that situation.  Otherwise r-m-w operations by pmap_inval_intr()
	 * may have problems.
	 */
	if (ATOMIC_CPUMASK_TESTANDCLR(smp_idleinvl_reqs, md->mi.gd_cpuid)) {
		ATOMIC_CPUMASK_NANDBIT(smp_invltlb_mask, md->mi.gd_cpuid);
		cpu_invltlb();
		cpu_mfence();
	}
#endif

	/*
	 * This is a real mess.  I'd like to just leave interrupts disabled
	 * but it can cause the lapic to deadlock if too many interrupts queue
	 * to it, due to the idiotic design of the lapic.  So instead we have
	 * to enter a critical section so normal interrupts are made pending
	 * and track whether this one was reentered.
	 */
	if (md->gd_xinvaltlb) {		/* reentrant on cpu */
		md->gd_xinvaltlb = 2;
		return;
	}
	md->gd_xinvaltlb = 1;

	/*
	 * Check only those cpus with active Xinvl* commands pending.
	 *
	 * We are going to enable interrupts so make sure we are in a
	 * critical section.  This is necessary to avoid deadlocking
	 * the lapic and to ensure that we execute our commands prior to
	 * any nominal interrupt or preemption.
	 *
	 * WARNING! It is very important that we only clear out but in
	 *	    smp_smurf_mask once for each interrupt we take.  In
	 *	    this case, we clear it on initial entry and only loop
	 *	    on the reentrancy detect (caused by another interrupt).
	 */
	cpumask = smp_invmask;
#ifdef LOOPMASK_IN
	ATOMIC_CPUMASK_ORBIT(smp_in_mask, md->mi.gd_cpuid);
#endif
loop:
	cpu_enable_intr();
	ATOMIC_CPUMASK_NANDBIT(smp_smurf_mask, md->mi.gd_cpuid);

	/*
	 * Specific page request(s), and we can't return until all bits
	 * are zero.
	 */
	for (;;) {
		int toolong;

		/*
		 * Also execute any pending full invalidation request in
		 * this loop.
		 */
		if (CPUMASK_TESTBIT(smp_invltlb_mask, md->mi.gd_cpuid)) {
			ATOMIC_CPUMASK_NANDBIT(smp_invltlb_mask,
					       md->mi.gd_cpuid);
			cpu_invltlb();
			cpu_mfence();
		}

#ifdef LOOPRECOVER
		if (tsc_frequency && rdtsc() - tsc_base > tsc_frequency) {
			/*
			 * cpuid 	- cpu doing the waiting
			 * invmask	- IPI in progress
			 * invltlb_mask - which ones are TLB invalidations?
			 */
			kprintf("smp_inval_intr %2d, WARNING blocked >1 sec "
				"inv=%08jx tlbm=%08jx "
				"smurf=%08jx "
#ifdef LOOPMASK_IN
				"in=%08jx "
#endif
				"idle=%08jx/%08jx\n",
				md->mi.gd_cpuid,
				smp_invmask.ary[0],
				smp_invltlb_mask.ary[0],
				smp_smurf_mask.ary[0],
#ifdef LOOPMASK_IN
				smp_in_mask.ary[0],
#endif
				smp_idleinvl_mask.ary[0],
				smp_idleinvl_reqs.ary[0]);
			tsc_base = rdtsc();
			toolong = 1;
		} else {
			toolong = 0;
		}
#else
		toolong = 0;
#endif

		/*
		 * We can only add bits to the cpumask to test during the
		 * loop because the smp_invmask bit is cleared once the
		 * originator completes the command (the targets may still
		 * be cycling their own completions in this loop, afterwords).
		 *
		 * lfence required prior to all tests as this Xinvltlb
		 * interrupt could race the originator (already be in progress
		 * wnen the originator decides to issue, due to an issue by
		 * another cpu).
		 */
		cpu_lfence();
		CPUMASK_ORMASK(cpumask, smp_invmask);
		/*cpumask = smp_active_mask;*/	/* XXX */
		cpu_lfence();

		if (pmap_inval_intr(&cpumask, toolong) == 0) {
			/*
			 * Clear our smurf mask to allow new IPIs, but deal
			 * with potential races.
			 */
			break;
		}

		/*
		 * Test if someone sent us another invalidation IPI, break
		 * out so we can take it to avoid deadlocking the lapic
		 * interrupt queue (? stupid intel, amd).
		 */
		if (md->gd_xinvaltlb == 2)
			break;
		/*
		if (CPUMASK_TESTBIT(smp_smurf_mask, md->mi.gd_cpuid))
			break;
		*/
	}

	/*
	 * Full invalidation request
	 */
	if (CPUMASK_TESTBIT(smp_invltlb_mask, md->mi.gd_cpuid)) {
		ATOMIC_CPUMASK_NANDBIT(smp_invltlb_mask,
				       md->mi.gd_cpuid);
		cpu_invltlb();
		cpu_mfence();
	}

	/*
	 * Check to see if another Xinvltlb interrupt occurred and loop up
	 * if it did.
	 */
	cpu_disable_intr();
	if (md->gd_xinvaltlb == 2) {
		md->gd_xinvaltlb = 1;
		goto loop;
	}
#ifdef LOOPMASK_IN
	ATOMIC_CPUMASK_NANDBIT(smp_in_mask, md->mi.gd_cpuid);
#endif
	md->gd_xinvaltlb = 0;
}

void
cpu_wbinvd_on_all_cpus_callback(void *arg)
{
	wbinvd();
}

/*
 * When called the executing CPU will send an IPI to all other CPUs
 * requesting that they halt execution.
 *
 * Usually (but not necessarily) called with 'other_cpus' as its arg.
 *
 *  - Signals all CPUs in map to stop.
 *  - Waits for each to stop.
 *
 * Returns:
 *  -1: error
 *   0: NA
 *   1: ok
 *
 * XXX FIXME: this is not MP-safe, needs a lock to prevent multiple CPUs
 *            from executing at same time.
 */
int
stop_cpus(cpumask_t map)
{
	cpumask_t mask;

	CPUMASK_ANDMASK(map, smp_active_mask);

	/* send the Xcpustop IPI to all CPUs in map */
	selected_apic_ipi(map, XCPUSTOP_OFFSET, APIC_DELMODE_FIXED);

	do {
		mask = stopped_cpus;
		CPUMASK_ANDMASK(mask, map);
		/* spin */
	} while (CPUMASK_CMPMASKNEQ(mask, map));

	return 1;
}


/*
 * Called by a CPU to restart stopped CPUs. 
 *
 * Usually (but not necessarily) called with 'stopped_cpus' as its arg.
 *
 *  - Signals all CPUs in map to restart.
 *  - Waits for each to restart.
 *
 * Returns:
 *  -1: error
 *   0: NA
 *   1: ok
 */
int
restart_cpus(cpumask_t map)
{
	cpumask_t mask;

	/* signal other cpus to restart */
	mask = map;
	CPUMASK_ANDMASK(mask, smp_active_mask);
	cpu_ccfence();
	started_cpus = mask;
	cpu_ccfence();

	/* wait for each to clear its bit */
	while (CPUMASK_CMPMASKNEQ(stopped_cpus, map))
		cpu_pause();

	return 1;
}

/*
 * This is called once the mpboot code has gotten us properly relocated
 * and the MMU turned on, etc.   ap_init() is actually the idle thread,
 * and when it returns the scheduler will call the real cpu_idle() main
 * loop for the idlethread.  Interrupts are disabled on entry and should
 * remain disabled at return.
 */
void
ap_init(void)
{
	int	cpu_id;

	/*
	 * Adjust smp_startup_mask to signal the BSP that we have started
	 * up successfully.  Note that we do not yet hold the BGL.  The BSP
	 * is waiting for our signal.
	 *
	 * We can't set our bit in smp_active_mask yet because we are holding
	 * interrupts physically disabled and remote cpus could deadlock
	 * trying to send us an IPI.
	 */
	ATOMIC_CPUMASK_ORBIT(smp_startup_mask, mycpu->gd_cpuid);
	cpu_mfence();

	/*
	 * Interlock for LAPIC initialization.  Wait until mp_finish_lapic is
	 * non-zero, then get the MP lock.
	 *
	 * Note: We are in a critical section.
	 *
	 * Note: we are the idle thread, we can only spin.
	 *
	 * Note: The load fence is memory volatile and prevents the compiler
	 * from improperly caching mp_finish_lapic, and the cpu from improperly
	 * caching it.
	 */
	while (mp_finish_lapic == 0) {
		cpu_pause();
		cpu_lfence();
	}
#if 0
	while (try_mplock() == 0) {
		cpu_pause();
		cpu_lfence();
	}
#endif

	if (cpu_feature & CPUID_TSC) {
		/*
		 * The BSP is constantly updating tsc0_offset, figure out
		 * the relative difference to synchronize ktrdump.
		 */
		tsc_offsets[mycpu->gd_cpuid] = rdtsc() - tsc0_offset;
	}

	/* BSP may have changed PTD while we're waiting for the lock */
	cpu_invltlb();

	/* Build our map of 'other' CPUs. */
	mycpu->gd_other_cpus = smp_startup_mask;
	ATOMIC_CPUMASK_NANDBIT(mycpu->gd_other_cpus, mycpu->gd_cpuid);

	/* A quick check from sanity claus */
	cpu_id = APICID_TO_CPUID(LAPIC_READID);
	if (mycpu->gd_cpuid != cpu_id) {
		kprintf("SMP: assigned cpuid = %d\n", mycpu->gd_cpuid);
		kprintf("SMP: actual cpuid = %d lapicid %d\n",
			cpu_id, LAPIC_READID);
#if 0 /* JGXXX */
		kprintf("PTD[MPPTDI] = %p\n", (void *)PTD[MPPTDI]);
#endif
		panic("cpuid mismatch! boom!!");
	}

	/* Initialize AP's local APIC for irq's */
	lapic_init(FALSE);

	/* LAPIC initialization is done */
	ATOMIC_CPUMASK_ORBIT(smp_lapic_mask, mycpu->gd_cpuid);
	cpu_mfence();

#if 0
	/* Let BSP move onto the next initialization stage */
	rel_mplock();
#endif

	/*
	 * Interlock for finalization.  Wait until mp_finish is non-zero,
	 * then get the MP lock.
	 *
	 * Note: We are in a critical section.
	 *
	 * Note: we are the idle thread, we can only spin.
	 *
	 * Note: The load fence is memory volatile and prevents the compiler
	 * from improperly caching mp_finish, and the cpu from improperly
	 * caching it.
	 */
	while (mp_finish == 0) {
		cpu_pause();
		cpu_lfence();
	}

	/* BSP may have changed PTD while we're waiting for the lock */
	cpu_invltlb();

	/* Set memory range attributes for this CPU to match the BSP */
	mem_range_AP_init();

	/*
	 * Once we go active we must process any IPIQ messages that may
	 * have been queued, because no actual IPI will occur until we
	 * set our bit in the smp_active_mask.  If we don't the IPI
	 * message interlock could be left set which would also prevent
	 * further IPIs.
	 *
	 * The idle loop doesn't expect the BGL to be held and while
	 * lwkt_switch() normally cleans things up this is a special case
	 * because we returning almost directly into the idle loop.
	 *
	 * The idle thread is never placed on the runq, make sure
	 * nothing we've done put it there.
	 */

	/*
	 * Hold a critical section and allow real interrupts to occur.  Zero
	 * any spurious interrupts which have accumulated, then set our
	 * smp_active_mask indicating that we are fully operational.
	 */
	crit_enter();
	__asm __volatile("sti; pause; pause"::);
	bzero(mdcpu->gd_ipending, sizeof(mdcpu->gd_ipending));
	ATOMIC_CPUMASK_ORBIT(smp_active_mask, mycpu->gd_cpuid);

	/*
	 * Wait until all cpus have set their smp_active_mask and have fully
	 * operational interrupts before proceeding.
	 *
	 * We need a final cpu_invltlb() because we would not have received
	 * any until we set our bit in smp_active_mask.
	 */
	while (mp_finish == 1) {
		cpu_pause();
		cpu_lfence();
	}
	cpu_invltlb();

	/*
	 * Initialize per-cpu clocks and do other per-cpu initialization.
	 * At this point code is expected to be able to use the full kernel
	 * API.
	 */
	initclocks_pcpu();	/* clock interrupts (via IPIs) */

	/*
	 * Since we may have cleaned up the interrupt triggers, manually
	 * process any pending IPIs before exiting our critical section.
	 * Once the critical section has exited, normal interrupt processing
	 * may occur.
	 */
	atomic_swap_int(&mycpu->gd_npoll, 0);
	lwkt_process_ipiq();
	crit_exit();

	/*
	 * Final final, allow the waiting BSP to resume the boot process,
	 * return 'into' the idle thread bootstrap.
	 */
	ATOMIC_CPUMASK_ORBIT(smp_finalize_mask, mycpu->gd_cpuid);
	KKASSERT((curthread->td_flags & TDF_RUNQ) == 0);
}

/*
 * Get SMP fully working before we start initializing devices.
 */
static
void
ap_finish(void)
{
	if (bootverbose)
		kprintf("Finish MP startup\n");
	rel_mplock();

	/*
	 * Wait for the active mask to complete, after which all cpus will
	 * be accepting interrupts.
	 */
	mp_finish = 1;
	while (CPUMASK_CMPMASKNEQ(smp_active_mask, smp_startup_mask)) {
		cpu_pause();
		cpu_lfence();
	}

	/*
	 * Wait for the finalization mask to complete, after which all cpus
	 * have completely finished initializing and are entering or are in
	 * their idle thread.
	 *
	 * BSP should have received all required invltlbs but do another
	 * one just in case.
	 */
	cpu_invltlb();
	mp_finish = 2;
	while (CPUMASK_CMPMASKNEQ(smp_finalize_mask, smp_startup_mask)) {
		cpu_pause();
		cpu_lfence();
	}

	while (try_mplock() == 0) {
		cpu_pause();
		cpu_lfence();
	}

	if (bootverbose) {
		kprintf("Active CPU Mask: %016jx\n",
			(uintmax_t)CPUMASK_LOWMASK(smp_active_mask));
	}
}

SYSINIT(finishsmp, SI_BOOT2_FINISH_SMP, SI_ORDER_FIRST, ap_finish, NULL);

/*
 * Interrupts must be hard-disabled by caller
 */
void
cpu_send_ipiq(int dcpu)
{
	if (CPUMASK_TESTBIT(smp_active_mask, dcpu))
                single_apic_ipi(dcpu, XIPIQ_OFFSET, APIC_DELMODE_FIXED);
}

#if 0	/* single_apic_ipi_passive() not working yet */
/*
 * Returns 0 on failure, 1 on success
 */
int
cpu_send_ipiq_passive(int dcpu)
{
        int r = 0;
	if (CPUMASK_TESTBIT(smp_active_mask, dcpu)) {
                r = single_apic_ipi_passive(dcpu, XIPIQ_OFFSET,
                                        APIC_DELMODE_FIXED);
        }
	return(r);
}
#endif

static void
mp_bsp_simple_setup(void)
{
	struct mdglobaldata *gd;
	size_t ipiq_size;

	/* build our map of 'other' CPUs */
	mycpu->gd_other_cpus = smp_startup_mask;
	CPUMASK_NANDBIT(mycpu->gd_other_cpus, mycpu->gd_cpuid);

	gd = (struct mdglobaldata *)mycpu;
	gd->gd_acpi_id = CPUID_TO_ACPIID(mycpu->gd_cpuid);

	ipiq_size = sizeof(struct lwkt_ipiq) * ncpus;
	mycpu->gd_ipiq = (void *)kmem_alloc(kernel_map, ipiq_size,
					    VM_SUBSYS_IPIQ);
	bzero(mycpu->gd_ipiq, ipiq_size);

	/* initialize arc4random. */
	arc4_init_pcpu(0);

	pmap_set_opt();

	if (cpu_feature & CPUID_TSC)
		tsc0_offset = rdtsc();
}


/*
 * CPU TOPOLOGY DETECTION FUNCTIONS
 */

/* Detect intel topology using CPUID 
 * Ref: http://www.intel.com/Assets/PDF/appnote/241618.pdf, pg 41
 */
static void
detect_intel_topology(int count_htt_cores)
{
	int shift = 0;
	int ecx_index = 0;
	int core_plus_logical_bits = 0;
	int cores_per_package;
	int logical_per_package;
	int logical_per_core;
	unsigned int p[4];

	if (cpu_high >= 0xb) {
		goto FUNC_B;

	} else if (cpu_high >= 0x4) {
		goto FUNC_4;

	} else {
		core_bits = 0;
		for (shift = 0; (1 << shift) < count_htt_cores; ++shift)
			;
		logical_CPU_bits = 1 << shift;
		return;
	}

FUNC_B:
	cpuid_count(0xb, FUNC_B_THREAD_LEVEL, p);

	/* if 0xb not supported - fallback to 0x4 */
	if (p[1] == 0 || (FUNC_B_TYPE(p[2]) != FUNC_B_THREAD_TYPE)) {
		goto FUNC_4;
	}

	logical_CPU_bits = FUNC_B_BITS_SHIFT_NEXT_LEVEL(p[0]);

	ecx_index = FUNC_B_THREAD_LEVEL + 1;
	do {
		cpuid_count(0xb, ecx_index, p);

		/* Check for the Core type in the implemented sub leaves. */
		if (FUNC_B_TYPE(p[2]) == FUNC_B_CORE_TYPE) {
			core_plus_logical_bits = FUNC_B_BITS_SHIFT_NEXT_LEVEL(p[0]);
			break;
		}

		ecx_index++;

	} while (FUNC_B_TYPE(p[2]) != FUNC_B_INVALID_TYPE);

	core_bits = core_plus_logical_bits - logical_CPU_bits;

	return;

FUNC_4:
	cpuid_count(0x4, 0, p);
	cores_per_package = FUNC_4_MAX_CORE_NO(p[0]) + 1;

	logical_per_package = count_htt_cores;
	logical_per_core = logical_per_package / cores_per_package;
	
	for (shift = 0; (1 << shift) < logical_per_core; ++shift)
		;
	logical_CPU_bits = shift;

	for (shift = 0; (1 << shift) < cores_per_package; ++shift)
		;
	core_bits = shift;

	return;
}

/* Detect AMD topology using CPUID
 * Ref: http://support.amd.com/us/Embedded_TechDocs/25481.pdf, last page
 */
static void
detect_amd_topology(int count_htt_cores)
{
	int shift = 0;
	if ((cpu_feature & CPUID_HTT) && (amd_feature2 & AMDID2_CMP)) {
		if (cpu_procinfo2 & AMDID_COREID_SIZE) {
			core_bits = (cpu_procinfo2 & AMDID_COREID_SIZE) >>
				    AMDID_COREID_SIZE_SHIFT;
		} else {
			core_bits = (cpu_procinfo2 & AMDID_CMP_CORES) + 1;
			for (shift = 0; (1 << shift) < core_bits; ++shift)
				;
			core_bits = shift;
		}
		logical_CPU_bits = count_htt_cores >> core_bits;
		for (shift = 0; (1 << shift) < logical_CPU_bits; ++shift)
			;
		logical_CPU_bits = shift;

		kprintf("core_bits %d logical_CPU_bits %d\n",
			core_bits - logical_CPU_bits, logical_CPU_bits);

		if (amd_feature2 & AMDID2_TOPOEXT) {
			u_int p[4];	/* eax,ebx,ecx,edx */
			int nodes;

			cpuid_count(0x8000001e, 0, p);

			switch(((p[1] >> 8) & 3) + 1) {
			case 1:
				logical_CPU_bits = 0;
				break;
			case 2:
				logical_CPU_bits = 1;
				break;
			case 3:
			case 4:
				logical_CPU_bits = 2;
				break;
			}

			/*
			 * Nodes are kind of a stand-in for packages*sockets,
			 * but can be thought of in terms of Numa domains.
			 */
			nodes = ((p[2] >> 8) & 7) + 1;
			switch(nodes) {
			case 8:
			case 7:
			case 6:
			case 5:
				--core_bits;
				/* fallthrough */
			case 4:
			case 3:
				--core_bits;
				/* fallthrough */
			case 2:
				--core_bits;
				/* fallthrough */
			case 1:
				break;
			}
			core_bits -= logical_CPU_bits;
			kprintf("%d-way htt, %d Nodes, %d cores/node\n",
				(int)(((p[1] >> 8) & 3) + 1),
				nodes,
				1 << core_bits);

		}
#if 0
		if (amd_feature2 & AMDID2_TOPOEXT) {
			u_int p[4];
			int i;
			int type;
			int level;
			int share_count;

			logical_CPU_bits = 0;
			core_bits = 0;

			for (i = 0; i < 256; ++i)  {
				cpuid_count(0x8000001d, i, p);
				type = p[0] & 0x1f;
				level = (p[0] >> 5) & 0x7;
				share_count = 1 + ((p[0] >> 14) & 0xfff);

				if (type == 0)
					break;
				kprintf("Topology probe i=%2d type=%d "
					"level=%d share_count=%d\n",
					i, type, level, share_count);
				shift = 0;
				while ((1 << shift) < share_count)
					++shift;

				switch(type) {
				case 1:
					/*
					 * CPUID_TYPE_SMT
					 *
					 * Logical CPU (SMT)
					 */
					logical_CPU_bits = shift;
					break;
				case 2:
					/*
					 * CPUID_TYPE_CORE
					 *
					 * Physical subdivision of a package
					 */
					core_bits = logical_CPU_bits +
						    shift;
					break;
				case 3:
					/*
					 * CPUID_TYPE_CACHE
					 *
					 * CPU L1/L2/L3 cache
					 */
					break;
				case 4:
					/*
					 * CPUID_TYPE_PKG
					 *
					 * Package aka chip, equivalent to
					 * socket
					 */
					break;
				}
			}
		}
#endif
	} else {
		for (shift = 0; (1 << shift) < count_htt_cores; ++shift)
			;
		core_bits = shift;
		logical_CPU_bits = 0;
	}
}

static void
amd_get_compute_unit_id(void *arg)
{
	u_int regs[4];

	do_cpuid(0x8000001e, regs);
	cpu_node_t * mynode = get_cpu_node_by_cpuid(mycpuid);

	/* 
	 * AMD - CPUID Specification September 2010
	 * page 34 - //ComputeUnitID = ebx[0:7]//
	 */
	mynode->compute_unit_id = regs[1] & 0xff;
}

int
fix_amd_topology(void)
{
	cpumask_t mask;

	if (cpu_vendor_id != CPU_VENDOR_AMD)
		return -1;
	if ((amd_feature2 & AMDID2_TOPOEXT) == 0)
		return -1;

	CPUMASK_ASSALLONES(mask);
	lwkt_cpusync_simple(mask, amd_get_compute_unit_id, NULL);

	kprintf("Compute unit iDS:\n");
	int i;
	for (i = 0; i < ncpus; i++) {
		kprintf("%d-%d; \n",
			i, get_cpu_node_by_cpuid(i)->compute_unit_id);
	}
	return 0;
}

/*
 * Calculate
 * - logical_CPU_bits
 * - core_bits
 * With the values above (for AMD or INTEL) we are able to generally
 * detect the CPU topology (number of cores for each level):
 * Ref: http://wiki.osdev.org/Detecting_CPU_Topology_(80x86)
 * Ref: http://www.multicoreinfo.com/research/papers/whitepapers/Intel-detect-topology.pdf
 */
void
detect_cpu_topology(void)
{
	static int topology_detected = 0;
	int count = 0;
	
	if (topology_detected)
		goto OUT;
	if ((cpu_feature & CPUID_HTT) == 0) {
		core_bits = 0;
		logical_CPU_bits = 0;
		goto OUT;
	}
	count = (cpu_procinfo & CPUID_HTT_CORES) >> CPUID_HTT_CORE_SHIFT;

	if (cpu_vendor_id == CPU_VENDOR_INTEL)
		detect_intel_topology(count);	
	else if (cpu_vendor_id == CPU_VENDOR_AMD)
		detect_amd_topology(count);
	topology_detected = 1;

OUT:
	if (bootverbose) {
		kprintf("Bits within APICID: logical_CPU_bits: %d; "
			"core_bits: %d\n",
			logical_CPU_bits, core_bits);
	}
}

/*
 * Interface functions to calculate chip_ID,
 * core_number and logical_number
 * Ref: http://wiki.osdev.org/Detecting_CPU_Topology_(80x86)
 */
int
get_chip_ID(int cpuid)
{
	return get_apicid_from_cpuid(cpuid) >>
	    (logical_CPU_bits + core_bits);
}

int
get_chip_ID_from_APICID(int apicid)
{
	return apicid >> (logical_CPU_bits + core_bits);
}

int
get_core_number_within_chip(int cpuid)
{
	return ((get_apicid_from_cpuid(cpuid) >> logical_CPU_bits) &
		((1 << core_bits) - 1));
}

int
get_logical_CPU_number_within_core(int cpuid)
{
	return (get_apicid_from_cpuid(cpuid) &
		((1 << logical_CPU_bits) - 1));
}
