[llvm] r291494 - TarWriter: Fix a bug in Ustar header.

Rui Ueyama via llvm-commits llvm-commits at lists.llvm.org
Tue Jan 10 12:03:20 PST 2017


On it.

On Tue, Jan 10, 2017 at 9:34 AM, Aleksey Shlyapnikov <alekseys at google.com>
wrote:

> This bot is broken too: http://lab.llvm.org:8011/
> builders/sanitizer-x86_64-linux-fast/builds/2015
>
> On Tue, Jan 10, 2017 at 5:29 AM, Rafael Avila de Espindola via
> llvm-commits <llvm-commits at lists.llvm.org> wrote:
>
>> Looks like the unit test has a leak:
>>
>> http://lab.llvm.org:8011/builders/sanitizer-x86_64-linux-
>> bootstrap/builds/611/steps/check-llvm%20asan/logs/stdio
>>
>> Cheers,
>> Rafael
>>
>>
>> Rui Ueyama via llvm-commits <llvm-commits at lists.llvm.org> writes:
>>
>> > Author: ruiu
>> > Date: Mon Jan  9 16:55:00 2017
>> > New Revision: 291494
>> >
>> > URL: http://llvm.org/viewvc/llvm-project?rev=291494&view=rev
>> > Log:
>> > TarWriter: Fix a bug in Ustar header.
>> >
>> > If we split a filename into `Name` and `Prefix`, `Prefix` is at most
>> > 145 bytes. We had a bug that didn't split a path correctly. This bug
>> > was pointed out by Rafael in the post commit review.
>> >
>> > This patch adds a unit test for TarWriter to verify the fix.
>> >
>> > Added:
>> >     llvm/trunk/unittests/Support/TarWriterTest.cpp
>> > Modified:
>> >     llvm/trunk/lib/Support/TarWriter.cpp
>> >     llvm/trunk/unittests/Support/CMakeLists.txt
>> >
>> > Modified: llvm/trunk/lib/Support/TarWriter.cpp
>> > URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Support/
>> TarWriter.cpp?rev=291494&r1=291493&r2=291494&view=diff
>> > ============================================================
>> ==================
>> > --- llvm/trunk/lib/Support/TarWriter.cpp (original)
>> > +++ llvm/trunk/lib/Support/TarWriter.cpp Mon Jan  9 16:55:00 2017
>> > @@ -122,7 +122,7 @@ static void writePaxHeader(raw_fd_ostrea
>> >  static std::pair<StringRef, StringRef> splitPath(StringRef Path) {
>> >    if (Path.size() <= sizeof(UstarHeader::Name))
>> >      return {"", Path};
>> > -  size_t Sep = Path.rfind('/', sizeof(UstarHeader::Name) + 1);
>> > +  size_t Sep = Path.rfind('/', sizeof(UstarHeader::Prefix) + 1);
>> >    if (Sep == StringRef::npos)
>> >      return {"", Path};
>> >    return {Path.substr(0, Sep), Path.substr(Sep + 1)};
>> >
>> > Modified: llvm/trunk/unittests/Support/CMakeLists.txt
>> > URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/Sup
>> port/CMakeLists.txt?rev=291494&r1=291493&r2=291494&view=diff
>> > ============================================================
>> ==================
>> > --- llvm/trunk/unittests/Support/CMakeLists.txt (original)
>> > +++ llvm/trunk/unittests/Support/CMakeLists.txt Mon Jan  9 16:55:00
>> 2017
>> > @@ -43,10 +43,11 @@ add_llvm_unittest(SupportTests
>> >    SpecialCaseListTest.cpp
>> >    StringPool.cpp
>> >    SwapByteOrderTest.cpp
>> > +  TarWriterTest.cpp
>> >    TargetParserTest.cpp
>> > -  Threading.cpp
>> >    ThreadLocalTest.cpp
>> >    ThreadPool.cpp
>> > +  Threading.cpp
>> >    TimerTest.cpp
>> >    TypeNameTest.cpp
>> >    TrailingObjectsTest.cpp
>> >
>> > Added: llvm/trunk/unittests/Support/TarWriterTest.cpp
>> > URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/Sup
>> port/TarWriterTest.cpp?rev=291494&view=auto
>> > ============================================================
>> ==================
>> > --- llvm/trunk/unittests/Support/TarWriterTest.cpp (added)
>> > +++ llvm/trunk/unittests/Support/TarWriterTest.cpp Mon Jan  9 16:55:00
>> 2017
>> > @@ -0,0 +1,88 @@
>> > +//===- llvm/unittest/Support/TarWriterTest.cpp
>> ----------------------------===//
>> > +//
>> > +//                     The LLVM Compiler Infrastructure
>> > +//
>> > +// This file is distributed under the University of Illinois Open
>> Source
>> > +// License. See LICENSE.TXT for details.
>> > +//
>> > +//===------------------------------------------------------
>> ----------------===//
>> > +
>> > +#include "llvm/Support/FileSystem.h"
>> > +#include "llvm/Support/MemoryBuffer.h"
>> > +#include "llvm/Support/TarWriter.h"
>> > +#include "gtest/gtest.h"
>> > +
>> > +using namespace llvm;
>> > +namespace {
>> > +
>> > +struct UstarHeader {
>> > +  char Name[100];
>> > +  char Mode[8];
>> > +  char Uid[8];
>> > +  char Gid[8];
>> > +  char Size[12];
>> > +  char Mtime[12];
>> > +  char Checksum[8];
>> > +  char TypeFlag;
>> > +  char Linkname[100];
>> > +  char Magic[6];
>> > +  char Version[2];
>> > +  char Uname[32];
>> > +  char Gname[32];
>> > +  char DevMajor[8];
>> > +  char DevMinor[8];
>> > +  char Prefix[155];
>> > +  char Pad[12];
>> > +};
>> > +
>> > +class TarWriterTest : public ::testing::Test {};
>> > +
>> > +static UstarHeader create(StringRef Base, StringRef Filename) {
>> > +  // Create a temporary file.
>> > +  SmallString<128> Path;
>> > +  std::error_code EC =
>> > +      sys::fs::createTemporaryFile("TarWriterTest", "tar", Path);
>> > +  EXPECT_FALSE((bool)EC);
>> > +
>> > +  // Create a tar file.
>> > +  Expected<std::unique_ptr<TarWriter>> TarOrErr =
>> TarWriter::create(Path, Base);
>> > +  EXPECT_TRUE((bool)TarOrErr);
>> > +  std::unique_ptr<TarWriter> Tar = std::move(*TarOrErr);
>> > +  Tar->append(Filename, "contents");
>> > +  Tar.release();
>> > +
>> > +  // Read the tar file.
>> > +  ErrorOr<std::unique_ptr<MemoryBuffer>> MBOrErr =
>> MemoryBuffer::getFile(Path);
>> > +  EXPECT_TRUE((bool)MBOrErr);
>> > +  std::unique_ptr<MemoryBuffer> MB = std::move(*MBOrErr);
>> > +  sys::fs::remove(Path);
>> > +  return *reinterpret_cast<const UstarHeader *>(MB->getBufferStart());
>> > +}
>> > +
>> > +TEST_F(TarWriterTest, Basics) {
>> > +  UstarHeader Hdr = create("base", "file");
>> > +  ASSERT_EQ("ustar", StringRef(Hdr.Magic));
>> > +  ASSERT_EQ("00", StringRef(Hdr.Version, 2));
>> > +  ASSERT_EQ("base/file", StringRef(Hdr.Name));
>> > +  ASSERT_EQ("00000000010", StringRef(Hdr.Size));
>> > +}
>> > +
>> > +TEST_F(TarWriterTest, LongFilename) {
>> > +  UstarHeader Hdr1 = create(
>> > +      "012345678", std::string(99, 'x') + "/" + std::string(44, 'x') +
>> "/foo");
>> > +  ASSERT_EQ("foo", StringRef(Hdr1.Name));
>> > +  ASSERT_EQ("012345678/" + std::string(99, 'x') + "/" +
>> std::string(44, 'x'),
>> > +            StringRef(Hdr1.Prefix));
>> > +
>> > +  UstarHeader Hdr2 = create(
>> > +      "012345678", std::string(99, 'x') + "/" + std::string(45, 'x') +
>> "/foo");
>> > +  ASSERT_EQ("foo", StringRef(Hdr2.Name));
>> > +  ASSERT_EQ("012345678/" + std::string(99, 'x') + "/" +
>> std::string(45, 'x'),
>> > +            StringRef(Hdr2.Prefix));
>> > +
>> > +  UstarHeader Hdr3 = create(
>> > +      "012345678", std::string(99, 'x') + "/" + std::string(46, 'x') +
>> "/foo");
>> > +  ASSERT_EQ(std::string(46, 'x') + "/foo", StringRef(Hdr3.Name));
>> > +  ASSERT_EQ("012345678/" + std::string(99, 'x'),
>> StringRef(Hdr3.Prefix));
>> > +}
>> > +}
>> >
>> >
>> > _______________________________________________
>> > llvm-commits mailing list
>> > llvm-commits at lists.llvm.org
>> > http://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-commits
>> _______________________________________________
>> llvm-commits mailing list
>> llvm-commits at lists.llvm.org
>> http://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-commits
>>
>
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20170110/1808b04e/attachment.html>


More information about the llvm-commits mailing list