krh@43: #include krh@43: #include krh@43: #include krh@43: #include krh@72: #include krh@72: #include krh@43: #include krh@75: #include krh@71: #include krh@94: #include krh@43: #include "razor.h" krh@151: #include "razor-internal.h" krh@43: krh@152: static const char system_repo_filename[] = "system.repo"; krh@171: static const char next_repo_filename[] = "system-next.repo"; krh@152: static const char rawhide_repo_filename[] = "rawhide.repo"; krh@152: static const char updated_repo_filename[] = "system-updated.repo"; krh@152: static const char razor_root_path[] = "/var/lib/razor"; krh@152: static const char root[] = "install"; krh@152: static const char *repo_filename = system_repo_filename; krh@43: krh@43: static int krh@43: command_list(int argc, const char *argv[]) krh@43: { krh@43: struct razor_set *set; krh@92: struct razor_package_iterator *pi; krh@92: struct razor_package *package; krh@92: const char *pattern = argv[0], *name, *version; krh@43: krh@43: set = razor_set_open(repo_filename); krh@92: pi = razor_package_iterator_create(set); krh@92: while (razor_package_iterator_next(pi, &package, &name, &version)) { krh@92: if (pattern && fnmatch(pattern, name, 0) != 0) krh@92: continue; krh@92: krh@92: printf("%s-%s\n", name, version); krh@92: } krh@92: razor_package_iterator_destroy(pi); krh@92: razor_set_destroy(set); krh@92: krh@92: return 0; krh@92: } krh@92: krh@92: static int krh@92: list_properties(const char *package_name, krh@92: enum razor_property_type required_type) krh@92: { krh@92: struct razor_set *set; krh@92: struct razor_property *property; krh@92: struct razor_package *package; krh@92: struct razor_property_iterator *pi; krh@92: const char *name, *version; krh@92: enum razor_property_type type; danw@109: enum razor_version_relation relation; krh@92: krh@92: set = razor_set_open(repo_filename); krh@92: if (package_name) krh@92: package = razor_set_get_package(set, package_name); krh@92: else krh@92: package = NULL; krh@92: krh@92: pi = razor_property_iterator_create(set, package); krh@92: while (razor_property_iterator_next(pi, &property, danw@109: &name, &relation, &version, danw@109: &type)) { krh@92: if (type != required_type) krh@92: continue; krh@92: if (version[0] == '\0') krh@92: printf("%s\n", name); krh@92: else danw@137: printf("%s %s %s\n", name, danw@137: razor_version_relations[relation], version); krh@92: } krh@92: razor_property_iterator_destroy(pi); krh@92: krh@43: razor_set_destroy(set); krh@43: krh@43: return 0; krh@43: } krh@43: krh@43: static int krh@43: command_list_requires(int argc, const char *argv[]) krh@43: { krh@92: return list_properties(argv[0], RAZOR_PROPERTY_REQUIRES); krh@43: } krh@43: krh@43: static int krh@43: command_list_provides(int argc, const char *argv[]) krh@43: { krh@92: return list_properties(argv[0], RAZOR_PROPERTY_PROVIDES); krh@43: } krh@43: krh@43: static int krh@67: command_list_obsoletes(int argc, const char *argv[]) krh@67: { krh@92: return list_properties(argv[0], RAZOR_PROPERTY_OBSOLETES); krh@67: } krh@67: krh@67: static int krh@67: command_list_conflicts(int argc, const char *argv[]) krh@67: { krh@92: return list_properties(argv[0], RAZOR_PROPERTY_CONFLICTS); krh@67: } krh@67: krh@67: static int krh@48: command_list_files(int argc, const char *argv[]) krh@48: { krh@48: struct razor_set *set; krh@48: krh@48: set = razor_set_open(repo_filename); krh@48: if (set == NULL) krh@48: return 1; krh@49: razor_set_list_files(set, argv[0]); krh@48: razor_set_destroy(set); krh@48: krh@48: return 0; krh@48: } krh@48: krh@48: static int krh@52: command_list_file_packages(int argc, const char *argv[]) krh@52: { krh@52: struct razor_set *set; krh@102: struct razor_package_iterator *pi; krh@102: struct razor_package *package; krh@102: const char *name, *version; krh@52: krh@52: set = razor_set_open(repo_filename); krh@52: if (set == NULL) krh@52: return 1; krh@102: krh@102: pi = razor_package_iterator_create_for_file(set, argv[0]); krh@102: while (razor_package_iterator_next(pi, &package, &name, &version)) krh@102: printf("%s-%s\n", name, version); krh@102: razor_package_iterator_destroy(pi); krh@102: krh@52: razor_set_destroy(set); krh@52: krh@52: return 0; krh@52: } krh@52: krh@56: static int krh@56: command_list_package_files(int argc, const char *argv[]) krh@56: { krh@56: struct razor_set *set; krh@56: krh@56: set = razor_set_open(repo_filename); krh@56: if (set == NULL) krh@56: return 1; krh@56: razor_set_list_package_files(set, argv[0]); krh@56: razor_set_destroy(set); krh@56: krh@56: return 0; krh@56: } krh@52: krh@101: static void krh@101: list_packages_for_property(struct razor_set *set, krh@101: struct razor_property *property) krh@101: { krh@101: struct razor_package_iterator *pi; krh@101: struct razor_package *package; krh@101: const char *name, *version; krh@101: krh@101: pi = razor_package_iterator_create_for_property(set, property); krh@101: while (razor_package_iterator_next(pi, &package, &name, &version)) krh@101: printf("%s-%s\n", name, version); krh@101: razor_package_iterator_destroy(pi); krh@101: } krh@101: krh@52: static int krh@100: list_property_packages(const char *ref_name, krh@100: const char *ref_version, krh@100: enum razor_property_type ref_type) krh@43: { krh@43: struct razor_set *set; krh@100: struct razor_property *property; krh@100: struct razor_property_iterator *pi; krh@100: const char *name, *version; krh@100: enum razor_property_type type; danw@109: enum razor_version_relation relation; krh@100: krh@100: if (ref_name == NULL) krh@100: return 0; krh@43: krh@43: set = razor_set_open(repo_filename); krh@100: if (set == NULL) krh@100: return 1; krh@100: krh@100: pi = razor_property_iterator_create(set, NULL); krh@100: while (razor_property_iterator_next(pi, &property, danw@109: &name, &relation, &version, danw@109: &type)) { krh@100: if (strcmp(ref_name, name) != 0) krh@100: continue; danw@109: if (ref_version && relation == RAZOR_VERSION_EQUAL && danw@109: strcmp(ref_version, version) != 0) krh@100: continue; krh@100: if (ref_type != type) krh@100: continue; krh@100: krh@101: list_packages_for_property(set, property); krh@100: } krh@100: razor_property_iterator_destroy(pi); krh@43: krh@43: return 0; krh@43: } krh@43: krh@43: static int krh@100: command_what_requires(int argc, const char *argv[]) krh@100: { krh@100: return list_property_packages(argv[0], argv[1], krh@100: RAZOR_PROPERTY_REQUIRES); krh@100: } krh@100: krh@100: static int krh@43: command_what_provides(int argc, const char *argv[]) krh@43: { krh@100: return list_property_packages(argv[0], argv[1], krh@100: RAZOR_PROPERTY_PROVIDES); krh@43: } krh@43: krh@73: static int krh@73: show_progress(void *clientp, krh@73: double dltotal, double dlnow, double ultotal, double ulnow) krh@73: { krh@73: const char *file = clientp; krh@73: krh@73: if (!dlnow < dltotal) krh@73: fprintf(stderr, "\rdownloading %s, %dkB/%dkB", krh@73: file, (int) dlnow / 1024, (int) dltotal / 1024); krh@73: else krh@73: fprintf(stderr, "\n"); krh@73: krh@73: return 0; krh@73: } krh@73: krh@73: static int krh@73: download_if_missing(CURL *curl, const char *url, const char *file) krh@73: { krh@73: struct stat buf; krh@73: char error[256]; krh@73: FILE *fp; krh@73: CURLcode res; krh@73: char buffer[256]; krh@73: krh@73: curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, error); krh@73: curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0); krh@73: curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, show_progress); krh@73: curl_easy_setopt(curl, CURLOPT_PROGRESSDATA, file); krh@73: krh@73: if (stat(file, &buf) < 0) { krh@73: fp = fopen(file, "w"); krh@73: snprintf(buffer, sizeof buffer, "%s/%s", url, file); krh@73: curl_easy_setopt(curl, CURLOPT_WRITEDATA, fp); krh@73: curl_easy_setopt(curl, CURLOPT_URL, buffer); krh@73: res = curl_easy_perform(curl); krh@73: fclose(fp); krh@73: if (res != CURLE_OK) { krh@73: fprintf(stderr, "curl error: %s\n", error); krh@73: unlink(file); krh@73: return -1; krh@73: } krh@73: } krh@73: krh@73: return 0; krh@73: } krh@73: krh@71: #define REPO_URL "http://download.fedora.redhat.com" \ krh@71: "/pub/fedora/linux/development/i386/os/repodata" krh@71: krh@43: static int krh@43: command_import_yum(int argc, const char *argv[]) krh@43: { krh@43: struct razor_set *set; krh@71: CURL *curl; krh@71: krh@71: curl = curl_easy_init(); krh@71: if (curl == NULL) krh@71: return 1; krh@71: krh@73: if (download_if_missing(curl, REPO_URL, "primary.xml.gz") < 0) krh@73: return -1; krh@73: if (download_if_missing(curl, REPO_URL, "filelists.xml.gz") < 0) krh@73: return -1; krh@71: curl_easy_cleanup(curl); krh@43: krh@70: set = razor_set_create_from_yum(); krh@43: if (set == NULL) krh@43: return 1; krh@43: razor_set_write(set, rawhide_repo_filename); krh@43: razor_set_destroy(set); krh@43: printf("wrote %s\n", rawhide_repo_filename); krh@43: krh@43: return 0; krh@43: } krh@43: krh@43: static int krh@43: command_import_rpmdb(int argc, const char *argv[]) krh@43: { krh@43: struct razor_set *set; krh@43: krh@43: set = razor_set_create_from_rpmdb(); krh@43: if (set == NULL) krh@43: return 1; krh@43: razor_set_write(set, repo_filename); krh@43: razor_set_destroy(set); krh@43: printf("wrote %s\n", repo_filename); krh@43: krh@43: return 0; krh@43: } krh@43: krh@43: static int krh@43: command_validate(int argc, const char *argv[]) krh@43: { krh@43: struct razor_set *set; krh@43: krh@43: set = razor_set_open(repo_filename); krh@43: if (set == NULL) krh@43: return 1; krh@43: razor_set_list_unsatisfied(set); krh@43: razor_set_destroy(set); krh@43: krh@43: return 0; krh@43: } krh@43: krh@43: static int krh@43: command_update(int argc, const char *argv[]) krh@43: { krh@43: struct razor_set *set, *upstream; danw@137: struct razor_transaction *trans; krh@43: krh@43: set = razor_set_open(repo_filename); krh@43: upstream = razor_set_open(rawhide_repo_filename); krh@43: if (set == NULL || upstream == NULL) krh@43: return 1; danw@137: trans = razor_transaction_create(set, upstream, argc, argv, 0, NULL); danw@137: razor_transaction_describe(trans); danw@137: if (trans->errors) danw@137: return 1; danw@137: danw@137: set = razor_transaction_run(trans); danw@137: razor_transaction_destroy(trans); krh@44: razor_set_write(set, updated_repo_filename); krh@43: razor_set_destroy(set); krh@43: razor_set_destroy(upstream); krh@43: printf("wrote system-updated.repo\n"); krh@43: krh@43: return 0; krh@43: } krh@43: danw@129: static int danw@129: command_remove(int argc, const char *argv[]) danw@129: { danw@129: struct razor_set *set; danw@137: struct razor_transaction *trans; danw@129: danw@129: set = razor_set_open(repo_filename); danw@129: if (set == NULL) danw@129: return 1; danw@137: trans = razor_transaction_create(set, NULL, 0, NULL, argc, argv); danw@137: razor_transaction_describe(trans); danw@137: if (trans->errors) danw@137: return 1; danw@137: danw@137: set = razor_transaction_run(trans); danw@137: razor_transaction_destroy(trans); danw@129: razor_set_write(set, updated_repo_filename); danw@129: razor_set_destroy(set); danw@129: printf("wrote system-updated.repo\n"); danw@129: danw@129: return 0; danw@129: } danw@129: krh@44: static void krh@44: print_diff(const char *name, krh@44: const char *old_version, const char *new_version, void *data) krh@44: { krh@44: if (old_version) krh@44: printf("removing %s %s\n", name, old_version); krh@44: else krh@44: printf("install %s %s\n", name, new_version); krh@44: } krh@44: krh@44: static int krh@44: command_diff(int argc, const char *argv[]) krh@44: { krh@44: struct razor_set *set, *updated; krh@44: krh@44: set = razor_set_open(repo_filename); krh@44: updated = razor_set_open(updated_repo_filename); krh@44: if (set == NULL || updated == NULL) krh@44: return 1; krh@44: krh@44: razor_set_diff(set, updated, print_diff, NULL); krh@44: krh@44: razor_set_destroy(set); krh@44: razor_set_destroy(updated); krh@44: krh@44: return 0; krh@44: } krh@44: krh@74: static int krh@75: command_import_rpms(int argc, const char *argv[]) krh@74: { krh@75: DIR *dir; krh@75: struct dirent *de; krh@75: struct razor_importer *importer; krh@75: struct razor_set *set; krh@77: struct razor_rpm *rpm; krh@75: int len; krh@75: char filename[256]; krh@75: const char *dirname = argv[0]; krh@75: krh@75: if (dirname == NULL) { krh@75: fprintf(stderr, "usage: razor import-rpms DIR\n"); krh@75: return -1; krh@75: } krh@75: krh@75: dir = opendir(dirname); krh@75: if (dir == NULL) { krh@75: fprintf(stderr, "couldn't read dir %s\n", dirname); krh@75: return -1; krh@75: } krh@75: krh@75: importer = razor_importer_new(); krh@75: krh@75: while (de = readdir(dir), de != NULL) { krh@75: len = strlen(de->d_name); krh@75: if (len < 5 || strcmp(de->d_name + len - 4, ".rpm") != 0) krh@75: continue; krh@75: snprintf(filename, sizeof filename, krh@75: "%s/%s", dirname, de->d_name); krh@77: rpm = razor_rpm_open(filename); krh@77: if (rpm == NULL) { krh@77: fprintf(stderr, krh@77: "failed to open rpm \"%s\"\n", filename); krh@77: continue; krh@77: } krh@77: if (razor_importer_add_rpm(importer, rpm)) { krh@75: fprintf(stderr, "couldn't import %s\n", filename); krh@75: break; krh@75: } krh@77: razor_rpm_close(rpm); krh@75: } krh@75: krh@75: if (de != NULL) { krh@75: razor_importer_destroy(importer); krh@75: return -1; krh@75: } krh@75: krh@75: set = razor_importer_finish(importer); krh@75: krh@75: razor_set_write(set, repo_filename); krh@75: razor_set_destroy(set); krh@75: printf("wrote %s\n", repo_filename); krh@74: krh@74: return 0; krh@74: } krh@74: krh@152: static struct razor_set * krh@152: create_set_from_rpms(int argc, const char *argv[]) krh@152: { krh@152: struct razor_importer *importer; krh@152: struct razor_rpm *rpm; krh@152: int i; krh@152: krh@152: importer = razor_importer_new(); krh@152: for (i = 0; i < argc; i++) { krh@152: rpm = razor_rpm_open(argv[i]); krh@152: if (rpm == NULL) { krh@152: fprintf(stderr, krh@152: "failed to open rpm \"%s\"\n", argv[i]); krh@152: continue; krh@152: } krh@152: if (razor_importer_add_rpm(importer, rpm)) { krh@152: fprintf(stderr, "couldn't import %s\n", argv[i]); krh@152: break; krh@152: } krh@152: razor_rpm_close(rpm); krh@152: } krh@152: krh@152: return razor_importer_finish(importer); krh@152: } krh@152: krh@152: static char ** krh@152: list_packages(int count, struct razor_set *set) krh@152: { krh@152: struct razor_package_iterator *pi; krh@152: struct razor_package *package; krh@152: const char *name, *version; krh@152: char **packages; krh@152: int i; krh@152: krh@152: packages = malloc(count * sizeof *packages); krh@152: pi = razor_package_iterator_create(set); krh@152: i = 0; krh@152: while (razor_package_iterator_next(pi, &package, &name, &version)) krh@175: packages[i++] = strdup(name); krh@152: razor_package_iterator_destroy(pi); krh@152: krh@152: return packages; krh@152: } krh@151: krh@77: static int krh@77: command_install(int argc, const char *argv[]) krh@77: { krh@171: struct razor_set *system, *upstream, *next; krh@152: struct razor_transaction *trans; krh@77: struct razor_rpm *rpm; krh@152: const char *filename; krh@171: char path[PATH_MAX], new_path[PATH_MAX], **packages; krh@152: int i; krh@151: krh@152: upstream = create_set_from_rpms(argc, argv); krh@152: snprintf(path, sizeof path, krh@152: "%s%s/%s", root, razor_root_path, system_repo_filename); krh@152: system = razor_set_open(path); krh@152: if (system == NULL) { krh@152: fprintf(stderr, "couldn't open system package database\n"); krh@151: return -1; krh@151: } krh@152: krh@152: packages = list_packages(argc, upstream); krh@152: trans = razor_transaction_create(system, upstream, danw@155: argc, (const char **)packages, danw@155: 0, NULL); danw@155: free(packages); krh@152: razor_transaction_describe(trans); krh@152: if (trans->errors) krh@152: return 1; krh@152: krh@152: /* FIXME: Use _finish() convention here? That is, a function krh@152: * that starts the computation and returns the result while krh@152: * destroying the transaction. Nice for transient objects krh@152: * such as the merger and the importer. Should we do that for krh@152: * transactions too, that is, razor_transaction_finish()? */ krh@171: next = razor_transaction_run(trans); krh@152: razor_transaction_destroy(trans); krh@152: krh@171: /* FIXME: Need razor_set_write_to_fd() so we can open it excl krh@171: * up front here or fail if it already exists. */ krh@171: snprintf(new_path, sizeof new_path, krh@171: "%s%s/%s", root, razor_root_path, next_repo_filename); krh@171: razor_set_write(next, path); krh@171: krh@171: razor_set_destroy(next); krh@171: razor_set_destroy(system); krh@171: razor_set_destroy(upstream); krh@171: krh@171: printf("wrote %s\n", new_path); krh@171: krh@152: for (i = 0; i < argc; i++) { krh@152: filename = argv[i]; krh@152: rpm = razor_rpm_open(argv[i]); krh@152: if (rpm == NULL) { krh@152: fprintf(stderr, "failed to open rpm %s\n", filename); krh@152: return -1; krh@152: } krh@152: if (razor_rpm_install(rpm, root) < 0) { krh@152: fprintf(stderr, krh@152: "failed to install rpm %s\n", filename); krh@152: return -1; krh@152: } krh@152: razor_rpm_close(rpm); krh@152: } krh@151: krh@171: /* Make it so. */ krh@171: rename(new_path, path); krh@171: printf("renamed %s to %s\n", new_path, path); krh@171: krh@151: return 0; krh@151: } krh@151: krh@151: static int krh@151: command_init(int argc, const char *argv[]) krh@151: { krh@77: struct stat buf; krh@151: struct razor_set *set; krh@151: char path[PATH_MAX]; krh@77: krh@77: if (stat(root, &buf) < 0) { krh@77: if (mkdir(root, 0777) < 0) { krh@77: fprintf(stderr, krh@77: "could not create install root \"%s\"\n", krh@77: root); krh@77: return -1; krh@77: } krh@77: fprintf(stderr, "created install root \"%s\"\n", root); krh@77: } else if (!S_ISDIR(buf.st_mode)) { krh@77: fprintf(stderr, krh@77: "install root \"%s\" exists, but is not a directory\n", krh@77: root); krh@77: return -1; krh@77: } krh@77: krh@151: if (razor_create_dir(root, razor_root_path) < 0) { krh@151: fprintf(stderr, "could not create %s%s\n", krh@151: root, razor_root_path); krh@77: return -1; krh@77: } krh@151: krh@151: set = razor_set_create(); krh@151: snprintf(path, sizeof path, "%s%s/%s", krh@152: root, razor_root_path, system_repo_filename); krh@151: if (razor_set_write(set, path) < 0) { krh@151: fprintf(stderr, "could not write initial package set\n"); krh@77: return -1; krh@77: } krh@151: razor_set_destroy(set); krh@77: krh@77: return 0; krh@77: } krh@77: krh@43: static struct { krh@43: const char *name; krh@43: const char *description; krh@43: int (*func)(int argc, const char *argv[]); krh@43: } razor_commands[] = { krh@43: { "list", "list all packages", command_list }, krh@67: { "list-requires", "list all requires for the given package", command_list_requires }, krh@75: { "list-provides", "list all provides for the given package", command_list_provides }, krh@75: { "list-obsoletes", "list all obsoletes for the given package", command_list_obsoletes }, krh@75: { "list-conflicts", "list all conflicts for the given package", command_list_conflicts }, krh@48: { "list-files", "list files for package set", command_list_files }, krh@52: { "list-file-packages", "list packages owning file", command_list_file_packages }, krh@56: { "list-package-files", "list files in package", command_list_package_files }, krh@43: { "what-requires", "list the packages that have the given requires", command_what_requires }, krh@43: { "what-provides", "list the packages that have the given provides", command_what_provides }, krh@75: { "import-yum", "import yum metadata files", command_import_yum }, krh@43: { "import-rpmdb", "import the system rpm database", command_import_rpmdb }, krh@75: { "import-rpms", "import rpms from the given directory", command_import_rpms }, krh@43: { "validate", "validate a package set", command_validate }, krh@44: { "update", "update all or specified packages", command_update }, danw@129: { "remove", "remove specified packages", command_remove }, krh@77: { "diff", "show diff between two package sets", command_diff }, krh@151: { "install", "install rpm", command_install }, krh@151: { "init", "init razor root", command_init } krh@43: }; krh@43: krh@43: static int krh@43: usage(void) krh@43: { krh@43: int i; krh@43: krh@43: printf("usage:\n"); krh@43: for (i = 0; i < ARRAY_SIZE(razor_commands); i++) krh@43: printf(" %-20s%s\n", krh@43: razor_commands[i].name, razor_commands[i].description); krh@43: krh@43: return 1; krh@43: } krh@43: krh@43: int krh@43: main(int argc, const char *argv[]) krh@43: { krh@43: char *repo; krh@43: int i; krh@43: krh@43: repo = getenv("RAZOR_REPO"); krh@43: if (repo != NULL) krh@43: repo_filename = repo; krh@43: krh@43: if (argc < 2) krh@43: return usage(); krh@43: krh@43: for (i = 0; i < ARRAY_SIZE(razor_commands); i++) krh@43: if (strcmp(razor_commands[i].name, argv[1]) == 0) krh@43: return razor_commands[i].func(argc - 2, argv + 2); krh@43: krh@43: return usage(); krh@43: }