Statistics
| Branch: | Tag: | Revision:

root / lib / rapi / client.py @ 5ed59e1e

History | View | Annotate | Download (32.6 kB)

1
#
2
#
3

    
4
# Copyright (C) 2010 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

    
22
"""Ganeti RAPI client."""
23

    
24
# No Ganeti-specific modules should be imported. The RAPI client is supposed to
25
# be standalone.
26

    
27
import httplib
28
import urllib2
29
import logging
30
import simplejson
31
import socket
32
import urllib
33
import OpenSSL
34
import distutils.version
35

    
36

    
37
GANETI_RAPI_PORT = 5080
38
GANETI_RAPI_VERSION = 2
39

    
40
HTTP_DELETE = "DELETE"
41
HTTP_GET = "GET"
42
HTTP_PUT = "PUT"
43
HTTP_POST = "POST"
44
HTTP_OK = 200
45
HTTP_NOT_FOUND = 404
46
HTTP_APP_JSON = "application/json"
47

    
48
REPLACE_DISK_PRI = "replace_on_primary"
49
REPLACE_DISK_SECONDARY = "replace_on_secondary"
50
REPLACE_DISK_CHG = "replace_new_secondary"
51
REPLACE_DISK_AUTO = "replace_auto"
52

    
53
NODE_ROLE_DRAINED = "drained"
54
NODE_ROLE_MASTER_CANDIATE = "master-candidate"
55
NODE_ROLE_MASTER = "master"
56
NODE_ROLE_OFFLINE = "offline"
57
NODE_ROLE_REGULAR = "regular"
58

    
59
# Internal constants
60
_REQ_DATA_VERSION_FIELD = "__version__"
61
_INST_CREATE_REQV1 = "instance-create-reqv1"
62

    
63

    
64
class Error(Exception):
65
  """Base error class for this module.
66

67
  """
68
  pass
69

    
70

    
71
class CertificateError(Error):
72
  """Raised when a problem is found with the SSL certificate.
73

74
  """
75
  pass
76

    
77

    
78
class GanetiApiError(Error):
79
  """Generic error raised from Ganeti API.
80

81
  """
82
  def __init__(self, msg, code=None):
83
    Error.__init__(self, msg)
84
    self.code = code
85

    
86

    
87
def FormatX509Name(x509_name):
88
  """Formats an X509 name.
89

90
  @type x509_name: OpenSSL.crypto.X509Name
91

92
  """
93
  try:
94
    # Only supported in pyOpenSSL 0.7 and above
95
    get_components_fn = x509_name.get_components
96
  except AttributeError:
97
    return repr(x509_name)
98
  else:
99
    return "".join("/%s=%s" % (name, value)
100
                   for name, value in get_components_fn())
101

    
102

    
103
class CertAuthorityVerify:
104
  """Certificate verificator for SSL context.
105

106
  Configures SSL context to verify server's certificate.
107

108
  """
109
  _CAPATH_MINVERSION = "0.9"
110
  _DEFVFYPATHS_MINVERSION = "0.9"
111

    
112
  _PYOPENSSL_VERSION = OpenSSL.__version__
113
  _PARSED_PYOPENSSL_VERSION = distutils.version.LooseVersion(_PYOPENSSL_VERSION)
114

    
115
  _SUPPORT_CAPATH = (_PARSED_PYOPENSSL_VERSION >= _CAPATH_MINVERSION)
116
  _SUPPORT_DEFVFYPATHS = (_PARSED_PYOPENSSL_VERSION >= _DEFVFYPATHS_MINVERSION)
117

    
118
  def __init__(self, cafile=None, capath=None, use_default_verify_paths=False):
119
    """Initializes this class.
120

121
    @type cafile: string
122
    @param cafile: In which file we can find the certificates
123
    @type capath: string
124
    @param capath: In which directory we can find the certificates
125
    @type use_default_verify_paths: bool
126
    @param use_default_verify_paths: Whether the platform provided CA
127
                                     certificates are to be used for
128
                                     verification purposes
129

130
    """
131
    self._cafile = cafile
132
    self._capath = capath
133
    self._use_default_verify_paths = use_default_verify_paths
134

    
135
    if self._capath is not None and not self._SUPPORT_CAPATH:
136
      raise Error(("PyOpenSSL %s has no support for a CA directory,"
137
                   " version %s or above is required") %
138
                  (self._PYOPENSSL_VERSION, self._CAPATH_MINVERSION))
139

    
140
    if self._use_default_verify_paths and not self._SUPPORT_DEFVFYPATHS:
141
      raise Error(("PyOpenSSL %s has no support for using default verification"
142
                   " paths, version %s or above is required") %
143
                  (self._PYOPENSSL_VERSION, self._DEFVFYPATHS_MINVERSION))
144

    
145
  @staticmethod
146
  def _VerifySslCertCb(logger, _, cert, errnum, errdepth, ok):
