debug-tools/host_manager_11ad/UdpNetworkInterface.cpp

/*
 * Copyright (c) 2017, The Linux Foundation. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above
 *       copyright notice, this list of conditions and the following
 *       disclaimer in the documentation and/or other materials provided
 *       with the distribution.
 *     * Neither the name of The Linux Foundation nor the names of its
 *       contributors may be used to endorse or promote products derived
 *       from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#ifdef __linux
#include <stdlib.h>
#include <string.h>
#include <cerrno>
#include <unistd.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <net/if.h>
#include <netinet/in.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <dirent.h>
#include <stdio.h>

#elif _WINDOWS
#include <windows.h>

#else
#include <dirent.h>
#include <sys/socket.h>
#include <net/route.h>
#include <net/if.h>
#include <arpa/inet.h>
#endif

#include <iostream>
#include <fstream>
#include <sstream>
#include "DebugLogger.h"
#include "UdpNetworkInterface.h"

UdpNetworkInterface::UdpNetworkInterface(std::string broadcastIp, int portIn, int portOut)
{
    m_portIn = portIn;
    m_portOut = portOut;
    m_broadcastIp = broadcastIp;
#ifndef _WINDOWS
    // create UDP socket
    m_socket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
    if (m_socket < 0)
    {
        std::string error = "Can't open UDP socket";
        LOG_WARNING << error << std::endl;
        throw error;
    }

    // set broadcast flag on
    int enabled = 1;
    if (setsockopt(m_socket, SOL_SOCKET, SO_BROADCAST, (char*)&enabled, sizeof(enabled)) < 0)
    {
        std::string error = "Can't set broadcast option for udp socket, error: ";
        error += strerror(errno);

        LOG_WARNING << error << std::endl;
        shutdown(m_socket, SHUT_RDWR);
        throw error;
    }

    // bind socket to portIn
    struct sockaddr_in address;
    address.sin_family = AF_INET;
    address.sin_addr.s_addr = INADDR_ANY;
    address.sin_port = htons(m_portIn);
    if (::bind(m_socket, (struct sockaddr *)&address, sizeof(struct sockaddr_in)) < 0)
    {
        std::string error = "Can't bind socket to port, error: ";
        error += strerror(errno);

        LOG_WARNING << error << std::endl;
        throw error;
    }
#else
    std::string error = "UDP socket is supported on linux OS only";
    throw error;
#endif
}

int UdpNetworkInterface::Send(std::string message)
{
#ifndef _WINDOWS
    struct sockaddr_in dstAddress;
    dstAddress.sin_family = AF_INET;
    inet_pton(AF_INET, m_broadcastIp.c_str(), &dstAddress.sin_addr.s_addr);
    dstAddress.sin_port = htons(m_portOut);
    int messageSize = message.length() * sizeof(char);
    int result = sendto(m_socket, message.c_str(), messageSize, 0, (sockaddr*)&dstAddress, sizeof(dstAddress));
    LOG_VERBOSE << "INFO : sendto with sock_out=" << m_socket << ", message=" << message << " messageSize=" << messageSize << " returned with " << result << std::endl;
    if (result < 0)
    {
        LOG_WARNING << "ERROR : Cannot send udp broadcast message, error " << ": " << strerror(errno) << std::endl;
        return 0;
    }
    return messageSize;
#else
    return 0;
#endif
}

const char* UdpNetworkInterface::Receive(int len)
{
#ifndef _WINDOWS
    char* buf = new char[len];
    if (nullptr == buf)
    {
        LOG_ERROR << "Cannot allocate receive buffer for UDP messages" << std::endl;
        return "";
    }

    if (recvfrom(m_socket, buf, len, 0, NULL, 0) < 0)
    {
        LOG_WARNING << "Can't receive from port " << m_portIn << std::endl;
        return "";
    }
    return buf;
#else
    return "";
#endif
}

void UdpNetworkInterface::Close()
{
#ifndef _WINDOWS
    shutdown(m_socket, SHUT_RDWR);
#endif
}