mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2025-05-06 11:12:54 +00:00
f7e152d049
At the moment, this processes the RIFF chunk structure and extracts the ICCP chunk, so that `icc` can now print ICC profiles embedded in webp files. (And are image files really more than containers of icc profiles?) It doesn't even decode image dimensions yet. The lossy format is a VP8 video frame. Once we get to that, we might want to move all the image decoders into a new LibImageDecoders that depends on both LibGfx and LibVideo. (Other newer image formats like heic and av1f also use video frames for image data.)
184 lines
12 KiB
C++
184 lines
12 KiB
C++
/*
|
||
* Copyright (c) 2020, Andreas Kling <kling@serenityos.org>
|
||
*
|
||
* SPDX-License-Identifier: BSD-2-Clause
|
||
*/
|
||
|
||
#include <AK/LexicalPath.h>
|
||
#include <AK/StringBuilder.h>
|
||
#include <LibCore/MimeData.h>
|
||
|
||
namespace Core {
|
||
|
||
Vector<DeprecatedString> MimeData::formats() const
|
||
{
|
||
Vector<DeprecatedString> mime_types;
|
||
mime_types.ensure_capacity(m_data.size());
|
||
for (auto& it : m_data)
|
||
mime_types.unchecked_append(it.key);
|
||
return mime_types;
|
||
}
|
||
|
||
Vector<URL> MimeData::urls() const
|
||
{
|
||
auto it = m_data.find("text/uri-list");
|
||
if (it == m_data.end())
|
||
return {};
|
||
Vector<URL> urls;
|
||
for (auto& line : StringView(it->value).split_view('\n')) {
|
||
urls.append(URL(line));
|
||
}
|
||
return urls;
|
||
}
|
||
|
||
void MimeData::set_urls(Vector<URL> const& urls)
|
||
{
|
||
StringBuilder builder;
|
||
for (auto& url : urls) {
|
||
builder.append(url.to_deprecated_string());
|
||
builder.append('\n');
|
||
}
|
||
set_data("text/uri-list", builder.to_byte_buffer());
|
||
}
|
||
|
||
DeprecatedString MimeData::text() const
|
||
{
|
||
return DeprecatedString::copy(m_data.get("text/plain").value_or({}));
|
||
}
|
||
|
||
void MimeData::set_text(DeprecatedString const& text)
|
||
{
|
||
set_data("text/plain", text.to_byte_buffer());
|
||
}
|
||
|
||
StringView guess_mime_type_based_on_filename(StringView path)
|
||
{
|
||
if (path.ends_with(".pbm"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/x‑portable‑bitmap"sv;
|
||
if (path.ends_with(".pgm"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/x‑portable‑graymap"sv;
|
||
if (path.ends_with(".png"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/png"sv;
|
||
if (path.ends_with(".ppm"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/x‑portable‑pixmap"sv;
|
||
if (path.ends_with(".gif"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/gif"sv;
|
||
if (path.ends_with(".bmp"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/bmp"sv;
|
||
if (path.ends_with(".jpg"sv, CaseSensitivity::CaseInsensitive) || path.ends_with(".jpeg"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/jpeg"sv;
|
||
if (path.ends_with(".qoi"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/x-qoi"sv;
|
||
if (path.ends_with(".svg"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/svg+xml"sv;
|
||
if (path.ends_with(".tga"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/x-targa"sv;
|
||
if (path.ends_with(".webp"sv, CaseSensitivity::CaseInsensitive))
|
||
return "image/webp"sv;
|
||
if (path.ends_with(".md"sv, CaseSensitivity::CaseInsensitive))
|
||
return "text/markdown"sv;
|
||
if (path.ends_with(".html"sv, CaseSensitivity::CaseInsensitive) || path.ends_with(".htm"sv, CaseSensitivity::CaseInsensitive))
|
||
return "text/html"sv;
|
||
if (path.ends_with(".css"sv, CaseSensitivity::CaseInsensitive))
|
||
return "text/css"sv;
|
||
if (path.ends_with(".icc"sv, CaseSensitivity::CaseInsensitive) || path.ends_with(".icm"sv, CaseSensitivity::CaseInsensitive))
|
||
return "application/vnd.iccprofile"sv;
|
||
if (path.ends_with(".js"sv, CaseSensitivity::CaseInsensitive))
|
||
return "application/javascript"sv;
|
||
if (path.ends_with(".json"sv, CaseSensitivity::CaseInsensitive))
|
||
return "application/json"sv;
|
||
if (path.ends_with(".zip"sv, CaseSensitivity::CaseInsensitive))
|
||
return "application/zip"sv;
|
||
if (path.ends_with(".md"sv, CaseSensitivity::CaseInsensitive))
|
||
return "text/markdown"sv;
|
||
if (path.ends_with("/"sv, CaseSensitivity::CaseInsensitive))
|
||
return "text/html"sv;
|
||
if (path.ends_with(".csv"sv, CaseSensitivity::CaseInsensitive))
|
||
return "text/csv"sv;
|
||
if (path.ends_with(".sheets"sv, CaseSensitivity::CaseInsensitive))
|
||
return "application/x-sheets+json"sv;
|
||
// FIXME: Share this, TextEditor and HackStudio language detection somehow.
|
||
auto basename = LexicalPath::basename(path);
|
||
if (path.ends_with(".cpp"sv, CaseSensitivity::CaseInsensitive)
|
||
|| path.ends_with(".c"sv, CaseSensitivity::CaseInsensitive)
|
||
|| path.ends_with(".hpp"sv, CaseSensitivity::CaseInsensitive)
|
||
|| path.ends_with(".h"sv, CaseSensitivity::CaseInsensitive)
|
||
|| path.ends_with(".gml"sv, CaseSensitivity::CaseInsensitive)
|
||
|| path.ends_with(".ini"sv, CaseSensitivity::CaseInsensitive)
|
||
|| path.ends_with(".ipc"sv, CaseSensitivity::CaseInsensitive)
|
||
|| path.ends_with(".txt"sv, CaseSensitivity::CaseInsensitive)
|
||
|| basename == "CMakeLists.txt"
|
||
|| basename == ".history"
|
||
|| basename == ".shellrc")
|
||
return "text/plain"sv;
|
||
return "application/octet-stream"sv;
|
||
}
|
||
|
||
#define ENUMERATE_HEADER_CONTENTS \
|
||
__ENUMERATE_MIME_TYPE_HEADER(blend, "extra/blender", 0, 7, 'B', 'L', 'E', 'N', 'D', 'E', 'R') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(bmp, "image/bmp", 0, 2, 'B', 'M') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(bzip2, "application/x-bzip2", 0, 3, 'B', 'Z', 'h') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(compressed_iso, "extra/isz", 0, 4, 'I', 's', 'Z', '!') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(elf, "extra/elf", 0, 4, 0x7F, 'E', 'L', 'F') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(ext, "extra/ext", 0x438, 2, 0x53, 0xEF) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(flac, "audio/flac", 0, 4, 'f', 'L', 'a', 'C') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(gif_87, "image/gif", 0, 6, 'G', 'I', 'F', '8', '7', 'a') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(gif_89, "image/gif", 0, 6, 'G', 'I', 'F', '8', '9', 'a') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(gzip, "application/gzip", 0, 2, 0x1F, 0x8B) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(icc, "application/vnd.iccprofile", 36, 4, 'a', 'c', 's', 'p') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(iso9660_0, "extra/iso-9660", 0x8001, 5, 0x43, 0x44, 0x30, 0x30, 0x31) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(iso9660_1, "extra/iso-9660", 0x8801, 5, 0x43, 0x44, 0x30, 0x30, 0x31) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(iso9660_2, "extra/iso-9660", 0x9001, 5, 0x43, 0x44, 0x30, 0x30, 0x31) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(jpeg, "image/jpeg", 0, 4, 0xFF, 0xD8, 0xFF, 0xDB) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(jpeg_huh, "image/jpeg", 0, 4, 0xFF, 0xD8, 0xFF, 0xEE) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(jpeg_jfif, "image/jpeg", 0, 12, 0xFF, 0xD8, 0xFF, 0xE0, 0x00, 0x10, 'J', 'F', 'I', 'F', 0x00, 0x01) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(lua_bytecode, "extra/lua-bytecode", 0, 4, 0x1B, 'L', 'u', 'a') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(midi, "audio/midi", 0, 4, 0x4D, 0x54, 0x68, 0x64) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(mkv, "extra/matroska", 0, 4, 0x1A, 0x45, 0xDF, 0xA3) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(mp3, "audio/mpeg", 0, 2, 0xFF, 0xFB) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(nesrom, "extra/nes-rom", 0, 4, 'N', 'E', 'S', 0x1A) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(pbm, "image/x-portable-bitmap", 0, 3, 0x50, 0x31, 0x0A) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(pdf, "application/pdf", 0, 5, 0x25, 'P', 'D', 'F', 0x2D) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(pgm, "image/x-portable-graymap", 0, 3, 0x50, 0x32, 0x0A) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(png, "image/png", 0, 8, 0x89, 'P', 'N', 'G', 0x0D, 0x0A, 0x1A, 0x0A) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(ppm, "image/x-portable-pixmap", 0, 3, 0x50, 0x33, 0x0A) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(qcow, "extra/qcow", 0, 3, 'Q', 'F', 'I') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(qoi, "image/x-qoi", 0, 4, 'q', 'o', 'i', 'f') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(rtf, "application/rtf", 0, 6, 0x7B, 0x5C, 0x72, 0x74, 0x66, 0x31) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(sevenzip, "application/x-7z-compressed", 0, 6, 0x37, 0x7A, 0xBC, 0xAF, 0x27, 0x1C) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(shell, "text/x-shellscript", 0, 10, '#', '!', '/', 'b', 'i', 'n', '/', 's', 'h', '\n') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(sqlite, "extra/sqlite", 0, 16, 'S', 'Q', 'L', 'i', 't', 'e', ' ', 'f', 'o', 'r', 'm', 'a', 't', ' ', '3', 0x00) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(tar, "application/tar", 0x101, 5, 0x75, 0x73, 0x74, 0x61, 0x72) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(zip, "application/zip", 0, 2, 0x50, 0x4B) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(tiff, "image/tiff", 0, 4, 'I', 'I', '*', 0x00) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(tiff_bigendian, "image/tiff", 0, 4, 'M', 'M', 0x00, '*') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(wasm, "application/wasm", 0, 4, 0x00, 'a', 's', 'm') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(wav, "audio/wave", 8, 4, 'W', 'A', 'V', 'E') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(webp, "image/webp", 8, 4, 'W', 'E', 'B', 'P') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(win_31x_archive, "extra/win-31x-compressed", 0, 4, 'K', 'W', 'A', 'J') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(win_95_archive, "extra/win-95-compressed", 0, 4, 'S', 'Z', 'D', 'D') \
|
||
__ENUMERATE_MIME_TYPE_HEADER(zlib_0, "extra/raw-zlib", 0, 2, 0x78, 0x01) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(zlib_1, "extra/raw-zlib", 0, 2, 0x78, 0x5E) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(zlib_2, "extra/raw-zlib", 0, 2, 0x78, 0x9C) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(zlib_3, "extra/raw-zlib", 0, 2, 0x78, 0xDA) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(zlib_4, "extra/raw-zlib", 0, 2, 0x78, 0x20) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(zlib_5, "extra/raw-zlib", 0, 2, 0x78, 0x7D) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(zlib_6, "extra/raw-zlib", 0, 2, 0x78, 0xBB) \
|
||
__ENUMERATE_MIME_TYPE_HEADER(zlib_7, "extra/raw-zlib", 0, 2, 0x78, 0xF9)
|
||
|
||
#define __ENUMERATE_MIME_TYPE_HEADER(var_name, mime_type, pattern_offset, pattern_size, ...) \
|
||
static const u8 var_name##_arr[pattern_size] = { __VA_ARGS__ }; \
|
||
static constexpr ReadonlyBytes var_name = ReadonlyBytes { var_name##_arr, pattern_size };
|
||
ENUMERATE_HEADER_CONTENTS
|
||
#undef __ENUMERATE_MIME_TYPE_HEADER
|
||
|
||
Optional<DeprecatedString> guess_mime_type_based_on_sniffed_bytes(ReadonlyBytes bytes)
|
||
{
|
||
#define __ENUMERATE_MIME_TYPE_HEADER(var_name, mime_type, pattern_offset, pattern_size, ...) \
|
||
if (static_cast<ssize_t>(bytes.size()) >= pattern_offset && bytes.slice(pattern_offset).starts_with(var_name)) \
|
||
return mime_type;
|
||
ENUMERATE_HEADER_CONTENTS;
|
||
#undef __ENUMERATE_MIME_TYPE_HEADER
|
||
return {};
|
||
}
|
||
}
|