| #include "sd_event_loop.hpp" |
| |
| #include "main.hpp" |
| #include "message_handler.hpp" |
| |
| #include <netinet/in.h> |
| #include <sys/ioctl.h> |
| #include <sys/socket.h> |
| #include <systemd/sd-daemon.h> |
| |
| #include <phosphor-logging/log.hpp> |
| |
| namespace eventloop |
| { |
| using namespace phosphor::logging; |
| |
| static int udp623Handler(sd_event_source* es, int fd, uint32_t revents, |
| void* userdata) |
| { |
| std::shared_ptr<udpsocket::Channel> channelPtr; |
| struct timeval timeout; |
| timeout.tv_sec = SELECT_CALL_TIMEOUT; |
| timeout.tv_usec = 0; |
| |
| try |
| { |
| channelPtr.reset(new udpsocket::Channel(fd, timeout)); |
| |
| // Initialize the Message Handler with the socket channel |
| message::Handler msgHandler(channelPtr); |
| |
| std::unique_ptr<message::Message> inMessage; |
| |
| // Read the incoming IPMI packet |
| inMessage = msgHandler.receive(); |
| if (inMessage == nullptr) |
| { |
| return 0; |
| } |
| |
| // Execute the Command |
| auto outMessage = msgHandler.executeCommand(*(inMessage.get())); |
| if (outMessage == nullptr) |
| { |
| return 0; |
| } |
| |
| // Send the response IPMI Message |
| msgHandler.send(*(outMessage.get())); |
| } |
| catch (std::exception& e) |
| { |
| log<level::ERR>("Executing the IPMI message failed"); |
| log<level::ERR>(e.what()); |
| } |
| |
| return 0; |
| } |
| |
| static int consoleInputHandler(sd_event_source* es, int fd, uint32_t revents, |
| void* userdata) |
| { |
| try |
| { |
| int readSize = 0; |
| |
| if (ioctl(fd, FIONREAD, &readSize) < 0) |
| { |
| log<level::ERR>("ioctl failed for FIONREAD:", |
| entry("ERRNO=%d", errno)); |
| return 0; |
| } |
| |
| std::vector<uint8_t> buffer(readSize); |
| auto bufferSize = buffer.size(); |
| ssize_t readDataLen = 0; |
| |
| readDataLen = read(fd, buffer.data(), bufferSize); |
| |
| // Update the Console buffer with data read from the socket |
| if (readDataLen > 0) |
| { |
| buffer.resize(readDataLen); |
| std::get<sol::Manager&>(singletonPool).dataBuffer.write(buffer); |
| } |
| else if (readDataLen == 0) |
| { |
| log<level::ERR>("Connection Closed for host console socket"); |
| } |
| else if (readDataLen < 0) // Error |
| { |
| log<level::ERR>("Reading from host console socket failed:", |
| entry("ERRNO=%d", errno)); |
| } |
| } |
| catch (std::exception& e) |
| { |
| log<level::ERR>(e.what()); |
| } |
| |
| return 0; |
| } |
| |
| static int charAccTimerHandler(sd_event_source* s, uint64_t usec, |
| void* userdata) |
| { |
| auto bufferSize = std::get<sol::Manager&>(singletonPool).dataBuffer.size(); |
| |
| try |
| { |
| // The instance is hardcoded to 1, in the case of supporting multiple |
| // payload instances we would need to populate it from userdata |
| uint8_t instance = 1; |
| |
| if (bufferSize > 0) |
| { |
| auto& context = |
| std::get<sol::Manager&>(singletonPool).getContext(instance); |
| |
| int rc = context.sendOutboundPayload(); |
| |
| if (rc == 0) |
| { |
| return 0; |
| } |
| } |
| |
| std::get<eventloop::EventLoop&>(singletonPool) |
| .switchTimer(instance, Timers::ACCUMULATE, true); |
| } |
| catch (std::exception& e) |
| { |
| log<level::ERR>(e.what()); |
| } |
| |
| return 0; |
| } |
| |
| static int retryTimerHandler(sd_event_source* s, uint64_t usec, void* userdata) |
| { |
| try |
| { |
| // The instance is hardcoded to 1, in the case of supporting multiple |
| // payload instances we would need to populate it from userdata |
| uint8_t instance = 1; |
| |
| auto& context = |
| std::get<sol::Manager&>(singletonPool).getContext(instance); |
| |
| if (context.retryCounter) |
| { |
| --context.retryCounter; |
| std::get<eventloop::EventLoop&>(singletonPool) |
| .switchTimer(instance, Timers::RETRY, true); |
| context.resendPayload(sol::Context::noClear); |
| } |
| else |
| { |
| context.retryCounter = context.maxRetryCount; |
| context.resendPayload(sol::Context::clear); |
| std::get<eventloop::EventLoop&>(singletonPool) |
| .switchTimer(instance, Timers::RETRY, false); |
| std::get<eventloop::EventLoop&>(singletonPool) |
| .switchTimer(instance, Timers::ACCUMULATE, true); |
| } |
| } |
| catch (std::exception& e) |
| { |
| log<level::ERR>(e.what()); |
| } |
| |
| return 0; |
| } |
| |
| int EventLoop::startEventLoop(sd_event* events) |
| { |
| int fd = -1; |
| int r = 0; |
| int listen_fd; |
| sigset_t ss; |
| sd_event_source* source = nullptr; |
| auto bus = ipmid_get_sd_bus_connection(); |
| |
| event = events; |
| // Attach the bus to sd_event to service user requests |
| r = sd_bus_attach_event(bus, event, SD_EVENT_PRIORITY_NORMAL); |
| if (r < 0) |
| { |
| goto finish; |
| } |
| |
| if (sigemptyset(&ss) < 0 || sigaddset(&ss, SIGTERM) < 0 || |
| sigaddset(&ss, SIGINT) < 0) |
| { |
| r = -errno; |
| goto finish; |
| } |
| |
| /* Block SIGTERM first, so that the event loop can handle it */ |
| if (sigprocmask(SIG_BLOCK, &ss, nullptr) < 0) |
| { |
| r = -errno; |
| goto finish; |
| } |
| |
| /* Let's make use of the default handler and "floating" reference features |
| * of sd_event_add_signal() */ |
| r = sd_event_add_signal(event, nullptr, SIGTERM, nullptr, nullptr); |
| if (r < 0) |
| { |
| goto finish; |
| } |
| |
| r = sd_event_add_signal(event, nullptr, SIGINT, nullptr, nullptr); |
| if (r < 0) |
| { |
| goto finish; |
| } |
| |
| // Create our own socket if SysD did not supply one. |
| listen_fd = sd_listen_fds(0); |
| if (listen_fd == 1) |
| { |
| fd = SD_LISTEN_FDS_START; |
| } |
| else if (listen_fd > 1) |
| { |
| log<level::ERR>("Too many file descriptors received"); |
| goto finish; |
| } |
| else |
| { |
| struct sockaddr_in address; |
| if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) == 0) |
| { |
| r = -errno; |
| log<level::ERR>("Unable to manually open socket"); |
| goto finish; |
| } |
| |
| address.sin_family = AF_INET; |
| address.sin_addr.s_addr = INADDR_ANY; |
| address.sin_port = htons(IPMI_STD_PORT); |
| |
| if (bind(fd, (struct sockaddr*)&address, sizeof(address)) < 0) |
| { |
| r = -errno; |
| log<level::ERR>("Unable to bind socket"); |
| goto finish; |
| } |
| } |
| |
| r = sd_event_add_io(event, &source, fd, EPOLLIN, udp623Handler, nullptr); |
| if (r < 0) |
| { |
| goto finish; |
| } |
| |
| udpIPMI.reset(source); |
| source = nullptr; |
| |
| r = sd_event_loop(event); |
| |
| finish: |
| |
| if (fd >= 0) |
| { |
| (void)close(fd); |
| } |
| |
| if (r < 0) |
| { |
| log<level::ERR>("Event Loop Failure:", |
| entry("FAILURE=%s", strerror(-r))); |
| } |
| |
| return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS; |
| } |
| |
| void EventLoop::startHostConsole(const sol::CustomFD& fd) |
| { |
| int rc = 0; |
| |
| if ((fd() == -1) || hostConsole.get()) |
| { |
| throw std::runtime_error("Console descriptor already added"); |
| } |
| |
| sd_event_source* source = nullptr; |
| |
| // Add the fd to the event loop for EPOLLIN |
| rc = sd_event_add_io(event, &source, fd(), EPOLLIN, consoleInputHandler, |
| nullptr); |
| if (rc < 0) |
| { |
| throw std::runtime_error("Failed to add socket descriptor"); |
| } |
| |
| hostConsole.reset(source); |
| source = nullptr; |
| } |
| |
| void EventLoop::stopHostConsole() |
| { |
| if (hostConsole.get()) |
| { |
| // Disable the host console payload |
| int rc = sd_event_source_set_enabled(hostConsole.get(), SD_EVENT_OFF); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to disable the host console socket", |
| entry("RC=%d", rc)); |
| } |
| |
| hostConsole.reset(); |
| } |
| } |
| |
| void EventLoop::startSOLPayloadInstance(uint8_t payloadInst, |
| IntervalType accumulateInterval, |
| IntervalType retryInterval) |
| { |
| auto instance = payloadInst; |
| sd_event_source* accTimerSource = nullptr; |
| sd_event_source* retryTimerSource = nullptr; |
| int rc = 0; |
| uint64_t currentTime = 0; |
| |
| rc = sd_event_now(event, CLOCK_MONOTONIC, ¤tTime); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to get the current timestamp", |
| entry("RC=%d", rc)); |
| throw std::runtime_error("Failed to get current timestamp"); |
| } |
| |
| // Create character accumulate timer |
| rc = sd_event_add_time(event, &accTimerSource, CLOCK_MONOTONIC, |
| currentTime + accumulateInterval.count(), 0, |
| charAccTimerHandler, static_cast<void*>(&instance)); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to setup the accumulate timer", |
| entry("RC=%d", rc)); |
| throw std::runtime_error("Failed to setup accumulate timer"); |
| } |
| |
| // Create retry interval timer and add to the event loop |
| rc = sd_event_add_time(event, &retryTimerSource, CLOCK_MONOTONIC, |
| currentTime + retryInterval.count(), 0, |
| retryTimerHandler, static_cast<void*>(&instance)); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to setup the retry timer", entry("RC=%d", rc)); |
| throw std::runtime_error("Failed to setup retry timer"); |
| } |
| |
| // Enable the Character Accumulate Timer |
| rc = sd_event_source_set_enabled(accTimerSource, SD_EVENT_ONESHOT); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to enable the accumulate timer", |
| entry("RC=%d", rc)); |
| throw std::runtime_error("Failed to enable accumulate timer"); |
| } |
| |
| // Disable the Retry Interval Timer |
| rc = sd_event_source_set_enabled(retryTimerSource, SD_EVENT_OFF); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to disable the retry timer", |
| entry("RC=%d", rc)); |
| throw std::runtime_error("Failed to disable retry timer"); |
| } |
| |
| EventSource accEventSource(accTimerSource); |
| EventSource retryEventSource(retryTimerSource); |
| accTimerSource = nullptr; |
| retryTimerSource = nullptr; |
| |
| TimerMap timer; |
| timer.emplace(Timers::ACCUMULATE, std::make_tuple(std::move(accEventSource), |
| accumulateInterval)); |
| timer.emplace(Timers::RETRY, |
| std::make_tuple(std::move(retryEventSource), retryInterval)); |
| payloadInfo.emplace(instance, std::move(timer)); |
| } |
| |
| void EventLoop::stopSOLPayloadInstance(uint8_t payloadInst) |
| { |
| auto iter = payloadInfo.find(payloadInst); |
| if (iter == payloadInfo.end()) |
| { |
| log<level::ERR>("SOL Payload instance not found", |
| entry("PAYLOADINST=%d", payloadInst)); |
| throw std::runtime_error("SOL payload instance not found"); |
| } |
| |
| int rc = 0; |
| |
| /* Destroy the character accumulate timer event source */ |
| rc = sd_event_source_set_enabled( |
| (std::get<0>(iter->second.at(Timers::ACCUMULATE))).get(), SD_EVENT_OFF); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to disable the character accumulate timer", |
| entry("RC=%d", rc)); |
| payloadInfo.erase(payloadInst); |
| throw std::runtime_error("Failed to disable accumulate timer"); |
| } |
| |
| /* Destroy the retry interval timer event source */ |
| rc = sd_event_source_set_enabled( |
| (std::get<0>(iter->second.at(Timers::RETRY))).get(), SD_EVENT_OFF); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to disable the retry timer", |
| entry("RC=%d", rc)); |
| payloadInfo.erase(payloadInst); |
| throw std::runtime_error("Failed to disable retry timer"); |
| } |
| |
| payloadInfo.erase(payloadInst); |
| } |
| |
| void EventLoop::switchTimer(uint8_t payloadInst, Timers type, bool status) |
| { |
| auto iter = payloadInfo.find(payloadInst); |
| if (iter == payloadInfo.end()) |
| { |
| log<level::ERR>("SOL Payload instance not found", |
| entry("PAYLOADINST=%d", payloadInst)); |
| throw std::runtime_error("SOL Payload instance not found"); |
| } |
| |
| int rc = 0; |
| auto source = (std::get<0>(iter->second.at(type))).get(); |
| auto interval = std::get<1>(iter->second.at(type)); |
| |
| // Turn OFF the timer |
| if (!status) |
| { |
| rc = sd_event_source_set_enabled(source, SD_EVENT_OFF); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to disable the timer", entry("RC=%d", rc)); |
| throw std::runtime_error("Failed to disable timer"); |
| } |
| return; |
| } |
| |
| // Turn ON the timer |
| uint64_t currentTime = 0; |
| rc = sd_event_now(event, CLOCK_MONOTONIC, ¤tTime); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to get the current timestamp", |
| entry("RC=%d", rc)); |
| throw std::runtime_error("Failed to get current timestamp"); |
| } |
| |
| rc = sd_event_source_set_time(source, currentTime + interval.count()); |
| if (rc < 0) |
| { |
| log<level::ERR>("sd_event_source_set_time function failed", |
| entry("RC=%d", rc)); |
| throw std::runtime_error("sd_event_source_set_time function failed"); |
| } |
| |
| rc = sd_event_source_set_enabled(source, SD_EVENT_ONESHOT); |
| if (rc < 0) |
| { |
| log<level::ERR>("Failed to enable the timer", entry("RC=%d", rc)); |
| throw std::runtime_error("Failed to enable timer"); |
| } |
| } |
| |
| } // namespace eventloop |