You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
389 lines
8.9 KiB
389 lines
8.9 KiB
/***************************************************************************
|
|
smb4k_mv - This is the move utility of Smb4K
|
|
-------------------
|
|
begin : Sa Nov 19 2005
|
|
copyright : (C) 2005-2007 by Alexander Reinholdt
|
|
email : dustpuppy@users.berlios.de
|
|
***************************************************************************/
|
|
|
|
/***************************************************************************
|
|
* 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., 51 Franklin Street, Fifth Floor, Boston, *
|
|
* MA 02110-1301 USA *
|
|
***************************************************************************/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include <config.h>
|
|
#endif
|
|
|
|
#include <locale.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
#include <getopt.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <iostream>
|
|
using namespace std;
|
|
|
|
#define SMB4K_MV_VERSION "0.4"
|
|
|
|
void info()
|
|
{
|
|
cout << "This is smb4k_mv (version " << SMB4K_MV_VERSION << "), the move utility of Smb4K" << endl;
|
|
cout << "Copyright (C) 2005-2007, Alexander Reinholdt" << endl;
|
|
cout << endl;
|
|
cout << "Usage:" << endl;
|
|
cout << " smb4k_mv {user}:{group} {perms} {src} {dest}" << endl;
|
|
cout << " smb4k_mv --help" << endl;
|
|
cout << " smb4k_mv --version" << endl;
|
|
cout << endl;
|
|
cout << "Arguments:" << endl;
|
|
cout << " {user}\tThe (new) owner of the file" << endl;
|
|
cout << endl;
|
|
cout << " {group}\tThe (new) group of the file" << endl;
|
|
cout << endl;
|
|
cout << " {perms}\tThe new permissions of the file" << endl;
|
|
cout << endl;
|
|
cout << " {src}\t\tThe (full) path of the source file" << endl;
|
|
cout << endl;
|
|
cout << " {dest}\tThe destination where the file will be moved to" << endl;
|
|
cout << endl;
|
|
cout << " --help\tDisplay this help screen and exit." << endl;
|
|
cout << " --version\tDisplay the version information and exit." << endl;
|
|
cout << endl;
|
|
}
|
|
|
|
|
|
void version()
|
|
{
|
|
cout << "Version: " << SMB4K_MV_VERSION << endl;
|
|
}
|
|
|
|
|
|
bool find_program( const char *name, char *path )
|
|
{
|
|
const char *paths[] = { "/bin/", "/sbin/", "/usr/bin/", "/usr/sbin/", "/usr/local/bin/", "/usr/local/sbin/" };
|
|
string file = "";
|
|
|
|
for ( uint i = 0; i < sizeof( paths ) / sizeof( char * ); i++ )
|
|
{
|
|
string p( paths[i] );
|
|
p.append( name );
|
|
|
|
if ( access( p.c_str(), X_OK ) == 0 )
|
|
{
|
|
file.assign( p );
|
|
break;
|
|
}
|
|
}
|
|
|
|
if ( !strcmp( file.c_str(), "" ) )
|
|
{
|
|
cerr << "smb4k_mv: Could not find " << name << " binary" << endl;
|
|
|
|
return false;
|
|
}
|
|
|
|
int len = strlen( file.c_str() ) + 1;
|
|
(void) strncpy( path, file.c_str(), len );
|
|
path[len-1] = '\0';
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
int main( int argc, char *argv[], char *envp[] )
|
|
{
|
|
(void) setlocale( LC_ALL, "" );
|
|
|
|
if ( argc < 2 )
|
|
{
|
|
info();
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
char *user_group = NULL;
|
|
char *permissions = NULL;
|
|
char *source = NULL;
|
|
char *destination = NULL;
|
|
|
|
int c;
|
|
|
|
while ( 1 )
|
|
{
|
|
int option_index = 0;
|
|
|
|
static struct option long_options[] =
|
|
{
|
|
{ "help", 0, 0, 0 },
|
|
{ "version", 0, 0, 0 },
|
|
{ 0, 0, 0, 0 }
|
|
};
|
|
|
|
c = getopt_long( argc, argv, "", long_options, &option_index );
|
|
|
|
if ( c == -1 )
|
|
{
|
|
break;
|
|
}
|
|
|
|
switch ( c )
|
|
{
|
|
case 0:
|
|
{
|
|
// Get the length of the option string, create a
|
|
// new string and copy the option into it:
|
|
int len = strlen( long_options[option_index].name ) + 1;
|
|
char opt[len];
|
|
opt[0] = '\0';
|
|
(void) strncpy( opt, long_options[option_index].name, len );
|
|
opt[len-1] = '\0';
|
|
|
|
// Now check which option has been provided:
|
|
if ( !strncmp( opt, "help", len ) )
|
|
{
|
|
info();
|
|
|
|
// That's it. Exit here.
|
|
exit( EXIT_SUCCESS );
|
|
}
|
|
else if ( !strncmp( opt, "version", len ) )
|
|
{
|
|
version();
|
|
|
|
// That's it. Exit here.
|
|
exit( EXIT_SUCCESS );
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
|
|
break;
|
|
}
|
|
case '?':
|
|
{
|
|
// Abort the program if an unknown option
|
|
// is encountered:
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
default:
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if ( optind < argc )
|
|
{
|
|
while ( optind < argc )
|
|
{
|
|
if ( optind == 1 )
|
|
{
|
|
if ( strchr( argv[optind], ':' ) )
|
|
{
|
|
int len = strlen( argv[optind] ) + 1;
|
|
user_group = new char[len];
|
|
user_group[0] = '\0';
|
|
(void) strncpy( user_group, argv[optind], len );
|
|
user_group[len-1] = '\0';
|
|
}
|
|
else
|
|
{
|
|
cerr << "smb4k_mv: First argument must contain a colon" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
}
|
|
else if ( optind == 2 )
|
|
{
|
|
int i = 0;
|
|
|
|
while ( argv[optind][i] )
|
|
{
|
|
if ( !isdigit( argv[optind][i] ) )
|
|
{
|
|
cerr << "smb4k_mv: Second argument must only contain digits" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
i++;
|
|
}
|
|
|
|
int len = strlen( argv[optind] ) + 1;
|
|
permissions = new char[len];
|
|
permissions[0] = '\0';
|
|
(void) strncpy( permissions, argv[optind], len );
|
|
permissions[len-1] = '\0';
|
|
}
|
|
else if ( optind == 3 )
|
|
{
|
|
// We only want regular files to be moved:
|
|
struct stat file_stat;
|
|
|
|
if ( lstat( argv[optind], &file_stat ) == -1 )
|
|
{
|
|
int err = errno;
|
|
cerr << "smb4k_mv: " << strerror( err ) << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
if ( !S_ISREG( file_stat.st_mode ) ||
|
|
S_ISFIFO( file_stat.st_mode ) ||
|
|
S_ISLNK( file_stat.st_mode ) )
|
|
{
|
|
cerr << "smb4k_mv: '" << argv[optind] << "' is not a regular file" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
int len = strlen( argv[optind] ) + 1;
|
|
source = new char[len];
|
|
source[0] = '\0';
|
|
(void) strncpy( source, argv[optind], len );
|
|
source[len-1] = '\0';
|
|
}
|
|
else if ( optind == 4 )
|
|
{
|
|
// Be sure that the destination either does not
|
|
// exist or that it is a regular file:
|
|
struct stat file_stat;
|
|
bool file_exists = true;
|
|
|
|
if ( lstat( argv[optind], &file_stat ) == -1 )
|
|
{
|
|
int err = errno;
|
|
|
|
// If the file does not exist, that's very good
|
|
// for our purposes. Do not error out in that case.
|
|
if ( err != ENOENT )
|
|
{
|
|
cerr << "smb4k_mv: " << strerror( err ) << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
else
|
|
{
|
|
file_exists = false;
|
|
}
|
|
}
|
|
|
|
if ( file_exists )
|
|
{
|
|
if ( !S_ISREG( file_stat.st_mode ) ||
|
|
S_ISFIFO( file_stat.st_mode ) ||
|
|
S_ISLNK( file_stat.st_mode ) )
|
|
{
|
|
cerr << "smb4k_mv: '" << argv[optind] << "' is not a regular file" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
}
|
|
|
|
int len = strlen( argv[optind] ) + 1;
|
|
destination = new char[len];
|
|
destination[0] = '\0';
|
|
(void) strncpy( destination, argv[optind], len );
|
|
destination[len-1] = '\0';
|
|
}
|
|
else
|
|
{
|
|
cerr << "smb4k_mv: There are too many arguments" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
optind++;
|
|
}
|
|
}
|
|
|
|
int i;
|
|
|
|
// Change owner and group:
|
|
char chown_prog[255];
|
|
chown_prog[0] = '\0';
|
|
|
|
if ( !find_program( "chown", chown_prog ) )
|
|
{
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
char *chown_argv[] = { chown_prog, user_group, source, NULL };
|
|
|
|
if ( fork() == 0 )
|
|
{
|
|
if ( (i = execve( chown_argv[0], chown_argv, envp )) == -1 )
|
|
{
|
|
int err = errno;
|
|
cerr << "smb4k_mv: " << strerror( err ) << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
wait( &i );
|
|
}
|
|
|
|
// Apply new permissions:
|
|
char chmod_prog[255];
|
|
chmod_prog[0] = '\0';
|
|
|
|
if ( !find_program( "chmod", chmod_prog ) )
|
|
{
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
char *chmod_argv[] = { chmod_prog, permissions, source, NULL };
|
|
|
|
if ( fork() == 0 )
|
|
{
|
|
if ( (i = execve( chmod_argv[0], chmod_argv, envp )) == -1 )
|
|
{
|
|
int err = errno;
|
|
cerr << "smb4k_mv: " << strerror( err ) << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
wait( &i );
|
|
}
|
|
|
|
// Now, finally, move the file:
|
|
char mv_prog[255];
|
|
mv_prog[0] = '\0';
|
|
|
|
if ( !find_program( "mv", mv_prog ) )
|
|
{
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
char *mv_argv[] = { mv_prog, source, destination, NULL };
|
|
|
|
if ( fork() == 0 )
|
|
{
|
|
if ( (i = execve( mv_argv[0], mv_argv, envp )) == -1 )
|
|
{
|
|
int err = errno;
|
|
cerr << "smb4k_mv: " << strerror( err ) << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
wait( &i );
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|