//
// ZoneMinder Video Device Fixer, $Date$, $Revision$
// Copyright (C) 2003  Philip Coombes
// 
// 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.
// 

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <fcntl.h>
#include <limits.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <mysql/mysql.h>

#include "zm.h"
#include "zm_db.h"

bool fixDevice( const char *device_path )
{
	struct stat stat_buf;

	if ( stat( device_path, &stat_buf ) < 0 )
	{
		Error(( "Can't stat %s: %s", device_path, strerror(errno)));
		return( false );
	}

	uid_t uid = getuid();
	gid_t gid = getgid();

	mode_t mask = 0; 
	if ( uid == stat_buf.st_uid )
	{
		// If we are the owner
		mask = 00600;
	}
	else if ( gid == stat_buf.st_gid )
	{
		// If we are in the owner group
		mask = 00060;
	}
	else
	{
		// We are neither the owner nor in the group
		mask = 00006;
	}

	mode_t mode = stat_buf.st_mode;
	if ( (mode & mask) == mask )
	{
		Debug( 1, ( "Permissions on %s are ok at %o", device_path, mode ));
		return( true );
	}
	mode |= mask;

	Info(( "Resetting permissions on %s to %o", device_path, mode ));
	if ( chmod( device_path, mode ) < 0 )
	{
		Error(( "Can't chmod %s to %o: %s", device_path, mode, strerror(errno)));
		return( false );
	}
	return( true );
}

bool fixVideoDevice( int device )
{
	char device_path[64];

	snprintf( device_path, sizeof(device_path), "/dev/video%d", device );

	return( fixDevice( device_path ) );
}

int main( int argc, char *argv[] )
{
	char dbg_name_string[16] = "zmfix";
	zm_dbg_name = dbg_name_string;

	zmDbgInit();

	if ( argc > 1 && !strcmp( argv[1], "-a" ) )
	{
		// Do all devices we can find
		for ( int device = 0; device < 256; device++ )
		{
			if ( !fixVideoDevice( device ) )
			{
				break;
			}
		}
	}
	else
	{
		// Only do registered devices
		zmDbConnect( ZM_DB_USERA, ZM_DB_PASSA );

		static char sql[BUFSIZ];
		//snprintf( sql, sizeof(sql), "select distinct Device from Monitors where Function != 'None' and Type = 'Local'" );
		snprintf( sql, sizeof(sql), "select distinct Device from Monitors where Type = 'Local'" );
		if ( mysql_query( &dbconn, sql ) )
		{
			Error(( "Can't run query: %s", mysql_error( &dbconn ) ));
			exit( mysql_errno( &dbconn ) );
		}

		MYSQL_RES *result = mysql_store_result( &dbconn );
		if ( !result )
		{
			Error(( "Can't use query result: %s", mysql_error( &dbconn ) ));
			exit( mysql_errno( &dbconn ) );
		}

		for( int i = 0; MYSQL_ROW dbrow = mysql_fetch_row( result ); i++ )
		{
			int device = atoi(dbrow[0]);
			fixVideoDevice( device );
		}

		if ( mysql_errno( &dbconn ) )
		{
			Error(( "Can't fetch row: %s", mysql_error( &dbconn ) ));
			exit( mysql_errno( &dbconn ) );
		}
		// Yadda yadda
		mysql_free_result( result );

		if ( (bool)config.Item( ZM_OPT_X10 ) )
		{
			const char *x10_port = (const char *)config.Item( ZM_X10_DEVICE );

			if ( x10_port )
			{
				fixDevice( x10_port );
			}
		}
	}
	return( 0 );
}