/mandos/trunk : revision 487

3

#

4

# Mandos Monitor - Control and monitor the Mandos server

5

#

6

7

6

7

8

#

9

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

10

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

22

# Contact the authors at <mandos@fukt.bsnet.se>.

23

#

24

25

from __future__ import division, absolute_import, with_statement

25

from __future__ import division, absolute_import, print_function, unicode_literals

26

27

import sys

28

import os

42

43

import locale

44

45

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

45

locale.setlocale(locale.LC_ALL, '')

46

47

import logging

48

logging.getLogger('dbus.proxies').setLevel(logging.CRITICAL)

51

domain = 'se.bsnet.fukt'

52

server_interface = domain + '.Mandos'

53

client_interface = domain + '.Mandos.Client'

54

version = "1.2.1"

54

version = "1.3.0"

55

56

# Always run in monochrome mode

57

urwid.curses_display.curses.has_colors = lambda : False

65

"Parse an ISO 8601 date string to a datetime.datetime()"

66

if not iso:

67

return None

68

d, t = iso.split(u"T", 1)

69

year, month, day = d.split(u"-", 2)

70

hour, minute, second = t.split(u":", 2)

68

d, t = iso.split("T", 1)

69

year, month, day = d.split("-", 2)

70

hour, minute, second = t.split(":", 2)

71

second, fraction = divmod(float(second), 1)

72

return datetime.datetime(int(year),

73

int(month),

86

self.proxy = proxy_object # Mandos Client proxy object

87

88

self.properties = dict()

89

self.proxy.connect_to_signal(u"PropertyChanged",

89

self.proxy.connect_to_signal("PropertyChanged",

90

self.property_changed,

91

client_interface,

92

byte_arrays=True)

127

self.last_checker_failed = False

128

129

# The widget shown normally

130

self._text_widget = urwid.Text(u"")

130

self._text_widget = urwid.Text("")

131

# The widget shown when we have focus

132

self._focus_text_widget = urwid.Text(u"")

132

self._focus_text_widget = urwid.Text("")

133

super(MandosClientWidget, self).__init__(

134

update_hook=update_hook, delete_hook=delete_hook,

135

*args, **kwargs)

137

self.opened = False

138

139

last_checked_ok = isoformat_to_datetime(self.properties

140

[u"LastCheckedOK"])

140

["LastCheckedOK"])

141

if last_checked_ok is None:

142

self.last_checker_failed = True

143

else:

146

> datetime.timedelta

147

(milliseconds=

148

self.properties

149

[u"Interval"]))

149

["Interval"]))

150

151

if self.last_checker_failed:

152

self.using_timer(True)

154

if self.need_approval:

155

self.using_timer(True)

156

157

