1 /*
2  * Create a squashfs filesystem.  This is a highly compressed read only
3  * filesystem.
4  *
5  * Copyright (c) 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011,
6  * 2012, 2013, 2014
7  * Phillip Lougher <phillip@squashfs.org.uk>
8  *
9  * This program is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU General Public License
11  * as published by the Free Software Foundation; either version 2,
12  * or (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
22  *
23  * mksquashfs.c
24  */
25 
26 #define FALSE 0
27 #define TRUE 1
28 #define MAX_LINE 16384
29 
30 #include <pwd.h>
31 #include <grp.h>
32 #include <time.h>
33 #include <unistd.h>
34 #include <stdio.h>
35 #include <stddef.h>
36 #include <sys/types.h>
37 #include <sys/stat.h>
38 #include <fcntl.h>
39 #include <errno.h>
40 #include <dirent.h>
41 #include <string.h>
42 #include <stdlib.h>
43 #include <signal.h>
44 #include <setjmp.h>
45 #include <sys/types.h>
46 #include <sys/mman.h>
47 #include <pthread.h>
48 #include <regex.h>
49 #include <fnmatch.h>
50 #include <sys/wait.h>
51 #include <limits.h>
52 #include <ctype.h>
53 
54 #ifndef linux
55 #define __BYTE_ORDER BYTE_ORDER
56 #define __BIG_ENDIAN BIG_ENDIAN
57 #define __LITTLE_ENDIAN LITTLE_ENDIAN
58 #include <sys/sysctl.h>
59 #else
60 #include <endian.h>
61 #include <sys/sysinfo.h>
62 #endif
63 
64 #include "squashfs_fs.h"
65 #include "squashfs_swap.h"
66 #include "mksquashfs.h"
67 #include "sort.h"
68 #include "pseudo.h"
69 #include "compressor.h"
70 #include "xattr.h"
71 #include "action.h"
72 #include "error.h"
73 #include "progressbar.h"
74 #include "info.h"
75 #include "caches-queues-lists.h"
76 #include "read_fs.h"
77 #include "restore.h"
78 #include "process_fragments.h"
79 
80 /* ANDROID CHANGES START*/
81 #ifdef ANDROID
82 #include "android.h"
83 int android_config = FALSE;
84 char *context_file = NULL;
85 char *mount_point = NULL;
86 char *target_out_path = NULL;
87 #endif
88 /* ANDROID CHANGES END */
89 
90 int delete = FALSE;
91 int fd;
92 struct squashfs_super_block sBlk;
93 
94 /* filesystem flags for building */
95 int comp_opts = FALSE;
96 int no_xattrs = XATTR_DEF;
97 int noX = FALSE;
98 int duplicate_checking = TRUE;
99 int noF = FALSE;
100 int no_fragments = FALSE;
101 int always_use_fragments = FALSE;
102 int noI = FALSE;
103 int noD = FALSE;
104 int silent = TRUE;
105 int exportable = TRUE;
106 int sparse_files = TRUE;
107 int old_exclude = TRUE;
108 int use_regex = FALSE;
109 int nopad = FALSE;
110 int exit_on_error = FALSE;
111 
112 long long global_uid = -1, global_gid = -1;
113 
114 /* superblock attributes */
115 int block_size = SQUASHFS_FILE_SIZE, block_log;
116 unsigned int id_count = 0;
117 int file_count = 0, sym_count = 0, dev_count = 0, dir_count = 0, fifo_count = 0,
118 	sock_count = 0;
119 
120 /* write position within data section */
121 long long bytes = 0, total_bytes = 0;
122 
123 /* in memory directory table - possibly compressed */
124 char *directory_table = NULL;
125 unsigned int directory_bytes = 0, directory_size = 0, total_directory_bytes = 0;
126 
127 /* cached directory table */
128 char *directory_data_cache = NULL;
129 unsigned int directory_cache_bytes = 0, directory_cache_size = 0;
130 
131 /* in memory inode table - possibly compressed */
132 char *inode_table = NULL;
133 unsigned int inode_bytes = 0, inode_size = 0, total_inode_bytes = 0;
134 
135 /* cached inode table */
136 char *data_cache = NULL;
137 unsigned int cache_bytes = 0, cache_size = 0, inode_count = 0;
138 
139 /* inode lookup table */
140 squashfs_inode *inode_lookup_table = NULL;
141 
142 /* in memory directory data */
143 #define I_COUNT_SIZE		128
144 #define DIR_ENTRIES		32
145 #define INODE_HASH_SIZE		65536
146 #define INODE_HASH_MASK		(INODE_HASH_SIZE - 1)
147 #define INODE_HASH(dev, ino)	(ino & INODE_HASH_MASK)
148 
149 struct cached_dir_index {
150 	struct squashfs_dir_index	index;
151 	char				*name;
152 };
153 
154 struct directory {
155 	unsigned int		start_block;
156 	unsigned int		size;
157 	unsigned char		*buff;
158 	unsigned char		*p;
159 	unsigned int		entry_count;
160 	unsigned char		*entry_count_p;
161 	unsigned int		i_count;
162 	unsigned int		i_size;
163 	struct cached_dir_index	*index;
164 	unsigned char		*index_count_p;
165 	unsigned int		inode_number;
166 };
167 
168 struct inode_info *inode_info[INODE_HASH_SIZE];
169 
170 /* hash tables used to do fast duplicate searches in duplicate check */
171 struct file_info *dupl[65536];
172 int dup_files = 0;
173 
174 /* exclude file handling */
175 /* list of exclude dirs/files */
176 struct exclude_info {
177 	dev_t			st_dev;
178 	ino_t			st_ino;
179 };
180 
181 #define EXCLUDE_SIZE 8192
182 int exclude = 0;
183 struct exclude_info *exclude_paths = NULL;
184 int old_excluded(char *filename, struct stat *buf);
185 
186 struct path_entry {
187 	char *name;
188 	regex_t *preg;
189 	struct pathname *paths;
190 };
191 
192 struct pathname {
193 	int names;
194 	struct path_entry *name;
195 };
196 
197 struct pathnames {
198 	int count;
199 	struct pathname *path[0];
200 };
201 #define PATHS_ALLOC_SIZE 10
202 
203 struct pathnames *paths = NULL;
204 struct pathname *path = NULL;
205 struct pathname *stickypath = NULL;
206 int excluded(char *name, struct pathnames *paths, struct pathnames **new);
207 
208 int fragments = 0;
209 
210 #define FRAG_SIZE 32768
211 
212 struct squashfs_fragment_entry *fragment_table = NULL;
213 int fragments_outstanding = 0;
214 
215 int fragments_locked = FALSE;
216 
217 /* current inode number for directories and non directories */
218 unsigned int inode_no = 1;
219 unsigned int root_inode_number = 0;
220 
221 /* list of source dirs/files */
222 int source = 0;
223 char **source_path;
224 
225 /* list of root directory entries read from original filesystem */
226 int old_root_entries = 0;
227 struct old_root_entry_info {
228 	char			*name;
229 	struct inode_info	inode;
230 };
231 struct old_root_entry_info *old_root_entry;
232 
233 /* restore orignal filesystem state if appending to existing filesystem is
234  * cancelled */
235 int appending = FALSE;
236 char *sdata_cache, *sdirectory_data_cache, *sdirectory_compressed;
237 
238 long long sbytes, stotal_bytes;
239 
240 unsigned int sinode_bytes, scache_bytes, sdirectory_bytes,
241 	sdirectory_cache_bytes, sdirectory_compressed_bytes,
242 	stotal_inode_bytes, stotal_directory_bytes,
243 	sinode_count = 0, sfile_count, ssym_count, sdev_count,
244 	sdir_count, sfifo_count, ssock_count, sdup_files;
245 int sfragments;
246 int threads;
247 
248 /* flag whether destination file is a block device */
249 int block_device = FALSE;
250 
251 /* flag indicating whether files are sorted using sort list(s) */
252 int sorted = FALSE;
253 
254 /* save destination file name for deleting on error */
255 char *destination_file = NULL;
256 
257 /* recovery file for abnormal exit on appending */
258 char *recovery_file = NULL;
259 int recover = TRUE;
260 
261 struct id *id_hash_table[ID_ENTRIES];
262 struct id *id_table[SQUASHFS_IDS], *sid_table[SQUASHFS_IDS];
263 unsigned int uid_count = 0, guid_count = 0;
264 unsigned int sid_count = 0, suid_count = 0, sguid_count = 0;
265 
266 struct cache *reader_buffer, *fragment_buffer, *reserve_cache;
267 struct cache *bwriter_buffer, *fwriter_buffer;
268 struct queue *to_reader, *to_deflate, *to_writer, *from_writer,
269 	*to_frag, *locked_fragment, *to_process_frag;
270 struct seq_queue *to_main;
271 pthread_t reader_thread, writer_thread, main_thread;
272 pthread_t *deflator_thread, *frag_deflator_thread, *frag_thread;
273 pthread_t *restore_thread = NULL;
274 pthread_mutex_t	fragment_mutex = PTHREAD_MUTEX_INITIALIZER;
275 pthread_mutex_t	pos_mutex = PTHREAD_MUTEX_INITIALIZER;
276 pthread_mutex_t	dup_mutex = PTHREAD_MUTEX_INITIALIZER;
277 
278 /* user options that control parallelisation */
279 int processors = -1;
280 int bwriter_size;
281 
282 /* compression operations */
283 struct compressor *comp = NULL;
284 int compressor_opt_parsed = FALSE;
285 void *stream = NULL;
286 
287 /* xattr stats */
288 unsigned int xattr_bytes = 0, total_xattr_bytes = 0;
289 
290 /* fragment to file mapping used when appending */
291 int append_fragments = 0;
292 struct append_file **file_mapping;
293 
294 /* root of the in-core directory structure */
295 struct dir_info *root_dir;
296 
297 static char *read_from_disk(long long start, unsigned int avail_bytes);
298 void add_old_root_entry(char *name, squashfs_inode inode, int inode_number,
299 	int type);
300 struct file_info *duplicate(long long file_size, long long bytes,
301 	unsigned int **block_list, long long *start, struct fragment **fragment,
302 	struct file_buffer *file_buffer, int blocks, unsigned short checksum,
303 	int checksum_flag);
304 struct dir_info *dir_scan1(char *, char *, struct pathnames *,
305 	struct dir_ent *(_readdir)(struct dir_info *), int);
306 void dir_scan2(struct dir_info *dir, struct pseudo *pseudo);
307 void dir_scan3(struct dir_info *dir);
308 void dir_scan4(struct dir_info *dir);
309 void dir_scan5(struct dir_info *dir);
310 void dir_scan6(struct dir_info *dir);
311 void dir_scan7(squashfs_inode *inode, struct dir_info *dir_info);
312 struct file_info *add_non_dup(long long file_size, long long bytes,
313 	unsigned int *block_list, long long start, struct fragment *fragment,
314 	unsigned short checksum, unsigned short fragment_checksum,
315 	int checksum_flag, int checksum_frag_flag);
316 long long generic_write_table(int, void *, int, void *, int);
317 void restorefs();
318 struct dir_info *scan1_opendir(char *pathname, char *subpath, int depth);
319 void write_filesystem_tables(struct squashfs_super_block *sBlk, int nopad);
320 unsigned short get_checksum_mem(char *buff, int bytes);
321 void check_usable_phys_mem(int total_mem);
322 
323 
prep_exit()324 void prep_exit()
325 {
326 	if(restore_thread) {
327 		if(pthread_self() == *restore_thread) {
328 			/*
329 			 * Recursive failure when trying to restore filesystem!
330 			 * Nothing to do except to exit, otherwise we'll just
331 			 * appear to hang.  The user should be able to restore
332 			 * from the recovery file (which is why it was added, in
333 			 * case of catastrophic failure in Mksquashfs)
334 			 */
335 			exit(1);
336 		} else {
337 			/* signal the restore thread to restore */
338 			pthread_kill(*restore_thread, SIGUSR1);
339 			pthread_exit(NULL);
340 		}
341 	} else if(delete) {
342 		if(destination_file && !block_device)
343 			unlink(destination_file);
344 	} else if(recovery_file)
345 		unlink(recovery_file);
346 }
347 
348 
add_overflow(int a,int b)349 int add_overflow(int a, int b)
350 {
351 	return (INT_MAX - a) < b;
352 }
353 
354 
shift_overflow(int a,int shift)355 int shift_overflow(int a, int shift)
356 {
357 	return (INT_MAX >> shift) < a;
358 }
359 
360 
multiply_overflow(int a,int multiplier)361 int multiply_overflow(int a, int multiplier)
362 {
363 	return (INT_MAX / multiplier) < a;
364 }
365 
366 
multiply_overflowll(long long a,int multiplier)367 int multiply_overflowll(long long a, int multiplier)
368 {
369 	return (LLONG_MAX / multiplier) < a;
370 }
371 
372 
373 #define MKINODE(A)	((squashfs_inode)(((squashfs_inode) inode_bytes << 16) \
374 			+ (((char *)A) - data_cache)))
375 
376 
restorefs()377 void restorefs()
378 {
379 	ERROR("Exiting - restoring original filesystem!\n\n");
380 
381 	bytes = sbytes;
382 	memcpy(data_cache, sdata_cache, cache_bytes = scache_bytes);
383 	memcpy(directory_data_cache, sdirectory_data_cache,
384 		sdirectory_cache_bytes);
385 	directory_cache_bytes = sdirectory_cache_bytes;
386 	inode_bytes = sinode_bytes;
387 	directory_bytes = sdirectory_bytes;
388  	memcpy(directory_table + directory_bytes, sdirectory_compressed,
389 		sdirectory_compressed_bytes);
390  	directory_bytes += sdirectory_compressed_bytes;
391 	total_bytes = stotal_bytes;
392 	total_inode_bytes = stotal_inode_bytes;
393 	total_directory_bytes = stotal_directory_bytes;
394 	inode_count = sinode_count;
395 	file_count = sfile_count;
396 	sym_count = ssym_count;
397 	dev_count = sdev_count;
398 	dir_count = sdir_count;
399 	fifo_count = sfifo_count;
400 	sock_count = ssock_count;
401 	dup_files = sdup_files;
402 	fragments = sfragments;
403 	id_count = sid_count;
404 	restore_xattrs();
405 	write_filesystem_tables(&sBlk, nopad);
406 	exit(1);
407 }
408 
409 
sighandler()410 void sighandler()
411 {
412 	EXIT_MKSQUASHFS();
413 }
414 
415 
mangle2(void * strm,char * d,char * s,int size,int block_size,int uncompressed,int data_block)416 int mangle2(void *strm, char *d, char *s, int size,
417 	int block_size, int uncompressed, int data_block)
418 {
419 	int error, c_byte = 0;
420 
421 	if(!uncompressed) {
422 		c_byte = compressor_compress(comp, strm, d, s, size, block_size,
423 			 &error);
424 		if(c_byte == -1)
425 			BAD_ERROR("mangle2:: %s compress failed with error "
426 				"code %d\n", comp->name, error);
427 	}
428 
429 	if(c_byte == 0 || c_byte >= size) {
430 		memcpy(d, s, size);
431 		return size | (data_block ? SQUASHFS_COMPRESSED_BIT_BLOCK :
432 			SQUASHFS_COMPRESSED_BIT);
433 	}
434 
435 	return c_byte;
436 }
437 
438 
mangle(char * d,char * s,int size,int block_size,int uncompressed,int data_block)439 int mangle(char *d, char *s, int size, int block_size,
440 	int uncompressed, int data_block)
441 {
442 	return mangle2(stream, d, s, size, block_size, uncompressed,
443 		data_block);
444 }
445 
446 
get_inode(int req_size)447 void *get_inode(int req_size)
448 {
449 	int data_space;
450 	unsigned short c_byte;
451 
452 	while(cache_bytes >= SQUASHFS_METADATA_SIZE) {
453 		if((inode_size - inode_bytes) <
454 				((SQUASHFS_METADATA_SIZE << 1)) + 2) {
455 			void *it = realloc(inode_table, inode_size +
456 				(SQUASHFS_METADATA_SIZE << 1) + 2);
457 			if(it == NULL)
458 				MEM_ERROR();
459 			inode_table = it;
460 			inode_size += (SQUASHFS_METADATA_SIZE << 1) + 2;
461 		}
462 
463 		c_byte = mangle(inode_table + inode_bytes + BLOCK_OFFSET,
464 			data_cache, SQUASHFS_METADATA_SIZE,
465 			SQUASHFS_METADATA_SIZE, noI, 0);
466 		TRACE("Inode block @ 0x%x, size %d\n", inode_bytes, c_byte);
467 		SQUASHFS_SWAP_SHORTS(&c_byte, inode_table + inode_bytes, 1);
468 		inode_bytes += SQUASHFS_COMPRESSED_SIZE(c_byte) + BLOCK_OFFSET;
469 		total_inode_bytes += SQUASHFS_METADATA_SIZE + BLOCK_OFFSET;
470 		memmove(data_cache, data_cache + SQUASHFS_METADATA_SIZE,
471 			cache_bytes - SQUASHFS_METADATA_SIZE);
472 		cache_bytes -= SQUASHFS_METADATA_SIZE;
473 	}
474 
475 	data_space = (cache_size - cache_bytes);
476 	if(data_space < req_size) {
477 			int realloc_size = cache_size == 0 ?
478 				((req_size + SQUASHFS_METADATA_SIZE) &
479 				~(SQUASHFS_METADATA_SIZE - 1)) : req_size -
480 				data_space;
481 
482 			void *dc = realloc(data_cache, cache_size +
483 				realloc_size);
484 			if(dc == NULL)
485 				MEM_ERROR();
486 			cache_size += realloc_size;
487 			data_cache = dc;
488 	}
489 
490 	cache_bytes += req_size;
491 
492 	return data_cache + cache_bytes - req_size;
493 }
494 
495 
read_bytes(int fd,void * buff,int bytes)496 int read_bytes(int fd, void *buff, int bytes)
497 {
498 	int res, count;
499 
500 	for(count = 0; count < bytes; count += res) {
501 		res = read(fd, buff + count, bytes - count);
502 		if(res < 1) {
503 			if(res == 0)
504 				goto bytes_read;
505 			else if(errno != EINTR) {
506 				ERROR("Read failed because %s\n",
507 						strerror(errno));
508 				return -1;
509 			} else
510 				res = 0;
511 		}
512 	}
513 
514 bytes_read:
515 	return count;
516 }
517 
518 
read_fs_bytes(int fd,long long byte,int bytes,void * buff)519 int read_fs_bytes(int fd, long long byte, int bytes, void *buff)
520 {
521 	off_t off = byte;
522 	int res = 1;
523 
524 	TRACE("read_fs_bytes: reading from position 0x%llx, bytes %d\n",
525 		byte, bytes);
526 
527 	pthread_cleanup_push((void *) pthread_mutex_unlock, &pos_mutex);
528 	pthread_mutex_lock(&pos_mutex);
529 	if(lseek(fd, off, SEEK_SET) == -1) {
530 		ERROR("read_fs_bytes: Lseek on destination failed because %s, "
531 			"offset=0x%llx\n", strerror(errno), off);
532 		res = 0;
533 	} else if(read_bytes(fd, buff, bytes) < bytes) {
534 		ERROR("Read on destination failed\n");
535 		res = 0;
536 	}
537 
538 	pthread_cleanup_pop(1);
539 	return res;
540 }
541 
542 
write_bytes(int fd,void * buff,int bytes)543 int write_bytes(int fd, void *buff, int bytes)
544 {
545 	int res, count;
546 
547 	for(count = 0; count < bytes; count += res) {
548 		res = write(fd, buff + count, bytes - count);
549 		if(res == -1) {
550 			if(errno != EINTR) {
551 				ERROR("Write failed because %s\n",
552 						strerror(errno));
553 				return -1;
554 			}
555 			res = 0;
556 		}
557 	}
558 
559 	return 0;
560 }
561 
562 
write_destination(int fd,long long byte,int bytes,void * buff)563 void write_destination(int fd, long long byte, int bytes, void *buff)
564 {
565 	off_t off = byte;
566 
567 	pthread_cleanup_push((void *) pthread_mutex_unlock, &pos_mutex);
568 	pthread_mutex_lock(&pos_mutex);
569 
570 	if(lseek(fd, off, SEEK_SET) == -1) {
571 		ERROR("write_destination: Lseek on destination "
572 			"failed because %s, offset=0x%llx\n", strerror(errno),
573 			off);
574 		BAD_ERROR("Probably out of space on output %s\n",
575 			block_device ? "block device" : "filesystem");
576 	}
577 
578 	if(write_bytes(fd, buff, bytes) == -1)
579 		BAD_ERROR("Failed to write to output %s\n",
580 			block_device ? "block device" : "filesystem");
581 
582 	pthread_cleanup_pop(1);
583 }
584 
585 
write_inodes()586 long long write_inodes()
587 {
588 	unsigned short c_byte;
589 	int avail_bytes;
590 	char *datap = data_cache;
591 	long long start_bytes = bytes;
592 
593 	while(cache_bytes) {
594 		if(inode_size - inode_bytes <
595 				((SQUASHFS_METADATA_SIZE << 1) + 2)) {
596 			void *it = realloc(inode_table, inode_size +
597 				((SQUASHFS_METADATA_SIZE << 1) + 2));
598 			if(it == NULL)
599 				MEM_ERROR();
600 			inode_size += (SQUASHFS_METADATA_SIZE << 1) + 2;
601 			inode_table = it;
602 		}
603 		avail_bytes = cache_bytes > SQUASHFS_METADATA_SIZE ?
604 			SQUASHFS_METADATA_SIZE : cache_bytes;
605 		c_byte = mangle(inode_table + inode_bytes + BLOCK_OFFSET, datap,
606 			avail_bytes, SQUASHFS_METADATA_SIZE, noI, 0);
607 		TRACE("Inode block @ 0x%x, size %d\n", inode_bytes, c_byte);
608 		SQUASHFS_SWAP_SHORTS(&c_byte, inode_table + inode_bytes, 1);
609 		inode_bytes += SQUASHFS_COMPRESSED_SIZE(c_byte) + BLOCK_OFFSET;
610 		total_inode_bytes += avail_bytes + BLOCK_OFFSET;
611 		datap += avail_bytes;
612 		cache_bytes -= avail_bytes;
613 	}
614 
615 	write_destination(fd, bytes, inode_bytes,  inode_table);
616 	bytes += inode_bytes;
617 
618 	return start_bytes;
619 }
620 
621 
write_directories()622 long long write_directories()
623 {
624 	unsigned short c_byte;
625 	int avail_bytes;
626 	char *directoryp = directory_data_cache;
627 	long long start_bytes = bytes;
628 
629 	while(directory_cache_bytes) {
630 		if(directory_size - directory_bytes <
631 				((SQUASHFS_METADATA_SIZE << 1) + 2)) {
632 			void *dt = realloc(directory_table,
633 				directory_size + ((SQUASHFS_METADATA_SIZE << 1)
634 				+ 2));
635 			if(dt == NULL)
636 				MEM_ERROR();
637 			directory_size += (SQUASHFS_METADATA_SIZE << 1) + 2;
638 			directory_table = dt;
639 		}
640 		avail_bytes = directory_cache_bytes > SQUASHFS_METADATA_SIZE ?
641 			SQUASHFS_METADATA_SIZE : directory_cache_bytes;
642 		c_byte = mangle(directory_table + directory_bytes +
643 			BLOCK_OFFSET, directoryp, avail_bytes,
644 			SQUASHFS_METADATA_SIZE, noI, 0);
645 		TRACE("Directory block @ 0x%x, size %d\n", directory_bytes,
646 			c_byte);
647 		SQUASHFS_SWAP_SHORTS(&c_byte,
648 			directory_table + directory_bytes, 1);
649 		directory_bytes += SQUASHFS_COMPRESSED_SIZE(c_byte) +
650 			BLOCK_OFFSET;
651 		total_directory_bytes += avail_bytes + BLOCK_OFFSET;
652 		directoryp += avail_bytes;
653 		directory_cache_bytes -= avail_bytes;
654 	}
655 	write_destination(fd, bytes, directory_bytes, directory_table);
656 	bytes += directory_bytes;
657 
658 	return start_bytes;
659 }
660 
661 
write_id_table()662 long long write_id_table()
663 {
664 	unsigned int id_bytes = SQUASHFS_ID_BYTES(id_count);
665 	unsigned int p[id_count];
666 	int i;
667 
668 	TRACE("write_id_table: ids %d, id_bytes %d\n", id_count, id_bytes);
669 	for(i = 0; i < id_count; i++) {
670 		TRACE("write_id_table: id index %d, id %d", i, id_table[i]->id);
671 		SQUASHFS_SWAP_INTS(&id_table[i]->id, p + i, 1);
672 	}
673 
674 	return generic_write_table(id_bytes, p, 0, NULL, noI);
675 }
676 
677 
get_id(unsigned int id)678 struct id *get_id(unsigned int id)
679 {
680 	int hash = ID_HASH(id);
681 	struct id *entry = id_hash_table[hash];
682 
683 	for(; entry; entry = entry->next)
684 		if(entry->id == id)
685 			break;
686 
687 	return entry;
688 }
689 
690 
create_id(unsigned int id)691 struct id *create_id(unsigned int id)
692 {
693 	int hash = ID_HASH(id);
694 	struct id *entry = malloc(sizeof(struct id));
695 	if(entry == NULL)
696 		MEM_ERROR();
697 	entry->id = id;
698 	entry->index = id_count ++;
699 	entry->flags = 0;
700 	entry->next = id_hash_table[hash];
701 	id_hash_table[hash] = entry;
702 	id_table[entry->index] = entry;
703 	return entry;
704 }
705 
706 
get_uid(unsigned int uid)707 unsigned int get_uid(unsigned int uid)
708 {
709 	struct id *entry = get_id(uid);
710 
711 	if(entry == NULL) {
712 		if(id_count == SQUASHFS_IDS)
713 			BAD_ERROR("Out of uids!\n");
714 		entry = create_id(uid);
715 	}
716 
717 	if((entry->flags & ISA_UID) == 0) {
718 		entry->flags |= ISA_UID;
719 		uid_count ++;
720 	}
721 
722 	return entry->index;
723 }
724 
725 
get_guid(unsigned int guid)726 unsigned int get_guid(unsigned int guid)
727 {
728 	struct id *entry = get_id(guid);
729 
730 	if(entry == NULL) {
731 		if(id_count == SQUASHFS_IDS)
732 			BAD_ERROR("Out of gids!\n");
733 		entry = create_id(guid);
734 	}
735 
736 	if((entry->flags & ISA_GID) == 0) {
737 		entry->flags |= ISA_GID;
738 		guid_count ++;
739 	}
740 
741 	return entry->index;
742 }
743 
744 
745 #define ALLOC_SIZE 128
746 
_pathname(struct dir_ent * dir_ent,char * pathname,int * size)747 char *_pathname(struct dir_ent *dir_ent, char *pathname, int *size)
748 {
749 	if(pathname == NULL) {
750 		pathname = malloc(ALLOC_SIZE);
751 		if(pathname == NULL)
752 			MEM_ERROR();
753 	}
754 
755 	for(;;) {
756 		int res = snprintf(pathname, *size, "%s/%s",
757 			dir_ent->our_dir->pathname,
758 			dir_ent->source_name ? : dir_ent->name);
759 
760 		if(res < 0)
761 			BAD_ERROR("snprintf failed in pathname\n");
762 		else if(res >= *size) {
763 			/*
764 			 * pathname is too small to contain the result, so
765 			 * increase it and try again
766 			 */
767 			*size = (res + ALLOC_SIZE) & ~(ALLOC_SIZE - 1);
768 			pathname = realloc(pathname, *size);
769 			if(pathname == NULL)
770 				MEM_ERROR();
771 		} else
772 			break;
773 	}
774 
775 	return pathname;
776 }
777 
778 
pathname(struct dir_ent * dir_ent)779 char *pathname(struct dir_ent *dir_ent)
780 {
781 	static char *pathname = NULL;
782 	static int size = ALLOC_SIZE;
783 
784 	if (dir_ent->nonstandard_pathname)
785 		return dir_ent->nonstandard_pathname;
786 
787 	return pathname = _pathname(dir_ent, pathname, &size);
788 }
789 
790 
pathname_reader(struct dir_ent * dir_ent)791 char *pathname_reader(struct dir_ent *dir_ent)
792 {
793 	static char *pathname = NULL;
794 	static int size = ALLOC_SIZE;
795 
796 	if (dir_ent->nonstandard_pathname)
797 		return dir_ent->nonstandard_pathname;
798 
799 	return pathname = _pathname(dir_ent, pathname, &size);
800 }
801 
802 
subpathname(struct dir_ent * dir_ent)803 char *subpathname(struct dir_ent *dir_ent)
804 {
805 	static char *subpath = NULL;
806 	static int size = ALLOC_SIZE;
807 	int res;
808 
809 	if(subpath == NULL) {
810 		subpath = malloc(ALLOC_SIZE);
811 		if(subpath == NULL)
812 			MEM_ERROR();
813 	}
814 
815 	for(;;) {
816 		if(dir_ent->our_dir->subpath[0] != '\0')
817 			res = snprintf(subpath, size, "%s/%s",
818 				dir_ent->our_dir->subpath, dir_ent->name);
819 		else
820 			res = snprintf(subpath, size, "/%s", dir_ent->name);
821 
822 		if(res < 0)
823 			BAD_ERROR("snprintf failed in subpathname\n");
824 		else if(res >= size) {
825 			/*
826 			 * subpath is too small to contain the result, so
827 			 * increase it and try again
828 			 */
829 			size = (res + ALLOC_SIZE) & ~(ALLOC_SIZE - 1);
830 			subpath = realloc(subpath, size);
831 			if(subpath == NULL)
832 				MEM_ERROR();
833 		} else
834 			break;
835 	}
836 
837 	return subpath;
838 }
839 
840 
get_inode_no(struct inode_info * inode)841 inline unsigned int get_inode_no(struct inode_info *inode)
842 {
843 	return inode->inode_number;
844 }
845 
846 
get_parent_no(struct dir_info * dir)847 inline unsigned int get_parent_no(struct dir_info *dir)
848 {
849 	return dir->depth ? get_inode_no(dir->dir_ent->inode) : inode_no;
850 }
851 
852 
create_inode(squashfs_inode * i_no,struct dir_info * dir_info,struct dir_ent * dir_ent,int type,long long byte_size,long long start_block,unsigned int offset,unsigned int * block_list,struct fragment * fragment,struct directory * dir_in,long long sparse)853 int create_inode(squashfs_inode *i_no, struct dir_info *dir_info,
854 	struct dir_ent *dir_ent, int type, long long byte_size,
855 	long long start_block, unsigned int offset, unsigned int *block_list,
856 	struct fragment *fragment, struct directory *dir_in, long long sparse)
857 {
858 	struct stat *buf = &dir_ent->inode->buf;
859 	union squashfs_inode_header inode_header;
860 	struct squashfs_base_inode_header *base = &inode_header.base;
861 	void *inode;
862 	char *filename = pathname(dir_ent);
863 	int nlink = dir_ent->inode->nlink;
864 	int xattr = read_xattrs(dir_ent);
865 
866 	switch(type) {
867 	case SQUASHFS_FILE_TYPE:
868 		if(dir_ent->inode->nlink > 1 ||
869 				byte_size >= (1LL << 32) ||
870 				start_block >= (1LL << 32) ||
871 				sparse || IS_XATTR(xattr))
872 			type = SQUASHFS_LREG_TYPE;
873 		break;
874 	case SQUASHFS_DIR_TYPE:
875 		if(dir_info->dir_is_ldir || IS_XATTR(xattr))
876 			type = SQUASHFS_LDIR_TYPE;
877 		break;
878 	case SQUASHFS_SYMLINK_TYPE:
879 		if(IS_XATTR(xattr))
880 			type = SQUASHFS_LSYMLINK_TYPE;
881 		break;
882 	case SQUASHFS_BLKDEV_TYPE:
883 		if(IS_XATTR(xattr))
884 			type = SQUASHFS_LBLKDEV_TYPE;
885 		break;
886 	case SQUASHFS_CHRDEV_TYPE:
887 		if(IS_XATTR(xattr))
888 			type = SQUASHFS_LCHRDEV_TYPE;
889 		break;
890 	case SQUASHFS_FIFO_TYPE:
891 		if(IS_XATTR(xattr))
892 			type = SQUASHFS_LFIFO_TYPE;
893 		break;
894 	case SQUASHFS_SOCKET_TYPE:
895 		if(IS_XATTR(xattr))
896 			type = SQUASHFS_LSOCKET_TYPE;
897 		break;
898 	}
899 
900 	base->mode = SQUASHFS_MODE(buf->st_mode);
901 	base->uid = get_uid((unsigned int) global_uid == -1 ?
902 		buf->st_uid : global_uid);
903 	base->inode_type = type;
904 	base->guid = get_guid((unsigned int) global_gid == -1 ?
905 		buf->st_gid : global_gid);
906 	base->mtime = buf->st_mtime;
907 	base->inode_number = get_inode_no(dir_ent->inode);
908 
909 	if(type == SQUASHFS_FILE_TYPE) {
910 		int i;
911 		struct squashfs_reg_inode_header *reg = &inode_header.reg;
912 		size_t off = offsetof(struct squashfs_reg_inode_header, block_list);
913 
914 		inode = get_inode(sizeof(*reg) + offset * sizeof(unsigned int));
915 		reg->file_size = byte_size;
916 		reg->start_block = start_block;
917 		reg->fragment = fragment->index;
918 		reg->offset = fragment->offset;
919 		SQUASHFS_SWAP_REG_INODE_HEADER(reg, inode);
920 		SQUASHFS_SWAP_INTS(block_list, inode + off, offset);
921 		TRACE("File inode, file_size %lld, start_block 0x%llx, blocks "
922 			"%d, fragment %d, offset %d, size %d\n", byte_size,
923 			start_block, offset, fragment->index, fragment->offset,
924 			fragment->size);
925 		for(i = 0; i < offset; i++)
926 			TRACE("Block %d, size %d\n", i, block_list[i]);
927 	}
928 	else if(type == SQUASHFS_LREG_TYPE) {
929 		int i;
930 		struct squashfs_lreg_inode_header *reg = &inode_header.lreg;
931 		size_t off = offsetof(struct squashfs_lreg_inode_header, block_list);
932 
933 		inode = get_inode(sizeof(*reg) + offset * sizeof(unsigned int));
934 		reg->nlink = nlink;
935 		reg->file_size = byte_size;
936 		reg->start_block = start_block;
937 		reg->fragment = fragment->index;
938 		reg->offset = fragment->offset;
939 		if(sparse && sparse >= byte_size)
940 			sparse = byte_size - 1;
941 		reg->sparse = sparse;
942 		reg->xattr = xattr;
943 		SQUASHFS_SWAP_LREG_INODE_HEADER(reg, inode);
944 		SQUASHFS_SWAP_INTS(block_list, inode + off, offset);
945 		TRACE("Long file inode, file_size %lld, start_block 0x%llx, "
946 			"blocks %d, fragment %d, offset %d, size %d, nlink %d"
947 			"\n", byte_size, start_block, offset, fragment->index,
948 			fragment->offset, fragment->size, nlink);
949 		for(i = 0; i < offset; i++)
950 			TRACE("Block %d, size %d\n", i, block_list[i]);
951 	}
952 	else if(type == SQUASHFS_LDIR_TYPE) {
953 		int i;
954 		unsigned char *p;
955 		struct squashfs_ldir_inode_header *dir = &inode_header.ldir;
956 		struct cached_dir_index *index = dir_in->index;
957 		unsigned int i_count = dir_in->i_count;
958 		unsigned int i_size = dir_in->i_size;
959 
960 		if(byte_size >= 1 << 27)
961 			BAD_ERROR("directory greater than 2^27-1 bytes!\n");
962 
963 		inode = get_inode(sizeof(*dir) + i_size);
964 		dir->inode_type = SQUASHFS_LDIR_TYPE;
965 		dir->nlink = dir_ent->dir->directory_count + 2;
966 		dir->file_size = byte_size;
967 		dir->offset = offset;
968 		dir->start_block = start_block;
969 		dir->i_count = i_count;
970 		dir->parent_inode = get_parent_no(dir_ent->our_dir);
971 		dir->xattr = xattr;
972 
973 		SQUASHFS_SWAP_LDIR_INODE_HEADER(dir, inode);
974 		p = inode + offsetof(struct squashfs_ldir_inode_header, index);
975 		for(i = 0; i < i_count; i++) {
976 			SQUASHFS_SWAP_DIR_INDEX(&index[i].index, p);
977 			p += offsetof(struct squashfs_dir_index, name);
978 			memcpy(p, index[i].name, index[i].index.size + 1);
979 			p += index[i].index.size + 1;
980 		}
981 		TRACE("Long directory inode, file_size %lld, start_block "
982 			"0x%llx, offset 0x%x, nlink %d\n", byte_size,
983 			start_block, offset, dir_ent->dir->directory_count + 2);
984 	}
985 	else if(type == SQUASHFS_DIR_TYPE) {
986 		struct squashfs_dir_inode_header *dir = &inode_header.dir;
987 
988 		inode = get_inode(sizeof(*dir));
989 		dir->nlink = dir_ent->dir->directory_count + 2;
990 		dir->file_size = byte_size;
991 		dir->offset = offset;
992 		dir->start_block = start_block;
993 		dir->parent_inode = get_parent_no(dir_ent->our_dir);
994 		SQUASHFS_SWAP_DIR_INODE_HEADER(dir, inode);
995 		TRACE("Directory inode, file_size %lld, start_block 0x%llx, "
996 			"offset 0x%x, nlink %d\n", byte_size, start_block,
997 			offset, dir_ent->dir->directory_count + 2);
998 	}
999 	else if(type == SQUASHFS_CHRDEV_TYPE || type == SQUASHFS_BLKDEV_TYPE) {
1000 		struct squashfs_dev_inode_header *dev = &inode_header.dev;
1001 		unsigned int major = major(buf->st_rdev);
1002 		unsigned int minor = minor(buf->st_rdev);
1003 
1004 		if(major > 0xfff) {
1005 			ERROR("Major %d out of range in device node %s, "
1006 				"truncating to %d\n", major, filename,
1007 				major & 0xfff);
1008 			major &= 0xfff;
1009 		}
1010 		if(minor > 0xfffff) {
1011 			ERROR("Minor %d out of range in device node %s, "
1012 				"truncating to %d\n", minor, filename,
1013 				minor & 0xfffff);
1014 			minor &= 0xfffff;
1015 		}
1016 		inode = get_inode(sizeof(*dev));
1017 		dev->nlink = nlink;
1018 		dev->rdev = (major << 8) | (minor & 0xff) |
1019 				((minor & ~0xff) << 12);
1020 		SQUASHFS_SWAP_DEV_INODE_HEADER(dev, inode);
1021 		TRACE("Device inode, rdev 0x%x, nlink %d\n", dev->rdev, nlink);
1022 	}
1023 	else if(type == SQUASHFS_LCHRDEV_TYPE || type == SQUASHFS_LBLKDEV_TYPE) {
1024 		struct squashfs_ldev_inode_header *dev = &inode_header.ldev;
1025 		unsigned int major = major(buf->st_rdev);
1026 		unsigned int minor = minor(buf->st_rdev);
1027 
1028 		if(major > 0xfff) {
1029 			ERROR("Major %d out of range in device node %s, "
1030 				"truncating to %d\n", major, filename,
1031 				major & 0xfff);
1032 			major &= 0xfff;
1033 		}
1034 		if(minor > 0xfffff) {
1035 			ERROR("Minor %d out of range in device node %s, "
1036 				"truncating to %d\n", minor, filename,
1037 				minor & 0xfffff);
1038 			minor &= 0xfffff;
1039 		}
1040 		inode = get_inode(sizeof(*dev));
1041 		dev->nlink = nlink;
1042 		dev->rdev = (major << 8) | (minor & 0xff) |
1043 				((minor & ~0xff) << 12);
1044 		dev->xattr = xattr;
1045 		SQUASHFS_SWAP_LDEV_INODE_HEADER(dev, inode);
1046 		TRACE("Device inode, rdev 0x%x, nlink %d\n", dev->rdev, nlink);
1047 	}
1048 	else if(type == SQUASHFS_SYMLINK_TYPE) {
1049 		struct squashfs_symlink_inode_header *symlink = &inode_header.symlink;
1050 		int byte = strlen(dir_ent->inode->symlink);
1051 		size_t off = offsetof(struct squashfs_symlink_inode_header, symlink);
1052 
1053 		inode = get_inode(sizeof(*symlink) + byte);
1054 		symlink->nlink = nlink;
1055 		symlink->symlink_size = byte;
1056 		SQUASHFS_SWAP_SYMLINK_INODE_HEADER(symlink, inode);
1057 		strncpy(inode + off, dir_ent->inode->symlink, byte);
1058 		TRACE("Symbolic link inode, symlink_size %d, nlink %d\n", byte,
1059 			nlink);
1060 	}
1061 	else if(type == SQUASHFS_LSYMLINK_TYPE) {
1062 		struct squashfs_symlink_inode_header *symlink = &inode_header.symlink;
1063 		int byte = strlen(dir_ent->inode->symlink);
1064 		size_t off = offsetof(struct squashfs_symlink_inode_header, symlink);
1065 
1066 		inode = get_inode(sizeof(*symlink) + byte +
1067 						sizeof(unsigned int));
1068 		symlink->nlink = nlink;
1069 		symlink->symlink_size = byte;
1070 		SQUASHFS_SWAP_SYMLINK_INODE_HEADER(symlink, inode);
1071 		strncpy(inode + off, dir_ent->inode->symlink, byte);
1072 		SQUASHFS_SWAP_INTS(&xattr, inode + off + byte, 1);
1073 		TRACE("Symbolic link inode, symlink_size %d, nlink %d\n", byte,
1074 			nlink);
1075 	}
1076 	else if(type == SQUASHFS_FIFO_TYPE || type == SQUASHFS_SOCKET_TYPE) {
1077 		struct squashfs_ipc_inode_header *ipc = &inode_header.ipc;
1078 
1079 		inode = get_inode(sizeof(*ipc));
1080 		ipc->nlink = nlink;
1081 		SQUASHFS_SWAP_IPC_INODE_HEADER(ipc, inode);
1082 		TRACE("ipc inode, type %s, nlink %d\n", type ==
1083 			SQUASHFS_FIFO_TYPE ? "fifo" : "socket", nlink);
1084 	}
1085 	else if(type == SQUASHFS_LFIFO_TYPE || type == SQUASHFS_LSOCKET_TYPE) {
1086 		struct squashfs_lipc_inode_header *ipc = &inode_header.lipc;
1087 
1088 		inode = get_inode(sizeof(*ipc));
1089 		ipc->nlink = nlink;
1090 		ipc->xattr = xattr;
1091 		SQUASHFS_SWAP_LIPC_INODE_HEADER(ipc, inode);
1092 		TRACE("ipc inode, type %s, nlink %d\n", type ==
1093 			SQUASHFS_FIFO_TYPE ? "fifo" : "socket", nlink);
1094 	} else
1095 		BAD_ERROR("Unrecognised inode %d in create_inode\n", type);
1096 
1097 	*i_no = MKINODE(inode);
1098 	inode_count ++;
1099 
1100 	TRACE("Created inode 0x%llx, type %d, uid %d, guid %d\n", *i_no, type,
1101 		base->uid, base->guid);
1102 
1103 	return TRUE;
1104 }
1105 
1106 
add_dir(squashfs_inode inode,unsigned int inode_number,char * name,int type,struct directory * dir)1107 void add_dir(squashfs_inode inode, unsigned int inode_number, char *name,
1108 	int type, struct directory *dir)
1109 {
1110 	unsigned char *buff;
1111 	struct squashfs_dir_entry idir;
1112 	unsigned int start_block = inode >> 16;
1113 	unsigned int offset = inode & 0xffff;
1114 	unsigned int size = strlen(name);
1115 	size_t name_off = offsetof(struct squashfs_dir_entry, name);
1116 
1117 	if(size > SQUASHFS_NAME_LEN) {
1118 		size = SQUASHFS_NAME_LEN;
1119 		ERROR("Filename is greater than %d characters, truncating! ..."
1120 			"\n", SQUASHFS_NAME_LEN);
1121 	}
1122 
1123 	if(dir->p + sizeof(struct squashfs_dir_entry) + size +
1124 			sizeof(struct squashfs_dir_header)
1125 			>= dir->buff + dir->size) {
1126 		buff = realloc(dir->buff, dir->size += SQUASHFS_METADATA_SIZE);
1127 		if(buff == NULL)
1128 			MEM_ERROR();
1129 
1130 		dir->p = (dir->p - dir->buff) + buff;
1131 		if(dir->entry_count_p)
1132 			dir->entry_count_p = (dir->entry_count_p - dir->buff +
1133 			buff);
1134 		dir->index_count_p = dir->index_count_p - dir->buff + buff;
1135 		dir->buff = buff;
1136 	}
1137 
1138 	if(dir->entry_count == 256 || start_block != dir->start_block ||
1139 			((dir->entry_count_p != NULL) &&
1140 			((dir->p + sizeof(struct squashfs_dir_entry) + size -
1141 			dir->index_count_p) > SQUASHFS_METADATA_SIZE)) ||
1142 			((long long) inode_number - dir->inode_number) > 32767
1143 			|| ((long long) inode_number - dir->inode_number)
1144 			< -32768) {
1145 		if(dir->entry_count_p) {
1146 			struct squashfs_dir_header dir_header;
1147 
1148 			if((dir->p + sizeof(struct squashfs_dir_entry) + size -
1149 					dir->index_count_p) >
1150 					SQUASHFS_METADATA_SIZE) {
1151 				if(dir->i_count % I_COUNT_SIZE == 0) {
1152 					dir->index = realloc(dir->index,
1153 						(dir->i_count + I_COUNT_SIZE) *
1154 						sizeof(struct cached_dir_index));
1155 					if(dir->index == NULL)
1156 						MEM_ERROR();
1157 				}
1158 				dir->index[dir->i_count].index.index =
1159 					dir->p - dir->buff;
1160 				dir->index[dir->i_count].index.size = size - 1;
1161 				dir->index[dir->i_count++].name = name;
1162 				dir->i_size += sizeof(struct squashfs_dir_index)
1163 					+ size;
1164 				dir->index_count_p = dir->p;
1165 			}
1166 
1167 			dir_header.count = dir->entry_count - 1;
1168 			dir_header.start_block = dir->start_block;
1169 			dir_header.inode_number = dir->inode_number;
1170 			SQUASHFS_SWAP_DIR_HEADER(&dir_header,
1171 				dir->entry_count_p);
1172 
1173 		}
1174 
1175 
1176 		dir->entry_count_p = dir->p;
1177 		dir->start_block = start_block;
1178 		dir->entry_count = 0;
1179 		dir->inode_number = inode_number;
1180 		dir->p += sizeof(struct squashfs_dir_header);
1181 	}
1182 
1183 	idir.offset = offset;
1184 	idir.type = type;
1185 	idir.size = size - 1;
1186 	idir.inode_number = ((long long) inode_number - dir->inode_number);
1187 	SQUASHFS_SWAP_DIR_ENTRY(&idir, dir->p);
1188 	strncpy((char *) dir->p + name_off, name, size);
1189 	dir->p += sizeof(struct squashfs_dir_entry) + size;
1190 	dir->entry_count ++;
1191 }
1192 
1193 
write_dir(squashfs_inode * inode,struct dir_info * dir_info,struct directory * dir)1194 void write_dir(squashfs_inode *inode, struct dir_info *dir_info,
1195 	struct directory *dir)
1196 {
1197 	unsigned int dir_size = dir->p - dir->buff;
1198 	int data_space = directory_cache_size - directory_cache_bytes;
1199 	unsigned int directory_block, directory_offset, i_count, index;
1200 	unsigned short c_byte;
1201 
1202 	if(data_space < dir_size) {
1203 		int realloc_size = directory_cache_size == 0 ?
1204 			((dir_size + SQUASHFS_METADATA_SIZE) &
1205 			~(SQUASHFS_METADATA_SIZE - 1)) : dir_size - data_space;
1206 
1207 		void *dc = realloc(directory_data_cache,
1208 			directory_cache_size + realloc_size);
1209 		if(dc == NULL)
1210 			MEM_ERROR();
1211 		directory_cache_size += realloc_size;
1212 		directory_data_cache = dc;
1213 	}
1214 
1215 	if(dir_size) {
1216 		struct squashfs_dir_header dir_header;
1217 
1218 		dir_header.count = dir->entry_count - 1;
1219 		dir_header.start_block = dir->start_block;
1220 		dir_header.inode_number = dir->inode_number;
1221 		SQUASHFS_SWAP_DIR_HEADER(&dir_header, dir->entry_count_p);
1222 		memcpy(directory_data_cache + directory_cache_bytes, dir->buff,
1223 			dir_size);
1224 	}
1225 	directory_offset = directory_cache_bytes;
1226 	directory_block = directory_bytes;
1227 	directory_cache_bytes += dir_size;
1228 	i_count = 0;
1229 	index = SQUASHFS_METADATA_SIZE - directory_offset;
1230 
1231 	while(1) {
1232 		while(i_count < dir->i_count &&
1233 				dir->index[i_count].index.index < index)
1234 			dir->index[i_count++].index.start_block =
1235 				directory_bytes;
1236 		index += SQUASHFS_METADATA_SIZE;
1237 
1238 		if(directory_cache_bytes < SQUASHFS_METADATA_SIZE)
1239 			break;
1240 
1241 		if((directory_size - directory_bytes) <
1242 					((SQUASHFS_METADATA_SIZE << 1) + 2)) {
1243 			void *dt = realloc(directory_table,
1244 				directory_size + (SQUASHFS_METADATA_SIZE << 1)
1245 				+ 2);
1246 			if(dt == NULL)
1247 				MEM_ERROR();
1248 			directory_size += SQUASHFS_METADATA_SIZE << 1;
1249 			directory_table = dt;
1250 		}
1251 
1252 		c_byte = mangle(directory_table + directory_bytes +
1253 				BLOCK_OFFSET, directory_data_cache,
1254 				SQUASHFS_METADATA_SIZE, SQUASHFS_METADATA_SIZE,
1255 				noI, 0);
1256 		TRACE("Directory block @ 0x%x, size %d\n", directory_bytes,
1257 			c_byte);
1258 		SQUASHFS_SWAP_SHORTS(&c_byte,
1259 			directory_table + directory_bytes, 1);
1260 		directory_bytes += SQUASHFS_COMPRESSED_SIZE(c_byte) +
1261 			BLOCK_OFFSET;
1262 		total_directory_bytes += SQUASHFS_METADATA_SIZE + BLOCK_OFFSET;
1263 		memmove(directory_data_cache, directory_data_cache +
1264 			SQUASHFS_METADATA_SIZE, directory_cache_bytes -
1265 			SQUASHFS_METADATA_SIZE);
1266 		directory_cache_bytes -= SQUASHFS_METADATA_SIZE;
1267 	}
1268 
1269 	create_inode(inode, dir_info, dir_info->dir_ent, SQUASHFS_DIR_TYPE,
1270 		dir_size + 3, directory_block, directory_offset, NULL, NULL,
1271 		dir, 0);
1272 
1273 #ifdef SQUASHFS_TRACE
1274 	{
1275 		unsigned char *dirp;
1276 		int count;
1277 
1278 		TRACE("Directory contents of inode 0x%llx\n", *inode);
1279 		dirp = dir->buff;
1280 		while(dirp < dir->p) {
1281 			char buffer[SQUASHFS_NAME_LEN + 1];
1282 			struct squashfs_dir_entry idir, *idirp;
1283 			struct squashfs_dir_header dirh;
1284 			SQUASHFS_SWAP_DIR_HEADER((struct squashfs_dir_header *) dirp,
1285 				&dirh);
1286 			count = dirh.count + 1;
1287 			dirp += sizeof(struct squashfs_dir_header);
1288 
1289 			TRACE("\tStart block 0x%x, count %d\n",
1290 				dirh.start_block, count);
1291 
1292 			while(count--) {
1293 				idirp = (struct squashfs_dir_entry *) dirp;
1294 				SQUASHFS_SWAP_DIR_ENTRY(idirp, &idir);
1295 				strncpy(buffer, idirp->name, idir.size + 1);
1296 				buffer[idir.size + 1] = '\0';
1297 				TRACE("\t\tname %s, inode offset 0x%x, type "
1298 					"%d\n", buffer, idir.offset, idir.type);
1299 				dirp += sizeof(struct squashfs_dir_entry) + idir.size +
1300 					1;
1301 			}
1302 		}
1303 	}
1304 #endif
1305 	dir_count ++;
1306 }
1307 
1308 
get_fragment(struct fragment * fragment)1309 static struct file_buffer *get_fragment(struct fragment *fragment)
1310 {
1311 	struct squashfs_fragment_entry *disk_fragment;
1312 	struct file_buffer *buffer, *compressed_buffer;
1313 	long long start_block;
1314 	int res, size, index = fragment->index;
1315 	char locked;
1316 
1317 	/*
1318 	 * Lookup fragment block in cache.
1319 	 * If the fragment block doesn't exist, then get the compressed version
1320 	 * from the writer cache or off disk, and decompress it.
1321 	 *
1322 	 * This routine has two things which complicate the code:
1323 	 *
1324 	 *	1. Multiple threads can simultaneously lookup/create the
1325 	 *	   same buffer.  This means a buffer needs to be "locked"
1326 	 *	   when it is being filled in, to prevent other threads from
1327 	 *	   using it when it is not ready.  This is because we now do
1328 	 *	   fragment duplicate checking in parallel.
1329 	 *	2. We have two caches which need to be checked for the
1330 	 *	   presence of fragment blocks: the normal fragment cache
1331 	 *	   and a "reserve" cache.  The reserve cache is used to
1332 	 *	   prevent an unnecessary pipeline stall when the fragment cache
1333 	 *	   is full of fragments waiting to be compressed.
1334 	 */
1335 
1336 	if(fragment->index == SQUASHFS_INVALID_FRAG)
1337 		return NULL;
1338 
1339 	pthread_cleanup_push((void *) pthread_mutex_unlock, &dup_mutex);
1340 	pthread_mutex_lock(&dup_mutex);
1341 
1342 again:
1343 	buffer = cache_lookup_nowait(fragment_buffer, index, &locked);
1344 	if(buffer) {
1345 		pthread_mutex_unlock(&dup_mutex);
1346 		if(locked)
1347 			/* got a buffer being filled in.  Wait for it */
1348 			cache_wait_unlock(buffer);
1349 		goto finished;
1350 	}
1351 
1352 	/* not in fragment cache, is it in the reserve cache? */
1353 	buffer = cache_lookup_nowait(reserve_cache, index, &locked);
1354 	if(buffer) {
1355 		pthread_mutex_unlock(&dup_mutex);
1356 		if(locked)
1357 			/* got a buffer being filled in.  Wait for it */
1358 			cache_wait_unlock(buffer);
1359 		goto finished;
1360 	}
1361 
1362 	/* in neither cache, try to get it from the fragment cache */
1363 	buffer = cache_get_nowait(fragment_buffer, index);
1364 	if(!buffer) {
1365 		/*
1366 		 * no room, get it from the reserve cache, this is
1367 		 * dimensioned so it will always have space (no more than
1368 		 * processors + 1 can have an outstanding reserve buffer)
1369 		 */
1370 		buffer = cache_get_nowait(reserve_cache, index);
1371 		if(!buffer) {
1372 			/* failsafe */
1373 			ERROR("no space in reserve cache\n");
1374 			goto again;
1375 		}
1376 	}
1377 
1378 	pthread_mutex_unlock(&dup_mutex);
1379 
1380 	compressed_buffer = cache_lookup(fwriter_buffer, index);
1381 
1382 	pthread_cleanup_push((void *) pthread_mutex_unlock, &fragment_mutex);
1383 	pthread_mutex_lock(&fragment_mutex);
1384 	disk_fragment = &fragment_table[index];
1385 	size = SQUASHFS_COMPRESSED_SIZE_BLOCK(disk_fragment->size);
1386 	start_block = disk_fragment->start_block;
1387 	pthread_cleanup_pop(1);
1388 
1389 	if(SQUASHFS_COMPRESSED_BLOCK(disk_fragment->size)) {
1390 		int error;
1391 		char *data;
1392 
1393 		if(compressed_buffer)
1394 			data = compressed_buffer->data;
1395 		else {
1396 			data = read_from_disk(start_block, size);
1397 			if(data == NULL) {
1398 				ERROR("Failed to read fragment from output"
1399 					" filesystem\n");
1400 				BAD_ERROR("Output filesystem corrupted?\n");
1401 			}
1402 		}
1403 
1404 		res = compressor_uncompress(comp, buffer->data, data, size,
1405 			block_size, &error);
1406 		if(res == -1)
1407 			BAD_ERROR("%s uncompress failed with error code %d\n",
1408 				comp->name, error);
1409 	} else if(compressed_buffer)
1410 		memcpy(buffer->data, compressed_buffer->data, size);
1411 	else {
1412 		res = read_fs_bytes(fd, start_block, size, buffer->data);
1413 		if(res == 0) {
1414 			ERROR("Failed to read fragment from output "
1415 				"filesystem\n");
1416 			BAD_ERROR("Output filesystem corrupted?\n");
1417 		}
1418 	}
1419 
1420 	cache_unlock(buffer);
1421 	cache_block_put(compressed_buffer);
1422 
1423 finished:
1424 	pthread_cleanup_pop(0);
1425 
1426 	return buffer;
1427 }
1428 
1429 
get_fragment_checksum(struct file_info * file)1430 unsigned short get_fragment_checksum(struct file_info *file)
1431 {
1432 	struct file_buffer *frag_buffer;
1433 	struct append_file *append;
1434 	int res, index = file->fragment->index;
1435 	unsigned short checksum;
1436 
1437 	if(index == SQUASHFS_INVALID_FRAG)
1438 		return 0;
1439 
1440 	pthread_cleanup_push((void *) pthread_mutex_unlock, &dup_mutex);
1441 	pthread_mutex_lock(&dup_mutex);
1442 	res = file->have_frag_checksum;
1443 	checksum = file->fragment_checksum;
1444 	pthread_cleanup_pop(1);
1445 
1446 	if(res)
1447 		return checksum;
1448 
1449 	frag_buffer = get_fragment(file->fragment);
1450 
1451 	pthread_cleanup_push((void *) pthread_mutex_unlock, &dup_mutex);
1452 
1453 	for(append = file_mapping[index]; append; append = append->next) {
1454 		int offset = append->file->fragment->offset;
1455 		int size = append->file->fragment->size;
1456 		unsigned short cksum =
1457 			get_checksum_mem(frag_buffer->data + offset, size);
1458 
1459 		if(file == append->file)
1460 			checksum = cksum;
1461 
1462 		pthread_mutex_lock(&dup_mutex);
1463 		append->file->fragment_checksum = cksum;
1464 		append->file->have_frag_checksum = TRUE;
1465 		pthread_mutex_unlock(&dup_mutex);
1466 	}
1467 
1468 	cache_block_put(frag_buffer);
1469 	pthread_cleanup_pop(0);
1470 
1471 	return checksum;
1472 }
1473 
1474 
lock_fragments()1475 void lock_fragments()
1476 {
1477 	pthread_cleanup_push((void *) pthread_mutex_unlock, &fragment_mutex);
1478 	pthread_mutex_lock(&fragment_mutex);
1479 	fragments_locked = TRUE;
1480 	pthread_cleanup_pop(1);
1481 }
1482 
1483 
unlock_fragments()1484 void unlock_fragments()
1485 {
1486 	int frg, size;
1487 	struct file_buffer *write_buffer;
1488 
1489 	pthread_cleanup_push((void *) pthread_mutex_unlock, &fragment_mutex);
1490 	pthread_mutex_lock(&fragment_mutex);
1491 
1492 	/*
1493 	 * Note queue_empty() is inherently racy with respect to concurrent
1494 	 * queue get and pushes.  We avoid this because we're holding the
1495 	 * fragment_mutex which ensures no other threads can be using the
1496 	 * queue at this time.
1497 	 */
1498 	while(!queue_empty(locked_fragment)) {
1499 		write_buffer = queue_get(locked_fragment);
1500 		frg = write_buffer->block;
1501 		size = SQUASHFS_COMPRESSED_SIZE_BLOCK(fragment_table[frg].size);
1502 		fragment_table[frg].start_block = bytes;
1503 		write_buffer->block = bytes;
1504 		bytes += size;
1505 		fragments_outstanding --;
1506 		queue_put(to_writer, write_buffer);
1507 		TRACE("fragment_locked writing fragment %d, compressed size %d"
1508 			"\n", frg, size);
1509 	}
1510 	fragments_locked = FALSE;
1511 	pthread_cleanup_pop(1);
1512 }
1513 
1514 /* Called with the fragment_mutex locked */
add_pending_fragment(struct file_buffer * write_buffer,int c_byte,int fragment)1515 void add_pending_fragment(struct file_buffer *write_buffer, int c_byte,
1516 	int fragment)
1517 {
1518 	fragment_table[fragment].size = c_byte;
1519 	write_buffer->block = fragment;
1520 
1521 	queue_put(locked_fragment, write_buffer);
1522 }
1523 
1524 
write_fragment(struct file_buffer * fragment)1525 void write_fragment(struct file_buffer *fragment)
1526 {
1527 	if(fragment == NULL)
1528 		return;
1529 
1530 	pthread_cleanup_push((void *) pthread_mutex_unlock, &fragment_mutex);
1531 	pthread_mutex_lock(&fragment_mutex);
1532 	fragment_table[fragment->block].unused = 0;
1533 	fragments_outstanding ++;
1534 	queue_put(to_frag, fragment);
1535 	pthread_cleanup_pop(1);
1536 }
1537 
1538 
allocate_fragment()1539 struct file_buffer *allocate_fragment()
1540 {
1541 	struct file_buffer *fragment = cache_get(fragment_buffer, fragments);
1542 
1543 	pthread_cleanup_push((void *) pthread_mutex_unlock, &fragment_mutex);
1544 	pthread_mutex_lock(&fragment_mutex);
1545 
1546 	if(fragments % FRAG_SIZE == 0) {
1547 		void *ft = realloc(fragment_table, (fragments +
1548 			FRAG_SIZE) * sizeof(struct squashfs_fragment_entry));
1549 		if(ft == NULL)
1550 			MEM_ERROR();
1551 		fragment_table = ft;
1552 	}
1553 
1554 	fragment->size = 0;
1555 	fragment->block = fragments ++;
1556 
1557 	pthread_cleanup_pop(1);
1558 
1559 	return fragment;
1560 }
1561 
1562 
1563 static struct fragment empty_fragment = {SQUASHFS_INVALID_FRAG, 0, 0};
1564 
1565 
free_fragment(struct fragment * fragment)1566 void free_fragment(struct fragment *fragment)
1567 {
1568 	if(fragment != &empty_fragment)
1569 		free(fragment);
1570 }
1571 
1572 
get_and_fill_fragment(struct file_buffer * file_buffer,struct dir_ent * dir_ent)1573 struct fragment *get_and_fill_fragment(struct file_buffer *file_buffer,
1574 	struct dir_ent *dir_ent)
1575 {
1576 	struct fragment *ffrg;
1577 	struct file_buffer **fragment;
1578 
1579 	if(file_buffer == NULL || file_buffer->size == 0)
1580 		return &empty_fragment;
1581 
1582 	fragment = eval_frag_actions(root_dir, dir_ent);
1583 
1584 	if((*fragment) && (*fragment)->size + file_buffer->size > block_size) {
1585 		write_fragment(*fragment);
1586 		*fragment = NULL;
1587 	}
1588 
1589 	ffrg = malloc(sizeof(struct fragment));
1590 	if(ffrg == NULL)
1591 		MEM_ERROR();
1592 
1593 	if(*fragment == NULL)
1594 		*fragment = allocate_fragment();
1595 
1596 	ffrg->index = (*fragment)->block;
1597 	ffrg->offset = (*fragment)->size;
1598 	ffrg->size = file_buffer->size;
1599 	memcpy((*fragment)->data + (*fragment)->size, file_buffer->data,
1600 		file_buffer->size);
1601 	(*fragment)->size += file_buffer->size;
1602 
1603 	return ffrg;
1604 }
1605 
1606 
generic_write_table(int length,void * buffer,int length2,void * buffer2,int uncompressed)1607 long long generic_write_table(int length, void *buffer, int length2,
1608 	void *buffer2, int uncompressed)
1609 {
1610 	int meta_blocks = (length + SQUASHFS_METADATA_SIZE - 1) /
1611 		SQUASHFS_METADATA_SIZE;
1612 	long long *list, start_bytes;
1613 	int compressed_size, i, list_size = meta_blocks * sizeof(long long);
1614 	unsigned short c_byte;
1615 	char cbuffer[(SQUASHFS_METADATA_SIZE << 2) + 2];
1616 
1617 #ifdef SQUASHFS_TRACE
1618 	long long obytes = bytes;
1619 	int olength = length;
1620 #endif
1621 
1622 	list = malloc(list_size);
1623 	if(list == NULL)
1624 		MEM_ERROR();
1625 
1626 	for(i = 0; i < meta_blocks; i++) {
1627 		int avail_bytes = length > SQUASHFS_METADATA_SIZE ?
1628 			SQUASHFS_METADATA_SIZE : length;
1629 		c_byte = mangle(cbuffer + BLOCK_OFFSET, buffer + i *
1630 			SQUASHFS_METADATA_SIZE , avail_bytes,
1631 			SQUASHFS_METADATA_SIZE, uncompressed, 0);
1632 		SQUASHFS_SWAP_SHORTS(&c_byte, cbuffer, 1);
1633 		list[i] = bytes;
1634 		compressed_size = SQUASHFS_COMPRESSED_SIZE(c_byte) +
1635 			BLOCK_OFFSET;
1636 		TRACE("block %d @ 0x%llx, compressed size %d\n", i, bytes,
1637 			compressed_size);
1638 		write_destination(fd, bytes, compressed_size, cbuffer);
1639 		bytes += compressed_size;
1640 		total_bytes += avail_bytes;
1641 		length -= avail_bytes;
1642 	}
1643 
1644 	start_bytes = bytes;
1645 	if(length2) {
1646 		write_destination(fd, bytes, length2, buffer2);
1647 		bytes += length2;
1648 		total_bytes += length2;
1649 	}
1650 
1651 	SQUASHFS_INSWAP_LONG_LONGS(list, meta_blocks);
1652 	write_destination(fd, bytes, list_size, list);
1653 	bytes += list_size;
1654 	total_bytes += list_size;
1655 
1656 	TRACE("generic_write_table: total uncompressed %d compressed %lld\n",
1657 		olength, bytes - obytes);
1658 
1659 	free(list);
1660 
1661 	return start_bytes;
1662 }
1663 
1664 
write_fragment_table()1665 long long write_fragment_table()
1666 {
1667 	unsigned int frag_bytes = SQUASHFS_FRAGMENT_BYTES(fragments);
1668 	int i;
1669 
1670 	TRACE("write_fragment_table: fragments %d, frag_bytes %d\n", fragments,
1671 		frag_bytes);
1672 	for(i = 0; i < fragments; i++) {
1673 		TRACE("write_fragment_table: fragment %d, start_block 0x%llx, "
1674 			"size %d\n", i, fragment_table[i].start_block,
1675 			fragment_table[i].size);
1676 		SQUASHFS_INSWAP_FRAGMENT_ENTRY(&fragment_table[i]);
1677 	}
1678 
1679 	return generic_write_table(frag_bytes, fragment_table, 0, NULL, noF);
1680 }
1681 
1682 
1683 char read_from_file_buffer[SQUASHFS_FILE_MAX_SIZE];
read_from_disk(long long start,unsigned int avail_bytes)1684 static char *read_from_disk(long long start, unsigned int avail_bytes)
1685 {
1686 	int res;
1687 
1688 	res = read_fs_bytes(fd, start, avail_bytes, read_from_file_buffer);
1689 	if(res == 0)
1690 		return NULL;
1691 
1692 	return read_from_file_buffer;
1693 }
1694 
1695 
1696 char read_from_file_buffer2[SQUASHFS_FILE_MAX_SIZE];
read_from_disk2(long long start,unsigned int avail_bytes)1697 char *read_from_disk2(long long start, unsigned int avail_bytes)
1698 {
1699 	int res;
1700 
1701 	res = read_fs_bytes(fd, start, avail_bytes, read_from_file_buffer2);
1702 	if(res == 0)
1703 		return NULL;
1704 
1705 	return read_from_file_buffer2;
1706 }
1707 
1708 
1709 /*
1710  * Compute 16 bit BSD checksum over the data
1711  */
get_checksum(char * buff,int bytes,unsigned short chksum)1712 unsigned short get_checksum(char *buff, int bytes, unsigned short chksum)
1713 {
1714 	unsigned char *b = (unsigned char *) buff;
1715 
1716 	while(bytes --) {
1717 		chksum = (chksum & 1) ? (chksum >> 1) | 0x8000 : chksum >> 1;
1718 		chksum += *b++;
1719 	}
1720 
1721 	return chksum;
1722 }
1723 
1724 
get_checksum_disk(long long start,long long l,unsigned int * blocks)1725 unsigned short get_checksum_disk(long long start, long long l,
1726 	unsigned int *blocks)
1727 {
1728 	unsigned short chksum = 0;
1729 	unsigned int bytes;
1730 	struct file_buffer *write_buffer;
1731 	int i;
1732 
1733 	for(i = 0; l; i++)  {
1734 		bytes = SQUASHFS_COMPRESSED_SIZE_BLOCK(blocks[i]);
1735 		if(bytes == 0) /* sparse block */
1736 			continue;
1737 		write_buffer = cache_lookup(bwriter_buffer, start);
1738 		if(write_buffer) {
1739 			chksum = get_checksum(write_buffer->data, bytes,
1740 				chksum);
1741 			cache_block_put(write_buffer);
1742 		} else {
1743 			void *data = read_from_disk(start, bytes);
1744 			if(data == NULL) {
1745 				ERROR("Failed to checksum data from output"
1746 					" filesystem\n");
1747 				BAD_ERROR("Output filesystem corrupted?\n");
1748 			}
1749 
1750 			chksum = get_checksum(data, bytes, chksum);
1751 		}
1752 
1753 		l -= bytes;
1754 		start += bytes;
1755 	}
1756 
1757 	return chksum;
1758 }
1759 
1760 
get_checksum_mem(char * buff,int bytes)1761 unsigned short get_checksum_mem(char *buff, int bytes)
1762 {
1763 	return get_checksum(buff, bytes, 0);
1764 }
1765 
1766 
get_checksum_mem_buffer(struct file_buffer * file_buffer)1767 unsigned short get_checksum_mem_buffer(struct file_buffer *file_buffer)
1768 {
1769 	if(file_buffer == NULL)
1770 		return 0;
1771 	else
1772 		return get_checksum(file_buffer->data, file_buffer->size, 0);
1773 }
1774 
1775 
1776 #define DUP_HASH(a) (a & 0xffff)
add_file(long long start,long long file_size,long long file_bytes,unsigned int * block_listp,int blocks,unsigned int fragment,int offset,int bytes)1777 void add_file(long long start, long long file_size, long long file_bytes,
1778 	unsigned int *block_listp, int blocks, unsigned int fragment,
1779 	int offset, int bytes)
1780 {
1781 	struct fragment *frg;
1782 	unsigned int *block_list = block_listp;
1783 	struct file_info *dupl_ptr = dupl[DUP_HASH(file_size)];
1784 	struct append_file *append_file;
1785 	struct file_info *file;
1786 
1787 	if(!duplicate_checking || file_size == 0)
1788 		return;
1789 
1790 	for(; dupl_ptr; dupl_ptr = dupl_ptr->next) {
1791 		if(file_size != dupl_ptr->file_size)
1792 			continue;
1793 		if(blocks != 0 && start != dupl_ptr->start)
1794 			continue;
1795 		if(fragment != dupl_ptr->fragment->index)
1796 			continue;
1797 		if(fragment != SQUASHFS_INVALID_FRAG && (offset !=
1798 				dupl_ptr->fragment->offset || bytes !=
1799 				dupl_ptr->fragment->size))
1800 			continue;
1801 		return;
1802 	}
1803 
1804 	frg = malloc(sizeof(struct fragment));
1805 	if(frg == NULL)
1806 		MEM_ERROR();
1807 
1808 	frg->index = fragment;
1809 	frg->offset = offset;
1810 	frg->size = bytes;
1811 
1812 	file = add_non_dup(file_size, file_bytes, block_list, start, frg, 0, 0,
1813 		FALSE, FALSE);
1814 
1815 	if(fragment == SQUASHFS_INVALID_FRAG)
1816 		return;
1817 
1818 	append_file = malloc(sizeof(struct append_file));
1819 	if(append_file == NULL)
1820 		MEM_ERROR();
1821 
1822 	append_file->file = file;
1823 	append_file->next = file_mapping[fragment];
1824 	file_mapping[fragment] = append_file;
1825 }
1826 
1827 
pre_duplicate(long long file_size)1828 int pre_duplicate(long long file_size)
1829 {
1830 	struct file_info *dupl_ptr = dupl[DUP_HASH(file_size)];
1831 
1832 	for(; dupl_ptr; dupl_ptr = dupl_ptr->next)
1833 		if(dupl_ptr->file_size == file_size)
1834 			return TRUE;
1835 
1836 	return FALSE;
1837 }
1838 
1839 
add_non_dup(long long file_size,long long bytes,unsigned int * block_list,long long start,struct fragment * fragment,unsigned short checksum,unsigned short fragment_checksum,int checksum_flag,int checksum_frag_flag)1840 struct file_info *add_non_dup(long long file_size, long long bytes,
1841 	unsigned int *block_list, long long start, struct fragment *fragment,
1842 	unsigned short checksum, unsigned short fragment_checksum,
1843 	int checksum_flag, int checksum_frag_flag)
1844 {
1845 	struct file_info *dupl_ptr = malloc(sizeof(struct file_info));
1846 
1847 	if(dupl_ptr == NULL)
1848 		MEM_ERROR();
1849 
1850 	dupl_ptr->file_size = file_size;
1851 	dupl_ptr->bytes = bytes;
1852 	dupl_ptr->block_list = block_list;
1853 	dupl_ptr->start = start;
1854 	dupl_ptr->fragment = fragment;
1855 	dupl_ptr->checksum = checksum;
1856 	dupl_ptr->fragment_checksum = fragment_checksum;
1857 	dupl_ptr->have_frag_checksum = checksum_frag_flag;
1858 	dupl_ptr->have_checksum = checksum_flag;
1859 
1860 	pthread_cleanup_push((void *) pthread_mutex_unlock, &dup_mutex);
1861         pthread_mutex_lock(&dup_mutex);
1862 	dupl_ptr->next = dupl[DUP_HASH(file_size)];
1863 	dupl[DUP_HASH(file_size)] = dupl_ptr;
1864 	dup_files ++;
1865 	pthread_cleanup_pop(1);
1866 
1867 	return dupl_ptr;
1868 }
1869 
1870 
frag_duplicate(struct file_buffer * file_buffer,char * dont_put)1871 struct fragment *frag_duplicate(struct file_buffer *file_buffer, char *dont_put)
1872 {
1873 	struct file_info *dupl_ptr;
1874 	struct file_buffer *buffer;
1875 	struct file_info *dupl_start = file_buffer->dupl_start;
1876 	long long file_size = file_buffer->file_size;
1877 	unsigned short checksum = file_buffer->checksum;
1878 	int res;
1879 
1880 	if(file_buffer->duplicate) {
1881 		TRACE("Found duplicate file, fragment %d, size %d, offset %d, "
1882 			"checksum 0x%x\n", dupl_start->fragment->index,
1883 			file_size, dupl_start->fragment->offset, checksum);
1884 		*dont_put = TRUE;
1885 		return dupl_start->fragment;
1886 	} else {
1887 		*dont_put = FALSE;
1888 		dupl_ptr = dupl[DUP_HASH(file_size)];
1889 	}
1890 
1891 	for(; dupl_ptr && dupl_ptr != dupl_start; dupl_ptr = dupl_ptr->next) {
1892 		if(file_size == dupl_ptr->file_size && file_size ==
1893 				dupl_ptr->fragment->size) {
1894 			if(get_fragment_checksum(dupl_ptr) == checksum) {
1895 				buffer = get_fragment(dupl_ptr->fragment);
1896 				res = memcmp(file_buffer->data, buffer->data +
1897 					dupl_ptr->fragment->offset, file_size);
1898 				cache_block_put(buffer);
1899 				if(res == 0)
1900 					break;
1901 			}
1902 		}
1903 	}
1904 
1905 	if(!dupl_ptr || dupl_ptr == dupl_start)
1906 		return NULL;
1907 
1908 	TRACE("Found duplicate file, fragment %d, size %d, offset %d, "
1909 		"checksum 0x%x\n", dupl_ptr->fragment->index, file_size,
1910 		dupl_ptr->fragment->offset, checksum);
1911 
1912 	return dupl_ptr->fragment;
1913 }
1914 
1915 
duplicate(long long file_size,long long bytes,unsigned int ** block_list,long long * start,struct fragment ** fragment,struct file_buffer * file_buffer,int blocks,unsigned short checksum,int checksum_flag)1916 struct file_info *duplicate(long long file_size, long long bytes,
1917 	unsigned int **block_list, long long *start, struct fragment **fragment,
1918 	struct file_buffer *file_buffer, int blocks, unsigned short checksum,
1919 	int checksum_flag)
1920 {
1921 	struct file_info *dupl_ptr = dupl[DUP_HASH(file_size)];
1922 	int frag_bytes = file_buffer ? file_buffer->size : 0;
1923 	unsigned short fragment_checksum = file_buffer ?
1924 		file_buffer->checksum : 0;
1925 
1926 	for(; dupl_ptr; dupl_ptr = dupl_ptr->next)
1927 		if(file_size == dupl_ptr->file_size && bytes == dupl_ptr->bytes
1928 				 && frag_bytes == dupl_ptr->fragment->size) {
1929 			long long target_start, dup_start = dupl_ptr->start;
1930 			int block;
1931 
1932 			if(memcmp(*block_list, dupl_ptr->block_list, blocks *
1933 					sizeof(unsigned int)) != 0)
1934 				continue;
1935 
1936 			if(checksum_flag == FALSE) {
1937 				checksum = get_checksum_disk(*start, bytes,
1938 					*block_list);
1939 				checksum_flag = TRUE;
1940 			}
1941 
1942 			if(!dupl_ptr->have_checksum) {
1943 				dupl_ptr->checksum =
1944 					get_checksum_disk(dupl_ptr->start,
1945 					dupl_ptr->bytes, dupl_ptr->block_list);
1946 				dupl_ptr->have_checksum = TRUE;
1947 			}
1948 
1949 			if(checksum != dupl_ptr->checksum ||
1950 					fragment_checksum !=
1951 					get_fragment_checksum(dupl_ptr))
1952 				continue;
1953 
1954 			target_start = *start;
1955 			for(block = 0; block < blocks; block ++) {
1956 				int size = SQUASHFS_COMPRESSED_SIZE_BLOCK
1957 					((*block_list)[block]);
1958 				struct file_buffer *target_buffer = NULL;
1959 				struct file_buffer *dup_buffer = NULL;
1960 				char *target_data, *dup_data;
1961 				int res;
1962 
1963 				if(size == 0)
1964 					continue;
1965 				target_buffer = cache_lookup(bwriter_buffer,
1966 					target_start);
1967 				if(target_buffer)
1968 					target_data = target_buffer->data;
1969 				else {
1970 					target_data =
1971 						read_from_disk(target_start,
1972 						size);
1973 					if(target_data == NULL) {
1974 						ERROR("Failed to read data from"
1975 							" output filesystem\n");
1976 						BAD_ERROR("Output filesystem"
1977 							" corrupted?\n");
1978 					}
1979 				}
1980 
1981 				dup_buffer = cache_lookup(bwriter_buffer,
1982 					dup_start);
1983 				if(dup_buffer)
1984 					dup_data = dup_buffer->data;
1985 				else {
1986 					dup_data = read_from_disk2(dup_start,
1987 						size);
1988 					if(dup_data == NULL) {
1989 						ERROR("Failed to read data from"
1990 							" output filesystem\n");
1991 						BAD_ERROR("Output filesystem"
1992 							" corrupted?\n");
1993 					}
1994 				}
1995 
1996 				res = memcmp(target_data, dup_data, size);
1997 				cache_block_put(target_buffer);
1998 				cache_block_put(dup_buffer);
1999 				if(res != 0)
2000 					break;
2001 				target_start += size;
2002 				dup_start += size;
2003 			}
2004 			if(block == blocks) {
2005 				struct file_buffer *frag_buffer =
2006 					get_fragment(dupl_ptr->fragment);
2007 
2008 				if(frag_bytes == 0 ||
2009 						memcmp(file_buffer->data,
2010 						frag_buffer->data +
2011 						dupl_ptr->fragment->offset,
2012 						frag_bytes) == 0) {
2013 					TRACE("Found duplicate file, start "
2014 						"0x%llx, size %lld, checksum "
2015 						"0x%x, fragment %d, size %d, "
2016 						"offset %d, checksum 0x%x\n",
2017 						dupl_ptr->start,
2018 						dupl_ptr->bytes,
2019 						dupl_ptr->checksum,
2020 						dupl_ptr->fragment->index,
2021 						frag_bytes,
2022 						dupl_ptr->fragment->offset,
2023 						fragment_checksum);
2024 					*block_list = dupl_ptr->block_list;
2025 					*start = dupl_ptr->start;
2026 					*fragment = dupl_ptr->fragment;
2027 					cache_block_put(frag_buffer);
2028 					return 0;
2029 				}
2030 				cache_block_put(frag_buffer);
2031 			}
2032 		}
2033 
2034 
2035 	return add_non_dup(file_size, bytes, *block_list, *start, *fragment,
2036 		checksum, fragment_checksum, checksum_flag, TRUE);
2037 }
2038 
2039 
is_fragment(struct inode_info * inode)2040 inline int is_fragment(struct inode_info *inode)
2041 {
2042 	off_t file_size = inode->buf.st_size;
2043 
2044 	/*
2045 	 * If this block is to be compressed differently to the
2046 	 * fragment compression then it cannot be a fragment
2047 	 */
2048 	if(inode->noF != noF)
2049 		return FALSE;
2050 
2051 	return !inode->no_fragments && file_size && (file_size < block_size ||
2052 		(inode->always_use_fragments && file_size & (block_size - 1)));
2053 }
2054 
2055 
put_file_buffer(struct file_buffer * file_buffer)2056 void put_file_buffer(struct file_buffer *file_buffer)
2057 {
2058 	/*
2059 	 * Decide where to send the file buffer:
2060 	 * - compressible non-fragment blocks go to the deflate threads,
2061 	 * - fragments go to the process fragment threads,
2062 	 * - all others go directly to the main thread
2063 	 */
2064 	if(file_buffer->error) {
2065 		file_buffer->fragment = 0;
2066 		seq_queue_put(to_main, file_buffer);
2067 	} else if (file_buffer->file_size == 0)
2068 		seq_queue_put(to_main, file_buffer);
2069  	else if(file_buffer->fragment)
2070 		queue_put(to_process_frag, file_buffer);
2071 	else
2072 		queue_put(to_deflate, file_buffer);
2073 }
2074 
2075 
2076 static int seq = 0;
reader_read_process(struct dir_ent * dir_ent)2077 void reader_read_process(struct dir_ent *dir_ent)
2078 {
2079 	long long bytes = 0;
2080 	struct inode_info *inode = dir_ent->inode;
2081 	struct file_buffer *prev_buffer = NULL, *file_buffer;
2082 	int status, byte, res, child;
2083 	int file = pseudo_exec_file(get_pseudo_file(inode->pseudo_id), &child);
2084 
2085 	if(!file) {
2086 		file_buffer = cache_get_nohash(reader_buffer);
2087 		file_buffer->sequence = seq ++;
2088 		goto read_err;
2089 	}
2090 
2091 	while(1) {
2092 		file_buffer = cache_get_nohash(reader_buffer);
2093 		file_buffer->sequence = seq ++;
2094 		file_buffer->noD = inode->noD;
2095 
2096 		byte = read_bytes(file, file_buffer->data, block_size);
2097 		if(byte == -1)
2098 			goto read_err2;
2099 
2100 		file_buffer->size = byte;
2101 		file_buffer->file_size = -1;
2102 		file_buffer->error = FALSE;
2103 		file_buffer->fragment = FALSE;
2104 		bytes += byte;
2105 
2106 		if(byte == 0)
2107 			break;
2108 
2109 		/*
2110 		 * Update progress bar size.  This is done
2111 		 * on every block rather than waiting for all blocks to be
2112 		 * read incase write_file_process() is running in parallel
2113 		 * with this.  Otherwise the current progress bar position
2114 		 * may get ahead of the progress bar size.
2115 		 */
2116 		progress_bar_size(1);
2117 
2118 		if(prev_buffer)
2119 			put_file_buffer(prev_buffer);
2120 		prev_buffer = file_buffer;
2121 	}
2122 
2123 	/*
2124  	 * Update inode file size now that the size of the dynamic pseudo file
2125 	 * is known.  This is needed for the -info option.
2126 	 */
2127 	inode->buf.st_size = bytes;
2128 
2129 	res = waitpid(child, &status, 0);
2130 	close(file);
2131 
2132 	if(res == -1 || !WIFEXITED(status) || WEXITSTATUS(status) != 0)
2133 		goto read_err;
2134 
2135 	if(prev_buffer == NULL)
2136 		prev_buffer = file_buffer;
2137 	else {
2138 		cache_block_put(file_buffer);
2139 		seq --;
2140 	}
2141 	prev_buffer->file_size = bytes;
2142 	prev_buffer->fragment = is_fragment(inode);
2143 	put_file_buffer(prev_buffer);
2144 
2145 	return;
2146 
2147 read_err2:
2148 	close(file);
2149 read_err:
2150 	if(prev_buffer) {
2151 		cache_block_put(file_buffer);
2152 		seq --;
2153 		file_buffer = prev_buffer;
2154 	}
2155 	file_buffer->error = TRUE;
2156 	put_file_buffer(file_buffer);
2157 }
2158 
2159 
reader_read_file(struct dir_ent * dir_ent)2160 void reader_read_file(struct dir_ent *dir_ent)
2161 {
2162 	struct stat *buf = &dir_ent->inode->buf, buf2;
2163 	struct file_buffer *file_buffer;
2164 	int blocks, file, res;
2165 	long long bytes, read_size;
2166 	struct inode_info *inode = dir_ent->inode;
2167 
2168 	if(inode->read)
2169 		return;
2170 
2171 	inode->read = TRUE;
2172 again:
2173 	bytes = 0;
2174 	read_size = buf->st_size;
2175 	blocks = (read_size + block_size - 1) >> block_log;
2176 
2177 	file = open(pathname_reader(dir_ent), O_RDONLY);
2178 	if(file == -1) {
2179 		file_buffer = cache_get_nohash(reader_buffer);
2180 		file_buffer->sequence = seq ++;
2181 		goto read_err2;
2182 	}
2183 
2184 	do {
2185 		file_buffer = cache_get_nohash(reader_buffer);
2186 		file_buffer->file_size = read_size;
2187 		file_buffer->sequence = seq ++;
2188 		file_buffer->noD = inode->noD;
2189 		file_buffer->error = FALSE;
2190 
2191 		/*
2192 		 * Always try to read block_size bytes from the file rather
2193 		 * than expected bytes (which will be less than the block_size
2194 		 * at the file tail) to check that the file hasn't grown
2195 		 * since being stated.  If it is longer (or shorter) than
2196 		 * expected, then restat, and try again.  Note the special
2197 		 * case where the file is an exact multiple of the block_size
2198 		 * is dealt with later.
2199 		 */
2200 		file_buffer->size = read_bytes(file, file_buffer->data,
2201 			block_size);
2202 		if(file_buffer->size == -1)
2203 			goto read_err;
2204 
2205 		bytes += file_buffer->size;
2206 
2207 		if(blocks > 1) {
2208 			/* non-tail block should be exactly block_size */
2209 			if(file_buffer->size < block_size)
2210 				goto restat;
2211 
2212 			file_buffer->fragment = FALSE;
2213 			put_file_buffer(file_buffer);
2214 		}
2215 	} while(-- blocks > 0);
2216 
2217 	/* Overall size including tail should match */
2218 	if(read_size != bytes)
2219 		goto restat;
2220 
2221 	if(read_size && read_size % block_size == 0) {
2222 		/*
2223 		 * Special case where we've not tried to read past the end of
2224 		 * the file.  We expect to get EOF, i.e. the file isn't larger
2225 		 * than we expect.
2226 		 */
2227 		char buffer;
2228 		int res;
2229 
2230 		res = read_bytes(file, &buffer, 1);
2231 		if(res == -1)
2232 			goto read_err;
2233 
2234 		if(res != 0)
2235 			goto restat;
2236 	}
2237 
2238 	file_buffer->fragment = is_fragment(inode);
2239 	put_file_buffer(file_buffer);
2240 
2241 	close(file);
2242 
2243 	return;
2244 
2245 restat:
2246 	res = fstat(file, &buf2);
2247 	if(res == -1) {
2248 		ERROR("Cannot stat dir/file %s because %s\n",
2249 			pathname_reader(dir_ent), strerror(errno));
2250 		goto read_err;
2251 	}
2252 
2253 	if(read_size != buf2.st_size) {
2254 		close(file);
2255 		memcpy(buf, &buf2, sizeof(struct stat));
2256 		file_buffer->error = 2;
2257 		put_file_buffer(file_buffer);
2258 		goto again;
2259 	}
2260 read_err:
2261 	close(file);
2262 read_err2:
2263 	file_buffer->error = TRUE;
2264 	put_file_buffer(file_buffer);
2265 }
2266 
2267 
reader_scan(struct dir_info * dir)2268 void reader_scan(struct dir_info *dir) {
2269 	struct dir_ent *dir_ent = dir->list;
2270 
2271 	for(; dir_ent; dir_ent = dir_ent->next) {
2272 		struct stat *buf = &dir_ent->inode->buf;
2273 		if(dir_ent->inode->root_entry)
2274 			continue;
2275 
2276 		if(IS_PSEUDO_PROCESS(dir_ent->inode)) {
2277 			reader_read_process(dir_ent);
2278 			continue;
2279 		}
2280 
2281 		switch(buf->st_mode & S_IFMT) {
2282 			case S_IFREG:
2283 				reader_read_file(dir_ent);
2284 				break;
2285 			case S_IFDIR:
2286 				reader_scan(dir_ent->dir);
2287 				break;
2288 		}
2289 	}
2290 }
2291 
2292 
reader(void * arg)2293 void *reader(void *arg)
2294 {
2295 	if(!sorted)
2296 		reader_scan(queue_get(to_reader));
2297 	else {
2298 		int i;
2299 		struct priority_entry *entry;
2300 
2301 		queue_get(to_reader);
2302 		for(i = 65535; i >= 0; i--)
2303 			for(entry = priority_list[i]; entry;
2304 							entry = entry->next)
2305 				reader_read_file(entry->dir);
2306 	}
2307 
2308 	pthread_exit(NULL);
2309 }
2310 
2311 
writer(void * arg)2312 void *writer(void *arg)
2313 {
2314 	while(1) {
2315 		struct file_buffer *file_buffer = queue_get(to_writer);
2316 		off_t off;
2317 
2318 		if(file_buffer == NULL) {
2319 			queue_put(from_writer, NULL);
2320 			continue;
2321 		}
2322 
2323 		off = file_buffer->block;
2324 
2325 		pthread_cleanup_push((void *) pthread_mutex_unlock, &pos_mutex);
2326 		pthread_mutex_lock(&pos_mutex);
2327 
2328 		if(lseek(fd, off, SEEK_SET) == -1) {
2329 			ERROR("writer: Lseek on destination failed because "
2330 				"%s, offset=0x%llx\n", strerror(errno), off);
2331 			BAD_ERROR("Probably out of space on output "
2332 				"%s\n", block_device ? "block device" :
2333 				"filesystem");
2334 		}
2335 
2336 		if(write_bytes(fd, file_buffer->data,
2337 				file_buffer->size) == -1)
2338 			BAD_ERROR("Failed to write to output %s\n",
2339 				block_device ? "block device" : "filesystem");
2340 
2341 		pthread_cleanup_pop(1);
2342 
2343 		cache_block_put(file_buffer);
2344 	}
2345 }
2346 
2347 
all_zero(struct file_buffer * file_buffer)2348 int all_zero(struct file_buffer *file_buffer)
2349 {
2350 	int i;
2351 	long entries = file_buffer->size / sizeof(long);
2352 	long *p = (long *) file_buffer->data;
2353 
2354 	for(i = 0; i < entries && p[i] == 0; i++);
2355 
2356 	if(i == entries) {
2357 		for(i = file_buffer->size & ~(sizeof(long) - 1);
2358 			i < file_buffer->size && file_buffer->data[i] == 0;
2359 			i++);
2360 
2361 		return i == file_buffer->size;
2362 	}
2363 
2364 	return 0;
2365 }
2366 
2367 
deflator(void * arg)2368 void *deflator(void *arg)
2369 {
2370 	struct file_buffer *write_buffer = cache_get_nohash(bwriter_buffer);
2371 	void *stream = NULL;
2372 	int res;
2373 
2374 	res = compressor_init(comp, &stream, block_size, 1);
2375 	if(res)
2376 		BAD_ERROR("deflator:: compressor_init failed\n");
2377 
2378 	while(1) {
2379 		struct file_buffer *file_buffer = queue_get(to_deflate);
2380 
2381 		if(sparse_files && all_zero(file_buffer)) {
2382 			file_buffer->c_byte = 0;
2383 			seq_queue_put(to_main, file_buffer);
2384 		} else {
2385 			write_buffer->c_byte = mangle2(stream,
2386 				write_buffer->data, file_buffer->data,
2387 				file_buffer->size, block_size,
2388 				file_buffer->noD, 1);
2389 			write_buffer->sequence = file_buffer->sequence;
2390 			write_buffer->file_size = file_buffer->file_size;
2391 			write_buffer->block = file_buffer->block;
2392 			write_buffer->size = SQUASHFS_COMPRESSED_SIZE_BLOCK
2393 				(write_buffer->c_byte);
2394 			write_buffer->fragment = FALSE;
2395 			write_buffer->error = FALSE;
2396 			cache_block_put(file_buffer);
2397 			seq_queue_put(to_main, write_buffer);
2398 			write_buffer = cache_get_nohash(bwriter_buffer);
2399 		}
2400 	}
2401 }
2402 
2403 
frag_deflator(void * arg)2404 void *frag_deflator(void *arg)
2405 {
2406 	void *stream = NULL;
2407 	int res;
2408 
2409 	res = compressor_init(comp, &stream, block_size, 1);
2410 	if(res)
2411 		BAD_ERROR("frag_deflator:: compressor_init failed\n");
2412 
2413 	pthread_cleanup_push((void *) pthread_mutex_unlock, &fragment_mutex);
2414 
2415 	while(1) {
2416 		int c_byte, compressed_size;
2417 		struct file_buffer *file_buffer = queue_get(to_frag);
2418 		struct file_buffer *write_buffer =
2419 			cache_get(fwriter_buffer, file_buffer->block);
2420 
2421 		c_byte = mangle2(stream, write_buffer->data, file_buffer->data,
2422 			file_buffer->size, block_size, noF, 1);
2423 		compressed_size = SQUASHFS_COMPRESSED_SIZE_BLOCK(c_byte);
2424 		write_buffer->size = compressed_size;
2425 		pthread_mutex_lock(&fragment_mutex);
2426 		if(fragments_locked == FALSE) {
2427 			fragment_table[file_buffer->block].size = c_byte;
2428 			fragment_table[file_buffer->block].start_block = bytes;
2429 			write_buffer->block = bytes;
2430 			bytes += compressed_size;
2431 			fragments_outstanding --;
2432 			queue_put(to_writer, write_buffer);
2433 			pthread_mutex_unlock(&fragment_mutex);
2434 			TRACE("Writing fragment %lld, uncompressed size %d, "
2435 				"compressed size %d\n", file_buffer->block,
2436 				file_buffer->size, compressed_size);
2437 		} else {
2438 				add_pending_fragment(write_buffer, c_byte,
2439 					file_buffer->block);
2440 				pthread_mutex_unlock(&fragment_mutex);
2441 		}
2442 		cache_block_put(file_buffer);
2443 	}
2444 
2445 	pthread_cleanup_pop(0);
2446 }
2447 
2448 
get_file_buffer()2449 struct file_buffer *get_file_buffer()
2450 {
2451 	struct file_buffer *file_buffer = seq_queue_get(to_main);
2452 
2453 	return file_buffer;
2454 }
2455 
2456 
write_file_empty(squashfs_inode * inode,struct dir_ent * dir_ent,struct file_buffer * file_buffer,int * duplicate_file)2457 void write_file_empty(squashfs_inode *inode, struct dir_ent *dir_ent,
2458 	struct file_buffer *file_buffer, int *duplicate_file)
2459 {
2460 	file_count ++;
2461 	*duplicate_file = FALSE;
2462 	cache_block_put(file_buffer);
2463 	create_inode(inode, NULL, dir_ent, SQUASHFS_FILE_TYPE, 0, 0, 0,
2464 		 NULL, &empty_fragment, NULL, 0);
2465 }
2466 
2467 
write_file_frag(squashfs_inode * inode,struct dir_ent * dir_ent,struct file_buffer * file_buffer,int * duplicate_file)2468 void write_file_frag(squashfs_inode *inode, struct dir_ent *dir_ent,
2469 	struct file_buffer *file_buffer, int *duplicate_file)
2470 {
2471 	int size = file_buffer->file_size;
2472 	struct fragment *fragment;
2473 	unsigned short checksum = file_buffer->checksum;
2474 	char dont_put;
2475 
2476 	fragment = frag_duplicate(file_buffer, &dont_put);
2477 	*duplicate_file = !fragment;
2478 	if(!fragment) {
2479 		fragment = get_and_fill_fragment(file_buffer, dir_ent);
2480 		if(duplicate_checking)
2481 			add_non_dup(size, 0, NULL, 0, fragment, 0, checksum,
2482 				TRUE, TRUE);
2483 	}
2484 
2485 	if(dont_put)
2486 		free(file_buffer);
2487 	else
2488 		cache_block_put(file_buffer);
2489 
2490 	total_bytes += size;
2491 	file_count ++;
2492 
2493 	inc_progress_bar();
2494 
2495 	create_inode(inode, NULL, dir_ent, SQUASHFS_FILE_TYPE, size, 0,
2496 			0, NULL, fragment, NULL, 0);
2497 
2498 	if(!duplicate_checking)
2499 		free_fragment(fragment);
2500 }
2501 
2502 
write_file_process(squashfs_inode * inode,struct dir_ent * dir_ent,struct file_buffer * read_buffer,int * duplicate_file)2503 int write_file_process(squashfs_inode *inode, struct dir_ent *dir_ent,
2504 	struct file_buffer *read_buffer, int *duplicate_file)
2505 {
2506 	long long read_size, file_bytes, start;
2507 	struct fragment *fragment;
2508 	unsigned int *block_list = NULL;
2509 	int block = 0, status;
2510 	long long sparse = 0;
2511 	struct file_buffer *fragment_buffer = NULL;
2512 
2513 	*duplicate_file = FALSE;
2514 
2515 	lock_fragments();
2516 
2517 	file_bytes = 0;
2518 	start = bytes;
2519 	while (1) {
2520 		read_size = read_buffer->file_size;
2521 		if(read_buffer->fragment)
2522 			fragment_buffer = read_buffer;
2523 		else {
2524 			block_list = realloc(block_list, (block + 1) *
2525 				sizeof(unsigned int));
2526 			if(block_list == NULL)
2527 				MEM_ERROR();
2528 			block_list[block ++] = read_buffer->c_byte;
2529 			if(read_buffer->c_byte) {
2530 				read_buffer->block = bytes;
2531 				bytes += read_buffer->size;
2532 				cache_hash(read_buffer, read_buffer->block);
2533 				file_bytes += read_buffer->size;
2534 				queue_put(to_writer, read_buffer);
2535 			} else {
2536 				sparse += read_buffer->size;
2537 				cache_block_put(read_buffer);
2538 			}
2539 		}
2540 		inc_progress_bar();
2541 
2542 		if(read_size != -1)
2543 			break;
2544 
2545 		read_buffer = get_file_buffer();
2546 		if(read_buffer->error)
2547 			goto read_err;
2548 	}
2549 
2550 	unlock_fragments();
2551 	fragment = get_and_fill_fragment(fragment_buffer, dir_ent);
2552 
2553 	if(duplicate_checking)
2554 		add_non_dup(read_size, file_bytes, block_list, start, fragment,
2555 			0, fragment_buffer ? fragment_buffer->checksum : 0,
2556 			FALSE, TRUE);
2557 	cache_block_put(fragment_buffer);
2558 	file_count ++;
2559 	total_bytes += read_size;
2560 
2561 	create_inode(inode, NULL, dir_ent, SQUASHFS_FILE_TYPE, read_size, start,
2562 		 block, block_list, fragment, NULL, sparse);
2563 
2564 	if(duplicate_checking == FALSE) {
2565 		free(block_list);
2566 		free_fragment(fragment);
2567 	}
2568 
2569 	return 0;
2570 
2571 read_err:
2572 	dec_progress_bar(block);
2573 	status = read_buffer->error;
2574 	bytes = start;
2575 	if(!block_device) {
2576 		int res;
2577 
2578 		queue_put(to_writer, NULL);
2579 		if(queue_get(from_writer) != 0)
2580 			EXIT_MKSQUASHFS();
2581 		res = ftruncate(fd, bytes);
2582 		if(res != 0)
2583 			BAD_ERROR("Failed to truncate dest file because %s\n",
2584 				strerror(errno));
2585 	}
2586 	unlock_fragments();
2587 	free(block_list);
2588 	cache_block_put(read_buffer);
2589 	return status;
2590 }
2591 
2592 
write_file_blocks_dup(squashfs_inode * inode,struct dir_ent * dir_ent,struct file_buffer * read_buffer,int * duplicate_file)2593 int write_file_blocks_dup(squashfs_inode *inode, struct dir_ent *dir_ent,
2594 	struct file_buffer *read_buffer, int *duplicate_file)
2595 {
2596 	int block, thresh;
2597 	long long read_size = read_buffer->file_size;
2598 	long long file_bytes, dup_start, start;
2599 	struct fragment *fragment;
2600 	struct file_info *dupl_ptr;
2601 	int blocks = (read_size + block_size - 1) >> block_log;
2602 	unsigned int *block_list, *block_listp;
2603 	struct file_buffer **buffer_list;
2604 	int status;
2605 	long long sparse = 0;
2606 	struct file_buffer *fragment_buffer = NULL;
2607 
2608 	block_list = malloc(blocks * sizeof(unsigned int));
2609 	if(block_list == NULL)
2610 		MEM_ERROR();
2611 	block_listp = block_list;
2612 
2613 	buffer_list = malloc(blocks * sizeof(struct file_buffer *));
2614 	if(buffer_list == NULL)
2615 		MEM_ERROR();
2616 
2617 	lock_fragments();
2618 
2619 	file_bytes = 0;
2620 	start = dup_start = bytes;
2621 	thresh = blocks > bwriter_size ? blocks - bwriter_size : 0;
2622 
2623 	for(block = 0; block < blocks;) {
2624 		if(read_buffer->fragment) {
2625 			block_list[block] = 0;
2626 			buffer_list[block] = NULL;
2627 			fragment_buffer = read_buffer;
2628 			blocks = read_size >> block_log;
2629 		} else {
2630 			block_list[block] = read_buffer->c_byte;
2631 
2632 			if(read_buffer->c_byte) {
2633 				read_buffer->block = bytes;
2634 				bytes += read_buffer->size;
2635 				file_bytes += read_buffer->size;
2636 				cache_hash(read_buffer, read_buffer->block);
2637 				if(block < thresh) {
2638 					buffer_list[block] = NULL;
2639 					queue_put(to_writer, read_buffer);
2640 				} else
2641 					buffer_list[block] = read_buffer;
2642 			} else {
2643 				buffer_list[block] = NULL;
2644 				sparse += read_buffer->size;
2645 				cache_block_put(read_buffer);
2646 			}
2647 		}
2648 		inc_progress_bar();
2649 
2650 		if(++block < blocks) {
2651 			read_buffer = get_file_buffer();
2652 			if(read_buffer->error)
2653 				goto read_err;
2654 		}
2655 	}
2656 
2657 	dupl_ptr = duplicate(read_size, file_bytes, &block_listp, &dup_start,
2658 		&fragment, fragment_buffer, blocks, 0, FALSE);
2659 
2660 	if(dupl_ptr) {
2661 		*duplicate_file = FALSE;
2662 		for(block = thresh; block < blocks; block ++)
2663 			if(buffer_list[block])
2664 				queue_put(to_writer, buffer_list[block]);
2665 		fragment = get_and_fill_fragment(fragment_buffer, dir_ent);
2666 		dupl_ptr->fragment = fragment;
2667 	} else {
2668 		*duplicate_file = TRUE;
2669 		for(block = thresh; block < blocks; block ++)
2670 			cache_block_put(buffer_list[block]);
2671 		bytes = start;
2672 		if(thresh && !block_device) {
2673 			int res;
2674 
2675 			queue_put(to_writer, NULL);
2676 			if(queue_get(from_writer) != 0)
2677 				EXIT_MKSQUASHFS();
2678 			res = ftruncate(fd, bytes);
2679 			if(res != 0)
2680 				BAD_ERROR("Failed to truncate dest file because"
2681 					"  %s\n", strerror(errno));
2682 		}
2683 	}
2684 
2685 	unlock_fragments();
2686 	cache_block_put(fragment_buffer);
2687 	free(buffer_list);
2688 	file_count ++;
2689 	total_bytes += read_size;
2690 
2691 	/*
2692 	 * sparse count is needed to ensure squashfs correctly reports a
2693  	 * a smaller block count on stat calls to sparse files.  This is
2694  	 * to ensure intelligent applications like cp correctly handle the
2695  	 * file as a sparse file.  If the file in the original filesystem isn't
2696  	 * stored as a sparse file then still store it sparsely in squashfs, but
2697  	 * report it as non-sparse on stat calls to preserve semantics
2698  	 */
2699 	if(sparse && (dir_ent->inode->buf.st_blocks << 9) >= read_size)
2700 		sparse = 0;
2701 
2702 	create_inode(inode, NULL, dir_ent, SQUASHFS_FILE_TYPE, read_size,
2703 		dup_start, blocks, block_listp, fragment, NULL, sparse);
2704 
2705 	if(*duplicate_file == TRUE)
2706 		free(block_list);
2707 
2708 	return 0;
2709 
2710 read_err:
2711 	dec_progress_bar(block);
2712 	status = read_buffer->error;
2713 	bytes = start;
2714 	if(thresh && !block_device) {
2715 		int res;
2716 
2717 		queue_put(to_writer, NULL);
2718 		if(queue_get(from_writer) != 0)
2719 			EXIT_MKSQUASHFS();
2720 		res = ftruncate(fd, bytes);
2721 		if(res != 0)
2722 			BAD_ERROR("Failed to truncate dest file because %s\n",
2723 				strerror(errno));
2724 	}
2725 	unlock_fragments();
2726 	for(blocks = thresh; blocks < block; blocks ++)
2727 		cache_block_put(buffer_list[blocks]);
2728 	free(buffer_list);
2729 	free(block_list);
2730 	cache_block_put(read_buffer);
2731 	return status;
2732 }
2733 
2734 
write_file_blocks(squashfs_inode * inode,struct dir_ent * dir_ent,struct file_buffer * read_buffer,int * dup)2735 int write_file_blocks(squashfs_inode *inode, struct dir_ent *dir_ent,
2736 	struct file_buffer *read_buffer, int *dup)
2737 {
2738 	long long read_size = read_buffer->file_size;
2739 	long long file_bytes, start;
2740 	struct fragment *fragment;
2741 	unsigned int *block_list;
2742 	int block, status;
2743 	int blocks = (read_size + block_size - 1) >> block_log;
2744 	long long sparse = 0;
2745 	struct file_buffer *fragment_buffer = NULL;
2746 
2747 	if(pre_duplicate(read_size))
2748 		return write_file_blocks_dup(inode, dir_ent, read_buffer, dup);
2749 
2750 	*dup = FALSE;
2751 
2752 	block_list = malloc(blocks * sizeof(unsigned int));
2753 	if(block_list == NULL)
2754 		MEM_ERROR();
2755 
2756 	lock_fragments();
2757 
2758 	file_bytes = 0;
2759 	start = bytes;
2760 	for(block = 0; block < blocks;) {
2761 		if(read_buffer->fragment) {
2762 			block_list[block] = 0;
2763 			fragment_buffer = read_buffer;
2764 			blocks = read_size >> block_log;
2765 		} else {
2766 			block_list[block] = read_buffer->c_byte;
2767 			if(read_buffer->c_byte) {
2768 				read_buffer->block = bytes;
2769 				bytes += read_buffer->size;
2770 				cache_hash(read_buffer, read_buffer->block);
2771 				file_bytes += read_buffer->size;
2772 				queue_put(to_writer, read_buffer);
2773 			} else {
2774 				sparse += read_buffer->size;
2775 				cache_block_put(read_buffer);
2776 			}
2777 		}
2778 		inc_progress_bar();
2779 
2780 		if(++block < blocks) {
2781 			read_buffer = get_file_buffer();
2782 			if(read_buffer->error)
2783 				goto read_err;
2784 		}
2785 	}
2786 
2787 	unlock_fragments();
2788 	fragment = get_and_fill_fragment(fragment_buffer, dir_ent);
2789 
2790 	if(duplicate_checking)
2791 		add_non_dup(read_size, file_bytes, block_list, start, fragment,
2792 			0, fragment_buffer ? fragment_buffer->checksum : 0,
2793 			FALSE, TRUE);
2794 	cache_block_put(fragment_buffer);
2795 	file_count ++;
2796 	total_bytes += read_size;
2797 
2798 	/*
2799 	 * sparse count is needed to ensure squashfs correctly reports a
2800  	 * a smaller block count on stat calls to sparse files.  This is
2801  	 * to ensure intelligent applications like cp correctly handle the
2802  	 * file as a sparse file.  If the file in the original filesystem isn't
2803  	 * stored as a sparse file then still store it sparsely in squashfs, but
2804  	 * report it as non-sparse on stat calls to preserve semantics
2805  	 */
2806 	if(sparse && (dir_ent->inode->buf.st_blocks << 9) >= read_size)
2807 		sparse = 0;
2808 
2809 	create_inode(inode, NULL, dir_ent, SQUASHFS_FILE_TYPE, read_size, start,
2810 		 blocks, block_list, fragment, NULL, sparse);
2811 
2812 	if(duplicate_checking == FALSE) {
2813 		free(block_list);
2814 		free_fragment(fragment);
2815 	}
2816 
2817 	return 0;
2818 
2819 read_err:
2820 	dec_progress_bar(block);
2821 	status = read_buffer->error;
2822 	bytes = start;
2823 	if(!block_device) {
2824 		int res;
2825 
2826 		queue_put(to_writer, NULL);
2827 		if(queue_get(from_writer) != 0)
2828 			EXIT_MKSQUASHFS();
2829 		res = ftruncate(fd, bytes);
2830 		if(res != 0)
2831 			BAD_ERROR("Failed to truncate dest file because %s\n",
2832 				strerror(errno));
2833 	}
2834 	unlock_fragments();
2835 	free(block_list);
2836 	cache_block_put(read_buffer);
2837 	return status;
2838 }
2839 
2840 
write_file(squashfs_inode * inode,struct dir_ent * dir,int * dup)2841 void write_file(squashfs_inode *inode, struct dir_ent *dir, int *dup)
2842 {
2843 	int status;
2844 	struct file_buffer *read_buffer;
2845 
2846 again:
2847 	read_buffer = get_file_buffer();
2848 	status = read_buffer->error;
2849 
2850 	if(status)
2851 		cache_block_put(read_buffer);
2852 	else if(read_buffer->file_size == -1)
2853 		status = write_file_process(inode, dir, read_buffer, dup);
2854 	else if(read_buffer->file_size == 0)
2855 		write_file_empty(inode, dir, read_buffer, dup);
2856 	else if(read_buffer->fragment && read_buffer->c_byte)
2857 		write_file_frag(inode, dir, read_buffer, dup);
2858 	else
2859 		status = write_file_blocks(inode, dir, read_buffer, dup);
2860 
2861 	if(status == 2) {
2862 		ERROR("File %s changed size while reading filesystem, "
2863 			"attempting to re-read\n", pathname(dir));
2864 		goto again;
2865 	} else if(status == 1) {
2866 		ERROR_START("Failed to read file %s", pathname(dir));
2867 		ERROR_EXIT(", creating empty file\n");
2868 		write_file_empty(inode, dir, NULL, dup);
2869 	}
2870 }
2871 
2872 
2873 #define BUFF_SIZE 512
2874 char *name;
2875 char *basename_r();
2876 
getbase(char * pathname)2877 char *getbase(char *pathname)
2878 {
2879 	static char *b_buffer = NULL;
2880 	static int b_size = BUFF_SIZE;
2881 	char *result;
2882 
2883 	if(b_buffer == NULL) {
2884 		b_buffer = malloc(b_size);
2885 		if(b_buffer == NULL)
2886 			MEM_ERROR();
2887 	}
2888 
2889 	while(1) {
2890 		if(*pathname != '/') {
2891 			result = getcwd(b_buffer, b_size);
2892 			if(result == NULL && errno != ERANGE)
2893 				BAD_ERROR("Getcwd failed in getbase\n");
2894 
2895 			/* enough room for pathname + "/" + '\0' terminator? */
2896 			if(result && strlen(pathname) + 2 <=
2897 						b_size - strlen(b_buffer)) {
2898 				strcat(strcat(b_buffer, "/"), pathname);
2899 				break;
2900 			}
2901 		} else if(strlen(pathname) < b_size) {
2902 			strcpy(b_buffer, pathname);
2903 			break;
2904 		}
2905 
2906 		/* Buffer not large enough, realloc and try again */
2907 		b_buffer = realloc(b_buffer, b_size += BUFF_SIZE);
2908 		if(b_buffer == NULL)
2909 			MEM_ERROR();
2910 	}
2911 
2912 	name = b_buffer;
2913 	if(((result = basename_r()) == NULL) || (strcmp(result, "..") == 0))
2914 		return NULL;
2915 	else
2916 		return result;
2917 }
2918 
2919 
basename_r()2920 char *basename_r()
2921 {
2922 	char *s;
2923 	char *p;
2924 	int n = 1;
2925 
2926 	for(;;) {
2927 		s = name;
2928 		if(*name == '\0')
2929 			return NULL;
2930 		if(*name != '/') {
2931 			while(*name != '\0' && *name != '/') name++;
2932 			n = name - s;
2933 		}
2934 		while(*name == '/') name++;
2935 		if(strncmp(s, ".", n) == 0)
2936 			continue;
2937 		if((*name == '\0') || (strncmp(s, "..", n) == 0) ||
2938 				((p = basename_r()) == NULL)) {
2939 			s[n] = '\0';
2940 			return s;
2941 		}
2942 		if(strcmp(p, "..") == 0)
2943 			continue;
2944 		return p;
2945 	}
2946 }
2947 
2948 
lookup_inode3(struct stat * buf,int pseudo,int id,char * symlink,int bytes)2949 struct inode_info *lookup_inode3(struct stat *buf, int pseudo, int id,
2950 	char *symlink, int bytes)
2951 {
2952 	int ino_hash = INODE_HASH(buf->st_dev, buf->st_ino);
2953 	struct inode_info *inode;
2954 
2955 	/*
2956 	 * Look-up inode in hash table, if it already exists we have a
2957 	 * hard-link, so increment the nlink count and return it.
2958 	 * Don't do the look-up for directories because we don't hard-link
2959 	 * directories.
2960 	 */
2961 	if ((buf->st_mode & S_IFMT) != S_IFDIR) {
2962 		for(inode = inode_info[ino_hash]; inode; inode = inode->next) {
2963 			if(memcmp(buf, &inode->buf, sizeof(struct stat)) == 0) {
2964 				inode->nlink ++;
2965 				return inode;
2966 			}
2967 		}
2968 	}
2969 
2970 	inode = malloc(sizeof(struct inode_info) + bytes);
2971 	if(inode == NULL)
2972 		MEM_ERROR();
2973 
2974 	if(bytes)
2975 		memcpy(&inode->symlink, symlink, bytes);
2976 	memcpy(&inode->buf, buf, sizeof(struct stat));
2977 	inode->read = FALSE;
2978 	inode->root_entry = FALSE;
2979 	inode->pseudo_file = pseudo;
2980 	inode->pseudo_id = id;
2981 	inode->inode = SQUASHFS_INVALID_BLK;
2982 	inode->nlink = 1;
2983 	inode->inode_number = 0;
2984 
2985 	/*
2986 	 * Copy filesystem wide defaults into inode, these filesystem
2987 	 * wide defaults may be altered on an individual inode basis by
2988 	 * user specified actions
2989 	 *
2990 	*/
2991 	inode->no_fragments = no_fragments;
2992 	inode->always_use_fragments = always_use_fragments;
2993 	inode->noD = noD;
2994 	inode->noF = noF;
2995 
2996 	inode->next = inode_info[ino_hash];
2997 	inode_info[ino_hash] = inode;
2998 
2999 	return inode;
3000 }
3001 
3002 
lookup_inode2(struct stat * buf,int pseudo,int id)3003 struct inode_info *lookup_inode2(struct stat *buf, int pseudo, int id)
3004 {
3005 	return lookup_inode3(buf, pseudo, id, NULL, 0);
3006 }
3007 
3008 
lookup_inode(struct stat * buf)3009 inline struct inode_info *lookup_inode(struct stat *buf)
3010 {
3011 	return lookup_inode2(buf, 0, 0);
3012 }
3013 
3014 
alloc_inode_no(struct inode_info * inode,unsigned int use_this)3015 inline void alloc_inode_no(struct inode_info *inode, unsigned int use_this)
3016 {
3017 	if (inode->inode_number == 0) {
3018 		inode->inode_number = use_this ? : inode_no ++;
3019 		if((inode->buf.st_mode & S_IFMT) == S_IFREG)
3020 			progress_bar_size((inode->buf.st_size + block_size - 1)
3021 								 >> block_log);
3022 	}
3023 }
3024 
3025 
create_dir_entry(char * name,char * source_name,char * nonstandard_pathname,struct dir_info * dir)3026 inline struct dir_ent *create_dir_entry(char *name, char *source_name,
3027 	char *nonstandard_pathname, struct dir_info *dir)
3028 {
3029 	struct dir_ent *dir_ent = malloc(sizeof(struct dir_ent));
3030 	if(dir_ent == NULL)
3031 		MEM_ERROR();
3032 
3033 	dir_ent->name = name;
3034 	dir_ent->source_name = source_name;
3035 	dir_ent->nonstandard_pathname = nonstandard_pathname;
3036 	dir_ent->our_dir = dir;
3037 	dir_ent->inode = NULL;
3038 	dir_ent->next = NULL;
3039 
3040 	return dir_ent;
3041 }
3042 
3043 
add_dir_entry(struct dir_ent * dir_ent,struct dir_info * sub_dir,struct inode_info * inode_info)3044 inline void add_dir_entry(struct dir_ent *dir_ent, struct dir_info *sub_dir,
3045 	struct inode_info *inode_info)
3046 {
3047 	struct dir_info *dir = dir_ent->our_dir;
3048 
3049 	if(sub_dir)
3050 		sub_dir->dir_ent = dir_ent;
3051 
3052 /* ANDROID CHANGES START*/
3053 #ifdef ANDROID
3054 	if (android_config) {
3055 		if (mount_point) {
3056 			char *mounted_path;
3057 			alloc_mounted_path(mount_point, subpathname(dir_ent), &mounted_path);
3058 			android_fs_config(mounted_path, &inode_info->buf, target_out_path);
3059 			free(mounted_path);
3060 		} else {
3061 			android_fs_config(pathname(dir_ent), &inode_info->buf, target_out_path);
3062 		}
3063 	}
3064 #endif
3065 /* ANDROID CHANGES END */
3066 
3067 	dir_ent->inode = inode_info;
3068 	dir_ent->dir = sub_dir;
3069 
3070 	dir_ent->next = dir->list;
3071 	dir->list = dir_ent;
3072 	dir->count++;
3073 }
3074 
3075 /* ANDROID CHANGES START*/
3076 #ifdef ANDROID
3077 /* Weird linker bug that complains those inline functions are undefined. */
3078 extern inline void add_dir_entry(struct dir_ent *dir_ent, struct dir_info *sub_dir,
3079 	struct inode_info *inode_info);
3080 extern inline void add_dir_entry2(char *name, char *source_name,
3081 	char *nonstandard_pathname, struct dir_info *sub_dir,
3082 	struct inode_info *inode_info, struct dir_info *dir);
3083 #endif
3084 /* ANDROID CHANGES END */
3085 
add_dir_entry2(char * name,char * source_name,char * nonstandard_pathname,struct dir_info * sub_dir,struct inode_info * inode_info,struct dir_info * dir)3086 inline void add_dir_entry2(char *name, char *source_name,
3087 	char *nonstandard_pathname, struct dir_info *sub_dir,
3088 	struct inode_info *inode_info, struct dir_info *dir)
3089 {
3090 	struct dir_ent *dir_ent = create_dir_entry(name, source_name,
3091 		nonstandard_pathname, dir);
3092 
3093 
3094 	add_dir_entry(dir_ent, sub_dir, inode_info);
3095 }
3096 
3097 
free_dir_entry(struct dir_ent * dir_ent)3098 inline void free_dir_entry(struct dir_ent *dir_ent)
3099 {
3100 	if(dir_ent->name)
3101 		free(dir_ent->name);
3102 
3103 	if(dir_ent->source_name)
3104 		free(dir_ent->source_name);
3105 
3106 	if(dir_ent->nonstandard_pathname)
3107 		free(dir_ent->nonstandard_pathname);
3108 
3109 	/* if this entry has been associated with an inode, then we need
3110 	 * to update the inode nlink count.  Orphaned inodes are harmless, and
3111 	 * is easier to leave them than go to the bother of deleting them */
3112 	if(dir_ent->inode && !dir_ent->inode->root_entry)
3113 		dir_ent->inode->nlink --;
3114 
3115 	free(dir_ent);
3116 }
3117 
3118 
add_excluded(struct dir_info * dir)3119 inline void add_excluded(struct dir_info *dir)
3120 {
3121 	dir->excluded ++;
3122 }
3123 
3124 
dir_scan(squashfs_inode * inode,char * pathname,struct dir_ent * (_readdir)(struct dir_info *),int progress)3125 void dir_scan(squashfs_inode *inode, char *pathname,
3126 	struct dir_ent *(_readdir)(struct dir_info *), int progress)
3127 {
3128 	struct stat buf;
3129 	struct dir_ent *dir_ent;
3130 
3131 	root_dir = dir_scan1(pathname, "", paths, _readdir, 1);
3132 	if(root_dir == NULL)
3133 		return;
3134 
3135 	/* Create root directory dir_ent and associated inode, and connect
3136 	 * it to the root directory dir_info structure */
3137 	dir_ent = create_dir_entry("", NULL, pathname,
3138 						scan1_opendir("", "", 0));
3139 
3140 	if(pathname[0] == '\0') {
3141 		/*
3142  		 * dummy top level directory, if multiple sources specified on
3143 		 * command line
3144 		 */
3145 		memset(&buf, 0, sizeof(buf));
3146 		buf.st_mode = S_IRWXU | S_IRWXG | S_IRWXO | S_IFDIR;
3147 		buf.st_uid = getuid();
3148 		buf.st_gid = getgid();
3149 		buf.st_mtime = time(NULL);
3150 		buf.st_dev = 0;
3151 		buf.st_ino = 0;
3152 		dir_ent->inode = lookup_inode2(&buf, PSEUDO_FILE_OTHER, 0);
3153 	} else {
3154 		if(lstat(pathname, &buf) == -1)
3155 			/* source directory has disappeared? */
3156 			BAD_ERROR("Cannot stat source directory %s because %s\n",
3157 				pathname, strerror(errno));
3158 /* ANDROID CHANGES START*/
3159 #ifdef ANDROID
3160 		if (android_config)
3161 			if (mount_point)
3162 				android_fs_config(mount_point, &buf, target_out_path);
3163 			else
3164 				android_fs_config(pathname, &buf, target_out_path);
3165 #endif
3166 /* ANDROID CHANGES END */
3167 		dir_ent->inode = lookup_inode(&buf);
3168 	}
3169 
3170 	dir_ent->dir = root_dir;
3171 	root_dir->dir_ent = dir_ent;
3172 
3173 	/*
3174 	 * Process most actions and any pseudo files
3175 	 */
3176 	if(actions() || get_pseudo())
3177 		dir_scan2(root_dir, get_pseudo());
3178 
3179 	/*
3180 	 * Process move actions
3181 	 */
3182 	if(move_actions()) {
3183 		dir_scan3(root_dir);
3184 		do_move_actions();
3185 	}
3186 
3187 	/*
3188 	 * Process prune actions
3189 	 */
3190 	if(prune_actions())
3191 		dir_scan4(root_dir);
3192 
3193 	/*
3194 	 * Process empty actions
3195 	 */
3196 	if(empty_actions())
3197 		dir_scan5(root_dir);
3198 
3199  	/*
3200 	 * Sort directories and compute the inode numbers
3201 	 */
3202 	dir_scan6(root_dir);
3203 
3204 	alloc_inode_no(dir_ent->inode, root_inode_number);
3205 
3206 	eval_actions(root_dir, dir_ent);
3207 
3208 	if(sorted)
3209 		generate_file_priorities(root_dir, 0,
3210 			&root_dir->dir_ent->inode->buf);
3211 
3212 	if(appending) {
3213 		sigset_t sigmask;
3214 
3215 		restore_thread = init_restore_thread();
3216 		sigemptyset(&sigmask);
3217 		sigaddset(&sigmask, SIGINT);
3218 		sigaddset(&sigmask, SIGTERM);
3219 		sigaddset(&sigmask, SIGUSR1);
3220 		if(pthread_sigmask(SIG_BLOCK, &sigmask, NULL) == -1)
3221 			BAD_ERROR("Failed to set signal mask\n");
3222 		write_destination(fd, SQUASHFS_START, 4, "\0\0\0\0");
3223 	}
3224 
3225 	queue_put(to_reader, root_dir);
3226 
3227 	set_progressbar_state(progress);
3228 
3229 	if(sorted)
3230 		sort_files_and_write(root_dir);
3231 
3232 	dir_scan7(inode, root_dir);
3233 	dir_ent->inode->inode = *inode;
3234 	dir_ent->inode->type = SQUASHFS_DIR_TYPE;
3235 }
3236 
3237 
3238 /*
3239  * dir_scan1 routines...
3240  * These scan the source directories into memory for processing.
3241  * Exclude actions are processed here (in contrast to the other actions)
3242  * because they affect what is scanned.
3243  */
scan1_opendir(char * pathname,char * subpath,int depth)3244 struct dir_info *scan1_opendir(char *pathname, char *subpath, int depth)
3245 {
3246 	struct dir_info *dir;
3247 
3248 	dir = malloc(sizeof(struct dir_info));
3249 	if(dir == NULL)
3250 		MEM_ERROR();
3251 
3252 	if(pathname[0] != '\0') {
3253 		dir->linuxdir = opendir(pathname);
3254 		if(dir->linuxdir == NULL) {
3255 			free(dir);
3256 			return NULL;
3257 		}
3258 	}
3259 
3260 	dir->pathname = strdup(pathname);
3261 	dir->subpath = strdup(subpath);
3262 	dir->count = 0;
3263 	dir->directory_count = 0;
3264 	dir->dir_is_ldir = TRUE;
3265 	dir->list = NULL;
3266 	dir->depth = depth;
3267 	dir->excluded = 0;
3268 
3269 	return dir;
3270 }
3271 
3272 
scan1_encomp_readdir(struct dir_info * dir)3273 struct dir_ent *scan1_encomp_readdir(struct dir_info *dir)
3274 {
3275 	static int index = 0;
3276 
3277 	if(dir->count < old_root_entries) {
3278 		int i;
3279 
3280 		for(i = 0; i < old_root_entries; i++) {
3281 			if(old_root_entry[i].inode.type == SQUASHFS_DIR_TYPE)
3282 				dir->directory_count ++;
3283 			add_dir_entry2(old_root_entry[i].name, NULL, NULL, NULL,
3284 				&old_root_entry[i].inode, dir);
3285 		}
3286 	}
3287 
3288 	while(index < source) {
3289 		char *basename = NULL;
3290 		char *dir_name = getbase(source_path[index]);
3291 		int pass = 1, res;
3292 
3293 		if(dir_name == NULL) {
3294 			ERROR_START("Bad source directory %s",
3295 				source_path[index]);
3296 			ERROR_EXIT(" - skipping ...\n");
3297 			index ++;
3298 			continue;
3299 		}
3300 		dir_name = strdup(dir_name);
3301 		for(;;) {
3302 			struct dir_ent *dir_ent = dir->list;
3303 
3304 			for(; dir_ent && strcmp(dir_ent->name, dir_name) != 0;
3305 				dir_ent = dir_ent->next);
3306 			if(dir_ent == NULL)
3307 				break;
3308 			ERROR("Source directory entry %s already used! - trying"
3309 				" ", dir_name);
3310 			if(pass == 1)
3311 				basename = dir_name;
3312 			else
3313 				free(dir_name);
3314 			res = asprintf(&dir_name, "%s_%d", basename, pass++);
3315 			if(res == -1)
3316 				BAD_ERROR("asprintf failed in "
3317 					"scan1_encomp_readdir\n");
3318 			ERROR("%s\n", dir_name);
3319 		}
3320 		return create_dir_entry(dir_name, basename,
3321 			strdup(source_path[index ++]), dir);
3322 	}
3323 	return NULL;
3324 }
3325 
3326 
scan1_single_readdir(struct dir_info * dir)3327 struct dir_ent *scan1_single_readdir(struct dir_info *dir)
3328 {
3329 	struct dirent *d_name;
3330 	int i;
3331 
3332 	if(dir->count < old_root_entries) {
3333 		for(i = 0; i < old_root_entries; i++) {
3334 			if(old_root_entry[i].inode.type == SQUASHFS_DIR_TYPE)
3335 				dir->directory_count ++;
3336 			add_dir_entry2(old_root_entry[i].name, NULL, NULL, NULL,
3337 				&old_root_entry[i].inode, dir);
3338 		}
3339 	}
3340 
3341 	if((d_name = readdir(dir->linuxdir)) != NULL) {
3342 		char *basename = NULL;
3343 		char *dir_name = strdup(d_name->d_name);
3344 		int pass = 1, res;
3345 
3346 		for(;;) {
3347 			struct dir_ent *dir_ent = dir->list;
3348 
3349 			for(; dir_ent && strcmp(dir_ent->name, dir_name) != 0;
3350 				dir_ent = dir_ent->next);
3351 			if(dir_ent == NULL)
3352 				break;
3353 			ERROR("Source directory entry %s already used! - trying"
3354 				" ", dir_name);
3355 			if (pass == 1)
3356 				basename = dir_name;
3357 			else
3358 				free(dir_name);
3359 			res = asprintf(&dir_name, "%s_%d", d_name->d_name, pass++);
3360 			if(res == -1)
3361 				BAD_ERROR("asprintf failed in "
3362 					"scan1_single_readdir\n");
3363 			ERROR("%s\n", dir_name);
3364 		}
3365 		return create_dir_entry(dir_name, basename, NULL, dir);
3366 	}
3367 
3368 	return NULL;
3369 }
3370 
3371 
scan1_readdir(struct dir_info * dir)3372 struct dir_ent *scan1_readdir(struct dir_info *dir)
3373 {
3374 	struct dirent *d_name = readdir(dir->linuxdir);
3375 
3376 	return d_name ?
3377 		create_dir_entry(strdup(d_name->d_name), NULL, NULL, dir) :
3378 		NULL;
3379 }
3380 
3381 
scan1_freedir(struct dir_info * dir)3382 void scan1_freedir(struct dir_info *dir)
3383 {
3384 	if(dir->pathname[0] != '\0')
3385 		closedir(dir->linuxdir);
3386 }
3387 
3388 
dir_scan1(char * filename,char * subpath,struct pathnames * paths,struct dir_ent * (_readdir)(struct dir_info *),int depth)3389 struct dir_info *dir_scan1(char *filename, char *subpath,
3390 	struct pathnames *paths,
3391 	struct dir_ent *(_readdir)(struct dir_info *), int depth)
3392 {
3393 	struct dir_info *dir = scan1_opendir(filename, subpath, depth);
3394 	struct dir_ent *dir_ent;
3395 
3396 	if(dir == NULL) {
3397 		ERROR_START("Could not open %s", filename);
3398 		ERROR_EXIT(", skipping...\n");
3399 		return NULL;
3400 	}
3401 
3402 	while((dir_ent = _readdir(dir))) {
3403 		struct dir_info *sub_dir;
3404 		struct stat buf;
3405 		struct pathnames *new = NULL;
3406 		char *filename = pathname(dir_ent);
3407 		char *subpath = NULL;
3408 		char *dir_name = dir_ent->name;
3409 
3410 		if(strcmp(dir_name, ".") == 0 || strcmp(dir_name, "..") == 0) {
3411 			free_dir_entry(dir_ent);
3412 			continue;
3413 		}
3414 
3415 		if(lstat(filename, &buf) == -1) {
3416 			ERROR_START("Cannot stat dir/file %s because %s",
3417 				filename, strerror(errno));
3418 			ERROR_EXIT(", ignoring\n");
3419 			free_dir_entry(dir_ent);
3420 			continue;
3421 		}
3422 
3423 		if((buf.st_mode & S_IFMT) != S_IFREG &&
3424 					(buf.st_mode & S_IFMT) != S_IFDIR &&
3425 					(buf.st_mode & S_IFMT) != S_IFLNK &&
3426 					(buf.st_mode & S_IFMT) != S_IFCHR &&
3427 					(buf.st_mode & S_IFMT) != S_IFBLK &&
3428 					(buf.st_mode & S_IFMT) != S_IFIFO &&
3429 					(buf.st_mode & S_IFMT) != S_IFSOCK) {
3430 			ERROR_START("File %s has unrecognised filetype %d",
3431 				filename, buf.st_mode & S_IFMT);
3432 			ERROR_EXIT(", ignoring\n");
3433 			free_dir_entry(dir_ent);
3434 			continue;
3435 		}
3436 
3437 		if((old_exclude && old_excluded(filename, &buf)) ||
3438 			(!old_exclude && excluded(dir_name, paths, &new))) {
3439 			add_excluded(dir);
3440 			free_dir_entry(dir_ent);
3441 			continue;
3442 		}
3443 
3444 		if(exclude_actions()) {
3445 			subpath = subpathname(dir_ent);
3446 
3447 			if(eval_exclude_actions(dir_name, filename, subpath,
3448 							&buf, depth, dir_ent)) {
3449 				add_excluded(dir);
3450 				free_dir_entry(dir_ent);
3451 				continue;
3452 			}
3453 		}
3454 
3455 		switch(buf.st_mode & S_IFMT) {
3456 		case S_IFDIR:
3457 			if(subpath == NULL)
3458 				subpath = subpathname(dir_ent);
3459 
3460 			sub_dir = dir_scan1(filename, subpath, new,
3461 					scan1_readdir, depth + 1);
3462 			if(sub_dir) {
3463 				dir->directory_count ++;
3464 				add_dir_entry(dir_ent, sub_dir,
3465 							lookup_inode(&buf));
3466 			} else
3467 				free_dir_entry(dir_ent);
3468 			break;
3469 		case S_IFLNK: {
3470 			int byte;
3471 			static char buff[65536]; /* overflow safe */
3472 
3473 			byte = readlink(filename, buff, 65536);
3474 			if(byte == -1) {
3475 				ERROR_START("Failed to read symlink %s",
3476 								filename);
3477 				ERROR_EXIT(", ignoring\n");
3478 			} else if(byte == 65536) {
3479 				ERROR_START("Symlink %s is greater than 65536 "
3480 							"bytes!", filename);
3481 				ERROR_EXIT(", ignoring\n");
3482 			} else {
3483 				/* readlink doesn't 0 terminate the returned
3484 				 * path */
3485 				buff[byte] = '\0';
3486 				add_dir_entry(dir_ent, NULL, lookup_inode3(&buf,
3487 							 0, 0, buff, byte + 1));
3488 			}
3489 			break;
3490 		}
3491 		default:
3492 			add_dir_entry(dir_ent, NULL, lookup_inode(&buf));
3493 		}
3494 
3495 		free(new);
3496 	}
3497 
3498 	scan1_freedir(dir);
3499 
3500 	return dir;
3501 }
3502 
3503 
3504 /*
3505  * dir_scan2 routines...
3506  * This processes most actions and any pseudo files
3507  */
scan2_readdir(struct dir_info * dir,struct dir_ent * dir_ent)3508 struct dir_ent *scan2_readdir(struct dir_info *dir, struct dir_ent *dir_ent)
3509 {
3510 	if (dir_ent == NULL)
3511 		dir_ent = dir->list;
3512 	else
3513 		dir_ent = dir_ent->next;
3514 
3515 	for(; dir_ent && dir_ent->inode->root_entry; dir_ent = dir_ent->next);
3516 
3517 	return dir_ent;
3518 }
3519 
3520 
scan2_lookup(struct dir_info * dir,char * name)3521 struct dir_ent *scan2_lookup(struct dir_info *dir, char *name)
3522 {
3523 	struct dir_ent *dir_ent = dir->list;
3524 
3525 	for(; dir_ent && strcmp(dir_ent->name, name) != 0;
3526 					dir_ent = dir_ent->next);
3527 
3528 	return dir_ent;
3529 }
3530 
3531 
dir_scan2(struct dir_info * dir,struct pseudo * pseudo)3532 void dir_scan2(struct dir_info *dir, struct pseudo *pseudo)
3533 {
3534 	struct dir_ent *dir_ent = NULL;
3535 	struct pseudo_entry *pseudo_ent;
3536 	struct stat buf;
3537 	static int pseudo_ino = 1;
3538 
3539 	while((dir_ent = scan2_readdir(dir, dir_ent)) != NULL) {
3540 		struct inode_info *inode_info = dir_ent->inode;
3541 		struct stat *buf = &inode_info->buf;
3542 		char *name = dir_ent->name;
3543 
3544 		eval_actions(root_dir, dir_ent);
3545 
3546 		if((buf->st_mode & S_IFMT) == S_IFDIR)
3547 			dir_scan2(dir_ent->dir, pseudo_subdir(name, pseudo));
3548 	}
3549 
3550 	while((pseudo_ent = pseudo_readdir(pseudo)) != NULL) {
3551 		dir_ent = scan2_lookup(dir, pseudo_ent->name);
3552 		if(pseudo_ent->dev->type == 'm') {
3553 			struct stat *buf;
3554 			if(dir_ent == NULL) {
3555 				ERROR_START("Pseudo modify file \"%s\" does "
3556 					"not exist in source filesystem.",
3557 					pseudo_ent->pathname);
3558 				ERROR_EXIT("  Ignoring.\n");
3559 				continue;
3560 			}
3561 			if(dir_ent->inode->root_entry) {
3562 				ERROR_START("Pseudo modify file \"%s\" is a "
3563 					"pre-existing file in the filesystem "
3564 					"being appended to.  It cannot be "\
3565 					"modified.", pseudo_ent->pathname);
3566 				ERROR_EXIT("  Ignoring.\n");
3567 				continue;
3568 			}
3569 			buf = &dir_ent->inode->buf;
3570 			buf->st_mode = (buf->st_mode & S_IFMT) |
3571 				pseudo_ent->dev->mode;
3572 			buf->st_uid = pseudo_ent->dev->uid;
3573 			buf->st_gid = pseudo_ent->dev->gid;
3574 			continue;
3575 		}
3576 
3577 		if(dir_ent) {
3578 			if(dir_ent->inode->root_entry) {
3579 				ERROR_START("Pseudo file \"%s\" is a "
3580 					"pre-existing file in the filesystem "
3581 					"being appended to.",
3582 					pseudo_ent->pathname);
3583 				ERROR_EXIT("  Ignoring.\n");
3584 			} else {
3585 				ERROR_START("Pseudo file \"%s\" exists in "
3586 					"source filesystem \"%s\".",
3587 					pseudo_ent->pathname,
3588 					pathname(dir_ent));
3589 				ERROR_EXIT("\nIgnoring, exclude it (-e/-ef) to "
3590 					"override.\n");
3591 			}
3592 			continue;
3593 		}
3594 
3595 		memset(&buf, 0, sizeof(buf));
3596 		buf.st_mode = pseudo_ent->dev->mode;
3597 		buf.st_uid = pseudo_ent->dev->uid;
3598 		buf.st_gid = pseudo_ent->dev->gid;
3599 		buf.st_rdev = makedev(pseudo_ent->dev->major,
3600 			pseudo_ent->dev->minor);
3601 		buf.st_mtime = time(NULL);
3602 		buf.st_ino = pseudo_ino ++;
3603 
3604 		if(pseudo_ent->dev->type == 'd') {
3605 			struct dir_ent *dir_ent =
3606 				create_dir_entry(pseudo_ent->name, NULL,
3607 						pseudo_ent->pathname, dir);
3608 			char *subpath = strdup(subpathname(dir_ent));
3609 			struct dir_info *sub_dir = scan1_opendir("", subpath,
3610 						dir->depth + 1);
3611 			if(sub_dir == NULL) {
3612 				ERROR_START("Could not create pseudo directory "
3613 					"\"%s\"", pseudo_ent->pathname);
3614 				ERROR_EXIT(", skipping...\n");
3615 				free(subpath);
3616 				pseudo_ino --;
3617 				continue;
3618 			}
3619 			dir_scan2(sub_dir, pseudo_ent->pseudo);
3620 			dir->directory_count ++;
3621 			add_dir_entry(dir_ent, sub_dir,
3622 				lookup_inode2(&buf, PSEUDO_FILE_OTHER, 0));
3623 		} else if(pseudo_ent->dev->type == 'f') {
3624 			add_dir_entry2(pseudo_ent->name, NULL,
3625 				pseudo_ent->pathname, NULL,
3626 				lookup_inode2(&buf, PSEUDO_FILE_PROCESS,
3627 				pseudo_ent->dev->pseudo_id), dir);
3628 		} else {
3629 			add_dir_entry2(pseudo_ent->name, NULL,
3630 				pseudo_ent->pathname, NULL,
3631 				lookup_inode2(&buf, PSEUDO_FILE_OTHER, 0), dir);
3632 		}
3633 	}
3634 }
3635 
3636 
3637 /*
3638  * dir_scan3 routines...
3639  * This processes the move action
3640  */
dir_scan3(struct dir_info * dir)3641 void dir_scan3(struct dir_info *dir)
3642 {
3643 	struct dir_ent *dir_ent = NULL;
3644 
3645 	while((dir_ent = scan2_readdir(dir, dir_ent)) != NULL) {
3646 
3647 		eval_move_actions(root_dir, dir_ent);
3648 
3649 		if((dir_ent->inode->buf.st_mode & S_IFMT) == S_IFDIR)
3650 			dir_scan3(dir_ent->dir);
3651 	}
3652 }
3653 
3654 
3655 /*
3656  * dir_scan4 routines...
3657  * This processes the prune action.  This action is designed to do fine
3658  * grained tuning of the in-core directory structure after the exclude,
3659  * move and pseudo actions have been performed.  This allows complex
3660  * tests to be performed which are impossible at exclude time (i.e.
3661  * tests which rely on the in-core directory structure)
3662  */
free_dir(struct dir_info * dir)3663 void free_dir(struct dir_info *dir)
3664 {
3665 	struct dir_ent *dir_ent = dir->list;
3666 
3667 	while(dir_ent) {
3668 		struct dir_ent *tmp = dir_ent;
3669 
3670 		if((dir_ent->inode->buf.st_mode & S_IFMT) == S_IFDIR)
3671 			free_dir(dir_ent->dir);
3672 
3673 		dir_ent = dir_ent->next;
3674 		free_dir_entry(tmp);
3675 	}
3676 
3677 	free(dir->pathname);
3678 	free(dir->subpath);
3679 	free(dir);
3680 }
3681 
3682 
dir_scan4(struct dir_info * dir)3683 void dir_scan4(struct dir_info *dir)
3684 {
3685 	struct dir_ent *dir_ent = dir->list, *prev = NULL;
3686 
3687 	while(dir_ent) {
3688 		if(dir_ent->inode->root_entry) {
3689 			prev = dir_ent;
3690 			dir_ent = dir_ent->next;
3691 			continue;
3692 		}
3693 
3694 		if((dir_ent->inode->buf.st_mode & S_IFMT) == S_IFDIR)
3695 			dir_scan4(dir_ent->dir);
3696 
3697 		if(eval_prune_actions(root_dir, dir_ent)) {
3698 			struct dir_ent *tmp = dir_ent;
3699 
3700 			if((dir_ent->inode->buf.st_mode & S_IFMT) == S_IFDIR) {
3701 				free_dir(dir_ent->dir);
3702 				dir->directory_count --;
3703 			}
3704 
3705 			dir->count --;
3706 
3707 			/* remove dir_ent from list */
3708 			dir_ent = dir_ent->next;
3709 			if(prev)
3710 				prev->next = dir_ent;
3711 			else
3712 				dir->list = dir_ent;
3713 
3714 			/* free it */
3715 			free_dir_entry(tmp);
3716 
3717 			add_excluded(dir);
3718 			continue;
3719 		}
3720 
3721 		prev = dir_ent;
3722 		dir_ent = dir_ent->next;
3723 	}
3724 }
3725 
3726 
3727 /*
3728  * dir_scan5 routines...
3729  * This processes the empty action.  This action has to be processed after
3730  * all other actions because the previous exclude and move actions and the
3731  * pseudo actions affect whether a directory is empty
3732  */
dir_scan5(struct dir_info * dir)3733 void dir_scan5(struct dir_info *dir)
3734 {
3735 	struct dir_ent *dir_ent = dir->list, *prev = NULL;
3736 
3737 	while(dir_ent) {
3738 		if(dir_ent->inode->root_entry) {
3739 			prev = dir_ent;
3740 			dir_ent = dir_ent->next;
3741 			continue;
3742 		}
3743 
3744 		if((dir_ent->inode->buf.st_mode & S_IFMT) == S_IFDIR) {
3745 			dir_scan5(dir_ent->dir);
3746 
3747 			if(eval_empty_actions(root_dir, dir_ent)) {
3748 				struct dir_ent *tmp = dir_ent;
3749 
3750 				/*
3751 				 * delete sub-directory, this is by definition
3752 				 * empty
3753 				 */
3754 				free(dir_ent->dir->pathname);
3755 				free(dir_ent->dir->subpath);
3756 				free(dir_ent->dir);
3757 
3758 				/* remove dir_ent from list */
3759 				dir_ent = dir_ent->next;
3760 				if(prev)
3761 					prev->next = dir_ent;
3762 				else
3763 					dir->list = dir_ent;
3764 
3765 				/* free it */
3766 				free_dir_entry(tmp);
3767 
3768 				/* update counts */
3769 				dir->directory_count --;
3770 				dir->count --;
3771 				add_excluded(dir);
3772 				continue;
3773 			}
3774 		}
3775 
3776 		prev = dir_ent;
3777 		dir_ent = dir_ent->next;
3778 	}
3779 }
3780 
3781 
3782 /*
3783  * dir_scan6 routines...
3784  * This sorts every directory and computes the inode numbers
3785  */
3786 
3787 /*
3788  * Bottom up linked list merge sort.
3789  *
3790  * Qsort and other O(n log n) algorithms work well with arrays but not
3791  * linked lists.  Merge sort another O(n log n) sort algorithm on the other hand
3792  * is not ideal for arrays (as it needs an additonal n storage locations
3793  * as sorting is not done in place), but it is ideal for linked lists because
3794  * it doesn't require any extra storage,
3795  */
sort_directory(struct dir_info * dir)3796 void sort_directory(struct dir_info *dir)
3797 {
3798 	struct dir_ent *cur, *l1, *l2, *next;
3799 	int len1, len2, stride = 1;
3800 
3801 	if(dir->list == NULL || dir->count < 2)
3802 		return;
3803 
3804 	/*
3805 	 * We can consider our linked-list to be made up of stride length
3806 	 * sublists.  Eacn iteration around this loop merges adjacent
3807 	 * stride length sublists into larger 2*stride sublists.  We stop
3808 	 * when stride becomes equal to the entire list.
3809 	 *
3810 	 * Initially stride = 1 (by definition a sublist of 1 is sorted), and
3811 	 * these 1 element sublists are merged into 2 element sublists,  which
3812 	 * are then merged into 4 element sublists and so on.
3813 	 */
3814 	do {
3815 		l2 = dir->list; /* head of current linked list */
3816 		cur = NULL; /* empty output list */
3817 
3818 		/*
3819 		 * Iterate through the linked list, merging adjacent sublists.
3820 		 * On each interation l2 points to the next sublist pair to be
3821 		 * merged (if there's only one sublist left this is simply added
3822 		 * to the output list)
3823 		 */
3824 		while(l2) {
3825 			l1 = l2;
3826 			for(len1 = 0; l2 && len1 < stride; len1 ++, l2 = l2->next);
3827 			len2 = stride;
3828 
3829 			/*
3830 			 * l1 points to first sublist.
3831 			 * l2 points to second sublist.
3832 			 * Merge them onto the output list
3833 			 */
3834 			while(len1 && l2 && len2) {
3835 				if(strcmp(l1->name, l2->name) <= 0) {
3836 					next = l1;
3837 					l1 = l1->next;
3838 					len1 --;
3839 				} else {
3840 					next = l2;
3841 					l2 = l2->next;
3842 					len2 --;
3843 				}
3844 
3845 				if(cur) {
3846 					cur->next = next;
3847 					cur = next;
3848 				} else
3849 					dir->list = cur = next;
3850 			}
3851 			/*
3852 			 * One sublist is now empty, copy the other one onto the
3853 			 * output list
3854 			 */
3855 			for(; len1; len1 --, l1 = l1->next) {
3856 				if(cur) {
3857 					cur->next = l1;
3858 					cur = l1;
3859 				} else
3860 					dir->list = cur = l1;
3861 			}
3862 			for(; l2 && len2; len2 --, l2 = l2->next) {
3863 				if(cur) {
3864 					cur->next = l2;
3865 					cur = l2;
3866 				} else
3867 					dir->list = cur = l2;
3868 			}
3869 		}
3870 		cur->next = NULL;
3871 		stride = stride << 1;
3872 	} while(stride < dir->count);
3873 }
3874 
3875 
dir_scan6(struct dir_info * dir)3876 void dir_scan6(struct dir_info *dir)
3877 {
3878 	struct dir_ent *dir_ent;
3879 	unsigned int byte_count = 0;
3880 
3881 	sort_directory(dir);
3882 
3883 	for(dir_ent = dir->list; dir_ent; dir_ent = dir_ent->next) {
3884 		byte_count += strlen(dir_ent->name) +
3885 			sizeof(struct squashfs_dir_entry);
3886 
3887 		if(dir_ent->inode->root_entry)
3888 			continue;
3889 
3890 		alloc_inode_no(dir_ent->inode, 0);
3891 
3892 		if((dir_ent->inode->buf.st_mode & S_IFMT) == S_IFDIR)
3893 			dir_scan6(dir_ent->dir);
3894 	}
3895 
3896 	if((dir->count < 257 && byte_count < SQUASHFS_METADATA_SIZE))
3897 		dir->dir_is_ldir = FALSE;
3898 }
3899 
3900 
3901 /*
3902  * dir_scan6 routines...
3903  * This generates the filesystem metadata and writes it out to the destination
3904  */
scan7_init_dir(struct directory * dir)3905 void scan7_init_dir(struct directory *dir)
3906 {
3907 	dir->buff = malloc(SQUASHFS_METADATA_SIZE);
3908 	if(dir->buff == NULL)
3909 		MEM_ERROR();
3910 
3911 	dir->size = SQUASHFS_METADATA_SIZE;
3912 	dir->p = dir->index_count_p = dir->buff;
3913 	dir->entry_count = 256;
3914 	dir->entry_count_p = NULL;
3915 	dir->index = NULL;
3916 	dir->i_count = dir->i_size = 0;
3917 }
3918 
3919 
scan7_readdir(struct directory * dir,struct dir_info * dir_info,struct dir_ent * dir_ent)3920 struct dir_ent *scan7_readdir(struct directory *dir, struct dir_info *dir_info,
3921 	struct dir_ent *dir_ent)
3922 {
3923 	if (dir_ent == NULL)
3924 		dir_ent = dir_info->list;
3925 	else
3926 		dir_ent = dir_ent->next;
3927 
3928 	for(; dir_ent && dir_ent->inode->root_entry; dir_ent = dir_ent->next)
3929 		add_dir(dir_ent->inode->inode, dir_ent->inode->inode_number,
3930 			dir_ent->name, dir_ent->inode->type, dir);
3931 
3932 	return dir_ent;
3933 }
3934 
3935 
scan7_freedir(struct directory * dir)3936 void scan7_freedir(struct directory *dir)
3937 {
3938 	if(dir->index)
3939 		free(dir->index);
3940 	free(dir->buff);
3941 }
3942 
3943 
dir_scan7(squashfs_inode * inode,struct dir_info * dir_info)3944 void dir_scan7(squashfs_inode *inode, struct dir_info *dir_info)
3945 {
3946 	int squashfs_type;
3947 	int duplicate_file;
3948 	struct directory dir;
3949 	struct dir_ent *dir_ent = NULL;
3950 
3951 	scan7_init_dir(&dir);
3952 
3953 	while((dir_ent = scan7_readdir(&dir, dir_info, dir_ent)) != NULL) {
3954 		struct stat *buf = &dir_ent->inode->buf;
3955 
3956 		update_info(dir_ent);
3957 
3958 		if(dir_ent->inode->inode == SQUASHFS_INVALID_BLK) {
3959 			switch(buf->st_mode & S_IFMT) {
3960 				case S_IFREG:
3961 					squashfs_type = SQUASHFS_FILE_TYPE;
3962 					write_file(inode, dir_ent,
3963 						&duplicate_file);
3964 					INFO("file %s, uncompressed size %lld "
3965 						"bytes %s\n",
3966 						subpathname(dir_ent),
3967 						(long long) buf->st_size,
3968 						duplicate_file ?  "DUPLICATE" :
3969 						 "");
3970 					break;
3971 
3972 				case S_IFDIR:
3973 					squashfs_type = SQUASHFS_DIR_TYPE;
3974 					dir_scan7(inode, dir_ent->dir);
3975 					break;
3976 
3977 				case S_IFLNK:
3978 					squashfs_type = SQUASHFS_SYMLINK_TYPE;
3979 					create_inode(inode, NULL, dir_ent,
3980 						squashfs_type, 0, 0, 0, NULL,
3981 						NULL, NULL, 0);
3982 					INFO("symbolic link %s inode 0x%llx\n",
3983 						subpathname(dir_ent), *inode);
3984 					sym_count ++;
3985 					break;
3986 
3987 				case S_IFCHR:
3988 					squashfs_type = SQUASHFS_CHRDEV_TYPE;
3989 					create_inode(inode, NULL, dir_ent,
3990 						squashfs_type, 0, 0, 0, NULL,
3991 						NULL, NULL, 0);
3992 					INFO("character device %s inode 0x%llx"
3993 						"\n", subpathname(dir_ent),
3994 						*inode);
3995 					dev_count ++;
3996 					break;
3997 
3998 				case S_IFBLK:
3999 					squashfs_type = SQUASHFS_BLKDEV_TYPE;
4000 					create_inode(inode, NULL, dir_ent,
4001 						squashfs_type, 0, 0, 0, NULL,
4002 						NULL, NULL, 0);
4003 					INFO("block device %s inode 0x%llx\n",
4004 						subpathname(dir_ent), *inode);
4005 					dev_count ++;
4006 					break;
4007 
4008 				case S_IFIFO:
4009 					squashfs_type = SQUASHFS_FIFO_TYPE;
4010 					create_inode(inode, NULL, dir_ent,
4011 						squashfs_type, 0, 0, 0, NULL,
4012 						NULL, NULL, 0);
4013 					INFO("fifo %s inode 0x%llx\n",
4014 						subpathname(dir_ent), *inode);
4015 					fifo_count ++;
4016 					break;
4017 
4018 				case S_IFSOCK:
4019 					squashfs_type = SQUASHFS_SOCKET_TYPE;
4020 					create_inode(inode, NULL, dir_ent,
4021 						squashfs_type, 0, 0, 0, NULL,
4022 						NULL, NULL, 0);
4023 					INFO("unix domain socket %s inode "
4024 						"0x%llx\n",
4025 						subpathname(dir_ent), *inode);
4026 					sock_count ++;
4027 					break;
4028 
4029 				default:
4030 					BAD_ERROR("%s unrecognised file type, "
4031 						"mode is %x\n",
4032 						subpathname(dir_ent),
4033 						buf->st_mode);
4034 			}
4035 			dir_ent->inode->inode = *inode;
4036 			dir_ent->inode->type = squashfs_type;
4037 		 } else {
4038 			*inode = dir_ent->inode->inode;
4039 			squashfs_type = dir_ent->inode->type;
4040 			switch(squashfs_type) {
4041 				case SQUASHFS_FILE_TYPE:
4042 					if(!sorted)
4043 						INFO("file %s, uncompressed "
4044 							"size %lld bytes LINK"
4045 							"\n",
4046 							subpathname(dir_ent),
4047 							(long long)
4048 							buf->st_size);
4049 					break;
4050 				case SQUASHFS_SYMLINK_TYPE:
4051 					INFO("symbolic link %s inode 0x%llx "
4052 						"LINK\n", subpathname(dir_ent),
4053 						 *inode);
4054 					break;
4055 				case SQUASHFS_CHRDEV_TYPE:
4056 					INFO("character device %s inode 0x%llx "
4057 						"LINK\n", subpathname(dir_ent),
4058 						*inode);
4059 					break;
4060 				case SQUASHFS_BLKDEV_TYPE:
4061 					INFO("block device %s inode 0x%llx "
4062 						"LINK\n", subpathname(dir_ent),
4063 						*inode);
4064 					break;
4065 				case SQUASHFS_FIFO_TYPE:
4066 					INFO("fifo %s inode 0x%llx LINK\n",
4067 						subpathname(dir_ent), *inode);
4068 					break;
4069 				case SQUASHFS_SOCKET_TYPE:
4070 					INFO("unix domain socket %s inode "
4071 						"0x%llx LINK\n",
4072 						subpathname(dir_ent), *inode);
4073 					break;
4074 			}
4075 		}
4076 
4077 		add_dir(*inode, get_inode_no(dir_ent->inode), dir_ent->name,
4078 			squashfs_type, &dir);
4079 	}
4080 
4081 	write_dir(inode, dir_info, &dir);
4082 	INFO("directory %s inode 0x%llx\n", subpathname(dir_info->dir_ent),
4083 		*inode);
4084 
4085 	scan7_freedir(&dir);
4086 }
4087 
4088 
slog(unsigned int block)4089 unsigned int slog(unsigned int block)
4090 {
4091 	int i;
4092 
4093 	for(i = 12; i <= 20; i++)
4094 		if(block == (1 << i))
4095 			return i;
4096 	return 0;
4097 }
4098 
4099 
old_excluded(char * filename,struct stat * buf)4100 int old_excluded(char *filename, struct stat *buf)
4101 {
4102 	int i;
4103 
4104 	for(i = 0; i < exclude; i++)
4105 		if((exclude_paths[i].st_dev == buf->st_dev) &&
4106 				(exclude_paths[i].st_ino == buf->st_ino))
4107 			return TRUE;
4108 	return FALSE;
4109 }
4110 
4111 
4112 #define ADD_ENTRY(buf) \
4113 	if(exclude % EXCLUDE_SIZE == 0) { \
4114 		exclude_paths = realloc(exclude_paths, (exclude + EXCLUDE_SIZE) \
4115 			* sizeof(struct exclude_info)); \
4116 		if(exclude_paths == NULL) \
4117 			MEM_ERROR(); \
4118 	} \
4119 	exclude_paths[exclude].st_dev = buf.st_dev; \
4120 	exclude_paths[exclude++].st_ino = buf.st_ino;
old_add_exclude(char * path)4121 int old_add_exclude(char *path)
4122 {
4123 	int i;
4124 	char *filename;
4125 	struct stat buf;
4126 
4127 	if(path[0] == '/' || strncmp(path, "./", 2) == 0 ||
4128 			strncmp(path, "../", 3) == 0) {
4129 		if(lstat(path, &buf) == -1) {
4130 			ERROR_START("Cannot stat exclude dir/file %s because "
4131 				"%s", path, strerror(errno));
4132 			ERROR_EXIT(", ignoring\n");
4133 			return TRUE;
4134 		}
4135 		ADD_ENTRY(buf);
4136 		return TRUE;
4137 	}
4138 
4139 	for(i = 0; i < source; i++) {
4140 		int res = asprintf(&filename, "%s/%s", source_path[i], path);
4141 		if(res == -1)
4142 			BAD_ERROR("asprintf failed in old_add_exclude\n");
4143 		if(lstat(filename, &buf) == -1) {
4144 			if(!(errno == ENOENT || errno == ENOTDIR)) {
4145 				ERROR_START("Cannot stat exclude dir/file %s "
4146 					"because %s", filename, strerror(errno));
4147 				ERROR_EXIT(", ignoring\n");
4148 			}
4149 			free(filename);
4150 			continue;
4151 		}
4152 		free(filename);
4153 		ADD_ENTRY(buf);
4154 	}
4155 	return TRUE;
4156 }
4157 
4158 
add_old_root_entry(char * name,squashfs_inode inode,int inode_number,int type)4159 void add_old_root_entry(char *name, squashfs_inode inode, int inode_number,
4160 	int type)
4161 {
4162 	old_root_entry = realloc(old_root_entry,
4163 		sizeof(struct old_root_entry_info) * (old_root_entries + 1));
4164 	if(old_root_entry == NULL)
4165 		MEM_ERROR();
4166 
4167 	old_root_entry[old_root_entries].name = strdup(name);
4168 	old_root_entry[old_root_entries].inode.inode = inode;
4169 	old_root_entry[old_root_entries].inode.inode_number = inode_number;
4170 	old_root_entry[old_root_entries].inode.type = type;
4171 	old_root_entry[old_root_entries++].inode.root_entry = TRUE;
4172 }
4173 
4174 
initialise_threads(int readq,int fragq,int bwriteq,int fwriteq,int freelst,char * destination_file)4175 void initialise_threads(int readq, int fragq, int bwriteq, int fwriteq,
4176 	int freelst, char *destination_file)
4177 {
4178 	int i;
4179 	sigset_t sigmask, old_mask;
4180 	int total_mem = readq;
4181 	int reader_size;
4182 	int fragment_size;
4183 	int fwriter_size;
4184 	/*
4185 	 * bwriter_size is global because it is needed in
4186 	 * write_file_blocks_dup()
4187 	 */
4188 
4189 	/*
4190 	 * Never allow the total size of the queues to be larger than
4191 	 * physical memory
4192 	 *
4193 	 * When adding together the possibly user supplied values, make
4194 	 * sure they've not been deliberately contrived to overflow an int
4195 	 */
4196 	if(add_overflow(total_mem, fragq))
4197 		BAD_ERROR("Queue sizes rediculously too large\n");
4198 	total_mem += fragq;
4199 	if(add_overflow(total_mem, bwriteq))
4200 		BAD_ERROR("Queue sizes rediculously too large\n");
4201 	total_mem += bwriteq;
4202 	if(add_overflow(total_mem, fwriteq))
4203 		BAD_ERROR("Queue sizes rediculously too large\n");
4204 	total_mem += fwriteq;
4205 
4206 	check_usable_phys_mem(total_mem);
4207 
4208 	/*
4209 	 * convert from queue size in Mbytes to queue size in
4210 	 * blocks.
4211 	 *
4212 	 * This isn't going to overflow an int unless there exists
4213 	 * systems with more than 8 Petabytes of RAM!
4214 	 */
4215 	reader_size = readq << (20 - block_log);
4216 	fragment_size = fragq << (20 - block_log);
4217 	bwriter_size = bwriteq << (20 - block_log);
4218 	fwriter_size = fwriteq << (20 - block_log);
4219 
4220 	/*
4221 	 * setup signal handlers for the main thread, these cleanup
4222 	 * deleting the destination file, if appending the
4223 	 * handlers for SIGTERM and SIGINT will be replaced with handlers
4224 	 * allowing the user to press ^C twice to restore the existing
4225 	 * filesystem.
4226 	 *
4227 	 * SIGUSR1 is an internal signal, which is used by the sub-threads
4228 	 * to tell the main thread to terminate, deleting the destination file,
4229 	 * or if necessary restoring the filesystem on appending
4230 	 */
4231 	signal(SIGTERM, sighandler);
4232 	signal(SIGINT, sighandler);
4233 	signal(SIGUSR1, sighandler);
4234 
4235 	/* block SIGQUIT and SIGHUP, these are handled by the info thread */
4236 	sigemptyset(&sigmask);
4237 	sigaddset(&sigmask, SIGQUIT);
4238 	sigaddset(&sigmask, SIGHUP);
4239 	if(pthread_sigmask(SIG_BLOCK, &sigmask, NULL) == -1)
4240 		BAD_ERROR("Failed to set signal mask in intialise_threads\n");
4241 
4242 	/*
4243 	 * temporarily block these signals, so the created sub-threads
4244 	 * will ignore them, ensuring the main thread handles them
4245 	 */
4246 	sigemptyset(&sigmask);
4247 	sigaddset(&sigmask, SIGINT);
4248 	sigaddset(&sigmask, SIGTERM);
4249 	sigaddset(&sigmask, SIGUSR1);
4250 	if(pthread_sigmask(SIG_BLOCK, &sigmask, &old_mask) == -1)
4251 		BAD_ERROR("Failed to set signal mask in intialise_threads\n");
4252 
4253 	if(processors == -1) {
4254 #ifndef linux
4255 		int mib[2];
4256 		size_t len = sizeof(processors);
4257 
4258 		mib[0] = CTL_HW;
4259 #ifdef HW_AVAILCPU
4260 		mib[1] = HW_AVAILCPU;
4261 #else
4262 		mib[1] = HW_NCPU;
4263 #endif
4264 
4265 		if(sysctl(mib, 2, &processors, &len, NULL, 0) == -1) {
4266 			ERROR_START("Failed to get number of available "
4267 				"processors.");
4268 			ERROR_EXIT("  Defaulting to 1\n");
4269 			processors = 1;
4270 		}
4271 #else
4272 		processors = sysconf(_SC_NPROCESSORS_ONLN);
4273 #endif
4274 	}
4275 
4276 	if(multiply_overflow(processors, 3) ||
4277 			multiply_overflow(processors * 3, sizeof(pthread_t)))
4278 		BAD_ERROR("Processors too large\n");
4279 
4280 	deflator_thread = malloc(processors * 3 * sizeof(pthread_t));
4281 	if(deflator_thread == NULL)
4282 		MEM_ERROR();
4283 
4284 	frag_deflator_thread = &deflator_thread[processors];
4285 	frag_thread = &frag_deflator_thread[processors];
4286 
4287 	to_reader = queue_init(1);
4288 	to_deflate = queue_init(reader_size);
4289 	to_process_frag = queue_init(reader_size);
4290 	to_writer = queue_init(bwriter_size + fwriter_size);
4291 	from_writer = queue_init(1);
4292 	to_frag = queue_init(fragment_size);
4293 	locked_fragment = queue_init(fragment_size);
4294 	to_main = seq_queue_init();
4295 	reader_buffer = cache_init(block_size, reader_size, 0, 0);
4296 	bwriter_buffer = cache_init(block_size, bwriter_size, 1, freelst);
4297 	fwriter_buffer = cache_init(block_size, fwriter_size, 1, freelst);
4298 	fragment_buffer = cache_init(block_size, fragment_size, 1, 0);
4299 	reserve_cache = cache_init(block_size, processors + 1, 1, 0);
4300 	pthread_create(&reader_thread, NULL, reader, NULL);
4301 	pthread_create(&writer_thread, NULL, writer, NULL);
4302 	init_progress_bar();
4303 	init_info();
4304 
4305 	for(i = 0; i < processors; i++) {
4306 		if(pthread_create(&deflator_thread[i], NULL, deflator, NULL))
4307 			BAD_ERROR("Failed to create thread\n");
4308 		if(pthread_create(&frag_deflator_thread[i], NULL, frag_deflator,
4309 				NULL) != 0)
4310 			BAD_ERROR("Failed to create thread\n");
4311 		if(pthread_create(&frag_thread[i], NULL, frag_thrd,
4312 				(void *) destination_file) != 0)
4313 			BAD_ERROR("Failed to create thread\n");
4314 	}
4315 
4316 	main_thread = pthread_self();
4317 
4318 	printf("Parallel mksquashfs: Using %d processor%s\n", processors,
4319 			processors == 1 ? "" : "s");
4320 
4321 	/* Restore the signal mask for the main thread */
4322 	if(pthread_sigmask(SIG_SETMASK, &old_mask, NULL) == -1)
4323 		BAD_ERROR("Failed to set signal mask in intialise_threads\n");
4324 }
4325 
4326 
write_inode_lookup_table()4327 long long write_inode_lookup_table()
4328 {
4329 	int i, inode_number, lookup_bytes = SQUASHFS_LOOKUP_BYTES(inode_count);
4330 	void *it;
4331 
4332 	if(inode_count == sinode_count)
4333 		goto skip_inode_hash_table;
4334 
4335 	it = realloc(inode_lookup_table, lookup_bytes);
4336 	if(it == NULL)
4337 		MEM_ERROR();
4338 	inode_lookup_table = it;
4339 
4340 	for(i = 0; i < INODE_HASH_SIZE; i ++) {
4341 		struct inode_info *inode;
4342 
4343 		for(inode = inode_info[i]; inode; inode = inode->next) {
4344 
4345 			inode_number = get_inode_no(inode);
4346 
4347 			/* The empty action will produce orphaned inode
4348 			 * entries in the inode_info[] table.  These
4349 			 * entries because they are orphaned will not be
4350 			 * allocated an inode number in dir_scan5(), so
4351 			 * skip any entries with the default dummy inode
4352 			 * number of 0 */
4353 			if(inode_number == 0)
4354 				continue;
4355 
4356 			SQUASHFS_SWAP_LONG_LONGS(&inode->inode,
4357 				&inode_lookup_table[inode_number - 1], 1);
4358 
4359 		}
4360 	}
4361 
4362 skip_inode_hash_table:
4363 	return generic_write_table(lookup_bytes, inode_lookup_table, 0, NULL,
4364 		noI);
4365 }
4366 
4367 
get_component(char * target,char ** targname)4368 char *get_component(char *target, char **targname)
4369 {
4370 	char *start;
4371 
4372 	while(*target == '/')
4373 		target ++;
4374 
4375 	start = target;
4376 	while(*target != '/' && *target != '\0')
4377 		target ++;
4378 
4379 	*targname = strndup(start, target - start);
4380 
4381 	while(*target == '/')
4382 		target ++;
4383 
4384 	return target;
4385 }
4386 
4387 
free_path(struct pathname * paths)4388 void free_path(struct pathname *paths)
4389 {
4390 	int i;
4391 
4392 	for(i = 0; i < paths->names; i++) {
4393 		if(paths->name[i].paths)
4394 			free_path(paths->name[i].paths);
4395 		free(paths->name[i].name);
4396 		if(paths->name[i].preg) {
4397 			regfree(paths->name[i].preg);
4398 			free(paths->name[i].preg);
4399 		}
4400 	}
4401 
4402 	free(paths);
4403 }
4404 
4405 
add_path(struct pathname * paths,char * target,char * alltarget)4406 struct pathname *add_path(struct pathname *paths, char *target, char *alltarget)
4407 {
4408 	char *targname;
4409 	int i, error;
4410 
4411 	target = get_component(target, &targname);
4412 
4413 	if(paths == NULL) {
4414 		paths = malloc(sizeof(struct pathname));
4415 		if(paths == NULL)
4416 			MEM_ERROR();
4417 
4418 		paths->names = 0;
4419 		paths->name = NULL;
4420 	}
4421 
4422 	for(i = 0; i < paths->names; i++)
4423 		if(strcmp(paths->name[i].name, targname) == 0)
4424 			break;
4425 
4426 	if(i == paths->names) {
4427 		/* allocate new name entry */
4428 		paths->names ++;
4429 		paths->name = realloc(paths->name, (i + 1) *
4430 			sizeof(struct path_entry));
4431 		if(paths->name == NULL)
4432 			MEM_ERROR();
4433 		paths->name[i].name = targname;
4434 		paths->name[i].paths = NULL;
4435 		if(use_regex) {
4436 			paths->name[i].preg = malloc(sizeof(regex_t));
4437 			if(paths->name[i].preg == NULL)
4438 				MEM_ERROR();
4439 			error = regcomp(paths->name[i].preg, targname,
4440 				REG_EXTENDED|REG_NOSUB);
4441 			if(error) {
4442 				char str[1024]; /* overflow safe */
4443 
4444 				regerror(error, paths->name[i].preg, str, 1024);
4445 				BAD_ERROR("invalid regex %s in export %s, "
4446 					"because %s\n", targname, alltarget,
4447 					str);
4448 			}
4449 		} else
4450 			paths->name[i].preg = NULL;
4451 
4452 		if(target[0] == '\0')
4453 			/* at leaf pathname component */
4454 			paths->name[i].paths = NULL;
4455 		else
4456 			/* recurse adding child components */
4457 			paths->name[i].paths = add_path(NULL, target,
4458 				alltarget);
4459 	} else {
4460 		/* existing matching entry */
4461 		free(targname);
4462 
4463 		if(paths->name[i].paths == NULL) {
4464 			/* No sub-directory which means this is the leaf
4465 			 * component of a pre-existing exclude which subsumes
4466 			 * the exclude currently being added, in which case stop
4467 			 * adding components */
4468 		} else if(target[0] == '\0') {
4469 			/* at leaf pathname component and child components exist
4470 			 * from more specific excludes, delete as they're
4471 			 * subsumed by this exclude */
4472 			free_path(paths->name[i].paths);
4473 			paths->name[i].paths = NULL;
4474 		} else
4475 			/* recurse adding child components */
4476 			add_path(paths->name[i].paths, target, alltarget);
4477 	}
4478 
4479 	return paths;
4480 }
4481 
4482 
add_exclude(char * target)4483 void add_exclude(char *target)
4484 {
4485 
4486 	if(target[0] == '/' || strncmp(target, "./", 2) == 0 ||
4487 			strncmp(target, "../", 3) == 0)
4488 		BAD_ERROR("/, ./ and ../ prefixed excludes not supported with "
4489 			"-wildcards or -regex options\n");
4490 	else if(strncmp(target, "... ", 4) == 0)
4491 		stickypath = add_path(stickypath, target + 4, target + 4);
4492 	else
4493 		path = add_path(path, target, target);
4494 }
4495 
4496 
display_path(int depth,struct pathname * paths)4497 void display_path(int depth, struct pathname *paths)
4498 {
4499 	int i, n;
4500 
4501 	if(paths == NULL)
4502 		return;
4503 
4504 	for(i = 0; i < paths->names; i++) {
4505 		for(n = 0; n < depth; n++)
4506 			printf("\t");
4507 		printf("%d: %s\n", depth, paths->name[i].name);
4508 		display_path(depth + 1, paths->name[i].paths);
4509 	}
4510 }
4511 
4512 
display_path2(struct pathname * paths,char * string)4513 void display_path2(struct pathname *paths, char *string)
4514 {
4515 	int i;
4516 	char *path;
4517 
4518 	if(paths == NULL) {
4519 		printf("%s\n", string);
4520 		return;
4521 	}
4522 
4523 	for(i = 0; i < paths->names; i++) {
4524 		int res = asprintf(&path, "%s/%s", string, paths->name[i].name);
4525 		if(res == -1)
4526 			BAD_ERROR("asprintf failed in display_path2\n");
4527 		display_path2(paths->name[i].paths, path);
4528 		free(path);
4529 	}
4530 }
4531 
4532 
add_subdir(struct pathnames * paths,struct pathname * path)4533 struct pathnames *add_subdir(struct pathnames *paths, struct pathname *path)
4534 {
4535 	int count = paths == NULL ? 0 : paths->count;
4536 
4537 	if(count % PATHS_ALLOC_SIZE == 0) {
4538 		paths = realloc(paths, sizeof(struct pathnames) +
4539 			(count + PATHS_ALLOC_SIZE) * sizeof(struct pathname *));
4540 		if(paths == NULL)
4541 			MEM_ERROR();
4542 	}
4543 
4544 	paths->path[count] = path;
4545 	paths->count = count  + 1;
4546 	return paths;
4547 }
4548 
4549 
excluded_match(char * name,struct pathname * path,struct pathnames ** new)4550 int excluded_match(char *name, struct pathname *path, struct pathnames **new)
4551 {
4552 	int i;
4553 
4554 	for(i = 0; i < path->names; i++) {
4555 		int match = use_regex ?
4556 			regexec(path->name[i].preg, name, (size_t) 0,
4557 					NULL, 0) == 0 :
4558 			fnmatch(path->name[i].name, name,
4559 				FNM_PATHNAME|FNM_PERIOD|FNM_EXTMATCH) == 0;
4560 
4561 		if(match) {
4562 			 if(path->name[i].paths == NULL || new == NULL)
4563 				/* match on a leaf component, any subdirectories
4564 			 	* in the filesystem should be excluded */
4565 				return TRUE;
4566 			else
4567 				/* match on a non-leaf component, add any
4568 				 * subdirectories to the new set of
4569 				 * subdirectories to scan for this name */
4570 				*new = add_subdir(*new, path->name[i].paths);
4571 		}
4572 	}
4573 
4574 	return FALSE;
4575 }
4576 
4577 
excluded(char * name,struct pathnames * paths,struct pathnames ** new)4578 int excluded(char *name, struct pathnames *paths, struct pathnames **new)
4579 {
4580 	int n;
4581 
4582 	if(stickypath && excluded_match(name, stickypath, NULL))
4583 		return TRUE;
4584 
4585 	for(n = 0; paths && n < paths->count; n++) {
4586 		int res = excluded_match(name, paths->path[n], new);
4587 		if(res) {
4588 			free(*new);
4589 			*new = NULL;
4590 			return TRUE;
4591 		}
4592 	}
4593 
4594 	/*
4595 	 * Either:
4596 	 * -  no matching names found, return empty new search set, or
4597 	 * -  one or more matches with sub-directories found (no leaf matches),
4598 	 *    in which case return new search set.
4599 	 *
4600 	 * In either case return FALSE as we don't want to exclude this entry
4601 	 */
4602 	return FALSE;
4603 }
4604 
4605 
process_exclude_file(char * argv)4606 void process_exclude_file(char *argv)
4607 {
4608 	FILE *fd;
4609 	char buffer[MAX_LINE + 1]; /* overflow safe */
4610 	char *filename;
4611 
4612 	fd = fopen(argv, "r");
4613 	if(fd == NULL)
4614 		BAD_ERROR("Failed to open exclude file \"%s\" because %s\n",
4615 			argv, strerror(errno));
4616 
4617 	while(fgets(filename = buffer, MAX_LINE + 1, fd) != NULL) {
4618 		int len = strlen(filename);
4619 
4620 		if(len == MAX_LINE && filename[len - 1] != '\n')
4621 			/* line too large */
4622 			BAD_ERROR("Line too long when reading "
4623 				"exclude file \"%s\", larger than %d "
4624 				"bytes\n", argv, MAX_LINE);
4625 
4626 		/*
4627 		 * Remove '\n' terminator if it exists (the last line
4628 		 * in the file may not be '\n' terminated)
4629 		 */
4630 		if(len && filename[len - 1] == '\n')
4631 			filename[len - 1] = '\0';
4632 
4633 		/* Skip any leading whitespace */
4634 		while(isspace(*filename))
4635 			filename ++;
4636 
4637 		/* if comment line, skip */
4638 		if(*filename == '#')
4639 			continue;
4640 
4641 		/*
4642 		 * check for initial backslash, to accommodate
4643 		 * filenames with leading space or leading # character
4644 		 */
4645 		if(*filename == '\\')
4646 			filename ++;
4647 
4648 		/* if line is now empty after skipping characters, skip it */
4649 		if(*filename == '\0')
4650 			continue;
4651 
4652 		if(old_exclude)
4653 			old_add_exclude(filename);
4654 		else
4655 			add_exclude(filename);
4656 	}
4657 
4658 	if(ferror(fd))
4659 		BAD_ERROR("Reading exclude file \"%s\" failed because %s\n",
4660 			argv, strerror(errno));
4661 
4662 	fclose(fd);
4663 }
4664 
4665 
4666 #define RECOVER_ID "Squashfs recovery file v1.0\n"
4667 #define RECOVER_ID_SIZE 28
4668 
write_recovery_data(struct squashfs_super_block * sBlk)4669 void write_recovery_data(struct squashfs_super_block *sBlk)
4670 {
4671 	int res, recoverfd, bytes = sBlk->bytes_used - sBlk->inode_table_start;
4672 	pid_t pid = getpid();
4673 	char *metadata;
4674 	char header[] = RECOVER_ID;
4675 
4676 	if(recover == FALSE) {
4677 		printf("No recovery data option specified.\n");
4678 		printf("Skipping saving recovery file.\n\n");
4679 		return;
4680 	}
4681 
4682 	metadata = malloc(bytes);
4683 	if(metadata == NULL)
4684 		MEM_ERROR();
4685 
4686 	res = read_fs_bytes(fd, sBlk->inode_table_start, bytes, metadata);
4687 	if(res == 0) {
4688 		ERROR("Failed to read append filesystem metadata\n");
4689 		BAD_ERROR("Filesystem corrupted?\n");
4690 	}
4691 
4692 	res = asprintf(&recovery_file, "squashfs_recovery_%s_%d",
4693 		getbase(destination_file), pid);
4694 	if(res == -1)
4695 		MEM_ERROR();
4696 
4697 	recoverfd = open(recovery_file, O_CREAT | O_TRUNC | O_RDWR, S_IRWXU);
4698 	if(recoverfd == -1)
4699 		BAD_ERROR("Failed to create recovery file, because %s.  "
4700 			"Aborting\n", strerror(errno));
4701 
4702 	if(write_bytes(recoverfd, header, RECOVER_ID_SIZE) == -1)
4703 		BAD_ERROR("Failed to write recovery file, because %s\n",
4704 			strerror(errno));
4705 
4706 	if(write_bytes(recoverfd, sBlk, sizeof(struct squashfs_super_block)) == -1)
4707 		BAD_ERROR("Failed to write recovery file, because %s\n",
4708 			strerror(errno));
4709 
4710 	if(write_bytes(recoverfd, metadata, bytes) == -1)
4711 		BAD_ERROR("Failed to write recovery file, because %s\n",
4712 			strerror(errno));
4713 
4714 	close(recoverfd);
4715 	free(metadata);
4716 
4717 	printf("Recovery file \"%s\" written\n", recovery_file);
4718 	printf("If Mksquashfs aborts abnormally (i.e. power failure), run\n");
4719 	printf("mksquashfs dummy %s -recover %s\n", destination_file,
4720 		recovery_file);
4721 	printf("to restore filesystem\n\n");
4722 }
4723 
4724 
read_recovery_data(char * recovery_file,char * destination_file)4725 void read_recovery_data(char *recovery_file, char *destination_file)
4726 {
4727 	int fd, recoverfd, bytes;
4728 	struct squashfs_super_block orig_sBlk, sBlk;
4729 	char *metadata;
4730 	int res;
4731 	struct stat buf;
4732 	char header[] = RECOVER_ID;
4733 	char header2[RECOVER_ID_SIZE];
4734 
4735 	recoverfd = open(recovery_file, O_RDONLY);
4736 	if(recoverfd == -1)
4737 		BAD_ERROR("Failed to open recovery file because %s\n",
4738 			strerror(errno));
4739 
4740 	if(stat(destination_file, &buf) == -1)
4741 		BAD_ERROR("Failed to stat destination file, because %s\n",
4742 			strerror(errno));
4743 
4744 	fd = open(destination_file, O_RDWR);
4745 	if(fd == -1)
4746 		BAD_ERROR("Failed to open destination file because %s\n",
4747 			strerror(errno));
4748 
4749 	res = read_bytes(recoverfd, header2, RECOVER_ID_SIZE);
4750 	if(res == -1)
4751 		BAD_ERROR("Failed to read recovery file, because %s\n",
4752 			strerror(errno));
4753 	if(res < RECOVER_ID_SIZE)
4754 		BAD_ERROR("Recovery file appears to be truncated\n");
4755 	if(strncmp(header, header2, RECOVER_ID_SIZE) !=0 )
4756 		BAD_ERROR("Not a recovery file\n");
4757 
4758 	res = read_bytes(recoverfd, &sBlk, sizeof(struct squashfs_super_block));
4759 	if(res == -1)
4760 		BAD_ERROR("Failed to read recovery file, because %s\n",
4761 			strerror(errno));
4762 	if(res < sizeof(struct squashfs_super_block))
4763 		BAD_ERROR("Recovery file appears to be truncated\n");
4764 
4765 	res = read_fs_bytes(fd, 0, sizeof(struct squashfs_super_block), &orig_sBlk);
4766 	if(res == 0) {
4767 		ERROR("Failed to read superblock from output filesystem\n");
4768 		BAD_ERROR("Output filesystem is empty!\n");
4769 	}
4770 
4771 	if(memcmp(((char *) &sBlk) + 4, ((char *) &orig_sBlk) + 4,
4772 			sizeof(struct squashfs_super_block) - 4) != 0)
4773 		BAD_ERROR("Recovery file and destination file do not seem to "
4774 			"match\n");
4775 
4776 	bytes = sBlk.bytes_used - sBlk.inode_table_start;
4777 
4778 	metadata = malloc(bytes);
4779 	if(metadata == NULL)
4780 		MEM_ERROR();
4781 
4782 	res = read_bytes(recoverfd, metadata, bytes);
4783 	if(res == -1)
4784 		BAD_ERROR("Failed to read recovery file, because %s\n",
4785 			strerror(errno));
4786 	if(res < bytes)
4787 		BAD_ERROR("Recovery file appears to be truncated\n");
4788 
4789 	write_destination(fd, 0, sizeof(struct squashfs_super_block), &sBlk);
4790 
4791 	write_destination(fd, sBlk.inode_table_start, bytes, metadata);
4792 
4793 	close(recoverfd);
4794 	close(fd);
4795 
4796 	printf("Successfully wrote recovery file \"%s\".  Exiting\n",
4797 		recovery_file);
4798 
4799 	exit(0);
4800 }
4801 
4802 
write_filesystem_tables(struct squashfs_super_block * sBlk,int nopad)4803 void write_filesystem_tables(struct squashfs_super_block *sBlk, int nopad)
4804 {
4805 	int i;
4806 
4807 	sBlk->fragments = fragments;
4808 	sBlk->no_ids = id_count;
4809 	sBlk->inode_table_start = write_inodes();
4810 	sBlk->directory_table_start = write_directories();
4811 	sBlk->fragment_table_start = write_fragment_table();
4812 	sBlk->lookup_table_start = exportable ? write_inode_lookup_table() :
4813 		SQUASHFS_INVALID_BLK;
4814 	sBlk->id_table_start = write_id_table();
4815 	sBlk->xattr_id_table_start = write_xattrs();
4816 
4817 	TRACE("sBlk->inode_table_start 0x%llx\n", sBlk->inode_table_start);
4818 	TRACE("sBlk->directory_table_start 0x%llx\n",
4819 		sBlk->directory_table_start);
4820 	TRACE("sBlk->fragment_table_start 0x%llx\n", sBlk->fragment_table_start);
4821 	if(exportable)
4822 		TRACE("sBlk->lookup_table_start 0x%llx\n",
4823 			sBlk->lookup_table_start);
4824 
4825 	sBlk->bytes_used = bytes;
4826 
4827 	sBlk->compression = comp->id;
4828 
4829 	SQUASHFS_INSWAP_SUPER_BLOCK(sBlk);
4830 	write_destination(fd, SQUASHFS_START, sizeof(*sBlk), sBlk);
4831 
4832 	if(!nopad && (i = bytes & (4096 - 1))) {
4833 		char temp[4096] = {0};
4834 		write_destination(fd, bytes, 4096 - i, temp);
4835 	}
4836 
4837 	close(fd);
4838 
4839 	if(recovery_file)
4840 		unlink(recovery_file);
4841 
4842 	total_bytes += total_inode_bytes + total_directory_bytes +
4843 		sizeof(struct squashfs_super_block) + total_xattr_bytes;
4844 
4845 	printf("\n%sSquashfs %d.%d filesystem, %s compressed, data block size"
4846 		" %d\n", exportable ? "Exportable " : "", SQUASHFS_MAJOR,
4847 		SQUASHFS_MINOR, comp->name, block_size);
4848 	printf("\t%s data, %s metadata, %s fragments, %s xattrs\n",
4849 		noD ? "uncompressed" : "compressed", noI ?  "uncompressed" :
4850 		"compressed", no_fragments ? "no" : noF ? "uncompressed" :
4851 		"compressed", no_xattrs ? "no" : noX ? "uncompressed" :
4852 		"compressed");
4853 	printf("\tduplicates are %sremoved\n", duplicate_checking ? "" :
4854 		"not ");
4855 	printf("Filesystem size %.2f Kbytes (%.2f Mbytes)\n", bytes / 1024.0,
4856 		bytes / (1024.0 * 1024.0));
4857 	printf("\t%.2f%% of uncompressed filesystem size (%.2f Kbytes)\n",
4858 		((float) bytes / total_bytes) * 100.0, total_bytes / 1024.0);
4859 	printf("Inode table size %d bytes (%.2f Kbytes)\n",
4860 		inode_bytes, inode_bytes / 1024.0);
4861 	printf("\t%.2f%% of uncompressed inode table size (%d bytes)\n",
4862 		((float) inode_bytes / total_inode_bytes) * 100.0,
4863 		total_inode_bytes);
4864 	printf("Directory table size %d bytes (%.2f Kbytes)\n",
4865 		directory_bytes, directory_bytes / 1024.0);
4866 	printf("\t%.2f%% of uncompressed directory table size (%d bytes)\n",
4867 		((float) directory_bytes / total_directory_bytes) * 100.0,
4868 		total_directory_bytes);
4869 	if(total_xattr_bytes) {
4870 		printf("Xattr table size %d bytes (%.2f Kbytes)\n",
4871 			xattr_bytes, xattr_bytes / 1024.0);
4872 		printf("\t%.2f%% of uncompressed xattr table size (%d bytes)\n",
4873 			((float) xattr_bytes / total_xattr_bytes) * 100.0,
4874 			total_xattr_bytes);
4875 	}
4876 	if(duplicate_checking)
4877 		printf("Number of duplicate files found %d\n", file_count -
4878 			dup_files);
4879 	else
4880 		printf("No duplicate files removed\n");
4881 	printf("Number of inodes %d\n", inode_count);
4882 	printf("Number of files %d\n", file_count);
4883 	if(!no_fragments)
4884 		printf("Number of fragments %d\n", fragments);
4885 	printf("Number of symbolic links  %d\n", sym_count);
4886 	printf("Number of device nodes %d\n", dev_count);
4887 	printf("Number of fifo nodes %d\n", fifo_count);
4888 	printf("Number of socket nodes %d\n", sock_count);
4889 	printf("Number of directories %d\n", dir_count);
4890 	printf("Number of ids (unique uids + gids) %d\n", id_count);
4891 	printf("Number of uids %d\n", uid_count);
4892 
4893 	for(i = 0; i < id_count; i++) {
4894 		if(id_table[i]->flags & ISA_UID) {
4895 			struct passwd *user = getpwuid(id_table[i]->id);
4896 			printf("\t%s (%d)\n", user == NULL ? "unknown" :
4897 				user->pw_name, id_table[i]->id);
4898 		}
4899 	}
4900 
4901 	printf("Number of gids %d\n", guid_count);
4902 
4903 	for(i = 0; i < id_count; i++) {
4904 		if(id_table[i]->flags & ISA_GID) {
4905 			struct group *group = getgrgid(id_table[i]->id);
4906 			printf("\t%s (%d)\n", group == NULL ? "unknown" :
4907 				group->gr_name, id_table[i]->id);
4908 		}
4909 	}
4910 }
4911 
4912 
parse_numberll(char * start,long long * res,int size)4913 int parse_numberll(char *start, long long *res, int size)
4914 {
4915 	char *end;
4916 	long long number;
4917 
4918 	errno = 0; /* To distinguish success/failure after call */
4919 
4920 	number = strtoll(start, &end, 10);
4921 
4922 	/*
4923 	 * check for strtoll underflow or overflow in conversion, and other
4924 	 * errors.
4925 	 */
4926 	if((errno == ERANGE && (number == LLONG_MIN || number == LLONG_MAX)) ||
4927 			(errno != 0 && number == 0))
4928 		return 0;
4929 
4930 	/* reject negative numbers as invalid */
4931 	if(number < 0)
4932 		return 0;
4933 
4934 	if(size) {
4935 		/*
4936 		 * Check for multiplier and trailing junk.
4937 		 * But first check that a number exists before the
4938 		 * multiplier
4939 		 */
4940 		if(end == start)
4941 			return 0;
4942 
4943 		switch(end[0]) {
4944 		case 'g':
4945 		case 'G':
4946 			if(multiply_overflowll(number, 1073741824))
4947 				return 0;
4948 			number *= 1073741824;
4949 
4950 			if(end[1] != '\0')
4951 				/* trailing junk after multiplier, but
4952 				 * allow it to be "bytes" */
4953 				if(strcmp(end + 1, "bytes"))
4954 					return 0;
4955 
4956 			break;
4957 		case 'm':
4958 		case 'M':
4959 			if(multiply_overflowll(number, 1048576))
4960 				return 0;
4961 			number *= 1048576;
4962 
4963 			if(end[1] != '\0')
4964 				/* trailing junk after multiplier, but
4965 				 * allow it to be "bytes" */
4966 				if(strcmp(end + 1, "bytes"))
4967 					return 0;
4968 
4969 			break;
4970 		case 'k':
4971 		case 'K':
4972 			if(multiply_overflowll(number, 1024))
4973 				return 0;
4974 			number *= 1024;
4975 
4976 			if(end[1] != '\0')
4977 				/* trailing junk after multiplier, but
4978 				 * allow it to be "bytes" */
4979 				if(strcmp(end + 1, "bytes"))
4980 					return 0;
4981 
4982 			break;
4983 		case '\0':
4984 			break;
4985 		default:
4986 			/* trailing junk after number */
4987 			return 0;
4988 		}
4989 	} else if(end[0] != '\0')
4990 		/* trailing junk after number */
4991 		return 0;
4992 
4993 	*res = number;
4994 	return 1;
4995 }
4996 
4997 
parse_number(char * start,int * res,int size)4998 int parse_number(char *start, int *res, int size)
4999 {
5000 	long long number;
5001 
5002 	if(!parse_numberll(start, &number, size))
5003 		return 0;
5004 
5005 	/* check if long result will overflow signed int */
5006 	if(number > INT_MAX)
5007 		return 0;
5008 
5009 	*res = (int) number;
5010 	return 1;
5011 }
5012 
5013 
parse_num(char * arg,int * res)5014 int parse_num(char *arg, int *res)
5015 {
5016 	return parse_number(arg, res, 0);
5017 }
5018 
5019 
get_physical_memory()5020 int get_physical_memory()
5021 {
5022 	/*
5023 	 * Long longs are used here because with PAE, a 32-bit
5024 	 * machine can have more than 4GB of physical memory
5025 	 *
5026 	 * sysconf(_SC_PHYS_PAGES) relies on /proc being mounted.
5027 	 * If it isn't fail.
5028 	 */
5029 	long long num_pages = sysconf(_SC_PHYS_PAGES);
5030 	long long page_size = sysconf(_SC_PAGESIZE);
5031 	int phys_mem = num_pages * page_size >> 20;
5032 
5033 	if(num_pages == -1 || page_size == -1)
5034 		return 0;
5035 
5036 	if(phys_mem < SQUASHFS_LOWMEM)
5037 		BAD_ERROR("Mksquashfs requires more physical memory than is "
5038 			"available!\n");
5039 
5040 	return phys_mem;
5041 }
5042 
5043 
check_usable_phys_mem(int total_mem)5044 void check_usable_phys_mem(int total_mem)
5045 {
5046 	/*
5047 	 * We want to allow users to use as much of their physical
5048 	 * memory as they wish.  However, for practical reasons there are
5049 	 * limits which need to be imposed, to protect users from themselves
5050 	 * and to prevent people from using Mksquashfs as a DOS attack by using
5051 	 * all physical memory.   Mksquashfs uses memory to cache data from disk
5052 	 * to optimise performance.  It is pointless to ask it to use more
5053 	 * than 75% of physical memory, as this causes thrashing and it is thus
5054 	 * self-defeating.
5055 	 */
5056 	int mem = get_physical_memory();
5057 
5058 	mem = (mem >> 1) + (mem >> 2); /* 75% */
5059 
5060 	if(total_mem > mem && mem) {
5061 		ERROR("Total memory requested is more than 75%% of physical "
5062 						"memory.\n");
5063 		ERROR("Mksquashfs uses memory to cache data from disk to "
5064 						"optimise performance.\n");
5065 		ERROR("It is pointless to ask it to use more than this amount "
5066 						"of memory, as this\n");
5067 		ERROR("causes thrashing and it is thus self-defeating.\n");
5068 		BAD_ERROR("Requested memory size too large\n");
5069 	}
5070 
5071 	if(sizeof(void *) == 4 && total_mem > 2048) {
5072 		/*
5073 		 * If we're running on a kernel with PAE or on a 64-bit kernel,
5074 		 * then the 75% physical memory limit can still easily exceed
5075 		 * the addressable memory by this process.
5076 		 *
5077 		 * Due to the typical kernel/user-space split (1GB/3GB, or
5078 		 * 2GB/2GB), we have to conservatively assume the 32-bit
5079 		 * processes can only address 2-3GB.  So refuse if the user
5080 		 * tries to allocate more than 2GB.
5081 		 */
5082 		ERROR("Total memory requested may exceed maximum "
5083 				"addressable memory by this process\n");
5084 		BAD_ERROR("Requested memory size too large\n");
5085 	}
5086 }
5087 
5088 
get_default_phys_mem()5089 int get_default_phys_mem()
5090 {
5091 	/*
5092 	 * get_physical_memory() relies on /proc being mounted.
5093 	 * If it fails, issue a warning, and use
5094 	 * SQUASHFS_LOWMEM / SQUASHFS_TAKE as default,
5095 	 * and allow a larger value to be set with -mem.
5096 	 */
5097 	int mem = get_physical_memory();
5098 
5099 	if(mem == 0) {
5100 		mem = SQUASHFS_LOWMEM / SQUASHFS_TAKE;
5101 
5102 		ERROR("Warning: Cannot get size of physical memory, probably "
5103 				"because /proc is missing.\n");
5104 		ERROR("Warning: Defaulting to minimal use of %d Mbytes, use "
5105 				"-mem to set a better value,\n", mem);
5106 		ERROR("Warning: or fix /proc.\n");
5107 	} else
5108 		mem /= SQUASHFS_TAKE;
5109 
5110 	if(sizeof(void *) == 4 && mem > 640) {
5111 		/*
5112 		 * If we're running on a kernel with PAE or on a 64-bit kernel,
5113 		 * the default memory usage can exceed the addressable
5114 		 * memory by this process.
5115 		 * Due to the typical kernel/user-space split (1GB/3GB, or
5116 		 * 2GB/2GB), we have to conservatively assume the 32-bit
5117 		 * processes can only address 2-3GB.  So limit the  default
5118 		 * usage to 640M, which gives room for other data.
5119 		 */
5120 		mem = 640;
5121 	}
5122 
5123 	return mem;
5124 }
5125 
5126 
calculate_queue_sizes(int mem,int * readq,int * fragq,int * bwriteq,int * fwriteq)5127 void calculate_queue_sizes(int mem, int *readq, int *fragq, int *bwriteq,
5128 							int *fwriteq)
5129 {
5130 	*readq = mem / SQUASHFS_READQ_MEM;
5131 	*bwriteq = mem / SQUASHFS_BWRITEQ_MEM;
5132 	*fwriteq = mem / SQUASHFS_FWRITEQ_MEM;
5133 	*fragq = mem - *readq - *bwriteq - *fwriteq;
5134 }
5135 
5136 
5137 #define VERSION() \
5138 	printf("mksquashfs version 4.3-git (2014/09/12)\n");\
5139 	printf("copyright (C) 2014 Phillip Lougher "\
5140 		"<phillip@squashfs.org.uk>\n\n"); \
5141 	printf("This program is free software; you can redistribute it and/or"\
5142 		"\n");\
5143 	printf("modify it under the terms of the GNU General Public License"\
5144 		"\n");\
5145 	printf("as published by the Free Software Foundation; either version "\
5146 		"2,\n");\
5147 	printf("or (at your option) any later version.\n\n");\
5148 	printf("This program is distributed in the hope that it will be "\
5149 		"useful,\n");\
5150 	printf("but WITHOUT ANY WARRANTY; without even the implied warranty "\
5151 		"of\n");\
5152 	printf("MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the"\
5153 		"\n");\
5154 	printf("GNU General Public License for more details.\n");
main(int argc,char * argv[])5155 int main(int argc, char *argv[])
5156 {
5157 	struct stat buf, source_buf;
5158 	int res, i;
5159 	char *b, *root_name = NULL;
5160 	int keep_as_directory = FALSE;
5161 	squashfs_inode inode;
5162 	int readq;
5163 	int fragq;
5164 	int bwriteq;
5165 	int fwriteq;
5166 	int total_mem = get_default_phys_mem();
5167 	int progress = TRUE;
5168 	int force_progress = FALSE;
5169 	struct file_buffer **fragment = NULL;
5170 
5171 	if(argc > 1 && strcmp(argv[1], "-version") == 0) {
5172 		VERSION();
5173 		exit(0);
5174 	}
5175 
5176 	block_log = slog(block_size);
5177 	calculate_queue_sizes(total_mem, &readq, &fragq, &bwriteq, &fwriteq);
5178 
5179         for(i = 1; i < argc && argv[i][0] != '-'; i++);
5180 	if(i < 3)
5181 		goto printOptions;
5182 	source_path = argv + 1;
5183 	source = i - 2;
5184 
5185 	/*
5186 	 * Scan the command line for -comp xxx option, this is to ensure
5187 	 * any -X compressor specific options are passed to the
5188 	 * correct compressor
5189 	 */
5190 	for(; i < argc; i++) {
5191 		struct compressor *prev_comp = comp;
5192 
5193 		if(strcmp(argv[i], "-comp") == 0) {
5194 			if(++i == argc) {
5195 				ERROR("%s: -comp missing compression type\n",
5196 					argv[0]);
5197 				exit(1);
5198 			}
5199 			comp = lookup_compressor(argv[i]);
5200 			if(!comp->supported) {
5201 				ERROR("%s: Compressor \"%s\" is not supported!"
5202 					"\n", argv[0], argv[i]);
5203 				ERROR("%s: Compressors available:\n", argv[0]);
5204 				display_compressors("", COMP_DEFAULT);
5205 				exit(1);
5206 			}
5207 			if(prev_comp != NULL && prev_comp != comp) {
5208 				ERROR("%s: -comp multiple conflicting -comp"
5209 					" options specified on command line"
5210 					", previously %s, now %s\n", argv[0],
5211 					prev_comp->name, comp->name);
5212 				exit(1);
5213 			}
5214 			compressor_opt_parsed = 1;
5215 
5216 		} else if(strcmp(argv[i], "-e") == 0)
5217 			break;
5218 		else if(strcmp(argv[i], "-root-becomes") == 0 ||
5219 				strcmp(argv[i], "-ef") == 0 ||
5220 				strcmp(argv[i], "-pf") == 0 ||
5221 				strcmp(argv[i], "-vaf") == 0 ||
5222 				strcmp(argv[i], "-comp") == 0)
5223 			i++;
5224 	}
5225 
5226 	/*
5227 	 * if no -comp option specified lookup default compressor.  Note the
5228 	 * Makefile ensures the default compressor has been built, and so we
5229 	 * don't need to to check for failure here
5230 	 */
5231 	if(comp == NULL)
5232 		comp = lookup_compressor(COMP_DEFAULT);
5233 
5234 	for(i = source + 2; i < argc; i++) {
5235 		if(strcmp(argv[i], "-action") == 0 ||
5236 				strcmp(argv[i], "-a") ==0) {
5237 			if(++i == argc) {
5238 				ERROR("%s: %s missing action\n",
5239 					argv[0], argv[i - 1]);
5240 				exit(1);
5241 			}
5242 			res = parse_action(argv[i], ACTION_LOG_NONE);
5243 			if(res == 0)
5244 				exit(1);
5245 
5246 		} else if(strcmp(argv[i], "-verbose-action") == 0 ||
5247 				strcmp(argv[i], "-va") ==0) {
5248 			if(++i == argc) {
5249 				ERROR("%s: %s missing action\n",
5250 					argv[0], argv[i - 1]);
5251 				exit(1);
5252 			}
5253 			res = parse_action(argv[i], ACTION_LOG_VERBOSE);
5254 			if(res == 0)
5255 				exit(1);
5256 
5257 		} else if(strcmp(argv[i], "-true-action") == 0 ||
5258 				strcmp(argv[i], "-ta") ==0) {
5259 			if(++i == argc) {
5260 				ERROR("%s: %s missing action\n",
5261 					argv[0], argv[i - 1]);
5262 				exit(1);
5263 			}
5264 			res = parse_action(argv[i], ACTION_LOG_TRUE);
5265 			if(res == 0)
5266 				exit(1);
5267 
5268 		} else if(strcmp(argv[i], "-false-action") == 0 ||
5269 				strcmp(argv[i], "-fa") ==0) {
5270 			if(++i == argc) {
5271 				ERROR("%s: %s missing action\n",
5272 					argv[0], argv[i - 1]);
5273 				exit(1);
5274 			}
5275 			res = parse_action(argv[i], ACTION_LOG_FALSE);
5276 			if(res == 0)
5277 				exit(1);
5278 
5279 		} else if(strcmp(argv[i], "-action-file") == 0 ||
5280 				strcmp(argv[i], "-af") ==0) {
5281 			if(++i == argc) {
5282 				ERROR("%s: %s missing filename\n", argv[0],
5283 							argv[i - 1]);
5284 				exit(1);
5285 			}
5286 			if(read_action_file(argv[i], ACTION_LOG_NONE) == FALSE)
5287 				exit(1);
5288 
5289 		} else if(strcmp(argv[i], "-verbose-action-file") == 0 ||
5290 				strcmp(argv[i], "-vaf") ==0) {
5291 			if(++i == argc) {
5292 				ERROR("%s: %s missing filename\n", argv[0],
5293 							argv[i - 1]);
5294 				exit(1);
5295 			}
5296 			if(read_action_file(argv[i], ACTION_LOG_VERBOSE) == FALSE)
5297 				exit(1);
5298 
5299 		} else if(strcmp(argv[i], "-true-action-file") == 0 ||
5300 				strcmp(argv[i], "-taf") ==0) {
5301 			if(++i == argc) {
5302 				ERROR("%s: %s missing filename\n", argv[0],
5303 							argv[i - 1]);
5304 				exit(1);
5305 			}
5306 			if(read_action_file(argv[i], ACTION_LOG_TRUE) == FALSE)
5307 				exit(1);
5308 
5309 		} else if(strcmp(argv[i], "-false-action-file") == 0 ||
5310 				strcmp(argv[i], "-faf") ==0) {
5311 			if(++i == argc) {
5312 				ERROR("%s: %s missing filename\n", argv[0],
5313 							argv[i - 1]);
5314 				exit(1);
5315 			}
5316 			if(read_action_file(argv[i], ACTION_LOG_FALSE) == FALSE)
5317 				exit(1);
5318 
5319 		} else if(strcmp(argv[i], "-comp") == 0)
5320 			/* parsed previously */
5321 			i++;
5322 
5323 		else if(strncmp(argv[i], "-X", 2) == 0) {
5324 			int args;
5325 
5326 			if(strcmp(argv[i] + 2, "help") == 0)
5327 				goto print_compressor_options;
5328 
5329 			args = compressor_options(comp, argv + i, argc - i);
5330 			if(args < 0) {
5331 				if(args == -1) {
5332 					ERROR("%s: Unrecognised compressor"
5333 						" option %s\n", argv[0],
5334 						argv[i]);
5335 					if(!compressor_opt_parsed)
5336 						ERROR("%s: Did you forget to"
5337 							" specify -comp?\n",
5338 							argv[0]);
5339 print_compressor_options:
5340 					ERROR("%s: selected compressor \"%s\""
5341 						".  Options supported: %s\n",
5342 						argv[0], comp->name,
5343 						comp->usage ? "" : "none");
5344 					if(comp->usage)
5345 						comp->usage();
5346 				}
5347 				exit(1);
5348 			}
5349 			i += args;
5350 
5351 		} else if(strcmp(argv[i], "-pf") == 0) {
5352 			if(++i == argc) {
5353 				ERROR("%s: -pf missing filename\n", argv[0]);
5354 				exit(1);
5355 			}
5356 			if(read_pseudo_file(argv[i]) == FALSE)
5357 				exit(1);
5358 		} else if(strcmp(argv[i], "-p") == 0) {
5359 			if(++i == argc) {
5360 				ERROR("%s: -p missing pseudo file definition\n",
5361 					argv[0]);
5362 				exit(1);
5363 			}
5364 			if(read_pseudo_def(argv[i]) == FALSE)
5365 				exit(1);
5366 		} else if(strcmp(argv[i], "-recover") == 0) {
5367 			if(++i == argc) {
5368 				ERROR("%s: -recover missing recovery file\n",
5369 					argv[0]);
5370 				exit(1);
5371 			}
5372 			read_recovery_data(argv[i], argv[source + 1]);
5373 		} else if(strcmp(argv[i], "-no-recovery") == 0)
5374 			recover = FALSE;
5375 		else if(strcmp(argv[i], "-wildcards") == 0) {
5376 			old_exclude = FALSE;
5377 			use_regex = FALSE;
5378 		} else if(strcmp(argv[i], "-regex") == 0) {
5379 			old_exclude = FALSE;
5380 			use_regex = TRUE;
5381 		} else if(strcmp(argv[i], "-no-sparse") == 0)
5382 			sparse_files = FALSE;
5383 		else if(strcmp(argv[i], "-no-progress") == 0)
5384 			progress = FALSE;
5385 		else if(strcmp(argv[i], "-progress") == 0)
5386 			force_progress = TRUE;
5387 		else if(strcmp(argv[i], "-no-exports") == 0)
5388 			exportable = FALSE;
5389 		else if(strcmp(argv[i], "-processors") == 0) {
5390 			if((++i == argc) || !parse_num(argv[i], &processors)) {
5391 				ERROR("%s: -processors missing or invalid "
5392 					"processor number\n", argv[0]);
5393 				exit(1);
5394 			}
5395 			if(processors < 1) {
5396 				ERROR("%s: -processors should be 1 or larger\n",
5397 					argv[0]);
5398 				exit(1);
5399 			}
5400 		} else if(strcmp(argv[i], "-read-queue") == 0) {
5401 			if((++i == argc) || !parse_num(argv[i], &readq)) {
5402 				ERROR("%s: -read-queue missing or invalid "
5403 					"queue size\n", argv[0]);
5404 				exit(1);
5405 			}
5406 			if(readq < 1) {
5407 				ERROR("%s: -read-queue should be 1 megabyte or "
5408 					"larger\n", argv[0]);
5409 				exit(1);
5410 			}
5411 		} else if(strcmp(argv[i], "-write-queue") == 0) {
5412 			if((++i == argc) || !parse_num(argv[i], &bwriteq)) {
5413 				ERROR("%s: -write-queue missing or invalid "
5414 					"queue size\n", argv[0]);
5415 				exit(1);
5416 			}
5417 			if(bwriteq < 2) {
5418 				ERROR("%s: -write-queue should be 2 megabytes "
5419 					"or larger\n", argv[0]);
5420 				exit(1);
5421 			}
5422 			fwriteq = bwriteq >> 1;
5423 			bwriteq -= fwriteq;
5424 		} else if(strcmp(argv[i], "-fragment-queue") == 0) {
5425 			if((++i == argc) || !parse_num(argv[i], &fragq)) {
5426 				ERROR("%s: -fragment-queue missing or invalid "
5427 					"queue size\n", argv[0]);
5428 				exit(1);
5429 			}
5430 			if(fragq < 1) {
5431 				ERROR("%s: -fragment-queue should be 1 "
5432 					"megabyte or larger\n", argv[0]);
5433 				exit(1);
5434 			}
5435 		} else if(strcmp(argv[i], "-mem") == 0) {
5436 			long long number;
5437 
5438 			if((++i == argc) ||
5439 					!parse_numberll(argv[i], &number, 1)) {
5440 				ERROR("%s: -mem missing or invalid mem size\n",
5441 					 argv[0]);
5442 				exit(1);
5443 			}
5444 
5445 			/*
5446 			 * convert from bytes to Mbytes, ensuring the value
5447 			 * does not overflow a signed int
5448 			 */
5449 			if(number >= (1LL << 51)) {
5450 				ERROR("%s: -mem invalid mem size\n", argv[0]);
5451 				exit(1);
5452 			}
5453 
5454 			total_mem = number / 1048576;
5455 			if(total_mem < (SQUASHFS_LOWMEM / SQUASHFS_TAKE)) {
5456 				ERROR("%s: -mem should be %d Mbytes or "
5457 					"larger\n", argv[0],
5458 					SQUASHFS_LOWMEM / SQUASHFS_TAKE);
5459 				exit(1);
5460 			}
5461 			calculate_queue_sizes(total_mem, &readq, &fragq,
5462 				&bwriteq, &fwriteq);
5463 		} else if(strcmp(argv[i], "-b") == 0) {
5464 			if(++i == argc) {
5465 				ERROR("%s: -b missing block size\n", argv[0]);
5466 				exit(1);
5467 			}
5468 			if(!parse_number(argv[i], &block_size, 1)) {
5469 				ERROR("%s: -b invalid block size\n", argv[0]);
5470 				exit(1);
5471 			}
5472 			if((block_log = slog(block_size)) == 0) {
5473 				ERROR("%s: -b block size not power of two or "
5474 					"not between 4096 and 1Mbyte\n",
5475 					argv[0]);
5476 				exit(1);
5477 			}
5478 		} else if(strcmp(argv[i], "-ef") == 0) {
5479 			if(++i == argc) {
5480 				ERROR("%s: -ef missing filename\n", argv[0]);
5481 				exit(1);
5482 			}
5483 		} else if(strcmp(argv[i], "-no-duplicates") == 0)
5484 			duplicate_checking = FALSE;
5485 
5486 		else if(strcmp(argv[i], "-no-fragments") == 0)
5487 			no_fragments = TRUE;
5488 
5489 		 else if(strcmp(argv[i], "-always-use-fragments") == 0)
5490 			always_use_fragments = TRUE;
5491 
5492 		 else if(strcmp(argv[i], "-sort") == 0) {
5493 			if(++i == argc) {
5494 				ERROR("%s: -sort missing filename\n", argv[0]);
5495 				exit(1);
5496 			}
5497 		} else if(strcmp(argv[i], "-all-root") == 0 ||
5498 				strcmp(argv[i], "-root-owned") == 0)
5499 			global_uid = global_gid = 0;
5500 
5501 		else if(strcmp(argv[i], "-force-uid") == 0) {
5502 			if(++i == argc) {
5503 				ERROR("%s: -force-uid missing uid or user\n",
5504 					argv[0]);
5505 				exit(1);
5506 			}
5507 			if((global_uid = strtoll(argv[i], &b, 10)), *b =='\0') {
5508 				if(global_uid < 0 || global_uid >
5509 						(((long long) 1 << 32) - 1)) {
5510 					ERROR("%s: -force-uid uid out of range"
5511 						"\n", argv[0]);
5512 					exit(1);
5513 				}
5514 			} else {
5515 				struct passwd *uid = getpwnam(argv[i]);
5516 				if(uid)
5517 					global_uid = uid->pw_uid;
5518 				else {
5519 					ERROR("%s: -force-uid invalid uid or "
5520 						"unknown user\n", argv[0]);
5521 					exit(1);
5522 				}
5523 			}
5524 		} else if(strcmp(argv[i], "-force-gid") == 0) {
5525 			if(++i == argc) {
5526 				ERROR("%s: -force-gid missing gid or group\n",
5527 					argv[0]);
5528 				exit(1);
5529 			}
5530 			if((global_gid = strtoll(argv[i], &b, 10)), *b =='\0') {
5531 				if(global_gid < 0 || global_gid >
5532 						(((long long) 1 << 32) - 1)) {
5533 					ERROR("%s: -force-gid gid out of range"
5534 						"\n", argv[0]);
5535 					exit(1);
5536 				}
5537 			} else {
5538 				struct group *gid = getgrnam(argv[i]);
5539 				if(gid)
5540 					global_gid = gid->gr_gid;
5541 				else {
5542 					ERROR("%s: -force-gid invalid gid or "
5543 						"unknown group\n", argv[0]);
5544 					exit(1);
5545 				}
5546 			}
5547 		} else if(strcmp(argv[i], "-noI") == 0 ||
5548 				strcmp(argv[i], "-noInodeCompression") == 0)
5549 			noI = TRUE;
5550 
5551 		else if(strcmp(argv[i], "-noD") == 0 ||
5552 				strcmp(argv[i], "-noDataCompression") == 0)
5553 			noD = TRUE;
5554 
5555 		else if(strcmp(argv[i], "-noF") == 0 ||
5556 				strcmp(argv[i], "-noFragmentCompression") == 0)
5557 			noF = TRUE;
5558 
5559 		else if(strcmp(argv[i], "-noX") == 0 ||
5560 				strcmp(argv[i], "-noXattrCompression") == 0)
5561 			noX = TRUE;
5562 
5563 		else if(strcmp(argv[i], "-no-xattrs") == 0)
5564 			no_xattrs = TRUE;
5565 
5566 		else if(strcmp(argv[i], "-xattrs") == 0)
5567 			no_xattrs = FALSE;
5568 
5569 /* ANDROID CHANGES START*/
5570 #ifdef ANDROID
5571 		else if(strcmp(argv[i], "-context-file") == 0) {
5572 			if(++i == argc) {
5573 				ERROR("%s: -context-file: missing file name\n",
5574 					argv[0]);
5575 				exit(1);
5576 			}
5577 			context_file = argv[i];
5578 		}
5579 #endif
5580 /* ANDROID CHANGES END */
5581 		else if(strcmp(argv[i], "-nopad") == 0)
5582 			nopad = TRUE;
5583 
5584 		else if(strcmp(argv[i], "-info") == 0)
5585 			silent = FALSE;
5586 
5587 		else if(strcmp(argv[i], "-e") == 0)
5588 			break;
5589 
5590 		else if(strcmp(argv[i], "-noappend") == 0)
5591 			delete = TRUE;
5592 
5593 		else if(strcmp(argv[i], "-keep-as-directory") == 0)
5594 			keep_as_directory = TRUE;
5595 /* ANDROID CHANGES START*/
5596 #ifdef ANDROID
5597 		else if(strcmp(argv[i], "-android-fs-config") == 0)
5598 			android_config = TRUE;
5599 		else if(strcmp(argv[i], "-mount-point") == 0) {
5600 			if(++i == argc) {
5601 				ERROR("%s: -mount-point: missing mount point name\n",
5602 					argv[0]);
5603 				exit(1);
5604 			}
5605 			mount_point = argv[i];
5606 		}
5607 		else if(strcmp(argv[i], "-product-out") == 0) {
5608 			if(++i == argc) {
5609 				ERROR("%s: -product-out: missing path name\n",
5610 					argv[0]);
5611 				exit(1);
5612 			}
5613 			target_out_path = argv[i];
5614 		}
5615 #endif
5616 /* ANDROID CHANGES END */
5617 
5618 		else if(strcmp(argv[i], "-exit-on-error") == 0)
5619 			exit_on_error = TRUE;
5620 
5621 		else if(strcmp(argv[i], "-root-becomes") == 0) {
5622 			if(++i == argc) {
5623 				ERROR("%s: -root-becomes: missing name\n",
5624 					argv[0]);
5625 				exit(1);
5626 			}
5627 			root_name = argv[i];
5628 		} else if(strcmp(argv[i], "-version") == 0) {
5629 			VERSION();
5630 		} else {
5631 			ERROR("%s: invalid option\n\n", argv[0]);
5632 printOptions:
5633 			ERROR("SYNTAX:%s source1 source2 ...  dest [options] "
5634 				"[-e list of exclude\ndirs/files]\n", argv[0]);
5635 			ERROR("\nFilesystem build options:\n");
5636 			ERROR("-comp <comp>\t\tselect <comp> compression\n");
5637 			ERROR("\t\t\tCompressors available:\n");
5638 			display_compressors("\t\t\t", COMP_DEFAULT);
5639 			ERROR("-b <block_size>\t\tset data block to "
5640 				"<block_size>.  Default 128 Kbytes\n");
5641 			ERROR("\t\t\tOptionally a suffix of K or M can be"
5642 				" given to specify\n\t\t\tKbytes or Mbytes"
5643 				" respectively\n");
5644 			ERROR("-no-exports\t\tdon't make the filesystem "
5645 				"exportable via NFS\n");
5646 			ERROR("-no-sparse\t\tdon't detect sparse files\n");
5647 			ERROR("-no-xattrs\t\tdon't store extended attributes"
5648 				NOXOPT_STR "\n");
5649 			ERROR("-xattrs\t\t\tstore extended attributes" XOPT_STR
5650 				"\n");
5651 /* ANDROID CHANGES START*/
5652 #ifdef ANDROID
5653 			ERROR("-context-file <file>\tApply selinux security "
5654 				"xattrs from context-file instead\n\t\t\t"
5655 				"of reading xattrs from file system\n");
5656 #endif
5657 /* ANDROID CHANGES END */
5658 			ERROR("-noI\t\t\tdo not compress inode table\n");
5659 			ERROR("-noD\t\t\tdo not compress data blocks\n");
5660 			ERROR("-noF\t\t\tdo not compress fragment blocks\n");
5661 			ERROR("-noX\t\t\tdo not compress extended "
5662 				"attributes\n");
5663 			ERROR("-no-fragments\t\tdo not use fragments\n");
5664 			ERROR("-always-use-fragments\tuse fragment blocks for "
5665 				"files larger than block size\n");
5666 			ERROR("-no-duplicates\t\tdo not perform duplicate "
5667 				"checking\n");
5668 			ERROR("-all-root\t\tmake all files owned by root\n");
5669 			ERROR("-force-uid uid\t\tset all file uids to uid\n");
5670 			ERROR("-force-gid gid\t\tset all file gids to gid\n");
5671 			ERROR("-nopad\t\t\tdo not pad filesystem to a multiple "
5672 				"of 4K\n");
5673 			ERROR("-keep-as-directory\tif one source directory is "
5674 				"specified, create a root\n");
5675 			ERROR("\t\t\tdirectory containing that directory, "
5676 				"rather than the\n");
5677 			ERROR("\t\t\tcontents of the directory\n");
5678 /* ANDROID CHANGES START*/
5679 #ifdef ANDROID
5680 			ERROR("-android-fs-config\tuse android fs config "
5681 				"for mode, uid, and gids of inodes\n");
5682 			ERROR("-mount-point <name>\tNeed to be provided when "
5683 				"android-fs-config or context-file\n\t\t\tare "
5684 				"enabled and source directory is not mount point\n");
5685 			ERROR("-product-out <path>\tPRODUCT_OUT directory to "
5686                                 "read device specific FS rules files from\n");
5687 #endif
5688 /* ANDROID CHANGES END */
5689 			ERROR("\nFilesystem filter options:\n");
5690 			ERROR("-p <pseudo-definition>\tAdd pseudo file "
5691 				"definition\n");
5692 			ERROR("-pf <pseudo-file>\tAdd list of pseudo file "
5693 				"definitions\n");
5694 			ERROR("-sort <sort_file>\tsort files according to "
5695 				"priorities in <sort_file>.  One\n");
5696 			ERROR("\t\t\tfile or dir with priority per line.  "
5697 				"Priority -32768 to\n");
5698 			ERROR("\t\t\t32767, default priority 0\n");
5699 			ERROR("-ef <exclude_file>\tlist of exclude dirs/files."
5700 				"  One per line\n");
5701 			ERROR("-wildcards\t\tAllow extended shell wildcards "
5702 				"(globbing) to be used in\n\t\t\texclude "
5703 				"dirs/files\n");
5704 			ERROR("-regex\t\t\tAllow POSIX regular expressions to "
5705 				"be used in exclude\n\t\t\tdirs/files\n");
5706 			ERROR("\nFilesystem append options:\n");
5707 			ERROR("-noappend\t\tdo not append to existing "
5708 				"filesystem\n");
5709 			ERROR("-root-becomes <name>\twhen appending source "
5710 				"files/directories, make the\n");
5711 			ERROR("\t\t\toriginal root become a subdirectory in "
5712 				"the new root\n");
5713 			ERROR("\t\t\tcalled <name>, rather than adding the new "
5714 				"source items\n");
5715 			ERROR("\t\t\tto the original root\n");
5716 			ERROR("\nMksquashfs runtime options:\n");
5717 			ERROR("-version\t\tprint version, licence and "
5718 				"copyright message\n");
5719 			ERROR("-exit-on-error\t\ttreat normally ignored errors "
5720 				"as fatal\n");
5721 			ERROR("-recover <name>\t\trecover filesystem data "
5722 				"using recovery file <name>\n");
5723 			ERROR("-no-recovery\t\tdon't generate a recovery "
5724 				"file\n");
5725 			ERROR("-info\t\t\tprint files written to filesystem\n");
5726 			ERROR("-no-progress\t\tdon't display the progress "
5727 				"bar\n");
5728 			ERROR("-progress\t\tdisplay progress bar when using "
5729 				"the -info option\n");
5730 			ERROR("-processors <number>\tUse <number> processors."
5731 				"  By default will use number of\n");
5732 			ERROR("\t\t\tprocessors available\n");
5733 			ERROR("-mem <size>\t\tUse <size> physical memory.  "
5734 				"Currently set to %dM\n", total_mem);
5735 			ERROR("\t\t\tOptionally a suffix of K, M or G can be"
5736 				" given to specify\n\t\t\tKbytes, Mbytes or"
5737 				" Gbytes respectively\n");
5738 			ERROR("\nMiscellaneous options:\n");
5739 			ERROR("-root-owned\t\talternative name for -all-root"
5740 				"\n");
5741 			ERROR("-noInodeCompression\talternative name for -noI"
5742 				"\n");
5743 			ERROR("-noDataCompression\talternative name for -noD"
5744 				"\n");
5745 			ERROR("-noFragmentCompression\talternative name for "
5746 				"-noF\n");
5747 			ERROR("-noXattrCompression\talternative name for "
5748 				"-noX\n");
5749 			ERROR("\n-Xhelp\t\t\tprint compressor options for"
5750 				" selected compressor\n");
5751 			ERROR("\nCompressors available and compressor specific "
5752 				"options:\n");
5753 			display_compressor_usage(COMP_DEFAULT);
5754 			exit(1);
5755 		}
5756 	}
5757 
5758 	/*
5759 	 * Some compressors may need the options to be checked for validity
5760 	 * once all the options have been processed
5761 	 */
5762 	res = compressor_options_post(comp, block_size);
5763 	if(res)
5764 		EXIT_MKSQUASHFS();
5765 
5766 	/*
5767 	 * If the -info option has been selected then disable the
5768 	 * progress bar unless it has been explicitly enabled with
5769 	 * the -progress option
5770 	 */
5771 	if(!silent)
5772 		progress = force_progress;
5773 
5774 #ifdef SQUASHFS_TRACE
5775 	/*
5776 	 * Disable progress bar if full debug tracing is enabled.
5777 	 * The progress bar in this case just gets in the way of the
5778 	 * debug trace output
5779 	 */
5780 	progress = FALSE;
5781 #endif
5782 
5783 	for(i = 0; i < source; i++)
5784 		if(lstat(source_path[i], &source_buf) == -1) {
5785 			fprintf(stderr, "Cannot stat source directory \"%s\" "
5786 				"because %s\n", source_path[i],
5787 				strerror(errno));
5788 			EXIT_MKSQUASHFS();
5789 		}
5790 
5791 	destination_file = argv[source + 1];
5792 	if(stat(argv[source + 1], &buf) == -1) {
5793 		if(errno == ENOENT) { /* Does not exist */
5794 			fd = open(argv[source + 1], O_CREAT | O_TRUNC | O_RDWR,
5795 				S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
5796 			if(fd == -1) {
5797 				perror("Could not create destination file");
5798 				exit(1);
5799 			}
5800 			delete = TRUE;
5801 		} else {
5802 			perror("Could not stat destination file");
5803 			exit(1);
5804 		}
5805 
5806 	} else {
5807 		if(S_ISBLK(buf.st_mode)) {
5808 			if((fd = open(argv[source + 1], O_RDWR)) == -1) {
5809 				perror("Could not open block device as "
5810 					"destination");
5811 				exit(1);
5812 			}
5813 			block_device = 1;
5814 
5815 		} else if(S_ISREG(buf.st_mode))	 {
5816 			fd = open(argv[source + 1], (delete ? O_TRUNC : 0) |
5817 				O_RDWR);
5818 			if(fd == -1) {
5819 				perror("Could not open regular file for "
5820 					"writing as destination");
5821 				exit(1);
5822 			}
5823 		}
5824 		else {
5825 			ERROR("Destination not block device or regular file\n");
5826 			exit(1);
5827 		}
5828 
5829 	}
5830 
5831 	/*
5832 	 * process the exclude files - must be done afer destination file has
5833 	 * been possibly created
5834 	 */
5835 	for(i = source + 2; i < argc; i++)
5836 		if(strcmp(argv[i], "-ef") == 0)
5837 			/*
5838 			 * Note presence of filename arg has already
5839 			 * been checked
5840 			 */
5841 			process_exclude_file(argv[++i]);
5842 		else if(strcmp(argv[i], "-e") == 0)
5843 			break;
5844 		else if(strcmp(argv[i], "-root-becomes") == 0 ||
5845 				strcmp(argv[i], "-sort") == 0 ||
5846 				strcmp(argv[i], "-pf") == 0 ||
5847 				strcmp(argv[i], "-af") == 0 ||
5848 				strcmp(argv[i], "-vaf") == 0 ||
5849 				strcmp(argv[i], "-comp") == 0)
5850 			i++;
5851 
5852 	if(i != argc) {
5853 		if(++i == argc) {
5854 			ERROR("%s: -e missing arguments\n", argv[0]);
5855 			EXIT_MKSQUASHFS();
5856 		}
5857 		while(i < argc)
5858 			if(old_exclude)
5859 				old_add_exclude(argv[i++]);
5860 			else
5861 				add_exclude(argv[i++]);
5862 	}
5863 
5864 	/* process the sort files - must be done afer the exclude files  */
5865 	for(i = source + 2; i < argc; i++)
5866 		if(strcmp(argv[i], "-sort") == 0) {
5867 			int res = read_sort_file(argv[++i], source,
5868 								source_path);
5869 			if(res == FALSE)
5870 				BAD_ERROR("Failed to read sort file\n");
5871 			sorted ++;
5872 		} else if(strcmp(argv[i], "-e") == 0)
5873 			break;
5874 		else if(strcmp(argv[i], "-root-becomes") == 0 ||
5875 				strcmp(argv[i], "-ef") == 0 ||
5876 				strcmp(argv[i], "-pf") == 0 ||
5877 				strcmp(argv[i], "-af") == 0 ||
5878 				strcmp(argv[i], "-vaf") == 0 ||
5879 				strcmp(argv[i], "-comp") == 0)
5880 			i++;
5881 
5882 	if(!delete) {
5883 	        comp = read_super(fd, &sBlk, argv[source + 1]);
5884 	        if(comp == NULL) {
5885 			ERROR("Failed to read existing filesystem - will not "
5886 				"overwrite - ABORTING!\n");
5887 			ERROR("To force Mksquashfs to write to this block "
5888 				"device or file use -noappend\n");
5889 			EXIT_MKSQUASHFS();
5890 		}
5891 
5892 		block_log = slog(block_size = sBlk.block_size);
5893 		noI = SQUASHFS_UNCOMPRESSED_INODES(sBlk.flags);
5894 		noD = SQUASHFS_UNCOMPRESSED_DATA(sBlk.flags);
5895 		noF = SQUASHFS_UNCOMPRESSED_FRAGMENTS(sBlk.flags);
5896 		noX = SQUASHFS_UNCOMPRESSED_XATTRS(sBlk.flags);
5897 		no_fragments = SQUASHFS_NO_FRAGMENTS(sBlk.flags);
5898 		always_use_fragments = SQUASHFS_ALWAYS_FRAGMENTS(sBlk.flags);
5899 		duplicate_checking = SQUASHFS_DUPLICATES(sBlk.flags);
5900 		exportable = SQUASHFS_EXPORTABLE(sBlk.flags);
5901 		no_xattrs = SQUASHFS_NO_XATTRS(sBlk.flags);
5902 		comp_opts = SQUASHFS_COMP_OPTS(sBlk.flags);
5903 	}
5904 
5905 	initialise_threads(readq, fragq, bwriteq, fwriteq, delete,
5906 		destination_file);
5907 
5908 	res = compressor_init(comp, &stream, SQUASHFS_METADATA_SIZE, 0);
5909 	if(res)
5910 		BAD_ERROR("compressor_init failed\n");
5911 
5912 	if(delete) {
5913 		int size;
5914 		void *comp_data = compressor_dump_options(comp, block_size,
5915 			&size);
5916 
5917 		printf("Creating %d.%d filesystem on %s, block size %d.\n",
5918 			SQUASHFS_MAJOR, SQUASHFS_MINOR, argv[source + 1], block_size);
5919 
5920 		/*
5921 		 * store any compressor specific options after the superblock,
5922 		 * and set the COMP_OPT flag to show that the filesystem has
5923 		 * compressor specfic options
5924 		 */
5925 		if(comp_data) {
5926 			unsigned short c_byte = size | SQUASHFS_COMPRESSED_BIT;
5927 
5928 			SQUASHFS_INSWAP_SHORTS(&c_byte, 1);
5929 			write_destination(fd, sizeof(struct squashfs_super_block),
5930 				sizeof(c_byte), &c_byte);
5931 			write_destination(fd, sizeof(struct squashfs_super_block) +
5932 				sizeof(c_byte), size, comp_data);
5933 			bytes = sizeof(struct squashfs_super_block) + sizeof(c_byte)
5934 				+ size;
5935 			comp_opts = TRUE;
5936 		} else
5937 			bytes = sizeof(struct squashfs_super_block);
5938 	} else {
5939 		unsigned int last_directory_block, inode_dir_offset,
5940 			inode_dir_file_size, root_inode_size,
5941 			inode_dir_start_block, uncompressed_data,
5942 			compressed_data, inode_dir_inode_number,
5943 			inode_dir_parent_inode;
5944 		unsigned int root_inode_start =
5945 			SQUASHFS_INODE_BLK(sBlk.root_inode),
5946 			root_inode_offset =
5947 			SQUASHFS_INODE_OFFSET(sBlk.root_inode);
5948 
5949 		if((bytes = read_filesystem(root_name, fd, &sBlk, &inode_table,
5950 				&data_cache, &directory_table,
5951 				&directory_data_cache, &last_directory_block,
5952 				&inode_dir_offset, &inode_dir_file_size,
5953 				&root_inode_size, &inode_dir_start_block,
5954 				&file_count, &sym_count, &dev_count, &dir_count,
5955 				&fifo_count, &sock_count, &total_bytes,
5956 				&total_inode_bytes, &total_directory_bytes,
5957 				&inode_dir_inode_number,
5958 				&inode_dir_parent_inode, add_old_root_entry,
5959 				&fragment_table, &inode_lookup_table)) == 0) {
5960 			ERROR("Failed to read existing filesystem - will not "
5961 				"overwrite - ABORTING!\n");
5962 			ERROR("To force Mksquashfs to write to this block "
5963 				"device or file use -noappend\n");
5964 			EXIT_MKSQUASHFS();
5965 		}
5966 		if((append_fragments = fragments = sBlk.fragments)) {
5967 			fragment_table = realloc((char *) fragment_table,
5968 				((fragments + FRAG_SIZE - 1) & ~(FRAG_SIZE - 1))
5969 				 * sizeof(struct squashfs_fragment_entry));
5970 			if(fragment_table == NULL)
5971 				BAD_ERROR("Out of memory in save filesystem state\n");
5972 		}
5973 
5974 		printf("Appending to existing %d.%d filesystem on %s, block "
5975 			"size %d\n", SQUASHFS_MAJOR, SQUASHFS_MINOR, argv[source + 1],
5976 			block_size);
5977 		printf("All -b, -noI, -noD, -noF, -noX, no-duplicates, no-fragments, "
5978 			"-always-use-fragments,\n-exportable and -comp options "
5979 			"ignored\n");
5980 		printf("\nIf appending is not wanted, please re-run with "
5981 			"-noappend specified!\n\n");
5982 
5983 		compressed_data = (inode_dir_offset + inode_dir_file_size) &
5984 			~(SQUASHFS_METADATA_SIZE - 1);
5985 		uncompressed_data = (inode_dir_offset + inode_dir_file_size) &
5986 			(SQUASHFS_METADATA_SIZE - 1);
5987 
5988 		/* save original filesystem state for restoring ... */
5989 		sfragments = fragments;
5990 		sbytes = bytes;
5991 		sinode_count = sBlk.inodes;
5992 		scache_bytes = root_inode_offset + root_inode_size;
5993 		sdirectory_cache_bytes = uncompressed_data;
5994 		sdata_cache = malloc(scache_bytes);
5995 		if(sdata_cache == NULL)
5996 			BAD_ERROR("Out of memory in save filesystem state\n");
5997 		sdirectory_data_cache = malloc(sdirectory_cache_bytes);
5998 		if(sdirectory_data_cache == NULL)
5999 			BAD_ERROR("Out of memory in save filesystem state\n");
6000 		memcpy(sdata_cache, data_cache, scache_bytes);
6001 		memcpy(sdirectory_data_cache, directory_data_cache +
6002 			compressed_data, sdirectory_cache_bytes);
6003 		sinode_bytes = root_inode_start;
6004 		stotal_bytes = total_bytes;
6005 		stotal_inode_bytes = total_inode_bytes;
6006 		stotal_directory_bytes = total_directory_bytes +
6007 			compressed_data;
6008 		sfile_count = file_count;
6009 		ssym_count = sym_count;
6010 		sdev_count = dev_count;
6011 		sdir_count = dir_count + 1;
6012 		sfifo_count = fifo_count;
6013 		ssock_count = sock_count;
6014 		sdup_files = dup_files;
6015 		sid_count = id_count;
6016 		write_recovery_data(&sBlk);
6017 		save_xattrs();
6018 		appending = TRUE;
6019 
6020 		/*
6021 		 * set the filesystem state up to be able to append to the
6022 		 * original filesystem.  The filesystem state differs depending
6023 		 * on whether we're appending to the original root directory, or
6024 		 * if the original root directory becomes a sub-directory
6025 		 * (root-becomes specified on command line, here root_name !=
6026 		 * NULL)
6027 		 */
6028 		inode_bytes = inode_size = root_inode_start;
6029 		directory_size = last_directory_block;
6030 		cache_size = root_inode_offset + root_inode_size;
6031 		directory_cache_size = inode_dir_offset + inode_dir_file_size;
6032 		if(root_name) {
6033 			sdirectory_bytes = last_directory_block;
6034 			sdirectory_compressed_bytes = 0;
6035 			root_inode_number = inode_dir_parent_inode;
6036 			inode_no = sBlk.inodes + 2;
6037 			directory_bytes = last_directory_block;
6038 			directory_cache_bytes = uncompressed_data;
6039 			memmove(directory_data_cache, directory_data_cache +
6040 				compressed_data, uncompressed_data);
6041 			cache_bytes = root_inode_offset + root_inode_size;
6042 			add_old_root_entry(root_name, sBlk.root_inode,
6043 				inode_dir_inode_number, SQUASHFS_DIR_TYPE);
6044 			total_directory_bytes += compressed_data;
6045 			dir_count ++;
6046 		} else {
6047 			sdirectory_compressed_bytes = last_directory_block -
6048 				inode_dir_start_block;
6049 			sdirectory_compressed =
6050 				malloc(sdirectory_compressed_bytes);
6051 			if(sdirectory_compressed == NULL)
6052 				BAD_ERROR("Out of memory in save filesystem "
6053 					"state\n");
6054 			memcpy(sdirectory_compressed, directory_table +
6055 				inode_dir_start_block,
6056 				sdirectory_compressed_bytes);
6057 			sdirectory_bytes = inode_dir_start_block;
6058 			root_inode_number = inode_dir_inode_number;
6059 			inode_no = sBlk.inodes + 1;
6060 			directory_bytes = inode_dir_start_block;
6061 			directory_cache_bytes = inode_dir_offset;
6062 			cache_bytes = root_inode_offset;
6063 		}
6064 
6065 		inode_count = file_count + dir_count + sym_count + dev_count +
6066 			fifo_count + sock_count;
6067 	}
6068 
6069 	if(path)
6070 		paths = add_subdir(paths, path);
6071 
6072 	dump_actions();
6073 	dump_pseudos();
6074 
6075 	if(delete && !keep_as_directory && source == 1 &&
6076 			S_ISDIR(source_buf.st_mode))
6077 		dir_scan(&inode, source_path[0], scan1_readdir, progress);
6078 	else if(!keep_as_directory && source == 1 &&
6079 			S_ISDIR(source_buf.st_mode))
6080 		dir_scan(&inode, source_path[0], scan1_single_readdir, progress);
6081 	else
6082 		dir_scan(&inode, "", scan1_encomp_readdir, progress);
6083 	sBlk.root_inode = inode;
6084 	sBlk.inodes = inode_count;
6085 	sBlk.s_magic = SQUASHFS_MAGIC;
6086 	sBlk.s_major = SQUASHFS_MAJOR;
6087 	sBlk.s_minor = SQUASHFS_MINOR;
6088 	sBlk.block_size = block_size;
6089 	sBlk.block_log = block_log;
6090 	sBlk.flags = SQUASHFS_MKFLAGS(noI, noD, noF, noX, no_fragments,
6091 		always_use_fragments, duplicate_checking, exportable,
6092 		no_xattrs, comp_opts);
6093 	sBlk.mkfs_time = time(NULL);
6094 
6095 	disable_info();
6096 
6097 	while((fragment = get_frag_action(fragment)))
6098 		write_fragment(*fragment);
6099 	unlock_fragments();
6100 	pthread_cleanup_push((void *) pthread_mutex_unlock, &fragment_mutex);
6101 	pthread_mutex_lock(&fragment_mutex);
6102 	while(fragments_outstanding) {
6103 		pthread_mutex_unlock(&fragment_mutex);
6104 		sched_yield();
6105 		pthread_mutex_lock(&fragment_mutex);
6106 	}
6107 	pthread_cleanup_pop(1);
6108 
6109 	queue_put(to_writer, NULL);
6110 	if(queue_get(from_writer) != 0)
6111 		EXIT_MKSQUASHFS();
6112 
6113 	set_progressbar_state(FALSE);
6114 	write_filesystem_tables(&sBlk, nopad);
6115 
6116 	return 0;
6117 }
6118