1 #include "block_list.h"
2 #include "block_range.h"
3 #include <stdio.h>
4 #include <sys/types.h>
5 #include <sys/stat.h>
6 #include <fcntl.h>
7
8 struct block_list {
9 FILE *f;
10 const char *mountpoint;
11
12 struct {
13 const char *filename;
14 struct block_range *head;
15 struct block_range *tail;
16 } entry;
17 };
18
init(const char * file,const char * mountpoint)19 static void *init(const char *file, const char *mountpoint)
20 {
21 struct block_list *params = malloc(sizeof(*params));
22
23 if (!params)
24 return NULL;
25 params->mountpoint = mountpoint;
26 params->f = fopen(file, "w+");
27 if (!params->f) {
28 free(params);
29 return NULL;
30 }
31 return params;
32 }
33
start_new_file(char * path,ext2_ino_t ino EXT2FS_ATTR ((unused)),struct ext2_inode * inode EXT2FS_ATTR ((unused)),void * data)34 static int start_new_file(char *path, ext2_ino_t ino EXT2FS_ATTR((unused)),
35 struct ext2_inode *inode EXT2FS_ATTR((unused)),
36 void *data)
37 {
38 struct block_list *params = data;
39
40 params->entry.head = params->entry.tail = NULL;
41 params->entry.filename = LINUX_S_ISREG(inode->i_mode) ? path : NULL;
42 return 0;
43 }
44
add_block(ext2_filsys fs EXT2FS_ATTR ((unused)),blk64_t blocknr,int metadata,void * data)45 static int add_block(ext2_filsys fs EXT2FS_ATTR((unused)), blk64_t blocknr,
46 int metadata, void *data)
47 {
48 struct block_list *params = data;
49
50 if (params->entry.filename && !metadata)
51 add_blocks_to_range(¶ms->entry.head, ¶ms->entry.tail,
52 blocknr, blocknr);
53 return 0;
54 }
55
inline_data(void * inline_data EXT2FS_ATTR ((unused)),void * data EXT2FS_ATTR ((unused)))56 static int inline_data(void *inline_data EXT2FS_ATTR((unused)),
57 void *data EXT2FS_ATTR((unused)))
58 {
59 return 0;
60 }
61
end_new_file(void * data)62 static int end_new_file(void *data)
63 {
64 struct block_list *params = data;
65
66 if (!params->entry.filename || !params->entry.head)
67 return 0;
68 if (fprintf(params->f, "%s%s ", params->mountpoint,
69 params->entry.filename) < 0
70 || write_block_ranges(params->f, params->entry.head, " ")
71 || fwrite("\n", 1, 1, params->f) != 1)
72 return -1;
73
74 delete_block_ranges(params->entry.head);
75 return 0;
76 }
77
cleanup(void * data)78 static int cleanup(void *data)
79 {
80 struct block_list *params = data;
81
82 fclose(params->f);
83 free(params);
84 return 0;
85 }
86
87 struct fsmap_format block_list_format = {
88 .init = init,
89 .start_new_file = start_new_file,
90 .add_block = add_block,
91 .inline_data = inline_data,
92 .end_new_file = end_new_file,
93 .cleanup = cleanup,
94 };
95