xref: /openbmc/pldm/pldmd/pldmd.cpp (revision c4e80cca)
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_pdr.hpp"
8 #include "dbus_impl_requester.hpp"
9 #include "host-bmc/dbus_to_host_effecters.hpp"
10 #include "host-bmc/host_pdr_handler.hpp"
11 #include "invoker.hpp"
12 #include "libpldmresponder/base.hpp"
13 #include "libpldmresponder/bios.hpp"
14 #include "libpldmresponder/fru.hpp"
15 #include "libpldmresponder/platform.hpp"
16 
17 #include <err.h>
18 #include <getopt.h>
19 #include <poll.h>
20 #include <stdlib.h>
21 #include <sys/socket.h>
22 #include <sys/types.h>
23 #include <sys/un.h>
24 #include <unistd.h>
25 
26 #include <sdeventplus/event.hpp>
27 #include <sdeventplus/source/io.hpp>
28 
29 #include <cstdio>
30 #include <cstring>
31 #include <fstream>
32 #include <iomanip>
33 #include <iostream>
34 #include <iterator>
35 #include <memory>
36 #include <sstream>
37 #include <stdexcept>
38 #include <string>
39 #include <vector>
40 
41 #ifdef OEM_IBM
42 #include "libpldmresponder/file_io.hpp"
43 #endif
44 
45 constexpr uint8_t MCTP_MSG_TYPE_PLDM = 1;
46 
47 using namespace pldm::responder;
48 using namespace pldm;
49 using namespace sdeventplus;
50 using namespace sdeventplus::source;
51 
52 static Response processRxMsg(const std::vector<uint8_t>& requestMsg,
53                              Invoker& invoker, dbus_api::Requester& requester)
54 {
55 
56     Response response;
57     uint8_t eid = requestMsg[0];
58     uint8_t type = requestMsg[1];
59     pldm_header_info hdrFields{};
60     auto hdr = reinterpret_cast<const pldm_msg_hdr*>(
61         requestMsg.data() + sizeof(eid) + sizeof(type));
62     if (PLDM_SUCCESS != unpack_pldm_header(hdr, &hdrFields))
63     {
64         std::cerr << "Empty PLDM request header \n";
65     }
66     else if (PLDM_RESPONSE != hdrFields.msg_type)
67     {
68         auto request = reinterpret_cast<const pldm_msg*>(hdr);
69         size_t requestLen = requestMsg.size() - sizeof(struct pldm_msg_hdr) -
70                             sizeof(eid) - sizeof(type);
71         try
72         {
73             response = invoker.handle(hdrFields.pldm_type, hdrFields.command,
74                                       request, requestLen);
75         }
76         catch (const std::out_of_range& e)
77         {
78             uint8_t completion_code = PLDM_ERROR_UNSUPPORTED_PLDM_CMD;
79             response.resize(sizeof(pldm_msg_hdr));
80             auto responseHdr = reinterpret_cast<pldm_msg_hdr*>(response.data());
81             pldm_header_info header{};
82             header.msg_type = PLDM_RESPONSE;
83             header.instance = hdrFields.instance;
84             header.pldm_type = hdrFields.pldm_type;
85             header.command = hdrFields.command;
86             auto result = pack_pldm_header(&header, responseHdr);
87             if (PLDM_SUCCESS != result)
88             {
89                 std::cerr << "Failed adding response header \n";
90             }
91             response.insert(response.end(), completion_code);
92         }
93     }
94     else
95     {
96         requester.markFree(eid, hdr->instance_id);
97     }
98     return response;
99 }
100 
101 void printBuffer(const std::vector<uint8_t>& buffer)
102 {
103     std::ostringstream tempStream;
104     tempStream << "Buffer Data: ";
105     if (!buffer.empty())
106     {
107         for (int byte : buffer)
108         {
109             tempStream << std::setfill('0') << std::setw(2) << std::hex << byte
110                        << " ";
111         }
112     }
113     std::cout << tempStream.str().c_str() << std::endl;
114 }
115 
116 void optionUsage(void)
117 {
118     std::cerr << "Usage: pldmd [options]\n";
119     std::cerr << "Options:\n";
120     std::cerr
121         << "  --verbose=<0/1>  0 - Disable verbosity, 1 - Enable verbosity\n";
122     std::cerr << "Defaulted settings:  --verbose=0 \n";
123 }
124 
125 int main(int argc, char** argv)
126 {
127 
128     bool verbose = false;
129     static struct option long_options[] = {
130         {"verbose", required_argument, 0, 'v'}, {0, 0, 0, 0}};
131 
132     auto argflag = getopt_long(argc, argv, "v:", long_options, nullptr);
133     switch (argflag)
134     {
135         case 'v':
136             switch (std::stoi(optarg))
137             {
138                 case 0:
139                     verbose = false;
140                     break;
141                 case 1:
142                     verbose = true;
143                     break;
144                 default:
145                     optionUsage();
146                     break;
147             }
148             break;
149         default:
150             optionUsage();
151             break;
152     }
153 
154     /* Create local socket. */
155     int returnCode = 0;
156     int sockfd = socket(AF_UNIX, SOCK_SEQPACKET, 0);
157     if (-1 == sockfd)
158     {
159         returnCode = -errno;
160         std::cerr << "Failed to create the socket, RC= " << returnCode << "\n";
161         exit(EXIT_FAILURE);
162     }
163 
164     auto event = Event::get_default();
165     std::unique_ptr<pldm_pdr, decltype(&pldm_pdr_destroy)> pdrRepo(
166         pldm_pdr_init(), pldm_pdr_destroy);
167     std::unique_ptr<pldm_entity_association_tree,
168                     decltype(&pldm_entity_association_tree_destroy)>
169         entityTree(pldm_entity_association_tree_init(),
170                    pldm_entity_association_tree_destroy);
171     auto& bus = pldm::utils::DBusHandler::getBus();
172     dbus_api::Requester dbusImplReq(bus, "/xyz/openbmc_project/pldm");
173     std::unique_ptr<HostPDRHandler> hostPDRHandler;
174     std::unique_ptr<pldm::host_effecters::HostEffecterParser>
175         hostEffecterParser;
176     DBusHandler dbusHandler;
177     auto hostEID = pldm::utils::readHostEID();
178     if (hostEID)
179     {
180         hostPDRHandler = std::make_unique<HostPDRHandler>(
181             sockfd, hostEID, event, pdrRepo.get(), entityTree.get(),
182             dbusImplReq);
183         hostEffecterParser =
184             std::make_unique<pldm::host_effecters::HostEffecterParser>(
185                 &dbusImplReq, sockfd, pdrRepo.get(), &dbusHandler,
186                 HOST_JSONS_DIR, verbose);
187     }
188 
189     Invoker invoker{};
190     invoker.registerHandler(PLDM_BASE, std::make_unique<base::Handler>());
191     invoker.registerHandler(PLDM_BIOS, std::make_unique<bios::Handler>());
192     auto fruHandler = std::make_unique<fru::Handler>(
193         FRU_JSONS_DIR, pdrRepo.get(), entityTree.get());
194     // FRU table is built lazily when a FRU command or Get PDR command is
195     // handled. To enable building FRU table, the FRU handler is passed to the
196     // Platform handler.
197     invoker.registerHandler(PLDM_PLATFORM,
198                             std::make_unique<platform::Handler>(
199                                 dbusHandler, PDR_JSONS_DIR, EVENTS_JSONS_DIR,
200                                 pdrRepo.get(), hostPDRHandler.get(),
201                                 fruHandler.get()));
202     invoker.registerHandler(PLDM_FRU, std::move(fruHandler));
203 
204 #ifdef OEM_IBM
205     invoker.registerHandler(PLDM_OEM, std::make_unique<oem_ibm::Handler>());
206 #endif
207 
208     pldm::utils::CustomFD socketFd(sockfd);
209 
210     struct sockaddr_un addr
211     {};
212     addr.sun_family = AF_UNIX;
213     const char path[] = "\0mctp-mux";
214     memcpy(addr.sun_path, path, sizeof(path) - 1);
215     int result = connect(socketFd(), reinterpret_cast<struct sockaddr*>(&addr),
216                          sizeof(path) + sizeof(addr.sun_family) - 1);
217     if (-1 == result)
218     {
219         returnCode = -errno;
220         std::cerr << "Failed to connect to the socket, RC= " << returnCode
221                   << "\n";
222         exit(EXIT_FAILURE);
223     }
224 
225     result = write(socketFd(), &MCTP_MSG_TYPE_PLDM, sizeof(MCTP_MSG_TYPE_PLDM));
226     if (-1 == result)
227     {
228         returnCode = -errno;
229         std::cerr << "Failed to send message type as pldm to mctp, RC= "
230                   << returnCode << "\n";
231         exit(EXIT_FAILURE);
232     }
233 
234     dbus_api::Pdr dbusImplPdr(bus, "/xyz/openbmc_project/pldm", pdrRepo.get());
235     auto callback = [verbose, &invoker, &dbusImplReq](IO& /*io*/, int fd,
236                                                       uint32_t revents) {
237         if (!(revents & EPOLLIN))
238         {
239             return;
240         }
241 
242         // Outgoing message.
243         struct iovec iov[2]{};
244 
245         // This structure contains the parameter information for the response
246         // message.
247         struct msghdr msg
248         {};
249 
250         int returnCode = 0;
251         ssize_t peekedLength = recv(fd, nullptr, 0, MSG_PEEK | MSG_TRUNC);
252         if (0 == peekedLength)
253         {
254             std::cerr << "Socket has been closed \n";
255         }
256         else if (peekedLength <= -1)
257         {
258             returnCode = -errno;
259             std::cerr << "recv system call failed, RC= " << returnCode << "\n";
260         }
261         else
262         {
263             std::vector<uint8_t> requestMsg(peekedLength);
264             auto recvDataLength = recv(
265                 fd, static_cast<void*>(requestMsg.data()), peekedLength, 0);
266             if (recvDataLength == peekedLength)
267             {
268                 if (verbose)
269                 {
270                     std::cout << "Received Msg" << std::endl;
271                     printBuffer(requestMsg);
272                 }
273                 if (MCTP_MSG_TYPE_PLDM != requestMsg[1])
274                 {
275                     // Skip this message and continue.
276                     std::cerr << "Encountered Non-PLDM type message"
277                               << "\n";
278                 }
279                 else
280                 {
281                     // process message and send response
282                     auto response =
283                         processRxMsg(requestMsg, invoker, dbusImplReq);
284                     if (!response.empty())
285                     {
286                         if (verbose)
287                         {
288                             std::cout << "Sending Msg" << std::endl;
289                             printBuffer(response);
290                         }
291 
292                         iov[0].iov_base = &requestMsg[0];
293                         iov[0].iov_len =
294                             sizeof(requestMsg[0]) + sizeof(requestMsg[1]);
295                         iov[1].iov_base = response.data();
296                         iov[1].iov_len = response.size();
297 
298                         msg.msg_iov = iov;
299                         msg.msg_iovlen = sizeof(iov) / sizeof(iov[0]);
300 
301                         int result = sendmsg(fd, &msg, 0);
302                         if (-1 == result)
303                         {
304                             returnCode = -errno;
305                             std::cerr << "sendto system call failed, RC= "
306                                       << returnCode << "\n";
307                         }
308                     }
309                 }
310             }
311             else
312             {
313                 std::cerr
314                     << "Failure to read peeked length packet. peekedLength= "
315                     << peekedLength << " recvDataLength=" << recvDataLength
316                     << "\n";
317             }
318         }
319     };
320 
321     bus.attach_event(event.get(), SD_EVENT_PRIORITY_NORMAL);
322     bus.request_name("xyz.openbmc_project.PLDM");
323     IO io(event, socketFd(), EPOLLIN, std::move(callback));
324     event.loop();
325 
326     result = shutdown(sockfd, SHUT_RDWR);
327     if (-1 == result)
328     {
329         returnCode = -errno;
330         std::cerr << "Failed to shutdown the socket, RC=" << returnCode << "\n";
331         exit(EXIT_FAILURE);
332     }
333     exit(EXIT_FAILURE);
334 }
335