PolytreeDE/rofi
1
0
Fork 0
mirror of https://github.com/davatorium/rofi.git synced 2025-07-31 21:59:25 -04:00
Code Releases Activity

Merge pull request #105 from Tblue/master

Browse source 
SSH config file parser: Recognize multiple host names in a "Host" line
This commit is contained in:
Dave Davenport 2015-01-14 21:51:19 +01:00
commit 3a4c1f1f3f
1 changed files with 34 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

67
source/ssh-dialog.c
View file

@ -49,6 +49,10 @@
#define SSH_CACHE_FILE "rofi-2.sshcache" #define SSH_CACHE_FILE "rofi-2.sshcache"
// Used in get_ssh() when splitting lines from the user's
// SSH config file into tokens.
#define SSH_TOKEN_DELIM "= \t\r\n"
static inline int execshssh ( const char *host ) static inline int execshssh ( const char *host )
{ {
/** /**
@ -137,52 +141,49 @@ static char ** get_ssh ( unsigned int *length )
if ( fd != NULL ) { if ( fd != NULL ) {
char buffer[1024]; char buffer[1024];
while ( fgets ( buffer, 1024, fd ) != NULL ) { while ( fgets ( buffer, sizeof( buffer ), fd ) ) {
char *token = &buffer[0]; // Each line is either empty, a comment line starting with a '#'
// Skip initial spaces. // character or of the form "keyword [=] arguments", where there may
while ( ( *token ) != '\n' && ( *token ) != '\0' && isspace ( *token ) ) { // be multiple (possibly quoted) arguments separated by whitespace.
token++; // The keyword is separated from its arguments by whitespace OR by
} // optional whitespace and a '=' character.
// Skip empty lines. char *token = strtok( buffer, SSH_TOKEN_DELIM );
if ( ( *token ) == '\n' || ( *token ) == '\0' ) {
// Skip empty lines and comment lines. Also skip lines where the
// keyword is not "Host".
if ( ! token || *token == '#' || g_ascii_strcasecmp( token, "Host" ) ) {
continue; continue;
} }
// Check for "Host[::space::]"
if ( strncasecmp ( token, "Host", 4 ) == 0 && isspace ( token[4] ) ) {
int start = 0, stop = 0;
buffer[strlen ( buffer ) - 1] = '\0';
for ( start = 4; isspace ( buffer[start] ); start++ ) { // Now we know that this is a "Host" line.
; // The "Host" keyword is followed by one more host names separated
// by whitespace; while host names may be quoted with double quotes
// to represent host names containing spaces, we don't support this
// (how many host names contain spaces?).
while ( ( token = strtok( NULL, SSH_TOKEN_DELIM ) ) ) {
// We do not want to show wildcard entries, as you cannot ssh to them.
if ( *token == '!' || strpbrk( token, "*?" ) ) {
continue;
} }
for ( stop = start; !isspace ( buffer[stop] ); stop++ ) { // Is this host name already in the history file?
// We do not want to show wildcard entries, as you cannot ssh to them. // This is a nice little penalty, but doable? time will tell.
if ( buffer[stop] == '?' || buffer[stop] == '*' ) { // given num_favorites is max 25.
stop = start; int found = 0;
for ( unsigned int j = 0; j < num_favorites; j++ ) {
if ( ! g_ascii_strcasecmp ( token, retv[j] ) ) {
found = 1;
break; break;
} }
} }
int found = 0; if ( found ) {
if ( start == stop ) {
continue;
}
// This is a nice little penalty, but doable? time will tell.
// given num_favorites is max 25.
for ( unsigned int j = 0; found == 0 && j < num_favorites; j++ ) {
if ( strncasecmp ( &buffer[start], retv[j], stop - start ) == 0 ) {
found = 1;
}
}
if ( found == 1 ) {
continue; continue;
} }
// Add this host name to the list.
retv = g_realloc ( retv, ( ( *length ) + 2 ) * sizeof ( char* ) ); retv = g_realloc ( retv, ( ( *length ) + 2 ) * sizeof ( char* ) );
retv[( *length )] = strndup ( &buffer[start], stop - start ); retv[( *length )] = g_strdup ( token );
retv[( *length ) + 1] = NULL; retv[( *length ) + 1] = NULL;
( *length )++; ( *length )++;
} }