docs/examples/ftp-wildcard.c

   1 /***************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * Copyright (C) 1998 - 2015, 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 http://curl.haxx.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  ***************************************************************************/
  22 /* <DESC>
  23  * FTP wildcard pattern matching
  24  * </DESC>
  25  */
  26 #include <curl/curl.h>
  27 #include <stdio.h>
  28
  29 struct callback_data {
  30   FILE *output;
  31 };
  32
  33 static long file_is_coming(struct curl_fileinfo *finfo,
  34                            struct callback_data *data,
  35                            int remains);
  36
  37 static long file_is_downloaded(struct callback_data *data);
  38
  39 static size_t write_it(char *buff, size_t size, size_t nmemb,
  40                        void *cb_data);
  41
  42 int main(int argc, char **argv)
  43 {
  44   int rc = CURLE_OK;
  45
  46   /* curl easy handle */
  47   CURL *handle;
  48
  49   /* help data */
  50   struct callback_data data = { 0 };
  51
  52   /* global initialization */
  53   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 an 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
  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, "w");
 123     if(!data->output) {
 124       return CURL_CHUNK_BGN_FUNC_FAIL;
 125     }
 126   }
 127
 128   return CURL_CHUNK_BGN_FUNC_OK;
 129 }
 130
 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
 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 }