ganeti-noded 16.2 KB
Newer Older
Iustin Pop's avatar
Iustin Pop committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#!/usr/bin/python
#

# Copyright (C) 2006, 2007 Google Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301, USA.


"""Ganeti node daemon"""

24
25
26
# functions in this module need to have a given name structure, so:
# pylint: disable-msg=C0103

Iustin Pop's avatar
Iustin Pop committed
27
28
29
import os
import sys
import traceback
Guido Trotter's avatar
Guido Trotter committed
30
import SocketServer
31
import errno
Iustin Pop's avatar
Iustin Pop committed
32
import logging
Iustin Pop's avatar
Iustin Pop committed
33
34
35
36
37
38
39
40
41
42

from optparse import OptionParser


from ganeti import backend
from ganeti import logger
from ganeti import constants
from ganeti import objects
from ganeti import errors
from ganeti import ssconf
43
from ganeti import http
44
from ganeti import utils
Iustin Pop's avatar
Iustin Pop committed
45

Iustin Pop's avatar
Iustin Pop committed
46
47
_EXIT_GANETI_NODED = False

Iustin Pop's avatar
Iustin Pop committed
48

49
class NodeDaemonRequestHandler(http.HTTPRequestHandler):
50
51
52
53
54
  """The server implementation.

  This class holds all methods exposed over the RPC interface.

  """
55
56
  def HandleRequest(self):
    """Handle a request.
Iustin Pop's avatar
Iustin Pop committed
57

58
    """
59
60
61
62
63
    global _EXIT_GANETI_NODED

    if self.command.upper() != "PUT":
      raise http.HTTPBadRequest()

Iustin Pop's avatar
Iustin Pop committed
64
65
66
67
    path = self.path
    if path.startswith("/"):
      path = path[1:]

68
69
70
    method = getattr(self, "perspective_%s" % path, None)
    if method is None:
      raise httperror.HTTPNotFound()
Iustin Pop's avatar
Iustin Pop committed
71

Iustin Pop's avatar
Iustin Pop committed
72
    try:
73
      return method(self.post_data)
74
75
    except errors.QuitGanetiException, err:
      _EXIT_GANETI_NODED = True
Iustin Pop's avatar
Iustin Pop committed
76
77
78

  # the new block devices  --------------------------

79
80
81
82
83
  @staticmethod
  def perspective_blockdev_create(params):
    """Create a block device.

    """
84
    bdev_s, size, owner, on_primary, info = params
85
    bdev = objects.Disk.FromDict(bdev_s)
Iustin Pop's avatar
Iustin Pop committed
86
87
    if bdev is None:
      raise ValueError("can't unserialize data!")
88
    return backend.CreateBlockDevice(bdev, size, owner, on_primary, info)
Iustin Pop's avatar
Iustin Pop committed
89

90
91
92
93
94
  @staticmethod
  def perspective_blockdev_remove(params):
    """Remove a block device.

    """
Iustin Pop's avatar
Iustin Pop committed
95
    bdev_s = params[0]
96
    bdev = objects.Disk.FromDict(bdev_s)
Iustin Pop's avatar
Iustin Pop committed
97
98
    return backend.RemoveBlockDevice(bdev)

Iustin Pop's avatar
Iustin Pop committed
99
100
101
102
103
104
105
106
  @staticmethod
  def perspective_blockdev_rename(params):
    """Remove a block device.

    """
    devlist = [(objects.Disk.FromDict(ds), uid) for ds, uid in params]
    return backend.RenameBlockDevices(devlist)

107
108
109
110
111
  @staticmethod
  def perspective_blockdev_assemble(params):
    """Assemble a block device.

    """
112
    bdev_s, owner, on_primary = params
113
    bdev = objects.Disk.FromDict(bdev_s)
Iustin Pop's avatar
Iustin Pop committed
114
115
    if bdev is None:
      raise ValueError("can't unserialize data!")
116
    return backend.AssembleBlockDevice(bdev, owner, on_primary)
Iustin Pop's avatar
Iustin Pop committed
117

118
119
120
121
122
  @staticmethod
  def perspective_blockdev_shutdown(params):
    """Shutdown a block device.

    """
Iustin Pop's avatar
Iustin Pop committed
123
    bdev_s = params[0]
124
    bdev = objects.Disk.FromDict(bdev_s)
Iustin Pop's avatar
Iustin Pop committed
125
126
127
128
    if bdev is None:
      raise ValueError("can't unserialize data!")
    return backend.ShutdownBlockDevice(bdev)

129
  @staticmethod
130
  def perspective_blockdev_addchildren(params):
