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
|
#!/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
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_file(self, temporary_config, patch_fullpath, 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
headers = {
'KernelVersion' : self.kernel_version,
'UUID': self.uuid
}
# Static patch and config filename
files = {'patch': ('01.patch', open(patch_fullpath, 'rb'), 'multipart/form-data', {'Expires': '0'}),
'config': ('config', open(temporary_config.name, 'rb'), 'multipart/form-data', {'Expires': '0'})}
print(str(files))
temporary_config.close()
r = requests.post(url, files=files, headers=headers)
print('send file: ' + str(r.json()))
r_dict = r.json()
return r_dict
def build_livepatch(self):
url = self.server_url+'/elivepatch/api/v1.0/build_livepatch'
payload = {
'KernelVersion': self.kernel_version,
'UUID' : self.uuid
}
r = requests.post(url, json=payload)
print(r.json())
def get_livepatch(self):
from io import BytesIO
url = self.server_url+'/elivepatch/api/v1.0/send_livepatch'
payload = {
'KernelVersion': self.kernel_version,
'UUID' : self.uuid
}
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()
if os.path.exists('myfile.ko'):
elivepatch_uuid_dir = os.path.join('..', 'elivepatch-'+ self.uuid)
if not os.path.exists(elivepatch_uuid_dir):
os.makedirs(elivepatch_uuid_dir)
shutil.move("myfile.ko", os.path.join(elivepatch_uuid_dir, 'livepatch.ko'))
print('livepatch saved in ' + elivepatch_uuid_dir + '/ folder')
else:
print('livepatch not received')
|
GitWeb