#[derive(PartialEq, Eq, Debug, Clone, Hash)]
pub enum Record {
Data {
offset: u16,
value: Vec<u8>,
},
EndOfFile,
ExtendedSegmentAddress(u16),
StartSegmentAddress {
cs: u16,
ip: u16,
},
ExtendedLinearAddress(u16),
StartLinearAddress(u32),
}
impl Record {
pub fn record_type(&self) -> u8 {
match self {
Record::Data { .. } => types::DATA,
Record::EndOfFile => types::END_OF_FILE,
Record::ExtendedSegmentAddress(..) => types::EXTENDED_SEGMENT_ADDRESS,
Record::StartSegmentAddress { .. } => types::START_SEGMENT_ADDRESS,
Record::ExtendedLinearAddress(..) => types::EXTENDED_LINEAR_ADDRESS,
Record::StartLinearAddress(..) => types::START_LINEAR_ADDRESS,
}
}
}
pub mod types {
pub const DATA: u8 = 0x00;
pub const END_OF_FILE: u8 = 0x01;
pub const EXTENDED_SEGMENT_ADDRESS: u8 = 0x02;
pub const START_SEGMENT_ADDRESS: u8 = 0x03;
pub const EXTENDED_LINEAR_ADDRESS: u8 = 0x04;
pub const START_LINEAR_ADDRESS: u8 = 0x05;
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_record_type() {
let data_record = Record::Data {
offset: 0u16,
value: Vec::new(),
};
assert_eq!(data_record.record_type(), 0x00);
let eof_record = Record::EndOfFile;
assert_eq!(eof_record.record_type(), 0x01);
let extended_segment_address_record = Record::ExtendedSegmentAddress(0);
assert_eq!(extended_segment_address_record.record_type(), 0x02);
let start_segment_address_record = Record::StartSegmentAddress { cs: 0, ip: 0 };
assert_eq!(start_segment_address_record.record_type(), 0x03);
let extended_linear_address_record = Record::ExtendedLinearAddress(0);
assert_eq!(extended_linear_address_record.record_type(), 0x04);
let start_linear_address_record = Record::StartLinearAddress(0);
assert_eq!(start_linear_address_record.record_type(), 0x05);
}
}