main.c revision 212029 
1/*
2 *
3 * FreeBSD install - a package for the installation and maintainance
4 * of non-core utilities.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 *    notice, this list of conditions and the following disclaimer in the
13 *    documentation and/or other materials provided with the distribution.
14 *
15 * Jordan K. Hubbard
16 * 18 July 1993
17 *
18 * This is the add module.
19 */
20
21#include <sys/cdefs.h>
22__FBSDID("$FreeBSD: head/usr.sbin/pkg_install/add/main.c 212029 2010-08-30 21:58:52Z nwhitehorn $");
23
24#include <sys/param.h>
25#include <sys/sysctl.h>
26#include <err.h>
27#include <getopt.h>
28
29#include <pkg.h>
30#include "add.h"
31
32char	*Prefix		= NULL;
33Boolean	PrefixRecursive	= FALSE;
34char	*Chroot		= NULL;
35Boolean	NoInstall	= FALSE;
36Boolean	NoRecord	= FALSE;
37Boolean Remote		= FALSE;
38Boolean KeepPackage	= FALSE;
39Boolean FailOnAlreadyInstalled	= TRUE;
40Boolean IgnoreDeps	= FALSE;
41
42char	*Mode		= NULL;
43char	*Owner		= NULL;
44char	*Group		= NULL;
45char	*PkgName	= NULL;
46char	*PkgAddCmd	= NULL;
47char	*Directory	= NULL;
48char	FirstPen[FILENAME_MAX];
49add_mode_t AddMode	= NORMAL;
50
51char	**pkgs;
52
53struct {
54	int lowver;	/* Lowest version number to match */
55	int hiver;	/* Highest version number to match */
56	const char *directory;	/* Directory it lives in */
57} releases[] = {
58	{ 410000, 410000, "/packages-4.1-release" },
59	{ 420000, 420000, "/packages-4.2-release" },
60	{ 430000, 430000, "/packages-4.3-release" },
61	{ 440000, 440000, "/packages-4.4-release" },
62	{ 450000, 450000, "/packages-4.5-release" },
63	{ 460000, 460001, "/packages-4.6-release" },
64	{ 460002, 460099, "/packages-4.6.2-release" },
65	{ 470000, 470099, "/packages-4.7-release" },
66	{ 480000, 480099, "/packages-4.8-release" },
67	{ 490000, 490099, "/packages-4.9-release" },
68	{ 491000, 491099, "/packages-4.10-release" },
69	{ 492000, 492099, "/packages-4.11-release" },
70	{ 500000, 500099, "/packages-5.0-release" },
71	{ 501000, 501099, "/packages-5.1-release" },
72	{ 502000, 502009, "/packages-5.2-release" },
73	{ 502010, 502099, "/packages-5.2.1-release" },
74	{ 503000, 503099, "/packages-5.3-release" },
75	{ 504000, 504099, "/packages-5.4-release" },
76	{ 505000, 505099, "/packages-5.5-release" },
77	{ 600000, 600099, "/packages-6.0-release" },
78	{ 601000, 601099, "/packages-6.1-release" },
79	{ 602000, 602099, "/packages-6.2-release" },
80	{ 603000, 603099, "/packages-6.3-release" },
81	{ 604000, 604099, "/packages-6.4-release" },
82	{ 700000, 700099, "/packages-7.0-release" },
83	{ 701000, 701099, "/packages-7.1-release" },
84	{ 702000, 702099, "/packages-7.2-release" },
85	{ 800000, 800499, "/packages-8.0-release" },
86	{ 300000, 399000, "/packages-3-stable" },
87	{ 400000, 499000, "/packages-4-stable" },
88	{ 502100, 502128, "/packages-5-current" },
89	{ 503100, 599000, "/packages-5-stable" },
90	{ 600100, 699000, "/packages-6-stable" },
91	{ 700100, 799000, "/packages-7-stable" },
92	{ 800500, 899000, "/packages-8-stable" },
93	{ 900000, 999000, "/packages-9-current" },
94	{ 0, 9999999, "/packages-current" },
95	{ 0, 0, NULL }
96};
97
98static char *getpackagesite(void);
99int getosreldate(void);
100
101static void usage(void);
102
103static char opts[] = "hviIRfFnrp:P:SMt:C:K";
104static struct option longopts[] = {
105	{ "chroot",	required_argument,	NULL,		'C' },
106	{ "dry-run",	no_argument,		NULL,		'n' },
107	{ "force",	no_argument,		NULL,		'f' },
108	{ "help",	no_argument,		NULL,		'h' },
109	{ "keep",	no_argument,		NULL,		'K' },
110	{ "master",	no_argument,		NULL,		'M' },
111	{ "no-deps",	no_argument,		NULL,		'i' },
112	{ "no-record",	no_argument,		NULL,		'R' },
113	{ "no-script",	no_argument,		NULL,		'I' },
114	{ "prefix",	required_argument,	NULL,		'p' },
115	{ "remote",	no_argument,		NULL,		'r' },
116	{ "template",	required_argument,	NULL,		't' },
117	{ "slave",	no_argument,		NULL,		'S' },
118	{ "verbose",	no_argument,		NULL,		'v' },
119	{ NULL,		0,			NULL,		0 }
120};
121
122int
123main(int argc, char **argv)
124{
125    int ch, error;
126    char **start;
127    char *cp, *packagesite = NULL, *remotepkg = NULL, *ptr;
128    static char temppackageroot[MAXPATHLEN];
129    static char pkgaddpath[MAXPATHLEN];
130
131    pkg_wrap(PKG_INSTALL_VERSION, argv);
132
133    if (*argv[0] != '/' && strchr(argv[0], '/') != NULL)
134	PkgAddCmd = realpath(argv[0], pkgaddpath);
135    else
136	PkgAddCmd = argv[0];
137
138    start = argv;
139    while ((ch = getopt_long(argc, argv, opts, longopts, NULL)) != -1) {
140	switch(ch) {
141	case 'v':
142	    Verbose++;
143	    break;
144
145	case 'p':
146	    Prefix = optarg;
147	    PrefixRecursive = FALSE;
148	    break;
149
150	case 'P':
151	    Prefix = optarg;
152	    PrefixRecursive = TRUE;
153	    break;
154
155	case 'I':
156	    NoInstall = TRUE;
157	    break;
158
159	case 'R':
160	    NoRecord = TRUE;
161	    break;
162
163	case 'f':
164	    Force = TRUE;
165	    break;
166
167	case 'F':
168	    FailOnAlreadyInstalled = FALSE;
169	    break;
170
171	case 'K':
172	    KeepPackage = TRUE;
173	    break;
174
175	case 'n':
176	    Fake = TRUE;
177	    break;
178
179	case 'r':
180	    Remote = TRUE;
181	    break;
182
183	case 't':
184	    if (strlcpy(FirstPen, optarg, sizeof(FirstPen)) >= sizeof(FirstPen))
185		errx(1, "-t Argument too long.");
186	    break;
187
188	case 'S':
189	    AddMode = SLAVE;
190	    break;
191
192	case 'M':
193	    AddMode = MASTER;
194	    break;
195
196	case 'C':
197	    Chroot = optarg;
198	    break;
199
200	case 'i':
201	    IgnoreDeps = TRUE;
202	    break;
203
204	case 'h':
205	default:
206	    usage();
207	    break;
208	}
209    }
210    argc -= optind;
211    argv += optind;
212
213    if (AddMode != SLAVE) {
214	pkgs = (char **)malloc((argc+1) * sizeof(char *));
215	for (ch = 0; ch <= argc; pkgs[ch++] = NULL) ;
216
217	/* Get all the remaining package names, if any */
218	for (ch = 0; *argv; ch++, argv++) {
219	    char temp[MAXPATHLEN];
220    	    if (Remote) {
221		if ((packagesite = getpackagesite()) == NULL)
222		    errx(1, "package name too long");
223		if (strlcpy(temppackageroot, packagesite,
224		    sizeof(temppackageroot)) >= sizeof(temppackageroot))
225		    errx(1, "package name too long");
226		if (strlcat(temppackageroot, *argv, sizeof(temppackageroot))
227		    >= sizeof(temppackageroot))
228		    errx(1, "package name too long");
229		remotepkg = temppackageroot;
230		if (!((ptr = strrchr(remotepkg, '.')) && ptr[1] == 't' &&
231			(ptr[2] == 'b' || ptr[2] == 'g') && ptr[3] == 'z' &&
232			!ptr[4]))
233		    if (strlcat(remotepkg, ".tbz",
234			sizeof(temppackageroot)) >= sizeof(temppackageroot))
235			errx(1, "package name too long");
236    	    }
237	    if (!strcmp(*argv, "-"))	/* stdin? */
238		pkgs[ch] = (char *)"-";
239	    else if (isURL(*argv)) {  	/* preserve URLs */
240		if (strlcpy(temp, *argv, sizeof(temp)) >= sizeof(temp))
241		    errx(1, "package name too long");
242		pkgs[ch] = strdup(temp);
243	    }
244	    else if ((Remote) && isURL(remotepkg)) {
245	    	if (strlcpy(temp, remotepkg, sizeof(temp)) >= sizeof(temp))
246		    errx(1, "package name too long");
247		pkgs[ch] = strdup(temp);
248	    } else {			/* expand all pathnames to fullnames */
249		if (fexists(*argv)) /* refers to a file directly */
250		    pkgs[ch] = strdup(realpath(*argv, temp));
251		else {		/* look for the file in the expected places */
252		    if (!(cp = fileFindByPath(NULL, *argv))) {
253			/* let pkg_do() fail later, so that error is reported */
254			if (strlcpy(temp, *argv, sizeof(temp)) >= sizeof(temp))
255			    errx(1, "package name too long");
256			pkgs[ch] = strdup(temp);
257		    } else {
258			if (strlcpy(temp, cp, sizeof(temp)) >= sizeof(temp))
259			    errx(1, "package name too long");
260			pkgs[ch] = strdup(temp);
261		    }
262		}
263	    }
264	    if (packagesite != NULL)
265		packagesite[0] = '\0';
266	}
267    }
268    /* If no packages, yelp */
269    if (!ch) {
270	warnx("missing package name(s)");
271	usage();
272    }
273    else if (ch > 1 && AddMode == MASTER) {
274	warnx("only one package name may be specified with master mode");
275	usage();
276    }
277    /* Perform chroot if requested */
278    if (Chroot != NULL) {
279	if (chroot(Chroot))
280	    errx(1, "chroot to %s failed", Chroot);
281    }
282    /* Make sure the sub-execs we invoke get found */
283    setenv("PATH",
284	   "/sbin:/bin:/usr/sbin:/usr/bin:/usr/local/sbin:/usr/local/bin:/usr/X11R6/bin",
285	   1);
286
287    /* Set a reasonable umask */
288    umask(022);
289
290    if ((error = pkg_perform(pkgs)) != 0) {
291	if (Verbose)
292	    warnx("%d package addition(s) failed", error);
293	return error;
294    }
295    else
296	return 0;
297}
298
299static char *
300getpackagesite(void)
301{
302    int reldate, i;
303    static char sitepath[MAXPATHLEN];
304    int archmib[] = { CTL_HW, HW_MACHINE_ARCH };
305    char arch[64];
306    size_t archlen = sizeof(arch);
307
308    if (getenv("PACKAGESITE")) {
309	if (strlcpy(sitepath, getenv("PACKAGESITE"), sizeof(sitepath))
310	    >= sizeof(sitepath))
311	    return NULL;
312	return sitepath;
313    }
314
315    if (getenv("PACKAGEROOT")) {
316	if (strlcpy(sitepath, getenv("PACKAGEROOT"), sizeof(sitepath))
317	    >= sizeof(sitepath))
318	    return NULL;
319    } else {
320	if (strlcat(sitepath, "ftp://ftp.freebsd.org", sizeof(sitepath))
321	    >= sizeof(sitepath))
322	    return NULL;
323    }
324
325    if (strlcat(sitepath, "/pub/FreeBSD/ports/", sizeof(sitepath))
326	>= sizeof(sitepath))
327	return NULL;
328
329    if (sysctl(archmib, 2, arch, &archlen, NULL, 0) == -1)
330	return NULL;
331    arch[archlen-1] = 0;
332    if (strlcat(sitepath, arch, sizeof(sitepath)) >= sizeof(sitepath))
333	return NULL;
334
335    reldate = getosreldate();
336    for(i = 0; releases[i].directory != NULL; i++) {
337	if (reldate >= releases[i].lowver && reldate <= releases[i].hiver) {
338	    if (strlcat(sitepath, releases[i].directory, sizeof(sitepath))
339		>= sizeof(sitepath))
340		return NULL;
341	    break;
342	}
343    }
344
345    if (strlcat(sitepath, "/Latest/", sizeof(sitepath)) >= sizeof(sitepath))
346	return NULL;
347
348    return sitepath;
349
350}
351
352static void
353usage(void)
354{
355    fprintf(stderr, "%s\n%s\n",
356	"usage: pkg_add [-viInfFrRMSK] [-t template] [-p prefix] [-P prefix] [-C chrootdir]",
357	"               pkg-name [pkg-name ...]");
358    exit(1);
359}
360