test: add tar unit test

Author: ingve

test/CMakeLists.txt

@@ -85,6 +85,7 @@ set(TEST_SOURCES
   ${TEST}/util/unit/percent_encoding_test.cpp
   ${TEST}/util/unit/ringbuffer.cpp
   ${TEST}/util/unit/statman.cpp
+  ${TEST}/util/unit/tar_test.cpp
   ${TEST}/util/unit/uri_test.cpp
 )
 
@@ -208,13 +209,20 @@ endif("${ARCH}" STREQUAL "ARCH_ARMv7")
 if(SILENT_BUILD)
   message(STATUS "NOTE: Building with some warnings turned off")
   set_property(SOURCE ${SOURCES} APPEND_STRING PROPERTY COMPILE_FLAGS
-  "-Wno-unused-variable -Wno-unused-parameter -Wno-sign-compare -Wno-format")
+  " -Wno-unused-variable -Wno-unused-parameter -Wno-sign-compare -Wno-format")
 endif()
 
 add_executable(unittests ${SOURCES})
 install(TARGETS unittests DESTINATION ${TEST})
 install(DIRECTORY lest/include/lest DESTINATION ${CMAKE_INSTALL_PREFIX}/include)
 
+add_custom_command(
+  TARGET unittests PRE_BUILD
+  COMMAND ${CMAKE_COMMAND} -E tar cf ${CMAKE_CURRENT_BINARY_DIR}/test-single.tar ${CMAKE_CURRENT_SOURCE_DIR}/CMakeLists.txt
+  COMMAND ${CMAKE_COMMAND} -E tar cf ${CMAKE_CURRENT_BINARY_DIR}/test-multiple.tar ${CMAKE_CURRENT_SOURCE_DIR}/*.py
+  COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_CURRENT_SOURCE_DIR}/CMakeLists.txt ${CMAKE_CURRENT_BINARY_DIR}/test-invalid.tar
+)
+
 add_custom_target(
   cppcheck
   COMMAND cppcheck

test/util/unit/tar_test.cpp

@@ -0,0 +1,86 @@
+// This file is a part of the IncludeOS unikernel - www.includeos.org
+//
+// Copyright 2015-2017 Oslo and Akershus University College of Applied Sciences
+// and Alfred Bratterud
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include <common.cxx>
+#include <tar>
+#include <sys/mman.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
+
+CASE("Reading single entry tar file")
+{
+  tar::Reader r;
+  struct stat st;
+  int res = stat("test-single.tar", &st);
+  EXPECT(res != -1);
+  size_t size = st.st_size;
+  int fd = open("test-single.tar", O_RDONLY);
+  EXPECT_NOT(fd == -1);
+  const uint8_t *mem = (const uint8_t *)mmap(0, size, PROT_READ, MAP_PRIVATE, fd, 0);
+  tar::Tar tar = r.read_uncompressed(mem, size);
+  auto names = tar.element_names();
+  EXPECT(names.size() == 1);
+  const auto& elements = tar.elements();
+  EXPECT(elements.size() == 1);
+  const auto& e = elements.at(0);
+  EXPECT_NOT(e.is_dir());
+  EXPECT(e.typeflag_is_set());
+  EXPECT_NOT(e.is_empty());
+  EXPECT(e.is_ustar());
+  EXPECT_NOT(e.is_tar_gz());
+  EXPECT(e.num_content_blocks() > 0);
+  EXPECT(e.size() > 0);
+  const auto& name = e.name();
+  EXPECT(name.find("CMakeLists.txt") != std::string::npos);
+  EXPECT_THROWS_AS(auto& missing_element = tar.element("not_there.txt"), tar::Tar_exception);
+  close(fd);
+}
+
+CASE("Reading multiple entry tar file")
+{
+  tar::Reader r;
+  struct stat st;
+  int res = stat("test-multiple.tar", &st);
+  EXPECT(res != -1);
+  size_t size = st.st_size;
+  int fd = open("test-multiple.tar", O_RDONLY);
+  EXPECT_NOT(fd == -1);
+  const uint8_t *mem = (const uint8_t *)mmap(0, size, PROT_READ, MAP_PRIVATE, fd, 0);
+  tar::Tar tar = r.read_uncompressed(mem, size);
+  auto names = tar.element_names();
+  EXPECT(names.size() > 1);
+  const auto& elements = tar.elements();
+  EXPECT(elements.size() > 1);
+  const auto& e = elements.at(elements.size() - 1);
+  EXPECT(e.name().find(".py") != std::string::npos);
+  close(fd);
+}
+
+CASE("Reading invalid tar file")
+{
+  tar::Reader r;
+  struct stat st;
+  int res = stat("test-invalid.tar", &st);
+  EXPECT(res != -1);
+  size_t size = st.st_size;
+  int fd = open("test-invalid.tar", O_RDONLY);
+  EXPECT_NOT(fd == -1);
+  const uint8_t *mem = (const uint8_t *)mmap(0, size, PROT_READ, MAP_PRIVATE, fd, 0);
+  EXPECT_THROWS_AS(tar::Tar tar = r.read_uncompressed(mem, size), tar::Tar_exception);
+  close(fd);
+}