tfc_random.c

   1 /*
   2  * tfcrypt -- high security Threefish encryption tool.
   3  *
   4  * tfcrypt is copyrighted:
   5  * Copyright (C) 2012-2018 Andrey Rys. All rights reserved.
   6  *
   7  * tfcrypt is licensed to you under the terms of std. MIT/X11 license:
   8  *
   9  * Permission is hereby granted, free of charge, to any person obtaining
  10  * a copy of this software and associated documentation files (the
  11  * "Software"), to deal in the Software without restriction, including
  12  * without limitation the rights to use, copy, modify, merge, publish,
  13  * distribute, sublicense, and/or sell copies of the Software, and to
  14  * permit persons to whom the Software is furnished to do so, subject to
  15  * the following conditions:
  16  *
  17  * The above copyright notice and this permission notice shall be
  18  * included in all copies or substantial portions of the Software.
  19  *
  20  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
  23  * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
  24  * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  25  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  26  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27  */
  28
  29 #include "tfcrypt.h"
  30
  31 static void get_urandom(const char *src, void *buf, size_t size)
  32 {
  33         tfc_byte *ubuf = buf;
  34         int fd = -1;
  35         size_t sz = size, rd;
  36
  37         if (src == NULL) fd = -1;
  38         else fd = open(src, O_RDONLY);
  39
  40         if (fd == -1) fd = open("/dev/urandom", O_RDONLY);
  41         if (fd == -1) fd = open("/dev/arandom", O_RDONLY);
  42         if (fd == -1) fd = open("/dev/prandom", O_RDONLY);
  43         if (fd == -1) fd = open("/dev/srandom", O_RDONLY);
  44         if (fd == -1) fd = open("/dev/random", O_RDONLY);
  45         if (fd == -1) xerror(NO, YES, YES, "random source is required (tried %s)", src);
  46
  47 _again: rd = xread(fd, ubuf, sz);
  48         if (rd < sz && rd != NOSIZE) {
  49                 ubuf += rd;
  50                 sz -= rd;
  51                 goto _again;
  52         }
  53
  54         xclose(fd);
  55 }
  56
  57 static tfc_yesno tfc_random_initialised;
  58
  59 static void tfc_initrandom(const char *rndsrc)
  60 {
  61         tfc_byte k[TF_KEY_SIZE];
  62
  63         if (tfc_random_initialised == YES) return;
  64
  65         get_urandom(rndsrc, k, TF_KEY_SIZE);
  66         tf_prng_seedkey(k);
  67         memset(k, 0, TF_KEY_SIZE);
  68
  69         tfc_random_initialised = YES;
  70 }
  71
  72 void tfc_finirandom(void)
  73 {
  74         tf_prng_seedkey(NULL);
  75         tfc_random_initialised = NO;
  76 }
  77
  78 void tfc_getrandom(void *buf, size_t sz)
  79 {
  80         if (tfc_random_initialised == NO) tfc_initrandom(randsource);
  81         tf_prng_genrandom(buf, sz);
  82 }
  83
  84 void gen_write_bytes(const char *foutname, tfc_fsize offset, tfc_fsize nrbytes)
  85 {
  86         static tfc_fsize wrpos = NOFSIZE;
  87         int fd, x;
  88         size_t lblock, lio, lrem;
  89         tfc_byte *pblk;
  90
  91         for (x = 1; x < NSIG; x++) signal(x, SIG_IGN);
  92         memset(&sigact, 0, sizeof(sigact));
  93         sigact.sa_flags = SA_RESTART;
  94         sigact.sa_handler = print_crypt_status;
  95         sigaction(SIGUSR1, &sigact, NULL);
  96         sigaction(SIGTSTP, &sigact, NULL);
  97         sigaction(SIGALRM, &sigact, NULL);
  98         if (status_timer) setup_next_alarm(status_timer);
  99         sigact.sa_handler = change_status_width;
 100         sigaction(SIGQUIT, &sigact, NULL);
 101         sigact.sa_handler = change_status_timer;
 102         sigaction(SIGUSR2, &sigact, NULL);
 103         if (quiet == NO) {
 104                 sigact.sa_handler = print_crypt_status;
 105                 sigaction(SIGINT, &sigact, NULL);
 106                 sigaction(SIGTERM, &sigact, NULL);
 107         }
 108         else {
 109                 sigact.sa_handler = exit_sigterm;
 110                 sigaction(SIGINT, &sigact, NULL);
 111                 sigaction(SIGTERM, &sigact, NULL);
 112         }
 113         memset(&sigact, 0, sizeof(struct sigaction));
 114
 115         tfc_getcurtime(&delta_time);
 116
 117         if (do_less_stats) do_less_stats = NO;
 118         else do_less_stats = YES;
 119
 120         if (!foutname) {
 121                 fd = 1;
 122                 foutname = TFC_STDOUT_NAME;
 123         }
 124         else if (!strcmp(foutname, "-")) {
 125                 fd = 1;
 126                 foutname = TFC_STDOUT_NAME;
 127         }
 128         else fd = open(foutname, O_WRONLY | O_CREAT | O_LARGEFILE | write_flags, 0666);
 129         if (fd == -1) xerror(NO, NO, YES, "%s", foutname);
 130
 131         if (offset) {
 132                 if (lseek(fd, offset, SEEK_SET) == -1)
 133                         xerror(ignore_seek_errors, NO, NO, "%s: seek failed", foutname);
 134         }
 135
 136         if (ctr_mode == TFC_MODE_PLAIN) memset(srcblk, 0, sizeof(srcblk));
 137
 138         if (verbose) tfc_nfsay(stderr, "%s: writing %lld bytes to %s ... ",
 139                 progname, nrbytes, foutname);
 140
 141         errno = 0;
 142         do_stop = NO;
 143         while (1) {
 144                 if (do_stop) break;
 145                 pblk = srcblk;
 146                 lblock = lrem = blk_len_adj(nrbytes, total_processed_src, blksize);
 147
 148                 if (ctr_mode != TFC_MODE_PLAIN) tfc_getrandom(srcblk, lblock);
 149
 150                 if (error_action == TFC_ERRACT_SYNC) wrpos = tfc_fdgetpos(fd);
 151 _wagain:        lio = xwrite(fd, pblk, lrem);
 152                 if (lio == NOSIZE) {
 153                         if (errno != EIO && catch_all_errors != YES)
 154                                 xerror(NO, NO, YES, "%s", foutname);
 155                         switch (error_action) {
 156                                 case TFC_ERRACT_CONT: xerror(YES, NO, YES, "%s", foutname); goto _wagain; break;
 157                                 case TFC_ERRACT_SYNC:
 158                                 case TFC_ERRACT_LSYNC:
 159                                         xerror(YES, NO, YES, "%s", foutname);
 160                                         if (wrpos == NOFSIZE) lseek(fd, lblock, SEEK_CUR);
 161                                         else lseek(fd, wrpos + lblock, SEEK_SET);
 162                                         break;
 163                                 default: xerror(NO, NO, YES, "%s", foutname); break;
 164                         }
 165                 }
 166                 if (do_fsync && fsync(fd) == -1) xerror(NO, NO, YES, "%s", foutname);
 167                 if (lio < lrem) {
 168                         pblk += lio;
 169                         lrem -= lio;
 170                         goto _wagain;
 171                 }
 172
 173                 total_processed_src += lblock;
 174                 delta_processed += lblock;
 175                 total_processed_dst = total_processed_src;
 176                 if (total_processed_src >= nrbytes) break;
 177         }
 178
 179         if (verbose) tfc_esay("done!");
 180         if (verbose || status_timer) print_crypt_status(0);
 181
 182         xclose(fd);
 183         xexit(0);
 184 }