ganeti-noded 21.1 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
33
import signal
Iustin Pop's avatar
Iustin Pop committed
34
35
36
37
38
39
40

from optparse import OptionParser

from ganeti import backend
from ganeti import constants
from ganeti import objects
from ganeti import errors
41
from ganeti import jstore
42
from ganeti import daemon
43
from ganeti import http
44
from ganeti import utils
Iustin Pop's avatar
Iustin Pop committed
45

46
47
import ganeti.http.server

Iustin Pop's avatar
Iustin Pop committed
48

49
50
51
queue_lock = None


52
53
54
55
def _RequireJobQueueLock(fn):
  """Decorator for job queue manipulating functions.

  """
56
57
  QUEUE_LOCK_TIMEOUT = 10

58
59
  def wrapper(*args, **kwargs):
    # Locking in exclusive, blocking mode because there could be several
60
    # children running at the same time. Waiting up to 10 seconds.
61
    queue_lock.Exclusive(blocking=True, timeout=QUEUE_LOCK_TIMEOUT)
62
63
64
65
    try:
      return fn(*args, **kwargs)
    finally:
      queue_lock.Unlock()
66

67
68
69
  return wrapper


70
class NodeHttpServer(http.server.HttpServer):
71
72
73
74
75
  """The server implementation.

  This class holds all methods exposed over the RPC interface.

  """
76
  def __init__(self, *args, **kwargs):
77
    http.server.HttpServer.__init__(self, *args, **kwargs)
78
79
80
    self.noded_pid = os.getpid()

  def HandleRequest(self, req):
81
    """Handle a request.
Iustin Pop's avatar
Iustin Pop committed
82

83
    """
84
    if req.request_method.upper() != http.HTTP_PUT:
85
      raise http.HttpBadRequest()
86

87
    path = req.request_path
Iustin Pop's avatar
Iustin Pop committed
88
89
90
    if path.startswith("/"):
      path = path[1:]

91
92
    method = getattr(self, "perspective_%s" % path, None)
    if method is None:
93
      raise http.HttpNotFound()
Iustin Pop's avatar
Iustin Pop committed
94

Iustin Pop's avatar
Iustin Pop committed
95
    try:
96
      try:
97
        return method(req.request_body)
98
99
100
      except:
        logging.exception("Error in RPC call")
        raise
101
    except errors.QuitGanetiException, err:
102
      # Tell parent to quit
103
      os.kill(self.noded_pid, signal.SIGTERM)
Iustin Pop's avatar
Iustin Pop committed
104
105
106

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

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

    """
112
    bdev_s, size, owner, on_primary, info = 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.BlockdevCreate(bdev, size, owner, on_primary, info)
Iustin Pop's avatar
Iustin Pop committed
117

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

    """
Iustin Pop's avatar
Iustin Pop committed
123
    bdev_s = params[0]
124
    bdev = objects.Disk.FromDict(bdev_s)
125
    return backend.BlockdevRemove(bdev)
Iustin Pop's avatar
Iustin Pop committed
126

Iustin Pop's avatar
Iustin Pop committed
127
128
129
130
131
132
  @staticmethod
  def perspective_blockdev_rename(params):
    """Remove a block device.

    """
    devlist = [(objects.Disk.FromDict(ds), uid) for ds, uid in params]
133
    return backend.BlockdevRename(devlist)
Iustin Pop's avatar
Iustin Pop committed
134

135
136
137
138
139
  @staticmethod
  def perspective_blockdev_assemble(params):
    """Assemble a block device.

    """
140
    bdev_s, owner, on_primary = params
141
    bdev = objects.Disk.FromDict(bdev_s)
Iustin Pop's avatar
Iustin Pop committed
142
143
    if bdev is None:
      raise ValueError("can't unserialize data!")
144
    return backend.BlockdevAssemble(bdev, owner, on_primary)
Iustin Pop's avatar
Iustin Pop committed
145