147
    """Callback for SSL certificate verification.
148

149
    @param logger: Logging object
150

151
    """
152
    if ok:
153
      log_fn = logger.debug
154
    else:
155
      log_fn = logger.error
156

    
157
    log_fn("Verifying SSL certificate at depth %s, subject '%s', issuer '%s'",
158
           errdepth, FormatX509Name(cert.get_subject()),
159
           FormatX509Name(cert.get_issuer()))
160

    
161
    if not ok:
162
      try:
163
        # Only supported in pyOpenSSL 0.7 and above
164
        # pylint: disable-msg=E1101
165
        fn = OpenSSL.crypto.X509_verify_cert_error_string
166
      except AttributeError:
167
        errmsg = ""
168
      else:
169
        errmsg = ":%s" % fn(errnum)
170

    
171
      logger.error("verify error:num=%s%s", errnum, errmsg)
172

    
173
    return ok
174

    
175
  def __call__(self, ctx, logger):
176
    """Configures an SSL context to verify certificates.
177

178
    @type ctx: OpenSSL.SSL.Context
179
    @param ctx: SSL context
180

181
    """
182
    if self._use_default_verify_paths:
183
      ctx.set_default_verify_paths()
184

    
185
    if self._cafile or self._capath:
186
      if self._SUPPORT_CAPATH:
187
        ctx.load_verify_locations(self._cafile, self._capath)
188
      else:
189
        ctx.load_verify_locations(self._cafile)
190

    
191
    ctx.set_verify(OpenSSL.SSL.VERIFY_PEER,
192
                   lambda conn, cert, errnum, errdepth, ok: \
193
                     self._VerifySslCertCb(logger, conn, cert,
194
                                           errnum, errdepth, ok))
195

    
196

    
197
class _HTTPSConnectionOpenSSL(httplib.HTTPSConnection):
198
  """HTTPS Connection handler that verifies the SSL certificate.
199

200
  """
201
  def __init__(self, *args, **kwargs):
202
    """Initializes this class.
203

204
    """
205
    httplib.HTTPSConnection.__init__(self, *args, **kwargs)
206
    self._logger = None
207
    self._config_ssl_verification = None
208

    
209
  def Setup(self, logger, config_ssl_verification):
210
    """Sets the SSL verification config function.
211

212
    @param logger: Logging object
213
    @type config_ssl_verification: callable
214

215
    """
216
    assert self._logger is None
217
    assert self._config_ssl_verification is None
218

    
219
    self._logger = logger
220
    self._config_ssl_verification = config_ssl_verification
221

    
222
  def connect(self):
223
    """Connect to the server specified when the object was created.
224

225
    This ensures that SSL certificates are verified.
226

227
    """
228
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
229

    
230
    ctx = OpenSSL.SSL.Context(OpenSSL.SSL.TLSv1_METHOD)
231
    ctx.set_options(OpenSSL.SSL.OP_NO_SSLv2)
232

    
233
    if self._config_ssl_verification:
234
      self._config_ssl_verification(ctx, self._logger)
235

    
236
    ssl = OpenSSL.SSL.Connection(ctx, sock)
237
    ssl.connect((self.host, self.port))
238

    
239
    self.sock = httplib.FakeSocket(sock, ssl)
240

    
241

    
242
class _HTTPSHandler(urllib2.HTTPSHandler):
243
  def __init__(self, logger, config_ssl_verification):
244
    """Initializes this class.
245

246
    @param logger: Logging object
247
    @type config_ssl_verification: callable
248
    @param config_ssl_verification: Function to configure SSL context for
249
                                    certificate verification
250

251
    """
252
    urllib2.HTTPSHandler.__init__(self)
253
    self._logger = logger
254
    self._config_ssl_verification = config_ssl_verification
255

    
256
  def _CreateHttpsConnection(self, *args, **kwargs):
257
    """Wrapper around L{_HTTPSConnectionOpenSSL} to add SSL verification.
258

259
    This wrapper is necessary provide a compatible API to urllib2.
260

261
    """
262
    conn = _HTTPSConnectionOpenSSL(*args, **kwargs)
263
    conn.Setup(self._logger, self._config_ssl_verification)
264
    return conn
265

    
266
  def https_open(self, req):
267
    """Creates HTTPS connection.
268

269
    Called by urllib2.
270

271
    """
272
    return self.do_open(self._CreateHttpsConnection, req)
273

    
274

    
275
class _RapiRequest(urllib2.Request):
276
  def __init__(self, method, url, headers, data):
277
    """Initializes this class.
278

279
    """
280
    urllib2.Request.__init__(self, url, data=data, headers=headers)
281
    self._method = method
282

    
283
  def get_method(self):
284
    """Returns the HTTP request method.
285

286
    """
287
    return self._method
