// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2008 University of California
//
// BOINC is free software; you can redistribute it and/or modify it
// under the terms of the GNU Lesser General Public License
// as published by the Free Software Foundation,
// either version 3 of the License, or (at your option) any later version.
//
// BOINC is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with BOINC.  If not, see <http://www.gnu.org/licenses/>.

#ifndef BOINC_NETWORK_H
#define BOINC_NETWORK_H

#include <string.h>
#ifdef _WIN32
#include "boinc_win.h"
// WxWidgets can't deal with modern network code (winsock2.h)
// so use old code on Win
#define sockaddr_storage sockaddr_in
#else
#include <sys/select.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#endif

struct FDSET_GROUP {
    fd_set read_fds;
    fd_set write_fds;
    fd_set exc_fds;
    int max_fd;

    void zero() {
        FD_ZERO(&read_fds);
        FD_ZERO(&write_fds);
        FD_ZERO(&exc_fds);
        max_fd = -1;
    }
};

extern bool is_localhost(sockaddr_storage& s);
extern bool same_ip_addr(sockaddr_storage& s1, sockaddr_storage& s2);
extern int resolve_hostname(const char* hostname, sockaddr_storage& ip_addr);
extern int resolve_hostname_or_ip_addr(const char* hostname, sockaddr_storage& ip_addr);
extern int boinc_socket(int& sock, int protocol=AF_INET);
    // create a stream-mode socket of the given family
extern int boinc_socket_asynch(int sock, bool asynch);
    // make an existing socket asynchronous or synchronous
extern void boinc_close_socket(int sock);
extern int get_socket_error(int fd);
extern const char* socket_error_str();
extern void reset_dns();
extern int boinc_get_port(bool is_remote, int& port);
extern bool network_connected();

#ifdef _WIN32
inline int addr_len(sockaddr_storage&) {
    return (int) sizeof(sockaddr_in);
}
#else
inline int addr_len(sockaddr_storage& s) {
    if (s.ss_family == AF_INET6) {
        return (int) sizeof(sockaddr_in6);
    }
    return (int) sizeof(sockaddr_in);
}
#endif

#if defined(_WIN32) && !defined(__CYGWIN32__)
typedef int BOINC_SOCKLEN_T;
#else
typedef socklen_t BOINC_SOCKLEN_T;
#endif

#if defined(_WIN32) && defined(USE_WINSOCK)
extern int WinsockInitialize();
extern int WinsockCleanup();
#endif
#endif
