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

CA认证

开发平台:

WINDOWS

  1. /* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
  2. /* 
  3.  * The contents of this file are subject to the Mozilla Public
  4.  * License Version 1.1 (the "License"); you may not use this file
  5.  * except in compliance with the License. You may obtain a copy of
  6.  * the License at http://www.mozilla.org/MPL/
  7.  * 
  8.  * Software distributed under the License is distributed on an "AS
  9.  * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
  10.  * implied. See the License for the specific language governing
  11.  * rights and limitations under the License.
  12.  * 
  13.  * The Original Code is the Netscape security libraries.
  14.  * 
  15.  * The Initial Developer of the Original Code is Netscape
  16.  * Communications Corporation.  Portions created by Netscape are 
  17.  * Copyright (C) 1994-2000 Netscape Communications Corporation.  All
  18.  * Rights Reserved.
  19.  * 
  20.  * Contributor(s):
  21.  * 
  22.  * Alternatively, the contents of this file may be used under the
  23.  * terms of the GNU General Public License Version 2 or later (the
  24.  * "GPL"), in which case the provisions of the GPL are applicable 
  25.  * instead of those above.  If you wish to allow use of your 
  26.  * version of this file only under the terms of the GPL and not to
  27.  * allow others to use your version of this file under the MPL,
  28.  * indicate your decision by deleting the provisions above and
  29.  * replace them with the notice and other provisions required by
  30.  * the GPL.  If you do not delete the provisions above, a recipient
  31.  * may use your version of this file under either the MPL or the
  32.  * GPL.
  33.  */
  34. #if defined(XP_UNIX) || defined(XP_BEOS) || defined(XP_OS2)
  35. #include <sys/types.h>
  36. #include <sys/socket.h>
  37. #include <netinet/in.h>
  38. #else
  39. #ifdef XP_MAC
  40. #else /* windows */
  41. #include <windows.h>
  42. #include <winsock.h>
  43. #endif
  44. #endif
  45. #include <errno.h>
  46. #include "cmtcmn.h"
  47. #include "cmtutils.h"
  48. #include "messages.h"
  49. #include "rsrcids.h"
  50. CMTStatus CMT_OpenSSLConnection(PCMT_CONTROL control, CMTSocket sock,
  51.                                 SSMSSLConnectionRequestType flags, 
  52.                                 CMUint32 port, char * hostIP, 
  53.                                 char * hostName, CMBool forceHandshake, void* clientContext)
  54. {
  55. CMTItem message;
  56.     SSLDataConnectionRequest request;
  57.     DataConnectionReply reply;
  58.     CMUint32 sent;
  59. /* Do some parameter checking */
  60. if (!control || !hostIP || !hostName) {
  61. goto loser;
  62. }
  63.     request.flags = flags;
  64.     request.port = port;
  65.     request.hostIP = hostIP;
  66.     request.hostName = hostName;
  67.     request.forceHandshake = forceHandshake;
  68.     request.clientContext = CMT_CopyPtrToItem(clientContext);
  69.     /* Encode message */
  70.     if (CMT_EncodeMessage(SSLDataConnectionRequestTemplate, &message, &request) != CMTSuccess) {
  71.         goto loser;
  72.     }
  73.     /* Set the message request type */
  74.     message.type = SSM_REQUEST_MESSAGE | SSM_DATA_CONNECTION | SSM_SSL_CONNECTION;
  75.     
  76. /* Send the message and get the response */
  77.     if (CMT_SendMessage(control, &message) == CMTFailure) {
  78.         goto loser;
  79.     }
  80.     /* Validate the message reply type */
  81.     if (message.type != (SSM_REPLY_OK_MESSAGE | SSM_DATA_CONNECTION | SSM_SSL_CONNECTION)) {
  82.         goto loser;
  83.     }
  84.     /* Decode message */
  85.     if (CMT_DecodeMessage(DataConnectionReplyTemplate, &reply, &message) != CMTSuccess) {
  86.         goto loser;
  87.     }
  88. /* Success */
  89. if (reply.result == 0) {
  90.   if (control->sockFuncs.connect(sock, reply.port, NULL) != CMTSuccess) {
  91.             goto loser;
  92.   }
  93.   sent = CMT_WriteThisMany(control, sock, control->nonce.data, 
  94.                                control->nonce.len);
  95.   if (sent != control->nonce.len) {
  96.             goto loser;
  97.   }
  98.   
  99.   /* Save connection info */
  100.   if (CMT_AddDataConnection(control, sock, reply.connID)
  101.       != CMTSuccess) {
  102.     goto loser;
  103.   }
  104.   return CMTSuccess;
  105. loser:
  106. return CMTFailure;
  107. }
  108. CMTStatus CMT_GetSSLDataErrorCode(PCMT_CONTROL control, CMTSocket sock,
  109.                                   CMInt32* errorCode)
  110. {
  111.     CMUint32 connID;
  112.     if (!control || !errorCode) {
  113.         goto loser;
  114.     }
  115.     /* get the data connection */
  116.     if (CMT_GetDataConnectionID(control, sock, &connID) != CMTSuccess) {
  117.         goto loser;
  118.     }
  119.     /* get the PR error */
  120.     if (CMT_GetNumericAttribute(control, connID, SSM_FID_SSLDATA_ERROR_VALUE,
  121.                                 errorCode) != CMTSuccess) {
  122.         goto loser;
  123.     }
  124.     return CMTSuccess;
  125. loser:
  126.     return CMTFailure;
  127. }
  128. CMTStatus CMT_ReleaseSSLSocketStatus(PCMT_CONTROL control, CMTSocket sock)
  129. {
  130.     CMUint32 connectionID;
  131.     if (!control || !sock) {
  132.         goto loser;
  133.     }
  134.     if (CMT_GetDataConnectionID(control, sock, &connectionID) != CMTSuccess) {
  135.         goto loser;
  136.     }
  137.     if (CMT_SetNumericAttribute(control, connectionID, 
  138.                                 SSM_FID_SSLDATA_DISCARD_SOCKET_STATUS,
  139.                                 0) != CMTSuccess) {
  140.         goto loser;
  141.     }
  142.     return CMTSuccess;
  143.  loser:
  144.     return CMTFailure;
  145. }
  146. CMTStatus CMT_GetSSLSocketStatus(PCMT_CONTROL control, CMTSocket sock, 
  147.                                  CMTItem* pickledStatus, CMInt32* level)
  148. {
  149.     CMUint32 connectionID;
  150.     SingleNumMessage request;
  151.     CMTItem message;
  152.     PickleSecurityStatusReply reply;
  153.     if (!control || !pickledStatus || !level) {
  154.         goto loser;
  155.     }
  156. /* get the data connection */
  157.     if (CMT_GetDataConnectionID(control, sock, &connectionID) != CMTSuccess) {
  158. goto loser;
  159. }
  160.     /* set up the request */
  161.     request.value = connectionID;
  162.     /* encode the request */
  163.     if (CMT_EncodeMessage(SingleNumMessageTemplate, &message, &request) !=
  164.         CMTSuccess) {
  165.         goto loser;
  166.     }
  167.     /* set the message request type */
  168.     message.type = SSM_REQUEST_MESSAGE | SSM_RESOURCE_ACTION | 
  169.         SSM_CONSERVE_RESOURCE | SSM_PICKLE_SECURITY_STATUS;
  170.     /* send the message and get the response */
  171.     if (CMT_SendMessage(control, &message) == CMTFailure) {
  172.         goto loser;
  173.     }
  174.     /* validate the message reply type */
  175.     if (message.type != (SSM_REPLY_OK_MESSAGE | SSM_RESOURCE_ACTION |
  176.                          SSM_CONSERVE_RESOURCE | SSM_PICKLE_SECURITY_STATUS)) {
  177.         goto loser;
  178.     }
  179.     /* decode the reply */
  180.     if (CMT_DecodeMessage(PickleSecurityStatusReplyTemplate, &reply, &message)
  181.         != CMTSuccess) {
  182.         goto loser;
  183.     }
  184.     /* success */
  185.     if (reply.result == 0) {
  186.         *pickledStatus = reply.blob;
  187.         *level = reply.securityLevel;
  188.         return CMTSuccess;
  189.     }
  190. loser:
  191.     return CMTFailure;
  192. }
  193. CMTStatus CMT_OpenTLSConnection(PCMT_CONTROL control, CMTSocket sock,
  194.                                 CMUint32 port, char* hostIP, char* hostName)
  195. {
  196.     TLSDataConnectionRequest request;
  197.     CMTItem message;
  198.     DataConnectionReply reply;
  199.     CMUint32 sent;
  200.     /* do some parameter checking */
  201.     if (!control || !hostIP || !hostName) {
  202.         goto loser;
  203.     }
  204.     request.port = port;
  205.     request.hostIP = hostIP;
  206.     request.hostName = hostName;
  207.     /* encode the message */
  208.     if (CMT_EncodeMessage(TLSDataConnectionRequestTemplate, &message, &request)
  209.         != CMTSuccess) {
  210.         goto loser;
  211.     }
  212.     /* set the message request type */
  213.     message.type = SSM_REQUEST_MESSAGE | SSM_DATA_CONNECTION | 
  214.         SSM_TLS_CONNECTION;
  215.     /* send the message and get the response */
  216.     if (CMT_SendMessage(control, &message) == CMTFailure) {
  217.         goto loser;
  218.     }
  219.     /* validate the message reply type */
  220.     if (message.type != (SSM_REPLY_OK_MESSAGE | SSM_DATA_CONNECTION | 
  221.                          SSM_TLS_CONNECTION)) {
  222.         goto loser;
  223.     }
  224.     /* decode the message */
  225.     if (CMT_DecodeMessage(DataConnectionReplyTemplate, &reply, &message) !=
  226.         CMTSuccess) {
  227.         goto loser;
  228.     }
  229.     /* success */
  230.     if (reply.result == 0) {
  231.         if (control->sockFuncs.connect(sock, reply.port, NULL) != CMTSuccess) {
  232.             goto loser;
  233.         }
  234.         sent = CMT_WriteThisMany(control, sock, control->nonce.data, 
  235.                                  control->nonce.len);
  236.         if (sent != control->nonce.len) {
  237.             goto loser;
  238.         }
  239.         /* save connection info */
  240.         if (CMT_AddDataConnection(control, sock, reply.connID) != CMTSuccess) {
  241.             goto loser;
  242.         }
  243.         return CMTSuccess;
  244.     }
  245. loser:
  246.     return CMTFailure;
  247. }
  248. CMTStatus CMT_TLSStepUp(PCMT_CONTROL control, CMTSocket sock, 
  249.                         void* clientContext)
  250. {
  251.     TLSStepUpRequest request;
  252.     SingleNumMessage reply;
  253.     CMTItem message;
  254.     CMUint32 connectionID;
  255.     /* check arguments */
  256.     if (!control || !sock) {
  257.         goto loser;
  258.     }
  259.     /* get the data connection ID */
  260.     if (CMT_GetDataConnectionID(control, sock, &connectionID) != CMTSuccess) {
  261.         goto loser;
  262.     }
  263.     /* set up the request */
  264.     request.connID = connectionID;
  265.     request.clientContext = CMT_CopyPtrToItem(clientContext);
  266.     /* encode the request */
  267.     if (CMT_EncodeMessage(TLSStepUpRequestTemplate, &message, &request) !=
  268.         CMTSuccess) {
  269.         goto loser;
  270.     }
  271.     /* set the message request type */
  272.     message.type = SSM_REQUEST_MESSAGE | SSM_RESOURCE_ACTION | SSM_TLS_STEPUP;
  273.     /* send the message and get the response */
  274.     if (CMT_SendMessage(control, &message) == CMTFailure) {
  275.         goto loser;
  276.     }
  277.     /* validate the message reply type */
  278.     if (message.type != (SSM_REPLY_OK_MESSAGE | SSM_RESOURCE_ACTION |
  279.                          SSM_TLS_STEPUP)) {
  280.         goto loser;
  281.     }
  282.     /* decode the reply */
  283.     if (CMT_DecodeMessage(SingleNumMessageTemplate, &reply, &message) !=
  284.         CMTSuccess) {
  285.         goto loser;
  286.     }
  287.     return (CMTStatus) reply.value;
  288. loser:
  289.     return CMTFailure;
  290. }
  291. CMTStatus CMT_OpenSSLProxyConnection(PCMT_CONTROL control, CMTSocket sock,
  292.                                      CMUint32 port, char* hostIP, 
  293.                                      char* hostName)
  294. {
  295.     TLSDataConnectionRequest request;
  296.     CMTItem message;
  297.     DataConnectionReply reply;
  298.     CMUint32 sent;
  299.     /* do some parameter checking */
  300.     if (!control || !hostIP || !hostName) {
  301.         goto loser;
  302.     }
  303.     request.port = port;
  304.     request.hostIP = hostIP;
  305.     request.hostName = hostName;
  306.     /* encode the message */
  307.     if (CMT_EncodeMessage(TLSDataConnectionRequestTemplate, &message, &request)
  308.         != CMTSuccess) {
  309.         goto loser;
  310.     }
  311.     /* set the message request type */
  312.     message.type = SSM_REQUEST_MESSAGE | SSM_DATA_CONNECTION | 
  313.         SSM_PROXY_CONNECTION;
  314.     /* send the message and get the response */
  315.     if (CMT_SendMessage(control, &message) == CMTFailure) {
  316.         goto loser;
  317.     }
  318.     /* validate the message reply type */
  319.     if (message.type != (SSM_REPLY_OK_MESSAGE | SSM_DATA_CONNECTION | 
  320.                          SSM_PROXY_CONNECTION)) {
  321.         goto loser;
  322.     }
  323.     /* decode the message */
  324.     if (CMT_DecodeMessage(DataConnectionReplyTemplate, &reply, &message) !=
  325.         CMTSuccess) {
  326.         goto loser;
  327.     }
  328.     /* success */
  329.     if (reply.result == 0) {
  330.         if (control->sockFuncs.connect(sock, reply.port, NULL) != CMTSuccess) {
  331.             goto loser;
  332.         }
  333.         sent = CMT_WriteThisMany(control, sock, control->nonce.data, 
  334.                                  control->nonce.len);
  335.         if (sent != control->nonce.len) {
  336.             goto loser;
  337.         }
  338.         /* save connection info */
  339.         if (CMT_AddDataConnection(control, sock, reply.connID) != CMTSuccess) {
  340.             goto loser;
  341.         }
  342.         return CMTSuccess;
  343.     }
  344. loser:
  345.     return CMTFailure;
  346. }
  347. CMTStatus CMT_ProxyStepUp(PCMT_CONTROL control, CMTSocket sock, 
  348.                         void* clientContext, char* remoteUrl)
  349. {
  350.     ProxyStepUpRequest request;
  351.     SingleNumMessage reply;
  352.     CMTItem message;
  353.     CMUint32 connectionID;
  354.     /* check arguments */
  355.     if (!control || !sock || !remoteUrl) {
  356.         goto loser;
  357.     }
  358.     /* get the data connection ID */
  359.     if (CMT_GetDataConnectionID(control, sock, &connectionID) != CMTSuccess) {
  360.         goto loser;
  361.     }
  362.     /* set up the request */
  363.     request.connID = connectionID;
  364.     request.clientContext = CMT_CopyPtrToItem(clientContext);
  365.     request.url = remoteUrl;
  366.     /* encode the request */
  367.     if (CMT_EncodeMessage(ProxyStepUpRequestTemplate, &message, &request) !=
  368.         CMTSuccess) {
  369.         goto loser;
  370.     }
  371.     /* set the message request type */
  372.     message.type = SSM_REQUEST_MESSAGE | SSM_RESOURCE_ACTION | 
  373.         SSM_PROXY_STEPUP;
  374.     /* send the message and get the response */
  375.     if (CMT_SendMessage(control, &message) == CMTFailure) {
  376.         goto loser;
  377.     }
  378.     /* validate the message reply type */
  379.     if (message.type != (SSM_REPLY_OK_MESSAGE | SSM_RESOURCE_ACTION |
  380.                          SSM_PROXY_STEPUP)) {
  381.         goto loser;
  382.     }
  383.     /* decode the reply */
  384.     if (CMT_DecodeMessage(SingleNumMessageTemplate, &reply, &message) !=
  385.         CMTSuccess) {
  386.         goto loser;
  387.     }
  388.     return (CMTStatus) reply.value;
  389. loser:
  390.     return CMTFailure;
  391. }