Swift SIL: add some APIs

Author: eeckstein

SwiftCompilerSources/Sources/Basic/Utils.swift

@@ -64,11 +64,21 @@ public struct StringRef : CustomStringConvertible, NoReflectionChildren {
 
   public var string: String { _bridged.string }
   public var description: String { string }
-  
+
+  public var count: Int {
+    Int(_bridged.__bytes_endUnsafe() - _bridged.__bytes_beginUnsafe())
+  }
+
+  public subscript(index: Int) -> UInt8 {
+    let buffer = UnsafeBufferPointer<UInt8>(start: _bridged.__bytes_beginUnsafe(),
+                                            count: count)
+    return buffer[index]
+  }
+
   public static func ==(lhs: StringRef, rhs: StaticString) -> Bool {
     let lhsBuffer = UnsafeBufferPointer<UInt8>(
       start: lhs._bridged.__bytes_beginUnsafe(),
-      count: Int(lhs._bridged.__bytes_endUnsafe() - lhs._bridged.__bytes_beginUnsafe()))
+      count: lhs.count)
     return rhs.withUTF8Buffer { (rhsBuffer: UnsafeBufferPointer<UInt8>) in
       if lhsBuffer.count != rhsBuffer.count { return false }
       return lhsBuffer.elementsEqual(rhsBuffer, by: ==)

SwiftCompilerSources/Sources/SIL/ApplySite.swift

@@ -115,6 +115,13 @@ extension ApplySite {
     }
     return nil
   }
+
+  public func hasSemanticsAttribute(_ attr: StaticString) -> Bool {
+    if let callee = referencedFunction {
+      return callee.hasSemanticsAttribute(attr)
+    }
+    return false
+  }
 }
 
 public protocol FullApplySite : ApplySite {

SwiftCompilerSources/Sources/SIL/Function.swift

@@ -126,6 +126,8 @@ final public class Function : CustomStringConvertible, HasShortDescription, Hash
 
   public var isTransparent: Bool { bridged.isTransparent() }
 
+  public var isAsync: Bool { bridged.isAsync() }
+
   /// True if this is a `[global_init]` function.
   ///
   /// Such a function is typically a global addressor which calls the global's
@@ -222,6 +224,9 @@ final public class Function : CustomStringConvertible, HasShortDescription, Hash
     }
   }
 
+  public var isSerialized: Bool { bridged.isSerialized() }
+  public var hasValidLinkageForFragileRef: Bool { bridged.hasValidLinkageForFragileRef() }
+
   /// True, if the function runs with a swift 5.1 runtime.
   /// Note that this is function specific, because inlinable functions are de-serialized
   /// in a client module, which might be compiled with a different deployment target.

SwiftCompilerSources/Sources/SIL/Location.swift

@@ -37,4 +37,8 @@ public struct Location: Equatable, CustomStringConvertible {
   public func hasSameSourceLocation(as other: Location) -> Bool {
     bridged.hasSameSourceLocation(other.bridged)
   }
+
+  public static var artificialUnreachableLocation: Location {
+    Location(bridged: swift.SILDebugLocation.getArtificialUnreachableLocation())
+  }
 }

SwiftCompilerSources/Sources/SIL/Operand.swift

@@ -107,6 +107,19 @@ public struct UseList : CollectionLikeSequence {
     return nil
   }
 