146
147
148
149
150
  @staticmethod
  def perspective_blockdev_shutdown(params):
    """Shutdown a block device.

    """
Iustin Pop's avatar
Iustin Pop committed
151
    bdev_s = params[0]
152
    bdev = objects.Disk.FromDict(bdev_s)
Iustin Pop's avatar
Iustin Pop committed
153
154
    if bdev is None:
      raise ValueError("can't unserialize data!")
155
    return backend.BlockdevShutdown(bdev)
Iustin Pop's avatar
Iustin Pop committed
156

157
  @staticmethod
158
  def perspective_blockdev_addchildren(params):
159
160
161
162
163
164
    """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
165
    bdev_s, ndev_s = params
166
    bdev = objects.Disk.FromDict(bdev_s)
167
168
    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
169
      raise ValueError("can't unserialize data!")
170
    return backend.BlockdevAddchildren(bdev, ndevs)
Iustin Pop's avatar
Iustin Pop committed
171

172
  @staticmethod
173
  def perspective_blockdev_removechildren(params):
174
175
176
177
178
179
    """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
180
    bdev_s, ndev_s = params
181
    bdev = objects.Disk.FromDict(bdev_s)
182
183
    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
184
      raise ValueError("can't unserialize data!")
185
    return backend.BlockdevRemovechildren(bdev, ndevs)
Iustin Pop's avatar
Iustin Pop committed
186

187
188
189
190
191
  @staticmethod
  def perspective_blockdev_getmirrorstatus(params):
    """Return the mirror status for a list of disks.

    """
192
    disks = [objects.Disk.FromDict(dsk_s)
Iustin Pop's avatar
Iustin Pop committed
193
            for dsk_s in params]
194
    return backend.BlockdevGetmirrorstatus(disks)
Iustin Pop's avatar
Iustin Pop committed
195

196
197
198
199
200
201
202
  @staticmethod
  def perspective_blockdev_find(params):
    """Expose the FindBlockDevice functionality for a disk.

    This will try to find but not activate a disk.

    """
203
    disk = objects.Disk.FromDict(params[0])
204
    return backend.BlockdevFind(disk)
Iustin Pop's avatar
Iustin Pop committed
205

206
207
208
209
210
211
212
213
214
  @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.

    """
215
    cfbd = objects.Disk.FromDict(params[0])
216
    return backend.BlockdevSnapshot(cfbd)
Iustin Pop's avatar
Iustin Pop committed
217

218
219
220
221
222
223
224
  @staticmethod
  def perspective_blockdev_grow(params):
    """Grow a stack of devices.

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

227
228
229
230
231
  @staticmethod
  def perspective_blockdev_close(params):
    """Closes the given block devices.

    """
232
    disks = [objects.Disk.FromDict(cf) for cf in params[1]]
233
    return backend.BlockdevClose(params[0], disks)
234

Iustin Pop's avatar
Iustin Pop committed
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
  # blockdev/drbd specific methods ----------

  @staticmethod
  def perspective_drbd_disconnect_net(params):
    """Disconnects the network connection of drbd disks.

    Note that this is only valid for drbd disks, so the members of the
    disk list must all be drbd devices.

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

  @staticmethod
  def perspective_drbd_attach_net(params):
    """Attaches the network connection of drbd disks.

    Note that this is only valid for drbd disks, so the members of the
    disk list must all be drbd devices.

    """
    nodes_ip, disks, instance_name, multimaster = params
    disks = [objects.Disk.FromDict(cf) for cf in disks]
259
260
    return backend.DrbdAttachNet(nodes_ip, disks,
                                     instance_name, multimaster)
Iustin Pop's avatar
Iustin Pop committed
261
262
263
264
265
266
267
268
269
270
271
272
273

  @staticmethod
  def perspective_drbd_wait_sync(params):
    """Wait until DRBD disks are synched.

    Note that this is only valid for drbd disks, so the members of the
    disk list must all be drbd devices.

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

