1 // Copyright (c) 2009, Google Inc.
2 // All rights reserved.
3 //
4 // Redistribution and use in source and binary forms, with or without
5 // modification, are permitted provided that the following conditions are
6 // met:
7 //
8 //     * Redistributions of source code must retain the above copyright
9 // notice, this list of conditions and the following disclaimer.
10 //     * Redistributions in binary form must reproduce the above
11 // copyright notice, this list of conditions and the following disclaimer
12 // in the documentation and/or other materials provided with the
13 // distribution.
14 //     * Neither the name of Google Inc. nor the names of its
15 // contributors may be used to endorse or promote products derived from
16 // this software without specific prior written permission.
17 //
18 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 
30 #include <dlfcn.h>
31 
32 #include <iostream>
33 #include <string>
34 
35 #include "common/linux/libcurl_wrapper.h"
36 #include "common/using_std_string.h"
37 
38 namespace google_breakpad {
LibcurlWrapper()39 LibcurlWrapper::LibcurlWrapper()
40     : init_ok_(false),
41       formpost_(NULL),
42       lastptr_(NULL),
43       headerlist_(NULL) {
44   curl_lib_ = dlopen("libcurl.so", RTLD_NOW);
45   if (!curl_lib_) {
46     curl_lib_ = dlopen("libcurl.so.4", RTLD_NOW);
47   }
48   if (!curl_lib_) {
49     curl_lib_ = dlopen("libcurl.so.3", RTLD_NOW);
50   }
51   if (!curl_lib_) {
52     std::cout << "Could not find libcurl via dlopen";
53     return;
54   }
55   std::cout << "LibcurlWrapper init succeeded";
56   init_ok_ = true;
57   return;
58 }
59 
~LibcurlWrapper()60 LibcurlWrapper::~LibcurlWrapper() {}
61 
SetProxy(const string & proxy_host,const string & proxy_userpwd)62 bool LibcurlWrapper::SetProxy(const string& proxy_host,
63                               const string& proxy_userpwd) {
64   if (!init_ok_) {
65     return false;
66   }
67   // Set proxy information if necessary.
68   if (!proxy_host.empty()) {
69     (*easy_setopt_)(curl_, CURLOPT_PROXY, proxy_host.c_str());
70   } else {
71     std::cout << "SetProxy called with empty proxy host.";
72     return false;
73   }
74   if (!proxy_userpwd.empty()) {
75     (*easy_setopt_)(curl_, CURLOPT_PROXYUSERPWD, proxy_userpwd.c_str());
76   } else {
77     std::cout << "SetProxy called with empty proxy username/password.";
78     return false;
79   }
80   std::cout << "Set proxy host to " << proxy_host;
81   return true;
82 }
83 
AddFile(const string & upload_file_path,const string & basename)84 bool LibcurlWrapper::AddFile(const string& upload_file_path,
85                              const string& basename) {
86   if (!init_ok_) {
87     return false;
88   }
89   std::cout << "Adding " << upload_file_path << " to form upload.";
90   // Add form file.
91   (*formadd_)(&formpost_, &lastptr_,
92               CURLFORM_COPYNAME, basename.c_str(),
93               CURLFORM_FILE, upload_file_path.c_str(),
94               CURLFORM_END);
95 
96   return true;
97 }
98 
99 // Callback to get the response data from server.
WriteCallback(void * ptr,size_t size,size_t nmemb,void * userp)100 static size_t WriteCallback(void *ptr, size_t size,
101                             size_t nmemb, void *userp) {
102   if (!userp)
103     return 0;
104 
105   string *response = reinterpret_cast<string *>(userp);
106   size_t real_size = size * nmemb;
107   response->append(reinterpret_cast<char *>(ptr), real_size);
108   return real_size;
109 }
110 
SendRequest(const string & url,const std::map<string,string> & parameters,int * http_status_code,string * http_header_data,string * http_response_data)111 bool LibcurlWrapper::SendRequest(const string& url,
112                                  const std::map<string, string>& parameters,
113                                  int* http_status_code,
114                                  string* http_header_data,
115                                  string* http_response_data) {
116   (*easy_setopt_)(curl_, CURLOPT_URL, url.c_str());
117   std::map<string, string>::const_iterator iter = parameters.begin();
118   for (; iter != parameters.end(); ++iter)
119     (*formadd_)(&formpost_, &lastptr_,
120                 CURLFORM_COPYNAME, iter->first.c_str(),
121                 CURLFORM_COPYCONTENTS, iter->second.c_str(),
122                 CURLFORM_END);
123 
124   (*easy_setopt_)(curl_, CURLOPT_HTTPPOST, formpost_);
125   if (http_response_data != NULL) {
126     http_response_data->clear();
127     (*easy_setopt_)(curl_, CURLOPT_WRITEFUNCTION, WriteCallback);
128     (*easy_setopt_)(curl_, CURLOPT_WRITEDATA,
129                      reinterpret_cast<void *>(http_response_data));
130   }
131   if (http_header_data != NULL) {
132     http_header_data->clear();
133     (*easy_setopt_)(curl_, CURLOPT_HEADERFUNCTION, WriteCallback);
134     (*easy_setopt_)(curl_, CURLOPT_HEADERDATA,
135                      reinterpret_cast<void *>(http_header_data));
136   }
137 
138   CURLcode err_code = CURLE_OK;
139   err_code = (*easy_perform_)(curl_);
140   easy_strerror_ = reinterpret_cast<const char* (*)(CURLcode)>
141                        (dlsym(curl_lib_, "curl_easy_strerror"));
142 
143   if (http_status_code != NULL) {
144     (*easy_getinfo_)(curl_, CURLINFO_RESPONSE_CODE, http_status_code);
145   }
146 
147 #ifndef NDEBUG
148   if (err_code != CURLE_OK)
149     fprintf(stderr, "Failed to send http request to %s, error: %s\n",
150             url.c_str(),
151             (*easy_strerror_)(err_code));
152 #endif
153   if (headerlist_ != NULL) {
154     (*slist_free_all_)(headerlist_);
155   }
156 
157   (*easy_cleanup_)(curl_);
158   if (formpost_ != NULL) {
159     (*formfree_)(formpost_);
160   }
161 
162   return err_code == CURLE_OK;
163 }
164 
Init()165 bool LibcurlWrapper::Init() {
166   if (!init_ok_) {
167     std::cout << "Init_OK was not true in LibcurlWrapper::Init(), check earlier log messages";
168     return false;
169   }
170 
171   if (!SetFunctionPointers()) {
172     std::cout << "Could not find function pointers";
173     init_ok_ = false;
174     return false;
175   }
176 
177   curl_ = (*easy_init_)();
178 
179   last_curl_error_ = "No Error";
180 
181   if (!curl_) {
182     dlclose(curl_lib_);
183     std::cout << "Curl initialization failed";
184     return false;
185   }
186 
187   // Disable 100-continue header.
188   char buf[] = "Expect:";
189 
190   headerlist_ = (*slist_append_)(headerlist_, buf);
191   (*easy_setopt_)(curl_, CURLOPT_HTTPHEADER, headerlist_);
192   return true;
193 }
194 
195 #define SET_AND_CHECK_FUNCTION_POINTER(var, function_name, type) \
196   var = reinterpret_cast<type>(dlsym(curl_lib_, function_name)); \
197   if (!var) { \
198     std::cout << "Could not find libcurl function " << function_name; \
199     init_ok_ = false; \
200     return false; \
201   }
202 
SetFunctionPointers()203 bool LibcurlWrapper::SetFunctionPointers() {
204 
205   SET_AND_CHECK_FUNCTION_POINTER(easy_init_,
206                                  "curl_easy_init",
207                                  CURL*(*)());
208 
209   SET_AND_CHECK_FUNCTION_POINTER(easy_setopt_,
210                                  "curl_easy_setopt",
211                                  CURLcode(*)(CURL*, CURLoption, ...));
212 
213   SET_AND_CHECK_FUNCTION_POINTER(formadd_, "curl_formadd",
214       CURLFORMcode(*)(curl_httppost**, curl_httppost**, ...));
215 
216   SET_AND_CHECK_FUNCTION_POINTER(slist_append_, "curl_slist_append",
217       curl_slist*(*)(curl_slist*, const char*));
218 
219   SET_AND_CHECK_FUNCTION_POINTER(easy_perform_,
220                                  "curl_easy_perform",
221                                  CURLcode(*)(CURL*));
222 
223   SET_AND_CHECK_FUNCTION_POINTER(easy_cleanup_,
224                                  "curl_easy_cleanup",
225                                  void(*)(CURL*));
226 
227   SET_AND_CHECK_FUNCTION_POINTER(easy_getinfo_,
228                                  "curl_easy_getinfo",
229                                  CURLcode(*)(CURL *, CURLINFO info, ...));
230 
231   SET_AND_CHECK_FUNCTION_POINTER(slist_free_all_,
232                                  "curl_slist_free_all",
233                                  void(*)(curl_slist*));
234 
235   SET_AND_CHECK_FUNCTION_POINTER(formfree_,
236                                  "curl_formfree",
237                                  void(*)(curl_httppost*));
238   return true;
239 }
240 
241 }
242