service.cpp

// This file is a part of the IncludeOS unikernel - www.includeos.org
//
// Copyright 2015-2016 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 <os>
#include <cstdio>
#include <cassert>
#include <fs/disk.hpp>

// Includes std::string internal_banana
#include "banana.ascii"
static std::shared_ptr<fs::Disk> disk;

const uint64_t SIZE = 128*1024*1024;
const std::string shallow_banana{"/banana.txt"};
const std::string deep_banana{"/dir1/dir2/dir3/dir4/dir5/dir6/banana.txt"};

void is_done() {
  static int counter = 0;
  if (++counter == 3) INFO("FAT32","SUCCESS\n");
}

void test2()
{
  INFO("FAT32", "Remounting disk.");

  CHECKSERT(not disk->empty(), "Disk not empty");
  CHECKSERT(disk->dev().size() == SIZE / 512, "Disk size is %llu bytes", SIZE);

  disk->init_fs(disk->MBR,
  [] (fs::error_t err, auto& fs)
  {
    CHECKSERT(not err, "Filesystem mounted on VBR1");

    fs.stat(shallow_banana,
    [] (auto err, const auto& ent) {
      INFO("FAT32", "Shallow banana");

      CHECKSERT(not err, "Stat %s", shallow_banana.c_str());

      CHECKSERT(ent.is_valid(), "Stat file in root dir");
      CHECKSERT(ent.is_file(), "Entity is file");
      CHECKSERT(!ent.is_dir(), "Entity is not directory");
      CHECKSERT(ent.name() == "banana.txt", "Name is 'banana.txt'");
      is_done();
    });

    fs.read_file(shallow_banana,
    [] (fs::error_t err, fs::buffer_t buf)
    {
      INFO("FAT32", "Read file");
      if (err) {
        printf("Read error: %s\n", err.to_string().c_str());
      }
      CHECKSERT(not err, "read_file: Read %s asynchronously", shallow_banana.c_str());
      printf("%s\n", internal_banana.c_str());
      std::string banana((const char*) buf->data(), buf->size());
      CHECKSERT(banana == internal_banana, "Correct shallow banana");
      is_done();
    });

    fs.stat(deep_banana,
    [] (auto err, const auto& ent) {
      INFO("FAT32", "Deep banana");
      auto& fs = disk->fs();
      CHECKSERT(not err, "Stat %s", deep_banana.c_str());
      CHECKSERT(ent.is_valid(), "Stat file in deep dir");
      CHECKSERT(ent.is_file(), "Entity is file");
      CHECKSERT(!ent.is_dir(), "Entity is not directory");

      CHECKSERT(ent.name() == "banana.txt", "Name is 'banana.txt'");

      // asynch file reading test
      fs.read(ent, 0, ent.size(),
      [] (fs::error_t err, fs::buffer_t buf)
      {
        INFO("FAT32", "Read inside stat");
        CHECKSERT(not err, "read: Read %s asynchronously", deep_banana.c_str());

        std::string banana((const char*) buf->data(), buf->size());
        CHECKSERT(banana == internal_banana, "Correct deep fried banana");
        is_done();
      });
    });

  });
}

void Service::start()
{
  auto& device = hw::Devices::drive(0);
  disk = std::make_shared<fs::Disk> (device);

  INFO("FAT32", "Running tests for FAT32");
  CHECKSERT(disk, "Disk created");

  // which means that the disk can't be empty
  CHECKSERT(not disk->empty(), "Disk not empty");
  // verify that the size is indeed N sectors
  CHECKSERT(disk->dev().size() == SIZE / 512, "Disk size is %llu sectors", SIZE / 512);

  // auto-mount filesystem
  disk->init_fs(
  [] (fs::error_t err, auto& fs)
  {
    CHECKSERT(!err, "Filesystem auto-initializedd");

    const std::string fsname = "FAT16";
    CHECKSERT(fs.name() == fsname, "Filesystem recognized as FAT16");

    fs.ls("/",
    [] (fs::error_t err, auto ents) {
      CHECKSERT(not err, "Listing root directory");
      CHECKSERT(ents->size() == 2, "Exactly two ents in root dir");

      auto& e = ents->at(0);
      CHECKSERT(e.is_file(), "Ent is a file");
      CHECKSERT(e.name() == "banana.txt", "Ents name is 'banana.txt'");

      test2();
    });
  });
}