1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
#include "../include/dchat/Storage.hpp"
#include "../include/dchat/env.hpp"
#include "../include/dchat/FileUtil.hpp"
#include <boost/filesystem/convenience.hpp>
#include <sibs/SafeSerializer.hpp>
#include <sibs/SafeDeserializer.hpp>
#if OS_FAMILY == OS_FAMILY_POSIX
#include <pwd.h>
#endif
namespace dchat
{
boost::filesystem::path getHomeDir()
{
#if OS_FAMILY == OS_FAMILY_POSIX
const char *homeDir = getenv("HOME");
if(!homeDir)
{
passwd *pw = getpwuid(getuid());
homeDir = pw->pw_dir;
}
return boost::filesystem::path(homeDir);
#elif OS_FAMILY == OS_FAMILY_WINDOWS
BOOL ret;
HANDLE hToken;
std::wstring homeDir;
DWORD homeDirLen = MAX_PATH;
homeDir.resize(homeDirLen);
if (!OpenProcessToken(GetCurrentProcess(), TOKEN_READ, &hToken))
throw std::runtime_error("Failed to open process token");
if (!GetUserProfileDirectory(hToken, &homeDir[0], &homeDirLen))
{
CloseHandle(hToken);
throw std::runtime_error("Failed to get home directory");
}
CloseHandle(hToken);
homeDir.resize(wcslen(homeDir.c_str()));
return boost::filesystem::path(homeDir);
#endif
}
boost::filesystem::path getDchatDir()
{
boost::filesystem::path dchatHomeDir = getHomeDir() / ".local" / "share" / "dchat";
boost::filesystem::create_directories(dchatHomeDir);
return dchatHomeDir;
}
boost::filesystem::path getImagesDir()
{
boost::filesystem::path imagesDir = getDchatDir() / "images";
boost::filesystem::create_directories(imagesDir);
return imagesDir;
}
void loadBindsFromFile(LoadBindsCallbackFunc callbackFunc)
{
assert(callbackFunc);
StringView fileContent;
try
{
fileContent = getFileContent(getDchatDir() / "binds");
sibs::SafeDeserializer deserializer((const u8*)fileContent.data, fileContent.size);
while(!deserializer.empty())
{
u8 keySize = deserializer.extract<u8>();
std::string key;
key.resize(keySize);
deserializer.extract((u8*)&key[0], keySize);
u8 valueSize = deserializer.extract<u8>();
std::string value;
value.resize(valueSize);
deserializer.extract((u8*)&value[0], valueSize);
callbackFunc(key, value);
}
}
catch(FileException &e)
{
fprintf(stderr, "Failed to read binds from file, reason: %s\n", e.what());
}
delete[] fileContent.data;
}
void replaceBindsInFile(const std::unordered_map<std::string, std::string> &binds)
{
sibs::SafeSerializer serializer;
for(auto &it : binds)
{
serializer.add((u8)it.first.size());
serializer.add((const u8*)it.first.data(), it.first.size());
serializer.add((u8)it.second.size());
serializer.add((const u8*)it.second.data(), it.second.size());
}
fileReplace(getDchatDir() / "binds", StringView((const char*)serializer.getBuffer().data(), serializer.getBuffer().size()));
}
}
|
