/mandos/trunk : revision 821

1

#!/usr/bin/python2.7

1

#!/usr/bin/python

2

# -*- mode: python; coding: utf-8 -*-

3

#

4

# Mandos server - give out binary blobs to connecting clients.

11

# "AvahiService" class, and some lines in "main".

12

#

13

# Everything else is

14

15

14

15

16

#

17

# This program is free software: you can redistribute it and/or modify

18

# it under the terms of the GNU General Public License as published by

44

import argparse

45

import datetime

46

import errno

47

import gnutls.crypto

48

import gnutls.connection

49

import gnutls.errors

50

import gnutls.library.functions

51

import gnutls.library.constants

52

import gnutls.library.types

53

47

try:

54

48

import ConfigParser as configparser

55

49

except ImportError:

83

77

import dbus

84

78

import dbus.service

85

79

try:

86

import gobject

80

from gi.repository import GObject

87

81

except ImportError:

88

from gi.repository import GObject as gobject

82

import gobject as GObject

89

83

import avahi

90

84

from dbus.mainloop.glib import DBusGMainLoop

91

85

import ctypes

104

98

if sys.version_info.major == 2:

105

99

str = unicode

106

100

107

version = "1.6.9"

101

version = "1.7.3"

108

102

stored_state_file = "clients.pickle"

109

103

110

104

logger = logging.getLogger()

157

151

158

152

def __init__(self):

159

153

self.tempdir = tempfile.mkdtemp(prefix="mandos-")

154

self.gpg = "gpg"

155

try:

156

output = subprocess.check_output(["gpgconf"])

157

for line in output.splitlines():

158

name, text, path = line.split(":")

159

if name == "gpg":

160

self.gpg = path

161

break

162

except OSError as e:

163

if e.errno != errno.ENOENT:

164

raise

160

165

self.gnupgargs = ['--batch',

161

'--home', self.tempdir,

166

'--homedir', self.tempdir,

162

167

'--force-mdc',

163

168

'--quiet',

164

169

'--no-use-agent']

203

208

dir=self.tempdir) as passfile:

204

209

passfile.write(passphrase)

205

210

passfile.flush()

206

proc = subprocess.Popen(['gpg', '--symmetric',

211

proc = subprocess.Popen([self.gpg, '--symmetric',

207

212

'--passphrase-file',

208

213

passfile.name]

209

214

+ self.gnupgargs,

221

226

dir = self.tempdir) as passfile:

222

227

passfile.write(passphrase)

223

228

passfile.flush()

224

proc = subprocess.Popen(['gpg', '--decrypt',

229

proc = subprocess.Popen([self.gpg, '--decrypt',

225

230

'--passphrase-file',

226

231

passfile.name]

227

232

+ self.gnupgargs,

435

440

.format(self.name)))

436

441

return ret

437

442

443

# Pretend that we have a GnuTLS module

444

class GnuTLS(object):

445

"""This isn't so much a class as it is a module-like namespace.

446

It is instantiated once, and simulates having a GnuTLS module."""

447

448

_library = ctypes.cdll.LoadLibrary(

449

ctypes.util.find_library("gnutls"))

450

_need_version = "3.3.0"

451

def __init__(self):

452

# Need to use class name "GnuTLS" here, since this method is

453

# called before the assignment to the "gnutls" global variable

454

# happens.

455

if GnuTLS.check_version(self._need_version) is None:

456

raise GnuTLS.Error("Needs GnuTLS {} or later"

457

.format(self._need_version))

458

459

# Unless otherwise indicated, the constants and types below are

460

# all from the gnutls/gnutls.h C header file.

461

462

# Constants

463

E_SUCCESS = 0

464

E_INTERRUPTED = -52

465

E_AGAIN = -28

466

CRT_OPENPGP = 2

467

CLIENT = 2

468

SHUT_RDWR = 0

469

CRD_CERTIFICATE = 1

470

E_NO_CERTIFICATE_FOUND = -49

471

OPENPGP_FMT_RAW = 0 # gnutls/openpgp.h

472

473

# Types

474

class session_int(ctypes.Structure):

475

_fields_ = []

476

session_t = ctypes.POINTER(session_int)

477

class certificate_credentials_st(ctypes.Structure):

478

_fields_ = []

479

certificate_credentials_t = ctypes.POINTER(

480

certificate_credentials_st)

481

certificate_type_t = ctypes.c_int

482

class datum_t(ctypes.Structure):

483

_fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),

484

('size', ctypes.c_uint)]

485

class openpgp_crt_int(ctypes.Structure):

486

_fields_ = []

487

openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)

488

openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h

489

log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)

490

credentials_type_t = ctypes.c_int #

491

transport_ptr_t = ctypes.c_void_p

492

close_request_t = ctypes.c_int

493

494

# Exceptions

495

class Error(Exception):

496

# We need to use the class name "GnuTLS" here, since this

497

# exception might be raised from within GnuTLS.__init__,

498

# which is called before the assignment to the "gnutls"

499

# global variable has happened.

500

def __init__(self, message = None, code = None, args=()):

501

# Default usage is by a message string, but if a return

502

# code is passed, convert it to a string with

503

# gnutls.strerror()

504

self.code = code

505

if message is None and code is not None:

506

message = GnuTLS.strerror(code)

507

return super(GnuTLS.Error, self).__init__(

508

message, *args)

509

510

class CertificateSecurityError(Error):

511

pass

512

513

# Classes

514

class Credentials(object):

515

def __init__(self):

516

self._c_object = gnutls.certificate_credentials_t()

517

gnutls.certificate_allocate_credentials(

518

ctypes.byref(self._c_object))

519

self.type = gnutls.CRD_CERTIFICATE

520

521

def __del__(self):

522

gnutls.certificate_free_credentials(self._c_object)

523

524

class ClientSession(object):

525

def __init__(self, socket, credentials = None):

526

self._c_object = gnutls.session_t()

527

gnutls.init(ctypes.byref(self._c_object), gnutls.CLIENT)

528

gnutls.set_default_priority(self._c_object)

529

gnutls.transport_set_ptr(self._c_object, socket.fileno())

530

gnutls.handshake_set_private_extensions(self._c_object,

531

True)

532

self.socket = socket

