---

connection.cpp

Go to the documentation of this file.

#include "sysconfig.h"
#include "connection.h"


//##ModelId=3ED44BF10253
Connection::Connection(SOCKET sock)
        : mSock(sock), mpPendingInput(NULL), mpPendingOutput(NULL),
mDisconnectable(false) {}

//##ModelId=3ED44BF10255
Connection::~Connection() {
    delete[] mpPendingOutput;
    delete[] mpPendingInput;
}

//##ModelId=3ED44BF1025E
Connection::Connection(const Connection& r_conn)
        : mDisconnectable(r_conn.mDisconnectable), mSock(r_conn.mSock),
mpPendingInput(NULL), mpPendingOutput(NULL) {
    mInBuffer = r_conn.mInBuffer;
    mOutBuffer = r_conn.mOutBuffer;
    if (r_conn.mpPendingInput) {
        mpPendingInput = new char[strlen(r_conn.mpPendingInput) + 1];
        strcpy(mpPendingInput, r_conn.mpPendingInput);
    }
    if (r_conn.mpPendingOutput) {
        mpPendingOutput = new char[strlen(r_conn.mpPendingOutput) + 1];
        strcpy(mpPendingOutput, r_conn.mpPendingOutput);
    }
}

//##ModelId=3ED44BF10288
Connection & Connection::operator=(const Connection& r_conn) {
    if (this == &r_conn)
        return * this;
    mInBuffer = r_conn.mInBuffer;
    mOutBuffer = r_conn.mOutBuffer;
    if (r_conn.mpPendingInput) {
        mpPendingInput = new char[strlen(r_conn.mpPendingInput) + 1];
        strcpy(mpPendingInput, r_conn.mpPendingInput);
    }
    if (r_conn.mpPendingOutput) {
        mpPendingOutput = new char[strlen(r_conn.mpPendingOutput) + 1];
        strcpy(mpPendingOutput, r_conn.mpPendingOutput);
    }
    return *this;
}

//##ModelId=3ED44BF10260
void Connection::HandleInput() {
    string strin;
    char inbuf[4096];
    // If there was input leftover from the last call to HandleInput
    // prepend it to the input string and clear it.
    if (mpPendingInput) {
        strin.append(mpPendingInput);
        delete[] mpPendingInput;
        mpPendingInput = NULL;
    }
    // Loop to gather all socket input in 4K chunks.
    while (true) {
        int nbytes = recv (mSock, inbuf, 4096, 0);
        if (nbytes == 0) {
            // Connection is scheduled for disconnection, and we'll exit
            mDisconnectable = true;
            return ;
        } else if (nbytes == SOCKET_ERROR) {
            int err = WSAGetLastError ();
            if (err == WSAEWOULDBLOCK) {
                // Indicates no more input can be squeezed from this socket.
                // Let's bust this loop
                break;
            } else {
                // We have a serious error
                // Connection is scheduled for disconnection, and we'll exit
                cout << "Error-Connection(recv):" << err << " on " << mSock << endl;
                mDisconnectable = true;
                return ;
            }
        }
        // Append everything read to our string.
        inbuf[nbytes] = '\0';
        strin.append(inbuf);
    }
    // Loop to parse all input into lines for our buffer
    while (true) {
        int pos = strin.find("\r\n");
        if (pos != (int)strin.npos) {
            // A line was found; save it to buffer and remove from input string
            mInBuffer.push_back(strin.substr(0, pos));
            strin.erase(0, pos + 2);
        } else {
            // No       more complete lines found
            if (!strin.empty()) {
                // Save leftovers into pending input array
                mpPendingInput = new char[strin.size() + 1];
                strcpy(mpPendingInput, strin.c_str());
            }
            break;
        }
    }
}

//##ModelId=3ED44BF10268
void Connection::HandleOutput() {
    string strout;
    // If there is output left to be sent from last time append
    // it to our output string
    if (mpPendingOutput) {
        strout.append(mpPendingOutput);
        delete[] mpPendingOutput;
        mpPendingOutput = NULL;
    }
    // If there is any output waiting in line buffer append the first line
    if (!mOutBuffer.empty()) {
        strout.append(mOutBuffer.front());
        mOutBuffer.pop_front();
    }
    // Loop until our output string is empty
    while (!strout.empty()) {
        const char * pstr = strout.c_str();
        while (pstr) {  // Loop until we have written this string
            int nbytes = send (mSock, pstr, strlen(pstr), 0);
            if (nbytes == SOCKET_ERROR) {
                int err = WSAGetLastError ();
                if (err == WSAEWOULDBLOCK) {
                    // socket will block - save unwritten output in pending array
                    // and leave
                    mpPendingOutput = new char[strlen(pstr) + 1];
                    strcpy(mpPendingOutput, pstr);
                    return ;
                }
                // severe error set socket as disconnectable and leave
                cout << "Error-Connection(recv):" << err << " on " << mSock << endl;
                mDisconnectable = true;
                return ;
            } else if (nbytes < (int)strlen(pstr)) {
                pstr += nbytes;
            } else {
                pstr = NULL;
            }
        }
        strout.erase();
        // fetch next line in buffer if not empty
        if (!mOutBuffer.empty()) {
            strout.append(mOutBuffer.front());
            mOutBuffer.pop_front();
        }
    }
}

//##ModelId=3ED44BF10272
void Connection::Disconnect() {
    char inbuf[4096];
    bool done = false;
    shutdown(mSock, SD_SEND);
    while (!done) {
        int nbytes = recv (mSock, inbuf, 4096, 0);
        if (!nbytes || nbytes == SOCKET_ERROR)
            done = true;
    }
    closesocket(mSock);
}

//##ModelId=3ED44BF10274
void Connection::SendMsg(const string& r_msg) {
    mOutBuffer.push_back(r_msg);
}

//##ModelId=3ED44BF1027D
string* Connection::ReadMsg() {
    if (!mInBuffer.empty()) {
        string msg = mInBuffer.front();
        mInBuffer.pop_front();
        return new string(msg);
    }
    return NULL;
}

//##ModelId=3ED44BF1026A
bool Connection::CanBeDisconnected() {
    return mDisconnectable;
}

//##ModelId=3ED44BF10286
SOCKET Connection::GetSocket() {
    return mSock;
}

---