Iustin Pop's avatar
Iustin Pop committed
274
275
  # export/import  --------------------------

276
277
278
279
280
  @staticmethod
  def perspective_snapshot_export(params):
    """Export a given snapshot.

    """
281
    disk = objects.Disk.FromDict(params[0])
Iustin Pop's avatar
Iustin Pop committed
282
    dest_node = params[1]
283
    instance = objects.Instance.FromDict(params[2])
284
    cluster_name = params[3]
285
286
287
    dev_idx = params[4]
    return backend.ExportSnapshot(disk, dest_node, instance,
                                  cluster_name, dev_idx)
288
289
290
291

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

293
    """
294
295
    instance = objects.Instance.FromDict(params[0])
    snap_disks = [objects.Disk.FromDict(str_data)
Iustin Pop's avatar
Iustin Pop committed
296
297
298
                  for str_data in params[1]]
    return backend.FinalizeExport(instance, snap_disks)

299
300
301
302
303
304
305
306
307
308
  @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
309
310
311
312
    if einfo is None:
      return einfo
    return einfo.Dumps()

313
314
315
316
317
318
319
320
321
  @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
322
323
    return backend.ListExports()

324
325
326
327
328
  @staticmethod
  def perspective_export_remove(params):
    """Remove an export.

    """
Iustin Pop's avatar
Iustin Pop committed
329
330
331
332
333
    export = params[0]
    return backend.RemoveExport(export)

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

334
335
336
337
338
  @staticmethod
  def perspective_volume_list(params):
    """Query the list of logical volumes in a given volume group.

    """
Iustin Pop's avatar
Iustin Pop committed
339
340
341
    vgname = params[0]
    return backend.GetVolumeList(vgname)

342
343
344
345
346
  @staticmethod
  def perspective_vg_list(params):
    """Query the list of volume groups.

    """
Iustin Pop's avatar
Iustin Pop committed
347
348
349
350
    return backend.ListVolumeGroups()

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

351
352
353
354
355
  @staticmethod
  def perspective_bridges_exist(params):
    """Check if all bridges given exist on this node.

    """
Iustin Pop's avatar
Iustin Pop committed
356
357
358
359
360
    bridges_list = params[0]
    return backend.BridgesExist(bridges_list)

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

361
362
363
364
365
  @staticmethod
  def perspective_instance_os_add(params):
    """Install an OS on a given instance.

    """
366
    inst_s = params[0]
367
    inst = objects.Instance.FromDict(inst_s)
368
    return backend.InstanceOsAdd(inst)
Iustin Pop's avatar
Iustin Pop committed
369

370
371
372
373
374
  @staticmethod
  def perspective_instance_run_rename(params):
    """Runs the OS rename script for an instance.

    """
375
    inst_s, old_name = params
376
    inst = objects.Instance.FromDict(inst_s)
377
    return backend.RunRenameInstance(inst, old_name)
378

379
380
381
382
383
  @staticmethod
  def perspective_instance_os_import(params):
    """Run the import function of an OS onto a given instance.

    """
384
    inst_s, src_node, src_images, cluster_name = params
385
    inst = objects.Instance.FromDict(inst_s)
386
387
    return backend.ImportOSIntoInstance(inst, src_node, src_images,
                                        cluster_name)
Iustin Pop's avatar
Iustin Pop committed
388

389
390
391
392
393
  @staticmethod
  def perspective_instance_shutdown(params):
    """Shutdown an instance.

    """
394
    instance = objects.Instance.FromDict(params[0])
395
    return backend.InstanceShutdown(instance)
Iustin Pop's avatar
Iustin Pop committed
396

397
398
399
400
401
  @staticmethod
  def perspective_instance_start(params):
    """Start an instance.

    """
402
    instance = objects.Instance.FromDict(params[0])
403
    return backend.StartInstance(instance)
Iustin Pop's avatar
Iustin Pop committed
404

405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
  @staticmethod
  def perspective_migration_info(params):
    """Gather information about an instance to be migrated.

    """
    instance = objects.Instance.FromDict(params[0])
    return backend.MigrationInfo(instance)

  @staticmethod
  def perspective_accept_instance(params):
    """Prepare the node to accept an instance.

    """
    instance, info, target = params
    instance = objects.Instance.FromDict(instance)
    return backend.AcceptInstance(instance, info, target)

  @staticmethod
  def perspective_finalize_migration(params):
    """Finalize the instance migration.

    """
    instance, info, success = params
    instance = objects.Instance.FromDict(instance)
    return backend.FinalizeMigration(instance, info, success)

431
432
433
434
435
436
  @staticmethod
  def perspective_instance_migrate(params):
    """Migrates an instance.

    """
    instance, target, live = params
437
    instance = objects.Instance.FromDict(instance)
438
439
    return backend.MigrateInstance(instance, target, live)

440
441
442
443
444
445
446
  @staticmethod
  def perspective_instance_reboot(params):
    """Reboot an instance.

    """
    instance = objects.Instance.FromDict(params[0])
    reboot_type = params[1]
447
    return backend.InstanceReboot(instance, reboot_type)
448

449
450
451
452
453
  @staticmethod
  def perspective_instance_info(params):
    """Query instance information.

    """
Iustin Pop's avatar
Iustin Pop committed
454
    return backend.GetInstanceInfo(params[0], params[1])
Iustin Pop's avatar
Iustin Pop committed
455

456
457
458
459
460
461
462
463
  @staticmethod
  def perspective_instance_migratable(params):
    """Query whether the specified instance can be migrated.

    """
    instance = objects.Instance.FromDict(params[0])
    return backend.GetInstanceMigratable(instance)

464
465
466
467
468
  @staticmethod
  def perspective_all_instances_info(params):
    """Query information about all instances.

    """
469
    return backend.GetAllInstancesInfo(params[0])
Iustin Pop's avatar
Iustin Pop committed
470

471
472
473
474
475
  @staticmethod
  def perspective_instance_list(params):
    """Query the list of running instances.

    """
476
    return backend.GetInstanceList(params[0])
Iustin Pop's avatar
Iustin Pop committed
477
478
479

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

480
481
482
483
484
  @staticmethod
  def perspective_node_tcp_ping(params):
    """Do a TcpPing on the remote node.

    """
485
486
    return utils.TcpPing(params[1], params[2], timeout=params[3],
                         live_port_needed=params[4], source=params[0])
487

488
489
490
491
492
493
494
  @staticmethod
  def perspective_node_has_ip_address(params):
    """Checks if a node has the given ip address.

    """
    return utils.OwnIpAddress(params[0])

495
496
497
498
499
  @staticmethod
  def perspective_node_info(params):
    """Query node information.

    """
500
501
    vgname, hypervisor_type = params
    return backend.GetNodeInfo(vgname, hypervisor_type)
Iustin Pop's avatar
Iustin Pop committed
502

503
504
505
506
507
  @staticmethod
  def perspective_node_add(params):
    """Complete the registration of this node in the cluster.

    """
Iustin Pop's avatar
Iustin Pop committed
508
509
510
    return backend.AddNode(params[0], params[1], params[2],
                           params[3], params[4], params[5])

511
512
513
514
515
  @staticmethod
  def perspective_node_verify(params):
    """Run a verify sequence on this node.

    """
516
    return backend.VerifyNode(params[0], params[1])
Iustin Pop's avatar
Iustin Pop committed
517

518
519
520
521
522
  @staticmethod
  def perspective_node_start_master(params):
    """Promote this node to master status.

    """
523
    return backend.StartMaster(params[0])
Iustin Pop's avatar
Iustin Pop committed
524

525
526
527
528
529
  @staticmethod
  def perspective_node_stop_master(params):
    """Demote this node from master status.

    """
530
    return backend.StopMaster(params[0])
Iustin Pop's avatar
Iustin Pop committed
531

532
533
534
535
536
  @staticmethod
  def perspective_node_leave_cluster(params):
    """Cleanup after leaving a cluster.

    """
Iustin Pop's avatar
Iustin Pop committed
537
538
    return backend.LeaveCluster()

539
540
541
542
543
  @staticmethod
  def perspective_node_volumes(params):
    """Query the list of all logical volume groups.

    """
544
545
    return backend.NodeVolumes()

546
547
548
549
550
551
552
553
  @staticmethod
  def perspective_node_demote_from_mc(params):
    """Demote a node from the master candidate role.

    """
    return backend.DemoteFromMC()


Iustin Pop's avatar
Iustin Pop committed
554
555
  # cluster --------------------------

556
557
558
559
560
  @staticmethod
  def perspective_version(params):
    """Query version information.

    """
Iustin Pop's avatar
Iustin Pop committed
561
562
    return constants.PROTOCOL_VERSION

563
564
565
566
567
568
569
570
  @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
571
572
    return backend.UploadFile(*params)

573
574
575
576
577
578
  @staticmethod
  def perspective_master_info(params):
    """Query master information.

    """
    return backend.GetMasterInfo()
Iustin Pop's avatar
Iustin Pop committed
579

580
581
582
583
584
  @staticmethod
  def perspective_write_ssconf_files(params):
    """Write ssconf files.

    """
585
586
    (values,) = params
    return backend.WriteSsconfFiles(values)
587

Iustin Pop's avatar
Iustin Pop committed
588
589
  # os -----------------------

590
591
592
593
594
  @staticmethod
  def perspective_os_diagnose(params):
    """Query detailed information about existing OSes.

    """
595
    return [os_obj.ToDict() for os_obj in backend.DiagnoseOS()]
Iustin Pop's avatar
Iustin Pop committed
596

597
598
599
600
601
  @staticmethod
  def perspective_os_get(params):
    """Query information about a given OS.

    """
Iustin Pop's avatar
Iustin Pop committed
602
603
    name = params[0]
    try:
604
      os_obj = backend.OSFromDisk(name)
Iustin Pop's avatar
Iustin Pop committed
605
    except errors.InvalidOS, err:
606
607
      os_obj = objects.OS.FromInvalidOS(err)
    return os_obj.ToDict()
Iustin Pop's avatar
Iustin Pop committed
608
609
610

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

611
612
613
614
615
  @staticmethod
  def perspective_hooks_runner(params):
    """Run hook scripts.

    """
Iustin Pop's avatar
Iustin Pop committed
616
617
618
619
    hpath, phase, env = params
    hr = backend.HooksRunner()
    return hr.RunHooks(hpath, phase, env)

620
621
622
623
624
625
626
627
628
629
630
  # iallocator -----------------

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

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

631
632
633
634
635
636
637
638
639
640
  # test -----------------------

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

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

641
642
  # file storage ---------------

643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
  @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)

669
670
  # jobs ------------------------

671
  @staticmethod
672
  @_RequireJobQueueLock
673
674
675
676
677
  def perspective_jobqueue_update(params):
    """Update job queue.

    """
    (file_name, content) = params
678
    return backend.JobQueueUpdate(file_name, content)
679
680

  @staticmethod
681
  @_RequireJobQueueLock
682
683
684
685
686
687
  def perspective_jobqueue_purge(params):
    """Purge job queue.

    """
    return backend.JobQueuePurge()

688
689
690
691
692
693
  @staticmethod
  @_RequireJobQueueLock
  def perspective_jobqueue_rename(params):
    """Rename a job queue file.

    """
694
695
    # TODO: What if a file fails to rename?
    return [backend.JobQueueRename(old, new) for old, new in params]
696

697
698
699
700
701
702
703
704
705
  @staticmethod
  def perspective_jobqueue_set_drain(params):
    """Set/unset the queue drain flag.

    """
    drain_flag = params[0]
    return backend.JobQueueSetDrainFlag(drain_flag)


706
707
708
709
710
711
712
713
714
715
  # hypervisor ---------------

  @staticmethod
  def perspective_hypervisor_validate_params(params):
    """Validate the hypervisor parameters.

    """
    (hvname, hvparams) = params
    return backend.ValidateHVParams(hvname, hvparams)

Iustin Pop's avatar
Iustin Pop committed
716
717
718
719

def ParseOptions():
  """Parse the command line options.