533

if credentials is None:

534

credentials = gnutls.Credentials()

535

gnutls.credentials_set(self._c_object, credentials.type,

536

ctypes.cast(credentials._c_object,

537

ctypes.c_void_p))

538

self.credentials = credentials

539

540

def __del__(self):

541

gnutls.deinit(self._c_object)

542

543

def handshake(self):

544

return gnutls.handshake(self._c_object)

545

546

def send(self, data):

547

data = bytes(data)

548

data_len = len(data)

549

while data_len > 0:

550

data_len -= gnutls.record_send(self._c_object,

551

data[-data_len:],

552

data_len)

553

554

def bye(self):

555

return gnutls.bye(self._c_object, gnutls.SHUT_RDWR)

556

557

# Error handling functions

558

def _error_code(result):

559

"""A function to raise exceptions on errors, suitable

560

for the 'restype' attribute on ctypes functions"""

561

if result >= 0:

562

return result

563

if result == gnutls.E_NO_CERTIFICATE_FOUND:

564

raise gnutls.CertificateSecurityError(code = result)

565

raise gnutls.Error(code = result)

566

567

def _retry_on_error(result, func, arguments):

568

"""A function to retry on some errors, suitable

569

for the 'errcheck' attribute on ctypes functions"""

570

while result < 0:

571

if result not in (gnutls.E_INTERRUPTED, gnutls.E_AGAIN):

572

return _error_code(result)

573

result = func(*arguments)

574

return result

575

576

# Unless otherwise indicated, the function declarations below are

577

# all from the gnutls/gnutls.h C header file.

578

579

# Functions

580

priority_set_direct = _library.gnutls_priority_set_direct

581

priority_set_direct.argtypes = [session_t, ctypes.c_char_p,

582

ctypes.POINTER(ctypes.c_char_p)]

583

priority_set_direct.restype = _error_code

584

585

init = _library.gnutls_init

586

init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]

587

init.restype = _error_code

588

589

set_default_priority = _library.gnutls_set_default_priority

590

set_default_priority.argtypes = [session_t]

591

set_default_priority.restype = _error_code

592

593

record_send = _library.gnutls_record_send

594

record_send.argtypes = [session_t, ctypes.c_void_p,

595

ctypes.c_size_t]

596

record_send.restype = ctypes.c_ssize_t

597

record_send.errcheck = _retry_on_error

598

599

certificate_allocate_credentials = (

600

_library.gnutls_certificate_allocate_credentials)

601

certificate_allocate_credentials.argtypes = [

602

ctypes.POINTER(certificate_credentials_t)]

603

certificate_allocate_credentials.restype = _error_code

604

605

certificate_free_credentials = (

606

_library.gnutls_certificate_free_credentials)

607

certificate_free_credentials.argtypes = [certificate_credentials_t]

608

certificate_free_credentials.restype = None

609

610

handshake_set_private_extensions = (

611

_library.gnutls_handshake_set_private_extensions)

612

handshake_set_private_extensions.argtypes = [session_t,

613

ctypes.c_int]

614

handshake_set_private_extensions.restype = None

615

616

credentials_set = _library.gnutls_credentials_set

617

credentials_set.argtypes = [session_t, credentials_type_t,

618

ctypes.c_void_p]

619

credentials_set.restype = _error_code

620

621

strerror = _library.gnutls_strerror

622

strerror.argtypes = [ctypes.c_int]

623

strerror.restype = ctypes.c_char_p

624

625

certificate_type_get = _library.gnutls_certificate_type_get

626

certificate_type_get.argtypes = [session_t]

627

certificate_type_get.restype = _error_code

628

629

certificate_get_peers = _library.gnutls_certificate_get_peers

630

certificate_get_peers.argtypes = [session_t,

631

ctypes.POINTER(ctypes.c_uint)]

632

certificate_get_peers.restype = ctypes.POINTER(datum_t)

633

634

global_set_log_level = _library.gnutls_global_set_log_level

635

global_set_log_level.argtypes = [ctypes.c_int]

636

global_set_log_level.restype = None

637

638

global_set_log_function = _library.gnutls_global_set_log_function

639

global_set_log_function.argtypes = [log_func]

640

global_set_log_function.restype = None

641

642

deinit = _library.gnutls_deinit

643

deinit.argtypes = [session_t]

644

deinit.restype = None

645

646

handshake = _library.gnutls_handshake

647

handshake.argtypes = [session_t]

648

handshake.restype = _error_code

649

handshake.errcheck = _retry_on_error

650

651

transport_set_ptr = _library.gnutls_transport_set_ptr

652

transport_set_ptr.argtypes = [session_t, transport_ptr_t]

653

transport_set_ptr.restype = None

654

655

bye = _library.gnutls_bye

656

bye.argtypes = [session_t, close_request_t]

657

bye.restype = _error_code

658

bye.errcheck = _retry_on_error

659

660

check_version = _library.gnutls_check_version

661

check_version.argtypes = [ctypes.c_char_p]

662

check_version.restype = ctypes.c_char_p

663

664

# All the function declarations below are from gnutls/openpgp.h

665

666

openpgp_crt_init = _library.gnutls_openpgp_crt_init

667

openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]

668

openpgp_crt_init.restype = _error_code

669

670

openpgp_crt_import = _library.gnutls_openpgp_crt_import

671

openpgp_crt_import.argtypes = [openpgp_crt_t,

672

ctypes.POINTER(datum_t),

673

openpgp_crt_fmt_t]

674

openpgp_crt_import.restype = _error_code

675

676

openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self

677

openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,

678

ctypes.POINTER(ctypes.c_uint)]

679

openpgp_crt_verify_self.restype = _error_code

680

681

openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit

682

openpgp_crt_deinit.argtypes = [openpgp_crt_t]

683

openpgp_crt_deinit.restype = None

684

685

openpgp_crt_get_fingerprint = (

686

_library.gnutls_openpgp_crt_get_fingerprint)

687

openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,

688

ctypes.c_void_p,

689

ctypes.POINTER(

690

ctypes.c_size_t)]

691

openpgp_crt_get_fingerprint.restype = _error_code

692

693

# Remove non-public functions

694

del _error_code, _retry_on_error

695

