src.cpp

#include <windows.h>
#include <stdio.h>
#include <filter.h>

#define GROUP_CHAR '/'
#define FILE_LINE_MAX_LEN 256

const char setting_effect_name[] = "_Setting.txt";
const char setting_filter_name[] = "_SettingF.txt";
const char exedit_filename[] = "exedit.auf";
char effpath[MAX_PATH + 1];
char filpath[MAX_PATH + 1];
char last_group_name[FILE_LINE_MAX_LEN + 1];
inline static struct {
    FILTER** LoadedFilterTable = (FILTER**)(0x187c98);
    int* ExeditFilterCount_ptr = (int*)(0x146248);
    char* add_effect_str = (char*)(0xa4d30);
    char* basic_effect_str = (char*)(0xa4d44);
    HMENU* hMenu_r_click = (HMENU*)0x168598;
    HMENU* hMenu_dlg_v_effect = (HMENU*)0x158d2c;
    HMENU(__cdecl* AddExeditMediaMenuItem)(HMENU hmenu, LPCSTR name, BOOL flag) = (decltype(AddExeditMediaMenuItem))0x39650;
    void(__cdecl* AddExeditMediaMenu)(HMENU menu, UINT_PTR uIDNewItem, LPCSTR lpNewItem, char* group_name) = (decltype(AddExeditMediaMenu))0x39970;
    void(__cdecl* AddExeditMenu)(HMENU menu, int uIDNewItem, int flag) = (decltype(AddExeditMenu))0x39a10;
    BOOL(__cdecl* ReadAlias)(char* file_name, char* folder_name) = (decltype(ReadAlias))0x39cf0;
    char* (__cdecl* GetExaPath)(char* file_name, char* folder_name) = (decltype(GetExaPath))0x39ed0;
    int(__cdecl* GetExaType)(char* path) = (decltype(GetExaType))0x2a2e0;
    int(__cdecl* GetAliasUID)(char* file_name, char* folder_name, int uIDItemOffset) = (decltype(GetAliasUID))0x283f0;
}ee;

FILTER_DLL filter_dll = {
    FILTER_FLAG_ALWAYS_ACTIVE | FILTER_FLAG_NO_CONFIG,
    NULL,NULL,
    const_cast<char*>("エフェクト並び替え"),
    NULL,NULL,NULL,
    NULL,NULL,
    NULL,NULL,NULL,
    NULL,
    func_init
};
EXTERN_C FILTER_DLL __declspec(dllexport)* __stdcall GetFilterTable() {
    return &filter_dll;
}


DWORD init_exeditfp(FILTER* fp) {
	SYS_INFO si;
	fp->exfunc->get_sys_info(NULL, &si);
    
	for (int i = 0; i < si.filter_n; i++) {
		FILTER* efp = (FILTER*)fp->exfunc->get_filterp(i);
        if (efp->information != NULL) {
            if (!lstrcmpA(efp->information, "拡張編集(exedit) version 0.92 by ＫＥＮくん")) {
                return (DWORD)efp->dll_hinst;
            }
        }
	}
    return 0;
}


DWORD exedit_dll_hinst;
BOOL exedit_ReplaceCall(DWORD exedit_address, void* new_address) {
    DWORD oldProtect;
    DWORD* address = (DWORD*)(exedit_address + exedit_dll_hinst);
    if (!VirtualProtect(address, 4, PAGE_EXECUTE_READWRITE, &oldProtect)) {
        return FALSE;
    }
    *address = (DWORD)new_address - (DWORD)address - 4;
    return VirtualProtect(address, 4, oldProtect, &oldProtect);
}

static char mod_path[MAX_PATH + 1];
char* get_file_name(FILTER* fp) {
    HMODULE mod = fp->dll_hinst;
    if (mod == nullptr) {
        GetModuleHandleExA(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS, (LPCSTR)fp, &mod);
    }
    GetModuleFileNameA(mod, mod_path, MAX_PATH);

    char* ptr = mod_path;
    for (int i = 0; i < MAX_PATH; i++) {
        if (IsDBCSLeadByteEx(0, mod_path[i])) {
            i++;
        } else if (mod_path[i] == '\0') {
            break;
        } else if (mod_path[i] == '\\' || mod_path[i] == '/') {
            ptr = mod_path + i + 1;
        }
    }
    return ptr;
}

