blob: a7aeeb5ff890b2447b9f180d4439cc3f2889608b [file] [log] [blame]
Alex Deymo20891f92015-10-12 17:28:04 -07001// Copyright 2015 The Chromium OS 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#include <err.h>
Alex Deymoe4458302017-10-26 16:40:01 +02006#include <fcntl.h>
Tianjie Xu1f1cdb22017-11-20 11:05:55 -08007#include <getopt.h>
Alex Deymoe4458302017-10-26 16:40:01 +02008#include <sys/mman.h>
9#include <sys/stat.h>
10#include <sys/types.h>
11#include <unistd.h>
12
13#include <stddef.h>
14#include <stdio.h>
15#include <stdlib.h>
16
Tianjie Xu1f1cdb22017-11-20 11:05:55 -080017#include <iostream>
Alex Deymoe4458302017-10-26 16:40:01 +020018#include <limits>
Alex Deymo20891f92015-10-12 17:28:04 -070019
Alex Deymoddf9db52017-03-02 16:10:41 -080020#include "bsdiff/bsdiff.h"
Tianjie Xu1f1cdb22017-11-20 11:05:55 -080021#include "bsdiff/bsdiff_arguments.h"
22#include "bsdiff/constants.h"
Alex Deymoe4458302017-10-26 16:40:01 +020023#include "bsdiff/patch_writer_factory.h"
24
25namespace {
26
27// mmap() the passed |filename| to read-only memory and store in |filesize| the
28// size of the file. To release the memory, call munmap with the returned
29// pointer and filesize. In case of error returns nullptr.
30void* MapFile(const char* filename, size_t* filesize) {
31 int fd = open(filename, O_RDONLY);
32 if (fd < 0) {
33 perror("open()");
34 return nullptr;
35 }
36
37 struct stat st;
38 fstat(fd, &st);
39 if (static_cast<uint64_t>(st.st_size) > std::numeric_limits<size_t>::max()) {
40 fprintf(stderr, "File too big\n");
41 close(fd);
42 return nullptr;
43 }
44 *filesize = st.st_size;
45
46 void* ret = mmap(nullptr, st.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
47 if (ret == MAP_FAILED) {
48 perror("mmap()");
49 close(fd);
50 return nullptr;
51 }
52 close(fd);
53 return ret;
54}
55
56// Generate bsdiff patch from the |old_filename| file to the |new_filename|
Tianjie Xu1f1cdb22017-11-20 11:05:55 -080057// file with options in |arguments|. Store the resulting patch in a new
58// |patch_filename| file. Returns 0 on success.
Alex Deymoe4458302017-10-26 16:40:01 +020059int GenerateBsdiffFromFiles(const char* old_filename,
60 const char* new_filename,
Tianjie Xu1f1cdb22017-11-20 11:05:55 -080061 const char* patch_filename,
62 const bsdiff::BsdiffArguments& arguments) {
63 size_t oldsize;
Alex Deymoe4458302017-10-26 16:40:01 +020064 uint8_t* old_buf = static_cast<uint8_t*>(MapFile(old_filename, &oldsize));
Tianjie Xu1f1cdb22017-11-20 11:05:55 -080065 if (!old_buf) {
66 return 1;
Alex Deymoe4458302017-10-26 16:40:01 +020067 }
68
Tianjie Xu1f1cdb22017-11-20 11:05:55 -080069 size_t newsize;
70 uint8_t* new_buf = static_cast<uint8_t*>(MapFile(new_filename, &newsize));
71 if (!new_buf) {
Alex Deymoe4458302017-10-26 16:40:01 +020072 munmap(old_buf, oldsize);
Tianjie Xu1f1cdb22017-11-20 11:05:55 -080073 return 1;
74 }
Alex Deymoe4458302017-10-26 16:40:01 +020075
Tianjie Xu1f1cdb22017-11-20 11:05:55 -080076 std::unique_ptr<bsdiff::PatchWriterInterface> patch_writer;
Alex Deymoe790a3b2017-11-09 18:09:11 +010077 std::vector<uint8_t> raw_data;
78
Tianjie Xu1f1cdb22017-11-20 11:05:55 -080079 if (arguments.format() == bsdiff::BsdiffFormat::kLegacy) {
80 patch_writer = bsdiff::CreateBsdiffPatchWriter(patch_filename);
81 } else if (arguments.format() == bsdiff::BsdiffFormat::kBsdf2) {
Tianjie Xu2e70b552018-03-02 16:22:10 -080082 patch_writer = bsdiff::CreateBSDF2PatchWriter(patch_filename,
Tianjie Xu77833b62018-03-07 18:13:47 -080083 arguments.compressor_types(),
Tianjie Xu2e70b552018-03-02 16:22:10 -080084 arguments.brotli_quality());
Alex Deymoe790a3b2017-11-09 18:09:11 +010085 } else if (arguments.format() == bsdiff::BsdiffFormat::kEndsley) {
Tianjie Xu2e70b552018-03-02 16:22:10 -080086 patch_writer = bsdiff::CreateEndsleyPatchWriter(
Tianjie Xu77833b62018-03-07 18:13:47 -080087 &raw_data, arguments.compressor_types()[0], arguments.brotli_quality());
Tianjie Xu1f1cdb22017-11-20 11:05:55 -080088 } else {
89 std::cerr << "unexpected bsdiff format." << std::endl;
90 return 1;
91 }
92
Alex Deymocdd45ca2018-02-14 19:18:49 +010093 int ret = bsdiff::bsdiff(old_buf, oldsize, new_buf, newsize,
94 arguments.min_length(), patch_writer.get(), nullptr);
95
96 munmap(old_buf, oldsize);
97 munmap(new_buf, newsize);
98
Alex Deymoe790a3b2017-11-09 18:09:11 +010099 if (!ret && arguments.format() == bsdiff::BsdiffFormat::kEndsley) {
100 // Store the raw_data on disk.
101 FILE* fp = fopen(patch_filename, "wb");
102 if (!fp) {
103 perror("Opening the patch file");
104 return 1;
105 }
106 if (raw_data.size() != fwrite(raw_data.data(), 1, raw_data.size(), fp)) {
107 perror("Writing to the patch file");
108 ret = 1;
109 }
110 fclose(fp);
111 }
Alex Deymocdd45ca2018-02-14 19:18:49 +0100112 return ret;
Tianjie Xu1f1cdb22017-11-20 11:05:55 -0800113}
114
115void PrintUsage(const std::string& proc_name) {
116 std::cerr << "usage: " << proc_name
117 << " [options] oldfile newfile patchfile\n";
Alex Deymoe790a3b2017-11-09 18:09:11 +0100118 std::cerr << " --format <legacy|bsdiff40|bsdf2|endsley> The format of the"
119 " bsdiff patch.\n"
120 << " --minlen LEN The minimum match length "
121 "required to consider a match in the algorithm.\n"
122 << " --type <bz2|brotli|nocompression> The algorithm to compress "
Tianjie Xu77833b62018-03-07 18:13:47 -0800123 "the patch, bsdf2 format only. Multiple supported compressors "
124 "should be split by ':', e.g. bz2:brotli.\n"
Tianjie Xu2e70b552018-03-02 16:22:10 -0800125 << " --brotli_quality Quality of the brotli "
126 "compressor.\n";
Alex Deymoe4458302017-10-26 16:40:01 +0200127}
128
129} // namespace
Alex Deymo20891f92015-10-12 17:28:04 -0700130
131int main(int argc, char* argv[]) {
Tianjie Xu1f1cdb22017-11-20 11:05:55 -0800132 bsdiff::BsdiffArguments arguments;
Alex Deymo20891f92015-10-12 17:28:04 -0700133
Tianjie Xu1f1cdb22017-11-20 11:05:55 -0800134 if (!arguments.ParseCommandLine(argc, argv)) {
135 PrintUsage(argv[0]);
136 return 1;
137 }
138
139 // The optind will be updated in ParseCommandLine to parse the options; and
140 // we expect the rest of the arguments to be oldfile, newfile, patchfile.
141 if (!arguments.IsValid() || argc - optind != 3) {
142 PrintUsage(argv[0]);
143 return 1;
144 }
145
146 return GenerateBsdiffFromFiles(argv[optind], argv[optind + 1],
147 argv[optind + 2], arguments);
Alex Deymo20891f92015-10-12 17:28:04 -0700148}