razor: main.c@2555ce3e2567 (annotated)

krh@212	1	/*
krh@212	2	* Copyright (C) 2008 Kristian Høgsberg <krh@redhat.com>
krh@212	3	* Copyright (C) 2008 Red Hat, Inc
krh@212	4	*
krh@212	5	* This program is free software; you can redistribute it and/or modify
krh@212	6	* it under the terms of the GNU General Public License as published by
krh@212	7	* the Free Software Foundation; either version 2 of the License, or
krh@212	8	* (at your option) any later version.
krh@212	9	*
krh@212	10	* This program is distributed in the hope that it will be useful,
krh@212	11	* but WITHOUT ANY WARRANTY; without even the implied warranty of
krh@212	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
krh@212	13	* GNU General Public License for more details.
krh@212	14	*
krh@212	15	* You should have received a copy of the GNU General Public License along
krh@212	16	* with this program; if not, write to the Free Software Foundation, Inc.,
krh@212	17	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
krh@212	18	*/
krh@212	19
krh@43	20	#include <stdlib.h>
krh@43	21	#include <stddef.h>
krh@43	22	#include <stdio.h>
krh@186	23	#include <stdint.h>
krh@43	24	#include <string.h>
krh@72	25	#include <sys/stat.h>
krh@43	26	#include <unistd.h>
krh@197	27	#include <fcntl.h>
krh@75	28	#include <dirent.h>
krh@71	29	#include <curl/curl.h>
krh@94	30	#include <fnmatch.h>
krh@206	31	#include <errno.h>
krh@43	32	#include "razor.h"
krh@151	33	#include "razor-internal.h"
krh@43	34
krh@152	35	static const char system_repo_filename[] = "system.repo";
krh@171	36	static const char next_repo_filename[] = "system-next.repo";
krh@152	37	static const char rawhide_repo_filename[] = "rawhide.repo";
krh@152	38	static const char updated_repo_filename[] = "system-updated.repo";
krh@152	39	static const char razor_root_path[] = "/var/lib/razor";
krh@152	40	static const char root[] = "install";
krh@152	41	static const char *repo_filename = system_repo_filename;
krh@231	42	static const char *yum_url;
krh@43	43
krh@43	44	static int
krh@43	45	command_list(int argc, const char *argv[])
krh@43	46	{
krh@43	47	struct razor_set *set;
krh@92	48	struct razor_package_iterator *pi;
krh@92	49	struct razor_package *package;
krh@192	50	const char pattern, name, version, arch;
krh@187	51	int only_names = 0, i = 0;
krh@43	52
krh@193	53	if (i < argc && strcmp(argv[i], "--only-names") == 0) {
krh@187	54	only_names = 1;
krh@187	55	i++;
krh@187	56	}
krh@187	57
krh@187	58	pattern = argv[i];
krh@43	59	set = razor_set_open(repo_filename);
krh@92	60	pi = razor_package_iterator_create(set);
krh@192	61	while (razor_package_iterator_next(pi, &package,
krh@192	62	&name, &version, &arch)) {
krh@92	63	if (pattern && fnmatch(pattern, name, 0) != 0)
krh@92	64	continue;
krh@92	65
krh@187	66	if (only_names)
krh@187	67	printf("%s\n", name);
krh@187	68	else
krh@192	69	printf("%s-%s.%s\n", name, version, arch);
krh@92	70	}
krh@92	71	razor_package_iterator_destroy(pi);
krh@92	72	razor_set_destroy(set);
krh@92	73
krh@92	74	return 0;
krh@92	75	}
krh@92	76
krh@92	77	static int
krh@92	78	list_properties(const char *package_name,
krh@92	79	enum razor_property_type required_type)
krh@92	80	{
krh@230	81	static const char *relation_string[] = { "<", "<=", "=", ">=", ">" };
krh@92	82	struct razor_set *set;
krh@92	83	struct razor_property *property;
krh@92	84	struct razor_package *package;
krh@92	85	struct razor_property_iterator *pi;
krh@92	86	const char name, version;
krh@92	87	enum razor_property_type type;
danw@109	88	enum razor_version_relation relation;
krh@92	89
krh@92	90	set = razor_set_open(repo_filename);
krh@92	91	if (package_name)
krh@92	92	package = razor_set_get_package(set, package_name);
krh@92	93	else
krh@92	94	package = NULL;
krh@92	95
krh@92	96	pi = razor_property_iterator_create(set, package);
krh@92	97	while (razor_property_iterator_next(pi, &property,
danw@109	98	&name, &relation, &version,
danw@109	99	&type)) {
krh@92	100	if (type != required_type)
krh@92	101	continue;
krh@92	102	if (version[0] == '\0')
krh@92	103	printf("%s\n", name);
krh@92	104	else
danw@137	105	printf("%s %s %s\n", name,
krh@230	106	relation_string[relation], version);
krh@92	107	}
krh@92	108	razor_property_iterator_destroy(pi);
krh@92	109
krh@43	110	razor_set_destroy(set);
krh@43	111
krh@43	112	return 0;
krh@43	113	}
krh@43	114
krh@43	115	static int
krh@43	116	command_list_requires(int argc, const char *argv[])
krh@43	117	{
krh@92	118	return list_properties(argv[0], RAZOR_PROPERTY_REQUIRES);
krh@43	119	}
krh@43	120
krh@43	121	static int
krh@43	122	command_list_provides(int argc, const char *argv[])
krh@43	123	{
krh@92	124	return list_properties(argv[0], RAZOR_PROPERTY_PROVIDES);
krh@43	125	}
krh@43	126
krh@43	127	static int
krh@67	128	command_list_obsoletes(int argc, const char *argv[])
krh@67	129	{
krh@92	130	return list_properties(argv[0], RAZOR_PROPERTY_OBSOLETES);
krh@67	131	}
krh@67	132
krh@67	133	static int
krh@67	134	command_list_conflicts(int argc, const char *argv[])
krh@67	135	{
krh@92	136	return list_properties(argv[0], RAZOR_PROPERTY_CONFLICTS);
krh@67	137	}
krh@67	138
krh@67	139	static int
krh@48	140	command_list_files(int argc, const char *argv[])
krh@48	141	{
krh@48	142	struct razor_set *set;
krh@48	143
krh@48	144	set = razor_set_open(repo_filename);
krh@48	145	if (set == NULL)
krh@48	146	return 1;
krh@49	147	razor_set_list_files(set, argv[0]);
krh@48	148	razor_set_destroy(set);
krh@48	149
krh@48	150	return 0;
krh@48	151	}
krh@48	152
krh@48	153	static int
krh@52	154	command_list_file_packages(int argc, const char *argv[])
krh@52	155	{
krh@52	156	struct razor_set *set;
krh@102	157	struct razor_package_iterator *pi;
krh@102	158	struct razor_package *package;
krh@192	159	const char name, version, *arch;
krh@52	160
krh@52	161	set = razor_set_open(repo_filename);
krh@52	162	if (set == NULL)
krh@52	163	return 1;
krh@102	164
krh@102	165	pi = razor_package_iterator_create_for_file(set, argv[0]);
krh@192	166	while (razor_package_iterator_next(pi, &package,
krh@192	167	&name, &version, &arch))
krh@102	168	printf("%s-%s\n", name, version);
krh@102	169	razor_package_iterator_destroy(pi);
krh@102	170
krh@52	171	razor_set_destroy(set);
krh@52	172
krh@52	173	return 0;
krh@52	174	}
krh@52	175
krh@56	176	static int
krh@56	177	command_list_package_files(int argc, const char *argv[])
krh@56	178	{
krh@56	179	struct razor_set *set;
krh@56	180
krh@56	181	set = razor_set_open(repo_filename);
krh@56	182	if (set == NULL)
krh@56	183	return 1;
krh@56	184	razor_set_list_package_files(set, argv[0]);
krh@56	185	razor_set_destroy(set);
krh@56	186
krh@56	187	return 0;
krh@56	188	}
krh@52	189
krh@101	190	static void
krh@101	191	list_packages_for_property(struct razor_set *set,
krh@101	192	struct razor_property *property)
krh@101	193	{
krh@101	194	struct razor_package_iterator *pi;
krh@101	195	struct razor_package *package;
krh@192	196	const char name, version, *arch;
krh@101	197
krh@101	198	pi = razor_package_iterator_create_for_property(set, property);
krh@192	199	while (razor_package_iterator_next(pi, &package,
krh@192	200	&name, &version, &arch))
krh@192	201	printf("%s-%s.%s\n", name, version, arch);
krh@101	202	razor_package_iterator_destroy(pi);
krh@101	203	}
krh@101	204
krh@52	205	static int
krh@100	206	list_property_packages(const char *ref_name,
krh@100	207	const char *ref_version,
krh@100	208	enum razor_property_type ref_type)
krh@43	209	{
krh@43	210	struct razor_set *set;
krh@100	211	struct razor_property *property;
krh@100	212	struct razor_property_iterator *pi;
krh@100	213	const char name, version;
krh@100	214	enum razor_property_type type;
danw@109	215	enum razor_version_relation relation;
krh@100	216
krh@100	217	if (ref_name == NULL)
krh@100	218	return 0;
krh@43	219
krh@43	220	set = razor_set_open(repo_filename);
krh@100	221	if (set == NULL)
krh@100	222	return 1;
krh@100	223
krh@100	224	pi = razor_property_iterator_create(set, NULL);
krh@100	225	while (razor_property_iterator_next(pi, &property,
danw@109	226	&name, &relation, &version,
danw@109	227	&type)) {
krh@100	228	if (strcmp(ref_name, name) != 0)
krh@100	229	continue;
danw@109	230	if (ref_version && relation == RAZOR_VERSION_EQUAL &&
danw@109	231	strcmp(ref_version, version) != 0)
krh@100	232	continue;
krh@100	233	if (ref_type != type)
krh@100	234	continue;
krh@100	235
krh@101	236	list_packages_for_property(set, property);
krh@100	237	}
krh@100	238	razor_property_iterator_destroy(pi);
krh@43	239
krh@43	240	return 0;
krh@43	241	}
krh@43	242
krh@43	243	static int
krh@100	244	command_what_requires(int argc, const char *argv[])
krh@100	245	{
krh@100	246	return list_property_packages(argv[0], argv[1],
krh@100	247	RAZOR_PROPERTY_REQUIRES);
krh@100	248	}
krh@100	249
krh@100	250	static int
krh@43	251	command_what_provides(int argc, const char *argv[])
krh@43	252	{
krh@100	253	return list_property_packages(argv[0], argv[1],
krh@100	254	RAZOR_PROPERTY_PROVIDES);
krh@43	255	}
krh@43	256
krh@73	257	static int
krh@73	258	show_progress(void *clientp,
krh@73	259	double dltotal, double dlnow, double ultotal, double ulnow)
krh@73	260	{
krh@73	261	const char *file = clientp;
krh@73	262
krh@73	263	if (!dlnow < dltotal)
krh@73	264	fprintf(stderr, "\rdownloading %s, %dkB/%dkB",
krh@73	265	file, (int) dlnow / 1024, (int) dltotal / 1024);
krh@73	266
krh@73	267	return 0;
krh@73	268	}
krh@73	269
krh@73	270	static int
krh@194	271	download_if_missing(const char url, const char file)
krh@73	272	{
krh@194	273	CURL *curl;
krh@73	274	struct stat buf;
krh@73	275	char error[256];
krh@73	276	FILE *fp;
krh@73	277	CURLcode res;
krh@204	278	long response;
krh@73	279
krh@194	280	curl = curl_easy_init();
krh@194	281	if (curl == NULL)
krh@194	282	return 1;
krh@194	283
krh@73	284	curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, error);
krh@73	285	curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0);
krh@73	286	curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, show_progress);
krh@73	287	curl_easy_setopt(curl, CURLOPT_PROGRESSDATA, file);
krh@73	288
krh@73	289	if (stat(file, &buf) < 0) {
krh@73	290	fp = fopen(file, "w");
krh@207	291	if (fp == NULL) {
krh@207	292	fprintf(stderr,
krh@207	293	"failed to open %s for writing\n", file);
krh@207	294	return -1;
krh@207	295	}
krh@73	296	curl_easy_setopt(curl, CURLOPT_WRITEDATA, fp);
krh@187	297	curl_easy_setopt(curl, CURLOPT_URL, url);
krh@73	298	res = curl_easy_perform(curl);
krh@73	299	fclose(fp);
krh@73	300	if (res != CURLE_OK) {
krh@73	301	fprintf(stderr, "curl error: %s\n", error);
krh@73	302	unlink(file);
krh@73	303	return -1;
krh@73	304	}
krh@204	305	res = curl_easy_getinfo(curl,
krh@204	306	CURLINFO_RESPONSE_CODE, &response);
krh@204	307	if (res != CURLE_OK) {
krh@204	308	fprintf(stderr, "curl error: %s\n", error);
krh@204	309	unlink(file);
krh@204	310	return -1;
krh@204	311	}
krh@204	312	if (response != 200) {
krh@204	313	fprintf(stderr, " - failed %ld\n", response);
krh@204	314	unlink(file);
krh@204	315	return -1;
krh@204	316	}
krh@188	317	fprintf(stderr, "\n");
krh@73	318	}
krh@73	319
krh@194	320	curl_easy_cleanup(curl);
krh@194	321
krh@73	322	return 0;
krh@73	323	}
krh@73	324
krh@231	325	#define YUM_URL "http://download.fedora.redhat.com" \
krh@187	326	"/pub/fedora/linux/development/i386/os"
krh@71	327
krh@43	328	static int
krh@43	329	command_import_yum(int argc, const char *argv[])
krh@43	330	{
krh@43	331	struct razor_set *set;
krh@231	332	char buffer[512];
krh@71	333
krh@231	334	printf("downloading from %s.\n", yum_url);
krh@231	335	snprintf(buffer, sizeof buffer,
krh@231	336	"%s/repodata/primary.xml.gz", yum_url);
krh@231	337	if (download_if_missing(buffer, "primary.xml.gz") < 0)
krh@73	338	return -1;
krh@231	339	snprintf(buffer, sizeof buffer,
krh@231	340	"%s/repodata/filelists.xml.gz", yum_url);
krh@231	341	if (download_if_missing(buffer, "filelists.xml.gz") < 0)
krh@73	342	return -1;
krh@43	343
krh@70	344	set = razor_set_create_from_yum();
krh@43	345	if (set == NULL)
krh@43	346	return 1;
krh@43	347	razor_set_write(set, rawhide_repo_filename);
krh@43	348	razor_set_destroy(set);
krh@43	349	printf("wrote %s\n", rawhide_repo_filename);
krh@43	350
krh@43	351	return 0;
krh@43	352	}
krh@43	353
krh@43	354	static int
krh@43	355	command_import_rpmdb(int argc, const char *argv[])
krh@43	356	{
krh@43	357	struct razor_set *set;
krh@43	358
krh@43	359	set = razor_set_create_from_rpmdb();
krh@43	360	if (set == NULL)
krh@43	361	return 1;
krh@43	362	razor_set_write(set, repo_filename);
krh@43	363	razor_set_destroy(set);
krh@43	364	printf("wrote %s\n", repo_filename);
krh@43	365
krh@43	366	return 0;
krh@43	367	}
krh@43	368
krh@43	369	static int
krh@208	370	mark_packages_for_update(struct razor_transaction *trans,
krh@208	371	struct razor_set set, const char pattern)
krh@208	372	{
krh@208	373	struct razor_package_iterator *pi;
krh@208	374	struct razor_package *package;
krh@208	375	const char name, version, *arch;
krh@208	376	int matches = 0;
krh@208	377
krh@208	378	pi = razor_package_iterator_create(set);
krh@208	379	while (razor_package_iterator_next(pi, &package,
krh@208	380	&name, &version, &arch)) {
krh@208	381	if (pattern && fnmatch(pattern, name, 0) == 0) {
krh@230	382	razor_transaction_update_package(trans, package);
krh@208	383	matches++;
krh@208	384	}
krh@208	385	}
krh@208	386	razor_package_iterator_destroy(pi);
krh@208	387
krh@208	388	return matches;
krh@208	389	}
krh@208	390
krh@208	391	static int
krh@208	392	mark_packages_for_removal(struct razor_transaction *trans,
krh@208	393	struct razor_set set, const char pattern)
krh@208	394	{
krh@208	395	struct razor_package_iterator *pi;
krh@208	396	struct razor_package *package;
krh@208	397	const char name, version, *arch;
krh@208	398	int matches = 0;
krh@208	399
krh@208	400	pi = razor_package_iterator_create(set);
krh@208	401	while (razor_package_iterator_next(pi, &package,
krh@208	402	&name, &version, &arch)) {
krh@208	403	if (pattern && fnmatch(pattern, name, 0) == 0) {
krh@208	404	razor_transaction_remove_package(trans, package);
krh@208	405	matches++;
krh@208	406	}
krh@208	407	}
krh@208	408	razor_package_iterator_destroy(pi);
krh@208	409
krh@208	410	return matches;
krh@208	411	}
krh@208	412
krh@208	413	static int
krh@43	414	command_update(int argc, const char *argv[])
krh@43	415	{
krh@43	416	struct razor_set set, upstream;
danw@137	417	struct razor_transaction *trans;
krh@208	418	int i, errors;
krh@43	419
krh@43	420	set = razor_set_open(repo_filename);
krh@43	421	upstream = razor_set_open(rawhide_repo_filename);
krh@43	422	if (set == NULL \|\| upstream == NULL)
krh@43	423	return 1;
krh@208	424
krh@208	425	trans = razor_transaction_create(set, upstream);
krh@208	426	if (argc == 0)
krh@208	427	razor_transaction_update_all(trans);
krh@208	428	for (i = 0; i < argc; i++) {
krh@230	429	if (mark_packages_for_update(trans, set, argv[i]) == 0) {
krh@208	430	fprintf(stderr, "no match for %s\n", argv[i]);
krh@208	431	return 1;
krh@208	432	}
krh@208	433	}
krh@208	434
krh@210	435	errors = razor_transaction_resolve(trans);
krh@190	436	if (errors)
danw@137	437	return 1;
danw@137	438
krh@196	439	set = razor_transaction_finish(trans);
krh@44	440	razor_set_write(set, updated_repo_filename);
krh@43	441	razor_set_destroy(set);
krh@43	442	razor_set_destroy(upstream);
krh@43	443	printf("wrote system-updated.repo\n");
krh@43	444
krh@43	445	return 0;
krh@43	446	}
krh@43	447
danw@129	448	static int
danw@129	449	command_remove(int argc, const char *argv[])
danw@129	450	{
danw@129	451	struct razor_set *set;
danw@137	452	struct razor_transaction *trans;
krh@208	453	int i, errors;
danw@129	454
danw@129	455	set = razor_set_open(repo_filename);
danw@129	456	if (set == NULL)
danw@129	457	return 1;
krh@208	458
krh@208	459	trans = razor_transaction_create(set, NULL);
krh@208	460	for (i = 0; i < argc; i++) {
krh@208	461	if (mark_packages_for_removal(trans, set, argv[i]) == 0) {
krh@208	462	fprintf(stderr, "no match for %s\n", argv[i]);
krh@208	463	return 1;
krh@208	464	}
krh@208	465	}
krh@208	466
krh@210	467	errors = razor_transaction_resolve(trans);
krh@190	468	if (errors)
danw@137	469	return 1;
danw@137	470
krh@196	471	set = razor_transaction_finish(trans);
danw@129	472	razor_set_write(set, updated_repo_filename);
danw@129	473	razor_set_destroy(set);
danw@129	474	printf("wrote system-updated.repo\n");
danw@129	475
danw@129	476	return 0;
danw@129	477	}
danw@129	478
krh@44	479	static void
krh@44	480	print_diff(const char *name,
krh@192	481	const char old_version, const char new_version, const char *arch,
krh@192	482	void *data)
krh@44	483	{
krh@44	484	if (old_version)
krh@44	485	printf("removing %s %s\n", name, old_version);
krh@44	486	else
krh@44	487	printf("install %s %s\n", name, new_version);
krh@44	488	}
krh@44	489
krh@44	490	static int
krh@44	491	command_diff(int argc, const char *argv[])
krh@44	492	{
krh@44	493	struct razor_set set, updated;
krh@44	494
krh@44	495	set = razor_set_open(repo_filename);
krh@44	496	updated = razor_set_open(updated_repo_filename);
krh@44	497	if (set == NULL \|\| updated == NULL)
krh@44	498	return 1;
krh@44	499
krh@44	500	razor_set_diff(set, updated, print_diff, NULL);
krh@44	501
krh@44	502	razor_set_destroy(set);
krh@44	503	razor_set_destroy(updated);
krh@44	504
krh@44	505	return 0;
krh@44	506	}
krh@44	507
krh@74	508	static int
krh@75	509	command_import_rpms(int argc, const char *argv[])
krh@74	510	{
krh@75	511	DIR *dir;
krh@75	512	struct dirent *de;
krh@75	513	struct razor_importer *importer;
krh@75	514	struct razor_set *set;
krh@77	515	struct razor_rpm *rpm;
krh@75	516	int len;
krh@75	517	char filename[256];
krh@75	518	const char *dirname = argv[0];
krh@75	519
krh@75	520	if (dirname == NULL) {
krh@75	521	fprintf(stderr, "usage: razor import-rpms DIR\n");
krh@75	522	return -1;
krh@75	523	}
krh@75	524
krh@75	525	dir = opendir(dirname);
krh@75	526	if (dir == NULL) {
krh@75	527	fprintf(stderr, "couldn't read dir %s\n", dirname);
krh@75	528	return -1;
krh@75	529	}
krh@75	530
krh@75	531	importer = razor_importer_new();
krh@75	532
krh@75	533	while (de = readdir(dir), de != NULL) {
krh@75	534	len = strlen(de->d_name);
krh@75	535	if (len < 5 \|\| strcmp(de->d_name + len - 4, ".rpm") != 0)
krh@75	536	continue;
krh@75	537	snprintf(filename, sizeof filename,
krh@75	538	"%s/%s", dirname, de->d_name);
krh@77	539	rpm = razor_rpm_open(filename);
krh@77	540	if (rpm == NULL) {
krh@77	541	fprintf(stderr,
krh@77	542	"failed to open rpm \"%s\"\n", filename);
krh@77	543	continue;
krh@77	544	}
krh@77	545	if (razor_importer_add_rpm(importer, rpm)) {
krh@75	546	fprintf(stderr, "couldn't import %s\n", filename);
krh@75	547	break;
krh@75	548	}
krh@77	549	razor_rpm_close(rpm);
krh@75	550	}
krh@75	551
krh@75	552	if (de != NULL) {
krh@75	553	razor_importer_destroy(importer);
krh@75	554	return -1;
krh@75	555	}
krh@75	556
krh@75	557	set = razor_importer_finish(importer);
krh@75	558
krh@75	559	razor_set_write(set, repo_filename);
krh@75	560	razor_set_destroy(set);
krh@75	561	printf("wrote %s\n", repo_filename);
krh@74	562
krh@74	563	return 0;
krh@74	564	}
krh@74	565
krh@232	566	/* The image data struct encapsulates filesystem conventions and the
krh@232	567	* locking protocol. FIXME: Should this be razor_root?*/
krh@232	568	struct razor_root {
krh@232	569	struct razor_set *system;
krh@232	570	int fd;
krh@232	571	char path[PATH_MAX];
krh@232	572	char new_path[PATH_MAX];
krh@232	573	};
krh@232	574
krh@232	575	#define RAZOR_ROOT_OPEN_WRITE 0x01
krh@232	576
krh@232	577	int
krh@232	578	razor_root_create(const char *root);
krh@232	579	struct razor_root *
krh@232	580	razor_root_open(const char *root, int flags);
krh@232	581	struct razor_transaction *
krh@232	582	razor_root_create_transaction(struct razor_root *image,
krh@232	583	struct razor_set *upstream);
krh@232	584	int
krh@232	585	razor_root_close(struct razor_root *image);
krh@232	586	void
krh@232	587	razor_root_update(struct razor_root image, struct razor_set next);
krh@232	588	int
krh@232	589	razor_root_commit(struct razor_root *image);
krh@232	590
krh@232	591	int
krh@232	592	razor_root_create(const char *root)
krh@232	593	{
krh@232	594	struct stat buf;
krh@232	595	struct razor_set *set;
krh@232	596	char path[PATH_MAX];
krh@232	597
krh@232	598	if (stat(root, &buf) < 0) {
krh@232	599	if (mkdir(root, 0777) < 0) {
krh@232	600	fprintf(stderr,
krh@232	601	"could not create install root \"%s\"\n",
krh@232	602	root);
krh@232	603	return -1;
krh@232	604	}
krh@232	605	fprintf(stderr, "created install root \"%s\"\n", root);
krh@232	606	} else if (!S_ISDIR(buf.st_mode)) {
krh@232	607	fprintf(stderr,
krh@232	608	"install root \"%s\" exists, but is not a directory\n",
krh@232	609	root);
krh@232	610	return -1;
krh@232	611	}
krh@232	612
krh@232	613	snprintf(path, sizeof path, "%s/%s",
krh@232	614	razor_root_path, system_repo_filename);
krh@232	615	if (razor_create_dir(root, path) < 0) {
krh@232	616	fprintf(stderr, "could not create %s%s\n",
krh@232	617	root, razor_root_path);
krh@232	618	return -1;
krh@232	619	}
krh@232	620
krh@232	621	set = razor_set_create();
krh@232	622	snprintf(path, sizeof path, "%s%s/%s",
krh@232	623	root, razor_root_path, system_repo_filename);
krh@232	624	if (stat(root, &buf) == 0) {
krh@232	625	fprintf(stderr,
krh@232	626	"a razor install root is already initialized\n");
krh@232	627	return -1;
krh@232	628	}
krh@232	629	if (razor_set_write(set, path) < 0) {
krh@232	630	fprintf(stderr, "could not write initial package set\n");
krh@232	631	return -1;
krh@232	632	}
krh@232	633	razor_set_destroy(set);
krh@232	634
krh@232	635	return 0;
krh@232	636	}
krh@232	637
krh@232	638	struct razor_root *
krh@232	639	razor_root_open(const char *root, int flags)
krh@232	640	{
krh@232	641	struct razor_root *image;
krh@232	642
krh@232	643	image = malloc(sizeof *image);
krh@232	644	if (image == NULL)
krh@232	645	return NULL;
krh@232	646
krh@232	647	/* Create the new next repo file up front to ensure exclusive
krh@232	648	* access. */
krh@232	649	snprintf(image->new_path, sizeof image->new_path,
krh@232	650	"%s%s/%s", root, root, next_repo_filename);
krh@232	651	image->fd = open(image->new_path,
krh@232	652	O_CREAT \| O_WRONLY \| O_TRUNC \| O_EXCL, 0666);
krh@232	653	if (image->fd < 0) {
krh@232	654	fprintf(stderr, "failed to get lock file, "
krh@232	655	"maybe previous operation crashed?\n");
krh@232	656
krh@232	657	/* FIXME: Use fcntl advisory locking on the system
krh@232	658	* package set file to figure out whether previous
krh@232	659	* operation crashed or is still in progress. */
krh@232	660
krh@232	661	free(image);
krh@232	662	return NULL;
krh@232	663	}
krh@232	664
krh@232	665	snprintf(image->path, sizeof image->path,
krh@232	666	"%s%s/%s", root, razor_root_path, system_repo_filename);
krh@232	667	image->system = razor_set_open(image->path);
krh@232	668	if (image->system == NULL) {
krh@232	669	unlink(image->new_path);
krh@232	670	close(image->fd);
krh@232	671	free(image);
krh@232	672	return NULL;
krh@232	673	}
krh@232	674
krh@232	675	return image;
krh@232	676	}
krh@232	677
krh@232	678	struct razor_transaction *
krh@232	679	razor_root_create_transaction(struct razor_root *image,
krh@232	680	struct razor_set *upstream)
krh@232	681	{
krh@232	682	/* FIXME: This should take a number of upstream repos. */
krh@232	683	return razor_transaction_create(image->system, upstream);
krh@232	684	}
krh@232	685
krh@232	686	int
krh@232	687	razor_root_close(struct razor_root *image)
krh@232	688	{
krh@232	689	unlink(image->new_path);
krh@232	690	close(image->fd);
krh@232	691	free(image);
krh@232	692
krh@232	693	return 0;
krh@232	694	}
krh@232	695
krh@232	696	void
krh@232	697	razor_root_update(struct razor_root image, struct razor_set next)
krh@232	698	{
krh@232	699	razor_set_write_to_fd(next, image->fd);
krh@232	700
krh@232	701	/* Sync the new repo file so the new package set is on disk
krh@232	702	* before we start upgrading. */
krh@232	703	fsync(image->fd);
krh@232	704	printf("wrote %s\n", image->new_path);
krh@232	705	}
krh@232	706
krh@232	707	int
krh@232	708	razor_root_commit(struct razor_root *image)
krh@232	709	{
krh@232	710	/* Make it so. */
krh@232	711	rename(image->new_path, image->path);
krh@232	712	printf("renamed %s to %s\n", image->new_path, image->path);
krh@232	713	close(image->fd);
krh@232	714	free(image);
krh@232	715
krh@232	716	return 0;
krh@232	717	}
krh@232	718
krh@194	719	static void
krh@194	720	download_package(const char *name,
krh@194	721	const char *old_version,
krh@194	722	const char *new_version,
krh@194	723	const char *arch,
krh@194	724	void *data)
krh@152	725	{
krh@194	726	char file[PATH_MAX], url[256];
krh@201	727	const char *v;
krh@204	728	int *errors = data;
krh@194	729
krh@194	730	if (old_version)
krh@194	731	return;
krh@194	732
krh@201	733	/* Skip epoch */
krh@201	734	v = strchr(new_version, ':');
krh@201	735	if (v != NULL)
krh@201	736	v = v + 1;
krh@201	737	else
krh@201	738	v = new_version;
krh@201	739
krh@194	740	snprintf(url, sizeof url,
krh@231	741	"%s/Packages/%s-%s.%s.rpm", yum_url, name, v, arch);
krh@194	742	snprintf(file, sizeof file,
krh@201	743	"rpms/%s-%s.%s.rpm", name, v, arch);
krh@194	744	if (download_if_missing(url, file) < 0)
krh@204	745	(*errors)++;
krh@194	746	}
krh@194	747
krh@194	748	static void
krh@194	749	install_package(const char *name,
krh@194	750	const char *old_version,
krh@194	751	const char *new_version,
krh@194	752	const char *arch,
krh@194	753	void *data)
krh@194	754	{
krh@201	755	const char v, root = data;
krh@194	756	char file[PATH_MAX];
krh@152	757	struct razor_rpm *rpm;
krh@152	758
krh@194	759	if (old_version) {
krh@194	760	printf("removing %s %s not handled\n", name, old_version);
krh@194	761	return;
krh@152	762	}
krh@152	763
krh@201	764	/* Skip epoch */
krh@201	765	v = strchr(new_version, ':');
krh@201	766	if (v != NULL)
krh@201	767	v = v + 1;
krh@201	768	else
krh@201	769	v = new_version;
krh@201	770
krh@201	771	printf("install %s %s\n", name, v);
krh@201	772	snprintf(file, sizeof file, "rpms/%s-%s.%s.rpm", name, v, arch);
krh@152	773
krh@194	774	rpm = razor_rpm_open(file);
krh@194	775	if (rpm == NULL) {
krh@194	776	fprintf(stderr, "failed to open rpm %s\n", file);
krh@194	777	return;
krh@194	778	}
krh@194	779	if (razor_rpm_install(rpm, root) < 0) {
krh@194	780	fprintf(stderr,
krh@194	781	"failed to install rpm %s\n", file);
krh@194	782	return;
krh@194	783	}
krh@194	784	razor_rpm_close(rpm);
krh@152	785	}
krh@151	786
krh@77	787	static int
krh@77	788	command_install(int argc, const char *argv[])
krh@77	789	{
krh@232	790	struct razor_root *root;
krh@232	791	struct razor_set upstream, next;
krh@152	792	struct razor_transaction *trans;
krh@232	793	int i = 0, errors, dependencies = 1;
krh@210	794
krh@210	795	if (i < argc && strcmp(argv[i], "--no-dependencies") == 0) {
krh@210	796	dependencies = 0;
krh@210	797	i++;
krh@210	798	}
krh@197	799
krh@232	800	root = razor_root_open(razor_root_path, RAZOR_ROOT_OPEN_WRITE);
krh@232	801	upstream = razor_set_open(rawhide_repo_filename);
krh@232	802	trans = razor_root_create_transaction(root, upstream);
krh@197	803
krh@210	804	for (; i < argc; i++) {
krh@208	805	if (mark_packages_for_update(trans, upstream, argv[i]) == 0) {
krh@208	806	fprintf(stderr, "no package matched %s\n", argv[i]);
krh@232	807	razor_root_close(root);
krh@208	808	return 1;
krh@208	809	}
krh@208	810	}
krh@208	811
krh@210	812	if (dependencies) {
krh@210	813	errors = razor_transaction_resolve(trans);
krh@210	814	if (errors) {
krh@232	815	razor_root_close(root);
krh@210	816	return 1;
krh@210	817	}
krh@197	818	}
krh@152	819
krh@196	820	next = razor_transaction_finish(trans);
krh@152	821
krh@232	822	razor_root_update(root, next);
krh@194	823
krh@206	824	if (mkdir("rpms", 0777) && errno != EEXIST) {
krh@206	825	fprintf(stderr, "failed to create rpms directory.\n");
krh@232	826	razor_root_close(root);
krh@206	827	return 1;
krh@206	828	}
krh@206	829
krh@232	830	razor_set_diff(root->system, next, download_package, &errors);
krh@204	831	if (errors > 0) {
krh@204	832	fprintf(stderr, "failed to download %d packages\n", errors);
krh@232	833	razor_root_close(root);
krh@204	834	return 1;
krh@204	835	}
krh@194	836
krh@200	837	/* FIXME: We need to figure out the right install order here,
krh@200	838	* so the post and pre scripts can run. */
krh@232	839	razor_set_diff(root->system, next, install_package, (void *) root);
krh@171	840
krh@171	841	razor_set_destroy(next);
krh@171	842	razor_set_destroy(upstream);
krh@171	843
krh@232	844	return razor_root_commit(root);
krh@151	845	}
krh@151	846
krh@151	847	static int
krh@151	848	command_init(int argc, const char *argv[])
krh@151	849	{
krh@232	850	return razor_root_create(root);
krh@77	851	}
krh@77	852
krh@187	853	static int
krh@187	854	command_download(int argc, const char *argv[])
krh@187	855	{
krh@187	856	struct razor_set *set;
krh@187	857	struct razor_package_iterator *pi;
krh@187	858	struct razor_package *package;
krh@192	859	const char pattern = argv[0], name, version, arch;
krh@187	860	char url[256], file[256];
krh@203	861	int matches = 0;
krh@187	862
krh@206	863	if (mkdir("rpms", 0777) && errno != EEXIST) {
krh@206	864	fprintf(stderr, "failed to create rpms directory.\n");
krh@206	865	return 1;
krh@206	866	}
krh@206	867
krh@187	868	set = razor_set_open(rawhide_repo_filename);
krh@187	869	pi = razor_package_iterator_create(set);
krh@192	870	while (razor_package_iterator_next(pi, &package,
krh@192	871	&name, &version, &arch)) {
krh@187	872	if (pattern && fnmatch(pattern, name, 0) != 0)
krh@187	873	continue;
krh@187	874
krh@203	875	matches++;
krh@187	876	snprintf(url, sizeof url,
krh@231	877	"%s/Packages/%s-%s.%s.rpm",
krh@231	878	yum_url, name, version, arch);
krh@187	879	snprintf(file, sizeof file,
krh@203	880	"rpms/%s-%s.%s.rpm", name, version, arch);
krh@204	881	download_if_missing(url, file);
krh@187	882	}
krh@187	883	razor_package_iterator_destroy(pi);
krh@187	884	razor_set_destroy(set);
krh@187	885
krh@203	886	if (matches == 0)
krh@203	887	fprintf(stderr, "no packages matched \"%s\"\n", pattern);
krh@203	888	else if (matches == 1)
krh@203	889	fprintf(stderr, "downloaded 1 package\n");
krh@203	890	else
krh@205	891	fprintf(stderr, "downloaded %d packages\n", matches);
krh@203	892
krh@187	893	return 0;
krh@187	894	}
krh@187	895
krh@43	896	static struct {
krh@43	897	const char *name;
krh@43	898	const char *description;
krh@43	899	int (func)(int argc, const char argv[]);
krh@43	900	} razor_commands[] = {
krh@43	901	{ "list", "list all packages", command_list },
krh@67	902	{ "list-requires", "list all requires for the given package", command_list_requires },
krh@75	903	{ "list-provides", "list all provides for the given package", command_list_provides },
krh@75	904	{ "list-obsoletes", "list all obsoletes for the given package", command_list_obsoletes },
krh@75	905	{ "list-conflicts", "list all conflicts for the given package", command_list_conflicts },
krh@48	906	{ "list-files", "list files for package set", command_list_files },
krh@52	907	{ "list-file-packages", "list packages owning file", command_list_file_packages },
krh@56	908	{ "list-package-files", "list files in package", command_list_package_files },
krh@43	909	{ "what-requires", "list the packages that have the given requires", command_what_requires },
krh@43	910	{ "what-provides", "list the packages that have the given provides", command_what_provides },
krh@75	911	{ "import-yum", "import yum metadata files", command_import_yum },
krh@43	912	{ "import-rpmdb", "import the system rpm database", command_import_rpmdb },
krh@75	913	{ "import-rpms", "import rpms from the given directory", command_import_rpms },
krh@44	914	{ "update", "update all or specified packages", command_update },
danw@129	915	{ "remove", "remove specified packages", command_remove },
krh@77	916	{ "diff", "show diff between two package sets", command_diff },
krh@151	917	{ "install", "install rpm", command_install },
krh@187	918	{ "init", "init razor root", command_init },
krh@187	919	{ "download", "download packages", command_download }
krh@43	920	};
krh@43	921
krh@43	922	static int
krh@43	923	usage(void)
krh@43	924	{
krh@43	925	int i;
krh@43	926
krh@43	927	printf("usage:\n");
krh@43	928	for (i = 0; i < ARRAY_SIZE(razor_commands); i++)
krh@43	929	printf(" %-20s%s\n",
krh@43	930	razor_commands[i].name, razor_commands[i].description);
krh@43	931
krh@43	932	return 1;
krh@43	933	}
krh@43	934
krh@43	935	int
krh@43	936	main(int argc, const char *argv[])
krh@43	937	{
krh@43	938	char *repo;
krh@43	939	int i;
krh@43	940
krh@43	941	repo = getenv("RAZOR_REPO");
krh@43	942	if (repo != NULL)
krh@43	943	repo_filename = repo;
krh@43	944
krh@231	945	yum_url = getenv("YUM_URL");
krh@231	946	if (yum_url == NULL)
krh@231	947	yum_url = YUM_URL;
krh@231	948
krh@43	949	if (argc < 2)
krh@43	950	return usage();
krh@43	951
krh@43	952	for (i = 0; i < ARRAY_SIZE(razor_commands); i++)
krh@43	953	if (strcmp(razor_commands[i].name, argv[1]) == 0)
krh@43	954	return razor_commands[i].func(argc - 2, argv + 2);
krh@43	955
krh@43	956	return usage();
krh@43	957	}

author	Kristian H?gsberg <krh@redhat.com>
	Mon Jun 09 16:01:34 2008 -0400 (2008-06-09)
changeset 233	2555ce3e2567
parent 231	fef9808171ff
child 234	7f5d32472bef
permissions	-rw-r--r--