123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211 |
- // CODYlib -*- mode:c++ -*-
- // Copyright (C) 2020 Nathan Sidwell, nathan@acm.org
- // License: Apache v2.0
- // Cody
- #include "internal.hh"
- // OS
- #include <fcntl.h>
- #include <unistd.h>
- #include <sys/stat.h>
- #include <sys/types.h>
- #if ((defined (__unix__) \
- && defined _POSIX_C_SOURCE \
- && (_POSIX_C_SOURCE - 0) >= 200809L) \
- || (defined (__Apple__) \
- && defined (__ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__) \
- && __ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__ >= 101000))
- // Autoconf test?
- #define HAVE_FSTATAT 1
- #else
- #define HAVE_FSTATAT 0
- #endif
- // Resolver code
- #if __windows__
- inline bool IsDirSep (char c)
- {
- return c == '/' || c == '\\';
- }
- inline bool IsAbsPath (char const *str)
- {
- // IIRC windows has the concept of per-drive current directories,
- // which make drive-using paths confusing. Let's not get into that.
- return IsDirSep (str)
- || (((str[0] >= 'A' && str[0] <= 'Z')
- || (str[0] >= 'a' && str[0] <= 'z'))&& str[1] == ':');
- }
- #else
- inline bool IsDirSep (char c)
- {
- return c == '/';
- }
- inline bool IsAbsPath (char const *str)
- {
- return IsDirSep (str[0]);
- }
- #endif
- constexpr char DIR_SEPARATOR = '/';
- constexpr char DOT_REPLACE = ','; // Replace . directories
- constexpr char COLON_REPLACE = '-'; // Replace : (partition char)
- constexpr char const REPO_DIR[] = "cmi.cache";
- namespace Cody {
- Resolver::~Resolver ()
- {
- }
- char const *Resolver::GetCMISuffix ()
- {
- return "cmi";
- }
- std::string Resolver::GetCMIName (std::string const &module)
- {
- std::string result;
- result.reserve (module.size () + 8);
- bool is_header = false;
- bool is_abs = false;
- if (IsAbsPath (module.c_str ()))
- is_header = is_abs = true;
- else if (module.front () == '.' && IsDirSep (module.c_str ()[1]))
- is_header = true;
- if (is_abs)
- {
- result.push_back ('.');
- result.append (module);
- }
- else
- result = std::move (module);
- if (is_header)
- {
- if (!is_abs)
- result[0] = DOT_REPLACE;
- /* Map .. to DOT_REPLACE, DOT_REPLACE. */
- for (size_t ix = 1; ; ix++)
- {
- ix = result.find ('.', ix);
- if (ix == result.npos)
- break;
- if (ix + 2 > result.size ())
- break;
- if (result[ix + 1] != '.')
- continue;
- if (!IsDirSep (result[ix - 1]))
- continue;
- if (!IsDirSep (result[ix + 2]))
- continue;
- result[ix] = DOT_REPLACE;
- result[ix + 1] = DOT_REPLACE;
- }
- }
- else if (COLON_REPLACE != ':')
- {
- // There can only be one colon in a module name
- auto colon = result.find (':');
- if (colon != result.npos)
- result[colon] = COLON_REPLACE;
- }
- if (char const *suffix = GetCMISuffix ())
- {
- result.push_back ('.');
- result.append (suffix);
- }
- return result;
- }
- void Resolver::WaitUntilReady (Server *)
- {
- }
- Resolver *Resolver::ConnectRequest (Server *s, unsigned version,
- std::string &, std::string &)
- {
- if (version > Version)
- s->ErrorResponse ("version mismatch");
- else
- s->ConnectResponse ("default");
- return this;
- }
- int Resolver::ModuleRepoRequest (Server *s)
- {
- s->PathnameResponse (REPO_DIR);
- return 0;
- }
- // Deprecated resolver functions
- int Resolver::ModuleExportRequest (Server *s, Flags, std::string &module)
- {
- auto cmi = GetCMIName (module);
- s->PathnameResponse (cmi);
- return 0;
- }
- int Resolver::ModuleImportRequest (Server *s, Flags, std::string &module)
- {
- auto cmi = GetCMIName (module);
- s->PathnameResponse (cmi);
- return 0;
- }
- int Resolver::ModuleCompiledRequest (Server *s, Flags, std::string &)
- {
- s->OKResponse ();
- return 0;
- }
- int Resolver::IncludeTranslateRequest (Server *s, Flags, std::string &include)
- {
- bool xlate = false;
- // This is not the most efficient
- auto cmi = GetCMIName (include);
- struct stat statbuf;
- #if HAVE_FSTATAT
- int fd_dir = open (REPO_DIR, O_RDONLY | O_CLOEXEC | O_DIRECTORY);
- if (fd_dir >= 0
- && fstatat (fd_dir, cmi.c_str (), &statbuf, 0) == 0
- && S_ISREG (statbuf.st_mode))
- // Sadly can't easily check if this process has read access,
- // except by trying to open it.
- xlate = true;
- if (fd_dir >= 0)
- close (fd_dir);
- #else
- std::string append = REPO_DIR;
- append.push_back (DIR_SEPARATOR);
- append.append (cmi);
- if (stat (append.c_str (), &statbuf) == 0
- || S_ISREG (statbuf.st_mode))
- xlate = true;
- #endif
- if (xlate)
- s->PathnameResponse (cmi);
- else
- s->BoolResponse (false);
- return 0;
- }
- void Resolver::ErrorResponse (Server *server, std::string &&msg)
- {
- server->ErrorResponse (msg);
- }
- }
|