source: trunk/debathena/third/schroot/bin/schroot-listmounts/schroot-listmounts-main.cc @ 24167

/* Copyright © 2005-2007  Roger Leigh <rleigh@debian.org>
 *
 * schroot is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * schroot is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see
 * <http://www.gnu.org/licenses/>.
 *
 *********************************************************************/

#include <config.h>

#include <sbuild/sbuild-mntstream.h>

#include "schroot-listmounts-main.h"

#include <cerrno>
#include <climits>
#include <cstdio>
#include <cstdlib>
#include <ctime>
#include <iostream>
#include <locale>

#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>

#include <boost/format.hpp>

using std::endl;
using boost::format;
using sbuild::_;
using sbuild::N_;
using namespace schroot_listmounts;

namespace
{

  typedef std::pair<main::error_code,const char *> emap;

  /**
   * This is a list of the supported error codes.  It's used to
   * construct the real error codes map.
   */
  emap init_errors[] =
    {
      // TRANSLATORS: %1% = file
      emap(main::FIND,  N_("Failed to find '%1%'"))
    };

}

template<>
sbuild::error<main::error_code>::map_type
sbuild::error<main::error_code>::error_strings
(init_errors,
 init_errors + (sizeof(init_errors) / sizeof(init_errors[0])));

main::main (options::ptr& options):
  schroot_base::main("schroot-listmounts",
                     // TRANSLATORS: '...' is an ellipsis e.g. U+2026,
                     // and '-' is an em-dash.
                     _("[OPTION...] - list mount points"),
                     options,
                     false),
  opts(options)
{
}

main::~main ()
{
}

void
main::action_listmounts ()
{
  std::string to_find = sbuild::normalname(this->opts->mountpoint);

  {
    // NOTE: This is a non-standard GNU extension.
    char *rpath = realpath(to_find.c_str(), NULL);
    if (rpath == 0)
      throw error(to_find, FIND, strerror(errno));

    to_find = rpath;
    free(rpath);
    rpath = 0;
  }

  // Check mounts.
  sbuild::mntstream mounts("/proc/mounts");
  sbuild::mntstream::mntentry entry;
  sbuild::string_list mountlist;

  while (mounts >> entry)
    {
      std::string mount_dir(entry.directory);
      if (to_find == "/" ||
          (mount_dir.find(to_find) == 0 &&
           (// Names are the same.
            mount_dir.size() == to_find.size() ||
            // Must have a following /, or not the same directory.
            (mount_dir.size() > to_find.size() &&
             mount_dir[to_find.size()] == '/'))))
        mountlist.push_back(mount_dir);
    }

  for (sbuild::string_list::const_reverse_iterator pos = mountlist.rbegin();
       pos != mountlist.rend();
       ++pos)
    std::cout << *pos << '\n';
  std::cout << std::flush;
}

int
main::run_impl ()
{
  if (this->opts->action == options::ACTION_HELP)
    action_help(std::cerr);
  else if (this->opts->action == options::ACTION_VERSION)
    action_version(std::cerr);
  else if (this->opts->action == options::ACTION_LISTMOUNTS)
    action_listmounts();
  else
    assert(0); // Invalid action.

  return EXIT_SUCCESS;
}