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
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Itay Marom
Cisco Systems, Inc.
Copyright (c) 2015-2015 Cisco Systems, Inc.
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.
"""
try:
# support import for Python 2
import outer_packages
except ImportError:
# support import for Python 3
import client.outer_packages
from client_utils.jsonrpc_client import JsonRpcClient, BatchMessage
from client_utils.packet_builder import CTRexPktBuilder
import json
from common.trex_streams import *
import argparse
import tempfile
import subprocess
import os
class SimRun(object):
def __init__ (self, yaml_file, dp_core_count, core_index, packet_limit, output_filename, is_valgrind, is_gdb):
self.yaml_file = yaml_file
self.output_filename = output_filename
self.dp_core_count = dp_core_count
self.core_index = core_index
self.packet_limit = packet_limit
self.is_valgrind = is_valgrind
self.is_gdb = is_gdb
# dummies
self.handler = 0
self.port_id = 0
self.mul = {"op": "abs",
"type": "raw",
"value": 1}
self.duration = -1
def load_yaml_file (self):
streams_db = CStreamsDB()
stream_list = streams_db.load_yaml_file(self.yaml_file)
streams_json = []
for stream in stream_list.compiled:
stream_json = {"id":1,
"jsonrpc": "2.0",
"method": "add_stream",
"params": {"handler": self.handler,
"port_id": self.port_id,
"stream_id": stream.stream_id,
"stream": stream.stream}
}
streams_json.append(stream_json)
return streams_json
def generate_start_cmd (self):
return {"id":1,
"jsonrpc": "2.0",
"method": "start_traffic",
"params": {"handler": self.handler,
"port_id": self.port_id,
"mul": self.mul,
"duration": self.duration}
}
def run (self):
# load the streams
cmds_json = (self.load_yaml_file())
cmds_json.append(self.generate_start_cmd())
f = tempfile.NamedTemporaryFile(delete = False)
f.write(json.dumps(cmds_json))
f.close()
try:
cmd = ['bp-sim-64-debug', '--sl', '--cores', str(self.dp_core_count), '--core_index', str(self.core_index), '-f', f.name, '-o', self.output_filename]
if self.is_valgrind:
cmd = ['valgrind', '--leak-check=full'] + cmd
elif self.is_gdb:
cmd = ['gdb', '--args'] + cmd
subprocess.call(cmd)
finally:
os.unlink(f.name)
def is_valid_file(filename):
if not os.path.isfile(filename):
raise argparse.ArgumentTypeError("The file '%s' does not exist" % filename)
return filename
def unsigned_int (x):
x = int(x)
if x <= 0:
raise argparse.ArgumentTypeError("argument must be >= 1")
return x
def setParserOptions():
parser = argparse.ArgumentParser(prog="stl_sim.py")
parser.add_argument("input_file",
help = "input file in YAML or Python format",
type = is_valid_file)
parser.add_argument("output_file",
help = "output file in ERF format")
parser.add_argument("-c", "--cores",
help = "DP core count [default is 1]",
default = 1,
type = int,
choices = xrange(1, 9))
parser.add_argument("-n", "--core_index",
help = "DP core index to examine [default is 0]",
default = 0,
type = int)
parser.add_argument("-j", "--join",
help = "run and join output from 0..core_count [default is False]",
default = False,
type = bool)
parser.add_argument("-l", "--limit",
help = "limit test total packet count [default is 5000]",
default = 5000,
type = unsigned_int)
group = parser.add_mutually_exclusive_group()
group.add_argument("-x", "--valgrind",
help = "run under valgrind [default is False]",
action = "store_true",
default = False)
group.add_argument("-g", "--gdb",
help = "run under GDB [default is False]",
action = "store_true",
default = False)
return parser
def validate_args (parser, options):
if options.core_index < 0 or options.core_index >= options.cores:
parser.error("DP core index valid range is 0 to {0}".format(options.cores - 1))
def main ():
parser = setParserOptions()
options = parser.parse_args()
validate_args(parser, options)
r = SimRun(options.input_file,
options.cores,
options.core_index,
options.limit,
options.output_file,
options.valgrind,
options.gdb)
r.run()
if __name__ == '__main__':
main()
|
