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 190533d0b0SPatrick Venture #include "blob_errors.hpp" 202bc23fe1SPatrick Venture #include "tool_errors.hpp" 210533d0b0SPatrick Venture 2200887597SPatrick Venture #include <algorithm> 23339dece8SPatrick Venture #include <cstring> 24af69625fSPatrick Venture #include <memory> 25af69625fSPatrick Venture 26*9b534f06SPatrick Venture namespace host_tool 27*9b534f06SPatrick Venture { 28*9b534f06SPatrick Venture 292bc23fe1SPatrick Venture void updaterMain(BlobInterface* blob, DataInterface* handler, 3000887597SPatrick Venture const std::string& imagePath, const std::string& signaturePath) 31bf58cd64SPatrick Venture { 32af69625fSPatrick Venture /* TODO(venture): Add optional parameter to specify the flash type, default 33af69625fSPatrick Venture * to legacy for now. 34af69625fSPatrick Venture */ 3500887597SPatrick Venture std::string goalFirmware = "/flash/image"; 3600887597SPatrick Venture 370bf8bf0cSPatrick Venture /* Get list of blob_ids, check for /flash/image, or /flash/tarball. 380bf8bf0cSPatrick Venture * TODO(venture) the mechanism doesn't care, but the caller of burn_my_bmc 390bf8bf0cSPatrick Venture * will have in mind which they're sending and we need to verify it's 400bf8bf0cSPatrick Venture * available and use it. 410bf8bf0cSPatrick Venture */ 4200887597SPatrick Venture std::vector<std::string> blobs = blob->getBlobList(); 43339dece8SPatrick Venture auto blobInst = std::find_if( 44339dece8SPatrick Venture blobs.begin(), blobs.end(), [&goalFirmware](const auto& iter) { 45339dece8SPatrick Venture /* Running into weird scenarios where the string comparison doesn't 46339dece8SPatrick Venture * work. TODO: revisit. 47339dece8SPatrick Venture */ 48339dece8SPatrick Venture return (0 == std::memcmp(goalFirmware.c_str(), iter.c_str(), 49339dece8SPatrick Venture goalFirmware.length())); 50339dece8SPatrick Venture // return (goalFirmware.compare(iter)); 51339dece8SPatrick Venture }); 5200887597SPatrick Venture if (blobInst == blobs.end()) 5300887597SPatrick Venture { 542bc23fe1SPatrick Venture throw ToolException(goalFirmware + " not found"); 5500887597SPatrick Venture } 56af69625fSPatrick Venture 57af69625fSPatrick Venture /* Call stat on /flash/image (or /flash/tarball) and check if data interface 5800887597SPatrick Venture * is supported. 5900887597SPatrick Venture */ 60339dece8SPatrick Venture StatResponse stat; 61339dece8SPatrick Venture try 62339dece8SPatrick Venture { 63339dece8SPatrick Venture stat = blob->getStat(goalFirmware); 64339dece8SPatrick Venture } 65339dece8SPatrick Venture catch (const BlobException& b) 66339dece8SPatrick Venture { 67339dece8SPatrick Venture throw ToolException("blob exception received: " + 68339dece8SPatrick Venture std::string(b.what())); 69339dece8SPatrick Venture } 70339dece8SPatrick Venture 71aa32a36aSPatrick Venture auto supported = handler->supportedType(); 72aa32a36aSPatrick Venture if ((stat.blob_state & supported) == 0) 738a55dcbdSPatrick Venture { 742bc23fe1SPatrick Venture throw ToolException("data interface selected not supported."); 758a55dcbdSPatrick Venture } 76af69625fSPatrick Venture 770533d0b0SPatrick Venture /* Yay, our data handler is supported. */ 780533d0b0SPatrick Venture std::uint16_t session; 790533d0b0SPatrick Venture try 800533d0b0SPatrick Venture { 81aa32a36aSPatrick Venture session = blob->openBlob(goalFirmware, supported); 820533d0b0SPatrick Venture } 830533d0b0SPatrick Venture catch (const BlobException& b) 840533d0b0SPatrick Venture { 852bc23fe1SPatrick Venture throw ToolException("blob exception received: " + 862bc23fe1SPatrick Venture std::string(b.what())); 870533d0b0SPatrick Venture } 880533d0b0SPatrick Venture 89fd6aaec8SPatrick Venture /* Send over the firmware image. */ 90fd6aaec8SPatrick Venture if (!handler->sendContents(imagePath, session)) 91fd6aaec8SPatrick Venture { 922bc23fe1SPatrick Venture throw ToolException("Failed to send contents of " + imagePath); 93fd6aaec8SPatrick Venture } 94fd6aaec8SPatrick Venture 95fd6aaec8SPatrick Venture /* Send over the hash contents. */ 96fd6aaec8SPatrick Venture /* Trigger the verification. */ 97fd6aaec8SPatrick Venture /* Check the verification. */ 98fd6aaec8SPatrick Venture 992bc23fe1SPatrick Venture return; 100bf58cd64SPatrick Venture } 101*9b534f06SPatrick Venture 102*9b534f06SPatrick Venture } // namespace host_tool 103