Rev 3386 | Rev 4327 | Go to most recent revision | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed
Rev 3386 | Rev 4153 | ||
---|---|---|---|
Line 36... | Line 36... | ||
36 | * @brief Support for loading TMPFS file system dump. |
36 | * @brief Support for loading TMPFS file system dump. |
37 | */ |
37 | */ |
38 | 38 | ||
39 | #include "tmpfs.h" |
39 | #include "tmpfs.h" |
40 | #include "../../vfs/vfs.h" |
40 | #include "../../vfs/vfs.h" |
41 | #include <ipc/ipc.h> |
- | |
42 | #include <async.h> |
- | |
43 | #include <errno.h> |
41 | #include <errno.h> |
44 | #include <stdlib.h> |
42 | #include <stdlib.h> |
45 | #include <string.h> |
43 | #include <string.h> |
46 | #include <sys/types.h> |
44 | #include <sys/types.h> |
47 | #include <as.h> |
45 | #include <as.h> |
48 | #include <libfs.h> |
46 | #include <libblock.h> |
49 | #include <ipc/services.h> |
- | |
50 | #include <ipc/devmap.h> |
- | |
51 | #include <sys/mman.h> |
- | |
52 | #include <byteorder.h> |
47 | #include <byteorder.h> |
53 | 48 | ||
54 | #define TMPFS_BLOCK_SIZE 1024 |
49 | #define TMPFS_BLOCK_SIZE 1024 |
55 | 50 | ||
56 | struct rdentry { |
51 | struct rdentry { |
57 | uint8_t type; |
52 | uint8_t type; |
58 | uint32_t len; |
53 | uint32_t len; |
59 | } __attribute__((packed)); |
54 | } __attribute__((packed)); |
60 | 55 | ||
61 | static bool |
56 | static bool |
62 | tmpfs_restore_recursion(int phone, void *block, off_t *bufpos, size_t *buflen, |
57 | tmpfs_restore_recursion(int dev, off_t *bufpos, size_t *buflen, off_t *pos, |
63 | off_t *pos, tmpfs_dentry_t *parent) |
58 | tmpfs_dentry_t *parent) |
64 | { |
59 | { |
65 | struct rdentry entry; |
60 | struct rdentry entry; |
66 | libfs_ops_t *ops = &tmpfs_libfs_ops; |
61 | libfs_ops_t *ops = &tmpfs_libfs_ops; |
- | 62 | int rc; |
|
67 | 63 | ||
68 | do { |
64 | do { |
69 | char *fname; |
65 | char *fname; |
70 | tmpfs_dentry_t *node; |
66 | tmpfs_dentry_t *node; |
71 | uint32_t size; |
67 | uint32_t size; |
72 | 68 | ||
73 | if (!libfs_blockread(phone, block, bufpos, buflen, pos, &entry, |
69 | if (block_read(dev, bufpos, buflen, pos, &entry, sizeof(entry), |
74 | sizeof(entry), TMPFS_BLOCK_SIZE)) |
70 | TMPFS_BLOCK_SIZE) != EOK) |
75 | return false; |
71 | return false; |
76 | 72 | ||
77 | entry.len = uint32_t_le2host(entry.len); |
73 | entry.len = uint32_t_le2host(entry.len); |
78 | 74 | ||
79 | switch (entry.type) { |
75 | switch (entry.type) { |
Line 82... | Line 78... | ||
82 | case TMPFS_FILE: |
78 | case TMPFS_FILE: |
83 | fname = malloc(entry.len + 1); |
79 | fname = malloc(entry.len + 1); |
84 | if (fname == NULL) |
80 | if (fname == NULL) |
85 | return false; |
81 | return false; |
86 | 82 | ||
87 | node = (tmpfs_dentry_t *) ops->create(L_FILE); |
83 | node = (tmpfs_dentry_t *) ops->create(dev, L_FILE); |
88 | if (node == NULL) { |
84 | if (node == NULL) { |
89 | free(fname); |
85 | free(fname); |
90 | return false; |
86 | return false; |
91 | } |
87 | } |
92 | 88 | ||
93 | if (!libfs_blockread(phone, block, bufpos, buflen, pos, |
89 | if (block_read(dev, bufpos, buflen, pos, fname, |
94 | fname, entry.len, TMPFS_BLOCK_SIZE)) { |
90 | entry.len, TMPFS_BLOCK_SIZE) != EOK) { |
95 | ops->destroy((void *) node); |
91 | ops->destroy((void *) node); |
96 | free(fname); |
92 | free(fname); |
97 | return false; |
93 | return false; |
98 | } |
94 | } |
99 | fname[entry.len] = 0; |
95 | fname[entry.len] = 0; |
100 | 96 | ||
101 | if (!ops->link((void *) parent, (void *) node, fname)) { |
97 | rc = ops->link((void *) parent, (void *) node, fname); |
- | 98 | if (rc != EOK) { |
|
102 | ops->destroy((void *) node); |
99 | ops->destroy((void *) node); |
103 | free(fname); |
100 | free(fname); |
104 | return false; |
101 | return false; |
105 | } |
102 | } |
106 | free(fname); |
103 | free(fname); |
107 | 104 | ||
108 | if (!libfs_blockread(phone, block, bufpos, buflen, pos, |
105 | if (block_read(dev, bufpos, buflen, pos, &size, |
109 | &size, sizeof(size), TMPFS_BLOCK_SIZE)) |
106 | sizeof(size), TMPFS_BLOCK_SIZE) != EOK) |
110 | return false; |
107 | return false; |
111 | 108 | ||
112 | size = uint32_t_le2host(size); |
109 | size = uint32_t_le2host(size); |
113 | 110 | ||
114 | node->data = malloc(size); |
111 | node->data = malloc(size); |
115 | if (node->data == NULL) |
112 | if (node->data == NULL) |
116 | return false; |
113 | return false; |
117 | 114 | ||
118 | node->size = size; |
115 | node->size = size; |
119 | if (!libfs_blockread(phone, block, bufpos, buflen, pos, |
116 | if (block_read(dev, bufpos, buflen, pos, node->data, |
120 | node->data, size, TMPFS_BLOCK_SIZE)) |
117 | size, TMPFS_BLOCK_SIZE) != EOK) |
121 | return false; |
118 | return false; |
122 | 119 | ||
123 | break; |
120 | break; |
124 | case TMPFS_DIRECTORY: |
121 | case TMPFS_DIRECTORY: |
125 | fname = malloc(entry.len + 1); |
122 | fname = malloc(entry.len + 1); |
126 | if (fname == NULL) |
123 | if (fname == NULL) |
127 | return false; |
124 | return false; |
128 | 125 | ||
129 | node = (tmpfs_dentry_t *) ops->create(L_DIRECTORY); |
126 | node = (tmpfs_dentry_t *) ops->create(dev, L_DIRECTORY); |
130 | if (node == NULL) { |
127 | if (node == NULL) { |
131 | free(fname); |
128 | free(fname); |
132 | return false; |
129 | return false; |
133 | } |
130 | } |
134 | 131 | ||
135 | if (!libfs_blockread(phone, block, bufpos, buflen, pos, |
132 | if (block_read(dev, bufpos, buflen, pos, fname, |
136 | fname, entry.len, TMPFS_BLOCK_SIZE)) { |
133 | entry.len, TMPFS_BLOCK_SIZE) != EOK) { |
137 | ops->destroy((void *) node); |
134 | ops->destroy((void *) node); |
138 | free(fname); |
135 | free(fname); |
139 | return false; |
136 | return false; |
140 | } |
137 | } |
141 | fname[entry.len] = 0; |
138 | fname[entry.len] = 0; |
142 | 139 | ||
143 | if (!ops->link((void *) parent, (void *) node, fname)) { |
140 | rc = ops->link((void *) parent, (void *) node, fname); |
- | 141 | if (rc != EOK) { |
|
144 | ops->destroy((void *) node); |
142 | ops->destroy((void *) node); |
145 | free(fname); |
143 | free(fname); |
146 | return false; |
144 | return false; |
147 | } |
145 | } |
148 | free(fname); |
146 | free(fname); |
149 | 147 | ||
150 | if (!tmpfs_restore_recursion(phone, block, bufpos, |
148 | if (!tmpfs_restore_recursion(dev, bufpos, buflen, pos, |
151 | buflen, pos, node)) |
149 | node)) |
152 | return false; |
150 | return false; |
153 | 151 | ||
154 | break; |
152 | break; |
155 | default: |
153 | default: |
156 | return false; |
154 | return false; |
Line 161... | Line 159... | ||
161 | } |
159 | } |
162 | 160 | ||
163 | bool tmpfs_restore(dev_handle_t dev) |
161 | bool tmpfs_restore(dev_handle_t dev) |
164 | { |
162 | { |
165 | libfs_ops_t *ops = &tmpfs_libfs_ops; |
163 | libfs_ops_t *ops = &tmpfs_libfs_ops; |
- | 164 | int rc; |
|
166 | 165 | ||
167 | void *block = mmap(NULL, TMPFS_BLOCK_SIZE, |
166 | rc = block_init(dev, TMPFS_BLOCK_SIZE); |
168 | PROTO_READ | PROTO_WRITE, MAP_ANONYMOUS | MAP_PRIVATE, 0, 0); |
- | |
169 | - | ||
170 | if (block == NULL) |
167 | if (rc != EOK) |
171 | return false; |
- | |
172 | - | ||
173 | int phone = ipc_connect_me_to(PHONE_NS, SERVICE_DEVMAP, |
- | |
174 | DEVMAP_CONNECT_TO_DEVICE, dev); |
- | |
175 | - | ||
176 | if (phone < 0) { |
- | |
177 | munmap(block, TMPFS_BLOCK_SIZE); |
- | |
178 | return false; |
168 | return false; |
179 | } |
- | |
180 | - | ||
181 | if (ipc_share_out_start(phone, block, AS_AREA_READ | AS_AREA_WRITE) != |
- | |
182 | EOK) |
- | |
183 | goto error; |
- | |
184 | 169 | ||
185 | off_t bufpos = 0; |
170 | off_t bufpos = 0; |
186 | size_t buflen = 0; |
171 | size_t buflen = 0; |
187 | off_t pos = 0; |
172 | off_t pos = 0; |
188 | 173 | ||
189 | char tag[6]; |
174 | char tag[6]; |
190 | if (!libfs_blockread(phone, block, &bufpos, &buflen, &pos, tag, 5, |
175 | if (block_read(dev, &bufpos, &buflen, &pos, tag, 5, |
191 | TMPFS_BLOCK_SIZE)) |
176 | TMPFS_BLOCK_SIZE) != EOK) |
192 | goto error; |
177 | goto error; |
193 | 178 | ||
194 | tag[5] = 0; |
179 | tag[5] = 0; |
195 | if (strcmp(tag, "TMPFS") != 0) |
180 | if (strcmp(tag, "TMPFS") != 0) |
196 | goto error; |
181 | goto error; |
197 | 182 | ||
198 | if (!tmpfs_restore_recursion(phone, block, &bufpos, &buflen, &pos, |
183 | if (!tmpfs_restore_recursion(dev, &bufpos, &buflen, &pos, |
199 | ops->root_get(dev))) |
184 | ops->root_get(dev))) |
200 | goto error; |
185 | goto error; |
201 | 186 | ||
202 | ipc_hangup(phone); |
187 | block_fini(dev); |
203 | munmap(block, TMPFS_BLOCK_SIZE); |
- | |
204 | return true; |
188 | return true; |
205 | 189 | ||
206 | error: |
190 | error: |
207 | ipc_hangup(phone); |
191 | block_fini(dev); |
208 | munmap(block, TMPFS_BLOCK_SIZE); |
- | |
209 | return false; |
192 | return false; |
210 | } |
193 | } |
211 | 194 | ||
212 | /** |
195 | /** |
213 | * @} |
196 | * @} |