288

    
289

    
290
class GanetiRapiClient(object):
291
  """Ganeti RAPI client.
292

293
  """
294
  USER_AGENT = "Ganeti RAPI Client"
295
  _json_encoder = simplejson.JSONEncoder(sort_keys=True)
296

    
297
  def __init__(self, host, port=GANETI_RAPI_PORT,
298
               username=None, password=None,
299
               config_ssl_verification=None, ignore_proxy=False,
300
               logger=logging):
301
    """Constructor.
302

303
    @type host: string
304
    @param host: the ganeti cluster master to interact with
305
    @type port: int
306
    @param port: the port on which the RAPI is running (default is 5080)
307
    @type username: string
308
    @param username: the username to connect with
309
    @type password: string
310
    @param password: the password to connect with
311
    @type config_ssl_verification: callable
312
    @param config_ssl_verification: Function to configure SSL context for
313
                                    certificate verification
314
    @type ignore_proxy: bool
315
    @param ignore_proxy: Whether to ignore proxy settings
316
    @param logger: Logging object
317

318
    """
319
    self._host = host
320
    self._port = port
321
    self._logger = logger
322

    
323
    self._base_url = "https://%s:%s" % (host, port)
324

    
325
    handlers = [_HTTPSHandler(self._logger, config_ssl_verification)]
326

    
327
    if username is not None:
328
      pwmgr = urllib2.HTTPPasswordMgrWithDefaultRealm()
329
      pwmgr.add_password(None, self._base_url, username, password)
330
      handlers.append(urllib2.HTTPBasicAuthHandler(pwmgr))
331
    elif password:
332
      raise Error("Specified password without username")
333

    
334
    if ignore_proxy:
335
      handlers.append(urllib2.ProxyHandler({}))
336

    
337
    self._http = urllib2.build_opener(*handlers) # pylint: disable-msg=W0142
338

    
339
    self._headers = {
340
      "Accept": HTTP_APP_JSON,
341
      "Content-type": HTTP_APP_JSON,
342
      "User-Agent": self.USER_AGENT,
343
      }
344

    
345
  @staticmethod
346
  def _EncodeQuery(query):
347
    """Encode query values for RAPI URL.
348

349
    @type query: list of two-tuples
350
    @param query: Query arguments
351
    @rtype: list
352
    @return: Query list with encoded values
353

354
    """
355
    result = []
356

    
357
    for name, value in query:
358
      if value is None:
359
        result.append((name, ""))
360

    
361
      elif isinstance(value, bool):
362
        # Boolean values must be encoded as 0 or 1
363
        result.append((name, int(value)))
364

    
365
      elif isinstance(value, (list, tuple, dict)):
366
        raise ValueError("Invalid query data type %r" % type(value).__name__)
367

    
368
      else:
369
        result.append((name, value))
370

    
371
    return result
372

    
373
  def _SendRequest(self, method, path, query, content):
374
    """Sends an HTTP request.
375

376
    This constructs a full URL, encodes and decodes HTTP bodies, and
377
    handles invalid responses in a pythonic way.
378

379
    @type method: string
380
    @param method: HTTP method to use
381
    @type path: string
382
    @param path: HTTP URL path
383
    @type query: list of two-tuples
384
    @param query: query arguments to pass to urllib.urlencode
385
    @type content: str or None
386
    @param content: HTTP body content
387

388
    @rtype: str
389
    @return: JSON-Decoded response
390

391
    @raises CertificateError: If an invalid SSL certificate is found
392
    @raises GanetiApiError: If an invalid response is returned
393

394
    """
395
    assert path.startswith("/")
396

    
397
    if content:
398
      encoded_content = self._json_encoder.encode(content)
399
    else:
400
      encoded_content = None
401

    
402
    # Build URL
403
    urlparts = [self._base_url, path]
404
    if query:
405
      urlparts.append("?")
406
      urlparts.append(urllib.urlencode(self._EncodeQuery(query)))
407

    
408
    url = "".join(urlparts)
409

    
410
    self._logger.debug("Sending request %s %s to %s:%s"
411
                       " (headers=%r, content=%r)",
412
                       method, url, self._host, self._port, self._headers,
413
                       encoded_content)
414

    
415
    req = _RapiRequest(method, url, self._headers, encoded_content)
416

    
417
    try:
418
      resp = self._http.open(req)
419
      encoded_response_content = resp.read()
420
    except (OpenSSL.SSL.Error, OpenSSL.crypto.Error), err:
421
      raise CertificateError("SSL issue: %s (%r)" % (err, err))
422
    except urllib2.HTTPError, err:
423
      raise GanetiApiError(str(err), code=err.code)
424

    
425
    if encoded_response_content:
426
      response_content = simplejson.loads(encoded_response_content)
427
    else:
428
      response_content = None
429

    
430
    # TODO: Are there other status codes that are valid? (redirect?)
431
    if resp.code != HTTP_OK:
