1
2
3
4
5
6
7
8
9
10
11#include <linux/cpufeature.h>
12
13#include <asm/cmdline.h>
14
15#include "cpu.h"
16
17#undef pr_fmt
18#define pr_fmt(fmt) "tsx: " fmt
19
20enum tsx_ctrl_states tsx_ctrl_state __ro_after_init = TSX_CTRL_NOT_SUPPORTED;
21
22void tsx_disable(void)
23{
24 u64 tsx;
25
26 rdmsrl(MSR_IA32_TSX_CTRL, tsx);
27
28
29 tsx |= TSX_CTRL_RTM_DISABLE;
30
31
32
33
34
35
36
37 tsx |= TSX_CTRL_CPUID_CLEAR;
38
39 wrmsrl(MSR_IA32_TSX_CTRL, tsx);
40}
41
42void tsx_enable(void)
43{
44 u64 tsx;
45
46 rdmsrl(MSR_IA32_TSX_CTRL, tsx);
47
48
49 tsx &= ~TSX_CTRL_RTM_DISABLE;
50
51
52
53
54
55
56 tsx &= ~TSX_CTRL_CPUID_CLEAR;
57
58 wrmsrl(MSR_IA32_TSX_CTRL, tsx);
59}
60
61static bool __init tsx_ctrl_is_supported(void)
62{
63 u64 ia32_cap = x86_read_arch_cap_msr();
64
65
66
67
68
69
70
71
72
73
74
75
76 return !!(ia32_cap & ARCH_CAP_TSX_CTRL_MSR);
77}
78
79static enum tsx_ctrl_states x86_get_tsx_auto_mode(void)
80{
81 if (boot_cpu_has_bug(X86_BUG_TAA))
82 return TSX_CTRL_DISABLE;
83
84 return TSX_CTRL_ENABLE;
85}
86
87void __init tsx_init(void)
88{
89 char arg[5] = {};
90 int ret;
91
92 if (!tsx_ctrl_is_supported())
93 return;
94
95 ret = cmdline_find_option(boot_command_line, "tsx", arg, sizeof(arg));
96 if (ret >= 0) {
97 if (!strcmp(arg, "on")) {
98 tsx_ctrl_state = TSX_CTRL_ENABLE;
99 } else if (!strcmp(arg, "off")) {
100 tsx_ctrl_state = TSX_CTRL_DISABLE;
101 } else if (!strcmp(arg, "auto")) {
102 tsx_ctrl_state = x86_get_tsx_auto_mode();
103 } else {
104 tsx_ctrl_state = TSX_CTRL_DISABLE;
105 pr_err("invalid option, defaulting to off\n");
106 }
107 } else {
108
109 if (IS_ENABLED(CONFIG_X86_INTEL_TSX_MODE_AUTO))
110 tsx_ctrl_state = x86_get_tsx_auto_mode();
111 else if (IS_ENABLED(CONFIG_X86_INTEL_TSX_MODE_OFF))
112 tsx_ctrl_state = TSX_CTRL_DISABLE;
113 else
114 tsx_ctrl_state = TSX_CTRL_ENABLE;
115 }
116
117 if (tsx_ctrl_state == TSX_CTRL_DISABLE) {
118 tsx_disable();
119
120
121
122
123
124
125 setup_clear_cpu_cap(X86_FEATURE_RTM);
126 setup_clear_cpu_cap(X86_FEATURE_HLE);
127 } else if (tsx_ctrl_state == TSX_CTRL_ENABLE) {
128
129
130
131
132
133
134
135 tsx_enable();
136
137
138
139
140
141 setup_force_cpu_cap(X86_FEATURE_RTM);
142 setup_force_cpu_cap(X86_FEATURE_HLE);
143 }
144}
145