Removes trailing spaces. Removes tabs.
[vlc.git] / src / network / acl.c
1 /*****************************************************************************
2  * acl.c:
3  *****************************************************************************
4  * Copyright © 2005-2007 Rémi Denis-Courmont
5  * $Id$
6  *
7  * Authors: Rémi Denis-Courmont <rem # videolan.org>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27 #include <vlc/vlc.h>
28
29 #include <ctype.h>
30 #include <vlc_acl.h>
31
32 #include <errno.h>
33
34 #include <vlc_network.h>
35 #include <vlc_charset.h>
36
37 /* FIXME: rwlock on acl, but libvlc doesn't implement rwlock */
38 typedef struct vlc_acl_entry_t
39 {
40     uint8_t    host[17];
41     uint8_t    i_bytes_match;
42     uint8_t    i_bits_mask;
43     vlc_bool_t b_allow;
44 } vlc_acl_entry_t;
45
46 struct vlc_acl_t
47 {
48     vlc_object_t    *p_owner;
49     unsigned         i_size;
50     vlc_acl_entry_t *p_entries;
51     vlc_bool_t       b_allow_default;
52 };
53
54 static int ACL_Resolve( vlc_object_t *p_this, uint8_t *p_bytes,
55                         const char *psz_ip )
56 {
57     struct addrinfo hints, *res;
58     int i_family;
59
60     memset (&hints, 0, sizeof (hints));
61     hints.ai_socktype = SOCK_STREAM; /* doesn't matter */
62     hints.ai_flags = AI_NUMERICHOST;
63
64     if( vlc_getaddrinfo( p_this, psz_ip, 0, &hints, &res ) )
65     {
66         msg_Err( p_this, "invalid IP address %s", psz_ip );
67         return -1;
68     }
69
70     p_bytes[16] = 0; /* avoids overflowing when i_bytes_match = 16 */
71
72     i_family = res->ai_addr->sa_family;
73     switch( i_family )
74     {
75         case AF_INET:
76         {
77             struct sockaddr_in *addr;
78
79             addr = (struct sockaddr_in *)res->ai_addr;
80             memset( p_bytes, 0, 12 );
81             memcpy( p_bytes + 12, &addr->sin_addr, 4 );
82             break;
83         }
84
85 #if defined (HAVE_GETADDRINFO) || defined (WIN32)
86         /* unfortunately many people define AF_INET6
87            though they don't have struct sockaddr_in6 */
88         case AF_INET6:
89         {
90             struct sockaddr_in6 *addr;
91
92             addr = (struct sockaddr_in6 *)res->ai_addr;
93             memcpy( p_bytes, &addr->sin6_addr, 16 );
94             break;
95         }
96 #endif
97
98         default:
99             msg_Err( p_this, "unknown address family" );
100             vlc_freeaddrinfo( res );
101             return -1;
102     }
103
104     vlc_freeaddrinfo( res );
105     return i_family;
106 }
107
108
109 /**
110  * Check if a given address passes an access control list.
111  *
112  * @param p_acl pre-existing ACL to match the address against
113  * @param psz_ip numeric IPv4/IPv6 address
114  *
115  * @return 0 if the first matching ACL entry is an access grant,
116  * 1 if the first matching ACL entry is a denial of access,
117  * -1 on error.
118  */
119 int ACL_Check( vlc_acl_t *p_acl, const char *psz_ip )
120 {
121     const vlc_acl_entry_t *p_cur, *p_end;
122     uint8_t host[17];
123
124     if( p_acl == NULL )
125         return -1;
126
127     p_cur = p_acl->p_entries;
128     p_end = p_cur + p_acl->i_size;
129
130     if( ACL_Resolve( p_acl->p_owner, host, psz_ip ) < 0 )
131         return -1;
132
133     while (p_cur < p_end)
134     {
135         unsigned i;
136
137         i = p_cur->i_bytes_match;
138         if( (memcmp( p_cur->host, host, i ) == 0)
139          && (((p_cur->host[i] ^ host[i]) & p_cur->i_bits_mask) == 0) )
140             return !p_cur->b_allow;
141
142         p_cur++;
143     }
144
145     return !p_acl->b_allow_default;
146 }
147
148 /**
149  * Adds an item to an ACL.
150  * Items are always matched in the same order as they are added.
151  */
152 int ACL_AddNet( vlc_acl_t *p_acl, const char *psz_ip, int i_len,
153                 vlc_bool_t b_allow )
154 {
155     vlc_acl_entry_t *p_ent;
156     unsigned i_size;
157     div_t d;
158     int i_family;
159
160     i_size = p_acl->i_size;
161     p_ent = (vlc_acl_entry_t *)realloc( p_acl->p_entries,
162                                         ++p_acl->i_size * sizeof( *p_ent ) );
163
164     if( p_ent == NULL )
165         return -1;
166
167     p_acl->p_entries = p_ent;
168     p_ent += i_size;
169
170     i_family = ACL_Resolve( p_acl->p_owner, p_ent->host, psz_ip );
171     if( i_family < 0 )
172     {
173         /*
174          * I'm lazy : memory space will be re-used in the next ACL_Add call...
175          * or not.
176          */
177         p_acl->i_size--;
178         return -1;
179     }
180
181     if( i_len >= 0 )
182     {
183         if( i_family == AF_INET )
184             i_len += 96;
185
186         if( i_len > 128 )
187             i_len = 128;
188         else
189         if( i_len < 0 )
190             i_len = 0;
191     }
192     else
193         i_len = 128; /* ACL_AddHost */
194
195     d = div( i_len, 8 );
196     p_ent->i_bytes_match = d.quot;
197     p_ent->i_bits_mask = 0xff << (8 - d.rem);
198
199     p_ent->b_allow = b_allow;
200     return 0;
201 }
202
203
204 /**
205  * Creates an empty ACL.
206  *
207  * @param b_allow whether to grant (VLC_TRUE) or deny (VLC_FALSE) access
208  * by default (ie if none of the ACL entries matched).
209  *
210  * @return an ACL object. NULL in case of error.
211  */
212 vlc_acl_t *__ACL_Create( vlc_object_t *p_this, vlc_bool_t b_allow )
213 {
214     vlc_acl_t *p_acl;
215
216     p_acl = (vlc_acl_t *)malloc( sizeof( *p_acl ) );
217     if( p_acl == NULL )
218         return NULL;
219
220     vlc_object_yield( p_this );
221     p_acl->p_owner = p_this;
222     p_acl->i_size = 0;
223     p_acl->p_entries = NULL;
224     p_acl->b_allow_default = b_allow;
225
226     return p_acl;
227 }
228
229
230 /**
231  * Perform a deep copy of an existing ACL.
232  *
233  * @param p_this object to attach the copy to.
234  * @param p_acl ACL object to be copied.
235  *
236  * @return a new ACL object, or NULL on error.
237  */
238 vlc_acl_t *__ACL_Duplicate( vlc_object_t *p_this, const vlc_acl_t *p_acl )
239 {
240     vlc_acl_t *p_dupacl;
241
242     if( p_acl == NULL )
243         return NULL;
244
245     p_dupacl = (vlc_acl_t *)malloc( sizeof( *p_dupacl ) );
246     if( p_dupacl == NULL )
247         return NULL;
248
249     if( p_acl->i_size )
250     {
251         p_dupacl->p_entries = (vlc_acl_entry_t *)
252             malloc( p_acl->i_size * sizeof( vlc_acl_entry_t ) );
253
254         if( p_dupacl->p_entries == NULL )
255         {
256             free( p_dupacl );
257             return NULL;
258         }
259
260         memcpy( p_dupacl->p_entries, p_acl->p_entries,
261                 p_acl->i_size * sizeof( vlc_acl_entry_t ) );
262     }
263     else
264         p_dupacl->p_entries = NULL;
265
266     vlc_object_yield( p_this );
267     p_dupacl->p_owner = p_this;
268     p_dupacl->i_size = p_acl->i_size;
269     p_dupacl->b_allow_default = p_acl->b_allow_default;
270
271     return p_dupacl;
272 }
273
274
275 /**
276  * Releases all resources associated with an ACL object.
277  */
278 void ACL_Destroy( vlc_acl_t *p_acl )
279 {
280     if( p_acl != NULL )
281     {
282         if( p_acl->p_entries != NULL )
283             free( p_acl->p_entries );
284
285         vlc_object_release( p_acl->p_owner );
286         free( p_acl );
287     }
288 }
289
290
291 /**
292  * Reads ACL entries from a file.
293  *
294  * @param p_acl ACL object in which to insert parsed entries.
295  * @param psz_patch filename from which to parse entries.
296  *
297  * @return 0 on success, -1 on error.
298  */
299 int ACL_LoadFile( vlc_acl_t *p_acl, const char *psz_path )
300 {
301     FILE *file;
302
303     if( p_acl == NULL )
304         return -1;
305
306     file = utf8_fopen( psz_path, "r" );
307     if( file == NULL )
308         return -1;
309
310     msg_Dbg( p_acl->p_owner, "find .hosts in dir=%s", psz_path );
311
312     while( !feof( file ) )
313     {
314         char line[1024], *psz_ip, *ptr;
315
316         if( fgets( line, sizeof( line ), file ) == NULL )
317         {
318             if( ferror( file ) )
319             {
320                 msg_Err( p_acl->p_owner, "error reading %s : %s\n", psz_path,
321                         strerror( errno ) );
322                 goto error;
323             }
324             continue;
325         }
326
327         /* fgets() is cool : never overflow, always nul-terminate */
328         psz_ip = line;
329
330         /* skips blanks - cannot overflow given '\0' is not space */
331         while( isspace( *psz_ip ) )
332             psz_ip++;
333
334         if( *psz_ip == '\0' ) /* empty/blank line */
335             continue;
336
337         ptr = strchr( psz_ip, '\n' );
338         if( ptr == NULL )
339         {
340             msg_Warn( p_acl->p_owner, "skipping overly long line in %s\n",
341                       psz_path);
342             do
343             {
344                 if( fgets( line, sizeof( line ), file ) == NULL )
345                 {
346                      if( ferror( file ) )
347                      {
348                          msg_Err( p_acl->p_owner, "error reading %s : %s\n",
349                                   psz_path, strerror( errno ) );
350                      }
351                      goto error;
352                 }
353             }
354             while( strchr( line, '\n' ) == NULL);
355
356             continue; /* skip unusable line */
357         }
358
359         /* skips comment-only line */
360         if( *psz_ip == '#' )
361             continue;
362
363         /* looks for first space, CR, LF, etc. or end-of-line comment */
364         /* (there is at least a linefeed) */
365         for( ptr = psz_ip; ( *ptr != '#' ) && !isspace( *ptr ); ptr++ );
366
367         *ptr = '\0';
368
369         msg_Dbg( p_acl->p_owner, "restricted to %s", psz_ip );
370
371         ptr = strchr( psz_ip, '/' );
372         if( ptr != NULL )
373             *ptr++ = '\0'; /* separate address from mask length */
374
375         if( (ptr != NULL)
376             ? ACL_AddNet( p_acl, psz_ip, atoi( ptr ), VLC_TRUE )
377             : ACL_AddHost( p_acl, psz_ip, VLC_TRUE ) )
378         {
379             msg_Err( p_acl->p_owner, "cannot add ACL from %s", psz_path );
380             goto error;
381         }
382     }
383
384     fclose( file );
385     return 0;
386
387 error:
388     fclose( file );
389     return -1;
390 }
391