const int snap_single(CamConfig *ccfg ){ const char *snapfname; int presnaps; if( !(snapfname = camconfig_query_str( ccfg, SEC_MAIN, "output_snapfile"))) return 0; presnaps = camconfig_query_def_int( ccfg, SEC_MAIN, "output_presnaps", 0 ); if( picture_single( ccfg, snapfname, presnaps ) == -1 ){ camserv_log( "snap_single", "Failed to snap picture!" ); } return 1; }
static void dump_cfg_options( CamConfig *ccfg ){ char vid_section[ 1024 ], key[ 1024 ]; const char *path, *val; ModInfo *minfo; int i, nfilters; if( video_query_active_section( ccfg, vid_section, sizeof( vid_section ))){ path = camconfig_query_str( ccfg, vid_section, "path" ); fprintf( stderr, "*** Module: \"%s\"\n", path ); if( (minfo = modinfo_query_so( path )) != NULL ){ modinfo_dump( minfo ); modinfo_destroy( minfo ); printf("\n"); } else fprintf( stderr, "Failed to dump video device; \"%s\"\n", path ); } nfilters = camconfig_query_def_int( ccfg, SEC_FILTERS, "num_filters", 0 ); for( i=0; i< nfilters; i++ ){ sprintf( key, "filter%d_section", i ); if( (val = camconfig_query_str( ccfg, SEC_FILTERS, key )) == NULL ){ fprintf( stderr, "KEY/VALUE \"%s\" not found!\n", key ); continue; } strncpy( key, val, sizeof( key ) - 1 ); key[ sizeof( key ) - 1 ] = '\0'; if( (path = camconfig_query_str( ccfg, key, "path" )) != NULL ){ if( (minfo = modinfo_query_so( path )) != NULL ){ fprintf( stderr, "*** Filter: \"%s\"\n", path ); modinfo_dump( minfo ); modinfo_destroy( minfo ); printf("\n"); } else fprintf( stderr, "Failed to dump filter: \"%s\"\n", path ); } } }
int main_loop( CamConfig *ccfg, Socket *picture_sock, char *picture_mem ){ Socket *listen_socket; SockSet *readset = NULL, *writeset = NULL; list_t *client_sockets; lnode_t *node; int cfg_listen_port, highest_fd, picture_client_ready; int num_sclients, num_clients; ClientInfo *clientinfo, *clientinfo2; if( (client_sockets = list_create( -1 )) == NULL) return -1; cfg_listen_port = camconfig_query_def_int( ccfg, SEC_SOCKET, "listen_port", CAMCONFIG_DEF_LISTEN_PORT ); if( (readset = sockset_new()) == NULL || (writeset = sockset_new()) == NULL ) { camserv_log( MODNAME, "Error allocating memory for socksets!"); if( readset ) sockset_dest( readset ); if( writeset ) sockset_dest( writeset ); list_destroy( client_sockets ); return -1; } if((listen_socket = socket_serve_tcp( NULL, cfg_listen_port, 100 )) == NULL ) { camserv_log( MODNAME, "Error setting up socket on port \"%d\". Exiting", cfg_listen_port ); list_destroy( client_sockets ); sockset_dest( readset ); sockset_dest( writeset ); return -1; } highest_fd = MAX( socket_query_fd( listen_socket ), socket_query_fd( picture_sock )); clientinfo = clientinfo_new( listen_socket ); clientinfo2 = clientinfo_new( picture_sock ); if( !clientinfo || !clientinfo2 || sockset_add_fd( readset, listen_socket, clientinfo ) == -1 || sockset_add_fd( readset, picture_sock, clientinfo2 ) == -1 ) { camserv_log( MODNAME, "Error adding initial sockets to sockset!"); sockset_dest( readset ); sockset_dest( writeset ); if( clientinfo ) clientinfo_dest( clientinfo ); if( clientinfo2 ) clientinfo_dest( clientinfo2 ); list_destroy( client_sockets ); return -1; } num_clients = 0; num_sclients = 0; picture_client_ready = 1; setup_signals(); Abort = 0; while( !Abort ){ int sel_res, i, nset_socks; void **set_socks; /* Only need to execute this if we have a streaming client */ if( (num_sclients > 0) && picture_client_ready == 1 ){ send( socket_query_fd( picture_sock ), "0", sizeof( "0" ), 0 ); picture_client_ready = 0; } sockset_reset( readset ); sockset_reset( writeset ); sel_res = sockset_select( highest_fd + 1, readset, writeset, NULL ); /* Service the event */ if( sel_res == -1 ){ camserv_log( MODNAME, "select() failure: %s", strerror( errno )); break; } else if( sel_res == 0 ){ camserv_log( MODNAME, "Unexpected select() fall through!" ); continue; } /* Readable sockets */ set_socks = sockset_query_socks( readset ); nset_socks = sockset_query_nsocks( readset ); for( i=0; i< nset_socks; i++ ){ ClientInfo *new_cinfo; clientinfo = set_socks[ i ]; if( clientinfo->socket == listen_socket ) { /* New client */ if( (new_cinfo = accept_client( listen_socket )) == NULL ) continue; if( (node = lnode_create( new_cinfo )) == NULL ){ clientinfo_dest( new_cinfo ); continue; } if( sockset_add_fd( readset, new_cinfo->socket, new_cinfo ) == -1 ){ camserv_log( MODNAME, "Failed to add socket %d to socket read set!", socket_query_fd( new_cinfo->socket )); clientinfo_dest( new_cinfo ); lnode_destroy( node ); continue; } if( socket_query_fd( new_cinfo->socket ) > highest_fd ) highest_fd = socket_query_fd( new_cinfo->socket ); list_append( client_sockets, node ); num_clients++; /* Init resource limit for this client */ new_cinfo->create_time = time( NULL ); new_cinfo->bytes = 0; new_cinfo->frames = 0; new_cinfo->max_seconds = camconfig_query_def_int( ccfg, SEC_SOCKET, "max_seconds", 0 ); new_cinfo->max_bytes = camconfig_query_def_int( ccfg, SEC_SOCKET, "max_bytes", 0 ); new_cinfo->max_frames = camconfig_query_def_int( ccfg, SEC_SOCKET, "max_frames", 0 ); /* Send fresh request for a picture */ send( socket_query_fd( picture_sock ), "0", sizeof( "0" ), 0 ); picture_client_ready = 0; /* Put this read socket on hold until the picture comes back */ sockset_hold( readset, new_cinfo->socket ); } else { char cmdbuf[ 1024 ]; int readlen; clientinfo = set_socks[ i ]; /* Regular joe client, set readable */ if( (readlen = read( socket_query_fd( clientinfo->socket), cmdbuf, sizeof( cmdbuf ) - 1)) <= 0 ) { camserv_log( MODNAME, "Closing socket: %s", socket_query_remote_name( clientinfo->socket )); if (clientinfo->client_type == CLIENT_T_BROWSER || clientinfo->client_type == CLIENT_T_PROXY) { num_sclients--; } client_remove( client_sockets, clientinfo ); sockset_del_fd( readset, clientinfo->socket ); sockset_unhold_all( writeset ); sockset_del_fd( writeset, clientinfo->socket ); clientinfo_dest( clientinfo ); num_clients--; } else { if( clientinfo->socket == picture_sock ) { if( dispatch_pictaker( cmdbuf, picture_mem ) == -1 ) camserv_log( MODNAME, "Pictaker dispatch failure!"); sockset_unhold_all( writeset ); /* Release the read hold as the picture has now been taken */ sockset_unhold_all( readset ); picture_client_ready = 1; } else { /* Information from a regular client */ cmdbuf[ readlen ] = '\0'; if( clientinfo->client_type == CLIENT_T_UNINIT ) { char *preamble; int pre_size; /* Figure out what type of client we have */ if( !strncmp( cmdbuf, "GET", 3 )) { if( strstr( cmdbuf, "/singleframe" )) { clientinfo->client_type = CLIENT_T_SINGLE; } else { clientinfo->client_type = CLIENT_T_BROWSER; num_sclients++; } } else if( !strncmp( cmdbuf, "PROXY", 5 )) { clientinfo->client_type = CLIENT_T_PROXY; /* Here we are in the same state as being done writing a pic */ clientinfo->state = CINFO_STATE_PICTURE; num_sclients++; databuf_buf_set( clientinfo->writebuf, NULL, 0 ); } else clientinfo->client_type = CLIENT_T_BROWSER; if( clientinfo->client_type != CLIENT_T_PROXY ) { /* Send the initial preamble. Only now we can decide which type of preamble to send (single vs. multi-part) */ if( clientinfo->client_type == CLIENT_T_SINGLE ) preamble = get_single_preamble_text( &pre_size ); else preamble = get_multi_preamble_text( &pre_size ); databuf_buf_set( clientinfo->writebuf, preamble, pre_size ); } if( sockset_add_fd( writeset, clientinfo->socket, clientinfo ) == -1 ) { camserv_log( MODNAME, "Failed to add socket %d to write set!", socket_query_fd( clientinfo->socket )); } } } } } } if( set_socks != NULL ) free( set_socks ); /* Writable sockets */ set_socks = sockset_query_socks( writeset ); nset_socks = sockset_query_nsocks( writeset ); for( i=0; i< nset_socks; i++ ){ ClientInfo *cinfo; cinfo = set_socks[ i ]; if( cinfo->client_type == CLIENT_T_BROWSER || cinfo->client_type == CLIENT_T_SINGLE ) { int result; if( (result = write_regular_client( cinfo, writeset )) != 0 ){ /* result: 1=close requested, -1=error detected */ if( result == -1 ) camserv_log( MODNAME, "Databuf write error on socket: %s\n", socket_query_remote_name( cinfo->socket )); if (cinfo->client_type == CLIENT_T_BROWSER) { num_sclients--; } client_remove( client_sockets, cinfo ); sockset_del_fd( readset, cinfo->socket ); sockset_del_fd( writeset, cinfo->socket ); clientinfo_dest( cinfo ); num_clients--; } } else { if( write_proxy_client( cinfo, writeset ) == -1 ){ camserv_log( MODNAME, "Databuf write error on socket: %d", socket_query_fd( cinfo->socket )); /* Should be proxy, but better check */ if (cinfo->client_type == CLIENT_T_PROXY) { num_sclients--; } client_remove( client_sockets, cinfo ); sockset_del_fd( readset, cinfo->socket ); sockset_del_fd( writeset, cinfo->socket ); clientinfo_dest( cinfo ); num_clients--; } } } if( set_socks != NULL ) free( set_socks ); } camserv_log( MODNAME, "Aborting."); sockset_dest( readset ); sockset_dest( writeset ); for( node = list_first( client_sockets) ; node; node=list_next( client_sockets, node )) { clientinfo_dest( node->data ); } /* Tell the picture taker to get out! Get out! */ camserv_log( MODNAME, "Closing picture taker"); send( socket_query_fd( picture_sock ), "9", sizeof( "9" ), 0 ); sleep( 3 ); camserv_log( MODNAME, "done\n"); list_destroy_nodes( client_sockets ); list_destroy( client_sockets ); socket_dest( listen_socket ); return 0; }
int main( int argc, char *argv[] ){ Socket **localsocks; CamConfig *camcfg; char *shm_segment, tmpbuf[ 1024 ], cfg_path[ MAXPATHLEN ]; int fd, shm_alloc, donecfg; extern int errno; donecfg = 0; if (argc >= 2) { strncpy( cfg_path, argv[ 1 ], sizeof( cfg_path ) ); cfg_path[ sizeof( cfg_path ) - 1 ] = '\0'; camserv_log( "main", "Trying to read config file \"%s\": ", cfg_path); if( (camcfg = read_ccfg( cfg_path )) == NULL ){ camserv_log( "main", "Error reading config \"%s\": %s", cfg_path, strerror( errno )); } else { camserv_log( "main", "Success reading config \"%s\"", cfg_path); donecfg=1; } } else { fprintf( stderr, "camserv v%s - by Jon Travis ([email protected])\n", VERSION ); fprintf( stderr, "Syntax: %s <cfg file>\n", argv[0] ); fprintf( stderr, "Will try %s/camserv.cfg\n", DATDIR); if (!donecfg) { snprintf( cfg_path, sizeof( cfg_path ), "%s/camserv.cfg", DATDIR ); cfg_path[ sizeof( cfg_path ) - 1 ] = '\0'; camserv_log( "main", "Trying to read config file \"%s\": ", cfg_path); if( (camcfg = read_ccfg( cfg_path )) == NULL ){ camserv_log( "main", "Error reading config \"%s\": %s", cfg_path, strerror( errno )); } else { camserv_log( "main", "Success reading config \"%s\"", cfg_path); donecfg=1; } } } if (!donecfg) { camserv_log( "main", "Error finding config file, exit!"); return(-1); } /* If we took a single snapshot, we are all done */ if( snap_single( camcfg )) return 0; if( (localsocks = socket_unix_pair( SOCK_DGRAM )) == NULL ){ camserv_log( "main", "Error creating communication sockets between procs"); return -1; } /* Setup a temp file for making our shm */ strcpy( tmpbuf, "/tmp/CAMSERV_XXXXXX" ); if( (fd = mkstemp( tmpbuf )) == -1 ){ camserv_log( "main", "Couldn't create temporary file: %s", tmpbuf ); strcpy( tmpbuf, argv[ 0 ] ); /* Last resort */ } else { close( fd ); } shm_alloc = camconfig_query_def_int( camcfg, SEC_MAIN, "shm_alloc", PICTURE_MALLOC ); if( shm_alloc < PICTURE_MALLOC ) camserv_log( "main", "Allocated %d bytes for SHM [RISKY RISKY!]", shm_alloc); if( (Shmid = shm_setup( tmpbuf, /* Allocate generous ammount */ shm_alloc, &shm_segment) ) == -1 ){ socket_unix_pair_dest( localsocks ); return -1; } unlink( tmpbuf ); /* Start the picture taker thread */ CPid = picture_taker( shm_segment, PICTURE_MALLOC, camcfg, localsocks[ 0 ]); if( CPid == -1 ){ /* Failure setting up camerastuffs */ camserv_log( "main", "Picture taker could not be created!"); socket_unix_pair_dest( localsocks ); return -1; } if( main_loop( camcfg, localsocks[ 1 ], shm_segment ) == -1 ){ camserv_log( "main", "Main loop exited abnormally"); socket_unix_pair_dest( localsocks ); if( CPid != -1 ) kill( CPid, SIGINT ); return -1; } socket_unix_pair_dest( localsocks ); return 0; }