aboutsummaryrefslogtreecommitdiff
path: root/lldb/tools/lldb-fuzzer/utils/TempFile.cpp
blob: c5c16ec19df7a762672e4dc0c4a780b03c7b8230 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
//===-- TempFile.cpp ------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//

#include "llvm/Support/FileSystem.h"
#include <TempFile.h>

using namespace lldb_fuzzer;
using namespace llvm;

TempFile::~TempFile() {
  if (!m_path.empty())
    sys::fs::remove(m_path.str(), true);
}

std::unique_ptr<TempFile> TempFile::Create(uint8_t *data, size_t size) {
  int fd;
  std::unique_ptr<TempFile> temp_file = std::make_unique<TempFile>();
  std::error_code ec = sys::fs::createTemporaryFile("lldb-fuzzer", "input", fd,
                                                    temp_file->m_path);
  if (ec)
    return nullptr;

  raw_fd_ostream os(fd, true);
  os.write(reinterpret_cast<const char *>(data), size);
  os.close();

  return temp_file;
}