<commit_before>// Multiprocessor support
// Search memory for MP description structures.
// http://developer.intel.com/design/pentium/datashts/24201606.pdf

#include "types.h"
#include "defs.h"
#include "param.h"
#include "memlayout.h"
#include "mp.h"
#include "x86.h"
#include "mmu.h"
#include "proc.h"

struct cpu cpus[NCPU];
int ismp;
int ncpu;
uchar ioapicid;

static uchar
sum(uchar *addr, int len)
{
  int i, sum;

  sum = 0;
  for(i=0; i<len; i++)
    sum += addr[i];
  return sum;
}

// Look for an MP structure in the len bytes at addr.
static struct mp*
mpsearch1(uint a, int len)
{
  uchar *e, *p, *addr;

  addr = P2V(a);
  e = addr+len;
  for(p = addr; p < e; p += sizeof(struct mp))
    if(memcmp(p, "_MP_", 4) == 0 && sum(p, sizeof(struct mp)) == 0)
      return (struct mp*)p;
  return 0;
}

// Search for the MP Floating Pointer Structure, which according to the
// spec is in one of the following three locations:
// 1) in the first KB of the EBDA;
// 2) in the last KB of system base memory;
// 3) in the BIOS ROM between 0xF0000 and 0xFFFFF.
static struct mp*
mpsearch(void)
{
  uchar *bda;
  uint p;
  struct mp *mp;

  bda = (uchar *) P2V(0x400);
  if((p = ((bda[0x0F]<<8)| bda[0x0E]) << 4)){
    if((mp = mpsearch1(p, 1024)))
      return mp;
  } else {
    p = ((bda[0x14]<<8)|bda[0x13])*1024;
    if((mp = mpsearch1(p-1024, 1024)))
      return mp;
  }
  return mpsearch1(0xF0000, 0x10000);
}

// Search for an MP configuration table.  For now,
// don't accept the default configurations (physaddr == 0).
// Check for correct signature, calculate the checksum and,
// if correct, check the version.
// To do: check extended table checksum.
static struct mpconf*
mpconfig(struct mp **pmp)
{
  struct mpconf *conf;
  struct mp *mp;

  if((mp = mpsearch()) == 0 || mp->physaddr == 0)
    return 0;
  conf = (struct mpconf*) P2V((uint) mp->physaddr);
  if(memcmp(conf, "PCMP", 4) != 0)
    return 0;
  if(conf->specrev != 1 && conf->specrev != 4)
    return 0;
  if(sum((uchar*)conf, conf->length) != 0)
    return 0;
  *pmp = mp;
  return conf;
}

void
mpinit(void)
{
  uchar *p, *e;
  struct mp *mp;
  struct mpconf *conf;
  struct mpproc *proc;
  struct mpioapic *mioapic;

  if((conf = mpconfig(&mp)) == 0)
    return;
  ismp = 1;
  lapic = (uint*)conf->lapicaddr;
  for(p=(uchar*)(conf+1), e=(uchar*)conf+conf->length; p<e; ){
    switch(*p){
    case MPPROC:
      proc = (struct mpproc*)p;
      if(!(proc->flags & MP_PROC_ENABLED)){
        cprintf("mpinit: ignoring cpu %d, not enabled\n", proc->apicid);
        p += sizeof(struct mpproc);
        continue;
      }
      if(!(proc->feature & MP_PROC_APIC)){
        cprintf("mpinit: cpu %d has no working APIC, ignored\n", proc->apicid);
        p += sizeof(struct mpproc);
        continue;
      }
      if(ncpu >= NCPU){
        cprintf("mpinit: more than %d cpus, ignoring cpu%d\n",
          NCPU, proc->apicid);
        p += sizeof(struct mpproc);
        continue;
      }
      if(ncpu != proc->apicid){
        cprintf("mpinit: ncpu=%d apicid=%d\n", ncpu, proc->apicid);
      }
      cpus[ncpu].apicid = proc->apicid;  // apicid may differ from ncpu
      ncpu++;
      p += sizeof(struct mpproc);
      continue;
    case MPIOAPIC:
      mioapic = (struct mpioapic*)p;
      if(!(mioapic->flags & MP_APIC_ENABLED)){
        cprintf("mpinit: ioapic %d disabled, ignored\n", mioapic->apicid);
        p += sizeof(struct mpioapic);
        continue;
      }
      if(ioapic == 0){
        ioapic = (volatile struct ioapic*)mioapic->addr;
        ioapicid = mioapic->apicid;
      } else {
        cprintf("mpinit: ignored extra ioapic %d\n", mioapic->apicid);
      }
      p += sizeof(struct mpioapic);
      continue;
    case MPBUS:
    case MPIOINTR:
    case MPLINTR:
      p += 8;
      continue;
    default:
      ismp = 0;
      break;
    }
  }
  if(!ismp){
    // Didn't like what we found; fall back to no MP.
    ncpu = 1;
    lapic = 0;
    ioapic = 0;
    ioapicid = 0;
    return;
  }

  if(mp->imcrp){
    // Bochs doesn't support IMCR, so this doesn't run on Bochs.
    // But it would on real hardware.
    outb(0x22, 0x70);   // Select IMCR
    outb(0x23, inb(0x23) | 1);  // Mask external interrupts.
  }
}
<commit_msg>Comment about APIC id versus CPU id confusion.<commit_after>// Multiprocessor support
// Search memory for MP description structures.
// http://developer.intel.com/design/pentium/datashts/24201606.pdf

#include "types.h"
#include "defs.h"
#include "param.h"
#include "memlayout.h"
#include "mp.h"
#include "x86.h"
#include "mmu.h"
#include "proc.h"

