1 /* 2 * Copyright (c) 2003,2004,2008 The DragonFly Project. All rights reserved. 3 * 4 * This code is derived from software contributed to The DragonFly Project 5 * by David Xu <davidxu@t2t2.com> and Matthew Dillon <dillon@backplane.com> 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 * 11 * 1. Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in 15 * the documentation and/or other materials provided with the 16 * distribution. 17 * 3. Neither the name of The DragonFly Project nor the names of its 18 * contributors may be used to endorse or promote products derived 19 * from this software without specific, prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 22 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 23 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 24 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 25 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 26 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING, 27 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 28 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 29 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 30 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 31 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 32 * SUCH DAMAGE. 33 */ 34 35 #include <sys/param.h> 36 #include <sys/systm.h> 37 #include <sys/sysmsg.h> 38 #include <sys/kernel.h> 39 #include <sys/proc.h> 40 #include <sys/sysent.h> 41 #include <sys/sysctl.h> 42 #include <sys/tls.h> 43 #include <sys/reg.h> 44 #include <sys/globaldata.h> 45 46 #include <sys/thread2.h> 47 48 #include <machine/cpu.h> 49 #include <machine/clock.h> 50 #include <machine/specialreg.h> 51 #include <machine/segments.h> 52 #include <machine/md_var.h> 53 #include <machine/pcb_ext.h> 54 #include <machine/globaldata.h> /* CPU_prvspace */ 55 #include <machine/smp.h> 56 #include <machine/pcb.h> 57 #include <machine/thread.h> 58 #include <machine/vmm.h> 59 60 /* 61 * set a TLS descriptor. For x86_64 descriptor 0 identifies %fs and 62 * descriptor 1 identifies %gs, and 0 is returned in sysmsg_result. 63 * 64 * Returns the value userland needs to load into %gs representing the 65 * TLS descriptor or -1 on error. 66 * 67 * (int which, struct tls_info *info, size_t infosize) 68 * 69 * MPSAFE 70 */ 71 int 72 sys_set_tls_area(struct sysmsg *sysmsg, const struct set_tls_area_args *uap) 73 { 74 struct tls_info info; 75 int error; 76 int i; 77 78 /* 79 * Sanity checks 80 * 81 * which 0 == %fs, which 1 == %gs 82 */ 83 i = uap->which; 84 if (i < 0 || i > 1) 85 return (ERANGE); 86 if (uap->infosize < 0) 87 return (EINVAL); 88 89 /* 90 * Maintain forwards compatibility with future extensions. 91 */ 92 if (uap->infosize != sizeof(info)) { 93 bzero(&info, sizeof(info)); 94 error = copyin(uap->info, &info, 95 min(sizeof(info), uap->infosize)); 96 } else { 97 error = copyin(uap->info, &info, sizeof(info)); 98 } 99 if (error) 100 return (error); 101 if (info.size < -1) 102 return (EINVAL); 103 104 /* 105 * For x86_64 we can only adjust FSBASE and GSBASE 106 */ 107 curthread->td_tls.info[i] = info; 108 set_user_TLS(); 109 sysmsg->sysmsg_result = 0; /* segment descriptor $0 */ 110 return(0); 111 } 112 113 /* 114 * Return the specified TLS descriptor to userland. 115 * 116 * Returns the value userland needs to load into %gs representing the 117 * TLS descriptor or -1 on error. 118 * 119 * (int which, struct tls_info *info, size_t infosize) 120 * 121 * MPSAFE 122 */ 123 int 124 sys_get_tls_area(struct sysmsg *sysmsg, const struct get_tls_area_args *uap) 125 { 126 struct tls_info info; 127 int error; 128 int i; 129 130 /* 131 * Sanity checks 132 */ 133 i = uap->which; 134 if (i < 0 || i > 1) 135 return (ERANGE); 136 if (uap->infosize < 0) 137 return (EINVAL); 138 139 info = curthread->td_tls.info[i]; 140 141 error = copyout(&info, uap->info, min(sizeof(info), uap->infosize)); 142 return(error); 143 } 144 145 /* 146 * Install the TLS. 147 * 148 * It shouldn't be possible for a preemptive thread switch to do anything 149 * more than set gd_user_fs and wrmsr for us. Even though there is a window 150 * where gd_user_fs/gd_user_gs do not match the MSRs no preemptive thread 151 * switch should ever switch to any heavy weight thread other than our own. 152 * 153 * Still, use a critical section to be safe. 154 * 155 * MPSAFE 156 */ 157 void 158 set_user_TLS(void) 159 { 160 struct mdglobaldata *gd = mdcpu; 161 thread_t td = gd->mi.gd_curthread; 162 163 crit_enter_quick(td); 164 td->td_pcb->pcb_fsbase = (register_t)td->td_tls.info[0].base; 165 td->td_pcb->pcb_gsbase = (register_t)td->td_tls.info[1].base; 166 if (gd->gd_user_fs != td->td_pcb->pcb_fsbase) { 167 gd->gd_user_fs = td->td_pcb->pcb_fsbase; 168 wrmsr(MSR_FSBASE, gd->gd_user_fs); 169 } 170 if (gd->gd_user_gs != td->td_pcb->pcb_gsbase) { 171 gd->gd_user_gs = td->td_pcb->pcb_gsbase; 172 wrmsr(MSR_KGSBASE, gd->gd_user_gs); 173 } 174 175 if (td->td_vmm) { 176 vmm_vm_set_tls_area(); 177 } 178 179 clear_quickret(); 180 crit_exit_quick(td); 181 } 182 183