Project

General

Profile

Download (9.42 KB) Statistics
| Branch: | Tag: | Revision:

root/software/pwmgr/pwmgr.py @ 97b0fc46

5a945a4a dsorber
#!/usr/bin/python3
import os
import shutil
import subprocess
feddc602 dsorber
import sys

import gi
gi.require_version('Gtk', '3.0')
from gi.repository import Gtk, Pango

5a945a4a dsorber
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'

feddc602 dsorber
class SearchDialog(Gtk.Dialog):

def __init__(self, parent):
5a945a4a dsorber
Gtk.Dialog.__init__(self, 'Search', parent,
feddc602 dsorber
Gtk.DialogFlags.MODAL, buttons=(
Gtk.STOCK_FIND, Gtk.ResponseType.OK,
Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL))

box = self.get_content_area()

5a945a4a dsorber
label = Gtk.Label('Insert text you want to search for:')
feddc602 dsorber
box.add(label)

self.entry = Gtk.Entry()
box.add(self.entry)

self.show_all()
5a945a4a dsorber
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
"""
97b0fc46 dsorber
dialog = Gtk.MessageDialog(parent,
5a945a4a dsorber
Gtk.DialogFlags.MODAL |
Gtk.DialogFlags.DESTROY_WITH_PARENT,
Gtk.MessageType.QUESTION,
Gtk.ButtonsType.OK_CANCEL,
message)

97b0fc46 dsorber
dialog.set_title(title)
5a945a4a dsorber
97b0fc46 dsorber
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)
5a945a4a dsorber
97b0fc46 dsorber
dialog.show_all()
response = dialog.run()
text = pw_entry.get_text()
dialog.destroy()
5a945a4a dsorber
if (response == Gtk.ResponseType.OK) and (text != ''):
return text
else:
return None
feddc602 dsorber
class TextViewWindow(Gtk.Window):

def __init__(self):
5a945a4a dsorber
Gtk.Window.__init__(self, title='pwmgr')
feddc602 dsorber
5a945a4a dsorber
self.set_default_size(800, 600)
feddc602 dsorber
self.grid = Gtk.Grid()
self.add(self.grid)

self.create_textview()
self.create_toolbar()
5a945a4a dsorber
self.create_bottom()
self.mounted = False
self.pwd_file_fd = None
self.pwd = None
feddc602 dsorber
def create_toolbar(self):
toolbar = Gtk.Toolbar()
self.grid.attach(toolbar, 0, 0, 3, 1)

5a945a4a dsorber
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)
feddc602 dsorber
toolbar.insert(Gtk.SeparatorToolItem(), 3)

button_clear = Gtk.ToolButton()
5a945a4a dsorber
button_clear.set_icon_name('edit-clear-symbolic')
button_clear.connect('clicked', self.on_clear_clicked)
toolbar.insert(button_clear, 4)
feddc602 dsorber
5a945a4a dsorber
toolbar.insert(Gtk.SeparatorToolItem(), 5)
feddc602 dsorber
button_search = Gtk.ToolButton()
5a945a4a dsorber
button_search.set_icon_name('system-search-symbolic')
button_search.connect('clicked', self.on_search_clicked)
toolbar.insert(button_search, 6)
feddc602 dsorber
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()
5a945a4a dsorber
self.textview.set_editable(False)
self.textview.set_cursor_visible(False)
feddc602 dsorber
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)
5a945a4a dsorber
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')
97b0fc46 dsorber
if self.pwd is None:
return
5a945a4a dsorber
#~ 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')
feddc602 dsorber
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():
5a945a4a dsorber
"""
NOTE: this *must* be called using sudo or gksudo
"""
feddc602 dsorber
win = TextViewWindow()
5a945a4a dsorber
win.connect('delete-event', win.close_pwd_file)
feddc602 dsorber
win.show_all()
Gtk.main()
if __name__ == '__main__':
sys.exit(main())