f292884004
geom_uzip module. This is based on utility I wrote some 3 years ago for a hack for md(4), which functionally was close to what geom_uzip does today. Since I don't have a time to test that it compiles/works on other arches, stick it to i386 only. Will do it later. Unlike original cloop util, this one embedds FreeBSD-compatible shell code into the generated image, not Linux one. Unfortunately severe space restriction imposed by the CLOOP format doesn't allow to put conditional code which will work both on Linux and FreeBSD. In fact it was quite a challenge to fit necessary FreeBSD code into 127 bytes. ;-)
260 lines
5.8 KiB
C
260 lines
5.8 KiB
C
/*
|
|
* ----------------------------------------------------------------------------
|
|
* "THE BEER-WARE LICENSE" (Revision 42):
|
|
* <sobomax@FreeBSD.ORG> wrote this file. As long as you retain this notice you
|
|
* can do whatever you want with this stuff. If we meet some day, and you think
|
|
* this stuff is worth it, you can buy me a beer in return. Maxim Sobolev
|
|
* ----------------------------------------------------------------------------
|
|
*
|
|
* $FreeBSD$
|
|
*
|
|
*/
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/endian.h>
|
|
#include <sys/param.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/uio.h>
|
|
#include <netinet/in.h>
|
|
#include <zlib.h>
|
|
#include <err.h>
|
|
#include <fcntl.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#define CLSTSIZE 16384
|
|
#define DEFAULT_SUFX ".uzip"
|
|
|
|
#define CLOOP_MAGIC_LEN 128
|
|
static char CLOOP_MAGIC_START[] = "#!/bin/sh\n#V2.0 Format\n"
|
|
"m=geom_uzip\n(kldstat -n $m 2>&-||kldload $m)>&-&&"
|
|
"mount_cd9660 /dev/`mdconfig -af $0`.uzip $1\nexit $?\n";
|
|
|
|
/*
|
|
* Maximum allowed valid block size (to prevent foot-shooting)
|
|
*/
|
|
#define MAX_BLKSZ (MAXPHYS - MAXPHYS / 1000 - 12)
|
|
|
|
static char *readblock(int, char *, u_int32_t);
|
|
static void usage(void);
|
|
static void *safe_malloc(size_t);
|
|
static void cleanup(void);
|
|
|
|
static char *cleanfile = NULL;
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
char *iname, *oname, *obuf, *ibuf;
|
|
uint64_t *toc;
|
|
int fdr, fdw, i, opt, verbose, tmp;
|
|
struct iovec iov[2];
|
|
struct stat sb;
|
|
uLongf destlen;
|
|
uint64_t offset;
|
|
struct cloop_header {
|
|
char magic[CLOOP_MAGIC_LEN]; /* cloop magic */
|
|
uint32_t blksz; /* block size */
|
|
uint32_t nblocks; /* number of blocks */
|
|
} hdr;
|
|
|
|
memset(&hdr, 0, sizeof(hdr));
|
|
hdr.blksz = CLSTSIZE;
|
|
strcpy(hdr.magic, CLOOP_MAGIC_START);
|
|
oname = NULL;
|
|
verbose = 0;
|
|
|
|
while((opt = getopt(argc, argv, "o:s:v")) != -1) {
|
|
switch(opt) {
|
|
case 'o':
|
|
oname = optarg;
|
|
break;
|
|
|
|
case 's':
|
|
tmp = atoi(optarg);
|
|
if (tmp <= 0) {
|
|
errx(1, "invalid cluster size specified: %s",
|
|
optarg);
|
|
/* Not reached */
|
|
}
|
|
if (tmp % DEV_BSIZE != 0) {
|
|
errx(1, "cluster size should be multiple of %d",
|
|
DEV_BSIZE);
|
|
/* Not reached */
|
|
}
|
|
if (tmp > MAX_BLKSZ) {
|
|
errx(1, "cluster size can't be more than %d",
|
|
MAX_BLKSZ);
|
|
/* Not reached */
|
|
}
|
|
hdr.blksz = tmp;
|
|
break;
|
|
|
|
case 'v':
|
|
verbose = 1;
|
|
break;
|
|
|
|
default:
|
|
usage();
|
|
/* Not reached */
|
|
}
|
|
}
|
|
argc -= optind;
|
|
argv += optind;
|
|
|
|
if (argc != 1) {
|
|
usage();
|
|
/* Not reached */
|
|
}
|
|
|
|
iname = argv[0];
|
|
if (oname == NULL) {
|
|
asprintf(&oname, "%s%s", iname, DEFAULT_SUFX);
|
|
if (oname == NULL) {
|
|
err(1, "can't allocate memory");
|
|
/* Not reached */
|
|
}
|
|
}
|
|
|
|
obuf = safe_malloc(compressBound(hdr.blksz));
|
|
ibuf = safe_malloc(hdr.blksz);
|
|
|
|
signal(SIGHUP, exit);
|
|
signal(SIGINT, exit);
|
|
signal(SIGTERM, exit);
|
|
signal(SIGXCPU, exit);
|
|
signal(SIGXFSZ, exit);
|
|
atexit(cleanup);
|
|
|
|
if (stat(iname, &sb) != 0) {
|
|
err(1, "%s", iname);
|
|
/* Not reached */
|
|
}
|
|
if ((sb.st_size % hdr.blksz) != 0) {
|
|
errx(1, "%s: incorrect image: file size is not multiple of %d",
|
|
iname, hdr.blksz);
|
|
/* Not reached */
|
|
}
|
|
hdr.nblocks = sb.st_size / hdr.blksz;
|
|
toc = safe_malloc((hdr.nblocks + 1) * sizeof(*toc));
|
|
|
|
fdr = open(iname, O_RDONLY);
|
|
if (fdr < 0) {
|
|
err(1, "%s", iname);
|
|
/* Not reached */
|
|
}
|
|
fdw = open(oname, O_WRONLY | O_TRUNC | O_CREAT,
|
|
S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
|
|
if (fdw < 0) {
|
|
err(1, "%s", oname);
|
|
/* Not reached */
|
|
}
|
|
cleanfile = oname;
|
|
|
|
/* Prepare header that we will write later when we have index ready. */
|
|
iov[0].iov_base = (char *)&hdr;
|
|
iov[0].iov_len = sizeof(hdr);
|
|
iov[1].iov_base = (char *)toc;
|
|
iov[1].iov_len = (hdr.nblocks + 1) * sizeof(*toc);
|
|
offset = iov[0].iov_len + iov[1].iov_len;
|
|
|
|
/* Reserve space for header */
|
|
lseek(fdw, offset, SEEK_SET);
|
|
|
|
if (verbose != 0)
|
|
fprintf(stderr, "Data size: %ld bytes, number of clusters: "
|
|
"%ld, index lengh: %ld bytes\n", (long)sb.st_size,
|
|
(long)(hdr.nblocks), ((long)hdr.nblocks + 1) * sizeof(*toc));
|
|
|
|
for(i = 0; i == 0 || ibuf != NULL; i++) {
|
|
ibuf = readblock(fdr, ibuf, hdr.blksz);
|
|
if (ibuf != NULL) {
|
|
destlen = compressBound(hdr.blksz);
|
|
memset(obuf, 0, destlen);
|
|
if (compress2(obuf, &destlen, ibuf, hdr.blksz, Z_BEST_COMPRESSION) != Z_OK) {
|
|
errx(1, "can't compress data: compress2() failed");
|
|
/* Not reached */
|
|
}
|
|
#if 0
|
|
/*
|
|
* We don't really need those two leading bytes. Moreover, they
|
|
* confuse oldest decompression routine presented in the
|
|
* FreeBSD kernel, so they should be omitted.
|
|
*/
|
|
destlen -= 2;
|
|
#endif
|
|
} else {
|
|
destlen = DEV_BSIZE - (offset % DEV_BSIZE);
|
|
memset(obuf, 0, destlen);
|
|
}
|
|
if (write(fdw, obuf, destlen) < 0) {
|
|
err(1, "%s", oname);
|
|
/* Not reached */
|
|
}
|
|
toc[i] = htobe64(offset);
|
|
offset += destlen;
|
|
}
|
|
close(fdr);
|
|
|
|
if (verbose != 0)
|
|
fprintf(stderr, "compressed data to %llu bytes.\n", offset);
|
|
|
|
/* Convert to big endian */
|
|
hdr.blksz = htonl(hdr.blksz);
|
|
hdr.nblocks = htonl(hdr.nblocks);
|
|
/* Write headers into pre-allocated space */
|
|
lseek(fdw, 0, SEEK_SET);
|
|
if (writev(fdw, iov, 2) < 0) {
|
|
err(1, "%s", oname);
|
|
/* Not reached */
|
|
}
|
|
cleanfile = NULL;
|
|
close(fdw);
|
|
|
|
exit(0);
|
|
}
|
|
|
|
static char *
|
|
readblock(int fd, char *ibuf, u_int32_t clstsize) {
|
|
int numread;
|
|
|
|
bzero(ibuf, clstsize);
|
|
numread = read(fd, ibuf, clstsize);
|
|
if (numread < 0) {
|
|
err(1, "read() failed");
|
|
/* Not reached */
|
|
}
|
|
if (numread == 0) {
|
|
return NULL;
|
|
}
|
|
return ibuf;
|
|
}
|
|
|
|
static void
|
|
usage(void) {
|
|
|
|
fprintf(stderr, "usage: mkuzip [-v] [-o outfile] [-s cluster_size] infile\n");
|
|
exit(1);
|
|
}
|
|
|
|
static void *
|
|
safe_malloc(size_t size) {
|
|
void *retval;
|
|
|
|
retval = malloc(size);
|
|
if (retval == NULL) {
|
|
err(1, "can't allocate memory");
|
|
/* Not reached */
|
|
}
|
|
return retval;
|
|
}
|
|
|
|
static void
|
|
cleanup(void) {
|
|
|
|
if (cleanfile != NULL)
|
|
unlink(cleanfile);
|
|
}
|