1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
import os
import shutil
import tempfile
import subprocess
class ManaGer(object):
def __init__(self):
self.tmp_patch_folder = os.path.join('/tmp', 'elivepatch')
if not os.path.exists(self.tmp_patch_folder):
os.mkdir(self.tmp_patch_folder)
def list(self, kernel_version):
kernel_sources = 'gentoo-sources'
patch_filename = []
# search previous livepatch patch folder
for (dirpath, dirnames, filenames) in os.walk(self.tmp_patch_folder):
if filenames and not dirnames:
for filename in filenames:
if filename.endswith('.patch'):
print('dirpath: '+str(dirpath),'filename: '+str(filename))
incremental_patch_fullpath = os.path.join(dirpath, filename)
print(incremental_patch_fullpath)
patch_filename.append(incremental_patch_fullpath)
# search eapply_user patches
# local basedir=${PORTAGE_CONFIGROOT%/}/etc/portage/patches
try:
portage_configroot = os.environ['PORTAGE_CONFIGROOT']
except:
portage_configroot = os.path.join('/etc', 'portage', 'patches')
kernel_patch_basedir_PN = os.path.join(portage_configroot, 'sys-kernel',
kernel_sources)
kernel_patch_basedir_P = os.path.join(portage_configroot, 'sys-kernel',
kernel_sources + '-' + kernel_version)
basedir = [kernel_patch_basedir_PN, kernel_patch_basedir_P]
for path in basedir:
for (dirpath, dirnames, filenames) in os.walk(path):
if filenames and not dirnames:
for filename in filenames:
if filename.endswith('.patch'):
print('dirpath: '+str(dirpath),'filename: '+str(filename))
incremental_patch_fullpath = os.path.join(dirpath, filename)
print(incremental_patch_fullpath)
patch_filename.append(incremental_patch_fullpath)
print('List of current patches:')
return patch_filename
def load(self, patch_fulldir, livepatch_fulldir):
try:
_command(['sudo', 'kpatch', 'load', livepatch_fulldir])
print('patch_fulldir:' + str(patch_fulldir) + ' livepatch_fulldir: '+ str(livepatch_fulldir))
self.save(patch_fulldir, livepatch_fulldir)
except:
print('failed to load the livepatch')
def save(self, patch, livepatch):
i = 0
while os.path.exists(os.path.join(self.tmp_patch_folder, "elivepatch_%s" % i)):
i += 1
path_folder = os.path.join(self.tmp_patch_folder, "elivepatch_%s" % i)
os.mkdir(path_folder)
shutil.copy(patch, os.path.join(path_folder, "elivepatch.patch"))
try:
shutil.copy(livepatch, os.path.join(path_folder, "elivepatch.ko"))
except:
pass
def _command(bashCommand, kernel_source_dir=None, env=None):
"""
Popen override function
:param bashCommand: List of command arguments to execute
:param kernel_source_dir: the source directory of the kernel
:return: void
"""
# Inherit the parent environment and update the private copy
if env:
process_env = os.environ.copy()
process_env.update(env)
env = process_env
if kernel_source_dir:
print(bashCommand)
process = subprocess.Popen(bashCommand, stdout=subprocess.PIPE, cwd=kernel_source_dir, env=env)
output, error = process.communicate()
print(output)
else:
print(bashCommand)
process = subprocess.Popen(bashCommand, stdout=subprocess.PIPE, env=env)
output, error = process.communicate()
print(output)
|