io.c 3.65 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
/* io.c
 *
 * Miscellaneous functions used by the example programs.
 */

/* nettle, low-level cryptographics library
 *
 * Copyright (C) 2002 Niels Mller
 *  
 * 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
 * the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
 * MA 02111-1307, USA.
 */

#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
70
71
void
werror(const char *format, ...)
{
  if (!quiet_flag)
    {
      va_list args;
      va_start(args, format);
      vfprintf(stderr, format, args);
      va_end(args);
    }
}

#define MIN(a,b) (((a) < (b)) ? (a) : (b))

unsigned
72
read_file(const char *name, unsigned max_size, char **contents)
73
74
75
{
  unsigned size;
  unsigned done;
76
  char *buffer;
77
78
79
80
  FILE *f;
    
  f = fopen(name, "rb");
  if (!f)
81
    {
82
      werror("Opening `%s' failed: %s\n", name, strerror(errno));
83
84
      return 0;
    }
85
  buffer = NULL;
86
87
88
89
90

  if (max_size && max_size < 100)
    size = max_size;
  else
    size = 100;
91
92
93
94
95
96
97

  /* FIXME: The use of feof and ferror in this loop is a bit confused
     (but I think it is still correct). We should check the return
     value of fread, and call feof and/or ferror when we get a short
     item count. */	

  for (done = 0;
98
99
100
101
102
       (!max_size || done < max_size) && !feof(f);
       size *= 2)
    {
      char *p;

103
      if (max_size && size > max_size)
104
105
106
	size = max_size;

      /* Space for terminating NUL */
107
      p = realloc(buffer, size + 1);
108
109
110
111
112

      if (!p)
	{
	fail:
	  fclose(f);
113
114
	  free(buffer);
	  *contents = NULL;
115
116
117
	  return 0;
	}

118
      buffer = p;
119
120
121
122
123
124
125
126
127
      done += fread(buffer + done, 1, size - done, f);

      if (ferror(f))
	goto fail;
    }
  
  fclose(f);

  /* NUL-terminate the data. */
128
129
130
  buffer[done] = '\0';
  *contents = buffer;
  
131
132
133
134
135
136
137
138
139
140
141
142
143
144
  return done;
}

int
write_file(const char *name, unsigned size, const char *buffer)
{
  FILE *f = fopen(name, "wb");
  unsigned res;
  
  if (!f)
    return 0;

  res = fwrite(buffer, 1, size, f);
  
Niels Möller's avatar
Niels Möller committed
145
  if (res < size)
146
147
    res = 0;

Niels Möller's avatar
Niels Möller committed
148
149
150
151
152
153
154
  return fclose(f) == 0 && res > 0;
}

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

156
  return res == size;
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
}

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);

175
176
  free(buffer);

177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
  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;
    }
}