432
      if isinstance(response_content, dict):
433
        msg = ("%s %s: %s" %
434
               (response_content["code"],
435
                response_content["message"],
436
                response_content["explain"]))
437
      else:
438
        msg = str(response_content)
439

    
440
      raise GanetiApiError(msg, code=resp.code)
441

    
442
    return response_content
443

    
444
  def GetVersion(self):
445
    """Gets the Remote API version running on the cluster.
446

447
    @rtype: int
448
    @return: Ganeti Remote API version
449

450
    """
451
    return self._SendRequest(HTTP_GET, "/version", None, None)
452

    
453
  def GetFeatures(self):
454
    """Gets the list of optional features supported by RAPI server.
455

456
    @rtype: list
457
    @return: List of optional features
458

459
    """
460
    try:
461
      return self._SendRequest(HTTP_GET, "/%s/features" % GANETI_RAPI_VERSION,
462
                               None, None)
463
    except GanetiApiError, err:
464
      # Older RAPI servers don't support this resource
465
      if err.code == HTTP_NOT_FOUND:
466
        return []
467

    
468
      raise
469

    
470
  def GetOperatingSystems(self):
471
    """Gets the Operating Systems running in the Ganeti cluster.
472

473
    @rtype: list of str
474
    @return: operating systems
475

476
    """
477
    return self._SendRequest(HTTP_GET, "/%s/os" % GANETI_RAPI_VERSION,
478
                             None, None)
479

    
480
  def GetInfo(self):
481
    """Gets info about the cluster.
482

483
    @rtype: dict
484
    @return: information about the cluster
485

486
    """
487
    return self._SendRequest(HTTP_GET, "/%s/info" % GANETI_RAPI_VERSION,
488
                             None, None)
489

    
490
  def GetClusterTags(self):
491
    """Gets the cluster tags.
492

493
    @rtype: list of str
494
    @return: cluster tags
495

496
    """
497
    return self._SendRequest(HTTP_GET, "/%s/tags" % GANETI_RAPI_VERSION,
498
                             None, None)
499

    
500
  def AddClusterTags(self, tags, dry_run=False):
501
    """Adds tags to the cluster.
502

503
    @type tags: list of str
504
    @param tags: tags to add to the cluster
505
    @type dry_run: bool
506
    @param dry_run: whether to perform a dry run
507

508
    @rtype: int
509
    @return: job id
510

511
    """
512
    query = [("tag", t) for t in tags]
513
    if dry_run:
514
      query.append(("dry-run", 1))
515

    
516
    return self._SendRequest(HTTP_PUT, "/%s/tags" % GANETI_RAPI_VERSION,
517
                             query, None)
518

    
519
  def DeleteClusterTags(self, tags, dry_run=False):
520
    """Deletes tags from the cluster.
521

522
    @type tags: list of str
523
    @param tags: tags to delete
524
    @type dry_run: bool
525
    @param dry_run: whether to perform a dry run
526

527
    """
528
    query = [("tag", t) for t in tags]
529
    if dry_run:
530
      query.append(("dry-run", 1))
531

    
532
    return self._SendRequest(HTTP_DELETE, "/%s/tags" % GANETI_RAPI_VERSION,
533
                             query, None)
534

    
535
  def GetInstances(self, bulk=False):
536
    """Gets information about instances on the cluster.
537

538
    @type bulk: bool
539
    @param bulk: whether to return all information about all instances
540

541
    @rtype: list of dict or list of str
542
    @return: if bulk is True, info about the instances, else a list of instances
543

544
    """
545
    query = []
546
    if bulk:
547
      query.append(("bulk", 1))
548

    
549
    instances = self._SendRequest(HTTP_GET,
550
                                  "/%s/instances" % GANETI_RAPI_VERSION,
551
                                  query, None)
552
    if bulk:
553
      return instances
554
    else:
555
      return [i["id"] for i in instances]
556

    
557
  def GetInstance(self, instance):
558
    """Gets information about an instance.
559

560
    @type instance: str
561
    @param instance: instance whose info to return
562

563
    @rtype: dict
564
    @return: info about the instance
565

566
    """
567
    return self._SendRequest(HTTP_GET,
568
                             ("/%s/instances/%s" %
569
                              (GANETI_RAPI_VERSION, instance)), None, None)
570

    
571
  def GetInstanceInfo(self, instance, static=None):
572
    """Gets information about an instance.
573

574
    @type instance: string
575
    @param instance: Instance name
576
    @rtype: string
577
    @return: Job ID
578

579
    """
580
    if static is not None:
581
      query = [("static", static)]
582
    else:
583
      query = None
584

    
585
    return self._SendRequest(HTTP_GET,
586
                             ("/%s/instances/%s/info" %
587
                              (GANETI_RAPI_VERSION, instance)), query, None)
