blob: d6cb8ca8e4c29f77ea74b1d1cf0384d072daea37 [file] [log] [blame]
Simran Basia9f41032012-05-11 14:21:58 -07001# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
Todd Broche505b8d2011-03-21 18:19:54 -07002# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4"""Servo Server."""
Kevin Chengc49494e2016-07-25 12:13:38 -07005import datetime
6import fcntl
Simran Basia9f41032012-05-11 14:21:58 -07007import fnmatch
Todd Broche505b8d2011-03-21 18:19:54 -07008import logging
Simran Basia9f41032012-05-11 14:21:58 -07009import os
Aseda Aboagye1d8477b2017-05-10 17:24:31 -070010import re
Todd Broche505b8d2011-03-21 18:19:54 -070011import SimpleXMLRPCServer
Simran Basia9f41032012-05-11 14:21:58 -070012import subprocess
13import tempfile
Ruben Rodriguez Buchillona5d8b922018-09-03 16:51:03 +080014import threading
Todd Broch7a91c252012-02-03 12:37:45 -080015import time
Wai-Hong Tam1f9e9a72017-05-02 14:14:46 -070016import usb
Todd Broche505b8d2011-03-21 18:19:54 -070017
Wai-Hong Tam4c09eff2017-02-17 11:46:19 -080018import drv as servo_drv
Aaron.Chuang88eff332014-07-31 08:32:00 +080019import bbadc
Simran Basia9ad25e2013-04-23 11:57:00 -070020import bbi2c
Simran Basi5492bde2013-05-16 17:08:47 -070021import bbgpio
Simran Basi949309b2013-05-31 15:12:15 -070022import bbuart
Aseda Aboagyea4922212015-11-20 15:19:08 -080023import ec3po_interface
Todd Broche505b8d2011-03-21 18:19:54 -070024import ftdigpio
25import ftdii2c
Todd Brochdbb09982011-10-02 07:14:26 -070026import ftdi_common
Todd Broch47c43f42011-05-26 15:11:31 -070027import ftdiuart
Rong Changc6c8c022014-08-11 14:07:11 +080028import i2cbus
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -080029import keyboard_handlers
Simran Basie750a342013-03-12 13:45:26 -070030import servo_interfaces
Kevin Cheng16304d12016-07-08 11:56:55 -070031import servo_postinit
Nick Sanders97bc4462016-01-04 15:37:31 -080032import stm32gpio
33import stm32i2c
34import stm32uart
Todd Broche505b8d2011-03-21 18:19:54 -070035
Wai-Hong Tam4c09eff2017-02-17 11:46:19 -080036HwDriverError = servo_drv.hw_driver.HwDriverError
Aseda Aboagyea4922212015-11-20 15:19:08 -080037
Todd Broche505b8d2011-03-21 18:19:54 -070038MAX_I2C_CLOCK_HZ = 100000
39
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -070040
Todd Broche505b8d2011-03-21 18:19:54 -070041class ServodError(Exception):
42 """Exception class for servod."""
43
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -070044
Todd Broche505b8d2011-03-21 18:19:54 -070045class Servod(object):
46 """Main class for Servo debug/controller Daemon."""
Simran Basia9f41032012-05-11 14:21:58 -070047
Kevin Cheng4b4f0022016-09-09 02:37:07 -070048 # This is the key to get the main serial used in the _serialnames dict.
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -070049 MAIN_SERIAL = 'main'
50 SERVO_MICRO_SERIAL = 'servo_micro'
51 CCD_SERIAL = 'ccd'
Kevin Cheng4b4f0022016-09-09 02:37:07 -070052
Ruben Rodriguez Buchillona5d8b922018-09-03 16:51:03 +080053 # Timeout to wait for interfaces to become available again if reinitialization
54 # is taking place. In seconds.
55 INTERFACE_AVAILABILITY_TIMEOUT = 60
56
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -070057 def init_servo_interfaces(self, vendor, product, serialname, interfaces):
Kevin Chengdc3befd2016-07-15 12:34:00 -070058 """Init the servo interfaces with the given interfaces.
59
60 We don't use the self._{vendor,product,serialname} attributes because we
61 want to allow other callers to initialize other interfaces that may not
62 be associated with the initialized attributes (e.g. a servo v4 servod object
63 that wants to also initialize a servo micro interface).
64
65 Args:
66 vendor: USB vendor id of FTDI device.
67 product: USB product id of FTDI device.
68 serialname: String of device serialname/number as defined in FTDI
69 eeprom.
70 interfaces: List of strings of interface types the server will
71 instantiate.
72
73 Raises:
74 ServodError if unable to locate init method for particular interface.
75 """
Mary Ruthven13389642017-02-14 12:15:34 -080076 # If it is a new device add it to the list
Wai-Hong Tam1f9e9a72017-05-02 14:14:46 -070077 device = (vendor, product, serialname)
Mary Ruthven13389642017-02-14 12:15:34 -080078 if device not in self._devices:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -070079 self._devices.append(device)
Mary Ruthven13389642017-02-14 12:15:34 -080080
Kevin Chengdc3befd2016-07-15 12:34:00 -070081 # Extend the interface list if we need to.
82 interfaces_len = len(interfaces)
83 interface_list_len = len(self._interface_list)
84 if interfaces_len > interface_list_len:
85 self._interface_list += [None] * (interfaces_len - interface_list_len)
86
Kevin Chengdc3befd2016-07-15 12:34:00 -070087 for i, interface in enumerate(interfaces):
88 is_ftdi_interface = False
89 if type(interface) is dict:
90 name = interface['name']
91 # Store interface index for those that care about it.
92 interface['index'] = i
93 elif type(interface) is str and interface != 'dummy':
94 name = interface
Wai-Hong Tam564c1702017-04-24 09:23:38 -070095 # It's a FTDI related interface. #0 is reserved for no use.
96 interface = ((i - 1) % ftdi_common.MAX_FTDI_INTERFACES_PER_DEVICE) + 1
Kevin Chengdc3befd2016-07-15 12:34:00 -070097 is_ftdi_interface = True
98 elif type(interface) is str and interface == 'dummy':
99 # 'dummy' reserves the interface for future use. Typically the
100 # interface will be managed by external third-party tools like
101 # openOCD for JTAG or flashrom for SPI. In the case of servo V4,
102 # it serves as a placeholder for servo micro interfaces.
103 continue
104 else:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700105 raise ServodError('Illegal interface type %s' % type(interface))
Kevin Chengdc3befd2016-07-15 12:34:00 -0700106
107 # servos with multiple FTDI are guaranteed to have contiguous USB PIDs
Wai-Hong Tam564c1702017-04-24 09:23:38 -0700108 product_increment = 0
109 if is_ftdi_interface:
110 product_increment = (i - 1) / ftdi_common.MAX_FTDI_INTERFACES_PER_DEVICE
111 if product_increment:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700112 self._logger.info('Use the next FTDI part @ pid = 0x%04x',
Wai-Hong Tam564c1702017-04-24 09:23:38 -0700113 product + product_increment)
Kevin Chengdc3befd2016-07-15 12:34:00 -0700114
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700115 self._logger.info('Initializing interface %d to %s', i, name)
Kevin Chengdc3befd2016-07-15 12:34:00 -0700116 try:
117 func = getattr(self, '_init_%s' % name)
118 except AttributeError:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700119 raise ServodError('Unable to locate init for interface %s' % name)
Wai-Hong Tam564c1702017-04-24 09:23:38 -0700120 result = func(vendor, product + product_increment, serialname, interface)
Kevin Chengdc3befd2016-07-15 12:34:00 -0700121
122 if isinstance(result, tuple):
123 result_len = len(result)
Wai-Hong Tamd8a94d62017-04-28 10:11:51 -0700124 self._interface_list[i:(i + result_len)] = result
Kevin Chengdc3befd2016-07-15 12:34:00 -0700125 else:
Wai-Hong Tamd8a94d62017-04-28 10:11:51 -0700126 self._interface_list[i] = result
Kevin Chengdc3befd2016-07-15 12:34:00 -0700127
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700128 def __init__(self, config, vendor, product, serialname=None, interfaces=None,
129 board='', version=None, usbkm232=None):
Todd Broche505b8d2011-03-21 18:19:54 -0700130 """Servod constructor.
131
132 Args:
133 config: instance of SystemConfig containing all controls for
134 particular Servod invocation
135 vendor: usb vendor id of FTDI device
136 product: usb product id of FTDI device
Todd Brochad034442011-05-25 15:05:29 -0700137 serialname: string of device serialname/number as defined in FTDI eeprom.
Todd Brochdbb09982011-10-02 07:14:26 -0700138 interfaces: list of strings of interface types the server will instantiate
Simran Basia23c1392013-08-06 14:59:10 -0700139 version: String. Servo board version. Examples: servo_v1, servo_v2,
140 servo_v2_r0, servo_v3
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800141 usbkm232: String. Optional. Path to USB-KM232 device which allow for
Kevin Chengdc3befd2016-07-15 12:34:00 -0700142 sending keyboard commands to DUTs that do not have built in
Wai-Hong Tam7b9f2992017-10-24 16:07:14 -0700143 keyboards. Used in FAFT tests. Use None for on board AVR MCU.
144 e.g. '/dev/ttyUSB0' or None.
Todd Brochdbb09982011-10-02 07:14:26 -0700145
146 Raises:
147 ServodError: if unable to locate init method for particular interface
Todd Broche505b8d2011-03-21 18:19:54 -0700148 """
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700149 self._logger = logging.getLogger('Servod')
150 self._logger.debug('')
Ruben Rodriguez Buchillona5d8b922018-09-03 16:51:03 +0800151 self._ifaces_available = threading.Event()
152 # Initially interfaces should be available.
153 self._ifaces_available.set()
Todd Broche505b8d2011-03-21 18:19:54 -0700154 self._vendor = vendor
155 self._product = product
Mary Ruthven13389642017-02-14 12:15:34 -0800156 self._devices = []
Kevin Cheng4b4f0022016-09-09 02:37:07 -0700157 self._serialnames = {self.MAIN_SERIAL: serialname}
Todd Broche505b8d2011-03-21 18:19:54 -0700158 self._syscfg = config
159 # list of objects (Fi2c, Fgpio) to physical interfaces (gpio, i2c) that ftdi
160 # interfaces are mapped to
161 self._interface_list = []
162 # Dict of Dict to map control name, function name to to tuple (params, drv)
163 # Ex) _drv_dict[name]['get'] = (params, drv)
164 self._drv_dict = {}
J. Richard Barnettee2820552013-03-14 16:13:46 -0700165 self._board = board
Wai-Hong Tam416cf612017-09-19 11:39:21 -0700166 self._base_board = ''
Simran Basia23c1392013-08-06 14:59:10 -0700167 self._version = version
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800168 self._usbkm232 = usbkm232
Todd Brochdbb09982011-10-02 07:14:26 -0700169 if not interfaces:
Todd Brochb21d8042014-05-15 12:54:54 -0700170 try:
171 interfaces = servo_interfaces.INTERFACE_BOARDS[board][vendor][product]
172 except KeyError:
173 interfaces = servo_interfaces.INTERFACE_DEFAULTS[vendor][product]
Dino Lic89d8c82018-01-11 09:56:47 +0800174 self._interfaces = interfaces
Todd Brochdbb09982011-10-02 07:14:26 -0700175
Kevin Chengdc3befd2016-07-15 12:34:00 -0700176 self.init_servo_interfaces(vendor, product, serialname, interfaces)
Kevin Cheng16304d12016-07-08 11:56:55 -0700177 servo_postinit.post_init(self)
Danny Chan662b6022015-11-04 17:34:53 -0800178
Mary Ruthven13389642017-02-14 12:15:34 -0800179 def reinitialize(self):
180 """Reinitialize all interfaces that support reinitialization"""
Mary Ruthven13389642017-02-14 12:15:34 -0800181 for i, interface in enumerate(self._interface_list):
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700182 if hasattr(interface, 'reinitialize'):
183 interface.reinitialize()
184 else:
185 self._logger.debug('interface %d has no reset functionality', i)
Ruben Rodriguez Buchillona5d8b922018-09-03 16:51:03 +0800186 # Indicate interfaces are safe to use again.
187 self._ifaces_available.set()
Mary Ruthven13389642017-02-14 12:15:34 -0800188
Wai-Hong Tam4544c302017-05-24 19:44:53 -0700189 def get_servo_interfaces(self, position, size):
190 """Get the list of servo interfaces.
191
192 Args:
193 position: The index the first interface to get.
194 size: The number of the interfaces.
195 """
196 return self._interface_list[position:(position + size)]
197
198 def set_servo_interfaces(self, position, interfaces):
199 """Set the list of servo interfaces.
200
201 Args:
202 position: The index the first interface to set.
203 interfaces: The list of interfaces to set.
204 """
205 size = len(interfaces)
206 self._interface_list[position:(position + size)] = interfaces
207
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800208 def _init_keyboard_handler(self, servo, board=''):
209 """Initialize the correct keyboard handler for board.
210
Kevin Chengdc3befd2016-07-15 12:34:00 -0700211 Args:
212 servo: servo object.
213 board: string, board name.
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800214
Kevin Chengdc3befd2016-07-15 12:34:00 -0700215 Returns:
Wai-Hong Tam7b9f2992017-10-24 16:07:14 -0700216 keyboard handler object, or None if no keyboard supported.
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800217 """
218 if board == 'parrot':
219 return keyboard_handlers.ParrotHandler(servo)
220 elif board == 'stout':
221 return keyboard_handlers.StoutHandler(servo)
PeggyChuang4f07d872015-08-07 12:11:38 +0800222 elif board in ('buddy', 'cranky', 'guado', 'jecht', 'mccloud', 'monroe',
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700223 'ninja', 'nyan_kitty', 'panther', 'rikku', 'stumpy', 'sumo',
224 'tidus', 'tricky', 'veyron_fievel', 'veyron_mickey',
Shelley Chen94cd2352017-07-26 11:36:45 -0700225 'veyron_rialto', 'veyron_tiger', 'zako'):
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800226 if self._usbkm232 is None:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700227 logging.info('No device path specified for usbkm232 handler. Use '
228 'the servo atmega chip to handle.')
Wai-Hong Tam7b9f2992017-10-24 16:07:14 -0700229
Danny Chan662b6022015-11-04 17:34:53 -0800230 # Use servo onboard keyboard emulator.
Wai-Hong Tam7b9f2992017-10-24 16:07:14 -0700231 if not self._syscfg.is_control('atmega_rst'):
232 logging.warn('No atmega in servo board. So no keyboard support.')
233 return None
234
Nick Sanders78423782015-11-09 14:28:19 -0800235 self.set('atmega_rst', 'on')
Nick Sandersbc836282015-12-08 21:19:23 -0800236 self.set('at_hwb', 'off')
Nick Sanders78423782015-11-09 14:28:19 -0800237 self.set('atmega_rst', 'off')
Danny Chan662b6022015-11-04 17:34:53 -0800238 self._usbkm232 = self.get('atmega_pty')
Wai-Hong Tam7b9f2992017-10-24 16:07:14 -0700239
Kevin Cheng810fc782016-11-01 12:36:46 -0700240 # We don't need to set the atmega uart settings if we're a servo v4.
Aseda Aboagye1d8477b2017-05-10 17:24:31 -0700241 if 'servo_v4' not in self._version:
Kevin Cheng810fc782016-11-01 12:36:46 -0700242 self.set('atmega_baudrate', '9600')
243 self.set('atmega_bits', 'eight')
244 self.set('atmega_parity', 'none')
245 self.set('atmega_sbits', 'one')
246 self.set('usb_mux_sel4', 'on')
247 self.set('usb_mux_oe4', 'on')
248 # Allow atmega bootup time.
249 time.sleep(1.0)
Wai-Hong Tam7b9f2992017-10-24 16:07:14 -0700250
Danny Chan662b6022015-11-04 17:34:53 -0800251 self._logger.info('USBKM232: %s', self._usbkm232)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800252 return keyboard_handlers.USBkm232Handler(servo, self._usbkm232)
253 else:
Tom Wai-Hong Tamd64164c2015-04-29 07:59:45 +0800254 # The following boards don't use Chrome EC.
255 if board in ('alex', 'butterfly', 'lumpy', 'zgb'):
256 return keyboard_handlers.MatrixKeyboardHandler(servo)
257 return keyboard_handlers.ChromeECHandler(servo)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800258
Ruben Rodriguez Buchillona16374b2018-06-20 16:45:00 -0700259 def close(self):
260 """Servod turn down logic."""
261 for i, interface in enumerate(self._interface_list):
262 self._logger.info('Turning down interface %d' % i)
263 if hasattr(interface, 'close'):
264 interface.close()
Todd Broch3ec8df02012-11-20 10:53:03 -0800265
Kevin Chengdc3befd2016-07-15 12:34:00 -0700266 def _init_ftdi_dummy(self, vendor, product, serialname, interface):
Kevin Cheng042f4932016-07-19 10:46:00 -0700267 """Dummy interface for ftdi devices.
268
269 This is a dummy function specifically for ftdi devices to not initialize
270 anything but to help pad the interface list.
271
272 Returns:
273 None.
274 """
275 return None
276
Kevin Chengdc3befd2016-07-15 12:34:00 -0700277 def _init_ftdi_gpio(self, vendor, product, serialname, interface):
Todd Broche505b8d2011-03-21 18:19:54 -0700278 """Initialize gpio driver interface and open for use.
279
280 Args:
281 interface: interface number of FTDI device to use.
282
283 Returns:
284 Instance object of interface.
Todd Broch6de9dc62012-04-09 15:23:53 -0700285
286 Raises:
287 ServodError: If init fails
Todd Broche505b8d2011-03-21 18:19:54 -0700288 """
Kevin Chengdc3befd2016-07-15 12:34:00 -0700289 fobj = ftdigpio.Fgpio(vendor, product, interface, serialname)
Todd Broch6de9dc62012-04-09 15:23:53 -0700290 try:
291 fobj.open()
292 except ftdigpio.FgpioError as e:
293 raise ServodError('Opening gpio interface. %s ( %d )' % (e.msg, e.value))
294
Todd Broche505b8d2011-03-21 18:19:54 -0700295 return fobj
296
Kevin Chengdc3befd2016-07-15 12:34:00 -0700297 def _init_stm32_uart(self, vendor, product, serialname, interface):
Nick Sanders97bc4462016-01-04 15:37:31 -0800298 """Initialize stm32 uart interface and open for use
299
300 Note, the uart runs in a separate thread. Users wishing to
301 interact with it will query control for the pty's pathname and connect
302 with their favorite console program. For example:
303 cu -l /dev/pts/22
304
305 Args:
306 interface: dict of interface parameters.
307
308 Returns:
309 Instance object of interface
310
311 Raises:
312 ServodError: Raised on init failure.
313 """
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700314 self._logger.info('Suart: interface: %s' % interface)
315 sobj = stm32uart.Suart(vendor, product, interface['interface'], serialname)
Nick Sanders97bc4462016-01-04 15:37:31 -0800316
317 try:
318 sobj.run()
319 except stm32uart.SuartError as e:
320 raise ServodError('Running uart interface. %s ( %d )' % (e.msg, e.value))
321
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700322 self._logger.info('%s' % sobj.get_pty())
Nick Sanders97bc4462016-01-04 15:37:31 -0800323 return sobj
324
Kevin Chengdc3befd2016-07-15 12:34:00 -0700325 def _init_stm32_gpio(self, vendor, product, serialname, interface):
Nick Sanders97bc4462016-01-04 15:37:31 -0800326 """Initialize stm32 gpio interface.
327 Args:
Wai-Hong Tam564c1702017-04-24 09:23:38 -0700328 interface: dict of interface parameters.
Nick Sanders97bc4462016-01-04 15:37:31 -0800329
330 Returns:
331 Instance object of interface
332
333 Raises:
334 SgpioError: Raised on init failure.
335 """
Kevin Cheng71a046f2016-06-13 16:37:58 -0700336 interface_number = interface
337 # Interface could be a dict.
338 if type(interface) is dict:
339 interface_number = interface['interface']
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700340 self._logger.info('Sgpio: interface: %s' % interface_number)
Kevin Chengdc3befd2016-07-15 12:34:00 -0700341 return stm32gpio.Sgpio(vendor, product, interface_number, serialname)
Nick Sanders97bc4462016-01-04 15:37:31 -0800342
Kevin Chengdc3befd2016-07-15 12:34:00 -0700343 def _init_stm32_i2c(self, vendor, product, serialname, interface):
Nick Sanders97bc4462016-01-04 15:37:31 -0800344 """Initialize stm32 USB to I2C bridge interface and open for use
345
346 Args:
Wai-Hong Tam564c1702017-04-24 09:23:38 -0700347 interface: dict of interface parameters.
Nick Sanders97bc4462016-01-04 15:37:31 -0800348
349 Returns:
350 Instance object of interface.
351
352 Raises:
353 Si2cError: Raised on init failure.
354 """
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700355 self._logger.info('Si2cBus: interface: %s' % interface)
Nick Sandersa3649712016-03-01 16:53:52 -0800356 port = interface.get('port', 0)
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700357 return stm32i2c.Si2cBus(vendor, product, interface['interface'], port=port,
358 serialname=serialname)
Nick Sanders97bc4462016-01-04 15:37:31 -0800359
Kevin Chengdc3befd2016-07-15 12:34:00 -0700360 def _init_bb_adc(self, vendor, product, serialname, interface):
Aaron.Chuang88eff332014-07-31 08:32:00 +0800361 """Initalize beaglebone ADC interface."""
362 return bbadc.BBadc()
363
Kevin Chengdc3befd2016-07-15 12:34:00 -0700364 def _init_bb_gpio(self, vendor, product, serialname, interface):
Simran Basie750a342013-03-12 13:45:26 -0700365 """Initalize beaglebone gpio interface."""
Simran Basi5492bde2013-05-16 17:08:47 -0700366 return bbgpio.BBgpio()
Simran Basie750a342013-03-12 13:45:26 -0700367
Kevin Chengdc3befd2016-07-15 12:34:00 -0700368 def _init_ftdi_i2c(self, vendor, product, serialname, interface):
Todd Broche505b8d2011-03-21 18:19:54 -0700369 """Initialize i2c interface and open for use.
370
371 Args:
372 interface: interface number of FTDI device to use
373
374 Returns:
375 Instance object of interface
Todd Broch6de9dc62012-04-09 15:23:53 -0700376
377 Raises:
378 ServodError: If init fails
Todd Broche505b8d2011-03-21 18:19:54 -0700379 """
Kevin Chengdc3befd2016-07-15 12:34:00 -0700380 fobj = ftdii2c.Fi2c(vendor, product, interface, serialname)
Todd Broch6de9dc62012-04-09 15:23:53 -0700381 try:
382 fobj.open()
383 except ftdii2c.Fi2cError as e:
384 raise ServodError('Opening i2c interface. %s ( %d )' % (e.msg, e.value))
385
Todd Broche505b8d2011-03-21 18:19:54 -0700386 # Set the frequency of operation of the i2c bus.
387 # TODO(tbroch) make configureable
388 fobj.setclock(MAX_I2C_CLOCK_HZ)
Todd Broch6de9dc62012-04-09 15:23:53 -0700389
Todd Broche505b8d2011-03-21 18:19:54 -0700390 return fobj
391
Simran Basie750a342013-03-12 13:45:26 -0700392 # TODO (sbasi) crbug.com/187489 - Implement bb_i2c.
393 def _init_bb_i2c(self, interface):
394 """Initalize beaglebone i2c interface."""
Simran Basia9ad25e2013-04-23 11:57:00 -0700395 return bbi2c.BBi2c(interface)
Simran Basie750a342013-03-12 13:45:26 -0700396
Kevin Chengdc3befd2016-07-15 12:34:00 -0700397 def _init_dev_i2c(self, vendor, product, serialname, interface):
Rong Changc6c8c022014-08-11 14:07:11 +0800398 """Initalize Linux i2c-dev interface."""
399 return i2cbus.I2CBus('/dev/i2c-%d' % interface['bus_num'])
400
Kevin Chengdc3befd2016-07-15 12:34:00 -0700401 def _init_ftdi_uart(self, vendor, product, serialname, interface):
Simran Basie750a342013-03-12 13:45:26 -0700402 """Initialize ftdi uart inteface and open for use
Todd Broch47c43f42011-05-26 15:11:31 -0700403
404 Note, the uart runs in a separate thread (pthreads). Users wishing to
405 interact with it will query control for the pty's pathname and connect
406 with there favorite console program. For example:
407 cu -l /dev/pts/22
408
409 Args:
410 interface: interface number of FTDI device to use
411
412 Returns:
413 Instance object of interface
Todd Broch6de9dc62012-04-09 15:23:53 -0700414
415 Raises:
416 ServodError: If init fails
Todd Broch47c43f42011-05-26 15:11:31 -0700417 """
Kevin Chengdc3befd2016-07-15 12:34:00 -0700418 fobj = ftdiuart.Fuart(vendor, product, interface, serialname)
Todd Broch6de9dc62012-04-09 15:23:53 -0700419 try:
420 fobj.run()
421 except ftdiuart.FuartError as e:
422 raise ServodError('Running uart interface. %s ( %d )' % (e.msg, e.value))
423
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700424 self._logger.info('%s' % fobj.get_pty())
Todd Broch47c43f42011-05-26 15:11:31 -0700425 return fobj
426
Simran Basie750a342013-03-12 13:45:26 -0700427 # TODO (sbasi) crbug.com/187492 - Implement bbuart.
Kevin Chengdc3befd2016-07-15 12:34:00 -0700428 def _init_bb_uart(self, vendor, product, serialname, interface):
Simran Basie750a342013-03-12 13:45:26 -0700429 """Initalize beaglebone uart interface."""
Simran Basi949309b2013-05-31 15:12:15 -0700430 logging.debug('UART INTERFACE: %s', interface)
431 return bbuart.BBuart(interface)
Simran Basie750a342013-03-12 13:45:26 -0700432
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700433 def _init_ftdi_gpiouart(self, vendor, product, serialname, interface):
Todd Broch888da782011-10-07 14:29:09 -0700434 """Initialize special gpio + uart interface and open for use
435
436 Note, the uart runs in a separate thread (pthreads). Users wishing to
437 interact with it will query control for the pty's pathname and connect
438 with there favorite console program. For example:
439 cu -l /dev/pts/22
440
441 Args:
442 interface: interface number of FTDI device to use
443
444 Returns:
445 Instance objects of interface
Todd Broch6de9dc62012-04-09 15:23:53 -0700446
447 Raises:
448 ServodError: If init fails
Todd Broch888da782011-10-07 14:29:09 -0700449 """
Kevin Chengce7dafd2016-08-02 11:11:38 -0700450 fgpio = self._init_ftdi_gpio(vendor, product, serialname, interface)
Kevin Chengdc3befd2016-07-15 12:34:00 -0700451 fuart = ftdiuart.Fuart(vendor, product, interface, serialname, fgpio._fc)
Todd Broch6de9dc62012-04-09 15:23:53 -0700452 try:
453 fuart.run()
454 except ftdiuart.FuartError as e:
455 raise ServodError('Running uart interface. %s ( %d )' % (e.msg, e.value))
456
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700457 self._logger.info('uart pty: %s' % fuart.get_pty())
Todd Broch888da782011-10-07 14:29:09 -0700458 return fgpio, fuart
459
Kevin Chengdc3befd2016-07-15 12:34:00 -0700460 def _init_ec3po_uart(self, vendor, product, serialname, interface):
Aseda Aboagyea4922212015-11-20 15:19:08 -0800461 """Initialize EC-3PO console interpreter interface.
462
463 Args:
464 interface: A dictionary representing the interface.
465
466 Returns:
467 An EC3PO object representing the EC-3PO interface or None if there's no
468 interface for the USB PD UART.
469 """
Wai-Hong Tam6c0fa592017-04-21 12:41:33 -0700470 raw_uart_name = interface['raw_pty']
Nick Sanders116ed9e2018-03-09 19:05:16 -0800471 raw_uart_source = interface['source']
Wai-Hong Tam6c0fa592017-04-21 12:41:33 -0700472 if self._syscfg.is_control(raw_uart_name):
Nick Sanders97bc4462016-01-04 15:37:31 -0800473 raw_ec_uart = self.get(raw_uart_name)
Nick Sanders116ed9e2018-03-09 19:05:16 -0800474 return ec3po_interface.EC3PO(raw_ec_uart, raw_uart_source)
Aseda Aboagyea4922212015-11-20 15:19:08 -0800475 else:
Wai-Hong Tam6c0fa592017-04-21 12:41:33 -0700476 # The overlay doesn't have the raw PTY defined, therefore we can skip
477 # initializing this interface since no control relies on it.
478 self._logger.debug(
479 'Skip initializing EC3PO for %s, no control specified.',
480 raw_uart_name)
481 return None
Aseda Aboagyea4922212015-11-20 15:19:08 -0800482
Tom Wai-Hong Tam28f0a5f2012-08-21 12:49:57 +0800483 def _camel_case(self, string):
484 output = ''
485 for s in string.split('_'):
486 if output:
487 output += s.capitalize()
488 else:
489 output = s
490 return output
491
Wai-Hong Tam4544c302017-05-24 19:44:53 -0700492 def clear_cached_drv(self):
493 """Clear the cached drivers.
494
495 The drivers are cached in the Dict _drv_dict when a control is got or set.
496 When the servo interfaces are relocated, the cached values may become wrong.
497 Should call this method to clear the cached values.
498 """
499 self._drv_dict = {}
500
Todd Broche505b8d2011-03-21 18:19:54 -0700501 def _get_param_drv(self, control_name, is_get=True):
502 """Get access to driver for a given control.
503
504 Note, some controls have different parameter dictionaries for 'getting' the
505 control's value versus 'setting' it. Boolean is_get distinguishes which is
506 being requested.
507
508 Args:
509 control_name: string name of control
510 is_get: boolean to determine
511
512 Returns:
513 tuple (param, drv) where:
514 param: param dictionary for control
515 drv: instance object of driver for particular control
516
517 Raises:
518 ServodError: Error occurred while examining params dict
519 """
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700520 self._logger.debug('')
Todd Broche505b8d2011-03-21 18:19:54 -0700521 # if already setup just return tuple from driver dict
522 if control_name in self._drv_dict:
523 if is_get and ('get' in self._drv_dict[control_name]):
524 return self._drv_dict[control_name]['get']
525 if not is_get and ('set' in self._drv_dict[control_name]):
526 return self._drv_dict[control_name]['set']
527
528 params = self._syscfg.lookup_control_params(control_name, is_get)
529 if 'drv' not in params:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700530 self._logger.error('Unable to determine driver for %s' % control_name)
Todd Broche505b8d2011-03-21 18:19:54 -0700531 raise ServodError("'drv' key not found in params dict")
532 if 'interface' not in params:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700533 self._logger.error('Unable to determine interface for %s' % control_name)
Todd Broche505b8d2011-03-21 18:19:54 -0700534 raise ServodError("'interface' key not found in params dict")
Simran Basi668be0e2013-08-07 11:54:50 -0700535
Aseda Aboagye1d8477b2017-05-10 17:24:31 -0700536 # Find the candidate servos. Using servo_v4 with a servo_micro connected as
537 # an example, the following shows the priority for selecting the interface.
538 #
539 # 1. The full name. (e.g. - 'servo_v4_with_servo_micro_interface')
540 # 2. servo_micro_interface
541 # 3. servo_v4_interface
542 # 4. Fallback to the default, interface.
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700543 candidates = [self._version]
Aseda Aboagye1d8477b2017-05-10 17:24:31 -0700544 candidates.extend(reversed(self._version.split('_with_')))
545
546 interface_id = 'unknown'
547 for c in candidates:
548 interface_name = '%s_interface' % c
549 if interface_name in params:
550 interface_id = params[interface_name]
551 self._logger.debug('Using %s parameter.' % interface_name)
552 break
553
554 # Use the default interface value if we couldn't find a more specific
555 # interface.
556 if interface_id == 'unknown':
557 interface_id = params['interface']
558 self._logger.debug('Using default interface parameter.')
559
J. Richard Barnette275d9fd2014-02-11 14:38:54 -0800560 if interface_id == 'servo':
561 interface = self
Simran Basi668be0e2013-08-07 11:54:50 -0700562 else:
Wai-Hong Tam564c1702017-04-24 09:23:38 -0700563 index = int(interface_id)
J. Richard Barnette275d9fd2014-02-11 14:38:54 -0800564 interface = self._interface_list[index]
Simran Basi668be0e2013-08-07 11:54:50 -0700565
Todd Broche505b8d2011-03-21 18:19:54 -0700566 drv_name = params['drv']
Wai-Hong Tam4c09eff2017-02-17 11:46:19 -0800567 drv_module = getattr(servo_drv, drv_name)
Tom Wai-Hong Tam28f0a5f2012-08-21 12:49:57 +0800568 drv_class = getattr(drv_module, self._camel_case(drv_name))
Todd Broche505b8d2011-03-21 18:19:54 -0700569 drv = drv_class(interface, params)
570 if control_name not in self._drv_dict:
571 self._drv_dict[control_name] = {}
572 if is_get:
573 self._drv_dict[control_name]['get'] = (params, drv)
574 else:
575 self._drv_dict[control_name]['set'] = (params, drv)
576 return (params, drv)
577
578 def doc_all(self):
579 """Return all documenation for controls.
580
581 Returns:
582 string of <doc> text in config file (xml) and the params dictionary for
583 all controls.
584
585 For example:
586 warm_reset :: Reset the device warmly
587 ------------------------> {'interface': '1', 'map': 'onoff_i', ... }
588 """
589 return self._syscfg.display_config()
590
591 def doc(self, name):
592 """Retreive doc string in system config file for given control name.
593
594 Args:
595 name: name string of control to get doc string
596
597 Returns:
598 doc string of name
599
600 Raises:
601 NameError: if fails to locate control
602 """
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700603 self._logger.debug('name(%s)' % (name))
Todd Broche505b8d2011-03-21 18:19:54 -0700604 if self._syscfg.is_control(name):
605 return self._syscfg.get_control_docstring(name)
606 else:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700607 raise NameError('No control %s' % name)
Todd Broche505b8d2011-03-21 18:19:54 -0700608
Ruben Rodriguez Buchillon247d88f2018-08-03 18:07:01 +0800609 def safe_switch_usbkey_power(self, power_state, _=None):
Kevin Cheng5595b342016-09-29 15:51:01 -0700610 """Toggle the usb power safely.
611
Kevin Chengc49494e2016-07-25 12:13:38 -0700612 Args:
Kevin Cheng5595b342016-09-29 15:51:01 -0700613 power_state: The setting to set for the usbkey power.
Ruben Rodriguez Buchillon247d88f2018-08-03 18:07:01 +0800614 _: to conform to current API
Kevin Chengc49494e2016-07-25 12:13:38 -0700615
Kevin Cheng5595b342016-09-29 15:51:01 -0700616 Returns:
617 An empty string to appease the xmlrpc gods.
618 """
Ruben Rodriguez Buchillon247d88f2018-08-03 18:07:01 +0800619 self.set('image_usbkey_pwr', power_state)
Kevin Cheng5595b342016-09-29 15:51:01 -0700620 return ''
621
Ruben Rodriguez Buchillon247d88f2018-08-03 18:07:01 +0800622 def safe_switch_usbkey(self, mux_direction, _=0):
Kevin Cheng5595b342016-09-29 15:51:01 -0700623 """Toggle the usb direction safely.
624
Kevin Cheng5595b342016-09-29 15:51:01 -0700625 Args:
Wai-Hong Tamf93f9a22018-02-06 14:24:46 -0800626 mux_direction: "servo_sees_usbkey" or "dut_sees_usbkey".
Ruben Rodriguez Buchillon247d88f2018-08-03 18:07:01 +0800627 _: to conform to current API
Kevin Cheng5595b342016-09-29 15:51:01 -0700628
629 Returns:
630 An empty string to appease the xmlrpc gods.
631 """
Ruben Rodriguez Buchillon247d88f2018-08-03 18:07:01 +0800632 self.set('image_usbkey_direction', mux_direction)
Kevin Cheng5595b342016-09-29 15:51:01 -0700633 return ''
634
Ruben Rodriguez Buchillon68958f82018-08-03 18:24:28 +0800635 def probe_host_usb_dev(self, _=0):
Simran Basia9f41032012-05-11 14:21:58 -0700636 """Probe the USB disk device plugged in the servo from the host side.
637
Kevin Cheng5595b342016-09-29 15:51:01 -0700638 Args:
Ruben Rodriguez Buchillon68958f82018-08-03 18:24:28 +0800639 _: to conform to current API
Kevin Cheng5595b342016-09-29 15:51:01 -0700640
Simran Basia9f41032012-05-11 14:21:58 -0700641 Returns:
Kevin Chengc49494e2016-07-25 12:13:38 -0700642 USB disk path if one and only one USB disk path is found, otherwise an
Simran Basia9f41032012-05-11 14:21:58 -0700643 """
Ruben Rodriguez Buchillon68958f82018-08-03 18:24:28 +0800644 return self.get('image_usbkey_dev')
Kevin Chengc49494e2016-07-25 12:13:38 -0700645
Ruben Rodriguez Buchillon68958f82018-08-03 18:24:28 +0800646 def download_image_to_usb(self, image_path, _=0):
Simran Basia9f41032012-05-11 14:21:58 -0700647 """Download image and save to the USB device found by probe_host_usb_dev.
648 If the image_path is a URL, it will download this url to the USB path;
649 otherwise it will simply copy the image_path's contents to the USB path.
650
651 Args:
652 image_path: path or url to the recovery image.
Ruben Rodriguez Buchillon68958f82018-08-03 18:24:28 +0800653 _: to conform to current API
Simran Basia9f41032012-05-11 14:21:58 -0700654
655 Returns:
656 True|False: True if process completed successfully, False if error
Ruben Rodriguez Buchillon4e00f0e2018-08-28 15:21:47 +0800657 occurred.
Simran Basia9f41032012-05-11 14:21:58 -0700658 """
Simran Basia9f41032012-05-11 14:21:58 -0700659 try:
Ruben Rodriguez Buchillon4e00f0e2018-08-28 15:21:47 +0800660 self.set('download_image_to_usb_dev', image_path)
661 return True
662 except Exception:
Simran Basia9f41032012-05-11 14:21:58 -0700663 return False
Simran Basia9f41032012-05-11 14:21:58 -0700664
665 def make_image_noninteractive(self):
666 """Makes the recovery image noninteractive.
667
668 A noninteractive image will reboot automatically after installation
669 instead of waiting for the USB device to be removed to initiate a system
670 reboot.
671
672 Mounts partition 1 of the image stored on usb_dev and creates a file
673 called "non_interactive" so that the image will become noninteractive.
674
675 Returns:
676 True|False: True if process completed successfully, False if error
677 occurred.
678 """
679 result = True
Ruben Rodriguez Buchillon68958f82018-08-03 18:24:28 +0800680 usb_dev = self.get('image_usbkey_dev')
Simran Basia9f41032012-05-11 14:21:58 -0700681 if not usb_dev:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700682 self._logger.error('No usb device connected to servo')
Simran Basia9f41032012-05-11 14:21:58 -0700683 return False
684 # Create TempDirectory
685 tmpdir = tempfile.mkdtemp()
686 if tmpdir:
687 # Mount drive to tmpdir.
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700688 partition_1 = '%s1' % usb_dev
689 rc = subprocess.call(['mount', partition_1, tmpdir])
Simran Basia9f41032012-05-11 14:21:58 -0700690 if rc == 0:
691 # Create file 'non_interactive'
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700692 non_interactive_file = os.path.join(tmpdir, 'non_interactive')
Simran Basia9f41032012-05-11 14:21:58 -0700693 try:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700694 open(non_interactive_file, 'w').close()
Simran Basia9f41032012-05-11 14:21:58 -0700695 except IOError as e:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700696 self._logger.error('Failed to create file %s : %s ( %d )',
Simran Basia9f41032012-05-11 14:21:58 -0700697 non_interactive_file, e.strerror, e.errno)
698 result = False
699 except BaseException as e:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700700 self._logger.error('Unexpected Exception creating file %s : %s',
Simran Basia9f41032012-05-11 14:21:58 -0700701 non_interactive_file, str(e))
702 result = False
703 # Unmount drive regardless if file creation worked or not.
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700704 rc = subprocess.call(['umount', partition_1])
Simran Basia9f41032012-05-11 14:21:58 -0700705 if rc != 0:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700706 self._logger.error('Failed to unmount USB Device')
Simran Basia9f41032012-05-11 14:21:58 -0700707 result = False
708 else:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700709 self._logger.error('Failed to mount USB Device')
Simran Basia9f41032012-05-11 14:21:58 -0700710 result = False
711
712 # Delete tmpdir. May throw exception if 'umount' failed.
713 try:
714 os.rmdir(tmpdir)
715 except OSError as e:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700716 self._logger.error('Failed to remove temp directory %s : %s', tmpdir,
717 str(e))
Simran Basia9f41032012-05-11 14:21:58 -0700718 return False
719 except BaseException as e:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700720 self._logger.error('Unexpected Exception removing tempdir %s : %s',
Simran Basia9f41032012-05-11 14:21:58 -0700721 tmpdir, str(e))
722 return False
723 else:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700724 self._logger.error('Failed to create temp directory.')
Simran Basia9f41032012-05-11 14:21:58 -0700725 return False
726 return result
727
Todd Broch352b4b22013-03-22 09:48:40 -0700728 def set_get_all(self, cmds):
729 """Set &| get one or more control values.
730
731 Args:
732 cmds: list of control[:value] to get or set.
733
734 Returns:
735 rv: list of responses from calling get or set methods.
736 """
737 rv = []
738 for cmd in cmds:
739 if ':' in cmd:
740 (control, value) = cmd.split(':')
741 rv.append(self.set(control, value))
742 else:
743 rv.append(self.get(cmd))
744 return rv
745
Aseda Aboagye6921f602017-08-01 14:45:38 -0700746 def get_serial_number(self, name):
747 """Returns the desired serial number from the serialnames dict.
748
749 Args:
750 name: A string which is the key into the _serialnames dictionary.
751
752 Returns:
753 A string containing the serial number or "unknown".
754 """
755 if not name:
756 name = 'main'
757
758 try:
759 return self._serialnames[name]
760 except KeyError:
761 self._logger.debug("'%s_serialname' not found!", name)
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700762 return 'unknown'
Aseda Aboagye6921f602017-08-01 14:45:38 -0700763
Todd Broche505b8d2011-03-21 18:19:54 -0700764 def get(self, name):
765 """Get control value.
766
767 Args:
768 name: name string of control
769
770 Returns:
771 Response from calling drv get method. Value is reformatted based on
772 control's dictionary parameters
773
774 Raises:
775 HwDriverError: Error occurred while using drv
Ruben Rodriguez Buchillona5d8b922018-09-03 16:51:03 +0800776 ServodError: if interfaces are not available within timeout period
Todd Broche505b8d2011-03-21 18:19:54 -0700777 """
Ruben Rodriguez Buchillona5d8b922018-09-03 16:51:03 +0800778 if not self._ifaces_available.wait(self.INTERFACE_AVAILABILITY_TIMEOUT):
779 raise ServodError('Timed out waiting for interfaces to become available.')
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700780 self._logger.debug('name(%s)' % (name))
Wai-Hong Tambafeca72017-10-05 14:22:12 -0700781 # This route is to retrieve serialnames on servo v4, which
782 # connects to multiple servo-micros or CCD, like the controls,
783 # 'ccd_serialname', 'servo_micro_for_soraka_serialname', etc.
784 # TODO(aaboagye): Refactor it.
785 if 'serialname' in name:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700786 return self.get_serial_number(name.split('serialname')[0].strip('_'))
Wai-Hong Tambafeca72017-10-05 14:22:12 -0700787
Todd Broche505b8d2011-03-21 18:19:54 -0700788 (param, drv) = self._get_param_drv(name)
789 try:
790 val = drv.get()
791 rd_val = self._syscfg.reformat_val(param, val)
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700792 self._logger.debug('%s = %s' % (name, rd_val))
Todd Broche505b8d2011-03-21 18:19:54 -0700793 return rd_val
Todd Brochfbc499d2011-06-16 16:09:58 -0700794 except AttributeError, error:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700795 self._logger.error('Getting %s: %s' % (name, error))
Todd Brochfbc499d2011-06-16 16:09:58 -0700796 raise
Vic Yangbe6cf262012-09-10 10:40:56 +0800797 except HwDriverError:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700798 self._logger.error('Getting %s' % (name))
Todd Broche505b8d2011-03-21 18:19:54 -0700799 raise
Todd Brochd6061672012-05-11 15:52:47 -0700800
Todd Broche505b8d2011-03-21 18:19:54 -0700801 def get_all(self, verbose):
802 """Get all controls values.
803
804 Args:
805 verbose: Boolean on whether to return doc info as well
806
807 Returns:
808 string creating from trying to get all values of all controls. In case of
809 error attempting access to control, response is 'ERR'.
810 """
Vadim Bendeburyb07944c2013-01-16 10:47:10 -0800811 rsp = []
Todd Broche505b8d2011-03-21 18:19:54 -0700812 for name in self._syscfg.syscfg_dict['control']:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700813 self._logger.debug('name = %s' % name)
Todd Broche505b8d2011-03-21 18:19:54 -0700814 try:
815 value = self.get(name)
816 except Exception:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700817 value = 'ERR'
Todd Broche505b8d2011-03-21 18:19:54 -0700818 pass
819 if verbose:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700820 rsp.append('GET %s = %s :: %s' % (name, value, self.doc(name)))
Todd Broche505b8d2011-03-21 18:19:54 -0700821 else:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700822 rsp.append('%s:%s' % (name, value))
Vadim Bendeburyb07944c2013-01-16 10:47:10 -0800823 return '\n'.join(sorted(rsp))
Todd Broche505b8d2011-03-21 18:19:54 -0700824
825 def set(self, name, wr_val_str):
826 """Set control.
827
828 Args:
829 name: name string of control
830 wr_val_str: value string to write. Can be integer, float or a
831 alpha-numerical that is mapped to a integer or float.
832
833 Raises:
834 HwDriverError: Error occurred while using driver
Ruben Rodriguez Buchillona5d8b922018-09-03 16:51:03 +0800835 ServodError: if interfaces are not available within timeout period
Todd Broche505b8d2011-03-21 18:19:54 -0700836 """
Ruben Rodriguez Buchillona5d8b922018-09-03 16:51:03 +0800837 if not self._ifaces_available.wait(self.INTERFACE_AVAILABILITY_TIMEOUT):
838 raise ServodError('Timed out waiting for interfaces to become available.')
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700839 self._logger.debug('name(%s) wr_val(%s)' % (name, wr_val_str))
Todd Broche505b8d2011-03-21 18:19:54 -0700840 (params, drv) = self._get_param_drv(name, False)
841 wr_val = self._syscfg.resolve_val(params, wr_val_str)
842 try:
843 drv.set(wr_val)
Vic Yangbe6cf262012-09-10 10:40:56 +0800844 except HwDriverError:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700845 self._logger.error('Setting %s -> %s' % (name, wr_val_str))
Todd Broche505b8d2011-03-21 18:19:54 -0700846 raise
Ruben Rodriguez Buchillonb5fe0f12018-05-09 10:19:56 +0800847 # TODO(crbug.com/841097) Figure out why despite allow_none=True for both
848 # xmlrpc server & client I still have to return something to appease the
Todd Broche505b8d2011-03-21 18:19:54 -0700849 # marshall/unmarshall
850 return True
851
Todd Brochd6061672012-05-11 15:52:47 -0700852 def hwinit(self, verbose=False):
853 """Initialize all controls.
854
855 These values are part of the system config XML files of the form
856 init=<value>. This command should be used by clients wishing to return the
857 servo and DUT its connected to a known good/safe state.
858
Vadim Bendeburybb51dd42013-01-31 13:47:46 -0800859 Note that initialization errors are ignored (as in some cases they could
860 be caused by DUT firmware deficiencies). This might need to be fine tuned
861 later.
862
Todd Brochd6061672012-05-11 15:52:47 -0700863 Args:
864 verbose: boolean, if True prints info about control initialized.
865 Otherwise prints nothing.
Vadim Bendebury5934e4b2013-02-06 13:57:54 -0800866
867 Returns:
868 This function is called across RPC and as such is expected to return
869 something unless transferring 'none' across is allowed. Hence adding a
870 dummy return value to make things simpler.
Todd Brochd6061672012-05-11 15:52:47 -0700871 """
Todd Brochd9acf0a2012-12-05 13:43:06 -0800872 for control_name, value in self._syscfg.hwinit:
Todd Broch3ec8df02012-11-20 10:53:03 -0800873 try:
John Carey6fe2bbf2015-08-31 16:13:03 -0700874 # Workaround for bug chrome-os-partner:42349. Without this check, the
875 # gpio will briefly pulse low if we set it from high to high.
876 if self.get(control_name) != value:
Aseda Aboagyea849d462016-05-04 17:08:16 -0700877 self.set(control_name, value)
878 if verbose:
879 self._logger.info('Initialized %s to %s', control_name, value)
Todd Broch3ec8df02012-11-20 10:53:03 -0800880 except Exception as e:
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700881 self._logger.error('Problem initializing %s -> %s :: %s', control_name,
882 value, str(e))
Nick Sandersbc836282015-12-08 21:19:23 -0800883
884 # Init keyboard after all the intefaces are up.
885 self._keyboard = self._init_keyboard_handler(self, self._board)
Vadim Bendebury5934e4b2013-02-06 13:57:54 -0800886 return True
Todd Broch3ec8df02012-11-20 10:53:03 -0800887
Todd Broche505b8d2011-03-21 18:19:54 -0700888 def echo(self, echo):
889 """Dummy echo function for testing/examples.
890
891 Args:
892 echo: string to echo back to client
893 """
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700894 self._logger.debug('echo(%s)' % (echo))
895 return 'ECH0ING: %s' % (echo)
Todd Broche505b8d2011-03-21 18:19:54 -0700896
J. Richard Barnettee2820552013-03-14 16:13:46 -0700897 def get_board(self):
898 """Return the board specified at startup, if any."""
899 return self._board
900
Wai-Hong Tam416cf612017-09-19 11:39:21 -0700901 def get_base_board(self):
902 """Returns the board name of the base if present.
903
904 Returns:
905 A string of the board name, or '' if not present.
906 """
907 # The value is set in servo_postinit.
908 return self._base_board
909
Simran Basia23c1392013-08-06 14:59:10 -0700910 def get_version(self):
911 """Get servo board version."""
912 return self._version
913
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800914 def power_long_press(self):
915 """Simulate a long power button press."""
916 # After a long power press, the EC may ignore the next power
917 # button press (at least on Alex). To guarantee that this
918 # won't happen, we need to allow the EC one second to
919 # collect itself.
Ruben Rodriguez Buchillon0f467942018-07-27 18:02:32 +0800920 return self.set('power_key', 'long_press')
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800921
922 def power_normal_press(self):
923 """Simulate a normal power button press."""
Ruben Rodriguez Buchillon0f467942018-07-27 18:02:32 +0800924 return self.set('power_key', 'press')
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800925
926 def power_short_press(self):
927 """Simulate a short power button press."""
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530928 return self.set('power_key', 'short_press')
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800929
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530930 def power_key(self, press_secs=''):
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800931 """Simulate a power button press.
932
933 Args:
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530934 press_secs: Time in seconds to simulate the keypress.
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800935 """
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530936 return self.set('power_key', 'press' if press_secs is '' else press_secs)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800937
938 def ctrl_d(self, press_secs=''):
939 """Simulate Ctrl-d simultaneous button presses."""
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530940 return self.set('ctrl_d', 'tab' if press_secs is '' else press_secs)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800941
Victor Dodone539cea2016-03-29 18:50:17 -0700942 def ctrl_u(self, press_secs=''):
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800943 """Simulate Ctrl-u simultaneous button presses."""
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530944 return self.set('ctrl_u', 'tab' if press_secs is '' else press_secs)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800945
946 def ctrl_enter(self, press_secs=''):
947 """Simulate Ctrl-enter simultaneous button presses."""
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530948 return self.set('ctrl_enter', 'tab' if press_secs is '' else press_secs)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800949
950 def d_key(self, press_secs=''):
951 """Simulate Enter key button press."""
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530952 return self.set('d_key', 'tab' if press_secs is '' else press_secs)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800953
954 def ctrl_key(self, press_secs=''):
955 """Simulate Enter key button press."""
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530956 return self.set('ctrl_key', 'tab' if press_secs is '' else press_secs)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800957
958 def enter_key(self, press_secs=''):
959 """Simulate Enter key button press."""
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530960 return self.set('enter_key', 'tab' if press_secs is '' else press_secs)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800961
962 def refresh_key(self, press_secs=''):
963 """Simulate Refresh key (F3) button press."""
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530964 return self.set('refresh_key', 'tab' if press_secs is '' else press_secs)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800965
966 def ctrl_refresh_key(self, press_secs=''):
967 """Simulate Ctrl and Refresh (F3) simultaneous press.
968
969 This key combination is an alternative of Space key.
970 """
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530971 return self.set('ctrl_refresh_key', ('tab' if press_secs is '' else
972 press_secs))
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800973
974 def imaginary_key(self, press_secs=''):
975 """Simulate imaginary key button press.
976
977 Maps to a key that doesn't physically exist.
978 """
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530979 return self.set('imaginary_key', 'tab' if press_secs is '' else press_secs)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800980
Vincent Palatin3acbbe52016-07-19 17:40:12 +0200981 def sysrq_x(self, press_secs=''):
982 """Simulate Alt VolumeUp X simultaneous press.
983
984 This key combination is the kernel system request (sysrq) x.
985 """
Lenine Ajagappane400d7d22018-09-05 02:29:21 +0530986 return self.set('sysrq_x', 'tab' if press_secs is '' else press_secs)
Vincent Palatin3acbbe52016-07-19 17:40:12 +0200987
Kevin Cheng4b4f0022016-09-09 02:37:07 -0700988 def get_servo_serials(self):
989 """Return all the serials associated with this process."""
990 return self._serialnames
991
992
Todd Broche505b8d2011-03-21 18:19:54 -0700993def test():
994 """Integration testing.
995
996 TODO(tbroch) Enhance integration test and add unittest (see mox)
997 """
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -0700998 logging.basicConfig(
999 level=logging.DEBUG,
1000 format='%(asctime)s - %(name)s - ' + '%(levelname)s - %(message)s')
Todd Broche505b8d2011-03-21 18:19:54 -07001001 # configure server & listen
1002 servod_obj = Servod(1)
Wai-Hong Tam564c1702017-04-24 09:23:38 -07001003 # 5 == number of interfaces on a FT4232H device
1004 for i in xrange(1, 5):
1005 if i == 2:
Todd Broche505b8d2011-03-21 18:19:54 -07001006 # its an i2c interface ... see __init__ for details and TODO to make
1007 # this configureable
1008 servod_obj._interface_list[i].wr_rd(0x21, [0], 1)
1009 else:
1010 # its a gpio interface
1011 servod_obj._interface_list[i].wr_rd(0)
1012
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -07001013 server = SimpleXMLRPCServer.SimpleXMLRPCServer(('localhost', 9999),
Todd Broche505b8d2011-03-21 18:19:54 -07001014 allow_none=True)
1015 server.register_introspection_functions()
1016 server.register_multicall_functions()
1017 server.register_instance(servod_obj)
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -07001018 logging.info('Listening on localhost port 9999')
Todd Broche505b8d2011-03-21 18:19:54 -07001019 server.serve_forever()
1020
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -07001021
1022if __name__ == '__main__':
Todd Broche505b8d2011-03-21 18:19:54 -07001023 test()
1024
1025 # simple client transaction would look like
Puthikorn Voravootivat01ead152018-03-23 15:38:40 -07001026 """remote_uri = 'http://localhost:9999' client = xmlrpclib.ServerProxy(remote_uri, verbose=False) send_str = "Hello_there" print "Sent " + send_str + ", Recv " + client.echo(send_str)
1027
Todd Broche505b8d2011-03-21 18:19:54 -07001028 """