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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
#pragma once
#include <app.h>
#include <boost/uuid/uuid.hpp>
#include <boost/uuid/uuid_generators.hpp>
#include <boost/uuid/uuid_io.hpp>
#include <dbus_singleton.hpp>
#include <cstdio>
#include <fstream>
#include <memory>
namespace crow
{
namespace image_upload
{
std::unique_ptr<sdbusplus::bus::match::match> fwUpdateMatcher;
inline void uploadImageHandler(const crow::Request& req, crow::Response& res,
const std::string& filename)
{
// Only allow one FW update at a time
if (fwUpdateMatcher != nullptr)
{
res.addHeader("Retry-After", "30");
res.result(boost::beast::http::status::service_unavailable);
res.end();
return;
}
// Make this const static so it survives outside this method
static boost::asio::steady_timer timeout(*req.ioService,
std::chrono::seconds(5));
timeout.expires_after(std::chrono::seconds(15));
auto timeoutHandler = [&res](const boost::system::error_code& ec) {
fwUpdateMatcher = nullptr;
if (ec == asio::error::operation_aborted)
{
// expected, we were canceled before the timer completed.
return;
}
BMCWEB_LOG_ERROR << "Timed out waiting for Version interface";
if (ec)
{
BMCWEB_LOG_ERROR << "Async_wait failed " << ec;
return;
}
res.result(boost::beast::http::status::bad_request);
res.jsonValue = {
{"data",
{{"description",
"Version already exists or failed to be extracted"}}},
{"message", "400 Bad Request"},
{"status", "error"}};
res.end();
};
std::function<void(sdbusplus::message::message&)> callback =
[&res](sdbusplus::message::message& m) {
BMCWEB_LOG_DEBUG << "Match fired";
sdbusplus::message::object_path path;
std::vector<std::pair<
std::string,
std::vector<std::pair<std::string, std::variant<std::string>>>>>
interfaces;
m.read(path, interfaces);
if (std::find_if(interfaces.begin(), interfaces.end(),
[](const auto& i) {
return i.first ==
"xyz.openbmc_project.Software.Version";
}) != interfaces.end())
{
timeout.cancel();
std::size_t index = path.str.rfind('/');
if (index != std::string::npos)
{
path.str.erase(0, index + 1);
}
res.jsonValue = {{"data", std::move(path.str)},
{"message", "200 OK"},
{"status", "ok"}};
BMCWEB_LOG_DEBUG << "ending response";
res.end();
fwUpdateMatcher = nullptr;
}
};
fwUpdateMatcher = std::make_unique<sdbusplus::bus::match::match>(
*crow::connections::systemBus,
"interface='org.freedesktop.DBus.ObjectManager',type='signal',"
"member='InterfacesAdded',path='/xyz/openbmc_project/software'",
callback);
std::string filepath(
"/tmp/images/" +
boost::uuids::to_string(boost::uuids::random_generator()()));
BMCWEB_LOG_DEBUG << "Writing file to " << filepath;
std::ofstream out(filepath, std::ofstream::out | std::ofstream::binary |
std::ofstream::trunc);
out << req.body;
out.close();
timeout.async_wait(timeoutHandler);
}
template <typename... Middlewares>
void requestRoutes(Crow<Middlewares...>& app)
{
BMCWEB_ROUTE(app, "/upload/image/<str>")
.requires({"ConfigureComponents", "ConfigureManager"})
.methods("POST"_method,
"PUT"_method)([](const crow::Request& req, crow::Response& res,
const std::string& filename) {
uploadImageHandler(req, res, filename);
});
BMCWEB_ROUTE(app, "/upload/image")
.requires({"ConfigureComponents", "ConfigureManager"})
.methods("POST"_method, "PUT"_method)(
[](const crow::Request& req, crow::Response& res) {
uploadImageHandler(req, res, "");
});
}
} // namespace image_upload
} // namespace crow
|
