update/
install.rs

1// Copyright 2020 The Fuchsia Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5use anyhow::{anyhow, Context, Error};
6use fidl_fuchsia_update_installer::{InstallerMarker, InstallerProxy, RebootControllerMarker};
7use fidl_fuchsia_update_installer_ext::{self as installer, start_update, Options, StateId};
8use fuchsia_component::client::connect_to_protocol;
9use fuchsia_url::AbsolutePackageUrl;
10use futures::prelude::*;
11
12pub async fn handle_force_install(
13    update_pkg_url: String,
14    reboot: bool,
15    service_initiated: bool,
16) -> Result<(), Error> {
17    let installer = connect_to_protocol::<InstallerMarker>()
18        .context("connecting to fuchsia.update.installer")?;
19    handle_force_install_impl(update_pkg_url, reboot, service_initiated, &installer).await
20}
21
22async fn handle_force_install_impl(
23    update_pkg_url: String,
24    reboot: bool,
25    service_initiated: bool,
26    installer: &InstallerProxy,
27) -> Result<(), Error> {
28    let pkg_url =
29        AbsolutePackageUrl::parse(&update_pkg_url).context("parsing update package url")?;
30
31    let options = Options {
32        initiator: if service_initiated {
33            installer::Initiator::Service
34        } else {
35            installer::Initiator::User
36        },
37        should_write_recovery: true,
38        allow_attach_to_existing_attempt: true,
39    };
40
41    let (reboot_controller, reboot_controller_server_end) =
42        fidl::endpoints::create_proxy::<RebootControllerMarker>();
43
44    let mut update_attempt =
45        start_update(&pkg_url, options, installer, Some(reboot_controller_server_end))
46            .await
47            .context("starting update")?;
48
49    println!(
50        "Installing an update.
51Progress reporting is based on the fraction of packages resolved, so if one package is much
52larger than the others, then the reported progress could appear to stall near the end.
53Until the update process is improved to have more granular reporting, try using
54    ffx inspect show 'core/pkg-resolver'
55for more detail on the progress of update-related downloads.
56"
57    );
58    if !reboot {
59        reboot_controller.detach().context("notify installer do not reboot")?;
60    }
61    while let Some(state) = update_attempt.try_next().await.context("getting next state")? {
62        println!("State: {state:?}");
63        if state.id() == StateId::WaitToReboot {
64            if reboot {
65                return Ok(());
66            }
67        } else if state.is_success() {
68            return Ok(());
69        } else if state.is_failure() {
70            anyhow::bail!("Encountered failure state");
71        }
72    }
73
74    Err(anyhow!("Installation ended unexpectedly"))
75}
update/install.rs

update/
install.rs
