blob: 4d13271ee96c5f5d8f39bcd74bdc2039e2ac7c2b [file] [log] [blame]
Andrew Geissler53be16b2017-09-06 16:23:27 -05001#!/bin/env python
2
3import argparse
4import requests
Charles P. Hofer88d15b32017-09-12 11:32:55 -05005import tarfile
6import time
Andrew Geissler53be16b2017-09-06 16:23:27 -05007import json
8
9import urllib3
Charles P. Hofere3ec3572017-09-15 13:30:37 -050010#import _sysconfigdata
Andrew Geissler53be16b2017-09-06 16:23:27 -050011urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
12
13
14class BMC:
15 def __init__(self, server):
16 self.url = "https://{0}/".format(server)
17 self.session = requests.Session()
18 self.login()
19
20 def login(self):
21 r = self.session.post(self.url + 'login',
22 json={'data': ['root', '0penBmc']},
23 verify=False)
24 j = r.json()
25 if j['status'] != 'ok':
26 raise Exception("Failed to login: \n" + r.text)
27
28 def list_sfw(self):
29 r = self.session.get(self.url + 'xyz/openbmc_project/software/',
30 verify=False)
31 j = r.json()
32 if j['status'] != 'ok':
Andrew Geissler942eca22017-09-08 11:20:17 -050033 raise Exception("Failed to query software: \n" + r.text)
Andrew Geissler53be16b2017-09-06 16:23:27 -050034
35 events = j['data']
36
37 return events
38
39 def get_image(self, image):
40 r = self.session.get(self.url + image, verify=False)
41
42 j = r.json()
43 if j['status'] != 'ok':
Andrew Geissler942eca22017-09-08 11:20:17 -050044 raise Exception("Failed to get image " + image + ": \n" + r.text)
Andrew Geissler53be16b2017-09-06 16:23:27 -050045
46 return j['data']
47
48 def upload_image(self, image):
Charles P. Hofer88d15b32017-09-12 11:32:55 -050049
Andrew Geissler53be16b2017-09-06 16:23:27 -050050 data = open(image,'rb').read()
51 r = self.session.post(self.url + "/upload/image",
52 data=data,
53 headers={'Content-Type': 'application/octet-stream'},
54 verify=False)
55 j = r.json()
56 if j['status'] != 'ok':
57 raise Exception("Failed to get event " + image + ": \n" + r.text)
58
59 return j['data']
60
Andrew Geissler942eca22017-09-08 11:20:17 -050061 def activate_image(self, image_id):
Charles P. Hofer88d15b32017-09-12 11:32:55 -050062 r = self.session.put(self.url + "/xyz/openbmc_project/software/" + image_id + "/attr/RequestedActivation",
Andrew Geissler64da2182017-09-08 14:06:42 -050063 json={'data': 'xyz.openbmc_project.Software.Activation.RequestedActivations.Active'},
Andrew Geissler942eca22017-09-08 11:20:17 -050064 verify=False)
Charles P. Hofer88d15b32017-09-12 11:32:55 -050065
Andrew Geissler942eca22017-09-08 11:20:17 -050066 j = r.json()
67 if j['status'] != 'ok':
68 raise Exception("Failed to activate image " + image_id + ": \n" + r.text)
69
70 return j['data']
71
Charles P. Hofer88d15b32017-09-12 11:32:55 -050072 def update_auto(self, image, reboot):
73 image_version = self.__get_image_version(image)
74 self.upload_image(image)
75 image_id = self.__wait_for_image_id(image_version)
76 self.activate_image(image_id)
77 self.__wait_for_activation(image_id)
78 if reboot:
79 self.reboot()
80
81 def reboot(self):
82 r = self.session.post(
83 self.url + '/org/openbmc/control/bmc0/action/warmReset',
84 headers={'Content-Type': 'application/json'},
85 data='{"data":[]}',
86 verify=False)
87
88 j = r.json()
89 if j['status'] != 'ok':
90 raise Exception("Failed to reboot BMC:\n" + r.text)
91
92 def __get_image_version(self, image_file_path):
93 # Open up the manfest file.
94 image_tar = tarfile.open(name=image_file_path, mode='r')
95 manifest = image_tar.extractfile('MANIFEST')
96
97 # Find version line.
98 for line in manifest:
99 if line.startswith('version='):
100 manifest.close()
101 image_tar.close()
102 return line.split('version=', 1)[1].strip()
103
104 # If we didn't find the version line, print an error and return false.
105 manifest.close()
106 image_tar.close()
107 raise Exception("Could not find version line in image manifest")
108
109 def __wait_for_image_id(self, image_version):
110 # Try 8 times, once every 15 seconds.
111 for attempt in range(8):
112 software = self.list_sfw()
113 # Look for our the image with the given version in software
114 for path in software:
115 image = self.get_image(path)
116 if image_version == image['Version']:
117 return path.split('/')[-1]
118 time.sleep(15)
119 return False
120
121 def __wait_for_activation(self, image_id):
122 # Keep waiting until the image is active or activation fails.
123 active = False
124 while not active:
125 image = self.get_image("/xyz/openbmc_project/software/" + image_id)
126 if 'xyz.openbmc_project.Software.Activation.Activations.Active' \
127 == image['Activation']:
128 print 'Activation Progress: 100%'
129 active = True
130 elif 'xyz.openbmc_project.Software.Activation.Activations.Activating' \
131 == image['Activation']:
132 print 'Activation Progress: ' + str(image['Progress']) + '%'
133 else:
134 raise Exception("Image activation failed. The BMC has set " \
135 + "the'Activation' property to " + image['Activation'])
136 time.sleep(15)
Andrew Geissler53be16b2017-09-06 16:23:27 -0500137
138def do_list_sfw(args):
139 s = BMC(server=args.server)
140 for e in s.list_sfw():
141 info = s.get_image(e)
142 print(e)
143 print json.dumps(info, indent=4)
Charles P. Hofer88d15b32017-09-12 11:32:55 -0500144
Andrew Geissler53be16b2017-09-06 16:23:27 -0500145def do_view_image(args):
146 s = BMC(server=args.server)
147 print json.dumps(s.get_image(args.image), indent=4)
148
149def do_upload_image(args):
150 s = BMC(server=args.server)
151 s.upload_image(args.image)
152
Andrew Geissler942eca22017-09-08 11:20:17 -0500153def do_activate_image(args):
154 s = BMC(server=args.server)
155 s.activate_image(args.image_id)
156
Charles P. Hofer88d15b32017-09-12 11:32:55 -0500157def do_update_auto(args):
158 s = BMC(server=args.server)
159 s.update_auto(args.image, args.reboot)
160
Andrew Geissler53be16b2017-09-06 16:23:27 -0500161parser = argparse.ArgumentParser()
162parser.add_argument('--server', help='hostname or IP of BMC', type=str,
163 required=True)
164
165subparsers = parser.add_subparsers()
166list_events = subparsers.add_parser('list', help='List all software images on BMC')
167list_events.set_defaults(func=do_list_sfw)
168
169image_view = subparsers.add_parser('view', help='View info of input image')
170image_view.add_argument('image', help='The image to analyze')
171image_view.set_defaults(func=do_view_image)
172
173image_upload = subparsers.add_parser('upload', help='Upload input image')
174image_upload.add_argument('image', help='The image to upload')
175image_upload.set_defaults(func=do_upload_image)
176
Andrew Geissler942eca22017-09-08 11:20:17 -0500177image_activate = subparsers.add_parser('activate', help='Activate input image id')
178image_activate.add_argument('image_id', help='The image id to activate')
179image_activate.set_defaults(func=do_activate_image)
180
Charles P. Hofer88d15b32017-09-12 11:32:55 -0500181image_update_auto = subparsers.add_parser('update_auto', help='Upload and '
182 + 'activate an image, and then reboot the BMC to apply the update '
183 + 'if necessary')
184image_update_auto.add_argument('image', help='The image to update to')
185image_update_auto.add_argument(
186 '--reboot',
Charles P. Hofer88d15b32017-09-12 11:32:55 -0500187 action='store_true',
188 default=False,
189 help='Set if the BMC should reboot after the update')
Charles P. Hofer88d15b32017-09-12 11:32:55 -0500190image_update_auto.set_defaults(func=do_update_auto)
191
Andrew Geissler53be16b2017-09-06 16:23:27 -0500192args = parser.parse_args()
193
194if 'func' in args:
195 args.func(args)
196else:
197 parser.print_help()