static char name_and_file[FILE_LINE_MAX_LEN + 1];
char* get_name_and_file(FILTER* fp) {
    char* filename = get_file_name(fp);
    if (lstrcmpA(filename, exedit_filename)) {
        wsprintfA(name_and_file, "%s:%s", fp->name, filename);
        return name_and_file;
    } else {
        return fp->name;
    }
}


void fputsn(char const* _Buffer, FILE* _Stream) {
    fputs(_Buffer, _Stream);
    fputc('\n', _Stream);
}
void fputcn(int _Character, FILE* _Stream) {
    fputc(_Character, _Stream);
    fputc('\n', _Stream);
}

void fput_alias(char const* folder_name, char const* file_name, FILE* _Stream) {
    if (folder_name != NULL && folder_name[0] != '\0') {
        fputs(folder_name, _Stream);
    }
    fputc('\\', _Stream);
    fputsn(file_name, _Stream);
}
void fput_group(char const* group_name, FILE* _Stream) {
    if (group_name != NULL && group_name[0] != '\0') {
        if (lstrcmpA(last_group_name, group_name) != 0) {
            lstrcpyA(last_group_name, group_name);
            fputc(GROUP_CHAR, _Stream);
            fputsn(group_name, _Stream);
        }
    } else {
        if (last_group_name[0] != '\0') {
            last_group_name[0] = '\0';
            fputcn(GROUP_CHAR, _Stream);
        }
    }
}
int split_file_folder(char* str) {
    int file_name_ptr = 0;
    for (int i = 0; i < FILE_LINE_MAX_LEN; i++) {
        if (IsDBCSLeadByteEx(0, str[i])) {
            i++;
        } else if (str[i] == '\0') {
            break;
        } else if (str[i] == '\\' || str[i] == '/') {
            if (file_name_ptr) {
                return -1;
            }
            str[i] = '\0';
            file_name_ptr = i + 1;

        }
    }
    return file_name_ptr;
}
void str_chg_n20(char* str) {
    for (int i = 0; i < FILE_LINE_MAX_LEN; i++) {
        if (IsDBCSLeadByteEx(0, str[i])) {
            i++;
        } else if (str[i] == '\n') {
            str[i] = '\0';
            break;
        } else if (str[i] == '\0') {
            break;
        }
    }
}

BOOL is_video_filter_effect_tl(FILTER* efp) {
    return ((efp->flag & 0x52000a0) == 0x4000020 && (efp->flag & 0x38));
}
BOOL is_video_filter_effect_dlg(FILTER* efp) {
    return ((efp->flag & 0x5200120) == 0x4000020);
}
BOOL is_video_filter_effect(FILTER* efp) {
    return is_video_filter_effect_tl(efp) || is_video_filter_effect_dlg(efp);
}
int get_effect_idx(char* name) {
    for (int i = 0; i < *ee.ExeditFilterCount_ptr; i++) {
        FILTER* efp = ee.LoadedFilterTable[i];
        if (is_video_filter_effect(efp)) {
            if (lstrcmpA(get_name_and_file(efp), name) == 0) {
                return i;
            }
        }
    }
    return -1;
}

BOOL is_video_filter_object_tl(FILTER* efp) {
    return (((efp->flag & 0x4000000) == 0 || (efp->flag & 0x38) == 0) && (efp->flag & 0x1008080) == 0);
}
BOOL is_video_filter_object_dlg(FILTER* efp) {
    return (((efp->flag & 0x4000000) == 0 || (efp->flag & 0x38) == 0) && (efp->flag & 0x1208100) == 0);
}
BOOL is_audio_filter_object_dlg(FILTER* efp) {
    return (((efp->flag & 0x4000000) == 0 || (efp->flag & 0x38) == 0) && (efp->flag & 0x1208100) == 0x200000);
}
BOOL is_filter_object(FILTER* efp) {
    return is_video_filter_object_tl(efp) || is_video_filter_object_dlg(efp) || is_audio_filter_object_dlg(efp);
}

