| OLD | NEW |
|---|
| (Empty) |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "update_engine/set_bootable_flag_action.h" | |
| 6 #include <sys/stat.h> | |
| 7 #include <sys/types.h> | |
| 8 #include <errno.h> | |
| 9 #include <fcntl.h> | |
| 10 #include <string> | |
| 11 #include <vector> | |
| 12 #include "update_engine/subprocess.h" | |
| 13 #include "update_engine/utils.h" | |
| 14 | |
| 15 using std::string; | |
| 16 using std::vector; | |
| 17 | |
| 18 namespace chromeos_update_engine { | |
| 19 | |
| 20 namespace { | |
| 21 const char* const kGpt = "/usr/bin/gpt"; | |
| 22 const ssize_t kPmbrLength = 512; | |
| 23 } | |
| 24 | |
| 25 void SetBootableFlagAction::PerformAction() { | |
| 26 ScopedActionCompleter completer(processor_, this); | |
| 27 | |
| 28 TEST_AND_RETURN(HasInputObject()); | |
| 29 const InstallPlan install_plan = GetInputObject(); | |
| 30 TEST_AND_RETURN(!install_plan.install_path.empty()); | |
| 31 const string partition = install_plan.install_path; | |
| 32 string root_device = utils::RootDevice(partition); | |
| 33 string partition_number = utils::PartitionNumber(partition); | |
| 34 | |
| 35 utils::BootLoader boot_loader; | |
| 36 TEST_AND_RETURN(utils::GetBootloader(&boot_loader)); | |
| 37 | |
| 38 // For now, only support Syslinux bootloader | |
| 39 TEST_AND_RETURN(boot_loader == utils::BootLoader_SYSLINUX); | |
| 40 | |
| 41 string temp_file; | |
| 42 TEST_AND_RETURN(utils::MakeTempFile("/tmp/pmbr_copy.XXXXXX", | |
| 43 &temp_file, | |
| 44 NULL)); | |
| 45 ScopedPathUnlinker temp_file_unlinker(temp_file); | |
| 46 | |
| 47 // Copy existing PMBR to temp file | |
| 48 vector<char> buf(kPmbrLength); | |
| 49 { | |
| 50 int fd = open(root_device.c_str(), O_RDONLY); | |
| 51 TEST_AND_RETURN(fd >= 0); | |
| 52 ScopedFdCloser fd_closer(&fd); | |
| 53 ssize_t bytes_read; | |
| 54 TEST_AND_RETURN(utils::PReadAll(fd, &buf[0], buf.size(), 0, &bytes_read)); | |
| 55 } | |
| 56 TEST_AND_RETURN(utils::WriteFile(temp_file.c_str(), &buf[0], buf.size())); | |
| 57 | |
| 58 // Call gpt tool to do the work | |
| 59 vector<string> command; | |
| 60 command.push_back(kGpt); | |
| 61 command.push_back("-S"); | |
| 62 command.push_back("boot"); | |
| 63 command.push_back("-i"); | |
| 64 command.push_back(partition_number); | |
| 65 command.push_back("-b"); | |
| 66 command.push_back(temp_file); | |
| 67 command.push_back(root_device); | |
| 68 int rc = 0; | |
| 69 Subprocess::SynchronousExec(command, &rc); | |
| 70 TEST_AND_RETURN(rc == 0); | |
| 71 | |
| 72 completer.set_code(kActionCodeSuccess); | |
| 73 if (HasOutputPipe()) | |
| 74 SetOutputObject(GetInputObject()); | |
| 75 } | |
| 76 | |
| 77 } // namespace chromeos_update_engine | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 4719002:
AU: Remove obsolete SetBootableFlagAction. (Closed)
Base URL: ssh://git@gitrw.chromium.org:9222/update_engine.git@master