1ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross/*
2ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross * Copyright (C) 2010 The Android Open Source Project
3ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross *
4ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross * Licensed under the Apache License, Version 2.0 (the "License");
5ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross * you may not use this file except in compliance with the License.
6ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross * You may obtain a copy of the License at
7ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross *
8ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross *      http://www.apache.org/licenses/LICENSE-2.0
9ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross *
10ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross * Unless required by applicable law or agreed to in writing, software
11ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross * distributed under the License is distributed on an "AS IS" BASIS,
12ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross * See the License for the specific language governing permissions and
14ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross * limitations under the License.
15ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross */
16ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
17ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#ifndef _EXT4_UTILS_H_
18ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define _EXT4_UTILS_H_
19ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
2033f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#define _GNU_SOURCE
2133f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#define _FILE_OFFSET_BITS 64
2233f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#define _LARGEFILE64_SOURCE
2333f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#include <sys/types.h>
2433f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#include <unistd.h>
2533f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross
26ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#include <sys/types.h>
27ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#include <errno.h>
28ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#include <stdarg.h>
29ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#include <stdio.h>
30ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#include <stdlib.h>
31ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#include <string.h>
322ae7663e1e064000356ee9e49ebd08bdddc545f9Ken Sumrall#include <setjmp.h>
33ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
3433f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#if defined(__APPLE__) && defined(__MACH__)
3533f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#define lseek64 lseek
36fe4a03182b5b17d333511c72406f926a791345d3Colin Cross#define ftruncate64 ftruncate
37fe4a03182b5b17d333511c72406f926a791345d3Colin Cross#define mmap64 mmap
3833f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#define off64_t off_t
3933f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#endif
4033f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross
4133f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#ifdef __BIONIC__
4233f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Crossextern void*  __mmap2(void *, size_t, int, int, int, off_t);
4333f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Crossstatic inline void *mmap64(void *addr, size_t length, int prot, int flags,
4433f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross        int fd, off64_t offset)
4533f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross{
4633f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross    return __mmap2(addr, length, prot, flags, fd, offset >> 12);
4733f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross}
4833f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross#endif
4933f96c66e9a1f2e266a75e5e84c091dffa6ef118Colin Cross
50ec0a2e83dc66d67addeb90e83144187691852a3eColin Crossextern int force;
51ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
52ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define warn(fmt, args...) do { fprintf(stderr, "warning: %s: " fmt "\n", __func__, ## args); } while (0)
532ae7663e1e064000356ee9e49ebd08bdddc545f9Ken Sumrall#define error(fmt, args...) do { fprintf(stderr, "error: %s: " fmt "\n", __func__, ## args); if (!force) longjmp(setjmp_env, EXIT_FAILURE); } while (0)
54a7ed433f2dc0116627a93b18fbb260f0665ca0cbColin Cross#define error_errno(s, args...) error(s ": %s", ##args, strerror(errno))
552ae7663e1e064000356ee9e49ebd08bdddc545f9Ken Sumrall#define critical_error(fmt, args...) do { fprintf(stderr, "critical error: %s: " fmt "\n", __func__, ## args); longjmp(setjmp_env, EXIT_FAILURE); } while (0)
56a7ed433f2dc0116627a93b18fbb260f0665ca0cbColin Cross#define critical_error_errno(s, args...) critical_error(s ": %s", ##args, strerror(errno))
57ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
58ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define EXT4_SUPER_MAGIC 0xEF53
59ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define EXT4_JNL_BACKUP_BLOCKS 1
60ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
61ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define min(a, b) ((a) < (b) ? (a) : (b))
62ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
63ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define DIV_ROUND_UP(x, y) (((x) + (y) - 1)/(y))
646bd2b5d80db5630cb73e3e5a4be3b38afa241c5aColin Cross#define ALIGN(x, y) ((y) * DIV_ROUND_UP((x), (y)))
65ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
66ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define __le64 u64
67ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define __le32 u32
68ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define __le16 u16
69ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
70ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define __be64 u64
71ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define __be32 u32
72ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define __be16 u16
73ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
74ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define __u64 u64
75ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define __u32 u32
76ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define __u16 u16
77ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#define __u8 u8
78ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
79ec0a2e83dc66d67addeb90e83144187691852a3eColin Crosstypedef unsigned long long u64;
80435a8b61e925e3efb22fce08612efe210e83f791Ken Sumralltypedef signed long long s64;
81ec0a2e83dc66d67addeb90e83144187691852a3eColin Crosstypedef unsigned int u32;
82ec0a2e83dc66d67addeb90e83144187691852a3eColin Crosstypedef unsigned short int u16;
83ec0a2e83dc66d67addeb90e83144187691852a3eColin Crosstypedef unsigned char u8;
84ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
85ec0a2e83dc66d67addeb90e83144187691852a3eColin Crossstruct block_group_info;
86ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
87ec0a2e83dc66d67addeb90e83144187691852a3eColin Crossstruct ext2_group_desc {
88ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	__le32 bg_block_bitmap;
89ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	__le32 bg_inode_bitmap;
90ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	__le32 bg_inode_table;
91ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	__le16 bg_free_blocks_count;
92ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	__le16 bg_free_inodes_count;
93ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	__le16 bg_used_dirs_count;
94ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	__le16 bg_pad;
95ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	__le32 bg_reserved[3];
96ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross};
97ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
98ec0a2e83dc66d67addeb90e83144187691852a3eColin Crossstruct fs_info {
99435a8b61e925e3efb22fce08612efe210e83f791Ken Sumrall	s64 len;	/* If set to 0, ask the block device for the size,
100435a8b61e925e3efb22fce08612efe210e83f791Ken Sumrall			 * if less than 0, reserve that much space at the
101435a8b61e925e3efb22fce08612efe210e83f791Ken Sumrall			 * end of the partition, else use the size given. */
102ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 block_size;
103ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 blocks_per_group;
104ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 inodes_per_group;
105ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 inode_size;
106ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 inodes;
107ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 journal_blocks;
108ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u16 feat_ro_compat;
109ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u16 feat_compat;
110ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u16 feat_incompat;
11122742ce739a046a079b2e1b03342a25472dfa352Colin Cross	u32 bg_desc_reserve_blocks;
112ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	const char *label;
113e4b5ae8ab07e698b95f004c9226000b02f853abcColin Cross	u8 no_journal;
114ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross};
115ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
116ec0a2e83dc66d67addeb90e83144187691852a3eColin Crossstruct fs_aux_info {
117ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	struct ext4_super_block *sb;
118107a9f161babc20daf915311146b0e864d3b4157Ken Sumrall	struct ext4_super_block **backup_sb;
119ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	struct ext2_group_desc *bg_desc;
120ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	struct block_group_info *bgs;
121ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 first_data_block;
122ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u64 len_blocks;
123ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 inode_table_blocks;
124ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 groups;
125ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 bg_desc_blocks;
126ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 default_i_flags;
127ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 blocks_per_ind;
128ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 blocks_per_dind;
129ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	u32 blocks_per_tind;
130ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross};
131ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
132ec0a2e83dc66d67addeb90e83144187691852a3eColin Crossextern struct fs_info info;
133ec0a2e83dc66d67addeb90e83144187691852a3eColin Crossextern struct fs_aux_info aux_info;
134ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
1352ae7663e1e064000356ee9e49ebd08bdddc545f9Ken Sumrallextern jmp_buf setjmp_env;
1362ae7663e1e064000356ee9e49ebd08bdddc545f9Ken Sumrall
137ec0a2e83dc66d67addeb90e83144187691852a3eColin Crossstatic inline int log_2(int j)
138ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross{
139ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	int i;
140ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
141ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	for (i = 0; j > 0; i++)
142ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross		j >>= 1;
143ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
144ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross	return i - 1;
145ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross}
146ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross
147881cca2f88ddcce86483b3ba95546b5641de8c0eColin Crossint ext4_bg_has_super_block(int bg);
148c2470654d4b4db09a7052fc5fa108ac21f1b1948Colin Crossvoid write_ext4_image(const char *filename, int gz, int sparse, int crc,
149c2470654d4b4db09a7052fc5fa108ac21f1b1948Colin Cross		int wipe);
150881cca2f88ddcce86483b3ba95546b5641de8c0eColin Crossvoid ext4_create_fs_aux_info(void);
151881cca2f88ddcce86483b3ba95546b5641de8c0eColin Crossvoid ext4_free_fs_aux_info(void);
152881cca2f88ddcce86483b3ba95546b5641de8c0eColin Crossvoid ext4_fill_in_sb(void);
153881cca2f88ddcce86483b3ba95546b5641de8c0eColin Crossvoid ext4_create_resize_inode(void);
154881cca2f88ddcce86483b3ba95546b5641de8c0eColin Crossvoid ext4_create_journal_inode(void);
155881cca2f88ddcce86483b3ba95546b5641de8c0eColin Crossvoid ext4_update_free(void);
156b781330b1acae2e5706bbda8d81e5f7575f40e2aColin Crossvoid ext4_queue_sb(void);
157881cca2f88ddcce86483b3ba95546b5641de8c0eColin Crossu64 get_file_size(const char *filename);
158881cca2f88ddcce86483b3ba95546b5641de8c0eColin Crossu64 parse_num(const char *arg);
159a7ed433f2dc0116627a93b18fbb260f0665ca0cbColin Crossvoid ext4_parse_sb(struct ext4_super_block *sb);
160881cca2f88ddcce86483b3ba95546b5641de8c0eColin Cross
161ec0a2e83dc66d67addeb90e83144187691852a3eColin Cross#endif
162