1331722Seadler/* 2230429Skib * Copyright (c) 2011 Konstantin Belousov <kib@FreeBSD.org> 3230429Skib * All rights reserved. 4230429Skib * 5230429Skib * Redistribution and use in source and binary forms, with or without 6230429Skib * modification, are permitted provided that the following conditions 7230429Skib * are met: 8230429Skib * 9230429Skib * 1. Redistributions of source code must retain the above copyright 10230429Skib * notice, this list of conditions and the following disclaimer. 11230429Skib * 2. Redistributions in binary form must reproduce the above copyright 12230429Skib * notice, this list of conditions and the following disclaimer in the 13230429Skib * documentation and/or other materials provided with the distribution. 14230429Skib * 15230429Skib * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 16230429Skib * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17230429Skib * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18230429Skib * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 19230429Skib * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20230429Skib * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21230429Skib * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22230429Skib * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23230429Skib * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24230429Skib * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25230429Skib */ 26230429Skib 27230429Skib#include <sys/cdefs.h> 28230429Skib__FBSDID("$FreeBSD$"); 29230429Skib 30230429Skib#include <sys/types.h> 31230429Skib#include <sys/ucontext.h> 32230429Skib#include <errno.h> 33230429Skib#include <stdarg.h> 34230429Skib#include <stdlib.h> 35230429Skib#include <machine/cpufunc.h> 36230429Skib#include <machine/fpu.h> 37230429Skib#include <machine/specialreg.h> 38230429Skib#include <machine/sysarch.h> 39230429Skib 40230429Skibstatic int xstate_sz = -1; 41230429Skib 42230864Skibint 43230429Skib__getcontextx_size(void) 44230429Skib{ 45230429Skib u_int p[4]; 46230429Skib 47230429Skib if (xstate_sz == -1) { 48230429Skib do_cpuid(1, p); 49230429Skib if ((p[2] & CPUID2_OSXSAVE) != 0) { 50230429Skib cpuid_count(0xd, 0x0, p); 51230429Skib xstate_sz = p[1] - sizeof(struct savefpu); 52230429Skib } else 53230429Skib xstate_sz = 0; 54230429Skib } 55230429Skib 56230429Skib return (sizeof(ucontext_t) + xstate_sz); 57230429Skib} 58230429Skib 59230429Skibint 60251047Skib__fillcontextx2(char *ctx) 61230429Skib{ 62230429Skib struct amd64_get_xfpustate xfpu; 63230429Skib ucontext_t *ucp; 64230429Skib 65230429Skib ucp = (ucontext_t *)ctx; 66230429Skib if (xstate_sz != 0) { 67230429Skib xfpu.addr = (char *)(ucp + 1); 68230429Skib xfpu.len = xstate_sz; 69230429Skib if (sysarch(AMD64_GET_XFPUSTATE, &xfpu) == -1) 70230429Skib return (-1); 71230429Skib ucp->uc_mcontext.mc_xfpustate = (__register_t)xfpu.addr; 72230429Skib ucp->uc_mcontext.mc_xfpustate_len = xstate_sz; 73230429Skib ucp->uc_mcontext.mc_flags |= _MC_HASFPXSTATE; 74230429Skib } else { 75230429Skib ucp->uc_mcontext.mc_xfpustate = 0; 76230429Skib ucp->uc_mcontext.mc_xfpustate_len = 0; 77230429Skib } 78230429Skib return (0); 79230429Skib} 80230429Skib 81251047Skibint 82251047Skib__fillcontextx(char *ctx) 83251047Skib{ 84251047Skib ucontext_t *ucp; 85251047Skib 86251047Skib ucp = (ucontext_t *)ctx; 87251047Skib if (getcontext(ucp) == -1) 88251047Skib return (-1); 89251047Skib __fillcontextx2(ctx); 90251047Skib return (0); 91251047Skib} 92251047Skib 93230429Skib__weak_reference(__getcontextx, getcontextx); 94230429Skib 95230429Skibucontext_t * 96230429Skib__getcontextx(void) 97230429Skib{ 98230429Skib char *ctx; 99230429Skib int error; 100230429Skib 101230429Skib ctx = malloc(__getcontextx_size()); 102230429Skib if (ctx == NULL) 103230429Skib return (NULL); 104230429Skib if (__fillcontextx(ctx) == -1) { 105230429Skib error = errno; 106230429Skib free(ctx); 107230429Skib errno = error; 108230429Skib return (NULL); 109230429Skib } 110230429Skib return ((ucontext_t *)ctx); 111230429Skib} 112