openbmc-sfw - openbmc/openbmc-tools - Gitiles

 #!/bin/env python

 import argparse
 import requests
 import tarfile
 import time
 import json

 import urllib3
 #import _sysconfigdata
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)


 class BMC:
     def __init__(self, server):
         self.url = "https://{0}/".format(server)
         self.session = requests.Session()
         self.login()

     def login(self):
         r = self.session.post(self.url + 'login',
                               json={'data': ['root', '0penBmc']},
                               verify=False)
         j = r.json()
         if j['status'] != 'ok':
             raise Exception("Failed to login: \n" + r.text)

     def list_sfw(self):
         r = self.session.get(self.url + 'xyz/openbmc_project/software/',
                              verify=False)
         j = r.json()
         if j['status'] != 'ok':
             raise Exception("Failed to query software: \n" + r.text)

         events = j['data']

         return events

     def get_image(self, image):
         r = self.session.get(self.url + image, verify=False)

         j = r.json()
         if j['status'] != 'ok':
             raise Exception("Failed to get image " + image + ": \n" + r.text)

         return j['data']

     def upload_image(self, image):

         data = open(image,'rb').read()
         r = self.session.post(self.url + "/upload/image",
                               data=data,
                               headers={'Content-Type': 'application/octet-stream'},
                               verify=False)
         j = r.json()
         if j['status'] != 'ok':
             raise Exception("Failed to get event " + image + ": \n" + r.text)

         return j['data']

     def activate_image(self, image_id):
         r = self.session.put(self.url + "/xyz/openbmc_project/software/" + image_id + "/attr/RequestedActivation",
                              json={'data': 'xyz.openbmc_project.Software.Activation.RequestedActivations.Active'},
                              verify=False)

         j = r.json()
         if j['status'] != 'ok':
             raise Exception("Failed to activate image " + image_id + ": \n" + r.text)

         return j['data']

     def set_priority(self, image_id, priority):
         r = self.session.put(self.url + "/xyz/openbmc_project/software/" + image_id + "/attr/Priority",
                              json={'data': int(priority)},
                              verify=False)

         j = r.json()
         if j['status'] != 'ok':
             raise Exception("Failed to set priority of image " + image_id + ": \n" + r.text)

         return j['data']

     def delete_image(self, image_id):
         r = self.session.post(self.url + "/xyz/openbmc_project/software/" + image_id + "/action/delete",
                             headers={'Content-Type': 'application/json'},
                             data='{"data":[]}',
                             verify=False)

         j = r.json()
         if j['status'] != 'ok':
             raise Exception("Failed to delete image " + image_id + ": \n" + r.text)

         return j['data']

     def update_auto(self, image, reboot):
         image_version = self.__get_image_version(image)
         self.upload_image(image)
         image_id = self.__wait_for_image_id(image_version)
         self.activate_image(image_id)
         self.__wait_for_activation(image_id)
         if reboot:
             self.reboot()

     def reboot(self):
         r = self.session.post(
             self.url + '/org/openbmc/control/bmc0/action/warmReset',
             headers={'Content-Type': 'application/json'},
             data='{"data":[]}',
             verify=False)

         j = r.json()
         if j['status'] != 'ok':
             raise Exception("Failed to reboot BMC:\n" + r.text)

     def __get_image_version(self, image_file_path):
         # Open up the manfest file.
         image_tar = tarfile.open(name=image_file_path, mode='r')
         manifest = image_tar.extractfile('MANIFEST')

         # Find version line.
         for line in manifest:
             if line.startswith('version='):
                 manifest.close()
                 image_tar.close()
                 return line.split('version=', 1)[1].strip()

         # If we didn't find the version line, print an error and return false.
         manifest.close()
         image_tar.close()
         raise Exception("Could not find version line in image manifest")

     def __wait_for_image_id(self, image_version):
         # Try 8 times, once every 15 seconds.
         for attempt in range(8):
             software = self.list_sfw()
             # Look for our the image with the given version in software
             for path in software:
                 image = self.get_image(path)
                 if 'Version' in image and image_version == image['Version']:
                     return path.split('/')[-1]
             time.sleep(15)
         return False

     def __wait_for_activation(self, image_id):
         # Keep waiting until the image is active or activation fails.
         active = False
         while not active:
             image = self.get_image("/xyz/openbmc_project/software/" + image_id)
             if 'xyz.openbmc_project.Software.Activation.Activations.Active' \
                     == image['Activation']:
                 print 'Activation Progress: 100%'
                 active = True
             elif 'xyz.openbmc_project.Software.Activation.Activations.Activating' \
                     == image['Activation']:
                 print 'Activation Progress: ' + str(image['Progress']) + '%'
             else:
                 raise Exception("Image activation failed. The BMC has set " \
                     + "the'Activation' property to " + image['Activation'])
             time.sleep(15)

 def do_list_sfw(args):
     s = BMC(server=args.server)
     for e in s.list_sfw():
         if (e == '/xyz/openbmc_project/software/active'):
             continue
         info = s.get_image(e)
         if (((info['Purpose'] == 'xyz.openbmc_project.Software.Version.VersionPurpose.BMC') and
              (args.bmc or not args.host)) or \
            ((info['Purpose'] == 'xyz.openbmc_project.Software.Version.VersionPurpose.Host') and
             (args.host or not args.bmc))):
             print(e)
             print json.dumps(info, indent=4)

 def do_view_image(args):
     s = BMC(server=args.server)
     print json.dumps(s.get_image(args.image), indent=4)

 def do_upload_image(args):
     s = BMC(server=args.server)
     s.upload_image(args.image)

 def do_activate_image(args):
     s = BMC(server=args.server)
     s.activate_image(args.image_id)

 def do_update_auto(args):
     s = BMC(server=args.server)
     s.update_auto(args.image, args.reboot)

 def do_delete_image(args):
     s = BMC(server=args.server)
     s.delete_image(args.image_id)

 def do_set_priority(args):
     s = BMC(server=args.server)
     s.set_priority(args.image_id,args.priority)

 parser = argparse.ArgumentParser()
 parser.add_argument('--server', help='hostname or IP of BMC', type=str,
                     required=True)

 subparsers = parser.add_subparsers()
 list_events = subparsers.add_parser('list', help='List all software images on BMC')
 list_events.set_defaults(func=do_list_sfw)
 list_events.add_argument(
     '--bmc',
     action='store_true',
     default=False,
     help='Set if you want to see BMC images')
 list_events.add_argument(
     '--host',
     action='store_true',
     default=False,
     help='Set if you want to see Host images')

 image_view = subparsers.add_parser('view', help='View info of input image')
 image_view.add_argument('image', help='The image to analyze')
 image_view.set_defaults(func=do_view_image)

 image_upload = subparsers.add_parser('upload', help='Upload input image')
 image_upload.add_argument('image', help='The image to upload')
 image_upload.set_defaults(func=do_upload_image)

 image_activate = subparsers.add_parser('activate', help='Activate input image id')
 image_activate.add_argument('image_id', help='The image id to activate')
 image_activate.set_defaults(func=do_activate_image)

 image_update_auto = subparsers.add_parser('update_auto', help='Upload and '
     + 'activate an image, and then reboot the BMC to apply the update '
     + 'if necessary')
 image_update_auto.add_argument('image', help='The image to update to')
 image_update_auto.add_argument(
     '--reboot',
     action='store_true',
     default=False,
     help='Set if the BMC should reboot after the update')
 image_update_auto.set_defaults(func=do_update_auto)

 image_delete = subparsers.add_parser('delete', help='Delete input image id')
 image_delete.add_argument('image_id', help='The image id to delete')
 image_delete.set_defaults(func=do_delete_image)

 image_priority = subparsers.add_parser('priority', help='Set priority of input image')
 image_priority.add_argument('image_id', help='The image id to set priority of')
 image_priority.add_argument('priority', help='The priority to set')
 image_priority.set_defaults(func=do_set_priority)

 args = parser.parse_args()

 if 'func' in args:
     args.func(args)
 else:
     parser.print_help()
	#!/bin/env python

	import argparse
	import requests
	import tarfile
	import time
	import json

	import urllib3
	#import _sysconfigdata
	urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)


	class BMC:
	def __init__(self, server):
	self.url = "https://{0}/".format(server)
	self.session = requests.Session()
	self.login()

	def login(self):
	r = self.session.post(self.url + 'login',
	json={'data': ['root', '0penBmc']},
	verify=False)
	j = r.json()
	if j['status'] != 'ok':
	raise Exception("Failed to login: \n" + r.text)

	def list_sfw(self):
	r = self.session.get(self.url + 'xyz/openbmc_project/software/',
	verify=False)
	j = r.json()
	if j['status'] != 'ok':
	raise Exception("Failed to query software: \n" + r.text)

	events = j['data']

	return events

	def get_image(self, image):
	r = self.session.get(self.url + image, verify=False)

	j = r.json()
	if j['status'] != 'ok':
	raise Exception("Failed to get image " + image + ": \n" + r.text)

	return j['data']

	def upload_image(self, image):

	data = open(image,'rb').read()
	r = self.session.post(self.url + "/upload/image",
	data=data,
	headers={'Content-Type': 'application/octet-stream'},
	verify=False)
	j = r.json()
	if j['status'] != 'ok':
	raise Exception("Failed to get event " + image + ": \n" + r.text)

	return j['data']

	def activate_image(self, image_id):
	r = self.session.put(self.url + "/xyz/openbmc_project/software/" + image_id + "/attr/RequestedActivation",
	json={'data': 'xyz.openbmc_project.Software.Activation.RequestedActivations.Active'},
	verify=False)

	j = r.json()
	if j['status'] != 'ok':
	raise Exception("Failed to activate image " + image_id + ": \n" + r.text)

	return j['data']

	def set_priority(self, image_id, priority):
	r = self.session.put(self.url + "/xyz/openbmc_project/software/" + image_id + "/attr/Priority",
	json={'data': int(priority)},
	verify=False)

	j = r.json()
	if j['status'] != 'ok':
	raise Exception("Failed to set priority of image " + image_id + ": \n" + r.text)

	return j['data']

	def delete_image(self, image_id):
	r = self.session.post(self.url + "/xyz/openbmc_project/software/" + image_id + "/action/delete",
	headers={'Content-Type': 'application/json'},
	data='{"data":[]}',
	verify=False)

	j = r.json()
	if j['status'] != 'ok':
	raise Exception("Failed to delete image " + image_id + ": \n" + r.text)

	return j['data']

	def update_auto(self, image, reboot):
	image_version = self.__get_image_version(image)
	self.upload_image(image)
	image_id = self.__wait_for_image_id(image_version)
	self.activate_image(image_id)
	self.__wait_for_activation(image_id)
	if reboot:
	self.reboot()

	def reboot(self):
	r = self.session.post(
	self.url + '/org/openbmc/control/bmc0/action/warmReset',
	headers={'Content-Type': 'application/json'},
	data='{"data":[]}',
	verify=False)

	j = r.json()
	if j['status'] != 'ok':
	raise Exception("Failed to reboot BMC:\n" + r.text)

	def __get_image_version(self, image_file_path):
	# Open up the manfest file.
	image_tar = tarfile.open(name=image_file_path, mode='r')
	manifest = image_tar.extractfile('MANIFEST')

	# Find version line.
	for line in manifest:
	if line.startswith('version='):
	manifest.close()
	image_tar.close()
	return line.split('version=', 1)[1].strip()

	# If we didn't find the version line, print an error and return false.
	manifest.close()
	image_tar.close()
	raise Exception("Could not find version line in image manifest")

	def __wait_for_image_id(self, image_version):
	# Try 8 times, once every 15 seconds.
	for attempt in range(8):
	software = self.list_sfw()
	# Look for our the image with the given version in software
	for path in software:
	image = self.get_image(path)
	if 'Version' in image and image_version == image['Version']:
	return path.split('/')[-1]
	time.sleep(15)
	return False

	def __wait_for_activation(self, image_id):
	# Keep waiting until the image is active or activation fails.
	active = False
	while not active:
	image = self.get_image("/xyz/openbmc_project/software/" + image_id)
	if 'xyz.openbmc_project.Software.Activation.Activations.Active' \
	== image['Activation']:
	print 'Activation Progress: 100%'
	active = True
	elif 'xyz.openbmc_project.Software.Activation.Activations.Activating' \
	== image['Activation']:
	print 'Activation Progress: ' + str(image['Progress']) + '%'
	else:
	raise Exception("Image activation failed. The BMC has set " \
	+ "the'Activation' property to " + image['Activation'])
	time.sleep(15)

	def do_list_sfw(args):
	s = BMC(server=args.server)
	for e in s.list_sfw():
	if (e == '/xyz/openbmc_project/software/active'):
	continue
	info = s.get_image(e)
	if (((info['Purpose'] == 'xyz.openbmc_project.Software.Version.VersionPurpose.BMC') and
	(args.bmc or not args.host)) or \
	((info['Purpose'] == 'xyz.openbmc_project.Software.Version.VersionPurpose.Host') and
	(args.host or not args.bmc))):
	print(e)
	print json.dumps(info, indent=4)

	def do_view_image(args):
	s = BMC(server=args.server)
	print json.dumps(s.get_image(args.image), indent=4)

	def do_upload_image(args):
	s = BMC(server=args.server)
	s.upload_image(args.image)

	def do_activate_image(args):
	s = BMC(server=args.server)
	s.activate_image(args.image_id)

	def do_update_auto(args):
	s = BMC(server=args.server)
	s.update_auto(args.image, args.reboot)

	def do_delete_image(args):
	s = BMC(server=args.server)
	s.delete_image(args.image_id)

	def do_set_priority(args):
	s = BMC(server=args.server)
	s.set_priority(args.image_id,args.priority)

	parser = argparse.ArgumentParser()
	parser.add_argument('--server', help='hostname or IP of BMC', type=str,
	required=True)

	subparsers = parser.add_subparsers()
	list_events = subparsers.add_parser('list', help='List all software images on BMC')
	list_events.set_defaults(func=do_list_sfw)
	list_events.add_argument(
	'--bmc',
	action='store_true',
	default=False,
	help='Set if you want to see BMC images')
	list_events.add_argument(
	'--host',
	action='store_true',
	default=False,
	help='Set if you want to see Host images')

	image_view = subparsers.add_parser('view', help='View info of input image')
	image_view.add_argument('image', help='The image to analyze')
	image_view.set_defaults(func=do_view_image)

	image_upload = subparsers.add_parser('upload', help='Upload input image')
	image_upload.add_argument('image', help='The image to upload')
	image_upload.set_defaults(func=do_upload_image)

	image_activate = subparsers.add_parser('activate', help='Activate input image id')
	image_activate.add_argument('image_id', help='The image id to activate')
	image_activate.set_defaults(func=do_activate_image)

	image_update_auto = subparsers.add_parser('update_auto', help='Upload and '
	+ 'activate an image, and then reboot the BMC to apply the update '
	+ 'if necessary')
	image_update_auto.add_argument('image', help='The image to update to')
	image_update_auto.add_argument(
	'--reboot',
	action='store_true',
	default=False,
	help='Set if the BMC should reboot after the update')
	image_update_auto.set_defaults(func=do_update_auto)

	image_delete = subparsers.add_parser('delete', help='Delete input image id')
	image_delete.add_argument('image_id', help='The image id to delete')
	image_delete.set_defaults(func=do_delete_image)

	image_priority = subparsers.add_parser('priority', help='Set priority of input image')
	image_priority.add_argument('image_id', help='The image id to set priority of')
	image_priority.add_argument('priority', help='The priority to set')
	image_priority.set_defaults(func=do_set_priority)

	args = parser.parse_args()

	if 'func' in args:
	args.func(args)
	else:
	parser.print_help()