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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173

use crate::information_element::SbdHeader;
use crate::mo::session_status::SessionStatus;
use crate::Result;
use chrono::{DateTime, Utc};
#[cfg(feature = "serde-derive")]
use serde::{Deserialize, Serialize};
use std::io::{Read, Write};

/// A mobile-originated header.
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
#[cfg_attr(feature = "serde-derive", derive(Serialize, Deserialize))]
pub struct Header {
    /// The Iridium Gateway id for this message.
    pub auto_id: u32,
    /// The device id.
    #[cfg_attr(feature = "serde-derive",
    serde(serialize_with = "as_str", deserialize_with = "str_to_imei"))]
    pub imei: [u8; 15],
    /// The session status.
    pub session_status: SessionStatus,
    /// The mobile originated message sequence number.
    pub momsn: u16,
    /// The mobile terminated message sequence number.
    pub mtmsn: u16,
    /// The time of iridium session.
    pub time_of_session: DateTime<Utc>,
}

impl SbdHeader for Header {
    fn write_to(&self, write: &mut dyn Write) -> Result<()> {
        use crate::Error;
        use byteorder::{BigEndian, WriteBytesExt};

        write.write_u8(1)?;
        write.write_u16::<BigEndian>(28)?;
        write.write_u32::<BigEndian>(self.auto_id)?;
        write.write_all(&self.imei)?;
        write.write_u8(self.session_status as u8)?;
        write.write_u16::<BigEndian>(self.momsn)?;
        write.write_u16::<BigEndian>(self.mtmsn)?;
        let timestamp = self.time_of_session.timestamp();
        if timestamp < 0 {
            return Err(Error::NegativeTimestamp(timestamp));
        } else {
            write.write_u32::<BigEndian>(timestamp as u32)?;
        };
        Ok(())
    }

    fn imei(&self) -> &str {
        use std::str;
        str::from_utf8(&self.imei).expect("IMEI numbers are specified to be ascii number")
    }
    fn len(&self) -> usize {
        31
    }
    fn as_mo(&self) -> Option<&Header> {
        Some(&self)
    }
}

impl Header {
    pub fn read_from(read: &mut dyn Read) -> Result<Self> {
        use crate::Error;
        use byteorder::{BigEndian, ReadBytesExt};
        use chrono::TimeZone;

        let auto_id = read.read_u32::<BigEndian>().map_err(Error::Io)?;
        let mut imei = [0; 15];
        read.read_exact(&mut imei).map_err(Error::Io)?;
        let session_status = SessionStatus::new(read.read_u8().map_err(Error::Io)?)?;
        let momsn = read.read_u16::<BigEndian>().map_err(Error::Io)?;
        let mtmsn = read.read_u16::<BigEndian>().map_err(Error::Io)?;
        let time_of_session = read
            .read_u32::<BigEndian>()
            .map_err(Error::from)
            .map(|n| Utc.timestamp(i64::from(n), 0))?;
        Ok(Header {
            auto_id,
            imei,
            session_status,
            momsn,
            mtmsn,
            time_of_session,
        })
    }
    /// Returns this message's session status.
    ///
    /// # Examples
    ///
    /// ```
    /// use sbd_lib::Message;
    /// let message = Message::from_path("data/0-mo.sbd").unwrap();
    /// if let Some(header) =  message.header().as_mo() {
    ///     let session_status = header.session_status();
    /// }
    /// ```
    pub fn session_status(&self) -> SessionStatus {
        self.session_status
    }

    /// Returns this message's mobile originated message sequence number.
    ///
    /// # Examples
    ///
    /// ```
    /// use sbd_lib::Message;
    /// let message = Message::from_path("data/0-mo.sbd").unwrap();
    /// if let Some(header) =  message.header().as_mo() {
    ///     let momsn = header.momsn();
    /// }
    /// ```
    pub fn momsn(&self) -> u16 {
        self.momsn
    }

    /// Returns this message's mobile terminated message sequence number.
    ///
    /// # Examples
    ///
    /// ```
    /// use sbd_lib::Message;
    /// let message = Message::from_path("data/0-mo.sbd").unwrap();
    /// if let Some(header) = message.header().as_mo() {
    ///     let mtmsn = header.mtmsn();
    /// }
    /// ```
    pub fn mtmsn(&self) -> u16 {
        self.mtmsn
    }

    /// Returns this message's time of session.
    ///
    /// # Examples
    ///
    /// ```
    /// use sbd_lib::Message;
    /// let message = Message::from_path("data/0-mo.sbd").unwrap();
    /// if let Some(header) = message.header().as_mo() {
    ///     let time_of_session = header.time_of_session();
    /// }
    /// ```
    pub fn time_of_session(&self) -> DateTime<Utc> {
        self.time_of_session
    }
}

#[cfg(feature = "serde-derive")]
fn as_str<S>(imei: &[u8], serializer: S) -> std::result::Result<S::Ok, S::Error>
    where S: serde::Serializer
{
    use std::str;
    use serde::ser;
    match str::from_utf8(imei) {
        Ok(s) => serializer.serialize_str(s),
        _=> Err(ser::Error::custom("imei contains invalid UTF-8 characters")),
    }
}

#[cfg(feature = "serde-derive")]
pub fn str_to_imei<'de, D>(deserializer: D) -> std::result::Result<[u8; 15], D::Error>
    where D: serde::Deserializer<'de>
{
    use serde::de;
    let mut value = [0; 15];
    let data = <&'de [u8]>::deserialize(deserializer)?;
    if data.len() == value.len() {
        value.copy_from_slice(data);
        Ok(value)
    } else {
        Err(de::Error::custom("imei wrong len"))
    }
}