588

    
589
  def CreateInstance(self, mode, name, disk_template, disks, nics,
590
                     **kwargs):
591
    """Creates a new instance.
592

593
    More details for parameters can be found in the RAPI documentation.
594

595
    @type mode: string
596
    @param mode: Instance creation mode
597
    @type name: string
598
    @param name: Hostname of the instance to create
599
    @type disk_template: string
600
    @param disk_template: Disk template for instance (e.g. plain, diskless,
601
                          file, or drbd)
602
    @type disks: list of dicts
603
    @param disks: List of disk definitions
604
    @type nics: list of dicts
605
    @param nics: List of NIC definitions
606
    @type dry_run: bool
607
    @keyword dry_run: whether to perform a dry run
608

609
    @rtype: int
610
    @return: job id
611

612
    """
613
    query = []
614

    
615
    if kwargs.get("dry_run"):
616
      query.append(("dry-run", 1))
617

    
618
    if _INST_CREATE_REQV1 in self.GetFeatures():
619
      # All required fields for request data version 1
620
      body = {
621
        _REQ_DATA_VERSION_FIELD: 1,
622
        "mode": mode,
623
        "name": name,
624
        "disk_template": disk_template,
625
        "disks": disks,
626
        "nics": nics,
627
        }
628

    
629
      conflicts = set(kwargs.iterkeys()) & set(body.iterkeys())
630
      if conflicts:
631
        raise GanetiApiError("Required fields can not be specified as"
632
                             " keywords: %s" % ", ".join(conflicts))
633

    
634
      body.update((key, value) for key, value in kwargs.iteritems()
635
                  if key != "dry_run")
636
    else:
637
      # TODO: Implement instance creation request data version 0
638
      # When implementing version 0, care should be taken to refuse unknown
639
      # parameters and invalid values. The interface of this function must stay
640
      # exactly the same for version 0 and 1 (e.g. they aren't allowed to
641
      # require different data types).
642
      raise NotImplementedError("Support for instance creation request data"
643
                                " version 0 is not yet implemented")
644

    
645
    return self._SendRequest(HTTP_POST, "/%s/instances" % GANETI_RAPI_VERSION,
646
                             query, body)
647

    
648
  def DeleteInstance(self, instance, dry_run=False):
649
    """Deletes an instance.
650

651
    @type instance: str
652
    @param instance: the instance to delete
653

654
    @rtype: int
655
    @return: job id
656

657
    """
658
    query = []
659
    if dry_run:
660
      query.append(("dry-run", 1))
661

    
662
    return self._SendRequest(HTTP_DELETE,
663
                             ("/%s/instances/%s" %
664
                              (GANETI_RAPI_VERSION, instance)), query, None)
665

    
666
  def GetInstanceTags(self, instance):
667
    """Gets tags for an instance.
668

669
    @type instance: str
670
    @param instance: instance whose tags to return
671

672
    @rtype: list of str
673
    @return: tags for the instance
674

675
    """
676
    return self._SendRequest(HTTP_GET,
677
                             ("/%s/instances/%s/tags" %
678
                              (GANETI_RAPI_VERSION, instance)), None, None)
679

    
680
  def AddInstanceTags(self, instance, tags, dry_run=False):
681
    """Adds tags to an instance.
682

683
    @type instance: str
684
    @param instance: instance to add tags to
685
    @type tags: list of str
686
    @param tags: tags to add to the instance
687
    @type dry_run: bool
688
    @param dry_run: whether to perform a dry run
689

690
    @rtype: int
691
    @return: job id
692

693
    """
694
    query = [("tag", t) for t in tags]
695
    if dry_run:
696
      query.append(("dry-run", 1))
697

    
698
    return self._SendRequest(HTTP_PUT,
699
                             ("/%s/instances/%s/tags" %
700
                              (GANETI_RAPI_VERSION, instance)), query, None)
701

    
702
  def DeleteInstanceTags(self, instance, tags, dry_run=False):
703
    """Deletes tags from an instance.
704

705
    @type instance: str
706
    @param instance: instance to delete tags from
707
    @type tags: list of str
708
    @param tags: tags to delete
709
    @type dry_run: bool
710
    @param dry_run: whether to perform a dry run
711

712
    """
713
    query = [("tag", t) for t in tags]
714
    if dry_run:
715
      query.append(("dry-run", 1))
716

    
717
    return self._SendRequest(HTTP_DELETE,
718
                             ("/%s/instances/%s/tags" %
719
                              (GANETI_RAPI_VERSION, instance)), query, None)
720

    
721
  def RebootInstance(self, instance, reboot_type=None, ignore_secondaries=None,
722
                     dry_run=False):
723
    """Reboots an instance.
724

725
    @type instance: str
726
    @param instance: instance to rebot
727
    @type reboot_type: str
728
    @param reboot_type: one of: hard, soft, full
729
    @type ignore_secondaries: bool
730
    @param ignore_secondaries: if True, ignores errors for the secondary node
731
        while re-assembling disks (in hard-reboot mode only)
732
    @type dry_run: bool
733
    @param dry_run: whether to perform a dry run
734

735
    """
