xref: /curl/docs/examples/ftp-wildcard.c (revision 5b9955e0)
1 /***************************************************************************
2  *                                  _   _ ____  _
3  *  Project                     ___| | | |  _ \| |
4  *                             / __| | | | |_) | |
5  *                            | (__| |_| |  _ <| |___
6  *                             \___|\___/|_| \_\_____|
7  *
8  * Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
9  *
10  * This software is licensed as described in the file COPYING, which
11  * you should have received as part of this distribution. The terms
12  * are also available at https://curl.se/docs/copyright.html.
13  *
14  * You may opt to use, copy, modify, merge, publish, distribute and/or sell
15  * copies of the Software, and permit persons to whom the Software is
16  * furnished to do so, under the terms of the COPYING file.
17  *
18  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19  * KIND, either express or implied.
20  *
21  * SPDX-License-Identifier: curl
22  *
23  ***************************************************************************/
24 /* <DESC>
25  * FTP wildcard pattern matching
26  * </DESC>
27  */
28 #include <curl/curl.h>
29 #include <stdio.h>
30 
31 struct callback_data {
32   FILE *output;
33 };
34 
35 static long file_is_coming(struct curl_fileinfo *finfo,
36                            struct callback_data *data,
37                            int remains);
38 
39 static long file_is_downloaded(struct callback_data *data);
40 
41 static size_t write_it(char *buff, size_t size, size_t nmemb,
42                        void *cb_data);
43 
main(int argc,char ** argv)44 int main(int argc, char **argv)
45 {
46   /* curl easy handle */
47   CURL *handle;
48 
49   /* help data */
50   struct callback_data data = { 0 };
51 
52   /* global initialization */
53   CURLcode rc = curl_global_init(CURL_GLOBAL_ALL);
54   if(rc)
55     return (int)rc;
56 
57   /* initialization of easy handle */
58   handle = curl_easy_init();
59   if(!handle) {
60     curl_global_cleanup();
61     return CURLE_OUT_OF_MEMORY;
62   }
63 
64   /* turn on wildcard matching */
65   curl_easy_setopt(handle, CURLOPT_WILDCARDMATCH, 1L);
66 
67   /* callback is called before download of concrete file started */
68   curl_easy_setopt(handle, CURLOPT_CHUNK_BGN_FUNCTION, file_is_coming);
69 
70   /* callback is called after data from the file have been transferred */
71   curl_easy_setopt(handle, CURLOPT_CHUNK_END_FUNCTION, file_is_downloaded);
72 
73   /* this callback writes contents into files */
74   curl_easy_setopt(handle, CURLOPT_WRITEFUNCTION, write_it);
75 
76   /* put transfer data into callbacks */
77   curl_easy_setopt(handle, CURLOPT_CHUNK_DATA, &data);
78   curl_easy_setopt(handle, CURLOPT_WRITEDATA, &data);
79 
80   /* curl_easy_setopt(handle, CURLOPT_VERBOSE, 1L); */
81 
82   /* set a URL containing wildcard pattern (only in the last part) */
83   if(argc == 2)
84     curl_easy_setopt(handle, CURLOPT_URL, argv[1]);
85   else
86     curl_easy_setopt(handle, CURLOPT_URL, "ftp://example.com/test/*");
87 
88   /* and start transfer! */
89   rc = curl_easy_perform(handle);
90 
91   curl_easy_cleanup(handle);
92   curl_global_cleanup();
93   return (int)rc;
94 }
95 
file_is_coming(struct curl_fileinfo * finfo,struct callback_data * data,int remains)96 static long file_is_coming(struct curl_fileinfo *finfo,
97                            struct callback_data *data,
98                            int remains)
99 {
100   printf("%3d %40s %10luB ", remains, finfo->filename,
101          (unsigned long)finfo->size);
102 
103   switch(finfo->filetype) {
104   case CURLFILETYPE_DIRECTORY:
105     printf(" DIR\n");
106     break;
107   case CURLFILETYPE_FILE:
108     printf("FILE ");
109     break;
110   default:
111     printf("OTHER\n");
112     break;
113   }
114 
115   if(finfo->filetype == CURLFILETYPE_FILE) {
116     /* do not transfer files >= 50B */
117     if(finfo->size > 50) {
118       printf("SKIPPED\n");
119       return CURL_CHUNK_BGN_FUNC_SKIP;
120     }
121 
122     data->output = fopen(finfo->filename, "wb");
123     if(!data->output) {
124       return CURL_CHUNK_BGN_FUNC_FAIL;
125     }
126   }
127 
128   return CURL_CHUNK_BGN_FUNC_OK;
129 }
130 
file_is_downloaded(struct callback_data * data)131 static long file_is_downloaded(struct callback_data *data)
132 {
133   if(data->output) {
134     printf("DOWNLOADED\n");
135     fclose(data->output);
136     data->output = 0x0;
137   }
138   return CURL_CHUNK_END_FUNC_OK;
139 }
140 
write_it(char * buff,size_t size,size_t nmemb,void * cb_data)141 static size_t write_it(char *buff, size_t size, size_t nmemb,
142                        void *cb_data)
143 {
144   struct callback_data *data = cb_data;
145   size_t written = 0;
146   if(data->output)
147     written = fwrite(buff, size, nmemb, data->output);
148   else
149     /* listing output */
150     written = fwrite(buff, size, nmemb, stdout);
151   return written;
152 }
153