blob: f4f962ad216e411e409281e9f32ce1f8e7b5bdaa [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."""
Simran Basia9f41032012-05-11 14:21:58 -07005import fnmatch
Todd Broche505b8d2011-03-21 18:19:54 -07006import imp
7import logging
Simran Basia9f41032012-05-11 14:21:58 -07008import os
9import shutil
Todd Broche505b8d2011-03-21 18:19:54 -070010import SimpleXMLRPCServer
Simran Basia9f41032012-05-11 14:21:58 -070011import subprocess
12import tempfile
Todd Broch7a91c252012-02-03 12:37:45 -080013import time
Simran Basia9f41032012-05-11 14:21:58 -070014import urllib
Todd Broche505b8d2011-03-21 18:19:54 -070015
16# TODO(tbroch) deprecate use of relative imports
Vic Yangbe6cf262012-09-10 10:40:56 +080017from drv.hw_driver import HwDriverError
Aaron.Chuang88eff332014-07-31 08:32:00 +080018import bbadc
Simran Basia9ad25e2013-04-23 11:57:00 -070019import bbi2c
Simran Basi5492bde2013-05-16 17:08:47 -070020import bbgpio
Simran Basi949309b2013-05-31 15:12:15 -070021import bbuart
Aseda Aboagyea4922212015-11-20 15:19:08 -080022import ec3po_interface
Todd Broche505b8d2011-03-21 18:19:54 -070023import ftdigpio
24import ftdii2c
Todd Brochdbb09982011-10-02 07:14:26 -070025import ftdi_common
Todd Broch47c43f42011-05-26 15:11:31 -070026import ftdiuart
Rong Changc6c8c022014-08-11 14:07:11 +080027import i2cbus
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -080028import keyboard_handlers
Simran Basie750a342013-03-12 13:45:26 -070029import servo_interfaces
Todd Broche505b8d2011-03-21 18:19:54 -070030
Aseda Aboagyea4922212015-11-20 15:19:08 -080031
Todd Broche505b8d2011-03-21 18:19:54 -070032MAX_I2C_CLOCK_HZ = 100000
33
Todd Brochdbb09982011-10-02 07:14:26 -070034
Todd Broche505b8d2011-03-21 18:19:54 -070035class ServodError(Exception):
36 """Exception class for servod."""
37
38class Servod(object):
39 """Main class for Servo debug/controller Daemon."""
Simran Basia9f41032012-05-11 14:21:58 -070040 _USB_DETECTION_DELAY = 10
Fang Deng90377712013-06-03 15:51:48 -070041 _USB_POWEROFF_DELAY = 2
Simran Basia9f41032012-05-11 14:21:58 -070042 _HTTP_PREFIX = "http://"
Fang Deng90377712013-06-03 15:51:48 -070043 _USB_J3 = "usb_mux_sel1"
44 _USB_J3_TO_SERVO = "servo_sees_usbkey"
45 _USB_J3_TO_DUT = "dut_sees_usbkey"
46 _USB_J3_PWR = "prtctl4_pwren"
47 _USB_J3_PWR_ON = "on"
48 _USB_J3_PWR_OFF = "off"
Simran Basia9f41032012-05-11 14:21:58 -070049
J. Richard Barnettee2820552013-03-14 16:13:46 -070050 def __init__(self, config, vendor, product, serialname=None,
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -080051 interfaces=None, board="", version=None, usbkm232=None):
Todd Broche505b8d2011-03-21 18:19:54 -070052 """Servod constructor.
53
54 Args:
55 config: instance of SystemConfig containing all controls for
56 particular Servod invocation
57 vendor: usb vendor id of FTDI device
58 product: usb product id of FTDI device
Todd Brochad034442011-05-25 15:05:29 -070059 serialname: string of device serialname/number as defined in FTDI eeprom.
Todd Brochdbb09982011-10-02 07:14:26 -070060 interfaces: list of strings of interface types the server will instantiate
Simran Basia23c1392013-08-06 14:59:10 -070061 version: String. Servo board version. Examples: servo_v1, servo_v2,
62 servo_v2_r0, servo_v3
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -080063 usbkm232: String. Optional. Path to USB-KM232 device which allow for
64 sending keyboard commands to DUTs that do not have built in
Danny Chan662b6022015-11-04 17:34:53 -080065 keyboards. Used in FAFT tests. Use 'atmega' for on board AVR MCU.
66 e.g. '/dev/ttyUSB0' or 'atmega'
Todd Brochdbb09982011-10-02 07:14:26 -070067
68 Raises:
69 ServodError: if unable to locate init method for particular interface
Todd Broche505b8d2011-03-21 18:19:54 -070070 """
71 self._logger = logging.getLogger("Servod")
72 self._logger.debug("")
73 self._vendor = vendor
74 self._product = product
Todd Brochad034442011-05-25 15:05:29 -070075 self._serialname = serialname
Todd Broche505b8d2011-03-21 18:19:54 -070076 self._syscfg = config
77 # list of objects (Fi2c, Fgpio) to physical interfaces (gpio, i2c) that ftdi
78 # interfaces are mapped to
79 self._interface_list = []
80 # Dict of Dict to map control name, function name to to tuple (params, drv)
81 # Ex) _drv_dict[name]['get'] = (params, drv)
82 self._drv_dict = {}
J. Richard Barnettee2820552013-03-14 16:13:46 -070083 self._board = board
Simran Basia23c1392013-08-06 14:59:10 -070084 self._version = version
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -080085 self._usbkm232 = usbkm232
Todd Brochdbb09982011-10-02 07:14:26 -070086 # Note, interface i is (i - 1) in list
87 if not interfaces:
Todd Brochb21d8042014-05-15 12:54:54 -070088 try:
89 interfaces = servo_interfaces.INTERFACE_BOARDS[board][vendor][product]
90 except KeyError:
91 interfaces = servo_interfaces.INTERFACE_DEFAULTS[vendor][product]
Todd Brochdbb09982011-10-02 07:14:26 -070092
Simran Basia9ad25e2013-04-23 11:57:00 -070093 for i, interface in enumerate(interfaces):
94 is_ftdi_interface = False
95 if type(interface) is dict:
96 name = interface['name']
Aseda Aboagyea4922212015-11-20 15:19:08 -080097 # Store interface index for those that care about it.
98 interface['index'] = i
Simran Basia9ad25e2013-04-23 11:57:00 -070099 elif type(interface) is str:
Simran Basia9ad25e2013-04-23 11:57:00 -0700100 name = interface
Aseda Aboagyea4922212015-11-20 15:19:08 -0800101 # It's a FTDI related interface.
Simran Basia9ad25e2013-04-23 11:57:00 -0700102 interface = (i % ftdi_common.MAX_FTDI_INTERFACES_PER_DEVICE) + 1
103 is_ftdi_interface = True
104 else:
105 raise ServodError("Illegal interface type %s" % type(interface))
106
Todd Broch8a77a992012-01-27 09:46:08 -0800107 # servos with multiple FTDI are guaranteed to have contiguous USB PIDs
Simran Basia9ad25e2013-04-23 11:57:00 -0700108 if is_ftdi_interface and i and \
109 ((i % ftdi_common.MAX_FTDI_INTERFACES_PER_DEVICE) == 0):
Todd Broch8a77a992012-01-27 09:46:08 -0800110 self._product += 1
111 self._logger.info("Changing to next FTDI part @ pid = 0x%04x",
112 self._product)
113
Simran Basia9ad25e2013-04-23 11:57:00 -0700114 self._logger.info("Initializing interface %d to %s", i + 1, name)
Todd Brochdbb09982011-10-02 07:14:26 -0700115 try:
116 func = getattr(self, '_init_%s' % name)
117 except AttributeError:
118 raise ServodError("Unable to locate init for interface %s" % name)
Simran Basia9ad25e2013-04-23 11:57:00 -0700119 result = func(interface)
Todd Broch888da782011-10-07 14:29:09 -0700120 if isinstance(result, tuple):
121 self._interface_list.extend(result)
122 else:
123 self._interface_list.append(result)
Todd Broche505b8d2011-03-21 18:19:54 -0700124
Danny Chan662b6022015-11-04 17:34:53 -0800125
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800126 def _init_keyboard_handler(self, servo, board=''):
127 """Initialize the correct keyboard handler for board.
128
129 @param servo: servo object.
130 @param board: string, board name.
131
132 """
133 if board == 'parrot':
134 return keyboard_handlers.ParrotHandler(servo)
135 elif board == 'stout':
136 return keyboard_handlers.StoutHandler(servo)
PeggyChuang4f07d872015-08-07 12:11:38 +0800137 elif board in ('buddy', 'cranky', 'guado', 'jecht', 'mccloud', 'monroe',
Tom Wai-Hong Tamd64164c2015-04-29 07:59:45 +0800138 'ninja', 'nyan_kitty', 'panther', 'rikku', 'stumpy',
Jason Simmonse5cccd12015-08-25 16:05:25 -0700139 'sumo', 'tidus', 'tricky', 'veyron_mickey', 'veyron_rialto',
140 'zako'):
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800141 if self._usbkm232 is None:
Yusuf Mohsinally66844692014-05-22 10:37:52 -0700142 logging.warn("No device path specified for usbkm232 handler. Returning "
Tom Wai-Hong Tamd64164c2015-04-29 07:59:45 +0800143 "the MatrixKeyboardHandler, which is likely the wrong "
144 "keyboard handler for the board type specified.")
145 return keyboard_handlers.MatrixKeyboardHandler(servo)
Danny Chan662b6022015-11-04 17:34:53 -0800146 if self._usbkm232 == 'atmega':
147 # Use servo onboard keyboard emulator.
Nick Sanders78423782015-11-09 14:28:19 -0800148 self.set('atmega_rst', 'on')
Nick Sandersbc836282015-12-08 21:19:23 -0800149 self.set('at_hwb', 'off')
Nick Sanders78423782015-11-09 14:28:19 -0800150 self.set('atmega_rst', 'off')
Danny Chan662b6022015-11-04 17:34:53 -0800151 self._usbkm232 = self.get('atmega_pty')
152 self.set('atmega_baudrate', '9600')
153 self.set('atmega_bits', 'eight')
154 self.set('atmega_parity', 'none')
155 self.set('atmega_sbits', 'one')
156 self.set('usb_mux_sel4', 'on')
Nick Sandersbc836282015-12-08 21:19:23 -0800157 self.set('usb_mux_oe4', 'on')
Nick Sanders78423782015-11-09 14:28:19 -0800158 # Allow atmega bootup time.
159 time.sleep(1.0)
Danny Chan662b6022015-11-04 17:34:53 -0800160 self._logger.info('USBKM232: %s', self._usbkm232)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800161 return keyboard_handlers.USBkm232Handler(servo, self._usbkm232)
162 else:
Tom Wai-Hong Tamd64164c2015-04-29 07:59:45 +0800163 # The following boards don't use Chrome EC.
164 if board in ('alex', 'butterfly', 'lumpy', 'zgb'):
165 return keyboard_handlers.MatrixKeyboardHandler(servo)
166 return keyboard_handlers.ChromeECHandler(servo)
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800167
Todd Broch3ec8df02012-11-20 10:53:03 -0800168 def __del__(self):
169 """Servod deconstructor."""
170 for interface in self._interface_list:
171 del(interface)
172
Todd Brochb3048492012-01-15 21:52:41 -0800173 def _init_dummy(self, interface):
174 """Initialize dummy interface.
175
176 Dummy interface is just a mechanism to reserve that interface for non servod
177 interaction. Typically the interface will be managed by external
178 third-party tools like openOCD or urjtag for JTAG or flashrom for SPI
179 interfaces.
180
181 TODO(tbroch): Investigate merits of incorporating these third-party
182 interfaces into servod or creating a communication channel between them
183
184 Returns: None
185 """
186 return None
187
Simran Basie750a342013-03-12 13:45:26 -0700188 def _init_ftdi_gpio(self, interface):
Todd Broche505b8d2011-03-21 18:19:54 -0700189 """Initialize gpio driver interface and open for use.
190
191 Args:
192 interface: interface number of FTDI device to use.
193
194 Returns:
195 Instance object of interface.
Todd Broch6de9dc62012-04-09 15:23:53 -0700196
197 Raises:
198 ServodError: If init fails
Todd Broche505b8d2011-03-21 18:19:54 -0700199 """
Todd Brochad034442011-05-25 15:05:29 -0700200 fobj = ftdigpio.Fgpio(self._vendor, self._product, interface,
201 self._serialname)
Todd Broch6de9dc62012-04-09 15:23:53 -0700202 try:
203 fobj.open()
204 except ftdigpio.FgpioError as e:
205 raise ServodError('Opening gpio interface. %s ( %d )' % (e.msg, e.value))
206
Todd Broche505b8d2011-03-21 18:19:54 -0700207 return fobj
208
Aaron.Chuang88eff332014-07-31 08:32:00 +0800209 def _init_bb_adc(self, interface):
210 """Initalize beaglebone ADC interface."""
211 return bbadc.BBadc()
212
Simran Basie750a342013-03-12 13:45:26 -0700213 def _init_bb_gpio(self, interface):
214 """Initalize beaglebone gpio interface."""
Simran Basi5492bde2013-05-16 17:08:47 -0700215 return bbgpio.BBgpio()
Simran Basie750a342013-03-12 13:45:26 -0700216
217 def _init_ftdi_i2c(self, interface):
Todd Broche505b8d2011-03-21 18:19:54 -0700218 """Initialize i2c interface and open for use.
219
220 Args:
221 interface: interface number of FTDI device to use
222
223 Returns:
224 Instance object of interface
Todd Broch6de9dc62012-04-09 15:23:53 -0700225
226 Raises:
227 ServodError: If init fails
Todd Broche505b8d2011-03-21 18:19:54 -0700228 """
Todd Brochad034442011-05-25 15:05:29 -0700229 fobj = ftdii2c.Fi2c(self._vendor, self._product, interface,
230 self._serialname)
Todd Broch6de9dc62012-04-09 15:23:53 -0700231 try:
232 fobj.open()
233 except ftdii2c.Fi2cError as e:
234 raise ServodError('Opening i2c interface. %s ( %d )' % (e.msg, e.value))
235
Todd Broche505b8d2011-03-21 18:19:54 -0700236 # Set the frequency of operation of the i2c bus.
237 # TODO(tbroch) make configureable
238 fobj.setclock(MAX_I2C_CLOCK_HZ)
Todd Broch6de9dc62012-04-09 15:23:53 -0700239
Todd Broche505b8d2011-03-21 18:19:54 -0700240 return fobj
241
Simran Basie750a342013-03-12 13:45:26 -0700242 # TODO (sbasi) crbug.com/187489 - Implement bb_i2c.
243 def _init_bb_i2c(self, interface):
244 """Initalize beaglebone i2c interface."""
Simran Basia9ad25e2013-04-23 11:57:00 -0700245 return bbi2c.BBi2c(interface)
Simran Basie750a342013-03-12 13:45:26 -0700246
Rong Changc6c8c022014-08-11 14:07:11 +0800247 def _init_dev_i2c(self, interface):
248 """Initalize Linux i2c-dev interface."""
249 return i2cbus.I2CBus('/dev/i2c-%d' % interface['bus_num'])
250
Simran Basie750a342013-03-12 13:45:26 -0700251 def _init_ftdi_uart(self, interface):
252 """Initialize ftdi uart inteface and open for use
Todd Broch47c43f42011-05-26 15:11:31 -0700253
254 Note, the uart runs in a separate thread (pthreads). Users wishing to
255 interact with it will query control for the pty's pathname and connect
256 with there favorite console program. For example:
257 cu -l /dev/pts/22
258
259 Args:
260 interface: interface number of FTDI device to use
261
262 Returns:
263 Instance object of interface
Todd Broch6de9dc62012-04-09 15:23:53 -0700264
265 Raises:
266 ServodError: If init fails
Todd Broch47c43f42011-05-26 15:11:31 -0700267 """
Jeremy Thorpe9e110062012-10-25 10:40:00 -0700268 fobj = ftdiuart.Fuart(self._vendor, self._product, interface,
269 self._serialname)
Todd Broch6de9dc62012-04-09 15:23:53 -0700270 try:
271 fobj.run()
272 except ftdiuart.FuartError as e:
273 raise ServodError('Running uart interface. %s ( %d )' % (e.msg, e.value))
274
Todd Broch47c43f42011-05-26 15:11:31 -0700275 self._logger.info("%s" % fobj.get_pty())
276 return fobj
277
Simran Basie750a342013-03-12 13:45:26 -0700278 # TODO (sbasi) crbug.com/187492 - Implement bbuart.
279 def _init_bb_uart(self, interface):
280 """Initalize beaglebone uart interface."""
Simran Basi949309b2013-05-31 15:12:15 -0700281 logging.debug('UART INTERFACE: %s', interface)
282 return bbuart.BBuart(interface)
Simran Basie750a342013-03-12 13:45:26 -0700283
284 def _init_ftdi_gpiouart(self, interface):
Todd Broch888da782011-10-07 14:29:09 -0700285 """Initialize special gpio + uart interface and open for use
286
287 Note, the uart runs in a separate thread (pthreads). Users wishing to
288 interact with it will query control for the pty's pathname and connect
289 with there favorite console program. For example:
290 cu -l /dev/pts/22
291
292 Args:
293 interface: interface number of FTDI device to use
294
295 Returns:
296 Instance objects of interface
Todd Broch6de9dc62012-04-09 15:23:53 -0700297
298 Raises:
299 ServodError: If init fails
Todd Broch888da782011-10-07 14:29:09 -0700300 """
Simran Basie750a342013-03-12 13:45:26 -0700301 fgpio = self._init_ftdi_gpio(interface)
Jeremy Thorpe9e110062012-10-25 10:40:00 -0700302 fuart = ftdiuart.Fuart(self._vendor, self._product, interface,
303 self._serialname, fgpio._fc)
Todd Broch6de9dc62012-04-09 15:23:53 -0700304 try:
305 fuart.run()
306 except ftdiuart.FuartError as e:
307 raise ServodError('Running uart interface. %s ( %d )' % (e.msg, e.value))
308
Todd Broch888da782011-10-07 14:29:09 -0700309 self._logger.info("uart pty: %s" % fuart.get_pty())
310 return fgpio, fuart
311
Aseda Aboagyea4922212015-11-20 15:19:08 -0800312 def _init_ec3po_uart(self, interface):
313 """Initialize EC-3PO console interpreter interface.
314
315 Args:
316 interface: A dictionary representing the interface.
317
318 Returns:
319 An EC3PO object representing the EC-3PO interface or None if there's no
320 interface for the USB PD UART.
321 """
322 vid = self._vendor
323 pid = self._product
324 # The current PID might be incremented if there are multiple FTDI.
325 # Therefore, try rewinding the PID back one if we don't find the base PID in
326 # the SERVO_ID_DEFAULTS
327 if (vid, pid) not in servo_interfaces.SERVO_ID_DEFAULTS:
328 self._logger.debug('VID:PID pair not found. Rewinding PID back one...')
329 pid -= 1
330 self._logger.debug('vid:0x%04x, pid:0x%04x', vid, pid)
331
332 # Servo V2 / V3 should have the interface indicies in the same spot.
333 if ((vid, pid) in servo_interfaces.SERVO_V2_DEFAULTS or
334 (vid, pid) in servo_interfaces.SERVO_V3_DEFAULTS):
335 # Determine if it's a PD interface or just main EC console.
336 if interface['index'] == servo_interfaces.EC3PO_USBPD_INTERFACE_NUM:
337 try:
338 # Obtain the raw EC UART PTY and create the EC-3PO interface.
339 raw_ec_uart = self.get('raw_usbpd_uart_pty')
340 except NameError:
341 # This overlay doesn't have a USB PD MCU, so skip init.
342 self._logger.info('No PD MCU UART.')
343 return None
344 except AttributeError:
345 # This overlay has no get method for the interface so skip init.
346 self._logger.warn('No interface for PD MCU UART.')
347 return None
348
349 elif interface['index'] == servo_interfaces.EC3PO_EC_INTERFACE_NUM:
350 raw_ec_uart = self.get('raw_ec_uart_pty')
351
352 # Servo V3, miniservo, Toad, Reston, Fruitpie, or Plankton
353 elif ((vid, pid) in servo_interfaces.MINISERVO_ID_DEFAULTS or
354 (vid, pid) in servo_interfaces.TOAD_ID_DEFAULTS or
355 (vid, pid) in servo_interfaces.RESTON_ID_DEFAULTS or
356 (vid, pid) in servo_interfaces.FRUITPIE_ID_DEFAULTS or
357 (vid, pid) in servo_interfaces.PLANKTON_ID_DEFAULTS):
358 raw_ec_uart = self.get('raw_ec_uart_pty')
359
360 else:
361 raise ServodError(('Unexpected EC-3PO interface!'
362 ' (0x%04x:0x%04x) %r') % (vid, pid, interface))
363
364 return ec3po_interface.EC3PO(raw_ec_uart)
365
Tom Wai-Hong Tam28f0a5f2012-08-21 12:49:57 +0800366 def _camel_case(self, string):
367 output = ''
368 for s in string.split('_'):
369 if output:
370 output += s.capitalize()
371 else:
372 output = s
373 return output
374
Todd Broche505b8d2011-03-21 18:19:54 -0700375 def _get_param_drv(self, control_name, is_get=True):
376 """Get access to driver for a given control.
377
378 Note, some controls have different parameter dictionaries for 'getting' the
379 control's value versus 'setting' it. Boolean is_get distinguishes which is
380 being requested.
381
382 Args:
383 control_name: string name of control
384 is_get: boolean to determine
385
386 Returns:
387 tuple (param, drv) where:
388 param: param dictionary for control
389 drv: instance object of driver for particular control
390
391 Raises:
392 ServodError: Error occurred while examining params dict
393 """
394 self._logger.debug("")
395 # if already setup just return tuple from driver dict
396 if control_name in self._drv_dict:
397 if is_get and ('get' in self._drv_dict[control_name]):
398 return self._drv_dict[control_name]['get']
399 if not is_get and ('set' in self._drv_dict[control_name]):
400 return self._drv_dict[control_name]['set']
401
402 params = self._syscfg.lookup_control_params(control_name, is_get)
403 if 'drv' not in params:
404 self._logger.error("Unable to determine driver for %s" % control_name)
405 raise ServodError("'drv' key not found in params dict")
406 if 'interface' not in params:
407 self._logger.error("Unable to determine interface for %s" %
408 control_name)
Todd Broche505b8d2011-03-21 18:19:54 -0700409 raise ServodError("'interface' key not found in params dict")
Simran Basi668be0e2013-08-07 11:54:50 -0700410
J. Richard Barnette275d9fd2014-02-11 14:38:54 -0800411 interface_id = params.get(
412 '%s_interface' % self._version, params['interface'])
413 if interface_id == 'servo':
414 interface = self
Simran Basi668be0e2013-08-07 11:54:50 -0700415 else:
J. Richard Barnette275d9fd2014-02-11 14:38:54 -0800416 index = int(interface_id) - 1
417 interface = self._interface_list[index]
Simran Basi668be0e2013-08-07 11:54:50 -0700418
Todd Broche505b8d2011-03-21 18:19:54 -0700419 servo_pkg = imp.load_module('servo', *imp.find_module('servo'))
420 drv_pkg = imp.load_module('drv',
421 *imp.find_module('drv', servo_pkg.__path__))
422 drv_name = params['drv']
423 drv_module = getattr(drv_pkg, drv_name)
Tom Wai-Hong Tam28f0a5f2012-08-21 12:49:57 +0800424 drv_class = getattr(drv_module, self._camel_case(drv_name))
Todd Broche505b8d2011-03-21 18:19:54 -0700425 drv = drv_class(interface, params)
426 if control_name not in self._drv_dict:
427 self._drv_dict[control_name] = {}
428 if is_get:
429 self._drv_dict[control_name]['get'] = (params, drv)
430 else:
431 self._drv_dict[control_name]['set'] = (params, drv)
432 return (params, drv)
433
434 def doc_all(self):
435 """Return all documenation for controls.
436
437 Returns:
438 string of <doc> text in config file (xml) and the params dictionary for
439 all controls.
440
441 For example:
442 warm_reset :: Reset the device warmly
443 ------------------------> {'interface': '1', 'map': 'onoff_i', ... }
444 """
445 return self._syscfg.display_config()
446
447 def doc(self, name):
448 """Retreive doc string in system config file for given control name.
449
450 Args:
451 name: name string of control to get doc string
452
453 Returns:
454 doc string of name
455
456 Raises:
457 NameError: if fails to locate control
458 """
459 self._logger.debug("name(%s)" % (name))
460 if self._syscfg.is_control(name):
461 return self._syscfg.get_control_docstring(name)
462 else:
463 raise NameError("No control %s" %name)
464
Fang Deng90377712013-06-03 15:51:48 -0700465 def _switch_usbkey(self, mux_direction):
466 """Connect USB flash stick to either servo or DUT.
467
468 This function switches 'usb_mux_sel1' to provide electrical
469 connection between the USB port J3 and either servo or DUT side.
470
471 Switching the usb mux is accompanied by powercycling
472 of the USB stick, because it sometimes gets wedged if the mux
473 is switched while the stick power is on.
474
475 Args:
476 mux_direction: "servo_sees_usbkey" or "dut_sees_usbkey".
477 """
478 self.set(self._USB_J3_PWR, self._USB_J3_PWR_OFF)
479 time.sleep(self._USB_POWEROFF_DELAY)
480 self.set(self._USB_J3, mux_direction)
481 time.sleep(self._USB_POWEROFF_DELAY)
482 self.set(self._USB_J3_PWR, self._USB_J3_PWR_ON)
483 if mux_direction == self._USB_J3_TO_SERVO:
484 time.sleep(self._USB_DETECTION_DELAY)
485
Simran Basia9f41032012-05-11 14:21:58 -0700486 def _get_usb_port_set(self):
487 """Gets a set of USB disks currently connected to the system
488
489 Returns:
490 A set of USB disk paths.
491 """
492 usb_set = fnmatch.filter(os.listdir("/dev/"), "sd[a-z]")
493 return set(["/dev/" + dev for dev in usb_set])
494
495 def _probe_host_usb_dev(self):
496 """Probe the USB disk device plugged in the servo from the host side.
497
498 Method can fail by:
499 1) Having multiple servos connected and returning incorrect /dev/sdX of
500 another servo.
501 2) Finding multiple /dev/sdX and returning None.
502
503 Returns:
504 USB disk path if one and only one USB disk path is found, otherwise None.
505 """
Fang Deng90377712013-06-03 15:51:48 -0700506 original_value = self.get(self._USB_J3)
507 original_usb_power = self.get(self._USB_J3_PWR)
Simran Basia9f41032012-05-11 14:21:58 -0700508 # Make the host unable to see the USB disk.
Fang Deng90377712013-06-03 15:51:48 -0700509 if (original_usb_power == self._USB_J3_PWR_ON and
510 original_value != self._USB_J3_TO_DUT):
511 self._switch_usbkey(self._USB_J3_TO_DUT)
Simran Basia9f41032012-05-11 14:21:58 -0700512 no_usb_set = self._get_usb_port_set()
Simran Basia9f41032012-05-11 14:21:58 -0700513
Fang Deng90377712013-06-03 15:51:48 -0700514 # Make the host able to see the USB disk.
515 self._switch_usbkey(self._USB_J3_TO_SERVO)
Simran Basia9f41032012-05-11 14:21:58 -0700516 has_usb_set = self._get_usb_port_set()
Fang Deng90377712013-06-03 15:51:48 -0700517
Simran Basia9f41032012-05-11 14:21:58 -0700518 # Back to its original value.
Fang Deng90377712013-06-03 15:51:48 -0700519 if original_value != self._USB_J3_TO_SERVO:
520 self._switch_usbkey(original_value)
521 if original_usb_power != self._USB_J3_PWR_ON:
522 self.set(self._USB_J3_PWR, self._USB_J3_PWR_OFF)
523 time.sleep(self._USB_POWEROFF_DELAY)
524
Simran Basia9f41032012-05-11 14:21:58 -0700525 # Subtract the two sets to find the usb device.
526 diff_set = has_usb_set - no_usb_set
527 if len(diff_set) == 1:
528 return diff_set.pop()
529 else:
530 return None
531
532 def download_image_to_usb(self, image_path):
533 """Download image and save to the USB device found by probe_host_usb_dev.
534 If the image_path is a URL, it will download this url to the USB path;
535 otherwise it will simply copy the image_path's contents to the USB path.
536
537 Args:
538 image_path: path or url to the recovery image.
539
540 Returns:
541 True|False: True if process completed successfully, False if error
542 occurred.
543 Can't return None because XMLRPC doesn't allow it. PTAL at tbroch's
544 comment at the end of set().
545 """
546 self._logger.debug("image_path(%s)" % image_path)
547 self._logger.debug("Detecting USB stick device...")
548 usb_dev = self._probe_host_usb_dev()
549 if not usb_dev:
550 self._logger.error("No usb device connected to servo")
551 return False
552
553 try:
554 if image_path.startswith(self._HTTP_PREFIX):
555 self._logger.debug("Image path is a URL, downloading image")
556 urllib.urlretrieve(image_path, usb_dev)
557 else:
558 shutil.copyfile(image_path, usb_dev)
559 except IOError as e:
560 self._logger.error("Failed to transfer image to USB device: %s ( %d ) ",
561 e.strerror, e.errno)
562 return False
563 except urllib.ContentTooShortError:
564 self._logger.error("Failed to download URL: %s to USB device: %s",
565 image_path, usb_dev)
566 return False
567 except BaseException as e:
568 self._logger.error("Unexpected exception downloading %s to %s: %s",
569 image_path, usb_dev, str(e))
570 return False
J. Richard Barnettee4125af2013-02-26 18:31:56 -0800571 finally:
572 # We just plastered the partition table for a block device.
573 # Pass or fail, we mustn't go without telling the kernel about
574 # the change, or it will punish us with sporadic, hard-to-debug
575 # failures.
576 subprocess.call(["sync"])
577 subprocess.call(["blockdev", "--rereadpt", usb_dev])
Simran Basia9f41032012-05-11 14:21:58 -0700578 return True
579
580 def make_image_noninteractive(self):
581 """Makes the recovery image noninteractive.
582
583 A noninteractive image will reboot automatically after installation
584 instead of waiting for the USB device to be removed to initiate a system
585 reboot.
586
587 Mounts partition 1 of the image stored on usb_dev and creates a file
588 called "non_interactive" so that the image will become noninteractive.
589
590 Returns:
591 True|False: True if process completed successfully, False if error
592 occurred.
593 """
594 result = True
595 usb_dev = self._probe_host_usb_dev()
596 if not usb_dev:
597 self._logger.error("No usb device connected to servo")
598 return False
599 # Create TempDirectory
600 tmpdir = tempfile.mkdtemp()
601 if tmpdir:
602 # Mount drive to tmpdir.
603 partition_1 = "%s1" % usb_dev
604 rc = subprocess.call(["mount", partition_1, tmpdir])
605 if rc == 0:
606 # Create file 'non_interactive'
607 non_interactive_file = os.path.join(tmpdir, "non_interactive")
608 try:
609 open(non_interactive_file, "w").close()
610 except IOError as e:
611 self._logger.error("Failed to create file %s : %s ( %d )",
612 non_interactive_file, e.strerror, e.errno)
613 result = False
614 except BaseException as e:
615 self._logger.error("Unexpected Exception creating file %s : %s",
616 non_interactive_file, str(e))
617 result = False
618 # Unmount drive regardless if file creation worked or not.
619 rc = subprocess.call(["umount", partition_1])
620 if rc != 0:
621 self._logger.error("Failed to unmount USB Device")
622 result = False
623 else:
624 self._logger.error("Failed to mount USB Device")
625 result = False
626
627 # Delete tmpdir. May throw exception if 'umount' failed.
628 try:
629 os.rmdir(tmpdir)
630 except OSError as e:
631 self._logger.error("Failed to remove temp directory %s : %s",
632 tmpdir, str(e))
633 return False
634 except BaseException as e:
635 self._logger.error("Unexpected Exception removing tempdir %s : %s",
636 tmpdir, str(e))
637 return False
638 else:
639 self._logger.error("Failed to create temp directory.")
640 return False
641 return result
642
Todd Broch352b4b22013-03-22 09:48:40 -0700643 def set_get_all(self, cmds):
644 """Set &| get one or more control values.
645
646 Args:
647 cmds: list of control[:value] to get or set.
648
649 Returns:
650 rv: list of responses from calling get or set methods.
651 """
652 rv = []
653 for cmd in cmds:
654 if ':' in cmd:
655 (control, value) = cmd.split(':')
656 rv.append(self.set(control, value))
657 else:
658 rv.append(self.get(cmd))
659 return rv
660
Todd Broche505b8d2011-03-21 18:19:54 -0700661 def get(self, name):
662 """Get control value.
663
664 Args:
665 name: name string of control
666
667 Returns:
668 Response from calling drv get method. Value is reformatted based on
669 control's dictionary parameters
670
671 Raises:
672 HwDriverError: Error occurred while using drv
673 """
674 self._logger.debug("name(%s)" % (name))
Vadim Bendeburyc3a83cf2015-03-24 13:07:00 -0700675 if name == 'serialname':
676 if self._serialname:
677 return self._serialname
678 return 'unknown'
Todd Broche505b8d2011-03-21 18:19:54 -0700679 (param, drv) = self._get_param_drv(name)
680 try:
681 val = drv.get()
682 rd_val = self._syscfg.reformat_val(param, val)
Todd Brochb042e7a2011-12-14 17:41:36 -0800683 self._logger.debug("%s = %s" % (name, rd_val))
Todd Broche505b8d2011-03-21 18:19:54 -0700684 return rd_val
Todd Brochfbc499d2011-06-16 16:09:58 -0700685 except AttributeError, error:
686 self._logger.error("Getting %s: %s" % (name, error))
687 raise
Vic Yangbe6cf262012-09-10 10:40:56 +0800688 except HwDriverError:
Todd Broche505b8d2011-03-21 18:19:54 -0700689 self._logger.error("Getting %s" % (name))
690 raise
Todd Brochd6061672012-05-11 15:52:47 -0700691
Todd Broche505b8d2011-03-21 18:19:54 -0700692 def get_all(self, verbose):
693 """Get all controls values.
694
695 Args:
696 verbose: Boolean on whether to return doc info as well
697
698 Returns:
699 string creating from trying to get all values of all controls. In case of
700 error attempting access to control, response is 'ERR'.
701 """
Vadim Bendeburyb07944c2013-01-16 10:47:10 -0800702 rsp = []
Todd Broche505b8d2011-03-21 18:19:54 -0700703 for name in self._syscfg.syscfg_dict['control']:
704 self._logger.debug("name = %s" %name)
705 try:
706 value = self.get(name)
707 except Exception:
708 value = "ERR"
709 pass
710 if verbose:
Vadim Bendeburyb07944c2013-01-16 10:47:10 -0800711 rsp.append("GET %s = %s :: %s" % (name, value, self.doc(name)))
Todd Broche505b8d2011-03-21 18:19:54 -0700712 else:
Vadim Bendeburyb07944c2013-01-16 10:47:10 -0800713 rsp.append("%s:%s" % (name, value))
714 return '\n'.join(sorted(rsp))
Todd Broche505b8d2011-03-21 18:19:54 -0700715
716 def set(self, name, wr_val_str):
717 """Set control.
718
719 Args:
720 name: name string of control
721 wr_val_str: value string to write. Can be integer, float or a
722 alpha-numerical that is mapped to a integer or float.
723
724 Raises:
725 HwDriverError: Error occurred while using driver
726 """
727 self._logger.debug("name(%s) wr_val(%s)" % (name, wr_val_str))
728 (params, drv) = self._get_param_drv(name, False)
729 wr_val = self._syscfg.resolve_val(params, wr_val_str)
730 try:
731 drv.set(wr_val)
Vic Yangbe6cf262012-09-10 10:40:56 +0800732 except HwDriverError:
Todd Broche505b8d2011-03-21 18:19:54 -0700733 self._logger.error("Setting %s -> %s" % (name, wr_val_str))
734 raise
735 # TODO(tbroch) Figure out why despite allow_none=True for both xmlrpc server
736 # & client I still have to return something to appease the
737 # marshall/unmarshall
738 return True
739
Todd Brochd6061672012-05-11 15:52:47 -0700740 def hwinit(self, verbose=False):
741 """Initialize all controls.
742
743 These values are part of the system config XML files of the form
744 init=<value>. This command should be used by clients wishing to return the
745 servo and DUT its connected to a known good/safe state.
746
Vadim Bendeburybb51dd42013-01-31 13:47:46 -0800747 Note that initialization errors are ignored (as in some cases they could
748 be caused by DUT firmware deficiencies). This might need to be fine tuned
749 later.
750
Todd Brochd6061672012-05-11 15:52:47 -0700751 Args:
752 verbose: boolean, if True prints info about control initialized.
753 Otherwise prints nothing.
Vadim Bendebury5934e4b2013-02-06 13:57:54 -0800754
755 Returns:
756 This function is called across RPC and as such is expected to return
757 something unless transferring 'none' across is allowed. Hence adding a
758 dummy return value to make things simpler.
Todd Brochd6061672012-05-11 15:52:47 -0700759 """
Todd Brochd9acf0a2012-12-05 13:43:06 -0800760 for control_name, value in self._syscfg.hwinit:
Todd Broch3ec8df02012-11-20 10:53:03 -0800761 try:
John Carey6fe2bbf2015-08-31 16:13:03 -0700762 # Workaround for bug chrome-os-partner:42349. Without this check, the
763 # gpio will briefly pulse low if we set it from high to high.
764 if self.get(control_name) != value:
765 self.set(control_name, value)
Todd Broch3ec8df02012-11-20 10:53:03 -0800766 except Exception as e:
Todd Broch3ec8df02012-11-20 10:53:03 -0800767 self._logger.error("Problem initializing %s -> %s :: %s",
768 control_name, value, str(e))
Todd Brochd6061672012-05-11 15:52:47 -0700769 if verbose:
770 self._logger.info('Initialized %s to %s', control_name, value)
Nick Sandersbc836282015-12-08 21:19:23 -0800771
772 # Init keyboard after all the intefaces are up.
773 self._keyboard = self._init_keyboard_handler(self, self._board)
Vadim Bendebury5934e4b2013-02-06 13:57:54 -0800774 return True
Todd Broch3ec8df02012-11-20 10:53:03 -0800775
Todd Broche505b8d2011-03-21 18:19:54 -0700776 def echo(self, echo):
777 """Dummy echo function for testing/examples.
778
779 Args:
780 echo: string to echo back to client
781 """
782 self._logger.debug("echo(%s)" % (echo))
783 return "ECH0ING: %s" % (echo)
784
J. Richard Barnettee2820552013-03-14 16:13:46 -0700785 def get_board(self):
786 """Return the board specified at startup, if any."""
787 return self._board
788
Simran Basia23c1392013-08-06 14:59:10 -0700789 def get_version(self):
790 """Get servo board version."""
791 return self._version
792
Yusuf Mohsinally29e30d22014-01-14 15:29:17 -0800793 def power_long_press(self):
794 """Simulate a long power button press."""
795 # After a long power press, the EC may ignore the next power
796 # button press (at least on Alex). To guarantee that this
797 # won't happen, we need to allow the EC one second to
798 # collect itself.
799 self._keyboard.power_long_press()
800 return True
801
802 def power_normal_press(self):
803 """Simulate a normal power button press."""
804 self._keyboard.power_normal_press()
805 return True
806
807 def power_short_press(self):
808 """Simulate a short power button press."""
809 self._keyboard.power_short_press()
810 return True
811
812 def power_key(self, secs=''):
813 """Simulate a power button press.
814
815 Args:
816 secs: Time in seconds to simulate the keypress.
817 """
818 self._keyboard.power_key(secs)
819 return True
820
821 def ctrl_d(self, press_secs=''):
822 """Simulate Ctrl-d simultaneous button presses."""
823 self._keyboard.ctrl_d(press_secs)
824 return True
825
826 def ctrl_u(self):
827 """Simulate Ctrl-u simultaneous button presses."""
828 self._keyboard.ctrl_u()
829 return True
830
831 def ctrl_enter(self, press_secs=''):
832 """Simulate Ctrl-enter simultaneous button presses."""
833 self._keyboard.ctrl_enter(press_secs)
834 return True
835
836 def d_key(self, press_secs=''):
837 """Simulate Enter key button press."""
838 self._keyboard.d_key(press_secs)
839 return True
840
841 def ctrl_key(self, press_secs=''):
842 """Simulate Enter key button press."""
843 self._keyboard.ctrl_key(press_secs)
844 return True
845
846 def enter_key(self, press_secs=''):
847 """Simulate Enter key button press."""
848 self._keyboard.enter_key(press_secs)
849 return True
850
851 def refresh_key(self, press_secs=''):
852 """Simulate Refresh key (F3) button press."""
853 self._keyboard.refresh_key(press_secs)
854 return True
855
856 def ctrl_refresh_key(self, press_secs=''):
857 """Simulate Ctrl and Refresh (F3) simultaneous press.
858
859 This key combination is an alternative of Space key.
860 """
861 self._keyboard.ctrl_refresh_key(press_secs)
862 return True
863
864 def imaginary_key(self, press_secs=''):
865 """Simulate imaginary key button press.
866
867 Maps to a key that doesn't physically exist.
868 """
869 self._keyboard.imaginary_key(press_secs)
870 return True
871
Todd Brochdbb09982011-10-02 07:14:26 -0700872
Todd Broche505b8d2011-03-21 18:19:54 -0700873def test():
874 """Integration testing.
875
876 TODO(tbroch) Enhance integration test and add unittest (see mox)
877 """
878 logging.basicConfig(level=logging.DEBUG,
879 format="%(asctime)s - %(name)s - " +
880 "%(levelname)s - %(message)s")
881 # configure server & listen
882 servod_obj = Servod(1)
883 # 4 == number of interfaces on a FT4232H device
884 for i in xrange(4):
885 if i == 1:
886 # its an i2c interface ... see __init__ for details and TODO to make
887 # this configureable
888 servod_obj._interface_list[i].wr_rd(0x21, [0], 1)
889 else:
890 # its a gpio interface
891 servod_obj._interface_list[i].wr_rd(0)
892
893 server = SimpleXMLRPCServer.SimpleXMLRPCServer(("localhost", 9999),
894 allow_none=True)
895 server.register_introspection_functions()
896 server.register_multicall_functions()
897 server.register_instance(servod_obj)
898 logging.info("Listening on localhost port 9999")
899 server.serve_forever()
900
901if __name__ == "__main__":
902 test()
903
904 # simple client transaction would look like
905 """
906 remote_uri = 'http://localhost:9999'
907 client = xmlrpclib.ServerProxy(remote_uri, verbose=False)
908 send_str = "Hello_there"
909 print "Sent " + send_str + ", Recv " + client.echo(send_str)
910 """