1#!/usr/bin/env python 2 3import gobject 4import dbus 5import dbus.service 6import dbus.mainloop.glib 7from obmc.dbuslib.bindings import get_dbus, DbusProperties, DbusObjectManager 8 9DBUS_NAME = 'org.openbmc.control.Chassis' 10OBJ_NAME = '/org/openbmc/control/chassis0' 11CONTROL_INTF = 'org.openbmc.Control' 12 13MACHINE_ID = '/etc/machine-id' 14 15POWER_OFF = 0 16POWER_ON = 1 17 18BOOTED = 100 19 20 21class ChassisControlObject(DbusProperties, DbusObjectManager): 22 def getUuid(self): 23 uuid = "" 24 try: 25 with open(MACHINE_ID) as f: 26 data = f.readline().rstrip('\n') 27 if (len(data) == 32): 28 uuid = data 29 else: 30 print "ERROR: UUID is not formatted correctly: " + data 31 except: 32 print "ERROR: Unable to open uuid file: " + MACHINE_ID 33 34 return uuid 35 36 def __init__(self, bus, name): 37 super(ChassisControlObject, self).__init__( 38 conn=bus, 39 object_path=name) 40 ## load utilized objects 41 self.dbus_objects = { 42 'power_control': { 43 'bus_name': 'org.openbmc.control.Power', 44 'object_name': '/org/openbmc/control/power0', 45 'interface_name': 'org.openbmc.control.Power' 46 }, 47 'host_services': { 48 'bus_name': 'org.openbmc.HostServices', 49 'object_name': '/org/openbmc/HostServices', 50 'interface_name': 'org.openbmc.HostServices' 51 }, 52 'settings': { 53 'bus_name': 'org.openbmc.settings.Host', 54 'object_name': '/org/openbmc/settings/host0', 55 'interface_name': 'org.freedesktop.DBus.Properties' 56 }, 57 'systemd': { 58 'bus_name': 'org.freedesktop.systemd1', 59 'object_name': '/org/freedesktop/systemd1', 60 'interface_name': 'org.freedesktop.systemd1.Manager' 61 }, 62 } 63 64 # uuid 65 self.Set(DBUS_NAME, "uuid", self.getUuid()) 66 self.Set(DBUS_NAME, "reboot", 0) 67 68 bus.add_signal_receiver(self.power_button_signal_handler, 69 dbus_interface="org.openbmc.Button", 70 signal_name="Released", 71 path="/org/openbmc/buttons/power0") 72 bus.add_signal_receiver(self.long_power_button_signal_handler, 73 dbus_interface="org.openbmc.Button", 74 signal_name="PressedLong", 75 path="/org/openbmc/buttons/power0") 76 bus.add_signal_receiver(self.softreset_button_signal_handler, 77 dbus_interface="org.openbmc.Button", 78 signal_name="Released", 79 path="/org/openbmc/buttons/reset0") 80 81 bus.add_signal_receiver(self.host_watchdog_signal_handler, 82 dbus_interface="org.openbmc.Watchdog", 83 signal_name="WatchdogError") 84 85 bus.add_signal_receiver(self.SystemStateHandler, 86 signal_name="GotoSystemState") 87 88 def getInterface(self, name): 89 o = self.dbus_objects[name] 90 obj = bus.get_object(o['bus_name'], o['object_name'], introspect=False) 91 return dbus.Interface(obj, o['interface_name']) 92 93 @dbus.service.method(DBUS_NAME, 94 in_signature='', out_signature='') 95 def powerOn(self): 96 print "Turn on power and boot" 97 self.Set(DBUS_NAME, "reboot", 0) 98 intf = self.getInterface('systemd') 99 f = getattr(intf, 'StartUnit') 100 f.call_async('obmc-host-start@0.target', 'replace') 101 return None 102 103 @dbus.service.method(DBUS_NAME, 104 in_signature='', out_signature='') 105 def powerOff(self): 106 print "Turn off power" 107 108 intf = self.getInterface('systemd') 109 f = getattr(intf, 'StartUnit') 110 f.call_async('obmc-host-stop@0.target', 'replace') 111 return None 112 113 @dbus.service.method(DBUS_NAME, 114 in_signature='', out_signature='') 115 def softPowerOff(self): 116 print "Soft off power" 117 intf = self.getInterface('host_services') 118 ## host services will call power off when ready 119 intf.SoftPowerOff() 120 return None 121 122 @dbus.service.method(DBUS_NAME, 123 in_signature='', out_signature='') 124 def reboot(self): 125 print "Rebooting" 126 if self.getPowerState() == POWER_OFF: 127 self.powerOn() 128 else: 129 self.Set(DBUS_NAME, "reboot", 1) 130 self.powerOff() 131 return None 132 133 @dbus.service.method(DBUS_NAME, 134 in_signature='', out_signature='') 135 def softReboot(self): 136 print "Soft Rebooting" 137 if self.getPowerState() == POWER_OFF: 138 self.powerOn() 139 else: 140 self.Set(DBUS_NAME, "reboot", 1) 141 self.softPowerOff() 142 return None 143 144 @dbus.service.method(DBUS_NAME, 145 in_signature='', out_signature='') 146 def quiesce(self): 147 intf = self.getInterface('systemd') 148 f = getattr(intf, 'StartUnit') 149 f.call_async('obmc-host-quiesce@0.target', 'replace') 150 return None 151 152 @dbus.service.method(DBUS_NAME, 153 in_signature='', out_signature='i') 154 def getPowerState(self): 155 intf = self.getInterface('power_control') 156 return intf.getPowerState() 157 158 ## Signal handler 159 160 def SystemStateHandler(self, state_name): 161 if state_name in ["HOST_POWERED_OFF", "HOST_POWERED_ON"]: 162 intf = self.getInterface('settings') 163 intf.Set("org.openbmc.settings.Host", "system_state", state_name) 164 165 if (state_name == "HOST_POWERED_OFF" and self.Get(DBUS_NAME, 166 "reboot") == 1): 167 self.powerOn() 168 169 def power_button_signal_handler(self): 170 # toggle power, power-on / soft-power-off 171 state = self.getPowerState() 172 if state == POWER_OFF: 173 self.powerOn() 174 elif state == POWER_ON: 175 self.softPowerOff() 176 177 def long_power_button_signal_handler(self): 178 print "Long-press button, hard power off" 179 self.powerOff() 180 181 def softreset_button_signal_handler(self): 182 self.softReboot() 183 184 def host_watchdog_signal_handler(self): 185 print "Watchdog Error, Going to quiesce" 186 self.quiesce() 187 188 189if __name__ == '__main__': 190 dbus.mainloop.glib.DBusGMainLoop(set_as_default=True) 191 192 bus = get_dbus() 193 obj = ChassisControlObject(bus, OBJ_NAME) 194 mainloop = gobject.MainLoop() 195 196 obj.unmask_signals() 197 name = dbus.service.BusName(DBUS_NAME, bus) 198 199 print "Running ChassisControlService" 200 mainloop.run() 201 202# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4 203