WebSVN – HelenOS – Diff – /branches/dynload/uspace/srv/fs/tmpfs/tmpfs_dump.c


/*
 * Copyright (c) 2008 Jakub Jermar
 * Copyright (c) 2008 Martin Decky
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
 
/** @addtogroup fs
 * @{
 */
 
/**
 * @file    tmpfs_dump.c
 * @brief   Support for loading TMPFS file system dump.
 */
 
#include "tmpfs.h"
#include "../../vfs/vfs.h"
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <as.h>
#include <libblock.h>
#include <byteorder.h>
 
#define TMPFS_BLOCK_SIZE    1024
 
struct rdentry {
    uint8_t type;
    uint32_t len;
} __attribute__((packed));
 
static bool
tmpfs_restore_recursion(int dev, off_t *bufpos, size_t *buflen, off_t *pos,
    tmpfs_dentry_t *parent)
{
    struct rdentry entry;
    libfs_ops_t *ops = &tmpfs_libfs_ops;
    int rc;
   
    do {
        char *fname;
        tmpfs_dentry_t *node;
        uint32_t size;
       
        if (block_read(dev, bufpos, buflen, pos, &entry, sizeof(entry),
            TMPFS_BLOCK_SIZE) != EOK)
            return false;
       
        entry.len = uint32_t_le2host(entry.len);
       
        switch (entry.type) {
        case TMPFS_NONE:
            break;
        case TMPFS_FILE:
            fname = malloc(entry.len + 1);
            if (fname == NULL)
                return false;
           
            node = (tmpfs_dentry_t *) ops->create(dev, L_FILE);
            if (node == NULL) {
                free(fname);
                return false;
            }
           
            if (block_read(dev, bufpos, buflen, pos, fname,
                entry.len, TMPFS_BLOCK_SIZE) != EOK) {
                ops->destroy((void *) node);
                free(fname);
                return false;
            }
            fname[entry.len] = 0;
           
            rc = ops->link((void *) parent, (void *) node, fname);
            if (rc != EOK) {
                ops->destroy((void *) node);
                free(fname);
                return false;
            }
            free(fname);
           
            if (block_read(dev, bufpos, buflen, pos, &size,
                sizeof(size), TMPFS_BLOCK_SIZE) != EOK)
                return false;
           
            size = uint32_t_le2host(size);
           
            node->data = malloc(size);
            if (node->data == NULL)
                return false;
           
            node->size = size;
            if (block_read(dev, bufpos, buflen, pos, node->data,
                size, TMPFS_BLOCK_SIZE) != EOK)
                return false;
           
            break;
        case TMPFS_DIRECTORY:
            fname = malloc(entry.len + 1);
            if (fname == NULL)
                return false;
           
            node = (tmpfs_dentry_t *) ops->create(dev, L_DIRECTORY);
            if (node == NULL) {
                free(fname);
                return false;
            }
           
            if (block_read(dev, bufpos, buflen, pos, fname,
                entry.len, TMPFS_BLOCK_SIZE) != EOK) {
                ops->destroy((void *) node);
                free(fname);
                return false;
            }
            fname[entry.len] = 0;
 
            rc = ops->link((void *) parent, (void *) node, fname);
            if (rc != EOK) {
                ops->destroy((void *) node);
                free(fname);
                return false;
            }
            free(fname);
           
            if (!tmpfs_restore_recursion(dev, bufpos, buflen, pos,
                node))
                return false;
           
            break;
        default:
            return false;
        }
    } while (entry.type != TMPFS_NONE);
   
    return true;
}
 
bool tmpfs_restore(dev_handle_t dev)
{
    libfs_ops_t *ops = &tmpfs_libfs_ops;
    int rc;
 
    rc = block_init(dev, TMPFS_BLOCK_SIZE);
    if (rc != EOK)
        return false;
   
    off_t bufpos = 0;
    size_t buflen = 0;
    off_t pos = 0;
   
    char tag[6];
    if (block_read(dev, &bufpos, &buflen, &pos, tag, 5,
        TMPFS_BLOCK_SIZE) != EOK)
        goto error;
   
    tag[5] = 0;
    if (str_cmp(tag, "TMPFS") != 0)
        goto error;
   
    if (!tmpfs_restore_recursion(dev, &bufpos, &buflen, &pos,
        ops->root_get(dev)))
        goto error;
       
    block_fini(dev);
    return true;
   
error:
    block_fini(dev);
    return false;
}
 
/**
 * @}
 */
 