736
    query = []
737
    if reboot_type:
738
      query.append(("type", reboot_type))
739
    if ignore_secondaries is not None:
740
      query.append(("ignore_secondaries", ignore_secondaries))
741
    if dry_run:
742
      query.append(("dry-run", 1))
743

    
744
    return self._SendRequest(HTTP_POST,
745
                             ("/%s/instances/%s/reboot" %
746
                              (GANETI_RAPI_VERSION, instance)), query, None)
747

    
748
  def ShutdownInstance(self, instance, dry_run=False):
749
    """Shuts down an instance.
750

751
    @type instance: str
752
    @param instance: the instance to shut down
753
    @type dry_run: bool
754
    @param dry_run: whether to perform a dry run
755

756
    """
757
    query = []
758
    if dry_run:
759
      query.append(("dry-run", 1))
760

    
761
    return self._SendRequest(HTTP_PUT,
762
                             ("/%s/instances/%s/shutdown" %
763
                              (GANETI_RAPI_VERSION, instance)), query, None)
764

    
765
  def StartupInstance(self, instance, dry_run=False):
766
    """Starts up an instance.
767

768
    @type instance: str
769
    @param instance: the instance to start up
770
    @type dry_run: bool
771
    @param dry_run: whether to perform a dry run
772

773
    """
774
    query = []
775
    if dry_run:
776
      query.append(("dry-run", 1))
777

    
778
    return self._SendRequest(HTTP_PUT,
779
                             ("/%s/instances/%s/startup" %
780
                              (GANETI_RAPI_VERSION, instance)), query, None)
781

    
782
  def ReinstallInstance(self, instance, os, no_startup=False):
783
    """Reinstalls an instance.
784

785
    @type instance: str
786
    @param instance: the instance to reinstall
787
    @type os: str
788
    @param os: the os to reinstall
789
    @type no_startup: bool
790
    @param no_startup: whether to start the instance automatically
791

792
    """
793
    query = [("os", os)]
794
    if no_startup:
795
      query.append(("nostartup", 1))
796
    return self._SendRequest(HTTP_POST,
797
                             ("/%s/instances/%s/reinstall" %
798
                              (GANETI_RAPI_VERSION, instance)), query, None)
799

    
800
  def ReplaceInstanceDisks(self, instance, disks=None, mode=REPLACE_DISK_AUTO,
801
                           remote_node=None, iallocator=None, dry_run=False):
802
    """Replaces disks on an instance.
803

804
    @type instance: str
805
    @param instance: instance whose disks to replace
806
    @type disks: list of ints
807
    @param disks: Indexes of disks to replace
808
    @type mode: str
809
    @param mode: replacement mode to use (defaults to replace_auto)
810
    @type remote_node: str or None
811
    @param remote_node: new secondary node to use (for use with
812
        replace_new_secondary mode)
813
    @type iallocator: str or None
814
    @param iallocator: instance allocator plugin to use (for use with
815
                       replace_auto mode)
816
    @type dry_run: bool
817
    @param dry_run: whether to perform a dry run
818

819
    @rtype: int
820
    @return: job id
821

822
    """
823
    query = [
824
      ("mode", mode),
825
      ]
826

    
827
    if disks:
828
      query.append(("disks", ",".join(str(idx) for idx in disks)))
829

    
830
    if remote_node:
831
      query.append(("remote_node", remote_node))
832

    
833
    if iallocator:
834
      query.append(("iallocator", iallocator))
835

    
836
    if dry_run:
837
      query.append(("dry-run", 1))
838

    
839
    return self._SendRequest(HTTP_POST,
840
                             ("/%s/instances/%s/replace-disks" %
841
                              (GANETI_RAPI_VERSION, instance)), query, None)
842

    
843
  def GetJobs(self):
844
    """Gets all jobs for the cluster.
845

846
    @rtype: list of int
847
    @return: job ids for the cluster
848

849
    """
850
    return [int(j["id"])
851
            for j in self._SendRequest(HTTP_GET,
852
                                       "/%s/jobs" % GANETI_RAPI_VERSION,
853
                                       None, None)]
854

    
855
  def GetJobStatus(self, job_id):
856
    """Gets the status of a job.
857

858
    @type job_id: int
859
    @param job_id: job id whose status to query
860

861
    @rtype: dict
862
    @return: job status
863

864
    """
865
    return self._SendRequest(HTTP_GET,
866
                             "/%s/jobs/%s" % (GANETI_RAPI_VERSION, job_id),
867
                             None, None)
868

    
869
  def WaitForJobChange(self, job_id, fields, prev_job_info, prev_log_serial):
