Add Section retrieval

Author: CodaFi

Sources/LLVM/IRGlobal.swift

@@ -31,4 +31,18 @@ extension IRGlobal {
     get { return StorageClass(llvm: LLVMGetDLLStorageClass(asLLVM())) }
     set { LLVMSetDLLStorageClass(asLLVM(), newValue.llvm) }
   }
+
+  /// Retrieves the section associated with the symbol that will eventually be
+  /// emitted for this global value.
+  ///
+  /// - Note: Global `Alias` values may or may not be resolvable to any
+  ///   particular section given the state of the IR in an arbitrary module. A
+  ///   return value of the empty string indicates a failed section lookup.
+  public var section: String {
+    get {
+      guard let sname = LLVMGetSection(asLLVM()) else { return "" }
+      return String(cString: sname)
+    }
+    set { LLVMSetSection(asLLVM(), newValue) }
+  }
 }

Sources/LLVM/ObjectFile.swift

@@ -56,7 +56,6 @@ public struct Section {
     /// The parent sequence of this section.
     private let sectionIterator: LLVMSectionIteratorRef
 
-
     internal init(fromIterator si: LLVMSectionIteratorRef) {
         self.sectionIterator = si
         self.name = String(cString: LLVMGetSectionName(si))
@@ -72,6 +71,12 @@ public struct Section {
             sectionIterator: self.sectionIterator
         )
     }
+
+    /// Returns whether a symbol matching the given `Symbol` can be found in
+    /// this section.
+    public func contains(symbol: Symbol) -> Bool {
+        return LLVMGetSectionContainsSymbol(self.sectionIterator, symbol.symbolIterator) != 0
+    }
 }
 
 /// A sequence for iterating over the sections in an object file.
@@ -109,10 +114,14 @@ public struct Symbol {
     /// The address of the symbol in the object file.
     public let address: Int
 
+    /// The parent sequence of this symbol.
+    fileprivate let symbolIterator: LLVMSymbolIteratorRef
+
     internal init(fromIterator si: LLVMSymbolIteratorRef) {
         self.name = String(cString: LLVMGetSymbolName(si))
         self.size = Int(LLVMGetSymbolSize(si))
         self.address = Int(LLVMGetSymbolAddress(si))
+      self.symbolIterator = si
     }
 }
 
@@ -188,8 +197,6 @@ public class SymbolSequence: Sequence {
             return Symbol(fromIterator: self.llvm)
         }
     }
-    
-
 
     deinit {
         LLVMDisposeSymbolIterator(llvm)