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.
571 lines
13 KiB
571 lines
13 KiB
/***************************************************************************
|
|
smb4k_umount - This is the unmount utility of Smb4K.
|
|
-------------------
|
|
begin : Sa Sep 25 2004
|
|
copyright : (C) 2004-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
|
|
|
|
#if !defined(__FreeBSD__) && !defined(__sun)
|
|
#include <sys/statfs.h>
|
|
#elif defined(__sun)
|
|
#include <sys/types.h>
|
|
#include <sys/statvfs.h>
|
|
#elif defined(__FreeBSD__)
|
|
#include <sys/param.h>
|
|
#include <sys/mount.h>
|
|
#endif
|
|
|
|
#ifdef __linux__
|
|
#include <sys/utsname.h>
|
|
#endif
|
|
|
|
#include <locale.h>
|
|
#include <unistd.h>
|
|
#include <getopt.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <iostream>
|
|
using namespace std;
|
|
|
|
#define SMB4K_UMOUNT_VERSION "0.14"
|
|
|
|
|
|
void info()
|
|
{
|
|
cout << "This is smb4k_umount (version " << SMB4K_UMOUNT_VERSION << "), the unmount utility of Smb4K" << endl;
|
|
cout << "Copyright (C) 2004-2007, Alexander Reinholdt" << endl;
|
|
cout << endl;
|
|
cout << "Usage:" << endl;
|
|
#ifndef __FreeBSD__
|
|
cout << " smb4k_umount {mode} {options} {mountpoint}" << endl;
|
|
#else
|
|
cout << " smb4k_umount {mountpoint}" << endl;
|
|
#endif
|
|
cout << " smb4k_umount --help" << endl;
|
|
cout << " smb4k_umount --version" << endl;
|
|
cout << endl;
|
|
cout << "Arguments:" << endl;
|
|
cout << " {mode}" << endl;
|
|
#ifndef __FreeBSD__
|
|
cout << " --no-suid\tsmb4k_umount is run in normal user mode, so smbumount or" << endl;
|
|
cout << "\t\tumount.cifs is invoked." << endl;
|
|
cout << " --suid\tsmb4k_umount is run in super user mode, so umount is invoked." << endl;
|
|
cout << " -n\t\tThe same as the '--no-suid' argument." << endl;
|
|
cout << " -s\t\tThe same as the '--suid' argument." << endl;
|
|
cout << endl;
|
|
cout << " {options}" << endl;
|
|
cout << " -t <fs>\tThe file system that should be used for unmounting. Only 'smbfs'" << endl;
|
|
cout << "\t\tand 'cifs' are supported. All other file systems will result in" << endl;
|
|
cout << "\t\tan error. Please note, that this argument is mandatory." << endl;
|
|
#endif
|
|
#ifdef __linux__
|
|
cout << " -l\t\tPerform a lazy unmount. See the manual page of umount for" << endl;
|
|
cout << "\t\tmore information. Please note, that this argument is only" << endl;
|
|
cout << "\t\trecognized in super user mode, i.e. if '--suid' is supplied," << endl;
|
|
cout << "\t\tand that you need kernel version 2.4.11 or later." << endl;
|
|
#endif
|
|
cout << endl;
|
|
cout << " {mountpoint}\tThe path where the share is mounted 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_UMOUNT_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_umount: Could not find " << name << " binary" << endl;
|
|
|
|
return false;
|
|
}
|
|
|
|
int len = strlen( file.c_str() ) + 1;
|
|
strncpy( path, file.c_str(), len );
|
|
path[len-1] = '\0';
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool check_filesystem( const char *path, const char *fs )
|
|
{
|
|
bool ok = false;
|
|
|
|
#if !defined(__sun)
|
|
struct statfs filesystem;
|
|
#else
|
|
struct statvfs filesystem;
|
|
#endif
|
|
|
|
#if !defined(__sun) && !defined(__irix__)
|
|
if ( statfs( path, &filesystem ) == -1 )
|
|
#elif defined(__irix__)
|
|
if ( statfs( path, &filesystem, sizeof( filesystem ), 0 ) == -1 )
|
|
#else
|
|
if ( statvfs( path, &filesystem ) == -1 )
|
|
#endif
|
|
{
|
|
int err_code = errno;
|
|
|
|
if ( err_code != EIO && err_code != EACCES )
|
|
{
|
|
// ok is still FALSE
|
|
cerr << "smb4k_umount: " << strerror( err_code ) << endl;
|
|
}
|
|
else
|
|
{
|
|
ok = true; // Bypass the check below, because it would yield ok == FALSE
|
|
// and we want to be able to unmount broken shares as well.
|
|
}
|
|
|
|
return ok;
|
|
}
|
|
|
|
#if !defined(__FreeBSD__) && !defined(__sun) && !defined(__irix__)
|
|
// First entry is for CIFS, the second for SMBFS.
|
|
if ( (uint)filesystem.f_type == 0xFF534D42 && !strncmp( fs, "cifs", strlen( fs )+1 ) )
|
|
{
|
|
ok = true;
|
|
}
|
|
else if ( (uint)filesystem.f_type == 0x517B && !strncmp( fs, "smbfs", strlen( fs )+1 ) )
|
|
{
|
|
ok = true;
|
|
}
|
|
#elif defined(__FreeBSD__)
|
|
if ( !strncmp( filesystem.f_fstypename, fs, strlen( fs ) ) )
|
|
{
|
|
ok = true;
|
|
}
|
|
#elif defined(__sun)
|
|
if ( (uint)filesystem.f_basetype == 0xFF534D42 && !strncmp( fs, "cifs", strlen( fs )+1 ) )
|
|
{
|
|
ok = true;
|
|
}
|
|
else if ( (uint)filesystem.f_basetype == 0x517B && !strncmp( fs, "smbfs", strlen( fs )+1 ) )
|
|
{
|
|
ok = true;
|
|
}
|
|
#elif defined(__irix__)
|
|
if ( (uint)filesystem.f_fstyp == 0xFF534D42 && !strncmp( fs, "cifs", strlen( fs )+1 ) )
|
|
{
|
|
ok = true;
|
|
}
|
|
else if ( (uint)filesystem.f_basetype == 0x517B && !strncmp( fs, "smbfs", strlen( fs )+1 ) )
|
|
{
|
|
ok = true;
|
|
}
|
|
#endif
|
|
else
|
|
{
|
|
// ok is still FALSE.
|
|
cerr << "smb4k_umount: Wrong file system specified" << endl;
|
|
}
|
|
|
|
return ok;
|
|
}
|
|
|
|
|
|
int main( int argc, char *argv[], char *envp[] )
|
|
{
|
|
// First of all, set the locale
|
|
(void) setlocale( LC_ALL, "" );
|
|
|
|
if ( argc < 2 )
|
|
{
|
|
info();
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
int new_argc = argc + 1;
|
|
char *new_argv[new_argc];
|
|
int index = 0;
|
|
char path[255];
|
|
path[0] = '\0';
|
|
char *mountpoint = NULL;
|
|
#ifndef __FreeBSD__
|
|
char *filesystem = NULL;
|
|
bool normal_user_mode = true;
|
|
bool have_user_mode = false;
|
|
#ifdef __linux__
|
|
bool lazy_unmount = false;
|
|
#endif
|
|
#endif
|
|
|
|
// Get the options that were passed:
|
|
int c;
|
|
|
|
while ( 1 )
|
|
{
|
|
int option_index = 0;
|
|
|
|
static struct option long_options[] =
|
|
{
|
|
{ "help", 0, 0, 0 },
|
|
{ "version", 0, 0, 0 },
|
|
#ifndef __FreeBSD__
|
|
{ "suid", 0, 0, 0 },
|
|
{ "no-suid", 0, 0, 0 },
|
|
#endif
|
|
{ 0, 0, 0, 0 }
|
|
};
|
|
#ifdef __linux__
|
|
c = getopt_long( argc, argv, "t:nsl", long_options, &option_index );
|
|
#else
|
|
#ifndef __FreeBSD__
|
|
c = getopt_long( argc, argv, "t:ns", long_options, &option_index );
|
|
#else
|
|
c = getopt_long( argc, argv, "", long_options, &option_index );
|
|
#endif
|
|
#endif
|
|
|
|
if ( c == -1 )
|
|
{
|
|
break;
|
|
}
|
|
|
|
switch ( c )
|
|
{
|
|
case 0:
|
|
{
|
|
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';
|
|
|
|
if ( !strncmp( opt, "help", len ) )
|
|
{
|
|
info();
|
|
|
|
exit( EXIT_SUCCESS );
|
|
}
|
|
else if ( !strncmp( opt, "version", len ) )
|
|
{
|
|
version();
|
|
|
|
exit( EXIT_SUCCESS );
|
|
}
|
|
#ifndef __FreeBSD__
|
|
else if ( !strncmp( opt, "suid", len ) )
|
|
{
|
|
// Enter super user mode
|
|
|
|
normal_user_mode = false;
|
|
have_user_mode = true;
|
|
|
|
break;
|
|
}
|
|
else if ( !strncmp( opt, "no-suid", len ) )
|
|
{
|
|
// Enter normal user mode
|
|
|
|
normal_user_mode = true;
|
|
have_user_mode = true;
|
|
|
|
break;
|
|
}
|
|
#endif
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
|
|
break;
|
|
}
|
|
#ifndef __FreeBSD__
|
|
case 't':
|
|
{
|
|
// Get the length of the option argument:
|
|
int len = strlen( optarg ) + 1;
|
|
|
|
if ( strncmp( "smbfs", optarg, len) != 0 &&
|
|
strncmp( "cifs", optarg, len ) != 0 )
|
|
{
|
|
cerr << "smb4k_umount: File system " << optarg << " is not supported" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
if ( !filesystem )
|
|
{
|
|
filesystem = new char[len];
|
|
filesystem[0] = '\0';
|
|
(void) strncpy( filesystem, optarg, len );
|
|
filesystem[len-1] = '\0';
|
|
}
|
|
|
|
break;
|
|
}
|
|
case 's':
|
|
{
|
|
// Enter super user mode
|
|
normal_user_mode = false;
|
|
have_user_mode = true;
|
|
|
|
break;
|
|
}
|
|
case 'n':
|
|
{
|
|
// Enter normal user mode
|
|
normal_user_mode = true;
|
|
have_user_mode = true;
|
|
|
|
break;
|
|
}
|
|
#endif
|
|
#ifdef __linux__
|
|
case 'l':
|
|
{
|
|
// Initiate a lazy unmount. The umount binary
|
|
// will complain, if '-l' is not supported.
|
|
lazy_unmount = true;
|
|
|
|
break;
|
|
}
|
|
#endif
|
|
case '?':
|
|
{
|
|
// Abort the program if an unknown option
|
|
// is encountered:
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
default:
|
|
{
|
|
break;
|
|
}
|
|
};
|
|
}
|
|
|
|
#ifndef __FreeBSD__
|
|
// Error out if no user mode was specified.
|
|
if ( !have_user_mode )
|
|
{
|
|
cerr << "smb4k_umount: No mode was specified" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
// Error out if the user did not specify any file system.
|
|
if ( !filesystem )
|
|
{
|
|
cerr << "smb4k_umount: No file system was specified" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
if ( normal_user_mode )
|
|
{
|
|
int len = strlen( filesystem ) + 1;
|
|
|
|
if ( !strncmp( "smbfs", filesystem, len ) )
|
|
{
|
|
if ( !find_program( "smbumount", path ) )
|
|
{
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if ( !find_program( "umount.cifs", path ) )
|
|
{
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
}
|
|
|
|
len = strlen( path ) + 1;
|
|
|
|
new_argv[index] = new char[len];
|
|
new_argv[index][0] = '\0';
|
|
(void) strncpy( new_argv[index], path, len );
|
|
new_argv[index][len-1] = '\0';
|
|
|
|
index++;
|
|
}
|
|
else
|
|
{
|
|
if ( !find_program( "umount", path ) )
|
|
{
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
int len = strlen( path ) + 1;
|
|
|
|
new_argv[index] = new char[len];
|
|
new_argv[index][0] = '\0';
|
|
(void) strncpy( new_argv[index], path, len );
|
|
new_argv[index][len-1] = '\0';
|
|
|
|
index++;
|
|
|
|
len = strlen( "-t" ) + 1;
|
|
|
|
new_argv[index] = new char[len];
|
|
new_argv[index][0] = '\0';
|
|
(void) strncpy( new_argv[index], "-t", len );
|
|
new_argv[index][len-1] = '\0';
|
|
|
|
index++;
|
|
|
|
len = strlen( filesystem ) + 1;
|
|
|
|
new_argv[index] = new char[len];
|
|
new_argv[index][0] = '\0';
|
|
(void) strncpy( new_argv[index], filesystem, len );
|
|
new_argv[index][len-1] = '\0';
|
|
|
|
index++;
|
|
|
|
#ifdef __linux__
|
|
// Lazy unmount?
|
|
if ( lazy_unmount )
|
|
{
|
|
len = strlen( "-l" ) + 1;
|
|
|
|
new_argv[index] = new char[len];
|
|
new_argv[index][0] = '\0';
|
|
(void) strncpy( new_argv[index], "-l", len );
|
|
new_argv[index][len-1] = '\0';
|
|
|
|
index++;
|
|
}
|
|
#endif
|
|
}
|
|
#else
|
|
// We do not need to care about the user mode and
|
|
// we also need not to check for the file system,
|
|
// since there is only one.
|
|
if ( !find_program( "umount", path ) )
|
|
{
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
int length = strlen( path ) + 1;
|
|
new_argv[index] = new char[length];
|
|
new_argv[index][0] = '\0';
|
|
(void) strncpy( new_argv[index], path, length );
|
|
new_argv[index][length-1] = '\0';
|
|
|
|
index++;
|
|
#endif
|
|
|
|
// Add the mount point:
|
|
if ( optind < argc )
|
|
{
|
|
while ( optind < argc )
|
|
{
|
|
if ( !mountpoint )
|
|
{
|
|
if ( argv[optind][0] != '\057' )
|
|
{
|
|
cerr << "smb4k_umount: Argument " << optind << " is not a mount point" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
#ifndef __FreeBSD__
|
|
if ( !check_filesystem( argv[optind], filesystem ) )
|
|
#else
|
|
if ( !check_filesystem( argv[optind], "smbfs" ) )
|
|
#endif
|
|
{
|
|
// Error message is given by check_filesystem()
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
int len = strlen( argv[optind] ) + 1;
|
|
|
|
mountpoint = new char[len];
|
|
mountpoint[0] = '\0';
|
|
(void) strncpy( mountpoint, argv[optind], len );
|
|
mountpoint[len-1] = '\0';
|
|
|
|
optind++;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if ( !mountpoint )
|
|
{
|
|
cerr << "smb4k_umount: No mount point was specified" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
else
|
|
{
|
|
int len = strlen( mountpoint ) + 1;
|
|
new_argv[index] = new char[len];
|
|
new_argv[index][0] = '\0';
|
|
(void) strncpy( new_argv[index], mountpoint, len );
|
|
new_argv[index][len-1] = '\0';
|
|
|
|
index++;
|
|
}
|
|
|
|
|
|
if ( index >= new_argc )
|
|
{
|
|
cerr << "smb4k_umount: There are too many arguments" << endl;
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
// Terminate new_argv:
|
|
new_argv[index] = NULL;
|
|
|
|
// Execute command:
|
|
if ( execve( new_argv[0], new_argv, envp ) == -1 )
|
|
{
|
|
int err = errno;
|
|
cerr << "smb4k_umount: " << strerror( err ) << endl;
|
|
|
|
exit( EXIT_FAILURE );
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|