514 lines
16 KiB
C++
514 lines
16 KiB
C++
|
#include "SH_Datatrans.hpp"
|
|||
|
|
#include "dirent.h"
|
|||
|
|
#include <stdio.h>
|
|||
|
|
|
|||
|
|
|
|||
|
|
DataTrans *pDataTrans = DataTrans::instance();
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
struct DowloadFile {
|
|||
|
|
|
|||
|
|
const char *filename;
|
|||
|
|
FILE *stream;
|
|||
|
|
|
|||
|
|
};
|
|||
|
|
DataTrans::DataTrans():
|
|||
|
|
m_bDebug(false)
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
DataTrans::~DataTrans()
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
static int OnDebug(CURL *, curl_infotype itype, char * pData, size_t size, void *)
|
|||
|
|
{
|
|||
|
|
if(itype == CURLINFO_TEXT) {
|
|||
|
|
//printf("[TEXT]%s\n", pData);
|
|||
|
|
} else if(itype == CURLINFO_HEADER_IN) {
|
|||
|
|
print_info("[HEADER_IN]%s\n", pData);
|
|||
|
|
} else if(itype == CURLINFO_HEADER_OUT) {
|
|||
|
|
print_info("[HEADER_OUT]%s\n", pData);
|
|||
|
|
} else if(itype == CURLINFO_DATA_IN) {
|
|||
|
|
print_info("[DATA_IN]%s\n", pData);
|
|||
|
|
} else if(itype == CURLINFO_DATA_OUT) {
|
|||
|
|
print_info("[DATA_OUT]%s\n", pData);
|
|||
|
|
}
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
static size_t OnWriteData(void* buffer, size_t size, size_t nmemb, void* lpVoid)
|
|||
|
|
{
|
|||
|
|
std::string* str = dynamic_cast<std::string*>((std::string *)lpVoid);
|
|||
|
|
if( NULL == str || NULL == buffer ) {
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
char* pData = (char*)buffer;
|
|||
|
|
str->append(pData, size * nmemb);
|
|||
|
|
return nmemb;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
static size_t my_fwrite(void *buffer, size_t size, size_t nmemb, void *stream)
|
|||
|
|
{
|
|||
|
|
struct DowloadFile *out=(struct DowloadFile *)stream;
|
|||
|
|
if(out && !out->stream) {
|
|||
|
|
out->stream=fopen(out->filename, "wb");//打开文件进行写入
|
|||
|
|
if(!out->stream)
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
return fwrite(buffer, size, nmemb, out->stream);
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int DataTrans::download(char* pFilename,string& strUrl,string& strResponse,bool bDownload)
|
|||
|
|
{
|
|||
|
|
CURL *curl = NULL;
|
|||
|
|
CURLcode res;
|
|||
|
|
struct DowloadFile dlfile={
|
|||
|
|
pFilename, //定义下载到本地的文件位置和路径
|
|||
|
|
NULL
|
|||
|
|
};
|
|||
|
|
curl_global_init(CURL_GLOBAL_DEFAULT);
|
|||
|
|
curl = curl_easy_init(); //初始化一个curl指针
|
|||
|
|
if(curl) { //curl对象存在的情况下执行的操作
|
|||
|
|
//设置远端地址
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_URL,strUrl.c_str());
|
|||
|
|
if(bDownload){
|
|||
|
|
//执行写入文件流操作
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, my_fwrite);//当有数据被写入，回调函数被调用，
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &dlfile); //设置结构体的指针传递给回调函数
|
|||
|
|
}else{
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_READFUNCTION, NULL);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, OnWriteData);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&strResponse);
|
|||
|
|
}
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 8);//连接超时，这个数值如果设置太短可能导致数据请求不到就断开了
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 10);//接收数据时超时设置，如果10秒内数据未接收完，直接退出
|
|||
|
|
//启用时会汇报所有的信息，存放在STDERR或指定的CURLOPT_STDERR中
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_VERBOSE, 1L);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_USERPWD, "SUREN:SUREN");
|
|||
|
|
|
|||
|
|
//写入文件
|
|||
|
|
res = curl_easy_perform(curl);
|
|||
|
|
//释放curl对象
|
|||
|
|
curl_easy_cleanup(curl);
|
|||
|
|
if(res != CURLE_OK)
|
|||
|
|
{
|
|||
|
|
cout<<stderr<<res<<endl;
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
printf("strResponse = %s\n",strResponse.c_str());
|
|||
|
|
if(bDownload){
|
|||
|
|
if(dlfile.stream)
|
|||
|
|
{
|
|||
|
|
//关闭文件流
|
|||
|
|
fclose(dlfile.stream);
|
|||
|
|
}
|
|||
|
|
}else{
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
//释放全局curl对象
|
|||
|
|
curl_global_cleanup();
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int DataTrans::Post(const std::string & strUrl, const std::string & strPost, std::string & strResponse)
|
|||
|
|
{
|
|||
|
|
CURLcode res;
|
|||
|
|
CURL* curl = curl_easy_init();
|
|||
|
|
if(NULL == curl) {
|
|||
|
|
return CURLE_FAILED_INIT;
|
|||
|
|
}
|
|||
|
|
if(m_bDebug)
|
|||
|
|
{
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_VERBOSE, 1);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_DEBUGFUNCTION, OnDebug);
|
|||
|
|
}
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_URL, strUrl.c_str());
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_POST, 1);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, strPost.c_str());
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_READFUNCTION, NULL);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, OnWriteData);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&strResponse);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 3);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 3);
|
|||
|
|
res = curl_easy_perform(curl);
|
|||
|
|
curl_easy_cleanup(curl);
|
|||
|
|
return res;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
int DataTrans::Get(const std::string & strUrl, std::string & strResponse)
|
|||
|
|
{
|
|||
|
|
CURLcode res;
|
|||
|
|
CURL* curl = curl_easy_init();
|
|||
|
|
if(NULL == curl) {
|
|||
|
|
return CURLE_FAILED_INIT;
|
|||
|
|
}
|
|||
|
|
if(m_bDebug) {
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_VERBOSE, 1);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_DEBUGFUNCTION, OnDebug);
|
|||
|
|
}
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_URL, strUrl.c_str());
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_READFUNCTION, NULL);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, OnWriteData);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&strResponse);
|
|||
|
|
/**
|
|||
|
|
* 当多个线程都使用超时处理的时候，同时主线程中有sleep或是wait等操作。
|
|||
|
|
* 如果不设置这个选项，libcurl将会发信号打断这个wait从而导致程序退出。
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 3);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 3);
|
|||
|
|
res = curl_easy_perform(curl);
|
|||
|
|
curl_easy_cleanup(curl);
|
|||
|
|
return res;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
int DataTrans::Posts(const std::string & strUrl, const std::string & strPost, std::string & strResponse, const char * pCaPath)
|
|||
|
|
{
|
|||
|
|
CURLcode res;
|
|||
|
|
CURL* curl = curl_easy_init();
|
|||
|
|
if(NULL == curl) {
|
|||
|
|
return CURLE_FAILED_INIT;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if(m_bDebug) {
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_VERBOSE, 1);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_DEBUGFUNCTION, OnDebug);
|
|||
|
|
}
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_URL, strUrl.c_str());
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_POST, 1);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, strPost.c_str());
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_READFUNCTION, NULL);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, OnWriteData);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&strResponse);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1);
|
|||
|
|
if(NULL == pCaPath) {
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, false);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, false);
|
|||
|
|
} else {
|
|||
|
|
//缺省情况就是PEM，所以无需设置，另外支持DER
|
|||
|
|
//curl_easy_setopt(curl,CURLOPT_SSLCERTTYPE,"PEM");
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, true);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_CAINFO, pCaPath);
|
|||
|
|
}
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 3);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 3);
|
|||
|
|
res = curl_easy_perform(curl);
|
|||
|
|
curl_easy_cleanup(curl);
|
|||
|
|
return res;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
int DataTrans::Gets(const std::string & strUrl, std::string & strResponse, const char * pCaPath)
|
|||
|
|
{
|
|||
|
|
CURLcode res;
|
|||
|
|
CURL* curl = curl_easy_init();
|
|||
|
|
if(NULL == curl) {
|
|||
|
|
return CURLE_FAILED_INIT;
|
|||
|
|
}
|
|||
|
|
if(m_bDebug) {
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_VERBOSE, 1);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_DEBUGFUNCTION, OnDebug);
|
|||
|
|
}
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_URL, strUrl.c_str());
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_READFUNCTION, NULL);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, OnWriteData);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&strResponse);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1);
|
|||
|
|
if(NULL == pCaPath) {
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, false);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, false);
|
|||
|
|
} else {
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, true);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_CAINFO, pCaPath);
|
|||
|
|
}
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 3);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 3);
|
|||
|
|
res = curl_easy_perform(curl);
|
|||
|
|
curl_easy_cleanup(curl);
|
|||
|
|
return res;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
int DataTrans::upload_file(const std::string &strUrl, const std::string &filename, std::string &response)
|
|||
|
|
{
|
|||
|
|
CURL *curl;
|
|||
|
|
CURLM *multi_handle;
|
|||
|
|
int still_running;
|
|||
|
|
char res[60];
|
|||
|
|
struct curl_httppost *formpost = NULL;
|
|||
|
|
struct curl_httppost *lastptr = NULL;
|
|||
|
|
struct curl_slist *headerlist = NULL;
|
|||
|
|
static const char buf[] = "Expect:";
|
|||
|
|
curl_global_init(CURL_GLOBAL_ALL);
|
|||
|
|
/* Fill in the file upload field. This makes libcurl load data from
|
|||
|
|
the given file name when curl_easy_perform() is called. */
|
|||
|
|
|
|||
|
|
/* Fill in the filename field */
|
|||
|
|
curl_formadd(&formpost, &lastptr, CURLFORM_COPYNAME, "filename", CURLFORM_FILE, filename.c_str(), CURLFORM_END);
|
|||
|
|
curl_formadd(&formpost, &lastptr, CURLFORM_COPYNAME, "submit", CURLFORM_COPYCONTENTS, "upload", CURLFORM_END);
|
|||
|
|
|
|||
|
|
curl = curl_easy_init();
|
|||
|
|
multi_handle = curl_multi_init();
|
|||
|
|
/* initalize custom header list (stating that Expect: 100-continue is not wanted */
|
|||
|
|
headerlist = curl_slist_append(headerlist, buf);
|
|||
|
|
if (curl && multi_handle) {
|
|||
|
|
/* what URL that receives this POST */
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_URL, strUrl.c_str());
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_HTTPAUTH, CURLAUTH_DIGEST);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_VERBOSE, 1L);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, res);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headerlist);
|
|||
|
|
curl_easy_setopt(curl, CURLOPT_HTTPPOST, formpost);
|
|||
|
|
curl_multi_add_handle(multi_handle, curl);
|
|||
|
|
curl_multi_perform(multi_handle, &still_running);
|
|||
|
|
do {
|
|||
|
|
struct timeval timeout;
|
|||
|
|
int rc;
|
|||
|
|
fd_set fdread;
|
|||
|
|
fd_set fdwrite;
|
|||
|
|
fd_set fdexcep;
|
|||
|
|
int maxfd = -1;
|
|||
|
|
long curl_timeo = -1;
|
|||
|
|
FD_ZERO(&fdread);
|
|||
|
|
FD_ZERO(&fdwrite);
|
|||
|
|
FD_ZERO(&fdexcep);
|
|||
|
|
/* set a suitable timeout to play around with */
|
|||
|
|
timeout.tv_sec = 1;
|
|||
|
|
timeout.tv_usec = 0;
|
|||
|
|
curl_multi_timeout(multi_handle, &curl_timeo);
|
|||
|
|
if (curl_timeo >= 0) {
|
|||
|
|
timeout.tv_sec = curl_timeo / 1000;
|
|||
|
|
if (timeout.tv_sec > 1)
|
|||
|
|
timeout.tv_sec = 1;
|
|||
|
|
else
|
|||
|
|
timeout.tv_usec = (curl_timeo % 1000) * 1000;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/* get file descriptors from the transfers */
|
|||
|
|
curl_multi_fdset(multi_handle, &fdread, &fdwrite, &fdexcep, &maxfd);
|
|||
|
|
/* In a real-world program you OF COURSE check the return code of the
|
|||
|
|
function calls. On success, the value of maxfd is guaranteed to be
|
|||
|
|
greater or equal than -1. We call select(maxfd + 1, ...), specially in
|
|||
|
|
case of (maxfd == -1), we call select(0, ...), which is basically equal
|
|||
|
|
to sleep. */
|
|||
|
|
rc = select(maxfd + 1, &fdread, &fdwrite, &fdexcep, &timeout);
|
|||
|
|
switch (rc) {
|
|||
|
|
case -1:
|
|||
|
|
break;
|
|||
|
|
case 0:
|
|||
|
|
default:
|
|||
|
|
/* timeout or readable/writable sockets */
|
|||
|
|
print_info("perform!\n");
|
|||
|
|
curl_multi_perform(multi_handle, &still_running);
|
|||
|
|
print_info("running: %d!\n", still_running);
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
} while (still_running);
|
|||
|
|
curl_multi_cleanup(multi_handle);
|
|||
|
|
/* always cleanup */
|
|||
|
|
curl_easy_cleanup(curl);
|
|||
|
|
/* then cleanup the formpost chain */
|
|||
|
|
curl_formfree(formpost);
|
|||
|
|
/* free slist */
|
|||
|
|
curl_slist_free_all(headerlist);
|
|||
|
|
}
|
|||
|
|
return 0;
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// int DataTrans::Send_file_socket(const std::string &filename)
|
|||
|
|
// {
|
|||
|
|
// int socketfd;
|
|||
|
|
// struct sockaddr_in s_add,c_add;
|
|||
|
|
// unsigned short portnum = 9092;
|
|||
|
|
// int len;
|
|||
|
|
// char buffer[4096];
|
|||
|
|
// FILE *fp ;
|
|||
|
|
// int file_block_length = 0;
|
|||
|
|
// /*
|
|||
|
|
// *创建socket
|
|||
|
|
// */
|
|||
|
|
// if((socketfd = socket(AF_INET,SOCK_STREAM,0)) < 0)
|
|||
|
|
// {
|
|||
|
|
// print_error("Socket create error! \n");
|
|||
|
|
// return -1;
|
|||
|
|
// }
|
|||
|
|
// /*
|
|||
|
|
// *设置地址
|
|||
|
|
// */
|
|||
|
|
|
|||
|
|
// bzero(&s_add,sizeof(struct sockaddr_in));
|
|||
|
|
// s_add.sin_family = AF_INET;
|
|||
|
|
// //s_add.sin_addr.s_addr = inet_addr(GlobalConfig::ServerIP.c_str());
|
|||
|
|
// s_add.sin_addr.s_addr = inet_addr("192.168.1.19");
|
|||
|
|
// s_add.sin_port = htons(portnum);
|
|||
|
|
|
|||
|
|
|
|||
|
|
// if (connect(socketfd,(struct sockaddr *)(&s_add),sizeof(struct sockaddr)) < 0)
|
|||
|
|
// {
|
|||
|
|
// print_error("Connect failure!\n");
|
|||
|
|
// return -1;
|
|||
|
|
// }
|
|||
|
|
// else
|
|||
|
|
// print_info("Connect Success!\n");
|
|||
|
|
// std::string filename = filename.substr();
|
|||
|
|
// int file_name_length = (int)strlen(filename.c_str());
|
|||
|
|
// if(send(socketfd,(char *)&file_name_length, sizeof(int), 0) < 0)
|
|||
|
|
// {
|
|||
|
|
// print_error("Send File_Name_Length Failed!\n");
|
|||
|
|
// }else
|
|||
|
|
// {
|
|||
|
|
// print_info("Send File_Name_Length Success!\n");
|
|||
|
|
// }
|
|||
|
|
// //发送文件名
|
|||
|
|
// if(send(socketfd,filename.c_str(),file_name_length,0) < 0)
|
|||
|
|
// {
|
|||
|
|
// print_error("Send File_Name Failed!\n");
|
|||
|
|
// }else
|
|||
|
|
// {
|
|||
|
|
// print_info("Send File_Name Success!\n");
|
|||
|
|
// }
|
|||
|
|
// /*
|
|||
|
|
// *发送文件
|
|||
|
|
// */
|
|||
|
|
// fp = fopen(filename.c_str(), "r");
|
|||
|
|
|
|||
|
|
// if (fp == NULL)
|
|||
|
|
// {
|
|||
|
|
// print_error("File: %s Not Found!\n", filename.c_str());
|
|||
|
|
// }
|
|||
|
|
// else
|
|||
|
|
// {
|
|||
|
|
// bzero(buffer, BUFFER_SIZE);
|
|||
|
|
// while( (file_block_length = fread(buffer, sizeof(char), BUFFER_SIZE, fp)) > 0)
|
|||
|
|
// {
|
|||
|
|
// print_info("file_block_length = %d\n", file_block_length);
|
|||
|
|
// /*
|
|||
|
|
// *把数据写入buffer
|
|||
|
|
// */
|
|||
|
|
// if (send(socketfd, buffer, file_block_length, 0) < 0)
|
|||
|
|
// {
|
|||
|
|
// print_error("Send File:%s Failed!\n", filename.c_str());
|
|||
|
|
// break;
|
|||
|
|
// }
|
|||
|
|
// bzero(buffer, sizeof(buffer));
|
|||
|
|
// }
|
|||
|
|
// fclose(fp);
|
|||
|
|
// print_info("File: %s Transfer Finished!\n", filename.c_str());
|
|||
|
|
// }
|
|||
|
|
// close(socketfd);
|
|||
|
|
// return 0;
|
|||
|
|
// }
|
|||
|
|
|
|||
|
|
int DataTrans::Send_file_socket(const std::string &filename)
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int DataTrans::Send_Dir_socket(const char *dirname)
|
|||
|
|
{
|
|||
|
|
DIR *dirp;
|
|||
|
|
std::vector<string> v;
|
|||
|
|
std::string temp;
|
|||
|
|
struct dirent *dp;
|
|||
|
|
dirp = opendir(dirname);
|
|||
|
|
while ((dp = readdir(dirp)) != NULL) {
|
|||
|
|
if (strcmp(dp->d_name, ".") == 0 || strcmp(dp->d_name, "..") == 0)
|
|||
|
|
continue;
|
|||
|
|
v.push_back(std::string(dp->d_name));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int socketfd = socket(AF_INET,SOCK_STREAM,0);
|
|||
|
|
struct sockaddr_in s_add,c_add;
|
|||
|
|
unsigned short portnum = 12345;
|
|||
|
|
int len;
|
|||
|
|
char buffer[BUFFER_SIZE];
|
|||
|
|
FILE *fp ;
|
|||
|
|
int file_block_length = 0;
|
|||
|
|
/*
|
|||
|
|
*创建socket
|
|||
|
|
*/
|
|||
|
|
if(socketfd < 0) {
|
|||
|
|
print_error("Socket create error! \n");
|
|||
|
|
return -1;
|
|||
|
|
} else {
|
|||
|
|
print_info("Socket create success!\n");
|
|||
|
|
}
|
|||
|
|
/*
|
|||
|
|
*设置地址
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
bzero(&s_add,sizeof(struct sockaddr_in));
|
|||
|
|
s_add.sin_family = AF_INET;
|
|||
|
|
s_add.sin_addr.s_addr = inet_addr(GlobalConfig::ServerIP.c_str());
|
|||
|
|
s_add.sin_port = htons(PORT);
|
|||
|
|
|
|||
|
|
if (connect(socketfd,(struct sockaddr *)(&s_add),sizeof(struct sockaddr)) < 0) {
|
|||
|
|
print_error("Connect failure!\n");
|
|||
|
|
return -1;
|
|||
|
|
} else {
|
|||
|
|
print_info("Connect Success!\n");
|
|||
|
|
}
|
|||
|
|
std::vector<std::string>::iterator iter;
|
|||
|
|
for(iter = v.begin();iter != v.end();++iter) {
|
|||
|
|
temp = *iter;
|
|||
|
|
cout << temp.c_str() <<endl;
|
|||
|
|
int file_name_length = (int)strlen(temp.c_str());
|
|||
|
|
if (send(socketfd, (char *)&file_name_length, sizeof(int), 0) < 0) {//发送文件名长度
|
|||
|
|
|
|||
|
|
print_error("Send File_Name_Length Failed!\n");
|
|||
|
|
} else {
|
|||
|
|
print_info("Send File_Name_Length Success!\n");
|
|||
|
|
}
|
|||
|
|
//发送文件名
|
|||
|
|
if(send(socketfd,temp.c_str(),file_name_length,0) < 0) {
|
|||
|
|
print_error("Send File_Name Failed!\n");
|
|||
|
|
} else {
|
|||
|
|
print_info("Send File_Name Success!\n");
|
|||
|
|
}
|
|||
|
|
/*
|
|||
|
|
*发送文件
|
|||
|
|
*/
|
|||
|
|
std::string filename = std::string(dirname + temp);
|
|||
|
|
fp = fopen(filename.c_str(), "r");
|
|||
|
|
if (fp == NULL) {
|
|||
|
|
print_error("File: %s Not Found!\n", filename.c_str());
|
|||
|
|
} else {
|
|||
|
|
bzero(buffer, BUFFER_SIZE);
|
|||
|
|
while( (file_block_length = fread(buffer, sizeof(char), BUFFER_SIZE, fp)) > 0) {
|
|||
|
|
print_info("file_block_length = %d\n", file_block_length);
|
|||
|
|
/*
|
|||
|
|
*把数据写入buffer
|
|||
|
|
*/
|
|||
|
|
if (send(socketfd, buffer, file_block_length, 0) < 0) {
|
|||
|
|
print_error("Send File:%s Failed!\n", filename.c_str());
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
bzero(buffer, sizeof(buffer));
|
|||
|
|
}
|
|||
|
|
fclose(fp);
|
|||
|
|
print_info("File: %s Transfer Finished!\n", filename.c_str());
|
|||
|
|
}
|
|||
|
|
close(socketfd);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
void DataTrans::SetDebug(bool bDebug)
|
|||
|
|
{
|
|||
|
|
m_bDebug = bDebug;
|
|||
|
|
}
|