path: root/lib/Basic/SourceManager.cpp

//===--- SourceManager.cpp - Track and cache source files -----------------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
//  This file implements the SourceManager interface.
//
//===----------------------------------------------------------------------===//

#include "clang/Basic/SourceManager.h"
#include "clang/Basic/SourceManagerInternals.h"
#include "clang/Basic/Diagnostic.h"
#include "clang/Basic/FileManager.h"
#include "llvm/Support/Compiler.h"
#include "llvm/Support/MemoryBuffer.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/System/Path.h"
#include <algorithm>
#include <string>
#include <cstdio>

using namespace clang;
using namespace SrcMgr;
using llvm::MemoryBuffer;

//===----------------------------------------------------------------------===//
// SourceManager Helper Classes
//===----------------------------------------------------------------------===//

struct BufferResult::FailureData {
  const llvm::MemoryBuffer *Buffer;
  const char *FileName;
  std::string ErrorStr;  
};

BufferResult::BufferResult(const BufferResult &Other) {
  if (const llvm::MemoryBuffer *Buffer
                        = Other.Data.dyn_cast<const llvm::MemoryBuffer *>()) {
    Data = Buffer;
    return;
  }
  
  Data = new FailureData(*Other.Data.get<FailureData *>());
}

BufferResult::BufferResult(const char *FileName, llvm::StringRef ErrorStr,
                           const llvm::MemoryBuffer *Buffer) {
  FailureData *FD = new FailureData;
  FD->FileName = FileName;
  FD->ErrorStr = ErrorStr;
  FD->Buffer = Buffer;
  Data = FD;
}

BufferResult::~BufferResult() {
  if (FailureData *FD = Data.dyn_cast<FailureData *>())
    delete FD;
}

bool BufferResult::isInvalid() const {
  return Data.is<FailureData *>();
}

const llvm::MemoryBuffer *BufferResult::getBuffer(Diagnostic &Diags) const {
  llvm::StringRef FileName;
  std::string ErrorMsg;
  const llvm::MemoryBuffer *Result = getBuffer(FileName, ErrorMsg);
  if (!ErrorMsg.empty()) {
    Diags.Report(diag::err_cannot_open_file)
      << FileName << ErrorMsg;
  }
  return Result;
}

const llvm::MemoryBuffer *BufferResult::getBuffer(llvm::StringRef &FileName, 
                                                  std::string &Error) const {
  if (const llvm::MemoryBuffer *Buffer
                                  = Data.dyn_cast<const llvm::MemoryBuffer *>())
    return Buffer;
 
  FailureData *Fail = Data.get<FailureData *>();
  FileName = Fail->FileName;
  Error = Fail->ErrorStr;