telnet.c 20.2 KB
Newer Older
1 2 3
/*****************************************************************************
 * telnet.c: VLM interface plugin
 *****************************************************************************
4
 * Copyright (C) 2000-2006 the VideoLAN team
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
Antoine Cellerier's avatar
Antoine Cellerier committed
22
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23 24 25 26 27 28
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/

29 30 31 32
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

33
#include <vlc_common.h>
34
#include <vlc_plugin.h>
Clément Stenac's avatar
Clément Stenac committed
35 36
#include <vlc_interface.h>
#include <vlc_input.h>
37

38
#include <stdbool.h>
39 40 41 42 43 44

#include <errno.h>

#ifdef HAVE_UNISTD_H
#   include <unistd.h>
#endif
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
45
#ifdef HAVE_POLL
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
46 47
#   include <poll.h>
#endif
48

Clément Stenac's avatar
Clément Stenac committed
49 50 51
#include <vlc_network.h>
#include <vlc_url.h>
#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
#define TELNETHOST_TEXT N_( "Host" )
#define TELNETHOST_LONGTEXT N_( "This is the host on which the " \
74
    "interface will listen. It defaults to all network interfaces (0.0.0.0)." \
75 76 77 78 79
    " If you want this interface to be available only on the local " \
    "machine, enter \"127.0.0.1\"." )
#define TELNETPORT_TEXT N_( "Port" )
#define TELNETPORT_LONGTEXT N_( "This is the TCP port on which this " \
    "interface will listen. It defaults to 4212." )
80
#define TELNETPORT_DEFAULT 4212
81 82 83
#define TELNETPWD_TEXT N_( "Password" )
#define TELNETPWD_LONGTEXT N_( "A single administration password is used " \
    "to protect this interface. The default value is \"admin\"." )
84
#define TELNETPWD_DEFAULT "admin"
85

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

/*****************************************************************************
 * 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
static char *MessageToString( vlm_message_t *, int );
Laurent Aimar's avatar
Laurent Aimar committed
121
static void Write_message( telnet_client_t *, vlm_message_t *, const 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
 * 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
135 136
 * precedence.
 * This code relies upon the fact the url.i_port is 0 if the :PORT
137 138
 * option is missing from --telnet-host.
 */
139
static int getPort(intf_thread_t *p_intf, const vlc_url_t *url, int i_port)
Jean-Paul Saman's avatar
Jean-Paul Saman committed
140
{
141 142 143 144 145
    if (i_port == TELNETPORT_DEFAULT && url->i_port != 0)
        i_port = url->i_port;
    if (url->i_port != 0 && url->i_port != i_port)
        // Print error if two different ports have been specified
        msg_Warn( p_intf, "ignoring port %d (using %d)", url->i_port, i_port );
146 147 148
    return i_port;
}

149 150 151 152 153 154
/*****************************************************************************
 * Open: initialize dummy interface
 *****************************************************************************/
static int Open( vlc_object_t *p_this )
{
    intf_thread_t *p_intf = (intf_thread_t*) p_this;
155
    vlm_t *mediatheque;
156 157
    char *psz_address;
    vlc_url_t url;
158 159
    int i_telnetport;

160 161 162 163 164
    if( !(mediatheque = vlm_New( p_intf )) )
    {
        msg_Err( p_intf, "cannot start VLM" );
        return VLC_EGENERIC;
    }
165

166
    msg_Info( p_intf, "using the VLM interface plugin..." );
167

168 169
    i_telnetport = var_InheritInteger( p_intf, "telnet-port" );
    psz_address  = var_InheritString( p_intf, "telnet-host" );
170
    vlc_UrlParse(&url, psz_address, 0);
171
    free( psz_address );
172 173 174

    // There might be two ports given, resolve any potentially
    // conflict
175
    url.i_port = getPort(p_intf, &url, i_telnetport);
176 177

    p_intf->p_sys = malloc( sizeof( intf_sys_t ) );
178 179
    if( !p_intf->p_sys )
    {
180
        vlm_Delete( mediatheque );
181 182 183 184
        vlc_UrlClean( &url );
        return VLC_ENOMEM;
    }
    if( ( p_intf->p_sys->pi_fd = net_ListenTCP( p_intf, url.psz_host, url.i_port ) ) == NULL )
185 186
    {
        msg_Err( p_intf, "cannot listen for telnet" );
187
        vlm_Delete( mediatheque );
188
        vlc_UrlClean( &url );
189 190 191
        free( p_intf->p_sys );
        return VLC_EGENERIC;
    }
192
    msg_Info( p_intf,
193
              "telnet interface started on interface %s %d",
194
              url.psz_host, url.i_port );
195

196 197
    p_intf->p_sys->i_clients   = 0;
    p_intf->p_sys->clients     = NULL;
198 199
    p_intf->p_sys->mediatheque = mediatheque;
    p_intf->pf_run = Run;
200

201
    vlc_UrlClean( &url );
202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217
    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 );
