Paul Lawrence | 6198026 | 2015-03-16 15:35:55 -0700 | [diff] [blame] | 1 | #include <string> |
| 2 | #include <fstream> |
| 3 | |
| 4 | // key names for properties we use |
| 5 | namespace properties { |
| 6 | extern const char* key; |
| 7 | extern const char* ref; |
Paul Lawrence | 2ecc642 | 2015-04-15 14:11:07 -0700 | [diff] [blame] | 8 | extern const char* props; |
Paul Lawrence | bc2eb8c | 2015-04-28 22:13:04 +0000 | [diff] [blame] | 9 | extern const char* is_default; |
Paul Lawrence | 6198026 | 2015-03-16 15:35:55 -0700 | [diff] [blame] | 10 | } |
| 11 | |
| 12 | /** |
| 13 | * Class to store data on the unencrypted folder of a device. |
| 14 | * Note that the folder must exist before this class is constructed. |
| 15 | * All names must be valid single level (no '/') file or directory names |
| 16 | * Data is organized hierarchically so we can get a child folder |
| 17 | */ |
| 18 | class UnencryptedProperties |
| 19 | { |
| 20 | public: |
Paul Lawrence | bc2eb8c | 2015-04-28 22:13:04 +0000 | [diff] [blame] | 21 | // Get path of folder. Must create before using any properties |
| 22 | // This is to allow proper setting of SELinux policy |
| 23 | static std::string GetPath(const char* device); |
| 24 | |
Paul Lawrence | 6198026 | 2015-03-16 15:35:55 -0700 | [diff] [blame] | 25 | // Opens properties folder on named device. |
Paul Lawrence | bc2eb8c | 2015-04-28 22:13:04 +0000 | [diff] [blame] | 26 | // If folder does not exist, OK will return false, all |
Paul Lawrence | 6198026 | 2015-03-16 15:35:55 -0700 | [diff] [blame] | 27 | // getters will return default properties and setters will fail. |
| 28 | UnencryptedProperties(const char* device); |
| 29 | |
| 30 | // Get named object. Return default if object does not exist or error. |
Paul Lawrence | bc2eb8c | 2015-04-28 22:13:04 +0000 | [diff] [blame] | 31 | template<typename t> t Get(const char* name, t default_value = t()) const; |
Paul Lawrence | 6198026 | 2015-03-16 15:35:55 -0700 | [diff] [blame] | 32 | |
| 33 | // Set named object. Return true if success, false otherwise |
| 34 | template<typename t> bool Set(const char* name, t const& value); |
| 35 | |
| 36 | // Get child properties |
Paul Lawrence | bc2eb8c | 2015-04-28 22:13:04 +0000 | [diff] [blame] | 37 | UnencryptedProperties GetChild(const char* name) const; |
Paul Lawrence | 6198026 | 2015-03-16 15:35:55 -0700 | [diff] [blame] | 38 | |
| 39 | // Remove named object |
| 40 | bool Remove(const char* name); |
| 41 | |
Paul Lawrence | bc2eb8c | 2015-04-28 22:13:04 +0000 | [diff] [blame] | 42 | // Does folder exist? |
| 43 | bool OK() const; |
| 44 | |
Paul Lawrence | 6198026 | 2015-03-16 15:35:55 -0700 | [diff] [blame] | 45 | private: |
| 46 | UnencryptedProperties(); |
Paul Lawrence | 6198026 | 2015-03-16 15:35:55 -0700 | [diff] [blame] | 47 | std::string folder_; |
| 48 | }; |
| 49 | |
| 50 | |
| 51 | template<typename t> t UnencryptedProperties::Get(const char* name, |
Paul Lawrence | bc2eb8c | 2015-04-28 22:13:04 +0000 | [diff] [blame] | 52 | t default_value) const |
Paul Lawrence | 6198026 | 2015-03-16 15:35:55 -0700 | [diff] [blame] | 53 | { |
| 54 | if (!OK()) return default_value; |
| 55 | t value = default_value; |
| 56 | std::ifstream(folder_ + "/" + name) >> value; |
| 57 | return value; |
| 58 | } |
| 59 | |
| 60 | template<typename t> bool UnencryptedProperties::Set(const char* name, |
| 61 | t const& value) |
| 62 | { |
| 63 | if (!OK()) return false; |
| 64 | std::ofstream o(folder_ + "/" + name); |
| 65 | o << value; |
| 66 | return !o.fail(); |
| 67 | } |
| 68 | |
| 69 | // Specialized getters/setters for strings |
| 70 | template<> std::string UnencryptedProperties::Get(const char* name, |
Paul Lawrence | bc2eb8c | 2015-04-28 22:13:04 +0000 | [diff] [blame] | 71 | std::string default_value) const; |
Paul Lawrence | 6198026 | 2015-03-16 15:35:55 -0700 | [diff] [blame] | 72 | |
| 73 | template<> bool UnencryptedProperties::Set(const char* name, |
| 74 | std::string const& value); |