IPv4StringToNetworkOctet.py

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from struct import pack

from Peach.transformer import Transformer


class Ipv4StringToNetworkOctet(Transformer):
    """Convert a dot notation IPv4 address into a 4 byte octet representation.
    """

    def realEncode(self, data):
        data = data.split('.')
        for i in range(len(data)):
            try:
                data[i] = int(data[i])
            except Exception as e:
                data[i] = 0
        for i in range(len(data), 4):
            data.append(0)
        return pack('!BBBB', data[0], data[1], data[2], data[3])