telnet.c 17.4 KB
Newer Older
1 2 3
/*****************************************************************************
 * telnet.c: VLM interface plugin
 *****************************************************************************
4
 * Copyright (C) 2000-2005 the VideoLAN team
hartman's avatar
hartman committed
5
 * $Id$
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
 *
 * Authors: Simon Latapie <garf@videolan.org>
 *          Laurent Aimar <fenrir@videolan.org>
 *
 * 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
dionoea's avatar
dionoea committed
22
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
#include <stdlib.h>                                      /* malloc(), free() */

#include <vlc/vlc.h>
#include <vlc/intf.h>

#include <vlc/input.h>

#include <sys/stat.h>

#include <errno.h>
#include <fcntl.h>

#ifdef HAVE_SYS_TIME_H
#    include <sys/time.h>
#endif

#ifdef HAVE_UNISTD_H
#   include <unistd.h>
#endif

#include "network.h"
49
#include "vlc_url.h"
50

51
#include "vlc_vlm.h"
52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71

#define READ_MODE_PWD 1
#define READ_MODE_CMD 2
#define WRITE_MODE_PWD 3 // when we write the word "Password:"
#define WRITE_MODE_CMD 4

/* telnet commands */
#define TEL_WILL    251
#define TEL_WONT    252
#define TEL_DO      253
#define TEL_DONT    254
#define TEL_IAC     255
#define TEL_ECHO    1

/*****************************************************************************
 * Module descriptor
 *****************************************************************************/
static int  Open ( vlc_object_t * );
static void Close( vlc_object_t * );

72 73 74 75 76 77 78 79
#define TELNETHOST_TEXT N_( "Telnet interface listen host" )
#define TELNETHOST_LONGTEXT N_( "This is the host on which the telnet " \
    "interface will listen. It defaults to all network interfaces (0.0.0.0)." \
    " I you want the telnet interface to be available only on the local " \
    "machine, enter 127.0.0.1" )
#define TELNETPORT_TEXT N_( "Telnet interface port" )
#define TELNETPORT_LONGTEXT N_( "This is the TCP port on which the telnet " \
    "interface will listen. It defaults to 4212" )
80
#define TELNETPORT_DEFAULT 4212
81 82 83
#define TELNETPWD_TEXT N_( "Telnet interface password" )
#define TELNETPWD_LONGTEXT N_( "The telnet interface uses a single " \
    "administration password. The default value is \"admin\"." )
84
#define TELNETPWD_DEFAULT "admin"
85 86

vlc_module_begin();
87
    set_shortname( "Telnet" );
88
    set_category( CAT_INTERFACE );
zorglub's avatar
zorglub committed
89
    set_subcategory( SUBCAT_INTERFACE_CONTROL );
90
    add_string( "telnet-host", "", NULL, TELNETHOST_TEXT,
91 92
                 TELNETHOST_LONGTEXT, VLC_TRUE );
    add_integer( "telnet-port", TELNETPORT_DEFAULT, NULL, TELNETPORT_TEXT,
93
                 TELNETPORT_LONGTEXT, VLC_TRUE );
94
    add_string( "telnet-password", TELNETPWD_DEFAULT, NULL, TELNETPWD_TEXT,
95
                TELNETPWD_LONGTEXT, VLC_TRUE );
96
    set_description( _("VLM remote control interface") );
97
    add_category_hint( "VLM", NULL, VLC_FALSE );
98 99 100 101 102 103 104 105 106 107 108 109 110
    set_capability( "interface", 0 );
    set_callbacks( Open , Close );
vlc_module_end();

/*****************************************************************************
 * Local prototypes.
 *****************************************************************************/
static void Run( intf_thread_t * );

typedef struct
{
    int        i_mode; /* read or write */
    int        fd;
111
    char       buffer_read[1000]; // 1000 byte per command should be sufficient
112
    char      *buffer_write;
113 114
    char      *p_buffer_read;
    char      *p_buffer_write; // the position in the buffer
115 116
    int        i_buffer_write; // the number of byte we still have to send
    int        i_tel_cmd; // for specific telnet commands
117

118 119
} telnet_client_t;

120 121
static char *MessageToString( vlm_message_t *, int );
static void Write_message( telnet_client_t *, vlm_message_t *, char *, int );
122 123 124 125 126

struct intf_sys_t
{
   telnet_client_t **clients;
   int             i_clients;
127
   int             *pi_fd;
128
   vlm_t           *mediatheque;
129 130
};

131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
/* 
 * getPort: Decide which port to use. There are two possibilities to
 * specify a port: integrated in the --telnet-host option with :PORT
 * or using the --telnet-port option. The --telnet-port option has
 * precedence. 
 * This code relies upon the fact the url.i_port is 0 if the :PORT 
 * option is missing from --telnet-host.
 */