# Create the global "gnutls" object, simulating a module

696

gnutls = GnuTLS()

697

438

698

def call_pipe(connection, # : multiprocessing.Connection

439

699

func, *args, **kwargs):

440

700

"""This function is meant to be called by multiprocessing.Process

455

715

checker: subprocess.Popen(); a running checker process used

456

716

to see if the client lives.

457

717

'None' if no process is running.

458

checker_callback_tag: a gobject event source tag, or None

718

checker_callback_tag: a GObject event source tag, or None

459

719

checker_command: string; External command which is run to check

460

720

if client lives. %() expansions are done at

461

721

runtime with vars(self) as dict, so that for

462

722

instance %(name)s can be used in the command.

463

checker_initiator_tag: a gobject event source tag, or None

723

checker_initiator_tag: a GObject event source tag, or None

464

724

created: datetime.datetime(); (UTC) object creation

465

725

client_structure: Object describing what attributes a client has

466

726

and is used for storing the client at exit

467

727

current_checker_command: string; current running checker_command

468

disable_initiator_tag: a gobject event source tag, or None

728

disable_initiator_tag: a GObject event source tag, or None

469

729

enabled: bool()

470

730

fingerprint: string (40 or 32 hexadecimal digits); used to

471

731

uniquely identify the client

625

885

if not quiet:

626

886

logger.info("Disabling client %s", self.name)

627

887

if getattr(self, "disable_initiator_tag", None) is not None:

628

gobject.source_remove(self.disable_initiator_tag)

888

GObject.source_remove(self.disable_initiator_tag)

629

889

self.disable_initiator_tag = None

630

890

self.expires = None

631

891

if getattr(self, "checker_initiator_tag", None) is not None:

632

gobject.source_remove(self.checker_initiator_tag)

892

GObject.source_remove(self.checker_initiator_tag)

633

893

self.checker_initiator_tag = None

634

894

self.stop_checker()

635

895

self.enabled = False

636

896

if not quiet:

637

897

self.send_changedstate()

638

# Do not run this again if called by a gobject.timeout_add

898

# Do not run this again if called by a GObject.timeout_add

639

899

return False

640

900

641

901

def __del__(self):

645

905

# Schedule a new checker to be started an 'interval' from now,

646

906

# and every interval from then on.

647

907

if self.checker_initiator_tag is not None:

648

gobject.source_remove(self.checker_initiator_tag)

649

self.checker_initiator_tag = gobject.timeout_add(

908

GObject.source_remove(self.checker_initiator_tag)

909

self.checker_initiator_tag = GObject.timeout_add(

650

910

int(self.interval.total_seconds() * 1000),

651

911

self.start_checker)

652

912

# Schedule a disable() when 'timeout' has passed

653

913

if self.disable_initiator_tag is not None:

654

gobject.source_remove(self.disable_initiator_tag)

655

self.disable_initiator_tag = gobject.timeout_add(

914

GObject.source_remove(self.disable_initiator_tag)

915

self.disable_initiator_tag = GObject.timeout_add(

656

916

int(self.timeout.total_seconds() * 1000), self.disable)

657

917

# Also start a new checker *right now*.

658

918

self.start_checker()

694

954

if timeout is None:

695

955

timeout = self.timeout

696

956

if self.disable_initiator_tag is not None:

697

gobject.source_remove(self.disable_initiator_tag)

957

GObject.source_remove(self.disable_initiator_tag)

698

958

self.disable_initiator_tag = None

699

959

if getattr(self, "enabled", False):

700

self.disable_initiator_tag = gobject.timeout_add(

960

self.disable_initiator_tag = GObject.timeout_add(

701

961

int(timeout.total_seconds() * 1000), self.disable)

702

962

self.expires = datetime.datetime.utcnow() + timeout

703

963

758

1018

args = (pipe[1], subprocess.call, command),

759

1019

kwargs = popen_args)

760

1020

self.checker.start()

761

self.checker_callback_tag = gobject.io_add_watch(

762

pipe[0].fileno(), gobject.IO_IN,

1021

self.checker_callback_tag = GObject.io_add_watch(

1022

pipe[0].fileno(), GObject.IO_IN,

763

1023

self.checker_callback, pipe[0], command)

764

# Re-run this periodically if run by gobject.timeout_add

1024

# Re-run this periodically if run by GObject.timeout_add

765

1025

return True

766

1026

767

1027

def stop_checker(self):

768

1028

"""Force the checker process, if any, to stop."""

769

1029

if self.checker_callback_tag:

770

gobject.source_remove(self.checker_callback_tag)

1030

GObject.source_remove(self.checker_callback_tag)

771

1031

self.checker_callback_tag = None

772

1032

if getattr(self, "checker", None) is None:

773

1033

return

842

1102

access="r")

843

1103

def Property_dbus_property(self):

844

1104

return dbus.Boolean(False)

1105

1106

See also the DBusObjectWithAnnotations class.

845

1107

"""

846

1108

847

1109

def decorator(func):

869

1131

pass

870

1132

871

1133

872

class DBusObjectWithProperties(dbus.service.Object):

873

"""A D-Bus object with properties.

1134

class DBusObjectWithAnnotations(dbus.service.Object):

1135

"""A D-Bus object with annotations.

874

1136

875

Classes inheriting from this can use the dbus_service_property

876

decorator to expose methods as D-Bus properties. It exposes the

877

standard Get(), Set(), and GetAll() methods on the D-Bus.

1137

Classes inheriting from this can use the dbus_annotations

1138

decorator to add annotations to methods or signals.

878

1139

"""

879

1140

880

1141

@staticmethod

896

1157

for name, athing in

897

1158

inspect.getmembers(cls, self._is_dbus_thing(thing)))

898

1159

1160

@dbus.service.method(dbus.INTROSPECTABLE_IFACE,

1161

out_signature = "s",

1162

path_keyword = 'object_path',

1163

connection_keyword = 'connection')

1164

def Introspect(self, object_path, connection):

1165

"""Overloading of standard D-Bus method.

1166

1167

Inserts annotation tags on methods and signals.

1168

