mirror of
https://github.com/moparisthebest/curl
synced 2024-10-31 23:55:01 -04:00
9e4f6c5809
This allows a libcurl test to portably sleep for a given number of milliseconds.
157 lines
3.9 KiB
C
157 lines
3.9 KiB
C
/***************************************************************************
|
|
* _ _ ____ _
|
|
* Project ___| | | | _ \| |
|
|
* / __| | | | |_) | |
|
|
* | (__| |_| | _ <| |___
|
|
* \___|\___/|_| \_\_____|
|
|
*
|
|
* Copyright (C) 1998 - 2014, Daniel Stenberg, <daniel@haxx.se>, et al.
|
|
*
|
|
* This software is licensed as described in the file COPYING, which
|
|
* you should have received as part of this distribution. The terms
|
|
* are also available at http://curl.haxx.se/docs/copyright.html.
|
|
*
|
|
* You may opt to use, copy, modify, merge, publish, distribute and/or sell
|
|
* copies of the Software, and permit persons to whom the Software is
|
|
* furnished to do so, under the terms of the COPYING file.
|
|
*
|
|
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
|
|
* KIND, either express or implied.
|
|
*
|
|
***************************************************************************/
|
|
#include "test.h"
|
|
|
|
#ifdef HAVE_LOCALE_H
|
|
# include <locale.h> /* for setlocale() */
|
|
#endif
|
|
|
|
#ifdef HAVE_IO_H
|
|
# include <io.h> /* for setmode() */
|
|
#endif
|
|
|
|
#ifdef HAVE_FCNTL_H
|
|
# include <fcntl.h> /* for setmode() */
|
|
#endif
|
|
|
|
#ifdef CURLDEBUG
|
|
# define MEMDEBUG_NODEFINES
|
|
# include "memdebug.h"
|
|
#endif
|
|
|
|
int select_wrapper(int nfds, fd_set *rd, fd_set *wr, fd_set *exc,
|
|
struct timeval *tv)
|
|
{
|
|
if(nfds < 0) {
|
|
SET_SOCKERRNO(EINVAL);
|
|
return -1;
|
|
}
|
|
#ifdef USE_WINSOCK
|
|
/*
|
|
* Winsock select() requires that at least one of the three fd_set
|
|
* pointers is not NULL and points to a non-empty fdset. IOW Winsock
|
|
* select() can not be used to sleep without a single fd_set.
|
|
*/
|
|
if(!nfds) {
|
|
Sleep(1000*tv->tv_sec + tv->tv_usec/1000);
|
|
return 0;
|
|
}
|
|
#endif
|
|
return select(nfds, rd, wr, exc, tv);
|
|
}
|
|
|
|
void wait_ms(int ms)
|
|
{
|
|
struct timeval t;
|
|
t.tv_sec = ms/1000;
|
|
ms -= (int)t.tv_sec * 1000;
|
|
t.tv_usec = ms * 1000;
|
|
select_wrapper(0, NULL, NULL , NULL, &t);
|
|
}
|
|
|
|
char *libtest_arg2=NULL;
|
|
char *libtest_arg3=NULL;
|
|
int test_argc;
|
|
char **test_argv;
|
|
|
|
struct timeval tv_test_start; /* for test timing */
|
|
|
|
#ifdef UNITTESTS
|
|
int unitfail; /* for unittests */
|
|
#endif
|
|
|
|
#ifdef CURLDEBUG
|
|
static void memory_tracking_init(void)
|
|
{
|
|
char *env;
|
|
/* if CURL_MEMDEBUG is set, this starts memory tracking message logging */
|
|
env = curl_getenv("CURL_MEMDEBUG");
|
|
if(env) {
|
|
/* use the value as file name */
|
|
char fname[CURL_MT_LOGFNAME_BUFSIZE];
|
|
if(strlen(env) >= CURL_MT_LOGFNAME_BUFSIZE)
|
|
env[CURL_MT_LOGFNAME_BUFSIZE-1] = '\0';
|
|
strcpy(fname, env);
|
|
curl_free(env);
|
|
curl_memdebug(fname);
|
|
/* this weird stuff here is to make curl_free() get called
|
|
before curl_memdebug() as otherwise memory tracking will
|
|
log a free() without an alloc! */
|
|
}
|
|
/* if CURL_MEMLIMIT is set, this enables fail-on-alloc-number-N feature */
|
|
env = curl_getenv("CURL_MEMLIMIT");
|
|
if(env) {
|
|
char *endptr;
|
|
long num = strtol(env, &endptr, 10);
|
|
if((endptr != env) && (endptr == env + strlen(env)) && (num > 0))
|
|
curl_memlimit(num);
|
|
curl_free(env);
|
|
}
|
|
}
|
|
#else
|
|
# define memory_tracking_init() Curl_nop_stmt
|
|
#endif
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
char *URL;
|
|
|
|
#ifdef O_BINARY
|
|
# ifdef __HIGHC__
|
|
_setmode(stdout, O_BINARY);
|
|
# else
|
|
setmode(fileno(stdout), O_BINARY);
|
|
# endif
|
|
#endif
|
|
|
|
memory_tracking_init();
|
|
|
|
/*
|
|
* Setup proper locale from environment. This is needed to enable locale-
|
|
* specific behaviour by the C library in order to test for undesired side
|
|
* effects that could cause in libcurl.
|
|
*/
|
|
#ifdef HAVE_SETLOCALE
|
|
setlocale(LC_ALL, "");
|
|
#endif
|
|
|
|
if(argc< 2 ) {
|
|
fprintf(stderr, "Pass URL as argument please\n");
|
|
return 1;
|
|
}
|
|
|
|
test_argc = argc;
|
|
test_argv = argv;
|
|
|
|
if(argc>2)
|
|
libtest_arg2=argv[2];
|
|
|
|
if(argc>3)
|
|
libtest_arg3=argv[3];
|
|
|
|
URL = argv[1]; /* provide this to the rest */
|
|
|
|
fprintf(stderr, "URL: %s\n", URL);
|
|
|
|
return test(URL);
|
|
}
|