Iustin Pop's avatar
Iustin Pop committed
720
  @return: (options, args) as from OptionParser.parse_args()
Iustin Pop's avatar
Iustin Pop committed
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737

  """
  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


738
739
740
741
742
743
744
745
def EnsureRuntimeEnvironment():
  """Ensure our run-time environment is complete.

  Currently this creates directories which could be missing, either
  due to directories being on a tmpfs mount, or due to incomplete
  packaging.

  """
746
  dirs = [(val, constants.RUN_DIRS_MODE) for val in constants.SUB_RUN_DIRS]
747
  dirs.append((constants.LOG_OS_DIR, 0750))
748
  dirs.append((constants.LOCK_DIR, 1777))
749
750
751
752
753
754
755
756
757
758
759
760
761
762
  for dir_name, dir_mode in dirs:
    if not os.path.exists(dir_name):
      try:
        os.mkdir(dir_name, dir_mode)
      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)


Iustin Pop's avatar
Iustin Pop committed
763
def main():
764
765
766
  """Main function for the node daemon.

  """
767
768
  global queue_lock

Iustin Pop's avatar
Iustin Pop committed
769
  options, args = ParseOptions()
770
  utils.debug = options.debug
Iustin Pop's avatar
Iustin Pop committed
771
772
773
774

  if options.fork:
    utils.CloseFDs()

Iustin Pop's avatar
Iustin Pop committed
775
776
777
778
779
780
  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:
Michael Hanselmann's avatar
Michael Hanselmann committed
781
    port = utils.GetNodeDaemonPort()
Iustin Pop's avatar
Iustin Pop committed
782
783
784
785
  except errors.ConfigurationError, err:
    print "Cluster configuration incomplete: '%s'" % str(err)
    sys.exit(5)

786
  EnsureRuntimeEnvironment()
787

Iustin Pop's avatar
Iustin Pop committed
788
789
  # become a daemon
  if options.fork:
790
    utils.Daemonize(logfile=constants.LOG_NODESERVER)
Iustin Pop's avatar
Iustin Pop committed
791

792
  utils.WritePidFile(constants.NODED_PID)
793
  try:
Iustin Pop's avatar
Iustin Pop committed
794
795
    utils.SetupLogging(logfile=constants.LOG_NODESERVER, debug=options.debug,
                       stderr_logging=not options.fork)
796
    logging.info("ganeti node daemon startup")
797

798
799
800
801
    # Read SSL certificate
    ssl_params = http.HttpSslParams(ssl_key_path=constants.SSL_CERT_FILE,
                                    ssl_cert_path=constants.SSL_CERT_FILE)

802
803
    # Prepare job queue
    queue_lock = jstore.InitAndVerifyQueue(must_lock=False)
Iustin Pop's avatar
Iustin Pop committed
804

805
    mainloop = daemon.Mainloop()
806
807
    server = NodeHttpServer(mainloop, "", port,
                            ssl_params=ssl_params, ssl_verify_peer=True)
808
809
810
811
812
    server.Start()
    try:
      mainloop.Run()
    finally:
      server.Stop()
813
  finally:
814
    utils.RemovePidFile(constants.NODED_PID)
815

Iustin Pop's avatar
Iustin Pop committed
816

817
if __name__ == '__main__':
Iustin Pop's avatar
Iustin Pop committed
818
  main()