static int getPort(intf_thread_t *p_intf, vlc_url_t url, int i_port)
    {
    // Print error if two different ports have been specified
    if (url.i_port != 0  &&
        i_port != TELNETPORT_DEFAULT && 
        url.i_port != i_port )
    {
	    msg_Err( p_intf, "ignoring port %d and using %d", url.i_port,
                 i_port);
    }
    if (i_port != TELNETPORT_DEFAULT)
    {
	    return i_port;
    }
    if (url.i_port != 0)
    {
 	    return url.i_port;
    }
    // return default
    return i_port;
}

161 162 163 164 165 166
/*****************************************************************************
 * Open: initialize dummy interface
 *****************************************************************************/
static int Open( vlc_object_t *p_this )
{
    intf_thread_t *p_intf = (intf_thread_t*) p_this;
167
    vlm_t *mediatheque;
168 169
    char *psz_address;
    vlc_url_t url;
170 171
    int i_telnetport;

172 173 174 175 176
    if( !(mediatheque = vlm_New( p_intf )) )
    {
        msg_Err( p_intf, "cannot start VLM" );
        return VLC_EGENERIC;
    }
177

178
    msg_Info( p_intf, "Using the VLM interface plugin..." );
179

180
    i_telnetport = config_GetInt( p_intf, "telnet-port" );
181 182 183 184 185 186 187
    psz_address  = config_GetPsz( p_intf, "telnet-host" );

    vlc_UrlParse(&url, psz_address, 0);

    // There might be two ports given, resolve any potentially
    // conflict
    url.i_port = getPort(p_intf, url, i_telnetport);
188 189

    p_intf->p_sys = malloc( sizeof( intf_sys_t ) );
190
    if( ( p_intf->p_sys->pi_fd = net_ListenTCP( p_intf, url.psz_host, url.i_port ) )
191
                == NULL )
192 193
    {
        msg_Err( p_intf, "cannot listen for telnet" );
194 195
        vlc_UrlClean(&url);
        free( psz_address );
196 197 198
        free( p_intf->p_sys );
        return VLC_EGENERIC;
    }
199 200
    msg_Info( p_intf, 
              "Telnet interface started on interface %s %d",
201
              url.psz_host, url.i_port );
hartman's avatar
hartman committed
202

203 204
    p_intf->p_sys->i_clients   = 0;
    p_intf->p_sys->clients     = NULL;
205 206
    p_intf->p_sys->mediatheque = mediatheque;
    p_intf->pf_run = Run;
207

208
    vlc_UrlClean(&url);
209
    free( psz_address );
210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230
    return VLC_SUCCESS;
}

/*****************************************************************************
 * Close:
 *****************************************************************************/
static void Close( vlc_object_t *p_this )
{
    intf_thread_t *p_intf = (intf_thread_t*)p_this;
    intf_sys_t    *p_sys  = p_intf->p_sys;
    int i;

    for( i = 0; i < p_sys->i_clients; i++ )
    {
        telnet_client_t *cl = p_sys->clients[i];

        net_Close( cl->fd );
        free( cl );
    }
    if( p_sys->clients != NULL ) free( p_sys->clients );

231
    net_ListenClose( p_sys->pi_fd );
232 233 234 235 236 237 238 239 240 241 242 243 244

    vlm_Delete( p_sys->mediatheque );

    free( p_sys );
}

/*****************************************************************************
 * Run: main loop
 *****************************************************************************/
