X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libtest%2Fhttp.cc;h=465ce6186708c9d3c5eeb78fb0703343e4d7e056;hb=b6719ab469b16022edd830cb90bc476b17fa6743;hp=5309d50447a7dea2bdbadc047e2caea445917dc3;hpb=a2d3699da26720a399bd7f563ccdbccf26e610c5;p=awesomized%2Flibmemcached diff --git a/libtest/http.cc b/libtest/http.cc index 5309d504..465ce618 100644 --- a/libtest/http.cc +++ b/libtest/http.cc @@ -1,68 +1,126 @@ /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab: - * - * libtest * - * Copyright (C) 2011 Data Differential, http://datadifferential.com/ + * Data Differential YATL (i.e. libtest) library * - * This library is free software; you can redistribute it and/or - * modify it under the terms of the GNU Lesser General Public - * License as published by the Free Software Foundation; either - * version 3 of the License, or (at your option) any later version. + * Copyright (C) 2012 Data Differential, http://datadifferential.com/ * - * This library is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU - * Lesser General Public License for more details. + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are + * met: + * + * * Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * * Redistributions in binary form must reproduce the above + * copyright notice, this list of conditions and the following disclaimer + * in the documentation and/or other materials provided with the + * distribution. + * + * * The names of its contributors may not be used to endorse or + * promote products derived from this software without specific prior + * written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * - * You should have received a copy of the GNU Lesser General Public - * License along with this library; if not, write to the Free Software - * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ +#include "libtest/yatlcon.h" + #include -#if defined(HAVE_CURL_CURL_H) && HAVE_CURL_CURL_H +#if defined(HAVE_LIBCURL) && HAVE_LIBCURL #include #else class CURL; #endif -namespace libtest { -namespace http { -#define YATL_USERAGENT "YATL/1.0" +static void cleanup_curl(void) +{ +#if defined(HAVE_LIBCURL) && HAVE_LIBCURL + curl_global_cleanup(); +#endif +} -extern "C" size_t - http_get_result_callback(void *ptr, size_t size, size_t nmemb, void *data) +static void initialize_curl_startup() +{ +#if defined(HAVE_LIBCURL) && HAVE_LIBCURL + if (curl_global_init(CURL_GLOBAL_ALL)) { - size_t body_size= size * nmemb; - - vchar_t *_body= (vchar_t*)data; + FATAL("curl_global_init(CURL_GLOBAL_ALL) failed"); + } +#endif - _body->resize(size * nmemb); - memcpy(&_body[0], ptr, _body->size()); + if (atexit(cleanup_curl)) + { + FATAL("atexit() failed"); + } +} - return _body->size(); +static pthread_once_t start_key_once= PTHREAD_ONCE_INIT; +static void initialize_curl(void) +{ + int ret; + if ((ret= pthread_once(&start_key_once, initialize_curl_startup)) != 0) + { + FATAL(strerror(ret)); } +} + +namespace libtest { +namespace http { + +#define YATL_USERAGENT "YATL/1.0" + +static size_t http_get_result_callback(void *ptr, size_t size, size_t nmemb, void *data) +{ + vchar_t *_body= (vchar_t*)data; + _body->resize(size * nmemb); + memcpy(&(*_body)[0], ptr, _body->size()); + + return _body->size(); +} static void init(CURL *curl, const std::string& url) { + (void)http_get_result_callback; + (void)curl; + (void)url; +#if defined(HAVE_LIBCURL) && HAVE_LIBCURL if (HAVE_LIBCURL) { -#if defined(HAVE_LIBCURL) && HAVE_LIBCURL assert(curl); curl_easy_setopt(curl, CURLOPT_URL, url.c_str()); curl_easy_setopt(curl, CURLOPT_USERAGENT, YATL_USERAGENT); -#endif } +#endif +} + +HTTP::HTTP(const std::string& url_arg) : + _url(url_arg), + _response(0) +{ + initialize_curl(); } bool GET::execute() { + (void)init; + +#if defined(HAVE_LIBCURL) && HAVE_LIBCURL if (HAVE_LIBCURL) { -#if defined(HAVE_LIBCURL) && HAVE_LIBCURL CURL *curl= curl_easy_init(); init(curl, url()); @@ -75,18 +133,18 @@ bool GET::execute() curl_easy_cleanup(curl); - return retref == CURLE_OK; -#endif + return bool(retref == CURLE_OK); } +#endif return false; } bool POST::execute() { +#if defined(HAVE_LIBCURL) && HAVE_LIBCURL if (HAVE_LIBCURL) { -#if defined(HAVE_LIBCURL) && HAVE_LIBCURL CURL *curl= curl_easy_init();; init(curl, url()); @@ -98,17 +156,19 @@ bool POST::execute() curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, _response); curl_easy_cleanup(curl); -#endif + + return bool(retref == CURLE_OK); } +#endif return false; } bool TRACE::execute() { +#if defined(HAVE_LIBCURL) && HAVE_LIBCURL if (HAVE_LIBCURL) { -#if defined(HAVE_LIBCURL) && HAVE_LIBCURL CURL *curl= curl_easy_init();; init(curl, url()); @@ -123,18 +183,18 @@ bool TRACE::execute() curl_easy_cleanup(curl); return retref == CURLE_OK; -#endif } +#endif return false; } bool HEAD::execute() { +#if defined(HAVE_LIBCURL) && HAVE_LIBCURL if (HAVE_LIBCURL) { -#if defined(HAVE_LIBCURL) && HAVE_LIBCURL - CURL *curl= curl_easy_init();; + CURL *curl= curl_easy_init(); init(curl, url()); @@ -147,8 +207,8 @@ bool HEAD::execute() curl_easy_cleanup(curl); return retref == CURLE_OK; -#endif } +#endif return false; }