OpenSolaris_b135/cmd/split/split.c
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 1999 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
/* Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
/* All Rights Reserved */
#pragma ident "%Z%%M% %I% %E% SMI"
#include <stdio.h>
#include <sys/types.h>
#include <sys/statvfs.h>
#include <locale.h>
#include <malloc.h>
#include <string.h>
#include <sys/param.h>
#include <stdlib.h>
#include <wchar.h>
#include <widec.h>
#include <ctype.h>
#include <errno.h>
#define DEFAULT_LINES 1000
#define ONE_K 1024
#define ONE_M ONE_K*ONE_K
#ifndef TRUE
#define TRUE 1
#define FALSE 0
#endif
static void Usage();
static void next_file_name();
static char *progname;
static int suffix_length = 2;
int
main(int argc, char **argv)
{
long long line_count = 0;
long long byte_count = 0;
long long out;
char *fname = NULL;
char head[MAXPATHLEN];
char *output_file_name;
char *tail;
char *last;
FILE *in_file = NULL;
FILE *out_file = (FILE *)NULL;
int i;
int c;
wint_t wc;
int output_file_open;
struct statvfs stbuf;
int opt;
int non_standard_line_count = FALSE;
(void) setlocale(LC_ALL, "");
#if !defined(TEXT_DOMAIN) /* Should be defined by cc -D */
#define TEXT_DOMAIN "SYS_TEST" /* Use this only if it weren't */
#endif
(void) textdomain(TEXT_DOMAIN);
progname = argv[0];
/* check for explicit stdin "-" option */
for (i = 1; i < argc; i++) {
if (strcmp(argv[i], "-") == 0) {
in_file = stdin;
while (i < argc) {
argv[i] = argv[i + 1];
/* a "-" before "--" is an error */
if ((argv[i] != NULL) &&
(strcmp(argv[i], "--") == 0)) {
Usage();
}
i++;
}
argc--;
}
}
/* check for non-standard "-line-count" option */
for (i = 1; i < argc; i++) {
if (strcmp(argv[i], "--") == 0)
break;
if ((argv[i][0] == '-') && isdigit(argv[i][1])) {
if (strlen(&argv[i][1]) !=
strspn(&argv[i][1], "0123456789")) {
(void) fprintf(stderr, gettext(
"%s: Badly formed number\n"), progname);
Usage();
}
line_count = (long long) strtoll(&argv[i][1],
(char **)NULL, 10);
non_standard_line_count = TRUE;
while (i < argc) {
argv[i] = argv[i + 1];
i++;
}
argc--;
}
}
/* get options */
while ((opt = getopt(argc, argv, "a:b:l:")) != EOF) {
switch (opt) {
case 'a':
if (strcmp(optarg, "--") == 0) {
Usage();
}
suffix_length = (long long) strtoll(optarg,
(char **)NULL, 10);
if (suffix_length <= 0) {
(void) fprintf(stderr, gettext(
"%s: Invalid \"-a %s\" option\n"),
progname, optarg);
Usage();
}
break;
case 'b':
if (strcmp(optarg, "--") == 0) {
Usage();
}
byte_count = (long long) strtoll(optarg,
(char **)NULL, 10);
if (*(optarg + strspn(optarg, "0123456789")) == 'k')
byte_count *= ONE_K;
if (*(optarg + strspn(optarg, "0123456789")) == 'm')
byte_count *= ONE_M;
break;
case 'l':
if (strcmp(optarg, "--") == 0) {
Usage();
}
if (non_standard_line_count == TRUE) {
Usage();
}
line_count = (long long) strtoll(optarg,
(char **)NULL, 10);
break;
default:
Usage();
}
}
/* get input file */
if ((in_file == NULL) && (optind < argc)) {
if ((in_file = fopen(argv[optind++], "r")) == NULL) {
(void) perror("split");
return (1);
}
}
if (in_file == NULL) {
in_file = stdin;
}
/* get output file name */
if (optind < argc) {
output_file_name = argv[optind];
if ((tail = strrchr(output_file_name, '/')) == NULL) {
tail = output_file_name;
(void) getcwd(head, sizeof (head));
} else {
tail++;
(void) strcpy(head, output_file_name);
last = strrchr(head, '/');
*++last = '\0';
}
if (statvfs(head, &stbuf) < 0) {
perror(head);
return (1);
}
if (strlen(tail) > (stbuf.f_namemax - suffix_length)) {
(void) fprintf(stderr, gettext(
"%s: More than %d characters in file name\n"),
progname, stbuf.f_namemax - suffix_length);
Usage();
}
} else
output_file_name = "x";
/* check options */
if (((int)strlen(output_file_name) + suffix_length) > FILENAME_MAX) {
(void) fprintf(stderr, gettext(
"%s: Output file name too long\n"), progname);
return (1);
}
if (line_count && byte_count) {
Usage();
}
/* use default line count if none specified */
if (line_count == 0) {
line_count = DEFAULT_LINES;
}
/*
* allocate buffer for the filenames we'll construct; it must be
* big enough to hold the name in 'output_file_name' + an n-char
* suffix + NULL terminator
*/
if ((fname = (char *)malloc(strlen(output_file_name) +
suffix_length + 1)) == NULL) {
(void) perror("split");
return (1);
}
/* build first output file name */
for (i = 0; output_file_name[i]; i++) {
fname[i] = output_file_name[i];
}
while (i < (int)strlen(output_file_name) + suffix_length) {
fname[i++] = 'a';
}
if (suffix_length)
fname[i - 1] = 'a' - 1;
fname[i] = '\0';
for (; ; ) {
output_file_open = FALSE;
if (byte_count) {
for (out = 0; out < byte_count; out++) {
errno = 0;
c = getc(in_file);
if (c == EOF) {
if (errno != 0) {
int lerrno = errno;
(void) fprintf(stderr, gettext(
"%s: Read error at file offset %lld: %s, "
"aborting split\n"),
progname, ftello(in_file),
strerror(lerrno));
if (output_file_open == TRUE)
(void) fclose(out_file);
free(fname);
return (1);
}
if (output_file_open == TRUE)
(void) fclose(out_file);
free(fname);
return (0);
}
if (output_file_open == FALSE) {
next_file_name(fname);
if ((out_file = fopen(fname, "w")) == NULL) {
(void) perror("split");
free(fname);
return (1);
}
output_file_open = TRUE;
}
if (putc(c, out_file) == EOF) {
perror("split");
if (output_file_open == TRUE)
(void) fclose(out_file);
free(fname);
return (1);
}
}
} else {
for (out = 0; out < line_count; out++) {
do {
errno = 0;
wc = getwc(in_file);
if (wc == WEOF) {
if (errno != 0) {
if (errno == EILSEQ) {
(void) fprintf(stderr, gettext(
"%s: Invalid multibyte sequence "
"encountered at file offset %lld, "
"aborting split\n"),
progname, ftello(in_file));
} else {
(void) perror("split");
}
if (output_file_open == TRUE)
(void) fclose(out_file);
free(fname);
return (1);
}
if (output_file_open == TRUE)
(void) fclose(out_file);
free(fname);
return (0);
}
if (output_file_open == FALSE) {
next_file_name(fname);
if ((out_file = fopen(fname, "w")) == NULL) {
(void) perror("split");
free(fname);
return (1);
}
output_file_open = TRUE;
}
if (putwc(wc, out_file) == WEOF) {
(void) perror("split");
if (output_file_open == TRUE)
(void) fclose(out_file);
free(fname);
return (1);
}
} while (wc != '\n');
}
}
if (output_file_open == TRUE)
(void) fclose(out_file);
}
/*NOTREACHED*/
}
static void
next_file_name(char *name)
{
int i;
i = strlen(name) - 1;
while (i >= (int)(strlen(name) - suffix_length)) {
if (++name[i] <= 'z')
return;
name[i--] = 'a';
}
(void) fprintf(stderr, gettext(
"%s: Exhausted output file names, aborting split\n"),
progname);
exit(1);
}
static void
Usage()
{
(void) fprintf(stderr, gettext(
"Usage: %s [-l #] [-a #] [file [name]]\n"
" %s [-b #[k|m]] [-a #] [file [name]]\n"
" %s [-#] [-a #] [file [name]]\n"),
progname, progname, progname);
exit(1);
}