870
    """Waits for job changes.
871

872
    @type job_id: int
873
    @param job_id: Job ID for which to wait
874

875
    """
876
    body = {
877
      "fields": fields,
878
      "previous_job_info": prev_job_info,
879
      "previous_log_serial": prev_log_serial,
880
      }
881

    
882
    return self._SendRequest(HTTP_GET,
883
                             "/%s/jobs/%s/wait" % (GANETI_RAPI_VERSION, job_id),
884
                             None, body)
885

    
886
  def CancelJob(self, job_id, dry_run=False):
887
    """Cancels a job.
888

889
    @type job_id: int
890
    @param job_id: id of the job to delete
891
    @type dry_run: bool
892
    @param dry_run: whether to perform a dry run
893

894
    """
895
    query = []
896
    if dry_run:
897
      query.append(("dry-run", 1))
898

    
899
    return self._SendRequest(HTTP_DELETE,
900
                             "/%s/jobs/%s" % (GANETI_RAPI_VERSION, job_id),
901
                             query, None)
902

    
903
  def GetNodes(self, bulk=False):
904
    """Gets all nodes in the cluster.
905

906
    @type bulk: bool
907
    @param bulk: whether to return all information about all instances
908

909
    @rtype: list of dict or str
910
    @return: if bulk is true, info about nodes in the cluster,
911
        else list of nodes in the cluster
912

913
    """
914
    query = []
915
    if bulk:
916
      query.append(("bulk", 1))
917

    
918
    nodes = self._SendRequest(HTTP_GET, "/%s/nodes" % GANETI_RAPI_VERSION,
919
                              query, None)
920
    if bulk:
921
      return nodes
922
    else:
923
      return [n["id"] for n in nodes]
924

    
925
  def GetNode(self, node):
926
    """Gets information about a node.
927

928
    @type node: str
929
    @param node: node whose info to return
930

931
    @rtype: dict
932
    @return: info about the node
933

934
    """
935
    return self._SendRequest(HTTP_GET,
936
                             "/%s/nodes/%s" % (GANETI_RAPI_VERSION, node),
937
                             None, None)
938

    
939
  def EvacuateNode(self, node, iallocator=None, remote_node=None,
940
                   dry_run=False):
941
    """Evacuates instances from a Ganeti node.
942

943
    @type node: str
944
    @param node: node to evacuate
945
    @type iallocator: str or None
946
    @param iallocator: instance allocator to use
947
    @type remote_node: str
948
    @param remote_node: node to evaucate to
949
    @type dry_run: bool
950
    @param dry_run: whether to perform a dry run
951

952
    @rtype: int
953
    @return: job id
954

955
    @raises GanetiApiError: if an iallocator and remote_node are both specified
956

957
    """
958
    if iallocator and remote_node:
959
      raise GanetiApiError("Only one of iallocator or remote_node can be used")
960

    
961
    query = []
962
    if iallocator:
963
      query.append(("iallocator", iallocator))
964
    if remote_node:
965
      query.append(("remote_node", remote_node))
966
    if dry_run:
967
      query.append(("dry-run", 1))
968

    
969
    return self._SendRequest(HTTP_POST,
970
                             ("/%s/nodes/%s/evacuate" %
971
                              (GANETI_RAPI_VERSION, node)), query, None)
972

    
973
  def MigrateNode(self, node, live=True, dry_run=False):
974
    """Migrates all primary instances from a node.
975

976
    @type node: str
977
    @param node: node to migrate
978
    @type live: bool
979
    @param live: whether to use live migration
980
    @type dry_run: bool
981
    @param dry_run: whether to perform a dry run
982

983
    @rtype: int
984
    @return: job id
985

986
    """
987
    query = []
988
    if live:
989
      query.append(("live", 1))
990
    if dry_run:
991
      query.append(("dry-run", 1))
992

    
993
    return self._SendRequest(HTTP_POST,
994
                             ("/%s/nodes/%s/migrate" %
995
                              (GANETI_RAPI_VERSION, node)), query, None)
996

    
997
  def GetNodeRole(self, node):
998
    """Gets the current role for a node.
999

1000
    @type node: str
1001
    @param node: node whose role to return
1002

1003
    @rtype: str
1004
    @return: the current role for a node
1005

1006
    """
1007
    return self._SendRequest(HTTP_GET,
1008
                             ("/%s/nodes/%s/role" %
1009
                              (GANETI_RAPI_VERSION, node)), None, None)
1010

    
1011
  def SetNodeRole(self, node, role, force=False):
1012
    """Sets the role for a node.
1013

1014
    @type node: str
1015
    @param node: the node whose role to set
1016
    @type role: str
1017
    @param role: the role to set for the node
1018
    @type force: bool
1019
    @param force: whether to force the role change
1020

1021
    @rtype: int
1022
    @return: job id
1023

1024
    """
1025
    query = [
1026
      ("force", force),
1027
      ]