218
        free( cl->buffer_write );
219 220
        free( cl );
    }
221
    free( p_sys->clients );
222

223
    net_ListenClose( p_sys->pi_fd );
224 225 226 227 228 229 230 231 232 233 234 235

    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;
236
    char           *psz_password;
237 238 239 240
    unsigned        nlisten = 0;

    for (const int *pfd = p_sys->pi_fd; *pfd != -1; pfd++)
        nlisten++; /* How many listening sockets do we have? */
241

242
    /* FIXME: make sure config_* is cancel-safe */
243
    psz_password = var_InheritString( p_intf, "telnet-password" );
244
    vlc_cleanup_push( free, psz_password );
245

246
    for( ;; )
247
    {
248 249
        unsigned ncli = p_sys->i_clients;
        struct pollfd ufd[ncli + nlisten];
250

251
        for (unsigned i = 0; i < ncli; i++)
252 253 254
        {
            telnet_client_t *cl = p_sys->clients[i];

255
            ufd[i].fd = cl->fd;
256
            if( (cl->i_mode == WRITE_MODE_PWD) || (cl->i_mode == WRITE_MODE_CMD) )
257
                ufd[i].events = POLLOUT;
258
            else
259 260
                ufd[i].events = POLLIN;
            ufd[i].revents = 0;
261 262
        }

263
        for (unsigned i = 0; i < nlisten; i++)
264
        {
265 266 267
            ufd[ncli + i].fd = p_sys->pi_fd[i];
            ufd[ncli + i].events = POLLIN;
            ufd[ncli + i].revents = 0;
268
        }
269

270
        switch (poll (ufd, sizeof (ufd) / sizeof (ufd[0]), -1))
271
        {
272 273 274 275
            case -1:
                if (net_errno != EINTR)
                {
                    msg_Err (p_intf, "network poll error");
Pierre's avatar
Pierre committed
276 277 278 279 280
#ifndef WIN32
                    pause ();  /* We are screwed! */
#else
                    abort (); /* We are even more screwed! (no pause() in win32) */
#endif
281
                    break;
282 283 284
                }
            case 0:
                continue;
285 286
        }

287
        int canc = vlc_savecancel ();
288
        /* check if there is something to do with the socket */
289
        for (unsigned i = 0; i < ncli; i++)
290 291 292
        {
            telnet_client_t *cl = p_sys->clients[i];

293
            if (ufd[i].revents & (POLLERR|POLLHUP))
294
            {
295 296 297 298
            drop:
                net_Close( cl->fd );
                TAB_REMOVE( p_intf->p_sys->i_clients ,
                            p_intf->p_sys->clients , cl );
299
                free( cl->buffer_write );
300 301 302 303 304 305 306 307
                free( cl );
                continue;
            }

            if (ufd[i].revents & POLLOUT && (cl->i_buffer_write > 0))
            {
                ssize_t i_len;

308 309
                i_len = send( cl->fd, cl->p_buffer_write ,
                              cl->i_buffer_write, 0 );
310 311 312 313 314 315
                if( i_len > 0 )
                {
                    cl->p_buffer_write += i_len;
                    cl->i_buffer_write -= i_len;
                }
            }
316
            if (ufd[i].revents & POLLIN)
317
            {
318 319
                bool end = false;
                ssize_t i_recv;
320

321 322
                while( ((i_recv=recv( cl->fd, cl->p_buffer_read, 1, 0 )) > 0) &&
                       ((cl->p_buffer_read - cl->buffer_read) < 999) )
323 324 325
                {
                    switch( cl->i_tel_cmd )
                    {
326
                    case 0:
327
                        switch( *(uint8_t *)cl->p_buffer_read )
328 329 330 331 332
                        {
                        case '\r':
                            break;
                        case '\n':
                            *cl->p_buffer_read = '\n';
333
                            end = true;
334 335 336 337
                            break;
                        case TEL_IAC: // telnet specific command
                            cl->i_tel_cmd = 1;
                            cl->p_buffer_read++;
338
                            break;
339 340
                        default:
                            cl->p_buffer_read++;
341
                            break;
342 343 344
                        }
                        break;
                    case 1:
345
                        switch( *(uint8_t *)cl->p_buffer_read )
346 347 348 349 350 351 352
                        {
                        case TEL_WILL: case TEL_WONT:
                        case TEL_DO: case TEL_DONT:
                            cl->i_tel_cmd++;
                            cl->p_buffer_read++;
                            break;
                        default:
353
                            cl->i_tel_cmd = 0;
354
                            cl->p_buffer_read--;
355
                            break;
356 357 358 359 360
                        }
                        break;
                    case 2:
                        cl->i_tel_cmd = 0;
                        cl->p_buffer_read -= 2;
361 362
                        break;
                    }
363

364
                    if( end ) break;
365 366
                }

367
                if( (cl->p_buffer_read - cl->buffer_read) == 999 )
368
                {
369
                    Write_message( cl, NULL, "Line too long\r\n",
370
                                   cl->i_mode + 2 );
371
                }
372

373 374 375 376 377 378
#ifdef WIN32
                if( i_recv <= 0 && WSAGetLastError() == WSAEWOULDBLOCK )
                {
                    errno = EAGAIN;
                }
#endif
379
                if( i_recv == 0 || ( i_recv == -1 && ( end || errno != EAGAIN ) ) )
380
                    goto drop;
381 382 383 384
            }
        }

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

389
            if( cl->i_mode >= WRITE_MODE_PWD && cl->i_buffer_write == 0 )
390
            {
391
               // we have finished to send
392 393
               cl->i_mode -= 2; // corresponding READ MODE
            }
394 395
            else if( cl->i_mode == READ_MODE_PWD &&
                     *cl->p_buffer_read == '\n' )
396 397
            {
                *cl->p_buffer_read = '\0';
398
                if( !psz_password || !strcmp( psz_password, cl->buffer_read ) )
399
                {
400 401
                    Write_message( cl, NULL, "\xff\xfc\x01\r\nWelcome, "
                                   "Master\r\n> ", WRITE_MODE_CMD );
402 403 404 405
                }
                else
                {
                    /* wrong password */
406
                    Write_message( cl, NULL,
407
                                   "\r\nWrong password.\r\nPassword: ",
408
                                   WRITE_MODE_PWD );
409 410
                }
            }
411 412
            else if( cl->i_mode == READ_MODE_CMD &&
                     *cl->p_buffer_read == '\n' )
413 414 415 416 417 418
            {
                /* 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 ) )
                {
419
                    net_Close( cl->fd );
420 421
                    TAB_REMOVE( p_intf->p_sys->i_clients ,
                                p_intf->p_sys->clients , cl );
422
                    free( cl->buffer_write );
423 424 425 426 427
                    free( cl );
                }
                else if( !strncmp( cl->buffer_read, "shutdown", 8 ) )
                {
                    msg_Err( p_intf, "shutdown requested" );
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
428
                    libvlc_Quit( p_intf->p_libvlc );
429
                }
430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455
                else if( *cl->buffer_read == '@'
                          && strchr( cl->buffer_read, ' ' ) )
                {
                    /* Module specific commands (use same syntax as in the
                     * rc interface) */
                    char *psz_name = cl->buffer_read + 1;
                    char *psz_cmd, *psz_arg, *psz_msg;
                    int i_ret;

                    psz_cmd = strchr( cl->buffer_read, ' ' );
                    *psz_cmd = '\0';  psz_cmd++;
                    if( ( psz_arg = strchr( psz_cmd, '\n' ) ) ) *psz_arg = '\0';
                    if( ( psz_arg = strchr( psz_cmd, '\r' ) ) ) *psz_arg = '\0';
                    if( ( psz_arg = strchr( psz_cmd, ' ' ) )
                        && *psz_arg )
                    {
                        *psz_arg = '\0';
                        psz_arg++;
                    }

                    i_ret = var_Command( p_intf, psz_name, psz_cmd, psz_arg,
                                         &psz_msg );

                    if( psz_msg )
                    {
                        vlm_message_t *message;
456
                        message = vlm_MessageNew( "Module command", "%s", psz_msg );
457 458 459 460 461
                        Write_message( cl, message, NULL, WRITE_MODE_CMD );
                        vlm_MessageDelete( message );
                        free( psz_msg );
                    }
                }
462 463
                else
                {
464
                    vlm_message_t *message;
465 466 467 468

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

469 470
                    vlm_ExecuteCommand( p_sys->mediatheque, cl->buffer_read,
                                        &message );
471 472 473
                    if( !strncmp( cl->buffer_read, "help", 4 ) )
                    {
                        vlm_message_t *p_my_help =
474
                            vlm_MessageSimpleNew( "Telnet Specific Commands:" );
475
                        vlm_MessageAdd( p_my_help,
476
                            vlm_MessageSimpleNew( "logout, quit, exit" ) );
477
                        vlm_MessageAdd( p_my_help,
478
                            vlm_MessageSimpleNew( "shutdown" ) );
479
                        vlm_MessageAdd( p_my_help,
480
                            vlm_MessageSimpleNew( "@moduleinstance command argument" ) );
481 482
                        vlm_MessageAdd( message, p_my_help );
                    }
483
                    Write_message( cl, message, NULL, WRITE_MODE_CMD );
484
                    vlm_MessageDelete( message );
485 486 487
                }
            }
        }
