[Ocfs2-tools-devel] [PATCH 10/10] Ocfs2-tools: Adjust corrupt handlers for truncate_log.c

[Tristan Ye]

Signed-off-by: Tristan Ye <tristan.ye at oracle.com>
---
 fswreck/include/truncate_log.h |    8 +++++-
 fswreck/truncate_log.c         |   41 ++++++++++++++++++++++++++++++---------
 2 files changed, 37 insertions(+), 12 deletions(-)

diff --git a/fswreck/include/truncate_log.h b/fswreck/include/truncate_log.h
index b5b11f6..e5311bf 100644
--- a/fswreck/include/truncate_log.h
+++ b/fswreck/include/truncate_log.h
@@ -25,7 +25,11 @@
 #ifndef __TRUNCATE_LOG_H
 #define __TRUNCATE_LOG_H
 
-void mess_up_truncate_log_list(ocfs2_filesys *fs, uint16_t slotnum);
-void mess_up_truncate_log_rec(ocfs2_filesys *fs, uint16_t slotnum);
+#include "fsck_type.h"
+
+void mess_up_truncate_log_list(ocfs2_filesys *fs, enum fsck_type type,
+			       uint16_t slotnum);
+void mess_up_truncate_log_rec(ocfs2_filesys *fs, enum fsck_type type,
+			      uint16_t slotnum);
 
 #endif		/* __TRUNCATE_LOG_H */
diff --git a/fswreck/truncate_log.c b/fswreck/truncate_log.c
index b757013..13594f5 100644
--- a/fswreck/truncate_log.c
+++ b/fswreck/truncate_log.c
@@ -222,33 +222,54 @@ static void get_truncate_log(ocfs2_filesys *fs,
 	return;
 }
 
-void mess_up_truncate_log_list(ocfs2_filesys *fs, uint16_t slotnum)
+void mess_up_truncate_log_list(ocfs2_filesys *fs, enum fsck_type type,
+			       uint16_t slotnum)
 {
 	uint64_t blkno;
 	int i;
- 	enum fsck_type types[] = { DEALLOC_COUNT, DEALLOC_USED };
 
 	get_truncate_log(fs, slotnum, &blkno);
 
-	for ( i = 0; i < ARRAY_ELEMENTS(types); i++) 
-		damage_truncate_log(fs, blkno, types[i], i);
+	switch (type) {
+	case DEALLOC_COUNT:
+		i = 0;
+		break;
+	case DEALLOC_USED:
+		i = 1;
+		break;
+	default:
+		break;
+	}
+
+	damage_truncate_log(fs, blkno, type, i);
 		
 	return;
 }
 
-void mess_up_truncate_log_rec(ocfs2_filesys *fs, uint16_t slotnum)
+void mess_up_truncate_log_rec(ocfs2_filesys *fs, enum fsck_type type,
+			      uint16_t slotnum)
 {
 	uint64_t blkno;
 	int i;
- 	enum fsck_type types[] = {	TRUNCATE_REC_START_RANGE,
-	 				TRUNCATE_REC_WRAP,
-					TRUNCATE_REC_RANGE };
+
+	switch (type) {
+	case TRUNCATE_REC_START_RANGE:
+		i = 0;
+		break;
+	case TRUNCATE_REC_WRAP:
+		i = 1;
+		break;
+	case TRUNCATE_REC_RANGE:
+		i = 2;
+		break;
+	default:
+		break;
+	}
 
 	get_truncate_log(fs, slotnum, &blkno);
 
 	create_truncate_log(fs, blkno, 10, 10);
-	for ( i = 0; i < ARRAY_ELEMENTS(types); i++) 
-		damage_truncate_log(fs, blkno, types[i], i);
+	damage_truncate_log(fs, blkno, type, i);
 		
 	return;
 }
-- 
1.5.5