1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

// Copyright (c) The Move Contributors
// SPDX-License-Identifier: Apache-2.0

use super::reroot_path;
use clap::*;
use move_compiler::compiled_unit::{CompiledUnit, NamedCompiledModule};
use move_disassembler::disassembler::Disassembler;
use move_package::{compilation::compiled_package::CompiledUnitWithSource, BuildConfig};
use std::path::PathBuf;

/// Disassemble the Move bytecode pointed to
#[derive(Parser)]
#[clap(name = "disassemble")]
pub struct Disassemble {
    /// Start a disassembled bytecode-to-source explorer
    #[clap(long = "interactive")]
    pub interactive: bool,
    /// The package name. If not provided defaults to current package modules only
    #[clap(long = "package")]
    pub package_name: Option<String>,
    /// The name of the module or script in the package to disassemble
    #[clap(long = "name")]
    pub module_or_script_name: String,
}

impl Disassemble {
    pub fn execute(self, path: Option<PathBuf>, config: BuildConfig) -> anyhow::Result<()> {
        let rerooted_path = reroot_path(path)?;
        let Self {
            interactive,
            package_name,
            module_or_script_name,
        } = self;
        // Make sure the package is built
        let package = config.compile_package(&rerooted_path, &mut Vec::new())?;
        let needle_package = package_name
            .as_deref()
            .unwrap_or(package.compiled_package_info.package_name.as_str());
        match package
            .get_module_by_name(needle_package, &module_or_script_name)
            .ok()
        {
            None => anyhow::bail!(
                "Unable to find module or script with name '{}' in package '{}'",
                module_or_script_name,
                needle_package,
            ),
            Some(unit) => {
                // Once we find the compiled bytecode we're interested in, startup the bytecode
                // viewer, or run the disassembler depending on if we need to run interactively
                // or not.
                if interactive {
                    match unit {
                        CompiledUnitWithSource {
                            unit:
                                CompiledUnit::Module(NamedCompiledModule {
                                    module, source_map, ..
                                }),
                            source_path,
                        } => move_bytecode_viewer::start_viewer_in_memory(
                            module.clone(),
                            source_map.clone(),
                            source_path,
                        ),
                        _ => anyhow::bail!("Interactive disassembler not supported for scripts"),
                    }
                } else {
                    println!("{}", Disassembler::from_unit(&unit.unit).disassemble()?);
                }
            }
        }
        Ok(())
    }
}