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
26 namespace chromeos_update_engine {
27
28 namespace {
29 // Output the list of extents as (start_block, num_blocks) in the passed output
30 // stream.
OutputExtents(std::ostream * os,const google::protobuf::RepeatedPtrField<Extent> & extents)31 void 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
SetOperationBlob(const brillo::Blob & blob,BlobFileWriter * blob_file)39 bool AnnotatedOperation::SetOperationBlob(const brillo::Blob& blob,
40 BlobFileWriter* blob_file) {
41 if (blob.empty()) {
42 op.clear_data_offset();
43 op.clear_data_length();
44 return true;
45 }
46 off_t data_offset = blob_file->StoreBlob(blob);
47 TEST_AND_RETURN_FALSE(data_offset != -1);
48 op.set_data_offset(data_offset);
49 op.set_data_length(blob.size());
50 return true;
51 }
52
operator <<(std::ostream & os,const AnnotatedOperation & aop)53 std::ostream& operator<<(std::ostream& os, const AnnotatedOperation& aop) {
54 // For example, this prints:
55 // REPLACE_BZ 500 @3000
56 // name: /foo/bar
57 // dst: (123, 3) (127, 2)
58 os << InstallOperationTypeName(aop.op.type()) << " " << aop.op.data_length();
59 if (aop.op.data_length() > 0)
60 os << " @" << aop.op.data_offset();
61 if (!aop.name.empty()) {
62 os << std::endl << " name: " << aop.name;
63 }
64 if (aop.op.src_extents_size() != 0) {
65 os << std::endl << " src:";
66 OutputExtents(&os, aop.op.src_extents());
67 }
68 if (aop.op.dst_extents_size() != 0) {
69 os << std::endl << " dst:";
70 OutputExtents(&os, aop.op.dst_extents());
71 }
72 return os;
73 }
74
75 } // namespace chromeos_update_engine
76