"""

1169

xmlstring = dbus.service.Object.Introspect(self, object_path,

1170

connection)

1171

try:

1172

document = xml.dom.minidom.parseString(xmlstring)

1173

1174

for if_tag in document.getElementsByTagName("interface"):

1175

# Add annotation tags

1176

for typ in ("method", "signal"):

1177

for tag in if_tag.getElementsByTagName(typ):

1178

annots = dict()

1179

for name, prop in (self.

1180

_get_all_dbus_things(typ)):

1181

if (name == tag.getAttribute("name")

1182

and prop._dbus_interface

1183

== if_tag.getAttribute("name")):

1184

annots.update(getattr(

1185

prop, "_dbus_annotations", {}))

1186

for name, value in annots.items():

1187

ann_tag = document.createElement(

1188

"annotation")

1189

ann_tag.setAttribute("name", name)

1190

ann_tag.setAttribute("value", value)

1191

tag.appendChild(ann_tag)

1192

# Add interface annotation tags

1193

for annotation, value in dict(

1194

itertools.chain.from_iterable(

1195

annotations().items()

1196

for name, annotations

1197

in self._get_all_dbus_things("interface")

1198

if name == if_tag.getAttribute("name")

1199

)).items():

1200

ann_tag = document.createElement("annotation")

1201

ann_tag.setAttribute("name", annotation)

1202

ann_tag.setAttribute("value", value)

1203

if_tag.appendChild(ann_tag)

1204

# Fix argument name for the Introspect method itself

1205

if (if_tag.getAttribute("name")

1206

== dbus.INTROSPECTABLE_IFACE):

1207

for cn in if_tag.getElementsByTagName("method"):

1208

if cn.getAttribute("name") == "Introspect":

1209

for arg in cn.getElementsByTagName("arg"):

1210

if (arg.getAttribute("direction")

1211

== "out"):

1212

arg.setAttribute("name",

1213

"xml_data")

1214

xmlstring = document.toxml("utf-8")

1215

document.unlink()

1216

except (AttributeError, xml.dom.DOMException,

1217

xml.parsers.expat.ExpatError) as error:

1218

logger.error("Failed to override Introspection method",

1219

exc_info=error)

1220

return xmlstring

1221

1222

1223

class DBusObjectWithProperties(DBusObjectWithAnnotations):

1224

"""A D-Bus object with properties.

1225

1226

Classes inheriting from this can use the dbus_service_property

1227

decorator to expose methods as D-Bus properties. It exposes the

1228

standard Get(), Set(), and GetAll() methods on the D-Bus.

1229

"""

1230

899

1231

def _get_dbus_property(self, interface_name, property_name):

900

1232

"""Returns a bound method if one exists which is a D-Bus

901

1233

property with the specified name and interface.

911

1243

raise DBusPropertyNotFound("{}:{}.{}".format(

912

1244

self.dbus_object_path, interface_name, property_name))

913

1245

1246

@classmethod

1247

def _get_all_interface_names(cls):

1248

"""Get a sequence of all interfaces supported by an object"""

1249

return (name for name in set(getattr(getattr(x, attr),

1250

"_dbus_interface", None)

1251

for x in (inspect.getmro(cls))

1252

for attr in dir(x))

1253

if name is not None)

1254

914

1255

@dbus.service.method(dbus.PROPERTIES_IFACE,

915

1256

in_signature="ss",

916

1257

out_signature="v")

986

1327

987

1328

Inserts property tags and interface annotation tags.

988

1329

"""

989

xmlstring = dbus.service.Object.Introspect(self, object_path,

990

connection)

1330

xmlstring = DBusObjectWithAnnotations.Introspect(self,

1331

object_path,

1332

connection)

991

1333

try:

992

1334

document = xml.dom.minidom.parseString(xmlstring)

993

1335

1006

1348

if prop._dbus_interface

1007

1349

== if_tag.getAttribute("name")):

1008

1350

if_tag.appendChild(tag)

1009

# Add annotation tags

1010

for typ in ("method", "signal", "property"):

1011

for tag in if_tag.getElementsByTagName(typ):

1012

annots = dict()

1013

for name, prop in (self.

1014

_get_all_dbus_things(typ)):

1015

if (name == tag.getAttribute("name")

1016

and prop._dbus_interface

1017

== if_tag.getAttribute("name")):

1018

annots.update(getattr(

1019

prop, "_dbus_annotations", {}))

1020

for name, value in annots.items():

1021

ann_tag = document.createElement(

1022

"annotation")

1023

ann_tag.setAttribute("name", name)

1024

ann_tag.setAttribute("value", value)

1025

tag.appendChild(ann_tag)

1026

# Add interface annotation tags

1027

for annotation, value in dict(

1028

itertools.chain.from_iterable(

1029

annotations().items()

1030

for name, annotations

1031

in self._get_all_dbus_things("interface")

1032

if name == if_tag.getAttribute("name")

1033

)).items():

1034

ann_tag = document.createElement("annotation")

1035

ann_tag.setAttribute("name", annotation)

1036

ann_tag.setAttribute("value", value)

1037

if_tag.appendChild(ann_tag)

1351

# Add annotation tags for properties

1352

for tag in if_tag.getElementsByTagName("property"):

1353

annots = dict()

1354

for name, prop in self._get_all_dbus_things(

1355

"property"):

1356

if (name == tag.getAttribute("name")

1357

and prop._dbus_interface

1358

== if_tag.getAttribute("name")):

1359

annots.update(getattr(

1360

prop, "_dbus_annotations", {}))

1361

for name, value in annots.items():

1362

ann_tag = document.createElement(

1363

"annotation")

1364

ann_tag.setAttribute("name", name)

1365

ann_tag.setAttribute("value", value)

1366

tag.appendChild(ann_tag)

1038

1367

# Add the names to the return values for the

1039

1368

# "org.freedesktop.DBus.Properties" methods

1040

1369

if (if_tag.getAttribute("name")

1058

1387

exc_info=error)

1059

1388

return xmlstring

1060

1389

1390

try:

1391

dbus.OBJECT_MANAGER_IFACE

1392

except AttributeError:

1393

dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"

1394

1395

class DBusObjectWithObjectManager(DBusObjectWithAnnotations):

1396

"""A D-Bus object with an ObjectManager.

1397

1398

Classes inheriting from this exposes the standard

1399

