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