AU: Start checkpointing update progress.

download_action.h

 // Copyright (c) 2009 The Chromium OS Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_DOWNLOAD_ACTION_H__
 #define CHROMEOS_PLATFORM_UPDATE_ENGINE_DOWNLOAD_ACTION_H__
 
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <fcntl.h>
@@ skipping 31 matching lines @@
 
   // Called periodically after bytes are received. This method will be
   // invoked only if the download is active. |bytes_received| is the
   // number of bytes downloaded thus far. |total| is the number of
   // bytes expected.
   virtual void BytesReceived(uint64_t bytes_received, uint64_t total) = 0;
 };
 
 class DownloadAction;
 class NoneType;
+class PrefsInterface;
 
 template<>
 class ActionTraits<DownloadAction> {
  public:
   // Takes and returns an InstallPlan
   typedef InstallPlan InputObjectType;
   typedef InstallPlan OutputObjectType;
 };
 
 class DownloadAction : public Action<DownloadAction>,
                        public HttpFetcherDelegate {
  public:
   // Takes ownership of the passed in HttpFetcher. Useful for testing.
   // A good calling pattern is:
   // DownloadAction(new WhateverHttpFetcher);
-  DownloadAction(HttpFetcher* http_fetcher);
+  DownloadAction(PrefsInterface* prefs, HttpFetcher* http_fetcher);
   virtual ~DownloadAction();
   typedef ActionTraits<DownloadAction>::InputObjectType InputObjectType;
   typedef ActionTraits<DownloadAction>::OutputObjectType OutputObjectType;
   void PerformAction();
   void TerminateProcessing();
 
   // Testing
   void SetTestFileWriter(FileWriter* writer) {
     writer_ = writer;
   }
@@ skipping 11 matching lines @@
 
   DownloadActionDelegate* delegate() const { return delegate_; }
   void set_delegate(DownloadActionDelegate* delegate) {
     delegate_ = delegate;
   }
 
  private:
   // The InstallPlan passed in
   InstallPlan install_plan_;
 
+  // Update Engine preference store.
+  PrefsInterface* prefs_;
+
   // The FileWriter that downloaded data should be written to. It will
   // either point to *decompressing_file_writer_ or *delta_performer_.
   FileWriter* writer_;
 
   // These are used for full updates:
   scoped_ptr<GzipDecompressingFileWriter> decompressing_file_writer_;
   scoped_ptr<SplitFileWriter> split_file_writer_;
   scoped_ptr<DirectFileWriter> kernel_file_writer_;
   scoped_ptr<DirectFileWriter> rootfs_file_writer_;
   scoped_ptr<BufferedFileWriter> kernel_buffered_file_writer_;
@@ skipping 15 matching lines @@
   DISALLOW_COPY_AND_ASSIGN(DownloadAction);
 };
 
 // We want to be sure that we're compiled with large file support on linux,
 // just in case we find ourselves downloading large images.
 COMPILE_ASSERT(8 == sizeof(off_t), off_t_not_64_bit);
 
 }  // namespace chromeos_update_engine
 
 #endif  // CHROMEOS_PLATFORM_UPDATE_ENGINE_DOWNLOAD_ACTION_H__