Statistics
| Branch: | Tag: | Revision:

root / lib / client / gnt_network.py @ af3aea48

History | View | Annotate | Download (11.9 kB)

1
#
2
#
3

    
4
# Copyright (C) 2011, 2012 Google Inc.
5
#
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.
10
#
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.
15
#
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
19
# 02110-1301, USA.
20

    
21
"""IP pool related commands"""
22

    
23
# pylint: disable=W0401,W0614
24
# W0401: Wildcard import ganeti.cli
25
# W0614: Unused import %s from wildcard import (since we need cli)
26

    
27
import textwrap
28
import itertools
29

    
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
35

    
36

    
37
#: default list of fields for L{ListNetworks}
38
_LIST_DEF_FIELDS = ["name", "network", "gateway",
39
                    "network_type", "mac_prefix", "group_list", "tags"]
40

    
41

    
42
def _HandleReservedIPs(ips):
43
  if ips is None:
44
    return None
45
  elif not ips:
46
    return []
47
  else:
48
    return utils.UnescapeAndSplit(ips, sep=",")
49

    
50

    
51
def AddNetwork(opts, args):
52
  """Add a network to the cluster.
53

54
  @param opts: the command line options selected by the user
55
  @type args: list
56
  @param args: a list of length 1 with the network name to create
57
  @rtype: int
58
  @return: the desired exit code
59

60
  """
61
  (network_name, ) = args
62

    
63
  if opts.network is None:
64
    raise errors.OpPrereqError("The --network option must be given",
65
                               errors.ECODE_INVAL)
66

    
67
  if opts.tags is not None:
68
    tags = opts.tags.split(",")
69
  else:
70
    tags = []
71

    
72
  reserved_ips = _HandleReservedIPs(opts.add_reserved_ips)
73

    
74
  op = opcodes.OpNetworkAdd(network_name=network_name,
75
                            gateway=opts.gateway,
76
                            network=opts.network,
77
                            gateway6=opts.gateway6,
78
                            network6=opts.network6,
79
                            mac_prefix=opts.mac_prefix,
80
                            network_type=opts.network_type,
81
                            add_reserved_ips=reserved_ips,
82
                            conflicts_check=opts.conflicts_check,
83
                            tags=tags)
84
  SubmitOrSend(op, opts)
85

    
86

    
87
def _GetDefaultGroups(cl, groups):
88
  """Gets list of groups to operate on.
89

90
  If C{groups} doesn't contain groups, a list of all groups in the cluster is
91
  returned.
92

93
  @type cl: L{luxi.Client}
94
  @type groups: list
95
  @rtype: list
96

97
  """
98
  if groups:
99
    return groups
100

    
101
  return list(itertools.chain(*cl.QueryGroups([], ["uuid"], False)))
102

    
103

    
104
def ConnectNetwork(opts, args):
105
  """Map a network to a node group.
106

107
  @param opts: the command line options selected by the user
108
  @type args: list
109
  @param args: Network, mode, physlink and node groups
110
  @rtype: int
111
  @return: the desired exit code
112

113
  """
114
  cl = GetClient()
115

    
116
  (network, mode, link) = args[:3]
117
  groups = _GetDefaultGroups(cl, args[3:])
118

    
119
  # TODO: Change logic to support "--submit"
120
  for group in groups:
121
    op = opcodes.OpNetworkConnect(group_name=group,
122
                                  network_name=network,
123
                                  network_mode=mode,
124
                                  network_link=link,
125
                                  conflicts_check=opts.conflicts_check)
126
    SubmitOpCode(op, opts=opts, cl=cl)
127

    
128

    
129
def DisconnectNetwork(opts, args):
130
  """Unmap a network from a node group.
131

132
  @param opts: the command line options selected by the user
133
  @type args: list
134
  @param args: Network and node groups
135
  @rtype: int
136
  @return: the desired exit code
137

138
  """
139
  cl = GetClient()
140

    
141
  (network, ) = args[:1]
142
  groups = _GetDefaultGroups(cl, args[1:])
143

    
144
  # TODO: Change logic to support "--submit"
145
  for group in groups:
