tomb

tomb.cpp (5513B)

---

 /*  Tomb - encrypted storage undertaker
  *
  *  (c) Copyright 2015 Gianluca Montecchi <gian@grys.it>
  *
  * This source code is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Public License as published
  * by the Free Software Foundation; either version 3 of the License,
  * or (at your option) any later version.
  *
  * This source code 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.
  * Please refer to the GNU Public License for more details.
  *
  * You should have received a copy of the GNU Public License along with
  * this source code; if not, write to:
  * Free Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  */
 
 
 #include "tomb.h"
 #include <QDir>
 #include <QStorageInfo>
 #include <QMessageBox>
 #include <QQuickView>
 
 #include <mntent.h>
 
 Tomb::Tomb(QWidget *parent)
     : QDialog(parent)
 {
     struct mntent *ent;
     FILE *aFile;
 
 
     if (QApplication::arguments().length() > 1) {
         this->info = QFileInfo(QApplication::arguments().takeAt(1));
         this->tombName = QString(info.baseName());
         this->tombPath = info.path();
     } else {
         QMessageBox::critical(this, tr("tomb-qt-tray"), tr("You need to specify a Tomb File.\nExiting"), QMessageBox::Ok);
         exit(EXIT_FAILURE);
     }
 
 
     // Build the menù
     this->icon = QIcon( QCoreApplication::applicationDirPath() + QDir::separator()+QString("pixmaps/tomb_icon.png"));
     this->trayIcon = new QSystemTrayIcon(this->icon);
     this->trayIconMenu = new QMenu();
 
     this->tombBuildMenu();
 
     this->trayIcon->setContextMenu(this->trayIconMenu);
 
     this->trayIcon->setToolTip(QString(info.baseName()));
     this->trayIcon->show();
     this->trayIcon->showMessage(tr("Tomb encrypted undertaker"),tr("We started digging out bones"), QSystemTrayIcon::Information);
     if (QT_VERSION >= 0x050400) {
         for (auto volume : QStorageInfo::mountedVolumes()) {
             if (QString(volume.device()).contains(this->tombName) == true) {
                 this->tombMountPoint = QString(volume.rootPath());
                 break;
             }
         }
     } else {
          aFile = setmntent("/proc/mounts", "r");
          if (aFile == NULL) {
            perror("setmntent");
            exit(1);
          }
          while (NULL != (ent = getmntent(aFile))) {
              if (QString( ent->mnt_fsname).contains(this->tombName) == true) {
                  this->tombMountPoint = QString(ent->mnt_dir);
                  break;
              }
          }
          endmntent(aFile);
     }
 }
 
 void Tomb::closeEvent(QCloseEvent *event) {
     event->accept();
 }
 
 void Tomb::tombBuildMenu() {
 
     // Create the menu items
 //    this->tombOpen = new QAction(tr("Open"), this);
     //this->trayIconMenu->addAction(tombOpen);
 
     this->menu_tombExplore = new QAction(tr("Explore"), this);
     this->trayIconMenu->addAction(menu_tombExplore);
 
     this->trayIconMenu->addSeparator();
 
     this->menu_tombClose = new QAction(tr("Close"), this);
     this->trayIconMenu->addAction(menu_tombClose);
 
     this->menu_tombSlam = new QAction(tr("Slam"), this);
     this->trayIconMenu->addAction(menu_tombSlam);
 
     connect(this->menu_tombExplore,  SIGNAL(triggered()), this, SLOT(tombExplore()));
     connect(this->menu_tombClose,  SIGNAL(triggered()), this, SLOT(tombClose()));
     connect(this->menu_tombSlam,  SIGNAL(triggered()), this, SLOT(tombSlam()));
 
 }
 
 void Tomb::tombExplore() {
 
     QDesktopServices::openUrl(QUrl(QUrl::fromLocalFile(this->tombMountPoint)));
 
 }
 
 void Tomb::tombClose() {
 
     QProcess *tomb;
     tomb = new QProcess(this);
     tomb->start("pkexec", QStringList() << "tomb" << "close");
     connect(tomb, SIGNAL(finished(int , QProcess::ExitStatus )), this, SLOT(tombCheckCmdRet(int , QProcess::ExitStatus )));
     connect(tomb, SIGNAL(error(QProcess::ProcessError)), this, SLOT(tombStartError(QProcess::ProcessError)));
     tomb->waitForFinished(30000);
 
 }
 
 void Tomb::tombSlam() {
 
     QProcess *tomb;
     tomb = new QProcess(this);
     tomb->start("pkexec", QStringList() << "tomb" << "slam");
     connect(tomb, SIGNAL(finished(int , QProcess::ExitStatus )), this, SLOT(tombCheckCmdRet(int , QProcess::ExitStatus )));
     connect(tomb, SIGNAL(error(QProcess::ProcessError)), this, SLOT(tombStartError(QProcess::ProcessError)));
     tomb->waitForFinished(30000);
 
 }
 
 void Tomb::tombCheckCmdRet(int exitCode, QProcess::ExitStatus exitStatus) {
 
     if (exitStatus == QProcess::CrashExit) {
         QMessageBox::critical(this, tr("tomb-qt-tray"), tr("polkit is not installed"),QMessageBox::Ok);
     } else {
         if (exitCode != QProcess::NormalExit) {
             QMessageBox::critical(this, tr("tomb-qt-tray"), tr("The program crashed\nTry to run 'tomb close' in a console"),QMessageBox::Ok);
         }
     }
 
 }
 
 void Tomb::tombStartError(QProcess::ProcessError err) {
 
     QString msg = QString();
 
     switch (err) {
         case QProcess::FailedToStart :
             msg = tr("The process failed to start. Either the invoked program is missing, or you may have insufficient permissions to invoke the program.");
         break;
         case QProcess::Crashed:
         case QProcess::Timedout:
         case QProcess::ReadError:
         case QProcess::WriteError:
         case QProcess::UnknownError:
             break;
     }
     QMessageBox::critical(this, tr("tomb-qt-tray"), msg, QMessageBox::Ok);
 
 }
 
 
 Tomb::~Tomb() {
 
 }