mirror of https://github.com/BOINC/boinc.git
212 lines
6.4 KiB
C++
212 lines
6.4 KiB
C++
// This file is part of BOINC.
|
|
// http://boinc.berkeley.edu
|
|
// Copyright (C) 2008 University of California
|
|
//
|
|
// BOINC 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.
|
|
//
|
|
// BOINC 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.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with BOINC. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
// HTTP_OP represents an HTTP operation.
|
|
// There are variants for GET and POST,
|
|
// and for the data source/sink (see below).
|
|
|
|
// We use libcurl: http://curl.haxx.se/libcurl
|
|
|
|
#ifndef _HTTP_CURL_
|
|
#define _HTTP_CURL_
|
|
|
|
#include <curl/curl.h>
|
|
|
|
#include "network.h"
|
|
#include "proxy_info.h"
|
|
|
|
extern int curl_init();
|
|
extern int curl_cleanup();
|
|
|
|
#define HTTP_OP_NONE 0
|
|
#define HTTP_OP_GET 1
|
|
// data sink is a file (used for file download)
|
|
#define HTTP_OP_POST 2
|
|
// data source and sink are files (used for scheduler op)
|
|
#define HTTP_OP_POST2 5
|
|
// a POST operation where the request comes from a combination
|
|
// of a string and a file w/offset,
|
|
// and the reply goes into a memory buffer.
|
|
// Used for file upload
|
|
|
|
#define HTTP_STATE_IDLE 0
|
|
#define HTTP_STATE_CONNECTING 1
|
|
#define HTTP_STATE_DONE 2
|
|
|
|
struct PROJECT;
|
|
|
|
class HTTP_OP {
|
|
public:
|
|
HTTP_OP();
|
|
~HTTP_OP();
|
|
|
|
PROXY_INFO pi;
|
|
PROJECT* project; // associated project, if any
|
|
|
|
char m_url[1024];
|
|
char m_curl_ca_bundle_location[256];
|
|
// string needed for ssl support
|
|
char m_curl_user_credentials[128];
|
|
// string needed for proxy username/password
|
|
|
|
int content_length;
|
|
unsigned int trace_id;
|
|
char request_header[4096];
|
|
|
|
FILE* fileIn;
|
|
FILE* fileOut;
|
|
// output file for POST responses
|
|
CURL* curlEasy;
|
|
// the "easy curl" handle for this net_xfer request
|
|
struct curl_slist *pcurlList;
|
|
// curl slist for http headers
|
|
struct curl_httppost *pcurlFormStart;
|
|
// a pointer to a form item for POST
|
|
struct curl_httppost *pcurlFormEnd;
|
|
// a pointer to a form item for POST
|
|
unsigned char* pByte;
|
|
// pointer to bytes for reading via libcurl_read function
|
|
|
|
// request message stuff
|
|
//
|
|
char* req1;
|
|
// if not NULL, send this string first
|
|
char infile[256];
|
|
// then (is nonempty) this file
|
|
double file_offset;
|
|
// starting at this offset
|
|
|
|
// reply message stuff
|
|
//
|
|
// reply is always written to a file
|
|
char outfile[256];
|
|
// if specified, it's written to this file w/ optional offset
|
|
// otherwise it's written to a temp file
|
|
// if type POST2, copy output to req1 buffer
|
|
int req1_len;
|
|
// with this limit
|
|
|
|
long lSeek;
|
|
// offset within the file or memory buffer we're reading,
|
|
char error_msg[256];
|
|
// put Curl error message here
|
|
bool bTempOutfile;
|
|
// CMC -- flag that outfile is really a tempfile we should delete
|
|
bool bSentHeader;
|
|
// CMC -- a flag that I already sent the header
|
|
CURLcode CurlResult;
|
|
// CMC -- send up curl result code
|
|
|
|
bool want_download; // at most one should be true
|
|
bool want_upload;
|
|
long connect_error;
|
|
// errno from connect() (not used for anything)
|
|
long response;
|
|
// HTTP status code from server
|
|
// the above two MUST be long (not int)
|
|
// otherwise breaks on 64-bit machines
|
|
double start_time;
|
|
double bytes_xferred;
|
|
// Uncompressed bytes transferred.
|
|
// In the case of "post2" this includes only the file part
|
|
// In the case of restartable ops (file upload/download)
|
|
// this includes previous count (i.e. file offset)
|
|
double start_bytes_xferred;
|
|
// bytes_xferred at the start of this operation;
|
|
// used to compute transfer speed
|
|
double xfer_speed;
|
|
// tranfer rate based on elapsed time and bytes_xferred
|
|
// (hence doesn't reflect compression; used only for GUI)
|
|
int http_op_state; // values above
|
|
int http_op_type;
|
|
// HTTP_OP_* (see above)
|
|
int http_op_retval;
|
|
// Either:
|
|
// 0
|
|
// ERR_GETHOSTBYNAME (if no such host)
|
|
// ERR_CONNECT (if server down)
|
|
// ERR_HTTP_PERMANENT (if 404 or 416)
|
|
// ERR_HTTP_TRANSIENT (other failures)
|
|
bool is_background;
|
|
// this operation shouldn't trigger a reference-site check
|
|
// and a user alert if it fails.
|
|
// For example: notice RSS feed fetches
|
|
|
|
void reset();
|
|
void init(PROJECT*);
|
|
int get_ip_addr(int &ip_addr);
|
|
void close_socket();
|
|
void close_file();
|
|
void update_speed();
|
|
void set_speed_limit(bool is_upload, double bytes_sec);
|
|
void handle_messages(CURLMsg*);
|
|
|
|
//int init_head(const char* url);
|
|
int init_get(
|
|
PROJECT*, const char* url, const char* outfile,
|
|
bool del_old_file, double offset, double size
|
|
);
|
|
int init_post(
|
|
PROJECT*, const char* url, const char* infile, const char* outfile
|
|
);
|
|
int init_post2(
|
|
PROJECT*,
|
|
const char* url,
|
|
char* req1, // first part of request. ALSO USED FOR REPLY
|
|
int req1_len,
|
|
const char* infile, double offset // infile is NULL if no file sent
|
|
);
|
|
bool http_op_done();
|
|
void setup_proxy_session(bool no_proxy);
|
|
bool no_proxy_for_url(const char* url);
|
|
bool is_active() {
|
|
return curlEasy!=NULL;
|
|
}
|
|
|
|
private:
|
|
// take an init_get/post/post2 and turns it into a libcurl request
|
|
//
|
|
int libcurl_exec(const char* url, const char* in, const char* out,
|
|
double offset, double size, bool is_post
|
|
);
|
|
};
|
|
|
|
// represents a set of HTTP requests in progress
|
|
|
|
class HTTP_OP_SET {
|
|
std::vector<HTTP_OP*> http_ops;
|
|
public:
|
|
HTTP_OP_SET();
|
|
void insert(HTTP_OP*);
|
|
int remove(HTTP_OP*);
|
|
int nops();
|
|
|
|
double bytes_up, bytes_down;
|
|
// total bytes transferred
|
|
|
|
void get_fdset(FDSET_GROUP&);
|
|
void got_select(FDSET_GROUP&, double);
|
|
HTTP_OP* lookup_curl(CURL* pcurl);
|
|
// lookup by easycurl handle
|
|
void cleanup_temp_files();
|
|
|
|
};
|
|
|
|
extern char* get_user_agent_string();
|
|
|
|
#endif //__HTTP_CURL_
|