4 # Copyright (C) 2011, 2012, 2013 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
21 """IP pool related commands"""
23 # pylint: disable=W0401,W0614
24 # W0401: Wildcard import ganeti.cli
25 # W0614: Unused import %s from wildcard import (since we need cli)
30 from ganeti.cli import *
31 from ganeti import constants
32 from ganeti import opcodes
33 from ganeti import utils
34 from ganeti import errors
37 #: default list of fields for L{ListNetworks}
38 _LIST_DEF_FIELDS = ["name", "network", "gateway",
39 "mac_prefix", "group_list", "tags"]
42 def _HandleReservedIPs(ips):
48 return utils.UnescapeAndSplit(ips, sep=",")
51 def AddNetwork(opts, args):
52 """Add a network to the cluster.
54 @param opts: the command line options selected by the user
56 @param args: a list of length 1 with the network name to create
58 @return: the desired exit code
61 (network_name, ) = args
63 if opts.network is None:
64 raise errors.OpPrereqError("The --network option must be given",
67 if opts.tags is not None:
68 tags = opts.tags.split(",")
72 reserved_ips = _HandleReservedIPs(opts.add_reserved_ips)
74 op = opcodes.OpNetworkAdd(network_name=network_name,
77 gateway6=opts.gateway6,
78 network6=opts.network6,
79 mac_prefix=opts.mac_prefix,
80 add_reserved_ips=reserved_ips,
81 conflicts_check=opts.conflicts_check,
83 SubmitOrSend(op, opts)
86 def _GetDefaultGroups(cl, groups):
87 """Gets list of groups to operate on.
89 If C{groups} doesn't contain groups, a list of all groups in the cluster is
92 @type cl: L{luxi.Client}
100 return list(itertools.chain(*cl.QueryGroups([], ["uuid"], False)))
103 def ConnectNetwork(opts, args):
104 """Map a network to a node group.
106 @param opts: the command line options selected by the user
108 @param args: Network, mode, physlink and node groups
110 @return: the desired exit code
115 (network, mode, link) = args[:3]
116 groups = _GetDefaultGroups(cl, args[3:])
118 # TODO: Change logic to support "--submit"
120 op = opcodes.OpNetworkConnect(group_name=group,
121 network_name=network,
124 conflicts_check=opts.conflicts_check)
125 SubmitOpCode(op, opts=opts, cl=cl)
128 def DisconnectNetwork(opts, args):
129 """Unmap a network from a node group.
131 @param opts: the command line options selected by the user
133 @param args: Network and node groups
135 @return: the desired exit code
140 (network, ) = args[:1]
141 groups = _GetDefaultGroups(cl, args[1:])
143 # TODO: Change logic to support "--submit"
145 op = opcodes.OpNetworkDisconnect(group_name=group,
146 network_name=network)
147 SubmitOpCode(op, opts=opts, cl=cl)
150 def ListNetworks(opts, args):
151 """List Ip pools and their properties.
153 @param opts: the command line options selected by the user
155 @param args: networks to list, or empty for all
157 @return: the desired exit code
160 desired_fields = ParseFields(opts.output, _LIST_DEF_FIELDS)
163 (lambda data: utils.CommaJoin("%s (%s, %s)" % (name, mode, link)
164 for (name, mode, link) in data),
166 "inst_list": (",".join, False),
167 "tags": (",".join, False),
170 cl = GetClient(query=True)
171 return GenericList(constants.QR_NETWORK, desired_fields, args, None,
172 opts.separator, not opts.no_headers,
173 verbose=opts.verbose, format_override=fmtoverride,
177 def ListNetworkFields(opts, args):
178 """List network fields.
180 @param opts: the command line options selected by the user
182 @param args: fields to list, or empty for all
184 @return: the desired exit code
187 cl = GetClient(query=True)
189 return GenericListFields(constants.QR_NETWORK, args, opts.separator,
190 not opts.no_headers, cl=cl)
193 def ShowNetworkConfig(_, args):
194 """Show network information.
197 @param args: should either be an empty list, in which case
198 we show information about all nodes, or should contain
199 a list of networks (names or UUIDs) to be queried for information
201 @return: the desired exit code
205 result = cl.QueryNetworks(fields=["name", "network", "gateway",
206 "network6", "gateway6",
208 "free_count", "reserved_count",
209 "map", "group_list", "inst_list",
210 "external_reservations",
211 "serial_no", "uuid"],
212 names=args, use_locking=False)
214 for (name, network, gateway, network6, gateway6,
215 mac_prefix, free_count, reserved_count,
216 mapping, group_list, instances, ext_res, serial, uuid) in result:
217 size = free_count + reserved_count
218 ToStdout("Network name: %s", name)
219 ToStdout("UUID: %s", uuid)
220 ToStdout("Serial number: %d", serial)
221 ToStdout(" Subnet: %s", network)
222 ToStdout(" Gateway: %s", gateway)
223 ToStdout(" IPv6 Subnet: %s", network6)
224 ToStdout(" IPv6 Gateway: %s", gateway6)
225 ToStdout(" Mac Prefix: %s", mac_prefix)
226 ToStdout(" Size: %d", size)
227 ToStdout(" Free: %d (%.2f%%)", free_count,
228 100 * float(free_count) / float(size))
229 ToStdout(" Usage map:")
231 for line in textwrap.wrap(mapping, width=64):
232 ToStdout(" %s %s %d", str(idx).rjust(3), line.ljust(64), idx + 63)
234 ToStdout(" (X) used (.) free")
237 ToStdout(" externally reserved IPs:")
238 for line in textwrap.wrap(ext_res, width=64):
239 ToStdout(" %s" % line)
242 ToStdout(" connected to node groups:")
243 for group, nic_mode, nic_link in group_list:
244 ToStdout(" %s (%s on %s)", group, nic_mode, nic_link)
246 ToStdout(" not connected to any node group")
249 ToStdout(" used by %d instances:", len(instances))
250 for inst in instances:
251 ((ips, networks), ) = cl.QueryInstances([inst],
252 ["nic.ips", "nic.networks"],
255 l = lambda value: ", ".join(str(idx) + ":" + str(ip)
256 for idx, (ip, net) in enumerate(value)
259 ToStdout(" %s : %s", inst, l(zip(ips, networks)))
261 ToStdout(" not used by any instances")
264 def SetNetworkParams(opts, args):
265 """Modifies an IP address pool's parameters.
267 @param opts: the command line options selected by the user
269 @param args: should contain only one element, the node group name
272 @return: the desired exit code
275 # TODO: add "network": opts.network,
277 "gateway": opts.gateway,
278 "add_reserved_ips": _HandleReservedIPs(opts.add_reserved_ips),
279 "remove_reserved_ips": _HandleReservedIPs(opts.remove_reserved_ips),
280 "mac_prefix": opts.mac_prefix,
281 "gateway6": opts.gateway6,
282 "network6": opts.network6,
285 if all_changes.values().count(None) == len(all_changes):
286 ToStderr("Please give at least one of the parameters.")
289 # pylint: disable=W0142
290 op = opcodes.OpNetworkSetParams(network_name=args[0], **all_changes)
292 # TODO: add feedback to user, e.g. list the modifications
293 SubmitOrSend(op, opts)
296 def RemoveNetwork(opts, args):
297 """Remove an IP address pool from the cluster.
299 @param opts: the command line options selected by the user
301 @param args: a list of length 1 with the id of the IP address pool to remove
303 @return: the desired exit code
306 (network_name,) = args
307 op = opcodes.OpNetworkRemove(network_name=network_name, force=opts.force)
308 SubmitOrSend(op, opts)
313 AddNetwork, ARGS_ONE_NETWORK,
314 [DRY_RUN_OPT, NETWORK_OPT, GATEWAY_OPT, ADD_RESERVED_IPS_OPT,
315 MAC_PREFIX_OPT, NETWORK6_OPT, GATEWAY6_OPT,
316 NOCONFLICTSCHECK_OPT, TAG_ADD_OPT, PRIORITY_OPT, SUBMIT_OPT],
317 "<network_name>", "Add a new IP network to the cluster"),
319 ListNetworks, ARGS_MANY_NETWORKS,
320 [NOHDR_OPT, SEP_OPT, FIELDS_OPT, VERBOSE_OPT],
322 "Lists the IP networks in the cluster. The available fields can be shown"
323 " using the \"list-fields\" command (see the man page for details)."
324 " The default list is (in order): %s." % utils.CommaJoin(_LIST_DEF_FIELDS)),
326 ListNetworkFields, [ArgUnknown()], [NOHDR_OPT, SEP_OPT], "[fields...]",
327 "Lists all available fields for networks"),
329 ShowNetworkConfig, ARGS_MANY_NETWORKS, [],
330 "[<network_name>...]", "Show information about the network(s)"),
332 SetNetworkParams, ARGS_ONE_NETWORK,
333 [DRY_RUN_OPT, SUBMIT_OPT, ADD_RESERVED_IPS_OPT, REMOVE_RESERVED_IPS_OPT,
334 GATEWAY_OPT, MAC_PREFIX_OPT, NETWORK6_OPT, GATEWAY6_OPT,
336 "<network_name>", "Alters the parameters of a network"),
339 [ArgNetwork(min=1, max=1),
340 ArgChoice(min=1, max=1, choices=constants.NIC_VALID_MODES),
341 ArgUnknown(min=1, max=1),
343 [NOCONFLICTSCHECK_OPT, PRIORITY_OPT],
344 "<network_name> <mode> <link> [<node_group>...]",
345 "Map a given network to the specified node group"
346 " with given mode and link (netparams)"),
349 [ArgNetwork(min=1, max=1), ArgGroup()],
351 "<network_name> [<node_group>...]",
352 "Unmap a given network from a specified node group"),
354 RemoveNetwork, ARGS_ONE_NETWORK,
355 [FORCE_OPT, DRY_RUN_OPT, SUBMIT_OPT, PRIORITY_OPT],
356 "[--dry-run] <network_id>",
357 "Remove an (empty) network from the cluster"),
359 ListTags, ARGS_ONE_NETWORK, [],
360 "<network_name>", "List the tags of the given network"),
362 AddTags, [ArgNetwork(min=1, max=1), ArgUnknown()],
363 [TAG_SRC_OPT, PRIORITY_OPT, SUBMIT_OPT],
364 "<network_name> tag...", "Add tags to the given network"),
366 RemoveTags, [ArgNetwork(min=1, max=1), ArgUnknown()],
367 [TAG_SRC_OPT, PRIORITY_OPT, SUBMIT_OPT],
368 "<network_name> tag...", "Remove tags from given network"),
373 return GenericMain(commands, override={"tag_type": constants.TAG_NETWORK})