path: root/duser.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>
#include <limits.h>
#include <ctype.h>
#include <errno.h>
#include <tre/regex.h>
#include <getopt.h>
#include <unistd.h>
#include <fcntl.h>
#define REGEX_MAX	60
const char* list_path = "/internal/1/domotest/opt/majordomo/majordomo-1.94.3/lists/";
const char* regex_fmt = "\%s$";

typedef struct stats_t
{
	int lines;
	int files;
	int matches;
	int added;
	int deleted;
	int modified;
} stats_t;
//Global statistics struct
stats_t processed;

typedef struct record_t
{
	int index;
	int match;
	int pad1;
	char name[REGEX_MAX];
	int pad2;
	char file[PATH_MAX];
	int pad3;
} record_t;

char* basename(const char* path);
record_t* find_in_file(const char* filename, const char* needle);
int get_file_count(const char* path);
char** get_file_list(const char* path, int count);
void stats_init(stats_t *s);
int user_list(const char* needle);
int find_in_file_ex(record_t* rec);
int user_del(record_t* rec);

char* basename(const char* path)
{
	char *ptr = strrchr(path, '/');
	return ptr ? ptr + 1 : (char*)path;
}


int user_del(record_t* rec)
{
	FILE *tfp;
	FILE *fp;
	int fd;
	int verify;
	char buf[REGEX_MAX];
	char tmpfile[255];
	snprintf(tmpfile, sizeof(tmpfile), "/tmp/duser.%s.XXXXXX", basename(rec->file));
	if((fd = mkstemp(tmpfile)) < 0 || (tfp = fdopen(fd, "w+")) == NULL)
	{
		if(fd != -1)
		{
			close(fd);
			unlink(tmpfile);
		}
		fprintf(stderr, "%s: %s\n", tmpfile, strerror(errno));
		exit(1);
	}
	
	if((fp = fopen(rec->file, "r")) == NULL)
	{
		fprintf(stderr, "%s: %s\n", rec->file, strerror(errno));
		exit(1);
	}

	while(!feof(fp))
	{
		memset(buf, 0, sizeof(buf));
		fgets(buf, REGEX_MAX, fp);
		buf[strlen(buf) - 1] = '\0';
		if((strcmp(buf, rec->name)) != 0 )
		{
			buf[strlen(buf)] = '\n';
			write(fd, buf, strlen(buf));
		}
	}

	fclose(fp);
	close(fd);
//	unlink(tmpfile);

	return 0;
}
// VERIFY that the record is proper (useful for deletion of users)
int find_in_file_ex(record_t* rec)
{
	int match = -1;
	char buf[REGEX_MAX];
	FILE *fp;
	
	if((fp = fopen(rec->file, "r")) == NULL)
	{
		perror("find_in_file_ex");
		return -1;
	}
	
	while(!feof(fp))
	{
		fgets(buf, REGEX_MAX, fp);
		buf[strlen(buf) - 1] = '\0';
		if((strcmp(buf, rec->name)) == 0) 
		{
			match = 0;
			break;
		}
	}

	fclose(fp);
	return match;
}

record_t* find_in_file(const char* filename, const char* needle)
{
	regmatch_t pmatch[10];
	regex_t preg;
	record_t record, *rptr;
	rptr = &record;
	rptr->index = 0;
	rptr->match = 0;
	
	int index = 0;
	char* base = basename(filename);
	char regex[REGEX_MAX];
	FILE *fp;
	if((fp = fopen(filename, "r")) == NULL)
	{
		fprintf(stderr, "Failed to process: %s\nReason: %s\n", base, strerror(errno));
		exit(1);
	}
	snprintf(regex, REGEX_MAX, regex_fmt, needle);
	strncpy(rptr->file, filename, PATH_MAX);
	snprintf(rptr->name, REGEX_MAX, "%s", needle);
	regcomp(&preg, regex, REG_EXTENDED|REG_ICASE|REG_NOSUB);
	while(!feof(fp))
	{
		char cmp[REGEX_MAX];
		memset(cmp, 0L, REGEX_MAX);
		fgets(cmp, REGEX_MAX, fp);
		cmp[strlen(cmp)-1] = '\0';
		if((regexec(&preg, cmp, 3, pmatch, REG_NOTBOL)) == 0)
		{
			rptr->match = 1;
			rptr->index = index;
		}
		index++;
		processed.lines++;
	}

	if(rptr->match)
		return rptr;
	else
		return NULL;

}

