from trex_stl_lib.api import *
import os

# stream from pcap file. continues pps 10 in sec 

CP = os.path.join(os.path.dirname(__file__))

class STLS1(object):

    def create_stream (self):

        return STLProfile( [ STLStream( isg = 10.0, # start in delay 
                                        name    ='S0',
                                        packet = STLPktBuilder(pkt = os.path.join(CP, "yaml/udp_64B_no_crc.pcap")),
                                        mode = STLTXSingleBurst( pps = 10, total_pkts = 10),
                                        next = 'S1'), # point to next stream 

                             STLStream( self_start = False, # Stream is disabled. Will run because it is pointed from S0
                                        name    ='S1',
                                        packet  = STLPktBuilder(pkt = os.path.join(CP, "yaml/udp_594B_no_crc.pcap")),
                                        mode    = STLTXSingleBurst( pps = 10, total_pkts = 20),
                                        next    = 'S2' ),

                             STLStream(  self_start = False, # Stream is disabled. Will run because it is pointed from S1
                                         name   ='S2',
                                         packet = STLPktBuilder
# Copyright (c) 2021 Cisco and/or its affiliates.
# 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.

"""Linux limit library."""

from resources.libraries.python.ssh import exec_cmd_no_error

__all__ = [u"LimitUtil"]


class LimitUtil:
    """Class contains methods for getting or setting process resource limits."""

    @staticmethod
    def get_pid_limit(node, pid):
        """Get process resource limits.

        :param node: Node in the topology.
        :param pid: Process ID.
        :type node: dict
        :type pid: int
        """
        command = f"prlimit --noheadings --pid={pid}"
        message = f"Node {node[u'host']} failed to run: {command}"

        exec_cmd_no_error(node, command, sudo=True, message=message)

    @staticmethod
    def set_pid_limit(node, pid, resource, limit):
        """Set process resource limits.

        :param node: Node in the topology.
        :param pid: Process ID.
        :param resource: Resource to set limits.
        :param limit: Limit value.
        :type node: dict
        :type pid: int
        :type resource: str
        :type limit: str
        """
        command = f"prlimit --{resource}={limit} --pid={pid}"
        message = f"Node {node[u'host']} failed to run: {command}"

        exec_cmd_no_error(node, command, sudo=True, message=message)