aboutsummaryrefslogtreecommitdiffstats
path: root/GPL/traffic_profiles/trex/profile_trex_stateless_scale_class.py
blob: 33cf090da2fc803792b68df87e62013dac7a280f (plain)
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
# Copyright (c) 2023 Cisco and/or its affiliates.
#
# SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
#
# Licensed under the Apache License 2.0 or
# GNU General Public License v2.0 or later;  you may not use this file
# except in compliance with one of these Licenses. You
# may obtain a copy of the Licenses at:
#
#     http://www.apache.org/licenses/LICENSE-2.0
#     https://www.gnu.org/licenses/old-licenses/gpl-2.0-standalone.html
#
# Note: If this file is linked with Scapy, which is GPLv2+, your use of it
# must be under GPLv2+.  If at any point in the future it is no longer linked
# with Scapy (or other GPLv2+ licensed software), you are free to choose
# Apache 2.
#
# 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.

"""Base class for stream profiles for T-rex traffic generator.
"""

import socket
import struct

from random import choice
from string import ascii_letters

from trex.stl.api import *


class TrafficStreamsScaleClass:
    """Base class for stream profiles for T-rex traffic generator."""

    STREAM_TABLE = {
        "IMIX_v4": [
            {"size": 60, "pps": 28, "isg": 0},
            {"size": 590, "pps": 20, "isg": 0.1},
            {"size": 1514, "pps": 4, "isg": 0.2}
        ],
        "IMIX_v4_1": [
            {"size": 64, "pps": 28, "isg": 0},
            {"size": 570, "pps": 16, "isg": 0.1},
            {"size": 1518, "pps": 4, "isg": 0.2}
        ]
    }

    def __init__(self):
        # Default value of frame size, it will be overwritten by the value of
        # "framesize" parameter of "get_streams" method.
        self.framesize = 64

        # If needed, add your own parameters.

    def _gen_payload(self, length):
        """Generate payload.

        If needed, implement your own algorithm.

        :param length: Length of generated payload.
        :type length: int
        :returns: The generated payload.
        :rtype: str
        """
        payload = ""
        for _ in range(length):
            payload += choice(ascii_letters)

        return payload

    def _get_start_end_ipv6(self, start_ip, end_ip):
        """Get start host and number of hosts from IPv6 as integer.

        :param start_ip: Start IPv6.
        :param end_ip: End IPv6.
        :type start_ip: string
        :type end_ip: string
        :return: Start host, number of hosts.
        :rtype tuple of int
        :raises: ValueError if start_ip is greater then end_ip.
        :raises: socket.error if the IP addresses are not valid IPv6 addresses.
        """
        try:
            ip1 = socket.inet_pton(socket.AF_INET6, start_ip)
            ip2 = socket.inet_pton(socket.AF_INET6, end_ip)

            hi1, lo1 = struct.unpack("!QQ", ip1)
            hi2, lo2 = struct.unpack("!QQ", ip2)

            if ((hi1 << 64) | lo1) > ((hi2 << 64) | lo2):
                raise ValueError("IPv6: start_ip is greater then end_ip")

            return lo1, abs(int(lo1) - int(lo2))

        except socket.error as err:
            print(err)
            raise

    def define_packets(self):
        """Define the packets to be sent from the traffic generator.

        This method MUST return:

            return base_pkt_a, base_pkt_b, vm1, vm2

            vm1 and vm2 CAN be None.

        :returns: Packets to be sent from the traffic generator.
        :rtype: tuple
        """
        raise NotImplementedError

    def create_streams(self):
        """Create traffic streams.

        Implement your own traffic streams.

        :returns: Traffic streams.
        :rtype: list
        """
        pkts, vms = self.define_packets()

        # Frame size is defined as an integer, e.g. 64, 1518:
        if isinstance(self.framesize, int):
            pkt_streams = list()
            lat_streams = list()
            for i in range(len(pkts)):
                payload_len = max(0, self.framesize - len(pkts[i]) - 4)

                pkt = STLPktBuilder(
                    pkt=pkts[i] / self._gen_payload(payload_len),
                    vm=vms[i]
                )
                pkt_lat = STLPktBuilder(
                    pkt=pkts[i] / self._gen_payload(payload_len),
                    vm=vms[i]
                )
                pkt_streams.append(
                    STLStream(
                        packet=pkt,
                        isg=10.0 * (i // (len(pkts) // 2)),
                        mode=STLTXCont(pps=9000)
                    )
                )
                lat_streams.append(
                    STLStream(
                        packet=pkt_lat,
                        isg=10.0 * (i // (len(pkts) // 2)),
                        flow_stats=STLFlowLatencyStats(pg_id=i),
                        mode=STLTXCont(pps=9000)
                    )
                )

            streams = list()
            streams.extend(pkt_streams)
            streams.extend(lat_streams)
            return streams

        # Frame size is defined as a string, e.g.IMIX_v4_1:
        elif isinstance(self.framesize, str):
            pkt_streams = list()
            for i in range(len(pkts)):
                for stream in self.STREAM_TABLE[self.framesize]:
                    payload_len = max(0, stream["size"] - len(pkts[i]) - 4)

                    pkt = STLPktBuilder(
                        pkt=pkts[i] / self._gen_payload(payload_len),
                        vm=vms[i]
                    )
                    pkt_streams.append(
                        STLStream(
                            packet=pkt,
                            isg=stream["isg"],
                            mode=STLTXCont(pps=stream["pps"])
                        )
                    )
            return pkt_streams

    def get_streams(self, **kwargs):
        """Get traffic streams created by "create_streams" method.

        If needed, add your own parameters.

        :param kwargs: Key-value pairs used by "create_streams" method while
        creating streams.
        :returns: Traffic streams.
        :rtype: list
        """
        self.framesize = kwargs["framesize"]
        self.rate = kwargs["rate"]

        return self.create_streams()