AU: support for generating new style full updates.

generate_delta_main.cc

 // Copyright (c) 2010 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.
 
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <errno.h>
 #include <fcntl.h>
 #include <unistd.h>
 
@@ skipping 74 matching lines @@
       ssize_t bytes_read;
       CHECK(utils::PReadAll(fd, &buf[0], buf.size(), offset, &bytes_read));
       if (bytes_read == 0)
         break;
       CHECK_EQ(performer.Write(&buf[0], bytes_read), bytes_read);
     }
     CHECK_EQ(performer.Close(), 0);
     LOG(INFO) << "done applying delta.";
     return 0;
   }
-  CHECK(!FLAGS_old_dir.empty());
-  CHECK(!FLAGS_new_dir.empty());
-  CHECK(!FLAGS_old_image.empty());
   CHECK(!FLAGS_new_image.empty());
   CHECK(!FLAGS_out_file.empty());
-  CHECK(!FLAGS_old_kernel.empty());
   CHECK(!FLAGS_new_kernel.empty());
-  if ((!IsDir(FLAGS_old_dir.c_str())) || (!IsDir(FLAGS_new_dir.c_str()))) {
-    LOG(FATAL) << "old_dir or new_dir not directory";
+  if (FLAGS_old_image.empty()) {
+    LOG(INFO) << "Generating full update";
+  } else {
+    LOG(INFO) << "Generating delta update";
+    CHECK(!FLAGS_old_kernel.empty());
+    CHECK(!FLAGS_old_dir.empty());
+    CHECK(!FLAGS_new_dir.empty());
+    if ((!IsDir(FLAGS_old_dir.c_str())) || (!IsDir(FLAGS_new_dir.c_str()))) {
+      LOG(FATAL) << "old_dir or new_dir not directory";
+    }
   }
-
   DeltaDiffGenerator::GenerateDeltaUpdateFile(FLAGS_old_dir,
                                               FLAGS_old_image,
                                               FLAGS_new_dir,
                                               FLAGS_new_image,
                                               FLAGS_old_kernel,
                                               FLAGS_new_kernel,
                                               FLAGS_out_file,
                                               FLAGS_private_key);
 
   return 0;
 }
 
 }  // namespace {}
 
 }  // namespace chromeos_update_engine
 
 int main(int argc, char** argv) {
   return chromeos_update_engine::Main(argc, argv);
 }