1028

    
1029
    return self._SendRequest(HTTP_PUT,
1030
                             ("/%s/nodes/%s/role" %
1031
                              (GANETI_RAPI_VERSION, node)), query, role)
1032

    
1033
  def GetNodeStorageUnits(self, node, storage_type, output_fields):
1034
    """Gets the storage units for a node.
1035

1036
    @type node: str
1037
    @param node: the node whose storage units to return
1038
    @type storage_type: str
1039
    @param storage_type: storage type whose units to return
1040
    @type output_fields: str
1041
    @param output_fields: storage type fields to return
1042

1043
    @rtype: int
1044
    @return: job id where results can be retrieved
1045

1046
    """
1047
    query = [
1048
      ("storage_type", storage_type),
1049
      ("output_fields", output_fields),
1050
      ]
1051

    
1052
    return self._SendRequest(HTTP_GET,
1053
                             ("/%s/nodes/%s/storage" %
1054
                              (GANETI_RAPI_VERSION, node)), query, None)
1055

    
1056
  def ModifyNodeStorageUnits(self, node, storage_type, name, allocatable=None):
1057
    """Modifies parameters of storage units on the node.
1058

1059
    @type node: str
1060
    @param node: node whose storage units to modify
1061
    @type storage_type: str
1062
    @param storage_type: storage type whose units to modify
1063
    @type name: str
1064
    @param name: name of the storage unit
1065
    @type allocatable: bool or None
1066
    @param allocatable: Whether to set the "allocatable" flag on the storage
1067
                        unit (None=no modification, True=set, False=unset)
1068

1069
    @rtype: int
1070
    @return: job id
1071

1072
    """
1073
    query = [
1074
      ("storage_type", storage_type),
1075
      ("name", name),
1076
      ]
1077

    
1078
    if allocatable is not None:
1079
      query.append(("allocatable", allocatable))
1080

    
1081
    return self._SendRequest(HTTP_PUT,
1082
                             ("/%s/nodes/%s/storage/modify" %
1083
                              (GANETI_RAPI_VERSION, node)), query, None)
1084

    
1085
  def RepairNodeStorageUnits(self, node, storage_type, name):
1086
    """Repairs a storage unit on the node.
1087

1088
    @type node: str
1089
    @param node: node whose storage units to repair
1090
    @type storage_type: str
1091
    @param storage_type: storage type to repair
1092
    @type name: str
1093
    @param name: name of the storage unit to repair
1094

1095
    @rtype: int
1096
    @return: job id
1097

1098
    """
1099
    query = [
1100
      ("storage_type", storage_type),
1101
      ("name", name),
1102
      ]
1103

    
1104
    return self._SendRequest(HTTP_PUT,
1105
                             ("/%s/nodes/%s/storage/repair" %
1106
                              (GANETI_RAPI_VERSION, node)), query, None)
1107

    
1108
  def GetNodeTags(self, node):
1109
    """Gets the tags for a node.
1110

1111
    @type node: str
1112
    @param node: node whose tags to return
1113

1114
    @rtype: list of str
1115
    @return: tags for the node
1116

1117
    """
1118
    return self._SendRequest(HTTP_GET,
1119
                             ("/%s/nodes/%s/tags" %
1120
                              (GANETI_RAPI_VERSION, node)), None, None)
1121

    
1122
  def AddNodeTags(self, node, tags, dry_run=False):
1123
    """Adds tags to a node.
1124

1125
    @type node: str
1126
    @param node: node to add tags to
1127
    @type tags: list of str
1128
    @param tags: tags to add to the node
1129
    @type dry_run: bool
1130
    @param dry_run: whether to perform a dry run
1131

1132
    @rtype: int
1133
    @return: job id
1134

1135
    """
1136
    query = [("tag", t) for t in tags]
1137
    if dry_run:
1138
      query.append(("dry-run", 1))
1139

    
1140
    return self._SendRequest(HTTP_PUT,
1141
                             ("/%s/nodes/%s/tags" %
1142
                              (GANETI_RAPI_VERSION, node)), query, tags)
1143

    
1144
  def DeleteNodeTags(self, node, tags, dry_run=False):
1145
    """Delete tags from a node.
1146

1147
    @type node: str
1148
    @param node: node to remove tags from
1149
    @type tags: list of str
1150
    @param tags: tags to remove from the node
1151
    @type dry_run: bool
1152
    @param dry_run: whether to perform a dry run
1153

1154
    @rtype: int
1155
    @return: job id
1156

1157
    """
1158
    query = [("tag", t) for t in tags]
1159
    if dry_run:
1160
      query.append(("dry-run", 1))
1161

    
1162
    return self._SendRequest(HTTP_DELETE,
1163
                             ("/%s/nodes/%s/tags" %
1164
                              (GANETI_RAPI_VERSION, node)), query, None)