courgette_minimal_tool.cc revision 90dce4d38c5ff5333bea97d859d4e484e27edf0c 
1// Copyright (c) 2009 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5// 'courgette_minimal_tool' is not meant to be a serious command-line tool.  It
6// has the minimum logic to apply a Courgette patch to a file.  The main purpose
7// is to monitor the code size of the patcher.
8
9#include <string>
10
11#include "courgette/third_party/bsdiff.h"
12#include "courgette/courgette.h"
13#include "courgette/streams.h"
14
15void PrintHelp() {
16  fprintf(stderr,
17    "Usage:\n"
18    "  courgette_minimal_tool <old-file-input> <patch-file-input>"
19    " <new-file-output>\n"
20    "\n");
21}
22
23void UsageProblem(const char* message) {
24  fprintf(stderr, "%s\n", message);
25  PrintHelp();
26  exit(1);
27}
28
29void Problem(const char* message) {
30  fprintf(stderr, "%s\n", message);
31  exit(1);
32}
33
34#if defined(OS_WIN)
35int wmain(int argc, const wchar_t* argv[]) {
36#else
37int main(int argc, const char* argv[]) {
38#endif
39  if (argc != 4)
40    UsageProblem("bad args");
41
42  courgette::Status status =
43      courgette::ApplyEnsemblePatch(argv[1], argv[2], argv[3]);
44
45  if (status != courgette::C_OK) {
46    if (status == courgette::C_READ_OPEN_ERROR) Problem("Can't open file.");
47    if (status == courgette::C_WRITE_OPEN_ERROR) Problem("Can't open file.");
48    if (status == courgette::C_READ_ERROR) Problem("Can't read from file.");
49    if (status == courgette::C_WRITE_ERROR) Problem("Can't write to file.");
50    Problem("patch failed.");
51  }
52
53  return 0;
54}
55