mirror of
https://github.com/tstack/lnav.git
synced 2024-09-11 21:07:39 +03:00
158 lines
5.1 KiB
C++
158 lines
5.1 KiB
C++
/**
|
|
* 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 <assert.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "base/auto_fd.hh"
|
|
#include "config.h"
|
|
#include "line_buffer.hh"
|
|
|
|
using namespace std;
|
|
|
|
static const char* TEST_DATA
|
|
= "Hello, World!\n"
|
|
"Goodbye, World!\n";
|
|
|
|
static void
|
|
single_line(const char* data)
|
|
{
|
|
line_buffer lb;
|
|
auto_fd pi[2];
|
|
off_t off = 0;
|
|
|
|
assert(auto_fd::pipe(pi) == 0);
|
|
log_perror(write(pi[1], data, strlen(data)));
|
|
pi[1].reset();
|
|
|
|
lb.set_fd(pi[0]);
|
|
auto load_result = lb.load_next_line({off});
|
|
auto li = load_result.unwrap();
|
|
assert(data[strlen(data) - 1] == '\n' || li.li_partial);
|
|
assert(li.li_file_range.next_offset() == (off_t) strlen(data));
|
|
assert(li.li_file_range.fr_size == strlen(data));
|
|
|
|
auto next_load_result = lb.load_next_line(li.li_file_range);
|
|
assert(next_load_result.isOk());
|
|
assert(next_load_result.unwrap().li_file_range.empty());
|
|
assert(lb.get_file_size() != -1);
|
|
}
|
|
|
|
int
|
|
main(int argc, char* argv[])
|
|
{
|
|
int retval = EXIT_SUCCESS;
|
|
|
|
single_line("Dexter Morgan");
|
|
single_line("Rudy Morgan\n");
|
|
|
|
{
|
|
char fn_template[] = "test_line_buffer.XXXXXX";
|
|
|
|
auto fd = auto_fd(mkstemp(fn_template));
|
|
remove(fn_template);
|
|
line_buffer lb;
|
|
|
|
write(fd, TEST_DATA, strlen(TEST_DATA));
|
|
lseek(fd, SEEK_SET, 0);
|
|
|
|
lb.set_fd(fd);
|
|
|
|
shared_buffer_ref sbr;
|
|
|
|
auto result = lb.read_range({0, 1024});
|
|
|
|
assert(result.isErr());
|
|
}
|
|
|
|
{
|
|
static string first = "Hello";
|
|
static string second = ", World!";
|
|
static string third = "Goodbye, World!";
|
|
static string last = "\n";
|
|
|
|
line_buffer lb;
|
|
auto_fd pi[2];
|
|
off_t off = 0;
|
|
|
|
assert(auto_fd::pipe(pi) == 0);
|
|
log_perror(write(pi[1], first.c_str(), first.size()));
|
|
fcntl(pi[0], F_SETFL, O_NONBLOCK);
|
|
|
|
lb.set_fd(pi[0]);
|
|
auto load_result = lb.load_next_line({off});
|
|
auto li = load_result.unwrap();
|
|
assert(li.li_partial);
|
|
assert(li.li_file_range.fr_size == 5);
|
|
log_perror(write(pi[1], second.c_str(), second.size()));
|
|
auto load_result2 = lb.load_next_line({off});
|
|
li = load_result2.unwrap();
|
|
assert(li.li_partial);
|
|
assert(li.li_file_range.fr_size == 13);
|
|
log_perror(write(pi[1], last.c_str(), last.size()));
|
|
auto load_result3 = lb.load_next_line({off});
|
|
li = load_result3.unwrap();
|
|
assert(!li.li_partial);
|
|
assert(li.li_file_range.fr_size == 14);
|
|
auto load_result4 = lb.load_next_line(li.li_file_range);
|
|
li = load_result4.unwrap();
|
|
auto last_range = li.li_file_range;
|
|
assert(li.li_partial);
|
|
assert(li.li_file_range.empty());
|
|
log_perror(write(pi[1], third.c_str(), third.size()));
|
|
auto load_result5 = lb.load_next_line(last_range);
|
|
li = load_result5.unwrap();
|
|
assert(li.li_partial);
|
|
assert(li.li_file_range.fr_size == 15);
|
|
log_perror(write(pi[1], last.c_str(), last.size()));
|
|
auto load_result6 = lb.load_next_line(last_range);
|
|
li = load_result6.unwrap();
|
|
assert(!li.li_partial);
|
|
assert(li.li_file_range.fr_size == 16);
|
|
|
|
auto load_result7 = lb.load_next_line(li.li_file_range);
|
|
li = load_result7.unwrap();
|
|
assert(li.li_partial);
|
|
assert(li.li_file_range.empty());
|
|
assert(!lb.is_pipe_closed());
|
|
|
|
pi[1].reset();
|
|
|
|
auto load_result8 = lb.load_next_line(li.li_file_range);
|
|
li = load_result8.unwrap();
|
|
assert(!li.li_partial);
|
|
assert(li.li_file_range.empty());
|
|
assert(lb.is_pipe_closed());
|
|
}
|
|
|
|
return retval;
|
|
}
|