938 lines
28 KiB
C
938 lines
28 KiB
C
/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-
|
|
*
|
|
* Copyright (C) 2008 Richard Hughes <richard@hughsie.com>
|
|
*
|
|
* Licensed under the GNU General Public License Version 2
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <locale.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#include <signal.h>
|
|
#include <glib/gi18n.h>
|
|
#include <packagekit-glib2/packagekit.h>
|
|
#include <packagekit-glib2/packagekit-private.h>
|
|
|
|
#define PK_MAX_PATH_LEN 1023
|
|
|
|
typedef enum {
|
|
PK_CNF_POLICY_RUN,
|
|
PK_CNF_POLICY_INSTALL,
|
|
PK_CNF_POLICY_ASK,
|
|
PK_CNF_POLICY_WARN,
|
|
PK_CNF_POLICY_UNKNOWN
|
|
} PkCnfPolicy;
|
|
|
|
typedef struct {
|
|
PkCnfPolicy single_match;
|
|
PkCnfPolicy multiple_match;
|
|
PkCnfPolicy single_install;
|
|
PkCnfPolicy multiple_install;
|
|
gboolean software_source_search;
|
|
gboolean similar_name_search;
|
|
gchar **locations;
|
|
guint max_search_time;
|
|
} PkCnfPolicyConfig;
|
|
|
|
static PkTask *task = NULL;
|
|
static GCancellable *cancellable = NULL;
|
|
|
|
/* bash reserved code */
|
|
#define EXIT_COMMAND_NOT_FOUND 127
|
|
|
|
/**
|
|
*
|
|
* Swizzle ordering, e.g. amke -> make
|
|
**/
|
|
static void
|
|
pk_cnf_find_alternatives_swizzle (const gchar *cmd, guint len, GPtrArray *array)
|
|
{
|
|
guint i;
|
|
gchar *possible;
|
|
gchar swap;
|
|
|
|
/* */
|
|
for (i = 0; i < len-1; i++) {
|
|
possible = g_strdup (cmd);
|
|
swap = possible[i];
|
|
possible[i] = possible[i+1];
|
|
possible[i+1] = swap;
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
}
|
|
|
|
/**
|
|
*
|
|
* Replace some easily confused chars, e.g. gnome-power-managir to gnome-power-manager
|
|
**/
|
|
static void
|
|
pk_cnf_find_alternatives_replace (const gchar *cmd, guint len, GPtrArray *array)
|
|
{
|
|
guint i;
|
|
gchar *possible;
|
|
gchar temp;
|
|
|
|
/* replace some easily confused chars */
|
|
for (i = 0; i < len; i++) {
|
|
temp = cmd[i];
|
|
if (temp == 'i') {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = 'e';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
if (temp == 'e') {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = 'i';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
if (temp == 'i') {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = 'o';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
if (temp == 'c') {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = 's';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
if (temp == 's') {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = 'c';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
if (temp == 's') {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = 'z';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
if (temp == 'z') {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = 's';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
if (temp == 'k') {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = 'c';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
if (temp == 'c') {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = 'k';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
*
|
|
* Truncate first and last char, so lshall -> lshal
|
|
**/
|
|
static void
|
|
pk_cnf_find_alternatives_truncate (const gchar *cmd, guint len, GPtrArray *array)
|
|
{
|
|
guint i;
|
|
gchar *possible;
|
|
|
|
/* truncate last char */
|
|
possible = g_strdup (cmd);
|
|
possible[len-1] = '\0';
|
|
g_ptr_array_add (array, possible);
|
|
|
|
/* truncate first char */
|
|
possible = g_strdup (cmd);
|
|
for (i = 0; i < len-1; i++)
|
|
possible[i] = possible[i+1];
|
|
possible[len-1] = '\0';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
|
|
/**
|
|
*
|
|
* Remove double chars, e.g. gnome-power-manaager -> gnome-power-manager
|
|
**/
|
|
static void
|
|
pk_cnf_find_alternatives_remove_double (const gchar *cmd, guint len, GPtrArray *array)
|
|
{
|
|
guint i, j;
|
|
gchar *possible;
|
|
|
|
for (i = 1; i < len; i++) {
|
|
if (cmd[i-1] == cmd[i]) {
|
|
possible = g_strdup (cmd);
|
|
for (j = i; j < len; j++)
|
|
possible[j] = possible[j+1];
|
|
possible[len-1] = '\0';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
*
|
|
* Fix British spellings, e.g. colourdiff -> colordiff
|
|
**/
|
|
static void
|
|
pk_cnf_find_alternatives_locale (const gchar *cmd, guint len, GPtrArray *array)
|
|
{
|
|
guint i, j;
|
|
gchar *possible;
|
|
|
|
for (i = 1; i < len; i++) {
|
|
if (cmd[i-1] == 'o' && cmd[i] == 'u') {
|
|
possible = g_strdup (cmd);
|
|
for (j = i; j < len; j++)
|
|
possible[j] = possible[j+1];
|
|
possible[len-1] = '\0';
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
*
|
|
* Suggest Linux commands for Solaris commands
|
|
**/
|
|
static void
|
|
pk_cnf_find_alternatives_solaris (const gchar *cmd, guint len, GPtrArray *array)
|
|
{
|
|
const gchar *tmp;
|
|
g_autoptr(GHashTable) hash = NULL;
|
|
|
|
hash = g_hash_table_new (g_str_hash, g_str_equal);
|
|
g_hash_table_insert (hash, (gpointer) "smuser", (gpointer) "usermod");
|
|
g_hash_table_insert (hash, (gpointer) "logins", (gpointer) "usermod");
|
|
g_hash_table_insert (hash, (gpointer) "adb", (gpointer) "gdb");
|
|
g_hash_table_insert (hash, (gpointer) "add_drv", (gpointer) "modprobe");
|
|
g_hash_table_insert (hash, (gpointer) "modload", (gpointer) "modprobe");
|
|
g_hash_table_insert (hash, (gpointer) "modunload", (gpointer) "modprobe");
|
|
g_hash_table_insert (hash, (gpointer) "rem_drv", (gpointer) "modprobe");
|
|
g_hash_table_insert (hash, (gpointer) "audit", (gpointer) "auditctl");
|
|
g_hash_table_insert (hash, (gpointer) "auditreduce", (gpointer) "auditctl");
|
|
g_hash_table_insert (hash, (gpointer) "cfgadm", (gpointer) "lsmod");
|
|
g_hash_table_insert (hash, (gpointer) "clri", (gpointer) "fsck");
|
|
g_hash_table_insert (hash, (gpointer) "fsdb", (gpointer) "fsck");
|
|
g_hash_table_insert (hash, (gpointer) "volcheck", (gpointer) "fsck");
|
|
g_hash_table_insert (hash, (gpointer) "crle", (gpointer) "ldconfig");
|
|
g_hash_table_insert (hash, (gpointer) "devfsadm", (gpointer) "udevtrigger");
|
|
g_hash_table_insert (hash, (gpointer) "devlinks", (gpointer) "ln");
|
|
g_hash_table_insert (hash, (gpointer) "dfshares", (gpointer) "exportfs");
|
|
g_hash_table_insert (hash, (gpointer) "share", (gpointer) "exportfs");
|
|
g_hash_table_insert (hash, (gpointer) "shareall", (gpointer) "exportfs");
|
|
g_hash_table_insert (hash, (gpointer) "dladm", (gpointer) "ifconfig");
|
|
g_hash_table_insert (hash, (gpointer) "kstat", (gpointer) "ifconfig");
|
|
g_hash_table_insert (hash, (gpointer) "dtrace", (gpointer) "stap");
|
|
g_hash_table_insert (hash, (gpointer) "eeprom", (gpointer) "hwclock");
|
|
g_hash_table_insert (hash, (gpointer) "fcinfo", (gpointer) "lspci");
|
|
g_hash_table_insert (hash, (gpointer) "prtfru", (gpointer) "lspci");
|
|
g_hash_table_insert (hash, (gpointer) "fmthard", (gpointer) "fdisk");
|
|
g_hash_table_insert (hash, (gpointer) "format", (gpointer) "fdisk");
|
|
g_hash_table_insert (hash, (gpointer) "prtvtoc", (gpointer) "fdisk");
|
|
g_hash_table_insert (hash, (gpointer) "installboot", (gpointer) "mkbootdisk");
|
|
g_hash_table_insert (hash, (gpointer) "installpatch", (gpointer) "yum");
|
|
g_hash_table_insert (hash, (gpointer) "patchaddpkgadd", (gpointer) "yum");
|
|
g_hash_table_insert (hash, (gpointer) "pkgchk", (gpointer) "yum");
|
|
g_hash_table_insert (hash, (gpointer) "pkginfo", (gpointer) "yum");
|
|
g_hash_table_insert (hash, (gpointer) "pkgrm", (gpointer) "yum");
|
|
g_hash_table_insert (hash, (gpointer) "prodreg", (gpointer) "yum");
|
|
g_hash_table_insert (hash, (gpointer) "showrev", (gpointer) "yum");
|
|
g_hash_table_insert (hash, (gpointer) "isainfo", (gpointer) "uname");
|
|
g_hash_table_insert (hash, (gpointer) "luxadm", (gpointer) "systool");
|
|
g_hash_table_insert (hash, (gpointer) "mkfile", (gpointer) "touch");
|
|
g_hash_table_insert (hash, (gpointer) "mpathadm", (gpointer) "multipath");
|
|
g_hash_table_insert (hash, (gpointer) "stmsboot", (gpointer) "multipath");
|
|
g_hash_table_insert (hash, (gpointer) "ndd", (gpointer) "modinfo");
|
|
g_hash_table_insert (hash, (gpointer) "newfs", (gpointer) "mkfs");
|
|
g_hash_table_insert (hash, (gpointer) "pbind", (gpointer) "taskset");
|
|
g_hash_table_insert (hash, (gpointer) "pldd", (gpointer) "ldd");
|
|
g_hash_table_insert (hash, (gpointer) "praudit", (gpointer) "auditctl");
|
|
g_hash_table_insert (hash, (gpointer) "prstat", (gpointer) "ps");
|
|
g_hash_table_insert (hash, (gpointer) "prtconf", (gpointer) "dmesg");
|
|
g_hash_table_insert (hash, (gpointer) "psrinfo", (gpointer) "dmidecode");
|
|
g_hash_table_insert (hash, (gpointer) "sysdef", (gpointer) "dmidecode");
|
|
g_hash_table_insert (hash, (gpointer) "ptree", (gpointer) "pstree");
|
|
g_hash_table_insert (hash, (gpointer) "snoop", (gpointer) "tcpdump");
|
|
g_hash_table_insert (hash, (gpointer) "sotruss", (gpointer) "strace");
|
|
g_hash_table_insert (hash, (gpointer) "truss", (gpointer) "strace");
|
|
g_hash_table_insert (hash, (gpointer) "svcadm", (gpointer) "service");
|
|
g_hash_table_insert (hash, (gpointer) "svcs", (gpointer) "service");
|
|
g_hash_table_insert (hash, (gpointer) "swap", (gpointer) "swapon");
|
|
g_hash_table_insert (hash, (gpointer) "trapstat", (gpointer) "oprofile");
|
|
|
|
/* find anything that matches exactly */
|
|
tmp = g_hash_table_lookup (hash, cmd);
|
|
if (tmp != NULL)
|
|
g_ptr_array_add (array, g_strdup (tmp));
|
|
}
|
|
|
|
/**
|
|
*
|
|
* Remove double chars, e.g. Lshal -> lshal
|
|
**/
|
|
static void
|
|
pk_cnf_find_alternatives_case (const gchar *cmd, guint len, GPtrArray *array)
|
|
{
|
|
guint i;
|
|
gchar *possible;
|
|
gchar temp;
|
|
|
|
for (i = 0; i < len; i++) {
|
|
temp = g_ascii_tolower (cmd[i]);
|
|
if (temp != cmd[i]) {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = temp;
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
temp = g_ascii_toupper (cmd[i]);
|
|
if (temp != cmd[i]) {
|
|
possible = g_strdup (cmd);
|
|
possible[i] = temp;
|
|
g_ptr_array_add (array, possible);
|
|
}
|
|
}
|
|
|
|
/* all lower */
|
|
possible = g_strdup (cmd);
|
|
for (i = 0; i < len; i++)
|
|
possible[i] = g_ascii_tolower (cmd[i]);
|
|
if (strcmp (possible, cmd) != 0)
|
|
g_ptr_array_add (array, possible);
|
|
else
|
|
g_free (possible);
|
|
|
|
/* all upper */
|
|
possible = g_strdup (cmd);
|
|
for (i = 0; i < len; i++)
|
|
possible[i] = g_ascii_toupper (cmd[i]);
|
|
if (strcmp (possible, cmd) != 0)
|
|
g_ptr_array_add (array, possible);
|
|
else
|
|
g_free (possible);
|
|
}
|
|
|
|
/**
|
|
*
|
|
* Generate a list of commands it might be
|
|
**/
|
|
static GPtrArray *
|
|
pk_cnf_find_alternatives (const gchar *cmd, guint len)
|
|
{
|
|
GPtrArray *array;
|
|
const gchar *cmdt;
|
|
const gchar *cmdt2;
|
|
guint i, j;
|
|
gchar buffer_bin[PK_MAX_PATH_LEN+1];
|
|
gchar buffer_sbin[PK_MAX_PATH_LEN+1];
|
|
gboolean ret;
|
|
g_autoptr(GPtrArray) possible = NULL;
|
|
g_autoptr(GPtrArray) unique = NULL;
|
|
|
|
array = g_ptr_array_new_with_free_func (g_free);
|
|
possible = g_ptr_array_new_with_free_func (g_free);
|
|
unique = g_ptr_array_new ();
|
|
pk_cnf_find_alternatives_swizzle (cmd, len, possible);
|
|
pk_cnf_find_alternatives_replace (cmd, len, possible);
|
|
if (len > 3)
|
|
pk_cnf_find_alternatives_truncate (cmd, len, possible);
|
|
pk_cnf_find_alternatives_remove_double (cmd, len, possible);
|
|
pk_cnf_find_alternatives_case (cmd, len, possible);
|
|
pk_cnf_find_alternatives_locale (cmd, len, possible);
|
|
pk_cnf_find_alternatives_solaris (cmd, len, possible);
|
|
|
|
/* remove duplicates using a helper array */
|
|
for (i = 0; i < possible->len; i++) {
|
|
cmdt = g_ptr_array_index (possible, i);
|
|
ret = TRUE;
|
|
for (j=0; j<unique->len; j++) {
|
|
cmdt2 = g_ptr_array_index (unique, j);
|
|
if (strcmp (cmdt, cmdt2) == 0) {
|
|
ret = FALSE;
|
|
break;
|
|
}
|
|
}
|
|
/* only add if not duplicate */
|
|
if (ret)
|
|
g_ptr_array_add (unique, (gpointer) cmdt);
|
|
}
|
|
|
|
/* ITS4: ignore, source is constant size */
|
|
strncpy (buffer_bin, "/usr/bin/", PK_MAX_PATH_LEN);
|
|
|
|
/* ITS4: ignore, source is constant size */
|
|
strncpy (buffer_sbin, "/usr/sbin/", PK_MAX_PATH_LEN);
|
|
|
|
/* remove any that exist (fast path) */
|
|
for (i = 0; i < unique->len; i++) {
|
|
cmdt = g_ptr_array_index (unique, i);
|
|
|
|
/* ITS4: ignore, size is checked */
|
|
strncpy (&buffer_bin[9], cmdt, PK_MAX_PATH_LEN-9);
|
|
|
|
/* ITS4: ignore, size is checked */
|
|
strncpy (&buffer_sbin[10], cmdt, PK_MAX_PATH_LEN-10);
|
|
|
|
/* does file exist in bindir (common case) */
|
|
ret = g_file_test (buffer_bin, G_FILE_TEST_EXISTS | G_FILE_TEST_IS_EXECUTABLE);
|
|
if (ret) {
|
|
g_ptr_array_add (array, g_strdup (cmdt));
|
|
continue;
|
|
}
|
|
|
|
/* does file exist in sbindir */
|
|
ret = g_file_test (buffer_sbin, G_FILE_TEST_EXISTS | G_FILE_TEST_IS_EXECUTABLE);
|
|
if (ret)
|
|
g_ptr_array_add (array, g_strdup (cmdt));
|
|
}
|
|
return array;
|
|
}
|
|
|
|
static void
|
|
pk_cnf_progress_cb (PkProgress *progress, PkProgressType type, gpointer data)
|
|
{
|
|
PkStatusEnum status;
|
|
const gchar *text = NULL;
|
|
|
|
/* status */
|
|
if (type != PK_PROGRESS_TYPE_STATUS)
|
|
return;
|
|
|
|
g_object_get (progress,
|
|
"status", &status,
|
|
NULL);
|
|
|
|
switch (status) {
|
|
case PK_STATUS_ENUM_SETUP:
|
|
case PK_STATUS_ENUM_FINISHED:
|
|
case PK_STATUS_ENUM_QUERY:
|
|
break;
|
|
case PK_STATUS_ENUM_DOWNLOAD_REPOSITORY:
|
|
/* TRANSLATORS: downloading repo data so we can search */
|
|
text = _("Downloading details about the software repositories.");
|
|
break;
|
|
case PK_STATUS_ENUM_DOWNLOAD_FILELIST:
|
|
/* TRANSLATORS: downloading file lists so we can search */
|
|
text = _("Downloading filelists (this may take some time to complete).");
|
|
break;
|
|
case PK_STATUS_ENUM_WAITING_FOR_LOCK:
|
|
/* TRANSLATORS: waiting for native lock */
|
|
text = _("Waiting for package manager lock.");
|
|
break;
|
|
case PK_STATUS_ENUM_LOADING_CACHE:
|
|
/* TRANSLATORS: loading package cache so we can search */
|
|
text = _("Loading list of packages.");
|
|
break;
|
|
default:
|
|
/* fallback to default */
|
|
text = pk_status_enum_to_localised_text (status);
|
|
}
|
|
|
|
/* print to screen, still one line */
|
|
if (text != NULL)
|
|
g_print ("\n * %s... ", text);
|
|
}
|
|
|
|
static gboolean
|
|
pk_cnf_cancel_cb (GCancellable *_cancellable)
|
|
{
|
|
g_warning ("Cancelling request");
|
|
g_cancellable_cancel (cancellable);
|
|
return FALSE;
|
|
}
|
|
|
|
/**
|
|
* Find software we could install
|
|
**/
|
|
static gchar **
|
|
pk_cnf_find_available (const gchar *cmd, guint max_search_time)
|
|
{
|
|
PkPackage *item;
|
|
gchar **package_ids = NULL;
|
|
const gchar *prefixes[] = {"/usr/bin", "/usr/sbin", "/bin", "/sbin", NULL};
|
|
gchar **values = NULL;
|
|
g_autoptr(GError) error = NULL;
|
|
GPtrArray *array = NULL;
|
|
guint i;
|
|
guint len;
|
|
PkBitfield filters;
|
|
PkResults *results = NULL;
|
|
PkError *error_code = NULL;
|
|
guint cancel_id;
|
|
|
|
/* create new array of full paths */
|
|
len = g_strv_length ((gchar **)prefixes);
|
|
values = g_new0 (gchar *, len + 1);
|
|
for (i = 0; prefixes[i] != NULL; i++)
|
|
values[i] = g_build_filename (prefixes[i], cmd, NULL);
|
|
|
|
/* only allow searching for a limited amount of time */
|
|
cancel_id = g_timeout_add (max_search_time,
|
|
(GSourceFunc) pk_cnf_cancel_cb,
|
|
cancellable);
|
|
g_source_set_name_by_id (cancel_id, "[PkCommandNotFound] cancel");
|
|
|
|
/* do search */
|
|
filters = pk_bitfield_from_enums (PK_FILTER_ENUM_NOT_INSTALLED,
|
|
PK_FILTER_ENUM_NEWEST,
|
|
PK_FILTER_ENUM_ARCH, -1);
|
|
results = pk_client_search_files (PK_CLIENT(task), filters, values, cancellable,
|
|
NULL, NULL, &error);
|
|
if (results == NULL) {
|
|
if (!g_error_matches (error, PK_CLIENT_ERROR, PK_CLIENT_ERROR_INVALID_INPUT)) {
|
|
/* TRANSLATORS: we failed to find the package, this shouldn't happen */
|
|
g_printerr ("%s: %s\n", _("Failed to search for file"), error->message);
|
|
}
|
|
goto out;
|
|
}
|
|
|
|
/* check error code */
|
|
error_code = pk_results_get_error_code (results);
|
|
if (error_code != NULL) {
|
|
if (pk_error_get_code (error_code) == PK_ERROR_ENUM_TRANSACTION_CANCELLED) {
|
|
g_debug ("The search was cancelled as it was taking too long");
|
|
} else {
|
|
/* TRANSLATORS: the transaction failed in a way we could not expect */
|
|
g_printerr ("%s: %s, %s\n", _("Getting the list of files failed"),
|
|
pk_error_enum_to_string (pk_error_get_code (error_code)),
|
|
pk_error_get_details (error_code));
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
/* get the packages returned */
|
|
array = pk_results_get_package_array (results);
|
|
package_ids = g_new0 (gchar *, array->len+1);
|
|
for (i = 0; i < array->len; i++) {
|
|
item = g_ptr_array_index (array, i);
|
|
package_ids[i] = g_strdup (pk_package_get_id (item));
|
|
}
|
|
out:
|
|
if (cancel_id > 0)
|
|
g_source_remove (cancel_id);
|
|
if (error_code != NULL)
|
|
g_object_unref (error_code);
|
|
if (results != NULL)
|
|
g_object_unref (results);
|
|
if (array != NULL)
|
|
g_ptr_array_unref (array);
|
|
g_strfreev (values);
|
|
return package_ids;
|
|
}
|
|
|
|
static PkCnfPolicy
|
|
pk_cnf_get_policy_from_string (const gchar *policy_text)
|
|
{
|
|
if (policy_text == NULL)
|
|
return PK_CNF_POLICY_UNKNOWN;
|
|
if (strcmp (policy_text, "run") == 0)
|
|
return PK_CNF_POLICY_RUN;
|
|
if (strcmp (policy_text, "ask") == 0)
|
|
return PK_CNF_POLICY_ASK;
|
|
if (strcmp (policy_text, "warn") == 0)
|
|
return PK_CNF_POLICY_WARN;
|
|
return PK_CNF_POLICY_UNKNOWN;
|
|
}
|
|
|
|
static PkCnfPolicy
|
|
pk_cnf_get_policy_from_file (GKeyFile *file, const gchar *key)
|
|
{
|
|
PkCnfPolicy policy;
|
|
g_autofree gchar *policy_text = NULL;
|
|
g_autoptr(GError) error = NULL;
|
|
|
|
/* get from file */
|
|
policy_text = g_key_file_get_string (file, "CommandNotFound", key, &error);
|
|
if (policy_text == NULL) {
|
|
g_warning ("failed to get key %s: %s", key, error->message);
|
|
}
|
|
|
|
/* convert to enum */
|
|
policy = pk_cnf_get_policy_from_string (policy_text);
|
|
return policy;
|
|
}
|
|
|
|
static PkCnfPolicyConfig *
|
|
pk_cnf_get_config (void)
|
|
{
|
|
GKeyFile *file;
|
|
gboolean ret;
|
|
g_autofree gchar *path = NULL;
|
|
g_autoptr(GError) error = NULL;
|
|
PkCnfPolicyConfig *config;
|
|
|
|
/* create */
|
|
config = g_new0 (PkCnfPolicyConfig, 1);
|
|
|
|
/* set defaults if the conf file is not found */
|
|
config->single_match = PK_CNF_POLICY_UNKNOWN;
|
|
config->multiple_match = PK_CNF_POLICY_UNKNOWN;
|
|
config->single_install = PK_CNF_POLICY_UNKNOWN;
|
|
config->multiple_install = PK_CNF_POLICY_UNKNOWN;
|
|
config->software_source_search = TRUE;
|
|
config->similar_name_search = TRUE;
|
|
config->locations = NULL;
|
|
config->max_search_time = 5000;
|
|
|
|
/* load file */
|
|
file = g_key_file_new ();
|
|
path = g_build_filename (SYSCONFDIR, "PackageKit", "CommandNotFound.conf", NULL);
|
|
ret = g_key_file_load_from_file (file, path, G_KEY_FILE_NONE, &error);
|
|
if (!ret) {
|
|
g_printerr ("failed to load config file: %s\n", error->message);
|
|
goto out;
|
|
}
|
|
|
|
/* get data */
|
|
config->single_match = pk_cnf_get_policy_from_file (file, "SingleMatch");
|
|
config->multiple_match = pk_cnf_get_policy_from_file (file, "MultipleMatch");
|
|
config->single_install = pk_cnf_get_policy_from_file (file, "SingleInstall");
|
|
config->multiple_install = pk_cnf_get_policy_from_file (file, "MultipleInstall");
|
|
config->software_source_search = g_key_file_get_boolean (file, "CommandNotFound", "SoftwareSourceSearch", NULL);
|
|
config->similar_name_search = g_key_file_get_boolean (file, "CommandNotFound", "SimilarNameSearch", NULL);
|
|
config->locations = g_key_file_get_string_list (file, "CommandNotFound", "SearchLocations", NULL, NULL);
|
|
config->max_search_time = g_key_file_get_integer (file, "CommandNotFound", "MaxSearchTime", NULL);
|
|
|
|
/* fallback */
|
|
if (config->locations == NULL) {
|
|
g_warning ("not found SearchLocations, using fallback");
|
|
config->locations = g_strsplit ("/usr/bin;/usr/sbin", ";", -1);
|
|
}
|
|
if (config->max_search_time == 0) {
|
|
g_warning ("not found MaxSearchTime, using fallback");
|
|
config->max_search_time = 2000;
|
|
}
|
|
out:
|
|
g_key_file_free (file);
|
|
return config;
|
|
}
|
|
|
|
static gint
|
|
pk_cnf_spawn_command (const gchar *exec, gchar **arguments, guint arguments_count)
|
|
{
|
|
g_autofree gchar **argv = NULL;
|
|
gint exit_status = EXIT_FAILURE;
|
|
g_autoptr(GError) error = NULL;
|
|
|
|
/* ensure program starts on a fresh line */
|
|
g_print ("\n");
|
|
|
|
argv = g_new0 (gchar *, arguments_count + 2);
|
|
argv[0] = (gchar*) exec;
|
|
for (guint i = 0; i <= arguments_count; i++) {
|
|
argv[1+i] = arguments[i];
|
|
}
|
|
if (!g_spawn_sync (NULL, argv, NULL, G_SPAWN_SEARCH_PATH, NULL, NULL,
|
|
NULL, NULL, &exit_status, &error)) {
|
|
/* TRANSLATORS: we failed to launch the executable, the error follows */
|
|
g_printerr ("%s '%s': %s\n", _("Failed to launch:"), exec, error->message);
|
|
}
|
|
return exit_status;
|
|
}
|
|
|
|
static gboolean
|
|
pk_cnf_install_package_id (const gchar *package_id)
|
|
{
|
|
g_autoptr(GError) error = NULL;
|
|
g_autoptr(PkError) error_code = NULL;
|
|
g_autoptr(PkResults) results = NULL;
|
|
g_auto(GStrv) package_ids = NULL;
|
|
|
|
/* do install */
|
|
package_ids = pk_package_ids_from_id (package_id);
|
|
results = pk_task_install_packages_sync (task, package_ids, cancellable,
|
|
(PkProgressCallback) pk_cnf_progress_cb, NULL, &error);
|
|
if (results == NULL) {
|
|
/* TRANSLATORS: we failed to install the package */
|
|
g_printerr ("%s: %s\n", _("Failed to install packages"), error->message);
|
|
return FALSE;
|
|
}
|
|
|
|
/* check error code */
|
|
error_code = pk_results_get_error_code (results);
|
|
if (error_code != NULL) {
|
|
/* TRANSLATORS: the transaction failed in a way we could not expect */
|
|
g_printerr ("%s: %s, %s\n", _("The transaction failed"), pk_error_enum_to_string (pk_error_get_code (error_code)), pk_error_get_details (error_code));
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
static void
|
|
pk_cnf_sigint_handler (int sig)
|
|
{
|
|
g_debug ("Handling SIGINT");
|
|
|
|
/* restore default ASAP, as the cancel might hang */
|
|
signal (SIGINT, SIG_DFL);
|
|
|
|
/* hopefully, cancel client */
|
|
g_cancellable_cancel (cancellable);
|
|
|
|
/* kill ourselves */
|
|
g_debug ("Retrying SIGINT");
|
|
kill (getpid (), SIGINT);
|
|
}
|
|
|
|
static gboolean
|
|
pk_cnf_is_backend_fast_enough_to_do_search (void)
|
|
{
|
|
gboolean ret = FALSE;
|
|
gchar *backend;
|
|
GError *error = NULL;
|
|
PkControl *control = NULL;
|
|
|
|
/* Initialize PkControl */
|
|
control = pk_control_new ();
|
|
ret = pk_control_get_properties (control, cancellable, &error);
|
|
if (!ret) {
|
|
/* failed to contact the daemon */
|
|
g_error_free (error);
|
|
goto out;
|
|
}
|
|
|
|
/* Find backend name */
|
|
g_object_get (control, "backend-name", &backend, NULL);
|
|
|
|
/* Current list of too slow backends */
|
|
if (g_strcmp0 (backend, "yum") == 0) {
|
|
ret = FALSE;
|
|
goto out;
|
|
}
|
|
out:
|
|
if (control != NULL)
|
|
g_object_unref(control);
|
|
return ret;
|
|
}
|
|
|
|
int
|
|
main (int argc, char *argv[])
|
|
{
|
|
gboolean ret;
|
|
PkCnfPolicyConfig *config = NULL;
|
|
guint i;
|
|
guint len;
|
|
gchar *text;
|
|
const gchar *possible;
|
|
gchar **parts;
|
|
guint retval = EXIT_COMMAND_NOT_FOUND;
|
|
const gchar *shell = "bash";
|
|
const gchar *env_shell;
|
|
g_autofree gchar *shell_to_free = NULL;
|
|
g_autoptr(GPtrArray) array = NULL;
|
|
g_auto(GStrv) package_ids = NULL;
|
|
|
|
setlocale (LC_ALL, "");
|
|
bindtextdomain (GETTEXT_PACKAGE, PACKAGE_LOCALE_DIR);
|
|
bind_textdomain_codeset (GETTEXT_PACKAGE, "UTF-8");
|
|
textdomain (GETTEXT_PACKAGE);
|
|
|
|
/* don't show debugging, unless VERBOSE is specified */
|
|
pk_debug_add_log_domain (G_LOG_DOMAIN);
|
|
|
|
/* no input */
|
|
if (argv[1] == NULL)
|
|
goto out;
|
|
|
|
/* do stuff on ctrl-c */
|
|
signal (SIGINT, pk_cnf_sigint_handler);
|
|
|
|
/* get policy config */
|
|
config = pk_cnf_get_config ();
|
|
task = PK_TASK(pk_task_text_new ());
|
|
g_object_set (task,
|
|
"cache-age", G_MAXUINT,
|
|
"interactive", FALSE,
|
|
"background", FALSE,
|
|
NULL);
|
|
cancellable = g_cancellable_new ();
|
|
|
|
/* get length */
|
|
len = strlen (argv[1]);
|
|
if (len < 1)
|
|
goto out;
|
|
|
|
/* invalid */
|
|
if (g_strcmp0 (argv[1], ".") == 0 || g_strcmp0 (argv[1], "..") == 0)
|
|
goto out;
|
|
|
|
env_shell = g_getenv ("SHELL");
|
|
if (env_shell != NULL)
|
|
shell = shell_to_free = g_path_get_basename (env_shell);
|
|
|
|
/* TRANSLATORS: the prefix of all the output telling the user
|
|
* why it's not executing. NOTE: this is lowercase to mimic
|
|
* the style of bash itself -- apologies */
|
|
g_printerr ("%s: %s: %s...\n", shell, argv[1], _("command not found"));
|
|
|
|
/* ignore one char mistakes */
|
|
if (len < 2)
|
|
goto out;
|
|
|
|
/* user is not allowing CNF to do anything useful */
|
|
if (!config->software_source_search &&
|
|
!config->similar_name_search) {
|
|
goto out;
|
|
}
|
|
|
|
/* generate swizzles */
|
|
if (config->similar_name_search)
|
|
array = pk_cnf_find_alternatives (argv[1], len);
|
|
|
|
/* one exact possibility */
|
|
if (array != NULL && array->len == 1) {
|
|
possible = g_ptr_array_index (array, 0);
|
|
if (config->single_match == PK_CNF_POLICY_WARN) {
|
|
/* TRANSLATORS: tell the user what we think the command is */
|
|
g_printerr ("%s '%s'\n", _("Similar command is:"), possible);
|
|
goto out;
|
|
}
|
|
|
|
/* run */
|
|
if (config->single_match == PK_CNF_POLICY_RUN) {
|
|
retval = pk_cnf_spawn_command (possible, &argv[2], argc - 2);
|
|
goto out;
|
|
}
|
|
|
|
/* ask */
|
|
if (config->single_match == PK_CNF_POLICY_ASK) {
|
|
/* TRANSLATORS: Ask the user if we should run the similar command */
|
|
text = g_strdup_printf ("%s %s", _("Run similar command:"), possible);
|
|
ret = pk_console_get_prompt (text, TRUE);
|
|
if (ret)
|
|
retval = pk_cnf_spawn_command (possible, &argv[2], argc - 2);
|
|
g_free (text);
|
|
}
|
|
goto out;
|
|
|
|
/* multiple choice */
|
|
} else if (array != NULL && array->len > 1) {
|
|
if (config->multiple_match == PK_CNF_POLICY_WARN) {
|
|
/* TRANSLATORS: show the user a list of commands that they could have meant */
|
|
g_printerr ("%s:\n", _("Similar commands are:"));
|
|
for (i = 0; i < array->len; i++) {
|
|
possible = g_ptr_array_index (array, i);
|
|
g_printerr ("'%s'\n", possible);
|
|
}
|
|
|
|
/* ask */
|
|
} else if (config->multiple_match == PK_CNF_POLICY_ASK) {
|
|
/* TRANSLATORS: show the user a list of commands we could run */
|
|
g_printerr ("%s:\n", _("Similar commands are:"));
|
|
for (i = 0; i < array->len; i++) {
|
|
possible = g_ptr_array_index (array, i);
|
|
g_printerr ("%i\t'%s'\n", i+1, possible);
|
|
}
|
|
|
|
/* TRANSLATORS: ask the user to choose a file to run */
|
|
i = pk_console_get_number (_("Please choose a command to run"), array->len);
|
|
g_assert (i < array->len);
|
|
|
|
/* run command */
|
|
possible = g_ptr_array_index (array, i);
|
|
retval = pk_cnf_spawn_command (possible, &argv[2], argc - 2);
|
|
}
|
|
goto out;
|
|
|
|
/* only search using PackageKit if configured to do so */
|
|
} else if (config->software_source_search &&
|
|
pk_cnf_is_backend_fast_enough_to_do_search ()) {
|
|
package_ids = pk_cnf_find_available (argv[1], config->max_search_time);
|
|
if (package_ids == NULL)
|
|
goto out;
|
|
len = g_strv_length (package_ids);
|
|
if (len == 1) {
|
|
parts = pk_package_id_split (package_ids[0]);
|
|
if (config->single_install == PK_CNF_POLICY_WARN) {
|
|
/* TRANSLATORS: tell the user what package provides the command */
|
|
g_printerr ("%s '%s'\n", _("The package providing this file is:"), parts[PK_PACKAGE_ID_NAME]);
|
|
goto out;
|
|
}
|
|
|
|
/* ask */
|
|
if (config->single_install == PK_CNF_POLICY_ASK) {
|
|
/* TRANSLATORS: as the user if we want to install a package to provide the command */
|
|
text = g_strdup_printf (_("Install package '%s' to provide command '%s'?"), parts[PK_PACKAGE_ID_NAME], argv[1]);
|
|
ret = pk_console_get_prompt (text, FALSE);
|
|
g_free (text);
|
|
if (ret) {
|
|
ret = pk_cnf_install_package_id (package_ids[0]);
|
|
if (ret)
|
|
retval = pk_cnf_spawn_command (argv[1], &argv[2], argc - 2);
|
|
}
|
|
g_print ("\n");
|
|
goto out;
|
|
}
|
|
|
|
/* install */
|
|
if (config->single_install == PK_CNF_POLICY_INSTALL) {
|
|
ret = pk_cnf_install_package_id (package_ids[0]);
|
|
if (ret)
|
|
retval = pk_cnf_spawn_command (argv[1], &argv[2], argc - 2);
|
|
}
|
|
g_strfreev (parts);
|
|
goto out;
|
|
} else if (len > 1) {
|
|
if (config->multiple_install == PK_CNF_POLICY_WARN) {
|
|
/* TRANSLATORS: Show the user a list of packages that provide this command */
|
|
g_printerr ("%s\n", _("Packages providing this file are:"));
|
|
for (i = 0; package_ids[i] != NULL; i++) {
|
|
parts = pk_package_id_split (package_ids[i]);
|
|
g_printerr ("'%s'\n", parts[PK_PACKAGE_ID_NAME]);
|
|
g_strfreev (parts);
|
|
}
|
|
|
|
/* ask */
|
|
} else if (config->multiple_install == PK_CNF_POLICY_ASK) {
|
|
/* TRANSLATORS: Show the user a list of packages that they can install to provide this command */
|
|
g_printerr ("%s:\n", _("Suitable packages are:"));
|
|
for (i = 0; package_ids[i] != NULL; i++) {
|
|
parts = pk_package_id_split (package_ids[i]);
|
|
g_printerr ("%i\t'%s'\n", i+1, parts[PK_PACKAGE_ID_NAME]);
|
|
g_strfreev (parts);
|
|
}
|
|
|
|
/* TRANSLATORS: ask the user to choose a file to install */
|
|
i = pk_console_get_number (_("Please choose a package to install"), len);
|
|
if (i == 0) {
|
|
g_printerr ("%s\n", _("User aborted selection"));
|
|
goto out;
|
|
}
|
|
g_assert (i < len);
|
|
|
|
/* run command */
|
|
ret = pk_cnf_install_package_id (package_ids[i - 1]);
|
|
if (ret)
|
|
retval = pk_cnf_spawn_command (argv[1], &argv[2], argc - 2);
|
|
}
|
|
goto out;
|
|
}
|
|
}
|
|
out:
|
|
if (task != NULL)
|
|
g_object_unref (task);
|
|
if (cancellable != NULL)
|
|
g_object_unref (cancellable);
|
|
if (config != NULL) {
|
|
g_strfreev (config->locations);
|
|
g_free (config);
|
|
}
|
|
return retval;
|
|
}
|
|
|