xref: /openbmc/pldm/pldmd/pldmd.cpp (revision 3d34b2c2)
1 #include "libpldm/base.h"
2 #include "libpldm/bios.h"
3 #include "libpldm/pdr.h"
4 #include "libpldm/platform.h"
5 
6 #include "common/utils.hpp"
7 #include "dbus_impl_requester.hpp"
8 #include "fw-update/manager.hpp"
9 #include "invoker.hpp"
10 #include "requester/handler.hpp"
11 #include "requester/mctp_endpoint_discovery.hpp"
12 #include "requester/request.hpp"
13 
14 #include <err.h>
15 #include <getopt.h>
16 #include <poll.h>
17 #include <stdlib.h>
18 #include <sys/socket.h>
19 #include <sys/types.h>
20 #include <sys/un.h>
21 #include <unistd.h>
22 
23 #include <sdeventplus/event.hpp>
24 #include <sdeventplus/source/io.hpp>
25 
26 #include <cstdio>
27 #include <cstring>
28 #include <fstream>
29 #include <iomanip>
30 #include <iostream>
31 #include <iterator>
32 #include <memory>
33 #include <optional>
34 #include <sstream>
35 #include <stdexcept>
36 #include <string>
37 #include <vector>
38 
39 #ifdef LIBPLDMRESPONDER
40 #include "dbus_impl_pdr.hpp"
41 #include "host-bmc/dbus_to_event_handler.hpp"
42 #include "host-bmc/dbus_to_host_effecters.hpp"
43 #include "host-bmc/host_condition.hpp"
44 #include "host-bmc/host_pdr_handler.hpp"
45 #include "libpldmresponder/base.hpp"
46 #include "libpldmresponder/bios.hpp"
47 #include "libpldmresponder/fru.hpp"
48 #include "libpldmresponder/oem_handler.hpp"
49 #include "libpldmresponder/platform.hpp"
50 #include "xyz/openbmc_project/PLDM/Event/server.hpp"
51 #endif
52 
53 #ifdef OEM_IBM
54 #include "libpldmresponder/file_io.hpp"
55 #include "libpldmresponder/oem_ibm_handler.hpp"
56 #endif
57 
58 constexpr uint8_t MCTP_MSG_TYPE_PLDM = 1;
59 
60 using namespace pldm;
61 using namespace sdeventplus;
62 using namespace sdeventplus::source;
63 using namespace pldm::responder;
64 using namespace pldm::utils;
65 
66 static std::optional<Response>
67     processRxMsg(const std::vector<uint8_t>& requestMsg, Invoker& invoker,
68                  requester::Handler<requester::Request>& handler,
69                  fw_update::Manager* fwManager)
70 {
71     using type = uint8_t;
72     uint8_t eid = requestMsg[0];
73     pldm_header_info hdrFields{};
74     auto hdr = reinterpret_cast<const pldm_msg_hdr*>(
75         requestMsg.data() + sizeof(eid) + sizeof(type));
76     if (PLDM_SUCCESS != unpack_pldm_header(hdr, &hdrFields))
77     {
78         std::cerr << "Empty PLDM request header \n";
79         return std::nullopt;
80     }
81 
82     if (PLDM_RESPONSE != hdrFields.msg_type)
83     {
84         Response response;
85         auto request = reinterpret_cast<const pldm_msg*>(hdr);
86         size_t requestLen = requestMsg.size() - sizeof(struct pldm_msg_hdr) -
87                             sizeof(eid) - sizeof(type);
88         try
89         {
90             if (hdrFields.pldm_type != PLDM_FWUP)
91             {
92                 response =
93                     invoker.handle(hdrFields.pldm_type, hdrFields.command,
94                                    request, requestLen);
95             }
96             else
97             {
98                 response = fwManager->handleRequest(eid, hdrFields.command,
99                                                     request, requestLen);
100             }
101         }
102         catch (const std::out_of_range& e)
103         {
104             uint8_t completion_code = PLDM_ERROR_UNSUPPORTED_PLDM_CMD;
105             response.resize(sizeof(pldm_msg_hdr));
106             auto responseHdr = reinterpret_cast<pldm_msg_hdr*>(response.data());
107             pldm_header_info header{};
108             header.msg_type = PLDM_RESPONSE;
109             header.instance = hdrFields.instance;
110             header.pldm_type = hdrFields.pldm_type;
111             header.command = hdrFields.command;
112             if (PLDM_SUCCESS != pack_pldm_header(&header, responseHdr))
113             {
114                 std::cerr << "Failed adding response header \n";
115                 return std::nullopt;
116             }
117             response.insert(response.end(), completion_code);
118         }
119         return response;
120     }
121     else if (PLDM_RESPONSE == hdrFields.msg_type)
122     {
123         auto response = reinterpret_cast<const pldm_msg*>(hdr);
124         size_t responseLen = requestMsg.size() - sizeof(struct pldm_msg_hdr) -
125                              sizeof(eid) - sizeof(type);
126         handler.handleResponse(eid, hdrFields.instance, hdrFields.pldm_type,
127                                hdrFields.command, response, responseLen);
128     }
129     return std::nullopt;
130 }
131 
132 void optionUsage(void)
133 {
134     std::cerr << "Usage: pldmd [options]\n";
135     std::cerr << "Options:\n";
136     std::cerr
137         << "  --verbose=<0/1>  0 - Disable verbosity, 1 - Enable verbosity\n";
138     std::cerr << "Defaulted settings:  --verbose=0 \n";
139 }
140 
141 int main(int argc, char** argv)
142 {
143 
144     bool verbose = false;
145     static struct option long_options[] = {
146         {"verbose", required_argument, 0, 'v'}, {0, 0, 0, 0}};
147 
148     auto argflag = getopt_long(argc, argv, "v:", long_options, nullptr);
149     switch (argflag)
150     {
151         case 'v':
152             switch (std::stoi(optarg))
153             {
154                 case 0:
155                     verbose = false;
156                     break;
157                 case 1:
158                     verbose = true;
159                     break;
160                 default:
161                     optionUsage();
162                     exit(EXIT_FAILURE);
163             }
164             break;
165         case -1:
166             break;
167         default:
168             exit(EXIT_FAILURE);
169     }
170 
171     /* Create local socket. */
172     int returnCode = 0;
173     int sockfd = socket(AF_UNIX, SOCK_SEQPACKET, 0);
174     if (-1 == sockfd)
175     {
176         returnCode = -errno;
177         std::cerr << "Failed to create the socket, RC= " << returnCode << "\n";
178         exit(EXIT_FAILURE);
179     }
180 
181     auto event = Event::get_default();
182     auto& bus = pldm::utils::DBusHandler::getBus();
183     sdbusplus::server::manager::manager objManager(
184         bus, "/xyz/openbmc_project/software");
185     dbus_api::Requester dbusImplReq(bus, "/xyz/openbmc_project/pldm");
186 
187     Invoker invoker{};
188     requester::Handler<requester::Request> reqHandler(sockfd, event,
189                                                       dbusImplReq, verbose);
190 #ifdef LIBPLDMRESPONDER
191     using namespace pldm::state_sensor;
192     dbus_api::Host dbusImplHost(bus, "/xyz/openbmc_project/pldm");
193     std::unique_ptr<pldm_pdr, decltype(&pldm_pdr_destroy)> pdrRepo(
194         pldm_pdr_init(), pldm_pdr_destroy);
195     std::unique_ptr<pldm_entity_association_tree,
196                     decltype(&pldm_entity_association_tree_destroy)>
197         entityTree(pldm_entity_association_tree_init(),
198                    pldm_entity_association_tree_destroy);
199     std::unique_ptr<pldm_entity_association_tree,
200                     decltype(&pldm_entity_association_tree_destroy)>
201         bmcEntityTree(pldm_entity_association_tree_init(),
202                       pldm_entity_association_tree_destroy);
203     std::shared_ptr<HostPDRHandler> hostPDRHandler;
204     std::unique_ptr<pldm::host_effecters::HostEffecterParser>
205         hostEffecterParser;
206     std::unique_ptr<DbusToPLDMEvent> dbusToPLDMEventHandler;
207     DBusHandler dbusHandler;
208     auto hostEID = pldm::utils::readHostEID();
209     if (hostEID)
210     {
211         hostPDRHandler = std::make_shared<HostPDRHandler>(
212             sockfd, hostEID, event, pdrRepo.get(), EVENTS_JSONS_DIR,
213             entityTree.get(), bmcEntityTree.get(), dbusImplReq, &reqHandler);
214         // HostFirmware interface needs access to hostPDR to know if host
215         // is running
216         dbusImplHost.setHostPdrObj(hostPDRHandler);
217 
218         hostEffecterParser =
219             std::make_unique<pldm::host_effecters::HostEffecterParser>(
220                 &dbusImplReq, sockfd, pdrRepo.get(), &dbusHandler,
221                 HOST_JSONS_DIR, &reqHandler);
222         dbusToPLDMEventHandler = std::make_unique<DbusToPLDMEvent>(
223             sockfd, hostEID, dbusImplReq, &reqHandler);
224     }
225     std::unique_ptr<oem_platform::Handler> oemPlatformHandler{};
226 
227 #ifdef OEM_IBM
228     std::unique_ptr<pldm::responder::CodeUpdate> codeUpdate =
229         std::make_unique<pldm::responder::CodeUpdate>(&dbusHandler);
230     codeUpdate->clearDirPath(LID_STAGING_DIR);
231     oemPlatformHandler = std::make_unique<oem_ibm_platform::Handler>(
232         &dbusHandler, codeUpdate.get(), sockfd, hostEID, dbusImplReq, event,
233         &reqHandler);
234     codeUpdate->setOemPlatformHandler(oemPlatformHandler.get());
235     invoker.registerHandler(PLDM_OEM, std::make_unique<oem_ibm::Handler>(
236                                           oemPlatformHandler.get(), sockfd,
237                                           hostEID, &dbusImplReq, &reqHandler));
238 #endif
239     invoker.registerHandler(
240         PLDM_BIOS, std::make_unique<bios::Handler>(sockfd, hostEID,
241                                                    &dbusImplReq, &reqHandler));
242     auto fruHandler = std::make_unique<fru::Handler>(
243         FRU_JSONS_DIR, FRU_MASTER_JSON, pdrRepo.get(), entityTree.get(),
244         bmcEntityTree.get());
245     // FRU table is built lazily when a FRU command or Get PDR command is
246     // handled. To enable building FRU table, the FRU handler is passed to the
247     // Platform handler.
248     auto platformHandler = std::make_unique<platform::Handler>(
249         &dbusHandler, PDR_JSONS_DIR, pdrRepo.get(), hostPDRHandler.get(),
250         dbusToPLDMEventHandler.get(), fruHandler.get(),
251         oemPlatformHandler.get(), event, true);
252 #ifdef OEM_IBM
253     pldm::responder::oem_ibm_platform::Handler* oemIbmPlatformHandler =
254         dynamic_cast<pldm::responder::oem_ibm_platform::Handler*>(
255             oemPlatformHandler.get());
256     oemIbmPlatformHandler->setPlatformHandler(platformHandler.get());
257 #endif
258 
259     invoker.registerHandler(PLDM_PLATFORM, std::move(platformHandler));
260     invoker.registerHandler(
261         PLDM_BASE,
262         std::make_unique<base::Handler>(hostEID, dbusImplReq, event,
263                                         oemPlatformHandler.get(), &reqHandler));
264     invoker.registerHandler(PLDM_FRU, std::move(fruHandler));
265     dbus_api::Pdr dbusImplPdr(bus, "/xyz/openbmc_project/pldm", pdrRepo.get());
266     sdbusplus::xyz::openbmc_project::PLDM::server::Event dbusImplEvent(
267         bus, "/xyz/openbmc_project/pldm");
268 
269 #endif
270 
271     pldm::utils::CustomFD socketFd(sockfd);
272 
273     struct sockaddr_un addr
274     {};
275     addr.sun_family = AF_UNIX;
276     const char path[] = "\0mctp-mux";
277     memcpy(addr.sun_path, path, sizeof(path) - 1);
278     int result = connect(socketFd(), reinterpret_cast<struct sockaddr*>(&addr),
279                          sizeof(path) + sizeof(addr.sun_family) - 1);
280     if (-1 == result)
281     {
282         returnCode = -errno;
283         std::cerr << "Failed to connect to the socket, RC= " << returnCode
284                   << "\n";
285         exit(EXIT_FAILURE);
286     }
287 
288     result = write(socketFd(), &MCTP_MSG_TYPE_PLDM, sizeof(MCTP_MSG_TYPE_PLDM));
289     if (-1 == result)
290     {
291         returnCode = -errno;
292         std::cerr << "Failed to send message type as pldm to mctp, RC= "
293                   << returnCode << "\n";
294         exit(EXIT_FAILURE);
295     }
296 
297     std::unique_ptr<fw_update::Manager> fwManager =
298         std::make_unique<fw_update::Manager>(event, reqHandler, dbusImplReq);
299     std::unique_ptr<MctpDiscovery> mctpDiscoveryHandler =
300         std::make_unique<MctpDiscovery>(bus, fwManager.get());
301 
302     auto callback = [verbose, &invoker, &reqHandler,
303                      &fwManager](IO& io, int fd, uint32_t revents) {
304         if (!(revents & EPOLLIN))
305         {
306             return;
307         }
308 
309         // Outgoing message.
310         struct iovec iov[2]{};
311 
312         // This structure contains the parameter information for the response
313         // message.
314         struct msghdr msg
315         {};
316 
317         int returnCode = 0;
318         ssize_t peekedLength = recv(fd, nullptr, 0, MSG_PEEK | MSG_TRUNC);
319         if (0 == peekedLength)
320         {
321             // MCTP daemon has closed the socket this daemon is connected to.
322             // This may or may not be an error scenario, in either case the
323             // recovery mechanism for this daemon is to restart, and hence exit
324             // the event loop, that will cause this daemon to exit with a
325             // failure code.
326             io.get_event().exit(0);
327         }
328         else if (peekedLength <= -1)
329         {
330             returnCode = -errno;
331             std::cerr << "recv system call failed, RC= " << returnCode << "\n";
332         }
333         else
334         {
335             std::vector<uint8_t> requestMsg(peekedLength);
336             auto recvDataLength = recv(
337                 fd, static_cast<void*>(requestMsg.data()), peekedLength, 0);
338             if (recvDataLength == peekedLength)
339             {
340                 if (verbose)
341                 {
342                     printBuffer(Rx, requestMsg);
343                 }
344 
345                 if (MCTP_MSG_TYPE_PLDM != requestMsg[1])
346                 {
347                     // Skip this message and continue.
348                 }
349                 else
350                 {
351                     // process message and send response
352                     auto response = processRxMsg(requestMsg, invoker,
353                                                  reqHandler, fwManager.get());
354                     if (response.has_value())
355                     {
356                         if (verbose)
357                         {
358                             printBuffer(Tx, *response);
359                         }
360 
361                         iov[0].iov_base = &requestMsg[0];
362                         iov[0].iov_len =
363                             sizeof(requestMsg[0]) + sizeof(requestMsg[1]);
364                         iov[1].iov_base = (*response).data();
365                         iov[1].iov_len = (*response).size();
366 
367                         msg.msg_iov = iov;
368                         msg.msg_iovlen = sizeof(iov) / sizeof(iov[0]);
369 
370                         int result = sendmsg(fd, &msg, 0);
371                         if (-1 == result)
372                         {
373                             returnCode = -errno;
374                             std::cerr << "sendto system call failed, RC= "
375                                       << returnCode << "\n";
376                         }
377                     }
378                 }
379             }
380             else
381             {
382                 std::cerr
383                     << "Failure to read peeked length packet. peekedLength= "
384                     << peekedLength << " recvDataLength=" << recvDataLength
385                     << "\n";
386             }
387         }
388     };
389 
390     bus.attach_event(event.get(), SD_EVENT_PRIORITY_NORMAL);
391     bus.request_name("xyz.openbmc_project.PLDM");
392     IO io(event, socketFd(), EPOLLIN, std::move(callback));
393 #ifdef LIBPLDMRESPONDER
394     if (hostPDRHandler)
395     {
396         hostPDRHandler->setHostFirmwareCondition();
397     }
398 #endif
399     returnCode = event.loop();
400 
401     if (shutdown(sockfd, SHUT_RDWR))
402     {
403         std::perror("Failed to shutdown the socket");
404     }
405     if (returnCode)
406     {
407         exit(EXIT_FAILURE);
408     }
409 
410     exit(EXIT_SUCCESS);
411 }
412