488 489 490 491 492 493 494 495 496

        /* handle new connections */
        for (unsigned i = 0; i < nlisten; i++)
        {
            int fd;

            if (ufd[ncli + i].revents == 0)
                continue;

497
            fd = net_AcceptSingle (VLC_OBJECT(p_intf), ufd[ncli + i].fd);
498 499 500
            if (fd == -1)
                continue;

Rémi Duraffort's avatar
Rémi Duraffort committed
501
            telnet_client_t *cl = calloc( 1, sizeof( telnet_client_t ));
502 503 504 505 506 507 508 509 510 511 512 513 514 515
            if (cl == NULL)
            {
                net_Close (fd);
                continue;
            }

            cl->i_tel_cmd = 0;
            cl->fd = fd;
            cl->buffer_write = NULL;
            cl->p_buffer_write = cl->buffer_write;
            Write_message( cl, NULL,
                           "Password: \xff\xfb\x01" , WRITE_MODE_PWD );
            TAB_APPEND( p_sys->i_clients, p_sys->clients, cl );
        }
516
        vlc_restorecancel( canc );
517
    }
518
    vlc_cleanup_run ();
519 520
}

521
static void Write_message( telnet_client_t *client, vlm_message_t *message,
Laurent Aimar's avatar
Laurent Aimar committed
522
                           const char *string_message, int i_mode )
