1
0
mirror of https://github.com/systemd/systemd.git synced 2024-11-08 11:27:32 +03:00
systemd/src/gnome-ask-password-agent.vala

267 lines
8.4 KiB
Vala

/***
This file is part of systemd.
Copyright 2010 Lennart Poettering
systemd is free software; you can redistribute it and/or modify it
under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
systemd is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
General Public License for more details.
You should have received a copy of the GNU General Public License
along with systemd; If not, see <http://www.gnu.org/licenses/>.
***/
using Gtk;
using GLib;
using Linux;
using Posix;
using Notify;
[CCode (cheader_filename = "time.h")]
extern int clock_gettime(int id, out timespec ts);
public class PasswordDialog : Dialog {
public Entry entry;
public PasswordDialog(string message, string icon) {
set_title("System Password");
set_has_separator(false);
set_border_width(8);
set_default_response(ResponseType.OK);
set_icon_name(icon);
add_button(Stock.CANCEL, ResponseType.CANCEL);
add_button(Stock.OK, ResponseType.OK);
Container content = (Container) get_content_area();
Box hbox = new HBox(false, 16);
hbox.set_border_width(8);
content.add(hbox);
Image image = new Image.from_icon_name(icon, IconSize.DIALOG);
hbox.pack_start(image, false, false);
Box vbox = new VBox(false, 8);
hbox.pack_start(vbox, true, true);
Label label = new Label(message);
vbox.pack_start(label, false, false);
entry = new Entry();
entry.set_visibility(false);
entry.set_activates_default(true);
vbox.pack_start(entry, false, false);
entry.activate.connect(on_entry_activated);
show_all();
}
public void on_entry_activated() {
response(ResponseType.OK);
}
}
public class MyStatusIcon : StatusIcon {
File directory;
File current;
FileMonitor file_monitor;
string message;
string icon;
string socket;
PasswordDialog password_dialog;
public MyStatusIcon() throws GLib.Error {
GLib.Object(icon_name : "dialog-password");
set_title("System Password Request");
directory = File.new_for_path("/run/systemd/ask-password/");
file_monitor = directory.monitor_directory(0);
file_monitor.changed.connect(file_monitor_changed);
current = null;
look_for_password();
activate.connect(status_icon_activate);
}
void file_monitor_changed(GLib.File file, GLib.File? other_file, GLib.FileMonitorEvent event_type) {
if (!file.get_basename().has_prefix("ask."))
return;
if (event_type == FileMonitorEvent.CREATED ||
event_type == FileMonitorEvent.DELETED) {
try {
look_for_password();
} catch (Error e) {
show_error(e.message);
}
}
}
void look_for_password() throws GLib.Error {
if (current != null) {
if (!current.query_exists()) {
current = null;
if (password_dialog != null)
password_dialog.response(ResponseType.REJECT);
}
}
if (current == null) {
FileEnumerator enumerator = directory.enumerate_children("standard::name", FileQueryInfoFlags.NOFOLLOW_SYMLINKS);
FileInfo i;
while ((i = enumerator.next_file()) != null) {
if (!i.get_name().has_prefix("ask."))
continue;
current = directory.get_child(i.get_name());
if (load_password())
break;
current = null;
}
}
if (current == null)
set_visible(false);
}
bool load_password() throws GLib.Error {
KeyFile key_file = new KeyFile();
try {
timespec ts;
key_file.load_from_file(current.get_path(), KeyFileFlags.NONE);
string not_after_as_string = key_file.get_string("Ask", "NotAfter");
clock_gettime(1, out ts);
uint64 now = (ts.tv_sec * 1000000) + (ts.tv_nsec / 1000);
uint64 not_after;
if (not_after_as_string.scanf("%llu", out not_after) != 1)
return false;
if (not_after > 0 && not_after < now)
return false;
socket = key_file.get_string("Ask", "Socket");
} catch (GLib.Error e) {
return false;
}
try {
message = key_file.get_string("Ask", "Message").compress();
} catch (GLib.Error e) {
message = "Please Enter System Password!";
}
set_tooltip_text(message);
try {
icon = key_file.get_string("Ask", "Icon");
} catch (GLib.Error e) {
icon = "dialog-password";
}
set_from_icon_name(icon);
set_visible(true);
Notification n = new Notification(title, message, icon);
n.set_timeout(5000);
n.show();
return true;
}
void status_icon_activate() {
if (current == null)
return;
if (password_dialog != null) {
password_dialog.present();
return;
}
password_dialog = new PasswordDialog(message, icon);
int result = password_dialog.run();
string password = password_dialog.entry.get_text();
password_dialog.destroy();
password_dialog = null;
if (result == ResponseType.REJECT ||
result == ResponseType.DELETE_EVENT)
return;
int to_process;
try {
Process.spawn_async_with_pipes(
null,
{ "/usr/bin/pkexec", "/lib/systemd/systemd-reply-password", result == ResponseType.OK ? "1" : "0", socket },
null,
0,
null,
null,
out to_process,
null,
null);
OutputStream stream = new UnixOutputStream(to_process, true);
#if VALA_0_12
stream.write(password.data, null);
#else
stream.write(password, password.length, null);
#endif
} catch (Error e) {
show_error(e.message);
}
}
}
static const OptionEntry entries[] = {
{ null }
};
void show_error(string e) {
var m = new MessageDialog(null, 0, MessageType.ERROR, ButtonsType.CLOSE, "%s", e);
m.run();
m.destroy();
}
int main(string[] args) {
try {
Gtk.init_with_args(ref args, "[OPTION...]", entries, "systemd-ask-password-agent");
Notify.init("Password Agent");
MyStatusIcon i = new MyStatusIcon();
Gtk.main();
} catch (GLib.Error e) {
show_error(e.message);
}
return 0;
}