1 76193d7c 2003-09-30 devnull #include	"mk.h"
  2 76193d7c 2003-09-30 devnull #define	ARMAG	"!<arch>\n"
  3 76193d7c 2003-09-30 devnull #define	SARMAG	8
  4 76193d7c 2003-09-30 devnull 
  5 76193d7c 2003-09-30 devnull #define	ARFMAG	"`\n"
  6 76193d7c 2003-09-30 devnull #define SARNAME	16
  7 76193d7c 2003-09-30 devnull 
  8 76193d7c 2003-09-30 devnull struct	ar_hdr
  9 76193d7c 2003-09-30 devnull {
 10 76193d7c 2003-09-30 devnull 	char	name[SARNAME];
 11 76193d7c 2003-09-30 devnull 	char	date[12];
 12 76193d7c 2003-09-30 devnull 	char	uid[6];
 13 76193d7c 2003-09-30 devnull 	char	gid[6];
 14 76193d7c 2003-09-30 devnull 	char	mode[8];
 15 76193d7c 2003-09-30 devnull 	char	size[10];
 16 76193d7c 2003-09-30 devnull 	char	fmag[2];
 17 76193d7c 2003-09-30 devnull };
 18 76193d7c 2003-09-30 devnull #define	SAR_HDR	(SARNAME+44)
 19 76193d7c 2003-09-30 devnull 
 20 92a0a8b6 2004-04-21 devnull static int dolong = 1;
 21 76193d7c 2003-09-30 devnull 
 22 76193d7c 2003-09-30 devnull static void atimes(char *);
 23 76193d7c 2003-09-30 devnull static char *split(char*, char**);
 24 76193d7c 2003-09-30 devnull 
 25 76193d7c 2003-09-30 devnull long
 26 2b9172c7 2004-12-27 devnull readn(int f, void *av, long n)
 27 2b9172c7 2004-12-27 devnull {
 28 2b9172c7 2004-12-27 devnull 	char *a;
 29 2b9172c7 2004-12-27 devnull 	long m, t;
 30 2b9172c7 2004-12-27 devnull 
 31 2b9172c7 2004-12-27 devnull 	a = av;
 32 2b9172c7 2004-12-27 devnull 	t = 0;
 33 2b9172c7 2004-12-27 devnull 	while(t < n){
 34 2b9172c7 2004-12-27 devnull 		m = read(f, a+t, n-t);
 35 2b9172c7 2004-12-27 devnull 		if(m <= 0){
 36 2b9172c7 2004-12-27 devnull 			if(t == 0)
 37 2b9172c7 2004-12-27 devnull 				return m;
 38 2b9172c7 2004-12-27 devnull 			break;
 39 2b9172c7 2004-12-27 devnull 		}
 40 2b9172c7 2004-12-27 devnull 		t += m;
 41 2b9172c7 2004-12-27 devnull 	}
 42 2b9172c7 2004-12-27 devnull 	return t;
 43 2b9172c7 2004-12-27 devnull }
 44 2b9172c7 2004-12-27 devnull long
 45 76193d7c 2003-09-30 devnull atimeof(int force, char *name)
 46 76193d7c 2003-09-30 devnull {
 47 76193d7c 2003-09-30 devnull 	Symtab *sym;
 48 76193d7c 2003-09-30 devnull 	long t;
 49 76193d7c 2003-09-30 devnull 	char *archive, *member, buf[512];
 50 76193d7c 2003-09-30 devnull 
 51 76193d7c 2003-09-30 devnull 	archive = split(name, &member);
 52 76193d7c 2003-09-30 devnull 	if(archive == 0)
 53 76193d7c 2003-09-30 devnull 		Exit();
 54 76193d7c 2003-09-30 devnull 
 55 76193d7c 2003-09-30 devnull 	t = mtime(archive);
 56 76193d7c 2003-09-30 devnull 	sym = symlook(archive, S_AGG, 0);
 57 76193d7c 2003-09-30 devnull 	if(sym){
 58 76193d7c 2003-09-30 devnull 		if(force || (t > (long)sym->value)){
 59 76193d7c 2003-09-30 devnull 			atimes(archive);
 60 76193d7c 2003-09-30 devnull 			sym->value = (void *)t;
 61 76193d7c 2003-09-30 devnull 		}
 62 76193d7c 2003-09-30 devnull 	}
 63 76193d7c 2003-09-30 devnull 	else{
 64 76193d7c 2003-09-30 devnull 		atimes(archive);
 65 76193d7c 2003-09-30 devnull 		/* mark the aggegate as having been done */
 66 76193d7c 2003-09-30 devnull 		symlook(strdup(archive), S_AGG, "")->value = (void *)t;
 67 76193d7c 2003-09-30 devnull 	}
 68 76193d7c 2003-09-30 devnull 		/* truncate long member name to sizeof of name field in archive header */
 69 76193d7c 2003-09-30 devnull 	if(dolong)
 70 76193d7c 2003-09-30 devnull 		snprint(buf, sizeof(buf), "%s(%s)", archive, member);
 71 76193d7c 2003-09-30 devnull 	else
 72 76193d7c 2003-09-30 devnull 		snprint(buf, sizeof(buf), "%s(%.*s)", archive, SARNAME, member);
 73 76193d7c 2003-09-30 devnull 	sym = symlook(buf, S_TIME, 0);
 74 76193d7c 2003-09-30 devnull 	if (sym)
 75 76193d7c 2003-09-30 devnull 		return (long)sym->value;	/* uggh */
 76 76193d7c 2003-09-30 devnull 	return 0;
 77 76193d7c 2003-09-30 devnull }
 78 76193d7c 2003-09-30 devnull 
 79 76193d7c 2003-09-30 devnull void
 80 76193d7c 2003-09-30 devnull atouch(char *name)
 81 76193d7c 2003-09-30 devnull {
 82 76193d7c 2003-09-30 devnull 	char *archive, *member;
 83 76193d7c 2003-09-30 devnull 	int fd, i;
 84 76193d7c 2003-09-30 devnull 	struct ar_hdr h;
 85 76193d7c 2003-09-30 devnull 	long t;
 86 76193d7c 2003-09-30 devnull 
 87 76193d7c 2003-09-30 devnull 	archive = split(name, &member);
 88 76193d7c 2003-09-30 devnull 	if(archive == 0)
 89 76193d7c 2003-09-30 devnull 		Exit();
 90 76193d7c 2003-09-30 devnull 
 91 76193d7c 2003-09-30 devnull 	fd = open(archive, ORDWR);
 92 76193d7c 2003-09-30 devnull 	if(fd < 0){
 93 76193d7c 2003-09-30 devnull 		fd = create(archive, OWRITE, 0666);
 94 76193d7c 2003-09-30 devnull 		if(fd < 0){
 95 76193d7c 2003-09-30 devnull 			fprint(2, "create %s: %r\n", archive);
 96 76193d7c 2003-09-30 devnull 			Exit();
 97 76193d7c 2003-09-30 devnull 		}
 98 76193d7c 2003-09-30 devnull 		write(fd, ARMAG, SARMAG);
 99 76193d7c 2003-09-30 devnull 	}
100 76193d7c 2003-09-30 devnull 	if(symlook(name, S_TIME, 0)){
101 76193d7c 2003-09-30 devnull 		/* hoon off and change it in situ */
102 76193d7c 2003-09-30 devnull 		LSEEK(fd, SARMAG, 0);
103 76193d7c 2003-09-30 devnull 		while(read(fd, (char *)&h, sizeof(h)) == sizeof(h)){
104 76193d7c 2003-09-30 devnull 			for(i = SARNAME-1; i > 0 && h.name[i] == ' '; i--)
105 76193d7c 2003-09-30 devnull 					;
106 76193d7c 2003-09-30 devnull 			h.name[i+1]=0;
107 76193d7c 2003-09-30 devnull 			if(strcmp(member, h.name) == 0){
108 76193d7c 2003-09-30 devnull 				t = SARNAME-sizeof(h);	/* ughgghh */
109 76193d7c 2003-09-30 devnull 				LSEEK(fd, t, 1);
110 76193d7c 2003-09-30 devnull 				fprint(fd, "%-12ld", time(0));
111 76193d7c 2003-09-30 devnull 				break;
112 76193d7c 2003-09-30 devnull 			}
113 76193d7c 2003-09-30 devnull 			t = atol(h.size);
114 76193d7c 2003-09-30 devnull 			if(t&01) t++;
115 76193d7c 2003-09-30 devnull 			LSEEK(fd, t, 1);
116 76193d7c 2003-09-30 devnull 		}
117 76193d7c 2003-09-30 devnull 	}
118 76193d7c 2003-09-30 devnull 	close(fd);
119 76193d7c 2003-09-30 devnull }
120 76193d7c 2003-09-30 devnull 
121 76193d7c 2003-09-30 devnull static void
122 76193d7c 2003-09-30 devnull atimes(char *ar)
123 76193d7c 2003-09-30 devnull {
124 76193d7c 2003-09-30 devnull 	struct ar_hdr h;
125 76193d7c 2003-09-30 devnull 	long t;
126 d946e4dc 2004-04-21 devnull 	int fd, i, namelen;
127 d946e4dc 2004-04-21 devnull 	char buf[2048], *p, *strings;
128 d946e4dc 2004-04-21 devnull 	char name[1024];
129 d946e4dc 2004-04-21 devnull 	Symtab *sym;
130 76193d7c 2003-09-30 devnull 
131 d946e4dc 2004-04-21 devnull 	strings = nil;
132 76193d7c 2003-09-30 devnull 	fd = open(ar, OREAD);
133 76193d7c 2003-09-30 devnull 	if(fd < 0)
134 76193d7c 2003-09-30 devnull 		return;
135 76193d7c 2003-09-30 devnull 
136 76193d7c 2003-09-30 devnull 	if(read(fd, buf, SARMAG) != SARMAG){
137 76193d7c 2003-09-30 devnull 		close(fd);
138 76193d7c 2003-09-30 devnull 		return;
139 76193d7c 2003-09-30 devnull 	}
140 d946e4dc 2004-04-21 devnull 	while(readn(fd, (char *)&h, sizeof(h)) == sizeof(h)){
141 76193d7c 2003-09-30 devnull 		t = atol(h.date);
142 76193d7c 2003-09-30 devnull 		if(t == 0)	/* as it sometimes happens; thanks ken */
143 76193d7c 2003-09-30 devnull 			t = 1;
144 d946e4dc 2004-04-21 devnull 		namelen = 0;
145 d946e4dc 2004-04-21 devnull 		if(memcmp(h.name, "#1/", 3) == 0){	/* BSD */
146 d946e4dc 2004-04-21 devnull 			namelen = atoi(h.name+3);
147 d946e4dc 2004-04-21 devnull 			if(namelen >= sizeof name){
148 d946e4dc 2004-04-21 devnull 				namelen = 0;
149 d946e4dc 2004-04-21 devnull 				goto skip;
150 d946e4dc 2004-04-21 devnull 			}
151 d946e4dc 2004-04-21 devnull 			if(readn(fd, name, namelen) != namelen)
152 d946e4dc 2004-04-21 devnull 				break;
153 d946e4dc 2004-04-21 devnull 			name[namelen] = 0;
154 d946e4dc 2004-04-21 devnull 		}else if(memcmp(h.name, "// ", 2) == 0){ /* GNU */
155 d946e4dc 2004-04-21 devnull 			/* date, uid, gid, mode all ' ' */
156 d946e4dc 2004-04-21 devnull 			for(i=2; i<16+12+6+6+8; i++)
157 d946e4dc 2004-04-21 devnull 				if(h.name[i] != ' ')
158 d946e4dc 2004-04-21 devnull 					goto skip;
159 d946e4dc 2004-04-21 devnull 			t = atol(h.size);
160 d946e4dc 2004-04-21 devnull 			if(t&01)
161 d946e4dc 2004-04-21 devnull 				t++;
162 d946e4dc 2004-04-21 devnull 			free(strings);
163 d946e4dc 2004-04-21 devnull 			strings = malloc(t+1);
164 d946e4dc 2004-04-21 devnull 			if(strings){
165 d946e4dc 2004-04-21 devnull 				if(readn(fd, strings, t) != t){
166 d946e4dc 2004-04-21 devnull 					free(strings);
167 d946e4dc 2004-04-21 devnull 					strings = nil;
168 d946e4dc 2004-04-21 devnull 					break;
169 d946e4dc 2004-04-21 devnull 				}
170 d946e4dc 2004-04-21 devnull 				strings[t] = 0;
171 d946e4dc 2004-04-21 devnull 				continue;
172 d946e4dc 2004-04-21 devnull 			}
173 d946e4dc 2004-04-21 devnull 			goto skip;
174 74374cc8 2005-10-31 devnull 		}else if(strings && h.name[0]=='/' && isdigit((uchar)h.name[1])){
175 d946e4dc 2004-04-21 devnull 			i = strtol(h.name+1, &p, 10);
176 92a0a8b6 2004-04-21 devnull 			if(*p != ' ' || i >= strlen(strings))
177 d946e4dc 2004-04-21 devnull 				goto skip;
178 d946e4dc 2004-04-21 devnull 			p = strings+i;
179 d946e4dc 2004-04-21 devnull 			for(; *p && *p != '/'; p++)
180 76193d7c 2003-09-30 devnull 				;
181 d946e4dc 2004-04-21 devnull 			namelen = p-(strings+i);
182 d946e4dc 2004-04-21 devnull 			if(namelen >= sizeof name){
183 d946e4dc 2004-04-21 devnull 				namelen = 0;
184 d946e4dc 2004-04-21 devnull 				goto skip;
185 d946e4dc 2004-04-21 devnull 			}
186 d946e4dc 2004-04-21 devnull 			memmove(name, strings+i, namelen);
187 d946e4dc 2004-04-21 devnull 			name[namelen] = 0;
188 d946e4dc 2004-04-21 devnull 			namelen = 0;
189 d946e4dc 2004-04-21 devnull 		}else{
190 d946e4dc 2004-04-21 devnull 			strncpy(name, h.name, sizeof(h.name));
191 d946e4dc 2004-04-21 devnull 			for(i = sizeof(h.name)-1; i > 0 && name[i] == ' '; i--)
192 d946e4dc 2004-04-21 devnull 					;
193 d946e4dc 2004-04-21 devnull 			if(name[i] == '/')		/* system V bug */
194 d946e4dc 2004-04-21 devnull 				i--;
195 d946e4dc 2004-04-21 devnull 			name[i+1]=0;
196 d946e4dc 2004-04-21 devnull 		}
197 a29753a1 2004-04-21 devnull 		snprint(buf, sizeof buf, "%s(%s)", ar, name);
198 d946e4dc 2004-04-21 devnull 		sym = symlook(strdup(buf), S_TIME, (void *)t);
199 d946e4dc 2004-04-21 devnull 		sym->value = (void *)t;
200 d946e4dc 2004-04-21 devnull 	skip:
201 76193d7c 2003-09-30 devnull 		t = atol(h.size);
202 76193d7c 2003-09-30 devnull 		if(t&01) t++;
203 d946e4dc 2004-04-21 devnull 		t -= namelen;
204 76193d7c 2003-09-30 devnull 		LSEEK(fd, t, 1);
205 76193d7c 2003-09-30 devnull 	}
206 76193d7c 2003-09-30 devnull 	close(fd);
207 d946e4dc 2004-04-21 devnull 	free(strings);
208 76193d7c 2003-09-30 devnull }
209 76193d7c 2003-09-30 devnull 
210 76193d7c 2003-09-30 devnull static int
211 76193d7c 2003-09-30 devnull type(char *file)
212 76193d7c 2003-09-30 devnull {
213 76193d7c 2003-09-30 devnull 	int fd;
214 76193d7c 2003-09-30 devnull 	char buf[SARMAG];
215 76193d7c 2003-09-30 devnull 
216 76193d7c 2003-09-30 devnull 	fd = open(file, OREAD);
217 76193d7c 2003-09-30 devnull 	if(fd < 0){
218 76193d7c 2003-09-30 devnull 		if(symlook(file, S_BITCH, 0) == 0){
219 3842363a 2005-02-02 devnull 			if(strlen(file) < 2 || strcmp(file+strlen(file)-2, ".a") != 0)
220 3842363a 2005-02-02 devnull 				Bprint(&bout, "%s doesn't exist: assuming it will be an archive\n", file);
221 76193d7c 2003-09-30 devnull 			symlook(file, S_BITCH, (void *)file);
222 76193d7c 2003-09-30 devnull 		}
223 76193d7c 2003-09-30 devnull 		return 1;
224 76193d7c 2003-09-30 devnull 	}
225 76193d7c 2003-09-30 devnull 	if(read(fd, buf, SARMAG) != SARMAG){
226 76193d7c 2003-09-30 devnull 		close(fd);
227 76193d7c 2003-09-30 devnull 		return 0;
228 76193d7c 2003-09-30 devnull 	}
229 76193d7c 2003-09-30 devnull 	close(fd);
230 76193d7c 2003-09-30 devnull 	return !strncmp(ARMAG, buf, SARMAG);
231 76193d7c 2003-09-30 devnull }
232 76193d7c 2003-09-30 devnull 
233 76193d7c 2003-09-30 devnull static char*
234 76193d7c 2003-09-30 devnull split(char *name, char **member)
235 76193d7c 2003-09-30 devnull {
236 76193d7c 2003-09-30 devnull 	char *p, *q;
237 76193d7c 2003-09-30 devnull 
238 76193d7c 2003-09-30 devnull 	p = strdup(name);
239 76193d7c 2003-09-30 devnull 	q = utfrune(p, '(');
240 76193d7c 2003-09-30 devnull 	if(q){
241 76193d7c 2003-09-30 devnull 		*q++ = 0;
242 76193d7c 2003-09-30 devnull 		if(member)
243 76193d7c 2003-09-30 devnull 			*member = q;
244 76193d7c 2003-09-30 devnull 		q = utfrune(q, ')');
245 76193d7c 2003-09-30 devnull 		if (q)
246 76193d7c 2003-09-30 devnull 			*q = 0;
247 76193d7c 2003-09-30 devnull 		if(type(p))
248 76193d7c 2003-09-30 devnull 			return p;
249 76193d7c 2003-09-30 devnull 		free(p);
250 76193d7c 2003-09-30 devnull 		fprint(2, "mk: '%s' is not an archive\n", name);
251 76193d7c 2003-09-30 devnull 	}
252 76193d7c 2003-09-30 devnull 	return 0;
253 76193d7c 2003-09-30 devnull }