b36d2128a61e9521841e11eb903de1971de93473
[oota-llvm.git] / lib / VMCore / InlineAsm.cpp
1 //===-- InlineAsm.cpp - Implement the InlineAsm class ---------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file was developed by Chris Lattner and is distributed under the
6 // University of Illinois Open Source License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements the InlineAsm class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "llvm/InlineAsm.h"
15 #include "llvm/DerivedTypes.h"
16 #include <algorithm>
17 #include <cctype>
18 using namespace llvm;
19
20 // NOTE: when memoizing the function type, we have to be careful to handle the
21 // case when the type gets refined.
22
23 InlineAsm *InlineAsm::get(const FunctionType *Ty, const std::string &AsmString,
24                           const std::string &Constraints, bool hasSideEffects) {
25   // FIXME: memoize!
26   return new InlineAsm(Ty, AsmString, Constraints, hasSideEffects);  
27 }
28
29 InlineAsm::InlineAsm(const FunctionType *Ty, const std::string &asmString,
30                      const std::string &constraints, bool hasSideEffects)
31   : Value(PointerType::get(Ty), Value::InlineAsmVal), AsmString(asmString), 
32     Constraints(constraints), HasSideEffects(hasSideEffects) {
33
34   // Do various checks on the constraint string and type.
35   assert(Verify(Ty, constraints) && "Function type not legal for constraints!");
36 }
37
38 const FunctionType *InlineAsm::getFunctionType() const {
39   return cast<FunctionType>(getType()->getElementType());
40 }
41
42 /// Parse - Analyze the specified string (e.g. "==&{eax}") and fill in the
43 /// fields in this structure.  If the constraint string is not understood,
44 /// return true, otherwise return false.
45 bool InlineAsm::ConstraintInfo::Parse(const std::string &Str,
46                      std::vector<InlineAsm::ConstraintInfo> &ConstraintsSoFar) {
47   std::string::const_iterator I = Str.begin(), E = Str.end();
48   
49   // Initialize
50   Type = isInput;
51   isEarlyClobber = false;
52   isIndirectOutput = false;
53   hasMatchingInput = false;
54   isCommutative = false;
55   
56   // Parse the prefix.
57   if (*I == '~') {
58     Type = isClobber;
59     ++I;
60   } else if (*I == '=') {
61     ++I;
62     Type = isOutput;
63     if (I != E && *I == '=') {
64       isIndirectOutput = true;
65       ++I;
66     }
67   }
68   
69   if (I == E) return true;  // Just a prefix, like "==" or "~".
70   
71   // Parse the modifiers.
72   bool DoneWithModifiers = false;
73   while (!DoneWithModifiers) {
74     switch (*I) {
75     default:
76       DoneWithModifiers = true;
77       break;
78     case '&':     // Early clobber.
79       if (Type != isOutput ||      // Cannot early clobber anything but output.
80           isEarlyClobber)          // Reject &&&&&&
81         return true;
82       isEarlyClobber = true;
83       break;
84     case '%':     // Commutative.
85       if (Type == isClobber ||     // Cannot commute clobbers.
86           isCommutative)           // Reject %%%%%
87         return true;
88       isCommutative = true;
89       break;
90     case '#':     // Comment.
91     case '*':     // Register preferencing.
92       return true;     // Not supported.
93     }
94     
95     if (!DoneWithModifiers) {
96       ++I;
97       if (I == E) return true;   // Just prefixes and modifiers!
98     }
99   }
100   
101   // Parse the various constraints.
102   while (I != E) {
103     if (*I == '{') {   // Physical register reference.
104       // Find the end of the register name.
105       std::string::const_iterator ConstraintEnd = std::find(I+1, E, '}');
106       if (ConstraintEnd == E) return true;  // "{foo"
107       Codes.push_back(std::string(I, ConstraintEnd+1));
108       I = ConstraintEnd+1;
109     } else if (isdigit(*I)) {     // Matching Constraint
110       // Maximal munch numbers.
111       std::string::const_iterator NumStart = I;
112       while (I != E && isdigit(*I))
113         ++I;
114       Codes.push_back(std::string(NumStart, I));
115       unsigned N = atoi(Codes.back().c_str());
116       // Check that this is a valid matching constraint!
117       if (N >= ConstraintsSoFar.size() || ConstraintsSoFar[N].Type != isOutput||
118           Type != isInput)
119         return true;  // Invalid constraint number.
120       
121       // Note that operand #n has a matching input.
122       ConstraintsSoFar[N].hasMatchingInput = true;
123     } else {
124       // Single letter constraint.
125       Codes.push_back(std::string(I, I+1));
126       ++I;
127     }
128   }
129
130   return false;
131 }
132
133 std::vector<InlineAsm::ConstraintInfo>
134 InlineAsm::ParseConstraints(const std::string &Constraints) {
135   std::vector<ConstraintInfo> Result;
136   
137   // Scan the constraints string.
138   for (std::string::const_iterator I = Constraints.begin(), 
139        E = Constraints.end(); I != E; ) {
140     ConstraintInfo Info;
141
142     // Find the end of this constraint.
143     std::string::const_iterator ConstraintEnd = std::find(I, E, ',');
144
145     if (ConstraintEnd == I ||  // Empty constraint like ",,"
146         Info.Parse(std::string(I, ConstraintEnd), Result)) {
147       Result.clear();          // Erroneous constraint?
148       break;
149     }
150
151     Result.push_back(Info);
152     
153     // ConstraintEnd may be either the next comma or the end of the string.  In
154     // the former case, we skip the comma.
155     I = ConstraintEnd;
156     if (I != E) {
157       ++I;
158       if (I == E) { Result.clear(); break; }    // don't allow "xyz,"
159     }
160   }
161   
162   return Result;
163 }
164
165
166 /// Verify - Verify that the specified constraint string is reasonable for the
167 /// specified function type, and otherwise validate the constraint string.
168 bool InlineAsm::Verify(const FunctionType *Ty, const std::string &ConstStr) {
169   if (Ty->isVarArg()) return false;
170   
171   std::vector<ConstraintInfo> Constraints = ParseConstraints(ConstStr);
172   
173   // Error parsing constraints.
174   if (Constraints.empty() && !ConstStr.empty()) return false;
175   
176   unsigned NumOutputs = 0, NumInputs = 0, NumClobbers = 0;
177   
178   for (unsigned i = 0, e = Constraints.size(); i != e; ++i) {
179     switch (Constraints[i].Type) {
180     case InlineAsm::isOutput:
181       if (!Constraints[i].isIndirectOutput) {
182         if (NumInputs || NumClobbers) return false;  // outputs come first.
183         ++NumOutputs;
184         break;
185       }
186       // FALLTHROUGH for IndirectOutputs.
187     case InlineAsm::isInput:
188       if (NumClobbers) return false;               // inputs before clobbers.
189       ++NumInputs;
190       break;
191     case InlineAsm::isClobber:
192       ++NumClobbers;
193       break;
194     }
195   }
196     
197   if (NumOutputs > 1) return false;  // Only one result allowed so far.
198   
199   if ((Ty->getReturnType() != Type::VoidTy) != NumOutputs)
200     return false;   // NumOutputs = 1 iff has a result type.
201   
202   if (Ty->getNumParams() != NumInputs) return false;
203   return true;
204 }