Connection.cpp

Go to the documentation of this file.

#include "Connection.h"

Connection::Connection(int port){
        this->port = port;
        isConnected = false;
        
        WSAData wsaData;
        if(WSAStartup(MAKEWORD(2,2), &wsaData) != NO_ERROR)
                throw "Error at WSAStartup.";

        tcp = socket(AF_INET, SOCK_STREAM, 0);
        if(tcp == INVALID_SOCKET){
                WSACleanup();
                throw "Error creating socket.";
        }

        int on = 1;
        setsockopt(tcp, SOL_SOCKET, SO_REUSEADDR, (const char *)&on, sizeof(on));
        isOpen = false;
}


Connection::Connection(SOCKET s){
        tcp = s;
        this->port = 0;
        
        isConnected = false;
        isOpen = true;

}


int Connection::connectByIp(char *addr){
        struct hostent *hostentry;
        sockaddr_in serverInfo;

        hostentry = gethostbyname(addr);
        if(!hostentry)
                return(-1);
        serverInfo.sin_family = AF_INET;
        serverInfo.sin_port = htons(port);
        serverInfo.sin_addr = *((struct in_addr *)hostentry->h_addr);

        if(connect(tcp, (struct sockaddr *)&serverInfo, sizeof(serverInfo)) == SOCKET_ERROR){
                WSACleanup();
                return(-1);
        }
        isConnected = true;
        return(0);
}


Connection *Connection::waitForConnection(){
        SOCKET client;
        Connection *newConn;

        if(!isOpen){
                sin.sin_family = AF_INET;
                sin.sin_port = htons(port);
                sin.sin_addr.s_addr = INADDR_ANY;

                if(bind(tcp, (struct sockaddr *)&sin, sizeof(sin)) == SOCKET_ERROR){
                        WSACleanup();
                        LOG_ERROR(("ERROR BIND"));
                        return(NULL);
                }

                if(listen(tcp, 1) == SOCKET_ERROR){
                        LOG_ERROR(("ERROR LISTEN"));
                        WSACleanup();
                        closesocket(tcp);
                        return(NULL);
                }
        }
        client = accept(tcp, NULL, NULL);
        if(tcp == SOCKET_ERROR){
                LOG_ERROR(("ERROR ACCEPT"));
                WSACleanup();
                closesocket(tcp);
                return(NULL);
        }
        newConn = new Connection(tcp);
        tcp = client;
        isConnected = true;
        return(newConn);
}




//hmm...
int Connection::disconnect(){
        closesocket(tcp);
        return(0);
}



int Connection::recieveData(char *buf, int size, int flag){
        int br;
        if(!isConnected)
                return(-1);
        if((br = recv(tcp, buf, size, flag)) == SOCKET_ERROR){
                closesocket(tcp);
                return(-1);
        }
        return(br);
}


int Connection::sendData(char *buf, int size){
        int bs;
        if(!isConnected)
                return(-1);
        if((bs = send(tcp, buf, size, 0)) == SOCKET_ERROR){
                closesocket(tcp);
                return(-1);
        }
        return(bs);
}


int Connection::getLastError(){
        return(WSAGetLastError());
}


int Connection::getPort(){
        return(this->port);
}

---