/mandos/trunk : revision 1013

1

#!/usr/bin/python

2

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

3

4

from __future__ import division

2

# -*- mode: python; coding: utf-8; after-save-hook: (lambda () (let ((command (if (and (boundp 'tramp-file-name-structure) (string-match (car tramp-file-name-structure) (buffer-file-name))) (tramp-file-name-localname (tramp-dissect-file-name (buffer-file-name))) (buffer-file-name)))) (if (= (shell-command (format "%s --check" (shell-quote-argument command)) "*Test*") 0) (let ((w (get-buffer-window "*Test*"))) (if w (delete-window w)) (kill-buffer "*Test*")) (display-buffer "*Test*")))); -*-

3

#

4

# Mandos Monitor - Control and monitor the Mandos server

5

#

6

7

8

#

9

# This file is part of Mandos.

10

#

11

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

12

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

13

# the Free Software Foundation, either version 3 of the License, or

14

# (at your option) any later version.

15

#

16

# Mandos is distributed in the hope that it will be useful, but

17

# WITHOUT ANY WARRANTY; without even the implied warranty of

18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

19

# GNU General Public License for more details.

20

#

21

# You should have received a copy of the GNU General Public License

22

# along with Mandos. If not, see <http://www.gnu.org/licenses/>.

23

#

24

# Contact the authors at <mandos@recompile.se>.

25

#

26

27

from __future__ import (division, absolute_import, print_function,

28

unicode_literals)

29

30

try:

31

from future_builtins import *

32

except ImportError:

33

pass

34

5

35

import sys

6

import dbus

7

from optparse import OptionParser

36

import argparse

8

37

import locale

9

38

import datetime

10

39

import re

11

12

locale.setlocale(locale.LC_ALL, u'')

13

14

tablewords = {

15

'Name': u'Name',

16

'Enabled': u'Enabled',

17

'Timeout': u'Timeout',

18

'LastCheckedOK': u'Last Successful Check',

19

'Created': u'Created',

20

'Interval': u'Interval',

21

'Host': u'Host',

22

'Fingerprint': u'Fingerprint',

23

'CheckerRunning': u'Check Is Running',

24

'LastEnabled': u'Last Enabled',

25

'Checker': u'Checker',

26

}

27

defaultkeywords = ('Name', 'Enabled', 'Timeout', 'LastCheckedOK')

28

domain = 'se.bsnet.fukt'

29

busname = domain + '.Mandos'

30

server_path = '/'

31

server_interface = domain + '.Mandos'

32

client_interface = domain + '.Mandos.Client'

33

version = "1.0.14"

40

import os

41

import collections

42

import json

43

import unittest

44

import logging

45

46

import dbus

47

48

# Show warnings by default

49

if not sys.warnoptions:

50

import warnings

51

warnings.simplefilter("default")

52

53

log = logging.getLogger(sys.argv[0])

54

logging.basicConfig(level="INFO", # Show info level messages

55

format="%(message)s") # Show basic log messages

56

57

logging.captureWarnings(True) # Show warnings via the logging system

58

59

if sys.version_info.major == 2:

60

str = unicode

61

62

locale.setlocale(locale.LC_ALL, "")

63

64

domain = "se.recompile"

65

busname = domain + ".Mandos"

66

server_path = "/"

67

server_interface = domain + ".Mandos"

68

client_interface = domain + ".Mandos.Client"

69

version = "1.8.3"

70

71

34

72

try:

35

bus = dbus.SystemBus()

36

mandos_dbus_objc = bus.get_object(busname, server_path)

37

except dbus.exceptions.DBusException:

38

sys.exit(1)

39

40

mandos_serv = dbus.Interface(mandos_dbus_objc,

41

dbus_interface = server_interface)

42

mandos_clients = mandos_serv.GetAllClientsWithProperties()

73

dbus.OBJECT_MANAGER_IFACE

74

except AttributeError:

75

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

43

76

44

def timedelta_to_milliseconds(td):

45

"Convert a datetime.timedelta object to milliseconds"

46

