]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | #ifndef CEPH_RGW_TORRENT_H |
2 | #define CEPH_RGW_TORRENT_H | |
3 | ||
4 | #include <string> | |
5 | #include <list> | |
6 | #include <map> | |
7 | #include <set> | |
8 | ||
9 | #include "common/ceph_time.h" | |
10 | ||
11 | #include "rgw_rados.h" | |
12 | #include "rgw_common.h" | |
13 | ||
7c673cae FG |
14 | using ceph::crypto::SHA1; |
15 | ||
16 | struct req_state; | |
17 | ||
18 | #define RGW_OBJ_TORRENT "rgw.torrent" | |
19 | ||
20 | #define ANNOUNCE "announce" | |
21 | #define ANNOUNCE_LIST "announce-list" | |
22 | #define COMMENT "comment" | |
23 | #define CREATED_BY "created by" | |
24 | #define CREATION_DATE "creation date" | |
25 | #define ENCODING "encoding" | |
26 | #define LENGTH "length" | |
27 | #define NAME "name" | |
28 | #define PIECE_LENGTH "piece length" | |
29 | #define PIECES "pieces" | |
30 | #define INFO_PIECES "info" | |
31 | #define GET_TORRENT "torrent" | |
32 | ||
33 | class TorrentBencode | |
34 | { | |
35 | public: | |
36 | TorrentBencode() {} | |
37 | ~TorrentBencode() {} | |
38 | ||
39 | //control characters | |
40 | void bencode_dict(bufferlist& bl) { bl.append('d'); } | |
41 | void bencode_list(bufferlist& bl) { bl.append('l'); } | |
42 | void bencode_end(bufferlist& bl) { bl.append('e'); } | |
43 | ||
44 | //single values | |
45 | void bencode(int value, bufferlist& bl) | |
46 | { | |
47 | bl.append('i'); | |
48 | char info[100] = { 0 }; | |
49 | sprintf(info, "%d", value); | |
50 | bl.append(info, strlen(info)); | |
51 | bencode_end(bl); | |
52 | } | |
53 | ||
54 | //single values | |
55 | void bencode(const std::string& str, bufferlist& bl) | |
56 | { | |
57 | bencode_key(str, bl); | |
58 | } | |
59 | ||
60 | //dictionary elements | |
61 | void bencode(const std::string& key, int value, bufferlist& bl) | |
62 | { | |
63 | bencode_key(key, bl); | |
64 | bencode(value, bl); | |
65 | } | |
66 | ||
67 | //dictionary elements | |
68 | void bencode(const std::string& key, const std::string& value, bufferlist& bl) | |
69 | { | |
70 | bencode_key(key, bl); | |
71 | bencode(value, bl); | |
72 | } | |
73 | ||
74 | //key len | |
75 | void bencode_key(const std::string& key, bufferlist& bl) | |
76 | { | |
77 | int len = key.length(); | |
78 | char info[100] = { 0 }; | |
79 | sprintf(info, "%d:", len); | |
80 | bl.append(info, strlen(info)); | |
81 | bl.append(key.c_str(), len); | |
82 | } | |
83 | }; | |
84 | ||
85 | /* torrent file struct */ | |
86 | class seed | |
87 | { | |
88 | private: | |
89 | struct | |
90 | { | |
91 | int piece_length; // each piece length | |
92 | bufferlist sha1_bl; // save sha1 | |
93 | string name; // file name | |
94 | off_t len; // file total bytes | |
95 | }info; | |
96 | ||
97 | string announce; // tracker | |
98 | string origin; // origin | |
224ce89b | 99 | time_t create_date{0}; // time of the file created |
7c673cae FG |
100 | string comment; // comment |
101 | string create_by; // app name and version | |
102 | string encoding; // if encode use gbk rather than gtf-8 use this field | |
103 | uint64_t sha_len; // sha1 length | |
104 | bool is_torrent; // flag | |
105 | bufferlist bl; // bufflist ready to send | |
7c673cae | 106 | |
224ce89b WB |
107 | struct req_state *s{nullptr}; |
108 | RGWRados *store{nullptr}; | |
31f18b77 | 109 | SHA1 h; |
7c673cae FG |
110 | |
111 | TorrentBencode dencode; | |
112 | public: | |
113 | seed(); | |
114 | ~seed(); | |
115 | ||
116 | int get_params(); | |
117 | void init(struct req_state *p_req, RGWRados *p_store); | |
118 | void get_torrent_file(int &op_ret, RGWRados::Object::Read &read_op, | |
119 | uint64_t &total_len, bufferlist &bl_data, rgw_obj &obj); | |
120 | ||
121 | off_t get_data_len(); | |
122 | bool get_flag(); | |
123 | ||
7c673cae | 124 | void set_create_date(ceph::real_time& value); |
31f18b77 FG |
125 | void set_info_name(const string& value); |
126 | void update(bufferlist &bl); | |
127 | int complete(); | |
7c673cae FG |
128 | |
129 | private: | |
130 | void do_encode (); | |
131 | void set_announce(); | |
132 | void set_exist(bool exist); | |
133 | void set_info_pieces(char *buff); | |
7c673cae FG |
134 | void sha1(SHA1 *h, bufferlist &bl, off_t bl_len); |
135 | int save_torrent_file(); | |
136 | }; | |
137 | #endif /* CEPH_RGW_TORRENT_H */ |