security/nss/cmd/SSLsample/sslsample.h

   1 /* ***** BEGIN LICENSE BLOCK *****
   2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3  *
   4  * The contents of this file are subject to the Mozilla Public License Version
   5  * 1.1 (the "License"); you may not use this file except in compliance with
   6  * the License. You may obtain a copy of the License at
   7  * http://www.mozilla.org/MPL/
   8  *
   9  * Software distributed under the License is distributed on an "AS IS" basis,
  10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11  * for the specific language governing rights and limitations under the
  12  * License.
  13  *
  14  * The Original Code is the Netscape security libraries.
  15  *
  16  * The Initial Developer of the Original Code is
  17  * Netscape Communications Corporation.
  18  * Portions created by the Initial Developer are Copyright (C) 1994-2000
  19  * the Initial Developer. All Rights Reserved.
  20  *
  21  * Contributor(s):
  22  *
  23  * Alternatively, the contents of this file may be used under the terms of
  24  * either the GNU General Public License Version 2 or later (the "GPL"), or
  25  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  26  * in which case the provisions of the GPL or the LGPL are applicable instead
  27  * of those above. If you wish to allow use of your version of this file only
  28  * under the terms of either the GPL or the LGPL, and not to allow others to
  29  * use your version of this file under the terms of the MPL, indicate your
  30  * decision by deleting the provisions above and replace them with the notice
  31  * and other provisions required by the GPL or the LGPL. If you do not delete
  32  * the provisions above, a recipient may use your version of this file under
  33  * the terms of any one of the MPL, the GPL or the LGPL.
  34  *
  35  * ***** END LICENSE BLOCK ***** */
  36
  37 #ifndef SSLSAMPLE_H
  38 #define SSLSAMPLE_H
  39
  40 /* Generic header files */
  41
  42 #include <stdio.h>
  43 #include <string.h>
  44
  45 /* NSPR header files */
  46
  47 #include "nspr.h"
  48 #include "prerror.h"
  49 #include "prnetdb.h"
  50
  51 /* NSS header files */
  52
  53 #include "pk11func.h"
  54 #include "secitem.h"
  55 #include "ssl.h"
  56 #include "certt.h"
  57 #include "nss.h"
  58 #include "secder.h"
  59 #include "key.h"
  60 #include "sslproto.h"
  61
  62 /* Custom header files */
  63
  64 /*
  65 #include "sslerror.h"
  66 */
  67
  68 #define BUFFER_SIZE 10240
  69
  70 /* Declare SSL cipher suites. */
  71
  72 extern int cipherSuites[];
  73 extern int ssl2CipherSuites[];
  74 extern int ssl3CipherSuites[];
  75
  76 /* Data buffer read from a socket. */
  77 typedef struct DataBufferStr {
  78         char data[BUFFER_SIZE];
  79         int  index;
  80         int  remaining;
  81         int  dataStart;
  82         int  dataEnd;
  83 } DataBuffer;
  84
  85 /* SSL callback routines. */
  86
  87 char * myPasswd(PK11SlotInfo *info, PRBool retry, void *arg);
  88
  89 SECStatus myAuthCertificate(void *arg, PRFileDesc *socket,
  90                             PRBool checksig, PRBool isServer);
  91
  92 SECStatus myBadCertHandler(void *arg, PRFileDesc *socket);
  93
  94 void myHandshakeCallback(PRFileDesc *socket, void *arg);
  95
  96 SECStatus myGetClientAuthData(void *arg, PRFileDesc *socket,
  97                               struct CERTDistNamesStr *caNames,
  98                               struct CERTCertificateStr **pRetCert,
  99                               struct SECKEYPrivateKeyStr **pRetKey);
 100
 101 /* Disable all v2/v3 SSL ciphers. */
 102
 103 void disableAllSSLCiphers(void);
 104
 105
 106 /* Error and information utilities. */
 107
 108 void errWarn(char *function);
 109
 110 void exitErr(char *function);
 111
 112 void printSecurityInfo(PRFileDesc *fd);
 113
 114 /* Some simple thread management routines. */
 115
 116 #define MAX_THREADS 32
 117
 118 typedef SECStatus startFn(void *a, int b);
 119
 120 typedef enum { rs_idle = 0, rs_running = 1, rs_zombie = 2 } runState;
 121
 122 typedef struct perThreadStr {
 123         PRFileDesc *a;
 124         int         b;
 125         int         rv;
 126         startFn    *startFunc;
 127         PRThread   *prThread;
 128         PRBool      inUse;
 129         runState    running;
 130 } perThread;
 131
 132 typedef struct GlobalThreadMgrStr {
 133         PRLock    *threadLock;
 134         PRCondVar *threadStartQ;
 135         PRCondVar *threadEndQ;
 136         perThread  threads[MAX_THREADS];
 137         int        index;
 138         int        numUsed;
 139         int        numRunning;
 140 } GlobalThreadMgr;
 141
 142 void thread_wrapper(void * arg);
 143
 144 SECStatus launch_thread(GlobalThreadMgr *threadMGR,
 145                         startFn *startFunc, void *a, int b);
 146
 147 SECStatus reap_threads(GlobalThreadMgr *threadMGR);
 148
 149 void destroy_thread_data(GlobalThreadMgr *threadMGR);
 150
 151 /* Management of locked variables. */
 152
 153 struct lockedVarsStr {
 154         PRLock *    lock;
 155         int         count;
 156         int         waiters;
 157         PRCondVar * condVar;
 158 };
 159
 160 typedef struct lockedVarsStr lockedVars;
 161
 162 void lockedVars_Init(lockedVars *lv);
 163
 164 void lockedVars_Destroy(lockedVars *lv);
 165
 166 void lockedVars_WaitForDone(lockedVars *lv);
 167
 168 int lockedVars_AddToCount(lockedVars *lv, int addend);
 169
 170 /* Buffer stuff. */
 171
 172 static const char stopCmd[] = { "GET /stop " };
 173 static const char defaultHeader[] = {
 174         "HTTP/1.0 200 OK\r\n"
 175         "Server: SSL sample server\r\n"
 176         "Content-type: text/plain\r\n"
 177         "\r\n"
 178 };
 179
 180 #endif