return ((td.days * 24 * 60 * 60 * 1000)

47

+ (td.seconds * 1000)

48

+ (td.microseconds // 1000))

49

77

50

78

def milliseconds_to_string(ms):

51

79

td = datetime.timedelta(0, 0, 0, ms)

52

return (u"%(days)s%(hours)02d:%(minutes)02d:%(seconds)02d"

53

% { "days": "%dT" % td.days if td.days else "",

54

"hours": td.seconds // 3600,

55

"minutes": (td.seconds % 3600) // 60,

56

"seconds": td.seconds % 60,

57

})

80

return ("{days}{hours:02}:{minutes:02}:{seconds:02}"

81

.format(days="{}T".format(td.days) if td.days else "",

82

hours=td.seconds // 3600,

83

minutes=(td.seconds % 3600) // 60,

84

seconds=td.seconds % 60))

85

86

87

def rfc3339_duration_to_delta(duration):

88

"""Parse an RFC 3339 "duration" and return a datetime.timedelta

89

90

>>> rfc3339_duration_to_delta("P7D")

91

datetime.timedelta(7)

92

>>> rfc3339_duration_to_delta("PT60S")

93

datetime.timedelta(0, 60)

94

>>> rfc3339_duration_to_delta("PT60M")

95

datetime.timedelta(0, 3600)

96

>>> rfc3339_duration_to_delta("P60M")

97

datetime.timedelta(1680)

98

>>> rfc3339_duration_to_delta("PT24H")

99

datetime.timedelta(1)

100

>>> rfc3339_duration_to_delta("P1W")

101

datetime.timedelta(7)

102

>>> rfc3339_duration_to_delta("PT5M30S")

103

datetime.timedelta(0, 330)

104

>>> rfc3339_duration_to_delta("P1DT3M20S")

105

datetime.timedelta(1, 200)

106

>>> # Can not be empty:

107

>>> rfc3339_duration_to_delta("")

108

Traceback (most recent call last):

109

...

110

ValueError: Invalid RFC 3339 duration: u''

111

>>> # Must start with "P":

112

>>> rfc3339_duration_to_delta("1D")

113

Traceback (most recent call last):

114

...

115

ValueError: Invalid RFC 3339 duration: u'1D'

116

>>> # Must use correct order

117

>>> rfc3339_duration_to_delta("PT1S2M")

118

Traceback (most recent call last):

119

...

120

ValueError: Invalid RFC 3339 duration: u'PT1S2M'

121

>>> # Time needs time marker

122

>>> rfc3339_duration_to_delta("P1H2S")

123

Traceback (most recent call last):

124

...

125

ValueError: Invalid RFC 3339 duration: u'P1H2S'

126

>>> # Weeks can not be combined with anything else

127

>>> rfc3339_duration_to_delta("P1D2W")

128

Traceback (most recent call last):

129

...

130

ValueError: Invalid RFC 3339 duration: u'P1D2W'

131

>>> rfc3339_duration_to_delta("P2W2H")

132

Traceback (most recent call last):

133

...

134

ValueError: Invalid RFC 3339 duration: u'P2W2H'

135

"""

136

137

# Parsing an RFC 3339 duration with regular expressions is not

138

# possible - there would have to be multiple places for the same

139

# values, like seconds. The current code, while more esoteric, is

140

# cleaner without depending on a parsing library. If Python had a

141

# built-in library for parsing we would use it, but we'd like to

142

# avoid excessive use of external libraries.

143

144

# New type for defining tokens, syntax, and semantics all-in-one

145

Token = collections.namedtuple("Token", (

146

"regexp", # To match token; if "value" is not None, must have

147

# a "group" containing digits

148

"value", # datetime.timedelta or None

149

"followers")) # Tokens valid after this token

150

# RFC 3339 "duration" tokens, syntax, and semantics; taken from

151

# the "duration" ABNF definition in RFC 3339, Appendix A.

152

token_end = Token(re.compile(r"$"), None, frozenset())

153

token_second = Token(re.compile(r"(\d+)S"),

154

datetime.timedelta(seconds=1),

155

frozenset((token_end, )))

156

token_minute = Token(re.compile(r"(\d+)M"),

157

datetime.timedelta(minutes=1),

158

frozenset((token_second, token_end)))

159

token_hour = Token(re.compile(r"(\d+)H"),

160

datetime.timedelta(hours=1),

161

frozenset((token_minute, token_end)))

162

token_time = Token(re.compile(r"T"),

163

None,

164

frozenset((token_hour, token_minute,

165

token_second)))

166

token_day = Token(re.compile(r"(\d+)D"),

167

datetime.timedelta(days=1),

168

frozenset((token_time, token_end)))

169

token_month = Token(re.compile(r"(\d+)M"),

170

datetime.timedelta(weeks=4),

171

frozenset((token_day, token_end)))

172

token_year = Token(re.compile(r"(\d+)Y"),

173

datetime.timedelta(weeks=52),

174

frozenset((token_month, token_end)))

175

token_week = Token(re.compile(r"(\d+)W"),

176

datetime.timedelta(weeks=1),

177

frozenset((token_end, )))

178

token_duration = Token(re.compile(r"P"), None,

179

frozenset((token_year, token_month,

180

token_day, token_time,

181

token_week)))

182

# Define starting values:

183

# Value so far

184

value = datetime.timedelta()

185

found_token = None

186

# Following valid tokens

187

followers = frozenset((token_duration, ))

188

# String left to parse

189

s = duration

190

# Loop until end token is found

191

while found_token is not token_end:

192

# Search for any currently valid tokens

193

for token in followers:

194

match = token.regexp.match(s)

195

if match is not None:

196

# Token found

197

if token.value is not None:

198

# Value found, parse digits

199

factor = int(match.group(1), 10)

200

# Add to value so far

201

value += factor * token.value

202

# Strip token from string

203

s = token.regexp.sub("", s, 1)

204

# Go to found token

205

found_token = token

206

# Set valid next tokens

207

followers = found_token.followers

208

break

209

else:

210

# No currently valid tokens were found

211

raise ValueError("Invalid RFC 3339 duration: {!r}"

212

.format(duration))

213

# End token found

214

return value

58

215

59

216

60

217

def string_to_delta(interval):

61

"""Parse a string and return a datetime.timedelta

62

63

>>> string_to_delta('7d')

218

"""Parse a string and return a datetime.timedelta"""

219

220

try:

221

return rfc3339_duration_to_delta(interval)

222

except ValueError as e:

223

log.warning("%s - Parsing as pre-1.6.1 interval instead",

224

' '.join(e.args))

225

return parse_pre_1_6_1_interval(interval)

226

227

228

def parse_pre_1_6_1_interval(interval):

229

"""Parse an interval string as documented by Mandos before 1.6.1,

230

and return a datetime.timedelta

231

232

>>> parse_pre_1_6_1_interval('7d')

64

233

datetime.timedelta(7)

65

>>> string_to_delta('60s')

234

>>> parse_pre_1_6_1_interval('60s')

66

235

datetime.timedelta(0, 60)

67

>>> string_to_delta('60m')

236

>>> parse_pre_1_6_1_interval('60m')

68

237

datetime.timedelta(0, 3600)

69

>>> string_to_delta('24h')

238

>>> parse_pre_1_6_1_interval('24h')

70

239

datetime.timedelta(1)

71

>>> string_to_delta(u'1w')

240

>>> parse_pre_1_6_1_interval('1w')

72

241

datetime.timedelta(7)

73

>>> string_to_delta('5m 30s')

242

>>> parse_pre_1_6_1_interval('5m 30s')

74

243

datetime.timedelta(0, 330)

244

>>> parse_pre_1_6_1_interval('')

245

datetime.timedelta(0)

246

>>> # Ignore unknown characters, allow any order and repetitions

247

>>> parse_pre_1_6_1_interval('2dxy7zz11y3m5m')

248

datetime.timedelta(2, 480, 18000)

249

75

250

"""

76

timevalue = datetime.timedelta(0)

77

regexp = re.compile("\d+[dsmhw]")

78

79

for s in regexp.findall(interval):

80

try:

81

suffix = unicode(s[-1])

82

value = int(s[:-1])

83

if suffix == u"d":

84

delta = datetime.timedelta(value)

85

elif suffix == u"s":

86

delta = datetime.timedelta(0, value)

87

elif suffix == u"m":

88

delta = datetime.timedelta(0, 0, 0, 0, value)

89

elif suffix == u"h":

90

delta = datetime.timedelta(0, 0, 0, 0, 0, value)

91

elif suffix == u"w":

92

delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)

93

else:

94

raise ValueError

95

except (ValueError, IndexError):

96

raise ValueError

97

timevalue += delta

98

return timevalue

99

100

def print_clients(clients):

101

def valuetostring(value, keyword):

102

if type(value) is dbus.Boolean:

103

return u"Yes" if value else u"No"

104

if keyword in (u"timeout", u"interval"):

105

return milliseconds_to_string(value)

106

return unicode(value)

107

108

# Create format string to print table rows

109

format_string = u' '.join(u'%%-%ds' %

110

max(len(tablewords[key]),

111

max(len(valuetostring(client[key],

112

key))

113

for client in

114

clients))

115

for key in keywords)

116

# Print header line

117

print format_string % tuple(tablewords[key] for key in keywords)

118

for client in clients:

119

print format_string % tuple(valuetostring(client[key], key)

120

for key in keywords)

121

122

parser = OptionParser(version = "%%prog %s" % version)

123

parser.add_option("-a", "--all", action="store_true",

124

help="Print all fields")

125

parser.add_option("-e", "--enable", action="store_true",

126

help="Enable client")

127

parser.add_option("-d", "--disable", action="store_true",

128

help="disable client")

129

parser.add_option("-b", "--bump-timeout", action="store_true",

130

help="Bump timeout for client")

131

parser.add_option("--start-checker", action="store_true",

132

help="Start checker for client")

133

parser.add_option("--stop-checker", action="store_true",

134

help="Stop checker for client")

135

parser.add_option("-V", "--is-enabled", action="store_true",

136

help="Check if client is enabled")

137

parser.add_option("-r", "--remove", action="store_true",

138

help="Remove client")

139

parser.add_option("-c", "--checker", type="string",

140

help="Set checker command for client")

141

parser.add_option("-t", "--timeout", type="string",

142

help="Set timeout for client")

143

parser.add_option("-i", "--interval", type="string",

144

help="Set checker interval for client")

145

parser.add_option("-H", "--host", type="string",

146

help="Set host for client")

147

parser.add_option("-s", "--secret", type="string",

148

help="Set password blob (file) for client")

149

parser.add_option("-A", "--approve", action="store_true",

150

help="Approve any current client request")

151

parser.add_option("-D", "--deny", action="store_true",

152

help="Deny any current client request")

153

options, client_names = parser.parse_args()

154

155

# Compile list of clients to process

156

clients=[]

157

for name in client_names:

158

for path, client in mandos_clients.iteritems():

159

if client['name'] == name:

160

client_objc = bus.get_object(busname, path)

161

clients.append(client_objc)

162

break

163

else:

164

print >> sys.stderr, "Client not found on server: %r" % name

251

252

value = datetime.timedelta(0)

253

regexp = re.compile(r"(\d+)([dsmhw]?)")

254

255

for num, suffix in regexp.findall(interval):

256

if suffix == "d":

257

value += datetime.timedelta(int(num))

258

elif suffix == "s":

259

value += datetime.timedelta(0, int(num))

260

elif suffix == "m":

261

value += datetime.timedelta(0, 0, 0, 0, int(num))

262

elif suffix == "h":

263

value += datetime.timedelta(0, 0, 0, 0, 0, int(num))

264

elif suffix == "w":

265

value += datetime.timedelta(0, 0, 0, 0, 0, 0, int(num))

266

elif suffix == "":

267

value += datetime.timedelta(0, 0, 0, int(num))

268

return value

269

270

271

## Classes for commands.

272

273

# Abstract classes first

274

class Command(object):

275

"""Abstract class for commands"""

276

def run(self, mandos, clients):

277

"""Normal commands should implement run_on_one_client(), but

278

commands which want to operate on all clients at the same time

279

can override this run() method instead."""

280

self.mandos = mandos

281

for client in clients:

282

self.run_on_one_client(client)

283

284

class PrintCmd(Command):

285

"""Abstract class for commands printing client details"""

286

all_keywords = ("Name", "Enabled", "Timeout", "LastCheckedOK",

287

"Created", "Interval", "Host", "KeyID",

288

"Fingerprint", "CheckerRunning", "LastEnabled",

289

"ApprovalPending", "ApprovedByDefault",

290

"LastApprovalRequest", "ApprovalDelay",

291

"ApprovalDuration", "Checker", "ExtendedTimeout",

292

"Expires", "LastCheckerStatus")

293

def run(self, mandos, clients):

294

print(self.output(clients))

295

296

class PropertyCmd(Command):

297

"""Abstract class for Actions for setting one client property"""

298

def run_on_one_client(self, client):

299

"""Set the Client's D-Bus property"""

300

client.Set(client_interface, self.property, self.value_to_set,

301

dbus_interface=dbus.PROPERTIES_IFACE)

302

303

class ValueArgumentMixIn(object):

304

"""Mixin class for commands taking a value as argument"""

305

def __init__(self, value):

306

self.value_to_set = value

307

308

class MillisecondsValueArgumentMixIn(ValueArgumentMixIn):

309

"""Mixin class for commands taking a value argument as

310

milliseconds."""

311

@property

312

def value_to_set(self):

313

return self._vts

314

@value_to_set.setter

315

def value_to_set(self, value):

316

"""When setting, convert value to a datetime.timedelta"""

317

self._vts = string_to_delta(value).total_seconds() * 1000

318

319

# Actual (non-abstract) command classes

320

321

class PrintTableCmd(PrintCmd):

322

def __init__(self, verbose=False):

323

self.verbose = verbose

324

325

def output(self, clients):

326

if self.verbose:

327

keywords = self.all_keywords

328

else:

329

keywords = ("Name", "Enabled", "Timeout", "LastCheckedOK")

330

return str(self.TableOfClients(clients.values(), keywords))

331

332

class TableOfClients(object):

333

tableheaders = {

334

"Name": "Name",

335

"Enabled": "Enabled",

336

"Timeout": "Timeout",

337

"LastCheckedOK": "Last Successful Check",

338

"LastApprovalRequest": "Last Approval Request",

339

"Created": "Created",

340

"Interval": "Interval",

341

"Host": "Host",

342

"Fingerprint": "Fingerprint",

343

"KeyID": "Key ID",

344

"CheckerRunning": "Check Is Running",

345

"LastEnabled": "Last Enabled",

346

"ApprovalPending": "Approval Is Pending",

347

"ApprovedByDefault": "Approved By Default",

348

"ApprovalDelay": "Approval Delay",

349

"ApprovalDuration": "Approval Duration",

350

"Checker": "Checker",

351

"ExtendedTimeout": "Extended Timeout",

352

"Expires": "Expires",

353

"LastCheckerStatus": "Last Checker Status",

354

}

355

356

def __init__(self, clients, keywords, tableheaders=None):

357

self.clients = clients

358

self.keywords = keywords

359

if tableheaders is not None:

360

self.tableheaders = tableheaders

361

362

def __str__(self):

363

return "\n".join(self.rows())

364

365

if sys.version_info.major == 2:

366

__unicode__ = __str__

367

def __str__(self):

368

return str(self).encode(locale.getpreferredencoding())

369

370

def rows(self):

371

format_string = self.row_formatting_string()

372

rows = [self.header_line(format_string)]

373

rows.extend(self.client_line(client, format_string)

374

for client in self.clients)

375

return rows

376

377

def row_formatting_string(self):

378

"Format string used to format table rows"

379

return " ".join("{{{key}:{width}}}".format(

380

width=max(len(self.tableheaders[key]),

381

*(len(self.string_from_client(client, key))

382

for client in self.clients)),

383

key=key)

384

for key in self.keywords)

385

386

def string_from_client(self, client, key):

387

return self.valuetostring(client[key], key)

388

389

@staticmethod

390

def valuetostring(value, keyword):

391

if isinstance(value, dbus.Boolean):

392

return "Yes" if value else "No"

393

if keyword in ("Timeout", "Interval", "ApprovalDelay",

394

"ApprovalDuration", "ExtendedTimeout"):

395

return milliseconds_to_string(value)

396

return str(value)

397

398

def header_line(self, format_string):

399

return format_string.format(**self.tableheaders)

400

401

def client_line(self, client, format_string):

402

return format_string.format(

403

**{key: self.string_from_client(client, key)

404

for key in self.keywords})

405

406

407

408

class DumpJSONCmd(PrintCmd):

409

def output(self, clients):

410

data = {client["Name"]:

411

{key: self.dbus_boolean_to_bool(client[key])

412

for key in self.all_keywords}

413

for client in clients.values()}

414

return json.dumps(data, indent=4, separators=(',', ': '))

415

@staticmethod

416

def dbus_boolean_to_bool(value):

417

if isinstance(value, dbus.Boolean):

418

value = bool(value)

419

return value

420

421

class IsEnabledCmd(Command):

422

def run_on_one_client(self, client):

423

if self.is_enabled(client):

424

sys.exit(0)

165

425

sys.exit(1)

166

167

if not clients and mandos_clients.values():

168

keywords = defaultkeywords

169

if options.all:

170

keywords = ('Name', 'Enabled', 'Timeout', 'LastCheckedOK',

171

'Created', 'Interval', 'Host', 'Fingerprint',

172

'CheckerRunning', 'LastEnabled', 'Checker')

173

print_clients(mandos_clients.values())

174

175

# Process each client in the list by all selected options

176

for client in clients:

177

if options.remove:

178

mandos_serv.RemoveClient(client.__dbus_object_path__)

426

def is_enabled(self, client):

427

return client.Get(client_interface, "Enabled",

428

dbus_interface=dbus.PROPERTIES_IFACE)

429

430

class RemoveCmd(Command):

431

def run_on_one_client(self, client):

432

self.mandos.RemoveClient(client.__dbus_object_path__)

433

434

class ApproveCmd(Command):

435

def run_on_one_client(self, client):

436

client.Approve(dbus.Boolean(True),

437

dbus_interface=client_interface)

438

439

class DenyCmd(Command):

440

def run_on_one_client(self, client):

441

client.Approve(dbus.Boolean(False),

442

dbus_interface=client_interface)

443

444

class EnableCmd(PropertyCmd):

445

property = "Enabled"

446

value_to_set = dbus.Boolean(True)

447

448

class DisableCmd(PropertyCmd):

449

property = "Enabled"

450

value_to_set = dbus.Boolean(False)

451

452

class BumpTimeoutCmd(PropertyCmd):

453

property = "LastCheckedOK"

454

value_to_set = ""

455

456

class StartCheckerCmd(PropertyCmd):

457

property = "CheckerRunning"

458

value_to_set = dbus.Boolean(True)

459

460

class StopCheckerCmd(PropertyCmd):

461

property = "CheckerRunning"

462

value_to_set = dbus.Boolean(False)

463

464

class ApproveByDefaultCmd(PropertyCmd):

465

property = "ApprovedByDefault"

466

value_to_set = dbus.Boolean(True)

467

468

class DenyByDefaultCmd(PropertyCmd):

469

property = "ApprovedByDefault"

470

value_to_set = dbus.Boolean(False)

471

472

class SetCheckerCmd(PropertyCmd, ValueArgumentMixIn):

473

property = "Checker"

474

475

class SetHostCmd(PropertyCmd, ValueArgumentMixIn):

476

property = "Host"

477

478

class SetSecretCmd(PropertyCmd, ValueArgumentMixIn):

479

property = "Secret"

480

481

class SetTimeoutCmd(PropertyCmd, MillisecondsValueArgumentMixIn):

482

property = "Timeout"

483

484

class SetExtendedTimeoutCmd(PropertyCmd,

485

MillisecondsValueArgumentMixIn):

486

property = "ExtendedTimeout"

487

488

class SetIntervalCmd(PropertyCmd, MillisecondsValueArgumentMixIn):

489

property = "Interval"

490

491

class SetApprovalDelayCmd(PropertyCmd,

492

MillisecondsValueArgumentMixIn):

493

property = "ApprovalDelay"

494

495

class SetApprovalDurationCmd(PropertyCmd,

496

MillisecondsValueArgumentMixIn):

497

property = "ApprovalDuration"

498

499

def has_actions(options):

500

return any((options.enable,

501

options.disable,

502

options.bump_timeout,

503

options.start_checker,

504

options.stop_checker,

505

options.is_enabled,

506

options.remove,

507

options.checker is not None,

508

options.timeout is not None,

509

options.extended_timeout is not None,

510

options.interval is not None,

511

options.approved_by_default is not None,

512

options.approval_delay is not None,

513

options.approval_duration is not None,

514

options.host is not None,

515

options.secret is not None,

516

options.approve,

517

options.deny))

518

519

520

def commands_and_clients_from_options(args=None):

521

if args is None:

522

args=sys.argv[1:]

523

parser = argparse.ArgumentParser()

524

parser.add_argument("--version", action="version",

525

version="%(prog)s {}".format(version),

526

help="show version number and exit")

527

parser.add_argument("-a", "--all", action="store_true",

528

help="Select all clients")

529

parser.add_argument("-v", "--verbose", action="store_true",

530

help="Print all fields")

531

parser.add_argument("-j", "--dump-json", action="store_true",

532

help="Dump client data in JSON format")

533

enable_disable = parser.add_mutually_exclusive_group()

534

enable_disable.add_argument("-e", "--enable", action="store_true",

535

help="Enable client")

536

enable_disable.add_argument("-d", "--disable",

537

action="store_true",

538

help="disable client")

539

parser.add_argument("-b", "--bump-timeout", action="store_true",

540

help="Bump timeout for client")

541

start_stop_checker = parser.add_mutually_exclusive_group()

542

start_stop_checker.add_argument("--start-checker",

543

action="store_true",

544

help="Start checker for client")

545

start_stop_checker.add_argument("--stop-checker",

546

action="store_true",

547

help="Stop checker for client")

548

parser.add_argument("-V", "--is-enabled", action="store_true",

549

help="Check if client is enabled")

550

parser.add_argument("-r", "--remove", action="store_true",

551

help="Remove client")

552

parser.add_argument("-c", "--checker",

553

help="Set checker command for client")

554

parser.add_argument("-t", "--timeout",

555

help="Set timeout for client")

556

parser.add_argument("--extended-timeout",

557

help="Set extended timeout for client")

558

parser.add_argument("-i", "--interval",

559

help="Set checker interval for client")

560

approve_deny_default = parser.add_mutually_exclusive_group()

561

approve_deny_default.add_argument(

562

"--approve-by-default", action="store_true",

563

default=None, dest="approved_by_default",

564

help="Set client to be approved by default")

565

approve_deny_default.add_argument(

566

"--deny-by-default", action="store_false",

567

dest="approved_by_default",

568

help="Set client to be denied by default")

569

parser.add_argument("--approval-delay",

570

help="Set delay before client approve/deny")

571

parser.add_argument("--approval-duration",

572

help="Set duration of one client approval")

573

parser.add_argument("-H", "--host", help="Set host for client")

574

parser.add_argument("-s", "--secret",

575

type=argparse.FileType(mode="rb"),

576

help="Set password blob (file) for client")

577

approve_deny = parser.add_mutually_exclusive_group()

578

approve_deny.add_argument(

579

"-A", "--approve", action="store_true",

580

help="Approve any current client request")

581

approve_deny.add_argument("-D", "--deny", action="store_true",

582

help="Deny any current client request")

583

parser.add_argument("--check", action="store_true",

584

help="Run self-test")

585

parser.add_argument("client", nargs="*", help="Client name")

586

options = parser.parse_args(args=args)

587

588

if has_actions(options) and not (options.client or options.all):

589

parser.error("Options require clients names or --all.")

590

if options.verbose and has_actions(options):

591

parser.error("--verbose can only be used alone.")

592

if options.dump_json and (options.verbose

593

or has_actions(options)):

594

parser.error("--dump-json can only be used alone.")

595

if options.all and not has_actions(options):

596

parser.error("--all requires an action.")

597

if options.is_enabled and len(options.client) > 1:

598

parser.error("--is-enabled requires exactly one client")

599

600

commands = []

601

602

if options.dump_json:

603

commands.append(DumpJSONCmd())

604

179

605

if options.enable:

180

client.Enable(dbus_interface=client_interface)

606

commands.append(EnableCmd())

607

181

608

if options.disable:

182

client.Disable(dbus_interface=client_interface)

609

commands.append(DisableCmd())

610

183

611

if options.bump_timeout:

184

client.CheckedOK(dbus_interface=client_interface)

612

commands.append(BumpTimeoutCmd(options.bump_timeout))

613

185

614

if options.start_checker:

186

client.StartChecker(dbus_interface=client_interface)

615

commands.append(StartCheckerCmd())

616

187

617

if options.stop_checker:

188

client.StopChecker(dbus_interface=client_interface)

618

commands.append(StopCheckerCmd())

619

189

620

if options.is_enabled:

190

sys.exit(0 if client.Get(client_interface,

191

u"Enabled",

192

dbus_interface=dbus.PROPERTIES_IFACE)

193

else 1)

194

if options.checker:

195

client.Set(client_interface, u"Checker", options.checker,

196

dbus_interface=dbus.PROPERTIES_IFACE)

197

if options.host:

198

client.Set(client_interface, u"Host", options.host,

199

dbus_interface=dbus.PROPERTIES_IFACE)

200

if options.interval:

201

client.Set(client_interface, u"Interval",

202

timedelta_to_milliseconds

203

(string_to_delta(options.interval)),

204

dbus_interface=dbus.PROPERTIES_IFACE)

205

if options.timeout:

206

client.Set(client_interface, u"Timeout",

207

timedelta_to_milliseconds(string_to_delta

208

(options.timeout)),

209

dbus_interface=dbus.PROPERTIES_IFACE)

210

if options.secret:

211

client.Set(client_interface, u"Secret",

212

dbus.ByteArray(open(options.secret, u'rb').read()),

213

dbus_interface=dbus.PROPERTIES_IFACE)

621

commands.append(IsEnabledCmd())

622

623

if options.remove:

624

commands.append(RemoveCmd())

625

626

if options.checker is not None:

627

commands.append(SetCheckerCmd())

628

629

if options.timeout is not None:

630

commands.append(SetTimeoutCmd(options.timeout))

631

632

if options.extended_timeout:

633

commands.append(

634

SetExtendedTimeoutCmd(options.extended_timeout))

635

636

if options.interval is not None:

637

command.append(SetIntervalCmd(options.interval))

638

639

if options.approved_by_default is not None:

640

if options.approved_by_default:

641

command.append(ApproveByDefaultCmd())

642

else:

643

command.append(DenyByDefaultCmd())

644

645

if options.approval_delay is not None:

646

command.append(SetApprovalDelayCmd(options.approval_delay))

647

648

if options.approval_duration is not None:

649

command.append(

650

SetApprovalDurationCmd(options.approval_duration))

651

652

if options.host is not None:

653

command.append(SetHostCmd(options.host))

654

655

if options.secret is not None:

656

command.append(SetSecretCmd(options.secret))

657

214

658

if options.approve:

215

client.Approve(dbus.Boolean(True),

216

dbus_interface=client_interface)

659

commands.append(ApproveCmd())

660

217

661

if options.deny:

218

client.Approve(dbus.Boolean(False),

219

dbus_interface=client_interface)

662

commands.append(DenyCmd())

663

664

# If no command option has been given, show table of clients,

665

# optionally verbosely

666

if not commands:

667

commands.append(PrintTableCmd(verbose=options.verbose))

668

669

return commands, options.client

670

671

672

def main():

673

commands, clientnames = commands_and_clients_from_options()

674

675

try:

676

bus = dbus.SystemBus()

677

mandos_dbus_objc = bus.get_object(busname, server_path)

678

except dbus.exceptions.DBusException:

679

log.critical("Could not connect to Mandos server")

680

sys.exit(1)

681

682

mandos_serv = dbus.Interface(mandos_dbus_objc,

683

dbus_interface=server_interface)

684

mandos_serv_object_manager = dbus.Interface(

685

mandos_dbus_objc, dbus_interface=dbus.OBJECT_MANAGER_IFACE)

686

687

# Filter out log message from dbus module

688

dbus_logger = logging.getLogger("dbus.proxies")

689

class NullFilter(logging.Filter):

690

def filter(self, record):

691

return False

692

dbus_filter = NullFilter()

693

dbus_logger.addFilter(dbus_filter)

694

try:

695

try:

696

mandos_clients = {path: ifs_and_props[client_interface]

697

for path, ifs_and_props in

698

mandos_serv_object_manager

699

.GetManagedObjects().items()

700

if client_interface in ifs_and_props}

701

finally:

702

# restore dbus logger

703

dbus_logger.removeFilter(dbus_filter)

704

except dbus.exceptions.DBusException as e:

705

log.critical("Failed to access Mandos server through D-Bus:"

706

"\n%s", e)

707

sys.exit(1)

708

709

# Compile dict of (clients: properties) to process

710

clients = {}

711

712

if not clientnames:

713

clients = {bus.get_object(busname, path): properties

714

for path, properties in mandos_clients.items()}

715

else:

716

for name in clientnames:

717

for path, client in mandos_clients.items():

718

if client["Name"] == name:

719

client_objc = bus.get_object(busname, path)

720

clients[client_objc] = client

721

break

722

else:

723

log.critical("Client not found on server: %r", name)

724

sys.exit(1)

725

726

# Run all commands on clients

727

for command in commands:

728

command.run(mandos_serv, clients)

729

730

731

class Test_milliseconds_to_string(unittest.TestCase):

732

def test_all(self):

733

self.assertEqual(milliseconds_to_string(93785000),

734

"1T02:03:05")

735

def test_no_days(self):

736

self.assertEqual(milliseconds_to_string(7385000), "02:03:05")

737

def test_all_zero(self):

738

self.assertEqual(milliseconds_to_string(0), "00:00:00")

739

def test_no_fractional_seconds(self):

740

self.assertEqual(milliseconds_to_string(400), "00:00:00")

741

self.assertEqual(milliseconds_to_string(900), "00:00:00")

742

self.assertEqual(milliseconds_to_string(1900), "00:00:01")

743

744

class Test_string_to_delta(unittest.TestCase):

745

def test_handles_basic_rfc3339(self):

746

self.assertEqual(string_to_delta("PT2H"),

747

datetime.timedelta(0, 7200))

748

def test_falls_back_to_pre_1_6_1_with_warning(self):

749

# assertLogs only exists in Python 3.4

750

if hasattr(self, "assertLogs"):

751

with self.assertLogs(log, logging.WARNING):

752

value = string_to_delta("2h")

753

else:

754

class WarningFilter(logging.Filter):

755

"""Don't show, but record the presence of, warnings"""

756

def filter(self, record):

757

is_warning = record.levelno >= logging.WARNING

758

self.found = is_warning or getattr(self, "found",

759

False)

760

return not is_warning

761

warning_filter = WarningFilter()

762

log.addFilter(warning_filter)

763

try:

764

value = string_to_delta("2h")

765

finally:

766

log.removeFilter(warning_filter)

767

self.assertTrue(getattr(warning_filter, "found", False))

768

self.assertEqual(value, datetime.timedelta(0, 7200))

769

770

771

class TestCmd(unittest.TestCase):

772

"""Abstract class for tests of command classes"""

773

def setUp(self):

774

testcase = self

775

class MockClient(object):

776

def __init__(self, name, **attributes):

777

self.__dbus_object_path__ = "objpath_{}".format(name)

778

self.attributes = attributes

779

self.attributes["Name"] = name

780

self.calls = []

781

def Set(self, interface, property, value, dbus_interface):

782

testcase.assertEqual(interface, client_interface)

783

testcase.assertEqual(dbus_interface,

784

dbus.PROPERTIES_IFACE)

785

self.attributes[property] = value

786

self.calls.append(("Set", (interface, property, value,

787

dbus_interface)))

788

def Get(self, interface, property, dbus_interface):

789

testcase.assertEqual(interface, client_interface)

790

testcase.assertEqual(dbus_interface,

791

dbus.PROPERTIES_IFACE)

792

self.calls.append(("Get", (interface, property,

793

dbus_interface)))

794

return self.attributes[property]

795

def __getitem__(self, key):

796

return self.attributes[key]

797

def __setitem__(self, key, value):

798

self.attributes[key] = value

799

self.clients = collections.OrderedDict([

800

("foo",

801

MockClient(

802

"foo",

803

KeyID=("92ed150794387c03ce684574b1139a65"

804

"94a34f895daaaf09fd8ea90a27cddb12"),

805

Secret=b"secret",

806

Host="foo.example.org",

807

Enabled=dbus.Boolean(True),

808

Timeout=300000,

809

LastCheckedOK="2019-02-03T00:00:00",

810

Created="2019-01-02T00:00:00",

811

Interval=120000,

812

Fingerprint=("778827225BA7DE539C5A"

813

"7CFA59CFF7CDBD9A5920"),

814

CheckerRunning=dbus.Boolean(False),

815

LastEnabled="2019-01-03T00:00:00",

816

ApprovalPending=dbus.Boolean(False),

817

ApprovedByDefault=dbus.Boolean(True),

818

LastApprovalRequest="",

819

ApprovalDelay=0,

820

ApprovalDuration=1000,

821

Checker="fping -q -- %(host)s",

822

ExtendedTimeout=900000,

823

Expires="2019-02-04T00:00:00",

824

LastCheckerStatus=0)),

825

("barbar",

826

MockClient(

827

"barbar",

828

KeyID=("0558568eedd67d622f5c83b35a115f79"

829

"6ab612cff5ad227247e46c2b020f441c"),

830

Secret=b"secretbar",

831

Host="192.0.2.3",

832

Enabled=dbus.Boolean(True),

833

Timeout=300000,

834

LastCheckedOK="2019-02-04T00:00:00",

835

Created="2019-01-03T00:00:00",

836

Interval=120000,

837

Fingerprint=("3E393AEAEFB84C7E89E2"

838

"F547B3A107558FCA3A27"),

839

CheckerRunning=dbus.Boolean(True),

840

LastEnabled="2019-01-04T00:00:00",

841

ApprovalPending=dbus.Boolean(False),

842

ApprovedByDefault=dbus.Boolean(False),

843

LastApprovalRequest="2019-01-03T00:00:00",

844

ApprovalDelay=30000,

845

ApprovalDuration=1000,

846

Checker=":",

847

ExtendedTimeout=900000,

848

Expires="2019-02-05T00:00:00",

849

LastCheckerStatus=-2)),

850

])

851

852

class TestPrintTableCmd(TestCmd):

853

def test_normal(self):

854

output = PrintTableCmd().output(self.clients)

855

expected_output = """

856

Name Enabled Timeout Last Successful Check

857

foo Yes 00:05:00 2019-02-03T00:00:00

858

barbar Yes 00:05:00 2019-02-04T00:00:00

859

"""[1:-1]

860

self.assertEqual(output, expected_output)

861

def test_verbose(self):

862

output = PrintTableCmd(verbose=True).output(self.clients)

863

expected_output = """

864

Name Enabled Timeout Last Successful Check Created Interval Host Key ID Fingerprint Check Is Running Last Enabled Approval Is Pending Approved By Default Last Approval Request Approval Delay Approval Duration Checker Extended Timeout Expires Last Checker Status

865

foo Yes 00:05:00 2019-02-03T00:00:00 2019-01-02T00:00:00 00:02:00 foo.example.org 92ed150794387c03ce684574b1139a6594a34f895daaaf09fd8ea90a27cddb12 778827225BA7DE539C5A7CFA59CFF7CDBD9A5920 No 2019-01-03T00:00:00 No Yes 00:00:00 00:00:01 fping -q -- %(host)s 00:15:00 2019-02-04T00:00:00 0

866

barbar Yes 00:05:00 2019-02-04T00:00:00 2019-01-03T00:00:00 00:02:00 192.0.2.3 0558568eedd67d622f5c83b35a115f796ab612cff5ad227247e46c2b020f441c 3E393AEAEFB84C7E89E2F547B3A107558FCA3A27 Yes 2019-01-04T00:00:00 No No 2019-01-03T00:00:00 00:00:30 00:00:01 : 00:15:00 2019-02-05T00:00:00 -2

867

"""[1:-1]

868

self.assertEqual(output, expected_output)

869

def test_one_client(self):

870

output = PrintTableCmd().output({"foo": self.clients["foo"]})

871

expected_output = """

872

Name Enabled Timeout Last Successful Check

873

foo Yes 00:05:00 2019-02-03T00:00:00

874

"""[1:-1]

875

self.assertEqual(output, expected_output)

876

877

class TestDumpJSONCmd(TestCmd):

878

def setUp(self):

879

self.expected_json = {

880

"foo": {

881

"Name": "foo",

882

"KeyID": ("92ed150794387c03ce684574b1139a65"

883

"94a34f895daaaf09fd8ea90a27cddb12"),

884

"Host": "foo.example.org",

885

"Enabled": True,

886

"Timeout": 300000,

887

"LastCheckedOK": "2019-02-03T00:00:00",

888

"Created": "2019-01-02T00:00:00",

889

"Interval": 120000,

890

"Fingerprint": ("778827225BA7DE539C5A"

891

"7CFA59CFF7CDBD9A5920"),

892

"CheckerRunning": False,

893

"LastEnabled": "2019-01-03T00:00:00",

894

"ApprovalPending": False,

895

"ApprovedByDefault": True,

896

"LastApprovalRequest": "",

897

"ApprovalDelay": 0,

898

"ApprovalDuration": 1000,

899

"Checker": "fping -q -- %(host)s",

900

"ExtendedTimeout": 900000,

901

"Expires": "2019-02-04T00:00:00",

902

"LastCheckerStatus": 0,

903

},

904

"barbar": {

905

"Name": "barbar",

906

"KeyID": ("0558568eedd67d622f5c83b35a115f79"

907

"6ab612cff5ad227247e46c2b020f441c"),

908

"Host": "192.0.2.3",

909

"Enabled": True,

910

"Timeout": 300000,

911

"LastCheckedOK": "2019-02-04T00:00:00",

912

"Created": "2019-01-03T00:00:00",

913

"Interval": 120000,

914

"Fingerprint": ("3E393AEAEFB84C7E89E2"

915

"F547B3A107558FCA3A27"),

916

"CheckerRunning": True,

917

"LastEnabled": "2019-01-04T00:00:00",

918

"ApprovalPending": False,

919

"ApprovedByDefault": False,

920

"LastApprovalRequest": "2019-01-03T00:00:00",

921

"ApprovalDelay": 30000,

922

"ApprovalDuration": 1000,

923

"Checker": ":",

924

"ExtendedTimeout": 900000,

925

"Expires": "2019-02-05T00:00:00",

926

"LastCheckerStatus": -2,

927

},

928

}

929

return super(TestDumpJSONCmd, self).setUp()

930

def test_normal(self):

931

json_data = json.loads(DumpJSONCmd().output(self.clients))

932

self.assertDictEqual(json_data, self.expected_json)

933

def test_one_client(self):

934

clients = {"foo": self.clients["foo"]}

935

json_data = json.loads(DumpJSONCmd().output(clients))

936

expected_json = {"foo": self.expected_json["foo"]}

937

self.assertDictEqual(json_data, expected_json)

938

939

class TestIsEnabledCmd(TestCmd):

940

def test_is_enabled(self):

941

self.assertTrue(all(IsEnabledCmd().is_enabled(client)

942

for client in self.clients.values()))

943

def test_is_enabled_does_get_attribute(self):

944

client = self.clients["foo"]

945

self.assertTrue(IsEnabledCmd().is_enabled(client))

946

self.assertListEqual(client.calls,

947

[("Get",

948

("se.recompile.Mandos.Client",

949

"Enabled",

950

"org.freedesktop.DBus.Properties"))])

951

def test_is_enabled_run_exits_successfully(self):

952

client = self.clients["foo"]

953

with self.assertRaises(SystemExit) as e:

954

IsEnabledCmd().run_on_one_client(client)

955

if e.exception.code is not None:

956

self.assertEqual(e.exception.code, 0)

957

else:

958

self.assertIsNone(e.exception.code)

959

def test_is_enabled_run_exits_with_failure(self):

960

client = self.clients["foo"]

961

client["Enabled"] = dbus.Boolean(False)

962

with self.assertRaises(SystemExit) as e:

963

IsEnabledCmd().run_on_one_client(client)

964

if isinstance(e.exception.code, int):

965

self.assertNotEqual(e.exception.code, 0)

966

else:

967

self.assertIsNotNone(e.exception.code)

968

969

970

971

def should_only_run_tests():

972

parser = argparse.ArgumentParser(add_help=False)

973

parser.add_argument("--check", action='store_true')

974

args, unknown_args = parser.parse_known_args()

975

run_tests = args.check

976

if run_tests:

977

# Remove --check argument from sys.argv

978

sys.argv[1:] = unknown_args

979

return run_tests

980

981

# Add all tests from doctest strings

982

def load_tests(loader, tests, none):

983

import doctest

984

tests.addTests(doctest.DocTestSuite())

985

return tests

986

987

if __name__ == "__main__":

988

if should_only_run_tests():

989

# Call using ./tdd-python-script --check [--verbose]

990

unittest.main()

991

else:

992

main()