1//===- llvm/unittest/Support/FileOutputBuffer.cpp - unit tests ------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#include "llvm/Support/FileOutputBuffer.h"
11#include "llvm/Support/ErrorHandling.h"
12#include "llvm/Support/FileSystem.h"
13#include "llvm/Support/Path.h"
14#include "llvm/Support/raw_ostream.h"
15#include "gtest/gtest.h"
16
17using namespace llvm;
18using namespace llvm::sys;
19
20#define ASSERT_NO_ERROR(x)                                                     \
21  if (std::error_code ASSERT_NO_ERROR_ec = x) {                                \
22    errs() << #x ": did not return errc::success.\n"                           \
23           << "error number: " << ASSERT_NO_ERROR_ec.value() << "\n"           \
24           << "error message: " << ASSERT_NO_ERROR_ec.message() << "\n";       \
25  } else {                                                                     \
26  }
27
28namespace {
29TEST(FileOutputBuffer, Test) {
30  // Create unique temporary directory for these tests
31  SmallString<128> TestDirectory;
32  {
33    ASSERT_NO_ERROR(
34        fs::createUniqueDirectory("FileOutputBuffer-test", TestDirectory));
35  }
36
37  // TEST 1: Verify commit case.
38  SmallString<128> File1(TestDirectory);
39	File1.append("/file1");
40  {
41    std::unique_ptr<FileOutputBuffer> Buffer;
42    ASSERT_NO_ERROR(FileOutputBuffer::create(File1, 8192, Buffer));
43    // Start buffer with special header.
44    memcpy(Buffer->getBufferStart(), "AABBCCDDEEFFGGHHIIJJ", 20);
45    // Write to end of buffer to verify it is writable.
46    memcpy(Buffer->getBufferEnd() - 20, "AABBCCDDEEFFGGHHIIJJ", 20);
47    // Commit buffer.
48    ASSERT_NO_ERROR(Buffer->commit());
49  }
50
51  // Verify file is correct size.
52  uint64_t File1Size;
53  ASSERT_NO_ERROR(fs::file_size(Twine(File1), File1Size));
54  ASSERT_EQ(File1Size, 8192ULL);
55  ASSERT_NO_ERROR(fs::remove(File1.str()));
56
57 	// TEST 2: Verify abort case.
58  SmallString<128> File2(TestDirectory);
59	File2.append("/file2");
60  {
61    std::unique_ptr<FileOutputBuffer> Buffer2;
62    ASSERT_NO_ERROR(FileOutputBuffer::create(File2, 8192, Buffer2));
63    // Fill buffer with special header.
64    memcpy(Buffer2->getBufferStart(), "AABBCCDDEEFFGGHHIIJJ", 20);
65    // Do *not* commit buffer.
66  }
67  // Verify file does not exist (because buffer not committed).
68  bool Exists = false;
69  ASSERT_NO_ERROR(fs::exists(Twine(File2), Exists));
70  EXPECT_FALSE(Exists);
71  ASSERT_NO_ERROR(fs::remove(File2.str()));
72
73  // TEST 3: Verify sizing down case.
74  SmallString<128> File3(TestDirectory);
75	File3.append("/file3");
76  {
77    std::unique_ptr<FileOutputBuffer> Buffer;
78    ASSERT_NO_ERROR(FileOutputBuffer::create(File3, 8192000, Buffer));
79    // Start buffer with special header.
80    memcpy(Buffer->getBufferStart(), "AABBCCDDEEFFGGHHIIJJ", 20);
81    // Write to end of buffer to verify it is writable.
82    memcpy(Buffer->getBufferEnd() - 20, "AABBCCDDEEFFGGHHIIJJ", 20);
83    // Commit buffer, but size down to smaller size
84    ASSERT_NO_ERROR(Buffer->commit(5000));
85  }
86
87  // Verify file is correct size.
88  uint64_t File3Size;
89  ASSERT_NO_ERROR(fs::file_size(Twine(File3), File3Size));
90  ASSERT_EQ(File3Size, 5000ULL);
91  ASSERT_NO_ERROR(fs::remove(File3.str()));
92
93  // TEST 4: Verify file can be made executable.
94  SmallString<128> File4(TestDirectory);
95	File4.append("/file4");
96  {
97    std::unique_ptr<FileOutputBuffer> Buffer;
98    ASSERT_NO_ERROR(FileOutputBuffer::create(File4, 8192, Buffer,
99                                              FileOutputBuffer::F_executable));
100    // Start buffer with special header.
101    memcpy(Buffer->getBufferStart(), "AABBCCDDEEFFGGHHIIJJ", 20);
102    // Commit buffer.
103    ASSERT_NO_ERROR(Buffer->commit());
104  }
105  // Verify file exists and is executable.
106  fs::file_status Status;
107  ASSERT_NO_ERROR(fs::status(Twine(File4), Status));
108  bool IsExecutable = (Status.permissions() & fs::owner_exe);
109  EXPECT_TRUE(IsExecutable);
110  ASSERT_NO_ERROR(fs::remove(File4.str()));
111
112  // Clean up.
113  ASSERT_NO_ERROR(fs::remove(TestDirectory.str()));
114}
115} // anonymous namespace
116