131
132
133
134
135
136
    """Add a child to a mirror device.

    Note: this is only valid for mirror devices. It's the caller's duty
    to send a correct disk, otherwise we raise an error.

    """
Iustin Pop's avatar
Iustin Pop committed
137
    bdev_s, ndev_s = params
138
    bdev = objects.Disk.FromDict(bdev_s)
139
140
    ndevs = [objects.Disk.FromDict(disk_s) for disk_s in ndev_s]
    if bdev is None or ndevs.count(None) > 0:
Iustin Pop's avatar
Iustin Pop committed
141
      raise ValueError("can't unserialize data!")
142
    return backend.MirrorAddChildren(bdev, ndevs)
Iustin Pop's avatar
Iustin Pop committed
143

144
  @staticmethod
145
  def perspective_blockdev_removechildren(params):
146
147
148
149
150
151
    """Remove a child from a mirror device.

    This is only valid for mirror devices, of course. It's the callers
    duty to send a correct disk, otherwise we raise an error.

    """
Iustin Pop's avatar
Iustin Pop committed
152
    bdev_s, ndev_s = params
153
    bdev = objects.Disk.FromDict(bdev_s)
154
155
    ndevs = [objects.Disk.FromDict(disk_s) for disk_s in ndev_s]
    if bdev is None or ndevs.count(None) > 0:
Iustin Pop's avatar
Iustin Pop committed
156
      raise ValueError("can't unserialize data!")
157
    return backend.MirrorRemoveChildren(bdev, ndevs)
Iustin Pop's avatar
Iustin Pop committed
158

159
160
161
162
163
  @staticmethod
  def perspective_blockdev_getmirrorstatus(params):
    """Return the mirror status for a list of disks.

    """
164
    disks = [objects.Disk.FromDict(dsk_s)
Iustin Pop's avatar
Iustin Pop committed
165
166
167
            for dsk_s in params]
    return backend.GetMirrorStatus(disks)

168
169
170
171
172
173
174
  @staticmethod
  def perspective_blockdev_find(params):
    """Expose the FindBlockDevice functionality for a disk.

    This will try to find but not activate a disk.

    """
175
    disk = objects.Disk.FromDict(params[0])
Iustin Pop's avatar
Iustin Pop committed
176
177
    return backend.FindBlockDevice(disk)

178
179
180
181
182
183
184
185
186
  @staticmethod
  def perspective_blockdev_snapshot(params):
    """Create a snapshot device.

    Note that this is only valid for LVM disks, if we get passed
    something else we raise an exception. The snapshot device can be
    remove by calling the generic block device remove call.

    """
187
    cfbd = objects.Disk.FromDict(params[0])
Iustin Pop's avatar
Iustin Pop committed
188
189
    return backend.SnapshotBlockDevice(cfbd)

190
191
192
193
194
195
196
197
198
  @staticmethod
  def perspective_blockdev_grow(params):
    """Grow a stack of devices.

    """
    cfbd = objects.Disk.FromDict(params[0])
    amount = params[1]
    return backend.GrowBlockDevice(cfbd, amount)

199
200
201
202
203
204
205
206
  @staticmethod
  def perspective_blockdev_close(params):
    """Closes the given block devices.

    """
    disks = [objects.Disk.FromDict(cf) for cf in params]
    return backend.CloseBlockDevices(disks)

Iustin Pop's avatar
Iustin Pop committed
207
208
  # export/import  --------------------------

209
210
211
212
213
  @staticmethod
  def perspective_snapshot_export(params):
    """Export a given snapshot.

    """
214
    disk = objects.Disk.FromDict(params[0])
Iustin Pop's avatar
Iustin Pop committed
215
    dest_node = params[1]
216
    instance = objects.Instance.FromDict(params[2])
217
218
219
220
221
    return backend.ExportSnapshot(disk, dest_node, instance)

  @staticmethod
  def perspective_finalize_export(params):
    """Expose the finalize export functionality.
Iustin Pop's avatar
Iustin Pop committed
222

223
    """
224
225
    instance = objects.Instance.FromDict(params[0])
    snap_disks = [objects.Disk.FromDict(str_data)
Iustin Pop's avatar
Iustin Pop committed
226
227
228
                  for str_data in params[1]]
    return backend.FinalizeExport(instance, snap_disks)

229
230
231
232
233
234
235
236
237
238
  @staticmethod
  def perspective_export_info(params):
    """Query information about an existing export on this node.

    The given path may not contain an export, in which case we return
    None.

    """
    path = params[0]
    einfo = backend.ExportInfo(path)
