Board logo

标题: [其他] cmdex 超越capi 自定义可扩展cmd [打印本页]

作者: misaki    时间: 2017-6-1 15:23     标题: cmdex 超越capi 自定义可扩展cmd

cmdex,ex即extension。之所以说超越capi,是因为capi有很多缺陷:
而cmdex虽然不直接提供扩展的命令,但提供了用于扩展的接口,使每个人都能定义自己的cmd。

特性:

扩展方式:
      程序启动时加载所在目录的ext.dll,并调用其导出的ExtCallBack函数,参数含义可自行测试,例子如下:
  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <windows.h>
  4. #define DLL_EXPORT __declspec(dllexport)
  5. #define HANDLED     0
  6. #define UNHANDLED   1
  7. #ifdef __cplusplus
  8. extern "C"
  9. {
  10. #endif
  11. int DLL_EXPORT ExtCallBack(const wchar_t * cmd, const wchar_t * cmdline);
  12. #ifdef __cplusplus
  13. }
  14. #endif
  15. wchar_t * mycmd = L"mycmd";
  16. int DLL_EXPORT ExtCallBack(const wchar_t * cmd, const wchar_t * cmdline)
  17. {
  18.     if(wcsnicmp(cmd, mycmd, wcslen(mycmd)) == 0)
  19.     {
  20.         wprintf(L"cmd:%s\n", cmd);
  21.         wprintf(L"lastcmdline:%s\n", cmdline);
  22.         fflush(stdout);
  23.         return HANDLED;//命令已处理
  24.     }
  25.     return UNHANDLED;//命令未处理,继续查找外部程序
  26. }
复制代码
用法随你喜欢,可以简单的添加命令,甚至可以做成一个插件管理器,把常用的第三方做成动态库供加载使用。
附件是程序本体和例子编译成的dll。
链接: https://pan.baidu.com/s/1RB4-lLK4z26I_NT3Wz9Kyg 提取码: schc
作者: bbaa    时间: 2017-6-3 13:16

capix欢迎你http://www.bathome.net/viewthrea ... amp;highlight=capix
作者: misaki    时间: 2017-6-4 00:50

回复 2# bbaa


    capix只是避免了被误杀和改善了兼容性,其他方面并不比capi优越,当然也比不上这个。    而且定位不同,capix是在扩展cmd的功能,而这个是一个提供扩展的平台,让你们随意hack cmd
作者: 老刘1号    时间: 2017-6-4 07:42

本帖最后由 老刘1号 于 2017-6-4 07:50 编辑

回复 3# misaki


    貌似很不错的样子
作者: bbaa    时间: 2017-6-4 11:54

回复 3# misaki


    Capix貌似也能扩展cmd吧
作者: Byaidu    时间: 2017-6-12 20:09

很厉害的样子,但只能注入一个DLL,不方便以后扩展
作者: bailong360    时间: 2017-6-12 20:34

本帖最后由 bailong360 于 2017-6-13 16:37 编辑

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

再提一个建议:如楼上所说只能加载一个dll局限性很大,除非再把ext.dll做成一个加载器.然而这样不如让cmdex提供一个load命令,用来加载其他dll.这样不同开发者开发的dll就可以同时使用
------------------------
测试了一下 发现有一个bug
  1. @echo off
  2. set a=%s%
  3. my%__%cmd %a%
  4. pause
复制代码
这样一段代码,执行后输出的是
  1. cmd:mycmd
  2. lastcmdline:my%__%cmd %a%
  3. 请按任意键继续. . .
复制代码
%__%和%a%没有被扩展,不过"cmd:mycmd"倒是得到了正确结果 望改进
作者: misaki    时间: 2017-6-14 21:52

回复 6# Byaidu

我上面提到可以把扩展dll做成插件管理器,这样就可以无限制扩展了,比较懒所以没有自己做
作者: misaki    时间: 2017-6-14 22:00

回复 7# bailong360


    cmdline就只是上一条命令而已,并不是扩展环境后的,因为bat的命令不是按行来算的,比如复合命令和转义换行,所以为了灵活性就直接暴露出来而不是我来处理
    某种意义上这样可以写出不属于bat的语法,比如带括号的函数,等号赋值等.

    实现的本质还是hook api,hook了wcsicmp和wcschr

    ps:好久没在bat吧看见你了,我贴吧id是gzhmbk
作者: bailong360    时间: 2017-6-15 12:10

本帖最后由 bailong360 于 2017-6-15 12:18 编辑

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

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

交给开发者来处理命令行感觉有利有弊吧
作者: misaki    时间: 2017-6-16 22:12

回复 10# bailong360

soga,因为bat效率实在太低了,所以这点hook影响不大

