net.c 37.9 KB
Newer Older
Laurent Aimar's avatar
   
Laurent Aimar committed
1
2
3
/*****************************************************************************
 * net.c:
 *****************************************************************************
4
 * Copyright (C) 2004-2005 the VideoLAN team
5
 * $Id$
Laurent Aimar's avatar
   
Laurent Aimar committed
6
7
 *
 * Authors: Laurent Aimar <fenrir@videolan.org>
8
 *          Rémi Denis-Courmont <rem # videolan.org>
Laurent Aimar's avatar
   
Laurent Aimar committed
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
 *
 * 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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
#include <stdlib.h>
#include <vlc/vlc.h>

31
#include <errno.h>
Laurent Aimar's avatar
   
Laurent Aimar committed
32
33
34
35
36
37
38
39
40

#ifdef HAVE_FCNTL_H
#   include <fcntl.h>
#endif

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

Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
41
#if defined( WIN32 ) || defined( UNDER_CE )
42
43
44
#   if defined(UNDER_CE) && defined(sockaddr_storage)
#       undef sockaddr_storage
#   endif
45
#   include <io.h>
Laurent Aimar's avatar
   
Laurent Aimar committed
46
47
#   include <winsock2.h>
#   include <ws2tcpip.h>
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
48
#   define ENETUNREACH WSAENETUNREACH
Laurent Aimar's avatar
   
Laurent Aimar committed
49
50
#else
#   include <sys/socket.h>
51
#   include <netinet/in.h>
Eric Petit's avatar
Eric Petit committed
52
53
54
#   ifdef HAVE_ARPA_INET_H
#       include <arpa/inet.h>
#   endif
55
#   include <netdb.h>
Laurent Aimar's avatar
   
Laurent Aimar committed
56
57
#endif

gbazin's avatar
gbazin committed
58
59
60
61
#ifdef HAVE_UNISTD_H
#   include <unistd.h>
#endif

Laurent Aimar's avatar
   
Laurent Aimar committed
62
63
#include "network.h"

64
65
66
67
68
69
70
71
72
73
74
75
76
77
#ifndef INADDR_ANY
#   define INADDR_ANY  0x00000000
#endif
#ifndef INADDR_NONE
#   define INADDR_NONE 0xFFFFFFFF
#endif

static int SocksNegociate( vlc_object_t *, int fd, int i_socks_version,
                           char *psz_socks_user, char *psz_socks_passwd );
static int SocksHandshakeTCP( vlc_object_t *,
                              int fd, int i_socks_version,
                              char *psz_socks_user, char *psz_socks_passwd,
                              const char *psz_host, int i_port );

78
79
80
81
82
83
84
85
86
static int net_Socket( vlc_object_t *p_this, int i_family, int i_socktype,
                       int i_protocol )
{
    int fd, i_val;

    fd = socket( i_family, i_socktype, i_protocol );
    if( fd == -1 )
    {
#if defined(WIN32) || defined(UNDER_CE)
87
88
89
        if( WSAGetLastError ( ) != WSAEAFNOSUPPORT )
            msg_Warn( p_this, "cannot create socket (%i)",
                      WSAGetLastError() );
90
#else
91
92
93
        if( errno != EAFNOSUPPORT )
            msg_Warn( p_this, "cannot create socket (%s)",
                      strerror( errno ) );
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
#endif
        return -1;
    }

        /* Set to non-blocking */
#if defined( WIN32 ) || defined( UNDER_CE )
    {
        unsigned long i_dummy = 1;
        if( ioctlsocket( fd, FIONBIO, &i_dummy ) != 0 )
            msg_Err( p_this, "cannot set socket to non-blocking mode" );
    }
#else
    if( ( ( i_val = fcntl( fd, F_GETFL, 0 ) ) < 0 ) ||
        ( fcntl( fd, F_SETFL, i_val | O_NONBLOCK ) < 0 ) )
        msg_Err( p_this, "cannot set socket to non-blocking mode (%s)",
                 strerror( errno ) );
#endif

    i_val = 1;
    setsockopt( fd, SOL_SOCKET, SO_REUSEADDR, (void *)&i_val,
                sizeof( i_val ) );

#ifdef IPV6_V6ONLY
    /*
     * Accepts only IPv6 connections on IPv6 sockets
     * (and open an IPv4 socket later as well if needed).
     * Only Linux and FreeBSD can map IPv4 connections on IPv6 sockets,
     * so this allows for more uniform handling across platforms. Besides,
     * it makes sure that IPv4 addresses will be printed as w.x.y.z rather
     * than ::ffff:w.x.y.z
     */
    if( i_family == AF_INET6 )
        setsockopt( fd, IPPROTO_IPV6, IPV6_V6ONLY, (void *)&i_val,
                    sizeof( i_val ) );
#endif

#if defined( WIN32 ) || defined( UNDER_CE )
131
132
133
# ifndef IPV6_PROTECTION_LEVEL
#  define IPV6_PROTECTION_LEVEL 23
# endif
134
135
    if( i_family == AF_INET6 )
    {
136
        i_val = 30 /*PROTECTION_LEVEL_UNRESTRICTED*/;
137
138
        setsockopt( fd, IPPROTO_IPV6, IPV6_PROTECTION_LEVEL,
                   (const char*)&i_val, sizeof( i_val ) );
139
140
141
142
143
    }
#endif
    return fd;
}

Laurent Aimar's avatar
   
Laurent Aimar committed
144
145
146
147
148
/*****************************************************************************
 * __net_OpenTCP:
 *****************************************************************************
 * Open a TCP connection and return a handle
 *****************************************************************************/
hartman's avatar
   
hartman committed
149
int __net_OpenTCP( vlc_object_t *p_this, const char *psz_host, int i_port )
Laurent Aimar's avatar
   
