1bf58cd64SPatrick Venture /* 2bf58cd64SPatrick Venture * Copyright 2018 Google Inc. 3bf58cd64SPatrick Venture * 4bf58cd64SPatrick Venture * Licensed under the Apache License, Version 2.0 (the "License"); 5bf58cd64SPatrick Venture * you may not use this file except in compliance with the License. 6bf58cd64SPatrick Venture * You may obtain a copy of the License at 7bf58cd64SPatrick Venture * 8bf58cd64SPatrick Venture * http://www.apache.org/licenses/LICENSE-2.0 9bf58cd64SPatrick Venture * 10bf58cd64SPatrick Venture * Unless required by applicable law or agreed to in writing, software 11bf58cd64SPatrick Venture * distributed under the License is distributed on an "AS IS" BASIS, 12bf58cd64SPatrick Venture * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13bf58cd64SPatrick Venture * See the License for the specific language governing permissions and 14bf58cd64SPatrick Venture * limitations under the License. 15bf58cd64SPatrick Venture */ 16bf58cd64SPatrick Venture 17bf58cd64SPatrick Venture #include "updater.hpp" 18bf58cd64SPatrick Venture 1984778b8dSPatrick Venture #include "flags.hpp" 2001123b2aSPatrick Venture #include "handler.hpp" 213ecb3503SPatrick Venture #include "status.hpp" 222bc23fe1SPatrick Venture #include "tool_errors.hpp" 237dad86fdSPatrick Venture #include "util.hpp" 240533d0b0SPatrick Venture 259b37b095SPatrick Venture #include <ipmiblob/blob_errors.hpp> 269b37b095SPatrick Venture 2700887597SPatrick Venture #include <algorithm> 28339dece8SPatrick Venture #include <cstring> 29af69625fSPatrick Venture #include <memory> 302a927e87SPatrick Venture #include <string> 31d61b0ff8SPatrick Venture #include <thread> 32c9792e75SPatrick Venture #include <unordered_map> 3355646decSPatrick Venture #include <vector> 34af69625fSPatrick Venture 359b534f06SPatrick Venture namespace host_tool 369b534f06SPatrick Venture { 379b534f06SPatrick Venture 38*203ad804SWilly Tu void updaterMain(UpdateHandlerInterface* updater, ipmiblob::BlobInterface* blob, 39*203ad804SWilly Tu const std::string& imagePath, const std::string& signaturePath, 406749ba1cSBrandon Kim const std::string& layoutType, bool ignoreUpdate) 4155646decSPatrick Venture { 42c498caa4SPatrick Venture /* TODO: validate the layoutType isn't a special value such as: 'update', 43c498caa4SPatrick Venture * 'verify', or 'hash' 44c498caa4SPatrick Venture */ 45c498caa4SPatrick Venture std::string layout = "/flash/" + layoutType; 469f937c45SPatrick Venture 479f937c45SPatrick Venture bool goalSupported = updater->checkAvailable(layout); 4855646decSPatrick Venture if (!goalSupported) 4955646decSPatrick Venture { 50060be01fSBenjamin Fair throw ToolException("Goal firmware not supported"); 5155646decSPatrick Venture } 5255646decSPatrick Venture 53*203ad804SWilly Tu // Clean all active blobs to support multiple stages 54*203ad804SWilly Tu // Check for any active blobs and delete the first one found to reset the 55*203ad804SWilly Tu // BMC's phosphor-ipmi-flash state machine then clean any leftover artifacts 56*203ad804SWilly Tu const auto blobList = blob->getBlobList(); 57*203ad804SWilly Tu for (const auto& activeBlob : blobList) 58*203ad804SWilly Tu { 59*203ad804SWilly Tu // Prefix is /flash/active/ 60*203ad804SWilly Tu if (activeBlob.find("/flash/active/", 0) == 0) 61*203ad804SWilly Tu { 62*203ad804SWilly Tu std::fprintf(stderr, "Found an active blob, deleting %s\n", 63*203ad804SWilly Tu activeBlob.c_str()); 64*203ad804SWilly Tu blob->deleteBlob(activeBlob); 65*203ad804SWilly Tu updater->cleanArtifacts(); 66*203ad804SWilly Tu break; 67*203ad804SWilly Tu } 68*203ad804SWilly Tu } 69*203ad804SWilly Tu 70060be01fSBenjamin Fair /* Yay, our layout type is supported. */ 715f2fcc4eSPatrick Venture try 725f2fcc4eSPatrick Venture { 7355646decSPatrick Venture /* Send over the firmware image. */ 7455646decSPatrick Venture std::fprintf(stderr, "Sending over the firmware image.\n"); 759f937c45SPatrick Venture updater->sendFile(layout, imagePath); 7655646decSPatrick Venture 7755646decSPatrick Venture /* Send over the hash contents. */ 7855646decSPatrick Venture std::fprintf(stderr, "Sending over the hash file.\n"); 791d5a31c9SPatrick Venture updater->sendFile(ipmi_flash::hashBlobId, signaturePath); 8055646decSPatrick Venture 815f2fcc4eSPatrick Venture /* Trigger the verification by opening and committing the verify file. 825f2fcc4eSPatrick Venture */ 8355646decSPatrick Venture std::fprintf(stderr, "Opening the verification file\n"); 846749ba1cSBrandon Kim if (updater->verifyFile(ipmi_flash::verifyBlobId, false)) 8555646decSPatrick Venture { 8655646decSPatrick Venture std::fprintf(stderr, "succeeded\n"); 8755646decSPatrick Venture } 8855646decSPatrick Venture else 8955646decSPatrick Venture { 9055646decSPatrick Venture std::fprintf(stderr, "failed\n"); 9114713becSPatrick Venture throw ToolException("Verification failed"); 9214713becSPatrick Venture } 9314713becSPatrick Venture 9414713becSPatrick Venture /* Trigger the update by opening and committing the update file. */ 9514713becSPatrick Venture std::fprintf(stderr, "Opening the update file\n"); 966749ba1cSBrandon Kim if (updater->verifyFile(ipmi_flash::updateBlobId, ignoreUpdate)) 9714713becSPatrick Venture { 9814713becSPatrick Venture std::fprintf(stderr, "succeeded\n"); 9914713becSPatrick Venture } 10014713becSPatrick Venture else 10114713becSPatrick Venture { 1025f2fcc4eSPatrick Venture /* Depending on the update mechanism used, this may be 1035f2fcc4eSPatrick Venture * uninteresting. For instance, for the static layout, we use the 1045f2fcc4eSPatrick Venture * reboot update mechanism. Which doesn't always lead to a 1055f2fcc4eSPatrick Venture * successful return before the BMC starts shutting down services. 10614713becSPatrick Venture */ 10714713becSPatrick Venture std::fprintf(stderr, "failed\n"); 10814713becSPatrick Venture throw ToolException("Update failed"); 10955646decSPatrick Venture } 110bf58cd64SPatrick Venture } 1115f2fcc4eSPatrick Venture catch (...) 1125f2fcc4eSPatrick Venture { 1135f2fcc4eSPatrick Venture updater->cleanArtifacts(); 1145f2fcc4eSPatrick Venture throw; 1155f2fcc4eSPatrick Venture } 1165f2fcc4eSPatrick Venture } 1179b534f06SPatrick Venture 1189b534f06SPatrick Venture } // namespace host_tool 119