Skip to content

Commit 513632d

Browse files
committed
runtime: check amd64 microarchitecture level at startup
Make Go runtime throw if it's been compiled to assume instruction set extensions that aren't available on the CPU. Updates #485064
1 parent 7d67f8d commit 513632d

File tree

1 file changed

+71
-2
lines changed

1 file changed

+71
-2
lines changed

src/runtime/asm_amd64.s

+71-2
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,30 @@ GLOBL _rt0_amd64_lib_argc<>(SB),NOPTR, $8
7878
DATA _rt0_amd64_lib_argv<>(SB)/8, $0
7979
GLOBL _rt0_amd64_lib_argv<>(SB),NOPTR, $8
8080

81+
#ifdef GOAMD64_v2
82+
DATA bad_proc_msg<>+0x00(SB)/78, $"This program can only be run on processors with v2 microarchitecture support.\n"
83+
#endif
84+
85+
#ifdef GOAMD64_v3
86+
DATA bad_proc_msg<>+0x00(SB)/78, $"This program can only be run on processors with v3 microarchitecture support.\n"
87+
#endif
88+
89+
GLOBL bad_proc_msg<>(SB), RODATA, $78
90+
91+
// Define a list of AMD64 microarchitcture level features
92+
// https://en.wikipedia.org/wiki/X86-64#Microarchitecture_levels
93+
94+
// SSE3 SSSE3 CMPXCHNG16 SSE4.1 SSE4.2 POPCNT
95+
#define V2_FEATURES_CX (1 << 0 | 1 << 9 | 1 << 13 | 1 << 19 | 1 << 20 | 1 << 23)
96+
// LAHF/SAHF
97+
#define V2_EXT_FEATURES_CX (1 << 0)
98+
// FMA MOVBE OSXSAVE AVX F16C
99+
#define V3_FEATURES_CX (V2_FEATURES_CX | 1 << 12 | 1 << 22 | 1 << 27 | 1 << 28 | 1 << 29)
100+
// ABM (FOR LZNCT)
101+
#define V3_EXT_FEATURES_CX (V2_EXT_FEATURES_CX | 1 << 5)
102+
// BMI1 AVX2 BMI2
103+
#define V3_EXT_FEATURES_BX (1 << 3 | 1 << 5 | 1 << 8)
104+
81105
TEXT runtime·rt0_go(SB),NOSPLIT|TOPFRAME,$0
82106
// copy arguments forward on an even stack
83107
MOVQ DI, AX // argc
@@ -101,22 +125,67 @@ TEXT runtime·rt0_go(SB),NOSPLIT|TOPFRAME,$0
101125
CPUID
102126
MOVL AX, SI
103127
CMPL AX, $0
104-
JE nocpuinfo
128+
#ifdef GOAMD64_v1
129+
JE nocpuinfo
130+
#else
131+
JNE has_cpuinfo
132+
bad_proc: // show that the program requires MMX.
133+
MOVQ $2, 0(SP)
134+
MOVQ $bad_proc_msg<>(SB), AX
135+
MOVQ AX, 8(SP)
136+
MOVQ $78, 16(SP)
137+
CALL runtime·write(SB)
138+
MOVQ $1, 0(SP)
139+
CALL runtime·exit(SB)
140+
CALL runtime·abort(SB)
141+
#endif
105142

143+
has_cpuinfo:
106144
CMPL BX, $0x756E6547 // "Genu"
107145
JNE notintel
108146
CMPL DX, $0x49656E69 // "ineI"
109147
JNE notintel
110148
CMPL CX, $0x6C65746E // "ntel"
111149
JNE notintel
112150
MOVB $1, runtime·isIntel(SB)
113-
notintel:
114151

152+
notintel:
115153
// Load EAX=1 cpuid flags
116154
MOVL $1, AX
117155
CPUID
118156
MOVL AX, runtime·processorVersionInfo(SB)
119157

158+
#ifdef GOAMD64_v2
159+
ANDL $V2_FEATURES_CX, CX
160+
CMPL CX, $V2_FEATURES_CX
161+
JNZ bad_proc
162+
MOVL $0x80000001, AX
163+
CPUID
164+
TESTQ $V2_EXT_FEATURES_CX, CX
165+
JZ bad_proc
166+
#endif
167+
168+
#ifdef GOAMD64_v3
169+
ANDL $V3_FEATURES_CX, CX
170+
CMPL CX, $V3_FEATURES_CX
171+
JNZ bad_proc
172+
MOVL $7, AX
173+
MOVL $0, CX
174+
CPUID
175+
ANDL $V3_EXT_FEATURES_BX, BX
176+
CMPL BX, $V3_EXT_FEATURES_BX
177+
JNZ bad_proc
178+
MOVL $0x80000000, AX
179+
CPUID
180+
CMPL AX, $0x80000001
181+
JL bad_proc
182+
MOVL $0x80000001, AX
183+
CPUID
184+
ANDL $V3_EXT_FEATURES_CX, CX
185+
CMPL CX, $V3_EXT_FEATURES_CX
186+
JNZ bad_proc
187+
#endif
188+
120189
nocpuinfo:
121190
// if there is an _cgo_init, call it.
122191
MOVQ _cgo_init(SB), AX

0 commit comments

Comments
 (0)