Source code:
/*----------------------------------------------------------------
// Copyright (C) 2008 ShanDong University At Weihai
// FileName：Struct_type.cpp
// Function Description：this program can imitate search virus from the given folder
// Created By：Alwin Xu
// Created Date: 2009-2-4 ~ 2009-2-5
// Modified By：
// Modified Description：
// Modified Date:
//----------------------------------------------------------------*/
#include "stdafx.h"
#include <iostream>
#include <windows.h>
#include <string.h>
#include <fstream>
using namespace std;
/////////////////////////////////
char virusArr[10][10];
static int count=0;
void initArr();
void printVirus(string strsrc,char* filename);
void searchVirus(char *filename);
void SearchAllFiles(char foldername[]);
bool CheckName(char *isFolder);
char* CutPath(char *foldername);
char* CutPath(char *foldername,int number);
//////////main()///////////////////
int _tmain(int argc, _TCHAR* argv[])
{
char check;
int flag = 1;
while(flag)
{
if(count<10)
{
cout<<"add a new virus:"<<endl;//add new virus.
cin>>virusArr[count];
count++;
cout<<"add virus again(Y or N):"<<endl;
cin>>check;
if('Y'==check || 'y'==check)
flag = 1;
else
flag = 0;
}
}
char foldername[20];
cout<<"enter you foldername(10): ";
cin>>foldername;
SearchAllFiles(foldername);
return 0;
}
//////////initialize virusArr./////////////////
void initArr()
{
memset(virusArr,0,100);
}
////////////check folder//////////////////////////////////////
bool CheckName(char *isFolder)//check if the name is a folder's name
{
bool sign =false;
int len = strlen(isFolder);
int count=0;
while(*isFolder++ != '.' && count++ <len);
if(count < len-1)
{
sign = true;
}
return sign;
}
//////////cut the last """/////////////////////////////////////////////
char* CutPath(char *foldername)
{
char *temp = foldername;
int len = strlen(foldername);
for(int i=len;i>0;i--)
{
if(foldername[i]=='""')
{
foldername[i+1]='"0';
i=0;
}
}
return temp;
}
/////////cut path according number///////////////////////////////////
char* CutPath(char *foldername,int number)
{
char *temp = foldername;
int len = strlen(foldername);
int count = 0;
for(int i=len;i>0;i--)
{
if(foldername[i]=='""')
{
count++;
}
if(number == count)
{
foldername[i+1]='"0';
i = 0;
}
}
return temp;
}
//////////////search virus from a file//////////////////////
void searchVirus(char *filename)
{
if(!filename)
{
exit(0);
}
ifstream isopen(filename);//open file
if(isopen.fail())
{
cout<<"the file may be not exsit,please check it."<<endl;
exit(0);
}
else
{
char ch;
string filecontent="";
while(!isopen.eof())
{
isopen.get(ch);
filecontent += ch;
}
printVirus(filecontent,filename);//print virus we have found.
filecontent="";
isopen.close();
}
}
///////////////print////////////////////////
void printVirus(string strsrc,char* filename)
{
cout<<"file path: "<<filename<<endl;
bool haveOrNot = false;
for(int i=0;i<count;i++)
{
int length = strsrc.length();
if(strsrc.find(virusArr[i])<length)
{
printf(""a");
cout<<"find virus: "<<virusArr[i]<<endl;
haveOrNot = true;
}
}
if(false == haveOrNot)
{
cout<<"no virus!"<<endl;
}
}
/////////////search all files from a given folder./////////////////////////
void SearchAllFiles(char foldername[])
{
WIN32_FIND_DATA FindFileData;
HANDLE hFind = INVALID_HANDLE_VALUE;
char DirSpec[MAX_PATH]; // directory specification
DWORD dwError;
char temp[20];
int sign =0;
strcpy(temp,foldername);
strcat(foldername,"""*.*");
hFind = FindFirstFile(foldername, &FindFileData);
if (hFind == INVALID_HANDLE_VALUE)
{
printf ("Invalid file handle. Error is %u"n", GetLastError());
}
else
{
std::wcout<<FindFileData.cFileName<<endl;
FindNextFile(hFind, &FindFileData);
while (FindNextFile(hFind, &FindFileData) != 0)
{
char transfer[15]={0};
strcpy(transfer,FindFileData.cFileName);
if(CheckName(transfer))
{
if(sign == 0)
{
strncat(temp,"""",2);
strncat(temp,transfer,15);
searchVirus(temp);
sign = 1;
}
else
{
CutPath(temp);
strncat(temp,transfer,15);
searchVirus(temp);
}
}
else
{
if(sign == 0)
{
strncat(temp,"""",2);
strncat(temp,transfer,15);
SearchAllFiles(temp);
CutPath(temp,2);
sign = 1;
}
else
{
CutPath(temp);
strncat(temp,transfer,15);
SearchAllFiles(temp);
CutPath(temp,2);
}
}
}
dwError = GetLastError();
FindClose(hFind);
if (dwError != ERROR_NO_MORE_FILES)
{
printf ("FindNextFile error. Error is %u"n", dwError);
}
}
}