/*
* Copyright (C) 2002 - 2005 Tomasz Kojm <tkojm@clamav.net>
*
* 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., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#if HAVE_CONFIG_H
#include "clamav-config.h"
#endif
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <grp.h>
#include <dirent.h>
#include <errno.h>
#include "shared.h"
#include "manager.h"
#include "others.h"
#include "options.h"
#include "treewalk.h"
#include "defaults.h"
#include "memory.h"
#include "output.h"
int treewalk(const char *dirname, struct cl_node *root, const struct passwd *user, const struct optstruct *opt, const struct cl_limits *limits, int options, unsigned int depth)
{
DIR *dd;
struct dirent *dent;
struct stat statbuf;
char *fname;
int scanret = 0, included;
unsigned int maxdepth;
struct optnode *optnode;
char *argument;
if(optl(opt, "exclude-dir")) {
argument = getfirstargl(opt, "exclude-dir", &optnode);
while(argument) {
if(match_regex(dirname, argument) == 1) {
if(!printinfected)
mprintf("%s: Excluded\n", dirname);
return 0;
}
argument = getnextargl(&optnode, "exclude-dir");
}
}
if(optl(opt, "include-dir")) {
included = 0;
argument = getfirstargl(opt, "include-dir", &optnode);
while(argument && !included) {
if(match_regex(dirname, argument) == 1) {
included = 1;
break;
}
argument = getnextargl(&optnode, "include");
}
if(!included) {
if(!printinfected)
mprintf("%s: Excluded\n", dirname);
return 0;
}
}
if(optl(opt, "max-dir-recursion"))
maxdepth = atoi(getargl(opt, "max-dir-recursion"));
else
maxdepth = 15;
if(depth > maxdepth)
return 0;
claminfo.dirs++;
depth++;
if((dd = opendir(dirname)) != NULL) {
while((dent = readdir(dd))) {
#ifndef C_INTERIX
if(dent->d_ino)
#endif
{
if(strcmp(dent->d_name, ".") && strcmp(dent->d_name, "..")) {
/* build the full name */
fname = mcalloc(strlen(dirname) + strlen(dent->d_name) + 2, sizeof(char));
sprintf(fname, "%s/%s", dirname, dent->d_name);
/* stat the file */
if(lstat(fname, &statbuf) != -1) {
if(S_ISDIR(statbuf.st_mode) && !S_ISLNK(statbuf.st_mode) && recursion) {
if(treewalk(fname, root, user, opt, limits, options, depth) == 1)
scanret++;
} else {
if(S_ISREG(statbuf.st_mode))
scanret += scanfile(fname, root, user, opt, limits, options);
}
}
free(fname);
}
}
}
} else {
if(!printinfected)
mprintf("%s: Can't open directory.\n", dirname);
return 53;
}
closedir(dd);
if(scanret)
return 1;
else
return 0;
}
int rmdirs(const char *dirname)
{
DIR *dd;
struct dirent *dent;
struct stat maind, statbuf;
char *fname;
if((dd = opendir(dirname)) != NULL) {
while(stat(dirname, &maind) != -1) {
if(!rmdir(dirname)) break;
if(errno != ENOTEMPTY && errno != EEXIST && errno != EBADF) {
mprintf("@Can't remove temporary directory %s: %s\n", dirname, strerror(errno));
closedir(dd);
return 0;
}
while((dent = readdir(dd))) {
#ifndef C_INTERIX
if(dent->d_ino)
#endif
{
if(strcmp(dent->d_name, ".") && strcmp(dent->d_name, "..")) {
fname = mcalloc(strlen(dirname) + strlen(dent->d_name) + 2, sizeof(char));
sprintf(fname, "%s/%s", dirname, dent->d_name);
/* stat the file */
if(lstat(fname, &statbuf) != -1) {
if(S_ISDIR(statbuf.st_mode) && !S_ISLNK(statbuf.st_mode)) {
if(rmdir(fname) == -1) { /* can't be deleted */
if(errno == EACCES) {
mprintf("@Can't remove some temporary directories due to access problem.\n");
closedir(dd);
return 0;
}
rmdirs(fname);
}
} else
unlink(fname);
}
free(fname);
}
}
}
rewinddir(dd);
}
} else {
if(!printinfected)
mprintf("%s: Can't open directory.\n", dirname);
return 53;
}
closedir(dd);
return 0;
}
int clamav_rmdirs(const char *dir)
{
#ifndef C_CYGWIN
struct passwd *user;
#endif
pid_t pid;
int status;
switch(pid = fork()) {
case -1:
return -1;
case 0:
#ifndef C_CYGWIN
if(!geteuid()) {
if((user = getpwnam(UNPUSER)) == NULL)
return -3;
#ifdef HAVE_SETGROUPS
if(setgroups(1, &user->pw_gid)) {
fprintf(stderr, "ERROR: setgroups() failed.\n");
return -3;
}
#endif
if(setgid(user->pw_gid)) {
fprintf(stderr, "ERROR: setgid(%d) failed.\n", (int) user->pw_gid);
return -3;
}
if(setuid(user->pw_uid)) {
fprintf(stderr, "ERROR: setuid(%d) failed.\n", (int) user->pw_uid);
return -3;
}
}
#endif
rmdirs(dir);
exit(0);
break;
default:
waitpid(pid, &status, 0);
if(WIFEXITED(status))
return 0;
else
return -2;
}
}
int fixperms(const char *dirname)
{
DIR *dd;
struct dirent *dent;
struct stat statbuf;
char *fname;
int scanret = 0;
if((dd = opendir(dirname)) != NULL) {
while((dent = readdir(dd))) {
#ifndef C_INTERIX
if(dent->d_ino)
#endif
{
if(strcmp(dent->d_name, ".") && strcmp(dent->d_name, "..")) {
/* build full name */
fname = mcalloc(strlen(dirname) + strlen(dent->d_name) + 2, sizeof(char));
sprintf(fname, "%s/%s", dirname, dent->d_name);
/* stat the file */
if(lstat(fname, &statbuf) != -1) {
if(S_ISDIR(statbuf.st_mode) && !S_ISLNK(statbuf.st_mode)) {
chmod(fname, 0700);
fixperms(fname);
} else if(S_ISREG(statbuf.st_mode))
chmod(fname, 0700);
}
free(fname);
}
}
}
} else {
if(!printinfected)
mprintf("%s: Can't open directory.\n", dirname);
return 53;
}
closedir(dd);
if(scanret)
return 1;
else
return 0;
}
int du(const char *dirname, struct s_du *n)
{
DIR *dd;
struct dirent *dent;
struct stat statbuf;
char *fname;
if((dd = opendir(dirname)) != NULL) {
while((dent = readdir(dd))) {
#ifndef C_INTERIX
if(dent->d_ino)
#endif
{
if(strcmp(dent->d_name, ".") && strcmp(dent->d_name, "..")) {
n->files++;
/* build the full name */
fname = mcalloc(strlen(dirname) + strlen(dent->d_name) + 2, sizeof(char));
sprintf(fname, "%s/%s", dirname, dent->d_name);
/* stat the file */
if(lstat(fname, &statbuf) != -1) {
if(S_ISDIR(statbuf.st_mode) && !S_ISLNK(statbuf.st_mode)) {
du(fname, n);
} else {
n->space += statbuf.st_size / 1024;
}
}
free(fname);
}
}
}
} else {
if(!printinfected)
mprintf("%s: Can't open directory.\n", dirname);
return 53;
}
closedir(dd);
return 0;
}