]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- |
2 | // vim: ts=8 sw=2 smarttab | |
3 | /* | |
4 | * Ceph - scalable distributed file system | |
5 | * | |
6 | * Copyright (C) 2004-2006 Sage Weil <sage@newdream.net> | |
7 | * | |
8 | * This is free software; you can redistribute it and/or | |
9 | * modify it under the terms of the GNU Lesser General Public | |
10 | * License version 2.1, as published by the Free Software | |
11 | * Foundation. See file COPYING. | |
12 | * | |
13 | */ | |
14 | ||
15 | #include <stdlib.h> | |
16 | #include <time.h> | |
17 | #include <errno.h> | |
18 | ||
19 | #include "include/types.h" | |
20 | #include "include/rados/librados.hpp" | |
21 | ||
20effc67 | 22 | using namespace std; |
7c673cae FG |
23 | using namespace librados; |
24 | ||
25 | void buf_to_hex(const unsigned char *buf, int len, char *str) | |
26 | { | |
27 | str[0] = '\0'; | |
28 | for (int i = 0; i < len; i++) { | |
29 | sprintf(&str[i*2], "%02x", (int)buf[i]); | |
30 | } | |
31 | } | |
32 | ||
33 | ||
34 | #define ID_SIZE 8 | |
35 | ||
36 | #define ACL_RD 0x1 | |
37 | #define ACL_WR 0x2 | |
38 | ||
39 | struct ACLID { | |
40 | char id[ID_SIZE + 1]; | |
41 | ||
42 | void encode(bufferlist& bl) const { | |
43 | bl.append((const char *)id, ID_SIZE); | |
44 | } | |
11fdf7f2 | 45 | void decode(bufferlist::const_iterator& iter) { |
7c673cae FG |
46 | iter.copy(ID_SIZE, (char *)id); |
47 | } | |
48 | }; | |
49 | WRITE_CLASS_ENCODER(ACLID) | |
50 | ||
51 | typedef __u32 ACLFlags; | |
52 | ||
53 | ||
54 | inline bool operator<(const ACLID& l, const ACLID& r) | |
55 | { | |
56 | return (memcmp(&l, &r, ID_SIZE) < 0); | |
57 | } | |
58 | ||
59 | struct ACLPair { | |
60 | ACLID id; | |
61 | ACLFlags flags; | |
62 | }; | |
63 | ||
64 | class ObjectACLs { | |
65 | map<ACLID, ACLFlags> acls_map; | |
66 | ||
67 | public: | |
68 | ||
69 | void encode(bufferlist& bl) const { | |
11fdf7f2 TL |
70 | using ceph::encode; |
71 | encode(acls_map, bl); | |
7c673cae | 72 | } |
11fdf7f2 TL |
73 | void decode(bufferlist::const_iterator& bl) { |
74 | using ceph::decode; | |
75 | decode(acls_map, bl); | |
7c673cae FG |
76 | } |
77 | ||
78 | int read_acl(ACLID& id, ACLFlags *flags); | |
79 | void set_acl(ACLID& id, ACLFlags flags); | |
80 | }; | |
81 | WRITE_CLASS_ENCODER(ObjectACLs) | |
82 | ||
83 | int ObjectACLs::read_acl(ACLID& id, ACLFlags *flags) | |
84 | { | |
85 | if (!flags) | |
86 | return -EINVAL; | |
87 | ||
88 | map<ACLID, ACLFlags>::iterator iter = acls_map.find(id); | |
89 | ||
90 | if (iter == acls_map.end()) | |
91 | return -ENOENT; | |
92 | ||
93 | *flags = iter->second; | |
94 | ||
95 | return 0; | |
96 | } | |
97 | ||
98 | void ObjectACLs::set_acl(ACLID& id, ACLFlags flags) | |
99 | { | |
100 | acls_map[id] = flags; | |
101 | } | |
102 | ||
103 | ||
104 | ||
105 | class ACLEntity | |
106 | { | |
107 | string name; | |
108 | map<ACLID, ACLEntity> groups; | |
109 | }; | |
110 | ||
111 | typedef map<ACLID, ACLEntity> tACLIDEntityMap; | |
112 | ||
113 | static map<ACLID, ACLEntity> users; | |
114 | static map<ACLID, ACLEntity> groups; | |
115 | ||
116 | void get_user(ACLID& aclid, ACLEntity *entity) | |
117 | { | |
118 | //users.find(aclid); | |
119 | } | |
120 | ||
121 | ||
122 | ||
123 | ||
124 | ||
125 | int main(int argc, const char **argv) | |
126 | { | |
127 | Rados rados; | |
128 | if (rados.init(NULL) < 0) { | |
129 | cerr << "couldn't initialize rados!" << std::endl; | |
130 | exit(1); | |
131 | } | |
132 | if (rados.conf_read_file(NULL)) { | |
133 | cerr << "couldn't read Ceph configuration file!" << std::endl; | |
134 | exit(1); | |
135 | } | |
136 | if (rados.connect() < 0) { | |
137 | cerr << "couldn't connect to cluster!" << std::endl; | |
138 | exit(1); | |
139 | } | |
140 | ||
141 | time_t tm; | |
142 | bufferlist bl, bl2; | |
143 | char buf[128]; | |
144 | ||
145 | time(&tm); | |
146 | snprintf(buf, 128, "%s", ctime(&tm)); | |
147 | bl.append(buf, strlen(buf)); | |
148 | ||
149 | const char *oid = "bar"; | |
150 | ||
151 | IoCtx io_ctx; | |
152 | int r = rados.ioctx_create("data", io_ctx); | |
153 | cout << "open io_ctx result = " << r << " pool = " << io_ctx.get_pool_name() << std::endl; | |
154 | ||
155 | ACLID id; | |
156 | ||
157 | snprintf(id.id, sizeof(id.id), "%.8x", 0x1234); | |
158 | cout << "id=" << id.id << std::endl; | |
159 | ||
160 | r = io_ctx.exec(oid, "acl", "get", bl, bl2); | |
161 | cout << "exec(acl get) returned " << r | |
162 | << " len=" << bl2.length() << std::endl; | |
163 | ObjectACLs oa; | |
164 | if (r >= 0) { | |
11fdf7f2 | 165 | auto iter = bl2.cbegin(); |
7c673cae FG |
166 | oa.decode(iter); |
167 | } | |
168 | ||
169 | oa.set_acl(id, ACL_RD); | |
170 | bl.clear(); | |
171 | oa.encode(bl); | |
172 | r = io_ctx.exec(oid, "acl", "set", bl, bl2); | |
173 | cout << "exec(acl set) returned " << r | |
174 | << " len=" << bl2.length() << std::endl; | |
175 | ||
176 | const unsigned char *md5 = (const unsigned char *)bl2.c_str(); | |
177 | char md5_str[bl2.length()*2 + 1]; | |
178 | buf_to_hex(md5, bl2.length(), md5_str); | |
179 | cout << "md5 result=" << md5_str << std::endl; | |
180 | ||
181 | int size = io_ctx.read(oid, bl2, 128, 0); | |
182 | cout << "read result=" << bl2.c_str() << std::endl; | |
183 | cout << "size=" << size << std::endl; | |
184 | ||
185 | return 0; | |
186 | } | |
187 |