forked from OSchip/llvm-project
				
			
		
			
				
	
	
		
			115 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			115 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			C++
		
	
	
	
//===- llvm-reduce.cpp - The LLVM Delta Reduction utility -----------------===//
 | 
						|
//
 | 
						|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 | 
						|
// See https://llvm.org/LICENSE.txt for license information.
 | 
						|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 | 
						|
//
 | 
						|
//===----------------------------------------------------------------------===//
 | 
						|
//
 | 
						|
// This program tries to reduce an IR test case for a given interesting-ness
 | 
						|
// test. It runs multiple delta debugging passes in order to minimize the input
 | 
						|
// file. It's worth noting that this is a part of the bugpoint redesign
 | 
						|
// proposal, and thus a *temporary* tool that will eventually be integrated
 | 
						|
// into the bugpoint tool itself.
 | 
						|
//
 | 
						|
//===----------------------------------------------------------------------===//
 | 
						|
 | 
						|
#include "DeltaManager.h"
 | 
						|
#include "llvm/ADT/SmallString.h"
 | 
						|
#include "llvm/IR/LLVMContext.h"
 | 
						|
#include "llvm/IR/Verifier.h"
 | 
						|
#include "llvm/IRReader/IRReader.h"
 | 
						|
#include "llvm/Support/CommandLine.h"
 | 
						|
#include "llvm/Support/InitLLVM.h"
 | 
						|
#include "llvm/Support/SourceMgr.h"
 | 
						|
#include "llvm/Support/raw_ostream.h"
 | 
						|
#include <system_error>
 | 
						|
#include <vector>
 | 
						|
 | 
						|
using namespace llvm;
 | 
						|
 | 
						|
static cl::opt<bool> Help("h", cl::desc("Alias for -help"), cl::Hidden);
 | 
						|
static cl::opt<bool> Version("v", cl::desc("Alias for -version"), cl::Hidden);
 | 
						|
 | 
						|
static cl::opt<std::string> InputFilename(cl::Positional, cl::Required,
 | 
						|
                                          cl::desc("<input llvm ll/bc file>"));
 | 
						|
 | 
						|
static cl::opt<std::string>
 | 
						|
    TestFilename("test", cl::Required,
 | 
						|
                 cl::desc("Name of the interesting-ness test to be run"));
 | 
						|
 | 
						|
static cl::list<std::string>
 | 
						|
    TestArguments("test-arg", cl::ZeroOrMore,
 | 
						|
                  cl::desc("Arguments passed onto the interesting-ness test"));
 | 
						|
 | 
						|
static cl::opt<std::string>
 | 
						|
    OutputFilename("output",
 | 
						|
                   cl::desc("Specify the output file. default: reduced.ll"));
 | 
						|
static cl::alias OutputFileAlias("o", cl::desc("Alias for -output"),
 | 
						|
                                 cl::aliasopt(OutputFilename));
 | 
						|
 | 
						|
static cl::opt<bool>
 | 
						|
    ReplaceInput("in-place",
 | 
						|
                 cl::desc("WARNING: This option will replace your input file "
 | 
						|
                          "with the reduced version!"));
 | 
						|
 | 
						|
// Parses IR into a Module and verifies it
 | 
						|
static std::unique_ptr<Module> parseInputFile(StringRef Filename,
 | 
						|
                                              LLVMContext &Ctxt) {
 | 
						|
  SMDiagnostic Err;
 | 
						|
  std::unique_ptr<Module> Result = parseIRFile(Filename, Err, Ctxt);
 | 
						|
  if (!Result) {
 | 
						|
    Err.print("llvm-reduce", errs());
 | 
						|
    return Result;
 | 
						|
  }
 | 
						|
 | 
						|
  if (verifyModule(*Result, &errs())) {
 | 
						|
    errs() << "Error: " << Filename << " - input module is broken!\n";
 | 
						|
    return std::unique_ptr<Module>();
 | 
						|
  }
 | 
						|
 | 
						|
  return Result;
 | 
						|
}
 | 
						|
 | 
						|
int main(int argc, char **argv) {
 | 
						|
  InitLLVM X(argc, argv);
 | 
						|
 | 
						|
  cl::ParseCommandLineOptions(argc, argv, "LLVM automatic testcase reducer.\n");
 | 
						|
 | 
						|
  LLVMContext Context;
 | 
						|
  std::unique_ptr<Module> OriginalProgram =
 | 
						|
      parseInputFile(InputFilename, Context);
 | 
						|
 | 
						|
  // Initialize test environment
 | 
						|
  TestRunner Tester(TestFilename, TestArguments);
 | 
						|
  Tester.setProgram(std::move(OriginalProgram));
 | 
						|
 | 
						|
  // Try to reduce code
 | 
						|
  runDeltaPasses(Tester);
 | 
						|
 | 
						|
  if (!Tester.getProgram()) {
 | 
						|
    errs() << "\nCouldnt reduce input :/\n";
 | 
						|
  } else {
 | 
						|
    // Print reduced file to STDOUT
 | 
						|
    if (OutputFilename == "-")
 | 
						|
      Tester.getProgram()->print(outs(), nullptr);
 | 
						|
    else {
 | 
						|
      if (ReplaceInput) // In-place
 | 
						|
        OutputFilename = InputFilename.c_str();
 | 
						|
      else if (OutputFilename.empty())
 | 
						|
        OutputFilename = "reduced.ll";
 | 
						|
 | 
						|
      std::error_code EC;
 | 
						|
      raw_fd_ostream Out(OutputFilename, EC);
 | 
						|
      if (EC) {
 | 
						|
        errs() << "Error opening output file: " << EC.message() << "!\n";
 | 
						|
        exit(1);
 | 
						|
      }
 | 
						|
      Tester.getProgram()->print(Out, /*AnnotationWriter=*/nullptr);
 | 
						|
      errs() << "\nDone reducing! Reduced testcase: " << OutputFilename << "\n";
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  return 0;
 | 
						|
}
 |