1 /*
2 * Copyright 2005 Sun Microsystems, Inc. All rights reserved.
3 * Use is subject to license terms.
4 */
5
6 /* Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
7 /* All Rights Reserved */
8
9 /*
10 * Copyright (c) 1980 Regents of the University of California.
11 * All rights reserved. The Berkeley software License Agreement
12 * specifies the terms and conditions for redistribution.
13 */
14
15 #pragma ident "%Z%%M% %I% %E% SMI"
16
17 #include <stdio.h>
18 #include <locale.h>
19 #include <assert.h>
20
21 extern void err();
22 extern int newkeys();
23 extern int recopy();
24 extern void whash();
25
26 int
main(int argc,char * argv[])27 main(int argc, char *argv[])
28 {
29 /*
30 * Make inverted file indexes. Reads a stream from mkey which
31 * gives record pointer items and keys. Generates set of files
32 * a. NHASH pointers to file b.
33 * b. lists of record numbers.
34 * c. record pointer items.
35 *
36 * these files are named xxx.ia, xxx.ib, xxx.ic;
37 * where xxx is taken from arg1.
38 * If the files exist they are updated.
39 */
40
41 FILE *fa, *fb, *fc, *fta, *ftb, *ftc;
42 FILE *fd = NULL;
43 int nhash = 256;
44 int appflg = 1;
45 int keepkey = 0, pipein = 0;
46 char nma[100], nmb[100], nmc[100], com[100], nmd[100];
47 char tmpa[20], tmpb[20], tmpc[20];
48 char *remove = NULL;
49 int chatty = 0, docs, hashes, fp[2], fr, fw, pfork, pwait, status;
50 int i, j, k;
51 long keys;
52 int iflong = 0;
53 char *sortdir;
54
55 (void) setlocale(LC_ALL, "");
56
57 #if !defined(TEXT_DOMAIN)
58 #define TEXT_DOMAIN "SYS_TEST"
59 #endif
60 (void) textdomain(TEXT_DOMAIN);
61
62 sortdir = (access("/crp/tmp", 06) == 0) ? "/crp/tmp" : "/usr/tmp";
63 while (argc > 1 && argv[1][0] == '-') {
64 switch (argv[1][1]) {
65 case 'h': /* size of hash table */
66 nhash = atoi(argv[1]+2);
67 break;
68 case 'n': /* new, don't append */
69 appflg = 0;
70 break;
71 case 'a': /* append to old file */
72 appflg = 1;
73 break;
74 case 'v': /* verbose output */
75 chatty = 1;
76 break;
77 case 'd': /* keep keys on file .id for check on searching */
78 keepkey = 1;
79 break;
80 case 'p': /* pipe into sort (saves space, costs time) */
81 pipein = 1;
82 break;
83 case 'i': /* input is on file, not stdin */
84 close(0);
85 if (open(argv[2], 0) != 0)
86 err(gettext("Can't read input %s"), argv[2]);
87 if (argv[1][2] == 'u') /* unlink */
88 remove = argv[2];
89 argc--;
90 argv++;
91 break;
92 }
93 argc--;
94 argv++;
95 }
96 strcpy(nma, argc >= 2 ? argv[1] : "Index");
97 strcpy(nmb, nma);
98 strcpy(nmc, nma);
99 strcpy(nmd, nma);
100 strcat(nma, ".ia");
101 strcat(nmb, ".ib");
102 strcat(nmc, ".ic");
103 strcat(nmd, ".id");
104
105 sprintf(tmpa, "junk%di", getpid());
106 if (pipein) {
107 sprintf(com, "/usr/bin/sort -T %s -o %s", sortdir, tmpa);
108 fta = popen(com, "w");
109 } else { /* use tmp file */
110 fta = fopen(tmpa, "w");
111 assert(fta != NULL);
112 }
113 fb = 0;
114 if (appflg) {
115 if (fb = fopen(nmb, "r")) {
116 sprintf(tmpb, "junk%dj", getpid());
117 ftb = fopen(tmpb, "w");
118 if (ftb == NULL)
119 err(gettext("Can't get scratch file %s"), tmpb);
120 nhash = recopy(ftb, fb, fopen(nma, "r"));
121 fclose(ftb);
122 } else
123 appflg = 0;
124 }
125 fc = fopen(nmc, appflg ? "a" : "w");
126 if (keepkey)
127 fd = keepkey ? fopen(nmd, "w") : 0;
128 docs = newkeys(fta, stdin, fc, nhash, fd, &iflong);
129 fclose(stdin);
130 if (remove != NULL)
131 unlink(remove);
132 fclose(fta);
133 if (pipein) {
134 pclose(fta);
135 }
136 else
137 {
138 sprintf(com, "sort -T %s %s -o %s", sortdir, tmpa, tmpa);
139 system(com);
140 }
141 if (appflg) {
142 sprintf(tmpc, "junk%dk", getpid());
143 sprintf(com, "mv %s %s", tmpa, tmpc);
144 system(com);
145 sprintf(com, "sort -T %s -m %s %s -o %s", sortdir,
146 tmpb, tmpc, tmpa);
147 system(com);
148 }
149 fta = fopen(tmpa, "r");
150 fa = fopen(nma, "w");
151 fb = fopen(nmb, "w");
152 whash(fta, fa, fb, nhash, iflong, &keys, &hashes);
153 fclose(fta);
154 #ifndef D1
155 unlink(tmpa);
156 #endif
157 if (appflg) {
158 unlink(tmpb);
159 unlink(tmpc);
160 }
161 if (chatty)
162 printf(gettext("%ld key occurrences, %d hashes, %d docs\n"),
163 keys, hashes, docs);
164
165 return (0);
166 }
167