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])
Iustin Pop's avatar
Iustin Pop committed
395
396
    return backend.ShutdownInstance(instance)

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

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

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
431
  @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)

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

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

441
442
443
444
445
446
447
448
449
450
  @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)

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    """
Iustin Pop's avatar
Iustin Pop committed
539
540
    return backend.LeaveCluster()

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

    """
546
547
    return backend.NodeVolumes()

548
549
550
551
552
553
554
555
  @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
556
557
  # cluster --------------------------

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

    """
Iustin Pop's avatar
Iustin Pop committed
563
564
    return constants.PROTOCOL_VERSION

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

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

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

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

    """
587
588
    (values,) = params
    return backend.WriteSsconfFiles(values)
589

Iustin Pop's avatar
Iustin Pop committed
590
591
  # os -----------------------

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

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

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

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

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

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

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

622
623
624
625
626
627
628
629
630
631
632
  # iallocator -----------------

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

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

633
634
635
636
637
638
639
640
641
642
  # test -----------------------

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

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

643
644
  # file storage ---------------

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

671
672
  # jobs ------------------------

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

    """
    (file_name, content) = params
680
    return backend.JobQueueUpdate(file_name, content)
681
682

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

    """
    return backend.JobQueuePurge()

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

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

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

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


708
709
710
711
712
713
714
715
716
717
  # 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
718
719
720
721

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

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

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


740
741
742
743
744
745
746
747
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.

  """
748
  dirs = [(val, constants.RUN_DIRS_MODE) for val in constants.SUB_RUN_DIRS]
749
  dirs.append((constants.LOG_OS_DIR, 0750))
750
  dirs.append((constants.LOCK_DIR, 1777))
751
752
753
754
755
756
757
758
759
760
761
762
763
764
  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
765
def main():
766
767
768
  """Main function for the node daemon.

  """
769
770
  global queue_lock

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

  if options.fork:
    utils.CloseFDs()

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

788
  EnsureRuntimeEnvironment()
789

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

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

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

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

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

Iustin Pop's avatar
Iustin Pop committed
818

819
if __name__ == '__main__':
Iustin Pop's avatar
Iustin Pop committed
820
  main()