Iustin Pop's avatar
Iustin Pop committed
239
240
241
242
    if einfo is None:
      return einfo
    return einfo.Dumps()

243
244
245
246
247
248
249
250
251
  @staticmethod
  def perspective_export_list(params):
    """List the available exports on this node.

    Note that as opposed to export_info, which may query data about an
    export in any path, this only queries the standard Ganeti path
    (constants.EXPORT_DIR).

    """
Iustin Pop's avatar
Iustin Pop committed
252
253
    return backend.ListExports()

254
255
256
257
258
  @staticmethod
  def perspective_export_remove(params):
    """Remove an export.

    """
Iustin Pop's avatar
Iustin Pop committed
259
260
261
262
263
    export = params[0]
    return backend.RemoveExport(export)

  # volume  --------------------------

264
265
266
267
268
  @staticmethod
  def perspective_volume_list(params):
    """Query the list of logical volumes in a given volume group.

    """
Iustin Pop's avatar
Iustin Pop committed
269
270
271
    vgname = params[0]
    return backend.GetVolumeList(vgname)

272
273
274
275
276
  @staticmethod
  def perspective_vg_list(params):
    """Query the list of volume groups.

    """
Iustin Pop's avatar
Iustin Pop committed
277
278
279
280
    return backend.ListVolumeGroups()

  # bridge  --------------------------

281
282
283
284
285
  @staticmethod
  def perspective_bridges_exist(params):
    """Check if all bridges given exist on this node.

    """
Iustin Pop's avatar
Iustin Pop committed
286
287
288
289
290
    bridges_list = params[0]
    return backend.BridgesExist(bridges_list)

  # instance  --------------------------

291
292
293
294
295
  @staticmethod
  def perspective_instance_os_add(params):
    """Install an OS on a given instance.

    """
Iustin Pop's avatar
Iustin Pop committed
296
    inst_s, os_disk, swap_disk = params
297
    inst = objects.Instance.FromDict(inst_s)
Iustin Pop's avatar
Iustin Pop committed
298
299
    return backend.AddOSToInstance(inst, os_disk, swap_disk)

300
301
302
303
304
305
  @staticmethod
  def perspective_instance_run_rename(params):
    """Runs the OS rename script for an instance.

    """
    inst_s, old_name, os_disk, swap_disk = params
306
    inst = objects.Instance.FromDict(inst_s)
307
308
    return backend.RunRenameInstance(inst, old_name, os_disk, swap_disk)

309
310
311
312
313
  @staticmethod
  def perspective_instance_os_import(params):
    """Run the import function of an OS onto a given instance.

    """
Iustin Pop's avatar
Iustin Pop committed
314
    inst_s, os_disk, swap_disk, src_node, src_image = params
315
    inst = objects.Instance.FromDict(inst_s)
Iustin Pop's avatar
Iustin Pop committed
316
317
318
    return backend.ImportOSIntoInstance(inst, os_disk, swap_disk,
                                        src_node, src_image)

319
320
321
322
323
  @staticmethod
  def perspective_instance_shutdown(params):
    """Shutdown an instance.

    """
324
    instance = objects.Instance.FromDict(params[0])
Iustin Pop's avatar
Iustin Pop committed
325
326
    return backend.ShutdownInstance(instance)

327
328
329
330
331
  @staticmethod
  def perspective_instance_start(params):
    """Start an instance.

    """
332
    instance = objects.Instance.FromDict(params[0])
Iustin Pop's avatar
Iustin Pop committed
333
334
335
    extra_args = params[1]
    return backend.StartInstance(instance, extra_args)

336
337
338
339
340
341
342
343
  @staticmethod
  def perspective_instance_migrate(params):
    """Migrates an instance.

    """
    instance, target, live = params
    return backend.MigrateInstance(instance, target, live)

344
345
346
347
348
349
350
351
352
353
  @staticmethod
  def perspective_instance_reboot(params):
    """Reboot an instance.

    """
    instance = objects.Instance.FromDict(params[0])
    reboot_type = params[1]
    extra_args = params[2]
    return backend.RebootInstance(instance, reboot_type, extra_args)

354
355
356
357
358
  @staticmethod
  def perspective_instance_info(params):
    """Query instance information.

    """
Iustin Pop's avatar
Iustin Pop committed
359
360
    return backend.GetInstanceInfo(params[0])

361
362
363
364
365
  @staticmethod
  def perspective_all_instances_info(params):
    """Query information about all instances.

    """
Iustin Pop's avatar
Iustin Pop committed
366
367
    return backend.GetAllInstancesInfo()