确实有利有弊,不过还是觉得从程序本体抽离出来更好,方便改动
作者: bailong360    时间: 2017-6-16 22:16

本帖最后由 bailong360 于 2017-6-16 22:21 编辑

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

试着编写了一个AddCmd  tcc编译通过
将变量扩展和参数都处理了一下,应该是个通常情况下比较友好的接口
  1. #include <stdio.h>
  2. #include <windows.h>
  3. #include <shellapi.h>
  4. #define DLL_EXPORT __declspec(dllexport)
  5. #define HANDLED     0
  6. #define UNHANDLED   1
  7. #define MAX_CMD_LENGTH 512
  8. typedef struct cmdTable {
  9.     wchar_t cmd[MAX_CMD_LENGTH];
  10.     int     (*func)(int, wchar_t *[]);
  11.     struct  cmdTable *next;
  12. } CMDTABLE;
  13. DLL_EXPORT int ExtCallBack(wchar_t *, wchar_t *);
  14. char* WcharToChar(wchar_t *wstr);
  15. int AddCmd(int argc, wchar_t *cmdline[]);
  16. CMDTABLE cmd_Load = {L"AddCmd", AddCmd, NULL};
  17. CMDTABLE *head = &cmd_Load;
  18. CMDTABLE *tail = &cmd_Load;
  19. DLL_EXPORT int ExtCallBack(wchar_t *cmd, wchar_t *cmdline)
  20. {
  21.     CMDTABLE *current = head;
  22.     do {
  23.         if (!wcsnicmp(cmd, current->cmd, wcslen(current->cmd))) {
  24.             int       argc;
  25.             wchar_t **argv;
  26.             wchar_t newcmdline[MAX_CMD_LENGTH];
  27.             ExpandEnvironmentStringsW(cmdline, newcmdline, MAX_CMD_LENGTH + 1);
  28.             newcmdline[wcslen(newcmdline) - 2] = '\0'; //去除末尾的\n
  29.             argv = CommandLineToArgvW(newcmdline, &argc);
  30.             current->func(argc, argv);
  31.             LocalFree(argv);
  32.             return HANDLED;
  33.         }
  34.         current = current -> next;
  35.     } while (current != NULL);
  36.     return UNHANDLED;
  37. }
  38. char* WcharToChar(wchar_t *wstr)
  39. {
  40.     int len   = WideCharToMultiByte(CP_ACP, 0, wstr, -1, NULL, 0, NULL, NULL);
  41.     char *str = (char *)malloc(len * sizeof(char));
  42.     WideCharToMultiByte(CP_ACP, 0, wstr, -1, str, len, NULL, NULL);
  43.     return str;
  44. }
  45. int AddCmd(int argc, wchar_t *argv[])
  46. {
  47.     if (argc < 4) {
  48.         puts("Usage: AddCmd YourCmd YourDll YourFunc");
  49.         puts("**YourFunc should be defined like this \"int YourFunc(int argc, wchar_t *argv[])\"");
  50.         return 0;
  51.     }
  52.     HMODULE hDLL   = LoadLibraryW(argv[2]);
  53.     char *funcName = WcharToChar(argv[3]);
  54.     void *func     = (void *)GetProcAddress(hDLL, funcName);
  55.    
  56.     if (hDLL == NULL) {
  57.         fprintf(stderr, "ERROR:can't load dll \"%S\"", argv[2]);
  58.         free(funcName);
  59.         return 0;
  60.     } else if (func == NULL) {
  61.         fprintf(stderr, "ERROR:can't load func \"%s\"", funcName);
  62.         free(funcName);
  63.         return 0;
  64.     }
  65.     tail->next = (CMDTABLE *)malloc(sizeof(CMDTABLE));
  66.     wcscpy(tail->next->cmd, argv[1]);
  67.     tail->next->func = (void *)GetProcAddress(hDLL, funcName);
  68.     tail->next->next = NULL;
  69.     tail = tail->next;
  70.     free(funcName);
  71.     return 0;
  72. }
复制代码

作者: misaki    时间: 2017-6-17 19:51

回复 12# bailong360

试了xpsp3和8.1的,所以8.1之前应该没问题,win10没试过
作者: bailong360    时间: 2017-6-17 21:21

回复 13# misaki
刚刚又想到了一个问题
ExpanEnvironmentStringsW只能扩展以%%包裹的变量,!!好像不好处理- -
作者: misaki    时间: 2017-6-18 12:23

回复 14# bailong360


   发帖的时候忘记说了,这个和cmd有点不同,如果自己处理命令行的话,%%也是变量延迟扩展的效果,实现可以参考nt4的源码吧




欢迎光临 批处理之家 (http://bbs.bathome.net/) Powered by Discuz! 7.2