1 /*
2 * Copyright (c) 1987, 1993, 1994
3 * The Regents of the University of California. All rights reserved.
4 *
5 * %sccs.include.redist.c%
6 */
7
8 #ifndef lint
9 static char copyright[] =
10 "@(#) Copyright (c) 1987, 1993, 1994\n\
11 The Regents of the University of California. All rights reserved.\n";
12 #endif /* not lint */
13
14 #ifndef lint
15 static char sccsid[] = "@(#)split.c 8.3 (Berkeley) 04/25/94";
16 #endif /* not lint */
17
18 #include <sys/param.h>
19
20 #include <ctype.h>
21 #include <err.h>
22 #include <fcntl.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <unistd.h>
27
28 #define DEFLINE 1000 /* Default num lines per file. */
29
30 long bytecnt; /* Byte count to split on. */
31 long numlines; /* Line count to split on. */
32 int file_open; /* If a file open. */
33 int ifd = -1, ofd = -1; /* Input/output file descriptors. */
34 char bfr[MAXBSIZE]; /* I/O buffer. */
35 char fname[MAXPATHLEN]; /* File name prefix. */
36
37 void newfile __P((void));
38 void split1 __P((void));
39 void split2 __P((void));
40 void usage __P((void));
41
42 int
main(argc,argv)43 main(argc, argv)
44 int argc;
45 char *argv[];
46 {
47 int ch;
48 char *ep, *p;
49
50 while ((ch = getopt(argc, argv, "-0123456789b:l:")) != EOF)
51 switch (ch) {
52 case '0': case '1': case '2': case '3': case '4':
53 case '5': case '6': case '7': case '8': case '9':
54 /*
55 * Undocumented kludge: split was originally designed
56 * to take a number after a dash.
57 */
58 if (numlines == 0) {
59 p = argv[optind - 1];
60 if (p[0] == '-' && p[1] == ch && !p[2])
61 numlines = strtol(++p, &ep, 10);
62 else
63 numlines =
64 strtol(argv[optind] + 1, &ep, 10);
65 if (numlines <= 0 || *ep)
66 errx(1,
67 "%s: illegal line count.", optarg);
68 }
69 break;
70 case '-': /* Undocumented: historic stdin flag. */
71 if (ifd != -1)
72 usage();
73 ifd = 0;
74 break;
75 case 'b': /* Byte count. */
76 if ((bytecnt = strtol(optarg, &ep, 10)) <= 0 ||
77 *ep != '\0' && *ep != 'k' && *ep != 'm')
78 errx(1, "%s: illegal byte count.", optarg);
79 if (*ep == 'k')
80 bytecnt *= 1024;
81 else if (*ep == 'm')
82 bytecnt *= 1048576;
83 break;
84 case 'l': /* Line count. */
85 if (numlines != 0)
86 usage();
87 if ((numlines = strtol(optarg, &ep, 10)) <= 0 || *ep)
88 errx(1, "%s: illegal line count.", optarg);
89 break;
90 default:
91 usage();
92 }
93 argv += optind;
94 argc -= optind;
95
96 if (*argv != NULL)
97 if (ifd == -1) { /* Input file. */
98 if ((ifd = open(*argv, O_RDONLY, 0)) < 0)
99 err(1, "%s", *argv);
100 ++argv;
101 }
102 if (*argv != NULL) /* File name prefix. */
103 (void)strcpy(fname, *argv++);
104 if (*argv != NULL)
105 usage();
106
107 if (numlines == 0)
108 numlines = DEFLINE;
109 else if (bytecnt)
110 usage();
111
112 if (ifd == -1) /* Stdin by default. */
113 ifd = 0;
114
115 if (bytecnt) {
116 split1();
117 exit (0);
118 }
119 split2();
120 exit(0);
121 }
122
123 /*
124 * split1 --
125 * Split the input by bytes.
126 */
127 void
split1()128 split1()
129 {
130 long bcnt;
131 int dist, len;
132 char *C;
133
134 for (bcnt = 0;;)
135 switch (len = read(ifd, bfr, MAXBSIZE)) {
136 case 0:
137 exit(0);
138 case -1:
139 err(1, "read");
140 /* NOTREACHED */
141 default:
142 if (!file_open) {
143 newfile();
144 file_open = 1;
145 }
146 if (bcnt + len >= bytecnt) {
147 dist = bytecnt - bcnt;
148 if (write(ofd, bfr, dist) != dist)
149 err(1, "write");
150 len -= dist;
151 for (C = bfr + dist; len >= bytecnt;
152 len -= bytecnt, C += bytecnt) {
153 newfile();
154 if (write(ofd,
155 C, (int)bytecnt) != bytecnt)
156 err(1, "write");
157 }
158 if (len) {
159 newfile();
160 if (write(ofd, C, len) != len)
161 err(1, "write");
162 } else
163 file_open = 0;
164 bcnt = len;
165 } else {
166 bcnt += len;
167 if (write(ofd, bfr, len) != len)
168 err(1, "write");
169 }
170 }
171 }
172
173 /*
174 * split2 --
175 * Split the input by lines.
176 */
177 void
split2()178 split2()
179 {
180 long lcnt;
181 int len, bcnt;
182 char *Ce, *Cs;
183
184 for (lcnt = 0;;)
185 switch (len = read(ifd, bfr, MAXBSIZE)) {
186 case 0:
187 exit(0);
188 case -1:
189 err(1, "read");
190 /* NOTREACHED */
191 default:
192 if (!file_open) {
193 newfile();
194 file_open = 1;
195 }
196 for (Cs = Ce = bfr; len--; Ce++)
197 if (*Ce == '\n' && ++lcnt == numlines) {
198 bcnt = Ce - Cs + 1;
199 if (write(ofd, Cs, bcnt) != bcnt)
200 err(1, "write");
201 lcnt = 0;
202 Cs = Ce + 1;
203 if (len)
204 newfile();
205 else
206 file_open = 0;
207 }
208 if (Cs < Ce) {
209 bcnt = Ce - Cs;
210 if (write(ofd, Cs, bcnt) != bcnt)
211 err(1, "write");
212 }
213 }
214 }
215
216 /*
217 * newfile --
218 * Open a new output file.
219 */
220 void
newfile()221 newfile()
222 {
223 static long fnum;
224 static int defname;
225 static char *fpnt;
226
227 if (ofd == -1) {
228 if (fname[0] == '\0') {
229 fname[0] = 'x';
230 fpnt = fname + 1;
231 defname = 1;
232 } else {
233 fpnt = fname + strlen(fname);
234 defname = 0;
235 }
236 ofd = fileno(stdout);
237 }
238 /*
239 * Hack to increase max files; original code wandered through
240 * magic characters. Maximum files is 3 * 26 * 26 == 2028
241 */
242 #define MAXFILES 676
243 if (fnum == MAXFILES) {
244 if (!defname || fname[0] == 'z')
245 errx(1, "too many files.");
246 ++fname[0];
247 fnum = 0;
248 }
249 fpnt[0] = fnum / 26 + 'a';
250 fpnt[1] = fnum % 26 + 'a';
251 ++fnum;
252 if (!freopen(fname, "w", stdout))
253 err(1, "%s", fname);
254 }
255
256 void
usage()257 usage()
258 {
259 (void)fprintf(stderr,
260 "usage: split [-b byte_count] [-l line_count] [file [prefix]]\n");
261 exit(1);
262 }
263