2022-01-11 16:49:26 +00:00
|
|
|
|
#!/usr/bin/env python3
|
2017-06-06 08:40:32 +00:00
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
2017-09-03 16:15:48 +00:00
|
|
|
|
import base64
|
2022-01-11 16:49:26 +00:00
|
|
|
|
import subprocess
|
|
|
|
|
import configparser
|
|
|
|
|
import gi
|
|
|
|
|
gi.require_version("Gtk", "3.0")
|
|
|
|
|
from gi.repository import Gtk, Gdk, GLib
|
2017-09-03 16:15:48 +00:00
|
|
|
|
from os import path
|
2017-06-06 08:40:32 +00:00
|
|
|
|
|
|
|
|
|
# ===== Connection settings: ======
|
|
|
|
|
# ===== freerdp-x11:
|
|
|
|
|
|
2017-09-03 16:15:48 +00:00
|
|
|
|
string_l = "xfreerdp /drive:share,/media /bpp:15 /jpeg /cert-ignore /v:192.168.1.128"
|
|
|
|
|
string_r = "xfreerdp /drive:share,/media /bpp:8 /jpeg /cert-ignore /v:external.domain:3389"
|
2017-06-06 08:40:32 +00:00
|
|
|
|
|
|
|
|
|
# ===== rdesktop:
|
2017-09-03 16:15:48 +00:00
|
|
|
|
#string_l = "rdesktop -z -a 15 -k en-us -r disk:share=/media 192.168.1.128"
|
|
|
|
|
#string_r = "rdesktop -z -a 8 -k en-us -r disk:share=/media external.domain:3389"
|
|
|
|
|
|
|
|
|
|
conf_file = path.expanduser("~") + '/.config/pygtkrdesktop.conf'
|
2017-06-06 08:40:32 +00:00
|
|
|
|
|
2022-01-11 16:49:26 +00:00
|
|
|
|
class Table(Gtk.Window):
|
2017-06-06 08:40:32 +00:00
|
|
|
|
|
2017-09-03 16:15:48 +00:00
|
|
|
|
def Connect(self, widget, data=None):
|
2017-06-06 08:40:32 +00:00
|
|
|
|
name = login_entry.get_text()
|
|
|
|
|
password = password_entry.get_text()
|
2017-09-03 16:15:48 +00:00
|
|
|
|
SaveLogin(name, password)
|
2017-06-06 08:40:32 +00:00
|
|
|
|
if ( data == "local" ):
|
|
|
|
|
string = string_l + " /u:" + name + " /p:" + password
|
|
|
|
|
# ===== rdesktop:
|
|
|
|
|
# string = string_l + " -u " + name + " -p " + password
|
|
|
|
|
if ( data == "remote" ):
|
|
|
|
|
string = string_r + " /u:" + name + " /p:" + password
|
|
|
|
|
# ===== rdesktop:
|
|
|
|
|
# string = string_r + " -u " + name + " -p " + password
|
2017-09-03 16:15:48 +00:00
|
|
|
|
|
|
|
|
|
if fullscreen_check.get_active():
|
|
|
|
|
string += " /f"
|
|
|
|
|
# ===== rdesktop:
|
|
|
|
|
# string += " -f"
|
|
|
|
|
else:
|
|
|
|
|
string += " /w:800 /h:600"
|
|
|
|
|
# ===== rdesktop:
|
|
|
|
|
# string += ""
|
|
|
|
|
|
2017-06-06 08:40:32 +00:00
|
|
|
|
p = subprocess.Popen(string, shell=True, stderr=subprocess.PIPE, )
|
|
|
|
|
streamdata = p.communicate()[0]
|
|
|
|
|
rc = p.returncode
|
|
|
|
|
|
|
|
|
|
if ( (rc == 0) or (rc == 62)):
|
2022-01-11 16:49:26 +00:00
|
|
|
|
Gtk.main_quit()
|
2017-06-06 08:40:32 +00:00
|
|
|
|
else:
|
|
|
|
|
if (rc == 132):
|
|
|
|
|
status_entry_label.set_markup('<b><span color="red">Error: wrong login or password</span></b>');
|
|
|
|
|
else:
|
|
|
|
|
status_entry_label.set_markup('<b><span color="red">Error: check your connection</span></b>');
|
|
|
|
|
|
|
|
|
|
# Uncomment to check errorcode:
|
|
|
|
|
# print "Error code", rc
|
|
|
|
|
|
2017-09-03 16:15:48 +00:00
|
|
|
|
def LoginEnterPressed(self, widget, event):
|
2022-01-11 16:49:26 +00:00
|
|
|
|
if Gdk.keyval_name(event.keyval) == 'Return':
|
2017-09-03 16:15:48 +00:00
|
|
|
|
password_entry.grab_focus()
|
|
|
|
|
return True
|
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
def PasswordEnterPressed(self, widget, event):
|
2022-01-11 16:49:26 +00:00
|
|
|
|
if Gdk.keyval_name(event.keyval) == 'Return':
|
2017-09-03 16:15:48 +00:00
|
|
|
|
self.Connect(widget, "local")
|
|
|
|
|
return True
|
|
|
|
|
return False
|
|
|
|
|
|
2017-06-06 08:40:32 +00:00
|
|
|
|
def delete_event(self, widget, event, data=None):
|
2022-01-11 16:49:26 +00:00
|
|
|
|
Gtk.main_quit()
|
2017-06-06 08:40:32 +00:00
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
def __init__(self):
|
2022-01-11 16:49:26 +00:00
|
|
|
|
super().__init__(title="pyGtkRDP")
|
|
|
|
|
self.window = Gtk.Window()
|
2017-06-06 08:40:32 +00:00
|
|
|
|
self.window.set_title("Connection...")
|
|
|
|
|
self.window.connect("delete_event", self.delete_event)
|
|
|
|
|
self.window.set_border_width(10)
|
2022-01-11 16:49:26 +00:00
|
|
|
|
table = Gtk.Table(n_rows=9, n_columns=2, homogeneous=True)
|
2017-06-06 08:40:32 +00:00
|
|
|
|
self.window.add(table)
|
|
|
|
|
|
2022-01-11 16:49:26 +00:00
|
|
|
|
login_entry_label = Gtk.Label (label="Username:")
|
2017-06-06 08:40:32 +00:00
|
|
|
|
table.attach(login_entry_label, 0, 2, 0, 1)
|
|
|
|
|
login_entry_label.show()
|
|
|
|
|
|
|
|
|
|
login_entry.set_max_length(20)
|
|
|
|
|
table.attach(login_entry, 0, 2, 1, 2)
|
2017-09-03 16:15:48 +00:00
|
|
|
|
if conf.has_option("login", "saved"):
|
|
|
|
|
login_entry.set_text(conf.get("login", "saved"))
|
2017-06-06 08:40:32 +00:00
|
|
|
|
login_entry.show()
|
2017-09-03 16:15:48 +00:00
|
|
|
|
login_entry.connect('key-press-event', self.LoginEnterPressed)
|
2017-06-06 08:40:32 +00:00
|
|
|
|
|
2022-01-11 16:49:26 +00:00
|
|
|
|
password_entry_label = Gtk.Label (label="Password:")
|
2017-06-06 08:40:32 +00:00
|
|
|
|
table.attach(password_entry_label, 0, 2, 2, 3)
|
|
|
|
|
password_entry_label.show()
|
|
|
|
|
|
|
|
|
|
password_entry.set_max_length(30)
|
|
|
|
|
password_entry.set_visibility(False)
|
|
|
|
|
table.attach(password_entry, 0, 2, 3, 4)
|
|
|
|
|
password_entry.show()
|
2017-09-03 16:15:48 +00:00
|
|
|
|
if conf.has_option("login", "saved"):
|
|
|
|
|
password_entry.grab_focus()
|
|
|
|
|
if conf.has_option("login", "password"):
|
2022-01-11 16:49:26 +00:00
|
|
|
|
password_entry.set_text(base64.b32decode(base64.b16decode(base64.b64decode
|
|
|
|
|
(conf.get("login", "password")))).decode('utf-8'))
|
2017-09-03 16:15:48 +00:00
|
|
|
|
password_entry.connect('key-press-event', self.PasswordEnterPressed)
|
2017-06-06 08:40:32 +00:00
|
|
|
|
|
|
|
|
|
table.attach(status_entry_label, 0, 2, 4, 5)
|
|
|
|
|
status_entry_label.show()
|
|
|
|
|
|
2017-09-03 16:15:48 +00:00
|
|
|
|
fullscreen_check.set_tooltip_text("If checked – remote connection will be fullscreen, if no – windowed. Press CTRL+ALT+ENTER to toggle fullscreen.")
|
|
|
|
|
if conf.has_option("login", "fullscreen"):
|
|
|
|
|
if conf.get("login", "fullscreen") == 'True':
|
|
|
|
|
fullscreen_check.set_active(True)
|
|
|
|
|
else:
|
|
|
|
|
fullscreen_check.set_active(False)
|
|
|
|
|
else:
|
|
|
|
|
fullscreen_check.set_active(True)
|
|
|
|
|
table.attach(fullscreen_check, 0, 2, 5, 6)
|
|
|
|
|
fullscreen_check.show()
|
|
|
|
|
|
|
|
|
|
password_check.set_tooltip_text("If checked – passwrod will be saved in file. It's unsafe feature, please, be careful.")
|
|
|
|
|
if conf.has_option("login", "savepassword"):
|
|
|
|
|
if conf.get("login", "savepassword") == 'True':
|
|
|
|
|
password_check.set_active(True)
|
|
|
|
|
else:
|
|
|
|
|
password_check.set_active(False)
|
|
|
|
|
else:
|
|
|
|
|
password_check.set_active(False)
|
|
|
|
|
table.attach(password_check, 0, 2, 6, 7)
|
|
|
|
|
password_check.show()
|
|
|
|
|
|
2022-01-11 16:49:26 +00:00
|
|
|
|
local_button = Gtk.Button(label="Local connection")
|
2017-09-03 16:15:48 +00:00
|
|
|
|
local_button.set_tooltip_text("Connection via LAN (using internal IP.")
|
|
|
|
|
local_button.connect("clicked", self.Connect, "local")
|
|
|
|
|
table.attach(local_button, 0, 1, 7, 8)
|
2017-06-06 08:40:32 +00:00
|
|
|
|
local_button.show()
|
|
|
|
|
|
2022-01-11 16:49:26 +00:00
|
|
|
|
remote_button = Gtk.Button(label="External connection")
|
2017-09-03 16:15:48 +00:00
|
|
|
|
remote_button.set_tooltip_text("Connection via WAN (using external IP).")
|
|
|
|
|
remote_button.connect("clicked", self.Connect, "remote")
|
|
|
|
|
table.attach(remote_button, 1, 2, 7, 8)
|
2017-06-06 08:40:32 +00:00
|
|
|
|
remote_button.show()
|
|
|
|
|
|
2022-01-11 16:49:26 +00:00
|
|
|
|
button = Gtk.Button(label="Exit")
|
|
|
|
|
button.connect("clicked", lambda w: Gtk.main_quit())
|
2017-09-03 16:15:48 +00:00
|
|
|
|
table.attach(button, 0, 2, 8, 9)
|
2017-06-06 08:40:32 +00:00
|
|
|
|
button.show()
|
|
|
|
|
|
|
|
|
|
table.show()
|
|
|
|
|
self.window.show()
|
|
|
|
|
|
2017-09-03 16:15:48 +00:00
|
|
|
|
def SaveLogin(login, password):
|
|
|
|
|
if not conf.has_section("login"):
|
|
|
|
|
conf.add_section("login")
|
|
|
|
|
conf.set("login", "saved", login)
|
|
|
|
|
conf.set("login", "fullscreen", fullscreen_check.get_active())
|
|
|
|
|
conf.set("login", "savepassword", password_check.get_active())
|
|
|
|
|
if password_check.get_active():
|
2022-01-11 16:49:26 +00:00
|
|
|
|
password = password.encode()
|
|
|
|
|
password = base64.b64encode(base64.b16encode(
|
|
|
|
|
base64.b32encode(password)))
|
|
|
|
|
password = password.decode('utf-8')
|
|
|
|
|
conf.set("login", "password", password)
|
2017-09-03 16:15:48 +00:00
|
|
|
|
else:
|
|
|
|
|
conf.remove_option("login", "password")
|
|
|
|
|
conf.write(open(conf_file, "w"))
|
|
|
|
|
|
2017-06-06 08:40:32 +00:00
|
|
|
|
def main():
|
2022-01-11 16:49:26 +00:00
|
|
|
|
Gtk.main()
|
2017-06-06 08:40:32 +00:00
|
|
|
|
return 0
|
|
|
|
|
|
2022-01-11 16:49:26 +00:00
|
|
|
|
login_entry = Gtk.Entry()
|
|
|
|
|
password_entry = Gtk.Entry()
|
|
|
|
|
status_entry_label = Gtk.Label (label="CTRL+ALT+ENTER - toggle fullscreen")
|
|
|
|
|
fullscreen_check = Gtk.CheckButton (label="Fullscreen mode")
|
|
|
|
|
password_check = Gtk.CheckButton (label="Save password (unsafe)")
|
2017-06-06 08:40:32 +00:00
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
2022-01-11 16:49:26 +00:00
|
|
|
|
conf = configparser.RawConfigParser()
|
2017-09-03 16:15:48 +00:00
|
|
|
|
if not path.exists(conf_file):
|
|
|
|
|
open(conf_file, 'w')
|
|
|
|
|
conf.read(conf_file)
|
2017-06-06 08:40:32 +00:00
|
|
|
|
Table()
|
|
|
|
|
main()
|