tests/libtest/lib525.c

   1 /***************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * Copyright (C) 1998 - 2017, 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.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 "test.h"
  23
  24 #include <fcntl.h>
  25
  26 #include "testutil.h"
  27 #include "warnless.h"
  28 #include "memdebug.h"
  29
  30 #define TEST_HANG_TIMEOUT 60 * 1000
  31
  32 int test(char *URL)
  33 {
  34   int res = 0;
  35   CURL *curl = NULL;
  36   FILE *hd_src = NULL;
  37   int hd;
  38   struct_stat file_info;
  39   CURLM *m = NULL;
  40   int running;
  41
  42   start_test_timing();
  43
  44   if(!libtest_arg2) {
  45 #ifdef LIB529
  46     /* test 529 */
  47     fprintf(stderr, "Usage: lib529 [url] [uploadfile]\n");
  48 #else
  49     /* test 525 */
  50     fprintf(stderr, "Usage: lib525 [url] [uploadfile]\n");
  51 #endif
  52     return TEST_ERR_USAGE;
  53   }
  54
  55   hd_src = fopen(libtest_arg2, "rb");
  56   if(NULL == hd_src) {
  57     fprintf(stderr, "fopen failed with error: %d (%s)\n",
  58             errno, strerror(errno));
  59     fprintf(stderr, "Error opening file: (%s)\n", libtest_arg2);
  60     return TEST_ERR_FOPEN;
  61   }
  62
  63   /* get the file size of the local file */
  64   hd = fstat(fileno(hd_src), &file_info);
  65   if(hd == -1) {
  66     /* can't open file, bail out */
  67     fprintf(stderr, "fstat() failed with error: %d (%s)\n",
  68             errno, strerror(errno));
  69     fprintf(stderr, "ERROR: cannot open file (%s)\n", libtest_arg2);
  70     fclose(hd_src);
  71     return TEST_ERR_FSTAT;
  72   }
  73
  74   res_global_init(CURL_GLOBAL_ALL);
  75   if(res) {
  76     fclose(hd_src);
  77     return res;
  78   }
  79
  80   easy_init(curl);
  81
  82   /* enable uploading */
  83   easy_setopt(curl, CURLOPT_UPLOAD, 1L);
  84
  85   /* specify target */
  86   easy_setopt(curl, CURLOPT_URL, URL);
  87
  88   /* go verbose */
  89   easy_setopt(curl, CURLOPT_VERBOSE, 1L);
  90
  91   /* use active FTP */
  92   easy_setopt(curl, CURLOPT_FTPPORT, "-");
  93
  94   /* now specify which file to upload */
  95   easy_setopt(curl, CURLOPT_READDATA, hd_src);
  96
  97   /* NOTE: if you want this code to work on Windows with libcurl as a DLL, you
  98      MUST also provide a read callback with CURLOPT_READFUNCTION. Failing to
  99      do so will give you a crash since a DLL may not use the variable's memory
 100      when passed in to it from an app like this. */
 101
 102   /* Set the size of the file to upload (optional).  If you give a *_LARGE
 103      option you MUST make sure that the type of the passed-in argument is a
 104      curl_off_t. If you use CURLOPT_INFILESIZE (without _LARGE) you must
 105      make sure that to pass in a type 'long' argument. */
 106   easy_setopt(curl, CURLOPT_INFILESIZE_LARGE, (curl_off_t)file_info.st_size);
 107
 108   multi_init(m);
 109
 110   multi_add_handle(m, curl);
 111
 112   for(;;) {
 113     struct timeval interval;
 114     fd_set rd, wr, exc;
 115     int maxfd = -99;
 116
 117     interval.tv_sec = 1;
 118     interval.tv_usec = 0;
 119
 120     multi_perform(m, &running);
 121
 122     abort_on_test_timeout();
 123
 124     if(!running)
 125       break; /* done */
 126
 127     FD_ZERO(&rd);
 128     FD_ZERO(&wr);
 129     FD_ZERO(&exc);
 130
 131     multi_fdset(m, &rd, &wr, &exc, &maxfd);
 132
 133     /* At this point, maxfd is guaranteed to be greater or equal than -1. */
 134
 135     select_test(maxfd + 1, &rd, &wr, &exc, &interval);
 136
 137     abort_on_test_timeout();
 138   }
 139
 140 test_cleanup:
 141
 142 #ifdef LIB529
 143   /* test 529 */
 144   /* proper cleanup sequence - type PA */
 145   curl_multi_remove_handle(m, curl);
 146   curl_multi_cleanup(m);
 147   curl_easy_cleanup(curl);
 148   curl_global_cleanup();
 149 #else
 150   /* test 525 */
 151   /* proper cleanup sequence - type PB */
 152   curl_multi_remove_handle(m, curl);
 153   curl_easy_cleanup(curl);
 154   curl_multi_cleanup(m);
 155   curl_global_cleanup();
 156 #endif
 157
 158   /* close the local file */
 159   fclose(hd_src);
 160
 161   return res;
 162 }