Rev 3674	Rev 4348
1	/*	1	/*
2	* Copyright (c) 2008 Jakub Jermar	2	* Copyright (c) 2008 Jakub Jermar
3	* Copyright (c) 2008 Martin Decky	3	* Copyright (c) 2008 Martin Decky
4	* All rights reserved.	4	* All rights reserved.
5	*	5	*
6	* Redistribution and use in source and binary forms, with or without	6	* Redistribution and use in source and binary forms, with or without
7	* modification, are permitted provided that the following conditions	7	* modification, are permitted provided that the following conditions
8	* are met:	8	* are met:
9	*	9	*
10	* - Redistributions of source code must retain the above copyright	10	* - Redistributions of source code must retain the above copyright
11	* notice, this list of conditions and the following disclaimer.	11	* notice, this list of conditions and the following disclaimer.
12	* - Redistributions in binary form must reproduce the above copyright	12	* - Redistributions in binary form must reproduce the above copyright
13	* notice, this list of conditions and the following disclaimer in the	13	* notice, this list of conditions and the following disclaimer in the
14	* documentation and/or other materials provided with the distribution.	14	* documentation and/or other materials provided with the distribution.
15	* - The name of the author may not be used to endorse or promote products	15	* - The name of the author may not be used to endorse or promote products
16	* derived from this software without specific prior written permission.	16	* derived from this software without specific prior written permission.
17	*	17	*
18	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR	18	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES	19	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.	20	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,	21	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT	22	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,	23	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY	24	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT	25	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF	26	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.	27	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28	*/	28	*/
29		29
30	/** @addtogroup fs	30	/** @addtogroup fs
31	* @{	31	* @{
32	*/	32	*/
33		33
34	/**	34	/**
35	* @file tmpfs_dump.c	35	* @file tmpfs_dump.c
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 <errno.h>	41	#include <errno.h>
42	#include <stdlib.h>	42	#include <stdlib.h>
43	#include <string.h>	43	#include <string.h>
44	#include <sys/types.h>	44	#include <sys/types.h>
45	#include <as.h>	45	#include <as.h>
46	#include <libblock.h>	46	#include <libblock.h>
47	#include <byteorder.h>	47	#include <byteorder.h>
48		48
49	#define TMPFS_BLOCK_SIZE 1024	49	#define TMPFS_BLOCK_SIZE 1024
50		50
51	struct rdentry {	51	struct rdentry {
52	uint8_t type;	52	uint8_t type;
53	uint32_t len;	53	uint32_t len;
54	} __attribute__((packed));	54	} __attribute__((packed));
55		55
56	static bool	56	static bool
57	tmpfs_restore_recursion(int dev, off_t bufpos, size_t buflen, off_t *pos,	57	tmpfs_restore_recursion(int dev, off_t bufpos, size_t buflen, off_t *pos,
58	tmpfs_dentry_t *parent)	58	tmpfs_dentry_t *parent)
59	{	59	{
60	struct rdentry entry;	60	struct rdentry entry;
61	libfs_ops_t *ops = &tmpfs_libfs_ops;	61	libfs_ops_t *ops = &tmpfs_libfs_ops;
62	int rc;	62	int rc;
63		63
64	do {	64	do {
65	char *fname;	65	char *fname;
66	tmpfs_dentry_t *node;	66	tmpfs_dentry_t *node;
67	uint32_t size;	67	uint32_t size;
68		68
69	if (block_read(dev, bufpos, buflen, pos, &entry, sizeof(entry),	69	if (block_read(dev, bufpos, buflen, pos, &entry, sizeof(entry),
70	TMPFS_BLOCK_SIZE) != EOK)	70	TMPFS_BLOCK_SIZE) != EOK)
71	return false;	71	return false;
72		72
73	entry.len = uint32_t_le2host(entry.len);	73	entry.len = uint32_t_le2host(entry.len);
74		74
75	switch (entry.type) {	75	switch (entry.type) {
76	case TMPFS_NONE:	76	case TMPFS_NONE:
77	break;	77	break;
78	case TMPFS_FILE:	78	case TMPFS_FILE:
79	fname = malloc(entry.len + 1);	79	fname = malloc(entry.len + 1);
80	if (fname == NULL)	80	if (fname == NULL)
81	return false;	81	return false;
82		82
83	node = (tmpfs_dentry_t *) ops->create(dev, L_FILE);	83	node = (tmpfs_dentry_t *) ops->create(dev, L_FILE);
84	if (node == NULL) {	84	if (node == NULL) {
85	free(fname);	85	free(fname);
86	return false;	86	return false;
87	}	87	}
88		88
89	if (block_read(dev, bufpos, buflen, pos, fname,	89	if (block_read(dev, bufpos, buflen, pos, fname,
90	entry.len, TMPFS_BLOCK_SIZE) != EOK) {	90	entry.len, TMPFS_BLOCK_SIZE) != EOK) {
91	ops->destroy((void *) node);	91	ops->destroy((void *) node);
92	free(fname);	92	free(fname);
93	return false;	93	return false;
94	}	94	}
95	fname[entry.len] = 0;	95	fname[entry.len] = 0;
96		96
97	rc = ops->link((void ) parent, (void ) node, fname);	97	rc = ops->link((void ) parent, (void ) node, fname);
98	if (rc != EOK) {	98	if (rc != EOK) {
99	ops->destroy((void *) node);	99	ops->destroy((void *) node);
100	free(fname);	100	free(fname);
101	return false;	101	return false;
102	}	102	}
103	free(fname);	103	free(fname);
104		104
105	if (block_read(dev, bufpos, buflen, pos, &size,	105	if (block_read(dev, bufpos, buflen, pos, &size,
106	sizeof(size), TMPFS_BLOCK_SIZE) != EOK)	106	sizeof(size), TMPFS_BLOCK_SIZE) != EOK)
107	return false;	107	return false;
108		108
109	size = uint32_t_le2host(size);	109	size = uint32_t_le2host(size);
110		110
111	node->data = malloc(size);	111	node->data = malloc(size);
112	if (node->data == NULL)	112	if (node->data == NULL)
113	return false;	113	return false;
114		114
115	node->size = size;	115	node->size = size;
116	if (block_read(dev, bufpos, buflen, pos, node->data,	116	if (block_read(dev, bufpos, buflen, pos, node->data,
117	size, TMPFS_BLOCK_SIZE) != EOK)	117	size, TMPFS_BLOCK_SIZE) != EOK)
118	return false;	118	return false;
119		119
120	break;	120	break;
121	case TMPFS_DIRECTORY:	121	case TMPFS_DIRECTORY:
122	fname = malloc(entry.len + 1);	122	fname = malloc(entry.len + 1);
123	if (fname == NULL)	123	if (fname == NULL)
124	return false;	124	return false;
125		125
126	node = (tmpfs_dentry_t *) ops->create(dev, L_DIRECTORY);	126	node = (tmpfs_dentry_t *) ops->create(dev, L_DIRECTORY);
127	if (node == NULL) {	127	if (node == NULL) {
128	free(fname);	128	free(fname);
129	return false;	129	return false;
130	}	130	}
131		131
132	if (block_read(dev, bufpos, buflen, pos, fname,	132	if (block_read(dev, bufpos, buflen, pos, fname,
133	entry.len, TMPFS_BLOCK_SIZE) != EOK) {	133	entry.len, TMPFS_BLOCK_SIZE) != EOK) {
134	ops->destroy((void *) node);	134	ops->destroy((void *) node);
135	free(fname);	135	free(fname);
136	return false;	136	return false;
137	}	137	}
138	fname[entry.len] = 0;	138	fname[entry.len] = 0;
139		139
140	rc = ops->link((void ) parent, (void ) node, fname);	140	rc = ops->link((void ) parent, (void ) node, fname);
141	if (rc != EOK) {	141	if (rc != EOK) {
142	ops->destroy((void *) node);	142	ops->destroy((void *) node);
143	free(fname);	143	free(fname);
144	return false;	144	return false;
145	}	145	}
146	free(fname);	146	free(fname);
147		147
148	if (!tmpfs_restore_recursion(dev, bufpos, buflen, pos,	148	if (!tmpfs_restore_recursion(dev, bufpos, buflen, pos,
149	node))	149	node))
150	return false;	150	return false;
151		151
152	break;	152	break;
153	default:	153	default:
154	return false;	154	return false;
155	}	155	}
156	} while (entry.type != TMPFS_NONE);	156	} while (entry.type != TMPFS_NONE);
157		157
158	return true;	158	return true;
159	}	159	}
160		160
161	bool tmpfs_restore(dev_handle_t dev)	161	bool tmpfs_restore(dev_handle_t dev)
162	{	162	{
163	libfs_ops_t *ops = &tmpfs_libfs_ops;	163	libfs_ops_t *ops = &tmpfs_libfs_ops;
164	int rc;	164	int rc;
165		165
166	rc = block_init(dev, TMPFS_BLOCK_SIZE);	166	rc = block_init(dev, TMPFS_BLOCK_SIZE);
167	if (rc != EOK)	167	if (rc != EOK)
168	return false;	168	return false;
169		169
170	off_t bufpos = 0;	170	off_t bufpos = 0;
171	size_t buflen = 0;	171	size_t buflen = 0;
172	off_t pos = 0;	172	off_t pos = 0;
173		173
174	char tag[6];	174	char tag[6];
175	if (block_read(dev, &bufpos, &buflen, &pos, tag, 5,	175	if (block_read(dev, &bufpos, &buflen, &pos, tag, 5,
176	TMPFS_BLOCK_SIZE) != EOK)	176	TMPFS_BLOCK_SIZE) != EOK)
177	goto error;	177	goto error;
178		178
179	tag[5] = 0;	179	tag[5] = 0;
180	if (strcmp(tag, "TMPFS") != 0)	180	if (str_cmp(tag, "TMPFS") != 0)
181	goto error;	181	goto error;
182		182
183	if (!tmpfs_restore_recursion(dev, &bufpos, &buflen, &pos,	183	if (!tmpfs_restore_recursion(dev, &bufpos, &buflen, &pos,
184	ops->root_get(dev)))	184	ops->root_get(dev)))
185	goto error;	185	goto error;
186		186
187	block_fini(dev);	187	block_fini(dev);
188	return true;	188	return true;
189		189
190	error:	190	error:
191	block_fini(dev);	191	block_fini(dev);
192	return false;	192	return false;
193	}	193	}
194		194
195	/**	195	/**
196	* @}	196	* @}
197	*/	197	*/
198		198

Subversion Repositories HelenOS

(root)/branches/dynload/uspace/srv/fs/tmpfs/tmpfs_dump.c @ 3674 – Rev 3674 → 4348