Skip to content

Commit d651285

Browse files
committed
Modify Makefile for Azure, refactor code in main.rs
Revert changes to do_measurements Refactor modified: sh_script/Azure/Makefile modified: sh_script/fuzzing.sh modified: src/migtd/src/bin/migtd/main.rs Add formatting. modified: sh_script/Azure/Makefile modified: sh_script/fuzzing.sh modified: src/migtd/src/bin/migtd/main.rs Increase timeout for build_AzCVMEmu_policy_and_test.sh to complete. modified: .github/workflows/integration-emu.yml modified: sh_script/Azure/Makefile modified: sh_script/fuzzing.sh modified: src/migtd/src/bin/migtd/main.rs Remove non Makefile changes.
1 parent 4e19941 commit d651285

File tree

2 files changed

+36
-10
lines changed

2 files changed

+36
-10
lines changed

sh_script/fuzzing.sh

100644100755
File mode changed.

src/migtd/src/bin/migtd/main.rs

Lines changed: 36 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,7 @@ fn basic_info() {
139139
info!("MigTD Version - {}\n", MIGTD_VERSION);
140140
}
141141

142+
#[cfg(not(feature = "policy_v2"))]
142143
fn do_measurements() {
143144
// Get the event log recorded by firmware
144145
let event_log = event_log::get_event_log_mut().expect("Failed to get the event log");
@@ -147,18 +148,29 @@ fn do_measurements() {
147148
measure_test_feature(event_log);
148149
return;
149150
}
150-
151-
#[cfg(feature = "policy_v2")]
152-
get_policy_issuer_chain_and_measure(event_log);
153-
154151
// Get migration td policy from CFV and measure it into RMTR
155152
get_policy_and_measure(event_log);
156153

157154
// Get root certificate from CFV and measure it into RMTR
158-
#[cfg(not(feature = "policy_v2"))]
159155
get_ca_and_measure(event_log);
160156
}
161157

158+
#[cfg(feature = "policy_v2")]
159+
fn do_measurements() {
160+
// Get the event log recorded by firmware
161+
let event_log = event_log::get_event_log_mut().expect("Failed to get the event log");
162+
163+
if cfg!(feature = "test_disable_ra_and_accept_all") {
164+
measure_test_feature(event_log);
165+
return;
166+
}
167+
168+
get_policy_issuer_chain_and_measure(event_log);
169+
170+
// Get migration td policy from CFV and measure it into RMTR
171+
get_policy_and_measure(event_log);
172+
}
173+
162174
fn measure_test_feature(event_log: &mut [u8]) {
163175
// Measure and extend the migtd test feature to RTMR
164176
event_log::write_tagged_event_log(
@@ -171,19 +183,33 @@ fn measure_test_feature(event_log: &mut [u8]) {
171183
.expect("Failed to log migtd test feature");
172184
}
173185

186+
#[cfg(not(feature = "policy_v2"))]
187+
fn get_policy_and_measure(event_log: &mut [u8]) {
188+
// Read migration policy from CFV
189+
let policy = config::get_policy().expect("Fail to get policy from CFV\n");
190+
191+
let event_data = policy;
192+
193+
// Measure and extend the migration policy to RTMR
194+
event_log::write_tagged_event_log(
195+
event_log,
196+
MR_INDEX_POLICY,
197+
policy,
198+
TAGGED_EVENT_ID_POLICY,
199+
event_data,
200+
)
201+
.expect("Failed to log migration policy");
202+
}
203+
204+
#[cfg(feature = "policy_v2")]
174205
fn get_policy_and_measure(event_log: &mut [u8]) {
175206
// Read migration policy from CFV
176207
let policy = config::get_policy().expect("Fail to get policy from CFV\n");
177208

178-
#[cfg(feature = "policy_v2")]
179209
let version = initialize_policy();
180210

181-
#[cfg(feature = "policy_v2")]
182211
let event_data = version.as_bytes();
183212

184-
#[cfg(not(feature = "policy_v2"))]
185-
let event_data = policy;
186-
187213
// Measure and extend the migration policy to RTMR
188214
event_log::write_tagged_event_log(
189215
event_log,

0 commit comments

Comments
 (0)