123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136 |
- // Copyright 2021 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package typesniffer
-
- import (
- "bytes"
- "encoding/base64"
- "encoding/hex"
- "strings"
- "testing"
-
- "github.com/stretchr/testify/assert"
- )
-
- func TestDetectContentTypeLongerThanSniffLen(t *testing.T) {
- // Pre-condition: Shorter than sniffLen detects SVG.
- assert.Equal(t, "image/svg+xml", DetectContentType([]byte(`<!-- Comment --><svg></svg>`)).contentType)
- // Longer than sniffLen detects something else.
- assert.NotEqual(t, "image/svg+xml", DetectContentType([]byte(`<!-- `+strings.Repeat("x", sniffLen)+` --><svg></svg>`)).contentType)
- }
-
- func TestIsTextFile(t *testing.T) {
- assert.True(t, DetectContentType([]byte{}).IsText())
- assert.True(t, DetectContentType([]byte("lorem ipsum")).IsText())
- }
-
- func TestIsSvgImage(t *testing.T) {
- assert.True(t, DetectContentType([]byte("<svg></svg>")).IsSvgImage())
- assert.True(t, DetectContentType([]byte(" <svg></svg>")).IsSvgImage())
- assert.True(t, DetectContentType([]byte(`<svg width="100"></svg>`)).IsSvgImage())
- assert.True(t, DetectContentType([]byte(`<?xml version="1.0" encoding="UTF-8"?><svg></svg>`)).IsSvgImage())
- assert.True(t, DetectContentType([]byte(`<!-- Comment -->
- <svg></svg>`)).IsSvgImage())
- assert.True(t, DetectContentType([]byte(`<!-- Multiple -->
- <!-- Comments -->
- <svg></svg>`)).IsSvgImage())
- assert.True(t, DetectContentType([]byte(`<!-- Multiline
- Comment -->
- <svg></svg>`)).IsSvgImage())
- assert.True(t, DetectContentType([]byte(`<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1 Basic//EN"
- "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11-basic.dtd">
- <svg></svg>`)).IsSvgImage())
- assert.True(t, DetectContentType([]byte(`<?xml version="1.0" encoding="UTF-8"?>
- <!-- Comment -->
- <svg></svg>`)).IsSvgImage())
- assert.True(t, DetectContentType([]byte(`<?xml version="1.0" encoding="UTF-8"?>
- <!-- Multiple -->
- <!-- Comments -->
- <svg></svg>`)).IsSvgImage())
- assert.True(t, DetectContentType([]byte(`<?xml version="1.0" encoding="UTF-8"?>
- <!-- Multline
- Comment -->
- <svg></svg>`)).IsSvgImage())
- assert.True(t, DetectContentType([]byte(`<?xml version="1.0" encoding="UTF-8"?>
- <!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
- <!-- Multline
- Comment -->
- <svg></svg>`)).IsSvgImage())
-
- // the DetectContentType should work for incomplete data, because only beginning bytes are used for detection
- assert.True(t, DetectContentType([]byte(`<svg>....`)).IsSvgImage())
-
- assert.False(t, DetectContentType([]byte{}).IsSvgImage())
- assert.False(t, DetectContentType([]byte("svg")).IsSvgImage())
- assert.False(t, DetectContentType([]byte("<svgfoo></svgfoo>")).IsSvgImage())
- assert.False(t, DetectContentType([]byte("text<svg></svg>")).IsSvgImage())
- assert.False(t, DetectContentType([]byte("<html><body><svg></svg></body></html>")).IsSvgImage())
- assert.False(t, DetectContentType([]byte(`<script>"<svg></svg>"</script>`)).IsSvgImage())
- assert.False(t, DetectContentType([]byte(`<!-- <svg></svg> inside comment -->
- <foo></foo>`)).IsSvgImage())
- assert.False(t, DetectContentType([]byte(`<?xml version="1.0" encoding="UTF-8"?>
- <!-- <svg></svg> inside comment -->
- <foo></foo>`)).IsSvgImage())
-
- assert.False(t, DetectContentType([]byte(`
- <!-- comment1 -->
- <div>
- <!-- comment2 -->
- <svg></svg>
- </div>
- `)).IsSvgImage())
-
- assert.False(t, DetectContentType([]byte(`
- <!-- comment1
- -->
- <div>
- <!-- comment2
- -->
- <svg></svg>
- </div>
- `)).IsSvgImage())
- assert.False(t, DetectContentType([]byte(`<html><body><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg></svg></body></html>`)).IsSvgImage())
- assert.False(t, DetectContentType([]byte(`<html><body><?xml version="1.0" encoding="UTF-8"?><svg></svg></body></html>`)).IsSvgImage())
- }
-
- func TestIsPDF(t *testing.T) {
- pdf, _ := base64.StdEncoding.DecodeString("JVBERi0xLjYKJcOkw7zDtsOfCjIgMCBvYmoKPDwvTGVuZ3RoIDMgMCBSL0ZpbHRlci9GbGF0ZURlY29kZT4+CnN0cmVhbQp4nF3NPwsCMQwF8D2f4s2CNYk1baF0EHRwOwg4iJt/NsFb/PpevUE4Mjwe")
- assert.True(t, DetectContentType(pdf).IsPDF())
- assert.False(t, DetectContentType([]byte("plain text")).IsPDF())
- }
-
- func TestIsVideo(t *testing.T) {
- mp4, _ := base64.StdEncoding.DecodeString("AAAAGGZ0eXBtcDQyAAAAAGlzb21tcDQyAAEI721vb3YAAABsbXZoZAAAAADaBlwX2gZcFwAAA+gA")
- assert.True(t, DetectContentType(mp4).IsVideo())
- assert.False(t, DetectContentType([]byte("plain text")).IsVideo())
- }
-
- func TestIsAudio(t *testing.T) {
- mp3, _ := base64.StdEncoding.DecodeString("SUQzBAAAAAABAFRYWFgAAAASAAADbWFqb3JfYnJhbmQAbXA0MgBUWFhYAAAAEQAAA21pbm9yX3Zl")
- assert.True(t, DetectContentType(mp3).IsAudio())
- assert.False(t, DetectContentType([]byte("plain text")).IsAudio())
-
- assert.True(t, DetectContentType([]byte("ID3Toy\000")).IsAudio())
- assert.True(t, DetectContentType([]byte("ID3Toy\n====\t* hi 🌞, ...")).IsText()) // test ID3 tag for plain text
- assert.True(t, DetectContentType([]byte("ID3Toy\n====\t* hi 🌞, ..."+"🌛"[0:2])).IsText()) // test ID3 tag with incomplete UTF8 char
- }
-
- func TestDetectContentTypeFromReader(t *testing.T) {
- mp3, _ := base64.StdEncoding.DecodeString("SUQzBAAAAAABAFRYWFgAAAASAAADbWFqb3JfYnJhbmQAbXA0MgBUWFhYAAAAEQAAA21pbm9yX3Zl")
- st, err := DetectContentTypeFromReader(bytes.NewReader(mp3))
- assert.NoError(t, err)
- assert.True(t, st.IsAudio())
- }
-
- func TestDetectContentTypeOgg(t *testing.T) {
- oggAudio, _ := hex.DecodeString("4f67675300020000000000000000352f0000000000007dc39163011e01766f72626973000000000244ac0000000000000071020000000000b8014f6767530000")
- st, err := DetectContentTypeFromReader(bytes.NewReader(oggAudio))
- assert.NoError(t, err)
- assert.True(t, st.IsAudio())
-
- oggVideo, _ := hex.DecodeString("4f676753000200000000000000007d9747ef000000009b59daf3012a807468656f7261030201001e00110001e000010e00020000001e00000001000001000001")
- st, err = DetectContentTypeFromReader(bytes.NewReader(oggVideo))
- assert.NoError(t, err)
- assert.True(t, st.IsVideo())
- }
|