sallyport 0.6.3

// SPDX-License-Identifier: Apache-2.0

use super::super::types::Argv;
use super::Alloc;
use crate::guest::alloc::{Allocator, Collect, Collector, Output};
use crate::libc::{clockid_t, timespec, SYS_clock_gettime};
use crate::Result;

use core::ffi::c_long;

pub struct ClockGettime<'a> {
    pub clockid: clockid_t,
    pub tp: &'a mut timespec,
}

unsafe impl<'a> Alloc<'a> for ClockGettime<'a> {
    const NUM: c_long = SYS_clock_gettime;

    type Argv = Argv<2>;
    type Ret = ();

    type Staged = Output<'a, timespec, &'a mut timespec>;
    type Committed = Self::Staged;
    type Collected = Result<()>;

    fn stage(self, alloc: &mut impl Allocator) -> Result<(Self::Argv, Self::Staged)> {
        let tp = Output::stage(alloc, self.tp)?;
        Ok((Argv([self.clockid as _, tp.offset()]), tp))
    }

    fn collect(
        tp: Self::Committed,
        ret: Result<Self::Ret>,
        col: &impl Collector,
    ) -> Self::Collected {
        if ret.is_ok() {
            tp.collect(col);
        };
        ret
    }
}