例外がスロー ブレークポイントが発生しました ヒープが原因かなと考えています。

Question

前提・実現したいこと

C言語初心者です。
指定ファイル内を探索してディレクトリ・ファイル名を出力するシステムを作成しています。
現状ブレークポイントが発生しました。と出ますがCSV出力されている状態です。
また出力の際正常に対象全てのデータが出力されることもあれば途中まで出力されることもあります。
(Sample2やSample4で終わったりしています。)
検索してみるとヒープが破壊されていることが原因なのかなと考えているのですが
どのように修正すればよいか分からないといった状況です。
よろしくお願いいたします。

発生している問題・エラーメッセージ

例外が発生しました。
file_search.exe によってブレークポイントが発生しました。

該当のソースコード

Sampleファイル
https://gyazo.com/c8e4387bf035c97eeb96ad42b844f837
呼び出し履歴
https://gyazo.com/936af4fdc360b075db5ccbcc9a031a44
ブレークポイント発生箇所(exe_common.inl)
https://gyazo.com/1ab31c3a03fc8af1fb0fa9a2524ec853

main.c
#define _CRT_SECURE_NO_WARNINGS

#include <Windows.h>
#include <stdio.h>
#include <locale.h>
#include <stdlib.h>
#include <io.h>
#include <sys/stat.h>
#include <direct.h>
#include <string.h>
#include <wchar.h>
#include <limits.h>

void FileSearch(const wchar_t* dirpath, int count);
char file_name[1024];
const char output_dir[1024];
FILE* file;

