画像形式の判定

著者：北本 敦

公開日：2023/08/31

最終更新日：2023/08/31

カテゴリー：技術情報

タグ：C＃

北本です。
投稿の間隔が大きく空いてしまい申し訳ありません。

前回の内容は、WebClientでのサーバーから終わりのわからない連番のJPGファイルをダウンロードする際、404エラーで終わりを判定しようとすると、ファイルの非存在時にエラーページのHTMLにリダイレクトされる仕様になっているような場合にうまくいかないという話でした。今回はこの対策を考えます。

ここでは、ファイルの中身から画像ファイルであるかどうかを判定することにします。画像ファイルの先頭にはその形式を示すヘッダ情報が含まれていますので、その数バイト分を確認することで画像形式を判断することができます。例えば、JPEGの場合であれば先頭の2バイトが必ず「0xFF, 0xD8」になっています。

他の画像形式の例も挙げておきます。

では、以上を踏まえて前回のC#のコードを修正してみます。

using System.Collections.Generic;
using System.IO;
using System.Net;

namespace ImageDownloader
{
    class Program
    {
        static void Main(string[] args)
        {
            // 実行ファイルのディレクトリ配下のdownloadフォルダを保存先とする
            string saveFolder = @"./download";
            if (!Directory.Exists(saveFolder))
            {
                // フォルダが存在しなければ作成
                Directory.CreateDirectory(saveFolder);
            }

            using (WebClient wc = new WebClient())
            {
                string urlFormat = @"http://localhost:8080/test/test.php?image_name={0}_{1}.jpg";

                int i = 1;  // 親番
                int j = 1;  // 枝番
                while (true)
                {
                    while (true)
                    {
                        string url = string.Format(urlFormat, i, j);
                        string fileName = string.Format(@"saved_image_{0}_{1}.jpg", i, j);
                        string saveFilePath = Path.Combine(saveFolder, fileName);
                        
                        byte[] data;
                        try
                        {
                            // ファイルのバイト列を取得
                            data = wc.DownloadData(url);
                        }
                        catch
                        {
                            // ダウンロードに失敗
                            break;
                        }

                        if (IsJpg(data))
                        {
                            using (var fs = new FileStream(saveFilePath, FileMode.Create, FileAccess.Write))
                            {
                                using (var bw = new BinaryWriter(fs))
                                {
                                    // ファイルを保存
                                    bw.Write(data);
                                }
                            }
                        }
                        else
                        {
                            // JPG画像ではない
                            break;
                        }

                        // 枝番を1つ進める
                        j++;
                    }

                    if (j == 1)
                    {
                        // 枝番「1」が存在しなかった場合、親番の最後の数を超えたとみなし終了
                        break;
                    }

                    // 親番を1つ進め、枝番を1に戻す
                    i++;
                    j = 1;
                }
            }
        }

        /// <summary>
        /// 対象ファイルのバイト列の先頭がheaderに指定したバイト列に一致するか
        /// </summary>
        /// <param name="data">対象ファイルのバイト列</param>
        /// <param name="header">比較対象のヘッダのバイト列</param>
        /// <returns>一致すればtrue、そうでなければfalse</returns>
        static bool IsMatchFileHeader(byte[] data, List<byte> header)
        {
            if (data.Length < header.Count)
            {
                return false;
            }

            for (int i = 0; i < header.Count; i++)
            {
                if (data[i] != header[i])
                {
                    return false;
                }
            }
            return true;
        }

        /// <summary>
        /// JPGファイルであるか
        /// </summary>
        /// <param name="data">対象ファイルのバイト列</param>
        /// <returns>JPGであればtrue、そうでなければfalse</returns>
        static bool IsJpg(byte[] data)
        {
            List<byte> header = new List<byte> { 0xFF, 0xD8 };
            return IsMatchFileHeader(data, header);
        }
    }
}

IsMatchFileHeader、IsJpgというメソッドを追加しています。IsJpgがファイルのバイト列を受け取ってJPG形式であるかを判定するメソッドです。IsMatchFileHeaderはIsJpgから呼び出しているメソッドで、引数dataに指定したバイト列の先頭が、引数headerに指定したバイト列に一致しているかを判定しています。
もし、PNG、BMP、GIFの判定をしたければ、以下のようなメソッドを作成すればOKです。

        static bool IsPng (byte[] data)
        {
            List<byte> header = new List<byte> { 0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A };
            return IsMatchFileHeader(data, header);
        }

        static bool IsGif(byte[] data)
        {
            List<byte> header = new List<byte> { 0x47, 0x49, 0x46 };
            return IsMatchFileHeader(data, header);
        }

        static bool IsBmp(byte[] data)
        {
            List<byte> header = new List<byte> { 0x42, 0x4D };
            return IsMatchFileHeader(data, header);
        }

また、前回はWebClient.DownloadFileでURLから直接ファイルを保存していましたが、WebClient.DownloadDataでバイト列として受け取り、それを前述のIsJpgメソッドでJPGファイルであるかを判定し、そうである場合のみファイルを保存するように変更しています。

以上、今回は先頭バイトから画像形式を判定する方法について取り上げました。
WebClientでの画像ダウンロードに関する話題はこれで終わりとします。