# pylint: disable=no-member
import subprocess
import attr
from ..factory import target_factory
from ..protocol import BootstrapProtocol
from ..step import step
from .common import Driver
from ..util.managedfile import ManagedFile
from ..util.timeout import Timeout
from ..util.helper import processwrapper
[docs]@target_factory.reg_driver
@attr.s(eq=False)
class MXSUSBDriver(Driver, BootstrapProtocol):
bindings = {
"loader": {"MXSUSBLoader", "NetworkMXSUSBLoader"},
}
image = attr.ib(default=None)
[docs] def __attrs_post_init__(self):
super().__attrs_post_init__()
# FIXME make sure we always have an environment or config
if self.target.env:
self.tool = self.target.env.config.get_tool('mxs-usb-loader') or 'mxs-usb-loader'
else:
self.tool = 'mxs-usb-loader'
[docs] def on_activate(self):
pass
[docs] def on_deactivate(self):
pass
[docs] @Driver.check_active
@step(args=['filename'])
def load(self, filename=None):
if filename is None and self.image is not None:
filename = self.target.env.config.get_image_path(self.image)
mf = ManagedFile(filename, self.loader)
mf.sync_to_resource()
processwrapper.check_output(
self.loader.command_prefix + [self.tool, "0", mf.get_remote_path()]
)
[docs]@target_factory.reg_driver
@attr.s(eq=False)
class IMXUSBDriver(Driver, BootstrapProtocol):
bindings = {
"loader": {"IMXUSBLoader", "NetworkIMXUSBLoader", "MXSUSBLoader", "NetworkMXSUSBLoader"},
}
image = attr.ib(default=None)
[docs] def __attrs_post_init__(self):
super().__attrs_post_init__()
# FIXME make sure we always have an environment or config
if self.target.env:
self.tool = self.target.env.config.get_tool('imx-usb-loader') or 'imx-usb-loader'
else:
self.tool = 'imx-usb-loader'
[docs] def on_activate(self):
pass
[docs] def on_deactivate(self):
pass
[docs] @Driver.check_active
@step(args=['filename'])
def load(self, filename=None):
if filename is None and self.image is not None:
filename = self.target.env.config.get_image_path(self.image)
mf = ManagedFile(filename, self.loader)
mf.sync_to_resource()
processwrapper.check_output(
self.loader.command_prefix +
[self.tool, "-p", str(self.loader.path), "-c", mf.get_remote_path()]
)
[docs]@target_factory.reg_driver
@attr.s(eq=False)
class RKUSBDriver(Driver, BootstrapProtocol):
bindings = {
"loader": {"RKUSBLoader", "NetworkRKUSBLoader"},
}
image = attr.ib(default=None)
usb_loader = attr.ib(default=None)
[docs] def __attrs_post_init__(self):
super().__attrs_post_init__()
# FIXME make sure we always have an environment or config
if self.target.env:
self.tool = self.target.env.config.get_tool('rk-usb-loader') or 'rk-usb-loader'
else:
self.tool = 'rk-usb-loader'
[docs] def on_activate(self):
pass
[docs] def on_deactivate(self):
pass
[docs] @Driver.check_active
@step(args=['filename'])
def load(self, filename=None):
if self.target.env:
usb_loader = self.target.env.config.get_image_path(self.usb_loader)
mf = ManagedFile(usb_loader, self.loader)
mf.sync_to_resource()
timeout = Timeout(3.0)
while True:
try:
processwrapper.check_output(
self.loader.command_prefix +
[self.tool, 'db', mf.get_remote_path()]
)
break
except subprocess.CalledProcessError:
if timeout.expired:
raise
if filename is None and self.image is not None:
filename = self.target.env.config.get_image_path(self.image)
mf = ManagedFile(filename, self.loader)
mf.sync_to_resource()
timeout = Timeout(3.0)
while True:
try:
processwrapper.check_output(
self.loader.command_prefix +
[self.tool, 'wl', '0x40', mf.get_remote_path()]
)
break
except subprocess.CalledProcessError:
if timeout.expired:
raise
[docs]@target_factory.reg_driver
@attr.s(eq=False)
class UUUDriver(Driver, BootstrapProtocol):
bindings = {
"loader": {"IMXUSBLoader", "NetworkIMXUSBLoader", "MXSUSBLoader", "NetworkMXSUSBLoader"},
}
image = attr.ib(default=None)
cmd = attr.ib(default='spl', validator=attr.validators.instance_of(str))
[docs] def __attrs_post_init__(self):
super().__attrs_post_init__()
# FIXME make sure we always have an environment or config
if self.target.env:
self.tool = self.target.env.config.get_tool('uuu-loader') or 'uuu-loader'
else:
self.tool = 'uuu-loader'
[docs] def on_activate(self):
pass
[docs] def on_deactivate(self):
pass
[docs] @Driver.check_active
@step(args=['filename'])
def load(self, filename=None):
if filename is None and self.image is not None:
filename = self.target.env.config.get_image_path(self.image)
mf = ManagedFile(filename, self.loader)
mf.sync_to_resource()
processwrapper.check_output(
self.loader.command_prefix + [self.tool, mf.get_remote_path(), self.cmd]
)