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
|
/* SPDX-License-Identifier: LGPL-3.0-or-later */
/*
* read_table.c
*
* Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at>
*/
#define SQFS_BUILDING_DLL
#include "config.h"
#include "sqfs/meta_reader.h"
#include "sqfs/error.h"
#include "sqfs/table.h"
#include "sqfs/block.h"
#include "sqfs/io.h"
#include "util.h"
#include <stdlib.h>
int sqfs_read_table(sqfs_file_t *file, sqfs_compressor_t *cmp,
size_t table_size, sqfs_u64 location, sqfs_u64 lower_limit,
sqfs_u64 upper_limit, void **out)
{
size_t diff, block_count, blk_idx = 0;
sqfs_u64 start, *locations;
sqfs_meta_reader_t *m;
void *data, *ptr;
int err;
data = malloc(table_size);
if (data == NULL)
return SQFS_ERROR_ALLOC;
/* restore list from image */
block_count = table_size / SQFS_META_BLOCK_SIZE;
if ((table_size % SQFS_META_BLOCK_SIZE) != 0)
++block_count;
locations = alloc_array(sizeof(sqfs_u64), block_count);
if (locations == NULL) {
err = SQFS_ERROR_ALLOC;
goto fail_data;
}
err = file->read_at(file, location, locations,
sizeof(sqfs_u64) * block_count);
if (err)
goto fail_idx;
/* Read the actual data */
m = sqfs_meta_reader_create(file, cmp, lower_limit, upper_limit);
if (m == NULL) {
err = SQFS_ERROR_ALLOC;
goto fail_idx;
}
ptr = data;
while (table_size > 0) {
start = le64toh(locations[blk_idx++]);
err = sqfs_meta_reader_seek(m, start, 0);
if (err)
goto fail;
diff = SQFS_META_BLOCK_SIZE;
if (diff > table_size)
diff = table_size;
err = sqfs_meta_reader_read(m, ptr, diff);
if (err)
goto fail;
ptr = (char *)ptr + diff;
table_size -= diff;
}
sqfs_meta_reader_destroy(m);
free(locations);
*out = data;
return 0;
fail:
sqfs_meta_reader_destroy(m);
fail_idx:
free(locations);
fail_data:
free(data);
*out = NULL;
return err;
}
|
