1 files changed, 128 insertions, 0 deletions
diff --git a/utils/zenutils/source/shared/pe.cpp b/utils/zenutils/source/shared/pe.cpp
new file mode 100755
index 0000000..c86ec6c
--- /dev/null
+++ b/utils/zenutils/source/shared/pe.cpp
@@ -0,0 +1,128 @@
+/* zenutils - Utilities for working with creative firmwares.
+ * Copyright 2007 (c) Rasmus Ry <rasmus.ry{at}gmail.com>
+ *
+ * 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, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#include "pe.h"
+
+
+shared::pe_file::pe_file(PeLib::PeFile* pef) : _pef(pef)
+{
+}
+shared::pe_file::~pe_file()
+{
+    if (_pef != NULL)
+        delete _pef;
+}
+
+bool shared::pe_file::is_valid() const
+{
+    if (_pef->getBits() == 32)
+    {
+        PeLib::PeHeader32& pef32 = static_cast<PeLib::PeFile32*>(_pef)->peHeader();
+        if (!pef32.isValid())
+            return false;
+        return true;
+    }
+    else if (_pef->getBits() == 64)
+    {
+        PeLib::PeHeader64& pef64 = static_cast<PeLib::PeFile64*>(_pef)->peHeader();
+        if (!pef64.isValid())
+            return false;
+        return true;
+    }
+    return false;
+}
+
+bool shared::pe_file::read(const std::string& filename)
+{
+    if (_pef != NULL)
+    {
+        delete _pef;
+        _pef = NULL;
+    }
+
+    _pef = PeLib::openPeFile(filename);
+    if (!_pef)
+    {
+        return false;
+    }
+    if (_pef->readMzHeader())
+    {
+        delete _pef;
+        return false;
+    }
+    if (!_pef->mzHeader().isValid())
+    {
+        delete _pef;
+        return false;
+    }
+    if (_pef->readPeHeader())
+    {
+        delete _pef;
+        return false;
+    }
+    if (!is_valid())
+    {
+        delete _pef;
+        return false;
+    }
+    return true;
+}
+
+bool shared::pe_file::find_section(const std::string& name, section_info& info) const
+{
+    if (_pef->getBits() == 32)
+        return find_section(static_cast<PeLib::PeFile32*>(_pef),
+                            name, info);
+    else if (_pef->getBits() == 64)
+        return find_section(static_cast<PeLib::PeFile64*>(_pef),
+                            name, info);
+    return false;
+}
+
+bool shared::pe_file::add_section(const std::string& name,
+                                  const bytes& buffer, section_info& info)
+{
+    if (_pef->getBits() == 32)
+    {
+         return add_section(static_cast<PeLib::PeFile32*>(_pef),
+                            name, buffer, info);
+    }
+    else if (_pef->getBits() == 64)
+    {
+         return add_section(static_cast<PeLib::PeFile64*>(_pef),
+                            name, buffer, info);
+    }
+    return false;
+}
+
+dword shared::pe_file::get_image_base() const
+{
+    if (_pef->getBits() == 32)
+        return static_cast<PeLib::PeFile32*>(_pef)->peHeader().getImageBase();
+    else
+        return static_cast<PeLib::PeFile64*>(_pef)->peHeader().getImageBase();
+    return 0;
+}
+dword shared::pe_file::pa_to_va(dword pa) const
+{
+    if (_pef->getBits() == 32)
+        return static_cast<PeLib::PeFile32*>(_pef)->peHeader().offsetToVa(pa);
+    else
+        return static_cast<PeLib::PeFile64*>(_pef)->peHeader().offsetToVa(pa);
+    return 0;
+}