sdmmc_core/command/class/class2/cmd18/
arg.rs

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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
use crate::lib_bitfield;
use crate::result::{Error, Result};

lib_bitfield! {
    /// Argumentument for CMD18.
    pub Argument(u32): u32 {
        /// Reads data blocks starting at `data_address` until a `STOP_TRANSMISSION` command is sent.
        pub data_address: 31, 0;
    }
}

impl Argument {
    /// Represents the byte length of the [Argument].
    pub const LEN: usize = 4;
    /// Represents the default value of the [Argument].
    pub const DEFAULT: u32 = 0;

    /// Creates a new [Argument].
    pub const fn new() -> Self {
        Self(Self::DEFAULT)
    }

    /// Converts a [`u32`] into an [Argument].
    pub const fn from_bits(val: u32) -> Self {
        Self(val)
    }

    /// Converts a [`u32`] into an [Argument].
    pub const fn try_from_bits(val: u32) -> Result<Self> {
        Ok(Self::from_bits(val))
    }

    /// Converts the [Argument] into a byte array.
    pub const fn bytes(&self) -> [u8; Self::LEN] {
        self.0.to_be_bytes()
    }

    /// Attempts to convert a byte slice into an [Argument].
    pub const fn try_from_bytes(val: &[u8]) -> Result<Self> {
        match val.len() {
            len if len < Self::LEN => Err(Error::invalid_length(len, Self::LEN)),
            _ => Ok(Self::from_bits(u32::from_be_bytes([
                val[0], val[1], val[2], val[3],
            ]))),
        }
    }
}

impl Default for Argument {
    fn default() -> Self {
        Self::new()
    }
}

impl From<u32> for Argument {
    fn from(val: u32) -> Self {
        Self::from_bits(val)
    }
}

impl From<Argument> for u32 {
    fn from(val: Argument) -> Self {
        val.bits()
    }
}

impl From<Argument> for [u8; Argument::LEN] {
    fn from(val: Argument) -> Self {
        val.bytes()
    }
}

impl TryFrom<&[u8]> for Argument {
    type Error = Error;

    fn try_from(val: &[u8]) -> Result<Self> {
        Self::try_from_bytes(val)
    }
}

impl<const N: usize> TryFrom<&[u8; N]> for Argument {
    type Error = Error;

    fn try_from(val: &[u8; N]) -> Result<Self> {
        Self::try_from_bytes(val.as_ref())
    }
}

impl<const N: usize> TryFrom<[u8; N]> for Argument {
    type Error = Error;

    fn try_from(val: [u8; N]) -> Result<Self> {
        Self::try_from_bytes(val.as_ref())
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_fields() {
        (1..=u32::BITS)
            .map(|r| ((1u64 << r) - 1) as u32)
            .for_each(|data_address| {
                let raw = data_address;
                let raw_bytes = raw.to_be_bytes();

                let mut exp_arg = Argument(raw);

                assert_eq!(Argument::try_from_bits(raw), Ok(exp_arg));
                assert_eq!(Argument::from_bits(raw), exp_arg);
                assert_eq!(Argument::from(raw), exp_arg);

                assert_eq!(Argument::try_from_bytes(&raw_bytes), Ok(exp_arg));
                assert_eq!(Argument::try_from(raw_bytes), Ok(exp_arg));
                assert_eq!(Argument::try_from(&raw_bytes), Ok(exp_arg));

                assert_eq!(exp_arg.bytes(), raw_bytes);
                assert_eq!(<[u8; Argument::LEN]>::from(exp_arg), raw_bytes);

                assert_eq!(exp_arg.data_address(), data_address);

                exp_arg.set_data_address(0);
                assert_eq!(exp_arg.data_address(), 0);

                exp_arg.set_data_address(data_address);
                assert_eq!(exp_arg.data_address(), data_address);
            });
    }
}