lib/libcaps/globaldata.c

   1 /*
   2  * GLOBALDATA.C
   3  *
   4  * Copyright (c) 2003 Matthew Dillon <dillon@backplane.com>
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26  * SUCH DAMAGE.
  27  *
  28  * $DragonFly: src/lib/libcaps/globaldata.c,v 1.3 2003/12/07 04:21:52 dillon Exp $
  29  */
  30
  31 #include "defs.h"
  32
  33 struct globaldata gdary[MAXVCPU];
  34 u_int mp_lock;
  35 int ncpus = 1;
  36 int smp_active = 1;
  37 u_int32_t stopped_cpus;
  38 char *panicstr;
  39
  40 /*
  41  * Master globaldata init
  42  */
  43 void
  44 globaldata_init(thread_t td)
  45 {
  46     mi_gdinit1(&gdary[0], 0);
  47     mi_gdinit2(&gdary[0]);
  48
  49     /*
  50      * If a 'main' thread is passed make it the current thread and mark it
  51      * as currently running, but not on the run queue.
  52      */
  53     if (td) {
  54         gdary[0].gd_curthread = td;
  55         lwkt_init_thread(td, NULL, TDF_RUNNING|TDF_SYSTHREAD, mycpu);
  56     }
  57 }
  58
  59 /*
  60  * per-cpu globaldata init.  Calls lwkt_gdinit() and md_gdinit*().  Returns
  61  * with the target cpu left in a critical section.
  62  */
  63 void
  64 mi_gdinit1(globaldata_t gd, int cpuid)
  65 {
  66     bzero(gd, sizeof(*gd));
  67     TAILQ_INIT(&gd->gd_tdfreeq);
  68     gd->gd_cpuid = cpuid;
  69     gd->gd_self = gd;
  70     gd->gd_upcall.upc_magic = UPCALL_MAGIC;
  71     gd->gd_upcall.upc_critoff = offsetof(struct thread, td_pri);
  72     gd->gd_ipiq = malloc(sizeof(lwkt_ipiq) * MAXVCPU);
  73     bzero(gd->gd_ipiq, sizeof(lwkt_ipiq) * MAXVCPU);
  74     md_gdinit1(gd);
  75 }
  76
  77 void
  78 mi_gdinit2(globaldata_t gd)
  79 {
  80     gd->gd_upcid = upc_register(&gd->gd_upcall, upc_callused_wrapper,
  81                                 (void *)lwkt_process_ipiq, gd);
  82     if (gd->gd_upcid < 0)
  83         panic("upc_register: failed on cpu %d\n", gd->gd_cpuid);
  84     md_gdinit2(gd);
  85     lwkt_gdinit(gd);
  86 }
  87
  88 globaldata_t
  89 globaldata_find(int cpu)
  90 {
  91     KKASSERT(cpu >= 0 && cpu < ncpus);
  92     return(&gdary[cpu]);
  93 }
  94
  95 void *
  96 libcaps_alloc_stack(int stksize)
  97 {
  98     return(malloc(stksize));
  99 }
 100
 101 void
 102 libcaps_free_stack(void *stk, int stksize)
 103 {
 104     free(stk);
 105 }
 106
 107 /*
 108  * Process any pending upcalls now.  Remember there is a dispatch interlock
 109  * if upc_pending is non-zero, so we have to set it to zero.  If we are in
 110  * a critical section this function is a NOP.
 111  */
 112 void
 113 splz(void)
 114 {
 115     globaldata_t gd = mycpu;
 116
 117     if (gd->gd_upcall.upc_pending) {
 118         gd->gd_upcall.upc_pending = 0;
 119         upc_control(UPC_CONTROL_DISPATCH, -1, (void *)1);
 120     }
 121 }
 122
 123 int
 124 need_resched(void)
 125 {
 126     return(0);
 127 }
 128
 129 void
 130 cpu_send_ipiq(int dcpu)
 131 {
 132     upc_control(UPC_CONTROL_DISPATCH, gdary[dcpu].gd_upcid, (void *)TDPRI_CRIT);
 133 }
 134
 135 void
 136 cpu_halt(void)
 137 {
 138     upc_control(UPC_CONTROL_WAIT, -1, (void *)TDPRI_CRIT);
 139 }
 140
 141 __dead2 void
 142 panic(const char *ctl, ...)
 143 {
 144     va_list va;
 145
 146     va_start(va, ctl);
 147     vfprintf(stderr, ctl, va);
 148     va_end(va);
 149     abort();
 150 }
 151
 152 /*
 153  * Create a new virtual cpu.  The cpuid is returned and may be used
 154  * in later lwkt_create() calls.
 155  */
 156 static int
 157 caps_vcpu_start(void *vgd)
 158 {
 159     mi_gdinit2(vgd);    /* sets %gs */
 160     cpu_rfork_start();
 161 }
 162
 163 int
 164 caps_fork_vcpu(void)
 165 {
 166     int cpuid;
 167     globaldata_t gd;
 168     char stack[8192];
 169
 170     if ((cpuid = ncpus) >= MAXVCPU)
 171         return(-1);
 172     ++ncpus;
 173     gd = &gdary[cpuid];
 174     mi_gdinit1(gd, cpuid);
 175     rfork_thread(RFMEM|RFPROC|RFSIGSHARE, stack + sizeof(stack),
 176                 caps_vcpu_start, gd);
 177     while (gd->gd_pid == 0)
 178         ;
 179     /* XXX wait for upcall setup */
 180     return(cpuid);
 181 }
 182