io.c 3.51 KB
Newer Older
1
2
3
4
5
6
7
/* io.c
 *
 * Miscellaneous functions used by the example programs.
 */

/* nettle, low-level cryptographics library
 *
Niels Möller's avatar
Niels Möller committed
8
 * Copyright (C) 2002 Niels Möller
9
10
11
12
13
14
15
16
17
18
19
20
21
 *  
 * The nettle library is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or (at your
 * option) any later version.
 * 
 * The nettle library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
 * License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with the nettle library; see the file COPYING.LIB.  If not, write to
22
23
 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02111-1301, USA.
24
25
26
27
 */

#if HAVE_CONFIG_H
# include "config.h"
28
#endif
29
30
31
32

#include <stdarg.h>
#include <stdlib.h>

33
34
35
36
/* For errno and strerror */
#include <errno.h>
#include <string.h>

37
38
#include "io.h"

39
40
41
42
43
#define RANDOM_DEVICE "/dev/urandom"
#define BUFSIZE 1000

int quiet_flag = 0;

Niels Möller's avatar
Niels Möller committed
44
45
46
47
48
49
50
51
52
53
54
55
56
void *
xalloc(size_t size)
{
  void *p = malloc(size);
  if (!p)
    {
      fprintf(stderr, "Virtual memory exhausted.\n");
      abort();
    }

  return p;
}

57
58
59
60
61
62
63
64
65
66
67
68
69
void
werror(const char *format, ...)
{
  if (!quiet_flag)
    {
      va_list args;
      va_start(args, format);
      vfprintf(stderr, format, args);
      va_end(args);
    }
}

unsigned
70
read_file(const char *name, unsigned max_size, char **contents)
71
{
Niels Möller's avatar
Niels Möller committed
72
  unsigned size, done;
73
  char *buffer;
74
75
76
77
  FILE *f;
    
  f = fopen(name, "rb");
  if (!f)
78
    {
79
      werror("Opening `%s' failed: %s\n", name, strerror(errno));
80
81
      return 0;
    }
82

Niels Möller's avatar
Niels Möller committed
83
  size = 100;
84

Niels Möller's avatar
Niels Möller committed
85
  for (buffer = NULL, done = 0;; size *= 2)
86
87
88
    {
      char *p;

89
      if (max_size && size > max_size)
90
91
92
	size = max_size;

      /* Space for terminating NUL */
93
      p = realloc(buffer, size + 1);
94
95
96
97
98

      if (!p)
	{
	fail:
	  fclose(f);
99
100
	  free(buffer);
	  *contents = NULL;
101
102
103
	  return 0;
	}

104
      buffer = p;
105
106
      done += fread(buffer + done, 1, size - done, f);

Niels Möller's avatar
Niels Möller committed
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
      if (done < size)
	{
	  /* Short count means EOF or read error */
	  if (ferror(f))
	    {
	      fprintf (stderr, "Reading `%s' failed: %s\n",
		       name, strerror(errno));

	      goto fail;
	    }
	  if (done == 0)
	    /* Treat empty file as error */
	    goto fail;

	  break;
	}

      if (size == max_size)
	break;
126
127
128
129
130
    }
  
  fclose(f);

  /* NUL-terminate the data. */
131
132
133
  buffer[done] = '\0';
  *contents = buffer;
  
134
135
136
137
  return done;
}

int
138
write_string(FILE *f, unsigned size, const char *buffer)
139
{
140
  size_t res = fwrite(buffer, 1, size, f);
141

142
  return res == size;
Niels Möller's avatar
Niels Möller committed
143
144
145
}

int
146
write_file(const char *name, unsigned size, const char *buffer)
Niels Möller's avatar
Niels Möller committed
147
{
148
149
150
151
152
  FILE *f = fopen(name, "wb");
  int res;
  
  if (!f)
    return 0;
153

154
155
  res = write_string(f, size, buffer);
  return fclose(f) == 0 && res;
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
}

int
simple_random(struct yarrow256_ctx *ctx, const char *name)
{
  unsigned length;
  char *buffer;

  if (name)
    length = read_file(name, 0, &buffer);
  else
    length = read_file(RANDOM_DEVICE, 20, &buffer);
  
  if (!length)
    return 0;

  yarrow256_seed(ctx, length, buffer);

174
175
  free(buffer);

176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
  return 1;
}

int
hash_file(const struct nettle_hash *hash, void *ctx, FILE *f)
{
  for (;;)
    {
      char buffer[BUFSIZE];
      size_t res = fread(buffer, 1, sizeof(buffer), f);
      if (ferror(f))
	return 0;
      
      hash->update(ctx, res, buffer);
      if (feof(f))
	return 1;
    }
}