1 // 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>
6 #include <fcntl.h>
7 #include <getopt.h>
8 #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
17 #include <iostream>
18 #include <limits>
19
20 #include "bsdiff/bsdiff.h"
21 #include "bsdiff/bsdiff_arguments.h"
22 #include "bsdiff/constants.h"
23 #include "bsdiff/patch_writer_factory.h"
24
25 namespace {
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.
MapFile(const char * filename,size_t * filesize)30 void* 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|
57 // file with options in |arguments|. Store the resulting patch in a new
58 // |patch_filename| file. Returns 0 on success.
GenerateBsdiffFromFiles(const char * old_filename,const char * new_filename,const char * patch_filename,const bsdiff::BsdiffArguments & arguments)59 int GenerateBsdiffFromFiles(const char* old_filename,
60 const char* new_filename,
61 const char* patch_filename,
62 const bsdiff::BsdiffArguments& arguments) {
63 size_t oldsize;
64 uint8_t* old_buf = static_cast<uint8_t*>(MapFile(old_filename, &oldsize));
65 if (!old_buf) {
66 return 1;
67 }
68
69 size_t newsize;
70 uint8_t* new_buf = static_cast<uint8_t*>(MapFile(new_filename, &newsize));
71 if (!new_buf) {
72 munmap(old_buf, oldsize);
73 return 1;
74 }
75
76 std::unique_ptr<bsdiff::PatchWriterInterface> patch_writer;
77 std::vector<uint8_t> raw_data;
78
79 if (arguments.format() == bsdiff::BsdiffFormat::kLegacy) {
80 patch_writer = bsdiff::CreateBsdiffPatchWriter(patch_filename);
81 } else if (arguments.format() == bsdiff::BsdiffFormat::kBsdf2) {
82 patch_writer = bsdiff::CreateBSDF2PatchWriter(patch_filename,
83 arguments.compressor_types(),
84 arguments.brotli_quality());
85 } else if (arguments.format() == bsdiff::BsdiffFormat::kEndsley) {
86 patch_writer = bsdiff::CreateEndsleyPatchWriter(
87 &raw_data, arguments.compressor_types()[0], arguments.brotli_quality());
88 } else {
89 std::cerr << "unexpected bsdiff format." << std::endl;
90 return 1;
91 }
92
93 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
99 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 }
112 return ret;
113 }
114
PrintUsage(const std::string & proc_name)115 void PrintUsage(const std::string& proc_name) {
116 std::cerr << "usage: " << proc_name
117 << " [options] oldfile newfile patchfile\n";
118 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 "
123 "the patch, bsdf2 format only. Multiple supported compressors "
124 "should be split by ':', e.g. bz2:brotli.\n"
125 << " --brotli_quality Quality of the brotli "
126 "compressor.\n";
127 }
128
129 } // namespace
130
main(int argc,char * argv[])131 int main(int argc, char* argv[]) {
132 bsdiff::BsdiffArguments arguments;
133
134 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);
148 }
149