-/***********************************************************\r
- ar.c -- main file\r
-***********************************************************/\r
-\r
-static char *usage =\r
- "ar -- compression archiver -- written by Haruhiko Okumura\n"\r
- " PC-VAN:SCIENCE CompuServe:74050,1022\n"\r
- " NIFTY-Serve:PAF01022 INTERNET:74050.1022@compuserve.com\n"\r
- "Usage: ar command archive [file ...]\n"\r
- "Commands:\n"\r
- " a: Add files to archive (replace if present)\n"\r
- " x: Extract files from archive\n"\r
- " r: Replace files in archive\n"\r
- " d: Delete files from archive\n"\r
- " p: Print files on standard output\n"\r
- " l: List contents of archive\n"\r
- "If no files are named, all files in archive are processed,\n"\r
- " except for commands 'a' and 'd'.\n"\r
- "You may copy, distribute, and rewrite this program freely.\n";\r
-\r
-/***********************************************************\r
-\r
-Structure of archive block (low order byte first):\r
------preheader\r
- 1 basic header size\r
- = 25 + strlen(filename) (= 0 if end of archive)\r
- 1 basic header algebraic sum (mod 256)\r
------basic header\r
- 5 method ("-lh0-" = stored, "-lh5-" = compressed)\r
- 4 compressed size (including extended headers)\r
- 4 original size\r
- 4 not used\r
- 1 0x20\r
- 1 0x01\r
- 1 filename length (x)\r
- x filename\r
- 2 original file's CRC\r
- 1 0x20\r
- 2 first extended header size (0 if none)\r
------first extended header, etc.\r
------compressed file\r
-\r
-***********************************************************/\r
-\r
-#include <stdlib.h>\r
-#include <string.h>\r
-#include <ctype.h>\r
-#include "ar.h"\r
-\r
-#define FNAME_MAX (255 - 25) /* max strlen(filename) */\r
-#define namelen header[19]\r
-#define filename ((char *)&header[20])\r
-\r
-int unpackable; /* global, set in io.c */\r
-ulong compsize, origsize; /* global */\r
-\r
-static uchar buffer[DICSIZ];\r
-static uchar header[255];\r
-static uchar headersize, headersum;\r
-static uint file_crc;\r
-static char *temp_name;\r
-\r
-static uint ratio(ulong a, ulong b) /* [(1000a + [b/2]) / b] */\r
-{\r
- int i;\r
-\r
- for (i = 0; i < 3; i++)\r
- if (a <= ULONG_MAX / 10) a *= 10; else b /= 10;\r
- if ((ulong)(a + (b >> 1)) < a) { a >>= 1; b >>= 1; }\r
- if (b == 0) return 0;\r
- return (uint)((a + (b >> 1)) / b);\r
-}\r
-\r
-static void put_to_header(int i, int n, ulong x)\r
-{\r
- while (--n >= 0) {\r
- header[i++] = (uchar)((uint)x & 0xFF); x >>= 8;\r
- }\r
-}\r
-\r
-static ulong get_from_header(int i, int n)\r
-{\r
- ulong s;\r
-\r
- s = 0;\r
- while (--n >= 0) s = (s << 8) + header[i + n]; /* little endian */\r
- return s;\r
-}\r
-\r
-static uint calc_headersum(void)\r
-{\r
- int i;\r
- uint s;\r
-\r
- s = 0;\r
- for (i = 0; i < headersize; i++) s += header[i];\r
- return s & 0xFF;\r
-}\r
-\r
-static int read_header(void)\r
-{\r
- headersize = (uchar) fgetc(arcfile);\r
- if (headersize == 0) return 0; /* end of archive */\r
- headersum = (uchar) fgetc(arcfile);\r
- fread_crc(header, headersize, arcfile); /* CRC not used */\r
- if (calc_headersum() != headersum) error("Header sum error");\r
- compsize = get_from_header(5, 4);\r
- origsize = get_from_header(9, 4);\r
- file_crc = (uint)get_from_header(headersize - 5, 2);\r
- filename[namelen] = '\0';\r
- return 1; /* success */\r
-}\r
-\r
-static void write_header(void)\r
-{\r
- fputc(headersize, outfile);\r
- /* We've destroyed file_crc by null-terminating filename. */\r
- put_to_header(headersize - 5, 2, (ulong)file_crc);\r
- fputc(calc_headersum(), outfile);\r
- fwrite_crc(header, headersize, outfile); /* CRC not used */\r
-}\r
-\r
-static void skip(void)\r
-{\r
- fseek(arcfile, compsize, SEEK_CUR);\r
-}\r
-\r
-static void copy(void)\r
-{\r
- uint n;\r
-\r
- write_header();\r
- while (compsize != 0) {\r
- n = (uint)((compsize > DICSIZ) ? DICSIZ : compsize);\r
- if (fread ((char *)buffer, 1, n, arcfile) != n)\r
- error("Can't read");\r
- if (fwrite((char *)buffer, 1, n, outfile) != n)\r
- error("Can't write");\r
- compsize -= n;\r
- }\r
-}\r
-\r
-static void store(void)\r
-{\r
- uint n;\r
-\r
- origsize = 0;\r
- crc = INIT_CRC;\r
- while ((n = fread((char *)buffer, 1, DICSIZ, infile)) != 0) {\r
- fwrite_crc(buffer, n, outfile); origsize += n;\r
- }\r
- compsize = origsize;\r
-}\r
-\r
-static int add(int replace_flag)\r
-{\r
- long headerpos, arcpos;\r
- uint r;\r
-\r
- if ((infile = fopen(filename, "rb")) == NULL) {\r
- fprintf(stderr, "Can't open %s\n", filename);\r
- return 0; /* failure */\r
- }\r
- if (replace_flag) {\r
- printf("Replacing %s ", filename); skip();\r
- } else\r
- printf("Adding %s ", filename);\r
- headerpos = ftell(outfile);\r
- namelen = strlen(filename);\r
- headersize = 25 + namelen;\r
- memcpy(header, "-lh5-", 5); /* compress */\r
- write_header(); /* temporarily */\r
- arcpos = ftell(outfile);\r
- origsize = compsize = 0; unpackable = 0;\r
- crc = INIT_CRC; encode();\r
- if (unpackable) {\r
- header[3] = '0'; /* store */\r
- rewind(infile);\r
- fseek(outfile, arcpos, SEEK_SET);\r
- store();\r
- }\r
- file_crc = crc ^ INIT_CRC;\r
- fclose(infile);\r
- put_to_header(5, 4, compsize);\r
- put_to_header(9, 4, origsize);\r
- memcpy(header + 13, "\0\0\0\0\x20\x01", 6);\r
- memcpy(header + headersize - 3, "\x20\0\0", 3);\r
- fseek(outfile, headerpos, SEEK_SET);\r
- write_header(); /* true header */\r
- fseek(outfile, 0L, SEEK_END);\r
- r = ratio(compsize, origsize);\r
- printf(" %d.%d%%\n", r / 10, r % 10);\r
- return 1; /* success */\r
-}\r
-\r
-int get_line(char *s, int n)\r
-{\r
- int i, c;\r
-\r
- i = 0;\r
- while ((c = getchar()) != EOF && c != '\n')\r
- if (i < n) s[i++] = (char)c;\r
- s[i] = '\0';\r
- return i;\r
-}\r
-\r
-static void extract(int to_file)\r
-{\r
- int n, method;\r
- uint ext_headersize;\r
-\r
- if (to_file) {\r
- while ((outfile = fopen(filename, "wb")) == NULL) {\r
- fprintf(stderr, "Can't open %s\nNew filename: ", filename);\r
- if (get_line(filename, FNAME_MAX) == 0) {\r
- fprintf(stderr, "Not extracted\n");\r
- skip(); return;\r
- }\r
- namelen = strlen(filename);\r
- }\r
- printf("Extracting %s ", filename);\r
- } else {\r
- outfile = stdout;\r
- printf("===== %s =====\n", filename);\r
- }\r
- crc = INIT_CRC;\r
- method = header[3]; header[3] = ' ';\r
- if (! strchr("045", method) || memcmp("-lh -", header, 5)) {\r
- fprintf(stderr, "Unknown method: %u\n", method);\r
- skip();\r
- } else {\r
- ext_headersize = (uint)get_from_header(headersize - 2, 2);\r
- while (ext_headersize != 0) {\r
- fprintf(stderr, "There's an extended header of size %u.\n",\r
- ext_headersize);\r
- compsize -= ext_headersize;\r
- if (fseek(arcfile, ext_headersize - 2, SEEK_CUR))\r
- error("Can't read");\r
- ext_headersize = fgetc(arcfile);\r
- ext_headersize += (uint)fgetc(arcfile) << 8;\r
- }\r
- crc = INIT_CRC;\r
- if (method != '0') decode_start();\r
- while (origsize != 0) {\r
- n = (uint)((origsize > DICSIZ) ? DICSIZ : origsize);\r
- if (method != '0') decode(n, buffer);\r
- else if (fread((char *)buffer, 1, n, arcfile) != n)\r
- error("Can't read");\r
- fwrite_crc(buffer, n, outfile);\r
- if (outfile != stdout) putc('.', stderr);\r
- origsize -= n;\r
- }\r
- }\r
- if (to_file) fclose(outfile); else outfile = NULL;\r
- printf("\n");\r
- if ((crc ^ INIT_CRC) != file_crc)\r
- fprintf(stderr, "CRC error\n");\r
-}\r
-\r
-static void list_start(void)\r
-{\r
- printf("Filename Original Compressed Ratio CRC Method\n");\r
-}\r
-\r
-static void list(void)\r
-{\r
- uint r;\r
-\r
- printf("%-14s", filename);\r
- if (namelen > 14) printf("\n ");\r
- r = ratio(compsize, origsize);\r
- printf(" %10lu %10lu %u.%03u %04X %5.5s\n",\r
- origsize, compsize, r / 1000, r % 1000, file_crc, header);\r
-}\r
-\r
-static int match(char *s1, char *s2)\r
-{\r
- for ( ; ; ) {\r
- while (*s2 == '*' || *s2 == '?') {\r
- if (*s2++ == '*')\r
- while (*s1 && *s1 != *s2) s1++;\r
- else if (*s1 == 0)\r
- return 0;\r
- else s1++;\r
- }\r
- if (*s1 != *s2) return 0;\r
- if (*s1 == 0 ) return 1;\r
- s1++; s2++;\r
- }\r
-}\r
-\r
-static int search(int argc, char *argv[])\r
-{\r
- int i;\r
-\r
- if (argc == 3) return 1;\r
- for (i = 3; i < argc; i++)\r
- if (match(filename, argv[i])) return 1;\r
- return 0;\r
-}\r
-\r
-static void exitfunc(void)\r
-{\r
- fclose(outfile); remove(temp_name);\r
-}\r
-\r
-int main(int argc, char *argv[])\r
-{\r
- int i, j, cmd, count, nfiles, found, done;\r
-\r
- /* Check command line arguments. */\r
- if (argc < 3\r
- || argv[1][1] != '\0'\r
- || ! strchr("AXRDPL", cmd = toupper(argv[1][0]))\r
- || (argc == 3 && strchr("AD", cmd)))\r
- error(usage);\r
-\r
- /* Wildcards used? */\r
- for (i = 3; i < argc; i++)\r
- if (strpbrk(argv[i], "*?")) break;\r
- if (cmd == 'A' && i < argc)\r
- error("Filenames may not contain '*' and '?'");\r
- if (i < argc) nfiles = -1; /* contains wildcards */\r
- else nfiles = argc - 3; /* number of files to process */\r
-\r
- /* Open archive. */\r
- arcfile = fopen(argv[2], "rb");\r
- if (arcfile == NULL && cmd != 'A')\r
- error("Can't open archive '%s'", argv[2]);\r
-\r
- /* Open temporary file. */\r
- if (strchr("ARD", cmd)) {\r
- temp_name = tmpnam(NULL);\r
- outfile = fopen(temp_name, "wb");\r
- if (outfile == NULL)\r
- error("Can't open temporary file");\r
- atexit(exitfunc);\r
- } else temp_name = NULL;\r
-\r
- make_crctable(); count = done = 0;\r
-\r
- if (cmd == 'A') {\r
- for (i = 3; i < argc; i++) {\r
- for (j = 3; j < i; j++)\r
- if (strcmp(argv[j], argv[i]) == 0) break;\r
- if (j == i) {\r
- strcpy(filename, argv[i]);\r
- if (add(0)) count++; else argv[i][0] = 0;\r
- } else nfiles--;\r
- }\r
- if (count == 0 || arcfile == NULL) done = 1;\r
- }\r
-\r
- while (! done && read_header()) {\r
- found = search(argc, argv);\r
- switch (cmd) {\r
- case 'R':\r
- if (found) {\r
- if (add(1)) count++; else copy();\r
- } else copy();\r
- break;\r
- case 'A': case 'D':\r
- if (found) {\r
- count += (cmd == 'D'); skip();\r
- } else copy();\r
- break;\r
- case 'X': case 'P':\r
- if (found) {\r
- extract(cmd == 'X');\r
- if (++count == nfiles) done = 1;\r
- } else skip();\r
- break;\r
- case 'L':\r
- if (found) {\r
- if (count == 0) list_start();\r
- list();\r
- if (++count == nfiles) done = 1;\r
- }\r
- skip(); break;\r
- }\r
- }\r
-\r
- if (temp_name != NULL && count != 0) {\r
- fputc(0, outfile); /* end of archive */\r
- if (ferror(outfile) || fclose(outfile) == EOF)\r
- error("Can't write");\r
- remove(argv[2]); rename(temp_name, argv[2]);\r
- }\r
-\r
- printf(" %d files\n", count);\r
- return EXIT_SUCCESS;\r
-}\r
+/***********************************************************
+ ar.c -- main file
+***********************************************************/
+
+static char *version = "0.01";
+
+static char *usage =
+ "ar -- compression archiver -- written by Haruhiko Okumura\n"
+ " PC-VAN:SCIENCE CompuServe:74050,1022\n"
+ " NIFTY-Serve:PAF01022 INTERNET:74050.1022@compuserve.com\n"
+ "Usage: ar command archive [file ...]\n"
+ "Commands:\n"
+ " a: Add files to archive (replace if present)\n"
+ " x: Extract files from archive\n"
+ " r: Replace files in archive\n"
+ " d: Delete files from archive\n"
+ " p: Print files on standard output\n"
+ " l: List contents of archive\n"
+ "If no files are named, all files in archive are processed,\n"
+ " except for commands 'a' and 'd'.\n"
+ "You may copy, distribute, and rewrite this program freely.\n";
+
+/***********************************************************
+
+Structure of archive block (low order byte first):
+-----preheader
+ 1 basic header size
+ = 25 + strlen(filename) (= 0 if end of archive)
+ 1 basic header algebraic sum (mod 256)
+-----basic header
+ 5 method ("-lh0-" = stored, "-lh5-" = compressed)
+ 4 compressed size (including extended headers)
+ 4 original size
+ 4 not used
+ 1 0x20
+ 1 0x01
+ 1 filename length (x)
+ x filename
+ 2 original file's CRC
+ 1 0x20
+ 2 first extended header size (0 if none)
+-----first extended header, etc.
+-----compressed file
+
+***********************************************************/
+
+#include <stdlib.h>
+#include <string.h>
+#include <ctype.h>
+#include <errno.h>
+#include <sys/stat.h>
+#include <time.h>
+#include <utime.h>
+#include <unistd.h>
+#include "ar.h"
+
+struct lha_method methods[] = {
+ /* id, dicbit, pbit, maxmatch */
+ /* note: dicbit == 0 means no compress */
+ /* (1U << pbit) > np(dicbit+1) */
+ {"-lh0-", 0, 0, 0}, /* 0: no compress */
+ {"-lh1-", 12, 0, 60}, /* 1: 2^12 = 4KB dynamic huffman (LHarc) */
+ {"-lh2-", 13, 0,256}, /* 2: 2^13 = 8KB dynamic huffman */
+ {"-lh3-", 13, 0,256}, /* 3: 2^13 = 8KB static huffman */
+ {"-lh4-", 12, 4,256}, /* 4: 2^12 = 4KB static huffman (pos and len)*/
+ {"-lh5-", 13, 4,256}, /* 5: 2^13 = 8KB static huffman (pos and len)*/
+ {"-lh6-", 15, 5,256}, /* 6: 2^15 = 32KB static huffman (pos and len)*/
+ {"-lh7-", 16, 5,256}, /* 7: 2^16 = 64KB static huffman (pos and len)*/
+ {"-lzs-", 11, 0, 17}, /* 8: 2^11 = 2KB (LArc) */
+ {"-lz5-", 12, 0, 17}, /* 9: 2^12 = 4KB (LArc) */
+ {"-lz4-", 0, 0, 0}, /*10: no compress (LArc) */
+ {"-lhd-", 0, 0, 0}, /*11: directory */
+};
+
+struct lha_opts opts;
+
+struct lha_method *
+which_method(char *id)
+{
+ int i;
+
+ for (i = 0; i < sizeof(methods)/sizeof(methods[0]); i++) {
+ if (strncmp(id, methods[i].id, sizeof(methods[0].id)) == 0) {
+ return &methods[i];
+ }
+ }
+ return NULL;
+}
+
+static void
+print_usage()
+{
+ printf("%s", usage);
+ exit(0);
+}
+
+static void
+print_version()
+{
+ printf("version %s\n", version);
+ exit(0);
+}
+
+uint
+ratio(ulong a, ulong b)
+{ /* [(1000a + [b/2]) / b] */
+ int i;
+
+ for (i = 0; i < 3; i++)
+ if (a <= ULONG_MAX / 10)
+ a *= 10;
+ else
+ b /= 10;
+ if ((ulong) (a + (b >> 1)) < a) {
+ a >>= 1;
+ b >>= 1;
+ }
+ if (b == 0)
+ return 0;
+ return (uint) ((a + (b >> 1)) / b);
+}
+
+void
+skip(FILE *fp, struct lzh_header *h)
+{
+ int i;
+ if (opts.archive_to_stdio)
+ for (i = 0; i < h->compsize; i++)
+ fgetc(fp);
+ else
+ fseek(fp, h->compsize, SEEK_CUR);
+}
+
+static void
+copy(FILE *arcfile, FILE *outfile, struct lzh_header *h)
+{
+ uint n;
+ uchar buffer[MAXDICSIZ];
+
+ write_header(outfile, h);
+ while (h->compsize != 0) {
+ n = (uint) ((h->compsize > sizeof(buffer)) ? sizeof(buffer) : h->compsize);
+ if (fread((char *) buffer, 1, n, arcfile) != n)
+ error("Can't read");
+ if (fwrite((char *) buffer, 1, n, outfile) != n)
+ error("Can't write");
+ h->compsize -= n;
+ }
+}
+
+int
+get_line(char *s, int n)
+{
+ int i, c;
+
+ i = 0;
+ while ((c = getchar()) != EOF && c != '\n')
+ if (i < n)
+ s[i++] = (char) c;
+ s[i] = '\0';
+ return i;
+}
+
+static int
+match(char *s1, char *s2)
+{
+ for (;;) {
+ while (*s2 == '*' || *s2 == '?') {
+ if (*s2++ == '*')
+ while (*s1 && *s1 != *s2)
+ s1++;
+ else if (*s1 == 0)
+ return 0;
+ else
+ s1++;
+ }
+ if (*s1 != *s2)
+ return 0;
+ if (*s1 == 0)
+ return 1;
+ s1++;
+ s2++;
+ }
+}
+
+static int
+search(int argc, char *argv[], struct lzh_header *h)
+{
+ int i;
+
+ if (argc == 0)
+ return -1;
+ for (i = 0; i < argc; i++)
+ if (argv[i] && match(h->filename, argv[i]))
+ return i+1;
+ return 0;
+}
+
+#include "getopt_long.h"
+
+void
+parse_args(int argc, char **argv)
+{
+ int c;
+
+ for (;;) {
+ /* int this_option_optind = optind ? optind : 1; */
+ int option_index = 0;
+
+ enum {
+ LHA_OPT_HELP = 128,
+ LHA_OPT_VERSION,
+ };
+
+ static struct option long_options[] = {
+ /* name, has_arg, *flag, val */
+ /* has_arg:
+ no_argument (0)
+ required_argument (1)
+ optional_argument (2)
+ flag:
+ NULL: getopt_long() return val
+ non-NULL: getopt_long() return 0, and *flag set val.
+ */
+ {"help", no_argument, NULL, LHA_OPT_HELP},
+ {"version", no_argument, NULL, LHA_OPT_VERSION},
+ {0, 0, 0, 0}
+ };
+
+ c = getopt_long(argc, argv, "012fgo[567]q[012]vw:z",
+ long_options, &option_index);
+
+ if (c == -1) break; /* end of parsing options */
+
+ switch (c) {
+ case '?':
+ print_usage();
+ break;
+ case 0:
+ /* set value by long option */
+ break;
+ case '0': case '1': case '2':
+ /* header level */
+ opts.header_level = c - '0';
+ break;
+ case 'f':
+ opts.force_extract = 1;
+ break;
+ case 'g':
+ opts.generic = 1;
+ opts.header_level = 0;
+ break;
+ case 'o':
+ /* compress method */
+ {
+ int idx = 1; /* -o means -lh1- method */
+
+ if (optarg)
+ idx = *optarg - '0'; /* -lh[567]- method */
+
+ opts.method = &methods[idx];
+ }
+ break;
+ case 'q':
+ /* quiet mode */
+ opts.quiet = 2; /* -q is equivalent to -q2 */
+ if (optarg)
+ opts.quiet = *optarg - '0';
+ break;
+ case 'v':
+ /* verbose mode */
+ opts.verbose = 1;
+ break;
+
+ case 'w':
+ /* extract directory */
+ if (!optarg)
+ error("extract directory does not specified for `-w'");
+ if (*optarg == '=')
+ optarg++;
+
+ opts.outdir = optarg;
+ break;
+ case 'z': /* no compress */
+ opts.nocompress = 1;
+ break;
+ case LHA_OPT_HELP:
+ print_usage();
+ break;
+ case LHA_OPT_VERSION:
+ print_version();
+ break;
+ default:
+ break;
+ }
+ }
+}
+
+FILE *
+open_tempfile()
+{
+ FILE *outfile;
+
+ outfile = tmpfile();
+ if (outfile == NULL)
+ error("Can't open temporary file");
+
+ return outfile;
+}
+
+static void
+op_add(int cmd, char *archive_file, int argc, char **argv)
+{
+ int i, count, found, done;
+ struct lzh_header h;
+ struct lzh_ostream w, *wp;
+ FILE *arcfile = NULL;
+ FILE *outfile = NULL;
+
+ wp = &w;
+
+ count = done = 0;
+
+ outfile = open_tempfile();
+ wp->fp = outfile;
+ wp->buf = 0;
+ if (*argv == 0)
+ error("archived files are not specified.");
+
+ arcfile = fopen(archive_file, "rb");
+ if (arcfile == NULL)
+ error("Can't open archive '%s'", archive_file);
+
+ while (!done && read_header(arcfile, &h)) {
+
+ found = search(argc, argv, &h);
+ if (found>0) {
+ argv[found-1] = 0;
+
+ if (cmd == 'u') {
+ time_t mtime;
+
+ if (file_mtime(h.filename, &mtime) == -1 || h.mtime > mtime) {
+ copy(arcfile, outfile, &h);
+ continue;
+ }
+ }
+
+ if (add(wp, 1, h.filename, h.namelen)) {
+ skip(arcfile, &h);
+ count++;
+ }
+ else
+ copy(arcfile, outfile, &h);
+ }
+ else
+ copy(arcfile, outfile, &h);
+ }
+
+ for (i = 0; i < argc; i++) {
+ if (argv[i]) {
+ count++;
+ add(wp, 0, argv[i], strlen(argv[i]));
+ }
+ }
+
+ if (opts.quiet < 2)
+ printf(" %d files\n", count);
+
+ if (count > 0) {
+ fputc(0, outfile); /* end of archive */
+ if (ferror(outfile))
+ error("Can't write");
+ if (!opts.archive_to_stdio)
+ unlink(archive_file);
+
+ fclose(arcfile);
+ rewind(outfile);
+ if (copy_stream_to_file(outfile, archive_file) == -1)
+ error("fail to copy_stream_to_file(): temp -> %s",archive_file);
+ }
+}
+
+static void
+op_create(int cmd, char *archive_file, int argc, char **argv)
+{
+ int i;
+ struct lzh_ostream w, *wp;
+ FILE *outfile = NULL;
+
+ wp = &w;
+
+ outfile = open_tempfile();
+ wp->fp = outfile;
+ wp->buf = 0;
+ if (*argv == 0)
+ error("archived files are not specified.");
+
+ for (i = 0; i < argc; i++) {
+ add(wp, 0, argv[i], strlen(argv[i]));
+ }
+
+ fputc(0, outfile); /* end of archive */
+ if (ferror(outfile))
+ error("Can't write");
+ rewind(outfile);
+ if (opts.archive_to_stdio) {
+ if (copy_stream(outfile, stdout) == -1)
+ error("fail to copy_stream_to_file(): temp -> %s","stdout");
+ }
+ else {
+ if (copy_stream_to_file(outfile, archive_file) == -1)
+ error("fail to copy_stream_to_file(): temp -> %s",archive_file);
+ }
+
+ return;
+}
+
+static void
+op_delete(int cmd, char *archive_file, int argc, char **argv)
+{
+ int count, found, done;
+ struct lzh_header h;
+ int arc_count;
+ struct lzh_ostream w, *wp;
+ FILE *arcfile = NULL;
+ FILE *outfile = NULL;
+
+ wp = &w;
+
+ count = done = 0;
+
+ if (argc == 0) {
+ message("No files given in argument, do nothing.");
+ return;
+ }
+ outfile = open_tempfile();
+
+ /* Open archive. */
+ if (opts.archive_to_stdio) {
+ arcfile = stdin;
+ }
+ else {
+ arcfile = fopen(archive_file, "rb");
+ }
+ if (arcfile == NULL)
+ error("Can't open archive '%s'", archive_file);
+
+ arc_count = 0;
+
+ while (!done && read_header(arcfile, &h)) {
+
+ arc_count++;
+
+ found = search(argc, argv, &h);
+ if (found) {
+ count++;
+ message("'%s' deleted", h.filename);
+ skip(arcfile, &h);
+ }
+ else
+ copy(arcfile, outfile, &h);
+ }
+
+ if (opts.quiet < 2)
+ printf(" %d files\n", count);
+
+ if (count > 0) {
+ fputc(0, outfile); /* end of archive */
+ if (ferror(outfile))
+ error("Can't write");
+ if (!opts.archive_to_stdio)
+ unlink(archive_file);
+
+ fclose(arcfile);
+ rewind(outfile);
+
+ if (arc_count > count) {
+ if (copy_stream_to_file(outfile, archive_file) == -1)
+ error("fail to copy_stream_to_file(): temp -> %s",archive_file);
+ }
+ else {
+ message("The archive file \"%s\" was removed because it would be empty.", archive_file);
+ }
+ }
+}
+
+static void
+op_extract(int cmd, char *archive_file, int argc, char **argv)
+{
+ int count, nfiles, found, done;
+ struct lzh_header h;
+ struct lzh_istream r, *rp;
+ FILE *arcfile = NULL;
+
+ rp = &r;
+
+ count = done = nfiles = 0;
+
+ /* Open archive. */
+ if (opts.archive_to_stdio) {
+ arcfile = stdin;
+ }
+ else {
+ arcfile = fopen(archive_file, "rb");
+ }
+ if (arcfile == NULL)
+ error("Can't open archive '%s'", archive_file);
+
+ /* change directory to extract dir */
+ if (cmd == 'x') {
+ if (opts.outdir) {
+ if (mkdir(opts.outdir, 0777) == -1) {
+ if (errno != EEXIST)
+ error("cannot make directory \"%s\"", opts.outdir);
+ }
+
+ if (chdir(opts.outdir) == -1)
+ error("cannot change directory \"%s\"", opts.outdir);
+ }
+ }
+
+ while (!done && read_header(arcfile, &h)) {
+
+ found = search(argc, argv, &h);
+ if (found != 0) {
+ rp->fp = arcfile;
+ rp->compsize = h.compsize;
+ extract(rp, cmd == 'x', &h);
+ if (++count == nfiles)
+ done = 1;
+ }
+ else
+ skip(arcfile, &h);
+ }
+
+ if (cmd != 'p') {
+ if (opts.quiet < 2)
+ printf(" %d files\n", count);
+ }
+}
+
+static void
+op_list(int cmd, char *archive_file, int argc, char **argv)
+{
+ int count, nfiles, found, done;
+ struct lzh_header h;
+ struct lzh_istream r, *rp;
+ FILE *arcfile = NULL;
+
+ rp = &r;
+
+ count = done = nfiles = 0;
+
+ /* Open archive. */
+ if (opts.archive_to_stdio) {
+ arcfile = stdin;
+ }
+ else {
+ arcfile = fopen(archive_file, "rb");
+ }
+ if (arcfile == NULL)
+ error("Can't open archive '%s'", archive_file);
+
+ while (!done && read_header(arcfile, &h)) {
+
+ found = search(argc, argv, &h);
+
+ if (found != 0) {
+ if (count == 0)
+ list_start();
+ list(&h);
+ if (++count == nfiles)
+ done = 1;
+ }
+ skip(arcfile, &h);
+ }
+
+ if (opts.quiet < 2)
+ printf(" %d files\n", count);
+}
+
+int
+main(int argc, char *argv[])
+{
+ int cmd;
+ char *archive_file;
+
+ INITIALIZE_OPTS(opts);
+
+ if (argv[1] == 0)
+ print_usage();
+
+ /*take a command character */
+ {
+ char *arg1;
+
+ arg1 = argv[1];
+ if (arg1[0] == '-')
+ arg1++;
+ if (arg1[0] == 0)
+ print_usage();
+
+ cmd = *arg1;
+ if (arg1[1] == 0) {
+ /* -<cmd> -<opts> ... */
+ argv++;
+ argc--;
+ }
+ else {
+ /* -<cmd><opts> => -<opts> */
+ *arg1 = '-';
+ }
+ }
+
+ parse_args(argc, argv);
+ argv += optind;
+ argc -= optind;
+
+ archive_file = argv[0];
+
+ if (strcmp(archive_file, "-") == 0)
+ opts.archive_to_stdio = 1;
+ if (opts.archive_to_stdio)
+ opts.quiet = 2;
+
+ argv++;
+ argc--;
+
+ make_crctable();
+
+ switch (cmd) {
+ case 'a':
+ case 'u':
+ if (opts.archive_to_stdio || !file_exists(archive_file)) {
+ op_create(cmd, archive_file, argc, argv);
+ }
+ else {
+ op_add(cmd, archive_file, argc, argv);
+ }
+ break;
+
+ case 'c':
+ op_create(cmd, archive_file, argc, argv);
+ break;
+
+ case 'd':
+ op_delete(cmd, archive_file, argc, argv);
+ break;
+
+ case 'x':
+ case 'p':
+ op_extract(cmd, archive_file, argc, argv);
+ break;
+
+ case 'l':
+ case 'v':
+ op_list(cmd, archive_file, argc, argv);
+ break;
+
+ default:
+ print_usage();
+ break;
+ }
+ return EXIT_SUCCESS;
+}
OSDN Git Service
