6 #include "XrdVersion.hh"
10 #include <curl/curl.h>
20 curl_slist_free_all(m_headers);
22 if (m_curl) {curl_easy_setopt(m_curl, CURLOPT_HTTPHEADER, m_headers);}
29 m_push = other.m_push;
30 m_recv_status_line = other.m_recv_status_line;
31 m_recv_all_headers = other.m_recv_all_headers;
32 m_offset = other.m_offset;
33 m_start_offset = other.m_start_offset;
34 m_status_code = other.m_status_code;
35 m_content_length = other.m_content_length;
36 m_stream = other.m_stream;
37 m_curl = other.m_curl;
38 m_headers = other.m_headers;
39 m_headers_copy = other.m_headers_copy;
40 m_resp_protocol = other.m_resp_protocol;
41 m_is_transfer_state = other.m_is_transfer_state;
42 curl_easy_setopt(m_curl, CURLOPT_HEADERDATA,
this);
43 if (m_is_transfer_state) {
45 curl_easy_setopt(m_curl, CURLOPT_READDATA,
this);
47 curl_easy_setopt(m_curl, CURLOPT_WRITEDATA,
this);
50 tpcForwardCreds = other.tpcForwardCreds;
51 other.m_headers_copy.clear();
53 other.m_headers = NULL;
54 other.m_stream = NULL;
58 bool State::InstallHandlers(
CURL *curl) {
59 curl_easy_setopt(curl, CURLOPT_USERAGENT,
"xrootd-tpc/" XrdVERSION);
60 curl_easy_setopt(curl, CURLOPT_HEADERFUNCTION, &State::HeaderCB);
61 curl_easy_setopt(curl, CURLOPT_HEADERDATA,
this);
62 if(m_is_transfer_state) {
64 curl_easy_setopt(curl, CURLOPT_UPLOAD, 1);
65 curl_easy_setopt(curl, CURLOPT_READFUNCTION, &State::ReadCB);
66 curl_easy_setopt(curl, CURLOPT_READDATA,
this);
69 curl_easy_setopt(curl, CURLOPT_INFILESIZE_LARGE, buf.st_size);
72 curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, &State::WriteCB);
73 curl_easy_setopt(curl, CURLOPT_WRITEDATA,
this);
76 curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
78 curl_easy_setopt(curl,CURLOPT_UNRESTRICTED_AUTH,1L);
83 curl_version_info_data *curl_ver = curl_version_info(CURLVERSION_NOW);
84 if (curl_ver->age > 0 && curl_ver->version_num >= 0x072600) {
86 curl_easy_setopt(curl, CURLOPT_LOW_SPEED_TIME, 2*60);
87 curl_easy_setopt(curl, CURLOPT_LOW_SPEED_LIMIT, 10*1024);
96 struct curl_slist *list = NULL;
97 for (std::map<std::string, std::string>::const_iterator hdr_iter = req.
headers.begin();
100 if (!strcasecmp(hdr_iter->first.c_str(),
"copy-header")) {
101 list = curl_slist_append(list, hdr_iter->second.c_str());
102 m_headers_copy.emplace_back(hdr_iter->second);
105 if (!strncasecmp(hdr_iter->first.c_str(),
"transferheader",14)) {
106 std::stringstream ss;
107 ss << hdr_iter->first.substr(14) <<
": " << hdr_iter->second;
108 list = curl_slist_append(list, ss.str().c_str());
109 m_headers_copy.emplace_back(ss.str());
113 curl_easy_setopt(m_curl, CURLOPT_HTTPHEADER, list);
121 m_content_length = -1;
122 m_recv_all_headers =
false;
123 m_recv_status_line =
false;
126 size_t State::HeaderCB(
char *buffer,
size_t size,
size_t nitems,
void *userdata)
129 std::string header(buffer, size*nitems);
130 return obj->Header(header);
133 int State::Header(
const std::string &header) {
135 if (m_recv_all_headers) {
136 m_recv_all_headers =
false;
137 m_recv_status_line =
false;
139 if (!m_recv_status_line) {
140 std::stringstream ss(header);
143 m_resp_protocol = item;
147 m_status_code = std::stol(item);
151 m_recv_status_line =
true;
152 }
else if (header.size() == 0 || header ==
"\n" || header ==
"\r\n") {
153 m_recv_all_headers =
true;
155 else if (header !=
"\r\n") {
157 std::size_t found = header.find(
":");
158 if (found != std::string::npos) {
159 std::string header_name = header.substr(0, found);
160 std::transform(header_name.begin(), header_name.end(), header_name.begin(), ::tolower);
161 std::string header_value = header.substr(found+1);
162 if (header_name ==
"content-length")
165 m_content_length = std::stoll(header_value);
179 return header.size();
182 size_t State::WriteCB(
void *buffer,
size_t size,
size_t nitems,
void *userdata) {
188 obj->m_error_buf += std::string(
static_cast<char*
>(buffer),
189 std::min(
static_cast<size_t>(1024), size*nitems));
191 if (obj->m_error_buf.size() >= 1024)
196 return obj->Write(
static_cast<char*
>(buffer), size*nitems);
199 ssize_t State::Write(
char *buffer,
size_t size) {
200 ssize_t retval = m_stream->
Write(m_start_offset + m_offset, buffer, size,
false);
215 ssize_t retval = m_stream->
Write(m_start_offset + m_offset, 0, 0,
true);
225 size_t State::ReadCB(
void *buffer,
size_t size,
size_t nitems,
void *userdata) {
229 return obj->Read(
static_cast<char*
>(buffer), size*nitems);
232 int State::Read(
char *buffer,
size_t size) {
233 int retval = m_stream->
Read(m_start_offset + m_offset, buffer, size);
243 CURL *curl = curl_easy_duphandle(m_curl);
245 throw std::runtime_error(
"Failed to duplicate existing curl handle.");
248 State *state =
new State(0, *m_stream, curl, m_push, tpcForwardCreds);
251 state->m_headers_copy.reserve(m_headers_copy.size());
252 for (std::vector<std::string>::const_iterator header_iter = m_headers_copy.begin();
253 header_iter != m_headers_copy.end();
255 state->m_headers = curl_slist_append(state->m_headers, header_iter->c_str());
256 state->m_headers_copy.push_back(*header_iter);
258 curl_easy_setopt(curl, CURLOPT_HTTPHEADER, NULL);
259 curl_easy_setopt(curl, CURLOPT_HTTPHEADER, state->m_headers);
266 m_start_offset = offset;
268 m_content_length = size;
269 std::stringstream ss;
270 ss << offset <<
"-" << (offset+size-1);
271 curl_easy_setopt(m_curl, CURLOPT_RANGE, ss.str().c_str());
298 #if LIBCURL_VERSION_NUM >= 0x071500
299 char *curl_ip = NULL;
300 CURLcode rc = curl_easy_getinfo(m_curl, CURLINFO_PRIMARY_IP, &curl_ip);
301 if ((rc != CURLE_OK) || !curl_ip) {
305 rc = curl_easy_getinfo(m_curl, CURLINFO_PRIMARY_PORT, &curl_port);
306 if ((rc != CURLE_OK) || !curl_port) {
309 std::stringstream ss;
315 if (NULL == strchr(curl_ip,
':'))
316 ss <<
"tcp:" << curl_ip <<
":" << curl_port;
318 ss <<
"tcp:[" << curl_ip <<
"]:" << curl_port;
int stat(const char *path, struct stat *buf)
void getline(uchar *buff, int blen)
int GetStatusCode() const
void CopyHeaders(XrdHttpExtReq &req)
void SetTransferParameters(off_t offset, size_t size)
std::string GetConnectionDescription()
int AvailableBuffers() const
int Read(off_t offset, char *buffer, size_t size)
ssize_t Write(off_t offset, const char *buffer, size_t size, bool force)
std::string GetErrorMessage() const
size_t AvailableBuffers() const
std::map< std::string, std::string > & headers