368
369
370
371
372
  @staticmethod
  def perspective_instance_list(params):
    """Query the list of running instances.

    """
Iustin Pop's avatar
Iustin Pop committed
373
374
375
376
    return backend.GetInstanceList()

  # node --------------------------

377
378
379
380
381
  @staticmethod
  def perspective_node_tcp_ping(params):
    """Do a TcpPing on the remote node.

    """
382
383
    return utils.TcpPing(params[1], params[2], timeout=params[3],
                         live_port_needed=params[4], source=params[0])
384

385
386
387
388
389
  @staticmethod
  def perspective_node_info(params):
    """Query node information.

    """
Iustin Pop's avatar
Iustin Pop committed
390
391
392
    vgname = params[0]
    return backend.GetNodeInfo(vgname)

393
394
395
396
397
  @staticmethod
  def perspective_node_add(params):
    """Complete the registration of this node in the cluster.

    """
Iustin Pop's avatar
Iustin Pop committed
398
399
400
    return backend.AddNode(params[0], params[1], params[2],
                           params[3], params[4], params[5])

401
402
403
404
405
  @staticmethod
  def perspective_node_verify(params):
    """Run a verify sequence on this node.

    """
Iustin Pop's avatar
Iustin Pop committed
406
407
    return backend.VerifyNode(params[0])

408
409
410
411
412
  @staticmethod
  def perspective_node_start_master(params):
    """Promote this node to master status.

    """
Iustin Pop's avatar
Iustin Pop committed
413
414
    return backend.StartMaster()

415
416
417
418
419
  @staticmethod
  def perspective_node_stop_master(params):
    """Demote this node from master status.

    """
Iustin Pop's avatar
Iustin Pop committed
420
421
    return backend.StopMaster()

422
423
424
425
426
  @staticmethod
  def perspective_node_leave_cluster(params):
    """Cleanup after leaving a cluster.

    """
Iustin Pop's avatar
Iustin Pop committed
427
428
    return backend.LeaveCluster()

429
430
431
432
433
  @staticmethod
  def perspective_node_volumes(params):
    """Query the list of all logical volume groups.

    """
434
435
    return backend.NodeVolumes()

Iustin Pop's avatar
Iustin Pop committed
436
437
  # cluster --------------------------

438
439
440
441
442
  @staticmethod
  def perspective_version(params):
    """Query version information.

    """
Iustin Pop's avatar
Iustin Pop committed
443
444
    return constants.PROTOCOL_VERSION

445
446
447
448
449
450
451
452
  @staticmethod
  def perspective_upload_file(params):
    """Upload a file.

    Note that the backend implementation imposes strict rules on which
    files are accepted.

    """
Iustin Pop's avatar
Iustin Pop committed
453
454
455
456
457
    return backend.UploadFile(*params)


  # os -----------------------

458
459
460
461
462
  @staticmethod
  def perspective_os_diagnose(params):
    """Query detailed information about existing OSes.

    """
463
    return [os.ToDict() for os in backend.DiagnoseOS()]
Iustin Pop's avatar
Iustin Pop committed
464

465
466
467
468
469
  @staticmethod
  def perspective_os_get(params):
    """Query information about a given OS.

    """
Iustin Pop's avatar
Iustin Pop committed
470
471
    name = params[0]
    try:
472
      os_obj = backend.OSFromDisk(name)
Iustin Pop's avatar
Iustin Pop committed
473
    except errors.InvalidOS, err:
474
475
      os_obj = objects.OS.FromInvalidOS(err)
    return os_obj.ToDict()
Iustin Pop's avatar
Iustin Pop committed
476
477
478

  # hooks -----------------------

479
480
481
482
483
  @staticmethod
  def perspective_hooks_runner(params):
    """Run hook scripts.

    """
Iustin Pop's avatar
Iustin Pop committed
484
485
486
487
    hpath, phase, env = params
    hr = backend.HooksRunner()
    return hr.RunHooks(hpath, phase, env)

488
489
490
491
492
493
494
495
496
497
498
  # iallocator -----------------

  @staticmethod
  def perspective_iallocator_runner(params):
    """Run an iallocator script.

    """
    name, idata = params
    iar = backend.IAllocatorRunner()
    return iar.Run(name, idata)

499
500
501
502
503
504
505
506
507
508
  # test -----------------------

  @staticmethod
  def perspective_test_delay(params):
    """Run test delay.

    """
    duration = params[0]
    return utils.TestDelay(duration)

