6 #include "rgw_torrent.h"
7 #include "include/str_list.h"
8 #include "include/rados/librados.hpp"
10 #define dout_subsys ceph_subsys_rgw
13 using ceph::crypto::MD5;
14 using namespace librados;
15 using namespace boost;
16 using ceph::crypto::SHA1;
20 seed::info.piece_length = 0;
28 seed::info.sha1_bl.clear();
34 void seed::init(struct req_state *p_req, RGWRados *p_store)
40 int seed::get_torrent_file(RGWRados::Object::Read &read_op,
42 ceph::bufferlist &bl_data,
45 /* add other field if config is set */
46 dencode.bencode_dict(bl);
50 dencode.bencode(COMMENT, comment, bl);
52 if (!create_by.empty())
54 dencode.bencode(CREATED_BY, create_by, bl);
56 if (!encoding.empty())
58 dencode.bencode(ENCODING, encoding, bl);
62 map<string, bufferlist> m;
64 get_obj_bucket_and_oid_loc(obj, oid, key);
65 ldout(s->cct, 0) << "NOTICE: head obj oid= " << oid << dendl;
67 obj_key.insert(RGW_OBJ_TORRENT);
68 const int op_ret = read_op.state.io_ctx.omap_get_vals_by_keys(oid, obj_key, &m);
71 ldout(s->cct, 0) << "ERROR: failed to omap_get_vals_by_keys op_ret = "
76 map<string, bufferlist>::iterator iter;
77 for (iter = m.begin(); iter != m.end(); ++iter)
79 bufferlist bl_tmp = iter->second;
80 char *pbuff = bl_tmp.c_str();
81 bl.append(pbuff, bl_tmp.length());
83 dencode.bencode_end(bl);
86 total_len = bl.length();
95 void seed::update(bufferlist &bl)
101 info.len += bl.length();
102 sha1(&h, bl, bl.length());
107 uint64_t remain = info.len%info.piece_length;
108 uint8_t remain_len = ((remain > 0)? 1 : 0);
109 sha_len = (info.len/info.piece_length + remain_len)*CEPH_CRYPTO_SHA1_DIGESTSIZE;
112 /* produce torrent data */
115 /* save torrent data into OMAP */
116 ret = save_torrent_file();
119 ldout(s->cct, 0) << "ERROR: failed to save_torrent_file() ret= "<< ret << dendl;
126 off_t seed::get_data_len()
131 void seed::set_create_date(ceph::real_time& value)
133 utime_t date = ceph::real_clock::to_timespec(value);
134 create_date = date.sec();
137 void seed::set_info_pieces(char *buff)
139 info.sha1_bl.append(buff, CEPH_CRYPTO_SHA1_DIGESTSIZE);
142 void seed::set_info_name(const string& value)
147 void seed::sha1(SHA1 *h, bufferlist &bl, off_t bl_len)
149 off_t num = bl_len/info.piece_length;
151 remain = bl_len%info.piece_length;
153 char *pstr = bl.c_str();
157 for (off_t i = 0; i < num; i++)
159 memset(sha, 0x00, sizeof(sha));
160 h->Update((byte *)pstr, info.piece_length);
161 h->Final((byte *)sha);
162 set_info_pieces(sha);
163 pstr += info.piece_length;
169 memset(sha, 0x00, sizeof(sha));
170 h->Update((byte *)pstr, remain);
171 h->Final((byte *)sha);
172 set_info_pieces(sha);
176 int seed::get_params()
179 info.piece_length = g_conf->rgw_torrent_sha_unit;
180 create_by = g_conf->rgw_torrent_createby;
181 encoding = g_conf->rgw_torrent_encoding;
182 origin = g_conf->rgw_torrent_origin;
183 comment = g_conf->rgw_torrent_comment;
184 announce = g_conf->rgw_torrent_tracker;
186 /* tracker and tracker list is empty, set announce to origin */
187 if (announce.empty() && !origin.empty())
195 void seed::set_announce()
197 list<string> announce_list; // used to get announce list from conf
198 get_str_list(announce, ",", announce_list);
200 if (announce_list.empty())
202 ldout(s->cct, 5) << "NOTICE: announce_list is empty " << dendl;
206 list<string>::iterator iter = announce_list.begin();
207 dencode.bencode_key(ANNOUNCE, bl);
208 dencode.bencode_key((*iter), bl);
210 dencode.bencode_key(ANNOUNCE_LIST, bl);
211 dencode.bencode_list(bl);
212 for (; iter != announce_list.end(); ++iter)
214 dencode.bencode_list(bl);
215 dencode.bencode_key((*iter), bl);
216 dencode.bencode_end(bl);
218 dencode.bencode_end(bl);
221 void seed::do_encode()
223 /*Only encode create_date and sha1 info*/
224 /*Other field will be added if confi is set when run get torrent*/
225 dencode.bencode(CREATION_DATE, create_date, bl);
227 dencode.bencode_key(INFO_PIECES, bl);
228 dencode.bencode_dict(bl);
229 dencode.bencode(LENGTH, info.len, bl);
230 dencode.bencode(NAME, info.name, bl);
231 dencode.bencode(PIECE_LENGTH, info.piece_length, bl);
233 char info_sha[100] = { 0 };
234 sprintf(info_sha, "%" PRIu64, sha_len);
235 string sha_len_str = info_sha;
236 dencode.bencode_key(PIECES, bl);
237 bl.append(sha_len_str.c_str(), sha_len_str.length());
239 bl.append(info.sha1_bl.c_str(), sha_len);
240 dencode.bencode_end(bl);
243 int seed::save_torrent_file()
246 string key = RGW_OBJ_TORRENT;
247 rgw_obj obj(s->bucket, s->object.name);
250 store->obj_to_raw(s->bucket_info.placement_rule, obj, &raw_obj);
252 op_ret = store->omap_set(raw_obj, key, bl);
255 ldout(s->cct, 0) << "ERROR: failed to omap_set() op_ret = " << op_ret << dendl;