mirror of
https://github.com/virt-manager/virt-manager.git
synced 2024-12-23 17:34:21 +03:00
ffab20782f
Like ECHOPROMPT which is needed for libssh hosts file management. Sync the two implementations a bit
219 lines
6.1 KiB
Python
219 lines
6.1 KiB
Python
# Copyright (C) 2012-2013 Red Hat, Inc.
|
|
# Copyright (C) 2012 Cole Robinson <crobinso@redhat.com>
|
|
#
|
|
# This work is licensed under the GNU GPLv2 or later.
|
|
# See the COPYING file in the top-level directory.
|
|
|
|
import collections
|
|
import logging
|
|
import os
|
|
import re
|
|
import time
|
|
|
|
from gi.repository import GLib
|
|
from gi.repository import Gio
|
|
|
|
import libvirt
|
|
|
|
|
|
def do_we_have_session():
|
|
pid = os.getpid()
|
|
try:
|
|
bus = Gio.bus_get_sync(Gio.BusType.SYSTEM, None)
|
|
except Exception:
|
|
logging.exception("Error getting system bus handle")
|
|
return
|
|
|
|
# Check systemd
|
|
try:
|
|
manager = Gio.DBusProxy.new_sync(bus, 0, None,
|
|
"org.freedesktop.login1",
|
|
"/org/freedesktop/login1",
|
|
"org.freedesktop.login1.Manager", None)
|
|
|
|
ret = manager.GetSessionByPID("(u)", pid)
|
|
logging.debug("Found login1 session=%s", ret)
|
|
return True
|
|
except Exception:
|
|
logging.exception("Couldn't connect to logind")
|
|
|
|
return False
|
|
|
|
|
|
def creds_dialog(creds, cbdata):
|
|
"""
|
|
Thread safe wrapper for libvirt openAuth user/pass callback
|
|
"""
|
|
retipc = []
|
|
|
|
def wrapper(fn, creds, cbdata):
|
|
try:
|
|
ret = fn(creds, cbdata)
|
|
except Exception:
|
|
logging.exception("Error from creds dialog")
|
|
ret = -1
|
|
retipc.append(ret)
|
|
|
|
GLib.idle_add(wrapper, _creds_dialog_main, creds, cbdata)
|
|
|
|
while not retipc:
|
|
time.sleep(.1)
|
|
|
|
return retipc[0]
|
|
|
|
|
|
def _creds_dialog_main(creds, cbdata):
|
|
"""
|
|
Libvirt openAuth callback for username/password credentials
|
|
"""
|
|
_conn = cbdata
|
|
from gi.repository import Gtk
|
|
|
|
dialog = Gtk.Dialog(_("Authentication required"), None, 0,
|
|
(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL,
|
|
Gtk.STOCK_OK, Gtk.ResponseType.OK))
|
|
label = []
|
|
entry = []
|
|
|
|
box = Gtk.Table(2, len(creds))
|
|
box.set_border_width(6)
|
|
box.set_row_spacings(6)
|
|
box.set_col_spacings(12)
|
|
|
|
def _on_ent_activate(ent):
|
|
idx = entry.index(ent)
|
|
|
|
if idx < len(entry) - 1:
|
|
entry[idx + 1].grab_focus()
|
|
else:
|
|
dialog.response(Gtk.ResponseType.OK)
|
|
|
|
row = 0
|
|
for cred in creds:
|
|
# Libvirt virConnectCredential
|
|
credtype, prompt, _challenge, _defresult, _result = cred
|
|
noecho = credtype in [
|
|
libvirt.VIR_CRED_PASSPHRASE, libvirt.VIR_CRED_NOECHOPROMPT]
|
|
if not prompt:
|
|
logging.error("No prompt for auth credtype=%s", credtype)
|
|
return -1
|
|
|
|
prompt += ": "
|
|
text_label = Gtk.Label(label=prompt)
|
|
text_label.set_alignment(0.0, 0.5)
|
|
label.append(text_label)
|
|
|
|
ent = Gtk.Entry()
|
|
if noecho:
|
|
ent.set_visibility(False)
|
|
ent.connect("activate", _on_ent_activate)
|
|
entry.append(ent)
|
|
|
|
box.attach(label[row], 0, 1, row, row + 1,
|
|
Gtk.AttachOptions.FILL, 0, 0, 0)
|
|
box.attach(entry[row], 1, 2, row, row + 1,
|
|
Gtk.AttachOptions.FILL, 0, 0, 0)
|
|
row = row + 1
|
|
|
|
vbox = dialog.get_child()
|
|
vbox.add(box)
|
|
|
|
dialog.show_all()
|
|
res = dialog.run()
|
|
dialog.hide()
|
|
|
|
if res == Gtk.ResponseType.OK:
|
|
row = 0
|
|
for cred in creds:
|
|
cred[4] = entry[row].get_text()
|
|
row = row + 1
|
|
ret = 0
|
|
else:
|
|
ret = -1
|
|
|
|
dialog.destroy()
|
|
return ret
|
|
|
|
|
|
def acquire_tgt():
|
|
"""
|
|
Try to get kerberos ticket if openAuth seems to require it
|
|
"""
|
|
logging.debug("In acquire tgt.")
|
|
try:
|
|
bus = Gio.bus_get_sync(Gio.BusType.SESSION, None)
|
|
ka = Gio.DBusProxy.new_sync(bus, 0, None,
|
|
"org.gnome.KrbAuthDialog",
|
|
"/org/gnome/KrbAuthDialog",
|
|
"org.freedesktop.KrbAuthDialog", None)
|
|
ret = ka.acquireTgt("(s)", "")
|
|
except Exception as e:
|
|
logging.info("Cannot acquire tgt %s", str(e))
|
|
ret = False
|
|
return ret
|
|
|
|
|
|
def connect_error(conn, errmsg, tb, warnconsole):
|
|
"""
|
|
Format connection error message
|
|
"""
|
|
errmsg = errmsg.strip(" \n")
|
|
tb = tb.strip(" \n")
|
|
hint = ""
|
|
show_errmsg = True
|
|
config = conn.config
|
|
|
|
if conn.is_remote():
|
|
logging.debug("connect_error: conn transport=%s",
|
|
conn.get_uri_transport())
|
|
if re.search(r"nc: .* -- 'U'", tb):
|
|
hint += _("The remote host requires a version of netcat/nc "
|
|
"which supports the -U option.")
|
|
show_errmsg = False
|
|
elif (conn.get_uri_transport() == "ssh" and
|
|
re.search(r"ssh-askpass", tb)):
|
|
|
|
askpass = (config.askpass_package and
|
|
config.askpass_package[0] or
|
|
"openssh-askpass")
|
|
hint += _("You need to install %s or "
|
|
"similar to connect to this host.") % askpass
|
|
show_errmsg = False
|
|
else:
|
|
hint += _("Verify that the 'libvirtd' daemon is running "
|
|
"on the remote host.")
|
|
|
|
elif conn.is_xen():
|
|
hint += _("Verify that:\n"
|
|
" - A Xen host kernel was booted\n"
|
|
" - The Xen service has been started")
|
|
|
|
else:
|
|
if warnconsole:
|
|
hint += _("Could not detect a local session: if you are "
|
|
"running virt-manager over ssh -X or VNC, you "
|
|
"may not be able to connect to libvirt as a "
|
|
"regular user. Try running as root.")
|
|
show_errmsg = False
|
|
elif re.search(r"libvirt-sock", tb):
|
|
hint += _("Verify that the 'libvirtd' daemon is running.")
|
|
show_errmsg = False
|
|
|
|
msg = _("Unable to connect to libvirt %s." % conn.get_uri())
|
|
if show_errmsg:
|
|
msg += "\n\n%s" % errmsg
|
|
if hint:
|
|
msg += "\n\n%s" % hint
|
|
|
|
msg = msg.strip("\n")
|
|
details = msg
|
|
details += "\n\n"
|
|
details += "Libvirt URI is: %s\n\n" % conn.get_uri()
|
|
details += tb
|
|
|
|
title = _("Virtual Machine Manager Connection Failure")
|
|
|
|
ConnectError = collections.namedtuple("ConnectError",
|
|
["msg", "details", "title"])
|
|
return ConnectError(msg, details, title)
|