ldid.hpp

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