GetManagedObjects call and the InterfacesAdded and

1400

InterfacesRemoved signals on the standard

1401

"org.freedesktop.DBus.ObjectManager" interface.

1402

1403

Note: No signals are sent automatically; they must be sent

1404

manually.

1405

"""

1406

@dbus.service.method(dbus.OBJECT_MANAGER_IFACE,

1407

out_signature = "a{oa{sa{sv}}}")

1408

def GetManagedObjects(self):

1409

"""This function must be overridden"""

1410

raise NotImplementedError()

1411

1412

@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE,

1413

signature = "oa{sa{sv}}")

1414

def InterfacesAdded(self, object_path, interfaces_and_properties):

1415

pass

1416

1417

@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature = "oas")

1418

def InterfacesRemoved(self, object_path, interfaces):

1419

pass

1420

1421

@dbus.service.method(dbus.INTROSPECTABLE_IFACE,

1422

out_signature = "s",

1423

path_keyword = 'object_path',

1424

connection_keyword = 'connection')

1425

def Introspect(self, object_path, connection):

1426

"""Overloading of standard D-Bus method.

1427

1428

Override return argument name of GetManagedObjects to be

1429

"objpath_interfaces_and_properties"

1430

"""

1431

xmlstring = DBusObjectWithAnnotations.Introspect(self,

1432

object_path,

1433

connection)

1434

try:

1435

document = xml.dom.minidom.parseString(xmlstring)

1436

1437

for if_tag in document.getElementsByTagName("interface"):

1438

# Fix argument name for the GetManagedObjects method

1439

if (if_tag.getAttribute("name")

1440

== dbus.OBJECT_MANAGER_IFACE):

1441

for cn in if_tag.getElementsByTagName("method"):

1442

if (cn.getAttribute("name")

1443

== "GetManagedObjects"):

1444

for arg in cn.getElementsByTagName("arg"):

1445

if (arg.getAttribute("direction")

1446

== "out"):

1447

arg.setAttribute(

1448

"name",

1449

"objpath_interfaces"

1450

"_and_properties")

1451

xmlstring = document.toxml("utf-8")

1452

document.unlink()

1453

except (AttributeError, xml.dom.DOMException,

1454

xml.parsers.expat.ExpatError) as error:

1455

logger.error("Failed to override Introspection method",

1456

exc_info = error)

1457

return xmlstring

1061

1458

1062

1459

def datetime_to_dbus(dt, variant_level=0):

1063

1460

"""Convert a UTC datetime.datetime() to a D-Bus type."""

1154

1551

func1 and func2 to the "call_both" function

1155

1552

outside of its arguments"""

1156

1553

1554

@functools.wraps(func2)

1157

1555

def call_both(*args, **kwargs):

1158

1556

"""This function will emit two D-Bus

1159

1557

signals by calling func1 and func2"""

1160

1558

func1(*args, **kwargs)

1161

1559

func2(*args, **kwargs)

1560

# Make wrapper function look like a D-Bus signal

1561

for name, attr in inspect.getmembers(func2):

1562

if name.startswith("_dbus_"):

1563

setattr(call_both, name, attr)

1162

1564

1163

1565

return call_both

1164

1566

# Create the "call_both" function and add it to

1376

1778

if exitstatus >= 0:

1377

1779

# Emit D-Bus signal

1378

1780

self.CheckerCompleted(dbus.Int16(exitstatus),

1379

dbus.Int64(0),

1781

# This is specific to GNU libC

1782

dbus.Int64(exitstatus << 8),

1380

1783

dbus.String(command))

1381

1784

else:

1382

1785

# Emit D-Bus signal

1383

1786

self.CheckerCompleted(dbus.Int16(-1),

1384

1787

dbus.Int64(

1385

self.last_checker_signal),

1788

# This is specific to GNU libC

1789

(exitstatus << 8)

1790

| self.last_checker_signal),

1386

1791

dbus.String(command))

1387

1792

return ret

1388

1793

1402

1807

1403

1808

def approve(self, value=True):

1404

1809

self.approved = value

1405

