Index: third_party/courgette/bsdiff_create.cc |
=================================================================== |
--- third_party/courgette/bsdiff_create.cc (revision 0) |
+++ third_party/courgette/bsdiff_create.cc (revision 0) |
@@ -0,0 +1,432 @@ |
+/* |
+ bsdiff.c -- Binary patch generator. |
+ |
+ Copyright 2003 Colin Percival |
+ |
+ For the terms under which this work may be distributed, please see |
+ the adjoining file "LICENSE". |
+ |
+ ChangeLog: |
+ 2005-05-05 - Use the modified header struct from bspatch.h; use 32-bit |
+ values throughout. |
+ --Benjamin Smedberg <benjamin@smedbergs.us> |
+ 2005-05-18 - Use the same CRC algorithm as bzip2, and leverage the CRC table |
+ provided by libbz2. |
+ --Darin Fisher <darin@meer.net> |
+ 2007-11-14 - Changed to use Crc from Lzma library instead of Bzip library |
+ --Rahul Kuchhal |
+ 2009-03-31 - Change to use Streams. Added lots of comments. |
+ --Stephen Adams <sra@chromium.org> |
+*/ |
+ |
+#include "third_party/courgette/bsdiff.h" |
+ |
+#include <stdlib.h> |
+#include <algorithm> |
+ |
+#include "base/logging.h" |
+#include "base/scoped_ptr.h" |
+#include "base/string_util.h" |
+#include "base/time.h" |
+ |
+#include "third_party/courgette/crc.h" |
+#include "third_party/courgette/streams.h" |
+ |
+namespace courgette { |
+ |
+// ------------------------------------------------------------------------ |
+// |
+// The following code is taken verbatim from 'bsdiff.c'. Please keep all the |
+// code formatting and variable names. The only changes from the original are |
+// replacing tabs with spaces, indentation, and using 'const'. |
+// |
+// The code appears to be a rewritten version of the suffix array algorithm |
+// presented in "Faster Suffix Sorting" by N. Jesper Larsson and Kunihiko |
+// Sadakane, special-cased for bytes. |
+ |
+static void |
+split(int *I,int *V,int start,int len,int h) |
+{ |
+ int i,j,k,x,tmp,jj,kk; |
+ |
+ if(len<16) { |
+ for(k=start;k<start+len;k+=j) { |
+ j=1;x=V[I[k]+h]; |
+ for(i=1;k+i<start+len;i++) { |
+ if(V[I[k+i]+h]<x) { |
+ x=V[I[k+i]+h]; |
+ j=0; |
+ }; |
+ if(V[I[k+i]+h]==x) { |
+ tmp=I[k+j];I[k+j]=I[k+i];I[k+i]=tmp; |
+ j++; |
+ }; |
+ }; |
+ for(i=0;i<j;i++) V[I[k+i]]=k+j-1; |
+ if(j==1) I[k]=-1; |
+ }; |
+ return; |
+ }; |
+ |
+ x=V[I[start+len/2]+h]; |
+ jj=0;kk=0; |
+ for(i=start;i<start+len;i++) { |
+ if(V[I[i]+h]<x) jj++; |
+ if(V[I[i]+h]==x) kk++; |
+ }; |
+ jj+=start;kk+=jj; |
+ |
+ i=start;j=0;k=0; |
+ while(i<jj) { |
+ if(V[I[i]+h]<x) { |
+ i++; |
+ } else if(V[I[i]+h]==x) { |
+ tmp=I[i];I[i]=I[jj+j];I[jj+j]=tmp; |
+ j++; |
+ } else { |
+ tmp=I[i];I[i]=I[kk+k];I[kk+k]=tmp; |
+ k++; |
+ }; |
+ }; |
+ |
+ while(jj+j<kk) { |
+ if(V[I[jj+j]+h]==x) { |
+ j++; |
+ } else { |
+ tmp=I[jj+j];I[jj+j]=I[kk+k];I[kk+k]=tmp; |
+ k++; |
+ }; |
+ }; |
+ |
+ if(jj>start) split(I,V,start,jj-start,h); |
+ |
+ for(i=0;i<kk-jj;i++) V[I[jj+i]]=kk-1; |
+ if(jj==kk-1) I[jj]=-1; |
+ |
+ if(start+len>kk) split(I,V,kk,start+len-kk,h); |
+} |
+ |
+static void |
+qsufsort(int *I,int *V,const unsigned char *old,int oldsize) |
+{ |
+ int buckets[256]; |
+ int i,h,len; |
+ |
+ for(i=0;i<256;i++) buckets[i]=0; |
+ for(i=0;i<oldsize;i++) buckets[old[i]]++; |
+ for(i=1;i<256;i++) buckets[i]+=buckets[i-1]; |
+ for(i=255;i>0;i--) buckets[i]=buckets[i-1]; |
+ buckets[0]=0; |
+ |
+ for(i=0;i<oldsize;i++) I[++buckets[old[i]]]=i; |
+ I[0]=oldsize; |
+ for(i=0;i<oldsize;i++) V[i]=buckets[old[i]]; |
+ V[oldsize]=0; |
+ for(i=1;i<256;i++) if(buckets[i]==buckets[i-1]+1) I[buckets[i]]=-1; |
+ I[0]=-1; |
+ |
+ for(h=1;I[0]!=-(oldsize+1);h+=h) { |
+ len=0; |
+ for(i=0;i<oldsize+1;) { |
+ if(I[i]<0) { |
+ len-=I[i]; |
+ i-=I[i]; |
+ } else { |
+ if(len) I[i-len]=-len; |
+ len=V[I[i]]+1-i; |
+ split(I,V,i,len,h); |
+ i+=len; |
+ len=0; |
+ }; |
+ }; |
+ if(len) I[i-len]=-len; |
+ }; |
+ |
+ for(i=0;i<oldsize+1;i++) I[V[i]]=i; |
+} |
+ |
+static int |
+matchlen(const unsigned char *old,int oldsize,const unsigned char *newbuf,int newsize) |
+{ |
+ int i; |
+ |
+ for(i=0;(i<oldsize)&&(i<newsize);i++) |
+ if(old[i]!=newbuf[i]) break; |
+ |
+ return i; |
+} |
+ |
+static int |
+search(int *I,const unsigned char *old,int oldsize, |
+ const unsigned char *newbuf,int newsize,int st,int en,int *pos) |
+{ |
+ int x,y; |
+ |
+ if(en-st<2) { |
+ x=matchlen(old+I[st],oldsize-I[st],newbuf,newsize); |
+ y=matchlen(old+I[en],oldsize-I[en],newbuf,newsize); |
+ |
+ if(x>y) { |
+ *pos=I[st]; |
+ return x; |
+ } else { |
+ *pos=I[en]; |
+ return y; |
+ } |
+ } |
+ |
+ x=st+(en-st)/2; |
+ if(memcmp(old+I[x],newbuf,std::min(oldsize-I[x],newsize))<0) { |
+ return search(I,old,oldsize,newbuf,newsize,x,en,pos); |
+ } else { |
+ return search(I,old,oldsize,newbuf,newsize,st,x,pos); |
+ } |
+} |
+ |
+// End of 'verbatim' code. |
+// ------------------------------------------------------------------------ |
+ |
+static void WriteHeader(SinkStream* stream, MBSPatchHeader* header) { |
+ stream->Write(header->tag, sizeof(header->tag)); |
+ stream->WriteVarint32(header->slen); |
+ stream->WriteVarint32(header->scrc32); |
+ stream->WriteVarint32(header->dlen); |
+ stream->WriteVarint32(header->cblen); |
+ stream->WriteVarint32(header->difflen); |
+ stream->WriteVarint32(header->extralen); |
+} |
+ |
+BSDiffStatus CreateBinaryPatch(SourceStream* old_stream, |
+ SourceStream* new_stream, |
+ SinkStream* patch_stream) |
+{ |
+ base::Time start_bsdiff_time = base::Time::Now(); |
+ LOG(INFO) << "Start bsdiff"; |
+ size_t initial_patch_stream_length = patch_stream->Length(); |
+ |
+ const uint8* old = old_stream->Buffer(); |
+ const int oldsize = old_stream->Remaining(); |
+ |
+ scoped_array<int> I(new(std::nothrow) int[oldsize + 1]); |
+ scoped_array<int> V(new(std::nothrow) int[oldsize + 1]); |
+ if (I == NULL || V == NULL) |
+ return MEM_ERROR; |
+ |
+ base::Time q_start_time = base::Time::Now(); |
+ qsufsort(I.get(), V.get(), old, oldsize); |
+ LOG(INFO) << " done qsufsort " |
+ << (base::Time::Now() - q_start_time).InSecondsF(); |
+ V.reset(); |
+ |
+ const uint8* newbuf = new_stream->Buffer(); |
+ const int newsize = new_stream->Remaining(); |
+ |
+ // Allocate newsize+1 bytes instead of newsize bytes to ensure that we never |
+ // try to malloc(0) and get a NULL pointer. |
+ |
+ scoped_array<uint8> diff_bytes_array(new(std::nothrow) uint8[newsize + 1]); |
+ scoped_array<uint8> extra_bytes_array(new(std::nothrow) uint8[newsize + 1]); |
+ if (diff_bytes_array == NULL || extra_bytes_array == NULL) |
+ return MEM_ERROR; |
+ |
+ uint8* diff_bytes = diff_bytes_array.get(); |
+ uint8* extra_bytes = extra_bytes_array.get(); |
+ int control_length = 0; |
+ int diff_bytes_length = 0; |
+ int diff_bytes_nonzero = 0; |
+ int extra_bytes_length = 0; |
+ int eblen = 0; |
+ |
+ SinkStream control_stream; |
+ |
+ // The patch format is a sequence of triples <copy,extra,seek> where 'copy' is |
+ // the number of bytes to copy from the old file (possibly with mistakes), |
+ // 'extra' is the number of bytes to copy from a stream of fresh bytes, and |
+ // 'seek' is an offset to move to the position to copy for the next triple. |
+ // |
+ // The invariant at the top of this loop is that we are committed to emitting |
+ // a triple for the part of |newbuf| surrounding a 'seed' match near |
+ // |lastscan|. We are searching for a second match that will be the 'seed' of |
+ // the next triple. As we scan through |newbuf|, one of four things can |
+ // happen at the current position |scan|: |
+ // |
+ // 1. We find a nice match that appears to be consistent with the current |
+ // seed. Continue scanning. It is likely that this match will become |
+ // part of the 'copy'. |
+ // |
+ // 2. We find match which does much better than extending the current seed |
+ // old match. Emit a triple for the current seed and take this match as |
+ // the new seed for a new triple. By 'much better' we remove 8 mismatched |
+ // bytes by taking the new seed. |
+ // |
+ // 3. There is not a good match. Continue scanning. These bytes will likely |
+ // become part of the 'extra'. |
+ // |
+ // 4. There is no match because we reached the end of the input, |newbuf|. |
+ |
+ // This is how the loop advances through the bytes of |newbuf|: |
+ // |
+ // ...012345678901234567890123456789... |
+ // ssssssssss Seed at |lastscan| |
+ // xxyyyxxyyxy |scan| forward, cases (3)(x) & (1)(y) |
+ // mmmmmmmm New match will start new seed case (2). |
+ // fffffffffffffff |lenf| = scan forward from |lastscan| |
+ // bbbb |lenb| = scan back from new seed |scan|. |
+ // ddddddddddddddd Emit diff bytes for the 'copy'. |
+ // xx Emit extra bytes. |
+ // ssssssssssss |lastscan = scan - lenb| is new seed. |
+ // x Cases (1) and (3) .... |
+ |
+ |
+ int lastscan = 0, lastpos = 0, lastoffset = 0; |
+ |
+ int scan = 0; |
+ int match_length = 0; |
+ |
+ while (scan < newsize) { |
+ int pos = 0; |
+ int oldscore = 0; // Count of how many bytes of the current match at |scan| |
+ // extend the match at |lastscan|. |
+ |
+ scan += match_length; |
+ for (int scsc = scan; scan < newsize; ++scan) { |
+ match_length = search(I.get(), old, oldsize, |
+ newbuf + scan, newsize - scan, |
+ 0, oldsize, &pos); |
+ |
+ for ( ; scsc < scan + match_length ; scsc++) |
+ if ((scsc + lastoffset < oldsize) && |
+ (old[scsc + lastoffset] == newbuf[scsc])) |
+ oldscore++; |
+ |
+ if ((match_length == oldscore) && (match_length != 0)) |
+ break; // Good continuing match, case (1) |
+ if (match_length > oldscore + 8) |
+ break; // New seed match, case (2) |
+ |
+ if ((scan + lastoffset < oldsize) && |
+ (old[scan + lastoffset] == newbuf[scan])) |
+ oldscore--; |
+ // Case (3) continues in this loop until we fall out of the loop (4). |
+ } |
+ |
+ if ((match_length != oldscore) || (scan == newsize)) { // Cases (2) and (4) |
+ // This next chunk of code finds the boundary between the bytes to be |
+ // copied as part of the current triple, and the bytes to be copied as |
+ // part of the next triple. The |lastscan| match is extended forwards as |
+ // far as possible provided doing to does not add too many mistakes. The |
+ // |scan| match is extended backwards in a similar way. |
+ |
+ // Extend the current match (if any) backwards. |lenb| is the maximal |
+ // extension for which less than half the byte positions in the extension |
+ // are wrong. |
+ int lenb = 0; |
+ if (scan < newsize) { // i.e. not case (4); there is a match to extend. |
+ int score = 0, Sb = 0; |
+ for (int i = 1; (scan >= lastscan + i) && (pos >= i); i++) { |
+ if (old[pos - i] == newbuf[scan - i]) score++; |
+ if (score*2 - i > Sb*2 - lenb) { Sb = score; lenb = i; } |
+ } |
+ } |
+ |
+ // Extend the lastscan match forward; |lenf| is the maximal extension for |
+ // which less than half of the byte positions in entire lastscan match are |
+ // wrong. There is a subtle point here: |lastscan| points to before the |
+ // seed match by |lenb| bytes from the previous iteration. This is why |
+ // the loop measures the total number of mistakes in the the match, not |
+ // just the from the match. |
+ int lenf = 0; |
+ { |
+ int score = 0, Sf = 0; |
+ for (int i = 0; (lastscan + i < scan) && (lastpos + i < oldsize); ) { |
+ if (old[lastpos + i] == newbuf[lastscan + i]) score++; |
+ i++; |
+ if (score*2 - i > Sf*2 - lenf) { Sf = score; lenf = i; } |
+ } |
+ } |
+ |
+ // If the extended scans overlap, pick a position in the overlap region |
+ // that maximizes the exact matching bytes. |
+ if (lastscan + lenf > scan - lenb) { |
+ int overlap = (lastscan + lenf) - (scan - lenb); |
+ int score = 0; |
+ int Ss = 0, lens = 0; |
+ for (int i = 0; i < overlap; i++) { |
+ if (newbuf[lastscan + lenf - overlap + i] == |
+ old[lastpos + lenf - overlap + i]) score++; |
+ if (newbuf[scan - lenb + i] == old[pos - lenb + i]) score--; |
+ if (score > Ss) { Ss = score; lens = i + 1; } |
+ } |
+ |
+ lenf += lens - overlap; |
+ lenb -= lens; |
+ }; |
+ |
+ for (int i = 0; i < lenf; i++) { |
+ uint8 diff_byte = newbuf[lastscan + i] - old[lastpos + i]; |
+ diff_bytes[diff_bytes_length + i] = diff_byte; |
+ if (diff_byte) |
+ ++diff_bytes_nonzero; |
+ } |
+ int gap = (scan - lenb) - (lastscan + lenf); |
+ for (int i = 0; i < gap; i++) |
+ extra_bytes[extra_bytes_length + i] = newbuf[lastscan + lenf + i]; |
+ |
+ diff_bytes_length += lenf; |
+ extra_bytes_length += gap; |
+ |
+ uint32 copy_count = lenf; |
+ uint32 extra_count = gap; |
+ int32 seek_adjustment = ((pos - lenb) - (lastpos + lenf)); |
+ |
+ control_stream.WriteVarint32(copy_count); |
+ control_stream.WriteVarint32(extra_count); |
+ control_stream.WriteVarint32Signed(seek_adjustment); |
+ ++control_length; |
+#ifdef DEBUG_bsmedberg |
+ LOG(INFO) << StringPrintf( |
+ "Writing a block: copy: %-8u extra: %-8u seek: %+-9d", |
+ copy_count, extra_count, seek_adjustment); |
+#endif |
+ |
+ lastscan = scan - lenb; // Include the backward extension in seed. |
+ lastpos = pos - lenb; // ditto. |
+ lastoffset = lastpos - lastscan; |
+ } |
+ } |
+ |
+ I.reset(); |
+ |
+ MBSPatchHeader header; |
+ // The string will have a null terminator that we don't use, hence '-1'. |
+ COMPILE_ASSERT(sizeof(MBS_PATCH_HEADER_TAG) - 1 == sizeof(header.tag), |
+ MBS_PATCH_HEADER_TAG_must_match_header_field_size); |
+ memcpy(header.tag, MBS_PATCH_HEADER_TAG, sizeof(header.tag)); |
+ header.slen = oldsize; |
+ header.scrc32 = CalculateCrc(old, oldsize); |
+ header.dlen = newsize; |
+ header.cblen = control_stream.Length(); |
+ header.difflen = diff_bytes_length; |
+ header.extralen = extra_bytes_length; |
+ |
+ WriteHeader(patch_stream, &header); |
+ |
+ patch_stream->Append(&control_stream); |
+ patch_stream->Write(diff_bytes, diff_bytes_length); |
+ patch_stream->Write(extra_bytes, extra_bytes_length); |
+ |
+ LOG(INFO) << "Control tuples: " << control_length |
+ << " copy bytes: " << diff_bytes_length |
+ << " mistakes: " << diff_bytes_nonzero |
+ << " extra bytes: " << extra_bytes_length; |
+ |
+ LOG(INFO) << "Uncompressed bsdiff patch size " |
+ << patch_stream->Length() - initial_patch_stream_length; |
+ |
+ LOG(INFO) << "End bsdiff " |
+ << (base::Time::Now() - start_bsdiff_time).InSecondsF(); |
+ |
+ return OK; |
+} |
+ |
+} // namespace |