4 # Copyright (C) 2006, 2007, 2008 Google Inc.
6 # This program is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
11 # This program is distributed in the hope that it will be useful, but
12 # WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 # General Public License for more details.
16 # You should have received a copy of the GNU General Public License
17 # along with this program; if not, write to the Free Software
18 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
22 """Global Configuration data for Ganeti.
24 This module provides the interface to a special case of cluster
25 configuration data, which is mostly static and available to all nodes.
33 from ganeti import errors
34 from ganeti import constants
35 from ganeti import utils
36 from ganeti import serializer
37 from ganeti import objects
40 SSCONF_LOCK_TIMEOUT = 10
42 RE_VALID_SSCONF_NAME = re.compile(r'^[-_a-z0-9]+$')
45 class SimpleConfigReader(object):
46 """Simple class to read configuration file.
49 def __init__(self, file_name=constants.CLUSTER_CONF_FILE):
50 """Initializes this class.
52 @type file_name: string
53 @param file_name: Configuration file path
56 self._file_name = file_name
57 self._last_inode = None
58 self._last_mtime = None
59 self._last_size = None
61 self._config_data = None
62 self._inst_ips_by_link = None
63 self._ip_to_inst_by_link = None
64 self._instances_ips = None
65 self._mc_primary_ips = None
66 self._nodes_primary_ips = None
68 # we need a forced reload at class init time, to initialize _last_*
69 self._Load(force=True)
71 def _Load(self, force=False):
72 """Loads (or reloads) the config file.
75 @param force: whether to force the reload without checking the mtime
77 @return: boolean value that says whether we reloaded the configuration or
78 not (because we decided it was already up-to-date)
82 cfg_stat = os.stat(self._file_name)
83 except EnvironmentError, err:
84 raise errors.ConfigurationError("Cannot stat config file %s: %s" %
85 (self._file_name, err))
86 inode = cfg_stat.st_ino
87 mtime = cfg_stat.st_mtime
88 size = cfg_stat.st_size
90 if (force or inode != self._last_inode or
91 mtime > self._last_mtime or
92 size != self._last_size):
93 self._last_inode = inode
94 self._last_mtime = mtime
95 self._last_size = size
101 self._config_data = serializer.Load(utils.ReadFile(self._file_name))
102 except EnvironmentError, err:
103 raise errors.ConfigurationError("Cannot read config file %s: %s" %
104 (self._file_name, err))
105 except ValueError, err:
106 raise errors.ConfigurationError("Cannot load config file %s: %s" %
107 (self._file_name, err))
109 self._ip_to_inst_by_link = {}
110 self._instances_ips = []
111 self._inst_ips_by_link = {}
112 c_nparams = self._config_data['cluster']['nicparams'][constants.PP_DEFAULT]
113 for iname in self._config_data['instances']:
114 instance = self._config_data['instances'][iname]
115 for nic in instance['nics']:
116 if 'ip' in nic and nic['ip']:
117 params = objects.FillDict(c_nparams, nic['nicparams'])
118 if not params['link'] in self._inst_ips_by_link:
119 self._inst_ips_by_link[params['link']] = []
120 self._ip_to_inst_by_link[params['link']] = {}
121 self._ip_to_inst_by_link[params['link']][nic['ip']] = iname
122 self._inst_ips_by_link[params['link']].append(nic['ip'])
124 self._nodes_primary_ips = []
125 self._mc_primary_ips = []
126 for node_name in self._config_data["nodes"]:
127 node = self._config_data["nodes"][node_name]
128 self._nodes_primary_ips.append(node["primary_ip"])
129 if node["master_candidate"]:
130 self._mc_primary_ips.append(node["primary_ip"])
134 # Clients can request a reload of the config file, so we export our internal
135 # _Load function as Reload.
138 def GetClusterName(self):
139 return self._config_data["cluster"]["cluster_name"]
141 def GetHostKey(self):
142 return self._config_data["cluster"]["rsahostkeypub"]
144 def GetMasterNode(self):
145 return self._config_data["cluster"]["master_node"]
147 def GetMasterIP(self):
148 return self._config_data["cluster"]["master_ip"]
150 def GetMasterNetdev(self):
151 return self._config_data["cluster"]["master_netdev"]
153 def GetFileStorageDir(self):
154 return self._config_data["cluster"]["file_storage_dir"]
156 def GetNodeList(self):
157 return self._config_data["nodes"].keys()
159 def GetConfigSerialNo(self):
160 return self._config_data["serial_no"]
162 def GetClusterSerialNo(self):
163 return self._config_data["cluster"]["serial_no"]
165 def GetDefaultNicParams(self):
166 return self._config_data["cluster"]["nicparams"][constants.PP_DEFAULT]
168 def GetDefaultNicLink(self):
169 return self.GetDefaultNicParams()[constants.NIC_LINK]
171 def GetNodeStatusFlags(self, node):
172 """Get a node's status flags
175 @param node: node name
176 @rtype: (bool, bool, bool)
177 @return: (master_candidate, drained, offline) (or None if no such node)
180 if node not in self._config_data["nodes"]:
183 master_candidate = self._config_data["nodes"][node]["master_candidate"]
184 drained = self._config_data["nodes"][node]["drained"]
185 offline = self._config_data["nodes"][node]["offline"]
186 return master_candidate, drained, offline
188 def GetInstanceByLinkIp(self, ip, link):
189 """Get instance name from its link and ip address.
192 @param ip: ip address
194 @param link: nic link
196 @return: instance name
200 link = self.GetDefaultNicLink()
201 if not link in self._ip_to_inst_by_link:
203 if not ip in self._ip_to_inst_by_link[link]:
205 return self._ip_to_inst_by_link[link][ip]
207 def GetNodePrimaryIp(self, node):
208 """Get a node's primary ip
211 @param node: node name
212 @rtype: string, or None
213 @return: node's primary ip, or None if no such node
216 if node not in self._config_data["nodes"]:
218 return self._config_data["nodes"][node]["primary_ip"]
220 def GetInstancePrimaryNode(self, instance):
221 """Get an instance's primary node
223 @type instance: string
224 @param instance: instance name
225 @rtype: string, or None
226 @return: primary node, or None if no such instance
229 if instance not in self._config_data["instances"]:
231 return self._config_data["instances"][instance]["primary_node"]
233 def GetNodesPrimaryIps(self):
234 return self._nodes_primary_ips
236 def GetMasterCandidatesPrimaryIps(self):
237 return self._mc_primary_ips
239 def GetInstancesIps(self, link):
240 """Get list of nic ips connected to a certain link.
243 @param link: nic link
245 @return: list of ips connected to that link
249 link = self.GetDefaultNicLink()
251 if link in self._inst_ips_by_link:
252 return self._inst_ips_by_link[link]
257 class SimpleStore(object):
258 """Interface to static cluster data.
260 This is different that the config.ConfigWriter and
261 SimpleConfigReader classes in that it holds data that will always be
262 present, even on nodes which don't have all the cluster data.
264 Other particularities of the datastore:
265 - keys are restricted to predefined values
268 _SS_FILEPREFIX = "ssconf_"
270 constants.SS_CLUSTER_NAME,
271 constants.SS_CLUSTER_TAGS,
272 constants.SS_FILE_STORAGE_DIR,
273 constants.SS_MASTER_CANDIDATES,
274 constants.SS_MASTER_CANDIDATES_IPS,
275 constants.SS_MASTER_IP,
276 constants.SS_MASTER_NETDEV,
277 constants.SS_MASTER_NODE,
278 constants.SS_NODE_LIST,
279 constants.SS_NODE_PRIMARY_IPS,
280 constants.SS_NODE_SECONDARY_IPS,
281 constants.SS_OFFLINE_NODES,
282 constants.SS_ONLINE_NODES,
283 constants.SS_INSTANCE_LIST,
284 constants.SS_RELEASE_VERSION,
288 def __init__(self, cfg_location=None):
289 if cfg_location is None:
290 self._cfg_dir = constants.DATA_DIR
292 self._cfg_dir = cfg_location
294 def KeyToFilename(self, key):
295 """Convert a given key into filename.
298 if key not in self._VALID_KEYS:
299 raise errors.ProgrammerError("Invalid key requested from SSConf: '%s'"
302 filename = self._cfg_dir + '/' + self._SS_FILEPREFIX + key
305 def _ReadFile(self, key):
306 """Generic routine to read keys.
308 This will read the file which holds the value requested. Errors
309 will be changed into ConfigurationErrors.
312 filename = self.KeyToFilename(key)
314 data = utils.ReadFile(filename, size=self._MAX_SIZE)
315 except EnvironmentError, err:
316 raise errors.ConfigurationError("Can't read from the ssconf file:"
318 data = data.rstrip('\n')
321 def WriteFiles(self, values):
322 """Writes ssconf files used by external scripts.
325 @param values: Dictionary of (name, value)
328 ssconf_lock = utils.FileLock.Open(constants.SSCONF_LOCK_FILE)
330 # Get lock while writing files
331 ssconf_lock.Exclusive(blocking=True, timeout=SSCONF_LOCK_TIMEOUT)
333 for name, value in values.iteritems():
334 if value and not value.endswith("\n"):
336 utils.WriteFile(self.KeyToFilename(name), data=value, mode=0444)
340 def GetFileList(self):
341 """Return the list of all config files.
343 This is used for computing node replication data.
346 return [self.KeyToFilename(key) for key in self._VALID_KEYS]
348 def GetClusterName(self):
349 """Get the cluster name.
352 return self._ReadFile(constants.SS_CLUSTER_NAME)
354 def GetFileStorageDir(self):
355 """Get the file storage dir.
358 return self._ReadFile(constants.SS_FILE_STORAGE_DIR)
360 def GetMasterCandidates(self):
361 """Return the list of master candidates.
364 data = self._ReadFile(constants.SS_MASTER_CANDIDATES)
365 nl = data.splitlines(False)
368 def GetMasterCandidatesIPList(self):
369 """Return the list of master candidates' primary IP.
372 data = self._ReadFile(constants.SS_MASTER_CANDIDATES_IPS)
373 nl = data.splitlines(False)
376 def GetMasterIP(self):
377 """Get the IP of the master node for this cluster.
380 return self._ReadFile(constants.SS_MASTER_IP)
382 def GetMasterNetdev(self):
383 """Get the netdev to which we'll add the master ip.
386 return self._ReadFile(constants.SS_MASTER_NETDEV)
388 def GetMasterNode(self):
389 """Get the hostname of the master node for this cluster.
392 return self._ReadFile(constants.SS_MASTER_NODE)
394 def GetNodeList(self):
395 """Return the list of cluster nodes.
398 data = self._ReadFile(constants.SS_NODE_LIST)
399 nl = data.splitlines(False)
402 def GetNodePrimaryIPList(self):
403 """Return the list of cluster nodes' primary IP.
406 data = self._ReadFile(constants.SS_NODE_PRIMARY_IPS)
407 nl = data.splitlines(False)
410 def GetNodeSecondaryIPList(self):
411 """Return the list of cluster nodes' secondary IP.
414 data = self._ReadFile(constants.SS_NODE_SECONDARY_IPS)
415 nl = data.splitlines(False)
418 def GetClusterTags(self):
419 """Return the cluster tags.
422 data = self._ReadFile(constants.SS_CLUSTER_TAGS)
423 nl = data.splitlines(False)
427 def GetMasterAndMyself(ss=None):
428 """Get the master node and my own hostname.
430 This can be either used for a 'soft' check (compared to CheckMaster,
431 which exits) or just for computing both at the same time.
433 The function does not handle any errors, these should be handled in
434 the caller (errors.ConfigurationError, errors.ResolverError).
436 @param ss: either a sstore.SimpleConfigReader or a
437 sstore.SimpleStore instance
439 @return: a tuple (master node name, my own name)
444 return ss.GetMasterNode(), utils.HostInfo().name
447 def CheckMaster(debug, ss=None):
448 """Checks the node setup.
450 If this is the master, the function will return. Otherwise it will
451 exit with an exit code based on the node status.
455 master_name, myself = GetMasterAndMyself(ss)
456 except errors.ConfigurationError, err:
457 print "Cluster configuration incomplete: '%s'" % str(err)
458 sys.exit(constants.EXIT_NODESETUP_ERROR)
459 except errors.ResolverError, err:
460 sys.stderr.write("Cannot resolve my own name (%s)\n" % err.args[0])
461 sys.exit(constants.EXIT_NODESETUP_ERROR)
463 if myself != master_name:
465 sys.stderr.write("Not master, exiting.\n")
466 sys.exit(constants.EXIT_NOTMASTER)
469 def CheckMasterCandidate(debug, ss=None):
470 """Checks the node setup.
472 If this is a master candidate, the function will return. Otherwise it will
473 exit with an exit code based on the node status.
479 myself = utils.HostInfo().name
480 candidates = ss.GetMasterCandidates()
481 except errors.ConfigurationError, err:
482 print "Cluster configuration incomplete: '%s'" % str(err)
483 sys.exit(constants.EXIT_NODESETUP_ERROR)
484 except errors.ResolverError, err:
485 sys.stderr.write("Cannot resolve my own name (%s)\n" % err.args[0])
486 sys.exit(constants.EXIT_NODESETUP_ERROR)
488 if myself not in candidates:
490 sys.stderr.write("Not master candidate, exiting.\n")
491 sys.exit(constants.EXIT_NOTCANDIDATE)