src/storage/flashmap/src/main.rs

// Copyright 2021 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

mod flashmap;
mod manager;
mod util;

use {
    crate::manager::Manager,
    anyhow::{self, Context},
    fidl_fuchsia_acpi_chromeos::DeviceMarker,
    fidl_fuchsia_nand_flashmap::ManagerRequestStream,
    fuchsia_component::server::ServiceFs,
    fuchsia_inspect::{component, health::Reporter},
    fuchsia_syslog::fx_log_warn,
    fuchsia_zircon as zx,
    futures::prelude::*,
    tracing,
};

const FLASH_PHYS_BASE: u64 = 0xff000000;

enum IncomingRequest {
    FlashmapManager(ManagerRequestStream),
}

#[fuchsia::main(logging = true)]
async fn main() -> Result<(), anyhow::Error> {
    let mut service_fs = ServiceFs::new_local();

    // Initialize inspect
    inspect_runtime::serve(component::inspector(), &mut service_fs)?;
    component::health().set_starting_up();

    let acpi_path = "/dev/class/chromeos-acpi/000";
    let device = fuchsia_component::client::connect_to_protocol_at_path::<DeviceMarker>(acpi_path)
        .context("Connecting to ChromeOS ACPI service")?;
    let address =
        match device.get_flashmap_address().await.context("Sending get flashmap address")? {
            Ok(value) => Some(value - FLASH_PHYS_BASE),
            Err(e) => {
                fx_log_warn!("Failed to get flashmap address: {:?}", zx::Status::from_raw(e));
                None
            }
        };

    let manager = Manager::new(address);

    service_fs.dir("svc").add_fidl_service(IncomingRequest::FlashmapManager);
    service_fs.take_and_serve_directory_handle().context("failed to serve outgoing namespace")?;

    component::health().set_ok();
    tracing::debug!("Initialized.");

    let manager_ref = &manager;
    service_fs
        .for_each_concurrent(None, |request: IncomingRequest| async move {
            match request {
                IncomingRequest::FlashmapManager(stream) => manager_ref.serve(stream).await,
            };
        })
        .await;

    Ok(())
}