diff options
Diffstat (limited to 'tools/atree/files.cpp')
| -rw-r--r-- | tools/atree/files.cpp | 357 |
1 files changed, 357 insertions, 0 deletions
diff --git a/tools/atree/files.cpp b/tools/atree/files.cpp new file mode 100644 index 0000000..5842378 --- /dev/null +++ b/tools/atree/files.cpp @@ -0,0 +1,357 @@ +#include "files.h" +#include <stdio.h> +#include <errno.h> +#include <sys/stat.h> +#include <unistd.h> +#include <dirent.h> +#include <fnmatch.h> + +static bool +is_comment_line(const char* p) +{ + while (*p && isspace(*p)) { + p++; + } + return *p == '#'; +} + +static string +path_append(const string& base, const string& leaf) +{ + string full = base; + if (base.length() > 0 && leaf.length() > 0) { + full += '/'; + } + full += leaf; + return full; +} + +static bool +is_whitespace_line(const char* p) +{ + while (*p) { + if (!isspace(*p)) { + return false; + } + p++; + } + return true; +} + +static bool +is_exclude_line(const char* p) { + while (*p) { + if (*p == '-') { + return true; + } + else if (isspace(*p)) { + p++; + } + else { + return false; + } + } + return false; +} + +void +split_line(const char* p, vector<string>* out) +{ + const char* q = p; + enum { WHITE, TEXT } state = WHITE; + while (*p) { + if (*p == '#') { + break; + } + + switch (state) + { + case WHITE: + if (!isspace(*p)) { + q = p; + state = TEXT; + } + break; + case TEXT: + if (isspace(*p)) { + if (q != p) { + out->push_back(string(q, p-q)); + } + state = WHITE; + } + break; + } + p++; + } + if (state == TEXT) { + out->push_back(string(q, p-q)); + } +} + +static void +add_file(vector<FileRecord>* files, const string& listFile, int listLine, + const string& sourceName, const string& outName) +{ + FileRecord rec; + rec.listFile = listFile; + rec.listLine = listLine; + rec.sourceName = sourceName; + rec.outName = outName; + files->push_back(rec); +} + +int +read_list_file(const string& filename, vector<FileRecord>* files, + vector<string>* excludes) +{ + int err = 0; + FILE* f = NULL; + long size; + char* buf = NULL; + char *p, *q; + int i, lineCount; + + f = fopen(filename.c_str(), "r"); + if (f == NULL) { + fprintf(stderr, "Could not open list file (%s): %s\n", + filename.c_str(), strerror(errno)); + err = errno; + goto cleanup; + } + + err = fseek(f, 0, SEEK_END); + if (err != 0) { + fprintf(stderr, "Could not seek to the end of file %s. (%s)\n", + filename.c_str(), strerror(errno)); + err = errno; + goto cleanup; + } + + size = ftell(f); + + err = fseek(f, 0, SEEK_SET); + if (err != 0) { + fprintf(stderr, "Could not seek to the beginning of file %s. (%s)\n", + filename.c_str(), strerror(errno)); + err = errno; + goto cleanup; + } + + buf = (char*)malloc(size+1); + if (buf == NULL) { + // (potentially large) + fprintf(stderr, "out of memory (%ld)\n", size); + err = ENOMEM; + goto cleanup; + } + + if (1 != fread(buf, size, 1, f)) { + fprintf(stderr, "error reading file %s. (%s)\n", + filename.c_str(), strerror(errno)); + err = errno; + goto cleanup; + } + + // split on lines + p = buf; + q = buf+size; + lineCount = 0; + while (p<q) { + if (*p == '\r' || *p == '\n') { + *p = '\0'; + lineCount++; + } + p++; + } + + // read lines + p = buf; + for (i=0; i<lineCount; i++) { + int len = strlen(p); + q = p + len + 1; + if (is_whitespace_line(p) || is_comment_line(p)) { + ; + } + else if (is_exclude_line(p)) { + while (*p != '-') p++; + p++; + excludes->push_back(string(p)); + } + else { + vector<string> words; + + split_line(p, &words); + +#if 0 + printf("[ "); + for (size_t k=0; k<words.size(); k++) { + printf("'%s' ", words[k].c_str()); + } + printf("]\n"); +#endif + + if (words.size() == 1) { + // pattern: DEST + add_file(files, filename, i+1, words[0], words[0]); + } + else if (words.size() == 2) { + // pattern: SRC DEST + add_file(files, filename, i+1, words[0], words[1]); + } + else { + fprintf(stderr, "%s:%d: bad format: %s\n", filename.c_str(), + i+1, p); + err = 1; + } + } + p = q; + } + +cleanup: + if (buf != NULL) { + free(buf); + } + if (f != NULL) { + fclose(f); + } + return err; +} + + +int +locate(FileRecord* rec, const vector<string>& search) +{ + int err; + + for (vector<string>::const_iterator it=search.begin(); + it!=search.end(); it++) { + string full = path_append(*it, rec->sourceName); + struct stat st; + err = stat(full.c_str(), &st); + if (err == 0) { + rec->sourceBase = *it; + rec->sourcePath = full; + rec->sourceMod = st.st_mtime; + rec->sourceIsDir = S_ISDIR(st.st_mode); + return 0; + } + } + + fprintf(stderr, "%s:%d: couldn't locate source file: %s\n", + rec->listFile.c_str(), rec->listLine, rec->sourceName.c_str()); + return 1; +} + +void +stat_out(const string& base, FileRecord* rec) +{ + rec->outPath = path_append(base, rec->outName); + + int err; + struct stat st; + err = stat(rec->outPath.c_str(), &st); + if (err == 0) { + rec->outMod = st.st_mtime; + rec->outIsDir = S_ISDIR(st.st_mode); + } else { + rec->outMod = 0; + rec->outIsDir = false; + } +} + +string +dir_part(const string& filename) +{ + int pos = filename.rfind('/'); + if (pos <= 0) { + return "."; + } + return filename.substr(0, pos); +} + +static void +add_more(const string& entry, bool isDir, + const FileRecord& rec, vector<FileRecord>*more) +{ + FileRecord r; + r.listFile = rec.listFile; + r.listLine = rec.listLine; + r.sourceName = path_append(rec.sourceName, entry); + r.sourcePath = path_append(rec.sourceBase, r.sourceName); + struct stat st; + int err = stat(r.sourcePath.c_str(), &st); + if (err == 0) { + r.sourceMod = st.st_mtime; + } + r.sourceIsDir = isDir; + r.outName = path_append(rec.outName, entry); + more->push_back(r); +} + +static bool +matches_excludes(const char* file, const vector<string>& excludes) +{ + for (vector<string>::const_iterator it=excludes.begin(); + it!=excludes.end(); it++) { + if (0 == fnmatch(it->c_str(), file, FNM_PERIOD)) { + return true; + } + } + return false; +} + +static int +list_dir(const string& path, const FileRecord& rec, + const vector<string>& excludes, + vector<FileRecord>* more) +{ + int err; + + string full = path_append(rec.sourceBase, rec.sourceName); + full = path_append(full, path); + + DIR *d = opendir(full.c_str()); + if (d == NULL) { + return errno; + } + + vector<string> dirs; + + struct dirent *ent; + while (NULL != (ent = readdir(d))) { + if (0 == strcmp(".", ent->d_name) + || 0 == strcmp("..", ent->d_name)) { + continue; + } + if (matches_excludes(ent->d_name, excludes)) { + continue; + } + string entry = path_append(path, ent->d_name); +#ifdef HAVE_DIRENT_D_TYPE + bool is_directory = (ent->d_type == DT_DIR); +#else + // If dirent.d_type is missing, then use stat instead + struct stat stat_buf; + stat(entry.c_str(), &stat_buf); + bool is_directory = S_ISDIR(stat_buf.st_mode); +#endif + add_more(entry, is_directory, rec, more); + if (is_directory) { + dirs.push_back(entry); + } + } + closedir(d); + + for (vector<string>::iterator it=dirs.begin(); it!=dirs.end(); it++) { + list_dir(*it, rec, excludes, more); + } + + return 0; +} + +int +list_dir(const FileRecord& rec, const vector<string>& excludes, + vector<FileRecord>* files) +{ + return list_dir("", rec, excludes, files); +} |