Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | |
Norman James | 471ab59 | 2015-08-30 22:29:40 -0500 | [diff] [blame] | 3 | import sys |
Adriana Kobylak | 08d3bdb | 2015-10-20 16:59:14 -0500 | [diff] [blame] | 4 | import uuid |
Norman James | 6f8d042 | 2015-09-14 18:48:00 -0500 | [diff] [blame] | 5 | #from gi.repository import GObject |
| 6 | import gobject |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 7 | import dbus |
| 8 | import dbus.service |
| 9 | import dbus.mainloop.glib |
Norman James | 5e792e3 | 2015-10-07 17:36:17 -0500 | [diff] [blame] | 10 | import Openbmc |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 11 | |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 12 | DBUS_NAME = 'org.openbmc.control.Chassis' |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 13 | OBJ_NAME = '/org/openbmc/control/'+sys.argv[1] |
Norman James | a3e47c4 | 2015-10-18 14:43:10 -0500 | [diff] [blame] | 14 | CONTROL_INTF = 'org.openbmc.Control' |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 15 | |
Norman James | 2a3d20b | 2015-08-20 07:09:33 -0500 | [diff] [blame] | 16 | POWER_OFF = 0 |
| 17 | POWER_ON = 1 |
| 18 | |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 19 | BOOTED = 100 |
| 20 | |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 21 | class ChassisControlObject(dbus.service.Object): |
| 22 | def __init__(self,bus,name): |
Norman James | 471ab59 | 2015-08-30 22:29:40 -0500 | [diff] [blame] | 23 | self.dbus_objects = { } |
Norman James | 90baede | 2015-09-02 20:32:49 -0500 | [diff] [blame] | 24 | |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 25 | dbus.service.Object.__init__(self,bus,name) |
Norman James | 90baede | 2015-09-02 20:32:49 -0500 | [diff] [blame] | 26 | ## load utilized objects |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 27 | self.dbus_objects = { |
| 28 | 'power_control' : { |
| 29 | 'bus_name' : 'org.openbmc.control.Power', |
Norman James | a3e47c4 | 2015-10-18 14:43:10 -0500 | [diff] [blame] | 30 | 'object_name' : '/org/openbmc/control/power0', |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 31 | 'interface_name' : 'org.openbmc.control.Power' |
| 32 | }, |
| 33 | 'identify_led' : { |
Norman James | a3e47c4 | 2015-10-18 14:43:10 -0500 | [diff] [blame] | 34 | 'bus_name' : 'org.openbmc.control.led', |
| 35 | 'object_name' : '/org/openbmc/led/IDENTIFY', |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 36 | 'interface_name' : 'org.openbmc.Led' |
| 37 | } |
Norman James | 471ab59 | 2015-08-30 22:29:40 -0500 | [diff] [blame] | 38 | } |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 39 | #self.power_sequence = 0 |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 40 | self.reboot = 0 |
| 41 | self.last_power_state = 0 |
Norman James | 90baede | 2015-09-02 20:32:49 -0500 | [diff] [blame] | 42 | |
Adriana Kobylak | 08d3bdb | 2015-10-20 16:59:14 -0500 | [diff] [blame] | 43 | #uuid |
| 44 | self.id = 0 |
| 45 | |
Norman James | 471ab59 | 2015-08-30 22:29:40 -0500 | [diff] [blame] | 46 | bus.add_signal_receiver(self.power_button_signal_handler, |
| 47 | dbus_interface = "org.openbmc.Button", signal_name = "ButtonPressed", |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 48 | path="/org/openbmc/buttons/PowerButton_0" ) |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 49 | bus.add_signal_receiver(self.host_watchdog_signal_handler, |
| 50 | dbus_interface = "org.openbmc.Watchdog", signal_name = "WatchdogError") |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 51 | bus.add_signal_receiver(self.SystemStateHandler,signal_name = "GotoSystemState") |
Norman James | 471ab59 | 2015-08-30 22:29:40 -0500 | [diff] [blame] | 52 | |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 53 | |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 54 | def getInterface(self,name): |
| 55 | o = self.dbus_objects[name] |
| 56 | obj = bus.get_object(o['bus_name'],o['object_name']) |
| 57 | return dbus.Interface(obj,o['interface_name']) |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 58 | |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 59 | @dbus.service.method(DBUS_NAME, |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 60 | in_signature='', out_signature='s') |
| 61 | def getID(self): |
Adriana Kobylak | 08d3bdb | 2015-10-20 16:59:14 -0500 | [diff] [blame] | 62 | if (self.id==0): |
| 63 | #calculate uuuid |
| 64 | self.id = uuid.uuid1() |
| 65 | return str(self.id) |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 66 | |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 67 | @dbus.service.method(DBUS_NAME, |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 68 | in_signature='', out_signature='') |
| 69 | def setIdentify(self): |
| 70 | print "Turn on identify" |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 71 | intf = self.getInterface('identify_led') |
| 72 | intf.setOn() |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 73 | return None |
| 74 | |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 75 | @dbus.service.method(DBUS_NAME, |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 76 | in_signature='', out_signature='') |
| 77 | def clearIdentify(self): |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 78 | print "Turn on identify" |
| 79 | intf = self.getInterface('identify_led') |
| 80 | intf.setOff() |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 81 | return None |
| 82 | |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 83 | @dbus.service.method(DBUS_NAME, |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 84 | in_signature='', out_signature='') |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 85 | def powerOn(self): |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 86 | print "Turn on power and boot" |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 87 | self.reboot = 0 |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 88 | if (self.getPowerState()==0): |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 89 | intf = self.getInterface('power_control') |
| 90 | intf.setPowerState(POWER_ON) |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 91 | return None |
| 92 | |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 93 | @dbus.service.method(DBUS_NAME, |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 94 | in_signature='', out_signature='') |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 95 | def powerOff(self): |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 96 | print "Turn off power" |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 97 | intf = self.getInterface('power_control') |
| 98 | intf.setPowerState(POWER_OFF) |
| 99 | return None |
| 100 | |
| 101 | @dbus.service.method(DBUS_NAME, |
| 102 | in_signature='', out_signature='') |
| 103 | def softPowerOff(self): |
| 104 | print "Soft off power" |
| 105 | ## Somehow tell host to shutdown via ipmi |
| 106 | return None |
| 107 | |
| 108 | @dbus.service.method(DBUS_NAME, |
| 109 | in_signature='', out_signature='') |
| 110 | def reboot(self): |
| 111 | print "Rebooting" |
| 112 | self.reboot=1 |
| 113 | intf.softPowerOff() |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 114 | return None |
| 115 | |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 116 | @dbus.service.method(DBUS_NAME, |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 117 | in_signature='', out_signature='i') |
| 118 | def getPowerState(self): |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 119 | intf = self.getInterface('power_control') |
| 120 | return intf.getPowerState() |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 121 | |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 122 | @dbus.service.method(DBUS_NAME, |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 123 | in_signature='', out_signature='') |
| 124 | def setDebugMode(self): |
| 125 | return None |
| 126 | |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 127 | @dbus.service.method(DBUS_NAME, |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 128 | in_signature='i', out_signature='') |
| 129 | def setPowerPolicy(self,policy): |
| 130 | return None |
| 131 | |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 132 | |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 133 | ## Signal handler |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 134 | |
| 135 | def SystemStateHandler(self,state_name): |
Norman James | 0e0c788 | 2015-10-19 08:58:22 -0500 | [diff] [blame] | 136 | if (state_name == "HOST_POWERED_OFF" and self.reboot==1): |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 137 | self.powerOn() |
| 138 | |
| 139 | |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 140 | def power_button_signal_handler(self): |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 141 | # toggle power |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 142 | state = self.getPowerState() |
Norman James | 2a3d20b | 2015-08-20 07:09:33 -0500 | [diff] [blame] | 143 | if state == POWER_OFF: |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 144 | self.powerOn() |
Norman James | 2a3d20b | 2015-08-20 07:09:33 -0500 | [diff] [blame] | 145 | elif state == POWER_ON: |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 146 | self.powerOff(); |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 147 | |
| 148 | # TODO: handle long press and reset |
| 149 | |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 150 | def host_watchdog_signal_handler(self): |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 151 | print "Watchdog Error, Hard Rebooting" |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 152 | self.reboot = 1 |
Norman James | 362a80f | 2015-09-14 14:04:39 -0500 | [diff] [blame] | 153 | self.powerOff() |
Norman James | 9e6acf9 | 2015-09-08 07:00:04 -0500 | [diff] [blame] | 154 | |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 155 | |
| 156 | if __name__ == '__main__': |
| 157 | dbus.mainloop.glib.DBusGMainLoop(set_as_default=True) |
| 158 | |
Norman James | 5e792e3 | 2015-10-07 17:36:17 -0500 | [diff] [blame] | 159 | bus = Openbmc.getDBus() |
Norman James | 3f97c5d | 2015-08-26 17:44:14 -0500 | [diff] [blame] | 160 | name = dbus.service.BusName(DBUS_NAME, bus) |
| 161 | obj = ChassisControlObject(bus, OBJ_NAME) |
Norman James | 6f8d042 | 2015-09-14 18:48:00 -0500 | [diff] [blame] | 162 | mainloop = gobject.MainLoop() |
Norman James | 81dbd35 | 2015-08-19 22:44:53 -0500 | [diff] [blame] | 163 | |
Norman James | e276510 | 2015-08-19 22:00:55 -0500 | [diff] [blame] | 164 | print "Running ChassisControlService" |
| 165 | mainloop.run() |
| 166 | |