int get_filter_idx(char* name) {
    for (int i = 0; i < *ee.ExeditFilterCount_ptr; i++) {
        FILTER* efp = ee.LoadedFilterTable[i];
        if (is_filter_object(efp)) {
            if (lstrcmpA(get_name_and_file(efp), name) == 0) {
                return i;
            }
        }
    }
    return -1;
}

#define ALIAS_TYPE_ENABLE   0b00001
#define ALIAS_TYPE_EFFECT   0b00010
#define ALIAS_TYPE_AUDIO    0b00100
BOOL is_alias_effect(char* file_name, char* folder_name){
    return (ee.GetExaType(ee.GetExaPath (file_name, folder_name)) & 0x1f) == 3;
}

void write_default_eff_list() {
    FILE* inifile;
    if (fopen_s(&inifile, effpath, "w") == 0) {
        BOOL basic_eff_complete = FALSE;
        for (int i = 0; i < *ee.ExeditFilterCount_ptr; i++) {
            FILTER* efp = ee.LoadedFilterTable[i];
            if (is_video_filter_effect(efp)) {
                if (efp->flag & 0x8000) {
                    if (!basic_eff_complete) {
                        fputc(GROUP_CHAR, inifile);
                        fputsn(ee.basic_effect_str, inifile);
                        for (int j = i; j < *ee.ExeditFilterCount_ptr; j++) {
                            FILTER* befp = ee.LoadedFilterTable[j];
                            if ((befp->flag & 0x8000) && is_video_filter_effect(befp)) {
                                fputsn(get_name_and_file(befp), inifile);
                            }
                        }
                        fputcn(GROUP_CHAR, inifile);
                        basic_eff_complete = TRUE;
                    }
                } else {
                    fputsn(get_name_and_file(efp), inifile);
                }
            }
        }
        fclose(inifile);
    }
}
void write_default_fil_list() {
    FILE* inifile;
    if (fopen_s(&inifile, filpath, "w") == 0) {
        for (int i = 0; i < *ee.ExeditFilterCount_ptr; i++) {
            FILTER* efp = ee.LoadedFilterTable[i];
            if (is_filter_object(efp)) {
                fputsn(get_name_and_file(efp), inifile);
            }
        }
        fclose(inifile);
    }
}


void set_last_group(char* str) {
    for (int i = 0; i < FILE_LINE_MAX_LEN; i++) {
        if (IsDBCSLeadByteEx(0, str[i])) {
            last_group_name[i] = str[i];
            i++;
        } else {
            if (str[i] == '\n') {
                last_group_name[i] = '\0';
                break;
            }
        }
        last_group_name[i] = str[i];
    }
}


void add_menu_apend(HMENU menu, int menu_uid, char* group, char* name) {
    if (group[0] != '\0') {
        menu = ee.AddExeditMediaMenuItem(menu, ee.add_effect_str, 0);
        menu = ee.AddExeditMediaMenuItem(menu, group, 0);
        ee.AddExeditMediaMenu(menu, menu_uid, name, NULL);
    } else {
        ee.AddExeditMediaMenu(menu, menu_uid, name, ee.add_effect_str);
    }
}

void add_menu(HMENU menu, int menu_uid, char* group, char* name) {
    if (group[0] != '\0') {
        menu = ee.AddExeditMediaMenuItem(menu, group, 0);
    }
    ee.AddExeditMediaMenu(menu, menu_uid, name, NULL);
}

