[llvm-commits] CVS: llvm/lib/Transforms/Scalar/LowerInvoke.cpp
Chris Lattner
lattner at cs.uiuc.edu
Sun Feb 8 16:15:01 PST 2004
Changes in directory llvm/lib/Transforms/Scalar:
LowerInvoke.cpp updated: 1.7 -> 1.8
---
Log message:
Improve compatibility with programs that already have a prototype for 'write',
even if it is wierd in some way.
---
Diffs of the changes: (+43 -12)
Index: llvm/lib/Transforms/Scalar/LowerInvoke.cpp
diff -u llvm/lib/Transforms/Scalar/LowerInvoke.cpp:1.7 llvm/lib/Transforms/Scalar/LowerInvoke.cpp:1.8
--- llvm/lib/Transforms/Scalar/LowerInvoke.cpp:1.7 Sun Feb 8 15:43:50 2004
+++ llvm/lib/Transforms/Scalar/LowerInvoke.cpp Sun Feb 8 16:14:44 2004
@@ -60,6 +60,7 @@
bool doInitialization(Module &M);
bool runOnFunction(Function &F);
private:
+ void writeAbortMessage(Instruction *IB);
bool insertCheapEHSupport(Function &F);
bool insertExpensiveEHSupport(Function &F);
};
@@ -148,12 +149,49 @@
}
// We need the 'write' and 'abort' functions for both models.
- WriteFn = M.getOrInsertFunction("write", Type::VoidTy, Type::IntTy,
- VoidPtrTy, Type::IntTy, 0);
AbortFn = M.getOrInsertFunction("abort", Type::VoidTy, 0);
+
+ // Unfortunately, 'write' can end up being prototyped in several different
+ // ways. If the user defines a three (or more) operand function named 'write'
+ // we will used their prototype. We _do not_ want to insert another instance
+ // of a write prototype, because we don't know that the funcresolve pass will
+ // run after us. If there is a definition of a write function, but it's not
+ // suitable for our uses, we just don't emit write calls. If there is no
+ // write prototype at all, we just add one.
+ if (Function *WF = M.getNamedFunction("write")) {
+ if (WF->getFunctionType()->getNumParams() > 3 ||
+ WF->getFunctionType()->isVarArg())
+ WriteFn = WF;
+ else
+ WriteFn = 0;
+ } else {
+ WriteFn = M.getOrInsertFunction("write", Type::VoidTy, Type::IntTy,
+ VoidPtrTy, Type::IntTy, 0);
+ }
return true;
}
+void LowerInvoke::writeAbortMessage(Instruction *IB) {
+ if (WriteFn) {
+ // These are the arguments we WANT...
+ std::vector<Value*> Args;
+ Args.push_back(ConstantInt::get(Type::IntTy, 2));
+ Args.push_back(AbortMessage);
+ Args.push_back(ConstantInt::get(Type::IntTy, AbortMessageLength));
+
+ // If the actual declaration of write disagrees, insert casts as
+ // appropriate.
+ const FunctionType *FT = WriteFn->getFunctionType();
+ unsigned NumArgs = FT->getNumParams();
+ for (unsigned i = 0; i != 3; ++i)
+ if (i < NumArgs && FT->getParamType(i) != Args[i]->getType())
+ Args[i] = ConstantExpr::getCast(cast<Constant>(Args[i]),
+ FT->getParamType(i));
+
+ new CallInst(WriteFn, Args, "", IB);
+ }
+}
+
bool LowerInvoke::insertCheapEHSupport(Function &F) {
bool Changed = false;
for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB)
@@ -177,11 +215,7 @@
++NumLowered; Changed = true;
} else if (UnwindInst *UI = dyn_cast<UnwindInst>(BB->getTerminator())) {
// Insert a new call to write(2, AbortMessage, AbortMessageLength);
- std::vector<Value*> Args;
- Args.push_back(ConstantInt::get(Type::IntTy, 2));
- Args.push_back(AbortMessage);
- Args.push_back(ConstantInt::get(Type::IntTy, AbortMessageLength));
- new CallInst(WriteFn, Args, "", UI);
+ writeAbortMessage(UI);
// Insert a call to abort()
new CallInst(AbortFn, std::vector<Value*>(), "", UI);
@@ -331,11 +365,8 @@
RI = TermBlock->getTerminator();
// Insert a new call to write(2, AbortMessage, AbortMessageLength);
- Idx[0] = ConstantInt::get(Type::IntTy, 2);
- Idx[1] = AbortMessage;
- Idx.push_back(ConstantInt::get(Type::IntTy, AbortMessageLength));
- new CallInst(WriteFn, Idx, "", RI);
-
+ writeAbortMessage(RI);
+
// Insert a call to abort()
new CallInst(AbortFn, std::vector<Value*>(), "", RI);
}
More information about the llvm-commits
mailing list