class Connection:
id = None
name = None
host = None
port = None
user = None
password = None
use_key = False
key_path = None
tunnels = None
model = None
def __init__(self):
self.tunnels = Tunnels()
def load(self):
# Create the Password Manager
password_manager = PasswordManager()
# Create the DBConnection
database = DBConnection()
# Create the SQL
sql = "SELECT * FROM connections WHERE id_connection = {}"
# Bind the value
sql = sql.format(self.id)
# Execute the query
rows = database.select_query(sql)
# Set the attrs
for row in rows:
self.name = row['name']
self.host = row['host']
self.port = row['port']
self.user = row['user']
self.password = password_manager.decrypt_password(row['passwd'])
self.use_key = True if int(row['use_key']) == 1 else False
self.key_path = row['key_path']
def load_tunnels(self):
# Clear the tunnels list form the collection
self.tunnels.clear_tunnels()
# Create the DBConnection
database = DBConnection()
# Create the SQL
sql = "SELECT * FROM tunnels WHERE id_connection = {}"
# Bind the value
sql = sql.format(self.id)
# Execute the query
rows = database.select_query(sql)
# Set the attrs
for row in rows:
tunnel = Tunnel()
tunnel.id = row['id_tunnel']
tunnel.id_connection = self.id
tunnel.local_port = row['local_port']
tunnel.address = row['address']
tunnel.remote_port = row['remote_port']
# Add on Tunnels
self.tunnels.add_tunnel(tunnel)
def save(self):
# Create the Password Manager
password_manager = PasswordManager()
# Create the DBConnection
database = DBConnection()
# Create the SQL
if self.id is None:
sql = "INSERT INTO connections (name, host, port, user, passwd, use_key, key_path)" \
" VALUES ('{}', '{}', {}, '{}', '{}', {}, '{}');"
else:
sql = "UPDATE connections SET name = '{}', host = '{}', port = {}, user = '******', passwd = '{}'," \
" use_key = {}, key_path = '{}' where id_connection = {}"
# Bind the values
sql = sql.format(
self.name,
self.host,
self.port,
self.user,
password_manager.encrypt_password(self.password),
1 if self.use_key else 0,
self.key_path,
self.id
)
# Execute the SQL
if database.execute_query(sql) > 0:
# Get the last row inserted
if self.id is None:
self.id = database.cursor.lastrowid
# Return true telling it's OK
return True
else:
# Return False for fail
return False
def get_tunnels(self):
return self.tunnels
def delete_tunnels(self):
# Check if have the connection ID
if self.id is not None:
# Create the DBConnection
database = DBConnection()
# Create the SQL
sql = "DELETE FROM tunnels WHERE id_connection = {}"
# Bind the value
sql = sql.format(self.id)
# Execute the query
if database.execute_query(sql) > 0:
# Return true telling it's OK
return True
else:
# Return False
return False
def delete(self):
# Check if have the connection ID
if self.id is not None:
# Create the DBConnection
database = DBConnection()
# Create the SQL
sql = "DELETE FROM connections WHERE id_connection = {}"
# Bind the value
sql = sql.format(self.id)
# Execute the query
if database.execute_query(sql) > 0:
# Return true telling it's OK
return True
else:
# Return False
return False
def get_model(self):
# Create the Model
self.model = Gtk.ListStore(str, str)
self.model.append(["Name", self.name])
self.model.append(["Host", self.host])
self.model.append(["Port", str(self.port)])
self.model.append(["User", self.user])
self.model.append(["Use Key", "Yes" if self.use_key else "No"])
self.model.append(["Key Path", self.key_path])
# Return the Model
return self.model
def generate_ssh_command(self):
# Get the settings
# Check if uses key
if self.use_key is True:
command = "/usr/bin/ssh {}@{} -i {}"
command = command.format(self.user, self.host, self.key_path)
else:
command = "/usr/bin/sshpass -p{} /usr/bin/ssh {}@{}"
command = command.format(self.password, self.user, self.host)
# Format other parameters
command += " -p {}"
command = command.format(self.port)
# Strict key checking
command += " -o StrictHostKeyChecking=no"
# Get the settings and apply here
settings = Settings()
settings.load()
# X11 forwarding
if settings.x11_forward == 1:
command += " -X"
if settings.request_compression == 1:
command += " -C"
if settings.force_ipv4 == 1:
command += " -4"
if settings.force_ipv6 == 1:
command += " -6"
# Get the tunnels
self.load_tunnels()
for tunnel in self.tunnels.get_tunnels():
command += " -L {}:{}:{}"
command = command.format(tunnel.local_port, tunnel.address, tunnel.remote_port)
# At the end, return the command
return command
class ConnectionWindowEvents:
window = None
builder = None
refresh_list_callback = None
tunnels = None
index_to_remove = None
connection = None
def __init__(self,
window,
builder,
refresh_list_callback,
connection=None):
# Copy the parameters
self.window = window
self.builder = builder
self.refresh_list_callback = refresh_list_callback
self.tunnels = Tunnels()
# If id_connection is not None, load the connection to Edit
if connection is not None:
self.connection = connection
self.load_connection()
# Connect the event notify::active of the switch key
switch = self.builder.get_object("switch_use_key")
switch.connect("notify::active", self.on_switch_use_key_activate)
def on_switch_use_key_activate(self, switch, active):
# Get the password fields
password = self.builder.get_object("txt_password")
confirm_password = self.builder.get_object("txt_password_confirm")
file_chooser = self.builder.get_object("filechooser_key")
if switch.get_active():
# Disable them
password.set_sensitive(False)
confirm_password.set_sensitive(False)
file_chooser.set_sensitive(True)
else:
# Enable them
password.set_sensitive(True)
confirm_password.set_sensitive(True)
file_chooser.set_sensitive(False)
def on_btn_add_clicked(self, btn):
NewTunnelWindow(self.on_added_tunnel_callback)
def on_tunnels_table_cursor_changed(self, treeview):
# Get the Index
(model, path) = treeview.get_selection().get_selected()
if model is not None and path is not None:
self.index_to_remove = model[path][3]
def on_added_tunnel_callback(self, tunnel):
# Add the tunnel on the list
self.tunnels.add_tunnel(tunnel)
# Add to the table the new model
table = self.builder.get_object("tunnels_table")
# Get and clean the model
model = table.get_model()
if model is not None:
model.clear()
# Set the new model
table.set_model(self.tunnels.get_tunnels_model())
def on_btn_remove_clicked(self, btn):
# Remove the line tunnel from the collection and rebuild the table
self.tunnels.remove_tunnel_at(self.index_to_remove)
# Rebuild the table
table = self.builder.get_object("tunnels_table")
# Get and clean the model
model = table.get_model()
if model is not None:
model.clear()
# Set the new model
table.set_model(self.tunnels.get_tunnels_model())
def on_btn_save_clicked(self, btn):
# First of all, validate the form
if self.validate_form():
# If OK, save the connection
if self.save_connection():
# Run the callback
if self.refresh_list_callback is not None:
# Call the callback to refresh the connection list
self.refresh_list_callback()
# Close the Window
self.window.destroy()
else:
# Show a message Box telling the save fails
MessageBox(
"Fail to save the connection. Check if have duplicate names. Try again"
)
def on_btn_cancel_clicked(self, btn):
self.window.destroy()
def load_connection(self):
# Get all objects
name = self.builder.get_object("txt_name")
host = self.builder.get_object("txt_host")
port = self.builder.get_object("txt_port")
user = self.builder.get_object("txt_user")
switch_key = self.builder.get_object("switch_use_key")
password = self.builder.get_object("txt_password")
confirm_password = self.builder.get_object("txt_password_confirm")
file_chooser = self.builder.get_object("filechooser_key")
table = self.builder.get_object("tunnels_table")
# Set the values
name.set_text(self.connection.name)
host.set_text(self.connection.host)
port.set_text(str(self.connection.port))
user.set_text(self.connection.user)
switch_key.set_active(self.connection.use_key)
password.set_text(self.connection.password)
confirm_password.set_text(self.connection.password)
if self.connection.key_path != 'None': # On SQLITE is saved as str 'None'
file_chooser.set_filename(self.connection.key_path)
# Check if use key is checked, and call the function to switch the fields
if self.connection.use_key:
self.on_switch_use_key_activate(switch_key,
self.connection.use_key)
# Also, load the tunnels
self.tunnels = self.connection.get_tunnels()
# Get and clean the model
model = table.get_model()
if model is not None:
model.clear()
# Set the new model
table.set_model(self.tunnels.get_tunnels_model())
def validate_form(self):
# Get all object that need be validated
name = self.builder.get_object("txt_name")
host = self.builder.get_object("txt_host")
port = self.builder.get_object("txt_port")
user = self.builder.get_object("txt_user")
switch_key = self.builder.get_object("switch_use_key")
password = self.builder.get_object("txt_password")
confirm_password = self.builder.get_object("txt_password_confirm")
file_chooser = self.builder.get_object("filechooser_key")
# Check for the name
if len(name.get_text()) == 0:
MessageBox("Name can't be empty!")
return False
# Check for the host
if len(host.get_text()) == 0:
MessageBox("Host can't be empty!")
return False
# Check for the port
port_number = port.get_text()
if len(port_number) == 0:
MessageBox("Port can't be empty!")
return False
elif not port_number.isdigit() or int(port_number) <= 0 or int(
port_number) > 65565:
MessageBox("Port must be between 1 and 65565")
return False
# Check for the user
if len(user.get_text()) == 0:
MessageBox("User can't be empty!")
return False
# If switch key is On, the filechooser must have a file
if switch_key.get_active():
# If file chooser haven't any file, return error
if file_chooser.get_filename() is None:
MessageBox("You must select a key file")
return False
else:
# Check if password is empty
if len(password.get_text()) == 0:
MessageBox("Password can't be empty!")
return False
# Check if the password are the same, if not, return error
if password.get_text() != confirm_password.get_text():
MessageBox("The passwords not match")
return False
# If all pass, return True
return True
def save_connection(self):
# Get all object that need be validated
name = self.builder.get_object("txt_name")
host = self.builder.get_object("txt_host")
port = self.builder.get_object("txt_port")
user = self.builder.get_object("txt_user")
switch_key = self.builder.get_object("switch_use_key")
password = self.builder.get_object("txt_password")
file_chooser = self.builder.get_object("filechooser_key")
# Create a new Connection
if self.connection is None:
self.connection = Connection()
self.connection.name = name.get_text()
self.connection.host = host.get_text()
self.connection.port = port.get_text()
self.connection.user = user.get_text()
self.connection.use_key = switch_key.get_active()
self.connection.key_path = file_chooser.get_filename()
self.connection.password = password.get_text()
# Save the connection
saved = self.connection.save()
# Delete the Tunnels
self.connection.delete_tunnels()
# Now saved the tunnels
for tunnel in self.tunnels.get_tunnels():
# Set the ID connection
tunnel.id_connection = self.connection.id
# Save the Tunnel
tunnel.save()
# return if is saved or not
return saved
class Connection:
id = None
name = None
host = None
port = None
user = None
password = None
use_key = False
key_path = None
tunnels = None
model = None
def __init__(self):
self.tunnels = Tunnels()
def load(self):
# Create the Password Manager
password_manager = PasswordManager()
# Create the DBConnection
database = DBConnection()
# Create the SQL
sql = "SELECT * FROM connections WHERE id_connection = {}"
# Bind the value
sql = sql.format(self.id)
# Execute the query
rows = database.select_query(sql)
# Set the attrs
for row in rows:
self.name = row['name']
self.host = row['host']
self.port = row['port']
self.user = row['user']
self.password = password_manager.decrypt_password(row['passwd'])
self.use_key = True if int(row['use_key']) == 1 else False
self.key_path = row['key_path']
def load_tunnels(self):
# Clear the tunnels list form the collection
self.tunnels.clear_tunnels()
# Create the DBConnection
database = DBConnection()
# Create the SQL
sql = "SELECT * FROM tunnels WHERE id_connection = {}"
# Bind the value
sql = sql.format(self.id)
# Execute the query
rows = database.select_query(sql)
# Set the attrs
for row in rows:
tunnel = Tunnel()
tunnel.id = row['id_tunnel']
tunnel.id_connection = self.id
tunnel.local_port = row['local_port']
tunnel.address = row['address']
tunnel.remote_port = row['remote_port']
# Add on Tunnels
self.tunnels.add_tunnel(tunnel)
def save(self):
# Create the Password Manager
password_manager = PasswordManager()
# Create the DBConnection
database = DBConnection()
# Create the SQL
if self.id is None:
sql = "INSERT INTO connections (name, host, port, user, passwd, use_key, key_path)" \
" VALUES ('{}', '{}', {}, '{}', '{}', {}, '{}');"
else:
sql = "UPDATE connections SET name = '{}', host = '{}', port = {}, user = '******', passwd = '{}'," \
" use_key = {}, key_path = '{}' where id_connection = {}"
# Bind the values
sql = sql.format(self.name, self.host, self.port, self.user,
password_manager.encrypt_password(self.password),
1 if self.use_key else 0, self.key_path, self.id)
# Execute the SQL
if database.execute_query(sql) > 0:
# Get the last row inserted
if self.id is None:
self.id = database.cursor.lastrowid
# Return true telling it's OK
return True
else:
# Return False for fail
return False
def get_tunnels(self):
return self.tunnels
def delete_tunnels(self):
# Check if have the connection ID
if self.id is not None:
# Create the DBConnection
database = DBConnection()
# Create the SQL
sql = "DELETE FROM tunnels WHERE id_connection = {}"
# Bind the value
sql = sql.format(self.id)
# Execute the query
if database.execute_query(sql) > 0:
# Return true telling it's OK
return True
else:
# Return False
return False
def delete(self):
# Check if have the connection ID
if self.id is not None:
# Create the DBConnection
database = DBConnection()
# Create the SQL
sql = "DELETE FROM connections WHERE id_connection = {}"
# Bind the value
sql = sql.format(self.id)
# Execute the query
if database.execute_query(sql) > 0:
# Return true telling it's OK
return True
else:
# Return False
return False
def get_model(self):
# Create the Model
self.model = Gtk.ListStore(str, str)
self.model.append(["Name", self.name])
self.model.append(["Host", self.host])
self.model.append(["Port", str(self.port)])
self.model.append(["User", self.user])
self.model.append(["Use Key", "Yes" if self.use_key else "No"])
self.model.append(["Key Path", self.key_path])
# Return the Model
return self.model
def generate_ssh_command(self):
# Get the settings
# Check if uses key
if self.use_key is True:
command = "/usr/bin/ssh {}@{} -i {}"
command = command.format(self.user, self.host, self.key_path)
else:
command = "/usr/bin/sshpass -p{} /usr/bin/ssh {}@{}"
command = command.format(self.password, self.user, self.host)
# Format other parameters
command += " -p {}"
command = command.format(self.port)
# Strict key checking
command += " -o StrictHostKeyChecking=no"
# Get the settings and apply here
settings = Settings()
settings.load()
# X11 forwarding
if settings.x11_forward == 1:
command += " -X"
if settings.request_compression == 1:
command += " -C"
if settings.force_ipv4 == 1:
command += " -4"
if settings.force_ipv6 == 1:
command += " -6"
# Get the tunnels
self.load_tunnels()
for tunnel in self.tunnels.get_tunnels():
command += " -L {}:{}:{}"
command = command.format(tunnel.local_port, tunnel.address,
tunnel.remote_port)
# At the end, return the command
return command