1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
#!/usr/bin/env python
"""
mbed SDK
Copyright (c) 2011-2018 ARM Limited
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
import telnetlib
import socket
from .conn_primitive import ConnectorPrimitive, ConnectorPrimitiveException
class FastmodelConnectorPrimitive(ConnectorPrimitive):
def __init__(self, name, config):
ConnectorPrimitive.__init__(self, name)
self.config = config
self.fm_config = config.get('fm_config', None)
self.platform_name = config.get('platform_name', None)
self.image_path = config.get('image_path', None)
self.polling_timeout = int(config.get('polling_timeout', 60))
# FastModel Agent tool-kit
self.fm_agent_module = None
self.resource = None
# Initialize FastModel
if self.__fastmodel_init():
# FastModel Launch load and run, equivalent to DUT connection, flashing and reset...
self.__fastmodel_launch()
self.__fastmodel_load(self.image_path)
self.__fastmodel_run()
def __fastmodel_init(self):
"""! Initialize models using fm_agent APIs """
self.logger.prn_inf("Initializing FastModel...")
try:
self.fm_agent_module = __import__("fm_agent")
except ImportError as e:
self.logger.prn_err("unable to load mbed-fastmodel-agent module. Check if the module install correctly.")
self.fm_agent_module = None
self.logger.prn_err("Importing failed : %s" % str(e))
raise ConnectorPrimitiveException("Importing failed : %s" % str(e))
try:
self.resource = self.fm_agent_module.FastmodelAgent(logger=self.logger)
self.resource.setup_simulator(self.platform_name,self.fm_config)
if self.__resource_allocated():
pass
except self.fm_agent_module.SimulatorError as e:
self.logger.prn_err("module fm_agent, create() failed: %s"% str(e))
raise ConnectorPrimitiveException("FastModel Initializing failed as throw SimulatorError!")
return True
def __fastmodel_launch(self):
"""! launch the FastModel"""
self.logger.prn_inf("Launching FastModel...")
try:
if not self.resource.start_simulator():
raise ConnectorPrimitiveException("FastModel running failed, run_simulator() return False!")
except self.fm_agent_module.SimulatorError as e:
self.logger.prn_err("start_simulator() failed: %s"% str(e))
raise ConnectorPrimitiveException("FastModel launching failed as throw FastModelError!")
def __fastmodel_run(self):
"""! Use fm_agent API to run the FastModel """
self.logger.prn_inf("Running FastModel...")
try:
if not self.resource.run_simulator():
raise ConnectorPrimitiveException("FastModel running failed, run_simulator() return False!")
except self.fm_agent_module.SimulatorError as e:
self.logger.prn_err("run_simulator() failed: %s"% str(e))
raise ConnectorPrimitiveException("FastModel running failed as throw SimulatorError!")
def __fastmodel_load(self, filename):
"""! Use fm_agent API to load image to FastModel, this is functional equivalent to flashing DUT"""
self.logger.prn_inf("loading FastModel with image '%s'..."% filename)
try:
if not self.resource.load_simulator(filename):
raise ConnectorPrimitiveException("FastModel loading failed, load_simulator() return False!")
except self.fm_agent_module.SimulatorError as e:
self.logger.prn_err("run_simulator() failed: %s"% str(e))
raise ConnectorPrimitiveException("FastModel loading failed as throw SimulatorError!")
def __resource_allocated(self):
"""! Check whether FastModel resource been allocated
@return True or throw an exception
"""
if self.resource:
return True
else:
self.logger.prn_err("FastModel resource not available!")
return False
def read(self, count):
"""! Read data from DUT, count is not used for FastModel"""
date = str()
if self.__resource_allocated():
try:
data = self.resource.read()
except self.fm_agent_module.SimulatorError as e:
self.logger.prn_err("FastmodelConnectorPrimitive.read() failed: %s"% str(e))
else:
return data
else:
return False
def write(self, payload, log=False):
"""! Write 'payload' to DUT"""
if self.__resource_allocated():
if log:
self.logger.prn_txd(payload)
try:
self.resource.write(payload)
except self.fm_agent_module.SimulatorError as e:
self.logger.prn_err("FastmodelConnectorPrimitive.write() failed: %s"% str(e))
else:
return True
else:
return False
def flush(self):
"""! flush not supported in FastModel_module"""
pass
def connected(self):
"""! return whether FastModel is connected """
if self.__resource_allocated():
return self.resource.is_simulator_alive
else:
return False
def finish(self):
"""! shutdown the FastModel and release the allocation """
if self.__resource_allocated():
try:
self.resource.shutdown_simulator()
self.resource = None
except self.fm_agent_module.SimulatorError as e:
self.logger.prn_err("FastmodelConnectorPrimitive.finish() failed: %s"% str(e))
def reset(self):
if self.__resource_allocated():
try:
if not self.resource.reset_simulator():
self.logger.prn_err("FastModel reset failed, reset_simulator() return False!")
except self.fm_agent_module.SimulatorError as e:
self.logger.prn_err("FastmodelConnectorPrimitive.reset() failed: %s"% str(e))
def __del__(self):
self.finish()
|