Laurent Aimar committed
150
{
151
152
    struct addrinfo hints, *res, *ptr;
    const char      *psz_realhost;
153
154
    char            *psz_socks;
    int             i_realport, i_val, i_handle = -1;
155
    vlc_bool_t      b_unreach = VLC_FALSE;
Laurent Aimar's avatar
   
Laurent Aimar committed
156

157
158
    if( i_port == 0 )
        i_port = 80; /* historical VLC thing */
Laurent Aimar's avatar
   
Laurent Aimar committed
159

160
161
    memset( &hints, 0, sizeof( hints ) );
    hints.ai_socktype = SOCK_STREAM;
Laurent Aimar's avatar
   
Laurent Aimar committed
162

163
164
    psz_socks = var_CreateGetString( p_this, "socks" );
    if( *psz_socks && *psz_socks != ':' )
Laurent Aimar's avatar
   
Laurent Aimar committed
165
    {
166
        char *psz = strchr( psz_socks, ':' );
167
168
169
170

        if( psz )
            *psz++ = '\0';

171
        psz_realhost = psz_socks;
172
        i_realport = ( psz != NULL ) ? atoi( psz ) : 1080;
173

174
        msg_Dbg( p_this, "net: connecting to %s port %d for %s port %d",
175
                 psz_realhost, i_realport, psz_host, i_port );
176
177
178
    }
    else
    {
179
        psz_realhost = psz_host;
180
        i_realport = i_port;
181

182
        msg_Dbg( p_this, "net: connecting to %s port %d", psz_realhost,
183
                 i_realport );
184
    }
185

186
    i_val = vlc_getaddrinfo( p_this, psz_realhost, i_realport, &hints, &res );
187
    if( i_val )
Laurent Aimar's avatar
   
Laurent Aimar committed
188
    {
189
        msg_Err( p_this, "cannot resolve %s port %d : %s", psz_realhost,
190
                 i_realport, vlc_gai_strerror( i_val ) );
191
        free( psz_socks );
Laurent Aimar's avatar
   
Laurent Aimar committed
192
193
194
        return -1;
    }

195
196
197
198
    for( ptr = res; (ptr != NULL) && (i_handle == -1); ptr = ptr->ai_next )
    {
        int fd;

199
200
        fd = net_Socket( p_this, ptr->ai_family, ptr->ai_socktype,
                         ptr->ai_protocol );
201
202
203
204
205
        if( fd == -1 )
            continue;

        if( connect( fd, ptr->ai_addr, ptr->ai_addrlen ) )
        {
206
            socklen_t i_val_size = sizeof( i_val );
207
            div_t d;
208
209
            struct timeval tv;
            vlc_value_t timeout;
210

211
212
213
#if defined( WIN32 ) || defined( UNDER_CE )
            if( WSAGetLastError() != WSAEWOULDBLOCK )
            {
214
215
216
217
218
                if( WSAGetLastError () == WSAENETUNREACH )
                    b_unreach = VLC_TRUE;
                else
                    msg_Warn( p_this, "connection to %s port %d failed (%d)",
                              psz_host, i_port, WSAGetLastError( ) );
219
220
221
222
223
224
                net_Close( fd );
                continue;
            }
#else
            if( errno != EINPROGRESS )
            {
225
226
227
228
229
                if( errno == ENETUNREACH )
                    b_unreach = VLC_FALSE;
                else
                    msg_Warn( p_this, "connection to %s port %d : %s", psz_host,
                              i_port, strerror( errno ) );
230
231
232
233
234
235
236
237
                net_Close( fd );
                continue;
            }
#endif

            var_Create( p_this, "ipv4-timeout",
                        VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
            var_Get( p_this, "ipv4-timeout", &timeout );
238
239
240
241
242
243
            if( timeout.i_int < 0 )
            {
                msg_Err( p_this, "invalid negative value for ipv4-timeout" );
                timeout.i_int = 0;
            }
            d = div( timeout.i_int, 100 );
244
245
246
247

            msg_Dbg( p_this, "connection in progress" );
            do
            {
248
249
                fd_set fds;

250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
                if( p_this->b_die )
                {
                    msg_Dbg( p_this, "connection aborted" );
                    net_Close( fd );
                    vlc_freeaddrinfo( res );
                    free( psz_socks );
                    return -1;
                }

                /* Initialize file descriptor set */
                FD_ZERO( &fds );
                FD_SET( fd, &fds );

                /* We'll wait 0.1 second if nothing happens */
                tv.tv_sec = 0;
265
                tv.tv_usec = (d.quot > 0) ? 100000 : (1000 * d.rem);
266
267

                i_val = select( fd + 1, NULL, &fds, NULL, &tv );
268
269
270
271
272
273
274
275
276
277

                if( d.quot <= 0 )
                {
                    msg_Dbg( p_this, "connection timed out" );
                    net_Close( fd );
                    fd = -1;
                    break;
                }

                d.quot--;
278
279
280
281
282
283
284
285
286
            }
            while( ( i_val == 0 ) || ( ( i_val < 0 ) &&
#if defined( WIN32 ) || defined( UNDER_CE )
                            ( WSAGetLastError() == WSAEWOULDBLOCK )
#else
                            ( errno == EINTR )
#endif
                     ) );

287
288
289
            if( fd == -1 )
                continue; /* timeout */

290
291
292
293
294
295
296
297
298
299
300
            if( i_val < 0 )
            {
                msg_Warn( p_this, "connection aborted (select failed)" );
                net_Close( fd );
                continue;
            }

#if !defined( SYS_BEOS ) && !defined( UNDER_CE )
            if( getsockopt( fd, SOL_SOCKET, SO_ERROR, (void*)&i_val,
                            &i_val_size ) == -1 || i_val != 0 )
            {
301
302
303
304
                if( i_val == ENETUNREACH )
                    b_unreach = VLC_TRUE;
                else
                {
305
#ifdef WIN32
306
307
                    msg_Warn( p_this, "connection to %s port %d failed (%d)",
                              psz_host, i_port, WSAGetLastError( ) );
308
#else
309
310
                    msg_Warn( p_this, "connection to %s port %d : %s", psz_host,
                              i_port, strerror( i_val ) );
311
#endif
312
                }
313
314
315
316
317
318
319
                net_Close( fd );
                continue;
            }
#endif
        }
        i_handle = fd; /* success! */
    }
320

321
322
    vlc_freeaddrinfo( res );

323
324
325
326
327
328
329
330
    if( i_handle == -1 )
    {
        if( b_unreach )
            msg_Err( p_this, "Host %s port %d is unreachable", psz_host,
                     i_port );
        return -1;
    }

331
    if( *psz_socks && *psz_socks != ':' )
332
333
334
335
    {
        char *psz_user = var_CreateGetString( p_this, "socks-user" );
        char *psz_pwd  = var_CreateGetString( p_this, "socks-pwd" );

336
        if( SocksHandshakeTCP( p_this, i_handle, 5, psz_user, psz_pwd,
337
338
339
                               psz_host, i_port ) )
        {
            msg_Err( p_this, "failed to use the SOCKS server" );
340
341
            net_Close( i_handle );
            i_handle = -1;
342
343
344
345
346
        }

        free( psz_user );
        free( psz_pwd );
    }
347
    free( psz_socks );
348

349
    return i_handle;
Laurent Aimar's avatar
   
Laurent Aimar committed
350
351
}

352

353
354
355
/*****************************************************************************
 * __net_ListenTCP:
 *****************************************************************************
356
357
 * Open TCP passive "listening" socket(s)
 * This function returns NULL in case of error.
358
 *****************************************************************************/
359
int *__net_ListenTCP( vlc_object_t *p_this, const char *psz_host, int i_port )
360
{
361
362
    struct addrinfo hints, *res, *ptr;
    int             i_val, *pi_handles, i_size;
363

364
365
366
    memset( &hints, 0, sizeof( hints ) );
    hints.ai_socktype = SOCK_STREAM;
    hints.ai_flags = AI_PASSIVE;
367

368
    msg_Dbg( p_this, "net: listening to %s port %d", psz_host, i_port );
369

370
    i_val = vlc_getaddrinfo( p_this, psz_host, i_port, &hints, &res );
371
    if( i_val )
372
    {
373
        msg_Err( p_this, "cannot resolve %s port %d : %s", psz_host, i_port,
374
375
                 vlc_gai_strerror( i_val ) );
        return NULL;
376
377
    }

378
379
    pi_handles = NULL;
    i_size = 1;
380

381
    for( ptr = res; ptr != NULL; ptr = ptr->ai_next )
382
    {
383
384
        int fd, *newpi;

385
386
        fd = net_Socket( p_this, ptr->ai_family, ptr->ai_socktype,
                         ptr->ai_protocol );
387
388
        if( fd == -1 )
            continue;
389

390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
        /* Bind the socket */
        if( bind( fd, ptr->ai_addr, ptr->ai_addrlen ) )
        {
#if defined(WIN32) || defined(UNDER_CE)
            msg_Warn( p_this, "cannot bind socket (%i)", WSAGetLastError( ) );
#else
            msg_Warn( p_this, "cannot bind socket (%s)", strerror( errno ) );
#endif
            net_Close( fd );
            continue;
        }
 
        /* Listen */
        if( listen( fd, 100 ) == -1 )
        {
#if defined(WIN32) || defined(UNDER_CE)
            msg_Err( p_this, "cannot bring socket in listening mode (%i)",
                     WSAGetLastError());
#else
            msg_Err( p_this, "cannot bring the socket in listening mode (%s)",
                     strerror( errno ) );
#endif
            net_Close( fd );
            continue;
        }

        newpi = (int *)realloc( pi_handles, (++i_size) * sizeof( int ) );
        if( newpi == NULL )
        {
            net_Close( fd );
            break;
        }
        else
        {
            newpi[i_size - 2] = fd;
            pi_handles = newpi;
        }
427
    }
428

Eric Petit's avatar
Eric Petit committed
429
    vlc_freeaddrinfo( res );
430

431
432
    if( pi_handles != NULL )
        pi_handles[i_size - 1] = -1;
433
    return pi_handles;
434
435
}

436
437
438
/*****************************************************************************
 * __net_Accept:
 *****************************************************************************
439
 * Accept a connection on a set of listening sockets and return it
440
 *****************************************************************************/
441
int __net_Accept( vlc_object_t *p_this, int *pi_fd, mtime_t i_wait )
442
443
444
445
446
{
    vlc_bool_t b_die = p_this->b_die, b_block = (i_wait < 0);

    while( p_this->b_die == b_die )
    {
447
448
449
450
451
452
        int i_val = -1, *pi, *pi_end;
        struct timeval timeout;
        fd_set fds_r, fds_e;

        pi = pi_fd;

453
454
455
        /* Initialize file descriptor set */
        FD_ZERO( &fds_r );
        FD_ZERO( &fds_e );
456
457
458
459
460
461
462
463
464
465
466
467

        for( pi = pi_fd; *pi != -1; pi++ )
        {
            int i_fd = *pi;

            if( i_fd > i_val )
                i_val = i_fd;

            FD_SET( i_fd, &fds_r );
            FD_SET( i_fd, &fds_e );
        }
        pi_end = pi;
468
469
470
471

        timeout.tv_sec = 0;
        timeout.tv_usec = b_block ? 500000 : i_wait;

472
473
        i_val = select( i_val + 1, &fds_r, NULL, &fds_e, &timeout );
        if( ( ( i_val < 0 ) && ( errno == EINTR ) ) || i_val == 0 )
474
        {
475
476
477
478
            if( b_block )
                continue;
            else
                return -1;
479
        }
480
        else if( i_val < 0 )
481
        {
gbazin's avatar
gbazin committed
482
#if defined(WIN32) || defined(UNDER_CE)
483
            msg_Err( p_this, "network select error (%i)", WSAGetLastError() );
484
#else
485
            msg_Err( p_this, "network select error (%s)", strerror( errno ) );
486
487
488
489
#endif
            return -1;
        }

490
        for( pi = pi_fd; *pi != -1; pi++ )
491
        {
492
493
494
495
496
497
498
499
            int i_fd = *pi;

            if( !FD_ISSET( i_fd, &fds_r ) && !FD_ISSET( i_fd, &fds_e ) )
                continue;

            i_val = accept( i_fd, NULL, 0 );
            if( i_val < 0 )
            {
gbazin's avatar
gbazin committed
500
#if defined(WIN32) || defined(UNDER_CE)
501
                msg_Err( p_this, "accept failed (%i)", WSAGetLastError() );
502
#else
503
                msg_Err( p_this, "accept failed (%s)", strerror( errno ) );
504
#endif
505
506
507
508
509
510
511
512
513
514
515
516
            }
            else
            {
                /*
                 * This round-robin trick ensures that the first sockets in
                 * pi_fd won't prevent the last ones from getting accept'ed.
                 */
                --pi_end;
                memmove( pi, pi + 1, pi_end - pi );
                *pi_end = i_fd;
                return i_val;
            }
517
518
519
520
521
522
        }
    }

    return -1;
}

Laurent Aimar's avatar
Laurent Aimar committed
523
524
525
526
527
/*****************************************************************************
 * __net_OpenUDP:
 *****************************************************************************
 * Open a UDP connection and return a handle
 *****************************************************************************/
528
529
int __net_OpenUDP( vlc_object_t *p_this, const char *psz_bind, int i_bind,
                   const char *psz_server, int i_server )
Laurent Aimar's avatar
Laurent Aimar committed
530
{
531
    vlc_value_t      v4, v6;
Laurent Aimar's avatar
Laurent Aimar committed
532
533
    void            *private;
    network_socket_t sock;
534
    module_t         *p_network = NULL;
Laurent Aimar's avatar
Laurent Aimar committed
535

gbazin's avatar
gbazin committed
536
537
538
    if( psz_server == NULL ) psz_server = "";
    if( psz_bind == NULL ) psz_bind = "";

Laurent Aimar's avatar
Laurent Aimar committed
539
540
541
542
543
544
    /* Prepare the network_socket_t structure */
    sock.psz_bind_addr   = psz_bind;
    sock.i_bind_port     = i_bind;
    sock.psz_server_addr = psz_server;
    sock.i_server_port   = i_server;
    sock.i_ttl           = 0;
545
546
547
    sock.v6only          = 0;
    sock.i_handle        = -1;

548
    msg_Dbg( p_this, "net: connecting to '[%s]:%d@[%s]:%d'",
Laurent Aimar's avatar
Laurent Aimar committed
549
             psz_server, i_server, psz_bind, i_bind );
550
551
552
553
554
555
556
557

    /* Check if we have force ipv4 or ipv6 */
    var_Create( p_this, "ipv4", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
    var_Get( p_this, "ipv4", &v4 );
    var_Create( p_this, "ipv6", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
    var_Get( p_this, "ipv6", &v6 );

    if( !v4.b_bool )
Laurent Aimar's avatar
Laurent Aimar committed
558
    {
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
        if( v6.b_bool )
            sock.v6only = 1;

        /* try IPv6 first (unless IPv4 forced) */
        private = p_this->p_private;
        p_this->p_private = (void*)&sock;
        p_network = module_Need( p_this, "network", "ipv6", VLC_TRUE );

        if( p_network != NULL )
            module_Unneed( p_this, p_network );

        p_this->p_private = private;

        /*
         * Check if the IP stack can receive IPv4 packets on IPv6 sockets.
         * If yes, then it is better to use the IPv6 socket.
         * Otherwise, if we also get an IPv4, we have to choose, so we use
         * IPv4 only.
         */
        if( ( sock.i_handle != -1 ) && ( ( sock.v6only == 0 ) || v6.b_bool ) )
            return sock.i_handle;
    }

    if( !v6.b_bool )
    {
        int fd6 = sock.i_handle;

        /* also try IPv4 (unless IPv6 forced) */
        private = p_this->p_private;
        p_this->p_private = (void*)&sock;
        p_network = module_Need( p_this, "network", "ipv4", VLC_TRUE );

        if( p_network != NULL )
            module_Unneed( p_this, p_network );

        p_this->p_private = private;

        if( fd6 != -1 )
        {
            if( sock.i_handle != -1 )
            {
                msg_Warn( p_this, "net: lame IPv6/IPv4 dual-stack present. "
                                  "Using only IPv4." );
                net_Close( fd6 );
            }
            else
                sock.i_handle = fd6;
        }
Laurent Aimar's avatar
Laurent Aimar committed
607
    }
608
609
610
611

    if( sock.i_handle == -1 )
        msg_Dbg( p_this, "net: connection to '[%s]:%d@[%s]:%d' failed",
                psz_server, i_server, psz_bind, i_bind );
Laurent Aimar's avatar
Laurent Aimar committed
612
613
614
615

    return sock.i_handle;
}

Laurent Aimar's avatar
   
Laurent Aimar committed
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
/*****************************************************************************
 * __net_Close:
 *****************************************************************************
 * Close a network handle
 *****************************************************************************/
void net_Close( int fd )
{
#ifdef UNDER_CE
    CloseHandle( (HANDLE)fd );
#elif defined( WIN32 )
    closesocket( fd );
#else
    close( fd );
#endif
}

632
633
634
635
636
637
638
639
640
641
642
643
void net_ListenClose( int *pi_fd )
{
    if( pi_fd != NULL )
    {
        int *pi;

        for( pi = pi_fd; *pi != -1; pi++ )
            net_Close( *pi );
        free( pi_fd );
    }
}

Laurent Aimar's avatar
   
Laurent Aimar committed
644
645
646
647
/*****************************************************************************
 * __net_Read:
 *****************************************************************************
 * Read from a network socket
648
 * If b_retry is true, then we repeat until we have read the right amount of
Laurent Aimar's avatar
   
Laurent Aimar committed
649
650
 * data
 *****************************************************************************/
651
652
int __net_Read( vlc_object_t *p_this, int fd, v_socket_t *p_vs,
                uint8_t *p_data, int i_data, vlc_bool_t b_retry )
Laurent Aimar's avatar
   
Laurent Aimar committed
653
654
{
    struct timeval  timeout;
655
    fd_set          fds_r, fds_e;
Laurent Aimar's avatar
   
Laurent Aimar committed
656
657
658
    int             i_recv;
    int             i_total = 0;
    int             i_ret;
659
    vlc_bool_t      b_die = p_this->b_die;
Laurent Aimar's avatar
   
Laurent Aimar committed
660
661
662
663
664

    while( i_data > 0 )
    {
        do
        {
665
            if( p_this->b_die != b_die )
Laurent Aimar's avatar
   
Laurent Aimar committed
666
667
668
669
670
            {
                return 0;
            }

            /* Initialize file descriptor set */
671
672
673
674
            FD_ZERO( &fds_r );
            FD_SET( fd, &fds_r );
            FD_ZERO( &fds_e );
            FD_SET( fd, &fds_e );
Laurent Aimar's avatar
   
Laurent Aimar committed
675
676
677
678

            /* We'll wait 0.5 second if nothing happens */
            timeout.tv_sec = 0;
            timeout.tv_usec = 500000;
679
680
681

        } while( (i_ret = select(fd + 1, &fds_r, NULL, &fds_e, &timeout)) == 0
                 || ( i_ret < 0 && errno == EINTR ) );
Laurent Aimar's avatar
   
Laurent Aimar committed
682
683
684

        if( i_ret < 0 )
        {
gbazin's avatar
gbazin committed
685
#if defined(WIN32) || defined(UNDER_CE)
686
            msg_Err( p_this, "network select error (%d)", WSAGetLastError() );
gbazin's avatar
gbazin committed
687
#else
Laurent Aimar's avatar
   
Laurent Aimar committed
688
            msg_Err( p_this, "network select error (%s)", strerror(errno) );
gbazin's avatar
gbazin committed
689
#endif
Laurent Aimar's avatar
   
Laurent Aimar committed
690
691
692
            return i_total > 0 ? i_total : -1;
        }

693
694
695
        if( ( i_recv = (p_vs != NULL)
              ? p_vs->pf_recv( p_vs->p_sys, p_data, i_data )
              : recv( fd, p_data, i_data, 0 ) ) < 0 )
Laurent Aimar's avatar
   
Laurent Aimar committed
696
        {
gbazin's avatar
gbazin committed
697
#if defined(WIN32) || defined(UNDER_CE)
698
699
700
701
702
            if( WSAGetLastError() == WSAEWOULDBLOCK )
            {
                /* only happens with p_vs (SSL) - not really an error */
            }
            else
Laurent Aimar's avatar
Laurent Aimar committed
703
704
705
706
            /* For udp only */
            /* On win32 recv() will fail if the datagram doesn't fit inside
             * the passed buffer, even though the buffer will be filled with
             * the first part of the datagram. */
zorglub's avatar
zorglub committed
707
            if( WSAGetLastError() == WSAEMSGSIZE )
Laurent Aimar's avatar
Laurent Aimar committed
708
            {
gbazin's avatar
gbazin committed
709
                msg_Err( p_this, "recv() failed. "
Laurent Aimar's avatar
Laurent Aimar committed
710
                         "Increase the mtu size (--mtu option)" );
711
                i_total += i_data;
Laurent Aimar's avatar
Laurent Aimar committed
712
            }
713
714
            else if( WSAGetLastError() == WSAEINTR ) continue;
            else msg_Err( p_this, "recv failed (%i)", WSAGetLastError() );
715
#else
716
            /* EAGAIN only happens with p_vs (TLS) and it's not an error */
717
718
            if( errno != EAGAIN )
                msg_Err( p_this, "recv failed (%s)", strerror(errno) );
719
#endif
Laurent Aimar's avatar
   
Laurent Aimar committed
720
721
            return i_total > 0 ? i_total : -1;
        }
722
723
724
725
726
        else if( i_recv == 0 )
        {
            /* Connection closed */
            b_retry = VLC_FALSE;
        }
Laurent Aimar's avatar
   
Laurent Aimar committed
727
728
729
730
731
732
733
734
735
736
737
738

        p_data += i_recv;
        i_data -= i_recv;
        i_total+= i_recv;
        if( !b_retry )
        {
            break;
        }
    }
    return i_total;
}

zorglub's avatar
zorglub committed
739
740
741
742
743
/*****************************************************************************
 * __net_ReadNonBlock:
 *****************************************************************************
 * Read from a network socket, non blocking mode (with timeout)
 *****************************************************************************/
744
745
int __net_ReadNonBlock( vlc_object_t *p_this, int fd, v_socket_t *p_vs,
                        uint8_t *p_data, int i_data, mtime_t i_wait)
zorglub's avatar
zorglub committed
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
{
    struct timeval  timeout;
    fd_set          fds_r, fds_e;
    int             i_recv;
    int             i_ret;

    /* Initialize file descriptor set */
    FD_ZERO( &fds_r );
    FD_SET( fd, &fds_r );
    FD_ZERO( &fds_e );
    FD_SET( fd, &fds_e );

    timeout.tv_sec = 0;
    timeout.tv_usec = i_wait;

    i_ret = select(fd + 1, &fds_r, NULL, &fds_e, &timeout);

    if( i_ret < 0 && errno == EINTR )
    {
        return 0;
    }
    else if( i_ret < 0 )
    {
gbazin's avatar
gbazin committed
769
#if defined(WIN32) || defined(UNDER_CE)
770
        msg_Err( p_this, "network select error (%d)", WSAGetLastError() );
gbazin's avatar
gbazin committed
771
#else
zorglub's avatar
zorglub committed
772
        msg_Err( p_this, "network select error (%s)", strerror(errno) );
gbazin's avatar
gbazin committed
773
#endif
zorglub's avatar
zorglub committed
774
775
776
777
778
779
780
781
        return -1;
    }
    else if( i_ret == 0)
    {
        return 0;
    }
    else
    {
gbazin's avatar
gbazin committed
782
783
784
#if !defined(UNDER_CE)
        if( fd == 0/*STDIN_FILENO*/ ) i_recv = read( fd, p_data, i_data ); else
#endif
785
786
        if( ( i_recv = (p_vs != NULL)
              ? p_vs->pf_recv( p_vs->p_sys, p_data, i_data )
787
              : recv( fd, p_data, i_data, 0 ) ) < 0 )
zorglub's avatar
zorglub committed
788
        {
gbazin's avatar
gbazin committed
789
#if defined(WIN32) || defined(UNDER_CE)
zorglub's avatar
zorglub committed
790
791
792
793
794
795
796
797
798
            /* For udp only */
            /* On win32 recv() will fail if the datagram doesn't fit inside
             * the passed buffer, even though the buffer will be filled with
             * the first part of the datagram. */
            if( WSAGetLastError() == WSAEMSGSIZE )
            {
                msg_Err( p_this, "recv() failed. "
                         "Increase the mtu size (--mtu option)" );
            }
799
            else msg_Err( p_this, "recv failed (%i)", WSAGetLastError() );
zorglub's avatar
zorglub committed
800
801
802
803
804
#else
            msg_Err( p_this, "recv failed (%s)", strerror(errno) );
#endif
            return -1;
        }
805
806

        return i_recv ? i_recv : -1;  /* !i_recv -> connection closed if tcp */
zorglub's avatar
zorglub committed
807
    }
808

zorglub's avatar
zorglub committed
809
810
811
812
    /* We will never be here */
    return -1;
}

zorglub's avatar
zorglub committed
813
814
815
/*****************************************************************************
 * __net_Select:
 *****************************************************************************
816
817
 * Read from several sockets (with timeout). Takes data from the first socket
 * that has some.
zorglub's avatar
zorglub committed
818
 *****************************************************************************/
819
820
int __net_Select( vlc_object_t *p_this, int *pi_fd, v_socket_t **pp_vs,
                  int i_fd, uint8_t *p_data, int i_data, mtime_t i_wait )
zorglub's avatar
zorglub committed
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
{
    struct timeval  timeout;
    fd_set          fds_r, fds_e;
    int             i_recv;
    int             i_ret;
    int             i;
    int             i_max_fd = 0;

    /* Initialize file descriptor set */
    FD_ZERO( &fds_r );
    FD_ZERO( &fds_e );

    for( i = 0 ; i < i_fd ; i++)
    {
        if( pi_fd[i] > i_max_fd ) i_max_fd = pi_fd[i];
        FD_SET( pi_fd[i], &fds_r );
        FD_SET( pi_fd[i], &fds_e );
    }

    timeout.tv_sec = 0;
    timeout.tv_usec = i_wait;

    i_ret = select( i_max_fd + 1, &fds_r, NULL, &fds_e, &timeout );

    if( i_ret < 0 && errno == EINTR )
    {
        return 0;
    }
    else if( i_ret < 0 )
    {
        msg_Err( p_this, "network select error (%s)", strerror(errno) );
        return -1;
    }
    else if( i_ret == 0 )
    {
        return 0;
    }
    else
    {
        for( i = 0 ; i < i_fd ; i++)
        {
            if( FD_ISSET( pi_fd[i], &fds_r ) )
            {
864
865
866
                i_recv = ((pp_vs != NULL) && (pp_vs[i] != NULL))
                         ? pp_vs[i]->pf_recv( pp_vs[i]->p_sys, p_data, i_data )
                         : recv( pi_fd[i], p_data, i_data, 0 );
867
                if( i_recv < 0 )
zorglub's avatar
zorglub committed
868
869
870
871
872
873
874
875
876
877
878
                {
#ifdef WIN32
                    /* For udp only */
                    /* On win32 recv() will fail if the datagram doesn't
                     * fit inside the passed buffer, even though the buffer
                     *  will be filled with the first part of the datagram. */
                    if( WSAGetLastError() == WSAEMSGSIZE )
                    {
                        msg_Err( p_this, "recv() failed. "
                             "Increase the mtu size (--mtu option)" );
                    }
879
880
                    else msg_Err( p_this, "recv failed (%i)",
                                  WSAGetLastError() );
zorglub's avatar
zorglub committed
881
#else
882
                    msg_Err( p_this, "recv failed (%s)", strerror(errno) );
zorglub's avatar
zorglub committed
883
884
885
886
887
888
889
890
891
892
893
894
895
896
#endif
                    return VLC_EGENERIC;
                }

                return i_recv;
            }
        }
    }

    /* We will never be here */
    return -1;
}


Laurent Aimar's avatar
   
Laurent Aimar committed
897
/* Write exact amount requested */
898
int __net_Write( vlc_object_t *p_this, int fd, v_socket_t *p_vs,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
899
                 const uint8_t *p_data, int i_data )
Laurent Aimar's avatar
   
Laurent Aimar committed
900
901
{
    struct timeval  timeout;
902
    fd_set          fds_w, fds_e;
Laurent Aimar's avatar
   
Laurent Aimar committed
903
904
905
906
    int             i_send;
    int             i_total = 0;
    int             i_ret;

907
908
    vlc_bool_t      b_die = p_this->b_die;

Laurent Aimar's avatar
   
Laurent Aimar committed
909
910
911
912
    while( i_data > 0 )
    {
        do
        {
913
            if( p_this->b_die != b_die )
Laurent Aimar's avatar
   
Laurent Aimar committed
914
915
916
917
918
            {
                return 0;
            }

            /* Initialize file descriptor set */
919
920
921
922
            FD_ZERO( &fds_w );
            FD_SET( fd, &fds_w );
            FD_ZERO( &fds_e );
            FD_SET( fd, &fds_e );
Laurent Aimar's avatar
   
Laurent Aimar committed
923
924
925
926

            /* We'll wait 0.5 second if nothing happens */
            timeout.tv_sec = 0;
            timeout.tv_usec = 500000;
927
928
929

        } while( (i_ret = select(fd + 1, NULL, &fds_w, &fds_e, &timeout)) == 0
                 || ( i_ret < 0 && errno == EINTR ) );
Laurent Aimar's avatar
   
Laurent Aimar committed
930
931
932

        if( i_ret < 0 )
        {
gbazin's avatar
gbazin committed
933
#if defined(WIN32) || defined(UNDER_CE)
934
            msg_Err( p_this, "network select error (%d)", WSAGetLastError() );
gbazin's avatar
gbazin committed
935
#else
Laurent Aimar's avatar
   
Laurent Aimar committed
936
            msg_Err( p_this, "network select error (%s)", strerror(errno) );
gbazin's avatar
gbazin committed
937
#endif
Laurent Aimar's avatar
   
Laurent Aimar committed
938
939
940
            return i_total > 0 ? i_total : -1;
        }

941
942
943
        if( ( i_send = (p_vs != NULL)
                       ? p_vs->pf_send( p_vs->p_sys, p_data, i_data )
                       : send( fd, p_data, i_data, 0 ) ) < 0 )
Laurent Aimar's avatar
   
Laurent Aimar committed
944
        {
945
946
947
            /* XXX With udp for example, it will issue a message if the host
             * isn't listening */
            /* msg_Err( p_this, "send failed (%s)", strerror(errno) ); */
Laurent Aimar's avatar
   
Laurent Aimar committed
948
949
950
951
952
953
954
955
956
957
            return i_total > 0 ? i_total : -1;
        }

        p_data += i_send;
        i_data -= i_send;
        i_total+= i_send;
    }
    return i_total;
}

958
char *__net_Gets( vlc_object_t *p_this, int fd, v_socket_t *p_vs )
Laurent Aimar's avatar
   
Laurent Aimar committed
959
{
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
960
961
    char *psz_line = NULL, *ptr = NULL;
    size_t  i_line = 0, i_max = 0;
Laurent Aimar's avatar
   
Laurent Aimar committed
962
963
964
965


    for( ;; )
    {
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
966
        if( i_line == i_max )
Laurent Aimar's avatar
   
Laurent Aimar committed
967
        {
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
968
969
970
            i_max += 1024;
            psz_line = realloc( psz_line, i_max );
            ptr = psz_line + i_line;
Laurent Aimar's avatar
   
Laurent Aimar committed
971
        }
972

973
        if( net_Read( p_this, fd, p_vs, (uint8_t *)ptr, 1, VLC_TRUE ) != 1 )
Laurent Aimar's avatar
   
Laurent Aimar committed
974
        {
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
975
976
977
978
979
            if( i_line == 0 )
            {
                free( psz_line );
                return NULL;
            }
Laurent Aimar's avatar
   
Laurent Aimar committed
980
981
982
            break;
        }

Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
983
984
        if ( *ptr == '\n' )
            break;
Laurent Aimar's avatar
   
Laurent Aimar committed
985

Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
986
987
        i_line++;
        ptr++;
Laurent Aimar's avatar
   
Laurent Aimar committed
988
989
    }

Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
990
991
    *ptr-- = '\0';

Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
992
    if( ( ptr >= psz_line ) && ( *ptr == '\r' ) )
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
993
994
        *ptr = '\0';

Laurent Aimar's avatar
   
Laurent Aimar committed
995
996
997
    return psz_line;
}

998
999
int net_Printf( vlc_object_t *p_this, int fd, v_socket_t *p_vs,
                const char *psz_fmt, ... )
Laurent Aimar's avatar
   
Laurent Aimar committed
1000
{
1001
    int i_ret;
Laurent Aimar's avatar
   
Laurent Aimar committed
1002
    va_list args;
1003
    va_start( args, psz_fmt );
1004
    i_ret = net_vaPrintf( p_this, fd, p_vs, psz_fmt, args );
1005
1006
1007
1008
1009
    va_end( args );

    return i_ret;
}

1010
1011
int __net_vaPrintf( vlc_object_t *p_this, int fd, v_socket_t *p_vs,
                    const char *psz_fmt, va_list args )
1012
{
Laurent Aimar's avatar
   
Laurent Aimar committed
1013
    char    *psz;
1014
    int     i_size, i_ret;
Laurent Aimar's avatar
   
Laurent Aimar committed
1015

1016
    i_size = vasprintf( &psz, psz_fmt, args );
1017
1018
    i_ret = __net_Write( p_this, fd, p_vs, (uint8_t *)psz, i_size ) < i_size
        ? -1 : i_size;
1019
1020
1021
    free( psz );

    return i_ret;
Laurent Aimar's avatar
   
Laurent Aimar committed
1022
}
1023
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
1141



/*****************************************************************************
 * SocksNegociate:
 *****************************************************************************
 * Negociate authentication with a SOCKS server.
 *****************************************************************************/
static int SocksNegociate( vlc_object_t *p_obj,
                           int fd, int i_socks_version,
                           char *psz_socks_user,
                           char *psz_socks_passwd )
{
    uint8_t buffer[128+2*256];
    int i_len;
    vlc_bool_t b_auth = VLC_FALSE;

    if( i_socks_version != 5 )
        return VLC_SUCCESS;

    /* We negociate authentication */

    if( psz_socks_user && psz_socks_passwd &&
        *psz_socks_user && *psz_socks_passwd )
        b_auth = VLC_TRUE;

    buffer[0] = i_socks_version;    /* SOCKS version */
    if( b_auth )
    {
        buffer[1] = 2;                  /* Number of methods */
        buffer[2] = 0x00;               /* - No auth required */
        buffer[3] = 0x02;               /* - USer/Password */
        i_len = 4;
    }
    else
    {
        buffer[1] = 1;                  /* Number of methods */
        buffer[2] = 0x00;               /* - No auth required */
        i_len = 3;
    }
    
    if( net_Write( p_obj, fd, NULL, buffer, i_len ) != i_len )
        return VLC_EGENERIC;
    if( net_Read( p_obj, fd, NULL, buffer, 2, VLC_TRUE ) != 2 )
        return VLC_EGENERIC;

    msg_Dbg( p_obj, "socks: v=%d method=%x", buffer[0], buffer[1] );

    if( buffer[1] == 0x00 )
    {
        msg_Dbg( p_obj, "socks: no authentication required" );
    }
    else if( buffer[1] == 0x02 )
    {
        int i_len1 = __MIN( strlen(psz_socks_user), 255 );
        int i_len2 = __MIN( strlen(psz_socks_passwd), 255 );
        msg_Dbg( p_obj, "socks: username/password authentication" );

        /* XXX: we don't support user/pwd > 255 (truncated)*/
        buffer[0] = i_socks_version;        /* Version */
        buffer[1] = i_len1;                 /* User length */
        memcpy( &buffer[2], psz_socks_user, i_len1 );
        buffer[2+i_len1] = i_len2;          /* Password length */
        memcpy( &buffer[2+i_len1+1], psz_socks_passwd, i_len2 );

        i_len = 3 + i_len1 + i_len2;

        if( net_Write( p_obj, fd, NULL, buffer, i_len ) != i_len )
            return VLC_EGENERIC;

        if( net_Read( p_obj, fd, NULL, buffer, 2, VLC_TRUE ) != 2 )
            return VLC_EGENERIC;

        msg_Dbg( p_obj, "socks: v=%d status=%x", buffer[0], buffer[1] );
        if( buffer[1] != 0x00 )
        {
            msg_Err( p_obj, "socks: authentication rejected" );
            return VLC_EGENERIC;
        }
    }
    else
    {
        if( b_auth )
            msg_Err( p_obj, "socks: unsupported authentication method %x",
                     buffer[0] );
        else
            msg_Err( p_obj, "socks: authentification needed" );
        return VLC_EGENERIC;
    }

    return VLC_SUCCESS;
}

/*****************************************************************************
 * SocksHandshakeTCP:
 *****************************************************************************
 * Open a TCP connection using a SOCKS server and return a handle (RFC 1928)
 *****************************************************************************/
static int SocksHandshakeTCP( vlc_object_t *p_obj,
                              int fd,
                              int i_socks_version,
                              char *psz_socks_user, char *psz_socks_passwd,
                              const char *psz_host, int i_port )
{
    uint8_t buffer[128+2*256];

    if( i_socks_version != 4 && i_socks_version != 5 )
    {
        msg_Warn( p_obj, "invalid socks protocol version %d", i_socks_version );
        i_socks_version = 5;
    }

    if( i_socks_version == 5 && 
        SocksNegociate( p_obj, fd, i_socks_version,
                        psz_socks_user, psz_socks_passwd ) )
        return VLC_EGENERIC;

    if( i_socks_version == 4 )
    {
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
1142
        struct addrinfo hints = { 0 }, *p_res;
1143
1144

        /* v4 only support ipv4 */
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
1145
        hints.ai_family = AF_INET;
1146
        if( vlc_getaddrinfo( p_obj, psz_host, 0, &hints, &p_res ) )
1147
1148
1149
1150
1151
            return VLC_EGENERIC;

        buffer[0] = i_socks_version;
        buffer[1] = 0x01;               /* CONNECT */
        SetWBE( &buffer[2], i_port );   /* Port */
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
1152
1153
1154
1155
        memcpy( &buffer[4],             /* Address */
                &((struct sockaddr_in *)(p_res->ai_addr))->sin_addr, 4 );
        vlc_freeaddrinfo( p_res );

1156
1157
1158
1159
1160
1161
1162
1163
1164
1165
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
1190
1191
1192
1193
1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
        buffer[8] = 0;                  /* Empty user id */

        if( net_Write( p_obj, fd, NULL, buffer, 9 ) != 9 )
            return VLC_EGENERIC;
        if( net_Read( p_obj, fd, NULL, buffer, 8, VLC_TRUE ) != 8 )
            return VLC_EGENERIC;

        msg_Dbg( p_obj, "socks: v=%d cd=%d",
                 buffer[0], buffer[1] );

        if( buffer[1] != 90 )
            return VLC_EGENERIC;
    }
    else if( i_socks_version == 5 )
    {
        int i_hlen = __MIN(strlen( psz_host ), 255);
        int i_len;

        buffer[0] = i_socks_version;    /* Version */
        buffer[1] = 0x01;               /* Cmd: connect */
        buffer[2] = 0x00;               /* Reserved */
        buffer[3] = 3;                  /* ATYP: for now domainname */

        buffer[4] = i_hlen;
        memcpy( &buffer[5], psz_host, i_hlen );
        SetWBE( &buffer[5+i_hlen], i_port );

        i_len = 5 + i_hlen + 2;


        if( net_Write( p_obj, fd, NULL, buffer, i_len ) != i_len )
            return VLC_EGENERIC;

        /* Read the header */
        if( net_Read( p_obj, fd, NULL, buffer, 5, VLC_TRUE ) != 5 )
            return VLC_EGENERIC;

        msg_Dbg( p_obj, "socks: v=%d rep=%d atyp=%d",
                 buffer[0], buffer[1], buffer[3] );

        if( buffer[1] != 0x00 )
        {
            msg_Err( p_obj, "socks: CONNECT request failed\n" );
            return VLC_EGENERIC;
        }

        /* Read the remaining bytes */
        if( buffer[3] == 0x01 )
            i_len = 4-1 + 2;
        else if( buffer[3] == 0x03 )
            i_len = buffer[4] + 2;
        else if( buffer[3] == 0x04 )
            i_len = 16-1+2;
        else 
            return VLC_EGENERIC;

        if( net_Read( p_obj, fd, NULL, buffer, i_len, VLC_TRUE ) != i_len )
            return VLC_EGENERIC;
    }

    return VLC_SUCCESS;
}
1218

Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
1219
1220
1221
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237
1238
1239
1240
1241
1242
1243
1244
1245
1246
1247
1248
1249
1250
1251
1252
1253
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
/*****************************************************************************
 * inet_pton replacement for obsolete and/or crap operating systems
 *****************************************************************************/
#ifndef HAVE_INET_PTON
int inet_pton(int af, const char *src, void *dst)
{
# ifdef WIN32
    /* As we already know, Microsoft always go its own way, so even if they do
     * provide IPv6, they don't provide the API. */
    struct sockaddr_storage addr;
    int len = sizeof( addr );

    /* Damn it, they didn't even put LPCSTR for the firs parameter!!! */
    char *workaround_for_ill_designed_api = strdup( src );
    
    if( !WSAStringToAddress( workaround_for_ill_designed_api, af, NULL,
                             (LPSOCKADDR)&addr, &len ) )
    {
        free( workaround_for_ill_designed_api );
        return -1;
    }
    free( workaround_for_ill_designed_api );

    switch( af )
    {
        case AF_INET6:
            memcpy( dst, &((struct sockaddr_in6 *)&addr)->sin6_addr, 16 );
            break;
            
        case AF_INET:
            memcpy( dst, &((struct sockaddr_in *)&addr)->sin_addr, 4 );
            break;

        default:
            WSASetLastError( WSAEAFNOSUPPORT );
            return -1;
    }
# else
    /* Assume IPv6 is not supported. */
    /* Would be safer and more simpler to use inet_aton() but it is most
     * likely not provided either. */
    uint32_t ipv4;

    if( af != AF_INET )
    {
        errno = EAFNOSUPPORT;
        return -1;
    }

    ipv4 = inet_addr( src );
    if( ipv4 == INADDR_NONE )
        return -1;

Eric Petit's avatar
Eric Petit committed
1272
    memcpy( dst, &ipv4, 4 );
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
1273
1274
1275
1276
# endif /* WIN32 */
    return 0;
}
#endif /* HAVE_INET_PTON */