int get_file_count(const char* path)
{
	DIR *dp;
	struct dirent *ep;
	int file_count = 0;

	if((dp = opendir(path)) == NULL)
	{
		perror("opendir");
		exit(1);
	}

	while((ep = readdir(dp)))
	{
		if(ep->d_type == DT_REG && !strstr(ep->d_name, "."))
		{
			file_count++;
		}
	}
	closedir(dp);

	return file_count;
}

void free_file_list(char** list)
{
	int i = 0;
	for( i = 0 ; list[i] != NULL ; i++ )
	{
		free(list[i]);
	}
	free(list);
}

char** get_file_list(const char* path, int count)
{
	DIR *dp;
	struct dirent *ep;
	int i = 0;

	if((dp = opendir(path)) == NULL)
	{
		perror("opendir");
		exit(1);
	}

	char **list = (char**)calloc(count, REGEX_MAX);
	if(list == NULL)
	{
		perror("calloc");
		exit(1);
	}
	while((ep = readdir(dp)))
	{
		if(ep->d_type == DT_REG && !strstr(ep->d_name, "."))
		{
			list[i] = (char*)malloc(sizeof(char) * strlen(ep->d_name)+1);
			memset(list[i], 0L, strlen(ep->d_name));
			strncpy(list[i], ep->d_name, strlen(ep->d_name));
			i++;
		}
	}
	closedir(dp);
	return list;
}

void stats_init(stats_t *s)
{
	s->lines = 0;
	s->files = 0;
	s->matches = 0;
	s->added = 0;
	s->deleted = 0;
	s->modified = 0;
}

int user_list(const char* needle)
{
	int verified = 0;
	processed.files = get_file_count(list_path);
	char** list;
	list = get_file_list(list_path, processed.files);

	printf("%20s%12s\n", "List", "At line");
	printf("\t\t%s\n", "================");
	int i = 0;
	for(i = 0 ; list[i] != NULL ; i++)
	{
		char tmp[PATH_MAX];
		sprintf(tmp, "%s%s", list_path, list[i]);
		record_t *rp;
		if((rp = find_in_file(tmp, needle)) != NULL)
		{
			verified = find_in_file_ex(rp);
			printf("%20s\t%5d -> %s(%d) -> %d\n", basename(rp->file), rp->index, verified ? "Yes" : "No", verified, user_del(rp));
			processed.matches++;
		}
	}
	free_file_list(list);	
	return 0;
}

void usage(const char* progname)
{
	printf("Domouser v0.1a - jhunk@stsci.edu\n");
	printf("Usage: %s\n", progname);
	printf("	-h		This usage statement\n");
	printf("	-a		Add a user to a list\n");
	printf("	-d		Delete a user from all lists\n");
	printf("	-l		Find and list a user in all lists\n");
	printf("\n");
	exit(0);
}

static const struct option options[] = {
	{"help",	no_argument,		NULL,	'h'},
	{"add",		required_argument,	NULL,	'a'},
	{"del",		required_argument,	NULL,	'd'},
	{"mod",		required_argument,	NULL,	'm'},
	{"list",	no_argument,		NULL,	'l'},
	{NULL,		0,					NULL,	0}
};
static const char *optstring = "hl:a:m:d";
int user_cmd_add = 0;
int user_cmd_del = 0;
int user_cmd_mod = 0;
int user_cmd_list = 0;
int user_cmd_noopt = 0;

int main(int argc, char* argv[])
{
	stats_init(&processed);
	char *needle;

	char c;
	while((c = getopt_long(argc, argv, optstring, options, NULL)) != -1)
	{
		switch(c)
		{
			case 'h':
				usage(argv[0]);
				break;
			case 'a':
				user_cmd_add = 1;
				break;
			case 'd':
				user_cmd_del = 1;
				break;
			case 'm':
				user_cmd_mod = 1;
				break;
			case 'l':
				user_cmd_list = 1;
				needle = strdup(optarg);
				break;
			case '?':
				usage(argv[0]);
				break;

			default:
				return 1;
				break;

		}
	}
	
	if(user_cmd_list || user_cmd_noopt)
	{
		user_list(needle);
	}
	

	printf("\n%d matches\t%d files\t%d lines parsed\n", 
		processed.matches, processed.files, processed.lines);
	return 0;
}