1 // Copyright 2022 Google LLC 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 #include "file-io.hpp" 16 17 #include <sdeventplus/event.hpp> 18 #include <sdeventplus/source/io.hpp> 19 #include <stdplus/fd/create.hpp> 20 #include <stdplus/fd/ops.hpp> 21 #include <stdplus/print.hpp> 22 23 using namespace std::string_view_literals; 24 25 // A privileged port that is reserved for querying BMC DHCP completion. 26 // This is well known by the clients querying the status. 27 constexpr uint16_t kListenPort = 23; 28 29 stdplus::ManagedFd createListener() 30 { 31 using namespace stdplus::fd; 32 auto sock = socket(SocketDomain::INet6, SocketType::Stream, 33 SocketProto::TCP); 34 setFileFlags(sock, getFileFlags(sock).set(stdplus::fd::FileFlag::NonBlock)); 35 sockaddr_in6 addr = {}; 36 addr.sin6_family = AF_INET6; 37 addr.sin6_port = htons(kListenPort); 38 bind(sock, addr); 39 listen(sock, 10); 40 return sock; 41 } 42 43 int main() 44 { 45 try 46 { 47 auto listener = createListener(); 48 auto event = sdeventplus::Event::get_default(); 49 sdeventplus::source::IO do_accept( 50 event, listener.get(), EPOLLIN | EPOLLET, 51 [&](sdeventplus::source::IO&, int, uint32_t) { 52 while (auto fd = stdplus::fd::accept(listener)) 53 { 54 std::string data; 55 try 56 { 57 data = fileRead(statusFile); 58 } 59 catch (const std::exception& e) 60 { 61 // we don't want to fail the upgrade process, set the status 62 // to ONGOING 63 data.push_back(2); 64 data.append("Failed to read status "); 65 data.append(e.what()); 66 } 67 68 stdplus::fd::sendExact(*fd, data, stdplus::fd::SendFlags(0)); 69 } 70 }); 71 return event.loop(); 72 } 73 catch (const std::exception& e) 74 { 75 stdplus::println(stderr, "Failed: {}", e.what()); 76 return 1; 77 } 78 } 79