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
106
107
108
109
|
#include "config.h"
#include <experimental/filesystem>
#include <cereal/archives/json.hpp>
#include <fstream>
#include "serialize.hpp"
#include <sdbusplus/server.hpp>
namespace phosphor
{
namespace software
{
namespace updater
{
namespace fs = std::experimental::filesystem;
void storeToFile(std::string versionId, uint8_t priority)
{
auto bus = sdbusplus::bus::new_default();
if (!fs::is_directory(PERSIST_DIR))
{
fs::create_directories(PERSIST_DIR);
}
std::string path = PERSIST_DIR + versionId;
std::ofstream os(path.c_str());
cereal::JSONOutputArchive oarchive(os);
oarchive(cereal::make_nvp("priority", priority));
std::string serviceFile = "obmc-flash-bmc-setenv@" + versionId + "\\x3d" +
std::to_string(priority) + ".service";
auto method = bus.new_method_call(SYSTEMD_BUSNAME, SYSTEMD_PATH,
SYSTEMD_INTERFACE, "StartUnit");
method.append(serviceFile, "replace");
bus.call_noreply(method);
}
bool restoreFromFile(std::string versionId, uint8_t& priority)
{
std::string path = PERSIST_DIR + versionId;
if (fs::exists(path))
{
std::ifstream is(path.c_str(), std::ios::in);
try
{
cereal::JSONInputArchive iarchive(is);
iarchive(cereal::make_nvp("priority", priority));
return true;
}
catch (cereal::RapidJSONException& e)
{
fs::remove(path);
}
}
// Find the mtd device "u-boot-env" to retrieve the environment variables
std::ifstream mtdDevices("/proc/mtd");
std::string device, devicePath;
try
{
while (std::getline(mtdDevices, device))
{
if (device.find("u-boot-env") != std::string::npos)
{
devicePath = "/dev/" + device.substr(0, device.find(':'));
break;
}
}
if (!devicePath.empty())
{
std::ifstream input(devicePath.c_str());
std::string envVars;
std::getline(input, envVars);
std::string versionVar = versionId + "=";
auto varPosition = envVars.find(versionVar);
if (varPosition != std::string::npos)
{
// Grab the environment variable for this versionId. These
// variables follow the format "versionId=priority\0"
auto var = envVars.substr(varPosition);
priority = std::stoi(var.substr(versionVar.length()));
return true;
}
}
}
catch (const std::exception& e)
{
}
return false;
}
void removeFile(std::string versionId)
{
std::string path = PERSIST_DIR + versionId;
if (fs::exists(path))
{
fs::remove(path);
}
}
} // namespace phosphor
} // namespace software
} // namespace openpower
|
