1 /*
2  * Copyright (C) 2016 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include "util.h"
18 
19 #include <stdio.h>
20 #include <stdlib.h>
21 
22 #include "libfdt.h"
23 #include "libufdt_sysdeps.h"
24 
25 static char *load_file_contents(FILE *fp, size_t *len_ptr) {
26   // Gets the file size.
27   fseek(fp, 0, SEEK_END);
28   size_t len = ftell(fp);
29   fseek(fp, 0, SEEK_SET);
30 
31   char *buf = malloc(len);
32   if (buf == NULL) {
33     return NULL;
34   }
35 
36   if (fread(buf, len, 1, fp) != 1) {
37     free(buf);
38     return NULL;
39   }
40 
41   if (len_ptr) {
42     *len_ptr = len;
43   }
44 
45   return buf;
46 }
47 
48 char *load_file(const char *filename, size_t *len_ptr) {
49   FILE *fp = fopen(filename, "r");
50   if (!fp) {
51     return NULL;
52   }
53 
54   char *buf = load_file_contents(fp, len_ptr);
55 
56   fclose(fp);
57 
58   return buf;
59 }
60 
61 int write_buf_to_file(const char *filename,
62                       const void *buf, size_t buf_size) {
63   int ret = 0;
64   FILE *fout = NULL;
65 
66   fout = fopen(filename, "wb");
67   if (!fout) {
68     ret = 1;
69     goto end;
70   }
71   if (fwrite(buf, 1, buf_size, fout) < 1) {
72     ret = 2;
73     goto end;
74   }
75 
76 end:
77   if (fout) fclose(fout);
78 
79   return ret;
80 }
81 
82 int write_fdt_to_file(const char *filename, const void *fdt) {
83   return write_buf_to_file(filename, fdt, fdt_totalsize(fdt));
84 }
85