void __cdecl AddExeditMenu_FilterEffect_RClick(HMENU menu, int uIDNewItem, int flag) {
    FILE* inifile;
    if (fopen_s(&inifile, effpath, "r") == 0) {

        DeleteMenu(menu, 100, 0);

        bool filter_called[512];
        for (int i = 0; i < 512; i++) {
            filter_called[i] = false;
        }

        char str[FILE_LINE_MAX_LEN + 1];
        last_group_name[0] = last_group_name[FILE_LINE_MAX_LEN - 1] = '\0';

        while (fgets(str, FILE_LINE_MAX_LEN, inifile)) {
            if (str[0] == GROUP_CHAR) {
                set_last_group(&str[1]);
            } else if (str[0] != '\n') {
                str_chg_n20(str);
                int filename_ptr = split_file_folder(str);
                char* folder_name;
                if (filename_ptr == 0) {
                    int filter_idx = get_effect_idx(str);
                    if (0 <= filter_idx) { // エフェクトが存在する
                        FILTER* efp = ee.LoadedFilterTable[filter_idx];
                        filter_called[filter_idx] = true;
                        if (is_video_filter_effect_tl(efp)) {
                            add_menu_apend(menu, uIDNewItem + filter_idx, last_group_name, efp->name);
                            continue;
                        }
                    }
                    folder_name = &last_group_name[FILE_LINE_MAX_LEN - 1];
                } else {
                    folder_name = str;
                }
                if (0 <= filename_ptr) { // exaフォルダは1階層まで
                    char* file_name = &str[filename_ptr];
                    if (is_alias_effect(file_name, folder_name)) {
                        int alias_menu_id = abs(ee.GetAliasUID(file_name, folder_name, 3000));
                        if (alias_menu_id) {
                            add_menu_apend(menu, alias_menu_id, last_group_name, file_name);
                        }
                    }
                }
            }
        }
        fclose(inifile);

        BOOL file_a_open = (fopen_s(&inifile, effpath, "a") == 0);
        for (int i = 0; i < *ee.ExeditFilterCount_ptr; i++) {
            if (!filter_called[i]) {
                FILTER* efp = ee.LoadedFilterTable[i];

                if ((efp->flag & 0x5000080) == 0x4000000 && (efp->flag & 0x38)) {
                    if ((efp->flag & 0x20) == 0) {
                        ee.AddExeditMediaMenu(menu, uIDNewItem + i, efp->name, (char*)((int)efp + 128));
                    } else if ((efp->flag & 0x200000) == 0) {
                        if (file_a_open) {
                            if (last_group_name[0] != '\0') {
                                last_group_name[0] = '\0';
                                fputcn(GROUP_CHAR, inifile);
                            }
                            fputsn(get_name_and_file(efp), inifile);
                        }
                        add_menu_apend(menu, uIDNewItem + i, last_group_name, efp->name);
                    }
                }
            }
        }
        if (file_a_open) {
            fclose(inifile);
        }
    } else {
        ee.AddExeditMenu(menu, uIDNewItem, flag);
    }
}
void __cdecl AddExeditMenu_FilterEffect_Dialog(HMENU menu, int uIDNewItem, int flag) {
    FILE* inifile;
    if (fopen_s(&inifile, effpath, "r") == 0) {

        DeleteMenu(menu, 100, 0);

        bool filter_called[512];
        for (int i = 0; i < 512; i++) {
            filter_called[i] = false;
        }

        char str[FILE_LINE_MAX_LEN + 1];
        last_group_name[0] = last_group_name[FILE_LINE_MAX_LEN - 1] = '\0';

        while (fgets(str, FILE_LINE_MAX_LEN, inifile)) {
            if (str[0] == GROUP_CHAR) {
                set_last_group(&str[1]);
            } else if (str[0] != '\n') {
                str_chg_n20(str);
                int filename_ptr = split_file_folder(str);
                char* folder_name;
                if (filename_ptr == 0) {
                    int filter_idx = get_effect_idx(str);
                    if (0 <= filter_idx) { // エフェクトが存在する
                        FILTER* efp = ee.LoadedFilterTable[filter_idx];
                        filter_called[filter_idx] = true;
                        if (is_video_filter_effect_dlg(efp)) {
                            add_menu(menu, uIDNewItem + filter_idx, last_group_name, efp->name);
                            continue;
                        }
                    }
                    folder_name = &last_group_name[FILE_LINE_MAX_LEN - 1];
                } else {
                    folder_name = str;
                }
                if (0 <= filename_ptr) { // exaフォルダは1階層まで
                    char* file_name = &str[filename_ptr];
                    if (is_alias_effect(file_name, folder_name)) {
                        int alias_menu_id = abs(ee.GetAliasUID(file_name, folder_name, 3000));
                        if (alias_menu_id) {
                            add_menu(menu, alias_menu_id, last_group_name, file_name);
                        }
                    }
                }
            }
        }
        fclose(inifile);

        last_group_name[0] = '\0';
        for (int i = 0; i < *ee.ExeditFilterCount_ptr; i++) {
            if (!filter_called[i]) {
                FILTER* efp = ee.LoadedFilterTable[i];
                if (is_video_filter_effect_dlg(efp)) {
                    add_menu(menu, uIDNewItem + i, last_group_name, efp->name);
                }
            }
        }
    } else {
        ee.AddExeditMenu(menu, uIDNewItem, flag);
    }
}

