1// Protocol Buffers - Google's data interchange format
2// Copyright 2008 Google Inc.  All rights reserved.
3// http://code.google.com/p/protobuf/
4//
5// Redistribution and use in source and binary forms, with or without
6// modification, are permitted provided that the following conditions are
7// met:
8//
9//     * Redistributions of source code must retain the above copyright
10// notice, this list of conditions and the following disclaimer.
11//     * Redistributions in binary form must reproduce the above
12// copyright notice, this list of conditions and the following disclaimer
13// in the documentation and/or other materials provided with the
14// distribution.
15//     * Neither the name of Google Inc. nor the names of its
16// contributors may be used to endorse or promote products derived from
17// this software without specific prior written permission.
18//
19// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30
31// Protocol Buffers - Google's data interchange format
32// Copyright 2008 Google Inc.  All rights reserved.
33// http://code.google.com/p/protobuf/
34//
35// Redistribution and use in source and binary forms, with or without
36// modification, are permitted provided that the following conditions are
37// met:
38//
39//     * Redistributions of source code must retain the above copyright
40// notice, this list of conditions and the following disclaimer.
41//     * Redistributions in binary form must reproduce the above
42// copyright notice, this list of conditions and the following disclaimer
43// in the documentation and/or other materials provided with the
44// distribution.
45//     * Neither the name of Google Inc. nor the names of its
46// contributors may be used to endorse or promote products derived from
47// this software without specific prior written permission.
48//
49// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
50// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
51// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
52// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
53// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
54// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
55// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
56// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
57// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
58// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
59// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
60
61// Author: ambrose@google.com (Ambrose Feinstein),
62//         kenton@google.com (Kenton Varda)
63//
64// Based on http://www.pkware.com/documents/casestudies/APPNOTE.TXT
65
66#include <google/protobuf/compiler/zip_writer.h>
67#include <google/protobuf/io/coded_stream.h>
68
69namespace google {
70namespace protobuf {
71namespace compiler {
72
73static const uint32 kCRC32Table[256] = {
74  0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419, 0x706af48f,
75  0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4, 0xe0d5e91e, 0x97d2d988,
76  0x09b64c2b, 0x7eb17cbd, 0xe7b82d07, 0x90bf1d91, 0x1db71064, 0x6ab020f2,
77  0xf3b97148, 0x84be41de, 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7,
78  0x136c9856, 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,
79  0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4, 0xa2677172,
80  0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b, 0x35b5a8fa, 0x42b2986c,
81  0xdbbbc9d6, 0xacbcf940, 0x32d86ce3, 0x45df5c75, 0xdcd60dcf, 0xabd13d59,
82  0x26d930ac, 0x51de003a, 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423,
83  0xcfba9599, 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
84  0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190, 0x01db7106,
85  0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f, 0x9fbfe4a5, 0xe8b8d433,
86  0x7807c9a2, 0x0f00f934, 0x9609a88e, 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d,
87  0x91646c97, 0xe6635c01, 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e,
88  0x6c0695ed, 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
89  0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3, 0xfbd44c65,
90  0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2, 0x4adfa541, 0x3dd895d7,
91  0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a, 0x346ed9fc, 0xad678846, 0xda60b8d0,
92  0x44042d73, 0x33031de5, 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa,
93  0xbe0b1010, 0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
94  0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17, 0x2eb40d81,
95  0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6, 0x03b6e20c, 0x74b1d29a,
96  0xead54739, 0x9dd277af, 0x04db2615, 0x73dc1683, 0xe3630b12, 0x94643b84,
97  0x0d6d6a3e, 0x7a6a5aa8, 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1,
98  0xf00f9344, 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,
99  0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a, 0x67dd4acc,
100  0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5, 0xd6d6a3e8, 0xa1d1937e,
101  0x38d8c2c4, 0x4fdff252, 0xd1bb67f1, 0xa6bc5767, 0x3fb506dd, 0x48b2364b,
102  0xd80d2bda, 0xaf0a1b4c, 0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55,
103  0x316e8eef, 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
104  0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe, 0xb2bd0b28,
105  0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31, 0x2cd99e8b, 0x5bdeae1d,
106  0x9b64c2b0, 0xec63f226, 0x756aa39c, 0x026d930a, 0x9c0906a9, 0xeb0e363f,
107  0x72076785, 0x05005713, 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38,
108  0x92d28e9b, 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,
109  0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1, 0x18b74777,
110  0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c, 0x8f659eff, 0xf862ae69,
111  0x616bffd3, 0x166ccf45, 0xa00ae278, 0xd70dd2ee, 0x4e048354, 0x3903b3c2,
112  0xa7672661, 0xd06016f7, 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc,
113  0x40df0b66, 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
114  0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605, 0xcdd70693,
115  0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8, 0x5d681b02, 0x2a6f2b94,
116  0xb40bbe37, 0xc30c8ea1, 0x5a05df1b, 0x2d02ef8d
117};
118
119static uint32 ComputeCRC32(const string &buf) {
120  uint32 x = ~0U;
121  for (int i = 0; i < buf.size(); ++i) {
122    unsigned char c = buf[i];
123    x = kCRC32Table[(x ^ c) & 0xff] ^ (x >> 8);
124  }
125  return ~x;
126}
127
128static void WriteShort(io::CodedOutputStream *out, uint16 val) {
129  uint8 p[2];
130  p[0] = static_cast<uint8>(val);
131  p[1] = static_cast<uint8>(val >> 8);
132  out->WriteRaw(p, 2);
133}
134
135ZipWriter::ZipWriter(io::ZeroCopyOutputStream* raw_output)
136  : raw_output_(raw_output) {}
137ZipWriter::~ZipWriter() {}
138
139bool ZipWriter::Write(const string& filename, const string& contents) {
140  FileInfo info;
141
142  info.name = filename;
143  uint16 filename_size = filename.size();
144  info.offset = raw_output_->ByteCount();
145  info.size = contents.size();
146  info.crc32 = ComputeCRC32(contents);
147
148  files_.push_back(info);
149
150  // write file header
151  io::CodedOutputStream output(raw_output_);
152  output.WriteLittleEndian32(0x04034b50);  // magic
153  WriteShort(&output, 10);  // version needed to extract
154  WriteShort(&output, 0);  // flags
155  WriteShort(&output, 0);  // compression method: stored
156  WriteShort(&output, 0);  // last modified time
157  WriteShort(&output, 0);  // last modified date
158  output.WriteLittleEndian32(info.crc32);  // crc-32
159  output.WriteLittleEndian32(info.size);  // compressed size
160  output.WriteLittleEndian32(info.size);  // uncompressed size
161  WriteShort(&output, filename_size);  // file name length
162  WriteShort(&output, 0);   // extra field length
163  output.WriteString(filename);  // file name
164  output.WriteString(contents);  // file data
165
166  return !output.HadError();
167}
168
169bool ZipWriter::WriteDirectory() {
170  uint16 num_entries = files_.size();
171  uint32 dir_ofs = raw_output_->ByteCount();
172
173  // write central directory
174  io::CodedOutputStream output(raw_output_);
175  for (int i = 0; i < num_entries; ++i) {
176    const string &filename = files_[i].name;
177    uint16 filename_size = filename.size();
178    uint32 crc32 = files_[i].crc32;
179    uint32 size = files_[i].size;
180    uint32 offset = files_[i].offset;
181
182    output.WriteLittleEndian32(0x02014b50);  // magic
183    WriteShort(&output, 10);  // version made by
184    WriteShort(&output, 10);  // version needed to extract
185    WriteShort(&output, 0);  // flags
186    WriteShort(&output, 0);  // compression method: stored
187    WriteShort(&output, 0);  // last modified time
188    WriteShort(&output, 0);  // last modified date
189    output.WriteLittleEndian32(crc32);  // crc-32
190    output.WriteLittleEndian32(size);  // compressed size
191    output.WriteLittleEndian32(size);  // uncompressed size
192    WriteShort(&output, filename_size);  // file name length
193    WriteShort(&output, 0);   // extra field length
194    WriteShort(&output, 0);   // file comment length
195    WriteShort(&output, 0);   // starting disk number
196    WriteShort(&output, 0);   // internal file attributes
197    output.WriteLittleEndian32(0);  // external file attributes
198    output.WriteLittleEndian32(offset);  // local header offset
199    output.WriteString(filename);  // file name
200  }
201  uint32 dir_len = output.ByteCount();
202
203  // write end of central directory marker
204  output.WriteLittleEndian32(0x06054b50);  // magic
205  WriteShort(&output, 0);  // disk number
206  WriteShort(&output, 0);  // disk with start of central directory
207  WriteShort(&output, num_entries);  // central directory entries (this disk)
208  WriteShort(&output, num_entries);  // central directory entries (total)
209  output.WriteLittleEndian32(dir_len);  // central directory byte size
210  output.WriteLittleEndian32(dir_ofs);  // central directory offset
211  WriteShort(&output, 0);   // comment length
212
213  return output.HadError();
214}
215
216}  // namespace compiler
217}  // namespace protobuf
218}  // namespace google
219