curl_multi_wait: Add parameter to return number of active sockets
[platform/upstream/curl.git] / tests / libtest / lib1500.c
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 #include "test.h"
23
24 #include "testutil.h"
25 #include "warnless.h"
26 #include "memdebug.h"
27
28 #define TEST_HANG_TIMEOUT 60 * 1000
29
30 int test(char *URL)
31 {
32   CURL* curls = NULL;
33   CURLM* multi = NULL;
34   int still_running;
35   int i = -1;
36   int res = 0;
37   CURLMsg *msg;
38
39   start_test_timing();
40
41   global_init(CURL_GLOBAL_ALL);
42
43   multi_init(multi);
44
45   easy_init(curls);
46
47   easy_setopt(curls, CURLOPT_URL, URL);
48   easy_setopt(curls, CURLOPT_HEADER, 1L);
49
50   multi_add_handle(multi, curls);
51
52   multi_perform(multi, &still_running);
53
54   abort_on_test_timeout();
55
56   while(still_running) {
57     int num;
58     res = curl_multi_wait(multi, NULL, 0, TEST_HANG_TIMEOUT, &num);
59     if (res != CURLM_OK) {
60       printf("curl_multi_wait() returned %d\n", res);
61       res = -1;
62       goto test_cleanup;
63     }
64     if (num != 1) {
65       printf("curl_multi_wait() returned on %d handle(s), expected 1\n", num);
66       res = -1;
67       goto test_cleanup;
68     }
69
70     abort_on_test_timeout();
71
72     multi_perform(multi, &still_running);
73
74     abort_on_test_timeout();
75   }
76
77   msg = curl_multi_info_read(multi, &still_running);
78   if(msg)
79     /* this should now contain a result code from the easy handle,
80        get it */
81     i = msg->data.result;
82
83 test_cleanup:
84
85   /* undocumented cleanup sequence - type UA */
86
87   curl_multi_cleanup(multi);
88   curl_easy_cleanup(curls);
89   curl_global_cleanup();
90
91   if(res)
92     i = res;
93
94   return i; /* return the final return code */
95 }