From 6a11a69b15b47e0eb0d558acdf192f69128f8ab0 Mon Sep 17 00:00:00 2001
From: Yann Collet <yann.collet.73@gmail.com>
Date: Fri, 20 Nov 2015 12:00:25 +0100
Subject: [PATCH] added : repcode generation to datagen

---
 programs/datagen.c | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/programs/datagen.c b/programs/datagen.c
index 23d7d15f..7fb98e83 100644
--- a/programs/datagen.c
+++ b/programs/datagen.c
@@ -138,6 +138,7 @@ void RDG_genBlock(void* buffer, size_t buffSize, size_t prefixSize, double match
     const U32 matchProba32 = (U32)(32768 * matchProba);
     size_t pos = prefixSize;
     U32* seed = seedPtr;
+    U32 prevOffset = 1;
 
     /* special case : sparse content */
     while (matchProba >= 1.0)
@@ -167,14 +168,16 @@ void RDG_genBlock(void* buffer, size_t buffSize, size_t prefixSize, double match
         {
             /* Copy (within 32K) */
             size_t match;
-            size_t d;
-            int length = RDG_RANDLENGTH + 4;
+            size_t length = RDG_RANDLENGTH + 4;
             U32 offset = RDG_RAND15BITS + 1;
+            U32 repeatOffset = (RDG_rand(seed) & 15) == 2;
+            if (repeatOffset) offset = prevOffset;
             if (offset > pos) offset = (U32)pos;
             match = pos - offset;
-            d = pos + length;
-            if (d > buffSize) d = buffSize;
-            while (pos < d) buffPtr[pos++] = buffPtr[match++];
+            if (length > buffSize-pos) length = buffSize-pos;
+            memcpy(buffPtr+pos, buffPtr+match, length);
+            pos += length;
+            prevOffset = offset;
         }
         else
         {