static void Run( intf_thread_t *p_intf )
{
    intf_sys_t     *p_sys = p_intf->p_sys;
    struct timeval  timeout;
245
    char           *psz_password;
246

247
    psz_password = config_GetPsz( p_intf, "telnet-password" );
248 249 250

    while( !p_intf->b_die )
    {
251 252 253
        fd_set fds_read, fds_write;
        int    i_handle_max = 0;
        int    i_ret, i_len, fd, i;
254 255

        /* if a new client wants to communicate */
256
        fd = net_Accept( p_intf, p_sys->pi_fd, p_sys->i_clients > 0 ? 0 : -1 );
257 258 259 260
        if( fd > 0 )
        {
            telnet_client_t *cl;

261
            /* to be non blocking */
262 263 264 265 266 267 268 269 270 271 272 273 274
#if defined( WIN32 ) || defined( UNDER_CE )
            {
                unsigned long i_dummy = 1;
                ioctlsocket( fd, FIONBIO, &i_dummy );
            }
#else
            fcntl( fd, F_SETFL, O_NONBLOCK );
#endif
            cl = malloc( sizeof( telnet_client_t ));
            cl->i_tel_cmd = 0;
            cl->fd = fd;
            cl->buffer_write = NULL;
            cl->p_buffer_write = cl->buffer_write;
275
            Write_message( cl, NULL, "Password: \xff\xfb\x01", WRITE_MODE_PWD );
276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301

            TAB_APPEND( p_sys->i_clients, p_sys->clients, cl );
        }

        /* to do a proper select */
        FD_ZERO( &fds_read );
        FD_ZERO( &fds_write );

        for( i = 0 ; i < p_sys->i_clients ; i++ )
        {
            telnet_client_t *cl = p_sys->clients[i];

            if( cl->i_mode == WRITE_MODE_PWD || cl->i_mode == WRITE_MODE_CMD )
            {
                FD_SET( cl->fd , &fds_write );
            }
            else
            {
                FD_SET( cl->fd , &fds_read );
            }
            i_handle_max = __MAX( i_handle_max, cl->fd );
        }

        timeout.tv_sec = 0;
        timeout.tv_usec = 500*1000;

302
        i_ret = select( i_handle_max + 1, &fds_read, &fds_write, 0, &timeout );
303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331
        if( i_ret == -1 && errno != EINTR )
        {
            msg_Warn( p_intf, "cannot select sockets" );
            msleep( 1000 );
            continue;
        }
        else if( i_ret <= 0 )
        {
            continue;
        }

        /* check if there is something to do with the socket */
        for( i = 0 ; i < p_sys->i_clients ; i++ )
        {
            telnet_client_t *cl = p_sys->clients[i];

            if( FD_ISSET(cl->fd , &fds_write) && cl->i_buffer_write > 0 )
            {
                i_len = send( cl->fd , cl->p_buffer_write ,
                              cl->i_buffer_write , 0 );
                if( i_len > 0 )
                {
                    cl->p_buffer_write += i_len;
                    cl->i_buffer_write -= i_len;
                }
            }
            else if( FD_ISSET( cl->fd, &fds_read) )
            {
                int i_end = 0;
332
                int i_recv;
333

334
                while( (i_recv=recv( cl->fd, cl->p_buffer_read, 1, 0 )) > 0 &&
335 336 337 338
                       cl->p_buffer_read - cl->buffer_read < 999 )
                {
                    switch( cl->i_tel_cmd )
                    {
339
                    case 0:
340
                        switch( *(uint8_t *)cl->p_buffer_read )
341 342 343 344 345 346 347 348 349 350
                        {
                        case '\r':
                            break;
                        case '\n':
                            *cl->p_buffer_read = '\n';
                            i_end = 1;
                            break;
                        case TEL_IAC: // telnet specific command
                            cl->i_tel_cmd = 1;
                            cl->p_buffer_read++;
351
                            break;
352 353
                        default:
                            cl->p_buffer_read++;
354
                            break;
355 356 357
                        }
                        break;
                    case 1:
358
                        switch( *(uint8_t *)cl->p_buffer_read )
359 360 361 362 363 364 365
                        {
                        case TEL_WILL: case TEL_WONT:
                        case TEL_DO: case TEL_DONT:
                            cl->i_tel_cmd++;
                            cl->p_buffer_read++;
                            break;
                        default:
366
                            cl->i_tel_cmd = 0;
367
                            cl->p_buffer_read--;
368
                            break;
369 370 371 372 373
                        }
                        break;
                    case 2:
                        cl->i_tel_cmd = 0;
                        cl->p_buffer_read -= 2;
374 375
                        break;
                    }
376 377

                    if( i_end != 0 ) break;
378 379
                }

380
                if( cl->p_buffer_read - cl->buffer_read == 999 )
381
                {
382
                    Write_message( cl, NULL, "Line too long\r\n",
383
                                   cl->i_mode + 2 );
384
                }
385 386 387 388 389 390 391 392

                if (i_recv == 0)
                {
                    net_Close( cl->fd );
                    TAB_REMOVE( p_intf->p_sys->i_clients ,
                                p_intf->p_sys->clients , cl );
                    free( cl );
                }
393 394 395 396 397 398 399 400
            }
        }

        /* and now we should bidouille the data we received / send */
        for( i = 0 ; i < p_sys->i_clients ; i++ )
        {
            telnet_client_t *cl = p_sys->clients[i];

401
            if( cl->i_mode >= WRITE_MODE_PWD && cl->i_buffer_write == 0 )
402
            {
403
               // we have finished to send
404 405
               cl->i_mode -= 2; // corresponding READ MODE
            }
406 407
            else if( cl->i_mode == READ_MODE_PWD &&
                     *cl->p_buffer_read == '\n' )
408 409
            {
                *cl->p_buffer_read = '\0';
410
                if( strcmp( psz_password, cl->buffer_read ) == 0 )
411
                {
412 413
                    Write_message( cl, NULL, "\xff\xfc\x01\r\nWelcome, "
                                   "Master\r\n> ", WRITE_MODE_CMD );
414 415 416 417
                }
                else
                {
                    /* wrong password */
418 419
                    Write_message( cl, NULL,
                                   "\r\nWrong password.\r\nPassword: ",
420
                                   WRITE_MODE_PWD );
421 422
                }
            }
423 424
            else if( cl->i_mode == READ_MODE_CMD &&
                     *cl->p_buffer_read == '\n' )
425 426 427 428 429 430
            {
                /* ok, here is a command line */
                if( !strncmp( cl->buffer_read, "logout", 6 ) ||
                    !strncmp( cl->buffer_read, "quit", 4 )  ||
                    !strncmp( cl->buffer_read, "exit", 4 ) )
                {
431
                    net_Close( cl->fd );
432 433
                    TAB_REMOVE( p_intf->p_sys->i_clients ,
                                p_intf->p_sys->clients , cl );
434 435 436 437 438 439 440 441 442
                    free( cl );
                }
                else if( !strncmp( cl->buffer_read, "shutdown", 8 ) )
                {
                    msg_Err( p_intf, "shutdown requested" );
                    p_intf->p_vlc->b_die = VLC_TRUE;
                }
                else
                {
443
                    vlm_message_t *message;
444 445 446 447

                    /* create a standard string */
                    *cl->p_buffer_read = '\0';

448 449 450
                    vlm_ExecuteCommand( p_sys->mediatheque, cl->buffer_read,
                                        &message );
                    Write_message( cl, message, NULL, WRITE_MODE_CMD );
451
                    vlm_MessageDelete( message );
452 453 454 455 456 457
                }
            }
        }
    }
}

