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
