Norman James | b4914ad | 2015-10-26 07:14:29 -0500 | [diff] [blame] | 1 | #!/usr/bin/python |
| 2 | |
| 3 | import sys |
Norman James | b4914ad | 2015-10-26 07:14:29 -0500 | [diff] [blame] | 4 | import dbus |
Andrew Jeffery | 0b6883d | 2017-10-11 22:08:55 +1030 | [diff] [blame] | 5 | import argparse |
Norman James | b4914ad | 2015-10-26 07:14:29 -0500 | [diff] [blame] | 6 | |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 7 | from dbus.mainloop.glib import DBusGMainLoop |
| 8 | import gobject |
Andrew Jeffery | b43f7ed | 2017-10-12 01:33:59 +1030 | [diff] [blame] | 9 | import os |
| 10 | import signal |
| 11 | import time |
| 12 | from subprocess import Popen |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 13 | |
Andrew Jeffery | c6b5deb | 2017-10-11 22:18:27 +1030 | [diff] [blame] | 14 | descriptors = { |
Andrew Geissler | cae038e | 2016-12-19 15:29:10 -0600 | [diff] [blame] | 15 | 'power': { |
| 16 | 'bus_name': 'org.openbmc.control.Power', |
| 17 | 'object_name': '/org/openbmc/control/power0', |
| 18 | 'interface_name': 'org.openbmc.control.Power' |
| 19 | }, |
Andrew Geissler | 2cbbb67 | 2017-05-17 10:52:49 -0500 | [diff] [blame] | 20 | 'chassison': { |
| 21 | 'bus_name': 'xyz.openbmc_project.State.Chassis', |
| 22 | 'object_name': '/xyz/openbmc_project/state/chassis0', |
| 23 | 'interface_name': 'xyz.openbmc_project.State.Chassis', |
| 24 | 'property': 'RequestedPowerTransition', |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 25 | 'value': 'xyz.openbmc_project.State.Chassis.Transition.On', |
| 26 | 'monitor': 'obmc-chassis-poweron@0.target', |
Andrew Geissler | 2cbbb67 | 2017-05-17 10:52:49 -0500 | [diff] [blame] | 27 | }, |
| 28 | 'chassisoff': { |
| 29 | 'bus_name': 'xyz.openbmc_project.State.Chassis', |
| 30 | 'object_name': '/xyz/openbmc_project/state/chassis0', |
| 31 | 'interface_name': 'xyz.openbmc_project.State.Chassis', |
| 32 | 'property': 'RequestedPowerTransition', |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 33 | 'value': 'xyz.openbmc_project.State.Chassis.Transition.Off', |
| 34 | 'monitor': 'obmc-chassis-hard-poweroff@0.target', |
Andrew Geissler | 2cbbb67 | 2017-05-17 10:52:49 -0500 | [diff] [blame] | 35 | }, |
Andrew Geissler | cae038e | 2016-12-19 15:29:10 -0600 | [diff] [blame] | 36 | 'poweron': { |
Andrew Geissler | 25b9a85 | 2016-12-19 16:14:09 -0600 | [diff] [blame] | 37 | 'bus_name': 'xyz.openbmc_project.State.Host', |
| 38 | 'object_name': '/xyz/openbmc_project/state/host0', |
| 39 | 'interface_name': 'xyz.openbmc_project.State.Host', |
| 40 | 'property': 'RequestedHostTransition', |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 41 | 'value': 'xyz.openbmc_project.State.Host.Transition.On', |
| 42 | 'monitor': 'obmc-host-start@0.target', |
Andrew Geissler | cae038e | 2016-12-19 15:29:10 -0600 | [diff] [blame] | 43 | }, |
| 44 | 'poweroff': { |
Andrew Geissler | 25b9a85 | 2016-12-19 16:14:09 -0600 | [diff] [blame] | 45 | 'bus_name': 'xyz.openbmc_project.State.Host', |
| 46 | 'object_name': '/xyz/openbmc_project/state/host0', |
| 47 | 'interface_name': 'xyz.openbmc_project.State.Host', |
| 48 | 'property': 'RequestedHostTransition', |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 49 | 'value': 'xyz.openbmc_project.State.Host.Transition.Off', |
| 50 | 'monitor': 'obmc-host-stop@0.target', |
Andrew Geissler | cae038e | 2016-12-19 15:29:10 -0600 | [diff] [blame] | 51 | }, |
Andrew Geissler | 8f8cbce | 2017-02-20 16:10:26 -0600 | [diff] [blame] | 52 | 'bmcstate': { |
| 53 | 'bus_name': 'xyz.openbmc_project.State.BMC', |
| 54 | 'object_name': '/xyz/openbmc_project/state/bmc0', |
| 55 | 'interface_name': 'xyz.openbmc_project.State.BMC', |
| 56 | 'property': 'CurrentBMCState', |
| 57 | }, |
| 58 | 'chassisstate': { |
| 59 | 'bus_name': 'xyz.openbmc_project.State.Chassis', |
| 60 | 'object_name': '/xyz/openbmc_project/state/chassis0', |
| 61 | 'interface_name': 'xyz.openbmc_project.State.Chassis', |
| 62 | 'property': 'CurrentPowerState', |
| 63 | }, |
| 64 | 'hoststate': { |
| 65 | 'bus_name': 'xyz.openbmc_project.State.Host', |
| 66 | 'object_name': '/xyz/openbmc_project/state/host0', |
| 67 | 'interface_name': 'xyz.openbmc_project.State.Host', |
| 68 | 'property': 'CurrentHostState', |
Andrew Geissler | cae038e | 2016-12-19 15:29:10 -0600 | [diff] [blame] | 69 | }, |
| 70 | 'bootprogress': { |
George Keishing | 3be0995 | 2017-08-18 09:48:03 -0500 | [diff] [blame] | 71 | 'bus_name': 'xyz.openbmc_project.State.Host', |
| 72 | 'object_name': '/xyz/openbmc_project/state/host0', |
| 73 | 'interface_name': 'xyz.openbmc_project.State.Boot.Progress', |
| 74 | 'property': 'BootProgress', |
Andrew Geissler | cae038e | 2016-12-19 15:29:10 -0600 | [diff] [blame] | 75 | }, |
Andrew Jeffery | 0b6883d | 2017-10-11 22:08:55 +1030 | [diff] [blame] | 76 | 'state' : ['bmcstate', 'chassisstate', 'hoststate'] |
Norman James | b4914ad | 2015-10-26 07:14:29 -0500 | [diff] [blame] | 77 | } |
| 78 | |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 79 | def run_set_property(dbus_bus, dbus_iface, descriptor, args): |
| 80 | mainloop = gobject.MainLoop() |
| 81 | |
| 82 | iface = descriptor['interface_name'] |
| 83 | prop = descriptor['property'] |
| 84 | |
| 85 | if 'monitor' not in descriptor: |
| 86 | dbus_iface.Set(iface, prop, descriptor['value']) |
| 87 | return True |
| 88 | |
| 89 | def property_listener(job, path, unit, state): |
| 90 | if descriptor['monitor'] != unit: |
| 91 | return |
| 92 | |
| 93 | property_listener.success = (state == 'done') |
| 94 | mainloop.quit() |
| 95 | |
| 96 | property_listener.success = True |
| 97 | |
Andrew Jeffery | b43f7ed | 2017-10-12 01:33:59 +1030 | [diff] [blame] | 98 | if args.wait and args.verbose: |
| 99 | pid = Popen(["/bin/journalctl", "-f", "--no-pager"]).pid |
| 100 | |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 101 | if args.wait: |
| 102 | sig_match = dbus_bus.add_signal_receiver(property_listener, "JobRemoved") |
| 103 | |
| 104 | dbus_iface.Set(iface, prop, descriptor['value']) |
| 105 | |
| 106 | if args.wait: |
| 107 | mainloop.run() |
| 108 | sig_match.remove() |
| 109 | |
Andrew Jeffery | b43f7ed | 2017-10-12 01:33:59 +1030 | [diff] [blame] | 110 | if args.wait and args.verbose: |
| 111 | # wait some time for the journal output |
| 112 | time.sleep(args.wait_tune) |
| 113 | os.kill(pid, signal.SIGTERM) |
| 114 | |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 115 | return property_listener.success |
| 116 | |
Andrew Jeffery | cf6a784 | 2017-10-12 14:47:43 +1030 | [diff] [blame^] | 117 | def get_dbus_obj(dbus_bus, bus, obj, args): |
| 118 | if not args.wait: |
| 119 | return dbus_bus.get_object(bus, obj) |
| 120 | |
| 121 | mainloop = gobject.MainLoop() |
| 122 | |
| 123 | def property_listener(job, path, unit, state): |
| 124 | if 'obmc-standby.target' == unit: |
| 125 | mainloop.quit() |
| 126 | |
| 127 | sig_match = dbus_bus.add_signal_receiver(property_listener, "JobRemoved") |
| 128 | try: |
| 129 | return dbus_bus.get_object(bus, obj) |
| 130 | except dbus.exceptions.DBusException as e: |
| 131 | if args.verbose: |
| 132 | pid = Popen(["/bin/journalctl", "-f", "--no-pager"]).pid |
| 133 | |
| 134 | mainloop.run() |
| 135 | |
| 136 | if args.verbose: |
| 137 | os.kill(pid, signal.SIGTERM) |
| 138 | finally: |
| 139 | sig_match.remove() |
| 140 | |
| 141 | return dbus_bus.get_object(bus, obj) |
| 142 | |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 143 | def run_one_command(dbus_bus, descriptor, args): |
Andrew Jeffery | c6b5deb | 2017-10-11 22:18:27 +1030 | [diff] [blame] | 144 | bus = descriptor['bus_name'] |
| 145 | obj = descriptor['object_name'] |
| 146 | iface = descriptor['interface_name'] |
Andrew Jeffery | cf6a784 | 2017-10-12 14:47:43 +1030 | [diff] [blame^] | 147 | dbus_obj = get_dbus_obj(dbus_bus, bus, obj, args) |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 148 | result = None |
Andrew Jeffery | e4dcaef | 2017-10-11 18:08:01 +1030 | [diff] [blame] | 149 | |
Andrew Jeffery | 08e149b | 2017-10-12 01:37:07 +1030 | [diff] [blame] | 150 | if 'property' in descriptor: |
Andrew Jeffery | e4dcaef | 2017-10-11 18:08:01 +1030 | [diff] [blame] | 151 | dbus_iface = dbus.Interface(dbus_obj, "org.freedesktop.DBus.Properties") |
Andrew Jeffery | 08e149b | 2017-10-12 01:37:07 +1030 | [diff] [blame] | 152 | if 'value' in descriptor: |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 153 | result = run_set_property(dbus_bus, dbus_iface, descriptor, args) |
Andrew Jeffery | e4dcaef | 2017-10-11 18:08:01 +1030 | [diff] [blame] | 154 | else: |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 155 | prop = descriptor['property'] |
Andrew Jeffery | e4dcaef | 2017-10-11 18:08:01 +1030 | [diff] [blame] | 156 | dbus_prop = dbus_iface.Get(iface, prop) |
| 157 | print '{:<20}: {}'.format(prop, str(dbus_prop)) |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 158 | result = True |
Andrew Jeffery | e4dcaef | 2017-10-11 18:08:01 +1030 | [diff] [blame] | 159 | else: |
| 160 | dbus_iface = dbus.Interface(dbus_obj, "org.freedesktop.DBus.Properties") |
| 161 | props = dbus_iface.GetAll(iface) |
| 162 | for p in props: |
Andrew Jeffery | bba82bd | 2017-10-11 22:20:30 +1030 | [diff] [blame] | 163 | print "{} = {}".format(p, str(props[p])) |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 164 | result = True |
Andrew Jeffery | e4dcaef | 2017-10-11 18:08:01 +1030 | [diff] [blame] | 165 | |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 166 | return result |
| 167 | |
| 168 | def run_all_commands(dbus_bus, recipe, args): |
Andrew Jeffery | c6b5deb | 2017-10-11 22:18:27 +1030 | [diff] [blame] | 169 | if isinstance(recipe, dict): |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 170 | return run_one_command(dbus_bus, recipe, args) |
Andrew Jeffery | 7b376e7 | 2017-10-11 18:12:05 +1030 | [diff] [blame] | 171 | |
Andrew Jeffery | c6b5deb | 2017-10-11 22:18:27 +1030 | [diff] [blame] | 172 | assert isinstance(recipe, list) |
| 173 | for command in recipe: |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 174 | descriptor = descriptors[command] |
| 175 | if not run_one_command(dbus_bus, descriptor, args): |
| 176 | print "Failed to execute command: {}".format(descriptor) |
| 177 | return False |
| 178 | |
| 179 | return True |
Andrew Jeffery | 7b376e7 | 2017-10-11 18:12:05 +1030 | [diff] [blame] | 180 | |
Andrew Jeffery | 44e6d9c | 2017-10-11 16:57:31 +1030 | [diff] [blame] | 181 | def main(): |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 182 | dbus.mainloop.glib.DBusGMainLoop(set_as_default=True) |
| 183 | |
Andrew Jeffery | 0b6883d | 2017-10-11 22:08:55 +1030 | [diff] [blame] | 184 | parser = argparse.ArgumentParser() |
Andrew Jeffery | b43f7ed | 2017-10-12 01:33:59 +1030 | [diff] [blame] | 185 | parser.add_argument('--verbose', '-v', action='store_true', |
| 186 | help="Verbose output") |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 187 | parser.add_argument('--wait', '-w', action='store_true', |
| 188 | help='Block until the state transition succeeds or fails') |
Andrew Jeffery | b43f7ed | 2017-10-12 01:33:59 +1030 | [diff] [blame] | 189 | parser.add_argument('--wait-tune', '-t', nargs='?', default=8, type=float, |
| 190 | # help='Seconds to wait for journal output to complete after receiving DBus signal', |
| 191 | help=argparse.SUPPRESS) |
Andrew Jeffery | c6b5deb | 2017-10-11 22:18:27 +1030 | [diff] [blame] | 192 | parser.add_argument('recipe', choices=sorted(descriptors.keys())) |
Andrew Jeffery | 0b6883d | 2017-10-11 22:08:55 +1030 | [diff] [blame] | 193 | args = parser.parse_args() |
Andrew Geissler | 8f8cbce | 2017-02-20 16:10:26 -0600 | [diff] [blame] | 194 | |
Andrew Jeffery | bdb7435 | 2017-10-11 17:29:48 +1030 | [diff] [blame] | 195 | dbus_bus = dbus.SystemBus() |
Andrew Jeffery | cf6a784 | 2017-10-12 14:47:43 +1030 | [diff] [blame^] | 196 | |
| 197 | # The only way to get a sensible backtrace with python 2 without stupid |
| 198 | # hoops is to let the uncaught exception handler do the work for you. |
| 199 | # Catching and binding an exception appears to overwrite the stack trace at |
| 200 | # the point of bind. |
| 201 | # |
| 202 | # So, if we're in verbose mode, don't try to catch the DBus exception. That |
| 203 | # way we can understand where it originated. |
| 204 | if args.verbose: |
| 205 | return run_all_commands(dbus_bus, descriptors[args.recipe], args) |
| 206 | |
| 207 | # Otherwise, we don't care about the traceback. Just catch it and print the |
| 208 | # error message. |
Andrew Jeffery | 0b6883d | 2017-10-11 22:08:55 +1030 | [diff] [blame] | 209 | try: |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 210 | return run_all_commands(dbus_bus, descriptors[args.recipe], args) |
Andrew Jeffery | cf6a784 | 2017-10-12 14:47:43 +1030 | [diff] [blame^] | 211 | except dbus.exceptions.DBusException as e: |
| 212 | print >> sys.stderr, "DBus error occurred: {}".format(e.get_dbus_message()) |
Andrew Jeffery | 0b6883d | 2017-10-11 22:08:55 +1030 | [diff] [blame] | 213 | finally: |
| 214 | dbus_bus.close() |
Andrew Jeffery | 44e6d9c | 2017-10-11 16:57:31 +1030 | [diff] [blame] | 215 | |
| 216 | if __name__ == "__main__": |
Andrew Jeffery | de4ed02 | 2017-10-12 01:26:55 +1030 | [diff] [blame] | 217 | sys.exit(0 if main() else 1) |