struct cpu cpus[NCPU];
int ismp;
int ncpu;
uchar ioapicid;

static uchar
sum(uchar *addr, int len)
{
  int i, sum;

  sum = 0;
  for(i=0; i<len; i++)
    sum += addr[i];
  return sum;
}

// Look for an MP structure in the len bytes at addr.
static struct mp*
mpsearch1(uint a, int len)
{
  uchar *e, *p, *addr;

  addr = P2V(a);
  e = addr+len;
  for(p = addr; p < e; p += sizeof(struct mp))
    if(memcmp(p, "_MP_", 4) == 0 && sum(p, sizeof(struct mp)) == 0)
      return (struct mp*)p;
  return 0;
}

// Search for the MP Floating Pointer Structure, which according to the
// spec is in one of the following three locations:
// 1) in the first KB of the EBDA;
// 2) in the last KB of system base memory;
// 3) in the BIOS ROM between 0xF0000 and 0xFFFFF.
static struct mp*
mpsearch(void)
{
  uchar *bda;
  uint p;
  struct mp *mp;

  bda = (uchar *) P2V(0x400);
  if((p = ((bda[0x0F]<<8)| bda[0x0E]) << 4)){
    if((mp = mpsearch1(p, 1024)))
      return mp;
  } else {
    p = ((bda[0x14]<<8)|bda[0x13])*1024;
    if((mp = mpsearch1(p-1024, 1024)))
      return mp;
  }
  return mpsearch1(0xF0000, 0x10000);
}

// Search for an MP configuration table.  For now,
// don't accept the default configurations (physaddr == 0).
// Check for correct signature, calculate the checksum and,
// if correct, check the version.
// To do: check extended table checksum.
static struct mpconf*
mpconfig(struct mp **pmp)
{
  struct mpconf *conf;
  struct mp *mp;

  if((mp = mpsearch()) == 0 || mp->physaddr == 0)
    return 0;
  conf = (struct mpconf*) P2V((uint) mp->physaddr);
  if(memcmp(conf, "PCMP", 4) != 0)
    return 0;
  if(conf->specrev != 1 && conf->specrev != 4)
    return 0;
  if(sum((uchar*)conf, conf->length) != 0)
    return 0;
  *pmp = mp;
  return conf;
}

void
mpinit(void)
{
  uchar *p, *e;
  struct mp *mp;
  struct mpconf *conf;
  struct mpproc *proc;
  struct mpioapic *mioapic;

  if((conf = mpconfig(&mp)) == 0)
    return;
  ismp = 1;
  lapic = (uint*)conf->lapicaddr;
  for(p=(uchar*)(conf+1), e=(uchar*)conf+conf->length; p<e; ){
    switch(*p){
    case MPPROC:
      proc = (struct mpproc*)p;
      if(!(proc->flags & MP_PROC_ENABLED)){
        cprintf("mpinit: ignoring cpu %d, not enabled\n", proc->apicid);
        p += sizeof(struct mpproc);
        continue;
      }
      if(!(proc->feature & MP_PROC_APIC)){
        cprintf("mpinit: cpu %d has no working APIC, ignored\n", proc->apicid);
        p += sizeof(struct mpproc);
        continue;
      }
      if(ncpu >= NCPU){
        cprintf("mpinit: more than %d cpus, ignoring cpu%d\n",
          NCPU, proc->apicid);
        p += sizeof(struct mpproc);
        continue;
      }
      // TODO: The following is dubious at best. We enforce a rule that the
      // APIC ids have to be consecutive, but the MP specification explicitly
      // says that they DON'T have to be. As long as QEMU is used to simulate
      // multiple SOCKETS everything seems fine, but as soon as we add CORES
      // the code here breaks (and fails to identify them). The confusion
      // between the id we assign here and the actual APIC id continues in
      // ioapic.c for example. A rewrite is in order.
      if(ncpu != proc->apicid){
        cprintf("mpinit: ncpu=%d apicid=%d\n", ncpu, proc->apicid);
      }
      cpus[ncpu].apicid = proc->apicid;  // apicid may differ from ncpu
      ncpu++;
      p += sizeof(struct mpproc);
      continue;
    case MPIOAPIC:
      mioapic = (struct mpioapic*)p;
      if(!(mioapic->flags & MP_APIC_ENABLED)){
        cprintf("mpinit: ioapic %d disabled, ignored\n", mioapic->apicid);
        p += sizeof(struct mpioapic);
        continue;
      }
      if(ioapic == 0){
        ioapic = (volatile struct ioapic*)mioapic->addr;
        ioapicid = mioapic->apicid;
      } else {
        cprintf("mpinit: ignored extra ioapic %d\n", mioapic->apicid);
      }
      p += sizeof(struct mpioapic);
      continue;
    case MPBUS:
    case MPIOINTR:
    case MPLINTR:
      p += 8;
      continue;
    default:
      ismp = 0;
      break;
    }
  }
  if(!ismp){
    // Didn't like what we found; fall back to no MP.
    ncpu = 1;
    lapic = 0;
    ioapic = 0;
    ioapicid = 0;
    return;
  }

  if(mp->imcrp){
    // Bochs doesn't support IMCR, so this doesn't run on Bochs.
    // But it would on real hardware.
    outb(0x22, 0x70);   // Select IMCR
    outb(0x23, inb(0x23) | 1);  // Mask external interrupts.
  }
}

{'fraction_non_alphanumeric': 0.14913936701832314, 'fraction_numerical': 0.0246529705719045, 'mean_word_length': 2.6154154957848252, 'pattern_counts': {'":': 0, '<': 21, '<?xml version=': 0, '>': 47, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '2776790', 'n_tokens_mistral': 3536, 'n_tokens_neox': 3346, 'n_words': 990}