1 /* 2 * Copyright 2019 Google Inc. 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #include "net.hpp" 18 19 #include "data.hpp" 20 #include "flags.hpp" 21 22 #include <errno.h> 23 #include <fcntl.h> 24 #include <netdb.h> 25 #include <sys/sendfile.h> 26 #include <sys/socket.h> 27 #include <sys/types.h> 28 29 #include <ipmiblob/blob_errors.hpp> 30 #include <stdplus/handle/managed.hpp> 31 32 #include <cstdint> 33 #include <cstring> 34 #include <memory> 35 #include <optional> 36 #include <string> 37 #include <vector> 38 39 namespace 40 { 41 42 void closefd(int&& fd, const internal::Sys*& sys) 43 { 44 sys->close(fd); 45 } 46 using Fd = stdplus::Managed<int, const internal::Sys*>::Handle<closefd>; 47 48 } // namespace 49 50 namespace host_tool 51 { 52 53 bool NetDataHandler::sendContents(const std::string& input, 54 std::uint16_t session) 55 { 56 constexpr size_t blockSize = 64 * 1024; 57 Fd inputFd(std::nullopt, sys); 58 59 { 60 inputFd.reset(sys->open(input.c_str(), O_RDONLY)); 61 if (*inputFd < 0) 62 { 63 (void)inputFd.release(); 64 std::fprintf(stderr, "Unable to open file: '%s'\n", input.c_str()); 65 return false; 66 } 67 68 std::int64_t fileSize = sys->getSize(input.c_str()); 69 if (fileSize == 0) 70 { 71 std::fprintf(stderr, 72 "Zero-length file, or other file access error\n"); 73 return false; 74 } 75 76 progress->start(fileSize); 77 } 78 79 Fd connFd(std::nullopt, sys); 80 81 { 82 struct addrinfo hints; 83 std::memset(&hints, 0, sizeof(hints)); 84 hints.ai_flags = AI_NUMERICHOST; 85 hints.ai_family = AF_UNSPEC; 86 hints.ai_socktype = SOCK_STREAM; 87 88 struct addrinfo *addrs, *addr; 89 int ret = sys->getaddrinfo(host.c_str(), port.c_str(), &hints, &addrs); 90 if (ret < 0) 91 { 92 std::fprintf(stderr, "Couldn't parse address %s with port %s: %s\n", 93 host.c_str(), port.c_str(), gai_strerror(ret)); 94 return false; 95 } 96 97 for (addr = addrs; addr != nullptr; addr = addr->ai_next) 98 { 99 connFd.reset(sys->socket(addr->ai_family, addr->ai_socktype, 100 addr->ai_protocol)); 101 if (*connFd == -1) 102 continue; 103 104 if (sys->connect(*connFd, addr->ai_addr, addr->ai_addrlen) != -1) 105 break; 106 } 107 108 // TODO: use stdplus Managed for the addrinfo structs 109 sys->freeaddrinfo(addrs); 110 111 if (addr == nullptr) 112 { 113 std::fprintf(stderr, "Failed to connect\n"); 114 return false; 115 } 116 } 117 118 try 119 { 120 int bytesSent = 0; 121 off_t offset = 0; 122 123 do 124 { 125 bytesSent = sys->sendfile(*connFd, *inputFd, &offset, blockSize); 126 if (bytesSent < 0) 127 { 128 std::fprintf(stderr, "Failed to send data to BMC: %s\n", 129 strerror(errno)); 130 return false; 131 } 132 else if (bytesSent > 0) 133 { 134 /* Ok, so the data is staged, now send the blob write with 135 * the details. 136 */ 137 struct ipmi_flash::ExtChunkHdr chunk; 138 chunk.length = bytesSent; 139 std::vector<std::uint8_t> chunkBytes(sizeof(chunk)); 140 std::memcpy(chunkBytes.data(), &chunk, sizeof(chunk)); 141 142 /* This doesn't return anything on success. */ 143 blob->writeBytes(session, offset - bytesSent, chunkBytes); 144 progress->updateProgress(bytesSent); 145 } 146 } while (bytesSent > 0); 147 } 148 catch (const ipmiblob::BlobException& b) 149 { 150 return false; 151 } 152 153 return true; 154 } 155 156 } // namespace host_tool 157