Cool_Tools/ImHex
1
0
Fork 0
mirror of synced 2024-11-12 02:00:52 +01:00
Code Releases Wiki Activity

Improved libmagic usage. Now supports additional databases

Browse Source
This commit is contained in:
WerWolv 2020-11-12 21:21:11 +01:00
parent d88bb877b6
commit a166bf4ed8
1 changed files with 45 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
source/views/view_information.cpp
View File

@ -4,13 +4,20 @@
#include "utils.hpp" #include "utils.hpp"
#include <algorithm>
#include <vector>
#include <cstring> #include <cstring>
#include <cmath> #include <cmath>
#include <filesystem>
#include <span>
#include <vector>
#include <magic.h> #include <magic.h>
#include <span>
#if defined(__EMX__) || defined (WIN32)
#define MAGIC_PATH_SEPARATOR ";"
#else
#define MAGIC_PATH_SEPARATOR ":"
#endif
namespace hex { namespace hex {
@ -71,20 +78,43 @@ namespace hex {
std::vector<u8> buffer(this->m_dataProvider->getSize(), 0x00); std::vector<u8> buffer(this->m_dataProvider->getSize(), 0x00);
this->m_dataProvider->read(0x00, buffer.data(), buffer.size()); this->m_dataProvider->read(0x00, buffer.data(), buffer.size());
this->m_fileDescription.clear();
this->m_mimeType.clear();
std::string magicFiles;
for (const auto &entry : std::filesystem::directory_iterator("magic")) {
if (entry.is_regular_file() && entry.path().extension() == ".mgc")
magicFiles += entry.path().string() + MAGIC_PATH_SEPARATOR;
}
magicFiles.pop_back();
{ {
magic_t cookie = magic_open(MAGIC_NONE); magic_t cookie = magic_open(MAGIC_NONE);
magic_load(cookie, "magic"); if (magic_load(cookie, magicFiles.c_str()) == -1)
goto skip_description;
this->m_fileDescription = magic_buffer(cookie, buffer.data(), buffer.size()); this->m_fileDescription = magic_buffer(cookie, buffer.data(), buffer.size());
skip_description:
magic_close(cookie); magic_close(cookie);
} }
{ {
magic_t cookie = magic_open(MAGIC_MIME); magic_t cookie = magic_open(MAGIC_MIME);
magic_load(cookie, "magic"); if (magic_load(cookie, magicFiles.c_str()) == -1)
goto skip_mime;
this->m_mimeType = magic_buffer(cookie, buffer.data(), buffer.size()); this->m_mimeType = magic_buffer(cookie, buffer.data(), buffer.size());
skip_mime:
magic_close(cookie); magic_close(cookie);
} }
this->m_shouldInvalidate = false; this->m_shouldInvalidate = false;
} }
} }
@ -115,14 +145,18 @@ namespace hex {
ImGui::Separator(); ImGui::Separator();
ImGui::NewLine(); ImGui::NewLine();
if (!this->m_fileDescription.empty()) {
ImGui::TextUnformatted("Description:"); ImGui::TextUnformatted("Description:");
ImGui::TextWrapped("%s", this->m_fileDescription.c_str()); ImGui::TextWrapped("%s", this->m_fileDescription.c_str());
ImGui::NewLine(); ImGui::NewLine();
}
if (!this->m_mimeType.empty()) {
ImGui::TextUnformatted("MIME Type:"); ImGui::TextUnformatted("MIME Type:");
ImGui::TextWrapped("%s", this->m_mimeType.c_str()); ImGui::TextWrapped("%s", this->m_mimeType.c_str());
ImGui::NewLine(); ImGui::NewLine();
} }
}
ImGui::EndChild(); ImGui::EndChild();
} }