diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 18:45:59 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 18:45:59 +0000 |
commit | 19fcec84d8d7d21e796c7624e521b60d28ee21ed (patch) | |
tree | 42d26aa27d1e3f7c0b8bd3fd14e7d7082f5008dc /src/rgw/rgw_torrent.h | |
parent | Initial commit. (diff) | |
download | ceph-upstream.tar.xz ceph-upstream.zip |
Adding upstream version 16.2.11+ds.upstream/16.2.11+dsupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/rgw/rgw_torrent.h')
-rw-r--r-- | src/rgw/rgw_torrent.h | 142 |
1 files changed, 142 insertions, 0 deletions
diff --git a/src/rgw/rgw_torrent.h b/src/rgw/rgw_torrent.h new file mode 100644 index 000000000..995923668 --- /dev/null +++ b/src/rgw/rgw_torrent.h @@ -0,0 +1,142 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab ft=cpp + +#ifndef CEPH_RGW_TORRENT_H +#define CEPH_RGW_TORRENT_H + +#include <string> +#include <list> +#include <map> +#include <set> + +#include "common/ceph_time.h" + +#include "rgw_rados.h" +#include "rgw_common.h" + +using ceph::crypto::SHA1; + +struct req_state; + +#define RGW_OBJ_TORRENT "rgw.torrent" + +#define ANNOUNCE "announce" +#define ANNOUNCE_LIST "announce-list" +#define COMMENT "comment" +#define CREATED_BY "created by" +#define CREATION_DATE "creation date" +#define ENCODING "encoding" +#define LENGTH "length" +#define NAME "name" +#define PIECE_LENGTH "piece length" +#define PIECES "pieces" +#define INFO_PIECES "info" +#define GET_TORRENT "torrent" + +class TorrentBencode +{ +public: + TorrentBencode() {} + ~TorrentBencode() {} + + //control characters + void bencode_dict(bufferlist& bl) { bl.append('d'); } + void bencode_list(bufferlist& bl) { bl.append('l'); } + void bencode_end(bufferlist& bl) { bl.append('e'); } + + //single values + void bencode(int value, bufferlist& bl) + { + bl.append('i'); + char info[100] = { 0 }; + sprintf(info, "%d", value); + bl.append(info, strlen(info)); + bencode_end(bl); + } + + //single values + void bencode(const std::string& str, bufferlist& bl) + { + bencode_key(str, bl); + } + + //dictionary elements + void bencode(const std::string& key, int value, bufferlist& bl) + { + bencode_key(key, bl); + bencode(value, bl); + } + + //dictionary elements + void bencode(const std::string& key, const std::string& value, bufferlist& bl) + { + bencode_key(key, bl); + bencode(value, bl); + } + + //key len + void bencode_key(const std::string& key, bufferlist& bl) + { + int len = key.length(); + char info[100] = { 0 }; + sprintf(info, "%d:", len); + bl.append(info, strlen(info)); + bl.append(key.c_str(), len); + } +}; + +/* torrent file struct */ +class seed +{ +private: + struct + { + int piece_length; // each piece length + bufferlist sha1_bl; // save sha1 + string name; // file name + off_t len; // file total bytes + }info; + + string announce; // tracker + string origin; // origin + time_t create_date{0}; // time of the file created + string comment; // comment + string create_by; // app name and version + string encoding; // if encode use gbk rather than gtf-8 use this field + uint64_t sha_len; // sha1 length + bool is_torrent; // flag + bufferlist bl; // bufflist ready to send + + struct req_state *s{nullptr}; + rgw::sal::RGWRadosStore *store{nullptr}; + SHA1 h; + + TorrentBencode dencode; +public: + seed(); + ~seed(); + + int get_params(); + void init(struct req_state *p_req, rgw::sal::RGWRadosStore *p_store); + int get_torrent_file(rgw::sal::RGWObject* object, + uint64_t &total_len, + ceph::bufferlist &bl_data, + rgw_obj &obj); + + off_t get_data_len(); + bool get_flag(); + + void set_create_date(ceph::real_time& value); + void set_info_name(const string& value); + void update(bufferlist &bl); + int complete(optional_yield y); + +private: + void do_encode (); + void set_announce(); + void set_exist(bool exist); + void set_info_pieces(char *buff); + void sha1(SHA1 *h, bufferlist &bl, off_t bl_len); + int save_torrent_file(optional_yield y); +}; +#endif /* CEPH_RGW_TORRENT_H */ |