diff options
author | Alan Brault <alan.brault@incruentatus.net> | 2011-05-02 12:07:47 -0400 |
---|---|---|
committer | Alan Brault <alan.brault@incruentatus.net> | 2011-05-02 12:10:28 -0400 |
commit | 255755cb2a3abf34440792748325c98c9a61f766 (patch) | |
tree | 4fc511a21810b4b5f6de55da11760af66d06e39f /src | |
parent | e3d75a13af695720397f0b28e745e2e48edbeede (diff) | |
download | duser-255755cb2a3abf34440792748325c98c9a61f766.tar.gz |
Partial autotools conversion to help out Joe
Diffstat (limited to 'src')
-rw-r--r-- | src/cfg.c | 87 | ||||
-rw-r--r-- | src/compat.c | 59 | ||||
-rw-r--r-- | src/duser.c | 307 | ||||
-rw-r--r-- | src/log.c | 77 | ||||
-rw-r--r-- | src/user.c | 577 | ||||
-rw-r--r-- | src/util.c | 95 |
6 files changed, 1202 insertions, 0 deletions
diff --git a/src/cfg.c b/src/cfg.c new file mode 100644 index 0000000..b30bd3f --- /dev/null +++ b/src/cfg.c @@ -0,0 +1,87 @@ +/** +* duser - Manage MajorDomo lists +* Copyright (C) 2011 Joseph Hunkeler <jhunkeler@gmail.com, jhunk@stsci.edu> +* +* This file is part of duser. +* +* duser 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 3 of the License, or +* (at your option) any later version. +* +* duser 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 duser. If not, see <http://www.gnu.org/licenses/>. +**/ + +#define _GNU_SOURCE + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <ctype.h> +#include <errno.h> +#include "duser.h" +#include "cfg.h" + +FILE *cfgfp; + +int cfg_open(const char* filename) +{ + if((cfgfp = fopen(filename, "r")) == NULL) + return errno; + else + return 0; +} + +void cfg_close() +{ + fclose(cfgfp); +} + +//Value of key +int cfg_get_key(char* val, const char* key) +{ + + if(cfgfp == NULL) + return -1; + + if((fseek(cfgfp, 0L, SEEK_SET)) != 0) + return errno; + + char buffer[CFG_MAX]; + char *buf = buffer; //= (char*)malloc(sizeof(char) * CFG_MAX+1); + if(buffer == NULL) + { + fprintf(stderr, "%s: %s\n", SELF, strerror(errno)); + exit(1); + } + + int i = 0; + while(!feof(cfgfp)) + { + memset(buffer, 0L, CFG_MAX); + fgets(buffer, CFG_MAX, cfgfp); + if((strcasestr(buffer, key))) + { + while(buf) + { + if(isspace(*buf)) + { + strncpy(val, &buf[i+1], CFG_MAX); + val[strlen(val)-1] = 0; + break; + } + buf++; + } + i++; + break; + } + } + + return 0; +} diff --git a/src/compat.c b/src/compat.c new file mode 100644 index 0000000..d729229 --- /dev/null +++ b/src/compat.c @@ -0,0 +1,59 @@ +/** +* duser - Manage MajorDomo lists +* Copyright (C) 2011 Joseph Hunkeler <jhunkeler@gmail.com, jhunk@stsci.edu> +* +* This file is part of duser. +* +* duser 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 3 of the License, or +* (at your option) any later version. +* +* duser 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 duser. If not, see <http://www.gnu.org/licenses/>. +**/ + +#include <stdio.h> +#include <string.h> +#include <ctype.h> +#include "compat.h" + +#ifndef HAVE_STRCHRNUL +#undef strchrnul +char *strchrnul(const char* s, int c) +{ + while(*s && *s != c) + { + s++; + } + return (char*)s; +} +#endif + +#ifndef HAVE_STRCASESTR +#undef strcasestr +char *strcasestr(const char *s, const char *find) +{ + char c, sc; + size_t len; + + if ((c = *find++) != 0) { + c = tolower((unsigned char)c); + len = strlen(find); + do { + do { + if ((sc = *s++) == 0) + return (NULL); + } while ((char)tolower((unsigned char)sc) != c); + } while (strncasecmp(s, find, len) != 0); + s--; + } + return ((char *)s); +} +#endif + diff --git a/src/duser.c b/src/duser.c new file mode 100644 index 0000000..8f5603f --- /dev/null +++ b/src/duser.c @@ -0,0 +1,307 @@ +/** +* duser - Manage MajorDomo lists +* Copyright (C) 2011 Joseph Hunkeler <jhunkeler@gmail.com, jhunk@stsci.edu> +* +* This file is part of duser. +* +* duser 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 3 of the License, or +* (at your option) any later version. +* +* duser 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 duser. If not, see <http://www.gnu.org/licenses/>. +**/ + +#define _GNU_SOURCE + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <dirent.h> +#include <limits.h> +#include <ctype.h> +#include <errno.h> +#include <sys/stat.h> +#include <unistd.h> +#include <time.h> +#include <fcntl.h> +#include <libgen.h> + +#include "version.h" +#include "cfg.h" +#include "duser.h" + +char list_path[PATH_MAX]; +char logfile[PATH_MAX]; +static char progname[FILENAME_MAX]; + +int CMD_FLAG_NOOPT = 0; +int CMD_FLAG_DEL = 0; +int CMD_FLAG_DEL_ALL = 0; +int CMD_FLAG_DEL_LIST = 0; +int CMD_FLAG_MOD = 0; +int CMD_FLAG_ADD = 0; +int CMD_FLAG_LIST = 0; +int CMD_FLAG_HELP = 0; +int CMD_FLAG_LOOK = 0; +int CMD_FLAG_NEW = 0; +int CMD_FLAG_NULL = 0; + +void version() +{ + printf("%s v%s.%s%s (%s %s)\n", + progname, VER_MAJOR, VER_MINOR, VER_OTH, __DATE__, __TIME__); +} + +void usage() +{ + version(); + printf("Usage: %s command [address [list]]\n", progname); + printf("Commands:\n"); + printf(" help This usage statement\n"); + printf(" add Add a user to a list\n"); + printf(" del Delete a user from a list\n"); + printf(" delA Delete a user from all lists\n"); + printf(" delL Completely delete a mailing list\n"); + printf(" mod Modify a user in a list\n"); + printf(" new Creates a new list\n"); + printf(" list Find and list a user in all lists\n"); + printf(" look Find user in specific list\n"); + printf("\n"); + exit(0); +} + +int main(int argc, char* argv[]) +{ + strncpy(progname, argv[0], strlen(argv[0])); + if((cfg_open(CFG_PATH)) == 0) + { + cfg_get_key(list_path, "path"); + cfg_get_key(logfile, "logfile"); + cfg_close(); + } + else + { + fprintf(stderr, "%s: %s: %s\n", SELF, CFG_PATH, strerror(errno)); + return -1; + } + + char filename[PATH_MAX]; + record_t *rec; + const char* needle = argv[2]; + const char* single_list = argv[3]; + + if(single_list) + snprintf(filename, PATH_MAX, "%s%s", list_path, single_list); + + if(argc < 3) + { + usage(); + return 0; + } + + stats_init(&processed); + user_cmd(argc, argv); + + if((strval(needle)) != 0) + { + fprintf(stderr, "%s: Invalid string\n", SELF); + exit(1); + } + + if(CMD_FLAG_DEL) + { + if(needle == NULL) + { + printf("You must specify an email address\n"); + return -1; + } + else if(single_list == NULL) + { + printf("You must specify a list to remove '%s' from\n", needle); + return -1; + } + + + if((rec = find_in_file(filename, needle)) == NULL) + { + printf("'%s' not found in '%s'\n", needle, basename(filename)); + return -1; + } + + printf("Please review the information below:\n\n"); + printf("Email: %s\n", rec->name); + printf("At line: %d\n", rec->index); + printf("In File: %s\n", basename(rec->file)); + printf("\nDo you wish to wish to delete this record? [y/N] "); + + char choice = getchar(); + if((user_choice(choice)) == 0) + { + if((user_del(rec)) > 0) + { + printf("Record deleted\n"); + COM(SELF, "Commmand: DELETE\n"); + COM(SELF, "'%s' deleted from '%s' at line %d\n", rec->name, basename(rec->file), rec->index); + } + } + else + { + printf("Aborting...\n"); + } + return 0; + } + + if(CMD_FLAG_DEL_ALL) + { + user_del_all(needle); + return 0; + } + + if(CMD_FLAG_DEL_LIST) + { + user_del_list(needle); + return 0; + } + + if(CMD_FLAG_ADD) + { + if(needle == NULL) + { + printf("You must specify an email address\n"); + return -1; + } + else if(single_list == NULL) + { + printf("You must specify a list in which to add '%s' to\n", needle); + return -1; + } + + printf("Please review the information below:\n\n");; + printf("Email: %s\n", needle); + printf("In File: %s\n", basename(filename)); + printf("\nDo you wish to wish to add this record? [y/N] "); + + char choice = getchar(); + if((user_choice(choice)) == 0) + { + if((user_add(filename, needle)) > 0) + { + printf("Record added\n"); + COM(SELF, "Commmand: ADD\n"); + COM(SELF, "'%s' added to '%s'\n", needle, basename(filename)); + } + } + else + { + printf("Aborting...\n"); + } + return 0; + } + if(CMD_FLAG_MOD) + { + if(needle == NULL) + { + printf("You must specify an email address\n"); + return -1; + } + else if(single_list == NULL) + { + printf("You must specify a list in which to modify '%s' in\n", needle); + return -1; + } + + fprintf(stderr, "Not implemented, sorry.\n"); + return 0; + } + if(CMD_FLAG_LIST) + { + if(needle == NULL) + { + printf("You must specify an email address\n"); + return -1; + } + user_list(needle); + return 0; + } + if(CMD_FLAG_LOOK) + { + if(needle == NULL) + { + printf("You must specify an email address\n"); + return -1; + } + else if(single_list == NULL) + { + printf("You must specify a list in which to find '%s' in\n", needle); + return -1; + } + + rec = find_in_file(filename, needle); + if(rec) + { + if(rec->match) + { + printf("Found '%s' at line %d of list '%s'\n", rec->name, rec->index, basename(rec->file)); + } + else + { + printf("Corrupt record? This is not supposed to happen.\n"); + return -1; + } + } + else + { + printf("Not found in '%s'\n", single_list); + } + return 0; + } + if(CMD_FLAG_NEW) + { + memset(filename, 0L, PATH_MAX); + snprintf(filename, PATH_MAX, "%s%s", list_path, needle); + + if(needle == NULL) + { + printf("You must specify a list to create\n"); + return -1; + } + + printf("Please review the information below:\n\n");; + printf("File: %s\n", filename); + printf("\nDo you wish to wish to create this list? [y/N] "); + + char choice = getchar(); + if((user_choice(choice)) == 0) + { + if((user_new_list(filename)) == 0) + { + printf("List added\n"); + COM(SELF, "Commmand: NEW\n"); + COM(SELF, "Created new list '%s'\n", basename(filename)); + } + } + else + { + printf("Aborting...\n"); + } + return 0; + } + if(CMD_FLAG_HELP) + { + usage(); + } + if(CMD_FLAG_NULL) + { + fprintf(stderr, "Command not specified\n"); + exit(1); + } + + return 0; +} diff --git a/src/log.c b/src/log.c new file mode 100644 index 0000000..c0c1051 --- /dev/null +++ b/src/log.c @@ -0,0 +1,77 @@ +/** +* duser - Manage MajorDomo lists +* Copyright (C) 2011 Joseph Hunkeler <jhunkeler@gmail.com, jhunk@stsci.edu> +* +* This file is part of duser. +* +* duser 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 3 of the License, or +* (at your option) any later version. +* +* duser 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 duser. If not, see <http://www.gnu.org/licenses/>. +**/ + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <stdarg.h> +#include <sys/types.h> +#include <unistd.h> +#include <limits.h> +#include <errno.h> +#include <time.h> +#include "duser.h" + +extern char logfile[PATH_MAX]; +//extern char* logfile; + +int logcleanup() +{ + int status = 0; + if((status = access(logfile, W_OK)) == 0) + { + status = unlink(logfile); + } + + return status; +} + +int COM(const char* func, char *format, ...) +{ + struct tm *logtm; + time_t logtime = time(NULL); + char timestr[64]; + char *str = (char*)malloc(sizeof(char) * 256); + char tmpstr[255]; + int n; + + FILE *logfp = NULL; + logfp = fopen(logfile, "a+"); + if(logfp == NULL) + { + fprintf(stderr, "Unable to open %s\n", logfile); + exit(1); + } + + va_list args; + va_start (args, format); + n = vsprintf (str, format, args); + va_end (args); + + logtm = localtime(&logtime); + + snprintf(timestr, sizeof(timestr), "%02d-%02d-%02d %02d:%02d:%02d", logtm->tm_year+1900, logtm->tm_mon+1, logtm->tm_mday, logtm->tm_hour, logtm->tm_min, logtm->tm_sec); + snprintf(tmpstr, sizeof(tmpstr), "%s %d _%s_: %s", timestr, getuid(), func, str); + fprintf(logfp, "%s", tmpstr); + fclose(logfp); + + free(str); + return 0; +} diff --git a/src/user.c b/src/user.c new file mode 100644 index 0000000..72a7c13 --- /dev/null +++ b/src/user.c @@ -0,0 +1,577 @@ +/** +* duser - Manage MajorDomo lists +* Copyright (C) 2011 Joseph Hunkeler <jhunkeler@gmail.com, jhunk@stsci.edu> +* +* This file is part of duser. +* +* duser 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 3 of the License, or +* (at your option) any later version. +* +* duser 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 duser. If not, see <http://www.gnu.org/licenses/>. +**/ + +#define _GNU_SOURCE + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <dirent.h> +#include <limits.h> +#include <ctype.h> +#include <errno.h> +#include <sys/stat.h> +#include <unistd.h> +#include <time.h> +#include <fcntl.h> +#include <libgen.h> +#ifdef _NLINUX_ +# define HAVE_STRCHRNUL +# define HAVE_STRCASESTR +#endif +#include "duser.h" + +extern char list_path[PATH_MAX]; +extern char logfile[PATH_MAX]; +static char progname[FILENAME_MAX]; +extern int CMD_FLAG_NOOPT; +extern int CMD_FLAG_DEL; +extern int CMD_FLAG_DEL_ALL; +extern int CMD_FLAG_DEL_LIST; +extern int CMD_FLAG_MOD; +extern int CMD_FLAG_ADD; +extern int CMD_FLAG_LIST; +extern int CMD_FLAG_HELP; +extern int CMD_FLAG_LOOK; +extern int CMD_FLAG_NEW; +extern int CMD_FLAG_NULL; + + +int user_del_list(const char* filename) +{ + char tmp[PATH_MAX]; + snprintf(tmp, PATH_MAX, "%s%s", list_path, filename); + + if((access(tmp, F_OK|W_OK)) != 0) + { + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, tmp, strerror(errno)); + return -1; + } + + printf("\n!!!WARNING!!!\n"); + printf("You are about to delete the mailing list '%s'\n\n", basename(tmp)); + printf("Are SURE you want to continue? [y/N]"); + int choice = getchar(); + if((user_choice(choice)) != 0) + { + printf("Aborting...\n"); + exit(1); + } + else + { + COM(SELF, "Command: DELETE LIST\n"); + if((unlink(tmp)) != 0) + { + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, tmp, strerror(errno)); + return errno; + } + else + { + COM(SELF, "List '%s' deleted\n", basename(tmp)); + printf("Deleted list: '%s'\n", basename(tmp)); + } + } + + return 0; +} + +int user_del_all(const char* needle) +{ + processed.files = get_file_count(list_path); + char** list; + char tmp[PATH_MAX]; + record_t *rp; + list = get_file_list(list_path, processed.files); + + int i = 0; + for(i = 0 ; list[i] != NULL ; i++) + { + rp = NULL; + snprintf(tmp, PATH_MAX, "%s%s", list_path, list[i]); + if((rp = find_in_file(tmp, needle)) != NULL) + { + printf("%20s\t%5d%23s\n", basename(rp->file), rp->index, rp->name); + processed.matches++; + } + } + + printf("\n%d matches\t%d files\t%d lines parsed\n", + processed.matches, processed.files, processed.lines); + + if(processed.matches < 1) + { + free_file_list(list); + return -1; + } + + printf("\n!!!WARNING!!!\n"); + printf("You are about to delete '%s' from every list\n", needle); + printf("that it appears in!\n\n"); + printf("Are SURE you want to continue? [y/N]"); + + int choice = getchar(); + if((user_choice(choice)) != 0) + { + printf("Aborting...\n"); + exit(1); + } + else + { + COM(SELF, "Commmand: DELETE\n"); + for(i = 0 ; list[i] != NULL ; i++) + { + rp = NULL; + snprintf(tmp, PATH_MAX, "%s%s", list_path, list[i]); + if((rp = find_in_file(tmp, needle)) != NULL) + { + if((user_del(rp)) > 0) + { + COM(SELF, "'%s' deleted from '%s' at line %d\n", rp->name, basename(rp->file), rp->index); + printf("'%s' deleted from '%s'\n", rp->name, basename(rp->file)); + } + } + } + } + free_file_list(list); + return 0; + +} + +int user_del(record_t* rec) +{ + FILE *tfp; + FILE *fp; + int i = 0; + int fd = 0; + int bytes = 0; + int bytes_total = 0; + 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, "r+")) == NULL) + { + if(fd != -1) + { + close(fd); + unlink(tmpfile); + } + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, tmpfile, strerror(errno)); + exit(1); + } + + if((fp = fopen(rec->file, "r")) == NULL) + { + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, 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((strncmp(buf, rec->name, strlen(rec->name))) != 0 && (i != rec->index)) + { + buf[strlen(buf)] = '\n'; + bytes = write(fd, buf, strlen(buf)); + } + i++; + } + //Rewind the temp file + lseek(fd, 0L, SEEK_SET); + //Close the original file + fclose(fp); + //Truncate original and copy data from tmp to original + if((fp = fopen(rec->file, "w+")) == NULL) + { + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, rec->file, strerror(errno)); + exit(1); + } + + i = 0; + while(!feof(tfp)) + { + memset(buf, 0, sizeof(buf)); + fgets(buf, REGEX_MAX, tfp); + buf[strlen(buf) - 1] = '\0'; + if((strncmp(buf, rec->name, strlen(rec->name))) != 0 && (i != rec->index)) + { + buf[strlen(buf)] = '\n'; + if(buf[0] == '\n') + buf[0] = '\0'; + + bytes = fwrite(buf, strlen(buf), 1, fp); + bytes_total += bytes; + } + } + + fclose(fp); + close(fd); + unlink(tmpfile); + + if(bytes_total) + return bytes_total; + + 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) + { + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, rec->file, strerror(errno)); + return -1; + } + + while(!feof(fp)) + { + fgets(buf, REGEX_MAX, fp); + buf[strlen(buf) - 1] = '\0'; + if((strncmp(buf, rec->name, strlen(rec->name))) == 0) + { + match = 1; + 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; + + FILE *fp; + char *fname = strdup(filename); + + if((fp = fopen(fname, "r")) == NULL) + { + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, basename(fname), strerror(errno)); + exit(1); + } + strncpy(rptr->file, fname, PATH_MAX); + while(!feof(fp)) + { + char cmp[REGEX_MAX]; + memset(cmp, 0L, REGEX_MAX); + fgets(cmp, REGEX_MAX, fp); + cmp[strlen(cmp)-1] = '\0'; + if((strfind(cmp, needle)) == 0) + { + snprintf(rptr->name, REGEX_MAX, "%s", cmp); + rptr->match = 1; + rptr->index = index; + } + index++; + processed.lines++; + } + + free(fname); + fclose(fp); + if(rptr->match) + return rptr; + else + return NULL; + +} + +int get_file_count(const char* path) +{ + DIR *dp = NULL; + struct dirent *ep = NULL; + int file_count = 0; + + if((dp = opendir(path)) == NULL) + { + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, path, strerror(errno)); + exit(1); + } + + while((ep = readdir(dp))) + { +#ifdef _NLINUX_ + char path[PATH_MAX]; + struct stat st; + snprintf(path, PATH_MAX, "%s%s", list_path, ep->d_name); + if((stat(path, &st)) == 0) + { + if(S_ISREG(st.st_mode)) + { + if(!strstr(ep->d_name, ".")) + { + file_count++; + } + } + } +#else + if(ep->d_type == DT_REG && !strstr(ep->d_name, ".")) + { + file_count++; + } +#endif + } + 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 = NULL; + struct dirent *ep = NULL; + 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))) + { +#ifdef _NLINUX_ + char path[PATH_MAX]; + struct stat st; + snprintf(path, PATH_MAX, "%s%s", list_path, ep->d_name); + if((stat(path, &st)) == 0) + { + if(S_ISREG(st.st_mode)) + { + if(!strstr(ep->d_name, ".")) + { + list[i] = (char*)malloc(sizeof(char) * strlen(ep->d_name)+1); + memset(list[i], 0L, strlen(ep->d_name)+1); + strncpy(list[i], ep->d_name, strlen(ep->d_name)+1); + i++; + } + } + } +#else + 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)+1); + strncpy(list[i], ep->d_name, strlen(ep->d_name)+1); + i++; + } +#endif + } + closedir(dp); + return list; +} + + +int user_list(const char* needle) +{ + processed.files = get_file_count(list_path); + char** list = get_file_list(list_path, processed.files); + if(list == NULL) + { + fprintf(stderr, "abort\n"); + exit(1); + } + + printf("%20s%12s%16s\n", "List", "At line", "Match"); + printf("\t\t%s\n", "====================================="); + int i = 0; + for(i = 0 ; list[i] != NULL ; i++) + { + char tmp[PATH_MAX]; + snprintf(tmp, PATH_MAX, "%s%s", list_path, list[i]); + record_t *rp; + if((rp = find_in_file(tmp, needle)) != NULL) + { + printf(FMTLIST, basename(rp->file), rp->index, rp->name); + processed.matches++; + } + } + free_file_list(list); + + printf("\n%d matches\t%d files\t%d lines parsed\n", + processed.matches, processed.files, processed.lines); + return 0; +} + +int user_add(const char* filename, const char* needle) +{ + int bytes = 0; + record_t *rp; + char *fname = strdup(filename); + char *newline = NULL; + char buf[REGEX_MAX]; + FILE *fp; + if((access(filename, W_OK|F_OK)) != 0) + { + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, filename, strerror(errno)); + return -1; + } + + if((fp = fopen(filename, "r+")) == NULL) + { + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, filename, strerror(errno)); + return -1; + } + + if((rp = find_in_file(filename, needle)) != NULL) + { + fprintf(stderr, "%s: '%s' already exists in '%s'\n", SELF, needle, basename(fname)); + return -1; + } + rewind(fp); + //Go to end of file and find out what's where + fseek(fp, 0L, SEEK_END); + fgets(buf, REGEX_MAX, fp); + + //Check for a newline at the end of the file. + //If so, add one. If not, do not add a preceding newline. + if((newline = strchrnul(buf, '\n')) != NULL) + { + snprintf(buf, REGEX_MAX, "%s\n", needle); + } + else + { + //probably very rarely used. + snprintf(buf, REGEX_MAX, "\n%s\n", needle); + } + + bytes = fwrite(buf, strlen(buf), 1, fp); + //fputs(buf, fp); + fflush(fp); + free(fname); + fclose(fp); + return bytes; +} + +int user_cmd(const int argc, char* argv[]) +{ + const char* cmd = argv[1]; + + if(argc < 2) + return CMD_FLAG_NULL; + + if(cmd) + { + if((strncmp(cmd, "del", strlen(cmd))) == 0) + { + CMD_FLAG_DEL = 1; + } + if((strncmp(cmd, "delA", strlen(cmd))) == 0) + { + CMD_FLAG_DEL_ALL = 1; + } + if((strncmp(cmd, "delL", strlen(cmd))) == 0) + { + CMD_FLAG_DEL_LIST = 1; + } + if((strncmp(cmd, "add", strlen(cmd))) == 0) + { + CMD_FLAG_ADD = 1; + } + if((strncmp(cmd, "mod", strlen(cmd))) == 0) + { + CMD_FLAG_MOD = 1; + } + if((strncmp(cmd, "new", strlen(cmd))) == 0) + { + CMD_FLAG_NEW = 1; + } + if((strncmp(cmd, "list", strlen(cmd))) == 0) + { + CMD_FLAG_LIST = 1; + } + if((strncmp(cmd, "look", strlen(cmd))) == 0) + { + CMD_FLAG_LOOK = 1; + } + if((strncmp(cmd, "help", strlen(cmd))) == 0) + { + CMD_FLAG_HELP = 1; + } + } + + return 0; +} + + +int user_new_list(const char* fname) +{ + char *filename = strdup(fname); + char *timestr; + char message[BUFSIZ]; + FILE *fp = NULL; + time_t ttm; + struct tm *tmptr; + + if((access(filename, F_OK)) == 0) + { + fprintf(stderr, "%s: %s: File already exists\n", SELF, basename(filename)); + return -1; + } + + if((fp = fopen(filename, "w+")) == NULL) + { + fprintf(stderr, "FATAL: %s: %s: %s\n", SELF, basename(filename), strerror(errno)); + return -1; + } + + time(&ttm); + tmptr = localtime(&ttm); + timestr = asctime(tmptr); + timestr[strlen(timestr)-1] = '\0'; + snprintf(message, BUFSIZ, "#\n# Created by %s on %s. UID %d\n#\n", basename(progname), timestr, getuid()); + fputs(message, fp); + fflush(fp); + fclose(fp); + + free(filename); + return 0; +} + + diff --git a/src/util.c b/src/util.c new file mode 100644 index 0000000..4d5f58c --- /dev/null +++ b/src/util.c @@ -0,0 +1,95 @@ +/** +* duser - Manage MajorDomo lists +* Copyright (C) 2011 Joseph Hunkeler <jhunkeler@gmail.com, jhunk@stsci.edu> +* +* This file is part of duser. +* +* duser 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 3 of the License, or +* (at your option) any later version. +* +* duser 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 duser. If not, see <http://www.gnu.org/licenses/>. +**/ + +#define _GNU_SOURCE +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <ctype.h> +#include "duser.h" + +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_choice(char c) +{ + if(c == 'y' || c == 'Y') + return 0; + + return 1; +} + +int check_cmd_string(char** args, const char* str2, int count) +{ + int i = 0; + while(i < count) + { + if((strncmp(args[i], str2, strlen(args[i]))) == 0) + { + return 0; + } + i++; + } + return -1; +} + +//I'm using this until regex can get its head out of its... +//toilet bowl. +int strfind(const char* str1, const char* str2) +{ + if((strcasestr(str1, str2)) != 0) + { + if((strcasecmp(str1, str2)) == 0) + return 0; + } + return -1; +} + +int strval(const char* str) +{ + const char* bad = "!#$%^&*()+={}[]|\\<>,"; + unsigned int i = 0; + unsigned int ibad = 0; + for(i = 0 ; i < strlen(str) ; i++) + { + unsigned char c = str[i]; + if((i == 0) && !(isalpha(c))) + { + return -1; + } + for(ibad = 0 ; ibad <= strlen(bad) ; ibad++) + { + if(str[i] == bad[ibad]) + { + return -1; + } + } + } + + return 0; +} + |