xref: /llvm-project/lldb/tools/lldb-fuzzer/utils/TempFile.cpp (revision 61efe14e21b2c47a848f6d7500ed05af17c64a9a)
1*61efe14eSJonas Devlieghere //===-- TempFile.cpp ------------------------------------------------------===//
2*61efe14eSJonas Devlieghere //
3*61efe14eSJonas Devlieghere // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*61efe14eSJonas Devlieghere // See https://llvm.org/LICENSE.txt for license information.
5*61efe14eSJonas Devlieghere // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*61efe14eSJonas Devlieghere //
7*61efe14eSJonas Devlieghere //===----------------------------------------------------------------------===//
8*61efe14eSJonas Devlieghere 
9*61efe14eSJonas Devlieghere #include "llvm/Support/FileSystem.h"
10*61efe14eSJonas Devlieghere #include <TempFile.h>
11*61efe14eSJonas Devlieghere 
12*61efe14eSJonas Devlieghere using namespace lldb_fuzzer;
13*61efe14eSJonas Devlieghere using namespace llvm;
14*61efe14eSJonas Devlieghere 
~TempFile()15*61efe14eSJonas Devlieghere TempFile::~TempFile() {
16*61efe14eSJonas Devlieghere   if (!m_path.empty())
17*61efe14eSJonas Devlieghere     sys::fs::remove(m_path.str(), true);
18*61efe14eSJonas Devlieghere }
19*61efe14eSJonas Devlieghere 
Create(uint8_t * data,size_t size)20*61efe14eSJonas Devlieghere std::unique_ptr<TempFile> TempFile::Create(uint8_t *data, size_t size) {
21*61efe14eSJonas Devlieghere   int fd;
22*61efe14eSJonas Devlieghere   std::unique_ptr<TempFile> temp_file = std::make_unique<TempFile>();
23*61efe14eSJonas Devlieghere   std::error_code ec = sys::fs::createTemporaryFile("lldb-fuzzer", "input", fd,
24*61efe14eSJonas Devlieghere                                                     temp_file->m_path);
25*61efe14eSJonas Devlieghere   if (ec)
26*61efe14eSJonas Devlieghere     return nullptr;
27*61efe14eSJonas Devlieghere 
28*61efe14eSJonas Devlieghere   raw_fd_ostream os(fd, true);
29*61efe14eSJonas Devlieghere   os.write(reinterpret_cast<const char *>(data), size);
30*61efe14eSJonas Devlieghere   os.close();
31*61efe14eSJonas Devlieghere 
32*61efe14eSJonas Devlieghere   return temp_file;
33*61efe14eSJonas Devlieghere }
34