Socket.h

/*
 * Copyright (c) 2008-2015, Hazelcast, Inc. All Rights Reserved.
 *
 * 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.
 */
#ifndef HAZELCAST_SOCKET
#define HAZELCAST_SOCKET


#include "hazelcast/client/Address.h"
#include "hazelcast/util/AtomicBoolean.h"
#include <string>

#if  defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
#pragma warning(push)
#pragma warning(disable: 4251) //for dll export 
#endif 

#if  defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
#pragma comment(lib, "Ws2_32.lib")
#include <winsock2.h>
#include <ws2tcpip.h>
#else

#include <unistd.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/wait.h>
#include <sys/errno.h>
#include <sys/select.h>
#include <fcntl.h>

#endif


#if  defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
    typedef int socklen_t;
#endif

#if !defined(MSG_NOSIGNAL)
#  define MSG_NOSIGNAL 0
#endif

namespace hazelcast {
    namespace client {
        class HAZELCAST_API Socket {
        public:
            Socket(int socketId);
            Socket(const client::Address &address);
            virtual ~Socket();
            int connect(int timeoutInMillis);

            virtual int send(const void *buffer, int len) const;

            int receive(void *buffer, int len, int flag = 0) const;

            int getSocketId() const;

            void setRemoteEndpoint(const client::Address &address);

            const client::Address &getRemoteEndpoint() const;
            void close();

            client::Address getAddress() const;

            void setBlocking(bool blocking);

        private:

            Socket(const Socket &rhs);

            Socket& operator=(const Socket &rhs);

            client::Address remoteEndpoint;

            struct addrinfo *serverInfo;
            int socketId;
            util::AtomicBoolean isOpen;

            #if  defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
            WSADATA wsa_data;
            #endif

        };

        struct HAZELCAST_API socketPtrComp {
            bool operator ()(Socket const *const &lhs, Socket const *const &rhs) const;
        };

    }
}

#if  defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
#pragma warning(pop)
#endif 

#endif /* HAZELCAST_SOCKET */