RawFs.cc

/*
 * Copyright (c) 2008-2014, Pedigree Developers
 *
 * Please see the CONTRIB file in the root of the source tree for a full
 * list of contributors.
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */
 
#include "RawFs.h"
#include "pedigree/kernel/Log.h"
#include "pedigree/kernel/machine/Device.h"
#include "pedigree/kernel/panic.h"
#include "pedigree/kernel/processor/types.h"
#include "pedigree/kernel/utilities/new"
 
#include "RawFsDir.h"
#include "RawFsFile.h"
#include "modules/Module.h"
#include "modules/system/vfs/VFS.h"
 
class Disk;
class File;
 
static RawFs* g_pRawFs = 0;
 
RawFsDir::~RawFsDir() = default;
 
RawFs::RawFs() : m_pRoot(0) {
  m_pRoot = new RawFsDir(String(""), this, 0);
}
 
RawFs::~RawFs() {
  if (shutdown() != SyncStatus::Success)
    panic("rawfs: filesystem shutdown failed");
  if (m_pRoot)
    delete m_pRoot;
}
 
Filesystem::SyncStatus RawFs::sync() {
  return m_pRoot->syncFiles(false) ? SyncStatus::Success : SyncStatus::IoError;
}
 
Filesystem::SyncStatus RawFs::shutdown() {
  return m_pRoot->syncFiles(true) ? SyncStatus::Success : SyncStatus::IoError;
}
 
File* RawFs::getRoot() const {
  return m_pRoot;
}
 
static bool hasDiskChildren(Device* pDev) {
  for (size_t i = 0; i < pDev->getNumChildren(); i++) {
    Device* pChild = pDev->getChild(i);
    if (pChild->getType() == Device::Disk)
      return true;
    if (hasDiskChildren(pChild))
      return true;
  }
  return false;
}
 
static void searchNode(Device* pDev, RawFsDir* pDir) {
  for (size_t i = 0; i < pDev->getNumChildren(); i++) {
    Device* pChild = pDev->getChild(i);
    if (pChild->getType() == Device::Disk) {
      String str;
      pChild->getName(str);
      if (hasDiskChildren(pChild)) {
        NOTICE("rawfs: adding dir '" << str << "'");
        RawFsDir* pDir2 = new RawFsDir(str, g_pRawFs, static_cast<File*>(pDir));
        pDir->addEntry(pDir2);
        pDir->addEntry(new RawFsFile(String("entire-disk"), g_pRawFs, static_cast<File*>(pDir),
                                     reinterpret_cast<Disk*>(pChild)));
        searchNode(pChild, pDir2);
      } else {
        NOTICE("rawfs: adding file '" << str << "'");
        pDir->addEntry(new RawFsFile(str, g_pRawFs, static_cast<File*>(pDir),
                                     reinterpret_cast<Disk*>(pChild)));
      }
    } else
      searchNode(pChild, pDir);
  }
}
 
static void rescanTree() {
  RawFsDir* pD = static_cast<RawFsDir*>(g_pRawFs->getRoot());
  pD->removeRecursive();
 
  // Find the disk topology in an intermediate form.
  // searchNode(&Device::root(), pD);
}
 
static bool init() {
  g_pRawFs = new RawFs();
  VFS::instance().addMountCallback(&rescanTree);
  rescanTree();
  VFS::instance().registerFilesystem(g_pRawFs, String("raw"));
 
  return true;
}
 
static void destroy() {
  if (!VFS::instance().removeMountCallback(&rescanTree)) {
    panic("rawfs mount callback was not registered during unload");
  }
  if (!VFS::instance().unregisterFilesystem(g_pRawFs, true, true))
    panic("rawfs: filesystem retirement failed");
  g_pRawFs = nullptr;
}
 
MODULE_INFO_RUNTIME_PINNED("rawfs", &init, &destroy, "vfs");