92 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			92 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /* vi: set sw=4 ts=4: */
 | |
| /* mkswap.c - format swap device (Linux v1 only)
 | |
|  *
 | |
|  * Copyright 2006 Rob Landley <rob@landley.net>
 | |
|  *
 | |
|  * Licensed under GPL version 2, see file LICENSE in this tarball for details.
 | |
|  */
 | |
| 
 | |
| #include "libbb.h"
 | |
| 
 | |
| #if ENABLE_SELINUX
 | |
| static void mkswap_selinux_setcontext(int fd, const char *path)
 | |
| {
 | |
| 	struct stat stbuf;
 | |
| 
 | |
| 	if (!is_selinux_enabled())
 | |
| 		return;
 | |
| 
 | |
| 	if (fstat(fd, &stbuf) < 0)
 | |
| 		bb_perror_msg_and_die("fstat failed");
 | |
| 	if (S_ISREG(stbuf.st_mode)) {
 | |
| 		security_context_t newcon;
 | |
| 		security_context_t oldcon = NULL;
 | |
| 		context_t context;
 | |
| 
 | |
| 		if (fgetfilecon_raw(fd, &oldcon) < 0) {
 | |
| 			if (errno != ENODATA)
 | |
| 				goto error;
 | |
| 			if (matchpathcon(path, stbuf.st_mode, &oldcon) < 0)
 | |
| 				goto error;
 | |
| 		}
 | |
| 		context = context_new(oldcon);
 | |
| 		if (!context || context_type_set(context, "swapfile_t"))
 | |
| 			goto error;
 | |
| 		newcon = context_str(context);
 | |
| 		if (!newcon)
 | |
| 			goto error;
 | |
| 		if (strcmp(oldcon, newcon) != 0 && fsetfilecon_raw(fd, newcon) < 0)
 | |
| 			goto error;
 | |
| 		if (ENABLE_FEATURE_CLEAN_UP) {
 | |
| 			context_free(context);
 | |
| 			freecon(oldcon);
 | |
| 		}
 | |
| 	}
 | |
| 	return;
 | |
|  error:
 | |
| 	bb_perror_msg_and_die("SELinux relabeling failed");
 | |
| }
 | |
| #else
 | |
| #define mkswap_selinux_setcontext(fd, path) ((void)0)
 | |
| #endif
 | |
| 
 | |
| int mkswap_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 | |
| int mkswap_main(int argc, char **argv)
 | |
| {
 | |
| 	int fd, pagesize;
 | |
| 	off_t len;
 | |
| 	unsigned int hdr[129];
 | |
| 
 | |
| 	// No options supported.
 | |
| 
 | |
| 	if (argc != 2) bb_show_usage();
 | |
| 
 | |
| 	// Figure out how big the device is and announce our intentions.
 | |
| 
 | |
| 	fd = xopen(argv[1], O_RDWR);
 | |
| 	len = fdlength(fd);
 | |
| 	pagesize = getpagesize();
 | |
| 	printf("Setting up swapspace version 1, size = %"OFF_FMT"d bytes\n",
 | |
| 			len - pagesize);
 | |
| 	mkswap_selinux_setcontext(fd, argv[1]);
 | |
| 
 | |
| 	// Make a header.
 | |
| 
 | |
| 	memset(hdr, 0, sizeof(hdr));
 | |
| 	hdr[0] = 1;
 | |
| 	hdr[1] = (len / pagesize) - 1;
 | |
| 
 | |
| 	// Write the header.  Sync to disk because some kernel versions check
 | |
| 	// signature on disk (not in cache) during swapon.
 | |
| 
 | |
| 	xlseek(fd, 1024, SEEK_SET);
 | |
| 	xwrite(fd, hdr, sizeof(hdr));
 | |
| 	xlseek(fd, pagesize - 10, SEEK_SET);
 | |
| 	xwrite(fd, "SWAPSPACE2", 10);
 | |
| 	fsync(fd);
 | |
| 
 | |
| 	if (ENABLE_FEATURE_CLEAN_UP) close(fd);
 | |
| 
 | |
| 	return 0;
 | |
| }
 |