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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
/* Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*
* Host functions for verified boot.
*/
/* TODO: change all 'return 0', 'return 1' into meaningful return codes */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "host_common.h"
#include "vboot_common.h"
char* StrCopy(char* dest, const char* src, int dest_size)
{
strncpy(dest, src, dest_size);
dest[dest_size - 1] = '\0';
return dest;
}
uint8_t* ReadFile(const char* filename, uint64_t* sizeptr)
{
FILE* f;
uint8_t* buf;
long size;
f = fopen(filename, "rb");
if (!f) {
fprintf(stderr, "Unable to open file %s\n", filename);
return NULL;
}
fseek(f, 0, SEEK_END);
size = ftell(f);
if (size < 0) {
fclose(f);
return NULL;
}
rewind(f);
buf = malloc(size);
if (!buf) {
fclose(f);
return NULL;
}
if(1 != fread(buf, size, 1, f)) {
fprintf(stderr, "Unable to read from file %s\n", filename);
fclose(f);
free(buf);
return NULL;
}
fclose(f);
if (sizeptr)
*sizeptr = size;
return buf;
}
char* ReadFileString(char* dest, int size, const char* filename)
{
char* got;
FILE* f;
f = fopen(filename, "rt");
if (!f)
return NULL;
got = fgets(dest, size, f);
fclose(f);
return got;
}
int ReadFileInt(const char* filename, unsigned* value)
{
char buf[64];
char* e = NULL;
if (!ReadFileString(buf, sizeof(buf), filename))
return -1;
/* Convert to integer. Allow characters after the int ("123 blah"). */
*value = (unsigned)strtoul(buf, &e, 0);
if (e == buf)
return -1; /* No characters consumed, so conversion failed */
return 0;
}
int ReadFileBit(const char* filename, int bitmask)
{
unsigned value;
if (ReadFileInt(filename, &value) < 0)
return -1;
else return (value & bitmask ? 1 : 0);
}
vb2_error_t WriteFile(const char* filename, const void *data, uint64_t size)
{
FILE *f = fopen(filename, "wb");
if (!f) {
fprintf(stderr, "Unable to open file %s\n", filename);
return 1;
}
if (1 != fwrite(data, size, 1, f)) {
fprintf(stderr, "Unable to write to file %s\n", filename);
fclose(f);
unlink(filename); /* Delete any partial file */
return 1;
}
fclose(f);
return 0;
}
|
