nw4r ut almost matching

1 files changed, 178 insertions, 0 deletions

diff --git a/src/nw4r/ut/ut_NandFileStream.cpp b/src/nw4r/ut/ut_NandFileStream.cpp
index e69de29b..29c7a0b8 100644
--- a/src/nw4r/ut/ut_NandFileStream.cpp
+++ b/src/nw4r/ut/ut_NandFileStream.cpp
@@ -0,0 +1,178 @@
+
+#include <nw4r/ut.h>
+
+namespace nw4r {
+namespace ut {
+
+NW4R_UT_RTTI_DEF_DERIVED(NandFileStream, FileStream);
+
+void NandFileStream::NandAsyncCallback_(long result, NANDCommandBlock *block) {
+    NandFileStream *self = reinterpret_cast<AsyncContext *>(block)->stream;
+
+    self->mIsBusy = false;
+    self->mResult = result;
+
+    if (self->mCallback != NULL) {
+        self->mCallback(result, self, self->mCallbackArg);
+    }
+}
+
+void NandFileStream::Initialize_() {
+    mCanRead = false;
+    mCanWrite = false;
+    mCloseOnDestroy = false;
+    mAllowClose = false;
+    mIsOpen = false;
+    mIsBusy = false;
+    mCallback = NULL;
+    mCallbackArg = NULL;
+    mResult = NAND_RESULT_OK;
+    mAsyncContext.stream = this;
+}
+
+NandFileStream::NandFileStream(const char *path, unsigned long access) {
+    Initialize_();
+    Open(path, access);
+}
+
+NandFileStream::NandFileStream(const NANDFileInfo *info, unsigned long access, bool close) {
+    Initialize_();
+    Open(info, access, close);
+}
+
+NandFileStream::~NandFileStream() {
+    if (mCloseOnDestroy) {
+        Close();
+    }
+}
+
+bool NandFileStream::Open(const char *path, unsigned long access) {
+    if (mCloseOnDestroy) {
+        Close();
+    }
+
+    mCanRead = access & NAND_ACCESS_READ;
+    mCanWrite = access & NAND_ACCESS_WRITE;
+
+    if (NANDOpen(path, &mAsyncContext.info, access) != NAND_RESULT_OK) {
+        return false;
+    }
+
+    if (mCanRead) {
+        unsigned long fileSize;
+
+        if (NANDGetLength(&mAsyncContext.info, &fileSize) != NAND_RESULT_OK) {
+            NANDClose(&mAsyncContext.info);
+            return false;
+        }
+
+        mFilePosition.SetFileSize(fileSize);
+    }
+
+    mFilePosition.Seek(0, SEEK_BEG);
+
+    mCloseOnDestroy = true;
+    mAllowClose = true;
+    mIsOpen = true;
+
+    return true;
+}
+
+bool NandFileStream::Open(const NANDFileInfo *info, unsigned long access, bool close) {
+    if (mCloseOnDestroy) {
+        Close();
+    }
+
+    mCanRead = access & NAND_ACCESS_READ;
+    mCanWrite = access & NAND_ACCESS_WRITE;
+
+    mAsyncContext.info = *info;
+
+    unsigned long fileSize;
+    if (mCanRead) {
+        if (NANDGetLength(&mAsyncContext.info, &fileSize) != NAND_RESULT_OK) {
+            if (close) {
+                NANDClose(&mAsyncContext.info);
+            }
+            return false;
+        }
+        mFilePosition.SetFileSize(fileSize);
+    }
+    mFilePosition.Seek(0, SEEK_BEG);
+
+    mCloseOnDestroy = false;
+    mAllowClose = close;
+    mIsOpen = true;
+    return true;
+}
+
+void NandFileStream::Close() {
+    if (mAllowClose && mIsOpen) {
+        NANDClose(&mAsyncContext.info);
+        mIsOpen = false;
+    }
+}
+
+long NandFileStream::Read(void *dst, unsigned long size) {
+    NANDSeek(&mAsyncContext.info, mFilePosition.Tell(), NAND_SEEK_BEG);
+
+    long result = NANDRead(&mAsyncContext.info, dst, size);
+    if (result > 0) {
+        mFilePosition.Skip(result);
+    }
+
+    return result;
+}
+
+bool NandFileStream::ReadAsync(void *dst, unsigned long size, AsyncCallback callback, void *arg) {
+    mCallback = callback;
+    mCallbackArg = arg;
+    mIsBusy = true;
+
+    NANDSeek(&mAsyncContext.info, mFilePosition.Tell(), NAND_SEEK_BEG);
+
+    bool success =
+            NANDReadAsync(&mAsyncContext.info, dst, size, NandAsyncCallback_, &mAsyncContext.block) == NAND_RESULT_OK;
+
+    if (success) {
+        mFilePosition.Skip(size);
+    } else {
+        mIsBusy = false;
+    }
+
+    return success;
+}
+
+long NandFileStream::Write(const void *src, unsigned long size) {
+    NANDSeek(&mAsyncContext.info, mFilePosition.Tell(), NAND_SEEK_BEG);
+    long result = NANDWrite(&mAsyncContext.info, src, size);
+    if (result > 0) {
+        mFilePosition.Append(result);
+    }
+    return result;
+}
+
+bool NandFileStream::WriteAsync(const void *src, unsigned long size, AsyncCallback callback, void *arg) {
+    mCallback = callback;
+    mCallbackArg = arg;
+    mIsBusy = true;
+
+    NANDSeek(&mAsyncContext.info, mFilePosition.Tell(), NAND_SEEK_BEG);
+
+    long result = NANDWriteAsync(&mAsyncContext.info, src, size, NandAsyncCallback_, &mAsyncContext.block);
+
+    if (result == NAND_RESULT_OK) {
+        mFilePosition.Append(size);
+    } else {
+        mIsBusy = false;
+    }
+
+    return result == NAND_RESULT_OK;
+}
+
+void NandFileStream::Seek(long offset, unsigned long origin) {
+    mFilePosition.Seek(offset, origin);
+}
+
+} // namespace ut
+} // namespace nw4r