michael@0: /*
michael@0:  * "Default" SSLSocket methods, used by sockets that do neither SSL nor socks.
michael@0:  *
michael@0:  * This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "cert.h"
michael@0: #include "ssl.h"
michael@0: #include "sslimpl.h"
michael@0: 
michael@0: #if defined(WIN32)
michael@0: #define MAP_ERROR(from,to) if (err == from) { PORT_SetError(to); }
michael@0: #define DEFINE_ERROR       PRErrorCode err = PR_GetError();
michael@0: #else
michael@0: #define MAP_ERROR(from,to)
michael@0: #define DEFINE_ERROR
michael@0: #endif
michael@0: 
michael@0: int ssl_DefConnect(sslSocket *ss, const PRNetAddr *sa)
michael@0: {
michael@0:     PRFileDesc *lower = ss->fd->lower;
michael@0:     int rv;
michael@0: 
michael@0:     rv = lower->methods->connect(lower, sa, ss->cTimeout);
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: int ssl_DefBind(sslSocket *ss, const PRNetAddr *addr)
michael@0: {
michael@0:     PRFileDesc *lower = ss->fd->lower;
michael@0:     int rv;
michael@0: 
michael@0:     rv = lower->methods->bind(lower, addr);
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: int ssl_DefListen(sslSocket *ss, int backlog)
michael@0: {
michael@0:     PRFileDesc *lower = ss->fd->lower;
michael@0:     int rv;
michael@0: 
michael@0:     rv = lower->methods->listen(lower, backlog);
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: int ssl_DefShutdown(sslSocket *ss, int how)
michael@0: {
michael@0:     PRFileDesc *lower = ss->fd->lower;
michael@0:     int rv;
michael@0: 
michael@0:     rv = lower->methods->shutdown(lower, how);
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: int ssl_DefRecv(sslSocket *ss, unsigned char *buf, int len, int flags)
michael@0: {
michael@0:     PRFileDesc *lower = ss->fd->lower;
michael@0:     int rv;
michael@0: 
michael@0:     rv = lower->methods->recv(lower, (void *)buf, len, flags, ss->rTimeout);
michael@0:     if (rv < 0) {
michael@0: 	DEFINE_ERROR
michael@0: 	MAP_ERROR(PR_SOCKET_SHUTDOWN_ERROR, PR_CONNECT_RESET_ERROR)
michael@0:     } else if (rv > len) {
michael@0: 	PORT_Assert(rv <= len);
michael@0: 	PORT_SetError(PR_BUFFER_OVERFLOW_ERROR);
michael@0: 	rv = SECFailure;
michael@0:     }
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: /* Default (unencrypted) send.
michael@0:  * For blocking sockets, always returns len or SECFailure, no short writes.
michael@0:  * For non-blocking sockets:
michael@0:  *   Returns positive count if any data was written, else returns SECFailure. 
michael@0:  *   Short writes may occur.  Does not return SECWouldBlock.
michael@0:  */
michael@0: int ssl_DefSend(sslSocket *ss, const unsigned char *buf, int len, int flags)
michael@0: {
michael@0:     PRFileDesc *lower = ss->fd->lower;
michael@0:     int sent = 0;
michael@0: 
michael@0: #if NSS_DISABLE_NAGLE_DELAYS
michael@0:     /* Although this is overkill, we disable Nagle delays completely for 
michael@0:     ** SSL sockets.
michael@0:     */
michael@0:     if (ss->opt.useSecurity && !ss->delayDisabled) {
michael@0: 	ssl_EnableNagleDelay(ss, PR_FALSE);   /* ignore error */
michael@0:     	ss->delayDisabled = 1;
michael@0:     }
michael@0: #endif
michael@0:     do {
michael@0: 	int rv = lower->methods->send(lower, (const void *)(buf + sent), 
michael@0: 	                              len - sent, flags, ss->wTimeout);
michael@0: 	if (rv < 0) {
michael@0: 	    PRErrorCode err = PR_GetError();
michael@0: 	    if (err == PR_WOULD_BLOCK_ERROR) {
michael@0: 		ss->lastWriteBlocked = 1;
michael@0: 		return sent ? sent : SECFailure;
michael@0: 	    }
michael@0: 	    ss->lastWriteBlocked = 0;
michael@0: 	    MAP_ERROR(PR_CONNECT_ABORTED_ERROR, PR_CONNECT_RESET_ERROR)
michael@0: 	    /* Loser */
michael@0: 	    return rv;
michael@0: 	}
michael@0: 	sent += rv;
michael@0: 	
michael@0: 	if (IS_DTLS(ss) && (len > sent)) { 
michael@0: 	    /* We got a partial write so just return it */
michael@0: 	    return sent;
michael@0: 	}
michael@0:     } while (len > sent);
michael@0:     ss->lastWriteBlocked = 0;
michael@0:     return sent;
michael@0: }
michael@0: 
michael@0: int ssl_DefRead(sslSocket *ss, unsigned char *buf, int len)
michael@0: {
michael@0:     PRFileDesc *lower = ss->fd->lower;
michael@0:     int rv;
michael@0: 
michael@0:     rv = lower->methods->read(lower, (void *)buf, len);
michael@0:     if (rv < 0) {
michael@0: 	DEFINE_ERROR
michael@0: 	MAP_ERROR(PR_SOCKET_SHUTDOWN_ERROR, PR_CONNECT_RESET_ERROR)
michael@0:     }
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: int ssl_DefWrite(sslSocket *ss, const unsigned char *buf, int len)
michael@0: {
michael@0:     PRFileDesc *lower = ss->fd->lower;
michael@0:     int sent = 0;
michael@0: 
michael@0:     do {
michael@0: 	int rv = lower->methods->write(lower, (const void *)(buf + sent), 
michael@0: 	                               len - sent);
michael@0: 	if (rv < 0) {
michael@0: 	    PRErrorCode err = PR_GetError();
michael@0: 	    if (err == PR_WOULD_BLOCK_ERROR) {
michael@0: 		ss->lastWriteBlocked = 1;
michael@0: 		return sent ? sent : SECFailure;
michael@0: 	    }
michael@0: 	    ss->lastWriteBlocked = 0;
michael@0: 	    MAP_ERROR(PR_CONNECT_ABORTED_ERROR, PR_CONNECT_RESET_ERROR)
michael@0: 	    /* Loser */
michael@0: 	    return rv;
michael@0: 	}
michael@0: 	sent += rv;
michael@0:     } while (len > sent);
michael@0:     ss->lastWriteBlocked = 0;
michael@0:     return sent;
michael@0: }
michael@0: 
michael@0: int ssl_DefGetpeername(sslSocket *ss, PRNetAddr *name)
michael@0: {
michael@0:     PRFileDesc *lower = ss->fd->lower;
michael@0:     int rv;
michael@0: 
michael@0:     rv = lower->methods->getpeername(lower, name);
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: int ssl_DefGetsockname(sslSocket *ss, PRNetAddr *name)
michael@0: {
michael@0:     PRFileDesc *lower = ss->fd->lower;
michael@0:     int rv;
michael@0: 
michael@0:     rv = lower->methods->getsockname(lower, name);
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: int ssl_DefClose(sslSocket *ss)
michael@0: {
michael@0:     PRFileDesc *fd;
michael@0:     PRFileDesc *popped;
michael@0:     int         rv;
michael@0: 
michael@0:     fd    = ss->fd;
michael@0: 
michael@0:     /* First, remove the SSL layer PRFileDesc from the socket's stack, 
michael@0:     ** then invoke the SSL layer's PRFileDesc destructor.
michael@0:     ** This must happen before the next layer down is closed.
michael@0:     */
michael@0:     PORT_Assert(fd->higher == NULL);
michael@0:     if (fd->higher) {
michael@0: 	PORT_SetError(PR_BAD_DESCRIPTOR_ERROR);
michael@0: 	return SECFailure;
michael@0:     }
michael@0:     ss->fd = NULL;
michael@0: 
michael@0:     /* PR_PopIOLayer will swap the contents of the top two PRFileDescs on
michael@0:     ** the stack, and then remove the second one.  This way, the address
michael@0:     ** of the PRFileDesc on the top of the stack doesn't change.
michael@0:     */
michael@0:     popped = PR_PopIOLayer(fd, PR_TOP_IO_LAYER); 
michael@0:     popped->dtor(popped);
michael@0: 
michael@0:     /* fd is now the PRFileDesc for the next layer down.
michael@0:     ** Now close the underlying socket. 
michael@0:     */
michael@0:     rv = fd->methods->close(fd);
michael@0: 
michael@0:     ssl_FreeSocket(ss);
michael@0: 
michael@0:     SSL_TRC(5, ("%d: SSL[%d]: closing, rv=%d errno=%d",
michael@0: 		SSL_GETPID(), fd, rv, PORT_GetError()));
michael@0:     return rv;
michael@0: }