1 files changed, 225 insertions, 0 deletions
diff --git a/third_party/rust/object/tests/round_trip/comdat.rs b/third_party/rust/object/tests/round_trip/comdat.rs
new file mode 100644
index 0000000000..7b697a04c0
--- /dev/null
+++ b/third_party/rust/object/tests/round_trip/comdat.rs
@@ -0,0 +1,225 @@
+#![cfg(all(feature = "read", feature = "write"))]
+
+use object::pe;
+use object::read::{Object, ObjectComdat, ObjectSection, ObjectSymbol};
+use object::{read, write};
+use object::{
+    Architecture, BinaryFormat, ComdatKind, Endianness, SectionKind, SymbolFlags, SymbolKind,
+    SymbolScope,
+};
+
+#[test]
+fn coff_x86_64_comdat() {
+    let mut object =
+        write::Object::new(BinaryFormat::Coff, Architecture::X86_64, Endianness::Little);
+
+    let (section1, offset) =
+        object.add_subsection(write::StandardSection::Text, b"s1", &[0, 1, 2, 3], 4);
+    object.section_symbol(section1);
+    let (section2, _) =
+        object.add_subsection(write::StandardSection::Data, b"s1", &[0, 1, 2, 3], 4);
+    object.section_symbol(section2);
+
+    let symbol = object.add_symbol(write::Symbol {
+        name: b"s1".to_vec(),
+        value: offset,
+        size: 4,
+        kind: SymbolKind::Data,
+        scope: SymbolScope::Linkage,
+        weak: false,
+        section: write::SymbolSection::Section(section1),
+        flags: SymbolFlags::None,
+    });
+
+    object.add_comdat(write::Comdat {
+        kind: ComdatKind::NoDuplicates,
+        symbol,
+        sections: vec![section1, section2],
+    });
+
+    let bytes = object.write().unwrap();
+
+    //std::fs::write(&"comdat.o", &bytes).unwrap();
+
+    let object = read::File::parse(&*bytes).unwrap();
+    assert_eq!(object.format(), BinaryFormat::Coff);
+    assert_eq!(object.architecture(), Architecture::X86_64);
+
+    let mut sections = object.sections();
+
+    let section1 = sections.next().unwrap();
+    println!("{:?}", section1);
+    let section1_index = section1.index();
+    assert_eq!(section1.name(), Ok(".text$s1"));
+    assert_eq!(section1.kind(), SectionKind::Text);
+    assert_eq!(section1.address(), 0);
+    assert_eq!(section1.size(), 4);
+
+    let section2 = sections.next().unwrap();
+    println!("{:?}", section2);
+    let section2_index = section2.index();
+    assert_eq!(section2.name(), Ok(".data$s1"));
+    assert_eq!(section2.kind(), SectionKind::Data);
+    assert_eq!(section2.address(), 0);
+    assert_eq!(section2.size(), 4);
+
+    let mut symbols = object.symbols();
+
+    let symbol = symbols.next().unwrap();
+    println!("{:?}", symbol);
+    assert_eq!(symbol.name(), Ok(".text$s1"));
+    assert_eq!(symbol.kind(), SymbolKind::Section);
+    assert_eq!(
+        symbol.section(),
+        read::SymbolSection::Section(section1.index())
+    );
+    assert_eq!(
+        symbol.flags(),
+        SymbolFlags::CoffSection {
+            selection: pe::IMAGE_COMDAT_SELECT_NODUPLICATES,
+            associative_section: None
+        }
+    );
+
+    let symbol = symbols.next().unwrap();
+    println!("{:?}", symbol);
+    assert_eq!(symbol.name(), Ok(".data$s1"));
+    assert_eq!(symbol.kind(), SymbolKind::Section);
+    assert_eq!(
+        symbol.section(),
+        read::SymbolSection::Section(section2.index())
+    );
+    assert_eq!(
+        symbol.flags(),
+        SymbolFlags::CoffSection {
+            selection: pe::IMAGE_COMDAT_SELECT_ASSOCIATIVE,
+            associative_section: Some(section1_index)
+        }
+    );
+
+    let symbol = symbols.next().unwrap();
+    let symbol_index = symbol.index();
+    println!("{:?}", symbol);
+    assert_eq!(symbol.name(), Ok("s1"));
+    assert_eq!(symbol.kind(), SymbolKind::Data);
+    assert_eq!(
+        symbol.section(),
+        read::SymbolSection::Section(section1.index())
+    );
+    assert_eq!(symbol.scope(), SymbolScope::Linkage);
+    assert_eq!(symbol.is_weak(), false);
+    assert_eq!(symbol.is_undefined(), false);
+    assert_eq!(symbol.address(), 0);
+
+    let symbol = symbols.next();
+    assert!(symbol.is_none(), "unexpected symbol {:?}", symbol);
+
+    let mut comdats = object.comdats();
+
+    let comdat = comdats.next().unwrap();
+    println!("{:?}", comdat);
+    assert_eq!(comdat.kind(), ComdatKind::NoDuplicates);
+    assert_eq!(comdat.symbol(), symbol_index);
+
+    let mut comdat_sections = comdat.sections();
+    assert_eq!(comdat_sections.next(), Some(section1_index));
+    assert_eq!(comdat_sections.next(), Some(section2_index));
+    assert_eq!(comdat_sections.next(), None);
+}
+
+#[test]
+fn elf_x86_64_comdat() {
+    let mut object =
+        write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
+
+    let (section1, offset) =
+        object.add_subsection(write::StandardSection::Text, b"s1", &[0, 1, 2, 3], 4);
+    let (section2, _) =
+        object.add_subsection(write::StandardSection::Data, b"s1", &[0, 1, 2, 3], 4);
+
+    let symbol = object.add_symbol(write::Symbol {
+        name: b"s1".to_vec(),
+        value: offset,
+        size: 4,
+        kind: SymbolKind::Data,
+        scope: SymbolScope::Linkage,
+        weak: false,
+        section: write::SymbolSection::Section(section1),
+        flags: SymbolFlags::None,
+    });
+
+    object.add_comdat(write::Comdat {
+        kind: ComdatKind::Any,
+        symbol,
+        sections: vec![section1, section2],
+    });
+
+    let bytes = object.write().unwrap();
+
+    //std::fs::write(&"comdat.o", &bytes).unwrap();
+
+    let object = read::File::parse(&*bytes).unwrap();
+    assert_eq!(object.format(), BinaryFormat::Elf);
+    assert_eq!(object.architecture(), Architecture::X86_64);
+
+    let mut sections = object.sections();
+
+    let section = sections.next().unwrap();
+    println!("{:?}", section);
+    assert_eq!(section.name(), Ok(""));
+
+    let section = sections.next().unwrap();
+    println!("{:?}", section);
+    assert_eq!(section.name(), Ok(".group"));
+
+    let section1 = sections.next().unwrap();
+    println!("{:?}", section1);
+    let section1_index = section1.index();
+    assert_eq!(section1.name(), Ok(".text.s1"));
+    assert_eq!(section1.kind(), SectionKind::Text);
+    assert_eq!(section1.address(), 0);
+    assert_eq!(section1.size(), 4);
+
+    let section2 = sections.next().unwrap();
+    println!("{:?}", section2);
+    let section2_index = section2.index();
+    assert_eq!(section2.name(), Ok(".data.s1"));
+    assert_eq!(section2.kind(), SectionKind::Data);
+    assert_eq!(section2.address(), 0);
+    assert_eq!(section2.size(), 4);
+
+    let mut symbols = object.symbols();
+
+    let symbol = symbols.next().unwrap();
+    println!("{:?}", symbol);
+    assert_eq!(symbol.name(), Ok(""));
+
+    let symbol = symbols.next().unwrap();
+    let symbol_index = symbol.index();
+    println!("{:?}", symbol);
+    assert_eq!(symbol.name(), Ok("s1"));
+    assert_eq!(symbol.kind(), SymbolKind::Data);
+    assert_eq!(
+        symbol.section(),
+        read::SymbolSection::Section(section1.index())
+    );
+    assert_eq!(symbol.scope(), SymbolScope::Linkage);
+    assert_eq!(symbol.is_weak(), false);
+    assert_eq!(symbol.is_undefined(), false);
+    assert_eq!(symbol.address(), 0);
+
+    let symbol = symbols.next();
+    assert!(symbol.is_none(), "unexpected symbol {:?}", symbol);
+
+    let mut comdats = object.comdats();
+
+    let comdat = comdats.next().unwrap();
+    println!("{:?}", comdat);
+    assert_eq!(comdat.kind(), ComdatKind::Any);
+    assert_eq!(comdat.symbol(), symbol_index);
+
+    let mut comdat_sections = comdat.sections();
+    assert_eq!(comdat_sections.next(), Some(section1_index));
+    assert_eq!(comdat_sections.next(), Some(section2_index));
+    assert_eq!(comdat_sections.next(), None);
+}