lokinet/vendor/cppbackport-master/lib/fs/filestatus.cpp

/* Copyright (c) 2015, Pollard Banknote Limited
   All rights reserved.

   Redistribution and use in source and binary forms, with or without
   modification, are permitted provided that the following conditions are met:

   1. Redistributions of source code must retain the above copyright notice,
   this list of conditions and the following disclaimer.

   2. Redistributions in binary form must reproduce the above copyright notice,
   this list of conditions and the following disclaimer in the documentation
   and/or other materials provided with the distribution.

   3. Neither the name of the copyright holder nor the names of its contributors
   may be used to endorse or promote products derived from this software without
   specific prior written permission.

   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
   AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
   LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
   CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
   SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
   INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
   CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
   ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
   POSSIBILITY OF SUCH DAMAGE.
 */
#include "filestatus.h"

#include <iostream>

#ifndef _MSC_VER
#include <unistd.h>
#endif
#include <sys/stat.h>
#include <sys/types.h>

#include "path.h"

#ifdef _MSC_VER
#include <Windows.h>
#include <stdint.h>
#include <io.h>

typedef unsigned short mode_t;
typedef uint32_t id_t; /* Internal uids/gids are 32-bits */

typedef SSIZE_T ssize_t;
#ifndef _OFF_T_DEFINED
typedef DWORD64 off_t;
#endif
typedef uint32_t uid_t; /* [???] user IDs */

#ifndef S_IFIFO
#define S_IFIFO 0010000 /* [XSI] named pipe (fifo) */
#endif

#ifndef S_IFBLK
#define S_IFBLK 0060000 /* [XSI] block special */
#endif

#ifndef S_IFLNK
#define S_IFLNK 0120000 /* [XSI] symbolic link */
#endif

#ifndef S_IFSOCK
#define S_IFSOCK 0140000 /* [XSI] socket */
#endif

#ifndef S_ISBLK
#define S_ISBLK(m) (((m)&S_IFMT) == S_IFBLK) /* block special */
#endif

#ifndef S_ISCHR
#define S_ISCHR(m) (((m)&S_IFMT) == S_IFCHR) /* char special */
#endif

#ifndef S_ISDIR
#define S_ISDIR(m) (((m)&S_IFMT) == S_IFDIR) /* directory */
#endif

#ifndef S_ISFIFO
#define S_ISFIFO(m) (((m)&S_IFMT) == S_IFIFO) /* fifo or socket */
#endif

#ifndef S_ISREG
#define S_ISREG(m) (((m)&S_IFMT) == S_IFREG) /* regular file */
#endif

#ifndef S_ISLNK
#define S_ISLNK(m) (((m)&S_IFMT) == S_IFLNK) /* symbolic link */
#endif

#ifndef S_ISSOCK
#define S_ISSOCK(m) (((m)&S_IFMT) == S_IFSOCK) /* socket */
#endif

#ifndef makedev
#define makedev(x, y) ((dev_t)(((x) << 24) | (y)))
#endif

#ifndef DT_UNKNOWN
#define DT_UNKNOWN 0
#endif

#ifndef DT_FIFO
#define DT_FIFO 1
#endif

#ifndef DT_CHR
#define DT_CHR 2
#endif

#ifndef DT_DIR
#define DT_DIR 4
#endif

#ifndef DT_BLK
#define DT_BLK 6
#endif

#ifndef DT_REG
#define DT_REG 8
#endif

#ifndef DT_LNK
#define DT_LNK 10
#endif

#ifndef DT_SOCK
#define DT_SOCK 12
#endif

#ifndef DT_WHT
#define DT_WHT 14
#endif
#endif

