2 files changed, 97 insertions, 0 deletions
diff --git a/src/common/DataStores/DB2FileSystemSource.cpp b/src/common/DataStores/DB2FileSystemSource.cpp
new file mode 100644
index 00000000000..930100e8d38
--- /dev/null
+++ b/src/common/DataStores/DB2FileSystemSource.cpp
@@ -0,0 +1,58 @@
+/*
+ * Copyright (C) 2008-2018 TrinityCore <https://www.trinitycore.org/>
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the
+ * Free Software Foundation; either version 2 of the License, or (at your
+ * option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "DB2FileSystemSource.h"
+#include <boost/filesystem/operations.hpp>
+
+DB2FileSystemSource::DB2FileSystemSource(std::string const& fileName)
+{
+    _fileName = fileName;
+    _file = fopen(_fileName.c_str(), "rb");
+}
+
+DB2FileSystemSource::~DB2FileSystemSource()
+{
+    if (_file)
+        fclose(_file);
+}
+
+bool DB2FileSystemSource::IsOpen() const
+{
+    return _file != nullptr;
+}
+
+bool DB2FileSystemSource::Read(void* buffer, std::size_t numBytes)
+{
+    return fread(buffer, numBytes, 1, _file) == 1;
+}
+
+std::size_t DB2FileSystemSource::GetPosition() const
+{
+    return ftell(_file);
+}
+
+std::size_t DB2FileSystemSource::GetFileSize() const
+{
+    boost::system::error_code error;
+    std::size_t size = boost::filesystem::file_size(_fileName, error);
+    return !error ? size : 0;
+}
+
+char const* DB2FileSystemSource::GetFileName() const
+{
+    return _fileName.c_str();
+}
diff --git a/src/common/DataStores/DB2FileSystemSource.h b/src/common/DataStores/DB2FileSystemSource.h
new file mode 100644
index 00000000000..eb838bbb846
--- /dev/null
+++ b/src/common/DataStores/DB2FileSystemSource.h
@@ -0,0 +1,39 @@
+/*
+ * Copyright (C) 2008-2018 TrinityCore <https://www.trinitycore.org/>
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the
+ * Free Software Foundation; either version 2 of the License, or (at your
+ * option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef DB2FileSystemSource_h__
+#define DB2FileSystemSource_h__
+
+#include "DB2FileLoader.h"
+#include <string>
+
+struct TC_COMMON_API DB2FileSystemSource : public DB2FileSource
+{
+    DB2FileSystemSource(std::string const& fileName);
+    ~DB2FileSystemSource();
+    bool IsOpen() const override;
+    bool Read(void* buffer, std::size_t numBytes) override;
+    std::size_t GetPosition() const override;
+    std::size_t GetFileSize() const override;
+    char const* GetFileName() const override;
+
+private:
+    std::string _fileName;
+    FILE* _file;
+};
+
+#endif // DB2FileSystemSource_h__