Add a "do not deduplicate" block flag

Signed-off-by: David Oberhollenzer <david.oberhollenzer@sigma-star.at>

2 files changed, 14 insertions, 10 deletions

diff --git a/lib/sqfs/block_processor/common.c b/lib/sqfs/block_processor/common.c
index ab83b5d..2c6e59e 100644
--- a/lib/sqfs/block_processor/common.c
+++ b/lib/sqfs/block_processor/common.c
@@ -142,11 +142,15 @@ int process_completed_fragment(sqfs_block_processor_t *proc, sqfs_block_t *frag,
 
 	proc->stats.total_frag_count += 1;
 
-	err = sqfs_frag_table_find_tail_end(proc->frag_tbl, frag->checksum,
-					    frag->size, &index, &offset);
-	if (err == 0) {
-		sqfs_inode_set_frag_location(*(frag->inode), index, offset);
-		return 0;
+	if (!(frag->flags & SQFS_BLK_DONT_DEDUPLICATE)) {
+		err = sqfs_frag_table_find_tail_end(proc->frag_tbl,
+						    frag->checksum, frag->size,
+						    &index, &offset);
+		if (err == 0) {
+			sqfs_inode_set_frag_location(*(frag->inode),
+						     index, offset);
+			return 0;
+		}
 	}
 
 	if (proc->frag_block != NULL) {
diff --git a/lib/sqfs/block_writer.c b/lib/sqfs/block_writer.c
index 73dcb14..e252c89 100644
--- a/lib/sqfs/block_writer.c
+++ b/lib/sqfs/block_writer.c
@@ -229,7 +229,7 @@ int sqfs_block_writer_write(sqfs_block_writer_t *wr, sqfs_u32 size,
 
 		if (count == 0) {
 			*location = 0;
-		} else {
+		} else if (!(flags & SQFS_BLK_DONT_DEDUPLICATE)) {
 			start = deduplicate_blocks(wr, count);
 			offset = wr->blocks[start].offset;
 
@@ -244,11 +244,11 @@ int sqfs_block_writer_write(sqfs_block_writer_t *wr, sqfs_u32 size,
 			} else {
 				wr->num_blocks = wr->file_start;
 			}
-		}
 
-		err = wr->file->truncate(wr->file, wr->start);
-		if (err)
-			return err;
+			err = wr->file->truncate(wr->file, wr->start);
+			if (err)
+				return err;
+		}
 
 		wr->stats.blocks_written = wr->num_blocks;
 		wr->stats.bytes_written = wr->start - wr->data_area_start;