1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab
#ifndef CEPH_RGW_REST_CLIENT_H
#define CEPH_RGW_REST_CLIENT_H
#include "rgw_http_client.h"
class RGWGetDataCB;
class RGWRESTSimpleRequest : public RGWHTTPClient {
protected:
int http_status;
int status;
string url;
map<string, string> out_headers;
param_vec_t params;
bufferlist::iterator *send_iter;
size_t max_response; /* we need this as we don't stream out response */
bufferlist response;
virtual int handle_header(const string& name, const string& val);
void append_param(string& dest, const string& name, const string& val);
void get_params_str(map<string, string>& extra_args, string& dest);
int sign_request(RGWAccessKey& key, RGWEnv& env, req_info& info);
public:
RGWRESTSimpleRequest(CephContext *_cct, const string& _url, param_vec_t *_headers,
param_vec_t *_params) : RGWHTTPClient(_cct), http_status(0), status(0),
url(_url), send_iter(NULL),
max_response(0) {
set_headers(_headers);
set_params(_params);
}
void set_headers(param_vec_t *_headers) {
if (_headers)
headers = *_headers;
}
void set_params(param_vec_t *_params) {
if (_params)
params = *_params;
}
int receive_header(void *ptr, size_t len) override;
int receive_data(void *ptr, size_t len) override;
int send_data(void *ptr, size_t len) override;
bufferlist& get_response() { return response; }
int execute(RGWAccessKey& key, const char *method, const char *resource);
int forward_request(RGWAccessKey& key, req_info& info, size_t max_response, bufferlist *inbl, bufferlist *outbl);
map<string, string>& get_out_headers() { return out_headers; }
int get_http_status() { return http_status; }
int get_status();
};
class RGWRESTStreamWriteRequest : public RGWRESTSimpleRequest {
Mutex lock;
list<bufferlist> pending_send;
RGWGetDataCB *cb;
RGWHTTPManager http_manager;
public:
int add_output_data(bufferlist& bl);
int send_data(void *ptr, size_t len) override;
RGWRESTStreamWriteRequest(CephContext *_cct, const string& _url, param_vec_t *_headers,
param_vec_t *_params) : RGWRESTSimpleRequest(_cct, _url, _headers, _params),
lock("RGWRESTStreamWriteRequest"), cb(NULL), http_manager(_cct) {}
~RGWRESTStreamWriteRequest() override;
int put_obj_init(RGWAccessKey& key, rgw_obj& obj, uint64_t obj_size, map<string, bufferlist>& attrs);
int complete(string& etag, real_time *mtime);
RGWGetDataCB *get_out_cb() { return cb; }
};
class RGWRESTStreamRWRequest : public RGWRESTSimpleRequest {
Mutex lock;
RGWGetDataCB *cb;
bufferlist outbl;
bufferlist in_data;
size_t chunk_ofs;
size_t ofs;
RGWHTTPManager http_manager;
const char *method;
uint64_t write_ofs;
protected:
int handle_header(const string& name, const string& val) override;
public:
int send_data(void *ptr, size_t len) override;
int receive_data(void *ptr, size_t len) override;
RGWRESTStreamRWRequest(CephContext *_cct, const char *_method, const string& _url, RGWGetDataCB *_cb,
param_vec_t *_headers, param_vec_t *_params) : RGWRESTSimpleRequest(_cct, _url, _headers, _params),
lock("RGWRESTStreamReadRequest"), cb(_cb),
chunk_ofs(0), ofs(0), http_manager(_cct), method(_method), write_ofs(0) {
}
virtual ~RGWRESTStreamRWRequest() override {}
int send_request(RGWAccessKey& key, map<string, string>& extra_headers, rgw_obj& obj, RGWHTTPManager *mgr = NULL);
int send_request(RGWAccessKey *key, map<string, string>& extra_headers, const string& resource, bufferlist *send_data = NULL /* optional input data */, RGWHTTPManager *mgr = NULL);
int complete_request(string& etag, real_time *mtime, uint64_t *psize, map<string, string>& attrs);
void set_outbl(bufferlist& _outbl) {
outbl.swap(_outbl);
}
void set_in_cb(RGWGetDataCB *_cb) { cb = _cb; }
};
class RGWRESTStreamReadRequest : public RGWRESTStreamRWRequest {
public:
RGWRESTStreamReadRequest(CephContext *_cct, const string& _url, RGWGetDataCB *_cb, param_vec_t *_headers,
param_vec_t *_params) : RGWRESTStreamRWRequest(_cct, "GET", _url, _cb, _headers, _params) {}
};
class RGWRESTStreamHeadRequest : public RGWRESTStreamRWRequest {
public:
RGWRESTStreamHeadRequest(CephContext *_cct, const string& _url, RGWGetDataCB *_cb, param_vec_t *_headers,
param_vec_t *_params) : RGWRESTStreamRWRequest(_cct, "HEAD", _url, _cb, _headers, _params) {}
};
#endif
|
