From ebc247399f8b0993320d4d6174fb5325f38b2896 Mon Sep 17 00:00:00 2001
From: Shuduo Sang <sangshuduo@gmail.com>
Date: Sat, 19 Jun 2021 15:46:28 +0800
Subject: [PATCH] Hotfix/sangshuduo/td 4765 random fail func (#6542)

* fix missed function.

* add function declaration back for random fail test.
---
 src/os/inc/osFile.h        | 18 +++++++++++++++++-
 src/os/src/detail/osFail.c |  6 +++---
 2 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/src/os/inc/osFile.h b/src/os/inc/osFile.h
index 262f19ad22..86c08c0a3a 100644
--- a/src/os/inc/osFile.h
+++ b/src/os/inc/osFile.h
@@ -53,8 +53,24 @@ int64_t taosFSendFile(FILE *outfile, FILE *infile, int64_t *offset, int64_t size
 void taosGetTmpfilePath(const char *fileNamePrefix, char *dstPath);
 void taosClose(FileFd fd);
 
+#ifdef TAOS_RANDOM_FILE_FAIL
+  void taosSetRandomFileFailFactor(int32_t factor);
+  void taosSetRandomFileFailOutput(const char *path);
+  #ifdef TAOS_RANDOM_FILE_FAIL_TEST
+    int64_t taosReadFileRandomFail(int32_t fd, void *buf, int32_t count, const char *file, uint32_t line);
+    int64_t taosWriteFileRandomFail(int32_t fd, void *buf, int32_t count, const char *file, uint32_t line);
+    int64_t taosLSeekRandomFail(int32_t fd, int64_t offset, int32_t whence, const char *file, uint32_t line);
+    #undef taosRead
+    #undef taosWrite
+    #undef taosLSeek
+    #define taosRead(fd, buf, count) taosReadFileRandomFail(fd, buf, count, __FILE__, __LINE__)
+    #define taosWrite(fd, buf, count) taosWriteFileRandomFail(fd, buf, count, __FILE__, __LINE__)
+    #define taosLSeek(fd, offset, whence) taosLSeekRandomFail(fd, offset, whence, __FILE__, __LINE__)
+  #endif
+#endif
+
 #ifdef __cplusplus
 }
 #endif
 
-#endif
\ No newline at end of file
+#endif
diff --git a/src/os/src/detail/osFail.c b/src/os/src/detail/osFail.c
index a99bcd01db..6ddeefc521 100644
--- a/src/os/src/detail/osFail.c
+++ b/src/os/src/detail/osFail.c
@@ -113,7 +113,7 @@ int64_t taosReadFileRandomFail(int32_t fd, void *buf, int32_t count, const char
     }
   }
 
-  return taosReadImp(fd, buf, count);
+  return taosRead(fd, buf, count);
 }
 
 int64_t taosWriteFileRandomFail(int32_t fd, void *buf, int32_t count, const char *file, uint32_t line) {
@@ -124,7 +124,7 @@ int64_t taosWriteFileRandomFail(int32_t fd, void *buf, int32_t count, const char
     }
   }
 
-  return taosWriteImp(fd, buf, count);
+  return taosWrite(fd, buf, count);
 }
 
 int64_t taosLSeekRandomFail(int32_t fd, int64_t offset, int32_t whence, const char *file, uint32_t line) {
@@ -135,7 +135,7 @@ int64_t taosLSeekRandomFail(int32_t fd, int64_t offset, int32_t whence, const ch
     }
   }
 
-  return taosLSeekImp(fd, offset, whence);
+  return taosLSeek(fd, offset, whence);
 }
 
 #endif //TAOS_RANDOM_FILE_FAIL
-- 
GitLab