void __cdecl AddExeditMenu_FilterObject_RClick(HMENU menu, int uIDNewItem, int flag) {
    FILE* inifile;
    if (fopen_s(&inifile, filpath, "r") == 0) {

        DeleteMenu(menu, 100, 0);

        bool filter_called[512];
        for (int i = 0; i < 512; i++) {
            filter_called[i] = false;
        }

        char str[FILE_LINE_MAX_LEN + 1];
        last_group_name[0] = last_group_name[FILE_LINE_MAX_LEN - 1] = '\0';

        while (fgets(str, FILE_LINE_MAX_LEN, inifile)) {
            if (str[0] == GROUP_CHAR) {
                set_last_group(&str[1]);
            } else if (str[0] != '\n') {
                str_chg_n20(str);
                int filter_idx = get_filter_idx(str);
                if (0 <= filter_idx) {
                    FILTER* efp = ee.LoadedFilterTable[filter_idx];
                    filter_called[filter_idx] = true;
                    if (is_video_filter_object_tl(efp)) {
                        add_menu(menu, uIDNewItem + filter_idx, last_group_name, efp->name);
                        continue;
                    }
                }
            }
        }
        fclose(inifile);

        BOOL file_a_open = (fopen_s(&inifile, filpath, "a") == 0);
        for (int i = 0; i < *ee.ExeditFilterCount_ptr; i++) {
            if (!filter_called[i]) {
                FILTER* efp = ee.LoadedFilterTable[i];

                if (is_filter_object(efp)) {
                    if (file_a_open) {
                        if (last_group_name[0] != '\0') {
                            last_group_name[0] = '\0';
                            fputcn(GROUP_CHAR, inifile);
                        }
                        fputsn(get_name_and_file(efp), inifile);
                    }
                    if (is_video_filter_object_tl(efp)) {
                        add_menu(menu, uIDNewItem + i, last_group_name, efp->name);
                    }
                }
            }
        }
        if (file_a_open) {
            fclose(inifile);
        }
    } else {
        ee.AddExeditMenu(menu, uIDNewItem, flag);
    }
}
void __cdecl AddExeditMenu_FilterObject_Dialog(HMENU menu, int uIDNewItem, int flag) {
    FILE* inifile;
    if (fopen_s(&inifile, filpath, "r") == 0) {

        DeleteMenu(menu, 100, 0);

        bool filter_called[512];
        for (int i = 0; i < 512; i++) {
            filter_called[i] = false;
        }

        char str[FILE_LINE_MAX_LEN + 1];
        last_group_name[0] = last_group_name[FILE_LINE_MAX_LEN - 1] = '\0';

        while (fgets(str, FILE_LINE_MAX_LEN, inifile)) {
            if (str[0] == GROUP_CHAR) {
                set_last_group(&str[1]);
            } else if (str[0] != '\n') {
                str_chg_n20(str);
                int filter_idx = get_filter_idx(str);
                if (0 <= filter_idx) {
                    FILTER* efp = ee.LoadedFilterTable[filter_idx];
                    filter_called[filter_idx] = true;
                    if (is_video_filter_object_dlg(efp)) {
                        add_menu(menu, uIDNewItem + filter_idx, last_group_name, efp->name);
                        continue;
                    }
                }
            }
        }
        fclose(inifile);

        last_group_name[0] = '\0';
        for (int i = 0; i < *ee.ExeditFilterCount_ptr; i++) {
            if (!filter_called[i]) {
                FILTER* efp = ee.LoadedFilterTable[i];
                if (is_video_filter_object_dlg(efp)) {
                    add_menu(menu, uIDNewItem + i, last_group_name, efp->name);
                }
            }
        }
    } else {
        ee.AddExeditMenu(menu, uIDNewItem, flag);
    }
}