146
    op = opcodes.OpNetworkDisconnect(group_name=group,
147
                                     network_name=network,
148
                                     conflicts_check=opts.conflicts_check)
149
    SubmitOpCode(op, opts=opts, cl=cl)
150

    
151

    
152
def ListNetworks(opts, args):
153
  """List Ip pools and their properties.
154

155
  @param opts: the command line options selected by the user
156
  @type args: list
157
  @param args: networks to list, or empty for all
158
  @rtype: int
159
  @return: the desired exit code
160

161
  """
162
  desired_fields = ParseFields(opts.output, _LIST_DEF_FIELDS)
163
  fmtoverride = {
164
    "group_list":
165
      (lambda data: utils.CommaJoin("%s (%s, %s)" % (name, mode, link)
166
                                    for (name, mode, link) in data),
167
       False),
168
    "inst_list": (",".join, False),
169
    "tags": (",".join, False),
170
    }
171

    
172
  return GenericList(constants.QR_NETWORK, desired_fields, args, None,
173
                     opts.separator, not opts.no_headers,
174
                     verbose=opts.verbose, format_override=fmtoverride)
175

    
176

    
177
def ListNetworkFields(opts, args):
178
  """List network fields.
179

180
  @param opts: the command line options selected by the user
181
  @type args: list
182
  @param args: fields to list, or empty for all
183
  @rtype: int
184
  @return: the desired exit code
185

186
  """
187
  return GenericListFields(constants.QR_NETWORK, args, opts.separator,
188
                           not opts.no_headers)
189

    
190

    
191
def ShowNetworkConfig(_, args):
192
  """Show network information.
193

194
  @type args: list
195
  @param args: should either be an empty list, in which case
196
      we show information about all nodes, or should contain
197
      a list of networks (names or UUIDs) to be queried for information
198
  @rtype: int
199
  @return: the desired exit code
200

201
  """
202
  cl = GetClient()
203
  result = cl.QueryNetworks(fields=["name", "network", "gateway",
204
                                    "network6", "gateway6",
205
                                    "mac_prefix", "network_type",
206
                                    "free_count", "reserved_count",
207
                                    "map", "group_list", "inst_list",
208
                                    "external_reservations",
209
                                    "serial_no", "uuid"],
210
                            names=args, use_locking=False)
211

    
212
  for (name, network, gateway, network6, gateway6,
213
       mac_prefix, network_type, free_count, reserved_count,
214
       mapping, group_list, instances, ext_res, serial, uuid) in result:
215
    size = free_count + reserved_count
216
    ToStdout("Network name: %s", name)
217
    ToStdout("UUID: %s", uuid)
218
    ToStdout("Serial number: %d", serial)
219
    ToStdout("  Subnet: %s", network)
220
    ToStdout("  Gateway: %s", gateway)
221
    ToStdout("  IPv6 Subnet: %s", network6)
222
    ToStdout("  IPv6 Gateway: %s", gateway6)
223
    ToStdout("  Mac Prefix: %s", mac_prefix)
224
    ToStdout("  Type: %s", network_type)
225
    ToStdout("  Size: %d", size)
226
    ToStdout("  Free: %d (%.2f%%)", free_count,
227
             100 * float(free_count) / float(size))
228
    ToStdout("  Usage map:")
229
    idx = 0
230
    for line in textwrap.wrap(mapping, width=64):
231
      ToStdout("     %s %s %d", str(idx).rjust(3), line.ljust(64), idx + 63)
232
      idx += 64
233
    ToStdout("         (X) used    (.) free")
234

    
235
    if ext_res:
236
      ToStdout("  externally reserved IPs:")
237
      for line in textwrap.wrap(ext_res, width=64):
238
        ToStdout("    %s" % line)
239

    
240
    if group_list:
241
      ToStdout("  connected to node groups:")
242
      for group in group_list:
243
        ToStdout("    %s", group)
244
    else:
245
      ToStdout("  not connected to any node group")
246

    
247
    if instances:
248
      ToStdout("  used by %d instances:", len(instances))
249
      for inst in instances:
250
        ((ips, networks), ) = cl.QueryInstances([inst],
251
                                                ["nic.ips", "nic.networks"],
252
                                                use_locking=False)
