2004-09-10 20:17:31 +00:00
|
|
|
/*
|
|
|
|
* ----------------------------------------------------------------------------
|
|
|
|
* "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";
|
|
|
|
|
|
|
|
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 */
|
|
|
|
}
|
2004-09-10 23:16:05 +00:00
|
|
|
if (compressBound(tmp) > MAXPHYS) {
|
|
|
|
errx(1, "cluster size is too large");
|
2004-09-10 20:17:31 +00:00
|
|
|
/* 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 */
|
|
|
|
}
|
2004-09-10 23:16:05 +00:00
|
|
|
hdr.nblocks = sb.st_size / hdr.blksz;
|
2004-09-10 20:17:31 +00:00
|
|
|
if ((sb.st_size % hdr.blksz) != 0) {
|
2004-09-10 23:16:05 +00:00
|
|
|
if (verbose != 0)
|
|
|
|
fprintf(stderr, "file size is not multiple "
|
|
|
|
"of %d, padding data\n", hdr.blksz);
|
|
|
|
hdr.nblocks++;
|
2004-09-10 20:17:31 +00:00
|
|
|
}
|
|
|
|
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)
|
2004-09-10 23:16:05 +00:00
|
|
|
fprintf(stderr, "data size %llu bytes, number of clusters "
|
|
|
|
"%u, index lengh %u bytes\n", sb.st_size,
|
|
|
|
hdr.nblocks, iov[1].iov_len);
|
2004-09-10 20:17:31 +00:00
|
|
|
|
|
|
|
for(i = 0; i == 0 || ibuf != NULL; i++) {
|
|
|
|
ibuf = readblock(fdr, ibuf, hdr.blksz);
|
|
|
|
if (ibuf != NULL) {
|
|
|
|
destlen = compressBound(hdr.blksz);
|
2004-09-10 23:16:05 +00:00
|
|
|
if (compress2(obuf, &destlen, ibuf, hdr.blksz,
|
|
|
|
Z_BEST_COMPRESSION) != Z_OK) {
|
|
|
|
errx(1, "can't compress data: compress2() "
|
|
|
|
"failed");
|
2004-09-10 20:17:31 +00:00
|
|
|
/* Not reached */
|
|
|
|
}
|
2004-09-10 23:16:05 +00:00
|
|
|
if (verbose != 0)
|
|
|
|
fprintf(stderr, "cluster #%d, in %u bytes, "
|
|
|
|
"out %lu bytes\n", i, hdr.blksz, destlen);
|
2004-09-10 20:17:31 +00:00
|
|
|
} else {
|
|
|
|
destlen = DEV_BSIZE - (offset % DEV_BSIZE);
|
|
|
|
memset(obuf, 0, destlen);
|
2004-09-10 23:16:05 +00:00
|
|
|
if (verbose != 0)
|
|
|
|
fprintf(stderr, "padding data with %lu bytes so "
|
|
|
|
"that file size is multiple of %d\n", destlen,
|
|
|
|
DEV_BSIZE);
|
2004-09-10 20:17:31 +00:00
|
|
|
}
|
|
|
|
if (write(fdw, obuf, destlen) < 0) {
|
|
|
|
err(1, "%s", oname);
|
|
|
|
/* Not reached */
|
|
|
|
}
|
|
|
|
toc[i] = htobe64(offset);
|
|
|
|
offset += destlen;
|
|
|
|
}
|
|
|
|
close(fdr);
|
|
|
|
|
|
|
|
if (verbose != 0)
|
2004-09-10 23:16:05 +00:00
|
|
|
fprintf(stderr, "compressed data to %llu bytes, saved %lld "
|
|
|
|
"bytes, %.2f%% decrease.\n", offset, (long long)(sb.st_size - offset),
|
|
|
|
100.0 * (long long)(sb.st_size - offset) / (float)sb.st_size);
|
2004-09-10 20:17:31 +00:00
|
|
|
|
|
|
|
/* 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 *
|
2004-09-10 23:16:05 +00:00
|
|
|
readblock(int fd, char *ibuf, u_int32_t clstsize)
|
|
|
|
{
|
2004-09-10 20:17:31 +00:00
|
|
|
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
|
2004-09-10 23:16:05 +00:00
|
|
|
usage(void)
|
|
|
|
{
|
2004-09-10 20:17:31 +00:00
|
|
|
|
|
|
|
fprintf(stderr, "usage: mkuzip [-v] [-o outfile] [-s cluster_size] infile\n");
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void *
|
2004-09-10 23:16:05 +00:00
|
|
|
safe_malloc(size_t size)
|
|
|
|
{
|
2004-09-10 20:17:31 +00:00
|
|
|
void *retval;
|
|
|
|
|
|
|
|
retval = malloc(size);
|
|
|
|
if (retval == NULL) {
|
|
|
|
err(1, "can't allocate memory");
|
|
|
|
/* Not reached */
|
|
|
|
}
|
|
|
|
return retval;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
2004-09-10 23:16:05 +00:00
|
|
|
cleanup(void)
|
|
|
|
{
|
2004-09-10 20:17:31 +00:00
|
|
|
|
|
|
|
if (cleanfile != NULL)
|
|
|
|
unlink(cleanfile);
|
|
|
|
}
|