1 #pragma once
  2 #include <stdio.h>
  3 #include <stdlib.h>
  4 #include <string.h>
  5 #include <unistd.h>
  6 #include <pthread.h>
  7 #include <time.h>
  8 #include <sys/stat.h>
  9 #include <signal.h>
 10 
 11 #include <curl/curl.h>
 12 #include <curl/easy.h>
 13 
 14 #include <string.h>
 15 #include <fstream>
 16 #include <iostream>
 17 
 18 /*
 19 g++ curl_test.cpp -I /usr/local/curl/include/ -I /usr/local/libssh2/include/ -I /usr/local/openssl/include/  -L /usr/local/curl/lib/ -L /usr/local/libssh2/lib/ -L /usr/local/openssl/lib/ -lrt -lcurl -lssh2 -lssl -lcrypto -ldl -lz
 20 */
 21 
 22 using namespace std;
 23 
 24 void gloale_init()
 25 {
 26     
 27     curl_global_init(CURL_GLOBAL_DEFAULT);
 28     return;
 29 }
 30 
 31 size_t read_callback(void *ptr, size_t size, size_t nmemb, void *stream) //回调函数
 32 {
 33     curl_off_t nread;
 34     size_t retcode = fread(ptr, size, nmemb, (FILE*)(stream));
 35     nread = (curl_off_t)retcode;
 36     return retcode;
 37 }
 38 
 39 
 40 size_t write_data(void *ptr, size_t size, size_t nmemb, void *stream)
 41 {
 42     int written = fwrite(ptr, size, nmemb, (FILE *)stream);
 43     return written;
 44 }
 45 
 46 size_t upload(const char * user,const char * passwd,const char * url, const char * path)
 47 {    
 48     CURL *curl;
 49     CURLcode res;
 50     string s1(user);
 51     string s2(passwd);
 52     string s3 = s1 + ":" + s2;
 53     cout<<s3<<endl;    
 54     // system("ls write_file");
 55     FILE* pSendFile = fopen(path,"r");
 56     if(pSendFile == NULL)
 57     {
 58         printf("open failed\n");
 59         return 1;
 60     }
 61 
 62     fseek(pSendFile, 0L, SEEK_END);
 63 
 64     size_t iFileSize = ftell(pSendFile);
 65 
 66     fseek(pSendFile, 0L, SEEK_SET);
 67     printf("begin easy_init\n"); 
 68  
 69     curl = curl_easy_init();
 70     printf("curl_easy_init success\n");
 71     if(curl) {
 72         curl_easy_setopt(curl, CURLOPT_URL,url);
 73     curl_easy_setopt(curl, CURLOPT_USERPWD,s3.c_str());  
 74     curl_easy_setopt(curl, CURLOPT_READFUNCTION, read_callback);
 75     curl_easy_setopt(curl, CURLOPT_READDATA, pSendFile);
 76     curl_easy_setopt(curl, CURLOPT_FTP_CREATE_MISSING_DIRS, 0);
 77     curl_easy_setopt(curl, CURLOPT_UPLOAD, 1);
 78     curl_easy_setopt(curl, CURLOPT_INFILESIZE, iFileSize);
 79  
 80     printf("curl_easy_setopt success");
 81     res = curl_easy_perform(curl);
 82  
 83     curl_easy_cleanup(curl);
 84  
 85     if(CURLE_OK != res) 
 86     {
 87  
 88         fprintf(stdout, "curl told us %d\n", res);
 89     }
 90   }
 91     fclose(pSendFile);
 92     curl_global_cleanup();
 93     return 0;
 94 }
 95 
 96 
 97 int download(const char * user,const char * passw,const char * url,const char * filePath)
 98 {
 99     CURL *curl;
100     CURLcode curl_code;
101     string s1(user);
102     string s2(passw);
103     string s3 = s1 + ":" + s2;
104     cout<<s3<<endl;    
105     curl = curl_easy_init();
106     curl_easy_setopt(curl, CURLOPT_URL, url);
107 //    curl_easy_setopt(curl, CURLOPT_USERPWD, s3.c_str());
108     curl_easy_setopt(curl, CURLOPT_USERNAME, user);
109     curl_easy_setopt(curl, CURLOPT_PASSWORD, passw);
110     
111     FILE *fp = fopen(filePath, "wb+");
112     if (NULL == fp)
113     {
114         curl_easy_cleanup(curl);
115         printf("fopen failed\n");
116         return -1;
117     }
118     
119     curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data);
120     curl_easy_setopt(curl, CURLOPT_WRITEDATA, fp);
121     curl_code = curl_easy_perform(curl);
122     printf("curl_code = %d\n",curl_code);
123     if (CURLE_OK != curl_code)
124     {
125         printf("perform failed\n");
126         curl_easy_cleanup(curl);
127         fclose(fp);
128         remove(filePath);        
129         return -1;
130     }
131     curl_easy_cleanup(curl);
132     
133     fclose(fp);
134 
135     return 0;
136 }
137 
138 int main(int argc, char *argv[])
139 {
140     gloale_init();
141     char * serverip = "192.168.6.100";
142     char * port = "22";
143     char * serverpath = "/root/2.xml.bak";
144     char *user = "root";
145     char * passwd = "123456";
146     char * savepath = "/root/2.xml";
147     char url[125] = {0};
148     
149     sprintf(url,"sftp://%s:%s/%s",serverip,port,serverpath);
150     printf("url: %s\n", url);
151     // download(user,passwd,url,savepath);
152     upload(user,passwd,url,savepath);
153 
154     return 0;
155 }