509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
  @staticmethod
  def perspective_file_storage_dir_create(params):
    """Create the file storage directory.

    """
    file_storage_dir = params[0]
    return backend.CreateFileStorageDir(file_storage_dir)

  @staticmethod
  def perspective_file_storage_dir_remove(params):
    """Remove the file storage directory.

    """
    file_storage_dir = params[0]
    return backend.RemoveFileStorageDir(file_storage_dir)

  @staticmethod
  def perspective_file_storage_dir_rename(params):
    """Rename the file storage directory.

    """
    old_file_storage_dir = params[0]
    new_file_storage_dir = params[1]
    return backend.RenameFileStorageDir(old_file_storage_dir,
                                        new_file_storage_dir)

Iustin Pop's avatar
Iustin Pop committed
535

536
537
538
539
540
class NodeDaemonHttpServer(http.HTTPServer):
  def __init__(self, server_address):
    http.HTTPServer.__init__(self, server_address, NodeDaemonRequestHandler)


Guido Trotter's avatar
Guido Trotter committed
541
542
543
544
545
546
547
548
549
class ForkingHTTPServer(SocketServer.ForkingMixIn, NodeDaemonHttpServer):
  """Forking HTTP Server.

  This inherits from ForkingMixIn and HTTPServer in order to fork for each
  request we handle. This allows more requests to be handled concurrently.

  """


Iustin Pop's avatar
Iustin Pop committed
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
def ParseOptions():
  """Parse the command line options.

  Returns:
    (options, args) as from OptionParser.parse_args()

  """
  parser = OptionParser(description="Ganeti node daemon",
                        usage="%prog [-f] [-d]",
                        version="%%prog (ganeti) %s" %
                        constants.RELEASE_VERSION)

  parser.add_option("-f", "--foreground", dest="fork",
                    help="Don't detach from the current terminal",
                    default=True, action="store_false")
  parser.add_option("-d", "--debug", dest="debug",
                    help="Enable some debug messages",
                    default=False, action="store_true")
  options, args = parser.parse_args()
  return options, args


def main():
573
574
575
  """Main function for the node daemon.

  """
Iustin Pop's avatar
Iustin Pop committed
576
  options, args = ParseOptions()
577
  utils.debug = options.debug
Iustin Pop's avatar
Iustin Pop committed
578
579
580
581
582
583
584
585
586
587
588
589
590
  for fname in (constants.SSL_CERT_FILE,):
    if not os.path.isfile(fname):
      print "config %s not there, will not run." % fname
      sys.exit(5)

  try:
    ss = ssconf.SimpleStore()
    port = ss.GetNodeDaemonPort()
    pwdata = ss.GetNodeDaemonPassword()
  except errors.ConfigurationError, err:
    print "Cluster configuration incomplete: '%s'" % str(err)
    sys.exit(5)

591
592
593
594
595
596
597
598
599
600
601
602
603
604
  # create the various SUB_RUN_DIRS, if not existing, so that we handle the
  # situation where RUN_DIR is tmpfs
  for dir_name in constants.SUB_RUN_DIRS:
    if not os.path.exists(dir_name):
      try:
        os.mkdir(dir_name, 0755)
      except EnvironmentError, err:
        if err.errno != errno.EEXIST:
          print ("Node setup wrong, cannot create directory %s: %s" %
                 (dir_name, err))
          sys.exit(5)
    if not os.path.isdir(dir_name):
      print ("Node setup wrong, %s is not a directory" % dir_name)
      sys.exit(5)
605

Iustin Pop's avatar
Iustin Pop committed
606
607
  # become a daemon
  if options.fork:
608
    utils.Daemonize(logfile=constants.LOG_NODESERVER)
Iustin Pop's avatar
Iustin Pop committed
609

Iustin Pop's avatar
Iustin Pop committed
610
611
  logger.SetupDaemon(logfile=constants.LOG_NODESERVER, debug=options.debug,
                     stderr_logging=not options.fork)
Iustin Pop's avatar
Iustin Pop committed
612
  logging.info("ganeti node daemon startup")
Iustin Pop's avatar
Iustin Pop committed
613

614
615
  global _EXIT_GANETI_NODED

Guido Trotter's avatar
Guido Trotter committed
616
617
618
619
620
621
  if options.fork:
    httpd = ForkingHTTPServer(('', port))
  else:
    httpd = NodeDaemonHttpServer(('', port))

  # FIXME: updating _EXIT_GANETI_NODED doesn't work when forking
622
623
  while (not _EXIT_GANETI_NODED):
    httpd.handle_request()
Iustin Pop's avatar
Iustin Pop committed
624
625


626
if __name__ == '__main__':
Iustin Pop's avatar
Iustin Pop committed
627
  main()