253

    
254
        l = lambda value: ", ".join(str(idx) + ":" + str(ip)
255
                                    for idx, (ip, net) in enumerate(value)
256
                                      if net == name)
257

    
258
        ToStdout("    %s : %s", inst, l(zip(ips, networks)))
259
    else:
260
      ToStdout("  not used by any instances")
261

    
262

    
263
def SetNetworkParams(opts, args):
264
  """Modifies an IP address pool's parameters.
265

266
  @param opts: the command line options selected by the user
267
  @type args: list
268
  @param args: should contain only one element, the node group name
269

270
  @rtype: int
271
  @return: the desired exit code
272

273
  """
274
  # TODO: add "network": opts.network,
275
  all_changes = {
276
    "gateway": opts.gateway,
277
    "add_reserved_ips": _HandleReservedIPs(opts.add_reserved_ips),
278
    "remove_reserved_ips": _HandleReservedIPs(opts.remove_reserved_ips),
279
    "mac_prefix": opts.mac_prefix,
280
    "network_type": opts.network_type,
281
    "gateway6": opts.gateway6,
282
    "network6": opts.network6,
283
  }
284

    
285
  if all_changes.values().count(None) == len(all_changes):
286
    ToStderr("Please give at least one of the parameters.")
287
    return 1
288

    
289
  # pylint: disable=W0142
290
  op = opcodes.OpNetworkSetParams(network_name=args[0], **all_changes)
291

    
292
  # TODO: add feedback to user, e.g. list the modifications
293
  SubmitOrSend(op, opts)
294

    
295

    
296
def RemoveNetwork(opts, args):
297
  """Remove an IP address pool from the cluster.
298

299
  @param opts: the command line options selected by the user
300
  @type args: list
301
  @param args: a list of length 1 with the id of the IP address pool to remove
302
  @rtype: int
303
  @return: the desired exit code
304

305
  """
306
  (network_name,) = args
307
  op = opcodes.OpNetworkRemove(network_name=network_name, force=opts.force)
308
  SubmitOrSend(op, opts)
309

    
310

    
311
commands = {
312
  "add": (
313
    AddNetwork, ARGS_ONE_NETWORK,
314
    [DRY_RUN_OPT, NETWORK_OPT, GATEWAY_OPT, ADD_RESERVED_IPS_OPT,
315
     MAC_PREFIX_OPT, NETWORK_TYPE_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"),
318
  "list": (
319
    ListNetworks, ARGS_MANY_NETWORKS,
320
    [NOHDR_OPT, SEP_OPT, FIELDS_OPT, VERBOSE_OPT],
321
    "[<network_id>...]",
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)),
325
  "list-fields": (
326
    ListNetworkFields, [ArgUnknown()], [NOHDR_OPT, SEP_OPT], "[fields...]",
327
    "Lists all available fields for networks"),
328
  "info": (
329
    ShowNetworkConfig, ARGS_MANY_NETWORKS, [],
330
    "[<network_name>...]", "Show information about the network(s)"),
331
  "modify": (
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, NETWORK_TYPE_OPT, NETWORK6_OPT, GATEWAY6_OPT,
335
     PRIORITY_OPT],
336
    "<network_name>", "Alters the parameters of a network"),
337
  "connect": (
338
    ConnectNetwork,
339
    [ArgNetwork(min=1, max=1),
340
     ArgChoice(min=1, max=1, choices=constants.NIC_VALID_MODES),
341
     ArgUnknown(min=1, max=1),
342
     ArgGroup()],
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)"),
347
  "disconnect": (
348
    DisconnectNetwork,
349
    [ArgNetwork(min=1, max=1), ArgGroup()],
350
    [NOCONFLICTSCHECK_OPT, PRIORITY_OPT],
351
    "<network_name> [<node_group>...]",
352
    "Unmap a given network from a specified node group"),
353
  "remove": (
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"),
358
  "list-tags": (
359
    ListTags, ARGS_ONE_NETWORK, [],
360
    "<network_name>", "List the tags of the given network"),
361
  "add-tags": (
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"),
365
  "remove-tags": (
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"),
369
}
370

    
371

    
372
def Main():
373
  return GenericMain(commands, override={"tag_type": constants.TAG_NETWORK})