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
31struct callback_data {
32  FILE *output;
33};
34
35static long file_is_coming(struct curl_fileinfo *finfo,
36                           struct callback_data *data,
37                           int remains);
38
39static long file_is_downloaded(struct callback_data *data);
40
41static size_t write_it(char *buff, size_t size, size_t nmemb,
42                       void *cb_data);
43
44int 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  int rc = curl_global_init(CURL_GLOBAL_ALL);
54  if(rc)
55    return 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 will write 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 rc;
94}
95
96static 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
131static 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
141static 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