ssldef.c
上传用户:lyxiangda
上传日期:2007-01-12
资源大小:3042k
文件大小:6k
源码类别:

CA认证

开发平台:

WINDOWS

  1. /*
  2.  * "Default" SSLSocket methods, used by sockets that do neither SSL nor socks.
  3.  *
  4.  * The contents of this file are subject to the Mozilla Public
  5.  * License Version 1.1 (the "License"); you may not use this file
  6.  * except in compliance with the License. You may obtain a copy of
  7.  * the License at http://www.mozilla.org/MPL/
  8.  * 
  9.  * Software distributed under the License is distributed on an "AS
  10.  * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
  11.  * implied. See the License for the specific language governing
  12.  * rights and limitations under the License.
  13.  * 
  14.  * The Original Code is the Netscape security libraries.
  15.  * 
  16.  * The Initial Developer of the Original Code is Netscape
  17.  * Communications Corporation.  Portions created by Netscape are 
  18.  * Copyright (C) 1994-2000 Netscape Communications Corporation.  All
  19.  * Rights Reserved.
  20.  * 
  21.  * Contributor(s):
  22.  * 
  23.  * Alternatively, the contents of this file may be used under the
  24.  * terms of the GNU General Public License Version 2 or later (the
  25.  * "GPL"), in which case the provisions of the GPL are applicable 
  26.  * instead of those above.  If you wish to allow use of your 
  27.  * version of this file only under the terms of the GPL and not to
  28.  * allow others to use your version of this file under the MPL,
  29.  * indicate your decision by deleting the provisions above and
  30.  * replace them with the notice and other provisions required by
  31.  * the GPL.  If you do not delete the provisions above, a recipient
  32.  * may use your version of this file under either the MPL or the
  33.  * GPL.
  34.  *
  35.  * $Id: ssldef.c,v 1.2 2000/10/07 02:22:22 nelsonb%netscape.com Exp $
  36.  */
  37. #include "cert.h"
  38. #include "ssl.h"
  39. #include "sslimpl.h"
  40. #if defined(WIN32)
  41. #define MAP_ERROR(from,to) if (err == from) { PORT_SetError(to); }
  42. #else
  43. #define MAP_ERROR(from,to)
  44. #endif
  45. int ssl_DefConnect(sslSocket *ss, const PRNetAddr *sa)
  46. {
  47.     PRFileDesc *lower = ss->fd->lower;
  48.     int rv;
  49.     rv = lower->methods->connect(lower, sa, ss->cTimeout);
  50.     return rv;
  51. }
  52. int ssl_DefBind(sslSocket *ss, const PRNetAddr *addr)
  53. {
  54.     PRFileDesc *lower = ss->fd->lower;
  55.     int rv;
  56.     rv = lower->methods->bind(lower, addr);
  57.     return rv;
  58. }
  59. int ssl_DefListen(sslSocket *ss, int backlog)
  60. {
  61.     PRFileDesc *lower = ss->fd->lower;
  62.     int rv;
  63.     rv = lower->methods->listen(lower, backlog);
  64.     return rv;
  65. }
  66. int ssl_DefShutdown(sslSocket *ss, int how)
  67. {
  68.     PRFileDesc *lower = ss->fd->lower;
  69.     int rv;
  70.     rv = lower->methods->shutdown(lower, how);
  71.     return rv;
  72. }
  73. int ssl_DefRecv(sslSocket *ss, unsigned char *buf, int len, int flags)
  74. {
  75.     PRFileDesc *lower = ss->fd->lower;
  76.     int rv;
  77.     rv = lower->methods->recv(lower, (void *)buf, len, flags, ss->rTimeout);
  78.     if (rv < 0) {
  79. PRErrorCode err = PR_GetError();
  80. MAP_ERROR(PR_SOCKET_SHUTDOWN_ERROR, PR_CONNECT_RESET_ERROR)
  81.     } else if (rv > len) {
  82. PORT_Assert(rv <= len);
  83. PORT_SetError(PR_BUFFER_OVERFLOW_ERROR);
  84. rv = SECFailure;
  85.     }
  86.     return rv;
  87. }
  88. /* Default (unencrypted) send.
  89.  * Returns SECSuccess or SECFailure,  NOT SECWouldBlock. 
  90.  * Returns positive count if any data was written. 
  91.  * ALWAYS check for a short write after calling ssl_DefSend.
  92.  */
  93. int ssl_DefSend(sslSocket *ss, const unsigned char *buf, int len, int flags)
  94. {
  95.     PRFileDesc *lower = ss->fd->lower;
  96.     int rv, count;
  97.     count = 0;
  98.     for (;;) {
  99. rv = lower->methods->send(lower, (const void *)buf, len,
  100.  flags, ss->wTimeout);
  101. if (rv < 0) {
  102.     PRErrorCode err = PR_GetError();
  103.     if (err == PR_WOULD_BLOCK_ERROR) {
  104. return count ? count : rv;
  105.     }
  106.     MAP_ERROR(PR_CONNECT_ABORTED_ERROR, PR_CONNECT_RESET_ERROR)
  107.     /* Loser */
  108.     return rv;
  109. }
  110. count += rv;
  111. if (rv < len) {
  112.     /* Short send. Send the rest in the next call */
  113.     buf += rv;
  114.     len -= rv;
  115.     continue;
  116. }
  117. break;
  118.     }
  119.     return count;
  120. }
  121. int ssl_DefRead(sslSocket *ss, unsigned char *buf, int len)
  122. {
  123.     PRFileDesc *lower = ss->fd->lower;
  124.     int rv;
  125.     rv = lower->methods->read(lower, (void *)buf, len);
  126.     if (rv < 0) {
  127. PRErrorCode err = PR_GetError();
  128. MAP_ERROR(PR_SOCKET_SHUTDOWN_ERROR, PR_CONNECT_RESET_ERROR)
  129.     }
  130.     return rv;
  131. }
  132. int ssl_DefWrite(sslSocket *ss, const unsigned char *buf, int len)
  133. {
  134.     PRFileDesc *lower = ss->fd->lower;
  135.     int rv, count;
  136.     count = 0;
  137.     for (;;) {
  138. rv = lower->methods->write(lower, (void *)buf, len);
  139. if (rv < 0) {
  140.     PRErrorCode err = PR_GetError();
  141.     if (err == PR_WOULD_BLOCK_ERROR) {
  142. return count ? count : rv;
  143.     }
  144.     MAP_ERROR(PR_CONNECT_ABORTED_ERROR, PR_CONNECT_RESET_ERROR)
  145.     /* Loser */
  146.     return rv;
  147. }
  148. count += rv;
  149. if (rv != len) {
  150.     /* Short write. Send the rest in the next call */
  151.     buf += rv;
  152.     len -= rv;
  153.     continue;
  154. }
  155. break;
  156.     }
  157.     return count;
  158. }
  159. int ssl_DefGetpeername(sslSocket *ss, PRNetAddr *name)
  160. {
  161.     PRFileDesc *lower = ss->fd->lower;
  162.     int rv;
  163.     rv = lower->methods->getpeername(lower, name);
  164.     return rv;
  165. }
  166. int ssl_DefGetsockname(sslSocket *ss, PRNetAddr *name)
  167. {
  168.     PRFileDesc *lower = ss->fd->lower;
  169.     int rv;
  170.     rv = lower->methods->getsockname(lower, name);
  171.     return rv;
  172. }
  173. int ssl_DefClose(sslSocket *ss)
  174. {
  175.     PRFileDesc *fd;
  176.     PRFileDesc *popped;
  177.     int         rv;
  178.     fd    = ss->fd;
  179.     /* First, remove the SSL layer PRFileDesc from the socket's stack, 
  180.     ** then invoke the SSL layer's PRFileDesc destructor.
  181.     ** This must happen before the next layer down is closed.
  182.     */
  183.     PORT_Assert(fd->higher == NULL);
  184.     if (fd->higher) {
  185. PORT_SetError(PR_BAD_DESCRIPTOR_ERROR);
  186. return SECFailure;
  187.     }
  188.     ss->fd = NULL;
  189.     /* PR_PopIOLayer will swap the contents of the top two PRFileDescs on
  190.     ** the stack, and then remove the second one.  This way, the address
  191.     ** of the PRFileDesc on the top of the stack doesn't change.
  192.     */
  193.     popped = PR_PopIOLayer(fd, PR_TOP_IO_LAYER); 
  194.     popped->dtor(popped);
  195.     /* fd is now the PRFileDesc for the next layer down.
  196.     ** Now close the underlying socket. 
  197.     */
  198.     rv = fd->methods->close(fd);
  199.     ssl_FreeSocket(ss);
  200.     SSL_TRC(5, ("%d: SSL[%d]: closing, rv=%d errno=%d",
  201. SSL_GETPID(), fd, rv, PORT_GetError()));
  202.     return rv;
  203. }