rld/src/linker.rs

use std::{fmt::Display, path::PathBuf};

use crate::{
    common::{Loadable, Output, Relocatable},
    error::Error,
};

pub struct Linker<R, O>
where
    R: Relocatable,
    O: Output<R>,
{
    relocatables: Vec<R>,
    loadable: Loadable,
    output: O,
}

impl<R, O> Linker<R, O>
where
    R: Relocatable,
    O: Output<R>,
{
    pub fn new(output: O) -> Self {
        Self {
            relocatables: Vec::new(),
            loadable: Loadable::default(),
            output,
        }
    }

    pub fn add_relocatable(&mut self, origin: PathBuf) -> Result<(), Error> {
        let r = R::new(origin, self.relocatables.len())?;

        // TODO: parallelize?
        for section in r.sections() {
            self.loadable.process_section(section?)?;
        }

        self.relocatables.push(r);

        Ok(())
    }

    pub fn link(mut self) -> Result<PathBuf, Error> {
        self.process_symbols()?;

        self.loadable.set_start_offset(4096); // TODO: get from .start symbol location
        self.output.finalize(&self.relocatables, &self.loadable)
    }

    fn process_symbols(&mut self) -> Result<(), Error> {
        for r in &self.relocatables {
            for (name, symbol) in r.symbols() {
                self.loadable.process_symbol(name, symbol)?;
            }
        }

        Ok(())
    }

    // pub fn object_path(origin: &Path) -> Result<PathBuf, Error> {
    //     let mut dest = std::fs::canonicalize(origin)?;
    //     if !dest.pop() {
    //         let err = std::io::Error::new(ErrorKind::Other, "Destination path invalid");
    //         Err(Error::IOError(Box::new(err)))
    //     } else {
    //         dest.push("rld.out");
    //         Ok(dest)
    //     }
    // }
}

impl<R, O> Display for Linker<R, O>
where
    R: Relocatable,
    O: Output<R>,
{
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        writeln!(f, "===Relocatables===")?;

        for r in self.relocatables.iter() {
            write!(f, "{}", r)?;
        }

        Ok(())
    }
}