#!/usr/bin/python3 import os import shutil import subprocess import sys import gi gi.require_version('Gtk', '3.0') from gi.repository import Gtk, Pango MOUNT_PATH = '/mnt/pw_ramdisk' PWD_FILE_PATH = '/home/dsorber/Desktop/network' ENC_FILE_PATH = '/home/dsorber/Documents/chicken_soup.enc' SCRATCH_FILE_PATH = '/mnt/pw_ramdisk/chicken_soup.txt' ENC_TMP_FILE_PATH = '/home/dsorber/Documents/chicken_soup.enc.tmp' VERSION = '0.0.1' class SearchDialog(Gtk.Dialog): def __init__(self, parent): Gtk.Dialog.__init__(self, 'Search', parent, Gtk.DialogFlags.MODAL, buttons=( Gtk.STOCK_FIND, Gtk.ResponseType.OK, Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL)) box = self.get_content_area() label = Gtk.Label('Insert text you want to search for:') box.add(label) self.entry = Gtk.Entry() box.add(self.entry) self.show_all() def get_user_pw(parent, message, title=''): """ Returns user input as a string or None If user does not input text it returns None, NOT AN EMPTY STRING. https://stackoverflow.com/questions/13970445/python-gtk3-how-to-add-a-gtk-entry-to-a-gtk-messagedialog """ dialog = Gtk.MessageDialog(parent, Gtk.DialogFlags.MODAL | Gtk.DialogFlags.DESTROY_WITH_PARENT, Gtk.MessageType.QUESTION, Gtk.ButtonsType.OK_CANCEL, message) dialog.set_title(title) box = dialog.get_content_area() pw_entry = Gtk.Entry() pw_entry.set_visibility(False) pw_entry.set_invisible_char('*') pw_entry.set_size_request(200, 0) box.add(pw_entry) #~ box.pack_end(pw_entry, False, False, 0) dialog.show_all() response = dialog.run() text = pw_entry.get_text() dialog.destroy() if (response == Gtk.ResponseType.OK) and (text != ''): return text else: return None class TextViewWindow(Gtk.Window): def __init__(self): Gtk.Window.__init__(self, title='pwmgr') self.set_default_size(800, 600) self.grid = Gtk.Grid() self.add(self.grid) self.create_textview() self.create_toolbar() self.create_bottom() self.mounted = False self.pwd_file_fd = None self.pwd = None def create_toolbar(self): toolbar = Gtk.Toolbar() self.grid.attach(toolbar, 0, 0, 3, 1) button_open = Gtk.ToolButton() button_open.set_icon_name('document-open') toolbar.insert(button_open, 0) button_open.connect('clicked', self.open_pwd_file) button_save = Gtk.ToolButton() button_save.set_icon_name('document-save') toolbar.insert(button_save, 1) button_save.connect('clicked', self.save_pwd_file) button_close = Gtk.ToolButton() button_close.set_icon_name('window-close') toolbar.insert(button_close, 2) button_close.connect('clicked', self.close_pwd_file, None) toolbar.insert(Gtk.SeparatorToolItem(), 3) button_clear = Gtk.ToolButton() button_clear.set_icon_name('edit-clear-symbolic') button_clear.connect('clicked', self.on_clear_clicked) toolbar.insert(button_clear, 4) toolbar.insert(Gtk.SeparatorToolItem(), 5) button_search = Gtk.ToolButton() button_search.set_icon_name('system-search-symbolic') button_search.connect('clicked', self.on_search_clicked) toolbar.insert(button_search, 6) def create_textview(self): scrolledwindow = Gtk.ScrolledWindow() scrolledwindow.set_hexpand(True) scrolledwindow.set_vexpand(True) self.grid.attach(scrolledwindow, 0, 1, 3, 1) self.textview = Gtk.TextView() self.textview.set_editable(False) self.textview.set_cursor_visible(False) self.textbuffer = self.textview.get_buffer() scrolledwindow.add(self.textview) self.tag_bold = self.textbuffer.create_tag("bold", weight=Pango.Weight.BOLD) self.tag_italic = self.textbuffer.create_tag("italic", style=Pango.Style.ITALIC) self.tag_underline = self.textbuffer.create_tag("underline", underline=Pango.Underline.SINGLE) self.tag_found = self.textbuffer.create_tag('found', background='orange') def create_bottom(self): label = Gtk.Label() label.set_text('version: {:s}'.format(VERSION)) label.set_justify(Gtk.Justification.RIGHT) self.grid.attach(label, 0, 2, 3, 1) def open_pwd_file(self, widget): """ Open handler """ # Get password self.pwd = get_user_pw(self, 'Please enter your password', 'Password') if self.pwd is None: return #~ print('User entered: {:s}'.format(self.pwd)) if not self.mounted: os.makedirs(MOUNT_PATH, 0o600, exist_ok=True) # Mount the tmpfs mnt_cmd = 'mount -t tmpfs -o size=16m tmpfs {:s}' mnt_cmd = mnt_cmd.format(MOUNT_PATH) subprocess.call(mnt_cmd, shell=True, timeout=5) self.mounted = True cmd = 'openssl enc -aes-256-cbc -d -in {:s} -out {:s} -pass pass:{:s}' cmd = cmd.format(ENC_FILE_PATH, SCRATCH_FILE_PATH, self.pwd) subprocess.call(cmd, shell=True, timeout=5) # Open and load file; make textview editable with open(SCRATCH_FILE_PATH, 'r') as pwd_file_fd: self.textbuffer.set_text(pwd_file_fd.read()) self.textview.set_editable(True) self.textview.set_cursor_visible(True) def save_pwd_file(self, widget): """ Save handler """ if self.mounted: with open(SCRATCH_FILE_PATH, 'w') as pwd_file_fd: pwd_file_fd.write( self.textbuffer.get_text(self.textbuffer.get_start_iter(), self.textbuffer.get_end_iter(), True)) print('Write complete') cmd = 'openssl enc -aes-256-cbc -salt -in {:s} -out {:s} -pass pass:{:s}' cmd = cmd.format(SCRATCH_FILE_PATH, ENC_TMP_FILE_PATH, self.pwd) subprocess.call(cmd, shell=True, timeout=5) print('enc complete') # Clear the buffer and disable the textview self.textbuffer.set_text('') self.textview.set_editable(False) self.textview.set_cursor_visible(False) def close_pwd_file(self, widget, other): """ Close handler """ rand_write_cmd = 'dd if=/dev/urandom of={:s} bs=16M count=1 status=none' rand_write_cmd = rand_write_cmd.format(os.path.join(MOUNT_PATH, 'data')) zero_write_cmd = 'dd if=/dev/zero of={:s} bs=16M count=1 status=none' zero_write_cmd = zero_write_cmd.format(os.path.join(MOUNT_PATH, 'data')) if self.mounted: #~ shutil.rmtree(os.path.join(MOUNT_PATH, '*'), ignore_errors=True) rm_cmd = 'rm -rf {:s}'.format(os.path.join(MOUNT_PATH, '*')) subprocess.call(rm_cmd, shell=True, timeout=5) # Overwrite a few times for ctr in range(10): print(ctr) subprocess.call(rand_write_cmd, shell=True, timeout=5) subprocess.call(rm_cmd, shell=True, timeout=5) subprocess.call(zero_write_cmd, shell=True, timeout=5) subprocess.call(rm_cmd, shell=True, timeout=5) umount_cmd = 'umount {:s}'.format(MOUNT_PATH) subprocess.call(umount_cmd, shell=True, timeout=5) Gtk.main_quit() print('CLOSE') def on_button_clicked(self, widget, tag): bounds = self.textbuffer.get_selection_bounds() if len(bounds) != 0: start, end = bounds self.textbuffer.apply_tag(tag, start, end) def on_clear_clicked(self, widget): start = self.textbuffer.get_start_iter() end = self.textbuffer.get_end_iter() self.textbuffer.remove_all_tags(start, end) def on_search_clicked(self, widget): dialog = SearchDialog(self) response = dialog.run() if response == Gtk.ResponseType.OK: cursor_mark = self.textbuffer.get_insert() start = self.textbuffer.get_iter_at_mark(cursor_mark) if start.get_offset() == self.textbuffer.get_char_count(): start = self.textbuffer.get_start_iter() self.search_and_mark(dialog.entry.get_text(), start) dialog.destroy() def search_and_mark(self, text, start): end = self.textbuffer.get_end_iter() match = start.forward_search(text, 0, end) if match != None: match_start, match_end = match self.textbuffer.apply_tag(self.tag_found, match_start, match_end) self.search_and_mark(text, match_end) def main(): """ NOTE: this *must* be called using sudo or gksudo """ win = TextViewWindow() win.connect('delete-event', win.close_pwd_file) win.show_all() Gtk.main() if __name__ == '__main__': sys.exit(main())