net/third_party/nss/ssl/ssldef.c

   1 /*
   2  * "Default" SSLSocket methods, used by sockets that do neither SSL nor socks.
   3  *
   4  * This Source Code Form is subject to the terms of the Mozilla Public
   5  * License, v. 2.0. If a copy of the MPL was not distributed with this
   6  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   7
   8 #include "cert.h"
   9 #include "ssl.h"
  10 #include "sslimpl.h"
  11
  12 #if defined(WIN32)
  13 #define MAP_ERROR(from,to) if (err == from) { PORT_SetError(to); }
  14 #define DEFINE_ERROR       PRErrorCode err = PR_GetError();
  15 #else
  16 #define MAP_ERROR(from,to)
  17 #define DEFINE_ERROR
  18 #endif
  19
  20 int ssl_DefConnect(sslSocket *ss, const PRNetAddr *sa)
  21 {
  22     PRFileDesc *lower = ss->fd->lower;
  23     int rv;
  24
  25     rv = lower->methods->connect(lower, sa, ss->cTimeout);
  26     return rv;
  27 }
  28
  29 int ssl_DefBind(sslSocket *ss, const PRNetAddr *addr)
  30 {
  31     PRFileDesc *lower = ss->fd->lower;
  32     int rv;
  33
  34     rv = lower->methods->bind(lower, addr);
  35     return rv;
  36 }
  37
  38 int ssl_DefListen(sslSocket *ss, int backlog)
  39 {
  40     PRFileDesc *lower = ss->fd->lower;
  41     int rv;
  42
  43     rv = lower->methods->listen(lower, backlog);
  44     return rv;
  45 }
  46
  47 int ssl_DefShutdown(sslSocket *ss, int how)
  48 {
  49     PRFileDesc *lower = ss->fd->lower;
  50     int rv;
  51
  52     rv = lower->methods->shutdown(lower, how);
  53     return rv;
  54 }
  55
  56 int ssl_DefRecv(sslSocket *ss, unsigned char *buf, int len, int flags)
  57 {
  58     PRFileDesc *lower = ss->fd->lower;
  59     int rv;
  60
  61     rv = lower->methods->recv(lower, (void *)buf, len, flags, ss->rTimeout);
  62     if (rv < 0) {
  63         DEFINE_ERROR
  64         MAP_ERROR(PR_SOCKET_SHUTDOWN_ERROR, PR_CONNECT_RESET_ERROR)
  65     } else if (rv > len) {
  66         PORT_Assert(rv <= len);
  67         PORT_SetError(PR_BUFFER_OVERFLOW_ERROR);
  68         rv = SECFailure;
  69     }
  70     return rv;
  71 }
  72
  73 /* Default (unencrypted) send.
  74  * For blocking sockets, always returns len or SECFailure, no short writes.
  75  * For non-blocking sockets:
  76  *   Returns positive count if any data was written, else returns SECFailure.
  77  *   Short writes may occur.  Does not return SECWouldBlock.
  78  */
  79 int ssl_DefSend(sslSocket *ss, const unsigned char *buf, int len, int flags)
  80 {
  81     PRFileDesc *lower = ss->fd->lower;
  82     int sent = 0;
  83
  84 #if NSS_DISABLE_NAGLE_DELAYS
  85     /* Although this is overkill, we disable Nagle delays completely for
  86     ** SSL sockets.
  87     */
  88     if (ss->opt.useSecurity && !ss->delayDisabled) {
  89         ssl_EnableNagleDelay(ss, PR_FALSE);   /* ignore error */
  90         ss->delayDisabled = 1;
  91     }
  92 #endif
  93     do {
  94         int rv = lower->methods->send(lower, (const void *)(buf + sent),
  95                                       len - sent, flags, ss->wTimeout);
  96         if (rv < 0) {
  97             PRErrorCode err = PR_GetError();
  98             if (err == PR_WOULD_BLOCK_ERROR) {
  99                 ss->lastWriteBlocked = 1;
 100                 return sent ? sent : SECFailure;
 101             }
 102             ss->lastWriteBlocked = 0;
 103             MAP_ERROR(PR_CONNECT_ABORTED_ERROR, PR_CONNECT_RESET_ERROR)
 104             /* Loser */
 105             return rv;
 106         }
 107         sent += rv;
 108
 109         if (IS_DTLS(ss) && (len > sent)) {
 110             /* We got a partial write so just return it */
 111             return sent;
 112         }
 113     } while (len > sent);
 114     ss->lastWriteBlocked = 0;
 115     return sent;
 116 }
 117
 118 int ssl_DefRead(sslSocket *ss, unsigned char *buf, int len)
 119 {
 120     PRFileDesc *lower = ss->fd->lower;
 121     int rv;
 122
 123     rv = lower->methods->read(lower, (void *)buf, len);
 124     if (rv < 0) {
 125         DEFINE_ERROR
 126         MAP_ERROR(PR_SOCKET_SHUTDOWN_ERROR, PR_CONNECT_RESET_ERROR)
 127     }
 128     return rv;
 129 }
 130
 131 int ssl_DefWrite(sslSocket *ss, const unsigned char *buf, int len)
 132 {
 133     PRFileDesc *lower = ss->fd->lower;
 134     int sent = 0;
 135
 136     do {
 137         int rv = lower->methods->write(lower, (const void *)(buf + sent),
 138                                        len - sent);
 139         if (rv < 0) {
 140             PRErrorCode err = PR_GetError();
 141             if (err == PR_WOULD_BLOCK_ERROR) {
 142                 ss->lastWriteBlocked = 1;
 143                 return sent ? sent : SECFailure;
 144             }
 145             ss->lastWriteBlocked = 0;
 146             MAP_ERROR(PR_CONNECT_ABORTED_ERROR, PR_CONNECT_RESET_ERROR)
 147             /* Loser */
 148             return rv;
 149         }
 150         sent += rv;
 151     } while (len > sent);
 152     ss->lastWriteBlocked = 0;
 153     return sent;
 154 }
 155
 156 int ssl_DefGetpeername(sslSocket *ss, PRNetAddr *name)
 157 {
 158     PRFileDesc *lower = ss->fd->lower;
 159     int rv;
 160
 161     rv = lower->methods->getpeername(lower, name);
 162     return rv;
 163 }
 164
 165 int ssl_DefGetsockname(sslSocket *ss, PRNetAddr *name)
 166 {
 167     PRFileDesc *lower = ss->fd->lower;
 168     int rv;
 169
 170     rv = lower->methods->getsockname(lower, name);
 171     return rv;
 172 }
 173
 174 int ssl_DefClose(sslSocket *ss)
 175 {
 176     PRFileDesc *fd;
 177     PRFileDesc *popped;
 178     int         rv;
 179
 180     fd    = ss->fd;
 181
 182     /* First, remove the SSL layer PRFileDesc from the socket's stack,
 183     ** then invoke the SSL layer's PRFileDesc destructor.
 184     ** This must happen before the next layer down is closed.
 185     */
 186     PORT_Assert(fd->higher == NULL);
 187     if (fd->higher) {
 188         PORT_SetError(PR_BAD_DESCRIPTOR_ERROR);
 189         return SECFailure;
 190     }
 191     ss->fd = NULL;
 192
 193     /* PR_PopIOLayer will swap the contents of the top two PRFileDescs on
 194     ** the stack, and then remove the second one.  This way, the address
 195     ** of the PRFileDesc on the top of the stack doesn't change.
 196     */
 197     popped = PR_PopIOLayer(fd, PR_TOP_IO_LAYER);
 198     popped->dtor(popped);
 199
 200     /* fd is now the PRFileDesc for the next layer down.
 201     ** Now close the underlying socket.
 202     */
 203     rv = fd->methods->close(fd);
 204
 205     ssl_FreeSocket(ss);
 206
 207     SSL_TRC(5, ("%d: SSL[%d]: closing, rv=%d errno=%d",
 208                 SSL_GETPID(), fd, rv, PORT_GetError()));
 209     return rv;
 210 }