OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	11/20/2022 06:47:16 AM	rwxr-xr-x
📄 AptAuth.py	3.08 KB	07/12/2019 01:59:10 PM	rw-r--r--
📄 CountryInformation.py	1.96 KB	07/12/2019 01:59:10 PM	rw-r--r--
📄 GoaAuth.py	3.91 KB	07/12/2019 01:59:10 PM	rw-r--r--
📄 LivepatchService.py	8.99 KB	07/12/2019 01:59:10 PM	rw-r--r--
📄 LivepatchSnap.py	4.24 KB	07/12/2019 01:59:10 PM	rw-r--r--
📄 MirrorTest.py	7.28 KB	07/12/2019 01:59:10 PM	rw-r--r--
📄 SoftwareProperties.py	31.71 KB	07/12/2019 02:08:41 PM	rw-r--r--
📄 __init__.py	1.39 KB	07/12/2019 01:59:10 PM	rw-r--r--
📁 __pycache__	-	10/14/2020 08:21:42 AM	rwxr-xr-x
📄 cloudarchive.py	4.57 KB	12/02/2019 02:21:38 PM	rw-r--r--
📁 dbus	-	10/14/2020 08:21:46 AM	rwxr-xr-x
📄 distro.py	2.16 KB	07/12/2019 02:08:41 PM	rw-r--r--
📄 ppa.py	14.6 KB	07/12/2019 02:08:41 PM	rw-r--r--
📄 shortcuts.py	1.79 KB	07/12/2019 01:59:10 PM	rw-r--r--

Editing: LivepatchSnap.py

#
#  Copyright (c) 2019 Canonical
#
#  Authors:
#       Andrea Azzarone <andrea.azzarone@canonical.com>
#
#  This program is free software; you can redistribute it and/or
#  modify it under the terms of the GNU General Public License as
#  published by the Free Software Foundation; either version 2 of the
#  License, or (at your option) any later version.
#
#  This program 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.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
#  USA

from gettext import gettext as _
import logging

import gi
from gi.repository import Gio, GLib

try:
    gi.require_version('Snapd', '1')
    from gi.repository import Snapd
except(ImportError, ValueError):
    pass

class LivepatchSnap(object):

# Constants
    SNAP_NAME = 'canonical-livepatch'

# Public API
    def __init__(self):
        self._snapd_client = Snapd.Client()
        self._cancellable = Gio.Cancellable()

def get_status(self):
        """ Get the status of canonical-livepatch snap.

Returns:
            Snapd.SnapStatus.Enun: An enum indicating the status of the snap.
        """
        snap = self._get_raw_snap()
        return snap.get_status() if snap else Snapd.SnapStatus.UNKNOWN

def enable_or_install(self):
        """Enable or install canonical-livepatch snap.

Returns:
            (True, '') if successful, (False, error_message) otherwise.
        """
        status = self.get_status()

if status == Snapd.SnapStatus.ACTIVE:
            logging.warning('{} snap is already active'.format(self.SNAP_NAME))
            return True, ''
        elif status == Snapd.SnapStatus.AVAILABLE:
            return self._install()
        elif status == Snapd.SnapStatus.INSTALLED:
            return self._enable()
        else:
            logging.warning('{} snap is in an unknown state'.format(self.SNAP_NAME))
            return False, _('Canonical Livepatch snap cannot be installed.')

# Private methods
    def _get_raw_snap(self):
        """Get the Sanpd.Snap raw object of the canonical-livepatch snapd.

Returns:
            Sanpd.Snap if successful, None otherwise.
        """
        try:
            snap = self._snapd_client.get_snap_sync(
                name=self.SNAP_NAME,
                cancellable=self._cancellable)
        except GLib.Error as e:
            logging.debug('Snapd.Client.get_snap_sync failed: {}'.format(e.message))
            snap = None

if snap:
            return snap

try:
            (snaps, ignored) = self._snapd_client.find_sync(
                flags=Snapd.FindFlags.MATCH_NAME,
                query=self.SNAP_NAME,
                cancellable=self._cancellable)
            snap = snaps[0]
        except GLib.Error as e:
            logging.debug('Snapd.Client.find_sync failed: {}'.format(e.message))

return snap

def _install(self):
        """Install canonical-livepatch snap.

Returns:
            (True, '') if successful, (False, error_message) otherwise.
        """
        assert self.get_status() == Snapd.SnapStatus.AVAILABLE

try:
            self._snapd_client.install2_sync(
                flags=Snapd.InstallFlags.NONE,
                name=self.SNAP_NAME,
                cancellable=self._cancellable)
        except GLib.Error as e:
            return False, _('Canonical Livepatch snap cannot be installed: {}'.format(e.message))
        else:
            return True, ''

def _enable(self):
        """Enable the canonical-livepatch snap.

Returns:
            (True, '') if successful, (False, error_message) otherwise.
        """
        assert self.get_status() == Snapd.SnapStatus.INSTALLED

try:
            self._snapd_client.enable_sync(
                name=self.SNAP_NAME,
                cancellable=self._cancellable)
        except GLib.Error as e:
            return False, _('Canonical Livepatch snap cannot be enabled: {}'.format(e.message))
        else:
            return True, ''