523
{
524 525
    char *psz_message;

526 527
    client->p_buffer_read = client->buffer_read;
    (client->p_buffer_read)[0] = 0; // if (cl->p_buffer_read)[0] = '\n'
528
    free( client->buffer_write );
529 530

    /* generate the psz_message string */
531
    if( message )
532
    {
533 534
        /* ok, look for vlm_message_t */
        psz_message = MessageToString( message, 0 );
535
    }
536
    else
537
    {
538
        /* it is a basic string_message */
539 540 541
        psz_message = strdup( string_message );
    }

542
    client->buffer_write = client->p_buffer_write = psz_message;
543
    client->i_buffer_write = strlen( psz_message );
544 545 546
    client->i_mode = i_mode;
}

547 548 549
/* 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 )
550
{
551 552 553
#define STRING_CR "\r\n"
#define STRING_TAIL "> "

554
    char *psz_message;
555
    int i, i_message = sizeof( STRING_TAIL );
556

557
    if( !message || !message->psz_name )
558
    {
559
        return strdup( STRING_CR STRING_TAIL );
560
    }
561
    else if( !i_level && !message->i_child && !message->psz_value  )
562
    {
563
        /* A command is successful. Don't write anything */
564
        return strdup( /*STRING_CR*/ STRING_TAIL );
565 566 567
    }

    i_message += strlen( message->psz_name ) + i_level * sizeof( "    " ) + 1;
568
    psz_message = xmalloc( i_message );
569
    *psz_message = 0;
570 571 572 573 574 575 576
    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 );
577
        psz_message = xrealloc( psz_message, i_message );
578 579 580
        strcat( psz_message, " : " );
        strcat( psz_message, message->psz_value );
        strcat( psz_message, STRING_CR );
581 582 583
    }
    else
    {
584
        i_message += sizeof( STRING_CR );
585
        psz_message = xrealloc( psz_message, i_message );
586 587
        strcat( psz_message, STRING_CR );
    }
588

589 590 591 592
    for( i = 0; i < message->i_child; i++ )
    {
        char *child_message =
            MessageToString( message->child[i], i_level + 1 );
593

594
        i_message += strlen( child_message );
595
        psz_message = xrealloc( psz_message, i_message );
596 597
        strcat( psz_message, child_message );
        free( child_message );
598
    }
599 600 601 602

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

    return psz_message;
603
}