BOOL __cdecl ReadAlias_wrap(char* file_name, char* folder_name) {
    int alias_menu_id = ee.GetAliasUID(file_name, folder_name, 3000);
    if (alias_menu_id <= 0) {
        return FALSE;
    }
    if (is_alias_effect(file_name, folder_name)) {
        FILE* inifile;
        if (fopen_s(&inifile, effpath, "a") == 0) {
            fput_group(folder_name, inifile);
            fput_alias(folder_name, file_name, inifile);
            fclose(inifile);
        }
        HMENU menu = GetSubMenu(GetSubMenu(*ee.hMenu_r_click, 0), 0);
        add_menu_apend(menu, alias_menu_id, folder_name, file_name);
        menu = GetSubMenu(GetSubMenu(*ee.hMenu_dlg_v_effect, 0), 0);
        add_menu(menu, alias_menu_id, folder_name, file_name);
        return TRUE;
    }
    return ee.ReadAlias(file_name, folder_name);
}
void AddExeditMediaMenu_wrap39e57(HMENU menu, UINT_PTR uIDNewItem, LPCSTR lpNewItem, char* group_name) {
    FILE* inifile;
    if (fopen_s(&inifile, effpath, "a") == 0) {
        fput_group(group_name, inifile);
        fput_alias(group_name, lpNewItem, inifile);

        fclose(inifile);
    }
    ee.AddExeditMediaMenu(menu, uIDNewItem, lpNewItem, group_name);
}

BOOL func_init(FILTER* fp) {

    // aufのパスを取得→設定ファイルのパスに変更　パスが250バイトを超えるプラグインは読み込まれないようなので"_Setting"を追加しても260バイトを超えない
    GetModuleFileNameA(fp->dll_hinst, effpath, MAX_PATH);
    lstrcpyA(filpath, effpath);
    lstrcpyA(&effpath[lstrlenA(effpath) - 4], setting_effect_name);
    lstrcpyA(&filpath[lstrlenA(filpath) - 4], setting_filter_name);


    exedit_dll_hinst = init_exeditfp(fp);
    if (exedit_dll_hinst == 0) {
        MessageBoxA(fp->hwnd, "拡張編集0.92が見つかりませんでした", fp->name, MB_OK);
        return TRUE;
    }

    int* ptr = (int*)&ee;
    for (int i = sizeof(ee) >> 2; 0 < i; i--) {
        *ptr += exedit_dll_hinst;
        ptr++;
    }

    last_group_name[0] = '\0';
    FILE* inifile;
    if (fopen_s(&inifile, effpath, "r+") == 0) {

        // 最後の文字が改行でなければ追加する
        fseek(inifile, -1, SEEK_END);
        if (fgetc(inifile) != '\n') {
            fputc('\n', inifile);
        }
        fclose(inifile);

        // filter effect
        exedit_ReplaceCall(0x4485f, &AddExeditMenu_FilterEffect_RClick);
        exedit_ReplaceCall(0x4489f, &AddExeditMenu_FilterEffect_Dialog);

        exedit_ReplaceCall(0x44a0e, &ReadAlias_wrap); // folder
        exedit_ReplaceCall(0x44a6d, &ReadAlias_wrap); // root
        
    } else {
        write_default_eff_list();

    }
    if (fopen_s(&inifile, filpath, "r+") == 0) {

        // 最後の文字が改行でなければ追加する
        fseek(inifile, -1, SEEK_END);
        if (fgetc(inifile) != '\n') {
            fputc('\n', inifile);
        }
        fclose(inifile);

        // filter object
        exedit_ReplaceCall(0x4487f, &AddExeditMenu_FilterObject_RClick);
        exedit_ReplaceCall(0x448be, &AddExeditMenu_FilterObject_Dialog);

    } else {
        write_default_fil_list();

    }
    exedit_ReplaceCall(0x39e57, &AddExeditMediaMenu_wrap39e57);
    
	return TRUE;
}