diff options
Diffstat (limited to 'test/drive_logfile.cc')
| -rw-r--r-- | test/drive_logfile.cc | 194 |
1 files changed, 194 insertions, 0 deletions
diff --git a/test/drive_logfile.cc b/test/drive_logfile.cc new file mode 100644 index 0000000..2da37dd --- /dev/null +++ b/test/drive_logfile.cc @@ -0,0 +1,194 @@ +/** + * Copyright (c) 2007-2012, Timothy Stack + * + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * * Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * * Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * * Neither the name of Timothy Stack nor the names of its contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ''AS IS'' AND ANY + * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY + * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; + * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND + * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS + * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include <algorithm> + +#include <assert.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <sys/stat.h> +#include <sys/types.h> +#include <unistd.h> + +#include "base/injector.hh" +#include "base/opt_util.hh" +#include "config.h" +#include "log_format.hh" +#include "log_format_loader.hh" +#include "logfile.hh" + +using namespace std; + +typedef enum { + MODE_NONE, + MODE_ECHO, + MODE_LINE_COUNT, + MODE_TIMES, + MODE_LEVELS, +} dl_mode_t; + +time_t +time(time_t* _unused) +{ + return 1194107018; +} + +int +main(int argc, char* argv[]) +{ + int c, retval = EXIT_SUCCESS; + dl_mode_t mode = MODE_NONE; + string expected_format; + + { + static auto builtin_formats + = injector::get<std::vector<std::shared_ptr<log_format>>>(); + auto& root_formats = log_format::get_root_formats(); + + log_format::get_root_formats().insert(root_formats.begin(), + builtin_formats.begin(), + builtin_formats.end()); + builtin_formats.clear(); + } + + { + std::vector<lnav::console::user_message> errors; + vector<ghc::filesystem::path> paths; + + getenv_opt("test_dir") | + [&paths](auto value) { paths.template emplace_back(value); }; + load_formats(paths, errors); + } + + while ((c = getopt(argc, argv, "ef:ltv")) != -1) { + switch (c) { + case 'f': + expected_format = optarg; + break; + case 'e': + mode = MODE_ECHO; + break; + case 'l': + mode = MODE_LINE_COUNT; + break; + case 't': + mode = MODE_TIMES; + break; + case 'v': + mode = MODE_LEVELS; + break; + } + } + + argc -= optind; + argv += optind; + + if (retval == EXIT_FAILURE) { + } else if (argc == 0) { + fprintf(stderr, "error: expecting log file name\n"); + } else { + logfile_open_options default_loo; + auto open_res = logfile::open(argv[0], default_loo); + + if (open_res.isErr()) { + fprintf(stderr, + "unable to open logfile: %s\n", + open_res.unwrapErr().c_str()); + return EXIT_FAILURE; + } + + auto lf = open_res.unwrap(); + struct stat st; + + stat(argv[0], &st); + assert(strcmp(argv[0], lf->get_filename().c_str()) == 0); + + lf->rebuild_index(); + assert(!lf->is_closed()); + lf->rebuild_index(); + assert(!lf->is_closed()); + lf->rebuild_index(); + assert(!lf->is_closed()); + assert(lf->get_activity().la_polls == 3); + if (expected_format.empty()) { + assert(lf->get_format() == nullptr); + } else { + // printf("%s %s\n", lf->get_format()->get_name().c_str(), + // expected_format.c_str()); + assert(lf->get_format() != nullptr); + assert(lf->get_format()->get_name().to_string() == expected_format); + } + if (!lf->is_compressed()) { + assert(lf->get_modified_time() == st.st_mtime); + } + + switch (mode) { + case MODE_NONE: + break; + case MODE_ECHO: + for (auto iter = lf->begin(); iter != lf->end(); ++iter) { + auto sbr = lf->read_line(iter).unwrap(); + + printf("%.*s\n", (int) sbr.length(), sbr.get_data()); + } + break; + case MODE_LINE_COUNT: + printf("%zd\n", lf->size()); + break; + case MODE_TIMES: + for (auto& iter : *lf) { + if (iter.is_ignored()) { + continue; + } + + char buffer[1024]; + time_t lt; + + lt = iter.get_time(); + strftime(buffer, + sizeof(buffer), + "%b %d %H:%M:%S %Y", + gmtime(<)); + printf("%s -- %03d\n", buffer, iter.get_millis()); + } + break; + case MODE_LEVELS: + for (auto& iter : *lf) { + log_level_t level = iter.get_level_and_flags(); + printf("%s 0x%x\n", + level_names[level & ~LEVEL__FLAGS], + level & LEVEL__FLAGS); + } + break; + } + } + + return retval; +} |