namespace
{
  ::cpp17::filesystem::file_status
  from_mode_t(mode_t m)
  {
    ::cpp17::filesystem::perms p =
        static_cast<::cpp17::filesystem::perms >(m & 0xFFF);

    ::cpp17::filesystem::file_type t = file_type::unknown;

    if(S_ISREG(m))
    {
      t = file_type::regular;
    }
    else if(S_ISDIR(m))
    {
      t = file_type::directory;
    }
    else if(S_ISCHR(m))
    {
      t = file_type::character;
    }
    else if(S_ISBLK(m))
    {
      t = file_type::block;
    }
    else if(S_ISFIFO(m))
    {
      t = file_type::fifo;
    }
#ifndef _WIN32  // these only work on cygnus or msys2!
    else if(S_ISLNK(m))
    {
      t = file_type::symlink;
    }
    else if(S_ISSOCK(m))
    {
      t = file_type::socket;
    }
#endif
    return ::cpp17::filesystem::file_status(t, p);
  }

}  // namespace

namespace cpp17
{
  namespace filesystem
  {
    file_status::file_status(const file_status& s) : t(s.t), p(s.p)
    {
    }

    file_status::file_status(file_type t_, perms p_) : t(t_), p(p_)
    {
    }

    file_status&
    file_status::operator=(const file_status& s)
    {
      t = s.t;
      p = s.p;

      return *this;
    }

    file_type
    file_status::type() const
    {
      return t;
    }

    void
    file_status::type(file_type t_)
    {
      t = t_;
    }

    perms
    file_status::permissions() const
    {
      return p;
    }

    void
    file_status::permissions(perms p_)
    {
      p = p_;
    }

    std::ostream&
    operator<<(std::ostream& os, const file_status& fs)
    {
      os << fs.type() << "; " << fs.permissions();

      return os;
    }

    file_status
    status(const path& path_)
    {
      if(!path_.empty())
      {
        struct stat st;

        if(::stat(path_.c_str(), &st) == 0)
        {
          return from_mode_t(st.st_mode);
        }
      }

      return file_status();
    }

    file_status
    symlink_status(const path& path_)
    {
      if(!path_.empty())
      {
        struct stat st;
#ifndef _WIN32
        if(::lstat(path_.c_str(), &st) == 0)
#else
        if(::stat(path_.c_str(), &st) == 0)
#endif
        {
          return from_mode_t(st.st_mode);
        }
      }
      return file_status();
    }

    bool
    status_known(file_status s)
    {
      return s.type() != file_type::none;
    }

    bool
    exists(file_status s)
    {
      return status_known(s) && s.type() != file_type::not_found;
    }

    bool
    exists(const path& p, std::error_code& ec)
    {
      return exists(status(p));
    }

    bool
    is_block_file(file_status s)
    {
      return s.type() == file_type::block;
    }

    bool
    is_block_file(const path& p)
    {
      return is_block_file(status(p));
    }

    bool
    is_character_file(file_status s)
    {
      return s.type() == file_type::character;
    }

    bool
    is_character_file(const path& p)
    {
      return is_character_file(status(p));
    }

    bool
    is_fifo(file_status s)
    {
      return s.type() == file_type::fifo;
    }

    bool
    is_fifo(const path& p)
    {
      return is_fifo(status(p));
    }

    bool
    is_other(file_status s)
    {
      return exists(s) && !is_regular_file(s) && !is_directory(s)
          && !is_symlink(s);
    }

    bool
    is_other(const path& p)
    {
      return is_other(status(p));
    }

    bool
    is_regular_file(file_status s)
    {
      return s.type() == file_type::regular;
    }

    bool
    is_regular_file(const path& p)
    {
      return is_regular_file(status(p));
    }

    bool
    is_socket(file_status s)
    {
      return s.type() == file_type::socket;
    }

    bool
    is_socket(const path& p)
    {
      return is_socket(status(p));
    }

    bool
    is_symlink(file_status s)
    {
      return s.type() == file_type::symlink;
    }

    bool
    is_symlink(const path& p)
    {
      return is_symlink(status(p));
    }

    bool
    is_directory(file_status s)
    {
      return s.type() == file_type::directory;
    }

    bool
    is_directory(const path& p)
    {
      return is_directory(status(p));
    }

    std::size_t
    file_size(const path& p)
    {
      if(!p.empty())
      {
        struct stat st;

        if(::stat(p.c_str(), &st) == 0)
        {
          return st.st_size;
        }
      }

      return std::size_t(-1);
    }
  }  // namespace filesystem
}  // namespace cpp17