458 459
static void Write_message( telnet_client_t *client, vlm_message_t *message,
                           char *string_message, int i_mode )
460
{
461 462
    char *psz_message;

463 464 465
    client->p_buffer_read = client->buffer_read;
    (client->p_buffer_read)[0] = 0; // if (cl->p_buffer_read)[0] = '\n'
    if( client->buffer_write ) free( client->buffer_write );
466 467

    /* generate the psz_message string */
468
    if( message )
469
    {
470 471
        /* ok, look for vlm_message_t */
        psz_message = MessageToString( message, 0 );
472
    }
473
    else
474
    {
475
        /* it is a basic string_message */
476 477 478
        psz_message = strdup( string_message );
    }

479
    client->buffer_write = client->p_buffer_write = psz_message;
480
    client->i_buffer_write = strlen( psz_message );
481 482 483
    client->i_mode = i_mode;
}

484 485 486
/* We need the level of the message to put a beautiful indentation.
 * first level is 0 */
static char *MessageToString( vlm_message_t *message, int i_level )
487
{
488 489 490
#define STRING_CR "\r\n"
#define STRING_TAIL "> "

491
    char *psz_message;
492
    int i, i_message = sizeof( STRING_TAIL );
493

494
    if( !message || !message->psz_name )
495
    {
496
        return strdup( STRING_CR STRING_TAIL );
497
    }
498
    else if( !i_level && !message->i_child && !message->psz_value  )
499
    {
500
        /* A command is successful. Don't write anything */
501
        return strdup( /*STRING_CR*/ STRING_TAIL );
502 503 504
    }

    i_message += strlen( message->psz_name ) + i_level * sizeof( "    " ) + 1;
505 506
    psz_message = malloc( i_message );
    *psz_message = 0;
507 508 509 510 511 512 513 514 515 516 517
    for( i = 0; i < i_level; i++ ) strcat( psz_message, "    " );
    strcat( psz_message, message->psz_name );

    if( message->psz_value )
    {
        i_message += sizeof( " : " ) + strlen( message->psz_value ) +
            sizeof( STRING_CR );
        psz_message = realloc( psz_message, i_message );
        strcat( psz_message, " : " );
        strcat( psz_message, message->psz_value );
        strcat( psz_message, STRING_CR );
518 519 520
    }
    else
    {
521 522 523 524
        i_message += sizeof( STRING_CR );
        psz_message = realloc( psz_message, i_message );
        strcat( psz_message, STRING_CR );
    }
525

526 527 528 529
    for( i = 0; i < message->i_child; i++ )
    {
        char *child_message =
            MessageToString( message->child[i], i_level + 1 );
530

531 532 533 534
        i_message += strlen( child_message );
        psz_message = realloc( psz_message, i_message );
        strcat( psz_message, child_message );
        free( child_message );
535
    }
536 537 538 539

    if( i_level == 0 ) strcat( psz_message, STRING_TAIL );

    return psz_message;
540
}