gobject.timeout_add(int(self.approval_duration.total_seconds()

1810

GObject.timeout_add(int(self.approval_duration.total_seconds()

1406

1811

* 1000), self._reset_approved)

1407

1812

self.send_changedstate()

1408

1813

1465

1870

self.checked_ok()

1466

1871

1467

1872

# Enable - method

1873

@dbus_annotations({"org.freedesktop.DBus.Deprecated": "true"})

1468

1874

@dbus.service.method(_interface)

1469

1875

def Enable(self):

1470

1876

"D-Bus method"

1471

1877

self.enable()

1472

1878

1473

1879

# StartChecker - method

1880

@dbus_annotations({"org.freedesktop.DBus.Deprecated": "true"})

1474

1881

@dbus.service.method(_interface)

1475

1882

def StartChecker(self):

1476

1883

"D-Bus method"

1477

1884

self.start_checker()

1478

1885

1479

1886

# Disable - method

1887

@dbus_annotations({"org.freedesktop.DBus.Deprecated": "true"})

1480

1888

@dbus.service.method(_interface)

1481

1889

def Disable(self):

1482

1890

"D-Bus method"

1483

1891

self.disable()

1484

1892

1485

1893

# StopChecker - method

1894

@dbus_annotations({"org.freedesktop.DBus.Deprecated": "true"})

1486

1895

@dbus.service.method(_interface)

1487

1896

def StopChecker(self):

1488

1897

self.stop_checker()

1524

1933

self.approval_duration = datetime.timedelta(0, 0, 0, value)

1525

1934

1526

1935

# Name - property

1936

@dbus_annotations(

1937

{"org.freedesktop.DBus.Property.EmitsChangedSignal": "const"})

1527

1938

@dbus_service_property(_interface, signature="s", access="read")

1528

1939

def Name_dbus_property(self):

1529

1940

return dbus.String(self.name)

1530

1941

1531

1942

# Fingerprint - property

1943

@dbus_annotations(

1944

{"org.freedesktop.DBus.Property.EmitsChangedSignal": "const"})

1532

1945

@dbus_service_property(_interface, signature="s", access="read")

1533

1946

def Fingerprint_dbus_property(self):

1534

1947

return dbus.String(self.fingerprint)

1543

1956

self.host = str(value)

1544

1957

1545

1958

# Created - property

1959

@dbus_annotations(

1960

{"org.freedesktop.DBus.Property.EmitsChangedSignal": "const"})

1546

1961

@dbus_service_property(_interface, signature="s", access="read")

1547

1962

def Created_dbus_property(self):

1548

1963

return datetime_to_dbus(self.created)

1609

2024

if (getattr(self, "disable_initiator_tag", None)

1610

2025

is None):

1611

2026

return

1612

gobject.source_remove(self.disable_initiator_tag)

1613

self.disable_initiator_tag = gobject.timeout_add(

2027

GObject.source_remove(self.disable_initiator_tag)

2028

self.disable_initiator_tag = GObject.timeout_add(

1614

2029

int((self.expires - now).total_seconds() * 1000),

1615

2030

self.disable)

1616

2031

1636

2051

return

1637

2052

if self.enabled:

1638

2053

# Reschedule checker run

1639

gobject.source_remove(self.checker_initiator_tag)

1640

self.checker_initiator_tag = gobject.timeout_add(

2054

GObject.source_remove(self.checker_initiator_tag)

2055

self.checker_initiator_tag = GObject.timeout_add(

1641

2056

value, self.start_checker)

1642

2057

self.start_checker() # Start one now, too

1643

2058

1663

2078

self.stop_checker()

1664

2079

1665

2080

# ObjectPath - property

2081

@dbus_annotations(

2082

{"org.freedesktop.DBus.Property.EmitsChangedSignal": "const",

2083

"org.freedesktop.DBus.Deprecated": "true"})

1666

2084

@dbus_service_property(_interface, signature="o", access="read")

1667

2085

def ObjectPath_dbus_property(self):

1668

2086

return self.dbus_object_path # is already a dbus.ObjectPath

1669

2087

1670

2088

# Secret = property

2089

@dbus_annotations(

2090

{"org.freedesktop.DBus.Property.EmitsChangedSignal":

2091

"invalidates"})

1671

2092

@dbus_service_property(_interface,

1672

2093

signature="ay",

1673

2094

access="write",

1719

2140

logger.debug("Pipe FD: %d",

1720

2141

self.server.child_pipe.fileno())

1721

2142

1722

session = gnutls.connection.ClientSession(

1723

self.request, gnutls.connection .X509Credentials())

1724

1725

# Note: gnutls.connection.X509Credentials is really a

1726

# generic GnuTLS certificate credentials object so long as

1727

# no X.509 keys are added to it. Therefore, we can use it

1728

# here despite using OpenPGP certificates.

2143

session = gnutls.ClientSession(self.request)

1729

2144

1730

2145

#priority = ':'.join(("NONE", "+VERS-TLS1.1",

1731

2146

# "+AES-256-CBC", "+SHA1",

1735

2150

priority = self.server.gnutls_priority

1736

2151

if priority is None:

1737

2152

priority = "NORMAL"

1738

gnutls.library.functions.gnutls_priority_set_direct(

1739

session._c_object, priority, None)

2153

gnutls.priority_set_direct(session._c_object, priority,

2154

None)

1740

2155

1741

2156

# Start communication using the Mandos protocol

1742

2157

# Get protocol number

1752

2167

# Start GnuTLS connection

1753

2168

try:

1754

2169

session.handshake()

1755

except gnutls.errors.GNUTLSError as error:

2170

except gnutls.Error as error:

1756

2171

logger.warning("Handshake failed: %s", error)

1757

2172

# Do not run session.bye() here: the session is not

1758

2173

# established. Just abandon the request.

1764

2179

try:

1765

2180

fpr = self.fingerprint(

1766

2181

self.peer_certificate(session))

1767

except (TypeError,

1768

gnutls.errors.GNUTLSError) as error:

2182

except (TypeError, gnutls.Error) as error:

1769

2183

logger.warning("Bad certificate: %s", error)

1770

2184

return

1771

2185

logger.debug("Fingerprint: %s", fpr)

1829

2243

else:

1830

2244

delay -= time2 - time

1831

2245

1832

sent_size = 0

1833

while sent_size < len(client.secret):

1834

try:

1835

sent = session.send(client.secret[sent_size:])

1836

except gnutls.errors.GNUTLSError as error:

1837

logger.warning("gnutls send failed",

1838

exc_info=error)

1839

return

1840

logger.debug("Sent: %d, remaining: %d", sent,

1841

len(client.secret) - (sent_size

1842

+ sent))

1843

sent_size += sent

2246

try:

2247

session.send(client.secret)

2248

except gnutls.Error as error:

2249

logger.warning("gnutls send failed",

2250

exc_info = error)

2251

return

1844

2252

1845

2253

logger.info("Sending secret to %s", client.name)

1846

2254

# bump the timeout using extended_timeout

1854

2262

client.approvals_pending -= 1

1855

2263

try:

1856

2264

session.bye()

1857

except gnutls.errors.GNUTLSError as error:

2265

except gnutls.Error as error:

1858

2266

logger.warning("GnuTLS bye failed",

1859

2267

exc_info=error)

1860

2268

1862

2270

def peer_certificate(session):

1863

2271

"Return the peer's OpenPGP certificate as a bytestring"

1864

2272

# If not an OpenPGP certificate...

1865

if (gnutls.library.functions.gnutls_certificate_type_get(

1866

session._c_object)

1867

!= gnutls.library.constants.GNUTLS_CRT_OPENPGP):

1868

# ...do the normal thing

1869

return session.peer_certificate

2273

if (gnutls.certificate_type_get(session._c_object)

2274

!= gnutls.CRT_OPENPGP):

2275

# ...return invalid data

2276

return b""

1870

2277

list_size = ctypes.c_uint(1)

1871

cert_list = (gnutls.library.functions

1872

.gnutls_certificate_get_peers

2278

cert_list = (gnutls.certificate_get_peers

1873

2279

(session._c_object, ctypes.byref(list_size)))

1874

2280

if not bool(cert_list) and list_size.value != 0:

1875

raise gnutls.errors.GNUTLSError("error getting peer"

1876

" certificate")

2281

raise gnutls.Error("error getting peer certificate")

1877

2282

if list_size.value == 0:

1878

2283

return None

1879

2284

cert = cert_list[0]

1883

2288

def fingerprint(openpgp):

1884

2289

"Convert an OpenPGP bytestring to a hexdigit fingerprint"

1885

2290

# New GnuTLS "datum" with the OpenPGP public key

1886

datum = gnutls.library.types.gnutls_datum_t(

2291

datum = gnutls.datum_t(

1887

2292

ctypes.cast(ctypes.c_char_p(openpgp),

1888

2293

ctypes.POINTER(ctypes.c_ubyte)),

1889

2294

ctypes.c_uint(len(openpgp)))

1890

2295

# New empty GnuTLS certificate

1891

crt = gnutls.library.types.gnutls_openpgp_crt_t()

1892

gnutls.library.functions.gnutls_openpgp_crt_init(

1893

ctypes.byref(crt))

2296

crt = gnutls.openpgp_crt_t()

2297

gnutls.openpgp_crt_init(ctypes.byref(crt))

1894

2298

# Import the OpenPGP public key into the certificate

1895

gnutls.library.functions.gnutls_openpgp_crt_import(

1896

crt, ctypes.byref(datum),

1897

gnutls.library.constants.GNUTLS_OPENPGP_FMT_RAW)

2299

gnutls.openpgp_crt_import(crt, ctypes.byref(datum),

2300

gnutls.OPENPGP_FMT_RAW)

1898

2301

# Verify the self signature in the key

1899

2302

crtverify = ctypes.c_uint()

1900

gnutls.library.functions.gnutls_openpgp_crt_verify_self(

1901

crt, 0, ctypes.byref(crtverify))

2303

gnutls.openpgp_crt_verify_self(crt, 0,

2304

ctypes.byref(crtverify))

1902

2305

if crtverify.value != 0:

1903

gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)

1904

raise gnutls.errors.CertificateSecurityError(

1905

"Verify failed")

2306

gnutls.openpgp_crt_deinit(crt)

2307

raise gnutls.CertificateSecurityError("Verify failed")

1906

2308

# New buffer for the fingerprint

1907

2309

buf = ctypes.create_string_buffer(20)

1908

2310

buf_len = ctypes.c_size_t()

1909

2311

# Get the fingerprint from the certificate into the buffer

1910

gnutls.library.functions.gnutls_openpgp_crt_get_fingerprint(

1911

crt, ctypes.byref(buf), ctypes.byref(buf_len))

2312

gnutls.openpgp_crt_get_fingerprint(crt, ctypes.byref(buf),

2313

ctypes.byref(buf_len))

1912

2314

# Deinit the certificate

1913

gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)

2315

gnutls.openpgp_crt_deinit(crt)

1914

2316

# Convert the buffer to a Python bytestring

1915

2317

fpr = ctypes.string_at(buf, buf_len.value)

1916

2318

# Convert the bytestring to hexadecimal notation

2060

2462

gnutls_priority GnuTLS priority string

2061

2463

use_dbus: Boolean; to emit D-Bus signals or not

2062

2464

2063

Assumes a gobject.MainLoop event loop.

2465

Assumes a GObject.MainLoop event loop.

2064

2466

"""

2065

2467

2066

2468

def __init__(self, server_address, RequestHandlerClass,

2091

2493

2092

2494

def add_pipe(self, parent_pipe, proc):

2093

2495

# Call "handle_ipc" for both data and EOF events

2094

gobject.io_add_watch(

2496

GObject.io_add_watch(

2095

2497

parent_pipe.fileno(),

2096

gobject.IO_IN | gobject.IO_HUP,

2498

GObject.IO_IN | GObject.IO_HUP,

2097

2499

functools.partial(self.handle_ipc,

2098

2500

parent_pipe = parent_pipe,

2099

2501

proc = proc))

2103

2505

proc = None,

2104

2506

client_object=None):

2105

2507

# error, or the other end of multiprocessing.Pipe has closed

2106

if condition & (gobject.IO_ERR | gobject.IO_HUP):

2508

if condition & (GObject.IO_ERR | GObject.IO_HUP):

2107

2509

# Wait for other process to exit

2108

2510

proc.join()

2109

2511

return False

2130

2532

parent_pipe.send(False)

2131

2533

return False

2132

2534

2133

gobject.io_add_watch(

2535

GObject.io_add_watch(

2134

2536

parent_pipe.fileno(),

2135

gobject.IO_IN | gobject.IO_HUP,

2537

GObject.IO_IN | GObject.IO_HUP,

2136

2538

functools.partial(self.handle_ipc,

2137

2539

parent_pipe = parent_pipe,

2138

2540

proc = proc,

2520

2922

logger.error("Could not open file %r", pidfilename,

2521

2923

exc_info=e)

2522

2924

2523

for name in ("_mandos", "mandos", "nobody"):

2925

for name, group in (("_mandos", "_mandos"),

2926

("mandos", "mandos"),

2927

("nobody", "nogroup")):

2524

2928

try:

2525

2929

uid = pwd.getpwnam(name).pw_uid

2526

gid = pwd.getpwnam(name).pw_gid

2930

gid = pwd.getpwnam(group).pw_gid

2527

2931

break

2528

2932

except KeyError:

2529

2933

continue

2542

2946

2543

2947

# "Use a log level over 10 to enable all debugging options."

2544

2948

# - GnuTLS manual

2545

gnutls.library.functions.gnutls_global_set_log_level(11)

2949

gnutls.global_set_log_level(11)

2546

2950

2547

@gnutls.library.types.gnutls_log_func

2951

@gnutls.log_func

2548

2952

def debug_gnutls(level, string):

2549

2953

logger.debug("GnuTLS: %s", string[:-1])

2550

2954

2551

gnutls.library.functions.gnutls_global_set_log_function(

2552

debug_gnutls)

2955

gnutls.global_set_log_function(debug_gnutls)

2553

2956

2554

2957

# Redirect stdin so all checkers get /dev/null

2555

2958

null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)

2562

2965

# Close all input and output, do double fork, etc.

2563

2966

daemon()

2564

2967

2565

# multiprocessing will use threads, so before we use gobject we

2566

# need to inform gobject that threads will be used.

2567

gobject.threads_init()

2968

# multiprocessing will use threads, so before we use GObject we

2969

# need to inform GObject that threads will be used.

2970

GObject.threads_init()

2568

2971

2569

2972

global main_loop

2570

2973

# From the Avahi example code

2571

2974

DBusGMainLoop(set_as_default=True)

2572

main_loop = gobject.MainLoop()

2975

main_loop = GObject.MainLoop()

2573

2976

bus = dbus.SystemBus()

2574

2977

# End of Avahi example code

2575

2978

if use_dbus:

2734

3137

2735

3138

@alternate_dbus_interfaces(

2736

3139

{ "se.recompile.Mandos": "se.bsnet.fukt.Mandos" })

2737

class MandosDBusService(DBusObjectWithProperties):

3140

class MandosDBusService(DBusObjectWithObjectManager):

2738

3141

"""A D-Bus proxy object"""

2739

3142

2740

3143

def __init__(self):

2742

3145

2743

3146

_interface = "se.recompile.Mandos"

2744

3147

2745

@dbus_interface_annotations(_interface)

2746

def _foo(self):

2747

return {

2748

"org.freedesktop.DBus.Property.EmitsChangedSignal":

2749

"false" }

2750

2751

3148

@dbus.service.signal(_interface, signature="o")

2752

3149

def ClientAdded(self, objpath):

2753

3150

"D-Bus signal"

2758

3155

"D-Bus signal"

2759

3156

pass

2760

3157

3158

@dbus_annotations({"org.freedesktop.DBus.Deprecated":

3159

"true"})

2761

3160

@dbus.service.signal(_interface, signature="os")

2762

3161

def ClientRemoved(self, objpath, name):

2763

3162

"D-Bus signal"

2764

3163

pass

2765

3164

3165

@dbus_annotations({"org.freedesktop.DBus.Deprecated":

3166

"true"})

2766

3167

@dbus.service.method(_interface, out_signature="ao")

2767

3168

def GetAllClients(self):

2768

3169

"D-Bus method"

2769

3170

return dbus.Array(c.dbus_object_path for c in

2770

3171

tcp_server.clients.itervalues())

2771

3172

3173

@dbus_annotations({"org.freedesktop.DBus.Deprecated":

3174

"true"})

2772

3175

@dbus.service.method(_interface,

2773

3176

out_signature="a{oa{sv}}")

2774

3177

def GetAllClientsWithProperties(self):

2775

3178

"D-Bus method"

2776

3179

return dbus.Dictionary(

2777

{ c.dbus_object_path: c.GetAll("")

3180

{ c.dbus_object_path: c.GetAll(

3181

"se.recompile.Mandos.Client")

2778

3182

for c in tcp_server.clients.itervalues() },

2779

3183

signature="oa{sv}")

2780

3184

2785

3189

if c.dbus_object_path == object_path:

2786

3190

del tcp_server.clients[c.name]

2787

3191

c.remove_from_connection()

2788

# Don't signal anything except ClientRemoved

3192

# Don't signal the disabling

2789

3193

c.disable(quiet=True)

2790

# Emit D-Bus signal

2791

self.ClientRemoved(object_path, c.name)

3194

# Emit D-Bus signal for removal

3195

self.client_removed_signal(c)

2792

3196

return

2793

3197

raise KeyError(object_path)

2794

3198

2795

3199

del _interface

3200

3201

@dbus.service.method(dbus.OBJECT_MANAGER_IFACE,

3202

out_signature = "a{oa{sa{sv}}}")

3203

def GetManagedObjects(self):

3204

"""D-Bus method"""

3205

return dbus.Dictionary(

3206

{ client.dbus_object_path:

3207

dbus.Dictionary(

3208

{ interface: client.GetAll(interface)

3209

for interface in

3210

client._get_all_interface_names()})

3211

for client in tcp_server.clients.values()})

3212

3213

def client_added_signal(self, client):

3214

"""Send the new standard signal and the old signal"""

3215

if use_dbus:

3216

# New standard signal

3217

self.InterfacesAdded(

3218

client.dbus_object_path,

3219

dbus.Dictionary(

3220

{ interface: client.GetAll(interface)

3221

for interface in

3222

client._get_all_interface_names()}))

3223

# Old signal

3224

self.ClientAdded(client.dbus_object_path)

3225

3226

def client_removed_signal(self, client):

3227

"""Send the new standard signal and the old signal"""

3228

if use_dbus:

3229

# New standard signal

3230

self.InterfacesRemoved(

3231

client.dbus_object_path,

3232

client._get_all_interface_names())

3233

# Old signal

3234

self.ClientRemoved(client.dbus_object_path,

3235

client.name)

2796

3236

2797

3237

mandos_dbus_service = MandosDBusService()

2798

3238

2863

3303

name, client = tcp_server.clients.popitem()

2864

3304

if use_dbus:

2865

3305

client.remove_from_connection()

2866

# Don't signal anything except ClientRemoved

3306

# Don't signal the disabling

2867

3307

client.disable(quiet=True)

3308

# Emit D-Bus signal for removal

2868

3309

if use_dbus:

2869

# Emit D-Bus signal

2870

mandos_dbus_service.ClientRemoved(

2871

client.dbus_object_path, client.name)

3310

mandos_dbus_service.client_removed_signal(client)

2872

3311

client_settings.clear()

2873

3312

2874

3313

atexit.register(cleanup)

2875

3314

2876

3315

for client in tcp_server.clients.itervalues():

2877

3316

if use_dbus:

2878

# Emit D-Bus signal

2879

mandos_dbus_service.ClientAdded(client.dbus_object_path)

3317

# Emit D-Bus signal for adding

3318

mandos_dbus_service.client_added_signal(client)

2880

3319

# Need to initiate checking of clients

2881

3320

if client.enabled:

2882

3321

client.init_checker()

2908

3347

sys.exit(1)

2909

3348

# End of Avahi example code

2910

3349

2911

gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,

3350

GObject.io_add_watch(tcp_server.fileno(), GObject.IO_IN,

2912

3351

lambda *args, **kwargs:

2913

3352

(tcp_server.handle_request

2914

3353

(*args[2:], **kwargs) or True))