Home | History | Annotate | Line # | Download | only in IR
      1 //===- llvm/InlineAsm.h - Class to represent inline asm strings -*- C++ -*-===//
      2 //
      3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
      4 // See https://llvm.org/LICENSE.txt for license information.
      5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
      6 //
      7 //===----------------------------------------------------------------------===//
      8 //
      9 // This class represents the inline asm strings, which are Value*'s that are
     10 // used as the callee operand of call instructions.  InlineAsm's are uniqued
     11 // like constants, and created via InlineAsm::get(...).
     12 //
     13 //===----------------------------------------------------------------------===//
     14 
     15 #ifndef LLVM_IR_INLINEASM_H
     16 #define LLVM_IR_INLINEASM_H
     17 
     18 #include "llvm/ADT/StringRef.h"
     19 #include "llvm/IR/Value.h"
     20 #include "llvm/Support/ErrorHandling.h"
     21 #include <cassert>
     22 #include <string>
     23 #include <vector>
     24 
     25 namespace llvm {
     26 
     27 class FunctionType;
     28 class PointerType;
     29 template <class ConstantClass> class ConstantUniqueMap;
     30 
     31 class InlineAsm final : public Value {
     32 public:
     33   enum AsmDialect {
     34     AD_ATT,
     35     AD_Intel
     36   };
     37 
     38 private:
     39   friend struct InlineAsmKeyType;
     40   friend class ConstantUniqueMap<InlineAsm>;
     41 
     42   std::string AsmString, Constraints;
     43   FunctionType *FTy;
     44   bool HasSideEffects;
     45   bool IsAlignStack;
     46   AsmDialect Dialect;
     47   bool CanThrow;
     48 
     49   InlineAsm(FunctionType *Ty, const std::string &AsmString,
     50             const std::string &Constraints, bool hasSideEffects,
     51             bool isAlignStack, AsmDialect asmDialect, bool canThrow);
     52 
     53   /// When the ConstantUniqueMap merges two types and makes two InlineAsms
     54   /// identical, it destroys one of them with this method.
     55   void destroyConstant();
     56 
     57 public:
     58   InlineAsm(const InlineAsm &) = delete;
     59   InlineAsm &operator=(const InlineAsm &) = delete;
     60 
     61   /// InlineAsm::get - Return the specified uniqued inline asm string.
     62   ///
     63   static InlineAsm *get(FunctionType *Ty, StringRef AsmString,
     64                         StringRef Constraints, bool hasSideEffects,
     65                         bool isAlignStack = false,
     66                         AsmDialect asmDialect = AD_ATT, bool canThrow = false);
     67 
     68   bool hasSideEffects() const { return HasSideEffects; }
     69   bool isAlignStack() const { return IsAlignStack; }
     70   AsmDialect getDialect() const { return Dialect; }
     71   bool canThrow() const { return CanThrow; }
     72 
     73   /// getType - InlineAsm's are always pointers.
     74   ///
     75   PointerType *getType() const {
     76     return reinterpret_cast<PointerType*>(Value::getType());
     77   }
     78 
     79   /// getFunctionType - InlineAsm's are always pointers to functions.
     80   ///
     81   FunctionType *getFunctionType() const;
     82 
     83   const std::string &getAsmString() const { return AsmString; }
     84   const std::string &getConstraintString() const { return Constraints; }
     85 
     86   /// Verify - This static method can be used by the parser to check to see if
     87   /// the specified constraint string is legal for the type.  This returns true
     88   /// if legal, false if not.
     89   ///
     90   static bool Verify(FunctionType *Ty, StringRef Constraints);
     91 
     92   // Constraint String Parsing
     93   enum ConstraintPrefix {
     94     isInput,            // 'x'
     95     isOutput,           // '=x'
     96     isClobber           // '~x'
     97   };
     98 
     99   using ConstraintCodeVector = std::vector<std::string>;
    100 
    101   struct SubConstraintInfo {
    102     /// MatchingInput - If this is not -1, this is an output constraint where an
    103     /// input constraint is required to match it (e.g. "0").  The value is the
    104     /// constraint number that matches this one (for example, if this is
    105     /// constraint #0 and constraint #4 has the value "0", this will be 4).
    106     int MatchingInput = -1;
    107 
    108     /// Code - The constraint code, either the register name (in braces) or the
    109     /// constraint letter/number.
    110     ConstraintCodeVector Codes;
    111 
    112     /// Default constructor.
    113     SubConstraintInfo() = default;
    114   };
    115 
    116   using SubConstraintInfoVector = std::vector<SubConstraintInfo>;
    117   struct ConstraintInfo;
    118   using ConstraintInfoVector = std::vector<ConstraintInfo>;
    119 
    120   struct ConstraintInfo {
    121     /// Type - The basic type of the constraint: input/output/clobber
    122     ///
    123     ConstraintPrefix Type = isInput;
    124 
    125     /// isEarlyClobber - "&": output operand writes result before inputs are all
    126     /// read.  This is only ever set for an output operand.
    127     bool isEarlyClobber = false;
    128 
    129     /// MatchingInput - If this is not -1, this is an output constraint where an
    130     /// input constraint is required to match it (e.g. "0").  The value is the
    131     /// constraint number that matches this one (for example, if this is
    132     /// constraint #0 and constraint #4 has the value "0", this will be 4).
    133     int MatchingInput = -1;
    134 
    135     /// hasMatchingInput - Return true if this is an output constraint that has
    136     /// a matching input constraint.
    137     bool hasMatchingInput() const { return MatchingInput != -1; }
    138 
    139     /// isCommutative - This is set to true for a constraint that is commutative
    140     /// with the next operand.
    141     bool isCommutative = false;
    142 
    143     /// isIndirect - True if this operand is an indirect operand.  This means
    144     /// that the address of the source or destination is present in the call
    145     /// instruction, instead of it being returned or passed in explicitly.  This
    146     /// is represented with a '*' in the asm string.
    147     bool isIndirect = false;
    148 
    149     /// Code - The constraint code, either the register name (in braces) or the
    150     /// constraint letter/number.
    151     ConstraintCodeVector Codes;
    152 
    153     /// isMultipleAlternative - '|': has multiple-alternative constraints.
    154     bool isMultipleAlternative = false;
    155 
    156     /// multipleAlternatives - If there are multiple alternative constraints,
    157     /// this array will contain them.  Otherwise it will be empty.
    158     SubConstraintInfoVector multipleAlternatives;
    159 
    160     /// The currently selected alternative constraint index.
    161     unsigned currentAlternativeIndex = 0;
    162 
    163     /// Default constructor.
    164     ConstraintInfo() = default;
    165 
    166     /// Parse - Analyze the specified string (e.g. "=*&{eax}") and fill in the
    167     /// fields in this structure.  If the constraint string is not understood,
    168     /// return true, otherwise return false.
    169     bool Parse(StringRef Str, ConstraintInfoVector &ConstraintsSoFar);
    170 
    171     /// selectAlternative - Point this constraint to the alternative constraint
    172     /// indicated by the index.
    173     void selectAlternative(unsigned index);
    174   };
    175 
    176   /// ParseConstraints - Split up the constraint string into the specific
    177   /// constraints and their prefixes.  If this returns an empty vector, and if
    178   /// the constraint string itself isn't empty, there was an error parsing.
    179   static ConstraintInfoVector ParseConstraints(StringRef ConstraintString);
    180 
    181   /// ParseConstraints - Parse the constraints of this inlineasm object,
    182   /// returning them the same way that ParseConstraints(str) does.
    183   ConstraintInfoVector ParseConstraints() const {
    184     return ParseConstraints(Constraints);
    185   }
    186 
    187   // Methods for support type inquiry through isa, cast, and dyn_cast:
    188   static bool classof(const Value *V) {
    189     return V->getValueID() == Value::InlineAsmVal;
    190   }
    191 
    192   // These are helper methods for dealing with flags in the INLINEASM SDNode
    193   // in the backend.
    194   //
    195   // The encoding of the flag word is currently:
    196   //   Bits 2-0 - A Kind_* value indicating the kind of the operand.
    197   //   Bits 15-3 - The number of SDNode operands associated with this inline
    198   //               assembly operand.
    199   //   If bit 31 is set:
    200   //     Bit 30-16 - The operand number that this operand must match.
    201   //                 When bits 2-0 are Kind_Mem, the Constraint_* value must be
    202   //                 obtained from the flags for this operand number.
    203   //   Else if bits 2-0 are Kind_Mem:
    204   //     Bit 30-16 - A Constraint_* value indicating the original constraint
    205   //                 code.
    206   //   Else:
    207   //     Bit 30-16 - The register class ID to use for the operand.
    208 
    209   enum : uint32_t {
    210     // Fixed operands on an INLINEASM SDNode.
    211     Op_InputChain = 0,
    212     Op_AsmString = 1,
    213     Op_MDNode = 2,
    214     Op_ExtraInfo = 3,    // HasSideEffects, IsAlignStack, AsmDialect.
    215     Op_FirstOperand = 4,
    216 
    217     // Fixed operands on an INLINEASM MachineInstr.
    218     MIOp_AsmString = 0,
    219     MIOp_ExtraInfo = 1,    // HasSideEffects, IsAlignStack, AsmDialect.
    220     MIOp_FirstOperand = 2,
    221 
    222     // Interpretation of the MIOp_ExtraInfo bit field.
    223     Extra_HasSideEffects = 1,
    224     Extra_IsAlignStack = 2,
    225     Extra_AsmDialect = 4,
    226     Extra_MayLoad = 8,
    227     Extra_MayStore = 16,
    228     Extra_IsConvergent = 32,
    229 
    230     // Inline asm operands map to multiple SDNode / MachineInstr operands.
    231     // The first operand is an immediate describing the asm operand, the low
    232     // bits is the kind:
    233     Kind_RegUse = 1,             // Input register, "r".
    234     Kind_RegDef = 2,             // Output register, "=r".
    235     Kind_RegDefEarlyClobber = 3, // Early-clobber output register, "=&r".
    236     Kind_Clobber = 4,            // Clobbered register, "~r".
    237     Kind_Imm = 5,                // Immediate.
    238     Kind_Mem = 6,                // Memory operand, "m".
    239 
    240     // Memory constraint codes.
    241     // These could be tablegenerated but there's little need to do that since
    242     // there's plenty of space in the encoding to support the union of all
    243     // constraint codes for all targets.
    244     Constraint_Unknown = 0,
    245     Constraint_es,
    246     Constraint_i,
    247     Constraint_m,
    248     Constraint_o,
    249     Constraint_v,
    250     Constraint_A,
    251     Constraint_Q,
    252     Constraint_R,
    253     Constraint_S,
    254     Constraint_T,
    255     Constraint_Um,
    256     Constraint_Un,
    257     Constraint_Uq,
    258     Constraint_Us,
    259     Constraint_Ut,
    260     Constraint_Uv,
    261     Constraint_Uy,
    262     Constraint_X,
    263     Constraint_Z,
    264     Constraint_ZC,
    265     Constraint_Zy,
    266     Constraints_Max = Constraint_Zy,
    267     Constraints_ShiftAmount = 16,
    268 
    269     Flag_MatchingOperand = 0x80000000
    270   };
    271 
    272   static unsigned getFlagWord(unsigned Kind, unsigned NumOps) {
    273     assert(((NumOps << 3) & ~0xffff) == 0 && "Too many inline asm operands!");
    274     assert(Kind >= Kind_RegUse && Kind <= Kind_Mem && "Invalid Kind");
    275     return Kind | (NumOps << 3);
    276   }
    277 
    278   static bool isRegDefKind(unsigned Flag){ return getKind(Flag) == Kind_RegDef;}
    279   static bool isImmKind(unsigned Flag) { return getKind(Flag) == Kind_Imm; }
    280   static bool isMemKind(unsigned Flag) { return getKind(Flag) == Kind_Mem; }
    281   static bool isRegDefEarlyClobberKind(unsigned Flag) {
    282     return getKind(Flag) == Kind_RegDefEarlyClobber;
    283   }
    284   static bool isClobberKind(unsigned Flag) {
    285     return getKind(Flag) == Kind_Clobber;
    286   }
    287 
    288   /// getFlagWordForMatchingOp - Augment an existing flag word returned by
    289   /// getFlagWord with information indicating that this input operand is tied
    290   /// to a previous output operand.
    291   static unsigned getFlagWordForMatchingOp(unsigned InputFlag,
    292                                            unsigned MatchedOperandNo) {
    293     assert(MatchedOperandNo <= 0x7fff && "Too big matched operand");
    294     assert((InputFlag & ~0xffff) == 0 && "High bits already contain data");
    295     return InputFlag | Flag_MatchingOperand | (MatchedOperandNo << 16);
    296   }
    297 
    298   /// getFlagWordForRegClass - Augment an existing flag word returned by
    299   /// getFlagWord with the required register class for the following register
    300   /// operands.
    301   /// A tied use operand cannot have a register class, use the register class
    302   /// from the def operand instead.
    303   static unsigned getFlagWordForRegClass(unsigned InputFlag, unsigned RC) {
    304     // Store RC + 1, reserve the value 0 to mean 'no register class'.
    305     ++RC;
    306     assert(!isImmKind(InputFlag) && "Immediates cannot have a register class");
    307     assert(!isMemKind(InputFlag) && "Memory operand cannot have a register class");
    308     assert(RC <= 0x7fff && "Too large register class ID");
    309     assert((InputFlag & ~0xffff) == 0 && "High bits already contain data");
    310     return InputFlag | (RC << 16);
    311   }
    312 
    313   /// Augment an existing flag word returned by getFlagWord with the constraint
    314   /// code for a memory constraint.
    315   static unsigned getFlagWordForMem(unsigned InputFlag, unsigned Constraint) {
    316     assert(isMemKind(InputFlag) && "InputFlag is not a memory constraint!");
    317     assert(Constraint <= 0x7fff && "Too large a memory constraint ID");
    318     assert(Constraint <= Constraints_Max && "Unknown constraint ID");
    319     assert((InputFlag & ~0xffff) == 0 && "High bits already contain data");
    320     return InputFlag | (Constraint << Constraints_ShiftAmount);
    321   }
    322 
    323   static unsigned convertMemFlagWordToMatchingFlagWord(unsigned InputFlag) {
    324     assert(isMemKind(InputFlag));
    325     return InputFlag & ~(0x7fff << Constraints_ShiftAmount);
    326   }
    327 
    328   static unsigned getKind(unsigned Flags) {
    329     return Flags & 7;
    330   }
    331 
    332   static unsigned getMemoryConstraintID(unsigned Flag) {
    333     assert(isMemKind(Flag));
    334     return (Flag >> Constraints_ShiftAmount) & 0x7fff;
    335   }
    336 
    337   /// getNumOperandRegisters - Extract the number of registers field from the
    338   /// inline asm operand flag.
    339   static unsigned getNumOperandRegisters(unsigned Flag) {
    340     return (Flag & 0xffff) >> 3;
    341   }
    342 
    343   /// isUseOperandTiedToDef - Return true if the flag of the inline asm
    344   /// operand indicates it is an use operand that's matched to a def operand.
    345   static bool isUseOperandTiedToDef(unsigned Flag, unsigned &Idx) {
    346     if ((Flag & Flag_MatchingOperand) == 0)
    347       return false;
    348     Idx = (Flag & ~Flag_MatchingOperand) >> 16;
    349     return true;
    350   }
    351 
    352   /// hasRegClassConstraint - Returns true if the flag contains a register
    353   /// class constraint.  Sets RC to the register class ID.
    354   static bool hasRegClassConstraint(unsigned Flag, unsigned &RC) {
    355     if (Flag & Flag_MatchingOperand)
    356       return false;
    357     unsigned High = Flag >> 16;
    358     // getFlagWordForRegClass() uses 0 to mean no register class, and otherwise
    359     // stores RC + 1.
    360     if (!High)
    361       return false;
    362     RC = High - 1;
    363     return true;
    364   }
    365 
    366   static std::vector<StringRef> getExtraInfoNames(unsigned ExtraInfo) {
    367     std::vector<StringRef> Result;
    368     if (ExtraInfo & InlineAsm::Extra_HasSideEffects)
    369       Result.push_back("sideeffect");
    370     if (ExtraInfo & InlineAsm::Extra_MayLoad)
    371       Result.push_back("mayload");
    372     if (ExtraInfo & InlineAsm::Extra_MayStore)
    373       Result.push_back("maystore");
    374     if (ExtraInfo & InlineAsm::Extra_IsConvergent)
    375       Result.push_back("isconvergent");
    376     if (ExtraInfo & InlineAsm::Extra_IsAlignStack)
    377       Result.push_back("alignstack");
    378 
    379     AsmDialect Dialect =
    380         InlineAsm::AsmDialect((ExtraInfo & InlineAsm::Extra_AsmDialect));
    381 
    382     if (Dialect == InlineAsm::AD_ATT)
    383       Result.push_back("attdialect");
    384     if (Dialect == InlineAsm::AD_Intel)
    385       Result.push_back("inteldialect");
    386 
    387     return Result;
    388   }
    389 
    390   static StringRef getKindName(unsigned Kind) {
    391     switch (Kind) {
    392     case InlineAsm::Kind_RegUse:
    393       return "reguse";
    394     case InlineAsm::Kind_RegDef:
    395       return "regdef";
    396     case InlineAsm::Kind_RegDefEarlyClobber:
    397       return "regdef-ec";
    398     case InlineAsm::Kind_Clobber:
    399       return "clobber";
    400     case InlineAsm::Kind_Imm:
    401       return "imm";
    402     case InlineAsm::Kind_Mem:
    403       return "mem";
    404     default:
    405       llvm_unreachable("Unknown operand kind");
    406     }
    407   }
    408 
    409   static StringRef getMemConstraintName(unsigned Constraint) {
    410     switch (Constraint) {
    411     case InlineAsm::Constraint_es:
    412       return "es";
    413     case InlineAsm::Constraint_i:
    414       return "i";
    415     case InlineAsm::Constraint_m:
    416       return "m";
    417     case InlineAsm::Constraint_o:
    418       return "o";
    419     case InlineAsm::Constraint_v:
    420       return "v";
    421     case InlineAsm::Constraint_Q:
    422       return "Q";
    423     case InlineAsm::Constraint_R:
    424       return "R";
    425     case InlineAsm::Constraint_S:
    426       return "S";
    427     case InlineAsm::Constraint_T:
    428       return "T";
    429     case InlineAsm::Constraint_Um:
    430       return "Um";
    431     case InlineAsm::Constraint_Un:
    432       return "Un";
    433     case InlineAsm::Constraint_Uq:
    434       return "Uq";
    435     case InlineAsm::Constraint_Us:
    436       return "Us";
    437     case InlineAsm::Constraint_Ut:
    438       return "Ut";
    439     case InlineAsm::Constraint_Uv:
    440       return "Uv";
    441     case InlineAsm::Constraint_Uy:
    442       return "Uy";
    443     case InlineAsm::Constraint_X:
    444       return "X";
    445     case InlineAsm::Constraint_Z:
    446       return "Z";
    447     case InlineAsm::Constraint_ZC:
    448       return "ZC";
    449     case InlineAsm::Constraint_Zy:
    450       return "Zy";
    451     default:
    452       llvm_unreachable("Unknown memory constraint");
    453     }
    454   }
    455 };
    456 
    457 } // end namespace llvm
    458 
    459 #endif // LLVM_IR_INLINEASM_H
    460