+++ /dev/null
-/*
- * miracle_connection.c -- DV Connection Handler
- * Copyright (C) 2002-2003 Ushodaya Enterprises Limited
- * Author: Charles Yates <charles.yates@pandora.be>
- *
- * 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-1307, USA.
- */
-
-#ifdef HAVE_CONFIG_H
-#include <config.h>
-#endif
-
-/* System header files */
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <signal.h>
-#include <sys/types.h>
-#include <unistd.h>
-#include <time.h>
-#include <netdb.h>
-#include <sys/socket.h>
-#include <arpa/inet.h>
-
-#include <valerie/valerie_socket.h>
-
-/* Application header files */
-#include "miracle_commands.h"
-#include "miracle_connection.h"
-#include "miracle_server.h"
-#include "miracle_log.h"
-
-/** This is a generic replacement for fgets which operates on a file
- descriptor. Unlike fgets, we can also specify a line terminator. Maximum
- of (max - 1) chars can be read into buf from fd. If we reach the
- end-of-file, *eof_chk is set to 1.
-*/
-
-int fdgetline( int fd, char *buf, int max, char line_terminator, int *eof_chk )
-{
- int count = 0;
- char tmp [1];
- *eof_chk = 0;
-
- if (fd)
- while (count < max - 1) {
- if (read (fd, tmp, 1) > 0) {
- if (tmp [0] != line_terminator)
- buf [count++] = tmp [0];
- else
- break;
-
-/* Is it an EOF character (ctrl-D, i.e. ascii 4)? If so we definitely want
- to break. */
-
- if (tmp [0] == 4) {
- *eof_chk = 1;
- break;
- }
- } else {
- *eof_chk = 1;
- break;
- }
- }
-
- buf [count] = '\0';
-
- return count;
-}
-
-static int connection_initiate( int );
-static int connection_send( int, valerie_response );
-static int connection_read( int, char *, int );
-static void connection_close( int );
-
-static int connection_initiate( int fd )
-{
- int error = 0;
- valerie_response response = valerie_response_init( );
- valerie_response_set_error( response, 100, "VTR Ready" );
- error = connection_send( fd, response );
- valerie_response_close( response );
- return error;
-}
-
-static int connection_send( int fd, valerie_response response )
-{
- int error = 0;
- int index = 0;
- int code = valerie_response_get_error_code( response );
-
- if ( code != -1 )
- {
- int items = valerie_response_count( response );
-
- if ( items == 0 )
- valerie_response_set_error( response, 500, "Unknown error" );
-
- if ( code == 200 && items > 2 )
- valerie_response_set_error( response, 201, "OK" );
- else if ( code == 200 && items > 1 )
- valerie_response_set_error( response, 202, "OK" );
-
- code = valerie_response_get_error_code( response );
- items = valerie_response_count( response );
-
- for ( index = 0; !error && index < items; index ++ )
- {
- char *line = valerie_response_get_line( response, index );
- int length = strlen( line );
- if ( length == 0 && index != valerie_response_count( response ) - 1 && write( fd, " ", 1 ) != 1 )
- error = -1;
- else if ( length > 0 && write( fd, line, length ) != length )
- error = -1;
- if ( write( fd, "\r\n", 2 ) != 2 )
- error = -1;
- }
-
- if ( ( code == 201 || code == 500 ) && strcmp( valerie_response_get_line( response, items - 1 ), "" ) )
- if ( write( fd, "\r\n", 2 ) != 2 )
- miracle_log( LOG_ERR, "write(\"\\r\\n\") failed!" );
- }
- else
- {
- const char *message = "500 Empty Response\r\n\r\n";
- if ( write( fd, message, strlen( message ) ) != strlen( message ))
- miracle_log( LOG_ERR, "write(%s) failed!", message );
- }
-
- return error;
-}
-
-static int connection_read( int fd, char *command, int length )
-{
- int eof_chk;
- int nchars = fdgetline( fd, command, length, '\n', &eof_chk );
- char *cr = strchr( command, '\r');
- if ( cr != NULL )
- cr[0] = '\0';
- if ( eof_chk || strncasecmp( command, "BYE", 3 ) == 0 )
- nchars = 0;
- return nchars;
-}
-
-int connection_status( int fd, valerie_notifier notifier )
-{
- int error = 0;
- int index = 0;
- valerie_status_t status;
- char text[ 10240 ];
- valerie_socket socket = valerie_socket_init_fd( fd );
-
- for ( index = 0; !error && index < MAX_UNITS; index ++ )
- {
- valerie_notifier_get( notifier, &status, index );
- valerie_status_serialise( &status, text, sizeof( text ) );
- error = valerie_socket_write_data( socket, text, strlen( text ) ) != strlen( text );
- }
-
- while ( !error )
- {
- if ( valerie_notifier_wait( notifier, &status ) == 0 )
- {
- valerie_status_serialise( &status, text, sizeof( text ) );
- error = valerie_socket_write_data( socket, text, strlen( text ) ) != strlen( text );
- }
- else
- {
- struct timeval tv = { 0, 0 };
- fd_set rfds;
-
- FD_ZERO( &rfds );
- FD_SET( fd, &rfds );
-
- if ( select( socket->fd + 1, &rfds, NULL, NULL, &tv ) )
- error = 1;
- }
- }
-
- valerie_socket_close( socket );
-
- return error;
-}
-
-static void connection_close( int fd )
-{
- close( fd );
-}
-
-void *parser_thread( void *arg )
-{
- struct hostent *he;
- connection_t *connection = arg;
- mlt_properties owner = connection->owner;
- char address[ 512 ];
- char command[ 1024 ];
- int fd = connection->fd;
- valerie_parser parser = connection->parser;
- valerie_response response = NULL;
-
- /* Get the connecting clients ip information */
- he = gethostbyaddr( (char *) &( connection->sin.sin_addr.s_addr ), sizeof(u_int32_t), AF_INET);
- if ( he != NULL )
- strcpy( address, he->h_name );
- else
- inet_ntop( AF_INET, &( connection->sin.sin_addr.s_addr), address, 32 );
-
- miracle_log( LOG_NOTICE, "Connection established with %s (%d)", address, fd );
-
- /* Execute the commands received. */
- if ( connection_initiate( fd ) == 0 )
- {
- int error = 0;
-
- while( !error && connection_read( fd, command, 1024 ) )
- {
- response = NULL;
-
- if ( !strncmp( command, "PUSH ", 5 ) )
- {
- char temp[ 20 ];
- int bytes;
- char *buffer = NULL;
- int total = 0;
- mlt_service service = NULL;
-
- connection_read( fd, temp, 20 );
- bytes = atoi( temp );
- buffer = malloc( bytes + 1 );
- while ( total < bytes )
- {
- int count = read( fd, buffer + total, bytes - total );
- if ( count >= 0 )
- total += count;
- else
- break;
- }
- buffer[ bytes ] = '\0';
- if ( bytes > 0 && total == bytes )
- {
- if ( mlt_properties_get( owner, "push-parser-off" ) == 0 )
- {
- service = ( mlt_service )mlt_factory_producer( NULL, "westley-xml", buffer );
- mlt_events_fire( owner, "push-received", &response, command, service, NULL );
- if ( response == NULL )
- response = valerie_parser_push( parser, command, service );
- }
- else
- {
- response = valerie_parser_received( parser, command, buffer );
- }
- }
- error = connection_send( fd, response );
- valerie_response_close( response );
- mlt_service_close( service );
- free( buffer );
- }
- else if ( strncmp( command, "STATUS", 6 ) )
- {
- mlt_events_fire( owner, "command-received", &response, command, NULL );
- if ( response == NULL )
- response = valerie_parser_execute( parser, command );
- miracle_log( LOG_INFO, "%s \"%s\" %d", address, command, valerie_response_get_error_code( response ) );
- error = connection_send( fd, response );
- valerie_response_close( response );
- }
- else
- {
- error = connection_status( fd, valerie_parser_get_notifier( parser ) );
- }
- }
- }
-
- /* Free the resources associated with this connection. */
- connection_close( fd );
-
- miracle_log( LOG_NOTICE, "Connection with %s (%d) closed", address, fd );
-
- free( connection );
-
- return NULL;
-}
projects / melted / blobdiff