Index: third_party/courgette/bsdiff_create.cc |
=================================================================== |
--- third_party/courgette/bsdiff_create.cc (revision 15668) |
+++ third_party/courgette/bsdiff_create.cc (working copy) |
@@ -1,432 +0,0 @@ |
-/* |
- 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 |