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         /* If bytes >= 0, read up to 'bytes' bytes.
  86          * if bytes == -1, read complete file. */
  87
  88         MS_STATIC unsigned char buf[BUFSIZE];
  89         struct stat sb;
  90         int i,ret=0,n;
  91         FILE *in;
  92
  93         if (file == NULL) return(0);
  94
  95         i=stat(file,&sb);
  96         /* If the state fails, put some crap in anyway */
  97         RAND_add(&sb,sizeof(sb),0);
  98         if (i < 0) return(0);
  99         if (bytes == 0) return(ret);
 100
 101         in=fopen(file,"rb");
 102         if (in == NULL) goto err;
 103         for (;;)
 104                 {
 105                 if (bytes > 0)
 106                         n = (bytes < BUFSIZE)?(int)bytes:BUFSIZE;
 107                 else
 108                         n = BUFSIZE;
 109                 i=fread(buf,1,n,in);
 110                 if (i <= 0) break;
 111                 /* even if n != i, use the full array */
 112                 RAND_add(buf,n,i);
 113                 ret+=i;
 114                 if (bytes > 0)
 115                         {
 116                         bytes-=n;
 117                         if (bytes == 0) break;
 118                         }
 119                 }
 120         fclose(in);
 121         memset(buf,0,BUFSIZE);
 122 err:
 123         return(ret);
 124         }
 125
 126 int RAND_write_file(const char *file)
 127         {
 128         unsigned char buf[BUFSIZE];
 129         int i,ret=0,err=0;
 130         FILE *out = NULL;
 131         int n;
 132
 133 #ifdef VMS
 134         /* Under VMS, fopen(file, "wb") will create a new version of the
 135            same file.  This is not good, so let's try updating an existing
 136            one, and create file only if it doesn't already exist. */
 137         out=fopen(file,"rb+");
 138         if (out == NULL && errno != ENOENT)
 139                 goto err;
 140 #endif
 141
 142         if (out == NULL)
 143                 {
 144 #if defined O_CREAT && defined O_EXCL
 145                 /* chmod(..., 0600) is too late to protect the file,
 146                  * permissions should be restrictive from the start */
 147                 int fd = open(file, O_CREAT | O_EXCL, 0600);
 148                 if (fd != -1)
 149                         out = fdopen(fd, "wb");
 150 #else
 151                 out=fopen(file,"wb");
 152 #endif
 153                 }
 154         if (out == NULL) goto err;
 155 #ifndef NO_CHMOD
 156         chmod(file,0600);
 157 #endif
 158         n=RAND_DATA;
 159         for (;;)
 160                 {
 161                 i=(n > BUFSIZE)?BUFSIZE:n;
 162                 n-=BUFSIZE;
 163                 if (RAND_bytes(buf,i) <= 0)
 164                         err=1;
 165                 i=fwrite(buf,1,i,out);
 166                 if (i <= 0)
 167                         {
 168                         ret=0;
 169                         break;
 170                         }
 171                 ret+=i;
 172                 if (n <= 0) break;
 173                 }
 174 #ifdef VMS
 175         /* We may have updated an existing file using mode "rb+",
 176          * now remove any old extra bytes */
 177         if (ret > 0)
 178                 ftruncate(fileno(out), ret);
 179 #endif
 180
 181         fclose(out);
 182         memset(buf,0,BUFSIZE);
 183 err:
 184         return(err ? -1 : ret);
 185         }
 186
 187 char *RAND_file_name(char *buf, int size)
 188         {
 189         char *s;
 190         char *ret=NULL;
 191
 192         s=getenv("RANDFILE");
 193         if (s != NULL)
 194                 {
 195                 strncpy(buf,s,size-1);
 196                 buf[size-1]='\0';
 197                 ret=buf;
 198                 }
 199         else
 200                 {
 201                 s=getenv("HOME");
 202                 if (s == NULL) return(RFILE);
 203                 if (((int)(strlen(s)+strlen(RFILE)+2)) > size)
 204                         return(RFILE);
 205                 strcpy(buf,s);
 206 #ifndef VMS
 207                 strcat(buf,"/");
 208 #endif
 209                 strcat(buf,RFILE);
 210                 ret=buf;
 211                 }
 212         return(ret);
 213         }