xref: /aosp_15_r20/external/curl/docs/examples/sftpuploadresume.c (revision 6236dae45794135f37c4eb022389c904c8b0090d)
1*6236dae4SAndroid Build Coastguard Worker /***************************************************************************
2*6236dae4SAndroid Build Coastguard Worker  *                                  _   _ ____  _
3*6236dae4SAndroid Build Coastguard Worker  *  Project                     ___| | | |  _ \| |
4*6236dae4SAndroid Build Coastguard Worker  *                             / __| | | | |_) | |
5*6236dae4SAndroid Build Coastguard Worker  *                            | (__| |_| |  _ <| |___
6*6236dae4SAndroid Build Coastguard Worker  *                             \___|\___/|_| \_\_____|
7*6236dae4SAndroid Build Coastguard Worker  *
8*6236dae4SAndroid Build Coastguard Worker  * Copyright (C) Daniel Stenberg, <[email protected]>, et al.
9*6236dae4SAndroid Build Coastguard Worker  *
10*6236dae4SAndroid Build Coastguard Worker  * This software is licensed as described in the file COPYING, which
11*6236dae4SAndroid Build Coastguard Worker  * you should have received as part of this distribution. The terms
12*6236dae4SAndroid Build Coastguard Worker  * are also available at https://curl.se/docs/copyright.html.
13*6236dae4SAndroid Build Coastguard Worker  *
14*6236dae4SAndroid Build Coastguard Worker  * You may opt to use, copy, modify, merge, publish, distribute and/or sell
15*6236dae4SAndroid Build Coastguard Worker  * copies of the Software, and permit persons to whom the Software is
16*6236dae4SAndroid Build Coastguard Worker  * furnished to do so, under the terms of the COPYING file.
17*6236dae4SAndroid Build Coastguard Worker  *
18*6236dae4SAndroid Build Coastguard Worker  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19*6236dae4SAndroid Build Coastguard Worker  * KIND, either express or implied.
20*6236dae4SAndroid Build Coastguard Worker  *
21*6236dae4SAndroid Build Coastguard Worker  * SPDX-License-Identifier: curl
22*6236dae4SAndroid Build Coastguard Worker  *
23*6236dae4SAndroid Build Coastguard Worker  ***************************************************************************/
24*6236dae4SAndroid Build Coastguard Worker /* <DESC>
25*6236dae4SAndroid Build Coastguard Worker  * Upload to SFTP, resuming a previously aborted transfer.
26*6236dae4SAndroid Build Coastguard Worker  * </DESC>
27*6236dae4SAndroid Build Coastguard Worker  */
28*6236dae4SAndroid Build Coastguard Worker 
29*6236dae4SAndroid Build Coastguard Worker #include <stdlib.h>
30*6236dae4SAndroid Build Coastguard Worker #include <stdio.h>
31*6236dae4SAndroid Build Coastguard Worker #include <curl/curl.h>
32*6236dae4SAndroid Build Coastguard Worker 
33*6236dae4SAndroid Build Coastguard Worker /* read data to upload */
readfunc(char * ptr,size_t size,size_t nmemb,void * stream)34*6236dae4SAndroid Build Coastguard Worker static size_t readfunc(char *ptr, size_t size, size_t nmemb, void *stream)
35*6236dae4SAndroid Build Coastguard Worker {
36*6236dae4SAndroid Build Coastguard Worker   FILE *f = (FILE *)stream;
37*6236dae4SAndroid Build Coastguard Worker   size_t n;
38*6236dae4SAndroid Build Coastguard Worker 
39*6236dae4SAndroid Build Coastguard Worker   if(ferror(f))
40*6236dae4SAndroid Build Coastguard Worker     return CURL_READFUNC_ABORT;
41*6236dae4SAndroid Build Coastguard Worker 
42*6236dae4SAndroid Build Coastguard Worker   n = fread(ptr, size, nmemb, f) * size;
43*6236dae4SAndroid Build Coastguard Worker 
44*6236dae4SAndroid Build Coastguard Worker   return n;
45*6236dae4SAndroid Build Coastguard Worker }
46*6236dae4SAndroid Build Coastguard Worker 
47*6236dae4SAndroid Build Coastguard Worker /*
48*6236dae4SAndroid Build Coastguard Worker  * sftpGetRemoteFileSize returns the remote file size in byte; -1 on error
49*6236dae4SAndroid Build Coastguard Worker  */
sftpGetRemoteFileSize(const char * i_remoteFile)50*6236dae4SAndroid Build Coastguard Worker static curl_off_t sftpGetRemoteFileSize(const char *i_remoteFile)
51*6236dae4SAndroid Build Coastguard Worker {
52*6236dae4SAndroid Build Coastguard Worker   CURLcode result = CURLE_GOT_NOTHING;
53*6236dae4SAndroid Build Coastguard Worker   curl_off_t remoteFileSizeByte = -1;
54*6236dae4SAndroid Build Coastguard Worker   CURL *curlHandlePtr = curl_easy_init();
55*6236dae4SAndroid Build Coastguard Worker 
56*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlHandlePtr, CURLOPT_VERBOSE, 1L);
57*6236dae4SAndroid Build Coastguard Worker 
58*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlHandlePtr, CURLOPT_URL, i_remoteFile);
59*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlHandlePtr, CURLOPT_NOPROGRESS, 1);
60*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlHandlePtr, CURLOPT_NOBODY, 1);
61*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlHandlePtr, CURLOPT_HEADER, 1);
62*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlHandlePtr, CURLOPT_FILETIME, 1);
63*6236dae4SAndroid Build Coastguard Worker 
64*6236dae4SAndroid Build Coastguard Worker   result = curl_easy_perform(curlHandlePtr);
65*6236dae4SAndroid Build Coastguard Worker   if(CURLE_OK == result) {
66*6236dae4SAndroid Build Coastguard Worker     result = curl_easy_getinfo(curlHandlePtr,
67*6236dae4SAndroid Build Coastguard Worker                                CURLINFO_CONTENT_LENGTH_DOWNLOAD_T,
68*6236dae4SAndroid Build Coastguard Worker                                &remoteFileSizeByte);
69*6236dae4SAndroid Build Coastguard Worker     if(result)
70*6236dae4SAndroid Build Coastguard Worker       return -1;
71*6236dae4SAndroid Build Coastguard Worker     printf("filesize: %lu\n", (unsigned long)remoteFileSizeByte);
72*6236dae4SAndroid Build Coastguard Worker   }
73*6236dae4SAndroid Build Coastguard Worker   curl_easy_cleanup(curlHandlePtr);
74*6236dae4SAndroid Build Coastguard Worker 
75*6236dae4SAndroid Build Coastguard Worker   return remoteFileSizeByte;
76*6236dae4SAndroid Build Coastguard Worker }
77*6236dae4SAndroid Build Coastguard Worker 
78*6236dae4SAndroid Build Coastguard Worker 
sftpResumeUpload(CURL * curlhandle,const char * remotepath,const char * localpath)79*6236dae4SAndroid Build Coastguard Worker static int sftpResumeUpload(CURL *curlhandle, const char *remotepath,
80*6236dae4SAndroid Build Coastguard Worker                             const char *localpath)
81*6236dae4SAndroid Build Coastguard Worker {
82*6236dae4SAndroid Build Coastguard Worker   FILE *f = NULL;
83*6236dae4SAndroid Build Coastguard Worker   CURLcode result = CURLE_GOT_NOTHING;
84*6236dae4SAndroid Build Coastguard Worker 
85*6236dae4SAndroid Build Coastguard Worker   curl_off_t remoteFileSizeByte = sftpGetRemoteFileSize(remotepath);
86*6236dae4SAndroid Build Coastguard Worker   if(-1 == remoteFileSizeByte) {
87*6236dae4SAndroid Build Coastguard Worker     printf("Error reading the remote file size: unable to resume upload\n");
88*6236dae4SAndroid Build Coastguard Worker     return -1;
89*6236dae4SAndroid Build Coastguard Worker   }
90*6236dae4SAndroid Build Coastguard Worker 
91*6236dae4SAndroid Build Coastguard Worker   f = fopen(localpath, "rb");
92*6236dae4SAndroid Build Coastguard Worker   if(!f) {
93*6236dae4SAndroid Build Coastguard Worker     perror(NULL);
94*6236dae4SAndroid Build Coastguard Worker     return 0;
95*6236dae4SAndroid Build Coastguard Worker   }
96*6236dae4SAndroid Build Coastguard Worker 
97*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlhandle, CURLOPT_UPLOAD, 1L);
98*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlhandle, CURLOPT_URL, remotepath);
99*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlhandle, CURLOPT_READFUNCTION, readfunc);
100*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlhandle, CURLOPT_READDATA, f);
101*6236dae4SAndroid Build Coastguard Worker 
102*6236dae4SAndroid Build Coastguard Worker #ifdef _WIN32
103*6236dae4SAndroid Build Coastguard Worker   _fseeki64(f, remoteFileSizeByte, SEEK_SET);
104*6236dae4SAndroid Build Coastguard Worker #else
105*6236dae4SAndroid Build Coastguard Worker   fseek(f, (long)remoteFileSizeByte, SEEK_SET);
106*6236dae4SAndroid Build Coastguard Worker #endif
107*6236dae4SAndroid Build Coastguard Worker   curl_easy_setopt(curlhandle, CURLOPT_APPEND, 1L);
108*6236dae4SAndroid Build Coastguard Worker   result = curl_easy_perform(curlhandle);
109*6236dae4SAndroid Build Coastguard Worker 
110*6236dae4SAndroid Build Coastguard Worker   fclose(f);
111*6236dae4SAndroid Build Coastguard Worker 
112*6236dae4SAndroid Build Coastguard Worker   if(result == CURLE_OK)
113*6236dae4SAndroid Build Coastguard Worker     return 1;
114*6236dae4SAndroid Build Coastguard Worker   else {
115*6236dae4SAndroid Build Coastguard Worker     fprintf(stderr, "%s\n", curl_easy_strerror(result));
116*6236dae4SAndroid Build Coastguard Worker     return 0;
117*6236dae4SAndroid Build Coastguard Worker   }
118*6236dae4SAndroid Build Coastguard Worker }
119*6236dae4SAndroid Build Coastguard Worker 
main(void)120*6236dae4SAndroid Build Coastguard Worker int main(void)
121*6236dae4SAndroid Build Coastguard Worker {
122*6236dae4SAndroid Build Coastguard Worker   const char *remote = "sftp://user:[email protected]/path/filename";
123*6236dae4SAndroid Build Coastguard Worker   const char *filename = "filename";
124*6236dae4SAndroid Build Coastguard Worker   CURL *curlhandle = NULL;
125*6236dae4SAndroid Build Coastguard Worker 
126*6236dae4SAndroid Build Coastguard Worker   curl_global_init(CURL_GLOBAL_ALL);
127*6236dae4SAndroid Build Coastguard Worker   curlhandle = curl_easy_init();
128*6236dae4SAndroid Build Coastguard Worker 
129*6236dae4SAndroid Build Coastguard Worker   if(!sftpResumeUpload(curlhandle, remote, filename)) {
130*6236dae4SAndroid Build Coastguard Worker     printf("resumed upload using curl %s failed\n", curl_version());
131*6236dae4SAndroid Build Coastguard Worker   }
132*6236dae4SAndroid Build Coastguard Worker 
133*6236dae4SAndroid Build Coastguard Worker   curl_easy_cleanup(curlhandle);
134*6236dae4SAndroid Build Coastguard Worker   curl_global_cleanup();
135*6236dae4SAndroid Build Coastguard Worker 
136*6236dae4SAndroid Build Coastguard Worker   return 0;
137*6236dae4SAndroid Build Coastguard Worker }
138