1b6f0625de2d4a733dab2b63f6c54f59c84cefee0Daniel Dunbar//===-- llvm-stress.cpp - Generate random LL files to stress-test LLVM ----===//
2fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem//
3fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem//                     The LLVM Compiler Infrastructure
4fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem//
5fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem// This file is distributed under the University of Illinois Open Source
6fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem// License. See LICENSE.TXT for details.
7fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem//
8fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem//===----------------------------------------------------------------------===//
9fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem//
10fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem// This program is a utility that generates random .ll files to stress-test
11fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem// different components in LLVM.
12fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem//
13fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem//===----------------------------------------------------------------------===//
14fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/LLVMContext.h"
15fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Module.h"
16fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/PassManager.h"
17fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Constants.h"
18fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Instruction.h"
19fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/CallGraphSCCPass.h"
20fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Assembly/PrintModulePass.h"
21fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Analysis/Verifier.h"
22fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Support/PassNameParser.h"
23fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Support/Debug.h"
24fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Support/ManagedStatic.h"
25fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Support/PluginLoader.h"
26fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Support/PrettyStackTrace.h"
27fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include "llvm/Support/ToolOutputFile.h"
28fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include <memory>
29fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include <sstream>
30fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include <set>
31fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include <vector>
32fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem#include <algorithm>
33fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemusing namespace llvm;
34fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
35fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstatic cl::opt<unsigned> SeedCL("seed",
36fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  cl::desc("Seed used for randomness"), cl::init(0));
37fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstatic cl::opt<unsigned> SizeCL("size",
38fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  cl::desc("The estimated size of the generated function (# of instrs)"),
39fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  cl::init(100));
40fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstatic cl::opt<std::string>
41fdc309cc4e95778d7615f3829917c4fe42086e1eNadav RotemOutputFilename("o", cl::desc("Override output filename"),
42fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem               cl::value_desc("filename"));
43fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
442cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkelstatic cl::opt<bool> GenHalfFloat("generate-half-float",
452cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel  cl::desc("Generate half-length floating-point values"), cl::init(false));
462cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkelstatic cl::opt<bool> GenX86FP80("generate-x86-fp80",
472cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel  cl::desc("Generate 80-bit X86 floating-point values"), cl::init(false));
482cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkelstatic cl::opt<bool> GenFP128("generate-fp128",
492cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel  cl::desc("Generate 128-bit floating-point values"), cl::init(false));
502cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkelstatic cl::opt<bool> GenPPCFP128("generate-ppc-fp128",
512cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel  cl::desc("Generate 128-bit PPC floating-point values"), cl::init(false));
522cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkelstatic cl::opt<bool> GenX86MMX("generate-x86-mmx",
532cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel  cl::desc("Generate X86 MMX floating-point values"), cl::init(false));
542cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel
55fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem/// A utility class to provide a pseudo-random number generator which is
56fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem/// the same across all platforms. This is somewhat close to the libc
57fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem/// implementation. Note: This is not a cryptographically secure pseudorandom
58fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem/// number generator.
59fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemclass Random {
60fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotempublic:
61fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// C'tor
62fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Random(unsigned _seed):Seed(_seed) {}
6383f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith
6483f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith  /// Return a random integer, up to a
6583f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith  /// maximum of 2**19 - 1.
6683f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith  uint32_t Rand() {
6783f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith    uint32_t Val = Seed + 0x000b07a1;
68fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Seed = (Val * 0x3c7c0ac1);
69fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Only lowest 19 bits are random-ish.
70fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return Seed & 0x7ffff;
71fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
72fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
7383f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith  /// Return a random 32 bit integer.
7483f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith  uint32_t Rand32() {
7583f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith    uint32_t Val = Rand();
7683f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith    Val &= 0xffff;
7783f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith    return Val | (Rand() << 16);
7883f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith  }
7983f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith
8083f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith  /// Return a random 64 bit integer.
8183f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith  uint64_t Rand64() {
8283f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith    uint64_t Val = Rand32();
8383f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith    return Val | (uint64_t(Rand32()) << 32);
8483f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith  }
85fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemprivate:
86fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  unsigned Seed;
87fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
88fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
89fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem/// Generate an empty function with a default argument list.
90fdc309cc4e95778d7615f3829917c4fe42086e1eNadav RotemFunction *GenEmptyFunction(Module *M) {
91fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Type Definitions
92fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::vector<Type*> ArgsTy;
93fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Define a few arguments
94fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  LLVMContext &Context = M->getContext();
95fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ArgsTy.push_back(PointerType::get(IntegerType::getInt8Ty(Context), 0));
96fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ArgsTy.push_back(PointerType::get(IntegerType::getInt32Ty(Context), 0));
97fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ArgsTy.push_back(PointerType::get(IntegerType::getInt64Ty(Context), 0));
98fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ArgsTy.push_back(IntegerType::getInt32Ty(Context));
99fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ArgsTy.push_back(IntegerType::getInt64Ty(Context));
100fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ArgsTy.push_back(IntegerType::getInt8Ty(Context));
101fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
102fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  FunctionType *FuncTy = FunctionType::get(Type::getVoidTy(Context), ArgsTy, 0);
103fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Pick a unique name to describe the input parameters
104fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::stringstream ss;
105fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ss<<"autogen_SD"<<SeedCL;
106fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Function *Func = Function::Create(FuncTy, GlobalValue::ExternalLinkage,
107fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                                    ss.str(), M);
108fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
109fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Func->setCallingConv(CallingConv::C);
110fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  return Func;
111fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem}
112fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
113fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem/// A base class, implementing utilities needed for
114fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem/// modifying and adding new random instructions.
115fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct Modifier {
116fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Used to store the randomly generated values.
117fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  typedef std::vector<Value*> PieceTable;
118fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
119fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotempublic:
120fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// C'tor
12108c833930726d5c33baac060a7aca7c1caf636b3Nadav Rotem  Modifier(BasicBlock *Block, PieceTable *PT, Random *R):
122b6f0625de2d4a733dab2b63f6c54f59c84cefee0Daniel Dunbar    BB(Block),PT(PT),Ran(R),Context(BB->getContext()) {}
123fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Add a new instruction.
124fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() = 0;
125fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Add N new instructions,
126fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void ActN(unsigned n) {
127fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    for (unsigned i=0; i<n; ++i)
128fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Act();
129fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
130fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
131fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemprotected:
132fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Return a random value from the list of known values.
133fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Value *getRandomVal() {
134fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    assert(PT->size());
135fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return PT->at(Ran->Rand() % PT->size());
136fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
137fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
138bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem  Constant *getRandomConstant(Type *Tp) {
139bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem    if (Tp->isIntegerTy()) {
140bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem      if (Ran->Rand() & 1)
141bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem        return ConstantInt::getAllOnesValue(Tp);
142bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem      return ConstantInt::getNullValue(Tp);
143bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem    } else if (Tp->isFloatingPointTy()) {
144bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem      if (Ran->Rand() & 1)
145bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem        return ConstantFP::getAllOnesValue(Tp);
146bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem      return ConstantFP::getNullValue(Tp);
147bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem    }
148bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem    return UndefValue::get(Tp);
149bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem  }
150bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem
151fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Return a random value with a known type.
152fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Value *getRandomValue(Type *Tp) {
153fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    unsigned index = Ran->Rand();
154fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    for (unsigned i=0; i<PT->size(); ++i) {
155fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Value *V = PT->at((index + i) % PT->size());
156fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (V->getType() == Tp)
157fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return V;
158fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
159fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
160fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // If the requested type was not found, generate a constant value.
161fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (Tp->isIntegerTy()) {
162fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (Ran->Rand() & 1)
163fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return ConstantInt::getAllOnesValue(Tp);
164fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      return ConstantInt::getNullValue(Tp);
165fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    } else if (Tp->isFloatingPointTy()) {
166fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (Ran->Rand() & 1)
167fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return ConstantFP::getAllOnesValue(Tp);
168fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      return ConstantFP::getNullValue(Tp);
169bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem    } else if (Tp->isVectorTy()) {
170bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem      VectorType *VTp = cast<VectorType>(Tp);
171bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem
172bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem      std::vector<Constant*> TempValues;
173bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem      TempValues.reserve(VTp->getNumElements());
174bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem      for (unsigned i = 0; i < VTp->getNumElements(); ++i)
175bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem        TempValues.push_back(getRandomConstant(VTp->getScalarType()));
176bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem
177bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem      ArrayRef<Constant*> VectorValue(TempValues);
178bfb7dfa756ffa48d2c968ffcade3295938495b6eNadav Rotem      return ConstantVector::get(VectorValue);
179fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
180fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
181fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return UndefValue::get(Tp);
182fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
183fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
184fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Return a random value of any pointer type.
185fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Value *getRandomPointerValue() {
186fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    unsigned index = Ran->Rand();
187fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    for (unsigned i=0; i<PT->size(); ++i) {
188fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Value *V = PT->at((index + i) % PT->size());
189fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (V->getType()->isPointerTy())
190fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return V;
191fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
192fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return UndefValue::get(pickPointerType());
193fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
194fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
195fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Return a random value of any vector type.
196fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Value *getRandomVectorValue() {
197fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    unsigned index = Ran->Rand();
198fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    for (unsigned i=0; i<PT->size(); ++i) {
199fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Value *V = PT->at((index + i) % PT->size());
200fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (V->getType()->isVectorTy())
201fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return V;
202fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
203fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return UndefValue::get(pickVectorType());
204fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
205fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
206fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Pick a random type.
207fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Type *pickType() {
208fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return (Ran->Rand() & 1 ? pickVectorType() : pickScalarType());
209fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
210fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
211fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Pick a random pointer type.
212fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Type *pickPointerType() {
213fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Type *Ty = pickType();
214fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return PointerType::get(Ty, 0);
215fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
216fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
217fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Pick a random vector type.
218fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Type *pickVectorType(unsigned len = (unsigned)-1) {
219fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Pick a random vector width in the range 2**0 to 2**4.
220fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // by adding two randoms we are generating a normal-like distribution
221fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // around 2**3.
222fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    unsigned width = 1<<((Ran->Rand() % 3) + (Ran->Rand() % 3));
223701de8fafc8db86a0a7df61b177720b1f681c60cDylan Noblesmith    Type *Ty;
224701de8fafc8db86a0a7df61b177720b1f681c60cDylan Noblesmith
225701de8fafc8db86a0a7df61b177720b1f681c60cDylan Noblesmith    // Vectors of x86mmx are illegal; keep trying till we get something else.
226701de8fafc8db86a0a7df61b177720b1f681c60cDylan Noblesmith    do {
227701de8fafc8db86a0a7df61b177720b1f681c60cDylan Noblesmith      Ty = pickScalarType();
228701de8fafc8db86a0a7df61b177720b1f681c60cDylan Noblesmith    } while (Ty->isX86_MMXTy());
229701de8fafc8db86a0a7df61b177720b1f681c60cDylan Noblesmith
230fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (len != (unsigned)-1)
231fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      width = len;
232fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return VectorType::get(Ty, width);
233fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
234fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
235fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Pick a random scalar type.
236fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Type *pickScalarType() {
2372cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel    Type *t = 0;
2382cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel    do {
2392cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      switch (Ran->Rand() % 30) {
2402cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 0: t = Type::getInt1Ty(Context); break;
2412cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 1: t = Type::getInt8Ty(Context); break;
2422cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 2: t = Type::getInt16Ty(Context); break;
2432cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 3: case 4:
2442cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 5: t = Type::getFloatTy(Context); break;
2452cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 6: case 7:
2462cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 8: t = Type::getDoubleTy(Context); break;
2472cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 9: case 10:
2482cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 11: t = Type::getInt32Ty(Context); break;
2492cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 12: case 13:
2502cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 14: t = Type::getInt64Ty(Context); break;
2512cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 15: case 16:
2522cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 17: if (GenHalfFloat) t = Type::getHalfTy(Context); break;
2532cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 18: case 19:
2542cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 20: if (GenX86FP80) t = Type::getX86_FP80Ty(Context); break;
2552cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 21: case 22:
2562cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 23: if (GenFP128) t = Type::getFP128Ty(Context); break;
2572cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 24: case 25:
2582cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 26: if (GenPPCFP128) t = Type::getPPC_FP128Ty(Context); break;
2592cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 27: case 28:
2602cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      case 29: if (GenX86MMX) t = Type::getX86_MMXTy(Context); break;
2612cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      default: llvm_unreachable("Invalid scalar value");
2622cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel      }
2632cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel    } while (t == 0);
2642cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel
2652cff6179ab40fc92ec5b6302b9ab5f7b0ee1e513Hal Finkel    return t;
266fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
267fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
268fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Basic block to populate
269fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  BasicBlock *BB;
270fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Value table
271fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  PieceTable *PT;
272fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Random number generator
273fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Random *Ran;
274fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  /// Context
275fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  LLVMContext &Context;
276fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
277fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
278fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct LoadModifier: public Modifier {
279b6f0625de2d4a733dab2b63f6c54f59c84cefee0Daniel Dunbar  LoadModifier(BasicBlock *BB, PieceTable *PT, Random *R):Modifier(BB, PT, R) {}
280fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
281fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Try to use predefined pointers. If non exist, use undef pointer value;
282fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Ptr = getRandomPointerValue();
283fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *V = new LoadInst(Ptr, "L", BB->getTerminator());
284fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    PT->push_back(V);
285fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
286fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
287fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
288fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct StoreModifier: public Modifier {
289fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  StoreModifier(BasicBlock *BB, PieceTable *PT, Random *R):Modifier(BB, PT, R) {}
290fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
291fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Try to use predefined pointers. If non exist, use undef pointer value;
292fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Ptr = getRandomPointerValue();
293fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Type  *Tp = Ptr->getType();
294fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Val = getRandomValue(Tp->getContainedType(0));
2952e851a9abf781bc5feef9be98f794dcb1caab548Nadav Rotem    Type  *ValTy = Val->getType();
296fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
297fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Do not store vectors of i1s because they are unsupported
298794c16ae856f827babb22f862f4e69118ad68b09Nadav Rotem    // by the codegen.
299794c16ae856f827babb22f862f4e69118ad68b09Nadav Rotem    if (ValTy->isVectorTy() && ValTy->getScalarSizeInBits() == 1)
300fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      return;
301fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
302fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    new StoreInst(Val, Ptr, BB->getTerminator());
303fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
304fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
305fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
306fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct BinModifier: public Modifier {
307fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  BinModifier(BasicBlock *BB, PieceTable *PT, Random *R):Modifier(BB, PT, R) {}
308fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
309fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
310fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Val0 = getRandomVal();
311fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Val1 = getRandomValue(Val0->getType());
312fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
313fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Don't handle pointer types.
314fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (Val0->getType()->isPointerTy() ||
315fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        Val1->getType()->isPointerTy())
316fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      return;
317fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
318fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Don't handle i1 types.
319fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (Val0->getType()->getScalarSizeInBits() == 1)
320fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      return;
321fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
322fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
323fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    bool isFloat = Val0->getType()->getScalarType()->isFloatingPointTy();
324fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Instruction* Term = BB->getTerminator();
325fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    unsigned R = Ran->Rand() % (isFloat ? 7 : 13);
326fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Instruction::BinaryOps Op;
327fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
328fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    switch (R) {
329fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    default: llvm_unreachable("Invalid BinOp");
330fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 0:{Op = (isFloat?Instruction::FAdd : Instruction::Add); break; }
331fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 1:{Op = (isFloat?Instruction::FSub : Instruction::Sub); break; }
332fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 2:{Op = (isFloat?Instruction::FMul : Instruction::Mul); break; }
333fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 3:{Op = (isFloat?Instruction::FDiv : Instruction::SDiv); break; }
334fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 4:{Op = (isFloat?Instruction::FDiv : Instruction::UDiv); break; }
335fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 5:{Op = (isFloat?Instruction::FRem : Instruction::SRem); break; }
336fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 6:{Op = (isFloat?Instruction::FRem : Instruction::URem); break; }
337fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 7: {Op = Instruction::Shl;  break; }
338fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 8: {Op = Instruction::LShr; break; }
339fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 9: {Op = Instruction::AShr; break; }
340fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 10:{Op = Instruction::And;  break; }
341fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 11:{Op = Instruction::Or;   break; }
342fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    case 12:{Op = Instruction::Xor;  break; }
343fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
344fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
345fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    PT->push_back(BinaryOperator::Create(Op, Val0, Val1, "B", Term));
346fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
347fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
348fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
349fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem/// Generate constant values.
350fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct ConstModifier: public Modifier {
351fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ConstModifier(BasicBlock *BB, PieceTable *PT, Random *R):Modifier(BB, PT, R) {}
352fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
353fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Type *Ty = pickType();
354fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
355fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (Ty->isVectorTy()) {
356fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      switch (Ran->Rand() % 2) {
357fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      case 0: if (Ty->getScalarType()->isIntegerTy())
358fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                return PT->push_back(ConstantVector::getAllOnesValue(Ty));
359fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      case 1: if (Ty->getScalarType()->isIntegerTy())
360fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                return PT->push_back(ConstantVector::getNullValue(Ty));
361fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      }
362fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
363fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
364fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (Ty->isFloatingPointTy()) {
36583f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith      // Generate 128 random bits, the size of the (currently)
36683f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith      // largest floating-point types.
36783f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith      uint64_t RandomBits[2];
36883f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith      for (unsigned i = 0; i < 2; ++i)
36983f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith        RandomBits[i] = Ran->Rand64();
37083f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith
37183f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith      APInt RandomInt(Ty->getPrimitiveSizeInBits(), makeArrayRef(RandomBits));
37283f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith
37383f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith      bool isIEEE = !Ty->isX86_FP80Ty() && !Ty->isPPC_FP128Ty();
37483f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith      APFloat RandomFloat(RandomInt, isIEEE);
37583f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith
376fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (Ran->Rand() & 1)
377fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return PT->push_back(ConstantFP::getNullValue(Ty));
37883f17f25fc560db4f756010a1bbe1f8eb1d74b12Dylan Noblesmith      return PT->push_back(ConstantFP::get(Ty->getContext(), RandomFloat));
379fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
380fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
381fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (Ty->isIntegerTy()) {
382fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      switch (Ran->Rand() % 7) {
383fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      case 0: if (Ty->isIntegerTy())
384fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                return PT->push_back(ConstantInt::get(Ty,
385fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                  APInt::getAllOnesValue(Ty->getPrimitiveSizeInBits())));
386fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      case 1: if (Ty->isIntegerTy())
387fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                return PT->push_back(ConstantInt::get(Ty,
388fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                  APInt::getNullValue(Ty->getPrimitiveSizeInBits())));
389fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      case 2: case 3: case 4: case 5:
390fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      case 6: if (Ty->isIntegerTy())
391fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                PT->push_back(ConstantInt::get(Ty, Ran->Rand()));
392fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      }
393fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
394fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
395fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
396fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
397fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
398fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct AllocaModifier: public Modifier {
399fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  AllocaModifier(BasicBlock *BB, PieceTable *PT, Random *R):Modifier(BB, PT, R){}
400fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
401fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
402fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Type *Tp = pickType();
403fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    PT->push_back(new AllocaInst(Tp, "A", BB->getFirstNonPHI()));
404fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
405fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
406fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
407fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct ExtractElementModifier: public Modifier {
408fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ExtractElementModifier(BasicBlock *BB, PieceTable *PT, Random *R):
409fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Modifier(BB, PT, R) {}
410fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
411fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
412fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Val0 = getRandomVectorValue();
413fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *V = ExtractElementInst::Create(Val0,
414fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem             ConstantInt::get(Type::getInt32Ty(BB->getContext()),
4157d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem             Ran->Rand() % cast<VectorType>(Val0->getType())->getNumElements()),
416fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem             "E", BB->getTerminator());
417fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return PT->push_back(V);
418fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
419fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
420fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
421fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct ShuffModifier: public Modifier {
422fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ShuffModifier(BasicBlock *BB, PieceTable *PT, Random *R):Modifier(BB, PT, R) {}
423fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
424fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
425fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Val0 = getRandomVectorValue();
426fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Val1 = getRandomValue(Val0->getType());
427fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
428fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    unsigned Width = cast<VectorType>(Val0->getType())->getNumElements();
429fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    std::vector<Constant*> Idxs;
430fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
431fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Type *I32 = Type::getInt32Ty(BB->getContext());
432fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    for (unsigned i=0; i<Width; ++i) {
433fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Constant *CI = ConstantInt::get(I32, Ran->Rand() % (Width*2));
434fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      // Pick some undef values.
435fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (!(Ran->Rand() % 5))
436fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        CI = UndefValue::get(I32);
437fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Idxs.push_back(CI);
438fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
439fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
440fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Constant *Mask = ConstantVector::get(Idxs);
441fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
442fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *V = new ShuffleVectorInst(Val0, Val1, Mask, "Shuff",
443fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                                     BB->getTerminator());
444fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    PT->push_back(V);
445fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
446fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
447fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
448fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct InsertElementModifier: public Modifier {
449fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  InsertElementModifier(BasicBlock *BB, PieceTable *PT, Random *R):
450fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Modifier(BB, PT, R) {}
451fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
452fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
453fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Val0 = getRandomVectorValue();
454fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Val1 = getRandomValue(Val0->getType()->getScalarType());
455fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
456fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *V = InsertElementInst::Create(Val0, Val1,
457fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem              ConstantInt::get(Type::getInt32Ty(BB->getContext()),
458fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem              Ran->Rand() % cast<VectorType>(Val0->getType())->getNumElements()),
459fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem              "I",  BB->getTerminator());
460fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return PT->push_back(V);
461fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
462fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
463fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
464fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
465fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct CastModifier: public Modifier {
466fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  CastModifier(BasicBlock *BB, PieceTable *PT, Random *R):Modifier(BB, PT, R) {}
467fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
468fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
469fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *V = getRandomVal();
470fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Type *VTy = V->getType();
471fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Type *DestTy = pickScalarType();
472fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
473fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Handle vector casts vectors.
474fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (VTy->isVectorTy()) {
475fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      VectorType *VecTy = cast<VectorType>(VTy);
476fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      DestTy = pickVectorType(VecTy->getNumElements());
477fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
478fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
4797d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem    // no need to cast.
480fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (VTy == DestTy) return;
481fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
482fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Pointers:
483fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (VTy->isPointerTy()) {
484fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (!DestTy->isPointerTy())
485fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        DestTy = PointerType::get(DestTy, 0);
486fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      return PT->push_back(
487fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        new BitCastInst(V, DestTy, "PC", BB->getTerminator()));
488fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
489fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
4907d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem    unsigned VSize = VTy->getScalarType()->getPrimitiveSizeInBits();
4917d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem    unsigned DestSize = DestTy->getScalarType()->getPrimitiveSizeInBits();
4927d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem
493fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Generate lots of bitcasts.
4947d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem    if ((Ran->Rand() & 1) && VSize == DestSize) {
495fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      return PT->push_back(
496fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        new BitCastInst(V, DestTy, "BC", BB->getTerminator()));
497fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
498fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
499fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Both types are integers:
500fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (VTy->getScalarType()->isIntegerTy() &&
501fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        DestTy->getScalarType()->isIntegerTy()) {
5027d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem      if (VSize > DestSize) {
503fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return PT->push_back(
504fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem          new TruncInst(V, DestTy, "Tr", BB->getTerminator()));
505fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      } else {
5067d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem        assert(VSize < DestSize && "Different int types with the same size?");
507fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        if (Ran->Rand() & 1)
508fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem          return PT->push_back(
509fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem            new ZExtInst(V, DestTy, "ZE", BB->getTerminator()));
510fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return PT->push_back(new SExtInst(V, DestTy, "Se", BB->getTerminator()));
511fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      }
512fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
513fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
514fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Fp to int.
515fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (VTy->getScalarType()->isFloatingPointTy() &&
516fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        DestTy->getScalarType()->isIntegerTy()) {
517fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (Ran->Rand() & 1)
518fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return PT->push_back(
519fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem          new FPToSIInst(V, DestTy, "FC", BB->getTerminator()));
520fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      return PT->push_back(new FPToUIInst(V, DestTy, "FC", BB->getTerminator()));
521fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
522fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
523fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Int to fp.
524fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (VTy->getScalarType()->isIntegerTy() &&
525fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        DestTy->getScalarType()->isFloatingPointTy()) {
526fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (Ran->Rand() & 1)
527fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return PT->push_back(
528fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem          new SIToFPInst(V, DestTy, "FC", BB->getTerminator()));
529fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      return PT->push_back(new UIToFPInst(V, DestTy, "FC", BB->getTerminator()));
530fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
531fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
532fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
533fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Both floats.
534fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (VTy->getScalarType()->isFloatingPointTy() &&
535fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        DestTy->getScalarType()->isFloatingPointTy()) {
5367d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem      if (VSize > DestSize) {
537fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return PT->push_back(
538fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem          new FPTruncInst(V, DestTy, "Tr", BB->getTerminator()));
5397d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem      } else if (VSize < DestSize) {
540fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        return PT->push_back(
541fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem          new FPExtInst(V, DestTy, "ZE", BB->getTerminator()));
542fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      }
5437d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem      // If VSize == DestSize, then the two types must be fp128 and ppc_fp128,
5447d719a5237d8da0aed188a400b9792b64dae5fc0Nadav Rotem      // for which there is no defined conversion. So do nothing.
545fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
546fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
547fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
548fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
549fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
550fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct SelectModifier: public Modifier {
551fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  SelectModifier(BasicBlock *BB, PieceTable *PT, Random *R):
552fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Modifier(BB, PT, R) {}
553fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
554fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
555fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    // Try a bunch of different select configuration until a valid one is found.
556fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Value *Val0 = getRandomVal();
557fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Value *Val1 = getRandomValue(Val0->getType());
558fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
559fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Type *CondTy = Type::getInt1Ty(Context);
560fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
561fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      // If the value type is a vector, and we allow vector select, then in 50%
562fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      // of the cases generate a vector select.
563fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      if (Val0->getType()->isVectorTy() && (Ran->Rand() % 1)) {
564fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        unsigned NumElem = cast<VectorType>(Val0->getType())->getNumElements();
565fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem        CondTy = VectorType::get(CondTy, NumElem);
566fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      }
567fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
568fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Value *Cond = getRandomValue(CondTy);
569fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Value *V = SelectInst::Create(Cond, Val0, Val1, "Sl", BB->getTerminator());
570fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      return PT->push_back(V);
571fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
572fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
573fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
574fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
575fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemstruct CmpModifier: public Modifier {
576fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  CmpModifier(BasicBlock *BB, PieceTable *PT, Random *R):Modifier(BB, PT, R) {}
577fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  virtual void Act() {
578fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
579fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Val0 = getRandomVal();
580fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *Val1 = getRandomValue(Val0->getType());
581fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
582fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (Val0->getType()->isPointerTy()) return;
583fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    bool fp = Val0->getType()->getScalarType()->isFloatingPointTy();
584fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
585fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    int op;
586fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (fp) {
587fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      op = Ran->Rand() %
588fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      (CmpInst::LAST_FCMP_PREDICATE - CmpInst::FIRST_FCMP_PREDICATE) +
589fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem       CmpInst::FIRST_FCMP_PREDICATE;
590fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    } else {
591fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      op = Ran->Rand() %
592fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      (CmpInst::LAST_ICMP_PREDICATE - CmpInst::FIRST_ICMP_PREDICATE) +
593fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem       CmpInst::FIRST_ICMP_PREDICATE;
594fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
595fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
596fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Value *V = CmpInst::Create(fp ? Instruction::FCmp : Instruction::ICmp,
597fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                               op, Val0, Val1, "Cmp", BB->getTerminator());
598fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return PT->push_back(V);
599fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
600fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem};
601fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
602fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemvoid FillFunction(Function *F) {
603fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Create a legal entry block.
604fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  BasicBlock *BB = BasicBlock::Create(F->getContext(), "BB", F);
605fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ReturnInst::Create(F->getContext(), BB);
606fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
607fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Create the value table.
608fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Modifier::PieceTable PT;
609fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Pick an initial seed value
610fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Random R(SeedCL);
611fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
612fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Consider arguments as legal values.
613fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  for (Function::arg_iterator it = F->arg_begin(), e = F->arg_end();
614fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem       it != e; ++it)
615fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    PT.push_back(it);
616fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
617fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // List of modifiers which add new random instructions.
618fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::vector<Modifier*> Modifiers;
619fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::auto_ptr<Modifier> LM(new LoadModifier(BB, &PT, &R));
620fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::auto_ptr<Modifier> SM(new StoreModifier(BB, &PT, &R));
621fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::auto_ptr<Modifier> EE(new ExtractElementModifier(BB, &PT, &R));
622fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::auto_ptr<Modifier> SHM(new ShuffModifier(BB, &PT, &R));
623fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::auto_ptr<Modifier> IE(new InsertElementModifier(BB, &PT, &R));
624fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::auto_ptr<Modifier> BM(new BinModifier(BB, &PT, &R));
625fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::auto_ptr<Modifier> CM(new CastModifier(BB, &PT, &R));
626fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::auto_ptr<Modifier> SLM(new SelectModifier(BB, &PT, &R));
627fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::auto_ptr<Modifier> PM(new CmpModifier(BB, &PT, &R));
628fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Modifiers.push_back(LM.get());
629fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Modifiers.push_back(SM.get());
630fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Modifiers.push_back(EE.get());
631fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Modifiers.push_back(SHM.get());
632fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Modifiers.push_back(IE.get());
633fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Modifiers.push_back(BM.get());
634fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Modifiers.push_back(CM.get());
635fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Modifiers.push_back(SLM.get());
636fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Modifiers.push_back(PM.get());
637fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
638fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Generate the random instructions
639fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  AllocaModifier AM(BB, &PT, &R); AM.ActN(5); // Throw in a few allocas
640fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  ConstModifier COM(BB, &PT, &R);  COM.ActN(40); // Throw in a few constants
641fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
642fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  for (unsigned i=0; i< SizeCL / Modifiers.size(); ++i)
643fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    for (std::vector<Modifier*>::iterator it = Modifiers.begin(),
644fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem         e = Modifiers.end(); it != e; ++it) {
645fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      (*it)->Act();
646fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
647fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
648fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  SM->ActN(5); // Throw in a few stores.
649fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem}
650fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
651fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemvoid IntroduceControlFlow(Function *F) {
652fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::set<Instruction*> BoolInst;
653fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  for (BasicBlock::iterator it = F->begin()->begin(),
654fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem       e = F->begin()->end(); it != e; ++it) {
655fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (it->getType() == IntegerType::getInt1Ty(F->getContext()))
656fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      BoolInst.insert(it);
657fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
658fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
659fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  for (std::set<Instruction*>::iterator it = BoolInst.begin(),
660fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem       e = BoolInst.end(); it != e; ++it) {
661fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Instruction *Instr = *it;
662fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    BasicBlock *Curr = Instr->getParent();
663fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    BasicBlock::iterator Loc= Instr;
664fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    BasicBlock *Next = Curr->splitBasicBlock(Loc, "CF");
665fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    Instr->moveBefore(Curr->getTerminator());
666fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    if (Curr != &F->getEntryBlock()) {
667fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      BranchInst::Create(Curr, Next, Instr, Curr->getTerminator());
668fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem      Curr->getTerminator()->eraseFromParent();
669fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    }
670fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
671fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem}
672fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
673fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotemint main(int argc, char **argv) {
674fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Init LLVM, call llvm_shutdown() on exit, parse args, etc.
675fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  llvm::PrettyStackTraceProgram X(argc, argv);
676fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  cl::ParseCommandLineOptions(argc, argv, "llvm codegen stress-tester\n");
677fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  llvm_shutdown_obj Y;
678fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
679fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::auto_ptr<Module> M(new Module("/tmp/autogen.bc", getGlobalContext()));
680fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Function *F = GenEmptyFunction(M.get());
681fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  FillFunction(F);
682fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  IntroduceControlFlow(F);
683fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
684fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Figure out what stream we are supposed to write to...
685fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  OwningPtr<tool_output_file> Out;
686fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  // Default to standard output.
687fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  if (OutputFilename.empty())
688fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    OutputFilename = "-";
689fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
690fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  std::string ErrorInfo;
691fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Out.reset(new tool_output_file(OutputFilename.c_str(), ErrorInfo,
692fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem                                 raw_fd_ostream::F_Binary));
693fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  if (!ErrorInfo.empty()) {
694fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    errs() << ErrorInfo << '\n';
695fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem    return 1;
696fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  }
697fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
698fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  PassManager Passes;
699fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Passes.add(createVerifierPass());
700fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Passes.add(createPrintModulePass(&Out->os()));
701fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Passes.run(*M.get());
702fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  Out->keep();
703fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem
704fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem  return 0;
705fdc309cc4e95778d7615f3829917c4fe42086e1eNadav Rotem}
706