← Back to branch summary

~ubuntu-branches/ubuntu/vivid/curl/vivid-security

« back to all changes in this revision

Viewing changes to tests/libtest/lib1515.c

  • Committer: Package Import Robot
  • Author(s): Michael Vogt
  • Date: 2014-04-30 13:34:14 UTC
  • mfrom: (76.1.1 utopic)
  • Revision ID: package-import@ubuntu.com-20140430133414-lal4vsnldlvsue7n
* Merge from Debian unstable.  Remaining changes:
  - Drop dependencies not in main:
    + Build-Depends: Drop stunnel4 and libssh2-1-dev.
    + Drop libssh2-1-dev from binary package Depends.
  - Add new libcurl3-udeb package.
  - Add new curl-udeb package.

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/libtest/lib1515.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
 
 
23
/*
 
24
 * Check for bugs #1303 and #1327: libcurl should never remove DNS entries
 
25
 * created via CURLOPT_RESOLVE, neither after DNS_CACHE_TIMEOUT elapses
 
26
 * (test1515) nor a dead connection is detected (test1616).
 
27
 */
 
28
 
 
29
#include "test.h"
 
30
#include "testutil.h"
 
31
#include "warnless.h"
 
32
#include "memdebug.h"
 
33
 
 
34
#define TEST_HANG_TIMEOUT 60 * 1000
 
35
 
 
36
#define DNS_TIMEOUT 1
 
37
 
 
38
#if defined(WIN32) || defined(_WIN32)
 
39
#define sleep(s) Sleep(s * 1000)
 
40
#endif
 
41
 
 
42
#define _MPRINTF_REPLACE
 
43
#include <curl/mprintf.h>
 
44
 
 
45
static int debug_callback(CURL *curl, curl_infotype info, char *msg, size_t len, void *ptr)
 
46
{
 
47
  (void)curl;
 
48
  (void)ptr;
 
49
 
 
50
  if(info == CURLINFO_TEXT)
 
51
    fprintf(stderr, "debug: %.*s", (int) len, msg);
 
52
 
 
53
  return 0;
 
54
}
 
55
 
 
56
static int do_one_request(CURLM *m, char *URL, char *resolve)
 
57
{
 
58
  CURL *curls;
 
59
  struct curl_slist *resolve_list = NULL;
 
60
  int still_running;
 
61
  int res = 0;
 
62
  CURLMsg *msg;
 
63
  int msgs_left;
 
64
 
 
65
  resolve_list = curl_slist_append(resolve_list, resolve);
 
66
 
 
67
  easy_init(curls);
 
68
 
 
69
  easy_setopt(curls, CURLOPT_URL, URL);
 
70
  easy_setopt(curls, CURLOPT_RESOLVE, resolve_list);
 
71
  easy_setopt(curls, CURLOPT_DEBUGFUNCTION, debug_callback);
 
72
  easy_setopt(curls, CURLOPT_VERBOSE, 1);
 
73
  easy_setopt(curls, CURLOPT_DNS_CACHE_TIMEOUT, DNS_TIMEOUT);
 
74
 
 
75
  multi_add_handle(m, curls);
 
76
  multi_perform(m, &still_running);
 
77
 
 
78
  abort_on_test_timeout();
 
79
 
 
80
  while(still_running) {
 
81
    struct timeval timeout;
 
82
    fd_set fdread, fdwrite, fdexcep;
 
83
    int maxfd = -99;
 
84
 
 
85
    FD_ZERO(&fdread);
 
86
    FD_ZERO(&fdwrite);
 
87
    FD_ZERO(&fdexcep);
 
88
    timeout.tv_sec = 1;
 
89
    timeout.tv_usec = 0;
 
90
 
 
91
    multi_fdset(m, &fdread, &fdwrite, &fdexcep, &maxfd);
 
92
    select_test(maxfd+1, &fdread, &fdwrite, &fdexcep, &timeout);
 
93
 
 
94
    abort_on_test_timeout();
 
95
    multi_perform(m, &still_running);
 
96
 
 
97
    abort_on_test_timeout();
 
98
  }
 
99
 
 
100
  while((msg = curl_multi_info_read(m, &msgs_left))) {
 
101
    if(msg->msg == CURLMSG_DONE && msg->easy_handle == curls) {
 
102
      res = msg->data.result;
 
103
      break;
 
104
    }
 
105
  }
 
106
 
 
107
test_cleanup:
 
108
 
 
109
  curl_multi_remove_handle(m, curls);
 
110
  curl_easy_cleanup(curls);
 
111
  curl_slist_free_all(resolve_list);
 
112
 
 
113
  return res;
 
114
}
 
115
 
 
116
int test(char *URL)
 
117
{
 
118
  CURLM* multi = NULL;
 
119
  int res = 0;
 
120
  char *address = libtest_arg2;
 
121
  char *port = libtest_arg3;
 
122
  char *path = URL;
 
123
  char dns_entry[256];
 
124
  int i;
 
125
  int count = 2;
 
126
 
 
127
  snprintf(dns_entry, sizeof(dns_entry), "testserver.example.com:%s:%s", port, address);
 
128
 
 
129
  start_test_timing();
 
130
 
 
131
  global_init(CURL_GLOBAL_ALL);
 
132
  multi_init(multi);
 
133
 
 
134
  for(i = 1; i <= count; i++) {
 
135
    char target_url[256];
 
136
    snprintf(target_url, sizeof(target_url), "http://testserver.example.com:%s%s%04d", port, path, i);
 
137
 
 
138
    /* second request must succeed like the first one */
 
139
    if((res = do_one_request(multi, target_url, dns_entry)))
 
140
      goto test_cleanup;
 
141
 
 
142
    if(i < count)
 
143
      sleep(DNS_TIMEOUT + 1);
 
144
  }
 
145
 
 
146
test_cleanup:
 
147
 
 
148
  curl_multi_cleanup(multi);
 
149
 
 
150
  return (int) res;
 
151
}