/*

 * "Default" SSLSocket methods, used by sockets that do neither SSL nor socks.

 *

 * This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "cert.h"

#include "ssl.h"

#include "sslimpl.h"

#if defined(WIN32)

#define MAP_ERROR(from, to) \

    if (err == from) {      \

        PORT_SetError(to);  \

    }

#define DEFINE_ERROR PRErrorCode err = PR_GetError();

#else

#define MAP_ERROR(from, to)

#define DEFINE_ERROR

#endif

int

ssl_DefConnect(sslSocket *ss, const PRNetAddr *sa)

{

    PRFileDesc *lower = ss->fd->lower;

    int rv;

    rv = lower->methods->connect(lower, sa, ss->cTimeout);

    return rv;

}

int

ssl_DefBind(sslSocket *ss, const PRNetAddr *addr)

{

    PRFileDesc *lower = ss->fd->lower;

    int rv;

    rv = lower->methods->bind(lower, addr);

    return rv;

}

int

ssl_DefListen(sslSocket *ss, int backlog)

{

    PRFileDesc *lower = ss->fd->lower;

    int rv;

    rv = lower->methods->listen(lower, backlog);

    return rv;

}

int

ssl_DefShutdown(sslSocket *ss, int how)

{

    PRFileDesc *lower = ss->fd->lower;

    int rv;

    rv = lower->methods->shutdown(lower, how);

    return rv;

}

int

ssl_DefRecv(sslSocket *ss, unsigned char *buf, int len, int flags)

{

    PRFileDesc *lower = ss->fd->lower;

    int rv;

    PORT_Assert(buf && len > 0);

    rv = lower->methods->recv(lower, (void *)buf, len, flags, ss->rTimeout);

    if (rv < 0) {

        DEFINE_ERROR

        MAP_ERROR(PR_SOCKET_SHUTDOWN_ERROR, PR_CONNECT_RESET_ERROR)

    } else if (rv > len) {

        PORT_Assert(rv <= len);

        PORT_SetError(PR_BUFFER_OVERFLOW_ERROR);

        rv = SECFailure;

    }

    return rv;

}

/* Default (unencrypted) send.

 * For blocking sockets, always returns len or SECFailure, no short writes.

 * For non-blocking sockets:

 *   Returns positive count if any data was written, else returns SECFailure.

 *   Short writes may occur.

 */

int

ssl_DefSend(sslSocket *ss, const unsigned char *buf, int len, int flags)

{

    PRFileDesc *lower = ss->fd->lower;

    int sent = 0;

#if NSS_DISABLE_NAGLE_DELAYS

    /* Although this is overkill, we disable Nagle delays completely for

    ** SSL sockets.

    */

    if (ss->opt.useSecurity && !ss->delayDisabled) {

        ssl_EnableNagleDelay(ss, PR_FALSE); /* ignore error */

        ss->delayDisabled = 1;

    }

#endif

    do {

        int rv = lower->methods->send(lower, (const void *)(buf + sent),

                                      len - sent, flags, ss->wTimeout);

        if (rv < 0) {

            PRErrorCode err = PR_GetError();

            if (err == PR_WOULD_BLOCK_ERROR) {

                ss->lastWriteBlocked = 1;

                return sent ? sent : SECFailure;

            }

            ss->lastWriteBlocked = 0;

            MAP_ERROR(PR_CONNECT_ABORTED_ERROR, PR_CONNECT_RESET_ERROR)

            /* Loser */

            return rv;

        }

        sent += rv;

        if (IS_DTLS(ss) && (len > sent)) {

            /* We got a partial write so just return it */

            return sent;

        }

    } while (len > sent);

    ss->lastWriteBlocked = 0;

    return sent;

}

int

ssl_DefRead(sslSocket *ss, unsigned char *buf, int len)

{

    PRFileDesc *lower = ss->fd->lower;

    int rv;

    rv = lower->methods->read(lower, (void *)buf, len);

    if (rv < 0) {

        DEFINE_ERROR

        MAP_ERROR(PR_SOCKET_SHUTDOWN_ERROR, PR_CONNECT_RESET_ERROR)

    }

    return rv;

}

int

ssl_DefWrite(sslSocket *ss, const unsigned char *buf, int len)

{

    PRFileDesc *lower = ss->fd->lower;

    int sent = 0;

    do {

        int rv = lower->methods->write(lower, (const void *)(buf + sent),

                                       len - sent);

        if (rv < 0) {

            PRErrorCode err = PR_GetError();

            if (err == PR_WOULD_BLOCK_ERROR) {

                ss->lastWriteBlocked = 1;

                return sent ? sent : SECFailure;

            }

            ss->lastWriteBlocked = 0;

            MAP_ERROR(PR_CONNECT_ABORTED_ERROR, PR_CONNECT_RESET_ERROR)

            /* Loser */

            return rv;

        }

        sent += rv;

    } while (len > sent);

    ss->lastWriteBlocked = 0;

    return sent;

}

int

ssl_DefGetpeername(sslSocket *ss, PRNetAddr *name)

{

    PRFileDesc *lower = ss->fd->lower;

    int rv;

    rv = lower->methods->getpeername(lower, name);

    return rv;

}

int

ssl_DefGetsockname(sslSocket *ss, PRNetAddr *name)

{

    PRFileDesc *lower = ss->fd->lower;

    int rv;

    rv = lower->methods->getsockname(lower, name);

    return rv;

}

int

ssl_DefClose(sslSocket *ss)

{

    PRFileDesc *fd;

    PRFileDesc *popped;

    int rv;

    fd = ss->fd;

    /* First, remove the SSL layer PRFileDesc from the socket's stack,

    ** then invoke the SSL layer's PRFileDesc destructor.

    ** This must happen before the next layer down is closed.

    */

    PORT_Assert(fd->higher == NULL);

    if (fd->higher) {

        PORT_SetError(PR_BAD_DESCRIPTOR_ERROR);

        return SECFailure;

    }

    ss->fd = NULL;

    /* PR_PopIOLayer will swap the contents of the top two PRFileDescs on

    ** the stack, and then remove the second one.  This way, the address

    ** of the PRFileDesc on the top of the stack doesn't change.

    */

    popped = PR_PopIOLayer(fd, PR_TOP_IO_LAYER);

    popped->dtor(popped);

    /* fd is now the PRFileDesc for the next layer down.

    ** Now close the underlying socket.

    */

    rv = fd->methods->close(fd);

    ssl_FreeSocket(ss);

    SSL_TRC(5, ("%d: SSL[%d]: closing, rv=%d errno=%d",

                SSL_GETPID(), fd, rv, PORT_GetError()));

    return rv;

}