/*
 * Copyright 2017, The Android Open Source Project
 *
 * 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.
 */

#pragma once

#include "result.h"

#include <arpa/inet.h>

class Message;

class Socket {
public:
    Socket();
    Socket(const Socket&) = delete;
    ~Socket();

    Socket& operator=(const Socket&) = delete;

    int get() const { return mSocketFd; }
    // Open a socket, |domain|, |type| and |protocol| are as described in the
    // man pages for socket.
    Result open(int domain, int type, int protocol);
    // Bind to a generic |sockaddr| of size |sockaddrLength|
    Result bind(const void* sockaddr, size_t sockaddrLength);
    // Bind to an IP |address| and |port|
    Result bindIp(in_addr_t address, uint16_t port);
    // Bind a raw socket to the interface with index |interfaceIndex|.
    Result bindRaw(unsigned int interfaceIndex);
    // Send data in |message| on an IP socket to
    // |destinationAddress|:|destinationPort|, the message will egress on the
    // interface specified by |interfaceIndex|
    Result sendOnInterface(unsigned int interfaceIndex,
                           in_addr_t destinationAddress,
                           uint16_t destinationPort,
                           const Message& message);
    // Send |message| as a UDP datagram on a raw socket. The source address of
    // the message will be |source|:|sourcePort| and the destination will be
    // |destination|:|destinationPort|. The message will be sent on the
    // interface indicated by |interfaceIndex|.
    Result sendRawUdp(in_addr_t source,
                      uint16_t sourcePort,
                      in_addr_t destination,
                      uint16_t destinationPort,
                      unsigned int interfaceIndex,
                      const Message& message);
    // Receive data on the socket and indicate which interface the data was
    // received on in |interfaceIndex|. The received data is placed in |message|
    Result receiveFromInterface(Message* message, unsigned int* interfaceIndex);
    // Receive UDP data on a raw socket. Expect that the protocol in the IP
    // header is UDP and that the port in the UDP header is |expectedPort|. If
    // the received data is valid then |isValid| will be set to true, otherwise
    // false. The validity check includes the expected values as well as basic
    // size requirements to fit the expected protocol headers.  The method will
    // only return an error result if the actual receiving fails.
    Result receiveRawUdp(uint16_t expectedPort,
                         Message* message,
                         bool* isValid);
    // Enable |optionName| on option |level|. These values are the same as used
    // in setsockopt calls.
    Result enableOption(int level, int optionName);
private:
    int mSocketFd;
};