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
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab
#include "test/librbd/test_support.h"
#include "include/rbd_types.h"
#include "gtest/gtest.h"
#include <sstream>
bool get_features(uint64_t *features) {
const char *c = getenv("RBD_FEATURES");
if (c == NULL) {
return false;
}
std::stringstream ss(c);
if (!(ss >> *features)) {
return false;
}
return true;
}
bool is_feature_enabled(uint64_t feature) {
uint64_t features;
return (get_features(&features) && (features & feature) == feature);
}
int create_image_full_pp(librbd::RBD &rbd, librados::IoCtx &ioctx,
const std::string &name, uint64_t size,
uint64_t features, bool old_format, int *order)
{
if (old_format) {
librados::Rados rados(ioctx);
int r = rados.conf_set("rbd_default_format", "1");
if (r < 0) {
return r;
}
return rbd.create(ioctx, name.c_str(), size, order);
} else if ((features & RBD_FEATURE_STRIPINGV2) != 0) {
uint64_t stripe_unit = IMAGE_STRIPE_UNIT;
if (*order) {
// use a conservative stripe_unit for non default order
stripe_unit = (1ull << (*order-1));
}
printf("creating image with stripe unit: %" PRIu64 ", stripe count: %" PRIu64 "\n",
stripe_unit, IMAGE_STRIPE_COUNT);
return rbd.create3(ioctx, name.c_str(), size, features, order, stripe_unit,
IMAGE_STRIPE_COUNT);
} else {
return rbd.create2(ioctx, name.c_str(), size, features, order);
}
}
int create_image_pp(librbd::RBD &rbd, librados::IoCtx &ioctx,
const std::string &name, uint64_t size) {
int order = 0;
uint64_t features = 0;
if (!get_features(&features)) {
// ensure old-format tests actually use the old format
librados::Rados rados(ioctx);
int r = rados.conf_set("rbd_default_format", "1");
if (r < 0) {
return r;
}
return rbd.create(ioctx, name.c_str(), size, &order);
} else {
return rbd.create2(ioctx, name.c_str(), size, features, &order);
}
}
int clone_image_pp(librbd::RBD &rbd, librbd::Image &p_image, librados::IoCtx &p_ioctx,
const char *p_name, const char *p_snap_name, librados::IoCtx &c_ioctx,
const char *c_name, uint64_t features)
{
uint64_t stripe_unit = p_image.get_stripe_unit();
uint64_t stripe_count = p_image.get_stripe_count();
librbd::image_info_t p_info;
int r = p_image.stat(p_info, sizeof(p_info));
if (r < 0) {
return r;
}
int c_order = p_info.order;
return rbd.clone2(p_ioctx, p_name, p_snap_name, c_ioctx, c_name,
features, &c_order, stripe_unit, stripe_count);
}
int get_image_id(librbd::Image &image, std::string *image_id)
{
int r = image.get_id(image_id);
if (r < 0) {
return r;
}
return 0;
}
int create_image_data_pool(librados::Rados &rados, std::string &data_pool, bool *created) {
std::string pool;
int r = rados.conf_get("rbd_default_data_pool", pool);
if (r != 0) {
return r;
} else if (pool.empty()) {
return 0;
}
r = rados.pool_create(pool.c_str());
if ((r == 0) || (r == -EEXIST)) {
data_pool = pool;
*created = (r == 0);
return 0;
}
librados::IoCtx ioctx;
r = rados.ioctx_create(pool.c_str(), ioctx);
if (r < 0) {
return r;
}
librbd::RBD rbd;
return rbd.pool_init(ioctx, true);
}
bool is_librados_test_stub(librados::Rados &rados) {
std::string fsid;
EXPECT_EQ(0, rados.cluster_fsid(&fsid));
return fsid == "00000000-1111-2222-3333-444444444444";
}
|
