crypto/rand/randfile.c

   1 /* crypto/rand/randfile.c */
   2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
   3  * All rights reserved.
   4  *
   5  * This package is an SSL implementation written
   6  * by Eric Young (eay@cryptsoft.com).
   7  * The implementation was written so as to conform with Netscapes SSL.
   8  *
   9  * This library is free for commercial and non-commercial use as long as
  10  * the following conditions are aheared to.  The following conditions
  11  * apply to all code found in this distribution, be it the RC4, RSA,
  12  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
  13  * included with this distribution is covered by the same copyright terms
  14  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
  15  *
  16  * Copyright remains Eric Young's, and as such any Copyright notices in
  17  * the code are not to be removed.
  18  * If this package is used in a product, Eric Young should be given attribution
  19  * as the author of the parts of the library used.
  20  * This can be in the form of a textual message at program startup or
  21  * in documentation (online or textual) provided with the package.
  22  *
  23  * Redistribution and use in source and binary forms, with or without
  24  * modification, are permitted provided that the following conditions
  25  * are met:
  26  * 1. Redistributions of source code must retain the copyright
  27  *    notice, this list of conditions and the following disclaimer.
  28  * 2. Redistributions in binary form must reproduce the above copyright
  29  *    notice, this list of conditions and the following disclaimer in the
  30  *    documentation and/or other materials provided with the distribution.
  31  * 3. All advertising materials mentioning features or use of this software
  32  *    must display the following acknowledgement:
  33  *    "This product includes cryptographic software written by
  34  *     Eric Young (eay@cryptsoft.com)"
  35  *    The word 'cryptographic' can be left out if the rouines from the library
  36  *    being used are not cryptographic related :-).
  37  * 4. If you include any Windows specific code (or a derivative thereof) from
  38  *    the apps directory (application code) you must include an acknowledgement:
  39  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
  40  *
  41  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
  42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  51  * SUCH DAMAGE.
  52  *
  53  * The licence and distribution terms for any publically available version or
  54  * derivative of this code cannot be changed.  i.e. this code cannot simply be
  55  * copied and put under another distribution licence
  56  * [including the GNU Public Licence.]
  57  */
  58
  59 #include <errno.h>
  60 #include <stdio.h>
  61 #include <stdlib.h>
  62 #include <string.h>
  63
  64 #include "openssl/e_os.h"
  65
  66 #ifndef NO_SYS_TYPES_H
  67 # include <sys/types.h>
  68 #endif
  69 #ifdef MAC_OS_pre_X
  70 # include <stat.h>
  71 #else
  72 # include <sys/stat.h>
  73 #endif
  74
  75 #include <openssl/rand.h>
  76
  77 #undef BUFSIZE
  78 #define BUFSIZE 1024
  79 #define RAND_DATA 1024
  80
  81 /* #define RFILE ".rnd" - defined in ../../e_os.h */
  82
  83 int RAND_load_file(const char *file, long bytes)
  84         {
  85         MS_STATIC unsigned char buf[BUFSIZE];
  86         struct stat sb;
  87         int i,ret=0,n;
  88         FILE *in;
  89
  90         if (file == NULL) return(0);
  91
  92         i=stat(file,&sb);
  93         /* If the state fails, put some crap in anyway */
  94         RAND_add(&sb,sizeof(sb),0);
  95         if (i < 0) return(0);
  96         if (bytes <= 0) return(ret);
  97
  98         in=fopen(file,"rb");
  99         if (in == NULL) goto err;
 100         for (;;)
 101                 {
 102                 n=(bytes < BUFSIZE)?(int)bytes:BUFSIZE;
 103                 i=fread(buf,1,n,in);
 104                 if (i <= 0) break;
 105                 /* even if n != i, use the full array */
 106                 RAND_add(buf,n,i);
 107                 ret+=i;
 108                 bytes-=n;
 109                 if (bytes <= 0) break;
 110                 }
 111         fclose(in);
 112         memset(buf,0,BUFSIZE);
 113 err:
 114         return(ret);
 115         }
 116
 117 int RAND_write_file(const char *file)
 118         {
 119         unsigned char buf[BUFSIZE];
 120         int i,ret=0,err=0;
 121         FILE *out = NULL;
 122         int n;
 123
 124         /* Under VMS, fopen(file, "wb") will create a new version of the
 125            same file.  This is not good, so let's try updating an existing
 126            one, and create file only if it doesn't already exist.  This
 127            should be completely harmless on system that have no file
 128            versions.                                    -- Richard Levitte */
 129         out=fopen(file,"rb+");
 130         if (out == NULL
 131 #ifdef ENOENT
 132             && errno == ENOENT
 133 #endif
 134            )
 135                 {
 136                 errno = 0;
 137 #if defined O_CREAT && defined O_EXCL
 138                 /* chmod(..., 0600) is too late to protect the file,
 139                  * permissions should be restrictive from the start */
 140                 {
 141                     int fd = open(file, O_CREAT | O_EXCL, 0600);
 142                     if (fd != -1)
 143                         out = fdopen(fd, "wb");
 144                 }
 145 #else
 146                 out=fopen(file,"wb");
 147 #endif
 148                 }
 149         if (out == NULL) goto err;
 150 #ifndef NO_CHMOD
 151         chmod(file,0600);
 152 #endif
 153         n=RAND_DATA;
 154         for (;;)
 155                 {
 156                 i=(n > BUFSIZE)?BUFSIZE:n;
 157                 n-=BUFSIZE;
 158                 if (RAND_bytes(buf,i) <= 0)
 159                         err=1;
 160                 i=fwrite(buf,1,i,out);
 161                 if (i <= 0)
 162                         {
 163                         ret=0;
 164                         break;
 165                         }
 166                 ret+=i;
 167                 if (n <= 0) break;
 168                 }
 169         if (ret > 0)
 170                 ftruncate(fileno(out), ret);
 171         fclose(out);
 172         memset(buf,0,BUFSIZE);
 173 err:
 174         return(err ? -1 : ret);
 175         }
 176
 177 char *RAND_file_name(char *buf, int size)
 178         {
 179         char *s;
 180         char *ret=NULL;
 181
 182         s=getenv("RANDFILE");
 183         if (s != NULL)
 184                 {
 185                 strncpy(buf,s,size-1);
 186                 buf[size-1]='\0';
 187                 ret=buf;
 188                 }
 189         else
 190                 {
 191                 s=getenv("HOME");
 192                 if (s == NULL) return(RFILE);
 193                 if (((int)(strlen(s)+strlen(RFILE)+2)) > size)
 194                         return(RFILE);
 195                 strcpy(buf,s);
 196 #ifndef VMS
 197                 strcat(buf,"/");
 198 #endif
 199                 strcat(buf,RFILE);
 200                 ret=buf;
 201                 }
 202         return(ret);
 203         }