bailong360

很cool,而且效率很高!
能否请教一下原理?

再提一个建议:如楼上所说只能加载一个dll局限性很大,除非再把ext.dll做成一个加载器.然而这样不如让cmdex提供一个load命令,用来加载其他dll.这样不同开发者开发的dll就可以同时使用
------------------------
测试了一下 发现有一个bug

@echo off
set a=%s%
my%__%cmd %a%
pause
复制代码

这样一段代码,执行后输出的是

cmd:mycmd
lastcmdline:my%__%cmd %a%

请按任意键继续. . .
复制代码

%__%和%a%没有被扩展,不过"cmd:mycmd"倒是得到了正确结果 望改进

bailong360

回复 9# misaki
前段时间备战高考 就没怎么来了
输入cmdex其实就看到你的ID了/手动斜眼

其实想到了你可能是hook了字符串处理的函数...但是觉得以这些函数的使用频率对效率影响应该比较大,就排除了这种可能性......看来功力尚浅啊
不过这倒是给了我一个新思路

交给开发者来处理命令行感觉有利有弊吧

bailong360

回复 11# misaki
测试过各系统下的兼容性吗? 感觉这个很有戏  

试着编写了一个AddCmd  tcc编译通过
将变量扩展和参数都处理了一下,应该是个通常情况下比较友好的接口

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

#define DLL_EXPORT __declspec(dllexport)
#define HANDLED     0
#define UNHANDLED   1
#define MAX_CMD_LENGTH 512

typedef struct cmdTable {
    wchar_t cmd[MAX_CMD_LENGTH];
    int     (*func)(int, wchar_t *[]);
    struct  cmdTable *next;
} CMDTABLE;

DLL_EXPORT int ExtCallBack(wchar_t *, wchar_t *);
char* WcharToChar(wchar_t *wstr);
int AddCmd(int argc, wchar_t *cmdline[]);

CMDTABLE cmd_Load = {L"AddCmd", AddCmd, NULL};
CMDTABLE *head = &cmd_Load;
CMDTABLE *tail = &cmd_Load;

DLL_EXPORT int ExtCallBack(wchar_t *cmd, wchar_t *cmdline)
{
    CMDTABLE *current = head;
    do {
        if (!wcsnicmp(cmd, current->cmd, wcslen(current->cmd))) {
            int       argc;
            wchar_t **argv;
            wchar_t newcmdline[MAX_CMD_LENGTH];
            ExpandEnvironmentStringsW(cmdline, newcmdline, MAX_CMD_LENGTH + 1);
            newcmdline[wcslen(newcmdline) - 2] = '\0'; //去除末尾的\n
            argv = CommandLineToArgvW(newcmdline, &argc);
            current->func(argc, argv);
            LocalFree(argv);
            return HANDLED;
        }
        current = current -> next;
    } while (current != NULL);
    return UNHANDLED;
}

char* WcharToChar(wchar_t *wstr)
{
    int len   = WideCharToMultiByte(CP_ACP, 0, wstr, -1, NULL, 0, NULL, NULL);
    char *str = (char *)malloc(len * sizeof(char));
    WideCharToMultiByte(CP_ACP, 0, wstr, -1, str, len, NULL, NULL);
    return str;
}

int AddCmd(int argc, wchar_t *argv[])
{
    if (argc < 4) {
        puts("Usage: AddCmd YourCmd YourDll YourFunc");
        puts("**YourFunc should be defined like this \"int YourFunc(int argc, wchar_t *argv[])\"");
        return 0;
    }
    HMODULE hDLL   = LoadLibraryW(argv[2]);
    char *funcName = WcharToChar(argv[3]);
    void *func     = (void *)GetProcAddress(hDLL, funcName);
    
    if (hDLL == NULL) {
        fprintf(stderr, "ERROR:can't load dll \"%S\"", argv[2]);
        free(funcName);
        return 0;
    } else if (func == NULL) {
        fprintf(stderr, "ERROR:can't load func \"%s\"", funcName);
        free(funcName);
        return 0;
    }
    tail->next = (CMDTABLE *)malloc(sizeof(CMDTABLE));
    wcscpy(tail->next->cmd, argv[1]);
    tail->next->func = (void *)GetProcAddress(hDLL, funcName);
    tail->next->next = NULL;
    tail = tail->next;
    free(funcName);
    return 0;
}
复制代码

bailong360

回复 13# misaki
刚刚又想到了一个问题
ExpanEnvironmentStringsW只能扩展以%%包裹的变量,!!好像不好处理- -