Alan Stokes | 3ef78d9 | 2021-09-08 11:51:06 +0100 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2021 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | //! Implementation of IIsolatedCompilationService, called from system server when compilation is |
| 18 | //! desired. |
| 19 | |
Alan Stokes | a2869d2 | 2021-09-22 09:06:41 +0100 | [diff] [blame] | 20 | use crate::instance_manager::InstanceManager; |
Alan Stokes | 8c84044 | 2021-11-26 15:54:30 +0000 | [diff] [blame] | 21 | use crate::odrefresh_task::OdrefreshTask; |
Alan Stokes | 9ca14ca | 2021-10-20 14:25:57 +0100 | [diff] [blame] | 22 | use android_system_composd::aidl::android::system::composd::{ |
| 23 | ICompilationTask::{BnCompilationTask, ICompilationTask}, |
| 24 | ICompilationTaskCallback::ICompilationTaskCallback, |
Victor Hsieh | bac923e | 2022-02-22 21:43:36 +0000 | [diff] [blame^] | 25 | IIsolatedCompilationService::{ |
| 26 | ApexSource::ApexSource, BnIsolatedCompilationService, IIsolatedCompilationService, |
| 27 | }, |
Alan Stokes | 3ef78d9 | 2021-09-08 11:51:06 +0100 | [diff] [blame] | 28 | }; |
Alan Stokes | cb732dc | 2021-11-16 15:18:13 +0000 | [diff] [blame] | 29 | use android_system_composd::binder::{ |
| 30 | self, BinderFeatures, ExceptionCode, Interface, Status, Strong, ThreadState, |
| 31 | }; |
Alan Stokes | 9ca14ca | 2021-10-20 14:25:57 +0100 | [diff] [blame] | 32 | use anyhow::{Context, Result}; |
Alan Stokes | 2d2e4db | 2022-01-28 16:41:52 +0000 | [diff] [blame] | 33 | use compos_aidl_interface::aidl::com::android::compos::ICompOsService::CompilationMode::CompilationMode; |
Alan Stokes | 126fd51 | 2021-12-16 15:00:01 +0000 | [diff] [blame] | 34 | use compos_common::binder::to_binder_result; |
Alan Stokes | 16fb855 | 2022-02-10 15:07:27 +0000 | [diff] [blame] | 35 | use compos_common::odrefresh::{PENDING_ARTIFACTS_SUBDIR, TEST_ARTIFACTS_SUBDIR}; |
Alan Stokes | ac9aa1a | 2021-12-14 11:32:13 +0000 | [diff] [blame] | 36 | use rustutils::{users::AID_ROOT, users::AID_SYSTEM}; |
Alan Stokes | e5e1d8d | 2021-11-19 16:31:14 +0000 | [diff] [blame] | 37 | use std::sync::Arc; |
Alan Stokes | 3ef78d9 | 2021-09-08 11:51:06 +0100 | [diff] [blame] | 38 | |
Alan Stokes | a2869d2 | 2021-09-22 09:06:41 +0100 | [diff] [blame] | 39 | pub struct IsolatedCompilationService { |
Alan Stokes | e5e1d8d | 2021-11-19 16:31:14 +0000 | [diff] [blame] | 40 | instance_manager: Arc<InstanceManager>, |
Alan Stokes | a2869d2 | 2021-09-22 09:06:41 +0100 | [diff] [blame] | 41 | } |
Alan Stokes | 3ef78d9 | 2021-09-08 11:51:06 +0100 | [diff] [blame] | 42 | |
Alan Stokes | e5e1d8d | 2021-11-19 16:31:14 +0000 | [diff] [blame] | 43 | pub fn new_binder( |
| 44 | instance_manager: Arc<InstanceManager>, |
| 45 | ) -> Strong<dyn IIsolatedCompilationService> { |
Alan Stokes | 69c610f | 2021-09-27 14:03:31 +0100 | [diff] [blame] | 46 | let service = IsolatedCompilationService { instance_manager }; |
Alan Stokes | 3ef78d9 | 2021-09-08 11:51:06 +0100 | [diff] [blame] | 47 | BnIsolatedCompilationService::new_binder(service, BinderFeatures::default()) |
| 48 | } |
| 49 | |
Alan Stokes | 3ef78d9 | 2021-09-08 11:51:06 +0100 | [diff] [blame] | 50 | impl Interface for IsolatedCompilationService {} |
| 51 | |
| 52 | impl IIsolatedCompilationService for IsolatedCompilationService { |
Alan Stokes | 6fc1837 | 2021-11-25 17:50:27 +0000 | [diff] [blame] | 53 | fn startStagedApexCompile( |
| 54 | &self, |
| 55 | callback: &Strong<dyn ICompilationTaskCallback>, |
| 56 | ) -> binder::Result<Strong<dyn ICompilationTask>> { |
| 57 | check_permissions()?; |
| 58 | to_binder_result(self.do_start_staged_apex_compile(callback)) |
| 59 | } |
| 60 | |
Alan Stokes | 9ca14ca | 2021-10-20 14:25:57 +0100 | [diff] [blame] | 61 | fn startTestCompile( |
| 62 | &self, |
Victor Hsieh | bac923e | 2022-02-22 21:43:36 +0000 | [diff] [blame^] | 63 | apex_source: ApexSource, |
Alan Stokes | 9ca14ca | 2021-10-20 14:25:57 +0100 | [diff] [blame] | 64 | callback: &Strong<dyn ICompilationTaskCallback>, |
| 65 | ) -> binder::Result<Strong<dyn ICompilationTask>> { |
Alan Stokes | 6fc1837 | 2021-11-25 17:50:27 +0000 | [diff] [blame] | 66 | check_permissions()?; |
Victor Hsieh | bac923e | 2022-02-22 21:43:36 +0000 | [diff] [blame^] | 67 | let prefer_staged = match apex_source { |
| 68 | ApexSource::NoStaged => false, |
| 69 | ApexSource::PreferStaged => true, |
| 70 | _ => unreachable!("Invalid ApexSource {:?}", apex_source), |
| 71 | }; |
| 72 | to_binder_result(self.do_start_test_compile(prefer_staged, callback)) |
Alan Stokes | b2cc79e | 2021-09-14 14:08:46 +0100 | [diff] [blame] | 73 | } |
| 74 | } |
| 75 | |
Alan Stokes | b2cc79e | 2021-09-14 14:08:46 +0100 | [diff] [blame] | 76 | impl IsolatedCompilationService { |
Alan Stokes | 6fc1837 | 2021-11-25 17:50:27 +0000 | [diff] [blame] | 77 | fn do_start_staged_apex_compile( |
| 78 | &self, |
| 79 | callback: &Strong<dyn ICompilationTaskCallback>, |
| 80 | ) -> Result<Strong<dyn ICompilationTask>> { |
Alan Stokes | 6542fdd | 2022-02-17 15:21:46 +0000 | [diff] [blame] | 81 | let comp_os = self.instance_manager.start_current_instance().context("Starting CompOS")?; |
Alan Stokes | 6fc1837 | 2021-11-25 17:50:27 +0000 | [diff] [blame] | 82 | |
Alan Stokes | 16fb855 | 2022-02-10 15:07:27 +0000 | [diff] [blame] | 83 | let target_dir_name = PENDING_ARTIFACTS_SUBDIR.to_owned(); |
Alan Stokes | 2d2e4db | 2022-01-28 16:41:52 +0000 | [diff] [blame] | 84 | let task = OdrefreshTask::start( |
| 85 | comp_os, |
| 86 | CompilationMode::NORMAL_COMPILE, |
| 87 | target_dir_name, |
| 88 | callback, |
| 89 | )?; |
Alan Stokes | 6fc1837 | 2021-11-25 17:50:27 +0000 | [diff] [blame] | 90 | |
| 91 | Ok(BnCompilationTask::new_binder(task, BinderFeatures::default())) |
| 92 | } |
| 93 | |
Alan Stokes | 9ca14ca | 2021-10-20 14:25:57 +0100 | [diff] [blame] | 94 | fn do_start_test_compile( |
| 95 | &self, |
Victor Hsieh | bac923e | 2022-02-22 21:43:36 +0000 | [diff] [blame^] | 96 | prefer_staged: bool, |
Alan Stokes | 9ca14ca | 2021-10-20 14:25:57 +0100 | [diff] [blame] | 97 | callback: &Strong<dyn ICompilationTaskCallback>, |
| 98 | ) -> Result<Strong<dyn ICompilationTask>> { |
Victor Hsieh | bac923e | 2022-02-22 21:43:36 +0000 | [diff] [blame^] | 99 | let comp_os = |
| 100 | self.instance_manager.start_test_instance(prefer_staged).context("Starting CompOS")?; |
Alan Stokes | b2cc79e | 2021-09-14 14:08:46 +0100 | [diff] [blame] | 101 | |
Alan Stokes | 16fb855 | 2022-02-10 15:07:27 +0000 | [diff] [blame] | 102 | let target_dir_name = TEST_ARTIFACTS_SUBDIR.to_owned(); |
Alan Stokes | 2d2e4db | 2022-01-28 16:41:52 +0000 | [diff] [blame] | 103 | let task = OdrefreshTask::start( |
| 104 | comp_os, |
| 105 | CompilationMode::TEST_COMPILE, |
| 106 | target_dir_name, |
| 107 | callback, |
| 108 | )?; |
Alan Stokes | 8c84044 | 2021-11-26 15:54:30 +0000 | [diff] [blame] | 109 | |
| 110 | Ok(BnCompilationTask::new_binder(task, BinderFeatures::default())) |
| 111 | } |
Victor Hsieh | 72c774c | 2021-11-18 15:52:28 -0800 | [diff] [blame] | 112 | } |
| 113 | |
Alan Stokes | 6fc1837 | 2021-11-25 17:50:27 +0000 | [diff] [blame] | 114 | fn check_permissions() -> binder::Result<()> { |
Victor Hsieh | 72c774c | 2021-11-18 15:52:28 -0800 | [diff] [blame] | 115 | let calling_uid = ThreadState::get_calling_uid(); |
| 116 | // This should only be called by system server, or root while testing |
| 117 | if calling_uid != AID_SYSTEM && calling_uid != AID_ROOT { |
| 118 | Err(Status::new_exception(ExceptionCode::SECURITY, None)) |
| 119 | } else { |
| 120 | Ok(()) |
| 121 | } |
| 122 | } |