self.proxy.connect_to_signal(u"CheckerCompleted",

157

self.proxy.connect_to_signal("CheckerCompleted",

158

self.checker_completed,

159

client_interface,

160

byte_arrays=True)

161

self.proxy.connect_to_signal(u"CheckerStarted",

161

self.proxy.connect_to_signal("CheckerStarted",

162

self.checker_started,

163

client_interface,

164

byte_arrays=True)

165

self.proxy.connect_to_signal(u"GotSecret",

165

self.proxy.connect_to_signal("GotSecret",

166

self.got_secret,

167

client_interface,

168

byte_arrays=True)

169

self.proxy.connect_to_signal(u"NeedApproval",

169

self.proxy.connect_to_signal("NeedApproval",

170

self.need_approval,

171

client_interface,

172

byte_arrays=True)

173

self.proxy.connect_to_signal(u"Rejected",

173

self.proxy.connect_to_signal("Rejected",

174

self.rejected,

175

client_interface,

176

byte_arrays=True)

178

def property_changed(self, property=None, value=None):

179

super(self, MandosClientWidget).property_changed(property,

180

value)

181

if property == u"ApprovalPending":

181

if property == "ApprovalPending":

182

using_timer(bool(value))

183

184

def using_timer(self, flag):

203

if self.last_checker_failed:

204

self.last_checker_failed = False

205

self.using_timer(False)

206

#self.logger(u'Checker for client %s (command "%s")'

207

# u' was successful'

208

# % (self.properties[u"Name"], command))

206

#self.logger('Checker for client %s (command "%s")'

207

# ' was successful'

208

# % (self.properties["Name"], command))

209

self.update()

210

return

211

# Checker failed

213

self.last_checker_failed = True

214

self.using_timer(True)

215

if os.WIFEXITED(condition):

216

self.logger(u'Checker for client %s (command "%s")'

217

u' failed with exit code %s'

218

% (self.properties[u"Name"], command,

216

self.logger('Checker for client %s (command "%s")'

217

' failed with exit code %s'

218

% (self.properties["Name"], command,

219

os.WEXITSTATUS(condition)))

220

elif os.WIFSIGNALED(condition):

221

self.logger(u'Checker for client %s (command "%s")'

222

u' was killed by signal %s'

223

% (self.properties[u"Name"], command,

221

self.logger('Checker for client %s (command "%s")'

222

' was killed by signal %s'

223

% (self.properties["Name"], command,

224

os.WTERMSIG(condition)))

225

elif os.WCOREDUMP(condition):

226

self.logger(u'Checker for client %s (command "%s")'

227

u' dumped core'

228

% (self.properties[u"Name"], command))

226

self.logger('Checker for client %s (command "%s")'

227

' dumped core'

228

% (self.properties["Name"], command))

229

else:

230

self.logger(u'Checker for client %s completed'

231

u' mysteriously')

230

self.logger('Checker for client %s completed'

231

' mysteriously')

232

self.update()

233

234

def checker_started(self, command):

235

#self.logger(u'Client %s started checker "%s"'

236

# % (self.properties[u"Name"], unicode(command)))

235

#self.logger('Client %s started checker "%s"'

236

# % (self.properties["Name"], unicode(command)))

237

pass

238

239

def got_secret(self):

240

self.last_checker_failed = False

241

self.logger(u'Client %s received its secret'

242

% self.properties[u"Name"])

241

self.logger('Client %s received its secret'

242

% self.properties["Name"])

243

244

def need_approval(self, timeout, default):

245

if not default:

246

message = u'Client %s needs approval within %s seconds'

246

message = 'Client %s needs approval within %s seconds'

247

else:

248

message = u'Client %s will get its secret in %s seconds'

248

message = 'Client %s will get its secret in %s seconds'

249

self.logger(message

250

% (self.properties[u"Name"], timeout/1000))

250

% (self.properties["Name"], timeout/1000))

251

self.using_timer(True)

252

253

def rejected(self, reason):

254

self.logger(u'Client %s was rejected; reason: %s'

255

% (self.properties[u"Name"], reason))

254

self.logger('Client %s was rejected; reason: %s'

255

% (self.properties["Name"], reason))

256

257

def selectable(self):

258

"""Make this a "selectable" widget.

259

This overrides the method from urwid.FlowWidget."""

260

return True

261

262

def rows(self, (maxcol,), focus=False):

262

def rows(self, maxcolrow, focus=False):

263

"""How many rows this widget will occupy might depend on

264

whether we have focus or not.

265

This overrides the method from urwid.FlowWidget"""

266

return self.current_widget(focus).rows((maxcol,), focus=focus)

266

return self.current_widget(focus).rows(maxcolrow, focus=focus)

267

268

def current_widget(self, focus=False):

269

if focus or self.opened:

273

def update(self):

274

"Called when what is visible on the screen should be updated."

275

# How to add standout mode to a style

276

with_standout = { u"normal": u"standout",

277

u"bold": u"bold-standout",

278

u"underline-blink":

279

u"underline-blink-standout",

280

u"bold-underline-blink":

281

u"bold-underline-blink-standout",

276

with_standout = { "normal": "standout",

277

"bold": "bold-standout",

278

"underline-blink":

279

"underline-blink-standout",

280

"bold-underline-blink":

281

"bold-underline-blink-standout",

282

}

283

284

# Rebuild focus and non-focus widgets using current properties

285

286

# Base part of a client. Name!

287

base = (u'%(name)s: '

288

% {u"name": self.properties[u"Name"]})

289

if not self.properties[u"Enabled"]:

290

message = u"DISABLED"

291

elif self.properties[u"ApprovalPending"]:

287

base = ('%(name)s: '

288

% {"name": self.properties["Name"]})

289

if not self.properties["Enabled"]:

290

message = "DISABLED"

291

elif self.properties["ApprovalPending"]:

292

timeout = datetime.timedelta(milliseconds

293

= self.properties

294

[u"ApprovalDelay"])

294

["ApprovalDelay"])

295

last_approval_request = isoformat_to_datetime(

296

self.properties[u"LastApprovalRequest"])

296

self.properties["LastApprovalRequest"])

297

if last_approval_request is not None:

298

timer = timeout - (datetime.datetime.utcnow()

299

- last_approval_request)

300

else:

301

timer = datetime.timedelta()

302

if self.properties[u"ApprovedByDefault"]:

303

message = u"Approval in %s. (d)eny?"

302

if self.properties["ApprovedByDefault"]:

303

message = "Approval in %s. (d)eny?"

304

else:

305

message = u"Denial in %s. (a)pprove?"

305

message = "Denial in %s. (a)pprove?"

306

message = message % unicode(timer).rsplit(".", 1)[0]

307

elif self.last_checker_failed:

308

timeout = datetime.timedelta(milliseconds

309

= self.properties

310

[u"Timeout"])

310

["Timeout"])

311

last_ok = isoformat_to_datetime(

312

max((self.properties[u"LastCheckedOK"]

313

or self.properties[u"Created"]),

314

self.properties[u"LastEnabled"]))

312

max((self.properties["LastCheckedOK"]

313

or self.properties["Created"]),

314

self.properties["LastEnabled"]))

315

timer = timeout - (datetime.datetime.utcnow() - last_ok)

316

message = (u'A checker has failed! Time until client'

317

u' gets disabled: %s'

316

message = ('A checker has failed! Time until client'

317

' gets disabled: %s'

318

% unicode(timer).rsplit(".", 1)[0])

319

else:

320

message = u"enabled"

320

message = "enabled"

321

self._text = "%s%s" % (base, message)

322

323

if not urwid.supports_unicode():

324

self._text = self._text.encode("ascii", "replace")

325

textlist = [(u"normal", self._text)]

325

textlist = [("normal", self._text)]

326

self._text_widget.set_text(textlist)

327

self._focus_text_widget.set_text([(with_standout[text[0]],

328

text[1])

348

if self.delete_hook is not None:

349

self.delete_hook(self)

350

351

def render(self, (maxcol,), focus=False):

351

def render(self, maxcolrow, focus=False):

352

"""Render differently if we have focus.

353

This overrides the method from urwid.FlowWidget"""

354

return self.current_widget(focus).render((maxcol,),

354

return self.current_widget(focus).render(maxcolrow,

355

focus=focus)

356

357

def keypress(self, (maxcol,), key):

357

def keypress(self, maxcolrow, key):

358

"""Handle keys.

359

This overrides the method from urwid.FlowWidget"""

360

if key == u"+":

361

self.proxy.Enable(dbus_interface = client_interface)

362

elif key == u"-":

363

self.proxy.Disable(dbus_interface = client_interface)

364

elif key == u"a":

360

if key == "+":

361

self.proxy.Enable(dbus_interface = client_interface,

362

ignore_reply=True)

363

elif key == "-":

364

self.proxy.Disable(dbus_interface = client_interface,

365

ignore_reply=True)

366

elif key == "a":

365

367

self.proxy.Approve(dbus.Boolean(True, variant_level=1),

366

dbus_interface = client_interface)

367

elif key == u"d":

368

dbus_interface = client_interface,

369

ignore_reply=True)

370

elif key == "d":

368

371

self.proxy.Approve(dbus.Boolean(False, variant_level=1),

369

dbus_interface = client_interface)

370

elif key == u"R" or key == u"_" or key == u"ctrl k":

372

dbus_interface = client_interface,

373

ignore_reply=True)

374

elif key == "R" or key == "_" or key == "ctrl k":

371

375

self.server_proxy_object.RemoveClient(self.proxy

372

.object_path)

373

elif key == u"s":

374

self.proxy.StartChecker(dbus_interface = client_interface)

375

elif key == u"S":

376

self.proxy.StopChecker(dbus_interface = client_interface)

377

elif key == u"C":

378

self.proxy.CheckedOK(dbus_interface = client_interface)

376

.object_path,

377

ignore_reply=True)

378

elif key == "s":

379

self.proxy.StartChecker(dbus_interface = client_interface,

380

ignore_reply=True)

381

elif key == "S":

382

self.proxy.StopChecker(dbus_interface = client_interface,

383

ignore_reply=True)

384

elif key == "C":

385

self.proxy.CheckedOK(dbus_interface = client_interface,

386

ignore_reply=True)

379

387

# xxx

380

# elif key == u"p" or key == "=":

388

# elif key == "p" or key == "=":

381

389

# self.proxy.pause()

382

# elif key == u"u" or key == ":":

390

# elif key == "u" or key == ":":

383

391

# self.proxy.unpause()

384

# elif key == u"RET":

392

# elif key == "RET":

385

393

# self.open()

386

394

else:

387

395

return key

403

411

"down" key presses, thus not allowing any containing widgets to

404

412

use them as an excuse to shift focus away from this widget.

405

413

"""

406

def keypress(self, (maxcol, maxrow), key):

407

ret = super(ConstrainedListBox, self).keypress((maxcol,

408

maxrow), key)

409

if ret in (u"up", u"down"):

414

def keypress(self, maxcolrow, key):

415

ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)

416

if ret in ("up", "down"):

410

417

return

411

418

return ret

412

419

421

428

self.screen = urwid.curses_display.Screen()

422

429

423

430

self.screen.register_palette((

424

(u"normal",

425

u"default", u"default", None),

426

(u"bold",

427

u"default", u"default", u"bold"),

428

(u"underline-blink",

429

u"default", u"default", u"underline"),

430

(u"standout",

431

u"default", u"default", u"standout"),

432

(u"bold-underline-blink",

433

u"default", u"default", (u"bold", u"underline")),

434

(u"bold-standout",

435

u"default", u"default", (u"bold", u"standout")),

436

(u"underline-blink-standout",

437

u"default", u"default", (u"underline", u"standout")),

438

(u"bold-underline-blink-standout",

439

u"default", u"default", (u"bold", u"underline",

440

u"standout")),

431

("normal",

432

"default", "default", None),

433

("bold",

434

"default", "default", "bold"),

435

("underline-blink",

436

"default", "default", "underline"),

437

("standout",

438

"default", "default", "standout"),

439

("bold-underline-blink",

440

"default", "default", ("bold", "underline")),

441

("bold-standout",

442

"default", "default", ("bold", "standout")),

443

("underline-blink-standout",

444

"default", "default", ("underline", "standout")),

445

("bold-underline-blink-standout",

446

"default", "default", ("bold", "underline",

447

"standout")),

441

448

))

442

449

443

450

if urwid.supports_unicode():

444

self.divider = u"─" # \u2500

445

#self.divider = u"━" # \u2501

451

self.divider = "─" # \u2500

452

#self.divider = "━" # \u2501

446

453

else:

447

#self.divider = u"-" # \u002d

448

self.divider = u"_" # \u005f

454

#self.divider = "-" # \u002d

455

self.divider = "_" # \u005f

449

456

450

457

self.screen.start()

451

458

465

472

# This keeps track of whether self.uilist currently has

466

473

# self.logbox in it or not

467

474

self.log_visible = True

468

self.log_wrap = u"any"

475

self.log_wrap = "any"

469

476

470

477

self.rebuild()

471

self.log_message_raw((u"bold",

472

u"Mandos Monitor version " + version))

473

self.log_message_raw((u"bold",

474

u"q: Quit ?: Help"))

478

self.log_message_raw(("bold",

479

"Mandos Monitor version " + version))

480

self.log_message_raw(("bold",

481

"q: Quit ?: Help"))

475

482

476

483

self.busname = domain + '.Mandos'

477

484

self.main_loop = gobject.MainLoop()

478

485

self.bus = dbus.SystemBus()

479

486

mandos_dbus_objc = self.bus.get_object(

480

self.busname, u"/", follow_name_owner_changes=True)

487

self.busname, "/", follow_name_owner_changes=True)

481

488

self.mandos_serv = dbus.Interface(mandos_dbus_objc,

482

489

dbus_interface

483

490

= server_interface)

488

495

mandos_clients = dbus.Dictionary()

489

496

490

497

(self.mandos_serv

491

.connect_to_signal(u"ClientRemoved",

498

.connect_to_signal("ClientRemoved",

492

499

self.find_and_remove_client,

493

500

dbus_interface=server_interface,

494

501

byte_arrays=True))

495

502

(self.mandos_serv

496

.connect_to_signal(u"ClientAdded",

503

.connect_to_signal("ClientAdded",

497

504

self.add_new_client,

498

505

dbus_interface=server_interface,

499

506

byte_arrays=True))

500

507

(self.mandos_serv

501

.connect_to_signal(u"ClientNotFound",

508

.connect_to_signal("ClientNotFound",

502

509

self.client_not_found,

503

510

dbus_interface=server_interface,

504

511

byte_arrays=True))

519

526

path=path)

520

527

521

528

def client_not_found(self, fingerprint, address):

522

self.log_message((u"Client with address %s and fingerprint %s"

523

u" could not be found" % (address,

529

self.log_message(("Client with address %s and fingerprint %s"

530

" could not be found" % (address,

524

531

fingerprint)))

525

532

526

533

def rebuild(self):

542

549

543

550

def log_message(self, message):

544

551

timestamp = datetime.datetime.now().isoformat()

545

self.log_message_raw(timestamp + u": " + message)

552

self.log_message_raw(timestamp + ": " + message)

546

553

547

554

def log_message_raw(self, markup):

548

555

"""Add a log message to the log buffer."""

551

558

and len(self.log) > self.max_log_length):

552

559

del self.log[0:len(self.log)-self.max_log_length-1]

553

560

self.logbox.set_focus(len(self.logbox.body.contents),

554

coming_from=u"above")

561

coming_from="above")

555

562

self.refresh()

556

563

557

564

def toggle_log_display(self):

558

565

"""Toggle visibility of the log buffer."""

559

566

self.log_visible = not self.log_visible

560

567

self.rebuild()

561

#self.log_message(u"Log visibility changed to: "

568

#self.log_message("Log visibility changed to: "

562

569

# + unicode(self.log_visible))

563

570

564

571

def change_log_display(self):

565

572

"""Change type of log display.

566

573

Currently, this toggles wrapping of text lines."""

567

if self.log_wrap == u"clip":

568

self.log_wrap = u"any"

574

if self.log_wrap == "clip":

575

self.log_wrap = "any"

569

576

else:

570

self.log_wrap = u"clip"

577

self.log_wrap = "clip"

571

578

for textwidget in self.log:

572

579

textwidget.set_wrap_mode(self.log_wrap)

573

#self.log_message(u"Wrap mode: " + self.log_wrap)

580

#self.log_message("Wrap mode: " + self.log_wrap)

574

581

575

582

def find_and_remove_client(self, path, name):

576

583

"""Find an client from its object path and remove it.

603

610

if path is None:

604

611

path = client.proxy.object_path

605

612

self.clients_dict[path] = client

606

self.clients.sort(None, lambda c: c.properties[u"Name"])

613

self.clients.sort(None, lambda c: c.properties["Name"])

607

614

self.refresh()

608

615

609

616

def remove_client(self, client, path=None):

640

647

641

648

def process_input(self, source, condition):

642

649

keys = self.screen.get_input()

643

translations = { u"ctrl n": u"down", # Emacs

644

u"ctrl p": u"up", # Emacs

645

u"ctrl v": u"page down", # Emacs

646

u"meta v": u"page up", # Emacs

647

u" ": u"page down", # less

648

u"f": u"page down", # less

649

u"b": u"page up", # less

650

u"j": u"down", # vi

651

u"k": u"up", # vi

650

translations = { "ctrl n": "down", # Emacs

651

"ctrl p": "up", # Emacs

652

"ctrl v": "page down", # Emacs

653

"meta v": "page up", # Emacs

654

" ": "page down", # less

655

"f": "page down", # less

656

"b": "page up", # less

657

"j": "down", # vi

658

"k": "up", # vi

652

659

}

653

660

for key in keys:

654

661

try:

656

663

except KeyError: # :-)

657

664

pass

658

665

659

if key == u"q" or key == u"Q":

666

if key == "q" or key == "Q":

660

667

self.stop()

661

668

break

662

elif key == u"window resize":

669

elif key == "window resize":

663

670

self.size = self.screen.get_cols_rows()

664

671

self.refresh()

665

elif key == u"\f": # Ctrl-L

672

elif key == "\f": # Ctrl-L

666

673

self.refresh()

667

elif key == u"l" or key == u"D":

674

elif key == "l" or key == "D":

668

675

self.toggle_log_display()

669

676

self.refresh()

670

elif key == u"w" or key == u"i":

677

elif key == "w" or key == "i":

671

678

self.change_log_display()

672

679

self.refresh()

673

elif key == u"?" or key == u"f1" or key == u"esc":

680

elif key == "?" or key == "f1" or key == "esc":

674

681

if not self.log_visible:

675

682

self.log_visible = True

676

683

self.rebuild()

677

self.log_message_raw((u"bold",

678

u" ".

679

join((u"q: Quit",

680

u"?: Help",

681

u"l: Log window toggle",

682

u"TAB: Switch window",

683

u"w: Wrap (log)"))))

684

self.log_message_raw((u"bold",

685

u" "

686

.join((u"Clients:",

687

u"+: Enable",

688

u"-: Disable",

689

u"R: Remove",

690

u"s: Start new checker",

691

u"S: Stop checker",

692

u"C: Checker OK",

693

u"a: Approve",

694

u"d: Deny"))))

684

self.log_message_raw(("bold",

685

" ".

686

join(("q: Quit",

687

"?: Help",

688

"l: Log window toggle",

689

"TAB: Switch window",

690

"w: Wrap (log)"))))

691

self.log_message_raw(("bold",

692

" "

693

.join(("Clients:",

694

"+: Enable",

695

"-: Disable",

696

"R: Remove",

697

"s: Start new checker",

698

"S: Stop checker",

699

"C: Checker OK",

700

"a: Approve",

701

"d: Deny"))))

695

702

self.refresh()

696

elif key == u"tab":

703

elif key == "tab":

697

704

if self.topwidget.get_focus() is self.logbox:

698

705

self.topwidget.set_focus(0)

699

706

else:

700

707

self.topwidget.set_focus(self.logbox)

701

708

self.refresh()

702

#elif (key == u"end" or key == u"meta >" or key == u"G"

703

# or key == u">"):

709

#elif (key == "end" or key == "meta >" or key == "G"

710

# or key == ">"):

704

711

# pass # xxx end-of-buffer

705

#elif (key == u"home" or key == u"meta <" or key == u"g"

706

# or key == u"<"):

712

#elif (key == "home" or key == "meta <" or key == "g"

713

# or key == "<"):

707

714

# pass # xxx beginning-of-buffer

708

#elif key == u"ctrl e" or key == u"$":

715

#elif key == "ctrl e" or key == "$":

709

716

# pass # xxx move-end-of-line

710

#elif key == u"ctrl a" or key == u"^":

717

#elif key == "ctrl a" or key == "^":

711

718

# pass # xxx move-beginning-of-line

712

#elif key == u"ctrl b" or key == u"meta (" or key == u"h":

719

#elif key == "ctrl b" or key == "meta (" or key == "h":

713

720

# pass # xxx left

714

#elif key == u"ctrl f" or key == u"meta )" or key == u"l":

721

#elif key == "ctrl f" or key == "meta )" or key == "l":

715

722

# pass # xxx right

716

#elif key == u"a":

723

#elif key == "a":

717

724

# pass # scroll up log

718

#elif key == u"z":

725

#elif key == "z":

719

726

# pass # scroll down log

720

727

elif self.topwidget.selectable():

721

728

self.topwidget.keypress(self.size, key)