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 } 211
