rofi/source/ssh-dialog.c

249 lines
7.2 KiB
C
Raw Normal View History

2014-01-21 09:01:55 +00:00
/**
2014-03-01 16:27:52 +00:00
* rofi
2014-01-21 09:01:55 +00:00
*
* MIT/X11 License
* Copyright 2013-2014 Qball Cow <qball@gmpclient.org>
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*
*/
#include <config.h>
2014-01-21 09:01:55 +00:00
#include <stdlib.h>
#include <stdio.h>
#include <X11/X.h>
#include <unistd.h>
#include <signal.h>
#include <sys/types.h>
#include <dirent.h>
#include <strings.h>
#include <string.h>
#include <ctype.h>
2014-01-21 13:56:25 +00:00
#include <errno.h>
2014-01-21 09:01:55 +00:00
2014-03-12 07:41:38 +00:00
#include "rofi.h"
#include "history.h"
2014-01-21 09:01:55 +00:00
#include "ssh-dialog.h"
#ifdef TIMING
#include <time.h>
#endif
#define SSH_CACHE_FILE "rofi-2.sshcache"
2014-01-21 09:01:55 +00:00
2014-03-22 20:04:19 +00:00
static inline int execshssh ( const char *host )
2014-01-21 09:01:55 +00:00
{
/**
* I am not happy about this code, it causes 7 mallocs and frees
*/
char **args = g_malloc_n ( 7, sizeof ( char* ) );
2014-05-22 07:33:32 +00:00
int i = 0;
args[i++] = g_strdup ( config.terminal_emulator );
2014-06-04 19:29:23 +00:00
if ( config.ssh_set_title ) {
args[i++] = g_strdup ( "-T" );
args[i++] = g_strdup_printf ( "ssh %s", host );
}
args[i++] = g_strdup ( "-e" );
args[i++] = g_strdup ( "ssh" );
args[i++] = g_strdup ( host );
args[i++] = NULL;
2014-08-27 17:44:15 +00:00
GError *error = NULL;
g_spawn_async ( NULL, args, NULL,
G_SPAWN_SEARCH_PATH,
2014-08-27 17:44:15 +00:00
NULL, NULL, NULL, &error );
2014-08-27 17:44:15 +00:00
if( error != NULL )
{
char *msg = g_strdup_printf("Failed to execute: 'ssh %s'\nError: '%s'", host,
error->message);
error_dialog(msg);
g_free(msg);
// print error.
g_error_free(error);
}
// Free the args list.
g_strfreev ( args );
return 0;
2014-01-21 09:01:55 +00:00
}
// execute sub-process
2014-03-22 20:04:19 +00:00
static pid_t exec_ssh ( const char *cmd )
2014-01-21 09:01:55 +00:00
{
2014-06-04 19:29:23 +00:00
if ( !cmd || !cmd[0] ) {
2014-03-22 20:04:19 +00:00
return -1;
}
2014-01-21 09:01:55 +00:00
execshssh ( cmd );
2014-01-21 09:01:55 +00:00
/**
* This happens in non-critical time (After launching app)
* It is allowed to be a bit slower.
*/
char *path = g_strdup_printf ( "%s/%s", cache_dir, SSH_CACHE_FILE );
history_set ( path, cmd );
g_free ( path );
return 0;
2014-01-21 09:01:55 +00:00
}
2014-03-22 20:04:19 +00:00
static void delete_ssh ( const char *cmd )
2014-02-01 13:06:08 +00:00
{
2014-06-04 19:29:23 +00:00
if ( !cmd || !cmd[0] ) {
2014-03-22 20:04:19 +00:00
return;
}
char *path = NULL;
path = g_strdup_printf ( "%s/%s", cache_dir, SSH_CACHE_FILE );
history_remove ( path, cmd );
g_free ( path );
2014-02-01 13:06:08 +00:00
}
2014-08-28 22:01:36 +00:00
static int ssh_sort_func ( const void *a, const void *b )
2014-01-21 09:01:55 +00:00
{
2014-03-22 20:04:19 +00:00
const char *astr = *( const char * const * ) a;
const char *bstr = *( const char * const * ) b;
return g_utf8_collate ( astr, bstr );
2014-01-21 09:01:55 +00:00
}
static char ** get_ssh ( unsigned int *length )
2014-01-21 09:01:55 +00:00
{
2014-03-22 20:04:19 +00:00
unsigned int num_favorites = 0;
char *path;
char **retv = NULL;
2014-01-21 09:01:55 +00:00
#ifdef TIMING
struct timespec start, stop;
2014-03-22 20:04:19 +00:00
clock_gettime ( CLOCK_REALTIME, &start );
2014-01-21 09:01:55 +00:00
#endif
2014-06-04 19:29:23 +00:00
if ( getenv ( "HOME" ) == NULL ) {
2014-03-22 20:04:19 +00:00
return NULL;
}
2014-01-21 09:01:55 +00:00
path = g_strdup_printf ( "%s/%s", cache_dir, SSH_CACHE_FILE );
retv = history_get_list ( path, length );
g_free ( path );
num_favorites = ( *length );
2014-05-13 20:11:42 +00:00
2014-05-22 07:33:32 +00:00
FILE *fd = NULL;
2014-03-22 20:04:19 +00:00
const char *hd = getenv ( "HOME" );
path = g_strdup_printf ( "%s/%s", hd, ".ssh/config" );
fd = fopen ( path, "r" );
2014-01-21 09:01:55 +00:00
2014-06-04 19:29:23 +00:00
if ( fd != NULL ) {
char buffer[1024];
2014-06-04 19:29:23 +00:00
while ( fgets ( buffer, 1024, fd ) != NULL ) {
if ( strncasecmp ( buffer, "Host", 4 ) == 0 ) {
2014-03-22 20:04:19 +00:00
int start = 0, stop = 0;
buffer[strlen ( buffer ) - 1] = '\0';
2014-06-04 19:29:23 +00:00
for ( start = 4; isspace ( buffer[start] ); start++ ) {
2014-03-22 20:04:19 +00:00
;
}
2014-01-21 09:13:42 +00:00
2014-03-22 20:04:19 +00:00
for ( stop = start; isalnum ( buffer[stop] ) ||
2014-01-26 15:47:26 +00:00
buffer[stop] == '_' ||
2014-06-04 19:29:23 +00:00
buffer[stop] == '.'; stop++ ) {
2014-03-22 20:04:19 +00:00
;
}
2014-01-21 09:13:42 +00:00
2014-01-21 09:35:59 +00:00
int found = 0;
2014-01-21 13:56:25 +00:00
2014-06-04 19:29:23 +00:00
if ( start == stop ) {
2014-03-22 20:04:19 +00:00
continue;
}
2014-01-22 09:01:45 +00:00
2014-01-21 09:35:59 +00:00
// This is a nice little penalty, but doable? time will tell.
// given num_favorites is max 25.
2014-06-04 19:29:23 +00:00
for ( unsigned int j = 0; found == 0 && j < num_favorites; j++ ) {
if ( strncasecmp ( &buffer[start], retv[j], stop - start ) == 0 ) {
2014-03-22 20:04:19 +00:00
found = 1;
}
2014-01-21 09:35:59 +00:00
}
2014-01-21 13:56:25 +00:00
2014-06-04 19:29:23 +00:00
if ( found == 1 ) {
2014-03-22 20:04:19 +00:00
continue;
}
2014-01-21 09:35:59 +00:00
retv = g_realloc ( retv, ( ( *length ) + 2 ) * sizeof ( char* ) );
retv[( *length )] = strndup ( &buffer[start], stop - start );
retv[( *length ) + 1] = NULL;
( *length )++;
2014-01-21 09:01:55 +00:00
}
}
2014-03-22 20:04:19 +00:00
fclose ( fd );
2014-01-21 09:01:55 +00:00
}
// TODO: check this is still fast enough. (takes 1ms on laptop.)
if ( ( *length ) > num_favorites ) {
2014-08-28 22:01:36 +00:00
qsort ( &retv[num_favorites], ( *length ) - num_favorites, sizeof ( char* ), ssh_sort_func );
2014-03-11 19:16:44 +00:00
}
g_free ( path );
2014-01-21 09:01:55 +00:00
#ifdef TIMING
2014-03-22 20:04:19 +00:00
clock_gettime ( CLOCK_REALTIME, &stop );
2014-01-21 09:01:55 +00:00
2014-06-04 19:29:23 +00:00
if ( stop.tv_sec != start.tv_sec ) {
2014-03-22 20:04:19 +00:00
stop.tv_nsec += ( stop.tv_sec - start.tv_sec ) * 1e9;
2014-01-21 09:01:55 +00:00
}
2014-03-22 20:04:19 +00:00
long diff = stop.tv_nsec - start.tv_nsec;
printf ( "Time elapsed: %ld us\n", diff / 1000 );
2014-01-21 09:01:55 +00:00
#endif
return retv;
}
SwitcherMode ssh_switcher_dialog ( char **input, G_GNUC_UNUSED void *data )
2014-01-21 09:01:55 +00:00
{
SwitcherMode retv = MODE_EXIT;
// act as a launcher
unsigned int cmd_list_length = 0;
char **cmd_list = get_ssh ( &cmd_list_length );
2014-01-21 09:01:55 +00:00
2014-06-04 19:29:23 +00:00
if ( cmd_list == NULL ) {
cmd_list = g_malloc_n ( 2, sizeof ( char * ) );
cmd_list[0] = g_strdup ( "No ssh hosts found" );
2014-01-21 09:01:55 +00:00
cmd_list[1] = NULL;
}
2014-03-22 20:04:19 +00:00
int shift = 0;
int selected_line = 0;
2014-08-03 15:05:06 +00:00
int mretv = menu ( cmd_list, cmd_list_length, input, "ssh:",
NULL, &shift, token_match, NULL, &selected_line, config.levenshtein_sort );
2014-06-04 19:29:23 +00:00
if ( mretv == MENU_NEXT ) {
2014-01-22 08:24:31 +00:00
retv = NEXT_DIALOG;
2014-03-22 20:04:19 +00:00
}
else if ( mretv == MENU_QUICK_SWITCH ) {
retv = selected_line;
}
2014-06-04 19:29:23 +00:00
else if ( mretv == MENU_OK && cmd_list[selected_line] != NULL ) {
2014-03-22 20:04:19 +00:00
exec_ssh ( cmd_list[selected_line] );
}
2014-06-04 19:29:23 +00:00
else if ( mretv == MENU_CUSTOM_INPUT && *input != NULL && *input[0] != '\0' ) {
2014-03-22 20:04:19 +00:00
exec_ssh ( *input );
}
2014-06-04 19:29:23 +00:00
else if ( mretv == MENU_ENTRY_DELETE && cmd_list[selected_line] ) {
2014-02-01 13:06:08 +00:00
delete_ssh ( cmd_list[selected_line] );
2014-02-01 13:39:49 +00:00
// Stay
retv = RELOAD_DIALOG;
2014-01-21 09:01:55 +00:00
}
g_strfreev ( cmd_list );
2014-01-21 09:01:55 +00:00
return retv;
}