int main(int argc, char* argv[])
{
    // 絶対パスの分割と取得
    char* search_file = argv[1];
    
    char path_dir[1024] = { '\0' };
    char* dirend;

    dirend = strrchr(search_file, '\');
    dirend = dirend + 1;

    for (int n = 0; n < (int)strlen(search_file); n++)
    {
        path_dir[n] = search_file[n];
        if (search_file + n == dirend - 1)
        {
            break;
        }
    }

    // パスの変数化
    sprintf(file_name, "%sOutput/output.csv", path_dir);
    sprintf(output_dir, "%sOutput", path_dir);
    strcat(path_dir, dirend);
    sprintf(path_dir,"%s\\", path_dir);

    //システム既定のロケールに設定する
    setlocale(LC_ALL, "");
    //マルチバイト文字をワイド文字に変換
    wchar_t* wide_path_dir[1024];
    mbstowcs(wide_path_dir, path_dir, sizeof(wide_path_dir) / sizeof(wide_path_dir[0]));

    if (argc > 1)
    {
        // Outputディレクトリがデスクトップに存在するか確認
        struct stat output;
        int existence = stat(output_dir, &output);
        // Outputディレクトリがなかった時作成する
        if (existence != 0)
        {
            (void)_mkdir(output_dir);
        }

        file = fopen(file_name, "w");
        fprintf(file, "%s\n", dirend);
        fclose(file);
    }
    
    // ディレクトリ名とファイル名、ファイルサイズを表示する
    FileSearch(wide_path_dir, 1);
    return 0;
}


// 引数として渡したディレクトリ以下のディレクトリとファイルを全て表示する
void FileSearch(const wchar_t* dirpath, int count)
{

    long long int size = 0;
    int i = 0;
    int datasize = 0;
    int pathsize = 0;

    // ハンドル
    HANDLE hFind;
    // ファイル情報
    WIN32_FIND_DATA win32fd;

    datasize = sizeof(win32fd);

    // ディレクトリにワイルドカードを追加
    // ファイル名も含んだパス
    wchar_t* dirwildpath;
    dirwildpath = (wchar_t*)malloc(sizeof(wchar_t) * datasize);
    // ファイルパスをコピー
    wcscpy(dirwildpath, dirpath);
    // ワイルドカードを追加
    wcscat(dirwildpath, L"*");

    // ハンドルを取得しファイル情報を保存する
    hFind = FindFirstFile(dirwildpath, &win32fd);

    // 何も見つからなかったとき
    if (hFind == INVALID_HANDLE_VALUE) {
        FindClose(hFind);
        return;
    }
    free(dirwildpath);
    while (FindNextFile(hFind, &win32fd))
    {
        // カレントディレクトリと親ディレクトリのときは何もしない
        if (wcscmp(win32fd.cFileName, L".") == 0
            || wcscmp(win32fd.cFileName, L"..") == 0 || wcscmp(win32fd.cFileName, L"desktop.ini") == 0)
        {
            continue;
        }

        // ディレクトリのときの処理
        else if (win32fd.dwFileAttributes == FILE_ATTRIBUTE_DIRECTORY)
        {
            file = fopen(file_name, "a");
            for (i = 0; i < count; i++)
            {
                fprintf(file, ",");
            }
            fwprintf(file, L"%ls\n", win32fd.cFileName);
            fclose(file);

            // 文字列バッファを更新
            pathsize = sizeof(dirpath);
            wchar_t* dir_buf = (wchar_t*)malloc(sizeof(wchar_t) * pathsize);
            /*if (count > 1)
            {
                free(dir_buf[count - 1]);
            }*/

            // ディレクトリをコピーする
            wcscpy(dir_buf, dirpath);
            wcscat(dir_buf, win32fd.cFileName);
            wcscat(dir_buf, L"\");
            // 再帰する
            FileSearch(dir_buf, count + 1);
        }
        else
        {
            // ファイルのとき
            file = fopen(file_name, "a");
            long long int fsize = fseek(file, 0, SEEK_END);
            fgetpos(file, &size);
            fseek(file, fsize, SEEK_SET);
            for (i = 0; i < count; i++)
            {
                fprintf(file, ",");
            }
            fwprintf(file, L"%ls,%lld[byte]\n", win32fd.cFileName, size);
            fclose(file);
        }
    }
    FindClose(hFind);
}

試したこと

動的配列のdirwildpathをFindClose(hFind)の前で解放してみたが変化なし
dir_bufはどのタイミングで解放していいか分かりません。

補足情報（FW/ツールのバージョンなど）

Windows10
Microsoft Visual C++ 2019

不足情報あれば随時追加させていただきます。

Answer 1

    WIN32_FIND_DATA win32fd;

    datasize = sizeof(win32fd);

    // ディレクトリにワイルドカードを追加
    // ファイル名も含んだパス
    wchar_t* dirwildpath;
    dirwildpath = (wchar_t*)malloc(sizeof(wchar_t) * datasize);

ここで sizeof(wchar_t) * datasize としているのは妙です。dirwildpath に入れたいデータサイズと WIN32_FIND_DATA のサイズは関係がありません。sizeof(wchar_t) * (wcslen(dirpath) + 2) ではないでしょうか。
(とは言え、WIN32_FIND_DATA はかなり大きいので、ここが原因で問題が起きているわけではないと思います。)

            pathsize = sizeof(dirpath);
            wchar_t* dir_buf = (wchar_t*)malloc(sizeof(wchar_t) * pathsize);

pathsize がポインタのサイズ(4 か 8 でしょう)になっています。そのため高い確率でバッファオーバーランしています。これも wcslen(dirpath) + wcslen(win32fd.cFileName) + 2 のようにすべきです。

• Cで文字列操作するのは間違いやすいので、C++でstd::wstringを活用するのをおすすめします。
• 極端に長い・深いパスを扱わなくてもよいなら、ファイル名用バッファのサイズを FILENAME_MAX などでの固定長にして、この長さを超えそうなら処理しない、というのも手です。

Comments

[shunxile]

>>int32_tさん
回答ありがとうございます。
修正したところ改善され出力されました。ありがとうございます。
いろいろと理解しきれてないことがあるのが分かりました。
+2というのはワイド文字の２バイトとワイルドカードの『¥』1文字ということで２×１ということでしょうか？

[int32_t]

+2 は、追加する '*' か '\' のぶんと、文字列終端の '\0' のぶんです。

[shunxile]

>>int32_tさん
そういうことでしたか！ありがとうございます。
またFILENAME＿MAXやstd::wstringなど知らないことだらけで
すごく勉強になりました。
個人的にもCの文字列に関して苦手意識もあったのですごく助かりました！
ありがとうございます。

Answer 2

#define _CRT_SECURE_NO_WARNINGS

#include <Windows.h>
#include <stdio.h>
#include <locale.h>
#include <stdlib.h>

void FileSearch(const wchar_t* dirpath, int depth) {

    // ハンドル
    HANDLE hFind;
    // ファイル情報
    WIN32_FIND_DATAW win32fd;

    wchar_t* dirwildpath = (wchar_t*)malloc( (wcslen(dirpath)+2) * sizeof(wchar_t) );
    wsprintf(dirwildpath, L"%s*", dirpath);

    hFind = FindFirstFileW(dirwildpath, &win32fd);
    free(dirwildpath);

    if (hFind == INVALID_HANDLE_VALUE) {
        FindClose(hFind);
        return;
    }

    do {
        if (   wcscmp(win32fd.cFileName, L".") == 0
            || wcscmp(win32fd.cFileName, L"..") == 0 
            || wcscmp(win32fd.cFileName, L"desktop.ini") == 0) {
            continue;
        }

        if (win32fd.dwFileAttributes == FILE_ATTRIBUTE_DIRECTORY) {
            for ( int i = 0; i < depth; ++i ) wprintf(L"    ");
            wprintf(L"dir.[%s]\n", win32fd.cFileName);

            size_t pathsize = wcslen(dirpath) + wcslen(win32fd.cFileName) + 2;
            wchar_t* dir_buf = (wchar_t*)malloc(sizeof(wchar_t) * pathsize);
            wsprintf(dir_buf,L"%s%s\", dirpath, win32fd.cFileName);
            FileSearch(dir_buf, depth+1);
            free(dir_buf);
        } else {
            for (int i = 0; i < depth; ++i) wprintf(L"    ");
            wprintf(L"file[%s]\n", win32fd.cFileName);
        }
    } while (FindNextFileW(hFind, &win32fd));
    FindClose(hFind);
}

int main() {
  setlocale(LC_ALL,"japanese");
  FileSearch(L".\", 0);
  return 0;
}