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
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab
#include "include/compat.h"
#include "CrushLocation.h"
#include "CrushWrapper.h"
#include "common/ceph_context.h"
#include "common/config.h"
#include "include/str_list.h"
#include "common/debug.h"
#include "common/errno.h"
#include "include/compat.h"
#include "common/SubProcess.h"
#include <vector>
int CrushLocation::update_from_conf()
{
if (cct->_conf->crush_location.length())
return _parse(cct->_conf->crush_location);
return 0;
}
int CrushLocation::_parse(const std::string& s)
{
std::multimap<std::string,std::string> new_crush_location;
std::vector<std::string> lvec;
get_str_vec(s, ";, \t", lvec);
int r = CrushWrapper::parse_loc_multimap(lvec, &new_crush_location);
if (r < 0) {
lderr(cct) << "warning: crush_location '" << cct->_conf->crush_location
<< "' does not parse, keeping original crush_location "
<< loc << dendl;
return -EINVAL;
}
std::lock_guard<std::mutex> l(lock);
loc.swap(new_crush_location);
lgeneric_dout(cct, 10) << "crush_location is " << loc << dendl;
return 0;
}
int CrushLocation::update_from_hook()
{
if (cct->_conf->crush_location_hook.length() == 0)
return 0;
if (0 != access(cct->_conf->crush_location_hook.c_str(), R_OK)) {
lderr(cct) << "the user define crush location hook: " << cct->_conf->crush_location_hook
<< " may not exist or can not access it" << dendl;
return errno;
}
SubProcessTimed hook(
cct->_conf->crush_location_hook.c_str(),
SubProcess::CLOSE, SubProcess::PIPE, SubProcess::PIPE,
cct->_conf->crush_location_hook_timeout);
hook.add_cmd_args(
"--cluster", cct->_conf->cluster.c_str(),
"--id", cct->_conf->name.get_id().c_str(),
"--type", cct->_conf->name.get_type_str(),
NULL);
int ret = hook.spawn();
if (ret != 0) {
lderr(cct) << "error: failed run " << cct->_conf->crush_location_hook << ": "
<< hook.err() << dendl;
return ret;
}
bufferlist bl;
ret = bl.read_fd(hook.get_stdout(), 100 * 1024);
if (ret < 0) {
lderr(cct) << "error: failed read stdout from "
<< cct->_conf->crush_location_hook
<< ": " << cpp_strerror(-ret) << dendl;
bufferlist err;
err.read_fd(hook.get_stderr(), 100 * 1024);
lderr(cct) << "stderr:\n";
err.hexdump(*_dout);
*_dout << dendl;
}
if (hook.join() != 0) {
lderr(cct) << "error: failed to join: " << hook.err() << dendl;
return -EINVAL;
}
if (ret < 0)
return ret;
std::string out;
bl.copy(0, bl.length(), out);
out.erase(out.find_last_not_of(" \n\r\t")+1);
return _parse(out);
}
int CrushLocation::init_on_startup()
{
if (cct->_conf->crush_location.length()) {
return update_from_conf();
}
if (cct->_conf->crush_location_hook.length()) {
return update_from_hook();
}
// start with a sane default
char hostname[HOST_NAME_MAX + 1];
int r = gethostname(hostname, sizeof(hostname));
if (r < 0)
strcpy(hostname, "unknown_host");
// use short hostname
for (unsigned i=0; hostname[i]; ++i) {
if (hostname[i] == '.') {
hostname[i] = '\0';
break;
}
}
std::lock_guard<std::mutex> l(lock);
loc.clear();
loc.insert(make_pair<std::string,std::string>("host", hostname));
loc.insert(make_pair<std::string,std::string>("root", "default"));
lgeneric_dout(cct, 10) << "crush_location is (default) " << loc << dendl;
return 0;
}
|
