tfc_random.c

   1 /*
   2  * tfcrypt -- high security Threefish encryption tool.
   3  *
   4  * tfcrypt is copyrighted:
   5  * Copyright (C) 2012-2019 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         xexit_no_nl = YES;
  92
  93         for (x = 1; x < NSIG; x++) signal(x, SIG_IGN);
  94         memset(&sigact, 0, sizeof(sigact));
  95         sigact.sa_flags = SA_RESTART;
  96         sigact.sa_handler = print_crypt_status;
  97         sigaction(SIGUSR1, &sigact, NULL);
  98         sigaction(SIGALRM, &sigact, NULL);
  99         if (status_timer) setup_next_alarm(status_timer > 1000000 ? 1000000 : status_timer);
 100         sigact.sa_handler = change_status_width;
 101         sigaction(SIGQUIT, &sigact, NULL);
 102         sigact.sa_handler = change_status_timer;
 103         sigaction(SIGUSR2, &sigact, NULL);
 104         if (quiet == NO) {
 105                 sigact.sa_handler = print_crypt_status;
 106                 sigaction(SIGINT, &sigact, NULL);
 107                 sigaction(SIGTERM, &sigact, NULL);
 108                 sigaction(SIGTSTP, &sigact, NULL);
 109         }
 110         else {
 111                 sigact.sa_handler = exit_sigterm;
 112                 sigaction(SIGINT, &sigact, NULL);
 113                 sigaction(SIGTERM, &sigact, NULL);
 114                 sigact.sa_handler = handle_sigtstp;
 115                 sigaction(SIGTSTP, &sigact, NULL);
 116         }
 117         memset(&sigact, 0, sizeof(struct sigaction));
 118
 119         tfc_getcurtime(&delta_time);
 120
 121         if (do_less_stats) do_less_stats = NO;
 122         else do_less_stats = YES;
 123
 124         if (!foutname) {
 125                 fd = 1;
 126                 foutname = TFC_STDOUT_NAME;
 127         }
 128         else if (!strcmp(foutname, "-")) {
 129                 fd = 1;
 130                 foutname = TFC_STDOUT_NAME;
 131         }
 132         else fd = open(foutname, O_WRONLY | O_CREAT | O_LARGEFILE | write_flags, 0666);
 133         if (fd == -1) xerror(NO, NO, YES, "%s", foutname);
 134
 135         if (offset) {
 136                 if (lseek(fd, offset, SEEK_SET) == -1)
 137                         xerror(ignore_seek_errors, NO, NO, "%s: seek failed", foutname);
 138         }
 139
 140         if (ctr_mode == TFC_MODE_PLAIN) memset(srcblk, 0, sizeof(srcblk));
 141
 142         if (verbose) tfc_nfsay(stderr, "%s: writing %lld bytes to %s ... ",
 143                 tfc_format_pid(progname), nrbytes, foutname);
 144
 145         errno = 0;
 146         do_stop = NO;
 147         while (1) {
 148                 if (do_stop) break;
 149                 pblk = srcblk;
 150                 lblock = lrem = blk_len_adj(nrbytes, total_processed_src, blksize);
 151
 152                 if (ctr_mode != TFC_MODE_PLAIN) tfc_getrandom(srcblk, lblock);
 153
 154                 if (error_action == TFC_ERRACT_SYNC) wrpos = tfc_fdgetpos(fd);
 155 _wagain:        lio = xwrite(fd, pblk, lrem);
 156                 if (lio == NOSIZE) {
 157                         if (errno != EIO && catch_all_errors != YES)
 158                                 xerror(NO, NO, YES, "%s", foutname);
 159                         switch (error_action) {
 160                                 case TFC_ERRACT_CONT: xerror(YES, NO, YES, "%s", foutname); goto _wagain; break;
 161                                 case TFC_ERRACT_SYNC:
 162                                 case TFC_ERRACT_LSYNC:
 163                                         xerror(YES, NO, YES, "%s", foutname);
 164                                         if (wrpos == NOFSIZE) lseek(fd, lblock, SEEK_CUR);
 165                                         else lseek(fd, wrpos + lblock, SEEK_SET);
 166                                         break;
 167                                 default: xerror(NO, NO, YES, "%s", foutname); break;
 168                         }
 169                 }
 170                 if (do_fsync && fsync(fd) == -1) xerror(NO, NO, YES, "%s", foutname);
 171                 if (lio < lrem) {
 172                         pblk += lio;
 173                         lrem -= lio;
 174                         goto _wagain;
 175                 }
 176
 177                 total_processed_src += lblock;
 178                 delta_processed += lblock;
 179                 total_processed_dst = total_processed_src;
 180                 if (total_processed_src >= nrbytes) break;
 181         }
 182
 183         if (verbose) tfc_esay("done!");
 184         if (verbose || status_timer) print_crypt_status(0);
 185         if (verbose) tfc_esay("\n");
 186
 187         xclose(fd);
 188         xexit(0);
 189 }