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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# (c) 2017, Alice Ferrazzi <alice.ferrazzi@gmail.com>
# Distributed under the terms of the GNU General Public License v2 or later
import requests
import os
import shutil
from elivepatch_client.client import patch
import sys
class ManaGer(object):
"""
RESTful client functions
"""
def __init__(self, server_url, kernel_version, uuid):
self.server_url = server_url
self.kernel_version = kernel_version
# universally unique identifier for support multiple request
self.uuid = uuid
def set_uuid(self, uuid):
self.uuid = uuid
def set_kernel_version(self, kernel_version):
self.kernel_version = kernel_version
def get_kernel_version(self):
return self.kernel_version
def get_uuid(self):
return self.uuid
def version(self):
url = self.server_url + '/elivepatch/api/v1.0/agent'
r = requests.get(url)
print(r.json())
def send_files(self, temporary_config, new_patch_fullpath, incremental_patches, api):
url = self.server_url+ api
# we are sending the file and the UUID
# The server is dividing user by UUID
# UUID is generated with python UUID
# TODO: add the UUID in the json location instead of headers
response_dict = None
headers = {
'KernelVersion' : self.kernel_version,
'UUID': self.uuid
}
# Static patch and config filename
patch_01 = open(new_patch_fullpath, 'rb')
files = [('patch', ('01.patch', patch_01, 'multipart/form-data', {'Expires': '0'})),
('patch', ('02.patch', patch_01, 'multipart/form-data', {'Expires': '0'})),
('patch', ('03.patch', patch_01, 'multipart/form-data', {'Expires': '0'})),
('config', ('config', open(temporary_config.name, 'rb'), 'multipart/form-data', {'Expires': '0'}))]
print(str(files))
temporary_config.close()
try:
response = requests.post(url, files=files, headers=headers)
print('send file: ' + str(response.json()))
response_dict = response.json()
except requests.exceptions.ConnectionError as e:
print('connection error: %s' % e)
sys.exit(1)
#except:
#self.catching_exceptions_exit(self.send_file)
return response_dict
def build_livepatch(self):
url = self.server_url+'/elivepatch/api/v1.0/build_livepatch'
payload = {
'KernelVersion': self.kernel_version,
'UUID' : self.uuid
}
try:
response = requests.post(url, json=payload)
print(response.json())
except:
self.catching_exceptions_exit(self.build_livepatch)
def get_livepatch(self, patch_folder):
from io import BytesIO
patch_manager = patch.ManaGer()
url = self.server_url+'/elivepatch/api/v1.0/send_livepatch'
payload = {
'KernelVersion': self.kernel_version,
'UUID' : self.uuid
}
try:
r = requests.get(url, json=payload)
if r.status_code == requests.codes.ok: # livepatch returned ok
try:
b= BytesIO(r.content)
with open('myfile.ko', 'wb') as out:
out.write(r.content)
r.close()
print(b)
except:
print('livepatch not found')
r.close()
except:
self.catching_exceptions_exit(self.get_livepatch)
elivepatch_uuid_dir = os.path.join('..', 'elivepatch-'+ self.uuid)
livepatch_fulldir = os.path.join(elivepatch_uuid_dir, 'livepatch.ko')
if os.path.exists('myfile.ko'):
if not os.path.exists(elivepatch_uuid_dir):
os.makedirs(elivepatch_uuid_dir)
shutil.copy("myfile.ko", livepatch_fulldir)
print('livepatch saved in ' + elivepatch_uuid_dir + '/ folder')
patch_manager.load(patch_folder, livepatch_fulldir)
else:
print('livepatch not received')
def catching_exceptions_exit(self, current_function):
e = sys.exc_info()
print( "Error %s: %s" % (current_function.__name__, str(e)) )
sys.exit(1)
|