1/***************************************************************************
2 *                                  _   _ ____  _
3 *  Project                     ___| | | |  _ \| |
4 *                             / __| | | | |_) | |
5 *                            | (__| |_| |  _ <| |___
6 *                             \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) 1998 - 2011, 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
23/* This test case is supposed to be identical to 547 except that this uses the
24 * multi interface and 547 is easy interface.
25 *
26 * argv1 = URL
27 * argv2 = proxy
28 * argv3 = proxyuser:password
29 */
30
31#include "test.h"
32#include "testutil.h"
33#include "warnless.h"
34#include "memdebug.h"
35
36#define TEST_HANG_TIMEOUT 60 * 1000
37
38#define UPLOADTHIS "this is the blurb we want to upload\n"
39
40static size_t readcallback(void  *ptr,
41                           size_t size,
42                           size_t nmemb,
43                           void *clientp)
44{
45  int *counter = (int *)clientp;
46
47  if(*counter) {
48    /* only do this once and then require a clearing of this */
49    fprintf(stderr, "READ ALREADY DONE!\n");
50    return 0;
51  }
52  (*counter)++; /* bump */
53
54  if(size * nmemb > strlen(UPLOADTHIS)) {
55    fprintf(stderr, "READ!\n");
56    strcpy(ptr, UPLOADTHIS);
57    return strlen(UPLOADTHIS);
58  }
59  fprintf(stderr, "READ NOT FINE!\n");
60  return 0;
61}
62static curlioerr ioctlcallback(CURL *handle,
63                               int cmd,
64                               void *clientp)
65{
66  int *counter = (int *)clientp;
67  (void)handle; /* unused */
68  if(cmd == CURLIOCMD_RESTARTREAD) {
69    fprintf(stderr, "REWIND!\n");
70    *counter = 0; /* clear counter to make the read callback restart */
71  }
72  return CURLIOE_OK;
73}
74
75
76int test(char *URL)
77{
78  int res = 0;
79  CURL *curl = NULL;
80  int counter=0;
81  CURLM *m = NULL;
82  int running=1;
83
84  start_test_timing();
85
86  global_init(CURL_GLOBAL_ALL);
87
88  easy_init(curl);
89
90  easy_setopt(curl, CURLOPT_URL, URL);
91  easy_setopt(curl, CURLOPT_VERBOSE, 1L);
92  easy_setopt(curl, CURLOPT_HEADER, 1L);
93
94  /* read the POST data from a callback */
95  easy_setopt(curl, CURLOPT_IOCTLFUNCTION, ioctlcallback);
96  easy_setopt(curl, CURLOPT_IOCTLDATA, &counter);
97  easy_setopt(curl, CURLOPT_READFUNCTION, readcallback);
98  easy_setopt(curl, CURLOPT_READDATA, &counter);
99  /* We CANNOT do the POST fine without setting the size (or choose chunked)! */
100  easy_setopt(curl, CURLOPT_POSTFIELDSIZE, strlen(UPLOADTHIS));
101
102  easy_setopt(curl, CURLOPT_POST, 1L);
103#ifdef CURL_DOES_CONVERSIONS
104  /* Convert the POST data to ASCII. */
105  easy_setopt(curl, CURLOPT_TRANSFERTEXT, 1L);
106#endif
107  easy_setopt(curl, CURLOPT_PROXY, libtest_arg2);
108  easy_setopt(curl, CURLOPT_PROXYUSERPWD, libtest_arg3);
109  easy_setopt(curl, CURLOPT_PROXYAUTH,
110                   (long) (CURLAUTH_NTLM | CURLAUTH_DIGEST | CURLAUTH_BASIC) );
111
112  multi_init(m);
113
114  multi_add_handle(m, curl);
115
116  while (running) {
117    struct timeval timeout;
118    fd_set fdread, fdwrite, fdexcep;
119    int maxfd = -99;
120
121    timeout.tv_sec = 0;
122    timeout.tv_usec = 100000L; /* 100 ms */
123
124    multi_perform(m, &running);
125
126    abort_on_test_timeout();
127
128#ifdef TPF
129    sleep(1); /* avoid ctl-10 dump */
130#endif
131
132    if(!running)
133      break; /* done */
134
135    FD_ZERO(&fdread);
136    FD_ZERO(&fdwrite);
137    FD_ZERO(&fdexcep);
138
139    multi_fdset(m, &fdread, &fdwrite, &fdexcep, &maxfd);
140
141    /* At this point, maxfd is guaranteed to be greater or equal than -1. */
142
143    select_test(maxfd+1, &fdread, &fdwrite, &fdexcep, &timeout);
144
145    abort_on_test_timeout();
146  }
147
148test_cleanup:
149
150  /* proper cleanup sequence - type PA */
151
152  curl_multi_remove_handle(m, curl);
153  curl_multi_cleanup(m);
154  curl_easy_cleanup(curl);
155  curl_global_cleanup();
156
157  return res;
158}
159
160