1 /*- 2 * Copyright (c) 2006, 2008 Stanislav Sedov <stas@FreeBSD.org>. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 15 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 16 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 17 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 18 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 19 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 20 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 21 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 23 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26 #include <sys/cdefs.h> 27 __FBSDID("$FreeBSD: release/10.0.0/usr.sbin/cpucontrol/amd.c 236504 2012-06-03 08:30:00Z avg $"); 28 29 #include <assert.h> 30 #include <stdio.h> 31 #include <stdlib.h> 32 #include <string.h> 33 #include <unistd.h> 34 #include <fcntl.h> 35 #include <err.h> 36 37 #include <sys/types.h> 38 #include <sys/stat.h> 39 #include <sys/mman.h> 40 #include <sys/ioctl.h> 41 #include <sys/ioccom.h> 42 #include <sys/cpuctl.h> 43 44 #include <machine/cpufunc.h> 45 #include <machine/specialreg.h> 46 47 #include "cpucontrol.h" 48 #include "amd.h" 49 50 int 51 amd_probe(int fd) 52 { 53 char vendor[13]; 54 int error; 55 cpuctl_cpuid_args_t idargs = { 56 .level = 0, 57 }; 58 59 error = ioctl(fd, CPUCTL_CPUID, &idargs); 60 if (error < 0) { 61 WARN(0, "ioctl()"); 62 return (1); 63 } 64 ((uint32_t *)vendor)[0] = idargs.data[1]; 65 ((uint32_t *)vendor)[1] = idargs.data[3]; 66 ((uint32_t *)vendor)[2] = idargs.data[2]; 67 vendor[12] = '\0'; 68 if (strncmp(vendor, AMD_VENDOR_ID, sizeof(AMD_VENDOR_ID)) != 0) 69 return (1); 70 return (0); 71 } 72 73 void 74 amd_update(const char *dev, const char *path) 75 { 76 int fd, devfd; 77 unsigned int i; 78 struct stat st; 79 uint32_t *fw_image; 80 amd_fw_header_t *fw_header; 81 uint32_t sum; 82 uint32_t signature; 83 uint32_t *fw_data; 84 size_t fw_size; 85 cpuctl_cpuid_args_t idargs = { 86 .level = 1, /* Request signature. */ 87 }; 88 cpuctl_update_args_t args; 89 int error; 90 91 assert(path); 92 assert(dev); 93 94 fd = -1; 95 fw_image = MAP_FAILED; 96 devfd = open(dev, O_RDWR); 97 if (devfd < 0) { 98 WARN(0, "could not open %s for writing", dev); 99 return; 100 } 101 error = ioctl(devfd, CPUCTL_CPUID, &idargs); 102 if (error < 0) { 103 WARN(0, "ioctl()"); 104 goto fail; 105 } 106 signature = idargs.data[0]; 107 WARNX(2, "found cpu family %#x model %#x " 108 "stepping %#x extfamily %#x extmodel %#x.", 109 (signature >> 8) & 0x0f, (signature >> 4) & 0x0f, 110 (signature >> 0) & 0x0f, (signature >> 20) & 0xff, 111 (signature >> 16) & 0x0f); 112 113 /* 114 * Open the firmware file. 115 */ 116 fd = open(path, O_RDONLY, 0); 117 if (fd < 0) { 118 WARN(0, "open(%s)", path); 119 goto fail; 120 } 121 error = fstat(fd, &st); 122 if (error != 0) { 123 WARN(0, "fstat(%s)", path); 124 goto fail; 125 } 126 if (st.st_size < 0 || (unsigned)st.st_size < sizeof(*fw_header)) { 127 WARNX(2, "file too short: %s", path); 128 goto fail; 129 } 130 /* 131 * mmap the whole image. 132 */ 133 fw_image = (uint32_t *)mmap(NULL, st.st_size, PROT_READ, 134 MAP_PRIVATE, fd, 0); 135 if (fw_image == MAP_FAILED) { 136 WARN(0, "mmap(%s)", path); 137 goto fail; 138 } 139 fw_header = (amd_fw_header_t *)fw_image; 140 if ((fw_header->magic >> 8) != AMD_MAGIC) { 141 WARNX(2, "%s is not a valid amd firmware: version mismatch", 142 path); 143 goto fail; 144 } 145 fw_data = (uint32_t *)(fw_header + 1); 146 fw_size = (st.st_size - sizeof(*fw_header)) / sizeof(uint32_t); 147 148 /* 149 * Check the primary checksum. 150 */ 151 sum = 0; 152 for (i = 0; i < fw_size; i++) 153 sum += fw_data[i]; 154 if (sum != fw_header->checksum) { 155 WARNX(2, "%s: update data checksum invalid", path); 156 goto fail; 157 } 158 if (signature == fw_header->signature) { 159 fprintf(stderr, "%s: updating cpu %s... ", path, dev); 160 161 args.data = fw_image; 162 args.size = st.st_size; 163 error = ioctl(devfd, CPUCTL_UPDATE, &args); 164 if (error < 0) { 165 fprintf(stderr, "failed.\n"); 166 warn("ioctl()"); 167 goto fail; 168 } 169 fprintf(stderr, "done.\n"); 170 } 171 172 fail: 173 if (fd >= 0) 174 close(fd); 175 if (devfd >= 0) 176 close(devfd); 177 if (fw_image != MAP_FAILED) 178 if(munmap(fw_image, st.st_size) != 0) 179 warn("munmap(%s)", path); 180 return; 181 } 182