在今天我们来实现那篇技术文章中的CMD工作集以及我的一点调错心得。
先介绍一下API
BOOL WINAPI ReadFile(
_In_ HANDLE hFile, // 读取的文件句柄
_Out_ LPVOID lpBuffer, // 保存读取缓冲字符数组
_In_ DWORD nNumberOfBytesToRead, // 缓冲数组的大小
_Out_opt_ LPDWORD lpNumberOfBytesRead, // 实际读出的大小
_Inout_opt_ LPOVERLAPPED lpOverlapped // 异步IO文件结构体
);
下面贴上源程序(c++):
#include <iostream>
#include <windows.h>
#include <string.h>
using namespace std;
void Jud_com(char *);
void welcome();
int main()
{
char comlist[MAX_PATH];
DWORD comlen;
HANDLE hinstream;
hinstream = GetStdHandle(STD_INPUT_HANDLE);
welcome();
while (true)
{
memset(comlist, 0, MAX_PATH);
cout <<"root@SKYF0cker >> #";
ReadFile(
hinstream,
comlist,
MAX_PATH,
&comlen,
NULL
);
cout <<"接收到命令"<<comlist<<endl;
Jud_com(comlist);
}
}
void welcome()
{
cout <<"SKYF0cker's CMD Tool [版本 0.0.1]n" << endl;
cout <<"学习自制 (c) http://blog.csdn.net/lihao008 欢迎交流"<< endl;
}
void Jud_com(char *x)
{
//char *temp = new char();
//temp = { 0 };
char mmp[MAX_PATH] = {0};
int i=0;
while (x[i]!= '\n'&&x[i]!= '\r')
{
mmp[i] = x[i];
i++;
}
if (strcmp(mmp,"fuckyou")==0)
{
cout <<"fuckyou too,bro"<< endl;
}
if ((strcmp(mmp, "exit")) == 0)
{
exit(0);
}
}
顺便还定制了一两个功能
运行结果:
运行成功
调错:
在最开始的时候,我在函数void Jud_com(char *x)中通过new定义了一个char *来储存字符串,但是后来发现字符串输出有了问题,定义但没有赋值的内存中,系统自动给他们赋了值,导致在strcmp时,不能正确判断。所以在最开始就要给数组一个值,比如:char mmp[MAX_PATH] = {0};这样就解决了。