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
|
#!/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 time
import requests
class ManaGer(object):
def __init__(self, server_url, kernel_version):
self.server_url = server_url
self.kernel_version = kernel_version
self.user_id = None
def set_user_id(self, user_id):
self.user_id = user_id
def set_kernel_version(self, kernel_version):
self.kernel_version = kernel_version
def get_kernel_version(self):
return self.kernel_version
def get_user_id(self):
return self.user_id
def version(self):
url = self.server_url + '/elivepatch/api/v1.0/agent'
r = requests.get(url)
print(r.text)
print(r.json())
def send_file(self, config_send_file, patch_send_file, config_file_name, patch_file_name, api):
url = self.server_url+ api
# we are sending the file and the UserID
# The server is dividing user by UserID
# UserID is generated with python UUID
# TODO: add the UserID in the json location instead of headers
headers = {
'KernelVersion' : self.kernel_version,
'UserID': self.user_id
}
files = {'patch': (patch_file_name, open(patch_send_file, 'rb'), 'multipart/form-data', {'Expires': '0'}),
'config': (config_file_name, open(config_send_file, 'rb'), 'multipart/form-data', {'Expires': '0'})}
print(str(files))
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,
'UserID' : self.user_id
}
r = requests.post(url, json=payload)
# print(r.text)
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,
'UserID' : self.user_id
}
r = requests.get(url, json=payload)
if r.status_code == requests.codes.ok: # livepatch returned ok
b= BytesIO(r.content)
with open('myfile.ko', 'wb') as out:
out.write(r.content)
r.close()
print(b)
else:
r.close()
time.sleep(5)
return self.get_livepatch() # try to get the livepatch again
|