ldid.hpp

   1 #ifndef LDID_HPP
   2 #define LDID_HPP
   3
   4 #include <cstdlib>
   5 #include <map>
   6 #include <set>
   7 #include <sstream>
   8 #include <streambuf>
   9 #include <string>
  10 #include <vector>
  11
  12 namespace ldid {
  13
  14 // I wish Apple cared about providing quality toolchains :/
  15
  16 template <typename Function_>
  17 class Functor;
  18
  19 template <typename Type_, typename... Args_>
  20 class Functor<Type_ (Args_...)> {
  21   public:
  22     virtual Type_ operator ()(Args_... args) const = 0;
  23 };
  24
  25 template <typename Function_>
  26 class FunctorImpl;
  27
  28 template <typename Value_, typename Type_, typename... Args_>
  29 class FunctorImpl<Type_ (Value_::*)(Args_...) const> :
  30     public Functor<Type_ (Args_...)>
  31 {
  32   private:
  33     const Value_ *value_;
  34
  35   public:
  36     FunctorImpl(const Value_ &value) :
  37         value_(&value)
  38     {
  39     }
  40
  41     virtual Type_ operator ()(Args_... args) const {
  42         return (*value_)(args...);
  43     }
  44 };
  45
  46 template <typename Function_>
  47 FunctorImpl<decltype(&Function_::operator())> fun(const Function_ &value) {
  48     return value;
  49 }
  50
  51 class Folder {
  52   public:
  53     virtual void Save(const std::string &path, bool edit, const void *flag, const Functor<void (std::streambuf &)> &code) = 0;
  54     virtual bool Look(const std::string &path) const = 0;
  55     virtual void Open(const std::string &path, const Functor<void (std::streambuf &, const void *)> &code) const = 0;
  56     virtual void Find(const std::string &path, const Functor<void (const std::string &)> &code, const Functor<void (const std::string &, const Functor<std::string ()> &)> &link) const = 0;
  57 };
  58
  59 class DiskFolder :
  60     public Folder
  61 {
  62   private:
  63     const std::string path_;
  64     std::map<std::string, std::string> commit_;
  65
  66   protected:
  67     std::string Path(const std::string &path) const;
  68
  69   private:
  70     void Find(const std::string &root, const std::string &base, const Functor<void (const std::string &)> &code, const Functor<void (const std::string &, const Functor<std::string ()> &)> &link) const;
  71
  72   public:
  73     DiskFolder(const std::string &path);
  74     ~DiskFolder();
  75
  76     virtual void Save(const std::string &path, bool edit, const void *flag, const Functor<void (std::streambuf &)> &code);
  77     virtual bool Look(const std::string &path) const;
  78     virtual void Open(const std::string &path, const Functor<void (std::streambuf &, const void *)> &code) const;
  79     virtual void Find(const std::string &path, const Functor<void (const std::string &)> &code, const Functor<void (const std::string &, const Functor<std::string ()> &)> &link) const;
  80 };
  81
  82 class SubFolder :
  83     public Folder
  84 {
  85   private:
  86     Folder &parent_;
  87     std::string path_;
  88
  89   public:
  90     SubFolder(Folder &parent, const std::string &path);
  91
  92     virtual void Save(const std::string &path, bool edit, const void *flag, const Functor<void (std::streambuf &)> &code);
  93     virtual bool Look(const std::string &path) const;
  94     virtual void Open(const std::string &path, const Functor<void (std::streambuf &, const void *)> &code) const;
  95     virtual void Find(const std::string &path, const Functor<void (const std::string &)> &code, const Functor<void (const std::string &, const Functor<std::string ()> &)> &link) const;
  96 };
  97
  98 class UnionFolder :
  99     public Folder
 100 {
 101   private:
 102     class StringBuffer :
 103         public std::stringbuf
 104     {
 105       public:
 106         StringBuffer() {
 107         }
 108
 109         StringBuffer(const StringBuffer &rhs) :
 110             std::stringbuf(rhs.str())
 111         {
 112         }
 113     };
 114
 115     Folder &parent_;
 116     std::set<std::string> deletes_;
 117
 118     std::map<std::string, std::string> remaps_;
 119     mutable std::map<std::string, std::pair<StringBuffer, const void *>> resets_;
 120
 121     std::string Map(const std::string &path) const;
 122     void Map(const std::string &path, const Functor<void (const std::string &)> &code, const std::string &file, const Functor<void (const Functor<void (std::streambuf &, const void *)> &)> &save) const;
 123
 124   public:
 125     UnionFolder(Folder &parent);
 126
 127     virtual void Save(const std::string &path, bool edit, const void *flag, const Functor<void (std::streambuf &)> &code);
 128     virtual bool Look(const std::string &path) const;
 129     virtual void Open(const std::string &path, const Functor<void (std::streambuf &, const void *)> &code) const;
 130     virtual void Find(const std::string &path, const Functor<void (const std::string &)> &code, const Functor<void (const std::string &, const Functor<std::string ()> &)> &link) const;
 131
 132     void operator ()(const std::string &from) {
 133         deletes_.insert(from);
 134     }
 135
 136     void operator ()(const std::string &from, const std::string &to) {
 137         operator ()(from);
 138         remaps_[to] = from;
 139     }
 140
 141     std::stringbuf &operator ()(const std::string &from, const void *flag) {
 142         operator ()(from);
 143         auto &reset(resets_[from]);
 144         reset.second = flag;
 145         return reset.first;
 146     }
 147 };
 148
 149 struct Bundle {
 150     std::string path;
 151     std::vector<char> hash;
 152 };
 153
 154 Bundle Sign(const std::string &root, Folder &folder, const std::string &key, const std::string &requirement, const Functor<std::string (const std::string &, const std::string &)> &alter);
 155
 156 typedef std::map<uint32_t, std::vector<char>> Slots;
 157
 158 std::vector<char> Sign(const void *idata, size_t isize, std::streambuf &output, const std::string &identifier, const std::string &entitlements, const std::string &requirement, const std::string &key, const Slots &slots);
 159
 160 }
 161
 162 #endif//LDID_HPP