cline: cline.c@9393eff3d2f9 (annotated)

cline.c@9393eff3d2f9 (annotated)

cline.c

Fri, 27 May 2011 13:20:15 +0200

author: Mike Becker <universe@uap-core.de>
date: Fri, 27 May 2011 13:20:15 +0200
changeset 5: 9393eff3d2f9
parent 4: c3acfb3b4957
child 6: be923400164c
permissions: -rw-r--r--

Fixed memory leak when exiting the programm ahead of time

 #include "cline.h"
 #include "functions.h"
 settings_t* new_settings_t() {
   settings_t *settings = malloc(sizeof(settings_t*));
   if (settings != NULL) {
   #ifdef _WIN32
     settings->fileSeparator      = '\\';
   #else
     settings->fileSeparator      = '/';
   #endif /* _WIN32 */
     settings->suffixc            = 1;
     settings->recursive          = false;
     settings->includeSuffixes    = false;
     settings->matchesOnly        = false;
   }
   return settings;
 }
 void destroy_settings_t(settings_t* settings) {
   if (settings->suffixv != NULL) {
     free(settings->suffixv);
   }
   free(settings);
 }
 void printHelpText(const char* prgName) {
   // Help text
   const char* helpText =
     "\nUsage:"
     "\n      %s [-hrm][-s suffix][<directory>]"
     "\n      %s [-hrm][-S suffix][<directory>]"
     "\n\nCounts the line terminator characters (\\n) within all"
     " files in the specified\ndirectory."
     "\n\nOptions:"
     "\n  -h, --help          - this help text"
     "\n  -m                  - print information about matching files only"
     "\n  -s <suffixes>       - only count files with these suffixes (separated"
     "\n                        by commas)"
     "\n  -S <suffixes>       - count any file except those with these suffixes"
     "\n                        (separated by commas)"
     "\n  -r, -R              - includes subdirectories"
     "\n\n"
     "The default call without any options is:"
     "\n  %s ./\n"
     "That means each file in each subdirectory is counted. If you want to count"
     "\nC source code in your working directory and its subdirectories, type:"
     "\n  %s -rs .c\n";
   printf(helpText, prgName, prgName, prgName, prgName);
 }
 int main(int argc, char** argv) {
   // Settings
   settings_t *settings = new_settings_t();
   if (settings == NULL) {
     fprintf(stderr, "Memory allocation failed.\n");
     return 1;
   }
   // Program name
   char* prgName = strrchr(argv[0], settings->fileSeparator);
   if (prgName == NULL) {
     prgName = argv[0];
   } else {
     prgName++;
   }
   // Defaults
   char* _suffix = " ";
   char _directory[3];
   _directory[0] = '.';
   _directory[1] = settings->fileSeparator;
   _directory[2] = 0;
   // Get arguments
   char* directory;
   char* suffix;
   bool showHelp = false;
   char checked = 0;
   for (int t = 1 ; t < argc ; t++) {
     int argflags = checkArgument(argv[t], "hsSrRm");
     // s
     if ((argflags & 2) > 0) {
       if ((checked & 1) > 0) {
         printHelpText(prgName);
         destroy_settings_t(settings);
         return 1;
       }
       settings->includeSuffixes = true;
       t++;
       if (t >= argc) {
         printHelpText(prgName);
         destroy_settings_t(settings);
         return 1;
       }
       suffix = argv[t];
       checked |= 1;
     }
     // S
     if ((argflags & 4) > 0) {
       if ((checked & 1) > 0) {
         printHelpText(prgName);
         destroy_settings_t(settings);
         return 1;
       }
       settings->includeSuffixes = false;
       t++;
       if (t >= argc) {
         printHelpText(prgName);
         destroy_settings_t(settings);
         return 1;
       }
       suffix = argv[t];
       checked |= 1;
     }
     // h
     if ((argflags & 1) > 0 || strcmp(argv[t], "--help") == 0) {
       if ((checked & 2) > 0) {
         printHelpText(prgName);
         destroy_settings_t(settings);
         return 1;
       }
       checked |= 2;
       showHelp = true;
     }
     // r, R
     if ((argflags & 24) > 0) {
       if ((checked & 4) > 0) {
         printHelpText(prgName);
         destroy_settings_t(settings);
         return 1;
       }
       checked |= 4;
       settings->recursive = true;
     }
     if ((argflags & 32) > 0) {
       if ((checked & 32) > 0) {
         printHelpText(prgName);
         destroy_settings_t(settings);
         return 1;
       }
       checked |= 32;
       settings->matchesOnly = true;
     }
     // other
     if (argflags == 0) {
       if ((checked & 8) > 0) {
         printHelpText(prgName);
         destroy_settings_t(settings);
         return 1;
       }
       checked |= 8;
       directory = argv[t];
     }
   }
   // Show help and quit
   if (showHelp) {
     printHelpText(prgName);
     destroy_settings_t(settings);
     return 0;
   }
   // Default values
   if ((checked & 1) == 0) {
     suffix = _suffix;
   }
   if ((checked & 8) == 0) {
     directory = _directory;
   }
   // Find tokens
   char* finder;
   finder = strchr(suffix, ',');
   while (finder != NULL) {
     settings->suffixc++;
     finder = strchr(finder+1, ',');
   }
   settings->suffixv = (char**) malloc(sizeof(char**)*settings->suffixc);
   if (settings->suffixv == NULL) {
     fprintf(stderr, "Memory allocation failed.\n");
     destroy_settings_t(settings);
     return 1;
   }
   finder = strtok(suffix, ",");
   int c = 0;
   while (finder != NULL) {
     settings->suffixv[c] = finder;
     c++;
     finder = strtok(NULL, ",");
   }
   // Open directory
   DIR *dir = opendir(directory);
   if (dir == NULL) {
     perror("Operation failed");
     destroy_settings_t(settings);
     return 1;
   }
   // Scan directory
   int lines = scanDirectory(dir, 0, directory, settings);
   closedir(dir);
   // Print double line and line count
   #ifdef _WIN32
     const int columns = 79;
   #else
     const int columns = 80;
   #endif /* _WIN32 */
   for (int t = 0 ; t < columns ; t++) {
     printf("=");
   }
   #ifdef _WIN32
     printf("\n%73d lines\n", lines);
   #else
     printf("\n%74d lines\n", lines);
   #endif /* _WIN32 */
   destroy_settings_t(settings);
   return 0;
 }

Mercurial > hg > cline / annotate

cline.c@9393eff3d2f9 (annotated)

cline.c