docs/examples/ftp-wildcard.c

   1 /***************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * Copyright (C) 1998 - 2014, 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 #include <curl/curl.h>
  23 #include <stdio.h>
  24
  25 struct callback_data {
  26   FILE *output;
  27 };
  28
  29 static long file_is_coming(struct curl_fileinfo *finfo,
  30                            struct callback_data *data,
  31                            int remains);
  32
  33 static long file_is_downloaded(struct callback_data *data);
  34
  35 static size_t write_it(char *buff, size_t size, size_t nmemb,
  36                        void *cb_data);
  37
  38 int main(int argc, char **argv)
  39 {
  40   int rc = CURLE_OK;
  41
  42   /* curl easy handle */
  43   CURL *handle;
  44
  45   /* help data */
  46   struct callback_data data = { 0 };
  47
  48   /* global initialization */
  49   rc = curl_global_init(CURL_GLOBAL_ALL);
  50   if(rc)
  51     return rc;
  52
  53   /* initialization of easy handle */
  54   handle = curl_easy_init();
  55   if(!handle) {
  56     curl_global_cleanup();
  57     return CURLE_OUT_OF_MEMORY;
  58   }
  59
  60   /* turn on wildcard matching */
  61   curl_easy_setopt(handle, CURLOPT_WILDCARDMATCH, 1L);
  62
  63   /* callback is called before download of concrete file started */
  64   curl_easy_setopt(handle, CURLOPT_CHUNK_BGN_FUNCTION, file_is_coming);
  65
  66   /* callback is called after data from the file have been transferred */
  67   curl_easy_setopt(handle, CURLOPT_CHUNK_END_FUNCTION, file_is_downloaded);
  68
  69   /* this callback will write contents into files */
  70   curl_easy_setopt(handle, CURLOPT_WRITEFUNCTION, write_it);
  71
  72   /* put transfer data into callbacks */
  73   curl_easy_setopt(handle, CURLOPT_CHUNK_DATA, &data);
  74   curl_easy_setopt(handle, CURLOPT_WRITEDATA, &data);
  75
  76   /* curl_easy_setopt(handle, CURLOPT_VERBOSE, 1L); */
  77
  78   /* set an URL containing wildcard pattern (only in the last part) */
  79   if(argc == 2)
  80     curl_easy_setopt(handle, CURLOPT_URL, argv[1]);
  81   else
  82     curl_easy_setopt(handle, CURLOPT_URL, "ftp://example.com/test/*");
  83
  84   /* and start transfer! */
  85   rc = curl_easy_perform(handle);
  86
  87   curl_easy_cleanup(handle);
  88   curl_global_cleanup();
  89   return rc;
  90 }
  91
  92 static long file_is_coming(struct curl_fileinfo *finfo,
  93                            struct callback_data *data,
  94                            int remains)
  95 {
  96   printf("%3d %40s %10luB ", remains, finfo->filename,
  97          (unsigned long)finfo->size);
  98
  99   switch(finfo->filetype) {
 100   case CURLFILETYPE_DIRECTORY:
 101     printf(" DIR\n");
 102     break;
 103   case CURLFILETYPE_FILE:
 104     printf("FILE ");
 105     break;
 106   default:
 107     printf("OTHER\n");
 108     break;
 109   }
 110
 111   if(finfo->filetype == CURLFILETYPE_FILE) {
 112     /* do not transfer files >= 50B */
 113     if(finfo->size > 50) {
 114       printf("SKIPPED\n");
 115       return CURL_CHUNK_BGN_FUNC_SKIP;
 116     }
 117
 118     data->output = fopen(finfo->filename, "w");
 119     if(!data->output) {
 120       return CURL_CHUNK_BGN_FUNC_FAIL;
 121     }
 122   }
 123
 124   return CURL_CHUNK_BGN_FUNC_OK;
 125 }
 126
 127 static long file_is_downloaded(struct callback_data *data)
 128 {
 129   if(data->output) {
 130     printf("DOWNLOADED\n");
 131     fclose(data->output);
 132     data->output = 0x0;
 133   }
 134   return CURL_CHUNK_END_FUNC_OK;
 135 }
 136
 137 static size_t write_it(char *buff, size_t size, size_t nmemb,
 138                        void *cb_data)
 139 {
 140   struct callback_data *data = cb_data;
 141   size_t written = 0;
 142   if(data->output)
 143     written = fwrite(buff, size, nmemb, data->output);
 144   else
 145     /* listing output */
 146     written = fwrite(buff, size, nmemb, stdout);
 147   return written;
 148 }