sys/types.h ==> sys/param.h for endian macros
[dragonfly.git] / release / write_mfs_in_kernel.c
1 /*
2  * ----------------------------------------------------------------------------
3  * "THE BEER-WARE LICENSE" (Revision 42):
4  * <phk@login.dknet.dk> wrote this file.  As long as you retain this notice you
5  * can do whatever you want with this stuff. If we meet some day, and you think
6  * this stuff is worth it, you can buy me a beer in return.   Poul-Henning Kamp
7  * ----------------------------------------------------------------------------
8  *
9  * $FreeBSD: src/release/write_mfs_in_kernel.c,v 1.7 1999/12/16 02:14:30 jkh Exp $
10  * $DragonFly: src/release/Attic/write_mfs_in_kernel.c,v 1.2 2003/06/17 04:27:17 dillon Exp $
11  *
12  * This program patches a filesystem into a kernel made with MD_ROOT
13  * option.
14  */
15
16 #include <stdio.h>
17 #include <stdlib.h>
18 #include <unistd.h>
19 #include <fcntl.h>
20 #include <sys/types.h>
21 #include <sys/param.h>
22 #include <sys/stat.h>
23 #include <ufs/ffs/fs.h>
24
25 static int force = 0;   /* don't check for zeros, may corrupt kernel */
26
27 int
28 main(int argc, char **argv)
29 {
30         unsigned char *buf_kernel, *buf_fs, *p,*q, *prog;
31         int fd_kernel, fd_fs, ch, errs=0;
32         struct stat st_kernel, st_fs;
33         u_long l;
34
35         prog= *argv;
36         while ((ch = getopt(argc, argv, "f")) != EOF)
37                 switch(ch) {                                                       
38                 case 'f':                                                          
39                         force = 1 - force;
40                         break;   
41                 default:
42                         errs++;
43                 }
44         argc -= optind;
45         argv += optind;
46
47         if (errs || argc != 2) {
48                 fprintf(stderr,"Usage:\n\t%s [-f] kernel fs\n", prog);
49                 exit(2);
50         }
51         --argv; /* original prog did not use getopt(3) */
52         fd_kernel = open(argv[1],O_RDWR);
53         if (fd_kernel < 0) { perror(argv[1]); exit(2); }
54         fstat(fd_kernel,&st_kernel);
55         fd_fs = open(argv[2],O_RDONLY);
56         if (fd_fs < 0) { perror(argv[2]); exit(2); }
57         fstat(fd_fs,&st_fs);
58         buf_kernel = malloc(st_kernel.st_size);
59         if (!buf_kernel) { perror("malloc"); exit(2); }
60         buf_fs = malloc(st_fs.st_size);
61         if (!buf_fs) { perror("malloc"); exit(2); }
62         if (st_kernel.st_size != read(fd_kernel,buf_kernel,st_kernel.st_size))
63                 { perror(argv[1]); exit(2); }
64         if (st_fs.st_size != read(fd_fs,buf_fs,st_fs.st_size))
65                 { perror(argv[2]); exit(2); }
66         for(l=0,p=buf_kernel; l < st_kernel.st_size - st_fs.st_size ; l++,p++ )
67                 if(*p == 'M' && !strcmp(p,"MFS Filesystem goes here"))
68                         goto found;
69         fprintf(stderr,"MFS filesystem signature not found in %s\n",argv[1]);
70         exit(1);
71 found:
72         if (!force)
73                 for(l=0,q= p + SBOFF; l < st_fs.st_size - SBOFF ; l++,q++ )
74                         if (*q)
75                                 goto fail;
76         memcpy(p+SBOFF,buf_fs+SBOFF,st_fs.st_size-SBOFF);
77         lseek(fd_kernel,0L,SEEK_SET);
78         if (st_kernel.st_size != write(fd_kernel,buf_kernel,st_kernel.st_size))
79                 { perror(argv[1]); exit(2); }
80         exit(0);
81 fail:
82         l += SBOFF;
83         fprintf(stderr,"Obstruction in kernel after %ld bytes (%ld Kbyte)\n",
84                 l, l/1024);
85         fprintf(stderr,"Filesystem is %ld bytes (%ld Kbyte)\n",
86                 (u_long)st_fs.st_size, (u_long)st_fs.st_size/1024);
87         exit(1);
88 }
89
90 /*
91  * I added a '-f' option to force writing the image into the kernel, even when
92  * there is already data (i.e. not zero) in the written area. This is useful
93  * to rewrite a changed MD-image. Beware: If the written image is larger than
94  * the space reserved in the kernel (with option MD_ROOT) then
95  * THIS WILL CORRUPT THE KERNEL!
96  *
97  */