+  public func getSingleUser<I: Instruction>(ofType: I.Type) -> I? {
+    var result: I? = nil
+    for use in self {
+      if let user = use.instruction as? I {
+        if result != nil {
+          return nil
+        }
+        result = user
+      }
+    }
+    return result
+  }
+
   public var isSingleUse: Bool { singleUse != nil }
 
   public func makeIterator() -> Iterator {

SwiftCompilerSources/Sources/SIL/Type.swift

@@ -19,6 +19,9 @@ public struct Type : CustomStringConvertible, NoReflectionChildren {
   public var isAddress: Bool { bridged.isAddress() }
   public var isObject: Bool { !isAddress }
 
+  public var addressType: Type { bridged.getAddressType().type }
+  public var objectType: Type { bridged.getObjectType().type }
+
   public func isTrivial(in function: Function) -> Bool {
     return bridged.isTrivial(function.bridged.getFunction())
   }
@@ -52,6 +55,19 @@ public struct Type : CustomStringConvertible, NoReflectionChildren {
 
   public var isOrContainsObjectiveCClass: Bool { bridged.isOrContainsObjectiveCClass() }
 
+  public var isBuiltinInteger: Bool { bridged.isBuiltinInteger() }
+  public var isBuiltinFloat: Bool { bridged.isBuiltinFloat() }
+  public var isBuiltinVector: Bool { bridged.isBuiltinVector() }
+  public var builtinVectorElementType: Type { bridged.getBuiltinVectorElementType().type }
+
+  public func isBuiltinInteger(withFixedWidth width: Int) -> Bool {
+    bridged.isBuiltinFixedWidthInteger(UInt32(width))
+  }
+
+  public func isExactSuperclass(of type: Type) -> Bool {
+    bridged.isExactSuperclassOf(type.bridged)
+  }
+
   public var tupleElements: TupleElementArray { TupleElementArray(type: self) }
 
   public func getNominalFields(in function: Function) -> NominalFieldsArray {
@@ -167,6 +183,8 @@ extension swift.SILType {
 public struct NominalTypeDecl : Equatable {
   let bridged: BridgedNominalTypeDecl
 
+  public var name: StringRef { StringRef(bridged: bridged.getName()) }
+
   public static func ==(lhs: NominalTypeDecl, rhs: NominalTypeDecl) -> Bool {
     lhs.bridged.decl == rhs.bridged.decl
   }

include/swift/SIL/SILBridging.h

@@ -241,6 +241,10 @@ struct BridgedFunction {
     return getFunction()->isTransparent() == swift::IsTransparent;
   }
 
+  bool isAsync() const {
+    return getFunction()->isAsync();
+  }
+
   bool isGlobalInitFunction() const {
     return getFunction()->isGlobalInit();
   }
@@ -271,6 +275,14 @@ struct BridgedFunction {
     return (InlineStrategy)getFunction()->getInlineStrategy();
   }
 
+  bool isSerialized() const {
+    return getFunction()->isSerialized();
+  }
+
+  bool hasValidLinkageForFragileRef() const {
+    return getFunction()->hasValidLinkageForFragileRef();
+  }
+
   bool needsStackProtection() const {
     return getFunction()->needsStackProtection();
   }
@@ -1208,6 +1220,11 @@ struct BridgedBuilder{
 
 struct BridgedNominalTypeDecl {
   swift::NominalTypeDecl * _Nonnull decl;
+
+  SWIFT_IMPORT_UNSAFE
+  llvm::StringRef getName() const {
+    return decl->getName().str();
+  }
 };
 
 // Passmanager and Context

include/swift/SIL/SILLocation.h

@@ -743,6 +743,10 @@ class SILDebugLocation {
     return getLocation().hasSameSourceLocation(rhs.getLocation()) &&
            getScope() == rhs.getScope();
   }
+
+  static SILDebugLocation getArtificialUnreachableLocation() {
+    return SILDebugLocation(ArtificialUnreachableLocation(), nullptr);
+  }
 };
 
 } // end swift namespace

include/swift/SIL/SILType.h

@@ -267,6 +267,29 @@ class SILType {
     });
   }
 
+  bool isBuiltinInteger() const {
+    return is<BuiltinIntegerType>();
+  }
+
+  bool isBuiltinFixedWidthInteger(unsigned width) const {
+    BuiltinIntegerType *bi = getAs<BuiltinIntegerType>();
+    return bi && bi->isFixedWidth(width);
+  }
+
+  bool isBuiltinFloat() const {
+    return is<BuiltinFloatType>();
+  }
+
+  bool isBuiltinVector() const {
+    return is<BuiltinVectorType>();
+  }
+
+  SWIFT_IMPORT_UNSAFE
+  SILType getBuiltinVectorElementType() const {
+    auto vector = castTo<BuiltinVectorType>();
+    return getPrimitiveObjectType(vector.getElementType());
+  }
+
   /// Retrieve the NominalTypeDecl for a type that maps to a Swift
   /// nominal or bound generic nominal type.
   SWIFT_IMPORT_UNSAFE