OLD | NEW |
1 // Copyright (c) 2010 The Chromium OS Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_FILESYSTEM_COPIER_ACTION_H__ | 5 #ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_FILESYSTEM_COPIER_ACTION_H__ |
6 #define CHROMEOS_PLATFORM_UPDATE_ENGINE_FILESYSTEM_COPIER_ACTION_H__ | 6 #define CHROMEOS_PLATFORM_UPDATE_ENGINE_FILESYSTEM_COPIER_ACTION_H__ |
7 | 7 |
8 #include <sys/stat.h> | 8 #include <sys/stat.h> |
9 #include <sys/types.h> | 9 #include <sys/types.h> |
10 | 10 |
(...skipping 19 matching lines...) Expand all Loading... |
30 class ActionTraits<FilesystemCopierAction> { | 30 class ActionTraits<FilesystemCopierAction> { |
31 public: | 31 public: |
32 // Takes the install plan as input | 32 // Takes the install plan as input |
33 typedef InstallPlan InputObjectType; | 33 typedef InstallPlan InputObjectType; |
34 // Passes the install plan as output | 34 // Passes the install plan as output |
35 typedef InstallPlan OutputObjectType; | 35 typedef InstallPlan OutputObjectType; |
36 }; | 36 }; |
37 | 37 |
38 class FilesystemCopierAction : public Action<FilesystemCopierAction> { | 38 class FilesystemCopierAction : public Action<FilesystemCopierAction> { |
39 public: | 39 public: |
40 explicit FilesystemCopierAction(bool copying_kernel_install_path); | 40 FilesystemCopierAction(bool copying_kernel_install_path, |
| 41 bool verify_hash); |
41 | 42 |
42 typedef ActionTraits<FilesystemCopierAction>::InputObjectType | 43 typedef ActionTraits<FilesystemCopierAction>::InputObjectType |
43 InputObjectType; | 44 InputObjectType; |
44 typedef ActionTraits<FilesystemCopierAction>::OutputObjectType | 45 typedef ActionTraits<FilesystemCopierAction>::OutputObjectType |
45 OutputObjectType; | 46 OutputObjectType; |
46 void PerformAction(); | 47 void PerformAction(); |
47 void TerminateProcessing(); | 48 void TerminateProcessing(); |
48 | 49 |
49 // Used for testing, so we can copy from somewhere other than root | 50 // Used for testing, so we can copy from somewhere other than root |
50 void set_copy_source(const std::string& path) { copy_source_ = path; } | 51 void set_copy_source(const std::string& path) { copy_source_ = path; } |
51 | 52 |
52 // Debugging/logging | 53 // Debugging/logging |
53 static std::string StaticType() { return "FilesystemCopierAction"; } | 54 static std::string StaticType() { return "FilesystemCopierAction"; } |
54 std::string Type() const { return StaticType(); } | 55 std::string Type() const { return StaticType(); } |
55 | 56 |
56 private: | 57 private: |
57 friend class FilesystemCopierActionTest; | 58 friend class FilesystemCopierActionTest; |
58 FRIEND_TEST(FilesystemCopierActionTest, RunAsRootDetermineFilesystemSizeTest); | 59 FRIEND_TEST(FilesystemCopierActionTest, RunAsRootDetermineFilesystemSizeTest); |
59 | 60 |
60 // Ping-pong buffers generally cycle through the following states: | 61 // Ping-pong buffers generally cycle through the following states: |
61 // Empty->Reading->Full->Writing->Empty. | 62 // Empty->Reading->Full->Writing->Empty. In hash verification mode the state |
| 63 // is never set to Writing. |
62 enum BufferState { | 64 enum BufferState { |
63 kBufferStateEmpty, | 65 kBufferStateEmpty, |
64 kBufferStateReading, | 66 kBufferStateReading, |
65 kBufferStateFull, | 67 kBufferStateFull, |
66 kBufferStateWriting | 68 kBufferStateWriting |
67 }; | 69 }; |
68 | 70 |
69 // Callbacks from glib when the read/write operation is done. | 71 // Callbacks from glib when the read/write operation is done. |
70 void AsyncReadReadyCallback(GObject *source_object, GAsyncResult *res); | 72 void AsyncReadReadyCallback(GObject *source_object, GAsyncResult *res); |
71 static void StaticAsyncReadReadyCallback(GObject *source_object, | 73 static void StaticAsyncReadReadyCallback(GObject *source_object, |
72 GAsyncResult *res, | 74 GAsyncResult *res, |
73 gpointer user_data); | 75 gpointer user_data); |
74 | 76 |
75 void AsyncWriteReadyCallback(GObject *source_object, GAsyncResult *res); | 77 void AsyncWriteReadyCallback(GObject *source_object, GAsyncResult *res); |
76 static void StaticAsyncWriteReadyCallback(GObject *source_object, | 78 static void StaticAsyncWriteReadyCallback(GObject *source_object, |
77 GAsyncResult *res, | 79 GAsyncResult *res, |
78 gpointer user_data); | 80 gpointer user_data); |
79 | 81 |
80 // Based on the state of the ping-pong buffers spawns appropriate read/write | 82 // Based on the state of the ping-pong buffers spawns appropriate read/write |
81 // actions asynchronously. | 83 // actions asynchronously. |
82 void SpawnAsyncActions(); | 84 void SpawnAsyncActions(); |
83 | 85 |
84 // Cleans up all the variables we use for async operations and tells the | 86 // Cleans up all the variables we use for async operations and tells the |
85 // ActionProcessor we're done w/ success as passed in. |cancelled_| should be | 87 // ActionProcessor we're done w/ |code| as passed in. |cancelled_| should be |
86 // true if TerminateProcessing() was called. | 88 // true if TerminateProcessing() was called. |
87 void Cleanup(bool success); | 89 void Cleanup(ActionExitCode code); |
88 | 90 |
89 // Determine, if possible, the source file system size to avoid copying the | 91 // Determine, if possible, the source file system size to avoid copying the |
90 // whole partition. Currently this supports only the root file system assuming | 92 // whole partition. Currently this supports only the root file system assuming |
91 // it's ext3-compatible. | 93 // it's ext3-compatible. |
92 void DetermineFilesystemSize(int fd); | 94 void DetermineFilesystemSize(int fd); |
93 | 95 |
94 // Returns the number of bytes to read based on the size of the buffer and the | |
95 // filesystem size. | |
96 int64_t GetBytesToRead(); | |
97 | |
98 // If true, this action is copying to the kernel_install_path from | 96 // If true, this action is copying to the kernel_install_path from |
99 // the install plan, otherwise it's copying just to the install_path. | 97 // the install plan, otherwise it's copying just to the install_path. |
100 const bool copying_kernel_install_path_; | 98 const bool copying_kernel_install_path_; |
101 | 99 |
| 100 // If true, this action is running in applied update hash verification mode -- |
| 101 // it computes a hash for the target install path and compares it against the |
| 102 // expected value. |
| 103 const bool verify_hash_; |
| 104 |
102 // The path to copy from. If empty (the default), the source is from the | 105 // The path to copy from. If empty (the default), the source is from the |
103 // passed in InstallPlan. | 106 // passed in InstallPlan. |
104 std::string copy_source_; | 107 std::string copy_source_; |
105 | 108 |
106 // If non-NULL, these are GUnixInputStream objects for the opened | 109 // If non-NULL, these are GUnixInputStream objects for the opened |
107 // source/destination partitions. | 110 // source/destination partitions. |
108 GInputStream* src_stream_; | 111 GInputStream* src_stream_; |
109 GOutputStream* dst_stream_; | 112 GOutputStream* dst_stream_; |
110 | 113 |
111 // Ping-pong buffers for storing data we read/write. Only one buffer is being | 114 // Ping-pong buffers for storing data we read/write. Only one buffer is being |
(...skipping 23 matching lines...) Expand all Loading... |
135 // field is initialized when the action is started and decremented as more | 138 // field is initialized when the action is started and decremented as more |
136 // bytes get copied. | 139 // bytes get copied. |
137 int64_t filesystem_size_; | 140 int64_t filesystem_size_; |
138 | 141 |
139 DISALLOW_COPY_AND_ASSIGN(FilesystemCopierAction); | 142 DISALLOW_COPY_AND_ASSIGN(FilesystemCopierAction); |
140 }; | 143 }; |
141 | 144 |
142 } // namespace chromeos_update_engine | 145 } // namespace chromeos_update_engine |
143 | 146 |
144 #endif // CHROMEOS_PLATFORM_UPDATE_ENGINE_FILESYSTEM_COPIER_ACTION_H__ | 147 #endif // CHROMEOS_PLATFORM_UPDATE_ENGINE_FILESYSTEM_COPIER_ACTION_H__ |
OLD | NEW |