123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246 |
- /***************************************************************************
- * _ _ ____ _
- * Project ___| | | | _ \| |
- * / __| | | | |_) | |
- * | (__| |_| | _ <| |___
- * \___|\___/|_| \_\_____|
- *
- * Copyright (C) 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 https://curl.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.
- *
- * SPDX-License-Identifier: curl
- *
- ***************************************************************************/
- #include "test.h"
- #include "memdebug.h"
- #define THREADS 2
- /* struct containing data of a thread */
- struct Tdata {
- CURLSH *share;
- char *url;
- };
- struct userdata {
- const char *text;
- int counter;
- };
- /* lock callback */
- static void my_lock(CURL *handle, curl_lock_data data,
- curl_lock_access laccess, void *useptr)
- {
- const char *what;
- struct userdata *user = (struct userdata *)useptr;
- (void)handle;
- (void)laccess;
- switch(data) {
- case CURL_LOCK_DATA_SHARE:
- what = "share";
- break;
- case CURL_LOCK_DATA_DNS:
- what = "dns";
- break;
- case CURL_LOCK_DATA_COOKIE:
- what = "cookie";
- break;
- case CURL_LOCK_DATA_SSL_SESSION:
- what = "ssl_session";
- break;
- default:
- fprintf(stderr, "lock: no such data: %d\n", (int)data);
- return;
- }
- printf("lock: %-6s [%s]: %d\n", what, user->text, user->counter);
- user->counter++;
- }
- /* unlock callback */
- static void my_unlock(CURL *handle, curl_lock_data data, void *useptr)
- {
- const char *what;
- struct userdata *user = (struct userdata *)useptr;
- (void)handle;
- switch(data) {
- case CURL_LOCK_DATA_SHARE:
- what = "share";
- break;
- case CURL_LOCK_DATA_DNS:
- what = "dns";
- break;
- case CURL_LOCK_DATA_COOKIE:
- what = "cookie";
- break;
- case CURL_LOCK_DATA_SSL_SESSION:
- what = "ssl_session";
- break;
- default:
- fprintf(stderr, "unlock: no such data: %d\n", (int)data);
- return;
- }
- printf("unlock: %-6s [%s]: %d\n", what, user->text, user->counter);
- user->counter++;
- }
- /* the dummy thread function */
- static void *fire(void *ptr)
- {
- CURLcode code;
- struct Tdata *tdata = (struct Tdata*)ptr;
- CURL *curl;
- curl = curl_easy_init();
- if(!curl) {
- fprintf(stderr, "curl_easy_init() failed\n");
- return NULL;
- }
- curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L);
- curl_easy_setopt(curl, CURLOPT_VERBOSE, 1L);
- curl_easy_setopt(curl, CURLOPT_URL, tdata->url);
- printf("CURLOPT_SHARE\n");
- curl_easy_setopt(curl, CURLOPT_SHARE, tdata->share);
- printf("PERFORM\n");
- code = curl_easy_perform(curl);
- if(code != CURLE_OK) {
- int i = 0;
- fprintf(stderr, "perform url '%s' repeat %d failed, curlcode %d\n",
- tdata->url, i, (int)code);
- }
- printf("CLEANUP\n");
- curl_easy_cleanup(curl);
- return NULL;
- }
- /* test function */
- int test(char *URL)
- {
- CURLcode res = CURLE_OK;
- CURLSHcode scode = CURLSHE_OK;
- char *url;
- struct Tdata tdata;
- CURL *curl;
- CURLSH *share;
- int i;
- struct userdata user;
- user.text = "Pigs in space";
- user.counter = 0;
- printf("GLOBAL_INIT\n");
- if(curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) {
- fprintf(stderr, "curl_global_init() failed\n");
- return TEST_ERR_MAJOR_BAD;
- }
- /* prepare share */
- printf("SHARE_INIT\n");
- share = curl_share_init();
- if(!share) {
- fprintf(stderr, "curl_share_init() failed\n");
- curl_global_cleanup();
- return TEST_ERR_MAJOR_BAD;
- }
- if(CURLSHE_OK == scode) {
- printf("CURLSHOPT_LOCKFUNC\n");
- scode = curl_share_setopt(share, CURLSHOPT_LOCKFUNC, my_lock);
- }
- if(CURLSHE_OK == scode) {
- printf("CURLSHOPT_UNLOCKFUNC\n");
- scode = curl_share_setopt(share, CURLSHOPT_UNLOCKFUNC, my_unlock);
- }
- if(CURLSHE_OK == scode) {
- printf("CURLSHOPT_USERDATA\n");
- scode = curl_share_setopt(share, CURLSHOPT_USERDATA, &user);
- }
- if(CURLSHE_OK == scode) {
- printf("CURL_LOCK_DATA_SSL_SESSION\n");
- scode = curl_share_setopt(share, CURLSHOPT_SHARE,
- CURL_LOCK_DATA_SSL_SESSION);
- }
- if(CURLSHE_OK != scode) {
- fprintf(stderr, "curl_share_setopt() failed\n");
- curl_share_cleanup(share);
- curl_global_cleanup();
- return TEST_ERR_MAJOR_BAD;
- }
- /* start treads */
- for(i = 1; i <= THREADS; i++) {
- /* set thread data */
- tdata.url = URL;
- tdata.share = share;
- /* simulate thread, direct call of "thread" function */
- printf("*** run %d\n",i);
- fire(&tdata);
- }
- /* fetch another one */
- printf("*** run %d\n", i);
- curl = curl_easy_init();
- if(!curl) {
- fprintf(stderr, "curl_easy_init() failed\n");
- curl_share_cleanup(share);
- curl_global_cleanup();
- return TEST_ERR_MAJOR_BAD;
- }
- url = URL;
- test_setopt(curl, CURLOPT_URL, url);
- printf("CURLOPT_SHARE\n");
- test_setopt(curl, CURLOPT_SHARE, share);
- printf("PERFORM\n");
- res = curl_easy_perform(curl);
- /* try to free share, expect to fail because share is in use */
- printf("try SHARE_CLEANUP...\n");
- scode = curl_share_cleanup(share);
- if(scode == CURLSHE_OK) {
- fprintf(stderr, "curl_share_cleanup succeed but error expected\n");
- share = NULL;
- }
- else {
- printf("SHARE_CLEANUP failed, correct\n");
- }
- test_cleanup:
- /* clean up last handle */
- printf("CLEANUP\n");
- curl_easy_cleanup(curl);
- /* free share */
- printf("SHARE_CLEANUP\n");
- scode = curl_share_cleanup(share);
- if(scode != CURLSHE_OK)
- fprintf(stderr, "curl_share_cleanup failed, code errno %d\n",
- (int)scode);
- printf("GLOBAL_CLEANUP\n");
- curl_global_cleanup();
- return res;
- }
|