annotated_operation.cc revision 726aeca1c665910593761a1990ee50f22d98f53f 
1//
2// Copyright (C) 2015 The Android Open Source Project
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 "update_engine/payload_generator/annotated_operation.h"
18
19#include <base/format_macros.h>
20#include <base/strings/string_number_conversions.h>
21#include <base/strings/stringprintf.h>
22
23#include "update_engine/common/utils.h"
24#include "update_engine/payload_consumer/payload_constants.h"
25
26namespace chromeos_update_engine {
27
28namespace {
29// Output the list of extents as (start_block, num_blocks) in the passed output
30// stream.
31void OutputExtents(std::ostream* os,
32                   const google::protobuf::RepeatedPtrField<Extent>& extents) {
33  for (const auto& extent : extents) {
34    *os << " (" << extent.start_block() << ", " << extent.num_blocks() << ")";
35  }
36}
37}  // namespace
38
39bool AnnotatedOperation::SetOperationBlob(const brillo::Blob& blob,
40                                          BlobFileWriter* blob_file) {
41  off_t data_offset = blob_file->StoreBlob(blob);
42  TEST_AND_RETURN_FALSE(data_offset != -1);
43  op.set_data_offset(data_offset);
44  op.set_data_length(blob.size());
45  return true;
46}
47
48std::ostream& operator<<(std::ostream& os, const AnnotatedOperation& aop) {
49  // For example, this prints:
50  // REPLACE_BZ 500 @3000
51  //   name: /foo/bar
52  //    dst: (123, 3) (127, 2)
53  os << InstallOperationTypeName(aop.op.type()) << " "  << aop.op.data_length();
54  if (aop.op.data_length() > 0)
55    os << " @" << aop.op.data_offset();
56  if (!aop.name.empty()) {
57    os << std::endl << "  name: " << aop.name;
58  }
59  if (aop.op.src_extents_size() != 0) {
60    os << std::endl << "   src:";
61    OutputExtents(&os, aop.op.src_extents());
62  }
63  if (aop.op.dst_extents_size() != 0) {
64    os << std::endl << "   dst:";
65    OutputExtents(&os, aop.op.dst_extents());
66  }
67  return os;
68}
69
70}  // namespace chromeos_update_engine
71