Как провалидировать что строка base64 является изображением?

Question

[AlexBoss]

Как проще всего провести валидацию строки base64 что она является картинкой(png или jpg).
Нашел вариант через проверку хедера. Хотелось бы что-то более изящное

Comments

[Алан Гибизов]

А как вы пробовали?

[AlexBoss]

Алан Гибизов, нашел вариант через проверку хедера. Хотелось бы что-то более изящное

[Модератор]

AlexBoss, для того, чтобы выполнить п.5.12 правил, вам необходимо дополнить вопрос своей попыткой решения, также привести, что не получается. Иначе вопрос может быть удалён.

[d'Ivan]

а нельзя ли заранее позаботиться о том что находится внутри Base64? Есть же всякие MIME types. По-моему, оно должно идти в паре.

Answer 1

[PavelMos]

https://docs.python.org/3/library/imghdr.html

работает с файлами и с объектами

но пишут что он тоже проверяет только заголовок файла . Но просмотрщики графики всё равно будут пытаться отобразить частично даже поврежденный файл, значит, само понятие, как валидный JPEG, не совсем корректно. Может быть, нужен полностью валидный, который откроется до конца ?

def what(file, h=None):
    f = None
    try:
        if h is None:
            if isinstance(file, basestring):
                f = open(file, 'rb')
                h = f.read(32)
            else:
                location = file.tell()
                h = file.read(32)
                file.seek(location)
        for tf in tests:
            res = tf(h, f)
            if res:
                return res
    finally:
        if f: f.close()
    return None

Answer 2

[Andrey]

png и jpg имеют фиксированный формат заголовков
тебе нужно просто найти этот формат и конвертнуть в base64
а потом просто найти вхождение в строке

например для png header такой
0x89 0x50 0x4e 0x47 0x0d 0x0a 0x1a 0x0a
base64 такой
iVBORw0KGg

Comments

[AlexBoss]

Тоже нашел этот вариант, но при этом нет гарантии что это является картинкой, т.к. остальные символы никак не проверяются

[Andrey]

AlexBoss, какие остальные ?
Начало base64 будет одинаковое

[AlexBoss]

seven5674, это понятно, что начало одинаковое и можно проверить по паттерну. Остальные символы в строке никак не проверяются

[Andrey]

AlexBoss, а зачем остальные проверять ?

[Василий Банников]

AlexBoss, ну ещё можешь попытаться прочитать файл дальше и проверить, что это не битое изображение ожидаемого размера.

Ещё можешь нейронку написать, которая проверит изображение на адекватность.
Но зачем всё это?